apps/dsaparam.c

   1 /*
   2  * Copyright 1995-2016 The OpenSSL Project Authors. All Rights Reserved.
   3  *
   4  * Licensed under the OpenSSL license (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 #ifdef OPENSSL_NO_DSA
  12 NON_EMPTY_TRANSLATION_UNIT
  13 #else
  14
  15 # include <stdio.h>
  16 # include <stdlib.h>
  17 # include <time.h>
  18 # include <string.h>
  19 # include "apps.h"
  20 # include <openssl/bio.h>
  21 # include <openssl/err.h>
  22 # include <openssl/bn.h>
  23 # include <openssl/dsa.h>
  24 # include <openssl/x509.h>
  25 # include <openssl/pem.h>
  26
  27 # ifdef GENCB_TEST
  28
  29 static int stop_keygen_flag = 0;
  30
  31 static void timebomb_sigalarm(int foo)
  32 {
  33     stop_keygen_flag = 1;
  34 }
  35
  36 # endif
  37
  38 static int dsa_cb(int p, int n, BN_GENCB *cb);
  39
  40 typedef enum OPTION_choice {
  41     OPT_ERR = -1, OPT_EOF = 0, OPT_HELP,
  42     OPT_INFORM, OPT_OUTFORM, OPT_IN, OPT_OUT, OPT_TEXT, OPT_C,
  43     OPT_NOOUT, OPT_GENKEY, OPT_RAND, OPT_ENGINE,
  44     OPT_TIMEBOMB
  45 } OPTION_CHOICE;
  46
  47 OPTIONS dsaparam_options[] = {
  48     {"help", OPT_HELP, '-', "Display this summary"},
  49     {"inform", OPT_INFORM, 'F', "Input format - DER or PEM"},
  50     {"in", OPT_IN, '<', "Input file"},
  51     {"outform", OPT_OUTFORM, 'F', "Output format - DER or PEM"},
  52     {"out", OPT_OUT, '>', "Output file"},
  53     {"text", OPT_TEXT, '-', "Print as text"},
  54     {"C", OPT_C, '-', "Output C code"},
  55     {"noout", OPT_NOOUT, '-', "No output"},
  56     {"genkey", OPT_GENKEY, '-', "Generate a DSA key"},
  57     {"rand", OPT_RAND, 's', "Files to use for random number input"},
  58 # ifdef GENCB_TEST
  59     {"timebomb", OPT_TIMEBOMB, 'p', "Interrupt keygen after 'pnum' seconds"},
  60 # endif
  61 # ifndef OPENSSL_NO_ENGINE
  62     {"engine", OPT_ENGINE, 's', "Use engine e, possibly a hardware device"},
  63 # endif
  64     {NULL}
  65 };
  66
  67 int dsaparam_main(int argc, char **argv)
  68 {
  69     DSA *dsa = NULL;
  70     BIO *in = NULL, *out = NULL;
  71     BN_GENCB *cb = NULL;
  72     int numbits = -1, num = 0, genkey = 0, need_rand = 0;
  73     int informat = FORMAT_PEM, outformat = FORMAT_PEM, noout = 0, C = 0;
  74     int ret = 1, i, text = 0, private = 0;
  75 # ifdef GENCB_TEST
  76     int timebomb = 0;
  77 # endif
  78     char *infile = NULL, *outfile = NULL, *prog, *inrand = NULL;
  79     OPTION_CHOICE o;
  80
  81     prog = opt_init(argc, argv, dsaparam_options);
  82     while ((o = opt_next()) != OPT_EOF) {
  83         switch (o) {
  84         case OPT_EOF:
  85         case OPT_ERR:
  86  opthelp:
  87             BIO_printf(bio_err, "%s: Use -help for summary.\n", prog);
  88             goto end;
  89         case OPT_HELP:
  90             opt_help(dsaparam_options);
  91             ret = 0;
  92             goto end;
  93         case OPT_INFORM:
  94             if (!opt_format(opt_arg(), OPT_FMT_PEMDER, &informat))
  95                 goto opthelp;
  96             break;
  97         case OPT_IN:
  98             infile = opt_arg();
  99             break;
 100         case OPT_OUTFORM:
 101             if (!opt_format(opt_arg(), OPT_FMT_PEMDER, &outformat))
 102                 goto opthelp;
 103             break;
 104         case OPT_OUT:
 105             outfile = opt_arg();
 106             break;
 107         case OPT_ENGINE:
 108             (void)setup_engine(opt_arg(), 0);
 109             break;
 110         case OPT_TIMEBOMB:
 111 # ifdef GENCB_TEST
 112             timebomb = atoi(opt_arg());
 113             break;
 114 # endif
 115         case OPT_TEXT:
 116             text = 1;
 117             break;
 118         case OPT_C:
 119             C = 1;
 120             break;
 121         case OPT_GENKEY:
 122             genkey = need_rand = 1;
 123             break;
 124         case OPT_RAND:
 125             inrand = opt_arg();
 126             need_rand = 1;
 127             break;
 128         case OPT_NOOUT:
 129             noout = 1;
 130             break;
 131         }
 132     }
 133     argc = opt_num_rest();
 134     argv = opt_rest();
 135
 136     if (argc == 1) {
 137         if (!opt_int(argv[0], &num) || num < 0)
 138             goto end;
 139         /* generate a key */
 140         numbits = num;
 141         need_rand = 1;
 142     }
 143     private = genkey ? 1 : 0;
 144
 145     in = bio_open_default(infile, 'r', informat);
 146     if (in == NULL)
 147         goto end;
 148     out = bio_open_owner(outfile, outformat, private);
 149     if (out == NULL)
 150         goto end;
 151
 152     if (need_rand) {
 153         app_RAND_load_file(NULL, (inrand != NULL));
 154         if (inrand != NULL)
 155             BIO_printf(bio_err, "%ld semi-random bytes loaded\n",
 156                        app_RAND_load_files(inrand));
 157     }
 158
 159     if (numbits > 0) {
 160         cb = BN_GENCB_new();
 161         if (cb == NULL) {
 162             BIO_printf(bio_err, "Error allocating BN_GENCB object\n");
 163             goto end;
 164         }
 165         BN_GENCB_set(cb, dsa_cb, bio_err);
 166         assert(need_rand);
 167         dsa = DSA_new();
 168         if (dsa == NULL) {
 169             BIO_printf(bio_err, "Error allocating DSA object\n");
 170             goto end;
 171         }
 172         BIO_printf(bio_err, "Generating DSA parameters, %d bit long prime\n",
 173                    num);
 174         BIO_printf(bio_err, "This could take some time\n");
 175 # ifdef GENCB_TEST
 176         if (timebomb > 0) {
 177             struct sigaction act;
 178             act.sa_handler = timebomb_sigalarm;
 179             act.sa_flags = 0;
 180             BIO_printf(bio_err,
 181                        "(though I'll stop it if not done within %d secs)\n",
 182                        timebomb);
 183             if (sigaction(SIGALRM, &act, NULL) != 0) {
 184                 BIO_printf(bio_err, "Error, couldn't set SIGALRM handler\n");
 185                 goto end;
 186             }
 187             alarm(timebomb);
 188         }
 189 # endif
 190         if (!DSA_generate_parameters_ex(dsa, num, NULL, 0, NULL, NULL, cb)) {
 191 # ifdef GENCB_TEST
 192             if (stop_keygen_flag) {
 193                 BIO_printf(bio_err, "DSA key generation time-stopped\n");
 194                 /* This is an asked-for behaviour! */
 195                 ret = 0;
 196                 goto end;
 197             }
 198 # endif
 199             ERR_print_errors(bio_err);
 200             BIO_printf(bio_err, "Error, DSA key generation failed\n");
 201             goto end;
 202         }
 203     } else if (informat == FORMAT_ASN1)
 204         dsa = d2i_DSAparams_bio(in, NULL);
 205     else
 206         dsa = PEM_read_bio_DSAparams(in, NULL, NULL, NULL);
 207     if (dsa == NULL) {
 208         BIO_printf(bio_err, "unable to load DSA parameters\n");
 209         ERR_print_errors(bio_err);
 210         goto end;
 211     }
 212
 213     if (text) {
 214         DSAparams_print(out, dsa);
 215     }
 216
 217     if (C) {
 218         const BIGNUM *p = NULL, *q = NULL, *g = NULL;
 219         unsigned char *data;
 220         int len, bits_p;
 221
 222         DSA_get0_pqg(dsa, &p, &q, &g);
 223         len = BN_num_bytes(p);
 224         bits_p = BN_num_bits(p);
 225
 226         data = app_malloc(len + 20, "BN space");
 227
 228         BIO_printf(bio_out, "DSA *get_dsa%d()\n{\n", bits_p);
 229         print_bignum_var(bio_out, p, "dsap", len, data);
 230         print_bignum_var(bio_out, q, "dsaq", len, data);
 231         print_bignum_var(bio_out, g, "dsag", len, data);
 232         BIO_printf(bio_out, "    DSA *dsa = DSA_new();\n"
 233                             "\n");
 234         BIO_printf(bio_out, "    if (dsa == NULL)\n"
 235                             "        return NULL;\n");
 236         BIO_printf(bio_out, "    dsa->p = BN_bin2bn(dsap_%d, sizeof (dsap_%d), NULL);\n",
 237                bits_p, bits_p);
 238         BIO_printf(bio_out, "    dsa->q = BN_bin2bn(dsaq_%d, sizeof (dsaq_%d), NULL);\n",
 239                bits_p, bits_p);
 240         BIO_printf(bio_out, "    dsa->g = BN_bin2bn(dsag_%d, sizeof (dsag_%d), NULL);\n",
 241                bits_p, bits_p);
 242         BIO_printf(bio_out, "    if (!dsa->p || !dsa->q || !dsa->g) {\n"
 243                             "        DSA_free(dsa);\n"
 244                             "        return NULL;\n"
 245                             "    }\n"
 246                             "    return(dsa);\n}\n");
 247         OPENSSL_free(data);
 248     }
 249
 250     if (!noout) {
 251         if (outformat == FORMAT_ASN1)
 252             i = i2d_DSAparams_bio(out, dsa);
 253         else
 254             i = PEM_write_bio_DSAparams(out, dsa);
 255         if (!i) {
 256             BIO_printf(bio_err, "unable to write DSA parameters\n");
 257             ERR_print_errors(bio_err);
 258             goto end;
 259         }
 260     }
 261     if (genkey) {
 262         DSA *dsakey;
 263
 264         assert(need_rand);
 265         if ((dsakey = DSAparams_dup(dsa)) == NULL)
 266             goto end;
 267         if (!DSA_generate_key(dsakey)) {
 268             ERR_print_errors(bio_err);
 269             DSA_free(dsakey);
 270             goto end;
 271         }
 272         assert(private);
 273         if (outformat == FORMAT_ASN1)
 274             i = i2d_DSAPrivateKey_bio(out, dsakey);
 275         else
 276             i = PEM_write_bio_DSAPrivateKey(out, dsakey, NULL, NULL, 0, NULL,
 277                                             NULL);
 278         DSA_free(dsakey);
 279     }
 280     if (need_rand)
 281         app_RAND_write_file(NULL);
 282     ret = 0;
 283  end:
 284     BN_GENCB_free(cb);
 285     BIO_free(in);
 286     BIO_free_all(out);
 287     DSA_free(dsa);
 288     return (ret);
 289 }
 290
 291 static int dsa_cb(int p, int n, BN_GENCB *cb)
 292 {
 293     char c = '*';
 294
 295     if (p == 0)
 296         c = '.';
 297     if (p == 1)
 298         c = '+';
 299     if (p == 2)
 300         c = '*';
 301     if (p == 3)
 302         c = '\n';
 303     BIO_write(BN_GENCB_get_arg(cb), &c, 1);
 304     (void)BIO_flush(BN_GENCB_get_arg(cb));
 305 # ifdef GENCB_TEST
 306     if (stop_keygen_flag)
 307         return 0;
 308 # endif
 309     return 1;
 310 }
 311 #endif