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
27 #include "wine/debug.h"
29 WINE_DEFAULT_DEBUG_CHANNEL(vbscript);
33 #define CTXARG_T DWORDLONG
34 #define IActiveScriptParseVtbl IActiveScriptParse64Vtbl
38 #define CTXARG_T DWORD
39 #define IActiveScriptParseVtbl IActiveScriptParse32Vtbl
44 IActiveScript IActiveScript_iface;
45 IActiveScriptParse IActiveScriptParse_iface;
46 IObjectSafety IObjectSafety_iface;
52 IActiveScriptSite *site;
58 static void change_state(VBScript *This, SCRIPTSTATE state)
60 if(This->state == state)
65 IActiveScriptSite_OnStateChange(This->site, state);
68 static void exec_queued_code(VBScript *This)
73 static HRESULT set_ctx_site(VBScript *This)
77 This->ctx->lcid = This->lcid;
79 hres = init_global(This->ctx);
83 IActiveScriptSite_AddRef(This->site);
84 This->ctx->site = This->site;
86 change_state(This, SCRIPTSTATE_INITIALIZED);
90 static void destroy_script(script_ctx_t *ctx)
92 while(!list_empty(&ctx->named_items)) {
93 named_item_t *iter = LIST_ENTRY(list_head(&ctx->named_items), named_item_t, entry);
95 list_remove(&iter->entry);
97 IDispatch_Release(iter->disp);
98 heap_free(iter->name);
103 IDispatch_Release(ctx->host_global);
105 IActiveScriptSite_Release(ctx->site);
107 IDispatchEx_Release(&ctx->script_obj->IDispatchEx_iface);
111 static void decrease_state(VBScript *This, SCRIPTSTATE state)
113 switch(This->state) {
114 case SCRIPTSTATE_CONNECTED:
115 change_state(This, SCRIPTSTATE_DISCONNECTED);
116 if(state == SCRIPTSTATE_DISCONNECTED)
119 case SCRIPTSTATE_STARTED:
120 case SCRIPTSTATE_DISCONNECTED:
121 if(This->state == SCRIPTSTATE_DISCONNECTED)
122 change_state(This, SCRIPTSTATE_INITIALIZED);
123 if(state == SCRIPTSTATE_INITIALIZED)
126 case SCRIPTSTATE_INITIALIZED:
127 case SCRIPTSTATE_UNINITIALIZED:
128 change_state(This, state);
131 IActiveScriptSite_Release(This->site);
137 if(state == SCRIPTSTATE_CLOSED) {
138 destroy_script(This->ctx);
148 static inline VBScript *impl_from_IActiveScript(IActiveScript *iface)
150 return CONTAINING_RECORD(iface, VBScript, IActiveScript_iface);
153 static HRESULT WINAPI VBScript_QueryInterface(IActiveScript *iface, REFIID riid, void **ppv)
155 VBScript *This = impl_from_IActiveScript(iface);
157 if(IsEqualGUID(riid, &IID_IUnknown)) {
158 TRACE("(%p)->(IID_IUnknown %p)\n", This, ppv);
159 *ppv = &This->IActiveScript_iface;
160 }else if(IsEqualGUID(riid, &IID_IActiveScript)) {
161 TRACE("(%p)->(IID_IActiveScript %p)\n", This, ppv);
162 *ppv = &This->IActiveScript_iface;
163 }else if(IsEqualGUID(riid, &IID_IActiveScriptParse)) {
164 TRACE("(%p)->(IID_IActiveScriptParse %p)\n", This, ppv);
165 *ppv = &This->IActiveScriptParse_iface;
166 }else if(IsEqualGUID(riid, &IID_IObjectSafety)) {
167 TRACE("(%p)->(IID_IObjectSafety %p)\n", This, ppv);
168 *ppv = &This->IObjectSafety_iface;
170 FIXME("(%p)->(%s %p)\n", This, debugstr_guid(riid), ppv);
172 return E_NOINTERFACE;
175 IUnknown_AddRef((IUnknown*)*ppv);
179 static ULONG WINAPI VBScript_AddRef(IActiveScript *iface)
181 VBScript *This = impl_from_IActiveScript(iface);
182 LONG ref = InterlockedIncrement(&This->ref);
184 TRACE("(%p) ref=%d\n", This, ref);
189 static ULONG WINAPI VBScript_Release(IActiveScript *iface)
191 VBScript *This = impl_from_IActiveScript(iface);
192 LONG ref = InterlockedDecrement(&This->ref);
194 TRACE("(%p) ref=%d\n", iface, ref);
198 IActiveScriptSite_Release(This->site);
205 static HRESULT WINAPI VBScript_SetScriptSite(IActiveScript *iface, IActiveScriptSite *pass)
207 VBScript *This = impl_from_IActiveScript(iface);
211 TRACE("(%p)->(%p)\n", This, pass);
219 if(InterlockedCompareExchange(&This->thread_id, GetCurrentThreadId(), 0))
223 IActiveScriptSite_AddRef(This->site);
225 hres = IActiveScriptSite_GetLCID(This->site, &lcid);
229 return This->ctx ? set_ctx_site(This) : S_OK;
232 static HRESULT WINAPI VBScript_GetScriptSite(IActiveScript *iface, REFIID riid,
235 VBScript *This = impl_from_IActiveScript(iface);
236 FIXME("(%p)->()\n", This);
240 static HRESULT WINAPI VBScript_SetScriptState(IActiveScript *iface, SCRIPTSTATE ss)
242 VBScript *This = impl_from_IActiveScript(iface);
244 TRACE("(%p)->(%d)\n", This, ss);
246 if(This->thread_id && GetCurrentThreadId() != This->thread_id)
249 if(ss == SCRIPTSTATE_UNINITIALIZED) {
250 if(This->state == SCRIPTSTATE_CLOSED)
253 decrease_state(This, SCRIPTSTATE_UNINITIALIZED);
261 case SCRIPTSTATE_STARTED:
262 case SCRIPTSTATE_CONNECTED: /* FIXME */
263 if(This->state == SCRIPTSTATE_CLOSED)
266 exec_queued_code(This);
268 case SCRIPTSTATE_INITIALIZED:
269 FIXME("unimplemented SCRIPTSTATE_INITIALIZED\n");
272 FIXME("unimplemented state %d\n", ss);
276 change_state(This, ss);
280 static HRESULT WINAPI VBScript_GetScriptState(IActiveScript *iface, SCRIPTSTATE *pssState)
282 VBScript *This = impl_from_IActiveScript(iface);
284 TRACE("(%p)->(%p)\n", This, pssState);
289 if(This->thread_id && This->thread_id != GetCurrentThreadId())
292 *pssState = This->state;
296 static HRESULT WINAPI VBScript_Close(IActiveScript *iface)
298 VBScript *This = impl_from_IActiveScript(iface);
300 TRACE("(%p)->()\n", This);
302 if(This->thread_id && This->thread_id != GetCurrentThreadId())
305 decrease_state(This, SCRIPTSTATE_CLOSED);
309 static HRESULT WINAPI VBScript_AddNamedItem(IActiveScript *iface, LPCOLESTR pstrName, DWORD dwFlags)
311 VBScript *This = impl_from_IActiveScript(iface);
313 IDispatch *disp = NULL;
316 TRACE("(%p)->(%s %x)\n", This, debugstr_w(pstrName), dwFlags);
318 if(This->thread_id != GetCurrentThreadId() || !This->ctx || This->state == SCRIPTSTATE_CLOSED)
321 if(dwFlags & SCRIPTITEM_GLOBALMEMBERS) {
324 hres = IActiveScriptSite_GetItemInfo(This->site, pstrName, SCRIPTINFO_IUNKNOWN, &unk, NULL);
326 WARN("GetItemInfo failed: %08x\n", hres);
330 hres = IUnknown_QueryInterface(unk, &IID_IDispatch, (void**)&disp);
331 IUnknown_Release(unk);
333 WARN("object does not implement IDispatch\n");
337 if(This->ctx->host_global)
338 IDispatch_Release(This->ctx->host_global);
339 IDispatch_AddRef(disp);
340 This->ctx->host_global = disp;
343 item = heap_alloc(sizeof(*item));
346 IDispatch_Release(disp);
347 return E_OUTOFMEMORY;
351 item->flags = dwFlags;
352 item->name = heap_strdupW(pstrName);
355 IDispatch_Release(disp);
357 return E_OUTOFMEMORY;
360 list_add_tail(&This->ctx->named_items, &item->entry);
364 static HRESULT WINAPI VBScript_AddTypeLib(IActiveScript *iface, REFGUID rguidTypeLib,
365 DWORD dwMajor, DWORD dwMinor, DWORD dwFlags)
367 VBScript *This = impl_from_IActiveScript(iface);
368 FIXME("(%p)->()\n", This);
372 static HRESULT WINAPI VBScript_GetScriptDispatch(IActiveScript *iface, LPCOLESTR pstrItemName, IDispatch **ppdisp)
374 VBScript *This = impl_from_IActiveScript(iface);
376 TRACE("(%p)->(%p)\n", This, ppdisp);
381 if(This->thread_id != GetCurrentThreadId() || !This->ctx->script_obj) {
386 *ppdisp = (IDispatch*)&This->ctx->script_obj->IDispatchEx_iface;
387 IDispatch_AddRef(*ppdisp);
391 static HRESULT WINAPI VBScript_GetCurrentScriptThreadID(IActiveScript *iface,
392 SCRIPTTHREADID *pstridThread)
394 VBScript *This = impl_from_IActiveScript(iface);
395 FIXME("(%p)->()\n", This);
399 static HRESULT WINAPI VBScript_GetScriptThreadID(IActiveScript *iface,
400 DWORD dwWin32ThreadId, SCRIPTTHREADID *pstidThread)
402 VBScript *This = impl_from_IActiveScript(iface);
403 FIXME("(%p)->()\n", This);
407 static HRESULT WINAPI VBScript_GetScriptThreadState(IActiveScript *iface,
408 SCRIPTTHREADID stidThread, SCRIPTTHREADSTATE *pstsState)
410 VBScript *This = impl_from_IActiveScript(iface);
411 FIXME("(%p)->()\n", This);
415 static HRESULT WINAPI VBScript_InterruptScriptThread(IActiveScript *iface,
416 SCRIPTTHREADID stidThread, const EXCEPINFO *pexcepinfo, DWORD dwFlags)
418 VBScript *This = impl_from_IActiveScript(iface);
419 FIXME("(%p)->()\n", This);
423 static HRESULT WINAPI VBScript_Clone(IActiveScript *iface, IActiveScript **ppscript)
425 VBScript *This = impl_from_IActiveScript(iface);
426 FIXME("(%p)->()\n", This);
430 static const IActiveScriptVtbl VBScriptVtbl = {
431 VBScript_QueryInterface,
434 VBScript_SetScriptSite,
435 VBScript_GetScriptSite,
436 VBScript_SetScriptState,
437 VBScript_GetScriptState,
439 VBScript_AddNamedItem,
441 VBScript_GetScriptDispatch,
442 VBScript_GetCurrentScriptThreadID,
443 VBScript_GetScriptThreadID,
444 VBScript_GetScriptThreadState,
445 VBScript_InterruptScriptThread,
449 static inline VBScript *impl_from_IActiveScriptParse(IActiveScriptParse *iface)
451 return CONTAINING_RECORD(iface, VBScript, IActiveScriptParse_iface);
454 static HRESULT WINAPI VBScriptParse_QueryInterface(IActiveScriptParse *iface, REFIID riid, void **ppv)
456 VBScript *This = impl_from_IActiveScriptParse(iface);
457 return IActiveScript_QueryInterface(&This->IActiveScript_iface, riid, ppv);
460 static ULONG WINAPI VBScriptParse_AddRef(IActiveScriptParse *iface)
462 VBScript *This = impl_from_IActiveScriptParse(iface);
463 return IActiveScript_AddRef(&This->IActiveScript_iface);
466 static ULONG WINAPI VBScriptParse_Release(IActiveScriptParse *iface)
468 VBScript *This = impl_from_IActiveScriptParse(iface);
469 return IActiveScript_Release(&This->IActiveScript_iface);
472 static HRESULT WINAPI VBScriptParse_InitNew(IActiveScriptParse *iface)
474 VBScript *This = impl_from_IActiveScriptParse(iface);
475 script_ctx_t *ctx, *old_ctx;
477 TRACE("(%p)\n", This);
482 ctx = heap_alloc_zero(sizeof(script_ctx_t));
484 return E_OUTOFMEMORY;
486 list_init(&ctx->named_items);
488 old_ctx = InterlockedCompareExchangePointer((void**)&This->ctx, ctx, NULL);
494 return This->site ? set_ctx_site(This) : S_OK;
497 static HRESULT WINAPI VBScriptParse_AddScriptlet(IActiveScriptParse *iface,
498 LPCOLESTR pstrDefaultName, LPCOLESTR pstrCode, LPCOLESTR pstrItemName,
499 LPCOLESTR pstrSubItemName, LPCOLESTR pstrEventName, LPCOLESTR pstrDelimiter,
500 CTXARG_T dwSourceContextCookie, ULONG ulStartingLineNumber, DWORD dwFlags,
501 BSTR *pbstrName, EXCEPINFO *pexcepinfo)
503 VBScript *This = impl_from_IActiveScriptParse(iface);
504 FIXME("(%p)->(%s %s %s %s %s %s %s %u %x %p %p)\n", This, debugstr_w(pstrDefaultName),
505 debugstr_w(pstrCode), debugstr_w(pstrItemName), debugstr_w(pstrSubItemName),
506 debugstr_w(pstrEventName), debugstr_w(pstrDelimiter), wine_dbgstr_longlong(dwSourceContextCookie),
507 ulStartingLineNumber, dwFlags, pbstrName, pexcepinfo);
511 static HRESULT WINAPI VBScriptParse_ParseScriptText(IActiveScriptParse *iface,
512 LPCOLESTR pstrCode, LPCOLESTR pstrItemName, IUnknown *punkContext,
513 LPCOLESTR pstrDelimiter, CTXARG_T dwSourceContextCookie, ULONG ulStartingLine,
514 DWORD dwFlags, VARIANT *pvarResult, EXCEPINFO *pexcepinfo)
516 VBScript *This = impl_from_IActiveScriptParse(iface);
520 TRACE("(%p)->(%s %s %p %s %s %u %x %p %p)\n", This, debugstr_w(pstrCode),
521 debugstr_w(pstrItemName), punkContext, debugstr_w(pstrDelimiter),
522 wine_dbgstr_longlong(dwSourceContextCookie), ulStartingLine, dwFlags, pvarResult, pexcepinfo);
524 hres = parse_script(&parser, pstrCode);
528 FIXME("compiling script not implemented\n");
532 static const IActiveScriptParseVtbl VBScriptParseVtbl = {
533 VBScriptParse_QueryInterface,
534 VBScriptParse_AddRef,
535 VBScriptParse_Release,
536 VBScriptParse_InitNew,
537 VBScriptParse_AddScriptlet,
538 VBScriptParse_ParseScriptText
541 static inline VBScript *impl_from_IObjectSafety(IObjectSafety *iface)
543 return CONTAINING_RECORD(iface, VBScript, IObjectSafety_iface);
546 static HRESULT WINAPI VBScriptSafety_QueryInterface(IObjectSafety *iface, REFIID riid, void **ppv)
548 VBScript *This = impl_from_IObjectSafety(iface);
549 return IActiveScript_QueryInterface(&This->IActiveScript_iface, riid, ppv);
552 static ULONG WINAPI VBScriptSafety_AddRef(IObjectSafety *iface)
554 VBScript *This = impl_from_IObjectSafety(iface);
555 return IActiveScript_AddRef(&This->IActiveScript_iface);
558 static ULONG WINAPI VBScriptSafety_Release(IObjectSafety *iface)
560 VBScript *This = impl_from_IObjectSafety(iface);
561 return IActiveScript_Release(&This->IActiveScript_iface);
564 #define SUPPORTED_OPTIONS (INTERFACESAFE_FOR_UNTRUSTED_DATA|INTERFACE_USES_DISPEX|INTERFACE_USES_SECURITY_MANAGER)
566 static HRESULT WINAPI VBScriptSafety_GetInterfaceSafetyOptions(IObjectSafety *iface, REFIID riid,
567 DWORD *pdwSupportedOptions, DWORD *pdwEnabledOptions)
569 VBScript *This = impl_from_IObjectSafety(iface);
571 TRACE("(%p)->(%s %p %p)\n", This, debugstr_guid(riid), pdwSupportedOptions, pdwEnabledOptions);
573 if(!pdwSupportedOptions || !pdwEnabledOptions)
576 *pdwSupportedOptions = SUPPORTED_OPTIONS;
577 *pdwEnabledOptions = This->safeopt;
581 static HRESULT WINAPI VBScriptSafety_SetInterfaceSafetyOptions(IObjectSafety *iface, REFIID riid,
582 DWORD dwOptionSetMask, DWORD dwEnabledOptions)
584 VBScript *This = impl_from_IObjectSafety(iface);
586 TRACE("(%p)->(%s %x %x)\n", This, debugstr_guid(riid), dwOptionSetMask, dwEnabledOptions);
588 if(dwOptionSetMask & ~SUPPORTED_OPTIONS)
591 This->safeopt = (dwEnabledOptions & dwOptionSetMask) | (This->safeopt & ~dwOptionSetMask) | INTERFACE_USES_DISPEX;
595 static const IObjectSafetyVtbl VBScriptSafetyVtbl = {
596 VBScriptSafety_QueryInterface,
597 VBScriptSafety_AddRef,
598 VBScriptSafety_Release,
599 VBScriptSafety_GetInterfaceSafetyOptions,
600 VBScriptSafety_SetInterfaceSafetyOptions
603 HRESULT WINAPI VBScriptFactory_CreateInstance(IClassFactory *iface, IUnknown *pUnkOuter, REFIID riid, void **ppv)
608 TRACE("(%p %s %p)\n", pUnkOuter, debugstr_guid(riid), ppv);
610 ret = heap_alloc_zero(sizeof(*ret));
612 return E_OUTOFMEMORY;
614 ret->IActiveScript_iface.lpVtbl = &VBScriptVtbl;
615 ret->IActiveScriptParse_iface.lpVtbl = &VBScriptParseVtbl;
616 ret->IObjectSafety_iface.lpVtbl = &VBScriptSafetyVtbl;
619 ret->state = SCRIPTSTATE_UNINITIALIZED;
620 ret->safeopt = INTERFACE_USES_DISPEX;
622 hres = IActiveScript_QueryInterface(&ret->IActiveScript_iface, riid, ppv);
623 IActiveScript_Release(&ret->IActiveScript_iface);