Annotation of embedaddon/ipsec-tools/src/racoon/throttle.c, revision 1.1.1.1

1.1       misho       1: /*     $NetBSD: throttle.c,v 1.7 2011/03/14 17:18:13 tteras Exp $      */
                      2: 
                      3: /* Id: throttle.c,v 1.5 2006/04/05 20:54:50 manubsd Exp */
                      4: 
                      5: /*
                      6:  * Copyright (C) 2004 Emmanuel Dreyfus
                      7:  * All rights reserved.
                      8:  * 
                      9:  * Redistribution and use in source and binary forms, with or without
                     10:  * modification, are permitted provided that the following conditions
                     11:  * are met:
                     12:  * 1. Redistributions of source code must retain the above copyright
                     13:  *    notice, this list of conditions and the following disclaimer.
                     14:  * 2. Redistributions in binary form must reproduce the above copyright
                     15:  *    notice, this list of conditions and the following disclaimer in the
                     16:  *    documentation and/or other materials provided with the distribution.
                     17:  * 3. Neither the name of the project nor the names of its contributors
                     18:  *    may be used to endorse or promote products derived from this software
                     19:  *    without specific prior written permission.
                     20:  * 
                     21:  * THIS SOFTWARE IS PROVIDED BY THE PROJECT AND CONTRIBUTORS ``AS IS'' AND
                     22:  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
                     23:  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
                     24:  * ARE DISCLAIMED.  IN NO EVENT SHALL THE PROJECT OR CONTRIBUTORS BE LIABLE
                     25:  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
                     26:  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
                     27:  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
                     28:  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
                     29:  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
                     30:  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
                     31:  * SUCH DAMAGE.
                     32:  */
                     33: 
                     34: #include "config.h"
                     35: 
                     36: #include <stdlib.h>
                     37: #include <string.h>
                     38: #include <sys/param.h>
                     39: #include <sys/queue.h>
                     40: #include <netinet/in.h>
                     41: #include <resolv.h>
                     42: 
                     43: #include "vmbuf.h"
                     44: #include "misc.h"
                     45: #include "plog.h"
                     46: #include "throttle.h"
                     47: #include "sockmisc.h"
                     48: #include "isakmp_var.h"
                     49: #include "isakmp.h"
                     50: #include "isakmp_xauth.h"
                     51: #include "isakmp_cfg.h"
                     52: #include "gcmalloc.h"
                     53: 
                     54: static struct throttle_list throttle_list =
                     55:        TAILQ_HEAD_INITIALIZER(throttle_list);
                     56: 
                     57: struct throttle_entry *
                     58: throttle_add(addr)
                     59:        struct sockaddr *addr;
                     60: {
                     61:        struct throttle_entry *te;
                     62:        struct timeval now, penalty;
                     63:        size_t len;
                     64: 
                     65:        len = sizeof(*te) 
                     66:            - sizeof(struct sockaddr_storage) 
                     67:            + sysdep_sa_len(addr);
                     68: 
                     69:        if ((te = racoon_malloc(len)) == NULL)
                     70:                return NULL;
                     71: 
                     72:        sched_get_monotonic_time(&now);
                     73:        penalty.tv_sec = isakmp_cfg_config.auth_throttle;
                     74:        penalty.tv_usec = 0;
                     75:        timeradd(&now, &penalty, &te->penalty_ends);
                     76: 
                     77:        memcpy(&te->host, addr, sysdep_sa_len(addr));
                     78:        TAILQ_INSERT_HEAD(&throttle_list, te, next);
                     79: 
                     80:        return te;
                     81: }
                     82: 
                     83: int
                     84: throttle_host(addr, authfail) 
                     85:        struct sockaddr *addr;
                     86:        int authfail;
                     87: {
                     88:        struct throttle_entry *te;
                     89:        struct timeval now, res;
                     90:        int found = 0;
                     91: 
                     92:        if (isakmp_cfg_config.auth_throttle == 0)
                     93:                return 0;
                     94: 
                     95:        sched_get_monotonic_time(&now);
                     96: restart:
                     97:        RACOON_TAILQ_FOREACH_REVERSE(te, &throttle_list, throttle_list, next) {
                     98:                /*
                     99:                 * Remove outdated entries
                    100:                 */
                    101:                if (timercmp(&te->penalty_ends, &now, <)) {
                    102:                        TAILQ_REMOVE(&throttle_list, te, next);
                    103:                        racoon_free(te);
                    104:                        goto restart;
                    105:                }
                    106: 
                    107:                if (cmpsaddr(addr, (struct sockaddr *) &te->host) <= CMPSADDR_WOP_MATCH) {
                    108:                        found = 1;
                    109:                        break;
                    110:                }
                    111:        }
                    112: 
                    113:        /* 
                    114:         * No match, if auth failed, allocate a new throttle entry
                    115:         * give no penalty even on error: this is the first time
                    116:         * and we are indulgent.
                    117:         */
                    118:        if (!found) {
                    119:                if (authfail) {
                    120:                        if ((te = throttle_add(addr)) == NULL) {
                    121:                                plog(LLV_ERROR, LOCATION, NULL, 
                    122:                                    "Throttle insertion failed\n");
                    123:                                return isakmp_cfg_config.auth_throttle;
                    124:                        }
                    125:                }
                    126:                return 0;
                    127:        } else {
                    128:                /*
                    129:                 * We had a match and auth failed, increase penalty.
                    130:                 */
                    131:                if (authfail) {
                    132:                        struct timeval remaining, penalty;
                    133: 
                    134:                        timersub(&te->penalty_ends, &now, &remaining);
                    135:                        penalty.tv_sec = isakmp_cfg_config.auth_throttle;
                    136:                        penalty.tv_usec = 0;
                    137:                        timeradd(&penalty, &remaining, &res);
                    138:                        if (res.tv_sec >= THROTTLE_PENALTY_MAX) {
                    139:                                res.tv_sec = THROTTLE_PENALTY_MAX;
                    140:                                res.tv_usec = 0;
                    141:                        }
                    142:                        timeradd(&now, &res, &te->penalty_ends);
                    143:                }
                    144:        }
                    145: 
                    146:        timersub(&te->penalty_ends, &now, &res);
                    147:        return res.tv_sec;
                    148: }
                    149: 

FreeBSD-CVSweb <freebsd-cvsweb@FreeBSD.org>