9b05a9494c23503702f88b27716468942b15d805
[paraslash.git] / gcrypt.c
1 /* Copyright (C) 2011 Andre Noll <maan@tuebingen.mpg.de>, see file COPYING. */
2
3 /** \file gcrypt.c Libgrcypt-based encryption/decryption routines. */
4
5 #include <regex.h>
6 #include <gcrypt.h>
7
8 #include "para.h"
9 #include "error.h"
10 #include "string.h"
11 #include "crypt.h"
12 #include "crypt_backend.h"
13 #include "fd.h"
14 #include "base64.h"
15
16 //#define GCRYPT_DEBUG 1
17
18 #ifdef GCRYPT_DEBUG
19 static void dump_buffer(const char *msg, unsigned char *buf, int len)
20 {
21         int i;
22
23         fprintf(stderr, "%s (%d bytes): ", msg, len);
24         for (i = 0; i < len; i++)
25                 fprintf(stderr, "%02x ", buf[i]);
26         fprintf(stderr, "\n");
27 }
28 #else
29 /** Empty. Define GCRYPT_DEBUG to dump buffers. */
30 #define dump_buffer(a, b, c)
31 #endif
32
33 void hash_function(const char *data, unsigned long len, unsigned char *hash)
34 {
35         gcry_error_t gret;
36         gcry_md_hd_t handle;
37         unsigned char *md;
38
39         gret = gcry_md_open(&handle, GCRY_MD_SHA1, 0);
40         assert(gret == 0);
41         gcry_md_write(handle, data, (size_t)len);
42         gcry_md_final(handle);
43         md = gcry_md_read(handle, GCRY_MD_SHA1);
44         assert(md);
45         memcpy(hash, md, HASH_SIZE);
46         gcry_md_close(handle);
47 }
48
49 void get_random_bytes_or_die(unsigned char *buf, int num)
50 {
51         gcry_randomize(buf, (size_t)num, GCRY_STRONG_RANDOM);
52 }
53
54 /*
55  * This is called at the beginning of every program that uses libgcrypt. The
56  * call to gcry_check_version() initializes the gcrypt library and checks that
57  * we have at least the minimal required version.
58  */
59 void crypt_init(void)
60 {
61         const char *req_ver = "1.5.0";
62         int seed;
63
64         if (!gcry_check_version(req_ver)) {
65                 PARA_EMERG_LOG("fatal: need at least libgcrypt-%s, have: %s\n",
66                         req_ver, gcry_check_version(NULL));
67                 exit(EXIT_FAILURE);
68         }
69
70         /*
71          * Allocate a pool of secure memory. This also drops privileges where
72          * needed.
73          */
74         gcry_control(GCRYCTL_INIT_SECMEM, 65536, 0);
75
76         /* Tell Libgcrypt that initialization has completed. */
77         gcry_control(GCRYCTL_INITIALIZATION_FINISHED, 0);
78
79         get_random_bytes_or_die((unsigned char *)&seed, sizeof(seed));
80         srandom(seed);
81 }
82
83 void crypt_shutdown(void)
84 {
85         /*
86          * WK does not see a way to apply a patch for the sake of Valgrind, so
87          * as of 2018 libgrypt has no deinitialization routine to free the
88          * resources on exit.
89          */
90 }
91
92 /** S-expression for the public part of an RSA key. */
93 #define RSA_PUBKEY_SEXP "(public-key (rsa (n %m) (e %m)))"
94 /** S-expression for a private RSA key. */
95 #define RSA_PRIVKEY_SEXP "(private-key (rsa (n %m) (e %m) (d %m) (p %m) (q %m) (u %m)))"
96 /** S-expression for decryption. */
97 #define RSA_DECRYPT_SEXP "(enc-val(flags oaep)(rsa(a %m)))"
98
99 struct asymmetric_key {
100         gcry_sexp_t sexp;
101         int num_bytes;
102 };
103
104 static const char *gcrypt_strerror(gcry_error_t gret)
105 {
106         return gcry_strerror(gcry_err_code(gret));
107 }
108
109 /** Private keys start with this header. */
110 #define PRIVATE_KEY_HEADER "-----BEGIN RSA PRIVATE KEY-----"
111 /** Private keys end with this footer. */
112 #define PRIVATE_KEY_FOOTER "-----END RSA PRIVATE KEY-----"
113
114 static int decode_key(const char *key_file, unsigned char **result,
115                 size_t *blob_size)
116 {
117         int ret, ret2, i, j;
118         void *map;
119         size_t map_size, key_size;
120         unsigned char *blob = NULL;
121         char *begin, *footer, *key;
122
123         ret = mmap_full_file(key_file, O_RDONLY, &map, &map_size, NULL);
124         if (ret < 0)
125                 goto out;
126         ret = -E_KEY_MARKER;
127         if (strncmp(map, PRIVATE_KEY_HEADER, strlen(PRIVATE_KEY_HEADER)))
128                 goto unmap;
129         footer = strstr(map, PRIVATE_KEY_FOOTER);
130         ret = -E_KEY_MARKER;
131         if (!footer)
132                 goto unmap;
133         begin = map + strlen(PRIVATE_KEY_HEADER);
134         /* skip whitespace at the beginning */
135         for (; begin < footer; begin++) {
136                 if (para_isspace(*begin))
137                         continue;
138                 break;
139         }
140         ret = -E_KEY_MARKER;
141         if (begin >= footer)
142                 goto unmap;
143
144         key_size = footer - begin;
145         key = para_malloc(key_size + 1);
146         for (i = 0, j = 0; begin + i < footer; i++) {
147                 if (para_isspace(begin[i]))
148                         continue;
149                 key[j++] = begin[i];
150         }
151         key[j] = '\0';
152         ret = base64_decode(key, j, (char **)&blob, blob_size);
153         free(key);
154 unmap:
155         ret2 = para_munmap(map, map_size);
156         if (ret >= 0 && ret2 < 0)
157                 ret = ret2;
158         if (ret < 0) {
159                 free(blob);
160                 blob = NULL;
161         }
162 out:
163         *result = blob;
164         return ret;
165 }
166
167 /** ASN Types and their code. */
168 enum asn1_types {
169         /** The next object is an integer. */
170         ASN1_TYPE_INTEGER = 0x2,
171         /** Bit string object. */
172         ASN1_TYPE_BIT_STRING = 0x03,
173         /** Keys start with one big type sequence. */
174         ASN1_TYPE_SEQUENCE = 0x30,
175 };
176
177 /* bit 6 has value 0 */
178 static inline bool is_primitive(unsigned char c)
179 {
180         return (c & (1<<6)) == 0;
181 }
182
183 static inline bool is_primitive_integer(unsigned char c)
184 {
185         if (!is_primitive(c))
186                 return false;
187         return (c & 0x1f) == ASN1_TYPE_INTEGER;
188 }
189
190 /* Bit 8 is zero (and bits 7-1 give the length) */
191 static inline bool is_short_form(unsigned char c)
192 {
193         return (c & 0x80) == 0;
194 }
195
196 static inline int get_short_form_length(unsigned char c)
197 {
198         return c & 0x7f;
199 }
200
201 static inline int get_long_form_num_length_bytes(unsigned char c)
202 {
203         return c & 0x7f;
204 }
205
206 /*
207  * Returns: Number of bytes scanned. This may differ from the value returned via
208  * bn_bytes because the latter does not include the ASN.1 prefix and a leading
209  * zero is not considered as an additional byte for bn_bytes.
210  */
211 static int read_bignum(unsigned char *start, unsigned char *end, gcry_mpi_t *bn,
212                 int *bn_bytes)
213 {
214         int i, bn_size;
215         gcry_error_t gret;
216         unsigned char *cp = start;
217
218         if (!is_primitive_integer(*cp))
219                 return -E_BAD_PRIVATE_KEY;
220         cp++;
221         if (is_short_form(*cp)) {
222                 bn_size = get_short_form_length(*cp);
223                 cp++;
224         } else {
225                 int num_bytes = get_long_form_num_length_bytes(*cp);
226                 if (cp + num_bytes > end)
227                         return -E_BAD_PRIVATE_KEY;
228                 if (num_bytes > 4) /* nobody has such a large modulus */
229                         return -E_BAD_PRIVATE_KEY;
230                 cp++;
231                 bn_size = 0;
232                 for (i = 0; i < num_bytes; i++, cp++)
233                         bn_size = (bn_size << 8) + *cp;
234         }
235         PARA_DEBUG_LOG("bn_size %d (0x%x)\n", bn_size, (unsigned)bn_size);
236         gret = gcry_mpi_scan(bn, GCRYMPI_FMT_STD, cp, bn_size, NULL);
237         if (gret) {
238                 PARA_ERROR_LOG("gcry_mpi_scan: %s\n",
239                         gcry_strerror(gcry_err_code(gret)));
240                 return-E_MPI_SCAN;
241         }
242         /*
243          * Don't take the first leading zero into account for the size of the
244          * bignum.
245          */
246         if (*cp == '\0') {
247                 cp++;
248                 bn_size--;
249         }
250         if (bn_bytes)
251                 *bn_bytes = bn_size;
252         cp += bn_size;
253 //      unsigned char *buf;
254 //      gcry_mpi_aprint(GCRYMPI_FMT_HEX, &buf, NULL, *bn);
255 //      PARA_CRIT_LOG("bn: %s\n", buf);
256         return cp - start;
257 }
258
259 static int find_privkey_bignum_offset(const unsigned char *data, int len)
260 {
261         const unsigned char *p = data, *end = data + len;
262
263         /* like the public key, the whole thing is contained in a sequence */
264         if (*p != ASN1_TYPE_SEQUENCE)
265                 return -E_ASN1_PARSE;
266         p++;
267         if (p >= end)
268                 return -E_ASN1_PARSE;
269         if (is_short_form(*p))
270                 p++;
271         else
272                 p += 1 + get_long_form_num_length_bytes(*p);
273         if (p >= end)
274                 return -E_ASN1_PARSE;
275
276         /* skip next integer */
277         if (*p != ASN1_TYPE_INTEGER)
278                 return -E_ASN1_PARSE;
279         p++;
280         if (p >= end)
281                 return -E_ASN1_PARSE;
282         if (is_short_form(*p))
283                 p += 1 + get_short_form_length(*p);
284         else
285                 p += 1 + get_long_form_num_length_bytes(*p);
286         if (p >= end)
287                 return -E_ASN1_PARSE;
288         return p - data;
289 }
290
291 static int get_private_key(const char *key_file, struct asymmetric_key **result)
292 {
293         gcry_mpi_t n = NULL, e = NULL, d = NULL, p = NULL, q = NULL,
294                 u = NULL;
295         unsigned char *blob, *cp, *end;
296         int ret, n_size;
297         gcry_error_t gret;
298         size_t erroff, blob_size;
299         gcry_sexp_t sexp;
300         struct asymmetric_key *key;
301
302         *result = NULL;
303         ret = decode_key(key_file, &blob, &blob_size);
304         if (ret < 0)
305                 return ret;
306         end = blob + blob_size;
307         ret = find_privkey_bignum_offset(blob, blob_size);
308         if (ret < 0)
309                 goto free_blob;
310         PARA_INFO_LOG("reading RSA params at offset %d\n", ret);
311         cp = blob + ret;
312
313         ret = read_bignum(cp, end, &n, &n_size);
314         if (ret < 0)
315                 goto free_blob;
316         cp += ret;
317
318         ret = read_bignum(cp, end, &e, NULL);
319         if (ret < 0)
320                 goto release_n;
321         cp += ret;
322
323         ret = read_bignum(cp, end, &d, NULL);
324         if (ret < 0)
325                 goto release_e;
326         cp += ret;
327
328         ret = read_bignum(cp, end, &p, NULL);
329         if (ret < 0)
330                 goto release_d;
331         cp += ret;
332
333         ret = read_bignum(cp, end, &q, NULL);
334         if (ret < 0)
335                 goto release_p;
336         cp += ret;
337         ret = read_bignum(cp, end, &u, NULL);
338         if (ret < 0)
339                 goto release_q;
340         /*
341          * OpenSSL uses slightly different parameters than gcrypt. To use these
342          * parameters we need to swap the values of p and q and recompute u.
343          */
344         if (gcry_mpi_cmp(p, q) > 0) {
345                 gcry_mpi_swap(p, q);
346                 gcry_mpi_invm(u, p, q);
347         }
348         gret = gcry_sexp_build(&sexp, &erroff, RSA_PRIVKEY_SEXP,
349                 n, e, d, p, q, u);
350
351         if (gret) {
352                 PARA_ERROR_LOG("offset %zu: %s\n", erroff,
353                         gcry_strerror(gcry_err_code(gret)));
354                 ret = -E_SEXP_BUILD;
355                 goto release_u;
356         }
357         key = para_malloc(sizeof(*key));
358         key->sexp = sexp;
359         *result = key;
360         ret = n_size * 8;
361         PARA_INFO_LOG("succesfully read %d bit private key\n", ret);
362 release_u:
363         gcry_mpi_release(u);
364 release_q:
365         gcry_mpi_release(q);
366 release_p:
367         gcry_mpi_release(p);
368 release_d:
369         gcry_mpi_release(d);
370 release_e:
371         gcry_mpi_release(e);
372 release_n:
373         gcry_mpi_release(n);
374 free_blob:
375         free(blob);
376         return ret;
377 }
378
379 int apc_get_pubkey(const char *key_file, struct asymmetric_key **result)
380 {
381         unsigned char *blob, *p, *end;
382         int ret;
383         gcry_error_t gret;
384         size_t nr_scanned, erroff, decoded_size;
385         gcry_mpi_t e, n;
386         gcry_sexp_t sexp;
387         struct asymmetric_key *key;
388
389         ret = decode_ssh_key(key_file, &blob, &decoded_size);
390         if (ret < 0)
391                 return ret;
392         p = blob + ret;
393         end = blob + decoded_size;
394         PARA_DEBUG_LOG("scanning modulus and public exponent\n");
395         gret = gcry_mpi_scan(&e, GCRYMPI_FMT_SSH, p, end - p, &nr_scanned);
396         if (gret) {
397                 ret = -E_MPI_SCAN;
398                 PARA_CRIT_LOG("%s\n", gcry_strerror(gcry_err_code(gret)));
399                 goto free_blob;
400         }
401         PARA_DEBUG_LOG("scanned e (%zu bytes)\n", nr_scanned);
402         p += nr_scanned;
403         if (p >= end)
404                 goto release_e;
405         gret = gcry_mpi_scan(&n, GCRYMPI_FMT_SSH, p, end - p, &nr_scanned);
406         if (gret) {
407                 ret = -E_MPI_SCAN;
408                 PARA_ERROR_LOG("%s\n", gcry_strerror(gcry_err_code(gret)));
409                 goto release_e;
410         }
411         PARA_DEBUG_LOG("scanned n (%zu bytes)\n", nr_scanned);
412         gret = gcry_sexp_build(&sexp, &erroff, RSA_PUBKEY_SEXP, n, e);
413         if (gret) {
414                 PARA_ERROR_LOG("offset %zu: %s\n", erroff,
415                         gcry_strerror(gcry_err_code(gret)));
416                 ret = -E_SEXP_BUILD;
417                 goto release_n;
418         }
419         ret = ROUND_DOWN(nr_scanned, 32);
420         PARA_INFO_LOG("successfully read %d bit ssh public key\n", ret * 8);
421         key = para_malloc(sizeof(*key));
422         key->num_bytes = ret;
423         key->sexp = sexp;
424         *result = key;
425 release_n:
426         gcry_mpi_release(n);
427 release_e:
428         gcry_mpi_release(e);
429 free_blob:
430         free(blob);
431         return ret;
432 }
433
434 void apc_free_pubkey(struct asymmetric_key *key)
435 {
436         if (!key)
437                 return;
438         gcry_sexp_release(key->sexp);
439         free(key);
440 }
441
442 static int decode_rsa(gcry_sexp_t sexp, unsigned char *outbuf, size_t *nbytes)
443 {
444         const char *p = gcry_sexp_nth_data(sexp, 1, nbytes);
445
446         if (!p)
447                 return -E_RSA_DECODE;
448         memcpy(outbuf, p, *nbytes);
449         return 1;
450 }
451
452 int apc_priv_decrypt(const char *key_file, unsigned char *outbuf,
453                 unsigned char *inbuf, int inlen)
454 {
455         gcry_error_t gret;
456         int ret;
457         struct asymmetric_key *priv;
458         gcry_mpi_t in_mpi = NULL;
459         gcry_sexp_t in, out, priv_key;
460         size_t nbytes;
461
462         ret = check_private_key_file(key_file);
463         if (ret < 0)
464                 return ret;
465         PARA_INFO_LOG("decrypting %d byte input\n", inlen);
466         /* key_file -> asymmetric key priv */
467         ret = get_private_key(key_file, &priv);
468         if (ret < 0)
469                 return ret;
470
471         /* asymmetric key priv -> sexp priv_key */
472         ret = -E_SEXP_FIND;
473         priv_key = gcry_sexp_find_token(priv->sexp, "private-key", 0);
474         if (!priv_key)
475                 goto free_key;
476
477         /* inbuf -> in_mpi */
478         gret = gcry_mpi_scan(&in_mpi, GCRYMPI_FMT_USG, inbuf,
479                 inlen, NULL);
480         if (gret) {
481                 PARA_ERROR_LOG("%s\n", gcrypt_strerror(gret));
482                 ret = -E_MPI_SCAN;
483                 goto key_release;
484         }
485         /* in_mpi -> in sexp */
486         gret = gcry_sexp_build(&in, NULL, RSA_DECRYPT_SEXP, in_mpi);
487         if (gret) {
488                 PARA_ERROR_LOG("%s\n", gcrypt_strerror(gret));
489                 ret = -E_SEXP_BUILD;
490                 goto in_mpi_release;
491         }
492
493         /* rsa decryption: in sexp -> out sexp */
494         gret = gcry_pk_decrypt(&out, in, priv_key);
495         if (gret) {
496                 PARA_ERROR_LOG("decrypt: %s\n", gcrypt_strerror(gret));
497                 ret = -E_SEXP_DECRYPT;
498                 goto in_release;
499         }
500         ret = decode_rsa(out, outbuf, &nbytes);
501         if (ret < 0)
502                 goto out_release;
503         PARA_INFO_LOG("successfully decrypted %zu byte message\n", nbytes);
504         ret = nbytes;
505 out_release:
506         gcry_sexp_release(out);
507 in_release:
508         gcry_sexp_release(in);
509 in_mpi_release:
510         gcry_mpi_release(in_mpi);
511 key_release:
512         gcry_sexp_release(priv_key);
513 free_key:
514         gcry_sexp_release(priv->sexp);
515         free(priv);
516         return ret;
517 }
518
519 int apc_pub_encrypt(struct asymmetric_key *pub, unsigned char *inbuf,
520                 unsigned len, unsigned char *outbuf)
521 {
522         gcry_error_t gret;
523         gcry_sexp_t pub_key, in, out, out_a;
524         gcry_mpi_t out_mpi = NULL;
525         size_t nbytes;
526         int ret;
527
528         PARA_INFO_LOG("encrypting %u byte input with %d-byte key\n", len, pub->num_bytes);
529
530         /* get pub key */
531         pub_key = gcry_sexp_find_token(pub->sexp, "public-key", 0);
532         if (!pub_key)
533                 return -E_SEXP_FIND;
534         gret = gcry_sexp_build(&in, NULL, "(data(flags oaep)(value %b))", len, inbuf);
535         if (gret) {
536                 PARA_ERROR_LOG("%s\n", gcrypt_strerror(gret));
537                 ret = -E_SEXP_BUILD;
538                 goto key_release;
539         }
540         /* rsa sexp encryption: in -> out */
541         gret = gcry_pk_encrypt(&out, in, pub_key);
542         if (gret) {
543                 PARA_ERROR_LOG("%s\n", gcrypt_strerror(gret));
544                 ret = -E_SEXP_ENCRYPT;
545                 goto in_release;
546         }
547         /* extract a, an MPI with the result of the RSA operation */
548         ret = -E_SEXP_FIND;
549         out_a = gcry_sexp_find_token(out, "a", 0);
550         if (!out_a)
551                 goto out_release;
552         /* convert sexp out_a -> out_mpi */
553         out_mpi = gcry_sexp_nth_mpi(out_a, 1, GCRYMPI_FMT_USG);
554         if (!out_mpi) {
555                 ret = -E_SEXP_FIND;
556                 goto out_a_release;
557         }
558         gret = gcry_mpi_print(GCRYMPI_FMT_USG, outbuf, 512 /* FIXME */, &nbytes, out_mpi);
559         if (gret) {
560                 PARA_ERROR_LOG("%s\n", gcrypt_strerror(gret));
561                 ret = -E_SEXP_ENCRYPT;
562                 goto out_mpi_release;
563         }
564         PARA_INFO_LOG("encrypted buffer is %zu bytes\n", nbytes);
565         dump_buffer("enc buf", outbuf, nbytes);
566         ret = nbytes;
567
568 out_mpi_release:
569         gcry_mpi_release(out_mpi);
570 out_a_release:
571         gcry_sexp_release(out_a);
572 out_release:
573         gcry_sexp_release(out);
574 in_release:
575         gcry_sexp_release(in);
576 key_release:
577         gcry_sexp_release(pub_key);
578         return ret;
579 }
580
581 struct stream_cipher {
582         gcry_cipher_hd_t handle;
583 };
584
585 struct stream_cipher *sc_new(const unsigned char *data, int len)
586 {
587         gcry_error_t gret;
588         struct stream_cipher *sc = para_malloc(sizeof(*sc));
589
590         assert(len >= 2 * AES_CRT128_BLOCK_SIZE);
591         gret = gcry_cipher_open(&sc->handle, GCRY_CIPHER_AES128,
592                 GCRY_CIPHER_MODE_CTR, 0);
593         assert(gret == 0);
594         gret = gcry_cipher_setkey(sc->handle, data,
595                 AES_CRT128_BLOCK_SIZE);
596         assert(gret == 0);
597         gret = gcry_cipher_setctr(sc->handle,
598                 data + AES_CRT128_BLOCK_SIZE, AES_CRT128_BLOCK_SIZE);
599         assert(gret == 0);
600         return sc;
601 }
602
603 void sc_free(struct stream_cipher *sc)
604 {
605         if (!sc)
606                 return;
607         gcry_cipher_close(sc->handle);
608         free(sc);
609 }
610
611 void sc_crypt(struct stream_cipher *sc, struct iovec *src, struct iovec *dst)
612 {
613         gcry_cipher_hd_t handle = sc->handle;
614         gcry_error_t gret;
615
616         /* perform in-place encryption */
617         *dst = *src;
618         gret = gcry_cipher_encrypt(handle, src->iov_base, src->iov_len,
619                 NULL, 0);
620         assert(gret == 0);
621 }