转：Linux设备驱动程序设计实例

Linux系统中，设备驱动程序是操作系统内核的重要组成部分，在 与硬件设备之间
建立了标准的抽象接口。通过这个接口，用户可以像处理普通文件一样，对硬件设
备进行打开(open)、关闭(close)、读写(read/write)等操作。通过分析和设计设
备驱动程序，可以深入理解Linux系统和进行系统开发。本文通过一个简单的例子
来说明设备驱动程序的设计。
1、程序清单

1. //　　　　　　MyDev.c　　　　　 2000年2月7日编写
2. #ifndef __KERNEL__
3. #　define __KERNEL__　　//按内核模块编译
4. #endif
5. #ifndef MODULE
6. #　define MODULE　　　　　　　　//设备驱动程序模块编译
7. #endif
8. #define DEVICE_NAME "MyDev"　　 
9. #define OPENSPK 1
10. #define CLOSESPK 2
11. //必要的头文件
12. #include <linux/module.h>　//同kernel.h,最基本的内核模块头文件
13. #include <linux/kernel.h>　//同module.h,最基本的内核模块头文件
14. #include <linux/sched.h>　 //这里包含了进行正确性检查的宏
15. #include <linux/fs.h>　　  //文件系统所必需的头文件
16. #include <asm/uaccess.h>　 //这里包含了内核空间与用户空间进行数据交换时的
17. 函数宏
18. #include <asm/io.h>　　　//I/O访问
19. int my_major=0;　　　　　//主设备号
20. static int Device_Open=0;
21. static char Message[]="This is from device driver";
22. char *Message_Ptr;
23. int my_open(struct inode *inode, struct file *file)
24. {//每当应用程序用open打开设备时，此函数被调用
25. 　printk ("/ndevice_open(%p,%p)/n", inode, file);
26. 　if (Device_Open)
27. 　　return -EBUSY;　　　//同时只能由一个应用程序打开
28. 　Device_Open++;
29. 　MOD_INC_USE_COUNT;　　//设备打开期间禁止卸载
30. 　return 0;
31. }
32. static void　my_release(struct inode *inode, struct file *file)
33. {//每当应用程序用close关闭设备时，此函数被调用
34. 　printk ("/ndevice_release(%p,%p)/n", inode, file);
35. 　Device_Open --;
36. 　MOD_DEC_USE_COUNT;　　//引用计数减1
37. }
38. ssize_t　my_read (struct file *f,char *buf,int size,loff_t off)
39. {//每当应用程序用read访问设备时，此函数被调用
40. 　　int bytes_read=0;　　　 
41. #ifdef DEBUG
42. 　　printk("/nmy_read is called. User buffer is %p,size is %d/n",buf,size);
43. #endif
44. if (verify_area(VERIFY_WRITE,buf,size)==-EFAULT) 
45. 　　return -EFAULT;
46. 　　　　Message_Ptr=Message;
47. 　　　　while(size && *Message_Ptr)
48. 　　　　{　　　 
49. 　　　　　　if(put_user(*(Message_Ptr++),buf++))　　//写数据到用户空间
50. 　　　　　　　　return -EINVAL;
51. 　　　　　　size --;
52. 　　　　　　　　bytes_read++;
53. 　　　　}
54. 　　　　return bytes_read; 
55. }
56. ssize_t my_write (struct file *f,const char *buf, int size,loff_t off)
57. {//每当应用程序用write访问设备时，此函数被调用
58. 　　int i;
59. 　　　　unsigned char uc;
60.     #ifdef DEBUG
61. 　　    printk("/nmy_write is called. User buffer is %p,size is %d/n",buf,size);
62.     #endif
63.     if (verify_area(VERIFY_WRITE,buf,size)==-EFAULT) 
64.         return -EFAULT;
65. 　　printk("/nData below is from user program:/n");
66. 　　for (i=0;i<size;i++)
67. 　　　　if(!get_user(uc,buf++)) //从用户空间读数据
68. 　　　　　　printk("%02x ",uc);
69. 　　return size; 
70. }
71. int my_ioctl(struct inode *inod,struct file *f,unsigned int arg1,
72. unsigned int arg2)
73. {//每当应用程序用ioctl访问设备时，此函数被调用
74.     #ifdef DEBUG
75. 　　   printk("/nmy_ioctl is called. Parameter is %p,size is %d/n",arg1);
76. #endif
77.     switch (arg1)
78.     {
79. 　　  case OPENSPK:
80. 　　　　　  printk("/nNow,open PC's speaker./n");
81. 　　　　　  outb(inb(0x61)|3,0x61); //打开计算机的扬声器
82. 　　　　　  break;
83. 　　  case CLOSESPK:
84. 　　　　　  printk("/nNow,close PC's speaker.");
85. 　　　　　  outb(inb(0x61)&0xfc,0x61);//关闭计算机的扬声器
86. 　　　　　  break;
87.     }
88. }
89. 　　　　
90. struct file_operations my_fops = {
91. 　　NULL,　　　　　/* lseek */
92. 　　my_read,
93. 　　my_write,
94. 　　NULL,
95. 　　NULL,
96. 　　my_ioctl,
97. 　　NULL,
98. 　　my_open,
99. 　　my_release,
100. 　　/* nothing more, fill with NULLs */
101. };
102. int init_module(void)
103. {//每当装配设备驱动程序时，系统自动调用此函数
104. 　　int result;
105. 　　result = register_chrdev(my_major,DEVICE_NAME,&my_fops); 
106. 　　if (result < 0) 
107.         return result;
108. 　　if (my_major == 0) 
109. 　　　　my_major = result; 
110. 　　printk("/nRegister Ok. major-number=%d/n",result);
111. 　  return 0;
112. }
113. void cleanup_module(void)
114. {//每当卸载设备驱动程序时，系统自动调用此函数
115. 　　printk("/nunload/n");
116. 　　unregister_chrdev(my_major, DEVICE_NAME);
117. }

