开通博客赚积分
发布资源赚积分
分类

源码开发语言/平台
当前位置：首页 > 源码/资料 > 游戏 > 模拟服务器 > 查看源码
httpdownload.cpp
资源名称：SwordOnline.rar　[点击查看]
上传用户：dzyhzl
上传日期：2019-04-29
资源大小：56270k
文件大小：51k
源码类别：

模拟服务器

开发平台：

C/C++

httpdownload.cpp：源码内容
  1. ///////////////////////////////////////////////////////////////////////////////
  2. // HttpDownload.cpp: implementation of the CHttpDownload class.
  3. ////////////////////////////////////////////////////////////////////////////////
  4. #include "stdafx.h"
  5. #include "HttpDownload.h"
  6. #include "SocksPacket.h"
  8. #define READ_BUFFER_SIZE (10 * 1024) // 缓冲大小 10K
  9. #define DEFAULT_SAVE_DIR _T("C:\") // 缺省保存目录
  10. #define DEFAULT_SAVE_FILE _T("Unknown.htm") // 缺省保存文件名
  11. #define DEFAULT_USERAGENT _T("KAVUpdate/1.0") // 缺省的UserAgent
  12. #define Check(a,b) if((a))return (b);
  14. // 用于BASE64编码、解码的静态常量
  15. CString CHttpDownload::m_strBase64TAB = _T( "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/" );
  16. int CHttpDownload::m_nBase64Mask[]= { 0, 1, 3, 7, 15, 31, 63, 127, 255 };
  18. ////////////////////////////////////////////////////////////////////////////////
  19. // 构造函数
  20. ////////////////////////////////////////////////////////////////////////////////
  21. CHttpDownload::CHttpDownload()
  22. {
  23. m_strDownloadUrl = _T("");
  24. m_strSavePath = _T("");
  25. m_strTempSavePath = _T("");
  26. m_bSaveResponse = TRUE; //是否需要保存返回
  28. // 停止下载
  29. m_bStopDownload = FALSE;
  30. m_hStopEvent = NULL;
  32. // 强制重新下载(不管已有的文件是否与远程文件相同)
  33. m_bForceDownload = FALSE;
  35. // 是否支持断点续传(假定不支持)
  36. m_bSupportResume = FALSE;
  38. // 文件以及下载大小
  39. m_dwFileSize = 0; // 文件总的大小
  40. m_dwFileDownloadedSize = 0; // 文件总共已经下载的大小
  42. m_dwDownloadSize = 0; // 本次Request需要下载的大小
  43. m_dwDownloadedSize = 0; // 本次Request已经下载的大小
  45. m_dwHeaderSize = 0; // HTTP协议头的长度
  46. m_strHeader = _T(""); // HTTP协议头
  48. // Referer
  49. m_strReferer = _T("");
  51. // UserAgent
  52. m_strUserAgent = DEFAULT_USERAGENT;
  54. // 超时TIMEOUT 连接超时、发送超时、接收超时(单位：毫秒)
  55. m_dwConnTimeout = HTTP_CONN_TIMEOUT;
  56. m_dwRecvTimeout = HTTP_RECV_TIMEOUT;
  57. m_dwSendTimeout = HTTP_SEND_TIMEOUT;
  59. // 重试机制
  60. m_nRetryType = HTTP_RETRY_NONE; //重试类型(0:不重试 1:重试一定次数 2:总是重试)
  61. m_nRetryTimes = 0; //重试次数
  62. m_nRetryDelay = 0; //重试延迟(单位：毫秒)
  63. m_nRetryMax = 0; //重试最大次数
  65. // 错误处理
  66. m_nErrorCount = 0; //错误次数
  67. m_strError = _T(""); //错误信息
  68. m_dwErrorCode = 0; //错误代码
  70. // 向其他窗口发送消息
  71. m_bNotify = FALSE; // 是否向外发送通知消息
  72. m_hNotifyWnd = NULL; // 被通知的窗口
  73. m_nNotifyMessage = 0; // 被通知的消息
  75. // 是否进行验证 : Request-Header: Authorization
  76. m_bAuthorization = FALSE;
  77. m_strUsername = _T("");
  78. m_strPassword = _T("");
  80. // 是否使用代理 
  81. m_bProxy = FALSE;
  82. m_strProxyServer = _T("");
  83. m_nProxyPort = 0;
  84. m_nProxyType = HTTP_PROXY_NONE;
  86. // 代理是否需要验证: Request-Header: Proxy-Authorization
  87. m_bProxyAuthorization = FALSE;
  88. m_strProxyUsername  = _T("");
  89. m_strProxyPassword = _T("");
  91. // 下载过程中所用的变量
  92. m_strServer = _T("");
  93. m_strObject = _T("");
  94. m_strFileName = _T("");
  95. m_nPort = DEFAULT_HTTP_PORT ;
  96. m_nVerb = HTTP_VERB_GET;
  97. m_strData = _T("");
  99. // 是否使用Cookie
  100. m_strCookie = _T("");
  101. m_bUseIECookie = FALSE;
  103. }
  106. ////////////////////////////////////////////////////////////////////////////////
  107. // 析构函数
  108. ////////////////////////////////////////////////////////////////////////////////
  109. CHttpDownload::~CHttpDownload()
  110. {
  111. CloseSocket();
  112. if( m_hStopEvent != NULL )
  113. {
  114. CloseHandle( m_hStopEvent );
  115. m_hStopEvent = NULL;
  116. }
  117. }
  119. //////////////////////////////////////////////////////////////////////////////////
  120. // 函数名：BOOL CreateSocket()
  121. // 用  途：创建SOCKET
  122. // 对全局变量的影响：无
  123. // 参  数：无
  124. // 返回值：BOOL
  125. // TRUE : 成功创建SOCKET
  126. // FALSE: 创建SOCKET失败
  127. ////////////////////////////////////////////////////////////////////////////////
  128. BOOL CHttpDownload::CreateSocket()
  129. {
  130. CloseSocket();
  131. return m_cBufSocket.Create(AF_INET,SOCK_STREAM,0,READ_BUFFER_SIZE);
  132. }
  135. //////////////////////////////////////////////////////////////////////////////////
  136. // 函数名：void CloseSocket()
  137. // 用  途：关闭SOCKET
  138. // 对全局变量的影响：无
  139. // 参  数：无
  140. // 返回值：void
  141. ////////////////////////////////////////////////////////////////////////////////
  142. void CHttpDownload::CloseSocket()
  143. {
  144. m_cBufSocket.Close(TRUE);
  145. }
  148. //////////////////////////////////////////////////////////////////////////////////
  149. // 函数名：int Download(
  150. // LPCTSTR lpszDownloadUrl,
  151. // LPCTSTR lpszSavePath /*= NULL*/,
  152. // BOOL bForceDownload /*= FALSE*/,
  153. // BOOL bSaveResponse /*= TRUE*/,
  154. // int nVerb /*= HTTP_VERB_GET*/,
  155. // LPCTSTR lpszData /*= NULL*/,
  156. // LPTSTR lpszFtpURL /*= NULL*/) 
  157. // 用  途：下载入口，调用此函数来开始下载过程
  158. // 对全局变量的影响：无
  159. // 参  数：
  160. // lpszDownloadUrl : 待下载的网络URL
  161. // lpszSavePath    : 本地保存路径(可以是全路径或者仅仅是目录，目录需以结尾)
  162. // bForceDownload  : 是否强制下载，不管文件是否与远程一致
  163. // bSaveResponse   : 是否保存服务器的返回
  164. // nVerb           : HTTP命令
  165. // lpszData        : 要在发送请求时传给服务器的数据(用于POST)
  166. // lpszFtpURL : 如果重定向到FTP,此指针用来保存FTP的URL
  167. // 返回值：int
  168. // HTTP_RESULT_STOP : 用户停止下载(调用了StopDownload函数)
  169. // HTTP_RESULT_FAIL : 下载失败
  170. // HTTP_RESULT_SUCCESS : 下载成功
  171. // HTTP_RESULT_SAMEAS : 本地已有完全一样的文件，不用再下载
  172. // HTTP_RESULT_REDIRECT_FTP: 重定向到FTP
  173. ////////////////////////////////////////////////////////////////////////////////
  174. int CHttpDownload::Download(LPCTSTR lpszDownloadUrl,LPCTSTR lpszSavePath /*= NULL*/,BOOL bForceDownload /*= FALSE*/,BOOL bSaveResponse /*= TRUE*/,int nVerb /*= HTTP_VERB_GET*/,LPCTSTR lpszData /*= NULL*/,LPTSTR lpszFtpURL /*= NULL*/ )
  175. {
  176. m_bStopDownload   = FALSE;
  177. m_bForceDownload  = bForceDownload;
  178. m_nRetryTimes   = 0;
  179. m_nErrorCount   = 0;
  180. m_strTempSavePath = _T("");
  182. m_bSaveResponse   = bSaveResponse;
  183. m_nVerb   = nVerb;
  184. m_strData   = _T("");
  185. if( lpszData != NULL )
  186. m_strData += lpszData;
  189. // 检验要下载的URL是否为空
  190. m_strDownloadUrl = lpszDownloadUrl;
  191. m_strDownloadUrl.TrimLeft();
  192. m_strDownloadUrl.TrimRight();
  193. if( m_strDownloadUrl.IsEmpty() )
  194. return HTTP_RESULT_FAIL;
  196. // 检验要下载的URL是否有效
  197. if ( !ParseURL(m_strDownloadUrl, m_strServer, m_strObject, m_nPort))
  198. {
  199. // 在前面加上"http://"再试
  200. m_strDownloadUrl = _T("http://") + m_strDownloadUrl;
  201. if ( !ParseURL(m_strDownloadUrl,m_strServer, m_strObject, m_nPort) )
  202. {
  203. TRACE(_T("Failed to parse the URL: %sn"), m_strDownloadUrl);
  204. return HTTP_RESULT_FAIL;
  205. }
  206. }
  208. // 检查本地保存路径
  209. m_strSavePath =  lpszSavePath;
  210. m_strSavePath.TrimLeft();
  211. m_strSavePath.TrimRight();
  212. if( m_strSavePath.IsEmpty() )
  213. m_strSavePath = DEFAULT_SAVE_DIR;
  215. if( m_strSavePath.Right(1) != '\' ) // 只有路径，没有文件名
  216. {
  217. m_strTempSavePath =  m_strSavePath;
  218. m_strTempSavePath += ".tmp";
  219. }
  221. // 创建停止事件
  222. if( m_hStopEvent == NULL )
  223. {
  224. m_hStopEvent = CreateEvent(NULL,TRUE,FALSE,NULL);
  225. if( m_hStopEvent == NULL )
  226. return HTTP_RESULT_FAIL;
  228. //seawind
  229. m_cBufSocket.m_hStopEvent = m_hStopEvent;
  230. }
  231. ResetEvent( m_hStopEvent );
  234. // 设置下载数据
  235. m_dwDownloadedSize = 0;
  236. m_dwFileDownloadedSize = 0;
  237. m_dwFileSize = 0;
  238. m_dwDownloadSize = 0;
  240. BOOL bSendOnce = TRUE; // 用于控制向hWndNotify窗口发送消息
  241. BOOL bSendRequestAgain = TRUE;
  243. while( bSendRequestAgain )
  244. {
  245. int nRequestRet = SendRequest( nVerb,lpszFtpURL ) ;
  247. //seawind
  248. //if (WaitForSingleObject(m_hStopEvent, 0) == WAIT_OBJECT_0)
  249. // return HTTP_RESULT_STOP;
  250. Check( m_bStopDownload, HTTP_RESULT_STOP);
  252. switch(nRequestRet)
  253. {
  254. case HTTP_REQUEST_SUCCESS:
  255. // 需要保存否?
  256. if( !m_bSaveResponse )
  257. {
  258. //关闭SOCKET
  259. CloseSocket();
  260. return HTTP_REQUEST_SUCCESS;
  261. }
  262. break;
  263. case HTTP_REQUEST_STOP:
  264. return HTTP_RESULT_STOP;
  265. break;
  266. case HTTP_REQUEST_FAIL:
  267. return HTTP_RESULT_FAIL;
  268. break;
  269. case HTTP_REQUEST_REDIRECT_FTP:
  270. return HTTP_RESULT_REDIRECT_FTP;
  271. break;
  272. case HTTP_REQUEST_ERROR:
  274. Check( m_bStopDownload, HTTP_RESULT_STOP); // 是否应该停止下载
  276. switch( m_nRetryType )
  277. {
  278. case HTTP_RETRY_NONE:
  279. return HTTP_RESULT_FAIL;
  280. break;
  281. case HTTP_RETRY_ALWAYS:
  282. if( m_nRetryDelay > 0 )
  283. {
  284. //为了让等待时也能退出，不能使用Sleep函数
  285. if( WaitForSingleObject(m_hStopEvent,m_nRetryDelay) == WAIT_OBJECT_0 )
  286. return HTTP_RESULT_STOP;
  287. }
  288. continue;
  289. break;
  290. case HTTP_RETRY_TIMES:
  291. if( m_nRetryTimes > m_nRetryMax )
  292. return HTTP_RESULT_FAIL;
  293. m_nRetryTimes++;
  295. if( m_nRetryDelay > 0 )
  296. {
  297. //为了让等待时也能退出，不能使用Sleep函数
  298. if( WaitForSingleObject(m_hStopEvent,m_nRetryDelay) == WAIT_OBJECT_0 )
  299. return HTTP_RESULT_STOP;
  300. }
  301. continue;
  302. break;
  303. default:
  304. return HTTP_RESULT_FAIL;
  305. break;
  306. }
  307. break;
  308. default:
  309. return HTTP_RESULT_FAIL;
  310. break;
  311. }
  313. if( m_strSavePath.Right(1) == '\' )
  314. m_strSavePath += m_strFileName;
  316. if( !m_bForceDownload ) // 非强制下载，检查Last-Modified
  317. {
  318. CFileStatus fileStatus;
  319. if (CFile::GetStatus(m_strSavePath,fileStatus))
  320. {
  321. // 可能会存在1秒的误差
  322. if (( fileStatus.m_mtime - m_TimeLastModified <=2 && m_TimeLastModified-fileStatus.m_mtime<=2 ) && (DWORD)fileStatus.m_size == m_dwFileSize )
  323. return HTTP_RESULT_SAMEAS;
  324. }
  325. }
  327. CFile fileDown;
  328. if(! fileDown.Open(m_strTempSavePath,CFile::modeCreate|CFile::modeNoTruncate|CFile::modeWrite|CFile::shareDenyWrite) )
  329. return HTTP_RESULT_FAIL;
  331. // 应该判断一下是否支持断点续传
  332. if( m_bSupportResume )
  333. {
  334. try
  335. {
  336. fileDown.SeekToEnd();
  337. }
  338. catch(CFileException* e)                                         
  339. {
  340. e->Delete();
  341. return HTTP_RESULT_FAIL;
  342. }
  343. }
  345. // 发送消息
  346. if( bSendOnce && m_bNotify )
  347. {
  348. HTTPDOWNLOADSTATUS DownloadStatus;
  350. DownloadStatus.dwFileSize  = m_dwFileSize;
  351. DownloadStatus.strFileName = m_strFileName;
  352. DownloadStatus.dwFileDownloadedSize  = m_dwFileDownloadedSize;
  354. DownloadStatus.nStatusType = HTTP_STATUS_FILESIZE;
  355. if (WaitForSingleObject(m_hStopEvent, 0) != WAIT_OBJECT_0)
  356. ::SendMessage(m_hNotifyWnd,m_nNotifyMessage,MSG_HTTPDOWNLOAD_STATUS,(LPARAM)&DownloadStatus);
  357. else
  358. return HTTP_RESULT_STOP;
  360. DownloadStatus.nStatusType = HTTP_STATUS_FILENAME;
  361. if (WaitForSingleObject(m_hStopEvent, 0) != WAIT_OBJECT_0)
  362. ::SendMessage(m_hNotifyWnd,m_nNotifyMessage,MSG_HTTPDOWNLOAD_STATUS,(LPARAM)&DownloadStatus);
  363. else
  364. return HTTP_RESULT_STOP;
  366. DownloadStatus.nStatusType = HTTP_STATUS_FILEDOWNLOADEDSIZE;
  367. if (WaitForSingleObject(m_hStopEvent, 0) != WAIT_OBJECT_0)
  368. ::SendMessage(m_hNotifyWnd,m_nNotifyMessage,MSG_HTTPDOWNLOAD_STATUS,(LPARAM)&DownloadStatus);
  369. else
  370. return HTTP_RESULT_STOP;
  372. bSendOnce = FALSE;
  373. }
  375. m_dwDownloadedSize = 0;
  376. // 现在开始读取数据
  377. char szReadBuf[READ_BUFFER_SIZE+1];
  379. /////////////////////////////////////////////
  380. // 如果m_dwDownloadSize = 0 (说明大小未知)
  381. if( m_dwDownloadSize == 0 )
  382. m_dwDownloadSize = 0xFFFFFFFF;
  383. ////////////////////////////////////////////
  384. bSendRequestAgain = FALSE;
  385. do
  386. {
  387. // 是否应该停止下载
  388. Check( m_bStopDownload, HTTP_RESULT_STOP);
  389. ZeroMemory(szReadBuf,READ_BUFFER_SIZE+1);
  390. int nRet =  m_cBufSocket.BSDGetData(szReadBuf,READ_BUFFER_SIZE,m_dwRecvTimeout);
  392. //seawind
  393. Check( m_bStopDownload, HTTP_RESULT_STOP);
  395. if (nRet <= 0)
  396. {
  397. ////////////////////////////////////////
  398. if( m_dwDownloadSize == 0xFFFFFFFF )
  399. break;
  400. ///////////////////////////////////////
  401. m_nErrorCount++;
  402. bSendRequestAgain = TRUE;
  403. break;// 跳出内层循环
  404. }
  406. // 将数据写入文件
  407. try
  408. {
  409. fileDown.Write(szReadBuf,nRet);
  410. }
  411. catch(CFileException* e)
  412. {
  413. e->Delete();
  414. bSendRequestAgain = TRUE;
  415. break;// 跳出内层循环
  416. }
  418. m_dwDownloadedSize += nRet;
  419. m_dwFileDownloadedSize += nRet;
  421. // 通知消息
  422. if( m_bNotify )
  423. {
  424. HTTPDOWNLOADSTATUS DownloadStatus;
  425. DownloadStatus.nStatusType = HTTP_STATUS_FILEDOWNLOADEDSIZE;
  426. DownloadStatus.dwFileDownloadedSize = m_dwFileDownloadedSize;
  427. DownloadStatus.dwFileSize = m_dwFileSize;
  428. DownloadStatus.strFileName = m_strFileName;
  430. if (WaitForSingleObject(m_hStopEvent, 0) != WAIT_OBJECT_0)
  431. ::SendMessage(m_hNotifyWnd,m_nNotifyMessage,MSG_HTTPDOWNLOAD_STATUS,(LPARAM)&DownloadStatus);
  432. else
  433. return HTTP_RESULT_STOP;
  434. }
  436. }while(m_dwDownloadedSize < m_dwDownloadSize);
  438. // 关闭文件
  439. fileDown.Close();
  441. } // WHILE LOOP
  443. //关闭SOCKET
  444. CloseSocket();
  446. // 文件改名
  447. //首先将已有的文件删除
  448. ::DeleteFile(m_strSavePath);
  450. //再将新下载的文件改名
  451. ::MoveFile(m_strTempSavePath, m_strSavePath);
  452.     
  453.     //再将新下载的文件的时间改回去
  454.     CFileStatus fileStatus;
  455.     
  456.     fileStatus.m_size = 0;
  457.     fileStatus.m_attribute = 0;
  458.     
  459.     try
  460.     {
  461.         CFile::GetStatus(m_strSavePath, fileStatus);
  462.         
  463.         fileStatus.m_mtime = m_TimeLastModified;
  464.         CFile::SetStatus(m_strSavePath,fileStatus);
  465.     }
  466.     catch(CFileException *e)
  467.     {
  468.         e->Delete();
  469.     }
  470.     
  471.     return HTTP_RESULT_SUCCESS;
  472. }
  475. //////////////////////////////////////////////////////////////////////////////////
  476. // 函数名：int SendRequest(
  477. // int nVerb /*= HTTP_VERB_GET*/,
  478. // LPTSTR lpszFtpURL/* = NULL*/ ) 
  479. // 用  途：发送HTTP请求
  480. // 对全局变量的影响：无
  481. // 参  数：
  482. // nVerb      : HTTP命令
  483. // lpszFtpURL : 如果重定向到FTP,此指针用来保存FTP的URL
  484. // 返回值：int
  485. // HTTP_REQUEST_STOP : 用户终止
  486. // HTTP_REQUEST_ERROR : 发生一般网络错误，可以重试
  487. // HTTP_REQUEST_SUCCESS : 请求成功
  488. // HTTP_REQUEST_FAIL : 发生验证错误，不用重试
  489. // HTTP_REQUEST_REDIRECT_FTP : 请求被重定向到FTP
  490. ////////////////////////////////////////////////////////////////////////////////
  491. int CHttpDownload::SendRequest(int nVerb /*= HTTP_VERB_GET*/,LPTSTR lpszFtpURL/* = NULL*/)
  492. {
  493. // 判断HTTP命令是否合法
  494. if( nVerb < HTTP_VERB_MIN || nVerb > HTTP_VERB_MAX )
  495. return HTTP_REQUEST_FAIL;
  497. // 保存HTTP命令
  498. m_nVerb = nVerb;
  500. while (TRUE)
  501. {
  502. CString strSend,strAuth,strHeader,strVerb,strCookie,strRange,strText;
  503. int nRet;
  504. char szReadBuf[1025];
  505. DWORD dwContentLength,dwStatusCode;
  507. strSend = _T("");
  508. strAuth = _T("");
  509. strHeader = _T("");
  510. strVerb = _T("");
  511. strCookie = _T("");
  512. strRange = _T("");
  513. strText = _T("");
  515. m_dwFileDownloadedSize = 0; // 文件已经下载大小
  516. m_dwDownloadSize    = 0; // 要下载大小
  517. strVerb = HTTP_VERB_STR[m_nVerb]; // HTTP命令
  519. ///////////////////////////////////////
  520. // 目前的版本中，此信息并没有用
  521. m_strHeader = _T("");
  522. m_dwHeaderSize = 0;
  523. //////////////////////////////////////
  525. Check( !CreateSocket(), HTTP_REQUEST_FAIL); // 创建SOCKET
  526. Check( m_bStopDownload, HTTP_REQUEST_STOP); // 是否应该停止下载
  528. // 建立连接
  529. nRet = MakeConnection( &m_cBufSocket,m_strServer,m_nPort);
  530. Check( nRet != HTTP_REQUEST_SUCCESS, nRet);
  531. Check( m_bStopDownload, HTTP_REQUEST_STOP); // 是否应该停止下载
  533. if( m_nProxyType == HTTP_PROXY_HTTPGET ) // HTTP_GET代理
  534. {
  535. strSend  = strVerb  + m_strDownloadUrl + " HTTP/1.1rn";
  536. if( m_bProxyAuthorization )
  537. {
  538. strAuth = _T("");
  539. Base64Encode(m_strProxyUsername+":"+m_strProxyPassword,strAuth);
  540. strSend += "Proxy-Authorization: Basic "+strAuth+"rn";
  541. }
  542. }
  543. else // 没有Proxy或者不是HTTP_GET代理
  544. strSend  = strVerb  + m_strObject + " HTTP/1.1rn";
  546. // 访问的页面是否是受保护的
  547. if( m_bAuthorization )
  548. {
  549. strAuth = _T("");
  550. Base64Encode(m_strUsername+":"+m_strPassword,strAuth);
  551. strSend += "Authorization: Basic "+strAuth+"rn";
  552. }
  554. strSend += "Host: " + m_strServer + "rn";
  555. strSend += "Accept: */*rn";
  556. strSend += "Pragma: no-cachern"; 
  557. strSend += "Cache-Control: no-cachern";
  558. strSend += "User-Agent: " + m_strUserAgent + "rn";
  559. if( !m_strReferer.IsEmpty() )
  560. strSend += "Referer: " + m_strReferer + "rn";
  561. strSend += "Connection: closern";
  563. // 判断是否使用Cookie
  564. if( m_bUseIECookie )
  565. {
  566. // 获取IE的Cookie
  567. if( GetIECookie(strCookie,m_strDownloadUrl) )
  568. m_strCookie = strCookie;
  569. else
  570. m_strCookie = _T("");
  571. }
  573. // Cookie
  574. if( !m_strCookie.IsEmpty() )
  575. strSend += "Cookie: " + m_strCookie + "rn";
  577. // 是否保存服务器的返回
  578. if( m_bSaveResponse )
  579. {
  580. // 判断m_strSavePath是否为路径
  581. GetFileName();
  582. if( m_strSavePath.Right(1) == '\' )
  583. {
  584. m_strTempSavePath = m_strSavePath;
  585. m_strTempSavePath += m_strFileName;
  586. m_strTempSavePath += ".tmp";
  587. }
  589. // 查看文件已经下载的长度
  590. CFileStatus fileDownStatus;
  591. if (CFile::GetStatus(m_strTempSavePath,fileDownStatus) && !m_bForceDownload )
  592. {
  593. m_dwFileDownloadedSize = fileDownStatus.m_size;
  594. if (m_dwFileDownloadedSize > 0)
  595. {
  596. strRange.Format(_T("Range: bytes=%d-rn"),m_dwFileDownloadedSize );
  597. }
  598. }
  599. strSend += strRange;
  600. }
  602. // Post?
  603. switch( m_nVerb )
  604. {
  605. case HTTP_VERB_GET:
  606. case HTTP_VERB_HEAD:
  607. // strSend += strRange;
  608. break;
  609. case HTTP_VERB_POST:
  610. strSend += "Content-Type: application/x-www-form-urlencodedrn";
  611. strText.Format("Content-Length: %drn", m_strData.GetLength() );
  612. strSend += strText;
  613. break;
  614. case HTTP_VERB_PUT:
  615. default:
  616. break;
  617. }
  619. //必须要加一个空行，否则Http服务器将不会应答
  620. strSend += "rn";
  622. // Post Data?
  623. switch( m_nVerb )
  624. {
  625. case HTTP_VERB_GET:
  626. case HTTP_VERB_HEAD:
  627. break;
  628. case HTTP_VERB_POST:
  629. case HTTP_VERB_PUT:
  630. strSend += m_strData;
  631. break;
  632. default:
  633. break;
  634. }
  636. //发送请求
  637. nRet =  m_cBufSocket.Send((LPCTSTR)strSend,strSend.GetLength(),m_dwSendTimeout);
  638. if( nRet < strSend.GetLength() )
  639. {
  640. if ( m_cBufSocket.GetLastError() == WSAETIMEDOUT) // 超时
  641. continue;
  642. else // 其他错误，可能是网络断了，等待一段时间后重试
  643. return HTTP_REQUEST_ERROR;
  644. }
  646. // 接收HTTP头
  647. BOOL bAgain = TRUE;
  648. while( bAgain )
  649. {
  650. bAgain = FALSE;
  651. strHeader.Empty();
  652. while( TRUE )
  653. {
  654. Check(m_bStopDownload,HTTP_REQUEST_STOP);
  655. ZeroMemory(szReadBuf,1025);
  656. nRet = m_cBufSocket.BSDGetString(szReadBuf,1024,m_dwRecvTimeout);
  658. //seawind
  659. Check(m_bStopDownload,HTTP_REQUEST_STOP);
  661. Check( nRet == SOCKET_FAIL ,HTTP_REQUEST_ERROR);
  663. if( szReadBuf[0] == '' ) // We have encountered "rnrn"
  664. break; 
  666. strHeader += szReadBuf;
  667. if( nRet == SOCKET_SUCCESS )
  668. strHeader += "rn";
  669. }
  671. ///////////////////////////////////////
  672. // 目前的版本中，此信息并没有用
  673. m_strHeader = strHeader;
  674. m_dwHeaderSize = m_strHeader.GetLength();
  675. //////////////////////////////////////
  676. nRet = GetInfo(strHeader,dwContentLength,dwStatusCode,m_TimeLastModified,strCookie,lpszFtpURL);
  677. switch ( nRet )
  678. {
  679. case HTTP_FAIL:
  680. return HTTP_REQUEST_FAIL;
  681. break;
  682. case HTTP_REDIRECT:
  683. m_nVerb = HTTP_VERB_GET;
  684. continue;
  685. break;
  686. case HTTP_REDIRECT_FTP:
  687. return HTTP_REQUEST_REDIRECT_FTP;
  688. break;
  689. case HTTP_SUCCESS:
  691. if( dwStatusCode >= 100  && dwStatusCode <200 )
  692. {
  693. bAgain = TRUE;
  694. continue; // Inner Loop
  695. }
  697. if( m_bSaveResponse )
  698. {
  699. m_dwDownloadSize = dwContentLength;
  700. // 应该判断一下服务器是否支持断点续传
  701. if( strRange.IsEmpty() )
  702. {
  703. m_dwFileSize = dwContentLength; // 整个文件的长度
  705. //seawind
  706. m_bSupportResume = FALSE;
  707. }
  708. else
  709. {
  710. if ( dwStatusCode == 206 ) //支持断点续传
  711. {
  712. m_dwFileSize = (dwContentLength == 0 )?0:(m_dwFileDownloadedSize +dwContentLength);
  713. m_bSupportResume = TRUE;
  714. }
  715. else //不支持断点续传
  716. {
  717. m_bSupportResume = FALSE;
  718. m_dwFileDownloadedSize = 0; //不支持断点续传，此值要设为0
  719. m_dwFileSize = dwContentLength;
  720. }
  721. }// strRange
  722. }// m_bSaveResponse
  723. return HTTP_REQUEST_SUCCESS;
  724. break;
  725. default:
  726. return HTTP_REQUEST_FAIL;
  727. break;
  728. }// SWITCH
  730. } // WHILE bAgain
  732. }// WHILE LOOP
  734. return HTTP_REQUEST_SUCCESS;
  735. }
  739. //////////////////////////////////////////////////////////////////////////////////
  740. // 函数名：void SetProxy(
  741. // LPCTSTR lpszProxyServer,
  742. // int nProxyPort,
  743. // BOOL bProxy/*=TRUE*/,
  744. // BOOL bProxyAuthorization /*= FALSE*/,
  745. // LPCTSTR lpszProxyUsername /*= NULL*/,
  746. // LPCTSTR lpszProxyPassword /*= NULL*/,
  747. // int nProxyType /*= HTTP_PROXY_HTTPGET*/ ) 
  748. // 用  途：设置代理及代理认证方式
  749. // 对全局变量的影响：无
  750. // 参  数：
  751. // lpszProxyServer     : 代理服务器的地址(IP地址或者域名)
  752. // nProxyPort          : 代理服务器的端口
  753. // bProxy              : 是否使用代理(该值为FALSE的时候，前两个参数将被忽略)
  754. // bProxyAuthorization : 代理是否需要验证
  755. // lpszProxyUsername   : 代理服务器验证的用户名
  756. // lpszProxyPassword   : 代理服务器验证的口令
  757. // nProxyType          : 代理服务器的类型
  758. // 返回值：void
  759. ////////////////////////////////////////////////////////////////////////////////
  760. void CHttpDownload::SetProxy(LPCTSTR lpszProxyServer,int nProxyPort,BOOL bProxy/*=TRUE*/,BOOL bProxyAuthorization /*= FALSE*/,LPCTSTR lpszProxyUsername /*= NULL*/,LPCTSTR lpszProxyPassword /*= NULL*/,int nProxyType /*= HTTP_PROXY_HTTPGET*/)
  761. {
  762. if( bProxy && nProxyType == HTTP_PROXY_USEIE ) // 使用IE的Proxy设置
  763. {
  764. if( !GetIEProxy(m_strProxyServer,m_nProxyPort,m_nProxyType) )
  765. {
  766. m_bProxy = FALSE;
  767. m_bProxyAuthorization = FALSE;
  768. m_nProxyPort = 0;
  769. m_nProxyType = HTTP_PROXY_NONE;
  770. m_strProxyServer = _T("");
  771. m_strProxyUsername = _T("");
  772. m_strProxyPassword = _T("");
  773. }
  774. else
  775. {
  776. m_bProxy = TRUE;
  777. if( lpszProxyUsername != NULL )
  778. {
  779. m_bProxyAuthorization = TRUE;
  780. m_strProxyUsername = lpszProxyUsername;
  781. m_strProxyPassword = lpszProxyPassword;
  782. }
  783. else
  784. {
  785. m_bProxyAuthorization = FALSE;
  786. m_strProxyUsername = _T("");
  787. m_strProxyPassword = _T("");
  788. }
  790. }
  791. return; // 返回
  792. }
  794. if( bProxy && lpszProxyServer != NULL)
  795. {
  796. m_bProxy = TRUE;
  797. m_strProxyServer = lpszProxyServer;
  798. m_nProxyPort = nProxyPort;
  799. m_nProxyType = nProxyType;
  801. if( bProxyAuthorization && lpszProxyUsername != NULL)
  802. {
  803. m_bProxyAuthorization = TRUE;
  804. m_strProxyUsername = lpszProxyUsername;
  805. m_strProxyPassword = lpszProxyPassword;
  806. }
  807. else
  808. {
  809. m_bProxyAuthorization = FALSE;
  810. m_strProxyUsername = _T("");
  811. m_strProxyPassword = _T("");
  812. }
  813. }
  814. else
  815. {
  816. m_bProxy = FALSE;
  817. m_bProxyAuthorization = FALSE;
  818. m_nProxyPort = 0;
  819. m_nProxyType = HTTP_PROXY_NONE;
  820. m_strProxyServer = _T("");
  821. m_strProxyUsername = _T("");
  822. m_strProxyPassword = _T("");
  823. }
  824. }
  827. //////////////////////////////////////////////////////////////////////////////////
  828. // 函数名：void SetAuthorization(
  829. // LPCTSTR lpszUsername,
  830. // LPCTSTR lpszPassword,
  831. // BOOL bAuthorization/* = TRUE*/ ) 
  832. // 用  途：设置WWW认证信息(访问被保护的页面时需要)
  833. // 对全局变量的影响：无
  834. // 参  数：
  835. // lpszUsername   : 访问页面的用户名
  836. // lpszPassword   : 口令
  837. // bAuthorization : 是否需要验证(此值为FALSE时，前两个参数将被忽略)
  838. // 返回值：void
  839. ////////////////////////////////////////////////////////////////////////////////
  840. void CHttpDownload::SetAuthorization(LPCTSTR lpszUsername,LPCTSTR lpszPassword,BOOL bAuthorization/* = TRUE*/ )
  841. {
  842. if( bAuthorization && lpszUsername != NULL )
  843. {
  844. m_bAuthorization = TRUE;
  845. m_strUsername  = lpszUsername;
  846. m_strPassword  = lpszPassword;
  847. }
  848. else
  849. {
  850. m_bAuthorization = FALSE;
  851. m_strUsername  = _T("");
  852. m_strPassword  = _T("");
  853. }
  854. }
  856. //////////////////////////////////////////////////////////////////////////////////
  857. // 函数名：void SetNotifyWnd(
  858. // HWND hNotifyWnd,
  859. // int nNotifyMsg,
  860. // BOOL bNotify /*= TRUE */ ) 
  861. // 用  途：设置是否需要发送消息给调用窗口
  862. // 对全局变量的影响：无
  863. // 参  数：
  864. // hNotifyWnd : 接收消息的窗口
  865. // nNotifyMsg : 消息ID
  866. // bNotify    : 是否发送消息(此值为FALSE时，前两个参数将被忽略)
  867. // 返回值：void
  868. ////////////////////////////////////////////////////////////////////////////////
  869. void CHttpDownload::SetNotifyWnd(HWND hNotifyWnd, int nNotifyMsg, BOOL bNotify /*= TRUE */)
  870. {
  871. if( bNotify && (hNotifyWnd != NULL) && ::IsWindow(hNotifyWnd) )
  872. {
  873. m_bNotify = TRUE;
  874. m_hNotifyWnd = hNotifyWnd;
  875. m_nNotifyMessage = nNotifyMsg;
  876. }
  877. else
  878. {
  879. m_bNotify = FALSE;
  880. m_hNotifyWnd = NULL;
  881. m_nNotifyMessage = 0;
  882. }
  883. }
  885. //////////////////////////////////////////////////////////////////////////////////
  886. // 函数名：void SetTimeout(
  887. // DWORD dwSendTimeout,
  888. // DWORD dwRecvTimeout,
  889. // DWORD dwConnTimeout ) 
  890. // 用  途：设置超时
  891. // 对全局变量的影响：无
  892. // 参  数：
  893. // dwSendTimeout : 发送超时
  894. // dwRecvTimeout : 接收超时
  895. // dwConnTimeout : 连接超时
  896. // 返回值：void
  897. ////////////////////////////////////////////////////////////////////////////////
  898. void CHttpDownload::SetTimeout(DWORD dwSendTimeout, DWORD dwRecvTimeout, DWORD dwConnTimeout)
  899. {
  900. if( dwSendTimeout > 0 )
  901. m_dwSendTimeout = dwSendTimeout;
  903. if( dwRecvTimeout > 0 )
  904. m_dwRecvTimeout = dwRecvTimeout;
  906. if( dwConnTimeout > 0 )
  907. m_dwConnTimeout = dwConnTimeout;
  910. }
  913. //////////////////////////////////////////////////////////////////////////////////
  914. // 函数名：void SetUserAgent( LPCTSTR lpszUserAgent ) 
  915. // 用  途：设置UserAgent
  916. // 对全局变量的影响：无
  917. // 参  数：
  918. // lpszUserAgent : 新的UserAgent(例如：Mozilla/4.0 (compatible; MSIE 5.0; Windows 98) )
  919. // 返回值：void
  920. ////////////////////////////////////////////////////////////////////////////////
  921. void CHttpDownload::SetUserAgent(LPCTSTR lpszUserAgent)
  922. {
  923. if( lpszUserAgent == NULL )
  924. m_strUserAgent = DEFAULT_USERAGENT;
  925. else
  926. m_strUserAgent = lpszUserAgent;
  928. if( m_strUserAgent.IsEmpty())
  929. m_strUserAgent = DEFAULT_USERAGENT;
  931. }
  933. //////////////////////////////////////////////////////////////////////////////////
  934. // 函数名：void SetReferer( LPCTSTR lpszReferer ) 
  935. // 用  途：设置Referer(提交者)
  936. // 对全局变量的影响：无
  937. // 参  数：
  938. // lpszReferer :  新的Referer
  939. // 返回值：void
  940. ////////////////////////////////////////////////////////////////////////////////
  941. void CHttpDownload::SetReferer(LPCTSTR lpszReferer)
  942. {
  943. if( lpszReferer != NULL )
  944. m_strReferer = lpszReferer;
  945. else
  946. m_strReferer = _T("");
  947. }
  950. //////////////////////////////////////////////////////////////////////////////////
  951. // 函数名：void SetRetry(
  952. // int nRetryType,
  953. // int nRetryDelay,
  954. // int nRetryMax ) 
  955. // 用  途：设置重试的机制
  956. // 对全局变量的影响：无
  957. // 参  数：
  958. // nRetryType  :  重试方式
  959. // HTTP_RETRY_NONE  - 不重试
  960. // HTTP_RETRY_TIMES - 重试一定次数
  961. // HTTP_RETRY_ALWAYS- 总是重试(可能陷入死循环)
  962. // nRetryDelay : 重试延迟
  963. // nRetryMax   : 最大重试次数(nRetryType为HTTP_RETRY_TIMES时，此值才有意义)
  964. // 返回值：void
  965. ////////////////////////////////////////////////////////////////////////////////
  966. void CHttpDownload::SetRetry(int nRetryType, int nRetryDelay, int nRetryMax)
  967. {
  968. m_nRetryType  = nRetryType;
  969. m_nRetryDelay = nRetryDelay;
  970. m_nRetryMax   = nRetryMax;
  972. // 检查一下m_nRetryMax,如果为0，设为缺省值
  973. //if( (HTTP_RETRY_TIMES == m_nRetryType) && (0 == m_nRetryMax) )
  974. // m_nRetryMax = DEFAULT_HTTP_RETRY_MAX;
  975. }
  978. //////////////////////////////////////////////////////////////////////////////////
  979. // 函数名：BOOL GetDownloadFileStatus(
  980. // IN LPCTSTR lpszDownloadUrl,
  981. // OUT DWORD &dwFileSize,
  982. // OUT CTime &FileTime ) 
  983. // 用  途：获取下载文件的状态
  984. // 对全局变量的影响：无
  985. // 参  数：
  986. // lpszDownloadUrl : 文件的URL
  987. // dwFileSize      : 用于保存文件的大小
  988. // FileTime        : 用于保存文件的修改时间
  989. // 返回值：BOOL
  990. ////////////////////////////////////////////////////////////////////////////////
  991. BOOL CHttpDownload::GetDownloadFileStatus(IN LPCTSTR lpszDownloadUrl,OUT DWORD &dwFileSize, OUT CTime &FileTime)
  992. {
  993. // 检验要下载的URL是否为空
  994. m_strDownloadUrl = lpszDownloadUrl;
  995. m_strDownloadUrl.TrimLeft();
  996. m_strDownloadUrl.TrimRight();
  997. if( m_strDownloadUrl.IsEmpty() )
  998. return FALSE;
  1000. // 检验要下载的URL是否有效
  1001. if ( !ParseURL(m_strDownloadUrl, m_strServer, m_strObject, m_nPort))
  1002. {
  1003. // 在前面加上"http://"再试
  1004. m_strDownloadUrl = _T("http://") + m_strDownloadUrl;
  1005. if ( !ParseURL(m_strDownloadUrl, m_strServer, m_strObject, m_nPort) )
  1006. {
  1007. TRACE(_T("Failed to parse the URL: %sn"), m_strDownloadUrl);
  1008. return FALSE;
  1009. }
  1010. }
  1012. m_strSavePath = "|";
  1013. m_strTempSavePath = "|";
  1014. m_bStopDownload = FALSE;
  1016. if ( SendRequest() !=  HTTP_REQUEST_SUCCESS )
  1017. return FALSE;
  1019. dwFileSize  = m_dwDownloadSize;
  1020. FileTime = m_TimeLastModified;
  1022. return TRUE;
  1023. }
  1025. //////////////////////////////////////////////////////////////////////////////////
  1026. // 函数名：BOOL ParseURL(
  1027. // IN LPCTSTR lpszURL,
  1028. // OUT CString &strServer,
  1029. // OUT CString &strObject,
  1030. // OUT int& nPort ) 
  1031. // 用  途：从URL里面拆分出Server和Object来
  1032. // 对全局变量的影响：无
  1033. // 参  数：
  1034. // lpszURL   : 待拆分的URL
  1035. // strServer : 服务器地址
  1036. // strObject : 服务器上的目标文件
  1037. // nPort     : 服务器的端口
  1038. // 返回值：BOOL
  1039. ////////////////////////////////////////////////////////////////////////////////
  1040. BOOL CHttpDownload::ParseURL(IN LPCTSTR lpszURL, OUT CString &strServer, OUT CString &strObject,OUT int& nPort)
  1041. {
  1042. CString strURL(lpszURL);
  1043. strURL.TrimLeft();
  1044. strURL.TrimRight();
  1046. // 清除数据
  1047. strServer = _T("");
  1048. strObject = _T("");
  1049. nPort   = 0;
  1051. int nPos = strURL.Find("://");
  1052. if( nPos == -1 )
  1053. return FALSE;
  1055. // 进一步验证是否为http://
  1056. CString strTemp = strURL.Left( nPos+lstrlen("://") );
  1057. strTemp.MakeLower();
  1058. if( strTemp.Compare("http://") != 0 )
  1059. return FALSE;
  1061. strURL = strURL.Mid( strTemp.GetLength() );
  1062. nPos = strURL.Find('/');
  1063. if ( nPos == -1 )
  1064. return FALSE;
  1066. strObject = strURL.Mid(nPos);
  1067. strTemp   = strURL.Left(nPos);
  1069. ///////////////////////////////////////////////////////////////
  1070. /// 注意：并没有考虑URL中有用户名和口令的情形和最后有#的情形
  1071. /// 例如：http://abc@def:www.yahoo.com:81/index.html#link1
  1072. /// 
  1073. //////////////////////////////////////////////////////////////
  1075. // 查找是否有端口号
  1076. nPos = strTemp.Find(":");
  1077. if( nPos == -1 )
  1078. {
  1079. strServer = strTemp;
  1080. nPort   = DEFAULT_HTTP_PORT;
  1081. }
  1082. else
  1083. {
  1084. strServer = strTemp.Left( nPos );
  1085. strTemp   = strTemp.Mid( nPos+1 );
  1086. nPort   = (int)_ttoi((LPCTSTR)strTemp);
  1087. }
  1088. return TRUE;
  1089. }
  1091. //////////////////////////////////////////////////////////////////////////////////
  1092. // 函数名：int GetInfo(
  1093. // IN LPCTSTR lpszHeader,
  1094. // OUT DWORD& dwContentLength,
  1095. // OUT DWORD& dwStatusCode,
  1096. // OUT CTime& TimeLastModified,
  1097. // OUT CString& strCookie,
  1098. // LPTSTR lpszFtpURL ) 
  1099. // 用  途：从返回的HTTP协议头里获得必要的信息
  1100. // 对全局变量的影响：无
  1101. // 参  数：
  1102. // lpszHeader       : HTTP协议头
  1103. // dwContentLength  : content-length字段的值
  1104. // dwStatusCode     : 状态码
  1105. // TimeLastModified : last-modified字段的值
  1106. // strCookie        : 用来保存Cookie
  1107. // lpszFtpURL       : 用来保存重定向到FTP的URL
  1108. // 返回值：int
  1109. // HTTP_FAIL : 不合法的HTTP协议头，失败
  1110. // HTTP_REDIRECT : 重定向
  1111. // HTTP_SUCCESS : 成功
  1112. ////////////////////////////////////////////////////////////////////////////////
  1113. int CHttpDownload::GetInfo(IN LPCTSTR lpszHeader,OUT DWORD& dwContentLength,OUT DWORD& dwStatusCode,OUT CTime& TimeLastModified,OUT CString& strCookie,LPTSTR lpszFtpURL)
  1114. {
  1115. dwContentLength = 0;
  1116. dwStatusCode = 0;
  1117. TimeLastModified= CTime::GetCurrentTime();
  1118. strCookie = _T("");
  1120. CString strHeader(lpszHeader);
  1121. strHeader.MakeLower();
  1123. //拆分出HTTP应答的头信息的第一行
  1124. int nPos = strHeader.Find("rn");
  1125. if (nPos == -1)
  1126. return HTTP_FAIL;
  1128. // 获得返回码: Status Code
  1129. CString strStatusLine = strHeader.Left(nPos);
  1130. strStatusLine.TrimLeft();
  1131. strStatusLine.TrimRight();
  1133. nPos = strStatusLine.Find(' ');
  1134. if( nPos == -1 )
  1135. return HTTP_FAIL;
  1136. strStatusLine = strStatusLine.Mid(nPos+1);
  1137. nPos = strStatusLine.Find(' ');
  1138. if( nPos == -1 )
  1139. return HTTP_FAIL;
  1140. strStatusLine = strStatusLine.Left(nPos);
  1141. dwStatusCode = (DWORD)_ttoi((LPCTSTR)strStatusLine);
  1143. // 检查返回码
  1144. if( dwStatusCode >= 300 && dwStatusCode < 400 ) //首先检测一下服务器的应答是否为重定向
  1145. {
  1146. nPos = strHeader.Find("location:");
  1147. if (nPos == -1)
  1148. return HTTP_FAIL;
  1150. CString strRedirectURL = strHeader.Mid(nPos + strlen("location:"));
  1151. nPos = strRedirectURL.Find("rn");
  1152. if (nPos == -1)
  1153. return HTTP_FAIL;
  1155. strRedirectURL = strRedirectURL.Left(nPos);
  1156. strRedirectURL.TrimLeft();
  1157. strRedirectURL.TrimRight();
  1159. // 设置Referer
  1160. m_strReferer = m_strDownloadUrl;
  1162. // 判断是否重定向到其他的服务器
  1163. // 从版本1.7开始，支持重定向到FTP服务器
  1164. if( strRedirectURL.Mid(0,6) == _T("ftp://") )
  1165. {
  1166. if( lpszFtpURL != NULL )
  1167. memcpy( lpszFtpURL,(LPCTSTR)strRedirectURL,strRedirectURL.GetLength() );
  1169. return HTTP_REDIRECT_FTP;
  1170. }
  1172. if( strRedirectURL.Mid(0,7) == _T("http://") )
  1173. {
  1174. m_strDownloadUrl = strRedirectURL;
  1175. // 检验要下载的URL是否有效
  1176. if ( !ParseURL(m_strDownloadUrl, m_strServer, m_strObject, m_nPort))
  1177. return HTTP_FAIL;
  1178. return HTTP_REDIRECT;
  1179. }
  1181. // 重定向到本服务器的其他地方
  1182. strRedirectURL.Replace("\","/");
  1184. // 是相对于根目录
  1185. if( strRedirectURL[0] == '/' )
  1186. {
  1187. m_strObject = strRedirectURL;
  1188. return HTTP_REDIRECT;
  1189. }
  1191. // 是相对当前目录
  1192. int nParentDirCount = 0;
  1193. nPos = strRedirectURL.Find("../");
  1194. while (nPos != -1)
  1195. {
  1196. strRedirectURL = strRedirectURL.Mid(nPos+3);
  1197. nParentDirCount++;
  1198. nPos = strRedirectURL.Find("../");
  1199. }
  1200. for (int i=0; i<=nParentDirCount; i++)
  1201. {
  1202. nPos = m_strDownloadUrl.ReverseFind('/');
  1203. if (nPos != -1)
  1204. m_strDownloadUrl = m_strDownloadUrl.Left(nPos);
  1205. }
  1206. m_strDownloadUrl = m_strDownloadUrl+"/"+strRedirectURL;
  1208. if ( !ParseURL(m_strDownloadUrl, m_strServer, m_strObject, m_nPort))
  1209. return HTTP_FAIL;
  1210. return HTTP_REDIRECT;
  1211. }
  1213. // 发生错误，不用重试(不管是服务器或者是客户端的)
  1214. if( dwStatusCode >=400 )
  1215. return HTTP_FAIL;
  1217. // 获取ContentLength(如果没有，就返回0，表示长度未知)
  1218. nPos = strHeader.Find("content-length:");
  1219. if( nPos != -1 )
  1220. {
  1221. CString strConentLen = strHeader.Mid(nPos + strlen("content-length:"));
  1222. nPos = strConentLen.Find("rn");
  1223. if (nPos != -1)
  1224. {
  1225. strConentLen = strConentLen.Left(nPos);
  1226. strConentLen.TrimLeft();
  1227. strConentLen.TrimRight();
  1229. // Content-Length:
  1230. dwContentLength = (DWORD) _ttoi( (LPCTSTR)strConentLen );
  1231. }
  1232. }
  1234. // 获取Last-Modified:
  1235. nPos = strHeader.Find("last-modified:");
  1236. if (nPos != -1)
  1237. {
  1238. CString strTime = strHeader.Mid(nPos + strlen("last-modified:"));
  1239. nPos = strTime.Find("rn");
  1240. if (nPos != -1)
  1241. {
  1242. strTime = strTime.Left(nPos);
  1243. strTime.TrimLeft();
  1244. strTime.TrimRight();
  1245. TimeLastModified = GetTime(strTime);
  1246. }
  1247. }
  1249. // 获取Cookie
  1250. nPos = strHeader.Find("set-cookie:");
  1251. if (nPos != -1)
  1252. {
  1253. CString strText = strHeader.Mid(nPos + strlen("set-cookie:"));
  1254. nPos = strText.Find("rn");
  1255. if (nPos != -1)
  1256. {
  1257. strCookie = strText.Left(nPos);
  1258. strCookie.TrimLeft();
  1259. strCookie.TrimRight();
  1261. int nPos1 = strCookie.Find("expires=");
  1262. int nPos2 = strCookie.Find("domain=");
  1263. int nPos3 = strCookie.Find("path=");
  1265. nPos = -1;
  1266. if( nPos1 != -1 )
  1267. nPos = nPos1;
  1269. if( nPos2 != -1 )
  1270. nPos = ( nPos == -1 )? nPos2:((nPos > nPos2 )?nPos2:nPos);
  1272. if( nPos3 != -1 )
  1273. nPos = ( nPos == -1 )? nPos3:((nPos > nPos3 )?nPos3:nPos);
  1275. if( nPos != -1 )
  1276. strCookie = strCookie.Left( nPos );
  1278. // 删除最后的;
  1279. if( strCookie.Right(1) == ";" )
  1280. strCookie = strCookie.Left( strCookie.GetLength() -1 );
  1281. }
  1283. }
  1284. return HTTP_SUCCESS;
  1285. }
  1287. //////////////////////////////////////////////////////////////////////////////////
  1288. // 函数名：CTime GetTime( LPCTSTR lpszTime ) 
  1289. // 用  途：将GMT字符串转化成时间
  1290. // 对全局变量的影响：无
  1291. // 参  数：
  1292. // lpszTime : 时间字符串
  1293. // 返回值：CTime
  1294. ////////////////////////////////////////////////////////////////////////////////
  1295. CTime CHttpDownload::GetTime(LPCTSTR lpszTime)
  1296. {
  1297. int nDay,nMonth,nYear,nHour,nMinute,nSecond;
  1299. CString strTime = lpszTime;
  1300. int nPos = strTime.Find(',');
  1301. if (nPos != -1)
  1302. {
  1303. strTime = strTime.Mid(nPos+1);
  1304. strTime.TrimLeft();
  1306. CString strDay,strMonth,strYear,strHour,strMinute,strSecond;
  1307. CString strAllMonth = "jan,feb,mar,apr,may,jun,jul,aug,sep,oct,nov,dec";
  1308. strDay = strTime.Left(2);
  1309. nDay = atoi(strDay);
  1310. strMonth = strTime.Mid(3,3);
  1311. strMonth.MakeLower();
  1312. nPos = strAllMonth.Find(strMonth);
  1313. if (nPos != -1)
  1314. {
  1315. strMonth.Format("%d",((nPos/4)+1));
  1316. nMonth = atoi(strMonth);
  1317. }
  1318. else
  1319. nMonth = 1;
  1320. strTime = strTime.Mid(6);
  1321. strTime.TrimLeft();
  1322. nPos = strTime.FindOneOf(" t");
  1323. if (nPos != -1)
  1324. {
  1325. strYear = strTime.Left(nPos);
  1326. nYear = atoi(strYear);
  1327. }
  1328. else
  1329. nYear = 2000;
  1330. strTime = strTime.Mid(nPos+1);
  1331. strHour = strTime.Left(2);
  1332. nHour = atoi(strHour);
  1333. strMinute = strTime.Mid(3,2);
  1334. nMinute = atoi(strMinute);
  1335. strSecond = strTime.Mid(6,2);
  1336. nSecond = atoi(strSecond);
  1337. }
  1339. CTime time(nYear,nMonth,nDay,nHour,nMinute,nSecond);
  1340. return time;
  1341. }
  1343. //////////////////////////////////////////////////////////////////////////////////
  1344. // 函数名：void GetFileName()
  1345. // 用  途：从下载URL中获取文件名
  1346. // 对全局变量的影响：无
  1347. // 参  数：无
  1348. // 返回值：void
  1349. ////////////////////////////////////////////////////////////////////////////////
  1350. void CHttpDownload::GetFileName()
  1351. {
  1352. // 获取的文件名
  1353. int nSlash = m_strObject.ReverseFind(_T('/'));
  1354. if (nSlash == -1)
  1355. nSlash = m_strObject.ReverseFind(_T('\'));
  1357. if (nSlash != -1 && m_strObject.GetLength() > 1)
  1358. m_strFileName = m_strObject.Right(m_strObject.GetLength() - nSlash - 1);
  1359. else
  1360. m_strFileName = m_strObject;
  1362. m_strFileName.TrimLeft();
  1363. m_strFileName.TrimRight();
  1364. // 文件名为空或无效
  1365. if( m_strFileName.IsEmpty() || !IsValidFileName(m_strFileName) )
  1366. m_strFileName = DEFAULT_SAVE_FILE;
  1367. }
  1369. //////////////////////////////////////////////////////////////////////////////////
  1370. // 函数名：void StopDownload()
  1371. // 用  途：终止下载 
  1372. // 对全局变量的影响：无
  1373. // 参  数：无
  1374. // 返回值：void
  1375. ////////////////////////////////////////////////////////////////////////////////
  1376. void CHttpDownload::StopDownload()
  1377. {
  1378. m_bStopDownload = TRUE;
  1380. if ( m_hStopEvent != NULL )
  1381. SetEvent(m_hStopEvent);
  1382. }
  1385. //////////////////////////////////////////////////////////////////////////////////
  1386. // 函数名：int Base64Decode(
  1387. // IN LPCTSTR lpszDecoding,
  1388. // OUT CString &strDecoded ) 
  1389. // 用  途：BASE64解码
  1390. // 对全局变量的影响：无
  1391. // 参  数：
  1392. // lpszDecoding : 待解码字符串
  1393. // strDecoded   : 解码结果
  1394. // 返回值：int，解码后的长度
  1395. ////////////////////////////////////////////////////////////////////////////////
  1396. int CHttpDownload::Base64Decode(IN LPCTSTR lpszDecoding, OUT CString &strDecoded)
  1397. {
  1398. int nIndex =0;
  1399. int nDigit;
  1400.     int nDecode[ 256 ];
  1401. int nSize;
  1402. int nNumBits = 6;
  1404. if( lpszDecoding == NULL )
  1405. return 0;
  1407. if( ( nSize = lstrlen(lpszDecoding) ) == 0 )
  1408. return 0;
  1410. // Build Decode Table
  1411. for( int i = 0; i < 256; i++ ) 
  1412. nDecode[i] = -2; // Illegal digit
  1413. for( i=0; i < 64; i++ )
  1414. {
  1415. nDecode[ m_strBase64TAB[ i ] ] = i;
  1416. nDecode[ '=' ] = -1; 
  1417.     }
  1419. // Clear the output buffer
  1420. strDecoded = _T("");
  1421. long lBitsStorage  =0;
  1422. int nBitsRemaining = 0;
  1423. int nScratch = 0;
  1424. UCHAR c;
  1426. // Decode the Input
  1427. for( nIndex = 0, i = 0; nIndex < nSize; nIndex++ )
  1428. {
  1429. c = lpszDecoding[ nIndex ];
  1431. // 忽略所有不合法的字符
  1432. if( c> 0x7F)
  1433. continue;
  1435. nDigit = nDecode[c];
  1436. if( nDigit >= 0 ) 
  1437. {
  1438. lBitsStorage = (lBitsStorage << nNumBits) | (nDigit & 0x3F);
  1439. nBitsRemaining += nNumBits;
  1440. while( nBitsRemaining > 7 ) 
  1441. {
  1442. nScratch = lBitsStorage >> (nBitsRemaining - 8);
  1443. strDecoded += (nScratch & 0xFF);
  1444. i++;
  1445. nBitsRemaining -= 8;
  1446. }
  1447. }
  1448.     }
  1450. return strDecoded.GetLength();
  1451. }
  1454. //////////////////////////////////////////////////////////////////////////////////
  1455. // 函数名：int Base64Encode(
  1456. // IN LPCTSTR lpszEncoding,
  1457. // OUT CString &strEncoded ) 
  1458. // 用  途：BASE64编码
  1459. // 对全局变量的影响：无
  1460. // 参  数：
  1461. // lpszEncoding : 待编码的字符串
  1462. // strEncoded   : 编码结果
  1463. // 返回值：int，编码后的长度
  1464. ////////////////////////////////////////////////////////////////////////////////
  1465. int CHttpDownload::Base64Encode(IN LPCTSTR lpszEncoding, OUT CString &strEncoded)
  1466. {
  1467. int nDigit;
  1468. int nNumBits = 6;
  1469. int nIndex = 0;
  1470. int nInputSize;
  1472. strEncoded = _T( "" );
  1473. if( lpszEncoding == NULL )
  1474. return 0;
  1476. if( ( nInputSize = lstrlen(lpszEncoding) ) == 0 )
  1477. return 0;
  1479. int nBitsRemaining = 0;
  1480. long lBitsStorage =0;
  1481. long lScratch =0;
  1482. int nBits;
  1483. UCHAR c;
  1485. while( nNumBits > 0 )
  1486. {
  1487. while( ( nBitsRemaining < nNumBits ) &&  ( nIndex < nInputSize ) ) 
  1488. {
  1489. c = lpszEncoding[ nIndex++ ];
  1490. lBitsStorage <<= 8;
  1491. lBitsStorage |= (c & 0xff);
  1492. nBitsRemaining += 8;
  1493. }
  1494. if( nBitsRemaining < nNumBits ) 
  1495. {
  1496. lScratch = lBitsStorage << ( nNumBits - nBitsRemaining );
  1497. nBits    = nBitsRemaining;
  1498. nBitsRemaining = 0;
  1499. }  
  1500. else 
  1501. {
  1502. lScratch = lBitsStorage >> ( nBitsRemaining - nNumBits );
  1503. nBits  = nNumBits;
  1504. nBitsRemaining -= nNumBits;
  1505. }
  1506. nDigit = (int)(lScratch & m_nBase64Mask[nNumBits]);
  1507. nNumBits = nBits;
  1508. if( nNumBits <=0 )
  1509. break;
  1511. strEncoded += m_strBase64TAB[ nDigit ];
  1512. }
  1513. // Pad with '=' as per RFC 1521
  1514. while( strEncoded.GetLength() % 4 != 0 )
  1515. strEncoded += '=';
  1517. return strEncoded.GetLength();
  1519. }
  1522. //////////////////////////////////////////////////////////////////////////////////
  1523. // 函数名：int MakeConnection(
  1524. // CBufSocket* pBufSocket,
  1525. // CString strServer,
  1526. // int nPort ) 
  1527. // 用  途：建立连接
  1528. // 对全局变量的影响：无
  1529. // 参  数：
  1530. // pBufSocket : CBufSocket类指针
  1531. // strServer  : 待连接服务器
  1532. // nPort      : 端口
  1533. // 返回值：int
  1534. ////////////////////////////////////////////////////////////////////////////////
  1535. int CHttpDownload::MakeConnection(CBufSocket* pBufSocket,CString strServer,int nPort)
  1536. {
  1537. CSocksPacket cSocks( pBufSocket );
  1538. DWORD dwIP;
  1539. BYTE byAuth,byAtyp;
  1540. SOCKSREPPACKET pack;
  1541. CString strAuth,strAddr;
  1543. strAuth = _T("");
  1544. strAddr = _T("");
  1546. switch( m_nProxyType )
  1547. {
  1548. case HTTP_PROXY_NONE:
  1549. Check( m_bStopDownload,HTTP_REQUEST_STOP);
  1550. Check( !pBufSocket->Connect(strServer,nPort,m_dwConnTimeout,TRUE),HTTP_REQUEST_ERROR);
  1551. break;
  1552. case HTTP_PROXY_HTTPGET:
  1553. Check( m_bStopDownload,HTTP_REQUEST_STOP);
  1554. Check( !pBufSocket->Connect(m_strProxyServer,m_nProxyPort,m_dwConnTimeout,TRUE),HTTP_REQUEST_ERROR);
  1555. break;
  1556. case HTTP_PROXY_SOCKS4A:
  1557. dwIP = CBufSocket::GetIP(strServer,TRUE);
  1558. if( dwIP == INADDR_NONE )
  1559. {
  1560. Check( m_bStopDownload,HTTP_REQUEST_STOP);
  1561. Check( !pBufSocket->Connect(m_strProxyServer,m_nProxyPort,m_dwConnTimeout,TRUE),HTTP_REQUEST_ERROR);
  1562. Check( m_bStopDownload,HTTP_REQUEST_STOP);
  1563. Check( !cSocks.SendSocks4aReq(CMD_CONNECT,nPort,strServer,m_strProxyUsername,m_dwSendTimeout),HTTP_REQUEST_ERROR);
  1565. Check( m_bStopDownload,HTTP_REQUEST_STOP);
  1566. Check( !cSocks.RecvPacket(PACKET_SOCKS4AREP,m_dwRecvTimeout),HTTP_REQUEST_ERROR);
  1567. Check( !cSocks.IsSocksOK(),HTTP_REQUEST_FAIL); // 请求有错误，重试可能是没有用的
  1568. break;// NOTICE:如果本地能够解析域名，可以使用SOCKS4 Proxy
  1569. }
  1570. case HTTP_PROXY_SOCKS4:
  1571. // 必须要得到Proxy Server的IP地址(不能为域名)
  1572. dwIP = CBufSocket::GetIP(strServer,TRUE);
  1573. Check( dwIP == INADDR_NONE,HTTP_REQUEST_ERROR);
  1575. Check( m_bStopDownload,HTTP_REQUEST_STOP);
  1576. Check( !pBufSocket->Connect(m_strProxyServer,m_nProxyPort,m_dwConnTimeout,TRUE),HTTP_REQUEST_ERROR);
  1577. Check( m_bStopDownload,HTTP_REQUEST_STOP);
  1578. Check( !cSocks.SendSocks4Req(CMD_CONNECT,nPort,dwIP,m_strProxyUsername,m_dwSendTimeout),HTTP_REQUEST_ERROR);
  1580. Check( m_bStopDownload,HTTP_REQUEST_STOP);
  1581. Check( !cSocks.RecvPacket(PACKET_SOCKS4REP,m_dwRecvTimeout),HTTP_REQUEST_ERROR);
  1582. Check( !cSocks.IsSocksOK(),HTTP_REQUEST_FAIL); // 请求有错误，重试可能是没有用的
  1583. break;
  1585. case HTTP_PROXY_SOCKS5:
  1587. Check( m_bStopDownload,HTTP_REQUEST_STOP);
  1588. Check( !pBufSocket->Connect(m_strProxyServer,m_nProxyPort,m_dwConnTimeout,TRUE),HTTP_REQUEST_ERROR);
  1590. if( m_bProxyAuthorization )
  1591. {
  1592. char ch = (char)AUTH_NONE;
  1593. strAuth += ch;
  1594. ch  = (char)AUTH_PASSWD;
  1595. strAuth += ch;
  1596. }
  1597. else
  1598. {
  1599. char ch = (char)AUTH_NONE;
  1600. strAuth += ch;
  1601. }
  1602. byAuth =(BYTE)strAuth.GetLength();
  1604. Check( m_bStopDownload,HTTP_REQUEST_STOP);
  1605. Check( !cSocks.SendSocks5AuthReq(byAuth,(LPCTSTR)strAuth,m_dwSendTimeout),HTTP_REQUEST_ERROR);
  1607. Check( m_bStopDownload,HTTP_REQUEST_STOP);
  1609. ZeroMemory(&pack,sizeof(SOCKSREPPACKET));
  1610. Check( !cSocks.RecvPacket(&pack,PACKET_SOCKS5AUTHREP,m_dwRecvTimeout),HTTP_REQUEST_ERROR);
  1611. Check( !cSocks.IsSocksOK(&pack,PACKET_SOCKS5AUTHREP),HTTP_REQUEST_FAIL); // 请求有错误，重试可能是没有用的
  1613. switch( pack.socks5AuthRep.byAuth )
  1614. {
  1615. case AUTH_NONE:
  1616. break;
  1617. case AUTH_PASSWD:
  1619. Check( !m_bProxyAuthorization,HTTP_REQUEST_FAIL);
  1620. Check( m_bStopDownload,HTTP_REQUEST_STOP);
  1621. Check( !cSocks.SendSocks5AuthPasswdReq(m_strProxyUsername,m_strProxyPassword,m_dwSendTimeout),HTTP_REQUEST_ERROR);
  1622. Check( m_bStopDownload,HTTP_REQUEST_STOP);
  1624. Check( !cSocks.RecvPacket(PACKET_SOCKS5AUTHPASSWDREP,m_dwRecvTimeout),HTTP_REQUEST_ERROR);
  1625. Check( !cSocks.IsSocksOK(),HTTP_REQUEST_FAIL); // 请求有错误，重试可能是没有用的
  1626. break;
  1627. case AUTH_GSSAPI:
  1628. case AUTH_CHAP:
  1629. case AUTH_UNKNOWN:
  1630. default:
  1631. return HTTP_REQUEST_FAIL;
  1632. break;
  1633. }
  1634. dwIP = CBufSocket::GetIP(strServer,TRUE);
  1635. if( dwIP != INADDR_NONE )
  1636. {
  1637. byAtyp = ATYP_IPV4ADDR;
  1638. //不用转换字节序
  1639. strAddr += (char)( (dwIP    ) &0x000000ff) ; 
  1640. strAddr += (char)( (dwIP>>8 ) &0x000000ff); 
  1641. strAddr += (char)( (dwIP>>16) &0x000000ff); 
  1642. strAddr += (char)( (dwIP>>24) &0x000000ff); 
  1644. }
  1645. else
  1646. {
  1647. byAtyp = ATYP_HOSTNAME;
  1648. char ch = (char)strServer.GetLength();
  1649. strAddr += ch;
  1650. strAddr += strServer;
  1651. }
  1653. Check( m_bStopDownload,HTTP_REQUEST_STOP);
  1654. Check( !cSocks.SendSocks5Req(CMD_CONNECT,byAtyp,strAddr,nPort,m_dwSendTimeout),HTTP_REQUEST_ERROR);
  1655. Check( m_bStopDownload,HTTP_REQUEST_STOP);
  1656. Check( !cSocks.RecvPacket(PACKET_SOCKS5REP,m_dwRecvTimeout),HTTP_REQUEST_ERROR);
  1657. Check( !cSocks.IsSocksOK(),HTTP_REQUEST_FAIL); // 请求有错误，重试可能是没有用的
  1658. break;
  1659. case HTTP_PROXY_HTTPCONNECT:
  1660. default:
  1661. return HTTP_REQUEST_FAIL;
  1662. break;
  1663. }
  1664. return HTTP_REQUEST_SUCCESS;
  1665. }
  1667. //////////////////////////////////////////////////////////////////////////////////
  1668. // 函数名：BOOL GetIEProxy(
  1669. // CString &strProxyServer,
  1670. // int &nProxyPort,
  1671. // int &nProxyType ) 
  1672. // 用  途：获取IE的Proxy设置
  1673. // 对全局变量的影响：无
  1674. // 参  数：
  1675. // strProxyServer : 代理服务器地址
  1676. // nProxyPort     : 代理服务器的端口
  1677. // nProxyType     : 代理服务器的类型
  1678. // 返回值：BOOL
  1679. ////////////////////////////////////////////////////////////////////////////////
  1680. BOOL CHttpDownload::GetIEProxy(CString &strProxyServer, int &nProxyPort, int &nProxyType)
  1681. {
  1682. // 读取注册表
  1683. HKEY hRegKey;
  1684. DWORD dwRet = RegOpenKeyEx(
  1685. HKEY_CURRENT_USER,
  1686. "Software\Microsoft\Windows\CurrentVersion\Internet Settings",
  1687. 0L,
  1688. KEY_QUERY_VALUE,&hRegKey);
  1689. if (dwRet != ERROR_SUCCESS)
  1690. return FALSE;
  1692. TCHAR szAddr[256] = { 0 };
  1693. DWORD dwLen = 256;
  1694. dwRet = RegQueryValueEx(hRegKey,"ProxyServer",NULL,NULL,(LPBYTE)szAddr,&dwLen);
  1695. if (dwRet != ERROR_SUCCESS)
  1696. {
  1697. RegCloseKey(hRegKey);
  1698. return FALSE;
  1699. }
  1700.     RegCloseKey(hRegKey);
  1701.     
  1702.     // 注：可能存在所有代理同一的情况，此时无 http=...ftp=...信息
  1703.     //    comment by Linsuyi, 2002/01/22  23:30
  1704.     // 
  1705.     // 分析Proxy的设置
  1706.     //http=193.168.10.1:80;socks=192.168.10.235:1080
  1707.     //193.168.10.1:1090
  1708.     //ftp=193.168.10.1:1090;gopher=193.168.10.1:1090;https=193.168.10.1:1090;socks=193.168.10.2:1080
  1709.     int nPos = -1;
  1710.     CString strProxy = szAddr;
  1711.     
  1712.     if (strProxy.IsEmpty())
  1713.         return FALSE;
  1714.     
  1715.     strProxy.MakeLower();
  1716.     nProxyType = HTTP_PROXY_HTTPGET;
  1717.     
  1718.     nPos = strProxy.Find(';');
  1719.     if (nPos != -1)
  1720.     {
  1721.         nPos = strProxy.Find("http=");        // HTTP GET Proxy
  1722.         if (nPos != -1)
  1723.             strProxy = strProxy.Mid(nPos+strlen("http="));
  1724.         else if ((nPos = strProxy.Find("socks=")) != -1) // SOCKS Proxy ( SOCKS5 )
  1725.         {
  1726.             strProxy = strProxy.Mid( nPos+strlen("socks="));
  1727.             nProxyType = HTTP_PROXY_SOCKS5;
  1728.         }
  1729.         nPos = strProxy.Find(";");
  1730.         if (nPos != -1)
  1731.          strProxy = strProxy.Left(nPos);
  1732.     }
  1733.     
  1734.     nPos = strProxy.Find(":");
  1735.     strProxyServer = strProxy.Left(nPos);
  1736.     nProxyPort = _ttoi(strProxy.Mid(nPos+1));
  1737.     if (nProxyPort <= 0)
  1738.         return FALSE;
  1739.     
  1740.     return TRUE;
  1741. }
  1743. //////////////////////////////////////////////////////////////////////////////////
  1744. // 函数名：void void SetCookie(
  1745. // LPCTSTR lpszCookie,
  1746. // BOOL bUseIECookie /*= FALSE*/ ) 
  1747. // 用  途：设置Cookie
  1748. // 对全局变量的影响：无
  1749. // 参  数：
  1750. // lpszCookie   : Cookie值 
  1751. // bUseIECookie : 是否使用IE的Cookie值
  1752. // 返回值：void
  1753. ////////////////////////////////////////////////////////////////////////////////
  1754. void CHttpDownload::SetCookie(LPCTSTR lpszCookie,BOOL bUseIECookie /*= FALSE*/ )
  1755. {
  1756. if( bUseIECookie )
  1757. {
  1758. m_bUseIECookie = TRUE;
  1759. /*
  1760. CString strCookie;
  1761. if( GetIECookie(strCookie,m_strDownloadUrl) )
  1762. m_strCookie = strCookie;
  1763. else
  1764. m_strCookie = _T("");
  1765. */
  1767. }
  1768. else
  1769. {
  1770. m_bUseIECookie = FALSE;
  1772. if( lpszCookie != NULL)
  1773. m_strCookie = lpszCookie;
  1774. else
  1775. m_strCookie = _T("");
  1776. }
  1777. }
  1779. //////////////////////////////////////////////////////////////////////////////////
  1780. // 函数名：BOOL GetIECookie(
  1781. // CString& strCookie,
  1782. // LPCTSTR lpszURL )
  1783. // 用  途：获取IE的Cookie设置
  1784. // 对全局变量的影响：无
  1785. // 参  数：
  1786. // strCookie  : 用于保存Cookie
  1787. // lpszURL    : 服务器URL
  1788. // 返回值：BOOL
  1789. ////////////////////////////////////////////////////////////////////////////////
  1790. BOOL CHttpDownload::GetIECookie(CString& strCookie,LPCTSTR lpszURL )
  1791. {
  1792. /*
  1793. Cookies Folder:
  1794. HKEY_CURRENT_USERSoftwareMicrosoftWindowsCurrentVersionExplorerShell Folders 
  1795. Cookies = ***
  1797. The simpliest way to Get Cookie value is call :
  1798. CInternetSession::GetCookie(***,***,strCookie);
  1799. But this way need the Wininet.dll which i don't want to use,How can i do?
  1800. OR: You can call :
  1801. CInternetSession::GetCookie(***,***,strCookie);
  1802. before you use HttpDownload,and then call SetCookie(***);
  1803. */
  1804. strCookie = _T("");
  1805. return TRUE;
  1806. }
  1808. ////////////////////////////////////////////////////////////////////////////////
  1809. // 函数名：BOOL IsValidFileName( LPCTSTR lpszFileName ) 
  1810. // 用  途：判断文件名是否合法
  1811. // 对全局变量的影响：无
  1812. // 参  数：
  1813. // lpszFileName : 文件名
  1814. // 返回值：BOOL
  1815. ////////////////////////////////////////////////////////////////////////////////
  1816. BOOL CHttpDownload::IsValidFileName(LPCTSTR lpszFileName)
  1817. {
  1818. /*
  1819. Short and long file names must not contain the following characters:
  1820.    ?   |   >  <   :  /  *  "
  1821. In addition, short file names must not contain the following characters:
  1822. +  ,  ;  =  [  ] 
  1823. Short file names may not include a space, although a long file name may. 
  1824. No space is allowed preceding the vertical bar (|) separator for the short 
  1825. file name/long file name syntax. If a space exists after the separator, then 
  1826. the long file name must have a space at the beginning of the file name. No 
  1827. full-path syntax is allowed.
  1828. */
  1829. CString strFileName(lpszFileName);
  1830. if( strFileName.FindOneOf("\?|></"*:") != -1 )
  1831. return FALSE;
  1833. return TRUE;
  1834. }
  1836. //seawind
  1837. BOOL CHttpDownload::IsUserStop()
  1838. {
  1839. ASSERT(m_hStopEvent != NULL);
  1841. if (WaitForSingleObject(m_hStopEvent, 0) == WAIT_OBJECT_0)
  1842. return TRUE;
  1843. else
  1844. return FALSE;
  1845. }