WinSocket模型的探讨——select模型，认为这篇文章很不错。因此转载了。

360737395

查找了很多资料都找不到select模型的详细用法，《Windows网络必威官方》这本书上也只是写了一个简单的回应服务器，就连writefds的用法都没讲，也不知道什么时候利用“可写”来发文件。这些都是我的疑问，相信很多研究网络必威官方的同路人也碰到了我的这些问题。这些疑问在这篇文章中都解决了！耗费了偶很多的精力去猜测去思考！

游客，如果您要查看本帖隐藏内容请回复

henry_sea

沙发，支持楼主~~

zuiaixifan

网络必威官方是个难点~

360737395

就是难才好玩。。这篇文章还讲了 文件共享的问题。。

self001

  多谢分享

mengmengyu

瞧一瞧，是不是真的有那么好

flyboy

貌似不错。。。

tom1987

什么文章这么好？

22228971

谢谢 并且拿走了

jhlrxw

我先看一下 学习

xianni

沙发，支持楼主~~

360737395

/**********************************************************************************************************************/

第二个文件

/*/
文件：SocketClient.cpp

说明：
此文件是作为测试的客户端，实现了登录和取文件的功能。
和服务端的交互就是采用了发送命令、数据长度，然后发送具体的数据这样的顺序。
详细可看服务端的说明。

基本逻辑是这样的，客户端要先登录服务端，然后登录成功之后，才能进行相应的操作。

错误处理做得不太好，以后再补充了。

其他的如注释，结构，命名等的编码规范都用了个人比较喜欢的方式。

输出：
..\Bin\SocketClient.exe

用法：
可以 SocketClient Server_IP
或者直接启动SocketClient，会提示你输入服务端的IP

Todo:
下一步首先完成各个SOCKET的模型，然后公开自己的研究代码。
功能方面就是：
1、服务器可以指定共享文件夹
2、客户端可以列出服务器共享了哪些文件
3、客户端可以列出哪些用户在线，并可以发命令和其他用户聊天
/*/

1. 
   
2. #include <winsock2.h>
   
3. #pragma comment(lib, "WS2_32")
   
4. 
   
5. #include <iostream>
   
6. using namespace std;
   
7. 
   
8. #include <stdlib.h>
   
9. 
   
10. #include "..\Common Include\CommonSocket.h"
    
11. #include "..\Common Include\CommonCmd.h"
    
12. 
    
13. bool g_bAuth = false;
    
14. 
    
15. void GetFile(SOCKET sock);
    
16. bool Auth(SOCKET sock, char *szName, char *szPwd);
    
17. bool RegisterUser(SOCKET sock, char *szName, char *szPwd);
    
18. 
    
19. 
    
20. ///////////////////////////////////////////////////////////////////////
    
21. //
    
22. // 函数名       : Usage
    
23. // 功能描述     : 提示程序用法
    
24. // 返回值       : void
    
25. //
    
26. ///////////////////////////////////////////////////////////////////////
    
27. void Usage()
    
28. {
    
29. printf("*******************************************\n");
    
30. printf("Socket Client                            \n");
    
31. printf("Written by DYL                         \n");
    
32. printf("Email: dylgsy@163.com                 \n");
    
33. printf("Usage: SocketClient.exe Server_IP          \n");
    
34. printf("*******************************************\n");
    
35. }
    
36. 
    
37. 
    
38. ///////////////////////////////////////////////////////////////////////
    
39. //
    
40. // 函数名       : Menu
    
41. // 功能描述     : 选择服务的界面
    
42. // 返回值       : void
    
43. //
    
44. ///////////////////////////////////////////////////////////////////////
    
45. void Menu()
    
46. {
    
47. system("cls");
    
48. printf("********************************************\n");
    
49. printf("请选择操作：        \n\n");
    
50. printf("1、取得文件         \n");
    
51. printf("2、退出          \n");
    
52. printf("********************************************\n");
    
53. }
    
54. 
    
55. 
    
56. ///////////////////////////////////////////////////////////////////////
    
57. //
    
58. // 函数名       : LoginMenu
    
59. // 功能描述     : 用户登录的界面
    
60. // 返回值       : void
    
61. //
    
62. ///////////////////////////////////////////////////////////////////////
    
