最近项目开发中使用到TCP通信,虽然以前编过简单的TCP的Server和Client程序,但是对于如果并发处理多个客户端没有多想过,上网查了一下,使用select()来实现比较方便,但是没找到完整的例子,下面的例子比较完整,所以保存供以后忘了在看看。
select()简述(来自baidu百科)
确定一个或多个套接口的状态,如:需要则等待。
#include <winsock.h>
int PASCAL FAR select( int nfds, fd_set FAR* readfds, fd_set FAR* writefds, fd_set FAR* exceptfds, const struct timeval FAR* timeout);
nfds:是一个整数值,是指集合中所有文件描述符的范围,即所有文件描述符的最大值加1,不能错!在Windows中这个参数的值无所谓,可以设置不正确。
readfds:(可选)指针,指向一组等待可读性检查的套接口。
writefds:(可选)指针,指向一组等待可写性检查的套接口。
exceptfds:(可选)指针,指向一组等待错误检查的套接口。
timeout:select()最多等待时间,对阻塞操作则为NULL。
#include <winsock.h>
int PASCAL FAR select( int nfds, fd_set FAR* readfds, fd_set FAR* writefds, fd_set FAR* exceptfds, const struct timeval FAR* timeout);
nfds:是一个整数值,是指集合中所有文件描述符的范围,即所有文件描述符的最大值加1,不能错!在Windows中这个参数的值无所谓,可以设置不正确。
readfds:(可选)指针,指向一组等待可读性检查的套接口。
writefds:(可选)指针,指向一组等待可写性检查的套接口。
exceptfds:(可选)指针,指向一组等待错误检查的套接口。
timeout:select()最多等待时间,对阻塞操作则为NULL。
///
fds_set介绍(baidu百科)
select()机制中提供一fd_set的数据结构,实际上是一long类型的数组,每一个数组元素都能与一打开的文件句柄(不管是socket句柄,还是其他文件或命名管道或设备句柄)建立联系,建立联系的工作由程序员完成,当调用select()时,由内核根据IO状态修改fd_set的内容,由此来通知执行了select()的进程哪一socket或文件发生了可读或可写事件。
select()机制中提供一fd_set的数据结构,实际上是一long类型的数组,每一个数组元素都能与一打开的文件句柄(不管是socket句柄,还是其他文件或命名管道或设备句柄)建立联系,建立联系的工作由程序员完成,当调用select()时,由内核根据IO状态修改fd_set的内容,由此来通知执行了select()的进程哪一socket或文件发生了可读或可写事件。
结构体原型:
typedef struct fd_set {
u_int fd_count;
socket fd_array[FD_SETSIZE];
} fd_set;
fd_set set;
FD_ZERO(&set); /*将set清零使集合中不含任何fd*/
FD_SET(fd, &set); /*将fd加入set集合*/
FD_CLR(fd, &set); /*将fd从set集合中清除*/
FD_ISSET(fd, &set); /*在调用select()函数后,用FD_ISSET来检测fd在fdset集合中的状态是否变化返回整型,当检测到fd状态发生变化时返回真,否则,返回假(0)*/
以上式子中的fd为socket句柄。
FD_ZERO(&set); /*将set清零使集合中不含任何fd*/
FD_SET(fd, &set); /*将fd加入set集合*/
FD_CLR(fd, &set); /*将fd从set集合中清除*/
FD_ISSET(fd, &set); /*在调用select()函数后,用FD_ISSET来检测fd在fdset集合中的状态是否变化返回整型,当检测到fd状态发生变化时返回真,否则,返回假(0)*/
以上式子中的fd为socket句柄。
///
Select I/O模型来实现一个并发处理多个客户端的TCP服务器 <转>
#include <winsock2.h> #include <stdio.h> #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; }