多路IO转接服务器——poll模型

poll模型和select模型很相似，只是描述fd集合的方式不同，poll使用pollfd结构而不是select的fd_set结构，poll得能够监听的数量远远超过了1024，可以通过（ulimit -n 数字）来设置监控的socket描述符个数。

poll模型涉及到得函数有：

int poll(struct pollfd *fds, nfds_t nfds, int timeout);
struct pollfd {
int fd; /* 文件描述符*/
short events; /* 监控的事件*/
short revents; /* 监控事件中满足条件返回的事件*/
};

结构体pollfd中的events可以有以下几种取值，如果你想同时监控一个文件描述符的读、写属性，可以用 | 连接

POLLIN普通或带外优先数据可读,即POLLRDNORM | POLLRDBAND
POLLRDNORM-数据可读
POLLRDBAND-优先级带数据可读
POLLPRI 高优先级可读数据

POLLOUT普通或带外数据可写
POLLWRNORM-数据可写
POLLWRBAND-优先级带数据可写

POLLERR 发生错误
POLLHUP 发生挂起
POLLNVAL 描述字不是一个打开的文件

nfds 监控数组中有多少文件描述符需要被监控

timeout 毫秒级等待
-1：阻塞等，#define INFTIM -1 Linux中没有定义此宏
0：立即返回，不阻塞进程
>0：等待指定毫秒数，如当前系统时间精度不够毫秒，向上取值

poll函数返回值：成功返回被激活的文件描述符个数，如果设置timeout时间阻塞读，那么timeout时间到但是没有任何描述符满足条件，返回0，出错返回-1

服务器端代码：server.c

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <poll.h>
#include <errno.h>
#include <ctype.h>

#define MAXLINE 80
#define SERV_PORT 8000
#define OPEN_MAX 1024
int main(int argc, char *argv[])
{
	int i, j, maxi, listenfd, connfd, sockfd;
	int nready;
	ssize_t n;
	char buf[MAXLINE], str[INET_ADDRSTRLEN];
	socklen_t clilen;
	struct pollfd client[OPEN_MAX];
	struct sockaddr_in cliaddr, servaddr;
	listenfd = socket(AF_INET, SOCK_STREAM, 0);
	bzero(&servaddr, sizeof(servaddr));
	servaddr.sin_family = AF_INET;
	servaddr.sin_addr.s_addr = htonl(INADDR_ANY);
	servaddr.sin_port = htons(SERV_PORT);
	bind(listenfd, (struct sockaddr *)&servaddr, sizeof(servaddr));
	listen(listenfd, 20);
	client[0].fd = listenfd;
	client[0].events = POLLRDNORM; /* listenfd监听普通读事件 */
	for (i = 1; i < OPEN_MAX; i++)
		client[i].fd = -1; /* 用-1初始化client[]里剩下元素 */
	maxi = 0; /* client[]数组有效元素中最大元素下标 */
	for ( ; ; ) {
		nready = poll(client, maxi+1, -1); /* 阻塞 */
		if (client[0].revents & POLLRDNORM) { /* 有客户端链接请求 */
			clilen = sizeof(cliaddr);
			connfd = accept(listenfd, (struct sockaddr *)&cliaddr, &clilen);
			printf("received from %s at PORT %d\n",
			inet_ntop(AF_INET, &cliaddr.sin_addr, str, sizeof(str)),
			ntohs(cliaddr.sin_port));
			for (i = 1; i < OPEN_MAX; i++)
				if (client[i].fd < 0) {
				    client[i].fd = connfd; /* 找到client[]中空闲的位置，存放accept返回的connfd */
				    break;
				}
			if (i == OPEN_MAX)
				perr_exit("too many clients");
			client[i].events = POLLRDNORM; /* 设置刚刚返回的connfd，监控读事件 */
			if (i > maxi)
				maxi = i; /* 更新client[]中最大元素下标 */
			if (--nready <= 0)
				continue; /* 没有更多就绪事件时,继续回到poll阻塞 */
		}
		for (i = 1; i <= maxi; i++) { /* 检测client[] */
			if ( (sockfd = client[i].fd) < 0)
				continue;
			if (client[i].revents & (POLLRDNORM | POLLERR)) {
				if ( (n = read(sockfd, buf, MAXLINE)) < 0) {
					if (errno == ECONNRESET) { /* 当收到 RST标志时 */
					/* connection reset by client */
						printf("client[%d] aborted connection\n", i);
						close(sockfd);
						client[i].fd = -1;
					} 
					else
						perr_exit("read error");
				} else if (n == 0) {
					/* connection closed by client */
					printf("client[%d] closed connection\n", i);
					close(sockfd);
					client[i].fd = -1;
				} else {
					for (j = 0; j < n; j++)
				    	buf[j] = toupper(buf[j]);
					writen(sockfd, buf, n);
				}
				if (--nready <= 0)
					break; /* no more readable descriptors */
			}
		}
	}
	return 0;
}

客户端代码：client.c

#include <stdio.h>
#include <string.h>
#include <unistd.h>
#include <netinet/in.h>

#define MAXLINE 80
#define SERV_PORT 8000
int main(int argc, char *argv[])
{
	struct sockaddr_in servaddr;
	char buf[MAXLINE];
	int sockfd, n;
	sockfd = socket(AF_INET, SOCK_STREAM, 0);
	bzero(&servaddr, sizeof(servaddr));
	servaddr.sin_family = AF_INET;
	inet_pton(AF_INET, "127.0.0.1", &servaddr.sin_addr);
	servaddr.sin_port = htons(SERV_PORT);
	connect(sockfd, (struct sockaddr *)&servaddr, sizeof(servaddr));
	while (fgets(buf, MAXLINE, stdin) != NULL) {
		write(sockfd, buf, strlen(buf));
		n = Read(sockfd, buf, MAXLINE);
		if (n == 0)
			printf("the other side has been closed.\n");
		else
			write(STDOUT_FILENO, buf, n);
	}
	close(sockfd);
	return 0;
}

运行结果如下：