linux 驱动学习之poll与select

        在使用socket或串口的时候应用代码经常使用select来判断有没接收到数据，驱动需要做什么处理，应用层用户才能正确判断有数据收到并读取数据呢？使用select能够监视我们需要监视的文件描述符的变化情况——读写或是异常。
       先建立简单模型，应用使用select监听设备是否可读，当设备可读将数据读出后继续监听。已经有了一个监听设备可读状态的进程，还要打开终端使用echo xxx >/dev/moddev60向设备写入数据，设备驱动收到数据后控制设备变成可读状态，select进程select函数返回，查询设备可读后读出数据。代码为了演示做了简化不做并发出错处理等。

先看下应用c代码

#include <stdio.h>
#include <fcntl.h>
unsigned char rdBuf[1024];
int main (int *argc,char**argv)
{
   int fd,res;
	fd_set fset;
   FD_ZERO(&fset);
   fd=open("/dev/moduledev60",O_RDWR);
   if(fd<0){
	    printf("open devices error\n");
		return -1;
    }
    while(1)
	 {
	   FD_ZERO(&fset); 
	   FD_SET(fd,&fset); 
       res=select(fd+1,&fset,NULL,NULL,NULL);
       if(res==-1)
	   {
		 close(fd);  
	     return -1;
	   }
	   else if(res==0)  break;
	   else
	   { /*select 返回后判断可读*/
	     if(FD_ISSET(fd,&fset))
		 { 
		   read(fd,rdBuf,1024);	 
		   printf("read data:%s\n",rdBuf);
		 }
	   }
	 }
	close(fd);
	return 0;
}

驱动关键代码

struct file_operations ops=
{
   .owner=THIS_MODULE  ,
   .read =fileops_read ,
   .write=fileops_write,
   .poll =fileops_poll
};

DECLARE_WAIT_QUEUE_HEAD(rdWait);//poll_wait中使用
static unsigned char rdBuf[1024];//测试用读写缓存 写入数据和读取数据0结尾
static unsigned char rdFlag=0;   //可读标记 0不可读缓存内无数据  1可读

ssize_t fileops_read(struct file *filp, char __user *buff, size_t count, loff_t *offp)
{
  unsigned int len;
  if(rdFlag==0) return 0;        //不可读时直接返回
  len=strlen(rdBuf);             //读缓存字符串长  这个模型只做字符串（0结尾）的写入和读取
  copy_to_user(buff,rdBuf,len);  //
  rdFlag=0;                      //清标记
  return len;
}

ssize_t fileops_write(struct file *filp, const char __user *buff, size_t count, loff_t *offp)
{
  copy_from_user(rdBuf,buff,count);//
  rdBuf[count]=0;                  //补字符串结束标记
  rdFlag=1;                        //标记可读 poll函数检测这个标记
  wake_up_interruptible(&rdWait);  /*fileops_poll 函数使用了 poll_wait 加入了rdWait等待列队，执行后fileops_poll会被调用，如果不使用等待列队

                                    fileops_poll只在应用调用select时执行一次 这样有数据select也不会返回*/

  return count;
}

unsigned int fileops_poll(struct file *filp,poll_table *wait)
{
  /*这里测试只处理的可读标记 也可以增加可写标记*/
  int mask=0;
  poll_wait(filp,&rdWait,wait); //这个函数不阻塞的，当状态改变时，应该唤醒列队这个函数才会重新调用更新新的状态。
  if(rdFlag) 
  {
     mask=POLLIN|POLLRDNORM;
  }
  return mask; 
}

编译后测试结果

先运行应用，监听设备是否可读
[root@localhost ctest]# ./main 

打开另外一个终端，写入数据
[root@localhost ctest]# echo eeee>/dev/moduledev60
[root@localhost ctest]# echo eeee>/dev/moduledev60
[root@localhost ctest]# echo eeee>/dev/moduledev60

应用判断到可读后读出数据
[root@localhost ctest]# ./main 
read data:eeee

read data:eeee

read data:eeee