?  search:
summary | log | graphiclog1 | graphiclog2 | commit | commitdiff | tree | refs | edit | fork
blame | history | raw | HEAD
msvcrt: Add _snwprintf_l.
[wine/multimedia.git] / dlls / urlmon / urlmon_main.c
blob13c61dea771f995512ae69924bb346718d2dfe48
1 /*
2 * UrlMon
3 *
4 * Copyright (c) 2000 Patrik Stridvall
5 *
6 * This library is free software; you can redistribute it and/or
7 * modify it under the terms of the GNU Lesser General Public
8 * License as published by the Free Software Foundation; either
9 * version 2.1 of the License, or (at your option) any later version.
10 *
11 * This library is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14 * Lesser General Public License for more details.
15 *
16 * You should have received a copy of the GNU Lesser General Public
17 * License along with this library; if not, write to the Free Software
18 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
19 */
20
21 #include <stdarg.h>
22
23 #include "urlmon_main.h"
24
25 #include "winreg.h"
26
27 #define NO_SHLWAPI_REG
28 #include "shlwapi.h"
29 #include "advpub.h"
30
31 #include "wine/debug.h"
32
33 #include "urlmon.h"
34
35 WINE_DEFAULT_DEBUG_CHANNEL(urlmon);
36
37 LONG URLMON_refCount = 0;
38
39 static HMODULE hCabinet = NULL;
40 static DWORD urlmon_tls = TLS_OUT_OF_INDEXES;
41
42 static void init_session(BOOL);
43
44 static struct list tls_list = LIST_INIT(tls_list);
45
46 static CRITICAL_SECTION tls_cs;
47 static CRITICAL_SECTION_DEBUG tls_cs_dbg =
48 {
49 0, 0, &tls_cs,
50 { &tls_cs_dbg.ProcessLocksList, &tls_cs_dbg.ProcessLocksList },
51 0, 0, { (DWORD_PTR)(__FILE__ ": tls") }
52 };
53
54 static CRITICAL_SECTION tls_cs = { &tls_cs_dbg, -1, 0, 0, 0, 0 };
55
56 tls_data_t *get_tls_data(void)
57 {
58 tls_data_t *data;
59
60 if(urlmon_tls == TLS_OUT_OF_INDEXES) {
61 DWORD tls = TlsAlloc();
62 if(tls == TLS_OUT_OF_INDEXES)
63 return NULL;
64
65 tls = InterlockedCompareExchange((LONG*)&urlmon_tls, tls, TLS_OUT_OF_INDEXES);
66 if(tls != urlmon_tls)
67 TlsFree(tls);
68 }
69
70 data = TlsGetValue(urlmon_tls);
71 if(!data) {
72 data = heap_alloc_zero(sizeof(tls_data_t));
73 if(!data)
74 return NULL;
75
76 EnterCriticalSection(&tls_cs);
77 list_add_tail(&tls_list, &data->entry);
78 LeaveCriticalSection(&tls_cs);
79
80 TlsSetValue(urlmon_tls, data);
81 }
82
83 return data;
84 }
85
86 static void free_tls_list(void)
87 {
88 tls_data_t *data;
89
90 if(urlmon_tls == TLS_OUT_OF_INDEXES)
91 return;
92
93 while(!list_empty(&tls_list)) {
94 data = LIST_ENTRY(list_head(&tls_list), tls_data_t, entry);
95 list_remove(&data->entry);
96 heap_free(data);
97 }
98
99 TlsFree(urlmon_tls);
100 }
101
102 static void detach_thread(void)
103 {
104 tls_data_t *data;
105
106 if(urlmon_tls == TLS_OUT_OF_INDEXES)
107 return;
108
109 data = TlsGetValue(urlmon_tls);
110 if(!data)
111 return;
112
113 EnterCriticalSection(&tls_cs);
114 list_remove(&data->entry);
115 LeaveCriticalSection(&tls_cs);
116
117 if(data->notif_hwnd) {
118 WARN("notif_hwnd not destroyed\n");
119 DestroyWindow(data->notif_hwnd);
120 }
121
122 heap_free(data);
123 }
124
125 static void process_detach(void)
126 {
127 HINTERNET internet_session;
128
129 internet_session = get_internet_session(NULL);
130 if(internet_session)
131 InternetCloseHandle(internet_session);
132
133 if (hCabinet)
134 FreeLibrary(hCabinet);
135
136 init_session(FALSE);
137 free_session();
138 free_tls_list();
139 }
140
141 /***********************************************************************
142 * DllMain (URLMON.init)
143 */
144 BOOL WINAPI DllMain(HINSTANCE hinstDLL, DWORD fdwReason, LPVOID fImpLoad)
145 {
146 TRACE("%p 0x%x %p\n", hinstDLL, fdwReason, fImpLoad);
147
148 URLMON_DllMain( hinstDLL, fdwReason, fImpLoad );
149
150 switch(fdwReason) {
151 case DLL_PROCESS_ATTACH:
152 init_session(TRUE);
153 break;
154
155 case DLL_PROCESS_DETACH:
156 process_detach();
157 DeleteCriticalSection(&tls_cs);
158 break;
159
160 case DLL_THREAD_DETACH:
161 detach_thread();
162 break;
163 }
164 return TRUE;
165 }
166
167
168 /***********************************************************************
169 * DllInstall (URLMON.@)
170 */
171 HRESULT WINAPI DllInstall(BOOL bInstall, LPCWSTR cmdline)
172 {
173 FIXME("(%s, %s): stub\n", bInstall?"TRUE":"FALSE",
174 debugstr_w(cmdline));
175
176 return S_OK;
177 }
178
179 /***********************************************************************
180 * DllCanUnloadNow (URLMON.@)
181 */
182 HRESULT WINAPI DllCanUnloadNow(void)
183 {
184 return URLMON_refCount != 0 ? S_FALSE : S_OK;
185 }
186
187
188
189 /******************************************************************************
190 * Urlmon ClassFactory
191 */
192 typedef struct {
193 IClassFactory IClassFactory_iface;
194
195 HRESULT (*pfnCreateInstance)(IUnknown *pUnkOuter, LPVOID *ppObj);
196 } ClassFactory;
197
198 static inline ClassFactory *impl_from_IClassFactory(IClassFactory *iface)
199 {
200 return CONTAINING_RECORD(iface, ClassFactory, IClassFactory_iface);
201 }
202
203 static HRESULT WINAPI CF_QueryInterface(IClassFactory *iface, REFIID riid, LPVOID *ppv)
204 {
205 *ppv = NULL;
206
207 if(IsEqualGUID(riid, &IID_IUnknown)) {
208 TRACE("(%p)->(IID_IUnknown %p)\n", iface, ppv);
209 *ppv = iface;
210 }else if(IsEqualGUID(riid, &IID_IClassFactory)) {
211 TRACE("(%p)->(IID_IClassFactory %p)\n", iface, ppv);
212 *ppv = iface;
213 }
214
215 if(*ppv) {
216 IUnknown_AddRef((IUnknown*)*ppv);
217 return S_OK;
218 }
219
220 WARN("(%p)->(%s,%p),not found\n", iface, debugstr_guid(riid), ppv);
221 return E_NOINTERFACE;
222 }
223
224 static ULONG WINAPI CF_AddRef(IClassFactory *iface)
225 {
226 URLMON_LockModule();
227 return 2;
228 }
229
230 static ULONG WINAPI CF_Release(IClassFactory *iface)
231 {
232 URLMON_UnlockModule();
233 return 1;
234 }
235
236
237 static HRESULT WINAPI CF_CreateInstance(IClassFactory *iface, IUnknown *pOuter,
238 REFIID riid, LPVOID *ppobj)
239 {
240 ClassFactory *This = impl_from_IClassFactory(iface);
241 HRESULT hres;
242 LPUNKNOWN punk;
243
244 TRACE("(%p)->(%p,%s,%p)\n",This,pOuter,debugstr_guid(riid),ppobj);
245
246 *ppobj = NULL;
247 if(SUCCEEDED(hres = This->pfnCreateInstance(pOuter, (LPVOID *) &punk))) {
248 hres = IUnknown_QueryInterface(punk, riid, ppobj);
249 IUnknown_Release(punk);
250 }
251 return hres;
252 }
253
254 static HRESULT WINAPI CF_LockServer(LPCLASSFACTORY iface,BOOL dolock)
255 {
256 TRACE("(%d)\n", dolock);
257
258 if (dolock)
259 URLMON_LockModule();
260 else
261 URLMON_UnlockModule();
262
263 return S_OK;
264 }
265
266 static const IClassFactoryVtbl ClassFactoryVtbl =
267 {
268 CF_QueryInterface,
269 CF_AddRef,
270 CF_Release,
271 CF_CreateInstance,
272 CF_LockServer
273 };
274
275 static ClassFactory FileProtocolCF =
276 { { &ClassFactoryVtbl }, FileProtocol_Construct};
277 static ClassFactory FtpProtocolCF =
278 { { &ClassFactoryVtbl }, FtpProtocol_Construct};
279 static ClassFactory GopherProtocolCF =
280 { { &ClassFactoryVtbl }, GopherProtocol_Construct};
281 static ClassFactory HttpProtocolCF =
282 { { &ClassFactoryVtbl }, HttpProtocol_Construct};
283 static ClassFactory HttpSProtocolCF =
284 { { &ClassFactoryVtbl }, HttpSProtocol_Construct};
285 static ClassFactory MkProtocolCF =
286 { { &ClassFactoryVtbl }, MkProtocol_Construct};
287 static ClassFactory SecurityManagerCF =
288 { { &ClassFactoryVtbl }, SecManagerImpl_Construct};
289 static ClassFactory ZoneManagerCF =
290 { { &ClassFactoryVtbl }, ZoneMgrImpl_Construct};
291 static ClassFactory StdURLMonikerCF =
292 { { &ClassFactoryVtbl }, StdURLMoniker_Construct};
293 static ClassFactory MimeFilterCF =
294 { { &ClassFactoryVtbl }, MimeFilter_Construct};
295
296 struct object_creation_info
297 {
298 const CLSID *clsid;
299 IClassFactory *cf;
300 LPCWSTR protocol;
301 };
302
303 static const WCHAR wszFile[] = {'f','i','l','e',0};
304 static const WCHAR wszFtp[] = {'f','t','p',0};
305 static const WCHAR wszGopher[] = {'g','o','p','h','e','r',0};
306 static const WCHAR wszHttp[] = {'h','t','t','p',0};
307 static const WCHAR wszHttps[] = {'h','t','t','p','s',0};
308 static const WCHAR wszMk[] = {'m','k',0};
309
310 static const struct object_creation_info object_creation[] =
311 {
312 { &CLSID_FileProtocol, &FileProtocolCF.IClassFactory_iface, wszFile },
313 { &CLSID_FtpProtocol, &FtpProtocolCF.IClassFactory_iface, wszFtp },
314 { &CLSID_GopherProtocol, &GopherProtocolCF.IClassFactory_iface, wszGopher },
315 { &CLSID_HttpProtocol, &HttpProtocolCF.IClassFactory_iface, wszHttp },
316 { &CLSID_HttpSProtocol, &HttpSProtocolCF.IClassFactory_iface, wszHttps },
317 { &CLSID_MkProtocol, &MkProtocolCF.IClassFactory_iface, wszMk },
318 { &CLSID_InternetSecurityManager, &SecurityManagerCF.IClassFactory_iface, NULL },
319 { &CLSID_InternetZoneManager, &ZoneManagerCF.IClassFactory_iface, NULL },
320 { &CLSID_StdURLMoniker, &StdURLMonikerCF.IClassFactory_iface, NULL },
321 { &CLSID_DeCompMimeFilter, &MimeFilterCF.IClassFactory_iface, NULL }
322 };
323
324 static void init_session(BOOL init)
325 {
326 unsigned int i;
327
328 for(i=0; i < sizeof(object_creation)/sizeof(object_creation[0]); i++) {
329
330 if(object_creation[i].protocol)
331 register_urlmon_namespace(object_creation[i].cf, object_creation[i].clsid,
332 object_creation[i].protocol, init);
333 }
334 }
335
336 /*******************************************************************************
337 * DllGetClassObject [URLMON.@]
338 * Retrieves class object from a DLL object
339 *
340 * NOTES
341 * Docs say returns STDAPI
342 *
343 * PARAMS
344 * rclsid [I] CLSID for the class object
345 * riid [I] Reference to identifier of interface for class object
346 * ppv [O] Address of variable to receive interface pointer for riid
347 *
348 * RETURNS
349 * Success: S_OK
350 * Failure: CLASS_E_CLASSNOTAVAILABLE, E_OUTOFMEMORY, E_INVALIDARG,
351 * E_UNEXPECTED
352 */
353
354 HRESULT WINAPI DllGetClassObject(REFCLSID rclsid, REFIID riid, LPVOID *ppv)
355 {
356 unsigned int i;
357 HRESULT hr;
358
359 TRACE("(%s,%s,%p)\n", debugstr_guid(rclsid), debugstr_guid(riid), ppv);
360
361 for (i=0; i < sizeof(object_creation)/sizeof(object_creation[0]); i++)
362 {
363 if (IsEqualGUID(object_creation[i].clsid, rclsid))
364 return IClassFactory_QueryInterface(object_creation[i].cf, riid, ppv);
365 }
366
367 hr = URLMON_DllGetClassObject(rclsid, riid, ppv);
368 if(SUCCEEDED(hr))
369 return hr;
370
371 FIXME("%s: no class found.\n", debugstr_guid(rclsid));
372 return CLASS_E_CLASSNOTAVAILABLE;
373 }
374
375 static HRESULT register_inf(BOOL doregister)
376 {
377 HRESULT (WINAPI *pRegInstall)(HMODULE hm, LPCSTR pszSection, const STRTABLEA* pstTable);
378 HMODULE hAdvpack;
379
380 static const WCHAR wszAdvpack[] = {'a','d','v','p','a','c','k','.','d','l','l',0};
381
382 hAdvpack = LoadLibraryW(wszAdvpack);
383 pRegInstall = (void *)GetProcAddress(hAdvpack, "RegInstall");
384
385 return pRegInstall(hProxyDll, doregister ? "RegisterDll" : "UnregisterDll", NULL);
386 }
387
388 /***********************************************************************
389 * DllRegisterServer (URLMON.@)
390 */
391 HRESULT WINAPI DllRegisterServer(void)
392 {
393 HRESULT hr;
394
395 TRACE("\n");
396
397 hr = URLMON_DllRegisterServer();
398 return SUCCEEDED(hr) ? register_inf(TRUE) : hr;
399 }
400
401 /***********************************************************************
402 * DllUnregisterServer (URLMON.@)
403 */
404 HRESULT WINAPI DllUnregisterServer(void)
405 {
406 HRESULT hr;
407
408 TRACE("\n");
409
410 hr = URLMON_DllUnregisterServer();
411 return SUCCEEDED(hr) ? register_inf(FALSE) : hr;
412 }
413
414 /***********************************************************************
415 * DllRegisterServerEx (URLMON.@)
416 */
417 HRESULT WINAPI DllRegisterServerEx(void)
418 {
419 FIXME("(void): stub\n");
420
421 return E_FAIL;
422 }
423
424 /**************************************************************************
425 * IsValidURL (URLMON.@)
426 *
427 * Determines if a specified string is a valid URL.
428 *
429 * PARAMS
430 * pBC [I] ignored, should be NULL.
431 * szURL [I] string that represents the URL in question.
432 * dwReserved [I] reserved and must be zero.
433 *
434 * RETURNS
435 * Success: S_OK.
436 * Failure: S_FALSE.
437 * returns E_INVALIDARG if one or more of the args is invalid.
438 *
439 * TODO:
440 * test functionality against windows to see what a valid URL is.
441 */
442 HRESULT WINAPI IsValidURL(LPBC pBC, LPCWSTR szURL, DWORD dwReserved)
443 {
444 FIXME("(%p, %s, %d): stub\n", pBC, debugstr_w(szURL), dwReserved);
445
446 if (dwReserved || !szURL)
447 return E_INVALIDARG;
448
449 return S_OK;
450 }
451
452 /**************************************************************************
453 * FaultInIEFeature (URLMON.@)
454 *
455 * Undocumented. Appears to be used by native shdocvw.dll.
456 */
457 HRESULT WINAPI FaultInIEFeature( HWND hwnd, uCLSSPEC * pClassSpec,
458 QUERYCONTEXT *pQuery, DWORD flags )
459 {
460 FIXME("%p %p %p %08x\n", hwnd, pClassSpec, pQuery, flags);
461 return E_NOTIMPL;
462 }
463
464 /**************************************************************************
465 * CoGetClassObjectFromURL (URLMON.@)
466 */
467 HRESULT WINAPI CoGetClassObjectFromURL( REFCLSID rclsid, LPCWSTR szCodeURL, DWORD dwFileVersionMS,
468 DWORD dwFileVersionLS, LPCWSTR szContentType,
469 LPBINDCTX pBindCtx, DWORD dwClsContext, LPVOID pvReserved,
470 REFIID riid, LPVOID *ppv )
471 {
472 FIXME("(%s %s %d %d %s %p %d %p %s %p) Stub!\n", debugstr_guid(rclsid), debugstr_w(szCodeURL),
473 dwFileVersionMS, dwFileVersionLS, debugstr_w(szContentType), pBindCtx, dwClsContext, pvReserved,
474 debugstr_guid(riid), ppv);
475 return E_NOINTERFACE;
476 }
477
478 /***********************************************************************
479 * ReleaseBindInfo (URLMON.@)
480 *
481 * Release the resources used by the specified BINDINFO structure.
482 *
483 * PARAMS
484 * pbindinfo [I] BINDINFO to release.
485 *
486 * RETURNS
487 * Nothing.
488 */
489 void WINAPI ReleaseBindInfo(BINDINFO* pbindinfo)
490 {
491 DWORD size;
492
493 TRACE("(%p)\n", pbindinfo);
494
495 if(!pbindinfo || !(size = pbindinfo->cbSize))
496 return;
497
498 CoTaskMemFree(pbindinfo->szExtraInfo);
499 ReleaseStgMedium(&pbindinfo->stgmedData);
500
501 if(offsetof(BINDINFO, szExtraInfo) < size)
502 CoTaskMemFree(pbindinfo->szCustomVerb);
503
504 if(pbindinfo->pUnk && offsetof(BINDINFO, pUnk) < size)
505 IUnknown_Release(pbindinfo->pUnk);
506
507 memset(pbindinfo, 0, size);
508 pbindinfo->cbSize = size;
509 }
510
511 /***********************************************************************
512 * CopyStgMedium (URLMON.@)
513 */
514 HRESULT WINAPI CopyStgMedium(const STGMEDIUM *src, STGMEDIUM *dst)
515 {
516 TRACE("(%p %p)\n", src, dst);
517
518 if(!src || !dst)
519 return E_POINTER;
520
521 *dst = *src;
522
523 switch(dst->tymed) {
524 case TYMED_NULL:
525 break;
526 case TYMED_FILE:
527 if(src->u.lpszFileName && !src->pUnkForRelease) {
528 DWORD size = (strlenW(src->u.lpszFileName)+1)*sizeof(WCHAR);
529 dst->u.lpszFileName = CoTaskMemAlloc(size);
530 memcpy(dst->u.lpszFileName, src->u.lpszFileName, size);
531 }
532 break;
533 case TYMED_ISTREAM:
534 if(dst->u.pstm)
535 IStream_AddRef(dst->u.pstm);
536 break;
537 case TYMED_ISTORAGE:
538 if(dst->u.pstg)
539 IStorage_AddRef(dst->u.pstg);
540 break;
541 default:
542 FIXME("Unimplemented tymed %d\n", src->tymed);
543 }
544
545 if(dst->pUnkForRelease)
546 IUnknown_AddRef(dst->pUnkForRelease);
547
548 return S_OK;
549 }
550
551 static BOOL text_richtext_filter(const BYTE *b, DWORD size)
552 {
553 return size > 5 && !memcmp(b, "{\\rtf", 5);
554 }
555
556 static BOOL text_html_filter(const BYTE *b, DWORD size)
557 {
558 DWORD i;
559
560 if(size < 5)
561 return FALSE;
562
563 for(i=0; i < size-5; i++) {
564 if(b[i] == '<'
565 && (b[i+1] == 'h' || b[i+1] == 'H')
566 && (b[i+2] == 't' || b[i+2] == 'T')
567 && (b[i+3] == 'm' || b[i+3] == 'M')
568 && (b[i+4] == 'l' || b[i+4] == 'L'))
569 return TRUE;
570 }
571
572 return FALSE;
573 }
574
575 static BOOL audio_basic_filter(const BYTE *b, DWORD size)
576 {
577 return size > 4
578 && b[0] == '.' && b[1] == 's' && b[2] == 'n' && b[3] == 'd';
579 }
580
581 static BOOL audio_wav_filter(const BYTE *b, DWORD size)
582 {
583 return size > 12
584 && b[0] == 'R' && b[1] == 'I' && b[2] == 'F' && b[3] == 'F'
585 && b[8] == 'W' && b[9] == 'A' && b[10] == 'V' && b[11] == 'E';
586 }
587
588 static BOOL image_gif_filter(const BYTE *b, DWORD size)
589 {
590 return size >= 6
591 && (b[0] == 'G' || b[0] == 'g')
592 && (b[1] == 'I' || b[1] == 'i')
593 && (b[2] == 'F' || b[2] == 'f')
594 && b[3] == '8'
595 && (b[4] == '7' || b[4] == '9')
596 && (b[5] == 'A' || b[5] == 'a');
597 }
598
599 static BOOL image_pjpeg_filter(const BYTE *b, DWORD size)
600 {
601 return size > 2 && b[0] == 0xff && b[1] == 0xd8;
602 }
603
604 static BOOL image_tiff_filter(const BYTE *b, DWORD size)
605 {
606 static const BYTE magic1[] = {0x4d,0x4d,0x00,0x2a};
607 static const BYTE magic2[] = {0x49,0x49,0x2a,0xff};
608
609 return size >= 4 && (!memcmp(b, magic1, 4) || !memcmp(b, magic2, 4));
610 }
611
612 static BOOL image_xpng_filter(const BYTE *b, DWORD size)
613 {
614 static const BYTE xpng_header[] = {0x89,'P','N','G',0x0d,0x0a,0x1a,0x0a};
615 return size > sizeof(xpng_header) && !memcmp(b, xpng_header, sizeof(xpng_header));
616 }
617
618 static BOOL image_bmp_filter(const BYTE *b, DWORD size)
619 {
620 return size >= 14
621 && b[0] == 0x42 && b[1] == 0x4d
622 && *(const DWORD *)(b+6) == 0;
623 }
624
625 static BOOL video_avi_filter(const BYTE *b, DWORD size)
626 {
627 return size > 12
628 && b[0] == 'R' && b[1] == 'I' && b[2] == 'F' && b[3] == 'F'
629 && b[8] == 'A' && b[9] == 'V' && b[10] == 'I' && b[11] == 0x20;
630 }
631
632 static BOOL video_mpeg_filter(const BYTE *b, DWORD size)
633 {
634 return size > 4
635 && !b[0] && !b[1] && b[2] == 0x01
636 && (b[3] == 0xb3 || b[3] == 0xba);
637 }
638
639 static BOOL application_postscript_filter(const BYTE *b, DWORD size)
640 {
641 return size > 2 && b[0] == '%' && b[1] == '!';
642 }
643
644 static BOOL application_pdf_filter(const BYTE *b, DWORD size)
645 {
646 return size > 4 && b[0] == 0x25 && b[1] == 0x50 && b[2] == 0x44 && b[3] == 0x46;
647 }
648
649 static BOOL application_xzip_filter(const BYTE *b, DWORD size)
650 {
651 return size > 2 && b[0] == 0x50 && b[1] == 0x4b;
652 }
653
654 static BOOL application_xgzip_filter(const BYTE *b, DWORD size)
655 {
656 return size > 2 && b[0] == 0x1f && b[1] == 0x8b;
657 }
658
659 static BOOL application_java_filter(const BYTE *b, DWORD size)
660 {
661 return size > 4 && b[0] == 0xca && b[1] == 0xfe && b[2] == 0xba && b[3] == 0xbe;
662 }
663
664 static BOOL application_xmsdownload(const BYTE *b, DWORD size)
665 {
666 return size > 2 && b[0] == 'M' && b[1] == 'Z';
667 }
668
669 static BOOL text_plain_filter(const BYTE *b, DWORD size)
670 {
671 const BYTE *ptr;
672
673 for(ptr = b; ptr < b+size-1; ptr++) {
674 if(*ptr < 0x20 && *ptr != '\n' && *ptr != '\r' && *ptr != '\t')
675 return FALSE;
676 }
677
678 return TRUE;
679 }
680
681 static BOOL application_octet_stream_filter(const BYTE *b, DWORD size)
682 {
683 return TRUE;
684 }
685
686 static HRESULT find_mime_from_buffer(const BYTE *buf, DWORD size, const WCHAR *proposed_mime, WCHAR **ret_mime)
687 {
688 LPCWSTR ret = NULL;
689 DWORD len, i;
690
691 static const WCHAR text_htmlW[] = {'t','e','x','t','/','h','t','m','l',0};
692 static const WCHAR text_richtextW[] = {'t','e','x','t','/','r','i','c','h','t','e','x','t',0};
693 static const WCHAR audio_basicW[] = {'a','u','d','i','o','/','b','a','s','i','c',0};
694 static const WCHAR audio_wavW[] = {'a','u','d','i','o','/','w','a','v',0};
695 static const WCHAR image_gifW[] = {'i','m','a','g','e','/','g','i','f',0};
696 static const WCHAR image_pjpegW[] = {'i','m','a','g','e','/','p','j','p','e','g',0};
697 static const WCHAR image_tiffW[] = {'i','m','a','g','e','/','t','i','f','f',0};
698 static const WCHAR image_xpngW[] = {'i','m','a','g','e','/','x','-','p','n','g',0};
699 static const WCHAR image_bmpW[] = {'i','m','a','g','e','/','b','m','p',0};
700 static const WCHAR video_aviW[] = {'v','i','d','e','o','/','a','v','i',0};
701 static const WCHAR video_mpegW[] = {'v','i','d','e','o','/','m','p','e','g',0};
702 static const WCHAR app_postscriptW[] =
703 {'a','p','p','l','i','c','a','t','i','o','n','/','p','o','s','t','s','c','r','i','p','t',0};
704 static const WCHAR app_pdfW[] = {'a','p','p','l','i','c','a','t','i','o','n','/','p','d','f',0};
705 static const WCHAR app_xzipW[] = {'a','p','p','l','i','c','a','t','i','o','n','/',
706 'x','-','z','i','p','-','c','o','m','p','r','e','s','s','e','d',0};
707 static const WCHAR app_xgzipW[] = {'a','p','p','l','i','c','a','t','i','o','n','/',
708 'x','-','g','z','i','p','-','c','o','m','p','r','e','s','s','e','d',0};
709 static const WCHAR app_javaW[] = {'a','p','p','l','i','c','a','t','i','o','n','/',
710 'j','a','v','a',0};
711 static const WCHAR app_xmsdownloadW[] = {'a','p','p','l','i','c','a','t','i','o','n','/',
712 'x','-','m','s','d','o','w','n','l','o','a','d',0};
713 static const WCHAR text_plainW[] = {'t','e','x','t','/','p','l','a','i','n','\0'};
714 static const WCHAR app_octetstreamW[] = {'a','p','p','l','i','c','a','t','i','o','n','/',
715 'o','c','t','e','t','-','s','t','r','e','a','m','\0'};
716
717 static const struct {
718 LPCWSTR mime;
719 BOOL (*filter)(const BYTE *,DWORD);
720 } mime_filters[] = {
721 {text_htmlW, text_html_filter},
722 {text_richtextW, text_richtext_filter},
723 /* {audio_xaiffW, audio_xaiff_filter}, */
724 {audio_basicW, audio_basic_filter},
725 {audio_wavW, audio_wav_filter},
726 {image_gifW, image_gif_filter},
727 {image_pjpegW, image_pjpeg_filter},
728 {image_tiffW, image_tiff_filter},
729 {image_xpngW, image_xpng_filter},
730 /* {image_xbitmapW, image_xbitmap_filter}, */
731 {image_bmpW, image_bmp_filter},
732 /* {image_xjgW, image_xjg_filter}, */
733 /* {image_xemfW, image_xemf_filter}, */
734 /* {image_xwmfW, image_xwmf_filter}, */
735 {video_aviW, video_avi_filter},
736 {video_mpegW, video_mpeg_filter},
737 {app_postscriptW, application_postscript_filter},
738 /* {app_base64W, application_base64_filter}, */
739 /* {app_macbinhex40W, application_macbinhex40_filter}, */
740 {app_pdfW, application_pdf_filter},
741 /* {app_zcompressedW, application_xcompressed_filter}, */
742 {app_xzipW, application_xzip_filter},
743 {app_xgzipW, application_xgzip_filter},
744 {app_javaW, application_java_filter},
745 {app_xmsdownloadW, application_xmsdownload},
746 {text_plainW, text_plain_filter},
747 {app_octetstreamW, application_octet_stream_filter}
748 };
749
750 if(!buf || !size) {
751 if(!proposed_mime)
752 return E_FAIL;
753
754 len = strlenW(proposed_mime)+1;
755 *ret_mime = CoTaskMemAlloc(len*sizeof(WCHAR));
756 if(!*ret_mime)
757 return E_OUTOFMEMORY;
758
759 memcpy(*ret_mime, proposed_mime, len*sizeof(WCHAR));
760 return S_OK;
761 }
762
763 if(proposed_mime && strcmpW(proposed_mime, app_octetstreamW)) {
764 for(i=0; i < sizeof(mime_filters)/sizeof(*mime_filters); i++) {
765 if(!strcmpW(proposed_mime, mime_filters[i].mime))
766 break;
767 }
768
769 if(i == sizeof(mime_filters)/sizeof(*mime_filters) || mime_filters[i].filter(buf, size)) {
770 len = strlenW(proposed_mime)+1;
771 *ret_mime = CoTaskMemAlloc(len*sizeof(WCHAR));
772 if(!*ret_mime)
773 return E_OUTOFMEMORY;
774
775 memcpy(*ret_mime, proposed_mime, len*sizeof(WCHAR));
776 return S_OK;
777 }
778 }
779
780 i=0;
781 while(!ret) {
782 if(mime_filters[i].filter(buf, size))
783 ret = mime_filters[i].mime;
784 i++;
785 }
786
787 TRACE("found %s for %s\n", debugstr_w(ret), debugstr_an((const char*)buf, min(32, size)));
788
789 if(proposed_mime) {
790 if(i == sizeof(mime_filters)/sizeof(*mime_filters))
791 ret = proposed_mime;
792
793 /* text/html is a special case */
794 if(!strcmpW(proposed_mime, text_htmlW) && !strcmpW(ret, text_plainW))
795 ret = text_htmlW;
796 }
797
798 len = strlenW(ret)+1;
799 *ret_mime = CoTaskMemAlloc(len*sizeof(WCHAR));
800 if(!*ret_mime)
801 return E_OUTOFMEMORY;
802
803 memcpy(*ret_mime, ret, len*sizeof(WCHAR));
804 return S_OK;
805 }
806
807 /***********************************************************************
808 * FindMimeFromData (URLMON.@)
809 *
810 * Determines the Multipurpose Internet Mail Extensions (MIME) type from the data provided.
811 */
812 HRESULT WINAPI FindMimeFromData(LPBC pBC, LPCWSTR pwzUrl, LPVOID pBuffer,
813 DWORD cbSize, LPCWSTR pwzMimeProposed, DWORD dwMimeFlags,
814 LPWSTR* ppwzMimeOut, DWORD dwReserved)
815 {
816 TRACE("(%p,%s,%p,%d,%s,0x%x,%p,0x%x)\n", pBC, debugstr_w(pwzUrl), pBuffer, cbSize,
817 debugstr_w(pwzMimeProposed), dwMimeFlags, ppwzMimeOut, dwReserved);
818
819 if(dwMimeFlags)
820 WARN("dwMimeFlags=%08x\n", dwMimeFlags);
821 if(dwReserved)
822 WARN("dwReserved=%d\n", dwReserved);
823
824 /* pBC seams to not be used */
825
826 if(!ppwzMimeOut || (!pwzUrl && !pBuffer))
827 return E_INVALIDARG;
828
829 if(pwzMimeProposed || pBuffer)
830 return find_mime_from_buffer(pBuffer, cbSize, pwzMimeProposed, ppwzMimeOut);
831
832 if(pwzUrl) {
833 HKEY hkey;
834 DWORD res, size;
835 LPCWSTR ptr;
836 WCHAR mime[64];
837
838 static const WCHAR wszContentType[] =
839 {'C','o','n','t','e','n','t',' ','T','y','p','e','\0'};
840
841 ptr = strrchrW(pwzUrl, '.');
842 if(!ptr)
843 return E_FAIL;
844
845 res = RegOpenKeyW(HKEY_CLASSES_ROOT, ptr, &hkey);
846 if(res != ERROR_SUCCESS)
847 return HRESULT_FROM_WIN32(res);
848
849 size = sizeof(mime);
850 res = RegQueryValueExW(hkey, wszContentType, NULL, NULL, (LPBYTE)mime, &size);
851 RegCloseKey(hkey);
852 if(res != ERROR_SUCCESS)
853 return HRESULT_FROM_WIN32(res);
854
855 *ppwzMimeOut = CoTaskMemAlloc(size);
856 memcpy(*ppwzMimeOut, mime, size);
857 return S_OK;
858 }
859
860 return E_FAIL;
861 }
862
863 /***********************************************************************
864 * GetClassFileOrMime (URLMON.@)
865 *
866 * Determines the class ID from the bind context, file name or MIME type.
867 */
868 HRESULT WINAPI GetClassFileOrMime(LPBC pBC, LPCWSTR pszFilename,
869 LPVOID pBuffer, DWORD cbBuffer, LPCWSTR pszMimeType, DWORD dwReserved,
870 CLSID *pclsid)
871 {
872 FIXME("(%p, %s, %p, %d, %p, 0x%08x, %p): stub\n", pBC,
873 debugstr_w(pszFilename), pBuffer, cbBuffer, debugstr_w(pszMimeType),
874 dwReserved, pclsid);
875 return E_NOTIMPL;
876 }
877
878 /***********************************************************************
879 * Extract (URLMON.@)
880 */
881 HRESULT WINAPI Extract(void *dest, LPCSTR szCabName)
882 {
883 HRESULT (WINAPI *pExtract)(void *, LPCSTR);
884
885 if (!hCabinet)
886 hCabinet = LoadLibraryA("cabinet.dll");
887
888 if (!hCabinet) return HRESULT_FROM_WIN32(GetLastError());
889 pExtract = (void *)GetProcAddress(hCabinet, "Extract");
890 if (!pExtract) return HRESULT_FROM_WIN32(GetLastError());
891
892 return pExtract(dest, szCabName);
893 }
894
895 /***********************************************************************
896 * IsLoggingEnabledA (URLMON.@)
897 */
898 BOOL WINAPI IsLoggingEnabledA(LPCSTR url)
899 {
900 FIXME("(%s)\n", debugstr_a(url));
901 return FALSE;
902 }
903
904 /***********************************************************************
905 * IsLoggingEnabledW (URLMON.@)
906 */
907 BOOL WINAPI IsLoggingEnabledW(LPCWSTR url)
908 {
909 FIXME("(%s)\n", debugstr_w(url));
910 return FALSE;
911 }
912
913 /***********************************************************************
914 * IsProtectedModeURL (URLMON.111)
915 * Undocumented, added in IE7
916 */
917 BOOL WINAPI IsProtectedModeURL(const WCHAR *url)
918 {
919 FIXME("stub: %s\n", debugstr_w(url));
920 return TRUE;
921 }
922
923 /***********************************************************************
924 * LogSqmBits (URLMON.410)
925 * Undocumented, added in IE8
926 */
927 int WINAPI LogSqmBits(DWORD unk1, DWORD unk2)
928 {
929 FIXME("stub: %d %d\n", unk1, unk2);
930 return 0;
931 }
932
933 /***********************************************************************
934 * LogSqmUXCommandOffsetInternal (URLMON.423)
935 * Undocumented, added in IE8
936 */
937 void WINAPI LogSqmUXCommandOffsetInternal(DWORD unk1, DWORD unk2, DWORD unk3, DWORD unk4)
938 {
939 FIXME("stub: %d %d %d %d\n", unk1, unk2, unk3, unk4);
940 }
941
942 /***********************************************************************
943 * MapUriToBrowserEmulationState (URLMON.444)
944 * Undocumented, added in IE8
945 */
946 int WINAPI MapUriToBrowserEmulationState(DWORD unk1, DWORD unk2, DWORD unk3)
947 {
948 FIXME("stub: %d %d %d\n", unk1, unk2, unk3);
949 return 0;
950 }
951
952 /***********************************************************************
953 * MapBrowserEmulationModeToUserAgent (URLMON.445)
954 * Undocumented, added in IE8
955 */
956 int WINAPI MapBrowserEmulationModeToUserAgent(DWORD unk1, DWORD unk2)
957 {
958 FIXME("stub: %d %d\n", unk1, unk2);
959 return 0;
960 }
961
962 /***********************************************************************
963 * RegisterMediaTypes
964 * Added in IE3, registers known MIME-type strings.
965 */
966 HRESULT WINAPI RegisterMediaTypes(UINT types, LPCSTR *szTypes, CLIPFORMAT *cfTypes)
967 {
968 FIXME("stub: %u %p %p\n", types, szTypes, cfTypes);
969 return E_INVALIDARG;
970 }
Maarten's multimedia branch