TCP 协议文件传输示例

一个简单的传输的例子 单线程 阻塞模式.  

 以下是发送端代码

send.cpp

1. //---------------------------------------------------------------------------  
3. #include <iostream>  
5. #include "winsock2.h"   
6.   
7. using namespace std;  
11. #pragma comment(lib , "ws2_32.lib")   
12.   
13. //---------------------------------------------------------------------------   
14.   
15.   
16.   
17. WSADATA wsaData;   
18.   
19. UINT uTimerID;   
20.   
21. DWORD MAX_SIZE = 1024 * 4;   
22.   
23. DWORD FinishSize = 0;   
24.   
25. DWORD SendSize = MAX_SIZE;   
26.   
27. LPSTR SFileName;   
28.   
29. DWORD FSize;   
30.   
31. DWORD DWSpeed=0;   
32.   
33. DWORD OldFinish=0;   
34.   
35.   
36.   
37. MMRESULT (WINAPI *ptimeSetEvent)(UINT uDelay, UINT uResolution, LPTIMECALLBACK fptc, DWORD dwUser, UINT fuEvent);   
38.   
39. MMRESULT (WINAPI *ptimeKillEvent)(UINT uTimerID);   
40.   
41. HINSTANCE DLLInst;   
42.   
43.   
44.   
45. bool init()   
46.   
47. {   
48.   
49.     DLLInst=LoadLibrary("winmm.dll");   
50.   
51.     if(DLLInst)   
52.   
53.     {   
54.   
55.         ptimeSetEvent = (MMRESULT (WINAPI *)(UINT uDelay, UINT uResolution, LPTIMECALLBACK fptc, DWORD dwUser, UINT fuEvent))GetProcAddress(DLLInst, "timeSetEvent");   
56.   
57.         if(!ptimeSetEvent) return false;   
58.   
59.         ptimeKillEvent = (MMRESULT (WINAPI *)(UINT uTimerID))GetProcAddress(DLLInst, "timeKillEvent");   
60.   
61.         if(!ptimeKillEvent) return false;   
62.   
63.         return true;   
64.   
65.     }   
66.   
67.     return false;   
68.   
69. }   
70.   
71.   
72.   
73. void Byte2Human(const DWORD & byte)   
74.   
75. {   
76.   
77.     if(byte < 1024){   
78.   
79.         cout << ((float)((int)(byte*100+0.5))/100.0) << "B";   
80.   
81.         return;   
82.   
83.     }   
84.   
85.     float kb = byte/1024;   
86.   
87.     if (kb<1024){   
88.   
89.         cout << ((float)((int)(kb*100+0.5))/100.0) << "KB";   
90.   
91.         return ;   
92.   
93.     }   
94.   
95.     float mb = kb/1024;   
96.   
97.     if (mb<1024){   
98.   
99.         cout << ((float)((int)(mb*100+0.5))/100.0) << "MB";   
100.   
101.         return;   
102.   
103.     }   
104.   
105.     float gb = mb/1024;   
106.   
107.     cout << ((float)((int)(gb*100+0.5))/100.0) << "GB";   
108.   
109. }   
110.   
111.   
112.   
113. void WINAPI TimeProc(UINT uID, UINT uMsg, DWORD dwUser, DWORD dw1, DWORD dw2)   
114.   
115. {   
116.   
117.     DWSpeed = FinishSize-OldFinish;   
118.   
119.     OldFinish = FinishSize;   
120.   
121.     cout << '/r';   
122.   
123.     cout << "                                                                              ";   
124.   
125.     cout << '/r';   
126.   
127.     cout << "文件名:" << SFileName << " " << FinishSize << "/" << FSize << " ";   
128.   
129.     Byte2Human(DWSpeed);   
130.   
131.     cout << "/S";   
132.   
133. }   
134.   
135.   
136.   
137. int main(int argc, char* argv[])   
138.   
139. {   
140.   
141.     if(!init())   
142.   
143.     {   
144.   
145.         cout<<"初始化失败"<<endl;   
146.   
147.         return 0;   
148.   
149.     }   
150.   
151.     int ret;   
152.   
153.     if(argc < 4)   
154.   
155.     {   
156.   
157.         cout << "App.exe Host Port FileName" << endl;   
158.   
159.         return 0;   
160.   
161.     }   
162.   
163.   
164.   
165.     if((ret=WSAStartup(0x0202, &wsaData))!=0)   
166.   
167.     {   
168.   
169.         cout << "WSAStartup Failed!! Code:" << ret << endl;   
170.   
171.         return 0;   
172.   
173.     }   
174.   
175.     SOCKET sock;   
176.   
177.     struct sockaddr_in server;   
178.   
179.     sock=socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);   
180.   
181.     if(sock==INVALID_SOCKET)   
182.   
183.     {   
184.   
185.         cout << "创建socket出错" << endl;   
186.   
187.         return 0;   
188.   
189.     }   
190.   
191.     server.sin_family=AF_INET;   
192.   
193.     server.sin_port=htons(atoi(argv[2]));   
194.   
195.     server.sin_addr.s_addr=inet_addr(argv[1]);   
196.   
197.     cout << "正在连接 " << argv[1] << ":" << argv[2] << endl;   
198.   
199.     if(connect(sock, (struct sockaddr *)&server, sizeof(server))==SOCKET_ERROR)   
200.   
201.     {   
202.   
203.         cout << "连接失败" << endl;   
204.   
205.         return 0;   
206.   
207.     }   
208.   
209.     cout << "连接成功" << endl;   
210.   
211.   
212.   
213.     //打开文件准备发送   
214.   
215.     HANDLE hFile;   
216.   
217.     OFSTRUCT ofs;   
218.   
219.     HANDLE hMap;   
220.   
221.     void *pVoid;   
222.   
223.     SFileName = argv[3];   
224.   
225.     hFile=(HANDLE)OpenFile(argv[3], &ofs, OF_READWRITE | OF_SHARE_DENY_NONE);   
226.   
227.     if((HFILE)hFile==HFILE_ERROR)   
228.   
229.     {   
230.   
231.         cout << "打开文件失败 ErrorCode=" << GetLastError() << endl;   
232.   
233.         return 0;   
234.   
235.     }   
236.   
237.     FSize=GetFileSize((HANDLE)hFile, NULL);   
238.   
239.     hMap=CreateFileMapping(hFile, NULL, PAGE_READWRITE, 0, 0, NULL);   
240.   
241.     if(hMap==NULL)   
242.   
243.     {   
244.   
245.         cout << "打开映射失败 ErrorCode=" << GetLastError() << endl;   
246.   
247.         return 0;   
248.   
249.     }   
250.   
251.     pVoid=MapViewOfFile(hMap, FILE_MAP_ALL_ACCESS, 0, 0, 0);   
252.   
253.     if(pVoid==NULL)   
254.   
255.     {   
256.   
257.         cout << "映射数据失败 ErrorCode=" << GetLastError() << endl;   
258.   
259.         return 0;   
260.   
261.     }   
262.   
263.     char *p = (char *)pVoid;   
264.   
265.     cout << "正在发送文件 " << argv[3] << endl;   
266.   
267.     uTimerID=ptimeSetEvent(1000, 0, (LPTIMECALLBACK)TimeProc, NULL, TIME_PERIODIC|TIME_CALLBACK_FUNCTION);   
268.   
269.   
270.   
271.     while(1)   
272.   
273.     {   
274.   
275.         if(FSize - FinishSize < MAX_SIZE)   
276.   
277.         {   
278.   
279.             SendSize = FSize - FinishSize;   
280.   
281.         }   
282.   
283.         int ret = send(sock, p+FinishSize, SendSize, 0);   
284.   
285.         if(ret == SOCKET_ERROR )   
286.   
287.         {   
288.   
289.             cout << endl << "Ret=" << ret << " ErrorCode=" << WSAGetLastError() << endl;   
290.   
291.             return 0;   
292.   
293.         }   
294.   
295.         FinishSize += ret;   
296.   
297.         if(FinishSize == FSize) break;   
298.   
299.     }   
300.   
301.     closesocket(sock);   
302.   
303.     ptimeKillEvent(uTimerID);   
304.   
305.     CloseHandle(hMap);   
306.   
307.     CloseHandle(hFile);   
308.   
309.     cout << endl;   
310.   
311.     cout << "完成 发送数据量:" << FinishSize << endl;   
312.   
313.     return 0;   
314.   
315. }   
316.   
317. //---------------------------------------------------------------------------  

