urlmon: Win64 printf format warning fixes.
[wine.git] / dlls / urlmon / urlmon_main.c
blobf7e281eac6dda9ea6291bfd998d9481746ab2192
1 /*
2 * UrlMon
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
21 #include <stdarg.h>
23 #define COBJMACROS
25 #include "windef.h"
26 #include "winbase.h"
27 #include "winreg.h"
29 #define NO_SHLWAPI_REG
30 #include "shlwapi.h"
31 #include "wine/debug.h"
32 #include "wine/unicode.h"
34 #include "winuser.h"
35 #include "urlmon.h"
36 #include "urlmon_main.h"
37 #include "ole2.h"
39 WINE_DEFAULT_DEBUG_CHANNEL(urlmon);
41 LONG URLMON_refCount = 0;
43 HINSTANCE URLMON_hInstance = 0;
44 static HMODULE hCabinet = NULL;
46 DWORD urlmon_tls = 0;
48 static void init_session(BOOL);
50 /***********************************************************************
51 * DllMain (URLMON.init)
53 BOOL WINAPI DllMain(HINSTANCE hinstDLL, DWORD fdwReason, LPVOID fImpLoad)
55 TRACE("%p 0x%x %p\n", hinstDLL, fdwReason, fImpLoad);
57 switch(fdwReason) {
58 case DLL_PROCESS_ATTACH:
59 DisableThreadLibraryCalls(hinstDLL);
60 URLMON_hInstance = hinstDLL;
61 init_session(TRUE);
62 break;
64 case DLL_PROCESS_DETACH:
65 if(urlmon_tls)
66 TlsFree(urlmon_tls);
67 if (hCabinet)
68 FreeLibrary(hCabinet);
69 hCabinet = NULL;
70 init_session(FALSE);
71 URLMON_hInstance = 0;
72 break;
74 return TRUE;
78 /***********************************************************************
79 * DllInstall (URLMON.@)
81 HRESULT WINAPI DllInstall(BOOL bInstall, LPCWSTR cmdline)
83 FIXME("(%s, %s): stub\n", bInstall?"TRUE":"FALSE",
84 debugstr_w(cmdline));
86 return S_OK;
89 /***********************************************************************
90 * DllCanUnloadNow (URLMON.@)
92 HRESULT WINAPI DllCanUnloadNow(void)
94 return URLMON_refCount != 0 ? S_FALSE : S_OK;
99 /******************************************************************************
100 * Urlmon ClassFactory
102 typedef struct {
103 const IClassFactoryVtbl *lpClassFactoryVtbl;
105 HRESULT (*pfnCreateInstance)(IUnknown *pUnkOuter, LPVOID *ppObj);
106 } ClassFactory;
108 #define CLASSFACTORY(x) ((IClassFactory*) &(x)->lpClassFactoryVtbl)
110 static HRESULT WINAPI CF_QueryInterface(IClassFactory *iface, REFIID riid, LPVOID *ppv)
112 *ppv = NULL;
114 if(IsEqualGUID(riid, &IID_IUnknown)) {
115 TRACE("(%p)->(IID_IUnknown %p)\n", iface, ppv);
116 *ppv = iface;
117 }else if(IsEqualGUID(riid, &IID_IClassFactory)) {
118 TRACE("(%p)->(IID_IClassFactory %p)\n", iface, ppv);
119 *ppv = iface;
122 if(*ppv) {
123 IUnknown_AddRef((IUnknown*)*ppv);
124 return S_OK;
127 WARN("(%p)->(%s,%p),not found\n", iface, debugstr_guid(riid), ppv);
128 return E_NOINTERFACE;
131 static ULONG WINAPI CF_AddRef(IClassFactory *iface)
133 URLMON_LockModule();
134 return 2;
137 static ULONG WINAPI CF_Release(IClassFactory *iface)
139 URLMON_UnlockModule();
140 return 1;
144 static HRESULT WINAPI CF_CreateInstance(IClassFactory *iface, IUnknown *pOuter,
145 REFIID riid, LPVOID *ppobj)
147 ClassFactory *This = (ClassFactory*)iface;
148 HRESULT hres;
149 LPUNKNOWN punk;
151 TRACE("(%p)->(%p,%s,%p)\n",This,pOuter,debugstr_guid(riid),ppobj);
153 *ppobj = NULL;
154 if(SUCCEEDED(hres = This->pfnCreateInstance(pOuter, (LPVOID *) &punk))) {
155 hres = IUnknown_QueryInterface(punk, riid, ppobj);
156 IUnknown_Release(punk);
158 return hres;
161 static HRESULT WINAPI CF_LockServer(LPCLASSFACTORY iface,BOOL dolock)
163 TRACE("(%d)\n", dolock);
165 if (dolock)
166 URLMON_LockModule();
167 else
168 URLMON_UnlockModule();
170 return S_OK;
173 static const IClassFactoryVtbl ClassFactoryVtbl =
175 CF_QueryInterface,
176 CF_AddRef,
177 CF_Release,
178 CF_CreateInstance,
179 CF_LockServer
182 static const ClassFactory FileProtocolCF =
183 { &ClassFactoryVtbl, FileProtocol_Construct};
184 static const ClassFactory FtpProtocolCF =
185 { &ClassFactoryVtbl, FtpProtocol_Construct};
186 static const ClassFactory HttpProtocolCF =
187 { &ClassFactoryVtbl, HttpProtocol_Construct};
188 static const ClassFactory SecurityManagerCF =
189 { &ClassFactoryVtbl, SecManagerImpl_Construct};
190 static const ClassFactory ZoneManagerCF =
191 { &ClassFactoryVtbl, ZoneMgrImpl_Construct};
193 struct object_creation_info
195 const CLSID *clsid;
196 IClassFactory *cf;
197 LPCWSTR protocol;
200 static const WCHAR wszFile[] = {'f','i','l','e',0};
201 static const WCHAR wszFtp[] = {'f','t','p',0};
202 static const WCHAR wszHttp[] = {'h','t','t','p',0};
204 static const struct object_creation_info object_creation[] =
206 { &CLSID_FileProtocol, CLASSFACTORY(&FileProtocolCF), wszFile },
207 { &CLSID_FtpProtocol, CLASSFACTORY(&FtpProtocolCF), wszFtp },
208 { &CLSID_HttpProtocol, CLASSFACTORY(&HttpProtocolCF), wszHttp },
209 { &CLSID_InternetSecurityManager, CLASSFACTORY(&SecurityManagerCF), NULL },
210 { &CLSID_InternetZoneManager, CLASSFACTORY(&ZoneManagerCF), NULL }
213 static void init_session(BOOL init)
215 IInternetSession *session;
216 int i;
218 CoInternetGetSession(0, &session, 0);
220 for(i=0; i < sizeof(object_creation)/sizeof(object_creation[0]); i++) {
221 if(object_creation[i].protocol) {
222 if(init)
223 IInternetSession_RegisterNameSpace(session, object_creation[i].cf,
224 object_creation[i].clsid, object_creation[i].protocol, 0, NULL, 0);
225 else
226 IInternetSession_UnregisterNameSpace(session, object_creation[i].cf,
227 object_creation[i].protocol);
231 IInternetSession_Release(session);
234 /*******************************************************************************
235 * DllGetClassObject [URLMON.@]
236 * Retrieves class object from a DLL object
238 * NOTES
239 * Docs say returns STDAPI
241 * PARAMS
242 * rclsid [I] CLSID for the class object
243 * riid [I] Reference to identifier of interface for class object
244 * ppv [O] Address of variable to receive interface pointer for riid
246 * RETURNS
247 * Success: S_OK
248 * Failure: CLASS_E_CLASSNOTAVAILABLE, E_OUTOFMEMORY, E_INVALIDARG,
249 * E_UNEXPECTED
252 HRESULT WINAPI DllGetClassObject(REFCLSID rclsid, REFIID riid, LPVOID *ppv)
254 int i;
256 TRACE("(%s,%s,%p)\n", debugstr_guid(rclsid), debugstr_guid(riid), ppv);
258 for (i=0; i < sizeof(object_creation)/sizeof(object_creation[0]); i++)
260 if (IsEqualGUID(object_creation[i].clsid, rclsid))
261 return IClassFactory_QueryInterface(object_creation[i].cf, riid, ppv);
264 FIXME("%s: no class found.\n", debugstr_guid(rclsid));
265 return CLASS_E_CLASSNOTAVAILABLE;
269 /***********************************************************************
270 * DllRegisterServerEx (URLMON.@)
272 HRESULT WINAPI DllRegisterServerEx(void)
274 FIXME("(void): stub\n");
276 return E_FAIL;
279 /**************************************************************************
280 * UrlMkSetSessionOption (URLMON.@)
282 HRESULT WINAPI UrlMkSetSessionOption(DWORD dwOption, LPVOID pBuffer, DWORD dwBufferLength,
283 DWORD Reserved)
285 FIXME("(%#x, %p, %#x): stub\n", dwOption, pBuffer, dwBufferLength);
287 return S_OK;
290 static const CHAR Agent[] = "Mozilla/4.0 (compatible; MSIE 6.0; Windows NT 5.0)";
292 /**************************************************************************
293 * ObtainUserAgentString (URLMON.@)
295 HRESULT WINAPI ObtainUserAgentString(DWORD dwOption, LPSTR pcszUAOut, DWORD *cbSize)
297 FIXME("(%d, %p, %p): stub\n", dwOption, pcszUAOut, cbSize);
299 if(dwOption) {
300 ERR("dwOption: %d, must be zero\n", dwOption);
303 if (sizeof(Agent) < *cbSize)
304 *cbSize = sizeof(Agent);
305 lstrcpynA(pcszUAOut, Agent, *cbSize);
307 return S_OK;
310 HRESULT WINAPI CoInternetCompareUrl(LPCWSTR pwzUrl1, LPCWSTR pwzUrl2, DWORD dwCompareFlags)
312 TRACE("(%s,%s,%08x)\n", debugstr_w(pwzUrl1), debugstr_w(pwzUrl2), dwCompareFlags);
313 return UrlCompareW(pwzUrl1, pwzUrl2, dwCompareFlags)==0?S_OK:S_FALSE;
316 /**************************************************************************
317 * IsValidURL (URLMON.@)
319 * Determines if a specified string is a valid URL.
321 * PARAMS
322 * pBC [I] ignored, must be NULL.
323 * szURL [I] string that represents the URL in question.
324 * dwReserved [I] reserved and must be zero.
326 * RETURNS
327 * Success: S_OK.
328 * Failure: S_FALSE.
329 * returns E_INVALIDARG if one or more of the args is invalid.
331 * TODO:
332 * test functionality against windows to see what a valid URL is.
334 HRESULT WINAPI IsValidURL(LPBC pBC, LPCWSTR szURL, DWORD dwReserved)
336 FIXME("(%p, %s, %d): stub\n", pBC, debugstr_w(szURL), dwReserved);
338 if (pBC != NULL || dwReserved != 0)
339 return E_INVALIDARG;
341 return S_OK;
344 /**************************************************************************
345 * FaultInIEFeature (URLMON.@)
347 * Undocumented. Appears to be used by native shdocvw.dll.
349 HRESULT WINAPI FaultInIEFeature( HWND hwnd, uCLSSPEC * pClassSpec,
350 QUERYCONTEXT *pQuery, DWORD flags )
352 FIXME("%p %p %p %08x\n", hwnd, pClassSpec, pQuery, flags);
353 return E_NOTIMPL;
356 /**************************************************************************
357 * CoGetClassObjectFromURL (URLMON.@)
359 HRESULT WINAPI CoGetClassObjectFromURL( REFCLSID rclsid, LPCWSTR szCodeURL, DWORD dwFileVersionMS,
360 DWORD dwFileVersionLS, LPCWSTR szContentType,
361 LPBINDCTX pBindCtx, DWORD dwClsContext, LPVOID pvReserved,
362 REFIID riid, LPVOID *ppv )
364 FIXME("(%s %s %d %d %s %p %d %p %s %p) Stub!\n", debugstr_guid(rclsid), debugstr_w(szCodeURL),
365 dwFileVersionMS, dwFileVersionLS, debugstr_w(szContentType), pBindCtx, dwClsContext, pvReserved,
366 debugstr_guid(riid), ppv);
367 return E_NOINTERFACE;
370 /***********************************************************************
371 * ReleaseBindInfo (URLMON.@)
373 * Release the resources used by the specified BINDINFO structure.
375 * PARAMS
376 * pbindinfo [I] BINDINFO to release.
378 * RETURNS
379 * Nothing.
381 void WINAPI ReleaseBindInfo(BINDINFO* pbindinfo)
383 DWORD size;
385 TRACE("(%p)\n", pbindinfo);
387 if(!pbindinfo || !(size = pbindinfo->cbSize))
388 return;
390 CoTaskMemFree(pbindinfo->szExtraInfo);
391 ReleaseStgMedium(&pbindinfo->stgmedData);
393 if(offsetof(BINDINFO, szExtraInfo) < size)
394 CoTaskMemFree(pbindinfo->szCustomVerb);
397 if(pbindinfo->pUnk && offsetof(BINDINFO, pUnk) < size)
398 IUnknown_Release(pbindinfo->pUnk);
400 memset(pbindinfo, 0, size);
401 pbindinfo->cbSize = size;
404 /***********************************************************************
405 * FindMimeFromData (URLMON.@)
407 * Determines the Multipurpose Internet Mail Extensions (MIME) type from the data provided.
409 static BOOL text_html_filter(LPVOID buf, DWORD size)
411 const char *b = buf;
412 int i;
414 if(size < 5)
415 return FALSE;
417 for(i=0; i < size-5; i++) {
418 if(b[i] == '<'
419 && (b[i+1] == 'h' || b[i+1] == 'H')
420 && (b[i+2] == 't' || b[i+2] == 'T')
421 && (b[i+3] == 'm' || b[i+3] == 'M')
422 && (b[i+4] == 'l' || b[i+4] == 'L'))
423 return TRUE;
426 return FALSE;
429 static BOOL image_gif_filter(LPVOID buf, DWORD size)
431 const BYTE const *b = buf;
433 return size >= 6
434 && (b[0] == 'G' || b[0] == 'g')
435 && (b[1] == 'I' || b[1] == 'i')
436 && (b[2] == 'F' || b[2] == 'f')
437 && b[3] == '8'
438 && (b[4] == '7' || b[4] == '9')
439 && (b[5] == 'A' || b[5] == 'a');
442 static BOOL image_pjpeg_filter(LPVOID buf, DWORD size)
444 return size > 2 && *(BYTE*)buf == 0xff && *((BYTE*)buf+1) == 0xd8;
447 static BOOL image_xpng_filter(LPVOID buf, DWORD size)
449 static const BYTE xpng_header[] = {0x89,'P','N','G',0x0d,0x0a,0x1a,0x0a};
450 return size > sizeof(xpng_header) && !memcmp(buf, xpng_header, sizeof(xpng_header));
453 static BOOL image_bmp_filter(LPVOID buf, DWORD size)
455 return size >= 14
456 && *(BYTE*)buf == 0x42
457 && *((BYTE*)buf+1) == 0x4d
458 && *(DWORD*)((BYTE*)buf+6) == 0;
461 static BOOL text_plain_filter(LPVOID buf, DWORD size)
463 UCHAR *ptr;
465 for(ptr = buf; ptr < (UCHAR*)buf+size-1; ptr++) {
466 if(*ptr < 0x20 && *ptr != '\n' && *ptr != '\r' && *ptr != '\t')
467 return FALSE;
470 return TRUE;
473 static BOOL application_octet_stream_filter(LPVOID buf, DWORD size)
475 return TRUE;
478 HRESULT WINAPI FindMimeFromData(LPBC pBC, LPCWSTR pwzUrl, LPVOID pBuffer,
479 DWORD cbSize, LPCWSTR pwzMimeProposed, DWORD dwMimeFlags,
480 LPWSTR* ppwzMimeOut, DWORD dwReserved)
482 TRACE("(%p,%s,%p,%d,%s,0x%x,%p,0x%x)\n", pBC, debugstr_w(pwzUrl), pBuffer, cbSize,
483 debugstr_w(pwzMimeProposed), dwMimeFlags, ppwzMimeOut, dwReserved);
485 if(dwMimeFlags)
486 WARN("dwMimeFlags=%08x\n", dwMimeFlags);
487 if(dwReserved)
488 WARN("dwReserved=%d\n", dwReserved);
490 /* pBC seams to not be used */
492 if(!ppwzMimeOut || (!pwzUrl && !pBuffer))
493 return E_INVALIDARG;
495 if(pwzMimeProposed && (!pBuffer || (pBuffer && !cbSize))) {
496 DWORD len;
498 if(!pwzMimeProposed)
499 return E_FAIL;
501 len = strlenW(pwzMimeProposed)+1;
502 *ppwzMimeOut = CoTaskMemAlloc(len*sizeof(WCHAR));
503 memcpy(*ppwzMimeOut, pwzMimeProposed, len*sizeof(WCHAR));
504 return S_OK;
507 if(pBuffer) {
508 DWORD len;
509 LPCWSTR ret = NULL;
510 int i;
512 static const WCHAR wszTextHtml[] = {'t','e','x','t','/','h','t','m','l',0};
513 static const WCHAR wszImageGif[] = {'i','m','a','g','e','/','g','i','f',0};
514 static const WCHAR wszImagePjpeg[] = {'i','m','a','g','e','/','p','j','p','e','g',0};
515 static const WCHAR wszImageXPng[] = {'i','m','a','g','e','/','x','-','p','n','g',0};
516 static const WCHAR wszImageBmp[] = {'i','m','a','g','e','/','b','m','p',0};
517 static const WCHAR wszTextPlain[] = {'t','e','x','t','/','p','l','a','i','n','\0'};
518 static const WCHAR wszAppOctetStream[] = {'a','p','p','l','i','c','a','t','i','o','n','/',
519 'o','c','t','e','t','-','s','t','r','e','a','m','\0'};
521 static const struct {
522 LPCWSTR mime;
523 BOOL (*filter)(LPVOID,DWORD);
524 } mime_filters[] = {
525 {wszTextHtml, text_html_filter},
526 {wszImageGif, image_gif_filter},
527 {wszImagePjpeg, image_pjpeg_filter},
528 {wszImageXPng, image_xpng_filter},
529 {wszImageBmp, image_bmp_filter},
530 {wszTextPlain, text_plain_filter},
531 {wszAppOctetStream, application_octet_stream_filter}
534 if(!cbSize)
535 return E_FAIL;
537 if(pwzMimeProposed && strcmpW(pwzMimeProposed, wszAppOctetStream)) {
538 for(i=0; i < sizeof(mime_filters)/sizeof(*mime_filters); i++) {
539 if(!strcmpW(pwzMimeProposed, mime_filters[i].mime))
540 break;
543 if(i == sizeof(mime_filters)/sizeof(*mime_filters)
544 || mime_filters[i].filter(pBuffer, cbSize)) {
545 len = strlenW(pwzMimeProposed)+1;
546 *ppwzMimeOut = CoTaskMemAlloc(len*sizeof(WCHAR));
547 memcpy(*ppwzMimeOut, pwzMimeProposed, len*sizeof(WCHAR));
548 return S_OK;
552 i=0;
553 while(!ret) {
554 if(mime_filters[i].filter(pBuffer, cbSize))
555 ret = mime_filters[i].mime;
556 i++;
559 if(pwzMimeProposed) {
560 if(i == sizeof(mime_filters)/sizeof(*mime_filters))
561 ret = pwzMimeProposed;
563 /* text/html is a special case */
564 if(!strcmpW(pwzMimeProposed, wszTextHtml) && !strcmpW(ret, wszTextPlain))
565 ret = wszTextHtml;
568 len = strlenW(ret)+1;
569 *ppwzMimeOut = CoTaskMemAlloc(len*sizeof(WCHAR));
570 memcpy(*ppwzMimeOut, ret, len*sizeof(WCHAR));
571 return S_OK;
574 if(pwzUrl) {
575 HKEY hkey;
576 DWORD res, size;
577 LPCWSTR ptr;
578 WCHAR mime[64];
580 static const WCHAR wszContentType[] =
581 {'C','o','n','t','e','n','t',' ','T','y','p','e','\0'};
583 ptr = strrchrW(pwzUrl, '.');
584 if(!ptr)
585 return E_FAIL;
587 res = RegOpenKeyW(HKEY_CLASSES_ROOT, ptr, &hkey);
588 if(res != ERROR_SUCCESS)
589 return E_FAIL;
591 size = sizeof(mime);
592 res = RegQueryValueExW(hkey, wszContentType, NULL, NULL, (LPBYTE)mime, &size);
593 RegCloseKey(hkey);
594 if(res != ERROR_SUCCESS)
595 return E_FAIL;
597 *ppwzMimeOut = CoTaskMemAlloc(size);
598 memcpy(*ppwzMimeOut, mime, size);
599 return S_OK;
602 return E_FAIL;
605 /***********************************************************************
606 * Extract (URLMON.@)
608 HRESULT WINAPI Extract(void *dest, LPCSTR szCabName)
610 HRESULT (WINAPI *pExtract)(void *, LPCSTR);
612 if (!hCabinet)
613 hCabinet = LoadLibraryA("cabinet.dll");
615 if (!hCabinet) return HRESULT_FROM_WIN32(GetLastError());
616 pExtract = (void *)GetProcAddress(hCabinet, "Extract");
617 if (!pExtract) return HRESULT_FROM_WIN32(GetLastError());
619 return pExtract(dest, szCabName);