mirror of https://github.com/OISF/suricata
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
101 lines
2.2 KiB
C
101 lines
2.2 KiB
C
/* Copyright (c) 2008 Victor Julien <victor@inliniac.net> */
|
|
|
|
#include "eidps-common.h"
|
|
#include "threads.h"
|
|
#include "debug.h"
|
|
#include "flow-queue.h"
|
|
#include "flow-util.h"
|
|
|
|
void FlowEnqueue (FlowQueue *q, Flow *f) {
|
|
/* more packets in queue */
|
|
if (q->top != NULL) {
|
|
f->lnext = q->top;
|
|
q->top->lprev = f;
|
|
q->top = f;
|
|
/* only packet */
|
|
} else {
|
|
q->top = f;
|
|
q->bot = f;
|
|
}
|
|
q->len++;
|
|
#ifdef DBG_PERF
|
|
if (q->len > q->dbg_maxlen)
|
|
q->dbg_maxlen = q->len;
|
|
#endif /* DBG_PERF */
|
|
}
|
|
|
|
Flow *FlowDequeue (FlowQueue *q) {
|
|
mutex_lock(&q->mutex_q);
|
|
|
|
Flow *f = q->bot;
|
|
if (f == NULL) {
|
|
mutex_unlock(&q->mutex_q);
|
|
return NULL;
|
|
}
|
|
|
|
/* more packets in queue */
|
|
if (q->bot->lprev != NULL) {
|
|
q->bot = q->bot->lprev;
|
|
q->bot->lnext = NULL;
|
|
/* just the one we remove, so now empty */
|
|
} else {
|
|
q->top = NULL;
|
|
q->bot = NULL;
|
|
}
|
|
|
|
q->len--;
|
|
|
|
f->lnext = NULL;
|
|
f->lprev = NULL;
|
|
|
|
mutex_unlock(&q->mutex_q);
|
|
return f;
|
|
}
|
|
|
|
void FlowRequeue(Flow *f, FlowQueue *srcq, FlowQueue *dstq)
|
|
{
|
|
if (srcq != NULL)
|
|
{
|
|
mutex_lock(&srcq->mutex_q);
|
|
|
|
/* remove from old queue */
|
|
if (srcq->top == f)
|
|
srcq->top = f->lnext; /* remove from queue top */
|
|
if (srcq->bot == f)
|
|
srcq->bot = f->lprev; /* remove from queue bot */
|
|
if (f->lprev)
|
|
f->lprev->lnext = f->lnext; /* remove from flow prev */
|
|
if (f->lnext)
|
|
f->lnext->lprev = f->lprev; /* remove from flow next */
|
|
|
|
srcq->len--; /* adjust len */
|
|
|
|
f->lnext = NULL;
|
|
f->lprev = NULL;
|
|
|
|
/* don't unlock if src and dst are the same */
|
|
if (srcq != dstq) mutex_unlock(&srcq->mutex_q);
|
|
}
|
|
|
|
/* now put it in dst */
|
|
if (srcq != dstq) mutex_lock(&dstq->mutex_q);
|
|
|
|
/* add to new queue (append) */
|
|
f->lprev = dstq->bot;
|
|
if (f->lprev)
|
|
f->lprev->lnext = f;
|
|
f->lnext = NULL;
|
|
dstq->bot = f;
|
|
if (dstq->top == NULL)
|
|
dstq->top = f;
|
|
|
|
dstq->len++;
|
|
#ifdef DBG_PERF
|
|
if (dstq->len > dstq->dbg_maxlen)
|
|
dstq->dbg_maxlen = dstq->len;
|
|
#endif /* DBG_PERF */
|
|
|
|
mutex_unlock(&dstq->mutex_q);
|
|
}
|
|
|