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