2、设备驱动程序设计
　　Linux设备分为字符设备、块设备和网络设备。字符设备是不需要缓冲而直接
读写的设备，如串口、键盘、鼠标等，本例就是字符设备驱动程序；块设备的访问
通常需要缓冲来支持，以数据块为单位来读写，如磁盘设备等；网络设备是通过套
接字来访问的特殊设备。
1) 设备驱动程序和内核与应用程序的接口
　　无论哪种类型的设备，Linux都是通过在内核中维护特殊的设备控制块来与设
备驱动程序接口的。在字符设备和块设备的控制块中，有一个重要的数据结构
file_operations，该结构中包含了驱动程序提供给应用程序访问硬件设备的各种
方法，其定义如下（参见fs.h）：

1. struct file_operations {
2. loff_t (*llseek) (struct file *, loff_t, int);
3. //响应应用程序中lseek调用的函数指针
4. ssize_t (*read) (struct file *, char *, size_t, loff_t *);
5. //响应应用程序中read调用的函数指针
6. ssize_t (*write) (struct file *, const char *, size_t, loff_t *);
7. //响应应用程序中write调用的函数指针
8. int (*readdir) (struct file *, void *, filldir_t); 
9. //响应应用程序中readdir调用的函数指针 
10. unsigned int (*poll) (struct file *, struct poll_table_struct *);
11. //响应应用程序中select调用的函数指针
12. int (*ioctl) (struct inode *, struct file *, unsigned int, unsigned long);
13. //响应应用程序中ioctl调用的函数指针
14. int (*mmap) (struct file *, struct vm_area_struct *); 
15. //响应应用程序中mmap调用的函数指针
16. int (*open) (struct inode *, struct file *);　　
17. //响应应用程序中open调用的函数指针
18. int (*flush) (struct file *);　　　　　　　 
19. int (*release) (struct inode *, struct file *);　
20. //响应应用程序中close调用的函数指针
21. int (*fsync) (struct file *, struct dentry *);　 
22. int (*fasync) (int, struct file *, int);
23. int (*check_media_change) (kdev_t dev);
24. int (*revalidate) (kdev_t dev);
25. int (*lock) (struct file *, int, struct file_lock *);
26. };

