18、IMX6ULL學習筆記-linux設備樹之內核定時器之ioctl函數

一、設備驅動程序

#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 <linux/cdev.h>
#include <linux/device.h>
#include <linux/of.h>
#include <linux/of_address.h>
#include <linux/of_gpio.h>
#include <asm/mach/map.h>
#include <asm/uaccess.h>
#include <asm/io.h>
#define DeviceName 		"devicetree-timer-ioctl" //設備驅動名稱
#define DeviceNodes 	1 					  	 //設備節點個數
#define CLOSE_CMD       _IO(0XEF, 1)          	 //關閉命令               
#define OPEN_CMD        _IO(0XEF, 2)          	 //打開命令             
#define SETPERIOD_CMD   _IOW(0xEF, 3, int)    	 //設置週期
/* 定義一個設備結構體 */
struct mod_struct {
	dev_t devid;			     /* 設備號*/
	struct cdev cdev;		     /* cdev  */
	struct class *class;	  	 /* 類    */
	struct device *device;	 	 /* 設備  */
	int major;				     /* 主設備號 */
	int minor;				     /* 次設備號 */
	struct device_node	*nd;  	 /* 設備節點 */
	int gpio;					 /* GPIO編號*/
	struct timer_list timer;	 /* 定義一個定時器*/
	int timeperiod;				 /* 定時器週期*/
};
struct mod_struct mod_device;	 /* led設備 */
/*
 * @description		: 打開設備
 * @param - inode 	: 傳遞給驅動的inode
 * @param - filp 	: 設備文件，file結構體有個叫做private_data的成員變量
 * 					  一般在open的時候將private_data指向設備結構體。
 * @return 			: 0 成功;其他 失敗
 */
static int kernel_timer_open(struct inode *inode, struct file *filp)
{
	filp->private_data = &mod_device; /* 設置私有數據 */
	return 0;
}
/*
 * @description		: 從設備讀取數據 
 * @param - filp 	: 要打開的設備文件(文件描述符)
 * @param - buf 	: 返回給用戶空間的數據緩衝區
 * @param - cnt 	: 要讀取的數據長度
 * @param - offt 	: 相對於文件首地址的偏移
 * @return 			: 讀取的字節數，如果爲負值，表示讀取失敗
 */
static ssize_t kernel_timer_read(struct file *filp, char __user *buf, size_t cnt, loff_t *offt)
{
	return 0;
}
/*
 * @description		: 向設備寫數據 
 * @param - filp 	: 設備文件，表示打開的文件描述符
 * @param - buf 	: 要寫給設備寫入的數據
 * @param - cnt 	: 要寫入的數據長度
 * @param - offt 	: 相對於文件首地址的偏移
 * @return 			: 寫入的字節數，如果爲負值，表示寫入失敗
 */
static ssize_t kernel_timer_write(struct file *filp, const char __user *buf,size_t cnt, loff_t *offt)
{
	return 0;
}
/*
 * @description		: 關閉/釋放設備
 * @param - filp 	: 要關閉的設備文件(文件描述符)
 * @return 			: 0 成功;其他 失敗
 */
static int kernel_timer_release(struct inode *inode, struct file *filp)
{
	return 0;
}
static long kernel_timer_ioctl(struct file *file, unsigned int cmd, unsigned long arg)
{
    int ret = 0;
    int value = 0;
    struct mod_struct *dev = file->private_data;
	switch (cmd) {
	case CLOSE_CMD:
        del_timer_sync(&dev->timer);
		break;
	case OPEN_CMD:
        mod_timer(&dev->timer, jiffies + msecs_to_jiffies(dev->timeperiod));
		break;
	case SETPERIOD_CMD:
        ret = copy_from_user(&value, (int *)arg, sizeof(int));
        if(ret < 0) {
            return -EFAULT;
        }
        dev->timeperiod = value;
        mod_timer(&dev->timer, jiffies + msecs_to_jiffies(dev->timeperiod));
		break;
	}
	return ret;
}
/* 設備操作函數 */
static struct file_operations beep_fops = {
	.owner    	=  THIS_MODULE,
	.open     	=  kernel_timer_open,
	.read     	=  kernel_timer_read,
	.write    	=  kernel_timer_write,
	.release  	=  kernel_timer_release,
	.unlocked_ioctl  =  kernel_timer_ioctl,
};
/*
 * @description	: 定時器超時函數
 * @param 		: arg
 * @return 		: 無
 */
