[Linux 驱动] -- Linux 设备驱动--块设备（三）之程序设计

块设备驱动注册与注销

块设备驱动中的第1个工作通常是注册它们自己到内核，完成这个任务的函数是 register_blkdev()，其原型为：

int register_blkdev(unsigned int major, const char *name);

major 参数是块设备要使用的主设备号，name 为设备名，它会在 /proc/devices 中被显示。如果major为0，内核会自动分配一个新的主设备号 register_blkdev()函数的返回值就是这个主设备号。如果返回1为负值，表明发生了一个错误。

与 register_blkdev() 对应的注销函数是 unregister_blkdev()，其原型为：

int unregister_blkdev(unsigned int major, const char *name);

这里，传递给 unregister_blkdev() 的参数必须与传递给 register_blkdev() 的参数匹配，否则这个函数返回 -EINVAL。

块设备的请求队列操作

标准的请求处理程序能排序请求，并合并相邻的请求，如果一个块设备希望是用标准的请求处理程序，那它必须调用函数blk_init_queue来初始化请求队列。当处理在队列上的请求时，必须持有队列自旋锁。初始化请求队列

request_queue_t *blk_init_queue(request_fn_proc *rfn, spinlock_t *lock);

该函数的第1个参数是请求处理函数的指针，第2个参数是控制访问队列权限的自旋锁，这个函数会发生内存分配的行为，故它可能会失败，函数调用成功时，它返回指向初始化请求队列的指针，否则，返回NULL。这个函数一般在块设备的模块加载函数中调用。清除请求队列

void blk_cleanup_queue(request_queue_t *q);

这个函数完成将请求队列返回给系统的任务，一般在块设备驱动模块卸载函数中调用。

提取请求：

struct request *elv_next_request(request_queue_t *queue);

上述函数用于返回下一个要处理的请求（由 I/O 调度器决定），如果没有请求则返回NULL。

去除请求：

void blkdev_dequeue_request(struct request *req);

上述函数从队列中去除一个请求。如果驱动中同时从同一个队列中操作了多个请求，它必须以这样的方式将它们从队列中去除。

分配 “请求队列”：

request_queue_t *blk_alloc_queue(int gfp_mask);

对于 FLASH、RAM盘等完全随机访问的非机械设备，并不需要进行复杂的 I/O 调度，这个时候，应该使用上述函数分配一个 “请求队列”，并使用如下函数来绑定 "请求队列" 和 "制造请求" 函数。

void blk_queue_mask_request(request_queue_t *q, mask_request_fn *mfn);

void blk_queue_hardsect_size(request_queue_t *queue, unsigned short max);

该函数用于告知内核块设备硬件扇区的大小，所有由内核产生的请求都是这个大小的倍数并且被正确对界。但是，内核块设备层和驱动之间的通信还是以512字节扇区为单位进行。

步骤：

在块设备驱动的模块加载函数中通常需要完成如下工作：

分配、初始化请求队列，绑定请求队列和请求函数。
分配、初始化gendisk，给gendisk的major、fops、queue等成员赋值，最后添加gendisk。
注册块设备驱动。

在块设备驱动的模块卸载函数中通常需要与模块加载函数相反的工作：

清除请求队列。
删除gendisk和对gendisk的引用。
删除对块设备的引用，注销块设备驱动。

总结：

块设备的I/O操作方式与字符设备存在较大的不同，因而引入了 request_queue、request、bio 等一系列数据结构。在整个块设备的 I/O 操作中，贯穿于始终的就是 "请求"，字符设备的 I/O 操作则时这届进行不绕弯，块设备的 I/O 操作会排队和整合。

驱动的任务是处理请求，对请求的排队和整合由 I/O 调度算法解决，因此，块设备驱动的核心就是请求处理韩硕或 "制造请求" 函数。

