Formatting
[freeradius.git] / src / modules / rlm_realm / trustrouter.c
1 /*
2  *   This program is is free software; you can redistribute it and/or modify
3  *   it under the terms of the GNU General Public License, version 2 if the
4  *   License as published by the Free Software Foundation.
5  *
6  *   This program is distributed in the hope that it will be useful,
7  *   but WITHOUT ANY WARRANTY; without even the implied warranty of
8  *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
9  *   GNU General Public License for more details.
10  *
11  *   You should have received a copy of the GNU General Public License
12  *   along with this program; if not, write to the Free Software
13  *   Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
14  */
15
16 /**
17  * $Id$
18  * @file trustrouter.c
19  * @brief Integration with external trust router code
20  *
21  * @copyright 2014 Network RADIUS SARL
22  */
23 #include <trust_router/tid.h>
24 #include <freeradius-devel/radiusd.h>
25 #include <freeradius-devel/rad_assert.h>
26 #include <freeradius-devel/modules.h>
27 #include <freeradius-devel/realms.h>
28
29 #ifdef HAVE_TRUST_ROUTER_TR_DH_H
30 #include "trustrouter.h"
31
32 #include <trust_router/tr_dh.h>
33 static TIDC_INSTANCE *global_tidc = NULL;
34
35 struct resp_opaque {
36         REALM *orig_realm;
37         REALM *output_realm;
38         TID_RC result;
39         char err_msg[1024];
40         char *fr_realm_name;
41 };
42
43
44 bool tr_init(void) 
45 {
46         if (global_tidc) return true;
47
48         global_tidc = tidc_create();
49         if (!global_tidc) {
50                 DEBUG2( "tr_init: Error creating global TIDC instance.\n");
51                 return false;
52         }
53
54         if (!tidc_set_dh(global_tidc, tr_create_dh_params(NULL, 0))) {
55                 DEBUG2( "tr_init: Error creating client DH params.\n");
56                 return false;
57         }
58
59         return true;
60 }
61
62 static fr_tls_server_conf_t *construct_tls(TIDC_INSTANCE *inst,
63                                            home_server_t *hs,
64                                            TID_SRVR_BLK *server)
65 {
66         fr_tls_server_conf_t *tls;
67         unsigned char *key_buf = NULL;
68         ssize_t keylen;
69         char *hexbuf = NULL;
70         DH *aaa_server_dh;
71
72         tls = talloc_zero( hs, fr_tls_server_conf_t);
73         if (!tls) return NULL;
74
75         aaa_server_dh = tid_srvr_get_dh(server);
76         keylen = tr_compute_dh_key(&key_buf, aaa_server_dh->pub_key,
77                                    tidc_get_dh(inst));
78         if (keylen <= 0) {
79                 DEBUG2("DH error");
80                 goto error;
81         }
82
83         hexbuf = talloc_size(tls, keylen*2 + 1);
84         if (!hexbuf) goto error;
85
86         tr_bin_to_hex(key_buf, keylen, hexbuf, 2*keylen + 1);
87
88         tls->psk_password = hexbuf;
89         tls->psk_identity = talloc_strdup(tls, tid_srvr_get_key_name(server)->buf);
90
91         tls->cipher_list = talloc_strdup(tls, "PSK");
92         tls->fragment_size = 4200;
93         tls->ctx = tls_init_ctx(tls, 1);
94         if (!tls->ctx) goto error;
95
96         memset(key_buf, 0, keylen);
97         tr_dh_free(key_buf);
98         return tls;
99
100 error:
101         if (key_buf) {
102                 memset(key_buf, 0, keylen);
103                 tr_dh_free(key_buf);
104         }
105         if (hexbuf) memset(hexbuf, 0, keylen*2);
106
107         if (tls) talloc_free(tls);
108         return NULL;
109 }
110   
111 static char *build_pool_name(TALLOC_CTX *ctx, TID_RESP *resp)
112 {
113         size_t index, sa_len, sl;
114         TID_SRVR_BLK *server;
115         char *pool_name = NULL;
116         char addr_buf[256];
117         const struct sockaddr *sa;
118         pool_name = talloc_strdup(ctx, "hp-");
119
120         tid_resp_servers_foreach(resp, server, index) {
121                 tid_srvr_get_address(server, &sa, &sa_len);
122                 if (0 != getnameinfo(sa, sa_len,
123                                      addr_buf, sizeof(addr_buf)-1,
124                                      NULL, 0, NI_NUMERICHOST)) {
125                         DEBUG2("getnameinfo failed");
126                         return NULL;
127                 }
128
129                 sl = strlen(addr_buf);
130                 rad_assert(sl+2 <= sizeof(addr_buf));
131
132                 addr_buf[sl] = '-';
133                 addr_buf[sl+1] = '\0';
134
135                 pool_name = talloc_strdup_append(pool_name, addr_buf);
136         }
137
138         return pool_name;
139 }
140
141 static home_server_t *srvr_blk_to_home_server(TALLOC_CTX *ctx,
142                                               TIDC_INSTANCE *inst,
143                                               TID_SRVR_BLK *blk,
144                                               char const *realm_name)
145 {
146         home_server_t *hs = NULL;
147         const struct sockaddr *sa = NULL;
148         size_t sa_len = 0;
149         fr_ipaddr_t home_server_ip;
150         uint16_t port;
151         char nametemp[256];
152
153         rad_assert(blk != NULL);
154         tid_srvr_get_address(blk, &sa, &sa_len);
155         switch (sa->sa_family) {
156
157         case AF_INET: {
158                 const struct sockaddr_in *sin = (const struct sockaddr_in *) sa;
159                 home_server_ip.af = AF_INET;
160                 home_server_ip.scope = 0;
161                 home_server_ip.ipaddr.ip4addr = sin->sin_addr;
162                 port = ntohs(sin->sin_port);
163                 break;
164         }
165
166         case AF_INET6: {
167                 const struct sockaddr_in6 *sin6 = (const struct sockaddr_in6 *) sa;
168                 home_server_ip.af = AF_INET6;
169                 home_server_ip.scope = sin6->sin6_scope_id;
170                 home_server_ip.ipaddr.ip6addr = sin6->sin6_addr;
171                 break;
172         }
173
174         default:
175                 DEBUG2("Unknown address family in tid srvr block");
176                 return NULL;
177         }
178   
179         if (0 != getnameinfo(sa, sa_len,
180                              nametemp,
181                              sizeof nametemp,
182                              NULL, 0,
183                              NI_NUMERICHOST)) {
184                 DEBUG2("getnameinfo failed");
185                 return NULL;
186         }
187
188         hs = talloc_zero(ctx, home_server_t);
189         if (!hs) return NULL;
190
191         /*
192          *      All dynamic home servers are for authentication.
193          */
194         hs->type = HOME_TYPE_AUTH;
195         hs->ipaddr = home_server_ip;
196         hs->src_ipaddr.af = home_server_ip.af;
197         hs->name = talloc_asprintf(hs, "%s-for-%s", nametemp, realm_name);
198         hs->hostname = talloc_strdup(hs, nametemp);
199         hs->port = port;
200         hs->proto = IPPROTO_TCP;
201         hs->secret = talloc_strdup(hs, "radsec");
202         hs->response_window.tv_sec = 30;
203         hs->last_packet_recv = time(NULL);
204
205         hs->tls = construct_tls(inst, hs, blk);
206         if (!hs->tls) goto error;
207
208         realm_home_server_sanitize(hs, NULL);
209
210         return hs;
211 error:
212         talloc_free(hs);
213         return NULL;
214 }
215
216 static home_pool_t *servers_to_pool(TALLOC_CTX *ctx,
217                                     TIDC_INSTANCE *inst,
218                                     TID_RESP *resp,
219                                     const char *realm_name)
220 {
221         home_pool_t *pool = NULL;
222         size_t num_servers = 0, index;
223         TID_SRVR_BLK *server = NULL;
224
225         num_servers = tid_resp_get_num_servers(resp);
226
227         pool = talloc_zero_size(ctx, sizeof(*pool) + num_servers *sizeof(home_server_t *));
228         if (!pool) goto error;
229
230         pool->type = HOME_POOL_CLIENT_PORT_BALANCE;
231         pool->server_type = HOME_TYPE_AUTH;
232
233         pool->name = build_pool_name(pool, resp);
234         if (!pool->name) goto error;
235
236         pool->num_home_servers = num_servers;
237
238         tid_resp_servers_foreach(resp, server, index) {
239                 home_server_t *hs;
240
241                 hs = srvr_blk_to_home_server(pool, inst, server, realm_name);
242                 if (!hs) goto error;
243                 pool->servers[index] = hs;
244         }
245
246         return pool;
247
248 error:
249         if (pool) talloc_free(pool);
250
251         return NULL;
252 }
253
254 static void tr_response_func( TIDC_INSTANCE *inst,
255                               UNUSED TID_REQ *req, TID_RESP *resp,
256                               void *cookie)
257 {
258         struct resp_opaque  *opaque = (struct resp_opaque *) cookie;
259         REALM *nr = opaque->orig_realm;
260
261         if (tid_resp_get_result(resp) != TID_SUCCESS) {
262
263                 size_t err_msg_len;
264                 opaque->result = tid_resp_get_result(resp);
265                 memset(opaque->err_msg, 0, sizeof(opaque->err_msg));
266
267                 if (tid_resp_get_err_msg(resp)) {
268                         TR_NAME *err_msg = tid_resp_get_err_msg(resp);
269                         err_msg_len = err_msg->len+1;
270                         if (err_msg_len > sizeof(opaque->err_msg))
271                                 err_msg_len = sizeof(opaque->err_msg);
272                         strlcpy(opaque->err_msg, err_msg->buf, err_msg_len);
273                 }
274                 return;
275         }
276                 
277         if (!nr) {
278                 nr = talloc_zero(NULL, REALM);
279                 if (!nr) goto error;
280                 nr->name = talloc_move(nr, &opaque->fr_realm_name);
281                 nr->auth_pool = servers_to_pool(nr, inst, resp, opaque->fr_realm_name);
282                 if (!realm_realm_add(nr, NULL)) goto error;
283
284         } else {
285                 home_pool_t *old_pool = nr->auth_pool;
286                 home_pool_t *new_pool;
287
288                 new_pool = servers_to_pool(nr, inst, resp, opaque->fr_realm_name);
289                 if (!new_pool) {
290                         ERROR("Unable to recreate pool for %s", opaque->fr_realm_name);
291                         goto error;
292                 }
293                 nr->auth_pool = new_pool;
294
295                 /*
296                  *      Mark the old pool as "to be freed"
297                  */
298                 realm_pool_free(old_pool);
299         }
300
301         opaque->output_realm = nr;
302         return;
303                 
304 error:
305         if (nr && !opaque->orig_realm) {
306                 talloc_free(nr);
307         }
308
309         return;
310 }
311
312 static bool update_required(REALM const *r)
313 {
314         const home_pool_t *pool;
315         int i;
316         const home_server_t *server;
317         time_t now = time(NULL);
318
319         /*
320          *      No pool.  Not our realm.
321          */
322         if (!r->auth_pool) return false;
323
324         pool = r->auth_pool;
325
326         for (i = 0; i < pool->num_home_servers; i++) {
327                 server = pool->servers[i];
328
329                 /*
330                  *      The realm was loaded from the configuration
331                  *      files.
332                  */
333                 if (server->cs) return false;
334
335                 /*
336                  *      These values don't make sense.
337                  */
338                 if ((server->last_packet_recv > (now + 5)) || 
339                     (server->last_failed_open > (now + 5))) {
340                         continue;
341                 }
342
343                 /*
344                  *      This server has received a packet in the last
345                  *      5 minutes.  It doesn't need an update.
346                  */
347                 if ((now - server->last_packet_recv) < 300) {
348                         return false;
349                 }
350
351                 /*
352                  *      If we've opened in the last 10 minutes, then
353                  *      open rather than update.
354                  */
355                 if ((now - server->last_failed_open) > 600) {
356                         return false;
357                 }
358         }
359
360         return true;
361 }
362
363     
364
365 REALM *tr_query_realm(char const *realm,
366                       char const  *community,
367                       char const *rprealm,
368                       char const *trustrouter,
369                       unsigned int port)
370 {
371         int conn = 0;
372         int rcode;
373         gss_ctx_id_t gssctx;
374         struct resp_opaque cookie;
375
376         if (!realm) return NULL;
377
378         /* clear the cookie structure */
379         memset (&cookie, 0, sizeof(cookie));
380
381         cookie.fr_realm_name = talloc_asprintf(NULL,
382                                                "%s%%%s",
383                                                community, realm);
384
385         cookie.orig_realm = cookie.output_realm = realm_find(cookie.fr_realm_name);
386
387         if (cookie.orig_realm && !update_required(cookie.orig_realm)) {
388                 talloc_free(cookie.fr_realm_name);
389                 return cookie.orig_realm;
390         }
391     
392         /* Set-up TID connection */
393         DEBUG2("Opening TIDC connection to %s:%u", trustrouter, port);
394
395         conn = tidc_open_connection(global_tidc, (char *)trustrouter, port, &gssctx);
396         if (conn < 0) {
397                 /* Handle error */
398                 DEBUG2("Error in tidc_open_connection.\n");
399                 goto cleanup;
400         }
401
402         /* Send a TID request */
403         rcode = tidc_send_request(global_tidc, conn, gssctx, (char *)rprealm, 
404                                   (char *) realm, (char *)community, 
405                                   &tr_response_func, &cookie);
406         if (rcode < 0) {
407                 /* Handle error */
408                 DEBUG2("Error in tidc_send_request, rc = %d.\n", rcode);
409                 goto cleanup;
410         }
411
412 cleanup:
413         if (cookie.fr_realm_name)
414                 talloc_free(cookie.fr_realm_name);
415
416         return cookie.output_realm;
417 }
418 #endif  /* HAVE_TRUST_ROUTER_TR_DH_H */