diff options
author | Ashlee Young <ashlee@onosfw.com> | 2015-09-09 22:21:41 -0700 |
---|---|---|
committer | Ashlee Young <ashlee@onosfw.com> | 2015-09-09 22:21:41 -0700 |
commit | 8879b125d26e8db1a5633de5a9c692eb2d1c4f83 (patch) | |
tree | c7259d85a991b83dfa85ab2e339360669fc1f58e /framework/src/suricata/src/defrag-hash.h | |
parent | 13d05bc8458758ee39cb829098241e89616717ee (diff) |
suricata checkin based on commit id a4bce14770beee46a537eda3c3f6e8e8565d5d0a
Change-Id: I9a214fa0ee95e58fc640e50bd604dac7f42db48f
Diffstat (limited to 'framework/src/suricata/src/defrag-hash.h')
-rw-r--r-- | framework/src/suricata/src/defrag-hash.h | 103 |
1 files changed, 103 insertions, 0 deletions
diff --git a/framework/src/suricata/src/defrag-hash.h b/framework/src/suricata/src/defrag-hash.h new file mode 100644 index 00000000..2d48393a --- /dev/null +++ b/framework/src/suricata/src/defrag-hash.h @@ -0,0 +1,103 @@ +/* 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 __DEFRAG_HASH_H__ +#define __DEFRAG_HASH_H__ + +#include "decode.h" +#include "defrag.h" + +/** Spinlocks or Mutex for the flow buckets. */ +//#define DRLOCK_SPIN +#define DRLOCK_MUTEX + +#ifdef DRLOCK_SPIN + #ifdef DRLOCK_MUTEX + #error Cannot enable both DRLOCK_SPIN and DRLOCK_MUTEX + #endif +#endif + +#ifdef DRLOCK_SPIN + #define DRLOCK_TYPE SCSpinlock + #define DRLOCK_INIT(fb) SCSpinInit(&(fb)->lock, 0) + #define DRLOCK_DESTROY(fb) SCSpinDestroy(&(fb)->lock) + #define DRLOCK_LOCK(fb) SCSpinLock(&(fb)->lock) + #define DRLOCK_TRYLOCK(fb) SCSpinTrylock(&(fb)->lock) + #define DRLOCK_UNLOCK(fb) SCSpinUnlock(&(fb)->lock) +#elif defined DRLOCK_MUTEX + #define DRLOCK_TYPE SCMutex + #define DRLOCK_INIT(fb) SCMutexInit(&(fb)->lock, NULL) + #define DRLOCK_DESTROY(fb) SCMutexDestroy(&(fb)->lock) + #define DRLOCK_LOCK(fb) SCMutexLock(&(fb)->lock) + #define DRLOCK_TRYLOCK(fb) SCMutexTrylock(&(fb)->lock) + #define DRLOCK_UNLOCK(fb) SCMutexUnlock(&(fb)->lock) +#else + #error Enable DRLOCK_SPIN or DRLOCK_MUTEX +#endif + +typedef struct DefragTrackerHashRow_ { + DRLOCK_TYPE lock; + DefragTracker *head; + DefragTracker *tail; +} DefragTrackerHashRow; + +/** defrag tracker hash table */ +DefragTrackerHashRow *defragtracker_hash; + +#define DEFRAG_VERBOSE 0 +#define DEFRAG_QUIET 1 + +typedef struct DefragConfig_ { + uint64_t memcap; + uint32_t hash_rand; + uint32_t hash_size; + uint32_t prealloc; +} DefragConfig; + +/** \brief check if a memory alloc would fit in the memcap + * + * \param size memory allocation size to check + * + * \retval 1 it fits + * \retval 0 no fit + */ +#define DEFRAG_CHECK_MEMCAP(size) \ + ((((uint64_t)SC_ATOMIC_GET(defrag_memuse) + (uint64_t)(size)) <= defrag_config.memcap)) + +DefragConfig defrag_config; +SC_ATOMIC_DECLARE(unsigned long long int,defrag_memuse); +SC_ATOMIC_DECLARE(unsigned int,defragtracker_counter); +SC_ATOMIC_DECLARE(unsigned int,defragtracker_prune_idx); + +void DefragInitConfig(char quiet); +void DefragHashShutdown(void); + +DefragTracker *DefragLookupTrackerFromHash (Packet *); +DefragTracker *DefragGetTrackerFromHash (Packet *); +void DefragTrackerRelease(DefragTracker *); +void DefragTrackerClearMemory(DefragTracker *); +void DefragTrackerMoveToSpare(DefragTracker *); +uint32_t DefragTrackerSpareQueueGetSize(void); + +#endif /* __DEFRAG_HASH_H__ */ + |