37f232f61adf342cf3b65edcc30a22f4523d66d1
[tinc] / src / gcrypt / cipher.c
1 /*
2     cipher.c -- Symmetric block cipher handling
3     Copyright (C) 2007-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 "cipher.h"
23 #include "../cipher.h"
24 #include "../logger.h"
25 #include "../xalloc.h"
26
27 typedef enum gcry_cipher_algos cipher_algo_t;
28 typedef enum gcry_cipher_modes cipher_mode_t;
29
30 static struct {
31         const char *name;
32         cipher_algo_t algo;
33         cipher_mode_t mode;
34         nid_t nid;
35 } ciphertable[] = {
36         {"none", GCRY_CIPHER_NONE, GCRY_CIPHER_MODE_NONE, 0},
37
38         {NULL,       GCRY_CIPHER_BLOWFISH, GCRY_CIPHER_MODE_ECB, 92},
39         {"blowfish", GCRY_CIPHER_BLOWFISH, GCRY_CIPHER_MODE_CBC, 91},
40         {NULL,       GCRY_CIPHER_BLOWFISH, GCRY_CIPHER_MODE_CFB, 93},
41         {NULL,       GCRY_CIPHER_BLOWFISH, GCRY_CIPHER_MODE_OFB, 94},
42
43         {"aes-128-ecb", GCRY_CIPHER_AES, GCRY_CIPHER_MODE_ECB, 418},
44         {"aes-128-cbc", GCRY_CIPHER_AES, GCRY_CIPHER_MODE_CBC, 419},
45         {"aes-128-cfb", GCRY_CIPHER_AES, GCRY_CIPHER_MODE_CFB, 421},
46         {"aes-128-ofb", GCRY_CIPHER_AES, GCRY_CIPHER_MODE_OFB, 420},
47
48         {"aes-192-ecb", GCRY_CIPHER_AES192, GCRY_CIPHER_MODE_ECB, 422},
49         {"aes-192-cbc", GCRY_CIPHER_AES192, GCRY_CIPHER_MODE_CBC, 423},
50         {"aes-192-cfb", GCRY_CIPHER_AES192, GCRY_CIPHER_MODE_CFB, 425},
51         {"aes-192-ofb", GCRY_CIPHER_AES192, GCRY_CIPHER_MODE_OFB, 424},
52
53         {"aes-256-ecb", GCRY_CIPHER_AES256, GCRY_CIPHER_MODE_ECB, 426},
54         {"aes-256-cbc", GCRY_CIPHER_AES256, GCRY_CIPHER_MODE_CBC, 427},
55         {"aes-256-cfb", GCRY_CIPHER_AES256, GCRY_CIPHER_MODE_CFB, 429},
56         {"aes-256-ofb", GCRY_CIPHER_AES256, GCRY_CIPHER_MODE_OFB, 428},
57 };
58
59 static bool nametocipher(const char *name, cipher_algo_t *algo, cipher_mode_t *mode) {
60         for(size_t i = 0; i < sizeof(ciphertable) / sizeof(*ciphertable); i++) {
61                 if(ciphertable[i].name && !strcasecmp(name, ciphertable[i].name)) {
62                         *algo = ciphertable[i].algo;
63                         *mode = ciphertable[i].mode;
64                         return true;
65                 }
66         }
67
68         return false;
69 }
70
71 static bool nidtocipher(cipher_algo_t *algo, cipher_mode_t *mode, nid_t nid) {
72         for(size_t i = 0; i < sizeof(ciphertable) / sizeof(*ciphertable); i++) {
73                 if(nid == ciphertable[i].nid) {
74                         *algo = ciphertable[i].algo;
75                         *mode = ciphertable[i].mode;
76                         return true;
77                 }
78         }
79
80         return false;
81 }
82
83 static bool ciphertonid(nid_t *nid, cipher_algo_t algo, cipher_mode_t mode) {
84         for(size_t i = 0; i < sizeof(ciphertable) / sizeof(*ciphertable); i++) {
85                 if(algo == ciphertable[i].algo && mode == ciphertable[i].mode) {
86                         *nid = ciphertable[i].nid;
87                         return true;
88                 }
89         }
90
91         return false;
92 }
93
94 static bool cipher_open(cipher_t *cipher, cipher_algo_t algo, cipher_mode_t mode) {
95         gcry_error_t err;
96
97         if(!ciphertonid(&cipher->nid, algo, mode)) {
98                 logger(DEBUG_ALWAYS, LOG_DEBUG, "Cipher %d mode %d has no corresponding nid!", algo, mode);
99                 return false;
100         }
101
102         if((err = gcry_cipher_open(&cipher->handle, algo, mode, 0))) {
103                 logger(DEBUG_ALWAYS, LOG_DEBUG, "Unable to initialise cipher %d mode %d: %s", algo, mode, gcry_strerror(err));
104                 return false;
105         }
106
107         cipher->keylen = gcry_cipher_get_algo_keylen(algo);
108         cipher->blklen = gcry_cipher_get_algo_blklen(algo);
109         cipher->key = xmalloc(cipher->keylen + cipher->blklen);
110         cipher->padding = mode == GCRY_CIPHER_MODE_ECB || mode == GCRY_CIPHER_MODE_CBC;
111
112         return true;
113 }
114
115 bool cipher_open_by_name(cipher_t *cipher, const char *name) {
116         cipher_algo_t algo;
117         cipher_mode_t mode;
118
119         if(!nametocipher(name, &algo, &mode)) {
120                 logger(DEBUG_ALWAYS, LOG_DEBUG, "Unknown cipher name '%s'!", name);
121                 return false;
122         }
123
124         return cipher_open(cipher, algo, mode);
125 }
126
127 bool cipher_open_by_nid(cipher_t *cipher, nid_t nid) {
128         cipher_algo_t algo;
129         cipher_mode_t mode;
130
131         if(!nidtocipher(&algo, &mode, nid)) {
132                 logger(DEBUG_ALWAYS, LOG_DEBUG, "Unknown cipher ID %d!", nid);
133                 return false;
134         }
135
136         return cipher_open(cipher, algo, mode);
137 }
138
139 void cipher_close(cipher_t *cipher) {
140         if(cipher->handle) {
141                 gcry_cipher_close(cipher->handle);
142                 cipher->handle = NULL;
143         }
144
145         free(cipher->key);
146
147         memset(cipher, 0, sizeof(*cipher));
148 }
149
150 size_t cipher_keylength(const cipher_t *cipher) {
151         if(!cipher) {
152                 return 0;
153         }
154
155         return cipher->keylen + cipher->blklen;
156 }
157
158 uint64_t cipher_budget(const cipher_t *cipher) {
159         if(!cipher) {
160                 return UINT64_MAX; // NULL cipher
161         }
162
163         size_t ivlen = cipher->blklen;
164         size_t blklen = cipher->blklen;
165
166         size_t len = blklen > 1
167                      ? blklen
168                      : ivlen > 1 ? ivlen : 8;
169         size_t bits = len * 4 - 1;
170
171         return bits < 64
172                ? UINT64_C(1) << bits
173                : UINT64_MAX;
174 }
175
176 size_t cipher_blocksize(const cipher_t *cipher) {
177         if(!cipher || !cipher->blklen) {
178                 return 1;
179         }
180
181         return cipher->blklen;
182 }
183
184 bool cipher_set_key(cipher_t *cipher, void *key, bool encrypt) {
185         (void)encrypt;
186
187         memcpy(cipher->key, key, cipher->keylen + cipher->blklen);
188
189         gcry_cipher_setkey(cipher->handle, cipher->key, cipher->keylen);
190         gcry_cipher_setiv(cipher->handle, cipher->key + cipher->keylen, cipher->blklen);
191
192         return true;
193 }
194
195 bool cipher_set_key_from_rsa(cipher_t *cipher, void *key, size_t len, bool encrypt) {
196         (void)encrypt;
197
198         memcpy(cipher->key, (char *)key + len - cipher->keylen, cipher->keylen);
199         gcry_cipher_setkey(cipher->handle, cipher->key, cipher->keylen);
200
201         memcpy((char *)cipher->key + cipher->keylen, (char *)key + len - cipher->blklen - cipher->keylen, cipher->blklen);
202         gcry_cipher_setiv(cipher->handle, cipher->key + cipher->keylen, cipher->blklen);
203
204         return true;
205 }
206
207 bool cipher_encrypt(cipher_t *cipher, const void *indata, size_t inlen, void *outdata, size_t *outlen, bool oneshot) {
208         gcry_error_t err;
209         uint8_t pad[cipher->blklen];
210
211         if(cipher->padding) {
212                 if(!oneshot) {
213                         return false;
214                 }
215
216                 size_t reqlen = ((inlen + cipher->blklen) / cipher->blklen) * cipher->blklen;
217
218                 if(*outlen < reqlen) {
219                         logger(DEBUG_ALWAYS, LOG_ERR, "Error while encrypting: not enough room for padding");
220                         return false;
221                 }
222
223                 uint8_t padbyte = reqlen - inlen;
224                 inlen = reqlen - cipher->blklen;
225
226                 for(int i = 0; i < cipher->blklen; i++)
227                         if(i < cipher->blklen - padbyte) {
228                                 pad[i] = ((uint8_t *)indata)[inlen + i];
229                         } else {
230                                 pad[i] = padbyte;
231                         }
232         }
233
234         if(oneshot) {
235                 gcry_cipher_setiv(cipher->handle, cipher->key + cipher->keylen, cipher->blklen);
236         }
237
238         if((err = gcry_cipher_encrypt(cipher->handle, outdata, *outlen, indata, inlen))) {
239                 logger(DEBUG_ALWAYS, LOG_ERR, "Error while encrypting: %s", gcry_strerror(err));
240                 return false;
241         }
242
243         if(cipher->padding) {
244                 if((err = gcry_cipher_encrypt(cipher->handle, (char *)outdata + inlen, cipher->blklen, pad, cipher->blklen))) {
245                         logger(DEBUG_ALWAYS, LOG_ERR, "Error while encrypting: %s", gcry_strerror(err));
246                         return false;
247                 }
248
249                 inlen += cipher->blklen;
250         }
251
252         *outlen = inlen;
253         return true;
254 }
255
256 bool cipher_decrypt(cipher_t *cipher, const void *indata, size_t inlen, void *outdata, size_t *outlen, bool oneshot) {
257         gcry_error_t err;
258
259         if(oneshot) {
260                 gcry_cipher_setiv(cipher->handle, cipher->key + cipher->keylen, cipher->blklen);
261         }
262
263         if((err = gcry_cipher_decrypt(cipher->handle, outdata, *outlen, indata, inlen))) {
264                 logger(DEBUG_ALWAYS, LOG_ERR, "Error while decrypting: %s", gcry_strerror(err));
265                 return false;
266         }
267
268         if(cipher->padding) {
269                 if(!oneshot) {
270                         return false;
271                 }
272
273                 uint8_t padbyte = ((uint8_t *)outdata)[inlen - 1];
274
275                 if(padbyte == 0 || padbyte > cipher->blklen || padbyte > inlen) {
276                         logger(DEBUG_ALWAYS, LOG_ERR, "Error while decrypting: invalid padding");
277                         return false;
278                 }
279
280                 size_t origlen = inlen - padbyte;
281
282                 for(size_t i = inlen - 1; i >= origlen; i--)
283                         if(((uint8_t *)outdata)[i] != padbyte) {
284                                 logger(DEBUG_ALWAYS, LOG_ERR, "Error while decrypting: invalid padding");
285                                 return false;
286                         }
287
288                 *outlen = origlen;
289         } else {
290                 *outlen = inlen;
291         }
292
293         return true;
294 }
295
296 nid_t cipher_get_nid(const cipher_t *cipher) {
297         if(!cipher || !cipher->nid) {
298                 return 0;
299         }
300
301         return cipher->nid;
302 }
303
304 bool cipher_active(const cipher_t *cipher) {
305         return cipher->nid != 0;
306 }