点击(此处)折叠或打开

1. #include <linux/init.h>
   
2. #include <linux/module.h>
   
3. #include <linux/kernel.h>
   
4. #include <linux/kfifo.h>
   
5. 
   
6. MODULE_LICENSE("GPL");
   
7. MODULE_AUTHOR("zl");
   
8. MODULE_VERSION("V1.0");
   
9. MODULE_DESCRIPTION("kfifo test");
   
10. 
    
11. struct KfifoNode {
    
12.     unsigned int num;
    
13.     char *string;
    
14. };
    
15. 
    
16. char *array[] = {
    
17.     "abcdefg",
    
18.     "gfedcba",
    
19.     "aaaaa",
    
20.     "bbbb",
    
21.     "ccc",
    
22.     "dd",
    
23.     "e",
    
24.     "12345",
    
25.     "1234",
    
26.     "123",
    
27.     "12",
    
28.     "1",
    
29.     "1111",
    
30. };
    
31. 
    
32. #define TAB_SIZE(array) (sizeof(array)/sizeof(array[0]))
    
33. 
    
34. static struct kfifo *pkfifoSpace;
    
35. static spinlock_t fifoLock; 
    
36. 
    
37. static void kfifo_check(char* str, int line, struct kfifo* pkfifo) 
    
38. {
    
39.     if(pkfifo != NULL) 
    
40.     {
    
41.         printk("[%s-%d]: pkfifo->size = %d\t pkfifo->in = %d\t pkfifo->out = %d\t \n",str, line, pkfifo->size, pkfifo->in, pkfifo->out);
    
42.     }
    
43. }
    
44. 
    
45. static int __init mykfifo_init(void)
    
46. {
    
47.     int i; 
    
48.     struct KfifoNode *pstNode;
    
49. 
    
50.     pkfifoSpace = kfifo_alloc((sizeof(struct KfifoNode) << 4), GFP_KERNEL, &fifoLock);
    
51.     if (pkfifoSpace == NULL) 
    
52.     {
    
53.         printk("kfifo_alloc failed !\n");
    
54.         return -EFAULT;
    
55.     }
    
56.     spin_lock_init(&fifoLock); //Initial fifo spinlock 
    
57.     pstNode = kzalloc(sizeof(struct KfifoNode), GFP_KERNEL);
    
58. /****************************************************************/
    
59.     printk("*****************************************************\n");
    
60.     kfifo_check((char *)__func__, __LINE__, pkfifoSpace);
    
61.     for(i = 0; i < TAB_SIZE(array); i++)
    
62.     {
    
63.         pstNode->num = i;
    
64.         pstNode->string = (char *)array[i];
    
65.         kfifo_put(pkfifoSpace, (unsigned char *)pstNode, sizeof(struct KfifoNode)); //将数据写入缓冲区
    
66.         kfifo_check((char *)__func__, __LINE__, pkfifoSpace);
    
67.         printk("[%s-%d]:Num is: %d, Message is: %s\n", __func__, __LINE__, pstNode->num, pstNode->string);
    
68.     }
    
69. /***************************************************************/
    
70.     printk("-----------------------------------------------------\n");
    
71.     if(!kfifo_len(pkfifoSpace))
    
72.     {
    
73.         printk("[%s-%d]: kfifo_len return 0, test failed !!! \n", __func__, __LINE__);
    
74.         kfifo_reset(pkfifoSpace);
    
75.         kfifo_free(pkfifoSpace);
    
76. 
    
77.         return -1;
    
78.     }
    
79.     
    
80.     for(i = 0; i < TAB_SIZE(array); i++)
    
81.     {
    
82.         kfifo_get(pkfifoSpace, (unsigned char *)pstNode, sizeof(struct KfifoNode));
    
83.         kfifo_check((char *)__func__, __LINE__, pkfifoSpace);
    
84.         printk("[%s-%d]: Num is: %d, fifoMessage is: %s\n", __func__, __LINE__, pstNode->num, pstNode->string);
    
85.     }
    
86. /***************************************************************/
    
87.     printk("-----------------------------------------------------\n");
    
88.     kfifo_check((char *)__func__, __LINE__, pkfifoSpace);
    
89. 
    
90.     kfree(pstNode);
    
91.     kfifo_reset(pkfifoSpace);
    
92.     kfifo_free(pkfifoSpace);
    
93. 
    
94.     return 0;
    
95. 
    
96. }
    
97. 
    
98. static void __exit mykfifo_exit(void)
    
99. {
    
100.     printk("exit !\n");
     
101. }
     
102. 
     
103. module_init(mykfifo_init);
     
104. module_exit(mykfifo_exit);