fixed bug with multiple status server sent, some dtls fixes
[radsecproxy.git] / dtls.c
1 /*
2  * Copyright (C) 2008 Stig Venaas <venaas@uninett.no>
3  *
4  * Permission to use, copy, modify, and distribute this software for any
5  * purpose with or without fee is hereby granted, provided that the above
6  * copyright notice and this permission notice appear in all copies.
7  */
8
9 #include <signal.h>
10 #include <sys/socket.h>
11 #include <netinet/in.h>
12 #include <netdb.h>
13 #include <string.h>
14 #include <unistd.h>
15 #include <limits.h>
16 #ifdef SYS_SOLARIS9
17 #include <fcntl.h>
18 #endif
19 #include <sys/time.h>
20 #include <sys/types.h>
21 #include <sys/select.h>
22 #include <ctype.h>
23 #include <sys/wait.h>
24 #include <arpa/inet.h>
25 #include <regex.h>
26 #include <pthread.h>
27 #include <openssl/ssl.h>
28 #include <openssl/err.h>
29 #include "debug.h"
30 #include "list.h"
31 #include "hash.h"
32 #include "util.h"
33 #include "radsecproxy.h"
34 #include "dtls.h"
35
36 static int client4_sock = -1;
37 static int client6_sock = -1;
38
39 struct sessioncacheentry {
40     pthread_mutex_t mutex;
41     struct queue *rbios;
42     struct timeval expiry;
43 };
44
45 struct dtlsservernewparams {
46     struct sessioncacheentry *sesscache;
47     int sock;
48     struct sockaddr_storage addr;    
49 };
50
51 int udp2bio(int s, struct queue *q, int cnt) {
52     unsigned char *buf;
53     BIO *rbio;
54
55     if (cnt < 1)
56         return 0;
57     
58     buf = malloc(cnt);
59     if (!buf) {
60         unsigned char err;
61         debug(DBG_ERR, "udp2bio: malloc failed");
62         recv(s, &err, 1, 0);
63         return 0;
64     }
65
66     cnt = recv(s, buf, cnt, 0);
67     if (cnt < 1) {
68         debug(DBG_WARN, "udp2bio: recv failed");
69         free(buf);
70         return 0;
71     }
72
73     rbio = BIO_new_mem_buf(buf, cnt);
74     BIO_set_mem_eof_return(rbio, -1);
75
76     pthread_mutex_lock(&q->mutex);
77     if (!list_push(q->entries, rbio)) {
78         BIO_free(rbio);
79         pthread_mutex_unlock(&q->mutex);
80         return 0;
81     }
82     pthread_cond_signal(&q->cond);
83     pthread_mutex_unlock(&q->mutex);
84     return 1;
85 }
86
87 BIO *getrbio(SSL *ssl, struct queue *q, int timeout) {
88     BIO *rbio;
89     struct timeval now;
90     struct timespec to;
91
92     pthread_mutex_lock(&q->mutex);
93     if (!(rbio = (BIO *)list_shift(q->entries))) {
94         if (timeout) {
95             gettimeofday(&now, NULL);
96             memset(&to, 0, sizeof(struct timespec));
97             to.tv_sec = now.tv_sec + timeout;
98             pthread_cond_timedwait(&q->cond, &q->mutex, &to);
99         } else
100             pthread_cond_wait(&q->cond, &q->mutex);
101         rbio = (BIO *)list_shift(q->entries);
102     }
103     pthread_mutex_unlock(&q->mutex);
104     return rbio;
105 }
106
107 int dtlsread(SSL *ssl, struct queue *q, unsigned char *buf, int num, int timeout) {
108     int len, cnt;
109     BIO *rbio;
110     
111     for (len = 0; len < num; len += cnt) {
112         cnt = SSL_read(ssl, buf + len, num - len);
113         if (cnt <= 0)
114             switch (cnt = SSL_get_error(ssl, cnt)) {
115             case SSL_ERROR_WANT_READ:
116                 rbio = getrbio(ssl, q, timeout);
117                 if (!rbio)
118                     return 0;
119                 BIO_free(ssl->rbio);            
120                 ssl->rbio = rbio;
121                 cnt = 0;
122                 continue;
123             case SSL_ERROR_WANT_WRITE:
124                 cnt = 0;
125                 continue;
126             case SSL_ERROR_ZERO_RETURN:
127                 /* remote end sent close_notify, send one back */
128                 SSL_shutdown(ssl);
129                 return -1;
130             default:
131                 return -1;
132             }
133     }
134     return num;
135 }
136
137 /* accept if acc == 1, else connect */
138 SSL *dtlsacccon(uint8_t acc, SSL_CTX *ctx, int s, struct sockaddr *addr, struct queue *rbios) {
139     SSL *ssl;
140     int i, res;
141     unsigned long error;
142     BIO *mem0bio, *wbio;
143
144     ssl = SSL_new(ctx);
145     if (!ssl)
146         return NULL;
147     
148     mem0bio = BIO_new(BIO_s_mem());
149     BIO_set_mem_eof_return(mem0bio, -1);
150     wbio = BIO_new_dgram(s, BIO_NOCLOSE);
151     BIO_dgram_set_peer(wbio, addr);
152     SSL_set_bio(ssl, mem0bio, wbio);
153
154     for (i = 0; i < 5; i++) {
155         res = acc ? SSL_accept(ssl) : SSL_connect(ssl);
156         if (res > 0)
157             return ssl;
158         if (res == 0)
159             break;
160         if (SSL_get_error(ssl, res) == SSL_ERROR_WANT_READ) {
161             BIO_free(ssl->rbio);
162             ssl->rbio = getrbio(ssl, rbios, 5);
163             if (!ssl->rbio)
164                 break;
165         }
166         while ((error = ERR_get_error()))
167             debug(DBG_ERR, "dtls%st: DTLS: %s", acc ? "accep" : "connec", ERR_error_string(error, NULL));
168     }
169
170     SSL_free(ssl);
171     return NULL;
172 }
173
174 unsigned char *raddtlsget(SSL *ssl, struct queue *rbios, int timeout) {
175     int cnt, len;
176     unsigned char buf[4], *rad;
177
178     for (;;) {
179         cnt = dtlsread(ssl, rbios, buf, 4, timeout);
180         if (cnt < 1) {
181             debug(DBG_DBG, cnt ? "raddtlsget: connection lost" : "raddtlsget: timeout");
182             return NULL;
183         }
184
185         len = RADLEN(buf);
186         rad = malloc(len);
187         if (!rad) {
188             debug(DBG_ERR, "raddtlsget: malloc failed");
189             continue;
190         }
191         memcpy(rad, buf, 4);
192         
193         cnt = dtlsread(ssl, rbios, rad + 4, len - 4, timeout);
194         if (cnt < 1) {
195             debug(DBG_DBG, cnt ? "raddtlsget: connection lost" : "raddtlsget: timeout");
196             free(rad);
197             return NULL;
198         }
199         
200         if (len >= 20)
201             break;
202         
203         free(rad);
204         debug(DBG_WARN, "raddtlsget: packet smaller than minimum radius size");
205     }
206     
207     debug(DBG_DBG, "raddtlsget: got %d bytes", len);
208     return rad;
209 }
210
211 void *dtlsserverwr(void *arg) {
212     int cnt;
213     unsigned long error;
214     struct client *client = (struct client *)arg;
215     struct queue *replyq;
216     struct reply *reply;
217     
218     debug(DBG_DBG, "dtlsserverwr: starting for %s", client->conf->host);
219     replyq = client->replyq;
220     for (;;) {
221         pthread_mutex_lock(&replyq->mutex);
222         while (!list_first(replyq->entries)) {
223             if (client->ssl) {      
224                 debug(DBG_DBG, "dtlsserverwr: waiting for signal");
225                 pthread_cond_wait(&replyq->cond, &replyq->mutex);
226                 debug(DBG_DBG, "dtlsserverwr: got signal");
227             }
228             if (!client->ssl) {
229                 /* ssl might have changed while waiting */
230                 pthread_mutex_unlock(&replyq->mutex);
231                 debug(DBG_DBG, "dtlsserverwr: exiting as requested");
232                 ERR_remove_state(0);
233                 pthread_exit(NULL);
234             }
235         }
236         reply = (struct reply *)list_shift(replyq->entries);
237         pthread_mutex_unlock(&replyq->mutex);
238         cnt = SSL_write(client->ssl, reply->buf, RADLEN(reply->buf));
239         if (cnt > 0)
240             debug(DBG_DBG, "dtlsserverwr: sent %d bytes, Radius packet of length %d",
241                   cnt, RADLEN(reply->buf));
242         else
243             while ((error = ERR_get_error()))
244                 debug(DBG_ERR, "dtlsserverwr: SSL: %s", ERR_error_string(error, NULL));
245         free(reply->buf);
246         free(reply);
247     }
248 }
249
250 void dtlsserverrd(struct client *client) {
251     struct request rq;
252     pthread_t dtlsserverwrth;
253     
254     debug(DBG_DBG, "dtlsserverrd: starting for %s", client->conf->host);
255
256     if (pthread_create(&dtlsserverwrth, NULL, dtlsserverwr, (void *)client)) {
257         debug(DBG_ERR, "dtlsserverrd: pthread_create failed");
258         return;
259     }
260
261     for (;;) {
262         memset(&rq, 0, sizeof(struct request));
263         rq.buf = raddtlsget(client->ssl, client->rbios, IDLE_TIMEOUT);
264         if (!rq.buf) {
265             debug(DBG_ERR, "dtlsserverrd: connection from %s lost", client->conf->host);
266             break;
267         }
268         debug(DBG_DBG, "dtlsserverrd: got Radius message from %s", client->conf->host);
269         rq.from = client;
270         if (!radsrv(&rq)) {
271             debug(DBG_ERR, "dtlsserverrd: message authentication/validation failed, closing connection from %s", client->conf->host);
272             break;
273         }
274     }
275     
276     /* stop writer by setting ssl to NULL and give signal in case waiting for data */
277     client->ssl = NULL;
278
279     pthread_mutex_lock(&client->replyq->mutex);
280     pthread_cond_signal(&client->replyq->cond);
281     pthread_mutex_unlock(&client->replyq->mutex);
282     debug(DBG_DBG, "dtlsserverrd: waiting for writer to end");
283     pthread_join(dtlsserverwrth, NULL);
284     removeclientrqs(client);
285     debug(DBG_DBG, "dtlsserverrd: reader for %s exiting", client->conf->host);
286 }
287
288 void *dtlsservernew(void *arg) {
289     struct dtlsservernewparams *params = (struct dtlsservernewparams *)arg;
290     struct client *client;
291     struct clsrvconf *conf;
292     struct list_node *cur = NULL;
293     SSL *ssl = NULL;
294     X509 *cert = NULL;
295     uint8_t delay = 60;
296
297     debug(DBG_DBG, "dtlsservernew: starting");
298     conf = find_clconf(RAD_DTLS, (struct sockaddr *)&params->addr, NULL);
299     if (conf) {
300         ssl = dtlsacccon(1, conf->ssl_ctx, params->sock, (struct sockaddr *)&params->addr, params->sesscache->rbios);
301         if (!ssl)
302             goto exit;
303         cert = verifytlscert(ssl);
304         if (!cert)
305             goto exit;
306     }
307
308     while (conf) {
309         if (verifyconfcert(cert, conf)) {
310             X509_free(cert);
311             client = addclient(conf);
312             if (client) {
313                 client->sock = params->sock;
314                 client->rbios = params->sesscache->rbios;
315                 client->addr = params->addr;
316                 client->ssl = ssl;
317                 dtlsserverrd(client);
318                 removeclient(client);
319                 delay = 0;
320             } else {
321                 debug(DBG_WARN, "dtlsservernew: failed to create new client instance");
322             }
323             goto exit;
324         }
325         conf = find_clconf(RAD_DTLS, (struct sockaddr *)&params->addr, &cur);
326     }
327     debug(DBG_WARN, "dtlsservernew: ignoring request, no matching TLS client");
328
329     if (cert)
330         X509_free(cert);
331
332  exit:
333     SSL_shutdown(ssl);
334     SSL_free(ssl);
335     pthread_mutex_lock(&params->sesscache->mutex);
336     freebios(params->sesscache->rbios);
337     params->sesscache->rbios = NULL;
338     gettimeofday(&params->sesscache->expiry, NULL);
339     params->sesscache->expiry.tv_sec += delay;
340     pthread_mutex_unlock(&params->sesscache->mutex);
341     free(params);
342     ERR_remove_state(0);
343     pthread_exit(NULL);
344     debug(DBG_DBG, "dtlsservernew: exiting");
345 }
346
347 void cacheexpire(struct hash *cache, struct timeval *last) {
348     struct timeval now;
349     struct hash_entry *he;
350     struct sessioncacheentry *e;
351     
352     gettimeofday(&now, NULL);
353     if (now.tv_sec - last->tv_sec < 19)
354         return;
355
356     for (he = hash_first(cache); he; he = hash_next(he)) {
357         e = (struct sessioncacheentry *)he->data;
358         pthread_mutex_lock(&e->mutex);
359         if (!e->expiry.tv_sec || e->expiry.tv_sec > now.tv_sec) {
360             pthread_mutex_unlock(&e->mutex);
361             continue;
362         }
363         debug(DBG_DBG, "cacheexpire: freeing entry");
364         hash_extract(cache, he->key, he->keylen);
365         if (e->rbios) {
366             freebios(e->rbios);
367             e->rbios = NULL;
368         }
369         pthread_mutex_unlock(&e->mutex);
370         pthread_mutex_destroy(&e->mutex);
371     }
372     last->tv_sec = now.tv_sec;
373 }
374
375 void *udpdtlsserverrd(void *arg) {
376     int ndesc, cnt, s = *(int *)arg;
377     unsigned char buf[4];
378     struct sockaddr_storage from;
379     socklen_t fromlen = sizeof(from);
380     struct dtlsservernewparams *params;
381     fd_set readfds;
382     struct timeval timeout, lastexpiry;
383     pthread_t dtlsserverth;
384     struct hash *sessioncache;
385     struct sessioncacheentry *cacheentry;
386     
387     sessioncache = hash_create();
388     if (!sessioncache)
389         debugx(1, DBG_ERR, "udpdtlsserverrd: malloc failed");
390     gettimeofday(&lastexpiry, NULL);
391     
392     for (;;) {
393         FD_ZERO(&readfds);
394         FD_SET(s, &readfds);
395         memset(&timeout, 0, sizeof(struct timeval));
396         timeout.tv_sec = 60;
397         ndesc = select(s + 1, &readfds, NULL, NULL, &timeout);
398         if (ndesc < 1) {
399             cacheexpire(sessioncache, &lastexpiry);
400             continue;
401         }
402         cnt = recvfrom(s, buf, 4, MSG_PEEK | MSG_TRUNC, (struct sockaddr *)&from, &fromlen);
403         if (cnt == -1) {
404             debug(DBG_WARN, "udpdtlsserverrd: recv failed");
405             cacheexpire(sessioncache, &lastexpiry);
406             continue;
407         }
408         cacheentry = hash_read(sessioncache, &from, fromlen);
409         if (cacheentry) {
410             debug(DBG_DBG, "udpdtlsserverrd: cache hit");
411             pthread_mutex_lock(&cacheentry->mutex);
412             if (cacheentry->rbios) {
413                 if (udp2bio(s, cacheentry->rbios, cnt))
414                     debug(DBG_DBG, "udpdtlsserverrd: got DTLS in UDP from %s", addr2string((struct sockaddr *)&from, fromlen));
415             } else
416                 recv(s, buf, 1, 0);
417             pthread_mutex_unlock(&cacheentry->mutex);
418             cacheexpire(sessioncache, &lastexpiry);
419             continue;
420         }
421
422         /* from new source */
423         debug(DBG_DBG, "udpdtlsserverrd: cache miss");
424         params = malloc(sizeof(struct dtlsservernewparams));
425         if (!params) {
426             cacheexpire(sessioncache, &lastexpiry);
427             recv(s, buf, 1, 0);
428             continue;
429         }
430         memset(params, 0, sizeof(struct dtlsservernewparams));
431         params->sesscache = malloc(sizeof(struct sessioncacheentry));
432         if (!params->sesscache) {
433             free(params);
434             cacheexpire(sessioncache, &lastexpiry);
435             recv(s, buf, 1, 0);
436             continue;
437         }
438         memset(params->sesscache, 0, sizeof(struct sessioncacheentry));
439         pthread_mutex_init(&params->sesscache->mutex, NULL);
440         params->sesscache->rbios = newqueue();
441         if (hash_insert(sessioncache, &from, fromlen, params->sesscache)) {
442             params->sock = s;
443             memcpy(&params->addr, &from, fromlen);
444
445             if (udp2bio(s, params->sesscache->rbios, cnt)) {
446                 debug(DBG_DBG, "udpdtlsserverrd: got DTLS in UDP from %s", addr2string((struct sockaddr *)&from, fromlen));
447                 if (!pthread_create(&dtlsserverth, NULL, dtlsservernew, (void *)params)) {
448                     pthread_detach(dtlsserverth);
449                     cacheexpire(sessioncache, &lastexpiry);
450                     continue;
451                 }
452                 debug(DBG_ERR, "udpdtlsserverrd: pthread_create failed");
453             }
454             hash_extract(sessioncache, &from, fromlen);
455         }
456         freebios(params->sesscache->rbios);
457         pthread_mutex_destroy(&params->sesscache->mutex);
458         free(params->sesscache);
459         free(params);
460         cacheexpire(sessioncache, &lastexpiry);
461     }
462 }
463
464 int dtlsconnect(struct server *server, struct timeval *when, int timeout, char *text) {
465     struct timeval now;
466     time_t elapsed;
467     X509 *cert;
468     
469     debug(DBG_DBG, "dtlsconnect: called from %s", text);
470     pthread_mutex_lock(&server->lock);
471     if (when && memcmp(&server->lastconnecttry, when, sizeof(struct timeval))) {
472         /* already reconnected, nothing to do */
473         debug(DBG_DBG, "dtlsconnect(%s): seems already reconnected", text);
474         pthread_mutex_unlock(&server->lock);
475         return 1;
476     }
477
478     for (;;) {
479         gettimeofday(&now, NULL);
480         elapsed = now.tv_sec - server->lastconnecttry.tv_sec;
481
482         if (timeout && server->lastconnecttry.tv_sec && elapsed > timeout) {
483             debug(DBG_DBG, "dtlsconnect: timeout");
484             SSL_free(server->ssl);
485             server->ssl = NULL;
486             pthread_mutex_unlock(&server->lock);
487             return 0;
488         }
489
490         if (server->connectionok) {
491             server->connectionok = 0;
492             sleep(2);
493         } else if (elapsed < 1)
494             sleep(2);
495         else if (elapsed < 60) {
496             debug(DBG_INFO, "dtlsconnect: sleeping %lds", elapsed);
497             sleep(elapsed);
498         } else if (elapsed < 100000) {
499             debug(DBG_INFO, "dtlsconnect: sleeping %ds", 60);
500             sleep(60);
501         } else
502             server->lastconnecttry.tv_sec = now.tv_sec;  /* no sleep at startup */
503         debug(DBG_WARN, "dtlsconnect: trying to open DTLS connection to %s port %s", server->conf->host, server->conf->port);
504
505         SSL_free(server->ssl);
506         server->ssl = dtlsacccon(0, server->conf->ssl_ctx, server->sock, server->conf->addrinfo->ai_addr, server->rbios);
507         if (!server->ssl)
508             continue;
509         debug(DBG_DBG, "dtlsconnect: DTLS: ok");
510         
511         cert = verifytlscert(server->ssl);
512         if (!cert)
513             continue;
514         
515         if (verifyconfcert(cert, server->conf))
516             break;
517         X509_free(cert);
518     }
519     X509_free(cert);
520     debug(DBG_WARN, "dtlsconnect: DTLS connection to %s port %s up", server->conf->host, server->conf->port);
521     gettimeofday(&server->lastconnecttry, NULL);
522     pthread_mutex_unlock(&server->lock);
523     return 1;
524 }
525
526 int clientradputdtls(struct server *server, unsigned char *rad) {
527     int cnt;
528     size_t len;
529     unsigned long error;
530     struct clsrvconf *conf = server->conf;
531
532     if (!server->ssl)
533         return 0;
534     len = RADLEN(rad);
535     while ((cnt = SSL_write(server->ssl, rad, len)) <= 0) {
536         while ((error = ERR_get_error()))
537             debug(DBG_ERR, "clientradputdtls: DTLS: %s", ERR_error_string(error, NULL));
538     }
539     debug(DBG_DBG, "clientradputdtls: Sent %d bytes, Radius packet of length %d to DTLS peer %s", cnt, len, conf->host);
540     return 1;
541 }
542
543 /* reads UDP containing DTLS and passes it on to dtlsclientrd */
544 void *udpdtlsclientrd(void *arg) {
545     int cnt, s = *(int *)arg;
546     unsigned char buf[4];
547     struct sockaddr_storage from;
548     socklen_t fromlen = sizeof(from);
549     struct clsrvconf *conf;
550     fd_set readfds;
551     
552     for (;;) {
553         FD_ZERO(&readfds);
554         FD_SET(s, &readfds);
555         if (select(s + 1, &readfds, NULL, NULL, NULL) < 1)
556             continue;
557         cnt = recvfrom(s, buf, 4, MSG_PEEK | MSG_TRUNC, (struct sockaddr *)&from, &fromlen);
558         if (cnt == -1) {
559             debug(DBG_WARN, "udpdtlsclientrd: recv failed");
560             continue;
561         }
562         
563         conf = find_srvconf(RAD_DTLS, (struct sockaddr *)&from, NULL);
564         if (!conf) {
565             debug(DBG_WARN, "udpdtlsclientrd: got packet from wrong or unknown DTLS peer %s, ignoring", addr2string((struct sockaddr *)&from, fromlen));
566             recv(s, buf, 4, 0);
567             continue;
568         }
569         if (udp2bio(s, conf->servers->rbios, cnt))
570             debug(DBG_DBG, "radudpget: got DTLS in UDP from %s", addr2string((struct sockaddr *)&from, fromlen));
571     }
572 }
573
574 void *dtlsclientrd(void *arg) {
575     struct server *server = (struct server *)arg;
576     unsigned char *buf;
577     struct timeval lastconnecttry;
578     int secs;
579     
580     for (;;) {
581         /* yes, lastconnecttry is really necessary */
582         lastconnecttry = server->lastconnecttry;
583         for (secs = 0; !(buf = raddtlsget(server->ssl, server->rbios, 10)) && !server->lostrqs && secs < IDLE_TIMEOUT; secs += 10);
584         if (!buf) {
585             dtlsconnect(server, &lastconnecttry, 0, "dtlsclientrd");
586             continue;
587         }
588         if (!replyh(server, buf))
589             free(buf);
590     }
591     ERR_remove_state(0);
592     server->clientrdgone = 1;
593     return NULL;
594 }
595
596 void addserverextradtls(struct clsrvconf *conf) {
597     switch (conf->addrinfo->ai_family) {
598     case AF_INET:
599         if (client4_sock < 0) {
600             client4_sock = bindtoaddr(getsrcprotores(RAD_DTLS), AF_INET, 0, 1);
601             if (client4_sock < 0)
602                 debugx(1, DBG_ERR, "addserver: failed to create client socket for server %s", conf->host);
603         }
604         conf->servers->sock = client4_sock;
605         break;
606     case AF_INET6:
607         if (client6_sock < 0) {
608             client6_sock = bindtoaddr(getsrcprotores(RAD_DTLS), AF_INET6, 0, 1);
609             if (client6_sock < 0)
610                 debugx(1, DBG_ERR, "addserver: failed to create client socket for server %s", conf->host);
611         }
612         conf->servers->sock = client6_sock;
613         break;
614     default:
615         debugx(1, DBG_ERR, "addserver: unsupported address family");
616     }
617 }
618
619 void initextradtls() {
620     pthread_t cl4th, cl6th;
621     
622     if (client4_sock >= 0)
623         if (pthread_create(&cl4th, NULL, udpdtlsclientrd, (void *)&client4_sock))
624             debugx(1, DBG_ERR, "pthread_create failed");
625     if (client6_sock >= 0)
626         if (pthread_create(&cl6th, NULL, udpdtlsclientrd, (void *)&client6_sock))
627             debugx(1, DBG_ERR, "pthread_create failed");
628 }