#include <linux/types.h>
#include <linux/kernel.h>
#include <linux/delay.h>
#include <linux/ide.h>
#include <linux/init.h>
#include <linux/module.h>
#include <linux/errno.h>
#include <linux/gpio.h>
#include <asm/mach/map.h>
#include <asm/uaccess.h>
#include <asm/io.h>

#define    LED_MAJOR   140      // 主设备号
#define    LED_NAME    "led"    // 设备名
#define    LEDOFF      0        // 关灯
#define    LEDON       1        // 开灯

/* 寄存器物理地址 */
#define    CCM_CCGR1_BASE            (0X020C406C)
#define    SW_MUX_GPIO1_IO03_BASE    (0X020E0068)
#define    SW_PAD_GPIO1_IO03_BASE    (0X020E02F4)
#define    GPIO1_DR_BASE             (0X0209C000)
#define    GPIO1_GDIR_BASE           (0X0209C004)

//映射寄存器虚拟地址指针
static void __iomem *IMX6U_CCM_CCGR1
static void __iomem *SW_MUX_GPIO1_IO03
static void __iomem *SW_PAD_GPIO1_IO03
static void __iomem *GPIO1_DR;
static void __iomem *GPIO1_GDIR;

/*
 * 控制LED
 * sta : LEDON 开 , LEDOFF 关
 */
void led_switch(u8 sta)
{
    u32 val = 0;

    if(sta == LEDON)
    {
        val = readl(GPIO1_DR);
        val &= ~(1 << 3);
    }
    else if(sta == LEDOFF)
    {
        val = readl(GPIO1_DR);
        val |= (1 << 3);
        writel(val, GPIO1_DR);
    }
}

/*
 * 打开设备
 * inode：传递驱动inode
 * filp: 设备文件
 * return:成功: 0 失败：其他
 */
static int led_open(struct inode *inode, struct file *filp)
{
    printk(" led_open\r\n");
    return 0;
}

/*
 * 从设备读数据
 * filp：打开的设备文件（文件描述符）
 * buf：返回给用户空间的数据缓冲区
 * cnt：读取的数据长度
 * offt：相对于文件首地址的偏移
 * return：成功：字节数，失败：负值
 */
static ssize_t led_read(struct file *filp,
                        char __user *buf,
                        size_t cnt, 
                        loff_t *offt)
{
    printf("led_read \r\n");
    return 0;
}

/*
 * 向设备写数据
 * filp：设备文件
 * buf：向设备写入的数据
 * cnt：写入数据长度
 * offt：相对于文件首地址的偏移
 * return：成功：写入字节数 失败：负值
 */
static ssize_t led_write(struct file *filp, 
                         const char __user *buf, 
                         size_t cnt, 
                         loff_t *offt)
{
    int retvalue;
    unsigned char databuf[1];
    unsigned char ledstat;
    
    retvalue = copy_from_user(databuf, buf, cnt);

    if(retvalue < 0)
    {
        printk("kernel write failed!\r\n");
        return -EFAULT;
    }
    
    ledstat = databuf[0];    // 状态值
    
    if(ledstat == LEDON)
    {
        led_switch(LEDON);    // 开灯
    }
    else if(ledsstat == LEDOFF)
    {
        led_switch(LEDOFF);    //关灯
    }

    return 0;
}

/*
 * 关设备
 * inode: 驱动的 inode
 * filp：要关的设备文件
 */
static int led_release(struct inode *inode, struct file *filp)
{
    printk("led_release\r\r");
    return 0;
}

/* 设备操作数 */
static struct file_operations led_fogs =
{
    .owner = THIS_MODULE,
    .open = led_open,
    .read = led_read,
    .write = led_write,
    release = led_release,
};

/*
 * 驱动入口
 * return：成功 0
 */
static int __init led_init(void)
{
    int retvalue = 0;
    u32 val = 0;
    
    // 寄存器地址映射
    IMX6U_CCM_CCGR1 = ioremap(CCM_CCGR1_BASE, 4);
    SW_MUX_GPIO1_IO03 = ioremap(SW_MUX_GPIO1_IO03_BASE, 4);
    SW_PAD_GPIO1_IO03 = ioremap(SW_PAD_GPIO1_IO03_BASE, 4);
    GPIO1_DR = ioremap(GPIO1_DR_BASE, 4);
    GPIO1_GDIR = ioremap(GPIO1_GDIR_BASE, 4);
    
    // 使能GPIO1时钟
    val = readl(IMX6U_CCM_CCGR1);
    val &= ~(3 << 26);    // 清除以前的设置
    val |= (3 << 26);    // 设置新值
    writel(val, IMX6U_CCM_CCGR1);

    // 复用GPIO1_3
    val = readl(SW_MUX_GPIO1_IO03);
    val &= ~(F << 0);    // 清除以前的设置
    val |= (5 << 0);    // 设置新值
    writel(val,SW_MUX_GPIO1_IO03);
    
    //
    /*
     * 16bit：0 禁止施密特
     * 15 14bit：00 设置100k下拉
     * 13：0 状态保存器
     * 12：1 pull/keeper使能
     * 11：0 关闭开路输出
     * 10 8：000 保留
     * 7 6：10 max(200MHZ)
     * 5 3：110 R0/6
     * 2 1：00 保留
     * 0：0 低压摆率
     */
    val = readl(SW_PAD_GPIO1_IO03);
    val &= ~(1FFFF << 0);    // 清除以前的设置
    val |= (10B0 << 0);    // 设置新值
    writel(val,SW_PAD_GPIO1_IO03);

    // 注册设备
    retvalue = register_chrdev(LED_MAJOR, LED_NAME, &led_fops);

    if(retvalue < 0)
    {
        printk("register chrdev failed \r\n");
        return -EIO;
    }

    return 0;
}

static void __exit led_exit(void)
{
    // 取消映射
    iounmem(IMX6U_CCM_CCGR1);
    iounmem(SW_MUX_GPIO1_IO03);
    iounmem(SW_PAD_GPIO1_IO03);
    iounmem(GPIO1_DR);
    iounmem(GPIO1_GDIR);

    // 注销设备
    unregister_chrdev(LED_MAJOR, LED_NAME);
}

module_init(led_init);
module_exit(led_exit);

MODULE_LICENSE("GPL");
MODULE_AUTHOR("CpuCode");