Update all header guards.
[tinc] / src / openssl / rsagen.c
1 /*
2     rsagen.c -- RSA key generation and export
3     Copyright (C) 2008-2013 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 #include "../xalloc.h"
31
32 /* This function prettyprints the key generation process */
33
34 static int indicator(int a, int b, BN_GENCB *cb) {
35         switch (a) {
36                 case 0:
37                         fprintf(stderr, ".");
38                         break;
39
40                 case 1:
41                         fprintf(stderr, "+");
42                         break;
43
44                 case 2:
45                         fprintf(stderr, "-");
46                         break;
47
48                 case 3:
49                         switch (b) {
50                                 case 0:
51                                         fprintf(stderr, " p\n");
52                                         break;
53
54                                 case 1:
55                                         fprintf(stderr, " q\n");
56                                         break;
57
58                                 default:
59                                         fprintf(stderr, "?");
60                         }
61                         break;
62
63                 default:
64                         fprintf(stderr, "?");
65         }
66
67         return 1;
68 }
69
70 // Generate RSA key
71
72 #ifndef HAVE_BN_GENCB_NEW
73 BN_GENCB *BN_GENCB_new(void) {
74         return xzalloc(sizeof(BN_GENCB));
75 }
76
77 void BN_GENCB_free(BN_GENCB *cb) {
78         free(cb);
79 }
80 #endif
81
82 rsa_t *rsa_generate(size_t bits, unsigned long exponent) {
83         BIGNUM *bn_e = BN_new();
84         rsa_t *rsa = RSA_new();
85         BN_GENCB *cb = BN_GENCB_new();
86
87         if(!bn_e || !rsa || !cb)
88                 abort();
89
90         BN_set_word(bn_e, exponent);
91         BN_GENCB_set(cb, indicator, NULL);
92
93         int result = RSA_generate_key_ex(rsa, bits, bn_e, cb);
94
95         BN_GENCB_free(cb);
96         BN_free(bn_e);
97
98         if(!result) {
99                 fprintf(stderr, "Error during key generation!\n");
100                 RSA_free(rsa);
101                 return NULL;
102         }
103
104         return rsa;
105 }
106
107 // Write PEM RSA keys
108
109 bool rsa_write_pem_public_key(rsa_t *rsa, FILE *fp) {
110         return PEM_write_RSAPublicKey(fp, rsa);
111 }
112
113 bool rsa_write_pem_private_key(rsa_t *rsa, FILE *fp) {
114         return PEM_write_RSAPrivateKey(fp, rsa, NULL, NULL, 0, NULL, NULL);
115 }