I/O多路转接之select

什么是select？

首先它是系统调用，系统提供它用来实现多路复用输入/输出模型。利用它可以让我们的程序来监视多个文件句柄。程序会停在select这里等待,直到被监视的文件句柄有一个或多个发生了状态改变。

         

参数nfds代表要监听的文件描述符的最大值加1。接下来三个参数分别表示要监听事件对应的文件描述符集：可读文件描述符的集合,可写文件描述符的集合及异
常文件描述符的集合。与之相匹配的有一组宏：

         

FD_CLR(inr fd,fd_set* set);用来清除描述词组set中相关fd 的位
FD_ISSET(int fd,fd_set *set);用来测试描述词组set中相关fd 的位是否为真
FD_SET(int fd,fd_set*set);用来设置描述词组set中相关fd的位
FD_ZERO(fd_set *set);用来清除描述词组set的全部位
参数timeout为结构timeval,用来设置select()的等待时间,其结构定义如下:

         

如果参数timeout设为:
NULL:则表示select()没有timeout,select将一一直被阻塞,直到某个文件描述符上发生了
事件。
0:仅检测描述符集合的状态,然后立即返回,并不等待外部事件的发生。
特定的时间值:如果在指定的时间段里没有事件发生,select将超时返回。

函数返回值:
执行行成功则返回文文件描述词状态已改变的个数；如果返回0代表在描述词状态改变前已超过timeout时间,没有返回;当有错误发生生时则返回-1,错误原因存于errno,此时参数readfds,writefds,exceptfds和timeout的值变成不可预测。错误值可能为:

EBADF 文文件描述词为无无效的或该文文件已关闭；EINTR 此调用用被信号所中断；EINVAL 参数n 为负值；ENOMEM 核心内存不足。

select缺点:

(1)每次调用select,都需要把fd集合从用户态拷贝到内核态,这个开销在fd很多时会很大
(2)同时每次调用select都需要在内核遍历传递进来的所有fd,这个开销在fd很多时也很大
(3)select支持的文件描述符数量太小了,默认是1024

示例：

server：

#include <stdio.h>
#include <sys/socket.h>
#include <stdlib.h>
#include <sys/types.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <string.h>
#include <unistd.h>

int fds[64];
const int fds_nums = sizeof(fds) / sizeof(fds[0]);
static int startup(const char *_ip, int _port)
{
	int sock = socket(AF_INET, SOCK_STREAM, 0);
	if ( sock < 0 )
	{
		perror("scoket");
		exit(2);
	}
	

	struct sockaddr_in local;
	local.sin_family = AF_INET;
	local.sin_port = htons(_port);
	local.sin_addr.s_addr = inet_addr(_ip);

	if (bind(sock, (struct sockaddr*)&local, sizeof(local)))
	{
		perror("bind");
		exit(3);
	}

	if ( listen(sock, 5) < 0 )
	{
		perror("listen");
		exit(4);
	}
	
	return sock;
}

static void usage(const char *proc)
{
	printf("Usage: %s [ip] [port]\n", proc);
}

int main(int argc, char *argv[])
{
	if (argc != 3)
	{
		usage(argv[0]);	
		exit(1);
	}
		

	int i = 0;
	for (; i < fds_nums; ++i)
	{
		fds[i] = -1;
	}
	
	int listen_sock = startup(argv[1], atoi(argv[2]));
	
	int opt = 1;
	setsockopt(listen_sock, SOL_SOCKET, SO_REUSEADDR, &opt, sizeof(opt));

	fd_set rset;
	FD_ZERO(&rset);
	FD_SET(listen_sock, &rset);

	fds[0] = listen_sock;
	
	int done = 0;
	while (!done)
	{
		int max_fd = -1;
		for (i = 0; i < fds_nums; ++i)
		{
			if (fds[i] > 0)
			{
				FD_SET(fds[i], &rset);
				max_fd = max_fd > fds[i] ? max_fd:fds[i];
			}
		}

		struct timeval timeout = {5, 0};
		switch(select(max_fd + 1, &rset, NULL, NULL, NULL/*&timeout*/))
		{
		case 0:
			printf("timeout...\n");
			break;
		case -1:
			perror("select");
			break;
		default:
			{
				for(i = 0; i < fds_nums; ++i)
				{
					if (i==0 && FD_ISSET(listen_sock, &rset))
					{
						//some fds ready for io
						struct sockaddr_in peer;
						socklen_t len = sizeof(peer);
						int new_fd = accept(listen_sock, (struct sockaddr*)&peer, &len);
						if (new_fd > 0)
						{
							int j = 0;
							printf("get a new client: socket:%s:%d\n", inet_ntoa(peer.sin_addr), ntohs(peer.sin_port));
							for (j = 0; j < fds_nums; ++j)
							{
								if (fds[j] == -1)
								{
									fds[j] = new_fd;
									break;
								}
							}
							if (j == fds_nums)
							{
								close(new_fd);
							}

						}
					}	
					else
					{
						if (FD_ISSET(fds[i], &rset))
						{
							char buf[1024];
							memset(buf, '\0', sizeof(buf));
							ssize_t _s = read(fds[i], buf, sizeof(buf) - 1);
							if (_s > 0)
							{
								printf("client# %s\n", buf);
							}
							else if (_s == 0)
							{
								printf("client close...\n");
								close(fds[i]);
								fds[i] = -1;
							}
							else
							{
								printf("error...\n");
							}
						}
					}
				} // for
			}
			break;
		}
	}
	
	return 0;
}

把服务跑起来：

         

用telnet测试：