1 /*
2 * Copyright 2005-2006 Jacek Caban for CodeWeavers
3 *
4 * This library is free software; you can redistribute it and/or
5 * modify it under the terms of the GNU Lesser General Public
6 * License as published by the Free Software Foundation; either
7 * version 2.1 of the License, or (at your option) any later version.
8 *
9 * This library is distributed in the hope that it will be useful,
10 * but WITHOUT ANY WARRANTY; without even the implied warranty of
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
12 * Lesser General Public License for more details.
13 *
14 * You should have received a copy of the GNU Lesser General Public
15 * License along with this library; if not, write to the Free Software
16 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
17 */
18
19 #include "urlmon_main.h"
20 #include "winreg.h"
21
22 #include "wine/debug.h"
23
24 WINE_DEFAULT_DEBUG_CHANNEL(urlmon);
25
26 typedef struct {
27 LPWSTR protocol;
28 IClassFactory *cf;
29 CLSID clsid;
30 BOOL urlmon;
31
32 struct list entry;
33 } name_space;
34
35 typedef struct {
36 IClassFactory *cf;
37 CLSID clsid;
38 LPWSTR mime;
39
40 struct list entry;
41 } mime_filter;
42
43 static struct list name_space_list = LIST_INIT(name_space_list);
44 static struct list mime_filter_list = LIST_INIT(mime_filter_list);
45
46 static CRITICAL_SECTION session_cs;
47 static CRITICAL_SECTION_DEBUG session_cs_dbg =
48 {
49 0, 0, &session_cs,
50 { &session_cs_dbg.ProcessLocksList, &session_cs_dbg.ProcessLocksList },
51 0, 0, { (DWORD_PTR)(__FILE__ ": session") }
52 };
53 static CRITICAL_SECTION session_cs = { &session_cs_dbg, -1, 0, 0, 0, 0 };
54
55 static const WCHAR internet_settings_keyW[] =
56 {'S','O','F','T','W','A','R','E',
57 '\\','M','i','c','r','o','s','o','f','t',
58 '\\','W','i','n','d','o','w','s',
59 '\\','C','u','r','r','e','n','t','V','e','r','s','i','o','n',
60 '\\','I','n','t','e','r','n','e','t',' ','S','e','t','t','i','n','g','s',0};
61
find_name_space(LPCWSTR protocol)62 static name_space *find_name_space(LPCWSTR protocol)
63 {
64 name_space *iter;
65
66 LIST_FOR_EACH_ENTRY(iter, &name_space_list, name_space, entry) {
67 if(!wcsicmp(iter->protocol, protocol))
68 return iter;
69 }
70
71 return NULL;
72 }
73
get_protocol_cf(LPCWSTR schema,DWORD schema_len,CLSID * pclsid,IClassFactory ** ret)74 static HRESULT get_protocol_cf(LPCWSTR schema, DWORD schema_len, CLSID *pclsid, IClassFactory **ret)
75 {
76 WCHAR str_clsid[64];
77 HKEY hkey = NULL;
78 DWORD res, type, size;
79 CLSID clsid;
80 LPWSTR wszKey;
81 HRESULT hres;
82
83 static const WCHAR wszProtocolsKey[] =
84 {'P','R','O','T','O','C','O','L','S','\\','H','a','n','d','l','e','r','\\'};
85 static const WCHAR wszCLSID[] = {'C','L','S','I','D',0};
86
87 wszKey = heap_alloc(sizeof(wszProtocolsKey)+(schema_len+1)*sizeof(WCHAR));
88 memcpy(wszKey, wszProtocolsKey, sizeof(wszProtocolsKey));
89 memcpy(wszKey + ARRAY_SIZE(wszProtocolsKey), schema, (schema_len+1)*sizeof(WCHAR));
90
91 res = RegOpenKeyW(HKEY_CLASSES_ROOT, wszKey, &hkey);
92 heap_free(wszKey);
93 if(res != ERROR_SUCCESS) {
94 TRACE("Could not open protocol handler key\n");
95 return MK_E_SYNTAX;
96 }
97
98 size = sizeof(str_clsid);
99 res = RegQueryValueExW(hkey, wszCLSID, NULL, &type, (LPBYTE)str_clsid, &size);
100 RegCloseKey(hkey);
101 if(res != ERROR_SUCCESS || type != REG_SZ) {
102 WARN("Could not get protocol CLSID res=%d\n", res);
103 return MK_E_SYNTAX;
104 }
105
106 hres = CLSIDFromString(str_clsid, &clsid);
107 if(FAILED(hres)) {
108 WARN("CLSIDFromString failed: %08x\n", hres);
109 return hres;
110 }
111
112 if(pclsid)
113 *pclsid = clsid;
114
115 if(!ret)
116 return S_OK;
117
118 hres = CoGetClassObject(&clsid, CLSCTX_INPROC_SERVER, NULL, &IID_IClassFactory, (void**)ret);
119 return SUCCEEDED(hres) ? S_OK : MK_E_SYNTAX;
120 }
121
register_namespace(IClassFactory * cf,REFIID clsid,LPCWSTR protocol,BOOL urlmon_protocol)122 HRESULT register_namespace(IClassFactory *cf, REFIID clsid, LPCWSTR protocol, BOOL urlmon_protocol)
123 {
124 name_space *new_name_space;
125
126 new_name_space = heap_alloc(sizeof(name_space));
127
128 if(!urlmon_protocol)
129 IClassFactory_AddRef(cf);
130 new_name_space->cf = cf;
131 new_name_space->clsid = *clsid;
132 new_name_space->urlmon = urlmon_protocol;
133 new_name_space->protocol = heap_strdupW(protocol);
134
135 EnterCriticalSection(&session_cs);
136
137 list_add_head(&name_space_list, &new_name_space->entry);
138
139 LeaveCriticalSection(&session_cs);
140
141 return S_OK;
142 }
143
unregister_namespace(IClassFactory * cf,LPCWSTR protocol)144 static HRESULT unregister_namespace(IClassFactory *cf, LPCWSTR protocol)
145 {
146 name_space *iter;
147
148 EnterCriticalSection(&session_cs);
149
150 LIST_FOR_EACH_ENTRY(iter, &name_space_list, name_space, entry) {
151 if(iter->cf == cf && !wcsicmp(iter->protocol, protocol)) {
152 list_remove(&iter->entry);
153
154 LeaveCriticalSection(&session_cs);
155
156 if(!iter->urlmon)
157 IClassFactory_Release(iter->cf);
158 heap_free(iter->protocol);
159 heap_free(iter);
160 return S_OK;
161 }
162 }
163
164 LeaveCriticalSection(&session_cs);
165 return S_OK;
166 }
167
is_registered_protocol(LPCWSTR url)168 BOOL is_registered_protocol(LPCWSTR url)
169 {
170 DWORD schema_len;
171 WCHAR schema[64];
172 HRESULT hres;
173
174 hres = CoInternetParseUrl(url, PARSE_SCHEMA, 0, schema, ARRAY_SIZE(schema), &schema_len, 0);
175 if(FAILED(hres))
176 return FALSE;
177
178 return get_protocol_cf(schema, schema_len, NULL, NULL) == S_OK;
179 }
180
get_protocol_info(LPCWSTR url)181 IInternetProtocolInfo *get_protocol_info(LPCWSTR url)
182 {
183 IInternetProtocolInfo *ret = NULL;
184 IClassFactory *cf;
185 name_space *ns;
186 WCHAR schema[64];
187 DWORD schema_len;
188 HRESULT hres;
189
190 hres = CoInternetParseUrl(url, PARSE_SCHEMA, 0, schema, ARRAY_SIZE(schema), &schema_len, 0);
191 if(FAILED(hres) || !schema_len)
192 return NULL;
193
194 EnterCriticalSection(&session_cs);
195
196 ns = find_name_space(schema);
197 if(ns && !ns->urlmon) {
198 hres = IClassFactory_QueryInterface(ns->cf, &IID_IInternetProtocolInfo, (void**)&ret);
199 if(FAILED(hres))
200 hres = IClassFactory_CreateInstance(ns->cf, NULL, &IID_IInternetProtocolInfo, (void**)&ret);
201 }
202
203 LeaveCriticalSection(&session_cs);
204
205 if(ns && SUCCEEDED(hres))
206 return ret;
207
208 hres = get_protocol_cf(schema, schema_len, NULL, &cf);
209 if(FAILED(hres))
210 return NULL;
211
212 hres = IClassFactory_QueryInterface(cf, &IID_IInternetProtocolInfo, (void**)&ret);
213 if(FAILED(hres))
214 IClassFactory_CreateInstance(cf, NULL, &IID_IInternetProtocolInfo, (void**)&ret);
215 IClassFactory_Release(cf);
216
217 return ret;
218 }
219
get_protocol_handler(IUri * uri,CLSID * clsid,IClassFactory ** ret)220 HRESULT get_protocol_handler(IUri *uri, CLSID *clsid, IClassFactory **ret)
221 {
222 name_space *ns;
223 BSTR scheme;
224 HRESULT hres;
225
226 *ret = NULL;
227
228 /* FIXME: Avoid GetSchemeName call for known schemes */
229 hres = IUri_GetSchemeName(uri, &scheme);
230 if(FAILED(hres))
231 return hres;
232
233 EnterCriticalSection(&session_cs);
234
235 ns = find_name_space(scheme);
236 if(ns) {
237 *ret = ns->cf;
238 IClassFactory_AddRef(*ret);
239 if(clsid)
240 *clsid = ns->clsid;
241 }
242
243 LeaveCriticalSection(&session_cs);
244
245 hres = *ret ? S_OK : get_protocol_cf(scheme, SysStringLen(scheme), clsid, ret);
246 SysFreeString(scheme);
247 return hres;
248 }
249
get_mime_filter(LPCWSTR mime)250 IInternetProtocol *get_mime_filter(LPCWSTR mime)
251 {
252 static const WCHAR filtersW[] = {'P','r','o','t','o','c','o','l','s',
253 '\\','F','i','l','t','e','r',0 };
254 static const WCHAR CLSIDW[] = {'C','L','S','I','D',0};
255
256 IClassFactory *cf = NULL;
257 IInternetProtocol *ret;
258 mime_filter *iter;
259 HKEY hlist, hfilter;
260 WCHAR clsidw[64];
261 CLSID clsid;
262 DWORD res, type, size;
263 HRESULT hres;
264
265 EnterCriticalSection(&session_cs);
266
267 LIST_FOR_EACH_ENTRY(iter, &mime_filter_list, mime_filter, entry) {
268 if(!wcscmp(iter->mime, mime)) {
269 cf = iter->cf;
270 break;
271 }
272 }
273
274 LeaveCriticalSection(&session_cs);
275
276 if(cf) {
277 hres = IClassFactory_CreateInstance(cf, NULL, &IID_IInternetProtocol, (void**)&ret);
278 if(FAILED(hres)) {
279 WARN("CreateInstance failed: %08x\n", hres);
280 return NULL;
281 }
282
283 return ret;
284 }
285
286 res = RegOpenKeyW(HKEY_CLASSES_ROOT, filtersW, &hlist);
287 if(res != ERROR_SUCCESS) {
288 TRACE("Could not open MIME filters key\n");
289 return NULL;
290 }
291
292 res = RegOpenKeyW(hlist, mime, &hfilter);
293 CloseHandle(hlist);
294 if(res != ERROR_SUCCESS)
295 return NULL;
296
297 size = sizeof(clsidw);
298 res = RegQueryValueExW(hfilter, CLSIDW, NULL, &type, (LPBYTE)clsidw, &size);
299 CloseHandle(hfilter);
300 if(res!=ERROR_SUCCESS || type!=REG_SZ) {
301 WARN("Could not get filter CLSID for %s\n", debugstr_w(mime));
302 return NULL;
303 }
304
305 hres = CLSIDFromString(clsidw, &clsid);
306 if(FAILED(hres)) {
307 WARN("CLSIDFromString failed for %s (%x)\n", debugstr_w(mime), hres);
308 return NULL;
309 }
310
311 hres = CoCreateInstance(&clsid, NULL, CLSCTX_INPROC_SERVER, &IID_IInternetProtocol, (void**)&ret);
312 if(FAILED(hres)) {
313 WARN("CoCreateInstance failed: %08x\n", hres);
314 return NULL;
315 }
316
317 return ret;
318 }
319
InternetSession_QueryInterface(IInternetSession * iface,REFIID riid,void ** ppv)320 static HRESULT WINAPI InternetSession_QueryInterface(IInternetSession *iface,
321 REFIID riid, void **ppv)
322 {
323 TRACE("(%s %p)\n", debugstr_guid(riid), ppv);
324
325 if(IsEqualGUID(&IID_IUnknown, riid) || IsEqualGUID(&IID_IInternetSession, riid)) {
326 *ppv = iface;
327 IInternetSession_AddRef(iface);
328 return S_OK;
329 }
330
331 *ppv = NULL;
332 return E_NOINTERFACE;
333 }
334
InternetSession_AddRef(IInternetSession * iface)335 static ULONG WINAPI InternetSession_AddRef(IInternetSession *iface)
336 {
337 TRACE("()\n");
338 URLMON_LockModule();
339 return 2;
340 }
341
InternetSession_Release(IInternetSession * iface)342 static ULONG WINAPI InternetSession_Release(IInternetSession *iface)
343 {
344 TRACE("()\n");
345 URLMON_UnlockModule();
346 return 1;
347 }
348
InternetSession_RegisterNameSpace(IInternetSession * iface,IClassFactory * pCF,REFCLSID rclsid,LPCWSTR pwzProtocol,ULONG cPatterns,const LPCWSTR * ppwzPatterns,DWORD dwReserved)349 static HRESULT WINAPI InternetSession_RegisterNameSpace(IInternetSession *iface,
350 IClassFactory *pCF, REFCLSID rclsid, LPCWSTR pwzProtocol, ULONG cPatterns,
351 const LPCWSTR *ppwzPatterns, DWORD dwReserved)
352 {
353 TRACE("(%p %s %s %d %p %d)\n", pCF, debugstr_guid(rclsid), debugstr_w(pwzProtocol),
354 cPatterns, ppwzPatterns, dwReserved);
355
356 if(cPatterns || ppwzPatterns)
357 FIXME("patterns not supported\n");
358 if(dwReserved)
359 WARN("dwReserved = %d\n", dwReserved);
360
361 if(!pCF || !pwzProtocol)
362 return E_INVALIDARG;
363
364 return register_namespace(pCF, rclsid, pwzProtocol, FALSE);
365 }
366
InternetSession_UnregisterNameSpace(IInternetSession * iface,IClassFactory * pCF,LPCWSTR pszProtocol)367 static HRESULT WINAPI InternetSession_UnregisterNameSpace(IInternetSession *iface,
368 IClassFactory *pCF, LPCWSTR pszProtocol)
369 {
370 TRACE("(%p %s)\n", pCF, debugstr_w(pszProtocol));
371
372 if(!pCF || !pszProtocol)
373 return E_INVALIDARG;
374
375 return unregister_namespace(pCF, pszProtocol);
376 }
377
InternetSession_RegisterMimeFilter(IInternetSession * iface,IClassFactory * pCF,REFCLSID rclsid,LPCWSTR pwzType)378 static HRESULT WINAPI InternetSession_RegisterMimeFilter(IInternetSession *iface,
379 IClassFactory *pCF, REFCLSID rclsid, LPCWSTR pwzType)
380 {
381 mime_filter *filter;
382
383 TRACE("(%p %s %s)\n", pCF, debugstr_guid(rclsid), debugstr_w(pwzType));
384
385 filter = heap_alloc(sizeof(mime_filter));
386
387 IClassFactory_AddRef(pCF);
388 filter->cf = pCF;
389 filter->clsid = *rclsid;
390 filter->mime = heap_strdupW(pwzType);
391
392 EnterCriticalSection(&session_cs);
393
394 list_add_head(&mime_filter_list, &filter->entry);
395
396 LeaveCriticalSection(&session_cs);
397
398 return S_OK;
399 }
400
InternetSession_UnregisterMimeFilter(IInternetSession * iface,IClassFactory * pCF,LPCWSTR pwzType)401 static HRESULT WINAPI InternetSession_UnregisterMimeFilter(IInternetSession *iface,
402 IClassFactory *pCF, LPCWSTR pwzType)
403 {
404 mime_filter *iter;
405
406 TRACE("(%p %s)\n", pCF, debugstr_w(pwzType));
407
408 EnterCriticalSection(&session_cs);
409
410 LIST_FOR_EACH_ENTRY(iter, &mime_filter_list, mime_filter, entry) {
411 if(iter->cf == pCF && !wcscmp(iter->mime, pwzType)) {
412 list_remove(&iter->entry);
413
414 LeaveCriticalSection(&session_cs);
415
416 IClassFactory_Release(iter->cf);
417 heap_free(iter->mime);
418 heap_free(iter);
419 return S_OK;
420 }
421 }
422
423 LeaveCriticalSection(&session_cs);
424 return S_OK;
425 }
426
InternetSession_CreateBinding(IInternetSession * iface,LPBC pBC,LPCWSTR szUrl,IUnknown * pUnkOuter,IUnknown ** ppUnk,IInternetProtocol ** ppOInetProt,DWORD dwOption)427 static HRESULT WINAPI InternetSession_CreateBinding(IInternetSession *iface,
428 LPBC pBC, LPCWSTR szUrl, IUnknown *pUnkOuter, IUnknown **ppUnk,
429 IInternetProtocol **ppOInetProt, DWORD dwOption)
430 {
431 BindProtocol *protocol;
432 HRESULT hres;
433
434 TRACE("(%p %s %p %p %p %08x)\n", pBC, debugstr_w(szUrl), pUnkOuter, ppUnk,
435 ppOInetProt, dwOption);
436
437 if(pBC || pUnkOuter || ppUnk || dwOption)
438 FIXME("Unsupported arguments\n");
439
440 hres = create_binding_protocol(&protocol);
441 if(FAILED(hres))
442 return hres;
443
444 *ppOInetProt = (IInternetProtocol*)&protocol->IInternetProtocolEx_iface;
445 return S_OK;
446 }
447
InternetSession_SetSessionOption(IInternetSession * iface,DWORD dwOption,LPVOID pBuffer,DWORD dwBufferLength,DWORD dwReserved)448 static HRESULT WINAPI InternetSession_SetSessionOption(IInternetSession *iface,
449 DWORD dwOption, LPVOID pBuffer, DWORD dwBufferLength, DWORD dwReserved)
450 {
451 FIXME("(%08x %p %d %d)\n", dwOption, pBuffer, dwBufferLength, dwReserved);
452 return E_NOTIMPL;
453 }
454
455 static const IInternetSessionVtbl InternetSessionVtbl = {
456 InternetSession_QueryInterface,
457 InternetSession_AddRef,
458 InternetSession_Release,
459 InternetSession_RegisterNameSpace,
460 InternetSession_UnregisterNameSpace,
461 InternetSession_RegisterMimeFilter,
462 InternetSession_UnregisterMimeFilter,
463 InternetSession_CreateBinding,
464 InternetSession_SetSessionOption
465 };
466
467 static IInternetSession InternetSession = { &InternetSessionVtbl };
468
469 /***********************************************************************
470 * CoInternetGetSession (URLMON.@)
471 *
472 * Create a new internet session and return an IInternetSession interface
473 * representing it.
474 *
475 * PARAMS
476 * dwSessionMode [I] Mode for the internet session
477 * ppIInternetSession [O] Destination for creates IInternetSession object
478 * dwReserved [I] Reserved, must be 0.
479 *
480 * RETURNS
481 * Success: S_OK. ppIInternetSession contains the IInternetSession interface.
482 * Failure: E_INVALIDARG, if any argument is invalid, or
483 * E_OUTOFMEMORY if memory allocation fails.
484 */
CoInternetGetSession(DWORD dwSessionMode,IInternetSession ** ppIInternetSession,DWORD dwReserved)485 HRESULT WINAPI CoInternetGetSession(DWORD dwSessionMode, IInternetSession **ppIInternetSession,
486 DWORD dwReserved)
487 {
488 TRACE("(%d %p %d)\n", dwSessionMode, ppIInternetSession, dwReserved);
489
490 if(dwSessionMode)
491 ERR("dwSessionMode=%d\n", dwSessionMode);
492 if(dwReserved)
493 ERR("dwReserved=%d\n", dwReserved);
494
495 IInternetSession_AddRef(&InternetSession);
496 *ppIInternetSession = &InternetSession;
497 return S_OK;
498 }
499
500 /**************************************************************************
501 * UrlMkGetSessionOption (URLMON.@)
502 */
get_url_encoding(HKEY root,DWORD * encoding)503 static BOOL get_url_encoding(HKEY root, DWORD *encoding)
504 {
505 DWORD size = sizeof(DWORD), res, type;
506 HKEY hkey;
507
508 static const WCHAR wszUrlEncoding[] = {'U','r','l','E','n','c','o','d','i','n','g',0};
509
510 res = RegOpenKeyW(root, internet_settings_keyW, &hkey);
511 if(res != ERROR_SUCCESS)
512 return FALSE;
513
514 res = RegQueryValueExW(hkey, wszUrlEncoding, NULL, &type, (LPBYTE)encoding, &size);
515 RegCloseKey(hkey);
516
517 return res == ERROR_SUCCESS;
518 }
519
520 static LPWSTR user_agent;
521
ensure_useragent(void)522 static void ensure_useragent(void)
523 {
524 OSVERSIONINFOW info = {sizeof(info)};
525 const WCHAR *os_type, *is_nt;
526 WCHAR buf[512], *ret, *tmp;
527 DWORD res, idx=0;
528 size_t len, size;
529 BOOL is_wow;
530 HKEY key;
531
532 static const WCHAR formatW[] =
533 {'M','o','z','i','l','l','a','/','4','.','0',
534 ' ','(','c','o','m','p','a','t','i','b','l','e',';',
535 ' ','M','S','I','E',' ','8','.','0',';',
536 ' ','W','i','n','d','o','w','s',' ','%','s','%','d','.','%','d',';',
537 ' ','%','s','T','r','i','d','e','n','t','/','5','.','0',0};
538 static const WCHAR post_platform_keyW[] =
539 {'S','O','F','T','W','A','R','E',
540 '\\','M','i','c','r','o','s','o','f','t',
541 '\\','W','i','n','d','o','w','s',
542 '\\','C','u','r','r','e','n','t','V','e','r','s','i','o','n',
543 '\\','I','n','t','e','r','n','e','t',' ','S','e','t','t','i','n','g','s',
544 '\\','5','.','0','\\','U','s','e','r',' ','A','g','e','n','t',
545 '\\','P','o','s','t',' ','P','l','a','t','f','o','r','m',0};
546 static const WCHAR ntW[] = {'N','T',' ',0};
547 static const WCHAR win64W[] = {'W','i','n','6','4',';',' ','x','6','4',';',' ',0};
548 static const WCHAR wow64W[] = {'W','O','W','6','4',';',' ',0};
549 static const WCHAR emptyW[] = {0};
550
551 if(user_agent)
552 return;
553
554 GetVersionExW(&info);
555 is_nt = info.dwPlatformId == VER_PLATFORM_WIN32_NT ? ntW : emptyW;
556
557 if(sizeof(void*) == 8)
558 os_type = win64W;
559 else if(IsWow64Process(GetCurrentProcess(), &is_wow) && is_wow)
560 os_type = wow64W;
561 else
562 os_type = emptyW;
563
564 swprintf(buf, formatW, is_nt, info.dwMajorVersion, info.dwMinorVersion, os_type);
565 len = lstrlenW(buf);
566
567 size = len+40;
568 ret = heap_alloc(size * sizeof(WCHAR));
569 if(!ret)
570 return;
571
572 memcpy(ret, buf, len*sizeof(WCHAR));
573
574 res = RegOpenKeyW(HKEY_LOCAL_MACHINE, post_platform_keyW, &key);
575 if(res == ERROR_SUCCESS) {
576 DWORD value_len;
577
578 while(1) {
579 value_len = ARRAY_SIZE(buf);
580 res = RegEnumValueW(key, idx, buf, &value_len, NULL, NULL, NULL, NULL);
581 if(res != ERROR_SUCCESS)
582 break;
583 idx++;
584
585 if(len + value_len + 2 /* strlen("; ") */ + 1 /* trailing ')' */ >= size) {
586 tmp = heap_realloc(ret, (size*2+value_len)*sizeof(WCHAR));
587 if(!tmp)
588 break;
589 ret = tmp;
590 size = size*2+value_len;
591 }
592
593 ret[len++] = ';';
594 ret[len++] = ' ';
595 memcpy(ret+len, buf, value_len*sizeof(WCHAR));
596 len += value_len;
597 }
598
599 RegCloseKey(key);
600 }
601
602 ret[len++] = ')';
603 ret[len++] = 0;
604
605 user_agent = ret;
606 TRACE("Using user agent %s\n", debugstr_w(user_agent));
607 }
608
get_useragent(void)609 LPWSTR get_useragent(void)
610 {
611 LPWSTR ret;
612
613 ensure_useragent();
614
615 EnterCriticalSection(&session_cs);
616 ret = heap_strdupW(user_agent);
617 LeaveCriticalSection(&session_cs);
618
619 return ret;
620 }
621
UrlMkGetSessionOption(DWORD dwOption,LPVOID pBuffer,DWORD dwBufferLength,DWORD * pdwBufferLength,DWORD dwReserved)622 HRESULT WINAPI UrlMkGetSessionOption(DWORD dwOption, LPVOID pBuffer, DWORD dwBufferLength,
623 DWORD* pdwBufferLength, DWORD dwReserved)
624 {
625 TRACE("(%x, %p, %d, %p)\n", dwOption, pBuffer, dwBufferLength, pdwBufferLength);
626
627 if(dwReserved)
628 WARN("dwReserved = %d\n", dwReserved);
629
630 switch(dwOption) {
631 case URLMON_OPTION_USERAGENT: {
632 HRESULT hres = E_OUTOFMEMORY;
633 DWORD size;
634
635 if(!pdwBufferLength)
636 return E_INVALIDARG;
637
638 EnterCriticalSection(&session_cs);
639
640 ensure_useragent();
641 if(user_agent) {
642 size = WideCharToMultiByte(CP_ACP, 0, user_agent, -1, NULL, 0, NULL, NULL);
643 *pdwBufferLength = size;
644 if(size <= dwBufferLength) {
645 if(pBuffer)
646 WideCharToMultiByte(CP_ACP, 0, user_agent, -1, pBuffer, size, NULL, NULL);
647 else
648 hres = E_INVALIDARG;
649 }
650 }
651
652 LeaveCriticalSection(&session_cs);
653
654 /* Tests prove that we have to return E_OUTOFMEMORY on success. */
655 return hres;
656 }
657 case URLMON_OPTION_URL_ENCODING: {
658 DWORD encoding = 0;
659
660 if(!pBuffer || dwBufferLength < sizeof(DWORD) || !pdwBufferLength)
661 return E_INVALIDARG;
662
663 if(!get_url_encoding(HKEY_CURRENT_USER, &encoding))
664 get_url_encoding(HKEY_LOCAL_MACHINE, &encoding);
665
666 *pdwBufferLength = sizeof(DWORD);
667 *(DWORD*)pBuffer = encoding ? URL_ENCODING_DISABLE_UTF8 : URL_ENCODING_ENABLE_UTF8;
668 return S_OK;
669 }
670 default:
671 FIXME("unsupported option %x\n", dwOption);
672 }
673
674 return E_INVALIDARG;
675 }
676
677 /**************************************************************************
678 * UrlMkSetSessionOption (URLMON.@)
679 */
UrlMkSetSessionOption(DWORD dwOption,LPVOID pBuffer,DWORD dwBufferLength,DWORD Reserved)680 HRESULT WINAPI UrlMkSetSessionOption(DWORD dwOption, LPVOID pBuffer, DWORD dwBufferLength,
681 DWORD Reserved)
682 {
683 TRACE("(%x %p %x)\n", dwOption, pBuffer, dwBufferLength);
684
685 switch(dwOption) {
686 case URLMON_OPTION_USERAGENT: {
687 LPWSTR new_user_agent;
688 char *buf = pBuffer;
689 DWORD len, size;
690
691 if(!pBuffer || !dwBufferLength)
692 return E_INVALIDARG;
693
694 for(len=0; len<dwBufferLength && buf[len]; len++);
695
696 TRACE("Setting user agent %s\n", debugstr_an(buf, len));
697
698 size = MultiByteToWideChar(CP_ACP, 0, buf, len, NULL, 0);
699 new_user_agent = heap_alloc((size+1)*sizeof(WCHAR));
700 if(!new_user_agent)
701 return E_OUTOFMEMORY;
702 MultiByteToWideChar(CP_ACP, 0, buf, len, new_user_agent, size);
703 new_user_agent[size] = 0;
704
705 EnterCriticalSection(&session_cs);
706
707 heap_free(user_agent);
708 user_agent = new_user_agent;
709 update_user_agent(user_agent);
710
711 LeaveCriticalSection(&session_cs);
712 break;
713 }
714 default:
715 FIXME("Unknown option %x\n", dwOption);
716 return E_INVALIDARG;
717 }
718
719 return S_OK;
720 }
721
722 /**************************************************************************
723 * ObtainUserAgentString (URLMON.@)
724 */
ObtainUserAgentString(DWORD dwOption,LPSTR pcszUAOut,DWORD * cbSize)725 HRESULT WINAPI ObtainUserAgentString(DWORD dwOption, LPSTR pcszUAOut, DWORD *cbSize)
726 {
727 DWORD size;
728 HRESULT hres = E_FAIL;
729
730 TRACE("(%d %p %p)\n", dwOption, pcszUAOut, cbSize);
731
732 if(!pcszUAOut || !cbSize)
733 return E_INVALIDARG;
734
735 EnterCriticalSection(&session_cs);
736
737 ensure_useragent();
738 if(user_agent) {
739 size = WideCharToMultiByte(CP_ACP, 0, user_agent, -1, NULL, 0, NULL, NULL);
740
741 if(size <= *cbSize) {
742 WideCharToMultiByte(CP_ACP, 0, user_agent, -1, pcszUAOut, *cbSize, NULL, NULL);
743 hres = S_OK;
744 }else {
745 hres = E_OUTOFMEMORY;
746 }
747
748 *cbSize = size;
749 }
750
751 LeaveCriticalSection(&session_cs);
752 return hres;
753 }
754
free_session(void)755 void free_session(void)
756 {
757 name_space *ns_iter, *ns_last;
758 mime_filter *mf_iter, *mf_last;
759
760 LIST_FOR_EACH_ENTRY_SAFE(ns_iter, ns_last, &name_space_list, name_space, entry) {
761 if(!ns_iter->urlmon)
762 IClassFactory_Release(ns_iter->cf);
763 heap_free(ns_iter->protocol);
764 heap_free(ns_iter);
765 }
766
767 LIST_FOR_EACH_ENTRY_SAFE(mf_iter, mf_last, &mime_filter_list, mime_filter, entry) {
768 IClassFactory_Release(mf_iter->cf);
769 heap_free(mf_iter->mime);
770 heap_free(mf_iter);
771 }
772
773 heap_free(user_agent);
774 }
775