winhttp: Set last error for invalid and unimplemented header queries.

Signed-off-by: Hans Leidekker <hans@codeweavers.com>
Signed-off-by: Alexandre Julliard <julliard@winehq.org>
This commit is contained in:
Hans Leidekker 2019-08-02 11:01:38 +02:00 committed by Alexandre Julliard
parent d6a80a4833
commit 788f588f0a
2 changed files with 15 additions and 1 deletions

View File

@ -746,9 +746,15 @@ static BOOL query_headers( struct request *request, DWORD level, const WCHAR *na
return ret;
default:
if (attr >= ARRAY_SIZE(attribute_table) || !attribute_table[attr])
if (attr >= ARRAY_SIZE(attribute_table))
{
SetLastError( ERROR_INVALID_PARAMETER );
return FALSE;
}
if (!attribute_table[attr])
{
FIXME("attribute %u not implemented\n", attr);
SetLastError( ERROR_WINHTTP_HEADER_NOT_FOUND );
return FALSE;
}
TRACE("attribute %s\n", debugstr_w(attribute_table[attr]));

View File

@ -456,6 +456,14 @@ static void test_WinHttpSendRequest (void)
"Expected ERROR_SUCCESS got %u.\n", GetLastError());
ok(ret == TRUE, "WinHttpReceiveResponse failed: %u.\n", GetLastError());
SetLastError(0xdeadbeef);
ret = WinHttpQueryHeaders(request, WINHTTP_QUERY_ORIG_URI, NULL, NULL, &len, NULL);
ok(!ret && GetLastError() == ERROR_WINHTTP_HEADER_NOT_FOUND, "got %u\n", GetLastError());
SetLastError(0xdeadbeef);
ret = WinHttpQueryHeaders(request, WINHTTP_QUERY_MAX + 1, NULL, NULL, &len, NULL);
ok(!ret && GetLastError() == ERROR_INVALID_PARAMETER, "got %u\n", GetLastError());
bytes_rw = -1;
ret = WinHttpReadData(request, buffer, sizeof(buffer) - 1, &bytes_rw);
ok(ret == TRUE, "WinHttpReadData failed: %u.\n", GetLastError());