diff options
Diffstat (limited to 'framework/src/suricata/src/flow-queue.h')
-rw-r--r-- | framework/src/suricata/src/flow-queue.h | 85 |
1 files changed, 85 insertions, 0 deletions
diff --git a/framework/src/suricata/src/flow-queue.h b/framework/src/suricata/src/flow-queue.h new file mode 100644 index 00000000..b370cd22 --- /dev/null +++ b/framework/src/suricata/src/flow-queue.h @@ -0,0 +1,85 @@ +/* Copyright (C) 2007-2012 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> + */ + +#ifndef __FLOW_QUEUE_H__ +#define __FLOW_QUEUE_H__ + +#include "suricata-common.h" +#include "flow.h" + +/** Spinlocks or Mutex for the flow queues. */ +//#define FQLOCK_SPIN +#define FQLOCK_MUTEX + +#ifdef FQLOCK_SPIN + #ifdef FQLOCK_MUTEX + #error Cannot enable both FQLOCK_SPIN and FQLOCK_MUTEX + #endif +#endif + +/* Define a queue for storing flows */ +typedef struct FlowQueue_ +{ + Flow *top; + Flow *bot; + uint32_t len; +#ifdef DBG_PERF + uint32_t dbg_maxlen; +#endif /* DBG_PERF */ +#ifdef FQLOCK_MUTEX + SCMutex m; +#elif defined FQLOCK_SPIN + SCSpinlock s; +#else + #error Enable FQLOCK_SPIN or FQLOCK_MUTEX +#endif +} FlowQueue; + +#ifdef FQLOCK_SPIN + #define FQLOCK_INIT(q) SCSpinInit(&(q)->s, 0) + #define FQLOCK_DESTROY(q) SCSpinDestroy(&(q)->s) + #define FQLOCK_LOCK(q) SCSpinLock(&(q)->s) + #define FQLOCK_TRYLOCK(q) SCSpinTrylock(&(q)->s) + #define FQLOCK_UNLOCK(q) SCSpinUnlock(&(q)->s) +#elif defined FQLOCK_MUTEX + #define FQLOCK_INIT(q) SCMutexInit(&(q)->m, NULL) + #define FQLOCK_DESTROY(q) SCMutexDestroy(&(q)->m) + #define FQLOCK_LOCK(q) SCMutexLock(&(q)->m) + #define FQLOCK_TRYLOCK(q) SCMutexTrylock(&(q)->m) + #define FQLOCK_UNLOCK(q) SCMutexUnlock(&(q)->m) +#else + #error Enable FQLOCK_SPIN or FQLOCK_MUTEX +#endif + +/* prototypes */ +FlowQueue *FlowQueueNew(); +FlowQueue *FlowQueueInit(FlowQueue *); +void FlowQueueDestroy (FlowQueue *); + +void FlowEnqueue (FlowQueue *, Flow *); +Flow *FlowDequeue (FlowQueue *); + +void FlowMoveToSpare(Flow *); + +#endif /* __FLOW_QUEUE_H__ */ + |