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