JZ2440块设备驱动------用ram模拟块设备

/* 参考:drivers\block\xd.c
 * drivers\block\z2ram.c
 * 功能:便于实验观察，无论是向磁盘读写，都是先把所有的读操作完后，
 * 然后在完成所有的写操作
 *
 * 2016年6月26日22:41:51
 */




#include <linux/module.h>
#include <linux/errno.h>
#include <linux/interrupt.h>
#include <linux/mm.h>
#include <linux/fs.h>
#include <linux/kernel.h>
#include <linux/timer.h>
#include <linux/genhd.h>
#include <linux/hdreg.h>
#include <linux/ioport.h>
#include <linux/init.h>
#include <linux/wait.h>
#include <linux/blkdev.h>
#include <linux/blkpg.h>
#include <linux/delay.h>
#include <linux/io.h>


#include <asm/system.h>
#include <asm/uaccess.h>
#include <asm/dma.h>


static struct gendisk *ramblock_disk;
static request_queue_t * ramblock_queue;


#define RAMBLOCK_SIZE (1024*1024)
static unsigned char * ramblock_buf;


/*定义自旋锁*/
static DEFINE_SPINLOCK(ramblock_lock);
static int major;


/* 为了兼容老工具fdisk(分区工具),使用ramblock_getgeo函数获得几何属性，
 * 现在的磁盘没有这些东西，为了兼容老工具fdisk(分区工具)，假设我们有这些
 * 东西，
 *
 */
static int ramblock_getgeo(struct block_device *bdev, struct hd_geometry *geo)
{
/* 容量=heads*cylinders*sectors*512 */
geo->heads     = 2;/*磁头数(面数)*/
geo->cylinders = 32;/*柱面数(多少环)*/
geo->sectors   = RAMBLOCK_SIZE/2/32/512;
return 0;
}






static struct block_device_operations ramblock_fops = {
.owner = THIS_MODULE,
.getgeo = ramblock_getgeo;
};






/* struct request结构体中的sector成员是表示下一个扇区，如果是读操作，则sector
 * 代表源，如果是写目的
 * struct request结构体中的current_nr_sectors是当前要处理的扇区个数，表示长度
 * struct request结构体中的buffer,要是写的话代表源，读的话代表目的
 *
 */
 
/* 把request放入请求队列request_queue之前就已经优化好了
 *
 */
static void do_ramblock_request (request_queue_t * q)
{
static int cnt   = 0;
static int w_cnt = 0;
static int r_cnt = 0;
struct request *req;
// printk("do_ramblock_request %d\n",++cnt);
/* 以电梯调度算法从q队列里面取出下一个请求
* 在while里面是操作
*/
while ((req = elv_next_request(q)) != NULL) 
{
/*传输数据三要素:源，长度，目的*/
/*源/目的:*/
/*sector扇区从0开始，目的= 起始地址ramblock_buf + 偏移offset*/
unsigned long offset = req->sector *512;

/*目的/源*/
//req->buffer

/*长度:*/
unsigned long len   = req->current_nr_sectors *512;
/*读写的方向*/
if (rq_data_dir(req) == READ)
{
printk("do_ramblock_request read %d\n", ++r_cnt);
memcpy(req->buffer,ramblock_buf+offset,len);
}
else
{
printk("do_ramblock_request write %d\n", ++w_cnt);
memcpy(ramblock_buf+offset,req->buffer,len);


}


/* 结束
* 1表示成功 0表示失败
*/
end_request(req, 1); /* wrap up, 0 = fail, 1 = success */
}
}


static int ramblock_init(void)
{
/* 1 分配一个gendisk结构体*/
/*次设备号个数:分区个数+1,因为次设备号为0表示整个磁盘，不算分区*/
ramblock_disk = alloc_disk(16);//15个分区
/* 2设置*/

/* 2.1分配设置队列:提供读写能力
* 当构造请求队列完成时(已经做了优化)，调用请求处理函数do_ramblock_request，
* 把request放入请求队列request_queue之前就已经优化好了
* blk_init_queue的参数为执行队列的处理函数，自旋锁
*/


/* 每个块设备都有一个请求队列，每个请求队列单独执行I/O调度
* ，请求队列是由请求结构实例链接成的双向链表，链表以及整个
* 队列的信息用结构request_queue描述，称为请求队列对象结构
* 或请求队列结构。它存放了关于挂起请求的信息以及管理请求队
* 列（如：电梯算法）所需要的信息。结构成员request_fn是来
* 自设备驱动程序的请求处理函数。
* 
*/




 
ramblock_queue = blk_init_queue(do_ramblock_request,&ramblock_lock);

ramblock_disk->queue = ramblock_queue;
/* 2.2设置其他属性:比如容量*/

/*注册块设备，在PC机上执行 cat /proc/devices 可查看相关信息*/
major = register_blkdev(0,"ramblock");

ramblock_disk->major = major;
/* 第一个次设备号
* 从0开始到16个次设备号都对应这个块设备
*/
ramblock_disk->first_minor = 0;
/* 名字
* 字符串格式化命令，主要功能是把格式化的数据写入某个字符串中
*/
sprintf(ramblock_disk->disk_name, "ramblock");
/*操作函数必须要有，即使是空的*/
ramblock_disk->fops = &ramblock_fops;
/* 设置容量
* 单位是以扇区为单位的，一扇区为512字节
*/
set_capacity(ramblock_disk, RAMBLOCK_SIZE/512);


/*3硬件相关的操作*/
/* 分配内存
* GFP_KERNEL表示内核内存的正常分配. 可能睡眠.
* kzalloc函数分配的内存都被初始化为0，是为了室内和运行速度更快
*/
ramblock_buf =  kzalloc(RAMBLOCK_SIZE,GFP_KERNEL);
/* 4注册*/
add_disk(ramblock_disk);
return 0;


}


static void ramblock_exit(void)
{
unregister_blkdev(major,"ramblock");
del_gendisk(ramblock_disk);
put_disk(ramblock_disk);
blk_cleanup_queue(ramblock_queue);
kfree(ramblock_buf);


}


module_init(ramblock_init);
module_exit(ramblock_exit);
MODULE_LICENSE("GPL");