63. void LoginMenu()
    
64. {
    
65. cout << "请按任意键继续操作．" <<endl;
    
66. getchar();
    
67. system("cls");
    
68. printf("********************************************\n");
    
69. printf("请选择操作：        \n\n");
    
70. printf("1、登录          \n");
    
71. printf("2、注册          \n");
    
72. printf("3、退出          \n");
    
73. printf("********************************************\n");
    
74. }
    
75. 
    
76. 
    
77. ///////////////////////////////////////////////////////////////////////
    
78. //
    
79. // 函数名       : Login
    
80. // 功能描述     : 用户登录的界面逻辑
    
81. // 参数         : SOCKET sock
    
82. // 返回值       : bool
    
83. //
    
84. ///////////////////////////////////////////////////////////////////////
    
85. bool Login(SOCKET sock)
    
86. {
    
87. bool bGoOn = true;
    
88. while(bGoOn)
    
89. {
    
90.   LoginMenu();
    
91.   int nChoose = 0;
    
92.   cin >> nChoose;
    
93. 
    
94.   char szName[10];
    
95.   char szPwd[20];
    
96.   char szConfirmPwd[20];
    
97.   memset(szName, 0, 10);
    
98.   memset(szPwd, 0, 20);
    
99.   memset(szConfirmPwd, 0, 20);
    
100. 
     
101.   bool bGoOnLogin = true;
     
102. 
     
103.   switch(nChoose)
     
104.   {
     
105.   case 1:
     
106.    while(bGoOnLogin)
     
107.    {
     
108.     cout << "请输入你的用户名：";
     
109.     cin >> szName;
     
110.     cout << "请输入你的密码：";
     
111.     cin >> szPwd;
     
112.     if(Auth(sock, szName, szPwd))
     
113.     {
     
114.      return true;
     
115.     }
     
116.     else
     
117.     {
     
118.      char c;
     
119.      cout << "继续登录？y/n" << endl;
     
120.      cin >> c;
     
121.      switch(c)
     
122.      {
     
123.      case 'y':
     
124.       bGoOnLogin = true;
     
125.       break;
     
126.      case 'n':
     
127.       bGoOnLogin = false;
     
128.       break;
     
129.      default:
     
130.       break;
     
131.      }
     
132.     }
     
133.    }
     
134.    break;
     
135.    
     
136.   case 2:
     
137.    cout << "请输入你的用户名：";
     
138.    cin >> szName;
     
139.    cout << "请输入你的密码：";
     
140.    cin >> szPwd;
     
141.    cout << "请再次输入你的密码：";
     
142.    cin >> szConfirmPwd;
     
143.    if(strcmp(szPwd, szConfirmPwd) != 0)
     
144.    {
     
145.     cout << "前后密码不一致" << endl;
     
146.    }
     
147.    else
     
148.    {
     
149.     if(!RegisterUser(sock, szName, szPwd))
     
150.     {
     
151.      cout << "注册用户失败!" << endl;
     
152.     }
     
153.    }
     
154.    break;
     
155. 
     
156.   case 3:
     
157.    bGoOn = false;
     
158.    return false;
     
159.   default:
     
160.    break;
     
161.   }
     
162. }
     
163. 
     
164. return false;
     
165. }
     
166. 
     
167. void main(int argc, char *argv[])
     
168. {
     
169. system("cls");
     
170. char szServerIP[20];
     
171. memset(szServerIP, 0, 20);
     
172. 
     
173. if(argc != 2)
     
174. {
     
175.   cout << "请输入服务器IP：";
     
176.   cin >> szServerIP;
     
177. }
     
178. else
     
179. {
     
180.   strcpy(szServerIP, argv[1]);
     
181. }
     
182. InitWinsock();
     
183. SOCKET sockServer;
     
184. sockServer = ConnectServer(szServerIP, PORT, 1);
     
185. if(sockServer == NULL)
     
186. {
     
187.   OutErr("连接服务器失败！");
     
188.   return;
     
189. }
     
190. else
     
191. {
     
192.   OutMsg("已和服务器建立连接！");
     
193. }
     
194. 
     
195. // 要求用户登录
     
196. if(!Login(sockServer))
     
197.   return;
     
198. 
     
199. // 登录成功，让用户选择服务
     
200. int nChoose = 0;
     
201. bool bExit = false;
     
202. while(!bExit)
     
203. {
     
204.   Menu();
     
205.   cin >> nChoose;
     
206.   switch(nChoose)
     
207.   {
     
208.   case 1:  // 获取文件
     
209.    GetFile(sockServer);
     
210.    break;
     
211.   case 2:
     
212.    bExit = true;
     
213.    break;   
     
214.   default:
     
215.    break;
     
216.   }
     
217. }
     
218. shutdown(sockServer, SD_BOTH);
     
219. closesocket(sockServer);
     
220. 
     
221. }
     
