c++ - 上传文件时的 WinHTTP 问题

标签 c++ visual-c++ winhttp

在花了几天的时间把我的头撞到墙上后,我想我应该在这里问一下。

下面代码的问题是我基本上是在迭代一个目录,将文件上传到那里。所有文件都很小,大小约为 1KB,因此这不是大小问题。第一次上传顺利通过,所有后续调用都因 winhttp 仅发送 header 而中断。

代码如下:

BOOL NetworkManager::UploadFileToServer(wchar_t *pszURL, wchar_t *pszFilePath, wchar_t *_pszProxyAddress, wchar_t *pszServerAddress)
{
HINTERNET  hSession = NULL, 
hConnect = NULL,
hRequest = NULL;
BOOL bResults;
DWORD dwSize = 0;
DWORD dwContentLength = 0;
LPCWSTR pszProxyAddress = 0;
wchar_t wszContentLength[256] = { 0 };

pszProxyAddress = _pszProxyAddress;

printf("Trying to send %S\r\n", pszFilePath);

if(pszProxyAddress != NULL && wcslen(pszProxyAddress) < 4)
{
    pszProxyAddress = NULL;
}

HANDLE hFile = CreateFile(pszFilePath, GENERIC_READ, FILE_SHARE_READ, NULL, OPEN_EXISTING, NULL, NULL);

if(hFile == INVALID_HANDLE_VALUE)
{
    printf("UM: Unable to open the file for sending, aborting...\r\n");
    return FALSE;
}

DWORD dwFileSize = GetFileSize(hFile, NULL);

// Use WinHttpOpen to obtain a session handle.
if(pszProxyAddress == NULL)
{
    hSession = WinHttpOpen( L"Mozilla/5.0 (Windows NT 6.3; Trident/7.0; rv:11.0) like Gecko", 
                        WINHTTP_ACCESS_TYPE_DEFAULT_PROXY,
                        NULL,
                        WINHTTP_NO_PROXY_BYPASS, 0);
}
else
{
    hSession = WinHttpOpen( L"Mozilla/5.0 (Windows NT 6.3; Trident/7.0; rv:11.0) like Gecko", 
                        WINHTTP_ACCESS_TYPE_NAMED_PROXY,
                        pszProxyAddress, 
                        WINHTTP_NO_PROXY_BYPASS, 0);
}

// Specify an HTTP server.
if (hSession)
{
    hConnect = WinHttpConnect( hSession, _pszServerAddress,
                               INTERNET_DEFAULT_HTTPS_PORT, 0);
}
else
{
    printf("hSession failed, errorcode 0x%08x\r\n", GetLastError());
    return FALSE;
}

// Create an HTTP request handle.
if (hConnect)
{
    hRequest = WinHttpOpenRequest( hConnect, L"POST", L"upload.php",
                                   NULL, WINHTTP_NO_REFERER, 
                                   WINHTTP_DEFAULT_ACCEPT_TYPES, 
                                   WINHTTP_FLAG_SECURE);
}
else
{
    printf("hConnect failed, errorcode 0x%08x\r\n", GetLastError());
    WinHttpCloseHandle(hSession);
    return FALSE;
}
PHEAP_BUFFER pBuf = NULL;
DWORD dwBytesWritten = 0;
// Send a request.
if (hRequest)
{

DWORD options = SECURITY_FLAG_IGNORE_CERT_CN_INVALID | SECURITY_FLAG_IGNORE_CERT_DATE_INVALID  |  SECURITY_FLAG_IGNORE_UNKNOWN_CA ;

    bResults = WinHttpSetOption( hRequest, WINHTTP_OPTION_SECURITY_FLAGS , (LPVOID)&options, sizeof (DWORD) );

WinHttpAddRequestHeaders(hRequest, L"Content-Type: multipart/form-data; boundary=----BoundaryXu02", (ULONG) -1L, WINHTTP_ADDREQ_FLAG_ADD);

dwContentLength = strlen(pszFormHeader) + dwFileSize + strlen(pszFinalBoundary);        
DWORD dwTotalSent = 0;
pBuf = MemoryManager::AllocateHeapMemory(dwContentLength, 1);
DWORD dwBytesRead = 0;
strcat_s((PCHAR)pBuf->pBuffer, pBuf->dwBufferSize, pszFormHeader);
ReadFile(hFile, &pBuf->pBuffer[strlen(pszFormHeader)], dwFileSize, &dwBytesRead, NULL);
memcpy(&pBuf->pBuffer[strlen(pszFormHeader) + dwFileSize], pszFinalBoundary, strlen(pszFinalBoundary));

wsprintf(wszContentLength, L"Content-Length: %d", dwContentLength);

bResults = WinHttpSendRequest( hRequest, wszContentLength, -1, 0, 0, dwContentLength, 0);

printf("Sending out the request\r\n");

WinHttpWriteData(hRequest, pBuf->pBuffer, pBuf->dwBufferSize, &dwBytesWritten);
}
else
{
    printf("hRequest failed, errorcode 0x%08x\r\n", GetLastError());
    WinHttpCloseHandle(hSession);
    WinHttpCloseHandle(hConnect);
    return FALSE;
}


//WinHttpWriteData(hRequest, pBuf->pBuffer, pBuf->dwBufferSize, &dwBytesWritten);

 // End the request.
if (bResults)
{
    bResults = WinHttpReceiveResponse( hRequest, NULL);
}
else
{
    printf("hResults failed, errorcode 0x%08x\r\n");
    WinHttpCloseHandle(hRequest);
    WinHttpCloseHandle(hConnect);
    WinHttpCloseHandle(hSession);
    MemoryManager::FreeHeapMemory(pBuf);
    return FALSE;
}

WinHttpQueryDataAvailable(hRequest, &dwBytesWritten);

// Close any open handles.
if (hRequest) WinHttpCloseHandle(hRequest);
if (hConnect) WinHttpCloseHandle(hConnect);
if (hSession) WinHttpCloseHandle(hSession);
MemoryManager::FreeHeapMemory(pBuf);
CloseHandle(hFile);
DeleteFile(pszFilePath);
return TRUE;
}

