4 * Copyright (c) 2000 Patrik Stridvall
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.
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.
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
29 #define NO_SHLWAPI_REG
31 #include "wine/debug.h"
32 #include "wine/unicode.h"
36 #include "urlmon_main.h"
39 WINE_DEFAULT_DEBUG_CHANNEL(urlmon);
41 LONG URLMON_refCount = 0;
43 HINSTANCE URLMON_hInstance = 0;
44 static HMODULE hCabinet = NULL;
46 static void init_session(BOOL);
48 /***********************************************************************
49 * DllMain (URLMON.init)
51 BOOL WINAPI DllMain(HINSTANCE hinstDLL, DWORD fdwReason, LPVOID fImpLoad)
53 TRACE("%p 0x%x %p\n", hinstDLL, fdwReason, fImpLoad);
56 case DLL_PROCESS_ATTACH:
57 DisableThreadLibraryCalls(hinstDLL);
58 URLMON_hInstance = hinstDLL;
62 case DLL_PROCESS_DETACH:
64 FreeLibrary(hCabinet);
74 /***********************************************************************
75 * DllInstall (URLMON.@)
77 HRESULT WINAPI DllInstall(BOOL bInstall, LPCWSTR cmdline)
79 FIXME("(%s, %s): stub\n", bInstall?"TRUE":"FALSE",
85 /***********************************************************************
86 * DllCanUnloadNow (URLMON.@)
88 HRESULT WINAPI DllCanUnloadNow(void)
90 return URLMON_refCount != 0 ? S_FALSE : S_OK;
95 /******************************************************************************
99 const IClassFactoryVtbl *lpClassFactoryVtbl;
101 HRESULT (*pfnCreateInstance)(IUnknown *pUnkOuter, LPVOID *ppObj);
104 #define CLASSFACTORY(x) ((IClassFactory*) &(x)->lpClassFactoryVtbl)
106 static HRESULT WINAPI CF_QueryInterface(IClassFactory *iface, REFIID riid, LPVOID *ppv)
110 if(IsEqualGUID(riid, &IID_IUnknown)) {
111 TRACE("(%p)->(IID_IUnknown %p)\n", iface, ppv);
113 }else if(IsEqualGUID(riid, &IID_IClassFactory)) {
114 TRACE("(%p)->(IID_IClassFactory %p)\n", iface, ppv);
119 IUnknown_AddRef((IUnknown*)*ppv);
123 WARN("(%p)->(%s,%p),not found\n", iface, debugstr_guid(riid), ppv);
124 return E_NOINTERFACE;
127 static ULONG WINAPI CF_AddRef(IClassFactory *iface)
133 static ULONG WINAPI CF_Release(IClassFactory *iface)
135 URLMON_UnlockModule();
140 static HRESULT WINAPI CF_CreateInstance(IClassFactory *iface, IUnknown *pOuter,
141 REFIID riid, LPVOID *ppobj)
143 ClassFactory *This = (ClassFactory*)iface;
147 TRACE("(%p)->(%p,%s,%p)\n",This,pOuter,debugstr_guid(riid),ppobj);
150 if(SUCCEEDED(hres = This->pfnCreateInstance(pOuter, (LPVOID *) &punk))) {
151 hres = IUnknown_QueryInterface(punk, riid, ppobj);
152 IUnknown_Release(punk);
157 static HRESULT WINAPI CF_LockServer(LPCLASSFACTORY iface,BOOL dolock)
159 TRACE("(%d)\n", dolock);
164 URLMON_UnlockModule();
169 static const IClassFactoryVtbl ClassFactoryVtbl =
178 static const ClassFactory FileProtocolCF =
179 { &ClassFactoryVtbl, FileProtocol_Construct};
180 static const ClassFactory FtpProtocolCF =
181 { &ClassFactoryVtbl, FtpProtocol_Construct};
182 static const ClassFactory HttpProtocolCF =
183 { &ClassFactoryVtbl, HttpProtocol_Construct};
184 static const ClassFactory MkProtocolCF =
185 { &ClassFactoryVtbl, MkProtocol_Construct};
186 static const ClassFactory SecurityManagerCF =
187 { &ClassFactoryVtbl, SecManagerImpl_Construct};
188 static const ClassFactory ZoneManagerCF =
189 { &ClassFactoryVtbl, ZoneMgrImpl_Construct};
191 struct object_creation_info
198 static const WCHAR wszFile[] = {'f','i','l','e',0};
199 static const WCHAR wszFtp[] = {'f','t','p',0};
200 static const WCHAR wszHttp[] = {'h','t','t','p',0};
201 static const WCHAR wszMk[] = {'m','k',0};
203 static const struct object_creation_info object_creation[] =
205 { &CLSID_FileProtocol, CLASSFACTORY(&FileProtocolCF), wszFile },
206 { &CLSID_FtpProtocol, CLASSFACTORY(&FtpProtocolCF), wszFtp },
207 { &CLSID_HttpProtocol, CLASSFACTORY(&HttpProtocolCF), wszHttp },
208 { &CLSID_MkProtocol, CLASSFACTORY(&MkProtocolCF), wszMk },
209 { &CLSID_InternetSecurityManager, CLASSFACTORY(&SecurityManagerCF), NULL },
210 { &CLSID_InternetZoneManager, CLASSFACTORY(&ZoneManagerCF), NULL }
213 static void init_session(BOOL init)
215 IInternetSession *session;
218 CoInternetGetSession(0, &session, 0);
220 for(i=0; i < sizeof(object_creation)/sizeof(object_creation[0]); i++) {
221 if(object_creation[i].protocol) {
224 IInternetSession_RegisterNameSpace(session, object_creation[i].cf,
225 object_creation[i].clsid, object_creation[i].protocol, 0, NULL, 0);
226 /* make sure that the AddRef on the class factory doesn't keep us loaded */
227 URLMON_UnlockModule();
231 /* make sure that the Release on the class factory doesn't unload us */
233 IInternetSession_UnregisterNameSpace(session, object_creation[i].cf,
234 object_creation[i].protocol);
239 IInternetSession_Release(session);
242 /*******************************************************************************
243 * DllGetClassObject [URLMON.@]
244 * Retrieves class object from a DLL object
247 * Docs say returns STDAPI
250 * rclsid [I] CLSID for the class object
251 * riid [I] Reference to identifier of interface for class object
252 * ppv [O] Address of variable to receive interface pointer for riid
256 * Failure: CLASS_E_CLASSNOTAVAILABLE, E_OUTOFMEMORY, E_INVALIDARG,
260 HRESULT WINAPI DllGetClassObject(REFCLSID rclsid, REFIID riid, LPVOID *ppv)
264 TRACE("(%s,%s,%p)\n", debugstr_guid(rclsid), debugstr_guid(riid), ppv);
266 for (i=0; i < sizeof(object_creation)/sizeof(object_creation[0]); i++)
268 if (IsEqualGUID(object_creation[i].clsid, rclsid))
269 return IClassFactory_QueryInterface(object_creation[i].cf, riid, ppv);
272 FIXME("%s: no class found.\n", debugstr_guid(rclsid));
273 return CLASS_E_CLASSNOTAVAILABLE;
277 /***********************************************************************
278 * DllRegisterServerEx (URLMON.@)
280 HRESULT WINAPI DllRegisterServerEx(void)
282 FIXME("(void): stub\n");
287 /**************************************************************************
288 * UrlMkSetSessionOption (URLMON.@)
290 HRESULT WINAPI UrlMkSetSessionOption(DWORD dwOption, LPVOID pBuffer, DWORD dwBufferLength,
293 FIXME("(%#x, %p, %#x): stub\n", dwOption, pBuffer, dwBufferLength);
298 static const CHAR Agent[] = "Mozilla/4.0 (compatible; MSIE 6.0; Windows NT 5.0)";
300 /**************************************************************************
301 * ObtainUserAgentString (URLMON.@)
303 HRESULT WINAPI ObtainUserAgentString(DWORD dwOption, LPSTR pcszUAOut, DWORD *cbSize)
305 FIXME("(%d, %p, %p): stub\n", dwOption, pcszUAOut, cbSize);
307 if (pcszUAOut == NULL || cbSize == NULL)
310 if (*cbSize < sizeof(Agent))
312 *cbSize = sizeof(Agent);
313 return E_OUTOFMEMORY;
316 if (sizeof(Agent) < *cbSize)
317 *cbSize = sizeof(Agent);
318 lstrcpynA(pcszUAOut, Agent, *cbSize);
323 /**************************************************************************
324 * IsValidURL (URLMON.@)
326 * Determines if a specified string is a valid URL.
329 * pBC [I] ignored, must be NULL.
330 * szURL [I] string that represents the URL in question.
331 * dwReserved [I] reserved and must be zero.
336 * returns E_INVALIDARG if one or more of the args is invalid.
339 * test functionality against windows to see what a valid URL is.
341 HRESULT WINAPI IsValidURL(LPBC pBC, LPCWSTR szURL, DWORD dwReserved)
343 FIXME("(%p, %s, %d): stub\n", pBC, debugstr_w(szURL), dwReserved);
345 if (pBC != NULL || dwReserved != 0)
351 /**************************************************************************
352 * FaultInIEFeature (URLMON.@)
354 * Undocumented. Appears to be used by native shdocvw.dll.
356 HRESULT WINAPI FaultInIEFeature( HWND hwnd, uCLSSPEC * pClassSpec,
357 QUERYCONTEXT *pQuery, DWORD flags )
359 FIXME("%p %p %p %08x\n", hwnd, pClassSpec, pQuery, flags);
363 /**************************************************************************
364 * CoGetClassObjectFromURL (URLMON.@)
366 HRESULT WINAPI CoGetClassObjectFromURL( REFCLSID rclsid, LPCWSTR szCodeURL, DWORD dwFileVersionMS,
367 DWORD dwFileVersionLS, LPCWSTR szContentType,
368 LPBINDCTX pBindCtx, DWORD dwClsContext, LPVOID pvReserved,
369 REFIID riid, LPVOID *ppv )
371 FIXME("(%s %s %d %d %s %p %d %p %s %p) Stub!\n", debugstr_guid(rclsid), debugstr_w(szCodeURL),
372 dwFileVersionMS, dwFileVersionLS, debugstr_w(szContentType), pBindCtx, dwClsContext, pvReserved,
373 debugstr_guid(riid), ppv);
374 return E_NOINTERFACE;
377 /***********************************************************************
378 * ReleaseBindInfo (URLMON.@)
380 * Release the resources used by the specified BINDINFO structure.
383 * pbindinfo [I] BINDINFO to release.
388 void WINAPI ReleaseBindInfo(BINDINFO* pbindinfo)
392 TRACE("(%p)\n", pbindinfo);
394 if(!pbindinfo || !(size = pbindinfo->cbSize))
397 CoTaskMemFree(pbindinfo->szExtraInfo);
398 ReleaseStgMedium(&pbindinfo->stgmedData);
400 if(offsetof(BINDINFO, szExtraInfo) < size)
401 CoTaskMemFree(pbindinfo->szCustomVerb);
404 if(pbindinfo->pUnk && offsetof(BINDINFO, pUnk) < size)
405 IUnknown_Release(pbindinfo->pUnk);
407 memset(pbindinfo, 0, size);
408 pbindinfo->cbSize = size;
411 /***********************************************************************
412 * FindMimeFromData (URLMON.@)
414 * Determines the Multipurpose Internet Mail Extensions (MIME) type from the data provided.
416 static BOOL text_html_filter(const BYTE *b, DWORD size)
423 for(i=0; i < size-5; i++) {
425 && (b[i+1] == 'h' || b[i+1] == 'H')
426 && (b[i+2] == 't' || b[i+2] == 'T')
427 && (b[i+3] == 'm' || b[i+3] == 'M')
428 && (b[i+4] == 'l' || b[i+4] == 'L'))
435 static BOOL image_gif_filter(const BYTE *b, DWORD size)
438 && (b[0] == 'G' || b[0] == 'g')
439 && (b[1] == 'I' || b[1] == 'i')
440 && (b[2] == 'F' || b[2] == 'f')
442 && (b[4] == '7' || b[4] == '9')
443 && (b[5] == 'A' || b[5] == 'a');
446 static BOOL image_pjpeg_filter(const BYTE *b, DWORD size)
448 return size > 2 && b[0] == 0xff && b[1] == 0xd8;
451 static BOOL image_tiff_filter(const BYTE *b, DWORD size)
453 return size > 2 && b[0] == 0x4d && b[1] == 0x4d;
456 static BOOL image_xpng_filter(const BYTE *b, DWORD size)
458 static const BYTE xpng_header[] = {0x89,'P','N','G',0x0d,0x0a,0x1a,0x0a};
459 return size > sizeof(xpng_header) && !memcmp(b, xpng_header, sizeof(xpng_header));
462 static BOOL image_bmp_filter(const BYTE *b, DWORD size)
465 && b[0] == 0x42 && b[1] == 0x4d
466 && *(const DWORD *)(b+6) == 0;
469 static BOOL video_avi_filter(const BYTE *b, DWORD size)
472 && b[0] == 'R' && b[1] == 'I' && b[2] == 'F' && b[3] == 'F'
473 && b[8] == 'A' && b[9] == 'V' && b[10] == 'I' && b[11] == 0x20;
476 static BOOL video_mpeg_filter(const BYTE *b, DWORD size)
479 && !b[0] && !b[1] && b[2] == 0x01
480 && (b[3] == 0xb3 || b[3] == 0xba);
483 static BOOL application_pdf_filter(const BYTE *b, DWORD size)
485 return size > 4 && b[0] == 0x25 && b[1] == 0x50 && b[2] == 0x44 && b[3] == 0x46;
488 static BOOL application_xzip_filter(const BYTE *b, DWORD size)
490 return size > 2 && b[0] == 0x50 && b[1] == 0x4b;
493 static BOOL application_xgzip_filter(const BYTE *b, DWORD size)
495 return size > 2 && b[0] == 0x1f && b[1] == 0x8b;
498 static BOOL application_java_filter(const BYTE *b, DWORD size)
500 return size > 4 && b[0] == 0xca && b[1] == 0xfe && b[2] == 0xba && b[3] == 0xbe;
503 static BOOL application_xmsdownload(const BYTE *b, DWORD size)
505 return size > 2 && b[0] == 'M' && b[1] == 'Z';
508 static BOOL text_plain_filter(const BYTE *b, DWORD size)
512 for(ptr = b; ptr < b+size-1; ptr++) {
513 if(*ptr < 0x20 && *ptr != '\n' && *ptr != '\r' && *ptr != '\t')
520 static BOOL application_octet_stream_filter(const BYTE *b, DWORD size)
525 HRESULT WINAPI FindMimeFromData(LPBC pBC, LPCWSTR pwzUrl, LPVOID pBuffer,
526 DWORD cbSize, LPCWSTR pwzMimeProposed, DWORD dwMimeFlags,
527 LPWSTR* ppwzMimeOut, DWORD dwReserved)
529 TRACE("(%p,%s,%p,%d,%s,0x%x,%p,0x%x)\n", pBC, debugstr_w(pwzUrl), pBuffer, cbSize,
530 debugstr_w(pwzMimeProposed), dwMimeFlags, ppwzMimeOut, dwReserved);
533 WARN("dwMimeFlags=%08x\n", dwMimeFlags);
535 WARN("dwReserved=%d\n", dwReserved);
537 /* pBC seams to not be used */
539 if(!ppwzMimeOut || (!pwzUrl && !pBuffer))
542 if(pwzMimeProposed && (!pBuffer || (pBuffer && !cbSize))) {
548 len = strlenW(pwzMimeProposed)+1;
549 *ppwzMimeOut = CoTaskMemAlloc(len*sizeof(WCHAR));
550 memcpy(*ppwzMimeOut, pwzMimeProposed, len*sizeof(WCHAR));
555 const BYTE *buf = pBuffer;
560 static const WCHAR wszTextHtml[] = {'t','e','x','t','/','h','t','m','l',0};
561 static const WCHAR wszImageGif[] = {'i','m','a','g','e','/','g','i','f',0};
562 static const WCHAR wszImagePjpeg[] = {'i','m','a','g','e','/','p','j','p','e','g',0};
563 static const WCHAR wszImageTiff[] = {'i','m','a','g','e','/','t','i','f','f',0};
564 static const WCHAR wszImageXPng[] = {'i','m','a','g','e','/','x','-','p','n','g',0};
565 static const WCHAR wszImageBmp[] = {'i','m','a','g','e','/','b','m','p',0};
566 static const WCHAR wszVideoAvi[] = {'v','i','d','e','o','/','a','v','i',0};
567 static const WCHAR wszVideoMpeg[] = {'v','i','d','e','o','/','m','p','e','g',0};
568 static const WCHAR wszAppPdf[] = {'a','p','p','l','i','c','a','t','i','o','n','/',
570 static const WCHAR wszAppXZip[] = {'a','p','p','l','i','c','a','t','i','o','n','/',
571 'x','-','z','i','p','-','c','o','m','p','r','e','s','s','e','d',0};
572 static const WCHAR wszAppXGzip[] = {'a','p','p','l','i','c','a','t','i','o','n','/',
573 'x','-','g','z','i','p','-','c','o','m','p','r','e','s','s','e','d',0};
574 static const WCHAR wszAppJava[] = {'a','p','p','l','i','c','a','t','i','o','n','/',
576 static const WCHAR wszAppXMSDownload[] = {'a','p','p','l','i','c','a','t','i','o','n','/',
577 'x','-','m','s','d','o','w','n','l','o','a','d',0};
578 static const WCHAR wszTextPlain[] = {'t','e','x','t','/','p','l','a','i','n','\0'};
579 static const WCHAR wszAppOctetStream[] = {'a','p','p','l','i','c','a','t','i','o','n','/',
580 'o','c','t','e','t','-','s','t','r','e','a','m','\0'};
582 static const struct {
584 BOOL (*filter)(const BYTE *,DWORD);
586 {wszTextHtml, text_html_filter},
587 {wszImageGif, image_gif_filter},
588 {wszImagePjpeg, image_pjpeg_filter},
589 {wszImageTiff, image_tiff_filter},
590 {wszImageXPng, image_xpng_filter},
591 {wszImageBmp, image_bmp_filter},
592 {wszVideoAvi, video_avi_filter},
593 {wszVideoMpeg, video_mpeg_filter},
594 {wszAppPdf, application_pdf_filter},
595 {wszAppXZip, application_xzip_filter},
596 {wszAppXGzip, application_xgzip_filter},
597 {wszAppJava, application_java_filter},
598 {wszAppXMSDownload, application_xmsdownload},
599 {wszTextPlain, text_plain_filter},
600 {wszAppOctetStream, application_octet_stream_filter}
606 if(pwzMimeProposed && strcmpW(pwzMimeProposed, wszAppOctetStream)) {
607 for(i=0; i < sizeof(mime_filters)/sizeof(*mime_filters); i++) {
608 if(!strcmpW(pwzMimeProposed, mime_filters[i].mime))
612 if(i == sizeof(mime_filters)/sizeof(*mime_filters)
613 || mime_filters[i].filter(buf, cbSize)) {
614 len = strlenW(pwzMimeProposed)+1;
615 *ppwzMimeOut = CoTaskMemAlloc(len*sizeof(WCHAR));
616 memcpy(*ppwzMimeOut, pwzMimeProposed, len*sizeof(WCHAR));
623 if(mime_filters[i].filter(buf, cbSize))
624 ret = mime_filters[i].mime;
628 TRACE("found %s for data\n"
629 "%02x %02x %02x %02x %02x %02x %02x %02x %02x %02x %02x %02x %02x %02x %02x %02x\n",
630 debugstr_w(ret), buf[0],buf[1],buf[2],buf[3], buf[4],buf[5],buf[6],buf[7],
631 buf[8],buf[9],buf[10],buf[11], buf[12],buf[13],buf[14],buf[15]);
633 if(pwzMimeProposed) {
634 if(i == sizeof(mime_filters)/sizeof(*mime_filters))
635 ret = pwzMimeProposed;
637 /* text/html is a special case */
638 if(!strcmpW(pwzMimeProposed, wszTextHtml) && !strcmpW(ret, wszTextPlain))
642 len = strlenW(ret)+1;
643 *ppwzMimeOut = CoTaskMemAlloc(len*sizeof(WCHAR));
644 memcpy(*ppwzMimeOut, ret, len*sizeof(WCHAR));
654 static const WCHAR wszContentType[] =
655 {'C','o','n','t','e','n','t',' ','T','y','p','e','\0'};
657 ptr = strrchrW(pwzUrl, '.');
661 res = RegOpenKeyW(HKEY_CLASSES_ROOT, ptr, &hkey);
662 if(res != ERROR_SUCCESS)
663 return HRESULT_FROM_WIN32(res);
666 res = RegQueryValueExW(hkey, wszContentType, NULL, NULL, (LPBYTE)mime, &size);
668 if(res != ERROR_SUCCESS)
669 return HRESULT_FROM_WIN32(res);
671 *ppwzMimeOut = CoTaskMemAlloc(size);
672 memcpy(*ppwzMimeOut, mime, size);
679 /***********************************************************************
682 HRESULT WINAPI Extract(void *dest, LPCSTR szCabName)
684 HRESULT (WINAPI *pExtract)(void *, LPCSTR);
687 hCabinet = LoadLibraryA("cabinet.dll");
689 if (!hCabinet) return HRESULT_FROM_WIN32(GetLastError());
690 pExtract = (void *)GetProcAddress(hCabinet, "Extract");
691 if (!pExtract) return HRESULT_FROM_WIN32(GetLastError());
693 return pExtract(dest, szCabName);