[PATCH v3 09/10] random-dump.c: new test-binary to generate dump files
David Bremner
david at tethera.net
Sat Jan 14 17:40:23 PST 2012
From: David Bremner <bremner at debian.org>
This binary creates a "torture test" dump file for the new dump
format.
---
test/Makefile.local | 4 ++
test/basic | 2 +-
test/random-dump.c | 144 +++++++++++++++++++++++++++++++++++++++++++++++++++
3 files changed, 149 insertions(+), 1 deletions(-)
create mode 100644 test/random-dump.c
diff --git a/test/Makefile.local b/test/Makefile.local
index ba697f4..b59f837 100644
--- a/test/Makefile.local
+++ b/test/Makefile.local
@@ -16,6 +16,9 @@ $(dir)/arg-test: $(dir)/arg-test.o command-line-arguments.o util/libutil.a
$(dir)/hex-xcode: $(dir)/hex-xcode.o command-line-arguments.o util/libutil.a
$(call quiet,CC) -I. $^ -o $@ -ltalloc
+$(dir)/random-dump: $(dir)/random-dump.o command-line-arguments.o util/libutil.a
+ $(call quiet,CC) -I. $^ -o $@ -ltalloc -lm
+
$(dir)/smtp-dummy: $(smtp_dummy_modules)
$(call quiet,CC) $^ -o $@
@@ -25,6 +28,7 @@ $(dir)/symbol-test: $(dir)/symbol-test.o
.PHONY: test check
test-binaries: $(dir)/arg-test $(dir)/hex-xcode \
+ $(dir)/random-dump \
$(dir)/smtp-dummy $(dir)/symbol-test
test: all test-binaries
diff --git a/test/basic b/test/basic
index af57026..e3a6cef 100755
--- a/test/basic
+++ b/test/basic
@@ -54,7 +54,7 @@ test_begin_subtest 'Ensure that all available tests will be run by notmuch-test'
eval $(sed -n -e '/^TESTS="$/,/^"$/p' $TEST_DIRECTORY/notmuch-test)
tests_in_suite=$(for i in $TESTS; do echo $i; done | sort)
available=$(find "$TEST_DIRECTORY" -maxdepth 1 -type f -executable -printf '%f\n' | \
- sed -r -e "/^(aggregate-results.sh|notmuch-test|smtp-dummy|test-verbose|symbol-test|arg-test|hex-xcode)$/d" | \
+ sed -r -e "/^(aggregate-results.sh|notmuch-test|smtp-dummy|test-verbose|symbol-test|arg-test|hex-xcode|random-dump)$/d" | \
sort)
test_expect_equal "$tests_in_suite" "$available"
diff --git a/test/random-dump.c b/test/random-dump.c
new file mode 100644
index 0000000..1949425
--- /dev/null
+++ b/test/random-dump.c
@@ -0,0 +1,144 @@
+/*
+ Generate a random dump file in 'notmuch' format.
+ Generated message-id's and tags are intentionally nasty.
+
+ We restrict ourselves to 7 bit message-ids, because generating
+ random valid UTF-8 seems like work. And invalid UTF-8 can't be
+ round-tripped via Xapian.
+
+ */
+
+#include <stdlib.h>
+#include <assert.h>
+#include <talloc.h>
+#include <string.h>
+#include "math.h"
+#include "hex-escape.h"
+#include "command-line-arguments.h"
+
+static void
+hex_out (void *ctx, char *buf)
+{
+ static char *encoded_buf = NULL;
+ static size_t encoded_buf_size = 0;
+
+ if (hex_encode (ctx, buf, &encoded_buf, &encoded_buf_size) != HEX_SUCCESS) {
+ fprintf (stderr, "Hex encoding failed");
+ exit (1);
+ }
+
+ fputs (encoded_buf, stdout);
+}
+
+static void
+random_chars (char *buf, int from, int stop, int max_char,
+ const char *blacklist)
+{
+ int i;
+
+ for (i = from; i < stop; i++) {
+ do {
+ buf[i] = ' ' + (random () % (max_char - ' '));
+ } while (blacklist && strchr (blacklist, buf[i]));
+ }
+}
+
+static void
+random_tag (void *ctx, size_t len)
+{
+ static char *buf = NULL;
+ static size_t buf_len = 0;
+
+ int use = (random () % (len - 1)) + 1;
+
+ if (len > buf_len) {
+ buf = talloc_realloc (ctx, buf, char, len);
+ buf_len = len;
+ }
+
+ random_chars (buf, 0, use, 255, NULL);
+
+ buf[use] = '\0';
+
+ hex_out (ctx, buf);
+}
+
+static void
+random_message_id (void *ctx, size_t len)
+{
+ static char *buf = NULL;
+ static size_t buf_len = 0;
+
+ int lhs_len = (random () % (len / 2 - 1)) + 1;
+
+ int rhs_len = (random () % len / 2) + 1;
+
+ const char *blacklist = "\n\r@<>[]()";
+
+ if (len > buf_len) {
+ buf = talloc_realloc (ctx, buf, char, len);
+ buf_len = len;
+ }
+
+ random_chars (buf, 0, lhs_len, 127, blacklist);
+
+ buf[lhs_len] = '@';
+
+ random_chars (buf, lhs_len + 1, lhs_len + rhs_len + 1, 127, blacklist);
+
+ hex_out (ctx, buf);
+}
+
+int
+main (int argc, char **argv)
+{
+
+ void *ctx = talloc_new (NULL);
+ int num_lines = 500;
+ int max_tags = 10;
+ int message_id_len = 100;
+ int tag_len = 50;
+ int seed = 734569;
+
+ int pad_tag = 0, pad_mid = 0;
+
+ notmuch_opt_desc_t options[] = {
+ { NOTMUCH_OPT_INT, &num_lines, "num-lines", 'n', 0 },
+ { NOTMUCH_OPT_INT, &max_tags, "max-tags", 'm', 0 },
+ { NOTMUCH_OPT_INT, &message_id_len, "message-id-len", 'M', 0 },
+ { NOTMUCH_OPT_INT, &tag_len, "tag-len", 't', 0 },
+ { NOTMUCH_OPT_INT, &seed, "tag-len", 't', 0 },
+ { 0, 0, 0, 0, 0 }
+ };
+
+ int opt_index = parse_arguments (argc, argv, options, 1);
+
+ if (opt_index < 0)
+ exit (1);
+
+ pad_mid = ((int) log10 (num_lines) + 1);
+ pad_tag = ((int) log10 (max_tags)) + 1;
+
+ srandom (seed);
+
+ int line;
+ for (line = 0; line < num_lines; line++) {
+
+ printf ("%0*d-", pad_mid, line);
+
+ random_message_id (ctx, message_id_len);
+
+ int num_tags = random () % (max_tags + 1);
+
+ int j;
+ for (j = 0; j < num_tags; j++) {
+ printf (" %0*d-", pad_tag, j);
+ random_tag (ctx, tag_len);
+ }
+ putchar ('\n');
+ }
+
+ talloc_free (ctx);
+
+ return 0;
+}
--
1.7.7.3
More information about the notmuch
mailing list