static void timer_func(unsigned long arg)
{
    struct mod_struct *dev = (struct mod_struct*)arg;
    static int sta = 1;
    sta = !sta;
    gpio_set_value(dev->gpio, sta);
    mod_timer(&dev->timer, jiffies + msecs_to_jiffies(dev->timeperiod));
}
/*
 * @description	: 定時器超時函數
 * @param 		: dev
 * @return 		: 無
 */
int led_init(struct mod_struct *dev)
{
	int ret = 0;
	/* 1、獲取設備節點：/devicetree-leds-pincrl */
	dev->nd = of_find_node_by_path("/devicetree-leds-pincrl");
	if(dev->nd == NULL) {
			printk("devicetree-leds-pincrl node not find!\r\n");
			ret = -EINVAL;
			goto fail_fd;
	} else {
		printk("devicetree-leds-pincrl node find!\r\n");
	}
	/* 2、從設備節點下的led-gpio屬性裏獲取gpio編號 */
	dev->gpio = of_get_named_gpio(dev->nd, "leds-gpio", 0);
	if(dev->gpio < 0) {
			printk("can't get leds-gpio\r\n");
			ret = -EINVAL;
			goto fail_gpio;
	}
	printk("led-gpio num = %d\r\n", dev->gpio);
	/*3、申請gpio(檢查是否被佔用)*/
	ret = gpio_request(dev->gpio,"led2-gpio");
	if(ret)
	{
		ret = -EBUSY;
		printk("IO %d busy,can't request!\r\n",dev->gpio);
		goto fail_ioreq;
	}
	/*4、設置GPIO1_IO03爲輸出，並且輸出高電平，默認關閉LED燈 */
	ret = gpio_direction_output(dev->gpio,1);
	if(ret < 0) {
			printk("can't set gpio!\r\n");
			goto fail_ioset;
	}
	return 0;
fail_ioset:
	gpio_free(dev->gpio);
fail_ioreq:
fail_gpio:
fail_fd:
	return ret;
}
/*
 * @description	: 驅動出口函數
 * @param 		: 無
 * @return 		: 無
 */
static int __init kernel_timer_init(void)
{
	int ret = 0;
/*一、 註冊字符設備驅動 */
	/* 1、創建設備號 */
	if (mod_device.major)
	{		
		mod_device.devid = MKDEV(mod_device.major,0);//設備號起始點(major,0)
		//設備號起始點開始，申請DeviceNodes個設備號
		ret = register_chrdev_region(mod_device.devid,DeviceNodes,DeviceName);
	} 
	else {
		ret = alloc_chrdev_region(&mod_device.devid,0,DeviceNodes,DeviceName);	/* 申請設備號 */
		mod_device.major = MAJOR(mod_device.devid);	/* 獲取分配號的主設備號 */
		mod_device.minor = MINOR(mod_device.devid);	/* 獲取分配號的次設備號 */	
	}
	if(ret < 0){ printk("設備號申請失敗\r\n");goto fail_chrdev; }
	printk("mod_device major=%d,minor=%d\r\n",mod_device.major,mod_device.minor);	
	/* 2、初始化cdev並添加cdev */
	mod_device.cdev.owner = THIS_MODULE;
	cdev_init(&mod_device.cdev,&beep_fops);
	ret = cdev_add(&mod_device.cdev,mod_device.devid,DeviceNodes);
	if(ret < 0){ printk("添加cdev失敗！\r\n");goto fail_cdev; }
	/* 3、創建類 */
	mod_device.class = class_create(THIS_MODULE,DeviceName);
	if (IS_ERR(mod_device.class)) { ret = PTR_ERR(mod_device.class);goto fail_class; } 
	/* 4、創建設備節點*/
	mod_device.device = device_create(mod_device.class,NULL,mod_device.devid,NULL,"pinctrl-timer");
	if (IS_ERR(mod_device.device)) { ret = PTR_ERR(mod_device.device);goto fail_device; }
/*二、LED初始化*/
	ret = led_init(&mod_device);
	if(ret < 0){ printk("led_init fail!\r\n");goto fail_ledinit; }
/*三、初始化內核定時器*/
	init_timer(&mod_device.timer);
	mod_device.timer.function = timer_func;
	mod_device.timer.expires  = jiffies + msecs_to_jiffies(500);
	mod_device.timer.data     = (unsigned long)(&mod_device);
	add_timer(&mod_device.timer);
	return 0;
/*X、模塊加載失敗處理部分*/
fail_ledinit:
	device_destroy(mod_device.class,mod_device.devid);
fail_device:	
	class_destroy(mod_device.class);
fail_class:	
	cdev_del(&mod_device.cdev);
fail_cdev:
	unregister_chrdev_region(mod_device.devid,DeviceNodes);
fail_chrdev:
	return ret;
}
/*
 * @description	: 驅動出口函數
 * @param 		: 無
 * @return 		: 無
 */
