博客

// HttpDownLoadFile.h </p> <p>#ifndef _HTTP_DOWN_LOAD_FILE_H<br /> #define _HTTP_DOWN_LOAD_FILE_H</p> <p>#include <windows.h><br /> #include <wininet.h></p> <p>#pragma comment(lib,”wininet.lib”)<br /> #pragma comment(lib,”Winmm.lib”)</p> <p>class CHttpDownLoadFile<br /> {<br /> public:<br /> enum DownLoadMode //下载模式<br /> {<br /> Mode_BlockNoMsg = 1, //阻塞，不传递消息模式<br /> Mode_BlockAndMsg, //阻塞，传递消息模式<br /> Mode_UnBlockNoMsg, //非阻塞，不传递消息模式<br /> Mode_UnBlockAndMsg, //非阻塞，传递消息模式<br /> };</p> <p> enum DownLoadMsg //下载时发送的消息<br /> {<br /> DL_OK = 0x0001,<br /> DL_ERROR,<br /> DL_START,<br /> DL_FILESIZE,<br /> DL_READDATA,<br /> DL_TERMINATE,<br /> };</p> <p> enum DownLoadWndMsg //定义的windows消息<br /> {<br /> DL_MSG = WM_USER + 200,<br /> };</p> <p>private:<br /> HWND m_hWnd; // 下载时，消息的接收窗体<br /> int m_DownLoadMode; // 下载模式<br /> BOOL m_OverWrite; // 是否覆盖<br /> int m_ErrorCode; // 错误码<br /> DWORD m_Speed; // 下载速度<br /> DWORD m_ReadCount; // 已经下载的大小<br /> DWORD m_FileSize; // 文件大小<br /> DWORD m_MsgCode; // 下载时，发送的消息代码，初始化时等于 DL_MSG</p> <p> TCHAR m_OutputFileName[_MAX_PATH]; // 保存下载时的文件名<br /> TCHAR m_URL[_MAX_PATH]; // 资源的URL</p> <p> BOOL m_bDownLoadWebOK; // 是否下载完成<br /> DWORD m_WaitTime; // 下载的时间<br /> private:</p> <p> HINTERNET FhSession ; //会话句柄<br /> HINTERNET FhConnect ; //http连接句柄<br /> HINTERNET FhRequest ; //http请求句柄<br /> HANDLE hFile;</p> <p>public:<br /> CHttpDownLoadFile();<br /> CHttpDownLoadFile(HWND hWnd);<br /> ~CHttpDownLoadFile();<br /> public:<br /> LPSTR GetURL() { return m_URL; }<br /> void SetURL( LPSTR Value ) { strcpy(m_URL,Value); }</p> <p> LPSTR GetOutputFileName() { return m_OutputFileName; }<br /> void SetOutputFileName( LPSTR fn ) { strcpy(m_OutputFileName, fn); }</p> <p> BOOL SetOverWrite() { return m_OverWrite; }<br /> void SetOverWrite(BOOL Value) { m_OverWrite = Value; }</p> <p> void SetWndHandle(HWND hWnd) { m_hWnd = hWnd; }<br /> HWND GetWndHandle() { return m_hWnd; }</p> <p> void SetDownLoadMode(int mode) { m_DownLoadMode= mode; }<br /> int GetDownLoadMode() { return m_DownLoadMode; }</p> <p> void SetMsgCode(DWORD code) { m_MsgCode = code; }</p> <p> DWORD GetSpeed() { return m_Speed; }<br /> DWORD GetFileSize() { return m_FileSize; }<br /> DWORD GetBeenDownLoadedSize() { return m_ReadCount; }</p> <p>public:<br /> int StartDownLoad(); // 开始下载,并把下载的内容保存到文件里</p> <p>public:<br /> int _RealDownLoad(); // 实际下载<br /> void ParseURL(char *Url,char *pHostName,char *pFileName,DWORD* nPort);</p> <p>public:<br /> static int StartDownLoad(LPSTR url,<br /> LPSTR outputfile,<br /> int mode = Mode_BlockNoMsg,<br /> HWND hWnd = NULL,<br /> BOOL isOverWrite = FALSE);<br /> static ULONG WINAPI DownLoadProc(LPVOID pDownLoadFile);<br /> private:<br /> void DisposeError();<br /> };</p> <p>#endif</p> <p>//HttpDownLoadFile.cpp<br /> #include “HttpDownLoadFile.h”</p> <p>CHttpDownLoadFile::CHttpDownLoadFile()<br /> {<br /> m_hWnd = NULL;<br /> m_DownLoadMode = Mode_BlockNoMsg;<br /> m_OverWrite = TRUE;<br /> m_bDownLoadWebOK = FALSE;</p> <p> m_ErrorCode = 0;</p> <p> m_Speed = 0;<br /> m_ReadCount = 0;<br /> m_FileSize = 0;</p> <p> m_MsgCode = DL_MSG;<br /> m_WaitTime = 10000;</p> <p> hFile = NULL;</p> <p> memset(m_URL,0x00,sizeof(m_URL));<br /> memset(m_URL,0x00,sizeof(m_OutputFileName));<br /> }</p> <p>CHttpDownLoadFile::CHttpDownLoadFile(HWND hWnd)<br /> {<br /> m_hWnd = hWnd;<br /> m_DownLoadMode = Mode_BlockNoMsg;<br /> m_OverWrite = TRUE;<br /> m_bDownLoadWebOK = FALSE;</p> <p> m_ErrorCode = 0;</p> <p> m_Speed = 0;<br /> m_ReadCount = 0;<br /> m_FileSize = 0;</p> <p> m_MsgCode = DL_MSG;</p> <p> hFile = NULL;</p> <p> memset(m_URL,0x00,sizeof(m_URL));<br /> memset(m_URL,0x00,sizeof(m_OutputFileName));<br /> }</p> <p>int CHttpDownLoadFile::StartDownLoad()<br /> {<br /> if( m_DownLoadMode == Mode_BlockNoMsg || m_DownLoadMode == Mode_BlockAndMsg )<br /> {<br /> return this->_RealDownLoad();<br /> }else<br /> {<br /> ULONG threadID;<br /> HANDLE hThread = CreateThread(NULL,0,DownLoadProc,this,0,&threadID);<br /> CloseHandle(hThread);<br /> return TRUE;<br /> }<br /> }</p> <p>ULONG WINAPI CHttpDownLoadFile::DownLoadProc(LPVOID pDownLoadFile)<br /> {<br /> CHttpDownLoadFile* pDownLoad = (CHttpDownLoadFile*)(pDownLoadFile);<br /> return pDownLoad->_RealDownLoad();<br /> }</p> <p>int CHttpDownLoadFile::_RealDownLoad()<br /> {<br /> // 发送开始传输消息<br /> if( m_DownLoadMode == Mode_UnBlockAndMsg || m_DownLoadMode == Mode_BlockAndMsg)<br /> {<br /> ::SendMessage(m_hWnd,m_MsgCode,DL_START,0);<br /> }</p> <p> // 打开链接，开始下载<br /> FhSession = ::InternetOpen(“HttpDownLoadFile”, INTERNET_OPEN_TYPE_PRECONFIG, NULL, NULL, 0);// 同步方式<br /> if(FhSession == NULL || FhSession == INVALID_HANDLE_VALUE)<br /> {<br /> //出错处理<br /> this->DisposeError();<br /> return 0;<br /> }</p> <p> TCHAR server[_MAX_PATH];<br /> TCHAR fileName[_MAX_PATH];<br /> DWORD nPort = INTERNET_DEFAULT_HTTP_PORT;<br /> memset(server,0x00,_MAX_PATH);<br /> memset(fileName,0x00,_MAX_PATH);<br /> const char *FAcceptTypes[] = {“*/*”,NULL};</p> <p> this->ParseURL(m_URL,server,fileName,&nPort);<br /> // 得到下载后的文件名<br /> if( strcmp(m_OutputFileName,””) == 0)<br /> {<br /> strcpy(m_OutputFileName,fileName);<br /> }</p> <p> FhConnect = ::InternetConnect(FhSession, server, nPort,<br /> NULL, NULL, INTERNET_SERVICE_HTTP, 0, 0);<br /> if(FhConnect==NULL || FhConnect == INVALID_HANDLE_VALUE)<br /> {<br /> //出错处理<br /> this->DisposeError();<br /> return 0;<br /> }<br /> FhRequest = ::HttpOpenRequest(FhConnect, “GET”, fileName, “HTTP/1.0”, NULL, FAcceptTypes,<br /> INTERNET_FLAG_NO_CACHE_WRITE | INTERNET_FLAG_RELOAD, 0);<br /> if(FhRequest==NULL || FhRequest == INVALID_HANDLE_VALUE)<br /> {<br /> //出错处理<br /> this->DisposeError();<br /> return 0;<br /> }</p> <p> const char http_accept_header[] = “Accept: */*/r/n”;<br /> if (!::HttpAddRequestHeaders(FhRequest, http_accept_header, strlen(http_accept_header),<br /> HTTP_ADDREQ_FLAG_REPLACE))<br /> {<br /> //出错处理<br /> this->DisposeError();<br /> return 0;<br /> }</p> <p> const char data_type_header[] = “Content-Type: application/x-www-form-urlencoded/r/n”;<br /> if ( HttpAddRequestHeaders(FhRequest,data_type_header,strlen(data_type_header),<br /> HTTP_ADDREQ_FLAG_ADD|HTTP_ADDREQ_FLAG_REPLACE) == FALSE )<br /> {<br /> this->DisposeError();<br /> return 0;<br /> }</p> <p> if(::HttpSendRequest(FhRequest, NULL, 0, NULL, 0)==FALSE)<br /> {<br /> //出错处理<br /> this->DisposeError();<br /> return 0;<br /> }</p> <p> // 发送文件大小消息<br /> if( m_DownLoadMode == Mode_UnBlockAndMsg || m_DownLoadMode == Mode_BlockAndMsg)<br /> {<br /> char TmpBuf[64] = “/0″;<br /> DWORD bufLength = 64;<br /> memset(TmpBuf,0x00,64);<br /> if(::HttpQueryInfo(FhRequest, HTTP_QUERY_CONTENT_LENGTH, TmpBuf,&bufLength, NULL)==FALSE)<br /> {<br /> ::SendMessage(m_hWnd, DL_MSG, DL_ERROR, 0);<br /> }<br /> else<br /> {<br /> this->m_FileSize = atoi(TmpBuf);<br /> ::SendMessage(m_hWnd, DL_MSG, DL_FILESIZE, m_FileSize);<br /> }<br /> }</p> <p> // 创建文件<br /> hFile = CreateFile(m_OutputFileName,GENERIC_WRITE,FILE_SHARE_READ,NULL, /<br /> CREATE_ALWAYS,FILE_ATTRIBUTE_NORMAL,0);<br /> if( hFile == INVALID_HANDLE_VALUE )<br /> {<br /> return FALSE;<br /> }</p> <p> const int bufLength = 4096;<br /> char buf[bufLength];<br /> DWORD realWrite = 0;<br /> DWORD dwRead = 0;<br /> DWORD bRet = 0;</p> <p> memset(buf,0,bufLength);<br /> DWORD tick = ::timeGetTime();<br /> while(1)<br /> {<br /> bRet = ::InternetReadFile(FhRequest, (LPVOID)(buf), bufLength, &dwRead);<br /> if(dwRead<=0) break; //已读到文件尾<br /> if(bRet==FALSE)<br /> {<br /> this->DisposeError();<br /> return 0;<br /> }<br /> m_ReadCount = m_ReadCount+dwRead;<br /> if(::WriteFile(hFile,(void *)buf,dwRead,&realWrite,(LPOVERLAPPED)NULL)==0)<br /> {//写操作出错处理<br /> this->DisposeError();<br /> return 0;<br /> }<br /> if(::timeGetTime()-tick > 0)<br /> {<br /> m_Speed=(m_ReadCount/(::timeGetTime()-tick))*1000;<br /> }<br /> if( m_DownLoadMode == Mode_UnBlockAndMsg || m_DownLoadMode == Mode_BlockAndMsg)<br /> {<br /> ::SendMessage((HWND)m_hWnd, DL_MSG, DL_READDATA, (LPARAM)this);<br /> }<br /> }</p> <p> if(hFile)<br /> {<br /> ::CloseHandle(hFile);<br /> hFile = NULL;<br /> }<br /> if(FhRequest) {<br /> ::InternetCloseHandle(FhRequest);<br /> FhRequest = NULL;<br /> }<br /> if(FhConnect) {<br /> ::InternetCloseHandle(FhConnect);<br /> FhConnect = NULL;<br /> }<br /> if(FhSession) {<br /> ::InternetCloseHandle(FhSession);<br /> FhSession = NULL;<br /> }</p> <p> if( m_DownLoadMode == Mode_UnBlockAndMsg || m_DownLoadMode == Mode_BlockAndMsg)<br /> {<br /> ::SendMessage(m_hWnd,m_MsgCode,DL_TERMINATE,0);<br /> } </p> <p> return TRUE;<br /> }</p> <p>int CHttpDownLoadFile::StartDownLoad(LPSTR url,<br /> LPSTR outputfile,<br /> int mode,<br /> HWND hWnd,<br /> BOOL isOverWrite)<br /> {<br /> CHttpDownLoadFile* pDownLoad = new CHttpDownLoadFile;</p> <p> pDownLoad->SetWndHandle(hWnd);<br /> pDownLoad->SetOverWrite(isOverWrite);<br /> pDownLoad->SetOutputFileName(outputfile);<br /> pDownLoad->SetURL(url);<br /> pDownLoad->SetDownLoadMode(mode);</p> <p> return pDownLoad->StartDownLoad();<br /> }</p> <p>void CHttpDownLoadFile::ParseURL(char *Url,char *pHostName,char *pFileName,DWORD* nPort)<br /> {<br /> char *all=Url;<br /> char *p=strstr(all,”http://”);<br /> char Buffer[_MAX_PATH];<br /> memset(Buffer,0x00,_MAX_PATH);</p> <p> if(p)<br /> {<br /> int pos=p-all;<br /> all=all+pos+strlen(“http://”);<br /> p=strstr(all,”/”);<br /> if(p)<br /> {<br /> pos=p-all;<br /> memcpy(Buffer,all,pos);<br /> strcpy(pFileName,all+pos);</p> <p> p = strstr(Buffer,”:”);<br /> if(p)<br /> {<br /> pos = p – Buffer;<br /> memcpy(pHostName,Buffer,pos);<br /> *nPort = atoi(p+1);<br /> }else<br /> {<br /> strcpy(pHostName,Buffer);<br /> }<br /> }<br /> }<br /> }</p> <p>void CHttpDownLoadFile::DisposeError()<br /> {<br /> if(FhRequest) {<br /> ::InternetCloseHandle(FhRequest);<br /> FhRequest = NULL;<br /> }<br /> if(FhConnect) {<br /> ::InternetCloseHandle(FhConnect);<br /> FhConnect = NULL;<br /> }<br /> if(FhSession) {<br /> ::InternetCloseHandle(FhSession);<br /> FhSession = NULL;<br /> }<br /> if(hFile) {<br /> ::CloseHandle(hFile);<br /> hFile = NULL;<br /> }</p> <p> if( m_DownLoadMode == Mode_UnBlockAndMsg || m_DownLoadMode == Mode_BlockAndMsg)<br /> {<br /> ::SendMessage(m_hWnd,m_MsgCode,DL_ERROR,0);<br /> }<br /> }</p> <p>//main.cpp 测试<br /> int main(int argc,char* argv[])<br /> {<br /> CHttpDownLoadFile* pDown = new CHttpDownLoadFile();<br /> pDown->SetDownLoadMode(CHttpDownLoadFile::DownLoadMode::Mode_BlockNoMsg);<br /> pDown->SetURL(“http://www.brstime.com/”);<br /> pDown->SetOutputFileName(“brstime.html”);<br /> if( pDown->StartDownLoad() == TRUE)<br /> {<br /> printf(“down load ok!”);<br /> }<br /> return 1;<br /> }</p> <p>/*———————————<br /> CHttpDownLoadFile 的功能就是把网络上的网页，文件（测试过文本文件），下载到本地，并保存成文件。这个类的效率比较低，适用于下载比较小的文件，且不要求速度的场合。<br /> CHttpDownLoadFile 提供4中下载模式，即下载时是否发送消息，是否以阻塞的方式下载等。<br /> 测试用例中的下载模式是以阻塞且不发送消息的下载模式、下载一个网页。<br /> */

