Linux网络编程——tcp并发服务器（poll实现）

想详细彻底地了解poll或看懂下面的代码请参考《Linux网络编程——I/O复用之poll函数》

代码：

#include <string.h>
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <sys/select.h>
#include <sys/time.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <poll.h>
#include <errno.h>
#define OPEN_MAX 100

int main(int argc, char *argv[])
{
	//1.创建tcp监听套接字
	int sockfd = socket(AF_INET, SOCK_STREAM, 0);
	
	//2.绑定sockfd
	struct sockaddr_in my_addr;
	bzero(&my_addr, sizeof(my_addr));
	my_addr.sin_family = AF_INET;
	my_addr.sin_port = htons(8000);
	my_addr.sin_addr.s_addr = htonl(INADDR_ANY);
	bind(sockfd, (struct sockaddr *)&my_addr, sizeof(my_addr));
	
	//3.监听listen
	listen(sockfd, 10);
	
	//4.poll相应参数准备
	struct pollfd client[OPEN_MAX];
	int i = 0, maxi = 0;
	for(;i<OPEN_MAX; i++)
		client[i].fd = -1;//初始化poll结构中的文件描述符fd
	
	client[0].fd = sockfd;//需要监测的描述符
	client[0].events = POLLIN;//普通或优先级带数据可读
	
	//5.对已连接的客户端的数据处理
	while(1)
	{
		int ret = poll(client, maxi+1, -1);//对加入poll结构体数组所有元素进行监测
		
		//5.1监测sockfd(监听套接字)是否存在连接
		if((client[0].revents & POLLIN) == POLLIN )
		{
			struct sockaddr_in cli_addr;
			int clilen = sizeof(cli_addr);
			int connfd = 0;
			//5.1.1 从tcp完成连接中提取客户端
			connfd = accept(sockfd, (struct sockaddr *)&cli_addr, &clilen);
			
			//5.1.2 将提取到的connfd放入poll结构体数组中，以便于poll函数监测
			for(i=1; i<OPEN_MAX; i++)
			{
				if(client[i].fd < 0)
				{
					client[i].fd = connfd;
					client[i].events = POLLIN;
					break;
				}
			}
			
			//5.1.3 maxi更新
			if(i > maxi)
				maxi = i;
				
			//5.1.4 如果没有就绪的描述符，就继续poll监测，否则继续向下看
			if(--ret <= 0)
				continue;
		}
		
		//5.2继续响应就绪的描述符
		for(i=1; i<=maxi; i++)
		{
			if(client[i].fd < 0)
				continue;
			
			if(client[i].revents & (POLLIN | POLLERR))
			{
				int len = 0;
				char buf[128] = "";
				
				//5.2.1接受客户端数据
				if((len = recv(client[i].fd, buf, sizeof(buf), 0)) < 0)
				{
					if(errno == ECONNRESET)//tcp连接超时、RST
					{
						close(client[i].fd);
						client[i].fd = -1;
					}
					else
						perror("read error:");
					
				}
				else if(len == 0)//客户端关闭连接
				{
					close(client[i].fd);
					client[i].fd = -1;
				}
				else//正常接收到服务器的数据
					send(client[i].fd, buf, len, 0);
				
				//5.2.2所有的就绪描述符处理完了，就退出当前的for循环，继续poll监测
				if(--ret <= 0)
					break;
				
			}
		}
	}
	return 0;
}

运行结果：

源码下载：