Reformat all code using astyle.
[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
62                 break;
63
64         default:
65                 fprintf(stderr, "?");
66         }
67
68         return 1;
69 }
70
71 // Generate RSA key
72
73 #ifndef HAVE_BN_GENCB_NEW
74 BN_GENCB *BN_GENCB_new(void) {
75         return xzalloc(sizeof(BN_GENCB));
76 }
77
78 void BN_GENCB_free(BN_GENCB *cb) {
79         free(cb);
80 }
81 #endif
82
83 rsa_t *rsa_generate(size_t bits, unsigned long exponent) {
84         BIGNUM *bn_e = BN_new();
85         rsa_t *rsa = RSA_new();
86         BN_GENCB *cb = BN_GENCB_new();
87
88         if(!bn_e || !rsa || !cb) {
89                 abort();
90         }
91
92         BN_set_word(bn_e, exponent);
93         BN_GENCB_set(cb, indicator, NULL);
94
95         int result = RSA_generate_key_ex(rsa, bits, bn_e, cb);
96
97         BN_GENCB_free(cb);
98         BN_free(bn_e);
99
100         if(!result) {
101                 fprintf(stderr, "Error during key generation!\n");
102                 RSA_free(rsa);
103                 return NULL;
104         }
105
106         return rsa;
107 }
108
109 // Write PEM RSA keys
110
111 bool rsa_write_pem_public_key(rsa_t *rsa, FILE *fp) {
112         return PEM_write_RSAPublicKey(fp, rsa);
113 }
114
115 bool rsa_write_pem_private_key(rsa_t *rsa, FILE *fp) {
116         return PEM_write_RSAPrivateKey(fp, rsa, NULL, NULL, 0, NULL, NULL);
117 }