编写简单的ramdisk(无请求队列)

  最近在研究块设备驱动的编写,看了赵磊大牛的《写一个块设备驱动》,受益匪浅,虽然能看懂里面说的,但动手写写代码还是能加深理解的,下面实现的ramdisk写的很简单,如果有错误,欢迎大牛们指正哈!

分配一块内存区存放ram disk数据

  为了简单,我直接静态分配了一个大小为16MB的内存区,当然对于编写驱动来说这个空间有点大。不过就是为了简单嘛,可以理解。

 

#define SIMP_BLKDEV_BYTES            (16 * 1024 * 1024)
unsigned char simp_blkdev_data[SIMP_BLKDEV_BYTES];

 

分配一个请求队列

  可以通过函数blk_alloc_queue分配一个默认的请求队列,用该方法生成的请求对面没有设置默认的IO调度器。如果调用blk_init_queue函数分配一个请求队列,会设置默认的IO调度器。因为是编写ram disk,不需要访问外部设备,所以不需要使用IO调度器,故使用blk_alloc_queue来分配一个请求队列。

simp_blkdev_queue = blk_alloc_queue(GFP_KERNEL);

设置自己的make_request_fn函数

  blk_alloc_queue分配的请求队列中make_request_fn是没有被赋值的,这也导致了前面说的不会使用默认的IO调度器,那么我们就必须自己实现这个函数,因为上层代码向请求队列发生请求时都是通过这个函数来完成的。因为我们使用内存来模拟块设备,所以其实连请求队列都不需要,上面分配它仅仅为了让上层代码能够使用请求队列中的make_request_fn函数,否则上层代码会不知道去哪里调用make_request_fn

  对于上层代码发出的请求,可以直接用make_request_fn函数来完成请求并直接将结果返回给上层的代码。具体如下:

 

static void simp_blkdev_make_request(struct request_queue *q, struct bio *bio)
{
    struct bio_vec *bvec;
    int i;
    void *dsk_mem;
    ...
    dsk_mem = simp_blkdev_data + (bio->bi_sector << 9);
    /* 遍历 bio 中所有的 bvec */
    bio_for_each_segment(bvec, bio, i)
    {
        void *iovec_mem;

        switch(bio_rw(bio))
        {
            case READ:
            case READA:             /* 读和预读都进行同样的处理 */
                /* 用 kmalloc 将请求页映射到非线性映射区域进行
                    * 访问,这种方法主要是为了兼容高端内存,
                    * (bvec->bv_page 可能源于高端内存) */
                iovec_mem = kmap(bvec->bv_page) + bvec->bv_offset;
                memcpy(iovec_mem, dsk_mem, bvec->bv_len);
                kunmap(bvec->bv_page);
                break;
            case WRITE:
                    iovec_mem = kmap(bvec->bv_page) + bvec->bv_offset;
                    memcpy(dsk_mem, iovec_mem, bvec->bv_len);
                    kunmap(bvec->bv_page);
                    break;

            default:
                    printk(KERN_ERR SIMP_BLKDEV_DISKNAME
                            ":Unknown value of bio_rw: %lu
",
                            bio_rw(bio));
                    bio_endio(bio, -EIO);
                    return ;
            }
        dsk_mem += bvec->bv_len;
    }
    bio_endio(bio, 0);
}

 

  上面代码之间通过判断bio中的请求类型来判断具体的操作。bio_endio是用来返回给调用者make_request_fn执行结果的。有了上面自定义的make_request_fn,我们还要把该函数的地址赋值给请求队列中的make_request_fn,这样上层代码就可以使用我们自定义的这个函数了,这个可以通过函数blk_queue_make_request来完成,该函数的作用就是为请求队列绑定make_request_fn方法。

  从上面的代码中可以看出,我们直接在make_request_fn中完成了上层代码的请求,但通常的方法make_request_fn仅仅是更加上层的请求生成request结构,并将该request插入到请求队列中,再由请求队列中的request_fn来完成请求队列中的请求。为什么要把上层的请求先插入到请求队列中,而不是像我们上面那样直接处理请求呢?原因是这些请求大多数都是涉及到慢速的磁盘操作,缓存这些请求到请求队列中有利于合并相邻的请求和排序请求(IO调度程序要做的事情),这样有利于减少磁盘寻道的时间,大家都知道磁盘的寻道时间是非常慢的。而在这里由于我们涉及的只是内存操作,所有就没有必须用这么复杂的机制了,直接像处理字符设备请求那样,来一个请求就处理一个。

 

