理论证明,在一个生产者和一个消费者的情况下,两者之间的同步无需加锁,即可并发访问。Linux内核无锁队列kfifo完美践行了该理论,性能得到极大提升。工作中如果能使用到该定论,可以有效的提升程序的性能。
内核代码处处有惊喜,往往让人叹为观止,犹如桃花源,让你发现别有洞天的美景。kfifo的实现中使用如下几个trick:
- 一个整数如果是2的整数幂,那么对齐取余操作%,可以转化为与其&操作,提升效率
- 使用整数自动溢出构成的一个环来替代循环数组,简化判空、判满及各种长度运算
- 先操作buffer,再计数。实现无锁队列
1. kfifo结构体信息
kfifo是内核中一个First In First Out数据结构,循坏队列实现。
struct __kfifo {
unsigned int in;
unsigned int out;
unsigned int mask;
unsigned int esize;
void *data;
};
data:用于存放数据的缓存区
esize:缓存区每个元素的size(element size)
mask:缓冲区元素个数(size) - 1;使用&mask,替换%size,提升效率
in:队尾下标,入队列的offset为(in % size 或者 in & mask)
out:队首下标,出队列的offset为(out % size 或者 out & mask)
队尾下标in在有数据入队的时候,一直自增;队首下标out在有数据出队的时候,一直自增。内核使用了unsigned int溢出的特性,来实现循环队列,即in - out不管任何情况都为队列的长度,即使in < out,这里可以画图或者自己编程理解一下。
2. kfifo内存分配
int __kfifo_alloc(struct __kfifo *fifo, unsigned int size,
size_t esize, gfp_t gfp_mask)
{
/*
* round down to the next power of 2, since our 'let the indices
* wrap' technique works only in this case.
*/
size = roundup_pow_of_two(size);
fifo->in = 0;
fifo->out = 0;
fifo->esize = esize;
if (size < 2) {
fifo->data = NULL;
fifo->mask = 0;
return -EINVAL;
}
fifo->data = kmalloc(size * esize, gfp_mask)<