Add tests

Signed-off-by: Darshan Sen <raisinten@gmail.com>

.circleci/config.yml

@@ -110,7 +110,7 @@ commands:
       - attach_workspace:
           at: .
 
-      - run: make check
+      - run: make test
 
 ## JOBS ##
 

Makefile

@@ -3,7 +3,7 @@ include vendor/vendorpull/targets.mk
 include build/system.mk
 include build/deps.mk
 
-all: vendor compile patch check
+all: vendor compile patch test
 
 .PHONY: lief
 lief: dist/lief
@@ -15,6 +15,6 @@ dist/lief:
 	cd vendor/lief && python3 ./setup.py $(BUILD_OPTS) build_ext -b ../../$@ -j $(JOBS)
 
 
-.PHONY: check
-check:
-	$(MAKE) -C examples/
+.PHONY: test
+test:
+	./test.sh

README.markdown

@@ -29,7 +29,7 @@ $ ./postject.py --macho-segment-name __ELECTRON /Users/dsanders/electron/src/out
 ### Testing
 
 ```sh
-$ make check
+$ make test
 ```
 
 ## Design

test.sh

@@ -0,0 +1,27 @@
+#!/bin/sh
+
+set -o errexit
+set -o nounset
+
+for test_directory in tests/*
+do
+  if [ -d "$test_directory" ]
+  then
+    echo "---- Running $test_directory" 1>&2
+    TEMPORARY_DIRECTORY="$(mktemp -d)"
+    pwd="$PWD"
+    cd "$test_directory"
+    TEMPORARY_DIRECTORY="$TEMPORARY_DIRECTORY" ./test.sh \
+      && EXIT_CODE="$?" || EXIT_CODE="$?"
+    cd "$pwd"
+    rm -rf "$TEMPORARY_DIRECTORY"
+
+    if [ "$EXIT_CODE" = "0" ]
+    then
+      echo "\033[33;32m✓ PASS\x1b[0m $test_directory" 1>&2
+    else
+      echo "\033[33;31m× FAIL\x1b[0m $test_directory" 1>&2
+      exit "$EXIT_CODE"
+    fi
+  fi
+done

tests/inject_and_retrieve_data_1000b/test.c

@@ -0,0 +1,21 @@
+#include <stdio.h>
+#include <string.h>
+
+#include "../../postject-api.h"
+
+int main()
+{
+  size_t size;
+  const void *ptr = postject_find_resource("foobar", &size, NULL);
+
+  if (ptr && size > 0) {
+    char *str = (char *)malloc(size + 1);
+    memset(str, 0, size + 1);
+    strncpy(str, ptr, size);
+    printf("%s\n", str);
+  } else {
+    printf("Hello world\n");
+  }
+
+  return 0;
+}

tests/inject_and_retrieve_data_1000b/test.sh

@@ -0,0 +1,23 @@
+#!/bin/sh
+
+set -o errexit
+set -o nounset
+
+bin="$TEMPORARY_DIRECTORY/a.out"
+clang test.c -o "$bin"
+
+data="$(head -c 1000 /dev/urandom)"
+file="$TEMPORARY_DIRECTORY/data.txt"
+echo "$data" > "$file"
+
+../../postject.py --overwrite "$bin" "foobar" "$file"
+
+want="$data"
+have="$("$bin")"
+
+if [ "$have" != "$want" ]
+then
+	echo "have: \""$have"\"" 1>&2
+	echo "want: \""$want"\"" 1>&2
+	exit 1
+fi

tests/inject_and_retrieve_data_basic/test.c

@@ -0,0 +1,21 @@
+#include <stdio.h>
+#include <string.h>
+
+#include "../../postject-api.h"
+
+int main()
+{
+  size_t size;
+  const void *ptr = postject_find_resource("foobar", &size, NULL);
+
+  if (ptr && size > 0) {
+    char *str = (char *)malloc(size + 1);
+    memset(str, 0, size + 1);
+    strncpy(str, ptr, size);
+    printf("%s\n", str);
+  } else {
+    printf("Hello world\n");
+  }
+
+  return 0;
+}

tests/inject_and_retrieve_data_basic/test.sh

@@ -0,0 +1,23 @@
+#!/bin/sh
+
+set -o errexit
+set -o nounset
+
+bin="$TEMPORARY_DIRECTORY/a.out"
+clang test.c -o "$bin"
+
+data="hello world @ $(date)"
+file="$TEMPORARY_DIRECTORY/data.txt"
+echo "$data" > "$file"
+
+../../postject.py --overwrite "$bin" "foobar" "$file"
+
+want="$data"
+have="$("$bin")"
+
+if [ "$have" != "$want" ]
+then
+	echo "have: \""$have"\"" 1>&2
+	echo "want: \""$want"\"" 1>&2
+	exit 1
+fi