[HEIMDAL-646] malloc(0) checks for AIX
[heimdal.git] / lib / hcrypto / test_dh.c
blob61505ae8d9999766038e5381c93e3a3c192614c2
1 /*
2 * Copyright (c) 2007, Novell, Inc.
3 * Author: Matthias Koenig <mkoenig@suse.de>
5 * All rights reserved.
7 * Redistribution and use in source and binary forms, with or without
8 * modification, are permitted provided that the following conditions are met:
10 * * Redistributions of source code must retain the above copyright notice, this
11 * list of conditions and the following disclaimer.
13 * * Redistributions in binary form must reproduce the above copyright notice,
14 * this list of conditions and the following disclaimer in the documentation
15 * and/or other materials provided with the distribution.
17 * * Neither the name of the Novell nor the names of its contributors may be used
18 * to endorse or promote products derived from this software without specific
19 * prior written permission.
21 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
22 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
23 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
24 * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
25 * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
26 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
27 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
28 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
29 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
30 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
31 * POSSIBILITY OF SUCH DAMAGE.
34 /* openssl diffie-hellman test code
35 * works with openssl-0.9.8e
36 * primes with 3072 and 6144 bits as specified in RFC3526
37 * fail since openssl-0.9.8f
40 #include <config.h>
42 #include <stdio.h>
43 #include <ctype.h>
45 #include <roken.h>
46 #include <getarg.h>
48 #include <dh.h>
54 static int version_flag;
55 static int help_flag;
56 static int verbose;
58 static struct getargs args[] = {
59 { "verbose", 0, arg_flag, &verbose,
60 "verbose output from tests", NULL },
61 { "version", 0, arg_flag, &version_flag,
62 "print version", NULL },
63 { "help", 0, arg_flag, &help_flag,
64 NULL, NULL }
71 #define OAKLEY_PRIME_MODP768 \
72 "FFFFFFFF FFFFFFFF C90FDAA2 2168C234 C4C6628B 80DC1CD1" \
73 "29024E08 8A67CC74 020BBEA6 3B139B22 514A0879 8E3404DD" \
74 "EF9519B3 CD3A431B 302B0A6D F25F1437 4FE1356D 6D51C245" \
75 "E485B576 625E7EC6 F44C42E9 A63A3620 FFFFFFFF FFFFFFFF"
77 #define OAKLEY_PRIME_MODP1024 \
78 "FFFFFFFF FFFFFFFF C90FDAA2 2168C234 C4C6628B 80DC1CD1" \
79 "29024E08 8A67CC74 020BBEA6 3B139B22 514A0879 8E3404DD" \
80 "EF9519B3 CD3A431B 302B0A6D F25F1437 4FE1356D 6D51C245" \
81 "E485B576 625E7EC6 F44C42E9 A637ED6B 0BFF5CB6 F406B7ED" \
82 "EE386BFB 5A899FA5 AE9F2411 7C4B1FE6 49286651 ECE65381" \
83 "FFFFFFFF FFFFFFFF"
85 #define OAKLEY_PRIME_MODP1536 \
86 "FFFFFFFF FFFFFFFF C90FDAA2 2168C234 C4C6628B 80DC1CD1" \
87 "29024E08 8A67CC74 020BBEA6 3B139B22 514A0879 8E3404DD" \
88 "EF9519B3 CD3A431B 302B0A6D F25F1437 4FE1356D 6D51C245" \
89 "E485B576 625E7EC6 F44C42E9 A637ED6B 0BFF5CB6 F406B7ED" \
90 "EE386BFB 5A899FA5 AE9F2411 7C4B1FE6 49286651 ECE45B3D" \
91 "C2007CB8 A163BF05 98DA4836 1C55D39A 69163FA8 FD24CF5F" \
92 "83655D23 DCA3AD96 1C62F356 208552BB 9ED52907 7096966D" \
93 "670C354E 4ABC9804 F1746C08 CA237327 FFFFFFFF FFFFFFFF"
95 /* RFC 3526 */
96 #define OAKLEY_PRIME_MODP2048 \
97 "FFFFFFFF FFFFFFFF C90FDAA2 2168C234 C4C6628B 80DC1CD1" \
98 "29024E08 8A67CC74 020BBEA6 3B139B22 514A0879 8E3404DD" \
99 "EF9519B3 CD3A431B 302B0A6D F25F1437 4FE1356D 6D51C245" \
100 "E485B576 625E7EC6 F44C42E9 A637ED6B 0BFF5CB6 F406B7ED" \
101 "EE386BFB 5A899FA5 AE9F2411 7C4B1FE6 49286651 ECE45B3D" \
102 "C2007CB8 A163BF05 98DA4836 1C55D39A 69163FA8 FD24CF5F" \
103 "83655D23 DCA3AD96 1C62F356 208552BB 9ED52907 7096966D" \
104 "670C354E 4ABC9804 F1746C08 CA18217C 32905E46 2E36CE3B" \
105 "E39E772C 180E8603 9B2783A2 EC07A28F B5C55DF0 6F4C52C9" \
106 "DE2BCBF6 95581718 3995497C EA956AE5 15D22618 98FA0510" \
107 "15728E5A 8AACAA68 FFFFFFFF FFFFFFFF"
109 #define OAKLEY_PRIME_MODP3072 \
110 "FFFFFFFF FFFFFFFF C90FDAA2 2168C234 C4C6628B 80DC1CD1" \
111 "29024E08 8A67CC74 020BBEA6 3B139B22 514A0879 8E3404DD" \
112 "EF9519B3 CD3A431B 302B0A6D F25F1437 4FE1356D 6D51C245" \
113 "E485B576 625E7EC6 F44C42E9 A637ED6B 0BFF5CB6 F406B7ED" \
114 "EE386BFB 5A899FA5 AE9F2411 7C4B1FE6 49286651 ECE45B3D" \
115 "C2007CB8 A163BF05 98DA4836 1C55D39A 69163FA8 FD24CF5F" \
116 "83655D23 DCA3AD96 1C62F356 208552BB 9ED52907 7096966D" \
117 "670C354E 4ABC9804 F1746C08 CA18217C 32905E46 2E36CE3B" \
118 "E39E772C 180E8603 9B2783A2 EC07A28F B5C55DF0 6F4C52C9" \
119 "DE2BCBF6 95581718 3995497C EA956AE5 15D22618 98FA0510" \
120 "15728E5A 8AAAC42D AD33170D 04507A33 A85521AB DF1CBA64" \
121 "ECFB8504 58DBEF0A 8AEA7157 5D060C7D B3970F85 A6E1E4C7" \
122 "ABF5AE8C DB0933D7 1E8C94E0 4A25619D CEE3D226 1AD2EE6B" \
123 "F12FFA06 D98A0864 D8760273 3EC86A64 521F2B18 177B200C" \
124 "BBE11757 7A615D6C 770988C0 BAD946E2 08E24FA0 74E5AB31" \
125 "43DB5BFC E0FD108E 4B82D120 A93AD2CA FFFFFFFF FFFFFFFF"
127 #define OAKLEY_PRIME_MODP4096 \
128 "FFFFFFFF FFFFFFFF C90FDAA2 2168C234 C4C6628B 80DC1CD1" \
129 "29024E08 8A67CC74 020BBEA6 3B139B22 514A0879 8E3404DD" \
130 "EF9519B3 CD3A431B 302B0A6D F25F1437 4FE1356D 6D51C245" \
131 "E485B576 625E7EC6 F44C42E9 A637ED6B 0BFF5CB6 F406B7ED" \
132 "EE386BFB 5A899FA5 AE9F2411 7C4B1FE6 49286651 ECE45B3D" \
133 "C2007CB8 A163BF05 98DA4836 1C55D39A 69163FA8 FD24CF5F" \
134 "83655D23 DCA3AD96 1C62F356 208552BB 9ED52907 7096966D" \
135 "670C354E 4ABC9804 F1746C08 CA18217C 32905E46 2E36CE3B" \
136 "E39E772C 180E8603 9B2783A2 EC07A28F B5C55DF0 6F4C52C9" \
137 "DE2BCBF6 95581718 3995497C EA956AE5 15D22618 98FA0510" \
138 "15728E5A 8AAAC42D AD33170D 04507A33 A85521AB DF1CBA64" \
139 "ECFB8504 58DBEF0A 8AEA7157 5D060C7D B3970F85 A6E1E4C7" \
140 "ABF5AE8C DB0933D7 1E8C94E0 4A25619D CEE3D226 1AD2EE6B" \
141 "F12FFA06 D98A0864 D8760273 3EC86A64 521F2B18 177B200C" \
142 "BBE11757 7A615D6C 770988C0 BAD946E2 08E24FA0 74E5AB31" \
143 "43DB5BFC E0FD108E 4B82D120 A9210801 1A723C12 A787E6D7" \
144 "88719A10 BDBA5B26 99C32718 6AF4E23C 1A946834 B6150BDA" \
145 "2583E9CA 2AD44CE8 DBBBC2DB 04DE8EF9 2E8EFC14 1FBECAA6" \
146 "287C5947 4E6BC05D 99B2964F A090C3A2 233BA186 515BE7ED" \
147 "1F612970 CEE2D7AF B81BDD76 2170481C D0069127 D5B05AA9" \
148 "93B4EA98 8D8FDDC1 86FFB7DC 90A6C08F 4DF435C9 34063199" \
149 "FFFFFFFF FFFFFFFF"
151 #define OAKLEY_PRIME_MODP6144 \
152 "FFFFFFFF FFFFFFFF C90FDAA2 2168C234 C4C6628B 80DC1CD1" \
153 "29024E08 8A67CC74 020BBEA6 3B139B22 514A0879 8E3404DD" \
154 "EF9519B3 CD3A431B 302B0A6D F25F1437 4FE1356D 6D51C245" \
155 "E485B576 625E7EC6 F44C42E9 A637ED6B 0BFF5CB6 F406B7ED" \
156 "EE386BFB 5A899FA5 AE9F2411 7C4B1FE6 49286651 ECE45B3D" \
157 "C2007CB8 A163BF05 98DA4836 1C55D39A 69163FA8 FD24CF5F" \
158 "83655D23 DCA3AD96 1C62F356 208552BB 9ED52907 7096966D" \
159 "670C354E 4ABC9804 F1746C08 CA18217C 32905E46 2E36CE3B" \
160 "E39E772C 180E8603 9B2783A2 EC07A28F B5C55DF0 6F4C52C9" \
161 "DE2BCBF6 95581718 3995497C EA956AE5 15D22618 98FA0510" \
162 "15728E5A 8AAAC42D AD33170D 04507A33 A85521AB DF1CBA64" \
163 "ECFB8504 58DBEF0A 8AEA7157 5D060C7D B3970F85 A6E1E4C7" \
164 "ABF5AE8C DB0933D7 1E8C94E0 4A25619D CEE3D226 1AD2EE6B" \
165 "F12FFA06 D98A0864 D8760273 3EC86A64 521F2B18 177B200C" \
166 "BBE11757 7A615D6C 770988C0 BAD946E2 08E24FA0 74E5AB31" \
167 "43DB5BFC E0FD108E 4B82D120 A9210801 1A723C12 A787E6D7" \
168 "88719A10 BDBA5B26 99C32718 6AF4E23C 1A946834 B6150BDA" \
169 "2583E9CA 2AD44CE8 DBBBC2DB 04DE8EF9 2E8EFC14 1FBECAA6" \
170 "287C5947 4E6BC05D 99B2964F A090C3A2 233BA186 515BE7ED" \
171 "1F612970 CEE2D7AF B81BDD76 2170481C D0069127 D5B05AA9" \
172 "93B4EA98 8D8FDDC1 86FFB7DC 90A6C08F 4DF435C9 34028492" \
173 "36C3FAB4 D27C7026 C1D4DCB2 602646DE C9751E76 3DBA37BD" \
174 "F8FF9406 AD9E530E E5DB382F 413001AE B06A53ED 9027D831" \
175 "179727B0 865A8918 DA3EDBEB CF9B14ED 44CE6CBA CED4BB1B" \
176 "DB7F1447 E6CC254B 33205151 2BD7AF42 6FB8F401 378CD2BF" \
177 "5983CA01 C64B92EC F032EA15 D1721D03 F482D7CE 6E74FEF6" \
178 "D55E702F 46980C82 B5A84031 900B1C9E 59E7C97F BEC7E8F3" \
179 "23A97A7E 36CC88BE 0F1D45B7 FF585AC5 4BD407B2 2B4154AA" \
180 "CC8F6D7E BF48E1D8 14CC5ED2 0F8037E0 A79715EE F29BE328" \
181 "06A1D58B B7C5DA76 F550AA3D 8A1FBFF0 EB19CCB1 A313D55C" \
182 "DA56C9EC 2EF29632 387FE8D7 6E3C0468 043E8F66 3F4860EE" \
183 "12BF2D5B 0B7474D6 E694F91E 6DCC4024 FFFFFFFF FFFFFFFF"
185 #define OAKLEY_PRIME_MODP8192 \
186 "FFFFFFFF FFFFFFFF C90FDAA2 2168C234 C4C6628B 80DC1CD1" \
187 "29024E08 8A67CC74 020BBEA6 3B139B22 514A0879 8E3404DD" \
188 "EF9519B3 CD3A431B 302B0A6D F25F1437 4FE1356D 6D51C245" \
189 "E485B576 625E7EC6 F44C42E9 A637ED6B 0BFF5CB6 F406B7ED" \
190 "EE386BFB 5A899FA5 AE9F2411 7C4B1FE6 49286651 ECE45B3D" \
191 "C2007CB8 A163BF05 98DA4836 1C55D39A 69163FA8 FD24CF5F" \
192 "83655D23 DCA3AD96 1C62F356 208552BB 9ED52907 7096966D" \
193 "670C354E 4ABC9804 F1746C08 CA18217C 32905E46 2E36CE3B" \
194 "E39E772C 180E8603 9B2783A2 EC07A28F B5C55DF0 6F4C52C9" \
195 "DE2BCBF6 95581718 3995497C EA956AE5 15D22618 98FA0510" \
196 "15728E5A 8AAAC42D AD33170D 04507A33 A85521AB DF1CBA64" \
197 "ECFB8504 58DBEF0A 8AEA7157 5D060C7D B3970F85 A6E1E4C7" \
198 "ABF5AE8C DB0933D7 1E8C94E0 4A25619D CEE3D226 1AD2EE6B" \
199 "F12FFA06 D98A0864 D8760273 3EC86A64 521F2B18 177B200C" \
200 "BBE11757 7A615D6C 770988C0 BAD946E2 08E24FA0 74E5AB31" \
201 "43DB5BFC E0FD108E 4B82D120 A9210801 1A723C12 A787E6D7" \
202 "88719A10 BDBA5B26 99C32718 6AF4E23C 1A946834 B6150BDA" \
203 "2583E9CA 2AD44CE8 DBBBC2DB 04DE8EF9 2E8EFC14 1FBECAA6" \
204 "287C5947 4E6BC05D 99B2964F A090C3A2 233BA186 515BE7ED" \
205 "1F612970 CEE2D7AF B81BDD76 2170481C D0069127 D5B05AA9" \
206 "93B4EA98 8D8FDDC1 86FFB7DC 90A6C08F 4DF435C9 34028492" \
207 "36C3FAB4 D27C7026 C1D4DCB2 602646DE C9751E76 3DBA37BD" \
208 "F8FF9406 AD9E530E E5DB382F 413001AE B06A53ED 9027D831" \
209 "179727B0 865A8918 DA3EDBEB CF9B14ED 44CE6CBA CED4BB1B" \
210 "DB7F1447 E6CC254B 33205151 2BD7AF42 6FB8F401 378CD2BF" \
211 "5983CA01 C64B92EC F032EA15 D1721D03 F482D7CE 6E74FEF6" \
212 "D55E702F 46980C82 B5A84031 900B1C9E 59E7C97F BEC7E8F3" \
213 "23A97A7E 36CC88BE 0F1D45B7 FF585AC5 4BD407B2 2B4154AA" \
214 "CC8F6D7E BF48E1D8 14CC5ED2 0F8037E0 A79715EE F29BE328" \
215 "06A1D58B B7C5DA76 F550AA3D 8A1FBFF0 EB19CCB1 A313D55C" \
216 "DA56C9EC 2EF29632 387FE8D7 6E3C0468 043E8F66 3F4860EE" \
217 "12BF2D5B 0B7474D6 E694F91E 6DBE1159 74A3926F 12FEE5E4" \
218 "38777CB6 A932DF8C D8BEC4D0 73B931BA 3BC832B6 8D9DD300" \
219 "741FA7BF 8AFC47ED 2576F693 6BA42466 3AAB639C 5AE4F568" \
220 "3423B474 2BF1C978 238F16CB E39D652D E3FDB8BE FC848AD9" \
221 "22222E04 A4037C07 13EB57A8 1A23F0C7 3473FC64 6CEA306B" \
222 "4BCBC886 2F8385DD FA9D4B7F A2C087E8 79683303 ED5BDD3A" \
223 "062B3CF5 B3A278A6 6D2A13F8 3F44F82D DF310EE0 74AB6A36" \
224 "4597E899 A0255DC1 64F31CC5 0846851D F9AB4819 5DED7EA1" \
225 "B1D510BD 7EE74D73 FAF36BC3 1ECFA268 359046F4 EB879F92" \
226 "4009438B 481C6CD7 889A002E D5EE382B C9190DA6 FC026E47" \
227 "9558E447 5677E9AA 9E3050E2 765694DF C81F56E8 80B96E71" \
228 "60C980DD 98EDD3DF FFFFFFFF FFFFFFFF"
230 struct prime {
231 char *name;
232 char *value;
233 } primes[] = {
234 { "modp768", OAKLEY_PRIME_MODP768 },
235 { "modp1024", OAKLEY_PRIME_MODP1024 },
236 { "modp1536", OAKLEY_PRIME_MODP1536 },
237 { "modp2048", OAKLEY_PRIME_MODP2048 },
238 { "modp3072", OAKLEY_PRIME_MODP3072 },
239 { "modp4096", OAKLEY_PRIME_MODP4096 },
240 { "modp6144", OAKLEY_PRIME_MODP6144 },
241 { "modp8192", OAKLEY_PRIME_MODP8192 },
242 { NULL, NULL }
246 * exchange a string based "base" to a value.
249 static char *
250 str2val(const char *str, int base, size_t *len)
252 int f;
253 size_t i;
254 char *dst;
255 char *rp;
256 const char *p;
257 char b[3];
259 i = 0;
260 for (p = str; *p != '\0'; p++) {
261 if (isxdigit((int)*p))
262 i++;
263 else if (isspace((int)*p))
265 else
266 return NULL;
268 if (i == 0 || (i % 2) != 0)
269 return NULL;
270 i /= 2;
272 if ((dst = malloc(i)) == NULL)
273 return NULL;
275 i = 0;
276 f = 0;
277 for (rp = dst, p = str; *p != '\0'; p++) {
278 if (isxdigit((int)*p)) {
279 if (!f) {
280 b[0] = *p;
281 f = 1;
282 } else {
283 b[1] = *p;
284 b[2] = '\0';
285 *rp++ = (char)strtol(b, NULL, base);
286 i++;
287 f = 0;
292 *len = i;
294 return(dst);
297 static void set_prime(BIGNUM *p, char *str)
299 size_t len = 0;
300 unsigned char *prime;
302 prime = (unsigned char *)str2val(str, 16, &len);
303 if (prime == NULL)
304 errx(1, "failed to parse %s", str);
305 BN_bin2bn(prime, len, p);
308 static void set_generator(BIGNUM *g)
310 BN_set_word(g, 2);
313 static void print_secret(unsigned char *sec, size_t len)
315 size_t i;
317 for (i = 0; i < len; ++i)
318 printf("%x", sec[i]);
320 printf("\n");
323 static int check_prime(struct prime *pr)
325 DH *dh1, *dh2;
326 BIGNUM *p, *g;
327 unsigned char *sec1, *sec2;
328 size_t size;
329 int ret;
331 if (verbose)
332 printf("Testing %s\n", pr->name);
334 p = BN_new();
335 g = BN_new();
336 dh1 = DH_new();
337 dh2 = DH_new();
339 /* 1. set shared parameter */
340 set_prime(p, pr->value);
341 set_generator(g);
342 dh1->p = BN_dup(p);
343 dh1->g = BN_dup(g);
344 dh2->p = BN_dup(p);
345 dh2->g = BN_dup(g);
347 /* 2. set keys */
348 ret = DH_generate_key(dh1);
349 if (ret == 0) {
350 fprintf(stderr, "DH_generate_key\n");
351 exit(EXIT_FAILURE);
353 ret = DH_generate_key(dh2);
354 if (ret == 0) {
355 fprintf(stderr, "DH_generate_key\n");
356 exit(EXIT_FAILURE);
359 /* 3. compute shared secret */
360 size = DH_size(dh1);
361 if (size != DH_size(dh2)) {
362 fprintf(stderr, "size does not match!\n");
363 exit(EXIT_FAILURE);
365 sec1 = malloc(size);
366 sec2 = malloc(size);
367 if (!sec1 || !sec2) {
368 perror("malloc");
369 exit(EXIT_FAILURE);
371 ret = DH_compute_key(sec1, dh2->pub_key, dh1);
372 if (ret == -1) {
373 fprintf(stderr, "DH_compute_key");
374 exit(EXIT_FAILURE);
376 ret = DH_compute_key(sec2, dh1->pub_key, dh2);
377 if (ret == -1) {
378 fprintf(stderr, "DH_compute_key");
379 exit(EXIT_FAILURE);
382 /* 4. compare shared secret */
383 if (verbose) {
384 printf("shared secret 1\n");
385 print_secret(sec1, size);
386 printf("shared secret 2\n");
387 print_secret(sec2, size);
390 if (memcmp(sec1, sec2, size) == 0)
391 ret = 1;
392 else
393 ret = 0;
395 free(sec2);
396 free(sec1);
397 DH_free(dh2);
398 DH_free(dh1);
399 BN_free(g);
400 BN_free(p);
402 return ret;
409 static void
410 usage (int ret)
412 arg_printusage (args,
413 sizeof(args)/sizeof(*args),
414 NULL,
415 "");
416 exit (ret);
420 main(int argc, char **argv)
422 int idx = 0;
424 setprogname(argv[0]);
426 if(getarg(args, sizeof(args) / sizeof(args[0]), argc, argv, &idx))
427 usage(1);
429 if (help_flag)
430 usage(0);
432 if(version_flag){
433 print_version(NULL);
434 exit(0);
437 argc -= idx;
438 argv += idx;
441 struct prime *p = primes;
443 for (; p->name; ++p)
444 if (check_prime(p))
445 printf("%s: shared secret OK\n", p->name);
446 else
447 printf("%s: shared secret FAILURE\n", p->name);
449 return 0;
452 return 0;