dlls/jscript/tests/jscript.c

   1 /*
   2  * Copyright 2008 Jacek Caban for CodeWeavers
   3  *
   4  * This library is free software; you can redistribute it and/or
   5  * modify it under the terms of the GNU Lesser General Public
   6  * License as published by the Free Software Foundation; either
   7  * version 2.1 of the License, or (at your option) any later version.
   8  *
   9  * This library is distributed in the hope that it will be useful,
  10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  12  * Lesser General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU Lesser General Public
  15  * License along with this library; if not, write to the Free Software
  16  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  17  */
  18
  19 #define COBJMACROS
  20 #define CONST_VTABLE
  21
  22 #include <initguid.h>
  23 #include <ole2.h>
  24 #include <activscp.h>
  25 #include <objsafe.h>
  26
  27 #include "wine/test.h"
  28
  29 DEFINE_GUID(GUID_NULL,0,0,0,0,0,0,0,0,0,0,0);
  30
  31 static const CLSID CLSID_JScript =
  32     {0xf414c260,0x6ac0,0x11cf,{0xb6,0xd1,0x00,0xaa,0x00,0xbb,0xbb,0x58}};
  33
  34 #define DEFINE_EXPECT(func) \
  35     static BOOL expect_ ## func = FALSE, called_ ## func = FALSE
  36
  37 #define SET_EXPECT(func) \
  38     expect_ ## func = TRUE
  39
  40 #define SET_CALLED(func) \
  41     called_ ## func = TRUE
  42
  43 #define CHECK_EXPECT2(func) \
  44     do { \
  45         ok(expect_ ##func, "unexpected call " #func "\n"); \
  46         called_ ## func = TRUE; \
  47     }while(0)
  48
  49 #define CHECK_EXPECT(func) \
  50     do { \
  51         CHECK_EXPECT2(func); \
  52         expect_ ## func = FALSE; \
  53     }while(0)
  54
  55 #define CHECK_CALLED(func) \
  56     do { \
  57         ok(called_ ## func, "expected " #func "\n"); \
  58         expect_ ## func = called_ ## func = FALSE; \
  59     }while(0)
  60
  61 DEFINE_EXPECT(GetLCID);
  62 DEFINE_EXPECT(OnStateChange_STARTED);
  63 DEFINE_EXPECT(OnStateChange_CLOSED);
  64 DEFINE_EXPECT(OnStateChange_INITIALIZED);
  65 DEFINE_EXPECT(OnEnterScript);
  66 DEFINE_EXPECT(OnLeaveScript);
  67
  68 static HRESULT WINAPI ActiveScriptSite_QueryInterface(IActiveScriptSite *iface, REFIID riid, void **ppv)
  69 {
  70     *ppv = NULL;
  71
  72     if(IsEqualGUID(&IID_IUnknown, riid))
  73         *ppv = iface;
  74     else if(IsEqualGUID(&IID_IActiveScriptSite, riid))
  75         *ppv = iface;
  76     else
  77         return E_NOINTERFACE;
  78
  79     IUnknown_AddRef((IUnknown*)*ppv);
  80     return S_OK;
  81 }
  82
  83 static ULONG WINAPI ActiveScriptSite_AddRef(IActiveScriptSite *iface)
  84 {
  85     return 2;
  86 }
  87
  88 static ULONG WINAPI ActiveScriptSite_Release(IActiveScriptSite *iface)
  89 {
  90     return 1;
  91 }
  92
  93 static HRESULT WINAPI ActiveScriptSite_GetLCID(IActiveScriptSite *iface, LCID *plcid)
  94 {
  95     CHECK_EXPECT(GetLCID);
  96     return E_NOTIMPL;
  97 }
  98
  99 static HRESULT WINAPI ActiveScriptSite_GetItemInfo(IActiveScriptSite *iface, LPCOLESTR pstrName,
 100         DWORD dwReturnMask, IUnknown **ppiunkItem, ITypeInfo **ppti)
 101 {
 102     ok(0, "unexpected call\n");
 103     return E_NOTIMPL;
 104 }
 105
 106 static HRESULT WINAPI ActiveScriptSite_GetDocVersionString(IActiveScriptSite *iface, BSTR *pbstrVersion)
 107 {
 108     ok(0, "unexpected call\n");
 109     return E_NOTIMPL;
 110 }
 111
 112 static HRESULT WINAPI ActiveScriptSite_OnScriptTerminate(IActiveScriptSite *iface,
 113         const VARIANT *pvarResult, const EXCEPINFO *pexcepinfo)
 114 {
 115     ok(0, "unexpected call\n");
 116     return E_NOTIMPL;
 117 }
 118
 119 static HRESULT WINAPI ActiveScriptSite_OnStateChange(IActiveScriptSite *iface, SCRIPTSTATE ssScriptState)
 120 {
 121     switch(ssScriptState) {
 122     case SCRIPTSTATE_STARTED:
 123         CHECK_EXPECT(OnStateChange_STARTED);
 124         return S_OK;
 125     case SCRIPTSTATE_CLOSED:
 126         CHECK_EXPECT(OnStateChange_CLOSED);
 127         return S_OK;
 128     case SCRIPTSTATE_INITIALIZED:
 129         CHECK_EXPECT(OnStateChange_INITIALIZED);
 130         return S_OK;
 131     default:
 132         ok(0, "unexpected call %d\n", ssScriptState);
 133     }
 134
 135     return E_NOTIMPL;
 136 }
 137
 138 static HRESULT WINAPI ActiveScriptSite_OnScriptError(IActiveScriptSite *iface, IActiveScriptError *pscripterror)
 139 {
 140     ok(0, "unexpected call\n");
 141     return E_NOTIMPL;
 142 }
 143
 144 static HRESULT WINAPI ActiveScriptSite_OnEnterScript(IActiveScriptSite *iface)
 145 {
 146     CHECK_EXPECT(OnEnterScript);
 147     return S_OK;
 148 }
 149
 150 static HRESULT WINAPI ActiveScriptSite_OnLeaveScript(IActiveScriptSite *iface)
 151 {
 152     CHECK_EXPECT(OnLeaveScript);
 153     return S_OK;
 154 }
 155
 156 #undef ACTSCPSITE_THIS
 157
 158 static const IActiveScriptSiteVtbl ActiveScriptSiteVtbl = {
 159     ActiveScriptSite_QueryInterface,
 160     ActiveScriptSite_AddRef,
 161     ActiveScriptSite_Release,
 162     ActiveScriptSite_GetLCID,
 163     ActiveScriptSite_GetItemInfo,
 164     ActiveScriptSite_GetDocVersionString,
 165     ActiveScriptSite_OnScriptTerminate,
 166     ActiveScriptSite_OnStateChange,
 167     ActiveScriptSite_OnScriptError,
 168     ActiveScriptSite_OnEnterScript,
 169     ActiveScriptSite_OnLeaveScript
 170 };
 171
 172 static IActiveScriptSite ActiveScriptSite = { &ActiveScriptSiteVtbl };
 173
 174 static void test_safety(IUnknown *unk)
 175 {
 176     IObjectSafety *safety;
 177     DWORD supported, enabled;
 178     HRESULT hres;
 179
 180     hres = IUnknown_QueryInterface(unk, &IID_IObjectSafety, (void**)&safety);
 181     ok(hres == S_OK, "Could not get IObjectSafety: %08x\n", hres);
 182     if(FAILED(hres))
 183         return;
 184
 185     hres = IObjectSafety_GetInterfaceSafetyOptions(safety, &IID_NULL, &supported, NULL);
 186     ok(hres == E_POINTER, "GetInterfaceSafetyOptions failed: %08x, expected E_POINTER\n", hres);
 187     hres = IObjectSafety_GetInterfaceSafetyOptions(safety, &IID_NULL, NULL, &enabled);
 188     ok(hres == E_POINTER, "GetInterfaceSafetyOptions failed: %08x, expected E_POINTER\n", hres);
 189
 190     supported = enabled = 0xdeadbeef;
 191     hres = IObjectSafety_GetInterfaceSafetyOptions(safety, &IID_NULL, &supported, &enabled);
 192     ok(hres == S_OK, "GetInterfaceSafetyOptions failed: %08x\n", hres);
 193     ok(supported == (INTERFACESAFE_FOR_UNTRUSTED_DATA|INTERFACE_USES_DISPEX|INTERFACE_USES_SECURITY_MANAGER),
 194        "supported=%x\n", supported);
 195     ok(enabled == INTERFACE_USES_DISPEX, "enabled=%x\n", enabled);
 196
 197     supported = enabled = 0xdeadbeef;
 198     hres = IObjectSafety_GetInterfaceSafetyOptions(safety, &IID_IActiveScript, &supported, &enabled);
 199     ok(hres == S_OK, "GetInterfaceSafetyOptions failed: %08x\n", hres);
 200     ok(supported == (INTERFACESAFE_FOR_UNTRUSTED_DATA|INTERFACE_USES_DISPEX|INTERFACE_USES_SECURITY_MANAGER),
 201        "supported=%x\n", supported);
 202     ok(enabled == INTERFACE_USES_DISPEX, "enabled=%x\n", enabled);
 203
 204     supported = enabled = 0xdeadbeef;
 205     hres = IObjectSafety_GetInterfaceSafetyOptions(safety, &IID_IActiveScriptParse, &supported, &enabled);
 206     ok(hres == S_OK, "GetInterfaceSafetyOptions failed: %08x\n", hres);
 207     ok(supported == (INTERFACESAFE_FOR_UNTRUSTED_DATA|INTERFACE_USES_DISPEX|INTERFACE_USES_SECURITY_MANAGER),
 208        "supported=%x\n", supported);
 209     ok(enabled == INTERFACE_USES_DISPEX, "enabled=%x\n", enabled);
 210
 211     hres = IObjectSafety_SetInterfaceSafetyOptions(safety, &IID_IActiveScriptParse,
 212             INTERFACESAFE_FOR_UNTRUSTED_DATA|INTERFACE_USES_DISPEX|INTERFACE_USES_SECURITY_MANAGER
 213                 |INTERFACESAFE_FOR_UNTRUSTED_CALLER,
 214             INTERFACESAFE_FOR_UNTRUSTED_DATA|INTERFACE_USES_DISPEX|INTERFACE_USES_SECURITY_MANAGER);
 215     ok(hres == E_FAIL, "SetInterfaceSafetyOptions failed: %08x, expected E_FAIL\n", hres);
 216
 217     hres = IObjectSafety_SetInterfaceSafetyOptions(safety, &IID_IActiveScriptParse,
 218             INTERFACESAFE_FOR_UNTRUSTED_DATA|INTERFACE_USES_DISPEX|INTERFACE_USES_SECURITY_MANAGER,
 219             INTERFACESAFE_FOR_UNTRUSTED_DATA|INTERFACE_USES_DISPEX|INTERFACE_USES_SECURITY_MANAGER);
 220     ok(hres == S_OK, "SetInterfaceSafetyOptions failed: %08x\n", hres);
 221
 222     supported = enabled = 0xdeadbeef;
 223     hres = IObjectSafety_GetInterfaceSafetyOptions(safety, &IID_IActiveScriptParse, &supported, &enabled);
 224     ok(hres == S_OK, "GetInterfaceSafetyOptions failed: %08x\n", hres);
 225     ok(supported == (INTERFACESAFE_FOR_UNTRUSTED_DATA|INTERFACE_USES_DISPEX|INTERFACE_USES_SECURITY_MANAGER),
 226        "supported=%x\n", supported);
 227     ok(enabled == (INTERFACESAFE_FOR_UNTRUSTED_DATA|INTERFACE_USES_DISPEX|INTERFACE_USES_SECURITY_MANAGER),
 228        "enabled=%x\n", enabled);
 229
 230     IObjectSafety_Release(safety);
 231 }
 232
 233 static void test_jscript(void)
 234 {
 235     IActiveScriptParse *parse;
 236     IActiveScript *script;
 237     IUnknown *unk;
 238     ULONG ref;
 239     HRESULT hres;
 240
 241     hres = CoCreateInstance(&CLSID_JScript, NULL, CLSCTX_INPROC_SERVER|CLSCTX_INPROC_HANDLER,
 242             &IID_IUnknown, (void**)&unk);
 243     ok(hres == S_OK, "CoCreateInstance failed: %08x\n", hres);
 244     if(FAILED(hres))
 245         return;
 246
 247     hres = IUnknown_QueryInterface(unk, &IID_IActiveScript, (void**)&script);
 248     ok(hres == S_OK, "Could not get IActiveScript: %08x\n", hres);
 249
 250     hres = IUnknown_QueryInterface(unk, &IID_IActiveScriptParse, (void**)&parse);
 251     ok(hres == S_OK, "Could not get IActiveScriptParse: %08x\n", hres);
 252
 253     test_safety(unk);
 254
 255     hres = IActiveScriptParse_InitNew(parse);
 256     ok(hres == S_OK, "InitNew failed: %08x\n", hres);
 257
 258     hres = IActiveScriptParse_InitNew(parse);
 259     ok(hres == E_UNEXPECTED, "InitNew failed: %08x, expected E_UNEXPECTED\n", hres);
 260
 261     hres = IActiveScript_SetScriptSite(script, NULL);
 262     ok(hres == E_POINTER, "SetScriptSite failed: %08x, expected E_POINTER\n", hres);
 263
 264     SET_EXPECT(GetLCID);
 265     SET_EXPECT(OnStateChange_INITIALIZED);
 266     hres = IActiveScript_SetScriptSite(script, &ActiveScriptSite);
 267     ok(hres == S_OK, "SetScriptSite failed: %08x\n", hres);
 268     CHECK_CALLED(GetLCID);
 269     CHECK_CALLED(OnStateChange_INITIALIZED);
 270
 271     hres = IActiveScript_SetScriptSite(script, &ActiveScriptSite);
 272     ok(hres == E_UNEXPECTED, "SetScriptSite failed: %08x, expected E_UNEXPECTED\n", hres);
 273
 274     SET_EXPECT(OnStateChange_CLOSED);
 275     hres = IActiveScript_Close(script);
 276     ok(hres == S_OK, "Close failed: %08x\n", hres);
 277     CHECK_CALLED(OnStateChange_CLOSED);
 278
 279     IActiveScriptParse_Release(parse);
 280     IActiveScript_Release(script);
 281
 282     ref = IUnknown_Release(unk);
 283     ok(!ref, "ref = %d\n", ref);
 284 }
 285
 286 START_TEST(jscript)
 287 {
 288     CoInitialize(NULL);
 289
 290     test_jscript();
 291
 292     CoUninitialize();
 293 }