驱动第三次作业

编写LED驱动，创建三个设备文件，每一个设备文件和一个LED灯绑定，当操作这个设备文件时只能控制对应的这盏灯。

hend.h

#ifndef __HEAD_H__
#define __HEAD_H__

typedef struct
{
    volatile unsigned int MODER;   // 0x00
    volatile unsigned int OTYPER;  // 0x04
    volatile unsigned int OSPEEDR; // 0x08
    volatile unsigned int PUPDR;   // 0x0C
    volatile unsigned int IDR;     // 0x10
    volatile unsigned int ODR;     // 0x14
    volatile unsigned int BSRR;    // 0x18
    volatile unsigned int LCKR;    // 0x1C
    volatile unsigned int AFRL;    // 0x20
    volatile unsigned int AFRH;    // 0x24
    volatile unsigned int BRR;     // 0x28
    volatile unsigned int res;
    volatile unsigned int SECCFGR; // 0x30

} gpio_t;

#define GPIOA 0x50002000
#define GPIOB 0x50003000
#define GPIOC 0x50004000
#define GPIOD 0x50005000
#define GPIOE 0x50006000
#define GPIOF 0x50007000
#define GPIOG 0x50008000
#define GPIOH 0x50009000
#define GPIOI 0x5000A000
#define GPIOJ 0x5000B000
#define GPIOK 0x5000C000
#define GPIOZ 0x54004000

#define PHY_RCC 0X50000a28

#define LED_ON _IO('i', 1)
#define LED_OFF _IO('i', 0)

#endif

test.c

#include <stdio.h>
#include <stdlib.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <unistd.h>
#include <string.h>
#include <sys/ioctl.h>
#include "head.h"

int main(int argc, char const *argv[])
{
    int a, b, c;
    int fd;
    while (1)
    {
        printf("请选择要控制的硬件:1(led1) 2(led2) 3(led3) 4(退出)>>>");
        scanf("%d", &a);
        switch (a)
        {
        case 1:
            fd = open("/dev/led0", O_RDWR);
            if (fd < 0)
            {
                printf("打开设备文件失败\n");
                exit(-1);
            }
            while (1)
            {
                printf("请选择:0(关灯) 1(开灯) 3(退出)>>>");
                scanf("%d", &b);
                switch (b)
                {
                case 1:
                    ioctl(fd, LED_ON);
                    break;
                case 0:
                    ioctl(fd, LED_OFF);
                    break;
                }
                if (3 == b)
                {
                    close(fd);
                    break;
                }
            }

            break;
        case 2:
            fd = open("/dev/led1", O_RDWR);
            if (fd < 0)
            {
                printf("打开设备文件失败\n");
                exit(-1);
            }
            while (1)
            {
                printf("请选择:0(关灯) 1(开灯) 3(退出)>>>");
                scanf("%d", &b);
                switch (b)
                {
                case 1:
                    ioctl(fd, LED_ON);
                    break;
                case 0:
                    ioctl(fd, LED_OFF);
                    break;
                }

                if (3 == b)
                {
                    close(fd);
                    break;
                }
            }
            break;
        case 3:
            fd = open("/dev/led1", O_RDWR);
            if (fd < 0)
            {
                printf("打开设备文件失败\n");
                exit(-1);
            }
            while (1)
            {
                printf("请选择:0(关灯) 1(开灯) 3(退出)>>>");
                scanf("%d", &b);
                switch (b)
                {
                case 1:
                    ioctl(fd, LED_ON);
                    break;
                case 0:
                    ioctl(fd, LED_OFF);
                    break;
                }

                if (3 == b)
                {
                    close(fd);
                    break;
                }
            }
            break;
        }
        if (4 == a)
        {
            break;
        }
    }
    return 0;
}

led.c

#include <linux/init.h>
#include <linux/module.h>
#include <linux/cdev.h>
#include <linux/fs.h>
#include <linux/device.h>
#include <linux/slab.h>
#include "head.h"
#include <linux/io.h>
#include <linux/uaccess.h>

struct cdev *cdev;
unsigned int major = 0;
unsigned int minor = 0;
dev_t devno;
struct class *cls;
struct device *dev;
gpio_t *gpioe;
gpio_t *gpiof;
unsigned int *rcc;

int mycdev_open(struct inode *inode, struct file *file)
{
    unsigned int led_minor = MINOR(inode->i_rdev);
    file->private_data = (void *)led_minor;
    printk("%s:%s:%d\n", __FILE__, __func__, __LINE__);
    return 0;
}

long mycdev_ioctl(struct file *file, unsigned int cmd, unsigned long arg)
{
    unsigned int led_minor = (int)file->private_data;
    switch (led_minor)
    {
    case 0:
        switch (cmd)
        {
        case LED_ON:
            gpioe->ODR |= (0x1 << 10);
            break;
        case LED_OFF:
            gpioe->ODR &= (~(0x1 << 10));
            break;
        default:
            break;
        }
        break;
    case 1:
        switch (cmd)
        {
        case LED_ON:
            gpiof->ODR |= (0x1 << 10);
            break;
        case LED_OFF:
            gpiof->ODR &= (~(0x1 << 10));
            break;
        default:
            break;
        }
        break;
    case 2:
        switch (cmd)
        {
        case LED_ON:
            gpioe->ODR |= (0x1 << 8);
            break;
        case LED_OFF:
            gpioe->ODR &= (~(0x1 << 8));
            break;
        default:
            break;
        }
        break;
    }
    return 0;
}