多数情况下，只需为上面结构中的少数方法编写服务函数，其他均设为NULL即可。每一个可装配的设备驱动程序都必须有init_module和cleanup_module两个函数，装载和卸载设备时内核自动调用这两个函数。在init_module中，除了可以对硬件
设备进行检查和初始化外，还必须调用register_* 函数将设备登记到系统中。本例中是通过register_chrdev来登记的，如果是块设备或网络设备则应该用register_blkdev和register_netdev来登记。Register_chrdev 的主要功能是将设备名和结构file_operations登记到系统的设备控制块中。
2) 与应用程序的数据交换
由于设备驱动程序工作在内核存储空间，不能简单地用"="、"memcpy"等方法与应用程序交换数据。在头文件uaccess.h中定义了方法put_user(x, ptr)和get_user(x, ptr)，用于内核空间与用户空间的数据交换。值x的类型根据指针ptr的类型确定，请参见源代码中的my_read与my_write函数。
3) 与硬件设备的接口
Linux中为设备驱动程序访问I/O端口、硬件中断和DMA提供了简便方法，相应的头文件分别为io.h、irq.h、dma.h。由于篇辐限制，本例中只涉及到I/O端口访问。Linux提供的I/O端口访问方法主要有：inb()、inw()、outb()、outw()、inb_p()、inw_p()、outb_p()、outw_p()等。要注意的是，设备驱动程序在使用端口前，应该先用check_region()检查该端口的占用情况，如果指定的端口可用，则再用request_region()向系统登记。说明check_region()、request_region()的头文件是ioport.h。
4) 内存分配
设备驱动程序作为内核的一部分，不能使用虚拟内存，必须利用内核提供的kmalloc()与kfree()来申请和释放内核存储空间。Kmalloc()带两个参数，第一个要申请的是内存数量，在早期的版本中，这个数量必须是2的整数幂，如128、256。关于kmalloc()与kfree()的用法，可参考内核源程序中的malloc.h与slab.c程序。

3、程序的编译和访问
本例在Linux 2.2.x.x中可以成功编译和运行。先用下面的命令行进行编译：
gcc -Wall -O2 -c MyDev.c
该命令行中参数-Wall告诉编译程序显示警告信息；参数-O2是关于代码优化的设置，注意内核模块必须优化；参数-c规定只进行编译和汇编，不进行连接。正确编译后，形成MyDev.o文件。可以输入命令insmod MyDev.o来装载此程序。如果装配成功，则显示信息：Register Ok.major-number=xx。利用命令lsmod可以看到该模块被装配到系统中。为了访问该模块，应该用命令mknode来创建设备文件。下面的应用程序可以对创建的设备文件进行访问。

1. #include <stdio.h>
2. #include <fcntl.h>
3. #include <sys/ioctl.h>
4. #define DEVICE_NAME MyDev
5. #define OPENSPK 1
6. #define CLOSESPK 2
7. char buf[128];
8. int main()
9. {
10.     int f=open(DEVICE_NAME,O_RDRW);
11. 　　 if (f==-1) 
12.         return 1;
13. 　　printf("/nHit enter key to read device...");
14. 　　    read(f,buf,128);　printf(buf);
15.     printf("/nHit enter key to write device ...");
16.         write(f,"test",4);
17.     printf("/nHit enter key to open PC's speaker...");
18.         ioctl(f,OPENSPK);
19.     printf("/nHit enter key to close PC's speaker...");
20.     ioctl(f,CLOSESPK);
21. 　　close(f);
22. }