?? downloadhttp.cpp
字號:
// DownloadHttp.cpp: implementation of the CDownloadHttp class.
//
//////////////////////////////////////////////////////////////////////
#include "stdafx.h"
#include "NetDownMTR.h"
#include "DownloadHttp.h"
#ifdef _DEBUG
#undef THIS_FILE
static char THIS_FILE[]=__FILE__;
#define new DEBUG_NEW
#endif
void DownloadNotify ( int nIndex, UINT nNotityType, LPVOID lpNotifyData, LPVOID pDownloadMTR );
//////////////////////////////////////////////////////////////////////
// Construction/Destruction
//////////////////////////////////////////////////////////////////////
CDownloadHttp::CDownloadHttp()
{
}
CDownloadHttp::~CDownloadHttp()
{
}
BOOL CDownloadHttp::DownloadOnce()
{
// 不需要下載了
int nWillDownloadSize = Get_WillDownloadSize(); // 本次應(yīng)該下載的字節(jié)數(shù)
int nDownloadedSize = Get_DownloadedSize (); // 已下載字節(jié)數(shù)
if ( nWillDownloadSize > 0 && nDownloadedSize >= nWillDownloadSize )
return DownloadEnd(TRUE);
if ( !CDownloadPub::DownloadOnce () )
return DownloadEnd(FALSE);
char szTailData[NET_BUFFER_SIZE] = {0};
int nTailSize = sizeof(szTailData);
if ( !RequestHttpData ( TRUE, szTailData, &nTailSize ) )
return DownloadEnd(FALSE);
// 從HTTP服務(wù)器中讀取數(shù)據(jù)，并保存到文件中
return DownloadEnd ( RecvDataAndSaveToFile(m_SocketClient,szTailData, nTailSize) );
}
BOOL CDownloadHttp::RequestHttpData(BOOL bGet, char *szTailData/*=NULL*/, int *pnTailSize/*=NULL*/ )
{
int nTailSizeTemp = 0;
BOOL bRedirect = TRUE;
while ( bRedirect )
{
CString csReq = GetRequestStr ( bGet );
CString csResponse;
nTailSizeTemp = pnTailSize?(*pnTailSize):0;
if ( !SendRequest ( csReq, csResponse, szTailData, &nTailSizeTemp ) )
return FALSE;
if ( !ParseResponseString ( csResponse, bRedirect ) )
return FALSE;
}
if ( pnTailSize )
*pnTailSize = nTailSizeTemp;
return TRUE;
}
//
// 獲取遠(yuǎn)程站點(diǎn)信息，如：是否支持?jǐn)帱c(diǎn)續(xù)傳、要下載的文件大小和創(chuàng)建時(shí)間等
//
BOOL CDownloadHttp::GetRemoteSiteInfo()
{
if ( !CDownloadPub::GetRemoteSiteInfo() )
return FALSE;
if ( !RequestHttpData ( TRUE ) )
return FALSE;
m_SocketClient.Disconnect ();
return TRUE;
}
CString CDownloadHttp::GetRequestStr(BOOL bGet)
{
CString strVerb;
if( bGet )
strVerb = _T("GET ");
else
strVerb = _T("HEAD ");
CString csReq, strAuth, strRange;
csReq = strVerb + m_csObject + " HTTP/1.1\r\n";
if ( !m_csUsername.IsEmpty () )
{
strAuth = _T("");
Base64Encode ( m_csUsername + ":" + m_csPassword, strAuth );
csReq += "Authorization: Basic " + strAuth + "\r\n";
}
csReq += "Host: " + m_csServer + "\r\n";
csReq += "Accept: */*\r\n";
csReq += "Pragma: no-cache\r\n";
csReq += "Cache-Control: no-cache\r\n";
csReq += "User-Agent: "+m_csUserAgent+"\r\n";
if( !m_csReferer.IsEmpty() )
csReq += "Referer: "+m_csReferer+"\r\n";
csReq += "Connection: close\r\n";
// 指定要下載的文件范圍
CString csEndPos;
int nWillDownloadStartPos = Get_WillDownloadStartPos (); // 開始位置
int nWillDownloadSize = Get_WillDownloadSize(); // 本次應(yīng)該下載的字節(jié)數(shù)
int nDownloadedSize = Get_DownloadedSize (); // 已下載字節(jié)數(shù)
if ( nWillDownloadSize >= 0 )
csEndPos.Format ( "%d", nWillDownloadStartPos+nWillDownloadSize-1 );
ASSERT ( nWillDownloadSize < 0 || nDownloadedSize < nWillDownloadSize );
strRange.Format ( _T("Range: bytes=%d-%s\r\n"), nWillDownloadStartPos+nDownloadedSize, csEndPos );
csReq += strRange;
csReq += "\r\n";
return csReq;
}
//
// 向服務(wù)器提交請求，并得到返回字符串
//
BOOL CDownloadHttp::SendRequest(LPCTSTR lpszReq, CString &csResponse, char *szTailData/*=NULL*/, int *pnTailSize/*=NULL*/ )
{
m_SocketClient.Disconnect ();
if ( !Connect () ) return FALSE;
if ( !m_SocketClient.SendString ( lpszReq ) )
{
return FALSE;
}
for ( int i=0; ; i++ )
{
char szRecvBuf[NET_BUFFER_SIZE] = {0};
int nReadSize = m_SocketClient.Receive ( szRecvBuf, sizeof(szRecvBuf) );
if ( nReadSize <= 0 )
{
Log ( L_WARNING, "Receive response data failed" );
return FALSE;
}
csResponse += szRecvBuf;
char *p = strstr ( szRecvBuf, "\r\n\r\n" );
if ( p )
{
if ( szTailData && pnTailSize && *pnTailSize > 0 )
{
p += 4;
int nOtioseSize = nReadSize - int( p - szRecvBuf );
*pnTailSize = MIN ( nOtioseSize, *pnTailSize );
memcpy ( szTailData, p, *pnTailSize );
}
#ifdef _DEBUG
int nPos = csResponse.Find ( "\r\n\r\n", 0 );
CString csDump;
if ( nPos >= 0 ) csDump = csResponse.Left ( nPos );
else csDump = csResponse;
// TRACE ( "\n收到回應(yīng) <<<---------------------\n\n%s\n", csDump );
#endif
break;
}
}
return TRUE;
}
DWORD CDownloadHttp::GetResponseCode(CString csLineText)
{
csLineText.MakeLower ();
ASSERT ( csLineText.Find ( "http/", 0 ) >= 0 );
int nPos = csLineText.Find ( " ", 0 );
if ( nPos < 0 ) return 0;
CString csCode = csLineText.Mid ( nPos + 1 );
csCode.TrimLeft(); csCode.TrimRight();
nPos = csCode.Find ( " ", 0 );
if ( nPos < 0 ) nPos = csCode.GetLength() - 1;
csCode = csCode.Left ( nPos );
return (DWORD)atoi(csCode);
}
BOOL CDownloadHttp::ParseResponseString ( CString csResponseString, OUT BOOL &bRedirect )
{
bRedirect = FALSE;
// 獲取返回代碼
CString csOneLine = GetOneLine ( csResponseString );
DWORD dwResponseCode = GetResponseCode ( csOneLine );
if ( dwResponseCode < 1 )
{
Log ( L_WARNING, "Received error response code : %s", csOneLine );
return FALSE;
}
int nPos = 0;
// 請求文件被重定向
if( dwResponseCode >= 300 && dwResponseCode < 400 )
{
bRedirect = TRUE;
// 得到請求文件新的URL
CString csRedirectFileName = FindAfterFlagString ( "location:", csResponseString );
// 設(shè)置 Referer
m_csReferer = m_csDownloadUrl;
// 重定向到其他的服務(wù)器
nPos = csRedirectFileName.Find("://");
if ( nPos >= 0 )
{
m_csDownloadUrl = csRedirectFileName;
// 檢驗(yàn)要下載的URL是否有效
if ( !ParseURL ( m_csDownloadUrl, m_csServer, m_csObject, m_nPort, m_csProtocolType ) )
{
Log ( L_WARNING, "Redirect media path [%s] invalid", m_csDownloadUrl );
return FALSE;
}
return TRUE;
}
// 重定向到本服務(wù)器的其他地方
csRedirectFileName.Replace ( "\\", "/" );
// 重定向于根目錄
if( csRedirectFileName[0] == '/' )
{
m_csObject = csRedirectFileName;
DownloadNotify ( -1, NOTIFY_TYPE_GOT_REMOTE_FILENAME, (LPVOID)(LPCTSTR)(GetDownloadObjectFileName()), m_pDownloadMTR );
return TRUE;
}
// 定向于相對當(dāng)前目錄
int nParentDirCount = 0;
nPos = csRedirectFileName.Find ( "../" );
while ( nPos >= 0 )
{
csRedirectFileName = csRedirectFileName.Mid(nPos+3);
nParentDirCount++;
nPos = csRedirectFileName.Find("../");
}
for (int i=0; i<=nParentDirCount; i++)
{
nPos = m_csDownloadUrl.ReverseFind('/');
if (nPos != -1)
m_csDownloadUrl = m_csDownloadUrl.Left(nPos);
}
if ( csRedirectFileName.Find ( "./", 0 ) == 0 )
csRedirectFileName.Delete ( 0, 2 );
m_csDownloadUrl = m_csDownloadUrl+"/"+csRedirectFileName;
return ParseURL ( m_csDownloadUrl, m_csServer, m_csObject, m_nPort, m_csProtocolType );
}
// 請求被成功接收、理解和接受
else if( dwResponseCode >= 200 && dwResponseCode < 300 )
{
if ( m_nIndex == -1 ) // 主線程才需要獲取文件大小的信息
{
// 獲取 Content-Length
CString csDownFileLen = FindAfterFlagString ( "content-length:", csResponseString );
m_nFileTotalSize = (int) _ttoi( (LPCTSTR)csDownFileLen );
DownloadNotify ( -1, NOTIFY_TYPE_GOT_REMOTE_FILESIZE, (LPVOID)m_nFileTotalSize, m_pDownloadMTR );
int nWillDownloadStartPos = Get_WillDownloadStartPos (); // 開始位置
int nWillDownloadSize = Get_WillDownloadSize(); // 本次應(yīng)該下載的字節(jié)數(shù)
int nDownloadedSize = Get_DownloadedSize (); // 已下載字節(jié)數(shù)
if ( m_nFileTotalSize > 0 && nWillDownloadSize-nDownloadedSize > m_nFileTotalSize )
Set_WillDownloadSize ( m_nFileTotalSize-nDownloadedSize );
}
// 獲取服務(wù)器文件的最后修改時(shí)間
CString csModifiedTime = FindAfterFlagString ( "last-modified:", csResponseString );
if ( !csModifiedTime.IsEmpty() )
{
m_TimeLastModified = ConvertHttpTimeString(csModifiedTime);
}
if ( dwResponseCode == 206 ) // 支持?jǐn)帱c(diǎn)續(xù)傳
{
m_bSupportResume = TRUE;
}
else // 不支持?jǐn)帱c(diǎn)續(xù)傳
{
m_bSupportResume = FALSE;
}
return TRUE;
}
// Log ( L_WARNING, "Index.%d receive invalid code : %d", m_nIndex, dwResponseCode );
return FALSE;
}
CString CDownloadHttp::FindAfterFlagString(LPCTSTR lpszFoundStr, CString csOrg)
{
ASSERT ( lpszFoundStr && strlen(lpszFoundStr) > 0 );
CString csReturing, csFoundStr = GET_SAFE_STRING(lpszFoundStr);
csFoundStr.MakeLower ();
CString csOrgLower = csOrg;
csOrgLower.MakeLower ();
int nPos = csOrgLower.Find ( csFoundStr );
if ( nPos < 0 ) return "";
csReturing = csOrg.Mid ( nPos + csFoundStr.GetLength() );
nPos = csReturing.Find("\r\n");
if ( nPos < 0 ) return "";
csReturing = csReturing.Left(nPos);
csReturing.TrimLeft();
csReturing.TrimRight();
return csReturing;
}
//
// 將 HTTP 服務(wù)器表示的時(shí)間轉(zhuǎn)換為 CTime 格式，如：Wed, 16 May 2007 14:29:53 GMT
//
CTime CDownloadHttp::ConvertHttpTimeString(CString csTimeGMT)
{
CString csYear, csMonth, csDay, csTime;
CTime tReturning = -1;
int nPos = csTimeGMT.Find ( ",", 0 );
if ( nPos < 0 || nPos >= csTimeGMT.GetLength()-1 )
return tReturning;
csTimeGMT = csTimeGMT.Mid ( nPos + 1 );
csTimeGMT.TrimLeft(); csTimeGMT.TrimRight ();
// 日
nPos = csTimeGMT.Find ( " ", 0 );
if ( nPos < 0 || nPos >= csTimeGMT.GetLength()-1 )
return tReturning;
csDay = csTimeGMT.Left ( nPos );
csTimeGMT = csTimeGMT.Mid ( nPos + 1 );
csTimeGMT.TrimLeft(); csTimeGMT.TrimRight ();
// 月
nPos = csTimeGMT.Find ( " ", 0 );
if ( nPos < 0 || nPos >= csTimeGMT.GetLength()-1 )
return tReturning;
csMonth = csTimeGMT.Left ( nPos );
int nMonth = GetMouthByShortStr ( csMonth );
ASSERT ( nMonth >= 1 && nMonth <= 12 );
csMonth.Format ( "%02d", nMonth );
csTimeGMT = csTimeGMT.Mid ( nPos + 1 );
csTimeGMT.TrimLeft(); csTimeGMT.TrimRight ();
// 年
nPos = csTimeGMT.Find ( " ", 0 );
if ( nPos < 0 || nPos >= csTimeGMT.GetLength()-1 )
return tReturning;
csYear = csTimeGMT.Left ( nPos );
csTimeGMT = csTimeGMT.Mid ( nPos + 1 );
csTimeGMT.TrimLeft(); csTimeGMT.TrimRight ();
// 時(shí)間
nPos = csTimeGMT.Find ( " ", 0 );
if ( nPos < 0 || nPos >= csTimeGMT.GetLength()-1 )
return tReturning;
csTime = csTimeGMT.Left ( nPos );
csTimeGMT = csTimeGMT.Mid ( nPos + 1 );
CString csFileTimeInfo;
csFileTimeInfo.Format ( "%s-%s-%s %s", csYear, csMonth, csDay, csTime );
ConvertStrToCTime ( csFileTimeInfo.GetBuffer(0), tReturning );
return tReturning;
}
?? 快捷鍵說明
復(fù)制代碼
Ctrl + C
搜索代碼
Ctrl + F
全屏模式
F11
切換主題
Ctrl + Shift + D
顯示快捷鍵
?
增大字號
Ctrl + =
減小字號
Ctrl + -