mydev.c
#include <linux/init.h>
#include <linux/module.h>
#include <linux/fs.h>
#include <linux/uaccess.h>
#include <linux/io.h>
#include "head.h"
#define CNAME "mycdev"
int major;
char kbuf[128] = {0};
unsigned int *vir_led1_moder;
unsigned int *vir_led1_odr;
unsigned int *vir_led1_rcc;
unsigned int *vir_led2_moder;
unsigned int *vir_led2_odr;
unsigned int *vir_led2_rcc;
unsigned int *vir_led3_moder;
unsigned int *vir_led3_odr;
unsigned int *vir_led3_rcc;
int mycdev_open(struct inode *inode, struct file *file)
{
printk("%s:%s:%d\n", __FILE__, __func__, __LINE__);
return 0;
}
int mycdev_close(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;
if (size > sizeof(kbuf))
size = sizeof(kbuf);
ret = copy_to_user(ubuf, kbuf, size);
if (ret)
{
printk("数据从内核向用户拷贝失败\n");
return ret;
}
return size;
}
ssize_t mycdev_write(struct file *file, const char __user *ubuf, size_t size, loff_t *loff)
{
int ret;
if (size > sizeof(kbuf))
size = sizeof(kbuf);
ret = copy_from_user(kbuf, ubuf, size);
if (ret)
{
printk("数据从内核向用户拷贝失败\n");
return ret;
}
if (kbuf[0] == '1') //开灯
(*vir_led1_odr) |= (1 << 10);
else if (kbuf[0] == '2')
(*vir_led2_odr) |= (1 << 10);
else if (kbuf[0] == '3')
(*vir_led3_odr) |= (1 << 8);
else if (kbuf[0] == '0')
{
(*vir_led1_odr) &= ~(1 << 10);
(*vir_led2_odr) &= ~(1 << 10);
(*vir_led3_odr) &= ~(1 << 8);
}
return size;
}
struct file_operations fops = {
.open = mycdev_open,
.read = mycdev_read,
.write = mycdev_write,
.release = mycdev_close,
};
static int __init demo_init(void)
{
major = register_chrdev(0, CNAME, &fops); //驱动注册
if (major < 0)
{
printk("字符串设备驱动注册失败\n");
return major;
}
printk("字符设备驱动注册成功major=%d\n", major);
/***LED1****/
vir_led1_moder = ioremap(PHY_LED1_MODE, 4);
if (vir_led1_moder == NULL)
{
printk("vir_led1_moder 映射失败\n");
return -ENOMEM;
}
printk("vir_led1_moder 映射成功\n");
vir_led1_odr = ioremap(PHY_LED1_ODR, 4);
if (vir_led1_odr == NULL)
{
printk("vir_led1_odr 映射失败\n");
return -ENOMEM;
}
printk("vir_led1_odr 映射成功\n");
vir_led1_rcc = ioremap(PHY_LED1_RCC, 4);
if (vir_led1_rcc == NULL)
{
printk("vir_led1_rcc 映射失败\n");
return -ENOMEM;
}
printk("vir_led1_rcc 映射成功\n");
//寄存器初始化
(*vir_led1_moder) &= ~(3 << 20);
(*vir_led1_moder) |= (1 << 20);
(*vir_led1_odr) &= ~(1 << 10);
(*vir_led1_rcc) |= (1 << 4);
/***LED2****/
vir_led2_moder = ioremap(PHY_LED2_MODE, 4);
if (vir_led2_moder == NULL)
{
printk("vir_led2_moder 映射失败\n");
return -ENOMEM;
}
printk("vir_led2_moder 映射成功\n");
vir_led2_odr = ioremap(PHY_LED2_ODR, 4);
if (vir_led2_odr == NULL)
{
printk("vir_led2_odr 映射失败\n");
return -ENOMEM;
}
printk("vir_led2_odr 映射成功\n");
vir_led2_rcc = ioremap(PHY_LED2_RCC, 4);
if (vir_led2_rcc == NULL)
{
printk("vir_led2_rcc 映射失败\n");
return -ENOMEM;
}
printk("vir_led2_rcc 映射成功\n");
//寄存器初始化
(*vir_led2_moder) &= ~(3 << 20);
(*vir_led2_moder) |= (1 << 20);
(*vir_led2_odr) &= ~(1 << 10);
(*vir_led2_rcc) |= (1 << 5);
/***LED3****/
vir_led3_moder = ioremap(PHY_LED3_MODE, 4);
if (vir_led3_moder == NULL)
{
printk("vir_led3_moder 映射失败\n");
return -ENOMEM;
}
printk("vir_led3_moder 映射成功\n");
vir_led3_odr = ioremap(PHY_LED3_ODR, 4);
if (vir_led3_odr == NULL)
{
printk("vir_led3_odr 映射失败\n");
return -ENOMEM;
}
printk("vir_led3_odr 映射成功\n");
vir_led3_rcc = ioremap(PHY_LED3_RCC, 4);
if (vir_led3_rcc == NULL)
{
printk("vir_led3_rcc 映射失败\n");
return -ENOMEM;
}
printk("vir_led3_rcc 映射成功\n");
//寄存器初始化
(*vir_led3_moder) &= ~(3 << 16);
(*vir_led3_moder) |= (1 << 16);
(*vir_led3_odr) &= ~(1 << 8);
(*vir_led3_rcc) |= (1 << 4);
return 0;
}
static void __exit demo_exit(void)
{
unregister_chrdev(major, CNAME);
}
module_init(demo_init);
module_exit(demo_exit);
MODULE_LICENSE("GPL");
head.h
#ifndef __HEAD_H__
#define __HEAD_H__
#define PHY_LED1_MODE 0x50006000
#define PHY_LED1_ODR 0x50006014
#define PHY_LED1_RCC 0x50000A28
#define PHY_LED2_MODE 0x50007000
#define PHY_LED2_ODR 0x50007014
#define PHY_LED2_RCC 0x50000A28
#define PHY_LED3_MODE 0x50006000
#define PHY_LED3_ODR 0x50006014
#define PHY_LED3_RCC 0x50000A28
#endif
demo.c
#include <stdio.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <unistd.h>
#include <stdlib.h>
#include <string.h>
int main(int argc, char const *argv[])
{
char buf[128] = "";
int fd = open("/dev/mychdev", O_RDWR);
if (fd < 0)
{
printf("打开设备文件失败\n");
exit(-1);
}
printf("打开设备文件成功\n");
while (1)
{
printf("请输入命令 1:开启LED1\t2:开启LED2\t3:开启LED3\t0:关灯\n");
fgets(buf, sizeof(buf), stdin);
if (buf[0] != '1' && buf[0] != '2' && buf[0] != '3' && buf[0] != '0')
{
printf("输入错误\n");
continue;
}
buf[strlen(buf) - 1] = '\0';
write(fd, buf, sizeof(buf));
memset(buf, 0, sizeof(buf));
}
close(fd);
return 0;
}