From: Mike Frysinger <vapier@gentoo.org>
Date: Thu, 24 Dec 2015 05:02:05 +0000 (-0500)
Subject: start a testsuite
X-Git-Url: https://git.wh0rd.org/?a=commitdiff_plain;h=03f0954062385ac3c9b3fd93da0bd39e9a0a3066;p=elf2flt.git
start a testsuite
---
diff --git a/.gitignore b/.gitignore
index 3b5a980..cdb96e4 100644
--- a/.gitignore
+++ b/.gitignore
@@ -11,6 +11,10 @@ a.out
*.gdb
*.exe
+# test files
+*.bin
+*.out
+
# autotool files
autom4te.cache
config.cache
diff --git a/Makefile.in b/Makefile.in
index 5c7df77..a6feea6 100644
--- a/Makefile.in
+++ b/Makefile.in
@@ -1,4 +1,6 @@
srcdir = @srcdir@
+abs_top_srcdir = @abs_top_srcdir@
+abs_top_builddir = @abs_top_builddir@
VPATH = @srcdir@
prefix = @prefix@
exec_prefix = @exec_prefix@
@@ -87,6 +89,17 @@ ld-elf2flt.sh: $(srcdir)/ld-elf2flt.in
Makefile: $(srcdir)/Makefile.in
./config.status $@
+tests/flthdr/generate: tests/flthdr/generate.o
+ $(link)
+
+check-flthdr:
+ mkdir -p tests/flthdr
+ $(MAKE) tests/flthdr/generate
+ abs_top_srcdir="$(abs_top_srcdir)" abs_top_builddir="$(abs_top_builddir)" \
+ $(srcdir)/tests/flthdr/test.sh
+
+check: check-flthdr
+
clean:
-rm -f $(PROGS) *.$(OBJEXT) .deps
diff --git a/tests/flthdr/.gitignore b/tests/flthdr/.gitignore
new file mode 100644
index 0000000..af82cca
--- /dev/null
+++ b/tests/flthdr/.gitignore
@@ -0,0 +1 @@
+/generate
diff --git a/tests/flthdr/basic.good b/tests/flthdr/basic.good
new file mode 100644
index 0000000..efd36d5
--- /dev/null
+++ b/tests/flthdr/basic.good
@@ -0,0 +1,12 @@
+basic.bin
+ Magic: bFLT
+ Rev: 4
+ Build Date: not specified
+ Entry: 0x1234
+ Data Start: 0x5678
+ Data End: 0xabcd
+ BSS End: 0xdef14565
+ Stack Size: 0x10aa
+ Reloc Start: 0x3456
+ Reloc Count: 0x0
+ Flags: 0xff ( Load-to-Ram Has-PIC-GOT Gzip-Compressed Gzip-Data-Compressed Kernel-Traced-Load L1-Scratch-Stack )
diff --git a/tests/flthdr/generate.c b/tests/flthdr/generate.c
new file mode 100644
index 0000000..9f234c1
--- /dev/null
+++ b/tests/flthdr/generate.c
@@ -0,0 +1,127 @@
+/* Helper tool for generating simple flat files that flthdr can read.
+ * Written by Mike Frysinger <vapier@gentoo.org>
+ * Distributed under the terms of the GNU General Public License v2
+ */
+
+#include <err.h>
+#include <errno.h>
+#include <getopt.h>
+#include <stdint.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <unistd.h>
+
+#include <arpa/inet.h>
+
+#include "flat.h"
+
+void usage(int status)
+{
+ fprintf(status ? stderr : stdout,
+ "Usage: generate [options]\n"
+ "\n"
+ "Helper tool for generating flat files for testing.\n"
+ "Flat file is written to stdout by default.\n"
+ "\n"
+ "Options:\n"
+ " -m <str> Set the magic (must be 4 bytes)\n"
+ " -v <u32> Set the version (rev)\n"
+ " -e <u32> Set the entry point\n"
+ " -d <u32> Set the data start addr\n"
+ " -D <u32> Set the data end addr\n"
+ " -B <u32> Set the bss end addr\n"
+ " -s <u32> Set the stack size\n"
+ " -r <u32> Set the reloc start addr\n"
+ " -R <u32> Set the reloc count\n"
+ " -f <u32> Set the flags\n"
+ " -b <u32> Set the build date\n"
+ " -F <u32> Set the filler field (each -F sets 1 field)\n"
+ " -o <out> Write output to a file\n"
+ " -h This help output\n"
+ );
+ exit(status);
+}
+
+void set32(uint32_t *field, const char *val)
+{
+ char *end;
+ long lval;
+
+ errno = 0;
+ lval = strtol(val, &end, 0);
+ if (*end)
+ err(1, "invalid number: %s", val);
+ *field = htonl(lval);
+}
+
+int main(int argc, char *argv[])
+{
+ FILE *output = stdout;
+ struct flat_hdr hdr;
+ int c, fillerpos;
+
+ fillerpos = 0;
+ memset(&hdr, 0, sizeof(hdr));
+ memcpy(hdr.magic, "bFLT", 4);
+ hdr.rev = htonl(FLAT_VERSION);
+
+ while ((c = getopt(argc, argv, "ho:m:v:e:d:D:B:s:r:R:f:b:F:")) != -1) {
+ switch (c) {
+ case 'm': /* magic */
+ if (strlen(optarg) != 4)
+ errx(1, "magic must be 4 bytes");
+ memcpy(hdr.magic, optarg, 4);
+ break;
+ case 'v': /* revision */
+ set32(&hdr.rev, optarg);
+ break;
+ case 'e': /* entry */
+ set32(&hdr.entry, optarg);
+ break;
+ case 'd': /* data_start */
+ set32(&hdr.data_start, optarg);
+ break;
+ case 'D': /* data_end */
+ set32(&hdr.data_end, optarg);
+ break;
+ case 'B': /* bss_end */
+ set32(&hdr.bss_end, optarg);
+ break;
+ case 's': /* stack_size */
+ set32(&hdr.stack_size, optarg);
+ break;
+ case 'r': /* reloc_start */
+ set32(&hdr.reloc_start, optarg);
+ break;
+ case 'R': /* reloc_count */
+ set32(&hdr.reloc_count, optarg);
+ break;
+ case 'f': /* flags */
+ set32(&hdr.flags, optarg);
+ break;
+ case 'b': /* build_date */
+ set32(&hdr.build_date, optarg);
+ break;
+ case 'F': /* filler */
+ if (fillerpos == 5)
+ errx(1, "can only use -F 5 times");
+ set32(&hdr.filler[fillerpos++], optarg);
+ break;
+ case 'o': /* output */
+ output = fopen(optarg, "wbe");
+ if (!output)
+ err(1, "could not open %s", optarg);
+ break;
+ case 'h':
+ usage(0);
+ default:
+ usage(1);
+ }
+ }
+
+ fwrite(&hdr, sizeof(hdr), 1, output);
+ fclose(output);
+
+ return 0;
+}
diff --git a/tests/flthdr/multi.good b/tests/flthdr/multi.good
new file mode 100644
index 0000000..ddbf3e1
--- /dev/null
+++ b/tests/flthdr/multi.good
@@ -0,0 +1,5 @@
+Flag Rev Text Data BSS Stack Relocs RAM Filename
+-----------------------------------------------------------
+- 4 -64 0 0 0 0 0 multi.bin
+- 4 -64 0 0 0 0 0 multi.bin
+- 4 -64 0 0 0 0 0 multi.bin
diff --git a/tests/flthdr/test.sh b/tests/flthdr/test.sh
new file mode 100755
index 0000000..654041a
--- /dev/null
+++ b/tests/flthdr/test.sh
@@ -0,0 +1,55 @@
+#!/bin/bash
+# Test behavior of flthdr.
+# Written by Mike Frysinger <vapier@gentoo.org>
+# Distributed under the terms of the GNU General Public License v2
+
+. "${0%/*}"/../lib.sh "flthdr"
+
+start 'cli no flags'
+if flthdr >stdout 2>stderr; then
+ fail 'exit'
+elif [[ -s stdout || ! -s stderr ]]; then
+ fail 'output'
+else
+ rm stdout stderr
+ pass
+fi
+
+start 'cli help'
+if ! flthdr -h >stdout 2>stderr; then
+ fail 'exit'
+elif [[ -s stderr || ! -s stdout ]]; then
+ fail 'output'
+else
+ rm stdout stderr
+ pass
+fi
+
+start 'basic flags'
+generate -o basic.bin \
+ -e 0x1234 \
+ -d 0x5678 \
+ -D 0xabcd \
+ -r 0x3456 \
+ -B 0xdef14565 \
+ -s 0x10aa \
+ -f 0xff
+if ! flthdr basic.bin > basic.out; then
+ fail 'run'
+elif ! diff -u basic.out "${srcdir}"/basic.good; then
+ fail 'diff'
+else
+ rm basic.bin basic.out
+ pass
+fi
+
+start 'multi output'
+generate -o multi.bin
+if ! flthdr multi.bin multi.bin multi.bin > multi.out; then
+ fail 'run'
+elif ! diff -u multi.out "${srcdir}"/multi.good; then
+ fail 'diff'
+else
+ rm multi.bin multi.out
+ pass
+fi
diff --git a/tests/lib.sh b/tests/lib.sh
new file mode 100644
index 0000000..a352083
--- /dev/null
+++ b/tests/lib.sh
@@ -0,0 +1,43 @@
+#!/bin/bash
+# Common test code.
+# Written by Mike Frysinger <vapier@gentoo.org>
+# Distributed under the terms of the GNU General Public License v2
+
+set -e
+
+SUBDIR=$1
+
+if [ -z "${abs_top_srcdir}" ]; then
+ abs_top_srcdir=$(realpath "$(dirname "$(realpath "$0")")/../..")
+ abs_top_builddir=${abs_top_srcdir}
+fi
+srcdir="${abs_top_srcdir}/tests/${SUBDIR}"
+builddir="${abs_top_builddir}/tests/${SUBDIR}"
+cd "${builddir}"
+
+PATH="${builddir}:${abs_top_builddir}:${PATH}"
+
+FAIL_COUNT=0
+
+start() {
+ printf '%s: testing %s ... ' "${SUBDIR}" "$1"
+}
+
+pass() {
+ echo 'OK'
+}
+
+fail() {
+ echo "FAIL ($*)"
+ : $(( FAIL_COUNT += 1 ))
+}
+
+test_runner_exit() {
+ if [ ${FAIL_COUNT} -ne 0 ]; then
+ echo "${SUBDIR}: some tests (${FAIL_COUNT}) failed!"
+ exit 1
+ else
+ echo "${SUBDIR}: all tests passed!"
+ fi
+}
+trap test_runner_exit EXIT