Fix little oops on sizeof()
[asterisk/asterisk.git] / res / res_crypto.c
1 /*
2  * Asterisk -- A telephony toolkit for Linux.
3  *
4  * Provide Cryptographic Signature capability
5  * 
6  * Copyright (C) 1999, Mark Spencer
7  *
8  * Mark Spencer <markster@linux-support.net>
9  *
10  * This program is free software, distributed under the terms of
11  * the GNU General Public License
12  */
13
14 #include <sys/types.h>
15 #include <asterisk/file.h>
16 #include <asterisk/channel.h>
17 #include <asterisk/logger.h>
18 #include <asterisk/say.h>
19 #include <asterisk/module.h>
20 #include <asterisk/options.h>
21 #include <asterisk/crypto.h>
22 #include <asterisk/md5.h>
23 #include <asterisk/cli.h>
24 #include <asterisk/io.h>
25 #include <asterisk/lock.h>
26 #include <asterisk/utils.h>
27 #include <openssl/ssl.h>
28 #include <openssl/err.h>
29 #include <stdio.h>
30 #include <dirent.h>
31 #include <string.h>
32 #include <errno.h>
33 #include <unistd.h>
34 #include <fcntl.h>
35 #include "../asterisk.h"
36 #include "../astconf.h"
37
38 /*
39  * Asterisk uses RSA keys with SHA-1 message digests for its
40  * digital signatures.  The choice of RSA is due to its higher
41  * throughput on verification, and the choice of SHA-1 based
42  * on the recently discovered collisions in MD5's compression 
43  * algorithm and recommendations of avoiding MD5 in new schemes
44  * from various industry experts.
45  *
46  * We use OpenSSL to provide our crypto routines, although we never
47  * actually use full-up SSL
48  *
49  */
50
51 /*
52  * XXX This module is not very thread-safe.  It is for everyday stuff
53  *     like reading keys and stuff, but there are all kinds of weird
54  *     races with people running reload and key init at the same time
55  *     for example
56  *
57  * XXXX
58  */
59
60 AST_MUTEX_DEFINE_STATIC(keylock);
61
62 #define KEY_NEEDS_PASSCODE (1 << 16)
63
64 struct ast_key {
65         /* Name of entity */
66         char name[80];
67         /* File name */
68         char fn[256];
69         /* Key type (AST_KEY_PUB or AST_KEY_PRIV, along with flags from above) */
70         int ktype;
71         /* RSA structure (if successfully loaded) */
72         RSA *rsa;
73         /* Whether we should be deleted */
74         int delme;
75         /* FD for input (or -1 if no input allowed, or -2 if we needed input) */
76         int infd;
77         /* FD for output */
78         int outfd;
79         /* Last MD5 Digest */
80         unsigned char digest[16];
81         struct ast_key *next;
82 };
83
84 static struct ast_key *keys = NULL;
85
86
87 #if 0
88 static int fdprint(int fd, char *s)
89 {
90         return write(fd, s, strlen(s) + 1);
91 }
92 #endif
93 static int pw_cb(char *buf, int size, int rwflag, void *userdata)
94 {
95         struct ast_key *key = (struct ast_key *)userdata;
96         char prompt[256];
97         int res;
98         int tmp;
99         if (key->infd > -1) {
100                 snprintf(prompt, sizeof(prompt), ">>>> passcode for %s key '%s': ",
101                          key->ktype == AST_KEY_PRIVATE ? "PRIVATE" : "PUBLIC", key->name);
102                 write(key->outfd, prompt, strlen(prompt));
103                 memset(buf, 0, sizeof(buf));
104                 tmp = ast_hide_password(key->infd);
105                 memset(buf, 0, size);
106                 res = read(key->infd, buf, size);
107                 ast_restore_tty(key->infd, tmp);
108                 if (buf[strlen(buf) -1] == '\n')
109                         buf[strlen(buf) - 1] = '\0';
110                 return strlen(buf);
111         } else {
112                 /* Note that we were at least called */
113                 key->infd = -2;
114         }
115         return -1;
116 }
117
118 struct ast_key *ast_key_get(char *kname, int ktype)
119 {
120         struct ast_key *key;
121         ast_mutex_lock(&keylock);
122         key = keys;
123         while(key) {
124                 if (!strcmp(kname, key->name) &&
125                     (ktype == key->ktype))
126                         break;
127                 key = key->next;
128         }
129         ast_mutex_unlock(&keylock);
130         return key;
131 }
132
133 static struct ast_key *try_load_key (char *dir, char *fname, int ifd, int ofd, int *not2)
134 {
135         int ktype = 0;
136         char *c = NULL;
137         char ffname[256];
138         char digest[16];
139         FILE *f;
140         struct MD5Context md5;
141         struct ast_key *key;
142         static int notice = 0;
143         int found = 0;
144
145         /* Make sure its name is a public or private key */
146
147         if ((c = strstr(fname, ".pub")) && !strcmp(c, ".pub")) {
148                 ktype = AST_KEY_PUBLIC;
149         } else if ((c = strstr(fname, ".key")) && !strcmp(c, ".key")) {
150                 ktype = AST_KEY_PRIVATE;
151         } else
152                 return NULL;
153
154         /* Get actual filename */
155         snprintf(ffname, sizeof(ffname), "%s/%s", dir, fname);
156
157         ast_mutex_lock(&keylock);
158         key = keys;
159         while(key) {
160                 /* Look for an existing version already */
161                 if (!strcasecmp(key->fn, ffname)) 
162                         break;
163                 key = key->next;
164         }
165         ast_mutex_unlock(&keylock);
166
167         /* Open file */
168         f = fopen(ffname, "r");
169         if (!f) {
170                 ast_log(LOG_WARNING, "Unable to open key file %s: %s\n", ffname, strerror(errno));
171                 return NULL;
172         }
173         MD5Init(&md5);
174         while(!feof(f)) {
175                 /* Calculate a "whatever" quality md5sum of the key */
176                 char buf[256];
177                 fgets(buf, sizeof(buf), f);
178                 if (!feof(f)) {
179                         MD5Update(&md5, buf, strlen(buf));
180                 }
181         }
182         MD5Final(digest, &md5);
183         if (key) {
184                 /* If the MD5 sum is the same, and it isn't awaiting a passcode 
185                    then this is far enough */
186                 if (!memcmp(digest, key->digest, 16) &&
187                     !(key->ktype & KEY_NEEDS_PASSCODE)) {
188                         fclose(f);
189                         key->delme = 0;
190                         return NULL;
191                 } else {
192                         /* Preserve keytype */
193                         ktype = key->ktype;
194                         /* Recycle the same structure */
195                         found++;
196                 }
197         }
198
199         /* Make fname just be the normal name now */
200         *c = '\0';
201         if (!key) {
202                 key = (struct ast_key *)malloc(sizeof(struct ast_key));
203                 if (!key) {
204                         ast_log(LOG_WARNING, "Out of memory\n");
205                         fclose(f);
206                         return NULL;
207                 }
208                 memset(key, 0, sizeof(struct ast_key));
209         }
210         /* At this point we have a key structure (old or new).  Time to
211            fill it with what we know */
212         /* Gotta lock if this one already exists */
213         if (found)
214                 ast_mutex_lock(&keylock);
215         /* First the filename */
216         strncpy(key->fn, ffname, sizeof(key->fn) - 1);
217         /* Then the name */
218         strncpy(key->name, fname, sizeof(key->name) - 1);
219         key->ktype = ktype;
220         /* Yes, assume we're going to be deleted */
221         key->delme = 1;
222         /* Keep the key type */
223         memcpy(key->digest, digest, 16);
224         /* Can I/O takes the FD we're given */
225         key->infd = ifd;
226         key->outfd = ofd;
227         /* Reset the file back to the beginning */
228         rewind(f);
229         /* Now load the key with the right method */
230         if (ktype == AST_KEY_PUBLIC)
231                 key->rsa = PEM_read_RSA_PUBKEY(f, NULL, pw_cb, key);
232         else
233                 key->rsa = PEM_read_RSAPrivateKey(f, NULL, pw_cb, key);
234         fclose(f);
235         if (key->rsa) {
236                 /* Key loaded okay */
237                 key->ktype &= ~KEY_NEEDS_PASSCODE;
238                 if (option_verbose > 2)
239                         ast_verbose(VERBOSE_PREFIX_3 "Loaded %s key '%s'\n", key->ktype == AST_KEY_PUBLIC ? "PUBLIC" : "PRIVATE", key->name);
240                 if (option_debug)
241                         ast_log(LOG_DEBUG, "Key '%s' loaded OK\n", key->name);
242                 key->delme = 0;
243         } else if (key->infd != -2) {
244                 ast_log(LOG_WARNING, "Key load %s '%s' failed\n",key->ktype == AST_KEY_PUBLIC ? "PUBLIC" : "PRIVATE", key->name);
245                 if (ofd > -1) {
246                         ERR_print_errors_fp(stderr);
247                 } else
248                         ERR_print_errors_fp(stderr);
249         } else {
250                 ast_log(LOG_NOTICE, "Key '%s' needs passcode.\n", key->name);
251                 key->ktype |= KEY_NEEDS_PASSCODE;
252                 if (!notice) {
253                         if (!option_initcrypto) 
254                                 ast_log(LOG_NOTICE, "Add the '-i' flag to the asterisk command line if you want to automatically initialize passcodes at launch.\n");
255                         notice++;
256                 }
257                 /* Keep it anyway */
258                 key->delme = 0;
259                 /* Print final notice about "init keys" when done */
260                 *not2 = 1;
261         }
262         if (found)
263                 ast_mutex_unlock(&keylock);
264         if (!found) {
265                 ast_mutex_lock(&keylock);
266                 key->next = keys;
267                 keys = key;
268                 ast_mutex_unlock(&keylock);
269         }
270         return key;
271 }
272
273 #if 0
274
275 static void dump(unsigned char *src, int len)
276 {
277         int x; 
278         for (x=0;x<len;x++)
279                 printf("%02x", *(src++));
280         printf("\n");
281 }
282
283 static char *binary(int y, int len)
284 {
285         static char res[80];
286         int x;
287         memset(res, 0, sizeof(res));
288         for (x=0;x<len;x++) {
289                 if (y & (1 << x))
290                         res[(len - x - 1)] = '1';
291                 else
292                         res[(len - x - 1)] = '0';
293         }
294         return res;
295 }
296
297 #endif
298
299 int ast_sign_bin(struct ast_key *key, char *msg, int msglen, unsigned char *dsig)
300 {
301         unsigned char digest[20];
302         int siglen = 128;
303         int res;
304
305         if (key->ktype != AST_KEY_PRIVATE) {
306                 ast_log(LOG_WARNING, "Cannot sign with a private key\n");
307                 return -1;
308         }
309
310         /* Calculate digest of message */
311         SHA1((unsigned char *)msg, msglen, digest);
312
313         /* Verify signature */
314         res = RSA_sign(NID_sha1, digest, sizeof(digest), dsig, &siglen, key->rsa);
315         
316         if (!res) {
317                 ast_log(LOG_WARNING, "RSA Signature (key %s) failed\n", key->name);
318                 return -1;
319         }
320
321         if (siglen != 128) {
322                 ast_log(LOG_WARNING, "Unexpected signature length %d, expecting %d\n", (int)siglen, (int)128);
323                 return -1;
324         }
325
326         return 0;
327         
328 }
329
330 int ast_sign(struct ast_key *key, char *msg, char *sig)
331 {
332         unsigned char dsig[128];
333         int siglen = sizeof(dsig);
334         int res;
335         res = ast_sign_bin(key, msg, strlen(msg), dsig);
336         if (!res)
337                 /* Success -- encode (256 bytes max as documented) */
338                 ast_base64encode(sig, dsig, siglen, 256);
339         return res;
340         
341 }
342
343 int ast_check_signature_bin(struct ast_key *key, char *msg, int msglen, unsigned char *dsig)
344 {
345         unsigned char digest[20];
346         int res;
347
348         if (key->ktype != AST_KEY_PUBLIC) {
349                 /* Okay, so of course you really *can* but for our purposes
350                    we're going to say you can't */
351                 ast_log(LOG_WARNING, "Cannot check message signature with a private key\n");
352                 return -1;
353         }
354
355         /* Calculate digest of message */
356         SHA1((unsigned char *)msg, msglen, digest);
357
358         /* Verify signature */
359         res = RSA_verify(NID_sha1, digest, sizeof(digest), dsig, 128, key->rsa);
360         
361         if (!res) {
362                 ast_log(LOG_DEBUG, "Key failed verification\n");
363                 return -1;
364         }
365         /* Pass */
366         return 0;
367 }
368
369 int ast_check_signature(struct ast_key *key, char *msg, char *sig)
370 {
371         unsigned char dsig[128];
372         int res;
373
374         /* Decode signature */
375         res = ast_base64decode(dsig, sig, sizeof(dsig));
376         if (res != sizeof(dsig)) {
377                 ast_log(LOG_WARNING, "Signature improper length (expect %d, got %d)\n", (int)sizeof(dsig), (int)res);
378                 return -1;
379         }
380         res = ast_check_signature_bin(key, msg, strlen(msg), dsig);
381         return res;
382 }
383
384 static void crypto_load(int ifd, int ofd)
385 {
386         struct ast_key *key, *nkey, *last;
387         DIR *dir;
388         struct dirent *ent;
389         int note = 0;
390         /* Mark all keys for deletion */
391         ast_mutex_lock(&keylock);
392         key = keys;
393         while(key) {
394                 key->delme = 1;
395                 key = key->next;
396         }
397         ast_mutex_unlock(&keylock);
398         /* Load new keys */
399         dir = opendir((char *)ast_config_AST_KEY_DIR);
400         if (dir) {
401                 while((ent = readdir(dir))) {
402                         try_load_key((char *)ast_config_AST_KEY_DIR, ent->d_name, ifd, ofd, &note);
403                 }
404                 closedir(dir);
405         } else
406                 ast_log(LOG_WARNING, "Unable to open key directory '%s'\n", (char *)ast_config_AST_KEY_DIR);
407         if (note) {
408                 ast_log(LOG_NOTICE, "Please run the command 'init keys' to enter the passcodes for the keys\n");
409         }
410         ast_mutex_lock(&keylock);
411         key = keys;
412         last = NULL;
413         while(key) {
414                 nkey = key->next;
415                 if (key->delme) {
416                         ast_log(LOG_DEBUG, "Deleting key %s type %d\n", key->name, key->ktype);
417                         /* Do the delete */
418                         if (last)
419                                 last->next = nkey;
420                         else
421                                 keys = nkey;
422                         if (key->rsa)
423                                 RSA_free(key->rsa);
424                         free(key);
425                 } else 
426                         last = key;
427                 key = nkey;
428         }
429         ast_mutex_unlock(&keylock);
430 }
431
432 static void md52sum(char *sum, unsigned char *md5)
433 {
434         int x;
435         for (x=0;x<16;x++) 
436                 sum += sprintf(sum, "%02x", *(md5++));
437 }
438
439 static int show_keys(int fd, int argc, char *argv[])
440 {
441         struct ast_key *key;
442         char sum[16 * 2 + 1];
443
444         ast_mutex_lock(&keylock);
445         key = keys;
446         ast_cli(fd, "%-18s %-8s %-16s %-33s\n", "Key Name", "Type", "Status", "Sum");
447         while(key) {
448                 md52sum(sum, key->digest);
449                 ast_cli(fd, "%-18s %-8s %-16s %-33s\n", key->name, 
450                         (key->ktype & 0xf) == AST_KEY_PUBLIC ? "PUBLIC" : "PRIVATE",
451                         key->ktype & KEY_NEEDS_PASSCODE ? "[Needs Passcode]" : "[Loaded]", sum);
452                                 
453                 key = key->next;
454         }
455         ast_mutex_unlock(&keylock);
456         return RESULT_SUCCESS;
457 }
458
459 static int init_keys(int fd, int argc, char *argv[])
460 {
461         struct ast_key *key;
462         int ign;
463         char *kn;
464         char tmp[256] = "";
465
466         key = keys;
467         while(key) {
468                 /* Reload keys that need pass codes now */
469                 if (key->ktype & KEY_NEEDS_PASSCODE) {
470                         kn = key->fn + strlen(ast_config_AST_KEY_DIR) + 1;
471                         strncpy(tmp, kn, sizeof(tmp) - 1);
472                         try_load_key((char *)ast_config_AST_KEY_DIR, tmp, fd, fd, &ign);
473                 }
474                 key = key->next;
475         }
476         return RESULT_SUCCESS;
477 }
478
479 static char show_key_usage[] =
480 "Usage: show keys\n"
481 "       Displays information about RSA keys known by Asterisk\n";
482
483 static char init_keys_usage[] =
484 "Usage: init keys\n"
485 "       Initializes private keys (by reading in pass code from the user)\n";
486
487 static struct ast_cli_entry cli_show_keys = 
488 { { "show", "keys", NULL }, show_keys, "Displays RSA key information", show_key_usage };
489
490 static struct ast_cli_entry cli_init_keys = 
491 { { "init", "keys", NULL }, init_keys, "Initialize RSA key passcodes", init_keys_usage };
492
493 static int crypto_init(void)
494 {
495         SSL_library_init();
496         ERR_load_crypto_strings();
497         ast_cli_register(&cli_show_keys);
498         ast_cli_register(&cli_init_keys);
499         return 0;
500 }
501
502 int reload(void)
503 {
504         crypto_load(-1, -1);
505         return 0;
506 }
507
508 int load_module(void)
509 {
510         crypto_init();
511         if (option_initcrypto)
512                 crypto_load(STDIN_FILENO, STDOUT_FILENO);
513         else
514                 crypto_load(-1, -1);
515         return 0;
516 }
517
518 int unload_module(void)
519 {
520         /* Can't unload this once we're loaded */
521         return -1;
522 }
523
524 char *description(void)
525 {
526         return "Cryptographic Digital Signatures";
527 }
528
529 int usecount(void)
530 {
531         /* We should never be unloaded */
532         return 1;
533 }
534
535 char *key()
536 {
537         return ASTERISK_GPL_KEY;
538 }