21111 4.19作业

/************************************************************************************
* 文 件 名：head.h
* 创 建 者：黄 冠 博
* 版    本：V1.0
* 创建日期：2023-4-19
* 文件说明：在应用层控制3只LED的亮灭,蜂鸣器，风扇，马达
*************************************************************************************/
/************************************************************************************

                                       头函数                                          

*************************************************************************************/
#ifndef __HEAD_H__
#define __HEAD_H__

typedef struct{
    volatile unsigned int MODER;
    volatile unsigned int OTYPER;
    volatile unsigned int OSPEEDR;
    volatile unsigned int PUPDR;
    volatile unsigned int IDR;
    volatile unsigned int ODR;
}gpio_t;


#define PHY_LED1_ADDR 0x50006000//LED1
#define PHY_LED2_ADDR 0x50007000//LED2
#define PHY_LED3_ADDR 0x50006000//LED3
#define PHY_RCC_ADDR  0x50000A28//RCC
#define PHY_FAN_ADDR 0x50006000//风扇
#define PHY_BEEF_ADDR 0x50003000//蜂鸣器
#define PHY_MOTOR_ADDR 0x50007000//马达

#endif

/************************************************************************************
* 文 件 名：LED.c
* 创 建 者：黄 冠 博
* 版    本：V1.0
* 创建日期：2023-4-19
* 文件说明：在应用层控制3只LED的亮灭,蜂鸣器，风扇，马达
*************************************************************************************/
/************************************************************************************

                                       应用程序                                          

*************************************************************************************/
#include <stdio.h>
#include <stdlib.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <unistd.h>
#include <string.h>

int main(int argc,char const *argv[])
{
    char buf[128]={0};
    int fd=open("/dev/mychrdev",O_RDWR);
    if(fd<0)
    {
        printf("打开设备文件失败\n");
        exit(-1);
    }
    while(1)
    {
        printf("第一个数字 1 2 3代表第几只LED,4:风扇,5:蜂鸣器,6:马达\n");
        printf("第二个数字 1:LED亮 0:LED灭\n");
        printf("请输入两个数字：");
        //从终端读取
        fgets(buf,sizeof(buf),stdin);
        buf[strlen(buf)-1]='\0';
        //向设备文件中写
        write(fd,buf,sizeof(buf));
        //清空buf
        memset(buf,0,sizeof(buf));
    }

    close(fd);

    return 0;
}

