From 6fbf66fad3367b24fd6743bcd50254902fd9c8d5 Mon Sep 17 00:00:00 2001 From: james Date: Mon, 26 Sep 2005 05:28:27 +0000 Subject: This is the start of the BETA21 branch. It includes the --topology feature, and TAP-Win32 driver changes to allow non-admin access. git-svn-id: http://svn.openvpn.net/projects/openvpn/branches/BETA21/openvpn@580 e7ae566f-a301-0410-adde-c780ea21d3b5 --- mbuf.c | 187 +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 187 insertions(+) create mode 100644 mbuf.c (limited to 'mbuf.c') diff --git a/mbuf.c b/mbuf.c new file mode 100644 index 0000000..871bdba --- /dev/null +++ b/mbuf.c @@ -0,0 +1,187 @@ +/* + * OpenVPN -- An application to securely tunnel IP networks + * over a single TCP/UDP port, with support for SSL/TLS-based + * session authentication and key exchange, + * packet encryption, packet authentication, and + * packet compression. + * + * Copyright (C) 2002-2005 OpenVPN Solutions LLC + * + * This program is free software; you can redistribute it and/or modify + * it under the terms of the GNU General Public License version 2 + * as published by the Free Software Foundation. + * + * This program is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * along with this program (see the file COPYING included with this + * distribution); if not, write to the Free Software Foundation, Inc., + * 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA + */ + +#ifdef WIN32 +#include "config-win32.h" +#else +#include "config.h" +#endif + +#include "syshead.h" + +#if P2MP + +#include "buffer.h" +#include "error.h" +#include "misc.h" +#include "mbuf.h" + +#include "memdbg.h" + +struct mbuf_set * +mbuf_init (unsigned int size) +{ + struct mbuf_set *ret; + ALLOC_OBJ_CLEAR (ret, struct mbuf_set); + mutex_init (&ret->mutex); + ret->capacity = adjust_power_of_2 (size); + ALLOC_ARRAY (ret->array, struct mbuf_item, ret->capacity); + return ret; +} + +void +mbuf_free (struct mbuf_set *ms) +{ + if (ms) + { + int i; + for (i = 0; i < (int) ms->len; ++i) + { + struct mbuf_item *item = &ms->array[MBUF_INDEX(ms->head, i, ms->capacity)]; + mbuf_free_buf (item->buffer); + } + free (ms->array); + mutex_destroy (&ms->mutex); + free (ms); + } +} + +struct mbuf_buffer * +mbuf_alloc_buf (const struct buffer *buf) +{ + struct mbuf_buffer *ret; + ALLOC_OBJ (ret, struct mbuf_buffer); + ret->buf = clone_buf (buf); + ret->refcount = 1; + ret->flags = 0; + return ret; +} + +void +mbuf_free_buf (struct mbuf_buffer *mb) +{ + if (mb) + { + if (--mb->refcount <= 0) + { + free_buf (&mb->buf); + free (mb); + } + } +} + +void +mbuf_add_item (struct mbuf_set *ms, const struct mbuf_item *item) +{ + ASSERT (ms); + mutex_lock (&ms->mutex); + if (ms->len == ms->capacity) + { + struct mbuf_item rm; + ASSERT (mbuf_extract_item (ms, &rm, false)); + mbuf_free_buf (rm.buffer); + msg (D_MULTI_DROPPED, "MBUF: mbuf packet dropped"); + } + + ASSERT (ms->len < ms->capacity); + + ms->array[MBUF_INDEX(ms->head, ms->len, ms->capacity)] = *item; + if (++ms->len > ms->max_queued) + ms->max_queued = ms->len; + ++item->buffer->refcount; + mutex_unlock (&ms->mutex); +} + +bool +mbuf_extract_item (struct mbuf_set *ms, struct mbuf_item *item, const bool lock) +{ + bool ret = false; + if (ms) + { + if (lock) + mutex_lock (&ms->mutex); + while (ms->len) + { + *item = ms->array[ms->head]; + ms->head = MBUF_INDEX(ms->head, 1, ms->capacity); + --ms->len; + if (item->instance) /* ignore dereferenced instances */ + { + ret = true; + break; + } + } + if (lock) + mutex_unlock (&ms->mutex); + } + return ret; +} + +struct multi_instance * +mbuf_peek_dowork (struct mbuf_set *ms) +{ + struct multi_instance *ret = NULL; + if (ms) + { + int i; + mutex_lock (&ms->mutex); + for (i = 0; i < (int) ms->len; ++i) + { + struct mbuf_item *item = &ms->array[MBUF_INDEX(ms->head, i, ms->capacity)]; + if (item->instance) + { + ret = item->instance; + break; + } + } + mutex_unlock (&ms->mutex); + } + return ret; +} + +void +mbuf_dereference_instance (struct mbuf_set *ms, struct multi_instance *mi) +{ + if (ms) + { + int i; + mutex_lock (&ms->mutex); + for (i = 0; i < (int) ms->len; ++i) + { + struct mbuf_item *item = &ms->array[MBUF_INDEX(ms->head, i, ms->capacity)]; + if (item->instance == mi) + { + mbuf_free_buf (item->buffer); + item->buffer = NULL; + item->instance = NULL; + msg (D_MBUF, "MBUF: dereferenced queued packet"); + } + } + mutex_unlock (&ms->mutex); + } +} + +#else +static void dummy(void) {} +#endif /* P2MP */ -- cgit v1.2.3