Add support for ECCCDH test format.
[openssl.git] / fips / ecdh / fips_ecdhvs.c
1 /* fips/ecdh/fips_ecdhvs.c */
2 /* Written by Dr Stephen N Henson (steve@openssl.org) for the OpenSSL
3  * project.
4  */
5 /* ====================================================================
6  * Copyright (c) 2011 The OpenSSL Project.  All rights reserved.
7  *
8  * Redistribution and use in source and binary forms, with or without
9  * modification, are permitted provided that the following conditions
10  * are met:
11  *
12  * 1. Redistributions of source code must retain the above copyright
13  *    notice, this list of conditions and the following disclaimer. 
14  *
15  * 2. Redistributions in binary form must reproduce the above copyright
16  *    notice, this list of conditions and the following disclaimer in
17  *    the documentation and/or other materials provided with the
18  *    distribution.
19  *
20  * 3. All advertising materials mentioning features or use of this
21  *    software must display the following acknowledgment:
22  *    "This product includes software developed by the OpenSSL Project
23  *    for use in the OpenSSL Toolkit. (http://www.OpenSSL.org/)"
24  *
25  * 4. The names "OpenSSL Toolkit" and "OpenSSL Project" must not be used to
26  *    endorse or promote products derived from this software without
27  *    prior written permission. For written permission, please contact
28  *    licensing@OpenSSL.org.
29  *
30  * 5. Products derived from this software may not be called "OpenSSL"
31  *    nor may "OpenSSL" appear in their names without prior written
32  *    permission of the OpenSSL Project.
33  *
34  * 6. Redistributions of any form whatsoever must retain the following
35  *    acknowledgment:
36  *    "This product includes software developed by the OpenSSL Project
37  *    for use in the OpenSSL Toolkit (http://www.OpenSSL.org/)"
38  *
39  * THIS SOFTWARE IS PROVIDED BY THE OpenSSL PROJECT ``AS IS'' AND ANY
40  * EXPRESSED OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
41  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
42  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE OpenSSL PROJECT OR
43  * ITS CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
44  * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
45  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
46  * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
47  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
48  * STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
49  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED
50  * OF THE POSSIBILITY OF SUCH DAMAGE.
51  * ====================================================================
52  */
53
54
55 #define OPENSSL_FIPSAPI
56 #include <openssl/opensslconf.h>
57
58 #ifndef OPENSSL_FIPS
59 #include <stdio.h>
60
61 int main(int argc, char **argv)
62 {
63     printf("No FIPS ECDH support\n");
64     return(0);
65 }
66 #else
67
68 #include <openssl/crypto.h>
69 #include <openssl/bn.h>
70 #include <openssl/ecdh.h>
71 #include <openssl/fips.h>
72 #include <openssl/err.h>
73 #include <openssl/evp.h>
74 #include <string.h>
75 #include <ctype.h>
76
77 #include "fips_utl.h"
78
79 static const EVP_MD *parse_md(char *line)
80         {
81         char *p;
82         if (line[0] != '[' || line[1] != 'E')
83                 return NULL;
84         p = strchr(line, '-');
85         if (!p)
86                 return NULL;
87         line = p + 1;
88         p = strchr(line, ']');
89         if (!p)
90                 return NULL;
91         *p = 0;
92         p = line;
93         while(isspace(*p))
94                 p++;
95         if (!strcmp(p, "SHA1"))
96                 return EVP_sha1();
97         else if (!strcmp(p, "SHA224"))
98                 return EVP_sha224();
99         else if (!strcmp(p, "SHA256"))
100                 return EVP_sha256();
101         else if (!strcmp(p, "SHA384"))
102                 return EVP_sha384();
103         else if (!strcmp(p, "SHA512"))
104                 return EVP_sha512();
105         else
106                 return NULL;
107         }
108
109 static int lookup_curve2(char *cname)
110         {
111         char *p;
112         p = strchr(cname, ']');
113         if (!p)
114                 {
115                 fprintf(stderr, "Parse error: missing ]\n");
116                 return NID_undef;
117                 }
118         *p = 0;
119
120         if (!strcmp(cname, "B-163"))
121                 return NID_sect163r2;
122         if (!strcmp(cname, "B-233"))
123                 return NID_sect233r1;
124         if (!strcmp(cname, "B-283"))
125                 return NID_sect283r1;
126         if (!strcmp(cname, "B-409"))
127                 return NID_sect409r1;
128         if (!strcmp(cname, "B-571"))
129                 return NID_sect571r1;
130         if (!strcmp(cname, "K-163"))
131                 return NID_sect163k1;
132         if (!strcmp(cname, "K-233"))
133                 return NID_sect233k1;
134         if (!strcmp(cname, "K-283"))
135                 return NID_sect283k1;
136         if (!strcmp(cname, "K-409"))
137                 return NID_sect409k1;
138         if (!strcmp(cname, "K-571"))
139                 return NID_sect571k1;
140         if (!strcmp(cname, "P-192"))
141                 return NID_X9_62_prime192v1;
142         if (!strcmp(cname, "P-224"))
143                 return NID_secp224r1;
144         if (!strcmp(cname, "P-256"))
145                 return NID_X9_62_prime256v1;
146         if (!strcmp(cname, "P-384"))
147                 return NID_secp384r1;
148         if (!strcmp(cname, "P-521"))
149                 return NID_secp521r1;
150
151         fprintf(stderr, "Unknown Curve name %s\n", cname);
152         return NID_undef;
153         }
154
155 static int lookup_curve(char *cname)
156         {
157         char *p;
158         p = strchr(cname, ':');
159         if (!p)
160                 {
161                 fprintf(stderr, "Parse error: missing :\n");
162                 return NID_undef;
163                 }
164         cname = p + 1;
165         while(isspace(*cname))
166                 cname++;
167         return lookup_curve2(cname);
168         }
169
170 static EC_POINT *make_peer(EC_GROUP *group, BIGNUM *x, BIGNUM *y)
171         {
172         EC_POINT *peer;
173         int rv;
174         BN_CTX *c;
175         peer = EC_POINT_new(group);
176         if (!peer)
177                 return NULL;
178         c = BN_CTX_new();
179         if (EC_METHOD_get_field_type(EC_GROUP_method_of(group))
180                 == NID_X9_62_prime_field)
181                 rv = EC_POINT_set_affine_coordinates_GFp(group, peer, x, y, c);
182         else
183 #ifdef OPENSSL_NO_EC2M
184                 {
185                 fprintf(stderr, "ERROR: GF2m not supported\n");
186                 exit(1);
187                 }
188 #else
189                 rv = EC_POINT_set_affine_coordinates_GF2m(group, peer, x, y, c);
190 #endif
191
192         BN_CTX_free(c);
193         if (rv)
194                 return peer;
195         EC_POINT_free(peer);
196         return NULL;
197         }
198
199 static int ec_print_pubkey(FILE *out, EC_KEY *key, int add_e)
200         {
201         const EC_POINT *pt;
202         const EC_GROUP *grp;
203         const EC_METHOD *meth;
204         int rv;
205         BIGNUM *tx, *ty;
206         BN_CTX *ctx;
207         ctx = BN_CTX_new();
208         if (!ctx)
209                 return 0;
210         tx = BN_CTX_get(ctx);
211         ty = BN_CTX_get(ctx);
212         if (!tx || !ty)
213                 return 0;
214         grp = EC_KEY_get0_group(key);
215         pt = EC_KEY_get0_public_key(key);
216         meth = EC_GROUP_method_of(grp);
217         if (EC_METHOD_get_field_type(meth) == NID_X9_62_prime_field)
218                 rv = EC_POINT_get_affine_coordinates_GFp(grp, pt, tx, ty, ctx);
219         else
220 #ifdef OPENSSL_NO_EC2M
221                 {
222                 fprintf(stderr, "ERROR: GF2m not supported\n");
223                 exit(1);
224                 }
225 #else
226                 rv = EC_POINT_get_affine_coordinates_GF2m(grp, pt, tx, ty, ctx);
227 #endif
228
229         if (add_e)
230                 {
231                 do_bn_print_name(out, "QeIUTx", tx);
232                 do_bn_print_name(out, "QeIUTy", ty);
233                 }
234         else
235                 {
236                 do_bn_print_name(out, "QIUTx", tx);
237                 do_bn_print_name(out, "QIUTy", ty);
238                 }
239
240         BN_CTX_free(ctx);
241
242         return rv;
243
244         }
245
246 static void ec_output_Zhash(FILE *out, int exout, EC_GROUP *group,
247                         BIGNUM *ix, BIGNUM *iy, BIGNUM *id, BIGNUM *cx,
248                         BIGNUM *cy, const EVP_MD *md,
249                                 unsigned char *rhash, size_t rhashlen)
250         {
251         EC_KEY *ec = NULL;
252         EC_POINT *peerkey = NULL;
253         unsigned char *Z;
254         unsigned char chash[EVP_MAX_MD_SIZE];
255         int Zlen;
256         ec = EC_KEY_new();
257         EC_KEY_set_group(ec, group);
258         peerkey = make_peer(group, cx, cy);
259         if (rhash == NULL)
260                 {
261                 if (md)
262                         rhashlen = M_EVP_MD_size(md);
263                 EC_KEY_generate_key(ec);
264                 ec_print_pubkey(out, ec, md ? 1 : 0);
265                 }
266         else
267                 {
268                 EC_KEY_set_public_key_affine_coordinates(ec, ix, iy);
269                 EC_KEY_set_private_key(ec, id);
270                 }
271         Zlen = (EC_GROUP_get_degree(group) + 7)/8;
272         Z = OPENSSL_malloc(Zlen);
273         if (!Z)
274                 exit(1);
275         ECDH_compute_key(Z, Zlen, peerkey, ec, 0);
276         if (md)
277                 {
278                 if (exout)
279                         OutputValue("Z", Z, Zlen, out, 0);
280                 FIPS_digest(Z, Zlen, chash, NULL, md);
281                 OutputValue(rhash ? "IUTHashZZ" : "HashZZ",
282                                                 chash, rhashlen, out, 0);
283                 if (rhash)
284                         {
285                         fprintf(out, "Result = %s\n",
286                                 memcmp(chash, rhash, rhashlen) ? "F" : "P");
287                         }
288                 }
289         else
290                 OutputValue("ZIUT", Z, Zlen, out, 0);
291         OPENSSL_cleanse(Z, Zlen);
292         OPENSSL_free(Z);
293         EC_KEY_free(ec);
294         EC_POINT_free(peerkey);
295         }
296                 
297 int main(int argc,char **argv)
298         {
299         char **args = argv + 1;
300         int argn = argc - 1;
301         FILE *in, *out;
302         char buf[2048], lbuf[2048];
303         unsigned char *rhash = NULL;
304         long rhashlen;
305         BIGNUM *cx = NULL, *cy = NULL;
306         BIGNUM *id = NULL, *ix = NULL, *iy = NULL;
307         const EVP_MD *md = NULL;
308         EC_GROUP *group = NULL;
309         char *keyword = NULL, *value = NULL;
310         int do_verify = -1, exout = 0;
311
312         int curve_nids[5] = {0,0,0,0,0};
313         int param_set = -1;
314
315         fips_algtest_init();
316
317         if (argn && !strcmp(*args, "ecdhver"))
318                 {
319                 do_verify = 1;
320                 args++;
321                 argn--;
322                 }
323         else if (argn && !strcmp(*args, "ecdhgen"))
324                 {
325                 do_verify = 0;
326                 args++;
327                 argn--;
328                 }
329
330         if (argn && !strcmp(*args, "-exout"))
331                 {
332                 exout = 1;
333                 args++;
334                 argn--;
335                 }
336
337         if (do_verify == -1)
338                 {
339                 fprintf(stderr,"%s [ecdhver|ecdhgen|] [-exout] (infile outfile)\n",argv[0]);
340                 exit(1);
341                 }
342
343         if (argn == 2)
344                 {
345                 in = fopen(*args, "r");
346                 if (!in)
347                         {
348                         fprintf(stderr, "Error opening input file\n");
349                         exit(1);
350                         }
351                 out = fopen(args[1], "w");
352                 if (!out)
353                         {
354                         fprintf(stderr, "Error opening output file\n");
355                         exit(1);
356                         }
357                 }
358         else if (argn == 0)
359                 {
360                 in = stdin;
361                 out = stdout;
362                 }
363         else
364                 {
365                 fprintf(stderr,"%s [dhver|dhgen|] [-exout] (infile outfile)\n",argv[0]);
366                 exit(1);
367                 }
368
369         while (fgets(buf, sizeof(buf), in) != NULL)
370                 {
371                 fputs(buf, out);
372                 if (buf[0] == '[' && buf[1] == 'E')
373                         {
374                         int c = buf[2];
375                         if (c < 'A' || c > 'E')
376                                 goto parse_error;
377                         param_set = c - 'A';
378                         /* If just [E?] then initial paramset */
379                         if (buf[3] == ']')
380                                 continue;
381                         if (group)
382                                 EC_GROUP_free(group);
383                         group = EC_GROUP_new_by_curve_name(curve_nids[c - 'A']);
384                         }
385                 if (strlen(buf) > 10 && !strncmp(buf, "[Curve", 6))
386                         {
387                         int nid;
388                         if (param_set == -1)
389                                 goto parse_error;
390                         nid = lookup_curve(buf);
391                         if (nid == NID_undef)
392                                 goto parse_error;
393                         curve_nids[param_set] = nid;
394                         }
395
396                 if (strlen(buf) > 4 && buf[0] == '[' && buf[2] == '-')
397                         {
398                         int nid = lookup_curve2(buf + 1);
399                         if (nid == NID_undef)
400                                 goto parse_error;
401                         if (group)
402                                 EC_GROUP_free(group);
403                         group = EC_GROUP_new_by_curve_name(nid);
404                         }
405
406                 if (strlen(buf) > 6 && !strncmp(buf, "[E", 2))
407                         {
408                         md = parse_md(buf);
409                         if (md == NULL)
410                                 goto parse_error;
411                         continue;
412                         }
413                 if (!parse_line(&keyword, &value, lbuf, buf))
414                         continue;
415                 if (!strcmp(keyword, "QeCAVSx") || !strcmp(keyword, "QCAVSx"))
416                         {
417                         if (!do_hex2bn(&cx, value))
418                                 goto parse_error;
419                         }
420                 else if (!strcmp(keyword, "QeCAVSy") || !strcmp(keyword, "QCAVSy"))
421                         {
422                         if (!do_hex2bn(&cy, value))
423                                 goto parse_error;
424                         if (do_verify == 0)
425                                 ec_output_Zhash(out, exout, group,
426                                                 NULL, NULL, NULL,
427                                                 cx, cy, md, rhash, rhashlen);
428                         }
429                 else if (!strcmp(keyword, "deIUT"))
430                         {
431                         if (!do_hex2bn(&id, value))
432                                 goto parse_error;
433                         }
434                 else if (!strcmp(keyword, "QeIUTx"))
435                         {
436                         if (!do_hex2bn(&ix, value))
437                                 goto parse_error;
438                         }
439                 else if (!strcmp(keyword, "QeIUTy"))
440                         {
441                         if (!do_hex2bn(&iy, value))
442                                 goto parse_error;
443                         }
444                 else if (!strcmp(keyword, "CAVSHashZZ"))
445                         {
446                         if (!md)
447                                 goto parse_error;
448                         rhash = hex2bin_m(value, &rhashlen);
449                         if (!rhash || rhashlen != M_EVP_MD_size(md))
450                                 goto parse_error;
451                         ec_output_Zhash(out, exout, group, ix, iy, id, cx, cy,
452                                         md, rhash, rhashlen);
453                         }
454                 }
455         return 0;
456         parse_error:
457         fprintf(stderr, "Error Parsing request file\n");
458         exit(1);
459         }
460
461 #endif