Add section start/stop debug output.
[freeradius.git] / src / main / realms.c
1 /*
2  * realms.c     Realm handling code
3  *
4  * Version:     $Id$
5  *
6  *   This program is free software; you can redistribute it and/or modify
7  *   it under the terms of the GNU General Public License as published by
8  *   the Free Software Foundation; either version 2 of the License, or
9  *   (at your option) any later version.
10  *
11  *   This program is distributed in the hope that it will be useful,
12  *   but WITHOUT ANY WARRANTY; without even the implied warranty of
13  *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14  *   GNU General Public License for more details.
15  *
16  *   You should have received a copy of the GNU General Public License
17  *   along with this program; if not, write to the Free Software
18  *   Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
19  *
20  * Copyright 2000,2006  The FreeRADIUS server project
21  * Copyright 2000  Miquel van Smoorenburg <miquels@cistron.nl>
22  * Copyright 2000  Alan DeKok <aland@ox.org>
23  */
24
25 #include <freeradius-devel/ident.h>
26 RCSID("$Id$")
27
28 #include <freeradius-devel/radiusd.h>
29 #include <freeradius-devel/rad_assert.h>
30
31 #include <sys/stat.h>
32
33 #include <ctype.h>
34 #include <fcntl.h>
35
36 static rbtree_t *realms_byname = NULL;
37
38 static rbtree_t *home_servers_byaddr = NULL;
39 static rbtree_t *home_servers_byname = NULL;
40
41 static rbtree_t *home_pools_byname = NULL;
42
43 static int realm_name_cmp(const void *one, const void *two)
44 {
45         const REALM *a = one;
46         const REALM *b = two;
47
48         return strcasecmp(a->name, b->name);
49 }
50
51
52 static int home_server_name_cmp(const void *one, const void *two)
53 {
54         const home_server *a = one;
55         const home_server *b = two;
56
57         return strcasecmp(a->name, b->name);
58 }
59
60 static int home_server_addr_cmp(const void *one, const void *two)
61 {
62         const home_server *a = one;
63         const home_server *b = two;
64
65         if (a->port < b->port) return -1;
66         if (a->port > b->port) return +1;
67
68         return lrad_ipaddr_cmp(&a->ipaddr, &b->ipaddr);
69 }
70
71
72 static int home_pool_name_cmp(const void *one, const void *two)
73 {
74         const home_pool_t *a = one;
75         const home_pool_t *b = two;
76
77         return strcasecmp(a->name, b->name);
78 }
79
80
81 void realms_free(void)
82 {
83         rbtree_free(home_servers_byname);
84         home_servers_byname = NULL;
85
86         rbtree_free(home_servers_byaddr);
87         home_servers_byaddr = NULL;
88
89         rbtree_free(home_pools_byname);
90         home_pools_byname = NULL;
91
92         rbtree_free(realms_byname);
93         realms_byname = NULL;
94 }
95
96
97 int realms_init(const char *filename)
98 {
99         CONF_SECTION *cs;
100
101         if (realms_byname) return 1;
102
103         realms_byname = rbtree_create(realm_name_cmp, free, 0);
104         if (!realms_byname) {
105                 realms_free();
106                 return 0;
107         }
108
109         home_servers_byaddr = rbtree_create(home_server_addr_cmp, free, 0);
110         if (!home_servers_byaddr) {
111                 realms_free();
112                 return 0;
113         }
114
115         home_servers_byname = rbtree_create(home_server_name_cmp, NULL, 0);
116         if (!home_servers_byname) {
117                 realms_free();
118                 return 0;
119         }
120
121         home_pools_byname = rbtree_create(home_pool_name_cmp, free, 0);
122         if (!home_pools_byname) {
123                 realms_free();
124                 return 0;
125         }
126
127         for (cs = cf_subsection_find_next(mainconfig.config, NULL, "realm");
128              cs != NULL;
129              cs = cf_subsection_find_next(mainconfig.config, cs, "realm")) {
130                 if (!realm_add(filename, cs)) {
131                         realms_free();
132                         return 0;
133                 }
134         }
135
136         return 1;
137 }
138
139 static struct in_addr hs_ip4addr;
140 static struct in6_addr hs_ip6addr;
141 static char *hs_type = NULL;
142 static char *hs_check = NULL;
143
144 static CONF_PARSER home_server_config[] = {
145         { "ipaddr",  PW_TYPE_IPADDR,
146           0, &hs_ip4addr,  NULL },
147         { "ipv6addr",  PW_TYPE_IPV6ADDR,
148           0, &hs_ip6addr, NULL },
149
150         { "hostname",  PW_TYPE_STRING_PTR,
151           offsetof(home_server,hostname), NULL,  NULL},
152         { "port", PW_TYPE_INTEGER,
153           offsetof(home_server,port), NULL,   "0" },
154
155         { "type",  PW_TYPE_STRING_PTR,
156           0, &hs_type, NULL },
157
158         { "secret",  PW_TYPE_STRING_PTR,
159           offsetof(home_server,secret), NULL,  NULL},
160
161         { "response_window", PW_TYPE_INTEGER,
162           offsetof(home_server,response_window), NULL,   "30" },
163         { "max_outstanding", PW_TYPE_INTEGER,
164           offsetof(home_server,max_outstanding), NULL,   "65536" },
165
166         { "zombie_period", PW_TYPE_INTEGER,
167           offsetof(home_server,zombie_period), NULL,   "40" },
168         { "status_check", PW_TYPE_STRING_PTR,
169           0, &hs_check,   "none" },
170         { "ping_check", PW_TYPE_STRING_PTR,
171           0, &hs_check,   "none" },
172
173         { "ping_interval", PW_TYPE_INTEGER,
174           offsetof(home_server,ping_interval), NULL,   "30" },
175         { "check_interval", PW_TYPE_INTEGER,
176           offsetof(home_server,ping_interval), NULL,   "30" },
177         { "num_answers_to_alive", PW_TYPE_INTEGER,
178           offsetof(home_server,num_pings_to_alive), NULL,   "3" },
179         { "num_pings_to_alive", PW_TYPE_INTEGER,
180           offsetof(home_server,num_pings_to_alive), NULL,   "3" },
181         { "revive_interval", PW_TYPE_INTEGER,
182           offsetof(home_server,revive_interval), NULL,   "300" },
183         { "status_check_timeout", PW_TYPE_INTEGER,
184           offsetof(home_server,ping_timeout), NULL,   "4" },
185
186         { "username",  PW_TYPE_STRING_PTR,
187           offsetof(home_server,ping_user_name), NULL,  NULL},
188         { "password",  PW_TYPE_STRING_PTR,
189           offsetof(home_server,ping_user_password), NULL,  NULL},
190         
191         { NULL, -1, 0, NULL, NULL }             /* end the list */
192
193 };
194
195
196 static int home_server_add(const char *filename, CONF_SECTION *cs)
197 {
198         const char *name2;
199         home_server *home;
200
201         name2 = cf_section_name1(cs);
202         if (!name2 || (strcasecmp(name2, "home_server") != 0)) {
203                 radlog(L_ERR, "%s[%d]: Section is not a home_server.",
204                        filename, cf_section_lineno(cs));
205                 return 0;
206         }
207
208         name2 = cf_section_name2(cs);
209         if (!name2) {
210                 radlog(L_ERR, "%s[%d]: Home server section is missing a name.",
211                        filename, cf_section_lineno(cs));
212                 return 0;
213         }
214
215         home = rad_malloc(sizeof(*home));
216         memset(home, 0, sizeof(*home));
217
218         home->name = name2;
219         
220         memset(&hs_ip4addr, 0, sizeof(hs_ip4addr));
221         memset(&hs_ip6addr, 0, sizeof(hs_ip6addr));
222         cf_section_parse(cs, home, home_server_config);
223
224         if (!home->hostname && (htonl(hs_ip4addr.s_addr) == INADDR_NONE) &&
225             IN6_IS_ADDR_UNSPECIFIED(&hs_ip6addr)) {
226                 radlog(L_ERR, "%s[%d]: No hostname, IPv4 address, or IPv6 address defined for home server %s.",
227                        filename, cf_section_lineno(cs), name2);
228                 free(home);
229                 free(hs_type);
230                 hs_type = NULL;
231                 free(hs_check);
232                 hs_check = NULL;
233                 return 0;
234         }
235
236         /*
237          *      FIXME: Parse home->hostname!
238          *
239          *      Right now, only ipaddr && ip6addr are used.
240          *      The old-style parsing still allows hostnames.
241          */
242         if (htonl(hs_ip4addr.s_addr) != INADDR_NONE) {
243                 home->ipaddr.af = AF_INET;
244                 home->ipaddr.ipaddr.ip4addr = hs_ip4addr;
245
246         } else if (!IN6_IS_ADDR_UNSPECIFIED(&hs_ip6addr)) {
247                 home->ipaddr.af = AF_INET6;
248                 home->ipaddr.ipaddr.ip6addr = hs_ip6addr;
249
250         } else {
251                 radlog(L_ERR, "%s[%d]: FIXME: parse hostname for home server %s.",
252                        filename, cf_section_lineno(cs), name2);
253                 free(home);
254                 free(hs_type);
255                 hs_type = NULL;
256                 free(hs_check);
257                 hs_check = NULL;
258                 return 0;
259         }
260
261         if (!home->port || (home->port > 65535)) {
262                 radlog(L_ERR, "%s[%d]: No port, or invalid port defined for home server %s.",
263                        filename, cf_section_lineno(cs), name2);
264                 free(home);
265                 free(hs_type);
266                 hs_type = NULL;
267                 free(hs_check);
268                 hs_check = NULL;
269                 return 0;
270         }
271
272         if (0) {
273                 radlog(L_ERR, "%s[%d]: Fatal error!  Home server %s is ourselves!",
274                        filename, cf_section_lineno(cs), name2);
275                 free(home);
276                 free(hs_type);
277                 hs_type = NULL;
278                 free(hs_check);
279                 hs_check = NULL;
280                 return 0;
281         }
282
283         if (strcasecmp(hs_type, "auth") == 0) {
284                 home->type = HOME_TYPE_AUTH;
285
286         } else if (strcasecmp(hs_type, "acct") == 0) {
287                 home->type = HOME_TYPE_ACCT;
288
289         } else {
290                 radlog(L_ERR, "%s[%d]: Invalid type \"%s\" for home server %s.",
291                        filename, cf_section_lineno(cs), hs_type, name2);
292                 free(home);
293                 free(hs_type);
294                 hs_type = NULL;
295                 free(hs_check);
296                 hs_check = NULL;
297                 return 0;
298         }
299         free(hs_type);
300         hs_type = NULL;
301
302         if (!home->secret) {
303                 radlog(L_ERR, "%s[%d]: No shared secret defined for home server %s.",
304                        filename, cf_section_lineno(cs), name2);
305                 free(home);
306                 return 0;
307         }
308
309         if (strcasecmp(hs_check, "none") == 0) {
310                 home->ping_check = HOME_PING_CHECK_NONE;
311
312         } else if (strcasecmp(hs_check, "status-server") == 0) {
313                 home->ping_check = HOME_PING_CHECK_STATUS_SERVER;
314
315         } else if (strcasecmp(hs_check, "request") == 0) {
316                 home->ping_check = HOME_PING_CHECK_REQUEST;
317
318         } else {
319                 radlog(L_ERR, "%s[%d]: Invalid ping_check \"%s\" for home server %s.",
320                        filename, cf_section_lineno(cs), hs_check, name2);
321                 free(home);
322                 free(hs_check);
323                 hs_check = NULL;
324                 return 0;
325         }
326         free(hs_check);
327         hs_check = NULL;
328
329         if ((home->ping_check != HOME_PING_CHECK_NONE) &&
330             (home->ping_check != HOME_PING_CHECK_STATUS_SERVER)) {
331                 if (!home->ping_user_name) {
332                         radlog(L_INFO, "%s[%d]: You must supply a user name to enable ping checks",
333                                filename, cf_section_lineno(cs));
334                         free(home);
335                         return 0;
336                 }
337
338                 if ((home->type == HOME_TYPE_AUTH) && 
339                     !home->ping_user_password) {        
340                         radlog(L_INFO, "%s[%d]: You must supply a password to enable ping checks",
341                                filename, cf_section_lineno(cs));
342                         free(home);
343                         return 0;
344                 }
345         }
346
347         if (rbtree_finddata(home_servers_byaddr, home)) {
348                 radlog(L_INFO, "%s[%d]: Ignoring duplicate home server %s.",
349                        filename, cf_section_lineno(cs), name2);
350                 return 1;
351         }
352
353         if (!rbtree_insert(home_servers_byname, home)) {
354                 radlog(L_ERR, "%s[%d]: Internal error adding home server %s.",
355                        filename, cf_section_lineno(cs), name2);
356                 free(home);
357                 return 0;
358         }
359
360         if (!rbtree_insert(home_servers_byaddr, home)) {
361                 rbtree_deletebydata(home_servers_byname, home);
362                 radlog(L_ERR, "%s[%d]: Internal error adding home server %s.",
363                        filename, cf_section_lineno(cs), name2);
364                 free(home);
365                 return 0;
366         }
367
368         if (home->response_window < 5) home->response_window = 5;
369         if (home->response_window > 60) home->response_window = 60;
370
371         if (home->max_outstanding < 8) home->max_outstanding = 8;
372         if (home->max_outstanding > 65536*16) home->max_outstanding = 65536*16;
373
374         if (home->ping_interval < 6) home->ping_interval = 6;
375         if (home->ping_interval > 120) home->ping_interval = 120;
376
377         if (home->zombie_period < 20) home->zombie_period = 20;
378         if (home->zombie_period > 120) home->zombie_period = 120;
379
380         if (home->zombie_period < home->response_window) {
381                 home->zombie_period = home->response_window;
382         }
383
384         if (home->num_pings_to_alive < 3) home->num_pings_to_alive = 3;
385         if (home->num_pings_to_alive > 10) home->num_pings_to_alive = 10;
386
387         if (home->ping_timeout < 3) home->ping_timeout = 3;
388         if (home->ping_timeout > 10) home->ping_timeout = 10;
389
390         if (home->revive_interval < 60) home->revive_interval = 60;
391         if (home->revive_interval > 3600) home->revive_interval = 3600;
392
393         return 1;
394 }
395
396
397 static int server_pool_add(const char *filename, CONF_SECTION *cs)
398 {
399         const char *name2;
400         home_pool_t *pool = NULL;
401         const char *value;
402         CONF_PAIR *cp;
403         int num_home_servers;
404
405         name2 = cf_section_name1(cs);
406         if (!name2 || (strcasecmp(name2, "server_pool") != 0)) {
407                 radlog(L_ERR, "%s[%d]: Section is not a server_pool.",
408                        filename, cf_section_lineno(cs));
409                 return 0;
410         }
411
412         name2 = cf_section_name2(cs);
413         if (!name2) {
414                 radlog(L_ERR, "%s[%d]: Server pool section is missing a name.",
415                        filename, cf_section_lineno(cs));
416                 return 0;
417         }
418
419         DEBUG2(" server_pool %s {", name2);
420
421         num_home_servers = 0;
422         for (cp = cf_pair_find(cs, "home_server");
423              cp != NULL;
424              cp = cf_pair_find_next(cs, cp, "home_server")) {
425                 num_home_servers++;
426         }
427
428         if (num_home_servers == 0) {
429                 radlog(L_ERR, "%s[%d]: No home servers defined in pool %s",
430                        filename, cf_section_lineno(cs), name2);
431                 goto error;
432         }
433
434         pool = rad_malloc(sizeof(*pool) + num_home_servers * sizeof(pool->servers[0]));
435         memset(pool, 0, sizeof(*pool) + num_home_servers * sizeof(pool->servers[0]));
436
437         pool->type = HOME_POOL_FAIL_OVER;
438         pool->name = name2;
439
440         cp = cf_pair_find(cs, "type");
441         if (cp) {
442                 static LRAD_NAME_NUMBER pool_types[] = {
443                         { "load-balance", HOME_POOL_LOAD_BALANCE },
444                         { "fail-over", HOME_POOL_FAIL_OVER },
445                         { "round_robin", HOME_POOL_LOAD_BALANCE },
446                         { "fail_over", HOME_POOL_FAIL_OVER },
447                         { "client-balance", HOME_POOL_CLIENT_BALANCE },
448                         { NULL, 0 }
449                 };
450
451                 value = cf_pair_value(cp);
452                 if (!value) {
453                         radlog(L_ERR, "%s[%d]: No value given for type.",
454                                filename, cf_pair_lineno(cp));
455                         goto error;
456                 }
457
458                 pool->type = lrad_str2int(pool_types, value, 0);
459                 if (!pool->type) {
460                         radlog(L_ERR, "%s[%d]: Unknown type \"%s\".",
461                                filename, cf_pair_lineno(cp), value);
462                         goto error;
463                 }
464
465                 DEBUG2("\ttype = %s", name2, value);
466         }
467
468         for (cp = cf_pair_find(cs, "home_server");
469              cp != NULL;
470              cp = cf_pair_find_next(cs, cp, "home_server")) {
471                 home_server myhome, *home;
472
473                 value = cf_pair_value(cp);
474                 if (!value) {
475                         radlog(L_ERR, "%s[%d]: No value given for home_server.",
476                                filename, cf_pair_lineno(cp));
477                         goto error;
478                 }
479
480                 myhome.name = value;
481
482                 home = rbtree_finddata(home_servers_byname, &myhome);
483                 if (!home) {
484                         CONF_SECTION *server_cs;
485
486                         server_cs = cf_section_sub_find_name2(NULL,
487                                                               "home_server",
488                                                               value);
489                         if (!server_cs) {
490                                 radlog(L_ERR, "%s[%d]: Unknown home_server \"%s\".",
491                                        filename, cf_pair_lineno(cp), value);
492                                 goto error;
493                         }
494                         
495                         DEBUG2(" server_pool %s: Adding home_server %s",
496                                name2, value);
497
498                         if (!home_server_add(filename, server_cs)) {
499                                 goto error;
500                         }
501
502                         home = rbtree_finddata(home_servers_byname, &myhome);
503                         if (!home) {
504                                 rad_assert("Internal sanity check failed");
505                                 goto error;
506                         }
507                 }
508
509                 if (!pool->server_type) {
510                         rad_assert(home->type != 0);
511                         pool->server_type = home->type;
512
513                 } else if (pool->server_type != home->type) {
514                         radlog(L_ERR, "%s[%d]: Home server \"%s\" is not of the same type as previous servers in server pool %s",
515                                filename, cf_pair_lineno(cp), value, pool->name);
516                         goto error;
517                 }
518
519                 if (0) {
520                         DEBUG2("Warning: Duplicate home server %s in server pool %s", home->name, pool->name);
521                         continue;
522                 }
523
524                 DEBUG2("\thome_server = %s", name2, home->name);
525                 pool->servers[pool->num_home_servers] = home;
526                 pool->num_home_servers++;
527
528         } /* loop over home_server's */
529
530         if (!rbtree_insert(home_pools_byname, pool)) {
531                 rad_assert("Internal sanity check failed");
532                 goto error;
533         }
534
535         DEBUG2(" } # server_pool %s", name2);
536
537         rad_assert(pool->server_type != 0);
538         
539         return 1;
540
541  error:
542         DEBUG2(" } # server_pool %s", name2);
543         free(pool);
544         return 0;
545 }
546
547
548 static int old_server_add(const char *filename, int lineno,
549                           const char *name, const char *secret,
550                           home_pool_type_t ldflag, home_pool_t **pool_p,
551                           int type)
552 {
553         int i, insert_point, num_home_servers;
554         home_server myhome, *home;
555         home_pool_t mypool, *pool;
556         CONF_SECTION *cs;
557
558         /*
559          *      LOCAL realms get sanity checked, and nothing else happens.
560          */
561         if (strcmp(name, "LOCAL") == 0) {
562                 if (*pool_p) {
563                         radlog(L_ERR, "%s[%d]: Realm \"%s\" cannot be both LOCAL and remote", filename, lineno, name);
564                         return 0;
565                 }
566                 return 1;
567         }
568
569         mypool.name = name;
570         pool = rbtree_finddata(home_pools_byname, &mypool);
571         if (pool) {
572                 if (pool->type != ldflag) {
573                         radlog(L_ERR, "%s[%d]: Inconsistent ldflag for server pool \"%s\"", filename, lineno, name);
574                         return 0;
575                 }
576
577                 if (pool->server_type != type) {
578                         radlog(L_ERR, "%s[%d]: Inconsistent home server type for server pool \"%s\"", filename, lineno, name);
579                         return 0;
580                 }
581         }
582
583         myhome.name = name;
584         home = rbtree_finddata(home_servers_byname, &myhome);
585         if (home) {
586                 if (strcmp(home->secret, secret) != 0) {
587                         radlog(L_ERR, "%s[%d]: Inconsistent shared secret for home server \"%s\"", filename, lineno, name);
588                         return 0;
589                 }
590
591                 if (home->type != type) {
592                         radlog(L_ERR, "%s[%d]: Inconsistent type for home server \"%s\"", filename, lineno, name);
593                         return 0;
594                 }
595                 
596                 /*
597                  *      See if the home server is already listed
598                  *      in the pool.  If so, do nothing else.
599                  */
600                 if (pool) for (i = 0; i < pool->num_home_servers; i++) {
601                         if (pool->servers[i] == home) {
602                                 return 1;
603                         }
604                 }
605         }
606
607         /*
608          *      If we do have a pool, check that there is room to
609          *      insert the home server we've found, or the one that we
610          *      create here.
611          *
612          *      Note that we insert it into the LAST available
613          *      position, in order to maintain the same order as in
614          *      the configuration files.
615          */
616         insert_point = -1;
617         if (pool) {
618                 for (i = pool->num_home_servers - 1; i >= 0; i--) {
619                         if (pool->servers[i]) break;
620
621                         if (!pool->servers[i]) {
622                                 insert_point = i;
623                         }
624                 }
625
626                 if (insert_point < 0) {
627                         radlog(L_ERR, "%s[%d]: No room in pool to add home server \"%s\".  Please update the realm configuration to use the new-style home servers and server pools.", filename, lineno, name);
628                         return 0;
629                 }
630         }
631
632         /*
633          *      No home server, allocate one.
634          */
635         if (!home) {         
636                 const char *p;
637                 char *q;
638
639                 home = rad_malloc(sizeof(*home));
640                 memset(home, 0, sizeof(*home));
641
642                 home->name = name;
643                 home->hostname = name;
644                 home->type = type;
645                 home->secret = secret;
646
647                 p = strchr(name, ':');
648                 if (!p) {
649                         if (type == HOME_TYPE_AUTH) {
650                                 home->port = PW_AUTH_UDP_PORT;
651                         } else {
652                                 home->port = PW_ACCT_UDP_PORT;
653                         }
654
655                         p = name;
656                         q = NULL;
657
658                 } else if (p == name) {
659                                 radlog(L_ERR, "%s[%d]: Invalid hostname %s.",
660                                        filename, lineno, name);
661                                 free(home);
662                                 return 0;
663
664                 } else {
665                         home->port = atoi(p + 1);
666                         if ((home->port == 0) || (home->port > 65535)) {
667                                 radlog(L_ERR, "%s[%d]: Invalid port %s.",
668                                        filename, lineno, p + 1);
669                                 free(home);
670                                 return 0;
671                         }
672
673                         q = rad_malloc((p - name) + 1);
674                         memcpy(q, name, (p - name));
675                         q[p - name] = '\0';
676                         p = q;
677                 }
678
679                 if (ip_hton(p, AF_UNSPEC, &home->ipaddr) < 0) {
680                         radlog(L_ERR, "%s[%d]: Failed looking up hostname %s.",
681                                filename, lineno, p);
682                         free(home);
683                         free(q);
684                         return 0;
685                 }
686                 free(q);
687
688                 /*
689                  *      Use the old-style configuration.
690                  */
691                 home->max_outstanding = 65535*16;
692                 home->zombie_period = mainconfig.proxy_retry_delay * mainconfig.proxy_retry_count;
693                 if (home->zombie_period == 0) home->zombie_period =30;
694                 home->response_window = home->zombie_period - 1;
695
696                 home->ping_check = HOME_PING_CHECK_NONE;
697
698                 home->revive_interval = mainconfig.proxy_dead_time;
699
700                 if (rbtree_finddata(home_servers_byaddr, home)) {
701                         radlog(L_ERR, "%s[%d]: Home server %s has the same IP address as another home server.",
702                                filename, lineno, name);
703                         free(home);
704                         return 0;
705                 }
706
707                 if (!rbtree_insert(home_servers_byname, home)) {
708                         radlog(L_ERR, "%s[%d]: Internal error adding home server %s.",
709                                filename, lineno, name);
710                         free(home);
711                         return 0;
712                 }
713                 
714                 if (!rbtree_insert(home_servers_byaddr, home)) {
715                         rbtree_deletebydata(home_servers_byname, home);
716                         radlog(L_ERR, "%s[%d]: Internal error adding home server %s.",
717                                filename, lineno, name);
718                         free(home);
719                         return 0;
720                 }
721         }
722
723         /*
724          *      We now have a home server, see if we can insert it
725          *      into pre-existing pool.
726          */
727         if (insert_point >= 0) {
728                 rad_assert(pool != NULL);
729                 pool->servers[insert_point] = home;
730                 return 1;
731         }
732
733         rad_assert(pool == NULL);
734         rad_assert(home != NULL);
735
736         /*
737          *      Count the old-style realms of this name.
738          */
739         num_home_servers = 0;
740         for (cs = cf_section_sub_find_name2(mainconfig.config, "realm", name);
741              cs != NULL;
742              cs = cf_section_sub_find_name2(cs, "realm", name)) {
743                 num_home_servers++;
744         }
745
746
747         pool = rad_malloc(sizeof(*pool) + num_home_servers * sizeof(pool->servers[0]));
748         memset(pool, 0, sizeof(*pool) + num_home_servers * sizeof(pool->servers[0]));
749
750         pool->name = name;
751         pool->type = ldflag;
752         pool->server_type = type;
753         pool->num_home_servers = num_home_servers;
754         pool->servers[0] = home;
755
756         if (!rbtree_insert(home_pools_byname, pool)) {
757                 rad_assert("Internal sanity check failed");
758                 return 0;
759         }
760
761         *pool_p = pool;
762
763         return 1;
764 }
765
766 static int old_realm_config(const char *filename, CONF_SECTION *cs, REALM *r)
767 {
768         char *host;
769         const char *secret;
770         home_pool_type_t ldflag;
771
772         secret = cf_section_value_find(cs, "secret");
773
774         host = cf_section_value_find(cs, "ldflag");
775         if (!host ||
776             (strcasecmp(host, "fail_over") == 0)) {
777                 ldflag = HOME_POOL_FAIL_OVER;
778                 DEBUG2("\tldflag = fail_over", r->name);
779
780         } else if (strcasecmp(host, "round_robin") == 0) {
781                 ldflag = HOME_POOL_LOAD_BALANCE;
782                 DEBUG2("\tldflag = round_robin", r->name);
783
784         } else {
785                 radlog(L_ERR, "%s[%d]: Unknown value \"%s\" for ldflag",
786                        filename, cf_section_lineno(cs), host);
787                 return 0;
788         }
789
790         /*
791          *      Allow old-style if it doesn't exist, or if it exists and
792          *      it's LOCAL.
793          */
794         if (((host = cf_section_value_find(cs, "authhost")) != NULL) &&
795             (strcmp(host, "LOCAL") != 0)) {
796                 if (!secret) {
797                         radlog(L_ERR, "%s[%d]: No shared secret supplied for realm: %s",
798                                filename, cf_section_lineno(cs), r->name);
799                         return 0;
800                 }
801
802                 DEBUG2("\tauthhost = %s",  r->name, host);
803
804                 if (!old_server_add(filename, cf_section_lineno(cs),
805                                     host, secret, ldflag,
806                                     &r->auth_pool, HOME_TYPE_AUTH)) {
807                         return 0;
808                 }
809         }
810
811         if (((host = cf_section_value_find(cs, "accthost")) != NULL) &&
812             (strcmp(host, "LOCAL") != 0)) {
813                 if (!secret) {
814                         radlog(L_ERR, "%s[%d]: No shared secret supplied for realm: %s",
815                                filename, cf_section_lineno(cs), r->name);
816                         return 0;
817                 }
818
819                 DEBUG2("\taccthost = %s", r->name, host);
820
821                 if (!old_server_add(filename, cf_section_lineno(cs),
822                                     host, secret, ldflag,
823                                     &r->auth_pool, HOME_TYPE_ACCT)) {
824                         return 0;
825                 }
826         }
827
828         if (secret) DEBUG2("\tsecret = %s", r->name, secret);
829
830         return 1;
831         
832 }
833
834
835 static int add_pool_to_realm(const char *filename, int lineno,
836                              const char *name, home_pool_t **dest,
837                              int server_type)
838 {
839         home_pool_t mypool, *pool;
840
841         mypool.name = name;
842         pool = rbtree_finddata(home_pools_byname, &mypool);
843         if (!pool) {
844                 CONF_SECTION *pool_cs;
845
846                 pool_cs = cf_section_sub_find_name2(NULL, "server_pool",
847                                                     name);
848                 if (!pool_cs) {
849                         radlog(L_ERR, "%s[%d]: Failed to find server_pool \"%s\"",
850                                filename, lineno, name);
851                         return 0;
852                 }
853
854                 if (!server_pool_add(filename, pool_cs)) {
855                         return 0;
856                 }
857                 
858                 pool = rbtree_finddata(home_pools_byname, &mypool);
859                 if (!pool) {
860                         rad_assert("Internal sanity check failed");
861                         return 0;
862                 }
863         }
864
865         if (pool->server_type != server_type) {
866                 radlog(L_ERR, "%s[%d]: Incompatible server_pool \"%s\" (mixed auth_pool / acct_pool)",
867                        filename, lineno, name);
868                 return 0;
869         }
870
871         *dest = pool;
872
873         return 1;
874 }
875
876 int realm_add(const char *filename, CONF_SECTION *cs)
877 {
878         const char *name2;
879         char *pool = NULL;
880         REALM *r = NULL;
881         CONF_PAIR *cp;
882
883         name2 = cf_section_name1(cs);
884         if (!name2 || (strcasecmp(name2, "realm") != 0)) {
885                 radlog(L_ERR, "%s[%d]: Section is not a realm.",
886                        filename, cf_section_lineno(cs));
887                 return 0;
888         }
889
890         name2 = cf_section_name2(cs);
891         if (!name2) {
892                 radlog(L_ERR, "%s[%d]: Realm section is missing the realm name.",
893                        filename, cf_section_lineno(cs));
894                 return 0;
895         }
896
897         DEBUG2(" realm %s {", name2);
898
899         /*
900          *      The realm MAY already exist if it's an old-style realm.
901          *      In that case, merge the old-style realm with this one.
902          */
903         r = realm_find(name2);
904         if (r) {
905                 if (cf_pair_find(cs, "auth_pool") ||
906                     cf_pair_find(cs, "acct_pool")) {
907                         radlog(L_ERR, "%s[%d]: Duplicate realm \"%s\"",
908                                filename, cf_section_lineno(cs), name2);
909                         goto error;
910                 }
911
912                 if (!old_realm_config(filename, cs, r)) {
913                         goto error;
914                 }
915
916                 DEBUG2(" } # realm %s", name2);
917                 return 1;
918         }
919
920         r = rad_malloc(sizeof(*r));
921         memset(r, 0, sizeof(*r));
922
923         r->name = name2;
924
925         /*
926          *      Prefer new configuration to old one.
927          */
928         cp = cf_pair_find(cs, "auth_pool");
929         if (cp) pool = cf_pair_value(cp);
930         if (cp && pool) {
931                 DEBUG2(" realm %s: Adding server_pool %s",
932                        name2, pool);
933                 if (!add_pool_to_realm(filename, cf_pair_lineno(cp),
934                                        pool, &r->auth_pool, HOME_TYPE_AUTH)) {
935                         goto error;
936                 }
937                 DEBUG2("\tauth_pool = %s", name2, pool);
938         }
939
940         cp = cf_pair_find(cs, "acct_pool");
941         if (cp) pool = cf_pair_value(cp);
942         if (cp && pool) {
943                 DEBUG2(" realm %s: Adding server_pool %s",
944                        name2, pool);
945                 if (!add_pool_to_realm(filename, cf_pair_lineno(cp),
946                                        pool, &r->acct_pool, HOME_TYPE_ACCT)) {
947                         goto error;
948                 }
949                 DEBUG2("\tacct_pool = %s", name2, pool);
950         }
951
952         r->striprealm = 1;
953         
954         if ((cf_section_value_find(cs, "nostrip")) != NULL) {
955                 r->striprealm = 0;
956                 DEBUG2("\tnostrip", name2);
957         }
958
959         /*
960          *      We're a new-style realm.  Complain if we see the old
961          *      directives.
962          */
963         if (r->auth_pool || r->acct_pool) {
964                 if (((cp = cf_pair_find(cs, "authhost")) != NULL) ||
965                     ((cp = cf_pair_find(cs, "accthost")) != NULL) ||
966                     ((cp = cf_pair_find(cs, "secret")) != NULL) ||
967                     ((cp = cf_pair_find(cs, "ldflag")) != NULL)) {
968                         DEBUG2("WARNING: Ignoring old-style configuration entry \"%s\" in realm \"%s\"", cf_pair_attr(cp), r->name);
969                 }
970
971
972                 /*
973                  *      The realm MAY be an old-style realm, as there
974                  *      was no auth_pool or acct_pool.  Double-check
975                  *      it, just to be safe.
976                  */
977         } else if (!old_realm_config(filename, cs, r)) {
978                 goto error;
979         }
980
981         if (!rbtree_insert(realms_byname, r)) {
982                 rad_assert("Internal sanity check failed");
983                 goto error;
984         }
985
986         DEBUG2(" } # realm %s", name2);
987
988         return 1;
989
990  error:
991         DEBUG2(" } # realm %s", name2);
992         free(r);
993         return 0;
994 }
995
996
997 /*
998  *      Find a realm in the REALM list.
999  */
1000 REALM *realm_find(const char *name)
1001 {
1002         REALM myrealm;
1003
1004         if (!name) name = "NULL";
1005
1006         myrealm.name = name;
1007         return rbtree_finddata(realms_byname, &myrealm);
1008 }
1009
1010
1011 home_server *home_server_ldb(const char *realmname,
1012                              home_pool_t *pool, REQUEST *request)
1013 {
1014         int             start;
1015         int             count;
1016
1017         start = 0;
1018
1019         /*
1020          *      Determine how to pick choose the home server.
1021          */
1022         switch (pool->type) {
1023                 uint32_t hash;
1024
1025                 /*
1026                  *      Load balancing.  Pick one at random.
1027                  */
1028         case HOME_POOL_LOAD_BALANCE:
1029                 hash = lrad_rand();
1030                 start = hash % pool->num_home_servers;
1031                 break;
1032
1033                 /*
1034                  *      For load-balancing by client IP address, we
1035                  *      pick a home server by hashing the client IP.
1036                  *
1037                  *      This isn't as even a load distribution as
1038                  *      tracking the State attribute, but it's better
1039                  *      than nothing.
1040                  */
1041         case HOME_POOL_CLIENT_BALANCE:
1042                 switch (request->packet->src_ipaddr.af) {
1043                 case AF_INET:
1044                         hash = lrad_hash(&request->packet->src_ipaddr.ipaddr.ip4addr,
1045                                          sizeof(request->packet->src_ipaddr.ipaddr.ip4addr));
1046                         break;
1047                 case AF_INET6:
1048                         hash = lrad_hash(&request->packet->src_ipaddr.ipaddr.ip6addr,
1049                                          sizeof(request->packet->src_ipaddr.ipaddr.ip6addr));
1050                         break;
1051                 default:
1052                         hash = 0;
1053                         break;
1054                 }
1055                 start = hash % pool->num_home_servers;
1056                 break;
1057
1058         default:
1059                 start = 0;
1060                 break;
1061         }
1062
1063         /*
1064          *      Starting with the home server we chose, loop through
1065          *      all home servers.  If the current one is dead, skip
1066          *      it.  If it is too busy, skip it.
1067          *
1068          *      Otherwise, use it.
1069          */
1070         for (count = 0; count < pool->num_home_servers; count++) {
1071                 home_server *home = pool->servers[(start + count) % pool->num_home_servers];
1072
1073                 if (home->state == HOME_STATE_IS_DEAD) {
1074                         continue;
1075                 }
1076
1077                 /*
1078                  *      This home server is too busy.  Choose another one.
1079                  */
1080                 if (home->currently_outstanding >= home->max_outstanding) {
1081                         continue;
1082                 }
1083
1084                 return home;
1085         } /* loop over the home servers */
1086
1087         /*
1088          *      No live match found, and no fallback to the "DEFAULT"
1089          *      realm.  We fix this by blindly marking all servers as
1090          *      "live".  But only do it for ones that don't support
1091          *      "pings", as they will be marked live when they
1092          *      actually are live.
1093          */
1094         if (!mainconfig.proxy_fallback &&
1095             mainconfig.wake_all_if_all_dead) {
1096                 home_server *lb = NULL;
1097
1098                 for (count = 0; count < pool->num_home_servers; count++) {
1099                         home_server *home = pool->servers[count];
1100
1101                         if ((home->state == HOME_STATE_IS_DEAD) &&
1102                             (home->ping_check == HOME_PING_CHECK_NONE)) {
1103                                 home->state = HOME_STATE_ALIVE;
1104                                 if (!lb) lb = home;
1105                         }
1106                 }
1107
1108                 if (lb) return lb;
1109         }
1110
1111         /*
1112          *      Still nothing.  Look up the DEFAULT realm, but only
1113          *      if we weren't looking up the NULL or DEFAULT realms.
1114          */
1115         if (mainconfig.proxy_fallback &&
1116             realmname &&
1117             (strcmp(realmname, "NULL") != 0) &&
1118             (strcmp(realmname, "DEFAULT") != 0)) {
1119                 REALM *rd = realm_find("DEFAULT");
1120
1121                 if (!rd) return NULL;
1122
1123                 pool = NULL;
1124                 if (request->packet->code == PW_AUTHENTICATION_REQUEST) {
1125                         pool = rd->auth_pool;
1126                         
1127                 } else if (request->packet->code == PW_ACCOUNTING_REQUEST) {
1128                         pool = rd->acct_pool;
1129                 }
1130                 if (!pool) return NULL;
1131
1132                 DEBUG2("  Realm %s has no live home servers.  Falling back to the DEFAULT realm.", realmname);
1133                 return home_server_ldb(rd->name, pool, request);
1134         }
1135
1136         /*
1137          *      Still haven't found anything.  Oh well.
1138          */
1139         return NULL;
1140 }
1141
1142
1143 home_server *home_server_find(lrad_ipaddr_t *ipaddr, int port)
1144 {
1145         home_server myhome;
1146
1147         myhome.ipaddr = *ipaddr;
1148         myhome.port = port;
1149
1150         return rbtree_finddata(home_servers_byaddr, &myhome);
1151 }