網絡下載文件小程序(非原創,他人所寫)

#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

從一個由InternetOpenUrlFtpOpenFile,  HttpOpenRequest 函數打開的句柄中讀取數據。

SyntaxC++BOOL InternetReadFile( __in HINTERNET hFile, __out LPVOID lpBuffer, __in DWORD dwNumberOfBytesToRead, __out LPDWORD lpdwNumberOfBytesRead);Parameters

hFile [in]

InternetOpenUrlFtpOpenFile,  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)


發佈了28 篇原創文章 · 獲贊 4 · 訪問量 1萬+
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章