2022.11.6

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

char buf[128] = {0};

int main(int argc, char const *argv[])
{
    int fd = -1;
    fd = open("/dev/myled", O_RDWR);
    if (-1 == fd)
    {
        perror("open is error\n");
        exit(1);
    }

    while (1)
    {
        /*      printf("请输入'1','2','3'\n");
                fgets(buf, sizeof(buf), stdin);
                buf[strlen(buf) - 1] = '\0';*/
        buf[0] = '1';
        write(fd, buf, sizeof(buf));
        sleep(1);
        buf[0] = '0';
        write(fd, buf, sizeof(buf));
        sleep(1);

        buf[0] = '2';
        write(fd, buf, sizeof(buf));
        sleep(1);
        buf[0] = '0';
        write(fd, buf, sizeof(buf));
        sleep(1);

        buf[0] = '3';
        write(fd, buf, sizeof(buf));
        sleep(1);
        buf[0] = '0';
        write(fd, buf, sizeof(buf));
        sleep(1);
    }
    read(fd, buf, sizeof(buf));
    close(fd);

    return 0;
}

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

#define CNAME "myled"

int major;
char kbuf[128] = {0};

gpio_t *vir_led1;
gpio_t *vir_led2;
gpio_t *vir_led3;
volatile unsigned int *vir_rcc;

int leds_init(void)
{
    vir_rcc = ioremap(PHY_RCC_ADDR, 4);
    if (vir_rcc == NULL)
    {
        printk("vir_rcc faild\n");
        return -ENOMEM;
    }

    vir_led1 = ioremap(PHY_LED1_ADDR, sizeof(gpio_t));
    if (vir_led1 == NULL)
    {
        printk("vir_led1 faild\n");
        return -ENOMEM;
    }

    vir_led2 = ioremap(PHY_LED2_ADDR, sizeof(gpio_t));
    if (vir_led2 == NULL)
    {
        printk("vir_led2 faild\n");
        return -ENOMEM;
    }

    vir_led3 = ioremap(PHY_LED3_ADDR, sizeof(gpio_t));
    if (vir_led3 == NULL)
    {
        printk("vir_led3 faild\n");
        return -ENOMEM;
    }

    //初始化寄存器部分

    // rcc
    *vir_rcc |= (0x3 << 4);

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

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

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

    return 0;
}

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 __user *ubuf, size_t size, loff_t *loff)
{
    int 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("data copy failed\n");
        return -EIO;
    }

    return 0;
}
ssize_t mycdev_write(struct file *file, const char __user *ubuf, size_t size, loff_t *loff)
{
    int ret;
    printk("%s:%s:%d\n", __FILE__, __func__, __LINE__);
    if (size > sizeof(kbuf))
        size = sizeof(kbuf);
    ret = copy_from_user(kbuf, ubuf, size);
    if (ret)
    {
        printk("copy from user is error\n");
        return -EIO;
    }
    printk("copy from user kbuf = %s\n", kbuf);

    if (kbuf[0] == '1')
        vir_led1->ODR |= (0x1 << 10);
    else if (kbuf[0] == '2')
        vir_led2->ODR |= (0x1 << 10);
    else if (kbuf[0] == '3')
        vir_led3->ODR |= (0x1 << 8);
    else if (kbuf[0] == '0')
    {
        vir_led1->ODR &= ~(0x1 << 10);
        vir_led2->ODR &= ~(0x1 << 10);
        vir_led3->ODR &= ~(0x1 << 8);
    }
    return size;
}
int mycdev_close(struct inode *inode, struct file *file)
{
    printk("%s:%s:%d\n", __FILE__, __func__, __LINE__);
    return 0;
}

const struct file_operations fops =
    {
        .open = mycdev_open,
        .read = mycdev_read,
        .write = mycdev_write,
        .release = mycdev_close,
};

//入口
static int __init mycdev_init(void)
{
    // 1.注册字符设备驱动
    major = register_chrdev(0, CNAME, &fops);
    if (major < 0)
    {
        printk("error\n");
        return major;
    }

    printk("major=%d\n", major);

    leds_init();

    return 0;
}

//出口
static void __exit mycdev_exit(void)
{
    //取消映射
    iounmap(vir_led1);
    iounmap(vir_led2);
    iounmap(vir_led3);
    iounmap(vir_rcc);

    // 2.注销字符设备驱动
    unregister_chrdev(major, CNAME);
}

module_init(mycdev_init);
module_exit(mycdev_exit);

MODULE_LICENSE("GPL");