More testing

Signed-off-by: Mattias Andrée <maandree@kth.se>

1 files changed, 47 insertions, 6 deletions

diff --git a/common.h b/common.h
index e55ca46..f3dc8de 100644
--- a/common.h
+++ b/common.h
@@ -108,13 +108,16 @@ static int
 run_tests(const char *name, enum libhashsum_algorithm algorithm, size_t hash_size,
           struct testcase *testcases, size_t ntestcases, char hexsum[])
 {
-	struct libhashsum_hasher hasher;
+	struct libhashsum_hasher hasher, hasher2;
+	unsigned char custom_hashbuffer[2048];
+	unsigned char custom_hashbuffer2[sizeof(custom_hashbuffer)];
+	int use_custom_hashbuffer = 0;
 	char *input, *p, bitstr[sizeof(" + b\"1234567\"")];
 	unsigned char extra_bit;
 	int ok = 1, caseok;
 	size_t i, j, input_string_len, input_total_len, input_size;
 	size_t bits;
-	for (i = 0; i < ntestcases; i++) {
+	for (i = 0; i < ntestcases; i++, use_custom_hashbuffer ^= 1) {
 		if (libhashsum_init_hasher(&hasher, algorithm)) {
 			perror("libhashsum_init_hasher");
 			return 2;
@@ -177,17 +180,47 @@ run_tests(const char *name, enum libhashsum_algorithm algorithm, size_t hash_siz
 		}
 		if (bits)
 			*p = testcases[i].input[input_string_len];
-		if (hasher.finalise(&hasher, input, input_total_len, (unsigned)bits, input_size)) {
-			perror("hasher.finalise");
+		hasher.hash_output = use_custom_hashbuffer ? custom_hashbuffer : NULL;
+		if (hasher.finalise_const(&hasher, input, input_total_len, (unsigned)bits)) {
+			perror("hasher.finalise_const");
 			return 2;
 		}
 		if (!hasher.hash_output) {
-			fprintf(stderr, "hasher.finalise did not set hasher.hash_output\n");
+			fprintf(stderr, "hasher.finalise_const did not set hasher.hash_output\n");
 			return 2;
 		}
-		free(input);
+#ifndef NO_CUSTOM_HASH_BUFFER_SUPPORT
+		if (use_custom_hashbuffer && hasher.hash_output != custom_hashbuffer) {
+			fprintf(stderr, "hasher.finalise_const did keep custom hash buffer\n");
+			return 2;
+		}
+#endif
 		hex(hexsum, hasher.hash_output, hasher.hash_size);
 		ok &= caseok = !testcases[i].output || !strcmp(hexsum, testcases[i].output);
+		if (libhashsum_init_hasher(&hasher2, algorithm)) {
+			perror("libhashsum_init_hasher");
+			return 2;
+		}
+		hasher2.hash_output = use_custom_hashbuffer ? custom_hashbuffer2 : NULL;
+		if (hasher2.finalise(&hasher2, input, input_total_len, (unsigned)bits, input_size)) {
+			perror("hasher.finalise");
+			return 2;
+		}
+		if (!hasher2.hash_output) {
+			fprintf(stderr, "hasher.finalise did not set hasher.hash_output\n");
+			return 2;
+		}
+#ifndef NO_CUSTOM_HASH_BUFFER_SUPPORT
+		if (use_custom_hashbuffer && hasher2.hash_output != custom_hashbuffer2) {
+			fprintf(stderr, "hasher.finalise did keep custom hash buffer\n");
+			return 2;
+		}
+#endif
+		if (memcmp(hasher2.hash_output, hasher.hash_output, hasher.hash_size)) {
+			fprintf(stderr, "hasher.finalise and hasher.finalise_const produced different hashes\n");
+			return 2;
+		}
+		free(input);
 		input = escape(testcases[i].input, input_string_len);
 		bitstr[0] = '\0';
 		if (bits) {
@@ -229,6 +262,14 @@ run_tests(const char *name, enum libhashsum_algorithm algorithm, size_t hash_siz
 	return run_tests(NAME, LIBHASHSUM_##ID, LIBHASHSUM_##ID##_HASH_SIZE,\
 	                 testcases, sizeof(testcases) / sizeof(*testcases), hexsum)
 
+# define ASSERT(ASSERTION)\
+	do {\
+		if ((ASSERTION))\
+			break;\
+		fprintf(stderr, "assertion `%s` at line %i failed\n", #ASSERTION, __LINE__);\
+		exit(2);\
+	} while (0)
+
 #endif