基于stm32平台控制led、风扇、震动马达、蜂鸣器启停的案例

//驱动层文件mychrdev.c
#include 
#include 
#include 
#include 
#include 
#include "head.h"

int major;
char kbuf[128] = {0};
//映射至虚拟空间的GPIO
gpio_t *vir_led1;   //LED1
gpio_t *vir_led2;   //LED2
gpio_t *vir_led3;   //LED3
gpio_t *vir_beep;   //蜂鸣器
gpio_t *vir_fan;    //风扇
gpio_t *vir_moter;  //马达
unsigned int *vir_rcc;  //RCC
//
struct class *cls;
struct device *dev;
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)
{
    switch (cmd)
    {
    case _ON:
        switch (arg)
        {
        case 1:                       
            vir_led1->ODR |= (0x1 << 10);   // LED1开灯
            break;
        case 2:                           
            vir_led2->ODR |= (0x1 << 10);   // LED2开灯
            break;
        case 3:                         
            vir_led3->ODR |= (0x1 << 8);    // LED3开灯
            break;
        case 4:                          
            vir_moter->ODR |= (0x1 << 6);   // 马达开
            break;
        case 5:                        
            vir_fan->ODR |= (0x1 << 9);     // 风扇开
            break;
        case 6:                             
            vir_beep->ODR |= (0x1 << 6);    // 蜂鸣器开
            break;
        }

        break;
    case _OFF:
        switch (arg)
        {
        case 1:
            vir_led1->ODR &= (~(0X1 << 10));    //LED1关
            break;
        case 2:
            vir_led2->ODR &= (~(0X1 << 10));    //LED2关
            break;
        case 3:
            vir_led3->ODR &= (~(0X1 << 8));     //LED3关
            break;
        case 4:                              
            vir_moter->ODR &= (~(0X1 << 6));    //马达关
            break;
        case 5:                            
            vir_fan->ODR &= (~(0X1 << 9));      //风扇关
            break;
        case 6:                            
            vir_beep->ODR &= (~(0X1 << 6));     //蜂鸣器关
            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)
{
    // 寄存器地址的映射
    //GPIOB
    vir_beep = ioremap(GPIOB, sizeof(gpio_t));
    if (vir_beep == NULL)
    {
        printk("ioremap filed:%d\n", __LINE__);
        return -ENOMEM;
    }
    //GPIOE
    vir_led1 = ioremap(GPIOE, sizeof(gpio_t));
    if (vir_led1 == NULL)
    {
        printk("ioremap filed:%d\n", __LINE__);
        return -ENOMEM;
    }
    //GPIOF
    vir_led2 = ioremap(GPIOF, sizeof(gpio_t));
    if (vir_led2 == NULL)
    {
        printk("ioremap filed:%d\n", __LINE__);
        return -ENOMEM;
    }
    // GPIOE
    vir_led3 = vir_led1;
    vir_fan = vir_led1;
    // GPIOF
    vir_moter = vir_led2;
    vir_rcc = ioremap(PHY_RCC_ADDR, 4);
    if (vir_rcc == NULL)
    {
        printk("ioremap filed:%d\n", __LINE__);
        return -ENOMEM;
    }
    printk("w物理地址映射成功\n");
   
    // 寄存器的初始化
    // rcc
    // 使能GPIOEF
    (*vir_rcc) |= (3 << 4);
    // 使能GPIOB
    (*vir_rcc) |= (1 << 1);
    // 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_led1->MODER |= (1 << 16);
    vir_led1->ODR &= (~(1 << 8));
    // moter
    vir_moter->MODER &= (~(3 << 12));
    vir_moter->MODER |= (1 << 12);
    vir_moter->ODR &= (~(1 << 6));
    // fan
    vir_fan->MODER &= (~(3 << 18));
    vir_fan->MODER |= (1 << 18);
    vir_fan->ODR &= (~(1 << 9));
    // beep
    vir_beep->MODER &= (~(3 << 12));
    vir_beep->MODER |= (1 << 12);
    vir_beep->ODR &= (~(1 << 6));
    printk("w寄存器初始化成功\n");
    return 0;
}

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, "myled");
    if (IS_ERR(cls))
    {
        printk("向上提交目录失败\n");
        return -PTR_ERR(cls);
    }
    printk("向上提交目录信息成功\n");
    // 向上提交设备节点信息
    for (i = 0; i < 3; i++)
    {
        dev = device_create(cls, NULL, MKDEV(major, i), NULL, "myled%d", i);
        if (IS_ERR(dev))
        {
            printk("向上提交设备节点信息失败\n");
            return -PTR_ERR(dev);
        }
    }
    printk("向上提交设备节点成功\n");
    // 寄存器映射以及初始化
    all_led_init();

    return 0;
}
static void __exit mycdev_exit(void)
{
    // 取消地址映射
    iounmap(vir_led1);
    iounmap(vir_led2);
    iounmap(vir_rcc);
    // 销毁节点信息
    int i;
    for (i = 0; i < 3; i++)
    {
        device_destroy(cls, MKDEV(major, i));
    }
    // 销毁目录信息
    class_destroy(cls);
    // 注销字符设备驱动
    unregister_chrdev(major, "mychrdev");
}
module_init(mycdev_init);
module_exit(mycdev_exit);
MODULE_LICENSE("GPL");

//应用层文件test.c
#include
#include
#include 
#include 
#include 
#include
#include
#include
#include"head.h"
 
int main(int argc, char const *argv[])
{
    int a,b;
    char buf[128]={0};
    int fd=open("/dev/myled0",O_RDWR);
  if(fd<0)
    {
        printf("打开设备文件失败\n");
        exit(-1);
    }
    while(1)
    {
        //从终端读取
        printf("请输入指令\n");
        printf("0(关) 1(开)\n");
        printf("请输入>");
        scanf("%d",&a);
        printf("请输入要控制的灯 1(LED1) 2(LED2) 3(LED3)\n");
        printf("请输入要控制的设备 4(马达) 5(风扇) 6(蜂鸣器)>");
        scanf("%d",&b);
        switch(a)
        {
            case 1:
                ioctl(fd,_ON,b);//开
                break;
            case 0:
                ioctl(fd,_OFF,b);//关
                break;
        }
        
    }
    close(fd);
    return 0;
}  

//head.h
#ifndef __HEAD_H__
#define __HEAD_H__ 
#include "stm32mp1xx_gpio.h"
#define PHY_RCC_ADDR    0X50000A28
//功能码
  #define _ON _IOW('l',1,int)
  #define _OFF _IOW('l',0,int)
#endif 

你可能感兴趣的:(arm开发)