vc http 使用CHttpConnection下载文件

来源:互联网 发布:菠菜源码论坛 编辑:程序博客网 时间:2024/05/16 10:30

此函数可支持下载几M以上文件

BOOL Download(CString strURL, CString strName, CString strSavePath)

{
    CInternetSession session;
    CHttpConnection* pHttpConnection = NULL;
    CHttpFile* pHttpFile = NULL;
    CString strServer, strObject;
    INTERNET_PORT wPort;
    BOOL bReturn = FALSE;
    DWORD dwType;
    const int nTimeOut = 2000;
    session.SetOption(INTERNET_OPTION_CONNECT_TIMEOUT, nTimeOut); //重试之间的等待延时
    session.SetOption(INTERNET_OPTION_CONNECT_RETRIES, 1); //重试次数
    char* pszBuffer = NULL;
    try
   {
       AfxParseURL(strURL +_T("/")  +  strName, dwType, strServer, strObject, wPort);
       pHttpConnection = session.GetHttpConnection(strServer, wPort);
       pHttpFile = pHttpConnection->OpenRequest(CHttpConnection::HTTP_VERB_GET, strObject);
       if(pHttpFile->SendRequest() == FALSE) return FALSE;

      DWORD dwStateCode;
      pHttpFile->QueryInfoStatusCode(dwStateCode);
      if(dwStateCode == HTTP_STATUS_OK)
      {
CString strFilePath;
strFilePath = strSavePath;
//创建文件夹
//createDirectory(strFilePath);

HANDLE hFile = CreateFile(strFilePath + _T("\\") + strName, GENERIC_WRITE,
FILE_SHARE_WRITE, NULL, CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
if(hFile == INVALID_HANDLE_VALUE)
{
pHttpFile->Close();
pHttpConnection->Close();
session.Close();
return FALSE;
}
char szInfoBuffer[1000]; //返回消息
DWORD dwFileSize = 0; //文件长度
DWORD dwInfoBufferSize = sizeof(szInfoBuffer);
BOOL bResult = FALSE;
bResult = pHttpFile->QueryInfo(HTTP_QUERY_CONTENT_LENGTH,
(void*)szInfoBuffer, &dwInfoBufferSize,NULL);
dwFileSize = atoi(szInfoBuffer);
const int BUFFER_LENGTH = 1024 * 10;
pszBuffer = new char[BUFFER_LENGTH]; //读取文件的缓冲
DWORD dwWrite, dwTotalWrite;
dwWrite = dwTotalWrite = 0;
UINT nRead = pHttpFile->Read(pszBuffer, BUFFER_LENGTH); //读取服务器上数据
while(nRead > 0)
{
WriteFile(hFile, pszBuffer, nRead, &dwWrite, NULL); //写到本地文件
dwTotalWrite += dwWrite;
nRead = pHttpFile->Read(pszBuffer, BUFFER_LENGTH);
}
delete[] pszBuffer;
pszBuffer = NULL;
CloseHandle(hFile);
bReturn = TRUE;
}
}
catch(...)
{
}
if(pszBuffer != NULL)
{
delete[]pszBuffer;
}
if(pHttpFile != NULL)
{
pHttpFile->Close();
delete pHttpFile;
}
if(pHttpConnection != NULL)
{
pHttpConnection->Close();
delete pHttpConnection;
}
session.Close();
return bReturn;

}


经测试函数没有问题,但是http下载时存在一个问题就是网站上不能使用中文的文件名,或者目录名称,大概给网站的编码有问题,暂时没有研究。


0 0
原创粉丝点击