winhttp: Implement some more options.
[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
27 #include "wine/test.h"
28
29 static const WCHAR test_useragent[] =
30     {'W','i','n','e',' ','R','e','g','r','e','s','s','i','o','n',' ','T','e','s','t',0};
31 static const WCHAR test_server[] = {'w','i','n','e','h','q','.','o','r','g',0};
32
33 static void test_OpenRequest (void)
34 {
35     BOOL ret;
36     HINTERNET session, request, connection;
37
38     session = WinHttpOpen(test_useragent, WINHTTP_ACCESS_TYPE_DEFAULT_PROXY,
39         WINHTTP_NO_PROXY_NAME, WINHTTP_NO_PROXY_BYPASS, 0);
40     ok(session != NULL, "WinHttpOpen failed to open session.\n");
41
42     /* Test with a bad server name */
43     SetLastError(0xdeadbeef);
44     connection = WinHttpConnect(session, NULL, INTERNET_DEFAULT_HTTP_PORT, 0);
45     ok (connection == NULL, "WinHttpConnect succeeded in opening connection to NULL server argument.\n");
46     ok(GetLastError() == ERROR_INVALID_PARAMETER, "Expected ERROR_INVALID_PARAMETER, got %u.\n", GetLastError());
47
48     /* Test with a valid server name */
49     connection = WinHttpConnect (session, test_server, INTERNET_DEFAULT_HTTP_PORT, 0);
50     ok(connection != NULL, "WinHttpConnect failed to open a connection, error: %u.\n", GetLastError());
51
52     request = WinHttpOpenRequest(connection, NULL, NULL, NULL, WINHTTP_NO_REFERER,
53         WINHTTP_DEFAULT_ACCEPT_TYPES, 0);
54     if (request == NULL && GetLastError() == ERROR_WINHTTP_NAME_NOT_RESOLVED)
55     {
56         skip("Network unreachable, skipping.\n");
57         goto done;
58     }
59     ok(request != NULL, "WinHttpOpenrequest failed to open a request, error: %u.\n", GetLastError());
60
61     ret = WinHttpSendRequest(request, WINHTTP_NO_ADDITIONAL_HEADERS, 0, NULL, 0, 0, 0);
62     ok(ret == TRUE, "WinHttpSendRequest failed: %u\n", GetLastError());
63     ret = WinHttpCloseHandle(request);
64     ok(ret == TRUE, "WinHttpCloseHandle failed on closing request, got %d.\n", ret);
65
66  done:
67     ret = WinHttpCloseHandle(connection);
68     ok(ret == TRUE, "WinHttpCloseHandle failed on closing connection, got %d.\n", ret);
69     ret = WinHttpCloseHandle(session);
70     ok(ret == TRUE, "WinHttpCloseHandle failed on closing session, got %d.\n", ret);
71
72 }
73
74 static void test_SendRequest (void)
75 {
76     HINTERNET session, request, connection;
77     DWORD header_len, optional_len, total_len;
78     DWORD bytes_rw;
79     BOOL ret;
80     CHAR buffer[256];
81     int i;
82
83     static const WCHAR test_site[] = {'c','r','o','s','s','o','v','e','r','.',
84                                 'c','o','d','e','w','e','a','v','e','r','s','.','c','o','m',0};
85     static const WCHAR content_type[] =
86         {'C','o','n','t','e','n','t','-','T','y','p','e',':',' ','a','p','p','l','i','c','a','t','i','o','n',
87          '/','x','-','w','w','w','-','f','o','r','m','-','u','r','l','e','n','c','o','d','e','d',0};
88     static const WCHAR test_file[] = {'/','p','o','s','t','t','e','s','t','.','p','h','p',0};
89     static const WCHAR test_verb[] = {'P','O','S','T',0};
90     static CHAR post_data[] = "mode=Test";
91     static CHAR test_post[] = "mode => Test\\0\n";
92
93     header_len = -1L;
94     total_len = optional_len = sizeof(post_data);
95     memset(buffer, 0xff, sizeof(buffer));
96
97     session = WinHttpOpen(test_useragent, WINHTTP_ACCESS_TYPE_DEFAULT_PROXY,
98         WINHTTP_NO_PROXY_NAME, WINHTTP_NO_PROXY_BYPASS, 0);
99     ok(session != NULL, "WinHttpOpen failed to open session.\n");
100
101     connection = WinHttpConnect (session, test_site, INTERNET_DEFAULT_HTTP_PORT, 0);
102     ok(connection != NULL, "WinHttpConnect failed to open a connection, error: %u.\n", GetLastError());
103
104     request = WinHttpOpenRequest(connection, test_verb, test_file, NULL, WINHTTP_NO_REFERER,
105         WINHTTP_DEFAULT_ACCEPT_TYPES, WINHTTP_FLAG_BYPASS_PROXY_CACHE);
106     if (request == NULL && GetLastError() == ERROR_WINHTTP_NAME_NOT_RESOLVED)
107     {
108         skip("Network unreachable, skipping.\n");
109         goto done;
110     }
111     ok(request != NULL, "WinHttpOpenrequest failed to open a request, error: %u.\n", GetLastError());
112
113     ret = WinHttpSendRequest(request, content_type, header_len, post_data, optional_len, total_len, 0);
114     ok(ret == TRUE, "WinHttpSendRequest failed: %u\n", GetLastError());
115
116     for (i = 3; post_data[i]; i++)
117     {
118         bytes_rw = -1;
119         ret = WinHttpWriteData(request, &post_data[i], 1, &bytes_rw);
120         ok(ret == TRUE, "WinHttpWriteData failed: %u.\n", GetLastError());
121         ok(bytes_rw == 1, "WinHttpWriteData failed, wrote %u bytes instead of 1 byte.\n", bytes_rw);
122     }
123
124     ret = WinHttpReceiveResponse(request, NULL);
125     ok(ret == TRUE, "WinHttpReceiveResponse failed: %u.\n", GetLastError());
126
127     bytes_rw = -1;
128     ret = WinHttpReadData(request, buffer, sizeof(buffer) - 1, &bytes_rw);
129     ok(ret == TRUE, "WinHttpReadData failed: %u.\n", GetLastError());
130
131     ok(bytes_rw == strlen(test_post), "Read %u bytes instead of %d.\n", bytes_rw, lstrlen(test_post));
132     ok(strncmp(buffer, test_post, bytes_rw) == 0, "Data read did not match, got '%s'.\n", buffer);
133
134     ret = WinHttpCloseHandle(request);
135     ok(ret == TRUE, "WinHttpCloseHandle failed on closing request, got %d.\n", ret);
136  done:
137     ret = WinHttpCloseHandle(connection);
138     ok(ret == TRUE, "WinHttpCloseHandle failed on closing connection, got %d.\n", ret);
139     ret = WinHttpCloseHandle(session);
140     ok(ret == TRUE, "WinHttpCloseHandle failed on closing session, got %d.\n", ret);
141 }
142
143 static void test_WinHttpTimeFromSystemTime(void)
144 {
145     BOOL ret;
146     static const SYSTEMTIME time = {2008, 7, 1, 28, 10, 5, 52, 0};
147     static const WCHAR expected_string[] =
148         {'M','o','n',',',' ','2','8',' ','J','u','l',' ','2','0','0','8',' ',
149          '1','0',':','0','5',':','5','2',' ','G','M','T',0};
150     WCHAR time_string[WINHTTP_TIME_FORMAT_BUFSIZE+1];
151
152     ret = WinHttpTimeFromSystemTime(&time, time_string);
153     ok(ret == TRUE, "WinHttpTimeFromSystemTime failed: %u\n", GetLastError());
154     ok(memcmp(time_string, expected_string, sizeof(expected_string)) == 0,
155         "Time string returned did not match expected time string.\n");
156 }
157
158 static void test_WinHttpTimeToSystemTime(void)
159 {
160     BOOL ret;
161     SYSTEMTIME time;
162     static const SYSTEMTIME expected_time = {2008, 7, 1, 28, 10, 5, 52, 0};
163     static const WCHAR time_string1[] =
164         {'M','o','n',',',' ','2','8',' ','J','u','l',' ','2','0','0','8',' ',
165          +          '1','0',':','0','5',':','5','2',' ','G','M','T','\n',0};
166     static const WCHAR time_string2[] =
167         {' ','m','o','n',' ','2','8',' ','j','u','l',' ','2','0','0','8',' ',
168          '1','0',' ','0','5',' ','5','2','\n',0};
169
170     ret = WinHttpTimeToSystemTime(time_string1, &time);
171     ok(ret == TRUE, "WinHttpTimeToSystemTime failed: %u\n", GetLastError());
172     ok(memcmp(&time, &expected_time, sizeof(SYSTEMTIME)) == 0,
173         "Returned SYSTEMTIME structure did not match expected SYSTEMTIME structure.\n");
174
175     ret = WinHttpTimeToSystemTime(time_string2, &time);
176     ok(ret == TRUE, "WinHttpTimeToSystemTime failed: %u\n", GetLastError());
177     ok(memcmp(&time, &expected_time, sizeof(SYSTEMTIME)) == 0,
178         "Returned SYSTEMTIME structure did not match expected SYSTEMTIME structure.\n");
179 }
180
181 static void test_WinHttpAddHeaders(void)
182 {
183     HINTERNET session, request, connection;
184     BOOL ret;
185     WCHAR buffer[MAX_PATH];
186     WCHAR check_buffer[MAX_PATH];
187     DWORD index, len, oldlen;
188
189     static const WCHAR test_site[] = {'c','r','o','s','s','o','v','e','r','.',
190                                 'c','o','d','e','w','e','a','v','e','r','s','.','c','o','m',0};
191     static const WCHAR test_file[] = {'/','p','o','s','t','t','e','s','t','.','p','h','p',0};
192     static const WCHAR test_verb[] = {'P','O','S','T',0};
193
194     static const WCHAR test_header_begin[] =
195         {'P','O','S','T',' ','/','p','o','s','t','t','e','s','t','.','p','h','p',' ','H','T','T','P','/','1'};
196     static const WCHAR test_header_end[] = {'\r','\n','\r','\n',0};
197     static const WCHAR test_header_name[] = {'W','a','r','n','i','n','g',0};
198
199     static const WCHAR test_flag_coalesce[] = {'t','e','s','t','2',',',' ','t','e','s','t','4',0};
200     static const WCHAR test_flag_coalesce_comma[] =
201         {'t','e','s','t','2',',',' ','t','e','s','t','4',',',' ','t','e','s','t','5',0};
202     static const WCHAR test_flag_coalesce_semicolon[] =
203         {'t','e','s','t','2',',',' ','t','e','s','t','4',',',' ','t','e','s','t','5',';',' ','t','e','s','t','6',0};
204
205     static const WCHAR field[] = {'f','i','e','l','d',0};
206     static const WCHAR value[] = {'v','a','l','u','e',' ',0};
207
208     static const WCHAR test_headers[][14] =
209         {
210             {'W','a','r','n','i','n','g',':','t','e','s','t','1',0},
211             {'W','a','r','n','i','n','g',':','t','e','s','t','2',0},
212             {'W','a','r','n','i','n','g',':','t','e','s','t','3',0},
213             {'W','a','r','n','i','n','g',':','t','e','s','t','4',0},
214             {'W','a','r','n','i','n','g',':','t','e','s','t','5',0},
215             {'W','a','r','n','i','n','g',':','t','e','s','t','6',0},
216             {'W','a','r','n','i','n','g',':','t','e','s','t','7',0},
217             {0},
218             {':',0},
219             {'a',':',0},
220             {':','b',0},
221             {'c','d',0},
222             {' ','e',' ',':','f',0},
223             {'f','i','e','l','d',':',' ','v','a','l','u','e',' ',0}
224         };
225     static const WCHAR test_indices[][6] =
226         {
227             {'t','e','s','t','1',0},
228             {'t','e','s','t','2',0},
229             {'t','e','s','t','3',0},
230             {'t','e','s','t','4',0}
231         };
232
233     session = WinHttpOpen(test_useragent, WINHTTP_ACCESS_TYPE_DEFAULT_PROXY,
234         WINHTTP_NO_PROXY_NAME, WINHTTP_NO_PROXY_BYPASS, 0);
235     ok(session != NULL, "WinHttpOpen failed to open session.\n");
236
237     connection = WinHttpConnect (session, test_site, INTERNET_DEFAULT_HTTP_PORT, 0);
238     ok(connection != NULL, "WinHttpConnect failed to open a connection, error: %u.\n", GetLastError());
239
240     request = WinHttpOpenRequest(connection, test_verb, test_file, NULL, WINHTTP_NO_REFERER,
241         WINHTTP_DEFAULT_ACCEPT_TYPES, 0);
242     if (request == NULL && GetLastError() == ERROR_WINHTTP_NAME_NOT_RESOLVED)
243     {
244         skip("Network unreachable, skipping.\n");
245         goto done;
246     }
247     ok(request != NULL, "WinHttpOpenRequest failed to open a request, error: %u.\n", GetLastError());
248
249     index = 0;
250     len = sizeof(buffer);
251     ret = WinHttpQueryHeaders(request, WINHTTP_QUERY_CUSTOM | WINHTTP_QUERY_FLAG_REQUEST_HEADERS,
252         test_header_name, buffer, &len, &index);
253     ok(ret == FALSE, "WinHttpQueryHeaders unexpectedly succeeded, found 'Warning' header.\n");
254     ret = WinHttpAddRequestHeaders(request, test_headers[0], -1L, WINHTTP_ADDREQ_FLAG_ADD);
255     ok(ret == TRUE, "WinHttpAddRequestHeader failed to add new header, got %d with error %u.\n", ret, GetLastError());
256
257     index = 0;
258     len = sizeof(buffer);
259     ret = WinHttpQueryHeaders(request, WINHTTP_QUERY_CUSTOM | WINHTTP_QUERY_FLAG_REQUEST_HEADERS,
260         test_header_name, buffer, &len, &index);
261     ok(ret == TRUE, "WinHttpQueryHeaders failed: %u\n", GetLastError());
262     ok(index == 1, "WinHttpQueryHeaders failed: header index not incremented\n");
263     ok(memcmp(buffer, test_indices[0], sizeof(test_indices[0])) == 0, "WinHttpQueryHeaders failed: incorrect string returned\n");
264     ok(len == 5*sizeof(WCHAR), "WinHttpQueryHeaders failed: invalid length returned, expected 5, got %d\n", len);
265
266     ret = WinHttpQueryHeaders(request, WINHTTP_QUERY_CUSTOM | WINHTTP_QUERY_FLAG_REQUEST_HEADERS,
267         test_header_name, buffer, &len, &index);
268     ok(ret == FALSE, "WinHttpQueryHeaders unexpectedly succeeded, second index should not exist.\n");
269
270     /* Try to fetch the header info with a buffer that's big enough to fit the
271      * string but not the NULL terminator.
272      */
273     index = 0;
274     len = 5*sizeof(WCHAR);
275     memcpy(buffer, check_buffer, sizeof(buffer));
276     ret = WinHttpQueryHeaders(request, WINHTTP_QUERY_CUSTOM | WINHTTP_QUERY_FLAG_REQUEST_HEADERS,
277         test_header_name, buffer, &len, &index);
278     ok(ret == FALSE, "WinHttpQueryHeaders unexpectedly succeeded with a buffer that's too small.\n");
279     ok(memcmp(buffer, check_buffer, sizeof(buffer)) == 0,
280             "WinHttpQueryHeaders failed, modified the buffer when it should not have.\n");
281     ok(len == 6*sizeof(WCHAR), "WinHttpQueryHeaders returned invalid length, expected 12, got %d\n", len);
282
283     /* Try with a NULL buffer */
284     index = 0;
285     len = sizeof(buffer);
286     SetLastError(0xdeadbeef);
287     ret = WinHttpQueryHeaders(request, WINHTTP_QUERY_RAW_HEADERS_CRLF | WINHTTP_QUERY_FLAG_REQUEST_HEADERS,
288         test_header_name, NULL, &len, &index);
289     ok(ret == FALSE, "WinHttpQueryHeaders unexpectedly succeeded.\n");
290     todo_wine ok(GetLastError() == ERROR_INSUFFICIENT_BUFFER, "Expected ERROR_INSUFFICIENT_BUFFER, got %u\n", GetLastError());
291     ok(len > 40, "WinHttpQueryHeaders returned invalid length: expected greater than 40, got %d\n", len);
292     ok(index == 0, "WinHttpQueryHeaders incorrectly incremented header index.\n");
293
294     /* Try with a NULL buffer and a length that's too small */
295     index = 0;
296     len = 10;
297     SetLastError(0xdeadbeef);
298     ret = WinHttpQueryHeaders(request, WINHTTP_QUERY_RAW_HEADERS_CRLF | WINHTTP_QUERY_FLAG_REQUEST_HEADERS,
299         test_header_name, NULL, &len, &index);
300     ok(ret == FALSE, "WinHttpQueryHeaders unexpectedly succeeded.\n");
301     ok(GetLastError() == ERROR_INSUFFICIENT_BUFFER,
302         "WinHttpQueryHeaders set incorrect error: expected ERROR_INSUFFICENT_BUFFER, go %u\n", GetLastError());
303     ok(len > 40, "WinHttpQueryHeaders returned invalid length: expected greater than 40, got %d\n", len);
304     ok(index == 0, "WinHttpQueryHeaders incorrectly incremented header index.\n");
305
306     index = 0;
307     len = 0;
308     SetLastError(0xdeadbeef);
309     ret = WinHttpQueryHeaders(request, WINHTTP_QUERY_RAW_HEADERS_CRLF | WINHTTP_QUERY_FLAG_REQUEST_HEADERS,
310         test_header_name, NULL, &len, &index);
311     ok(ret == FALSE, "WinHttpQueryHeaders unexpectedly succeeded.\n");
312     ok(GetLastError() == ERROR_INSUFFICIENT_BUFFER,
313         "WinHttpQueryHeaders set incorrect error: expected ERROR_INSUFFICIENT_BUFFER, got %u\n", GetLastError());
314     ok(len > 40, "WinHttpQueryHeaders returned invalid length: expected greater than 40, got %d\n", len);
315     ok(index == 0, "WinHttpQueryHeaders failed: index was incremented.\n");
316
317     /* valid query */
318     oldlen = len;
319     index = 0;
320     len = sizeof(buffer);
321     memset(buffer, 0xff, sizeof(buffer));
322     ret = WinHttpQueryHeaders(request, WINHTTP_QUERY_RAW_HEADERS_CRLF | WINHTTP_QUERY_FLAG_REQUEST_HEADERS,
323         test_header_name, buffer, &len, &index);
324     ok(ret == TRUE, "WinHttpQueryHeaders failed: got %d\n", ret);
325     ok(len + sizeof(WCHAR) <= oldlen, "WinHttpQueryHeaders resulting length longer than advertized.\n");
326     ok((len < sizeof(buffer) - sizeof(WCHAR)) && buffer[len / sizeof(WCHAR)] == 0, "WinHttpQueryHeaders did not append NULL terminator\n");
327     ok(len == lstrlenW(buffer) * sizeof(WCHAR), "WinHttpQueryHeaders returned incorrect length.\n");
328     ok(memcmp(buffer, test_header_begin, sizeof(test_header_begin)) == 0,
329         "WinHttpQueryHeaders returned invalid beginning of header string.\n");
330     ok(memcmp(buffer + lstrlenW(buffer) - 4, test_header_end, sizeof(test_header_end)) == 0,
331         "WinHttpQueryHeaders returned invalid end of header string.\n");
332     ok(index == 0, "WinHttpQueryHeaders incremented header index.\n");
333
334     /* tests for more indices */
335     ret = WinHttpAddRequestHeaders(request, test_headers[1], -1L, WINHTTP_ADDREQ_FLAG_ADD);
336     ok(ret == TRUE, "WinHttpAddRequestHeaders failed to add duplicate header: %d\n", ret);
337
338     index = 0;
339     len = sizeof(buffer);
340     ret = WinHttpQueryHeaders(request, WINHTTP_QUERY_CUSTOM | WINHTTP_QUERY_FLAG_REQUEST_HEADERS,
341         test_header_name, buffer, &len, &index);
342     ok(ret == TRUE, "WinHttpQueryHeaders failed: %u\n", GetLastError());
343     ok(index == 1, "WinHttpQueryHeaders failed to increment index.\n");
344     ok(memcmp(buffer, test_indices[0], sizeof(test_indices[0])) == 0, "WinHttpQueryHeaders returned incorrect string.\n");
345
346     len = sizeof(buffer);
347     ret = WinHttpQueryHeaders(request, WINHTTP_QUERY_CUSTOM | WINHTTP_QUERY_FLAG_REQUEST_HEADERS,
348         test_header_name, buffer, &len, &index);
349     ok(ret == TRUE, "WinHttpQueryHeaders failed: %u\n", GetLastError());
350     ok(index == 2, "WinHttpQueryHeaders failed to increment index.\n");
351     ok(memcmp(buffer, test_indices[1], sizeof(test_indices[1])) == 0, "WinHttpQueryHeaders returned incorrect string.\n");
352
353     ret = WinHttpAddRequestHeaders(request, test_headers[2], -1L, WINHTTP_ADDREQ_FLAG_REPLACE);
354     ok(ret == TRUE, "WinHttpAddRequestHeaders failed to add duplicate header.\n");
355
356     index = 0;
357     len = sizeof(buffer);
358     ret = WinHttpQueryHeaders(request, WINHTTP_QUERY_CUSTOM | WINHTTP_QUERY_FLAG_REQUEST_HEADERS,
359         test_header_name, buffer, &len, &index);
360     ok(ret == TRUE, "WinHttpQueryHeaders failed: %u\n", GetLastError());
361     ok(index == 1, "WinHttpQueryHeaders failed to increment index.\n");
362     ok(memcmp(buffer, test_indices[1], sizeof(test_indices[1])) == 0, "WinHttpQueryHeaders returned incorrect string.\n");
363
364     len = sizeof(buffer);
365     ret = WinHttpQueryHeaders(request, WINHTTP_QUERY_CUSTOM | WINHTTP_QUERY_FLAG_REQUEST_HEADERS,
366         test_header_name, buffer, &len, &index);
367     ok(ret == TRUE, "WinHttpQueryHeaders failed: %u\n", GetLastError());
368     ok(index == 2, "WinHttpQueryHeaders failed to increment index.\n");
369     ok(memcmp(buffer, test_indices[2], sizeof(test_indices[2])) == 0, "WinHttpQueryHeaders returned incorrect string.\n");
370
371     /* add if new flag */
372     ret = WinHttpAddRequestHeaders(request, test_headers[3], -1L, WINHTTP_ADDREQ_FLAG_ADD_IF_NEW);
373     ok(ret == FALSE, "WinHttpAddRequestHeaders incorrectly replaced existing header.\n");
374
375     index = 0;
376     len = sizeof(buffer);
377     ret = WinHttpQueryHeaders(request, WINHTTP_QUERY_CUSTOM | WINHTTP_QUERY_FLAG_REQUEST_HEADERS,
378         test_header_name, buffer, &len, &index);
379     ok(ret == TRUE, "WinHttpQueryHeaders failed: %u\n", GetLastError());
380     ok(index == 1, "WinHttpQueryHeaders failed to increment index.\n");
381     ok(memcmp(buffer, test_indices[1], sizeof(test_indices[1])) == 0, "WinHttpQueryHeaders returned incorrect string.\n");
382
383     len = sizeof(buffer);
384     ret = WinHttpQueryHeaders(request, WINHTTP_QUERY_CUSTOM | WINHTTP_QUERY_FLAG_REQUEST_HEADERS,
385         test_header_name, buffer, &len, &index);
386     ok(ret == TRUE, "WinHttpQueryHeaders failed: %u\n", GetLastError());
387     ok(index == 2, "WinHttpQueryHeaders failed to increment index.\n");
388     ok(memcmp(buffer, test_indices[2], sizeof(test_indices[2])) == 0, "WinHttpQueryHeaders returned incorrect string.\n");
389
390     len = sizeof(buffer);
391     ret = WinHttpQueryHeaders(request, WINHTTP_QUERY_CUSTOM | WINHTTP_QUERY_FLAG_REQUEST_HEADERS,
392         test_header_name, buffer, &len, &index);
393     ok(ret == FALSE, "WinHttpQueryHeaders succeeded unexpectedly, found third header.\n");
394
395     /* coalesce flag */
396     ret = WinHttpAddRequestHeaders(request, test_headers[3], -1L, WINHTTP_ADDREQ_FLAG_COALESCE);
397     ok(ret == TRUE, "WinHttpAddRequestHeaders failed with flag WINHTTP_ADDREQ_FLAG_COALESCE.\n");
398
399     index = 0;
400     len = sizeof(buffer);
401     ret = WinHttpQueryHeaders(request, WINHTTP_QUERY_CUSTOM | WINHTTP_QUERY_FLAG_REQUEST_HEADERS,
402         test_header_name, buffer, &len, &index);
403     ok(ret == TRUE, "WinHttpQueryHeaders failed: %u\n", GetLastError());
404     ok(index == 1, "WinHttpQueryHeaders failed to increment index.\n");
405     ok(memcmp(buffer, test_flag_coalesce, sizeof(test_flag_coalesce)) == 0, "WinHttpQueryHeaders returned incorrect string.\n");
406
407     len = sizeof(buffer);
408     ret = WinHttpQueryHeaders(request, WINHTTP_QUERY_CUSTOM | WINHTTP_QUERY_FLAG_REQUEST_HEADERS,
409         test_header_name, buffer, &len, &index);
410     ok(ret == TRUE, "WinHttpQueryHeaders failed: %u\n", GetLastError());
411     ok(index == 2, "WinHttpQueryHeaders failed to increment index.\n");
412     ok(memcmp(buffer, test_indices[2], sizeof(test_indices[2])) == 0, "WinHttpQueryHeaders returned incorrect string.\n");
413
414     len = sizeof(buffer);
415     ret = WinHttpQueryHeaders(request, WINHTTP_QUERY_CUSTOM | WINHTTP_QUERY_FLAG_REQUEST_HEADERS,
416         test_header_name, buffer, &len, &index);
417     ok(ret == FALSE, "WinHttpQueryHeaders succeeded unexpectedly, found third header.\n");
418
419     /* coalesce with comma flag */
420     ret = WinHttpAddRequestHeaders(request, test_headers[4], -1L, WINHTTP_ADDREQ_FLAG_COALESCE_WITH_COMMA);
421     ok(ret == TRUE, "WinHttpAddRequestHeaders failed with flag WINHTTP_ADDREQ_FLAG_COALESCE_WITH_COMMA.\n");
422
423     index = 0;
424     len = sizeof(buffer);
425     ret = WinHttpQueryHeaders(request, WINHTTP_QUERY_CUSTOM | WINHTTP_QUERY_FLAG_REQUEST_HEADERS,
426         test_header_name, buffer, &len, &index);
427     ok(ret == TRUE, "WinHttpQueryHeaders failed: %u\n", GetLastError());
428     ok(index == 1, "WinHttpQueryHeaders failed to increment index.\n");
429     ok(memcmp(buffer, test_flag_coalesce_comma, sizeof(test_flag_coalesce_comma)) == 0,
430         "WinHttpQueryHeaders returned incorrect string.\n");
431
432     len = sizeof(buffer);
433     ret = WinHttpQueryHeaders(request, WINHTTP_QUERY_CUSTOM | WINHTTP_QUERY_FLAG_REQUEST_HEADERS,
434         test_header_name, buffer, &len, &index);
435     ok(ret == TRUE, "WinHttpQueryHeaders failed: %u\n", GetLastError());
436     ok(index == 2, "WinHttpQueryHeaders failed to increment index.\n");
437     ok(memcmp(buffer, test_indices[2], sizeof(test_indices[2])) == 0, "WinHttpQueryHeaders returned incorrect string.\n");
438
439     len = sizeof(buffer);
440     ret = WinHttpQueryHeaders(request, WINHTTP_QUERY_CUSTOM | WINHTTP_QUERY_FLAG_REQUEST_HEADERS,
441         test_header_name, buffer, &len, &index);
442     ok(ret == FALSE, "WinHttpQueryHeaders succeeded unexpectedly, found third header.\n");
443
444
445     /* coalesce with semicolon flag */
446     ret = WinHttpAddRequestHeaders(request, test_headers[5], -1L, WINHTTP_ADDREQ_FLAG_COALESCE_WITH_SEMICOLON);
447     ok(ret == TRUE, "WinHttpAddRequestHeaders failed with flag WINHTTP_ADDREQ_FLAG_COALESCE_WITH_SEMICOLON.\n");
448
449     index = 0;
450     len = sizeof(buffer);
451     ret = WinHttpQueryHeaders(request, WINHTTP_QUERY_CUSTOM | WINHTTP_QUERY_FLAG_REQUEST_HEADERS,
452         test_header_name, buffer, &len, &index);
453     ok(ret == TRUE, "WinHttpQueryHeaders failed: %u\n", GetLastError());
454     ok(index == 1, "WinHttpQueryHeaders failed to increment index.\n");
455     ok(memcmp(buffer, test_flag_coalesce_semicolon, sizeof(test_flag_coalesce_semicolon)) == 0,
456             "WinHttpQueryHeaders returned incorrect string.\n");
457
458     len = sizeof(buffer);
459     ret = WinHttpQueryHeaders(request, WINHTTP_QUERY_CUSTOM | WINHTTP_QUERY_FLAG_REQUEST_HEADERS,
460         test_header_name, buffer, &len, &index);
461     ok(ret == TRUE, "WinHttpQueryHeaders failed: %u\n", GetLastError());
462     ok(index == 2, "WinHttpQueryHeaders failed to increment index.\n");
463     ok(memcmp(buffer, test_indices[2], sizeof(test_indices[2])) == 0, "WinHttpQueryHeaders returned incorrect string.\n");
464
465     len = sizeof(buffer);
466     ret = WinHttpQueryHeaders(request, WINHTTP_QUERY_CUSTOM | WINHTTP_QUERY_FLAG_REQUEST_HEADERS,
467         test_header_name, buffer, &len, &index);
468     ok(ret == FALSE, "WinHttpQueryHeaders succeeded unexpectedly, found third header.\n");
469
470     /* add and replace flags */
471     ret = WinHttpAddRequestHeaders(request, test_headers[3], -1L, WINHTTP_ADDREQ_FLAG_ADD | WINHTTP_ADDREQ_FLAG_REPLACE);
472     ok(ret == TRUE, "WinHttpAddRequestHeaders failed with flag WINHTTP_ADDREQ_FLAG_ADD | WINHTTP_ADDREQ_FLAG_REPLACE.\n");
473
474     index = 0;
475     len = sizeof(buffer);
476     ret = WinHttpQueryHeaders(request, WINHTTP_QUERY_CUSTOM | WINHTTP_QUERY_FLAG_REQUEST_HEADERS,
477         test_header_name, buffer, &len, &index);
478     ok(ret == TRUE, "WinHttpQueryHeaders failed: %u\n", GetLastError());
479     ok(index == 1, "WinHttpQueryHeaders failed to increment index.\n");
480     ok(memcmp(buffer, test_indices[2], sizeof(test_indices[2])) == 0, "WinHttpQueryHeaders returned incorrect string.\n");
481
482     len = sizeof(buffer);
483     ret = WinHttpQueryHeaders(request, WINHTTP_QUERY_CUSTOM | WINHTTP_QUERY_FLAG_REQUEST_HEADERS,
484         test_header_name, buffer, &len, &index);
485     ok(ret == TRUE, "WinHttpQueryHeaders failed: %u\n", GetLastError());
486     ok(index == 2, "WinHttpQueryHeaders failed to increment index.\n");
487     ok(memcmp(buffer, test_indices[3], sizeof(test_indices[2])) == 0, "WinHttpQueryHeaders returned incorrect string.\n");
488
489     len = sizeof(buffer);
490     ret = WinHttpQueryHeaders(request, WINHTTP_QUERY_CUSTOM | WINHTTP_QUERY_FLAG_REQUEST_HEADERS,
491         test_header_name, buffer, &len, &index);
492     ok(ret == FALSE, "WinHttpQueryHeaders succeeded unexpectedly, found third header.\n");
493
494     ret = WinHttpAddRequestHeaders(request, test_headers[8], ~0UL, WINHTTP_ADDREQ_FLAG_ADD);
495     ok(!ret, "WinHttpAddRequestHeaders failed\n");
496
497     ret = WinHttpAddRequestHeaders(request, test_headers[9], ~0UL, WINHTTP_ADDREQ_FLAG_ADD);
498     ok(ret, "WinHttpAddRequestHeaders failed\n");
499
500     ret = WinHttpAddRequestHeaders(request, test_headers[10], ~0UL, WINHTTP_ADDREQ_FLAG_ADD);
501     ok(!ret, "WinHttpAddRequestHeaders failed\n");
502
503     ret = WinHttpAddRequestHeaders(request, test_headers[11], ~0UL, WINHTTP_ADDREQ_FLAG_ADD);
504     ok(!ret, "WinHttpAddRequestHeaders failed\n");
505
506     ret = WinHttpAddRequestHeaders(request, test_headers[12], ~0UL, WINHTTP_ADDREQ_FLAG_ADD);
507     ok(!ret, "WinHttpAddRequestHeaders failed\n");
508
509     ret = WinHttpAddRequestHeaders(request, test_headers[13], ~0UL, WINHTTP_ADDREQ_FLAG_ADD);
510     ok(ret, "WinHttpAddRequestHeaders failed\n");
511
512     index = 0;
513     buffer[0] = 0;
514     len = sizeof(buffer);
515     ret = WinHttpQueryHeaders(request, WINHTTP_QUERY_CUSTOM | WINHTTP_QUERY_FLAG_REQUEST_HEADERS,
516         field, buffer, &len, &index);
517     ok(ret, "WinHttpQueryHeaders failed: %u\n", GetLastError());
518     ok(!memcmp(buffer, value, sizeof(value)), "unexpected result\n");
519
520     ret = WinHttpCloseHandle(request);
521     ok(ret == TRUE, "WinHttpCloseHandle failed on closing request, got %d.\n", ret);
522  done:
523     ret = WinHttpCloseHandle(connection);
524     ok(ret == TRUE, "WinHttpCloseHandle failed on closing connection, got %d.\n", ret);
525     ret = WinHttpCloseHandle(session);
526     ok(ret == TRUE, "WinHttpCloseHandle failed on closing session, got %d.\n", ret);
527
528 }
529
530 static void test_secure_connection(void)
531 {
532     static const WCHAR google[] = {'w','w','w','.','g','o','o','g','l','e','.','c','o','m',0};
533
534     HANDLE ses, con, req;
535     DWORD size, status, policy;
536     BOOL ret;
537
538     ses = WinHttpOpen(test_useragent, 0, NULL, NULL, 0);
539     ok(ses != NULL, "failed to open session %u\n", GetLastError());
540
541     policy = WINHTTP_OPTION_REDIRECT_POLICY_ALWAYS;
542     ret = WinHttpSetOption(ses, WINHTTP_OPTION_REDIRECT_POLICY, &policy, sizeof(policy));
543     ok(ret, "failed to set redirect policy %u\n", GetLastError());
544
545     con = WinHttpConnect(ses, google, 443, 0);
546     ok(con != NULL, "failed to open a connection %u\n", GetLastError());
547
548     /* try without setting WINHTTP_FLAG_SECURE */
549     req = WinHttpOpenRequest(con, NULL, NULL, NULL, NULL, NULL, 0);
550     ok(req != NULL, "failed to open a request %u\n", GetLastError());
551
552     ret = WinHttpSendRequest(req, NULL, 0, NULL, 0, 0, 0);
553     ok(ret, "failed to send request %u\n", GetLastError());
554
555     ret = WinHttpReceiveResponse(req, NULL);
556     ok(!ret, "succeeded unexpectedly\n");
557
558     size = 0;
559     ret = WinHttpQueryHeaders(req, WINHTTP_QUERY_RAW_HEADERS_CRLF, NULL, NULL, &size, NULL);
560     ok(!ret, "succeeded unexpectedly\n");
561
562     WinHttpCloseHandle(req);
563
564     req = WinHttpOpenRequest(con, NULL, NULL, NULL, NULL, NULL, WINHTTP_FLAG_SECURE);
565     ok(req != NULL, "failed to open a request %u\n", GetLastError());
566
567     ret = WinHttpSendRequest(req, NULL, 0, NULL, 0, 0, 0);
568     ok(ret, "failed to send request %u\n", GetLastError());
569
570     ret = WinHttpReceiveResponse(req, NULL);
571     ok(ret, "failed to receive response %u\n", GetLastError());
572
573     size = sizeof(status);
574     ret = WinHttpQueryHeaders(req, WINHTTP_QUERY_STATUS_CODE | WINHTTP_QUERY_FLAG_NUMBER, NULL, &status, &size, NULL);
575     ok(ret, "failed unexpectedly %u\n", GetLastError());
576     ok(status == 200, "request failed unexpectedly %u\n", status);
577
578     size = 0;
579     ret = WinHttpQueryHeaders(req, WINHTTP_QUERY_RAW_HEADERS_CRLF, NULL, NULL, &size, NULL);
580     ok(!ret, "succeeded unexpectedly\n");
581
582     WinHttpCloseHandle(req);
583     WinHttpCloseHandle(con);
584     WinHttpCloseHandle(ses);
585 }
586
587 static void test_request_parameter_defaults(void)
588 {
589     static const WCHAR empty[] = {0};
590     static const WCHAR codeweavers[] = {'c','o','d','e','w','e','a','v','e','r','s','.','c','o','m',0};
591
592     HANDLE ses, con, req;
593     DWORD size, status;
594     BOOL ret;
595
596     ses = WinHttpOpen(test_useragent, 0, NULL, NULL, 0);
597     ok(ses != NULL, "failed to open session %u\n", GetLastError());
598
599     con = WinHttpConnect(ses, codeweavers, 0, 0);
600     ok(con != NULL, "failed to open a connection %u\n", GetLastError());
601
602     req = WinHttpOpenRequest(con, NULL, NULL, NULL, NULL, NULL, 0);
603     ok(req != NULL, "failed to open a request %u\n", GetLastError());
604
605     ret = WinHttpSendRequest(req, NULL, 0, NULL, 0, 0, 0);
606     ok(ret, "failed to send request %u\n", GetLastError());
607
608     ret = WinHttpReceiveResponse(req, NULL);
609     ok(ret, "failed to receive response %u\n", GetLastError());
610
611     size = sizeof(status);
612     ret = WinHttpQueryHeaders(req, WINHTTP_QUERY_STATUS_CODE | WINHTTP_QUERY_FLAG_NUMBER, NULL, &status, &size, NULL);
613     ok(ret, "failed unexpectedly %u\n", GetLastError());
614     ok(status == 200, "request failed unexpectedly %u\n", status);
615
616     WinHttpCloseHandle(req);
617
618     req = WinHttpOpenRequest(con, empty, empty, empty, NULL, NULL, 0);
619     ok(req != NULL, "failed to open a request %u\n", GetLastError());
620
621     ret = WinHttpSendRequest(req, NULL, 0, NULL, 0, 0, 0);
622     ok(ret, "failed to send request %u\n", GetLastError());
623
624     ret = WinHttpReceiveResponse(req, NULL);
625     ok(ret, "failed to receive response %u\n", GetLastError());
626
627     size = sizeof(status);
628     ret = WinHttpQueryHeaders(req, WINHTTP_QUERY_STATUS_CODE | WINHTTP_QUERY_FLAG_NUMBER, NULL, &status, &size, NULL);
629     ok(ret, "failed unexpectedly %u\n", GetLastError());
630     ok(status == 200, "request failed unexpectedly %u\n", status);
631
632     WinHttpCloseHandle(req);
633     WinHttpCloseHandle(con);
634     WinHttpCloseHandle(ses);
635 }
636
637 START_TEST (winhttp)
638 {
639     test_OpenRequest();
640     test_SendRequest();
641     test_WinHttpTimeFromSystemTime();
642     test_WinHttpTimeToSystemTime();
643     test_WinHttpAddHeaders();
644     test_secure_connection();
645     test_request_parameter_defaults();
646 }