3.16作业---用ioctl传递字符串

代码

mycdev.c

#include <linux/init.h>
#include <linux/module.h>
#include <linux/fs.h>
#include <linux/uaccess.h>
#include <linux/io.h>
#include "myled.h"
#include <linux/device.h>
 
#define CNAME "myled"
 
int major;
char kbuf[128] = {0};
gpio_t *virt_gpioe = NULL;
gpio_t *virt_gpiof = NULL;
unsigned int *virt_rcc = NULL;
#define LED1_ON (virt_gpioe->ODR |= (0x1 << 10))
#define LED1_OFF (virt_gpioe->ODR &= (~(0x1 << 10)))
#define LED2_ON (virt_gpiof->ODR |= (0x1 << 10))
#define LED2_OFF (virt_gpiof->ODR &= (~(0x1 << 10)))
#define LED3_ON (virt_gpioe->ODR |= (0x1 << 8))
#define LED3_OFF (virt_gpioe->ODR &= (~(0x1 << 8)))
 
struct class *cls;
struct device *dev;
 

 
int myled_open(struct inode *inode, struct file *file)
{
    printk("%s:%s:%d\n", __FILE__, __func__, __LINE__);
    return 0;
}
ssize_t myled_read(struct file *file, char __user *ubuf, size_t size, loff_t *loff)
{
    int ret;
    printk("%s:%s:%d\n", __FILE__, __func__, __LINE__);
    // 1.校验传输数据的大小，如果用户空间写的数据比内核空间数据大小大，需要更正大小
    if (size > sizeof(kbuf))
        size = sizeof(kbuf);
    // 2.将数据从用户空间拷贝到内核空间
    ret = copy_to_user(ubuf, kbuf, size);
    if (ret) // 3.判断是否错误
    {
        printk("copy to user is error\n");
        return -EIO;
    }
    return size; // 5.返回拷贝数据大小
}
ssize_t myled_write(struct file *file, const char __user *ubuf, size_t size, loff_t *loff)
{
    int ret;
    printk("%s:%s:%d\n", __FILE__, __func__, __LINE__);
    // 1.校验传输数据的大小，如果用户空间写的数据比内核空间数据大小大，需要更正大小
    if (size > sizeof(kbuf))
        size = sizeof(kbuf);
    // 2.将数据从用户空间拷贝到内核空间
    ret = copy_from_user(kbuf, ubuf, size);
    if (ret) // 3.判断是否错误
    {
        printk("copy from user is error\n");
        return -EIO;
    }
    // 4.打印传递数据内容
    printk("copy from user kbuf:%s\n", kbuf);
    // kbuf[0]:代表操作的是那一盏灯，kbuf[0] = 0  kbuf[0] = 1 kbuf[0] = 2
    // kbuf[1]:代表led灯的状态   kbuf[1] = 0  kbuf[1] = 1
    switch (kbuf[0])
    {
    case 0:
        kbuf[1] ? LED1_ON : LED1_OFF;
        break;
    case 1:
        kbuf[1] ? LED2_ON : LED2_OFF;
        break;
    case 2:
        kbuf[1] ? LED3_ON : LED3_OFF;
        break;
    }
    return size; // 5.返回拷贝数据大小
}
int myled_close(struct inode *inode, struct file *file)
{
    printk("%s:%s:%d\n", __FILE__, __func__, __LINE__);
    return 0;
}
long myled_ioctl(struct file *file, unsigned int cmd, unsigned long args)
{
    int whitch;
    int ret;
    char buf[128];
    image_t image;
    switch (cmd)
    {
    case LED_ON:
        ret = copy_from_user(&whitch, (void *)args, sizeof(int));
        if (ret)
        {
            printk("copy from user is error\n");
            return -EIO;
        }
        switch (whitch)
        {
        case LED1:
            LED1_ON;
            break;
        case LED2:
            LED2_ON;
            break;
        case LED3:
            LED3_ON;
            break;
        }
        break;
    case LED_OFF:
        ret = copy_from_user(&whitch, (void *)args, sizeof(int));
        if (ret)
        {
            printk("copy from user is error\n");
            return -EIO;
        }
        switch (whitch)
        {
        case LED1:
            LED1_OFF;
            break;
        case LED2:
            LED2_OFF;
            break;
        case LED3:
            LED3_OFF;
            break;
        }
        break;
    case UACCESS_BUF:
        ret = copy_from_user(buf, (void *)args, sizeof(buf));
        if (ret)
        {
            printk("copy from user is error\n");
            return -EIO;
        }
        printk("input:%s\n", buf);
        break;
    case UACCESS_STRUCT:
        ret = copy_from_user(&image, (void *)args, sizeof(image_t));
        if (ret)
        {
            printk("copy from user is error\n");
            return -EIO;
        }
        printk("内核空间: width=%d  high=%d\n",image.width,image.high);
 
        image.width+=10;
        image.high+=10;
        
        ret = copy_to_user((void *)args,&image,sizeof(image_t));
        if (ret) // 3.判断是否错误
        {
            printk("copy to user is error\n");
            return -EIO;
        }
        break;
    }
    return 0;
}
const struct file_operations fops = {
    .open = myled_open,
    .read = myled_read,
    .write = myled_write,
    .unlocked_ioctl = myled_ioctl,
    .release = myled_close,
};
static int __init mycdev_init(void)
{
    // 1.注册字符设备驱动
    major = register_chrdev(0, CNAME, &fops);
    if (major < 0) // 2.判断返回值
    {
        printk("register chrdev is error\n");
    }
    // 3.打印主设备号
    printk("register chrdev major=%d\n", major);
 
    // 4.将物理地址映射为虚拟地址
    // 4.1 将rcc地址映射
    virt_rcc = ioremap(PHY_RCC_LED1, 4);
    if (virt_rcc == NULL)
    {
        printk("rcc ioremap is error\n");
        return -ENOMEM;
    }
    // 映射GPIOE地址
    virt_gpioe = ioremap(PHY_GPIOE_ADDR, sizeof(gpio_t));
    if (virt_gpioe == NULL)
    {
        printk("virt_gpioe ioremap is error\n");
        return -ENOMEM;
    }
    // 映射GPIOF地址
    virt_gpiof = ioremap(PHY_GPIOF_ADDR, sizeof(gpio_t));
    if (virt_gpiof == NULL)
    {
        printk("virt_gpiof ioremap is error\n");
        return -ENOMEM;
    }
 
    // 5.对led1---->PE10引脚初始化  对led3---->PE8引脚初始化
    *virt_rcc |= (0x1 << 4);             // 5.1 使能GPIOE组时钟[4]=1
    virt_gpioe->MODER &= (~(0x3 << 20)); // 5.2 设置PE10引脚为输出模式 [21:20] = 01
    virt_gpioe->MODER |= (0x1 << 20);
    virt_gpioe->ODR &= (~(0x1 << 10)); // 5.3 设置PE10引脚输出低电平
 
    // 6. 对led2---->PF10引脚初始化
    *virt_rcc |= (0x1 << 5);             // 5.1 使能GPIOF组时钟[5]=1
    virt_gpiof->MODER &= (~(0x3 << 20)); // 5.2 设置PF10引脚为输出模式 [21:20] = 01
    virt_gpiof->MODER |= (0x1 << 20);
    virt_gpiof->ODR &= (~(0x1 << 10)); // 5.3 设置PF10引脚输出低电平
 
    // 7.对led3---->PE8引脚初始化
    virt_gpioe->MODER &= (~(0x3 << 16)); // 5.2 设置PE8引脚为输出模式 [17:16] = 01
    virt_gpioe->MODER |= (0x1 << 16);
    virt_gpioe->ODR &= (~(0x1 << 8)); // 5.3 设置PE8引脚输出低电平
 
    cls = class_create(THIS_MODULE, "myled");
    if (IS_ERR(cls))
    {
        PTR_ERR(cls);
    }
    dev = device_create(cls, NULL, MKDEV(major, 0), NULL, "myled");
    if (IS_ERR(dev))
    {
        PTR_ERR(dev);
    }
    return 0;
}
 
