?  search:
summary | log | graphiclog1 | graphiclog2 | commit | commitdiff | tree | refs | edit | fork
blame | history | raw | HEAD
gdi32: Don't overflow the buffer in GetGlyphOutline.
[wine/multimedia.git] / dlls / wininet / http.c
blob40f8d9a17ea482eb80f605fe37ca656e92fb13e4
1 /*
2 * Wininet - HTTP Implementation
3 *
4 * Copyright 1999 Corel Corporation
5 * Copyright 2002 CodeWeavers Inc.
6 * Copyright 2002 TransGaming Technologies Inc.
7 * Copyright 2004 Mike McCormack for CodeWeavers
8 * Copyright 2005 Aric Stewart for CodeWeavers
9 * Copyright 2006 Robert Shearman for CodeWeavers
10 * Copyright 2011 Jacek Caban for CodeWeavers
11 *
12 * Ulrich Czekalla
13 * David Hammerton
14 *
15 * This library is free software; you can redistribute it and/or
16 * modify it under the terms of the GNU Lesser General Public
17 * License as published by the Free Software Foundation; either
18 * version 2.1 of the License, or (at your option) any later version.
19 *
20 * This library is distributed in the hope that it will be useful,
21 * but WITHOUT ANY WARRANTY; without even the implied warranty of
22 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
23 * Lesser General Public License for more details.
24 *
25 * You should have received a copy of the GNU Lesser General Public
26 * License along with this library; if not, write to the Free Software
27 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
28 */
29
30 #include "config.h"
31 #include "wine/port.h"
32
33 #if defined(__MINGW32__) || defined (_MSC_VER)
34 #include <ws2tcpip.h>
35 #endif
36
37 #include <sys/types.h>
38 #ifdef HAVE_SYS_SOCKET_H
39 # include <sys/socket.h>
40 #endif
41 #ifdef HAVE_ARPA_INET_H
42 # include <arpa/inet.h>
43 #endif
44 #include <stdarg.h>
45 #include <stdio.h>
46 #include <stdlib.h>
47 #ifdef HAVE_UNISTD_H
48 # include <unistd.h>
49 #endif
50 #include <time.h>
51 #include <assert.h>
52 #ifdef HAVE_ZLIB
53 # include <zlib.h>
54 #endif
55
56 #include "windef.h"
57 #include "winbase.h"
58 #include "wininet.h"
59 #include "winerror.h"
60 #include "winternl.h"
61 #define NO_SHLWAPI_STREAM
62 #define NO_SHLWAPI_REG
63 #define NO_SHLWAPI_STRFCNS
64 #define NO_SHLWAPI_GDI
65 #include "shlwapi.h"
66 #include "sspi.h"
67 #include "wincrypt.h"
68 #include "winuser.h"
69
70 #include "internet.h"
71 #include "wine/debug.h"
72 #include "wine/exception.h"
73 #include "wine/unicode.h"
74
75 WINE_DEFAULT_DEBUG_CHANNEL(wininet);
76
77 static const WCHAR g_szHttp1_0[] = {'H','T','T','P','/','1','.','0',0};
78 static const WCHAR g_szHttp1_1[] = {'H','T','T','P','/','1','.','1',0};
79 static const WCHAR szOK[] = {'O','K',0};
80 static const WCHAR szDefaultHeader[] = {'H','T','T','P','/','1','.','0',' ','2','0','0',' ','O','K',0};
81 static const WCHAR hostW[] = { 'H','o','s','t',0 };
82 static const WCHAR szAuthorization[] = { 'A','u','t','h','o','r','i','z','a','t','i','o','n',0 };
83 static const WCHAR szProxy_Authorization[] = { 'P','r','o','x','y','-','A','u','t','h','o','r','i','z','a','t','i','o','n',0 };
84 static const WCHAR szStatus[] = { 'S','t','a','t','u','s',0 };
85 static const WCHAR szKeepAlive[] = {'K','e','e','p','-','A','l','i','v','e',0};
86 static const WCHAR szGET[] = { 'G','E','T', 0 };
87 static const WCHAR szHEAD[] = { 'H','E','A','D', 0 };
88 static const WCHAR szCrLf[] = {'\r','\n', 0};
89
90 static const WCHAR szAccept[] = { 'A','c','c','e','p','t',0 };
91 static const WCHAR szAccept_Charset[] = { 'A','c','c','e','p','t','-','C','h','a','r','s','e','t', 0 };
92 static const WCHAR szAccept_Encoding[] = { 'A','c','c','e','p','t','-','E','n','c','o','d','i','n','g',0 };
93 static const WCHAR szAccept_Language[] = { 'A','c','c','e','p','t','-','L','a','n','g','u','a','g','e',0 };
94 static const WCHAR szAccept_Ranges[] = { 'A','c','c','e','p','t','-','R','a','n','g','e','s',0 };
95 static const WCHAR szAge[] = { 'A','g','e',0 };
96 static const WCHAR szAllow[] = { 'A','l','l','o','w',0 };
97 static const WCHAR szCache_Control[] = { 'C','a','c','h','e','-','C','o','n','t','r','o','l',0 };
98 static const WCHAR szConnection[] = { 'C','o','n','n','e','c','t','i','o','n',0 };
99 static const WCHAR szContent_Base[] = { 'C','o','n','t','e','n','t','-','B','a','s','e',0 };
100 static const WCHAR szContent_Disposition[] = { 'C','o','n','t','e','n','t','-','D','i','s','p','o','s','i','t','i','o','n',0 };
101 static const WCHAR szContent_Encoding[] = { 'C','o','n','t','e','n','t','-','E','n','c','o','d','i','n','g',0 };
102 static const WCHAR szContent_ID[] = { 'C','o','n','t','e','n','t','-','I','D',0 };
103 static const WCHAR szContent_Language[] = { 'C','o','n','t','e','n','t','-','L','a','n','g','u','a','g','e',0 };
104 static const WCHAR szContent_Length[] = { 'C','o','n','t','e','n','t','-','L','e','n','g','t','h',0 };
105 static const WCHAR szContent_Location[] = { 'C','o','n','t','e','n','t','-','L','o','c','a','t','i','o','n',0 };
106 static const WCHAR szContent_MD5[] = { 'C','o','n','t','e','n','t','-','M','D','5',0 };
107 static const WCHAR szContent_Range[] = { 'C','o','n','t','e','n','t','-','R','a','n','g','e',0 };
108 static const WCHAR szContent_Transfer_Encoding[] = { 'C','o','n','t','e','n','t','-','T','r','a','n','s','f','e','r','-','E','n','c','o','d','i','n','g',0 };
109 static const WCHAR szContent_Type[] = { 'C','o','n','t','e','n','t','-','T','y','p','e',0 };
110 static const WCHAR szCookie[] = { 'C','o','o','k','i','e',0 };
111 static const WCHAR szDate[] = { 'D','a','t','e',0 };
112 static const WCHAR szFrom[] = { 'F','r','o','m',0 };
113 static const WCHAR szETag[] = { 'E','T','a','g',0 };
114 static const WCHAR szExpect[] = { 'E','x','p','e','c','t',0 };
115 static const WCHAR szExpires[] = { 'E','x','p','i','r','e','s',0 };
116 static const WCHAR szIf_Match[] = { 'I','f','-','M','a','t','c','h',0 };
117 static const WCHAR szIf_Modified_Since[] = { 'I','f','-','M','o','d','i','f','i','e','d','-','S','i','n','c','e',0 };
118 static const WCHAR szIf_None_Match[] = { 'I','f','-','N','o','n','e','-','M','a','t','c','h',0 };
119 static const WCHAR szIf_Range[] = { 'I','f','-','R','a','n','g','e',0 };
120 static const WCHAR szIf_Unmodified_Since[] = { 'I','f','-','U','n','m','o','d','i','f','i','e','d','-','S','i','n','c','e',0 };
121 static const WCHAR szLast_Modified[] = { 'L','a','s','t','-','M','o','d','i','f','i','e','d',0 };
122 static const WCHAR szLocation[] = { 'L','o','c','a','t','i','o','n',0 };
123 static const WCHAR szMax_Forwards[] = { 'M','a','x','-','F','o','r','w','a','r','d','s',0 };
124 static const WCHAR szMime_Version[] = { 'M','i','m','e','-','V','e','r','s','i','o','n',0 };
125 static const WCHAR szPragma[] = { 'P','r','a','g','m','a',0 };
126 static const WCHAR szProxy_Authenticate[] = { 'P','r','o','x','y','-','A','u','t','h','e','n','t','i','c','a','t','e',0 };
127 static const WCHAR szProxy_Connection[] = { 'P','r','o','x','y','-','C','o','n','n','e','c','t','i','o','n',0 };
128 static const WCHAR szPublic[] = { 'P','u','b','l','i','c',0 };
129 static const WCHAR szRange[] = { 'R','a','n','g','e',0 };
130 static const WCHAR szReferer[] = { 'R','e','f','e','r','e','r',0 };
131 static const WCHAR szRetry_After[] = { 'R','e','t','r','y','-','A','f','t','e','r',0 };
132 static const WCHAR szServer[] = { 'S','e','r','v','e','r',0 };
133 static const WCHAR szSet_Cookie[] = { 'S','e','t','-','C','o','o','k','i','e',0 };
134 static const WCHAR szTransfer_Encoding[] = { 'T','r','a','n','s','f','e','r','-','E','n','c','o','d','i','n','g',0 };
135 static const WCHAR szUnless_Modified_Since[] = { 'U','n','l','e','s','s','-','M','o','d','i','f','i','e','d','-','S','i','n','c','e',0 };
136 static const WCHAR szUpgrade[] = { 'U','p','g','r','a','d','e',0 };
137 static const WCHAR szURI[] = { 'U','R','I',0 };
138 static const WCHAR szUser_Agent[] = { 'U','s','e','r','-','A','g','e','n','t',0 };
139 static const WCHAR szVary[] = { 'V','a','r','y',0 };
140 static const WCHAR szVia[] = { 'V','i','a',0 };
141 static const WCHAR szWarning[] = { 'W','a','r','n','i','n','g',0 };
142 static const WCHAR szWWW_Authenticate[] = { 'W','W','W','-','A','u','t','h','e','n','t','i','c','a','t','e',0 };
143
144 static const WCHAR emptyW[] = {0};
145
146 #define HTTP_REFERER szReferer
147 #define HTTP_ACCEPT szAccept
148 #define HTTP_USERAGENT szUser_Agent
149
150 #define HTTP_ADDHDR_FLAG_ADD 0x20000000
151 #define HTTP_ADDHDR_FLAG_ADD_IF_NEW 0x10000000
152 #define HTTP_ADDHDR_FLAG_COALESCE 0x40000000
153 #define HTTP_ADDHDR_FLAG_COALESCE_WITH_COMMA 0x40000000
154 #define HTTP_ADDHDR_FLAG_COALESCE_WITH_SEMICOLON 0x01000000
155 #define HTTP_ADDHDR_FLAG_REPLACE 0x80000000
156 #define HTTP_ADDHDR_FLAG_REQ 0x02000000
157
158 #define COLLECT_TIME 60000
159
160 #define ARRAYSIZE(array) (sizeof(array)/sizeof((array)[0]))
161
162 struct HttpAuthInfo
163 {
164 LPWSTR scheme;
165 CredHandle cred;
166 CtxtHandle ctx;
167 TimeStamp exp;
168 ULONG attr;
169 ULONG max_token;
170 void *auth_data;
171 unsigned int auth_data_len;
172 BOOL finished; /* finished authenticating */
173 };
174
175
176 typedef struct _basicAuthorizationData
177 {
178 struct list entry;
179
180 LPWSTR host;
181 LPWSTR realm;
182 LPSTR authorization;
183 UINT authorizationLen;
184 } basicAuthorizationData;
185
186 typedef struct _authorizationData
187 {
188 struct list entry;
189
190 LPWSTR host;
191 LPWSTR scheme;
192 LPWSTR domain;
193 UINT domain_len;
194 LPWSTR user;
195 UINT user_len;
196 LPWSTR password;
197 UINT password_len;
198 } authorizationData;
199
200 static struct list basicAuthorizationCache = LIST_INIT(basicAuthorizationCache);
201 static struct list authorizationCache = LIST_INIT(authorizationCache);
202
203 static CRITICAL_SECTION authcache_cs;
204 static CRITICAL_SECTION_DEBUG critsect_debug =
205 {
206 0, 0, &authcache_cs,
207 { &critsect_debug.ProcessLocksList, &critsect_debug.ProcessLocksList },
208 0, 0, { (DWORD_PTR)(__FILE__ ": authcache_cs") }
209 };
210 static CRITICAL_SECTION authcache_cs = { &critsect_debug, -1, 0, 0, 0, 0 };
211
212 static DWORD HTTP_GetResponseHeaders(http_request_t *req, INT *len);
213 static DWORD HTTP_ProcessHeader(http_request_t *req, LPCWSTR field, LPCWSTR value, DWORD dwModifier);
214 static LPWSTR * HTTP_InterpretHttpHeader(LPCWSTR buffer);
215 static DWORD HTTP_InsertCustomHeader(http_request_t *req, LPHTTPHEADERW lpHdr);
216 static INT HTTP_GetCustomHeaderIndex(http_request_t *req, LPCWSTR lpszField, INT index, BOOL Request);
217 static BOOL HTTP_DeleteCustomHeader(http_request_t *req, DWORD index);
218 static LPWSTR HTTP_build_req( LPCWSTR *list, int len );
219 static DWORD HTTP_HttpQueryInfoW(http_request_t*, DWORD, LPVOID, LPDWORD, LPDWORD);
220 static LPWSTR HTTP_GetRedirectURL(http_request_t *req, LPCWSTR lpszUrl);
221 static UINT HTTP_DecodeBase64(LPCWSTR base64, LPSTR bin);
222 static BOOL HTTP_VerifyValidHeader(http_request_t *req, LPCWSTR field);
223 static BOOL drain_content(http_request_t*,BOOL);
224
225 static CRITICAL_SECTION connection_pool_cs;
226 static CRITICAL_SECTION_DEBUG connection_pool_debug =
227 {
228 0, 0, &connection_pool_cs,
229 { &critsect_debug.ProcessLocksList, &critsect_debug.ProcessLocksList },
230 0, 0, { (DWORD_PTR)(__FILE__ ": connection_pool_cs") }
231 };
232 static CRITICAL_SECTION connection_pool_cs = { &connection_pool_debug, -1, 0, 0, 0, 0 };
233
234 static struct list connection_pool = LIST_INIT(connection_pool);
235 static BOOL collector_running;
236
237 void server_addref(server_t *server)
238 {
239 InterlockedIncrement(&server->ref);
240 }
241
242 void server_release(server_t *server)
243 {
244 if(InterlockedDecrement(&server->ref))
245 return;
246
247 list_remove(&server->entry);
248
249 if(server->cert_chain)
250 CertFreeCertificateChain(server->cert_chain);
251 heap_free(server->name);
252 heap_free(server->scheme_host_port);
253 heap_free(server);
254 }
255
256 static BOOL process_host_port(server_t *server)
257 {
258 BOOL default_port;
259 size_t name_len;
260 WCHAR *buf;
261
262 static const WCHAR httpW[] = {'h','t','t','p',0};
263 static const WCHAR httpsW[] = {'h','t','t','p','s',0};
264 static const WCHAR formatW[] = {'%','s',':','/','/','%','s',':','%','u',0};
265
266 name_len = strlenW(server->name);
267 buf = heap_alloc((name_len + 10 /* strlen("://:<port>") */)*sizeof(WCHAR) + sizeof(httpsW));
268 if(!buf)
269 return FALSE;
270
271 sprintfW(buf, formatW, server->is_https ? httpsW : httpW, server->name, server->port);
272 server->scheme_host_port = buf;
273
274 server->host_port = server->scheme_host_port + 7 /* strlen("http://") */;
275 if(server->is_https)
276 server->host_port++;
277
278 default_port = server->port == (server->is_https ? INTERNET_DEFAULT_HTTPS_PORT : INTERNET_DEFAULT_HTTP_PORT);
279 server->canon_host_port = default_port ? server->name : server->host_port;
280 return TRUE;
281 }
282
283 server_t *get_server(const WCHAR *name, INTERNET_PORT port, BOOL is_https, BOOL do_create)
284 {
285 server_t *iter, *server = NULL;
286
287 if(port == INTERNET_INVALID_PORT_NUMBER)
288 port = INTERNET_DEFAULT_HTTP_PORT;
289
290 EnterCriticalSection(&connection_pool_cs);
291
292 LIST_FOR_EACH_ENTRY(iter, &connection_pool, server_t, entry) {
293 if(iter->port == port && !strcmpW(iter->name, name) && iter->is_https == is_https) {
294 server = iter;
295 server_addref(server);
296 break;
297 }
298 }
299
300 if(!server && do_create) {
301 server = heap_alloc_zero(sizeof(*server));
302 if(server) {
303 server->ref = 2; /* list reference and return */
304 server->port = port;
305 server->is_https = is_https;
306 list_init(&server->conn_pool);
307 server->name = heap_strdupW(name);
308 if(server->name && process_host_port(server)) {
309 list_add_head(&connection_pool, &server->entry);
310 }else {
311 heap_free(server);
312 server = NULL;
313 }
314 }
315 }
316
317 LeaveCriticalSection(&connection_pool_cs);
318
319 return server;
320 }
321
322 BOOL collect_connections(collect_type_t collect_type)
323 {
324 netconn_t *netconn, *netconn_safe;
325 server_t *server, *server_safe;
326 BOOL remaining = FALSE;
327 DWORD64 now;
328
329 now = GetTickCount64();
330
331 LIST_FOR_EACH_ENTRY_SAFE(server, server_safe, &connection_pool, server_t, entry) {
332 LIST_FOR_EACH_ENTRY_SAFE(netconn, netconn_safe, &server->conn_pool, netconn_t, pool_entry) {
333 if(collect_type > COLLECT_TIMEOUT || netconn->keep_until < now) {
334 TRACE("freeing %p\n", netconn);
335 list_remove(&netconn->pool_entry);
336 free_netconn(netconn);
337 }else {
338 remaining = TRUE;
339 }
340 }
341
342 if(collect_type == COLLECT_CLEANUP) {
343 list_remove(&server->entry);
344 list_init(&server->entry);
345 server_release(server);
346 }
347 }
348
349 return remaining;
350 }
351
352 static DWORD WINAPI collect_connections_proc(void *arg)
353 {
354 BOOL remaining_conns;
355
356 do {
357 /* FIXME: Use more sophisticated method */
358 Sleep(5000);
359
360 EnterCriticalSection(&connection_pool_cs);
361
362 remaining_conns = collect_connections(COLLECT_TIMEOUT);
363 if(!remaining_conns)
364 collector_running = FALSE;
365
366 LeaveCriticalSection(&connection_pool_cs);
367 }while(remaining_conns);
368
369 FreeLibraryAndExitThread(WININET_hModule, 0);
370 }
371
372 /***********************************************************************
373 * HTTP_GetHeader (internal)
374 *
375 * Headers section must be held
376 */
377 static LPHTTPHEADERW HTTP_GetHeader(http_request_t *req, LPCWSTR head)
378 {
379 int HeaderIndex = 0;
380 HeaderIndex = HTTP_GetCustomHeaderIndex(req, head, 0, TRUE);
381 if (HeaderIndex == -1)
382 return NULL;
383 else
384 return &req->custHeaders[HeaderIndex];
385 }
386
387 static WCHAR *get_host_header( http_request_t *req )
388 {
389 HTTPHEADERW *header;
390 WCHAR *ret = NULL;
391
392 EnterCriticalSection( &req->headers_section );
393 if ((header = HTTP_GetHeader( req, hostW ))) ret = heap_strdupW( header->lpszValue );
394 LeaveCriticalSection( &req->headers_section );
395 return ret;
396 }
397
398 struct data_stream_vtbl_t {
399 DWORD (*get_avail_data)(data_stream_t*,http_request_t*);
400 BOOL (*end_of_data)(data_stream_t*,http_request_t*);
401 DWORD (*read)(data_stream_t*,http_request_t*,BYTE*,DWORD,DWORD*,blocking_mode_t);
402 BOOL (*drain_content)(data_stream_t*,http_request_t*);
403 void (*destroy)(data_stream_t*);
404 };
405
406 typedef struct {
407 data_stream_t data_stream;
408
409 BYTE buf[READ_BUFFER_SIZE];
410 DWORD buf_size;
411 DWORD buf_pos;
412 DWORD chunk_size;
413 BOOL end_of_data;
414 } chunked_stream_t;
415
416 static inline void destroy_data_stream(data_stream_t *stream)
417 {
418 stream->vtbl->destroy(stream);
419 }
420
421 static void reset_data_stream(http_request_t *req)
422 {
423 destroy_data_stream(req->data_stream);
424 req->data_stream = &req->netconn_stream.data_stream;
425 req->read_pos = req->read_size = req->netconn_stream.content_read = 0;
426 req->read_chunked = req->read_gzip = FALSE;
427 }
428
429 #ifdef HAVE_ZLIB
430
431 typedef struct {
432 data_stream_t stream;
433 data_stream_t *parent_stream;
434 z_stream zstream;
435 BYTE buf[READ_BUFFER_SIZE];
436 DWORD buf_size;
437 DWORD buf_pos;
438 BOOL end_of_data;
439 } gzip_stream_t;
440
441 static DWORD gzip_get_avail_data(data_stream_t *stream, http_request_t *req)
442 {
443 /* Allow reading only from read buffer */
444 return 0;
445 }
446
447 static BOOL gzip_end_of_data(data_stream_t *stream, http_request_t *req)
448 {
449 gzip_stream_t *gzip_stream = (gzip_stream_t*)stream;
450 return gzip_stream->end_of_data
451 || (!gzip_stream->buf_size && gzip_stream->parent_stream->vtbl->end_of_data(gzip_stream->parent_stream, req));
452 }
453
454 static DWORD gzip_read(data_stream_t *stream, http_request_t *req, BYTE *buf, DWORD size,
455 DWORD *read, blocking_mode_t blocking_mode)
456 {
457 gzip_stream_t *gzip_stream = (gzip_stream_t*)stream;
458 z_stream *zstream = &gzip_stream->zstream;
459 DWORD current_read, ret_read = 0;
460 int zres;
461 DWORD res = ERROR_SUCCESS;
462
463 TRACE("(%d %d)\n", size, blocking_mode);
464
465 while(size && !gzip_stream->end_of_data) {
466 if(!gzip_stream->buf_size) {
467 if(gzip_stream->buf_pos) {
468 if(gzip_stream->buf_size)
469 memmove(gzip_stream->buf, gzip_stream->buf+gzip_stream->buf_pos, gzip_stream->buf_size);
470 gzip_stream->buf_pos = 0;
471 }
472 res = gzip_stream->parent_stream->vtbl->read(gzip_stream->parent_stream, req, gzip_stream->buf+gzip_stream->buf_size,
473 sizeof(gzip_stream->buf)-gzip_stream->buf_size, &current_read, blocking_mode);
474 gzip_stream->buf_size += current_read;
475 if(res != ERROR_SUCCESS)
476 break;
477
478 if(!current_read) {
479 if(blocking_mode != BLOCKING_DISALLOW) {
480 WARN("unexpected end of data\n");
481 gzip_stream->end_of_data = TRUE;
482 }
483 break;
484 }
485 }
486
487 zstream->next_in = gzip_stream->buf+gzip_stream->buf_pos;
488 zstream->avail_in = gzip_stream->buf_size;
489 zstream->next_out = buf+ret_read;
490 zstream->avail_out = size;
491 zres = inflate(&gzip_stream->zstream, 0);
492 current_read = size - zstream->avail_out;
493 size -= current_read;
494 ret_read += current_read;
495 gzip_stream->buf_size -= zstream->next_in - (gzip_stream->buf+gzip_stream->buf_pos);
496 gzip_stream->buf_pos = zstream->next_in-gzip_stream->buf;
497 if(zres == Z_STREAM_END) {
498 TRACE("end of data\n");
499 gzip_stream->end_of_data = TRUE;
500 inflateEnd(zstream);
501 }else if(zres != Z_OK) {
502 WARN("inflate failed %d: %s\n", zres, debugstr_a(zstream->msg));
503 if(!ret_read)
504 res = ERROR_INTERNET_DECODING_FAILED;
505 break;
506 }
507
508 if(ret_read && blocking_mode == BLOCKING_ALLOW)
509 blocking_mode = BLOCKING_DISALLOW;
510 }
511
512 TRACE("read %u bytes\n", ret_read);
513 *read = ret_read;
514 return res;
515 }
516
517 static BOOL gzip_drain_content(data_stream_t *stream, http_request_t *req)
518 {
519 gzip_stream_t *gzip_stream = (gzip_stream_t*)stream;
520 return gzip_stream->parent_stream->vtbl->drain_content(gzip_stream->parent_stream, req);
521 }
522
523 static void gzip_destroy(data_stream_t *stream)
524 {
525 gzip_stream_t *gzip_stream = (gzip_stream_t*)stream;
526
527 destroy_data_stream(gzip_stream->parent_stream);
528
529 if(!gzip_stream->end_of_data)
530 inflateEnd(&gzip_stream->zstream);
531 heap_free(gzip_stream);
532 }
533
534 static const data_stream_vtbl_t gzip_stream_vtbl = {
535 gzip_get_avail_data,
536 gzip_end_of_data,
537 gzip_read,
538 gzip_drain_content,
539 gzip_destroy
540 };
541
542 static voidpf wininet_zalloc(voidpf opaque, uInt items, uInt size)
543 {
544 return heap_alloc(items*size);
545 }
546
547 static void wininet_zfree(voidpf opaque, voidpf address)
548 {
549 heap_free(address);
550 }
551
552 static DWORD init_gzip_stream(http_request_t *req, BOOL is_gzip)
553 {
554 gzip_stream_t *gzip_stream;
555 int index, zres;
556
557 gzip_stream = heap_alloc_zero(sizeof(gzip_stream_t));
558 if(!gzip_stream)
559 return ERROR_OUTOFMEMORY;
560
561 gzip_stream->stream.vtbl = &gzip_stream_vtbl;
562 gzip_stream->zstream.zalloc = wininet_zalloc;
563 gzip_stream->zstream.zfree = wininet_zfree;
564
565 zres = inflateInit2(&gzip_stream->zstream, is_gzip ? 0x1f : -15);
566 if(zres != Z_OK) {
567 ERR("inflateInit failed: %d\n", zres);
568 heap_free(gzip_stream);
569 return ERROR_OUTOFMEMORY;
570 }
571
572 EnterCriticalSection( &req->headers_section );
573 index = HTTP_GetCustomHeaderIndex(req, szContent_Length, 0, FALSE);
574 if(index != -1)
575 HTTP_DeleteCustomHeader(req, index);
576 LeaveCriticalSection( &req->headers_section );
577
578 if(req->read_size) {
579 memcpy(gzip_stream->buf, req->read_buf+req->read_pos, req->read_size);
580 gzip_stream->buf_size = req->read_size;
581 req->read_pos = req->read_size = 0;
582 }
583
584 req->read_gzip = TRUE;
585 gzip_stream->parent_stream = req->data_stream;
586 req->data_stream = &gzip_stream->stream;
587 return ERROR_SUCCESS;
588 }
589
590 #else
591
592 static DWORD init_gzip_stream(http_request_t *req, BOOL is_gzip)
593 {
594 ERR("gzip stream not supported, missing zlib.\n");
595 return ERROR_SUCCESS;
596 }
597
598 #endif
599
600 /***********************************************************************
601 * HTTP_FreeTokens (internal)
602 *
603 * Frees table of pointers.
604 */
605 static void HTTP_FreeTokens(LPWSTR * token_array)
606 {
607 int i;
608 for (i = 0; token_array[i]; i++) heap_free(token_array[i]);
609 heap_free(token_array);
610 }
611
612 static void HTTP_FixURL(http_request_t *request)
613 {
614 static const WCHAR szSlash[] = { '/',0 };
615 static const WCHAR szHttp[] = { 'h','t','t','p',':','/','/', 0 };
616
617 /* If we don't have a path we set it to root */
618 if (NULL == request->path)
619 request->path = heap_strdupW(szSlash);
620 else /* remove \r and \n*/
621 {
622 int nLen = strlenW(request->path);
623 while ((nLen >0 ) && ((request->path[nLen-1] == '\r')||(request->path[nLen-1] == '\n')))
624 {
625 nLen--;
626 request->path[nLen]='\0';
627 }
628 /* Replace '\' with '/' */
629 while (nLen>0) {
630 nLen--;
631 if (request->path[nLen] == '\\') request->path[nLen]='/';
632 }
633 }
634
635 if(CSTR_EQUAL != CompareStringW( LOCALE_INVARIANT, NORM_IGNORECASE,
636 request->path, strlenW(request->path), szHttp, strlenW(szHttp) )
637 && request->path[0] != '/') /* not an absolute path ?? --> fix it !! */
638 {
639 WCHAR *fixurl = heap_alloc((strlenW(request->path) + 2)*sizeof(WCHAR));
640 *fixurl = '/';
641 strcpyW(fixurl + 1, request->path);
642 heap_free( request->path );
643 request->path = fixurl;
644 }
645 }
646
647 static WCHAR* build_request_header(http_request_t *request, const WCHAR *verb,
648 const WCHAR *path, const WCHAR *version, BOOL use_cr)
649 {
650 static const WCHAR szSpace[] = {' ',0};
651 static const WCHAR szColon[] = {':',' ',0};
652 static const WCHAR szCr[] = {'\r',0};
653 static const WCHAR szLf[] = {'\n',0};
654 LPWSTR requestString;
655 DWORD len, n;
656 LPCWSTR *req;
657 UINT i;
658
659 EnterCriticalSection( &request->headers_section );
660
661 /* allocate space for an array of all the string pointers to be added */
662 len = request->nCustHeaders * 5 + 10;
663 if (!(req = heap_alloc( len * sizeof(const WCHAR *) )))
664 {
665 LeaveCriticalSection( &request->headers_section );
666 return NULL;
667 }
668
669 /* add the verb, path and HTTP version string */
670 n = 0;
671 req[n++] = verb;
672 req[n++] = szSpace;
673 req[n++] = path;
674 req[n++] = szSpace;
675 req[n++] = version;
676 if (use_cr)
677 req[n++] = szCr;
678 req[n++] = szLf;
679
680 /* Append custom request headers */
681 for (i = 0; i < request->nCustHeaders; i++)
682 {
683 if (request->custHeaders[i].wFlags & HDR_ISREQUEST)
684 {
685 req[n++] = request->custHeaders[i].lpszField;
686 req[n++] = szColon;
687 req[n++] = request->custHeaders[i].lpszValue;
688 if (use_cr)
689 req[n++] = szCr;
690 req[n++] = szLf;
691
692 TRACE("Adding custom header %s (%s)\n",
693 debugstr_w(request->custHeaders[i].lpszField),
694 debugstr_w(request->custHeaders[i].lpszValue));
695 }
696 }
697 if (use_cr)
698 req[n++] = szCr;
699 req[n++] = szLf;
700 req[n] = NULL;
701
702 requestString = HTTP_build_req( req, 4 );
703 heap_free( req );
704 LeaveCriticalSection( &request->headers_section );
705 return requestString;
706 }
707
708 static WCHAR* build_response_header(http_request_t *request, BOOL use_cr)
709 {
710 static const WCHAR colonW[] = { ':',' ',0 };
711 static const WCHAR crW[] = { '\r',0 };
712 static const WCHAR lfW[] = { '\n',0 };
713 static const WCHAR status_fmt[] = { ' ','%','u',' ',0 };
714 const WCHAR **req;
715 WCHAR *ret, buf[14];
716 DWORD i, n = 0;
717
718 EnterCriticalSection( &request->headers_section );
719
720 if (!(req = heap_alloc( (request->nCustHeaders * 5 + 8) * sizeof(WCHAR *) )))
721 {
722 LeaveCriticalSection( &request->headers_section );
723 return NULL;
724 }
725
726 if (request->status_code)
727 {
728 req[n++] = request->version;
729 sprintfW(buf, status_fmt, request->status_code);
730 req[n++] = buf;
731 req[n++] = request->statusText;
732 if (use_cr)
733 req[n++] = crW;
734 req[n++] = lfW;
735 }
736
737 for(i = 0; i < request->nCustHeaders; i++)
738 {
739 if(!(request->custHeaders[i].wFlags & HDR_ISREQUEST)
740 && strcmpW(request->custHeaders[i].lpszField, szStatus))
741 {
742 req[n++] = request->custHeaders[i].lpszField;
743 req[n++] = colonW;
744 req[n++] = request->custHeaders[i].lpszValue;
745 if(use_cr)
746 req[n++] = crW;
747 req[n++] = lfW;
748
749 TRACE("Adding custom header %s (%s)\n",
750 debugstr_w(request->custHeaders[i].lpszField),
751 debugstr_w(request->custHeaders[i].lpszValue));
752 }
753 }
754 if(use_cr)
755 req[n++] = crW;
756 req[n++] = lfW;
757 req[n] = NULL;
758
759 ret = HTTP_build_req(req, 0);
760 heap_free(req);
761 LeaveCriticalSection( &request->headers_section );
762 return ret;
763 }
764
765 static void HTTP_ProcessCookies( http_request_t *request )
766 {
767 int HeaderIndex;
768 int numCookies = 0;
769 LPHTTPHEADERW setCookieHeader;
770
771 if(request->hdr.dwFlags & INTERNET_FLAG_NO_COOKIES)
772 return;
773
774 EnterCriticalSection( &request->headers_section );
775
776 while((HeaderIndex = HTTP_GetCustomHeaderIndex(request, szSet_Cookie, numCookies++, FALSE)) != -1)
777 {
778 HTTPHEADERW *host;
779 const WCHAR *data;
780 WCHAR *name;
781
782 setCookieHeader = &request->custHeaders[HeaderIndex];
783
784 if (!setCookieHeader->lpszValue)
785 continue;
786
787 host = HTTP_GetHeader(request, hostW);
788 if(!host)
789 continue;
790
791 data = strchrW(setCookieHeader->lpszValue, '=');
792 if(!data)
793 continue;
794
795 name = heap_strndupW(setCookieHeader->lpszValue, data-setCookieHeader->lpszValue);
796 if(!name)
797 continue;
798
799 data++;
800 set_cookie(host->lpszValue, request->path, name, data, INTERNET_COOKIE_HTTPONLY);
801 heap_free(name);
802 }
803
804 LeaveCriticalSection( &request->headers_section );
805 }
806
807 static void strip_spaces(LPWSTR start)
808 {
809 LPWSTR str = start;
810 LPWSTR end;
811
812 while (*str == ' ' && *str != '\0')
813 str++;
814
815 if (str != start)
816 memmove(start, str, sizeof(WCHAR) * (strlenW(str) + 1));
817
818 end = start + strlenW(start) - 1;
819 while (end >= start && *end == ' ')
820 {
821 *end = '\0';
822 end--;
823 }
824 }
825
826 static inline BOOL is_basic_auth_value( LPCWSTR pszAuthValue, LPWSTR *pszRealm )
827 {
828 static const WCHAR szBasic[] = {'B','a','s','i','c'}; /* Note: not nul-terminated */
829 static const WCHAR szRealm[] = {'r','e','a','l','m'}; /* Note: not nul-terminated */
830 BOOL is_basic;
831 is_basic = !strncmpiW(pszAuthValue, szBasic, ARRAYSIZE(szBasic)) &&
832 ((pszAuthValue[ARRAYSIZE(szBasic)] == ' ') || !pszAuthValue[ARRAYSIZE(szBasic)]);
833 if (is_basic && pszRealm)
834 {
835 LPCWSTR token;
836 LPCWSTR ptr = &pszAuthValue[ARRAYSIZE(szBasic)];
837 LPCWSTR realm;
838 ptr++;
839 *pszRealm=NULL;
840 token = strchrW(ptr,'=');
841 if (!token)
842 return TRUE;
843 realm = ptr;
844 while (*realm == ' ' && *realm != '\0')
845 realm++;
846 if(!strncmpiW(realm, szRealm, ARRAYSIZE(szRealm)) &&
847 (realm[ARRAYSIZE(szRealm)] == ' ' || realm[ARRAYSIZE(szRealm)] == '='))
848 {
849 token++;
850 while (*token == ' ' && *token != '\0')
851 token++;
852 if (*token == '\0')
853 return TRUE;
854 *pszRealm = heap_strdupW(token);
855 strip_spaces(*pszRealm);
856 }
857 }
858
859 return is_basic;
860 }
861
862 static void destroy_authinfo( struct HttpAuthInfo *authinfo )
863 {
864 if (!authinfo) return;
865
866 if (SecIsValidHandle(&authinfo->ctx))
867 DeleteSecurityContext(&authinfo->ctx);
868 if (SecIsValidHandle(&authinfo->cred))
869 FreeCredentialsHandle(&authinfo->cred);
870
871 heap_free(authinfo->auth_data);
872 heap_free(authinfo->scheme);
873 heap_free(authinfo);
874 }
875
876 static UINT retrieve_cached_basic_authorization(LPWSTR host, LPWSTR realm, LPSTR *auth_data)
877 {
878 basicAuthorizationData *ad;
879 UINT rc = 0;
880
881 TRACE("Looking for authorization for %s:%s\n",debugstr_w(host),debugstr_w(realm));
882
883 EnterCriticalSection(&authcache_cs);
884 LIST_FOR_EACH_ENTRY(ad, &basicAuthorizationCache, basicAuthorizationData, entry)
885 {
886 if (!strcmpiW(host,ad->host) && !strcmpW(realm,ad->realm))
887 {
888 TRACE("Authorization found in cache\n");
889 *auth_data = heap_alloc(ad->authorizationLen);
890 memcpy(*auth_data,ad->authorization,ad->authorizationLen);
891 rc = ad->authorizationLen;
892 break;
893 }
894 }
895 LeaveCriticalSection(&authcache_cs);
896 return rc;
897 }
898
899 static void cache_basic_authorization(LPWSTR host, LPWSTR realm, LPSTR auth_data, UINT auth_data_len)
900 {
901 struct list *cursor;
902 basicAuthorizationData* ad = NULL;
903
904 TRACE("caching authorization for %s:%s = %s\n",debugstr_w(host),debugstr_w(realm),debugstr_an(auth_data,auth_data_len));
905
906 EnterCriticalSection(&authcache_cs);
907 LIST_FOR_EACH(cursor, &basicAuthorizationCache)
908 {
909 basicAuthorizationData *check = LIST_ENTRY(cursor,basicAuthorizationData,entry);
910 if (!strcmpiW(host,check->host) && !strcmpW(realm,check->realm))
911 {
912 ad = check;
913 break;
914 }
915 }
916
917 if (ad)
918 {
919 TRACE("Found match in cache, replacing\n");
920 heap_free(ad->authorization);
921 ad->authorization = heap_alloc(auth_data_len);
922 memcpy(ad->authorization, auth_data, auth_data_len);
923 ad->authorizationLen = auth_data_len;
924 }
925 else
926 {
927 ad = heap_alloc(sizeof(basicAuthorizationData));
928 ad->host = heap_strdupW(host);
929 ad->realm = heap_strdupW(realm);
930 ad->authorization = heap_alloc(auth_data_len);
931 memcpy(ad->authorization, auth_data, auth_data_len);
932 ad->authorizationLen = auth_data_len;
933 list_add_head(&basicAuthorizationCache,&ad->entry);
934 TRACE("authorization cached\n");
935 }
936 LeaveCriticalSection(&authcache_cs);
937 }
938
939 static BOOL retrieve_cached_authorization(LPWSTR host, LPWSTR scheme,
940 SEC_WINNT_AUTH_IDENTITY_W *nt_auth_identity)
941 {
942 authorizationData *ad;
943
944 TRACE("Looking for authorization for %s:%s\n", debugstr_w(host), debugstr_w(scheme));
945
946 EnterCriticalSection(&authcache_cs);
947 LIST_FOR_EACH_ENTRY(ad, &authorizationCache, authorizationData, entry) {
948 if(!strcmpiW(host, ad->host) && !strcmpiW(scheme, ad->scheme)) {
949 TRACE("Authorization found in cache\n");
950
951 nt_auth_identity->User = heap_strdupW(ad->user);
952 nt_auth_identity->Password = heap_strdupW(ad->password);
953 nt_auth_identity->Domain = heap_alloc(sizeof(WCHAR)*ad->domain_len);
954 if(!nt_auth_identity->User || !nt_auth_identity->Password ||
955 (!nt_auth_identity->Domain && ad->domain_len)) {
956 heap_free(nt_auth_identity->User);
957 heap_free(nt_auth_identity->Password);
958 heap_free(nt_auth_identity->Domain);
959 break;
960 }
961
962 nt_auth_identity->Flags = SEC_WINNT_AUTH_IDENTITY_UNICODE;
963 nt_auth_identity->UserLength = ad->user_len;
964 nt_auth_identity->PasswordLength = ad->password_len;
965 memcpy(nt_auth_identity->Domain, ad->domain, sizeof(WCHAR)*ad->domain_len);
966 nt_auth_identity->DomainLength = ad->domain_len;
967 LeaveCriticalSection(&authcache_cs);
968 return TRUE;
969 }
970 }
971 LeaveCriticalSection(&authcache_cs);
972
973 return FALSE;
974 }
975
976 static void cache_authorization(LPWSTR host, LPWSTR scheme,
977 SEC_WINNT_AUTH_IDENTITY_W *nt_auth_identity)
978 {
979 authorizationData *ad;
980 BOOL found = FALSE;
981
982 TRACE("Caching authorization for %s:%s\n", debugstr_w(host), debugstr_w(scheme));
983
984 EnterCriticalSection(&authcache_cs);
985 LIST_FOR_EACH_ENTRY(ad, &authorizationCache, authorizationData, entry)
986 if(!strcmpiW(host, ad->host) && !strcmpiW(scheme, ad->scheme)) {
987 found = TRUE;
988 break;
989 }
990
991 if(found) {
992 heap_free(ad->user);
993 heap_free(ad->password);
994 heap_free(ad->domain);
995 } else {
996 ad = heap_alloc(sizeof(authorizationData));
997 if(!ad) {
998 LeaveCriticalSection(&authcache_cs);
999 return;
1000 }
1001
1002 ad->host = heap_strdupW(host);
1003 ad->scheme = heap_strdupW(scheme);
1004 list_add_head(&authorizationCache, &ad->entry);
1005 }
1006
1007 ad->user = heap_strndupW(nt_auth_identity->User, nt_auth_identity->UserLength);
1008 ad->password = heap_strndupW(nt_auth_identity->Password, nt_auth_identity->PasswordLength);
1009 ad->domain = heap_strndupW(nt_auth_identity->Domain, nt_auth_identity->DomainLength);
1010 ad->user_len = nt_auth_identity->UserLength;
1011 ad->password_len = nt_auth_identity->PasswordLength;
1012 ad->domain_len = nt_auth_identity->DomainLength;
1013
1014 if(!ad->host || !ad->scheme || !ad->user || !ad->password
1015 || (nt_auth_identity->Domain && !ad->domain)) {
1016 heap_free(ad->host);
1017 heap_free(ad->scheme);
1018 heap_free(ad->user);
1019 heap_free(ad->password);
1020 heap_free(ad->domain);
1021 list_remove(&ad->entry);
1022 heap_free(ad);
1023 }
1024
1025 LeaveCriticalSection(&authcache_cs);
1026 }
1027
1028 static BOOL HTTP_DoAuthorization( http_request_t *request, LPCWSTR pszAuthValue,
1029 struct HttpAuthInfo **ppAuthInfo,
1030 LPWSTR domain_and_username, LPWSTR password,
1031 LPWSTR host )
1032 {
1033 SECURITY_STATUS sec_status;
1034 struct HttpAuthInfo *pAuthInfo = *ppAuthInfo;
1035 BOOL first = FALSE;
1036 LPWSTR szRealm = NULL;
1037
1038 TRACE("%s\n", debugstr_w(pszAuthValue));
1039
1040 if (!pAuthInfo)
1041 {
1042 TimeStamp exp;
1043
1044 first = TRUE;
1045 pAuthInfo = heap_alloc(sizeof(*pAuthInfo));
1046 if (!pAuthInfo)
1047 return FALSE;
1048
1049 SecInvalidateHandle(&pAuthInfo->cred);
1050 SecInvalidateHandle(&pAuthInfo->ctx);
1051 memset(&pAuthInfo->exp, 0, sizeof(pAuthInfo->exp));
1052 pAuthInfo->attr = 0;
1053 pAuthInfo->auth_data = NULL;
1054 pAuthInfo->auth_data_len = 0;
1055 pAuthInfo->finished = FALSE;
1056
1057 if (is_basic_auth_value(pszAuthValue,NULL))
1058 {
1059 static const WCHAR szBasic[] = {'B','a','s','i','c',0};
1060 pAuthInfo->scheme = heap_strdupW(szBasic);
1061 if (!pAuthInfo->scheme)
1062 {
1063 heap_free(pAuthInfo);
1064 return FALSE;
1065 }
1066 }
1067 else
1068 {
1069 PVOID pAuthData;
1070 SEC_WINNT_AUTH_IDENTITY_W nt_auth_identity;
1071
1072 pAuthInfo->scheme = heap_strdupW(pszAuthValue);
1073 if (!pAuthInfo->scheme)
1074 {
1075 heap_free(pAuthInfo);
1076 return FALSE;
1077 }
1078
1079 if (domain_and_username)
1080 {
1081 WCHAR *user = strchrW(domain_and_username, '\\');
1082 WCHAR *domain = domain_and_username;
1083
1084 /* FIXME: make sure scheme accepts SEC_WINNT_AUTH_IDENTITY before calling AcquireCredentialsHandle */
1085
1086 pAuthData = &nt_auth_identity;
1087
1088 if (user) user++;
1089 else
1090 {
1091 user = domain_and_username;
1092 domain = NULL;
1093 }
1094
1095 nt_auth_identity.Flags = SEC_WINNT_AUTH_IDENTITY_UNICODE;
1096 nt_auth_identity.User = user;
1097 nt_auth_identity.UserLength = strlenW(nt_auth_identity.User);
1098 nt_auth_identity.Domain = domain;
1099 nt_auth_identity.DomainLength = domain ? user - domain - 1 : 0;
1100 nt_auth_identity.Password = password;
1101 nt_auth_identity.PasswordLength = strlenW(nt_auth_identity.Password);
1102
1103 cache_authorization(host, pAuthInfo->scheme, &nt_auth_identity);
1104 }
1105 else if(retrieve_cached_authorization(host, pAuthInfo->scheme, &nt_auth_identity))
1106 pAuthData = &nt_auth_identity;
1107 else
1108 /* use default credentials */
1109 pAuthData = NULL;
1110
1111 sec_status = AcquireCredentialsHandleW(NULL, pAuthInfo->scheme,
1112 SECPKG_CRED_OUTBOUND, NULL,
1113 pAuthData, NULL,
1114 NULL, &pAuthInfo->cred,
1115 &exp);
1116
1117 if(pAuthData && !domain_and_username) {
1118 heap_free(nt_auth_identity.User);
1119 heap_free(nt_auth_identity.Domain);
1120 heap_free(nt_auth_identity.Password);
1121 }
1122
1123 if (sec_status == SEC_E_OK)
1124 {
1125 PSecPkgInfoW sec_pkg_info;
1126 sec_status = QuerySecurityPackageInfoW(pAuthInfo->scheme, &sec_pkg_info);
1127 if (sec_status == SEC_E_OK)
1128 {
1129 pAuthInfo->max_token = sec_pkg_info->cbMaxToken;
1130 FreeContextBuffer(sec_pkg_info);
1131 }
1132 }
1133 if (sec_status != SEC_E_OK)
1134 {
1135 WARN("AcquireCredentialsHandleW for scheme %s failed with error 0x%08x\n",
1136 debugstr_w(pAuthInfo->scheme), sec_status);
1137 heap_free(pAuthInfo->scheme);
1138 heap_free(pAuthInfo);
1139 return FALSE;
1140 }
1141 }
1142 *ppAuthInfo = pAuthInfo;
1143 }
1144 else if (pAuthInfo->finished)
1145 return FALSE;
1146
1147 if ((strlenW(pszAuthValue) < strlenW(pAuthInfo->scheme)) ||
1148 strncmpiW(pszAuthValue, pAuthInfo->scheme, strlenW(pAuthInfo->scheme)))
1149 {
1150 ERR("authentication scheme changed from %s to %s\n",
1151 debugstr_w(pAuthInfo->scheme), debugstr_w(pszAuthValue));
1152 return FALSE;
1153 }
1154
1155 if (is_basic_auth_value(pszAuthValue,&szRealm))
1156 {
1157 int userlen;
1158 int passlen;
1159 char *auth_data = NULL;
1160 UINT auth_data_len = 0;
1161
1162 TRACE("basic authentication realm %s\n",debugstr_w(szRealm));
1163
1164 if (!domain_and_username)
1165 {
1166 if (host && szRealm)
1167 auth_data_len = retrieve_cached_basic_authorization(host, szRealm,&auth_data);
1168 if (auth_data_len == 0)
1169 {
1170 heap_free(szRealm);
1171 return FALSE;
1172 }
1173 }
1174 else
1175 {
1176 userlen = WideCharToMultiByte(CP_UTF8, 0, domain_and_username, lstrlenW(domain_and_username), NULL, 0, NULL, NULL);
1177 passlen = WideCharToMultiByte(CP_UTF8, 0, password, lstrlenW(password), NULL, 0, NULL, NULL);
1178
1179 /* length includes a nul terminator, which will be re-used for the ':' */
1180 auth_data = heap_alloc(userlen + 1 + passlen);
1181 if (!auth_data)
1182 {
1183 heap_free(szRealm);
1184 return FALSE;
1185 }
1186
1187 WideCharToMultiByte(CP_UTF8, 0, domain_and_username, -1, auth_data, userlen, NULL, NULL);
1188 auth_data[userlen] = ':';
1189 WideCharToMultiByte(CP_UTF8, 0, password, -1, &auth_data[userlen+1], passlen, NULL, NULL);
1190 auth_data_len = userlen + 1 + passlen;
1191 if (host && szRealm)
1192 cache_basic_authorization(host, szRealm, auth_data, auth_data_len);
1193 }
1194
1195 pAuthInfo->auth_data = auth_data;
1196 pAuthInfo->auth_data_len = auth_data_len;
1197 pAuthInfo->finished = TRUE;
1198 heap_free(szRealm);
1199 return TRUE;
1200 }
1201 else
1202 {
1203 LPCWSTR pszAuthData;
1204 SecBufferDesc out_desc, in_desc;
1205 SecBuffer out, in;
1206 unsigned char *buffer;
1207 ULONG context_req = ISC_REQ_CONNECTION | ISC_REQ_USE_DCE_STYLE |
1208 ISC_REQ_MUTUAL_AUTH | ISC_REQ_DELEGATE;
1209
1210 in.BufferType = SECBUFFER_TOKEN;
1211 in.cbBuffer = 0;
1212 in.pvBuffer = NULL;
1213
1214 in_desc.ulVersion = 0;
1215 in_desc.cBuffers = 1;
1216 in_desc.pBuffers = &in;
1217
1218 pszAuthData = pszAuthValue + strlenW(pAuthInfo->scheme);
1219 if (*pszAuthData == ' ')
1220 {
1221 pszAuthData++;
1222 in.cbBuffer = HTTP_DecodeBase64(pszAuthData, NULL);
1223 in.pvBuffer = heap_alloc(in.cbBuffer);
1224 HTTP_DecodeBase64(pszAuthData, in.pvBuffer);
1225 }
1226
1227 buffer = heap_alloc(pAuthInfo->max_token);
1228
1229 out.BufferType = SECBUFFER_TOKEN;
1230 out.cbBuffer = pAuthInfo->max_token;
1231 out.pvBuffer = buffer;
1232
1233 out_desc.ulVersion = 0;
1234 out_desc.cBuffers = 1;
1235 out_desc.pBuffers = &out;
1236
1237 sec_status = InitializeSecurityContextW(first ? &pAuthInfo->cred : NULL,
1238 first ? NULL : &pAuthInfo->ctx,
1239 first ? request->server->name : NULL,
1240 context_req, 0, SECURITY_NETWORK_DREP,
1241 in.pvBuffer ? &in_desc : NULL,
1242 0, &pAuthInfo->ctx, &out_desc,
1243 &pAuthInfo->attr, &pAuthInfo->exp);
1244 if (sec_status == SEC_E_OK)
1245 {
1246 pAuthInfo->finished = TRUE;
1247 pAuthInfo->auth_data = out.pvBuffer;
1248 pAuthInfo->auth_data_len = out.cbBuffer;
1249 TRACE("sending last auth packet\n");
1250 }
1251 else if (sec_status == SEC_I_CONTINUE_NEEDED)
1252 {
1253 pAuthInfo->auth_data = out.pvBuffer;
1254 pAuthInfo->auth_data_len = out.cbBuffer;
1255 TRACE("sending next auth packet\n");
1256 }
1257 else
1258 {
1259 ERR("InitializeSecurityContextW returned error 0x%08x\n", sec_status);
1260 heap_free(out.pvBuffer);
1261 destroy_authinfo(pAuthInfo);
1262 *ppAuthInfo = NULL;
1263 return FALSE;
1264 }
1265 }
1266
1267 return TRUE;
1268 }
1269
1270 /***********************************************************************
1271 * HTTP_HttpAddRequestHeadersW (internal)
1272 */
1273 static DWORD HTTP_HttpAddRequestHeadersW(http_request_t *request,
1274 LPCWSTR lpszHeader, DWORD dwHeaderLength, DWORD dwModifier)
1275 {
1276 LPWSTR lpszStart;
1277 LPWSTR lpszEnd;
1278 LPWSTR buffer;
1279 DWORD len, res = ERROR_HTTP_INVALID_HEADER;
1280
1281 TRACE("copying header: %s\n", debugstr_wn(lpszHeader, dwHeaderLength));
1282
1283 if( dwHeaderLength == ~0U )
1284 len = strlenW(lpszHeader);
1285 else
1286 len = dwHeaderLength;
1287 buffer = heap_alloc(sizeof(WCHAR)*(len+1));
1288 lstrcpynW( buffer, lpszHeader, len + 1);
1289
1290 lpszStart = buffer;
1291
1292 do
1293 {
1294 LPWSTR * pFieldAndValue;
1295
1296 lpszEnd = lpszStart;
1297
1298 while (*lpszEnd != '\0')
1299 {
1300 if (*lpszEnd == '\r' || *lpszEnd == '\n')
1301 break;
1302 lpszEnd++;
1303 }
1304
1305 if (*lpszStart == '\0')
1306 break;
1307
1308 if (*lpszEnd == '\r' || *lpszEnd == '\n')
1309 {
1310 *lpszEnd = '\0';
1311 lpszEnd++; /* Jump over newline */
1312 }
1313 TRACE("interpreting header %s\n", debugstr_w(lpszStart));
1314 if (*lpszStart == '\0')
1315 {
1316 /* Skip 0-length headers */
1317 lpszStart = lpszEnd;
1318 res = ERROR_SUCCESS;
1319 continue;
1320 }
1321 pFieldAndValue = HTTP_InterpretHttpHeader(lpszStart);
1322 if (pFieldAndValue)
1323 {
1324 res = HTTP_VerifyValidHeader(request, pFieldAndValue[0]);
1325 if (res == ERROR_SUCCESS)
1326 res = HTTP_ProcessHeader(request, pFieldAndValue[0],
1327 pFieldAndValue[1], dwModifier | HTTP_ADDHDR_FLAG_REQ);
1328 HTTP_FreeTokens(pFieldAndValue);
1329 }
1330
1331 lpszStart = lpszEnd;
1332 } while (res == ERROR_SUCCESS);
1333
1334 heap_free(buffer);
1335 return res;
1336 }
1337
1338 /***********************************************************************
1339 * HttpAddRequestHeadersW (WININET.@)
1340 *
1341 * Adds one or more HTTP header to the request handler
1342 *
1343 * NOTE
1344 * On Windows if dwHeaderLength includes the trailing '\0', then
1345 * HttpAddRequestHeadersW() adds it too. However this results in an
1346 * invalid HTTP header which is rejected by some servers so we probably
1347 * don't need to match Windows on that point.
1348 *
1349 * RETURNS
1350 * TRUE on success
1351 * FALSE on failure
1352 *
1353 */
1354 BOOL WINAPI HttpAddRequestHeadersW(HINTERNET hHttpRequest,
1355 LPCWSTR lpszHeader, DWORD dwHeaderLength, DWORD dwModifier)
1356 {
1357 http_request_t *request;
1358 DWORD res = ERROR_INTERNET_INCORRECT_HANDLE_TYPE;
1359
1360 TRACE("%p, %s, %i, %i\n", hHttpRequest, debugstr_wn(lpszHeader, dwHeaderLength), dwHeaderLength, dwModifier);
1361
1362 if (!lpszHeader)
1363 return TRUE;
1364
1365 request = (http_request_t*) get_handle_object( hHttpRequest );
1366 if (request && request->hdr.htype == WH_HHTTPREQ)
1367 res = HTTP_HttpAddRequestHeadersW( request, lpszHeader, dwHeaderLength, dwModifier );
1368 if( request )
1369 WININET_Release( &request->hdr );
1370
1371 if(res != ERROR_SUCCESS)
1372 SetLastError(res);
1373 return res == ERROR_SUCCESS;
1374 }
1375
1376 /***********************************************************************
1377 * HttpAddRequestHeadersA (WININET.@)
1378 *
1379 * Adds one or more HTTP header to the request handler
1380 *
1381 * RETURNS
1382 * TRUE on success
1383 * FALSE on failure
1384 *
1385 */
1386 BOOL WINAPI HttpAddRequestHeadersA(HINTERNET hHttpRequest,
1387 LPCSTR lpszHeader, DWORD dwHeaderLength, DWORD dwModifier)
1388 {
1389 DWORD len;
1390 LPWSTR hdr;
1391 BOOL r;
1392
1393 TRACE("%p, %s, %i, %i\n", hHttpRequest, debugstr_an(lpszHeader, dwHeaderLength), dwHeaderLength, dwModifier);
1394
1395 len = MultiByteToWideChar( CP_ACP, 0, lpszHeader, dwHeaderLength, NULL, 0 );
1396 hdr = heap_alloc(len*sizeof(WCHAR));
1397 MultiByteToWideChar( CP_ACP, 0, lpszHeader, dwHeaderLength, hdr, len );
1398 if( dwHeaderLength != ~0U )
1399 dwHeaderLength = len;
1400
1401 r = HttpAddRequestHeadersW( hHttpRequest, hdr, dwHeaderLength, dwModifier );
1402
1403 heap_free( hdr );
1404 return r;
1405 }
1406
1407 static void free_accept_types( WCHAR **accept_types )
1408 {
1409 WCHAR *ptr, **types = accept_types;
1410
1411 if (!types) return;
1412 while ((ptr = *types))
1413 {
1414 heap_free( ptr );
1415 types++;
1416 }
1417 heap_free( accept_types );
1418 }
1419
1420 static WCHAR **convert_accept_types( const char **accept_types )
1421 {
1422 unsigned int count;
1423 const char **types = accept_types;
1424 WCHAR **typesW;
1425 BOOL invalid_pointer = FALSE;
1426
1427 if (!types) return NULL;
1428 count = 0;
1429 while (*types)
1430 {
1431 __TRY
1432 {
1433 /* find out how many there are */
1434 if (*types && **types)
1435 {
1436 TRACE("accept type: %s\n", debugstr_a(*types));
1437 count++;
1438 }
1439 }
1440 __EXCEPT_PAGE_FAULT
1441 {
1442 WARN("invalid accept type pointer\n");
1443 invalid_pointer = TRUE;
1444 }
1445 __ENDTRY;
1446 types++;
1447 }
1448 if (invalid_pointer) return NULL;
1449 if (!(typesW = heap_alloc( sizeof(WCHAR *) * (count + 1) ))) return NULL;
1450 count = 0;
1451 types = accept_types;
1452 while (*types)
1453 {
1454 if (*types && **types) typesW[count++] = heap_strdupAtoW( *types );
1455 types++;
1456 }
1457 typesW[count] = NULL;
1458 return typesW;
1459 }
1460
1461 /***********************************************************************
1462 * HttpOpenRequestA (WININET.@)
1463 *
1464 * Open a HTTP request handle
1465 *
1466 * RETURNS
1467 * HINTERNET a HTTP request handle on success
1468 * NULL on failure
1469 *
1470 */
1471 HINTERNET WINAPI HttpOpenRequestA(HINTERNET hHttpSession,
1472 LPCSTR lpszVerb, LPCSTR lpszObjectName, LPCSTR lpszVersion,
1473 LPCSTR lpszReferrer , LPCSTR *lpszAcceptTypes,
1474 DWORD dwFlags, DWORD_PTR dwContext)
1475 {
1476 LPWSTR szVerb = NULL, szObjectName = NULL;
1477 LPWSTR szVersion = NULL, szReferrer = NULL, *szAcceptTypes = NULL;
1478 HINTERNET rc = NULL;
1479
1480 TRACE("(%p, %s, %s, %s, %s, %p, %08x, %08lx)\n", hHttpSession,
1481 debugstr_a(lpszVerb), debugstr_a(lpszObjectName),
1482 debugstr_a(lpszVersion), debugstr_a(lpszReferrer), lpszAcceptTypes,
1483 dwFlags, dwContext);
1484
1485 if (lpszVerb)
1486 {
1487 szVerb = heap_strdupAtoW(lpszVerb);
1488 if ( !szVerb )
1489 goto end;
1490 }
1491
1492 if (lpszObjectName)
1493 {
1494 szObjectName = heap_strdupAtoW(lpszObjectName);
1495 if ( !szObjectName )
1496 goto end;
1497 }
1498
1499 if (lpszVersion)
1500 {
1501 szVersion = heap_strdupAtoW(lpszVersion);
1502 if ( !szVersion )
1503 goto end;
1504 }
1505
1506 if (lpszReferrer)
1507 {
1508 szReferrer = heap_strdupAtoW(lpszReferrer);
1509 if ( !szReferrer )
1510 goto end;
1511 }
1512
1513 szAcceptTypes = convert_accept_types( lpszAcceptTypes );
1514 rc = HttpOpenRequestW(hHttpSession, szVerb, szObjectName, szVersion, szReferrer,
1515 (const WCHAR **)szAcceptTypes, dwFlags, dwContext);
1516
1517 end:
1518 free_accept_types(szAcceptTypes);
1519 heap_free(szReferrer);
1520 heap_free(szVersion);
1521 heap_free(szObjectName);
1522 heap_free(szVerb);
1523 return rc;
1524 }
1525
1526 /***********************************************************************
1527 * HTTP_EncodeBase64
1528 */
1529 static UINT HTTP_EncodeBase64( LPCSTR bin, unsigned int len, LPWSTR base64 )
1530 {
1531 UINT n = 0, x;
1532 static const CHAR HTTP_Base64Enc[] =
1533 "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/";
1534
1535 while( len > 0 )
1536 {
1537 /* first 6 bits, all from bin[0] */
1538 base64[n++] = HTTP_Base64Enc[(bin[0] & 0xfc) >> 2];
1539 x = (bin[0] & 3) << 4;
1540
1541 /* next 6 bits, 2 from bin[0] and 4 from bin[1] */
1542 if( len == 1 )
1543 {
1544 base64[n++] = HTTP_Base64Enc[x];
1545 base64[n++] = '=';
1546 base64[n++] = '=';
1547 break;
1548 }
1549 base64[n++] = HTTP_Base64Enc[ x | ( (bin[1]&0xf0) >> 4 ) ];
1550 x = ( bin[1] & 0x0f ) << 2;
1551
1552 /* next 6 bits 4 from bin[1] and 2 from bin[2] */
1553 if( len == 2 )
1554 {
1555 base64[n++] = HTTP_Base64Enc[x];
1556 base64[n++] = '=';
1557 break;
1558 }
1559 base64[n++] = HTTP_Base64Enc[ x | ( (bin[2]&0xc0 ) >> 6 ) ];
1560
1561 /* last 6 bits, all from bin [2] */
1562 base64[n++] = HTTP_Base64Enc[ bin[2] & 0x3f ];
1563 bin += 3;
1564 len -= 3;
1565 }
1566 base64[n] = 0;
1567 return n;
1568 }
1569
1570 #define CH(x) (((x) >= 'A' && (x) <= 'Z') ? (x) - 'A' : \
1571 ((x) >= 'a' && (x) <= 'z') ? (x) - 'a' + 26 : \
1572 ((x) >= '0' && (x) <= '9') ? (x) - '0' + 52 : \
1573 ((x) == '+') ? 62 : ((x) == '/') ? 63 : -1)
1574 static const signed char HTTP_Base64Dec[256] =
1575 {
1576 CH( 0),CH( 1),CH( 2),CH( 3),CH( 4),CH( 5),CH( 6),CH( 7),CH( 8),CH( 9),
1577 CH(10),CH(11),CH(12),CH(13),CH(14),CH(15),CH(16),CH(17),CH(18),CH(19),
1578 CH(20),CH(21),CH(22),CH(23),CH(24),CH(25),CH(26),CH(27),CH(28),CH(29),
1579 CH(30),CH(31),CH(32),CH(33),CH(34),CH(35),CH(36),CH(37),CH(38),CH(39),
1580 CH(40),CH(41),CH(42),CH(43),CH(44),CH(45),CH(46),CH(47),CH(48),CH(49),
1581 CH(50),CH(51),CH(52),CH(53),CH(54),CH(55),CH(56),CH(57),CH(58),CH(59),
1582 CH(60),CH(61),CH(62),CH(63),CH(64),CH(65),CH(66),CH(67),CH(68),CH(69),
1583 CH(70),CH(71),CH(72),CH(73),CH(74),CH(75),CH(76),CH(77),CH(78),CH(79),
1584 CH(80),CH(81),CH(82),CH(83),CH(84),CH(85),CH(86),CH(87),CH(88),CH(89),
1585 CH(90),CH(91),CH(92),CH(93),CH(94),CH(95),CH(96),CH(97),CH(98),CH(99),
1586 CH(100),CH(101),CH(102),CH(103),CH(104),CH(105),CH(106),CH(107),CH(108),CH(109),
1587 CH(110),CH(111),CH(112),CH(113),CH(114),CH(115),CH(116),CH(117),CH(118),CH(119),
1588 CH(120),CH(121),CH(122),CH(123),CH(124),CH(125),CH(126),CH(127),CH(128),CH(129),
1589 CH(130),CH(131),CH(132),CH(133),CH(134),CH(135),CH(136),CH(137),CH(138),CH(139),
1590 CH(140),CH(141),CH(142),CH(143),CH(144),CH(145),CH(146),CH(147),CH(148),CH(149),
1591 CH(150),CH(151),CH(152),CH(153),CH(154),CH(155),CH(156),CH(157),CH(158),CH(159),
1592 CH(160),CH(161),CH(162),CH(163),CH(164),CH(165),CH(166),CH(167),CH(168),CH(169),
1593 CH(170),CH(171),CH(172),CH(173),CH(174),CH(175),CH(176),CH(177),CH(178),CH(179),
1594 CH(180),CH(181),CH(182),CH(183),CH(184),CH(185),CH(186),CH(187),CH(188),CH(189),
1595 CH(190),CH(191),CH(192),CH(193),CH(194),CH(195),CH(196),CH(197),CH(198),CH(199),
1596 CH(200),CH(201),CH(202),CH(203),CH(204),CH(205),CH(206),CH(207),CH(208),CH(209),
1597 CH(210),CH(211),CH(212),CH(213),CH(214),CH(215),CH(216),CH(217),CH(218),CH(219),
1598 CH(220),CH(221),CH(222),CH(223),CH(224),CH(225),CH(226),CH(227),CH(228),CH(229),
1599 CH(230),CH(231),CH(232),CH(233),CH(234),CH(235),CH(236),CH(237),CH(238),CH(239),
1600 CH(240),CH(241),CH(242),CH(243),CH(244),CH(245),CH(246),CH(247),CH(248), CH(249),
1601 CH(250),CH(251),CH(252),CH(253),CH(254),CH(255),
1602 };
1603 #undef CH
1604
1605 /***********************************************************************
1606 * HTTP_DecodeBase64
1607 */
1608 static UINT HTTP_DecodeBase64( LPCWSTR base64, LPSTR bin )
1609 {
1610 unsigned int n = 0;
1611
1612 while(*base64)
1613 {
1614 signed char in[4];
1615
1616 if (base64[0] >= ARRAYSIZE(HTTP_Base64Dec) ||
1617 ((in[0] = HTTP_Base64Dec[base64[0]]) == -1) ||
1618 base64[1] >= ARRAYSIZE(HTTP_Base64Dec) ||
1619 ((in[1] = HTTP_Base64Dec[base64[1]]) == -1))
1620 {
1621 WARN("invalid base64: %s\n", debugstr_w(base64));
1622 return 0;
1623 }
1624 if (bin)
1625 bin[n] = (unsigned char) (in[0] << 2 | in[1] >> 4);
1626 n++;
1627
1628 if ((base64[2] == '=') && (base64[3] == '='))
1629 break;
1630 if (base64[2] > ARRAYSIZE(HTTP_Base64Dec) ||
1631 ((in[2] = HTTP_Base64Dec[base64[2]]) == -1))
1632 {
1633 WARN("invalid base64: %s\n", debugstr_w(&base64[2]));
1634 return 0;
1635 }
1636 if (bin)
1637 bin[n] = (unsigned char) (in[1] << 4 | in[2] >> 2);
1638 n++;
1639
1640 if (base64[3] == '=')
1641 break;
1642 if (base64[3] > ARRAYSIZE(HTTP_Base64Dec) ||
1643 ((in[3] = HTTP_Base64Dec[base64[3]]) == -1))
1644 {
1645 WARN("invalid base64: %s\n", debugstr_w(&base64[3]));
1646 return 0;
1647 }
1648 if (bin)
1649 bin[n] = (unsigned char) (((in[2] << 6) & 0xc0) | in[3]);
1650 n++;
1651
1652 base64 += 4;
1653 }
1654
1655 return n;
1656 }
1657
1658 /***********************************************************************
1659 * HTTP_InsertAuthorization
1660 *
1661 * Insert or delete the authorization field in the request header.
1662 */
1663 static BOOL HTTP_InsertAuthorization( http_request_t *request, struct HttpAuthInfo *pAuthInfo, LPCWSTR header )
1664 {
1665 if (pAuthInfo)
1666 {
1667 static const WCHAR wszSpace[] = {' ',0};
1668 static const WCHAR wszBasic[] = {'B','a','s','i','c',0};
1669 unsigned int len;
1670 WCHAR *authorization = NULL;
1671
1672 if (pAuthInfo->auth_data_len)
1673 {
1674 /* scheme + space + base64 encoded data (3/2/1 bytes data -> 4 bytes of characters) */
1675 len = strlenW(pAuthInfo->scheme)+1+((pAuthInfo->auth_data_len+2)*4)/3;
1676 authorization = heap_alloc((len+1)*sizeof(WCHAR));
1677 if (!authorization)
1678 return FALSE;
1679
1680 strcpyW(authorization, pAuthInfo->scheme);
1681 strcatW(authorization, wszSpace);
1682 HTTP_EncodeBase64(pAuthInfo->auth_data,
1683 pAuthInfo->auth_data_len,
1684 authorization+strlenW(authorization));
1685
1686 /* clear the data as it isn't valid now that it has been sent to the
1687 * server, unless it's Basic authentication which doesn't do
1688 * connection tracking */
1689 if (strcmpiW(pAuthInfo->scheme, wszBasic))
1690 {
1691 heap_free(pAuthInfo->auth_data);
1692 pAuthInfo->auth_data = NULL;
1693 pAuthInfo->auth_data_len = 0;
1694 }
1695 }
1696
1697 TRACE("Inserting authorization: %s\n", debugstr_w(authorization));
1698
1699 HTTP_ProcessHeader(request, header, authorization, HTTP_ADDHDR_FLAG_REQ | HTTP_ADDHDR_FLAG_REPLACE);
1700 heap_free(authorization);
1701 }
1702 return TRUE;
1703 }
1704
1705 static WCHAR *build_proxy_path_url(http_request_t *req)
1706 {
1707 DWORD size, len;
1708 WCHAR *url;
1709
1710 len = strlenW(req->server->scheme_host_port);
1711 size = len + strlenW(req->path) + 1;
1712 if(*req->path != '/')
1713 size++;
1714 url = heap_alloc(size * sizeof(WCHAR));
1715 if(!url)
1716 return NULL;
1717
1718 memcpy(url, req->server->scheme_host_port, len*sizeof(WCHAR));
1719 if(*req->path != '/')
1720 url[len++] = '/';
1721
1722 strcpyW(url+len, req->path);
1723
1724 TRACE("url=%s\n", debugstr_w(url));
1725 return url;
1726 }
1727
1728 static BOOL HTTP_DomainMatches(LPCWSTR server, LPCWSTR domain)
1729 {
1730 static const WCHAR localW[] = { '<','l','o','c','a','l','>',0 };
1731 BOOL ret = FALSE;
1732
1733 if (!strcmpiW( domain, localW ) && !strchrW( server, '.' ))
1734 ret = TRUE;
1735 else if (*domain == '*')
1736 {
1737 if (domain[1] == '.')
1738 {
1739 LPCWSTR dot;
1740
1741 /* For a hostname to match a wildcard, the last domain must match
1742 * the wildcard exactly. E.g. if the wildcard is *.a.b, and the
1743 * hostname is www.foo.a.b, it matches, but a.b does not.
1744 */
1745 dot = strchrW( server, '.' );
1746 if (dot)
1747 {
1748 int len = strlenW( dot + 1 );
1749
1750 if (len > strlenW( domain + 2 ))
1751 {
1752 LPCWSTR ptr;
1753
1754 /* The server's domain is longer than the wildcard, so it
1755 * could be a subdomain. Compare the last portion of the
1756 * server's domain.
1757 */
1758 ptr = dot + len + 1 - strlenW( domain + 2 );
1759 if (!strcmpiW( ptr, domain + 2 ))
1760 {
1761 /* This is only a match if the preceding character is
1762 * a '.', i.e. that it is a matching domain. E.g.
1763 * if domain is '*.b.c' and server is 'www.ab.c' they
1764 * do not match.
1765 */
1766 ret = *(ptr - 1) == '.';
1767 }
1768 }
1769 else
1770 ret = !strcmpiW( dot + 1, domain + 2 );
1771 }
1772 }
1773 }
1774 else
1775 ret = !strcmpiW( server, domain );
1776 return ret;
1777 }
1778
1779 static BOOL HTTP_ShouldBypassProxy(appinfo_t *lpwai, LPCWSTR server)
1780 {
1781 LPCWSTR ptr;
1782 BOOL ret = FALSE;
1783
1784 if (!lpwai->proxyBypass) return FALSE;
1785 ptr = lpwai->proxyBypass;
1786 do {
1787 LPCWSTR tmp = ptr;
1788
1789 ptr = strchrW( ptr, ';' );
1790 if (!ptr)
1791 ptr = strchrW( tmp, ' ' );
1792 if (ptr)
1793 {
1794 if (ptr - tmp < INTERNET_MAX_HOST_NAME_LENGTH)
1795 {
1796 WCHAR domain[INTERNET_MAX_HOST_NAME_LENGTH];
1797
1798 memcpy( domain, tmp, (ptr - tmp) * sizeof(WCHAR) );
1799 domain[ptr - tmp] = 0;
1800 ret = HTTP_DomainMatches( server, domain );
1801 }
1802 ptr += 1;
1803 }
1804 else if (*tmp)
1805 ret = HTTP_DomainMatches( server, tmp );
1806 } while (ptr && !ret);
1807 return ret;
1808 }
1809
1810 /***********************************************************************
1811 * HTTP_DealWithProxy
1812 */
1813 static BOOL HTTP_DealWithProxy(appinfo_t *hIC, http_session_t *session, http_request_t *request)
1814 {
1815 static const WCHAR protoHttp[] = { 'h','t','t','p',0 };
1816 static const WCHAR szHttp[] = { 'h','t','t','p',':','/','/',0 };
1817 static const WCHAR szFormat[] = { 'h','t','t','p',':','/','/','%','s',0 };
1818 WCHAR buf[INTERNET_MAX_HOST_NAME_LENGTH];
1819 WCHAR protoProxy[INTERNET_MAX_URL_LENGTH];
1820 DWORD protoProxyLen = INTERNET_MAX_URL_LENGTH;
1821 WCHAR proxy[INTERNET_MAX_URL_LENGTH];
1822 static WCHAR szNul[] = { 0 };
1823 URL_COMPONENTSW UrlComponents;
1824 server_t *new_server;
1825 BOOL is_https;
1826
1827 memset( &UrlComponents, 0, sizeof UrlComponents );
1828 UrlComponents.dwStructSize = sizeof UrlComponents;
1829 UrlComponents.lpszHostName = buf;
1830 UrlComponents.dwHostNameLength = INTERNET_MAX_HOST_NAME_LENGTH;
1831
1832 if (!INTERNET_FindProxyForProtocol(hIC->proxy, protoHttp, protoProxy, &protoProxyLen))
1833 return FALSE;
1834 if( CSTR_EQUAL != CompareStringW(LOCALE_SYSTEM_DEFAULT, NORM_IGNORECASE,
1835 protoProxy,strlenW(szHttp),szHttp,strlenW(szHttp)) )
1836 sprintfW(proxy, szFormat, protoProxy);
1837 else
1838 strcpyW(proxy, protoProxy);
1839 if( !InternetCrackUrlW(proxy, 0, 0, &UrlComponents) )
1840 return FALSE;
1841 if( UrlComponents.dwHostNameLength == 0 )
1842 return FALSE;
1843
1844 if( !request->path )
1845 request->path = szNul;
1846
1847 is_https = (UrlComponents.nScheme == INTERNET_SCHEME_HTTPS);
1848 if (is_https && UrlComponents.nPort == INTERNET_INVALID_PORT_NUMBER)
1849 UrlComponents.nPort = INTERNET_DEFAULT_HTTPS_PORT;
1850
1851 new_server = get_server(UrlComponents.lpszHostName, UrlComponents.nPort, is_https, TRUE);
1852 if(!new_server)
1853 return FALSE;
1854
1855 request->proxy = new_server;
1856
1857 TRACE("proxy server=%s port=%d\n", debugstr_w(new_server->name), new_server->port);
1858 return TRUE;
1859 }
1860
1861 static DWORD HTTP_ResolveName(http_request_t *request)
1862 {
1863 server_t *server = request->proxy ? request->proxy : request->server;
1864 socklen_t addr_len;
1865 void *addr;
1866
1867 if(server->addr_len)
1868 return ERROR_SUCCESS;
1869
1870 INTERNET_SendCallback(&request->hdr, request->hdr.dwContext,
1871 INTERNET_STATUS_RESOLVING_NAME,
1872 server->name,
1873 (strlenW(server->name)+1) * sizeof(WCHAR));
1874
1875 addr_len = sizeof(server->addr);
1876 if (!GetAddress(server->name, server->port, (struct sockaddr *)&server->addr, &addr_len))
1877 return ERROR_INTERNET_NAME_NOT_RESOLVED;
1878
1879 switch(server->addr.ss_family) {
1880 case AF_INET:
1881 addr = &((struct sockaddr_in *)&server->addr)->sin_addr;
1882 break;
1883 case AF_INET6:
1884 addr = &((struct sockaddr_in6 *)&server->addr)->sin6_addr;
1885 break;
1886 default:
1887 WARN("unsupported family %d\n", server->addr.ss_family);
1888 return ERROR_INTERNET_NAME_NOT_RESOLVED;
1889 }
1890
1891 server->addr_len = addr_len;
1892 inet_ntop(server->addr.ss_family, addr, server->addr_str, sizeof(server->addr_str));
1893 INTERNET_SendCallback(&request->hdr, request->hdr.dwContext,
1894 INTERNET_STATUS_NAME_RESOLVED,
1895 server->addr_str, strlen(server->addr_str)+1);
1896
1897 TRACE("resolved %s to %s\n", debugstr_w(server->name), server->addr_str);
1898 return ERROR_SUCCESS;
1899 }
1900
1901 static BOOL HTTP_GetRequestURL(http_request_t *req, LPWSTR buf)
1902 {
1903 static const WCHAR http[] = { 'h','t','t','p',':','/','/',0 };
1904 static const WCHAR https[] = { 'h','t','t','p','s',':','/','/',0 };
1905 static const WCHAR slash[] = { '/',0 };
1906 LPHTTPHEADERW host_header;
1907 LPCWSTR scheme;
1908
1909 host_header = HTTP_GetHeader(req, hostW);
1910 if(!host_header)
1911 return FALSE;
1912
1913 EnterCriticalSection( &req->headers_section );
1914
1915 if (req->hdr.dwFlags & INTERNET_FLAG_SECURE)
1916 scheme = https;
1917 else
1918 scheme = http;
1919 strcpyW(buf, scheme);
1920 strcatW(buf, host_header->lpszValue);
1921 if (req->path[0] != '/')
1922 strcatW(buf, slash);
1923 strcatW(buf, req->path);
1924
1925 LeaveCriticalSection( &req->headers_section );
1926 return TRUE;
1927 }
1928
1929
1930 /***********************************************************************
1931 * HTTPREQ_Destroy (internal)
1932 *
1933 * Deallocate request handle
1934 *
1935 */
1936 static void HTTPREQ_Destroy(object_header_t *hdr)
1937 {
1938 http_request_t *request = (http_request_t*) hdr;
1939 DWORD i;
1940
1941 TRACE("\n");
1942
1943 if(request->hCacheFile)
1944 CloseHandle(request->hCacheFile);
1945 if(request->req_file)
1946 req_file_release(request->req_file);
1947
1948 request->headers_section.DebugInfo->Spare[0] = 0;
1949 DeleteCriticalSection( &request->headers_section );
1950 request->read_section.DebugInfo->Spare[0] = 0;
1951 DeleteCriticalSection( &request->read_section );
1952 WININET_Release(&request->session->hdr);
1953
1954 destroy_authinfo(request->authInfo);
1955 destroy_authinfo(request->proxyAuthInfo);
1956
1957 if(request->server)
1958 server_release(request->server);
1959 if(request->proxy)
1960 server_release(request->proxy);
1961
1962 heap_free(request->path);
1963 heap_free(request->verb);
1964 heap_free(request->version);
1965 heap_free(request->statusText);
1966
1967 for (i = 0; i < request->nCustHeaders; i++)
1968 {
1969 heap_free(request->custHeaders[i].lpszField);
1970 heap_free(request->custHeaders[i].lpszValue);
1971 }
1972 destroy_data_stream(request->data_stream);
1973 heap_free(request->custHeaders);
1974 }
1975
1976 static void http_release_netconn(http_request_t *req, BOOL reuse)
1977 {
1978 TRACE("%p %p %x\n",req, req->netconn, reuse);
1979
1980 if(!is_valid_netconn(req->netconn))
1981 return;
1982
1983 if(reuse && req->netconn->keep_alive) {
1984 BOOL run_collector;
1985
1986 EnterCriticalSection(&connection_pool_cs);
1987
1988 list_add_head(&req->netconn->server->conn_pool, &req->netconn->pool_entry);
1989 req->netconn->keep_until = GetTickCount64() + COLLECT_TIME;
1990 req->netconn = NULL;
1991
1992 run_collector = !collector_running;
1993 collector_running = TRUE;
1994
1995 LeaveCriticalSection(&connection_pool_cs);
1996
1997 if(run_collector) {
1998 HANDLE thread = NULL;
1999 HMODULE module;
2000
2001 GetModuleHandleExW(GET_MODULE_HANDLE_EX_FLAG_FROM_ADDRESS, (const WCHAR*)WININET_hModule, &module);
2002 if(module)
2003 thread = CreateThread(NULL, 0, collect_connections_proc, NULL, 0, NULL);
2004 if(!thread) {
2005 EnterCriticalSection(&connection_pool_cs);
2006 collector_running = FALSE;
2007 LeaveCriticalSection(&connection_pool_cs);
2008
2009 if(module)
2010 FreeLibrary(module);
2011 }
2012 else
2013 CloseHandle(thread);
2014 }
2015 return;
2016 }
2017
2018 INTERNET_SendCallback(&req->hdr, req->hdr.dwContext,
2019 INTERNET_STATUS_CLOSING_CONNECTION, 0, 0);
2020
2021 close_netconn(req->netconn);
2022
2023 INTERNET_SendCallback(&req->hdr, req->hdr.dwContext,
2024 INTERNET_STATUS_CONNECTION_CLOSED, 0, 0);
2025 }
2026
2027 static BOOL HTTP_KeepAlive(http_request_t *request)
2028 {
2029 WCHAR szVersion[10];
2030 WCHAR szConnectionResponse[20];
2031 DWORD dwBufferSize = sizeof(szVersion);
2032 BOOL keepalive = FALSE;
2033
2034 /* as per RFC 2068, S8.1.2.1, if the client is HTTP/1.1 then assume that
2035 * the connection is keep-alive by default */
2036 if (HTTP_HttpQueryInfoW(request, HTTP_QUERY_VERSION, szVersion, &dwBufferSize, NULL) == ERROR_SUCCESS
2037 && !strcmpiW(szVersion, g_szHttp1_1))
2038 {
2039 keepalive = TRUE;
2040 }
2041
2042 dwBufferSize = sizeof(szConnectionResponse);
2043 if (HTTP_HttpQueryInfoW(request, HTTP_QUERY_PROXY_CONNECTION, szConnectionResponse, &dwBufferSize, NULL) == ERROR_SUCCESS
2044 || HTTP_HttpQueryInfoW(request, HTTP_QUERY_CONNECTION, szConnectionResponse, &dwBufferSize, NULL) == ERROR_SUCCESS)
2045 {
2046 keepalive = !strcmpiW(szConnectionResponse, szKeepAlive);
2047 }
2048
2049 return keepalive;
2050 }
2051
2052 static void HTTPREQ_CloseConnection(object_header_t *hdr)
2053 {
2054 http_request_t *req = (http_request_t*)hdr;
2055
2056 http_release_netconn(req, drain_content(req, FALSE));
2057 }
2058
2059 static DWORD str_to_buffer(const WCHAR *str, void *buffer, DWORD *size, BOOL unicode)
2060 {
2061 int len;
2062 if (unicode)
2063 {
2064 WCHAR *buf = buffer;
2065
2066 if (str) len = strlenW(str);
2067 else len = 0;
2068 if (*size < (len + 1) * sizeof(WCHAR))
2069 {
2070 *size = (len + 1) * sizeof(WCHAR);
2071 return ERROR_INSUFFICIENT_BUFFER;
2072 }
2073 if (str) strcpyW(buf, str);
2074 else buf[0] = 0;
2075
2076 *size = len;
2077 return ERROR_SUCCESS;
2078 }
2079 else
2080 {
2081 char *buf = buffer;
2082
2083 if (str) len = WideCharToMultiByte(CP_ACP, 0, str, -1, NULL, 0, NULL, NULL);
2084 else len = 1;
2085 if (*size < len)
2086 {
2087 *size = len;
2088 return ERROR_INSUFFICIENT_BUFFER;
2089 }
2090 if (str) WideCharToMultiByte(CP_ACP, 0, str, -1, buf, *size, NULL, NULL);
2091 else buf[0] = 0;
2092
2093 *size = len - 1;
2094 return ERROR_SUCCESS;
2095 }
2096 }
2097
2098 static DWORD HTTPREQ_QueryOption(object_header_t *hdr, DWORD option, void *buffer, DWORD *size, BOOL unicode)
2099 {
2100 http_request_t *req = (http_request_t*)hdr;
2101
2102 switch(option) {
2103 case INTERNET_OPTION_DIAGNOSTIC_SOCKET_INFO:
2104 {
2105 INTERNET_DIAGNOSTIC_SOCKET_INFO *info = buffer;
2106
2107 FIXME("INTERNET_DIAGNOSTIC_SOCKET_INFO stub\n");
2108
2109 if (*size < sizeof(INTERNET_DIAGNOSTIC_SOCKET_INFO))
2110 return ERROR_INSUFFICIENT_BUFFER;
2111 *size = sizeof(INTERNET_DIAGNOSTIC_SOCKET_INFO);
2112 /* FIXME: can't get a SOCKET from our connection since we don't use
2113 * winsock
2114 */
2115 info->Socket = 0;
2116 /* FIXME: get source port from req->netConnection */
2117 info->SourcePort = 0;
2118 info->DestPort = req->server->port;
2119 info->Flags = 0;
2120 if (HTTP_KeepAlive(req))
2121 info->Flags |= IDSI_FLAG_KEEP_ALIVE;
2122 if (req->proxy)
2123 info->Flags |= IDSI_FLAG_PROXY;
2124 if (is_valid_netconn(req->netconn) && req->netconn->secure)
2125 info->Flags |= IDSI_FLAG_SECURE;
2126
2127 return ERROR_SUCCESS;
2128 }
2129
2130 case 98:
2131 TRACE("Queried undocumented option 98, forwarding to INTERNET_OPTION_SECURITY_FLAGS\n");
2132 /* fall through */
2133 case INTERNET_OPTION_SECURITY_FLAGS:
2134 {
2135 DWORD flags;
2136
2137 if (*size < sizeof(ULONG))
2138 return ERROR_INSUFFICIENT_BUFFER;
2139
2140 *size = sizeof(DWORD);
2141 flags = is_valid_netconn(req->netconn) ? req->netconn->security_flags : req->security_flags | req->server->security_flags;
2142 *(DWORD *)buffer = flags;
2143
2144 TRACE("INTERNET_OPTION_SECURITY_FLAGS %x\n", flags);
2145 return ERROR_SUCCESS;
2146 }
2147
2148 case INTERNET_OPTION_HANDLE_TYPE:
2149 TRACE("INTERNET_OPTION_HANDLE_TYPE\n");
2150
2151 if (*size < sizeof(ULONG))
2152 return ERROR_INSUFFICIENT_BUFFER;
2153
2154 *size = sizeof(DWORD);
2155 *(DWORD*)buffer = INTERNET_HANDLE_TYPE_HTTP_REQUEST;
2156 return ERROR_SUCCESS;
2157
2158 case INTERNET_OPTION_URL: {
2159 static const WCHAR httpW[] = {'h','t','t','p',':','/','/',0};
2160 WCHAR url[INTERNET_MAX_URL_LENGTH];
2161 HTTPHEADERW *host;
2162
2163 TRACE("INTERNET_OPTION_URL\n");
2164
2165 EnterCriticalSection( &req->headers_section );
2166 host = HTTP_GetHeader(req, hostW);
2167 strcpyW(url, httpW);
2168 strcatW(url, host->lpszValue);
2169 strcatW(url, req->path);
2170 LeaveCriticalSection( &req->headers_section );
2171
2172 TRACE("INTERNET_OPTION_URL: %s\n",debugstr_w(url));
2173 return str_to_buffer(url, buffer, size, unicode);
2174 }
2175 case INTERNET_OPTION_USER_AGENT:
2176 return str_to_buffer(req->session->appInfo->agent, buffer, size, unicode);
2177 case INTERNET_OPTION_USERNAME:
2178 return str_to_buffer(req->session->userName, buffer, size, unicode);
2179 case INTERNET_OPTION_PASSWORD:
2180 return str_to_buffer(req->session->password, buffer, size, unicode);
2181 case INTERNET_OPTION_PROXY_USERNAME:
2182 return str_to_buffer(req->session->appInfo->proxyUsername, buffer, size, unicode);
2183 case INTERNET_OPTION_PROXY_PASSWORD:
2184 return str_to_buffer(req->session->appInfo->proxyPassword, buffer, size, unicode);
2185
2186 case INTERNET_OPTION_CACHE_TIMESTAMPS: {
2187 INTERNET_CACHE_ENTRY_INFOW *info;
2188 INTERNET_CACHE_TIMESTAMPS *ts = buffer;
2189 WCHAR url[INTERNET_MAX_URL_LENGTH];
2190 DWORD nbytes, error;
2191 BOOL ret;
2192
2193 TRACE("INTERNET_OPTION_CACHE_TIMESTAMPS\n");
2194
2195 if (*size < sizeof(*ts))
2196 {
2197 *size = sizeof(*ts);
2198 return ERROR_INSUFFICIENT_BUFFER;
2199 }
2200 nbytes = 0;
2201 HTTP_GetRequestURL(req, url);
2202 ret = GetUrlCacheEntryInfoW(url, NULL, &nbytes);
2203 error = GetLastError();
2204 if (!ret && error == ERROR_INSUFFICIENT_BUFFER)
2205 {
2206 if (!(info = heap_alloc(nbytes)))
2207 return ERROR_OUTOFMEMORY;
2208
2209 GetUrlCacheEntryInfoW(url, info, &nbytes);
2210
2211 ts->ftExpires = info->ExpireTime;
2212 ts->ftLastModified = info->LastModifiedTime;
2213
2214 heap_free(info);
2215 *size = sizeof(*ts);
2216 return ERROR_SUCCESS;
2217 }
2218 return error;
2219 }
2220
2221 case INTERNET_OPTION_DATAFILE_NAME: {
2222 DWORD req_size;
2223
2224 TRACE("INTERNET_OPTION_DATAFILE_NAME\n");
2225
2226 if(!req->req_file) {
2227 *size = 0;
2228 return ERROR_INTERNET_ITEM_NOT_FOUND;
2229 }
2230
2231 if(unicode) {
2232 req_size = (lstrlenW(req->req_file->file_name)+1) * sizeof(WCHAR);
2233 if(*size < req_size)
2234 return ERROR_INSUFFICIENT_BUFFER;
2235
2236 *size = req_size;
2237 memcpy(buffer, req->req_file->file_name, *size);
2238 return ERROR_SUCCESS;
2239 }else {
2240 req_size = WideCharToMultiByte(CP_ACP, 0, req->req_file->file_name, -1, NULL, 0, NULL, NULL);
2241 if (req_size > *size)
2242 return ERROR_INSUFFICIENT_BUFFER;
2243
2244 *size = WideCharToMultiByte(CP_ACP, 0, req->req_file->file_name,
2245 -1, buffer, *size, NULL, NULL);
2246 return ERROR_SUCCESS;
2247 }
2248 }
2249
2250 case INTERNET_OPTION_SECURITY_CERTIFICATE_STRUCT: {
2251 PCCERT_CONTEXT context;
2252
2253 if(!req->netconn)
2254 return ERROR_INTERNET_INVALID_OPERATION;
2255
2256 if(*size < sizeof(INTERNET_CERTIFICATE_INFOA)) {
2257 *size = sizeof(INTERNET_CERTIFICATE_INFOA);
2258 return ERROR_INSUFFICIENT_BUFFER;
2259 }
2260
2261 context = (PCCERT_CONTEXT)NETCON_GetCert(req->netconn);
2262 if(context) {
2263 INTERNET_CERTIFICATE_INFOA *info = (INTERNET_CERTIFICATE_INFOA*)buffer;
2264 DWORD len;
2265
2266 memset(info, 0, sizeof(*info));
2267 info->ftExpiry = context->pCertInfo->NotAfter;
2268 info->ftStart = context->pCertInfo->NotBefore;
2269 len = CertNameToStrA(context->dwCertEncodingType,
2270 &context->pCertInfo->Subject, CERT_SIMPLE_NAME_STR|CERT_NAME_STR_CRLF_FLAG, NULL, 0);
2271 info->lpszSubjectInfo = LocalAlloc(0, len);
2272 if(info->lpszSubjectInfo)
2273 CertNameToStrA(context->dwCertEncodingType,
2274 &context->pCertInfo->Subject, CERT_SIMPLE_NAME_STR|CERT_NAME_STR_CRLF_FLAG,
2275 info->lpszSubjectInfo, len);
2276 len = CertNameToStrA(context->dwCertEncodingType,
2277 &context->pCertInfo->Issuer, CERT_SIMPLE_NAME_STR|CERT_NAME_STR_CRLF_FLAG, NULL, 0);
2278 info->lpszIssuerInfo = LocalAlloc(0, len);
2279 if(info->lpszIssuerInfo)
2280 CertNameToStrA(context->dwCertEncodingType,
2281 &context->pCertInfo->Issuer, CERT_SIMPLE_NAME_STR|CERT_NAME_STR_CRLF_FLAG,
2282 info->lpszIssuerInfo, len);
2283 info->dwKeySize = NETCON_GetCipherStrength(req->netconn);
2284 CertFreeCertificateContext(context);
2285 return ERROR_SUCCESS;
2286 }
2287 return ERROR_NOT_SUPPORTED;
2288 }
2289 case INTERNET_OPTION_CONNECT_TIMEOUT:
2290 if (*size < sizeof(DWORD))
2291 return ERROR_INSUFFICIENT_BUFFER;
2292
2293 *size = sizeof(DWORD);
2294 *(DWORD *)buffer = req->connect_timeout;
2295 return ERROR_SUCCESS;
2296 case INTERNET_OPTION_REQUEST_FLAGS: {
2297 DWORD flags = 0;
2298
2299 if (*size < sizeof(DWORD))
2300 return ERROR_INSUFFICIENT_BUFFER;
2301
2302 /* FIXME: Add support for:
2303 * INTERNET_REQFLAG_FROM_CACHE
2304 * INTERNET_REQFLAG_CACHE_WRITE_DISABLED
2305 */
2306
2307 if(req->proxy)
2308 flags |= INTERNET_REQFLAG_VIA_PROXY;
2309 if(!req->status_code)
2310 flags |= INTERNET_REQFLAG_NO_HEADERS;
2311
2312 TRACE("INTERNET_OPTION_REQUEST_FLAGS returning %x\n", flags);
2313
2314 *size = sizeof(DWORD);
2315 *(DWORD*)buffer = flags;
2316 return ERROR_SUCCESS;
2317 }
2318 }
2319
2320 return INET_QueryOption(hdr, option, buffer, size, unicode);
2321 }
2322
2323 static DWORD HTTPREQ_SetOption(object_header_t *hdr, DWORD option, void *buffer, DWORD size)
2324 {
2325 http_request_t *req = (http_request_t*)hdr;
2326
2327 switch(option) {
2328 case 99: /* Undocumented, seems to be INTERNET_OPTION_SECURITY_FLAGS with argument validation */
2329 TRACE("Undocumented option 99\n");
2330
2331 if (!buffer || size != sizeof(DWORD))
2332 return ERROR_INVALID_PARAMETER;
2333 if(*(DWORD*)buffer & ~SECURITY_SET_MASK)
2334 return ERROR_INTERNET_OPTION_NOT_SETTABLE;
2335
2336 /* fall through */
2337 case INTERNET_OPTION_SECURITY_FLAGS:
2338 {
2339 DWORD flags;
2340
2341 if (!buffer || size != sizeof(DWORD))
2342 return ERROR_INVALID_PARAMETER;
2343 flags = *(DWORD *)buffer;
2344 TRACE("INTERNET_OPTION_SECURITY_FLAGS %08x\n", flags);
2345 flags &= SECURITY_SET_MASK;
2346 req->security_flags |= flags;
2347 if(is_valid_netconn(req->netconn))
2348 req->netconn->security_flags |= flags;
2349 return ERROR_SUCCESS;
2350 }
2351 case INTERNET_OPTION_CONNECT_TIMEOUT:
2352 if (!buffer || size != sizeof(DWORD)) return ERROR_INVALID_PARAMETER;
2353 req->connect_timeout = *(DWORD *)buffer;
2354 return ERROR_SUCCESS;
2355
2356 case INTERNET_OPTION_SEND_TIMEOUT:
2357 if (!buffer || size != sizeof(DWORD)) return ERROR_INVALID_PARAMETER;
2358 req->send_timeout = *(DWORD *)buffer;
2359 return ERROR_SUCCESS;
2360
2361 case INTERNET_OPTION_RECEIVE_TIMEOUT:
2362 if (!buffer || size != sizeof(DWORD)) return ERROR_INVALID_PARAMETER;
2363 req->receive_timeout = *(DWORD *)buffer;
2364 return ERROR_SUCCESS;
2365
2366 case INTERNET_OPTION_USERNAME:
2367 heap_free(req->session->userName);
2368 if (!(req->session->userName = heap_strdupW(buffer))) return ERROR_OUTOFMEMORY;
2369 return ERROR_SUCCESS;
2370
2371 case INTERNET_OPTION_PASSWORD:
2372 heap_free(req->session->password);
2373 if (!(req->session->password = heap_strdupW(buffer))) return ERROR_OUTOFMEMORY;
2374 return ERROR_SUCCESS;
2375
2376 case INTERNET_OPTION_PROXY_USERNAME:
2377 heap_free(req->session->appInfo->proxyUsername);
2378 if (!(req->session->appInfo->proxyUsername = heap_strdupW(buffer))) return ERROR_OUTOFMEMORY;
2379 return ERROR_SUCCESS;
2380
2381 case INTERNET_OPTION_PROXY_PASSWORD:
2382 heap_free(req->session->appInfo->proxyPassword);
2383 if (!(req->session->appInfo->proxyPassword = heap_strdupW(buffer))) return ERROR_OUTOFMEMORY;
2384 return ERROR_SUCCESS;
2385
2386 case INTERNET_OPTION_HTTP_DECODING:
2387 if(size != sizeof(BOOL))
2388 return ERROR_INVALID_PARAMETER;
2389 req->decoding = *(BOOL*)buffer;
2390 return ERROR_SUCCESS;
2391 }
2392
2393 return INET_SetOption(hdr, option, buffer, size);
2394 }
2395
2396 static void commit_cache_entry(http_request_t *req)
2397 {
2398 WCHAR url[INTERNET_MAX_URL_LENGTH];
2399
2400 TRACE("%p\n", req);
2401
2402 CloseHandle(req->hCacheFile);
2403 req->hCacheFile = NULL;
2404
2405 if(HTTP_GetRequestURL(req, url)) {
2406 WCHAR *header;
2407 DWORD header_len;
2408 BOOL res;
2409
2410 header = build_response_header(req, TRUE);
2411 header_len = (header ? strlenW(header) : 0);
2412 res = CommitUrlCacheEntryW(url, req->req_file->file_name, req->expires,
2413 req->last_modified, NORMAL_CACHE_ENTRY,
2414 header, header_len, NULL, 0);
2415 if(res)
2416 req->req_file->is_committed = TRUE;
2417 else
2418 WARN("CommitUrlCacheEntry failed: %u\n", GetLastError());
2419 heap_free(header);
2420 }
2421 }
2422
2423 static void create_cache_entry(http_request_t *req)
2424 {
2425 static const WCHAR no_cacheW[] = {'n','o','-','c','a','c','h','e',0};
2426 static const WCHAR no_storeW[] = {'n','o','-','s','t','o','r','e',0};
2427
2428 WCHAR url[INTERNET_MAX_URL_LENGTH];
2429 WCHAR file_name[MAX_PATH+1];
2430 BOOL b = TRUE;
2431
2432 /* FIXME: We should free previous cache file earlier */
2433 if(req->req_file) {
2434 req_file_release(req->req_file);
2435 req->req_file = NULL;
2436 }
2437 if(req->hCacheFile) {
2438 CloseHandle(req->hCacheFile);
2439 req->hCacheFile = NULL;
2440 }
2441
2442 if(req->hdr.dwFlags & INTERNET_FLAG_NO_CACHE_WRITE)
2443 b = FALSE;
2444
2445 if(b) {
2446 int header_idx;
2447
2448 EnterCriticalSection( &req->headers_section );
2449
2450 header_idx = HTTP_GetCustomHeaderIndex(req, szCache_Control, 0, FALSE);
2451 if(header_idx != -1) {
2452 WCHAR *ptr;
2453
2454 for(ptr=req->custHeaders[header_idx].lpszValue; *ptr; ) {
2455 WCHAR *end;
2456
2457 while(*ptr==' ' || *ptr=='\t')
2458 ptr++;
2459
2460 end = strchrW(ptr, ',');
2461 if(!end)
2462 end = ptr + strlenW(ptr);
2463
2464 if(!strncmpiW(ptr, no_cacheW, sizeof(no_cacheW)/sizeof(*no_cacheW)-1)
2465 || !strncmpiW(ptr, no_storeW, sizeof(no_storeW)/sizeof(*no_storeW)-1)) {
2466 b = FALSE;
2467 break;
2468 }
2469
2470 ptr = end;
2471 if(*ptr == ',')
2472 ptr++;
2473 }
2474 }
2475
2476 LeaveCriticalSection( &req->headers_section );
2477 }
2478
2479 if(!b) {
2480 if(!(req->hdr.dwFlags & INTERNET_FLAG_NEED_FILE))
2481 return;
2482
2483 FIXME("INTERNET_FLAG_NEED_FILE is not supported correctly\n");
2484 }
2485
2486 b = HTTP_GetRequestURL(req, url);
2487 if(!b) {
2488 WARN("Could not get URL\n");
2489 return;
2490 }
2491
2492 b = CreateUrlCacheEntryW(url, req->contentLength == ~0u ? 0 : req->contentLength, NULL, file_name, 0);
2493 if(!b) {
2494 WARN("Could not create cache entry: %08x\n", GetLastError());
2495 return;
2496 }
2497
2498 create_req_file(file_name, &req->req_file);
2499
2500 req->hCacheFile = CreateFileW(file_name, GENERIC_WRITE, FILE_SHARE_READ|FILE_SHARE_WRITE,
2501 NULL, CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
2502 if(req->hCacheFile == INVALID_HANDLE_VALUE) {
2503 WARN("Could not create file: %u\n", GetLastError());
2504 req->hCacheFile = NULL;
2505 return;
2506 }
2507
2508 if(req->read_size) {
2509 DWORD written;
2510
2511 b = WriteFile(req->hCacheFile, req->read_buf+req->read_pos, req->read_size, &written, NULL);
2512 if(!b)
2513 FIXME("WriteFile failed: %u\n", GetLastError());
2514
2515 if(req->data_stream->vtbl->end_of_data(req->data_stream, req))
2516 commit_cache_entry(req);
2517 }
2518 }
2519
2520 /* read some more data into the read buffer (the read section must be held) */
2521 static DWORD read_more_data( http_request_t *req, int maxlen )
2522 {
2523 DWORD res;
2524 int len;
2525
2526 if (req->read_pos)
2527 {
2528 /* move existing data to the start of the buffer */
2529 if(req->read_size)
2530 memmove( req->read_buf, req->read_buf + req->read_pos, req->read_size );
2531 req->read_pos = 0;
2532 }
2533
2534 if (maxlen == -1) maxlen = sizeof(req->read_buf);
2535
2536 res = NETCON_recv( req->netconn, req->read_buf + req->read_size,
2537 maxlen - req->read_size, BLOCKING_ALLOW, &len );
2538 if(res == ERROR_SUCCESS)
2539 req->read_size += len;
2540
2541 return res;
2542 }
2543
2544 /* remove some amount of data from the read buffer (the read section must be held) */
2545 static void remove_data( http_request_t *req, int count )
2546 {
2547 if (!(req->read_size -= count)) req->read_pos = 0;
2548 else req->read_pos += count;
2549 }
2550
2551 static DWORD read_line( http_request_t *req, LPSTR buffer, DWORD *len )
2552 {
2553 int count, bytes_read, pos = 0;
2554 DWORD res;
2555
2556 EnterCriticalSection( &req->read_section );
2557 for (;;)
2558 {
2559 BYTE *eol = memchr( req->read_buf + req->read_pos, '\n', req->read_size );
2560
2561 if (eol)
2562 {
2563 count = eol - (req->read_buf + req->read_pos);
2564 bytes_read = count + 1;
2565 }
2566 else count = bytes_read = req->read_size;
2567
2568 count = min( count, *len - pos );
2569 memcpy( buffer + pos, req->read_buf + req->read_pos, count );
2570 pos += count;
2571 remove_data( req, bytes_read );
2572 if (eol) break;
2573
2574 if ((res = read_more_data( req, -1 )))
2575 {
2576 WARN( "read failed %u\n", res );
2577 LeaveCriticalSection( &req->read_section );
2578 return res;
2579 }
2580 if (!req->read_size)
2581 {
2582 *len = 0;
2583 TRACE( "returning empty string\n" );
2584 LeaveCriticalSection( &req->read_section );
2585 return ERROR_SUCCESS;
2586 }
2587 }
2588 LeaveCriticalSection( &req->read_section );
2589
2590 if (pos < *len)
2591 {
2592 if (pos && buffer[pos - 1] == '\r') pos--;
2593 *len = pos + 1;
2594 }
2595 buffer[*len - 1] = 0;
2596 TRACE( "returning %s\n", debugstr_a(buffer));
2597 return ERROR_SUCCESS;
2598 }
2599
2600 /* check if we have reached the end of the data to read (the read section must be held) */
2601 static BOOL end_of_read_data( http_request_t *req )
2602 {
2603 return !req->read_size && req->data_stream->vtbl->end_of_data(req->data_stream, req);
2604 }
2605
2606 static DWORD read_http_stream(http_request_t *req, BYTE *buf, DWORD size, DWORD *read, blocking_mode_t blocking_mode)
2607 {
2608 DWORD res;
2609
2610 res = req->data_stream->vtbl->read(req->data_stream, req, buf, size, read, blocking_mode);
2611 assert(*read <= size);
2612
2613 if(req->hCacheFile) {
2614 if(*read) {
2615 BOOL bres;
2616 DWORD written;
2617
2618 bres = WriteFile(req->hCacheFile, buf, *read, &written, NULL);
2619 if(!bres)
2620 FIXME("WriteFile failed: %u\n", GetLastError());
2621 }
2622
2623 if(req->data_stream->vtbl->end_of_data(req->data_stream, req))
2624 commit_cache_entry(req);
2625 }
2626
2627 return res;
2628 }
2629
2630 /* fetch some more data into the read buffer (the read section must be held) */
2631 static DWORD refill_read_buffer(http_request_t *req, blocking_mode_t blocking_mode, DWORD *read_bytes)
2632 {
2633 DWORD res, read=0;
2634
2635 if(req->read_size == sizeof(req->read_buf))
2636 return ERROR_SUCCESS;
2637
2638 if(req->read_pos) {
2639 if(req->read_size)
2640 memmove(req->read_buf, req->read_buf+req->read_pos, req->read_size);
2641 req->read_pos = 0;
2642 }
2643
2644 res = read_http_stream(req, req->read_buf+req->read_size, sizeof(req->read_buf) - req->read_size,
2645 &read, blocking_mode);
2646 req->read_size += read;
2647
2648 TRACE("read %u bytes, read_size %u\n", read, req->read_size);
2649 if(read_bytes)
2650 *read_bytes = read;
2651 return res;
2652 }
2653
2654 /* return the size of data available to be read immediately (the read section must be held) */
2655 static DWORD get_avail_data( http_request_t *req )
2656 {
2657 return req->read_size + req->data_stream->vtbl->get_avail_data(req->data_stream, req);
2658 }
2659
2660 static DWORD netconn_get_avail_data(data_stream_t *stream, http_request_t *req)
2661 {
2662 netconn_stream_t *netconn_stream = (netconn_stream_t*)stream;
2663 DWORD avail = 0;
2664
2665 if(is_valid_netconn(req->netconn))
2666 NETCON_query_data_available(req->netconn, &avail);
2667 return netconn_stream->content_length == ~0u
2668 ? avail
2669 : min(avail, netconn_stream->content_length-netconn_stream->content_read);
2670 }
2671
2672 static BOOL netconn_end_of_data(data_stream_t *stream, http_request_t *req)
2673 {
2674 netconn_stream_t *netconn_stream = (netconn_stream_t*)stream;
2675 return netconn_stream->content_read == netconn_stream->content_length || !is_valid_netconn(req->netconn);
2676 }
2677
2678 static DWORD netconn_read(data_stream_t *stream, http_request_t *req, BYTE *buf, DWORD size,
2679 DWORD *read, blocking_mode_t blocking_mode)
2680 {
2681 netconn_stream_t *netconn_stream = (netconn_stream_t*)stream;
2682 DWORD res = ERROR_SUCCESS;
2683 int len = 0;
2684
2685 size = min(size, netconn_stream->content_length-netconn_stream->content_read);
2686
2687 if(size && is_valid_netconn(req->netconn)) {
2688 if((res = NETCON_recv(req->netconn, buf, size, blocking_mode, &len))) {
2689 len = 0;
2690 if(blocking_mode == BLOCKING_DISALLOW && res == WSAEWOULDBLOCK)
2691 res = ERROR_SUCCESS;
2692 else
2693 netconn_stream->content_length = netconn_stream->content_read;
2694 }else if(!len) {
2695 netconn_stream->content_length = netconn_stream->content_read;
2696 }
2697 }
2698
2699 netconn_stream->content_read += *read = len;
2700 TRACE("read %u bytes\n", len);
2701 return res;
2702 }
2703
2704 static BOOL netconn_drain_content(data_stream_t *stream, http_request_t *req)
2705 {
2706 netconn_stream_t *netconn_stream = (netconn_stream_t*)stream;
2707 BYTE buf[1024];
2708 int len;
2709
2710 if(netconn_end_of_data(stream, req))
2711 return TRUE;
2712
2713 do {
2714 if(NETCON_recv(req->netconn, buf, sizeof(buf), BLOCKING_DISALLOW, &len) != ERROR_SUCCESS)
2715 return FALSE;
2716
2717 netconn_stream->content_read += len;
2718 }while(netconn_stream->content_read < netconn_stream->content_length);
2719
2720 return TRUE;
2721 }
2722
2723 static void netconn_destroy(data_stream_t *stream)
2724 {
2725 }
2726
2727 static const data_stream_vtbl_t netconn_stream_vtbl = {
2728 netconn_get_avail_data,
2729 netconn_end_of_data,
2730 netconn_read,
2731 netconn_drain_content,
2732 netconn_destroy
2733 };
2734
2735 /* read some more data into the read buffer (the read section must be held) */
2736 static DWORD read_more_chunked_data(chunked_stream_t *stream, http_request_t *req, int maxlen)
2737 {
2738 DWORD res;
2739 int len;
2740
2741 assert(!stream->end_of_data);
2742
2743 if (stream->buf_pos)
2744 {
2745 /* move existing data to the start of the buffer */
2746 if(stream->buf_size)
2747 memmove(stream->buf, stream->buf + stream->buf_pos, stream->buf_size);
2748 stream->buf_pos = 0;
2749 }
2750
2751 if (maxlen == -1) maxlen = sizeof(stream->buf);
2752
2753 res = NETCON_recv( req->netconn, stream->buf + stream->buf_size,
2754 maxlen - stream->buf_size, BLOCKING_ALLOW, &len );
2755 if(res == ERROR_SUCCESS)
2756 stream->buf_size += len;
2757
2758 return res;
2759 }
2760
2761 /* remove some amount of data from the read buffer (the read section must be held) */
2762 static void remove_chunked_data(chunked_stream_t *stream, int count)
2763 {
2764 if (!(stream->buf_size -= count)) stream->buf_pos = 0;
2765 else stream->buf_pos += count;
2766 }
2767
2768 /* discard data contents until we reach end of line (the read section must be held) */
2769 static DWORD discard_chunked_eol(chunked_stream_t *stream, http_request_t *req)
2770 {
2771 DWORD res;
2772
2773 do
2774 {
2775 BYTE *eol = memchr(stream->buf + stream->buf_pos, '\n', stream->buf_size);
2776 if (eol)
2777 {
2778 remove_chunked_data(stream, (eol + 1) - (stream->buf + stream->buf_pos));
2779 break;
2780 }
2781 stream->buf_pos = stream->buf_size = 0; /* discard everything */
2782 if ((res = read_more_chunked_data(stream, req, -1)) != ERROR_SUCCESS) return res;
2783 } while (stream->buf_size);
2784 return ERROR_SUCCESS;
2785 }
2786
2787 /* read the size of the next chunk (the read section must be held) */
2788 static DWORD start_next_chunk(chunked_stream_t *stream, http_request_t *req)
2789 {
2790 DWORD chunk_size = 0, res;
2791
2792 assert(!stream->chunk_size || stream->chunk_size == ~0u);
2793
2794 if (stream->end_of_data) return ERROR_SUCCESS;
2795
2796 /* read terminator for the previous chunk */
2797 if(!stream->chunk_size && (res = discard_chunked_eol(stream, req)) != ERROR_SUCCESS)
2798 return res;
2799
2800 for (;;)
2801 {
2802 while (stream->buf_size)
2803 {
2804 char ch = stream->buf[stream->buf_pos];
2805 if (ch >= '0' && ch <= '9') chunk_size = chunk_size * 16 + ch - '0';
2806 else if (ch >= 'a' && ch <= 'f') chunk_size = chunk_size * 16 + ch - 'a' + 10;
2807 else if (ch >= 'A' && ch <= 'F') chunk_size = chunk_size * 16 + ch - 'A' + 10;
2808 else if (ch == ';' || ch == '\r' || ch == '\n')
2809 {
2810 TRACE( "reading %u byte chunk\n", chunk_size );
2811 stream->chunk_size = chunk_size;
2812 if (req->contentLength == ~0u) req->contentLength = chunk_size;
2813 else req->contentLength += chunk_size;
2814
2815 if (!chunk_size) stream->end_of_data = TRUE;
2816 return discard_chunked_eol(stream, req);
2817 }
2818 remove_chunked_data(stream, 1);
2819 }
2820 if ((res = read_more_chunked_data(stream, req, -1)) != ERROR_SUCCESS) return res;
2821 if (!stream->buf_size)
2822 {
2823 stream->chunk_size = 0;
2824 return ERROR_SUCCESS;
2825 }
2826 }
2827 }
2828
2829 static DWORD chunked_get_avail_data(data_stream_t *stream, http_request_t *req)
2830 {
2831 /* Allow reading only from read buffer */
2832 return 0;
2833 }
2834
2835 static BOOL chunked_end_of_data(data_stream_t *stream, http_request_t *req)
2836 {
2837 chunked_stream_t *chunked_stream = (chunked_stream_t*)stream;
2838 return chunked_stream->end_of_data;
2839 }
2840
2841 static DWORD chunked_read(data_stream_t *stream, http_request_t *req, BYTE *buf, DWORD size,
2842 DWORD *read, blocking_mode_t blocking_mode)
2843 {
2844 chunked_stream_t *chunked_stream = (chunked_stream_t*)stream;
2845 DWORD read_bytes = 0, ret_read = 0, res = ERROR_SUCCESS;
2846
2847 if(!chunked_stream->chunk_size || chunked_stream->chunk_size == ~0u) {
2848 res = start_next_chunk(chunked_stream, req);
2849 if(res != ERROR_SUCCESS)
2850 return res;
2851 }
2852
2853 while(size && chunked_stream->chunk_size && !chunked_stream->end_of_data) {
2854 if(chunked_stream->buf_size) {
2855 read_bytes = min(size, min(chunked_stream->buf_size, chunked_stream->chunk_size));
2856
2857 /* this could block */
2858 if(blocking_mode == BLOCKING_DISALLOW && read_bytes == chunked_stream->chunk_size)
2859 break;
2860
2861 memcpy(buf+ret_read, chunked_stream->buf+chunked_stream->buf_pos, read_bytes);
2862 remove_chunked_data(chunked_stream, read_bytes);
2863 }else {
2864 read_bytes = min(size, chunked_stream->chunk_size);
2865
2866 if(blocking_mode == BLOCKING_DISALLOW) {
2867 DWORD avail;
2868
2869 if(!is_valid_netconn(req->netconn) || !NETCON_query_data_available(req->netconn, &avail) || !avail)
2870 break;
2871 if(read_bytes > avail)
2872 read_bytes = avail;
2873
2874 /* this could block */
2875 if(read_bytes == chunked_stream->chunk_size)
2876 break;
2877 }
2878
2879 res = NETCON_recv(req->netconn, (char *)buf+ret_read, read_bytes, BLOCKING_ALLOW, (int*)&read_bytes);
2880 if(res != ERROR_SUCCESS)
2881 break;
2882 }
2883
2884 chunked_stream->chunk_size -= read_bytes;
2885 size -= read_bytes;
2886 ret_read += read_bytes;
2887 if(size && !chunked_stream->chunk_size) {
2888 assert(blocking_mode != BLOCKING_DISALLOW);
2889 res = start_next_chunk(chunked_stream, req);
2890 if(res != ERROR_SUCCESS)
2891 break;
2892 }
2893
2894 if(blocking_mode == BLOCKING_ALLOW)
2895 blocking_mode = BLOCKING_DISALLOW;
2896 }
2897
2898 TRACE("read %u bytes\n", ret_read);
2899 *read = ret_read;
2900 return res;
2901 }
2902
2903 static BOOL chunked_drain_content(data_stream_t *stream, http_request_t *req)
2904 {
2905 chunked_stream_t *chunked_stream = (chunked_stream_t*)stream;
2906
2907 remove_chunked_data(chunked_stream, chunked_stream->buf_size);
2908 return chunked_stream->end_of_data;
2909 }
2910
2911 static void chunked_destroy(data_stream_t *stream)
2912 {
2913 chunked_stream_t *chunked_stream = (chunked_stream_t*)stream;
2914 heap_free(chunked_stream);
2915 }
2916
2917 static const data_stream_vtbl_t chunked_stream_vtbl = {
2918 chunked_get_avail_data,
2919 chunked_end_of_data,
2920 chunked_read,
2921 chunked_drain_content,
2922 chunked_destroy
2923 };
2924
2925 /* set the request content length based on the headers */
2926 static DWORD set_content_length(http_request_t *request)
2927 {
2928 static const WCHAR szChunked[] = {'c','h','u','n','k','e','d',0};
2929 static const WCHAR headW[] = {'H','E','A','D',0};
2930 WCHAR encoding[20];
2931 DWORD size;
2932
2933 if(request->status_code == HTTP_STATUS_NO_CONTENT || !strcmpW(request->verb, headW)) {
2934 request->contentLength = request->netconn_stream.content_length = 0;
2935 return ERROR_SUCCESS;
2936 }
2937
2938 size = sizeof(request->contentLength);
2939 if (HTTP_HttpQueryInfoW(request, HTTP_QUERY_FLAG_NUMBER|HTTP_QUERY_CONTENT_LENGTH,
2940 &request->contentLength, &size, NULL) != ERROR_SUCCESS)
2941 request->contentLength = ~0u;
2942 request->netconn_stream.content_length = request->contentLength;
2943 request->netconn_stream.content_read = request->read_size;
2944
2945 size = sizeof(encoding);
2946 if (HTTP_HttpQueryInfoW(request, HTTP_QUERY_TRANSFER_ENCODING, encoding, &size, NULL) == ERROR_SUCCESS &&
2947 !strcmpiW(encoding, szChunked))
2948 {
2949 chunked_stream_t *chunked_stream;
2950
2951 chunked_stream = heap_alloc(sizeof(*chunked_stream));
2952 if(!chunked_stream)
2953 return ERROR_OUTOFMEMORY;
2954
2955 chunked_stream->data_stream.vtbl = &chunked_stream_vtbl;
2956 chunked_stream->buf_size = chunked_stream->buf_pos = 0;
2957 chunked_stream->chunk_size = ~0u;
2958 chunked_stream->end_of_data = FALSE;
2959
2960 if(request->read_size) {
2961 memcpy(chunked_stream->buf, request->read_buf+request->read_pos, request->read_size);
2962 chunked_stream->buf_size = request->read_size;
2963 request->read_size = request->read_pos = 0;
2964 }
2965
2966 request->data_stream = &chunked_stream->data_stream;
2967 request->contentLength = ~0u;
2968 request->read_chunked = TRUE;
2969 }
2970
2971 if(request->decoding) {
2972 int encoding_idx;
2973
2974 static const WCHAR deflateW[] = {'d','e','f','l','a','t','e',0};
2975 static const WCHAR gzipW[] = {'g','z','i','p',0};
2976
2977 EnterCriticalSection( &request->headers_section );
2978
2979 encoding_idx = HTTP_GetCustomHeaderIndex(request, szContent_Encoding, 0, FALSE);
2980 if(encoding_idx != -1) {
2981 if(!strcmpiW(request->custHeaders[encoding_idx].lpszValue, gzipW)) {
2982 HTTP_DeleteCustomHeader(request, encoding_idx);
2983 LeaveCriticalSection( &request->headers_section );
2984 return init_gzip_stream(request, TRUE);
2985 }
2986 if(!strcmpiW(request->custHeaders[encoding_idx].lpszValue, deflateW)) {
2987 HTTP_DeleteCustomHeader(request, encoding_idx);
2988 LeaveCriticalSection( &request->headers_section );
2989 return init_gzip_stream(request, FALSE);
2990 }
2991 }
2992
2993 LeaveCriticalSection( &request->headers_section );
2994 }
2995
2996 return ERROR_SUCCESS;
2997 }
2998
2999 static void send_request_complete(http_request_t *req, DWORD_PTR result, DWORD error)
3000 {
3001 INTERNET_ASYNC_RESULT iar;
3002
3003 iar.dwResult = result;
3004 iar.dwError = error;
3005
3006 INTERNET_SendCallback(&req->hdr, req->hdr.dwContext, INTERNET_STATUS_REQUEST_COMPLETE, &iar,
3007 sizeof(INTERNET_ASYNC_RESULT));
3008 }
3009
3010 static void HTTP_ReceiveRequestData(http_request_t *req, BOOL first_notif, DWORD *ret_size)
3011 {
3012 DWORD res, read = 0, avail = 0;
3013 blocking_mode_t mode;
3014
3015 TRACE("%p\n", req);
3016
3017 EnterCriticalSection( &req->read_section );
3018
3019 mode = first_notif && req->read_size ? BLOCKING_DISALLOW : BLOCKING_ALLOW;
3020 res = refill_read_buffer(req, mode, &read);
3021 if(res == ERROR_SUCCESS)
3022 avail = get_avail_data(req);
3023
3024 LeaveCriticalSection( &req->read_section );
3025
3026 if(res != ERROR_SUCCESS || (mode != BLOCKING_DISALLOW && !read)) {
3027 WARN("res %u read %u, closing connection\n", res, read);
3028 http_release_netconn(req, FALSE);
3029 }
3030
3031 if(res != ERROR_SUCCESS) {
3032 send_request_complete(req, 0, res);
3033 return;
3034 }
3035
3036 if(ret_size)
3037 *ret_size = avail;
3038 if(first_notif)
3039 avail = 0;
3040
3041 send_request_complete(req, req->session->hdr.dwInternalFlags & INET_OPENURL ? (DWORD_PTR)req->hdr.hInternet : 1, avail);
3042 }
3043
3044 /* read data from the http connection (the read section must be held) */
3045 static DWORD HTTPREQ_Read(http_request_t *req, void *buffer, DWORD size, DWORD *read, BOOL sync)
3046 {
3047 DWORD current_read = 0, ret_read = 0;
3048 blocking_mode_t blocking_mode;
3049 DWORD res = ERROR_SUCCESS;
3050
3051 blocking_mode = req->session->appInfo->hdr.dwFlags & INTERNET_FLAG_ASYNC ? BLOCKING_ALLOW : BLOCKING_WAITALL;
3052
3053 EnterCriticalSection( &req->read_section );
3054
3055 if(req->read_size) {
3056 ret_read = min(size, req->read_size);
3057 memcpy(buffer, req->read_buf+req->read_pos, ret_read);
3058 req->read_size -= ret_read;
3059 req->read_pos += ret_read;
3060 if(blocking_mode == BLOCKING_ALLOW)
3061 blocking_mode = BLOCKING_DISALLOW;
3062 }
3063
3064 if(ret_read < size) {
3065 res = read_http_stream(req, (BYTE*)buffer+ret_read, size-ret_read, &current_read, blocking_mode);
3066 ret_read += current_read;
3067 }
3068
3069 LeaveCriticalSection( &req->read_section );
3070
3071 *read = ret_read;
3072 TRACE( "retrieved %u bytes (%u)\n", ret_read, req->contentLength );
3073
3074 if(size && !ret_read)
3075 http_release_netconn(req, res == ERROR_SUCCESS);
3076
3077 return res;
3078 }
3079
3080 static BOOL drain_content(http_request_t *req, BOOL blocking)
3081 {
3082 BOOL ret;
3083
3084 if(!is_valid_netconn(req->netconn) || req->contentLength == -1)
3085 return FALSE;
3086
3087 if(!strcmpW(req->verb, szHEAD))
3088 return TRUE;
3089
3090 if(!blocking)
3091 return req->data_stream->vtbl->drain_content(req->data_stream, req);
3092
3093 EnterCriticalSection( &req->read_section );
3094
3095 while(1) {
3096 DWORD bytes_read, res;
3097 BYTE buf[4096];
3098
3099 res = HTTPREQ_Read(req, buf, sizeof(buf), &bytes_read, TRUE);
3100 if(res != ERROR_SUCCESS) {
3101 ret = FALSE;
3102 break;
3103 }
3104 if(!bytes_read) {
3105 ret = TRUE;
3106 break;
3107 }
3108 }
3109
3110 LeaveCriticalSection( &req->read_section );
3111 return ret;
3112 }
3113
3114 static DWORD HTTPREQ_ReadFile(object_header_t *hdr, void *buffer, DWORD size, DWORD *read)
3115 {
3116 http_request_t *req = (http_request_t*)hdr;
3117 DWORD res;
3118
3119 EnterCriticalSection( &req->read_section );
3120 if(hdr->dwError == INTERNET_HANDLE_IN_USE)
3121 hdr->dwError = ERROR_INTERNET_INTERNAL_ERROR;
3122
3123 res = HTTPREQ_Read(req, buffer, size, read, TRUE);
3124 if(res == ERROR_SUCCESS)
3125 res = hdr->dwError;
3126 LeaveCriticalSection( &req->read_section );
3127
3128 return res;
3129 }
3130
3131 typedef struct {
3132 task_header_t hdr;
3133 void *buf;
3134 DWORD size;
3135 DWORD *ret_read;
3136 } read_file_ex_task_t;
3137
3138 static void AsyncReadFileExProc(task_header_t *hdr)
3139 {
3140 read_file_ex_task_t *task = (read_file_ex_task_t*)hdr;
3141 http_request_t *req = (http_request_t*)task->hdr.hdr;
3142 DWORD res;
3143
3144 TRACE("INTERNETREADFILEEXW %p\n", task->hdr.hdr);
3145
3146 res = HTTPREQ_Read(req, task->buf, task->size, task->ret_read, TRUE);
3147 send_request_complete(req, res == ERROR_SUCCESS, res);
3148 }
3149
3150 static DWORD HTTPREQ_ReadFileEx(object_header_t *hdr, void *buf, DWORD size, DWORD *ret_read,
3151 DWORD flags, DWORD_PTR context)
3152 {
3153
3154 http_request_t *req = (http_request_t*)hdr;
3155 DWORD res, read, cread, error = ERROR_SUCCESS;
3156
3157 if (flags & ~(IRF_ASYNC|IRF_NO_WAIT))
3158 FIXME("these dwFlags aren't implemented: 0x%x\n", flags & ~(IRF_ASYNC|IRF_NO_WAIT));
3159
3160 INTERNET_SendCallback(&req->hdr, req->hdr.dwContext, INTERNET_STATUS_RECEIVING_RESPONSE, NULL, 0);
3161
3162 if (hdr->dwFlags & INTERNET_FLAG_ASYNC)
3163 {
3164 read_file_ex_task_t *task;
3165
3166 if (TryEnterCriticalSection( &req->read_section ))
3167 {
3168 if (get_avail_data(req))
3169 {
3170 res = HTTPREQ_Read(req, buf, size, &read, FALSE);
3171 LeaveCriticalSection( &req->read_section );
3172 goto done;
3173 }
3174 LeaveCriticalSection( &req->read_section );
3175 }
3176
3177 task = alloc_async_task(&req->hdr, AsyncReadFileExProc, sizeof(*task));
3178 task->buf = buf;
3179 task->size = size;
3180 task->ret_read = ret_read;
3181
3182 INTERNET_AsyncCall(&task->hdr);
3183
3184 return ERROR_IO_PENDING;
3185 }
3186
3187 read = 0;
3188
3189 EnterCriticalSection( &req->read_section );
3190 if(hdr->dwError == ERROR_SUCCESS)
3191 hdr->dwError = INTERNET_HANDLE_IN_USE;
3192 else if(hdr->dwError == INTERNET_HANDLE_IN_USE)
3193 hdr->dwError = ERROR_INTERNET_INTERNAL_ERROR;
3194
3195 while(1) {
3196 res = HTTPREQ_Read(req, (char*)buf+read, size-read, &cread, !(flags & IRF_NO_WAIT));
3197 if(res != ERROR_SUCCESS)
3198 break;
3199
3200 read += cread;
3201 if(read == size || end_of_read_data(req))
3202 break;
3203
3204 LeaveCriticalSection( &req->read_section );
3205
3206 INTERNET_SendCallback(&req->hdr, req->hdr.dwContext, INTERNET_STATUS_RESPONSE_RECEIVED,
3207 &cread, sizeof(cread));
3208 INTERNET_SendCallback(&req->hdr, req->hdr.dwContext,
3209 INTERNET_STATUS_RECEIVING_RESPONSE, NULL, 0);
3210
3211 EnterCriticalSection( &req->read_section );
3212 }
3213
3214 if(hdr->dwError == INTERNET_HANDLE_IN_USE)
3215 hdr->dwError = ERROR_SUCCESS;
3216 else
3217 error = hdr->dwError;
3218
3219 LeaveCriticalSection( &req->read_section );
3220
3221 done:
3222 *ret_read = read;
3223 if (res == ERROR_SUCCESS) {
3224 INTERNET_SendCallback(&req->hdr, req->hdr.dwContext, INTERNET_STATUS_RESPONSE_RECEIVED,
3225 &read, sizeof(read));
3226 }
3227
3228 return res==ERROR_SUCCESS ? error : res;
3229 }
3230
3231 static DWORD HTTPREQ_WriteFile(object_header_t *hdr, const void *buffer, DWORD size, DWORD *written)
3232 {
3233 DWORD res;
3234 http_request_t *request = (http_request_t*)hdr;
3235
3236 INTERNET_SendCallback(&request->hdr, request->hdr.dwContext, INTERNET_STATUS_SENDING_REQUEST, NULL, 0);
3237
3238 *written = 0;
3239 res = NETCON_send(request->netconn, buffer, size, 0, (LPINT)written);
3240 if (res == ERROR_SUCCESS)
3241 request->bytesWritten += *written;
3242
3243 INTERNET_SendCallback(&request->hdr, request->hdr.dwContext, INTERNET_STATUS_REQUEST_SENT, written, sizeof(DWORD));
3244 return res;
3245 }
3246
3247 typedef struct {
3248 task_header_t hdr;
3249 DWORD *ret_size;
3250 } http_data_available_task_t;
3251
3252 static void AsyncQueryDataAvailableProc(task_header_t *hdr)
3253 {
3254 http_data_available_task_t *task = (http_data_available_task_t*)hdr;
3255
3256 HTTP_ReceiveRequestData((http_request_t*)task->hdr.hdr, FALSE, task->ret_size);
3257 }
3258
3259 static DWORD HTTPREQ_QueryDataAvailable(object_header_t *hdr, DWORD *available, DWORD flags, DWORD_PTR ctx)
3260 {
3261 http_request_t *req = (http_request_t*)hdr;
3262
3263 TRACE("(%p %p %x %lx)\n", req, available, flags, ctx);
3264
3265 if (req->session->appInfo->hdr.dwFlags & INTERNET_FLAG_ASYNC)
3266 {
3267 http_data_available_task_t *task;
3268
3269 /* never wait, if we can't enter the section we queue an async request right away */
3270 if (TryEnterCriticalSection( &req->read_section ))
3271 {
3272 refill_read_buffer(req, BLOCKING_DISALLOW, NULL);
3273 if ((*available = get_avail_data( req ))) goto done;
3274 if (end_of_read_data( req )) goto done;
3275 LeaveCriticalSection( &req->read_section );
3276 }
3277
3278 task = alloc_async_task(&req->hdr, AsyncQueryDataAvailableProc, sizeof(*task));
3279 task->ret_size = available;
3280 INTERNET_AsyncCall(&task->hdr);
3281 return ERROR_IO_PENDING;
3282 }
3283
3284 EnterCriticalSection( &req->read_section );
3285
3286 if (!(*available = get_avail_data( req )) && !end_of_read_data( req ))
3287 {
3288 refill_read_buffer( req, BLOCKING_ALLOW, NULL );
3289 *available = get_avail_data( req );
3290 }
3291
3292 done:
3293 LeaveCriticalSection( &req->read_section );
3294
3295 TRACE( "returning %u\n", *available );
3296 return ERROR_SUCCESS;
3297 }
3298
3299 static DWORD HTTPREQ_LockRequestFile(object_header_t *hdr, req_file_t **ret)
3300 {
3301 http_request_t *req = (http_request_t*)hdr;
3302
3303 TRACE("(%p)\n", req);
3304
3305 if(!req->req_file) {
3306 WARN("No cache file name available\n");
3307 return ERROR_FILE_NOT_FOUND;
3308 }
3309
3310 *ret = req_file_addref(req->req_file);
3311 return ERROR_SUCCESS;
3312 }
3313
3314 static const object_vtbl_t HTTPREQVtbl = {
3315 HTTPREQ_Destroy,
3316 HTTPREQ_CloseConnection,
3317 HTTPREQ_QueryOption,
3318 HTTPREQ_SetOption,
3319 HTTPREQ_ReadFile,
3320 HTTPREQ_ReadFileEx,
3321 HTTPREQ_WriteFile,
3322 HTTPREQ_QueryDataAvailable,
3323 NULL,
3324 HTTPREQ_LockRequestFile
3325 };
3326
3327 /***********************************************************************
3328 * HTTP_HttpOpenRequestW (internal)
3329 *
3330 * Open a HTTP request handle
3331 *
3332 * RETURNS
3333 * HINTERNET a HTTP request handle on success
3334 * NULL on failure
3335 *
3336 */
3337 static DWORD HTTP_HttpOpenRequestW(http_session_t *session,
3338 LPCWSTR lpszVerb, LPCWSTR lpszObjectName, LPCWSTR lpszVersion,
3339 LPCWSTR lpszReferrer , LPCWSTR *lpszAcceptTypes,
3340 DWORD dwFlags, DWORD_PTR dwContext, HINTERNET *ret)
3341 {
3342 appinfo_t *hIC = session->appInfo;
3343 http_request_t *request;
3344 DWORD len;
3345
3346 TRACE("-->\n");
3347
3348 request = alloc_object(&session->hdr, &HTTPREQVtbl, sizeof(http_request_t));
3349 if(!request)
3350 return ERROR_OUTOFMEMORY;
3351
3352 request->hdr.htype = WH_HHTTPREQ;
3353 request->hdr.dwFlags = dwFlags;
3354 request->hdr.dwContext = dwContext;
3355 request->contentLength = ~0u;
3356
3357 request->netconn_stream.data_stream.vtbl = &netconn_stream_vtbl;
3358 request->data_stream = &request->netconn_stream.data_stream;
3359 request->connect_timeout = session->connect_timeout;
3360 request->send_timeout = session->send_timeout;
3361 request->receive_timeout = session->receive_timeout;
3362
3363 InitializeCriticalSection( &request->headers_section );
3364 request->headers_section.DebugInfo->Spare[0] = (DWORD_PTR)(__FILE__ ": http_request_t.headers_section");
3365
3366 InitializeCriticalSection( &request->read_section );
3367 request->read_section.DebugInfo->Spare[0] = (DWORD_PTR)(__FILE__ ": http_request_t.read_section");
3368
3369 WININET_AddRef( &session->hdr );
3370 request->session = session;
3371 list_add_head( &session->hdr.children, &request->hdr.entry );
3372
3373 request->server = get_server(session->hostName, session->hostPort, (dwFlags & INTERNET_FLAG_SECURE) != 0, TRUE);
3374 if(!request->server) {
3375 WININET_Release(&request->hdr);
3376 return ERROR_OUTOFMEMORY;
3377 }
3378
3379 if (dwFlags & INTERNET_FLAG_IGNORE_CERT_CN_INVALID)
3380 request->security_flags |= SECURITY_FLAG_IGNORE_CERT_CN_INVALID;
3381 if (dwFlags & INTERNET_FLAG_IGNORE_CERT_DATE_INVALID)
3382 request->security_flags |= SECURITY_FLAG_IGNORE_CERT_DATE_INVALID;
3383
3384 if (lpszObjectName && *lpszObjectName) {
3385 HRESULT rc;
3386
3387 len = 0;
3388 rc = UrlEscapeW(lpszObjectName, NULL, &len, URL_ESCAPE_SPACES_ONLY);
3389 if (rc != E_POINTER)
3390 len = strlenW(lpszObjectName)+1;
3391 request->path = heap_alloc(len*sizeof(WCHAR));
3392 rc = UrlEscapeW(lpszObjectName, request->path, &len,
3393 URL_ESCAPE_SPACES_ONLY);
3394 if (rc != S_OK)
3395 {
3396 ERR("Unable to escape string!(%s) (%d)\n",debugstr_w(lpszObjectName),rc);
3397 strcpyW(request->path,lpszObjectName);
3398 }
3399 }else {
3400 static const WCHAR slashW[] = {'/',0};
3401
3402 request->path = heap_strdupW(slashW);
3403 }
3404
3405 if (lpszReferrer && *lpszReferrer)
3406 HTTP_ProcessHeader(request, HTTP_REFERER, lpszReferrer, HTTP_ADDREQ_FLAG_ADD | HTTP_ADDHDR_FLAG_REQ);
3407
3408 if (lpszAcceptTypes)
3409 {
3410 int i;
3411 for (i = 0; lpszAcceptTypes[i]; i++)
3412 {
3413 if (!*lpszAcceptTypes[i]) continue;
3414 HTTP_ProcessHeader(request, HTTP_ACCEPT, lpszAcceptTypes[i],
3415 HTTP_ADDHDR_FLAG_COALESCE_WITH_COMMA |
3416 HTTP_ADDHDR_FLAG_REQ |
3417 (i == 0 ? HTTP_ADDHDR_FLAG_REPLACE : 0));
3418 }
3419 }
3420
3421 request->verb = heap_strdupW(lpszVerb && *lpszVerb ? lpszVerb : szGET);
3422 request->version = heap_strdupW(lpszVersion && *lpszVersion ? lpszVersion : g_szHttp1_1);
3423
3424 HTTP_ProcessHeader(request, hostW, request->server->canon_host_port, HTTP_ADDREQ_FLAG_ADD | HTTP_ADDHDR_FLAG_REQ);
3425
3426 if (hIC->proxy && hIC->proxy[0] && !HTTP_ShouldBypassProxy(hIC, session->hostName))
3427 HTTP_DealWithProxy( hIC, session, request );
3428
3429 INTERNET_SendCallback(&session->hdr, dwContext,
3430 INTERNET_STATUS_HANDLE_CREATED, &request->hdr.hInternet,
3431 sizeof(HINTERNET));
3432
3433 TRACE("<-- (%p)\n", request);
3434
3435 *ret = request->hdr.hInternet;
3436 return ERROR_SUCCESS;
3437 }
3438
3439 /***********************************************************************
3440 * HttpOpenRequestW (WININET.@)
3441 *
3442 * Open a HTTP request handle
3443 *
3444 * RETURNS
3445 * HINTERNET a HTTP request handle on success
3446 * NULL on failure
3447 *
3448 */
3449 HINTERNET WINAPI HttpOpenRequestW(HINTERNET hHttpSession,
3450 LPCWSTR lpszVerb, LPCWSTR lpszObjectName, LPCWSTR lpszVersion,
3451 LPCWSTR lpszReferrer , LPCWSTR *lpszAcceptTypes,
3452 DWORD dwFlags, DWORD_PTR dwContext)
3453 {
3454 http_session_t *session;
3455 HINTERNET handle = NULL;
3456 DWORD res;
3457
3458 TRACE("(%p, %s, %s, %s, %s, %p, %08x, %08lx)\n", hHttpSession,
3459 debugstr_w(lpszVerb), debugstr_w(lpszObjectName),
3460 debugstr_w(lpszVersion), debugstr_w(lpszReferrer), lpszAcceptTypes,
3461 dwFlags, dwContext);
3462 if(lpszAcceptTypes!=NULL)
3463 {
3464 int i;
3465 for(i=0;lpszAcceptTypes[i]!=NULL;i++)
3466 TRACE("\taccept type: %s\n",debugstr_w(lpszAcceptTypes[i]));
3467 }
3468
3469 session = (http_session_t*) get_handle_object( hHttpSession );
3470 if (NULL == session || session->hdr.htype != WH_HHTTPSESSION)
3471 {
3472 res = ERROR_INTERNET_INCORRECT_HANDLE_TYPE;
3473 goto lend;
3474 }
3475
3476 /*
3477 * My tests seem to show that the windows version does not
3478 * become asynchronous until after this point. And anyhow
3479 * if this call was asynchronous then how would you get the
3480 * necessary HINTERNET pointer returned by this function.
3481 *
3482 */
3483 res = HTTP_HttpOpenRequestW(session, lpszVerb, lpszObjectName,
3484 lpszVersion, lpszReferrer, lpszAcceptTypes,
3485 dwFlags, dwContext, &handle);
3486 lend:
3487 if( session )
3488 WININET_Release( &session->hdr );
3489 TRACE("returning %p\n", handle);
3490 if(res != ERROR_SUCCESS)
3491 SetLastError(res);
3492 return handle;
3493 }
3494
3495 static const LPCWSTR header_lookup[] = {
3496 szMime_Version, /* HTTP_QUERY_MIME_VERSION = 0 */
3497 szContent_Type, /* HTTP_QUERY_CONTENT_TYPE = 1 */
3498 szContent_Transfer_Encoding,/* HTTP_QUERY_CONTENT_TRANSFER_ENCODING = 2 */
3499 szContent_ID, /* HTTP_QUERY_CONTENT_ID = 3 */
3500 NULL, /* HTTP_QUERY_CONTENT_DESCRIPTION = 4 */
3501 szContent_Length, /* HTTP_QUERY_CONTENT_LENGTH = 5 */
3502 szContent_Language, /* HTTP_QUERY_CONTENT_LANGUAGE = 6 */
3503 szAllow, /* HTTP_QUERY_ALLOW = 7 */
3504 szPublic, /* HTTP_QUERY_PUBLIC = 8 */
3505 szDate, /* HTTP_QUERY_DATE = 9 */
3506 szExpires, /* HTTP_QUERY_EXPIRES = 10 */
3507 szLast_Modified, /* HTTP_QUERY_LAST_MODIFIED = 11 */
3508 NULL, /* HTTP_QUERY_MESSAGE_ID = 12 */
3509 szURI, /* HTTP_QUERY_URI = 13 */
3510 szFrom, /* HTTP_QUERY_DERIVED_FROM = 14 */
3511 NULL, /* HTTP_QUERY_COST = 15 */
3512 NULL, /* HTTP_QUERY_LINK = 16 */
3513 szPragma, /* HTTP_QUERY_PRAGMA = 17 */
3514 NULL, /* HTTP_QUERY_VERSION = 18 */
3515 szStatus, /* HTTP_QUERY_STATUS_CODE = 19 */
3516 NULL, /* HTTP_QUERY_STATUS_TEXT = 20 */
3517 NULL, /* HTTP_QUERY_RAW_HEADERS = 21 */
3518 NULL, /* HTTP_QUERY_RAW_HEADERS_CRLF = 22 */
3519 szConnection, /* HTTP_QUERY_CONNECTION = 23 */
3520 szAccept, /* HTTP_QUERY_ACCEPT = 24 */
3521 szAccept_Charset, /* HTTP_QUERY_ACCEPT_CHARSET = 25 */
3522 szAccept_Encoding, /* HTTP_QUERY_ACCEPT_ENCODING = 26 */
3523 szAccept_Language, /* HTTP_QUERY_ACCEPT_LANGUAGE = 27 */
3524 szAuthorization, /* HTTP_QUERY_AUTHORIZATION = 28 */
3525 szContent_Encoding, /* HTTP_QUERY_CONTENT_ENCODING = 29 */
3526 NULL, /* HTTP_QUERY_FORWARDED = 30 */
3527 NULL, /* HTTP_QUERY_FROM = 31 */
3528 szIf_Modified_Since, /* HTTP_QUERY_IF_MODIFIED_SINCE = 32 */
3529 szLocation, /* HTTP_QUERY_LOCATION = 33 */
3530 NULL, /* HTTP_QUERY_ORIG_URI = 34 */
3531 szReferer, /* HTTP_QUERY_REFERER = 35 */
3532 szRetry_After, /* HTTP_QUERY_RETRY_AFTER = 36 */
3533 szServer, /* HTTP_QUERY_SERVER = 37 */
3534 NULL, /* HTTP_TITLE = 38 */
3535 szUser_Agent, /* HTTP_QUERY_USER_AGENT = 39 */
3536 szWWW_Authenticate, /* HTTP_QUERY_WWW_AUTHENTICATE = 40 */
3537 szProxy_Authenticate, /* HTTP_QUERY_PROXY_AUTHENTICATE = 41 */
3538 szAccept_Ranges, /* HTTP_QUERY_ACCEPT_RANGES = 42 */
3539 szSet_Cookie, /* HTTP_QUERY_SET_COOKIE = 43 */
3540 szCookie, /* HTTP_QUERY_COOKIE = 44 */
3541 NULL, /* HTTP_QUERY_REQUEST_METHOD = 45 */
3542 NULL, /* HTTP_QUERY_REFRESH = 46 */
3543 szContent_Disposition, /* HTTP_QUERY_CONTENT_DISPOSITION = 47 */
3544 szAge, /* HTTP_QUERY_AGE = 48 */
3545 szCache_Control, /* HTTP_QUERY_CACHE_CONTROL = 49 */
3546 szContent_Base, /* HTTP_QUERY_CONTENT_BASE = 50 */
3547 szContent_Location, /* HTTP_QUERY_CONTENT_LOCATION = 51 */
3548 szContent_MD5, /* HTTP_QUERY_CONTENT_MD5 = 52 */
3549 szContent_Range, /* HTTP_QUERY_CONTENT_RANGE = 53 */
3550 szETag, /* HTTP_QUERY_ETAG = 54 */
3551 hostW, /* HTTP_QUERY_HOST = 55 */
3552 szIf_Match, /* HTTP_QUERY_IF_MATCH = 56 */
3553 szIf_None_Match, /* HTTP_QUERY_IF_NONE_MATCH = 57 */
3554 szIf_Range, /* HTTP_QUERY_IF_RANGE = 58 */
3555 szIf_Unmodified_Since, /* HTTP_QUERY_IF_UNMODIFIED_SINCE = 59 */
3556 szMax_Forwards, /* HTTP_QUERY_MAX_FORWARDS = 60 */
3557 szProxy_Authorization, /* HTTP_QUERY_PROXY_AUTHORIZATION = 61 */
3558 szRange, /* HTTP_QUERY_RANGE = 62 */
3559 szTransfer_Encoding, /* HTTP_QUERY_TRANSFER_ENCODING = 63 */
3560 szUpgrade, /* HTTP_QUERY_UPGRADE = 64 */
3561 szVary, /* HTTP_QUERY_VARY = 65 */
3562 szVia, /* HTTP_QUERY_VIA = 66 */
3563 szWarning, /* HTTP_QUERY_WARNING = 67 */
3564 szExpect, /* HTTP_QUERY_EXPECT = 68 */
3565 szProxy_Connection, /* HTTP_QUERY_PROXY_CONNECTION = 69 */
3566 szUnless_Modified_Since, /* HTTP_QUERY_UNLESS_MODIFIED_SINCE = 70 */
3567 };
3568
3569 #define LAST_TABLE_HEADER (sizeof(header_lookup)/sizeof(header_lookup[0]))
3570
3571 /***********************************************************************
3572 * HTTP_HttpQueryInfoW (internal)
3573 */
3574 static DWORD HTTP_HttpQueryInfoW(http_request_t *request, DWORD dwInfoLevel,
3575 LPVOID lpBuffer, LPDWORD lpdwBufferLength, LPDWORD lpdwIndex)
3576 {
3577 LPHTTPHEADERW lphttpHdr = NULL;
3578 BOOL request_only = dwInfoLevel & HTTP_QUERY_FLAG_REQUEST_HEADERS;
3579 INT requested_index = lpdwIndex ? *lpdwIndex : 0;
3580 DWORD level = (dwInfoLevel & ~HTTP_QUERY_MODIFIER_FLAGS_MASK);
3581 INT index = -1;
3582
3583 EnterCriticalSection( &request->headers_section );
3584
3585 /* Find requested header structure */
3586 switch (level)
3587 {
3588 case HTTP_QUERY_CUSTOM:
3589 if (!lpBuffer)
3590 {
3591 LeaveCriticalSection( &request->headers_section );
3592 return ERROR_INVALID_PARAMETER;
3593 }
3594 index = HTTP_GetCustomHeaderIndex(request, lpBuffer, requested_index, request_only);
3595 break;
3596 case HTTP_QUERY_RAW_HEADERS_CRLF:
3597 {
3598 LPWSTR headers;
3599 DWORD len = 0;
3600 DWORD res = ERROR_INVALID_PARAMETER;
3601
3602 if (request_only)
3603 headers = build_request_header(request, request->verb, request->path, request->version, TRUE);
3604 else
3605 headers = build_response_header(request, TRUE);
3606 if (!headers)
3607 {
3608 LeaveCriticalSection( &request->headers_section );
3609 return ERROR_OUTOFMEMORY;
3610 }
3611
3612 len = strlenW(headers) * sizeof(WCHAR);
3613 if (len + sizeof(WCHAR) > *lpdwBufferLength)
3614 {
3615 len += sizeof(WCHAR);
3616 res = ERROR_INSUFFICIENT_BUFFER;
3617 }
3618 else if (lpBuffer)
3619 {
3620 memcpy(lpBuffer, headers, len + sizeof(WCHAR));
3621 TRACE("returning data: %s\n", debugstr_wn(lpBuffer, len / sizeof(WCHAR)));
3622 res = ERROR_SUCCESS;
3623 }
3624 *lpdwBufferLength = len;
3625
3626 heap_free(headers);
3627 LeaveCriticalSection( &request->headers_section );
3628 return res;
3629 }
3630 case HTTP_QUERY_RAW_HEADERS:
3631 {
3632 LPWSTR headers;
3633 DWORD len;
3634
3635 if (request_only)
3636 headers = build_request_header(request, request->verb, request->path, request->version, FALSE);
3637 else
3638 headers = build_response_header(request, FALSE);
3639
3640 if (!headers)
3641 {
3642 LeaveCriticalSection( &request->headers_section );
3643 return ERROR_OUTOFMEMORY;
3644 }
3645
3646 len = strlenW(headers) * sizeof(WCHAR);
3647 if (len > *lpdwBufferLength)
3648 {
3649 *lpdwBufferLength = len;
3650 heap_free(headers);
3651 LeaveCriticalSection( &request->headers_section );
3652 return ERROR_INSUFFICIENT_BUFFER;
3653 }
3654
3655 if (lpBuffer)
3656 {
3657 DWORD i;
3658
3659 TRACE("returning data: %s\n", debugstr_wn(headers, len / sizeof(WCHAR)));
3660
3661 for (i = 0; i < len / sizeof(WCHAR); i++)
3662 {
3663 if (headers[i] == '\n')
3664 headers[i] = 0;
3665 }
3666 memcpy(lpBuffer, headers, len);
3667 }
3668 *lpdwBufferLength = len - sizeof(WCHAR);
3669
3670 heap_free(headers);
3671 LeaveCriticalSection( &request->headers_section );
3672 return ERROR_SUCCESS;
3673 }
3674 case HTTP_QUERY_STATUS_TEXT:
3675 if (request->statusText)
3676 {
3677 DWORD len = strlenW(request->statusText);
3678 if (len + 1 > *lpdwBufferLength/sizeof(WCHAR))
3679 {
3680 *lpdwBufferLength = (len + 1) * sizeof(WCHAR);
3681 LeaveCriticalSection( &request->headers_section );
3682 return ERROR_INSUFFICIENT_BUFFER;
3683 }
3684 if (lpBuffer)
3685 {
3686 memcpy(lpBuffer, request->statusText, (len + 1) * sizeof(WCHAR));
3687 TRACE("returning data: %s\n", debugstr_wn(lpBuffer, len));
3688 }
3689 *lpdwBufferLength = len * sizeof(WCHAR);
3690 LeaveCriticalSection( &request->headers_section );
3691 return ERROR_SUCCESS;
3692 }
3693 break;
3694 case HTTP_QUERY_VERSION:
3695 if (request->version)
3696 {
3697 DWORD len = strlenW(request->version);
3698 if (len + 1 > *lpdwBufferLength/sizeof(WCHAR))
3699 {
3700 *lpdwBufferLength = (len + 1) * sizeof(WCHAR);
3701 LeaveCriticalSection( &request->headers_section );
3702 return ERROR_INSUFFICIENT_BUFFER;
3703 }
3704 if (lpBuffer)
3705 {
3706 memcpy(lpBuffer, request->version, (len + 1) * sizeof(WCHAR));
3707 TRACE("returning data: %s\n", debugstr_wn(lpBuffer, len));
3708 }
3709 *lpdwBufferLength = len * sizeof(WCHAR);
3710 LeaveCriticalSection( &request->headers_section );
3711 return ERROR_SUCCESS;
3712 }
3713 break;
3714 case HTTP_QUERY_CONTENT_ENCODING:
3715 index = HTTP_GetCustomHeaderIndex(request, header_lookup[request->read_gzip ? HTTP_QUERY_CONTENT_TYPE : level],
3716 requested_index,request_only);
3717 break;
3718 case HTTP_QUERY_STATUS_CODE: {
3719 DWORD res = ERROR_SUCCESS;
3720
3721 if(request_only)
3722 {
3723 LeaveCriticalSection( &request->headers_section );
3724 return ERROR_HTTP_INVALID_QUERY_REQUEST;
3725 }
3726
3727 if(requested_index)
3728 break;
3729
3730 if(dwInfoLevel & HTTP_QUERY_FLAG_NUMBER) {
3731 if(*lpdwBufferLength >= sizeof(DWORD))
3732 *(DWORD*)lpBuffer = request->status_code;
3733 else
3734 res = ERROR_INSUFFICIENT_BUFFER;
3735 *lpdwBufferLength = sizeof(DWORD);
3736 }else {
3737 WCHAR buf[12];
3738 DWORD size;
3739 static const WCHAR formatW[] = {'%','u',0};
3740
3741 size = sprintfW(buf, formatW, request->status_code) * sizeof(WCHAR);
3742
3743 if(size <= *lpdwBufferLength) {
3744 memcpy(lpBuffer, buf, size+sizeof(WCHAR));
3745 }else {
3746 size += sizeof(WCHAR);
3747 res = ERROR_INSUFFICIENT_BUFFER;
3748 }
3749
3750 *lpdwBufferLength = size;
3751 }
3752 LeaveCriticalSection( &request->headers_section );
3753 return res;
3754 }
3755 default:
3756 assert (LAST_TABLE_HEADER == (HTTP_QUERY_UNLESS_MODIFIED_SINCE + 1));
3757
3758 if (level < LAST_TABLE_HEADER && header_lookup[level])
3759 index = HTTP_GetCustomHeaderIndex(request, header_lookup[level],
3760 requested_index,request_only);
3761 }
3762
3763 if (index >= 0)
3764 lphttpHdr = &request->custHeaders[index];
3765
3766 /* Ensure header satisfies requested attributes */
3767 if (!lphttpHdr ||
3768 ((dwInfoLevel & HTTP_QUERY_FLAG_REQUEST_HEADERS) &&
3769 (~lphttpHdr->wFlags & HDR_ISREQUEST)))
3770 {
3771 LeaveCriticalSection( &request->headers_section );
3772 return ERROR_HTTP_HEADER_NOT_FOUND;
3773 }
3774
3775 /* coalesce value to requested type */
3776 if (dwInfoLevel & HTTP_QUERY_FLAG_NUMBER && lpBuffer)
3777 {
3778 *(int *)lpBuffer = atoiW(lphttpHdr->lpszValue);
3779 TRACE(" returning number: %d\n", *(int *)lpBuffer);
3780 }
3781 else if (dwInfoLevel & HTTP_QUERY_FLAG_SYSTEMTIME && lpBuffer)
3782 {
3783 time_t tmpTime;
3784 struct tm tmpTM;
3785 SYSTEMTIME *STHook;
3786
3787 tmpTime = ConvertTimeString(lphttpHdr->lpszValue);
3788
3789 tmpTM = *gmtime(&tmpTime);
3790 STHook = (SYSTEMTIME *)lpBuffer;
3791 STHook->wDay = tmpTM.tm_mday;
3792 STHook->wHour = tmpTM.tm_hour;
3793 STHook->wMilliseconds = 0;
3794 STHook->wMinute = tmpTM.tm_min;
3795 STHook->wDayOfWeek = tmpTM.tm_wday;
3796 STHook->wMonth = tmpTM.tm_mon + 1;
3797 STHook->wSecond = tmpTM.tm_sec;
3798 STHook->wYear = 1900+tmpTM.tm_year;
3799
3800 TRACE(" returning time: %04d/%02d/%02d - %d - %02d:%02d:%02d.%02d\n",
3801 STHook->wYear, STHook->wMonth, STHook->wDay, STHook->wDayOfWeek,
3802 STHook->wHour, STHook->wMinute, STHook->wSecond, STHook->wMilliseconds);
3803 }
3804 else if (lphttpHdr->lpszValue)
3805 {
3806 DWORD len = (strlenW(lphttpHdr->lpszValue) + 1) * sizeof(WCHAR);
3807
3808 if (len > *lpdwBufferLength)
3809 {
3810 *lpdwBufferLength = len;
3811 LeaveCriticalSection( &request->headers_section );
3812 return ERROR_INSUFFICIENT_BUFFER;
3813 }
3814 if (lpBuffer)
3815 {
3816 memcpy(lpBuffer, lphttpHdr->lpszValue, len);
3817 TRACE("! returning string: %s\n", debugstr_w(lpBuffer));
3818 }
3819 *lpdwBufferLength = len - sizeof(WCHAR);
3820 }
3821 if (lpdwIndex) (*lpdwIndex)++;
3822
3823 LeaveCriticalSection( &request->headers_section );
3824 return ERROR_SUCCESS;
3825 }
3826
3827 /***********************************************************************
3828 * HttpQueryInfoW (WININET.@)
3829 *
3830 * Queries for information about an HTTP request
3831 *
3832 * RETURNS
3833 * TRUE on success
3834 * FALSE on failure
3835 *
3836 */
3837 BOOL WINAPI HttpQueryInfoW(HINTERNET hHttpRequest, DWORD dwInfoLevel,
3838 LPVOID lpBuffer, LPDWORD lpdwBufferLength, LPDWORD lpdwIndex)
3839 {
3840 http_request_t *request;
3841 DWORD res;
3842
3843 if (TRACE_ON(wininet)) {
3844 #define FE(x) { x, #x }
3845 static const wininet_flag_info query_flags[] = {
3846 FE(HTTP_QUERY_MIME_VERSION),
3847 FE(HTTP_QUERY_CONTENT_TYPE),
3848 FE(HTTP_QUERY_CONTENT_TRANSFER_ENCODING),
3849 FE(HTTP_QUERY_CONTENT_ID),
3850 FE(HTTP_QUERY_CONTENT_DESCRIPTION),
3851 FE(HTTP_QUERY_CONTENT_LENGTH),
3852 FE(HTTP_QUERY_CONTENT_LANGUAGE),
3853 FE(HTTP_QUERY_ALLOW),
3854 FE(HTTP_QUERY_PUBLIC),
3855 FE(HTTP_QUERY_DATE),
3856 FE(HTTP_QUERY_EXPIRES),
3857 FE(HTTP_QUERY_LAST_MODIFIED),
3858 FE(HTTP_QUERY_MESSAGE_ID),
3859 FE(HTTP_QUERY_URI),
3860 FE(HTTP_QUERY_DERIVED_FROM),
3861 FE(HTTP_QUERY_COST),
3862 FE(HTTP_QUERY_LINK),
3863 FE(HTTP_QUERY_PRAGMA),
3864 FE(HTTP_QUERY_VERSION),
3865 FE(HTTP_QUERY_STATUS_CODE),
3866 FE(HTTP_QUERY_STATUS_TEXT),
3867 FE(HTTP_QUERY_RAW_HEADERS),
3868 FE(HTTP_QUERY_RAW_HEADERS_CRLF),
3869 FE(HTTP_QUERY_CONNECTION),
3870 FE(HTTP_QUERY_ACCEPT),
3871 FE(HTTP_QUERY_ACCEPT_CHARSET),
3872 FE(HTTP_QUERY_ACCEPT_ENCODING),
3873 FE(HTTP_QUERY_ACCEPT_LANGUAGE),
3874 FE(HTTP_QUERY_AUTHORIZATION),
3875 FE(HTTP_QUERY_CONTENT_ENCODING),
3876 FE(HTTP_QUERY_FORWARDED),
3877 FE(HTTP_QUERY_FROM),
3878 FE(HTTP_QUERY_IF_MODIFIED_SINCE),
3879 FE(HTTP_QUERY_LOCATION),
3880 FE(HTTP_QUERY_ORIG_URI),
3881 FE(HTTP_QUERY_REFERER),
3882 FE(HTTP_QUERY_RETRY_AFTER),
3883 FE(HTTP_QUERY_SERVER),
3884 FE(HTTP_QUERY_TITLE),
3885 FE(HTTP_QUERY_USER_AGENT),
3886 FE(HTTP_QUERY_WWW_AUTHENTICATE),
3887 FE(HTTP_QUERY_PROXY_AUTHENTICATE),
3888 FE(HTTP_QUERY_ACCEPT_RANGES),
3889 FE(HTTP_QUERY_SET_COOKIE),
3890 FE(HTTP_QUERY_COOKIE),
3891 FE(HTTP_QUERY_REQUEST_METHOD),
3892 FE(HTTP_QUERY_REFRESH),
3893 FE(HTTP_QUERY_CONTENT_DISPOSITION),
3894 FE(HTTP_QUERY_AGE),
3895 FE(HTTP_QUERY_CACHE_CONTROL),
3896 FE(HTTP_QUERY_CONTENT_BASE),
3897 FE(HTTP_QUERY_CONTENT_LOCATION),
3898 FE(HTTP_QUERY_CONTENT_MD5),
3899 FE(HTTP_QUERY_CONTENT_RANGE),
3900 FE(HTTP_QUERY_ETAG),
3901 FE(HTTP_QUERY_HOST),
3902 FE(HTTP_QUERY_IF_MATCH),
3903 FE(HTTP_QUERY_IF_NONE_MATCH),
3904 FE(HTTP_QUERY_IF_RANGE),
3905 FE(HTTP_QUERY_IF_UNMODIFIED_SINCE),
3906 FE(HTTP_QUERY_MAX_FORWARDS),
3907 FE(HTTP_QUERY_PROXY_AUTHORIZATION),
3908 FE(HTTP_QUERY_RANGE),
3909 FE(HTTP_QUERY_TRANSFER_ENCODING),
3910 FE(HTTP_QUERY_UPGRADE),
3911 FE(HTTP_QUERY_VARY),
3912 FE(HTTP_QUERY_VIA),
3913 FE(HTTP_QUERY_WARNING),
3914 FE(HTTP_QUERY_CUSTOM)
3915 };
3916 static const wininet_flag_info modifier_flags[] = {
3917 FE(HTTP_QUERY_FLAG_REQUEST_HEADERS),
3918 FE(HTTP_QUERY_FLAG_SYSTEMTIME),
3919 FE(HTTP_QUERY_FLAG_NUMBER),
3920 FE(HTTP_QUERY_FLAG_COALESCE)
3921 };
3922 #undef FE
3923 DWORD info_mod = dwInfoLevel & HTTP_QUERY_MODIFIER_FLAGS_MASK;
3924 DWORD info = dwInfoLevel & HTTP_QUERY_HEADER_MASK;
3925 DWORD i;
3926
3927 TRACE("(%p, 0x%08x)--> %d\n", hHttpRequest, dwInfoLevel, info);
3928 TRACE(" Attribute:");
3929 for (i = 0; i < (sizeof(query_flags) / sizeof(query_flags[0])); i++) {
3930 if (query_flags[i].val == info) {
3931 TRACE(" %s", query_flags[i].name);
3932 break;
3933 }
3934 }
3935 if (i == (sizeof(query_flags) / sizeof(query_flags[0]))) {
3936 TRACE(" Unknown (%08x)", info);
3937 }
3938
3939 TRACE(" Modifier:");
3940 for (i = 0; i < (sizeof(modifier_flags) / sizeof(modifier_flags[0])); i++) {
3941 if (modifier_flags[i].val & info_mod) {
3942 TRACE(" %s", modifier_flags[i].name);
3943 info_mod &= ~ modifier_flags[i].val;
3944 }
3945 }
3946
3947 if (info_mod) {
3948 TRACE(" Unknown (%08x)", info_mod);
3949 }
3950 TRACE("\n");
3951 }
3952
3953 request = (http_request_t*) get_handle_object( hHttpRequest );
3954 if (NULL == request || request->hdr.htype != WH_HHTTPREQ)
3955 {
3956 res = ERROR_INTERNET_INCORRECT_HANDLE_TYPE;
3957 goto lend;
3958 }
3959
3960 if (lpBuffer == NULL)
3961 *lpdwBufferLength = 0;
3962 res = HTTP_HttpQueryInfoW( request, dwInfoLevel,
3963 lpBuffer, lpdwBufferLength, lpdwIndex);
3964
3965 lend:
3966 if( request )
3967 WININET_Release( &request->hdr );
3968
3969 TRACE("%u <--\n", res);
3970 if(res != ERROR_SUCCESS)
3971 SetLastError(res);
3972 return res == ERROR_SUCCESS;
3973 }
3974
3975 /***********************************************************************
3976 * HttpQueryInfoA (WININET.@)
3977 *
3978 * Queries for information about an HTTP request
3979 *
3980 * RETURNS
3981 * TRUE on success
3982 * FALSE on failure
3983 *
3984 */
3985 BOOL WINAPI HttpQueryInfoA(HINTERNET hHttpRequest, DWORD dwInfoLevel,
3986 LPVOID lpBuffer, LPDWORD lpdwBufferLength, LPDWORD lpdwIndex)
3987 {
3988 BOOL result;
3989 DWORD len;
3990 WCHAR* bufferW;
3991
3992 TRACE("%p %x\n", hHttpRequest, dwInfoLevel);
3993
3994 if((dwInfoLevel & HTTP_QUERY_FLAG_NUMBER) ||
3995 (dwInfoLevel & HTTP_QUERY_FLAG_SYSTEMTIME))
3996 {
3997 return HttpQueryInfoW( hHttpRequest, dwInfoLevel, lpBuffer,
3998 lpdwBufferLength, lpdwIndex );
3999 }
4000
4001 if (lpBuffer)
4002 {
4003 DWORD alloclen;
4004 len = (*lpdwBufferLength)*sizeof(WCHAR);
4005 if ((dwInfoLevel & HTTP_QUERY_HEADER_MASK) == HTTP_QUERY_CUSTOM)
4006 {
4007 alloclen = MultiByteToWideChar( CP_ACP, 0, lpBuffer, -1, NULL, 0 ) * sizeof(WCHAR);
4008 if (alloclen < len)
4009 alloclen = len;
4010 }
4011 else
4012 alloclen = len;
4013 bufferW = heap_alloc(alloclen);
4014 /* buffer is in/out because of HTTP_QUERY_CUSTOM */
4015 if ((dwInfoLevel & HTTP_QUERY_HEADER_MASK) == HTTP_QUERY_CUSTOM)
4016 MultiByteToWideChar( CP_ACP, 0, lpBuffer, -1, bufferW, alloclen / sizeof(WCHAR) );
4017 } else
4018 {
4019 bufferW = NULL;
4020 len = 0;
4021 }
4022
4023 result = HttpQueryInfoW( hHttpRequest, dwInfoLevel, bufferW,
4024 &len, lpdwIndex );
4025 if( result )
4026 {
4027 len = WideCharToMultiByte( CP_ACP,0, bufferW, len / sizeof(WCHAR) + 1,
4028 lpBuffer, *lpdwBufferLength, NULL, NULL );
4029 *lpdwBufferLength = len - 1;
4030
4031 TRACE("lpBuffer: %s\n", debugstr_a(lpBuffer));
4032 }
4033 else
4034 /* since the strings being returned from HttpQueryInfoW should be
4035 * only ASCII characters, it is reasonable to assume that all of
4036 * the Unicode characters can be reduced to a single byte */
4037 *lpdwBufferLength = len / sizeof(WCHAR);
4038
4039 heap_free( bufferW );
4040 return result;
4041 }
4042
4043 /***********************************************************************
4044 * HTTP_GetRedirectURL (internal)
4045 */
4046 static LPWSTR HTTP_GetRedirectURL(http_request_t *request, LPCWSTR lpszUrl)
4047 {
4048 static WCHAR szHttp[] = {'h','t','t','p',0};
4049 static WCHAR szHttps[] = {'h','t','t','p','s',0};
4050 http_session_t *session = request->session;
4051 URL_COMPONENTSW urlComponents;
4052 DWORD url_length = 0;
4053 LPWSTR orig_url;
4054 LPWSTR combined_url;
4055
4056 urlComponents.dwStructSize = sizeof(URL_COMPONENTSW);
4057 urlComponents.lpszScheme = (request->hdr.dwFlags & INTERNET_FLAG_SECURE) ? szHttps : szHttp;
4058 urlComponents.dwSchemeLength = 0;
4059 urlComponents.lpszHostName = request->server->name;
4060 urlComponents.dwHostNameLength = 0;
4061 urlComponents.nPort = request->server->port;
4062 urlComponents.lpszUserName = session->userName;
4063 urlComponents.dwUserNameLength = 0;
4064 urlComponents.lpszPassword = NULL;
4065 urlComponents.dwPasswordLength = 0;
4066 urlComponents.lpszUrlPath = request->path;
4067 urlComponents.dwUrlPathLength = 0;
4068 urlComponents.lpszExtraInfo = NULL;
4069 urlComponents.dwExtraInfoLength = 0;
4070
4071 if (!InternetCreateUrlW(&urlComponents, 0, NULL, &url_length) &&
4072 (GetLastError() != ERROR_INSUFFICIENT_BUFFER))
4073 return NULL;
4074
4075 orig_url = heap_alloc(url_length);
4076
4077 /* convert from bytes to characters */
4078 url_length = url_length / sizeof(WCHAR) - 1;
4079 if (!InternetCreateUrlW(&urlComponents, 0, orig_url, &url_length))
4080 {
4081 heap_free(orig_url);
4082 return NULL;
4083 }
4084
4085 url_length = 0;
4086 if (!InternetCombineUrlW(orig_url, lpszUrl, NULL, &url_length, ICU_ENCODE_SPACES_ONLY) &&
4087 (GetLastError() != ERROR_INSUFFICIENT_BUFFER))
4088 {
4089 heap_free(orig_url);
4090 return NULL;
4091 }
4092 combined_url = heap_alloc(url_length * sizeof(WCHAR));
4093
4094 if (!InternetCombineUrlW(orig_url, lpszUrl, combined_url, &url_length, ICU_ENCODE_SPACES_ONLY))
4095 {
4096 heap_free(orig_url);
4097 heap_free(combined_url);
4098 return NULL;
4099 }
4100 heap_free(orig_url);
4101 return combined_url;
4102 }
4103
4104
4105 /***********************************************************************
4106 * HTTP_HandleRedirect (internal)
4107 */
4108 static DWORD HTTP_HandleRedirect(http_request_t *request, LPCWSTR lpszUrl)
4109 {
4110 http_session_t *session = request->session;
4111 WCHAR path[INTERNET_MAX_PATH_LENGTH];
4112 int index;
4113
4114 if(lpszUrl[0]=='/')
4115 {
4116 /* if it's an absolute path, keep the same session info */
4117 lstrcpynW(path, lpszUrl, INTERNET_MAX_URL_LENGTH);
4118 }
4119 else
4120 {
4121 URL_COMPONENTSW urlComponents;
4122 WCHAR protocol[INTERNET_MAX_SCHEME_LENGTH];
4123 WCHAR hostName[INTERNET_MAX_HOST_NAME_LENGTH];
4124 WCHAR userName[INTERNET_MAX_USER_NAME_LENGTH];
4125 BOOL custom_port = FALSE;
4126
4127 static const WCHAR httpW[] = {'h','t','t','p',0};
4128 static const WCHAR httpsW[] = {'h','t','t','p','s',0};
4129
4130 userName[0] = 0;
4131 hostName[0] = 0;
4132 protocol[0] = 0;
4133
4134 urlComponents.dwStructSize = sizeof(URL_COMPONENTSW);
4135 urlComponents.lpszScheme = protocol;
4136 urlComponents.dwSchemeLength = INTERNET_MAX_SCHEME_LENGTH;
4137 urlComponents.lpszHostName = hostName;
4138 urlComponents.dwHostNameLength = INTERNET_MAX_HOST_NAME_LENGTH;
4139 urlComponents.lpszUserName = userName;
4140 urlComponents.dwUserNameLength = INTERNET_MAX_USER_NAME_LENGTH;
4141 urlComponents.lpszPassword = NULL;
4142 urlComponents.dwPasswordLength = 0;
4143 urlComponents.lpszUrlPath = path;
4144 urlComponents.dwUrlPathLength = INTERNET_MAX_PATH_LENGTH;
4145 urlComponents.lpszExtraInfo = NULL;
4146 urlComponents.dwExtraInfoLength = 0;
4147 if(!InternetCrackUrlW(lpszUrl, strlenW(lpszUrl), 0, &urlComponents))
4148 return INTERNET_GetLastError();
4149
4150 if(!strcmpiW(protocol, httpW)) {
4151 if(request->hdr.dwFlags & INTERNET_FLAG_SECURE) {
4152 TRACE("redirect from secure page to non-secure page\n");
4153 /* FIXME: warn about from secure redirect to non-secure page */
4154 request->hdr.dwFlags &= ~INTERNET_FLAG_SECURE;
4155 }
4156
4157 if(urlComponents.nPort == INTERNET_INVALID_PORT_NUMBER)
4158 urlComponents.nPort = INTERNET_DEFAULT_HTTP_PORT;
4159 else if(urlComponents.nPort != INTERNET_DEFAULT_HTTP_PORT)
4160 custom_port = TRUE;
4161 }else if(!strcmpiW(protocol, httpsW)) {
4162 if(!(request->hdr.dwFlags & INTERNET_FLAG_SECURE)) {
4163 TRACE("redirect from non-secure page to secure page\n");
4164 /* FIXME: notify about redirect to secure page */
4165 request->hdr.dwFlags |= INTERNET_FLAG_SECURE;
4166 }
4167
4168 if(urlComponents.nPort == INTERNET_INVALID_PORT_NUMBER)
4169 urlComponents.nPort = INTERNET_DEFAULT_HTTPS_PORT;
4170 else if(urlComponents.nPort != INTERNET_DEFAULT_HTTPS_PORT)
4171 custom_port = TRUE;
4172 }
4173
4174 heap_free(session->hostName);
4175
4176 if(custom_port) {
4177 int len;
4178 static const WCHAR fmt[] = {'%','s',':','%','u',0};
4179 len = lstrlenW(hostName);
4180 len += 7; /* 5 for strlen("65535") + 1 for ":" + 1 for '\0' */
4181 session->hostName = heap_alloc(len*sizeof(WCHAR));
4182 sprintfW(session->hostName, fmt, hostName, urlComponents.nPort);
4183 }
4184 else
4185 session->hostName = heap_strdupW(hostName);
4186 session->hostPort = urlComponents.nPort;
4187
4188 HTTP_ProcessHeader(request, hostW, session->hostName, HTTP_ADDREQ_FLAG_ADD | HTTP_ADDREQ_FLAG_REPLACE | HTTP_ADDHDR_FLAG_REQ);
4189
4190 heap_free(session->userName);
4191 session->userName = NULL;
4192 if (userName[0])
4193 session->userName = heap_strdupW(userName);
4194
4195 reset_data_stream(request);
4196
4197 if(strcmpiW(request->server->name, hostName) || request->server->port != urlComponents.nPort) {
4198 server_t *new_server;
4199
4200 new_server = get_server(hostName, urlComponents.nPort, urlComponents.nScheme == INTERNET_SCHEME_HTTPS, TRUE);
4201 server_release(request->server);
4202 request->server = new_server;
4203 }
4204 }
4205 heap_free(request->path);
4206 request->path=NULL;
4207 if (*path)
4208 {
4209 DWORD needed = 0;
4210 HRESULT rc;
4211
4212 rc = UrlEscapeW(path, NULL, &needed, URL_ESCAPE_SPACES_ONLY);
4213 if (rc != E_POINTER)
4214 needed = strlenW(path)+1;
4215 request->path = heap_alloc(needed*sizeof(WCHAR));
4216 rc = UrlEscapeW(path, request->path, &needed,
4217 URL_ESCAPE_SPACES_ONLY);
4218 if (rc != S_OK)
4219 {
4220 ERR("Unable to escape string!(%s) (%d)\n",debugstr_w(path),rc);
4221 strcpyW(request->path,path);
4222 }
4223 }
4224
4225 EnterCriticalSection( &request->headers_section );
4226
4227 /* Remove custom content-type/length headers on redirects. */
4228 index = HTTP_GetCustomHeaderIndex(request, szContent_Type, 0, TRUE);
4229 if (index != -1) HTTP_DeleteCustomHeader(request, index);
4230 index = HTTP_GetCustomHeaderIndex(request, szContent_Length, 0, TRUE);
4231 if (index != -1) HTTP_DeleteCustomHeader(request, index);
4232
4233 LeaveCriticalSection( &request->headers_section );
4234
4235 return ERROR_SUCCESS;
4236 }
4237
4238 /***********************************************************************
4239 * HTTP_build_req (internal)
4240 *
4241 * concatenate all the strings in the request together
4242 */
4243 static LPWSTR HTTP_build_req( LPCWSTR *list, int len )
4244 {
4245 LPCWSTR *t;
4246 LPWSTR str;
4247
4248 for( t = list; *t ; t++ )
4249 len += strlenW( *t );
4250 len++;
4251
4252 str = heap_alloc(len*sizeof(WCHAR));
4253 *str = 0;
4254
4255 for( t = list; *t ; t++ )
4256 strcatW( str, *t );
4257
4258 return str;
4259 }
4260
4261 static void HTTP_InsertCookies(http_request_t *request)
4262 {
4263 WCHAR *cookies;
4264 DWORD res;
4265
4266 res = get_cookie_header(request->server->name, request->path, &cookies);
4267 if(res != ERROR_SUCCESS || !cookies)
4268 return;
4269
4270 get_cookie_header(request->server->name, request->path, &cookies);
4271 HTTP_HttpAddRequestHeadersW(request, cookies, strlenW(cookies), HTTP_ADDREQ_FLAG_REPLACE);
4272 heap_free(cookies);
4273 }
4274
4275 static WORD HTTP_ParseWkday(LPCWSTR day)
4276 {
4277 static const WCHAR days[7][4] = {{ 's','u','n',0 },
4278 { 'm','o','n',0 },
4279 { 't','u','e',0 },
4280 { 'w','e','d',0 },
4281 { 't','h','u',0 },
4282 { 'f','r','i',0 },
4283 { 's','a','t',0 }};
4284 unsigned int i;
4285 for (i = 0; i < sizeof(days)/sizeof(*days); i++)
4286 if (!strcmpiW(day, days[i]))
4287 return i;
4288
4289 /* Invalid */
4290 return 7;
4291 }
4292
4293 static WORD HTTP_ParseMonth(LPCWSTR month)
4294 {
4295 static const WCHAR jan[] = { 'j','a','n',0 };
4296 static const WCHAR feb[] = { 'f','e','b',0 };
4297 static const WCHAR mar[] = { 'm','a','r',0 };
4298 static const WCHAR apr[] = { 'a','p','r',0 };
4299 static const WCHAR may[] = { 'm','a','y',0 };
4300 static const WCHAR jun[] = { 'j','u','n',0 };
4301 static const WCHAR jul[] = { 'j','u','l',0 };
4302 static const WCHAR aug[] = { 'a','u','g',0 };
4303 static const WCHAR sep[] = { 's','e','p',0 };
4304 static const WCHAR oct[] = { 'o','c','t',0 };
4305 static const WCHAR nov[] = { 'n','o','v',0 };
4306 static const WCHAR dec[] = { 'd','e','c',0 };
4307
4308 if (!strcmpiW(month, jan)) return 1;
4309 if (!strcmpiW(month, feb)) return 2;
4310 if (!strcmpiW(month, mar)) return 3;
4311 if (!strcmpiW(month, apr)) return 4;
4312 if (!strcmpiW(month, may)) return 5;
4313 if (!strcmpiW(month, jun)) return 6;
4314 if (!strcmpiW(month, jul)) return 7;
4315 if (!strcmpiW(month, aug)) return 8;
4316 if (!strcmpiW(month, sep)) return 9;
4317 if (!strcmpiW(month, oct)) return 10;
4318 if (!strcmpiW(month, nov)) return 11;
4319 if (!strcmpiW(month, dec)) return 12;
4320 /* Invalid */
4321 return 0;
4322 }
4323
4324 /* Parses the string pointed to by *str, assumed to be a 24-hour time HH:MM:SS,
4325 * optionally preceded by whitespace.
4326 * Upon success, returns TRUE, sets the wHour, wMinute, and wSecond fields of
4327 * st, and sets *str to the first character after the time format.
4328 */
4329 static BOOL HTTP_ParseTime(SYSTEMTIME *st, LPCWSTR *str)
4330 {
4331 LPCWSTR ptr = *str;
4332 WCHAR *nextPtr;
4333 unsigned long num;
4334
4335 while (isspaceW(*ptr))
4336 ptr++;
4337
4338 num = strtoulW(ptr, &nextPtr, 10);
4339 if (!nextPtr || nextPtr <= ptr || *nextPtr != ':')
4340 {
4341 ERR("unexpected time format %s\n", debugstr_w(ptr));
4342 return FALSE;
4343 }
4344 if (num > 23)
4345 {
4346 ERR("unexpected hour in time format %s\n", debugstr_w(ptr));
4347 return FALSE;
4348 }
4349 ptr = nextPtr + 1;
4350 st->wHour = (WORD)num;
4351 num = strtoulW(ptr, &nextPtr, 10);
4352 if (!nextPtr || nextPtr <= ptr || *nextPtr != ':')
4353 {
4354 ERR("unexpected time format %s\n", debugstr_w(ptr));
4355 return FALSE;
4356 }
4357 if (num > 59)
4358 {
4359 ERR("unexpected minute in time format %s\n", debugstr_w(ptr));
4360 return FALSE;
4361 }
4362 ptr = nextPtr + 1;
4363 st->wMinute = (WORD)num;
4364 num = strtoulW(ptr, &nextPtr, 10);
4365 if (!nextPtr || nextPtr <= ptr)
4366 {
4367 ERR("unexpected time format %s\n", debugstr_w(ptr));
4368 return FALSE;
4369 }
4370 if (num > 59)
4371 {
4372 ERR("unexpected second in time format %s\n", debugstr_w(ptr));
4373 return FALSE;
4374 }
4375 ptr = nextPtr + 1;
4376 *str = ptr;
4377 st->wSecond = (WORD)num;
4378 return TRUE;
4379 }
4380
4381 static BOOL HTTP_ParseDateAsAsctime(LPCWSTR value, FILETIME *ft)
4382 {
4383 static const WCHAR gmt[]= { 'G','M','T',0 };
4384 WCHAR day[4], *dayPtr, month[4], *monthPtr, *nextPtr;
4385 LPCWSTR ptr;
4386 SYSTEMTIME st = { 0 };
4387 unsigned long num;
4388
4389 for (ptr = value, dayPtr = day; *ptr && !isspaceW(*ptr) &&
4390 dayPtr - day < sizeof(day) / sizeof(day[0]) - 1; ptr++, dayPtr++)
4391 *dayPtr = *ptr;
4392 *dayPtr = 0;
4393 st.wDayOfWeek = HTTP_ParseWkday(day);
4394 if (st.wDayOfWeek >= 7)
4395 {
4396 ERR("unexpected weekday %s\n", debugstr_w(day));
4397 return FALSE;
4398 }
4399
4400 while (isspaceW(*ptr))
4401 ptr++;
4402
4403 for (monthPtr = month; !isspace(*ptr) &&
4404 monthPtr - month < sizeof(month) / sizeof(month[0]) - 1;
4405 monthPtr++, ptr++)
4406 *monthPtr = *ptr;
4407 *monthPtr = 0;
4408 st.wMonth = HTTP_ParseMonth(month);
4409 if (!st.wMonth || st.wMonth > 12)
4410 {
4411 ERR("unexpected month %s\n", debugstr_w(month));
4412 return FALSE;
4413 }
4414
4415 while (isspaceW(*ptr))
4416 ptr++;
4417
4418 num = strtoulW(ptr, &nextPtr, 10);
4419 if (!nextPtr || nextPtr <= ptr || !num || num > 31)
4420 {
4421 ERR("unexpected day %s\n", debugstr_w(ptr));
4422 return FALSE;
4423 }
4424 ptr = nextPtr;
4425 st.wDay = (WORD)num;
4426
4427 while (isspaceW(*ptr))
4428 ptr++;
4429
4430 if (!HTTP_ParseTime(&st, &ptr))
4431 return FALSE;
4432
4433 while (isspaceW(*ptr))
4434 ptr++;
4435
4436 num = strtoulW(ptr, &nextPtr, 10);
4437 if (!nextPtr || nextPtr <= ptr || num < 1601 || num > 30827)
4438 {
4439 ERR("unexpected year %s\n", debugstr_w(ptr));
4440 return FALSE;
4441 }
4442 ptr = nextPtr;
4443 st.wYear = (WORD)num;
4444
4445 while (isspaceW(*ptr))
4446 ptr++;
4447
4448 /* asctime() doesn't report a timezone, but some web servers do, so accept
4449 * with or without GMT.
4450 */
4451 if (*ptr && strcmpW(ptr, gmt))
4452 {
4453 ERR("unexpected timezone %s\n", debugstr_w(ptr));
4454 return FALSE;
4455 }
4456 return SystemTimeToFileTime(&st, ft);
4457 }
4458
4459 static BOOL HTTP_ParseRfc1123Date(LPCWSTR value, FILETIME *ft)
4460 {
4461 static const WCHAR gmt[]= { 'G','M','T',0 };
4462 WCHAR *nextPtr, day[4], month[4], *monthPtr;
4463 LPCWSTR ptr;
4464 unsigned long num;
4465 SYSTEMTIME st = { 0 };
4466
4467 ptr = strchrW(value, ',');
4468 if (!ptr)
4469 return FALSE;
4470 if (ptr - value != 3)
4471 {
4472 WARN("unexpected weekday %s\n", debugstr_wn(value, ptr - value));
4473 return FALSE;
4474 }
4475 memcpy(day, value, (ptr - value) * sizeof(WCHAR));
4476 day[3] = 0;
4477 st.wDayOfWeek = HTTP_ParseWkday(day);
4478 if (st.wDayOfWeek > 6)
4479 {
4480 WARN("unexpected weekday %s\n", debugstr_wn(value, ptr - value));
4481 return FALSE;
4482 }
4483 ptr++;
4484
4485 while (isspaceW(*ptr))
4486 ptr++;
4487
4488 num = strtoulW(ptr, &nextPtr, 10);
4489 if (!nextPtr || nextPtr <= ptr || !num || num > 31)
4490 {
4491 WARN("unexpected day %s\n", debugstr_w(value));
4492 return FALSE;
4493 }
4494 ptr = nextPtr;
4495 st.wDay = (WORD)num;
4496
4497 while (isspaceW(*ptr))
4498 ptr++;
4499
4500 for (monthPtr = month; !isspace(*ptr) &&
4501 monthPtr - month < sizeof(month) / sizeof(month[0]) - 1;
4502 monthPtr++, ptr++)
4503 *monthPtr = *ptr;
4504 *monthPtr = 0;
4505 st.wMonth = HTTP_ParseMonth(month);
4506 if (!st.wMonth || st.wMonth > 12)
4507 {
4508 WARN("unexpected month %s\n", debugstr_w(month));
4509 return FALSE;
4510 }
4511
4512 while (isspaceW(*ptr))
4513 ptr++;
4514
4515 num = strtoulW(ptr, &nextPtr, 10);
4516 if (!nextPtr || nextPtr <= ptr || num < 1601 || num > 30827)
4517 {
4518 ERR("unexpected year %s\n", debugstr_w(value));
4519 return FALSE;
4520 }
4521 ptr = nextPtr;
4522 st.wYear = (WORD)num;
4523
4524 if (!HTTP_ParseTime(&st, &ptr))
4525 return FALSE;
4526
4527 while (isspaceW(*ptr))
4528 ptr++;
4529
4530 if (strcmpW(ptr, gmt))
4531 {
4532 ERR("unexpected time zone %s\n", debugstr_w(ptr));
4533 return FALSE;
4534 }
4535 return SystemTimeToFileTime(&st, ft);
4536 }
4537
4538 static WORD HTTP_ParseWeekday(LPCWSTR day)
4539 {
4540 static const WCHAR days[7][10] = {{ 's','u','n','d','a','y',0 },
4541 { 'm','o','n','d','a','y',0 },
4542 { 't','u','e','s','d','a','y',0 },
4543 { 'w','e','d','n','e','s','d','a','y',0 },
4544 { 't','h','u','r','s','d','a','y',0 },
4545 { 'f','r','i','d','a','y',0 },
4546 { 's','a','t','u','r','d','a','y',0 }};
4547 unsigned int i;
4548 for (i = 0; i < sizeof(days)/sizeof(*days); i++)
4549 if (!strcmpiW(day, days[i]))
4550 return i;
4551
4552 /* Invalid */
4553 return 7;
4554 }
4555
4556 static BOOL HTTP_ParseRfc850Date(LPCWSTR value, FILETIME *ft)
4557 {
4558 static const WCHAR gmt[]= { 'G','M','T',0 };
4559 WCHAR *nextPtr, day[10], month[4], *monthPtr;
4560 LPCWSTR ptr;
4561 unsigned long num;
4562 SYSTEMTIME st = { 0 };
4563
4564 ptr = strchrW(value, ',');
4565 if (!ptr)
4566 return FALSE;
4567 if (ptr - value == 3)
4568 {
4569 memcpy(day, value, (ptr - value) * sizeof(WCHAR));
4570 day[3] = 0;
4571 st.wDayOfWeek = HTTP_ParseWkday(day);
4572 if (st.wDayOfWeek > 6)
4573 {
4574 ERR("unexpected weekday %s\n", debugstr_wn(value, ptr - value));
4575 return FALSE;
4576 }
4577 }
4578 else if (ptr - value < sizeof(day) / sizeof(day[0]))
4579 {
4580 memcpy(day, value, (ptr - value) * sizeof(WCHAR));
4581 day[ptr - value + 1] = 0;
4582 st.wDayOfWeek = HTTP_ParseWeekday(day);
4583 if (st.wDayOfWeek > 6)
4584 {
4585 ERR("unexpected weekday %s\n", debugstr_wn(value, ptr - value));
4586 return FALSE;
4587 }
4588 }
4589 else
4590 {
4591 ERR("unexpected weekday %s\n", debugstr_wn(value, ptr - value));
4592 return FALSE;
4593 }
4594 ptr++;
4595
4596 while (isspaceW(*ptr))
4597 ptr++;
4598
4599 num = strtoulW(ptr, &nextPtr, 10);
4600 if (!nextPtr || nextPtr <= ptr || !num || num > 31)
4601 {
4602 ERR("unexpected day %s\n", debugstr_w(value));
4603 return FALSE;
4604 }
4605 ptr = nextPtr;
4606 st.wDay = (WORD)num;
4607
4608 if (*ptr != '-')
4609 {
4610 ERR("unexpected month format %s\n", debugstr_w(ptr));
4611 return FALSE;
4612 }
4613 ptr++;
4614
4615 for (monthPtr = month; *ptr != '-' &&
4616 monthPtr - month < sizeof(month) / sizeof(month[0]) - 1;
4617 monthPtr++, ptr++)
4618 *monthPtr = *ptr;
4619 *monthPtr = 0;
4620 st.wMonth = HTTP_ParseMonth(month);
4621 if (!st.wMonth || st.wMonth > 12)
4622 {
4623 ERR("unexpected month %s\n", debugstr_w(month));
4624 return FALSE;
4625 }
4626
4627 if (*ptr != '-')
4628 {
4629 ERR("unexpected year format %s\n", debugstr_w(ptr));
4630 return FALSE;
4631 }
4632 ptr++;
4633
4634 num = strtoulW(ptr, &nextPtr, 10);
4635 if (!nextPtr || nextPtr <= ptr || num < 1601 || num > 30827)
4636 {
4637 ERR("unexpected year %s\n", debugstr_w(value));
4638 return FALSE;
4639 }
4640 ptr = nextPtr;
4641 st.wYear = (WORD)num;
4642
4643 if (!HTTP_ParseTime(&st, &ptr))
4644 return FALSE;
4645
4646 while (isspaceW(*ptr))
4647 ptr++;
4648
4649 if (strcmpW(ptr, gmt))
4650 {
4651 ERR("unexpected time zone %s\n", debugstr_w(ptr));
4652 return FALSE;
4653 }
4654 return SystemTimeToFileTime(&st, ft);
4655 }
4656
4657 static BOOL HTTP_ParseDate(LPCWSTR value, FILETIME *ft)
4658 {
4659 static const WCHAR zero[] = { '0',0 };
4660 BOOL ret;
4661
4662 if (!strcmpW(value, zero))
4663 {
4664 ft->dwLowDateTime = ft->dwHighDateTime = 0;
4665 ret = TRUE;
4666 }
4667 else if (strchrW(value, ','))
4668 {
4669 ret = HTTP_ParseRfc1123Date(value, ft);
4670 if (!ret)
4671 {
4672 ret = HTTP_ParseRfc850Date(value, ft);
4673 if (!ret)
4674 ERR("unexpected date format %s\n", debugstr_w(value));
4675 }
4676 }
4677 else
4678 {
4679 ret = HTTP_ParseDateAsAsctime(value, ft);
4680 if (!ret)
4681 ERR("unexpected date format %s\n", debugstr_w(value));
4682 }
4683 return ret;
4684 }
4685
4686 static void HTTP_ProcessExpires(http_request_t *request)
4687 {
4688 BOOL expirationFound = FALSE;
4689 int headerIndex;
4690
4691 EnterCriticalSection( &request->headers_section );
4692
4693 /* Look for a Cache-Control header with a max-age directive, as it takes
4694 * precedence over the Expires header.
4695 */
4696 headerIndex = HTTP_GetCustomHeaderIndex(request, szCache_Control, 0, FALSE);
4697 if (headerIndex != -1)
4698 {
4699 LPHTTPHEADERW ccHeader = &request->custHeaders[headerIndex];
4700 LPWSTR ptr;
4701
4702 for (ptr = ccHeader->lpszValue; ptr && *ptr; )
4703 {
4704 LPWSTR comma = strchrW(ptr, ','), end, equal;
4705
4706 if (comma)
4707 end = comma;
4708 else
4709 end = ptr + strlenW(ptr);
4710 for (equal = end - 1; equal > ptr && *equal != '='; equal--)
4711 ;
4712 if (*equal == '=')
4713 {
4714 static const WCHAR max_age[] = {
4715 'm','a','x','-','a','g','e',0 };
4716
4717 if (!strncmpiW(ptr, max_age, equal - ptr - 1))
4718 {
4719 LPWSTR nextPtr;
4720 unsigned long age;
4721
4722 age = strtoulW(equal + 1, &nextPtr, 10);
4723 if (nextPtr > equal + 1)
4724 {
4725 LARGE_INTEGER ft;
4726
4727 NtQuerySystemTime( &ft );
4728 /* Age is in seconds, FILETIME resolution is in
4729 * 100 nanosecond intervals.
4730 */
4731 ft.QuadPart += age * (ULONGLONG)1000000;
4732 request->expires.dwLowDateTime = ft.u.LowPart;
4733 request->expires.dwHighDateTime = ft.u.HighPart;
4734 expirationFound = TRUE;
4735 }
4736 }
4737 }
4738 if (comma)
4739 {
4740 ptr = comma + 1;
4741 while (isspaceW(*ptr))
4742 ptr++;
4743 }
4744 else
4745 ptr = NULL;
4746 }
4747 }
4748 if (!expirationFound)
4749 {
4750 headerIndex = HTTP_GetCustomHeaderIndex(request, szExpires, 0, FALSE);
4751 if (headerIndex != -1)
4752 {
4753 LPHTTPHEADERW expiresHeader = &request->custHeaders[headerIndex];
4754 FILETIME ft;
4755
4756 if (HTTP_ParseDate(expiresHeader->lpszValue, &ft))
4757 {
4758 expirationFound = TRUE;
4759 request->expires = ft;
4760 }
4761 }
4762 }
4763 if (!expirationFound)
4764 {
4765 LARGE_INTEGER t;
4766
4767 /* With no known age, default to 10 minutes until expiration. */
4768 NtQuerySystemTime( &t );
4769 t.QuadPart += 10 * 60 * (ULONGLONG)10000000;
4770 request->expires.dwLowDateTime = t.u.LowPart;
4771 request->expires.dwHighDateTime = t.u.HighPart;
4772 }
4773
4774 LeaveCriticalSection( &request->headers_section );
4775 }
4776
4777 static void HTTP_ProcessLastModified(http_request_t *request)
4778 {
4779 int headerIndex;
4780
4781 EnterCriticalSection( &request->headers_section );
4782
4783 headerIndex = HTTP_GetCustomHeaderIndex(request, szLast_Modified, 0, FALSE);
4784 if (headerIndex != -1)
4785 {
4786 LPHTTPHEADERW expiresHeader = &request->custHeaders[headerIndex];
4787 FILETIME ft;
4788
4789 if (HTTP_ParseDate(expiresHeader->lpszValue, &ft))
4790 request->last_modified = ft;
4791 }
4792
4793 LeaveCriticalSection( &request->headers_section );
4794 }
4795
4796 static void http_process_keep_alive(http_request_t *req)
4797 {
4798 int index;
4799
4800 EnterCriticalSection( &req->headers_section );
4801
4802 if ((index = HTTP_GetCustomHeaderIndex(req, szConnection, 0, FALSE)) != -1)
4803 req->netconn->keep_alive = !strcmpiW(req->custHeaders[index].lpszValue, szKeepAlive);
4804 else if ((index = HTTP_GetCustomHeaderIndex(req, szProxy_Connection, 0, FALSE)) != -1)
4805 req->netconn->keep_alive = !strcmpiW(req->custHeaders[index].lpszValue, szKeepAlive);
4806 else
4807 req->netconn->keep_alive = !strcmpiW(req->version, g_szHttp1_1);
4808
4809 LeaveCriticalSection( &req->headers_section );
4810 }
4811
4812 static DWORD open_http_connection(http_request_t *request, BOOL *reusing)
4813 {
4814 const BOOL is_https = (request->hdr.dwFlags & INTERNET_FLAG_SECURE) != 0;
4815 netconn_t *netconn = NULL;
4816 DWORD res;
4817
4818 reset_data_stream(request);
4819
4820 if (request->netconn)
4821 {
4822 if (is_valid_netconn(request->netconn) && NETCON_is_alive(request->netconn))
4823 {
4824 *reusing = TRUE;
4825 return ERROR_SUCCESS;
4826 }
4827 else
4828 {
4829 free_netconn(request->netconn);
4830 request->netconn = NULL;
4831 }
4832 }
4833
4834 res = HTTP_ResolveName(request);
4835 if(res != ERROR_SUCCESS)
4836 return res;
4837
4838 EnterCriticalSection(&connection_pool_cs);
4839
4840 while(!list_empty(&request->server->conn_pool)) {
4841 netconn = LIST_ENTRY(list_head(&request->server->conn_pool), netconn_t, pool_entry);
4842 list_remove(&netconn->pool_entry);
4843
4844 if(is_valid_netconn(netconn) && NETCON_is_alive(netconn))
4845 break;
4846
4847 TRACE("connection %p closed during idle\n", netconn);
4848 free_netconn(netconn);
4849 netconn = NULL;
4850 }
4851
4852 LeaveCriticalSection(&connection_pool_cs);
4853
4854 if(netconn) {
4855 TRACE("<-- reusing %p netconn\n", netconn);
4856 request->netconn = netconn;
4857 *reusing = TRUE;
4858 return ERROR_SUCCESS;
4859 }
4860
4861 TRACE("connecting to %s, proxy %s\n", debugstr_w(request->server->name),
4862 request->proxy ? debugstr_w(request->proxy->name) : "(null)");
4863
4864 INTERNET_SendCallback(&request->hdr, request->hdr.dwContext,
4865 INTERNET_STATUS_CONNECTING_TO_SERVER,
4866 request->server->addr_str,
4867 strlen(request->server->addr_str)+1);
4868
4869 res = create_netconn(is_https, request->proxy ? request->proxy : request->server, request->security_flags,
4870 (request->hdr.ErrorMask & INTERNET_ERROR_MASK_COMBINED_SEC_CERT) != 0,
4871 request->connect_timeout, &netconn);
4872 if(res != ERROR_SUCCESS) {
4873 ERR("create_netconn failed: %u\n", res);
4874 return res;
4875 }
4876
4877 request->netconn = netconn;
4878
4879 INTERNET_SendCallback(&request->hdr, request->hdr.dwContext,
4880 INTERNET_STATUS_CONNECTED_TO_SERVER,
4881 request->server->addr_str, strlen(request->server->addr_str)+1);
4882
4883 *reusing = FALSE;
4884 TRACE("Created connection to %s: %p\n", debugstr_w(request->server->name), netconn);
4885 return ERROR_SUCCESS;
4886 }
4887
4888 static char *build_ascii_request( const WCHAR *str, void *data, DWORD data_len, DWORD *out_len )
4889 {
4890 int len = WideCharToMultiByte( CP_ACP, 0, str, -1, NULL, 0, NULL, NULL );
4891 char *ret;
4892
4893 if (!(ret = heap_alloc( len + data_len ))) return NULL;
4894 WideCharToMultiByte( CP_ACP, 0, str, -1, ret, len, NULL, NULL );
4895 if (data_len) memcpy( ret + len - 1, data, data_len );
4896 *out_len = len + data_len - 1;
4897 ret[*out_len] = 0;
4898 return ret;
4899 }
4900
4901 /***********************************************************************
4902 * HTTP_HttpSendRequestW (internal)
4903 *
4904 * Sends the specified request to the HTTP server
4905 *
4906 * RETURNS
4907 * ERROR_SUCCESS on success
4908 * win32 error code on failure
4909 *
4910 */
4911 static DWORD HTTP_HttpSendRequestW(http_request_t *request, LPCWSTR lpszHeaders,
4912 DWORD dwHeaderLength, LPVOID lpOptional, DWORD dwOptionalLength,
4913 DWORD dwContentLength, BOOL bEndRequest)
4914 {
4915 static const WCHAR szContentLength[] =
4916 { 'C','o','n','t','e','n','t','-','L','e','n','g','t','h',':',' ','%','l','i','\r','\n',0 };
4917 BOOL redirected = FALSE, secure_proxy_connect = FALSE, loop_next;
4918 LPWSTR requestString = NULL;
4919 INT responseLen, cnt;
4920 WCHAR contentLengthStr[sizeof szContentLength/2 /* includes \r\n */ + 20 /* int */ ];
4921 DWORD res;
4922
4923 TRACE("--> %p\n", request);
4924
4925 assert(request->hdr.htype == WH_HHTTPREQ);
4926
4927 /* if the verb is NULL default to GET */
4928 if (!request->verb)
4929 request->verb = heap_strdupW(szGET);
4930
4931 if (dwContentLength || strcmpW(request->verb, szGET))
4932 {
4933 sprintfW(contentLengthStr, szContentLength, dwContentLength);
4934 HTTP_HttpAddRequestHeadersW(request, contentLengthStr, -1L, HTTP_ADDREQ_FLAG_ADD_IF_NEW);
4935 request->bytesToWrite = dwContentLength;
4936 }
4937 if (request->session->appInfo->agent)
4938 {
4939 WCHAR *agent_header;
4940 static const WCHAR user_agent[] = {'U','s','e','r','-','A','g','e','n','t',':',' ','%','s','\r','\n',0};
4941 int len;
4942
4943 len = strlenW(request->session->appInfo->agent) + strlenW(user_agent);
4944 agent_header = heap_alloc(len * sizeof(WCHAR));
4945 sprintfW(agent_header, user_agent, request->session->appInfo->agent);
4946
4947 HTTP_HttpAddRequestHeadersW(request, agent_header, strlenW(agent_header), HTTP_ADDREQ_FLAG_ADD_IF_NEW);
4948 heap_free(agent_header);
4949 }
4950 if (request->hdr.dwFlags & INTERNET_FLAG_PRAGMA_NOCACHE)
4951 {
4952 static const WCHAR pragma_nocache[] = {'P','r','a','g','m','a',':',' ','n','o','-','c','a','c','h','e','\r','\n',0};
4953 HTTP_HttpAddRequestHeadersW(request, pragma_nocache, strlenW(pragma_nocache), HTTP_ADDREQ_FLAG_ADD_IF_NEW);
4954 }
4955 if ((request->hdr.dwFlags & INTERNET_FLAG_NO_CACHE_WRITE) && strcmpW(request->verb, szGET))
4956 {
4957 static const WCHAR cache_control[] = {'C','a','c','h','e','-','C','o','n','t','r','o','l',':',
4958 ' ','n','o','-','c','a','c','h','e','\r','\n',0};
4959 HTTP_HttpAddRequestHeadersW(request, cache_control, strlenW(cache_control), HTTP_ADDREQ_FLAG_ADD_IF_NEW);
4960 }
4961
4962 /* add the headers the caller supplied */
4963 if( lpszHeaders && dwHeaderLength )
4964 HTTP_HttpAddRequestHeadersW(request, lpszHeaders, dwHeaderLength, HTTP_ADDREQ_FLAG_ADD | HTTP_ADDHDR_FLAG_REPLACE);
4965
4966 do
4967 {
4968 DWORD len, data_len = dwOptionalLength;
4969 BOOL reusing_connection;
4970 char *ascii_req;
4971
4972 loop_next = FALSE;
4973
4974 if(redirected) {
4975 request->contentLength = ~0u;
4976 request->bytesToWrite = 0;
4977 }
4978
4979 if (TRACE_ON(wininet))
4980 {
4981 HTTPHEADERW *host;
4982
4983 EnterCriticalSection( &request->headers_section );
4984 host = HTTP_GetHeader( request, hostW );
4985 TRACE("Going to url %s %s\n", debugstr_w(host->lpszValue), debugstr_w(request->path));
4986 LeaveCriticalSection( &request->headers_section );
4987 }
4988
4989 HTTP_FixURL(request);
4990 if (request->hdr.dwFlags & INTERNET_FLAG_KEEP_CONNECTION)
4991 {
4992 HTTP_ProcessHeader(request, szConnection, szKeepAlive, HTTP_ADDHDR_FLAG_REQ | HTTP_ADDHDR_FLAG_REPLACE);
4993 }
4994 HTTP_InsertAuthorization(request, request->authInfo, szAuthorization);
4995 HTTP_InsertAuthorization(request, request->proxyAuthInfo, szProxy_Authorization);
4996
4997 if (!(request->hdr.dwFlags & INTERNET_FLAG_NO_COOKIES))
4998 HTTP_InsertCookies(request);
4999
5000 res = open_http_connection(request, &reusing_connection);
5001 if (res != ERROR_SUCCESS)
5002 break;
5003
5004 if (!reusing_connection && (request->hdr.dwFlags & INTERNET_FLAG_SECURE))
5005 {
5006 if (request->proxy) secure_proxy_connect = TRUE;
5007 else
5008 {
5009 res = NETCON_secure_connect(request->netconn, request->server);
5010 if (res != ERROR_SUCCESS)
5011 {
5012 WARN("failed to upgrade to secure connection\n");
5013 http_release_netconn(request, FALSE);
5014 break;
5015 }
5016 }
5017 }
5018 if (secure_proxy_connect)
5019 {
5020 static const WCHAR connectW[] = {'C','O','N','N','E','C','T',0};
5021 const WCHAR *target = request->server->host_port;
5022 requestString = build_request_header(request, connectW, target, g_szHttp1_1, TRUE);
5023 }
5024 else if (request->proxy && !(request->hdr.dwFlags & INTERNET_FLAG_SECURE))
5025 {
5026 WCHAR *url = build_proxy_path_url(request);
5027 requestString = build_request_header(request, request->verb, url, request->version, TRUE);
5028 heap_free(url);
5029 }
5030 else
5031 requestString = build_request_header(request, request->verb, request->path, request->version, TRUE);
5032
5033 TRACE("Request header -> %s\n", debugstr_w(requestString) );
5034
5035 /* send the request as ASCII, tack on the optional data */
5036 if (!lpOptional || redirected || secure_proxy_connect)
5037 data_len = 0;
5038
5039 ascii_req = build_ascii_request( requestString, lpOptional, data_len, &len );
5040 TRACE("full request -> %s\n", debugstr_a(ascii_req) );
5041
5042 INTERNET_SendCallback(&request->hdr, request->hdr.dwContext,
5043 INTERNET_STATUS_SENDING_REQUEST, NULL, 0);
5044
5045 NETCON_set_timeout( request->netconn, TRUE, request->send_timeout );
5046 res = NETCON_send(request->netconn, ascii_req, len, 0, &cnt);
5047 heap_free( ascii_req );
5048 if(res != ERROR_SUCCESS) {
5049 TRACE("send failed: %u\n", res);
5050 if(!reusing_connection)
5051 break;
5052 http_release_netconn(request, FALSE);
5053 loop_next = TRUE;
5054 continue;
5055 }
5056
5057 request->bytesWritten = data_len;
5058
5059 INTERNET_SendCallback(&request->hdr, request->hdr.dwContext,
5060 INTERNET_STATUS_REQUEST_SENT,
5061 &len, sizeof(DWORD));
5062
5063 if (bEndRequest)
5064 {
5065 DWORD dwBufferSize;
5066
5067 INTERNET_SendCallback(&request->hdr, request->hdr.dwContext,
5068 INTERNET_STATUS_RECEIVING_RESPONSE, NULL, 0);
5069
5070 if (HTTP_GetResponseHeaders(request, &responseLen))
5071 {
5072 http_release_netconn(request, FALSE);
5073 res = ERROR_INTERNET_CONNECTION_ABORTED;
5074 goto lend;
5075 }
5076 /* FIXME: We should know that connection is closed before sending
5077 * headers. Otherwise wrong callbacks are executed */
5078 if(!responseLen && reusing_connection) {
5079 TRACE("Connection closed by server, reconnecting\n");
5080 http_release_netconn(request, FALSE);
5081 loop_next = TRUE;
5082 continue;
5083 }
5084
5085 INTERNET_SendCallback(&request->hdr, request->hdr.dwContext,
5086 INTERNET_STATUS_RESPONSE_RECEIVED, &responseLen,
5087 sizeof(DWORD));
5088
5089 http_process_keep_alive(request);
5090 HTTP_ProcessCookies(request);
5091 HTTP_ProcessExpires(request);
5092 HTTP_ProcessLastModified(request);
5093
5094 res = set_content_length(request);
5095 if(res != ERROR_SUCCESS)
5096 goto lend;
5097 if(!request->contentLength)
5098 http_release_netconn(request, TRUE);
5099
5100 if (!(request->hdr.dwFlags & INTERNET_FLAG_NO_AUTO_REDIRECT) && responseLen)
5101 {
5102 WCHAR *new_url, szNewLocation[INTERNET_MAX_URL_LENGTH];
5103 dwBufferSize=sizeof(szNewLocation);
5104 switch(request->status_code) {
5105 case HTTP_STATUS_REDIRECT:
5106 case HTTP_STATUS_MOVED:
5107 case HTTP_STATUS_REDIRECT_KEEP_VERB:
5108 case HTTP_STATUS_REDIRECT_METHOD:
5109 if(HTTP_HttpQueryInfoW(request,HTTP_QUERY_LOCATION,szNewLocation,&dwBufferSize,NULL) != ERROR_SUCCESS)
5110 break;
5111
5112 if (strcmpW(request->verb, szGET) && strcmpW(request->verb, szHEAD) &&
5113 request->status_code != HTTP_STATUS_REDIRECT_KEEP_VERB)
5114 {
5115 heap_free(request->verb);
5116 request->verb = heap_strdupW(szGET);
5117 }
5118 http_release_netconn(request, drain_content(request, FALSE));
5119 if ((new_url = HTTP_GetRedirectURL( request, szNewLocation )))
5120 {
5121 INTERNET_SendCallback(&request->hdr, request->hdr.dwContext, INTERNET_STATUS_REDIRECT,
5122 new_url, (strlenW(new_url) + 1) * sizeof(WCHAR));
5123 res = HTTP_HandleRedirect(request, new_url);
5124 if (res == ERROR_SUCCESS)
5125 {
5126 heap_free(requestString);
5127 loop_next = TRUE;
5128 }
5129 heap_free( new_url );
5130 }
5131 redirected = TRUE;
5132 }
5133 }
5134 if (!(request->hdr.dwFlags & INTERNET_FLAG_NO_AUTH) && res == ERROR_SUCCESS)
5135 {
5136 WCHAR szAuthValue[2048];
5137 dwBufferSize=2048;
5138 if (request->status_code == HTTP_STATUS_DENIED)
5139 {
5140 WCHAR *host = get_host_header( request );
5141 DWORD dwIndex = 0;
5142 while (HTTP_HttpQueryInfoW(request,HTTP_QUERY_WWW_AUTHENTICATE,szAuthValue,&dwBufferSize,&dwIndex) == ERROR_SUCCESS)
5143 {
5144 if (HTTP_DoAuthorization(request, szAuthValue,
5145 &request->authInfo,
5146 request->session->userName,
5147 request->session->password, host))
5148 {
5149 heap_free(requestString);
5150 if(!drain_content(request, TRUE)) {
5151 FIXME("Could not drain content\n");
5152 http_release_netconn(request, FALSE);
5153 }
5154 loop_next = TRUE;
5155 break;
5156 }
5157 }
5158 heap_free( host );
5159
5160 if(!loop_next) {
5161 TRACE("Cleaning wrong authorization data\n");
5162 destroy_authinfo(request->authInfo);
5163 request->authInfo = NULL;
5164 }
5165 }
5166 if (request->status_code == HTTP_STATUS_PROXY_AUTH_REQ)
5167 {
5168 DWORD dwIndex = 0;
5169 while (HTTP_HttpQueryInfoW(request,HTTP_QUERY_PROXY_AUTHENTICATE,szAuthValue,&dwBufferSize,&dwIndex) == ERROR_SUCCESS)
5170 {
5171 if (HTTP_DoAuthorization(request, szAuthValue,
5172 &request->proxyAuthInfo,
5173 request->session->appInfo->proxyUsername,
5174 request->session->appInfo->proxyPassword,
5175 NULL))
5176 {
5177 heap_free(requestString);
5178 if(!drain_content(request, TRUE)) {
5179 FIXME("Could not drain content\n");
5180 http_release_netconn(request, FALSE);
5181 }
5182 loop_next = TRUE;
5183 break;
5184 }
5185 }
5186
5187 if(!loop_next) {
5188 TRACE("Cleaning wrong proxy authorization data\n");
5189 destroy_authinfo(request->proxyAuthInfo);
5190 request->proxyAuthInfo = NULL;
5191 }
5192 }
5193 }
5194 if (secure_proxy_connect && request->status_code == HTTP_STATUS_OK)
5195 {
5196 int index;
5197
5198 res = NETCON_secure_connect(request->netconn, request->server);
5199 if (res != ERROR_SUCCESS)
5200 {
5201 WARN("failed to upgrade to secure proxy connection\n");
5202 http_release_netconn( request, FALSE );
5203 break;
5204 }
5205 EnterCriticalSection( &request->headers_section );
5206 index = HTTP_GetCustomHeaderIndex(request, szProxy_Authorization, 0, TRUE);
5207 if (index != -1) HTTP_DeleteCustomHeader(request, index);
5208 LeaveCriticalSection( &request->headers_section );
5209
5210 destroy_authinfo(request->proxyAuthInfo);
5211 request->proxyAuthInfo = NULL;
5212
5213 secure_proxy_connect = FALSE;
5214 loop_next = TRUE;
5215 }
5216 }
5217 else
5218 res = ERROR_SUCCESS;
5219 }
5220 while (loop_next);
5221
5222 lend:
5223 heap_free(requestString);
5224
5225 /* TODO: send notification for P3P header */
5226
5227 if(res == ERROR_SUCCESS)
5228 create_cache_entry(request);
5229
5230 if (request->session->appInfo->hdr.dwFlags & INTERNET_FLAG_ASYNC)
5231 {
5232 if (res == ERROR_SUCCESS) {
5233 if(bEndRequest && request->contentLength && request->bytesWritten == request->bytesToWrite)
5234 HTTP_ReceiveRequestData(request, TRUE, NULL);
5235 else
5236 send_request_complete(request,
5237 request->session->hdr.dwInternalFlags & INET_OPENURL ? (DWORD_PTR)request->hdr.hInternet : 1, 0);
5238 }else {
5239 send_request_complete(request, 0, res);
5240 }
5241 }
5242
5243 TRACE("<--\n");
5244 return res;
5245 }
5246
5247 typedef struct {
5248 task_header_t hdr;
5249 WCHAR *headers;
5250 DWORD headers_len;
5251 void *optional;
5252 DWORD optional_len;
5253 DWORD content_len;
5254 BOOL end_request;
5255 } send_request_task_t;
5256
5257 /***********************************************************************
5258 *
5259 * Helper functions for the HttpSendRequest(Ex) functions
5260 *
5261 */
5262 static void AsyncHttpSendRequestProc(task_header_t *hdr)
5263 {
5264 send_request_task_t *task = (send_request_task_t*)hdr;
5265 http_request_t *request = (http_request_t*)task->hdr.hdr;
5266
5267 TRACE("%p\n", request);
5268
5269 HTTP_HttpSendRequestW(request, task->headers, task->headers_len, task->optional,
5270 task->optional_len, task->content_len, task->end_request);
5271
5272 heap_free(task->headers);
5273 }
5274
5275
5276 static DWORD HTTP_HttpEndRequestW(http_request_t *request, DWORD dwFlags, DWORD_PTR dwContext)
5277 {
5278 DWORD dwBufferSize;
5279 INT responseLen;
5280 DWORD res = ERROR_SUCCESS;
5281
5282 if(!is_valid_netconn(request->netconn)) {
5283 WARN("Not connected\n");
5284 send_request_complete(request, 0, ERROR_INTERNET_OPERATION_CANCELLED);
5285 return ERROR_INTERNET_OPERATION_CANCELLED;
5286 }
5287
5288 INTERNET_SendCallback(&request->hdr, request->hdr.dwContext,
5289 INTERNET_STATUS_RECEIVING_RESPONSE, NULL, 0);
5290
5291 if (HTTP_GetResponseHeaders(request, &responseLen) || !responseLen)
5292 res = ERROR_HTTP_HEADER_NOT_FOUND;
5293
5294 INTERNET_SendCallback(&request->hdr, request->hdr.dwContext,
5295 INTERNET_STATUS_RESPONSE_RECEIVED, &responseLen, sizeof(DWORD));
5296
5297 /* process cookies here. Is this right? */
5298 http_process_keep_alive(request);
5299 HTTP_ProcessCookies(request);
5300 HTTP_ProcessExpires(request);
5301 HTTP_ProcessLastModified(request);
5302
5303 if ((res = set_content_length(request)) == ERROR_SUCCESS) {
5304 if(!request->contentLength)
5305 http_release_netconn(request, TRUE);
5306 }
5307
5308 if (res == ERROR_SUCCESS && !(request->hdr.dwFlags & INTERNET_FLAG_NO_AUTO_REDIRECT))
5309 {
5310 switch(request->status_code) {
5311 case HTTP_STATUS_REDIRECT:
5312 case HTTP_STATUS_MOVED:
5313 case HTTP_STATUS_REDIRECT_METHOD:
5314 case HTTP_STATUS_REDIRECT_KEEP_VERB: {
5315 WCHAR *new_url, szNewLocation[INTERNET_MAX_URL_LENGTH];
5316 dwBufferSize=sizeof(szNewLocation);
5317 if (HTTP_HttpQueryInfoW(request, HTTP_QUERY_LOCATION, szNewLocation, &dwBufferSize, NULL) != ERROR_SUCCESS)
5318 break;
5319
5320 if (strcmpW(request->verb, szGET) && strcmpW(request->verb, szHEAD) &&
5321 request->status_code != HTTP_STATUS_REDIRECT_KEEP_VERB)
5322 {
5323 heap_free(request->verb);
5324 request->verb = heap_strdupW(szGET);
5325 }
5326 http_release_netconn(request, drain_content(request, FALSE));
5327 if ((new_url = HTTP_GetRedirectURL( request, szNewLocation )))
5328 {
5329 INTERNET_SendCallback(&request->hdr, request->hdr.dwContext, INTERNET_STATUS_REDIRECT,
5330 new_url, (strlenW(new_url) + 1) * sizeof(WCHAR));
5331 res = HTTP_HandleRedirect(request, new_url);
5332 if (res == ERROR_SUCCESS)
5333 res = HTTP_HttpSendRequestW(request, NULL, 0, NULL, 0, 0, TRUE);
5334 heap_free( new_url );
5335 }
5336 }
5337 }
5338 }
5339
5340 if(res == ERROR_SUCCESS)
5341 create_cache_entry(request);
5342
5343 if (res == ERROR_SUCCESS && request->contentLength)
5344 HTTP_ReceiveRequestData(request, TRUE, NULL);
5345 else
5346 send_request_complete(request, res == ERROR_SUCCESS, res);
5347
5348 return res;
5349 }
5350
5351 /***********************************************************************
5352 * HttpEndRequestA (WININET.@)
5353 *
5354 * Ends an HTTP request that was started by HttpSendRequestEx
5355 *
5356 * RETURNS
5357 * TRUE if successful
5358 * FALSE on failure
5359 *
5360 */
5361 BOOL WINAPI HttpEndRequestA(HINTERNET hRequest,
5362 LPINTERNET_BUFFERSA lpBuffersOut, DWORD dwFlags, DWORD_PTR dwContext)
5363 {
5364 TRACE("(%p, %p, %08x, %08lx)\n", hRequest, lpBuffersOut, dwFlags, dwContext);
5365
5366 if (lpBuffersOut)
5367 {
5368 SetLastError(ERROR_INVALID_PARAMETER);
5369 return FALSE;
5370 }
5371
5372 return HttpEndRequestW(hRequest, NULL, dwFlags, dwContext);
5373 }
5374
5375 typedef struct {
5376 task_header_t hdr;
5377 DWORD flags;
5378 DWORD context;
5379 } end_request_task_t;
5380
5381 static void AsyncHttpEndRequestProc(task_header_t *hdr)
5382 {
5383 end_request_task_t *task = (end_request_task_t*)hdr;
5384 http_request_t *req = (http_request_t*)task->hdr.hdr;
5385
5386 TRACE("%p\n", req);
5387
5388 HTTP_HttpEndRequestW(req, task->flags, task->context);
5389 }
5390
5391 /***********************************************************************
5392 * HttpEndRequestW (WININET.@)
5393 *
5394 * Ends an HTTP request that was started by HttpSendRequestEx
5395 *
5396 * RETURNS
5397 * TRUE if successful
5398 * FALSE on failure
5399 *
5400 */
5401 BOOL WINAPI HttpEndRequestW(HINTERNET hRequest,
5402 LPINTERNET_BUFFERSW lpBuffersOut, DWORD dwFlags, DWORD_PTR dwContext)
5403 {
5404 http_request_t *request;
5405 DWORD res;
5406
5407 TRACE("%p %p %x %lx -->\n", hRequest, lpBuffersOut, dwFlags, dwContext);
5408
5409 if (lpBuffersOut)
5410 {
5411 SetLastError(ERROR_INVALID_PARAMETER);
5412 return FALSE;
5413 }
5414
5415 request = (http_request_t*) get_handle_object( hRequest );
5416
5417 if (NULL == request || request->hdr.htype != WH_HHTTPREQ)
5418 {
5419 SetLastError(ERROR_INTERNET_INCORRECT_HANDLE_TYPE);
5420 if (request)
5421 WININET_Release( &request->hdr );
5422 return FALSE;
5423 }
5424 request->hdr.dwFlags |= dwFlags;
5425
5426 if (request->session->appInfo->hdr.dwFlags & INTERNET_FLAG_ASYNC)
5427 {
5428 end_request_task_t *task;
5429
5430 task = alloc_async_task(&request->hdr, AsyncHttpEndRequestProc, sizeof(*task));
5431 task->flags = dwFlags;
5432 task->context = dwContext;
5433
5434 INTERNET_AsyncCall(&task->hdr);
5435 res = ERROR_IO_PENDING;
5436 }
5437 else
5438 res = HTTP_HttpEndRequestW(request, dwFlags, dwContext);
5439
5440 WININET_Release( &request->hdr );
5441 TRACE("%u <--\n", res);
5442 if(res != ERROR_SUCCESS)
5443 SetLastError(res);
5444 return res == ERROR_SUCCESS;
5445 }
5446
5447 /***********************************************************************
5448 * HttpSendRequestExA (WININET.@)
5449 *
5450 * Sends the specified request to the HTTP server and allows chunked
5451 * transfers.
5452 *
5453 * RETURNS
5454 * Success: TRUE
5455 * Failure: FALSE, call GetLastError() for more information.
5456 */
5457 BOOL WINAPI HttpSendRequestExA(HINTERNET hRequest,
5458 LPINTERNET_BUFFERSA lpBuffersIn,
5459 LPINTERNET_BUFFERSA lpBuffersOut,
5460 DWORD dwFlags, DWORD_PTR dwContext)
5461 {
5462 INTERNET_BUFFERSW BuffersInW;
5463 BOOL rc = FALSE;
5464 DWORD headerlen;
5465 LPWSTR header = NULL;
5466
5467 TRACE("(%p, %p, %p, %08x, %08lx)\n", hRequest, lpBuffersIn,
5468 lpBuffersOut, dwFlags, dwContext);
5469
5470 if (lpBuffersIn)
5471 {
5472 BuffersInW.dwStructSize = sizeof(LPINTERNET_BUFFERSW);
5473 if (lpBuffersIn->lpcszHeader)
5474 {
5475 headerlen = MultiByteToWideChar(CP_ACP,0,lpBuffersIn->lpcszHeader,
5476 lpBuffersIn->dwHeadersLength,0,0);
5477 header = heap_alloc(headerlen*sizeof(WCHAR));
5478 if (!(BuffersInW.lpcszHeader = header))
5479 {
5480 SetLastError(ERROR_OUTOFMEMORY);
5481 return FALSE;
5482 }
5483 BuffersInW.dwHeadersLength = MultiByteToWideChar(CP_ACP, 0,
5484 lpBuffersIn->lpcszHeader, lpBuffersIn->dwHeadersLength,
5485 header, headerlen);
5486 }
5487 else
5488 BuffersInW.lpcszHeader = NULL;
5489 BuffersInW.dwHeadersTotal = lpBuffersIn->dwHeadersTotal;
5490 BuffersInW.lpvBuffer = lpBuffersIn->lpvBuffer;
5491 BuffersInW.dwBufferLength = lpBuffersIn->dwBufferLength;
5492 BuffersInW.dwBufferTotal = lpBuffersIn->dwBufferTotal;
5493 BuffersInW.Next = NULL;
5494 }
5495
5496 rc = HttpSendRequestExW(hRequest, lpBuffersIn ? &BuffersInW : NULL, NULL, dwFlags, dwContext);
5497
5498 heap_free(header);
5499 return rc;
5500 }
5501
5502 /***********************************************************************
5503 * HttpSendRequestExW (WININET.@)
5504 *
5505 * Sends the specified request to the HTTP server and allows chunked
5506 * transfers
5507 *
5508 * RETURNS
5509 * Success: TRUE
5510 * Failure: FALSE, call GetLastError() for more information.
5511 */
5512 BOOL WINAPI HttpSendRequestExW(HINTERNET hRequest,
5513 LPINTERNET_BUFFERSW lpBuffersIn,
5514 LPINTERNET_BUFFERSW lpBuffersOut,
5515 DWORD dwFlags, DWORD_PTR dwContext)
5516 {
5517 http_request_t *request;
5518 http_session_t *session;
5519 appinfo_t *hIC;
5520 DWORD res;
5521
5522 TRACE("(%p, %p, %p, %08x, %08lx)\n", hRequest, lpBuffersIn,
5523 lpBuffersOut, dwFlags, dwContext);
5524
5525 request = (http_request_t*) get_handle_object( hRequest );
5526
5527 if (NULL == request || request->hdr.htype != WH_HHTTPREQ)
5528 {
5529 res = ERROR_INTERNET_INCORRECT_HANDLE_TYPE;
5530 goto lend;
5531 }
5532
5533 session = request->session;
5534 assert(session->hdr.htype == WH_HHTTPSESSION);
5535 hIC = session->appInfo;
5536 assert(hIC->hdr.htype == WH_HINIT);
5537
5538 if (hIC->hdr.dwFlags & INTERNET_FLAG_ASYNC)
5539 {
5540 send_request_task_t *task;
5541
5542 task = alloc_async_task(&request->hdr, AsyncHttpSendRequestProc, sizeof(*task));
5543 if (lpBuffersIn)
5544 {
5545 DWORD size = 0;
5546
5547 if (lpBuffersIn->lpcszHeader)
5548 {
5549 if (lpBuffersIn->dwHeadersLength == ~0u)
5550 size = (strlenW( lpBuffersIn->lpcszHeader ) + 1) * sizeof(WCHAR);
5551 else
5552 size = lpBuffersIn->dwHeadersLength * sizeof(WCHAR);
5553
5554 task->headers = heap_alloc(size);
5555 memcpy(task->headers, lpBuffersIn->lpcszHeader, size);
5556 }
5557 else task->headers = NULL;
5558
5559 task->headers_len = size / sizeof(WCHAR);
5560 task->optional = lpBuffersIn->lpvBuffer;
5561 task->optional_len = lpBuffersIn->dwBufferLength;
5562 task->content_len = lpBuffersIn->dwBufferTotal;
5563 }
5564 else
5565 {
5566 task->headers = NULL;
5567 task->headers_len = 0;
5568 task->optional = NULL;
5569 task->optional_len = 0;
5570 task->content_len = 0;
5571 }
5572
5573 task->end_request = FALSE;
5574
5575 INTERNET_AsyncCall(&task->hdr);
5576 res = ERROR_IO_PENDING;
5577 }
5578 else
5579 {
5580 if (lpBuffersIn)
5581 res = HTTP_HttpSendRequestW(request, lpBuffersIn->lpcszHeader, lpBuffersIn->dwHeadersLength,
5582 lpBuffersIn->lpvBuffer, lpBuffersIn->dwBufferLength,
5583 lpBuffersIn->dwBufferTotal, FALSE);
5584 else
5585 res = HTTP_HttpSendRequestW(request, NULL, 0, NULL, 0, 0, FALSE);
5586 }
5587
5588 lend:
5589 if ( request )
5590 WININET_Release( &request->hdr );
5591
5592 TRACE("<---\n");
5593 SetLastError(res);
5594 return res == ERROR_SUCCESS;
5595 }
5596
5597 /***********************************************************************
5598 * HttpSendRequestW (WININET.@)
5599 *
5600 * Sends the specified request to the HTTP server
5601 *
5602 * RETURNS
5603 * TRUE on success
5604 * FALSE on failure
5605 *
5606 */
5607 BOOL WINAPI HttpSendRequestW(HINTERNET hHttpRequest, LPCWSTR lpszHeaders,
5608 DWORD dwHeaderLength, LPVOID lpOptional ,DWORD dwOptionalLength)
5609 {
5610 http_request_t *request;
5611 http_session_t *session = NULL;
5612 appinfo_t *hIC = NULL;
5613 DWORD res = ERROR_SUCCESS;
5614
5615 TRACE("%p, %s, %i, %p, %i)\n", hHttpRequest,
5616 debugstr_wn(lpszHeaders, dwHeaderLength), dwHeaderLength, lpOptional, dwOptionalLength);
5617
5618 request = (http_request_t*) get_handle_object( hHttpRequest );
5619 if (NULL == request || request->hdr.htype != WH_HHTTPREQ)
5620 {
5621 res = ERROR_INTERNET_INCORRECT_HANDLE_TYPE;
5622 goto lend;
5623 }
5624
5625 session = request->session;
5626 if (NULL == session || session->hdr.htype != WH_HHTTPSESSION)
5627 {
5628 res = ERROR_INTERNET_INCORRECT_HANDLE_TYPE;
5629 goto lend;
5630 }
5631
5632 hIC = session->appInfo;
5633 if (NULL == hIC || hIC->hdr.htype != WH_HINIT)
5634 {
5635 res = ERROR_INTERNET_INCORRECT_HANDLE_TYPE;
5636 goto lend;
5637 }
5638
5639 if (hIC->hdr.dwFlags & INTERNET_FLAG_ASYNC)
5640 {
5641 send_request_task_t *task;
5642
5643 task = alloc_async_task(&request->hdr, AsyncHttpSendRequestProc, sizeof(*task));
5644 if (lpszHeaders)
5645 {
5646 DWORD size;
5647
5648 if (dwHeaderLength == ~0u) size = (strlenW(lpszHeaders) + 1) * sizeof(WCHAR);
5649 else size = dwHeaderLength * sizeof(WCHAR);
5650
5651 task->headers = heap_alloc(size);
5652 memcpy(task->headers, lpszHeaders, size);
5653 }
5654 else
5655 task->headers = NULL;
5656 task->headers_len = dwHeaderLength;
5657 task->optional = lpOptional;
5658 task->optional_len = dwOptionalLength;
5659 task->content_len = dwOptionalLength;
5660 task->end_request = TRUE;
5661
5662 INTERNET_AsyncCall(&task->hdr);
5663 res = ERROR_IO_PENDING;
5664 }
5665 else
5666 {
5667 res = HTTP_HttpSendRequestW(request, lpszHeaders,
5668 dwHeaderLength, lpOptional, dwOptionalLength,
5669 dwOptionalLength, TRUE);
5670 }
5671 lend:
5672 if( request )
5673 WININET_Release( &request->hdr );
5674
5675 SetLastError(res);
5676 return res == ERROR_SUCCESS;
5677 }
5678
5679 /***********************************************************************
5680 * HttpSendRequestA (WININET.@)
5681 *
5682 * Sends the specified request to the HTTP server
5683 *
5684 * RETURNS
5685 * TRUE on success
5686 * FALSE on failure
5687 *
5688 */
5689 BOOL WINAPI HttpSendRequestA(HINTERNET hHttpRequest, LPCSTR lpszHeaders,
5690 DWORD dwHeaderLength, LPVOID lpOptional ,DWORD dwOptionalLength)
5691 {
5692 BOOL result;
5693 LPWSTR szHeaders=NULL;
5694 DWORD nLen=dwHeaderLength;
5695 if(lpszHeaders!=NULL)
5696 {
5697 nLen=MultiByteToWideChar(CP_ACP,0,lpszHeaders,dwHeaderLength,NULL,0);
5698 szHeaders = heap_alloc(nLen*sizeof(WCHAR));
5699 MultiByteToWideChar(CP_ACP,0,lpszHeaders,dwHeaderLength,szHeaders,nLen);
5700 }
5701 result = HttpSendRequestW(hHttpRequest, szHeaders, nLen, lpOptional, dwOptionalLength);
5702 heap_free(szHeaders);
5703 return result;
5704 }
5705
5706 /***********************************************************************
5707 * HTTPSESSION_Destroy (internal)
5708 *
5709 * Deallocate session handle
5710 *
5711 */
5712 static void HTTPSESSION_Destroy(object_header_t *hdr)
5713 {
5714 http_session_t *session = (http_session_t*) hdr;
5715
5716 TRACE("%p\n", session);
5717
5718 WININET_Release(&session->appInfo->hdr);
5719
5720 heap_free(session->hostName);
5721 heap_free(session->password);
5722 heap_free(session->userName);
5723 }
5724
5725 static DWORD HTTPSESSION_QueryOption(object_header_t *hdr, DWORD option, void *buffer, DWORD *size, BOOL unicode)
5726 {
5727 http_session_t *ses = (http_session_t *)hdr;
5728
5729 switch(option) {
5730 case INTERNET_OPTION_HANDLE_TYPE:
5731 TRACE("INTERNET_OPTION_HANDLE_TYPE\n");
5732
5733 if (*size < sizeof(ULONG))
5734 return ERROR_INSUFFICIENT_BUFFER;
5735
5736 *size = sizeof(DWORD);
5737 *(DWORD*)buffer = INTERNET_HANDLE_TYPE_CONNECT_HTTP;
5738 return ERROR_SUCCESS;
5739 case INTERNET_OPTION_CONNECT_TIMEOUT:
5740 TRACE("INTERNET_OPTION_CONNECT_TIMEOUT\n");
5741
5742 if (*size < sizeof(DWORD))
5743 return ERROR_INSUFFICIENT_BUFFER;
5744
5745 *size = sizeof(DWORD);
5746 *(DWORD *)buffer = ses->connect_timeout;
5747 return ERROR_SUCCESS;
5748
5749 case INTERNET_OPTION_SEND_TIMEOUT:
5750 TRACE("INTERNET_OPTION_SEND_TIMEOUT\n");
5751
5752 if (*size < sizeof(DWORD))
5753 return ERROR_INSUFFICIENT_BUFFER;
5754
5755 *size = sizeof(DWORD);
5756 *(DWORD *)buffer = ses->send_timeout;
5757 return ERROR_SUCCESS;
5758
5759 case INTERNET_OPTION_RECEIVE_TIMEOUT:
5760 TRACE("INTERNET_OPTION_RECEIVE_TIMEOUT\n");
5761
5762 if (*size < sizeof(DWORD))
5763 return ERROR_INSUFFICIENT_BUFFER;
5764
5765 *size = sizeof(DWORD);
5766 *(DWORD *)buffer = ses->receive_timeout;
5767 return ERROR_SUCCESS;
5768 }
5769
5770 return INET_QueryOption(hdr, option, buffer, size, unicode);
5771 }
5772
5773 static DWORD HTTPSESSION_SetOption(object_header_t *hdr, DWORD option, void *buffer, DWORD size)
5774 {
5775 http_session_t *ses = (http_session_t*)hdr;
5776
5777 switch(option) {
5778 case INTERNET_OPTION_USERNAME:
5779 {
5780 heap_free(ses->userName);
5781 if (!(ses->userName = heap_strdupW(buffer))) return ERROR_OUTOFMEMORY;
5782 return ERROR_SUCCESS;
5783 }
5784 case INTERNET_OPTION_PASSWORD:
5785 {
5786 heap_free(ses->password);
5787 if (!(ses->password = heap_strdupW(buffer))) return ERROR_OUTOFMEMORY;
5788 return ERROR_SUCCESS;
5789 }
5790 case INTERNET_OPTION_PROXY_USERNAME:
5791 {
5792 heap_free(ses->appInfo->proxyUsername);
5793 if (!(ses->appInfo->proxyUsername = heap_strdupW(buffer))) return ERROR_OUTOFMEMORY;
5794 return ERROR_SUCCESS;
5795 }
5796 case INTERNET_OPTION_PROXY_PASSWORD:
5797 {
5798 heap_free(ses->appInfo->proxyPassword);
5799 if (!(ses->appInfo->proxyPassword = heap_strdupW(buffer))) return ERROR_OUTOFMEMORY;
5800 return ERROR_SUCCESS;
5801 }
5802 case INTERNET_OPTION_CONNECT_TIMEOUT:
5803 {
5804 if (!buffer || size != sizeof(DWORD)) return ERROR_INVALID_PARAMETER;
5805 ses->connect_timeout = *(DWORD *)buffer;
5806 return ERROR_SUCCESS;
5807 }
5808 case INTERNET_OPTION_SEND_TIMEOUT:
5809 {
5810 if (!buffer || size != sizeof(DWORD)) return ERROR_INVALID_PARAMETER;
5811 ses->send_timeout = *(DWORD *)buffer;
5812 return ERROR_SUCCESS;
5813 }
5814 case INTERNET_OPTION_RECEIVE_TIMEOUT:
5815 {
5816 if (!buffer || size != sizeof(DWORD)) return ERROR_INVALID_PARAMETER;
5817 ses->receive_timeout = *(DWORD *)buffer;
5818 return ERROR_SUCCESS;
5819 }
5820 default: break;
5821 }
5822
5823 return INET_SetOption(hdr, option, buffer, size);
5824 }
5825
5826 static const object_vtbl_t HTTPSESSIONVtbl = {
5827 HTTPSESSION_Destroy,
5828 NULL,
5829 HTTPSESSION_QueryOption,
5830 HTTPSESSION_SetOption,
5831 NULL,
5832 NULL,
5833 NULL,
5834 NULL,
5835 NULL
5836 };
5837
5838
5839 /***********************************************************************
5840 * HTTP_Connect (internal)
5841 *
5842 * Create http session handle
5843 *
5844 * RETURNS
5845 * HINTERNET a session handle on success
5846 * NULL on failure
5847 *
5848 */
5849 DWORD HTTP_Connect(appinfo_t *hIC, LPCWSTR lpszServerName,
5850 INTERNET_PORT serverPort, LPCWSTR lpszUserName,
5851 LPCWSTR lpszPassword, DWORD dwFlags, DWORD_PTR dwContext,
5852 DWORD dwInternalFlags, HINTERNET *ret)
5853 {
5854 http_session_t *session = NULL;
5855
5856 TRACE("-->\n");
5857
5858 if (!lpszServerName || !lpszServerName[0])
5859 return ERROR_INVALID_PARAMETER;
5860
5861 assert( hIC->hdr.htype == WH_HINIT );
5862
5863 session = alloc_object(&hIC->hdr, &HTTPSESSIONVtbl, sizeof(http_session_t));
5864 if (!session)
5865 return ERROR_OUTOFMEMORY;
5866
5867 /*
5868 * According to my tests. The name is not resolved until a request is sent
5869 */
5870
5871 session->hdr.htype = WH_HHTTPSESSION;
5872 session->hdr.dwFlags = dwFlags;
5873 session->hdr.dwContext = dwContext;
5874 session->hdr.dwInternalFlags |= dwInternalFlags;
5875
5876 WININET_AddRef( &hIC->hdr );
5877 session->appInfo = hIC;
5878 list_add_head( &hIC->hdr.children, &session->hdr.entry );
5879
5880 session->hostName = heap_strdupW(lpszServerName);
5881 if (lpszUserName && lpszUserName[0])
5882 session->userName = heap_strdupW(lpszUserName);
5883 if (lpszPassword && lpszPassword[0])
5884 session->password = heap_strdupW(lpszPassword);
5885 session->hostPort = serverPort;
5886 session->connect_timeout = hIC->connect_timeout;
5887 session->send_timeout = INFINITE;
5888 session->receive_timeout = INFINITE;
5889
5890 /* Don't send a handle created callback if this handle was created with InternetOpenUrl */
5891 if (!(session->hdr.dwInternalFlags & INET_OPENURL))
5892 {
5893 INTERNET_SendCallback(&hIC->hdr, dwContext,
5894 INTERNET_STATUS_HANDLE_CREATED, &session->hdr.hInternet,
5895 sizeof(HINTERNET));
5896 }
5897
5898 /*
5899 * an INTERNET_STATUS_REQUEST_COMPLETE is NOT sent here as per my tests on
5900 * windows
5901 */
5902
5903 TRACE("%p --> %p\n", hIC, session);
5904
5905 *ret = session->hdr.hInternet;
5906 return ERROR_SUCCESS;
5907 }
5908
5909 /***********************************************************************
5910 * HTTP_clear_response_headers (internal)
5911 *
5912 * clear out any old response headers
5913 */
5914 static void HTTP_clear_response_headers( http_request_t *request )
5915 {
5916 DWORD i;
5917
5918 EnterCriticalSection( &request->headers_section );
5919
5920 for( i=0; i<request->nCustHeaders; i++)
5921 {
5922 if( !request->custHeaders[i].lpszField )
5923 continue;
5924 if( !request->custHeaders[i].lpszValue )
5925 continue;
5926 if ( request->custHeaders[i].wFlags & HDR_ISREQUEST )
5927 continue;
5928 HTTP_DeleteCustomHeader( request, i );
5929 i--;
5930 }
5931
5932 LeaveCriticalSection( &request->headers_section );
5933 }
5934
5935 /***********************************************************************
5936 * HTTP_GetResponseHeaders (internal)
5937 *
5938 * Read server response
5939 *
5940 * RETURNS
5941 *
5942 * TRUE on success
5943 * FALSE on error
5944 */
5945 static DWORD HTTP_GetResponseHeaders(http_request_t *request, INT *len)
5946 {
5947 INT cbreaks = 0;
5948 WCHAR buffer[MAX_REPLY_LEN];
5949 DWORD buflen = MAX_REPLY_LEN;
5950 INT rc = 0;
5951 char bufferA[MAX_REPLY_LEN];
5952 LPWSTR status_code = NULL, status_text = NULL;
5953 DWORD res = ERROR_HTTP_INVALID_SERVER_RESPONSE;
5954 BOOL codeHundred = FALSE;
5955
5956 TRACE("-->\n");
5957
5958 if(!is_valid_netconn(request->netconn))
5959 goto lend;
5960
5961 /* clear old response headers (eg. from a redirect response) */
5962 HTTP_clear_response_headers( request );
5963
5964 NETCON_set_timeout( request->netconn, FALSE, request->receive_timeout );
5965 do {
5966 /*
5967 * We should first receive 'HTTP/1.x nnn OK' where nnn is the status code.
5968 */
5969 buflen = MAX_REPLY_LEN;
5970 if ((res = read_line(request, bufferA, &buflen)))
5971 goto lend;
5972
5973 if (!buflen) goto lend;
5974
5975 rc += buflen;
5976 MultiByteToWideChar( CP_ACP, 0, bufferA, buflen, buffer, MAX_REPLY_LEN );
5977 /* check is this a status code line? */
5978 if (!strncmpW(buffer, g_szHttp1_0, 4))
5979 {
5980 /* split the version from the status code */
5981 status_code = strchrW( buffer, ' ' );
5982 if( !status_code )
5983 goto lend;
5984 *status_code++=0;
5985
5986 /* split the status code from the status text */
5987 status_text = strchrW( status_code, ' ' );
5988 if( status_text )
5989 *status_text++=0;
5990
5991 request->status_code = atoiW(status_code);
5992
5993 TRACE("version [%s] status code [%s] status text [%s]\n",
5994 debugstr_w(buffer), debugstr_w(status_code), debugstr_w(status_text) );
5995
5996 codeHundred = request->status_code == HTTP_STATUS_CONTINUE;
5997 }
5998 else if (!codeHundred)
5999 {
6000 WARN("No status line at head of response (%s)\n", debugstr_w(buffer));
6001
6002 heap_free(request->version);
6003 heap_free(request->statusText);
6004
6005 request->status_code = HTTP_STATUS_OK;
6006 request->version = heap_strdupW(g_szHttp1_0);
6007 request->statusText = heap_strdupW(szOK);
6008
6009 goto lend;
6010 }
6011 } while (codeHundred);
6012
6013 /* Add status code */
6014 HTTP_ProcessHeader(request, szStatus, status_code,
6015 HTTP_ADDHDR_FLAG_REPLACE);
6016
6017 heap_free(request->version);
6018 heap_free(request->statusText);
6019
6020 request->version = heap_strdupW(buffer);
6021 request->statusText = heap_strdupW(status_text ? status_text : emptyW);
6022
6023 /* Restore the spaces */
6024 *(status_code-1) = ' ';
6025 if (status_text)
6026 *(status_text-1) = ' ';
6027
6028 /* Parse each response line */
6029 do
6030 {
6031 buflen = MAX_REPLY_LEN;
6032 if (!read_line(request, bufferA, &buflen) && buflen)
6033 {
6034 LPWSTR * pFieldAndValue;
6035
6036 TRACE("got line %s, now interpreting\n", debugstr_a(bufferA));
6037
6038 if (!bufferA[0]) break;
6039 MultiByteToWideChar( CP_ACP, 0, bufferA, buflen, buffer, MAX_REPLY_LEN );
6040
6041 pFieldAndValue = HTTP_InterpretHttpHeader(buffer);
6042 if (pFieldAndValue)
6043 {
6044 HTTP_ProcessHeader(request, pFieldAndValue[0], pFieldAndValue[1],
6045 HTTP_ADDREQ_FLAG_ADD );
6046 HTTP_FreeTokens(pFieldAndValue);
6047 }
6048 }
6049 else
6050 {
6051 cbreaks++;
6052 if (cbreaks >= 2)
6053 break;
6054 }
6055 }while(1);
6056
6057 res = ERROR_SUCCESS;
6058
6059 lend:
6060
6061 *len = rc;
6062 TRACE("<--\n");
6063 return res;
6064 }
6065
6066 /***********************************************************************
6067 * HTTP_InterpretHttpHeader (internal)
6068 *
6069 * Parse server response
6070 *
6071 * RETURNS
6072 *
6073 * Pointer to array of field, value, NULL on success.
6074 * NULL on error.
6075 */
6076 static LPWSTR * HTTP_InterpretHttpHeader(LPCWSTR buffer)
6077 {
6078 LPWSTR * pTokenPair;
6079 LPWSTR pszColon;
6080 INT len;
6081
6082 pTokenPair = heap_alloc_zero(sizeof(*pTokenPair)*3);
6083
6084 pszColon = strchrW(buffer, ':');
6085 /* must have two tokens */
6086 if (!pszColon)
6087 {
6088 HTTP_FreeTokens(pTokenPair);
6089 if (buffer[0])
6090 TRACE("No ':' in line: %s\n", debugstr_w(buffer));
6091 return NULL;
6092 }
6093
6094 pTokenPair[0] = heap_alloc((pszColon - buffer + 1) * sizeof(WCHAR));
6095 if (!pTokenPair[0])
6096 {
6097 HTTP_FreeTokens(pTokenPair);
6098 return NULL;
6099 }
6100 memcpy(pTokenPair[0], buffer, (pszColon - buffer) * sizeof(WCHAR));
6101 pTokenPair[0][pszColon - buffer] = '\0';
6102
6103 /* skip colon */
6104 pszColon++;
6105 len = strlenW(pszColon);
6106 pTokenPair[1] = heap_alloc((len + 1) * sizeof(WCHAR));
6107 if (!pTokenPair[1])
6108 {
6109 HTTP_FreeTokens(pTokenPair);
6110 return NULL;
6111 }
6112 memcpy(pTokenPair[1], pszColon, (len + 1) * sizeof(WCHAR));
6113
6114 strip_spaces(pTokenPair[0]);
6115 strip_spaces(pTokenPair[1]);
6116
6117 TRACE("field(%s) Value(%s)\n", debugstr_w(pTokenPair[0]), debugstr_w(pTokenPair[1]));
6118 return pTokenPair;
6119 }
6120
6121 /***********************************************************************
6122 * HTTP_ProcessHeader (internal)
6123 *
6124 * Stuff header into header tables according to <dwModifier>
6125 *
6126 */
6127
6128 #define COALESCEFLAGS (HTTP_ADDHDR_FLAG_COALESCE|HTTP_ADDHDR_FLAG_COALESCE_WITH_COMMA|HTTP_ADDHDR_FLAG_COALESCE_WITH_SEMICOLON)
6129
6130 static DWORD HTTP_ProcessHeader(http_request_t *request, LPCWSTR field, LPCWSTR value, DWORD dwModifier)
6131 {
6132 LPHTTPHEADERW lphttpHdr = NULL;
6133 INT index;
6134 BOOL request_only = dwModifier & HTTP_ADDHDR_FLAG_REQ;
6135 DWORD res = ERROR_HTTP_INVALID_HEADER;
6136
6137 TRACE("--> %s: %s - 0x%08x\n", debugstr_w(field), debugstr_w(value), dwModifier);
6138
6139 EnterCriticalSection( &request->headers_section );
6140
6141 /* REPLACE wins out over ADD */
6142 if (dwModifier & HTTP_ADDHDR_FLAG_REPLACE)
6143 dwModifier &= ~HTTP_ADDHDR_FLAG_ADD;
6144
6145 if (dwModifier & HTTP_ADDHDR_FLAG_ADD)
6146 index = -1;
6147 else
6148 index = HTTP_GetCustomHeaderIndex(request, field, 0, request_only);
6149
6150 if (index >= 0)
6151 {
6152 if (dwModifier & HTTP_ADDHDR_FLAG_ADD_IF_NEW)
6153 {
6154 LeaveCriticalSection( &request->headers_section );
6155 return ERROR_HTTP_INVALID_HEADER;
6156 }
6157 lphttpHdr = &request->custHeaders[index];
6158 }
6159 else if (value)
6160 {
6161 HTTPHEADERW hdr;
6162
6163 hdr.lpszField = (LPWSTR)field;
6164 hdr.lpszValue = (LPWSTR)value;
6165 hdr.wFlags = hdr.wCount = 0;
6166
6167 if (dwModifier & HTTP_ADDHDR_FLAG_REQ)
6168 hdr.wFlags |= HDR_ISREQUEST;
6169
6170 res = HTTP_InsertCustomHeader(request, &hdr);
6171 LeaveCriticalSection( &request->headers_section );
6172 return res;
6173 }
6174 /* no value to delete */
6175 else
6176 {
6177 LeaveCriticalSection( &request->headers_section );
6178 return ERROR_SUCCESS;
6179 }
6180
6181 if (dwModifier & HTTP_ADDHDR_FLAG_REQ)
6182 lphttpHdr->wFlags |= HDR_ISREQUEST;
6183 else
6184 lphttpHdr->wFlags &= ~HDR_ISREQUEST;
6185
6186 if (dwModifier & HTTP_ADDHDR_FLAG_REPLACE)
6187 {
6188 HTTP_DeleteCustomHeader( request, index );
6189
6190 if (value && value[0])
6191 {
6192 HTTPHEADERW hdr;
6193
6194 hdr.lpszField = (LPWSTR)field;
6195 hdr.lpszValue = (LPWSTR)value;
6196 hdr.wFlags = hdr.wCount = 0;
6197
6198 if (dwModifier & HTTP_ADDHDR_FLAG_REQ)
6199 hdr.wFlags |= HDR_ISREQUEST;
6200
6201 res = HTTP_InsertCustomHeader(request, &hdr);
6202 LeaveCriticalSection( &request->headers_section );
6203 return res;
6204 }
6205
6206 LeaveCriticalSection( &request->headers_section );
6207 return ERROR_SUCCESS;
6208 }
6209 else if (dwModifier & COALESCEFLAGS)
6210 {
6211 LPWSTR lpsztmp;
6212 WCHAR ch = 0;
6213 INT len = 0;
6214 INT origlen = strlenW(lphttpHdr->lpszValue);
6215 INT valuelen = strlenW(value);
6216
6217 if (dwModifier & HTTP_ADDHDR_FLAG_COALESCE_WITH_COMMA)
6218 {
6219 ch = ',';
6220 lphttpHdr->wFlags |= HDR_COMMADELIMITED;
6221 }
6222 else if (dwModifier & HTTP_ADDHDR_FLAG_COALESCE_WITH_SEMICOLON)
6223 {
6224 ch = ';';
6225 lphttpHdr->wFlags |= HDR_COMMADELIMITED;
6226 }
6227
6228 len = origlen + valuelen + ((ch > 0) ? 2 : 0);
6229
6230 lpsztmp = heap_realloc(lphttpHdr->lpszValue, (len+1)*sizeof(WCHAR));
6231 if (lpsztmp)
6232 {
6233 lphttpHdr->lpszValue = lpsztmp;
6234 /* FIXME: Increment lphttpHdr->wCount. Perhaps lpszValue should be an array */
6235 if (ch > 0)
6236 {
6237 lphttpHdr->lpszValue[origlen] = ch;
6238 origlen++;
6239 lphttpHdr->lpszValue[origlen] = ' ';
6240 origlen++;
6241 }
6242
6243 memcpy(&lphttpHdr->lpszValue[origlen], value, valuelen*sizeof(WCHAR));
6244 lphttpHdr->lpszValue[len] = '\0';
6245 res = ERROR_SUCCESS;
6246 }
6247 else
6248 {
6249 WARN("heap_realloc (%d bytes) failed\n",len+1);
6250 res = ERROR_OUTOFMEMORY;
6251 }
6252 }
6253 TRACE("<-- %d\n", res);
6254 LeaveCriticalSection( &request->headers_section );
6255 return res;
6256 }
6257
6258 /***********************************************************************
6259 * HTTP_GetCustomHeaderIndex (internal)
6260 *
6261 * Return index of custom header from header array
6262 * Headers section must be held
6263 */
6264 static INT HTTP_GetCustomHeaderIndex(http_request_t *request, LPCWSTR lpszField,
6265 int requested_index, BOOL request_only)
6266 {
6267 DWORD index;
6268
6269 TRACE("%s, %d, %d\n", debugstr_w(lpszField), requested_index, request_only);
6270
6271 for (index = 0; index < request->nCustHeaders; index++)
6272 {
6273 if (strcmpiW(request->custHeaders[index].lpszField, lpszField))
6274 continue;
6275
6276 if (request_only && !(request->custHeaders[index].wFlags & HDR_ISREQUEST))
6277 continue;
6278
6279 if (!request_only && (request->custHeaders[index].wFlags & HDR_ISREQUEST))
6280 continue;
6281
6282 if (requested_index == 0)
6283 break;
6284 requested_index --;
6285 }
6286
6287 if (index >= request->nCustHeaders)
6288 index = -1;
6289
6290 TRACE("Return: %d\n", index);
6291 return index;
6292 }
6293
6294
6295 /***********************************************************************
6296 * HTTP_InsertCustomHeader (internal)
6297 *
6298 * Insert header into array
6299 * Headers section must be held
6300 */
6301 static DWORD HTTP_InsertCustomHeader(http_request_t *request, LPHTTPHEADERW lpHdr)
6302 {
6303 INT count;
6304 LPHTTPHEADERW lph = NULL;
6305
6306 TRACE("--> %s: %s\n", debugstr_w(lpHdr->lpszField), debugstr_w(lpHdr->lpszValue));
6307 count = request->nCustHeaders + 1;
6308 if (count > 1)
6309 lph = heap_realloc_zero(request->custHeaders, sizeof(HTTPHEADERW) * count);
6310 else
6311 lph = heap_alloc_zero(sizeof(HTTPHEADERW) * count);
6312
6313 if (!lph)
6314 return ERROR_OUTOFMEMORY;
6315
6316 request->custHeaders = lph;
6317 request->custHeaders[count-1].lpszField = heap_strdupW(lpHdr->lpszField);
6318 request->custHeaders[count-1].lpszValue = heap_strdupW(lpHdr->lpszValue);
6319 request->custHeaders[count-1].wFlags = lpHdr->wFlags;
6320 request->custHeaders[count-1].wCount= lpHdr->wCount;
6321 request->nCustHeaders++;
6322
6323 return ERROR_SUCCESS;
6324 }
6325
6326
6327 /***********************************************************************
6328 * HTTP_DeleteCustomHeader (internal)
6329 *
6330 * Delete header from array
6331 * If this function is called, the index may change.
6332 * Headers section must be held
6333 */
6334 static BOOL HTTP_DeleteCustomHeader(http_request_t *request, DWORD index)
6335 {
6336 if( request->nCustHeaders <= 0 )
6337 return FALSE;
6338 if( index >= request->nCustHeaders )
6339 return FALSE;
6340 request->nCustHeaders--;
6341
6342 heap_free(request->custHeaders[index].lpszField);
6343 heap_free(request->custHeaders[index].lpszValue);
6344
6345 memmove( &request->custHeaders[index], &request->custHeaders[index+1],
6346 (request->nCustHeaders - index)* sizeof(HTTPHEADERW) );
6347 memset( &request->custHeaders[request->nCustHeaders], 0, sizeof(HTTPHEADERW) );
6348
6349 return TRUE;
6350 }
6351
6352
6353 /***********************************************************************
6354 * HTTP_VerifyValidHeader (internal)
6355 *
6356 * Verify the given header is not invalid for the given http request
6357 *
6358 */
6359 static BOOL HTTP_VerifyValidHeader(http_request_t *request, LPCWSTR field)
6360 {
6361 /* Accept-Encoding is stripped from HTTP/1.0 requests. It is invalid */
6362 if (!strcmpW(request->version, g_szHttp1_0) && !strcmpiW(field, szAccept_Encoding))
6363 return ERROR_HTTP_INVALID_HEADER;
6364
6365 return ERROR_SUCCESS;
6366 }
6367
6368 /***********************************************************************
6369 * IsHostInProxyBypassList (@)
6370 *
6371 * Undocumented
6372 *
6373 */
6374 BOOL WINAPI IsHostInProxyBypassList(DWORD flags, LPCSTR szHost, DWORD length)
6375 {
6376 FIXME("STUB: flags=%d host=%s length=%d\n",flags,szHost,length);
6377 return FALSE;
6378 }
Maarten's multimedia branch