1 /*
2 **  Copyright (c) 2005-2008 Sendmail, Inc. and its suppliers.
3 **    All rights reserved.
4 **
5 **  Copyright (c) 2009, 2011, 2012, The Trusted Domain Project.
6 **    All rights reserved.
7 */
8 
9 #include "build-config.h"
10 
11 /* system includes */
12 #include <sys/types.h>
13 #include <assert.h>
14 #include <string.h>
15 #include <stdio.h>
16 
17 #ifdef USE_GNUTLS
18 # include <gnutls/gnutls.h>
19 #endif /* USE_GNUTLS */
20 
21 /* libopendkim includes */
22 #include "../dkim.h"
23 #include "t-testdata.h"
24 
25 #define	MAXHEADER	4096
26 #define	SIGARRAY	10
27 
28 #define SIG2 "v=1; a=rsa-sha1; c=relaxed/relaxed; d=example.com; s=test;\r\n\tt=1172620939; bh=Z9ONHHsBrKN0pbfrOu025VfbdR4=;\r\n\th=Received:Received:Received:From:To:Date:Subject:Message-ID;\r\n\tb=Jf+j2RDZRkpIF1KaL5ByhHFPWj5RMeX5764IVlwIc11equjQND51K9FfL5pyjXvwj\r\n\t FoFPW0PGJb3liej6iDDEHgYpXR4p5qqlGx/C1Q9gf/MQN/Xlkv6ZXgR38QnWAfZxh5\r\n\t N1f5xUg+SJb5yBDoXklG62IRdia1Hq9MuiGumrGM="
29 
30 /*
31 **  MAIN -- program mainline
32 **
33 **  Parameters:
34 **  	The usual.
35 **
36 **  Return value:
37 **  	Exit status.
38 */
39 
40 int
main(int argc,char ** argv)41 main(int argc, char **argv)
42 {
43 #ifdef TEST_KEEP_FILES
44 	u_int flags;
45 #endif /* TEST_KEEP_FILES */
46 	DKIM_STAT status;
47 	u_int nsigs;
48 	DKIM *dkim;
49 	DKIM_LIB *lib;
50 	DKIM_SIGINFO *sig;
51 	dkim_query_t qtype = DKIM_QUERY_FILE;
52 	unsigned char hdr[MAXHEADER + 1];
53 	unsigned char signedhdrs[SIGARRAY][DKIM_MAXHEADER + 1];
54 
55 	printf("*** signed header content extraction\n");
56 
57 #ifdef USE_GNUTLS
58 	(void) gnutls_global_init();
59 #endif /* USE_GNUTLS */
60 
61 	/* instantiate the library */
62 	lib = dkim_init(NULL, NULL);
63 	assert(lib != NULL);
64 
65 #ifdef TEST_KEEP_FILES
66 	/* set flags */
67 	flags = (DKIM_LIBFLAGS_TMPFILES|DKIM_LIBFLAGS_KEEPFILES);
68 	(void) dkim_options(lib, DKIM_OP_SETOPT, DKIM_OPTS_FLAGS, &flags,
69 	                    sizeof flags);
70 #endif /* TEST_KEEP_FILES */
71 
72 	(void) dkim_options(lib, DKIM_OP_SETOPT, DKIM_OPTS_QUERYMETHOD,
73 	                    &qtype, sizeof qtype);
74 	(void) dkim_options(lib, DKIM_OP_SETOPT, DKIM_OPTS_QUERYINFO,
75 	                    KEYFILE, strlen(KEYFILE));
76 
77 	dkim = dkim_verify(lib, JOBID, NULL, &status);
78 	assert(dkim != NULL);
79 
80 	snprintf(hdr, sizeof hdr, "%s: %s", DKIM_SIGNHEADER, SIG2);
81 	status = dkim_header(dkim, hdr, strlen(hdr));
82 	assert(status == DKIM_STAT_OK);
83 
84 	status = dkim_header(dkim, HEADER01, strlen(HEADER01));
85 	assert(status == DKIM_STAT_OK);
86 
87 	status = dkim_header(dkim, HEADER02, strlen(HEADER02));
88 	assert(status == DKIM_STAT_OK);
89 
90 	status = dkim_header(dkim, HEADER03, strlen(HEADER03));
91 	assert(status == DKIM_STAT_OK);
92 
93 	status = dkim_header(dkim, HEADER04, strlen(HEADER04));
94 	assert(status == DKIM_STAT_OK);
95 
96 	status = dkim_header(dkim, HEADER05, strlen(HEADER05));
97 	assert(status == DKIM_STAT_OK);
98 
99 	status = dkim_header(dkim, HEADER06, strlen(HEADER06));
100 	assert(status == DKIM_STAT_OK);
101 
102 	status = dkim_header(dkim, HEADER07, strlen(HEADER07));
103 	assert(status == DKIM_STAT_OK);
104 
105 	status = dkim_header(dkim, HEADER08, strlen(HEADER08));
106 	assert(status == DKIM_STAT_OK);
107 
108 	status = dkim_header(dkim, HEADER09, strlen(HEADER09));
109 	assert(status == DKIM_STAT_OK);
110 
111 	status = dkim_header(dkim, HEADER10, strlen(HEADER09));
112 	assert(status == DKIM_STAT_OK);
113 
114 	status = dkim_eoh(dkim);
115 	assert(status == DKIM_STAT_OK);
116 
117 	status = dkim_body(dkim, BODY00, strlen(BODY00));
118 	assert(status == DKIM_STAT_OK);
119 
120 	status = dkim_body(dkim, BODY01, strlen(BODY01));
121 	assert(status == DKIM_STAT_OK);
122 
123 	status = dkim_body(dkim, BODY01A, strlen(BODY01A));
124 	assert(status == DKIM_STAT_OK);
125 	status = dkim_body(dkim, BODY01B, strlen(BODY01B));
126 	assert(status == DKIM_STAT_OK);
127 	status = dkim_body(dkim, BODY01C, strlen(BODY01C));
128 	assert(status == DKIM_STAT_OK);
129 	status = dkim_body(dkim, BODY01D, strlen(BODY01D));
130 	assert(status == DKIM_STAT_OK);
131 	status = dkim_body(dkim, BODY01E, strlen(BODY01E));
132 	assert(status == DKIM_STAT_OK);
133 
134 	status = dkim_body(dkim, BODY02, strlen(BODY02));
135 	assert(status == DKIM_STAT_OK);
136 
137 	status = dkim_body(dkim, BODY03, strlen(BODY03));
138 	assert(status == DKIM_STAT_OK);
139 
140 	status = dkim_body(dkim, BODY04, strlen(BODY04));
141 	assert(status == DKIM_STAT_OK);
142 
143 	status = dkim_body(dkim, BODY03, strlen(BODY03));
144 	assert(status == DKIM_STAT_OK);
145 
146 	status = dkim_body(dkim, BODY03, strlen(BODY03));
147 	assert(status == DKIM_STAT_OK);
148 
149 	status = dkim_body(dkim, BODY05, strlen(BODY05));
150 	assert(status == DKIM_STAT_OK);
151 
152 	status = dkim_body(dkim, BODY03, strlen(BODY03));
153 	assert(status == DKIM_STAT_OK);
154 
155 	status = dkim_body(dkim, BODY03, strlen(BODY03));
156 	assert(status == DKIM_STAT_OK);
157 
158 	status = dkim_eom(dkim, NULL);
159 	assert(status == DKIM_STAT_OK);
160 
161 	sig = dkim_getsignature(dkim);
162 	assert(sig != NULL);
163 	nsigs = SIGARRAY;
164 	status = dkim_sig_getsignedhdrs(dkim, sig, (u_char *) signedhdrs,
165 	                                DKIM_MAXHEADER + 1, &nsigs);
166 	assert(status == DKIM_STAT_OK);
167 	assert(nsigs == 8);
168 	assert(strcmp(signedhdrs[0], HEADER04) == 0);
169 	assert(strcmp(signedhdrs[1], HEADER03) == 0);
170 	assert(strcmp(signedhdrs[2], HEADER02) == 0);
171 	assert(strcmp(signedhdrs[3], HEADER05) == 0);
172 	assert(strcmp(signedhdrs[4], HEADER06) == 0);
173 	assert(strcmp(signedhdrs[5], HEADER07) == 0);
174 	assert(strcmp(signedhdrs[6], HEADER08) == 0);
175 	assert(strcmp(signedhdrs[7], HEADER09) == 0);
176 
177 	status = dkim_free(dkim);
178 	assert(status == DKIM_STAT_OK);
179 
180 	dkim_close(lib);
181 
182 	return 0;
183 }
184