尽管在块设备驱动中国仍然存在 block_device_operations 结构体及其成员函数，但其不再包含读写一类的成员函数，而只是包含打开、释放及 I/O 控制等与具体读写无关的函数。块设备驱动结构相当复杂的，但幸运的是，块设备不想字符设备那么包罗万象，它通常就是存储设备，而且驱动的主体已经由Linux内核提供，针对一个特定的硬件系统，驱动工程师所涉及到的工作往往只是编写少量的与硬件直接交互的代码。

#include <linux/init.h>  
#include <linux/module.h>  
#include <linux/kernel.h>  
#include <linux/fs.h>
#include <asm/uaccess.h>
#include <linux/spinlock.h>
#include <linux/sched.h>
#include <linux/types.h>
#include <linux/fcntl.h>
#include <linux/hdreg.h>
#include <linux/genhd.h>
#include <linux/blkdev.h>
 
#define MAXBUF 1024	
 
 
#define BLK_MAJOR 253
 
char blk_dev_name[]="blk_dev";
static char flash[1024*16];
 
 
int major;
spinlock_t lock;
struct gendisk *gd;
 
 
 
/*块设备数据传输*/
static void blk_transfer(unsigned long sector, unsigned long nsect, char *buffer, int write)
{
	int read = !write;
	if(read)
	{
		memcpy(buffer, flash+sector*512, nsect*512);
	}
	else
	{
		memcpy(flash+sector*512, buffer, nsect*512);
	}
}
 
/*块设备请求处理函数*/
static void blk_request_func(struct request_queue *q)
{
	struct request *req;
	while((req = elv_next_request(q)) != NULL)  
	{
		if(!blk_fs_request(req))
		{
			end_request(req, 0);
			continue;
		}
		
		blk_transfer(req->sector, req->current_nr_sectors, req->buffer, rq_data_dir(req));
		/*rq_data_dir从request获得数据传送的方向*/
		/*req->current_nr_sectors 在当前段中将完成的扇区数*/
		/*req->sector 将提交的下一个扇区*/
		end_request(req, 1);
	}
}
 
/*strcut block_device_operations*/
static  int blk_ioctl(struct block_device *dev, fmode_t no, unsigned cmd, unsigned long arg)
{
       return -ENOTTY;
}
 
static int blk_open (struct block_device *dev , fmode_t no)
{
	printk("blk mount succeed\n");
	return 0;
}
static int blk_release(struct gendisk *gd , fmode_t no)
{
	printk("blk umount succeed\n");
	return 0;
}
struct block_device_operations blk_ops=
{
	.owner = THIS_MODULE,
	.open = blk_open,
	.release = blk_release,
    .ioctl = blk_ioctl,
};
 
//-----------------------------------------------
 
static int __init block_module_init(void)
{
	
	
	if(!register_blkdev(BLK_MAJOR, blk_dev_name)) //注册一个块设备
	{
		major = BLK_MAJOR;	
		printk("regiser blk dev succeed\n");
	}
	else
	{
		return -EBUSY;
	}
	gd = alloc_disk(1);  //分配一个gendisk，分区是一个
	spin_lock_init(&lock); //初始化一个自旋锁
	gd->major = major;
	gd->first_minor = 0;   //第一个次设备号
	gd->fops = &blk_ops;   //关联操作函数
 
	gd->queue = blk_init_queue(blk_request_func, &lock); //初始化请求队列并关联到gendisk
 
	snprintf(gd->disk_name, 32, "blk%c", 'a');  
	blk_queue_hardsect_size(gd->queue, 512);  //设置扇区大小512字节
	set_capacity(gd, 32);  //设置块设备大小 512*32=16K
	add_disk(gd);
	printk("gendisk init success!\n");
	return 0;
}
static void __exit block_module_exit(void)
{
	blk_cleanup_queue(gd->queue);
	del_gendisk(gd); 
	unregister_blkdev(BLK_MAJOR, blk_dev_name);
	printk("block module exit succeed!\n");
}
 
module_init(block_module_init);
module_exit(block_module_exit);
 
MODULE_LICENSE("GPL");
MODULE_AUTHOR("gec");
//------------------------------------------------------------------------------