SM4 optimization for ARM by HW instruction
[openssl.git] / apps / ts.c
1 /*
2  * Copyright 2006-2021 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 <openssl/opensslconf.h>
11 #include <stdio.h>
12 #include <stdlib.h>
13 #include <string.h>
14 #include "apps.h"
15 #include "progs.h"
16 #include <openssl/bio.h>
17 #include <openssl/err.h>
18 #include <openssl/pem.h>
19 #include <openssl/rand.h>
20 #include <openssl/ts.h>
21 #include <openssl/bn.h>
22
23 /* Request nonce length, in bits (must be a multiple of 8). */
24 #define NONCE_LENGTH            64
25
26 /* Name of config entry that defines the OID file. */
27 #define ENV_OID_FILE            "oid_file"
28
29 /* Is |EXACTLY_ONE| of three pointers set? */
30 #define EXACTLY_ONE(a, b, c) \
31         (( a && !b && !c) || \
32          ( b && !a && !c) || \
33          ( c && !a && !b))
34
35 static ASN1_OBJECT *txt2obj(const char *oid);
36 static CONF *load_config_file(const char *configfile);
37
38 /* Query related functions. */
39 static int query_command(const char *data, const char *digest,
40                          const EVP_MD *md, const char *policy, int no_nonce,
41                          int cert, const char *in, const char *out, int text);
42 static TS_REQ *create_query(BIO *data_bio, const char *digest, const EVP_MD *md,
43                             const char *policy, int no_nonce, int cert);
44 static int create_digest(BIO *input, const char *digest,
45                          const EVP_MD *md, unsigned char **md_value);
46 static ASN1_INTEGER *create_nonce(int bits);
47
48 /* Reply related functions. */
49 static int reply_command(CONF *conf, const char *section, const char *engine,
50                          const char *queryfile, const char *passin, const char *inkey,
51                          const EVP_MD *md, const char *signer, const char *chain,
52                          const char *policy, const char *in, int token_in,
53                          const char *out, int token_out, int text);
54 static TS_RESP *read_PKCS7(BIO *in_bio);
55 static TS_RESP *create_response(CONF *conf, const char *section, const char *engine,
56                                 const char *queryfile, const char *passin,
57                                 const char *inkey, const EVP_MD *md, const char *signer,
58                                 const char *chain, const char *policy);
59 static ASN1_INTEGER *serial_cb(TS_RESP_CTX *ctx, void *data);
60 static ASN1_INTEGER *next_serial(const char *serialfile);
61 static int save_ts_serial(const char *serialfile, ASN1_INTEGER *serial);
62
63 /* Verify related functions. */
64 static int verify_command(const char *data, const char *digest, const char *queryfile,
65                           const char *in, int token_in,
66                           const char *CApath, const char *CAfile,
67                           const char *CAstore,
68                           char *untrusted, X509_VERIFY_PARAM *vpm);
69 static TS_VERIFY_CTX *create_verify_ctx(const char *data, const char *digest,
70                                         const char *queryfile,
71                                         const char *CApath, const char *CAfile,
72                                         const char *CAstore,
73                                         char *untrusted,
74                                         X509_VERIFY_PARAM *vpm);
75 static X509_STORE *create_cert_store(const char *CApath, const char *CAfile,
76                                      const char *CAstore, X509_VERIFY_PARAM *vpm);
77 static int verify_cb(int ok, X509_STORE_CTX *ctx);
78
79 typedef enum OPTION_choice {
80     OPT_COMMON,
81     OPT_ENGINE, OPT_CONFIG, OPT_SECTION, OPT_QUERY, OPT_DATA,
82     OPT_DIGEST, OPT_TSPOLICY, OPT_NO_NONCE, OPT_CERT,
83     OPT_IN, OPT_TOKEN_IN, OPT_OUT, OPT_TOKEN_OUT, OPT_TEXT,
84     OPT_REPLY, OPT_QUERYFILE, OPT_PASSIN, OPT_INKEY, OPT_SIGNER,
85     OPT_CHAIN, OPT_VERIFY, OPT_CAPATH, OPT_CAFILE, OPT_CASTORE, OPT_UNTRUSTED,
86     OPT_MD, OPT_V_ENUM, OPT_R_ENUM, OPT_PROV_ENUM
87 } OPTION_CHOICE;
88
89 const OPTIONS ts_options[] = {
90     OPT_SECTION("General"),
91     {"help", OPT_HELP, '-', "Display this summary"},
92     {"config", OPT_CONFIG, '<', "Configuration file"},
93     {"section", OPT_SECTION, 's', "Section to use within config file"},
94 #ifndef OPENSSL_NO_ENGINE
95     {"engine", OPT_ENGINE, 's', "Use engine, possibly a hardware device"},
96 #endif
97     {"inkey", OPT_INKEY, 's', "File with private key for reply"},
98     {"signer", OPT_SIGNER, 's', "Signer certificate file"},
99     {"chain", OPT_CHAIN, '<', "File with signer CA chain"},
100     {"CAfile", OPT_CAFILE, '<', "File with trusted CA certs"},
101     {"CApath", OPT_CAPATH, '/', "Path to trusted CA files"},
102     {"CAstore", OPT_CASTORE, ':', "URI to trusted CA store"},
103     {"untrusted", OPT_UNTRUSTED, '<', "Extra untrusted certs"},
104     {"token_in", OPT_TOKEN_IN, '-', "Input is a PKCS#7 file"},
105     {"token_out", OPT_TOKEN_OUT, '-', "Output is a PKCS#7 file"},
106     {"passin", OPT_PASSIN, 's', "Input file pass phrase source"},
107     {"", OPT_MD, '-', "Any supported digest"},
108
109     OPT_SECTION("Query"),
110     {"query", OPT_QUERY, '-', "Generate a TS query"},
111     {"data", OPT_DATA, '<', "File to hash"},
112     {"digest", OPT_DIGEST, 's', "Digest (as a hex string)"},
113     {"queryfile", OPT_QUERYFILE, '<', "File containing a TS query"},
114     {"cert", OPT_CERT, '-', "Put cert request into query"},
115     {"in", OPT_IN, '<', "Input file"},
116
117     OPT_SECTION("Verify"),
118     {"verify", OPT_VERIFY, '-', "Verify a TS response"},
119     {"reply", OPT_REPLY, '-', "Generate a TS reply"},
120     {"tspolicy", OPT_TSPOLICY, 's', "Policy OID to use"},
121     {"no_nonce", OPT_NO_NONCE, '-', "Do not include a nonce"},
122     {"out", OPT_OUT, '>', "Output file"},
123     {"text", OPT_TEXT, '-', "Output text (not DER)"},
124
125     OPT_R_OPTIONS,
126     OPT_V_OPTIONS,
127     OPT_PROV_OPTIONS,
128     {NULL}
129 };
130
131 /*
132  * This command is so complex, special help is needed.
133  */
134 static char* opt_helplist[] = {
135     "",
136     "Typical uses:",
137     " openssl ts -query [-rand file...] [-config file] [-data file]",
138     "    [-digest hexstring] [-tspolicy oid] [-no_nonce] [-cert]",
139     "    [-in file] [-out file] [-text]",
140     "",
141     " openssl ts -reply [-config file] [-section tsa_section]",
142     "    [-queryfile file] [-passin password]",
143     "    [-signer tsa_cert.pem] [-inkey private_key.pem]",
144     "    [-chain certs_file.pem] [-tspolicy oid]",
145     "    [-in file] [-token_in] [-out file] [-token_out]",
146 #ifndef OPENSSL_NO_ENGINE
147     "    [-text] [-engine id]",
148 #else
149     "    [-text]",
150 #endif
151     "",
152     " openssl ts -verify -CApath dir -CAfile root-cert.pem -CAstore uri",
153     "   -untrusted extra-certs.pem [-data file] [-digest hexstring]",
154     "    [-queryfile request.tsq] -in response.tsr [-token_in] ...",
155     NULL,
156 };
157
158 int ts_main(int argc, char **argv)
159 {
160     CONF *conf = NULL;
161     const char *CAfile = NULL, *prog;
162     char *untrusted = NULL;
163     const char *configfile = default_config_file, *engine = NULL;
164     const char *section = NULL, *digestname = NULL;
165     char **helpp;
166     char *password = NULL;
167     char *data = NULL, *digest = NULL, *policy = NULL;
168     char *in = NULL, *out = NULL, *queryfile = NULL, *passin = NULL;
169     char *inkey = NULL, *signer = NULL, *chain = NULL, *CApath = NULL;
170     char *CAstore = NULL;
171     EVP_MD *md = NULL;
172     OPTION_CHOICE o, mode = OPT_ERR;
173     int ret = 1, no_nonce = 0, cert = 0, text = 0;
174     int vpmtouched = 0;
175     X509_VERIFY_PARAM *vpm = NULL;
176     /* Input is ContentInfo instead of TimeStampResp. */
177     int token_in = 0;
178     /* Output is ContentInfo instead of TimeStampResp. */
179     int token_out = 0;
180
181     if ((vpm = X509_VERIFY_PARAM_new()) == NULL)
182         goto end;
183
184     opt_set_unknown_name("digest");
185     prog = opt_init(argc, argv, ts_options);
186     while ((o = opt_next()) != OPT_EOF) {
187         switch (o) {
188         case OPT_EOF:
189         case OPT_ERR:
190  opthelp:
191             BIO_printf(bio_err, "%s: Use -help for summary.\n", prog);
192             goto end;
193         case OPT_HELP:
194             opt_help(ts_options);
195             for (helpp = opt_helplist; *helpp; ++helpp)
196                 BIO_printf(bio_err, "%s\n", *helpp);
197             ret = 0;
198             goto end;
199         case OPT_CONFIG:
200             configfile = opt_arg();
201             break;
202         case OPT_SECTION:
203             section = opt_arg();
204             break;
205         case OPT_QUERY:
206         case OPT_REPLY:
207         case OPT_VERIFY:
208             if (mode != OPT_ERR) {
209                 BIO_printf(bio_err, "%s: Must give only one of -query, -reply, or -verify\n", prog);
210                 goto opthelp;
211             }
212             mode = o;
213             break;
214         case OPT_DATA:
215             data = opt_arg();
216             break;
217         case OPT_DIGEST:
218             digest = opt_arg();
219             break;
220         case OPT_R_CASES:
221             if (!opt_rand(o))
222                 goto end;
223             break;
224         case OPT_PROV_CASES:
225             if (!opt_provider(o))
226                 goto end;
227             break;
228         case OPT_TSPOLICY:
229             policy = opt_arg();
230             break;
231         case OPT_NO_NONCE:
232             no_nonce = 1;
233             break;
234         case OPT_CERT:
235             cert = 1;
236             break;
237         case OPT_IN:
238             in = opt_arg();
239             break;
240         case OPT_TOKEN_IN:
241             token_in = 1;
242             break;
243         case OPT_OUT:
244             out = opt_arg();
245             break;
246         case OPT_TOKEN_OUT:
247             token_out = 1;
248             break;
249         case OPT_TEXT:
250             text = 1;
251             break;
252         case OPT_QUERYFILE:
253             queryfile = opt_arg();
254             break;
255         case OPT_PASSIN:
256             passin = opt_arg();
257             break;
258         case OPT_INKEY:
259             inkey = opt_arg();
260             break;
261         case OPT_SIGNER:
262             signer = opt_arg();
263             break;
264         case OPT_CHAIN:
265             chain = opt_arg();
266             break;
267         case OPT_CAPATH:
268             CApath = opt_arg();
269             break;
270         case OPT_CAFILE:
271             CAfile = opt_arg();
272             break;
273         case OPT_CASTORE:
274             CAstore = opt_arg();
275             break;
276         case OPT_UNTRUSTED:
277             untrusted = opt_arg();
278             break;
279         case OPT_ENGINE:
280             engine = opt_arg();
281             break;
282         case OPT_MD:
283             digestname = opt_unknown();
284             break;
285         case OPT_V_CASES:
286             if (!opt_verify(o, vpm))
287                 goto end;
288             vpmtouched++;
289             break;
290         }
291     }
292
293     /* No extra arguments. */
294     if (!opt_check_rest_arg(NULL))
295         goto opthelp;
296     if (mode == OPT_ERR) {
297         BIO_printf(bio_err, "%s: Must give one of -query, -reply, or -verify\n", prog);
298         goto opthelp;
299     }
300
301     if (!app_RAND_load())
302         goto end;
303
304     if (!opt_md(digestname, &md))
305         goto opthelp;
306     if (mode == OPT_REPLY && passin &&
307         !app_passwd(passin, NULL, &password, NULL)) {
308         BIO_printf(bio_err, "Error getting password.\n");
309         goto end;
310     }
311
312     if ((conf = load_config_file(configfile)) == NULL)
313         goto end;
314     if (configfile != default_config_file && !app_load_modules(conf))
315         goto end;
316
317     /* Check parameter consistency and execute the appropriate function. */
318     if (mode == OPT_QUERY) {
319         if (vpmtouched)
320             goto opthelp;
321         if ((data != NULL) && (digest != NULL))
322             goto opthelp;
323         ret = !query_command(data, digest, md, policy, no_nonce, cert,
324                              in, out, text);
325     } else if (mode == OPT_REPLY) {
326         if (vpmtouched)
327             goto opthelp;
328         if ((in != NULL) && (queryfile != NULL))
329             goto opthelp;
330         if (in == NULL) {
331             if ((conf == NULL) || (token_in != 0))
332                 goto opthelp;
333         }
334         ret = !reply_command(conf, section, engine, queryfile,
335                              password, inkey, md, signer, chain, policy,
336                              in, token_in, out, token_out, text);
337
338     } else if (mode == OPT_VERIFY) {
339         if ((in == NULL) || !EXACTLY_ONE(queryfile, data, digest))
340             goto opthelp;
341         ret = !verify_command(data, digest, queryfile, in, token_in,
342                               CApath, CAfile, CAstore, untrusted,
343                               vpmtouched ? vpm : NULL);
344     } else {
345         goto opthelp;
346     }
347
348  end:
349     X509_VERIFY_PARAM_free(vpm);
350     EVP_MD_free(md);
351     NCONF_free(conf);
352     OPENSSL_free(password);
353     return ret;
354 }
355
356 /*
357  * Configuration file-related function definitions.
358  */
359
360 static ASN1_OBJECT *txt2obj(const char *oid)
361 {
362     ASN1_OBJECT *oid_obj = NULL;
363
364     if ((oid_obj = OBJ_txt2obj(oid, 0)) == NULL)
365         BIO_printf(bio_err, "cannot convert %s to OID\n", oid);
366
367     return oid_obj;
368 }
369
370 static CONF *load_config_file(const char *configfile)
371 {
372     CONF *conf = app_load_config(configfile);
373
374     if (conf != NULL) {
375         const char *p;
376
377         BIO_printf(bio_err, "Using configuration from %s\n", configfile);
378         p = NCONF_get_string(conf, NULL, ENV_OID_FILE);
379         if (p != NULL) {
380             BIO *oid_bio = BIO_new_file(p, "r");
381             if (!oid_bio)
382                 ERR_print_errors(bio_err);
383             else {
384                 OBJ_create_objects(oid_bio);
385                 BIO_free_all(oid_bio);
386             }
387         } else
388             ERR_clear_error();
389         if (!add_oid_section(conf))
390             ERR_print_errors(bio_err);
391     }
392     return conf;
393 }
394
395 /*
396  * Query-related method definitions.
397  */
398 static int query_command(const char *data, const char *digest, const EVP_MD *md,
399                          const char *policy, int no_nonce,
400                          int cert, const char *in, const char *out, int text)
401 {
402     int ret = 0;
403     TS_REQ *query = NULL;
404     BIO *in_bio = NULL;
405     BIO *data_bio = NULL;
406     BIO *out_bio = NULL;
407
408     /* Build query object. */
409     if (in != NULL) {
410         if ((in_bio = bio_open_default(in, 'r', FORMAT_ASN1)) == NULL)
411             goto end;
412         query = d2i_TS_REQ_bio(in_bio, NULL);
413     } else {
414         if (digest == NULL
415             && (data_bio = bio_open_default(data, 'r', FORMAT_ASN1)) == NULL)
416             goto end;
417         query = create_query(data_bio, digest, md, policy, no_nonce, cert);
418     }
419     if (query == NULL)
420         goto end;
421
422     if (text) {
423         if ((out_bio = bio_open_default(out, 'w', FORMAT_TEXT)) == NULL)
424             goto end;
425         if (!TS_REQ_print_bio(out_bio, query))
426             goto end;
427     } else {
428         if ((out_bio = bio_open_default(out, 'w', FORMAT_ASN1)) == NULL)
429             goto end;
430         if (!i2d_TS_REQ_bio(out_bio, query))
431             goto end;
432     }
433
434     ret = 1;
435
436  end:
437     ERR_print_errors(bio_err);
438     BIO_free_all(in_bio);
439     BIO_free_all(data_bio);
440     BIO_free_all(out_bio);
441     TS_REQ_free(query);
442     return ret;
443 }
444
445 static TS_REQ *create_query(BIO *data_bio, const char *digest, const EVP_MD *md,
446                             const char *policy, int no_nonce, int cert)
447 {
448     int ret = 0;
449     TS_REQ *ts_req = NULL;
450     int len;
451     TS_MSG_IMPRINT *msg_imprint = NULL;
452     X509_ALGOR *algo = NULL;
453     unsigned char *data = NULL;
454     ASN1_OBJECT *policy_obj = NULL;
455     ASN1_INTEGER *nonce_asn1 = NULL;
456
457     if (md == NULL && (md = EVP_get_digestbyname("sha256")) == NULL)
458         goto err;
459     if ((ts_req = TS_REQ_new()) == NULL)
460         goto err;
461     if (!TS_REQ_set_version(ts_req, 1))
462         goto err;
463     if ((msg_imprint = TS_MSG_IMPRINT_new()) == NULL)
464         goto err;
465     if ((algo = X509_ALGOR_new()) == NULL)
466         goto err;
467     if ((algo->algorithm = OBJ_nid2obj(EVP_MD_get_type(md))) == NULL)
468         goto err;
469     if ((algo->parameter = ASN1_TYPE_new()) == NULL)
470         goto err;
471     algo->parameter->type = V_ASN1_NULL;
472     if (!TS_MSG_IMPRINT_set_algo(msg_imprint, algo))
473         goto err;
474     if ((len = create_digest(data_bio, digest, md, &data)) == 0)
475         goto err;
476     if (!TS_MSG_IMPRINT_set_msg(msg_imprint, data, len))
477         goto err;
478     if (!TS_REQ_set_msg_imprint(ts_req, msg_imprint))
479         goto err;
480     if (policy && (policy_obj = txt2obj(policy)) == NULL)
481         goto err;
482     if (policy_obj && !TS_REQ_set_policy_id(ts_req, policy_obj))
483         goto err;
484
485     /* Setting nonce if requested. */
486     if (!no_nonce && (nonce_asn1 = create_nonce(NONCE_LENGTH)) == NULL)
487         goto err;
488     if (nonce_asn1 && !TS_REQ_set_nonce(ts_req, nonce_asn1))
489         goto err;
490     if (!TS_REQ_set_cert_req(ts_req, cert))
491         goto err;
492
493     ret = 1;
494  err:
495     if (!ret) {
496         TS_REQ_free(ts_req);
497         ts_req = NULL;
498         BIO_printf(bio_err, "could not create query\n");
499         ERR_print_errors(bio_err);
500     }
501     TS_MSG_IMPRINT_free(msg_imprint);
502     X509_ALGOR_free(algo);
503     OPENSSL_free(data);
504     ASN1_OBJECT_free(policy_obj);
505     ASN1_INTEGER_free(nonce_asn1);
506     return ts_req;
507 }
508
509 static int create_digest(BIO *input, const char *digest, const EVP_MD *md,
510                          unsigned char **md_value)
511 {
512     int md_value_len;
513     int rv = 0;
514     EVP_MD_CTX *md_ctx = NULL;
515
516     md_value_len = EVP_MD_get_size(md);
517     if (md_value_len < 0)
518         return 0;
519
520     if (input != NULL) {
521         unsigned char buffer[4096];
522         int length;
523
524         md_ctx = EVP_MD_CTX_new();
525         if (md_ctx == NULL)
526             return 0;
527         *md_value = app_malloc(md_value_len, "digest buffer");
528         if (!EVP_DigestInit(md_ctx, md))
529             goto err;
530         while ((length = BIO_read(input, buffer, sizeof(buffer))) > 0) {
531             if (!EVP_DigestUpdate(md_ctx, buffer, length))
532                 goto err;
533         }
534         if (!EVP_DigestFinal(md_ctx, *md_value, NULL))
535             goto err;
536         md_value_len = EVP_MD_get_size(md);
537     } else {
538         long digest_len;
539
540         *md_value = OPENSSL_hexstr2buf(digest, &digest_len);
541         if (*md_value == NULL || md_value_len != digest_len) {
542             OPENSSL_free(*md_value);
543             *md_value = NULL;
544             BIO_printf(bio_err, "bad digest, %d bytes "
545                        "must be specified\n", md_value_len);
546             return 0;
547         }
548     }
549     rv = md_value_len;
550  err:
551     EVP_MD_CTX_free(md_ctx);
552     return rv;
553 }
554
555 static ASN1_INTEGER *create_nonce(int bits)
556 {
557     unsigned char buf[20];
558     ASN1_INTEGER *nonce = NULL;
559     int len = (bits - 1) / 8 + 1;
560     int i;
561
562     if (len > (int)sizeof(buf))
563         goto err;
564     if (RAND_bytes(buf, len) <= 0)
565         goto err;
566
567     /* Find the first non-zero byte and creating ASN1_INTEGER object. */
568     for (i = 0; i < len && !buf[i]; ++i)
569         continue;
570     if ((nonce = ASN1_INTEGER_new()) == NULL)
571         goto err;
572     OPENSSL_free(nonce->data);
573     nonce->length = len - i;
574     nonce->data = app_malloc(nonce->length + 1, "nonce buffer");
575     memcpy(nonce->data, buf + i, nonce->length);
576     return nonce;
577
578  err:
579     BIO_printf(bio_err, "could not create nonce\n");
580     ASN1_INTEGER_free(nonce);
581     return NULL;
582 }
583
584 /*
585  * Reply-related method definitions.
586  */
587
588 static int reply_command(CONF *conf, const char *section, const char *engine,
589                          const char *queryfile, const char *passin, const char *inkey,
590                          const EVP_MD *md, const char *signer, const char *chain,
591                          const char *policy, const char *in, int token_in,
592                          const char *out, int token_out, int text)
593 {
594     int ret = 0;
595     TS_RESP *response = NULL;
596     BIO *in_bio = NULL;
597     BIO *query_bio = NULL;
598     BIO *inkey_bio = NULL;
599     BIO *signer_bio = NULL;
600     BIO *out_bio = NULL;
601
602     if (in != NULL) {
603         if ((in_bio = BIO_new_file(in, "rb")) == NULL)
604             goto end;
605         if (token_in) {
606             response = read_PKCS7(in_bio);
607         } else {
608             response = d2i_TS_RESP_bio(in_bio, NULL);
609         }
610     } else {
611         response = create_response(conf, section, engine, queryfile,
612                                    passin, inkey, md, signer, chain, policy);
613         if (response != NULL)
614             BIO_printf(bio_err, "Response has been generated.\n");
615         else
616             BIO_printf(bio_err, "Response is not generated.\n");
617     }
618     if (response == NULL)
619         goto end;
620
621     /* Write response. */
622     if (text) {
623         if ((out_bio = bio_open_default(out, 'w', FORMAT_TEXT)) == NULL)
624         goto end;
625         if (token_out) {
626             TS_TST_INFO *tst_info = TS_RESP_get_tst_info(response);
627             if (!TS_TST_INFO_print_bio(out_bio, tst_info))
628                 goto end;
629         } else {
630             if (!TS_RESP_print_bio(out_bio, response))
631                 goto end;
632         }
633     } else {
634         if ((out_bio = bio_open_default(out, 'w', FORMAT_ASN1)) == NULL)
635             goto end;
636         if (token_out) {
637             PKCS7 *token = TS_RESP_get_token(response);
638             if (!i2d_PKCS7_bio(out_bio, token))
639                 goto end;
640         } else {
641             if (!i2d_TS_RESP_bio(out_bio, response))
642                 goto end;
643         }
644     }
645
646     ret = 1;
647
648  end:
649     ERR_print_errors(bio_err);
650     BIO_free_all(in_bio);
651     BIO_free_all(query_bio);
652     BIO_free_all(inkey_bio);
653     BIO_free_all(signer_bio);
654     BIO_free_all(out_bio);
655     TS_RESP_free(response);
656     return ret;
657 }
658
659 /* Reads a PKCS7 token and adds default 'granted' status info to it. */
660 static TS_RESP *read_PKCS7(BIO *in_bio)
661 {
662     int ret = 0;
663     PKCS7 *token = NULL;
664     TS_TST_INFO *tst_info = NULL;
665     TS_RESP *resp = NULL;
666     TS_STATUS_INFO *si = NULL;
667
668     if ((token = d2i_PKCS7_bio(in_bio, NULL)) == NULL)
669         goto end;
670     if ((tst_info = PKCS7_to_TS_TST_INFO(token)) == NULL)
671         goto end;
672     if ((resp = TS_RESP_new()) == NULL)
673         goto end;
674     if ((si = TS_STATUS_INFO_new()) == NULL)
675         goto end;
676     if (!TS_STATUS_INFO_set_status(si, TS_STATUS_GRANTED))
677         goto end;
678     if (!TS_RESP_set_status_info(resp, si))
679         goto end;
680     TS_RESP_set_tst_info(resp, token, tst_info);
681     token = NULL;               /* Ownership is lost. */
682     tst_info = NULL;            /* Ownership is lost. */
683     ret = 1;
684
685  end:
686     PKCS7_free(token);
687     TS_TST_INFO_free(tst_info);
688     if (!ret) {
689         TS_RESP_free(resp);
690         resp = NULL;
691     }
692     TS_STATUS_INFO_free(si);
693     return resp;
694 }
695
696 static TS_RESP *create_response(CONF *conf, const char *section, const char *engine,
697                                 const char *queryfile, const char *passin,
698                                 const char *inkey, const EVP_MD *md, const char *signer,
699                                 const char *chain, const char *policy)
700 {
701     int ret = 0;
702     TS_RESP *response = NULL;
703     BIO *query_bio = NULL;
704     TS_RESP_CTX *resp_ctx = NULL;
705
706     if ((query_bio = BIO_new_file(queryfile, "rb")) == NULL)
707         goto end;
708     if ((section = TS_CONF_get_tsa_section(conf, section)) == NULL)
709         goto end;
710     if ((resp_ctx = TS_RESP_CTX_new()) == NULL)
711         goto end;
712     if (!TS_CONF_set_serial(conf, section, serial_cb, resp_ctx))
713         goto end;
714 #ifndef OPENSSL_NO_ENGINE
715     if (!TS_CONF_set_crypto_device(conf, section, engine))
716         goto end;
717 #endif
718     if (!TS_CONF_set_signer_cert(conf, section, signer, resp_ctx))
719         goto end;
720     if (!TS_CONF_set_certs(conf, section, chain, resp_ctx))
721         goto end;
722     if (!TS_CONF_set_signer_key(conf, section, inkey, passin, resp_ctx))
723         goto end;
724
725     if (md) {
726         if (!TS_RESP_CTX_set_signer_digest(resp_ctx, md))
727             goto end;
728     } else if (!TS_CONF_set_signer_digest(conf, section, NULL, resp_ctx)) {
729             goto end;
730     }
731
732     if (!TS_CONF_set_ess_cert_id_digest(conf, section, resp_ctx))
733         goto end;
734     if (!TS_CONF_set_def_policy(conf, section, policy, resp_ctx))
735         goto end;
736     if (!TS_CONF_set_policies(conf, section, resp_ctx))
737         goto end;
738     if (!TS_CONF_set_digests(conf, section, resp_ctx))
739         goto end;
740     if (!TS_CONF_set_accuracy(conf, section, resp_ctx))
741         goto end;
742     if (!TS_CONF_set_clock_precision_digits(conf, section, resp_ctx))
743         goto end;
744     if (!TS_CONF_set_ordering(conf, section, resp_ctx))
745         goto end;
746     if (!TS_CONF_set_tsa_name(conf, section, resp_ctx))
747         goto end;
748     if (!TS_CONF_set_ess_cert_id_chain(conf, section, resp_ctx))
749         goto end;
750     if ((response = TS_RESP_create_response(resp_ctx, query_bio)) == NULL)
751         goto end;
752     ret = 1;
753
754  end:
755     if (!ret) {
756         TS_RESP_free(response);
757         response = NULL;
758     }
759     TS_RESP_CTX_free(resp_ctx);
760     BIO_free_all(query_bio);
761     return response;
762 }
763
764 static ASN1_INTEGER *serial_cb(TS_RESP_CTX *ctx, void *data)
765 {
766     const char *serial_file = (const char *)data;
767     ASN1_INTEGER *serial = next_serial(serial_file);
768
769     if (serial == NULL) {
770         TS_RESP_CTX_set_status_info(ctx, TS_STATUS_REJECTION,
771                                     "Error during serial number "
772                                     "generation.");
773         TS_RESP_CTX_add_failure_info(ctx, TS_INFO_ADD_INFO_NOT_AVAILABLE);
774     } else {
775         save_ts_serial(serial_file, serial);
776     }
777
778     return serial;
779 }
780
781 static ASN1_INTEGER *next_serial(const char *serialfile)
782 {
783     int ret = 0;
784     BIO *in = NULL;
785     ASN1_INTEGER *serial = NULL;
786     BIGNUM *bn = NULL;
787
788     if ((serial = ASN1_INTEGER_new()) == NULL)
789         goto err;
790
791     if ((in = BIO_new_file(serialfile, "r")) == NULL) {
792         ERR_clear_error();
793         BIO_printf(bio_err, "Warning: could not open file %s for "
794                    "reading, using serial number: 1\n", serialfile);
795         if (!ASN1_INTEGER_set(serial, 1))
796             goto err;
797     } else {
798         char buf[1024];
799         if (!a2i_ASN1_INTEGER(in, serial, buf, sizeof(buf))) {
800             BIO_printf(bio_err, "unable to load number from %s\n",
801                        serialfile);
802             goto err;
803         }
804         if ((bn = ASN1_INTEGER_to_BN(serial, NULL)) == NULL)
805             goto err;
806         ASN1_INTEGER_free(serial);
807         serial = NULL;
808         if (!BN_add_word(bn, 1))
809             goto err;
810         if ((serial = BN_to_ASN1_INTEGER(bn, NULL)) == NULL)
811             goto err;
812     }
813     ret = 1;
814
815  err:
816     if (!ret) {
817         ASN1_INTEGER_free(serial);
818         serial = NULL;
819     }
820     BIO_free_all(in);
821     BN_free(bn);
822     return serial;
823 }
824
825 static int save_ts_serial(const char *serialfile, ASN1_INTEGER *serial)
826 {
827     int ret = 0;
828     BIO *out = NULL;
829
830     if ((out = BIO_new_file(serialfile, "w")) == NULL)
831         goto err;
832     if (i2a_ASN1_INTEGER(out, serial) <= 0)
833         goto err;
834     if (BIO_puts(out, "\n") <= 0)
835         goto err;
836     ret = 1;
837  err:
838     if (!ret)
839         BIO_printf(bio_err, "could not save serial number to %s\n",
840                    serialfile);
841     BIO_free_all(out);
842     return ret;
843 }
844
845
846 /*
847  * Verify-related method definitions.
848  */
849
850 static int verify_command(const char *data, const char *digest, const char *queryfile,
851                           const char *in, int token_in,
852                           const char *CApath, const char *CAfile,
853                           const char *CAstore, char *untrusted,
854                           X509_VERIFY_PARAM *vpm)
855 {
856     BIO *in_bio = NULL;
857     PKCS7 *token = NULL;
858     TS_RESP *response = NULL;
859     TS_VERIFY_CTX *verify_ctx = NULL;
860     int ret = 0;
861
862     if ((in_bio = BIO_new_file(in, "rb")) == NULL)
863         goto end;
864     if (token_in) {
865         if ((token = d2i_PKCS7_bio(in_bio, NULL)) == NULL)
866             goto end;
867     } else {
868         if ((response = d2i_TS_RESP_bio(in_bio, NULL)) == NULL)
869             goto end;
870     }
871
872     if ((verify_ctx = create_verify_ctx(data, digest, queryfile,
873                                         CApath, CAfile, CAstore, untrusted,
874                                         vpm)) == NULL)
875         goto end;
876
877     ret = token_in
878         ? TS_RESP_verify_token(verify_ctx, token)
879         : TS_RESP_verify_response(verify_ctx, response);
880
881  end:
882     printf("Verification: ");
883     if (ret)
884         printf("OK\n");
885     else {
886         printf("FAILED\n");
887         ERR_print_errors(bio_err);
888     }
889
890     BIO_free_all(in_bio);
891     PKCS7_free(token);
892     TS_RESP_free(response);
893     TS_VERIFY_CTX_free(verify_ctx);
894     return ret;
895 }
896
897 static TS_VERIFY_CTX *create_verify_ctx(const char *data, const char *digest,
898                                         const char *queryfile,
899                                         const char *CApath, const char *CAfile,
900                                         const char *CAstore,
901                                         char *untrusted,
902                                         X509_VERIFY_PARAM *vpm)
903 {
904     TS_VERIFY_CTX *ctx = NULL;
905     STACK_OF(X509) *certs;
906     BIO *input = NULL;
907     TS_REQ *request = NULL;
908     int ret = 0;
909     int f = 0;
910
911     if (data != NULL || digest != NULL) {
912         if ((ctx = TS_VERIFY_CTX_new()) == NULL)
913             goto err;
914         f = TS_VFY_VERSION | TS_VFY_SIGNER;
915         if (data != NULL) {
916             BIO *out = NULL;
917
918             f |= TS_VFY_DATA;
919             if ((out = BIO_new_file(data, "rb")) == NULL)
920                 goto err;
921             if (TS_VERIFY_CTX_set_data(ctx, out) == NULL) {
922                 BIO_free_all(out);
923                 goto err;
924             }
925         } else if (digest != NULL) {
926             long imprint_len;
927             unsigned char *hexstr = OPENSSL_hexstr2buf(digest, &imprint_len);
928             f |= TS_VFY_IMPRINT;
929             if (TS_VERIFY_CTX_set_imprint(ctx, hexstr, imprint_len) == NULL) {
930                 BIO_printf(bio_err, "invalid digest string\n");
931                 goto err;
932             }
933         }
934
935     } else if (queryfile != NULL) {
936         if ((input = BIO_new_file(queryfile, "rb")) == NULL)
937             goto err;
938         if ((request = d2i_TS_REQ_bio(input, NULL)) == NULL)
939             goto err;
940         if ((ctx = TS_REQ_to_TS_VERIFY_CTX(request, NULL)) == NULL)
941             goto err;
942     } else {
943         return NULL;
944     }
945
946     /* Add the signature verification flag and arguments. */
947     TS_VERIFY_CTX_add_flags(ctx, f | TS_VFY_SIGNATURE);
948
949     /* Initialising the X509_STORE object. */
950     if (TS_VERIFY_CTX_set_store(ctx,
951                                 create_cert_store(CApath, CAfile, CAstore, vpm))
952             == NULL)
953         goto err;
954
955     /* Loading any extra untrusted certificates. */
956     if (untrusted != NULL) {
957         certs = load_certs_multifile(untrusted, NULL, "extra untrusted certs",
958                                      vpm);
959         if (certs == NULL || TS_VERIFY_CTX_set_certs(ctx, certs) == NULL)
960             goto err;
961     }
962     ret = 1;
963
964  err:
965     if (!ret) {
966         TS_VERIFY_CTX_free(ctx);
967         ctx = NULL;
968     }
969     BIO_free_all(input);
970     TS_REQ_free(request);
971     return ctx;
972 }
973
974 static X509_STORE *create_cert_store(const char *CApath, const char *CAfile,
975                                      const char *CAstore, X509_VERIFY_PARAM *vpm)
976 {
977     X509_STORE *cert_ctx = NULL;
978     X509_LOOKUP *lookup = NULL;
979     OSSL_LIB_CTX *libctx = app_get0_libctx();
980     const char *propq = app_get0_propq();
981
982     cert_ctx = X509_STORE_new();
983     X509_STORE_set_verify_cb(cert_ctx, verify_cb);
984     if (CApath != NULL) {
985         lookup = X509_STORE_add_lookup(cert_ctx, X509_LOOKUP_hash_dir());
986         if (lookup == NULL) {
987             BIO_printf(bio_err, "memory allocation failure\n");
988             goto err;
989         }
990         if (!X509_LOOKUP_add_dir(lookup, CApath, X509_FILETYPE_PEM)) {
991             BIO_printf(bio_err, "Error loading directory %s\n", CApath);
992             goto err;
993         }
994     }
995
996     if (CAfile != NULL) {
997         lookup = X509_STORE_add_lookup(cert_ctx, X509_LOOKUP_file());
998         if (lookup == NULL) {
999             BIO_printf(bio_err, "memory allocation failure\n");
1000             goto err;
1001         }
1002         if (!X509_LOOKUP_load_file_ex(lookup, CAfile, X509_FILETYPE_PEM, libctx,
1003                                       propq)) {
1004             BIO_printf(bio_err, "Error loading file %s\n", CAfile);
1005             goto err;
1006         }
1007     }
1008
1009     if (CAstore != NULL) {
1010         lookup = X509_STORE_add_lookup(cert_ctx, X509_LOOKUP_store());
1011         if (lookup == NULL) {
1012             BIO_printf(bio_err, "memory allocation failure\n");
1013             goto err;
1014         }
1015         if (!X509_LOOKUP_load_store_ex(lookup, CAstore, libctx, propq)) {
1016             BIO_printf(bio_err, "Error loading store URI %s\n", CAstore);
1017             goto err;
1018         }
1019     }
1020
1021     if (vpm != NULL)
1022         X509_STORE_set1_param(cert_ctx, vpm);
1023
1024     return cert_ctx;
1025
1026  err:
1027     X509_STORE_free(cert_ctx);
1028     return NULL;
1029 }
1030
1031 static int verify_cb(int ok, X509_STORE_CTX *ctx)
1032 {
1033     return ok;
1034 }