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

源码开发语言/平台
当前位置：首页 > 源码/资料 > Windows编程 > 进程与线程 > 查看源码
DownloadHttp.cpp
资源名称：NetDownMTR.rar　[点击查看]
上传用户：oadesign
上传日期：2013-12-25
资源大小：265k
文件大小：11k
源码类别：

进程与线程

开发平台：

Visual C++

DownloadHttp.cpp：源码内容
  1. // DownloadHttp.cpp: implementation of the CDownloadHttp class.
  2. //
  3. //////////////////////////////////////////////////////////////////////
  5. #include "stdafx.h"
  6. #include "NetDownMTR.h"
  7. #include "DownloadHttp.h"
  9. #ifdef _DEBUG
  10. #undef THIS_FILE
  11. static char THIS_FILE[]=__FILE__;
  12. #define new DEBUG_NEW
  13. #endif
  15. void DownloadNotify ( int nIndex, UINT nNotityType, LPVOID lpNotifyData, LPVOID pDownloadMTR );
  17. //////////////////////////////////////////////////////////////////////
  18. // Construction/Destruction
  19. //////////////////////////////////////////////////////////////////////
  21. CDownloadHttp::CDownloadHttp()
  22. {
  24. }
  26. CDownloadHttp::~CDownloadHttp()
  27. {
  29. }
  31. BOOL CDownloadHttp::DownloadOnce()
  32. {
  33. // 不需要下载了
  34. int nWillDownloadSize = Get_WillDownloadSize(); // 本次应该下载的字节数
  35. int nDownloadedSize = Get_DownloadedSize (); // 已下载字节数
  36. if ( nWillDownloadSize > 0 && nDownloadedSize >= nWillDownloadSize )
  37. return DownloadEnd(TRUE);
  39. if ( !CDownloadPub::DownloadOnce () )
  40. return DownloadEnd(FALSE);
  42. char szTailData[NET_BUFFER_SIZE] = {0};
  43. int nTailSize = sizeof(szTailData);
  44. if ( !RequestHttpData ( TRUE, szTailData, &nTailSize ) )
  45. return DownloadEnd(FALSE);
  46. // 从HTTP服务器中读取数据，并保存到文件中
  47. return DownloadEnd ( RecvDataAndSaveToFile(m_SocketClient,szTailData, nTailSize) );
  49. }
  51. BOOL CDownloadHttp::RequestHttpData(BOOL bGet, char *szTailData/*=NULL*/, int *pnTailSize/*=NULL*/ )
  52. {
  53. int nTailSizeTemp = 0;
  54. BOOL bRedirect = TRUE;
  55. while ( bRedirect )
  56. {
  57. CString csReq = GetRequestStr ( bGet );
  58. CString csResponse;
  59. nTailSizeTemp = pnTailSize?(*pnTailSize):0;
  60. if ( !SendRequest ( csReq, csResponse, szTailData, &nTailSizeTemp ) )
  61. return FALSE;
  62. if ( !ParseResponseString ( csResponse, bRedirect ) )
  63. return FALSE;
  64. }
  65. if ( pnTailSize )
  66. *pnTailSize = nTailSizeTemp;
  68. return TRUE;
  69. }
  71. //
  72. // 获取远程站点信息，如：是否支持断点续传、要下载的文件大小和创建时间等
  73. //
  74. BOOL CDownloadHttp::GetRemoteSiteInfo()
  75. {
  76. if ( !CDownloadPub::GetRemoteSiteInfo() )
  77. return FALSE;
  79. if ( !RequestHttpData ( TRUE ) )
  80. return FALSE;
  82. m_SocketClient.Disconnect ();
  83. return TRUE;
  84. }
  86. CString CDownloadHttp::GetRequestStr(BOOL bGet)
  87. {
  88. CString strVerb;
  89. if( bGet )
  90. strVerb = _T("GET ");
  91. else
  92. strVerb = _T("HEAD ");
  94. CString csReq, strAuth, strRange;
  95. csReq  = strVerb  + m_csObject + " HTTP/1.1rn";
  97. if ( !m_csUsername.IsEmpty () )
  98. {
  99. strAuth = _T("");
  100. Base64Encode ( m_csUsername + ":" + m_csPassword, strAuth );
  101. csReq += "Authorization: Basic " + strAuth + "rn";
  102. }
  104. csReq += "Host: " + m_csServer + "rn";
  105. csReq += "Accept: */*rn";
  106. csReq += "Pragma: no-cachern"; 
  107. csReq += "Cache-Control: no-cachern";
  108. csReq += "User-Agent: "+m_csUserAgent+"rn";
  109. if( !m_csReferer.IsEmpty() )
  110. csReq += "Referer: "+m_csReferer+"rn";
  111. csReq += "Connection: closern";
  113. // 指定要下载的文件范围
  114. CString csEndPos;
  115. int nWillDownloadStartPos = Get_WillDownloadStartPos (); // 开始位置
  116. int nWillDownloadSize = Get_WillDownloadSize(); // 本次应该下载的字节数
  117. int nDownloadedSize = Get_DownloadedSize (); // 已下载字节数
  118. if ( nWillDownloadSize >= 0 )
  119. csEndPos.Format ( "%d", nWillDownloadStartPos+nWillDownloadSize-1 );
  120. ASSERT ( nWillDownloadSize < 0 || nDownloadedSize < nWillDownloadSize );
  121. strRange.Format ( _T("Range: bytes=%d-%srn"), nWillDownloadStartPos+nDownloadedSize, csEndPos );
  123. csReq += strRange;
  124. csReq += "rn";
  126. return csReq;
  127. }
  129. //
  130. // 向服务器提交请求，并得到返回字符串
  131. //
  132. BOOL CDownloadHttp::SendRequest(LPCTSTR lpszReq, CString &csResponse, char *szTailData/*=NULL*/, int *pnTailSize/*=NULL*/ )
  133. {
  134. m_SocketClient.Disconnect ();
  135. if ( !Connect () ) return FALSE;
  136. if ( !m_SocketClient.SendString ( lpszReq ) )
  137. {
  138. return FALSE;
  139. }
  141. for ( int i=0; ; i++ )
  142. {
  143. char szRecvBuf[NET_BUFFER_SIZE] = {0};
  144. int nReadSize = m_SocketClient.Receive ( szRecvBuf, sizeof(szRecvBuf) );
  145. if ( nReadSize <= 0 )
  146. {
  147. Log ( L_WARNING, "Receive response data failed" );
  148. return FALSE;
  149. }
  150. csResponse += szRecvBuf;
  151. char *p = strstr ( szRecvBuf, "rnrn" );
  152. if ( p )
  153. {
  154. if ( szTailData && pnTailSize && *pnTailSize > 0 )
  155. {
  156. p += 4;
  157. int nOtioseSize = nReadSize - int( p - szRecvBuf );
  158. *pnTailSize = MIN ( nOtioseSize, *pnTailSize );
  159. memcpy ( szTailData, p, *pnTailSize );
  160. }
  161. #ifdef _DEBUG
  162. int nPos = csResponse.Find ( "rnrn", 0 );
  163. CString csDump;
  164. if ( nPos >= 0 ) csDump = csResponse.Left ( nPos );
  165. else csDump = csResponse;
  166. // TRACE ( "n收到回应 <<<---------------------nn%sn", csDump );
  167. #endif
  168. break;
  169. }
  170. }
  172. return TRUE;
  173. }
  175. DWORD CDownloadHttp::GetResponseCode(CString csLineText)
  176. {
  177. csLineText.MakeLower ();
  178. ASSERT ( csLineText.Find ( "http/", 0 ) >= 0 );
  179. int nPos = csLineText.Find ( " ", 0 );
  180. if ( nPos < 0 ) return 0;
  181. CString csCode = csLineText.Mid ( nPos + 1 );
  182. csCode.TrimLeft(); csCode.TrimRight();
  183. nPos = csCode.Find ( " ", 0 );
  184. if ( nPos < 0 ) nPos = csCode.GetLength() - 1;
  185. csCode = csCode.Left ( nPos );
  187. return (DWORD)atoi(csCode);
  188. }
  190. BOOL CDownloadHttp::ParseResponseString ( CString csResponseString, OUT BOOL &bRedirect )
  191. {
  192. bRedirect = FALSE;
  193. // 获取返回代码
  194. CString csOneLine = GetOneLine ( csResponseString );
  195. DWORD dwResponseCode = GetResponseCode ( csOneLine );
  196. if ( dwResponseCode < 1 )
  197. {
  198. Log ( L_WARNING, "Received error response code : %s", csOneLine );
  199. return FALSE;
  200. }
  202. int nPos = 0;
  203. // 请求文件被重定向
  204. if( dwResponseCode >= 300 && dwResponseCode < 400 )
  205. {
  206. bRedirect = TRUE;
  207. // 得到请求文件新的URL
  208. CString csRedirectFileName = FindAfterFlagString ( "location:", csResponseString );
  210. // 设置 Referer
  211. m_csReferer = m_csDownloadUrl;
  213. // 重定向到其他的服务器
  214. nPos = csRedirectFileName.Find("://");
  215. if ( nPos >= 0 )
  216. {
  217. m_csDownloadUrl = csRedirectFileName;
  218. // 检验要下载的URL是否有效
  219. if ( !ParseURL ( m_csDownloadUrl, m_csServer, m_csObject, m_nPort, m_csProtocolType ) )
  220. {
  221. Log ( L_WARNING, "Redirect media path [%s] invalid", m_csDownloadUrl );
  222. return FALSE;
  223. }
  224. return TRUE;
  225. }
  227. // 重定向到本服务器的其他地方
  228. csRedirectFileName.Replace ( "\", "/" );
  229. // 重定向于根目录
  230. if( csRedirectFileName[0] == '/' )
  231. {
  232. m_csObject = csRedirectFileName;
  233. DownloadNotify ( -1, NOTIFY_TYPE_GOT_REMOTE_FILENAME, (LPVOID)(LPCTSTR)(GetDownloadObjectFileName()), m_pDownloadMTR );
  234. return TRUE;
  235. }
  237. // 定向于相对当前目录
  238. int nParentDirCount = 0;
  239. nPos = csRedirectFileName.Find ( "../" );
  240. while ( nPos >= 0 )
  241. {
  242. csRedirectFileName = csRedirectFileName.Mid(nPos+3);
  243. nParentDirCount++;
  244. nPos = csRedirectFileName.Find("../");
  245. }
  246. for (int i=0; i<=nParentDirCount; i++)
  247. {
  248. nPos = m_csDownloadUrl.ReverseFind('/');
  249. if (nPos != -1)
  250. m_csDownloadUrl = m_csDownloadUrl.Left(nPos);
  251. }
  252. if ( csRedirectFileName.Find ( "./", 0 ) == 0 )
  253. csRedirectFileName.Delete ( 0, 2 );
  254. m_csDownloadUrl = m_csDownloadUrl+"/"+csRedirectFileName;
  256. return ParseURL ( m_csDownloadUrl, m_csServer, m_csObject, m_nPort, m_csProtocolType );
  257. }
  258. // 请求被成功接收、理解和接受
  259. else if( dwResponseCode >= 200 && dwResponseCode < 300 )
  260. {
  261. if ( m_nIndex == -1 ) // 主线程才需要获取文件大小的信息
  262. {
  263. // 获取 Content-Length
  264. CString csDownFileLen = FindAfterFlagString ( "content-length:", csResponseString );
  265. m_nFileTotalSize = (int) _ttoi( (LPCTSTR)csDownFileLen );
  266. DownloadNotify ( -1, NOTIFY_TYPE_GOT_REMOTE_FILESIZE, (LPVOID)m_nFileTotalSize, m_pDownloadMTR );
  267. int nWillDownloadStartPos = Get_WillDownloadStartPos (); // 开始位置
  268. int nWillDownloadSize = Get_WillDownloadSize(); // 本次应该下载的字节数
  269. int nDownloadedSize = Get_DownloadedSize (); // 已下载字节数
  270. if ( m_nFileTotalSize > 0 && nWillDownloadSize-nDownloadedSize > m_nFileTotalSize )
  271. Set_WillDownloadSize ( m_nFileTotalSize-nDownloadedSize );
  272. }
  274. // 获取服务器文件的最后修改时间
  275. CString csModifiedTime = FindAfterFlagString ( "last-modified:", csResponseString );
  276. if ( !csModifiedTime.IsEmpty() )
  277. {
  278. m_TimeLastModified = ConvertHttpTimeString(csModifiedTime);
  279. }
  281. if ( dwResponseCode == 206 ) // 支持断点续传
  282. {
  283. m_bSupportResume = TRUE;
  284. }
  285. else // 不支持断点续传
  286. {
  287. m_bSupportResume = FALSE;
  288. }
  289. return TRUE;
  290. }
  292. // Log ( L_WARNING, "Index.%d receive invalid code : %d", m_nIndex, dwResponseCode );
  293. return FALSE;
  294. }
  296. CString CDownloadHttp::FindAfterFlagString(LPCTSTR lpszFoundStr, CString csOrg)
  297. {
  298. ASSERT ( lpszFoundStr && strlen(lpszFoundStr) > 0 );
  299. CString csReturing, csFoundStr = GET_SAFE_STRING(lpszFoundStr);
  300. csFoundStr.MakeLower ();
  301. CString csOrgLower = csOrg;
  302. csOrgLower.MakeLower ();
  303. int nPos = csOrgLower.Find ( csFoundStr );
  304. if ( nPos < 0 ) return "";
  305. csReturing = csOrg.Mid ( nPos + csFoundStr.GetLength() );
  306. nPos = csReturing.Find("rn");
  307. if ( nPos < 0 ) return "";
  308. csReturing = csReturing.Left(nPos);
  309. csReturing.TrimLeft();
  310. csReturing.TrimRight();
  312. return csReturing;
  313. }
  315. //
  316. // 将 HTTP 服务器表示的时间转换为 CTime 格式，如：Wed, 16 May 2007 14:29:53 GMT
  317. //
  318. CTime CDownloadHttp::ConvertHttpTimeString(CString csTimeGMT)
  319. {
  320. CString csYear, csMonth, csDay, csTime;
  321. CTime tReturning = -1;
  322. int nPos = csTimeGMT.Find ( ",", 0 );
  323. if ( nPos < 0 || nPos >= csTimeGMT.GetLength()-1 )
  324. return tReturning;
  325. csTimeGMT = csTimeGMT.Mid ( nPos + 1 );
  326. csTimeGMT.TrimLeft(); csTimeGMT.TrimRight ();
  328. // 日
  329. nPos = csTimeGMT.Find ( " ", 0 );
  330. if ( nPos < 0 || nPos >= csTimeGMT.GetLength()-1 )
  331. return tReturning;
  332. csDay = csTimeGMT.Left ( nPos );
  333. csTimeGMT = csTimeGMT.Mid ( nPos + 1 );
  334. csTimeGMT.TrimLeft(); csTimeGMT.TrimRight ();
  336. // 月
  337. nPos = csTimeGMT.Find ( " ", 0 );
  338. if ( nPos < 0 || nPos >= csTimeGMT.GetLength()-1 )
  339. return tReturning;
  340. csMonth = csTimeGMT.Left ( nPos );
  341. int nMonth = GetMouthByShortStr ( csMonth );
  342. ASSERT ( nMonth >= 1 && nMonth <= 12 );
  343. csMonth.Format ( "%02d", nMonth );
  344. csTimeGMT = csTimeGMT.Mid ( nPos + 1 );
  345. csTimeGMT.TrimLeft(); csTimeGMT.TrimRight ();
  347. // 年
  348. nPos = csTimeGMT.Find ( " ", 0 );
  349. if ( nPos < 0 || nPos >= csTimeGMT.GetLength()-1 )
  350. return tReturning;
  351. csYear = csTimeGMT.Left ( nPos );
  352. csTimeGMT = csTimeGMT.Mid ( nPos + 1 );
  353. csTimeGMT.TrimLeft(); csTimeGMT.TrimRight ();
  355. // 时间
  356. nPos = csTimeGMT.Find ( " ", 0 );
  357. if ( nPos < 0 || nPos >= csTimeGMT.GetLength()-1 )
  358. return tReturning;
  359. csTime = csTimeGMT.Left ( nPos );
  360. csTimeGMT = csTimeGMT.Mid ( nPos + 1 );
  362. CString csFileTimeInfo;
  363. csFileTimeInfo.Format ( "%s-%s-%s %s", csYear, csMonth, csDay, csTime );
  364. ConvertStrToCTime ( csFileTimeInfo.GetBuffer(0), tReturning );
  365. return tReturning;
  366. }