Annotation of embedaddon/ipsec-tools/src/racoon/throttle.c, revision 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>