winsock select 学习代码(2)
之前文章的改进版
服务器仅仅接受客户端发送的字符串并显示
客户端可以调节发送数目 但是不能超过64
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 | // SelectServer.cpp : 定义控制台应用程序的入口点。 // #include "stdafx.h" #include <winsock2.h> #include <conio.h> #pragma comment(lib,"ws2_32") #define SOCKET_MAXCNT 64 #define THREAD_NUM 2 #define CONNECT_PORT 8773 #define CONNECT_ADDR "127.0.0.1" #define DATA_BUF_SZIE 8192 typedef struct _SOCKET_INFORMATION { CHAR Buffer[DATA_BUF_SZIE]; WSABUF DataBuf; SOCKET Socket; OVERLAPPED Overlapped; } SOCKET_INFORMATION, * LPSOCKET_INFORMATION; DWORD totalSockets = 0; LPSOCKET_INFORMATION SocketArray[FD_SETSIZE]; BOOL CreateSocketInformation(SOCKET s) { LPSOCKET_INFORMATION SI; printf( "Accepted socket number %d\n" , s); if ((SI = (LPSOCKET_INFORMATION) GlobalAlloc(GPTR, sizeof (SOCKET_INFORMATION))) == NULL) { printf( "GlobalAlloc() failed with error %d\n" , GetLastError()); return FALSE; } // Prepare SocketInfo structure for use. SI->Socket = s; SocketArray[totalSockets] = SI; totalSockets++; return (TRUE); } void FreeSocketInformation(DWORD Index) { LPSOCKET_INFORMATION SI = SocketArray[Index]; DWORD i; closesocket(SI->Socket); printf( "Closing socket number %d\n" , SI->Socket); GlobalFree(SI); // Squash the socket array for (i = Index; i < totalSockets; i++) { SocketArray[i] = SocketArray[i + 1]; } totalSockets--; } int _tmain( int argc, _TCHAR* argv[]) { WSAData wsaData; SOCKADDR_IN InternetAddr = {0}; FD_SET ReadSet; DWORD i = 0; DWORD total = 0; DWORD Flags; DWORD RecvBytes = 0; SOCKET listenSock = INVALID_SOCKET; SOCKET acceptSock = INVALID_SOCKET; if ( WSAStartup(MAKEWORD(2,2),&wsaData) != 0 ) { printf( "WSAStartup error,exit\n" ); goto SOCKET_CLEAN; } listenSock = WSASocket(AF_INET,SOCK_STREAM,0,NULL,0, WSA_FLAG_OVERLAPPED); if (listenSock == INVALID_SOCKET) { printf( "WSASocket error \n" ); goto SOCKET_CLEAN; } InternetAddr.sin_family = AF_INET; InternetAddr.sin_addr.s_addr = htonl(INADDR_ANY); InternetAddr.sin_port = htons(CONNECT_PORT); if (bind(listenSock, (PSOCKADDR) &InternetAddr, sizeof (InternetAddr)) == SOCKET_ERROR) { printf( "bind() failed with error %d\n" , WSAGetLastError()); goto SOCKET_CLEAN; } if (listen(listenSock, 5)) { printf( "listen() failed with error %d\n" , WSAGetLastError()); goto SOCKET_CLEAN; } // Change the socket mode on the listening socket from blocking to // non-block so the application will not block waiting for requests. ULONG NonBlock = 1; if (ioctlsocket(listenSock, FIONBIO, &NonBlock) == SOCKET_ERROR) { printf( "ioctlsocket() failed with error %d\n" , WSAGetLastError()); goto SOCKET_CLEAN; } while (1) { FD_ZERO(&ReadSet); FD_SET(listenSock, &ReadSet); for (i = 0; i < totalSockets; i++) { FD_SET(SocketArray[i]->Socket, &ReadSet); } total = select (0,&ReadSet,NULL,NULL,NULL); if (total == SOCKET_ERROR) { printf( "select() failed with error %d\n" , WSAGetLastError()); goto SOCKET_CLEAN; } if (FD_ISSET(listenSock,&ReadSet)) { total--; acceptSock = accept(listenSock,NULL,NULL); if (acceptSock == INVALID_SOCKET) { if (WSAGetLastError() != WSAEWOULDBLOCK) { printf( "accept() failed with error %d\n" , WSAGetLastError()); goto SOCKET_CLEAN; } } else { NonBlock = 1; int ret = ioctlsocket(acceptSock,FIONBIO, &NonBlock); if (ret == SOCKET_ERROR) { printf( "ioctlsocket() failed with error %d\n" , WSAGetLastError()); goto SOCKET_CLEAN; } if (CreateSocketInformation(acceptSock) == FALSE) goto SOCKET_CLEAN; } } for (i = 0; total > 0 && i < totalSockets; i++) { LPSOCKET_INFORMATION SocketInfo = SocketArray[i]; SocketInfo->DataBuf.buf = SocketInfo->Buffer; SocketInfo->DataBuf.len = DATA_BUF_SZIE; if (FD_ISSET(SocketInfo->Socket,&ReadSet)) { total--; Flags = 0; ZeroMemory(SocketInfo->Buffer,DATA_BUF_SZIE); if (WSARecv(SocketInfo->Socket, &(SocketInfo->DataBuf), 1, &RecvBytes, &Flags, NULL, NULL) == SOCKET_ERROR) { if (WSAGetLastError() != WSAEWOULDBLOCK) { printf( "WSARecv() failed with error %d\n" , WSAGetLastError()); FreeSocketInformation(i); } continue ; } else { printf(SocketInfo->Buffer); if (RecvBytes == 0) { FreeSocketInformation(i); continue ; } } } } } SOCKET_CLEAN: if (listenSock != INVALID_SOCKET) { closesocket(listenSock); listenSock = INVALID_SOCKET; } if (acceptSock != INVALID_SOCKET) { closesocket(acceptSock); acceptSock = INVALID_SOCKET; } WSACleanup(); return 0; } |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 | // CommonClient.cpp : 定义控制台应用程序的入口点。 // #include "stdafx.h" #include <winsock2.h> #include <conio.h> #pragma comment(lib,"ws2_32") #define SOCKET_MAXCNT 64 #define THREAD_NUM 22 #define CONNECT_PORT 8773 #define CONNECT_ADDR "127.0.0.1" DWORD WINAPI SocketFunc(LPVOID pM) { char buf[512] = {0}; SOCKET sock; int result; struct sockaddr_in cli_addr; int num = *( int *)pM; sock = socket(AF_INET,SOCK_STREAM,0); cli_addr.sin_family = AF_INET; cli_addr.sin_port = htons(CONNECT_PORT); cli_addr.sin_addr.s_addr = inet_addr(CONNECT_ADDR); result = connect(sock,( struct sockaddr*)&cli_addr, sizeof (cli_addr)); if (result == SOCKET_ERROR) { printf( "%d thread: connect error\n" ,num); exit(1); } while (1) { sprintf(buf, "%d thread: %s" ,num, "hello world!\n" ); printf(buf); result = send(sock,buf,100,0); if (result == SOCKET_ERROR) { printf( "%d error \n" ,num); return 0; } Sleep(3000); } exit(1); return 0; } int _tmain( int argc, _TCHAR* argv[]) { int num[SOCKET_MAXCNT] = {0}; WSADATA wsa_data; WSAStartup(WINSOCK_VERSION,&wsa_data); // 初始化数组 传递给线程 区别各个线程号 for ( int i = 0 ;i < SOCKET_MAXCNT;i++) { num[i] = i; } for ( int i = 0; i < THREAD_NUM;i++) { HANDLE hThread = CreateThread(NULL,0,SocketFunc,&num[i],0,NULL); if (hThread != NULL) { CloseHandle(hThread); } } while (1) { Sleep(100000); } return 0; } |
作 者: itdef
欢迎转帖 请保持文本完整并注明出处
技术博客 http://www.cnblogs.com/itdef/
B站算法视频题解
https://space.bilibili.com/18508846
qq 151435887
gitee https://gitee.com/def/
欢迎c c++ 算法爱好者 windows驱动爱好者 服务器程序员沟通交流
如果觉得不错,欢迎点赞,你的鼓励就是我的动力
欢迎转帖 请保持文本完整并注明出处
技术博客 http://www.cnblogs.com/itdef/
B站算法视频题解
https://space.bilibili.com/18508846
qq 151435887
gitee https://gitee.com/def/
欢迎c c++ 算法爱好者 windows驱动爱好者 服务器程序员沟通交流
如果觉得不错,欢迎点赞,你的鼓励就是我的动力


【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· go语言实现终端里的倒计时
· 如何编写易于单元测试的代码
· 10年+ .NET Coder 心语,封装的思维:从隐藏、稳定开始理解其本质意义
· .NET Core 中如何实现缓存的预热?
· 从 HTTP 原因短语缺失研究 HTTP/2 和 HTTP/3 的设计差异
· 分享一个免费、快速、无限量使用的满血 DeepSeek R1 模型,支持深度思考和联网搜索!
· 基于 Docker 搭建 FRP 内网穿透开源项目(很简单哒)
· ollama系列1:轻松3步本地部署deepseek,普通电脑可用
· 按钮权限的设计及实现
· 25岁的心里话