aboutsummaryrefslogtreecommitdiff
path: root/mbuf.c
diff options
context:
space:
mode:
Diffstat (limited to '')
-rw-r--r--mbuf.c187
1 files changed, 187 insertions, 0 deletions
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 <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 */