这是来自服务器的访问日志:

xxx.xxx.xxx.244 - - [09/Jan/2014:16:39:33 +0200] "POST /upload.php HTTP/1.1" 200 1811 "-" "Mozilla/5.0 (Windows NT 6.3; Trident/7.0; rv:11.0) like Gecko"
xxx.xxx.xxx.244 - - [09/Jan/2014:16:39:33 +0200] "POST /upload.php HTTP/1.1" 200 295 "-" "Mozilla/5.0 (Windows NT 6.3; Trident/7.0; rv:11.0) like Gecko"
xxx.xxx.xxx.244 - - [09/Jan/2014:16:39:33 +0200] "POST /upload.php HTTP/1.1" 200 295 "-" "Mozilla/5.0 (Windows NT 6.3; Trident/7.0; rv:11.0) like Gecko"
xxx.xxx.xxx.244 - - [09/Jan/2014:16:39:33 +0200] "POST /upload.php HTTP/1.1" 200 295 "-" "Mozilla/5.0 (Windows NT 6.3; Trident/7.0; rv:11.0) like Gecko"
xxx.xxx.xxx.244 - - [09/Jan/2014:16:39:33 +0200] "POST /upload.php HTTP/1.1" 200 295 "-" "Mozilla/5.0 (Windows NT 6.3; Trident/7.0; rv:11.0) like Gecko"
xxx.xxx.xxx.244 - - [09/Jan/2014:16:39:34 +0200] "POST /upload.php HTTP/1.1" 200 295 "-" "Mozilla/5.0 (Windows NT 6.3; Trident/7.0; rv:11.0) like Gecko"
xxx.xxx.xxx.244 - - [09/Jan/2014:16:39:34 +0200] "POST /upload.php HTTP/1.1" 200 295 "-" "Mozilla/5.0 (Windows NT 6.3; Trident/7.0; rv:11.0) like Gecko"
xxx.xxx.xxx.244 - - [09/Jan/2014:16:39:34 +0200] "POST /upload.php HTTP/1.1" 200 295 "-" "Mozilla/5.0 (Windows NT 6.3; Trident/7.0; rv:11.0) like Gecko"
xxx.xxx.xxx.244 - - [09/Jan/2014:16:39:34 +0200] "POST /upload.php HTTP/1.1" 200 295 "-" "Mozilla/5.0 (Windows NT 6.3; Trident/7.0; rv:11.0) like Gecko"
xxx.xxx.xxx.244 - - [09/Jan/2014:16:39:34 +0200] "POST /upload.php HTTP/1.1" 200 295 "-" "Mozilla/5.0 (Windows NT 6.3; Trident/7.0; rv:11.0) like Gecko"
xxx.xxx.xxx.244 - - [09/Jan/2014:16:39:34 +0200] "POST /upload.php HTTP/1.1" 200 295 "-" "Mozilla/5.0 (Windows NT 6.3; Trident/7.0; rv:11.0) like Gecko"
xxx.xxx.xxx.244 - - [09/Jan/2014:16:39:34 +0200] "POST /upload.php HTTP/1.1" 200 295 "-" "Mozilla/5.0 (Windows NT 6.3; Trident/7.0; rv:11.0) like Gecko"
xxx.xxx.xxx.244 - - [09/Jan/2014:16:39:34 +0200] "POST /upload.php HTTP/1.1" 200 295 "-" "Mozilla/5.0 (Windows NT 6.3; Trident/7.0; rv:11.0) like Gecko"
xxx.xxx.xxx.244 - - [09/Jan/2014:16:39:34 +0200] "POST /upload.php HTTP/1.1" 200 295 "-" "Mozilla/5.0 (Windows NT 6.3; Trident/7.0; rv:11.0) like Gecko"

