用HTTP代理下载文件

 

1. //用HTTP代理下载文件详例
2. //代码如下:
4. //(1)一些基本变量
6. SOCKET HTTPSocket   = NULL; // 主socket
7. struct sockaddr_in SocketAddr = {0}; // address socket
8. struct sockaddr_in BindSocket = {0}; // for bind
10. int m_nRecvTimeout; // recieve timeout
11. int m_nSendTimeout; // send timeout
13. WSADATA wsaData;
15. //要下载文件部分.
16. //在BindSocket.sin_addr.s_addr = inet_addr (strHost)时,只能使用ip地址
17. //如果谁知道更好的方法，别忘了告诉我一下。
19. CString strHost="111.111.111.111 ";
20. CString DownLoadAddress="http://blog.csdn.net/andylin02";
21. CString hostFile="logo.gif";
22. int HttpPort=80;
24. //(2)一些函数，用来取得http头，和获取文件大小 
25. int GetFileLength(char *httpHeader)
26. {
27.     CString strHeader;
28.     int local;
29.     strHeader=(CString)httpHeader;
30.     local=strHeader.Find("Content-Length",0);
31.     local+=16;
32.     strHeader.Delete(0,local);
33.     local=strHeader.Find("/r");
34.     strHeader.SetAt(local,'/0');
35.     
36.     char temp[30];
37.     strcpy(temp,strHeader.GetBuffer(strHeader.GetLength()));
38.     
39.     return atoi(temp);
40. }
42. int GetHttpHeader(SOCKET sckDest, char *str)
43. {
44.     BOOL m_bResponsed=0;
45.     int m_nResponseHeaderSize;
47.     if(!m_bResponsed)
48.     {
49.         char c = 0;
50.         int nIndex = 0;
51.         BOOL bEndResponse = FALSE;
52.         while(!bEndResponse && nIndex < 1024)
53.         {
54.             recv(sckDest,&c,1,0);
55.             str[nIndex++] = c;
56.             if(nIndex >= 4)
57.             {
58.                 if(str[nIndex - 4] == /'/r/' && str[nIndex - 3] == /'/n/'
59.                     && str[nIndex - 2] == /'/r/' && str[nIndex - 1] == /'/n/')
60.                     bEndResponse = TRUE;
61.             }
62.         }
63.         m_nResponseHeaderSize = nIndex;
64.         m_bResponsed = TRUE;
65.     }
67.     return m_nResponseHeaderSize;
69. }
72. //(3)用来发送的部分
73. void szcopy(char* dest,const char* src,int nMaxBytes)
74. {
75.     int i_cntr=0;
76.     while ((src[i_cntr]!=/'/0/') || (i_cntr<nMaxBytes))
77.     {
78.         dest[i_cntr]=src[i_cntr++];
79.     }
81.     dest[i_cntr]=/'/0/';
82. }
84. BOOL SocketSend(SOCKET sckDest,const char* szHttp)
85. {
86.     char szSendHeader[MAXHEADERLENGTH];
87.     int iLen=strlen(szHttp);
89.     szcopy(szSendHeader,szHttp,iLen);
91.     if(send (sckDest ,(const char FAR *)szSendHeader ,iLen ,0)==SOCKET_ERROR)
92.     {
93.         closesocket(sckDest);
94.         AfxMessageBox("Error when send");
95.         return FALSE;
96.     }
98.     return TRUE;
99. }
101. BOOL SocketSend(SOCKET sckDest,CString szHttp)
102. {
103.     int iLen=szHttp.GetLength();
104.     if(send (sckDest,szHttp,iLen,0)==SOCKET_ERROR)
105.     {
106.         closesocket(sckDest);
107.         AfxMessageBox("Error when send");
108.         return FALSE;
109.     }
111.     return TRUE;
112. }
114. //(4)用于连接的函数
115. //这里是做了一些连接用的操作，分了两种情况
116. //   1)如果没有使用代理，则直接连到你指定的计算机
117. // 　2)如果使用了代理，则直接连到代理
119. BOOL CDLAngelDlg::ConnectHttp()
120. {
121.     message="正在建立连接/n";
122.     UpdateData(TRUE);
124.     if(m_combo == "HTTP") // m_combo 一个下拉条
125.     {
126.         HTTPSocket=socket(AF_INET,SOCK_STREAM,IPPROTO_TCP);
127.         SocketAddr.sin_addr.s_addr = inet_addr (m_ProxyAddr);
128.         SocketAddr.sin_family=AF_INET;
129.         SocketAddr.sin_port=htons(atoi(m_Port));
131.         struct fd_set fdSet;
132.         struct timeval tmvTimeout={0L,0L};
134.         FD_ZERO(&fdSet);
135.         FD_SET(HTTPSocket, &fdSet);
137.         if (select(0,&fdSet,NULL,NULL,&tmvTimeout)==SOCKET_ERROR)
138.         {
139.             closesocket(HTTPSocket);
140.             AfxMessageBox("Error when select.");
141.             return 0;
142.         }
144.         if (connect(HTTPSocket, (const struct sockaddr *)&SocketAddr, sizeof(SocketAddr))==SOCKET_ERROR)
145.         {
146.             message="/n代理连接失败/n";
147.             m_message.CleanText();
148.             m_message.AddText(message);
149.             return 0;
150.         }
152.         // 发送CONNCET请求令到代理服务器，用于和代理建立连接
153.         //代理服务器的地址和端口放在m_ProxyAddr,m_Port 里面
154.         CString temp;
155.         char tmpBuffer[1024];
156.         temp.Format("CONNECT %s:%s HTTP/1.1/r/nUser-Agent: MyApp/0.1/r/n/r/n",m_ProxyAddr,m_Port);
157.         if(!SocketSend(HTTPSocket,temp))
158.         {
159.             message="连接代理失败";
160.             return 0;
161.         }
163.         // 取得代理响应，如果连接代理成功，代理服务器将返回200 Connection established
164.         GetHttpHeader(HTTPSocket,tmpBuffer);
165.         temp=tmpBuffer;
167.         if(temp.Find("HTTP/1.0 200 Connection established",0)==-1)
168.         {
169.             message="连接代理失败/n";
170.             return 0;
171.         }
173.         message="代理连接完成/n";
174.         m_message.AddText("代理连接完成/n");
176.         return 1; // ----------〉这里是应该注意的，连接到代理后，就可以返回了，不需要再连接网上的另外一台机，代理服务器会自动转发数据，所以，连接完代理就像连接到网上另外一台机一样
177.     }
178.     else if(m_combo=="Socks4") // 这个，是为了给其他代理做准备
179.     {
180.         MessageBox("请注意，现在无法使用代理功能！");
181.     }
182.     else if(m_combo=="Socks5")
183.     {
184.         MessageBox("请注意，现在无法使用代理功能！");
185.     }
187.     // 如果没有使用代理，就要连接到网上的另一台机
188.     // 准备socket
189.     HTTPSocket=socket(AF_INET,SOCK_STREAM,IPPROTO_TCP);
191.     if (HTTPSocket==INVALID_SOCKET)
192.     {
193.         AfxMessageBox("Error when socket");
194.         return 0;
195.     }
197.     //设置超时
198.     struct linger zeroLinger;
199.     zeroLinger.l_onoff = 1;
200.     zeroLinger.l_linger = 0;
201.     if(setsockopt(HTTPSocket,SOL_SOCKET,SO_LINGER, (const char *)&zeroLinger, sizeof(zeroLinger))!=0)
202.     {
203.         closesocket(HTTPSocket);
204.         AfxMessageBox("Error when setscokopt(LINGER)");
205.         return 0;
206.     }
208.     //设置接收超时
209.     if (setsockopt(HTTPSocket,SOL_SOCKET,SO_RCVTIMEO,(const char *)&m_nRecvTimeout,sizeof(m_nRecvTimeout))!=0)
210.     {
211.         closesocket(HTTPSocket);
212.         AfxMessageBox("Error when setsockopt(RCVTIME).");
213.         return 0;
214.     }
216.     //设置发送超时
217.     if(setsockopt(HTTPSocket,SOL_SOCKET,SO_SNDTIMEO, (const char *)&m_nSendTimeout, sizeof(m_nSendTimeout))!=0)
218.     {
219.         closesocket(HTTPSocket);
220.         AfxMessageBox("Error when setsockopt(SNDTIMEO).");
221.         return 0;
222.     }
224.     SocketAddr.sin_addr.s_addr = htonl (INADDR_ANY);
225.     SocketAddr.sin_family=AF_INET;
227.     // 进行端口绑定
228.     if (bind (HTTPSocket, (const struct sockaddr FAR *)&SocketAddr, sizeof(SocketAddr))==SOCKET_ERROR)
229.     {
230.         closesocket(HTTPSocket);
231.         AfxMessageBox("Error when bind socket.");
232.         return 0;
233.     }
235.     //准备连接
236.     /// 准备连接信息
237.     BindSocket.sin_addr.s_addr = inet_addr (strHost);
238.     BindSocket.sin_family=AF_INET;
239.     BindSocket.sin_port=htons(HttpPort);
241.     struct fd_set fdSet;
242.     struct timeval tmvTimeout={0L,0L};
244.     FD_ZERO(&fdSet);
245.     FD_SET(HTTPSocket, &fdSet);
247.     if (select(0,&fdSet,NULL,NULL,&tmvTimeout)==SOCKET_ERROR)
248.     {
249.         closesocket(HTTPSocket);
250.         AfxMessageBox("Error when select.");
251.         return 0;
252.     }
254.     // 连接
257.     if (connect(HTTPSocket, (const struct sockaddr *)&BindSocket, sizeof(BindSocket))==SOCKET_ERROR)
258.     {
259.         AfxMessageBox("第一次连接失败，准备第二次连接");
260.         if (connect(HTTPSocket, (const struct sockaddr *)&BindSocket, sizeof(BindSocket))==SOCKET_ERROR)
261.         {
262.             closesocket(HTTPSocket);
263.             AfxMessageBox("连接失败");
264.             return 0;
265.         }
266.     }
268.     message="连接完成/n";
270.     return 1;
271. }
273. //(5)发送http请求，为下载数据进行准备
274. int CDLAngelDlg::SendHttpHeader()
275. {
276.     //进行下载 
277.     CString temp;
278.     BOOL bReturn;
279.     char tmpBuffer[MAXBLOCKSIZE];
281.     ///第1行:方法,请求的路径,版本
282.     temp="GET "+DownLoadAddress+hostFile+" HTTP/1.0/r/n";
283.     bReturn=SocketSend(HTTPSocket,temp);
284.     if(!bReturn)
285.     {
286.         message="发送请求失败";
287.         return 0;
288.     }
290.     ///第2行:主机
291.     temp="Host "+strHost+"/r/n";
292.     bReturn=SocketSend(HTTPSocket,temp);
293.     if(!bReturn)
294.     {
295.         message="发送请求失败";
296.         return 0;
297.     }
300.     ///第3行:接收的数据类型
301.     bReturn=SocketSend(HTTPSocket,"Accept: */*/r/n");
302.     if(!bReturn)
303.     {
304.         message="发送请求失败";
305.         return 0;
306.     }
309.     ///第4行：
310.     temp=DownLoadAddress;
311.     temp.Insert(0,"Referer ");
312.     temp+="/r/n";
313.     bReturn=SocketSend(HTTPSocket,temp);
314.     if(!bReturn)
315.     {
316.         message="发送请求失败";
317.         return 0;
318.     }
320.     ///第5行:浏览器类型
321.     bReturn=SocketSend(HTTPSocket,"User-Agent: Mozilla/4.0 (compatible; MSIE 5.0; Windows NT; DigExt; DTS Agent;)/r/n");
322.     if(!bReturn)
323.     {
324.         message="发送请求失败";
325.         return 0;
326.     }
328.     ///第6行:连接设置,保持
329.     // SocketSend(HTTPSocket,"Connection:Keep-Alive/r/n");
331.     ///第7行:Cookie.
333.     bReturn=SocketSend(HTTPSocket,"Cache-Control: no-cache/r/n");
334.     if(!bReturn)
335.     {
336.         message="发送请求失败";
337.         return 0;
338.     }
340.     bReturn=SocketSend(HTTPSocket,"Proxy-Connection: Keep-Alive/r/n");
341.     if(!bReturn)
342.     {
343.         message="发送请求失败";
344.         return 0;
345.     }
347.     /// 续传
348.     //Range是要下载的数据范围，对续传很重要
349.     if(continueFlag)
350.     {
351.         temp.Format("Range: bytes=%d- /r/n",conLength);
352.         bReturn=SocketSend(HTTPSocket,temp);
353.         if(!bReturn)
354.         {
355.             message="发送请求失败";
356.             return 0;
357.         }
358.     }
359.     
360.     ///最后一行:空行
361.     bReturn=SocketSend(HTTPSocket,"/r/n");
362.     if(!bReturn)
363.     {
364.         message="发送请求失败";
365.             return 0;
366.         }
368.         ///取得http头
369.         int i;
370.         i=GetHttpHeader(HTTPSocket,tmpBuffer);
371.         if(!i)
372.         {
373.             message="获取HTTP头出错";
374.             return 0;
375.         }
377.         //如果取得的http头含有404等字样，则表示连接出问题
378.         temp=tmpBuffer;
379.         if(temp.Find("404")!=-1)
380.         {
382.             return 0;
383.         }
385.         // 得到待下载文件的大小
387.         filelength=GetFileLength(tmpBuffer);
389.         return 1;
390. }
393. //这样，就连接到网上的另一台机了，如何下载数据，不用多说了吧
394.     while((num!=SOCKET_ERROR) && (num!=0))
395.     {
396.         num=recv(HTTPSocket, (char FAR *)tmpBuffer,(MAXBLOCKSIZE-1) ,0); 
397.         file.Write(tmpBuffer,num);
399.         if(ExitFlag)
400.         {
401.             file.Close();
402.             closesocket(HTTPSocket);
404.             DownComplete=1;
406.             m_message.CleanText();
407.             m_message.ShowColorText(RGB(128,128,0),DLCompleteMes);
409.             m_progress.ShowWindow(SW_HIDE);
410.             m_stopDownload.ShowWindow(SW_HIDE);
411.             _endthread();
412.         }
413.     }
414. }