使用 Wininet 下载二进制文件

Question

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

我想在不使用任何外部库的情况下执行此操作。

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

谢谢， 安德鲁

Answer 1

文档做出以下评论：

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

基本上，不能保证该函数准确读取 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" )