我的第一个驱动

lullaby2005

第一个简陋的字符设备驱动，只实现了open,write,read,close，不过作为入门，还是增加了一点信心
/*my_dev.c*/
               
               
                #include linux/init.h>
#include linux/module.h>
#include linux/config.h>
#include linux/kernel.h>
#include linux/slab.h>
#include linux/fs.h>
#include linux/types.h>
#include linux/errno.h>
#include linux/cdev.h>
#include linux/fcntl.h>
#include linux/seq_file.h>
#include asm/system.h>
#include asm/uaccess.h>
MODULE_LICENSE("GPL");
#define SIZE (4*1024)
static int minor_num = 0;
static int major_num = 0;
struct device
{
    char *buf;
    struct cdev my_cdev;
};
static struct device *g_my_device;
ssize_t my_read(struct file *filp, char *buf, size_t count, loff_t *f_pos);
ssize_t my_write(struct file *filp, const char *buf, size_t count, loff_t *f_pos);
int my_open(struct inode *inode, struct file *filp);
int my_release(struct inode *inode, struct file *filp);
struct file_operations my_fops =
{
    .owner   = THIS_MODULE,
    .read    = my_read,
    .write   = my_write,
    .open    = my_open,
        .release = my_release,
};
static int __init my_device_init(void)
{
    dev_t dev_num = 0;
    int res;
    int err;
    res = alloc_chrdev_region(&dev_num, minor_num, 1, "steven_device");
    major_num = MAJOR(dev_num);
    if (res  0)
    {
        printk("STEVEN alloc_chrdev_region error!\n");
        return -1;
    }
    g_my_device = kmalloc(sizeof(struct device), GFP_KERNEL);   
    if (NULL == g_my_device)
    {
        printk("STEVEN in %s[%d] kmalloc error!\n", __FUNCTION__, __LINE__);
    }
    memset(g_my_device, 0, sizeof(struct device));
    g_my_device->buf = kmalloc(sizeof(unsigned char) * SIZE, GFP_KERNEL);
    memset(g_my_device->buf, 0, sizeof(unsigned char) * SIZE);
    cdev_init(&(g_my_device->my_cdev), &my_fops);
    g_my_device->my_cdev.owner = THIS_MODULE;
    g_my_device->my_cdev.ops = &my_fops;
    err = cdev_add(&(g_my_device->my_cdev), dev_num, 1);
    if (err != 0)
        printk("STEVEN: cdev_add error!\n");
    return 0;
}
void __exit  my_device_exit(void)
{
    dev_t dev_num;
    dev_num = MKDEV(major_num, minor_num);
    cdev_del(&(g_my_device->my_cdev));
    kfree(g_my_device->buf);
    kfree(g_my_device);
    unregister_chrdev_region(dev_num, 1);
    return;
}
ssize_t my_read (struct file *filp, char __user *buf, size_t count, loff_t *f_pos)
{
    int ret;
    if ((g_my_device->buf)[0] == '\0')
    {
        ret = strlen("NULL");
        copy_to_user(buf, "NULL", strlen("NULL"));
        return 0;   
    }
    if (0 != copy_to_user(buf, g_my_device->buf, count))
    {
        printk("STEVEN: copy_to_user error!\n");
        return 0;
    }
    else
        return count;
}
ssize_t my_write(struct file *filp, const char __user *buf, size_t count, loff_t *f_pos)
{   
    memset(g_my_device->buf, 0, sizeof(char) * SIZE);
    if (0 != copy_from_user(g_my_device->buf, buf, count))   
    {
        printk("STEEVN: copy_from_user error!\n");
        return 0;
    }
    else
        return count;
}
int my_open(struct inode *inode, struct file *filp)
{
    struct device *my_device;
    my_device = container_of(inode->i_cdev, struct device, my_cdev);
    filp->private_data = my_device;
    return 0;
}
int my_release(struct inode *inode, struct file *filp)
{
    return 0;
}
module_init(my_device_init);
module_exit(my_device_exit);
/*read_test.c*/
#include stdio.h>
#include stdlib.h>
#include fcntl.h>
#include linux/stat.h>
#include linux/types.h>
int main ()
{
    int fd;
    int len;
    char buf[3100];
    memset(buf, 0, sizeof(buf));
    fd = open("/dev/steven_device", O_RDONLY);
    if (fd  0)
    {
        fprintf(stderr, "fd );
        exit(1);
    }
    len = read(fd, buf, 1000);
    if (len  0)
    {
        fprintf(stderr, "len );
        exit(1);
    }
    printf("%s\n", buf);
   
    close(fd);
    return 0;
}
/*write_test.c*/
#include stdio.h>
#include stdlib.h>
#include fcntl.h>
#include linux/stat.h>
#include linux/types.h>
int main (int argc, char *argv[])
{
    if (argc != 2)
    {
        fprintf(stderr, "Usage: ./write_test your_string\n");
        exit(1);
    }
    int fd;
    int len;
    fd = open("/dev/steven_device", O_WRONLY);
    if (fd  0)
    {
        fprintf(stderr, "STEVEN: fd );
        exit(1);
    }
    len = write(fd, argv[1], strlen(argv[1]));
    if (len  0)
    {
        fprintf(stderr, "STEVEN: len );
        exit(1);
    }
    close(fd);
    return 0;
}
/*Makefile*/
ifneq ($(KERNELRELEASE),)
    obj-m := my_dev.o
else
    KERNELDIR ?= /lib/modules/$(shell uname -r)/build
    PWD = $(shell pwd)
default:
    $(MAKE) -C $(KERNELDIR) M=$(PWD) modules
    gcc -o read_test read_test.c
    gcc -o write_test write_test.c
clean:
    rm -rf *.o *.ko *.mod* read_test write_test
install:
    /sbin/insmod my_dev.ko
uninstall:
    /sbin/rmmod my_dev.ko
endif
mknode命令是：
mknod /dev/steven_device c 254 0
/*254是在/proc/devices下面动态生成的主设备号*/
写的非常简陋，也许还有bug，不过这也算是偶在kernel编程方面迈出的第一步，相信后面会越来越好的！


本文来自ChinaUnix博客，如果查看原文请点：http://blog.chinaunix.net/u1/38534/showart_1089328.html