用IO多路复用之select与poll/epoll来浅谈网络聊天室

IO多路复用：

IO多路复用模型是建立在内核提供的多路分离函数select基础之上的，使用select函数可以避免同步非阻塞IO模型中轮询等待的问题。

select

该函数用于监视文件描述符的变化情况–读写或是异常

参数：

nfds: 通常被设置为select所监听的所有文件描述符中的最大值+1

		指定被监听的文件描述符的总数

		文件描述符从0开始

fd_set: 文件描述符集合

		fd_set结构体仅包含一个整数数组，该数组的每个元素的每一位(bit)

		标记一个文件描述符

		fd_set能容纳 的文件描述符数量由FD_SETSIZE指定

		限制了select所有处理的文件描述符数量 

		FD_ZERO(fd_set *fdset); //清除所有位

		FD_SET(int fd,fd_set *fdset);   //设置fd

		FD_CLR(int fd,fd_set *fdset);   //清除fd

		int FD_ISSET(int fd,fd_set *fdset); //测试

	readfds,writefds,exceptfds 分别用于记录要监听是否

		可读、可写、异常事件的文件描述符符合

	select函数调用返回时，内核将修改readfds,writefds,excepts

	文件描述符集合，保留有数据可读、可写、异常的文件描述符

timeout：

	设置select函数的超时时间

	返回select调用返回后剩余的时间，如果调用失败时timeout不确定

	如果timeout变量成员都为0，则select立即返回

	如果timeout取值为NULL,select将一直阻塞，直到某个文件描述符就绪

	struct timeval{

		long tv_sec;   //秒数

		long tv_usec;  //微秒数

	}

返回值：

	select成功时返回就绪(可读、可写和异常)文件描述符的总和

	如果在超时时间内滑任何文件描述符就绪，返回0

	select失败返回-1并设置errno

	在select等待期间，程序接收到信号，select立即返回-1，

	errno为EINTR

原理:

	把需要监听的文件描述符集合交给内核去测试，保留就绪的文件描述



	每一次都需要重新把所有需要监听的文件描述符加入到文件描述符集合中

	内核中需要遍历所有的文件描述符

	当select返回之后，还需要循环去所有的文件描述符去判断是否就绪

	

	当文件描述符数量增大时，效率其实是急剧下降的

	select还受到FD_SETSIZE的限制 

	当select返回之后，没有并行，而是串行 一个一个接收客户端的数据，转发

select()函数与Linux驱动程序的关系

当用户调用select系统调用时，select系统调用会先调用poll_initwait(&table)，然后调用驱动程序中 struct file_operations下的fop->poll函数，在这个函数里应该调用poll_wait()，将current加到某个等待队列(这里调用poll_wait())，并检查是否有效，如果无效就调用schedule_timeout();去睡眠。事件发生后，schedule_timeout()回来，调用fop->poll()，检查到可以运行，就调用poll_freewait(&table);从而完成select系统调用。重要的是fop->poll()里面要检查是否就绪，如果是，要返回相应标志。

聊天室

#include <stdio.h>
#include <stdlib.h>
#include <stdbool.h>
#include <string.h>
#include <errno.h>
#include <assert.h>
#include <unistd.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <sys/select.h>

#define NAME_LEN 48
#define MAX_CLIENTS 100
#define MSG_LEN 1024
typedef struct Client{
   
	int fd;
	struct sockaddr_in addr;
	char name[NAME_LEN];
}Client;

//全局变量 每个线程都会访问 
Client gcls[MAX_CLIENTS] = {
   };
int size = 0;

int init_server(const char *ip,unsigned short int port){
   
	int fd = socket(AF_INET,SOCK_STREAM,0);
	assert(fd != -1);
	struct sockaddr_in addr = {
   };
	addr.sin_family = AF_INET;
	addr.sin_port = htons(port);
	addr.sin_addr.s_addr = inet_addr(ip);
	socklen_t len = sizeof(addr);
	int ret = bind(fd,(const struct sockaddr*)&addr,len);
	assert(ret != -1);
	ret = listen(fd,MAX_CLIENTS);
	assert(ret != -1);
	return fd;
}

void broadcast(int fd,const char *msg){
   
	int i;
	for(i=0;i<size;i++){
   
		if(fd != gcls[i].fd){
   
			send(gcls[i].fd,msg,strlen(msg)+1,0);	
		}	
	}
}

void remove_client(int fd){
   
	int i;
	for(i=0;i<size;i++){
   
		if(fd == gcls[i].fd){
   
			close(fd);
			gcls[i] = gcls[--size];
			break;
		}	
	}
}

void select_fd(int fd){
   
	Client cls = {
   };
	socklen_t len = sizeof(cls.addr);
	int ret = 0,i,cnt;
	fd_set readfds; //可读文件描述符集
	int maxfd = 0;  //记录最大的文件描述符
	while(true){
   
		FD_ZERO(&readfds);//文件描述符集合清空
		maxfd = fd;
		FD_SET(fd,&readfds);//服务器的fd  用于判断是否有客户端连接
		for(i=0;i<size;i++){
   
			FD_SET(gcls[i].fd,&readfds);//和客户端交互的文件描述符
			if(gcls[i].fd > maxfd)
				maxfd = gcls[i].fd;
		}
		cnt = select(maxfd+1,&readfds,NULL,NULL