static void __exit kernel_timer_exit(void)/*按照相反順序註銷*/
{
	/*一、內核定定時器註銷部分*/
	del_timer(&mod_device.timer);/* 刪除timer */
	/*二、LED設備註銷部分*/
	gpio_set_value(mod_device.gpio,1);/* 卸載驅動的時候關閉LED */
	gpio_free(mod_device.gpio);/*註銷gpio*/		
	/*三、設備驅動註銷部分*/			
	device_destroy(mod_device.class,mod_device.devid);/*註銷設備節點*/
	class_destroy(mod_device.class);/*註銷類*/
	cdev_del(&mod_device.cdev);/*註銷cdev*/
	unregister_chrdev_region(mod_device.devid,DeviceNodes);/*註銷chrdev*/
}
module_init(kernel_timer_init);
module_exit(kernel_timer_exit);
MODULE_LICENSE("GPL");

二、app測試程序

#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <stdio.h>
#include <unistd.h>
#include <stdlib.h>
#include <string.h>
#include <sys/ioctl.h>
/*
 *argc:應用程序參數個數
 *argv[]:具體的參數內容，字符串形式 
 *./timerAPP  <filename>  
 * ./timerAPP /dev/timer
 */
#define CLOSE_CMD      _IO(0XEF, 1)          //關閉命令               
#define OPEN_CMD       _IO(0XEF, 2)          //打開命令             
#define SETPERIOD_CMD  _IOW(0xEF, 3, int)    //設置週期
int main(int argc, char *argv[])
{
    int fd, ret;
    char *filename;
    unsigned char databuf[1];
    unsigned int cmd;
    unsigned int arg;
    unsigned char str[100];

    if(argc != 2) {
        printf("Error Usage!\r\n");
        return -1;
    }

    filename = argv[1];

    fd = open(filename, O_RDWR);
    if(fd < 0) {
        printf("file %s open failed!\r\n", filename);
        return -1;
    }

    /* 循環讀取 */
    while(1) {
        printf("Input CMD:");
        ret = scanf("%d", &cmd);
        if(ret !=1 ) {
            gets(str);  /* 防止卡死 */
        }

        if(cmd == 1) {          /* 關閉 */
            ioctl(fd, CLOSE_CMD, &arg);  
        } else if(cmd == 2) {   /* 打開 */
            ioctl(fd, OPEN_CMD, &arg); 
        } else if(cmd == 3) {    /* 設置週期 */
            printf("Please input timer period->");
            ret = scanf("%d", &arg);
            printf("Your input message->%d\r\n",arg);
            if(ret !=1 ) {
                gets(str);
            }
            ioctl(fd, SETPERIOD_CMD, &arg); 
        }
    }
    close(fd);
    return 0;
}