misc.c

来自「我实现的一个基于零拷贝技术高速捕包的原型代码」· C语言 代码 · 共 56 行

#include "defs.h"

#define NEXT(x,radix)		(x+1)%radix
#define INC(x,radix)		x=NEXT(x,radix)

int enqueue(queue_t *q,uint val)
{
    if (!q)
        return 0;
    if(NEXT(q->tail,q->size) == q->head) /* full? */
        return -1;
    //多cpu下可能内核路径会交叉,读写者问题用锁保护

#ifdef __SMP__
    spin_lock(&q->lock);
#endif

    q->elem[q->tail] = val;
    INC(q->tail,q->size);
	
#ifdef __SMP__
    spin_unlock(&q->lock);
#endif
    return 0;
}

int dequeue(queue_t *q)
{
    if (!q)
        return 0;
    if(q->tail == q->head) /*empty ?*/
        return -1;
#ifdef __SMP__
    spin_lock(&q->lock);
#endif

    INC(q->head,q->size);

#ifdef __SMP__
    spin_unlock(&q->lock);
#endif

    return q->elem[q->head];
}

inline void init_queue(queue_t *q)
{
    if (!q)
        return;
    spin_lock_init(&q->lock)
    ;

    q->head = q->tail = 0;
}