/************************************************************************************
* 文 件 名：mychrdev.c
* 创 建 者：黄 冠 博
* 版    本：V1.0
* 创建日期：2023-4-19
* 文件说明：在应用层控制3只LED的亮灭,蜂鸣器，风扇，马达
*************************************************************************************/
/************************************************************************************

                                       驱动程序                                          

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


int major;
char kbuf[128]={0};
gpio_t *vir_led1;
gpio_t *vir_led2;
gpio_t *vir_led3;
gpio_t *vir_fan;
gpio_t *vir_beef;
gpio_t *vir_motor;

unsigned int *vir_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;
}
ssize_t mycdev_read(struct file *file,char *ubuf,size_t size,loff_t *lof)
{ 
    unsigned long ret;
    printk("%s:%s:%d\n",__FILE__,__func__,__LINE__);
   
    //向用户空间传送拷贝
    if(size>sizeof(kbuf))
        size=sizeof(kbuf);
    ret=copy_to_user(ubuf,kbuf,size);
    if(ret)//拷贝失败
    {
        printk("copy_to_user filed\n");
        return ret;
    }
    return 0;
}
ssize_t mycdev_write(struct file *file,const char *ubuf,size_t size,loff_t *lof)
{
    unsigned long ret;
    //从用户空间拷贝数据
    if(size>sizeof(kbuf))
        size=sizeof(kbuf);
    ret=copy_from_user(kbuf,ubuf,size);
    if(ret)
    {
        printk("copy_from_user filed\n");
        return ret;
    }
    switch(kbuf[0]){
        case '1'://LED1
            if(kbuf[1]=='0')//关灯
                vir_led1->ODR &=(~(1<<10));
            else//开灯
                vir_led1->ODR |= 1<<10;
        break;
        case '2'://LED2
            if(kbuf[1]=='0')//关灯
                vir_led2->ODR &=(~(1<<10));
            else//开灯
                vir_led2->ODR |= 1<<10;
        break;
        case '3'://LED3
            if(kbuf[1]=='0')//关灯
                vir_led3->ODR &=(~(1<<8));
            else//开灯
                vir_led3->ODR |= 1<<8;           
        break;
        case '4'://FAN
            if(kbuf[1]=='0')//关
                vir_fan->ODR &=(~(1<<9));
            else//开
                vir_fan->ODR |= 1<<9;
        break;
        case '5'://BEEF
            if(kbuf[1]=='0')//关
                vir_beef->ODR &=(~(1<<6));
            else//开
                vir_beef->ODR |= 1<<6;
        break;
        case '6'://MOTOR
            if(kbuf[1]=='0')//关
                vir_motor->ODR &=(~(1<<6));
            else//开
                vir_motor->ODR |= 1<<6;
        break;
        default:
        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,
    .read=mycdev_read,
    .write=mycdev_write,
    .release=mycdev_close,
};

//LED初始化
int all_led_init(void)
{
    //寄存器地址映射
    vir_led1=ioremap(PHY_LED1_ADDR,sizeof(gpio_t));
    if(vir_led1==NULL)
    {
        printk("ioremap failed:%d\n",__LINE__);
        return -ENOMEM;
    }
    vir_led2=ioremap(PHY_LED2_ADDR,sizeof(gpio_t));
    if(vir_led2==NULL)
    {
        printk("ioremap failed:%d\n",__LINE__);
        return -ENOMEM;
    }
    vir_led3=vir_led1;
    vir_beef=ioremap(PHY_BEEF_ADDR,sizeof(gpio_t));
    if(vir_beef==NULL)
    {
        printk("ioremap failed:%d\n",__LINE__);
        return -ENOMEM;
    }
    vir_motor=vir_led2;
    vir_fan=vir_led1;
    vir_rcc=ioremap(PHY_RCC_ADDR,4);  
    if(vir_rcc==NULL)
    {
        printk("ioremap failed:%d\n",__LINE__);
        return -ENOMEM;
    }
    printk("物理地址映射成功\n");

    //寄存器初始化
    //rcc
    (*vir_rcc) |= (1<<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<<18));
    //fan
    vir_fan->MODER &= (~(3 << 18));
    vir_fan->MODER |= (1 << 18);
    vir_fan->ODR &= (~(1 << 9));
    //beef
    vir_beep->MODER &= (~(3 << 12));
    vir_beep->MODER |= (1 << 12);
    vir_beep->ODR &= (~(1 << 6));
    //motor
    vir_motor->MODER &= (~(3 << 12));
    vir_motor->MODER |= (1 << 12);
    vir_motor->ODR &= (~(1 << 6));
    printk("寄存器初始化成功\n");

    return 0;

}
//出入口函数
//入口函数,内核模块安装时执行
static int __init mycdev_init(void)
{
    //字符设备驱动注册
    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);
    }
    printk("向上提交目录成功\n");
    //向上提交设备信息
    dev=device_create(cls,NULL,MKDEV(major,0),NULL,"mychrdev");
    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);
    //销毁设备信息
    device_destroy(cls,MKDEV(major,0));
    //销毁目录
    class_destroy(cls);
    //注销字符设备驱动
    unregister_chrdev(major,"mychrdev");
}
//声明入口、出口函数
module_init(mycdev_init);
module_exit(mycdev_exit);
//声明遵循GPL协议
MODULE_LICENSE("GPL");