3674057c5da9af495e17c15285ff40d71b3039d7
[tinc] / src / openssl / rsagen.c
1 /*
2     rsagen.c -- RSA key generation and export
3     Copyright (C) 2008-2022 Guus Sliepen <guus@tinc-vpn.org>
4
5     This program is free software; you can redistribute it and/or modify
6     it under the terms of the GNU General Public License as published by
7     the Free Software Foundation; either version 2 of the License, or
8     (at your option) any later version.
9
10     This program is distributed in the hope that it will be useful,
11     but WITHOUT ANY WARRANTY; without even the implied warranty of
12     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
13     GNU General Public License for more details.
14
15     You should have received a copy of the GNU General Public License along
16     with this program; if not, write to the Free Software Foundation, Inc.,
17     51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
18 */
19
20 #include "../system.h"
21
22 #include <openssl/pem.h>
23 #include <openssl/err.h>
24
25 #define TINC_RSA_INTERNAL
26 typedef RSA rsa_t;
27
28 #include "../logger.h"
29 #include "../rsagen.h"
30
31 /* This function prettyprints the key generation process */
32
33 static int indicator(int a, int b, BN_GENCB *cb) {
34         (void)cb;
35
36         switch(a) {
37         case 0:
38                 fprintf(stderr, ".");
39                 break;
40
41         case 1:
42                 fprintf(stderr, "+");
43                 break;
44
45         case 2:
46                 fprintf(stderr, "-");
47                 break;
48
49         case 3:
50                 switch(b) {
51                 case 0:
52                         fprintf(stderr, " p\n");
53                         break;
54
55                 case 1:
56                         fprintf(stderr, " q\n");
57                         break;
58
59                 default:
60                         fprintf(stderr, "?");
61                 }
62
63                 break;
64
65         default:
66                 fprintf(stderr, "?");
67         }
68
69         return 1;
70 }
71
72 // Generate RSA key
73
74 rsa_t *rsa_generate(size_t bits, unsigned long exponent) {
75         BIGNUM *bn_e = BN_new();
76         rsa_t *rsa = RSA_new();
77         BN_GENCB *cb = BN_GENCB_new();
78
79         if(!bn_e || !rsa || !cb) {
80                 abort();
81         }
82
83         BN_set_word(bn_e, exponent);
84         BN_GENCB_set(cb, indicator, NULL);
85
86         int result = RSA_generate_key_ex(rsa, (int) bits, bn_e, cb);
87
88         BN_GENCB_free(cb);
89         BN_free(bn_e);
90
91         if(!result) {
92                 fprintf(stderr, "Error during key generation!\n");
93                 RSA_free(rsa);
94                 return NULL;
95         }
96
97         return rsa;
98 }
99
100 // Write PEM RSA keys
101
102 bool rsa_write_pem_public_key(rsa_t *rsa, FILE *fp) {
103         return PEM_write_RSAPublicKey(fp, rsa);
104 }
105
106 bool rsa_write_pem_private_key(rsa_t *rsa, FILE *fp) {
107         return PEM_write_RSAPrivateKey(fp, rsa, NULL, NULL, 0, NULL, NULL);
108 }