winhttp: Store the context passed to WinHttpSendRequest.
[wine] / dlls / winhttp / tests / winhttp.c
1 /*
2  * WinHTTP - tests
3  *
4  * Copyright 2008 Google (Zac Brown)
5  *
6  * This library is free software; you can redistribute it and/or
7  * modify it under the terms of the GNU Lesser General Public
8  * License as published by the Free Software Foundation; either
9  * version 2.1 of the License, or (at your option) any later version.
10  *
11  * This library is distributed in the hope that it will be useful,
12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
14  * Lesser General Public License for more details.
15  *
16  * You should have received a copy of the GNU Lesser General Public
17  * License along with this library; if not, write to the Free Software
18  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
19  */
20
21 #include <stdarg.h>
22 #include <stdlib.h>
23 #include <windef.h>
24 #include <winbase.h>
25 #include <winhttp.h>
26 #include <wincrypt.h>
27 #include <winreg.h>
28
29 #include "wine/test.h"
30
31 static const WCHAR test_useragent[] =
32     {'W','i','n','e',' ','R','e','g','r','e','s','s','i','o','n',' ','T','e','s','t',0};
33 static const WCHAR test_server[] = {'w','i','n','e','h','q','.','o','r','g',0};
34
35 static void test_QueryOption(void)
36 {
37     BOOL ret;
38     HINTERNET session, request, connection;
39     DWORD feature, size;
40
41     SetLastError(0xdeadbeef);
42     session = WinHttpOpen(test_useragent, 0, 0, 0, 0);
43     ok(session != NULL, "WinHttpOpen failed to open session, error %u\n", GetLastError());
44
45     SetLastError(0xdeadbeef);
46     ret = WinHttpQueryOption(session, WINHTTP_OPTION_REDIRECT_POLICY, NULL, NULL);
47     ok(!ret, "should fail to set redirect policy %u\n", GetLastError());
48     ok(GetLastError() == ERROR_INVALID_PARAMETER,
49        "expected ERROR_INVALID_PARAMETER, got %u\n", GetLastError());
50
51     size = 0xdeadbeef;
52     SetLastError(0xdeadbeef);
53     ret = WinHttpQueryOption(session, WINHTTP_OPTION_REDIRECT_POLICY, NULL, &size);
54     ok(!ret, "should fail to query option\n");
55     ok(GetLastError() == ERROR_INSUFFICIENT_BUFFER,
56        "expected ERROR_INSUFFICIENT_BUFFER, got %u\n", GetLastError());
57     ok(size == 4, "expected 4, got %u\n", size);
58
59     feature = 0xdeadbeef;
60     size = sizeof(feature) - 1;
61     SetLastError(0xdeadbeef);
62     ret = WinHttpQueryOption(session, WINHTTP_OPTION_REDIRECT_POLICY, &feature, &size);
63     ok(!ret, "should fail to query option\n");
64     ok(GetLastError() == ERROR_INSUFFICIENT_BUFFER,
65        "expected ERROR_INSUFFICIENT_BUFFER, got %u\n", GetLastError());
66     ok(size == 4, "expected 4, got %u\n", size);
67
68     feature = 0xdeadbeef;
69     size = sizeof(feature) + 1;
70     SetLastError(0xdeadbeef);
71     ret = WinHttpQueryOption(session, WINHTTP_OPTION_REDIRECT_POLICY, &feature, &size);
72     ok(ret, "failed to query option %u\n", GetLastError());
73     ok(size == sizeof(feature), "WinHttpQueryOption should set the size: %u\n", size);
74     ok(feature == WINHTTP_OPTION_REDIRECT_POLICY_DISALLOW_HTTPS_TO_HTTP,
75        "expected WINHTTP_OPTION_REDIRECT_POLICY_DISALLOW_HTTPS_TO_HTTP, got %#x\n", feature);
76
77     SetLastError(0xdeadbeef);
78     ret = WinHttpSetOption(session, WINHTTP_OPTION_REDIRECT_POLICY, NULL, sizeof(feature));
79     ok(!ret, "should fail to set redirect policy %u\n", GetLastError());
80     ok(GetLastError() == ERROR_INVALID_PARAMETER,
81        "expected ERROR_INVALID_PARAMETER, got %u\n", GetLastError());
82
83     feature = WINHTTP_OPTION_REDIRECT_POLICY_ALWAYS;
84     SetLastError(0xdeadbeef);
85     ret = WinHttpSetOption(session, WINHTTP_OPTION_REDIRECT_POLICY, &feature, sizeof(feature) - 1);
86     ok(!ret, "should fail to set redirect policy %u\n", GetLastError());
87     ok(GetLastError() == ERROR_INSUFFICIENT_BUFFER,
88        "expected ERROR_INSUFFICIENT_BUFFER, got %u\n", GetLastError());
89
90     feature = WINHTTP_OPTION_REDIRECT_POLICY_ALWAYS;
91     SetLastError(0xdeadbeef);
92     ret = WinHttpSetOption(session, WINHTTP_OPTION_REDIRECT_POLICY, &feature, sizeof(feature) + 1);
93     ok(!ret, "should fail to set redirect policy %u\n", GetLastError());
94     ok(GetLastError() == ERROR_INSUFFICIENT_BUFFER,
95        "expected ERROR_INSUFFICIENT_BUFFER, got %u\n", GetLastError());
96
97     feature = WINHTTP_OPTION_REDIRECT_POLICY_ALWAYS;
98     SetLastError(0xdeadbeef);
99     ret = WinHttpSetOption(session, WINHTTP_OPTION_REDIRECT_POLICY, &feature, sizeof(feature));
100     ok(ret, "failed to set redirect policy %u\n", GetLastError());
101
102     feature = 0xdeadbeef;
103     size = sizeof(feature);
104     SetLastError(0xdeadbeef);
105     ret = WinHttpQueryOption(session, WINHTTP_OPTION_REDIRECT_POLICY, &feature, &size);
106     ok(ret, "failed to query option %u\n", GetLastError());
107     ok(feature == WINHTTP_OPTION_REDIRECT_POLICY_ALWAYS,
108        "expected WINHTTP_OPTION_REDIRECT_POLICY_ALWAYS, got %#x\n", feature);
109
110     feature = WINHTTP_DISABLE_COOKIES;
111     SetLastError(0xdeadbeef);
112     ret = WinHttpSetOption(session, WINHTTP_OPTION_DISABLE_FEATURE, &feature, sizeof(feature));
113     ok(!ret, "should fail to set disable feature for a session\n");
114     ok(GetLastError() == ERROR_WINHTTP_INCORRECT_HANDLE_TYPE,
115        "expected ERROR_WINHTTP_INCORRECT_HANDLE_TYPE, got %u\n", GetLastError());
116
117     SetLastError(0xdeadbeef);
118     connection = WinHttpConnect(session, test_server, INTERNET_DEFAULT_HTTP_PORT, 0);
119     ok(connection != NULL, "WinHttpConnect failed to open a connection, error: %u\n", GetLastError());
120
121     feature = WINHTTP_DISABLE_COOKIES;
122     SetLastError(0xdeadbeef);
123     ret = WinHttpSetOption(connection, WINHTTP_OPTION_DISABLE_FEATURE, &feature, sizeof(feature));
124     ok(!ret, "should fail to set disable feature for a connection\n");
125     ok(GetLastError() == ERROR_WINHTTP_INCORRECT_HANDLE_TYPE,
126        "expected ERROR_WINHTTP_INCORRECT_HANDLE_TYPE, got %u\n", GetLastError());
127
128     SetLastError(0xdeadbeef);
129     request = WinHttpOpenRequest(connection, NULL, NULL, NULL, WINHTTP_NO_REFERER,
130                                  WINHTTP_DEFAULT_ACCEPT_TYPES, 0);
131     if (request == NULL && GetLastError() == ERROR_WINHTTP_NAME_NOT_RESOLVED)
132     {
133         skip("Network unreachable, skipping the test\n");
134         goto done;
135     }
136
137     feature = 0xdeadbeef;
138     size = sizeof(feature);
139     SetLastError(0xdeadbeef);
140     ret = WinHttpQueryOption(request, WINHTTP_OPTION_DISABLE_FEATURE, &feature, &size);
141     ok(!ret, "should fail to query disable feature for a request\n");
142     ok(GetLastError() == ERROR_INVALID_PARAMETER,
143        "expected ERROR_INVALID_PARAMETER, got %u\n", GetLastError());
144
145     feature = 0;
146     size = sizeof(feature);
147     SetLastError(0xdeadbeef);
148     ret = WinHttpSetOption(request, WINHTTP_OPTION_DISABLE_FEATURE, &feature, sizeof(feature));
149     ok(ret, "failed to set feature %u\n", GetLastError());
150
151     feature = 0xffffffff;
152     size = sizeof(feature);
153     SetLastError(0xdeadbeef);
154     ret = WinHttpSetOption(request, WINHTTP_OPTION_DISABLE_FEATURE, &feature, sizeof(feature));
155     ok(ret, "failed to set feature %u\n", GetLastError());
156
157     feature = WINHTTP_DISABLE_COOKIES;
158     size = sizeof(feature);
159     SetLastError(0xdeadbeef);
160     ret = WinHttpSetOption(request, WINHTTP_OPTION_DISABLE_FEATURE, &feature, sizeof(feature));
161     ok(ret, "failed to set feature %u\n", GetLastError());
162
163     size = 0;
164     SetLastError(0xdeadbeef);
165     ret = WinHttpQueryOption(request, WINHTTP_OPTION_DISABLE_FEATURE, NULL, &size);
166     ok(!ret, "should fail to query disable feature for a request\n");
167     ok(GetLastError() == ERROR_INVALID_PARAMETER,
168        "expected ERROR_INVALID_PARAMETER, got %u\n", GetLastError());
169
170     SetLastError(0xdeadbeef);
171     ret = WinHttpCloseHandle(request);
172     ok(ret, "WinHttpCloseHandle failed on closing request: %u\n", GetLastError());
173
174 done:
175     SetLastError(0xdeadbeef);
176     ret = WinHttpCloseHandle(connection);
177     ok(ret, "WinHttpCloseHandle failed on closing connection: %u\n", GetLastError());
178     SetLastError(0xdeadbeef);
179     ret = WinHttpCloseHandle(session);
180     ok(ret, "WinHttpCloseHandle failed on closing session: %u\n", GetLastError());
181 }
182
183 static void test_OpenRequest (void)
184 {
185     BOOL ret;
186     HINTERNET session, request, connection;
187
188     session = WinHttpOpen(test_useragent, WINHTTP_ACCESS_TYPE_DEFAULT_PROXY,
189         WINHTTP_NO_PROXY_NAME, WINHTTP_NO_PROXY_BYPASS, 0);
190     ok(session != NULL, "WinHttpOpen failed to open session.\n");
191
192     /* Test with a bad server name */
193     SetLastError(0xdeadbeef);
194     connection = WinHttpConnect(session, NULL, INTERNET_DEFAULT_HTTP_PORT, 0);
195     ok (connection == NULL, "WinHttpConnect succeeded in opening connection to NULL server argument.\n");
196     ok(GetLastError() == ERROR_INVALID_PARAMETER, "Expected ERROR_INVALID_PARAMETER, got %u.\n", GetLastError());
197
198     /* Test with a valid server name */
199     connection = WinHttpConnect (session, test_server, INTERNET_DEFAULT_HTTP_PORT, 0);
200     ok(connection != NULL, "WinHttpConnect failed to open a connection, error: %u.\n", GetLastError());
201
202     request = WinHttpOpenRequest(connection, NULL, NULL, NULL, WINHTTP_NO_REFERER,
203         WINHTTP_DEFAULT_ACCEPT_TYPES, 0);
204     if (request == NULL && GetLastError() == ERROR_WINHTTP_NAME_NOT_RESOLVED)
205     {
206         skip("Network unreachable, skipping.\n");
207         goto done;
208     }
209     ok(request != NULL, "WinHttpOpenrequest failed to open a request, error: %u.\n", GetLastError());
210
211     ret = WinHttpSendRequest(request, WINHTTP_NO_ADDITIONAL_HEADERS, 0, NULL, 0, 0, 0);
212     ok(ret == TRUE, "WinHttpSendRequest failed: %u\n", GetLastError());
213     ret = WinHttpCloseHandle(request);
214     ok(ret == TRUE, "WinHttpCloseHandle failed on closing request, got %d.\n", ret);
215
216  done:
217     ret = WinHttpCloseHandle(connection);
218     ok(ret == TRUE, "WinHttpCloseHandle failed on closing connection, got %d.\n", ret);
219     ret = WinHttpCloseHandle(session);
220     ok(ret == TRUE, "WinHttpCloseHandle failed on closing session, got %d.\n", ret);
221
222 }
223
224 static void test_empty_headers_param(void)
225 {
226     static const WCHAR winehq[] = {'w','i','n','e','h','q','.','o','r','g',0};
227     static const WCHAR empty[]  = {0};
228     HANDLE ses, con, req;
229     BOOL ret;
230
231     ses = WinHttpOpen(test_useragent, 0, NULL, NULL, 0);
232     ok(ses != NULL, "failed to open session %u\n", GetLastError());
233
234     con = WinHttpConnect(ses, winehq, 80, 0);
235     ok(con != NULL, "failed to open a connection %u\n", GetLastError());
236
237     req = WinHttpOpenRequest(con, NULL, NULL, NULL, NULL, NULL, 0);
238     ok(req != NULL, "failed to open a request %u\n", GetLastError());
239
240     ret = WinHttpSendRequest(req, empty, 0, NULL, 0, 0, 0);
241     ok(ret, "failed to send request %u\n", GetLastError());
242
243     WinHttpCloseHandle(req);
244     WinHttpCloseHandle(con);
245     WinHttpCloseHandle(ses);
246 }
247
248 static void test_SendRequest (void)
249 {
250     HINTERNET session, request, connection;
251     DWORD header_len, optional_len, total_len, bytes_rw, size;
252     DWORD_PTR context;
253     BOOL ret;
254     CHAR buffer[256];
255     int i;
256
257     static const WCHAR test_site[] = {'c','r','o','s','s','o','v','e','r','.',
258                                 'c','o','d','e','w','e','a','v','e','r','s','.','c','o','m',0};
259     static const WCHAR content_type[] =
260         {'C','o','n','t','e','n','t','-','T','y','p','e',':',' ','a','p','p','l','i','c','a','t','i','o','n',
261          '/','x','-','w','w','w','-','f','o','r','m','-','u','r','l','e','n','c','o','d','e','d',0};
262     static const WCHAR test_file[] = {'/','p','o','s','t','t','e','s','t','.','p','h','p',0};
263     static const WCHAR test_verb[] = {'P','O','S','T',0};
264     static CHAR post_data[] = "mode=Test";
265     static CHAR test_post[] = "mode => Test\\0\n";
266
267     header_len = -1L;
268     total_len = optional_len = sizeof(post_data);
269     memset(buffer, 0xff, sizeof(buffer));
270
271     session = WinHttpOpen(test_useragent, WINHTTP_ACCESS_TYPE_DEFAULT_PROXY,
272         WINHTTP_NO_PROXY_NAME, WINHTTP_NO_PROXY_BYPASS, 0);
273     ok(session != NULL, "WinHttpOpen failed to open session.\n");
274
275     connection = WinHttpConnect (session, test_site, INTERNET_DEFAULT_HTTP_PORT, 0);
276     ok(connection != NULL, "WinHttpConnect failed to open a connection, error: %u.\n", GetLastError());
277
278     request = WinHttpOpenRequest(connection, test_verb, test_file, NULL, WINHTTP_NO_REFERER,
279         WINHTTP_DEFAULT_ACCEPT_TYPES, WINHTTP_FLAG_BYPASS_PROXY_CACHE);
280     if (request == NULL && GetLastError() == ERROR_WINHTTP_NAME_NOT_RESOLVED)
281     {
282         skip("Network unreachable, skipping.\n");
283         goto done;
284     }
285     ok(request != NULL, "WinHttpOpenrequest failed to open a request, error: %u.\n", GetLastError());
286
287     context = 0xdeadbeef;
288     ret = WinHttpSetOption(request, WINHTTP_OPTION_CONTEXT_VALUE, &context, sizeof(context));
289     ok(ret, "WinHttpSetOption failed: %u\n", GetLastError());
290
291     context++;
292     ret = WinHttpSendRequest(request, content_type, header_len, post_data, optional_len, total_len, context);
293     ok(ret == TRUE, "WinHttpSendRequest failed: %u\n", GetLastError());
294
295     context = 0;
296     size = sizeof(context);
297     ret = WinHttpQueryOption(request, WINHTTP_OPTION_CONTEXT_VALUE, &context, &size);
298     ok(ret, "WinHttpQueryOption failed: %u\n", GetLastError());
299     ok(context == 0xdeadbef0, "expected 0xdeadbef0, got %lx\n", context);
300
301     for (i = 3; post_data[i]; i++)
302     {
303         bytes_rw = -1;
304         ret = WinHttpWriteData(request, &post_data[i], 1, &bytes_rw);
305         if (ret)
306           ok(bytes_rw == 1, "WinHttpWriteData failed, wrote %u bytes instead of 1 byte.\n", bytes_rw);
307         else /* Since we already passed all optional data in WinHttpSendRequest Win7 fails our WinHttpWriteData call */
308         {
309           ok(GetLastError() == ERROR_INVALID_PARAMETER, "Expected ERROR_INVALID_PARAMETER got %u.\n", GetLastError());
310           ok(bytes_rw == -1, "Expected bytes_rw to remain unchanged.\n");
311         }
312     }
313
314     ret = WinHttpReceiveResponse(request, NULL);
315     ok(ret == TRUE, "WinHttpReceiveResponse failed: %u.\n", GetLastError());
316
317     bytes_rw = -1;
318     ret = WinHttpReadData(request, buffer, sizeof(buffer) - 1, &bytes_rw);
319     ok(ret == TRUE, "WinHttpReadData failed: %u.\n", GetLastError());
320
321     ok(bytes_rw == strlen(test_post), "Read %u bytes instead of %d.\n", bytes_rw, lstrlen(test_post));
322     ok(strncmp(buffer, test_post, bytes_rw) == 0, "Data read did not match, got '%s'.\n", buffer);
323
324     ret = WinHttpCloseHandle(request);
325     ok(ret == TRUE, "WinHttpCloseHandle failed on closing request, got %d.\n", ret);
326  done:
327     ret = WinHttpCloseHandle(connection);
328     ok(ret == TRUE, "WinHttpCloseHandle failed on closing connection, got %d.\n", ret);
329     ret = WinHttpCloseHandle(session);
330     ok(ret == TRUE, "WinHttpCloseHandle failed on closing session, got %d.\n", ret);
331 }
332
333 static void test_WinHttpTimeFromSystemTime(void)
334 {
335     BOOL ret;
336     static const SYSTEMTIME time = {2008, 7, 1, 28, 10, 5, 52, 0};
337     static const WCHAR expected_string[] =
338         {'M','o','n',',',' ','2','8',' ','J','u','l',' ','2','0','0','8',' ',
339          '1','0',':','0','5',':','5','2',' ','G','M','T',0};
340     WCHAR time_string[WINHTTP_TIME_FORMAT_BUFSIZE+1];
341
342     ret = WinHttpTimeFromSystemTime(&time, time_string);
343     ok(ret == TRUE, "WinHttpTimeFromSystemTime failed: %u\n", GetLastError());
344     ok(memcmp(time_string, expected_string, sizeof(expected_string)) == 0,
345         "Time string returned did not match expected time string.\n");
346 }
347
348 static void test_WinHttpTimeToSystemTime(void)
349 {
350     BOOL ret;
351     SYSTEMTIME time;
352     static const SYSTEMTIME expected_time = {2008, 7, 1, 28, 10, 5, 52, 0};
353     static const WCHAR time_string1[] =
354         {'M','o','n',',',' ','2','8',' ','J','u','l',' ','2','0','0','8',' ',
355          +          '1','0',':','0','5',':','5','2',' ','G','M','T','\n',0};
356     static const WCHAR time_string2[] =
357         {' ','m','o','n',' ','2','8',' ','j','u','l',' ','2','0','0','8',' ',
358          '1','0',' ','0','5',' ','5','2','\n',0};
359
360     ret = WinHttpTimeToSystemTime(time_string1, &time);
361     ok(ret == TRUE, "WinHttpTimeToSystemTime failed: %u\n", GetLastError());
362     ok(memcmp(&time, &expected_time, sizeof(SYSTEMTIME)) == 0,
363         "Returned SYSTEMTIME structure did not match expected SYSTEMTIME structure.\n");
364
365     ret = WinHttpTimeToSystemTime(time_string2, &time);
366     ok(ret == TRUE, "WinHttpTimeToSystemTime failed: %u\n", GetLastError());
367     ok(memcmp(&time, &expected_time, sizeof(SYSTEMTIME)) == 0,
368         "Returned SYSTEMTIME structure did not match expected SYSTEMTIME structure.\n");
369 }
370
371 static void test_WinHttpAddHeaders(void)
372 {
373     HINTERNET session, request, connection;
374     BOOL ret, reverse;
375     WCHAR buffer[MAX_PATH];
376     WCHAR check_buffer[MAX_PATH];
377     DWORD index, len, oldlen;
378
379     static const WCHAR test_site[] = {'c','r','o','s','s','o','v','e','r','.',
380                                 'c','o','d','e','w','e','a','v','e','r','s','.','c','o','m',0};
381     static const WCHAR test_file[] = {'/','p','o','s','t','t','e','s','t','.','p','h','p',0};
382     static const WCHAR test_verb[] = {'P','O','S','T',0};
383
384     static const WCHAR test_header_begin[] =
385         {'P','O','S','T',' ','/','p','o','s','t','t','e','s','t','.','p','h','p',' ','H','T','T','P','/','1'};
386     static const WCHAR full_path_test_header_begin[] =
387         {'P','O','S','T',' ','h','t','t','p',':','/','/','c','r','o','s','s','o','v','e','r','.','c','o','d','e','w','e','a','v','e','r','s','.','c','o','m',':','8','0','/','p','o','s','t','t','e','s','t','.','p','h','p',' ','H','T','T','P','/','1'};
388     static const WCHAR test_header_end[] = {'\r','\n','\r','\n',0};
389     static const WCHAR test_header_name[] = {'W','a','r','n','i','n','g',0};
390
391     static const WCHAR test_flag_coalesce[] = {'t','e','s','t','2',',',' ','t','e','s','t','4',0};
392     static const WCHAR test_flag_coalesce_reverse[] = {'t','e','s','t','3',',',' ','t','e','s','t','4',0};
393     static const WCHAR test_flag_coalesce_comma[] =
394         {'t','e','s','t','2',',',' ','t','e','s','t','4',',',' ','t','e','s','t','5',0};
395     static const WCHAR test_flag_coalesce_comma_reverse[] =
396         {'t','e','s','t','3',',',' ','t','e','s','t','4',',',' ','t','e','s','t','5',0};
397     static const WCHAR test_flag_coalesce_semicolon[] =
398         {'t','e','s','t','2',',',' ','t','e','s','t','4',',',' ','t','e','s','t','5',';',' ','t','e','s','t','6',0};
399     static const WCHAR test_flag_coalesce_semicolon_reverse[] =
400         {'t','e','s','t','3',',',' ','t','e','s','t','4',',',' ','t','e','s','t','5',';',' ','t','e','s','t','6',0};
401
402     static const WCHAR field[] = {'f','i','e','l','d',0};
403     static const WCHAR value[] = {'v','a','l','u','e',' ',0};
404     static const WCHAR value_nospace[] = {'v','a','l','u','e',0};
405
406     static const WCHAR test_headers[][14] =
407         {
408             {'W','a','r','n','i','n','g',':','t','e','s','t','1',0},
409             {'W','a','r','n','i','n','g',':','t','e','s','t','2',0},
410             {'W','a','r','n','i','n','g',':','t','e','s','t','3',0},
411             {'W','a','r','n','i','n','g',':','t','e','s','t','4',0},
412             {'W','a','r','n','i','n','g',':','t','e','s','t','5',0},
413             {'W','a','r','n','i','n','g',':','t','e','s','t','6',0},
414             {'W','a','r','n','i','n','g',':','t','e','s','t','7',0},
415             {0},
416             {':',0},
417             {'a',':',0},
418             {':','b',0},
419             {'c','d',0},
420             {' ','e',' ',':','f',0},
421             {'f','i','e','l','d',':',' ','v','a','l','u','e',' ',0}
422         };
423     static const WCHAR test_indices[][6] =
424         {
425             {'t','e','s','t','1',0},
426             {'t','e','s','t','2',0},
427             {'t','e','s','t','3',0},
428             {'t','e','s','t','4',0}
429         };
430
431     session = WinHttpOpen(test_useragent, WINHTTP_ACCESS_TYPE_DEFAULT_PROXY,
432         WINHTTP_NO_PROXY_NAME, WINHTTP_NO_PROXY_BYPASS, 0);
433     ok(session != NULL, "WinHttpOpen failed to open session.\n");
434
435     connection = WinHttpConnect (session, test_site, INTERNET_DEFAULT_HTTP_PORT, 0);
436     ok(connection != NULL, "WinHttpConnect failed to open a connection, error: %u.\n", GetLastError());
437
438     request = WinHttpOpenRequest(connection, test_verb, test_file, NULL, WINHTTP_NO_REFERER,
439         WINHTTP_DEFAULT_ACCEPT_TYPES, 0);
440     if (request == NULL && GetLastError() == ERROR_WINHTTP_NAME_NOT_RESOLVED)
441     {
442         skip("Network unreachable, skipping.\n");
443         goto done;
444     }
445     ok(request != NULL, "WinHttpOpenRequest failed to open a request, error: %u.\n", GetLastError());
446
447     index = 0;
448     len = sizeof(buffer);
449     ret = WinHttpQueryHeaders(request, WINHTTP_QUERY_CUSTOM | WINHTTP_QUERY_FLAG_REQUEST_HEADERS,
450         test_header_name, buffer, &len, &index);
451     ok(ret == FALSE, "WinHttpQueryHeaders unexpectedly succeeded, found 'Warning' header.\n");
452     ret = WinHttpAddRequestHeaders(request, test_headers[0], -1L, WINHTTP_ADDREQ_FLAG_ADD);
453     ok(ret == TRUE, "WinHttpAddRequestHeader failed to add new header, got %d with error %u.\n", ret, GetLastError());
454
455     index = 0;
456     len = sizeof(buffer);
457     ret = WinHttpQueryHeaders(request, WINHTTP_QUERY_CUSTOM | WINHTTP_QUERY_FLAG_REQUEST_HEADERS,
458         test_header_name, buffer, &len, &index);
459     ok(ret == TRUE, "WinHttpQueryHeaders failed: %u\n", GetLastError());
460     ok(index == 1, "WinHttpQueryHeaders failed: header index not incremented\n");
461     ok(memcmp(buffer, test_indices[0], sizeof(test_indices[0])) == 0, "WinHttpQueryHeaders failed: incorrect string returned\n");
462     ok(len == 5*sizeof(WCHAR), "WinHttpQueryHeaders failed: invalid length returned, expected 5, got %d\n", len);
463
464     ret = WinHttpQueryHeaders(request, WINHTTP_QUERY_CUSTOM | WINHTTP_QUERY_FLAG_REQUEST_HEADERS,
465         test_header_name, buffer, &len, &index);
466     ok(ret == FALSE, "WinHttpQueryHeaders unexpectedly succeeded, second index should not exist.\n");
467
468     /* Try to fetch the header info with a buffer that's big enough to fit the
469      * string but not the NULL terminator.
470      */
471     index = 0;
472     len = 5*sizeof(WCHAR);
473     memset(check_buffer, 0xab, sizeof(check_buffer));
474     memcpy(buffer, check_buffer, sizeof(buffer));
475     ret = WinHttpQueryHeaders(request, WINHTTP_QUERY_CUSTOM | WINHTTP_QUERY_FLAG_REQUEST_HEADERS,
476         test_header_name, buffer, &len, &index);
477     ok(ret == FALSE, "WinHttpQueryHeaders unexpectedly succeeded with a buffer that's too small.\n");
478     ok(memcmp(buffer, check_buffer, sizeof(buffer)) == 0,
479             "WinHttpQueryHeaders failed, modified the buffer when it should not have.\n");
480     ok(len == 6*sizeof(WCHAR), "WinHttpQueryHeaders returned invalid length, expected 12, got %d\n", len);
481
482     /* Try with a NULL buffer */
483     index = 0;
484     len = sizeof(buffer);
485     SetLastError(0xdeadbeef);
486     ret = WinHttpQueryHeaders(request, WINHTTP_QUERY_RAW_HEADERS_CRLF | WINHTTP_QUERY_FLAG_REQUEST_HEADERS,
487         test_header_name, NULL, &len, &index);
488     ok(ret == FALSE, "WinHttpQueryHeaders unexpectedly succeeded.\n");
489     ok(GetLastError() == ERROR_INSUFFICIENT_BUFFER, "Expected ERROR_INSUFFICIENT_BUFFER, got %u\n", GetLastError());
490     ok(len > 40, "WinHttpQueryHeaders returned invalid length: expected greater than 40, got %d\n", len);
491     ok(index == 0, "WinHttpQueryHeaders incorrectly incremented header index.\n");
492
493     /* Try with a NULL buffer and a length that's too small */
494     index = 0;
495     len = 10;
496     SetLastError(0xdeadbeef);
497     ret = WinHttpQueryHeaders(request, WINHTTP_QUERY_RAW_HEADERS_CRLF | WINHTTP_QUERY_FLAG_REQUEST_HEADERS,
498         test_header_name, NULL, &len, &index);
499     ok(ret == FALSE, "WinHttpQueryHeaders unexpectedly succeeded.\n");
500     ok(GetLastError() == ERROR_INSUFFICIENT_BUFFER,
501         "WinHttpQueryHeaders set incorrect error: expected ERROR_INSUFFICENT_BUFFER, go %u\n", GetLastError());
502     ok(len > 40, "WinHttpQueryHeaders returned invalid length: expected greater than 40, got %d\n", len);
503     ok(index == 0, "WinHttpQueryHeaders incorrectly incremented header index.\n");
504
505     index = 0;
506     len = 0;
507     SetLastError(0xdeadbeef);
508     ret = WinHttpQueryHeaders(request, WINHTTP_QUERY_RAW_HEADERS_CRLF | WINHTTP_QUERY_FLAG_REQUEST_HEADERS,
509         test_header_name, NULL, &len, &index);
510     ok(ret == FALSE, "WinHttpQueryHeaders unexpectedly succeeded.\n");
511     ok(GetLastError() == ERROR_INSUFFICIENT_BUFFER,
512         "WinHttpQueryHeaders set incorrect error: expected ERROR_INSUFFICIENT_BUFFER, got %u\n", GetLastError());
513     ok(len > 40, "WinHttpQueryHeaders returned invalid length: expected greater than 40, got %d\n", len);
514     ok(index == 0, "WinHttpQueryHeaders failed: index was incremented.\n");
515
516     /* valid query */
517     oldlen = len;
518     index = 0;
519     len = sizeof(buffer);
520     memset(buffer, 0xff, sizeof(buffer));
521     ret = WinHttpQueryHeaders(request, WINHTTP_QUERY_RAW_HEADERS_CRLF | WINHTTP_QUERY_FLAG_REQUEST_HEADERS,
522         test_header_name, buffer, &len, &index);
523     ok(ret == TRUE, "WinHttpQueryHeaders failed: got %d\n", ret);
524     ok(len + sizeof(WCHAR) <= oldlen, "WinHttpQueryHeaders resulting length longer than advertized.\n");
525     ok((len < sizeof(buffer) - sizeof(WCHAR)) && buffer[len / sizeof(WCHAR)] == 0, "WinHttpQueryHeaders did not append NULL terminator\n");
526     ok(len == lstrlenW(buffer) * sizeof(WCHAR), "WinHttpQueryHeaders returned incorrect length.\n");
527     ok(memcmp(buffer, test_header_begin, sizeof(test_header_begin)) == 0 ||
528         memcmp(buffer, full_path_test_header_begin, sizeof(full_path_test_header_begin)) == 0,
529         "WinHttpQueryHeaders returned invalid beginning of header string.\n");
530     ok(memcmp(buffer + lstrlenW(buffer) - 4, test_header_end, sizeof(test_header_end)) == 0,
531         "WinHttpQueryHeaders returned invalid end of header string.\n");
532     ok(index == 0, "WinHttpQueryHeaders incremented header index.\n");
533
534     index = 0;
535     len = 0;
536     SetLastError(0xdeadbeef);
537     ret = WinHttpQueryHeaders(request, WINHTTP_QUERY_RAW_HEADERS | WINHTTP_QUERY_FLAG_REQUEST_HEADERS,
538         test_header_name, NULL, &len, &index);
539     ok(ret == FALSE, "WinHttpQueryHeaders unexpectedly succeeded.\n");
540     ok(GetLastError() == ERROR_INSUFFICIENT_BUFFER,
541         "WinHttpQueryHeaders set incorrect error: expected ERROR_INSUFFICIENT_BUFFER, got %u\n", GetLastError());
542     ok(len > 40, "WinHttpQueryHeaders returned invalid length: expected greater than 40, got %d\n", len);
543     ok(index == 0, "WinHttpQueryHeaders failed: index was incremented.\n");
544
545     oldlen = len;
546     index = 0;
547     len = sizeof(buffer);
548     memset(buffer, 0xff, sizeof(buffer));
549     ret = WinHttpQueryHeaders(request, WINHTTP_QUERY_RAW_HEADERS | WINHTTP_QUERY_FLAG_REQUEST_HEADERS,
550         test_header_name, buffer, &len, &index);
551     ok(ret == TRUE, "WinHttpQueryHeaders failed %u\n", GetLastError());
552     ok(len + sizeof(WCHAR) <= oldlen, "resulting length longer than advertized\n");
553     ok((len < sizeof(buffer) - sizeof(WCHAR)) && !buffer[len / sizeof(WCHAR)] && !buffer[len / sizeof(WCHAR) - 1],
554         "no double NULL terminator\n");
555     ok(memcmp(buffer, test_header_begin, sizeof(test_header_begin)) == 0 ||
556         memcmp(buffer, full_path_test_header_begin, sizeof(full_path_test_header_begin)) == 0,
557         "invalid beginning of header string.\n");
558     ok(index == 0, "header index was incremented\n");
559
560     /* tests for more indices */
561     ret = WinHttpAddRequestHeaders(request, test_headers[1], -1L, WINHTTP_ADDREQ_FLAG_ADD);
562     ok(ret == TRUE, "WinHttpAddRequestHeaders failed to add duplicate header: %d\n", ret);
563
564     index = 0;
565     len = sizeof(buffer);
566     ret = WinHttpQueryHeaders(request, WINHTTP_QUERY_CUSTOM | WINHTTP_QUERY_FLAG_REQUEST_HEADERS,
567         test_header_name, buffer, &len, &index);
568     ok(ret == TRUE, "WinHttpQueryHeaders failed: %u\n", GetLastError());
569     ok(index == 1, "WinHttpQueryHeaders failed to increment index.\n");
570     ok(memcmp(buffer, test_indices[0], sizeof(test_indices[0])) == 0, "WinHttpQueryHeaders returned incorrect string.\n");
571
572     len = sizeof(buffer);
573     ret = WinHttpQueryHeaders(request, WINHTTP_QUERY_CUSTOM | WINHTTP_QUERY_FLAG_REQUEST_HEADERS,
574         test_header_name, buffer, &len, &index);
575     ok(ret == TRUE, "WinHttpQueryHeaders failed: %u\n", GetLastError());
576     ok(index == 2, "WinHttpQueryHeaders failed to increment index.\n");
577     ok(memcmp(buffer, test_indices[1], sizeof(test_indices[1])) == 0, "WinHttpQueryHeaders returned incorrect string.\n");
578
579     ret = WinHttpAddRequestHeaders(request, test_headers[2], -1L, WINHTTP_ADDREQ_FLAG_REPLACE);
580     ok(ret == TRUE, "WinHttpAddRequestHeaders failed to add duplicate header.\n");
581
582     index = 0;
583     len = sizeof(buffer);
584     ret = WinHttpQueryHeaders(request, WINHTTP_QUERY_CUSTOM | WINHTTP_QUERY_FLAG_REQUEST_HEADERS,
585         test_header_name, buffer, &len, &index);
586     ok(ret == TRUE, "WinHttpQueryHeaders failed: %u\n", GetLastError());
587     ok(index == 1, "WinHttpQueryHeaders failed to increment index.\n");
588     reverse = (memcmp(buffer, test_indices[1], sizeof(test_indices[1])) != 0); /* Win7 returns values in reverse order of adding */
589     ok(memcmp(buffer, test_indices[reverse ? 2 : 1], sizeof(test_indices[reverse ? 2 : 1])) == 0, "WinHttpQueryHeaders returned incorrect string.\n");
590
591     len = sizeof(buffer);
592     ret = WinHttpQueryHeaders(request, WINHTTP_QUERY_CUSTOM | WINHTTP_QUERY_FLAG_REQUEST_HEADERS,
593         test_header_name, buffer, &len, &index);
594     ok(ret == TRUE, "WinHttpQueryHeaders failed: %u\n", GetLastError());
595     ok(index == 2, "WinHttpQueryHeaders failed to increment index.\n");
596     ok(memcmp(buffer, test_indices[reverse ? 1 : 2], sizeof(test_indices[reverse ? 1 : 2])) == 0, "WinHttpQueryHeaders returned incorrect string.\n");
597
598     /* add if new flag */
599     ret = WinHttpAddRequestHeaders(request, test_headers[3], -1L, WINHTTP_ADDREQ_FLAG_ADD_IF_NEW);
600     ok(ret == FALSE, "WinHttpAddRequestHeaders incorrectly replaced existing header.\n");
601
602     index = 0;
603     len = sizeof(buffer);
604     ret = WinHttpQueryHeaders(request, WINHTTP_QUERY_CUSTOM | WINHTTP_QUERY_FLAG_REQUEST_HEADERS,
605         test_header_name, buffer, &len, &index);
606     ok(ret == TRUE, "WinHttpQueryHeaders failed: %u\n", GetLastError());
607     ok(index == 1, "WinHttpQueryHeaders failed to increment index.\n");
608     ok(memcmp(buffer, test_indices[reverse ? 2 : 1], sizeof(test_indices[reverse ? 2 : 1])) == 0, "WinHttpQueryHeaders returned incorrect string.\n");
609
610     len = sizeof(buffer);
611     ret = WinHttpQueryHeaders(request, WINHTTP_QUERY_CUSTOM | WINHTTP_QUERY_FLAG_REQUEST_HEADERS,
612         test_header_name, buffer, &len, &index);
613     ok(ret == TRUE, "WinHttpQueryHeaders failed: %u\n", GetLastError());
614     ok(index == 2, "WinHttpQueryHeaders failed to increment index.\n");
615     ok(memcmp(buffer, test_indices[reverse ? 1 : 2], sizeof(test_indices[reverse ? 1 : 2])) == 0, "WinHttpQueryHeaders returned incorrect string.\n");
616
617     len = sizeof(buffer);
618     ret = WinHttpQueryHeaders(request, WINHTTP_QUERY_CUSTOM | WINHTTP_QUERY_FLAG_REQUEST_HEADERS,
619         test_header_name, buffer, &len, &index);
620     ok(ret == FALSE, "WinHttpQueryHeaders succeeded unexpectedly, found third header.\n");
621
622     /* coalesce flag */
623     ret = WinHttpAddRequestHeaders(request, test_headers[3], -1L, WINHTTP_ADDREQ_FLAG_COALESCE);
624     ok(ret == TRUE, "WinHttpAddRequestHeaders failed with flag WINHTTP_ADDREQ_FLAG_COALESCE.\n");
625
626     index = 0;
627     len = sizeof(buffer);
628     ret = WinHttpQueryHeaders(request, WINHTTP_QUERY_CUSTOM | WINHTTP_QUERY_FLAG_REQUEST_HEADERS,
629         test_header_name, buffer, &len, &index);
630     ok(ret == TRUE, "WinHttpQueryHeaders failed: %u\n", GetLastError());
631     ok(index == 1, "WinHttpQueryHeaders failed to increment index.\n");
632     ok(memcmp(buffer, reverse ? test_flag_coalesce_reverse : test_flag_coalesce, sizeof(reverse ? test_flag_coalesce_reverse : test_flag_coalesce)) == 0, "WinHttpQueryHeaders returned incorrect string.\n");
633
634     len = sizeof(buffer);
635     ret = WinHttpQueryHeaders(request, WINHTTP_QUERY_CUSTOM | WINHTTP_QUERY_FLAG_REQUEST_HEADERS,
636         test_header_name, buffer, &len, &index);
637     ok(ret == TRUE, "WinHttpQueryHeaders failed: %u\n", GetLastError());
638     ok(index == 2, "WinHttpQueryHeaders failed to increment index.\n");
639     ok(memcmp(buffer, test_indices[reverse ? 1 : 2], sizeof(test_indices[reverse ? 1 : 2])) == 0, "WinHttpQueryHeaders returned incorrect string.\n");
640
641     len = sizeof(buffer);
642     ret = WinHttpQueryHeaders(request, WINHTTP_QUERY_CUSTOM | WINHTTP_QUERY_FLAG_REQUEST_HEADERS,
643         test_header_name, buffer, &len, &index);
644     ok(ret == FALSE, "WinHttpQueryHeaders succeeded unexpectedly, found third header.\n");
645
646     /* coalesce with comma flag */
647     ret = WinHttpAddRequestHeaders(request, test_headers[4], -1L, WINHTTP_ADDREQ_FLAG_COALESCE_WITH_COMMA);
648     ok(ret == TRUE, "WinHttpAddRequestHeaders failed with flag WINHTTP_ADDREQ_FLAG_COALESCE_WITH_COMMA.\n");
649
650     index = 0;
651     len = sizeof(buffer);
652     ret = WinHttpQueryHeaders(request, WINHTTP_QUERY_CUSTOM | WINHTTP_QUERY_FLAG_REQUEST_HEADERS,
653         test_header_name, buffer, &len, &index);
654     ok(ret == TRUE, "WinHttpQueryHeaders failed: %u\n", GetLastError());
655     ok(index == 1, "WinHttpQueryHeaders failed to increment index.\n");
656     ok(memcmp(buffer, reverse ? test_flag_coalesce_comma_reverse : test_flag_coalesce_comma, sizeof(reverse ? test_flag_coalesce_comma_reverse : test_flag_coalesce_comma)) == 0,
657         "WinHttpQueryHeaders returned incorrect string.\n");
658
659     len = sizeof(buffer);
660     ret = WinHttpQueryHeaders(request, WINHTTP_QUERY_CUSTOM | WINHTTP_QUERY_FLAG_REQUEST_HEADERS,
661         test_header_name, buffer, &len, &index);
662     ok(ret == TRUE, "WinHttpQueryHeaders failed: %u\n", GetLastError());
663     ok(index == 2, "WinHttpQueryHeaders failed to increment index.\n");
664     ok(memcmp(buffer, test_indices[reverse ? 1 : 2], sizeof(test_indices[reverse ? 1 : 2])) == 0, "WinHttpQueryHeaders returned incorrect string.\n");
665
666     len = sizeof(buffer);
667     ret = WinHttpQueryHeaders(request, WINHTTP_QUERY_CUSTOM | WINHTTP_QUERY_FLAG_REQUEST_HEADERS,
668         test_header_name, buffer, &len, &index);
669     ok(ret == FALSE, "WinHttpQueryHeaders succeeded unexpectedly, found third header.\n");
670
671
672     /* coalesce with semicolon flag */
673     ret = WinHttpAddRequestHeaders(request, test_headers[5], -1L, WINHTTP_ADDREQ_FLAG_COALESCE_WITH_SEMICOLON);
674     ok(ret == TRUE, "WinHttpAddRequestHeaders failed with flag WINHTTP_ADDREQ_FLAG_COALESCE_WITH_SEMICOLON.\n");
675
676     index = 0;
677     len = sizeof(buffer);
678     ret = WinHttpQueryHeaders(request, WINHTTP_QUERY_CUSTOM | WINHTTP_QUERY_FLAG_REQUEST_HEADERS,
679         test_header_name, buffer, &len, &index);
680     ok(ret == TRUE, "WinHttpQueryHeaders failed: %u\n", GetLastError());
681     ok(index == 1, "WinHttpQueryHeaders failed to increment index.\n");
682     ok(memcmp(buffer, reverse ? test_flag_coalesce_semicolon_reverse : test_flag_coalesce_semicolon, sizeof(reverse ? test_flag_coalesce_semicolon_reverse : test_flag_coalesce_semicolon)) == 0,
683             "WinHttpQueryHeaders returned incorrect string.\n");
684
685     len = sizeof(buffer);
686     ret = WinHttpQueryHeaders(request, WINHTTP_QUERY_CUSTOM | WINHTTP_QUERY_FLAG_REQUEST_HEADERS,
687         test_header_name, buffer, &len, &index);
688     ok(ret == TRUE, "WinHttpQueryHeaders failed: %u\n", GetLastError());
689     ok(index == 2, "WinHttpQueryHeaders failed to increment index.\n");
690     ok(memcmp(buffer, test_indices[reverse ? 1 : 2], sizeof(test_indices[reverse ? 1 : 2])) == 0, "WinHttpQueryHeaders returned incorrect string.\n");
691
692     len = sizeof(buffer);
693     ret = WinHttpQueryHeaders(request, WINHTTP_QUERY_CUSTOM | WINHTTP_QUERY_FLAG_REQUEST_HEADERS,
694         test_header_name, buffer, &len, &index);
695     ok(ret == FALSE, "WinHttpQueryHeaders succeeded unexpectedly, found third header.\n");
696
697     /* add and replace flags */
698     ret = WinHttpAddRequestHeaders(request, test_headers[3], -1L, WINHTTP_ADDREQ_FLAG_ADD | WINHTTP_ADDREQ_FLAG_REPLACE);
699     ok(ret == TRUE, "WinHttpAddRequestHeaders failed with flag WINHTTP_ADDREQ_FLAG_ADD | WINHTTP_ADDREQ_FLAG_REPLACE.\n");
700
701     index = 0;
702     len = sizeof(buffer);
703     ret = WinHttpQueryHeaders(request, WINHTTP_QUERY_CUSTOM | WINHTTP_QUERY_FLAG_REQUEST_HEADERS,
704         test_header_name, buffer, &len, &index);
705     ok(ret == TRUE, "WinHttpQueryHeaders failed: %u\n", GetLastError());
706     ok(index == 1, "WinHttpQueryHeaders failed to increment index.\n");
707     ok(memcmp(buffer, test_indices[reverse ? 3 : 2], sizeof(test_indices[reverse ? 3 : 2])) == 0, "WinHttpQueryHeaders returned incorrect string.\n");
708
709     len = sizeof(buffer);
710     ret = WinHttpQueryHeaders(request, WINHTTP_QUERY_CUSTOM | WINHTTP_QUERY_FLAG_REQUEST_HEADERS,
711         test_header_name, buffer, &len, &index);
712     ok(ret == TRUE, "WinHttpQueryHeaders failed: %u\n", GetLastError());
713     ok(index == 2, "WinHttpQueryHeaders failed to increment index.\n");
714     ok(memcmp(buffer, test_indices[reverse ? 1 : 3], sizeof(test_indices[reverse ? 1 : 3])) == 0, "WinHttpQueryHeaders returned incorrect string.\n");
715
716     len = sizeof(buffer);
717     ret = WinHttpQueryHeaders(request, WINHTTP_QUERY_CUSTOM | WINHTTP_QUERY_FLAG_REQUEST_HEADERS,
718         test_header_name, buffer, &len, &index);
719     ok(ret == FALSE, "WinHttpQueryHeaders succeeded unexpectedly, found third header.\n");
720
721     ret = WinHttpAddRequestHeaders(request, test_headers[8], ~0u, WINHTTP_ADDREQ_FLAG_ADD);
722     ok(!ret, "WinHttpAddRequestHeaders failed\n");
723
724     ret = WinHttpAddRequestHeaders(request, test_headers[9], ~0u, WINHTTP_ADDREQ_FLAG_ADD);
725     ok(ret, "WinHttpAddRequestHeaders failed\n");
726
727     ret = WinHttpAddRequestHeaders(request, test_headers[10], ~0u, WINHTTP_ADDREQ_FLAG_ADD);
728     ok(!ret, "WinHttpAddRequestHeaders failed\n");
729
730     ret = WinHttpAddRequestHeaders(request, test_headers[11], ~0u, WINHTTP_ADDREQ_FLAG_ADD);
731     ok(!ret, "WinHttpAddRequestHeaders failed\n");
732
733     ret = WinHttpAddRequestHeaders(request, test_headers[12], ~0u, WINHTTP_ADDREQ_FLAG_ADD);
734     ok(!ret, "WinHttpAddRequestHeaders failed\n");
735
736     ret = WinHttpAddRequestHeaders(request, test_headers[13], ~0u, WINHTTP_ADDREQ_FLAG_ADD);
737     ok(ret, "WinHttpAddRequestHeaders failed\n");
738
739     index = 0;
740     buffer[0] = 0;
741     len = sizeof(buffer);
742     ret = WinHttpQueryHeaders(request, WINHTTP_QUERY_CUSTOM | WINHTTP_QUERY_FLAG_REQUEST_HEADERS,
743         field, buffer, &len, &index);
744     ok(ret, "WinHttpQueryHeaders failed: %u\n", GetLastError());
745     ok(!memcmp(buffer, value, sizeof(value)) || ! memcmp(buffer, value_nospace, sizeof(value_nospace)), "unexpected result\n");
746
747     ret = WinHttpCloseHandle(request);
748     ok(ret == TRUE, "WinHttpCloseHandle failed on closing request, got %d.\n", ret);
749  done:
750     ret = WinHttpCloseHandle(connection);
751     ok(ret == TRUE, "WinHttpCloseHandle failed on closing connection, got %d.\n", ret);
752     ret = WinHttpCloseHandle(session);
753     ok(ret == TRUE, "WinHttpCloseHandle failed on closing session, got %d.\n", ret);
754
755 }
756
757 static void test_secure_connection(void)
758 {
759     static const WCHAR google[] = {'w','w','w','.','g','o','o','g','l','e','.','c','o','m',0};
760
761     HANDLE ses, con, req;
762     DWORD size, status, policy, bitness;
763     BOOL ret;
764     CERT_CONTEXT *cert;
765
766     ses = WinHttpOpen(test_useragent, 0, NULL, NULL, 0);
767     ok(ses != NULL, "failed to open session %u\n", GetLastError());
768
769     policy = WINHTTP_OPTION_REDIRECT_POLICY_ALWAYS;
770     ret = WinHttpSetOption(ses, WINHTTP_OPTION_REDIRECT_POLICY, &policy, sizeof(policy));
771     ok(ret, "failed to set redirect policy %u\n", GetLastError());
772
773     con = WinHttpConnect(ses, google, 443, 0);
774     ok(con != NULL, "failed to open a connection %u\n", GetLastError());
775
776     /* try without setting WINHTTP_FLAG_SECURE */
777     req = WinHttpOpenRequest(con, NULL, NULL, NULL, NULL, NULL, 0);
778     ok(req != NULL, "failed to open a request %u\n", GetLastError());
779
780     ret = WinHttpSendRequest(req, NULL, 0, NULL, 0, 0, 0);
781     ok(ret, "failed to send request %u\n", GetLastError());
782
783     ret = WinHttpReceiveResponse(req, NULL);
784     ok(!ret, "succeeded unexpectedly\n");
785
786     size = 0;
787     ret = WinHttpQueryHeaders(req, WINHTTP_QUERY_RAW_HEADERS_CRLF, NULL, NULL, &size, NULL);
788     ok(!ret, "succeeded unexpectedly\n");
789
790     WinHttpCloseHandle(req);
791
792     req = WinHttpOpenRequest(con, NULL, NULL, NULL, NULL, NULL, WINHTTP_FLAG_SECURE);
793     ok(req != NULL, "failed to open a request %u\n", GetLastError());
794
795     ret = WinHttpSendRequest(req, NULL, 0, NULL, 0, 0, 0);
796     ok(ret, "failed to send request %u\n", GetLastError());
797     if (!ret)
798     {
799         skip("secure connection failed, skipping remaining secure tests\n");
800         goto cleanup;
801     }
802
803     size = sizeof(cert);
804     ret = WinHttpQueryOption(req, WINHTTP_OPTION_SERVER_CERT_CONTEXT, &cert, &size );
805     ok(ret, "failed to retrieve certificate context %u\n", GetLastError());
806     if (ret)
807         CertFreeCertificateContext(cert);
808
809     size = sizeof(bitness);
810     ret = WinHttpQueryOption(req, WINHTTP_OPTION_SECURITY_KEY_BITNESS, &bitness, &size );
811     ok(ret, "failed to retrieve key bitness %u\n", GetLastError());
812
813     ret = WinHttpReceiveResponse(req, NULL);
814     ok(ret, "failed to receive response %u\n", GetLastError());
815
816     size = sizeof(status);
817     ret = WinHttpQueryHeaders(req, WINHTTP_QUERY_STATUS_CODE | WINHTTP_QUERY_FLAG_NUMBER, NULL, &status, &size, NULL);
818     ok(ret, "failed unexpectedly %u\n", GetLastError());
819     ok(status == 200, "request failed unexpectedly %u\n", status);
820
821     size = 0;
822     ret = WinHttpQueryHeaders(req, WINHTTP_QUERY_RAW_HEADERS_CRLF, NULL, NULL, &size, NULL);
823     ok(!ret, "succeeded unexpectedly\n");
824
825 cleanup:
826     WinHttpCloseHandle(req);
827     WinHttpCloseHandle(con);
828     WinHttpCloseHandle(ses);
829 }
830
831 static void test_request_parameter_defaults(void)
832 {
833     static const WCHAR empty[] = {0};
834     static const WCHAR codeweavers[] = {'c','o','d','e','w','e','a','v','e','r','s','.','c','o','m',0};
835
836     HANDLE ses, con, req;
837     DWORD size, status, error;
838     WCHAR *version;
839     BOOL ret;
840
841     ses = WinHttpOpen(test_useragent, 0, NULL, NULL, 0);
842     ok(ses != NULL, "failed to open session %u\n", GetLastError());
843
844     con = WinHttpConnect(ses, codeweavers, 0, 0);
845     ok(con != NULL, "failed to open a connection %u\n", GetLastError());
846
847     req = WinHttpOpenRequest(con, NULL, NULL, NULL, NULL, NULL, 0);
848     ok(req != NULL, "failed to open a request %u\n", GetLastError());
849
850     ret = WinHttpSendRequest(req, NULL, 0, NULL, 0, 0, 0);
851     ok(ret, "failed to send request %u\n", GetLastError());
852
853     ret = WinHttpReceiveResponse(req, NULL);
854     ok(ret, "failed to receive response %u\n", GetLastError());
855
856     size = sizeof(status);
857     ret = WinHttpQueryHeaders(req, WINHTTP_QUERY_STATUS_CODE | WINHTTP_QUERY_FLAG_NUMBER, NULL, &status, &size, NULL);
858     ok(ret, "failed unexpectedly %u\n", GetLastError());
859     ok(status == 200, "request failed unexpectedly %u\n", status);
860
861     WinHttpCloseHandle(req);
862
863     req = WinHttpOpenRequest(con, empty, empty, empty, NULL, NULL, 0);
864     ok(req != NULL, "failed to open a request %u\n", GetLastError());
865
866     ret = WinHttpSendRequest(req, NULL, 0, NULL, 0, 0, 0);
867     ok(ret, "failed to send request %u\n", GetLastError());
868
869     ret = WinHttpReceiveResponse(req, NULL);
870     ok(ret, "failed to receive response %u\n", GetLastError());
871
872     size = 0;
873     SetLastError(0xdeadbeef);
874     ret = WinHttpQueryHeaders(req, WINHTTP_QUERY_VERSION, NULL, NULL, &size, NULL);
875     error = GetLastError();
876     ok(!ret, "succeeded unexpectedly\n");
877     ok(error == ERROR_INSUFFICIENT_BUFFER, "expected ERROR_INSUFFICIENT_BUFFER, got %u\n", error);
878
879     version = HeapAlloc(GetProcessHeap(), 0, size);
880     ret = WinHttpQueryHeaders(req, WINHTTP_QUERY_VERSION, NULL, version, &size, NULL);
881     ok(ret, "failed unexpectedly %u\n", GetLastError());
882     ok(lstrlenW(version) == size / sizeof(WCHAR), "unexpected size %u\n", size);
883     HeapFree(GetProcessHeap(), 0, version);
884
885     size = sizeof(status);
886     ret = WinHttpQueryHeaders(req, WINHTTP_QUERY_STATUS_CODE | WINHTTP_QUERY_FLAG_NUMBER, NULL, &status, &size, NULL);
887     ok(ret, "failed unexpectedly %u\n", GetLastError());
888     ok(status == 200, "request failed unexpectedly %u\n", status);
889
890     WinHttpCloseHandle(req);
891     WinHttpCloseHandle(con);
892     WinHttpCloseHandle(ses);
893 }
894
895 static const WCHAR Connections[] = {
896     'S','o','f','t','w','a','r','e','\\',
897     'M','i','c','r','o','s','o','f','t','\\',
898     'W','i','n','d','o','w','s','\\',
899     'C','u','r','r','e','n','t','V','e','r','s','i','o','n','\\',
900     'I','n','t','e','r','n','e','t',' ','S','e','t','t','i','n','g','s','\\',
901     'C','o','n','n','e','c','t','i','o','n','s',0 };
902 static const WCHAR WinHttpSettings[] = {
903     'W','i','n','H','t','t','p','S','e','t','t','i','n','g','s',0 };
904
905 static DWORD get_default_proxy_reg_value( BYTE *buf, DWORD len, DWORD *type )
906 {
907     LONG l;
908     HKEY key;
909     DWORD ret = 0;
910
911     l = RegOpenKeyExW( HKEY_LOCAL_MACHINE, Connections, 0, KEY_READ, &key );
912     if (!l)
913     {
914         DWORD size = 0;
915
916         l = RegQueryValueExW( key, WinHttpSettings, NULL, type, NULL, &size );
917         if (!l)
918         {
919             if (size <= len)
920                 l = RegQueryValueExW( key, WinHttpSettings, NULL, type, buf,
921                     &size );
922             if (!l)
923                 ret = size;
924         }
925         RegCloseKey( key );
926     }
927     return ret;
928 }
929
930 static void set_default_proxy_reg_value( BYTE *buf, DWORD len, DWORD type )
931 {
932     LONG l;
933     HKEY key;
934
935     l = RegCreateKeyExW( HKEY_LOCAL_MACHINE, Connections, 0, NULL, 0,
936         KEY_WRITE, NULL, &key, NULL );
937     if (!l)
938     {
939         if (len)
940             RegSetValueExW( key, WinHttpSettings, 0, type, buf, len );
941         else
942             RegDeleteValueW( key, WinHttpSettings );
943         RegCloseKey( key );
944     }
945 }
946
947 static void test_set_default_proxy_config(void)
948 {
949     static const WCHAR wideString[] = { 0x226f, 0x575b, 0 };
950     static const WCHAR normalString[] = { 'f','o','o',0 };
951     DWORD type, len;
952     BYTE *saved_proxy_settings = NULL;
953     WINHTTP_PROXY_INFO info;
954     BOOL ret;
955
956     /* FIXME: it would be simpler to read the current settings using
957      * WinHttpGetDefaultProxyConfiguration and save them using
958      * WinHttpSetDefaultProxyConfiguration, but they appear to have a bug.
959      *
960      * If a proxy is configured in the registry, e.g. via 'proxcfg -p "foo"',
961      * the access type reported by WinHttpGetDefaultProxyConfiguration is 1,
962      * WINHTTP_ACCESS_TYPE_NO_PROXY, whereas it should be
963      * WINHTTP_ACCESS_TYPE_NAMED_PROXY.
964      * If WinHttpSetDefaultProxyConfiguration is called with dwAccessType = 1,
965      * the lpszProxy and lpszProxyBypass values are ignored.
966      * Thus, if a proxy is set with proxycfg, then calling
967      * WinHttpGetDefaultProxyConfiguration followed by
968      * WinHttpSetDefaultProxyConfiguration results in the proxy settings
969      * getting deleted from the registry.
970      *
971      * Instead I read the current registry value and restore it directly.
972      */
973     len = get_default_proxy_reg_value( NULL, 0, &type );
974     if (len)
975     {
976         saved_proxy_settings = HeapAlloc( GetProcessHeap(), 0, len );
977         len = get_default_proxy_reg_value( saved_proxy_settings, len, &type );
978     }
979
980     if (0)
981     {
982         /* Crashes on Vista and higher */
983         SetLastError(0xdeadbeef);
984         ret = WinHttpSetDefaultProxyConfiguration(NULL);
985         ok(!ret && GetLastError() == ERROR_INVALID_PARAMETER,
986             "expected ERROR_INVALID_PARAMETER, got %d\n", GetLastError());
987     }
988
989     /* test with invalid access type */
990     info.dwAccessType = 0xdeadbeef;
991     info.lpszProxy = info.lpszProxyBypass = NULL;
992     SetLastError(0xdeadbeef);
993     ret = WinHttpSetDefaultProxyConfiguration(&info);
994     ok(!ret && GetLastError() == ERROR_INVALID_PARAMETER,
995         "expected ERROR_INVALID_PARAMETER, got %d\n", GetLastError());
996
997     /* at a minimum, the proxy server must be set */
998     info.dwAccessType = WINHTTP_ACCESS_TYPE_NAMED_PROXY;
999     info.lpszProxy = info.lpszProxyBypass = NULL;
1000     SetLastError(0xdeadbeef);
1001     ret = WinHttpSetDefaultProxyConfiguration(&info);
1002     ok(!ret && GetLastError() == ERROR_INVALID_PARAMETER,
1003         "expected ERROR_INVALID_PARAMETER, got %d\n", GetLastError());
1004     info.lpszProxyBypass = normalString;
1005     SetLastError(0xdeadbeef);
1006     ret = WinHttpSetDefaultProxyConfiguration(&info);
1007     ok(!ret && GetLastError() == ERROR_INVALID_PARAMETER,
1008         "expected ERROR_INVALID_PARAMETER, got %d\n", GetLastError());
1009
1010     /* the proxy server can't have wide characters */
1011     info.lpszProxy = wideString;
1012     SetLastError(0xdeadbeef);
1013     ret = WinHttpSetDefaultProxyConfiguration(&info);
1014     ok((!ret && GetLastError() == ERROR_INVALID_PARAMETER) ||
1015         broken(ret), /* Earlier winhttp versions on W2K/XP */
1016         "expected ERROR_INVALID_PARAMETER, got %d\n", GetLastError());
1017
1018     info.lpszProxy = normalString;
1019     SetLastError(0xdeadbeef);
1020     ret = WinHttpSetDefaultProxyConfiguration(&info);
1021     if (ret)
1022     {
1023         ok(ret, "always true\n");
1024         set_default_proxy_reg_value( saved_proxy_settings, len, type );
1025     }
1026     else if (GetLastError() == ERROR_ACCESS_DENIED)
1027         skip("couldn't set default proxy configuration: access denied\n");
1028     else
1029         ok(ret, "WinHttpSetDefaultProxyConfiguration failed: %d\n",
1030            GetLastError());
1031 }
1032
1033 START_TEST (winhttp)
1034 {
1035     test_OpenRequest();
1036     test_SendRequest();
1037     test_WinHttpTimeFromSystemTime();
1038     test_WinHttpTimeToSystemTime();
1039     test_WinHttpAddHeaders();
1040     test_secure_connection();
1041     test_request_parameter_defaults();
1042     test_QueryOption();
1043     test_set_default_proxy_config();
1044     test_empty_headers_param();
1045 }