系统驱动作业

最新推荐文章于 2024-10-14 16:42:33 发布

2301_76948983

最新推荐文章于 2024-10-14 16:42:33 发布

阅读量57

点赞数

文章标签： linux arm

本文链接：https://blog.csdn.net/2301_76948983/article/details/131562791

版权

该代码实现了一个Linux字符设备驱动，通过ioctl系统调用控制GPIO端口来开关LED、蜂鸣器和风扇。驱动注册了几个设备节点，分别对应不同的硬件设备，并使用ioremap映射物理寄存器到虚拟地址进行操作。在用户空间，通过test.c程序与设备交互，实现对硬件的开启和关闭功能。

摘要由CSDN通过智能技术生成

通过ioctl函数选择不同硬件的控制，LED 蜂鸣器马达风扇

head.h

#ifndef __HEAD_H__
#define __HEAD_H__

//定义寄存器组织结构体
typedef struct {
    unsigned int MODER;
    unsigned int OTYPER;
    unsigned int OSPEEDR;
    unsigned int PUPDR;
    unsigned int IDR;
    unsigned int ODR;
}gpio_t;


//LED1->PE10
//LED2->PF10
//LED3->PE8
//蜂鸣器：TIM4_CH1 --------> PB6
//风扇：TIM1_CH1 --------> PE9
//马达：TIM16_CH1 --------> PF6

//定义灯、风扇、蜂鸣器、马达地址
#define PHY_LED1_ADDR 0x50006000
#define PHY_LED2_ADDR 0x50007000
#define PHY_LED3_ADDR 0x50006000
#define PHY_RCC_ADDR 0x50000A28
#define PHY_BEE_ADDR 0x50003000
#define PHY_FAN_ADDR 0x50006000
#define PHY_MOTOR_ADDR 0x50007000


#define ON _IOW('l',1,int)
#define OFF _IOW('l',0,int)

#endif

mychrdev.c

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

struct class *cls;
struct device *dev;

int major;
char kbuf[128] = "";

gpio_t *vir_led1;
gpio_t *vir_led2;
gpio_t *vir_led3;
gpio_t *vir_bee;
gpio_t *vir_fan;
gpio_t *vir_motor;
unsigned int *vir_rcc;

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