222. 
     
223. 
     
224. ///////////////////////////////////////////////////////////////////////
     
225. //
     
226. // 函数名       : Auth
     
227. // 功能描述     : 用户登录认证
     
228. // 参数         : SOCKET sock
     
229. // 参数         : char *szName
     
230. // 参数         : char *szPwd
     
231. // 返回值       : bool
     
232. //
     
233. ///////////////////////////////////////////////////////////////////////
     
234. bool Auth(SOCKET sock, char *szName, char *szPwd)
     
235. {
     
236. char szCmd[50];
     
237. memset(szCmd, 0, 50);
     
238. strcpy(szCmd, szName);
     
239. strcat(szCmd, " ");
     
240. strcat(szCmd, szPwd);
     
241. 
     
242. SCommand cmd;
     
243. cmd.CommandID = CMD_AUTHEN;
     
244. cmd.DataSize = strlen(szCmd);
     
245. 
     
246. int nRet;
     
247. nRet = SendFix(sock, (char *)&cmd, sizeof(cmd));
     
248. if(nRet == SOCKET_ERROR)
     
249. {
     
250.   OutErr("SendFix() failed!");
     
251.   return false;
     
252. }
     
253. else
     
254. {
     
255.   SendFix(sock, szCmd, strlen(szCmd));
     
256.   char szBuf[10];
     
257.   memset(szBuf, 0, 10);
     
258.   recv(sock, szBuf, 10, 0);
     
259.   if(strcmp(szBuf, "UP OK!") == 0)
     
260.   {
     
261.    cout << "登录成功。" << endl;
     
262.    g_bAuth = true;
     
263.   }
     
264.   else if(strcmp(szBuf, "U Err!") == 0)
     
265.   {
     
266.    cout << "此用户不存在。" << endl;
     
267.    g_bAuth = false;
     
268.   }
     
269.   else if(strcmp(szBuf, "P Err!") == 0)
     
270.   {
     
271.    cout << "密码错误。" << endl;
     
272.    g_bAuth = false;
     
273.   }
     
274. }
     
275. return g_bAuth;
     
276. }
     
277. 
     
278. 
     
279. ///////////////////////////////////////////////////////////////////////
     
280. //
     
281. // 函数名       : GetFile
     
282. // 功能描述     : 取得服务器的文件
     
283. // 参数         : SOCKET sock
     
284. // 返回值       : void
     
285. //
     
286. ///////////////////////////////////////////////////////////////////////
     
287. void GetFile(SOCKET sock)
     
