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