Rename "ping" check to "status" check. This means existing
[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
171         { "check_interval", PW_TYPE_INTEGER,
172           offsetof(home_server,ping_interval), NULL,   "30" },
173         { "num_answers_to_alive", PW_TYPE_INTEGER,
174           offsetof(home_server,num_pings_to_alive), NULL,   "3" },
175         { "revive_interval", PW_TYPE_INTEGER,
176           offsetof(home_server,revive_interval), NULL,   "300" },
177
178         { "username",  PW_TYPE_STRING_PTR,
179           offsetof(home_server,ping_user_name), NULL,  NULL},
180         { "password",  PW_TYPE_STRING_PTR,
181           offsetof(home_server,ping_user_password), NULL,  NULL},
182         
183         { NULL, -1, 0, NULL, NULL }             /* end the list */
184
185 };
186
187
188 static int home_server_add(const char *filename, CONF_SECTION *cs)
189 {
190         const char *name2;
191         home_server *home;
192
193         name2 = cf_section_name1(cs);
194         if (!name2 || (strcasecmp(name2, "home_server") != 0)) {
195                 radlog(L_ERR, "%s[%d]: Section is not a home_server.",
196                        filename, cf_section_lineno(cs));
197                 return 0;
198         }
199
200         name2 = cf_section_name2(cs);
201         if (!name2) {
202                 radlog(L_ERR, "%s[%d]: Home server section is missing a name.",
203                        filename, cf_section_lineno(cs));
204                 return 0;
205         }
206
207         home = rad_malloc(sizeof(*home));
208         memset(home, 0, sizeof(*home));
209
210         home->name = name2;
211         
212         memset(&hs_ip4addr, 0, sizeof(hs_ip4addr));
213         memset(&hs_ip6addr, 0, sizeof(hs_ip6addr));
214         cf_section_parse(cs, home, home_server_config);
215
216         if (!home->hostname && (htonl(hs_ip4addr.s_addr) == INADDR_NONE) &&
217             IN6_IS_ADDR_UNSPECIFIED(&hs_ip6addr)) {
218                 radlog(L_ERR, "%s[%d]: No hostname, IPv4 address, or IPv6 address defined for home server %s.",
219                        filename, cf_section_lineno(cs), name2);
220                 free(home);
221                 free(hs_type);
222                 hs_type = NULL;
223                 free(hs_check);
224                 hs_check = NULL;
225                 return 0;
226         }
227
228         /*
229          *      FIXME: Parse home->hostname!
230          *
231          *      Right now, only ipaddr && ip6addr are used.
232          *      The old-style parsing still allows hostnames.
233          */
234         if (htonl(hs_ip4addr.s_addr) != INADDR_NONE) {
235                 home->ipaddr.af = AF_INET;
236                 home->ipaddr.ipaddr.ip4addr = hs_ip4addr;
237
238         } else if (!IN6_IS_ADDR_UNSPECIFIED(&hs_ip6addr)) {
239                 home->ipaddr.af = AF_INET6;
240                 home->ipaddr.ipaddr.ip6addr = hs_ip6addr;
241
242         } else {
243                 radlog(L_ERR, "%s[%d]: FIXME: parse hostname for home server %s.",
244                        filename, cf_section_lineno(cs), name2);
245                 free(home);
246                 free(hs_type);
247                 hs_type = NULL;
248                 free(hs_check);
249                 hs_check = NULL;
250                 return 0;
251         }
252
253         if (!home->port || (home->port > 65535)) {
254                 radlog(L_ERR, "%s[%d]: No port, or invalid port defined for home server %s.",
255                        filename, cf_section_lineno(cs), name2);
256                 free(home);
257                 free(hs_type);
258                 hs_type = NULL;
259                 free(hs_check);
260                 hs_check = NULL;
261                 return 0;
262         }
263
264         if (0) {
265                 radlog(L_ERR, "%s[%d]: Fatal error!  Home server %s is ourselves!",
266                        filename, cf_section_lineno(cs), name2);
267                 free(home);
268                 free(hs_type);
269                 hs_type = NULL;
270                 free(hs_check);
271                 hs_check = NULL;
272                 return 0;
273         }
274
275         if (strcasecmp(hs_type, "auth") == 0) {
276                 home->type = HOME_TYPE_AUTH;
277
278         } else if (strcasecmp(hs_type, "acct") == 0) {
279                 home->type = HOME_TYPE_ACCT;
280
281         } else {
282                 radlog(L_ERR, "%s[%d]: Invalid type \"%s\" for home server %s.",
283                        filename, cf_section_lineno(cs), hs_type, name2);
284                 free(home);
285                 free(hs_type);
286                 hs_type = NULL;
287                 free(hs_check);
288                 hs_check = NULL;
289                 return 0;
290         }
291         free(hs_type);
292         hs_type = NULL;
293
294         if (!home->secret) {
295                 radlog(L_ERR, "%s[%d]: No shared secret defined for home server %s.",
296                        filename, cf_section_lineno(cs), name2);
297                 free(home);
298                 return 0;
299         }
300
301         if (strcasecmp(hs_check, "none") == 0) {
302                 home->ping_check = HOME_PING_CHECK_NONE;
303
304         } else if (strcasecmp(hs_check, "status-server") == 0) {
305                 home->ping_check = HOME_PING_CHECK_STATUS_SERVER;
306
307         } else if (strcasecmp(hs_check, "request") == 0) {
308                 home->ping_check = HOME_PING_CHECK_REQUEST;
309
310         } else {
311                 radlog(L_ERR, "%s[%d]: Invalid ping_check \"%s\" for home server %s.",
312                        filename, cf_section_lineno(cs), hs_check, name2);
313                 free(home);
314                 free(hs_check);
315                 hs_check = NULL;
316                 return 0;
317         }
318         free(hs_check);
319         hs_check = NULL;
320
321         if ((home->ping_check != HOME_PING_CHECK_NONE) &&
322             (home->ping_check != HOME_PING_CHECK_STATUS_SERVER)) {
323                 if (!home->ping_user_name) {
324                         radlog(L_INFO, "%s[%d]: You must supply a user name to enable ping checks",
325                                filename, cf_section_lineno(cs));
326                         free(home);
327                         return 0;
328                 }
329
330                 if ((home->type == HOME_TYPE_AUTH) && 
331                     !home->ping_user_password) {        
332                         radlog(L_INFO, "%s[%d]: You must supply a password to enable ping checks",
333                                filename, cf_section_lineno(cs));
334                         free(home);
335                         return 0;
336                 }
337         }
338
339         if (rbtree_finddata(home_servers_byaddr, home)) {
340                 radlog(L_INFO, "%s[%d]: Ignoring duplicate home server %s.",
341                        filename, cf_section_lineno(cs), name2);
342                 return 1;
343         }
344
345         if (!rbtree_insert(home_servers_byname, home)) {
346                 radlog(L_ERR, "%s[%d]: Internal error adding home server %s.",
347                        filename, cf_section_lineno(cs), name2);
348                 free(home);
349                 return 0;
350         }
351
352         if (!rbtree_insert(home_servers_byaddr, home)) {
353                 rbtree_deletebydata(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 (home->response_window < 5) home->response_window = 5;
361         if (home->response_window > 60) home->response_window = 60;
362
363         if (home->max_outstanding < 8) home->max_outstanding = 8;
364         if (home->max_outstanding > 65536*16) home->max_outstanding = 65536*16;
365
366         if (home->ping_interval < 6) home->ping_interval = 6;
367         if (home->ping_interval > 120) home->ping_interval = 120;
368
369         if (home->zombie_period < 20) home->zombie_period = 20;
370         if (home->zombie_period > 120) home->zombie_period = 120;
371
372         if (home->zombie_period < home->response_window) {
373                 home->zombie_period = home->response_window;
374         }
375
376         if (home->num_pings_to_alive < 3) home->num_pings_to_alive = 3;
377         if (home->num_pings_to_alive > 10) home->num_pings_to_alive = 10;
378
379         if (home->revive_interval < 60) home->revive_interval = 60;
380         if (home->revive_interval > 3600) home->revive_interval = 3600;
381
382         return 1;
383 }
384
385
386 static int server_pool_add(const char *filename, CONF_SECTION *cs)
387 {
388         const char *name2;
389         home_pool_t *pool;
390         const char *value;
391         CONF_PAIR *cp;
392         int num_home_servers;
393
394         name2 = cf_section_name1(cs);
395         if (!name2 || (strcasecmp(name2, "server_pool") != 0)) {
396                 radlog(L_ERR, "%s[%d]: Section is not a server_pool.",
397                        filename, cf_section_lineno(cs));
398                 return 0;
399         }
400
401         name2 = cf_section_name2(cs);
402         if (!name2) {
403                 radlog(L_ERR, "%s[%d]: Server pool section is missing a name.",
404                        filename, cf_section_lineno(cs));
405                 return 0;
406         }
407
408         num_home_servers = 0;
409         for (cp = cf_pair_find(cs, "home_server");
410              cp != NULL;
411              cp = cf_pair_find_next(cs, cp, "home_server")) {
412                 num_home_servers++;
413         }
414
415         if (num_home_servers == 0) {
416                 radlog(L_ERR, "%s[%d]: No home servers defined in pool %s",
417                        filename, cf_section_lineno(cs), name2);
418                 return 0;
419         }
420
421         pool = rad_malloc(sizeof(*pool) + num_home_servers * sizeof(pool->servers[0]));
422         memset(pool, 0, sizeof(*pool) + num_home_servers * sizeof(pool->servers[0]));
423
424         pool->type = HOME_POOL_FAIL_OVER;
425         pool->name = name2;
426
427         cp = cf_pair_find(cs, "type");
428         if (cp) {
429                 static LRAD_NAME_NUMBER pool_types[] = {
430                         { "load-balance", HOME_POOL_LOAD_BALANCE },
431                         { "fail-over", HOME_POOL_FAIL_OVER },
432                         { "round_robin", HOME_POOL_LOAD_BALANCE },
433                         { "fail_over", HOME_POOL_FAIL_OVER },
434                         { "client-balance", HOME_POOL_CLIENT_BALANCE },
435                         { NULL, 0 }
436                 };
437
438                 value = cf_pair_value(cp);
439                 if (!value) {
440                         radlog(L_ERR, "%s[%d]: No value given for type.",
441                                filename, cf_pair_lineno(cp));
442                         free(pool);
443                         return 0;
444                 }
445
446                 pool->type = lrad_str2int(pool_types, value, 0);
447                 if (!pool->type) {
448                         radlog(L_ERR, "%s[%d]: Unknown type \"%s\".",
449                                filename, cf_pair_lineno(cp), value);
450                         free(pool);
451                         return 0;
452                 }
453
454                 DEBUG2(" server_pool %s: type = %s", name2, value);
455         }
456
457         for (cp = cf_pair_find(cs, "home_server");
458              cp != NULL;
459              cp = cf_pair_find_next(cs, cp, "home_server")) {
460                 home_server myhome, *home;
461
462                 value = cf_pair_value(cp);
463                 if (!value) {
464                         radlog(L_ERR, "%s[%d]: No value given for home_server.",
465                                filename, cf_pair_lineno(cp));
466                         free(pool);
467                         return 0;
468                 }
469
470                 myhome.name = value;
471
472                 home = rbtree_finddata(home_servers_byname, &myhome);
473                 if (!home) {
474                         CONF_SECTION *server_cs;
475
476                         server_cs = cf_section_sub_find_name2(NULL,
477                                                               "home_server",
478                                                               value);
479                         if (!server_cs) {
480                                 radlog(L_ERR, "%s[%d]: Unknown home_server \"%s\".",
481                                        filename, cf_pair_lineno(cp), value);
482                                 free(pool);
483                                 return 0;
484                         }
485                         
486                         if (!home_server_add(filename, server_cs)) {
487                                 free(pool);
488                                 return 0;
489                         }
490
491                         home = rbtree_finddata(home_servers_byname, &myhome);
492                         if (!home) {
493                                 rad_assert("Internal sanity check failed");
494                                 return 0;
495                         }
496                 }
497
498                 if (!pool->server_type) {
499                         rad_assert(home->type != 0);
500                         pool->server_type = home->type;
501
502                 } else if (pool->server_type != home->type) {
503                         radlog(L_ERR, "%s[%d]: Home server \"%s\" is not of the same type as previous servers in server pool %s",
504                                filename, cf_pair_lineno(cp), value, pool->name);
505                         free(pool);
506                         return 0;
507                 }
508
509                 if (0) {
510                         DEBUG2("Warning: Duplicate home server %s in server pool %s", home->name, pool->name);
511                         continue;
512                 }
513
514                 DEBUG2(" server_pool %s: home_server = %s", name2, home->name);
515                 pool->servers[pool->num_home_servers] = home;
516                 pool->num_home_servers++;
517
518         } /* loop over home_server's */
519
520         if (!rbtree_insert(home_pools_byname, pool)) {
521                 rad_assert("Internal sanity check failed");
522                 return 0;
523         }
524
525         rad_assert(pool->server_type != 0);
526         
527         return 1;
528 }
529
530
531 static int old_server_add(const char *filename, int lineno,
532                           const char *name, const char *secret,
533                           home_pool_type_t ldflag, home_pool_t **pool_p,
534                           int type)
535 {
536         int i, insert_point, num_home_servers;
537         home_server myhome, *home;
538         home_pool_t mypool, *pool;
539         CONF_SECTION *cs;
540
541         /*
542          *      LOCAL realms get sanity checked, and nothing else happens.
543          */
544         if (strcmp(name, "LOCAL") == 0) {
545                 if (*pool_p) {
546                         radlog(L_ERR, "%s[%d]: Realm \"%s\" cannot be both LOCAL and remote", filename, lineno, name);
547                         return 0;
548                 }
549                 return 1;
550         }
551
552         mypool.name = name;
553         pool = rbtree_finddata(home_pools_byname, &mypool);
554         if (pool) {
555                 if (pool->type != ldflag) {
556                         radlog(L_ERR, "%s[%d]: Inconsistent ldflag for server pool \"%s\"", filename, lineno, name);
557                         return 0;
558                 }
559
560                 if (pool->server_type != type) {
561                         radlog(L_ERR, "%s[%d]: Inconsistent home server type for server pool \"%s\"", filename, lineno, name);
562                         return 0;
563                 }
564         }
565
566         myhome.name = name;
567         home = rbtree_finddata(home_servers_byname, &myhome);
568         if (home) {
569                 if (strcmp(home->secret, secret) != 0) {
570                         radlog(L_ERR, "%s[%d]: Inconsistent shared secret for home server \"%s\"", filename, lineno, name);
571                         return 0;
572                 }
573
574                 if (home->type != type) {
575                         radlog(L_ERR, "%s[%d]: Inconsistent type for home server \"%s\"", filename, lineno, name);
576                         return 0;
577                 }
578                 
579                 /*
580                  *      See if the home server is already listed
581                  *      in the pool.  If so, do nothing else.
582                  */
583                 if (pool) for (i = 0; i < pool->num_home_servers; i++) {
584                         if (pool->servers[i] == home) {
585                                 return 1;
586                         }
587                 }
588         }
589
590         /*
591          *      If we do have a pool, check that there is room to
592          *      insert the home server we've found, or the one that we
593          *      create here.
594          *
595          *      Note that we insert it into the LAST available
596          *      position, in order to maintain the same order as in
597          *      the configuration files.
598          */
599         insert_point = -1;
600         if (pool) {
601                 for (i = pool->num_home_servers - 1; i >= 0; i--) {
602                         if (pool->servers[i]) break;
603
604                         if (!pool->servers[i]) {
605                                 insert_point = i;
606                         }
607                 }
608
609                 if (insert_point < 0) {
610                         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);
611                         return 0;
612                 }
613         }
614
615         /*
616          *      No home server, allocate one.
617          */
618         if (!home) {         
619                 const char *p;
620                 char *q;
621
622                 home = rad_malloc(sizeof(*home));
623                 memset(home, 0, sizeof(*home));
624
625                 home->name = name;
626                 home->hostname = name;
627                 home->type = type;
628                 home->secret = secret;
629
630                 p = strchr(name, ':');
631                 if (!p) {
632                         if (type == HOME_TYPE_AUTH) {
633                                 home->port = PW_AUTH_UDP_PORT;
634                         } else {
635                                 home->port = PW_ACCT_UDP_PORT;
636                         }
637
638                         p = name;
639                         q = NULL;
640
641                 } else if (p == name) {
642                                 radlog(L_ERR, "%s[%d]: Invalid hostname %s.",
643                                        filename, lineno, name);
644                                 free(home);
645                                 return 0;
646
647                 } else {
648                         home->port = atoi(p + 1);
649                         if ((home->port == 0) || (home->port > 65535)) {
650                                 radlog(L_ERR, "%s[%d]: Invalid port %s.",
651                                        filename, lineno, p + 1);
652                                 free(home);
653                                 return 0;
654                         }
655
656                         q = rad_malloc((p - name) + 1);
657                         memcpy(q, name, (p - name));
658                         q[p - name] = '\0';
659                         p = q;
660                 }
661
662                 if (ip_hton(p, AF_UNSPEC, &home->ipaddr) < 0) {
663                         radlog(L_ERR, "%s[%d]: Failed looking up hostname %s.",
664                                filename, lineno, p);
665                         free(home);
666                         free(q);
667                         return 0;
668                 }
669                 free(q);
670
671                 /*
672                  *      Use the old-style configuration.
673                  */
674                 home->max_outstanding = 65535*16;
675                 home->zombie_period = mainconfig.proxy_retry_delay * mainconfig.proxy_retry_count;
676                 if (home->zombie_period == 0) home->zombie_period =30;
677                 home->response_window = home->zombie_period - 1;
678
679                 home->ping_check = HOME_PING_CHECK_NONE;
680
681                 home->revive_interval = mainconfig.proxy_dead_time;
682
683                 if (rbtree_finddata(home_servers_byaddr, home)) {
684                         radlog(L_ERR, "%s[%d]: Home server %s has the same IP address as another home server.",
685                                filename, lineno, name);
686                         free(home);
687                         return 0;
688                 }
689
690                 if (!rbtree_insert(home_servers_byname, home)) {
691                         radlog(L_ERR, "%s[%d]: Internal error adding home server %s.",
692                                filename, lineno, name);
693                         free(home);
694                         return 0;
695                 }
696                 
697                 if (!rbtree_insert(home_servers_byaddr, home)) {
698                         rbtree_deletebydata(home_servers_byname, home);
699                         radlog(L_ERR, "%s[%d]: Internal error adding home server %s.",
700                                filename, lineno, name);
701                         free(home);
702                         return 0;
703                 }
704         }
705
706         /*
707          *      We now have a home server, see if we can insert it
708          *      into pre-existing pool.
709          */
710         if (insert_point >= 0) {
711                 rad_assert(pool != NULL);
712                 pool->servers[insert_point] = home;
713                 return 1;
714         }
715
716         rad_assert(pool == NULL);
717         rad_assert(home != NULL);
718
719         /*
720          *      Count the old-style realms of this name.
721          */
722         num_home_servers = 0;
723         for (cs = cf_section_sub_find_name2(mainconfig.config, "realm", name);
724              cs != NULL;
725              cs = cf_section_sub_find_name2(cs, "realm", name)) {
726                 num_home_servers++;
727         }
728
729
730         pool = rad_malloc(sizeof(*pool) + num_home_servers * sizeof(pool->servers[0]));
731         memset(pool, 0, sizeof(*pool) + num_home_servers * sizeof(pool->servers[0]));
732
733         pool->name = name;
734         pool->type = ldflag;
735         pool->server_type = type;
736         pool->num_home_servers = num_home_servers;
737         pool->servers[0] = home;
738
739         if (!rbtree_insert(home_pools_byname, pool)) {
740                 rad_assert("Internal sanity check failed");
741                 return 0;
742         }
743
744         *pool_p = pool;
745
746         return 1;
747 }
748
749 static int old_realm_config(const char *filename, CONF_SECTION *cs, REALM *r)
750 {
751         char *host;
752         const char *secret;
753         home_pool_type_t ldflag;
754
755         secret = cf_section_value_find(cs, "secret");
756
757         host = cf_section_value_find(cs, "ldflag");
758         if (!host ||
759             (strcasecmp(host, "fail_over") == 0)) {
760                 ldflag = HOME_POOL_FAIL_OVER;
761                 DEBUG2("  realm %s: ldflag = fail_over", r->name);
762
763         } else if (strcasecmp(host, "round_robin") == 0) {
764                 ldflag = HOME_POOL_LOAD_BALANCE;
765                 DEBUG2("  realm %s: ldflag = round_robin", r->name);
766
767         } else {
768                 radlog(L_ERR, "%s[%d]: Unknown value \"%s\" for ldflag",
769                        filename, cf_section_lineno(cs), host);
770                 return 0;
771         }
772
773         /*
774          *      Allow old-style if it doesn't exist, or if it exists and
775          *      it's LOCAL.
776          */
777         if (((host = cf_section_value_find(cs, "authhost")) != NULL) &&
778             (strcmp(host, "LOCAL") != 0)) {
779                 if (!secret) {
780                         radlog(L_ERR, "%s[%d]: No shared secret supplied for realm: %s",
781                                filename, cf_section_lineno(cs), r->name);
782                         return 0;
783                 }
784
785                 DEBUG2("  realm %s: authhost = %s",  r->name, host);
786
787                 if (!old_server_add(filename, cf_section_lineno(cs),
788                                     host, secret, ldflag,
789                                     &r->auth_pool, HOME_TYPE_AUTH)) {
790                         return 0;
791                 }
792         }
793
794         if (((host = cf_section_value_find(cs, "accthost")) != 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("  realm %s: accthost = %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_ACCT)) {
807                         return 0;
808                 }
809         }
810
811         if (secret) DEBUG2("  realm %s: secret = %s", r->name, secret);
812
813         return 1;
814         
815 }
816
817
818 static int add_pool_to_realm(const char *filename, int lineno,
819                              const char *name, home_pool_t **dest,
820                              int server_type)
821 {
822         home_pool_t mypool, *pool;
823
824         mypool.name = name;
825         pool = rbtree_finddata(home_pools_byname, &mypool);
826         if (!pool) {
827                 CONF_SECTION *pool_cs;
828
829                 pool_cs = cf_section_sub_find_name2(NULL, "server_pool",
830                                                     name);
831                 if (!pool_cs) {
832                         radlog(L_ERR, "%s[%d]: Failed to find server_pool \"%s\"",
833                                filename, lineno, name);
834                         return 0;
835                 }
836
837                 if (!server_pool_add(filename, pool_cs)) {
838                         return 0;
839                 }
840                 
841                 pool = rbtree_finddata(home_pools_byname, &mypool);
842                 if (!pool) {
843                         rad_assert("Internal sanity check failed");
844                         return 0;
845                 }
846         }
847
848         if (pool->server_type != server_type) {
849                 radlog(L_ERR, "%s[%d]: Incompatible server_pool \"%s\" (mixed auth_pool / acct_pool)",
850                        filename, lineno, name);
851                 return 0;
852         }
853
854         *dest = pool;
855
856         return 1;
857 }
858
859 int realm_add(const char *filename, CONF_SECTION *cs)
860 {
861         const char *name2;
862         char *pool = NULL;
863         REALM *r;
864         CONF_PAIR *cp;
865
866         name2 = cf_section_name1(cs);
867         if (!name2 || (strcasecmp(name2, "realm") != 0)) {
868                 radlog(L_ERR, "%s[%d]: Section is not a realm.",
869                        filename, cf_section_lineno(cs));
870                 return 0;
871         }
872
873         name2 = cf_section_name2(cs);
874         if (!name2) {
875                 radlog(L_ERR, "%s[%d]: Realm section is missing the realm name.",
876                        filename, cf_section_lineno(cs));
877                 return 0;
878         }
879
880         /*
881          *      The realm MAY already exist if it's an old-style realm.
882          *      In that case, merge the old-style realm with this one.
883          */
884         r = realm_find(name2);
885         if (r) {
886                 if (cf_pair_find(cs, "auth_pool") ||
887                     cf_pair_find(cs, "acct_pool")) {
888                         radlog(L_ERR, "%s[%d]: Duplicate realm \"%s\"",
889                                filename, cf_section_lineno(cs), name2);
890                         return 0;
891                 }
892
893                 if (!old_realm_config(filename, cs, r)) {
894                         return 0;
895                 }
896
897                 return 1;
898         }
899
900         r = rad_malloc(sizeof(*r));
901         memset(r, 0, sizeof(*r));
902
903         r->name = name2;
904
905         /*
906          *      Prefer new configuration to old one.
907          */
908         cp = cf_pair_find(cs, "auth_pool");
909         if (cp) pool = cf_pair_value(cp);
910         if (cp && pool) {
911                 if (!add_pool_to_realm(filename, cf_pair_lineno(cp),
912                                        pool, &r->auth_pool, HOME_TYPE_AUTH)) {
913                         free(r);
914                         return 0;
915                 }
916                 DEBUG2(" realm %s: auth_pool = %s", name2, pool);
917         }
918
919         cp = cf_pair_find(cs, "acct_pool");
920         if (cp) pool = cf_pair_value(cp);
921         if (cp && pool) {
922                 if (!add_pool_to_realm(filename, cf_pair_lineno(cp),
923                                        pool, &r->acct_pool, HOME_TYPE_ACCT)) {
924                         free(r);
925                         return 0;
926                 }
927                 DEBUG2(" realm %s: acct_pool = %s", name2, pool);
928         }
929
930         r->striprealm = 1;
931         
932         if ((cf_section_value_find(cs, "nostrip")) != NULL) {
933                 r->striprealm = 0;
934                 DEBUG2(" realm %s: nostrip", name2);
935         }
936
937         /*
938          *      We're a new-style realm.  Complain if we see the old
939          *      directives.
940          */
941         if (r->auth_pool || r->acct_pool) {
942                 if (((cp = cf_pair_find(cs, "authhost")) != NULL) ||
943                     ((cp = cf_pair_find(cs, "accthost")) != NULL) ||
944                     ((cp = cf_pair_find(cs, "secret")) != NULL) ||
945                     ((cp = cf_pair_find(cs, "ldflag")) != NULL)) {
946                         DEBUG2("WARNING: Ignoring old-style configuration entry \"%s\" in realm \"%s\"", cf_pair_attr(cp), r->name);
947                 }
948
949
950                 /*
951                  *      The realm MAY be an old-style realm, as there
952                  *      was no auth_pool or acct_pool.  Double-check
953                  *      it, just to be safe.
954                  */
955         } else if (!old_realm_config(filename, cs, r)) {
956                 free(r);
957                 return 0;
958         }
959
960         if (!rbtree_insert(realms_byname, r)) {
961                 rad_assert("Internal sanity check failed");
962                 free(r);
963                 return 0;
964         }
965
966         return 1;
967 }
968
969
970 /*
971  *      Find a realm in the REALM list.
972  */
973 REALM *realm_find(const char *name)
974 {
975         REALM myrealm;
976
977         if (!name) name = "NULL";
978
979         myrealm.name = name;
980         return rbtree_finddata(realms_byname, &myrealm);
981 }
982
983
984 home_server *home_server_ldb(const char *realmname,
985                              home_pool_t *pool, REQUEST *request)
986 {
987         int             start;
988         int             count;
989
990         start = 0;
991
992         /*
993          *      Determine how to pick choose the home server.
994          */
995         switch (pool->type) {
996                 uint32_t hash;
997
998                 /*
999                  *      Load balancing.  Pick one at random.
1000                  */
1001         case HOME_POOL_LOAD_BALANCE:
1002                 hash = lrad_rand();
1003                 start = hash % pool->num_home_servers;
1004                 break;
1005
1006                 /*
1007                  *      For load-balancing by client IP address, we
1008                  *      pick a home server by hashing the client IP.
1009                  *
1010                  *      This isn't as even a load distribution as
1011                  *      tracking the State attribute, but it's better
1012                  *      than nothing.
1013                  */
1014         case HOME_POOL_CLIENT_BALANCE:
1015                 switch (request->packet->src_ipaddr.af) {
1016                 case AF_INET:
1017                         hash = lrad_hash(&request->packet->src_ipaddr.ipaddr.ip4addr,
1018                                          sizeof(request->packet->src_ipaddr.ipaddr.ip4addr));
1019                         break;
1020                 case AF_INET6:
1021                         hash = lrad_hash(&request->packet->src_ipaddr.ipaddr.ip6addr,
1022                                          sizeof(request->packet->src_ipaddr.ipaddr.ip6addr));
1023                         break;
1024                 default:
1025                         hash = 0;
1026                         break;
1027                 }
1028                 start = hash % pool->num_home_servers;
1029                 break;
1030
1031         default:
1032                 start = 0;
1033                 break;
1034         }
1035
1036         /*
1037          *      Starting with the home server we chose, loop through
1038          *      all home servers.  If the current one is dead, skip
1039          *      it.  If it is too busy, skip it.
1040          *
1041          *      Otherwise, use it.
1042          */
1043         for (count = 0; count < pool->num_home_servers; count++) {
1044                 home_server *home = pool->servers[(start + count) % pool->num_home_servers];
1045
1046                 if (home->state == HOME_STATE_IS_DEAD) {
1047                         continue;
1048                 }
1049
1050                 /*
1051                  *      This home server is too busy.  Choose another one.
1052                  */
1053                 if (home->currently_outstanding >= home->max_outstanding) {
1054                         continue;
1055                 }
1056
1057                 return home;
1058         } /* loop over the home servers */
1059
1060         /*
1061          *      No live match found, and no fallback to the "DEFAULT"
1062          *      realm.  We fix this by blindly marking all servers as
1063          *      "live".  But only do it for ones that don't support
1064          *      "pings", as they will be marked live when they
1065          *      actually are live.
1066          */
1067         if (!mainconfig.proxy_fallback &&
1068             mainconfig.wake_all_if_all_dead) {
1069                 home_server *lb = NULL;
1070
1071                 for (count = 0; count < pool->num_home_servers; count++) {
1072                         home_server *home = pool->servers[count];
1073
1074                         if ((home->state == HOME_STATE_IS_DEAD) &&
1075                             (home->ping_check == HOME_PING_CHECK_NONE)) {
1076                                 home->state = HOME_STATE_ALIVE;
1077                                 if (!lb) lb = home;
1078                         }
1079                 }
1080
1081                 if (lb) return lb;
1082         }
1083
1084         /*
1085          *      Still nothing.  Look up the DEFAULT realm, but only
1086          *      if we weren't looking up the NULL or DEFAULT realms.
1087          */
1088         if (mainconfig.proxy_fallback &&
1089             realmname &&
1090             (strcmp(realmname, "NULL") != 0) &&
1091             (strcmp(realmname, "DEFAULT") != 0)) {
1092                 REALM *rd = realm_find("DEFAULT");
1093
1094                 if (!rd) return NULL;
1095
1096                 pool = NULL;
1097                 if (request->packet->code == PW_AUTHENTICATION_REQUEST) {
1098                         pool = rd->auth_pool;
1099                         
1100                 } else if (request->packet->code == PW_ACCOUNTING_REQUEST) {
1101                         pool = rd->acct_pool;
1102                 }
1103                 if (!pool) return NULL;
1104
1105                 DEBUG2("  Realm %s has no live home servers.  Falling back to the DEFAULT realm.", realmname);
1106                 return home_server_ldb(rd->name, pool, request);
1107         }
1108
1109         /*
1110          *      Still haven't found anything.  Oh well.
1111          */
1112         return NULL;
1113 }
1114
1115
1116 home_server *home_server_find(lrad_ipaddr_t *ipaddr, int port)
1117 {
1118         home_server myhome;
1119
1120         myhome.ipaddr = *ipaddr;
1121         myhome.port = port;
1122
1123         return rbtree_finddata(home_servers_byaddr, &myhome);
1124 }