288. {
     
289. if(!g_bAuth)
     
290. {
     
291.   OutMsg("用户还没登录！请先登录");
     
292.   return;
     
293. }
     
294. 
     
295. char szSrcFile[MAX_PATH];
     
296. char szDstFile[MAX_PATH];
     
297. memset(szSrcFile, 0, MAX_PATH);
     
298. memset(szDstFile, 0, MAX_PATH);
     
299. 
     
300. cout << "你要取得Server上的文件：";
     
301. cin >> szSrcFile;
     
302. 
     
303. cout << "你要把文件存在哪里：";
     
304. cin >> szDstFile;
     
305. 
     
306. SCommand cmd;
     
307. cmd.CommandID = CMD_GETFILE;
     
308. cmd.DataSize = strlen(szSrcFile);
     
309. 
     
310. // 发送命令
     
311. SendFix(sock, (char *)&cmd, sizeof(cmd));
     
312. 
     
313. // 发送文件名
     
314. SendFix(sock, szSrcFile, strlen(szSrcFile));
     
315. 
     
316. // 接收文件长度
     
317. DWORD dwFileSize = 0;
     
318. RecvFix(sock, (char*)&dwFileSize, sizeof(dwFileSize));
     
319. 
     
320. if(dwFileSize == 0)
     
321. {
     
322.   OutMsg("文件不存在");
     
323.   return;
     
324. }
     
325. 
     
326. // 接收文件内容
     
327. DWORD dwLeft = dwFileSize;
     
328. char szBuf[1024];
     
329. HANDLE hFile = CreateFile(szDstFile, GENERIC_WRITE, FILE_SHARE_READ,
     
330.   NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL);
     
331. if(hFile == INVALID_HANDLE_VALUE)
     
332. {
     
333.   hFile = CreateFile(szDstFile, GENERIC_WRITE, FILE_SHARE_READ,
     
334.    NULL, CREATE_NEW, FILE_ATTRIBUTE_NORMAL, NULL);
     
335.   if(hFile == INVALID_HANDLE_VALUE)
     
336.   {
     
337.    OutErr("CreateFile failed!");
     
338.    return;
     
339.   }
     
340. }
     
341. while(dwLeft > 0)
     
342. {
     
343.   memset(szBuf, 0, 1024);
     
344.   // 这里是不确定文件内容的，所以要用recv，不能用RecvFix
     
345.   int nRead = recv(sock, szBuf, 1024, 0);
     
346.   if(nRead == SOCKET_ERROR)
     
347.    OutErr("RecvFix Error!");
     
348. 
     
349.   DWORD dwWritten = 0;
     
350.   if(!WriteFile(hFile, szBuf, nRead, &dwWritten, NULL))
     
351.   {
     
352.    OutErr("WriteFile error!");
     
353.    return;
     
354.   }
     
355.   dwLeft -= dwWritten;
     
356. }
     
357. 
     
358. CloseHandle(hFile);
     
359. 
     
360. OutMsg("接收文件成功！");
     
361. }
     
362. 
     
363. 
     
364. ///////////////////////////////////////////////////////////////////////
     
365. //
     
366. // 函数名       : RegisterUser
     
367. // 功能描述     : 注册新用户
     
368. // 参数         : SOCKET sock
     
369. // 参数         : char *szName
     
370. // 参数         : char *szPwd
     
371. // 返回值       : bool
     
372. //
     
373. ///////////////////////////////////////////////////////////////////////
     
374. bool RegisterUser(SOCKET sock, char *szName, char *szPwd)
     
375. {
     
376. char szCmd[50];
     
377. memset(szCmd, 0, 50);
     
378. strcpy(szCmd, szName);
     
379. strcat(szCmd, " ");
     
380. strcat(szCmd, szPwd);
     
381. 
     
382. SCommand cmd;
     
383. cmd.CommandID = CMD_REGISTER;
     
384. cmd.DataSize = strlen(szCmd);
     
385. 
     
386. // 发送命令
     
387. int nRet = SendFix(sock, (char *)&cmd, sizeof(cmd));
     
388. if(nRet == SOCKET_ERROR)
     
389. {
     
390.   OutErr("SendFix() failed!");
     
391.   return false;
     
392. }
     
393. else
     
394. {
     
395.   // 发送用户名和密码串
     
396.   SendFix(sock, szCmd, strlen(szCmd));
     
397.   char szBuf[10];
     
398.   memset(szBuf, 0, 10);
     
399.   
     
400.   recv(sock, szBuf, 10, 0);
     
401.   if(strcmp(szBuf, "REG OK!") == 0)
     
402.   {
     
403.    cout << "注册成功。" << endl;
     
404.    return true;
     
405.   }
     
406.   else if(strcmp(szBuf, "REG ERR!") == 0)
     
407.   {
     
408.    cout << "注册失败." << endl;
     
409.    return false;
     
410.   }
     
411. }
     
412. 
     
413. return false;
     
414. }
     

复制代码

360737395

/**********************************************************************************************************************/

第三个文件，公用的

/*/
文件: CommonSocket.h
说明:
实现了服务端和客户端一些公用的函数!
/*/

1. 
   
2. #ifndef __COMMONSOCKET_H__
   
3. #define __COMMONSOCKET_H__
   
4. 
   
5. #include <iostream>
   
