linux驱动阻塞和非阻塞

一、阻塞与非阻塞理解

1、阻塞
如果不能获取到设备资源，那么阻塞式 IO 就会将应用程序对应的线程挂起，直到设备资源可以获取为止

fd = open("/dev/xxx_dev", O_RDWR); /* 阻塞方式打开 */
ret = read(fd, &data, sizeof(data)); /* 读取数据 */

2、非阻塞
如果不能获取到设备资源，线程不会挂起，它要么一直轮询等待，直到设备资源可以使用，要么就直接放弃。

fd = open("/dev/xxx_dev", O_RDWR | O_NONBLOCK); /* 非阻塞方式打开 */
ret = read(fd, &data, sizeof(data)); /* 读取数据 */

O_NONBLOCK”，表示以非阻塞方式打开设备

二、阻塞在内核实现——等待队列

1)等待队列头
Linux 内核提供了等待队列(wait queue)来实现阻塞进程的唤醒工作。如果我们要在驱动中使用等待队列，必须创建并初始化一个等待队列头，等待队列头使用结构体wait_queue_head_t 表示， wait_queue_head_t 结构体定义在文件 include/linux/wait.h 中，

struct __wait_queue_head {
	spinlock_t lock;
	struct list_head task_list;
};
typedef struct __wait_queue_head wait_queue_head_t;

使用 init_waitqueue_head 函数初始化等待队列头

void init_waitqueue_head(wait_queue_head_t *q)

2）等待你队列项
等待队列头就是一个等待队列的头部，每个访问设备的进程都是一个队列项，当设备不可
用的时候就要将这些进程对应的等待队列项添加到等待队列里面，结构体 wait_queue_t 表示等待队列项。

struct __wait_queue {
	unsigned int flags;
	void *private;
	wait_queue_func_t func;
	struct list_head task_list;
};
typedef struct __wait_queue wait_queue_t;

3)、将队列项添加/移除等待队列头

当设备不可访问的时候就需要将进程对应的等待队列项添加到前面创建的等待队列头中，
只有添加到等待队列头中以后进程才能进入休眠态。当设备可以访问以后再将进程对应的等待队列项从等待队列头中移除即可

//添加  wait 等待队列项
void add_wait_queue(wait_queue_head_t *q,wait_queue_t *wait)
//移除  wait 等待队列项
void remove_wait_queue(wait_queue_head_t *q,wait_queue_t *wait)

4、等待唤醒
当设备可以使用的时候就要唤醒进入休眠态的进程

void wake_up(wait_queue_head_t *q)

void wake_up_interruptible(wait_queue_head_t *q)

三、非阻塞在内核实现——轮询

poll、 epoll 和 select 可以用于处理轮询，在应用程序通过 select、 epoll 或 poll 函数来查询设备是否可以操作。

1、select

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

参数readfds、 writefds 和 exceptfds：都是 fd_set 类型的， fd_set 类型变量的每一个位都代表了一个文件描述符。
readfds 用于监视指定描述符集的读变化，也就是监视这些文件是否可以读取，只要这些集合里面有一个文件可以读取那么 seclect 就会返回一个大于 0 的值表示文件可以读取。如果没有文件可以读取，那么就会根据 timeout 参数来判断是否超时。可以将 readfs设置为 NULL，表示不关心任何文件的读变化

比如我们现在要从一个设备文件中读取数据，那么就可以定义一个 fd_set 变量，这个变量
要传递给参数 readfds。

void FD_ZERO(fd_set *set)//将fd_set变量的所有位都清零
void FD_SET(int fd, fd_set *set)//将fd_set某个位置1
void FD_CLR(int fd, fd_set *set)//将fd_set某个位清零
int FD_ISSET(int fd, fd_set *set)

timeout:超时时间

struct timeval {
	long tv_sec; /* 秒 */
	long tv_usec; /* 微妙 */
};

返回值： 0，表示的话就表示超时发生，但是没有任何文件描述符可以进行操作；-1，发生错误；其他值，可以进行操作的文件描述符个数

select使用示例

