Adding upstream version 1:7.0.3.upstream/1%7.0.3

Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>

1 files changed, 236 insertions, 0 deletions

diff --git a/src/packet-queue.c b/src/packet-queue.c
new file mode 100644
index 0000000..81c7798
--- /dev/null
+++ b/src/packet-queue.c
@@ -0,0 +1,236 @@
+/* Copyright (C) 2007-2019 Open Information Security Foundation
+ *
+ * You can copy, redistribute or modify this Program 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
+ * version 2 along with this program; if not, write to the Free Software
+ * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
+ * 02110-1301, USA.
+ */
+
+/**
+ * \file
+ *
+ * \author Victor Julien <victor@inliniac.net>
+ *
+ * Packet Queue portion of the engine.
+ */
+
+#include "suricata-common.h"
+#include "decode.h"
+#include "packet-queue.h"
+#include "threads.h"
+#include "suricata.h"
+#include "util-var.h"
+#include "pkt-var.h"
+#include "util-validate.h"
+
+#ifdef DEBUG
+void PacketQueueValidateDebug(PacketQueue *q);
+void PacketQueueValidate(PacketQueue *q);
+
+void PacketQueueValidateDebug(PacketQueue *q)
+{
+    SCLogDebug("q->len %u, q->top %p, q->bot %p", q->len, q->top, q->bot);
+
+    if (q->len == 0) {
+        BUG_ON(q->top != NULL);
+        BUG_ON(q->bot != NULL);
+    } else if(q->len == 1) {
+        SCLogDebug("q->top->next %p, q->top->prev %p", q->top->next, q->top->prev);
+        SCLogDebug("q->bot->next %p, q->bot->prev %p", q->bot->next, q->bot->prev);
+
+        BUG_ON(q->top != q->bot);
+        BUG_ON(q->top->next != NULL);
+        BUG_ON(q->bot->next != NULL);
+        BUG_ON(q->top->prev != NULL);
+        BUG_ON(q->bot->prev != NULL);
+    } else if (q->len == 2) {
+        SCLogDebug("q->top->next %p, q->top->prev %p", q->top->next, q->top->prev);
+        SCLogDebug("q->bot->next %p, q->bot->prev %p", q->bot->next, q->bot->prev);
+
+        BUG_ON(q->top == NULL);
+        BUG_ON(q->bot == NULL);
+
+        BUG_ON(q->top == q->bot);
+
+        BUG_ON(q->top->prev != NULL);
+        BUG_ON(q->top->next != q->bot);
+
+        BUG_ON(q->bot->prev != q->top);
+        BUG_ON(q->bot->next != NULL);
+    } else {
+        BUG_ON(q->top == NULL);
+        BUG_ON(q->bot == NULL);
+
+        SCLogDebug("q->top->next %p, q->top->prev %p", q->top->next, q->top->prev);
+        SCLogDebug("q->bot->next %p, q->bot->prev %p", q->bot->next, q->bot->prev);
+
+        BUG_ON(q->top == q->bot);
+        BUG_ON(q->top->prev != NULL);
+        BUG_ON(q->bot->next != NULL);
+
+        BUG_ON(q->top->next == q->bot);
+        BUG_ON(q->bot->prev == q->top);
+
+        Packet *p, *pp;
+        for (p = q->top, pp = p->prev; p != NULL; pp = p, p = p->next) {
+            SCLogDebug("p %p, pp %p, p->next %p, p->prev %p", p, pp, p->next, p->prev);
+            BUG_ON(pp != p->prev);
+        }
+
+    }
+}
+
+#define BUGGER_ON(cond) { \
+    if ((cond)) { \
+        PacketQueueValidateDebug(q); \
+    } \
+}
+
+void PacketQueueValidate(PacketQueue *q)
+{
+    if (q->len == 0) {
+        BUGGER_ON(q->top != NULL);
+        BUGGER_ON(q->bot != NULL);
+    } else if(q->len == 1) {
+        BUGGER_ON(q->top != q->bot);
+        BUGGER_ON(q->top->next != NULL);
+        BUGGER_ON(q->bot->next != NULL);
+        BUGGER_ON(q->top->prev != NULL);
+        BUGGER_ON(q->bot->prev != NULL);
+    } else if (q->len == 2) {
+        BUGGER_ON(q->top == NULL);
+        BUGGER_ON(q->bot == NULL);
+
+        BUGGER_ON(q->top == q->bot);
+
+        BUGGER_ON(q->top->prev != NULL);
+        BUGGER_ON(q->top->next != q->bot);
+
+        BUGGER_ON(q->bot->prev != q->top);
+        BUGGER_ON(q->bot->next != NULL);
+    } else {
+        BUGGER_ON(q->top == NULL);
+        BUGGER_ON(q->bot == NULL);
+
+        BUGGER_ON(q->top == q->bot);
+        BUGGER_ON(q->top->prev != NULL);
+        BUGGER_ON(q->bot->next != NULL);
+
+        BUGGER_ON(q->top->next == q->bot);
+        BUGGER_ON(q->bot->prev == q->top);
+
+        Packet *p, *pp;
+        for (p = q->top, pp = p->prev; p != NULL; pp = p, p = p->next) {
+            BUGGER_ON(pp != p->prev);
+        }
+
+    }
+}
+#endif /* DEBUG */
+
+static inline void PacketEnqueueDo(PacketQueue *q, Packet *p)
+{
+    //PacketQueueValidateDebug(q);
+
+    if (p == NULL)
+        return;
+
+    /* more packets in queue */
+    if (q->top != NULL) {
+        p->prev = NULL;
+        p->next = q->top;
+        q->top->prev = p;
+        q->top = p;
+    /* only packet */
+    } else {
+        p->prev = NULL;
+        p->next = NULL;
+        q->top = p;
+        q->bot = p;
+    }
+    q->len++;
+#ifdef DBG_PERF
+    if (q->len > q->dbg_maxlen)
+        q->dbg_maxlen = q->len;
+#endif /* DBG_PERF */
+    //PacketQueueValidateDebug(q);
+}
+
+void PacketEnqueueNoLock(PacketQueueNoLock *qnl, Packet *p)
+{
+    DEBUG_VALIDATE_BUG_ON(p->pkt_src == 0);
+    PacketQueue *q = (PacketQueue *)qnl;
+    PacketEnqueueDo(q, p);
+}
+
+void PacketEnqueue (PacketQueue *q, Packet *p)
+{
+    PacketEnqueueDo(q, p);
+}
+
+static inline Packet *PacketDequeueDo (PacketQueue *q)
+{
+    //PacketQueueValidateDebug(q);
+    /* if the queue is empty there are no packets left. */
+    if (q->len == 0) {
+        return NULL;
+    }
+    q->len--;
+
+    /* pull the bottom packet from the queue */
+    Packet *p = q->bot;
+
+    /* more packets in queue */
+    if (q->bot->prev != NULL) {
+        q->bot = q->bot->prev;
+        q->bot->next = NULL;
+        /* just the one we remove, so now empty */
+    } else {
+        q->top = NULL;
+        q->bot = NULL;
+    }
+
+    //PacketQueueValidateDebug(q);
+    p->next = NULL;
+    p->prev = NULL;
+    return p;
+}
+
+Packet *PacketDequeueNoLock (PacketQueueNoLock *qnl)
+{
+    PacketQueue *q = (PacketQueue *)qnl;
+    Packet *p = PacketDequeueDo(q);
+    DEBUG_VALIDATE_BUG_ON(p != NULL && p->pkt_src == 0);
+    return p;
+}
+
+Packet *PacketDequeue (PacketQueue *q)
+{
+    return PacketDequeueDo(q);
+}
+
+PacketQueue *PacketQueueAlloc(void)
+{
+    PacketQueue *pq = SCCalloc(1, sizeof(*pq));
+    if (pq == NULL)
+        return NULL;
+    SCMutexInit(&pq->mutex_q, NULL);
+    SCCondInit(&pq->cond_q, NULL);
+    return pq;
+}
+
+void PacketQueueFree(PacketQueue *pq)
+{
+    SCCondDestroy(&pq->cond_q);
+    SCMutexDestroy(&pq->mutex_q);
+    SCFree(pq);
+}