Rework regtest runner

The C++ script is rewritten in Makefile and shell script for concision.

Author: McSinyx

.gitignore

@@ -1,6 +1,20 @@
 *.bin
 *.o
 *.a
+*.so
+*.exe
+*.out
 src/e9patch/e9loader_*.c
 e9patch
 e9tool
+test/regtest/bugs
+test/regtest/dl
+test/regtest/fini
+test/regtest/init
+test/regtest/inst
+test/regtest/patch
+test/regtest/test
+test/regtest/test.libc
+test/regtest/test.pie
+test/regtest/test_c
+test/regtest/test_c.debug

Makefile

@@ -1,4 +1,4 @@
-.PHONY: all clean install dev release debug sanitize
+.PHONY: all clean install check dev release debug sanitize check-debug
 
 #########################################################################
 # BUILD COMMON
@@ -66,6 +66,7 @@ clean:
 	$(MAKE) -C contrib/zydis clean
 	rm -rf $(E9PATCH_OBJS) $(E9TOOL_OBJS) e9patch e9tool \
         src/e9patch/e9loader_*.c e9loader_*.o e9loader_*.bin
+	$(MAKE) -C test/regtest clean-check
 
 src/e9patch/e9loader_elf.c: src/e9patch/e9loader_elf.cpp
 	$(CXX) -std=c++11 -Wall -fno-stack-protector -Wno-unused-function -fPIC \
@@ -82,6 +83,9 @@ src/e9patch/e9loader_pe.c: src/e9patch/e9loader_pe.cpp
 src/e9patch/e9elf.o: src/e9patch/e9loader_elf.c
 src/e9patch/e9pe.o: src/e9patch/e9loader_pe.c
 
+check: all
+	$(MAKE) -C test/regtest
+
 install: all
 	install -d "$(DESTDIR)$(PREFIX)/bin"
 	install -m 755 e9patch "$(DESTDIR)$(PREFIX)/bin/e9patch"
@@ -164,3 +168,6 @@ debug: dev
 
 sanitize: CXXFLAGS += -O0 -g -fsanitize=address
 sanitize: dev
+
+check-debug: debug
+	$(MAKE) -C test/regtest

test/regtest/Makefile

@@ -1,36 +1,72 @@
+.PHONY: check clean-check
+
 FCF_NONE := $(shell \
     if gcc -fcf-protection=none --version 2>&1 | grep -q 'unrecognized'; \
         then true; \
         else echo -fcf-protection=none; fi)
 
-all:
+BASE ::= test test.pie bugs test.libc libtest.so test_c test_c.debug example.so
+TRAMPOLINE ::= inst patch dl init fini
+IN ::= $(wildcard *.in)
+EXE ::= $(IN:.in=.exe)
+
+check: regtest $(EXE)
+	./$^
+
+%.exe: in=$(shell head -1 $<)
+%.exe: %.in $(BASE) $(TRAMPOLINE)
+	../../e9tool $(E9TOOL_OPTIONS) -M 'addr >= &"entry"' $(in)\
+		-E data..data_END -E data2...text -E .text..begin -o $@
+
+test:
 	gcc -x assembler-with-cpp -o test test.s -no-pie -nostdlib \
         -Wl,--section-start=.text=0xa000000 -Wl,--section-start=.bss=0xc000000 \
         -Wl,-z -Wl,max-page-size=4096 -DPIE=0
+
+test.pie:
 	gcc -x assembler-with-cpp -o test.pie test.s -pie -nostdlib \
         -Wl,--section-start=.text=0xa000000 -Wl,--section-start=.bss=0xc000000 \
         -Wl,-z -Wl,max-page-size=4096 -DPIE=1 \
 		-Wl,--export-dynamic
+
+bugs:
 	gcc -x assembler-with-cpp -o bugs bugs.s -no-pie -nostdlib \
         -Wl,--section-start=.text=0xa000000 -Wl,--section-start=.bss=0xc000000 \
         -Wl,-z -Wl,max-page-size=4096 -DPIE=0
+
+test.libc:
 	gcc -x assembler-with-cpp -o test.libc test_libc.s -pie -Wl,--export-dynamic
+
+libtest.so:
 	gcc -x assembler-with-cpp -shared -o libtest.so libtest.s 
+
+test_c:
 	gcc -O2 -fPIC $(FCF_NONE) -pie -o test_c test_c.c \
 		-Wl,--export-dynamic -U_FORTIFY_SOURCE
 	strip test_c
+
+test_c.debug:
 	gcc -O0 -g -fPIC -pie -o test_c.debug test_c.c
+
+inst:
 	../../e9compile.sh inst.c -I ../../examples/ 
+
+patch:
 	../../e9compile.sh patch.cpp -std=c++11 -I ../../examples/ 
+
+dl:
 	NO_SIMD_CHECK=1 ../../e9compile.sh dl.c -I ../../examples/
+
+init:
 	../../e9compile.sh init.c -I ../../examples/ 
+
+fini:
 	../../e9compile.sh fini.c -I ../../examples/
+
+example.so:
 	g++ -std=c++11 -fPIC -shared -o example.so -O2 \
         ../../examples/plugins/example.cpp -I ../../src/e9tool/
-	g++ -std=c++11 -pie -fPIC -o regtest regtest.cpp -O2
-	echo "XXX" > FILE.txt
-	chmod 0640 FILE.txt
 
-clean:
-	rm -f *.log *.out *.exe test test.pie test.libc libtest.so inst inst.o \
-        patch patch.o init init.o regtest
+clean-check:
+	rm -f $(BASE) $(TRAMPOLINE) $(EXE)
+	rm -f *.out

test/regtest/README.md

@@ -3,6 +3,4 @@ README
 
 To run the tests:
 
-    $ make
-    $ ./regtest
-
+    make E9TOOL_OPTIONS=

test/regtest/init_dso.cmd

@@ -1 +1,2 @@
-LD_PRELOAD=$PWD/init_dso.exe ./test.pie
+#!/bin/sh
+LD_PRELOAD=./init_dso.exe ./test.pie

test/regtest/init_dso_2.cmd

@@ -1 +1,2 @@
-LD_PRELOAD=$PWD/init_dso.exe ./test.pie a b c 1 2 3
+#!/bin/sh
+LD_PRELOAD=./init_dso.exe ./test.pie a b c 1 2 3

test/regtest/regtest

@@ -0,0 +1,25 @@
+#!/bin/sh
+fails=()
+for exe in $*
+do
+  tst=${exe%.exe}
+  cmd=$tst.cmd
+  out=$tst.out
+  exp=$tst.exp
+
+  if test -f $cmd
+  then ./exec.sh ./$cmd 1>$out 2>&1
+  else ./exec.sh ./$exe 1>$out 2>&1
+  fi
+
+  diff -u $out $exp
+  if test $? -ne 0
+  then fails+=($tst)
+  fi
+done
+
+if test "$fails"
+then
+  echo "Failing ${#fails[@]}/$# tests: ${fails[@]}"
+  exit 1
+fi