使用 Wininet 下载二进制文件

Question

我目前正在编写一个简单的程序，我想分发给我的朋友。我想要完成的是在启动程序时将一些外部二进制文件从互联网写入缓冲区。为此，我使用的是 Windows Internet（wininet）。目前，我正在使用 InternetReadFile 将文件写入我稍后在程序中使用的缓冲区。但是，文件并没有被完全读取，因为结果大小比服务器上文件的大小要小得多，当它应该相同时。

我想这样做，不使用任何外部库。

知道什么可以解决我的问题吗？

谢谢， 安德鲁

Answer 1

documentation 发表如下评论：

InternetReadFile 的操作与基本的 ReadFile 函数非常相似，但有一些例外。通常，InternetReadFile 从 HINTERNET 句柄中检索数据作为顺序字节流。每次调用 InternetReadFile 要读取的数据量由 dwNumberOfBytesToRead 参数指定，数据在 lpBuffer 参数中返回。正常读取为每次调用 InternetReadFile 检索指定的 dwNumberOfBytesToRead，直到到达文件末尾。 为确保检索到所有数据，应用程序必须继续调用 InternetReadFile 函数，直到该函数返回 TRUE 并且 lpdwNumberOfBytesRead 参数为零。

基本上，不能保证函数能准确读取dwNumberOfBytesToRead。使用lpdwNumberOfBytesRead 参数检查实际读取了多少字节。

此外，一旦总文件大小大于dwNumberOfBytesToRead，您将需要多次调用该调用。因为它一次不能读取超过dwNumberOfBytesToRead。

如果您预先知道总文件大小，则循环采用以下形式：

::DWORD error = ERROR_SUCCESS;
::BYTE data[SIZE]; // total file size.
::DWORD size = 0;
::DWORD read = 0;
do {
    ::BOOL result = ::InternetReadFile(stream, data+size, SIZE-size, &read);
    if ( result == FALSE ) {
        error = ::GetLastError();
    }
}
while ((error == ERROR_SUCCESS) && (read > 0) && ((size+=read) < SIZE));
  // check that `SIZE` was correct.
if (size != SIZE) {
}

如果没有，则需要将缓冲区中的数据写入另一个文件，而不是累积。

编辑（示例测试程序）：

这是一个获取 StackOverflow 首页的完整程序。这会以 1K 块下载大约 200K 的 HTML 代码，并检索整个页面。你能运行它看看它是否有效吗？

#include <Windows.h>
#include <Wininet.h>
#include <iostream>
#include <fstream>

namespace {

    ::HINTERNET netstart ()
    {
        const ::HINTERNET handle =
            ::InternetOpenW(0, INTERNET_OPEN_TYPE_DIRECT, 0, 0, 0);
        if ( handle == 0 )
        {
            const ::DWORD error = ::GetLastError();
            std::cerr
                << "InternetOpen(): " << error << "."
                << std::endl;
        }
        return (handle);
    }

    void netclose ( ::HINTERNET object )
    {
        const ::BOOL result = ::InternetCloseHandle(object);
        if ( result == FALSE )
        {
            const ::DWORD error = ::GetLastError();
            std::cerr
                << "InternetClose(): " << error << "."
                << std::endl;
        }
    }

    ::HINTERNET netopen ( ::HINTERNET session, ::LPCWSTR url )
    {
        const ::HINTERNET handle =
            ::InternetOpenUrlW(session, url, 0, 0, 0, 0);
        if ( handle == 0 )
        {
            const ::DWORD error = ::GetLastError();
            std::cerr
                << "InternetOpenUrl(): " << error << "."
                << std::endl;
        }
        return (handle);
    }

    void netfetch ( ::HINTERNET istream, std::ostream& ostream )
    {
        static const ::DWORD SIZE = 1024;
        ::DWORD error = ERROR_SUCCESS;
        ::BYTE data[SIZE];
        ::DWORD size = 0;
        do {
            ::BOOL result = ::InternetReadFile(istream, data, SIZE, &size);
            if ( result == FALSE )
            {
                error = ::GetLastError();
                std::cerr
                    << "InternetReadFile(): " << error << "."
                    << std::endl;
            }
            ostream.write((const char*)data, size);
        }
        while ((error == ERROR_SUCCESS) && (size > 0));
    }

}

int main ( int, char ** )
{
    const ::WCHAR URL[] = L"http://stackoverflow.com/";
    const ::HINTERNET session = ::netstart();
    if ( session != 0 )
    {
        const ::HINTERNET istream = ::netopen(session, URL);
        if ( istream != 0 )
        {
            std::ofstream ostream("output.txt", std::ios::binary);
            if ( ostream.is_open() ) {
                ::netfetch(istream, ostream);
            }
            else {
                std::cerr << "Could not open 'output.txt'." << std::endl;
            }
            ::netclose(istream);
        }
        ::netclose(session);
    }
}

#pragma comment ( lib, "Wininet.lib" )