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.
95 lines
2.4 KiB
C
95 lines
2.4 KiB
C
/* Copyright (C) 2007-2013 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>
|
|
*
|
|
* Simple queue handler
|
|
*/
|
|
|
|
#include "suricata.h"
|
|
#include "packet-queue.h"
|
|
#include "decode.h"
|
|
#include "threads.h"
|
|
#include "threadvars.h"
|
|
|
|
#include "tm-queuehandlers.h"
|
|
#include "tmqh-simple.h"
|
|
|
|
Packet *TmqhInputSimple(ThreadVars *t);
|
|
void TmqhOutputSimple(ThreadVars *t, Packet *p);
|
|
void TmqhInputSimpleShutdownHandler(ThreadVars *);
|
|
|
|
void TmqhSimpleRegister (void)
|
|
{
|
|
tmqh_table[TMQH_SIMPLE].name = "simple";
|
|
tmqh_table[TMQH_SIMPLE].InHandler = TmqhInputSimple;
|
|
tmqh_table[TMQH_SIMPLE].InShutdownHandler = TmqhInputSimpleShutdownHandler;
|
|
tmqh_table[TMQH_SIMPLE].OutHandler = TmqhOutputSimple;
|
|
}
|
|
|
|
Packet *TmqhInputSimple(ThreadVars *t)
|
|
{
|
|
PacketQueue *q = t->inq->pq;
|
|
|
|
StatsSyncCountersIfSignalled(t);
|
|
|
|
SCMutexLock(&q->mutex_q);
|
|
|
|
if (q->len == 0) {
|
|
/* if we have no packets in queue, wait... */
|
|
SCCondWait(&q->cond_q, &q->mutex_q);
|
|
}
|
|
|
|
if (q->len > 0) {
|
|
Packet *p = PacketDequeue(q);
|
|
SCMutexUnlock(&q->mutex_q);
|
|
return p;
|
|
} else {
|
|
/* return NULL if we have no pkt. Should only happen on signals. */
|
|
SCMutexUnlock(&q->mutex_q);
|
|
return NULL;
|
|
}
|
|
}
|
|
|
|
void TmqhInputSimpleShutdownHandler(ThreadVars *tv)
|
|
{
|
|
int i;
|
|
|
|
if (tv == NULL || tv->inq == NULL) {
|
|
return;
|
|
}
|
|
|
|
for (i = 0; i < (tv->inq->reader_cnt + tv->inq->writer_cnt); i++)
|
|
SCCondSignal(&tv->inq->pq->cond_q);
|
|
}
|
|
|
|
void TmqhOutputSimple(ThreadVars *t, Packet *p)
|
|
{
|
|
SCLogDebug("Packet %p, p->root %p, alloced %s", p, p->root, BOOL2STR(p->pool == NULL));
|
|
|
|
PacketQueue *q = t->outq->pq;
|
|
|
|
SCMutexLock(&q->mutex_q);
|
|
PacketEnqueue(q, p);
|
|
SCCondSignal(&q->cond_q);
|
|
SCMutexUnlock(&q->mutex_q);
|
|
}
|
|
|