1/*
2 * Copyright 2004-2021 The OpenSSL Project Authors. All Rights Reserved.
3 *
4 * Licensed under the Apache License 2.0 (the "License").  You may not use
5 * this file except in compliance with the License.  You can obtain a copy
6 * in the file LICENSE in the source distribution or at
7 * https://www.openssl.org/source/license.html
8 */
9
10#include <string.h>
11
12#include "apps.h"
13#include "progs.h"
14#include <openssl/bn.h>
15
16typedef enum OPTION_choice {
17    OPT_COMMON,
18    OPT_HEX, OPT_GENERATE, OPT_BITS, OPT_SAFE, OPT_CHECKS,
19    OPT_PROV_ENUM
20} OPTION_CHOICE;
21
22const OPTIONS prime_options[] = {
23    {OPT_HELP_STR, 1, '-', "Usage: %s [options] [number...]\n"},
24
25    OPT_SECTION("General"),
26    {"help", OPT_HELP, '-', "Display this summary"},
27    {"bits", OPT_BITS, 'p', "Size of number in bits"},
28    {"checks", OPT_CHECKS, 'p', "Number of checks"},
29
30    OPT_SECTION("Output"),
31    {"hex", OPT_HEX, '-', "Hex output"},
32    {"generate", OPT_GENERATE, '-', "Generate a prime"},
33    {"safe", OPT_SAFE, '-',
34     "When used with -generate, generate a safe prime"},
35
36    OPT_PROV_OPTIONS,
37
38    OPT_PARAMETERS(),
39    {"number", 0, 0, "Number(s) to check for primality if not generating"},
40    {NULL}
41};
42
43int prime_main(int argc, char **argv)
44{
45    BIGNUM *bn = NULL;
46    int hex = 0, generate = 0, bits = 0, safe = 0, ret = 1;
47    char *prog;
48    OPTION_CHOICE o;
49
50    prog = opt_init(argc, argv, prime_options);
51    while ((o = opt_next()) != OPT_EOF) {
52        switch (o) {
53        case OPT_EOF:
54        case OPT_ERR:
55opthelp:
56            BIO_printf(bio_err, "%s: Use -help for summary.\n", prog);
57            goto end;
58        case OPT_HELP:
59            opt_help(prime_options);
60            ret = 0;
61            goto end;
62        case OPT_HEX:
63            hex = 1;
64            break;
65        case OPT_GENERATE:
66            generate = 1;
67            break;
68        case OPT_BITS:
69            bits = atoi(opt_arg());
70            break;
71        case OPT_SAFE:
72            safe = 1;
73            break;
74        case OPT_CHECKS:
75            /* ignore parameter and argument */
76            opt_arg();
77            break;
78        case OPT_PROV_CASES:
79            if (!opt_provider(o))
80                goto end;
81            break;
82        }
83    }
84
85    /* Optional arguments are numbers to check. */
86    argc = opt_num_rest();
87    argv = opt_rest();
88    if (generate) {
89        if (argc != 0)
90            goto opthelp;
91    } else if (argc == 0) {
92        goto opthelp;
93    }
94
95    if (generate) {
96        char *s;
97
98        if (!bits) {
99            BIO_printf(bio_err, "Specify the number of bits.\n");
100            goto end;
101        }
102        bn = BN_new();
103        if (bn == NULL) {
104            BIO_printf(bio_err, "Out of memory.\n");
105            goto end;
106        }
107        if (!BN_generate_prime_ex(bn, bits, safe, NULL, NULL, NULL)) {
108            BIO_printf(bio_err, "Failed to generate prime.\n");
109            goto end;
110        }
111        s = hex ? BN_bn2hex(bn) : BN_bn2dec(bn);
112        if (s == NULL) {
113            BIO_printf(bio_err, "Out of memory.\n");
114            goto end;
115        }
116        BIO_printf(bio_out, "%s\n", s);
117        OPENSSL_free(s);
118    } else {
119        for ( ; *argv; argv++) {
120            int r;
121
122            if (hex)
123                r = BN_hex2bn(&bn, argv[0]);
124            else
125                r = BN_dec2bn(&bn, argv[0]);
126
127            if (!r) {
128                BIO_printf(bio_err, "Failed to process value (%s)\n", argv[0]);
129                goto end;
130            }
131
132            BN_print(bio_out, bn);
133            BIO_printf(bio_out, " (%s) %s prime\n",
134                       argv[0],
135                       BN_check_prime(bn, NULL, NULL)
136                           ? "is" : "is not");
137        }
138    }
139
140    ret = 0;
141 end:
142    BN_free(bn);
143    return ret;
144}
145