我完全不知道为什么世界上第一个 POST 成功但其余的去 sh*t :(

编辑:

忘记添加标题声明:

char *pszFormHeader = "------BoundaryXu02\r\nContent-Disposition: form-data; name=\"uploaded\"; filename=\"aviconv.dat\"\r\nContent-Type: application/octet-stream\r\n\r\n";
char *pszFinalBoundary = "\r\n------BoundaryXu02--\r\n";
char *pwzContentHeader = "Content-Type: multipart/form-data; boundary=----BoundaryXu02";
wchar_t wszContentLength[256] = { 0 };

最佳答案

好吧,终于弄清楚了代码存在的问题。

1) 为了防止潜在的代理干扰上传,在调用 WinHttpOpenRequest 时使用 WINHTTP_FLAG_REFRESH

2) 由于在此代码中您为每次上传创建新连接,因此某些服务器似乎感到困惑,因为 WinHTTP 的默认设置是在 header 中使用“Connection: keep-alive”。添加“连接:关闭”可解决该问题。

3) 似乎调用 WinHttpReceiveResponse 和 WinHttpQueryDataAvailable 是不够的。在最后添加 WinHttpReadData 调用后,一切开始正常运行。

4) 为确保每次我添加 CurrentTickCount 调用时上传使用不同的边界。只是为了确保沿途的愚蠢代理不会认为“我以前见过这个”

以下是通过SSL (HTTPS) 将文件上传到php 脚本的功能代码。我已经删除了引用我项目中其他组件的调用,并将它们替换为普通的 C 等效项。

我希望这对其他使用 WinHTTP 的人有所帮助:)

此外,请注意该函数会在发送后删除文件,因此不要只是复制/粘贴并在 c:\windows\system32\kernel32.dll 上测试...

免责声明:我确实知道代码仍有改进的余地,特别是在一些不安全的函数调用上,以及如果文件大小大于 DWORD,您会遇到问题。再一次,另一方面,如果您尝试像这样发送它们,WinHTTP 无论如何都会拒绝上传那么大的文件。

#include "stdafx.h"
#include <Windows.h>
#include <stdio.h>
#include <string.h>
#include <winhttp.h>

char *pszFormHeader = "------Boundary%08X\r\nContent-Disposition: form-data; name=\"uploaded\"; filename=\"%08x.dat\"\r\nContent-Type: application/octet-stream\r\n\r\n";
char *pszFinalBoundary = "\r\n------Boundary%08X--\r\n";
wchar_t *wszContentHeader = L"Content-Type: multipart/form-data; boundary=----Boundary%08X";
wchar_t wszContentLength[256] = { 0 };

#define BUFFER_SIZE_1KB 1024

BOOL UploadFileToServer(wchar_t *wszURL, wchar_t *wszFilePath, wchar_t *_wszProxyAddress, wchar_t *_wszServerAddress);

