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