implemented DoS protection with cookies and IP filter

This commit is contained in:
Martin Willi
2007-03-29 11:26:55 +00:00
parent 361e6df8c1
commit 891e69944f
2 changed files with 258 additions and 112 deletions
+239 -98
View File
@@ -33,32 +33,18 @@
#include <queues/jobs/job.h>
#include <queues/jobs/process_message_job.h>
typedef struct block_t block_t;
/**
* entry for a blocked IP
*/
struct block_t {
/**
* IP address to block
*/
host_t *ip;
/**
* lifetime for this block
*/
u_int32_t timeout;
};
/**
* destroy a block_t
*/
static void block_destroy(block_t *block)
{
block->ip->destroy(block->ip);
free(block);
}
/** length of the full cookie, including time (u_int32_t + SHA1()) */
#define COOKIE_LENGTH 24
/** lifetime of a cookie, in seconds */
#define COOKIE_LIFETIME 10
/** how many times to reuse the secret */
#define COOKIE_REUSE 10000
/** require cookies after half open IKE_SAs */
#define COOKIE_TRESHOLD 10
/** how many half open IKE_SAs per peer before blocking */
#define BLOCK_TRESHOLD 5
/** length of the secret to use for cookie calculation */
#define SECRET_LENGTH 16
typedef struct private_receiver_t private_receiver_t;
@@ -76,79 +62,189 @@ struct private_receiver_t {
*/
pthread_t assigned_thread;
/**
* List of blocked IPs
*/
linked_list_t *blocks;
/**
* mutex to exclusively access block list
*/
pthread_mutex_t mutex;
/**
* current secret to use for cookie calculation
*/
char secret[SECRET_LENGTH];
/**
* previous secret used to verify older cookies
*/
char secret_old[SECRET_LENGTH];
/**
* how many times we have used "secret" so far
*/
u_int32_t secret_used;
/**
* time we did the cookie switch
*/
u_int32_t secret_switch;
/**
* time offset to use, hides our system time
*/
u_int32_t secret_offset;
/**
* the randomizer to use for secret generation
*/
randomizer_t *randomizer;
/**
* hasher to use for cookie calculation
*/
hasher_t *hasher;
};
/**
* Implementation of receiver_t.block
* send a notify back to the sender
*/
static void block(private_receiver_t *this, host_t *ip, u_int32_t seconds)
static void send_notify(message_t *request, notify_type_t type, chunk_t data)
{
block_t *blocked = malloc_thing(block_t);
blocked->ip = ip->clone(ip);
blocked->timeout = time(NULL) + seconds;
DBG1(DBG_NET, "blocking %H for %ds", ip, seconds);
pthread_mutex_lock(&this->mutex);
this->blocks->insert_last(this->blocks, blocked);
pthread_mutex_unlock(&this->mutex);
if (request->get_request(request) &&
request->get_exchange_type(request) == IKE_SA_INIT)
{
message_t *response;
host_t *src, *dst;
packet_t *packet;
ike_sa_id_t *ike_sa_id;
response = message_create();
dst = request->get_source(request);
src = request->get_destination(request);
response->set_source(response, src->clone(src));
response->set_destination(response, dst->clone(dst));
response->set_exchange_type(response, request->get_exchange_type(request));
response->set_request(response, FALSE);
response->set_message_id(response, 0);
ike_sa_id = request->get_ike_sa_id(request);
ike_sa_id->switch_initiator(ike_sa_id);
response->set_ike_sa_id(response, ike_sa_id);
response->add_notify(response, FALSE, type, data);
if (response->generate(response, NULL, NULL, &packet) == SUCCESS)
{
charon->sender->send(charon->sender, packet);
response->destroy(response);
}
}
}
/**
* check if an IP is blocked
* build a cookie
*/
static bool is_blocked(private_receiver_t *this, host_t *ip)
static chunk_t cookie_build(private_receiver_t *this, message_t *message,
u_int32_t t, chunk_t secret)
{
bool found = FALSE;
u_int64_t spi = message->get_initiator_spi(message);
host_t *ip = message->get_source(message);
chunk_t input, hash = chunk_alloca(this->hasher->get_hash_size(this->hasher));
if (this->blocks->get_count(this->blocks))
/* COOKIE = t | sha1( IPi | SPIi | t | secret ) */
input = chunk_cata("cccc", ip->get_address(ip), chunk_from_thing(spi),
chunk_from_thing(t), secret);
this->hasher->get_hash(this->hasher, input, hash.ptr);
return chunk_cat("cc", chunk_from_thing(t), hash);
}
/**
* verify a received cookie
*/
static bool cookie_verify(private_receiver_t *this, message_t *message,
chunk_t cookie)
{
u_int32_t t, now;
chunk_t reference;
chunk_t secret;
now = time(NULL);
t = *(u_int32_t*)cookie.ptr;
if (cookie.len != COOKIE_LENGTH ||
t < now - this->secret_offset - COOKIE_LIFETIME)
{
iterator_t *iterator;
block_t *blocked;
u_int32_t now = time(NULL);
pthread_mutex_lock(&this->mutex);
iterator = this->blocks->create_iterator(this->blocks, TRUE);
while (iterator->iterate(iterator, (void**)&blocked))
{
if (now > blocked->timeout)
{
/* blocking expired, remove */
iterator->remove(iterator);
block_destroy(blocked);
continue;
}
if (!ip->ip_equals(ip, blocked->ip))
{
/* no match, get next */
continue;
}
/* blocked */
DBG2(DBG_NET, "received packet source address %H blocked", ip);
found = TRUE;
break;
}
iterator->destroy(iterator);
pthread_mutex_unlock(&this->mutex);
DBG2(DBG_NET, "received cookie lifetime expired, rejecting");
return FALSE;
}
return found;
/* check if cookie is derived from old_secret */
if (t + this->secret_offset > this->secret_switch)
{
secret = chunk_from_thing(this->secret);
}
else
{
secret = chunk_from_thing(this->secret_old);
}
/* compare own calculation against received */
reference = cookie_build(this, message, t, secret);
if (chunk_equals(reference, cookie))
{
chunk_free(&reference);
return TRUE;
}
chunk_free(&reference);
return FALSE;
}
/**
* check if cookies are required, and if so, a valid cookie is included
*/
static bool cookie_required(private_receiver_t *this, message_t *message)
{
bool failed = FALSE;
if (charon->ike_sa_manager->get_half_open_count(charon->ike_sa_manager,
NULL) >= COOKIE_TRESHOLD)
{
/* check for a cookie. We don't use our parser here and do it
* quick and dirty for performance reasons.
* we assume to cookie is the first payload (which is a MUST), and
* the cookies SPI length is zero. */
packet_t *packet = message->get_packet(message);
chunk_t data = packet->get_data(packet);
if (data.len <
IKE_HEADER_LENGTH + NOTIFY_PAYLOAD_HEADER_LENGTH + COOKIE_LENGTH ||
*(data.ptr + 16) != NOTIFY ||
*(u_int16_t*)(data.ptr + IKE_HEADER_LENGTH + 6) != htons(COOKIE))
{
/* no cookie found */
failed = TRUE;
}
else
{
data.ptr += IKE_HEADER_LENGTH + NOTIFY_PAYLOAD_HEADER_LENGTH;
data.len = COOKIE_LENGTH;
if (!cookie_verify(this, message, data))
{
DBG2(DBG_NET, "found cookie, but content invalid");
failed = TRUE;
}
}
packet->destroy(packet);
}
return failed;
}
/**
* check if peer has to many half open IKE_SAs
*/
static bool peer_to_aggressive(private_receiver_t *this, message_t *message)
{
if (charon->ike_sa_manager->get_half_open_count(charon->ike_sa_manager,
message->get_source(message)) >= BLOCK_TRESHOLD)
{
return TRUE;
}
return FALSE;
}
/**
* Implementation of receiver_t.receive_packets.
*/
static void receive_packets(private_receiver_t * this)
static void receive_packets(private_receiver_t *this)
{
packet_t *packet;
message_t *message;
@@ -160,18 +256,14 @@ static void receive_packets(private_receiver_t * this)
while (TRUE)
{
/* read in a packet */
if (charon->socket->receive(charon->socket, &packet) != SUCCESS)
{
DBG1(DBG_NET, "receiving from socket failed!");
continue;
}
if (is_blocked(this, packet->get_source(packet)))
{
packet->destroy(packet);
continue;
}
/* parse message header */
message = message_create_from_packet(packet);
if (message->parse_header(message) != SUCCESS)
{
@@ -181,22 +273,63 @@ static void receive_packets(private_receiver_t * this)
continue;
}
/* check IKE major version */
if (message->get_major_version(message) != IKE_MAJOR_VERSION)
{
DBG1(DBG_NET, "received unsupported IKE version %d.%d from %H, "
"ignored", message->get_major_version(message),
"sending INVALID_MAJOR_VERSION", message->get_major_version(message),
message->get_minor_version(message), packet->get_source(packet));
send_notify(message, INVALID_MAJOR_VERSION, chunk_empty);
message->destroy(message);
continue;
}
if (message->get_request(message) &&
message->get_exchange_type(message) == IKE_SA_INIT)
{
/* check for cookies */
if (cookie_required(this, message))
{
u_int32_t now = time(NULL);
chunk_t cookie = cookie_build(this, message, now - this->secret_offset,
chunk_from_thing(this->secret));
DBG2(DBG_NET, "received packet from: %#H to %#H",
message->get_source(message),
message->get_destination(message));
DBG2(DBG_NET, "sending COOKIE notify to %H",
message->get_source(message));
send_notify(message, COOKIE, cookie);
chunk_free(&cookie);
if (++this->secret_used > COOKIE_REUSE)
{
/* create new cookie */
DBG1(DBG_NET, "generating new cookie secret after %d uses",
this->secret_used);
memcpy(this->secret_old, this->secret, SECRET_LENGTH);
this->randomizer->get_pseudo_random_bytes(this->randomizer,
SECRET_LENGTH, this->secret);
this->secret_switch = now;
this->secret_used = 0;
}
message->destroy(message);
continue;
}
/* check if peer has not too many IKE_SAs half open */
if (peer_to_aggressive(this, message))
{
DBG1(DBG_NET, "ignoring IKE_SA setup from %H, "
"peer to aggressive", message->get_source(message));
message->destroy(message);
continue;
}
}
job = (job_t *)process_message_job_create(message);
charon->job_queue->add(charon->job_queue, job);
}
}
/**
* Implementation of receiver_t.destroy.
*/
@@ -204,7 +337,8 @@ static void destroy(private_receiver_t *this)
{
pthread_cancel(this->assigned_thread);
pthread_join(this->assigned_thread, NULL);
this->blocks->destroy_function(this->blocks, (void*)block_destroy);
this->randomizer->destroy(this->randomizer);
this->hasher->destroy(this->hasher);
free(this);
}
@@ -214,18 +348,25 @@ static void destroy(private_receiver_t *this)
receiver_t *receiver_create()
{
private_receiver_t *this = malloc_thing(private_receiver_t);
u_int32_t now = time(NULL);
this->public.block = (void(*)(receiver_t*,host_t*,u_int32_t)) block;
this->public.destroy = (void(*)(receiver_t*)) destroy;
if (pthread_create(&(this->assigned_thread), NULL, (void*(*)(void*))receive_packets, this) != 0)
this->randomizer = randomizer_create();
this->hasher = hasher_create(HASH_SHA1);
this->secret_switch = now;
this->secret_offset = random() % now;
this->secret_used = 0;
this->randomizer->get_pseudo_random_bytes(this->randomizer, SECRET_LENGTH,
this->secret);
memcpy(this->secret_old, this->secret, SECRET_LENGTH);
if (pthread_create(&this->assigned_thread, NULL,
(void*)receive_packets, this) != 0)
{
free(this);
charon->kill(charon, "unable to create receiver thread");
}
pthread_mutex_init(&this->mutex, NULL);
this->blocks = linked_list_create();
return &(this->public);
return &this->public;
}
+19 -14
View File
@@ -6,7 +6,7 @@
*/
/*
* Copyright (C) 2005-2006 Martin Willi
* Copyright (C) 2005-2007 Martin Willi
* Copyright (C) 2005 Jan Hutter
* Hochschule fuer Technik Rapperswil
*
@@ -33,7 +33,23 @@ typedef struct receiver_t receiver_t;
* @brief Receives packets from the socket and adds them to the job queue.
*
* The receiver starts a thread, wich reads on the blocking socket. A received
* packet is preparsed a process_message_job is queued in the job queue.
* packet is preparsed and a process_message_job is queued in the job queue.
*
* To endure DoS attacks, cookies are enabled when to many IKE_SAs are half
* open. The calculation of cookies is slightly different from the proposed
* method in RFC4306. We do not include a nonce, because we think the advantage
* we gain does not justify the overhead to parse the whole message.
* Instead of VersionIdOfSecret, we include a timestamp. This allows us to
* find out wich key was used for cookie creation. Further, we can set a
* lifetime for the cookie, which allows us to reuse the secret for a longer
* time.
* COOKIE = time | sha1( IPi | SPIi | time | secret )
*
* The secret is changed after a certain amount of cookies sent. The old
* secret is stored to allow a clean migration between secret changes.
*
* Further, the number of half-initiated IKE_SAs is limited per peer. This
* mades it impossible for a peer to flood the server with its real IP address.
*
* @b Constructors:
* - receiver_create()
@@ -41,15 +57,6 @@ typedef struct receiver_t receiver_t;
* @ingroup threads
*/
struct receiver_t {
/**
* @brief Block packets coming from a specific address.
*
* @param receiver receiver object
* @param ip IP address to block
* @param seconds for how long to block ip
*/
void (*block) (receiver_t *receiver, host_t *ip, u_int32_t seconds);
/**
* @brief Destroys a receiver_t object.
@@ -65,9 +72,7 @@ struct receiver_t {
* The receiver thread will start working, get data
* from the socket and add those packets to the job queue.
*
* @return
* - receiver_t object
* - NULL of thread could not be started
* @return receiver_t object
*
* @ingroup threads
*/