diff options
Diffstat (limited to '')
| -rw-r--r-- | mbuf.c | 187 |
1 files changed, 187 insertions, 0 deletions
@@ -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 <info@openvpn.net> + * + * 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 */ |