分配一个gendisk

  每个块设备都对应一个gendisk实例,这里也不例外,我们必须把为内存分配一个gendisk结构来把内存模拟为一个块设备。可以直接调用函数alloc_disk来分配一个gendisk结构。现在的问题是给设备分配一个什么设备号,可以可以随便选一个呢?显然是不行的,由于很多的设备号linux已经欲分配给了特定的设备,如果恰好选择了当前系统正在使用的设备号作为我们的设备号,那很显然最后会找不到我们的驱动程序,所以我们必须选一个系统一般都不用的设备号,打开linux/include/linux/major.h文件,我们会发现COMPAQ_SMART2_MAJORCOMPAQ_SMART2_MAJOR78个之多的设备号,并且貌似这个设备号很少使用,所以我们就可以选着它了。有了gendisk结构,接下来就是根据我们的需要初始化这个结构了,具体如下:

 

/* 分配一个 gendisk 结构 */
        simp_blkdev_disk = alloc_disk(1);
     /* 填充 gendisk 主要结构成员 */
        strcpy(simp_blkdev_disk->disk_name, SIMP_BLKDEV_DISKNAME);
        simp_blkdev_disk->major = SIMP_BLKDEV_DEVICEMAJOR;
        simp_blkdev_disk->first_minor = 0;
        simp_blkdev_disk->fops = &simp_blkdev_fops;
        simp_blkdev_disk->queue = simp_blkdev_queue;
        set_capacity(simp_blkdev_disk, SIMP_BLKDEV_BYTES >> 9);

 

  simp_blkdev_disk->disk_name设置磁盘的名字,这里我设置为cc这个名字最后会作为设备文件名字出现在/dev目录中,方便我们最后读写这个设备文件。

  simp_blkdev_disk->major是分配给我们这个虚拟设备的设备号,我选的是COMPAQ_SMART2_MAJOR,当然也可以选择其他的。simp_blkdev_disk->first_mino表示分配给设备的第一个从设备号,一般都是从0开始。simp_blkdev_disk->fops是根具体设备相关的底层函数集,由于我们用的是内存,所以这个只有简单的使用下面定义就行:

struct block_device_operations simp_blkdev_fops = 
{
        .owner = THIS_MODULE,
};

simp_blkdev_disk->queue就是前面创建的请求队列,set_capacity设置我们虚拟的这个设备的大小,以扇区为单位。

  到这里一切都准备就绪了,最后只要使用add_disk函数向内核注册我们虚拟的块设备就行了。

 

测试

  为了运行上面写得ram disk代码,最好的办法就是使用模块了。将上面的代码封装到模块中很简单,把创建请求队列和gendisk的代码都放在模块初始化代码中,把释放请求队列和gendisk的代码放在模块退出代码中。

 

static void __exit simp_blkdev_exit(void)
{
        del_gendisk(simp_blkdev_disk);        /* 删除 gendisk 结构 */
        put_disk(simp_blkdev_disk);        /* 释放一个该对象的引用 */
        blk_cleanup_queue(simp_blkdev_queue);    /* 清理请求队列 */
}

 

  一切都准备就绪了,现在直接编译模块,将模块插入到内核中就可以了。在将模块插入到内核后,可以查看系统消息:

  再查看/dev目录下,我们可以看到目录下多了cc这个文件。

 

  好了,现在就相当于我们有了一个块设备,大小为16MB,现在可以做的事就是格式化这个设备为ext4文件系统。

 

  接下来我们就可以将这个文件系统挂载到某个目录上进行操作了,我们可以在这个文件系统中创建文件,目录等。

 

 

 

 

 

 

原文地址:https://www.cnblogs.com/chengxuyuancc/p/3514812.html