rfc822: Allow test infrastructure to handle headers with minor errors
[ccan] / ccan / rfc822 / test / helper.c
1 #include <stdlib.h>
2 #include <stdio.h>
3
4 #include <ccan/talloc/talloc.h>
5 #include <ccan/failtest/failtest_override.h>
6 #include <ccan/failtest/failtest.h>
7
8 #include <ccan/rfc822/rfc822.h>
9
10 #include "helper.h"
11
12 /* failtest limitations mean we need these wrappers to test talloc
13  * failure paths. */
14 static void *malloc_wrapper(size_t size)
15 {
16         return malloc(size);
17 }
18
19 static void free_wrapper(void *ptr)
20 {
21         free(ptr);
22 }
23
24 static void *realloc_wrapper(void *ptr, size_t size)
25 {
26         return realloc(ptr, size);
27 }
28
29 #if 0
30 static void allocation_failure_exit(const char *s)
31 {
32         fprintf(stderr, "Allocation failure: %s", s);
33         exit(0);
34 }
35 #endif
36
37 static bool allocation_failed = false;
38
39 static void allocation_failure_continue(const char *s)
40 {
41         fprintf(stderr, "Allocation failure: %s", s);
42         allocation_failed = true;
43 }
44
45 void allocation_failure_check(void)
46 {
47         if (allocation_failed) {
48                 fprintf(stderr, "Exiting due to earlier failed allocation\n");
49                 exit(0);
50         }
51 }
52
53 void failtest_setup(int argc, char *argv[])
54 {
55         failtest_init(argc, argv);
56         rfc822_set_allocation_failure_handler(allocation_failure_continue);
57         talloc_set_allocator(malloc_wrapper, free_wrapper, realloc_wrapper);
58 }
59
60 void check_header(struct rfc822_msg *msg,
61                   struct rfc822_header *h,
62                   const char *name, const char *val,
63                   enum rfc822_header_errors experr, int crlf)
64 {
65         enum rfc822_header_errors errs;
66         struct bytestring hname, hvalue, hfull;
67         size_t namelen = strlen(name);
68         size_t valuelen = strlen(val);
69         size_t nln = crlf ? 2 : 1;
70         size_t fulllen = namelen + valuelen + 1 + nln;
71
72         errs = rfc822_header_errors(msg, h);
73         ok(errs == experr, "Header errors 0x%x != 0x%x", errs, experr);
74         allocation_failure_check();
75
76         hname = rfc822_header_raw_name(msg, h);
77         allocation_failure_check();
78
79         ok(hname.ptr && bytestring_eq(hname, bytestring_from_string(name)),
80            "Header name \"%.*s\"", (int)hname.len, hname.ptr);
81
82         hvalue = rfc822_header_raw_value(msg, h);
83         allocation_failure_check();
84
85         ok(hvalue.ptr && ((valuelen + nln) == hvalue.len)
86            && (memcmp(val, hvalue.ptr, valuelen) == 0)
87            && (!crlf || (hvalue.ptr[hvalue.len - 2] == '\r'))
88            && (hvalue.ptr[hvalue.len - 1] == '\n'),
89            "Header value");
90
91         hfull = rfc822_header_raw_content(msg, h);
92         allocation_failure_check();
93
94         ok(hfull.ptr && (fulllen == hfull.len)
95            && (memcmp(name, hfull.ptr, namelen) == 0)
96            && (hfull.ptr[namelen] == ':')
97            && (memcmp(val, hfull.ptr + namelen + 1, valuelen) == 0)
98            && (!crlf || (hfull.ptr[fulllen-2] == '\r'))
99            && (hfull.ptr[fulllen-1] == '\n'),
100            "Full header");
101 }