module2.c

#include <linux/init.h>
#include <linux/module.h>
#include <linux/kernel.h>
#include <linux/fs.h>
#include <asm/uaccess.h>

MODULE_LICENSE("GPL");
MODULE_AUTHOR("dimadivan");
MODULE_DESCRIPTION("Character device test");
MODULE_VERSION("0.01");

#define DEVICE_NAME "chrdev1"
#define EXAMPLE_MSG "example message\n"
#define MSG_BUFFER_LEN 17

// объявление функций символьного устройства
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 *);
static int major_num;
static int device_open_count = 0;
static char msg_buffer[MSG_BUFFER_LEN];
static char *msg_ptr;

// эта структура указывает на все функции нашего устройства
static struct file_operations file_ops = {
    .read = device_read,
    .write = device_write,
    .open = device_open,
    .release = device_release
};

// когда процесс читает наше устройство, вызывается эта функция
static ssize_t device_read(struct file *flip, char *buffer, size_t len, loff_t *offset)
{
    int bytes_read = 0;
    // если мы в конце, прыгаем обратно в начало
    if (*msg_ptr == 0)
    {
        msg_ptr = msg_buffer;
    }
    // Записать данные в буфер
    while (len && *msg_ptr)
    {
        // буфер находится в пользовательском пространстве, поэтому нельзя просто воспользоваться указателем. Функция put_user делает все нужное за нас
        put_user(*(msg_ptr++), buffer++); // источник, приемник
        len--;
        bytes_read++;
    }
    return bytes_read;
}

// вызывается, когда процесс пытается записать что-то в наше устройство
static ssize_t device_write(struct file *flip, const char *buffer, size_t len, loff_t *offset)
{
    // наше устройство только для чтения
    printk(KERN_ALERT "This operation is not supported.\n");
    return -EINVAL;
}

// вызывается, когда процесс открывает наше устройство
static int device_open(struct inode *inode, struct file *file)
{
    // если устройство уже открыто, возвращаем соответствующий код
    if (device_open_count)
    {
        return -EBUSY;
    }
    device_open_count++; // увеличить счетчик открытий
    try_module_get(THIS_MODULE); // увеличить счетчик использований
    return 0;
}

// вызывается, когда процесс закрывает наше устройство
static int device_release(struct inode *inode, struct file *file)
{
    // уменьшаем счетчик открытий и счетчик использований. Без этого, модуль не выгрузится.
    device_open_count--; // уменьшить счетчик открытий
    module_put(THIS_MODULE); // уменьшить счетчик использований
    return 0;
}

static int __init chrdev1_init(void)
{
    // запишем в буфер наше сообщение
    strncpy(msg_buffer, EXAMPLE_MSG, MSG_BUFFER_LEN);
    // приравняем msg_ptr к началу буфера
    msg_ptr = msg_buffer;
    // Пытаемся зарегистрировать символьное устройство
    major_num = register_chrdev(0, DEVICE_NAME, &file_ops);
    if (major_num < 0)
    {
        printk(KERN_ALERT "Could not register device: %d\n", major_num);
        return major_num;
    }
    else
    {
        printk(KERN_INFO "chrdev1 module loaded with device major number %d\n", major_num);
        return 0;
    }
}

static void __exit chrdev1_exit(void)
{
    // Чистим все за собой. Разрегистрируем символьное устройство.
    unregister_chrdev(major_num, DEVICE_NAME);
    printk(KERN_INFO "Goodbye, World!\n");
}

// регистрируем функции модуля
module_init(chrdev1_init);
module_exit(chrdev1_exit);