6. using namespace std;
   
7. 
   
8. #define OutErr(a) cout << (a) << endl \
   
9.       << "出错代码：" << WSAGetLastError() << endl \
   
10.       << "出错文件：" << __FILE__ << endl  \
    
11.       << "出错行数：" << __LINE__ << endl \
    
12. 
    
13. #define OutMsg(a) cout << (a) << endl;
    
14. 
    
15. ///////////////////////////////////////////////////////////////////////
    
16. //
    
17. // 函数名       : InitWinsock
    
18. // 功能描述     : 初始化WINSOCK
    
19. // 返回值       : void
    
20. //
    
21. ///////////////////////////////////////////////////////////////////////
    
22. void InitWinsock()
    
23. {
    
24. // 初始化WINSOCK
    
25. WSADATA wsd;
    
26. if( WSAStartup(MAKEWORD(2, 2), &wsd) != 0)
    
27. {
    
28.   OutErr("WSAStartup()");
    
29. }
    
30. }
    
31. 
    
32. ///////////////////////////////////////////////////////////////////////
    
33. //
    
34. // 函数名       : ConnectServer
    
35. // 功能描述     : 连接SERVER
    
36. // 参数         : char *lpszServerIP IP地址
    
37. // 参数         : int nPort    端口
    
38. // 返回值       : SOCKET    SERVER 的 Socket
    
39. //
    
40. ///////////////////////////////////////////////////////////////////////
    
41. SOCKET ConnectServer(char *lpszServerIP, int nPort, ULONG NonBlock)
    
42. {
    
43. SOCKET sServer = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
    
44. //ioctlsocket(sServer, FIONBIO, &NonBlock);
    
45. 
    
46. struct hostent *pHost = NULL;
    
47. struct sockaddr_in servAddr;
    
48. servAddr.sin_family = AF_INET;
    
49. servAddr.sin_port = htons(nPort);
    
50. servAddr.sin_addr.s_addr = inet_addr(lpszServerIP);
    
51. 
    
52. 
    
53. // 如果给的是主机的名字而不是IP地址
    
54. if(servAddr.sin_addr.s_addr == INADDR_NONE)
    
55. {
    
56.   pHost = gethostbyname( lpszServerIP );
    
57.   if(pHost == NULL)
    
58.   {
    
59.    OutErr("gethostbyname Failed!");
    
60.    return NULL;
    
61.   }
    
62.   memcpy(&servAddr.sin_addr, pHost->h_addr_list[0], pHost->h_length);
    
63. }
    
64. 
    
65. int nRet = 0;
    
66. nRet = connect(sServer, (struct sockaddr *)&servAddr, sizeof(servAddr));
    
67. if( nRet == SOCKET_ERROR )
    
68. {
    
69.   OutErr("connect failed!");
    
70.   return NULL;
    
71. }
    
72.   
    
73. return sServer;
    
74. }
    
75. 
    
76. ///////////////////////////////////////////////////////////////////////
    
77. //
    
78. // 函数名       : BindServer
    
79. // 功能描述     : 绑定端口
    
80. // 参数         : int nPort
    
81. // 返回值       : SOCKET
    
82. //
    
83. ///////////////////////////////////////////////////////////////////////
    
84. SOCKET BindServer(int nPort)
    
85. {
    
86. // 创建socket
    
87. SOCKET sServer = socket(AF_INET, SOCK_STREAM, 0);
    
88. 
    
89. // 绑定端口
    
90. struct sockaddr_in servAddr;
    
91. servAddr.sin_family = AF_INET;
    
92. servAddr.sin_port = htons(nPort);
    
93. servAddr.sin_addr.s_addr = htonl(INADDR_ANY);
    
94. 
    
95. if(bind(sServer, (struct sockaddr *)&servAddr, sizeof(servAddr)) < 0)
    
96. {
    
97.   OutErr("bind Failed!");
    
98.   return NULL;
    
99. }
    
100. 
     
101. // 设置监听队列为200
     
102. if(listen(sServer, 200) != 0)
     
103. {
     
104.   OutErr("listen Failed!");
     
105.   return NULL;
     
106. }
     
107. return sServer;
     
108. }
     
109. 
     
110. 
     
