#include <Windows.h>
#include <iostream>
#include <string>
#include <vector>
#include <algorithm>
#include <wininet.h>
#include <shellApi.h>
#include <Shlwapi.h>
#include <ShlObj.h>
#define INVALID_FILE_ATTRIBUTES ((DWORD)-1)
#pragma comment(lib, "shlwapi.lib")
#pragma comment(lib,"wininet")
using namespace std;
bool DownloadFile(LPCTSTR szUrl,LPCTSTR szLocalFile,BOOL bFailIfExists);
int main(int argc, char* argv[])
{
std::cout<<DownloadFile("http://www.baidu.com/","d:\\1.htm",TRUE)<<std::endl;
getchar();
return 0;
}
bool DownloadFile(LPCTSTR szUrl,LPCTSTR szLocalFile,BOOL bFailIfExists)
{
if (bFailIfExists &&
!PathIsDirectory(szLocalFile) &&
(GetFileAttributes(szLocalFile) != INVALID_FILE_ATTRIBUTES))
{
return false;
}
HANDLE hFile = INVALID_HANDLE_VALUE;
HINTERNET hInet = NULL;
HINTERNET hUrl = NULL;
DWORD dwBuf = 1024*1024,dwRead = 0; //1M
auto_ptr<char> szBuf(new char[dwBuf]);
memset(szBuf.get(),0,dwBuf);
std::string strTmp;
bool bRet = false;
try
{
hFile = CreateFile(szLocalFile,GENERIC_READ|GENERIC_WRITE,0,NULL,CREATE_ALWAYS,FILE_ATTRIBUTE_NORMAL,NULL);
if (hFile == INVALID_HANDLE_VALUE)
throw "error";
hInet = InternetOpen(NULL,INTERNET_OPEN_TYPE_PRECONFIG,NULL,NULL,0);
if (hInet == NULL)
throw "error";
hUrl = InternetOpenUrl(hInet,szUrl,NULL,0,
INTERNET_FLAG_TRANSFER_BINARY|INTERNET_FLAG_NO_CACHE_WRITE|INTERNET_FLAG_RELOAD,0);
if (hUrl == NULL)
throw "error";
for(;;)
{
if (!InternetReadFile(hUrl,szBuf.get(),dwBuf,&dwRead))
{
bRet = false;
break;
}
if (dwRead == 0)
{
bRet = true;
break;
}
//strTmp += std::string(szBuf,dwRead);
WriteFile(hFile,szBuf.get(),dwRead,&dwRead,NULL);
}
throw "ok";
}
catch(...)
{
if (hFile != INVALID_HANDLE_VALUE)
CloseHandle(hFile);
if (hUrl != NULL)
InternetCloseHandle(hUrl);
if (hInet != NULL)
InternetCloseHandle(hInet);
}
return bRet;
}
InternetReadFile函数
InternetReadFile Function
从一个由InternetOpenUrl, FtpOpenFile, 或 HttpOpenRequest 函数打开的句柄中读取数据。
SyntaxC++BOOL InternetReadFile( __in HINTERNET hFile, __out LPVOID lpBuffer, __in DWORD dwNumberOfBytesToRead, __out LPDWORD lpdwNumberOfBytesRead);Parameters
hFile [in]
由InternetOpenUrl, FtpOpenFile, 或 HttpOpenRequest 函数返回的句柄.
lpBuffer [out]
缓冲器指针
dwNumberOfBytesToRead [in]
欲读数据的字节量。
lpdwNumberOfBytesRead [out]
接收读取字节量的变量。该函数在做任何工作或错误检查之前都设置该值为零
返回值
成功:返回TRUE,失败,返回FALSE
备注
该函数的操作非常类似于 ReadFile 函数。典型情况下,该函数从一个作为字节顺序流的 HINTERNET 句柄中返回数据。每次函数读取的数据量都由 dwNumberOfBytesToRead 参数指定,并且数据返回到 lp
Buffer参数里。一个正常读取将持续到文件结束时为止。为了确保所有数据被接收,应用程序必须继续使用该函数直到函数返回TRUE且lpdwNumberOfBytesRead参数等于零值。此举在请求的数据被写入缓存中尤为重要,因为如不这样的话,缓存将不会正常的更新且下载的文件也将不会被允许写入到缓存中。注意:除了原始请求的命令设置为INTERNET_FLAG_NO_CACHE_WRITE标识的情况外,缓存操作都会自动进行。当应用程序使用InternetOpenUrl函数获得一个句柄时,WinINet试图将所有的数据都变得如同文件下载一样。
WinINet试图每次向lpBuffer缓冲中写入一行HTML。如果应用程序的缓冲区太小以至于不能承载一行的HTML数据,错误代码ERROR_INSUFFICIENT_BUFFER 将发送给应用程序,通知它需要更大的缓冲区。同样,转换的行数也不一定和缓冲区的大小匹配,所以该函数可以返回比lpBuffer缓冲少的数据。接下来的读取工作将返回所有的HTML数据。应用程序必须再次检查搜有数据是否都被接收。
(InternetReadFile函数说明转自:http://hi.baidu.com/fangenhong/item/5b11fd91777e6f1b924f41e4)