//---------------------------------------------------------------------------#include <iostream>#include "winsock2.h"using namespace std;#pragma comment(lib , "ws2_32.lib")//---------------------------------------------------------------------------WSADATA wsaData;UINT uTimerID;DWORD MAX_SIZE = 1024 * 4;DWORD FinishSize = 0;DWORD SendSize = MAX_SIZE;LPSTR SFileName;DWORD FSize;DWORD DWSpeed=0;DWORD OldFinish=0;MMRESULT (WINAPI *ptimeSetEvent)(UINT uDelay, UINT uResolution, LPTIMECALLBACK fptc, DWORD dwUser, UINT fuEvent);MMRESULT (WINAPI *ptimeKillEvent)(UINT uTimerID);HINSTANCE DLLInst;bool init(){DLLInst=LoadLibrary("winmm.dll");if(DLLInst)    {ptimeSetEvent = (MMRESULT (WINAPI *)(UINT uDelay, UINT uResolution, LPTIMECALLBACK fptc, DWORD dwUser, UINT fuEvent))GetProcAddress(DLLInst, "timeSetEvent");if(!ptimeSetEvent) return false;ptimeKillEvent = (MMRESULT (WINAPI *)(UINT uTimerID))GetProcAddress(DLLInst, "timeKillEvent");if(!ptimeKillEvent) return false;return true;}return false;}void Byte2Human(const DWORD & byte){if(byte < 1024){cout << ((float)((int)(byte*100+0.5))/100.0) << "B";return;}float kb = byte/1024;if (kb<1024){cout << ((float)((int)(kb*100+0.5))/100.0) << "KB";return ;}float mb = kb/1024;if (mb<1024){cout << ((float)((int)(mb*100+0.5))/100.0) << "MB";return;}float gb = mb/1024;cout << ((float)((int)(gb*100+0.5))/100.0) << "GB";}void WINAPI TimeProc(UINT uID, UINT uMsg, DWORD dwUser, DWORD dw1, DWORD dw2){DWSpeed = FinishSize-OldFinish;OldFinish = FinishSize;cout << '/r'; cout << "                                                                              ";cout << '/r';cout << "文件名:" << SFileName << " " << FinishSize << "/" << FSize << " ";Byte2Human(DWSpeed);cout << "/S";}int main(int argc, char* argv[]){if(!init()){cout<<"初始化失败"<<endl;return 0;}int ret;if(argc < 4){cout << "App.exe Host Port FileName" << endl;return 0;}if((ret=WSAStartup(0x0202, &wsaData))!=0){cout << "WSAStartup Failed!! Code:" << ret << endl;return 0;}SOCKET sock;struct sockaddr_in server;sock=socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);if(sock==INVALID_SOCKET){cout << "创建socket出错" << endl;return 0;}server.sin_family=AF_INET;server.sin_port=htons(atoi(argv[2]));server.sin_addr.s_addr=inet_addr(argv[1]);cout << "正在连接 " << argv[1] << ":" << argv[2] << endl;if(connect(sock, (struct sockaddr *)&server, sizeof(server))==SOCKET_ERROR){cout << "连接失败" << endl;return 0;}cout << "连接成功" << endl;//打开文件准备发送HANDLE hFile;OFSTRUCT ofs;HANDLE hMap;void *pVoid;SFileName = argv[3];hFile=(HANDLE)OpenFile(argv[3], &ofs, OF_READWRITE | OF_SHARE_DENY_NONE);if((HFILE)hFile==HFILE_ERROR){cout << "打开文件失败 ErrorCode=" << GetLastError() << endl;return 0;}FSize=GetFileSize((HANDLE)hFile, NULL);hMap=CreateFileMapping(hFile, NULL, PAGE_READWRITE, 0, 0, NULL);if(hMap==NULL){cout << "打开映射失败 ErrorCode=" << GetLastError() << endl;return 0;}pVoid=MapViewOfFile(hMap, FILE_MAP_ALL_ACCESS, 0, 0, 0);if(pVoid==NULL){cout << "映射数据失败 ErrorCode=" << GetLastError() << endl;return 0;}char *p = (char *)pVoid;cout << "正在发送文件 " << argv[3] << endl;uTimerID=ptimeSetEvent(1000, 0, (LPTIMECALLBACK)TimeProc, NULL, TIME_PERIODIC|TIME_CALLBACK_FUNCTION);while(1){if(FSize - FinishSize < MAX_SIZE){SendSize = FSize - FinishSize;}int ret = send(sock, p+FinishSize, SendSize, 0);if(ret == SOCKET_ERROR ){cout << endl << "Ret=" << ret << " ErrorCode=" << WSAGetLastError() << endl;return 0;}FinishSize += ret;if(FinishSize == FSize) break;}closesocket(sock);ptimeKillEvent(uTimerID);CloseHandle(hMap);CloseHandle(hFile);cout << endl;cout << "完成 发送数据量:" << FinishSize << endl;return 0;}//---------------------------------------------------------------------------

