网络编程基础：使用SELECT模型

【TCP】使用Select-服务端

Code:

1. //   
2. //  CopyRight(c) 2009, YOYO, All Rights Reserved.   
3. //  Author: LIN YiQian   
4. //  Created: 2009/09/17   
5. //  Describe: TCP Select 服务端   
6. //   
7. #include <WinSock2.h>   
8. #include <iostream>   
9. #include <set>   
10.   
11. #pragma comment(lib, "ws2_32.lib")   
12.   
13. typedef std::set<SOCKET> SOCKET_SET;   
14.   
15. void main(void)   
16. {   
17.     WSADATA wsaData;   
18.   
19.     if (WSAStartup( MAKEWORD(2, 2), &wsaData) != 0)   
20.     {   
21.         return;   
22.     }   
23.   
24.     if (LOBYTE(wsaData.wVersion) != 2 || HIBYTE(wsaData.wVersion) != 2)   
25.     {   
26.         WSACleanup();   
27.         return;   
28.     }   
29.   
30.     SOCKET  socketServer = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);   
31.   
32.     SOCKADDR_IN addrServer;   
33.     addrServer.sin_addr.S_un.S_addr = htonl(INADDR_ANY);   
34.     addrServer.sin_port = htons(9989);   
35.     addrServer.sin_family = AF_INET;   
36.   
37.     bind(socketServer, (SOCKADDR*)&addrServer, sizeof(SOCKADDR));   
38.        
39.     listen(socketServer, 4);   
40.   
41.     SOCKET          socketClient;   
42.     struct fd_set   fdRead;   
43.     int             nSize;   
44.     SOCKET_SET      setSockets;   
45.   
46.     char    szMsg[128];   
47.     ZeroMemory(szMsg, sizeof(szMsg));   
48.   
49.     int nBlockMilli = 0;   
50.     struct timeval timeout;   
51.     timeout.tv_sec = 0;   
52.     timeout.tv_usec = 100;   
53.   
54.     while (true)   
55.     {   
56.         FD_ZERO(&fdRead);       //  初始化fd_set   
57.         FD_SET(socketServer, &fdRead);  //  分配套接字句柄到对应的fd_set   
58.   
59.         select(FD_SETSIZE, &fdRead, NULL, NULL, &timeout);   
60.         nSize = sizeof(addrServer);   
61.   
62.         if (FD_ISSET(socketServer, &fdRead))    //  如果套接字句柄还在fd_set里，说明   
63.                                                 //  客户端已有connect的请求，马上就可以accept   
64.         {   
65.             socketClient = accept(socketServer, (SOCKADDR*)&addrServer, &nSize);   
66.             setSockets.insert(socketClient);   
67.         }   
68.   
69.         FD_ZERO(&fdRead);   
70.         for (SOCKET_SET::iterator iterSet = setSockets.begin(); iterSet != setSockets.end(); iterSet++)   
71.         {   
72.             FD_SET(*iterSet, &fdRead);   
73.         }   
74.   
75.         select(FD_SETSIZE, &fdRead, NULL, NULL, &timeout);   
76.         for (SOCKET_SET::iterator iterSet = setSockets.begin(); iterSet != setSockets.end(); iterSet++)   
77.         {   
78.             if (FD_ISSET(*iterSet, &fdRead))   
79.             {   
80.                 recv(*iterSet, szMsg, strlen(szMsg) + 1, 0);   
81.                 std::cout << szMsg << std::endl;   
82.             }   
83.         }   
84.     }   
85.   
86.     for (SOCKET_SET::iterator iterSet = setSockets.begin(); iterSet != setSockets.end(); iterSet++)   
87.     {   
88.         closesocket(*iterSet);   
89.     }   
90.   
91.     closesocket(socketServer);   
92.   
93.     WSACleanup();   
94.   
95.     return;   
96. }  

【TCP】使用Select-客户端

Code:

1. //   
2. //  CopyRight(c) 2009, YOYO, All Rights Reserved.   
3. //  Author: LIN YiQian   
4. //  Created: 2009/09/17   
5. //  Describe: TCP Select 客户端   
6. //   
7. #include <WinSock2.h>   
8. #include <iostream>   
9. #include <ctime>   
10.   
11. #pragma comment(lib, "ws2_32.lib")   
12.   
13. DWORD WINAPI ClientSocketThread(LPVOID lpParameter);   
14.   
15. void main(void)   
16. {   
17.     WSADATA wsaData;   
18.   
19.     if (WSAStartup(MAKEWORD(2, 2), &wsaData) != 0)   
20.     {   
21.         return;   
22.     }   
23.   
24.     if (LOBYTE(wsaData.wVersion) != 2 || HIBYTE(wsaData.wVersion) != 2)   
25.     {   
26.         WSACleanup();   
27.         return;   
28.     }   
29.   
30.     SYSTEM_INFO systemInfo;   
31.     GetSystemInfo(&systemInfo);   
32.   
33.     //  开启CPU个数2倍的线程   
34.     for (int i = 0; i < (systemInfo.dwNumberOfProcessors * 2); ++i)   
35.     {   
36.         HANDLE handle = CreateThread(NULL, 0, ClientSocketThread, NULL, 0, NULL);   
37.         if (handle == NULL)   
38.         {   
39.             std::cout << "创建Client线程失败!" << std::endl;   
40.         }   
41.         CloseHandle(handle);   
42.     }   
43.   
44.     while (true)   
45.     {   
46.     }   
47.   
48.     WSACleanup();   
49.   
50.     return;   
51. }   
52.   
53. DWORD WINAPI ClientSocketThread(LPVOID lpParameter)   
54. {   
55.     SOCKET socketClient = socket(AF_INET, SOCK_STREAM, 0);   
56.   
57.     SOCKADDR_IN addrServer;   
58.     addrServer.sin_addr.S_un.S_addr = inet_addr("127.0.0.1");   
59.     addrServer.sin_port = htons(9989);   
60.     addrServer.sin_family = AF_INET;   
61.   
62.     connect(socketClient, (SOCKADDR*)&addrServer, sizeof(SOCKADDR));   
63.   
64.     char szMsg[100];   
65.     srand(time(NULL));   
66.   
67.     while (true)   
68.     {   
69.         sprintf(szMsg, "Socket %d connect../n", socketClient);   
70.         send(socketClient, szMsg, strlen(szMsg) + 1, 0);   
71.   
72.         Sleep(100);   
73.     }   
74.   
75.     closesocket(socketClient);   
76. }