long mycdev_ioctl(struct file *file, unsigned int cmd, unsigned long arg)
{
    int which;
    // 通过拷贝函数获取ioctl第三个参数对应空间的数值
    int ret = copy_from_user(&which, (void *)arg, 4);
    if (ret)
    {
        printk("拷贝用户空间数据失败\n");
        return -EIO;
    }
    // 进行功能控制
    switch (cmd)
    {
    case ON: // 开
        switch (which)
        {
        case 1: // LED1
            vir_led1->ODR |= (0x1 << 10);
            break;
        case 2: // LED2
            vir_led2->ODR |= (0x1 << 10);
            break;
        case 3: // LED3
            vir_led3->ODR |= (0x1 << 8);
            break;
        case 4: // FAN
            vir_fan->ODR |= (0x1 << 9);
            break;
        case 5: // BEE
            vir_bee->ODR |= (0x1 << 6);
            break;
        case 6: // MOTOR
            vir_motor->ODR |= (0x1 << 6);
            break;
        }
        break;
    case OFF: // 关
        switch (which)
        {
        case 1: // LED1
            vir_led1->ODR &= (~(0X1 << 10));
            break;
        case 2: // LED2
            vir_led2->ODR &= (~(0X1 << 10));
            break;
        case 3: // LED3
            vir_led3->ODR &= (~(0X1 << 8));
            break;
        case 4: // FAN
            vir_fan->ODR &= (~(0x1 << 9));
            break;
        case 5: // BEE
            vir_bee->ODR &= (~(0x1 << 6));
            break;
        case 6: // MOTOR
            vir_motor->ODR &= (~(0x1 << 6));
            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,
};

static int __init mycdev_init(void)
{
    int i;

    major = register_chrdev(0, "mychrdev", &fops);
    if (major < 0)
    {
        printk("字符设备驱动注册失败\n");
        return major;
    }
    printk("字符设备驱动注册成功major=%d\n", major);

    // 向上提交目录
    cls = class_create(THIS_MODULE, "mychrdev");
    if (IS_ERR(cls))
    {
        printk("向上提交目录失败\n");
        return -PTR_ERR(cls);
    }
    // 向上提交设备节点信息
    // 为三盏灯创建三个设备文件
    for (i = 0; i < 3; i++)
    {
        dev = device_create(cls, NULL, MKDEV(major, i), NULL, "led%d", i);
        if (IS_ERR(dev))
        {
            printk("向上提交设备节点信息失败\n");
            return -PTR_ERR(dev);
        }
    }
    // 为蜂鸣器、风扇、马达创建设备文件
    dev = device_create(cls, NULL, MKDEV(major, 3), NULL, "fan");
    if (IS_ERR(dev))
    {
        printk("向上提交设备节点信息失败\n");
        return -PTR_ERR(dev);
    }
    dev = device_create(cls, NULL, MKDEV(major, 4), NULL, "bee");
    if (IS_ERR(dev))
    {
        printk("向上提交设备节点信息失败\n");
        return -PTR_ERR(dev);
    }
    dev = device_create(cls, NULL, MKDEV(major, 5), NULL, "motor");
    if (IS_ERR(dev))
    {
        printk("向上提交设备节点信息失败\n");
        return -PTR_ERR(dev);
    }

    // 映射物理寄存器

    vir_led1 = ioremap(PHY_LED1_ADDR, sizeof(gpio_t));
    if (vir_led1 == NULL)
    {
        printk("ioremap filed：%d\n", __LINE__);
        return -ENOMEM;
    }
    vir_led2 = ioremap(PHY_LED2_ADDR, sizeof(gpio_t));
    if (vir_led2 == NULL)
    {
        printk("ioremap filed：%d\n", __LINE__);
        return -ENOMEM;
    }
    vir_led3 = vir_led1;
    vir_rcc = ioremap(PHY_RCC_ADDR, 4);
    if (vir_rcc == NULL)
    {
        printk("ioremap filed：%d\n", __LINE__);
        return -ENOMEM;
    }
    vir_fan = ioremap(PHY_FAN_ADDR, sizeof(gpio_t));
    if (vir_fan == NULL)
    {
        printk("ioremap filed：%d\n", __LINE__);
        return -ENOMEM;
    }
    vir_bee = ioremap(PHY_BEE_ADDR, sizeof(gpio_t));
    if (vir_bee == NULL)
    {
        printk("ioremap filed：%d\n", __LINE__);
        return -ENOMEM;
    }
    vir_motor = ioremap(PHY_MOTOR_ADDR, sizeof(gpio_t));
    if (vir_motor == NULL)
    {
        printk("ioremap filed：%d\n", __LINE__);
        return -ENOMEM;
    }

    // 寄存器初始化
    (*vir_rcc) |= (0x1 << 1);
    (*vir_rcc) |= (0x3 << 4);
    // led1初始化
    vir_led1->MODER &= (~(3 << 20));
    vir_led1->MODER |= (1 << 20);
    vir_led1->ODR &= (~(1 << 10));
    // led2初始化
    vir_led2->MODER &= (~(3 << 20));
    vir_led2->MODER |= (1 << 20);
    vir_led2->ODR &= (~(1 << 10));
    // led3初始化
    vir_led3->MODER &= (~(3 << 16));
    vir_led3->MODER |= (1 << 16);
    vir_led3->ODR &= (~(1 << 8));
    // 风扇初始化
    vir_fan->MODER &= (~(3 << 18));
    vir_fan->MODER |= (1 << 18);
    vir_fan->ODR &= (~(1 << 9));
    // 蜂鸣器初始化
    vir_bee->MODER &= (~(3 << 12));
    vir_bee->MODER |= (1 << 12);
    vir_bee->ODR &= (~(1 << 6));
    // 马达初始化
    vir_motor->MODER &= (~(3 << 12));
    vir_motor->MODER |= (1 << 12);
    vir_motor->ODR &= (~(1 << 6));

    printk("硬件寄存器初始化成功\n");
    return 0;
}
static void __exit mycdev_exit(void)
{
    int i;
    // 取消寄存器地址映射
    iounmap(vir_led1);
    iounmap(vir_led2);
    iounmap(vir_rcc);
    iounmap(vir_fan);
    iounmap(vir_bee);
    iounmap(vir_motor);

    // 销毁设备节点信息
    for (i = 0; i < 3; i++)
    {
        device_destroy(cls, MKDEV(major, i));
    }
    device_destroy(cls, MKDEV(major, 3));
    device_destroy(cls, MKDEV(major, 4));
    device_destroy(cls, MKDEV(major, 5));
    // 销毁目录信息
    class_destroy(cls);
    // 销毁字符设备驱动
    // 字符设备驱动的注销
    unregister_chrdev(major, "mychrdev");
}
module_init(mycdev_init);
module_exit(mycdev_exit);
MODULE_LICENSE("GPL");

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"

// LED1->PE10
// LED2->PF10
// LED3->PE8
// 蜂鸣器：TIM4_CH1 --------> PB6
// 风扇：TIM1_CH1 --------> PE9
// 马达：TIM16_CH1 --------> PF6

int main(int argc, char const *argv[])
{

    int a, b;
    int fd_led = open("/dev/led0", O_RDWR);
    if (fd_led < 0)
    {
        printf("打开设备文件失败\n");
        exit(-1);
    }
    int fd_fan = open("/dev/fan", O_RDWR);
    if (fd_fan < 0)
    {
        printf("打开设备文件失败\n");
        exit(-1);
    }
    int fd_bee = open("/dev/bee", O_RDWR);
    if (fd_bee < 0)
    {
        printf("打开设备文件失败\n");
        exit(-1);
    }
    int fd_motor = open("/dev/motor", O_RDWR);
    if (fd_motor < 0)
    {
        printf("打开设备文件失败\n");
        exit(-1);
    }

    while (1)
    {
        // 从终端读取
        printf("请选择要实现的功能\n");
        printf("0(关) 1(开)>");
        scanf("%d", &a);
        printf("请输入要控制的设备\n");
        printf("1(LED1) 2(LED2) 3(LED3) 4(FAN) 5(BEE) 6(MOTOR)>");
        scanf("%d", &b);
        if (a == 1) // 开
        {
            switch (b)
            {
            case 1:
                ioctl(fd_led, ON, &b);
                break;
            case 2:
                ioctl(fd_led, ON, &b);
                break;
            case 3:
                ioctl(fd_led, ON, &b);
                break;
            case 4:
                ioctl(fd_fan, ON, &b);
                break;
            case 5:
                ioctl(fd_bee, ON, &b);
                break;
            case 6:
                ioctl(fd_motor, ON, &b);
                break;
            }
        }
        else if (a == 0) // 关
        {
            switch (b)
            {
            case 1:
                ioctl(fd_led, OFF, &b);
                break;
            case 2:
                ioctl(fd_led, OFF, &b);
                break;
            case 3:
                ioctl(fd_led, OFF, &b);
                break;
            case 4:
                ioctl(fd_fan, OFF, &b);
                break;
            case 5:
                ioctl(fd_bee, OFF, &b);
                break;
            case 6:
                ioctl(fd_motor, OFF, &b);
                break;
            }
        }
    }
    close(fd_led);
    close(fd_fan);
    close(fd_bee);
    close(fd_motor);
    return 0;
}