winhttp: Implement WinHttpQueryHeaders(WINHTTP_QUERY_VERSION).

This commit is contained in:
Hans Leidekker 2009-10-29 11:12:45 +01:00 committed by Alexandre Julliard
parent 39b1dbed7a
commit 846cc2b157
2 changed files with 33 additions and 1 deletions

View File

@ -652,6 +652,24 @@ static BOOL query_headers( request_t *request, DWORD level, LPCWSTR name, LPVOID
if (request_only) heap_free( headers );
return ret;
}
case WINHTTP_QUERY_VERSION:
{
DWORD len = (strlenW( request->version ) + 1) * sizeof(WCHAR);
if (len > *buflen)
{
set_last_error( ERROR_INSUFFICIENT_BUFFER );
*buflen = len;
return FALSE;
}
else if (buffer)
{
strcpyW( buffer, request->version );
TRACE("returning string: %s\n", debugstr_w(buffer));
ret = TRUE;
}
*buflen = len - sizeof(WCHAR);
return ret;
}
default:
{
if (attr >= sizeof(attribute_table)/sizeof(attribute_table[0]) || !attribute_table[attr])

View File

@ -816,7 +816,8 @@ static void test_request_parameter_defaults(void)
static const WCHAR codeweavers[] = {'c','o','d','e','w','e','a','v','e','r','s','.','c','o','m',0};
HANDLE ses, con, req;
DWORD size, status;
DWORD size, status, error;
WCHAR *version;
BOOL ret;
ses = WinHttpOpen(test_useragent, 0, NULL, NULL, 0);
@ -850,6 +851,19 @@ static void test_request_parameter_defaults(void)
ret = WinHttpReceiveResponse(req, NULL);
ok(ret, "failed to receive response %u\n", GetLastError());
size = 0;
SetLastError(0xdeadbeef);
ret = WinHttpQueryHeaders(req, WINHTTP_QUERY_VERSION, NULL, NULL, &size, NULL);
error = GetLastError();
ok(!ret, "succeeded unexpectedly\n");
ok(error == ERROR_INSUFFICIENT_BUFFER, "expected ERROR_INSUFFICIENT_BUFFER, got %u\n", error);
version = HeapAlloc(GetProcessHeap(), 0, size);
ret = WinHttpQueryHeaders(req, WINHTTP_QUERY_VERSION, NULL, version, &size, NULL);
ok(ret, "failed unexpectedly %u\n", GetLastError());
ok(lstrlenW(version) == size / sizeof(WCHAR), "unexpected size %u\n", size);
HeapFree(GetProcessHeap(), 0, version);
size = sizeof(status);
ret = WinHttpQueryHeaders(req, WINHTTP_QUERY_STATUS_CODE | WINHTTP_QUERY_FLAG_NUMBER, NULL, &status, &size, NULL);
ok(ret, "failed unexpectedly %u\n", GetLastError());