view plaincopy to clipboardprint?

1.    

view plaincopy to clipboardprint?

1. 以下是接收端代码  

以下是接收端代码

view plaincopy to clipboardprint?

1. recv.cpp<PRE class=csharp name="code">//---------------------------------------------------------------------------  
3. #include <iostream>  
4. #include "winsock2.h"   
5. using namespace std;</PRE>   
6. <PRE class=csharp name="code">  
7. #pragma comment(lib , "ws2_32.lib")   
8.   
9. //---------------------------------------------------------------------------</PRE>   
10. <PRE class=csharp name="code">   
11.   
12. WSADATA wsaData;   
13.   
14. DWORD FinishSize = 0;   
15.   
16. DWORD OldFinish=0;   
17.   
18. DWORD DWSpeed=0;   
19.   
20. UINT uTimerID;   
21.   
22.   
23.   
24. MMRESULT (WINAPI *ptimeSetEvent)(UINT uDelay, UINT uResolution, LPTIMECALLBACK fptc, DWORD dwUser, UINT fuEvent);   
25.   
26. MMRESULT (WINAPI *ptimeKillEvent)(UINT uTimerID);   
27.   
28. HINSTANCE DLLInst;   
29.   
30.   
31.   
32. bool init()   
33.   
34. {   
35.   
36.     DLLInst=LoadLibrary("winmm.dll");   
37.   
38.     if(DLLInst)   
39.   
40.     {   
41.   
42.         ptimeSetEvent = (MMRESULT (WINAPI *)(UINT uDelay, UINT uResolution, LPTIMECALLBACK fptc, DWORD dwUser, UINT fuEvent))GetProcAddress(DLLInst, "timeSetEvent");   
43.   
44.         if(!ptimeSetEvent) return false;   
45.   
46.         ptimeKillEvent = (MMRESULT (WINAPI *)(UINT uTimerID))GetProcAddress(DLLInst, "timeKillEvent");   
47.   
48.         if(!ptimeKillEvent) return false;   
49.   
50.         return true;   
51.   
52.     }   
53.   
54.     return false;   
55.   
56. }   
57.   
58.   
59.   
60. void Byte2Human(const DWORD & byte)   
61.   
62. {   
63.   
64.     if(byte < 1024){   
65.   
66.         cout << ((float)((int)(byte*100+0.5))/100.0) << "B";   
67.   
68.         return;   
69.   
70.     }   
71.   
72.     float kb = byte/1024;   
73.   
74.     if (kb<1024){   
75.   
76.         cout << ((float)((int)(kb*100+0.5))/100.0) << "KB";   
77.   
78.         return ;   
79.   
80.     }   
81.   
82.     float mb = kb/1024;   
83.   
84.     if (mb<1024){   
85.   
86.         cout << ((float)((int)(mb*100+0.5))/100.0) << "MB";   
87.   
88.         return;   
89.   
90.     }   
91.   
92.     float gb = mb/1024;   
93.   
94.     cout << ((float)((int)(gb*100+0.5))/100.0) << "GB";   
95.   
96. }   
97.   
98.   
99.   
100. void CALLBACK TimeProc(UINT uID, UINT uMsg, DWORD dwUser, DWORD dw1, DWORD dw2)   
101.   
102. {   
103.   
104.     DWSpeed = FinishSize-OldFinish;   
105.   
106.     OldFinish = FinishSize;   
107.   
108.     cout << '/r';   
109.   
110.     cout << "                                                                              ";   
111.   
112.     cout << '/r';   
113.   
114.     cout << "已接收 " << FinishSize << " ";   
115.   
116.     Byte2Human(DWSpeed);   
117.   
118.     cout << "/S";   
119.   
120. }   
121.   
122.   
123.   
124. int main(int argc, char* argv[])   
125.   
126. {   
127.   
128.     if(!init())   
129.   
130.     {   
131.   
132.         cout<<"初始化失败"<<endl;   
133.   
134.         return 0;   
135.   
136.     }   
137.   
138.   
139.   
140.     int ret;   
141.   
142.     if(argc < 3)   
143.   
144.     {   
145.   
146.         cout << "App.exe Port NewFileName" << endl;   
147.   
148.         return 0;   
149.   
150.     }   
151.   
152.   
153.   
154.     HANDLE hFile;   
155.   
156.     OFSTRUCT ofs;   
157.   
158.     HANDLE hMap;   
159.   
160.     void *pVoid;   
161.   
162.     hFile=(HANDLE)OpenFile(argv[2], &ofs, OF_CREATE | OF_READ);   
163.   
164.     if((HFILE)hFile==HFILE_ERROR)   
165.   
166.     {   
167.   
168.         cout << "打开文件失败";   
169.   
170.         ret = GetLastError();   
171.   
172.         cout << " ErrorCode=" << ret << endl;   
173.   
174.         return 0;   
175.   
176.     }   
177.   
178.   
179.   
180.     if((ret=WSAStartup(0x0202, &wsaData))!=0)   
181.   
182.     {   
183.   
184.         cout << "WSAStartup Failed!! Code:" << ret << endl;   
185.   
186.         return 0;   
187.   
188.     }   
189.   
190.   
191.   
192.     SOCKET sock, ssock;   
193.   
194.     struct sockaddr_in fsin, sin;   
195.   
196.     int alen;   
197.   
198.   
199.   
200.     memset(&sin, 0, sizeof(sin));   
201.   
202.     sin.sin_family=AF_INET;   
203.   
204.     sin.sin_addr.s_addr=inet_addr("");   
205.   
206.     sin.sin_port=htons(atoi(argv[1]));   
207.   
208.     sock=socket(PF_INET, SOCK_STREAM, 0);   
209.   
210.     if(sock==INVALID_SOCKET)   
211.   
212.     {   
213.   
214.         cout << "创建socket出错" << endl;   
215.   
216.         return 0;   
217.   
218.     }   
219.   
220.     if(bind(sock, (struct sockaddr*)&sin, sizeof(sin))==SOCKET_ERROR)   
221.   
222.     {   
223.   
224.         cout << "帮定Socket失败" << endl;   
225.   
226.         return 0;   
227.   
228.     }   
229.   
230.     if(listen(sock, SOMAXCONN)==SOCKET_ERROR)   
231.   
232.     {   
233.   
234.         cout << "监听失败" << endl;   
235.   
236.         return 0;   
237.   
238.     }   
239.   
240.     cout << "正在监听... 端口" << argv[1] << endl;   
241.   
242.     alen=sizeof(struct sockaddr);   
243.   
244.     ssock=accept(sock, (struct sockaddr *)&fsin, &alen);   
245.   
246.     if(ssock==INVALID_SOCKET)   
247.   
248.     {   
249.   
250.         cout << "接受请求失败" << endl;   
251.   
252.         return 0;   
253.   
254.     }   
255.   
256.     closesocket(sock);   
257.   
258.     char buf[4096];   
259.   
260.     DWORD dwWrite;   
261.   
262.     cout << "连接已建立" << endl;   
263.   
264.     uTimerID=ptimeSetEvent(1000, 0, (LPTIMECALLBACK)TimeProc, NULL, TIME_PERIODIC|TIME_CALLBACK_FUNCTION);   
265.   
266.   
267.   
268.     while(1)   
269.   
270.     {   
271.   
272.         memset(buf, 0x00, 4096);   
273.   
274.         ret = recv(ssock, buf, 4096, 0);   
275.   
276.         if(ret == SOCKET_ERROR )   
277.   
278.         {   
279.   
280.             cout << endl << "接收数据量:" << FinishSize << endl;   
281.   
282.             cout << "Ret=" << ret;   
283.   
284.             ret = WSAGetLastError();   
285.   
286.             cout << " ErrorCode=" << ret << endl;   
287.   
288.             return 0;   
289.   
290.         }   
291.   
292.         if(ret == 0)   
293.   
294.         {   
295.   
296.             cout << endl << "连接已断开" << endl;   
297.   
298.             break;   
299.   
300.         }   
301.   
302.         FinishSize += ret;   
303.   
304.         if(!WriteFile(hFile, buf, ret, &dwWrite, NULL))   
305.   
306.         {   
307.   
308.             cout << "写文件失败" << endl;   
309.   
310.             return 0;   
311.   
312.         }   
313.   
314.   
315.   
316.     }   
317.   
318.     ptimeKillEvent(uTimerID);   
319.   
320.     closesocket(sock);   
321.   
322.     CloseHandle(hFile);   
323.   
324.     cout << endl;   
325.   
326.     cout << "完成 接收数据量:" << FinishSize << endl;   
327.   
328.     return 0;   
329.   
330. }   
331.   
332. //---------------------------------------------------------------------------