ReactOS  0.4.14-dev-114-gc8cbd56
http.c
Go to the documentation of this file.
1 /*
2  * Wininet - HTTP Implementation
3  *
4  * Copyright 1999 Corel Corporation
5  * Copyright 2002 CodeWeavers Inc.
6  * Copyright 2002 TransGaming Technologies Inc.
7  * Copyright 2004 Mike McCormack for CodeWeavers
8  * Copyright 2005 Aric Stewart for CodeWeavers
9  * Copyright 2006 Robert Shearman for CodeWeavers
10  * Copyright 2011 Jacek Caban for CodeWeavers
11  *
12  * Ulrich Czekalla
13  * David Hammerton
14  *
15  * This library is free software; you can redistribute it and/or
16  * modify it under the terms of the GNU Lesser General Public
17  * License as published by the Free Software Foundation; either
18  * version 2.1 of the License, or (at your option) any later version.
19  *
20  * This library is distributed in the hope that it will be useful,
21  * but WITHOUT ANY WARRANTY; without even the implied warranty of
22  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
23  * Lesser General Public License for more details.
24  *
25  * You should have received a copy of the GNU Lesser General Public
26  * License along with this library; if not, write to the Free Software
27  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
28  */
29 
30 #include "config.h"
31 
32 #include <stdlib.h>
33 
34 #ifdef HAVE_ZLIB
35 # include <zlib.h>
36 #endif
37 
38 #include "winsock2.h"
39 #include "ws2ipdef.h"
40 
41 #include <stdarg.h>
42 #include <stdio.h>
43 #include <time.h>
44 #include <assert.h>
45 
46 #include "windef.h"
47 #include "winbase.h"
48 #include "wininet.h"
49 #include "winerror.h"
50 #include "winternl.h"
51 #define NO_SHLWAPI_STREAM
52 #define NO_SHLWAPI_REG
53 #define NO_SHLWAPI_STRFCNS
54 #define NO_SHLWAPI_GDI
55 #include "shlwapi.h"
56 #include "sspi.h"
57 #include "wincrypt.h"
58 #include "winuser.h"
59 
60 #include "internet.h"
61 #include "wine/debug.h"
62 #include "wine/exception.h"
63 #include "wine/unicode.h"
64 
66 
67 static const WCHAR g_szHttp1_0[] = {'H','T','T','P','/','1','.','0',0};
68 static const WCHAR g_szHttp1_1[] = {'H','T','T','P','/','1','.','1',0};
69 static const WCHAR szOK[] = {'O','K',0};
70 static const WCHAR hostW[] = { 'H','o','s','t',0 };
71 static const WCHAR szAuthorization[] = { 'A','u','t','h','o','r','i','z','a','t','i','o','n',0 };
72 static const WCHAR szProxy_Authorization[] = { 'P','r','o','x','y','-','A','u','t','h','o','r','i','z','a','t','i','o','n',0 };
73 static const WCHAR szStatus[] = { 'S','t','a','t','u','s',0 };
74 static const WCHAR szKeepAlive[] = {'K','e','e','p','-','A','l','i','v','e',0};
75 static const WCHAR szGET[] = { 'G','E','T', 0 };
76 static const WCHAR szHEAD[] = { 'H','E','A','D', 0 };
77 
78 static const WCHAR szAccept[] = { 'A','c','c','e','p','t',0 };
79 static const WCHAR szAccept_Charset[] = { 'A','c','c','e','p','t','-','C','h','a','r','s','e','t', 0 };
80 static const WCHAR szAccept_Encoding[] = { 'A','c','c','e','p','t','-','E','n','c','o','d','i','n','g',0 };
81 static const WCHAR szAccept_Language[] = { 'A','c','c','e','p','t','-','L','a','n','g','u','a','g','e',0 };
82 static const WCHAR szAccept_Ranges[] = { 'A','c','c','e','p','t','-','R','a','n','g','e','s',0 };
83 static const WCHAR szAge[] = { 'A','g','e',0 };
84 static const WCHAR szAllow[] = { 'A','l','l','o','w',0 };
85 static const WCHAR szCache_Control[] = { 'C','a','c','h','e','-','C','o','n','t','r','o','l',0 };
86 static const WCHAR szConnection[] = { 'C','o','n','n','e','c','t','i','o','n',0 };
87 static const WCHAR szContent_Base[] = { 'C','o','n','t','e','n','t','-','B','a','s','e',0 };
88 static const WCHAR szContent_Disposition[] = { 'C','o','n','t','e','n','t','-','D','i','s','p','o','s','i','t','i','o','n',0 };
89 static const WCHAR szContent_Encoding[] = { 'C','o','n','t','e','n','t','-','E','n','c','o','d','i','n','g',0 };
90 static const WCHAR szContent_ID[] = { 'C','o','n','t','e','n','t','-','I','D',0 };
91 static const WCHAR szContent_Language[] = { 'C','o','n','t','e','n','t','-','L','a','n','g','u','a','g','e',0 };
92 static const WCHAR szContent_Length[] = { 'C','o','n','t','e','n','t','-','L','e','n','g','t','h',0 };
93 static const WCHAR szContent_Location[] = { 'C','o','n','t','e','n','t','-','L','o','c','a','t','i','o','n',0 };
94 static const WCHAR szContent_MD5[] = { 'C','o','n','t','e','n','t','-','M','D','5',0 };
95 static const WCHAR szContent_Range[] = { 'C','o','n','t','e','n','t','-','R','a','n','g','e',0 };
96 static const WCHAR szContent_Transfer_Encoding[] = { 'C','o','n','t','e','n','t','-','T','r','a','n','s','f','e','r','-','E','n','c','o','d','i','n','g',0 };
97 static const WCHAR szContent_Type[] = { 'C','o','n','t','e','n','t','-','T','y','p','e',0 };
98 static const WCHAR szCookie[] = { 'C','o','o','k','i','e',0 };
99 static const WCHAR szDate[] = { 'D','a','t','e',0 };
100 static const WCHAR szFrom[] = { 'F','r','o','m',0 };
101 static const WCHAR szETag[] = { 'E','T','a','g',0 };
102 static const WCHAR szExpect[] = { 'E','x','p','e','c','t',0 };
103 static const WCHAR szExpires[] = { 'E','x','p','i','r','e','s',0 };
104 static const WCHAR szIf_Match[] = { 'I','f','-','M','a','t','c','h',0 };
105 static const WCHAR szIf_Modified_Since[] = { 'I','f','-','M','o','d','i','f','i','e','d','-','S','i','n','c','e',0 };
106 static const WCHAR szIf_None_Match[] = { 'I','f','-','N','o','n','e','-','M','a','t','c','h',0 };
107 static const WCHAR szIf_Range[] = { 'I','f','-','R','a','n','g','e',0 };
108 static const WCHAR szIf_Unmodified_Since[] = { 'I','f','-','U','n','m','o','d','i','f','i','e','d','-','S','i','n','c','e',0 };
109 static const WCHAR szLast_Modified[] = { 'L','a','s','t','-','M','o','d','i','f','i','e','d',0 };
110 static const WCHAR szLocation[] = { 'L','o','c','a','t','i','o','n',0 };
111 static const WCHAR szMax_Forwards[] = { 'M','a','x','-','F','o','r','w','a','r','d','s',0 };
112 static const WCHAR szMime_Version[] = { 'M','i','m','e','-','V','e','r','s','i','o','n',0 };
113 static const WCHAR szPragma[] = { 'P','r','a','g','m','a',0 };
114 static const WCHAR szProxy_Authenticate[] = { 'P','r','o','x','y','-','A','u','t','h','e','n','t','i','c','a','t','e',0 };
115 static const WCHAR szProxy_Connection[] = { 'P','r','o','x','y','-','C','o','n','n','e','c','t','i','o','n',0 };
116 static const WCHAR szPublic[] = { 'P','u','b','l','i','c',0 };
117 static const WCHAR szRange[] = { 'R','a','n','g','e',0 };
118 static const WCHAR szReferer[] = { 'R','e','f','e','r','e','r',0 };
119 static const WCHAR szRetry_After[] = { 'R','e','t','r','y','-','A','f','t','e','r',0 };
120 static const WCHAR szServer[] = { 'S','e','r','v','e','r',0 };
121 static const WCHAR szSet_Cookie[] = { 'S','e','t','-','C','o','o','k','i','e',0 };
122 static const WCHAR szTransfer_Encoding[] = { 'T','r','a','n','s','f','e','r','-','E','n','c','o','d','i','n','g',0 };
123 static const WCHAR szUnless_Modified_Since[] = { 'U','n','l','e','s','s','-','M','o','d','i','f','i','e','d','-','S','i','n','c','e',0 };
124 static const WCHAR szUpgrade[] = { 'U','p','g','r','a','d','e',0 };
125 static const WCHAR szURI[] = { 'U','R','I',0 };
126 static const WCHAR szUser_Agent[] = { 'U','s','e','r','-','A','g','e','n','t',0 };
127 static const WCHAR szVary[] = { 'V','a','r','y',0 };
128 static const WCHAR szVia[] = { 'V','i','a',0 };
129 static const WCHAR szWarning[] = { 'W','a','r','n','i','n','g',0 };
130 static const WCHAR szWWW_Authenticate[] = { 'W','W','W','-','A','u','t','h','e','n','t','i','c','a','t','e',0 };
131 
132 static const WCHAR emptyW[] = {0};
133 
134 #define HTTP_REFERER szReferer
135 #define HTTP_ACCEPT szAccept
136 #define HTTP_USERAGENT szUser_Agent
137 
138 #define HTTP_ADDHDR_FLAG_ADD 0x20000000
139 #define HTTP_ADDHDR_FLAG_ADD_IF_NEW 0x10000000
140 #define HTTP_ADDHDR_FLAG_COALESCE 0x40000000
141 #define HTTP_ADDHDR_FLAG_COALESCE_WITH_COMMA 0x40000000
142 #define HTTP_ADDHDR_FLAG_COALESCE_WITH_SEMICOLON 0x01000000
143 #define HTTP_ADDHDR_FLAG_REPLACE 0x80000000
144 #define HTTP_ADDHDR_FLAG_REQ 0x02000000
145 
146 #define COLLECT_TIME 60000
147 
148 #undef ARRAYSIZE
149 #define ARRAYSIZE(array) (sizeof(array)/sizeof((array)[0]))
150 
152 {
159  void *auth_data;
160  unsigned int auth_data_len;
161  BOOL finished; /* finished authenticating */
162 };
163 
164 
166 {
167  struct list entry;
168 
174 
175 typedef struct _authorizationData
176 {
177  struct list entry;
178 
188 
191 
194 {
195  0, 0, &authcache_cs,
197  0, 0, { (DWORD_PTR)(__FILE__ ": authcache_cs") }
198 };
199 static CRITICAL_SECTION authcache_cs = { &critsect_debug, -1, 0, 0, 0, 0 };
200 
207 static LPWSTR HTTP_build_req( LPCWSTR *list, int len );
209 static UINT HTTP_DecodeBase64(LPCWSTR base64, LPSTR bin);
211 
214 {
215  0, 0, &connection_pool_cs,
217  0, 0, { (DWORD_PTR)(__FILE__ ": connection_pool_cs") }
218 };
219 static CRITICAL_SECTION connection_pool_cs = { &connection_pool_debug, -1, 0, 0, 0, 0 };
220 
223 
225 {
227 }
228 
230 {
231  if(InterlockedDecrement(&server->ref))
232  return;
233 
234 #ifdef __REACTOS__
236 #endif
237  list_remove(&server->entry);
238 #ifdef __REACTOS__
240 #endif
241 
242  if(server->cert_chain)
243  CertFreeCertificateChain(server->cert_chain);
244  heap_free(server->name);
245  heap_free(server->scheme_host_port);
246  heap_free(server);
247 }
248 
250 {
251  BOOL default_port;
252  size_t name_len;
253  WCHAR *buf;
254 
255  static const WCHAR httpW[] = {'h','t','t','p',0};
256  static const WCHAR httpsW[] = {'h','t','t','p','s',0};
257  static const WCHAR formatW[] = {'%','s',':','/','/','%','s',':','%','u',0};
258 
259  name_len = strlenW(server->name);
260  buf = heap_alloc((name_len + 10 /* strlen("://:<port>") */)*sizeof(WCHAR) + sizeof(httpsW));
261  if(!buf)
262  return FALSE;
263 
264  sprintfW(buf, formatW, server->is_https ? httpsW : httpW, server->name, server->port);
265  server->scheme_host_port = buf;
266 
267  server->host_port = server->scheme_host_port + 7 /* strlen("http://") */;
268  if(server->is_https)
269  server->host_port++;
270 
271  default_port = server->port == (server->is_https ? INTERNET_DEFAULT_HTTPS_PORT : INTERNET_DEFAULT_HTTP_PORT);
272  server->canon_host_port = default_port ? server->name : server->host_port;
273  return TRUE;
274 }
275 
277 {
278  server_t *iter, *server = NULL;
279 
281 
283  if(iter->port == port && name.len == strlenW(iter->name) && !strncmpiW(iter->name, name.str, name.len)
284  && iter->is_https == is_https) {
285  server = iter;
287  break;
288  }
289  }
290 
291  if(!server && do_create) {
292  server = heap_alloc_zero(sizeof(*server));
293  if(server) {
294  server->ref = 2; /* list reference and return */
295  server->port = port;
296  server->is_https = is_https;
297  list_init(&server->conn_pool);
298  server->name = heap_strndupW(name.str, name.len);
299  if(server->name && process_host_port(server)) {
301  }else {
302  heap_free(server);
303  server = NULL;
304  }
305  }
306  }
307 
309 
310  return server;
311 }
312 
314 {
315  netconn_t *netconn, *netconn_safe;
316  server_t *server, *server_safe;
317  BOOL remaining = FALSE;
318  DWORD64 now;
319 
320 #ifdef __REACTOS__
321  now = GetTickCount();
322 #else
323  now = GetTickCount64();
324 #endif
325 
327  LIST_FOR_EACH_ENTRY_SAFE(netconn, netconn_safe, &server->conn_pool, netconn_t, pool_entry) {
328  if(collect_type > COLLECT_TIMEOUT || netconn->keep_until < now) {
329  TRACE("freeing %p\n", netconn);
330  list_remove(&netconn->pool_entry);
331  free_netconn(netconn);
332  }else {
333  remaining = TRUE;
334  }
335  }
336 
337  if(collect_type == COLLECT_CLEANUP) {
338  list_remove(&server->entry);
339  list_init(&server->entry);
341  }
342  }
343 
344  return remaining;
345 }
346 
348 {
349  BOOL remaining_conns;
350 
351  do {
352  /* FIXME: Use more sophisticated method */
353  Sleep(5000);
354 
356 
357  remaining_conns = collect_connections(COLLECT_TIMEOUT);
358  if(!remaining_conns)
360 
362  }while(remaining_conns);
363 
365 }
366 
367 /***********************************************************************
368  * HTTP_GetHeader (internal)
369  *
370  * Headers section must be held
371  */
373 {
374  int HeaderIndex = 0;
375  HeaderIndex = HTTP_GetCustomHeaderIndex(req, head, 0, TRUE);
376  if (HeaderIndex == -1)
377  return NULL;
378  else
379  return &req->custHeaders[HeaderIndex];
380 }
381 
383 {
385  WCHAR *ret = NULL;
386 
388  if ((header = HTTP_GetHeader( req, hostW ))) ret = heap_strdupW( header->lpszValue );
389  else ret = heap_strdupW( req->server->canon_host_port );
391  return ret;
392 }
393 
399 };
400 
401 typedef struct {
403 
408 
409  enum {
416  CHUNKED_STREAM_STATE_ERROR
417  } state;
419 
421 {
422  stream->vtbl->destroy(stream);
423 }
424 
426 {
429  req->read_pos = req->read_size = req->netconn_stream.content_read = 0;
430  req->read_gzip = FALSE;
431 }
432 
433 static void remove_header( http_request_t *request, const WCHAR *str, BOOL from_request )
434 {
435  int index;
436  EnterCriticalSection( &request->headers_section );
437  index = HTTP_GetCustomHeaderIndex( request, str, 0, from_request );
439  LeaveCriticalSection( &request->headers_section );
440 }
441 
442 #ifdef HAVE_ZLIB
443 
444 typedef struct {
446  data_stream_t *parent_stream;
447  z_stream zstream;
449  DWORD buf_size;
450  DWORD buf_pos;
451  BOOL end_of_data;
452 } gzip_stream_t;
453 
454 static BOOL gzip_end_of_data(data_stream_t *stream, http_request_t *req)
455 {
456  gzip_stream_t *gzip_stream = (gzip_stream_t*)stream;
457  return gzip_stream->end_of_data
458  || (!gzip_stream->buf_size && gzip_stream->parent_stream->vtbl->end_of_data(gzip_stream->parent_stream, req));
459 }
460 
461 static DWORD gzip_read(data_stream_t *stream, http_request_t *req, BYTE *buf, DWORD size,
462  DWORD *read, BOOL allow_blocking)
463 {
464  gzip_stream_t *gzip_stream = (gzip_stream_t*)stream;
465  z_stream *zstream = &gzip_stream->zstream;
466  DWORD current_read, ret_read = 0;
467  int zres;
469 
470  TRACE("(%d %x)\n", size, allow_blocking);
471 
472  while(size && !gzip_stream->end_of_data) {
473  if(!gzip_stream->buf_size) {
474  if(gzip_stream->buf_pos) {
475  if(gzip_stream->buf_size)
476  memmove(gzip_stream->buf, gzip_stream->buf+gzip_stream->buf_pos, gzip_stream->buf_size);
477  gzip_stream->buf_pos = 0;
478  }
479  res = gzip_stream->parent_stream->vtbl->read(gzip_stream->parent_stream, req, gzip_stream->buf+gzip_stream->buf_size,
480  sizeof(gzip_stream->buf)-gzip_stream->buf_size, &current_read, allow_blocking);
481  if(res != ERROR_SUCCESS)
482  break;
483 
484  gzip_stream->buf_size += current_read;
485  if(!current_read) {
486  WARN("unexpected end of data\n");
487  gzip_stream->end_of_data = TRUE;
488  break;
489  }
490  }
491 
492  zstream->next_in = gzip_stream->buf+gzip_stream->buf_pos;
493  zstream->avail_in = gzip_stream->buf_size;
494  zstream->next_out = buf+ret_read;
495  zstream->avail_out = size;
496  zres = inflate(&gzip_stream->zstream, 0);
497  current_read = size - zstream->avail_out;
498  size -= current_read;
499  ret_read += current_read;
500  gzip_stream->buf_size -= zstream->next_in - (gzip_stream->buf+gzip_stream->buf_pos);
501  gzip_stream->buf_pos = zstream->next_in-gzip_stream->buf;
502  if(zres == Z_STREAM_END) {
503  TRACE("end of data\n");
504  gzip_stream->end_of_data = TRUE;
505  inflateEnd(zstream);
506  }else if(zres != Z_OK) {
507  WARN("inflate failed %d: %s\n", zres, debugstr_a(zstream->msg));
508  if(!ret_read)
510  break;
511  }
512 
513  if(ret_read)
514  allow_blocking = FALSE;
515  }
516 
517  TRACE("read %u bytes\n", ret_read);
518  if(ret_read)
519  res = ERROR_SUCCESS;
520  *read = ret_read;
521  return res;
522 }
523 
524 static DWORD gzip_drain_content(data_stream_t *stream, http_request_t *req, BOOL allow_blocking)
525 {
526  gzip_stream_t *gzip_stream = (gzip_stream_t*)stream;
527  return gzip_stream->parent_stream->vtbl->drain_content(gzip_stream->parent_stream, req, allow_blocking);
528 }
529 
530 static void gzip_destroy(data_stream_t *stream)
531 {
532  gzip_stream_t *gzip_stream = (gzip_stream_t*)stream;
533 
534  destroy_data_stream(gzip_stream->parent_stream);
535 
536  if(!gzip_stream->end_of_data)
537  inflateEnd(&gzip_stream->zstream);
538  heap_free(gzip_stream);
539 }
540 
541 static const data_stream_vtbl_t gzip_stream_vtbl = {
542  gzip_end_of_data,
543  gzip_read,
544  gzip_drain_content,
545  gzip_destroy
546 };
547 
548 static voidpf wininet_zalloc(voidpf opaque, uInt items, uInt size)
549 {
550  return heap_alloc(items*size);
551 }
552 
553 static void wininet_zfree(voidpf opaque, voidpf address)
554 {
556 }
557 
558 static DWORD init_gzip_stream(http_request_t *req, BOOL is_gzip)
559 {
560  gzip_stream_t *gzip_stream;
561  int zres;
562 
563  gzip_stream = heap_alloc_zero(sizeof(gzip_stream_t));
564  if(!gzip_stream)
565  return ERROR_OUTOFMEMORY;
566 
567  gzip_stream->stream.vtbl = &gzip_stream_vtbl;
568  gzip_stream->zstream.zalloc = wininet_zalloc;
569  gzip_stream->zstream.zfree = wininet_zfree;
570 
571  zres = inflateInit2(&gzip_stream->zstream, is_gzip ? 0x1f : -15);
572  if(zres != Z_OK) {
573  ERR("inflateInit failed: %d\n", zres);
574  heap_free(gzip_stream);
575  return ERROR_OUTOFMEMORY;
576  }
577 
579 
580  if(req->read_size) {
581  memcpy(gzip_stream->buf, req->read_buf+req->read_pos, req->read_size);
582  gzip_stream->buf_size = req->read_size;
583  req->read_pos = req->read_size = 0;
584  }
585 
586  req->read_gzip = TRUE;
587  gzip_stream->parent_stream = req->data_stream;
588  req->data_stream = &gzip_stream->stream;
589  return ERROR_SUCCESS;
590 }
591 
592 #else
593 
595 {
596  ERR("gzip stream not supported, missing zlib.\n");
597  return ERROR_SUCCESS;
598 }
599 
600 #endif
601 
602 /***********************************************************************
603  * HTTP_FreeTokens (internal)
604  *
605  * Frees table of pointers.
606  */
607 static void HTTP_FreeTokens(LPWSTR * token_array)
608 {
609  int i;
610  for (i = 0; token_array[i]; i++) heap_free(token_array[i]);
611  heap_free(token_array);
612 }
613 
615 {
616  static const WCHAR szSlash[] = { '/',0 };
617  static const WCHAR szHttp[] = { 'h','t','t','p',':','/','/', 0 };
618 
619  /* If we don't have a path we set it to root */
620  if (NULL == request->path)
621  request->path = heap_strdupW(szSlash);
622  else /* remove \r and \n*/
623  {
624  int nLen = strlenW(request->path);
625  while ((nLen >0 ) && ((request->path[nLen-1] == '\r')||(request->path[nLen-1] == '\n')))
626  {
627  nLen--;
628  request->path[nLen]='\0';
629  }
630  /* Replace '\' with '/' */
631  while (nLen>0) {
632  nLen--;
633  if (request->path[nLen] == '\\') request->path[nLen]='/';
634  }
635  }
636 
638  request->path, strlenW(request->path), szHttp, strlenW(szHttp) )
639  && request->path[0] != '/') /* not an absolute path ?? --> fix it !! */
640  {
641  WCHAR *fixurl = heap_alloc((strlenW(request->path) + 2)*sizeof(WCHAR));
642  *fixurl = '/';
643  strcpyW(fixurl + 1, request->path);
644  heap_free( request->path );
645  request->path = fixurl;
646  }
647 }
648 
650  const WCHAR *path, const WCHAR *version, BOOL use_cr)
651 {
652  static const WCHAR szSpace[] = {' ',0};
653  static const WCHAR szColon[] = {':',' ',0};
654  static const WCHAR szCr[] = {'\r',0};
655  static const WCHAR szLf[] = {'\n',0};
656  LPWSTR requestString;
657  DWORD len, n;
658  LPCWSTR *req;
659  UINT i;
660 
661  EnterCriticalSection( &request->headers_section );
662 
663  /* allocate space for an array of all the string pointers to be added */
664  len = request->nCustHeaders * 5 + 10;
665  if (!(req = heap_alloc( len * sizeof(const WCHAR *) )))
666  {
667  LeaveCriticalSection( &request->headers_section );
668  return NULL;
669  }
670 
671  /* add the verb, path and HTTP version string */
672  n = 0;
673  req[n++] = verb;
674  req[n++] = szSpace;
675  req[n++] = path;
676  req[n++] = szSpace;
677  req[n++] = version;
678  if (use_cr)
679  req[n++] = szCr;
680  req[n++] = szLf;
681 
682  /* Append custom request headers */
683  for (i = 0; i < request->nCustHeaders; i++)
684  {
685  if (request->custHeaders[i].wFlags & HDR_ISREQUEST)
686  {
687  req[n++] = request->custHeaders[i].lpszField;
688  req[n++] = szColon;
689  req[n++] = request->custHeaders[i].lpszValue;
690  if (use_cr)
691  req[n++] = szCr;
692  req[n++] = szLf;
693 
694  TRACE("Adding custom header %s (%s)\n",
695  debugstr_w(request->custHeaders[i].lpszField),
696  debugstr_w(request->custHeaders[i].lpszValue));
697  }
698  }
699  if (use_cr)
700  req[n++] = szCr;
701  req[n++] = szLf;
702  req[n] = NULL;
703 
704  requestString = HTTP_build_req( req, 4 );
705  heap_free( req );
706  LeaveCriticalSection( &request->headers_section );
707  return requestString;
708 }
709 
711 {
712  static const WCHAR colonW[] = { ':',' ',0 };
713  static const WCHAR crW[] = { '\r',0 };
714  static const WCHAR lfW[] = { '\n',0 };
715  static const WCHAR status_fmt[] = { ' ','%','u',' ',0 };
716  const WCHAR **req;
717  WCHAR *ret, buf[14];
718  DWORD i, n = 0;
719 
720  EnterCriticalSection( &request->headers_section );
721 
722  if (!(req = heap_alloc( (request->nCustHeaders * 5 + 8) * sizeof(WCHAR *) )))
723  {
724  LeaveCriticalSection( &request->headers_section );
725  return NULL;
726  }
727 
728  if (request->status_code)
729  {
730  req[n++] = request->version;
731  sprintfW(buf, status_fmt, request->status_code);
732  req[n++] = buf;
733  req[n++] = request->statusText;
734  if (use_cr)
735  req[n++] = crW;
736  req[n++] = lfW;
737  }
738 
739  for(i = 0; i < request->nCustHeaders; i++)
740  {
741  if(!(request->custHeaders[i].wFlags & HDR_ISREQUEST)
742  && strcmpW(request->custHeaders[i].lpszField, szStatus))
743  {
744  req[n++] = request->custHeaders[i].lpszField;
745  req[n++] = colonW;
746  req[n++] = request->custHeaders[i].lpszValue;
747  if(use_cr)
748  req[n++] = crW;
749  req[n++] = lfW;
750 
751  TRACE("Adding custom header %s (%s)\n",
752  debugstr_w(request->custHeaders[i].lpszField),
753  debugstr_w(request->custHeaders[i].lpszValue));
754  }
755  }
756  if(use_cr)
757  req[n++] = crW;
758  req[n++] = lfW;
759  req[n] = NULL;
760 
761  ret = HTTP_build_req(req, 0);
762  heap_free(req);
763  LeaveCriticalSection( &request->headers_section );
764  return ret;
765 }
766 
768 {
769  int HeaderIndex;
770  int numCookies = 0;
771  LPHTTPHEADERW setCookieHeader;
772  WCHAR *path, *tmp;
773 
774  if(request->hdr.dwFlags & INTERNET_FLAG_NO_COOKIES)
775  return;
776 
778  if (!path)
779  return;
780 
781  tmp = strrchrW(path, '/');
782  if (tmp && tmp[1]) tmp[1] = 0;
783 
784  EnterCriticalSection( &request->headers_section );
785 
786  while((HeaderIndex = HTTP_GetCustomHeaderIndex(request, szSet_Cookie, numCookies++, FALSE)) != -1)
787  {
788  const WCHAR *data;
789  substr_t name;
790 
791  setCookieHeader = &request->custHeaders[HeaderIndex];
792 
793  if (!setCookieHeader->lpszValue)
794  continue;
795 
796  data = strchrW(setCookieHeader->lpszValue, '=');
797  if(!data)
798  continue;
799 
800  name = substr(setCookieHeader->lpszValue, data - setCookieHeader->lpszValue);
801  data++;
803  }
804 
805  LeaveCriticalSection( &request->headers_section );
806  heap_free(path);
807 }
808 
810 {
811  LPWSTR str = start;
812  LPWSTR end;
813 
814  while (*str == ' ')
815  str++;
816 
817  if (str != start)
818  memmove(start, str, sizeof(WCHAR) * (strlenW(str) + 1));
819 
820  end = start + strlenW(start) - 1;
821  while (end >= start && *end == ' ')
822  {
823  *end = '\0';
824  end--;
825  }
826 }
827 
828 static inline BOOL is_basic_auth_value( LPCWSTR pszAuthValue, LPWSTR *pszRealm )
829 {
830  static const WCHAR szBasic[] = {'B','a','s','i','c'}; /* Note: not nul-terminated */
831  static const WCHAR szRealm[] = {'r','e','a','l','m'}; /* Note: not nul-terminated */
832  BOOL is_basic;
833  is_basic = !strncmpiW(pszAuthValue, szBasic, ARRAYSIZE(szBasic)) &&
834  ((pszAuthValue[ARRAYSIZE(szBasic)] == ' ') || !pszAuthValue[ARRAYSIZE(szBasic)]);
835  if (is_basic && pszRealm)
836  {
837  LPCWSTR token;
838  LPCWSTR ptr = &pszAuthValue[ARRAYSIZE(szBasic)];
839  LPCWSTR realm;
840  ptr++;
841  *pszRealm=NULL;
842  token = strchrW(ptr,'=');
843  if (!token)
844  return TRUE;
845  realm = ptr;
846  while (*realm == ' ')
847  realm++;
848  if(!strncmpiW(realm, szRealm, ARRAYSIZE(szRealm)) &&
849  (realm[ARRAYSIZE(szRealm)] == ' ' || realm[ARRAYSIZE(szRealm)] == '='))
850  {
851  token++;
852  while (*token == ' ')
853  token++;
854  if (*token == '\0')
855  return TRUE;
856  *pszRealm = heap_strdupW(token);
857  strip_spaces(*pszRealm);
858  }
859  }
860 
861  return is_basic;
862 }
863 
864 static void destroy_authinfo( struct HttpAuthInfo *authinfo )
865 {
866  if (!authinfo) return;
867 
872 
873  heap_free(authinfo->auth_data);
876 }
877 
878 static UINT retrieve_cached_basic_authorization(const WCHAR *host, const WCHAR *realm, char **auth_data)
879 {
881  UINT rc = 0;
882 
883  TRACE("Looking for authorization for %s:%s\n",debugstr_w(host),debugstr_w(realm));
884 
887  {
888  if (!strcmpiW(host, ad->host) && (!realm || !strcmpW(realm, ad->realm)))
889  {
890  TRACE("Authorization found in cache\n");
891  *auth_data = heap_alloc(ad->authorizationLen);
892  memcpy(*auth_data,ad->authorization,ad->authorizationLen);
893  rc = ad->authorizationLen;
894  break;
895  }
896  }
898  return rc;
899 }
900 
901 static void cache_basic_authorization(LPWSTR host, LPWSTR realm, LPSTR auth_data, UINT auth_data_len)
902 {
903  struct list *cursor;
905 
906  TRACE("caching authorization for %s:%s = %s\n",debugstr_w(host),debugstr_w(realm),debugstr_an(auth_data,auth_data_len));
907 
910  {
912  if (!strcmpiW(host,check->host) && !strcmpW(realm,check->realm))
913  {
914  ad = check;
915  break;
916  }
917  }
918 
919  if (ad)
920  {
921  TRACE("Found match in cache, replacing\n");
923  ad->authorization = heap_alloc(auth_data_len);
924  memcpy(ad->authorization, auth_data, auth_data_len);
925  ad->authorizationLen = auth_data_len;
926  }
927  else
928  {
929  ad = heap_alloc(sizeof(basicAuthorizationData));
930  ad->host = heap_strdupW(host);
931  ad->realm = heap_strdupW(realm);
932  ad->authorization = heap_alloc(auth_data_len);
933  memcpy(ad->authorization, auth_data, auth_data_len);
934  ad->authorizationLen = auth_data_len;
936  TRACE("authorization cached\n");
937  }
939 }
940 
942  SEC_WINNT_AUTH_IDENTITY_W *nt_auth_identity)
943 {
944  authorizationData *ad;
945 
946  TRACE("Looking for authorization for %s:%s\n", debugstr_w(host), debugstr_w(scheme));
947 
950  if(!strcmpiW(host, ad->host) && !strcmpiW(scheme, ad->scheme)) {
951  TRACE("Authorization found in cache\n");
952 
953  nt_auth_identity->User = heap_strdupW(ad->user);
954  nt_auth_identity->Password = heap_strdupW(ad->password);
955  nt_auth_identity->Domain = heap_alloc(sizeof(WCHAR)*ad->domain_len);
956  if(!nt_auth_identity->User || !nt_auth_identity->Password ||
957  (!nt_auth_identity->Domain && ad->domain_len)) {
958  heap_free(nt_auth_identity->User);
959  heap_free(nt_auth_identity->Password);
960  heap_free(nt_auth_identity->Domain);
961  break;
962  }
963 
964  nt_auth_identity->Flags = SEC_WINNT_AUTH_IDENTITY_UNICODE;
965  nt_auth_identity->UserLength = ad->user_len;
966  nt_auth_identity->PasswordLength = ad->password_len;
967  memcpy(nt_auth_identity->Domain, ad->domain, sizeof(WCHAR)*ad->domain_len);
968  nt_auth_identity->DomainLength = ad->domain_len;
970  return TRUE;
971  }
972  }
974 
975  return FALSE;
976 }
977 
979  SEC_WINNT_AUTH_IDENTITY_W *nt_auth_identity)
980 {
981  authorizationData *ad;
982  BOOL found = FALSE;
983 
984  TRACE("Caching authorization for %s:%s\n", debugstr_w(host), debugstr_w(scheme));
985 
988  if(!strcmpiW(host, ad->host) && !strcmpiW(scheme, ad->scheme)) {
989  found = TRUE;
990  break;
991  }
992 
993  if(found) {
994  heap_free(ad->user);
995  heap_free(ad->password);
996  heap_free(ad->domain);
997  } else {
998  ad = heap_alloc(sizeof(authorizationData));
999  if(!ad) {
1001  return;
1002  }
1003 
1004  ad->host = heap_strdupW(host);
1005  ad->scheme = heap_strdupW(scheme);
1007  }
1008 
1009  ad->user = heap_strndupW(nt_auth_identity->User, nt_auth_identity->UserLength);
1010  ad->password = heap_strndupW(nt_auth_identity->Password, nt_auth_identity->PasswordLength);
1011  ad->domain = heap_strndupW(nt_auth_identity->Domain, nt_auth_identity->DomainLength);
1012  ad->user_len = nt_auth_identity->UserLength;
1013  ad->password_len = nt_auth_identity->PasswordLength;
1014  ad->domain_len = nt_auth_identity->DomainLength;
1015 
1016  if(!ad->host || !ad->scheme || !ad->user || !ad->password
1017  || (nt_auth_identity->Domain && !ad->domain)) {
1018  heap_free(ad->host);
1019  heap_free(ad->scheme);
1020  heap_free(ad->user);
1021  heap_free(ad->password);
1022  heap_free(ad->domain);
1023  list_remove(&ad->entry);
1024  heap_free(ad);
1025  }
1026 
1028 }
1029 
1031  struct HttpAuthInfo **ppAuthInfo,
1032  LPWSTR domain_and_username, LPWSTR password,
1033  LPWSTR host )
1034 {
1035  SECURITY_STATUS sec_status;
1036  struct HttpAuthInfo *pAuthInfo = *ppAuthInfo;
1037  BOOL first = FALSE;
1038  LPWSTR szRealm = NULL;
1039 
1040  TRACE("%s\n", debugstr_w(pszAuthValue));
1041 
1042  if (!pAuthInfo)
1043  {
1044  TimeStamp exp;
1045 
1046  first = TRUE;
1047  pAuthInfo = heap_alloc(sizeof(*pAuthInfo));
1048  if (!pAuthInfo)
1049  return FALSE;
1050 
1051  SecInvalidateHandle(&pAuthInfo->cred);
1052  SecInvalidateHandle(&pAuthInfo->ctx);
1053  memset(&pAuthInfo->exp, 0, sizeof(pAuthInfo->exp));
1054  pAuthInfo->attr = 0;
1055  pAuthInfo->auth_data = NULL;
1056  pAuthInfo->auth_data_len = 0;
1057  pAuthInfo->finished = FALSE;
1058 
1059  if (is_basic_auth_value(pszAuthValue,NULL))
1060  {
1061  static const WCHAR szBasic[] = {'B','a','s','i','c',0};
1062  pAuthInfo->scheme = heap_strdupW(szBasic);
1063  if (!pAuthInfo->scheme)
1064  {
1065  heap_free(pAuthInfo);
1066  return FALSE;
1067  }
1068  }
1069  else
1070  {
1071  PVOID pAuthData;
1072  SEC_WINNT_AUTH_IDENTITY_W nt_auth_identity;
1073 
1074  pAuthInfo->scheme = heap_strdupW(pszAuthValue);
1075  if (!pAuthInfo->scheme)
1076  {
1077  heap_free(pAuthInfo);
1078  return FALSE;
1079  }
1080 
1081  if (domain_and_username)
1082  {
1083  WCHAR *user = strchrW(domain_and_username, '\\');
1084  WCHAR *domain = domain_and_username;
1085 
1086  /* FIXME: make sure scheme accepts SEC_WINNT_AUTH_IDENTITY before calling AcquireCredentialsHandle */
1087 
1088  pAuthData = &nt_auth_identity;
1089 
1090  if (user) user++;
1091  else
1092  {
1093  user = domain_and_username;
1094  domain = NULL;
1095  }
1096 
1097  nt_auth_identity.Flags = SEC_WINNT_AUTH_IDENTITY_UNICODE;
1098  nt_auth_identity.User = user;
1099  nt_auth_identity.UserLength = strlenW(nt_auth_identity.User);
1100  nt_auth_identity.Domain = domain;
1101  nt_auth_identity.DomainLength = domain ? user - domain - 1 : 0;
1102  nt_auth_identity.Password = password;
1103  nt_auth_identity.PasswordLength = strlenW(nt_auth_identity.Password);
1104 
1105  cache_authorization(host, pAuthInfo->scheme, &nt_auth_identity);
1106  }
1107  else if(retrieve_cached_authorization(host, pAuthInfo->scheme, &nt_auth_identity))
1108  pAuthData = &nt_auth_identity;
1109  else
1110  /* use default credentials */
1111  pAuthData = NULL;
1112 
1113  sec_status = AcquireCredentialsHandleW(NULL, pAuthInfo->scheme,
1115  pAuthData, NULL,
1116  NULL, &pAuthInfo->cred,
1117  &exp);
1118 
1119  if(pAuthData && !domain_and_username) {
1120  heap_free(nt_auth_identity.User);
1121  heap_free(nt_auth_identity.Domain);
1122  heap_free(nt_auth_identity.Password);
1123  }
1124 
1125  if (sec_status == SEC_E_OK)
1126  {
1127  PSecPkgInfoW sec_pkg_info;
1128  sec_status = QuerySecurityPackageInfoW(pAuthInfo->scheme, &sec_pkg_info);
1129  if (sec_status == SEC_E_OK)
1130  {
1131  pAuthInfo->max_token = sec_pkg_info->cbMaxToken;
1132  FreeContextBuffer(sec_pkg_info);
1133  }
1134  }
1135  if (sec_status != SEC_E_OK)
1136  {
1137  WARN("AcquireCredentialsHandleW for scheme %s failed with error 0x%08x\n",
1138  debugstr_w(pAuthInfo->scheme), sec_status);
1139  heap_free(pAuthInfo->scheme);
1140  heap_free(pAuthInfo);
1141  return FALSE;
1142  }
1143  }
1144  *ppAuthInfo = pAuthInfo;
1145  }
1146  else if (pAuthInfo->finished)
1147  return FALSE;
1148 
1149  if ((strlenW(pszAuthValue) < strlenW(pAuthInfo->scheme)) ||
1150  strncmpiW(pszAuthValue, pAuthInfo->scheme, strlenW(pAuthInfo->scheme)))
1151  {
1152  ERR("authentication scheme changed from %s to %s\n",
1153  debugstr_w(pAuthInfo->scheme), debugstr_w(pszAuthValue));
1154  return FALSE;
1155  }
1156 
1157  if (is_basic_auth_value(pszAuthValue,&szRealm))
1158  {
1159  int userlen;
1160  int passlen;
1161  char *auth_data = NULL;
1162  UINT auth_data_len = 0;
1163 
1164  TRACE("basic authentication realm %s\n",debugstr_w(szRealm));
1165 
1166  if (!domain_and_username)
1167  {
1168  if (host && szRealm)
1170  if (auth_data_len == 0)
1171  {
1172  heap_free(szRealm);
1173  return FALSE;
1174  }
1175  }
1176  else
1177  {
1178  userlen = WideCharToMultiByte(CP_UTF8, 0, domain_and_username, lstrlenW(domain_and_username), NULL, 0, NULL, NULL);
1180 
1181  /* length includes a nul terminator, which will be re-used for the ':' */
1182  auth_data = heap_alloc(userlen + 1 + passlen);
1183  if (!auth_data)
1184  {
1185  heap_free(szRealm);
1186  return FALSE;
1187  }
1188 
1189  WideCharToMultiByte(CP_UTF8, 0, domain_and_username, -1, auth_data, userlen, NULL, NULL);
1190  auth_data[userlen] = ':';
1191  WideCharToMultiByte(CP_UTF8, 0, password, -1, &auth_data[userlen+1], passlen, NULL, NULL);
1192  auth_data_len = userlen + 1 + passlen;
1193  if (host && szRealm)
1195  }
1196 
1197  pAuthInfo->auth_data = auth_data;
1198  pAuthInfo->auth_data_len = auth_data_len;
1199  pAuthInfo->finished = TRUE;
1200  heap_free(szRealm);
1201  return TRUE;
1202  }
1203  else
1204  {
1205  LPCWSTR pszAuthData;
1206  SecBufferDesc out_desc, in_desc;
1207  SecBuffer out, in;
1208  unsigned char *buffer;
1211 
1212  in.BufferType = SECBUFFER_TOKEN;
1213  in.cbBuffer = 0;
1214  in.pvBuffer = NULL;
1215 
1216  in_desc.ulVersion = 0;
1217  in_desc.cBuffers = 1;
1218  in_desc.pBuffers = &in;
1219 
1220  pszAuthData = pszAuthValue + strlenW(pAuthInfo->scheme);
1221  if (*pszAuthData == ' ')
1222  {
1223  pszAuthData++;
1224  in.cbBuffer = HTTP_DecodeBase64(pszAuthData, NULL);
1225  in.pvBuffer = heap_alloc(in.cbBuffer);
1226  HTTP_DecodeBase64(pszAuthData, in.pvBuffer);
1227  }
1228 
1229  buffer = heap_alloc(pAuthInfo->max_token);
1230 
1231  out.BufferType = SECBUFFER_TOKEN;
1232  out.cbBuffer = pAuthInfo->max_token;
1233  out.pvBuffer = buffer;
1234 
1235  out_desc.ulVersion = 0;
1236  out_desc.cBuffers = 1;
1237  out_desc.pBuffers = &out;
1238 
1239  sec_status = InitializeSecurityContextW(first ? &pAuthInfo->cred : NULL,
1240  first ? NULL : &pAuthInfo->ctx,
1241  first ? request->server->name : NULL,
1242  context_req, 0, SECURITY_NETWORK_DREP,
1243  in.pvBuffer ? &in_desc : NULL,
1244  0, &pAuthInfo->ctx, &out_desc,
1245  &pAuthInfo->attr, &pAuthInfo->exp);
1246  if (sec_status == SEC_E_OK)
1247  {
1248  pAuthInfo->finished = TRUE;
1249  pAuthInfo->auth_data = out.pvBuffer;
1250  pAuthInfo->auth_data_len = out.cbBuffer;
1251  TRACE("sending last auth packet\n");
1252  }
1253  else if (sec_status == SEC_I_CONTINUE_NEEDED)
1254  {
1255  pAuthInfo->auth_data = out.pvBuffer;
1256  pAuthInfo->auth_data_len = out.cbBuffer;
1257  TRACE("sending next auth packet\n");
1258  }
1259  else
1260  {
1261  ERR("InitializeSecurityContextW returned error 0x%08x\n", sec_status);
1262  heap_free(out.pvBuffer);
1263  destroy_authinfo(pAuthInfo);
1264  *ppAuthInfo = NULL;
1265  return FALSE;
1266  }
1267  }
1268 
1269  return TRUE;
1270 }
1271 
1272 /***********************************************************************
1273  * HTTP_HttpAddRequestHeadersW (internal)
1274  */
1276  LPCWSTR lpszHeader, DWORD dwHeaderLength, DWORD dwModifier)
1277 {
1278  LPWSTR lpszStart;
1279  LPWSTR lpszEnd;
1280  LPWSTR buffer;
1282 
1283  TRACE("copying header: %s\n", debugstr_wn(lpszHeader, dwHeaderLength));
1284 
1285  if( dwHeaderLength == ~0U )
1286  len = strlenW(lpszHeader);
1287  else
1288  len = dwHeaderLength;
1289  buffer = heap_alloc(sizeof(WCHAR)*(len+1));
1290  lstrcpynW( buffer, lpszHeader, len + 1);
1291 
1292  lpszStart = buffer;
1293 
1294  do
1295  {
1296  LPWSTR * pFieldAndValue;
1297 
1298  lpszEnd = lpszStart;
1299 
1300  while (*lpszEnd != '\0')
1301  {
1302  if (*lpszEnd == '\r' || *lpszEnd == '\n')
1303  break;
1304  lpszEnd++;
1305  }
1306 
1307  if (*lpszStart == '\0')
1308  break;
1309 
1310  if (*lpszEnd == '\r' || *lpszEnd == '\n')
1311  {
1312  *lpszEnd = '\0';
1313  lpszEnd++; /* Jump over newline */
1314  }
1315  TRACE("interpreting header %s\n", debugstr_w(lpszStart));
1316  if (*lpszStart == '\0')
1317  {
1318  /* Skip 0-length headers */
1319  lpszStart = lpszEnd;
1320  res = ERROR_SUCCESS;
1321  continue;
1322  }
1323  pFieldAndValue = HTTP_InterpretHttpHeader(lpszStart);
1324  if (pFieldAndValue)
1325  {
1326  res = HTTP_ProcessHeader(request, pFieldAndValue[0],
1327  pFieldAndValue[1], dwModifier | HTTP_ADDHDR_FLAG_REQ);
1328  HTTP_FreeTokens(pFieldAndValue);
1329  }
1330 
1331  lpszStart = lpszEnd;
1332  } while (res == ERROR_SUCCESS);
1333 
1334  heap_free(buffer);
1335  return res;
1336 }
1337 
1338 /***********************************************************************
1339  * HttpAddRequestHeadersW (WININET.@)
1340  *
1341  * Adds one or more HTTP header to the request handler
1342  *
1343  * NOTE
1344  * On Windows if dwHeaderLength includes the trailing '\0', then
1345  * HttpAddRequestHeadersW() adds it too. However this results in an
1346  * invalid HTTP header which is rejected by some servers so we probably
1347  * don't need to match Windows on that point.
1348  *
1349  * RETURNS
1350  * TRUE on success
1351  * FALSE on failure
1352  *
1353  */
1355  LPCWSTR lpszHeader, DWORD dwHeaderLength, DWORD dwModifier)
1356 {
1359 
1360  TRACE("%p, %s, %i, %i\n", hHttpRequest, debugstr_wn(lpszHeader, dwHeaderLength), dwHeaderLength, dwModifier);
1361 
1362  if (!lpszHeader)
1363  return TRUE;
1364 
1365  request = (http_request_t*) get_handle_object( hHttpRequest );
1366  if (request && request->hdr.htype == WH_HHTTPREQ)
1367  res = HTTP_HttpAddRequestHeadersW( request, lpszHeader, dwHeaderLength, dwModifier );
1368  if( request )
1369  WININET_Release( &request->hdr );
1370 
1371  if(res != ERROR_SUCCESS)
1372  SetLastError(res);
1373  return res == ERROR_SUCCESS;
1374 }
1375 
1376 /***********************************************************************
1377  * HttpAddRequestHeadersA (WININET.@)
1378  *
1379  * Adds one or more HTTP header to the request handler
1380  *
1381  * RETURNS
1382  * TRUE on success
1383  * FALSE on failure
1384  *
1385  */
1387  LPCSTR lpszHeader, DWORD dwHeaderLength, DWORD dwModifier)
1388 {
1389  WCHAR *headers = NULL;
1390  BOOL r;
1391 
1392  TRACE("%p, %s, %i, %i\n", hHttpRequest, debugstr_an(lpszHeader, dwHeaderLength), dwHeaderLength, dwModifier);
1393 
1394  if(lpszHeader)
1395  headers = heap_strndupAtoW(lpszHeader, dwHeaderLength, &dwHeaderLength);
1396 
1397  r = HttpAddRequestHeadersW(hHttpRequest, headers, dwHeaderLength, dwModifier);
1398 
1399  heap_free(headers);
1400  return r;
1401 }
1402 
1403 static void free_accept_types( WCHAR **accept_types )
1404 {
1405  WCHAR *ptr, **types = accept_types;
1406 
1407  if (!types) return;
1408  while ((ptr = *types))
1409  {
1410  heap_free( ptr );
1411  types++;
1412  }
1413  heap_free( accept_types );
1414 }
1415 
1416 static WCHAR **convert_accept_types( const char **accept_types )
1417 {
1418  unsigned int count;
1419  const char **types = accept_types;
1420  WCHAR **typesW;
1421  BOOL invalid_pointer = FALSE;
1422 
1423  if (!types) return NULL;
1424  count = 0;
1425  while (*types)
1426  {
1427  __TRY
1428  {
1429  /* find out how many there are */
1430  if (*types && **types)
1431  {
1432  TRACE("accept type: %s\n", debugstr_a(*types));
1433  count++;
1434  }
1435  }
1437  {
1438  WARN("invalid accept type pointer\n");
1439  invalid_pointer = TRUE;
1440  }
1441  __ENDTRY;
1442  types++;
1443  }
1444  if (invalid_pointer) return NULL;
1445  if (!(typesW = heap_alloc( sizeof(WCHAR *) * (count + 1) ))) return NULL;
1446  count = 0;
1447  types = accept_types;
1448  while (*types)
1449  {
1450  if (*types && **types) typesW[count++] = heap_strdupAtoW( *types );
1451  types++;
1452  }
1453  typesW[count] = NULL;
1454  return typesW;
1455 }
1456 
1457 /***********************************************************************
1458  * HttpOpenRequestA (WININET.@)
1459  *
1460  * Open a HTTP request handle
1461  *
1462  * RETURNS
1463  * HINTERNET a HTTP request handle on success
1464  * NULL on failure
1465  *
1466  */
1468  LPCSTR lpszVerb, LPCSTR lpszObjectName, LPCSTR lpszVersion,
1469  LPCSTR lpszReferrer , LPCSTR *lpszAcceptTypes,
1470  DWORD dwFlags, DWORD_PTR dwContext)
1471 {
1472  LPWSTR szVerb = NULL, szObjectName = NULL;
1473  LPWSTR szVersion = NULL, szReferrer = NULL, *szAcceptTypes = NULL;
1474  HINTERNET rc = NULL;
1475 
1476  TRACE("(%p, %s, %s, %s, %s, %p, %08x, %08lx)\n", hHttpSession,
1477  debugstr_a(lpszVerb), debugstr_a(lpszObjectName),
1478  debugstr_a(lpszVersion), debugstr_a(lpszReferrer), lpszAcceptTypes,
1479  dwFlags, dwContext);
1480 
1481  if (lpszVerb)
1482  {
1483  szVerb = heap_strdupAtoW(lpszVerb);
1484  if ( !szVerb )
1485  goto end;
1486  }
1487 
1488  if (lpszObjectName)
1489  {
1490  szObjectName = heap_strdupAtoW(lpszObjectName);
1491  if ( !szObjectName )
1492  goto end;
1493  }
1494 
1495  if (lpszVersion)
1496  {
1497  szVersion = heap_strdupAtoW(lpszVersion);
1498  if ( !szVersion )
1499  goto end;
1500  }
1501 
1502  if (lpszReferrer)
1503  {
1504  szReferrer = heap_strdupAtoW(lpszReferrer);
1505  if ( !szReferrer )
1506  goto end;
1507  }
1508 
1509  szAcceptTypes = convert_accept_types( lpszAcceptTypes );
1510  rc = HttpOpenRequestW(hHttpSession, szVerb, szObjectName, szVersion, szReferrer,
1511  (const WCHAR **)szAcceptTypes, dwFlags, dwContext);
1512 
1513 end:
1514  free_accept_types(szAcceptTypes);
1515  heap_free(szReferrer);
1517  heap_free(szObjectName);
1518  heap_free(szVerb);
1519  return rc;
1520 }
1521 
1522 /***********************************************************************
1523  * HTTP_EncodeBase64
1524  */
1525 static UINT HTTP_EncodeBase64( LPCSTR bin, unsigned int len, LPWSTR base64 )
1526 {
1527  UINT n = 0, x;
1528  static const CHAR HTTP_Base64Enc[] =
1529  "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/";
1530 
1531  while( len > 0 )
1532  {
1533  /* first 6 bits, all from bin[0] */
1534  base64[n++] = HTTP_Base64Enc[(bin[0] & 0xfc) >> 2];
1535  x = (bin[0] & 3) << 4;
1536 
1537  /* next 6 bits, 2 from bin[0] and 4 from bin[1] */
1538  if( len == 1 )
1539  {
1540  base64[n++] = HTTP_Base64Enc[x];
1541  base64[n++] = '=';
1542  base64[n++] = '=';
1543  break;
1544  }
1545  base64[n++] = HTTP_Base64Enc[ x | ( (bin[1]&0xf0) >> 4 ) ];
1546  x = ( bin[1] & 0x0f ) << 2;
1547 
1548  /* next 6 bits 4 from bin[1] and 2 from bin[2] */
1549  if( len == 2 )
1550  {
1551  base64[n++] = HTTP_Base64Enc[x];
1552  base64[n++] = '=';
1553  break;
1554  }
1555  base64[n++] = HTTP_Base64Enc[ x | ( (bin[2]&0xc0 ) >> 6 ) ];
1556 
1557  /* last 6 bits, all from bin [2] */
1558  base64[n++] = HTTP_Base64Enc[ bin[2] & 0x3f ];
1559  bin += 3;
1560  len -= 3;
1561  }
1562  base64[n] = 0;
1563  return n;
1564 }
1565 
1566 static const signed char HTTP_Base64Dec[] =
1567 {
1568  -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, /* 0x00 */
1569  -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, /* 0x10 */
1570  -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, 62, -1, -1, -1, 63, /* 0x20 */
1571  52, 53, 54, 55, 56, 57, 58, 59, 60, 61, -1, -1, -1, -1, -1, -1, /* 0x30 */
1572  -1, 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, /* 0x40 */
1573  15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, -1, -1, -1, -1, -1, /* 0x50 */
1574  -1, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, /* 0x60 */
1575  41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51, -1, -1, -1, -1, -1 /* 0x70 */
1576 };
1577 
1578 /***********************************************************************
1579  * HTTP_DecodeBase64
1580  */
1582 {
1583  unsigned int n = 0;
1584 
1585  while(*base64)
1586  {
1587  signed char in[4];
1588 
1589  if (base64[0] >= ARRAYSIZE(HTTP_Base64Dec) ||
1590  ((in[0] = HTTP_Base64Dec[base64[0]]) == -1) ||
1591  base64[1] >= ARRAYSIZE(HTTP_Base64Dec) ||
1592  ((in[1] = HTTP_Base64Dec[base64[1]]) == -1))
1593  {
1594  WARN("invalid base64: %s\n", debugstr_w(base64));
1595  return 0;
1596  }
1597  if (bin)
1598  bin[n] = (unsigned char) (in[0] << 2 | in[1] >> 4);
1599  n++;
1600 
1601  if ((base64[2] == '=') && (base64[3] == '='))
1602  break;
1603  if (base64[2] > ARRAYSIZE(HTTP_Base64Dec) ||
1604  ((in[2] = HTTP_Base64Dec[base64[2]]) == -1))
1605  {
1606  WARN("invalid base64: %s\n", debugstr_w(&base64[2]));
1607  return 0;
1608  }
1609  if (bin)
1610  bin[n] = (unsigned char) (in[1] << 4 | in[2] >> 2);
1611  n++;
1612 
1613  if (base64[3] == '=')
1614  break;
1615  if (base64[3] > ARRAYSIZE(HTTP_Base64Dec) ||
1616  ((in[3] = HTTP_Base64Dec[base64[3]]) == -1))
1617  {
1618  WARN("invalid base64: %s\n", debugstr_w(&base64[3]));
1619  return 0;
1620  }
1621  if (bin)
1622  bin[n] = (unsigned char) (((in[2] << 6) & 0xc0) | in[3]);
1623  n++;
1624 
1625  base64 += 4;
1626  }
1627 
1628  return n;
1629 }
1630 
1631 static WCHAR *encode_auth_data( const WCHAR *scheme, const char *data, UINT data_len )
1632 {
1633  WCHAR *ret;
1634  UINT len, scheme_len = strlenW( scheme );
1635 
1636  /* scheme + space + base64 encoded data (3/2/1 bytes data -> 4 bytes of characters) */
1637  len = scheme_len + 1 + ((data_len + 2) * 4) / 3;
1638  if (!(ret = heap_alloc( (len + 1) * sizeof(WCHAR) ))) return NULL;
1639  memcpy( ret, scheme, scheme_len * sizeof(WCHAR) );
1640  ret[scheme_len] = ' ';
1641  HTTP_EncodeBase64( data, data_len, ret + scheme_len + 1 );
1642  return ret;
1643 }
1644 
1645 
1646 /***********************************************************************
1647  * HTTP_InsertAuthorization
1648  *
1649  * Insert or delete the authorization field in the request header.
1650  */
1652 {
1653  static const WCHAR wszBasic[] = {'B','a','s','i','c',0};
1654  WCHAR *host, *authorization = NULL;
1655 
1656  if (pAuthInfo)
1657  {
1658  if (pAuthInfo->auth_data_len)
1659  {
1660  if (!(authorization = encode_auth_data(pAuthInfo->scheme, pAuthInfo->auth_data, pAuthInfo->auth_data_len)))
1661  return FALSE;
1662 
1663  /* clear the data as it isn't valid now that it has been sent to the
1664  * server, unless it's Basic authentication which doesn't do
1665  * connection tracking */
1666  if (strcmpiW(pAuthInfo->scheme, wszBasic))
1667  {
1668  heap_free(pAuthInfo->auth_data);
1669  pAuthInfo->auth_data = NULL;
1670  pAuthInfo->auth_data_len = 0;
1671  }
1672  }
1673 
1674  TRACE("Inserting authorization: %s\n", debugstr_w(authorization));
1675 
1676  HTTP_ProcessHeader(request, header, authorization,
1678  heap_free(authorization);
1679  }
1681  {
1682  UINT data_len;
1683  char *data;
1684 
1685  if ((data_len = retrieve_cached_basic_authorization(host, NULL, &data)))
1686  {
1687  TRACE("Found cached basic authorization for %s\n", debugstr_w(host));
1688 
1689  if (!(authorization = encode_auth_data(wszBasic, data, data_len)))
1690  {
1691  heap_free(data);
1692  heap_free(host);
1693  return FALSE;
1694  }
1695 
1696  TRACE("Inserting authorization: %s\n", debugstr_w(authorization));
1697 
1698  HTTP_ProcessHeader(request, header, authorization,
1700  heap_free(data);
1701  heap_free(authorization);
1702  }
1703  heap_free(host);
1704  }
1705  return TRUE;
1706 }
1707 
1709 {
1710  DWORD size, len;
1711  WCHAR *url;
1712 
1714  size = len + strlenW(req->path) + 1;
1715  if(*req->path != '/')
1716  size++;
1717  url = heap_alloc(size * sizeof(WCHAR));
1718  if(!url)
1719  return NULL;
1720 
1721  memcpy(url, req->server->scheme_host_port, len*sizeof(WCHAR));
1722  if(*req->path != '/')
1723  url[len++] = '/';
1724 
1725  strcpyW(url+len, req->path);
1726 
1727  TRACE("url=%s\n", debugstr_w(url));
1728  return url;
1729 }
1730 
1732 {
1733  static const WCHAR localW[] = { '<','l','o','c','a','l','>',0 };
1734  const WCHAR *dot, *ptr;
1735  int len;
1736 
1737  if(domain.len == sizeof(localW)/sizeof(WCHAR)-1 && !strncmpiW(domain.str, localW, domain.len) && !strchrW(server, '.' ))
1738  return TRUE;
1739 
1740  if(domain.len && *domain.str != '*')
1741  return domain.len == strlenW(server) && !strncmpiW(server, domain.str, domain.len);
1742 
1743  if(domain.len < 2 || domain.str[1] != '.')
1744  return FALSE;
1745 
1746  /* For a hostname to match a wildcard, the last domain must match
1747  * the wildcard exactly. E.g. if the wildcard is *.a.b, and the
1748  * hostname is www.foo.a.b, it matches, but a.b does not.
1749  */
1750  dot = strchrW(server, '.');
1751  if(!dot)
1752  return FALSE;
1753 
1754  len = strlenW(dot + 1);
1755  if(len < domain.len - 2)
1756  return FALSE;
1757 
1758  /* The server's domain is longer than the wildcard, so it
1759  * could be a subdomain. Compare the last portion of the
1760  * server's domain.
1761  */
1762  ptr = dot + 1 + len - domain.len + 2;
1763  if(!strncmpiW(ptr, domain.str+2, domain.len-2))
1764  /* This is only a match if the preceding character is
1765  * a '.', i.e. that it is a matching domain. E.g.
1766  * if domain is '*.b.c' and server is 'www.ab.c' they
1767  * do not match.
1768  */
1769  return *(ptr - 1) == '.';
1770 
1771  return len == domain.len-2 && !strncmpiW(dot + 1, domain.str + 2, len);
1772 }
1773 
1775 {
1776  LPCWSTR ptr;
1777  BOOL ret = FALSE;
1778 
1779  if (!lpwai->proxyBypass) return FALSE;
1780  ptr = lpwai->proxyBypass;
1781  while(1) {
1782  LPCWSTR tmp = ptr;
1783 
1784  ptr = strchrW( ptr, ';' );
1785  if (!ptr)
1786  ptr = strchrW( tmp, ' ' );
1787  if (!ptr)
1788  ptr = tmp + strlenW(tmp);
1789  ret = HTTP_DomainMatches( server, substr(tmp, ptr-tmp) );
1790  if (ret || !*ptr)
1791  break;
1792  ptr++;
1793  }
1794  return ret;
1795 }
1796 
1797 /***********************************************************************
1798  * HTTP_DealWithProxy
1799  */
1801 {
1802  static const WCHAR protoHttp[] = { 'h','t','t','p',0 };
1803  static const WCHAR szHttp[] = { 'h','t','t','p',':','/','/',0 };
1804  static WCHAR szNul[] = { 0 };
1805  URL_COMPONENTSW UrlComponents = { sizeof(UrlComponents) };
1806  server_t *new_server = NULL;
1807  WCHAR *proxy;
1808 
1809  proxy = INTERNET_FindProxyForProtocol(hIC->proxy, protoHttp);
1810  if(!proxy)
1811  return FALSE;
1813  proxy, strlenW(szHttp), szHttp, strlenW(szHttp))) {
1814  WCHAR *proxy_url = heap_alloc(strlenW(proxy)*sizeof(WCHAR) + sizeof(szHttp));
1815  if(!proxy_url) {
1816  heap_free(proxy);
1817  return FALSE;
1818  }
1819  strcpyW(proxy_url, szHttp);
1820  strcatW(proxy_url, proxy);
1821  heap_free(proxy);
1822  proxy = proxy_url;
1823  }
1824 
1825  UrlComponents.dwHostNameLength = 1;
1826  if(InternetCrackUrlW(proxy, 0, 0, &UrlComponents) && UrlComponents.dwHostNameLength) {
1827  if( !request->path )
1828  request->path = szNul;
1829 
1830  new_server = get_server(substr(UrlComponents.lpszHostName, UrlComponents.dwHostNameLength),
1831  UrlComponents.nPort, UrlComponents.nScheme == INTERNET_SCHEME_HTTPS, TRUE);
1832  }
1833  heap_free(proxy);
1834  if(!new_server)
1835  return FALSE;
1836 
1837  request->proxy = new_server;
1838 
1839  TRACE("proxy server=%s port=%d\n", debugstr_w(new_server->name), new_server->port);
1840  return TRUE;
1841 }
1842 
1844 {
1845  server_t *server = request->proxy ? request->proxy : request->server;
1846  int addr_len;
1847 
1848  if(server->addr_len)
1849  return ERROR_SUCCESS;
1850 
1851  INTERNET_SendCallback(&request->hdr, request->hdr.dwContext,
1853  server->name,
1854  (strlenW(server->name)+1) * sizeof(WCHAR));
1855 
1856  addr_len = sizeof(server->addr);
1857  if (!GetAddress(server->name, server->port, (SOCKADDR*)&server->addr, &addr_len, server->addr_str))
1859 
1860  server->addr_len = addr_len;
1861  INTERNET_SendCallback(&request->hdr, request->hdr.dwContext,
1863  server->addr_str, strlen(server->addr_str)+1);
1864 
1865  TRACE("resolved %s to %s\n", debugstr_w(server->name), server->addr_str);
1866  return ERROR_SUCCESS;
1867 }
1868 
1870 {
1871  static const WCHAR http[] = { 'h','t','t','p',':','/','/',0 };
1872  static const WCHAR https[] = { 'h','t','t','p','s',':','/','/',0 };
1873  const WCHAR *host, *scheme;
1874  WCHAR *buf, *ptr;
1875  size_t len;
1876 
1877  host = req->server->canon_host_port;
1878 
1879  if (req->server->is_https)
1880  scheme = https;
1881  else
1882  scheme = http;
1883 
1884  len = strlenW(scheme) + strlenW(host) + (req->path[0] != '/' ? 1 : 0) + strlenW(req->path);
1885  ptr = buf = heap_alloc((len+1) * sizeof(WCHAR));
1886  if(buf) {
1887  strcpyW(ptr, scheme);
1888  ptr += strlenW(ptr);
1889 
1890  strcpyW(ptr, host);
1891  ptr += strlenW(ptr);
1892 
1893  if(req->path[0] != '/')
1894  *ptr++ = '/';
1895 
1896  strcpyW(ptr, req->path);
1897  ptr += strlenW(ptr);
1898  *ptr = 0;
1899  }
1900 
1901  return buf;
1902 }
1903 
1904 
1905 /***********************************************************************
1906  * HTTPREQ_Destroy (internal)
1907  *
1908  * Deallocate request handle
1909  *
1910  */
1912 {
1914  DWORD i;
1915 
1916  TRACE("\n");
1917 
1918  if(request->hCacheFile)
1919  CloseHandle(request->hCacheFile);
1920  if(request->req_file)
1921  req_file_release(request->req_file);
1922 
1923  request->headers_section.DebugInfo->Spare[0] = 0;
1924  DeleteCriticalSection( &request->headers_section );
1925  request->read_section.DebugInfo->Spare[0] = 0;
1926  DeleteCriticalSection( &request->read_section );
1927  WININET_Release(&request->session->hdr);
1928 
1929  destroy_authinfo(request->authInfo);
1930  destroy_authinfo(request->proxyAuthInfo);
1931 
1932  if(request->server)
1933  server_release(request->server);
1934  if(request->proxy)
1935  server_release(request->proxy);
1936 
1938  heap_free(request->verb);
1939  heap_free(request->version);
1940  heap_free(request->statusText);
1941 
1942  for (i = 0; i < request->nCustHeaders; i++)
1943  {
1944  heap_free(request->custHeaders[i].lpszField);
1945  heap_free(request->custHeaders[i].lpszValue);
1946  }
1947  destroy_data_stream(request->data_stream);
1948  heap_free(request->custHeaders);
1949 }
1950 
1951 static void http_release_netconn(http_request_t *req, BOOL reuse)
1952 {
1953  TRACE("%p %p %x\n",req, req->netconn, reuse);
1954 
1955  if(!is_valid_netconn(req->netconn))
1956  return;
1957 
1958 #ifndef __REACTOS__
1959  if(reuse && req->netconn->keep_alive) {
1960  BOOL run_collector;
1961 
1963 
1966  req->netconn = NULL;
1967 
1968  run_collector = !collector_running;
1970 
1972 
1973  if(run_collector) {
1974  HANDLE thread = NULL;
1975  HMODULE module;
1976 
1977  GetModuleHandleExW(GET_MODULE_HANDLE_EX_FLAG_FROM_ADDRESS, (const WCHAR*)WININET_hModule, &module);
1978  if(module)
1980  if(!thread) {
1984 
1985  if(module)
1987  }
1988  else
1990  }
1991  return;
1992  }
1993 #else
1994  /* Silence unused function warning */
1996 #endif
1997 
1998  INTERNET_SendCallback(&req->hdr, req->hdr.dwContext,
2000 
2001  close_netconn(req->netconn);
2002 
2003  INTERNET_SendCallback(&req->hdr, req->hdr.dwContext,
2005 }
2006 
2008 {
2009  WCHAR szVersion[10];
2010  WCHAR szConnectionResponse[20];
2011  DWORD dwBufferSize = sizeof(szVersion);
2012  BOOL keepalive = FALSE;
2013 
2014  /* as per RFC 2068, S8.1.2.1, if the client is HTTP/1.1 then assume that
2015  * the connection is keep-alive by default */
2018  {
2019  keepalive = TRUE;
2020  }
2021 
2022  dwBufferSize = sizeof(szConnectionResponse);
2023  if (HTTP_HttpQueryInfoW(request, HTTP_QUERY_PROXY_CONNECTION, szConnectionResponse, &dwBufferSize, NULL) == ERROR_SUCCESS
2024  || HTTP_HttpQueryInfoW(request, HTTP_QUERY_CONNECTION, szConnectionResponse, &dwBufferSize, NULL) == ERROR_SUCCESS)
2025  {
2026  keepalive = !strcmpiW(szConnectionResponse, szKeepAlive);
2027  }
2028 
2029  return keepalive;
2030 }
2031 
2033 {
2035 
2037 }
2038 
2039 static DWORD str_to_buffer(const WCHAR *str, void *buffer, DWORD *size, BOOL unicode)
2040 {
2041  int len;
2042  if (unicode)
2043  {
2044  WCHAR *buf = buffer;
2045 
2046  if (str) len = strlenW(str);
2047  else len = 0;
2048  if (*size < (len + 1) * sizeof(WCHAR))
2049  {
2050  *size = (len + 1) * sizeof(WCHAR);
2052  }
2053  if (str) strcpyW(buf, str);
2054  else buf[0] = 0;
2055 
2056  *size = len;
2057  return ERROR_SUCCESS;
2058  }
2059  else
2060  {
2061  char *buf = buffer;
2062 
2063  if (str) len = WideCharToMultiByte(CP_ACP, 0, str, -1, NULL, 0, NULL, NULL);
2064  else len = 1;
2065  if (*size < len)
2066  {
2067  *size = len;
2069  }
2070  if (str) WideCharToMultiByte(CP_ACP, 0, str, -1, buf, *size, NULL, NULL);
2071  else buf[0] = 0;
2072 
2073  *size = len - 1;
2074  return ERROR_SUCCESS;
2075  }
2076 }
2077 
2079 {
2081 
2082  switch(option) {
2084  {
2086 
2087  FIXME("INTERNET_DIAGNOSTIC_SOCKET_INFO stub\n");
2088 
2089  if (*size < sizeof(INTERNET_DIAGNOSTIC_SOCKET_INFO))
2092  /* FIXME: can't get a SOCKET from our connection since we don't use
2093  * winsock
2094  */
2095  info->Socket = 0;
2096  /* FIXME: get source port from req->netConnection */
2097  info->SourcePort = 0;
2098  info->DestPort = req->server->port;
2099  info->Flags = 0;
2100  if (HTTP_KeepAlive(req))
2101  info->Flags |= IDSI_FLAG_KEEP_ALIVE;
2102  if (req->proxy)
2103  info->Flags |= IDSI_FLAG_PROXY;
2104  if (is_valid_netconn(req->netconn) && req->netconn->secure)
2105  info->Flags |= IDSI_FLAG_SECURE;
2106 
2107  return ERROR_SUCCESS;
2108  }
2109 
2110  case 98:
2111  TRACE("Queried undocumented option 98, forwarding to INTERNET_OPTION_SECURITY_FLAGS\n");
2112  /* fall through */
2114  {
2115  DWORD flags;
2116 
2117  if (*size < sizeof(ULONG))
2119 
2120  *size = sizeof(DWORD);
2122  *(DWORD *)buffer = flags;
2123 
2124  TRACE("INTERNET_OPTION_SECURITY_FLAGS %x\n", flags);
2125  return ERROR_SUCCESS;
2126  }
2127 
2129  TRACE("INTERNET_OPTION_HANDLE_TYPE\n");
2130 
2131  if (*size < sizeof(ULONG))
2133 
2134  *size = sizeof(DWORD);
2136  return ERROR_SUCCESS;
2137 
2138  case INTERNET_OPTION_URL: {
2139  WCHAR *url;
2140  DWORD res;
2141 
2142  TRACE("INTERNET_OPTION_URL\n");
2143 
2144  url = compose_request_url(req);
2145  if(!url)
2146  return ERROR_OUTOFMEMORY;
2147 
2148  res = str_to_buffer(url, buffer, size, unicode);
2149  heap_free(url);
2150  return res;
2151  }
2153  return str_to_buffer(req->session->appInfo->agent, buffer, size, unicode);
2155  return str_to_buffer(req->session->userName, buffer, size, unicode);
2157  return str_to_buffer(req->session->password, buffer, size, unicode);
2159  return str_to_buffer(req->session->appInfo->proxyUsername, buffer, size, unicode);
2161  return str_to_buffer(req->session->appInfo->proxyPassword, buffer, size, unicode);
2162 
2166  DWORD nbytes, error;
2167  BOOL ret;
2168 
2169  TRACE("INTERNET_OPTION_CACHE_TIMESTAMPS\n");
2170 
2171  if(!req->req_file)
2172  return ERROR_FILE_NOT_FOUND;
2173 
2174  if (*size < sizeof(*ts))
2175  {
2176  *size = sizeof(*ts);
2178  }
2179 
2180  nbytes = 0;
2181  ret = GetUrlCacheEntryInfoW(req->req_file->url, NULL, &nbytes);
2182  error = GetLastError();
2183  if (!ret && error == ERROR_INSUFFICIENT_BUFFER)
2184  {
2185  if (!(info = heap_alloc(nbytes)))
2186  return ERROR_OUTOFMEMORY;
2187 
2188  GetUrlCacheEntryInfoW(req->req_file->url, info, &nbytes);
2189 
2190  ts->ftExpires = info->ExpireTime;
2191  ts->ftLastModified = info->LastModifiedTime;
2192 
2193  heap_free(info);
2194  *size = sizeof(*ts);
2195  return ERROR_SUCCESS;
2196  }
2197  return error;
2198  }
2199 
2201  DWORD req_size;
2202 
2203  TRACE("INTERNET_OPTION_DATAFILE_NAME\n");
2204 
2205  if(!req->req_file) {
2206  *size = 0;
2208  }
2209 
2210  if(unicode) {
2211  req_size = (lstrlenW(req->req_file->file_name)+1) * sizeof(WCHAR);
2212  if(*size < req_size)
2214 
2215  *size = req_size;
2216  memcpy(buffer, req->req_file->file_name, *size);
2217  return ERROR_SUCCESS;
2218  }else {
2219  req_size = WideCharToMultiByte(CP_ACP, 0, req->req_file->file_name, -1, NULL, 0, NULL, NULL);
2220  if (req_size > *size)
2222 
2224  -1, buffer, *size, NULL, NULL);
2225  return ERROR_SUCCESS;
2226  }
2227  }
2228 
2231 
2232  if(!req->netconn)
2234 
2235  if(*size < sizeof(INTERNET_CERTIFICATE_INFOA)) {
2236  *size = sizeof(INTERNET_CERTIFICATE_INFOA);
2238  }
2239 
2241  if(context) {
2243  DWORD len;
2244 
2245  memset(info, 0, sizeof(*info));
2246  info->ftExpiry = context->pCertInfo->NotAfter;
2247  info->ftStart = context->pCertInfo->NotBefore;
2248  len = CertNameToStrA(context->dwCertEncodingType,
2249  &context->pCertInfo->Subject, CERT_SIMPLE_NAME_STR|CERT_NAME_STR_CRLF_FLAG, NULL, 0);
2250  info->lpszSubjectInfo = LocalAlloc(0, len);
2251  if(info->lpszSubjectInfo)
2252  CertNameToStrA(context->dwCertEncodingType,
2254  info->lpszSubjectInfo, len);
2255  len = CertNameToStrA(context->dwCertEncodingType,
2256  &context->pCertInfo->Issuer, CERT_SIMPLE_NAME_STR|CERT_NAME_STR_CRLF_FLAG, NULL, 0);
2257  info->lpszIssuerInfo = LocalAlloc(0, len);
2258  if(info->lpszIssuerInfo)
2259  CertNameToStrA(context->dwCertEncodingType,
2261  info->lpszIssuerInfo, len);
2262  info->dwKeySize = NETCON_GetCipherStrength(req->netconn);
2264  return ERROR_SUCCESS;
2265  }
2266  return ERROR_NOT_SUPPORTED;
2267  }
2269  if (*size < sizeof(DWORD))
2271 
2272  *size = sizeof(DWORD);
2273  *(DWORD *)buffer = req->connect_timeout;
2274  return ERROR_SUCCESS;
2276  DWORD flags = 0;
2277 
2278  if (*size < sizeof(DWORD))
2280 
2281  /* FIXME: Add support for:
2282  * INTERNET_REQFLAG_FROM_CACHE
2283  * INTERNET_REQFLAG_CACHE_WRITE_DISABLED
2284  */
2285 
2286  if(req->proxy)
2288  if(!req->status_code)
2290 
2291  TRACE("INTERNET_OPTION_REQUEST_FLAGS returning %x\n", flags);
2292 
2293  *size = sizeof(DWORD);
2294  *(DWORD*)buffer = flags;
2295  return ERROR_SUCCESS;
2296  }
2297  }
2298 
2299  return INET_QueryOption(hdr, option, buffer, size, unicode);
2300 }
2301 
2303 {
2305 
2306  switch(option) {
2307  case 99: /* Undocumented, seems to be INTERNET_OPTION_SECURITY_FLAGS with argument validation */
2308  TRACE("Undocumented option 99\n");
2309 
2310  if (!buffer || size != sizeof(DWORD))
2311  return ERROR_INVALID_PARAMETER;
2312  if(*(DWORD*)buffer & ~SECURITY_SET_MASK)
2314 
2315  /* fall through */
2317  {
2318  DWORD flags;
2319 
2320  if (!buffer || size != sizeof(DWORD))
2321  return ERROR_INVALID_PARAMETER;
2322  flags = *(DWORD *)buffer;
2323  TRACE("INTERNET_OPTION_SECURITY_FLAGS %08x\n", flags);
2325  req->security_flags |= flags;
2326  if(is_valid_netconn(req->netconn))
2327  req->netconn->security_flags |= flags;
2328  return ERROR_SUCCESS;
2329  }
2331  if (!buffer || size != sizeof(DWORD)) return ERROR_INVALID_PARAMETER;
2332  req->connect_timeout = *(DWORD *)buffer;
2333  return ERROR_SUCCESS;
2334 
2336  if (!buffer || size != sizeof(DWORD)) return ERROR_INVALID_PARAMETER;
2337  req->send_timeout = *(DWORD *)buffer;
2338  return ERROR_SUCCESS;
2339 
2341  if (!buffer || size != sizeof(DWORD)) return ERROR_INVALID_PARAMETER;
2342  req->receive_timeout = *(DWORD *)buffer;
2343  return ERROR_SUCCESS;
2344 
2346  heap_free(req->session->userName);
2347  if (!(req->session->userName = heap_strdupW(buffer))) return ERROR_OUTOFMEMORY;
2348  return ERROR_SUCCESS;
2349 
2351  heap_free(req->session->password);
2352  if (!(req->session->password = heap_strdupW(buffer))) return ERROR_OUTOFMEMORY;
2353  return ERROR_SUCCESS;
2354 
2358  return ERROR_SUCCESS;
2359 
2363  return ERROR_SUCCESS;
2364 
2365  }
2366 
2367  return INET_SetOption(hdr, option, buffer, size);
2368 }
2369 
2371 {
2372  WCHAR *header;
2373  DWORD header_len;
2374  BOOL res;
2375 
2376  TRACE("%p\n", req);
2377 
2378  CloseHandle(req->hCacheFile);
2379  req->hCacheFile = NULL;
2380 
2382  header_len = (header ? strlenW(header) : 0);
2385  header, header_len, NULL, 0);
2386  if(res)
2387  req->req_file->is_committed = TRUE;
2388  else
2389  WARN("CommitUrlCacheEntry failed: %u\n", GetLastError());
2390  heap_free(header);
2391 }
2392 
2394 {
2395  static const WCHAR no_cacheW[] = {'n','o','-','c','a','c','h','e',0};
2396  static const WCHAR no_storeW[] = {'n','o','-','s','t','o','r','e',0};
2397 
2399  WCHAR *url;
2400  BOOL b = TRUE;
2401 
2402  /* FIXME: We should free previous cache file earlier */
2403  if(req->req_file) {
2404  req_file_release(req->req_file);
2405  req->req_file = NULL;
2406  }
2407  if(req->hCacheFile) {
2408  CloseHandle(req->hCacheFile);
2409  req->hCacheFile = NULL;
2410  }
2411 
2413  b = FALSE;
2414 
2415  if(b) {
2416  int header_idx;
2417 
2419 
2420  header_idx = HTTP_GetCustomHeaderIndex(req, szCache_Control, 0, FALSE);
2421  if(header_idx != -1) {
2422  WCHAR *ptr;
2423 
2424  for(ptr=req->custHeaders[header_idx].lpszValue; *ptr; ) {
2425  WCHAR *end;
2426 
2427  while(*ptr==' ' || *ptr=='\t')
2428  ptr++;
2429 
2430  end = strchrW(ptr, ',');
2431  if(!end)
2432  end = ptr + strlenW(ptr);
2433 
2434  if(!strncmpiW(ptr, no_cacheW, sizeof(no_cacheW)/sizeof(*no_cacheW)-1)
2435  || !strncmpiW(ptr, no_storeW, sizeof(no_storeW)/sizeof(*no_storeW)-1)) {
2436  b = FALSE;
2437  break;
2438  }
2439 
2440  ptr = end;
2441  if(*ptr == ',')
2442  ptr++;
2443  }
2444  }
2445 
2447  }
2448 
2449  if(!b) {
2450  if(!(req->hdr.dwFlags & INTERNET_FLAG_NEED_FILE))
2451  return;
2452 
2453  FIXME("INTERNET_FLAG_NEED_FILE is not supported correctly\n");
2454  }
2455 
2456  url = compose_request_url(req);
2457  if(!url) {
2458  WARN("Could not get URL\n");
2459  return;
2460  }
2461 
2462  b = CreateUrlCacheEntryW(url, req->contentLength == ~0u ? 0 : req->contentLength, NULL, file_name, 0);
2463  if(!b) {
2464  WARN("Could not create cache entry: %08x\n", GetLastError());
2465  return;
2466  }
2467 
2469  req->req_file->url = url;
2470 
2473  if(req->hCacheFile == INVALID_HANDLE_VALUE) {
2474  WARN("Could not create file: %u\n", GetLastError());
2475  req->hCacheFile = NULL;
2476  return;
2477  }
2478 
2479  if(req->read_size) {
2480  DWORD written;
2481 
2482  b = WriteFile(req->hCacheFile, req->read_buf+req->read_pos, req->read_size, &written, NULL);
2483  if(!b)
2484  FIXME("WriteFile failed: %u\n", GetLastError());
2485 
2486  if(req->data_stream->vtbl->end_of_data(req->data_stream, req))
2487  commit_cache_entry(req);
2488  }
2489 }
2490 
2491 /* read some more data into the read buffer (the read section must be held) */
2492 static DWORD read_more_data( http_request_t *req, int maxlen )
2493 {
2494  DWORD res;
2495  int len;
2496 
2497  if (req->read_pos)
2498  {
2499  /* move existing data to the start of the buffer */
2500  if(req->read_size)
2501  memmove( req->read_buf, req->read_buf + req->read_pos, req->read_size );
2502  req->read_pos = 0;
2503  }
2504 
2505  if (maxlen == -1) maxlen = sizeof(req->read_buf);
2506 
2507  res = NETCON_recv( req->netconn, req->read_buf + req->read_size,
2508  maxlen - req->read_size, TRUE, &len );
2509  if(res == ERROR_SUCCESS)
2510  req->read_size += len;
2511 
2512  return res;
2513 }
2514 
2515 /* remove some amount of data from the read buffer (the read section must be held) */
2516 static void remove_data( http_request_t *req, int count )
2517 {
2518  if (!(req->read_size -= count)) req->read_pos = 0;
2519  else req->read_pos += count;
2520 }
2521 
2523 {
2524  int count, bytes_read, pos = 0;
2525  DWORD res;
2526 
2528  for (;;)
2529  {
2530  BYTE *eol = memchr( req->read_buf + req->read_pos, '\n', req->read_size );
2531 
2532  if (eol)
2533  {
2534  count = eol - (req->read_buf + req->read_pos);
2535  bytes_read = count + 1;
2536  }
2537  else count = bytes_read = req->read_size;
2538 
2539  count = min( count, *len - pos );
2540  memcpy( buffer + pos, req->read_buf + req->read_pos, count );
2541  pos += count;
2542  remove_data( req, bytes_read );
2543  if (eol) break;
2544 
2545  if ((res = read_more_data( req, -1 )))
2546  {
2547  WARN( "read failed %u\n", res );
2549  return res;
2550  }
2551  if (!req->read_size)
2552  {
2553  *len = 0;
2554  TRACE( "returning empty string\n" );
2556  return ERROR_SUCCESS;
2557  }
2558  }
2560 
2561  if (pos < *len)
2562  {
2563  if (pos && buffer[pos - 1] == '\r') pos--;
2564  *len = pos + 1;
2565  }
2566  buffer[*len - 1] = 0;
2567  TRACE( "returning %s\n", debugstr_a(buffer));
2568  return ERROR_SUCCESS;
2569 }
2570 
2571 /* check if we have reached the end of the data to read (the read section must be held) */
2573 {
2574  return !req->read_size && req->data_stream->vtbl->end_of_data(req->data_stream, req);
2575 }
2576 
2578 {
2579  DWORD res;
2580 
2581  res = req->data_stream->vtbl->read(req->data_stream, req, buf, size, read, allow_blocking);
2582  if(res != ERROR_SUCCESS)
2583  *read = 0;
2584  assert(*read <= size);
2585 
2586  if(req->hCacheFile) {
2587  if(*read) {
2588  BOOL bres;
2589  DWORD written;
2590 
2591  bres = WriteFile(req->hCacheFile, buf, *read, &written, NULL);
2592  if(!bres)
2593  FIXME("WriteFile failed: %u\n", GetLastError());
2594  }
2595 
2596  if((res == ERROR_SUCCESS && !*read) || req->data_stream->vtbl->end_of_data(req->data_stream, req))
2597  commit_cache_entry(req);
2598  }
2599 
2600  return res;
2601 }
2602 
2603 /* fetch some more data into the read buffer (the read section must be held) */
2605 {
2606  DWORD res, read=0;
2607 
2608  if(req->read_size == sizeof(req->read_buf))
2609  return ERROR_SUCCESS;
2610 
2611  if(req->read_pos) {
2612  if(req->read_size)
2613  memmove(req->read_buf, req->read_buf+req->read_pos, req->read_size);
2614  req->read_pos = 0;
2615  }
2616 
2617  res = read_http_stream(req, req->read_buf+req->read_size, sizeof(req->read_buf) - req->read_size,
2618  &read, allow_blocking);
2619  if(res != ERROR_SUCCESS)
2620  return res;
2621 
2622  req->read_size += read;
2623 
2624  TRACE("read %u bytes, read_size %u\n", read, req->read_size);
2625  if(read_bytes)
2626  *read_bytes = read;
2627  return res;
2628 }
2629 
2631 {
2632  netconn_stream_t *netconn_stream = (netconn_stream_t*)stream;
2633  return netconn_stream->content_read == netconn_stream->content_length || !is_valid_netconn(req->netconn);
2634 }
2635 
2637  DWORD *read, BOOL allow_blocking)
2638 {
2639  netconn_stream_t *netconn_stream = (netconn_stream_t*)stream;
2641  int ret = 0;
2642 
2643  size = min(size, netconn_stream->content_length-netconn_stream->content_read);
2644 
2645  if(size && is_valid_netconn(req->netconn)) {
2646  res = NETCON_recv(req->netconn, buf, size, allow_blocking, &ret);
2647  if(res == ERROR_SUCCESS) {
2648  if(!ret)
2649  netconn_stream->content_length = netconn_stream->content_read;
2650  netconn_stream->content_read += ret;
2651  }
2652  }
2653 
2654  TRACE("res %u read %u bytes\n", res, ret);
2655  *read = ret;
2656  return res;
2657 }
2658 
2660 {
2661  netconn_stream_t *netconn_stream = (netconn_stream_t*)stream;
2662  BYTE buf[1024];
2663  int len, res;
2664  size_t size;
2665 
2666  if(netconn_stream->content_length == ~0u)
2667  return WSAEISCONN;
2668 
2669  while(netconn_stream->content_read < netconn_stream->content_length) {
2670  size = min(sizeof(buf), netconn_stream->content_length-netconn_stream->content_read);
2671  res = NETCON_recv(req->netconn, buf, size, allow_blocking, &len);
2672  if(res)
2673  return res;
2674  if(!len)
2675  return WSAECONNABORTED;
2676 
2677  netconn_stream->content_read += len;
2678  }
2679 
2680  return ERROR_SUCCESS;
2681 }
2682 
2684 {
2685 }
2686 
2689  netconn_read,
2692 };
2693 
2695 {
2696  assert(stream->buf_size);
2697 
2698  stream->buf_size--;
2699  return stream->buf[stream->buf_pos++];
2700 }
2701 
2703 {
2704  chunked_stream_t *chunked_stream = (chunked_stream_t*)stream;
2705  switch(chunked_stream->state) {
2706  case CHUNKED_STREAM_STATE_DISCARD_EOL_AT_END:
2707  case CHUNKED_STREAM_STATE_END_OF_STREAM:
2708  case CHUNKED_STREAM_STATE_ERROR:
2709  return TRUE;
2710  default:
2711  return FALSE;
2712  }
2713 }
2714 
2716  DWORD *read, BOOL allow_blocking)
2717 {
2718  chunked_stream_t *chunked_stream = (chunked_stream_t*)stream;
2719  DWORD ret_read = 0, res = ERROR_SUCCESS;
2720  BOOL continue_read = TRUE;
2721  int read_bytes;
2722  char ch;
2723 
2724  do {
2725  TRACE("state %d\n", chunked_stream->state);
2726 
2727  /* Ensure that we have data in the buffer for states that need it. */
2728  if(!chunked_stream->buf_size) {
2729  BOOL blocking_read = allow_blocking;
2730 
2731  switch(chunked_stream->state) {
2732  case CHUNKED_STREAM_STATE_DISCARD_EOL_AT_END:
2733  case CHUNKED_STREAM_STATE_DISCARD_EOL_AFTER_SIZE:
2734  /* never allow blocking after 0 chunk size */
2735  if(!chunked_stream->chunk_size)
2736  blocking_read = FALSE;
2737  /* fall through */
2738  case CHUNKED_STREAM_STATE_READING_CHUNK_SIZE:
2739  case CHUNKED_STREAM_STATE_DISCARD_EOL_AFTER_DATA:
2740  chunked_stream->buf_pos = 0;
2741  res = NETCON_recv(req->netconn, chunked_stream->buf, sizeof(chunked_stream->buf), blocking_read, &read_bytes);
2742  if(res == ERROR_SUCCESS && read_bytes) {
2743  chunked_stream->buf_size += read_bytes;
2744  }else if(res == WSAEWOULDBLOCK) {
2745  if(ret_read || allow_blocking)
2746  res = ERROR_SUCCESS;
2747  continue_read = FALSE;
2748  continue;
2749  }else {
2750  chunked_stream->state = CHUNKED_STREAM_STATE_ERROR;
2751  }
2752  break;
2753  default:
2754  break;
2755  }
2756  }
2757 
2758  switch(chunked_stream->state) {
2759  case CHUNKED_STREAM_STATE_READING_CHUNK_SIZE:
2760  ch = next_chunked_data_char(chunked_stream);
2761 
2762  if(ch >= '0' && ch <= '9') {
2763  chunked_stream->chunk_size = chunked_stream->chunk_size * 16 + ch - '0';
2764  }else if(ch >= 'a' && ch <= 'f') {
2765  chunked_stream->chunk_size = chunked_stream->chunk_size * 16 + ch - 'a' + 10;
2766  }else if (ch >= 'A' && ch <= 'F') {
2767  chunked_stream->chunk_size = chunked_stream->chunk_size * 16 + ch - 'A' + 10;
2768  }else if (ch == ';' || ch == '\r' || ch == '\n') {
2769  TRACE("reading %u byte chunk\n", chunked_stream->chunk_size);
2770  chunked_stream->buf_size++;
2771  chunked_stream->buf_pos--;
2772  if(req->contentLength == ~0u) req->contentLength = chunked_stream->chunk_size;
2773  else req->contentLength += chunked_stream->chunk_size;
2774  chunked_stream->state = CHUNKED_STREAM_STATE_DISCARD_EOL_AFTER_SIZE;
2775  }
2776  break;
2777 
2778  case CHUNKED_STREAM_STATE_DISCARD_EOL_AFTER_SIZE:
2779  ch = next_chunked_data_char(chunked_stream);
2780  if(ch == '\n')
2781  chunked_stream->state = chunked_stream->chunk_size
2782  ? CHUNKED_STREAM_STATE_READING_CHUNK
2783  : CHUNKED_STREAM_STATE_DISCARD_EOL_AT_END;
2784  else if(ch != '\r')
2785  WARN("unexpected char '%c'\n", ch);
2786  break;
2787 
2788  case CHUNKED_STREAM_STATE_READING_CHUNK:
2789  assert(chunked_stream->chunk_size);
2790  if(!size) {
2791  continue_read = FALSE;
2792  break;
2793  }
2794  read_bytes = min(size, chunked_stream->chunk_size);
2795 
2796  if(chunked_stream->buf_size) {
2797  if(read_bytes > chunked_stream->buf_size)
2798  read_bytes = chunked_stream->buf_size;
2799 
2800  memcpy(buf+ret_read, chunked_stream->buf+chunked_stream->buf_pos, read_bytes);
2801  chunked_stream->buf_pos += read_bytes;
2802  chunked_stream->buf_size -= read_bytes;
2803  }else {
2804  res = NETCON_recv(req->netconn, (char*)buf+ret_read, read_bytes,
2805  allow_blocking, (int*)&read_bytes);
2806  if(res != ERROR_SUCCESS) {
2807  continue_read = FALSE;
2808  break;
2809  }
2810 
2811  if(!read_bytes) {
2812  chunked_stream->state = CHUNKED_STREAM_STATE_ERROR;
2813  continue;
2814  }
2815  }
2816 
2817  chunked_stream->chunk_size -= read_bytes;
2818  size -= read_bytes;
2819  ret_read += read_bytes;
2820  if(!chunked_stream->chunk_size)
2821  chunked_stream->state = CHUNKED_STREAM_STATE_DISCARD_EOL_AFTER_DATA;
2822  allow_blocking = FALSE;
2823  break;
2824 
2825  case CHUNKED_STREAM_STATE_DISCARD_EOL_AFTER_DATA:
2826  ch = next_chunked_data_char(chunked_stream);
2827  if(ch == '\n')
2828  chunked_stream->state = CHUNKED_STREAM_STATE_READING_CHUNK_SIZE;
2829  else if(ch != '\r')
2830  WARN("unexpected char '%c'\n", ch);
2831  break;
2832 
2833  case CHUNKED_STREAM_STATE_DISCARD_EOL_AT_END:
2834  ch = next_chunked_data_char(chunked_stream);
2835  if(ch == '\n')
2836  chunked_stream->state = CHUNKED_STREAM_STATE_END_OF_STREAM;
2837  else if(ch != '\r')
2838  WARN("unexpected char '%c'\n", ch);
2839  break;
2840 
2841  case CHUNKED_STREAM_STATE_END_OF_STREAM:
2842  case CHUNKED_STREAM_STATE_ERROR:
2843  continue_read = FALSE;
2844  break;
2845  }
2846  } while(continue_read);
2847 
2848  if(ret_read)
2849  res = ERROR_SUCCESS;
2850  if(res != ERROR_SUCCESS)
2851  return res;
2852 
2853  TRACE("read %d bytes\n", ret_read);
2854  *read = ret_read;
2855  return ERROR_SUCCESS;
2856 }
2857 
2859 {
2860  chunked_stream_t *chunked_stream = (chunked_stream_t*)stream;
2861  BYTE buf[1024];
2862  DWORD size, res;
2863 
2864  while(chunked_stream->state != CHUNKED_STREAM_STATE_END_OF_STREAM
2865  && chunked_stream->state != CHUNKED_STREAM_STATE_ERROR) {
2866  res = chunked_read(stream, req, buf, sizeof(buf), &size, allow_blocking);
2867  if(res != ERROR_SUCCESS)
2868  return res;
2869  }
2870 
2871  if(chunked_stream->state != CHUNKED_STREAM_STATE_END_OF_STREAM)
2872  return ERROR_NO_DATA;
2873  return ERROR_SUCCESS;
2874 }
2875 
2877 {
2878  chunked_stream_t *chunked_stream = (chunked_stream_t*)stream;
2879  heap_free(chunked_stream);
2880 }
2881 
2884  chunked_read,
2887 };
2888 
2889 /* set the request content length based on the headers */
2891 {
2892  static const WCHAR szChunked[] = {'c','h','u','n','k','e','d',0};
2893  static const WCHAR headW[] = {'H','E','A','D',0};
2894  WCHAR encoding[20];
2895  DWORD size;
2896 
2897  if(request->status_code == HTTP_STATUS_NO_CONTENT || !strcmpW(request->verb, headW)) {
2898  request->contentLength = request->netconn_stream.content_length = 0;
2899  return ERROR_SUCCESS;
2900  }
2901 
2902  size = sizeof(request->contentLength);
2904  &request->contentLength, &size, NULL) != ERROR_SUCCESS)
2905  request->contentLength = ~0u;
2906  request->netconn_stream.content_length = request->contentLength;
2907  request->netconn_stream.content_read = request->read_size;
2908 
2909  size = sizeof(encoding);
2911  !strcmpiW(encoding, szChunked))
2912  {
2913  chunked_stream_t *chunked_stream;
2914 
2915  chunked_stream = heap_alloc(sizeof(*chunked_stream));
2916  if(!chunked_stream)
2917  return ERROR_OUTOFMEMORY;
2918 
2919  chunked_stream->data_stream.vtbl = &chunked_stream_vtbl;
2920  chunked_stream->buf_size = chunked_stream->buf_pos = 0;
2921  chunked_stream->chunk_size = 0;
2922  chunked_stream->state = CHUNKED_STREAM_STATE_READING_CHUNK_SIZE;
2923 
2924  if(request->read_size) {
2925  memcpy(chunked_stream->buf, request->read_buf+request->read_pos, request->read_size);
2926  chunked_stream->buf_size = request->read_size;
2927  request->read_size = request->read_pos = 0;
2928  }
2929 
2930  request->data_stream = &chunked_stream->data_stream;
2931  request->contentLength = ~0u;
2932  }
2933 
2934  if(request->hdr.decoding) {
2935  int encoding_idx;
2936 
2937  static const WCHAR deflateW[] = {'d','e','f','l','a','t','e',0};
2938  static const WCHAR gzipW[] = {'g','z','i','p',0};
2939 
2940  EnterCriticalSection( &request->headers_section );
2941 
2943  if(encoding_idx != -1) {
2944  if(!strcmpiW(request->custHeaders[encoding_idx].lpszValue, gzipW)) {
2945  HTTP_DeleteCustomHeader(request, encoding_idx);
2946  LeaveCriticalSection( &request->headers_section );
2947  return init_gzip_stream(request, TRUE);
2948  }
2949  if(!strcmpiW(request->custHeaders[encoding_idx].lpszValue, deflateW)) {
2950  HTTP_DeleteCustomHeader(request, encoding_idx);
2951  LeaveCriticalSection( &request->headers_section );
2952  return init_gzip_stream(request, FALSE);
2953  }
2954  }
2955 
2956  LeaveCriticalSection( &request->headers_section );
2957  }
2958 
2959  return ERROR_SUCCESS;
2960 }
2961 
2963 {
2965 
2966  iar.dwResult = result;
2967  iar.dwError = error;
2968 
2970  sizeof(INTERNET_ASYNC_RESULT));
2971 }
2972 
2974 {
2975  DWORD res, read = 0;
2976 
2977  TRACE("%p\n", req);
2978 
2980 
2981  res = refill_read_buffer(req, FALSE, &read);
2982  if(res == ERROR_SUCCESS)
2983  read += req->read_size;
2984 
2986 
2987  if(res != WSAEWOULDBLOCK && (res != ERROR_SUCCESS || !read)) {
2988  WARN("res %u read %u, closing connection\n", res, read);
2990  }
2991 
2992  if(res != ERROR_SUCCESS && res != WSAEWOULDBLOCK) {
2993  send_request_complete(req, 0, res);
2994  return;
2995  }
2996 
2998 }
2999 
3000 /* read data from the http connection (the read section must be held) */
3001 static DWORD HTTPREQ_Read(http_request_t *req, void *buffer, DWORD size, DWORD *read, BOOL allow_blocking)
3002 {
3003  DWORD current_read = 0, ret_read = 0;
3005 
3007 
3008  if(req->read_size) {
3009  ret_read = min(size, req->read_size);
3010  memcpy(buffer, req->read_buf+req->read_pos, ret_read);
3011  req->read_size -= ret_read;
3012  req->read_pos += ret_read;
3013  allow_blocking = FALSE;
3014  }
3015 
3016  if(ret_read < size) {
3017  res = read_http_stream(req, (BYTE*)buffer+ret_read, size-ret_read, &current_read, allow_blocking);
3018  if(res == ERROR_SUCCESS)
3019  ret_read += current_read;
3020  else if(res == WSAEWOULDBLOCK && ret_read)
3021  res = ERROR_SUCCESS;
3022  }
3023 
3025 
3026  *read = ret_read;
3027  TRACE( "retrieved %u bytes (res %u)\n", ret_read, res );
3028 
3029  if(res != WSAEWOULDBLOCK) {
3030  if(res != ERROR_SUCCESS)
3032  else if(!ret_read && drain_content(req, FALSE) == ERROR_SUCCESS)
3033  http_release_netconn(req, TRUE);
3034  }
3035 
3036  return res;
3037 }
3038 
3039 static DWORD drain_content(http_request_t *req, BOOL blocking)
3040 {
3041  DWORD res;
3042 
3043  TRACE("%p\n", req->netconn);
3044 
3045  if(!is_valid_netconn(req->netconn))
3046  return ERROR_NO_DATA;
3047 
3048  if(!strcmpW(req->verb, szHEAD))
3049  return ERROR_SUCCESS;
3050 
3052  res = req->data_stream->vtbl->drain_content(req->data_stream, req, blocking);
3054  return res;
3055 }
3056 
3057 typedef struct {
3059  void *buf;
3064 
3066 {
3068  http_request_t *req = (http_request_t*)task->hdr.hdr;
3069  DWORD res = ERROR_SUCCESS, read = task->read_pos, complete_arg = 0;
3070 
3071  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);
3072 
3073  if(task->buf) {
3074  DWORD read_bytes;
3075  while (read < task->size) {
3076  res = HTTPREQ_Read(req, (char*)task->buf + read, task->size - read, &read_bytes, TRUE);
3077  if (res != ERROR_SUCCESS || !read_bytes)
3078  break;
3079  read += read_bytes;
3080  }
3081  }else {
3083  res = refill_read_buffer(req, TRUE, &read);
3085 
3086  if(task->ret_read)
3087  complete_arg = read; /* QueryDataAvailable reports read bytes in request complete notification */
3088  if(res != ERROR_SUCCESS || !read)
3090  }
3091 
3092  TRACE("res %u read %u\n", res, read);
3093 
3094  if(task->ret_read)
3095  *task->ret_read = read;
3096 
3097  /* FIXME: We should report bytes transferred before decoding content. */
3099 
3100  if(res != ERROR_SUCCESS)
3101  complete_arg = res;
3102  send_request_complete(req, res == ERROR_SUCCESS, complete_arg);
3103 }
3104 
3105 static DWORD async_read(http_request_t *req, void *buf, DWORD size, DWORD read_pos, DWORD *ret_read)
3106 {
3107  read_file_task_t *task;
3108 
3109  task = alloc_async_task(&req->hdr, async_read_file_proc, sizeof(*task));
3110  if(!task)
3111  return ERROR_OUTOFMEMORY;
3112 
3113  task->buf = buf;
3114  task->size = size;
3115  task->read_pos = read_pos;
3116  task->ret_read = ret_read;
3117 
3118  INTERNET_AsyncCall(&task->hdr);
3119  return ERROR_IO_PENDING;
3120 }
3121 
3124 {
3126  DWORD res = ERROR_SUCCESS, read = 0, cread, error = ERROR_SUCCESS;
3127  BOOL allow_blocking, notify_received = FALSE;
3128 
3129  TRACE("(%p %p %u %x)\n", req, buf, size, flags);
3130 
3131  if (flags & ~(IRF_ASYNC|IRF_NO_WAIT))
3132  FIXME("these dwFlags aren't implemented: 0x%x\n", flags & ~(IRF_ASYNC|IRF_NO_WAIT));
3133 
3134  allow_blocking = !(req->session->appInfo->hdr.dwFlags & INTERNET_FLAG_ASYNC);
3135 
3136  if(allow_blocking || TryEnterCriticalSection(&req->read_section)) {
3137  if(allow_blocking)
3139  if(hdr->dwError == ERROR_SUCCESS)
3140  hdr->dwError = INTERNET_HANDLE_IN_USE;
3141  else if(hdr->dwError == INTERNET_HANDLE_IN_USE)
3143 
3144  if(req->read_size) {
3145  read = min(size, req->read_size);
3146  memcpy(buf, req->read_buf + req->read_pos, read);
3147  req->read_size -= read;
3148  req->read_pos += read;
3149  }
3150 
3151  if(read < size && (!read || !(flags & IRF_NO_WAIT)) && !end_of_read_data(req)) {
3155  notify_received = TRUE;
3156 
3157  while(read < size) {
3158  res = HTTPREQ_Read(req, (char*)buf+read, size-read, &cread, allow_blocking);
3159  read += cread;
3160  if (res != ERROR_SUCCESS || !cread)
3161  break;
3162  }
3163  }
3164 
3165  if(hdr->dwError == INTERNET_HANDLE_IN_USE)
3166  hdr->dwError = ERROR_SUCCESS;
3167  else
3168  error = hdr->dwError;
3169 
3171  }else {
3172  res = WSAEWOULDBLOCK;
3173  }
3174 
3175  if(res == WSAEWOULDBLOCK) {
3176  if(!(flags & IRF_NO_WAIT))
3177  return async_read(req, buf, size, read, ret_read);
3178  if(!read)
3179  return async_read(req, NULL, 0, 0, NULL);
3180  res = ERROR_SUCCESS;
3181  }
3182 
3183  *ret_read = read;
3184  if (res != ERROR_SUCCESS)
3185  return res;
3186 
3187  if(notify_received)
3189  &read, sizeof(read));
3190  return error;
3191 }
3192 
3194 {
3195  DWORD res;
3197 
3199 
3200  *written = 0;
3201  res = NETCON_send(request->netconn, buffer, size, 0, (LPINT)written);
3202  if (res == ERROR_SUCCESS)
3203  request->bytesWritten += *written;
3204 
3205  INTERNET_SendCallback(&request->hdr, request->hdr.dwContext, INTERNET_STATUS_REQUEST_SENT, written, sizeof(DWORD));
3206  return res;
3207 }
3208 
3210 {
3213  BOOL allow_blocking, notify_received = FALSE;
3214 
3215  TRACE("(%p %p %x %lx)\n", req, available, flags, ctx);
3216 
3217  if (flags & ~(IRF_ASYNC|IRF_NO_WAIT))
3218  FIXME("these dwFlags aren't implemented: 0x%x\n", flags & ~(IRF_ASYNC|IRF_NO_WAIT));
3219 
3220  *available = 0;
3221  allow_blocking = !(req->session->appInfo->hdr.dwFlags & INTERNET_FLAG_ASYNC);
3222 
3223  if(allow_blocking || TryEnterCriticalSection(&req->read_section)) {
3224  if(allow_blocking)
3226  if(hdr->dwError == ERROR_SUCCESS)
3227  hdr->dwError = INTERNET_HANDLE_IN_USE;
3228  else if(hdr->dwError == INTERNET_HANDLE_IN_USE)
3230 
3231  avail = req->read_size;
3232 
3233  if(!avail && !end_of_read_data(req)) {
3237  notify_received = TRUE;
3238 
3239  res = refill_read_buffer(req, allow_blocking, &avail);
3240  }
3241 
3242  if(hdr->dwError == INTERNET_HANDLE_IN_USE)
3243  hdr->dwError = ERROR_SUCCESS;
3244  else
3245  error = hdr->dwError;
3246 
3248  }else {
3249  res = WSAEWOULDBLOCK;
3250  }
3251 
3252  if(res == WSAEWOULDBLOCK)
3253  return async_read(req, NULL, 0, 0, available);
3254 
3255  if (res != ERROR_SUCCESS)
3256  return res;
3257 
3258  *available = avail;
3259  if(notify_received)
3261  &avail, sizeof(avail));
3262  return error;
3263 }
3264 
3266 {
3268 
3269  TRACE("(%p)\n", req);
3270 
3271  if(!req->req_file) {
3272  WARN("No cache file name available\n");
3273  return ERROR_FILE_NOT_FOUND;
3274  }
3275 
3276  *ret = req_file_addref(req->req_file);
3277  return ERROR_SUCCESS;
3278 }
3279 
3280 static const object_vtbl_t HTTPREQVtbl = {
3288  NULL,
3290 };
3291 
3292 /***********************************************************************
3293  * HTTP_HttpOpenRequestW (internal)
3294  *
3295  * Open a HTTP request handle
3296  *
3297  * RETURNS
3298  * HINTERNET a HTTP request handle on success
3299  * NULL on failure
3300  *
3301  */
3303  LPCWSTR lpszVerb, LPCWSTR lpszObjectName, LPCWSTR lpszVersion,
3304  LPCWSTR lpszReferrer , LPCWSTR *lpszAcceptTypes,
3305  DWORD dwFlags, DWORD_PTR dwContext, HINTERNET *ret)
3306 {
3307  appinfo_t *hIC = session->appInfo;
3309  DWORD port, len;
3310 
3311  TRACE("-->\n");
3312 
3313  request = alloc_object(&session->hdr, &HTTPREQVtbl, sizeof(http_request_t));
3314  if(!request)
3315  return ERROR_OUTOFMEMORY;
3316 
3317  request->hdr.htype = WH_HHTTPREQ;
3318  request->hdr.dwFlags = dwFlags;
3319  request->hdr.dwContext = dwContext;
3320  request->hdr.decoding = session->hdr.decoding;
3321  request->contentLength = ~0u;
3322 
3323  request->netconn_stream.data_stream.vtbl = &netconn_stream_vtbl;
3324  request->data_stream = &request->netconn_stream.data_stream;
3325  request->connect_timeout = session->connect_timeout;
3326  request->send_timeout = session->send_timeout;
3327  request->receive_timeout = session->receive_timeout;
3328 
3329  InitializeCriticalSection( &request->headers_section );
3330  request->headers_section.DebugInfo->Spare[0] = (DWORD_PTR)(__FILE__ ": http_request_t.headers_section");
3331 
3332  InitializeCriticalSection( &request->read_section );
3333  request->read_section.DebugInfo->Spare[0] = (DWORD_PTR)(__FILE__ ": http_request_t.read_section");
3334 
3335  WININET_AddRef( &session->hdr );
3336  request->session = session;
3337  list_add_head( &session->hdr.children, &request->hdr.entry );
3338 
3339  port = session->hostPort;
3341  port = (session->hdr.dwFlags & INTERNET_FLAG_SECURE) ?
3343 
3344  request->server = get_server(substrz(session->hostName), port, (dwFlags & INTERNET_FLAG_SECURE) != 0, TRUE);
3345  if(!request->server) {
3346  WININET_Release(&request->hdr);
3347  return ERROR_OUTOFMEMORY;
3348  }
3349 
3351  request->security_flags |= SECURITY_FLAG_IGNORE_CERT_CN_INVALID;
3354 
3355  if (lpszObjectName && *lpszObjectName) {
3356  HRESULT rc;
3357  WCHAR dummy;
3358 
3359  len = 1;
3360  rc = UrlCanonicalizeW(lpszObjectName, &dummy, &len, URL_ESCAPE_SPACES_ONLY);
3361  if (rc != E_POINTER)
3362  len = strlenW(lpszObjectName)+1;
3363  request->path = heap_alloc(len*sizeof(WCHAR));
3364  rc = UrlCanonicalizeW(lpszObjectName, request->path, &len,
3366  if (rc != S_OK)
3367  {
3368  ERR("Unable to escape string!(%s) (%d)\n",debugstr_w(lpszObjectName),rc);
3369  strcpyW(request->path,lpszObjectName);
3370  }
3371  }else {
3372  static const WCHAR slashW[] = {'/',0};
3373 
3375  }
3376 
3377  if (lpszReferrer && *lpszReferrer)
3379 
3380  if (lpszAcceptTypes)
3381  {
3382  int i;
3383  for (i = 0; lpszAcceptTypes[i]; i++)
3384  {
3385  if (!*lpszAcceptTypes[i]) continue;
3386  HTTP_ProcessHeader(request, HTTP_ACCEPT, lpszAcceptTypes[i],
3390  }
3391  }
3392 
3393  request->verb = heap_strdupW(lpszVerb && *lpszVerb ? lpszVerb : szGET);
3394  request->version = heap_strdupW(lpszVersion && *lpszVersion ? lpszVersion : g_szHttp1_1);
3395 
3396  if (hIC->proxy && hIC->proxy[0] && !HTTP_ShouldBypassProxy(hIC, session->hostName))
3397  HTTP_DealWithProxy( hIC, session, request );
3398 
3399  INTERNET_SendCallback(&session->hdr, dwContext,
3400  INTERNET_STATUS_HANDLE_CREATED, &request->hdr.hInternet,
3401  sizeof(HINTERNET));
3402 
3403  TRACE("<-- (%p)\n", request);
3404 
3405  *ret = request->hdr.hInternet;
3406  return ERROR_SUCCESS;
3407 }
3408 
3409 /***********************************************************************
3410  * HttpOpenRequestW (WININET.@)
3411  *
3412  * Open a HTTP request handle
3413  *
3414  * RETURNS
3415  * HINTERNET a HTTP request handle on success
3416  * NULL on failure
3417  *
3418  */
3420  LPCWSTR lpszVerb, LPCWSTR lpszObjectName, LPCWSTR lpszVersion,
3421  LPCWSTR lpszReferrer , LPCWSTR *lpszAcceptTypes,
3422  DWORD dwFlags, DWORD_PTR dwContext)
3423 {
3424  http_session_t *session;
3425  HINTERNET handle = NULL;
3426  DWORD res;
3427 
3428  TRACE("(%p, %s, %s, %s, %s, %p, %08x, %08lx)\n", hHttpSession,
3429  debugstr_w(lpszVerb), debugstr_w(lpszObjectName),
3430  debugstr_w(lpszVersion), debugstr_w(lpszReferrer), lpszAcceptTypes,
3431  dwFlags, dwContext);
3432  if(lpszAcceptTypes!=NULL)
3433  {
3434  int i;
3435  for(i=0;lpszAcceptTypes[i]!=NULL;i++)
3436  TRACE("\taccept type: %s\n",debugstr_w(lpszAcceptTypes[i]));
3437  }
3438 
3439  session = (http_session_t*) get_handle_object( hHttpSession );
3440  if (NULL == session || session->hdr.htype != WH_HHTTPSESSION)
3441  {
3443  goto lend;
3444  }
3445 
3446  /*
3447  * My tests seem to show that the windows version does not
3448  * become asynchronous until after this point. And anyhow
3449  * if this call was asynchronous then how would you get the
3450  * necessary HINTERNET pointer returned by this function.
3451  *
3452  */
3453  res = HTTP_HttpOpenRequestW(session, lpszVerb, lpszObjectName,
3454  lpszVersion, lpszReferrer, lpszAcceptTypes,
3455  dwFlags, dwContext, &handle);
3456 lend:
3457  if( session )
3458  WININET_Release( &session->hdr );
3459  TRACE("returning %p\n", handle);
3460  if(res != ERROR_SUCCESS)
3461  SetLastError(res);
3462  return handle;
3463 }
3464 
3465 static const LPCWSTR header_lookup[] = {
3466  szMime_Version, /* HTTP_QUERY_MIME_VERSION = 0 */
3467  szContent_Type, /* HTTP_QUERY_CONTENT_TYPE = 1 */
3468  szContent_Transfer_Encoding,/* HTTP_QUERY_CONTENT_TRANSFER_ENCODING = 2 */
3469  szContent_ID, /* HTTP_QUERY_CONTENT_ID = 3 */
3470  NULL, /* HTTP_QUERY_CONTENT_DESCRIPTION = 4 */
3471  szContent_Length, /* HTTP_QUERY_CONTENT_LENGTH = 5 */
3472  szContent_Language, /* HTTP_QUERY_CONTENT_LANGUAGE = 6 */
3473  szAllow, /* HTTP_QUERY_ALLOW = 7 */
3474  szPublic, /* HTTP_QUERY_PUBLIC = 8 */
3475  szDate, /* HTTP_QUERY_DATE = 9 */
3476  szExpires, /* HTTP_QUERY_EXPIRES = 10 */
3477  szLast_Modified, /* HTTP_QUERY_LAST_MODIFIED = 11 */
3478  NULL, /* HTTP_QUERY_MESSAGE_ID = 12 */
3479  szURI, /* HTTP_QUERY_URI = 13 */
3480  szFrom, /* HTTP_QUERY_DERIVED_FROM = 14 */
3481  NULL, /* HTTP_QUERY_COST = 15 */
3482  NULL, /* HTTP_QUERY_LINK = 16 */
3483  szPragma, /* HTTP_QUERY_PRAGMA = 17 */
3484  NULL, /* HTTP_QUERY_VERSION = 18 */
3485  szStatus, /* HTTP_QUERY_STATUS_CODE = 19 */
3486  NULL, /* HTTP_QUERY_STATUS_TEXT = 20 */
3487  NULL, /* HTTP_QUERY_RAW_HEADERS = 21 */
3488  NULL, /* HTTP_QUERY_RAW_HEADERS_CRLF = 22 */
3489  szConnection, /* HTTP_QUERY_CONNECTION = 23 */
3490  szAccept, /* HTTP_QUERY_ACCEPT = 24 */
3491  szAccept_Charset, /* HTTP_QUERY_ACCEPT_CHARSET = 25 */
3492  szAccept_Encoding, /* HTTP_QUERY_ACCEPT_ENCODING = 26 */
3493  szAccept_Language, /* HTTP_QUERY_ACCEPT_LANGUAGE = 27 */
3494  szAuthorization, /* HTTP_QUERY_AUTHORIZATION = 28 */
3495  szContent_Encoding, /* HTTP_QUERY_CONTENT_ENCODING = 29 */
3496  NULL, /* HTTP_QUERY_FORWARDED = 30 */
3497  NULL, /* HTTP_QUERY_FROM = 31 */
3498  szIf_Modified_Since, /* HTTP_QUERY_IF_MODIFIED_SINCE = 32 */
3499  szLocation, /* HTTP_QUERY_LOCATION = 33 */
3500  NULL, /* HTTP_QUERY_ORIG_URI = 34 */
3501  szReferer, /* HTTP_QUERY_REFERER = 35 */
3502  szRetry_After, /* HTTP_QUERY_RETRY_AFTER = 36 */
3503  szServer, /* HTTP_QUERY_SERVER = 37 */
3504  NULL, /* HTTP_TITLE = 38 */
3505  szUser_Agent, /* HTTP_QUERY_USER_AGENT = 39 */
3506  szWWW_Authenticate, /* HTTP_QUERY_WWW_AUTHENTICATE = 40 */
3507  szProxy_Authenticate, /* HTTP_QUERY_PROXY_AUTHENTICATE = 41 */
3508  szAccept_Ranges, /* HTTP_QUERY_ACCEPT_RANGES = 42 */
3509  szSet_Cookie, /* HTTP_QUERY_SET_COOKIE = 43 */
3510  szCookie, /* HTTP_QUERY_COOKIE = 44 */
3511  NULL, /* HTTP_QUERY_REQUEST_METHOD = 45 */
3512  NULL, /* HTTP_QUERY_REFRESH = 46 */
3513  szContent_Disposition, /* HTTP_QUERY_CONTENT_DISPOSITION = 47 */
3514  szAge, /* HTTP_QUERY_AGE = 48 */
3515  szCache_Control, /* HTTP_QUERY_CACHE_CONTROL = 49 */
3516  szContent_Base, /* HTTP_QUERY_CONTENT_BASE = 50 */
3517  szContent_Location, /* HTTP_QUERY_CONTENT_LOCATION = 51 */
3518  szContent_MD5, /* HTTP_QUERY_CONTENT_MD5 = 52 */
3519  szContent_Range, /* HTTP_QUERY_CONTENT_RANGE = 53 */
3520  szETag, /* HTTP_QUERY_ETAG = 54 */
3521  hostW, /* HTTP_QUERY_HOST = 55 */
3522  szIf_Match, /* HTTP_QUERY_IF_MATCH = 56 */
3523  szIf_None_Match, /* HTTP_QUERY_IF_NONE_MATCH = 57 */
3524  szIf_Range, /* HTTP_QUERY_IF_RANGE = 58 */
3525  szIf_Unmodified_Since, /* HTTP_QUERY_IF_UNMODIFIED_SINCE = 59 */
3526  szMax_Forwards, /* HTTP_QUERY_MAX_FORWARDS = 60 */
3527  szProxy_Authorization, /* HTTP_QUERY_PROXY_AUTHORIZATION = 61 */
3528  szRange, /* HTTP_QUERY_RANGE = 62 */
3529  szTransfer_Encoding, /* HTTP_QUERY_TRANSFER_ENCODING = 63 */
3530  szUpgrade, /* HTTP_QUERY_UPGRADE = 64 */
3531  szVary, /* HTTP_QUERY_VARY = 65 */
3532  szVia, /* HTTP_QUERY_VIA = 66 */
3533  szWarning, /* HTTP_QUERY_WARNING = 67 */
3534  szExpect, /* HTTP_QUERY_EXPECT = 68 */
3535  szProxy_Connection, /* HTTP_QUERY_PROXY_CONNECTION = 69 */
3536  szUnless_Modified_Since, /* HTTP_QUERY_UNLESS_MODIFIED_SINCE = 70 */
3537 };
3538 
3539 #define LAST_TABLE_HEADER (sizeof(header_lookup)/sizeof(header_lookup[0]))
3540 
3541 /***********************************************************************
3542  * HTTP_HttpQueryInfoW (internal)
3543  */
3546 {
3547  LPHTTPHEADERW lphttpHdr = NULL;
3548  BOOL request_only = !!(dwInfoLevel & HTTP_QUERY_FLAG_REQUEST_HEADERS);
3549  INT requested_index = lpdwIndex ? *lpdwIndex : 0;
3551  INT index = -1;
3552 
3553  EnterCriticalSection( &request->headers_section );
3554 
3555  /* Find requested header structure */
3556  switch (level)
3557  {
3558  case HTTP_QUERY_CUSTOM:
3559  if (!lpBuffer)
3560  {
3561  LeaveCriticalSection( &request->headers_section );
3562  return ERROR_INVALID_PARAMETER;
3563  }
3564  index = HTTP_GetCustomHeaderIndex(request, lpBuffer, requested_index, request_only);
3565  break;
3567  {
3568  LPWSTR headers;
3569  DWORD len = 0;
3571 
3572  if (request_only)
3574  else
3576  if (!headers)
3577  {
3578  LeaveCriticalSection( &request->headers_section );
3579  return ERROR_OUTOFMEMORY;
3580  }
3581 
3582  len = strlenW(headers) * sizeof(WCHAR);
3583  if (len + sizeof(WCHAR) > *lpdwBufferLength)
3584  {
3585  len += sizeof(WCHAR);
3587  }
3588  else if (lpBuffer)
3589  {
3590  memcpy(lpBuffer, headers, len + sizeof(WCHAR));
3591  TRACE("returning data: %s\n", debugstr_wn(lpBuffer, len / sizeof(WCHAR)));
3592  res = ERROR_SUCCESS;
3593  }
3594  *lpdwBufferLength = len;
3595 
3596  heap_free(headers);
3597  LeaveCriticalSection( &request->headers_section );
3598  return res;
3599  }
3601  {
3602  LPWSTR headers;
3603  DWORD len;
3604 
3605  if (request_only)
3607  else
3609 
3610  if (!headers)
3611  {
3612  LeaveCriticalSection( &request->headers_section );
3613  return ERROR_OUTOFMEMORY;
3614  }
3615 
3616  len = strlenW(headers) * sizeof(WCHAR);
3617  if (len > *lpdwBufferLength)
3618  {
3619  *lpdwBufferLength = len;
3620  heap_free(headers);
3621  LeaveCriticalSection( &request->headers_section );
3623  }
3624 
3625  if (lpBuffer)
3626  {
3627  DWORD i;
3628 
3629  TRACE("returning data: %s\n", debugstr_wn(headers, len / sizeof(WCHAR)));
3630 
3631  for (i = 0; i < len / sizeof(WCHAR); i++)
3632  {
3633  if (headers[i] == '\n')
3634  headers[i] = 0;
3635  }
3637  }
3638  *lpdwBufferLength = len - sizeof(WCHAR);
3639 
3640  heap_free(headers);
3641  LeaveCriticalSection( &request->headers_section );
3642  return ERROR_SUCCESS;
3643  }
3645  if (request->statusText)
3646  {
3647  DWORD len = strlenW(request->statusText);
3648  if (len + 1 > *lpdwBufferLength/sizeof(WCHAR))
3649  {
3650  *lpdwBufferLength = (len + 1) * sizeof(WCHAR);
3651  LeaveCriticalSection( &request->headers_section );
3653  }
3654  if (lpBuffer)
3655  {
3656  memcpy(lpBuffer, request->statusText, (len + 1) * sizeof(WCHAR));
3657  TRACE("returning data: %s\n", debugstr_wn(lpBuffer, len));
3658  }
3659  *lpdwBufferLength = len * sizeof(WCHAR);
3660  LeaveCriticalSection( &request->headers_section );
3661  return ERROR_SUCCESS;
3662  }
3663  break;
3664  case HTTP_QUERY_VERSION:
3665  if (request->version)
3666  {
3667  DWORD len = strlenW(request->version);
3668  if (len + 1 > *lpdwBufferLength/sizeof(WCHAR))
3669  {
3670  *lpdwBufferLength = (len + 1) * sizeof(WCHAR);
3671  LeaveCriticalSection( &request->headers_section );
3673  }
3674  if (lpBuffer)
3675  {
3676  memcpy(lpBuffer, request->version, (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;
3686  requested_index,request_only);
3687  break;
3688  case HTTP_QUERY_STATUS_CODE: {
3690 
3691  if(request_only)
3692  {
3693  LeaveCriticalSection( &request->headers_section );
3695  }
3696 
3697  if(requested_index)
3698  break;
3699 
3701  if(*lpdwBufferLength >= sizeof(DWORD))
3702  *(DWORD*)lpBuffer = request->status_code;
3703  else
3705  *lpdwBufferLength = sizeof(DWORD);
3706  }else {
3707  WCHAR buf[12];
3708  DWORD size;
3709  static const WCHAR formatW[] = {'%','u',0};
3710 
3711  size = sprintfW(buf, formatW, request->status_code) * sizeof(WCHAR);
3712 
3713  if(size <= *lpdwBufferLength) {
3714  memcpy(lpBuffer, buf, size+sizeof(WCHAR));
3715  }else {
3716  size += sizeof(WCHAR);
3718  }
3719 
3721  }
3722  LeaveCriticalSection( &request->headers_section );
3723  return res;
3724  }
3725  default:
3727 
3730  requested_index,request_only);
3731  }
3732 
3733  if (index >= 0)
3734  lphttpHdr = &request->custHeaders[index];
3735 
3736  /* Ensure header satisfies requested attributes */
3737  if (!lphttpHdr ||
3739  (~lphttpHdr->wFlags & HDR_ISREQUEST)))
3740  {
3741  LeaveCriticalSection( &request->headers_section );
3743  }
3744 
3745  /* coalesce value to requested type */
3747  {
3748  *(int *)lpBuffer = atoiW(lphttpHdr->lpszValue);
3749  TRACE(" returning number: %d\n", *(int *)lpBuffer);
3750  }
3752  {
3753  time_t tmpTime;
3754  struct tm tmpTM;
3755  SYSTEMTIME *STHook;
3756 
3757  tmpTime = ConvertTimeString(lphttpHdr->lpszValue);
3758 
3759  tmpTM = *gmtime(&tmpTime);
3760  STHook = (SYSTEMTIME *)lpBuffer;
3761  STHook->wDay = tmpTM.tm_mday;
3762  STHook->wHour = tmpTM.tm_hour;
3763  STHook->wMilliseconds = 0;
3764  STHook->wMinute = tmpTM.tm_min;
3765  STHook->wDayOfWeek = tmpTM.tm_wday;
3766  STHook->wMonth = tmpTM.tm_mon + 1;
3767  STHook->wSecond = tmpTM.tm_sec;
3768  STHook->wYear = 1900+tmpTM.tm_year;
3769 
3770  TRACE(" returning time: %04d/%02d/%02d - %d - %02d:%02d:%02d.%02d\n",
3771  STHook->wYear, STHook->wMonth, STHook->wDay, STHook->wDayOfWeek,
3772  STHook->wHour, STHook->wMinute, STHook->wSecond, STHook->wMilliseconds);
3773  }
3774  else if (lphttpHdr->lpszValue)
3775  {
3776  DWORD len = (strlenW(lphttpHdr->lpszValue) + 1) * sizeof(WCHAR);
3777 
3778  if (len > *lpdwBufferLength)
3779  {
3780  *lpdwBufferLength = len;
3781  LeaveCriticalSection( &request->headers_section );
3783  }
3784  if (lpBuffer)
3785  {
3786  memcpy(lpBuffer, lphttpHdr->lpszValue, len);
3787  TRACE("! returning string: %s\n", debugstr_w(lpBuffer));
3788  }
3789  *lpdwBufferLength = len - sizeof(WCHAR);
3790  }
3791  if (lpdwIndex) (*lpdwIndex)++;
3792 
3793  LeaveCriticalSection( &request->headers_section );
3794  return ERROR_SUCCESS;
3795 }
3796 
3797 /***********************************************************************
3798  * HttpQueryInfoW (WININET.@)
3799  *
3800  * Queries for information about an HTTP request
3801  *
3802  * RETURNS
3803  * TRUE on success
3804  * FALSE on failure
3805  *
3806  */
3809 {
3811  DWORD res;
3812 
3813  if (TRACE_ON(wininet)) {
3814 #define FE(x) { x, #x }
3815  static const wininet_flag_info query_flags[] = {
3829  FE(HTTP_QUERY_URI),
3864  FE(HTTP_QUERY_AGE),
3882  FE(HTTP_QUERY_VIA),
3885  };
3886  static const wininet_flag_info modifier_flags[] = {
3891  };
3892 #undef FE
3895  DWORD