驱动支持select函数

I/O多路转接至今还不是 POSIX的组成部分。SVR4和 4.3 + BSD都提供select函数以执行I/O多路转接。poll函数只由SVR4 提供。SVR4 实际上用poll实现select。I/O多路转接的基本思想是：先构造一张有关描述符的表，然后调用一个函数，它要到这些描述符中的一个已准备好进行 I/O时才返回。在返回时，它告诉进程哪一个描述符已准备好可以进行 I/O。

I/O多路转接在服务器端用的比较多，可以同时处理多个连接的接入，但是也有缺陷，貌似只能接受1024个接入，因此现在又了epoll，当然这不是讨论的重点了。

select的函数原型为：

int select(int  numfds,  fd_set  *readfds,  fd_set  *writefds,  fd_set  *exceptfds,  struct  timeval  *timeout);

其中 readfds、writefds、exceptfds 分别是被 select()监视的读、写和异常处理的文件描述符集合，numfds 的值是需要检查的号码最高的文件描述符加 1。timeout 参数是一个指向 struct timeval类型的指针，它可以使select()在等待timeout时间后若没有文件描述符准备好则返回。

struct timeval 数据结构为：  

struct timeval  

{  

    int tv_sec; /* seconds */  

    int tv_usec; /* microseconds */  

};

除此之外，我们还将使用下列 API：

FD_ZERO(fd_set *set)――清除一个文件描述符集；  

FD_SET(int fd,fd_set *set)――将一个文件描述符加入文件描述符集中；  

FD_CLR(int fd,fd_set *set)――将一个文件描述符从文件描述符集中清除；  

FD_ISSET(int fd,fd_set *set)――判断文件描述符是否被置位。

 

select 用于查询设备的状态，以便用户程序获知是否能对设备进行非阻塞的访问，需要设备驱动程序中的poll 函数支持。 驱动程序中 poll 函数中最主要用到的一个 API 是 poll_wait，其原型如下：

void poll_wait(struct file *filp, wait_queue_heat_t *queue, poll_table * wait);

poll_wait 函数所做的工作是把当前进程添加到 wait 参数指定的等待列表（poll_table）中。

需要说明的是，poll_wait 函数并不阻塞，程序中 poll_wait(filp, &outq, wait)这句话的意思并不是说一直等待 outq 信号量可获得，真正的阻塞动作是上层的 select/poll 函数中完成的。select/poll 会在一个循环中对每个需要监听的设备调用它们自己的 poll 支持函数以使得当前进程被加入各个设备的等待列表。若当前没有任何被监听的设备就绪，则内核进行调度（调用 schedule）让出 cpu 进入阻塞状态，schedule 返回时将再次循环检测是否有操作可以进行，如此反复；否则，若有任意一个设备就绪，select/poll 都立即返回。

 

针对前面的文章的程序，我们在驱动中加入对poll的支持，程序改进为：

/*  global_poll.c */

#include <linux/module.h>

#include <linux/init.h>

#include <linux/fs.h>

#include <asm/uaccess.h>

#include <linux/wait.h>

#include <linux/semaphore.h>

#include <linux/device.h>

#include <linux/cdev.h>

#include <linux/sched.h>

#include <linux/poll.h>

 

MODULE_LICENSE("GPL");

 

#define init_MUTEX(LOCKNAME) sema_init(LOCKNAME,1)

 #define DEVICE_NAME "CDEV_ZHU"

static struct class *cdev_class;

struct cdev dev_c;

dev_t dev;

 

static ssize_t globalvar_read(struct file *, char *, size_t, loff_t*);

static ssize_t globalvar_write(struct file *, const char *, size_t, loff_t*);

static unsigned int globalvar_poll(struct file *filp, poll_table *wait);

 

struct file_operations globalvar_fops =

{

read: globalvar_read,

write: globalvar_write,

poll: globalvar_poll,

};

 

static int global_var = 0;

