2 * util.c Various utility functions.
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.
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.
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
20 * Copyright 2000,2006 The FreeRADIUS server project
23 #include <freeradius-devel/ident.h>
26 #include <freeradius-devel/radiusd.h>
27 #include <freeradius-devel/rad_assert.h>
36 * The signal() function in Solaris 2.5.1 sets SA_NODEFER in
37 * sa_flags, which causes grief if signal() is called in the
38 * handler before the cause of the signal has been cleared.
39 * (Infinite recursion).
41 * The same problem appears on HPUX, so we avoid it, if we can.
43 * Using sigaction() to reset the signal handler fixes the problem,
44 * so where available, we prefer that solution.
47 void (*reset_signal(int signo, void (*func)(int)))(int)
50 struct sigaction act, oact;
52 memset(&act, 0, sizeof(act));
53 act.sa_handler = func;
54 sigemptyset(&act.sa_mask);
56 #ifdef SA_INTERRUPT /* SunOS */
57 act.sa_flags |= SA_INTERRUPT;
59 if (sigaction(signo, &act, &oact) < 0)
61 return oact.sa_handler;
65 * re-set by calling the 'signal' function, which
66 * may cause infinite recursion and core dumps due to
69 * However, the system is too dumb to implement sigaction(),
70 * so we don't have a choice.
79 * Per-request data, added by modules...
81 struct request_data_t {
87 void (*free_opaque)(void *);
91 * Add opaque data (with a "free" function) to a REQUEST.
93 * The unique ptr is meant to be a malloc'd module configuration,
94 * and the unique integer allows the caller to have multiple
95 * opaque data associated with a REQUEST.
97 int request_data_add(REQUEST *request,
98 void *unique_ptr, int unique_int,
99 void *opaque, void (*free_opaque)(void *))
101 request_data_t *this, **last, *next;
104 * Some simple sanity checks.
106 if (!request || !opaque) return -1;
109 for (last = &(request->data); *last != NULL; last = &((*last)->next)) {
110 if (((*last)->unique_ptr == unique_ptr) &&
111 ((*last)->unique_int == unique_int)) {
116 if (this->opaque && /* free it, if necessary */
118 this->free_opaque(this->opaque);
119 break; /* replace the existing entry */
123 if (!this) this = rad_malloc(sizeof(*this));
124 memset(this, 0, sizeof(*this));
127 this->unique_ptr = unique_ptr;
128 this->unique_int = unique_int;
129 this->opaque = opaque;
130 this->free_opaque = free_opaque;
139 * Get opaque data from a request.
141 void *request_data_get(REQUEST *request,
142 void *unique_ptr, int unique_int)
144 request_data_t **last;
146 if (!request) return NULL;
148 for (last = &(request->data); *last != NULL; last = &((*last)->next)) {
149 if (((*last)->unique_ptr == unique_ptr) &&
150 ((*last)->unique_int == unique_int)) {
151 request_data_t *this = *last;
152 void *ptr = this->opaque;
155 * Remove the entry from the list, and free it.
159 return ptr; /* don't free it, the caller does that */
163 return NULL; /* wasn't found, too bad... */
168 * Get opaque data from a request without removing it.
170 void *request_data_reference(REQUEST *request,
171 void *unique_ptr, int unique_int)
173 request_data_t **last;
175 for (last = &(request->data); *last != NULL; last = &((*last)->next)) {
176 if (((*last)->unique_ptr == unique_ptr) &&
177 ((*last)->unique_int == unique_int)) {
178 request_data_t *this = *last;
179 void *ptr = this->opaque;
185 return NULL; /* wasn't found, too bad... */
190 * Free a REQUEST struct.
192 void request_free(REQUEST **request_ptr)
196 if (!request_ptr || !*request_ptr) {
200 request = *request_ptr;
202 rad_assert(!request->in_request_hash);
204 rad_assert(!request->in_proxy_hash);
206 rad_assert(!request->ev);
209 rad_free(&request->packet);
213 rad_free(&request->proxy);
217 rad_free(&request->reply);
220 if (request->proxy_reply)
221 rad_free(&request->proxy_reply);
224 if (request->config_items)
225 pairfree(&request->config_items);
227 request->username = NULL;
228 request->password = NULL;
231 request_data_t *this, *next;
233 for (this = request->data; this != NULL; this = next) {
235 if (this->opaque && /* free it, if necessary */
237 this->free_opaque(this->opaque);
240 request->data = NULL;
244 (request->root->refcount > 0)) {
245 request->root->refcount--;
246 request->root = NULL;
251 request->coa->parent = NULL;
252 rad_assert(request->coa->ev == NULL);
253 request_free(&request->coa);
256 if (request->parent && (request->parent->coa == request)) {
257 request->parent->coa = NULL;
262 request->magic = 0x01020304; /* set the request to be nonsense */
264 request->client = NULL;
266 request->home_server = NULL;
268 talloc_free(request);
273 * Check a filename for sanity.
275 * Allow only uppercase/lowercase letters, numbers, and '-_/.'
277 int rad_checkfilename(const char *filename)
279 if (strspn(filename, "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789-_/.") == strlen(filename)) {
287 * Create possibly many directories.
289 * Note that the input directory name is NOT a constant!
290 * This is so that IF an error is returned, the 'directory' ptr
291 * points to the name of the file which caused the error.
293 int rad_mkdir(char *directory, mode_t mode)
300 * If the directory exists, don't do anything.
302 if (stat(directory, &st) == 0) {
307 * Look for the LAST directory name. Try to create that,
308 * failing on any error.
310 p = strrchr(directory, FR_DIR_SEP);
313 rcode = rad_mkdir(directory, mode);
316 * On error, we leave the directory name as the
317 * one which caused the error.
320 if (errno == EEXIST) return 0;
325 * Reset the directory delimiter, and go ask
326 * the system to make the directory.
334 * Having done everything successfully, we do the
335 * system call to actually go create the directory.
337 rcode = mkdir(directory, mode & 0777);
343 * Set things like sticky bits that aren't supported by
347 rcode = chmod(directory, mode);
355 * Module malloc() call, which does stuff if the malloc fails.
357 * This call ALWAYS succeeds!
359 void *rad_malloc(size_t size)
361 void *ptr = malloc(size);
364 radlog(L_ERR, "no memory");
372 void *rad_calloc(size_t size)
374 void *ptr = rad_malloc(size);
375 memset(ptr, 0, size);
379 void rad_const_free(const void *ptr)
384 memcpy(&tmp, &ptr, sizeof(tmp));
390 * Signature for free is dumb, and raises errors when we try
391 * to free const ptrs.
393 void rad_cfree(const void *ptr)
398 memcpy(&tmp, &ptr, sizeof(tmp));
403 * Logs an error message and aborts the program
407 void NEVER_RETURNS rad_assert_fail (const char *file, unsigned int line,
410 radlog(L_ERR, "ASSERT FAILED %s[%u]: %s", file, line, expr);
416 * Create a new REQUEST data structure.
418 REQUEST *request_alloc(void)
422 request = talloc_zero(NULL, REQUEST);
424 request->magic = REQUEST_MAGIC;
427 request->proxy = NULL;
429 request->reply = NULL;
431 request->proxy_reply = NULL;
433 request->config_items = NULL;
434 request->username = NULL;
435 request->password = NULL;
436 request->timestamp = time(NULL);
437 request->options = debug_flag; /* Default to global debug level */
439 request->module = "";
440 request->component = "<core>";
441 if (debug_flag) request->radlog = radlog_request;
448 * Create a new REQUEST, based on an old one.
450 * This function allows modules to inject fake requests
451 * into the server, for tunneled protocols like TTLS & PEAP.
453 REQUEST *request_alloc_fake(REQUEST *request)
457 fake = request_alloc();
459 fake->number = request->number;
460 #ifdef HAVE_PTHREAD_H
461 fake->child_pid = request->child_pid;
463 fake->parent = request;
464 fake->root = request->root;
465 fake->client = request->client;
468 * For new server support.
470 * FIXME: Key instead off of a "virtual server" data structure.
472 * FIXME: Permit different servers for inner && outer sessions?
474 fake->server = request->server;
476 fake->packet = rad_alloc(request, 1);
482 fake->reply = rad_alloc(request, 0);
488 fake->master_state = REQUEST_ACTIVE;
489 fake->child_state = REQUEST_RUNNING;
492 * Fill in the fake request.
494 fake->packet->sockfd = -1;
495 fake->packet->src_ipaddr = request->packet->src_ipaddr;
496 fake->packet->src_port = request->packet->src_port;
497 fake->packet->dst_ipaddr = request->packet->dst_ipaddr;
498 fake->packet->dst_port = 0;
501 * This isn't STRICTLY required, as the fake request MUST NEVER
502 * be put into the request list. However, it's still reasonable
505 fake->packet->id = fake->number & 0xff;
506 fake->packet->code = request->packet->code;
507 fake->timestamp = request->timestamp;
510 * Required for new identity support
512 fake->listener = request->listener;
515 * Fill in the fake reply, based on the fake request.
517 fake->reply->sockfd = fake->packet->sockfd;
518 fake->reply->src_ipaddr = fake->packet->dst_ipaddr;
519 fake->reply->src_port = fake->packet->dst_port;
520 fake->reply->dst_ipaddr = fake->packet->src_ipaddr;
521 fake->reply->dst_port = fake->packet->src_port;
522 fake->reply->id = fake->packet->id;
523 fake->reply->code = 0; /* UNKNOWN code */
526 * Copy debug information.
528 fake->options = request->options;
529 fake->radlog = request->radlog;
535 REQUEST *request_alloc_coa(REQUEST *request)
537 if (!request || request->coa) return NULL;
540 * Originate CoA requests only when necessary.
542 if ((request->packet->code != PW_AUTHENTICATION_REQUEST) &&
543 (request->packet->code != PW_ACCOUNTING_REQUEST)) return NULL;
545 request->coa = request_alloc_fake(request);
546 if (!request->coa) return NULL;
548 request->coa->packet->code = 0; /* unknown, as of yet */
549 request->coa->child_state = REQUEST_RUNNING;
550 request->coa->proxy = rad_alloc(request->coa, 0);
551 if (!request->coa->proxy) {
552 request_free(&request->coa);
561 * Copy a quoted string.
563 int rad_copy_string(char *to, const char *from)
575 } while (*from && (*from != quote));
577 if (*from != quote) return -1; /* not properly quoted */
590 int rad_copy_variable(char *to, const char *from)
602 sublen = rad_copy_string(to, from);
603 if (sublen < 0) return sublen;
609 case '}': /* end of variable expansion */
613 return length; /* proper end of variable */
621 case '%': /* start of variable expansion */
622 if (from[1] == '{') {
626 sublen = rad_copy_variable(to, from);
627 if (sublen < 0) return sublen;
632 } /* else FIXME: catch %%{ ?*/
640 } /* loop over the input string */
643 * We ended the string before a trailing '}'
653 int rad_pps(int *past, int *present, time_t *then, struct timeval *now)
657 if (*then != now->tv_sec) {
664 * Bootstrap PPS by looking at a percentage of
665 * the previous PPS. This lets us take a moving
666 * count, without doing a moving average. If
667 * we're a fraction "f" (0..1) into the current
668 * second, we can get a good guess for PPS by
671 * PPS = pps_now + pps_old * (1 - f)
673 * It's an instantaneous measurement, rather than
674 * a moving average. This will hopefully let it
675 * respond better to sudden spikes.
677 * Doing the calculations by thousands allows us
678 * to not overflow 2^32, AND to not underflow
679 * when we divide by USEC.
681 pps = USEC - now->tv_usec; /* useconds left in previous second */
682 pps /= 1000; /* scale to milliseconds */
683 pps *= *past; /* multiply by past count to get fraction */
684 pps /= 1000; /* scale to usec again */
685 pps += *present; /* add in current count */