Select I/O模型来实现一个并发处理多个客户端的TCP服务器

 

#include  
#include  
#pragma comment(lib, “ws2_32.lib”) 
int main() 
{ 
	// 加载win socket 
	WSADATA ws; 
	int ret; 
	ret = WSAStartup(MAKEWORD(2, 2), &ws); 
	if (ret != 0) 
	{ 
		printf(”WSAStartup() 失败!\n”); 
		return -1; 
	} 
	// 创建侦听SOCKET 
	SOCKET sListen; 
	sListen = socket(AF_INET, SOCK_STREAM, 0); 
	if (sListen == INVALID_SOCKET) 
	{ 
		printf(”socket() 失败!\n”); 
		return -1; 
	} 
	// 填充服务器地址结构 
	sockaddr_in servAddr; 
	servAddr.sin_family = AF_INET; 
	servAddr.sin_addr.s_addr = INADDR_ANY; 
	servAddr.sin_port = htons(8000);

	// 绑定服务器套接字 
	ret = bind(sListen, (sockaddr*)&servAddr, sizeof(servAddr)); 
	if (ret == SOCKET_ERROR) 
	{ 
		printf(”bind() 失败!\n”); 
		return -1; 
	}

	// 开始侦听 
	ret = listen(sListen, 5); 
	if (ret == SOCKET_ERROR) 
	{ 
		printf(”listen() 失败!\n”); 
		return -1; 
	} 
	printf(”服务器启动成功，在端口%d监听…\n”, ntohs(servAddr.sin_port)); 
	//使用select模型 
	// 创建套接字集合 
	fd_set allSockSet; // 总的套接字集合 
	fd_set readSet; // 可读套接字集合 
	fd_set writeSet; // 可写套接字集合

	FD_ZERO(&allSockSet); // 清空套接字集合 
	FD_SET(sListen, &allSockSet); // 将sListen套接字加入套接字集合中 
	char bufRecv[100]; // 接收缓冲区 
	// 进入服务器主循环 
	while(1) 
	{ 
		FD_ZERO(&readSet); // 清空可读套接字 
		FD_ZERO(&writeSet); // 清空可写套接字 
		readSet = allSockSet; // 赋值 
		writeSet = allSockSet; // 赋值 
		// 调用select函数，timeout设置为NULL 
		ret = select(0, &readSet, 0, NULL, NULL); 
		// 
		if (ret == SOCKET_ERROR) 
		{ 
			printf(”select() 失败!\n”); 
			return -1; 
		} 
		// 存在套接字的I/O已经准备好 
		if(ret > 0) 
		{ 
			// 遍历所有套接字 
			for (int i = 0; i < allSockSet.fd_count; ++i) 
			{ 
				SOCKET s = allSockSet.fd_array[i]; 
				// 存在可读的套接字 
				if (FD_ISSET(s, &readSet)) 
				{ 
					// 可读套接字为sListen 
					if (s == sListen) 
					{ 
						// 接收新的连接 
						sockaddr_in clientAddr; 
						int len = sizeof(clientAddr); 
						SOCKET sClient = accept(s, (sockaddr*)&clientAddr, &len); 
						// 将新创建的套接字加入到集合中 
						FD_SET(sClient, &allSockSet); 
						printf(”>>>>>有新的连接到来啦…\n”); 
						printf(”目前客户端数目为：%d\n”, allSockSet.fd_count - 1); 
					} 
					else // 接收客户端信息 
					{ 
						ret = recv(s, bufRecv, 100, 0); 
						// 接收错误 
						if (ret == SOCKET_ERROR) 
						{ 
							DWORD err = WSAGetLastError(); 
							if (err == WSAECONNRESET) 
							printf(”客户端被强行关闭\n”); 
							else 
							printf(”recv() 失败!”); 
							// 删除套接字 
							FD_CLR(s, &allSockSet); 
							printf(”目前客户端数目为：%d\n”, allSockSet.fd_count - 1); 
							break; 
						} 
						if (ret == 0) 
						{ 
							printf(”客户端已经退出!\n”); 
							// 删除套接字 
							FD_CLR(s, &allSockSet); 
							printf(”目前客户端数目为：%d\n”, allSockSet.fd_count - 1); 
							break; 
						} 
						bufRecv[ret] = ‘\0′; 
						printf(”收到的消息：%s\n”, bufRecv); 
					} // end else

				}// end if

			}// end for 
		} // end if 
	}//end while 
	return 0; 
}