111. ///////////////////////////////////////////////////////////////////////
     
112. //
     
113. // 函数名       : AcceptClient
     
114. // 功能描述     :
     
115. // 参数         : SOCKET sServer [in]
     
116. // 参数         : LPSTR lpszIP  [out] 返回客户端的IP地址
     
117. // 返回值       : SOCKET   [out] 返回客户端的socket
     
118. //
     
119. ///////////////////////////////////////////////////////////////////////
     
120. SOCKET AcceptClient(SOCKET sListen, LPSTR lpszIP)
     
121. {
     
122. struct sockaddr_in cliAddrTmp;
     
123. int cliAddrSize = sizeof(struct sockaddr_in);
     
124. SOCKET sClient = accept(sListen, (struct sockaddr *)&cliAddrTmp, &cliAddrSize);
     
125. if(sClient == INVALID_SOCKET)
     
126. {
     
127.   OutErr("accept failed!");
     
128.   return NULL;
     
129. }
     
130. sprintf(lpszIP, "%s", inet_ntoa(cliAddrTmp.sin_addr));
     
131. 
     
132. return sClient;
     
133. }
     
134. 
     
135. ///////////////////////////////////////////////////////////////////////
     
136. //
     
137. // 函数名       : RecvFix
     
138. // 功能描述     : 接收指定长度的数据，考虑非阻塞socket的情况
     
139. // 参数         : SOCKET socket [in]
     
140. // 参数         : char *data [in]
     
141. // 参数         : DWORD len  [in]
     
142. // 参数         : DWORD *retlen [out]
     
143. // 返回值       : bool
     
144. //
     
145. ///////////////////////////////////////////////////////////////////////
     
146. int RecvFix(SOCKET socket, char *data, DWORD len)
     
147. {
     
148. int retlen = 0;
     
149. int nLeft = len;
     
150. int nRead = 0;
     
151. char *pBuf = data;
     
152. while(nLeft > 0)
     
153. {
     
154.   nRead = recv(socket, pBuf, nLeft, 0);
     
155.   if(nRead == SOCKET_ERROR || nRead == 0)
     
156.   {
     
157.    if(WSAEWOULDBLOCK == WSAGetLastError())
     
158.     continue;
     
159.    else
     
160.     return nRead;
     
161.   }
     
162.   
     
163.   nLeft -= nRead;
     
164.   retlen += nRead;
     
165.   pBuf += nRead;
     
166. }
     
167. return nRead;
     
168. }
     
169. 
     
170. 
     
171. ///////////////////////////////////////////////////////////////////////
     
172. //
     
173. // 函数名       : SendFix
     
174. // 功能描述     : 发送指定长度的数据，考虑非阻塞socket的情况
     
175. // 参数         : SOCKET socket
     
176. // 参数         : char *data
     
177. // 参数         : DWORD len
     
178. // 参数         : DWORD *retlen
     
179. // 返回值       : bool
     
180. //
     
181. ///////////////////////////////////////////////////////////////////////
     
182. int SendFix(SOCKET socket, char *data, DWORD len)
     
183. {
     
184. int retlen = 0;
     
185. int nLeft = len;
     
186. int nWritten = 0;
     
187. const char *pBuf = data;
     
188. while(nLeft > 0)
     
189. {
     
190.   nWritten = send(socket, data, nLeft, 0);
     
191.   if(nWritten == SOCKET_ERROR || nWritten == 0)
     
192.   {
     
193.    if(WSAEWOULDBLOCK == WSAGetLastError())
     
194.     continue;
     
195.    else
     
196.     return nWritten;
     
197.   }
     
198. 
     
199.   
     
200.   nLeft -= nWritten;
     
201.   retlen += nWritten;
     
202.   pBuf += nWritten;
     
203. }
     
204. return nWritten;
     
205. }
     
206. 
     
207. 
     
