X-Git-Url: https://git.ozlabs.org/?a=blobdiff_plain;ds=sidebyside;f=ccan%2Frfc822%2Ftest%2Ftestdata.h;fp=ccan%2Frfc822%2Ftest%2Ftestdata.h;h=ceb3bcd892078e429509d1911988ad5ebbf45ac0;hb=de28abde1337e9255812ba720ffeef2c83a25dde;hp=0000000000000000000000000000000000000000;hpb=a3158e7994c9e05b3ef6683dea84ce0a6ae4609c;p=ccan

diff --git a/ccan/rfc822/test/testdata.h b/ccan/rfc822/test/testdata.h
new file mode 100644
index 00000000..ceb3bcd8
--- /dev/null
+++ b/ccan/rfc822/test/testdata.h
@@ -0,0 +1,124 @@
+#ifndef RFC822_TESTDATA_H
+#define RFC822_TESTDATA_H
+
+#include <ccan/talloc/talloc.h>
+#include <ccan/array_size/array_size.h>
+#include <ccan/foreach/foreach.h>
+
+struct testhdr {
+	const char *name, *val;
+};
+
+struct aexample {
+	const char *name;
+	struct testhdr *hdrs;
+	size_t nhdrs;
+	const char *body;
+};
+
+#define AEXAMPLE(s)				\
+	struct aexample s = {		\
+		.name = #s,			\
+		.hdrs = s##_hdrs,		\
+		.nhdrs = ARRAY_SIZE(s##_hdrs),	\
+		.body = s##_body,		\
+	};
+
+struct testhdr test_msg_1_hdrs[] = {
+	{"Date", "Tue, 22 Feb 2011 00:15:59 +1100"},
+	{"From", "Mister From <from@example.com>"},
+	{"To", "Mizz To <to@example.org>"},
+	{"Subject", "Some subject"},
+	{"Message-ID", "<20110221131559.GA28327@example>"},
+	{"MIME-Version", "1.0"},
+	{"Content-Type", "text/plain; charset=us-ascii"},
+	{"Content-Disposition", "inline"},
+};
+const char test_msg_1_body[] = "Test message\n";
+AEXAMPLE(test_msg_1);
+
+#define test_msg_empty_body_hdrs test_msg_1_hdrs
+const char test_msg_empty_body_body[] = "";
+AEXAMPLE(test_msg_empty_body);
+
+#define test_msg_nlnl_lf_hdrs test_msg_1_hdrs
+const char test_msg_nlnl_lf_body[] = "Message containing \n\n inside body\n";
+AEXAMPLE(test_msg_nlnl_lf);
+
+#define test_msg_nlnl_crlf_hdrs test_msg_1_hdrs
+const char test_msg_nlnl_crlf_body[] = "Message containing \r\n\r\n inside body\r\n";
+AEXAMPLE(test_msg_nlnl_crlf);
+
+#define test_msg_nlnl_mixed_hdrs test_msg_1_hdrs
+const char test_msg_nlnl_mixed_body[] = "Message containing both \n\n and \r\n\r\n inside body\n\r\n";
+AEXAMPLE(test_msg_nlnl_mixed);
+
+
+#define for_each_aexample(_e)				     \
+	foreach_ptr((_e), &test_msg_1, &test_msg_empty_body, \
+		    &test_msg_nlnl_lf, &test_msg_nlnl_crlf, \
+		    &test_msg_nlnl_mixed)
+
+#define for_each_aexample_buf(_e, _buf, _len)	\
+	for_each_aexample((_e)) 		\
+	if (((_buf) = assemble_msg((_e), &(_len))) != NULL)
+
+static inline int num_aexamples(void)
+{
+	const struct aexample *e;
+	int n = 0;
+
+	for_each_aexample(e)
+		n++;
+
+	return n;
+}
+
+static inline int num_aexample_hdrs(void)
+{
+	const struct aexample *e;
+	int n = 0;
+
+	for_each_aexample(e)
+		n += e->nhdrs;
+
+	return n;
+}
+
+static inline const char *assemble_msg(const struct aexample *e,
+				       size_t *len, int crlf)
+{
+	const char *nl = crlf ? "\r\n" : "\n";
+	int nln = crlf ? 2 : 1;
+	char *msg, *amsg;
+	size_t n = 0;
+	int i;
+
+	msg = talloc_strdup(NULL, "");
+	if (!msg)
+		return NULL;
+
+	for (i = 0; i < e->nhdrs; i++) {
+		amsg = talloc_asprintf_append(msg, "%s:%s%s", e->hdrs[i].name,
+					      e->hdrs[i].val, nl);
+		if (!amsg) {
+			talloc_free(msg);
+			return NULL;
+		}
+		msg = amsg;
+		n += strlen(e->hdrs[i].name) + strlen(e->hdrs[i].val) + 1 + nln;
+	}
+	amsg = talloc_asprintf_append(msg, "%s%s", nl, e->body);
+	if (!amsg) {
+		talloc_free(msg);
+		return NULL;
+	}
+	msg = amsg;
+	n += strlen(e->body) + nln;
+	*len = n;
+	return msg;
+}
+
+#define NLT(crlf)	((crlf) ? "CRLF" : "LF")
+
+#endif /* RFC822_TESTDATA_H */