static struct semaphore sem;

static wait_queue_head_t outq;

static int flag = 0;

 

static int __init globalvar_init(void)

{

int ret,err;

ret = alloc_chrdev_region(&dev,0,1,DEVICE_NAME) ;

if (ret)

{

printk("globalvar register failure");

}

else

{

 

cdev_init(&dev_c,&globalvar_fops);

 err = cdev_add(&dev_c,dev,1);

if(err)

{

printk(KERN_NOTICE "error %d adding FC_dev\n",err);

unregister_chrdev_region(dev, 1);

return err;

}

else

{

printk("device register success! \n");

}

 

cdev_class = class_create(THIS_MODULE,DEVICE_NAME);

if(IS_ERR(cdev_class))

{

printk("ERR:cannot create a cdev_class\n");

unregister_chrdev_region(dev, 1);

return -1;

}

device_create(cdev_class, NULL, dev, 0, DEVICE_NAME);

 

init_MUTEX(&sem);

init_waitqueue_head(&outq);

}

return ret;

}

static void __exit globalvar_exit(void)

{

device_destroy(cdev_class,dev);

class_destroy(cdev_class);

unregister_chrdev_region(dev,1);

printk("device ");

}

 

 

static ssize_t globalvar_read(struct file *filp, char *buf, size_t len, loff_t *off)

{

 

if (wait_event_interruptible(outq, flag != 0))

{

return - ERESTARTSYS;

}

 

if (down_interruptible(&sem))

{

return - ERESTARTSYS;

}

 

flag = 0;

if (copy_to_user(buf, &global_var, sizeof(int)))

{

up(&sem);

return - EFAULT;

}

 

up(&sem);

 

return sizeof(int);

}

 

static ssize_t globalvar_write(struct file *filp, const char *buf, size_t len, loff_t *off)

{

if (down_interruptible(&sem))

{

return - ERESTARTSYS;

}

if (copy_from_user(&global_var, buf, sizeof(int)))

{

up(&sem);

return - EFAULT;

}

up(&sem);

flag = 1;

 

wake_up_interruptible(&outq);

 

return sizeof(int);

}

 

static unsigned int globalvar_poll(struct file *filp, poll_table *wait)

{

unsigned int mask = 0;

 poll_wait(filp, &outq, wait);

 if(flag != 0)

{

mask |= POLLIN | POLLRDNORM;

}

 return mask;

}

module_init(globalvar_init);

module_exit(globalvar_exit);

 下面的应用程序等待/dev/globalvar_poll 可读，但是设置了 5 秒的等待超时，若超过 5 秒仍然没有数据可读，则输出“No data within 5 seconds”：

/* poll_read.c */

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

int main()
{
    int fd, num;
    fd_set rfds;
    struct timeval tv;
    fd = open("/dev/CDEV_ZHU", O_RDWR, S_IRUSR | S_IWUSR);
    if (fd != -1)
    {
        while (1)
        {
            //查看 globalvar 是否有输入
              FD_ZERO(&rfds);
            FD_SET(fd, &rfds);

            //设置超时时间为 5s
            tv.tv_sec = 5;
            tv.tv_usec = 0;

            select(fd + 1, &rfds, NULL, NULL, &tv); //注：select 的文件描述符要加"1"，最后一个参数设为“NULL”，则会永久等待
            //数据是否可获得？
              if (FD_ISSET(fd, &rfds))
            {
                  read(fd, &num, sizeof(int));               
                  printf("The globalvar is %d\n", num);

                //输入为 0，退出
                    if (num == 0)
                 {
                    close(fd);
                    break;
                 }
            }
            else
                printf("No data within 5 seconds.\n");
        }

    }
    else
    {
        printf("device open failure\n");
    }
   return 0;
}

 

还需要另外一个应用程序进行写入，可以使用上上篇文章“死锁”中的“globalvar_write.c”作为写入。测试结果如下：