208. /*
     
209. ///////////////////////////////////////////////////////////////////////
     
210. //
     
211. // 函数名       : SelectSend
     
212. // 功能描述     : 使用select模型来发送数据，没完成，所以注释掉了
     
213. // 参数         : SOCKET sock
     
214. // 参数         : FD_SET *wfds
     
215. // 参数         : char *data
     
216. // 参数         : DWORD len
     
217. // 返回值       : bool
     
218. //
     
219. ///////////////////////////////////////////////////////////////////////
     
220. bool SelectSend(SOCKET sock, FD_SET *wfds, char *data, DWORD len)
     
221. {
     
222. FD_ZERO(wfds);
     
223. FD_SET(sock, wfds);
     
224. 
     
225. if(select(0, NULL, wfds, NULL, NULL) == SOCKET_ERROR)
     
226. {
     
227.   OutErr("select() Failed!");
     
228.   return false;
     
229. }
     
230. // 如果是可以写的SOCKET，就一直写，直到返回WSAEWOULDBLOCK
     
231. if( FD_ISSET(sock, wfds) )
     
232. {
     
233.   int nLeft = len;
     
234.   while(nLeft > 0)
     
235.   {
     
236.    int nRet = send(sock, data, len, 0);
     
237.    if(nRet == SOCKET_ERROR)
     
238.     return false;
     
239.    nLeft -= nRet;
     
240.   }
     
241. }
     
242. 
     
243. return true;
     
244. }
     
245. 
     
246. 
     
247. ///////////////////////////////////////////////////////////////////////
     
248. //
     
249. // 函数名       : SelectRecv
     
250. // 功能描述     : 使用select模型来接收数据，没完成，所以注释掉了
     
251. // 参数         : SOCKET sock
     
252. // 参数         : FD_SET *rfds
     
253. // 参数         : char *data
     
254. // 参数         : DWORD len
     
255. // 返回值       : bool
     
256. //
     
257. ///////////////////////////////////////////////////////////////////////
     
258. bool SelectRecv(SOCKET sock, FD_SET *rfds, char *data, DWORD len)
     
259. {
     
260. FD_ZERO(rfds);
     
261. FD_SET(sock, rfds);
     
262. 
     
263. if(select(0, rfds, NULL, NULL, NULL) == SOCKET_ERROR)
     
264. {
     
265.   OutErr("select() Failed!");
     
266.   return false;
     
267. }
     
268. 
     
269. if( FD_ISSET(sock, rfds) )
     
270. {
     
271.   int nLeft = len;
     
272.   while(nLeft > 0)
     
273.   {
     
274.    int nRet = recv(sock, data, len, 0);
     
275.    if(nRet == SOCKET_ERROR)
     
276.     return false;
     
277.    nLeft -= nRet;
     
278.   }
     
279. }
     
280. return true;
     
281. }
     
282. */
     
283. 
     
284. 
     
285. #endif //__COMMONSOCKET_H__
     

复制代码

/**********************************************************************************************************************/

第四个文件，公用的

/*/
文件: CommonCmd.h
说明:
实现了服务端和客户端一些公用的数据结构，所以服务端和客户端都要包含。
其中有命令、SOCKET的当前状态等的定义。
/*/

1. 
   
2. #ifndef __COMMONCMD_H__
   
3. #define __COMMONCMD_H__
   
4. 
   
5. #define PORT 5050
   
6. 
   
7. // 命令定义
   
8. #define CMD_AUTHEN 1 // 登录认证
   
9. #define CMD_GETFILE 2 // 获取文件
   
10. #define CMD_REGISTER 3  // 注册用户
    
11. 
    
12. typedef struct tagCommand
    
13. {
    
14. int CommandID;  // 命令ID
    
15. DWORD DataSize;  // 后接数据的大小
    
16. }SCommand;
    
17. 
    
18. // 标志目前的SOCKET该做什么
    
19. enum ECurOp
    
20. {RecvCmd, RecvData, ExecCmd};
    
21. 
    
22. 
    
23. #endif //__COMMONCMD_H__
    

复制代码

好了，上面四个文件都搞好了，有兴趣的朋友自己去弄吧，希望对你们有用，好累啊（写文章是一件很累的事情）。有什么问题欢迎探讨！

redaiyu302

xie xie lou zhu de fengxiang

xulindong

本帖隐藏的内容需要回复才可以浏览

klazhen

沙发，支持楼主~~

ryosga

学习一下

yangxingyu

来看下。。。。基础要看好

erhuasu

最近也正在找相关的文章看，来学习一下了，多谢楼主分享

lianghan12344

good!!!!!!!!!!!!!!!!!!