Include sys/types.h, for FreeBSD.
[freeradius.git] / src / modules / rlm_realm / rlm_realm.c
1 /*
2  * rlm_realm.c  
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., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
19  *
20  * Copyright 2000  The FreeRADIUS server project
21  * FIXME add copyrights
22  */
23
24 #include "autoconf.h"
25
26 #include <stdio.h>
27 #include <stdlib.h>
28 #include <string.h>
29
30 #ifdef HAVE_SYS_TYPES_H
31 #include <sys/types.h>
32 #endif
33
34 #ifdef HAVE_NETINET_IN_H
35 #include <netinet/in.h>
36 #endif
37
38 #include "libradius.h"
39 #include "radiusd.h"
40 #include "modules.h"
41
42 static const char rcsid[] = "$Id$";
43
44 #define  REALM_FORMAT_PREFIX   0
45 #define  REALM_FORMAT_SUFFIX   1
46
47 typedef struct realm_config_t {
48         int        format;
49         char       *formatstring;
50         char       *delim;
51 } realm_config_t;
52
53 static CONF_PARSER module_config[] = {
54   { "format", PW_TYPE_STRING_PTR,
55     offsetof(realm_config_t,formatstring), NULL, "suffix" },
56   { "delimiter", PW_TYPE_STRING_PTR,
57     offsetof(realm_config_t,delim), NULL, "@" },
58   { NULL, -1, 0, NULL, NULL }    /* end the list */
59 };
60
61 /*
62  *      Internal function to cut down on duplicated code.
63  *
64  *      Returns NULL on don't proxy, realm otherwise.
65  */
66 static REALM *check_for_realm(void *instance, REQUEST *request)
67 {
68         int is_local;
69         char namebuf[MAX_STRING_LEN];
70         char *username;
71         char *realmname = NULL;
72         char *ptr;
73         VALUE_PAIR *vp;
74         REALM *realm;
75
76         struct realm_config_t *inst = instance;
77
78         /*
79          *      If the request has a proxy entry, then it's a proxy
80          *      reply, and we're walking through the module list again.
81          *
82          *      In that case, don't bother trying to proxy the request
83          *      again.
84          *
85          *      Also, if there's no User-Name attribute, we can't
86          *      proxy it, either.
87          */
88         if ((request->proxy != NULL) ||
89             (request->username == NULL)) {
90                 DEBUG2("    rlm_realm: Proxy reply, or no user name.  Ignoring.");
91                 return NULL;
92         }
93
94         /*
95          *      Check for 'Realm' attribute.  If it exists, then we've proxied
96          *      it already ( via another rlm_realm instance ) and should return.
97          */
98
99         if ( (vp = pairfind(request->packet->vps, PW_REALM)) != NULL ) {
100                 DEBUG2("    rlm_realm: Request already proxied.  Ignoring.");
101                 return NULL;
102         }
103
104         /*
105          *      We will be modifing this later, so we want our own copy
106          *      of it.
107          */
108         strNcpy(namebuf, (char *)request->username->strvalue, sizeof(namebuf));
109         username = namebuf;
110
111         switch(inst->format)
112         {
113
114         case REALM_FORMAT_SUFFIX:
115           
116           /* DEBUG2("  rlm_realm: Checking for suffix after \"%c\"", inst->delim[0]); */
117                 realmname = strrchr(username, inst->delim[0]);          
118                 if (realmname) {
119                         *realmname = '\0';
120                         realmname++;
121                 }
122                 break;
123                 
124         case REALM_FORMAT_PREFIX:
125                 
126                 /* DEBUG2("  rlm_realm: Checking for prefix before \"%c\"", inst->delim[0]); */
127                 
128                 ptr = strchr(username, inst->delim[0]);
129                 if (ptr) {
130                         *ptr = '\0';
131                      ptr++;
132                      realmname = username;
133                      username = ptr;
134                 }
135                 break;
136                
137         default:
138                 realmname = NULL;
139                 break;
140         }
141
142         DEBUG2("    rlm_realm: Looking up realm %s for User-Name = \"%s\"",
143                (realmname == NULL) ? "NULL" : realmname,
144                request->username->strvalue);
145
146         /*
147          *      Allow NULL realms.
148          */
149         realm = realm_find(realmname);
150         if (!realm) {
151                 DEBUG2("    rlm_realm: No such realm %s",
152                        (realmname == NULL) ? "NULL" : realmname);
153                 return NULL;
154         }
155         DEBUG2("    rlm_realm: Found realm %s", realm->realm);
156         
157         /*
158          *      If we've been told to strip the realm off, then do so.
159          */
160         if (realm->striprealm) {
161                 /*
162                  *      Create the Stripped-User-Name attribute, if it
163                  *      doesn't exist.
164                  *
165                  */
166                 if (request->username->attribute != PW_STRIPPED_USER_NAME) {
167                         vp = paircreate(PW_STRIPPED_USER_NAME, PW_TYPE_STRING);
168                         if (!vp) {
169                                 radlog(L_ERR|L_CONS, "no memory");
170                                 exit(1);
171                         }
172                         pairadd(&request->packet->vps, vp);
173                         DEBUG2("    rlm_realm: Adding Stripped-User-Name = \"%s\"", username);
174                 } else {
175                         vp = request->username;
176                         DEBUG2("    rlm_realm: Setting Stripped-User-Name = \"%s\"", username);
177                 }
178
179                 strcpy(vp->strvalue, username);
180                 vp->length = strlen((char *)vp->strvalue);
181                 request->username = vp;
182         }
183
184         DEBUG2("  rlm_realm: Proxying request from user %s to realm %s",
185                username, realm->realm);
186
187         /*
188          *      Figure out what to do with the request.
189          */
190         is_local = FALSE;
191         switch (request->packet->code) {
192         default:
193                 DEBUG2("rlm_realm: Unknown packet code %d\n",
194                        request->packet->code);
195                 return NULL;            /* don't do anything */
196                 
197                 /*
198                  *      Perhaps accounting proxying was turned off.
199                  */
200         case PW_ACCOUNTING_REQUEST:
201                 if (realm->acct_ipaddr == htonl(INADDR_NONE)) {
202                         DEBUG2("rlm_realm:  Accounting realm is LOCAL.");
203                         is_local = TRUE;
204                 }
205
206                 if (realm->acct_port == 0) {
207                         DEBUG2("rlm_realm:  acct_port is not set.  proxy cancelled");
208                         return NULL;
209                 }
210                 break;
211
212                 /*
213                  *      Perhaps authentication proxying was turned off.
214                  */
215         case PW_AUTHENTICATION_REQUEST:
216                 if (realm->ipaddr == htonl(INADDR_NONE)) {
217                         DEBUG2("rlm_realm:  Authentication realm is LOCAL.");
218                         is_local = TRUE;
219                 }
220
221                 if (realm->auth_port == 0) {
222                         DEBUG2("rlm_realm:  auth_port is not set.  proxy cancelled");
223                         return NULL;
224                 }
225                 break;
226         }
227
228         /*
229          *      Add the realm name to the request.
230          */
231         pairadd(&request->packet->vps, pairmake("Realm", realm->realm,
232                                                 T_OP_EQ));
233         DEBUG2("    rlm_realm: Adding Realm = \"%s\"", realm->realm);
234
235         /*
236          *      Local realm, don't proxy it.
237          */
238         if (is_local) {
239                 return NULL;
240         }
241
242         return realm;
243 }
244
245 /*
246  *      Add a "Proxy-To-Realm" attribute to the request.
247  */
248 static void add_proxy_to_realm(VALUE_PAIR **vps, REALM *realm)
249 {
250         VALUE_PAIR *vp;
251
252         /*
253          *      Tell the server to proxy this request to another
254          *      realm.
255          */
256         vp = pairmake("Proxy-To-Realm", realm->realm, T_OP_EQ);
257         if (!vp) {
258                 radlog(L_ERR|L_CONS, "no memory");
259                 exit(1);
260         }
261         
262         /*
263          *  Add it, even if it's already present.
264          */
265         pairadd(vps, vp);
266 }
267
268 /*
269  *  Perform the realm module instantiation.  Configuration info is
270  *  stored in *instance for later use.
271  */
272
273 static int realm_instantiate(CONF_SECTION *conf, void **instance)
274 {
275         struct realm_config_t *inst;
276
277         /* setup a storage area for instance data */
278         inst = rad_malloc(sizeof(struct realm_config_t));
279
280         if(cf_section_parse(conf, inst, module_config) < 0) {
281                free(inst);
282                return -1;
283         }
284
285         if(strcasecmp(inst->formatstring, "suffix") == 0) {
286              inst->format = REALM_FORMAT_SUFFIX;
287         } else if(strcasecmp(inst->formatstring, "prefix") == 0) {
288              inst->format = REALM_FORMAT_PREFIX;
289         } else {
290              radlog(L_ERR, "Bad value \"%s\" for realm format value", inst->formatstring);
291              free(inst);
292              return -1;
293         }
294         free(inst->formatstring);
295         if(strlen(inst->delim) != 1) {
296              radlog(L_ERR, "Bad value \"%s\" for realm delimiter value", inst->delim);
297              free(inst);
298              return -1;
299         }
300
301         *instance = inst;
302         return 0;
303
304 }
305
306
307
308  
309
310 /*
311  *  Examine a request for a username with an realm, and if it
312  *  corresponds to something in the realms file, set that realm as
313  *  Proxy-To.
314  *
315  *  This should very nearly duplicate the old proxy_send() code
316  */
317 static int realm_authorize(void *instance, REQUEST *request)
318 {
319         REALM *realm;
320
321         /*
322          *      Check if we've got to proxy the request.
323          *      If not, return without adding a Proxy-To-Realm
324          *      attribute.
325          */
326         realm = check_for_realm(instance, request);
327         if (!realm) {
328                 return RLM_MODULE_NOOP;
329         }
330
331         /*
332          *      Maybe add a Proxy-To-Realm attribute to the request.
333          */
334         DEBUG2("rlm_realm:  Preparing to proxy authentication request to realm %s\n",
335                realm->realm);
336         add_proxy_to_realm(&request->config_items, realm);
337
338         return RLM_MODULE_UPDATED; /* try the next module */
339 }
340
341 /*
342  * This does the exact same thing as the realm_authorize, it's just called
343  * differently.
344  */
345 static int realm_preacct(void *instance, REQUEST *request)
346 {
347         const char *name = (char *)request->username->strvalue;
348         REALM *realm;
349
350         if (!name)
351           return RLM_MODULE_OK;
352         
353
354         /*
355          *      Check if we've got to proxy the request.
356          *      If not, return without adding a Proxy-To-Realm
357          *      attribute.
358          */
359         realm = check_for_realm(instance, request);
360         if (!realm) {
361                 return RLM_MODULE_NOOP;
362         }
363
364
365         /*
366          *      Maybe add a Proxy-To-Realm attribute to the request.
367          */
368         DEBUG2("rlm_realm:  Preparing to proxy accounting request to realm %s\n",
369                realm->realm);
370         add_proxy_to_realm(&request->config_items, realm);
371
372         return RLM_MODULE_OK; /* try the next module */
373 }
374
375 static int realm_detach(void *instance)
376 {
377         struct realm_config_t *inst = instance;
378         free(inst->delim);
379         free(instance);
380         return 0;
381 }
382
383 /* globally exported name */
384 module_t rlm_realm = {
385   "realm",
386   0,                            /* type: reserved */
387   NULL,                         /* initialization */
388   realm_instantiate,            /* instantiation */
389   {
390           NULL,                 /* authentication */
391           realm_authorize,      /* authorization */
392           realm_preacct,        /* preaccounting */
393           NULL,                 /* accounting */
394           NULL                  /* checksimul */
395   },
396   realm_detach,                 /* detach */
397   NULL,                         /* destroy */
398 };