?  search:
summary | log | graphiclog1 | graphiclog2 | commit | commitdiff | tree | refs | edit | fork
blame | history | raw | HEAD
push cc8bc80451cc24f4d7cf75168b569f0ebfe19547
[wine/hacks.git] / dlls / wininet / http.c
blobe1a7ddc76d19aa327347b1bb6844bbb3ec89e2c7
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 *
11 * Ulrich Czekalla
12 * David Hammerton
13 *
14 * This library is free software; you can redistribute it and/or
15 * modify it under the terms of the GNU Lesser General Public
16 * License as published by the Free Software Foundation; either
17 * version 2.1 of the License, or (at your option) any later version.
18 *
19 * This library is distributed in the hope that it will be useful,
20 * but WITHOUT ANY WARRANTY; without even the implied warranty of
21 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
22 * Lesser General Public License for more details.
23 *
24 * You should have received a copy of the GNU Lesser General Public
25 * License along with this library; if not, write to the Free Software
26 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
27 */
28
29 #include "config.h"
30 #include "wine/port.h"
31
32 #if defined(__MINGW32__) || defined (_MSC_VER)
33 #include <ws2tcpip.h>
34 #endif
35
36 #include <sys/types.h>
37 #ifdef HAVE_SYS_SOCKET_H
38 # include <sys/socket.h>
39 #endif
40 #ifdef HAVE_ARPA_INET_H
41 # include <arpa/inet.h>
42 #endif
43 #include <stdarg.h>
44 #include <stdio.h>
45 #include <stdlib.h>
46 #ifdef HAVE_UNISTD_H
47 # include <unistd.h>
48 #endif
49 #include <time.h>
50 #include <assert.h>
51 #ifdef HAVE_ZLIB
52 # include <zlib.h>
53 #endif
54
55 #include "windef.h"
56 #include "winbase.h"
57 #include "wininet.h"
58 #include "winerror.h"
59 #define NO_SHLWAPI_STREAM
60 #define NO_SHLWAPI_REG
61 #define NO_SHLWAPI_STRFCNS
62 #define NO_SHLWAPI_GDI
63 #include "shlwapi.h"
64 #include "sspi.h"
65 #include "wincrypt.h"
66
67 #include "internet.h"
68 #include "wine/debug.h"
69 #include "wine/exception.h"
70 #include "wine/unicode.h"
71
72 WINE_DEFAULT_DEBUG_CHANNEL(wininet);
73
74 static const WCHAR g_szHttp1_0[] = {'H','T','T','P','/','1','.','0',0};
75 static const WCHAR g_szHttp1_1[] = {'H','T','T','P','/','1','.','1',0};
76 static const WCHAR hostW[] = { 'H','o','s','t',0 };
77 static const WCHAR szAuthorization[] = { 'A','u','t','h','o','r','i','z','a','t','i','o','n',0 };
78 static const WCHAR szProxy_Authorization[] = { 'P','r','o','x','y','-','A','u','t','h','o','r','i','z','a','t','i','o','n',0 };
79 static const WCHAR szStatus[] = { 'S','t','a','t','u','s',0 };
80 static const WCHAR szKeepAlive[] = {'K','e','e','p','-','A','l','i','v','e',0};
81 static const WCHAR szGET[] = { 'G','E','T', 0 };
82 static const WCHAR szCrLf[] = {'\r','\n', 0};
83
84 static const WCHAR szAccept[] = { 'A','c','c','e','p','t',0 };
85 static const WCHAR szAccept_Charset[] = { 'A','c','c','e','p','t','-','C','h','a','r','s','e','t', 0 };
86 static const WCHAR szAccept_Encoding[] = { 'A','c','c','e','p','t','-','E','n','c','o','d','i','n','g',0 };
87 static const WCHAR szAccept_Language[] = { 'A','c','c','e','p','t','-','L','a','n','g','u','a','g','e',0 };
88 static const WCHAR szAccept_Ranges[] = { 'A','c','c','e','p','t','-','R','a','n','g','e','s',0 };
89 static const WCHAR szAge[] = { 'A','g','e',0 };
90 static const WCHAR szAllow[] = { 'A','l','l','o','w',0 };
91 static const WCHAR szCache_Control[] = { 'C','a','c','h','e','-','C','o','n','t','r','o','l',0 };
92 static const WCHAR szConnection[] = { 'C','o','n','n','e','c','t','i','o','n',0 };
93 static const WCHAR szContent_Base[] = { 'C','o','n','t','e','n','t','-','B','a','s','e',0 };
94 static const WCHAR szContent_Encoding[] = { 'C','o','n','t','e','n','t','-','E','n','c','o','d','i','n','g',0 };
95 static const WCHAR szContent_ID[] = { 'C','o','n','t','e','n','t','-','I','D',0 };
96 static const WCHAR szContent_Language[] = { 'C','o','n','t','e','n','t','-','L','a','n','g','u','a','g','e',0 };
97 static const WCHAR szContent_Length[] = { 'C','o','n','t','e','n','t','-','L','e','n','g','t','h',0 };
98 static const WCHAR szContent_Location[] = { 'C','o','n','t','e','n','t','-','L','o','c','a','t','i','o','n',0 };
99 static const WCHAR szContent_MD5[] = { 'C','o','n','t','e','n','t','-','M','D','5',0 };
100 static const WCHAR szContent_Range[] = { 'C','o','n','t','e','n','t','-','R','a','n','g','e',0 };
101 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 };
102 static const WCHAR szContent_Type[] = { 'C','o','n','t','e','n','t','-','T','y','p','e',0 };
103 static const WCHAR szCookie[] = { 'C','o','o','k','i','e',0 };
104 static const WCHAR szDate[] = { 'D','a','t','e',0 };
105 static const WCHAR szFrom[] = { 'F','r','o','m',0 };
106 static const WCHAR szETag[] = { 'E','T','a','g',0 };
107 static const WCHAR szExpect[] = { 'E','x','p','e','c','t',0 };
108 static const WCHAR szExpires[] = { 'E','x','p','i','r','e','s',0 };
109 static const WCHAR szIf_Match[] = { 'I','f','-','M','a','t','c','h',0 };
110 static const WCHAR szIf_Modified_Since[] = { 'I','f','-','M','o','d','i','f','i','e','d','-','S','i','n','c','e',0 };
111 static const WCHAR szIf_None_Match[] = { 'I','f','-','N','o','n','e','-','M','a','t','c','h',0 };
112 static const WCHAR szIf_Range[] = { 'I','f','-','R','a','n','g','e',0 };
113 static const WCHAR szIf_Unmodified_Since[] = { 'I','f','-','U','n','m','o','d','i','f','i','e','d','-','S','i','n','c','e',0 };
114 static const WCHAR szLast_Modified[] = { 'L','a','s','t','-','M','o','d','i','f','i','e','d',0 };
115 static const WCHAR szLocation[] = { 'L','o','c','a','t','i','o','n',0 };
116 static const WCHAR szMax_Forwards[] = { 'M','a','x','-','F','o','r','w','a','r','d','s',0 };
117 static const WCHAR szMime_Version[] = { 'M','i','m','e','-','V','e','r','s','i','o','n',0 };
118 static const WCHAR szPragma[] = { 'P','r','a','g','m','a',0 };
119 static const WCHAR szProxy_Authenticate[] = { 'P','r','o','x','y','-','A','u','t','h','e','n','t','i','c','a','t','e',0 };
120 static const WCHAR szProxy_Connection[] = { 'P','r','o','x','y','-','C','o','n','n','e','c','t','i','o','n',0 };
121 static const WCHAR szPublic[] = { 'P','u','b','l','i','c',0 };
122 static const WCHAR szRange[] = { 'R','a','n','g','e',0 };
123 static const WCHAR szReferer[] = { 'R','e','f','e','r','e','r',0 };
124 static const WCHAR szRetry_After[] = { 'R','e','t','r','y','-','A','f','t','e','r',0 };
125 static const WCHAR szServer[] = { 'S','e','r','v','e','r',0 };
126 static const WCHAR szSet_Cookie[] = { 'S','e','t','-','C','o','o','k','i','e',0 };
127 static const WCHAR szTransfer_Encoding[] = { 'T','r','a','n','s','f','e','r','-','E','n','c','o','d','i','n','g',0 };
128 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 };
129 static const WCHAR szUpgrade[] = { 'U','p','g','r','a','d','e',0 };
130 static const WCHAR szURI[] = { 'U','R','I',0 };
131 static const WCHAR szUser_Agent[] = { 'U','s','e','r','-','A','g','e','n','t',0 };
132 static const WCHAR szVary[] = { 'V','a','r','y',0 };
133 static const WCHAR szVia[] = { 'V','i','a',0 };
134 static const WCHAR szWarning[] = { 'W','a','r','n','i','n','g',0 };
135 static const WCHAR szWWW_Authenticate[] = { 'W','W','W','-','A','u','t','h','e','n','t','i','c','a','t','e',0 };
136
137 #define MAXHOSTNAME 100
138 #define MAX_FIELD_VALUE_LEN 256
139 #define MAX_FIELD_LEN 256
140
141 #define HTTP_REFERER szReferer
142 #define HTTP_ACCEPT szAccept
143 #define HTTP_USERAGENT szUser_Agent
144
145 #define HTTP_ADDHDR_FLAG_ADD 0x20000000
146 #define HTTP_ADDHDR_FLAG_ADD_IF_NEW 0x10000000
147 #define HTTP_ADDHDR_FLAG_COALESCE 0x40000000
148 #define HTTP_ADDHDR_FLAG_COALESCE_WITH_COMMA 0x40000000
149 #define HTTP_ADDHDR_FLAG_COALESCE_WITH_SEMICOLON 0x01000000
150 #define HTTP_ADDHDR_FLAG_REPLACE 0x80000000
151 #define HTTP_ADDHDR_FLAG_REQ 0x02000000
152
153 #define ARRAYSIZE(array) (sizeof(array)/sizeof((array)[0]))
154
155 struct HttpAuthInfo
156 {
157 LPWSTR scheme;
158 CredHandle cred;
159 CtxtHandle ctx;
160 TimeStamp exp;
161 ULONG attr;
162 ULONG max_token;
163 void *auth_data;
164 unsigned int auth_data_len;
165 BOOL finished; /* finished authenticating */
166 };
167
168
169 struct gzip_stream_t {
170 #ifdef HAVE_ZLIB
171 z_stream zstream;
172 #endif
173 BYTE buf[8192];
174 DWORD buf_size;
175 DWORD buf_pos;
176 BOOL end_of_data;
177 };
178
179 static BOOL HTTP_OpenConnection(LPWININETHTTPREQW lpwhr);
180 static BOOL HTTP_GetResponseHeaders(LPWININETHTTPREQW lpwhr, BOOL clear);
181 static BOOL HTTP_ProcessHeader(LPWININETHTTPREQW lpwhr, LPCWSTR field, LPCWSTR value, DWORD dwModifier);
182 static LPWSTR * HTTP_InterpretHttpHeader(LPCWSTR buffer);
183 static BOOL HTTP_InsertCustomHeader(LPWININETHTTPREQW lpwhr, LPHTTPHEADERW lpHdr);
184 static INT HTTP_GetCustomHeaderIndex(LPWININETHTTPREQW lpwhr, LPCWSTR lpszField, INT index, BOOL Request);
185 static BOOL HTTP_DeleteCustomHeader(LPWININETHTTPREQW lpwhr, DWORD index);
186 static LPWSTR HTTP_build_req( LPCWSTR *list, int len );
187 static BOOL HTTP_HttpQueryInfoW(LPWININETHTTPREQW, DWORD, LPVOID, LPDWORD, LPDWORD);
188 static LPWSTR HTTP_GetRedirectURL(LPWININETHTTPREQW lpwhr, LPCWSTR lpszUrl);
189 static BOOL HTTP_HandleRedirect(LPWININETHTTPREQW lpwhr, LPCWSTR lpszUrl);
190 static UINT HTTP_DecodeBase64(LPCWSTR base64, LPSTR bin);
191 static BOOL HTTP_VerifyValidHeader(LPWININETHTTPREQW lpwhr, LPCWSTR field);
192 static void HTTP_DrainContent(WININETHTTPREQW *req);
193 static BOOL HTTP_FinishedReading(LPWININETHTTPREQW lpwhr);
194
195 static LPHTTPHEADERW HTTP_GetHeader(LPWININETHTTPREQW req, LPCWSTR head)
196 {
197 int HeaderIndex = 0;
198 HeaderIndex = HTTP_GetCustomHeaderIndex(req, head, 0, TRUE);
199 if (HeaderIndex == -1)
200 return NULL;
201 else
202 return &req->pCustHeaders[HeaderIndex];
203 }
204
205 #ifdef HAVE_ZLIB
206
207 static voidpf wininet_zalloc(voidpf opaque, uInt items, uInt size)
208 {
209 return HeapAlloc(GetProcessHeap(), 0, items*size);
210 }
211
212 static void wininet_zfree(voidpf opaque, voidpf address)
213 {
214 HeapFree(GetProcessHeap(), 0, address);
215 }
216
217 static void init_gzip_stream(WININETHTTPREQW *req)
218 {
219 gzip_stream_t *gzip_stream;
220 int zres;
221
222 gzip_stream = HeapAlloc(GetProcessHeap(), 0, sizeof(gzip_stream_t));
223 gzip_stream->zstream.zalloc = wininet_zalloc;
224 gzip_stream->zstream.zfree = wininet_zfree;
225 gzip_stream->zstream.opaque = NULL;
226 gzip_stream->zstream.next_in = NULL;
227 gzip_stream->zstream.avail_in = 0;
228 gzip_stream->zstream.next_out = NULL;
229 gzip_stream->zstream.avail_out = 0;
230 gzip_stream->buf_pos = 0;
231 gzip_stream->buf_size = 0;
232 gzip_stream->end_of_data = FALSE;
233
234 zres = inflateInit2(&gzip_stream->zstream, 0x1f);
235 if(zres != Z_OK) {
236 ERR("inflateInit failed: %d\n", zres);
237 HeapFree(GetProcessHeap(), 0, gzip_stream);
238 return;
239 }
240
241 req->gzip_stream = gzip_stream;
242 }
243
244 #else
245
246 static void init_gzip_stream(WININETHTTPREQW *req)
247 {
248 ERR("gzip stream not supported, missing zlib.\n");
249 }
250
251 #endif
252
253 /* set the request content length based on the headers */
254 static DWORD set_content_length( LPWININETHTTPREQW lpwhr )
255 {
256 static const WCHAR szChunked[] = {'c','h','u','n','k','e','d',0};
257 WCHAR encoding[20];
258 DWORD size;
259
260 size = sizeof(lpwhr->dwContentLength);
261 if (!HTTP_HttpQueryInfoW(lpwhr, HTTP_QUERY_FLAG_NUMBER|HTTP_QUERY_CONTENT_LENGTH,
262 &lpwhr->dwContentLength, &size, NULL))
263 lpwhr->dwContentLength = ~0u;
264
265 size = sizeof(encoding);
266 if (HTTP_HttpQueryInfoW(lpwhr, HTTP_QUERY_TRANSFER_ENCODING, encoding, &size, NULL) &&
267 !strcmpiW(encoding, szChunked))
268 {
269 lpwhr->dwContentLength = ~0u;
270 lpwhr->read_chunked = TRUE;
271 }
272
273 if(lpwhr->decoding) {
274 int encoding_idx;
275
276 static const WCHAR gzipW[] = {'g','z','i','p',0};
277
278 encoding_idx = HTTP_GetCustomHeaderIndex(lpwhr, szContent_Encoding, 0, FALSE);
279 if(encoding_idx != -1 && !strcmpiW(lpwhr->pCustHeaders[encoding_idx].lpszValue, gzipW))
280 init_gzip_stream(lpwhr);
281 }
282
283 return lpwhr->dwContentLength;
284 }
285
286 /***********************************************************************
287 * HTTP_Tokenize (internal)
288 *
289 * Tokenize a string, allocating memory for the tokens.
290 */
291 static LPWSTR * HTTP_Tokenize(LPCWSTR string, LPCWSTR token_string)
292 {
293 LPWSTR * token_array;
294 int tokens = 0;
295 int i;
296 LPCWSTR next_token;
297
298 if (string)
299 {
300 /* empty string has no tokens */
301 if (*string)
302 tokens++;
303 /* count tokens */
304 for (i = 0; string[i]; i++)
305 {
306 if (!strncmpW(string+i, token_string, strlenW(token_string)))
307 {
308 DWORD j;
309 tokens++;
310 /* we want to skip over separators, but not the null terminator */
311 for (j = 0; j < strlenW(token_string) - 1; j++)
312 if (!string[i+j])
313 break;
314 i += j;
315 }
316 }
317 }
318
319 /* add 1 for terminating NULL */
320 token_array = HeapAlloc(GetProcessHeap(), 0, (tokens+1) * sizeof(*token_array));
321 token_array[tokens] = NULL;
322 if (!tokens)
323 return token_array;
324 for (i = 0; i < tokens; i++)
325 {
326 int len;
327 next_token = strstrW(string, token_string);
328 if (!next_token) next_token = string+strlenW(string);
329 len = next_token - string;
330 token_array[i] = HeapAlloc(GetProcessHeap(), 0, (len+1)*sizeof(WCHAR));
331 memcpy(token_array[i], string, len*sizeof(WCHAR));
332 token_array[i][len] = '\0';
333 string = next_token+strlenW(token_string);
334 }
335 return token_array;
336 }
337
338 /***********************************************************************
339 * HTTP_FreeTokens (internal)
340 *
341 * Frees memory returned from HTTP_Tokenize.
342 */
343 static void HTTP_FreeTokens(LPWSTR * token_array)
344 {
345 int i;
346 for (i = 0; token_array[i]; i++)
347 HeapFree(GetProcessHeap(), 0, token_array[i]);
348 HeapFree(GetProcessHeap(), 0, token_array);
349 }
350
351 /* **********************************************************************
352 *
353 * Helper functions for the HttpSendRequest(Ex) functions
354 *
355 */
356 static void AsyncHttpSendRequestProc(WORKREQUEST *workRequest)
357 {
358 struct WORKREQ_HTTPSENDREQUESTW const *req = &workRequest->u.HttpSendRequestW;
359 LPWININETHTTPREQW lpwhr = (LPWININETHTTPREQW) workRequest->hdr;
360
361 TRACE("%p\n", lpwhr);
362
363 HTTP_HttpSendRequestW(lpwhr, req->lpszHeader,
364 req->dwHeaderLength, req->lpOptional, req->dwOptionalLength,
365 req->dwContentLength, req->bEndRequest);
366
367 HeapFree(GetProcessHeap(), 0, req->lpszHeader);
368 }
369
370 static void HTTP_FixURL( LPWININETHTTPREQW lpwhr)
371 {
372 static const WCHAR szSlash[] = { '/',0 };
373 static const WCHAR szHttp[] = { 'h','t','t','p',':','/','/', 0 };
374
375 /* If we don't have a path we set it to root */
376 if (NULL == lpwhr->lpszPath)
377 lpwhr->lpszPath = WININET_strdupW(szSlash);
378 else /* remove \r and \n*/
379 {
380 int nLen = strlenW(lpwhr->lpszPath);
381 while ((nLen >0 ) && ((lpwhr->lpszPath[nLen-1] == '\r')||(lpwhr->lpszPath[nLen-1] == '\n')))
382 {
383 nLen--;
384 lpwhr->lpszPath[nLen]='\0';
385 }
386 /* Replace '\' with '/' */
387 while (nLen>0) {
388 nLen--;
389 if (lpwhr->lpszPath[nLen] == '\\') lpwhr->lpszPath[nLen]='/';
390 }
391 }
392
393 if(CSTR_EQUAL != CompareStringW( LOCALE_SYSTEM_DEFAULT, NORM_IGNORECASE,
394 lpwhr->lpszPath, strlenW(lpwhr->lpszPath), szHttp, strlenW(szHttp) )
395 && lpwhr->lpszPath[0] != '/') /* not an absolute path ?? --> fix it !! */
396 {
397 WCHAR *fixurl = HeapAlloc(GetProcessHeap(), 0,
398 (strlenW(lpwhr->lpszPath) + 2)*sizeof(WCHAR));
399 *fixurl = '/';
400 strcpyW(fixurl + 1, lpwhr->lpszPath);
401 HeapFree( GetProcessHeap(), 0, lpwhr->lpszPath );
402 lpwhr->lpszPath = fixurl;
403 }
404 }
405
406 static LPWSTR HTTP_BuildHeaderRequestString( LPWININETHTTPREQW lpwhr, LPCWSTR verb, LPCWSTR path, LPCWSTR version )
407 {
408 LPWSTR requestString;
409 DWORD len, n;
410 LPCWSTR *req;
411 UINT i;
412 LPWSTR p;
413
414 static const WCHAR szSpace[] = { ' ',0 };
415 static const WCHAR szColon[] = { ':',' ',0 };
416 static const WCHAR sztwocrlf[] = {'\r','\n','\r','\n', 0};
417
418 /* allocate space for an array of all the string pointers to be added */
419 len = (lpwhr->nCustHeaders)*4 + 10;
420 req = HeapAlloc( GetProcessHeap(), 0, len*sizeof(LPCWSTR) );
421
422 /* add the verb, path and HTTP version string */
423 n = 0;
424 req[n++] = verb;
425 req[n++] = szSpace;
426 req[n++] = path;
427 req[n++] = szSpace;
428 req[n++] = version;
429
430 /* Append custom request headers */
431 for (i = 0; i < lpwhr->nCustHeaders; i++)
432 {
433 if (lpwhr->pCustHeaders[i].wFlags & HDR_ISREQUEST)
434 {
435 req[n++] = szCrLf;
436 req[n++] = lpwhr->pCustHeaders[i].lpszField;
437 req[n++] = szColon;
438 req[n++] = lpwhr->pCustHeaders[i].lpszValue;
439
440 TRACE("Adding custom header %s (%s)\n",
441 debugstr_w(lpwhr->pCustHeaders[i].lpszField),
442 debugstr_w(lpwhr->pCustHeaders[i].lpszValue));
443 }
444 }
445
446 if( n >= len )
447 ERR("oops. buffer overrun\n");
448
449 req[n] = NULL;
450 requestString = HTTP_build_req( req, 4 );
451 HeapFree( GetProcessHeap(), 0, req );
452
453 /*
454 * Set (header) termination string for request
455 * Make sure there's exactly two new lines at the end of the request
456 */
457 p = &requestString[strlenW(requestString)-1];
458 while ( (*p == '\n') || (*p == '\r') )
459 p--;
460 strcpyW( p+1, sztwocrlf );
461
462 return requestString;
463 }
464
465 static void HTTP_ProcessCookies( LPWININETHTTPREQW lpwhr )
466 {
467 int HeaderIndex;
468 int numCookies = 0;
469 LPHTTPHEADERW setCookieHeader;
470
471 while((HeaderIndex = HTTP_GetCustomHeaderIndex(lpwhr, szSet_Cookie, numCookies, FALSE)) != -1)
472 {
473 setCookieHeader = &lpwhr->pCustHeaders[HeaderIndex];
474
475 if (!(lpwhr->hdr.dwFlags & INTERNET_FLAG_NO_COOKIES) && setCookieHeader->lpszValue)
476 {
477 int len;
478 static const WCHAR szFmt[] = { 'h','t','t','p',':','/','/','%','s','%','s',0};
479 LPWSTR buf_url;
480 LPHTTPHEADERW Host;
481
482 Host = HTTP_GetHeader(lpwhr, hostW);
483 len = lstrlenW(Host->lpszValue) + 9 + lstrlenW(lpwhr->lpszPath);
484 buf_url = HeapAlloc(GetProcessHeap(), 0, len*sizeof(WCHAR));
485 sprintfW(buf_url, szFmt, Host->lpszValue, lpwhr->lpszPath);
486 InternetSetCookieW(buf_url, NULL, setCookieHeader->lpszValue);
487
488 HeapFree(GetProcessHeap(), 0, buf_url);
489 }
490 numCookies++;
491 }
492 }
493
494 static inline BOOL is_basic_auth_value( LPCWSTR pszAuthValue )
495 {
496 static const WCHAR szBasic[] = {'B','a','s','i','c'}; /* Note: not nul-terminated */
497 return !strncmpiW(pszAuthValue, szBasic, ARRAYSIZE(szBasic)) &&
498 ((pszAuthValue[ARRAYSIZE(szBasic)] == ' ') || !pszAuthValue[ARRAYSIZE(szBasic)]);
499 }
500
501 static BOOL HTTP_DoAuthorization( LPWININETHTTPREQW lpwhr, LPCWSTR pszAuthValue,
502 struct HttpAuthInfo **ppAuthInfo,
503 LPWSTR domain_and_username, LPWSTR password )
504 {
505 SECURITY_STATUS sec_status;
506 struct HttpAuthInfo *pAuthInfo = *ppAuthInfo;
507 BOOL first = FALSE;
508
509 TRACE("%s\n", debugstr_w(pszAuthValue));
510
511 if (!pAuthInfo)
512 {
513 TimeStamp exp;
514
515 first = TRUE;
516 pAuthInfo = HeapAlloc(GetProcessHeap(), 0, sizeof(*pAuthInfo));
517 if (!pAuthInfo)
518 return FALSE;
519
520 SecInvalidateHandle(&pAuthInfo->cred);
521 SecInvalidateHandle(&pAuthInfo->ctx);
522 memset(&pAuthInfo->exp, 0, sizeof(pAuthInfo->exp));
523 pAuthInfo->attr = 0;
524 pAuthInfo->auth_data = NULL;
525 pAuthInfo->auth_data_len = 0;
526 pAuthInfo->finished = FALSE;
527
528 if (is_basic_auth_value(pszAuthValue))
529 {
530 static const WCHAR szBasic[] = {'B','a','s','i','c',0};
531 pAuthInfo->scheme = WININET_strdupW(szBasic);
532 if (!pAuthInfo->scheme)
533 {
534 HeapFree(GetProcessHeap(), 0, pAuthInfo);
535 return FALSE;
536 }
537 }
538 else
539 {
540 PVOID pAuthData;
541 SEC_WINNT_AUTH_IDENTITY_W nt_auth_identity;
542
543 pAuthInfo->scheme = WININET_strdupW(pszAuthValue);
544 if (!pAuthInfo->scheme)
545 {
546 HeapFree(GetProcessHeap(), 0, pAuthInfo);
547 return FALSE;
548 }
549
550 if (domain_and_username)
551 {
552 WCHAR *user = strchrW(domain_and_username, '\\');
553 WCHAR *domain = domain_and_username;
554
555 /* FIXME: make sure scheme accepts SEC_WINNT_AUTH_IDENTITY before calling AcquireCredentialsHandle */
556
557 pAuthData = &nt_auth_identity;
558
559 if (user) user++;
560 else
561 {
562 user = domain_and_username;
563 domain = NULL;
564 }
565
566 nt_auth_identity.Flags = SEC_WINNT_AUTH_IDENTITY_UNICODE;
567 nt_auth_identity.User = user;
568 nt_auth_identity.UserLength = strlenW(nt_auth_identity.User);
569 nt_auth_identity.Domain = domain;
570 nt_auth_identity.DomainLength = domain ? user - domain - 1 : 0;
571 nt_auth_identity.Password = password;
572 nt_auth_identity.PasswordLength = strlenW(nt_auth_identity.Password);
573 }
574 else
575 /* use default credentials */
576 pAuthData = NULL;
577
578 sec_status = AcquireCredentialsHandleW(NULL, pAuthInfo->scheme,
579 SECPKG_CRED_OUTBOUND, NULL,
580 pAuthData, NULL,
581 NULL, &pAuthInfo->cred,
582 &exp);
583 if (sec_status == SEC_E_OK)
584 {
585 PSecPkgInfoW sec_pkg_info;
586 sec_status = QuerySecurityPackageInfoW(pAuthInfo->scheme, &sec_pkg_info);
587 if (sec_status == SEC_E_OK)
588 {
589 pAuthInfo->max_token = sec_pkg_info->cbMaxToken;
590 FreeContextBuffer(sec_pkg_info);
591 }
592 }
593 if (sec_status != SEC_E_OK)
594 {
595 WARN("AcquireCredentialsHandleW for scheme %s failed with error 0x%08x\n",
596 debugstr_w(pAuthInfo->scheme), sec_status);
597 HeapFree(GetProcessHeap(), 0, pAuthInfo->scheme);
598 HeapFree(GetProcessHeap(), 0, pAuthInfo);
599 return FALSE;
600 }
601 }
602 *ppAuthInfo = pAuthInfo;
603 }
604 else if (pAuthInfo->finished)
605 return FALSE;
606
607 if ((strlenW(pszAuthValue) < strlenW(pAuthInfo->scheme)) ||
608 strncmpiW(pszAuthValue, pAuthInfo->scheme, strlenW(pAuthInfo->scheme)))
609 {
610 ERR("authentication scheme changed from %s to %s\n",
611 debugstr_w(pAuthInfo->scheme), debugstr_w(pszAuthValue));
612 return FALSE;
613 }
614
615 if (is_basic_auth_value(pszAuthValue))
616 {
617 int userlen;
618 int passlen;
619 char *auth_data;
620
621 TRACE("basic authentication\n");
622
623 /* we don't cache credentials for basic authentication, so we can't
624 * retrieve them if the application didn't pass us any credentials */
625 if (!domain_and_username) return FALSE;
626
627 userlen = WideCharToMultiByte(CP_UTF8, 0, domain_and_username, lstrlenW(domain_and_username), NULL, 0, NULL, NULL);
628 passlen = WideCharToMultiByte(CP_UTF8, 0, password, lstrlenW(password), NULL, 0, NULL, NULL);
629
630 /* length includes a nul terminator, which will be re-used for the ':' */
631 auth_data = HeapAlloc(GetProcessHeap(), 0, userlen + 1 + passlen);
632 if (!auth_data)
633 return FALSE;
634
635 WideCharToMultiByte(CP_UTF8, 0, domain_and_username, -1, auth_data, userlen, NULL, NULL);
636 auth_data[userlen] = ':';
637 WideCharToMultiByte(CP_UTF8, 0, password, -1, &auth_data[userlen+1], passlen, NULL, NULL);
638
639 pAuthInfo->auth_data = auth_data;
640 pAuthInfo->auth_data_len = userlen + 1 + passlen;
641 pAuthInfo->finished = TRUE;
642
643 return TRUE;
644 }
645 else
646 {
647 LPCWSTR pszAuthData;
648 SecBufferDesc out_desc, in_desc;
649 SecBuffer out, in;
650 unsigned char *buffer;
651 ULONG context_req = ISC_REQ_CONNECTION | ISC_REQ_USE_DCE_STYLE |
652 ISC_REQ_MUTUAL_AUTH | ISC_REQ_DELEGATE;
653
654 in.BufferType = SECBUFFER_TOKEN;
655 in.cbBuffer = 0;
656 in.pvBuffer = NULL;
657
658 in_desc.ulVersion = 0;
659 in_desc.cBuffers = 1;
660 in_desc.pBuffers = &in;
661
662 pszAuthData = pszAuthValue + strlenW(pAuthInfo->scheme);
663 if (*pszAuthData == ' ')
664 {
665 pszAuthData++;
666 in.cbBuffer = HTTP_DecodeBase64(pszAuthData, NULL);
667 in.pvBuffer = HeapAlloc(GetProcessHeap(), 0, in.cbBuffer);
668 HTTP_DecodeBase64(pszAuthData, in.pvBuffer);
669 }
670
671 buffer = HeapAlloc(GetProcessHeap(), 0, pAuthInfo->max_token);
672
673 out.BufferType = SECBUFFER_TOKEN;
674 out.cbBuffer = pAuthInfo->max_token;
675 out.pvBuffer = buffer;
676
677 out_desc.ulVersion = 0;
678 out_desc.cBuffers = 1;
679 out_desc.pBuffers = &out;
680
681 sec_status = InitializeSecurityContextW(first ? &pAuthInfo->cred : NULL,
682 first ? NULL : &pAuthInfo->ctx,
683 first ? lpwhr->lpHttpSession->lpszServerName : NULL,
684 context_req, 0, SECURITY_NETWORK_DREP,
685 in.pvBuffer ? &in_desc : NULL,
686 0, &pAuthInfo->ctx, &out_desc,
687 &pAuthInfo->attr, &pAuthInfo->exp);
688 if (sec_status == SEC_E_OK)
689 {
690 pAuthInfo->finished = TRUE;
691 pAuthInfo->auth_data = out.pvBuffer;
692 pAuthInfo->auth_data_len = out.cbBuffer;
693 TRACE("sending last auth packet\n");
694 }
695 else if (sec_status == SEC_I_CONTINUE_NEEDED)
696 {
697 pAuthInfo->auth_data = out.pvBuffer;
698 pAuthInfo->auth_data_len = out.cbBuffer;
699 TRACE("sending next auth packet\n");
700 }
701 else
702 {
703 ERR("InitializeSecurityContextW returned error 0x%08x\n", sec_status);
704 pAuthInfo->finished = TRUE;
705 HeapFree(GetProcessHeap(), 0, out.pvBuffer);
706 return FALSE;
707 }
708 }
709
710 return TRUE;
711 }
712
713 /***********************************************************************
714 * HTTP_HttpAddRequestHeadersW (internal)
715 */
716 static BOOL HTTP_HttpAddRequestHeadersW(LPWININETHTTPREQW lpwhr,
717 LPCWSTR lpszHeader, DWORD dwHeaderLength, DWORD dwModifier)
718 {
719 LPWSTR lpszStart;
720 LPWSTR lpszEnd;
721 LPWSTR buffer;
722 BOOL bSuccess = FALSE;
723 DWORD len;
724
725 TRACE("copying header: %s\n", debugstr_wn(lpszHeader, dwHeaderLength));
726
727 if( dwHeaderLength == ~0U )
728 len = strlenW(lpszHeader);
729 else
730 len = dwHeaderLength;
731 buffer = HeapAlloc( GetProcessHeap(), 0, sizeof(WCHAR)*(len+1) );
732 lstrcpynW( buffer, lpszHeader, len + 1);
733
734 lpszStart = buffer;
735
736 do
737 {
738 LPWSTR * pFieldAndValue;
739
740 lpszEnd = lpszStart;
741
742 while (*lpszEnd != '\0')
743 {
744 if (*lpszEnd == '\r' || *lpszEnd == '\n')
745 break;
746 lpszEnd++;
747 }
748
749 if (*lpszStart == '\0')
750 break;
751
752 if (*lpszEnd == '\r' || *lpszEnd == '\n')
753 {
754 *lpszEnd = '\0';
755 lpszEnd++; /* Jump over newline */
756 }
757 TRACE("interpreting header %s\n", debugstr_w(lpszStart));
758 if (*lpszStart == '\0')
759 {
760 /* Skip 0-length headers */
761 lpszStart = lpszEnd;
762 bSuccess = TRUE;
763 continue;
764 }
765 pFieldAndValue = HTTP_InterpretHttpHeader(lpszStart);
766 if (pFieldAndValue)
767 {
768 bSuccess = HTTP_VerifyValidHeader(lpwhr, pFieldAndValue[0]);
769 if (bSuccess)
770 bSuccess = HTTP_ProcessHeader(lpwhr, pFieldAndValue[0],
771 pFieldAndValue[1], dwModifier | HTTP_ADDHDR_FLAG_REQ);
772 HTTP_FreeTokens(pFieldAndValue);
773 }
774
775 lpszStart = lpszEnd;
776 } while (bSuccess);
777
778 HeapFree(GetProcessHeap(), 0, buffer);
779
780 return bSuccess;
781 }
782
783 /***********************************************************************
784 * HttpAddRequestHeadersW (WININET.@)
785 *
786 * Adds one or more HTTP header to the request handler
787 *
788 * NOTE
789 * On Windows if dwHeaderLength includes the trailing '\0', then
790 * HttpAddRequestHeadersW() adds it too. However this results in an
791 * invalid Http header which is rejected by some servers so we probably
792 * don't need to match Windows on that point.
793 *
794 * RETURNS
795 * TRUE on success
796 * FALSE on failure
797 *
798 */
799 BOOL WINAPI HttpAddRequestHeadersW(HINTERNET hHttpRequest,
800 LPCWSTR lpszHeader, DWORD dwHeaderLength, DWORD dwModifier)
801 {
802 BOOL bSuccess = FALSE;
803 LPWININETHTTPREQW lpwhr;
804
805 TRACE("%p, %s, %i, %i\n", hHttpRequest, debugstr_wn(lpszHeader, dwHeaderLength), dwHeaderLength, dwModifier);
806
807 if (!lpszHeader)
808 return TRUE;
809
810 lpwhr = (LPWININETHTTPREQW) WININET_GetObject( hHttpRequest );
811 if (NULL == lpwhr || lpwhr->hdr.htype != WH_HHTTPREQ)
812 {
813 INTERNET_SetLastError(ERROR_INTERNET_INCORRECT_HANDLE_TYPE);
814 goto lend;
815 }
816 bSuccess = HTTP_HttpAddRequestHeadersW( lpwhr, lpszHeader, dwHeaderLength, dwModifier );
817 lend:
818 if( lpwhr )
819 WININET_Release( &lpwhr->hdr );
820
821 return bSuccess;
822 }
823
824 /***********************************************************************
825 * HttpAddRequestHeadersA (WININET.@)
826 *
827 * Adds one or more HTTP header to the request handler
828 *
829 * RETURNS
830 * TRUE on success
831 * FALSE on failure
832 *
833 */
834 BOOL WINAPI HttpAddRequestHeadersA(HINTERNET hHttpRequest,
835 LPCSTR lpszHeader, DWORD dwHeaderLength, DWORD dwModifier)
836 {
837 DWORD len;
838 LPWSTR hdr;
839 BOOL r;
840
841 TRACE("%p, %s, %i, %i\n", hHttpRequest, debugstr_an(lpszHeader, dwHeaderLength), dwHeaderLength, dwModifier);
842
843 len = MultiByteToWideChar( CP_ACP, 0, lpszHeader, dwHeaderLength, NULL, 0 );
844 hdr = HeapAlloc( GetProcessHeap(), 0, len*sizeof(WCHAR) );
845 MultiByteToWideChar( CP_ACP, 0, lpszHeader, dwHeaderLength, hdr, len );
846 if( dwHeaderLength != ~0U )
847 dwHeaderLength = len;
848
849 r = HttpAddRequestHeadersW( hHttpRequest, hdr, dwHeaderLength, dwModifier );
850
851 HeapFree( GetProcessHeap(), 0, hdr );
852
853 return r;
854 }
855
856 /***********************************************************************
857 * HttpEndRequestA (WININET.@)
858 *
859 * Ends an HTTP request that was started by HttpSendRequestEx
860 *
861 * RETURNS
862 * TRUE if successful
863 * FALSE on failure
864 *
865 */
866 BOOL WINAPI HttpEndRequestA(HINTERNET hRequest,
867 LPINTERNET_BUFFERSA lpBuffersOut, DWORD dwFlags, DWORD_PTR dwContext)
868 {
869 TRACE("(%p, %p, %08x, %08lx)\n", hRequest, lpBuffersOut, dwFlags, dwContext);
870
871 if (lpBuffersOut)
872 {
873 INTERNET_SetLastError(ERROR_INVALID_PARAMETER);
874 return FALSE;
875 }
876
877 return HttpEndRequestW(hRequest, NULL, dwFlags, dwContext);
878 }
879
880 static BOOL HTTP_HttpEndRequestW(LPWININETHTTPREQW lpwhr, DWORD dwFlags, DWORD_PTR dwContext)
881 {
882 BOOL rc = FALSE;
883 INT responseLen;
884 DWORD dwBufferSize;
885 INTERNET_ASYNC_RESULT iar;
886
887 INTERNET_SendCallback(&lpwhr->hdr, lpwhr->hdr.dwContext,
888 INTERNET_STATUS_RECEIVING_RESPONSE, NULL, 0);
889
890 responseLen = HTTP_GetResponseHeaders(lpwhr, TRUE);
891 if (responseLen)
892 rc = TRUE;
893
894 INTERNET_SendCallback(&lpwhr->hdr, lpwhr->hdr.dwContext,
895 INTERNET_STATUS_RESPONSE_RECEIVED, &responseLen, sizeof(DWORD));
896
897 /* process cookies here. Is this right? */
898 HTTP_ProcessCookies(lpwhr);
899
900 if (!set_content_length( lpwhr )) HTTP_FinishedReading(lpwhr);
901
902 if (!(lpwhr->hdr.dwFlags & INTERNET_FLAG_NO_AUTO_REDIRECT))
903 {
904 DWORD dwCode,dwCodeLength = sizeof(DWORD);
905 if (HTTP_HttpQueryInfoW(lpwhr, HTTP_QUERY_FLAG_NUMBER|HTTP_QUERY_STATUS_CODE, &dwCode, &dwCodeLength, NULL) &&
906 (dwCode == 302 || dwCode == 301 || dwCode == 303))
907 {
908 WCHAR *new_url, szNewLocation[INTERNET_MAX_URL_LENGTH];
909 dwBufferSize=sizeof(szNewLocation);
910 if (HTTP_HttpQueryInfoW(lpwhr, HTTP_QUERY_LOCATION, szNewLocation, &dwBufferSize, NULL))
911 {
912 /* redirects are always GETs */
913 HeapFree(GetProcessHeap(), 0, lpwhr->lpszVerb);
914 lpwhr->lpszVerb = WININET_strdupW(szGET);
915 HTTP_DrainContent(lpwhr);
916 if ((new_url = HTTP_GetRedirectURL( lpwhr, szNewLocation )))
917 {
918 INTERNET_SendCallback(&lpwhr->hdr, lpwhr->hdr.dwContext, INTERNET_STATUS_REDIRECT,
919 new_url, (strlenW(new_url) + 1) * sizeof(WCHAR));
920 rc = HTTP_HandleRedirect(lpwhr, new_url);
921 if (rc)
922 rc = HTTP_HttpSendRequestW(lpwhr, NULL, 0, NULL, 0, 0, TRUE);
923 HeapFree( GetProcessHeap(), 0, new_url );
924 }
925 }
926 }
927 }
928
929 iar.dwResult = (DWORD_PTR)lpwhr->hdr.hInternet;
930 iar.dwError = rc ? 0 : INTERNET_GetLastError();
931
932 INTERNET_SendCallback(&lpwhr->hdr, lpwhr->hdr.dwContext,
933 INTERNET_STATUS_REQUEST_COMPLETE, &iar,
934 sizeof(INTERNET_ASYNC_RESULT));
935 return rc;
936 }
937
938 static void AsyncHttpEndRequestProc(WORKREQUEST *work)
939 {
940 struct WORKREQ_HTTPENDREQUESTW const *req = &work->u.HttpEndRequestW;
941 LPWININETHTTPREQW lpwhr = (LPWININETHTTPREQW)work->hdr;
942
943 TRACE("%p\n", lpwhr);
944
945 HTTP_HttpEndRequestW(lpwhr, req->dwFlags, req->dwContext);
946 }
947
948 /***********************************************************************
949 * HttpEndRequestW (WININET.@)
950 *
951 * Ends an HTTP request that was started by HttpSendRequestEx
952 *
953 * RETURNS
954 * TRUE if successful
955 * FALSE on failure
956 *
957 */
958 BOOL WINAPI HttpEndRequestW(HINTERNET hRequest,
959 LPINTERNET_BUFFERSW lpBuffersOut, DWORD dwFlags, DWORD_PTR dwContext)
960 {
961 BOOL rc = FALSE;
962 LPWININETHTTPREQW lpwhr;
963
964 TRACE("-->\n");
965
966 if (lpBuffersOut)
967 {
968 INTERNET_SetLastError(ERROR_INVALID_PARAMETER);
969 return FALSE;
970 }
971
972 lpwhr = (LPWININETHTTPREQW) WININET_GetObject( hRequest );
973
974 if (NULL == lpwhr || lpwhr->hdr.htype != WH_HHTTPREQ)
975 {
976 INTERNET_SetLastError(ERROR_INTERNET_INCORRECT_HANDLE_TYPE);
977 if (lpwhr)
978 WININET_Release( &lpwhr->hdr );
979 return FALSE;
980 }
981 lpwhr->hdr.dwFlags |= dwFlags;
982
983 if (lpwhr->lpHttpSession->lpAppInfo->hdr.dwFlags & INTERNET_FLAG_ASYNC)
984 {
985 WORKREQUEST work;
986 struct WORKREQ_HTTPENDREQUESTW *request;
987
988 work.asyncproc = AsyncHttpEndRequestProc;
989 work.hdr = WININET_AddRef( &lpwhr->hdr );
990
991 request = &work.u.HttpEndRequestW;
992 request->dwFlags = dwFlags;
993 request->dwContext = dwContext;
994
995 INTERNET_AsyncCall(&work);
996 INTERNET_SetLastError(ERROR_IO_PENDING);
997 }
998 else
999 rc = HTTP_HttpEndRequestW(lpwhr, dwFlags, dwContext);
1000
1001 WININET_Release( &lpwhr->hdr );
1002 TRACE("%i <--\n",rc);
1003 return rc;
1004 }
1005
1006 /***********************************************************************
1007 * HttpOpenRequestW (WININET.@)
1008 *
1009 * Open a HTTP request handle
1010 *
1011 * RETURNS
1012 * HINTERNET a HTTP request handle on success
1013 * NULL on failure
1014 *
1015 */
1016 HINTERNET WINAPI HttpOpenRequestW(HINTERNET hHttpSession,
1017 LPCWSTR lpszVerb, LPCWSTR lpszObjectName, LPCWSTR lpszVersion,
1018 LPCWSTR lpszReferrer , LPCWSTR *lpszAcceptTypes,
1019 DWORD dwFlags, DWORD_PTR dwContext)
1020 {
1021 LPWININETHTTPSESSIONW lpwhs;
1022 HINTERNET handle = NULL;
1023
1024 TRACE("(%p, %s, %s, %s, %s, %p, %08x, %08lx)\n", hHttpSession,
1025 debugstr_w(lpszVerb), debugstr_w(lpszObjectName),
1026 debugstr_w(lpszVersion), debugstr_w(lpszReferrer), lpszAcceptTypes,
1027 dwFlags, dwContext);
1028 if(lpszAcceptTypes!=NULL)
1029 {
1030 int i;
1031 for(i=0;lpszAcceptTypes[i]!=NULL;i++)
1032 TRACE("\taccept type: %s\n",debugstr_w(lpszAcceptTypes[i]));
1033 }
1034
1035 lpwhs = (LPWININETHTTPSESSIONW) WININET_GetObject( hHttpSession );
1036 if (NULL == lpwhs || lpwhs->hdr.htype != WH_HHTTPSESSION)
1037 {
1038 INTERNET_SetLastError(ERROR_INTERNET_INCORRECT_HANDLE_TYPE);
1039 goto lend;
1040 }
1041
1042 /*
1043 * My tests seem to show that the windows version does not
1044 * become asynchronous until after this point. And anyhow
1045 * if this call was asynchronous then how would you get the
1046 * necessary HINTERNET pointer returned by this function.
1047 *
1048 */
1049 handle = HTTP_HttpOpenRequestW(lpwhs, lpszVerb, lpszObjectName,
1050 lpszVersion, lpszReferrer, lpszAcceptTypes,
1051 dwFlags, dwContext);
1052 lend:
1053 if( lpwhs )
1054 WININET_Release( &lpwhs->hdr );
1055 TRACE("returning %p\n", handle);
1056 return handle;
1057 }
1058
1059
1060 /***********************************************************************
1061 * HttpOpenRequestA (WININET.@)
1062 *
1063 * Open a HTTP request handle
1064 *
1065 * RETURNS
1066 * HINTERNET a HTTP request handle on success
1067 * NULL on failure
1068 *
1069 */
1070 HINTERNET WINAPI HttpOpenRequestA(HINTERNET hHttpSession,
1071 LPCSTR lpszVerb, LPCSTR lpszObjectName, LPCSTR lpszVersion,
1072 LPCSTR lpszReferrer , LPCSTR *lpszAcceptTypes,
1073 DWORD dwFlags, DWORD_PTR dwContext)
1074 {
1075 LPWSTR szVerb = NULL, szObjectName = NULL;
1076 LPWSTR szVersion = NULL, szReferrer = NULL, *szAcceptTypes = NULL;
1077 INT len, acceptTypesCount;
1078 HINTERNET rc = FALSE;
1079 LPCSTR *types;
1080
1081 TRACE("(%p, %s, %s, %s, %s, %p, %08x, %08lx)\n", hHttpSession,
1082 debugstr_a(lpszVerb), debugstr_a(lpszObjectName),
1083 debugstr_a(lpszVersion), debugstr_a(lpszReferrer), lpszAcceptTypes,
1084 dwFlags, dwContext);
1085
1086 if (lpszVerb)
1087 {
1088 len = MultiByteToWideChar(CP_ACP, 0, lpszVerb, -1, NULL, 0 );
1089 szVerb = HeapAlloc(GetProcessHeap(), 0, len * sizeof(WCHAR) );
1090 if ( !szVerb )
1091 goto end;
1092 MultiByteToWideChar(CP_ACP, 0, lpszVerb, -1, szVerb, len);
1093 }
1094
1095 if (lpszObjectName)
1096 {
1097 len = MultiByteToWideChar(CP_ACP, 0, lpszObjectName, -1, NULL, 0 );
1098 szObjectName = HeapAlloc(GetProcessHeap(), 0, len * sizeof(WCHAR) );
1099 if ( !szObjectName )
1100 goto end;
1101 MultiByteToWideChar(CP_ACP, 0, lpszObjectName, -1, szObjectName, len );
1102 }
1103
1104 if (lpszVersion)
1105 {
1106 len = MultiByteToWideChar(CP_ACP, 0, lpszVersion, -1, NULL, 0 );
1107 szVersion = HeapAlloc(GetProcessHeap(), 0, len * sizeof(WCHAR));
1108 if ( !szVersion )
1109 goto end;
1110 MultiByteToWideChar(CP_ACP, 0, lpszVersion, -1, szVersion, len );
1111 }
1112
1113 if (lpszReferrer)
1114 {
1115 len = MultiByteToWideChar(CP_ACP, 0, lpszReferrer, -1, NULL, 0 );
1116 szReferrer = HeapAlloc(GetProcessHeap(), 0, len * sizeof(WCHAR));
1117 if ( !szReferrer )
1118 goto end;
1119 MultiByteToWideChar(CP_ACP, 0, lpszReferrer, -1, szReferrer, len );
1120 }
1121
1122 if (lpszAcceptTypes)
1123 {
1124 acceptTypesCount = 0;
1125 types = lpszAcceptTypes;
1126 while (*types)
1127 {
1128 __TRY
1129 {
1130 /* find out how many there are */
1131 if (*types && **types)
1132 {
1133 TRACE("accept type: %s\n", debugstr_a(*types));
1134 acceptTypesCount++;
1135 }
1136 }
1137 __EXCEPT_PAGE_FAULT
1138 {
1139 WARN("invalid accept type pointer\n");
1140 }
1141 __ENDTRY;
1142 types++;
1143 }
1144 szAcceptTypes = HeapAlloc(GetProcessHeap(), 0, sizeof(WCHAR *) * (acceptTypesCount+1));
1145 if (!szAcceptTypes) goto end;
1146
1147 acceptTypesCount = 0;
1148 types = lpszAcceptTypes;
1149 while (*types)
1150 {
1151 __TRY
1152 {
1153 if (*types && **types)
1154 {
1155 len = MultiByteToWideChar(CP_ACP, 0, *types, -1, NULL, 0 );
1156 szAcceptTypes[acceptTypesCount] = HeapAlloc(GetProcessHeap(), 0, len * sizeof(WCHAR));
1157
1158 MultiByteToWideChar(CP_ACP, 0, *types, -1, szAcceptTypes[acceptTypesCount], len);
1159 acceptTypesCount++;
1160 }
1161 }
1162 __EXCEPT_PAGE_FAULT
1163 {
1164 /* ignore invalid pointer */
1165 }
1166 __ENDTRY;
1167 types++;
1168 }
1169 szAcceptTypes[acceptTypesCount] = NULL;
1170 }
1171
1172 rc = HttpOpenRequestW(hHttpSession, szVerb, szObjectName,
1173 szVersion, szReferrer,
1174 (LPCWSTR*)szAcceptTypes, dwFlags, dwContext);
1175
1176 end:
1177 if (szAcceptTypes)
1178 {
1179 acceptTypesCount = 0;
1180 while (szAcceptTypes[acceptTypesCount])
1181 {
1182 HeapFree(GetProcessHeap(), 0, szAcceptTypes[acceptTypesCount]);
1183 acceptTypesCount++;
1184 }
1185 HeapFree(GetProcessHeap(), 0, szAcceptTypes);
1186 }
1187 HeapFree(GetProcessHeap(), 0, szReferrer);
1188 HeapFree(GetProcessHeap(), 0, szVersion);
1189 HeapFree(GetProcessHeap(), 0, szObjectName);
1190 HeapFree(GetProcessHeap(), 0, szVerb);
1191
1192 return rc;
1193 }
1194
1195 /***********************************************************************
1196 * HTTP_EncodeBase64
1197 */
1198 static UINT HTTP_EncodeBase64( LPCSTR bin, unsigned int len, LPWSTR base64 )
1199 {
1200 UINT n = 0, x;
1201 static const CHAR HTTP_Base64Enc[] =
1202 "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/";
1203
1204 while( len > 0 )
1205 {
1206 /* first 6 bits, all from bin[0] */
1207 base64[n++] = HTTP_Base64Enc[(bin[0] & 0xfc) >> 2];
1208 x = (bin[0] & 3) << 4;
1209
1210 /* next 6 bits, 2 from bin[0] and 4 from bin[1] */
1211 if( len == 1 )
1212 {
1213 base64[n++] = HTTP_Base64Enc[x];
1214 base64[n++] = '=';
1215 base64[n++] = '=';
1216 break;
1217 }
1218 base64[n++] = HTTP_Base64Enc[ x | ( (bin[1]&0xf0) >> 4 ) ];
1219 x = ( bin[1] & 0x0f ) << 2;
1220
1221 /* next 6 bits 4 from bin[1] and 2 from bin[2] */
1222 if( len == 2 )
1223 {
1224 base64[n++] = HTTP_Base64Enc[x];
1225 base64[n++] = '=';
1226 break;
1227 }
1228 base64[n++] = HTTP_Base64Enc[ x | ( (bin[2]&0xc0 ) >> 6 ) ];
1229
1230 /* last 6 bits, all from bin [2] */
1231 base64[n++] = HTTP_Base64Enc[ bin[2] & 0x3f ];
1232 bin += 3;
1233 len -= 3;
1234 }
1235 base64[n] = 0;
1236 return n;
1237 }
1238
1239 #define CH(x) (((x) >= 'A' && (x) <= 'Z') ? (x) - 'A' : \
1240 ((x) >= 'a' && (x) <= 'z') ? (x) - 'a' + 26 : \
1241 ((x) >= '0' && (x) <= '9') ? (x) - '0' + 52 : \
1242 ((x) == '+') ? 62 : ((x) == '/') ? 63 : -1)
1243 static const signed char HTTP_Base64Dec[256] =
1244 {
1245 CH( 0),CH( 1),CH( 2),CH( 3),CH( 4),CH( 5),CH( 6),CH( 7),CH( 8),CH( 9),
1246 CH(10),CH(11),CH(12),CH(13),CH(14),CH(15),CH(16),CH(17),CH(18),CH(19),
1247 CH(20),CH(21),CH(22),CH(23),CH(24),CH(25),CH(26),CH(27),CH(28),CH(29),
1248 CH(30),CH(31),CH(32),CH(33),CH(34),CH(35),CH(36),CH(37),CH(38),CH(39),
1249 CH(40),CH(41),CH(42),CH(43),CH(44),CH(45),CH(46),CH(47),CH(48),CH(49),
1250 CH(50),CH(51),CH(52),CH(53),CH(54),CH(55),CH(56),CH(57),CH(58),CH(59),
1251 CH(60),CH(61),CH(62),CH(63),CH(64),CH(65),CH(66),CH(67),CH(68),CH(69),
1252 CH(70),CH(71),CH(72),CH(73),CH(74),CH(75),CH(76),CH(77),CH(78),CH(79),
1253 CH(80),CH(81),CH(82),CH(83),CH(84),CH(85),CH(86),CH(87),CH(88),CH(89),
1254 CH(90),CH(91),CH(92),CH(93),CH(94),CH(95),CH(96),CH(97),CH(98),CH(99),
1255 CH(100),CH(101),CH(102),CH(103),CH(104),CH(105),CH(106),CH(107),CH(108),CH(109),
1256 CH(110),CH(111),CH(112),CH(113),CH(114),CH(115),CH(116),CH(117),CH(118),CH(119),
1257 CH(120),CH(121),CH(122),CH(123),CH(124),CH(125),CH(126),CH(127),CH(128),CH(129),
1258 CH(130),CH(131),CH(132),CH(133),CH(134),CH(135),CH(136),CH(137),CH(138),CH(139),
1259 CH(140),CH(141),CH(142),CH(143),CH(144),CH(145),CH(146),CH(147),CH(148),CH(149),
1260 CH(150),CH(151),CH(152),CH(153),CH(154),CH(155),CH(156),CH(157),CH(158),CH(159),
1261 CH(160),CH(161),CH(162),CH(163),CH(164),CH(165),CH(166),CH(167),CH(168),CH(169),
1262 CH(170),CH(171),CH(172),CH(173),CH(174),CH(175),CH(176),CH(177),CH(178),CH(179),
1263 CH(180),CH(181),CH(182),CH(183),CH(184),CH(185),CH(186),CH(187),CH(188),CH(189),
1264 CH(190),CH(191),CH(192),CH(193),CH(194),CH(195),CH(196),CH(197),CH(198),CH(199),
1265 CH(200),CH(201),CH(202),CH(203),CH(204),CH(205),CH(206),CH(207),CH(208),CH(209),
1266 CH(210),CH(211),CH(212),CH(213),CH(214),CH(215),CH(216),CH(217),CH(218),CH(219),
1267 CH(220),CH(221),CH(222),CH(223),CH(224),CH(225),CH(226),CH(227),CH(228),CH(229),
1268 CH(230),CH(231),CH(232),CH(233),CH(234),CH(235),CH(236),CH(237),CH(238),CH(239),
1269 CH(240),CH(241),CH(242),CH(243),CH(244),CH(245),CH(246),CH(247),CH(248), CH(249),
1270 CH(250),CH(251),CH(252),CH(253),CH(254),CH(255),
1271 };
1272 #undef CH
1273
1274 /***********************************************************************
1275 * HTTP_DecodeBase64
1276 */
1277 static UINT HTTP_DecodeBase64( LPCWSTR base64, LPSTR bin )
1278 {
1279 unsigned int n = 0;
1280
1281 while(*base64)
1282 {
1283 signed char in[4];
1284
1285 if (base64[0] >= ARRAYSIZE(HTTP_Base64Dec) ||
1286 ((in[0] = HTTP_Base64Dec[base64[0]]) == -1) ||
1287 base64[1] >= ARRAYSIZE(HTTP_Base64Dec) ||
1288 ((in[1] = HTTP_Base64Dec[base64[1]]) == -1))
1289 {
1290 WARN("invalid base64: %s\n", debugstr_w(base64));
1291 return 0;
1292 }
1293 if (bin)
1294 bin[n] = (unsigned char) (in[0] << 2 | in[1] >> 4);
1295 n++;
1296
1297 if ((base64[2] == '=') && (base64[3] == '='))
1298 break;
1299 if (base64[2] > ARRAYSIZE(HTTP_Base64Dec) ||
1300 ((in[2] = HTTP_Base64Dec[base64[2]]) == -1))
1301 {
1302 WARN("invalid base64: %s\n", debugstr_w(&base64[2]));
1303 return 0;
1304 }
1305 if (bin)
1306 bin[n] = (unsigned char) (in[1] << 4 | in[2] >> 2);
1307 n++;
1308
1309 if (base64[3] == '=')
1310 break;
1311 if (base64[3] > ARRAYSIZE(HTTP_Base64Dec) ||
1312 ((in[3] = HTTP_Base64Dec[base64[3]]) == -1))
1313 {
1314 WARN("invalid base64: %s\n", debugstr_w(&base64[3]));
1315 return 0;
1316 }
1317 if (bin)
1318 bin[n] = (unsigned char) (((in[2] << 6) & 0xc0) | in[3]);
1319 n++;
1320
1321 base64 += 4;
1322 }
1323
1324 return n;
1325 }
1326
1327 /***********************************************************************
1328 * HTTP_InsertAuthorization
1329 *
1330 * Insert or delete the authorization field in the request header.
1331 */
1332 static BOOL HTTP_InsertAuthorization( LPWININETHTTPREQW lpwhr, struct HttpAuthInfo *pAuthInfo, LPCWSTR header )
1333 {
1334 if (pAuthInfo)
1335 {
1336 static const WCHAR wszSpace[] = {' ',0};
1337 static const WCHAR wszBasic[] = {'B','a','s','i','c',0};
1338 unsigned int len;
1339 WCHAR *authorization = NULL;
1340
1341 if (pAuthInfo->auth_data_len)
1342 {
1343 /* scheme + space + base64 encoded data (3/2/1 bytes data -> 4 bytes of characters) */
1344 len = strlenW(pAuthInfo->scheme)+1+((pAuthInfo->auth_data_len+2)*4)/3;
1345 authorization = HeapAlloc(GetProcessHeap(), 0, (len+1)*sizeof(WCHAR));
1346 if (!authorization)
1347 return FALSE;
1348
1349 strcpyW(authorization, pAuthInfo->scheme);
1350 strcatW(authorization, wszSpace);
1351 HTTP_EncodeBase64(pAuthInfo->auth_data,
1352 pAuthInfo->auth_data_len,
1353 authorization+strlenW(authorization));
1354
1355 /* clear the data as it isn't valid now that it has been sent to the
1356 * server, unless it's Basic authentication which doesn't do
1357 * connection tracking */
1358 if (strcmpiW(pAuthInfo->scheme, wszBasic))
1359 {
1360 HeapFree(GetProcessHeap(), 0, pAuthInfo->auth_data);
1361 pAuthInfo->auth_data = NULL;
1362 pAuthInfo->auth_data_len = 0;
1363 }
1364 }
1365
1366 TRACE("Inserting authorization: %s\n", debugstr_w(authorization));
1367
1368 HTTP_ProcessHeader(lpwhr, header, authorization, HTTP_ADDHDR_FLAG_REQ | HTTP_ADDHDR_FLAG_REPLACE);
1369
1370 HeapFree(GetProcessHeap(), 0, authorization);
1371 }
1372 return TRUE;
1373 }
1374
1375 static WCHAR *HTTP_BuildProxyRequestUrl(WININETHTTPREQW *req)
1376 {
1377 WCHAR new_location[INTERNET_MAX_URL_LENGTH], *url;
1378 DWORD size;
1379
1380 size = sizeof(new_location);
1381 if (HTTP_HttpQueryInfoW(req, HTTP_QUERY_LOCATION, new_location, &size, NULL))
1382 {
1383 if (!(url = HeapAlloc( GetProcessHeap(), 0, size + sizeof(WCHAR) ))) return NULL;
1384 strcpyW( url, new_location );
1385 }
1386 else
1387 {
1388 static const WCHAR slash[] = { '/',0 };
1389 static const WCHAR format[] = { 'h','t','t','p',':','/','/','%','s',':','%','d',0 };
1390 static const WCHAR formatSSL[] = { 'h','t','t','p','s',':','/','/','%','s',':','%','d',0 };
1391 WININETHTTPSESSIONW *session = req->lpHttpSession;
1392
1393 size = 16; /* "https://" + sizeof(port#) + ":/\0" */
1394 size += strlenW( session->lpszHostName ) + strlenW( req->lpszPath );
1395
1396 if (!(url = HeapAlloc( GetProcessHeap(), 0, size * sizeof(WCHAR) ))) return NULL;
1397
1398 if (req->hdr.dwFlags & INTERNET_FLAG_SECURE)
1399 sprintfW( url, formatSSL, session->lpszHostName, session->nHostPort );
1400 else
1401 sprintfW( url, format, session->lpszHostName, session->nHostPort );
1402 if (req->lpszPath[0] != '/') strcatW( url, slash );
1403 strcatW( url, req->lpszPath );
1404 }
1405 TRACE("url=%s\n", debugstr_w(url));
1406 return url;
1407 }
1408
1409 /***********************************************************************
1410 * HTTP_DealWithProxy
1411 */
1412 static BOOL HTTP_DealWithProxy( LPWININETAPPINFOW hIC,
1413 LPWININETHTTPSESSIONW lpwhs, LPWININETHTTPREQW lpwhr)
1414 {
1415 WCHAR buf[MAXHOSTNAME];
1416 WCHAR proxy[MAXHOSTNAME + 15]; /* 15 == "http://" + sizeof(port#) + ":/\0" */
1417 static WCHAR szNul[] = { 0 };
1418 URL_COMPONENTSW UrlComponents;
1419 static const WCHAR szHttp[] = { 'h','t','t','p',':','/','/',0 };
1420 static const WCHAR szFormat[] = { 'h','t','t','p',':','/','/','%','s',0 };
1421
1422 memset( &UrlComponents, 0, sizeof UrlComponents );
1423 UrlComponents.dwStructSize = sizeof UrlComponents;
1424 UrlComponents.lpszHostName = buf;
1425 UrlComponents.dwHostNameLength = MAXHOSTNAME;
1426
1427 if( CSTR_EQUAL != CompareStringW(LOCALE_SYSTEM_DEFAULT, NORM_IGNORECASE,
1428 hIC->lpszProxy,strlenW(szHttp),szHttp,strlenW(szHttp)) )
1429 sprintfW(proxy, szFormat, hIC->lpszProxy);
1430 else
1431 strcpyW(proxy, hIC->lpszProxy);
1432 if( !InternetCrackUrlW(proxy, 0, 0, &UrlComponents) )
1433 return FALSE;
1434 if( UrlComponents.dwHostNameLength == 0 )
1435 return FALSE;
1436
1437 if( !lpwhr->lpszPath )
1438 lpwhr->lpszPath = szNul;
1439
1440 if(UrlComponents.nPort == INTERNET_INVALID_PORT_NUMBER)
1441 UrlComponents.nPort = INTERNET_DEFAULT_HTTP_PORT;
1442
1443 HeapFree(GetProcessHeap(), 0, lpwhs->lpszServerName);
1444 lpwhs->lpszServerName = WININET_strdupW(UrlComponents.lpszHostName);
1445 lpwhs->nServerPort = UrlComponents.nPort;
1446
1447 TRACE("proxy server=%s port=%d\n", debugstr_w(lpwhs->lpszServerName), lpwhs->nServerPort);
1448 return TRUE;
1449 }
1450
1451 static BOOL HTTP_ResolveName(LPWININETHTTPREQW lpwhr)
1452 {
1453 char szaddr[32];
1454 LPWININETHTTPSESSIONW lpwhs = lpwhr->lpHttpSession;
1455
1456 INTERNET_SendCallback(&lpwhr->hdr, lpwhr->hdr.dwContext,
1457 INTERNET_STATUS_RESOLVING_NAME,
1458 lpwhs->lpszServerName,
1459 strlenW(lpwhs->lpszServerName)+1);
1460
1461 if (!GetAddress(lpwhs->lpszServerName, lpwhs->nServerPort,
1462 &lpwhs->socketAddress))
1463 {
1464 INTERNET_SetLastError(ERROR_INTERNET_NAME_NOT_RESOLVED);
1465 return FALSE;
1466 }
1467
1468 inet_ntop(lpwhs->socketAddress.sin_family, &lpwhs->socketAddress.sin_addr,
1469 szaddr, sizeof(szaddr));
1470 INTERNET_SendCallback(&lpwhr->hdr, lpwhr->hdr.dwContext,
1471 INTERNET_STATUS_NAME_RESOLVED,
1472 szaddr, strlen(szaddr)+1);
1473
1474 TRACE("resolved %s to %s\n", debugstr_w(lpwhs->lpszServerName), szaddr);
1475 return TRUE;
1476 }
1477
1478
1479 /***********************************************************************
1480 * HTTPREQ_Destroy (internal)
1481 *
1482 * Deallocate request handle
1483 *
1484 */
1485 static void HTTPREQ_Destroy(WININETHANDLEHEADER *hdr)
1486 {
1487 LPWININETHTTPREQW lpwhr = (LPWININETHTTPREQW) hdr;
1488 DWORD i;
1489
1490 TRACE("\n");
1491
1492 if(lpwhr->hCacheFile)
1493 CloseHandle(lpwhr->hCacheFile);
1494
1495 if(lpwhr->lpszCacheFile) {
1496 DeleteFileW(lpwhr->lpszCacheFile); /* FIXME */
1497 HeapFree(GetProcessHeap(), 0, lpwhr->lpszCacheFile);
1498 }
1499
1500 DeleteCriticalSection( &lpwhr->read_section );
1501 WININET_Release(&lpwhr->lpHttpSession->hdr);
1502
1503 if (lpwhr->pAuthInfo)
1504 {
1505 if (SecIsValidHandle(&lpwhr->pAuthInfo->ctx))
1506 DeleteSecurityContext(&lpwhr->pAuthInfo->ctx);
1507 if (SecIsValidHandle(&lpwhr->pAuthInfo->cred))
1508 FreeCredentialsHandle(&lpwhr->pAuthInfo->cred);
1509
1510 HeapFree(GetProcessHeap(), 0, lpwhr->pAuthInfo->auth_data);
1511 HeapFree(GetProcessHeap(), 0, lpwhr->pAuthInfo->scheme);
1512 HeapFree(GetProcessHeap(), 0, lpwhr->pAuthInfo);
1513 lpwhr->pAuthInfo = NULL;
1514 }
1515
1516 if (lpwhr->pProxyAuthInfo)
1517 {
1518 if (SecIsValidHandle(&lpwhr->pProxyAuthInfo->ctx))
1519 DeleteSecurityContext(&lpwhr->pProxyAuthInfo->ctx);
1520 if (SecIsValidHandle(&lpwhr->pProxyAuthInfo->cred))
1521 FreeCredentialsHandle(&lpwhr->pProxyAuthInfo->cred);
1522
1523 HeapFree(GetProcessHeap(), 0, lpwhr->pProxyAuthInfo->auth_data);
1524 HeapFree(GetProcessHeap(), 0, lpwhr->pProxyAuthInfo->scheme);
1525 HeapFree(GetProcessHeap(), 0, lpwhr->pProxyAuthInfo);
1526 lpwhr->pProxyAuthInfo = NULL;
1527 }
1528
1529 HeapFree(GetProcessHeap(), 0, lpwhr->lpszPath);
1530 HeapFree(GetProcessHeap(), 0, lpwhr->lpszVerb);
1531 HeapFree(GetProcessHeap(), 0, lpwhr->lpszRawHeaders);
1532 HeapFree(GetProcessHeap(), 0, lpwhr->lpszVersion);
1533 HeapFree(GetProcessHeap(), 0, lpwhr->lpszStatusText);
1534
1535 for (i = 0; i < lpwhr->nCustHeaders; i++)
1536 {
1537 HeapFree(GetProcessHeap(), 0, lpwhr->pCustHeaders[i].lpszField);
1538 HeapFree(GetProcessHeap(), 0, lpwhr->pCustHeaders[i].lpszValue);
1539 }
1540
1541 HeapFree(GetProcessHeap(), 0, lpwhr->pCustHeaders);
1542 HeapFree(GetProcessHeap(), 0, lpwhr);
1543 }
1544
1545 static void HTTPREQ_CloseConnection(WININETHANDLEHEADER *hdr)
1546 {
1547 LPWININETHTTPREQW lpwhr = (LPWININETHTTPREQW) hdr;
1548
1549 TRACE("%p\n",lpwhr);
1550
1551 #ifdef HAVE_ZLIB
1552 if(lpwhr->gzip_stream) {
1553 inflateEnd(&lpwhr->gzip_stream->zstream);
1554 HeapFree(GetProcessHeap(), 0, lpwhr->gzip_stream);
1555 lpwhr->gzip_stream = NULL;
1556 }
1557 #endif
1558
1559 if (!NETCON_connected(&lpwhr->netConnection))
1560 return;
1561
1562 INTERNET_SendCallback(&lpwhr->hdr, lpwhr->hdr.dwContext,
1563 INTERNET_STATUS_CLOSING_CONNECTION, 0, 0);
1564
1565 NETCON_close(&lpwhr->netConnection);
1566
1567 INTERNET_SendCallback(&lpwhr->hdr, lpwhr->hdr.dwContext,
1568 INTERNET_STATUS_CONNECTION_CLOSED, 0, 0);
1569 }
1570
1571 static BOOL HTTP_GetRequestURL(WININETHTTPREQW *req, LPWSTR buf)
1572 {
1573 LPHTTPHEADERW host_header;
1574
1575 static const WCHAR formatW[] = {'h','t','t','p',':','/','/','%','s','%','s',0};
1576
1577 host_header = HTTP_GetHeader(req, hostW);
1578 if(!host_header)
1579 return FALSE;
1580
1581 sprintfW(buf, formatW, host_header->lpszValue, req->lpszPath); /* FIXME */
1582 return TRUE;
1583 }
1584
1585 static DWORD HTTPREQ_QueryOption(WININETHANDLEHEADER *hdr, DWORD option, void *buffer, DWORD *size, BOOL unicode)
1586 {
1587 WININETHTTPREQW *req = (WININETHTTPREQW*)hdr;
1588
1589 switch(option) {
1590 case INTERNET_OPTION_HANDLE_TYPE:
1591 TRACE("INTERNET_OPTION_HANDLE_TYPE\n");
1592
1593 if (*size < sizeof(ULONG))
1594 return ERROR_INSUFFICIENT_BUFFER;
1595
1596 *size = sizeof(DWORD);
1597 *(DWORD*)buffer = INTERNET_HANDLE_TYPE_HTTP_REQUEST;
1598 return ERROR_SUCCESS;
1599
1600 case INTERNET_OPTION_URL: {
1601 WCHAR url[INTERNET_MAX_URL_LENGTH];
1602 HTTPHEADERW *host;
1603 DWORD len;
1604 WCHAR *pch;
1605
1606 static const WCHAR httpW[] = {'h','t','t','p',':','/','/',0};
1607
1608 TRACE("INTERNET_OPTION_URL\n");
1609
1610 host = HTTP_GetHeader(req, hostW);
1611 strcpyW(url, httpW);
1612 strcatW(url, host->lpszValue);
1613 if (NULL != (pch = strchrW(url + strlenW(httpW), ':')))
1614 *pch = 0;
1615 strcatW(url, req->lpszPath);
1616
1617 TRACE("INTERNET_OPTION_URL: %s\n",debugstr_w(url));
1618
1619 if(unicode) {
1620 len = (strlenW(url)+1) * sizeof(WCHAR);
1621 if(*size < len)
1622 return ERROR_INSUFFICIENT_BUFFER;
1623
1624 *size = len;
1625 strcpyW(buffer, url);
1626 return ERROR_SUCCESS;
1627 }else {
1628 len = WideCharToMultiByte(CP_ACP, 0, url, -1, buffer, *size, NULL, NULL);
1629 if(len > *size)
1630 return ERROR_INSUFFICIENT_BUFFER;
1631
1632 *size = len;
1633 return ERROR_SUCCESS;
1634 }
1635 }
1636
1637 case INTERNET_OPTION_CACHE_TIMESTAMPS: {
1638 INTERNET_CACHE_ENTRY_INFOW *info;
1639 INTERNET_CACHE_TIMESTAMPS *ts = buffer;
1640 WCHAR url[INTERNET_MAX_URL_LENGTH];
1641 DWORD nbytes, error;
1642 BOOL ret;
1643
1644 TRACE("INTERNET_OPTION_CACHE_TIMESTAMPS\n");
1645
1646 if (*size < sizeof(*ts))
1647 {
1648 *size = sizeof(*ts);
1649 return ERROR_INSUFFICIENT_BUFFER;
1650 }
1651 nbytes = 0;
1652 HTTP_GetRequestURL(req, url);
1653 ret = GetUrlCacheEntryInfoW(url, NULL, &nbytes);
1654 error = GetLastError();
1655 if (!ret && error == ERROR_INSUFFICIENT_BUFFER)
1656 {
1657 if (!(info = HeapAlloc(GetProcessHeap(), 0, nbytes)))
1658 return ERROR_OUTOFMEMORY;
1659
1660 GetUrlCacheEntryInfoW(url, info, &nbytes);
1661
1662 ts->ftExpires = info->ExpireTime;
1663 ts->ftLastModified = info->LastModifiedTime;
1664
1665 HeapFree(GetProcessHeap(), 0, info);
1666 *size = sizeof(*ts);
1667 return ERROR_SUCCESS;
1668 }
1669 return error;
1670 }
1671
1672 case INTERNET_OPTION_DATAFILE_NAME: {
1673 DWORD req_size;
1674
1675 TRACE("INTERNET_OPTION_DATAFILE_NAME\n");
1676
1677 if(!req->lpszCacheFile) {
1678 *size = 0;
1679 return ERROR_INTERNET_ITEM_NOT_FOUND;
1680 }
1681
1682 if(unicode) {
1683 req_size = (lstrlenW(req->lpszCacheFile)+1) * sizeof(WCHAR);
1684 if(*size < req_size)
1685 return ERROR_INSUFFICIENT_BUFFER;
1686
1687 *size = req_size;
1688 memcpy(buffer, req->lpszCacheFile, *size);
1689 return ERROR_SUCCESS;
1690 }else {
1691 req_size = WideCharToMultiByte(CP_ACP, 0, req->lpszCacheFile, -1, NULL, 0, NULL, NULL);
1692 if (req_size > *size)
1693 return ERROR_INSUFFICIENT_BUFFER;
1694
1695 *size = WideCharToMultiByte(CP_ACP, 0, req->lpszCacheFile,
1696 -1, buffer, *size, NULL, NULL);
1697 return ERROR_SUCCESS;
1698 }
1699 }
1700
1701 case INTERNET_OPTION_SECURITY_CERTIFICATE_STRUCT: {
1702 PCCERT_CONTEXT context;
1703
1704 if(*size < sizeof(INTERNET_CERTIFICATE_INFOW)) {
1705 *size = sizeof(INTERNET_CERTIFICATE_INFOW);
1706 return ERROR_INSUFFICIENT_BUFFER;
1707 }
1708
1709 context = (PCCERT_CONTEXT)NETCON_GetCert(&(req->netConnection));
1710 if(context) {
1711 INTERNET_CERTIFICATE_INFOW *info = (INTERNET_CERTIFICATE_INFOW*)buffer;
1712 DWORD len;
1713
1714 memset(info, 0, sizeof(INTERNET_CERTIFICATE_INFOW));
1715 info->ftExpiry = context->pCertInfo->NotAfter;
1716 info->ftStart = context->pCertInfo->NotBefore;
1717 if(unicode) {
1718 len = CertNameToStrW(context->dwCertEncodingType,
1719 &context->pCertInfo->Subject, CERT_SIMPLE_NAME_STR, NULL, 0);
1720 info->lpszSubjectInfo = LocalAlloc(0, len*sizeof(WCHAR));
1721 if(info->lpszSubjectInfo)
1722 CertNameToStrW(context->dwCertEncodingType,
1723 &context->pCertInfo->Subject, CERT_SIMPLE_NAME_STR,
1724 info->lpszSubjectInfo, len);
1725 len = CertNameToStrW(context->dwCertEncodingType,
1726 &context->pCertInfo->Issuer, CERT_SIMPLE_NAME_STR, NULL, 0);
1727 info->lpszIssuerInfo = LocalAlloc(0, len*sizeof(WCHAR));
1728 if (info->lpszIssuerInfo)
1729 CertNameToStrW(context->dwCertEncodingType,
1730 &context->pCertInfo->Issuer, CERT_SIMPLE_NAME_STR,
1731 info->lpszIssuerInfo, len);
1732 }else {
1733 INTERNET_CERTIFICATE_INFOA *infoA = (INTERNET_CERTIFICATE_INFOA*)info;
1734
1735 len = CertNameToStrA(context->dwCertEncodingType,
1736 &context->pCertInfo->Subject, CERT_SIMPLE_NAME_STR, NULL, 0);
1737 infoA->lpszSubjectInfo = LocalAlloc(0, len);
1738 if(infoA->lpszSubjectInfo)
1739 CertNameToStrA(context->dwCertEncodingType,
1740 &context->pCertInfo->Subject, CERT_SIMPLE_NAME_STR,
1741 infoA->lpszSubjectInfo, len);
1742 len = CertNameToStrA(context->dwCertEncodingType,
1743 &context->pCertInfo->Issuer, CERT_SIMPLE_NAME_STR, NULL, 0);
1744 infoA->lpszIssuerInfo = LocalAlloc(0, len);
1745 if(infoA->lpszIssuerInfo)
1746 CertNameToStrA(context->dwCertEncodingType,
1747 &context->pCertInfo->Issuer, CERT_SIMPLE_NAME_STR,
1748 infoA->lpszIssuerInfo, len);
1749 }
1750
1751 /*
1752 * Contrary to MSDN, these do not appear to be set.
1753 * lpszProtocolName
1754 * lpszSignatureAlgName
1755 * lpszEncryptionAlgName
1756 * dwKeySize
1757 */
1758 CertFreeCertificateContext(context);
1759 return ERROR_SUCCESS;
1760 }
1761 }
1762 }
1763
1764 return INET_QueryOption(option, buffer, size, unicode);
1765 }
1766
1767 static DWORD HTTPREQ_SetOption(WININETHANDLEHEADER *hdr, DWORD option, void *buffer, DWORD size)
1768 {
1769 WININETHTTPREQW *req = (WININETHTTPREQW*)hdr;
1770
1771 switch(option) {
1772 case INTERNET_OPTION_SEND_TIMEOUT:
1773 case INTERNET_OPTION_RECEIVE_TIMEOUT:
1774 TRACE("INTERNET_OPTION_SEND/RECEIVE_TIMEOUT\n");
1775
1776 if (size != sizeof(DWORD))
1777 return ERROR_INVALID_PARAMETER;
1778
1779 return NETCON_set_timeout(&req->netConnection, option == INTERNET_OPTION_SEND_TIMEOUT,
1780 *(DWORD*)buffer);
1781
1782 case INTERNET_OPTION_USERNAME:
1783 HeapFree(GetProcessHeap(), 0, req->lpHttpSession->lpszUserName);
1784 if (!(req->lpHttpSession->lpszUserName = WININET_strdupW(buffer))) return ERROR_OUTOFMEMORY;
1785 return ERROR_SUCCESS;
1786
1787 case INTERNET_OPTION_PASSWORD:
1788 HeapFree(GetProcessHeap(), 0, req->lpHttpSession->lpszPassword);
1789 if (!(req->lpHttpSession->lpszPassword = WININET_strdupW(buffer))) return ERROR_OUTOFMEMORY;
1790 return ERROR_SUCCESS;
1791 case INTERNET_OPTION_HTTP_DECODING:
1792 if(size != sizeof(BOOL))
1793 return ERROR_INVALID_PARAMETER;
1794 req->decoding = *(BOOL*)buffer;
1795 return ERROR_SUCCESS;
1796 }
1797
1798 return ERROR_INTERNET_INVALID_OPTION;
1799 }
1800
1801 /* read some more data into the read buffer (the read section must be held) */
1802 static BOOL read_more_data( WININETHTTPREQW *req, int maxlen )
1803 {
1804 int len;
1805
1806 if (req->read_pos)
1807 {
1808 /* move existing data to the start of the buffer */
1809 if(req->read_size)
1810 memmove( req->read_buf, req->read_buf + req->read_pos, req->read_size );
1811 req->read_pos = 0;
1812 }
1813
1814 if (maxlen == -1) maxlen = sizeof(req->read_buf);
1815
1816 if(!NETCON_recv( &req->netConnection, req->read_buf + req->read_size,
1817 maxlen - req->read_size, 0, &len ))
1818 return FALSE;
1819
1820 req->read_size += len;
1821 return TRUE;
1822 }
1823
1824 /* remove some amount of data from the read buffer (the read section must be held) */
1825 static void remove_data( WININETHTTPREQW *req, int count )
1826 {
1827 if (!(req->read_size -= count)) req->read_pos = 0;
1828 else req->read_pos += count;
1829 }
1830
1831 static BOOL read_line( WININETHTTPREQW *req, LPSTR buffer, DWORD *len )
1832 {
1833 int count, bytes_read, pos = 0;
1834
1835 EnterCriticalSection( &req->read_section );
1836 for (;;)
1837 {
1838 BYTE *eol = memchr( req->read_buf + req->read_pos, '\n', req->read_size );
1839
1840 if (eol)
1841 {
1842 count = eol - (req->read_buf + req->read_pos);
1843 bytes_read = count + 1;
1844 }
1845 else count = bytes_read = req->read_size;
1846
1847 count = min( count, *len - pos );
1848 memcpy( buffer + pos, req->read_buf + req->read_pos, count );
1849 pos += count;
1850 remove_data( req, bytes_read );
1851 if (eol) break;
1852
1853 if (!read_more_data( req, -1 ) || !req->read_size)
1854 {
1855 *len = 0;
1856 TRACE( "returning empty string\n" );
1857 LeaveCriticalSection( &req->read_section );
1858 return FALSE;
1859 }
1860 }
1861 LeaveCriticalSection( &req->read_section );
1862
1863 if (pos < *len)
1864 {
1865 if (pos && buffer[pos - 1] == '\r') pos--;
1866 *len = pos + 1;
1867 }
1868 buffer[*len - 1] = 0;
1869 TRACE( "returning %s\n", debugstr_a(buffer));
1870 return TRUE;
1871 }
1872
1873 /* discard data contents until we reach end of line (the read section must be held) */
1874 static BOOL discard_eol( WININETHTTPREQW *req )
1875 {
1876 do
1877 {
1878 BYTE *eol = memchr( req->read_buf + req->read_pos, '\n', req->read_size );
1879 if (eol)
1880 {
1881 remove_data( req, (eol + 1) - (req->read_buf + req->read_pos) );
1882 break;
1883 }
1884 req->read_pos = req->read_size = 0; /* discard everything */
1885 if (!read_more_data( req, -1 )) return FALSE;
1886 } while (req->read_size);
1887 return TRUE;
1888 }
1889
1890 /* read the size of the next chunk (the read section must be held) */
1891 static BOOL start_next_chunk( WININETHTTPREQW *req )
1892 {
1893 DWORD chunk_size = 0;
1894
1895 if (!req->dwContentLength) return TRUE;
1896 if (req->dwContentLength == req->dwContentRead)
1897 {
1898 /* read terminator for the previous chunk */
1899 if (!discard_eol( req )) return FALSE;
1900 req->dwContentLength = ~0u;
1901 req->dwContentRead = 0;
1902 }
1903 for (;;)
1904 {
1905 while (req->read_size)
1906 {
1907 char ch = req->read_buf[req->read_pos];
1908 if (ch >= '0' && ch <= '9') chunk_size = chunk_size * 16 + ch - '0';
1909 else if (ch >= 'a' && ch <= 'f') chunk_size = chunk_size * 16 + ch - 'a' + 10;
1910 else if (ch >= 'A' && ch <= 'F') chunk_size = chunk_size * 16 + ch - 'A' + 10;
1911 else if (ch == ';' || ch == '\r' || ch == '\n')
1912 {
1913 TRACE( "reading %u byte chunk\n", chunk_size );
1914 req->dwContentLength = chunk_size;
1915 req->dwContentRead = 0;
1916 if (!discard_eol( req )) return FALSE;
1917 return TRUE;
1918 }
1919 remove_data( req, 1 );
1920 }
1921 if (!read_more_data( req, -1 )) return FALSE;
1922 if (!req->read_size)
1923 {
1924 req->dwContentLength = req->dwContentRead = 0;
1925 return TRUE;
1926 }
1927 }
1928 }
1929
1930 /* check if we have reached the end of the data to read (the read section must be held) */
1931 static BOOL end_of_read_data( WININETHTTPREQW *req )
1932 {
1933 if (req->gzip_stream) return req->gzip_stream->end_of_data && !req->gzip_stream->buf_size;
1934 if (req->read_chunked) return (req->dwContentLength == 0);
1935 if (req->dwContentLength == ~0u) return FALSE;
1936 return (req->dwContentLength == req->dwContentRead);
1937 }
1938
1939 /* fetch some more data into the read buffer (the read section must be held) */
1940 static BOOL refill_buffer( WININETHTTPREQW *req )
1941 {
1942 int len = sizeof(req->read_buf);
1943
1944 if (req->read_chunked && (req->dwContentLength == ~0u || req->dwContentLength == req->dwContentRead))
1945 {
1946 if (!start_next_chunk( req )) return FALSE;
1947 }
1948
1949 if (req->dwContentLength != ~0u) len = min( len, req->dwContentLength - req->dwContentRead );
1950 if (len <= req->read_size) return TRUE;
1951
1952 if (!read_more_data( req, len )) return FALSE;
1953 if (!req->read_size) req->dwContentLength = req->dwContentRead = 0;
1954 return TRUE;
1955 }
1956
1957 static DWORD read_gzip_data(WININETHTTPREQW *req, BYTE *buf, int size, BOOL sync, int *read_ret)
1958 {
1959 DWORD ret = ERROR_SUCCESS;
1960 int read = 0;
1961
1962 #ifdef HAVE_ZLIB
1963 z_stream *zstream = &req->gzip_stream->zstream;
1964 int zres;
1965
1966 while(read < size && !req->gzip_stream->end_of_data) {
1967 if(!req->read_size) {
1968 if(!sync || !refill_buffer(req))
1969 break;
1970 }
1971
1972 zstream->next_in = req->read_buf+req->read_pos;
1973 zstream->avail_in = req->read_size;
1974 zstream->next_out = buf+read;
1975 zstream->avail_out = size-read;
1976 zres = inflate(zstream, Z_FULL_FLUSH);
1977 read = size - zstream->avail_out;
1978 remove_data(req, req->read_size-zstream->avail_in);
1979 if(zres == Z_STREAM_END) {
1980 TRACE("end of data\n");
1981 req->gzip_stream->end_of_data = TRUE;
1982 }else if(zres != Z_OK) {
1983 WARN("inflate failed %d\n", zres);
1984 if(!read)
1985 ret = ERROR_INTERNET_DECODING_FAILED;
1986 break;
1987 }
1988 }
1989 #endif
1990
1991 *read_ret = read;
1992 return ret;
1993 }
1994
1995 static void refill_gzip_buffer(WININETHTTPREQW *req)
1996 {
1997 DWORD res;
1998 int len;
1999
2000 if(!req->gzip_stream || !req->read_size || req->gzip_stream->buf_size == sizeof(req->gzip_stream->buf))
2001 return;
2002
2003 if(req->gzip_stream->buf_pos) {
2004 if(req->gzip_stream->buf_size)
2005 memmove(req->gzip_stream->buf, req->gzip_stream->buf + req->gzip_stream->buf_pos, req->gzip_stream->buf_size);
2006 req->gzip_stream->buf_pos = 0;
2007 }
2008
2009 res = read_gzip_data(req, req->gzip_stream->buf + req->gzip_stream->buf_size,
2010 sizeof(req->gzip_stream->buf) - req->gzip_stream->buf_size, FALSE, &len);
2011 if(res == ERROR_SUCCESS)
2012 req->gzip_stream->buf_size += len;
2013 }
2014
2015 /* return the size of data available to be read immediately (the read section must be held) */
2016 static DWORD get_avail_data( WININETHTTPREQW *req )
2017 {
2018 if (req->gzip_stream) {
2019 refill_gzip_buffer(req);
2020 return req->gzip_stream->buf_size;
2021 }
2022 if (req->read_chunked && (req->dwContentLength == ~0u || req->dwContentLength == req->dwContentRead))
2023 return 0;
2024 return min( req->read_size, req->dwContentLength - req->dwContentRead );
2025 }
2026
2027 static void HTTP_ReceiveRequestData(WININETHTTPREQW *req, BOOL first_notif)
2028 {
2029 INTERNET_ASYNC_RESULT iar;
2030
2031 TRACE("%p\n", req);
2032
2033 EnterCriticalSection( &req->read_section );
2034 if (refill_buffer( req )) {
2035 iar.dwResult = (DWORD_PTR)req->hdr.hInternet;
2036 iar.dwError = first_notif ? 0 : get_avail_data(req);
2037 }else {
2038 iar.dwResult = 0;
2039 iar.dwError = INTERNET_GetLastError();
2040 }
2041 LeaveCriticalSection( &req->read_section );
2042
2043 INTERNET_SendCallback(&req->hdr, req->hdr.dwContext, INTERNET_STATUS_REQUEST_COMPLETE, &iar,
2044 sizeof(INTERNET_ASYNC_RESULT));
2045 }
2046
2047 /* read data from the http connection (the read section must be held) */
2048 static DWORD HTTPREQ_Read(WININETHTTPREQW *req, void *buffer, DWORD size, DWORD *read, BOOL sync)
2049 {
2050 BOOL finished_reading = FALSE;
2051 int len, bytes_read = 0;
2052 DWORD ret = ERROR_SUCCESS;
2053
2054 EnterCriticalSection( &req->read_section );
2055
2056 if (req->read_chunked && (req->dwContentLength == ~0u || req->dwContentLength == req->dwContentRead))
2057 {
2058 if (!start_next_chunk( req )) goto done;
2059 }
2060
2061 if(req->gzip_stream) {
2062 if(req->gzip_stream->buf_size) {
2063 bytes_read = min(req->gzip_stream->buf_size, size);
2064 memcpy(buffer, req->gzip_stream->buf + req->gzip_stream->buf_pos, bytes_read);
2065 req->gzip_stream->buf_pos += bytes_read;
2066 req->gzip_stream->buf_size -= bytes_read;
2067 }else if(!req->read_size && !req->gzip_stream->end_of_data) {
2068 refill_buffer(req);
2069 }
2070
2071 if(size > bytes_read) {
2072 ret = read_gzip_data(req, (BYTE*)buffer+bytes_read, size-bytes_read, sync, &len);
2073 if(ret == ERROR_SUCCESS)
2074 bytes_read += len;
2075 }
2076
2077 finished_reading = req->gzip_stream->end_of_data && !req->gzip_stream->buf_size;
2078 }else {
2079 if (req->dwContentLength != ~0u) size = min( size, req->dwContentLength - req->dwContentRead );
2080
2081 if (req->read_size) {
2082 bytes_read = min( req->read_size, size );
2083 memcpy( buffer, req->read_buf + req->read_pos, bytes_read );
2084 remove_data( req, bytes_read );
2085 }
2086
2087 if (size > bytes_read && (!bytes_read || sync)) {
2088 if (NETCON_recv( &req->netConnection, (char *)buffer + bytes_read, size - bytes_read,
2089 sync ? MSG_WAITALL : 0, &len))
2090 bytes_read += len;
2091 /* always return success, even if the network layer returns an error */
2092 }
2093
2094 finished_reading = !bytes_read && req->dwContentRead == req->dwContentLength;
2095 }
2096 done:
2097 req->dwContentRead += bytes_read;
2098 *read = bytes_read;
2099
2100 TRACE( "retrieved %u bytes (%u/%u)\n", bytes_read, req->dwContentRead, req->dwContentLength );
2101 LeaveCriticalSection( &req->read_section );
2102
2103 if(ret == ERROR_SUCCESS && req->lpszCacheFile) {
2104 BOOL res;
2105 DWORD dwBytesWritten;
2106
2107 res = WriteFile(req->hCacheFile, buffer, bytes_read, &dwBytesWritten, NULL);
2108 if(!res)
2109 WARN("WriteFile failed: %u\n", GetLastError());
2110 }
2111
2112 if(finished_reading)
2113 HTTP_FinishedReading(req);
2114
2115 return ret;
2116 }
2117
2118
2119 static DWORD HTTPREQ_ReadFile(WININETHANDLEHEADER *hdr, void *buffer, DWORD size, DWORD *read)
2120 {
2121 WININETHTTPREQW *req = (WININETHTTPREQW*)hdr;
2122 return HTTPREQ_Read(req, buffer, size, read, TRUE);
2123 }
2124
2125 static void HTTPREQ_AsyncReadFileExAProc(WORKREQUEST *workRequest)
2126 {
2127 struct WORKREQ_INTERNETREADFILEEXA const *data = &workRequest->u.InternetReadFileExA;
2128 WININETHTTPREQW *req = (WININETHTTPREQW*)workRequest->hdr;
2129 INTERNET_ASYNC_RESULT iar;
2130 DWORD res;
2131
2132 TRACE("INTERNETREADFILEEXA %p\n", workRequest->hdr);
2133
2134 res = HTTPREQ_Read(req, data->lpBuffersOut->lpvBuffer,
2135 data->lpBuffersOut->dwBufferLength, &data->lpBuffersOut->dwBufferLength, TRUE);
2136
2137 iar.dwResult = res == ERROR_SUCCESS;
2138 iar.dwError = res;
2139
2140 INTERNET_SendCallback(&req->hdr, req->hdr.dwContext,
2141 INTERNET_STATUS_REQUEST_COMPLETE, &iar,
2142 sizeof(INTERNET_ASYNC_RESULT));
2143 }
2144
2145 static DWORD HTTPREQ_ReadFileExA(WININETHANDLEHEADER *hdr, INTERNET_BUFFERSA *buffers,
2146 DWORD flags, DWORD_PTR context)
2147 {
2148
2149 WININETHTTPREQW *req = (WININETHTTPREQW*)hdr;
2150 DWORD res;
2151
2152 if (flags & ~(IRF_ASYNC|IRF_NO_WAIT))
2153 FIXME("these dwFlags aren't implemented: 0x%x\n", flags & ~(IRF_ASYNC|IRF_NO_WAIT));
2154
2155 if (buffers->dwStructSize != sizeof(*buffers))
2156 return ERROR_INVALID_PARAMETER;
2157
2158 INTERNET_SendCallback(&req->hdr, req->hdr.dwContext, INTERNET_STATUS_RECEIVING_RESPONSE, NULL, 0);
2159
2160 if ((hdr->dwFlags & INTERNET_FLAG_ASYNC) && !get_avail_data(req))
2161 {
2162 WORKREQUEST workRequest;
2163
2164 if (TryEnterCriticalSection( &req->read_section ))
2165 {
2166 if (get_avail_data(req))
2167 {
2168 res = HTTPREQ_Read(req, buffers->lpvBuffer, buffers->dwBufferLength,
2169 &buffers->dwBufferLength, FALSE);
2170 LeaveCriticalSection( &req->read_section );
2171 goto done;
2172 }
2173 LeaveCriticalSection( &req->read_section );
2174 }
2175
2176 workRequest.asyncproc = HTTPREQ_AsyncReadFileExAProc;
2177 workRequest.hdr = WININET_AddRef(&req->hdr);
2178 workRequest.u.InternetReadFileExA.lpBuffersOut = buffers;
2179
2180 INTERNET_AsyncCall(&workRequest);
2181
2182 return ERROR_IO_PENDING;
2183 }
2184
2185 res = HTTPREQ_Read(req, buffers->lpvBuffer, buffers->dwBufferLength, &buffers->dwBufferLength,
2186 !(flags & IRF_NO_WAIT));
2187
2188 done:
2189 if (res == ERROR_SUCCESS) {
2190 DWORD size = buffers->dwBufferLength;
2191 INTERNET_SendCallback(&req->hdr, req->hdr.dwContext, INTERNET_STATUS_RESPONSE_RECEIVED,
2192 &size, sizeof(size));
2193 }
2194
2195 return res;
2196 }
2197
2198 static void HTTPREQ_AsyncReadFileExWProc(WORKREQUEST *workRequest)
2199 {
2200 struct WORKREQ_INTERNETREADFILEEXW const *data = &workRequest->u.InternetReadFileExW;
2201 WININETHTTPREQW *req = (WININETHTTPREQW*)workRequest->hdr;
2202 INTERNET_ASYNC_RESULT iar;
2203 DWORD res;
2204
2205 TRACE("INTERNETREADFILEEXW %p\n", workRequest->hdr);
2206
2207 res = HTTPREQ_Read(req, data->lpBuffersOut->lpvBuffer,
2208 data->lpBuffersOut->dwBufferLength, &data->lpBuffersOut->dwBufferLength, TRUE);
2209
2210 iar.dwResult = res == ERROR_SUCCESS;
2211 iar.dwError = res;
2212
2213 INTERNET_SendCallback(&req->hdr, req->hdr.dwContext,
2214 INTERNET_STATUS_REQUEST_COMPLETE, &iar,
2215 sizeof(INTERNET_ASYNC_RESULT));
2216 }
2217
2218 static DWORD HTTPREQ_ReadFileExW(WININETHANDLEHEADER *hdr, INTERNET_BUFFERSW *buffers,
2219 DWORD flags, DWORD_PTR context)
2220 {
2221
2222 WININETHTTPREQW *req = (WININETHTTPREQW*)hdr;
2223 DWORD res;
2224
2225 if (flags & ~(IRF_ASYNC|IRF_NO_WAIT))
2226 FIXME("these dwFlags aren't implemented: 0x%x\n", flags & ~(IRF_ASYNC|IRF_NO_WAIT));
2227
2228 if (buffers->dwStructSize != sizeof(*buffers))
2229 return ERROR_INVALID_PARAMETER;
2230
2231 INTERNET_SendCallback(&req->hdr, req->hdr.dwContext, INTERNET_STATUS_RECEIVING_RESPONSE, NULL, 0);
2232
2233 if (hdr->dwFlags & INTERNET_FLAG_ASYNC)
2234 {
2235 WORKREQUEST workRequest;
2236
2237 if (TryEnterCriticalSection( &req->read_section ))
2238 {
2239 if (get_avail_data(req))
2240 {
2241 res = HTTPREQ_Read(req, buffers->lpvBuffer, buffers->dwBufferLength,
2242 &buffers->dwBufferLength, FALSE);
2243 LeaveCriticalSection( &req->read_section );
2244 goto done;
2245 }
2246 LeaveCriticalSection( &req->read_section );
2247 }
2248
2249 workRequest.asyncproc = HTTPREQ_AsyncReadFileExWProc;
2250 workRequest.hdr = WININET_AddRef(&req->hdr);
2251 workRequest.u.InternetReadFileExW.lpBuffersOut = buffers;
2252
2253 INTERNET_AsyncCall(&workRequest);
2254
2255 return ERROR_IO_PENDING;
2256 }
2257
2258 res = HTTPREQ_Read(req, buffers->lpvBuffer, buffers->dwBufferLength, &buffers->dwBufferLength,
2259 !(flags & IRF_NO_WAIT));
2260
2261 done:
2262 if (res == ERROR_SUCCESS) {
2263 DWORD size = buffers->dwBufferLength;
2264 INTERNET_SendCallback(&req->hdr, req->hdr.dwContext, INTERNET_STATUS_RESPONSE_RECEIVED,
2265 &size, sizeof(size));
2266 }
2267
2268 return res;
2269 }
2270
2271 static BOOL HTTPREQ_WriteFile(WININETHANDLEHEADER *hdr, const void *buffer, DWORD size, DWORD *written)
2272 {
2273 BOOL ret;
2274 LPWININETHTTPREQW lpwhr = (LPWININETHTTPREQW)hdr;
2275
2276 INTERNET_SendCallback(&lpwhr->hdr, lpwhr->hdr.dwContext, INTERNET_STATUS_SENDING_REQUEST, NULL, 0);
2277
2278 *written = 0;
2279 if ((ret = NETCON_send(&lpwhr->netConnection, buffer, size, 0, (LPINT)written)))
2280 lpwhr->dwBytesWritten += *written;
2281
2282 INTERNET_SendCallback(&lpwhr->hdr, lpwhr->hdr.dwContext, INTERNET_STATUS_REQUEST_SENT, written, sizeof(DWORD));
2283 return ret;
2284 }
2285
2286 static void HTTPREQ_AsyncQueryDataAvailableProc(WORKREQUEST *workRequest)
2287 {
2288 WININETHTTPREQW *req = (WININETHTTPREQW*)workRequest->hdr;
2289
2290 HTTP_ReceiveRequestData(req, FALSE);
2291 }
2292
2293 static DWORD HTTPREQ_QueryDataAvailable(WININETHANDLEHEADER *hdr, DWORD *available, DWORD flags, DWORD_PTR ctx)
2294 {
2295 WININETHTTPREQW *req = (WININETHTTPREQW*)hdr;
2296
2297 TRACE("(%p %p %x %lx)\n", req, available, flags, ctx);
2298
2299 if (req->lpHttpSession->lpAppInfo->hdr.dwFlags & INTERNET_FLAG_ASYNC)
2300 {
2301 WORKREQUEST workRequest;
2302
2303 /* never wait, if we can't enter the section we queue an async request right away */
2304 if (TryEnterCriticalSection( &req->read_section ))
2305 {
2306 if ((*available = get_avail_data( req ))) goto done;
2307 if (end_of_read_data( req )) goto done;
2308 LeaveCriticalSection( &req->read_section );
2309 }
2310
2311 workRequest.asyncproc = HTTPREQ_AsyncQueryDataAvailableProc;
2312 workRequest.hdr = WININET_AddRef( &req->hdr );
2313
2314 INTERNET_AsyncCall(&workRequest);
2315
2316 return ERROR_IO_PENDING;
2317 }
2318
2319 EnterCriticalSection( &req->read_section );
2320
2321 if (!(*available = get_avail_data( req )) && !end_of_read_data( req ))
2322 {
2323 refill_buffer( req );
2324 *available = get_avail_data( req );
2325 }
2326
2327 done:
2328 if (*available == sizeof(req->read_buf) && !req->gzip_stream) /* check if we have even more pending in the socket */
2329 {
2330 DWORD extra;
2331 if (NETCON_query_data_available(&req->netConnection, &extra))
2332 *available = min( *available + extra, req->dwContentLength - req->dwContentRead );
2333 }
2334 LeaveCriticalSection( &req->read_section );
2335
2336 TRACE( "returning %u\n", *available );
2337 return ERROR_SUCCESS;
2338 }
2339
2340 static const HANDLEHEADERVtbl HTTPREQVtbl = {
2341 HTTPREQ_Destroy,
2342 HTTPREQ_CloseConnection,
2343 HTTPREQ_QueryOption,
2344 HTTPREQ_SetOption,
2345 HTTPREQ_ReadFile,
2346 HTTPREQ_ReadFileExA,
2347 HTTPREQ_ReadFileExW,
2348 HTTPREQ_WriteFile,
2349 HTTPREQ_QueryDataAvailable,
2350 NULL
2351 };
2352
2353 /***********************************************************************
2354 * HTTP_HttpOpenRequestW (internal)
2355 *
2356 * Open a HTTP request handle
2357 *
2358 * RETURNS
2359 * HINTERNET a HTTP request handle on success
2360 * NULL on failure
2361 *
2362 */
2363 HINTERNET WINAPI HTTP_HttpOpenRequestW(LPWININETHTTPSESSIONW lpwhs,
2364 LPCWSTR lpszVerb, LPCWSTR lpszObjectName, LPCWSTR lpszVersion,
2365 LPCWSTR lpszReferrer , LPCWSTR *lpszAcceptTypes,
2366 DWORD dwFlags, DWORD_PTR dwContext)
2367 {
2368 LPWININETAPPINFOW hIC = NULL;
2369 LPWININETHTTPREQW lpwhr;
2370 LPWSTR lpszHostName = NULL;
2371 HINTERNET handle = NULL;
2372 static const WCHAR szHostForm[] = {'%','s',':','%','u',0};
2373 DWORD len;
2374
2375 TRACE("-->\n");
2376
2377 assert( lpwhs->hdr.htype == WH_HHTTPSESSION );
2378 hIC = lpwhs->lpAppInfo;
2379
2380 lpwhr = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(WININETHTTPREQW));
2381 if (NULL == lpwhr)
2382 {
2383 INTERNET_SetLastError(ERROR_OUTOFMEMORY);
2384 goto lend;
2385 }
2386 lpwhr->hdr.htype = WH_HHTTPREQ;
2387 lpwhr->hdr.vtbl = &HTTPREQVtbl;
2388 lpwhr->hdr.dwFlags = dwFlags;
2389 lpwhr->hdr.dwContext = dwContext;
2390 lpwhr->hdr.refs = 1;
2391 lpwhr->hdr.lpfnStatusCB = lpwhs->hdr.lpfnStatusCB;
2392 lpwhr->hdr.dwInternalFlags = lpwhs->hdr.dwInternalFlags & INET_CALLBACKW;
2393 lpwhr->dwContentLength = ~0u;
2394 InitializeCriticalSection( &lpwhr->read_section );
2395
2396 WININET_AddRef( &lpwhs->hdr );
2397 lpwhr->lpHttpSession = lpwhs;
2398 list_add_head( &lpwhs->hdr.children, &lpwhr->hdr.entry );
2399
2400 lpszHostName = HeapAlloc(GetProcessHeap(), 0, sizeof(WCHAR) *
2401 (strlenW(lpwhs->lpszHostName) + 7 /* length of ":65535" + 1 */));
2402 if (NULL == lpszHostName)
2403 {
2404 INTERNET_SetLastError(ERROR_OUTOFMEMORY);
2405 goto lend;
2406 }
2407
2408 handle = WININET_AllocHandle( &lpwhr->hdr );
2409 if (NULL == handle)
2410 {
2411 INTERNET_SetLastError(ERROR_OUTOFMEMORY);
2412 goto lend;
2413 }
2414
2415 if (!NETCON_init(&lpwhr->netConnection, dwFlags & INTERNET_FLAG_SECURE))
2416 {
2417 InternetCloseHandle( handle );
2418 handle = NULL;
2419 goto lend;
2420 }
2421
2422 if (lpszObjectName && *lpszObjectName) {
2423 HRESULT rc;
2424
2425 len = 0;
2426 rc = UrlEscapeW(lpszObjectName, NULL, &len, URL_ESCAPE_SPACES_ONLY);
2427 if (rc != E_POINTER)
2428 len = strlenW(lpszObjectName)+1;
2429 lpwhr->lpszPath = HeapAlloc(GetProcessHeap(), 0, len*sizeof(WCHAR));
2430 rc = UrlEscapeW(lpszObjectName, lpwhr->lpszPath, &len,
2431 URL_ESCAPE_SPACES_ONLY);
2432 if (rc != S_OK)
2433 {
2434 ERR("Unable to escape string!(%s) (%d)\n",debugstr_w(lpszObjectName),rc);
2435 strcpyW(lpwhr->lpszPath,lpszObjectName);
2436 }
2437 }else {
2438 static const WCHAR slashW[] = {'/',0};
2439
2440 lpwhr->lpszPath = WININET_strdupW(slashW);
2441 }
2442
2443 if (lpszReferrer && *lpszReferrer)
2444 HTTP_ProcessHeader(lpwhr, HTTP_REFERER, lpszReferrer, HTTP_ADDREQ_FLAG_ADD | HTTP_ADDHDR_FLAG_REQ);
2445
2446 if (lpszAcceptTypes)
2447 {
2448 int i;
2449 for (i = 0; lpszAcceptTypes[i]; i++)
2450 {
2451 if (!*lpszAcceptTypes[i]) continue;
2452 HTTP_ProcessHeader(lpwhr, HTTP_ACCEPT, lpszAcceptTypes[i],
2453 HTTP_ADDHDR_FLAG_COALESCE_WITH_COMMA |
2454 HTTP_ADDHDR_FLAG_REQ |
2455 (i == 0 ? HTTP_ADDHDR_FLAG_REPLACE : 0));
2456 }
2457 }
2458
2459 lpwhr->lpszVerb = WININET_strdupW(lpszVerb && *lpszVerb ? lpszVerb : szGET);
2460
2461 if (lpszVersion)
2462 lpwhr->lpszVersion = WININET_strdupW(lpszVersion);
2463 else
2464 lpwhr->lpszVersion = WININET_strdupW(g_szHttp1_1);
2465
2466 if (lpwhs->nHostPort != INTERNET_INVALID_PORT_NUMBER &&
2467 lpwhs->nHostPort != INTERNET_DEFAULT_HTTP_PORT &&
2468 lpwhs->nHostPort != INTERNET_DEFAULT_HTTPS_PORT)
2469 {
2470 sprintfW(lpszHostName, szHostForm, lpwhs->lpszHostName, lpwhs->nHostPort);
2471 HTTP_ProcessHeader(lpwhr, hostW, lpszHostName,
2472 HTTP_ADDREQ_FLAG_ADD | HTTP_ADDHDR_FLAG_REQ);
2473 }
2474 else
2475 HTTP_ProcessHeader(lpwhr, hostW, lpwhs->lpszHostName,
2476 HTTP_ADDREQ_FLAG_ADD | HTTP_ADDHDR_FLAG_REQ);
2477
2478 if (lpwhs->nServerPort == INTERNET_INVALID_PORT_NUMBER)
2479 lpwhs->nServerPort = (dwFlags & INTERNET_FLAG_SECURE ?
2480 INTERNET_DEFAULT_HTTPS_PORT :
2481 INTERNET_DEFAULT_HTTP_PORT);
2482
2483 if (lpwhs->nHostPort == INTERNET_INVALID_PORT_NUMBER)
2484 lpwhs->nHostPort = (dwFlags & INTERNET_FLAG_SECURE ?
2485 INTERNET_DEFAULT_HTTPS_PORT :
2486 INTERNET_DEFAULT_HTTP_PORT);
2487
2488 if (NULL != hIC->lpszProxy && hIC->lpszProxy[0] != 0)
2489 HTTP_DealWithProxy( hIC, lpwhs, lpwhr );
2490
2491 INTERNET_SendCallback(&lpwhs->hdr, dwContext,
2492 INTERNET_STATUS_HANDLE_CREATED, &handle,
2493 sizeof(handle));
2494
2495 lend:
2496 HeapFree(GetProcessHeap(), 0, lpszHostName);
2497 if( lpwhr )
2498 WININET_Release( &lpwhr->hdr );
2499
2500 TRACE("<-- %p (%p)\n", handle, lpwhr);
2501 return handle;
2502 }
2503
2504 /* read any content returned by the server so that the connection can be
2505 * reused */
2506 static void HTTP_DrainContent(WININETHTTPREQW *req)
2507 {
2508 DWORD bytes_read;
2509
2510 if (!NETCON_connected(&req->netConnection)) return;
2511
2512 if (req->dwContentLength == -1)
2513 {
2514 NETCON_close(&req->netConnection);
2515 return;
2516 }
2517
2518 do
2519 {
2520 char buffer[2048];
2521 if (HTTPREQ_Read(req, buffer, sizeof(buffer), &bytes_read, TRUE) != ERROR_SUCCESS)
2522 return;
2523 } while (bytes_read);
2524 }
2525
2526 static const LPCWSTR header_lookup[] = {
2527 szMime_Version, /* HTTP_QUERY_MIME_VERSION = 0 */
2528 szContent_Type, /* HTTP_QUERY_CONTENT_TYPE = 1 */
2529 szContent_Transfer_Encoding,/* HTTP_QUERY_CONTENT_TRANSFER_ENCODING = 2 */
2530 szContent_ID, /* HTTP_QUERY_CONTENT_ID = 3 */
2531 NULL, /* HTTP_QUERY_CONTENT_DESCRIPTION = 4 */
2532 szContent_Length, /* HTTP_QUERY_CONTENT_LENGTH = 5 */
2533 szContent_Language, /* HTTP_QUERY_CONTENT_LANGUAGE = 6 */
2534 szAllow, /* HTTP_QUERY_ALLOW = 7 */
2535 szPublic, /* HTTP_QUERY_PUBLIC = 8 */
2536 szDate, /* HTTP_QUERY_DATE = 9 */
2537 szExpires, /* HTTP_QUERY_EXPIRES = 10 */
2538 szLast_Modified, /* HTTP_QUERY_LAST_MODIFIED = 11 */
2539 NULL, /* HTTP_QUERY_MESSAGE_ID = 12 */
2540 szURI, /* HTTP_QUERY_URI = 13 */
2541 szFrom, /* HTTP_QUERY_DERIVED_FROM = 14 */
2542 NULL, /* HTTP_QUERY_COST = 15 */
2543 NULL, /* HTTP_QUERY_LINK = 16 */
2544 szPragma, /* HTTP_QUERY_PRAGMA = 17 */
2545 NULL, /* HTTP_QUERY_VERSION = 18 */
2546 szStatus, /* HTTP_QUERY_STATUS_CODE = 19 */
2547 NULL, /* HTTP_QUERY_STATUS_TEXT = 20 */
2548 NULL, /* HTTP_QUERY_RAW_HEADERS = 21 */
2549 NULL, /* HTTP_QUERY_RAW_HEADERS_CRLF = 22 */
2550 szConnection, /* HTTP_QUERY_CONNECTION = 23 */
2551 szAccept, /* HTTP_QUERY_ACCEPT = 24 */
2552 szAccept_Charset, /* HTTP_QUERY_ACCEPT_CHARSET = 25 */
2553 szAccept_Encoding, /* HTTP_QUERY_ACCEPT_ENCODING = 26 */
2554 szAccept_Language, /* HTTP_QUERY_ACCEPT_LANGUAGE = 27 */
2555 szAuthorization, /* HTTP_QUERY_AUTHORIZATION = 28 */
2556 szContent_Encoding, /* HTTP_QUERY_CONTENT_ENCODING = 29 */
2557 NULL, /* HTTP_QUERY_FORWARDED = 30 */
2558 NULL, /* HTTP_QUERY_FROM = 31 */
2559 szIf_Modified_Since, /* HTTP_QUERY_IF_MODIFIED_SINCE = 32 */
2560 szLocation, /* HTTP_QUERY_LOCATION = 33 */
2561 NULL, /* HTTP_QUERY_ORIG_URI = 34 */
2562 szReferer, /* HTTP_QUERY_REFERER = 35 */
2563 szRetry_After, /* HTTP_QUERY_RETRY_AFTER = 36 */
2564 szServer, /* HTTP_QUERY_SERVER = 37 */
2565 NULL, /* HTTP_TITLE = 38 */
2566 szUser_Agent, /* HTTP_QUERY_USER_AGENT = 39 */
2567 szWWW_Authenticate, /* HTTP_QUERY_WWW_AUTHENTICATE = 40 */
2568 szProxy_Authenticate, /* HTTP_QUERY_PROXY_AUTHENTICATE = 41 */
2569 szAccept_Ranges, /* HTTP_QUERY_ACCEPT_RANGES = 42 */
2570 szSet_Cookie, /* HTTP_QUERY_SET_COOKIE = 43 */
2571 szCookie, /* HTTP_QUERY_COOKIE = 44 */
2572 NULL, /* HTTP_QUERY_REQUEST_METHOD = 45 */
2573 NULL, /* HTTP_QUERY_REFRESH = 46 */
2574 NULL, /* HTTP_QUERY_CONTENT_DISPOSITION = 47 */
2575 szAge, /* HTTP_QUERY_AGE = 48 */
2576 szCache_Control, /* HTTP_QUERY_CACHE_CONTROL = 49 */
2577 szContent_Base, /* HTTP_QUERY_CONTENT_BASE = 50 */
2578 szContent_Location, /* HTTP_QUERY_CONTENT_LOCATION = 51 */
2579 szContent_MD5, /* HTTP_QUERY_CONTENT_MD5 = 52 */
2580 szContent_Range, /* HTTP_QUERY_CONTENT_RANGE = 53 */
2581 szETag, /* HTTP_QUERY_ETAG = 54 */
2582 hostW, /* HTTP_QUERY_HOST = 55 */
2583 szIf_Match, /* HTTP_QUERY_IF_MATCH = 56 */
2584 szIf_None_Match, /* HTTP_QUERY_IF_NONE_MATCH = 57 */
2585 szIf_Range, /* HTTP_QUERY_IF_RANGE = 58 */
2586 szIf_Unmodified_Since, /* HTTP_QUERY_IF_UNMODIFIED_SINCE = 59 */
2587 szMax_Forwards, /* HTTP_QUERY_MAX_FORWARDS = 60 */
2588 szProxy_Authorization, /* HTTP_QUERY_PROXY_AUTHORIZATION = 61 */
2589 szRange, /* HTTP_QUERY_RANGE = 62 */
2590 szTransfer_Encoding, /* HTTP_QUERY_TRANSFER_ENCODING = 63 */
2591 szUpgrade, /* HTTP_QUERY_UPGRADE = 64 */
2592 szVary, /* HTTP_QUERY_VARY = 65 */
2593 szVia, /* HTTP_QUERY_VIA = 66 */
2594 szWarning, /* HTTP_QUERY_WARNING = 67 */
2595 szExpect, /* HTTP_QUERY_EXPECT = 68 */
2596 szProxy_Connection, /* HTTP_QUERY_PROXY_CONNECTION = 69 */
2597 szUnless_Modified_Since, /* HTTP_QUERY_UNLESS_MODIFIED_SINCE = 70 */
2598 };
2599
2600 #define LAST_TABLE_HEADER (sizeof(header_lookup)/sizeof(header_lookup[0]))
2601
2602 /***********************************************************************
2603 * HTTP_HttpQueryInfoW (internal)
2604 */
2605 static BOOL HTTP_HttpQueryInfoW( LPWININETHTTPREQW lpwhr, DWORD dwInfoLevel,
2606 LPVOID lpBuffer, LPDWORD lpdwBufferLength, LPDWORD lpdwIndex)
2607 {
2608 LPHTTPHEADERW lphttpHdr = NULL;
2609 BOOL bSuccess = FALSE;
2610 BOOL request_only = dwInfoLevel & HTTP_QUERY_FLAG_REQUEST_HEADERS;
2611 INT requested_index = lpdwIndex ? *lpdwIndex : 0;
2612 DWORD level = (dwInfoLevel & ~HTTP_QUERY_MODIFIER_FLAGS_MASK);
2613 INT index = -1;
2614
2615 /* Find requested header structure */
2616 switch (level)
2617 {
2618 case HTTP_QUERY_CUSTOM:
2619 if (!lpBuffer) return FALSE;
2620 index = HTTP_GetCustomHeaderIndex(lpwhr, lpBuffer, requested_index, request_only);
2621 break;
2622
2623 case HTTP_QUERY_CONTENT_LENGTH:
2624 if(lpwhr->gzip_stream) {
2625 INTERNET_SetLastError(ERROR_HTTP_HEADER_NOT_FOUND);
2626 return FALSE;
2627 }
2628
2629 index = HTTP_GetCustomHeaderIndex(lpwhr, header_lookup[level],
2630 requested_index,request_only);
2631 break;
2632
2633 case HTTP_QUERY_RAW_HEADERS_CRLF:
2634 {
2635 LPWSTR headers;
2636 DWORD len = 0;
2637 BOOL ret = FALSE;
2638
2639 if (request_only)
2640 headers = HTTP_BuildHeaderRequestString(lpwhr, lpwhr->lpszVerb, lpwhr->lpszPath, lpwhr->lpszVersion);
2641 else
2642 headers = lpwhr->lpszRawHeaders;
2643
2644 if (headers)
2645 len = strlenW(headers) * sizeof(WCHAR);
2646
2647 if (len + sizeof(WCHAR) > *lpdwBufferLength)
2648 {
2649 len += sizeof(WCHAR);
2650 INTERNET_SetLastError(ERROR_INSUFFICIENT_BUFFER);
2651 ret = FALSE;
2652 }
2653 else if (lpBuffer)
2654 {
2655 if (headers)
2656 memcpy(lpBuffer, headers, len + sizeof(WCHAR));
2657 else
2658 {
2659 len = strlenW(szCrLf) * sizeof(WCHAR);
2660 memcpy(lpBuffer, szCrLf, sizeof(szCrLf));
2661 }
2662 TRACE("returning data: %s\n", debugstr_wn(lpBuffer, len / sizeof(WCHAR)));
2663 ret = TRUE;
2664 }
2665 *lpdwBufferLength = len;
2666
2667 if (request_only)
2668 HeapFree(GetProcessHeap(), 0, headers);
2669 return ret;
2670 }
2671 case HTTP_QUERY_RAW_HEADERS:
2672 {
2673 LPWSTR * ppszRawHeaderLines = HTTP_Tokenize(lpwhr->lpszRawHeaders, szCrLf);
2674 DWORD i, size = 0;
2675 LPWSTR pszString = lpBuffer;
2676
2677 for (i = 0; ppszRawHeaderLines[i]; i++)
2678 size += strlenW(ppszRawHeaderLines[i]) + 1;
2679
2680 if (size + 1 > *lpdwBufferLength/sizeof(WCHAR))
2681 {
2682 HTTP_FreeTokens(ppszRawHeaderLines);
2683 *lpdwBufferLength = (size + 1) * sizeof(WCHAR);
2684 INTERNET_SetLastError(ERROR_INSUFFICIENT_BUFFER);
2685 return FALSE;
2686 }
2687 if (pszString)
2688 {
2689 for (i = 0; ppszRawHeaderLines[i]; i++)
2690 {
2691 DWORD len = strlenW(ppszRawHeaderLines[i]);
2692 memcpy(pszString, ppszRawHeaderLines[i], (len+1)*sizeof(WCHAR));
2693 pszString += len+1;
2694 }
2695 *pszString = '\0';
2696 TRACE("returning data: %s\n", debugstr_wn(lpBuffer, size));
2697 }
2698 *lpdwBufferLength = size * sizeof(WCHAR);
2699 HTTP_FreeTokens(ppszRawHeaderLines);
2700
2701 return TRUE;
2702 }
2703 case HTTP_QUERY_STATUS_TEXT:
2704 if (lpwhr->lpszStatusText)
2705 {
2706 DWORD len = strlenW(lpwhr->lpszStatusText);
2707 if (len + 1 > *lpdwBufferLength/sizeof(WCHAR))
2708 {
2709 *lpdwBufferLength = (len + 1) * sizeof(WCHAR);
2710 INTERNET_SetLastError(ERROR_INSUFFICIENT_BUFFER);
2711 return FALSE;
2712 }
2713 if (lpBuffer)
2714 {
2715 memcpy(lpBuffer, lpwhr->lpszStatusText, (len + 1) * sizeof(WCHAR));
2716 TRACE("returning data: %s\n", debugstr_wn(lpBuffer, len));
2717 }
2718 *lpdwBufferLength = len * sizeof(WCHAR);
2719 return TRUE;
2720 }
2721 break;
2722 case HTTP_QUERY_VERSION:
2723 if (lpwhr->lpszVersion)
2724 {
2725 DWORD len = strlenW(lpwhr->lpszVersion);
2726 if (len + 1 > *lpdwBufferLength/sizeof(WCHAR))
2727 {
2728 *lpdwBufferLength = (len + 1) * sizeof(WCHAR);
2729 INTERNET_SetLastError(ERROR_INSUFFICIENT_BUFFER);
2730 return FALSE;
2731 }
2732 if (lpBuffer)
2733 {
2734 memcpy(lpBuffer, lpwhr->lpszVersion, (len + 1) * sizeof(WCHAR));
2735 TRACE("returning data: %s\n", debugstr_wn(lpBuffer, len));
2736 }
2737 *lpdwBufferLength = len * sizeof(WCHAR);
2738 return TRUE;
2739 }
2740 break;
2741 case HTTP_QUERY_CONTENT_ENCODING:
2742 index = HTTP_GetCustomHeaderIndex(lpwhr, header_lookup[lpwhr->gzip_stream ? HTTP_QUERY_CONTENT_TYPE : level],
2743 requested_index,request_only);
2744 break;
2745 default:
2746 assert (LAST_TABLE_HEADER == (HTTP_QUERY_UNLESS_MODIFIED_SINCE + 1));
2747
2748 if (level < LAST_TABLE_HEADER && header_lookup[level])
2749 index = HTTP_GetCustomHeaderIndex(lpwhr, header_lookup[level],
2750 requested_index,request_only);
2751 }
2752
2753 if (index >= 0)
2754 lphttpHdr = &lpwhr->pCustHeaders[index];
2755
2756 /* Ensure header satisfies requested attributes */
2757 if (!lphttpHdr ||
2758 ((dwInfoLevel & HTTP_QUERY_FLAG_REQUEST_HEADERS) &&
2759 (~lphttpHdr->wFlags & HDR_ISREQUEST)))
2760 {
2761 INTERNET_SetLastError(ERROR_HTTP_HEADER_NOT_FOUND);
2762 return bSuccess;
2763 }
2764
2765 if (lpdwIndex && level != HTTP_QUERY_STATUS_CODE) (*lpdwIndex)++;
2766
2767 /* coalesce value to requested type */
2768 if (dwInfoLevel & HTTP_QUERY_FLAG_NUMBER && lpBuffer)
2769 {
2770 *(int *)lpBuffer = atoiW(lphttpHdr->lpszValue);
2771 TRACE(" returning number: %d\n", *(int *)lpBuffer);
2772 bSuccess = TRUE;
2773 }
2774 else if (dwInfoLevel & HTTP_QUERY_FLAG_SYSTEMTIME && lpBuffer)
2775 {
2776 time_t tmpTime;
2777 struct tm tmpTM;
2778 SYSTEMTIME *STHook;
2779
2780 tmpTime = ConvertTimeString(lphttpHdr->lpszValue);
2781
2782 tmpTM = *gmtime(&tmpTime);
2783 STHook = (SYSTEMTIME *)lpBuffer;
2784 STHook->wDay = tmpTM.tm_mday;
2785 STHook->wHour = tmpTM.tm_hour;
2786 STHook->wMilliseconds = 0;
2787 STHook->wMinute = tmpTM.tm_min;
2788 STHook->wDayOfWeek = tmpTM.tm_wday;
2789 STHook->wMonth = tmpTM.tm_mon + 1;
2790 STHook->wSecond = tmpTM.tm_sec;
2791 STHook->wYear = tmpTM.tm_year;
2792 bSuccess = TRUE;
2793
2794 TRACE(" returning time: %04d/%02d/%02d - %d - %02d:%02d:%02d.%02d\n",
2795 STHook->wYear, STHook->wMonth, STHook->wDay, STHook->wDayOfWeek,
2796 STHook->wHour, STHook->wMinute, STHook->wSecond, STHook->wMilliseconds);
2797 }
2798 else if (lphttpHdr->lpszValue)
2799 {
2800 DWORD len = (strlenW(lphttpHdr->lpszValue) + 1) * sizeof(WCHAR);
2801
2802 if (len > *lpdwBufferLength)
2803 {
2804 *lpdwBufferLength = len;
2805 INTERNET_SetLastError(ERROR_INSUFFICIENT_BUFFER);
2806 return bSuccess;
2807 }
2808 if (lpBuffer)
2809 {
2810 memcpy(lpBuffer, lphttpHdr->lpszValue, len);
2811 TRACE(" returning string: %s\n", debugstr_w(lpBuffer));
2812 }
2813 *lpdwBufferLength = len - sizeof(WCHAR);
2814 bSuccess = TRUE;
2815 }
2816 return bSuccess;
2817 }
2818
2819 /***********************************************************************
2820 * HttpQueryInfoW (WININET.@)
2821 *
2822 * Queries for information about an HTTP request
2823 *
2824 * RETURNS
2825 * TRUE on success
2826 * FALSE on failure
2827 *
2828 */
2829 BOOL WINAPI HttpQueryInfoW(HINTERNET hHttpRequest, DWORD dwInfoLevel,
2830 LPVOID lpBuffer, LPDWORD lpdwBufferLength, LPDWORD lpdwIndex)
2831 {
2832 BOOL bSuccess = FALSE;
2833 LPWININETHTTPREQW lpwhr;
2834
2835 if (TRACE_ON(wininet)) {
2836 #define FE(x) { x, #x }
2837 static const wininet_flag_info query_flags[] = {
2838 FE(HTTP_QUERY_MIME_VERSION),
2839 FE(HTTP_QUERY_CONTENT_TYPE),
2840 FE(HTTP_QUERY_CONTENT_TRANSFER_ENCODING),
2841 FE(HTTP_QUERY_CONTENT_ID),
2842 FE(HTTP_QUERY_CONTENT_DESCRIPTION),
2843 FE(HTTP_QUERY_CONTENT_LENGTH),
2844 FE(HTTP_QUERY_CONTENT_LANGUAGE),
2845 FE(HTTP_QUERY_ALLOW),
2846 FE(HTTP_QUERY_PUBLIC),
2847 FE(HTTP_QUERY_DATE),
2848 FE(HTTP_QUERY_EXPIRES),
2849 FE(HTTP_QUERY_LAST_MODIFIED),
2850 FE(HTTP_QUERY_MESSAGE_ID),
2851 FE(HTTP_QUERY_URI),
2852 FE(HTTP_QUERY_DERIVED_FROM),
2853 FE(HTTP_QUERY_COST),
2854 FE(HTTP_QUERY_LINK),
2855 FE(HTTP_QUERY_PRAGMA),
2856 FE(HTTP_QUERY_VERSION),
2857 FE(HTTP_QUERY_STATUS_CODE),
2858 FE(HTTP_QUERY_STATUS_TEXT),
2859 FE(HTTP_QUERY_RAW_HEADERS),
2860 FE(HTTP_QUERY_RAW_HEADERS_CRLF),
2861 FE(HTTP_QUERY_CONNECTION),
2862 FE(HTTP_QUERY_ACCEPT),
2863 FE(HTTP_QUERY_ACCEPT_CHARSET),
2864 FE(HTTP_QUERY_ACCEPT_ENCODING),
2865 FE(HTTP_QUERY_ACCEPT_LANGUAGE),
2866 FE(HTTP_QUERY_AUTHORIZATION),
2867 FE(HTTP_QUERY_CONTENT_ENCODING),
2868 FE(HTTP_QUERY_FORWARDED),
2869 FE(HTTP_QUERY_FROM),
2870 FE(HTTP_QUERY_IF_MODIFIED_SINCE),
2871 FE(HTTP_QUERY_LOCATION),
2872 FE(HTTP_QUERY_ORIG_URI),
2873 FE(HTTP_QUERY_REFERER),
2874 FE(HTTP_QUERY_RETRY_AFTER),
2875 FE(HTTP_QUERY_SERVER),
2876 FE(HTTP_QUERY_TITLE),
2877 FE(HTTP_QUERY_USER_AGENT),
2878 FE(HTTP_QUERY_WWW_AUTHENTICATE),
2879 FE(HTTP_QUERY_PROXY_AUTHENTICATE),
2880 FE(HTTP_QUERY_ACCEPT_RANGES),
2881 FE(HTTP_QUERY_SET_COOKIE),
2882 FE(HTTP_QUERY_COOKIE),
2883 FE(HTTP_QUERY_REQUEST_METHOD),
2884 FE(HTTP_QUERY_REFRESH),
2885 FE(HTTP_QUERY_CONTENT_DISPOSITION),
2886 FE(HTTP_QUERY_AGE),
2887 FE(HTTP_QUERY_CACHE_CONTROL),
2888 FE(HTTP_QUERY_CONTENT_BASE),
2889 FE(HTTP_QUERY_CONTENT_LOCATION),
2890 FE(HTTP_QUERY_CONTENT_MD5),
2891 FE(HTTP_QUERY_CONTENT_RANGE),
2892 FE(HTTP_QUERY_ETAG),
2893 FE(HTTP_QUERY_HOST),
2894 FE(HTTP_QUERY_IF_MATCH),
2895 FE(HTTP_QUERY_IF_NONE_MATCH),
2896 FE(HTTP_QUERY_IF_RANGE),
2897 FE(HTTP_QUERY_IF_UNMODIFIED_SINCE),
2898 FE(HTTP_QUERY_MAX_FORWARDS),
2899 FE(HTTP_QUERY_PROXY_AUTHORIZATION),
2900 FE(HTTP_QUERY_RANGE),
2901 FE(HTTP_QUERY_TRANSFER_ENCODING),
2902 FE(HTTP_QUERY_UPGRADE),
2903 FE(HTTP_QUERY_VARY),
2904 FE(HTTP_QUERY_VIA),
2905 FE(HTTP_QUERY_WARNING),
2906 FE(HTTP_QUERY_CUSTOM)
2907 };
2908 static const wininet_flag_info modifier_flags[] = {
2909 FE(HTTP_QUERY_FLAG_REQUEST_HEADERS),
2910 FE(HTTP_QUERY_FLAG_SYSTEMTIME),
2911 FE(HTTP_QUERY_FLAG_NUMBER),
2912 FE(HTTP_QUERY_FLAG_COALESCE)
2913 };
2914 #undef FE
2915 DWORD info_mod = dwInfoLevel & HTTP_QUERY_MODIFIER_FLAGS_MASK;
2916 DWORD info = dwInfoLevel & HTTP_QUERY_HEADER_MASK;
2917 DWORD i;
2918
2919 TRACE("(%p, 0x%08x)--> %d\n", hHttpRequest, dwInfoLevel, dwInfoLevel);
2920 TRACE(" Attribute:");
2921 for (i = 0; i < (sizeof(query_flags) / sizeof(query_flags[0])); i++) {
2922 if (query_flags[i].val == info) {
2923 TRACE(" %s", query_flags[i].name);
2924 break;
2925 }
2926 }
2927 if (i == (sizeof(query_flags) / sizeof(query_flags[0]))) {
2928 TRACE(" Unknown (%08x)", info);
2929 }
2930
2931 TRACE(" Modifier:");
2932 for (i = 0; i < (sizeof(modifier_flags) / sizeof(modifier_flags[0])); i++) {
2933 if (modifier_flags[i].val & info_mod) {
2934 TRACE(" %s", modifier_flags[i].name);
2935 info_mod &= ~ modifier_flags[i].val;
2936 }
2937 }
2938
2939 if (info_mod) {
2940 TRACE(" Unknown (%08x)", info_mod);
2941 }
2942 TRACE("\n");
2943 }
2944
2945 lpwhr = (LPWININETHTTPREQW) WININET_GetObject( hHttpRequest );
2946 if (NULL == lpwhr || lpwhr->hdr.htype != WH_HHTTPREQ)
2947 {
2948 INTERNET_SetLastError(ERROR_INTERNET_INCORRECT_HANDLE_TYPE);
2949 goto lend;
2950 }
2951
2952 if (lpBuffer == NULL)
2953 *lpdwBufferLength = 0;
2954 bSuccess = HTTP_HttpQueryInfoW( lpwhr, dwInfoLevel,
2955 lpBuffer, lpdwBufferLength, lpdwIndex);
2956
2957 lend:
2958 if( lpwhr )
2959 WININET_Release( &lpwhr->hdr );
2960
2961 TRACE("%d <--\n", bSuccess);
2962 return bSuccess;
2963 }
2964
2965 /***********************************************************************
2966 * HttpQueryInfoA (WININET.@)
2967 *
2968 * Queries for information about an HTTP request
2969 *
2970 * RETURNS
2971 * TRUE on success
2972 * FALSE on failure
2973 *
2974 */
2975 BOOL WINAPI HttpQueryInfoA(HINTERNET hHttpRequest, DWORD dwInfoLevel,
2976 LPVOID lpBuffer, LPDWORD lpdwBufferLength, LPDWORD lpdwIndex)
2977 {
2978 BOOL result;
2979 DWORD len;
2980 WCHAR* bufferW;
2981
2982 if((dwInfoLevel & HTTP_QUERY_FLAG_NUMBER) ||
2983 (dwInfoLevel & HTTP_QUERY_FLAG_SYSTEMTIME))
2984 {
2985 return HttpQueryInfoW( hHttpRequest, dwInfoLevel, lpBuffer,
2986 lpdwBufferLength, lpdwIndex );
2987 }
2988
2989 if (lpBuffer)
2990 {
2991 DWORD alloclen;
2992 len = (*lpdwBufferLength)*sizeof(WCHAR);
2993 if ((dwInfoLevel & HTTP_QUERY_HEADER_MASK) == HTTP_QUERY_CUSTOM)
2994 {
2995 alloclen = MultiByteToWideChar( CP_ACP, 0, lpBuffer, -1, NULL, 0 ) * sizeof(WCHAR);
2996 if (alloclen < len)
2997 alloclen = len;
2998 }
2999 else
3000 alloclen = len;
3001 bufferW = HeapAlloc( GetProcessHeap(), 0, alloclen );
3002 /* buffer is in/out because of HTTP_QUERY_CUSTOM */
3003 if ((dwInfoLevel & HTTP_QUERY_HEADER_MASK) == HTTP_QUERY_CUSTOM)
3004 MultiByteToWideChar( CP_ACP, 0, lpBuffer, -1, bufferW, alloclen / sizeof(WCHAR) );
3005 } else
3006 {
3007 bufferW = NULL;
3008 len = 0;
3009 }
3010
3011 result = HttpQueryInfoW( hHttpRequest, dwInfoLevel, bufferW,
3012 &len, lpdwIndex );
3013 if( result )
3014 {
3015 len = WideCharToMultiByte( CP_ACP,0, bufferW, len / sizeof(WCHAR) + 1,
3016 lpBuffer, *lpdwBufferLength, NULL, NULL );
3017 *lpdwBufferLength = len - 1;
3018
3019 TRACE("lpBuffer: %s\n", debugstr_a(lpBuffer));
3020 }
3021 else
3022 /* since the strings being returned from HttpQueryInfoW should be
3023 * only ASCII characters, it is reasonable to assume that all of
3024 * the Unicode characters can be reduced to a single byte */
3025 *lpdwBufferLength = len / sizeof(WCHAR);
3026
3027 HeapFree(GetProcessHeap(), 0, bufferW );
3028
3029 return result;
3030 }
3031
3032 /***********************************************************************
3033 * HttpSendRequestExA (WININET.@)
3034 *
3035 * Sends the specified request to the HTTP server and allows chunked
3036 * transfers.
3037 *
3038 * RETURNS
3039 * Success: TRUE
3040 * Failure: FALSE, call GetLastError() for more information.
3041 */
3042 BOOL WINAPI HttpSendRequestExA(HINTERNET hRequest,
3043 LPINTERNET_BUFFERSA lpBuffersIn,
3044 LPINTERNET_BUFFERSA lpBuffersOut,
3045 DWORD dwFlags, DWORD_PTR dwContext)
3046 {
3047 INTERNET_BUFFERSW BuffersInW;
3048 BOOL rc = FALSE;
3049 DWORD headerlen;
3050 LPWSTR header = NULL;
3051
3052 TRACE("(%p, %p, %p, %08x, %08lx)\n", hRequest, lpBuffersIn,
3053 lpBuffersOut, dwFlags, dwContext);
3054
3055 if (lpBuffersIn)
3056 {
3057 BuffersInW.dwStructSize = sizeof(LPINTERNET_BUFFERSW);
3058 if (lpBuffersIn->lpcszHeader)
3059 {
3060 headerlen = MultiByteToWideChar(CP_ACP,0,lpBuffersIn->lpcszHeader,
3061 lpBuffersIn->dwHeadersLength,0,0);
3062 header = HeapAlloc(GetProcessHeap(),0,headerlen*sizeof(WCHAR));
3063 if (!(BuffersInW.lpcszHeader = header))
3064 {
3065 INTERNET_SetLastError(ERROR_OUTOFMEMORY);
3066 return FALSE;
3067 }
3068 BuffersInW.dwHeadersLength = MultiByteToWideChar(CP_ACP, 0,
3069 lpBuffersIn->lpcszHeader, lpBuffersIn->dwHeadersLength,
3070 header, headerlen);
3071 }
3072 else
3073 BuffersInW.lpcszHeader = NULL;
3074 BuffersInW.dwHeadersTotal = lpBuffersIn->dwHeadersTotal;
3075 BuffersInW.lpvBuffer = lpBuffersIn->lpvBuffer;
3076 BuffersInW.dwBufferLength = lpBuffersIn->dwBufferLength;
3077 BuffersInW.dwBufferTotal = lpBuffersIn->dwBufferTotal;
3078 BuffersInW.Next = NULL;
3079 }
3080
3081 rc = HttpSendRequestExW(hRequest, lpBuffersIn ? &BuffersInW : NULL, NULL, dwFlags, dwContext);
3082
3083 HeapFree(GetProcessHeap(),0,header);
3084
3085 return rc;
3086 }
3087
3088 /***********************************************************************
3089 * HttpSendRequestExW (WININET.@)
3090 *
3091 * Sends the specified request to the HTTP server and allows chunked
3092 * transfers
3093 *
3094 * RETURNS
3095 * Success: TRUE
3096 * Failure: FALSE, call GetLastError() for more information.
3097 */
3098 BOOL WINAPI HttpSendRequestExW(HINTERNET hRequest,
3099 LPINTERNET_BUFFERSW lpBuffersIn,
3100 LPINTERNET_BUFFERSW lpBuffersOut,
3101 DWORD dwFlags, DWORD_PTR dwContext)
3102 {
3103 BOOL ret = FALSE;
3104 LPWININETHTTPREQW lpwhr;
3105 LPWININETHTTPSESSIONW lpwhs;
3106 LPWININETAPPINFOW hIC;
3107
3108 TRACE("(%p, %p, %p, %08x, %08lx)\n", hRequest, lpBuffersIn,
3109 lpBuffersOut, dwFlags, dwContext);
3110
3111 lpwhr = (LPWININETHTTPREQW) WININET_GetObject( hRequest );
3112
3113 if (NULL == lpwhr || lpwhr->hdr.htype != WH_HHTTPREQ)
3114 {
3115 INTERNET_SetLastError(ERROR_INTERNET_INCORRECT_HANDLE_TYPE);
3116 goto lend;
3117 }
3118
3119 lpwhs = lpwhr->lpHttpSession;
3120 assert(lpwhs->hdr.htype == WH_HHTTPSESSION);
3121 hIC = lpwhs->lpAppInfo;
3122 assert(hIC->hdr.htype == WH_HINIT);
3123
3124 if (hIC->hdr.dwFlags & INTERNET_FLAG_ASYNC)
3125 {
3126 WORKREQUEST workRequest;
3127 struct WORKREQ_HTTPSENDREQUESTW *req;
3128
3129 workRequest.asyncproc = AsyncHttpSendRequestProc;
3130 workRequest.hdr = WININET_AddRef( &lpwhr->hdr );
3131 req = &workRequest.u.HttpSendRequestW;
3132 if (lpBuffersIn)
3133 {
3134 if (lpBuffersIn->lpcszHeader)
3135 /* FIXME: this should use dwHeadersLength or may not be necessary at all */
3136 req->lpszHeader = WININET_strdupW(lpBuffersIn->lpcszHeader);
3137 else
3138 req->lpszHeader = NULL;
3139 req->dwHeaderLength = lpBuffersIn->dwHeadersLength;
3140 req->lpOptional = lpBuffersIn->lpvBuffer;
3141 req->dwOptionalLength = lpBuffersIn->dwBufferLength;
3142 req->dwContentLength = lpBuffersIn->dwBufferTotal;
3143 }
3144 else
3145 {
3146 req->lpszHeader = NULL;
3147 req->dwHeaderLength = 0;
3148 req->lpOptional = NULL;
3149 req->dwOptionalLength = 0;
3150 req->dwContentLength = 0;
3151 }
3152
3153 req->bEndRequest = FALSE;
3154
3155 INTERNET_AsyncCall(&workRequest);
3156 /*
3157 * This is from windows.
3158 */
3159 INTERNET_SetLastError(ERROR_IO_PENDING);
3160 }
3161 else
3162 {
3163 if (lpBuffersIn)
3164 ret = HTTP_HttpSendRequestW(lpwhr, lpBuffersIn->lpcszHeader, lpBuffersIn->dwHeadersLength,
3165 lpBuffersIn->lpvBuffer, lpBuffersIn->dwBufferLength,
3166 lpBuffersIn->dwBufferTotal, FALSE);
3167 else
3168 ret = HTTP_HttpSendRequestW(lpwhr, NULL, 0, NULL, 0, 0, FALSE);
3169 }
3170
3171 lend:
3172 if ( lpwhr )
3173 WININET_Release( &lpwhr->hdr );
3174
3175 TRACE("<---\n");
3176 return ret;
3177 }
3178
3179 /***********************************************************************
3180 * HttpSendRequestW (WININET.@)
3181 *
3182 * Sends the specified request to the HTTP server
3183 *
3184 * RETURNS
3185 * TRUE on success
3186 * FALSE on failure
3187 *
3188 */
3189 BOOL WINAPI HttpSendRequestW(HINTERNET hHttpRequest, LPCWSTR lpszHeaders,
3190 DWORD dwHeaderLength, LPVOID lpOptional ,DWORD dwOptionalLength)
3191 {
3192 LPWININETHTTPREQW lpwhr;
3193 LPWININETHTTPSESSIONW lpwhs = NULL;
3194 LPWININETAPPINFOW hIC = NULL;
3195 BOOL r;
3196
3197 TRACE("%p, %s, %i, %p, %i)\n", hHttpRequest,
3198 debugstr_wn(lpszHeaders, dwHeaderLength), dwHeaderLength, lpOptional, dwOptionalLength);
3199
3200 lpwhr = (LPWININETHTTPREQW) WININET_GetObject( hHttpRequest );
3201 if (NULL == lpwhr || lpwhr->hdr.htype != WH_HHTTPREQ)
3202 {
3203 INTERNET_SetLastError(ERROR_INTERNET_INCORRECT_HANDLE_TYPE);
3204 r = FALSE;
3205 goto lend;
3206 }
3207
3208 lpwhs = lpwhr->lpHttpSession;
3209 if (NULL == lpwhs || lpwhs->hdr.htype != WH_HHTTPSESSION)
3210 {
3211 INTERNET_SetLastError(ERROR_INTERNET_INCORRECT_HANDLE_TYPE);
3212 r = FALSE;
3213 goto lend;
3214 }
3215
3216 hIC = lpwhs->lpAppInfo;
3217 if (NULL == hIC || hIC->hdr.htype != WH_HINIT)
3218 {
3219 INTERNET_SetLastError(ERROR_INTERNET_INCORRECT_HANDLE_TYPE);
3220 r = FALSE;
3221 goto lend;
3222 }
3223
3224 if (hIC->hdr.dwFlags & INTERNET_FLAG_ASYNC)
3225 {
3226 WORKREQUEST workRequest;
3227 struct WORKREQ_HTTPSENDREQUESTW *req;
3228
3229 workRequest.asyncproc = AsyncHttpSendRequestProc;
3230 workRequest.hdr = WININET_AddRef( &lpwhr->hdr );
3231 req = &workRequest.u.HttpSendRequestW;
3232 if (lpszHeaders)
3233 {
3234 DWORD size;
3235
3236 if (dwHeaderLength == ~0u) size = (strlenW(lpszHeaders) + 1) * sizeof(WCHAR);
3237 else size = dwHeaderLength * sizeof(WCHAR);
3238
3239 req->lpszHeader = HeapAlloc(GetProcessHeap(), 0, size);
3240 memcpy(req->lpszHeader, lpszHeaders, size);
3241 }
3242 else
3243 req->lpszHeader = 0;
3244 req->dwHeaderLength = dwHeaderLength;
3245 req->lpOptional = lpOptional;
3246 req->dwOptionalLength = dwOptionalLength;
3247 req->dwContentLength = dwOptionalLength;
3248 req->bEndRequest = TRUE;
3249
3250 INTERNET_AsyncCall(&workRequest);
3251 /*
3252 * This is from windows.
3253 */
3254 INTERNET_SetLastError(ERROR_IO_PENDING);
3255 r = FALSE;
3256 }
3257 else
3258 {
3259 r = HTTP_HttpSendRequestW(lpwhr, lpszHeaders,
3260 dwHeaderLength, lpOptional, dwOptionalLength,
3261 dwOptionalLength, TRUE);
3262 }
3263 lend:
3264 if( lpwhr )
3265 WININET_Release( &lpwhr->hdr );
3266 return r;
3267 }
3268
3269 /***********************************************************************
3270 * HttpSendRequestA (WININET.@)
3271 *
3272 * Sends the specified request to the HTTP server
3273 *
3274 * RETURNS
3275 * TRUE on success
3276 * FALSE on failure
3277 *
3278 */
3279 BOOL WINAPI HttpSendRequestA(HINTERNET hHttpRequest, LPCSTR lpszHeaders,
3280 DWORD dwHeaderLength, LPVOID lpOptional ,DWORD dwOptionalLength)
3281 {
3282 BOOL result;
3283 LPWSTR szHeaders=NULL;
3284 DWORD nLen=dwHeaderLength;
3285 if(lpszHeaders!=NULL)
3286 {
3287 nLen=MultiByteToWideChar(CP_ACP,0,lpszHeaders,dwHeaderLength,NULL,0);
3288 szHeaders=HeapAlloc(GetProcessHeap(),0,nLen*sizeof(WCHAR));
3289 MultiByteToWideChar(CP_ACP,0,lpszHeaders,dwHeaderLength,szHeaders,nLen);
3290 }
3291 result=HttpSendRequestW(hHttpRequest, szHeaders, nLen, lpOptional, dwOptionalLength);
3292 HeapFree(GetProcessHeap(),0,szHeaders);
3293 return result;
3294 }
3295
3296 /***********************************************************************
3297 * HTTP_GetRedirectURL (internal)
3298 */
3299 static LPWSTR HTTP_GetRedirectURL(LPWININETHTTPREQW lpwhr, LPCWSTR lpszUrl)
3300 {
3301 static WCHAR szHttp[] = {'h','t','t','p',0};
3302 static WCHAR szHttps[] = {'h','t','t','p','s',0};
3303 LPWININETHTTPSESSIONW lpwhs = lpwhr->lpHttpSession;
3304 URL_COMPONENTSW urlComponents;
3305 DWORD url_length = 0;
3306 LPWSTR orig_url;
3307 LPWSTR combined_url;
3308
3309 urlComponents.dwStructSize = sizeof(URL_COMPONENTSW);
3310 urlComponents.lpszScheme = (lpwhr->hdr.dwFlags & INTERNET_FLAG_SECURE) ? szHttps : szHttp;
3311 urlComponents.dwSchemeLength = 0;
3312 urlComponents.lpszHostName = lpwhs->lpszHostName;
3313 urlComponents.dwHostNameLength = 0;
3314 urlComponents.nPort = lpwhs->nHostPort;
3315 urlComponents.lpszUserName = lpwhs->lpszUserName;
3316 urlComponents.dwUserNameLength = 0;
3317 urlComponents.lpszPassword = NULL;
3318 urlComponents.dwPasswordLength = 0;
3319 urlComponents.lpszUrlPath = lpwhr->lpszPath;
3320 urlComponents.dwUrlPathLength = 0;
3321 urlComponents.lpszExtraInfo = NULL;
3322 urlComponents.dwExtraInfoLength = 0;
3323
3324 if (!InternetCreateUrlW(&urlComponents, 0, NULL, &url_length) &&
3325 (GetLastError() != ERROR_INSUFFICIENT_BUFFER))
3326 return NULL;
3327
3328 orig_url = HeapAlloc(GetProcessHeap(), 0, url_length);
3329
3330 /* convert from bytes to characters */
3331 url_length = url_length / sizeof(WCHAR) - 1;
3332 if (!InternetCreateUrlW(&urlComponents, 0, orig_url, &url_length))
3333 {
3334 HeapFree(GetProcessHeap(), 0, orig_url);
3335 return NULL;
3336 }
3337
3338 url_length = 0;
3339 if (!InternetCombineUrlW(orig_url, lpszUrl, NULL, &url_length, ICU_ENCODE_SPACES_ONLY) &&
3340 (GetLastError() != ERROR_INSUFFICIENT_BUFFER))
3341 {
3342 HeapFree(GetProcessHeap(), 0, orig_url);
3343 return NULL;
3344 }
3345 combined_url = HeapAlloc(GetProcessHeap(), 0, url_length * sizeof(WCHAR));
3346
3347 if (!InternetCombineUrlW(orig_url, lpszUrl, combined_url, &url_length, ICU_ENCODE_SPACES_ONLY))
3348 {
3349 HeapFree(GetProcessHeap(), 0, orig_url);
3350 HeapFree(GetProcessHeap(), 0, combined_url);
3351 return NULL;
3352 }
3353 HeapFree(GetProcessHeap(), 0, orig_url);
3354 return combined_url;
3355 }
3356
3357
3358 /***********************************************************************
3359 * HTTP_HandleRedirect (internal)
3360 */
3361 static BOOL HTTP_HandleRedirect(LPWININETHTTPREQW lpwhr, LPCWSTR lpszUrl)
3362 {
3363 LPWININETHTTPSESSIONW lpwhs = lpwhr->lpHttpSession;
3364 LPWININETAPPINFOW hIC = lpwhs->lpAppInfo;
3365 BOOL using_proxy = hIC->lpszProxy && hIC->lpszProxy[0];
3366 WCHAR path[INTERNET_MAX_URL_LENGTH];
3367 int index;
3368
3369 if(lpszUrl[0]=='/')
3370 {
3371 /* if it's an absolute path, keep the same session info */
3372 lstrcpynW(path, lpszUrl, INTERNET_MAX_URL_LENGTH);
3373 }
3374 else
3375 {
3376 URL_COMPONENTSW urlComponents;
3377 WCHAR protocol[32], hostName[MAXHOSTNAME], userName[1024];
3378 static WCHAR szHttp[] = {'h','t','t','p',0};
3379 static WCHAR szHttps[] = {'h','t','t','p','s',0};
3380
3381 userName[0] = 0;
3382 hostName[0] = 0;
3383 protocol[0] = 0;
3384
3385 urlComponents.dwStructSize = sizeof(URL_COMPONENTSW);
3386 urlComponents.lpszScheme = protocol;
3387 urlComponents.dwSchemeLength = 32;
3388 urlComponents.lpszHostName = hostName;
3389 urlComponents.dwHostNameLength = MAXHOSTNAME;
3390 urlComponents.lpszUserName = userName;
3391 urlComponents.dwUserNameLength = 1024;
3392 urlComponents.lpszPassword = NULL;
3393 urlComponents.dwPasswordLength = 0;
3394 urlComponents.lpszUrlPath = path;
3395 urlComponents.dwUrlPathLength = 2048;
3396 urlComponents.lpszExtraInfo = NULL;
3397 urlComponents.dwExtraInfoLength = 0;
3398 if(!InternetCrackUrlW(lpszUrl, strlenW(lpszUrl), 0, &urlComponents))
3399 return FALSE;
3400
3401 if (!strncmpW(szHttp, urlComponents.lpszScheme, strlenW(szHttp)) &&
3402 (lpwhr->hdr.dwFlags & INTERNET_FLAG_SECURE))
3403 {
3404 TRACE("redirect from secure page to non-secure page\n");
3405 /* FIXME: warn about from secure redirect to non-secure page */
3406 lpwhr->hdr.dwFlags &= ~INTERNET_FLAG_SECURE;
3407 }
3408 if (!strncmpW(szHttps, urlComponents.lpszScheme, strlenW(szHttps)) &&
3409 !(lpwhr->hdr.dwFlags & INTERNET_FLAG_SECURE))
3410 {
3411 TRACE("redirect from non-secure page to secure page\n");
3412 /* FIXME: notify about redirect to secure page */
3413 lpwhr->hdr.dwFlags |= INTERNET_FLAG_SECURE;
3414 }
3415
3416 if (urlComponents.nPort == INTERNET_INVALID_PORT_NUMBER)
3417 {
3418 if (lstrlenW(protocol)>4) /*https*/
3419 urlComponents.nPort = INTERNET_DEFAULT_HTTPS_PORT;
3420 else /*http*/
3421 urlComponents.nPort = INTERNET_DEFAULT_HTTP_PORT;
3422 }
3423
3424 #if 0
3425 /*
3426 * This upsets redirects to binary files on sourceforge.net
3427 * and gives an html page instead of the target file
3428 * Examination of the HTTP request sent by native wininet.dll
3429 * reveals that it doesn't send a referrer in that case.
3430 * Maybe there's a flag that enables this, or maybe a referrer
3431 * shouldn't be added in case of a redirect.
3432 */
3433
3434 /* consider the current host as the referrer */
3435 if (lpwhs->lpszServerName && *lpwhs->lpszServerName)
3436 HTTP_ProcessHeader(lpwhr, HTTP_REFERER, lpwhs->lpszServerName,
3437 HTTP_ADDHDR_FLAG_REQ|HTTP_ADDREQ_FLAG_REPLACE|
3438 HTTP_ADDHDR_FLAG_ADD_IF_NEW);
3439 #endif
3440
3441 HeapFree(GetProcessHeap(), 0, lpwhs->lpszHostName);
3442 if (urlComponents.nPort != INTERNET_DEFAULT_HTTP_PORT &&
3443 urlComponents.nPort != INTERNET_DEFAULT_HTTPS_PORT)
3444 {
3445 int len;
3446 static const WCHAR fmt[] = {'%','s',':','%','i',0};
3447 len = lstrlenW(hostName);
3448 len += 7; /* 5 for strlen("65535") + 1 for ":" + 1 for '\0' */
3449 lpwhs->lpszHostName = HeapAlloc(GetProcessHeap(), 0, len*sizeof(WCHAR));
3450 sprintfW(lpwhs->lpszHostName, fmt, hostName, urlComponents.nPort);
3451 }
3452 else
3453 lpwhs->lpszHostName = WININET_strdupW(hostName);
3454
3455 HTTP_ProcessHeader(lpwhr, hostW, lpwhs->lpszHostName, HTTP_ADDREQ_FLAG_ADD | HTTP_ADDREQ_FLAG_REPLACE | HTTP_ADDHDR_FLAG_REQ);
3456
3457 HeapFree(GetProcessHeap(), 0, lpwhs->lpszUserName);
3458 lpwhs->lpszUserName = NULL;
3459 if (userName[0])
3460 lpwhs->lpszUserName = WININET_strdupW(userName);
3461
3462 if (!using_proxy)
3463 {
3464 if (strcmpiW(lpwhs->lpszServerName, hostName) || lpwhs->nServerPort != urlComponents.nPort)
3465 {
3466 HeapFree(GetProcessHeap(), 0, lpwhs->lpszServerName);
3467 lpwhs->lpszServerName = WININET_strdupW(hostName);
3468 lpwhs->nServerPort = urlComponents.nPort;
3469
3470 NETCON_close(&lpwhr->netConnection);
3471 if (!HTTP_ResolveName(lpwhr)) return FALSE;
3472 if (!NETCON_init(&lpwhr->netConnection, lpwhr->hdr.dwFlags & INTERNET_FLAG_SECURE)) return FALSE;
3473 lpwhr->read_pos = lpwhr->read_size = 0;
3474 lpwhr->read_chunked = FALSE;
3475 }
3476 }
3477 else
3478 TRACE("Redirect through proxy\n");
3479 }
3480
3481 HeapFree(GetProcessHeap(), 0, lpwhr->lpszPath);
3482 lpwhr->lpszPath=NULL;
3483 if (*path)
3484 {
3485 DWORD needed = 0;
3486 HRESULT rc;
3487
3488 rc = UrlEscapeW(path, NULL, &needed, URL_ESCAPE_SPACES_ONLY);
3489 if (rc != E_POINTER)
3490 needed = strlenW(path)+1;
3491 lpwhr->lpszPath = HeapAlloc(GetProcessHeap(), 0, needed*sizeof(WCHAR));
3492 rc = UrlEscapeW(path, lpwhr->lpszPath, &needed,
3493 URL_ESCAPE_SPACES_ONLY);
3494 if (rc != S_OK)
3495 {
3496 ERR("Unable to escape string!(%s) (%d)\n",debugstr_w(path),rc);
3497 strcpyW(lpwhr->lpszPath,path);
3498 }
3499 }
3500
3501 /* Remove custom content-type/length headers on redirects. */
3502 index = HTTP_GetCustomHeaderIndex(lpwhr, szContent_Type, 0, TRUE);
3503 if (0 <= index)
3504 HTTP_DeleteCustomHeader(lpwhr, index);
3505 index = HTTP_GetCustomHeaderIndex(lpwhr, szContent_Length, 0, TRUE);
3506 if (0 <= index)
3507 HTTP_DeleteCustomHeader(lpwhr, index);
3508
3509 return TRUE;
3510 }
3511
3512 /***********************************************************************
3513 * HTTP_build_req (internal)
3514 *
3515 * concatenate all the strings in the request together
3516 */
3517 static LPWSTR HTTP_build_req( LPCWSTR *list, int len )
3518 {
3519 LPCWSTR *t;
3520 LPWSTR str;
3521
3522 for( t = list; *t ; t++ )
3523 len += strlenW( *t );
3524 len++;
3525
3526 str = HeapAlloc( GetProcessHeap(), 0, len*sizeof(WCHAR) );
3527 *str = 0;
3528
3529 for( t = list; *t ; t++ )
3530 strcatW( str, *t );
3531
3532 return str;
3533 }
3534
3535 static BOOL HTTP_SecureProxyConnect(LPWININETHTTPREQW lpwhr)
3536 {
3537 LPWSTR lpszPath;
3538 LPWSTR requestString;
3539 INT len;
3540 INT cnt;
3541 INT responseLen;
3542 char *ascii_req;
3543 BOOL ret;
3544 static const WCHAR szConnect[] = {'C','O','N','N','E','C','T',0};
3545 static const WCHAR szFormat[] = {'%','s',':','%','d',0};
3546 LPWININETHTTPSESSIONW lpwhs = lpwhr->lpHttpSession;
3547
3548 TRACE("\n");
3549
3550 lpszPath = HeapAlloc( GetProcessHeap(), 0, (lstrlenW( lpwhs->lpszHostName ) + 13)*sizeof(WCHAR) );
3551 sprintfW( lpszPath, szFormat, lpwhs->lpszHostName, lpwhs->nHostPort );
3552 requestString = HTTP_BuildHeaderRequestString( lpwhr, szConnect, lpszPath, g_szHttp1_1 );
3553 HeapFree( GetProcessHeap(), 0, lpszPath );
3554
3555 len = WideCharToMultiByte( CP_ACP, 0, requestString, -1,
3556 NULL, 0, NULL, NULL );
3557 len--; /* the nul terminator isn't needed */
3558 ascii_req = HeapAlloc( GetProcessHeap(), 0, len );
3559 WideCharToMultiByte( CP_ACP, 0, requestString, -1,
3560 ascii_req, len, NULL, NULL );
3561 HeapFree( GetProcessHeap(), 0, requestString );
3562
3563 TRACE("full request -> %s\n", debugstr_an( ascii_req, len ) );
3564
3565 ret = NETCON_send( &lpwhr->netConnection, ascii_req, len, 0, &cnt );
3566 HeapFree( GetProcessHeap(), 0, ascii_req );
3567 if (!ret || cnt < 0)
3568 return FALSE;
3569
3570 responseLen = HTTP_GetResponseHeaders( lpwhr, TRUE );
3571 if (!responseLen)
3572 return FALSE;
3573
3574 return TRUE;
3575 }
3576
3577 static void HTTP_InsertCookies(LPWININETHTTPREQW lpwhr)
3578 {
3579 static const WCHAR szUrlForm[] = {'h','t','t','p',':','/','/','%','s','%','s',0};
3580 LPWSTR lpszCookies, lpszUrl = NULL;
3581 DWORD nCookieSize, size;
3582 LPHTTPHEADERW Host = HTTP_GetHeader(lpwhr, hostW);
3583
3584 size = (strlenW(Host->lpszValue) + strlenW(szUrlForm) + strlenW(lpwhr->lpszPath)) * sizeof(WCHAR);
3585 if (!(lpszUrl = HeapAlloc(GetProcessHeap(), 0, size))) return;
3586 sprintfW( lpszUrl, szUrlForm, Host->lpszValue, lpwhr->lpszPath);
3587
3588 if (InternetGetCookieW(lpszUrl, NULL, NULL, &nCookieSize))
3589 {
3590 int cnt = 0;
3591 static const WCHAR szCookie[] = {'C','o','o','k','i','e',':',' ',0};
3592
3593 size = sizeof(szCookie) + nCookieSize * sizeof(WCHAR) + sizeof(szCrLf);
3594 if ((lpszCookies = HeapAlloc(GetProcessHeap(), 0, size)))
3595 {
3596 cnt += sprintfW(lpszCookies, szCookie);
3597 InternetGetCookieW(lpszUrl, NULL, lpszCookies + cnt, &nCookieSize);
3598 strcatW(lpszCookies, szCrLf);
3599
3600 HTTP_HttpAddRequestHeadersW(lpwhr, lpszCookies, strlenW(lpszCookies), HTTP_ADDREQ_FLAG_REPLACE);
3601 HeapFree(GetProcessHeap(), 0, lpszCookies);
3602 }
3603 }
3604 HeapFree(GetProcessHeap(), 0, lpszUrl);
3605 }
3606
3607 /***********************************************************************
3608 * HTTP_HttpSendRequestW (internal)
3609 *
3610 * Sends the specified request to the HTTP server
3611 *
3612 * RETURNS
3613 * TRUE on success
3614 * FALSE on failure
3615 *
3616 */
3617 BOOL WINAPI HTTP_HttpSendRequestW(LPWININETHTTPREQW lpwhr, LPCWSTR lpszHeaders,
3618 DWORD dwHeaderLength, LPVOID lpOptional, DWORD dwOptionalLength,
3619 DWORD dwContentLength, BOOL bEndRequest)
3620 {
3621 INT cnt;
3622 BOOL bSuccess = FALSE, redirected = FALSE;
3623 LPWSTR requestString = NULL;
3624 INT responseLen;
3625 BOOL loop_next;
3626 INTERNET_ASYNC_RESULT iar;
3627 static const WCHAR szPost[] = { 'P','O','S','T',0 };
3628 static const WCHAR szContentLength[] =
3629 { 'C','o','n','t','e','n','t','-','L','e','n','g','t','h',':',' ','%','l','i','\r','\n',0 };
3630 WCHAR contentLengthStr[sizeof szContentLength/2 /* includes \r\n */ + 20 /* int */ ];
3631
3632 TRACE("--> %p\n", lpwhr);
3633
3634 assert(lpwhr->hdr.htype == WH_HHTTPREQ);
3635
3636 /* if the verb is NULL default to GET */
3637 if (!lpwhr->lpszVerb)
3638 lpwhr->lpszVerb = WININET_strdupW(szGET);
3639
3640 if (dwContentLength || strcmpW(lpwhr->lpszVerb, szGET))
3641 {
3642 sprintfW(contentLengthStr, szContentLength, dwContentLength);
3643 HTTP_HttpAddRequestHeadersW(lpwhr, contentLengthStr, -1L, HTTP_ADDREQ_FLAG_REPLACE);
3644 lpwhr->dwBytesToWrite = dwContentLength;
3645 }
3646 if (lpwhr->lpHttpSession->lpAppInfo->lpszAgent)
3647 {
3648 WCHAR *agent_header;
3649 static const WCHAR user_agent[] = {'U','s','e','r','-','A','g','e','n','t',':',' ','%','s','\r','\n',0};
3650 int len;
3651
3652 len = strlenW(lpwhr->lpHttpSession->lpAppInfo->lpszAgent) + strlenW(user_agent);
3653 agent_header = HeapAlloc(GetProcessHeap(), 0, len * sizeof(WCHAR));
3654 sprintfW(agent_header, user_agent, lpwhr->lpHttpSession->lpAppInfo->lpszAgent);
3655
3656 HTTP_HttpAddRequestHeadersW(lpwhr, agent_header, strlenW(agent_header), HTTP_ADDREQ_FLAG_ADD_IF_NEW);
3657 HeapFree(GetProcessHeap(), 0, agent_header);
3658 }
3659 if (lpwhr->hdr.dwFlags & INTERNET_FLAG_PRAGMA_NOCACHE)
3660 {
3661 static const WCHAR pragma_nocache[] = {'P','r','a','g','m','a',':',' ','n','o','-','c','a','c','h','e','\r','\n',0};
3662 HTTP_HttpAddRequestHeadersW(lpwhr, pragma_nocache, strlenW(pragma_nocache), HTTP_ADDREQ_FLAG_ADD_IF_NEW);
3663 }
3664 if ((lpwhr->hdr.dwFlags & INTERNET_FLAG_NO_CACHE_WRITE) && !strcmpW(lpwhr->lpszVerb, szPost))
3665 {
3666 static const WCHAR cache_control[] = {'C','a','c','h','e','-','C','o','n','t','r','o','l',':',
3667 ' ','n','o','-','c','a','c','h','e','\r','\n',0};
3668 HTTP_HttpAddRequestHeadersW(lpwhr, cache_control, strlenW(cache_control), HTTP_ADDREQ_FLAG_ADD_IF_NEW);
3669 }
3670
3671 do
3672 {
3673 DWORD len;
3674 char *ascii_req;
3675
3676 loop_next = FALSE;
3677
3678 /* like native, just in case the caller forgot to call InternetReadFile
3679 * for all the data */
3680 HTTP_DrainContent(lpwhr);
3681 lpwhr->dwContentRead = 0;
3682
3683 if (TRACE_ON(wininet))
3684 {
3685 LPHTTPHEADERW Host = HTTP_GetHeader(lpwhr, hostW);
3686 TRACE("Going to url %s %s\n", debugstr_w(Host->lpszValue), debugstr_w(lpwhr->lpszPath));
3687 }
3688
3689 HTTP_FixURL(lpwhr);
3690 if (lpwhr->hdr.dwFlags & INTERNET_FLAG_KEEP_CONNECTION)
3691 {
3692 HTTP_ProcessHeader(lpwhr, szConnection, szKeepAlive, HTTP_ADDHDR_FLAG_REQ | HTTP_ADDHDR_FLAG_REPLACE);
3693 }
3694 HTTP_InsertAuthorization(lpwhr, lpwhr->pAuthInfo, szAuthorization);
3695 HTTP_InsertAuthorization(lpwhr, lpwhr->pProxyAuthInfo, szProxy_Authorization);
3696
3697 if (!(lpwhr->hdr.dwFlags & INTERNET_FLAG_NO_COOKIES))
3698 HTTP_InsertCookies(lpwhr);
3699
3700 /* add the headers the caller supplied */
3701 if( lpszHeaders && dwHeaderLength )
3702 {
3703 HTTP_HttpAddRequestHeadersW(lpwhr, lpszHeaders, dwHeaderLength,
3704 HTTP_ADDREQ_FLAG_ADD | HTTP_ADDHDR_FLAG_REPLACE);
3705 }
3706
3707 if (lpwhr->lpHttpSession->lpAppInfo->lpszProxy && lpwhr->lpHttpSession->lpAppInfo->lpszProxy[0])
3708 {
3709 WCHAR *url = HTTP_BuildProxyRequestUrl(lpwhr);
3710 requestString = HTTP_BuildHeaderRequestString(lpwhr, lpwhr->lpszVerb, url, lpwhr->lpszVersion);
3711 HeapFree(GetProcessHeap(), 0, url);
3712 }
3713 else
3714 requestString = HTTP_BuildHeaderRequestString(lpwhr, lpwhr->lpszVerb, lpwhr->lpszPath, lpwhr->lpszVersion);
3715
3716
3717 TRACE("Request header -> %s\n", debugstr_w(requestString) );
3718
3719 /* Send the request and store the results */
3720 if (!HTTP_OpenConnection(lpwhr))
3721 goto lend;
3722
3723 /* send the request as ASCII, tack on the optional data */
3724 if (!lpOptional || redirected)
3725 dwOptionalLength = 0;
3726 len = WideCharToMultiByte( CP_ACP, 0, requestString, -1,
3727 NULL, 0, NULL, NULL );
3728 ascii_req = HeapAlloc( GetProcessHeap(), 0, len + dwOptionalLength );
3729 WideCharToMultiByte( CP_ACP, 0, requestString, -1,
3730 ascii_req, len, NULL, NULL );
3731 if( lpOptional )
3732 memcpy( &ascii_req[len-1], lpOptional, dwOptionalLength );
3733 len = (len + dwOptionalLength - 1);
3734 ascii_req[len] = 0;
3735 TRACE("full request -> %s\n", debugstr_a(ascii_req) );
3736
3737 INTERNET_SendCallback(&lpwhr->hdr, lpwhr->hdr.dwContext,
3738 INTERNET_STATUS_SENDING_REQUEST, NULL, 0);
3739
3740 NETCON_send(&lpwhr->netConnection, ascii_req, len, 0, &cnt);
3741 HeapFree( GetProcessHeap(), 0, ascii_req );
3742
3743 lpwhr->dwBytesWritten = dwOptionalLength;
3744
3745 INTERNET_SendCallback(&lpwhr->hdr, lpwhr->hdr.dwContext,
3746 INTERNET_STATUS_REQUEST_SENT,
3747 &len, sizeof(DWORD));
3748
3749 if (bEndRequest)
3750 {
3751 DWORD dwBufferSize;
3752 DWORD dwStatusCode;
3753
3754 INTERNET_SendCallback(&lpwhr->hdr, lpwhr->hdr.dwContext,
3755 INTERNET_STATUS_RECEIVING_RESPONSE, NULL, 0);
3756
3757 if (cnt < 0)
3758 goto lend;
3759
3760 responseLen = HTTP_GetResponseHeaders(lpwhr, TRUE);
3761 if (responseLen)
3762 bSuccess = TRUE;
3763
3764 INTERNET_SendCallback(&lpwhr->hdr, lpwhr->hdr.dwContext,
3765 INTERNET_STATUS_RESPONSE_RECEIVED, &responseLen,
3766 sizeof(DWORD));
3767
3768 HTTP_ProcessCookies(lpwhr);
3769
3770 if (!set_content_length( lpwhr )) HTTP_FinishedReading(lpwhr);
3771
3772 dwBufferSize = sizeof(dwStatusCode);
3773 if (!HTTP_HttpQueryInfoW(lpwhr,HTTP_QUERY_FLAG_NUMBER|HTTP_QUERY_STATUS_CODE,
3774 &dwStatusCode,&dwBufferSize,NULL))
3775 dwStatusCode = 0;
3776
3777 if (!(lpwhr->hdr.dwFlags & INTERNET_FLAG_NO_AUTO_REDIRECT) && bSuccess)
3778 {
3779 WCHAR *new_url, szNewLocation[INTERNET_MAX_URL_LENGTH];
3780 dwBufferSize=sizeof(szNewLocation);
3781 if ((dwStatusCode==HTTP_STATUS_REDIRECT || dwStatusCode==HTTP_STATUS_MOVED) &&
3782 HTTP_HttpQueryInfoW(lpwhr,HTTP_QUERY_LOCATION,szNewLocation,&dwBufferSize,NULL))
3783 {
3784 /* redirects are always GETs */
3785 HeapFree(GetProcessHeap(), 0, lpwhr->lpszVerb);
3786 lpwhr->lpszVerb = WININET_strdupW(szGET);
3787
3788 HTTP_DrainContent(lpwhr);
3789 if ((new_url = HTTP_GetRedirectURL( lpwhr, szNewLocation )))
3790 {
3791 INTERNET_SendCallback(&lpwhr->hdr, lpwhr->hdr.dwContext, INTERNET_STATUS_REDIRECT,
3792 new_url, (strlenW(new_url) + 1) * sizeof(WCHAR));
3793 bSuccess = HTTP_HandleRedirect(lpwhr, new_url);
3794 if (bSuccess)
3795 {
3796 HeapFree(GetProcessHeap(), 0, requestString);
3797 loop_next = TRUE;
3798 }
3799 HeapFree( GetProcessHeap(), 0, new_url );
3800 }
3801 redirected = TRUE;
3802 }
3803 }
3804 if (!(lpwhr->hdr.dwFlags & INTERNET_FLAG_NO_AUTH) && bSuccess)
3805 {
3806 WCHAR szAuthValue[2048];
3807 dwBufferSize=2048;
3808 if (dwStatusCode == HTTP_STATUS_DENIED)
3809 {
3810 DWORD dwIndex = 0;
3811 while (HTTP_HttpQueryInfoW(lpwhr,HTTP_QUERY_WWW_AUTHENTICATE,szAuthValue,&dwBufferSize,&dwIndex))
3812 {
3813 if (HTTP_DoAuthorization(lpwhr, szAuthValue,
3814 &lpwhr->pAuthInfo,
3815 lpwhr->lpHttpSession->lpszUserName,
3816 lpwhr->lpHttpSession->lpszPassword))
3817 {
3818 loop_next = TRUE;
3819 break;
3820 }
3821 }
3822 }
3823 if (dwStatusCode == HTTP_STATUS_PROXY_AUTH_REQ)
3824 {
3825 DWORD dwIndex = 0;
3826 while (HTTP_HttpQueryInfoW(lpwhr,HTTP_QUERY_PROXY_AUTHENTICATE,szAuthValue,&dwBufferSize,&dwIndex))
3827 {
3828 if (HTTP_DoAuthorization(lpwhr, szAuthValue,
3829 &lpwhr->pProxyAuthInfo,
3830 lpwhr->lpHttpSession->lpAppInfo->lpszProxyUsername,
3831 lpwhr->lpHttpSession->lpAppInfo->lpszProxyPassword))
3832 {
3833 loop_next = TRUE;
3834 break;
3835 }
3836 }
3837 }
3838 }
3839 }
3840 else
3841 bSuccess = TRUE;
3842 }
3843 while (loop_next);
3844
3845 if(bSuccess) {
3846 WCHAR url[INTERNET_MAX_URL_LENGTH];
3847 WCHAR cacheFileName[MAX_PATH+1];
3848 BOOL b;
3849
3850 b = HTTP_GetRequestURL(lpwhr, url);
3851 if(!b) {
3852 WARN("Could not get URL\n");
3853 goto lend;
3854 }
3855
3856 b = CreateUrlCacheEntryW(url, lpwhr->dwContentLength > 0 ? lpwhr->dwContentLength : 0, NULL, cacheFileName, 0);
3857 if(b) {
3858 lpwhr->lpszCacheFile = WININET_strdupW(cacheFileName);
3859 lpwhr->hCacheFile = CreateFileW(lpwhr->lpszCacheFile, GENERIC_WRITE, FILE_SHARE_READ|FILE_SHARE_WRITE,
3860 NULL, CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
3861 if(lpwhr->hCacheFile == INVALID_HANDLE_VALUE) {
3862 WARN("Could not create file: %u\n", GetLastError());
3863 lpwhr->hCacheFile = NULL;
3864 }
3865 }else {
3866 WARN("Could not create cache entry: %08x\n", GetLastError());
3867 }
3868 }
3869
3870 lend:
3871
3872 HeapFree(GetProcessHeap(), 0, requestString);
3873
3874 /* TODO: send notification for P3P header */
3875
3876 if (lpwhr->lpHttpSession->lpAppInfo->hdr.dwFlags & INTERNET_FLAG_ASYNC)
3877 {
3878 if (bSuccess)
3879 {
3880 if (lpwhr->dwBytesWritten == lpwhr->dwBytesToWrite) HTTP_ReceiveRequestData(lpwhr, TRUE);
3881 else
3882 {
3883 iar.dwResult = (DWORD_PTR)lpwhr->hdr.hInternet;
3884 iar.dwError = 0;
3885
3886 INTERNET_SendCallback(&lpwhr->hdr, lpwhr->hdr.dwContext,
3887 INTERNET_STATUS_REQUEST_COMPLETE, &iar,
3888 sizeof(INTERNET_ASYNC_RESULT));
3889 }
3890 }
3891 else
3892 {
3893 iar.dwResult = (DWORD_PTR)lpwhr->hdr.hInternet;
3894 iar.dwError = INTERNET_GetLastError();
3895
3896 INTERNET_SendCallback(&lpwhr->hdr, lpwhr->hdr.dwContext,
3897 INTERNET_STATUS_REQUEST_COMPLETE, &iar,
3898 sizeof(INTERNET_ASYNC_RESULT));
3899 }
3900 }
3901
3902 TRACE("<--\n");
3903 if (bSuccess) INTERNET_SetLastError(ERROR_SUCCESS);
3904 return bSuccess;
3905 }
3906
3907 /***********************************************************************
3908 * HTTPSESSION_Destroy (internal)
3909 *
3910 * Deallocate session handle
3911 *
3912 */
3913 static void HTTPSESSION_Destroy(WININETHANDLEHEADER *hdr)
3914 {
3915 LPWININETHTTPSESSIONW lpwhs = (LPWININETHTTPSESSIONW) hdr;
3916
3917 TRACE("%p\n", lpwhs);
3918
3919 WININET_Release(&lpwhs->lpAppInfo->hdr);
3920
3921 HeapFree(GetProcessHeap(), 0, lpwhs->lpszHostName);
3922 HeapFree(GetProcessHeap(), 0, lpwhs->lpszServerName);
3923 HeapFree(GetProcessHeap(), 0, lpwhs->lpszPassword);
3924 HeapFree(GetProcessHeap(), 0, lpwhs->lpszUserName);
3925 HeapFree(GetProcessHeap(), 0, lpwhs);
3926 }
3927
3928 static DWORD HTTPSESSION_QueryOption(WININETHANDLEHEADER *hdr, DWORD option, void *buffer, DWORD *size, BOOL unicode)
3929 {
3930 switch(option) {
3931 case INTERNET_OPTION_HANDLE_TYPE:
3932 TRACE("INTERNET_OPTION_HANDLE_TYPE\n");
3933
3934 if (*size < sizeof(ULONG))
3935 return ERROR_INSUFFICIENT_BUFFER;
3936
3937 *size = sizeof(DWORD);
3938 *(DWORD*)buffer = INTERNET_HANDLE_TYPE_CONNECT_HTTP;
3939 return ERROR_SUCCESS;
3940 }
3941
3942 return INET_QueryOption(option, buffer, size, unicode);
3943 }
3944
3945 static DWORD HTTPSESSION_SetOption(WININETHANDLEHEADER *hdr, DWORD option, void *buffer, DWORD size)
3946 {
3947 WININETHTTPSESSIONW *ses = (WININETHTTPSESSIONW*)hdr;
3948
3949 switch(option) {
3950 case INTERNET_OPTION_USERNAME:
3951 {
3952 HeapFree(GetProcessHeap(), 0, ses->lpszUserName);
3953 if (!(ses->lpszUserName = WININET_strdupW(buffer))) return ERROR_OUTOFMEMORY;
3954 return ERROR_SUCCESS;
3955 }
3956 case INTERNET_OPTION_PASSWORD:
3957 {
3958 HeapFree(GetProcessHeap(), 0, ses->lpszPassword);
3959 if (!(ses->lpszPassword = WININET_strdupW(buffer))) return ERROR_OUTOFMEMORY;
3960 return ERROR_SUCCESS;
3961 }
3962 default: break;
3963 }
3964
3965 return ERROR_INTERNET_INVALID_OPTION;
3966 }
3967
3968 static const HANDLEHEADERVtbl HTTPSESSIONVtbl = {
3969 HTTPSESSION_Destroy,
3970 NULL,
3971 HTTPSESSION_QueryOption,
3972 HTTPSESSION_SetOption,
3973 NULL,
3974 NULL,
3975 NULL,
3976 NULL,
3977 NULL
3978 };
3979
3980
3981 /***********************************************************************
3982 * HTTP_Connect (internal)
3983 *
3984 * Create http session handle
3985 *
3986 * RETURNS
3987 * HINTERNET a session handle on success
3988 * NULL on failure
3989 *
3990 */
3991 HINTERNET HTTP_Connect(LPWININETAPPINFOW hIC, LPCWSTR lpszServerName,
3992 INTERNET_PORT nServerPort, LPCWSTR lpszUserName,
3993 LPCWSTR lpszPassword, DWORD dwFlags, DWORD_PTR dwContext,
3994 DWORD dwInternalFlags)
3995 {
3996 LPWININETHTTPSESSIONW lpwhs = NULL;
3997 HINTERNET handle = NULL;
3998
3999 TRACE("-->\n");
4000
4001 if (!lpszServerName || !lpszServerName[0])
4002 {
4003 INTERNET_SetLastError(ERROR_INVALID_PARAMETER);
4004 goto lerror;
4005 }
4006
4007 assert( hIC->hdr.htype == WH_HINIT );
4008
4009 lpwhs = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(WININETHTTPSESSIONW));
4010 if (NULL == lpwhs)
4011 {
4012 INTERNET_SetLastError(ERROR_OUTOFMEMORY);
4013 goto lerror;
4014 }
4015
4016 /*
4017 * According to my tests. The name is not resolved until a request is sent
4018 */
4019
4020 lpwhs->hdr.htype = WH_HHTTPSESSION;
4021 lpwhs->hdr.vtbl = &HTTPSESSIONVtbl;
4022 lpwhs->hdr.dwFlags = dwFlags;
4023 lpwhs->hdr.dwContext = dwContext;
4024 lpwhs->hdr.dwInternalFlags = dwInternalFlags | (hIC->hdr.dwInternalFlags & INET_CALLBACKW);
4025 lpwhs->hdr.refs = 1;
4026 lpwhs->hdr.lpfnStatusCB = hIC->hdr.lpfnStatusCB;
4027
4028 WININET_AddRef( &hIC->hdr );
4029 lpwhs->lpAppInfo = hIC;
4030 list_add_head( &hIC->hdr.children, &lpwhs->hdr.entry );
4031
4032 handle = WININET_AllocHandle( &lpwhs->hdr );
4033 if (NULL == handle)
4034 {
4035 ERR("Failed to alloc handle\n");
4036 INTERNET_SetLastError(ERROR_OUTOFMEMORY);
4037 goto lerror;
4038 }
4039
4040 if(hIC->lpszProxy && hIC->dwAccessType == INTERNET_OPEN_TYPE_PROXY) {
4041 if(strchrW(hIC->lpszProxy, ' '))
4042 FIXME("Several proxies not implemented.\n");
4043 if(hIC->lpszProxyBypass)
4044 FIXME("Proxy bypass is ignored.\n");
4045 }
4046 if (lpszServerName && lpszServerName[0])
4047 {
4048 lpwhs->lpszServerName = WININET_strdupW(lpszServerName);
4049 lpwhs->lpszHostName = WININET_strdupW(lpszServerName);
4050 }
4051 if (lpszUserName && lpszUserName[0])
4052 lpwhs->lpszUserName = WININET_strdupW(lpszUserName);
4053 if (lpszPassword && lpszPassword[0])
4054 lpwhs->lpszPassword = WININET_strdupW(lpszPassword);
4055 lpwhs->nServerPort = nServerPort;
4056 lpwhs->nHostPort = nServerPort;
4057
4058 /* Don't send a handle created callback if this handle was created with InternetOpenUrl */
4059 if (!(lpwhs->hdr.dwInternalFlags & INET_OPENURL))
4060 {
4061 INTERNET_SendCallback(&hIC->hdr, dwContext,
4062 INTERNET_STATUS_HANDLE_CREATED, &handle,
4063 sizeof(handle));
4064 }
4065
4066 lerror:
4067 if( lpwhs )
4068 WININET_Release( &lpwhs->hdr );
4069
4070 /*
4071 * an INTERNET_STATUS_REQUEST_COMPLETE is NOT sent here as per my tests on
4072 * windows
4073 */
4074
4075 TRACE("%p --> %p (%p)\n", hIC, handle, lpwhs);
4076 return handle;
4077 }
4078
4079
4080 /***********************************************************************
4081 * HTTP_OpenConnection (internal)
4082 *
4083 * Connect to a web server
4084 *
4085 * RETURNS
4086 *
4087 * TRUE on success
4088 * FALSE on failure
4089 */
4090 static BOOL HTTP_OpenConnection(LPWININETHTTPREQW lpwhr)
4091 {
4092 BOOL bSuccess = FALSE;
4093 LPWININETHTTPSESSIONW lpwhs;
4094 LPWININETAPPINFOW hIC = NULL;
4095 char szaddr[32];
4096
4097 TRACE("-->\n");
4098
4099
4100 if (lpwhr->hdr.htype != WH_HHTTPREQ)
4101 {
4102 INTERNET_SetLastError(ERROR_INVALID_PARAMETER);
4103 goto lend;
4104 }
4105
4106 if (NETCON_connected(&lpwhr->netConnection))
4107 {
4108 bSuccess = TRUE;
4109 goto lend;
4110 }
4111 if (!HTTP_ResolveName(lpwhr)) goto lend;
4112
4113 lpwhs = lpwhr->lpHttpSession;
4114
4115 hIC = lpwhs->lpAppInfo;
4116 inet_ntop(lpwhs->socketAddress.sin_family, &lpwhs->socketAddress.sin_addr,
4117 szaddr, sizeof(szaddr));
4118 INTERNET_SendCallback(&lpwhr->hdr, lpwhr->hdr.dwContext,
4119 INTERNET_STATUS_CONNECTING_TO_SERVER,
4120 szaddr,
4121 strlen(szaddr)+1);
4122
4123 if (!NETCON_create(&lpwhr->netConnection, lpwhs->socketAddress.sin_family,
4124 SOCK_STREAM, 0))
4125 {
4126 WARN("Socket creation failed: %u\n", INTERNET_GetLastError());
4127 goto lend;
4128 }
4129
4130 if (!NETCON_connect(&lpwhr->netConnection, (struct sockaddr *)&lpwhs->socketAddress,
4131 sizeof(lpwhs->socketAddress)))
4132 goto lend;
4133
4134 if (lpwhr->hdr.dwFlags & INTERNET_FLAG_SECURE)
4135 {
4136 /* Note: we differ from Microsoft's WinINet here. they seem to have
4137 * a bug that causes no status callbacks to be sent when starting
4138 * a tunnel to a proxy server using the CONNECT verb. i believe our
4139 * behaviour to be more correct and to not cause any incompatibilities
4140 * because using a secure connection through a proxy server is a rare
4141 * case that would be hard for anyone to depend on */
4142 if (hIC->lpszProxy && !HTTP_SecureProxyConnect(lpwhr))
4143 goto lend;
4144
4145 if (!NETCON_secure_connect(&lpwhr->netConnection, lpwhs->lpszHostName))
4146 {
4147 WARN("Couldn't connect securely to host\n");
4148 goto lend;
4149 }
4150 }
4151
4152 INTERNET_SendCallback(&lpwhr->hdr, lpwhr->hdr.dwContext,
4153 INTERNET_STATUS_CONNECTED_TO_SERVER,
4154 szaddr, strlen(szaddr)+1);
4155
4156 bSuccess = TRUE;
4157
4158 lend:
4159 lpwhr->read_pos = lpwhr->read_size = 0;
4160 lpwhr->read_chunked = FALSE;
4161
4162 TRACE("%d <--\n", bSuccess);
4163 return bSuccess;
4164 }
4165
4166
4167 /***********************************************************************
4168 * HTTP_clear_response_headers (internal)
4169 *
4170 * clear out any old response headers
4171 */
4172 static void HTTP_clear_response_headers( LPWININETHTTPREQW lpwhr )
4173 {
4174 DWORD i;
4175
4176 for( i=0; i<lpwhr->nCustHeaders; i++)
4177 {
4178 if( !lpwhr->pCustHeaders[i].lpszField )
4179 continue;
4180 if( !lpwhr->pCustHeaders[i].lpszValue )
4181 continue;
4182 if ( lpwhr->pCustHeaders[i].wFlags & HDR_ISREQUEST )
4183 continue;
4184 HTTP_DeleteCustomHeader( lpwhr, i );
4185 i--;
4186 }
4187 }
4188
4189 /***********************************************************************
4190 * HTTP_GetResponseHeaders (internal)
4191 *
4192 * Read server response
4193 *
4194 * RETURNS
4195 *
4196 * TRUE on success
4197 * FALSE on error
4198 */
4199 static INT HTTP_GetResponseHeaders(LPWININETHTTPREQW lpwhr, BOOL clear)
4200 {
4201 INT cbreaks = 0;
4202 WCHAR buffer[MAX_REPLY_LEN];
4203 DWORD buflen = MAX_REPLY_LEN;
4204 BOOL bSuccess = FALSE;
4205 INT rc = 0;
4206 static const WCHAR szHundred[] = {'1','0','0',0};
4207 char bufferA[MAX_REPLY_LEN];
4208 LPWSTR status_code, status_text;
4209 DWORD cchMaxRawHeaders = 1024;
4210 LPWSTR lpszRawHeaders = HeapAlloc(GetProcessHeap(), 0, (cchMaxRawHeaders+1)*sizeof(WCHAR));
4211 LPWSTR temp;
4212 DWORD cchRawHeaders = 0;
4213
4214 TRACE("-->\n");
4215
4216 /* clear old response headers (eg. from a redirect response) */
4217 if (clear) HTTP_clear_response_headers( lpwhr );
4218
4219 if (!NETCON_connected(&lpwhr->netConnection))
4220 goto lend;
4221
4222 do {
4223 /*
4224 * We should first receive 'HTTP/1.x nnn OK' where nnn is the status code.
4225 */
4226 buflen = MAX_REPLY_LEN;
4227 if (!read_line(lpwhr, bufferA, &buflen))
4228 goto lend;
4229 rc += buflen;
4230 MultiByteToWideChar( CP_ACP, 0, bufferA, buflen, buffer, MAX_REPLY_LEN );
4231
4232 /* split the version from the status code */
4233 status_code = strchrW( buffer, ' ' );
4234 if( !status_code )
4235 goto lend;
4236 *status_code++=0;
4237
4238 /* split the status code from the status text */
4239 status_text = strchrW( status_code, ' ' );
4240 if( !status_text )
4241 goto lend;
4242 *status_text++=0;
4243
4244 TRACE("version [%s] status code [%s] status text [%s]\n",
4245 debugstr_w(buffer), debugstr_w(status_code), debugstr_w(status_text) );
4246
4247 } while (!strcmpW(status_code, szHundred)); /* ignore "100 Continue" responses */
4248
4249 /* Add status code */
4250 HTTP_ProcessHeader(lpwhr, szStatus, status_code,
4251 HTTP_ADDHDR_FLAG_REPLACE);
4252
4253 HeapFree(GetProcessHeap(),0,lpwhr->lpszVersion);
4254 HeapFree(GetProcessHeap(),0,lpwhr->lpszStatusText);
4255
4256 lpwhr->lpszVersion= WININET_strdupW(buffer);
4257 lpwhr->lpszStatusText = WININET_strdupW(status_text);
4258
4259 /* Restore the spaces */
4260 *(status_code-1) = ' ';
4261 *(status_text-1) = ' ';
4262
4263 /* regenerate raw headers */
4264 while (cchRawHeaders + buflen + strlenW(szCrLf) > cchMaxRawHeaders)
4265 cchMaxRawHeaders *= 2;
4266 temp = HeapReAlloc(GetProcessHeap(), 0, lpszRawHeaders, (cchMaxRawHeaders+1)*sizeof(WCHAR));
4267 if (temp == NULL) goto lend;
4268 lpszRawHeaders = temp;
4269 memcpy(lpszRawHeaders+cchRawHeaders, buffer, (buflen-1)*sizeof(WCHAR));
4270 cchRawHeaders += (buflen-1);
4271 memcpy(lpszRawHeaders+cchRawHeaders, szCrLf, sizeof(szCrLf));
4272 cchRawHeaders += sizeof(szCrLf)/sizeof(szCrLf[0])-1;
4273 lpszRawHeaders[cchRawHeaders] = '\0';
4274
4275 /* Parse each response line */
4276 do
4277 {
4278 buflen = MAX_REPLY_LEN;
4279 if (read_line(lpwhr, bufferA, &buflen))
4280 {
4281 LPWSTR * pFieldAndValue;
4282
4283 TRACE("got line %s, now interpreting\n", debugstr_a(bufferA));
4284
4285 if (!bufferA[0]) break;
4286 MultiByteToWideChar( CP_ACP, 0, bufferA, buflen, buffer, MAX_REPLY_LEN );
4287
4288 pFieldAndValue = HTTP_InterpretHttpHeader(buffer);
4289 if (pFieldAndValue)
4290 {
4291 while (cchRawHeaders + buflen + strlenW(szCrLf) > cchMaxRawHeaders)
4292 cchMaxRawHeaders *= 2;
4293 temp = HeapReAlloc(GetProcessHeap(), 0, lpszRawHeaders, (cchMaxRawHeaders+1)*sizeof(WCHAR));
4294 if (temp == NULL) goto lend;
4295 lpszRawHeaders = temp;
4296 memcpy(lpszRawHeaders+cchRawHeaders, buffer, (buflen-1)*sizeof(WCHAR));
4297 cchRawHeaders += (buflen-1);
4298 memcpy(lpszRawHeaders+cchRawHeaders, szCrLf, sizeof(szCrLf));
4299 cchRawHeaders += sizeof(szCrLf)/sizeof(szCrLf[0])-1;
4300 lpszRawHeaders[cchRawHeaders] = '\0';
4301
4302 HTTP_ProcessHeader(lpwhr, pFieldAndValue[0], pFieldAndValue[1],
4303 HTTP_ADDREQ_FLAG_ADD );
4304
4305 HTTP_FreeTokens(pFieldAndValue);
4306 }
4307 }
4308 else
4309 {
4310 cbreaks++;
4311 if (cbreaks >= 2)
4312 break;
4313 }
4314 }while(1);
4315
4316 HeapFree(GetProcessHeap(), 0, lpwhr->lpszRawHeaders);
4317 lpwhr->lpszRawHeaders = lpszRawHeaders;
4318 TRACE("raw headers: %s\n", debugstr_w(lpszRawHeaders));
4319 bSuccess = TRUE;
4320
4321 lend:
4322
4323 TRACE("<--\n");
4324 if (bSuccess)
4325 return rc;
4326 else
4327 {
4328 HeapFree(GetProcessHeap(), 0, lpszRawHeaders);
4329 return 0;
4330 }
4331 }
4332
4333
4334 static void strip_spaces(LPWSTR start)
4335 {
4336 LPWSTR str = start;
4337 LPWSTR end;
4338
4339 while (*str == ' ' && *str != '\0')
4340 str++;
4341
4342 if (str != start)
4343 memmove(start, str, sizeof(WCHAR) * (strlenW(str) + 1));
4344
4345 end = start + strlenW(start) - 1;
4346 while (end >= start && *end == ' ')
4347 {
4348 *end = '\0';
4349 end--;
4350 }
4351 }
4352
4353
4354 /***********************************************************************
4355 * HTTP_InterpretHttpHeader (internal)
4356 *
4357 * Parse server response
4358 *
4359 * RETURNS
4360 *
4361 * Pointer to array of field, value, NULL on success.
4362 * NULL on error.
4363 */
4364 static LPWSTR * HTTP_InterpretHttpHeader(LPCWSTR buffer)
4365 {
4366 LPWSTR * pTokenPair;
4367 LPWSTR pszColon;
4368 INT len;
4369
4370 pTokenPair = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(*pTokenPair)*3);
4371
4372 pszColon = strchrW(buffer, ':');
4373 /* must have two tokens */
4374 if (!pszColon)
4375 {
4376 HTTP_FreeTokens(pTokenPair);
4377 if (buffer[0])
4378 TRACE("No ':' in line: %s\n", debugstr_w(buffer));
4379 return NULL;
4380 }
4381
4382 pTokenPair[0] = HeapAlloc(GetProcessHeap(), 0, (pszColon - buffer + 1) * sizeof(WCHAR));
4383 if (!pTokenPair[0])
4384 {
4385 HTTP_FreeTokens(pTokenPair);
4386 return NULL;
4387 }
4388 memcpy(pTokenPair[0], buffer, (pszColon - buffer) * sizeof(WCHAR));
4389 pTokenPair[0][pszColon - buffer] = '\0';
4390
4391 /* skip colon */
4392 pszColon++;
4393 len = strlenW(pszColon);
4394 pTokenPair[1] = HeapAlloc(GetProcessHeap(), 0, (len + 1) * sizeof(WCHAR));
4395 if (!pTokenPair[1])
4396 {
4397 HTTP_FreeTokens(pTokenPair);
4398 return NULL;
4399 }
4400 memcpy(pTokenPair[1], pszColon, (len + 1) * sizeof(WCHAR));
4401
4402 strip_spaces(pTokenPair[0]);
4403 strip_spaces(pTokenPair[1]);
4404
4405 TRACE("field(%s) Value(%s)\n", debugstr_w(pTokenPair[0]), debugstr_w(pTokenPair[1]));
4406 return pTokenPair;
4407 }
4408
4409 /***********************************************************************
4410 * HTTP_ProcessHeader (internal)
4411 *
4412 * Stuff header into header tables according to <dwModifier>
4413 *
4414 */
4415
4416 #define COALESCEFLAGS (HTTP_ADDHDR_FLAG_COALESCE|HTTP_ADDHDR_FLAG_COALESCE_WITH_COMMA|HTTP_ADDHDR_FLAG_COALESCE_WITH_SEMICOLON)
4417
4418 static BOOL HTTP_ProcessHeader(LPWININETHTTPREQW lpwhr, LPCWSTR field, LPCWSTR value, DWORD dwModifier)
4419 {
4420 LPHTTPHEADERW lphttpHdr = NULL;
4421 BOOL bSuccess = FALSE;
4422 INT index = -1;
4423 BOOL request_only = dwModifier & HTTP_ADDHDR_FLAG_REQ;
4424
4425 TRACE("--> %s: %s - 0x%08x\n", debugstr_w(field), debugstr_w(value), dwModifier);
4426
4427 /* REPLACE wins out over ADD */
4428 if (dwModifier & HTTP_ADDHDR_FLAG_REPLACE)
4429 dwModifier &= ~HTTP_ADDHDR_FLAG_ADD;
4430
4431 if (dwModifier & HTTP_ADDHDR_FLAG_ADD)
4432 index = -1;
4433 else
4434 index = HTTP_GetCustomHeaderIndex(lpwhr, field, 0, request_only);
4435
4436 if (index >= 0)
4437 {
4438 if (dwModifier & HTTP_ADDHDR_FLAG_ADD_IF_NEW)
4439 {
4440 return FALSE;
4441 }
4442 lphttpHdr = &lpwhr->pCustHeaders[index];
4443 }
4444 else if (value)
4445 {
4446 HTTPHEADERW hdr;
4447
4448 hdr.lpszField = (LPWSTR)field;
4449 hdr.lpszValue = (LPWSTR)value;
4450 hdr.wFlags = hdr.wCount = 0;
4451
4452 if (dwModifier & HTTP_ADDHDR_FLAG_REQ)
4453 hdr.wFlags |= HDR_ISREQUEST;
4454
4455 return HTTP_InsertCustomHeader(lpwhr, &hdr);
4456 }
4457 /* no value to delete */
4458 else return TRUE;
4459
4460 if (dwModifier & HTTP_ADDHDR_FLAG_REQ)
4461 lphttpHdr->wFlags |= HDR_ISREQUEST;
4462 else
4463 lphttpHdr->wFlags &= ~HDR_ISREQUEST;
4464
4465 if (dwModifier & HTTP_ADDHDR_FLAG_REPLACE)
4466 {
4467 HTTP_DeleteCustomHeader( lpwhr, index );
4468
4469 if (value)
4470 {
4471 HTTPHEADERW hdr;
4472
4473 hdr.lpszField = (LPWSTR)field;
4474 hdr.lpszValue = (LPWSTR)value;
4475 hdr.wFlags = hdr.wCount = 0;
4476
4477 if (dwModifier & HTTP_ADDHDR_FLAG_REQ)
4478 hdr.wFlags |= HDR_ISREQUEST;
4479
4480 return HTTP_InsertCustomHeader(lpwhr, &hdr);
4481 }
4482
4483 return TRUE;
4484 }
4485 else if (dwModifier & COALESCEFLAGS)
4486 {
4487 LPWSTR lpsztmp;
4488 WCHAR ch = 0;
4489 INT len = 0;
4490 INT origlen = strlenW(lphttpHdr->lpszValue);
4491 INT valuelen = strlenW(value);
4492
4493 if (dwModifier & HTTP_ADDHDR_FLAG_COALESCE_WITH_COMMA)
4494 {
4495 ch = ',';
4496 lphttpHdr->wFlags |= HDR_COMMADELIMITED;
4497 }
4498 else if (dwModifier & HTTP_ADDHDR_FLAG_COALESCE_WITH_SEMICOLON)
4499 {
4500 ch = ';';
4501 lphttpHdr->wFlags |= HDR_COMMADELIMITED;
4502 }
4503
4504 len = origlen + valuelen + ((ch > 0) ? 2 : 0);
4505
4506 lpsztmp = HeapReAlloc(GetProcessHeap(), 0, lphttpHdr->lpszValue, (len+1)*sizeof(WCHAR));
4507 if (lpsztmp)
4508 {
4509 lphttpHdr->lpszValue = lpsztmp;
4510 /* FIXME: Increment lphttpHdr->wCount. Perhaps lpszValue should be an array */
4511 if (ch > 0)
4512 {
4513 lphttpHdr->lpszValue[origlen] = ch;
4514 origlen++;
4515 lphttpHdr->lpszValue[origlen] = ' ';
4516 origlen++;
4517 }
4518
4519 memcpy(&lphttpHdr->lpszValue[origlen], value, valuelen*sizeof(WCHAR));
4520 lphttpHdr->lpszValue[len] = '\0';
4521 bSuccess = TRUE;
4522 }
4523 else
4524 {
4525 WARN("HeapReAlloc (%d bytes) failed\n",len+1);
4526 INTERNET_SetLastError(ERROR_OUTOFMEMORY);
4527 }
4528 }
4529 TRACE("<-- %d\n",bSuccess);
4530 return bSuccess;
4531 }
4532
4533
4534 /***********************************************************************
4535 * HTTP_FinishedReading (internal)
4536 *
4537 * Called when all content from server has been read by client.
4538 *
4539 */
4540 static BOOL HTTP_FinishedReading(LPWININETHTTPREQW lpwhr)
4541 {
4542 WCHAR szVersion[10];
4543 WCHAR szConnectionResponse[20];
4544 DWORD dwBufferSize = sizeof(szVersion);
4545 BOOL keepalive = FALSE;
4546
4547 TRACE("\n");
4548
4549 /* as per RFC 2068, S8.1.2.1, if the client is HTTP/1.1 then assume that
4550 * the connection is keep-alive by default */
4551 if (HTTP_HttpQueryInfoW(lpwhr, HTTP_QUERY_VERSION, szVersion,
4552 &dwBufferSize, NULL) &&
4553 !strcmpiW(szVersion, g_szHttp1_1))
4554 {
4555 keepalive = TRUE;
4556 }
4557
4558 dwBufferSize = sizeof(szConnectionResponse);
4559 if (HTTP_HttpQueryInfoW(lpwhr, HTTP_QUERY_PROXY_CONNECTION, szConnectionResponse, &dwBufferSize, NULL) ||
4560 HTTP_HttpQueryInfoW(lpwhr, HTTP_QUERY_CONNECTION, szConnectionResponse, &dwBufferSize, NULL))
4561 {
4562 keepalive = !strcmpiW(szConnectionResponse, szKeepAlive);
4563 }
4564
4565 if (!keepalive)
4566 {
4567 HTTPREQ_CloseConnection(&lpwhr->hdr);
4568 }
4569
4570 /* FIXME: store data in the URL cache here */
4571
4572 return TRUE;
4573 }
4574
4575
4576 /***********************************************************************
4577 * HTTP_GetCustomHeaderIndex (internal)
4578 *
4579 * Return index of custom header from header array
4580 *
4581 */
4582 static INT HTTP_GetCustomHeaderIndex(LPWININETHTTPREQW lpwhr, LPCWSTR lpszField,
4583 int requested_index, BOOL request_only)
4584 {
4585 DWORD index;
4586
4587 TRACE("%s\n", debugstr_w(lpszField));
4588
4589 for (index = 0; index < lpwhr->nCustHeaders; index++)
4590 {
4591 if (strcmpiW(lpwhr->pCustHeaders[index].lpszField, lpszField))
4592 continue;
4593
4594 if (request_only && !(lpwhr->pCustHeaders[index].wFlags & HDR_ISREQUEST))
4595 continue;
4596
4597 if (!request_only && (lpwhr->pCustHeaders[index].wFlags & HDR_ISREQUEST))
4598 continue;
4599
4600 if (requested_index == 0)
4601 break;
4602 requested_index --;
4603 }
4604
4605 if (index >= lpwhr->nCustHeaders)
4606 index = -1;
4607
4608 TRACE("Return: %d\n", index);
4609 return index;
4610 }
4611
4612
4613 /***********************************************************************
4614 * HTTP_InsertCustomHeader (internal)
4615 *
4616 * Insert header into array
4617 *
4618 */
4619 static BOOL HTTP_InsertCustomHeader(LPWININETHTTPREQW lpwhr, LPHTTPHEADERW lpHdr)
4620 {
4621 INT count;
4622 LPHTTPHEADERW lph = NULL;
4623 BOOL r = FALSE;
4624
4625 TRACE("--> %s: %s\n", debugstr_w(lpHdr->lpszField), debugstr_w(lpHdr->lpszValue));
4626 count = lpwhr->nCustHeaders + 1;
4627 if (count > 1)
4628 lph = HeapReAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, lpwhr->pCustHeaders, sizeof(HTTPHEADERW) * count);
4629 else
4630 lph = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(HTTPHEADERW) * count);
4631
4632 if (NULL != lph)
4633 {
4634 lpwhr->pCustHeaders = lph;
4635 lpwhr->pCustHeaders[count-1].lpszField = WININET_strdupW(lpHdr->lpszField);
4636 lpwhr->pCustHeaders[count-1].lpszValue = WININET_strdupW(lpHdr->lpszValue);
4637 lpwhr->pCustHeaders[count-1].wFlags = lpHdr->wFlags;
4638 lpwhr->pCustHeaders[count-1].wCount= lpHdr->wCount;
4639 lpwhr->nCustHeaders++;
4640 r = TRUE;
4641 }
4642 else
4643 {
4644 INTERNET_SetLastError(ERROR_OUTOFMEMORY);
4645 }
4646
4647 return r;
4648 }
4649
4650
4651 /***********************************************************************
4652 * HTTP_DeleteCustomHeader (internal)
4653 *
4654 * Delete header from array
4655 * If this function is called, the indexs may change.
4656 */
4657 static BOOL HTTP_DeleteCustomHeader(LPWININETHTTPREQW lpwhr, DWORD index)
4658 {
4659 if( lpwhr->nCustHeaders <= 0 )
4660 return FALSE;
4661 if( index >= lpwhr->nCustHeaders )
4662 return FALSE;
4663 lpwhr->nCustHeaders--;
4664
4665 HeapFree(GetProcessHeap(), 0, lpwhr->pCustHeaders[index].lpszField);
4666 HeapFree(GetProcessHeap(), 0, lpwhr->pCustHeaders[index].lpszValue);
4667
4668 memmove( &lpwhr->pCustHeaders[index], &lpwhr->pCustHeaders[index+1],
4669 (lpwhr->nCustHeaders - index)* sizeof(HTTPHEADERW) );
4670 memset( &lpwhr->pCustHeaders[lpwhr->nCustHeaders], 0, sizeof(HTTPHEADERW) );
4671
4672 return TRUE;
4673 }
4674
4675
4676 /***********************************************************************
4677 * HTTP_VerifyValidHeader (internal)
4678 *
4679 * Verify the given header is not invalid for the given http request
4680 *
4681 */
4682 static BOOL HTTP_VerifyValidHeader(LPWININETHTTPREQW lpwhr, LPCWSTR field)
4683 {
4684 /* Accept-Encoding is stripped from HTTP/1.0 requests. It is invalid */
4685 if (!strcmpW(lpwhr->lpszVersion, g_szHttp1_0) && !strcmpiW(field, szAccept_Encoding))
4686 return FALSE;
4687
4688 return TRUE;
4689 }
4690
4691 /***********************************************************************
4692 * IsHostInProxyBypassList (@)
4693 *
4694 * Undocumented
4695 *
4696 */
4697 BOOL WINAPI IsHostInProxyBypassList(DWORD flags, LPCSTR szHost, DWORD length)
4698 {
4699 FIXME("STUB: flags=%d host=%s length=%d\n",flags,szHost,length);
4700 return FALSE;
4701 }
useful hacks and other patches not (yet) suitable for mainline