|  | /* | 
|  | * Internet Security and Zone Manager | 
|  | * | 
|  | * Copyright (c) 2004 Huw D M Davies | 
|  | * Copyright 2004 Jacek Caban | 
|  | * | 
|  | * This library is free software; you can redistribute it and/or | 
|  | * modify it under the terms of the GNU Lesser General Public | 
|  | * License as published by the Free Software Foundation; either | 
|  | * version 2.1 of the License, or (at your option) any later version. | 
|  | * | 
|  | * This library is distributed in the hope that it will be useful, | 
|  | * but WITHOUT ANY WARRANTY; without even the implied warranty of | 
|  | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU | 
|  | * Lesser General Public License for more details. | 
|  | * | 
|  | * You should have received a copy of the GNU Lesser General Public | 
|  | * License along with this library; if not, write to the Free Software | 
|  | * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA | 
|  | */ | 
|  |  | 
|  | #include <stdarg.h> | 
|  | #include <stdio.h> | 
|  |  | 
|  | #define COBJMACROS | 
|  |  | 
|  | #include "windef.h" | 
|  | #include "winbase.h" | 
|  | #include "winuser.h" | 
|  | #include "winreg.h" | 
|  | #include "wine/debug.h" | 
|  | #include "ole2.h" | 
|  | #include "wine/unicode.h" | 
|  | #include "urlmon.h" | 
|  | #include "urlmon_main.h" | 
|  |  | 
|  | WINE_DEFAULT_DEBUG_CHANNEL(urlmon); | 
|  |  | 
|  | /*********************************************************************** | 
|  | *           InternetSecurityManager implementation | 
|  | * | 
|  | */ | 
|  | typedef struct { | 
|  | const IInternetSecurityManagerVtbl* lpInternetSecurityManagerVtbl; | 
|  |  | 
|  | LONG ref; | 
|  |  | 
|  | IInternetSecurityMgrSite *mgrsite; | 
|  | IInternetSecurityManager *custom_manager; | 
|  | } SecManagerImpl; | 
|  |  | 
|  | #define SECMGR_THIS(iface) DEFINE_THIS(SecManagerImpl, InternetSecurityManager, iface) | 
|  |  | 
|  | static HRESULT map_url_to_zone(LPCWSTR url, DWORD *zone) | 
|  | { | 
|  | WCHAR schema[64]; | 
|  | DWORD res, size=0; | 
|  | HKEY hkey; | 
|  | HRESULT hres; | 
|  |  | 
|  | static const WCHAR wszZoneMapProtocolKey[] = | 
|  | {'S','o','f','t','w','a','r','e','\\', | 
|  | 'M','i','c','r','o','s','o','f','t','\\', | 
|  | 'W','i','n','d','o','w','s','\\', | 
|  | 'C','u','r','r','e','n','t','V','e','r','s','i','o','n','\\', | 
|  | 'I','n','t','e','r','n','e','t',' ','S','e','t','t','i','n','g','s','\\', | 
|  | 'Z','o','n','e','M','a','p','\\', | 
|  | 'P','r','o','t','o','c','o','l','D','e','f','a','u','l','t','s',0}; | 
|  | static const WCHAR wszFile[] = {'f','i','l','e',0}; | 
|  |  | 
|  | hres = CoInternetParseUrl(url, PARSE_SCHEMA, 0, schema, sizeof(schema)/sizeof(WCHAR), &size, 0); | 
|  | if(FAILED(hres)) | 
|  | return hres; | 
|  | if(!*schema) | 
|  | return 0x80041001; | 
|  |  | 
|  | /* file protocol is a special case */ | 
|  | if(!strcmpW(schema, wszFile)) { | 
|  | WCHAR path[MAX_PATH]; | 
|  |  | 
|  | hres = CoInternetParseUrl(url, PARSE_PATH_FROM_URL, 0, path, | 
|  | sizeof(path)/sizeof(WCHAR), &size, 0); | 
|  |  | 
|  | if(SUCCEEDED(hres) && strchrW(path, '\\')) { | 
|  | *zone = 0; | 
|  | return S_OK; | 
|  | } | 
|  | } | 
|  |  | 
|  | WARN("domains are not yet implemented\n"); | 
|  |  | 
|  | res = RegOpenKeyW(HKEY_CURRENT_USER, wszZoneMapProtocolKey, &hkey); | 
|  | if(res != ERROR_SUCCESS) { | 
|  | ERR("Could not open key %s\n", debugstr_w(wszZoneMapProtocolKey)); | 
|  | return E_UNEXPECTED; | 
|  | } | 
|  |  | 
|  | size = sizeof(DWORD); | 
|  | res = RegQueryValueExW(hkey, schema, NULL, NULL, (PBYTE)zone, &size); | 
|  | if(res == ERROR_SUCCESS) | 
|  | return S_OK; | 
|  |  | 
|  | res = RegOpenKeyW(HKEY_LOCAL_MACHINE, wszZoneMapProtocolKey, &hkey); | 
|  | if(res != ERROR_SUCCESS) { | 
|  | ERR("Could not open key %s\n", debugstr_w(wszZoneMapProtocolKey)); | 
|  | return E_UNEXPECTED; | 
|  | } | 
|  |  | 
|  | size = sizeof(DWORD); | 
|  | res = RegQueryValueExW(hkey, schema, NULL, NULL, (PBYTE)zone, &size); | 
|  | if(res == ERROR_SUCCESS) | 
|  | return S_OK; | 
|  |  | 
|  | *zone = 3; | 
|  | return S_OK; | 
|  | } | 
|  |  | 
|  | static HRESULT WINAPI SecManagerImpl_QueryInterface(IInternetSecurityManager* iface,REFIID riid,void** ppvObject) | 
|  | { | 
|  | SecManagerImpl *This = SECMGR_THIS(iface); | 
|  |  | 
|  | TRACE("(%p)->(%s,%p)\n",This,debugstr_guid(riid),ppvObject); | 
|  |  | 
|  | /* Perform a sanity check on the parameters.*/ | 
|  | if ( (This==0) || (ppvObject==0) ) | 
|  | return E_INVALIDARG; | 
|  |  | 
|  | /* Initialize the return parameter */ | 
|  | *ppvObject = 0; | 
|  |  | 
|  | /* Compare the riid with the interface IDs implemented by this object.*/ | 
|  | if (IsEqualIID(&IID_IUnknown, riid) || | 
|  | IsEqualIID(&IID_IInternetSecurityManager, riid)) | 
|  | *ppvObject = iface; | 
|  |  | 
|  | /* Check that we obtained an interface.*/ | 
|  | if (!*ppvObject) { | 
|  | WARN("not supported interface %s\n", debugstr_guid(riid)); | 
|  | return E_NOINTERFACE; | 
|  | } | 
|  |  | 
|  | /* Query Interface always increases the reference count by one when it is successful */ | 
|  | IInternetSecurityManager_AddRef(iface); | 
|  |  | 
|  | return S_OK; | 
|  | } | 
|  |  | 
|  | static ULONG WINAPI SecManagerImpl_AddRef(IInternetSecurityManager* iface) | 
|  | { | 
|  | SecManagerImpl *This = SECMGR_THIS(iface); | 
|  | ULONG refCount = InterlockedIncrement(&This->ref); | 
|  |  | 
|  | TRACE("(%p) ref=%lu\n", This, refCount); | 
|  |  | 
|  | return refCount; | 
|  | } | 
|  |  | 
|  | static ULONG WINAPI SecManagerImpl_Release(IInternetSecurityManager* iface) | 
|  | { | 
|  | SecManagerImpl *This = SECMGR_THIS(iface); | 
|  | ULONG refCount = InterlockedDecrement(&This->ref); | 
|  |  | 
|  | TRACE("(%p) ref=%lu\n", This, refCount); | 
|  |  | 
|  | /* destroy the object if there's no more reference on it */ | 
|  | if (!refCount){ | 
|  | if(This->mgrsite) | 
|  | IInternetSecurityMgrSite_Release(This->mgrsite); | 
|  | if(This->custom_manager) | 
|  | IInternetSecurityManager_Release(This->custom_manager); | 
|  |  | 
|  | HeapFree(GetProcessHeap(),0,This); | 
|  |  | 
|  | URLMON_UnlockModule(); | 
|  | } | 
|  |  | 
|  | return refCount; | 
|  | } | 
|  |  | 
|  | static HRESULT WINAPI SecManagerImpl_SetSecuritySite(IInternetSecurityManager *iface, | 
|  | IInternetSecurityMgrSite *pSite) | 
|  | { | 
|  | SecManagerImpl *This = SECMGR_THIS(iface); | 
|  |  | 
|  | TRACE("(%p)->(%p)\n", This, pSite); | 
|  |  | 
|  | if(This->mgrsite) | 
|  | IInternetSecurityMgrSite_Release(This->mgrsite); | 
|  |  | 
|  | if(This->custom_manager) { | 
|  | IInternetSecurityManager_Release(This->custom_manager); | 
|  | This->custom_manager = NULL; | 
|  | } | 
|  |  | 
|  | This->mgrsite = pSite; | 
|  |  | 
|  | if(pSite) { | 
|  | IServiceProvider *servprov; | 
|  | HRESULT hres; | 
|  |  | 
|  | IInternetSecurityMgrSite_AddRef(pSite); | 
|  |  | 
|  | hres = IInternetSecurityMgrSite_QueryInterface(pSite, &IID_IServiceProvider, | 
|  | (void**)&servprov); | 
|  | if(SUCCEEDED(hres)) { | 
|  | IServiceProvider_QueryService(servprov, &SID_SInternetSecurityManager, | 
|  | &IID_IInternetSecurityManager, (void**)&This->custom_manager); | 
|  | IServiceProvider_Release(servprov); | 
|  | } | 
|  | } | 
|  |  | 
|  | return S_OK; | 
|  | } | 
|  |  | 
|  | static HRESULT WINAPI SecManagerImpl_GetSecuritySite(IInternetSecurityManager *iface, | 
|  | IInternetSecurityMgrSite **ppSite) | 
|  | { | 
|  | SecManagerImpl *This = SECMGR_THIS(iface); | 
|  |  | 
|  | TRACE("(%p)->(%p)\n", This, ppSite); | 
|  |  | 
|  | if(!ppSite) | 
|  | return E_INVALIDARG; | 
|  |  | 
|  | if(This->mgrsite) | 
|  | IInternetSecurityMgrSite_AddRef(This->mgrsite); | 
|  |  | 
|  | *ppSite = This->mgrsite; | 
|  | return S_OK; | 
|  | } | 
|  |  | 
|  | static HRESULT WINAPI SecManagerImpl_MapUrlToZone(IInternetSecurityManager *iface, | 
|  | LPCWSTR pwszUrl, DWORD *pdwZone, | 
|  | DWORD dwFlags) | 
|  | { | 
|  | SecManagerImpl *This = SECMGR_THIS(iface); | 
|  | LPWSTR url; | 
|  | DWORD size; | 
|  | HRESULT hres; | 
|  |  | 
|  | TRACE("(%p)->(%s %p %08lx)\n", iface, debugstr_w(pwszUrl), pdwZone, dwFlags); | 
|  |  | 
|  | if(This->custom_manager) { | 
|  | hres = IInternetSecurityManager_MapUrlToZone(This->custom_manager, | 
|  | pwszUrl, pdwZone, dwFlags); | 
|  | if(hres != INET_E_DEFAULT_ACTION) | 
|  | return hres; | 
|  | } | 
|  |  | 
|  | if(!pwszUrl) | 
|  | return E_INVALIDARG; | 
|  |  | 
|  | if(dwFlags) | 
|  | FIXME("not supported flags: %08lx\n", dwFlags); | 
|  |  | 
|  | size = (strlenW(pwszUrl)+16) * sizeof(WCHAR); | 
|  | url = HeapAlloc(GetProcessHeap(), 0, size); | 
|  |  | 
|  | hres = CoInternetParseUrl(pwszUrl, PARSE_SECURITY_URL, 0, url, size/sizeof(WCHAR), &size, 0); | 
|  | if(FAILED(hres)) | 
|  | memcpy(url, pwszUrl, size); | 
|  |  | 
|  | hres = map_url_to_zone(url, pdwZone); | 
|  |  | 
|  | HeapFree(GetProcessHeap(), 0, url); | 
|  |  | 
|  | return hres; | 
|  | } | 
|  |  | 
|  | static HRESULT WINAPI SecManagerImpl_GetSecurityId(IInternetSecurityManager *iface, | 
|  | LPCWSTR pwszUrl, BYTE *pbSecurityId, DWORD *pcbSecurityId, DWORD_PTR dwReserved) | 
|  | { | 
|  | SecManagerImpl *This = SECMGR_THIS(iface); | 
|  | LPWSTR buf, ptr, ptr2; | 
|  | DWORD size, zone, len; | 
|  | HRESULT hres; | 
|  |  | 
|  | static const WCHAR wszFile[] = {'f','i','l','e',':'}; | 
|  |  | 
|  | TRACE("(%p)->(%s %p %p %08lx)\n", iface, debugstr_w(pwszUrl), pbSecurityId, | 
|  | pcbSecurityId, dwReserved); | 
|  |  | 
|  | if(This->custom_manager) { | 
|  | hres = IInternetSecurityManager_GetSecurityId(This->custom_manager, | 
|  | pwszUrl, pbSecurityId, pcbSecurityId, dwReserved); | 
|  | if(hres != INET_E_DEFAULT_ACTION) | 
|  | return hres; | 
|  | } | 
|  |  | 
|  | if(!pwszUrl || !pbSecurityId || !pcbSecurityId) | 
|  | return E_INVALIDARG; | 
|  |  | 
|  | if(dwReserved) | 
|  | FIXME("dwReserved is not supported\n"); | 
|  |  | 
|  | len = strlenW(pwszUrl)+1; | 
|  | buf = HeapAlloc(GetProcessHeap(), 0, (len+16)*sizeof(WCHAR)); | 
|  |  | 
|  | hres = CoInternetParseUrl(pwszUrl, PARSE_SECURITY_URL, 0, buf, len, &size, 0); | 
|  | if(FAILED(hres)) | 
|  | memcpy(buf, pwszUrl, len*sizeof(WCHAR)); | 
|  |  | 
|  | hres = map_url_to_zone(buf, &zone); | 
|  | if(FAILED(hres)) { | 
|  | HeapFree(GetProcessHeap(), 0, buf); | 
|  | return hres == 0x80041001 ? E_INVALIDARG : hres; | 
|  | } | 
|  |  | 
|  | /* file protocol is a special case */ | 
|  | if(strlenW(pwszUrl) >= sizeof(wszFile)/sizeof(WCHAR) | 
|  | && !memcmp(buf, wszFile, sizeof(wszFile))) { | 
|  |  | 
|  | static const BYTE secidFile[] = {'f','i','l','e',':'}; | 
|  |  | 
|  | if(*pcbSecurityId < sizeof(secidFile)+sizeof(zone)) | 
|  | return HRESULT_FROM_WIN32(ERROR_INSUFFICIENT_BUFFER); | 
|  |  | 
|  | memcpy(pbSecurityId, secidFile, sizeof(secidFile)); | 
|  | *(DWORD*)(pbSecurityId+sizeof(secidFile)) = zone; | 
|  |  | 
|  | *pcbSecurityId = sizeof(secidFile)+sizeof(zone); | 
|  | return S_OK; | 
|  | } | 
|  |  | 
|  | ptr = strchrW(buf, ':'); | 
|  | ptr2 = ++ptr; | 
|  | while(*ptr2 == '/') | 
|  | ptr2++; | 
|  | if(ptr2 != ptr) | 
|  | memmove(ptr, ptr2, (strlenW(ptr2)+1)*sizeof(WCHAR)); | 
|  |  | 
|  | ptr = strchrW(ptr, '/'); | 
|  | if(ptr) | 
|  | *ptr = 0; | 
|  |  | 
|  | len = WideCharToMultiByte(CP_ACP, 0, buf, -1, NULL, 0, NULL, NULL)-1; | 
|  |  | 
|  | if(len+sizeof(DWORD) > *pcbSecurityId) | 
|  | return HRESULT_FROM_WIN32(ERROR_INSUFFICIENT_BUFFER); | 
|  |  | 
|  | WideCharToMultiByte(CP_ACP, 0, buf, -1, (LPSTR)pbSecurityId, -1, NULL, NULL); | 
|  | HeapFree(GetProcessHeap(), 0, buf); | 
|  |  | 
|  | *(DWORD*)(pbSecurityId+len) = zone; | 
|  |  | 
|  | *pcbSecurityId = len+sizeof(DWORD); | 
|  |  | 
|  | return S_OK; | 
|  | } | 
|  |  | 
|  |  | 
|  | static HRESULT WINAPI SecManagerImpl_ProcessUrlAction(IInternetSecurityManager *iface, | 
|  | LPCWSTR pwszUrl, DWORD dwAction, | 
|  | BYTE *pPolicy, DWORD cbPolicy, | 
|  | BYTE *pContext, DWORD cbContext, | 
|  | DWORD dwFlags, DWORD dwReserved) | 
|  | { | 
|  | SecManagerImpl *This = SECMGR_THIS(iface); | 
|  | HRESULT hres; | 
|  |  | 
|  | TRACE("(%p)->(%s %08lx %p %08lx %p %08lx %08lx %08lx)\n", iface, debugstr_w(pwszUrl), dwAction, | 
|  | pPolicy, cbPolicy, pContext, cbContext, dwFlags, dwReserved); | 
|  |  | 
|  | if(This->custom_manager) { | 
|  | hres = IInternetSecurityManager_ProcessUrlAction(This->custom_manager, pwszUrl, dwAction, | 
|  | pPolicy, cbPolicy, pContext, cbContext, dwFlags, dwReserved); | 
|  | if(hres != INET_E_DEFAULT_ACTION) | 
|  | return hres; | 
|  | } | 
|  |  | 
|  | FIXME("Default action is not implemented\n"); | 
|  | return E_NOTIMPL; | 
|  | } | 
|  |  | 
|  |  | 
|  | static HRESULT WINAPI SecManagerImpl_QueryCustomPolicy(IInternetSecurityManager *iface, | 
|  | LPCWSTR pwszUrl, REFGUID guidKey, | 
|  | BYTE **ppPolicy, DWORD *pcbPolicy, | 
|  | BYTE *pContext, DWORD cbContext, | 
|  | DWORD dwReserved) | 
|  | { | 
|  | SecManagerImpl *This = SECMGR_THIS(iface); | 
|  | HRESULT hres; | 
|  |  | 
|  | TRACE("(%p)->(%s %s %p %p %p %08lx %08lx )\n", iface, debugstr_w(pwszUrl), debugstr_guid(guidKey), | 
|  | ppPolicy, pcbPolicy, pContext, cbContext, dwReserved); | 
|  |  | 
|  | if(This->custom_manager) { | 
|  | hres = IInternetSecurityManager_QueryCustomPolicy(This->custom_manager, pwszUrl, guidKey, | 
|  | ppPolicy, pcbPolicy, pContext, cbContext, dwReserved); | 
|  | if(hres != INET_E_DEFAULT_ACTION) | 
|  | return hres; | 
|  | } | 
|  |  | 
|  | FIXME("Default action is not implemented\n"); | 
|  | return E_NOTIMPL; | 
|  | } | 
|  |  | 
|  | static HRESULT WINAPI SecManagerImpl_SetZoneMapping(IInternetSecurityManager *iface, | 
|  | DWORD dwZone, LPCWSTR pwszPattern, DWORD dwFlags) | 
|  | { | 
|  | SecManagerImpl *This = SECMGR_THIS(iface); | 
|  | HRESULT hres; | 
|  |  | 
|  | TRACE("(%p)->(%08lx %s %08lx)\n", iface, dwZone, debugstr_w(pwszPattern),dwFlags); | 
|  |  | 
|  | if(This->custom_manager) { | 
|  | hres = IInternetSecurityManager_SetZoneMapping(This->custom_manager, dwZone, | 
|  | pwszPattern, dwFlags); | 
|  | if(hres != INET_E_DEFAULT_ACTION) | 
|  | return hres; | 
|  | } | 
|  |  | 
|  | FIXME("Default action is not implemented\n"); | 
|  | return E_NOTIMPL; | 
|  | } | 
|  |  | 
|  | static HRESULT WINAPI SecManagerImpl_GetZoneMappings(IInternetSecurityManager *iface, | 
|  | DWORD dwZone, IEnumString **ppenumString, DWORD dwFlags) | 
|  | { | 
|  | SecManagerImpl *This = SECMGR_THIS(iface); | 
|  | HRESULT hres; | 
|  |  | 
|  | TRACE("(%p)->(%08lx %p %08lx)\n", iface, dwZone, ppenumString,dwFlags); | 
|  |  | 
|  | if(This->custom_manager) { | 
|  | hres = IInternetSecurityManager_GetZoneMappings(This->custom_manager, dwZone, | 
|  | ppenumString, dwFlags); | 
|  | if(hres != INET_E_DEFAULT_ACTION) | 
|  | return hres; | 
|  | } | 
|  |  | 
|  | FIXME("Default action is not implemented\n"); | 
|  | return E_NOTIMPL; | 
|  | } | 
|  |  | 
|  | static const IInternetSecurityManagerVtbl VT_SecManagerImpl = | 
|  | { | 
|  | SecManagerImpl_QueryInterface, | 
|  | SecManagerImpl_AddRef, | 
|  | SecManagerImpl_Release, | 
|  | SecManagerImpl_SetSecuritySite, | 
|  | SecManagerImpl_GetSecuritySite, | 
|  | SecManagerImpl_MapUrlToZone, | 
|  | SecManagerImpl_GetSecurityId, | 
|  | SecManagerImpl_ProcessUrlAction, | 
|  | SecManagerImpl_QueryCustomPolicy, | 
|  | SecManagerImpl_SetZoneMapping, | 
|  | SecManagerImpl_GetZoneMappings | 
|  | }; | 
|  |  | 
|  | HRESULT SecManagerImpl_Construct(IUnknown *pUnkOuter, LPVOID *ppobj) | 
|  | { | 
|  | SecManagerImpl *This; | 
|  |  | 
|  | TRACE("(%p,%p)\n",pUnkOuter,ppobj); | 
|  | This = HeapAlloc(GetProcessHeap(), 0, sizeof(*This)); | 
|  |  | 
|  | /* Initialize the virtual function table. */ | 
|  | This->lpInternetSecurityManagerVtbl = &VT_SecManagerImpl; | 
|  |  | 
|  | This->ref = 1; | 
|  | This->mgrsite = NULL; | 
|  | This->custom_manager = NULL; | 
|  |  | 
|  | *ppobj = This; | 
|  |  | 
|  | URLMON_LockModule(); | 
|  |  | 
|  | return S_OK; | 
|  | } | 
|  |  | 
|  | /*********************************************************************** | 
|  | *           InternetZoneManager implementation | 
|  | * | 
|  | */ | 
|  | typedef struct { | 
|  | const IInternetZoneManagerVtbl* lpVtbl; | 
|  | LONG ref; | 
|  | } ZoneMgrImpl; | 
|  |  | 
|  | static HRESULT open_zone_key(DWORD zone, HKEY *hkey, URLZONEREG zone_reg) | 
|  | { | 
|  | static const WCHAR wszZonesKey[] = | 
|  | {'S','o','f','t','w','a','r','e','\\', | 
|  | 'M','i','c','r','o','s','o','f','t','\\', | 
|  | 'W','i','n','d','o','w','s','\\', | 
|  | 'C','u','r','r','e','n','t','V','e','r','s','i','o','n','\\', | 
|  | 'I','n','t','e','r','n','e','t',' ','S','e','t','t','i','n','g','s','\\', | 
|  | 'Z','o','n','e','s','\\',0}; | 
|  | static const WCHAR wszFormat[] = {'%','s','%','l','d',0}; | 
|  |  | 
|  | WCHAR key_name[sizeof(wszZonesKey)/sizeof(WCHAR)+8]; | 
|  | HKEY parent_key; | 
|  | DWORD res; | 
|  |  | 
|  | switch(zone_reg) { | 
|  | case URLZONEREG_DEFAULT: /* FIXME: TEST */ | 
|  | case URLZONEREG_HKCU: | 
|  | parent_key = HKEY_CURRENT_USER; | 
|  | break; | 
|  | case URLZONEREG_HKLM: | 
|  | parent_key = HKEY_LOCAL_MACHINE; | 
|  | break; | 
|  | default: | 
|  | WARN("Unknown URLZONEREG: %d\n", zone_reg); | 
|  | return E_FAIL; | 
|  | }; | 
|  |  | 
|  | wsprintfW(key_name, wszFormat, wszZonesKey, zone); | 
|  |  | 
|  | res = RegOpenKeyW(parent_key, key_name, hkey); | 
|  |  | 
|  | if(res != ERROR_SUCCESS) { | 
|  | WARN("RegOpenKey failed\n"); | 
|  | return E_INVALIDARG; | 
|  | } | 
|  |  | 
|  | return S_OK; | 
|  | } | 
|  |  | 
|  | /******************************************************************** | 
|  | *      IInternetZoneManager_QueryInterface | 
|  | */ | 
|  | static HRESULT WINAPI ZoneMgrImpl_QueryInterface(IInternetZoneManager* iface, REFIID riid, void** ppvObject) | 
|  | { | 
|  | ZoneMgrImpl* This = (ZoneMgrImpl*)iface; | 
|  |  | 
|  | TRACE("(%p)->(%s,%p)\n", This, debugstr_guid(riid), ppvObject); | 
|  |  | 
|  | if(!This || !ppvObject) | 
|  | return E_INVALIDARG; | 
|  |  | 
|  | if(!IsEqualIID(&IID_IUnknown, riid) && !IsEqualIID(&IID_IInternetZoneManager, riid)) { | 
|  | FIXME("Unknown interface: %s\n", debugstr_guid(riid)); | 
|  | *ppvObject = NULL; | 
|  | return E_NOINTERFACE; | 
|  | } | 
|  |  | 
|  | *ppvObject = iface; | 
|  | IInternetZoneManager_AddRef(iface); | 
|  |  | 
|  | return S_OK; | 
|  | } | 
|  |  | 
|  | /******************************************************************** | 
|  | *      IInternetZoneManager_AddRef | 
|  | */ | 
|  | static ULONG WINAPI ZoneMgrImpl_AddRef(IInternetZoneManager* iface) | 
|  | { | 
|  | ZoneMgrImpl* This = (ZoneMgrImpl*)iface; | 
|  | ULONG refCount = InterlockedIncrement(&This->ref); | 
|  |  | 
|  | TRACE("(%p)->(ref before=%lu)\n",This, refCount - 1); | 
|  |  | 
|  | return refCount; | 
|  | } | 
|  |  | 
|  | /******************************************************************** | 
|  | *      IInternetZoneManager_Release | 
|  | */ | 
|  | static ULONG WINAPI ZoneMgrImpl_Release(IInternetZoneManager* iface) | 
|  | { | 
|  | ZoneMgrImpl* This = (ZoneMgrImpl*)iface; | 
|  | ULONG refCount = InterlockedDecrement(&This->ref); | 
|  |  | 
|  | TRACE("(%p)->(ref before=%lu)\n",This, refCount + 1); | 
|  |  | 
|  | if(!refCount) { | 
|  | HeapFree(GetProcessHeap(), 0, This); | 
|  | URLMON_UnlockModule(); | 
|  | } | 
|  |  | 
|  | return refCount; | 
|  | } | 
|  |  | 
|  | /******************************************************************** | 
|  | *      IInternetZoneManager_GetZoneAttributes | 
|  | */ | 
|  | static HRESULT WINAPI ZoneMgrImpl_GetZoneAttributes(IInternetZoneManager* iface, | 
|  | DWORD dwZone, | 
|  | ZONEATTRIBUTES* pZoneAttributes) | 
|  | { | 
|  | FIXME("(%p)->(%ld %p) stub\n", iface, dwZone, pZoneAttributes); | 
|  | return E_NOTIMPL; | 
|  | } | 
|  |  | 
|  | /******************************************************************** | 
|  | *      IInternetZoneManager_SetZoneAttributes | 
|  | */ | 
|  | static HRESULT WINAPI ZoneMgrImpl_SetZoneAttributes(IInternetZoneManager* iface, | 
|  | DWORD dwZone, | 
|  | ZONEATTRIBUTES* pZoneAttributes) | 
|  | { | 
|  | FIXME("(%p)->(%08lx %p) stub\n", iface, dwZone, pZoneAttributes); | 
|  | return E_NOTIMPL; | 
|  | } | 
|  |  | 
|  | /******************************************************************** | 
|  | *      IInternetZoneManager_GetZoneCustomPolicy | 
|  | */ | 
|  | static HRESULT WINAPI ZoneMgrImpl_GetZoneCustomPolicy(IInternetZoneManager* iface, | 
|  | DWORD dwZone, | 
|  | REFGUID guidKey, | 
|  | BYTE** ppPolicy, | 
|  | DWORD* pcbPolicy, | 
|  | URLZONEREG ulrZoneReg) | 
|  | { | 
|  | FIXME("(%p)->(%08lx %s %p %p %08x) stub\n", iface, dwZone, debugstr_guid(guidKey), | 
|  | ppPolicy, pcbPolicy, ulrZoneReg); | 
|  | return E_NOTIMPL; | 
|  | } | 
|  |  | 
|  | /******************************************************************** | 
|  | *      IInternetZoneManager_SetZoneCustomPolicy | 
|  | */ | 
|  | static HRESULT WINAPI ZoneMgrImpl_SetZoneCustomPolicy(IInternetZoneManager* iface, | 
|  | DWORD dwZone, | 
|  | REFGUID guidKey, | 
|  | BYTE* ppPolicy, | 
|  | DWORD cbPolicy, | 
|  | URLZONEREG ulrZoneReg) | 
|  | { | 
|  | FIXME("(%p)->(%08lx %s %p %08lx %08x) stub\n", iface, dwZone, debugstr_guid(guidKey), | 
|  | ppPolicy, cbPolicy, ulrZoneReg); | 
|  | return E_NOTIMPL; | 
|  | } | 
|  |  | 
|  | /******************************************************************** | 
|  | *      IInternetZoneManager_GetZoneActionPolicy | 
|  | */ | 
|  | static HRESULT WINAPI ZoneMgrImpl_GetZoneActionPolicy(IInternetZoneManager* iface, | 
|  | DWORD dwZone, DWORD dwAction, BYTE* pPolicy, DWORD cbPolicy, URLZONEREG urlZoneReg) | 
|  | { | 
|  | WCHAR action[16]; | 
|  | HKEY hkey; | 
|  | LONG res; | 
|  | DWORD size = cbPolicy; | 
|  | HRESULT hres; | 
|  |  | 
|  | static const WCHAR wszFormat[] = {'%','l','X',0}; | 
|  |  | 
|  | TRACE("(%p)->(%ld %08lx %p %ld %d)\n", iface, dwZone, dwAction, pPolicy, | 
|  | cbPolicy, urlZoneReg); | 
|  |  | 
|  | if(!pPolicy) | 
|  | return E_INVALIDARG; | 
|  |  | 
|  | hres = open_zone_key(dwZone, &hkey, urlZoneReg); | 
|  | if(FAILED(hres)) | 
|  | return hres; | 
|  |  | 
|  | wsprintfW(action, wszFormat, dwAction); | 
|  |  | 
|  | res = RegQueryValueExW(hkey, action, NULL, NULL, pPolicy, &size); | 
|  | if(res == ERROR_MORE_DATA) { | 
|  | hres = E_INVALIDARG; | 
|  | }else if(res == ERROR_FILE_NOT_FOUND) { | 
|  | hres = E_FAIL; | 
|  | }else if(res != ERROR_SUCCESS) { | 
|  | ERR("RegQueryValue failed: %ld\n", res); | 
|  | hres = E_UNEXPECTED; | 
|  | } | 
|  |  | 
|  | RegCloseKey(hkey); | 
|  |  | 
|  | return hres; | 
|  | } | 
|  |  | 
|  | /******************************************************************** | 
|  | *      IInternetZoneManager_SetZoneActionPolicy | 
|  | */ | 
|  | static HRESULT WINAPI ZoneMgrImpl_SetZoneActionPolicy(IInternetZoneManager* iface, | 
|  | DWORD dwZone, | 
|  | DWORD dwAction, | 
|  | BYTE* pPolicy, | 
|  | DWORD cbPolicy, | 
|  | URLZONEREG urlZoneReg) | 
|  | { | 
|  | FIXME("(%p)->(%08lx %08lx %p %08lx %08x) stub\n", iface, dwZone, dwAction, pPolicy, | 
|  | cbPolicy, urlZoneReg); | 
|  | return E_NOTIMPL; | 
|  | } | 
|  |  | 
|  | /******************************************************************** | 
|  | *      IInternetZoneManager_PromptAction | 
|  | */ | 
|  | static HRESULT WINAPI ZoneMgrImpl_PromptAction(IInternetZoneManager* iface, | 
|  | DWORD dwAction, | 
|  | HWND hwndParent, | 
|  | LPCWSTR pwszUrl, | 
|  | LPCWSTR pwszText, | 
|  | DWORD dwPromptFlags) | 
|  | { | 
|  | FIXME("%p %08lx %p %s %s %08lx\n", iface, dwAction, hwndParent, | 
|  | debugstr_w(pwszUrl), debugstr_w(pwszText), dwPromptFlags ); | 
|  | return E_NOTIMPL; | 
|  | } | 
|  |  | 
|  | /******************************************************************** | 
|  | *      IInternetZoneManager_LogAction | 
|  | */ | 
|  | static HRESULT WINAPI ZoneMgrImpl_LogAction(IInternetZoneManager* iface, | 
|  | DWORD dwAction, | 
|  | LPCWSTR pwszUrl, | 
|  | LPCWSTR pwszText, | 
|  | DWORD dwLogFlags) | 
|  | { | 
|  | FIXME("(%p)->(%08lx %s %s %08lx) stub\n", iface, dwAction, debugstr_w(pwszUrl), | 
|  | debugstr_w(pwszText), dwLogFlags); | 
|  | return E_NOTIMPL; | 
|  | } | 
|  |  | 
|  | /******************************************************************** | 
|  | *      IInternetZoneManager_CreateZoneEnumerator | 
|  | */ | 
|  | static HRESULT WINAPI ZoneMgrImpl_CreateZoneEnumerator(IInternetZoneManager* iface, | 
|  | DWORD* pdwEnum, | 
|  | DWORD* pdwCount, | 
|  | DWORD dwFlags) | 
|  | { | 
|  | FIXME("(%p)->(%p %p %08lx) stub\n", iface, pdwEnum, pdwCount, dwFlags); | 
|  | return E_NOTIMPL; | 
|  | } | 
|  |  | 
|  | /******************************************************************** | 
|  | *      IInternetZoneManager_GetZoneAt | 
|  | */ | 
|  | static HRESULT WINAPI ZoneMgrImpl_GetZoneAt(IInternetZoneManager* iface, | 
|  | DWORD dwEnum, | 
|  | DWORD dwIndex, | 
|  | DWORD* pdwZone) | 
|  | { | 
|  | FIXME("(%p)->(%08lx %08lx %p) stub\n", iface, dwEnum, dwIndex, pdwZone); | 
|  | return E_NOTIMPL; | 
|  | } | 
|  |  | 
|  | /******************************************************************** | 
|  | *      IInternetZoneManager_DestroyZoneEnumerator | 
|  | */ | 
|  | static HRESULT WINAPI ZoneMgrImpl_DestroyZoneEnumerator(IInternetZoneManager* iface, | 
|  | DWORD dwEnum) | 
|  | { | 
|  | FIXME("(%p)->(%08lx) stub\n", iface, dwEnum); | 
|  | return E_NOTIMPL; | 
|  | } | 
|  |  | 
|  | /******************************************************************** | 
|  | *      IInternetZoneManager_CopyTemplatePoliciesToZone | 
|  | */ | 
|  | static HRESULT WINAPI ZoneMgrImpl_CopyTemplatePoliciesToZone(IInternetZoneManager* iface, | 
|  | DWORD dwTemplate, | 
|  | DWORD dwZone, | 
|  | DWORD dwReserved) | 
|  | { | 
|  | FIXME("(%p)->(%08lx %08lx %08lx) stub\n", iface, dwTemplate, dwZone, dwReserved); | 
|  | return E_NOTIMPL; | 
|  | } | 
|  |  | 
|  | /******************************************************************** | 
|  | *      IInternetZoneManager_Construct | 
|  | */ | 
|  | static const IInternetZoneManagerVtbl ZoneMgrImplVtbl = { | 
|  | ZoneMgrImpl_QueryInterface, | 
|  | ZoneMgrImpl_AddRef, | 
|  | ZoneMgrImpl_Release, | 
|  | ZoneMgrImpl_GetZoneAttributes, | 
|  | ZoneMgrImpl_SetZoneAttributes, | 
|  | ZoneMgrImpl_GetZoneCustomPolicy, | 
|  | ZoneMgrImpl_SetZoneCustomPolicy, | 
|  | ZoneMgrImpl_GetZoneActionPolicy, | 
|  | ZoneMgrImpl_SetZoneActionPolicy, | 
|  | ZoneMgrImpl_PromptAction, | 
|  | ZoneMgrImpl_LogAction, | 
|  | ZoneMgrImpl_CreateZoneEnumerator, | 
|  | ZoneMgrImpl_GetZoneAt, | 
|  | ZoneMgrImpl_DestroyZoneEnumerator, | 
|  | ZoneMgrImpl_CopyTemplatePoliciesToZone, | 
|  | }; | 
|  |  | 
|  | HRESULT ZoneMgrImpl_Construct(IUnknown *pUnkOuter, LPVOID *ppobj) | 
|  | { | 
|  | ZoneMgrImpl* ret = HeapAlloc(GetProcessHeap(), 0, sizeof(ZoneMgrImpl)); | 
|  |  | 
|  | TRACE("(%p %p)\n", pUnkOuter, ppobj); | 
|  | ret->lpVtbl = &ZoneMgrImplVtbl; | 
|  | ret->ref = 1; | 
|  | *ppobj = (IInternetZoneManager*)ret; | 
|  |  | 
|  | URLMON_LockModule(); | 
|  |  | 
|  | return S_OK; | 
|  | } | 
|  |  | 
|  | /*********************************************************************** | 
|  | *           CoInternetCreateSecurityManager (URLMON.@) | 
|  | * | 
|  | */ | 
|  | HRESULT WINAPI CoInternetCreateSecurityManager( IServiceProvider *pSP, | 
|  | IInternetSecurityManager **ppSM, DWORD dwReserved ) | 
|  | { | 
|  | TRACE("%p %p %ld\n", pSP, ppSM, dwReserved ); | 
|  |  | 
|  | if(pSP) | 
|  | FIXME("pSP not supported\n"); | 
|  |  | 
|  | return SecManagerImpl_Construct(NULL, (void**) ppSM); | 
|  | } | 
|  |  | 
|  | /******************************************************************** | 
|  | *      CoInternetCreateZoneManager (URLMON.@) | 
|  | */ | 
|  | HRESULT WINAPI CoInternetCreateZoneManager(IServiceProvider* pSP, IInternetZoneManager** ppZM, DWORD dwReserved) | 
|  | { | 
|  | TRACE("(%p %p %lx)\n", pSP, ppZM, dwReserved); | 
|  | return ZoneMgrImpl_Construct(NULL, (void**)ppZM); | 
|  | } |