/* 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++) {

        SCMutexLock(&tv->inq->pq->mutex_q);

        SCCondSignal(&tv->inq->pq->cond_q);

        SCMutexUnlock(&tv->inq->pq->mutex_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);

}