网络编程学习2：select模型

选择(select)模型

select模型是一个广泛在Winsock中使用的I/O模型。称它为select模型，是因为它主要是使用select函数来管理I/O的。

1 接口

// select函数可以确定一个或者多个套接字的状态。如果套接字上没有网络事件发生，便进入等待状态，以便执行同步I/O。
int select(
	int nfds,           // 忽略，仅是为了与Berkeley套接字兼容
	fd_set * readfds,   // 指向一个套接字集合，用来检查其可读性
	fd_set * writefds,  // 指向一个套接字集合，用来检查其可写性
	fd_set * exceptfds, // 指向一个套接字集合，用来检查错误
	const struct timeval * timeout // 指定此函数等待的最长时间，如果为NULL，则最长时间为无限大。
);
// 注：函数调用成功，返回发生网络事件的所有套接字数量的总和。如果超过了时间限制，返回0，失败则返回SOCKET_ERROR.

// fd_set结构可以把多个套接字连在一起，形成一个套接字集合。select函数可以测试这个集合中哪些套接字有事件发生。
typedef struct fd_set{
	u_int fd_count;               // 下面数组的大小
	SOCKET fd_array[FD_SETSIZE];  // 套接字句柄数组
}fd_set;
// 下面是WINSOCK定义的4个操作fd_set套接字集合的宏
FD_ZERO(*set)      // 初始化set为空集合。集合在使用前应该总是清空
FD_CLR(s, &set);   // 从set移除套接字s
FD_ISSET(s, &set); // 检查s是不是set的成员，如果是返回TRUE
FD_SET(s, &set);   // 添加套接字到集合

// 最后的参数timeout是timeval结构的指针，它指定了select函数等待的最长时间。如果设为NULL,select将会无限阻塞，直到有网络事件发生。
typedef struct timeval
{
	long tv_sec;     // 指示等待多少秒
	long tv_usec;    // 指示等待多少毫秒
}timeval;

2 实例

CInitSock theSock;              // 初始化Winsock库
int main()
{
	USHORT nPort = 4567;        // 此服务器监听的端口号
	// 创建监听套接字
	SOCKET sListen = ::socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
	sockaddr_in sin;
	sin.sin_family = AF_INET;
	sin.sin_port = htons(nPort);
	sin.sin_addr.S_un.S_addr = INADDR_ANY;
	
	// 绑定套接字到本地机器
	if(::bind(sListen, (sockaddr*)&sin, sizeof(sin)) == SOCKET_ERROR){
		printf("failed bind()\n");
		return -1;
	}
	// 进入监听模式
	::listen(sListen, 5);
	// select模型处理过程
	// 1)初始化一个套接字集合fdSocket,添加监听套接字句柄到这个集合
	fd_set fdSocket;         // 所有可用套接字集合
	FD_ZERO(&fdSocket);
	FD_SET(sListen, &fdSocket);
	
	while(TRUE){
		// 2) 将fdSocket集合的一个拷贝fdReset传递给select函数，当有事件发生时，select函数移除fdRead集合中没有未决I/O操作的套接字句柄，然后返回。
		fd_set fdRead = fdSocket;
		int nRet = ::select(0, &fdRead, NULL, NULL, NULL);
		if(nRet > 0)
		{
			// 3) 通过将原来fdSocket集合与select处理过的fdRead集合比较,确定都有哪些套接字有未决I/O，并进一步处理这些I/O。
			for(int i = 0; i < (int)fdSocket.fd_count;i++){
				if(FD_ISSET(fdSocket.fd_array[i], &fdRead))		
				{
					// 监听套接字接收到新连接
					if(fdSocket.fd_array[i] == sListen)
					{
						if(fdSocket.fd_count < FD_SETSIZE)
						{
							sockaddr_in addrRemote;
							int nAddrLen = sizeof(addrRemote);
							SOCKET sNew = ::accept(sListen, (SOCKADDR*)&addrRemote, &nAddrLen);
							FD_SET(sNew, &fdSocket);
							printf("接收到连接%s\n", ::inet_ntoa(addrRemote.sin_addr));
						}
						else
						{
							printf("too much connections!\n");
							continue;
						}
					}
					else
					{
						char szText[256];
						int nRecv = ::recv(fdSocket.fd_array[i], szText, strlen(szText), 0);
						// 可读
						if(nRecv > 0)
						{
							szText[nRecv] = '\0';
							printf("接收到数据:%s\n", szText);
						}
						// 连接关闭、重启或者中断
						else
						{
							::closesocket(fdSocket.fd_array[i]);
							FD_CLR(fdSocket.fd_array[i], &fdSocket);
						}
					}
				}
			}
		}
		else
		{
			printf("Failed select()\n");
			break;
		}
	}
	return 0;
}