Add tests for injecting large resources

RaisinTen · RaisinTen · commit 87437b362ba7 · 2022-09-20T12:42:20.000+05:30
Fixes: #12 Signed-off-by: Darshan Sen <raisinten@gmail.com>
diff --git a/.circleci/config.yml b/.circleci/config.yml
@@ -110,7 +110,7 @@ commands:
       - attach_workspace:
           at: .
 
-      - run: make check
+      - run: make test
 
 ## JOBS ##
 
diff --git a/Makefile b/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
diff --git a/README.markdown b/README.markdown
@@ -29,7 +29,7 @@ $ ./postject.py --macho-segment-name __ELECTRON /Users/dsanders/electron/src/out
 ### Testing
 
 ```sh
-$ make check
+$ make test
 ```
 
 ## Design
diff --git a/examples/Makefile b/examples/Makefile
diff --git a/examples/test.S b/examples/test.S
diff --git a/examples/test.c b/examples/test.c
diff --git a/examples/test.cpp b/examples/test.cpp
diff --git a/examples/test.sh b/examples/test.sh
diff --git a/test.list b/test.list
@@ -0,0 +1,3 @@
+tests/inject_and_retrieve_data_1KB
+tests/inject_and_retrieve_data_1MB
+tests/inject_and_retrieve_data_1GB
diff --git a/test.sh b/test.sh
@@ -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
diff --git a/tests/inject_and_retrieve_data_1GB/test.c b/tests/inject_and_retrieve_data_1GB/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;
+}
diff --git a/tests/inject_and_retrieve_data_1GB/test.sh b/tests/inject_and_retrieve_data_1GB/test.sh
@@ -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"
diff --git a/tests/inject_and_retrieve_data_1KB/test.c b/tests/inject_and_retrieve_data_1KB/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;
+}
diff --git a/tests/inject_and_retrieve_data_1KB/test.sh b/tests/inject_and_retrieve_data_1KB/test.sh
@@ -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"
diff --git a/tests/inject_and_retrieve_data_1MB/test.c b/tests/inject_and_retrieve_data_1MB/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;
+}
diff --git a/tests/inject_and_retrieve_data_1MB/test.sh b/tests/inject_and_retrieve_data_1MB/test.sh
@@ -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"

Original file line number	Diff line number	Diff line change
`@@ -0,0 +1,3 @@`
	`1`	`+tests/inject_and_retrieve_data_1KB`
	`2`	`+tests/inject_and_retrieve_data_1MB`
	`3`	`+tests/inject_and_retrieve_data_1GB`