urlmon: Added CoInternetIsFeatureZoneElevationEnabled implementation.

diff --git a/dlls/urlmon/internet.c b/dlls/urlmon/internet.c
index 72830f50a4d297cdba0b6491981df935cc9070ae..d4a49b5e55c44a221ef124d1523d9d6a46f7ff11 100644 (file)
--- a/dlls/urlmon/internet.c
+++ b/dlls/urlmon/internet.c
@@ -742,3 +742,53 @@ HRESULT WINAPI CoInternetIsFeatureEnabledForUrl(INTERNETFEATURELIST FeatureEntry
     }
     }
 }
+
+/***********************************************************************
+ *             CoInternetIsFeatureZoneElevationEnabled (URLMON.@)
+ */
+HRESULT WINAPI CoInternetIsFeatureZoneElevationEnabled(LPCWSTR szFromURL, LPCWSTR szToURL,
+        IInternetSecurityManager *pSecMgr, DWORD dwFlags)
+{
+    HRESULT hres;
+
+    TRACE("(%s %s %p %x)\n", debugstr_w(szFromURL), debugstr_w(szToURL), pSecMgr, dwFlags);
+
+    if(!pSecMgr || !szToURL)
+        return CoInternetIsFeatureEnabled(FEATURE_ZONE_ELEVATION, dwFlags);
+
+    switch(dwFlags) {
+    case GET_FEATURE_FROM_THREAD:
+    case GET_FEATURE_FROM_THREAD_LOCALMACHINE:
+    case GET_FEATURE_FROM_THREAD_INTRANET:
+    case GET_FEATURE_FROM_THREAD_TRUSTED:
+    case GET_FEATURE_FROM_THREAD_INTERNET:
+    case GET_FEATURE_FROM_THREAD_RESTRICTED:
+        FIXME("unsupported flags %x\n", dwFlags);
+        return E_NOTIMPL;
+
+    case GET_FEATURE_FROM_PROCESS:
+        hres = CoInternetIsFeatureEnabled(FEATURE_ZONE_ELEVATION, dwFlags);
+        if(hres != S_OK)
+            return hres;
+        /* fall through */
+
+    default: {
+        DWORD policy = URLPOLICY_DISALLOW;
+
+        hres = IInternetSecurityManager_ProcessUrlAction(pSecMgr, szToURL,
+                URLACTION_FEATURE_ZONE_ELEVATION, (BYTE*)&policy, sizeof(DWORD),
+                NULL, 0, PUAF_NOUI, 0);
+        if(FAILED(hres))
+            return S_OK;
+
+        switch(policy) {
+        case URLPOLICY_ALLOW:
+            return S_FALSE;
+        case URLPOLICY_QUERY:
+            FIXME("Ask user dialog not implemented\n");
+        default:
+            return S_OK;
+        }
+    }
+    }
+}

diff --git a/dlls/urlmon/urlmon.spec b/dlls/urlmon/urlmon.spec
index 28af3c5089af15f958c4a7d8ee680d1c73a3e18a..76bf718ec0578a165b69cec29c108f930c65e0b1 100644 (file)
--- a/dlls/urlmon/urlmon.spec
+++ b/dlls/urlmon/urlmon.spec
@@ -22,6 +22,7 @@
 @ stdcall CoInternetGetSession(long ptr long)
 @ stdcall CoInternetIsFeatureEnabled(long long)
 @ stdcall CoInternetIsFeatureEnabledForUrl(long long wstr ptr)
+@ stdcall CoInternetIsFeatureZoneElevationEnabled(wstr wstr ptr long)
 @ stdcall CoInternetParseUrl(wstr long long wstr long ptr long)
 @ stdcall CoInternetParseIUri(ptr long long wstr long ptr long)
 @ stdcall CoInternetQueryInfo(ptr long long ptr long ptr long)