在上文的基础上，增加了保存模式功能，即提供把文件内容保存到string对象里，而不是写到文件。

// HttpDownLoadFile.h </p> <p>#ifndef _HTTP_DOWN_LOAD_FILE_H<br /> #define _HTTP_DOWN_LOAD_FILE_H</p> <p>#include <windows.h><br /> #include <wininet.h><br /> #include <string><br /> using std::string;</p> <p>#pragma comment(lib,”wininet.lib”)<br /> #pragma comment(lib,”Winmm.lib”)</p> <p>class CHttpDownLoadFile<br /> {<br /> public:<br /> enum DownLoadMode //下载模式<br /> {<br /> Mode_BlockNoMsg = 1, //阻塞，不传递消息模式<br /> Mode_BlockAndMsg, //阻塞，传递消息模式<br /> Mode_UnBlockNoMsg, //非阻塞，不传递消息模式<br /> Mode_UnBlockAndMsg, //非阻塞，传递消息模式<br /> };</p> <p> enum DownLoadMsg //下载时发送的消息<br /> {<br /> DL_OK = 0x0001,<br /> DL_ERROR,<br /> DL_START,<br /> DL_FILESIZE,<br /> DL_READDATA,<br /> DL_TERMINATE,<br /> };</p> <p> enum DownLoadWndMsg //定义的windows消息<br /> {<br /> DL_MSG = WM_USER + 200,<br /> };</p> <p> enum ESaveMode //保存模式<br /> {<br /> Mode_SaveToFile = 1, //保存到文件<br /> Mode_SaveToBuffer, //保存到缓冲区<br /> };</p> <p>private:<br /> HWND m_hWnd; // 下载时，消息的接收窗体<br /> int m_DownLoadMode; // 下载模式<br /> BOOL m_OverWrite; // 是否覆盖<br /> int m_ErrorCode; // 错误码<br /> DWORD m_Speed; // 下载速度<br /> DWORD m_ReadCount; // 已经下载的大小<br /> DWORD m_FileSize; // 文件大小<br /> DWORD m_MsgCode; // 下载时，发送的消息代码，初始化时等于 DL_MSG</p> <p> TCHAR m_OutputFileName[_MAX_PATH]; // 保存下载时的文件名<br /> TCHAR m_URL[_MAX_PATH]; // 资源的URL</p> <p> BOOL m_bDownLoadWebOK; // 是否下载完成<br /> DWORD m_WaitTime; // 下载的时间</p> <p> int m_SaveMode; // 保存模式<br /> private:</p> <p> HINTERNET FhSession ; //会话句柄<br /> HINTERNET FhConnect ; //http连接句柄<br /> HINTERNET FhRequest ; //http请求句柄<br /> HANDLE hFile; //保存内容的文件<br /> string* m_Buffer; //保存内容的缓冲区</p> <p>public:<br /> CHttpDownLoadFile();<br /> CHttpDownLoadFile(HWND hWnd);<br /> ~CHttpDownLoadFile();<br /> public:<br /> LPSTR GetURL() { return m_URL; }<br /> void SetURL( LPSTR Value ) { strcpy(m_URL,Value); }</p> <p> LPSTR GetOutputFileName() { return m_OutputFileName; }<br /> void SetOutputFileName( LPSTR fn ) { strcpy(m_OutputFileName, fn); m_SaveMode = Mode_SaveToBuffer; }</p> <p> BOOL SetOverWrite() { return m_OverWrite; }<br /> void SetOverWrite(BOOL Value) { m_OverWrite = Value; }</p> <p> void SetWndHandle(HWND hWnd) { m_hWnd = hWnd; }<br /> HWND GetWndHandle() { return m_hWnd; }</p> <p> void SetDownLoadMode(int mode) { m_DownLoadMode= mode; }<br /> int GetDownLoadMode() { return m_DownLoadMode; }</p> <p> void SetBuffer(string* buffer){ m_Buffer = buffer; m_SaveMode = Mode_SaveToBuffer; }<br /> string* GetBuffer() { return m_Buffer; }</p> <p> void SetSaveMode(int mode) { m_SaveMode = mode; }<br /> int GetSaveMode() { return m_SaveMode; }</p> <p> void SetMsgCode(DWORD code) { m_MsgCode = code; }</p> <p> DWORD GetSpeed() { return m_Speed; }<br /> DWORD GetFileSize() { return m_FileSize; }<br /> DWORD GetBeenDownLoadedSize() { return m_ReadCount; }</p> <p>public:<br /> int StartDownLoad(); // 开始下载,并把下载的内容保存到文件里。<br /> int StartDownLoad(string* buffer); // 开始下载，把内容保存到缓冲区里。<br /> int StartDownLoad(LPSTR fn); // 开始下载，把内容保存到文件里。<br /> public:<br /> int _RealDownLoad(); // 实际下载<br /> int _SaveToFile(); // 保存到文件<br /> int _SaveToBuffer(); // 保存到缓冲<br /> void ParseURL(char *Url,char *pHostName,char *pFileName,DWORD* nPort);</p> <p>public:<br /> static int StartDownLoad(LPSTR url,<br /> LPSTR outputFileName,<br /> string* buffer,<br /> int downmode = Mode_BlockNoMsg,<br /> int savemode = Mode_SaveToFile,<br /> HWND hWnd = NULL<br /> );<br /> static ULONG WINAPI DownLoadProc(LPVOID pDownLoadFile);<br /> private:<br /> void DisposeError();<br /> };</p> <p>#endif</p> <p>//HttpDownLoadFile.cpp<br /> #include “HttpDownLoadFile.h”</p> <p>CHttpDownLoadFile::CHttpDownLoadFile()<br /> {<br /> m_hWnd = NULL;<br /> m_DownLoadMode = Mode_BlockNoMsg;<br /> m_SaveMode = Mode_SaveToFile;<br /> m_OverWrite = TRUE;<br /> m_bDownLoadWebOK = FALSE;</p> <p> m_ErrorCode = 0;</p> <p> m_Speed = 0;<br /> m_ReadCount = 0;<br /> m_FileSize = 0;</p> <p> m_MsgCode = DL_MSG;<br /> m_WaitTime = 10000;</p> <p> hFile = NULL;<br /> m_Buffer = NULL;</p> <p> memset(m_URL,0x00,sizeof(m_URL));<br /> memset(m_URL,0x00,sizeof(m_OutputFileName));<br /> }</p> <p>CHttpDownLoadFile::CHttpDownLoadFile(HWND hWnd)<br /> {<br /> m_hWnd = hWnd;<br /> m_DownLoadMode = Mode_BlockNoMsg;<br /> m_SaveMode = Mode_SaveToFile;<br /> m_OverWrite = TRUE;<br /> m_bDownLoadWebOK = FALSE;</p> <p> m_ErrorCode = 0;</p> <p> m_Speed = 0;<br /> m_ReadCount = 0;<br /> m_FileSize = 0;</p> <p> m_MsgCode = DL_MSG;</p> <p> hFile = NULL;<br /> m_Buffer = NULL;</p> <p> memset(m_URL,0x00,sizeof(m_URL));<br /> memset(m_URL,0x00,sizeof(m_OutputFileName));<br /> }</p> <p>int CHttpDownLoadFile::StartDownLoad(string* buffer)<br /> {<br /> this->SetBuffer(buffer);<br /> return this->StartDownLoad();<br /> }</p> <p>int CHttpDownLoadFile::StartDownLoad(LPSTR fn)<br /> {<br /> this->SetOutputFileName(fn);<br /> return this->StartDownLoad();<br /> }</p> <p>int CHttpDownLoadFile::StartDownLoad()<br /> {<br /> if( m_DownLoadMode == Mode_BlockNoMsg || m_DownLoadMode == Mode_BlockAndMsg )<br /> {<br /> return this->_RealDownLoad();<br /> }else<br /> {<br /> ULONG threadID;<br /> HANDLE hThread = CreateThread(NULL,0,DownLoadProc,this,0,&threadID);<br /> CloseHandle(hThread);<br /> return TRUE;<br /> }<br /> }</p> <p>ULONG WINAPI CHttpDownLoadFile::DownLoadProc(LPVOID pDownLoadFile)<br /> {<br /> CHttpDownLoadFile* pDownLoad = (CHttpDownLoadFile*)(pDownLoadFile);<br /> return pDownLoad->_RealDownLoad();<br /> }</p> <p>int CHttpDownLoadFile::_RealDownLoad()<br /> {<br /> // 发送开始传输消息<br /> if( m_DownLoadMode == Mode_UnBlockAndMsg || m_DownLoadMode == Mode_BlockAndMsg)<br /> {<br /> ::SendMessage(m_hWnd,m_MsgCode,DL_START,0);<br /> }</p> <p> // 打开链接，开始下载<br /> FhSession = ::InternetOpen(“HttpDownLoadFile”, INTERNET_OPEN_TYPE_PRECONFIG, NULL, NULL, 0);// 同步方式<br /> if(FhSession == NULL || FhSession == INVALID_HANDLE_VALUE)<br /> {<br /> //出错处理<br /> this->DisposeError();<br /> return 0;<br /> }</p> <p> TCHAR server[_MAX_PATH];<br /> TCHAR fileName[_MAX_PATH];<br /> DWORD nPort = INTERNET_DEFAULT_HTTP_PORT;<br /> memset(server,0x00,_MAX_PATH);<br /> memset(fileName,0x00,_MAX_PATH);<br /> const char *FAcceptTypes[] = {“*/*”,NULL};</p> <p> this->ParseURL(m_URL,server,fileName,&nPort);<br /> // 得到下载后的文件名<br /> if( strcmp(m_OutputFileName,””) == 0)<br /> {<br /> strcpy(m_OutputFileName,fileName);<br /> }</p> <p> FhConnect = ::InternetConnect(FhSession, server, nPort,<br /> NULL, NULL, INTERNET_SERVICE_HTTP, 0, 0);<br /> if(FhConnect==NULL || FhConnect == INVALID_HANDLE_VALUE)<br /> {<br /> //出错处理<br /> this->DisposeError();<br /> return 0;<br /> }<br /> FhRequest = ::HttpOpenRequest(FhConnect, “GET”, fileName, “HTTP/1.0”, NULL, FAcceptTypes,<br /> INTERNET_FLAG_NO_CACHE_WRITE | INTERNET_FLAG_RELOAD, 0);<br /> if(FhRequest==NULL || FhRequest == INVALID_HANDLE_VALUE)<br /> {<br /> //出错处理<br /> this->DisposeError();<br /> return 0;<br /> }</p> <p> const char http_accept_header[] = “Accept: */*/r/n”;<br /> if (!::HttpAddRequestHeaders(FhRequest, http_accept_header, strlen(http_accept_header),<br /> HTTP_ADDREQ_FLAG_REPLACE))<br /> {<br /> //出错处理<br /> this->DisposeError();<br /> return 0;<br /> }</p> <p> const char data_type_header[] = “Content-Type: application/x-www-form-urlencoded/r/n”;<br /> if ( HttpAddRequestHeaders(FhRequest,data_type_header,strlen(data_type_header),<br /> HTTP_ADDREQ_FLAG_ADD|HTTP_ADDREQ_FLAG_REPLACE) == FALSE )<br /> {<br /> this->DisposeError();<br /> return 0;<br /> }</p> <p> if(::HttpSendRequest(FhRequest, NULL, 0, NULL, 0)==FALSE)<br /> {<br /> //出错处理<br /> this->DisposeError();<br /> return 0;<br /> }</p> <p> // 发送文件大小消息<br /> if( m_DownLoadMode == Mode_UnBlockAndMsg || m_DownLoadMode == Mode_BlockAndMsg)<br /> {<br /> char TmpBuf[64] = “/0″;<br /> DWORD bufLength = 64;<br /> memset(TmpBuf,0x00,64);<br /> if(::HttpQueryInfo(FhRequest, HTTP_QUERY_CONTENT_LENGTH, TmpBuf,&bufLength, NULL)==FALSE)<br /> {<br /> ::SendMessage(m_hWnd, DL_MSG, DL_ERROR, 0);<br /> }<br /> else<br /> {<br /> this->m_FileSize = atoi(TmpBuf);<br /> ::SendMessage(m_hWnd, DL_MSG, DL_FILESIZE, m_FileSize);<br /> }<br /> }</p> <p> if( this->m_SaveMode == Mode_SaveToFile )<br /> {<br /> this->_SaveToFile();<br /> }else<br /> {<br /> this->_SaveToBuffer();<br /> }</p> <p> if(FhRequest) {<br /> ::InternetCloseHandle(FhRequest);<br /> FhRequest = NULL;<br /> }<br /> if(FhConnect) {<br /> ::InternetCloseHandle(FhConnect);<br /> FhConnect = NULL;<br /> }<br /> if(FhSession) {<br /> ::InternetCloseHandle(FhSession);<br /> FhSession = NULL;<br /> }</p> <p> if( m_DownLoadMode == Mode_UnBlockAndMsg || m_DownLoadMode == Mode_BlockAndMsg)<br /> {<br /> ::SendMessage(m_hWnd,m_MsgCode,DL_TERMINATE,0);<br /> } </p> <p> return TRUE;<br /> }</p> <p>// 保存到文件<br /> int CHttpDownLoadFile::_SaveToFile()<br /> {<br /> // 创建文件<br /> hFile = CreateFile(m_OutputFileName,GENERIC_WRITE,FILE_SHARE_READ,NULL, /<br /> CREATE_ALWAYS,FILE_ATTRIBUTE_NORMAL,0);<br /> if( hFile == INVALID_HANDLE_VALUE )<br /> {<br /> return FALSE;<br /> }</p> <p> const int bufLength = 4096;<br /> char buf[bufLength];<br /> DWORD realWrite = 0;<br /> DWORD dwRead = 0;<br /> DWORD bRet = 0;</p> <p> memset(buf,0,bufLength);<br /> DWORD tick = ::timeGetTime();<br /> while(1)<br /> {<br /> bRet = ::InternetReadFile(FhRequest, (LPVOID)(buf), bufLength, &dwRead);<br /> if(dwRead<=0) break; //已读到文件尾<br /> if(bRet==FALSE)<br /> {<br /> this->DisposeError();<br /> return 0;<br /> }<br /> m_ReadCount = m_ReadCount+dwRead;<br /> if(::WriteFile(hFile,(void *)buf,dwRead,&realWrite,(LPOVERLAPPED)NULL)==0)<br /> {//写操作出错处理<br /> this->DisposeError();<br /> return 0;<br /> }<br /> if(::timeGetTime()-tick > 0)<br /> {<br /> m_Speed=(m_ReadCount/(::timeGetTime()-tick))*1000;<br /> }<br /> if( m_DownLoadMode == Mode_UnBlockAndMsg || m_DownLoadMode == Mode_BlockAndMsg)<br /> {<br /> ::SendMessage((HWND)m_hWnd, DL_MSG, DL_READDATA, (LPARAM)this);<br /> }<br /> }</p> <p> if(hFile)<br /> {<br /> ::CloseHandle(hFile);<br /> hFile = NULL;<br /> }<br /> return 1;<br /> }</p> <p>// 保存到缓冲<br /> int CHttpDownLoadFile::_SaveToBuffer()<br /> {<br /> const int bufLength = 4096;<br /> char buf[bufLength];<br /> DWORD dwRead = 0;<br /> DWORD bRet = 0;</p> <p> memset(buf,0,bufLength);<br /> DWORD tick = ::timeGetTime();</p> <p> if( m_Buffer == NULL )<br /> {<br /> m_Buffer = new string;<br /> }</p> <p> while(1)<br /> {<br /> bRet = ::InternetReadFile(FhRequest, (LPVOID)(buf), bufLength, &dwRead);<br /> if(dwRead<=0) break; //已读到文件尾</p> <p> m_ReadCount = m_ReadCount+dwRead;<br /> *m_Buffer += buf;</p> <p> if(::timeGetTime()-tick > 0)<br /> {<br /> m_Speed=(m_ReadCount/(::timeGetTime()-tick))*1000;<br /> }<br /> if( m_DownLoadMode == Mode_UnBlockAndMsg || m_DownLoadMode == Mode_BlockAndMsg)<br /> {<br /> ::SendMessage((HWND)m_hWnd, DL_MSG, DL_READDATA, (LPARAM)this);<br /> }<br /> }<br /> return 1;<br /> }</p> <p>int CHttpDownLoadFile::StartDownLoad(LPSTR url,<br /> LPSTR outputFileName,<br /> string* buffer,<br /> int downmode,<br /> int savemode,<br /> HWND hWnd)<br /> {<br /> CHttpDownLoadFile* pDownLoad = new CHttpDownLoadFile;</p> <p> pDownLoad->SetURL(url);<br /> pDownLoad->SetOutputFileName(outputFileName);<br /> pDownLoad->SetBuffer(buffer);<br /> pDownLoad->SetDownLoadMode(downmode);<br /> pDownLoad->SetSaveMode(savemode);<br /> pDownLoad->SetWndHandle(hWnd);</p> <p> return pDownLoad->StartDownLoad();<br /> }</p> <p>void CHttpDownLoadFile::ParseURL(char *Url,char *pHostName,char *pFileName,DWORD* nPort)<br /> {<br /> char *all=Url;<br /> char *p=strstr(all,”http://”);<br /> char Buffer[_MAX_PATH];<br /> memset(Buffer,0x00,_MAX_PATH);</p> <p> if(p)<br /> {<br /> int pos=p-all;<br /> all=all+pos+strlen(“http://”);<br /> p=strstr(all,”/”);<br /> if(p)<br /> {<br /> pos=p-all;<br /> memcpy(Buffer,all,pos);<br /> strcpy(pFileName,all+pos);</p> <p> p = strstr(Buffer,”:”);<br /> if(p)<br /> {<br /> pos = p – Buffer;<br /> memcpy(pHostName,Buffer,pos);<br /> *nPort = atoi(p+1);<br /> }else<br /> {<br /> strcpy(pHostName,Buffer);<br /> }<br /> }<br /> }<br /> }</p> <p>void CHttpDownLoadFile::DisposeError()<br /> {<br /> if(FhRequest) {<br /> ::InternetCloseHandle(FhRequest);<br /> FhRequest = NULL;<br /> }<br /> if(FhConnect) {<br /> ::InternetCloseHandle(FhConnect);<br /> FhConnect = NULL;<br /> }<br /> if(FhSession) {<br /> ::InternetCloseHandle(FhSession);<br /> FhSession = NULL;<br /> }<br /> if(hFile) {<br /> ::CloseHandle(hFile);<br /> hFile = NULL;<br /> }</p> <p> if( m_DownLoadMode == Mode_UnBlockAndMsg || m_DownLoadMode == Mode_BlockAndMsg)<br /> {<br /> ::SendMessage(m_hWnd,m_MsgCode,DL_ERROR,0);<br /> }<br /> }