int main(int argc, char **argv)
{
    UploadFileToServer(L"upload.php", L"c:\\sample1.txt", NULL, L"www.example.com");
    UploadFileToServer(L"upload.php", L"c:\\sample2.txt", NULL, L"www.example.com");

}

BOOL UploadFileToServer(wchar_t *wszURL, wchar_t *wszFilePath, wchar_t *_wszProxyAddress, wchar_t *_wszServerAddress)
{
    /*
     * Declarations and initializations
     */
    HINTERNET  hSession = NULL, hConnect = NULL, hRequest = NULL;
    BOOL bResults;
    DWORD dwSize = 0;
    DWORD dwContentLength = 0;
    DWORD dwBytesToRead = 0;
    DWORD dwBytesRead = 0;
    DWORD dwBytesWritten = 0;
    DWORD dwBoundaryValue = 0;
    DWORD dwFileSize = 0;
    DWORD options = SECURITY_FLAG_IGNORE_CERT_CN_INVALID | SECURITY_FLAG_IGNORE_CERT_DATE_INVALID  |  SECURITY_FLAG_IGNORE_UNKNOWN_CA ;
    LPCWSTR wszProxyAddress = 0;
    wchar_t wszContentLength[256] = { 0 };

    PUCHAR pResponse = 0;
    PCHAR pFormHeader = 0;
    PCHAR pFinalBoundary = 0;
    PUCHAR pBuf = 0;
    wchar_t *pContentHeader = 0;

    /*
     * Preparations, set up the content headers
     */
    pFormHeader = (PCHAR) calloc(BUFFER_SIZE_1KB, 1);
    pFinalBoundary = (PCHAR) calloc(BUFFER_SIZE_1KB, 1);
    pContentHeader = (wchar_t *) calloc(BUFFER_SIZE_1KB, 1);

    dwBoundaryValue = GetTickCount();
    wszProxyAddress = _wszProxyAddress;

    sprintf_s(pFormHeader, BUFFER_SIZE_1KB, pszFormHeader, dwBoundaryValue, dwBoundaryValue);
    sprintf_s(pFinalBoundary, BUFFER_SIZE_1KB, pszFinalBoundary, dwBoundaryValue);
    wsprintf(pContentHeader, wszContentHeader, dwBoundaryValue);

    if(wszProxyAddress != NULL && wcslen(wszProxyAddress) < 4)
    {
        wszProxyAddress = NULL;
    }

    HANDLE hFile = CreateFile(wszFilePath, GENERIC_READ, FILE_SHARE_READ, NULL, OPEN_EXISTING, NULL, NULL);

    if(hFile == INVALID_HANDLE_VALUE)
    {
        free(pFormHeader);
        free(pFinalBoundary);
        free(pContentHeader);
        printf("Unable to open the file for sending, aborting...\r\n");
        return FALSE;
    }

    dwFileSize = GetFileSize(hFile, NULL);

    if(dwFileSize == 0)
    {
        free(pFormHeader);
        free(pFinalBoundary);
        free(pContentHeader);
        printf("The file is 0 bytes in size, cannot send it\r\n");
        return FALSE;
    }

    if(wszProxyAddress == NULL)
    {
        hSession = WinHttpOpen( L"Mozilla/5.0 (Windows NT 6.3; Trident/7.0; rv:11.0) like Gecko", 
                            WINHTTP_ACCESS_TYPE_DEFAULT_PROXY,
                            NULL,
                            WINHTTP_NO_PROXY_BYPASS, 0);
    }
    else
    {
        hSession = WinHttpOpen( L"Mozilla/5.0 (Windows NT 6.3; Trident/7.0; rv:11.0) like Gecko", 
                            WINHTTP_ACCESS_TYPE_NAMED_PROXY,
                            wszProxyAddress, 
                            WINHTTP_NO_PROXY_BYPASS, 0);
    }


    if (hSession)
    {
        hConnect = WinHttpConnect( hSession, _wszServerAddress,
                                INTERNET_DEFAULT_HTTPS_PORT, 0);
    }
    else
    {
        free(pFormHeader);
        free(pFinalBoundary);
        free(pContentHeader);
        printf("hSession failed, errorcode 0x%08x\r\n", GetLastError());
        return FALSE;
    }

    if (hConnect)
    {
        hRequest = WinHttpOpenRequest( hConnect, L"POST", wszURL,
                                   NULL, WINHTTP_NO_REFERER, 
                                   WINHTTP_DEFAULT_ACCEPT_TYPES, 
                                   WINHTTP_FLAG_SECURE | WINHTTP_FLAG_REFRESH);
    }
    else
    {
        free(pFormHeader);
        free(pFinalBoundary);
        free(pContentHeader);
        printf("hConnect failed, errorcode 0x%08x\r\n", GetLastError());
        WinHttpCloseHandle(hSession);
        return FALSE;
    }

if (hRequest)
{
    bResults = WinHttpSetOption( hRequest, WINHTTP_OPTION_SECURITY_FLAGS , (LPVOID)&options, sizeof (DWORD) );

    WinHttpAddRequestHeaders(hRequest, pContentHeader, (ULONG) -1L, WINHTTP_ADDREQ_FLAG_ADD);
    WinHttpAddRequestHeaders(hRequest, L"Connection: close", (ULONG) -1L, WINHTTP_ADDREQ_FLAG_ADD | WINHTTP_ADDREQ_FLAG_REPLACE);

    dwContentLength = strlen(pFormHeader) + dwFileSize + strlen(pFinalBoundary);        

    pBuf = (PUCHAR)calloc(dwContentLength, 1);

    strcat_s((PCHAR)pBuf, dwContentLength, pFormHeader);

    ReadFile(hFile, &pBuf[strlen(pFormHeader)], dwFileSize, &dwBytesRead, NULL);

    memcpy(&pBuf[strlen(pFormHeader) + dwFileSize], pFinalBoundary, strlen(pFinalBoundary));

    wsprintf(wszContentLength, L"Content-Length: %d", dwContentLength);

    bResults = WinHttpSendRequest( hRequest,
                                    wszContentLength,
                                    -1, pBuf, dwContentLength, 
                                   dwContentLength, 0);

}
else
{
    free(pFormHeader);
    free(pFinalBoundary);
    free(pContentHeader);
    printf("hRequest failed, errorcode 0x%08x\r\n", GetLastError());
    WinHttpCloseHandle(hSession);
    WinHttpCloseHandle(hConnect);
    return FALSE;
}

if (bResults)
{
    bResults = WinHttpReceiveResponse( hRequest, NULL);
}
else
{
    printf("hResults failed, errorcode 0x%08x\r\n");
    WinHttpCloseHandle(hRequest);
    WinHttpCloseHandle(hConnect);
    WinHttpCloseHandle(hSession);
    free(pBuf);
    free(pFormHeader);
    free(pFinalBoundary);
    free(pContentHeader);
    return FALSE;
}

WinHttpQueryDataAvailable(hRequest, &dwBytesToRead);

pResponse = (PUCHAR)calloc(dwBytesToRead, 1);

WinHttpReadData(hRequest, pResponse, dwBytesToRead, &dwBytesRead);

free(pResponse);
free(pFormHeader);
free(pFinalBoundary);
free(pContentHeader);
// Close any open handles.
if (hRequest) WinHttpCloseHandle(hRequest);
if (hConnect) WinHttpCloseHandle(hConnect);
if (hSession) WinHttpCloseHandle(hSession);

free(pBuf);
CloseHandle(hFile);

DeleteFile(wszFilePath);

return TRUE;

关于c++ - 上传文件时的 WinHTTP 问题,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/21023548/

相关文章:

c++ - malloc如何分配内存以及Heap的大小是多少?

c++ - concurrent_vector 无效数据

windows - 控制 WriteProfileString

c++ - (Visual-)字符串文字的 C++ 模板类型推断 - VS 2010 与 VS 2017

c++ - 允许在没有 Active Directory 信任的情况下进行 winhttp 委派

ssl - 使用 WinHTTP 通过 CONNECT 和 POST 发送用户代理?

c - 使用 WinHTTP (C++) 获取请求头

请求从 C++ 到非标量类型的转换

c++ - 为什么在Excel VBA中调用exe不输出文件?

c++ - 无法从 AES-128-GCM 获得正确的输出