1 #include <ccan/foreach/foreach.h>
2 #include <ccan/failtest/failtest_override.h>
3 #include <ccan/failtest/failtest.h>
4 #include <ccan/tap/tap.h>
8 #define CCAN_RFC822_DEBUG
10 #include <ccan/rfc822/rfc822.h>
12 #include <ccan/rfc822/rfc822.c>
17 #define NAME_TEMPLATE "X-Bad-Header"
18 #define NAME_TEMPLATE_LEN (strlen(NAME_TEMPLATE))
20 const char bad_name_template[] =
21 NAME_TEMPLATE ": This is a good header that will become bad\n";
23 static bool bad_name(const char *buf, char c)
25 struct rfc822_msg *msg;
26 struct rfc822_header *hdr;
27 enum rfc822_header_errors errs;
28 struct bytestring hname;
30 msg = rfc822_start(NULL, buf, strlen(buf));
32 allocation_failure_check();
34 hdr = rfc822_first_header(msg);
37 allocation_failure_check();
39 errs = rfc822_header_errors(msg, hdr);
41 allocation_failure_check();
43 ok1(!(errs & ~RFC822_HDR_BAD_NAME_CHARS));
45 /* Check raw_name still works properly */
46 hname = rfc822_header_raw_name(msg, hdr);
48 allocation_failure_check();
50 ok1(hname.ptr && hname.len == NAME_TEMPLATE_LEN);
51 ok1((hname.ptr[0] == c) &&
52 (memcmp(hname.ptr + 1, NAME_TEMPLATE + 1, NAME_TEMPLATE_LEN - 1) == 0));
56 allocation_failure_check();
58 return !!(errs & RFC822_HDR_BAD_NAME_CHARS);
61 int main(int argc, char *argv[])
65 failtest_setup(argc, argv);
67 plan_tests(5 * (1 + 3 + 4));
69 ok1(!bad_name(bad_name_template, bad_name_template[0]));
72 foreach_int(c, 'a', 'Z', '3') {
73 char *tmp = strdup(bad_name_template);
77 ok1(!bad_name(tmp, c));
83 foreach_int(c, ' ', '\t', '\b', '\x80') {
84 char *tmp = strdup(bad_name_template);
88 ok1(bad_name(tmp, c));
93 /* This exits depending on whether all tests passed */
94 failtest_exit(exit_status());