zapret/nfq/packet_queue.c

#include <stdlib.h>
#include <string.h>

#include "packet_queue.h"

void rawpacket_queue_init(struct rawpacket_tailhead *q)
{
	TAILQ_INIT(q);
}
void rawpacket_free(struct rawpacket *rp)
{
	if (rp) free(rp->packet);
	free(rp);
}
struct rawpacket *rawpacket_dequeue(struct rawpacket_tailhead *q)
{
	struct rawpacket *rp;
	rp = TAILQ_FIRST(q);
	if (rp)	TAILQ_REMOVE(q, rp, next);
	return rp;
}
void rawpacket_queue_destroy(struct rawpacket_tailhead *q)
{
	struct rawpacket *rp;
	while((rp = rawpacket_dequeue(q))) rawpacket_free(rp);
}

struct rawpacket *rawpacket_queue(struct rawpacket_tailhead *q,const struct sockaddr_storage* dst,uint32_t fwmark,const char *ifout,const void *data,size_t len)
{
	struct rawpacket *rp = malloc(sizeof(struct rawpacket));
	if (!rp) return NULL;

	rp->packet = malloc(len);
	if (!rp->packet)
	{
		free(rp);
		return NULL;
	}
	
	rp->dst = *dst;
	rp->fwmark = fwmark;
	if (ifout)
	{
	    strncpy(rp->ifout,ifout,sizeof(rp->ifout));
	    rp->ifout[sizeof(rp->ifout)-1]=0;
	}
	else
	    rp->ifout[0]=0;
	memcpy(rp->packet,data,len);
	rp->len=len;
	
	TAILQ_INSERT_TAIL(q, rp, next);
	
	return rp;
}

unsigned int rawpacket_queue_count(const struct rawpacket_tailhead *q)
{
	const struct rawpacket *rp;
	unsigned int ct=0;
	TAILQ_FOREACH(rp, q, next) ct++;
	return ct;
}
bool rawpacket_queue_empty(const struct rawpacket_tailhead *q)
{
	return !TAILQ_FIRST(q);
}