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