select模型Client——》Server

////////////////////////////////////////////////////// // select.cpp文件 #include <stdio.h> #include <winsock2.h> #pragma comment(lib, "WS2_32") // 链接到WS2_32.lib class CInitSock { public: CInitSock(BYTE minorVer = 2, BYTE majorVer = 2) { // 初始化WS2_32.dll WSADATA wsaData; WORD sockVersion = MAKEWORD(minorVer, majorVer); if(::WSAStartup(sockVersion, &wsaData) != 0) { exit(0); } } ~CInitSock() { ::WSACleanup(); } }; CInitSock initSock; // 初始化Winsock库 int main1(); int main2(); void main() { printf("Please select a item:/n"); printf("s:Server/n"); printf("c:Client/n"); char a; while(1){ scanf("%c",&a); if(a=='s'){main1();break;} else if(a=='c'){main2();break;} else continue; } } int main1() { 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集合的一个拷贝fdRead传递给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) // （1）监听套节字接收到新连接 { 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]; char temp; int nRecv = ::recv(fdSocket.fd_array[i], szText, strlen(szText), 0); printf("接收数据缓冲区长度：%d /n",strlen(szText)); printf("接收到数据长度：%d /n",nRecv); if(nRecv > 0) // （2）可读 { temp=szText[nRecv]; szText[nRecv] = '/0'; printf("接收到数据：%s /n", szText); szText[nRecv]=temp; } else // （3）连接关闭、重启或者中断 { ::closesocket(fdSocket.fd_array[i]); FD_CLR(fdSocket.fd_array[i], &fdSocket); } } } } } else { printf(" Failed select() /n"); break; } } return 0; } //Client int main2() { // 创建套节字 SOCKET s = ::socket(AF_INET, SOCK_STREAM, IPPROTO_TCP); if(s == INVALID_SOCKET) { printf(" Failed socket() /n"); return 0; } // 也可以在这里调用bind函数绑定一个本地地址 // 否则系统将会自动安排 // 填写远程地址信息 sockaddr_in servAddr; servAddr.sin_family = AF_INET; servAddr.sin_port = htons(4567); // 注意，这里要填写服务器程序（TCPServer程序）所在机器的IP地址 // 如果你的计算机没有联网，直接使用127.0.0.1即可 servAddr.sin_addr.S_un.S_addr = inet_addr("127.0.0.1"); if(::connect(s, (sockaddr*)&servAddr, sizeof(servAddr)) == -1) { printf(" Failed connect() /n"); return 0; } char szText[256]; while(TRUE){ scanf("%s",szText); //puts(szText); // 向服务器端发送数据 ::send(s, szText, strlen(szText), 0); } // 关闭套节字 ::closesocket(s); return 0; }