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 
27 #define SIG2 "v=1; a=rsa-sha256; c=simple/relaxed; d=example.com; s=test;\r\n\tt=1172620939; bh=QVUr2KBvm7/Q/ustiYzOlFMN9G8IMqBzUX81BdpjSDI=;\r\n\tl=340; h=Received:Received:Received:From:To:Date:Subject:Message-ID;\r\n\tb=OkSP9/9KTuHkXzc0KLoRcc5AcW97SA3RLPHuxdbTlCHyJgsyqrqPkOM85AdMJuYmH\r\n\t UiE1k12ge/AULlnlsBNLLRzO9aOpPVx4rLLpssFpPd0bWP37Y7apy0kM2ZrZDrexEo\r\n\t zPIEyU7aOnbLt/OTNSpiFwh51YGcBdlAtBS6bGEg="
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 	u_int flags;
43 	DKIM_STAT status;
44 	uint64_t fixed_time;
45 	DKIM *dkim;
46 	DKIM_LIB *lib;
47 	dkim_sigkey_t key;
48 	unsigned char hdr[MAXHEADER + 1];
49 
50 #ifdef USE_GNUTLS
51 	(void) gnutls_global_init();
52 #endif /* USE_GNUTLS */
53 
54 	/* instantiate the library */
55 	lib = dkim_init(NULL, NULL);
56 	assert(lib != NULL);
57 
58 	if (!dkim_libfeature(lib, DKIM_FEATURE_SHA256))
59 	{
60 		printf("*** simple/relaxed rsa-sha256 signing with lengths SKIPPED\n");
61 		dkim_close(lib);
62 		return 0;
63 	}
64 
65 	printf("*** simple/relaxed rsa-sha256 signing with lengths\n");
66 
67 #ifdef TEST_KEEP_FILES
68 	/* set flags */
69 	flags = (DKIM_LIBFLAGS_TMPFILES|DKIM_LIBFLAGS_KEEPFILES);
70 	(void) dkim_options(lib, DKIM_OP_SETOPT, DKIM_OPTS_FLAGS, &flags,
71 	                    sizeof flags);
72 #endif /* TEST_KEEP_FILES */
73 
74 	(void) dkim_options(lib, DKIM_OP_GETOPT, DKIM_OPTS_FLAGS, &flags,
75 	                    sizeof flags);
76 	flags |= DKIM_LIBFLAGS_SIGNLEN;
77 	(void) dkim_options(lib, DKIM_OP_SETOPT, DKIM_OPTS_FLAGS, &flags,
78 	                    sizeof flags);
79 
80 	key = KEY;
81 
82 	dkim = dkim_sign(lib, JOBID, NULL, key, SELECTOR, DOMAIN,
83 	                 DKIM_CANON_SIMPLE, DKIM_CANON_RELAXED,
84 	                 DKIM_SIGN_RSASHA256, -1L, &status);
85 	assert(dkim != NULL);
86 
87 	/* fix signing time */
88 	fixed_time = 1172620939;
89 	(void) dkim_options(lib, DKIM_OP_SETOPT, DKIM_OPTS_FIXEDTIME,
90 	                    &fixed_time, sizeof fixed_time);
91 
92 	status = dkim_header(dkim, HEADER02, strlen(HEADER02));
93 	assert(status == DKIM_STAT_OK);
94 
95 	status = dkim_header(dkim, HEADER03, strlen(HEADER03));
96 	assert(status == DKIM_STAT_OK);
97 
98 	status = dkim_header(dkim, HEADER04, strlen(HEADER04));
99 	assert(status == DKIM_STAT_OK);
100 
101 	status = dkim_header(dkim, HEADER05, strlen(HEADER05));
102 	assert(status == DKIM_STAT_OK);
103 
104 	status = dkim_header(dkim, HEADER06, strlen(HEADER06));
105 	assert(status == DKIM_STAT_OK);
106 
107 	status = dkim_header(dkim, HEADER07, strlen(HEADER07));
108 	assert(status == DKIM_STAT_OK);
109 
110 	status = dkim_header(dkim, HEADER08, strlen(HEADER08));
111 	assert(status == DKIM_STAT_OK);
112 
113 	status = dkim_header(dkim, HEADER09, strlen(HEADER09));
114 	assert(status == DKIM_STAT_OK);
115 
116 	status = dkim_eoh(dkim);
117 	assert(status == DKIM_STAT_OK);
118 
119 	status = dkim_body(dkim, BODY00, strlen(BODY00));
120 	assert(status == DKIM_STAT_OK);
121 
122 	status = dkim_body(dkim, BODY01, strlen(BODY01));
123 	assert(status == DKIM_STAT_OK);
124 
125 	status = dkim_body(dkim, BODY01A, strlen(BODY01A));
126 	assert(status == DKIM_STAT_OK);
127 	status = dkim_body(dkim, BODY01B, strlen(BODY01B));
128 	assert(status == DKIM_STAT_OK);
129 	status = dkim_body(dkim, BODY01C, strlen(BODY01C));
130 	assert(status == DKIM_STAT_OK);
131 	status = dkim_body(dkim, BODY01D, strlen(BODY01D));
132 	assert(status == DKIM_STAT_OK);
133 	status = dkim_body(dkim, BODY01E, strlen(BODY01E));
134 	assert(status == DKIM_STAT_OK);
135 
136 	status = dkim_body(dkim, BODY02, strlen(BODY02));
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, BODY04, strlen(BODY04));
143 	assert(status == DKIM_STAT_OK);
144 
145 	status = dkim_body(dkim, BODY03, strlen(BODY03));
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, BODY05, strlen(BODY05));
152 	assert(status == DKIM_STAT_OK);
153 
154 	status = dkim_body(dkim, BODY03, strlen(BODY03));
155 	assert(status == DKIM_STAT_OK);
156 
157 	status = dkim_body(dkim, BODY03, strlen(BODY03));
158 	assert(status == DKIM_STAT_OK);
159 
160 	status = dkim_eom(dkim, NULL);
161 	assert(status == DKIM_STAT_OK);
162 
163 	memset(hdr, '\0', sizeof hdr);
164 	status = dkim_getsighdr(dkim, hdr, sizeof hdr,
165 	                        strlen(DKIM_SIGNHEADER) + 2);
166 	assert(status == DKIM_STAT_OK);
167 	assert(strcmp(SIG2, hdr) == 0);
168 
169 	status = dkim_free(dkim);
170 	assert(status == DKIM_STAT_OK);
171 
172 	dkim_close(lib);
173 
174 	return 0;
175 }
176