X-Git-Url: http://git.ozlabs.org/?p=ccan;a=blobdiff_plain;f=ccan%2Frfc822%2Ftest%2Frun-unfold.c;fp=ccan%2Frfc822%2Ftest%2Frun-unfold.c;h=b9529269a01e1dbd70b93f7ae26101d47485c179;hp=0000000000000000000000000000000000000000;hb=de28abde1337e9255812ba720ffeef2c83a25dde;hpb=a3158e7994c9e05b3ef6683dea84ce0a6ae4609c

diff --git a/ccan/rfc822/test/run-unfold.c b/ccan/rfc822/test/run-unfold.c
new file mode 100644
index 00000000..b9529269
--- /dev/null
+++ b/ccan/rfc822/test/run-unfold.c
@@ -0,0 +1,121 @@
+#include <ccan/foreach/foreach.h>
+#include <ccan/failtest/failtest_override.h>
+#include <ccan/failtest/failtest.h>
+#include <ccan/tap/tap.h>
+#include <stdlib.h>
+#include <string.h>
+
+#define CCAN_RFC822_DEBUG
+
+#include <ccan/rfc822/rfc822.h>
+
+#include <ccan/rfc822/rfc822.c>
+
+#include "testdata.h"
+#include "helper.h"
+
+#define UNFOLDED " This is a string with\tlots of \tplaces to fold"
+#define FOLD_POINTS 11
+
+#define BEFORE "Date: Tue, 22 Feb 2011 00:15:59 +1100\n" \
+	"From: Mister From <from@example.com>\n" \
+	"To: Mizz To <to@example.org>\n" \
+	"Subject:"
+
+#define AFTER "Message-ID: <20110221131559.GA28327@example>\n" \
+	"\n" \
+	"body"
+
+static struct bytestring fold_and_assemble(int foldat, int crlf, int truncated)
+{
+	char *buf, *p;
+	int i, n = 0;
+
+	buf = talloc_array(NULL, char, strlen(BEFORE) + strlen(AFTER) + 3*strlen(UNFOLDED) + 2);
+	if (!buf)
+		exit(0);
+
+	memcpy(buf, BEFORE, strlen(BEFORE));
+
+	p = buf + strlen(BEFORE);
+
+	for (i = 0; i < strlen(UNFOLDED); i++) {
+		if (rfc822_iswsp(UNFOLDED[i])) {
+			n++;
+			if ((foldat == -1) || (foldat == n)) {
+				if (crlf)
+					*p++ = '\r';
+				*p++ = '\n';
+			}
+		}
+		*p++ = UNFOLDED[i];
+	}
+
+	if (!truncated) {
+		if (crlf)
+			*p++ = '\r';
+		*p++ = '\n';
+
+		memcpy(p, AFTER, strlen(AFTER));
+		p += strlen(AFTER);
+	}
+
+	return bytestring(buf, p - buf);
+}
+
+static void check_folded_header(const char *buf, size_t len)
+{
+	struct rfc822_msg *msg;
+	struct rfc822_header *hdr;
+	struct bytestring hunfold;
+
+	msg = rfc822_start(NULL, buf, len);
+	allocation_failure_check();
+
+	hdr = rfc822_first_header(msg);
+	allocation_failure_check();
+	hdr = rfc822_next_header(msg, hdr);
+	allocation_failure_check();
+	hdr = rfc822_next_header(msg, hdr);
+	allocation_failure_check();
+
+	/* This is the one we care about */
+	hdr = rfc822_next_header(msg, hdr);
+	allocation_failure_check();
+
+	ok(hdr && (rfc822_header_errors(msg, hdr) == 0), "Folded header valid");
+	allocation_failure_check();
+
+	hunfold = rfc822_header_unfolded_value(msg, hdr);
+	allocation_failure_check();
+
+	ok(hunfold.len == strlen(UNFOLDED), "Unfolded length %d, should be %d",
+	   hunfold.len, strlen(UNFOLDED));
+	ok1(memcmp(hunfold.ptr, UNFOLDED, hunfold.len) == 0);
+
+	rfc822_free(msg);
+	allocation_failure_check();
+}
+
+int main(int argc, char *argv[])
+{
+	struct bytestring msgbuf;
+	int crlf, truncated, i;
+
+	failtest_setup(argc, argv);
+
+	plan_tests(3 * 2 * 2 * (FOLD_POINTS + 2));
+
+	foreach_int(crlf, 0, 1) {
+		foreach_int(truncated, 0, 1) {
+			for (i = -1; i <= FOLD_POINTS; i++) {
+				msgbuf = fold_and_assemble(i, crlf, truncated);
+				check_folded_header(msgbuf.ptr, msgbuf.len);
+				talloc_free(msgbuf.ptr);
+			}
+		}
+	}
+
+	/* This exits depending on whether all tests passed */
+	failtest_exit(exit_status());
+}