testutil/init.c rename to testutil/testutil_init.c
[openssl.git] / apps / pkeyutl.c
1 /*
2  * Copyright 2006-2018 The OpenSSL Project Authors. All Rights Reserved.
3  *
4  * Licensed under the Apache License 2.0 (the "License").  You may not use
5  * this file except in compliance with the License.  You can obtain a copy
6  * in the file LICENSE in the source distribution or at
7  * https://www.openssl.org/source/license.html
8  */
9
10 #include "apps.h"
11 #include "progs.h"
12 #include <string.h>
13 #include <openssl/err.h>
14 #include <openssl/pem.h>
15 #include <openssl/evp.h>
16 #include <sys/stat.h>
17
18 #define KEY_NONE        0
19 #define KEY_PRIVKEY     1
20 #define KEY_PUBKEY      2
21 #define KEY_CERT        3
22
23 static EVP_PKEY_CTX *init_ctx(const char *kdfalg, int *pkeysize,
24                               const char *keyfile, int keyform, int key_type,
25                               char *passinarg, int pkey_op, ENGINE *e,
26                               const int impl, int rawin, EVP_PKEY **ppkey);
27
28 static int setup_peer(EVP_PKEY_CTX *ctx, int peerform, const char *file,
29                       ENGINE *e);
30
31 static int do_keyop(EVP_PKEY_CTX *ctx, int pkey_op,
32                     unsigned char *out, size_t *poutlen,
33                     const unsigned char *in, size_t inlen);
34
35 static int do_raw_keyop(int pkey_op, EVP_PKEY_CTX *ctx,
36                         const EVP_MD *md, EVP_PKEY *pkey, BIO *in,
37                         int filesize, unsigned char *sig, int siglen,
38                         unsigned char **out, size_t *poutlen);
39
40 typedef enum OPTION_choice {
41     OPT_ERR = -1, OPT_EOF = 0, OPT_HELP,
42     OPT_ENGINE, OPT_ENGINE_IMPL, OPT_IN, OPT_OUT,
43     OPT_PUBIN, OPT_CERTIN, OPT_ASN1PARSE, OPT_HEXDUMP, OPT_SIGN,
44     OPT_VERIFY, OPT_VERIFYRECOVER, OPT_REV, OPT_ENCRYPT, OPT_DECRYPT,
45     OPT_DERIVE, OPT_SIGFILE, OPT_INKEY, OPT_PEERKEY, OPT_PASSIN,
46     OPT_PEERFORM, OPT_KEYFORM, OPT_PKEYOPT, OPT_PKEYOPT_PASSIN, OPT_KDF,
47     OPT_KDFLEN, OPT_R_ENUM,
48     OPT_RAWIN, OPT_DIGEST
49 } OPTION_CHOICE;
50
51 const OPTIONS pkeyutl_options[] = {
52     {"help", OPT_HELP, '-', "Display this summary"},
53     {"in", OPT_IN, '<', "Input file - default stdin"},
54     {"rawin", OPT_RAWIN, '-', "Indicate the input data is in raw form"},
55     {"digest", OPT_DIGEST, 's',
56      "Specify the digest algorithm when signing the raw input data"},
57     {"out", OPT_OUT, '>', "Output file - default stdout"},
58     {"pubin", OPT_PUBIN, '-', "Input is a public key"},
59     {"certin", OPT_CERTIN, '-', "Input is a cert with a public key"},
60     {"asn1parse", OPT_ASN1PARSE, '-', "asn1parse the output data"},
61     {"hexdump", OPT_HEXDUMP, '-', "Hex dump output"},
62     {"sign", OPT_SIGN, '-', "Sign input data with private key"},
63     {"verify", OPT_VERIFY, '-', "Verify with public key"},
64     {"verifyrecover", OPT_VERIFYRECOVER, '-',
65      "Verify with public key, recover original data"},
66     {"rev", OPT_REV, '-', "Reverse the order of the input buffer"},
67     {"encrypt", OPT_ENCRYPT, '-', "Encrypt input data with public key"},
68     {"decrypt", OPT_DECRYPT, '-', "Decrypt input data with private key"},
69     {"derive", OPT_DERIVE, '-', "Derive shared secret"},
70     {"kdf", OPT_KDF, 's', "Use KDF algorithm"},
71     {"kdflen", OPT_KDFLEN, 'p', "KDF algorithm output length"},
72     {"sigfile", OPT_SIGFILE, '<', "Signature file (verify operation only)"},
73     {"inkey", OPT_INKEY, 's', "Input private key file"},
74     {"peerkey", OPT_PEERKEY, 's', "Peer key file used in key derivation"},
75     {"passin", OPT_PASSIN, 's', "Input file pass phrase source"},
76     {"peerform", OPT_PEERFORM, 'E', "Peer key format - default PEM"},
77     {"keyform", OPT_KEYFORM, 'E', "Private key format - default PEM"},
78     {"pkeyopt", OPT_PKEYOPT, 's', "Public key options as opt:value"},
79     {"pkeyopt_passin", OPT_PKEYOPT_PASSIN, 's',
80      "Public key option that is read as a passphrase argument opt:passphrase"},
81     OPT_R_OPTIONS,
82 #ifndef OPENSSL_NO_ENGINE
83     {"engine", OPT_ENGINE, 's', "Use engine, possibly a hardware device"},
84     {"engine_impl", OPT_ENGINE_IMPL, '-',
85      "Also use engine given by -engine for crypto operations"},
86 #endif
87     {NULL}
88 };
89
90 int pkeyutl_main(int argc, char **argv)
91 {
92     BIO *in = NULL, *out = NULL;
93     ENGINE *e = NULL;
94     EVP_PKEY_CTX *ctx = NULL;
95     EVP_PKEY *pkey = NULL;
96     char *infile = NULL, *outfile = NULL, *sigfile = NULL, *passinarg = NULL;
97     char hexdump = 0, asn1parse = 0, rev = 0, *prog;
98     unsigned char *buf_in = NULL, *buf_out = NULL, *sig = NULL;
99     OPTION_CHOICE o;
100     int buf_inlen = 0, siglen = -1, keyform = FORMAT_PEM, peerform = FORMAT_PEM;
101     int keysize = -1, pkey_op = EVP_PKEY_OP_SIGN, key_type = KEY_PRIVKEY;
102     int engine_impl = 0;
103     int ret = 1, rv = -1;
104     size_t buf_outlen;
105     const char *inkey = NULL;
106     const char *peerkey = NULL;
107     const char *kdfalg = NULL;
108     int kdflen = 0;
109     STACK_OF(OPENSSL_STRING) *pkeyopts = NULL;
110     STACK_OF(OPENSSL_STRING) *pkeyopts_passin = NULL;
111     int rawin = 0;
112     const EVP_MD *md = NULL;
113     int filesize = -1;
114
115     prog = opt_init(argc, argv, pkeyutl_options);
116     while ((o = opt_next()) != OPT_EOF) {
117         switch (o) {
118         case OPT_EOF:
119         case OPT_ERR:
120  opthelp:
121             BIO_printf(bio_err, "%s: Use -help for summary.\n", prog);
122             goto end;
123         case OPT_HELP:
124             opt_help(pkeyutl_options);
125             ret = 0;
126             goto end;
127         case OPT_IN:
128             infile = opt_arg();
129             break;
130         case OPT_OUT:
131             outfile = opt_arg();
132             break;
133         case OPT_SIGFILE:
134             sigfile = opt_arg();
135             break;
136         case OPT_ENGINE_IMPL:
137             engine_impl = 1;
138             break;
139         case OPT_INKEY:
140             inkey = opt_arg();
141             break;
142         case OPT_PEERKEY:
143             peerkey = opt_arg();
144             break;
145         case OPT_PASSIN:
146             passinarg = opt_arg();
147             break;
148         case OPT_PEERFORM:
149             if (!opt_format(opt_arg(), OPT_FMT_PDE, &peerform))
150                 goto opthelp;
151             break;
152         case OPT_KEYFORM:
153             if (!opt_format(opt_arg(), OPT_FMT_PDE, &keyform))
154                 goto opthelp;
155             break;
156         case OPT_R_CASES:
157             if (!opt_rand(o))
158                 goto end;
159             break;
160         case OPT_ENGINE:
161             e = setup_engine(opt_arg(), 0);
162             break;
163         case OPT_PUBIN:
164             key_type = KEY_PUBKEY;
165             break;
166         case OPT_CERTIN:
167             key_type = KEY_CERT;
168             break;
169         case OPT_ASN1PARSE:
170             asn1parse = 1;
171             break;
172         case OPT_HEXDUMP:
173             hexdump = 1;
174             break;
175         case OPT_SIGN:
176             pkey_op = EVP_PKEY_OP_SIGN;
177             break;
178         case OPT_VERIFY:
179             pkey_op = EVP_PKEY_OP_VERIFY;
180             break;
181         case OPT_VERIFYRECOVER:
182             pkey_op = EVP_PKEY_OP_VERIFYRECOVER;
183             break;
184         case OPT_ENCRYPT:
185             pkey_op = EVP_PKEY_OP_ENCRYPT;
186             break;
187         case OPT_DECRYPT:
188             pkey_op = EVP_PKEY_OP_DECRYPT;
189             break;
190         case OPT_DERIVE:
191             pkey_op = EVP_PKEY_OP_DERIVE;
192             break;
193         case OPT_KDF:
194             pkey_op = EVP_PKEY_OP_DERIVE;
195             key_type = KEY_NONE;
196             kdfalg = opt_arg();
197             break;
198         case OPT_KDFLEN:
199             kdflen = atoi(opt_arg());
200             break;
201         case OPT_REV:
202             rev = 1;
203             break;
204         case OPT_PKEYOPT:
205             if ((pkeyopts == NULL &&
206                  (pkeyopts = sk_OPENSSL_STRING_new_null()) == NULL) ||
207                 sk_OPENSSL_STRING_push(pkeyopts, opt_arg()) == 0) {
208                 BIO_puts(bio_err, "out of memory\n");
209                 goto end;
210             }
211             break;
212         case OPT_PKEYOPT_PASSIN:
213             if ((pkeyopts_passin == NULL &&
214                  (pkeyopts_passin = sk_OPENSSL_STRING_new_null()) == NULL) ||
215                 sk_OPENSSL_STRING_push(pkeyopts_passin, opt_arg()) == 0) {
216                 BIO_puts(bio_err, "out of memory\n");
217                 goto end;
218             }
219             break;
220         case OPT_RAWIN:
221             rawin = 1;
222             break;
223         case OPT_DIGEST:
224             if (!opt_md(opt_arg(), &md))
225                 goto end;
226             break;
227         }
228     }
229     argc = opt_num_rest();
230     if (argc != 0)
231         goto opthelp;
232
233     if (rawin && pkey_op != EVP_PKEY_OP_SIGN && pkey_op != EVP_PKEY_OP_VERIFY) {
234         BIO_printf(bio_err,
235                    "%s: -rawin can only be used with -sign or -verify\n",
236                    prog);
237         goto opthelp;
238     }
239
240     if (md != NULL && !rawin) {
241         BIO_printf(bio_err,
242                    "%s: -digest can only be used with -rawin\n",
243                    prog);
244         goto opthelp;
245     }
246
247     if (rawin && rev) {
248         BIO_printf(bio_err, "%s: -rev cannot be used with raw input\n",
249                    prog);
250         goto opthelp;
251     }
252
253     if (kdfalg != NULL) {
254         if (kdflen == 0) {
255             BIO_printf(bio_err,
256                        "%s: no KDF length given (-kdflen parameter).\n", prog);
257             goto opthelp;
258         }
259     } else if (inkey == NULL) {
260         BIO_printf(bio_err,
261                    "%s: no private key given (-inkey parameter).\n", prog);
262         goto opthelp;
263     } else if (peerkey != NULL && pkey_op != EVP_PKEY_OP_DERIVE) {
264         BIO_printf(bio_err,
265                    "%s: no peer key given (-peerkey parameter).\n", prog);
266         goto opthelp;
267     }
268     ctx = init_ctx(kdfalg, &keysize, inkey, keyform, key_type,
269                    passinarg, pkey_op, e, engine_impl, rawin, &pkey);
270     if (ctx == NULL) {
271         BIO_printf(bio_err, "%s: Error initializing context\n", prog);
272         ERR_print_errors(bio_err);
273         goto end;
274     }
275     if (peerkey != NULL && !setup_peer(ctx, peerform, peerkey, e)) {
276         BIO_printf(bio_err, "%s: Error setting up peer key\n", prog);
277         ERR_print_errors(bio_err);
278         goto end;
279     }
280     if (pkeyopts != NULL) {
281         int num = sk_OPENSSL_STRING_num(pkeyopts);
282         int i;
283
284         for (i = 0; i < num; ++i) {
285             const char *opt = sk_OPENSSL_STRING_value(pkeyopts, i);
286
287             if (pkey_ctrl_string(ctx, opt) <= 0) {
288                 BIO_printf(bio_err, "%s: Can't set parameter \"%s\":\n",
289                            prog, opt);
290                 ERR_print_errors(bio_err);
291                 goto end;
292             }
293         }
294     }
295     if (pkeyopts_passin != NULL) {
296         int num = sk_OPENSSL_STRING_num(pkeyopts_passin);
297         int i;
298
299         for (i = 0; i < num; i++) {
300             char *opt = sk_OPENSSL_STRING_value(pkeyopts_passin, i);
301             char *passin = strchr(opt, ':');
302             char *passwd;
303
304             if (passin == NULL) {
305                 /* Get password interactively */
306                 char passwd_buf[4096];
307                 BIO_snprintf(passwd_buf, sizeof(passwd_buf), "Enter %s: ", opt);
308                 EVP_read_pw_string(passwd_buf, sizeof(passwd_buf) - 1,
309                                    passwd_buf, 0);
310                 passwd = OPENSSL_strdup(passwd_buf);
311                 if (passwd == NULL) {
312                     BIO_puts(bio_err, "out of memory\n");
313                     goto end;
314                 }
315             } else {
316                 /* Get password as a passin argument: First split option name
317                  * and passphrase argument into two strings */
318                 *passin = 0;
319                 passin++;
320                 if (app_passwd(passin, NULL, &passwd, NULL) == 0) {
321                     BIO_printf(bio_err, "failed to get '%s'\n", opt);
322                     goto end;
323                 }
324             }
325
326             if (EVP_PKEY_CTX_ctrl_str(ctx, opt, passwd) <= 0) {
327                 BIO_printf(bio_err, "%s: Can't set parameter \"%s\":\n",
328                            prog, opt);
329                 goto end;
330             }
331             OPENSSL_free(passwd);
332         }
333     }
334
335     if (sigfile != NULL && (pkey_op != EVP_PKEY_OP_VERIFY)) {
336         BIO_printf(bio_err,
337                    "%s: Signature file specified for non verify\n", prog);
338         goto end;
339     }
340
341     if (sigfile == NULL && (pkey_op == EVP_PKEY_OP_VERIFY)) {
342         BIO_printf(bio_err,
343                    "%s: No signature file specified for verify\n", prog);
344         goto end;
345     }
346
347     if (pkey_op != EVP_PKEY_OP_DERIVE) {
348         in = bio_open_default(infile, 'r', FORMAT_BINARY);
349         if (infile != NULL) {
350             struct stat st;
351
352             if (stat(infile, &st) == 0 && st.st_size <= INT_MAX)
353                 filesize = (int)st.st_size;
354         }
355         if (in == NULL)
356             goto end;
357     }
358     out = bio_open_default(outfile, 'w', FORMAT_BINARY);
359     if (out == NULL)
360         goto end;
361
362     if (sigfile != NULL) {
363         BIO *sigbio = BIO_new_file(sigfile, "rb");
364
365         if (sigbio == NULL) {
366             BIO_printf(bio_err, "Can't open signature file %s\n", sigfile);
367             goto end;
368         }
369         siglen = bio_to_mem(&sig, keysize * 10, sigbio);
370         BIO_free(sigbio);
371         if (siglen < 0) {
372             BIO_printf(bio_err, "Error reading signature data\n");
373             goto end;
374         }
375     }
376
377     /* Raw input data is handled elsewhere */
378     if (in != NULL && !rawin) {
379         /* Read the input data */
380         buf_inlen = bio_to_mem(&buf_in, keysize * 10, in);
381         if (buf_inlen < 0) {
382             BIO_printf(bio_err, "Error reading input Data\n");
383             goto end;
384         }
385         if (rev) {
386             size_t i;
387             unsigned char ctmp;
388             size_t l = (size_t)buf_inlen;
389             for (i = 0; i < l / 2; i++) {
390                 ctmp = buf_in[i];
391                 buf_in[i] = buf_in[l - 1 - i];
392                 buf_in[l - 1 - i] = ctmp;
393             }
394         }
395     }
396
397     /* Sanity check the input if the input is not raw */
398     if (!rawin
399             && buf_inlen > EVP_MAX_MD_SIZE
400             && (pkey_op == EVP_PKEY_OP_SIGN
401                 || pkey_op == EVP_PKEY_OP_VERIFY)) {
402         BIO_printf(bio_err,
403                    "Error: The input data looks too long to be a hash\n");
404         goto end;
405     }
406
407     if (pkey_op == EVP_PKEY_OP_VERIFY) {
408         if (rawin) {
409             rv = do_raw_keyop(pkey_op, ctx, md, pkey, in, filesize, sig, siglen,
410                               NULL, 0);
411         } else {
412             rv = EVP_PKEY_verify(ctx, sig, (size_t)siglen,
413                                  buf_in, (size_t)buf_inlen);
414         }
415         if (rv == 1) {
416             BIO_puts(out, "Signature Verified Successfully\n");
417             ret = 0;
418         } else {
419             BIO_puts(out, "Signature Verification Failure\n");
420         }
421         goto end;
422     }
423     if (kdflen != 0) {
424         buf_outlen = kdflen;
425         rv = 1;
426     } else {
427         if (rawin) {
428             /* rawin allocates the buffer in do_raw_keyop() */
429             rv = do_raw_keyop(pkey_op, ctx, md, pkey, in, filesize, NULL, 0,
430                               &buf_out, (size_t *)&buf_outlen);
431         } else {
432             rv = do_keyop(ctx, pkey_op, NULL, (size_t *)&buf_outlen,
433                           buf_in, (size_t)buf_inlen);
434             if (rv > 0 && buf_outlen != 0) {
435                 buf_out = app_malloc(buf_outlen, "buffer output");
436                 rv = do_keyop(ctx, pkey_op,
437                               buf_out, (size_t *)&buf_outlen,
438                               buf_in, (size_t)buf_inlen);
439             }
440         }
441     }
442     if (rv <= 0) {
443         if (pkey_op != EVP_PKEY_OP_DERIVE) {
444             BIO_puts(bio_err, "Public Key operation error\n");
445         } else {
446             BIO_puts(bio_err, "Key derivation failed\n");
447         }
448         ERR_print_errors(bio_err);
449         goto end;
450     }
451     ret = 0;
452
453     if (asn1parse) {
454         if (!ASN1_parse_dump(out, buf_out, buf_outlen, 1, -1))
455             ERR_print_errors(bio_err);
456     } else if (hexdump) {
457         BIO_dump(out, (char *)buf_out, buf_outlen);
458     } else {
459         BIO_write(out, buf_out, buf_outlen);
460     }
461
462  end:
463     EVP_PKEY_CTX_free(ctx);
464     release_engine(e);
465     BIO_free(in);
466     BIO_free_all(out);
467     OPENSSL_free(buf_in);
468     OPENSSL_free(buf_out);
469     OPENSSL_free(sig);
470     sk_OPENSSL_STRING_free(pkeyopts);
471     sk_OPENSSL_STRING_free(pkeyopts_passin);
472     return ret;
473 }
474
475 static EVP_PKEY_CTX *init_ctx(const char *kdfalg, int *pkeysize,
476                               const char *keyfile, int keyform, int key_type,
477                               char *passinarg, int pkey_op, ENGINE *e,
478                               const int engine_impl, int rawin,
479                               EVP_PKEY **ppkey)
480 {
481     EVP_PKEY *pkey = NULL;
482     EVP_PKEY_CTX *ctx = NULL;
483     ENGINE *impl = NULL;
484     char *passin = NULL;
485     int rv = -1;
486     X509 *x;
487     if (((pkey_op == EVP_PKEY_OP_SIGN) || (pkey_op == EVP_PKEY_OP_DECRYPT)
488          || (pkey_op == EVP_PKEY_OP_DERIVE))
489         && (key_type != KEY_PRIVKEY && kdfalg == NULL)) {
490         BIO_printf(bio_err, "A private key is needed for this operation\n");
491         goto end;
492     }
493     if (!app_passwd(passinarg, NULL, &passin, NULL)) {
494         BIO_printf(bio_err, "Error getting password\n");
495         goto end;
496     }
497     switch (key_type) {
498     case KEY_PRIVKEY:
499         pkey = load_key(keyfile, keyform, 0, passin, e, "Private Key");
500         break;
501
502     case KEY_PUBKEY:
503         pkey = load_pubkey(keyfile, keyform, 0, NULL, e, "Public Key");
504         break;
505
506     case KEY_CERT:
507         x = load_cert(keyfile, keyform, "Certificate");
508         if (x) {
509             pkey = X509_get_pubkey(x);
510             X509_free(x);
511         }
512         break;
513
514     case KEY_NONE:
515         break;
516
517     }
518
519 #ifndef OPENSSL_NO_ENGINE
520     if (engine_impl)
521         impl = e;
522 #endif
523
524     if (kdfalg != NULL) {
525         int kdfnid = OBJ_sn2nid(kdfalg);
526
527         if (kdfnid == NID_undef) {
528             kdfnid = OBJ_ln2nid(kdfalg);
529             if (kdfnid == NID_undef) {
530                 BIO_printf(bio_err, "The given KDF \"%s\" is unknown.\n",
531                            kdfalg);
532                 goto end;
533             }
534         }
535         ctx = EVP_PKEY_CTX_new_id(kdfnid, impl);
536     } else {
537         if (pkey == NULL)
538             goto end;
539
540 #ifndef OPENSSL_NO_EC
541         /* SM2 needs a special treatment */
542         if (EVP_PKEY_id(pkey) == EVP_PKEY_EC) {
543             EC_KEY *eckey = NULL;
544             const EC_GROUP *group = NULL;
545             int nid;
546
547             if ((eckey = EVP_PKEY_get0_EC_KEY(pkey)) == NULL
548                     || (group = EC_KEY_get0_group(eckey)) == NULL
549                     || (nid = EC_GROUP_get_curve_name(group)) == 0)
550                 goto end;
551             if (nid == NID_sm2)
552                 EVP_PKEY_set_alias_type(pkey, EVP_PKEY_SM2);
553         }
554 #endif
555         *pkeysize = EVP_PKEY_size(pkey);
556         ctx = EVP_PKEY_CTX_new(pkey, impl);
557         if (ppkey != NULL)
558             *ppkey = pkey;
559         EVP_PKEY_free(pkey);
560     }
561
562     if (ctx == NULL)
563         goto end;
564
565     /*
566      * If rawin then we don't need to actually initialise the EVP_PKEY_CTX
567      * itself. That will get initialised during EVP_DigestSignInit or
568      * EVP_DigestVerifyInit.
569      */
570     if (rawin) {
571         rv = 1;
572     } else {
573         switch (pkey_op) {
574         case EVP_PKEY_OP_SIGN:
575             rv = EVP_PKEY_sign_init(ctx);
576             break;
577
578         case EVP_PKEY_OP_VERIFY:
579             rv = EVP_PKEY_verify_init(ctx);
580             break;
581
582         case EVP_PKEY_OP_VERIFYRECOVER:
583             rv = EVP_PKEY_verify_recover_init(ctx);
584             break;
585
586         case EVP_PKEY_OP_ENCRYPT:
587             rv = EVP_PKEY_encrypt_init(ctx);
588             break;
589
590         case EVP_PKEY_OP_DECRYPT:
591             rv = EVP_PKEY_decrypt_init(ctx);
592             break;
593
594         case EVP_PKEY_OP_DERIVE:
595             rv = EVP_PKEY_derive_init(ctx);
596             break;
597         }
598     }
599
600     if (rv <= 0) {
601         EVP_PKEY_CTX_free(ctx);
602         ctx = NULL;
603     }
604
605  end:
606     OPENSSL_free(passin);
607     return ctx;
608
609 }
610
611 static int setup_peer(EVP_PKEY_CTX *ctx, int peerform, const char *file,
612                       ENGINE *e)
613 {
614     EVP_PKEY *peer = NULL;
615     ENGINE *engine = NULL;
616     int ret;
617
618     if (peerform == FORMAT_ENGINE)
619         engine = e;
620     peer = load_pubkey(file, peerform, 0, NULL, engine, "Peer Key");
621     if (peer == NULL) {
622         BIO_printf(bio_err, "Error reading peer key %s\n", file);
623         ERR_print_errors(bio_err);
624         return 0;
625     }
626
627     ret = EVP_PKEY_derive_set_peer(ctx, peer);
628
629     EVP_PKEY_free(peer);
630     if (ret <= 0)
631         ERR_print_errors(bio_err);
632     return ret;
633 }
634
635 static int do_keyop(EVP_PKEY_CTX *ctx, int pkey_op,
636                     unsigned char *out, size_t *poutlen,
637                     const unsigned char *in, size_t inlen)
638 {
639     int rv = 0;
640     switch (pkey_op) {
641     case EVP_PKEY_OP_VERIFYRECOVER:
642         rv = EVP_PKEY_verify_recover(ctx, out, poutlen, in, inlen);
643         break;
644
645     case EVP_PKEY_OP_SIGN:
646         rv = EVP_PKEY_sign(ctx, out, poutlen, in, inlen);
647         break;
648
649     case EVP_PKEY_OP_ENCRYPT:
650         rv = EVP_PKEY_encrypt(ctx, out, poutlen, in, inlen);
651         break;
652
653     case EVP_PKEY_OP_DECRYPT:
654         rv = EVP_PKEY_decrypt(ctx, out, poutlen, in, inlen);
655         break;
656
657     case EVP_PKEY_OP_DERIVE:
658         rv = EVP_PKEY_derive(ctx, out, poutlen);
659         break;
660
661     }
662     return rv;
663 }
664
665 #define TBUF_MAXSIZE 2048
666
667 static int do_raw_keyop(int pkey_op, EVP_PKEY_CTX *ctx,
668                         const EVP_MD *md, EVP_PKEY *pkey, BIO *in,
669                         int filesize, unsigned char *sig, int siglen,
670                         unsigned char **out, size_t *poutlen)
671 {
672     int rv = 0;
673     EVP_MD_CTX *mctx = NULL;
674     unsigned char tbuf[TBUF_MAXSIZE];
675     unsigned char *mbuf = NULL;
676     int buf_len = 0;
677
678     if ((mctx = EVP_MD_CTX_new()) == NULL) {
679         BIO_printf(bio_err, "Error: out of memory\n");
680         return rv;
681     }
682     EVP_MD_CTX_set_pkey_ctx(mctx, ctx);
683
684     /* Some algorithms only support oneshot digests */
685     if (EVP_PKEY_id(pkey) == EVP_PKEY_ED25519
686             || EVP_PKEY_id(pkey) == EVP_PKEY_ED448) {
687         if (filesize < 0) {
688             BIO_printf(bio_err,
689                        "Error: unable to determine file size for oneshot operation\n");
690             goto end;
691         }
692         mbuf = app_malloc(filesize, "oneshot sign/verify buffer");
693         switch(pkey_op) {
694         case EVP_PKEY_OP_VERIFY:
695             if (EVP_DigestVerifyInit(mctx, NULL, md, NULL, pkey) != 1)
696                 goto end;
697             buf_len = BIO_read(in, mbuf, filesize);
698             if (buf_len != filesize) {
699                 BIO_printf(bio_err, "Error reading raw input data\n");
700                 goto end;
701             }
702             rv = EVP_DigestVerify(mctx, sig, (size_t)siglen, mbuf, buf_len);
703             break;
704         case EVP_PKEY_OP_SIGN:
705             if (EVP_DigestSignInit(mctx, NULL, md, NULL, pkey) != 1)
706                 goto end;
707             buf_len = BIO_read(in, mbuf, filesize);
708             if (buf_len != filesize) {
709                 BIO_printf(bio_err, "Error reading raw input data\n");
710                 goto end;
711             }
712             rv = EVP_DigestSign(mctx, NULL, poutlen, mbuf, buf_len);
713             if (rv == 1 && out != NULL) {
714                 *out = app_malloc(*poutlen, "buffer output");
715                 rv = EVP_DigestSign(mctx, *out, poutlen, mbuf, buf_len);
716             }
717             break;
718         }
719         goto end;
720     }
721
722     switch(pkey_op) {
723     case EVP_PKEY_OP_VERIFY:
724         if (EVP_DigestVerifyInit(mctx, NULL, md, NULL, pkey) != 1)
725             goto end;
726         for (;;) {
727             buf_len = BIO_read(in, tbuf, TBUF_MAXSIZE);
728             if (buf_len == 0)
729                 break;
730             if (buf_len < 0) {
731                 BIO_printf(bio_err, "Error reading raw input data\n");
732                 goto end;
733             }
734             rv = EVP_DigestVerifyUpdate(mctx, tbuf, (size_t)buf_len);
735             if (rv != 1) {
736                 BIO_printf(bio_err, "Error verifying raw input data\n");
737                 goto end;
738             }
739         }
740         rv = EVP_DigestVerifyFinal(mctx, sig, (size_t)siglen);
741         break;
742     case EVP_PKEY_OP_SIGN:
743         if (EVP_DigestSignInit(mctx, NULL, md, NULL, pkey) != 1)
744             goto end;
745         for (;;) {
746             buf_len = BIO_read(in, tbuf, TBUF_MAXSIZE);
747             if (buf_len == 0)
748                 break;
749             if (buf_len < 0) {
750                 BIO_printf(bio_err, "Error reading raw input data\n");
751                 goto end;
752             }
753             rv = EVP_DigestSignUpdate(mctx, tbuf, (size_t)buf_len);
754             if (rv != 1) {
755                 BIO_printf(bio_err, "Error signing raw input data\n");
756                 goto end;
757             }
758         }
759         rv = EVP_DigestSignFinal(mctx, NULL, poutlen);
760         if (rv == 1 && out != NULL) {
761             *out = app_malloc(*poutlen, "buffer output");
762             rv = EVP_DigestSignFinal(mctx, *out, poutlen);
763         }
764         break;
765     }
766
767  end:
768     OPENSSL_free(mbuf);
769     EVP_MD_CTX_free(mctx);
770     return rv;
771 }