free NULL cleanup
[openssl.git] / apps / ts.c
1 /* apps/ts.c */
2 /*
3  * Written by Zoltan Glozik (zglozik@stones.com) for the OpenSSL project
4  * 2002.
5  */
6 /* ====================================================================
7  * Copyright (c) 2001 The OpenSSL Project.  All rights reserved.
8  *
9  * Redistribution and use in source and binary forms, with or without
10  * modification, are permitted provided that the following conditions
11  * are met:
12  *
13  * 1. Redistributions of source code must retain the above copyright
14  *    notice, this list of conditions and the following disclaimer.
15  *
16  * 2. Redistributions in binary form must reproduce the above copyright
17  *    notice, this list of conditions and the following disclaimer in
18  *    the documentation and/or other materials provided with the
19  *    distribution.
20  *
21  * 3. All advertising materials mentioning features or use of this
22  *    software must display the following acknowledgment:
23  *    "This product includes software developed by the OpenSSL Project
24  *    for use in the OpenSSL Toolkit. (http://www.OpenSSL.org/)"
25  *
26  * 4. The names "OpenSSL Toolkit" and "OpenSSL Project" must not be used to
27  *    endorse or promote products derived from this software without
28  *    prior written permission. For written permission, please contact
29  *    licensing@OpenSSL.org.
30  *
31  * 5. Products derived from this software may not be called "OpenSSL"
32  *    nor may "OpenSSL" appear in their names without prior written
33  *    permission of the OpenSSL Project.
34  *
35  * 6. Redistributions of any form whatsoever must retain the following
36  *    acknowledgment:
37  *    "This product includes software developed by the OpenSSL Project
38  *    for use in the OpenSSL Toolkit (http://www.OpenSSL.org/)"
39  *
40  * THIS SOFTWARE IS PROVIDED BY THE OpenSSL PROJECT ``AS IS'' AND ANY
41  * EXPRESSED OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
42  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
43  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE OpenSSL PROJECT OR
44  * ITS CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
45  * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
46  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
47  * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
48  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
49  * STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
50  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED
51  * OF THE POSSIBILITY OF SUCH DAMAGE.
52  * ====================================================================
53  *
54  * This product includes cryptographic software written by Eric Young
55  * (eay@cryptsoft.com).  This product includes software written by Tim
56  * Hudson (tjh@cryptsoft.com).
57  *
58  */
59
60 #include <stdio.h>
61 #include <stdlib.h>
62 #include <string.h>
63 #include "apps.h"
64 #include <openssl/bio.h>
65 #include <openssl/err.h>
66 #include <openssl/pem.h>
67 #include <openssl/rand.h>
68 #include <openssl/ts.h>
69 #include <openssl/bn.h>
70
71 #undef PROG
72 #define PROG    ts_main
73
74 /* Length of the nonce of the request in bits (must be a multiple of 8). */
75 #define NONCE_LENGTH            64
76
77 /* Macro definitions for the configuration file. */
78 #define ENV_OID_FILE            "oid_file"
79
80 /* Local function declarations. */
81
82 static ASN1_OBJECT *txt2obj(const char *oid);
83 static CONF *load_config_file(const char *configfile);
84
85 /* Query related functions. */
86 static int query_command(const char *data, char *digest,
87                          const EVP_MD *md, const char *policy, int no_nonce,
88                          int cert, const char *in, const char *out, int text);
89 static BIO *BIO_open_with_default(const char *file, const char *mode,
90                                   FILE *default_fp);
91 static TS_REQ *create_query(BIO *data_bio, char *digest, const EVP_MD *md,
92                             const char *policy, int no_nonce, int cert);
93 static int create_digest(BIO *input, char *digest,
94                          const EVP_MD *md, unsigned char **md_value);
95 static ASN1_INTEGER *create_nonce(int bits);
96
97 /* Reply related functions. */
98 static int reply_command(CONF *conf, char *section, char *engine,
99                          char *queryfile, char *passin, char *inkey,
100                          char *signer, char *chain, const char *policy,
101                          char *in, int token_in, char *out, int token_out,
102                          int text);
103 static TS_RESP *read_PKCS7(BIO *in_bio);
104 static TS_RESP *create_response(CONF *conf, const char *section, char *engine,
105                                 char *queryfile, char *passin, char *inkey,
106                                 char *signer, char *chain,
107                                 const char *policy);
108 static ASN1_INTEGER *serial_cb(TS_RESP_CTX *ctx, void *data);
109 static ASN1_INTEGER *next_serial(const char *serialfile);
110 static int save_ts_serial(const char *serialfile, ASN1_INTEGER *serial);
111
112 /* Verify related functions. */
113 static int verify_command(char *data, char *digest, char *queryfile,
114                           char *in, int token_in,
115                           char *ca_path, char *ca_file, char *untrusted);
116 static TS_VERIFY_CTX *create_verify_ctx(char *data, char *digest,
117                                         char *queryfile,
118                                         char *ca_path, char *ca_file,
119                                         char *untrusted);
120 static X509_STORE *create_cert_store(char *ca_path, char *ca_file);
121 static int verify_cb(int ok, X509_STORE_CTX *ctx);
122
123 /* Main function definition. */
124 int MAIN(int, char **);
125
126 int MAIN(int argc, char **argv)
127 {
128     int ret = 1;
129     char *configfile = NULL;
130     char *section = NULL;
131     CONF *conf = NULL;
132     enum mode {
133         CMD_NONE, CMD_QUERY, CMD_REPLY, CMD_VERIFY
134     } mode = CMD_NONE;
135     char *data = NULL;
136     char *digest = NULL;
137     const EVP_MD *md = NULL;
138     char *rnd = NULL;
139     char *policy = NULL;
140     int no_nonce = 0;
141     int cert = 0;
142     char *in = NULL;
143     char *out = NULL;
144     int text = 0;
145     char *queryfile = NULL;
146     char *passin = NULL;        /* Password source. */
147     char *password = NULL;      /* Password itself. */
148     char *inkey = NULL;
149     char *signer = NULL;
150     char *chain = NULL;
151     char *ca_path = NULL;
152     char *ca_file = NULL;
153     char *untrusted = NULL;
154     char *engine = NULL;
155     /* Input is ContentInfo instead of TimeStampResp. */
156     int token_in = 0;
157     /* Output is ContentInfo instead of TimeStampResp. */
158     int token_out = 0;
159     int free_bio_err = 0;
160
161     ERR_load_crypto_strings();
162     apps_startup();
163
164     if (bio_err == NULL && (bio_err = BIO_new(BIO_s_file())) != NULL) {
165         BIO_set_fp(bio_err, stderr, BIO_NOCLOSE | BIO_FP_TEXT);
166         free_bio_err = 1;
167     }
168
169     if (!load_config(bio_err, NULL))
170         goto cleanup;
171
172     for (argc--, argv++; argc > 0; argc--, argv++) {
173         if (strcmp(*argv, "-config") == 0) {
174             if (argc-- < 1)
175                 goto usage;
176             configfile = *++argv;
177         } else if (strcmp(*argv, "-section") == 0) {
178             if (argc-- < 1)
179                 goto usage;
180             section = *++argv;
181         } else if (strcmp(*argv, "-query") == 0) {
182             if (mode != CMD_NONE)
183                 goto usage;
184             mode = CMD_QUERY;
185         } else if (strcmp(*argv, "-data") == 0) {
186             if (argc-- < 1)
187                 goto usage;
188             data = *++argv;
189         } else if (strcmp(*argv, "-digest") == 0) {
190             if (argc-- < 1)
191                 goto usage;
192             digest = *++argv;
193         } else if (strcmp(*argv, "-rand") == 0) {
194             if (argc-- < 1)
195                 goto usage;
196             rnd = *++argv;
197         } else if (strcmp(*argv, "-policy") == 0) {
198             if (argc-- < 1)
199                 goto usage;
200             policy = *++argv;
201         } else if (strcmp(*argv, "-no_nonce") == 0) {
202             no_nonce = 1;
203         } else if (strcmp(*argv, "-cert") == 0) {
204             cert = 1;
205         } else if (strcmp(*argv, "-in") == 0) {
206             if (argc-- < 1)
207                 goto usage;
208             in = *++argv;
209         } else if (strcmp(*argv, "-token_in") == 0) {
210             token_in = 1;
211         } else if (strcmp(*argv, "-out") == 0) {
212             if (argc-- < 1)
213                 goto usage;
214             out = *++argv;
215         } else if (strcmp(*argv, "-token_out") == 0) {
216             token_out = 1;
217         } else if (strcmp(*argv, "-text") == 0) {
218             text = 1;
219         } else if (strcmp(*argv, "-reply") == 0) {
220             if (mode != CMD_NONE)
221                 goto usage;
222             mode = CMD_REPLY;
223         } else if (strcmp(*argv, "-queryfile") == 0) {
224             if (argc-- < 1)
225                 goto usage;
226             queryfile = *++argv;
227         } else if (strcmp(*argv, "-passin") == 0) {
228             if (argc-- < 1)
229                 goto usage;
230             passin = *++argv;
231         } else if (strcmp(*argv, "-inkey") == 0) {
232             if (argc-- < 1)
233                 goto usage;
234             inkey = *++argv;
235         } else if (strcmp(*argv, "-signer") == 0) {
236             if (argc-- < 1)
237                 goto usage;
238             signer = *++argv;
239         } else if (strcmp(*argv, "-chain") == 0) {
240             if (argc-- < 1)
241                 goto usage;
242             chain = *++argv;
243         } else if (strcmp(*argv, "-verify") == 0) {
244             if (mode != CMD_NONE)
245                 goto usage;
246             mode = CMD_VERIFY;
247         } else if (strcmp(*argv, "-CApath") == 0) {
248             if (argc-- < 1)
249                 goto usage;
250             ca_path = *++argv;
251         } else if (strcmp(*argv, "-CAfile") == 0) {
252             if (argc-- < 1)
253                 goto usage;
254             ca_file = *++argv;
255         } else if (strcmp(*argv, "-untrusted") == 0) {
256             if (argc-- < 1)
257                 goto usage;
258             untrusted = *++argv;
259         } else if (strcmp(*argv, "-engine") == 0) {
260             if (argc-- < 1)
261                 goto usage;
262             engine = *++argv;
263         } else if ((md = EVP_get_digestbyname(*argv + 1)) != NULL) {
264             /* empty. */
265         } else
266             goto usage;
267     }
268
269     /* Seed the random number generator if it is going to be used. */
270     if (mode == CMD_QUERY && !no_nonce) {
271         if (!app_RAND_load_file(NULL, bio_err, 1) && rnd == NULL)
272             BIO_printf(bio_err, "warning, not much extra random "
273                        "data, consider using the -rand option\n");
274         if (rnd != NULL)
275             BIO_printf(bio_err, "%ld semi-random bytes loaded\n",
276                        app_RAND_load_files(rnd));
277     }
278
279     /* Get the password if required. */
280     if (mode == CMD_REPLY && passin &&
281         !app_passwd(bio_err, passin, NULL, &password, NULL)) {
282         BIO_printf(bio_err, "Error getting password.\n");
283         goto cleanup;
284     }
285
286     /*
287      * Check consistency of parameters and execute the appropriate function.
288      */
289     switch (mode) {
290     case CMD_NONE:
291         goto usage;
292     case CMD_QUERY:
293         /*
294          * Data file and message imprint cannot be specified at the same
295          * time.
296          */
297         ret = data != NULL && digest != NULL;
298         if (ret)
299             goto usage;
300         /* Load the config file for possible policy OIDs. */
301         conf = load_config_file(configfile);
302         ret = !query_command(data, digest, md, policy, no_nonce, cert,
303                              in, out, text);
304         break;
305     case CMD_REPLY:
306         conf = load_config_file(configfile);
307         if (in == NULL) {
308             ret = !(queryfile != NULL && conf != NULL && !token_in);
309             if (ret)
310                 goto usage;
311         } else {
312             /* 'in' and 'queryfile' are exclusive. */
313             ret = !(queryfile == NULL);
314             if (ret)
315                 goto usage;
316         }
317
318         ret = !reply_command(conf, section, engine, queryfile,
319                              password, inkey, signer, chain, policy,
320                              in, token_in, out, token_out, text);
321         break;
322     case CMD_VERIFY:
323         ret = !(((queryfile && !data && !digest)
324                  || (!queryfile && data && !digest)
325                  || (!queryfile && !data && digest))
326                 && in != NULL);
327         if (ret)
328             goto usage;
329
330         ret = !verify_command(data, digest, queryfile, in, token_in,
331                               ca_path, ca_file, untrusted);
332     }
333
334     goto cleanup;
335
336  usage:
337     BIO_printf(bio_err, "usage:\n"
338                "ts -query [-rand file%cfile%c...] [-config configfile] "
339                "[-data file_to_hash] [-digest digest_bytes]"
340                "[-md2|-md4|-md5|-sha|-sha1|-mdc2|-ripemd160] "
341                "[-policy object_id] [-no_nonce] [-cert] "
342                "[-in request.tsq] [-out request.tsq] [-text]\n",
343                LIST_SEPARATOR_CHAR, LIST_SEPARATOR_CHAR);
344     BIO_printf(bio_err, "or\n"
345                "ts -reply [-config configfile] [-section tsa_section] "
346                "[-queryfile request.tsq] [-passin password] "
347                "[-signer tsa_cert.pem] [-inkey private_key.pem] "
348                "[-chain certs_file.pem] [-policy object_id] "
349                "[-in response.tsr] [-token_in] "
350                "[-out response.tsr] [-token_out] [-text] [-engine id]\n");
351     BIO_printf(bio_err, "or\n"
352                "ts -verify [-data file_to_hash] [-digest digest_bytes] "
353                "[-queryfile request.tsq] "
354                "-in response.tsr [-token_in] "
355                "-CApath ca_path -CAfile ca_file.pem "
356                "-untrusted cert_file.pem\n");
357  cleanup:
358     /* Clean up. */
359     app_RAND_write_file(NULL, bio_err);
360     NCONF_free(conf);
361     OPENSSL_free(password);
362     OBJ_cleanup();
363     if (free_bio_err) {
364         BIO_free_all(bio_err);
365         bio_err = NULL;
366     }
367
368     OPENSSL_EXIT(ret);
369 }
370
371 /*
372  * Configuration file-related function definitions.
373  */
374
375 static ASN1_OBJECT *txt2obj(const char *oid)
376 {
377     ASN1_OBJECT *oid_obj = NULL;
378
379     if (!(oid_obj = OBJ_txt2obj(oid, 0)))
380         BIO_printf(bio_err, "cannot convert %s to OID\n", oid);
381
382     return oid_obj;
383 }
384
385 static CONF *load_config_file(const char *configfile)
386 {
387     CONF *conf = NULL;
388     long errorline = -1;
389
390     if (!configfile)
391         configfile = getenv("OPENSSL_CONF");
392     if (!configfile)
393         configfile = getenv("SSLEAY_CONF");
394
395     if (configfile &&
396         (!(conf = NCONF_new(NULL)) ||
397          NCONF_load(conf, configfile, &errorline) <= 0)) {
398         if (errorline <= 0)
399             BIO_printf(bio_err, "error loading the config file "
400                        "'%s'\n", configfile);
401         else
402             BIO_printf(bio_err, "error on line %ld of config file "
403                        "'%s'\n", errorline, configfile);
404     }
405
406     if (conf != NULL) {
407         const char *p;
408
409         BIO_printf(bio_err, "Using configuration from %s\n", configfile);
410         p = NCONF_get_string(conf, NULL, ENV_OID_FILE);
411         if (p != NULL) {
412             BIO *oid_bio = BIO_new_file(p, "r");
413             if (!oid_bio)
414                 ERR_print_errors(bio_err);
415             else {
416                 OBJ_create_objects(oid_bio);
417                 BIO_free_all(oid_bio);
418             }
419         } else
420             ERR_clear_error();
421         if (!add_oid_section(bio_err, conf))
422             ERR_print_errors(bio_err);
423     }
424     return conf;
425 }
426
427 /*
428  * Query-related method definitions.
429  */
430
431 static int query_command(const char *data, char *digest, const EVP_MD *md,
432                          const char *policy, int no_nonce,
433                          int cert, const char *in, const char *out, int text)
434 {
435     int ret = 0;
436     TS_REQ *query = NULL;
437     BIO *in_bio = NULL;
438     BIO *data_bio = NULL;
439     BIO *out_bio = NULL;
440
441     /* Build query object either from file or from scratch. */
442     if (in != NULL) {
443         if ((in_bio = BIO_new_file(in, "rb")) == NULL)
444             goto end;
445         query = d2i_TS_REQ_bio(in_bio, NULL);
446     } else {
447         /*
448          * Open the file if no explicit digest bytes were specified.
449          */
450         if (!digest && !(data_bio = BIO_open_with_default(data, "rb", stdin)))
451             goto end;
452         /* Creating the query object. */
453         query = create_query(data_bio, digest, md, policy, no_nonce, cert);
454         /* Saving the random number generator state. */
455     }
456     if (query == NULL)
457         goto end;
458
459     /* Write query either in ASN.1 or in text format. */
460     if ((out_bio = BIO_open_with_default(out, "wb", stdout)) == NULL)
461         goto end;
462     if (text) {
463         /* Text output. */
464         if (!TS_REQ_print_bio(out_bio, query))
465             goto end;
466     } else {
467         /* ASN.1 output. */
468         if (!i2d_TS_REQ_bio(out_bio, query))
469             goto end;
470     }
471
472     ret = 1;
473
474  end:
475     ERR_print_errors(bio_err);
476
477     /* Clean up. */
478     BIO_free_all(in_bio);
479     BIO_free_all(data_bio);
480     BIO_free_all(out_bio);
481     TS_REQ_free(query);
482
483     return ret;
484 }
485
486 static BIO *BIO_open_with_default(const char *file, const char *mode,
487                                   FILE *default_fp)
488 {
489     return file == NULL ? BIO_new_fp(default_fp, BIO_NOCLOSE)
490         : BIO_new_file(file, mode);
491 }
492
493 static TS_REQ *create_query(BIO *data_bio, char *digest, const EVP_MD *md,
494                             const char *policy, int no_nonce, int cert)
495 {
496     int ret = 0;
497     TS_REQ *ts_req = NULL;
498     int len;
499     TS_MSG_IMPRINT *msg_imprint = NULL;
500     X509_ALGOR *algo = NULL;
501     unsigned char *data = NULL;
502     ASN1_OBJECT *policy_obj = NULL;
503     ASN1_INTEGER *nonce_asn1 = NULL;
504
505     /* Setting default message digest. */
506     if (!md && !(md = EVP_get_digestbyname("sha1")))
507         goto err;
508
509     /* Creating request object. */
510     if (!(ts_req = TS_REQ_new()))
511         goto err;
512
513     /* Setting version. */
514     if (!TS_REQ_set_version(ts_req, 1))
515         goto err;
516
517     /* Creating and adding MSG_IMPRINT object. */
518     if (!(msg_imprint = TS_MSG_IMPRINT_new()))
519         goto err;
520
521     /* Adding algorithm. */
522     if (!(algo = X509_ALGOR_new()))
523         goto err;
524     if (!(algo->algorithm = OBJ_nid2obj(EVP_MD_type(md))))
525         goto err;
526     if (!(algo->parameter = ASN1_TYPE_new()))
527         goto err;
528     algo->parameter->type = V_ASN1_NULL;
529     if (!TS_MSG_IMPRINT_set_algo(msg_imprint, algo))
530         goto err;
531
532     /* Adding message digest. */
533     if ((len = create_digest(data_bio, digest, md, &data)) == 0)
534         goto err;
535     if (!TS_MSG_IMPRINT_set_msg(msg_imprint, data, len))
536         goto err;
537
538     if (!TS_REQ_set_msg_imprint(ts_req, msg_imprint))
539         goto err;
540
541     /* Setting policy if requested. */
542     if (policy && !(policy_obj = txt2obj(policy)))
543         goto err;
544     if (policy_obj && !TS_REQ_set_policy_id(ts_req, policy_obj))
545         goto err;
546
547     /* Setting nonce if requested. */
548     if (!no_nonce && !(nonce_asn1 = create_nonce(NONCE_LENGTH)))
549         goto err;
550     if (nonce_asn1 && !TS_REQ_set_nonce(ts_req, nonce_asn1))
551         goto err;
552
553     /* Setting certificate request flag if requested. */
554     if (!TS_REQ_set_cert_req(ts_req, cert))
555         goto err;
556
557     ret = 1;
558  err:
559     if (!ret) {
560         TS_REQ_free(ts_req);
561         ts_req = NULL;
562         BIO_printf(bio_err, "could not create query\n");
563     }
564     TS_MSG_IMPRINT_free(msg_imprint);
565     X509_ALGOR_free(algo);
566     OPENSSL_free(data);
567     ASN1_OBJECT_free(policy_obj);
568     ASN1_INTEGER_free(nonce_asn1);
569     return ts_req;
570 }
571
572 static int create_digest(BIO *input, char *digest, const EVP_MD *md,
573                          unsigned char **md_value)
574 {
575     int md_value_len;
576
577     md_value_len = EVP_MD_size(md);
578     if (md_value_len < 0)
579         goto err;
580     if (input) {
581         /* Digest must be computed from an input file. */
582         EVP_MD_CTX md_ctx;
583         unsigned char buffer[4096];
584         int length;
585
586         *md_value = OPENSSL_malloc(md_value_len);
587         if (*md_value == 0)
588             goto err;
589
590         EVP_DigestInit(&md_ctx, md);
591         while ((length = BIO_read(input, buffer, sizeof(buffer))) > 0) {
592             EVP_DigestUpdate(&md_ctx, buffer, length);
593         }
594         if (!EVP_DigestFinal(&md_ctx, *md_value, NULL))
595             return 0;
596     } else {
597         /* Digest bytes are specified with digest. */
598         long digest_len;
599         *md_value = string_to_hex(digest, &digest_len);
600         if (!*md_value || md_value_len != digest_len) {
601             OPENSSL_free(*md_value);
602             *md_value = NULL;
603             BIO_printf(bio_err, "bad digest, %d bytes "
604                        "must be specified\n", md_value_len);
605             goto err;
606         }
607     }
608
609     return md_value_len;
610  err:
611     return 0;
612 }
613
614 static ASN1_INTEGER *create_nonce(int bits)
615 {
616     unsigned char buf[20];
617     ASN1_INTEGER *nonce = NULL;
618     int len = (bits - 1) / 8 + 1;
619     int i;
620
621     /* Generating random byte sequence. */
622     if (len > (int)sizeof(buf))
623         goto err;
624     if (RAND_bytes(buf, len) <= 0)
625         goto err;
626
627     /* Find the first non-zero byte and creating ASN1_INTEGER object. */
628     for (i = 0; i < len && !buf[i]; ++i) ;
629     if (!(nonce = ASN1_INTEGER_new()))
630         goto err;
631     OPENSSL_free(nonce->data);
632     /* Allocate at least one byte. */
633     nonce->length = len - i;
634     if (!(nonce->data = OPENSSL_malloc(nonce->length + 1)))
635         goto err;
636     memcpy(nonce->data, buf + i, nonce->length);
637
638     return nonce;
639  err:
640     BIO_printf(bio_err, "could not create nonce\n");
641     ASN1_INTEGER_free(nonce);
642     return NULL;
643 }
644
645 /*
646  * Reply-related method definitions.
647  */
648
649 static int reply_command(CONF *conf, char *section, char *engine,
650                          char *queryfile, char *passin, char *inkey,
651                          char *signer, char *chain, const char *policy,
652                          char *in, int token_in,
653                          char *out, int token_out, int text)
654 {
655     int ret = 0;
656     TS_RESP *response = NULL;
657     BIO *in_bio = NULL;
658     BIO *query_bio = NULL;
659     BIO *inkey_bio = NULL;
660     BIO *signer_bio = NULL;
661     BIO *out_bio = NULL;
662
663     /* Build response object either from response or query. */
664     if (in != NULL) {
665         if ((in_bio = BIO_new_file(in, "rb")) == NULL)
666             goto end;
667         if (token_in) {
668             /*
669              * We have a ContentInfo (PKCS7) object, add 'granted' status
670              * info around it.
671              */
672             response = read_PKCS7(in_bio);
673         } else {
674             /* We have a ready-made TS_RESP object. */
675             response = d2i_TS_RESP_bio(in_bio, NULL);
676         }
677     } else {
678         response = create_response(conf, section, engine, queryfile,
679                                    passin, inkey, signer, chain, policy);
680         if (response)
681             BIO_printf(bio_err, "Response has been generated.\n");
682         else
683             BIO_printf(bio_err, "Response is not generated.\n");
684     }
685     if (response == NULL)
686         goto end;
687
688     /* Write response either in ASN.1 or text format. */
689     if ((out_bio = BIO_open_with_default(out, "wb", stdout)) == NULL)
690         goto end;
691     if (text) {
692         /* Text output. */
693         if (token_out) {
694             TS_TST_INFO *tst_info = TS_RESP_get_tst_info(response);
695             if (!TS_TST_INFO_print_bio(out_bio, tst_info))
696                 goto end;
697         } else {
698             if (!TS_RESP_print_bio(out_bio, response))
699                 goto end;
700         }
701     } else {
702         /* ASN.1 DER output. */
703         if (token_out) {
704             PKCS7 *token = TS_RESP_get_token(response);
705             if (!i2d_PKCS7_bio(out_bio, token))
706                 goto end;
707         } else {
708             if (!i2d_TS_RESP_bio(out_bio, response))
709                 goto end;
710         }
711     }
712
713     ret = 1;
714
715  end:
716     ERR_print_errors(bio_err);
717
718     /* Clean up. */
719     BIO_free_all(in_bio);
720     BIO_free_all(query_bio);
721     BIO_free_all(inkey_bio);
722     BIO_free_all(signer_bio);
723     BIO_free_all(out_bio);
724     TS_RESP_free(response);
725
726     return ret;
727 }
728
729 /* Reads a PKCS7 token and adds default 'granted' status info to it. */
730 static TS_RESP *read_PKCS7(BIO *in_bio)
731 {
732     int ret = 0;
733     PKCS7 *token = NULL;
734     TS_TST_INFO *tst_info = NULL;
735     TS_RESP *resp = NULL;
736     TS_STATUS_INFO *si = NULL;
737
738     /* Read PKCS7 object and extract the signed time stamp info. */
739     if (!(token = d2i_PKCS7_bio(in_bio, NULL)))
740         goto end;
741     if (!(tst_info = PKCS7_to_TS_TST_INFO(token)))
742         goto end;
743
744     /* Creating response object. */
745     if (!(resp = TS_RESP_new()))
746         goto end;
747
748     /* Create granted status info. */
749     if (!(si = TS_STATUS_INFO_new()))
750         goto end;
751     if (!(ASN1_INTEGER_set(si->status, TS_STATUS_GRANTED)))
752         goto end;
753     if (!TS_RESP_set_status_info(resp, si))
754         goto end;
755
756     /* Setting encapsulated token. */
757     TS_RESP_set_tst_info(resp, token, tst_info);
758     token = NULL;               /* Ownership is lost. */
759     tst_info = NULL;            /* Ownership is lost. */
760
761     ret = 1;
762  end:
763     PKCS7_free(token);
764     TS_TST_INFO_free(tst_info);
765     if (!ret) {
766         TS_RESP_free(resp);
767         resp = NULL;
768     }
769     TS_STATUS_INFO_free(si);
770     return resp;
771 }
772
773 static TS_RESP *create_response(CONF *conf, const char *section, char *engine,
774                                 char *queryfile, char *passin, char *inkey,
775                                 char *signer, char *chain, const char *policy)
776 {
777     int ret = 0;
778     TS_RESP *response = NULL;
779     BIO *query_bio = NULL;
780     TS_RESP_CTX *resp_ctx = NULL;
781
782     if (!(query_bio = BIO_new_file(queryfile, "rb")))
783         goto end;
784
785     /* Getting TSA configuration section. */
786     if (!(section = TS_CONF_get_tsa_section(conf, section)))
787         goto end;
788
789     /* Setting up response generation context. */
790     if (!(resp_ctx = TS_RESP_CTX_new()))
791         goto end;
792
793     /* Setting serial number provider callback. */
794     if (!TS_CONF_set_serial(conf, section, serial_cb, resp_ctx))
795         goto end;
796 #ifndef OPENSSL_NO_ENGINE
797     /* Setting default OpenSSL engine. */
798     if (!TS_CONF_set_crypto_device(conf, section, engine))
799         goto end;
800 #endif
801
802     /* Setting TSA signer certificate. */
803     if (!TS_CONF_set_signer_cert(conf, section, signer, resp_ctx))
804         goto end;
805
806     /* Setting TSA signer certificate chain. */
807     if (!TS_CONF_set_certs(conf, section, chain, resp_ctx))
808         goto end;
809
810     /* Setting TSA signer private key. */
811     if (!TS_CONF_set_signer_key(conf, section, inkey, passin, resp_ctx))
812         goto end;
813
814     /* Setting default policy OID. */
815     if (!TS_CONF_set_def_policy(conf, section, policy, resp_ctx))
816         goto end;
817
818     /* Setting acceptable policy OIDs. */
819     if (!TS_CONF_set_policies(conf, section, resp_ctx))
820         goto end;
821
822     /* Setting the acceptable one-way hash algorithms. */
823     if (!TS_CONF_set_digests(conf, section, resp_ctx))
824         goto end;
825
826     /* Setting guaranteed time stamp accuracy. */
827     if (!TS_CONF_set_accuracy(conf, section, resp_ctx))
828         goto end;
829
830     /* Setting the precision of the time. */
831     if (!TS_CONF_set_clock_precision_digits(conf, section, resp_ctx))
832         goto end;
833
834     /* Setting the ordering flaf if requested. */
835     if (!TS_CONF_set_ordering(conf, section, resp_ctx))
836         goto end;
837
838     /* Setting the TSA name required flag if requested. */
839     if (!TS_CONF_set_tsa_name(conf, section, resp_ctx))
840         goto end;
841
842     /* Setting the ESS cert id chain flag if requested. */
843     if (!TS_CONF_set_ess_cert_id_chain(conf, section, resp_ctx))
844         goto end;
845
846     /* Creating the response. */
847     if (!(response = TS_RESP_create_response(resp_ctx, query_bio)))
848         goto end;
849
850     ret = 1;
851  end:
852     if (!ret) {
853         TS_RESP_free(response);
854         response = NULL;
855     }
856     TS_RESP_CTX_free(resp_ctx);
857     BIO_free_all(query_bio);
858
859     return response;
860 }
861
862 static ASN1_INTEGER *serial_cb(TS_RESP_CTX *ctx, void *data)
863 {
864     const char *serial_file = (const char *)data;
865     ASN1_INTEGER *serial = next_serial(serial_file);
866
867     if (!serial) {
868         TS_RESP_CTX_set_status_info(ctx, TS_STATUS_REJECTION,
869                                     "Error during serial number "
870                                     "generation.");
871         TS_RESP_CTX_add_failure_info(ctx, TS_INFO_ADD_INFO_NOT_AVAILABLE);
872     } else
873         save_ts_serial(serial_file, serial);
874
875     return serial;
876 }
877
878 static ASN1_INTEGER *next_serial(const char *serialfile)
879 {
880     int ret = 0;
881     BIO *in = NULL;
882     ASN1_INTEGER *serial = NULL;
883     BIGNUM *bn = NULL;
884
885     if (!(serial = ASN1_INTEGER_new()))
886         goto err;
887
888     if (!(in = BIO_new_file(serialfile, "r"))) {
889         ERR_clear_error();
890         BIO_printf(bio_err, "Warning: could not open file %s for "
891                    "reading, using serial number: 1\n", serialfile);
892         if (!ASN1_INTEGER_set(serial, 1))
893             goto err;
894     } else {
895         char buf[1024];
896         if (!a2i_ASN1_INTEGER(in, serial, buf, sizeof(buf))) {
897             BIO_printf(bio_err, "unable to load number from %s\n",
898                        serialfile);
899             goto err;
900         }
901         if (!(bn = ASN1_INTEGER_to_BN(serial, NULL)))
902             goto err;
903         ASN1_INTEGER_free(serial);
904         serial = NULL;
905         if (!BN_add_word(bn, 1))
906             goto err;
907         if (!(serial = BN_to_ASN1_INTEGER(bn, NULL)))
908             goto err;
909     }
910     ret = 1;
911  err:
912     if (!ret) {
913         ASN1_INTEGER_free(serial);
914         serial = NULL;
915     }
916     BIO_free_all(in);
917     BN_free(bn);
918     return serial;
919 }
920
921 static int save_ts_serial(const char *serialfile, ASN1_INTEGER *serial)
922 {
923     int ret = 0;
924     BIO *out = NULL;
925
926     if (!(out = BIO_new_file(serialfile, "w")))
927         goto err;
928     if (i2a_ASN1_INTEGER(out, serial) <= 0)
929         goto err;
930     if (BIO_puts(out, "\n") <= 0)
931         goto err;
932     ret = 1;
933  err:
934     if (!ret)
935         BIO_printf(bio_err, "could not save serial number to %s\n",
936                    serialfile);
937     BIO_free_all(out);
938     return ret;
939 }
940
941 /*
942  * Verify-related method definitions.
943  */
944
945 static int verify_command(char *data, char *digest, char *queryfile,
946                           char *in, int token_in,
947                           char *ca_path, char *ca_file, char *untrusted)
948 {
949     BIO *in_bio = NULL;
950     PKCS7 *token = NULL;
951     TS_RESP *response = NULL;
952     TS_VERIFY_CTX *verify_ctx = NULL;
953     int ret = 0;
954
955     /* Decode the token (PKCS7) or response (TS_RESP) files. */
956     if (!(in_bio = BIO_new_file(in, "rb")))
957         goto end;
958     if (token_in) {
959         if (!(token = d2i_PKCS7_bio(in_bio, NULL)))
960             goto end;
961     } else {
962         if (!(response = d2i_TS_RESP_bio(in_bio, NULL)))
963             goto end;
964     }
965
966     if (!(verify_ctx = create_verify_ctx(data, digest, queryfile,
967                                          ca_path, ca_file, untrusted)))
968         goto end;
969
970     /* Checking the token or response against the request. */
971     ret = token_in ?
972         TS_RESP_verify_token(verify_ctx, token) :
973         TS_RESP_verify_response(verify_ctx, response);
974
975  end:
976     printf("Verification: ");
977     if (ret)
978         printf("OK\n");
979     else {
980         printf("FAILED\n");
981         /* Print errors, if there are any. */
982         ERR_print_errors(bio_err);
983     }
984
985     /* Clean up. */
986     BIO_free_all(in_bio);
987     PKCS7_free(token);
988     TS_RESP_free(response);
989     TS_VERIFY_CTX_free(verify_ctx);
990     return ret;
991 }
992
993 static TS_VERIFY_CTX *create_verify_ctx(char *data, char *digest,
994                                         char *queryfile,
995                                         char *ca_path, char *ca_file,
996                                         char *untrusted)
997 {
998     TS_VERIFY_CTX *ctx = NULL;
999     BIO *input = NULL;
1000     TS_REQ *request = NULL;
1001     int ret = 0;
1002
1003     if (data != NULL || digest != NULL) {
1004         if (!(ctx = TS_VERIFY_CTX_new()))
1005             goto err;
1006         ctx->flags = TS_VFY_VERSION | TS_VFY_SIGNER;
1007         if (data != NULL) {
1008             ctx->flags |= TS_VFY_DATA;
1009             if (!(ctx->data = BIO_new_file(data, "rb")))
1010                 goto err;
1011         } else if (digest != NULL) {
1012             long imprint_len;
1013             ctx->flags |= TS_VFY_IMPRINT;
1014             if (!(ctx->imprint = string_to_hex(digest, &imprint_len))) {
1015                 BIO_printf(bio_err, "invalid digest string\n");
1016                 goto err;
1017             }
1018             ctx->imprint_len = imprint_len;
1019         }
1020
1021     } else if (queryfile != NULL) {
1022         /*
1023          * The request has just to be read, decoded and converted to a verify
1024          * context object.
1025          */
1026         if (!(input = BIO_new_file(queryfile, "rb")))
1027             goto err;
1028         if (!(request = d2i_TS_REQ_bio(input, NULL)))
1029             goto err;
1030         if (!(ctx = TS_REQ_to_TS_VERIFY_CTX(request, NULL)))
1031             goto err;
1032     } else
1033         return NULL;
1034
1035     /* Add the signature verification flag and arguments. */
1036     ctx->flags |= TS_VFY_SIGNATURE;
1037
1038     /* Initialising the X509_STORE object. */
1039     if (!(ctx->store = create_cert_store(ca_path, ca_file)))
1040         goto err;
1041
1042     /* Loading untrusted certificates. */
1043     if (untrusted && !(ctx->certs = TS_CONF_load_certs(untrusted)))
1044         goto err;
1045
1046     ret = 1;
1047  err:
1048     if (!ret) {
1049         TS_VERIFY_CTX_free(ctx);
1050         ctx = NULL;
1051     }
1052     BIO_free_all(input);
1053     TS_REQ_free(request);
1054     return ctx;
1055 }
1056
1057 static X509_STORE *create_cert_store(char *ca_path, char *ca_file)
1058 {
1059     X509_STORE *cert_ctx = NULL;
1060     X509_LOOKUP *lookup = NULL;
1061     int i;
1062
1063     /* Creating the X509_STORE object. */
1064     cert_ctx = X509_STORE_new();
1065
1066     /* Setting the callback for certificate chain verification. */
1067     X509_STORE_set_verify_cb(cert_ctx, verify_cb);
1068
1069     /* Adding a trusted certificate directory source. */
1070     if (ca_path) {
1071         lookup = X509_STORE_add_lookup(cert_ctx, X509_LOOKUP_hash_dir());
1072         if (lookup == NULL) {
1073             BIO_printf(bio_err, "memory allocation failure\n");
1074             goto err;
1075         }
1076         i = X509_LOOKUP_add_dir(lookup, ca_path, X509_FILETYPE_PEM);
1077         if (!i) {
1078             BIO_printf(bio_err, "Error loading directory %s\n", ca_path);
1079             goto err;
1080         }
1081     }
1082
1083     /* Adding a trusted certificate file source. */
1084     if (ca_file) {
1085         lookup = X509_STORE_add_lookup(cert_ctx, X509_LOOKUP_file());
1086         if (lookup == NULL) {
1087             BIO_printf(bio_err, "memory allocation failure\n");
1088             goto err;
1089         }
1090         i = X509_LOOKUP_load_file(lookup, ca_file, X509_FILETYPE_PEM);
1091         if (!i) {
1092             BIO_printf(bio_err, "Error loading file %s\n", ca_file);
1093             goto err;
1094         }
1095     }
1096
1097     return cert_ctx;
1098  err:
1099     X509_STORE_free(cert_ctx);
1100     return NULL;
1101 }
1102
1103 static int verify_cb(int ok, X509_STORE_CTX *ctx)
1104 {
1105     /*-
1106     char buf[256];
1107
1108     if (!ok)
1109             {
1110             X509_NAME_oneline(X509_get_subject_name(ctx->current_cert),
1111                               buf, sizeof(buf));
1112             printf("%s\n", buf);
1113             printf("error %d at %d depth lookup: %s\n",
1114                    ctx->error, ctx->error_depth,
1115                     X509_verify_cert_error_string(ctx->error));
1116             }
1117     */
1118
1119     return ok;
1120 }