Linux内核有一些基本的数据结构,这些数据结构是Linux实现的基础,对于链表相信大家都不陌生,但是Linux内核中的链表与平常平常我们所使用的链表略有不同,第一次遇到或许会感到困惑。
先来看一个链表的节点,对于一个节点,分为两部分,一部分是数据,另一部分是串联数据的指针。Linux链表节点的定义如下(以下代码皆为3.5版本):
// include/linux/types.h
struct list_head {
struct list_head *next, *prev;
};
这里的定义有些奇怪,因为仅有前后节点的指针,并没有数据,就像一串链子,只有线没有线上的珠子,肯定是无法使用,那Linux内核如何把这些“珠子”附着到线上的呢?
来看一个简单的例子:
struce simple {
int data;
struct list_head list;
};
simple结构体的list成员指向下一个或者上一个simple的list,这样便把节点串联起来了,data作为“珠子”附着在list线上,但这样仍然有一个问题,list成员仅仅指向下一个simple的list成员,那从list成员如何得到simple节点的地址呢?
答案是根据list成员的地址以及list成员在simple的位置便可以计算出simple对象的地址,这样有些繁琐,Linux提供了一个宏,可以简化这个过程:
// include/linux/list.h
/**
* list_entry - get the struct for this entry
* @ptr: the &struct list_head pointer.
* @type: the type of the struct this is embedded in.
* @member: the name of the list_struct within the struct.
*/
#define list_entry(ptr, type, member) \
container_of(ptr, type, member)
// include/linux/kernel.h
#define container_of(ptr, type, member) ({ \
const typeof( ((type *)0)->member ) *__mptr = (ptr); \
(type *)( (char *)__mptr - offsetof(type,member) );})
#undef offsetof
#ifdef __compiler_offsetof
#define offsetof(TYPE,MEMBER) __compiler_offsetof(TYPE,MEMBER)
#else
#define offsetof(TYPE, MEMBER) ((size_t) &((TYPE *)0)->MEMBER)
#endif
#endif /* __KERNEL__ */
可以看到,list_entry直接调用了container_of,container_of分为两句,((type *)0)->member可以获得member在结构体type中的偏移,假设有一个结构体在地址0的位置,那么成员的地址便是成员对结构体的偏移,typeof是gcc的扩展,用于获取变量的类型,offsetof(type,member) 获取member成员在type中的偏移,然后使用member成员的指针ptr(复制成__mptr)减去偏移,即是结构体的地址。在我们的例子中,从list成员的地址获取simple结构的地址如下:
simple * p = list_entry(ptr, struct simple, list);
这样便解决了从list_head上获取附着的数据的问题。接下来需要解决对链表的增删改查的问题:
一、初始化链表:
初始化链表有两种方法,LIST_HEAD_INIT和LIST_HEAD
#define LIST_HEAD_INIT(name) { &(name), &(name) }
#define LIST_HEAD(name) \
struct list_head name = LIST_HEAD_INIT(name)
static inline void INIT_LIST_HEAD(struct list_head *list)
{
list->next = list;
list->prev = list;
}
创建一个指向自身的节点。
二、插入:
在节点后插入新节点list_add_tail,和在节点前插入新节点list_add:
/**
* list_add - add a new entry
* @new: new entry to be added
* @head: list head to add it after
*
* Insert a new entry after the specified head.
* This is good for implementing stacks.
*/
static inline void list_add(struct list_head *new, struct list_head *head)
{
__list_add(new, head, head->next);
}
/**
* list_add_tail - add a new entry
* @new: new entry to be added
* @head: list head to add it before
*
* Insert a new entry before the specified head.
* This is useful for implementing queues.
*/
static inline void list_add_tail(struct list_head *new, struct list_head *head)
{
__list_add(new, head->prev, head);
}
其中 __list_add 只是普通的链表操作,并无特别之处,可参见Linux源码查看实现。
三、删除节点:
static inline void list_del(struct list_head *entry)
{
__list_del(entry->prev, entry->next);
entry->next = LIST_POISON1;
entry->prev = LIST_POISON2;
}
__list_del 把entry从链表中删除,之后把entry链表指针复制成非空指针(如果使用会出现段错误)
四、检查是否空链表
判断一个链表是否为空,只需要看头节点是否指向自己便可:
static inline int list_empty(const struct list_head *head)
{
return head->next == head;
}
五、遍历
遍历是这几种操作中最为复杂的,有四个函数:
#define list_for_each(pos, head) \
for (pos = (head)->next; pos != (head); pos = pos->next)
#define list_for_each_prev(pos, head) \
for (pos = (head)->prev; pos != (head); pos = pos->prev)
#define list_for_each_entry(pos, head, member) \
for (pos = list_entry((head)->next, typeof(*pos), member); \
&pos->member != (head); \
pos = list_entry(pos->member.next, typeof(*pos), member))
#define list_for_each_entry_reverse(pos, head, member) \
for (pos = list_entry((head)->prev, typeof(*pos), member); \
&pos->member != (head); \
pos = list_entry(pos->member.prev, typeof(*pos), member))
list_for_each 和 list_for_each_prev 较为简单,一个向后遍历,另一个向前遍历,list_for_each_entry和list_for_each_entry_reverse功能相似,不过不是对list_head操作,而是直接对结构体操作,如我们这里的simple结构。根据之前的叙述也不难理解函数实现,只是在list_head上调用了list_entry获取了完整结构。
六、实例
千言万语不如一个例子来的直观,我们通过一个简单的例子说明一下如何使用内核链表:
#include <linux/list.h>
#include <linux/kernel.h>
#include <stdio.h>
struct simple {
int data;
struct list_head list;
};
int main()
{
int i = 0;
struct simple * p;
struct list_head * pos;
LIST_HEAD(head);
for (i = 0; i < 10; i++) {
p = (struct simple*)malloc(sizeof(struct simple));
p->data = i * 10;
list_add_tail(&p->list, &head);
}
list_for_each_entry(p, &head, list) {
printf("for %d\n", p->data);
}
while (!list_empty(&head)) {
pos = head.next;
p = (struct simple*)list_entry(pos,
struct simple, list);
list_del(pos);
printf("del %d\n", p->data);
free(p);
}
return 0;
}
编译参数为
gcc -D__KERNEL__ -I/usr/src/linux-headers-3.2.0-27-generic/include/ -I/usr/src/linux-headers-3.2.0-27-generic/arch/ia64/include/ simple.c
其中头文件中都是内核函数,需要宏__KERNEL__,否则大部分定义会被忽略。
转载自:http://tech.fancymore.com/page/143.html#more-143