리눅스 커널 모듈 프로그래밍 - mychardev

리눅스 커널 모듈 프로그래밍 - mychardev

이전 글에서 올렸던 이 문서 이후 예제를 이용하여 아주 간단한 Character Device Driver를 작성했다. 사실상 거의 코드를 가져다 썼지만 내용 이해를 위해 printk를 몇번 하여 좀더 보기 편해졌다.

#include <linux/kernel.h>
#include <linux/module.h>
#include <linux/fs.h>
#include <asm/uaccess.h>
 
int init_module(void);
void cleanup_module(void);
static int device_open(struct inode *, struct file *);
static int device_release(struct inode *, struct file *);
static ssize_t device_read(struct file *, char *, size_t, loff_t *);
static ssize_t device_write(struct file *, const char *, size_t, loff_t *);
 
#define SUCCESS 0
#define DEVICE_NAME "mychardev"
#define BUF_LEN 80
 
static int Major;
static int Device_Open = 0;
 
static char msg[BUF_LEN];
static char *msg_Ptr;
static struct file_operations fops = {
    .read = device_read,
    .write = device_write,
    .release = device_release,
    .open = device_open
};

#include "mychardev.h"
 
int init_module(void)
{
    printk(KERN_INFO "[+] Module Init\n");
    Major = register_chrdev(0, DEVICE_NAME, &fops);
    if (Major < 0) {
        printk(KERN_INFO "[-] Registration chrdev failed with : %d\n", Major);
        return Major;
    }
    printk(KERN_INFO "[+] Registration chrdev success : %d\n", Major);
    printk(KERN_INFO "[+] Create device command : mknod /dev/%s c %d 0\n", DEVICE_NAME, Major);
    return SUCCESS;
}
 
void cleanup_module(void)
{
    unregister_chrdev(Major, DEVICE_NAME);
    printk(KERN_INFO "[+] Module Exit\n");
}
 
// Ex : cat /dev/mychardev
static int device_open(struct inode *inode, struct file *file)
{
    static int counter = 0;
    printk("[*] Device Open\n");
    if(Device_Open)
        return -EBUSY;
    Device_Open++;
    sprintf(msg, "- %d times read\n", ++counter);
    msg_Ptr = msg;
    try_module_get(THIS_MODULE);
    return SUCCESS;
}
 
static int device_release(struct inode *inode, struct file *file)
{
    printk("[*] Device Release\n");
    Device_Open--;
    module_put(THIS_MODULE);
    return 0;
}
 
static ssize_t device_read(struct file *flip, char *buffer, size_t length, loff_t *offset)
{
    int bytes_read = 0;
    printk("[*] Device read\n");
    if(*msg_Ptr == 0)
        return 0;
 
    while(length && *msg_Ptr)
    {
        put_user(*(msg_Ptr++), buffer++);
        length--;
        bytes_read++;
    }
    return bytes_read;
}
 
// Ex : echo "test" > /dev/mychardev
static ssize_t device_write(struct file *flip, const char *buffer, size_t length, loff_t *offset)
{
    printk("[*] Device Write");
    printk("[-] Not supported operation\n");
    return -EINVAL;
}

결과 화면은 다음과 같다.