void main(void)
{
	int ret, fd; /* 要监视的文件描述符 */
	fd_set readfds; /* 读操作文件描述符集 */
	struct timeval timeout; /* 超时结构体 */
	
	fd = open("dev_xxx", O_RDWR | O_NONBLOCK); /* 非阻塞式访问 */
	
	FD_ZERO(&readfds); /* 清除 readfds */
	FD_SET(fd, &readfds); /* 将 fd 添加到 readfds 里面 */
	
	 /* 构造超时时间 */
	timeout.tv_sec = 0;
	timeout.tv_usec = 500000; /* 500ms */
	
	ret = select(fd + 1, &readfds, NULL, NULL, &timeout);
	switch (ret) {
		case 0: /* 超时 */
			printf("timeout!\r\n");
			break;
		case -1: /* 错误 */
			printf("error!\r\n");
			break;
		default: /* 可以读取数据 */
			 if(FD_ISSET(fd, &readfds)) { /* 判断是否为 fd 文件描述符 */
			 /* 使用 read 函数读取数据 */
			}
			 break;
	}
 }

2、 poll 函数

在单个线程中， select 函数能够监视的文件描述符数量有最大的限制，一般为 1024
poll 函数本质上和 select 没有太大的差别，但是 poll 函数没有最大文件描述符限制，

int poll(struct pollfd *fds,
				nfds_t nfds,
				int timeout)

fds 要监视的文件描述符集合以及要监视的事件结构体， pollfd 结构体如下所示

struct pollfd {
		int fd; /* 文件描述符 */
		short events; /* 请求的事件 */
		short revents; /* 返回的事件 */
};

events事件类型

POLLIN 有数据可以读取。
POLLPRI 有紧急的数据需要读取。
POLLOUT 可以写数据。
POLLERR 指定的文件描述符发生错误。
POLLHUP 指定的文件描述符挂起。
POLLNVAL 无效的请求。
POLLRDNORM 等同于 POLLIN

nfds： poll 函数要监视的文件描述符数量。
timeout： 超时时间，单位为 ms。

返回值：返回 revents 域中不为 0 的 pollfd 结构体个数，也就是发生事件或错误的文件描述
符数量； 0，超时； -1，发生错误，并且设置 errno 为错误类型

poll使用

void main(void)
{
	int ret;
	int fd; /* 要监视的文件描述符 */
	struct pollfd fds;
	
	fd = open(filename, O_RDWR | O_NONBLOCK); /* 非阻塞式访问 */
	
	/* 构造结构体 */
	fds.fd = fd;
	fds.events = POLLIN; /* 监视数据是否可以读取 */
	
	ret = poll(&fds, 1, 500); /* 轮询文件是否可操作，超时 500ms */
	if (ret) { /* 数据有效 */
			 ......
			 /* 读取数据 */
			 ......
	 } else if (ret == 0) { /* 超时 */
		......
	 } else if (ret < 0) { /* 错误 */
		 ......
	 }
 }

3、 epoll 函数

epoll 就是为处理大并发而准备的，一般常常在网络编程中使用 epoll 函数。

select 与 poll缺点
selcet 和 poll 函数都会随着所监听的 fd 数量的增加，出现效率低下的问题，而且poll 函数每次必须遍历所有的描述符来检查就绪的描述符，这个过程很浪费时间。

1）创建一个 epoll 句柄

int epoll_create(int size)

2）向其中添加要监视的文件描述符以及监视的事件

int epoll_ctl(int epfd,
				int op,
				int fd,
				struct epoll_event *event)

op： 表示要对 epfd(epoll 句柄)进行的操作，可以设置为：

EPOLL_CTL_ADD 向 epfd 添加文件参数 fd 表示的描述符。
EPOLL_CTL_MOD 修改参数 fd 的 event 事件。
EPOLL_CTL_DEL 从 epfd 中删除 fd 描述符

event： 要监视的事件类型

struct epoll_event {
			uint32_t events; /* epoll 事件 */
			epoll_data_t data; /* 用户数据 */
};

events 成员变量表示要监视的事件,事件可以组合使用

EPOLLIN 有数据可以读取。
EPOLLOUT 可以写数据
EPOLLPRI 有紧急的数据需要读取。
EPOLLERR 指定的文件描述符发生错误。
EPOLLHUP 指定的文件描述符挂起。
EPOLLET 设置 epoll 为边沿触发，默认触发模式为水平触发。
EPOLLONESHOT 一次性的监视，当监视完成以后还需要再次监视某个 fd，那么就需要将
fd 重新添加到 epoll 里面

3）等待事件的发生

int epoll_wait(int epfd,
				struct epoll_event *events,
				int maxevents,
				int timeout)