2 * Copyright 2011 Jacek Caban for CodeWeavers
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.
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.
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
25 #include "wine/debug.h"
27 WINE_DEFAULT_DEBUG_CHANNEL(vbscript);
31 #define CTXARG_T DWORDLONG
32 #define IActiveScriptParseVtbl IActiveScriptParse64Vtbl
36 #define CTXARG_T DWORD
37 #define IActiveScriptParseVtbl IActiveScriptParse32Vtbl
42 IActiveScript IActiveScript_iface;
43 IActiveScriptParse IActiveScriptParse_iface;
44 IObjectSafety IObjectSafety_iface;
50 IActiveScriptSite *site;
56 static void change_state(VBScript *This, SCRIPTSTATE state)
58 if(This->state == state)
63 IActiveScriptSite_OnStateChange(This->site, state);
66 static void exec_queued_code(VBScript *This)
71 static HRESULT set_ctx_site(VBScript *This)
75 This->ctx->lcid = This->lcid;
77 hres = init_global(This->ctx);
81 IActiveScriptSite_AddRef(This->site);
82 This->ctx->site = This->site;
84 change_state(This, SCRIPTSTATE_INITIALIZED);
88 static void destroy_script(script_ctx_t *ctx)
90 while(!list_empty(&ctx->code_list))
91 release_vbscode(LIST_ENTRY(list_head(&ctx->code_list), vbscode_t, entry));
93 while(!list_empty(&ctx->named_items)) {
94 named_item_t *iter = LIST_ENTRY(list_head(&ctx->named_items), named_item_t, entry);
96 list_remove(&iter->entry);
98 IDispatch_Release(iter->disp);
99 heap_free(iter->name);
104 IDispatch_Release(ctx->host_global);
106 IActiveScriptSite_Release(ctx->site);
108 IDispatchEx_Release(&ctx->script_obj->IDispatchEx_iface);
112 static void decrease_state(VBScript *This, SCRIPTSTATE state)
114 switch(This->state) {
115 case SCRIPTSTATE_CONNECTED:
116 change_state(This, SCRIPTSTATE_DISCONNECTED);
117 if(state == SCRIPTSTATE_DISCONNECTED)
120 case SCRIPTSTATE_STARTED:
121 case SCRIPTSTATE_DISCONNECTED:
122 if(This->state == SCRIPTSTATE_DISCONNECTED)
123 change_state(This, SCRIPTSTATE_INITIALIZED);
124 if(state == SCRIPTSTATE_INITIALIZED)
127 case SCRIPTSTATE_INITIALIZED:
128 case SCRIPTSTATE_UNINITIALIZED:
129 change_state(This, state);
132 IActiveScriptSite_Release(This->site);
138 if(state == SCRIPTSTATE_CLOSED) {
139 destroy_script(This->ctx);
149 static inline VBScript *impl_from_IActiveScript(IActiveScript *iface)
151 return CONTAINING_RECORD(iface, VBScript, IActiveScript_iface);
154 static HRESULT WINAPI VBScript_QueryInterface(IActiveScript *iface, REFIID riid, void **ppv)
156 VBScript *This = impl_from_IActiveScript(iface);
158 if(IsEqualGUID(riid, &IID_IUnknown)) {
159 TRACE("(%p)->(IID_IUnknown %p)\n", This, ppv);
160 *ppv = &This->IActiveScript_iface;
161 }else if(IsEqualGUID(riid, &IID_IActiveScript)) {
162 TRACE("(%p)->(IID_IActiveScript %p)\n", This, ppv);
163 *ppv = &This->IActiveScript_iface;
164 }else if(IsEqualGUID(riid, &IID_IActiveScriptParse)) {
165 TRACE("(%p)->(IID_IActiveScriptParse %p)\n", This, ppv);
166 *ppv = &This->IActiveScriptParse_iface;
167 }else if(IsEqualGUID(riid, &IID_IObjectSafety)) {
168 TRACE("(%p)->(IID_IObjectSafety %p)\n", This, ppv);
169 *ppv = &This->IObjectSafety_iface;
171 FIXME("(%p)->(%s %p)\n", This, debugstr_guid(riid), ppv);
173 return E_NOINTERFACE;
176 IUnknown_AddRef((IUnknown*)*ppv);
180 static ULONG WINAPI VBScript_AddRef(IActiveScript *iface)
182 VBScript *This = impl_from_IActiveScript(iface);
183 LONG ref = InterlockedIncrement(&This->ref);
185 TRACE("(%p) ref=%d\n", This, ref);
190 static ULONG WINAPI VBScript_Release(IActiveScript *iface)
192 VBScript *This = impl_from_IActiveScript(iface);
193 LONG ref = InterlockedDecrement(&This->ref);
195 TRACE("(%p) ref=%d\n", iface, ref);
199 IActiveScriptSite_Release(This->site);
206 static HRESULT WINAPI VBScript_SetScriptSite(IActiveScript *iface, IActiveScriptSite *pass)
208 VBScript *This = impl_from_IActiveScript(iface);
212 TRACE("(%p)->(%p)\n", This, pass);
220 if(InterlockedCompareExchange(&This->thread_id, GetCurrentThreadId(), 0))
224 IActiveScriptSite_AddRef(This->site);
226 hres = IActiveScriptSite_GetLCID(This->site, &lcid);
230 return This->ctx ? set_ctx_site(This) : S_OK;
233 static HRESULT WINAPI VBScript_GetScriptSite(IActiveScript *iface, REFIID riid,
236 VBScript *This = impl_from_IActiveScript(iface);
237 FIXME("(%p)->()\n", This);
241 static HRESULT WINAPI VBScript_SetScriptState(IActiveScript *iface, SCRIPTSTATE ss)
243 VBScript *This = impl_from_IActiveScript(iface);
245 TRACE("(%p)->(%d)\n", This, ss);
247 if(This->thread_id && GetCurrentThreadId() != This->thread_id)
250 if(ss == SCRIPTSTATE_UNINITIALIZED) {
251 if(This->state == SCRIPTSTATE_CLOSED)
254 decrease_state(This, SCRIPTSTATE_UNINITIALIZED);
262 case SCRIPTSTATE_STARTED:
263 case SCRIPTSTATE_CONNECTED: /* FIXME */
264 if(This->state == SCRIPTSTATE_CLOSED)
267 exec_queued_code(This);
269 case SCRIPTSTATE_INITIALIZED:
270 FIXME("unimplemented SCRIPTSTATE_INITIALIZED\n");
273 FIXME("unimplemented state %d\n", ss);
277 change_state(This, ss);
281 static HRESULT WINAPI VBScript_GetScriptState(IActiveScript *iface, SCRIPTSTATE *pssState)
283 VBScript *This = impl_from_IActiveScript(iface);
285 TRACE("(%p)->(%p)\n", This, pssState);
290 if(This->thread_id && This->thread_id != GetCurrentThreadId())
293 *pssState = This->state;
297 static HRESULT WINAPI VBScript_Close(IActiveScript *iface)
299 VBScript *This = impl_from_IActiveScript(iface);
301 TRACE("(%p)->()\n", This);
303 if(This->thread_id && This->thread_id != GetCurrentThreadId())
306 decrease_state(This, SCRIPTSTATE_CLOSED);
310 static HRESULT WINAPI VBScript_AddNamedItem(IActiveScript *iface, LPCOLESTR pstrName, DWORD dwFlags)
312 VBScript *This = impl_from_IActiveScript(iface);
314 IDispatch *disp = NULL;
317 TRACE("(%p)->(%s %x)\n", This, debugstr_w(pstrName), dwFlags);
319 if(This->thread_id != GetCurrentThreadId() || !This->ctx || This->state == SCRIPTSTATE_CLOSED)
322 if(dwFlags & SCRIPTITEM_GLOBALMEMBERS) {
325 hres = IActiveScriptSite_GetItemInfo(This->site, pstrName, SCRIPTINFO_IUNKNOWN, &unk, NULL);
327 WARN("GetItemInfo failed: %08x\n", hres);
331 hres = IUnknown_QueryInterface(unk, &IID_IDispatch, (void**)&disp);
332 IUnknown_Release(unk);
334 WARN("object does not implement IDispatch\n");
338 if(This->ctx->host_global)
339 IDispatch_Release(This->ctx->host_global);
340 IDispatch_AddRef(disp);
341 This->ctx->host_global = disp;
344 item = heap_alloc(sizeof(*item));
347 IDispatch_Release(disp);
348 return E_OUTOFMEMORY;
352 item->flags = dwFlags;
353 item->name = heap_strdupW(pstrName);
356 IDispatch_Release(disp);
358 return E_OUTOFMEMORY;
361 list_add_tail(&This->ctx->named_items, &item->entry);
365 static HRESULT WINAPI VBScript_AddTypeLib(IActiveScript *iface, REFGUID rguidTypeLib,
366 DWORD dwMajor, DWORD dwMinor, DWORD dwFlags)
368 VBScript *This = impl_from_IActiveScript(iface);
369 FIXME("(%p)->()\n", This);
373 static HRESULT WINAPI VBScript_GetScriptDispatch(IActiveScript *iface, LPCOLESTR pstrItemName, IDispatch **ppdisp)
375 VBScript *This = impl_from_IActiveScript(iface);
377 TRACE("(%p)->(%p)\n", This, ppdisp);
382 if(This->thread_id != GetCurrentThreadId() || !This->ctx->script_obj) {
387 *ppdisp = (IDispatch*)&This->ctx->script_obj->IDispatchEx_iface;
388 IDispatch_AddRef(*ppdisp);
392 static HRESULT WINAPI VBScript_GetCurrentScriptThreadID(IActiveScript *iface,
393 SCRIPTTHREADID *pstridThread)
395 VBScript *This = impl_from_IActiveScript(iface);
396 FIXME("(%p)->()\n", This);
400 static HRESULT WINAPI VBScript_GetScriptThreadID(IActiveScript *iface,
401 DWORD dwWin32ThreadId, SCRIPTTHREADID *pstidThread)
403 VBScript *This = impl_from_IActiveScript(iface);
404 FIXME("(%p)->()\n", This);
408 static HRESULT WINAPI VBScript_GetScriptThreadState(IActiveScript *iface,
409 SCRIPTTHREADID stidThread, SCRIPTTHREADSTATE *pstsState)
411 VBScript *This = impl_from_IActiveScript(iface);
412 FIXME("(%p)->()\n", This);
416 static HRESULT WINAPI VBScript_InterruptScriptThread(IActiveScript *iface,
417 SCRIPTTHREADID stidThread, const EXCEPINFO *pexcepinfo, DWORD dwFlags)
419 VBScript *This = impl_from_IActiveScript(iface);
420 FIXME("(%p)->()\n", This);
424 static HRESULT WINAPI VBScript_Clone(IActiveScript *iface, IActiveScript **ppscript)
426 VBScript *This = impl_from_IActiveScript(iface);
427 FIXME("(%p)->()\n", This);
431 static const IActiveScriptVtbl VBScriptVtbl = {
432 VBScript_QueryInterface,
435 VBScript_SetScriptSite,
436 VBScript_GetScriptSite,
437 VBScript_SetScriptState,
438 VBScript_GetScriptState,
440 VBScript_AddNamedItem,
442 VBScript_GetScriptDispatch,
443 VBScript_GetCurrentScriptThreadID,
444 VBScript_GetScriptThreadID,
445 VBScript_GetScriptThreadState,
446 VBScript_InterruptScriptThread,
450 static inline VBScript *impl_from_IActiveScriptParse(IActiveScriptParse *iface)
452 return CONTAINING_RECORD(iface, VBScript, IActiveScriptParse_iface);
455 static HRESULT WINAPI VBScriptParse_QueryInterface(IActiveScriptParse *iface, REFIID riid, void **ppv)
457 VBScript *This = impl_from_IActiveScriptParse(iface);
458 return IActiveScript_QueryInterface(&This->IActiveScript_iface, riid, ppv);
461 static ULONG WINAPI VBScriptParse_AddRef(IActiveScriptParse *iface)
463 VBScript *This = impl_from_IActiveScriptParse(iface);
464 return IActiveScript_AddRef(&This->IActiveScript_iface);
467 static ULONG WINAPI VBScriptParse_Release(IActiveScriptParse *iface)
469 VBScript *This = impl_from_IActiveScriptParse(iface);
470 return IActiveScript_Release(&This->IActiveScript_iface);
473 static HRESULT WINAPI VBScriptParse_InitNew(IActiveScriptParse *iface)
475 VBScript *This = impl_from_IActiveScriptParse(iface);
476 script_ctx_t *ctx, *old_ctx;
478 TRACE("(%p)\n", This);
483 ctx = heap_alloc_zero(sizeof(script_ctx_t));
485 return E_OUTOFMEMORY;
487 list_init(&ctx->code_list);
488 list_init(&ctx->named_items);
490 old_ctx = InterlockedCompareExchangePointer((void**)&This->ctx, ctx, NULL);
496 return This->site ? set_ctx_site(This) : S_OK;
499 static HRESULT WINAPI VBScriptParse_AddScriptlet(IActiveScriptParse *iface,
500 LPCOLESTR pstrDefaultName, LPCOLESTR pstrCode, LPCOLESTR pstrItemName,
501 LPCOLESTR pstrSubItemName, LPCOLESTR pstrEventName, LPCOLESTR pstrDelimiter,
502 CTXARG_T dwSourceContextCookie, ULONG ulStartingLineNumber, DWORD dwFlags,
503 BSTR *pbstrName, EXCEPINFO *pexcepinfo)
505 VBScript *This = impl_from_IActiveScriptParse(iface);
506 FIXME("(%p)->(%s %s %s %s %s %s %s %u %x %p %p)\n", This, debugstr_w(pstrDefaultName),
507 debugstr_w(pstrCode), debugstr_w(pstrItemName), debugstr_w(pstrSubItemName),
508 debugstr_w(pstrEventName), debugstr_w(pstrDelimiter), wine_dbgstr_longlong(dwSourceContextCookie),
509 ulStartingLineNumber, dwFlags, pbstrName, pexcepinfo);
513 static HRESULT WINAPI VBScriptParse_ParseScriptText(IActiveScriptParse *iface,
514 LPCOLESTR pstrCode, LPCOLESTR pstrItemName, IUnknown *punkContext,
515 LPCOLESTR pstrDelimiter, CTXARG_T dwSourceContextCookie, ULONG ulStartingLine,
516 DWORD dwFlags, VARIANT *pvarResult, EXCEPINFO *pexcepinfo)
518 VBScript *This = impl_from_IActiveScriptParse(iface);
522 TRACE("(%p)->(%s %s %p %s %s %u %x %p %p)\n", This, debugstr_w(pstrCode),
523 debugstr_w(pstrItemName), punkContext, debugstr_w(pstrDelimiter),
524 wine_dbgstr_longlong(dwSourceContextCookie), ulStartingLine, dwFlags, pvarResult, pexcepinfo);
526 if(This->thread_id != GetCurrentThreadId() || This->state == SCRIPTSTATE_CLOSED)
529 hres = compile_script(This->ctx, pstrCode, &code);
533 FIXME("executing script not implemented\n");
537 static const IActiveScriptParseVtbl VBScriptParseVtbl = {
538 VBScriptParse_QueryInterface,
539 VBScriptParse_AddRef,
540 VBScriptParse_Release,
541 VBScriptParse_InitNew,
542 VBScriptParse_AddScriptlet,
543 VBScriptParse_ParseScriptText
546 static inline VBScript *impl_from_IObjectSafety(IObjectSafety *iface)
548 return CONTAINING_RECORD(iface, VBScript, IObjectSafety_iface);
551 static HRESULT WINAPI VBScriptSafety_QueryInterface(IObjectSafety *iface, REFIID riid, void **ppv)
553 VBScript *This = impl_from_IObjectSafety(iface);
554 return IActiveScript_QueryInterface(&This->IActiveScript_iface, riid, ppv);
557 static ULONG WINAPI VBScriptSafety_AddRef(IObjectSafety *iface)
559 VBScript *This = impl_from_IObjectSafety(iface);
560 return IActiveScript_AddRef(&This->IActiveScript_iface);
563 static ULONG WINAPI VBScriptSafety_Release(IObjectSafety *iface)
565 VBScript *This = impl_from_IObjectSafety(iface);
566 return IActiveScript_Release(&This->IActiveScript_iface);
569 #define SUPPORTED_OPTIONS (INTERFACESAFE_FOR_UNTRUSTED_DATA|INTERFACE_USES_DISPEX|INTERFACE_USES_SECURITY_MANAGER)
571 static HRESULT WINAPI VBScriptSafety_GetInterfaceSafetyOptions(IObjectSafety *iface, REFIID riid,
572 DWORD *pdwSupportedOptions, DWORD *pdwEnabledOptions)
574 VBScript *This = impl_from_IObjectSafety(iface);
576 TRACE("(%p)->(%s %p %p)\n", This, debugstr_guid(riid), pdwSupportedOptions, pdwEnabledOptions);
578 if(!pdwSupportedOptions || !pdwEnabledOptions)
581 *pdwSupportedOptions = SUPPORTED_OPTIONS;
582 *pdwEnabledOptions = This->safeopt;
586 static HRESULT WINAPI VBScriptSafety_SetInterfaceSafetyOptions(IObjectSafety *iface, REFIID riid,
587 DWORD dwOptionSetMask, DWORD dwEnabledOptions)
589 VBScript *This = impl_from_IObjectSafety(iface);
591 TRACE("(%p)->(%s %x %x)\n", This, debugstr_guid(riid), dwOptionSetMask, dwEnabledOptions);
593 if(dwOptionSetMask & ~SUPPORTED_OPTIONS)
596 This->safeopt = (dwEnabledOptions & dwOptionSetMask) | (This->safeopt & ~dwOptionSetMask) | INTERFACE_USES_DISPEX;
600 static const IObjectSafetyVtbl VBScriptSafetyVtbl = {
601 VBScriptSafety_QueryInterface,
602 VBScriptSafety_AddRef,
603 VBScriptSafety_Release,
604 VBScriptSafety_GetInterfaceSafetyOptions,
605 VBScriptSafety_SetInterfaceSafetyOptions
608 HRESULT WINAPI VBScriptFactory_CreateInstance(IClassFactory *iface, IUnknown *pUnkOuter, REFIID riid, void **ppv)
613 TRACE("(%p %s %p)\n", pUnkOuter, debugstr_guid(riid), ppv);
615 ret = heap_alloc_zero(sizeof(*ret));
617 return E_OUTOFMEMORY;
619 ret->IActiveScript_iface.lpVtbl = &VBScriptVtbl;
620 ret->IActiveScriptParse_iface.lpVtbl = &VBScriptParseVtbl;
621 ret->IObjectSafety_iface.lpVtbl = &VBScriptSafetyVtbl;
624 ret->state = SCRIPTSTATE_UNINITIALIZED;
625 ret->safeopt = INTERFACE_USES_DISPEX;
627 hres = IActiveScript_QueryInterface(&ret->IActiveScript_iface, riid, ppv);
628 IActiveScript_Release(&ret->IActiveScript_iface);