八、 指定回调函数
本节讲的内容十分关键。不管Linux驱动程序的功能多么复杂还是多么“酷”,都必须允许用户空间的应用程序与内核空间的驱动程序进行交互才有意义。而最 常用的交互方式就是读写设备文件。通过file_operations.read和file_operations.write成员变量可以分别指定读写 设备文件要调用的回调函数指针。
在本节将为word_count.c添加两个函数:word_count_read和word_count_write。这两个函数分别处理从设备文件读 数据和向设备文件写数据的动作。本节的例子先不考虑word_count要实现的统计单词数的功能,先用word_count_read和 word_count_write函数做一个读写设备文件数据的实验,以便让读者了解如何与设备文件交互数据。本节编写的word_count.c文件是 一个分支,读者可在word_count/read_write目录找到word_count.c文件。可以用该文件覆盖word_count目录下的同 名文件测试本节的例子。
本例的功能是向设备文件/dev/wordcount写入数据后,都可以从/dev/wordcount设备文件中读出这些数据(只能读取一次)。下面先看看本例的完整的代码。
#include #include #include #include #include #include #define DEVICE_NAME 'wordcount' // 定义设备文件名 static unsigned char mem[10000]; // 保存向设备文件写入的数据 static char read_flag = 'y'; // y:已从设备文件读取数据 n:未从设备文件读取数据 static int written_count = 0; // 向设备文件写入数据的字节数 // 从设备文件读取数据时调用该函数 // file:指向设备文件、buf:保存可读取的数据 count:可读取的字节数 ppos:读取数据的偏移量 static ssize_t word_count_read(struct file *file, char __user *buf, size_t count, loff_t *ppos) { // 如果还没有读取设备文件中的数据,可以进行读取 if(read_flag == 'n') { // 将内核空间的数据复制到用户空间,buf中的数据就是从设备文件中读出的数据 copy_to_user(buf, (void*) mem, written_count); // 向日志输出已读取的字节数 printk('read count:%d', (int) written_count); // 设置数据已读状态 read_flag = 'y'; return written_count; } // 已经从设备文件读取数据,不能再次读取数据 else { return 0; } } // 向设备文件写入数据时调用该函数 // file:指向设备文件、buf:保存写入的数据 count:写入数据的字节数 ppos:写入数据的偏移量 static ssize_t word_count_write(struct file *file, const char __user *buf, size_t count, loff_t *ppos) { // 将用户空间的数据复制到内核空间,mem中的数据就是向设备文件写入的数据 copy_from_user(mem, buf, count); // 设置数据的未读状态 read_flag = 'n'; // 保存写入数据的字节数 written_count = count; // 向日志输出已写入的字节数 printk('written count:%d', (int)count); return count; } // 描述与设备文件触发的事件对应的回调函数指针 // 需要设置read和write成员变量,系统才能调用处理读写设备文件动作的函数 static struct file_operations dev_fops = { .owner = THIS_MODULE, .read = word_count_read, .write = word_count_write }; // 描述设备文件的信息 static struct miscdevice misc = { .minor = MISC_DYNAMIC_MINOR, .name = DEVICE_NAME, .fops = &dev_fops }; // 初始化Linux驱动 static int word_count_init(void) { int ret; // 建立设备文件 ret = misc_register(&misc); // 输出日志信息 printk('word_count_init_successn'); return ret; } // 卸载Linux驱动 static void word_count_exit(void) { // 删除设备文件 misc_deregister(&misc); // 输出日志信息 printk('word_init_exit_successn'); } // 注册初始化Linux驱动的函数 module_init( word_count_init); // 注册卸载Linux驱动的函数 module_exit( word_count_exit); MODULE_AUTHOR('lining'); MODULE_DESCRIPTION('statistics of word count.'); MODULE_ALIAS('word count module.'); MODULE_LICENSE('GPL'); |
编写上面代码需要了解如下几点。
1. word_count_read和word_count_write函数的参数基本相同,只有第2个参数buf稍微一点差异。 word_count_read函数的buf参数类型是char*,而word_count_write函数的buf参数类型是const char*,这就意味着word_count_write函数中的buf参数值无法修改。word_count_read函数中的buf参数表示从设备文 件读出的数据,也就是说,buf中的数据都可能由设备文件读出,至于可以读出多少数据,取决于word_count_read函数的返回值。如果 word_count_read函数返回n,则可以从buf读出n个字符。当然,如果n为0,表示无法读出任何的字符。如果n小于0,表示发生了某种错误 (n为错误代码)。word_count_write函数中的buf表示由用户空间的应用程序写入的数据。buf参数前有一个“__user”宏,表示 buf的内存区域位于用户空间。
2. 由于内核空间的程序不能直接访问用户空间中的数据,因此,需要在word_count_read和word_count_write函数中分别使用 copy_to_user和copy_from_user函数将数据从内核空间复制到用户空间或从用户空间复制到内核空间。
3. 本例只能从设备文件读一次数据。也就是说,写一次数据,读一次数据后,第二次无法再从设备文件读出任何数据。除非再次写入数据。这个功能是通过 read_flag变量控制的。当read_flag变量值为n,表示还没有读过设备文件,在word_count_read函数中会正常读取数据。如果 read_flag变量值为y,表示已经读过设备文件中的数据,word_count_read函数会直接返回0。应用程序将无法读取任何数据。
4. 实际上word_count_read函数的count参数表示的就是从设备文件读取的字节数。但因为使用cat命令测试word_count驱动时。直 接读取了32768个字节。因此count参数就没什么用了(值总是32768)。所以要在word_count_write函数中将写入的字节数保存, 在word_count_read函数中直接使用写入的字节数。也就是说,写入多少个字节,就读出多少个字节。
5. 所有写入的数据都保存在mem数组中。该数组定义为10000个字符,因此写入的数据字节数不能超过10000,否则将会溢出。
为了方便读者测试本节的例子,笔者编写了几个Shell脚本文件,允许在UbuntuLinux、S3C6410开发板和Android模拟器上测试 word_count驱动。其中有一个负责调度的脚本文件build.sh。本书所有的例子都会有一个build.sh脚本文件,执行这个脚本文件就会要 求用户选择将源代码编译到那个平台,选择菜单如图6-11所示。用户可以输入1、2或3选择编译平台。如果直接按回车键,默认值会选择第1个编译平台 (UbuntuLinux)。
build.sh脚本文件的代码如下:
source /root/drivers/common.sh # select_target是一个函数,用语显示图6-11所示的选择菜单,并接收用户的输入 # 改函数在common.sh文件中定义 select_target if [ $selected_target == 1 ]; then source ./build_ubuntu.sh # 执行编译成Ubuntu Linux平台驱动的脚本文件 elif [ $selected_target == 2 ]; then source ./build_s3c6410.sh # 执行编译成s3c6410平台驱动的脚本文件 elif [ $selected_target == 3 ]; then source ./build_emulator.sh # 执行编译成Android模拟器平台驱动的脚本文件 fi |
在build.sh脚本文件中涉及到了3个脚本文件(build_ubuntu.sh、build_s3c6410.sh和 build_emulator.sh),这3个脚本文件的代码类似,只是选择的Linux内核版本不同。对于S3C6410和Android模拟器平台, 编译完后Linux驱动,会自动将编译好的Linux驱动文件(*.so文件)上传到相应平台的/data/local目录,并安装Linux驱动。例 如,build_s3c6410.sh脚本文件的代码如下:
source /root/drivers/common.sh # S3C6410_KERNEL_PATH变量是适用S3C6410平台的Linux内核源代码的路径, # 该变量以及其它类似变量都在common.sh脚本文件中定义 make -C $S3C6410_KERNEL_PATH M=${PWD} find_devices # 如果什么都选择,直接退出 if [ '$selected_device' == '' ]; then exit else # 上传驱动程序(word_count.ko) adb -s $selected_device push ${PWD}/word_count.ko /data/local # 判断word_count驱动是否存在 testing=$(adb -s $selected_device shell lsmod | grep 'word_count') if [ '$testing' != '' ]; then # 删除已经存在的word_count驱动 adb -s $selected_device shell rmmod word_count fi # 在S3C6410开发板中安装word_count驱动 adb -s $selected_device shell 'insmod /data/local/word_count.ko' fi |
使用上面的脚本文件,需要在read_write目录建立一个Makefile文件,内容如下:
obj-m := word_count.o
现在执行build.sh脚本文件,选择要编译的平台,并执行下面的命令向/dev/word_count设备文件写入数据。
# echo ‘hello lining’ > /dev/wordcount
然后执行如下的命令从/dev/word_count设备文件读取数据。
# cat /dev/wordcount
如果输出“hello lining”,说明测试成功。
注意:如 果在S3C6410开发板和Android模拟器上测试word_count驱动,需要执行shell.sh脚本文件或adb shell命令进入相应平台的终端。其中shell.sh脚本在/root/drivers目录中。这两种方式的区别是如果有多个Android设备和 PC相连时,shell.sh脚本会出现一个类似图6-11所示的选择菜单,用户可以选择进入哪个Android设备的终端,而adb shell命令必须要加-s命令行参数指定Android设备的ID才可以进入相应Android设备的终端。
九、实现统计单词数的算法
本节开始编写word_count驱动的业务逻辑:统计单词数。本节实现的算法将由空格、制表符(ASCII:9)、回车符(ASCII:13)和换行符 (ASCII:10)分隔的字符串算做一个单词,该算法同时考虑了有多个分隔符(空格符、制表符、回车符和换行符)的情况。下面是word_count驱 动完整的代码。在代码中包含了统计单词数的函数get_word_count。
#include #include #include #include #include #include #define DEVICE_NAME 'wordcount' // 定义设备文件名 static unsigned char mem[10000]; // 保存向设备文件写入的数据 static int word_count = 0; // 单词数 #define TRUE -1 #define FALSE 0 // 判断指定字符是否为空格(包括空格符、制表符、回车符和换行符) static char is_spacewhite(char c) { if(c == ' ' || c == 9 || c == 13 || c == 10) return TRUE; else return FALSE; } // 统计单词数 static int get_word_count(const char *buf) { int n = 1; int i = 0; char c = ' '; char flag = 0; // 处理多个空格分隔的情况,0:正常情况,1:已遇到一个空格 if(*buf == ' |