Gabriel Ivăncescu : mshtml: Forward SID_SInternetHostSecurityManager of the document obj to the doc node.

[Alexandre Julliard]

Module: wine
Branch: master
Commit: 370a339055918c028eabe46dbe2677d674b735b7
URL:    https://gitlab.winehq.org/wine/wine/-/commit/370a339055918c028eabe46dbe2677d...

Author: Gabriel Ivăncescu 
Date:   Mon Jan 29 16:59:39 2024 +0200

mshtml: Forward SID_SInternetHostSecurityManager of the document obj to the doc node.

Signed-off-by: Gabriel Ivăncescu

---

dlls/mshtml/service.c      |  5 +++++
 dlls/mshtml/tests/script.c | 17 +++++++++++++++++
 2 files changed, 22 insertions(+)

diff --git a/dlls/mshtml/service.c b/dlls/mshtml/service.c
index 01dd9fee836..6042dca4111 100644
--- a/dlls/mshtml/service.c
+++ b/dlls/mshtml/service.c
@@ -425,6 +425,11 @@ static HRESULT WINAPI DocObjServiceProvider_QueryService(IServiceProvider *iface
         return IOleUndoManager_QueryInterface(This->undomgr, riid, ppv);
     }

+    if(IsEqualGUID(&SID_SInternetHostSecurityManager, guidService)) {
+        TRACE("SID_SInternetHostSecurityManager\n");
+        return IInternetHostSecurityManager_QueryInterface(&This->doc_node->IInternetHostSecurityManager_iface, riid, ppv);
+    }
+
     if(IsEqualGUID(&SID_SContainerDispatch, guidService)) {
         TRACE("SID_SContainerDispatch\n");
         return IHTMLDocument2_QueryInterface(&This->IHTMLDocument2_iface, riid, ppv);
diff --git a/dlls/mshtml/tests/script.c b/dlls/mshtml/tests/script.c
index 35c3680093d..b38a36f3462 100644
--- a/dlls/mshtml/tests/script.c
+++ b/dlls/mshtml/tests/script.c
@@ -4541,7 +4541,9 @@ static void test_exec_script(IHTMLDocument2 *doc, const WCHAR *codew, const WCHA

static void test_simple_script(void)
 {
+    IInternetHostSecurityManager *sec_mgr, *sec_mgr2;
     IHTMLDocument2 *doc_node;
+    IServiceProvider *sp;
     IHTMLWindow2 *window;
     IHTMLDocument2 *doc;
     HRESULT hres;
@@ -4621,6 +4623,21 @@ static void test_simple_script(void)
     hres = IHTMLWindow2_get_document(window, &doc_node);
     ok(hres == S_OK, "get_document failed: %08lx\n", hres);

+    hres = IHTMLDocument2_QueryInterface(doc, &IID_IServiceProvider, (void**)&sp);
+    ok(hres == S_OK, "Could not get IServiceProvider iface: %08lx\n", hres);
+    hres = IServiceProvider_QueryService(sp, &SID_SInternetHostSecurityManager, &IID_IInternetHostSecurityManager, (void**)&sec_mgr);
+    ok(hres == S_OK, "QueryService failed: %08lx\n", hres);
+    IServiceProvider_Release(sp);
+
+    hres = IHTMLDocument2_QueryInterface(doc_node, &IID_IServiceProvider, (void**)&sp);
+    ok(hres == S_OK, "Could not get IServiceProvider iface: %08lx\n", hres);
+    hres = IServiceProvider_QueryService(sp, &SID_SInternetHostSecurityManager, &IID_IInternetHostSecurityManager, (void**)&sec_mgr2);
+    ok(hres == S_OK, "QueryService failed: %08lx\n", hres);
+    ok(iface_cmp(sec_mgr, sec_mgr2), "sec_mgr != sec_mgr2\n");
+    IInternetHostSecurityManager_Release(sec_mgr2);
+    IInternetHostSecurityManager_Release(sec_mgr);
+    IServiceProvider_Release(sp);
+
     SET_EXPECT(SetScriptState_DISCONNECTED);
     SET_EXPECT(Close);
     SET_EXPECT(Close2);