Add tests for injecting large resources

Fixes: #12
Signed-off-by: Darshan Sen <[email protected]>

Author: RaisinTen

.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

examples/Makefile

@@ -0,0 +1,3 @@
+tests/inject_and_retrieve_data_1KB
+tests/inject_and_retrieve_data_1MB
+tests/inject_and_retrieve_data_1GB

examples/test.S

@@ -0,0 +1,24 @@
+#!/bin/sh
+
+set -o errexit
+set -o nounset
+
+while IFS= read -r test_directory
+do
+  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
+done < test.list

examples/test.c

@@ -0,0 +1,17 @@
+#include <assert.h>
+#include <stdio.h>
+
+#include "../../postject-api.h"
+
+int main()
+{
+  size_t size;
+  const void *ptr = postject_find_resource("foobar", &size, NULL);
+
+  assert(ptr != NULL);
+  assert(size > 0);
+
+  fwrite(ptr, size, 1, stdout);
+
+  return 0;
+}

examples/test.cpp

@@ -0,0 +1,18 @@
+#!/bin/sh
+
+set -o errexit
+set -o nounset
+
+bin="$TEMPORARY_DIRECTORY/a.out"
+cc test.c -o "$bin"
+
+input="$TEMPORARY_DIRECTORY/input.txt"
+head -c 1073741824 /dev/urandom > "$input"
+
+../../postject.py --overwrite "$bin" "foobar" "$input"
+
+output="$TEMPORARY_DIRECTORY/output.txt"
+
+"$bin" > "$output"
+
+diff "$input" "$output"

examples/test.sh

@@ -0,0 +1,17 @@
+#include <assert.h>
+#include <stdio.h>
+
+#include "../../postject-api.h"
+
+int main()
+{
+  size_t size;
+  const void *ptr = postject_find_resource("foobar", &size, NULL);
+
+  assert(ptr != NULL);
+  assert(size > 0);
+
+  fwrite(ptr, size, 1, stdout);
+
+  return 0;
+}

test.list

@@ -0,0 +1,18 @@
+#!/bin/sh
+
+set -o errexit
+set -o nounset
+
+bin="$TEMPORARY_DIRECTORY/a.out"
+cc test.c -o "$bin"
+
+input="$TEMPORARY_DIRECTORY/input.txt"
+head -c 1024 /dev/urandom > "$input"
+
+../../postject.py --overwrite "$bin" "foobar" "$input"
+
+output="$TEMPORARY_DIRECTORY/output.txt"
+
+"$bin" > "$output"
+
+diff "$input" "$output"

test.sh

@@ -0,0 +1,17 @@
+#include <assert.h>
+#include <stdio.h>
+
+#include "../../postject-api.h"
+
+int main()
+{
+  size_t size;
+  const void *ptr = postject_find_resource("foobar", &size, NULL);
+
+  assert(ptr != NULL);
+  assert(size > 0);
+
+  fwrite(ptr, size, 1, stdout);
+
+  return 0;
+}

tests/inject_and_retrieve_data_1GB/test.c

@@ -0,0 +1,18 @@
+#!/bin/sh
+
+set -o errexit
+set -o nounset
+
+bin="$TEMPORARY_DIRECTORY/a.out"
+cc test.c -o "$bin"
+
+input="$TEMPORARY_DIRECTORY/input.txt"
+head -c 1048576 /dev/urandom > "$input"
+
+../../postject.py --overwrite "$bin" "foobar" "$input"
+
+output="$TEMPORARY_DIRECTORY/output.txt"
+
+"$bin" > "$output"
+
+diff "$input" "$output"