1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
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 = tal_arr(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 %zd, should be %zd",
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);
tal_free(msgbuf.ptr);
}
}
}
/* This exits depending on whether all tests passed */
failtest_exit(exit_status());
}
|
