Virtual_blkdev.c
//============================================================================
// Name : Virtual_blkdev
// Author : Haier
// Version : 0.1
// Copyright : Your copyright notice
// Description : blkdev driver in c, Ansi-style , compile by GCC, Linux 2.6.32
//============================================================================
#include <linux/module.h>
#include <linux/moduleparam.h>
#include <linux/init.h>
#include <linux/sched.h>
#include <linux/kernel.h>
#include <linux/slab.h>
#include <linux/fs.h>
#include <linux/errno.h>
#include <linux/timer.h>
#include <linux/types.h>
#include <linux/fcntl.h>
#include <linux/hdreg.h>
#include <linux/kdev_t.h>
#include <linux/vmalloc.h>
#include <linux/genhd.h>
#include <linux/blkdev.h>
#include <linux/buffer_head.h>
#include <linux/bio.h>
#define VIRTUAL_BLKDEV_DEVICEMAJOR COMPAQ_SMART2_MAJOR
#define VIRTUAL_BLKDEV_DISKNAME "Virtual_blkdev"
#define VIRTUAL_BLKDEV_BYTES (8*1024*1024)
static struct request_queue *Virtual_blkdev_queue;
static struct gendisk *Virtual_blkdev_disk;
unsigned char Virtual_blkdev_data[VIRTUAL_BLKDEV_BYTES];
static void Virtual_blkdev_do_request(struct request_queue *q)
{
struct request *req;
int err = 0;
while((req = blk_fetch_request(q)) != NULL)
{
if(((blk_rq_pos(req) + blk_rq_sectors(req)) << 9) > VIRTUAL_BLKDEV_BYTES)
{
printk(KERN_ERR VIRTUAL_BLKDEV_DISKNAME": bad request: block=&llu, count=&u\n",
(unsigned long long)blk_rq_pos(req), blk_rq_sectors(req));
err = -EIO;
if (!__blk_end_request_cur(req, err))
__blk_end_request_cur(req, 0);
continue;
}
switch(rq_data_dir(req))
{
case READ:
memcpy(req->buffer, Virtual_blkdev_data + (blk_rq_pos(req) << 9),
blk_rq_sectors(req) << 9);
__blk_end_request_cur(req, 1);
break;
case WRITE:
memcpy(Virtual_blkdev_data + (blk_rq_pos(req) << 9),
req->buffer, blk_rq_sectors(req) << 9);
__blk_end_request_cur(req, 1);
break;
default:
break;
}
}
}
struct block_device_operations Virtual_blkdev_fops =
{
.owner = THIS_MODULE,
};
static int __init Virtual_blkdev_init(void)
{
int ret;
Virtual_blkdev_disk = alloc_disk(1);
if(!Virtual_blkdev_disk)
{
ret = -ENOMEM;
goto err_alloc_disk;
}
Virtual_blkdev_queue = blk_init_queue(Virtual_blkdev_do_request, NULL);
if(!Virtual_blkdev_queue)
{
ret = -ENOMEM;
goto err_init_queue;
}
strcpy(Virtual_blkdev_disk->disk_name, VIRTUAL_BLKDEV_DISKNAME);
Virtual_blkdev_disk->major = VIRTUAL_BLKDEV_DEVICEMAJOR;
Virtual_blkdev_disk->first_minor = 0;
Virtual_blkdev_disk->fops = &Virtual_blkdev_fops;
Virtual_blkdev_disk->queue = Virtual_blkdev_queue;
set_capacity(Virtual_blkdev_disk, VIRTUAL_BLKDEV_BYTES>>9);
add_disk(Virtual_blkdev_disk);
return 0;
err_init_queue:
put_disk(Virtual_blkdev_disk);
err_alloc_disk:
return ret;
}
static void __exit Virtual_blkdev_exit(void)
{
del_gendisk(Virtual_blkdev_disk);
put_disk(Virtual_blkdev_disk);
blk_cleanup_queue(Virtual_blkdev_queue);
}
module_init(Virtual_blkdev_init);
module_exit(Virtual_blkdev_exit);
Makefile
target = Virtual_blkdev
ifeq ($(KERNELRELEASE),)
KERNELDIR ?= /lib/modules/`uname -r`/build
modules:
$(MAKE) -C $(KERNELDIR) M=`pwd` modules
modules_install:
insmod $(target).ko
uninstall:
rmmod $(target).ko
clean:
rm -rf *.o *.mod.c *.ko *.symvers *.order *.unsigned
else
obj-m := $(target).o
endif