int mycdev_close(struct inode *inode, struct file *file)
{
    printk("%s:%s:%d\n", __FILE__, __func__, __LINE__);
    return 0;
}

struct file_operations fops = {
    .open = mycdev_open,
    .unlocked_ioctl = mycdev_ioctl,
    .release = mycdev_close,
};

int all_led_init(void)
{
    // 映射物理寄存器
    gpioe = ioremap(GPIOE, sizeof(gpio_t));
    if (gpioe == NULL)
    {
        printk("GPIOE地址映射表失败\n");
        return -EFAULT;
    }

    gpiof = ioremap(GPIOF, sizeof(gpio_t));
    if (gpiof == NULL)
    {
        printk("GPIOF地址映射表失败\n");
        return -EFAULT;
    }

    rcc = ioremap(PHY_RCC, 4);
    if (rcc == NULL)
    {
        printk("RCC地址映射表失败\n");
        return -EFAULT;
    }

    // 寄存器初始化
    (*rcc) |= (0x3 << 4);

    gpioe->MODER &= (~(0x3 << 20));
    gpioe->MODER |= (0x1 << 20);
    gpioe->ODR &= (~(0x1 << 10));

    gpioe->MODER &= (~(0x3 << 16));
    gpioe->MODER |= (0x1 << 16);
    gpioe->ODR &= (~(0x1 << 8));

    gpiof->MODER &= (~(0x3 << 20));
    gpiof->MODER |= (0x1 << 20);
    gpiof->ODR &= (~(0x1 << 10));

    printk("硬件寄存器初始化成功\n");
    return 0;
}

static int __init mycdev_init(void)
{
    int ret, i;
    // 1.分配字符设备驱动对象空间 cdev_alloc
    cdev = cdev_alloc();
    if (cdev == NULL)
    {
        printk("申请字符设备驱动对象空间失败\n");
        ret = -EFAULT;
        goto out1;
    }
    printk("字符设备驱动对象申请成功\n");
    // 2.字符设备驱动对象部分初始化 cdev_init
    cdev_init(cdev, &fops);
    // 3.申请设备号 register_chrdev_region/alloc_chrdev_region
    if (major > 0) // 静态申请设备号
    {
        ret = register_chrdev_region(MKDEV(major, minor), 3, "led");
        if (ret)
        {
            printk("静态指定设备号失败\n");
            goto out2;
        }
    }
    else
    {
        ret = alloc_chrdev_region(&devno, minor, 3, "led");
        if (ret)
        {
            printk("动态申请设备号失败\n");
            goto out2;
        }
        major = MAJOR(devno);
        minor = MINOR(devno);
    }
    printk("申请设备号成功\n");
    // 4.注册字符设备驱动对象 cdev_add()
    ret = cdev_add(cdev, MKDEV(major, minor), 3);
    if (ret)
    {
        printk("注册字符设备驱动对象失败\n");
        goto out3;
    }
    printk("注册字符设备驱动对象成功\n");
    // 5.向上提交目录
    cls = class_create(THIS_MODULE, "led");
    if (IS_ERR(cls))
    {
        printk("向上提交目录失败\n");
        ret = -PTR_ERR(cls);
        goto out4;
    }
    printk("向上提交目录成功\n");
    // 6.向上提交设备节点
    for (i = 0; i < 3; i++)
    {
        dev = device_create(cls, NULL, MKDEV(major, i), NULL, "led%d", i);
        if (IS_ERR(dev))
        {
            printk("向上提交节点信息失败\n");
            ret = PTR_ERR(dev);
            goto out5;
        }
    }
    printk("向上提交设备节点信息成功\n");
    all_led_init();
    return 0;
out5:
    for (--i; i >= 0; i--)
    {
        device_destroy(cls, MKDEV(major, i));
    }
    class_destroy(cls);
out4:
    cdev_del(cdev);
out3:
    unregister_chrdev_region(MKDEV(major, minor), 3);
out2:
    kfree(cdev);
out1:
    return ret;
}

static void __exit mycdev_exit(void)
{
    int i;
    // 1.销毁设备信息 device_destroy
    for (i = 0; i < 3; i++)
    {
        device_destroy(cls, MKDEV(major, i));
    }
    // 2.销毁目录 class_destroy
    class_destroy(cls);
    // 3.注销对象 cdev_del()
    cdev_del(cdev);
    // 4.释放设备号 unregister_chrdev_region()
    unregister_chrdev_region(MKDEV(major, minor), 3);
    // 5.释放对象空间 kfree()
    kfree(cdev);
}

module_init(mycdev_init);
module_exit(mycdev_exit);
MODULE_LICENSE("GPL");