static void __exit mycdev_exit(void)
{
    device_destroy(cls, MKDEV(major, 0));
    class_destroy(cls);
    // 2.取消地址映射
    iounmap(virt_rcc);
    iounmap(virt_gpioe);
    iounmap(virt_gpiof);
    // 1.注销字符设备驱动
    unregister_chrdev(major, CNAME);
}
module_init(mycdev_init);
module_exit(mycdev_exit);
MODULE_LICENSE("GPL");

myled.h

#ifndef __MYLED_H__
#define __MYLED_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;
 
enum{
    LED1,
    LED2,
    LED3
};
typedef struct 
{
    int width;
    int high;
}image_t;
//GPIOE基地址
#define PHY_GPIOE_ADDR 0x50006000
#define PHY_GPIOF_ADDR 0x50007000
//RCC基地址：0x50000A28
#define PHY_RCC_LED1 0x50000A28
 
#define LED_ON _IOW('a',1,int)
#define LED_OFF _IOW('a',0,int)
#define UACCESS_BUF _IOW('a',1,char[128])
#define UACCESS_STRUCT _IOW('a',1,image_t)
 
 
 
#endif
 

test.c

#include <stdio.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <unistd.h>
#include <errno.h>
#include <string.h>
#include <sys/ioctl.h>
#include "myled.h"
  
int main(int argc, const char *argv[])
{
    int whitch;
    char buf[128] = {0};
    image_t image;
    image.high=200;
    image.width=10;
    int fd = -1;
 
    fd = open("/dev/myled", O_RDWR);
    if (fd == -1)
    {
        perror("open is error\n");
        return -1;
    }
    ioctl(fd,UACCESS_STRUCT,&image);
    printf("用户空间: width=%d  high=%d\n",image.width,image.high);
 
    printf("please input------>");
    fgets(buf,sizeof(buf),stdin);
    buf[strlen(buf)-1]=0;
    ioctl(fd,UACCESS_BUF,buf);
 
    

    close(fd);
    return 0;
}