1 /*
2 * Copyright 2011 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
20 #include <assert.h>
21
22 #include "vbscript.h"
23 #include "objsafe.h"
24
25 #include "wine/debug.h"
26
27 WINE_DEFAULT_DEBUG_CHANNEL(vbscript);
28
29 #ifdef _WIN64
30
31 #define CTXARG_T DWORDLONG
32 #define IActiveScriptParseVtbl IActiveScriptParse64Vtbl
33
34 #else
35
36 #define CTXARG_T DWORD
37 #define IActiveScriptParseVtbl IActiveScriptParse32Vtbl
38
39 #endif
40
41 struct VBScript {
42 IActiveScript IActiveScript_iface;
43 IActiveScriptParse IActiveScriptParse_iface;
44 IObjectSafety IObjectSafety_iface;
45
46 LONG ref;
47
48 DWORD safeopt;
49 SCRIPTSTATE state;
50 IActiveScriptSite *site;
51 script_ctx_t *ctx;
52 LONG thread_id;
53 LCID lcid;
54 };
55
56 static void change_state(VBScript *This, SCRIPTSTATE state)
57 {
58 if(This->state == state)
59 return;
60
61 This->state = state;
62 if(This->site)
63 IActiveScriptSite_OnStateChange(This->site, state);
64 }
65
66 static inline BOOL is_started(VBScript *This)
67 {
68 return This->state == SCRIPTSTATE_STARTED
69 || This->state == SCRIPTSTATE_CONNECTED
70 || This->state == SCRIPTSTATE_DISCONNECTED;
71 }
72
73 static HRESULT exec_global_code(script_ctx_t *ctx, vbscode_t *code)
74 {
75 HRESULT hres;
76
77 code->global_executed = TRUE;
78
79 IActiveScriptSite_OnEnterScript(ctx->site);
80 hres = exec_script(ctx, &code->global_code, NULL, NULL, NULL);
81 IActiveScriptSite_OnLeaveScript(ctx->site);
82
83 return hres;
84 }
85
86 static void exec_queued_code(script_ctx_t *ctx)
87 {
88 vbscode_t *iter;
89
90 LIST_FOR_EACH_ENTRY(iter, &ctx->code_list, vbscode_t, entry) {
91 if(!iter->global_executed)
92 exec_global_code(ctx, iter);
93 }
94 }
95
96 static HRESULT set_ctx_site(VBScript *This)
97 {
98 HRESULT hres;
99
100 This->ctx->lcid = This->lcid;
101
102 hres = init_global(This->ctx);
103 if(FAILED(hres))
104 return hres;
105
106 IActiveScriptSite_AddRef(This->site);
107 This->ctx->site = This->site;
108
109 change_state(This, SCRIPTSTATE_INITIALIZED);
110 return S_OK;
111 }
112
113 static void release_script(script_ctx_t *ctx)
114 {
115 collect_objects(ctx);
116
117 release_dynamic_vars(ctx->global_vars);
118 ctx->global_vars = NULL;
119
120 while(!list_empty(&ctx->named_items)) {
121 named_item_t *iter = LIST_ENTRY(list_head(&ctx->named_items), named_item_t, entry);
122
123 list_remove(&iter->entry);
124 if(iter->disp)
125 IDispatch_Release(iter->disp);
126 heap_free(iter->name);
127 heap_free(iter);
128 }
129
130 if(ctx->host_global) {
131 IDispatch_Release(ctx->host_global);
132 ctx->host_global = NULL;
133 }
134
135 if(ctx->secmgr) {
136 IInternetHostSecurityManager_Release(ctx->secmgr);
137 ctx->secmgr = NULL;
138 }
139
140 if(ctx->site) {
141 IActiveScriptSite_Release(ctx->site);
142 ctx->site = NULL;
143 }
144
145 if(ctx->err_obj) {
146 IDispatchEx_Release(&ctx->err_obj->IDispatchEx_iface);
147 ctx->err_obj = NULL;
148 }
149
150 if(ctx->global_obj) {
151 IDispatchEx_Release(&ctx->global_obj->IDispatchEx_iface);
152 ctx->global_obj = NULL;
153 }
154
155 if(ctx->script_obj) {
156 IDispatchEx_Release(&ctx->script_obj->IDispatchEx_iface);
157 ctx->script_obj = NULL;
158 }
159
160 vbsheap_free(&ctx->heap);
161 vbsheap_init(&ctx->heap);
162 }
163
164 static void destroy_script(script_ctx_t *ctx)
165 {
166 while(!list_empty(&ctx->code_list))
167 release_vbscode(LIST_ENTRY(list_head(&ctx->code_list), vbscode_t, entry));
168
169 release_script(ctx);
170 heap_free(ctx);
171 }
172
173 static void decrease_state(VBScript *This, SCRIPTSTATE state)
174 {
175 switch(This->state) {
176 case SCRIPTSTATE_CONNECTED:
177 change_state(This, SCRIPTSTATE_DISCONNECTED);
178 if(state == SCRIPTSTATE_DISCONNECTED)
179 return;
180 /* FALLTHROUGH */
181 case SCRIPTSTATE_STARTED:
182 case SCRIPTSTATE_DISCONNECTED:
183 if(This->state == SCRIPTSTATE_DISCONNECTED)
184 change_state(This, SCRIPTSTATE_INITIALIZED);
185 if(state == SCRIPTSTATE_INITIALIZED)
186 break;
187 /* FALLTHROUGH */
188 case SCRIPTSTATE_INITIALIZED:
189 case SCRIPTSTATE_UNINITIALIZED:
190 change_state(This, state);
191
192 if(This->site) {
193 IActiveScriptSite_Release(This->site);
194 This->site = NULL;
195 }
196
197 if(This->ctx)
198 release_script(This->ctx);
199
200 This->thread_id = 0;
201 break;
202 case SCRIPTSTATE_CLOSED:
203 break;
204 default:
205 assert(0);
206 }
207 }
208
209 static inline VBScript *impl_from_IActiveScript(IActiveScript *iface)
210 {
211 return CONTAINING_RECORD(iface, VBScript, IActiveScript_iface);
212 }
213
214 static HRESULT WINAPI VBScript_QueryInterface(IActiveScript *iface, REFIID riid, void **ppv)
215 {
216 VBScript *This = impl_from_IActiveScript(iface);
217
218 if(IsEqualGUID(riid, &IID_IUnknown)) {
219 TRACE("(%p)->(IID_IUnknown %p)\n", This, ppv);
220 *ppv = &This->IActiveScript_iface;
221 }else if(IsEqualGUID(riid, &IID_IActiveScript)) {
222 TRACE("(%p)->(IID_IActiveScript %p)\n", This, ppv);
223 *ppv = &This->IActiveScript_iface;
224 }else if(IsEqualGUID(riid, &IID_IActiveScriptParse)) {
225 TRACE("(%p)->(IID_IActiveScriptParse %p)\n", This, ppv);
226 *ppv = &This->IActiveScriptParse_iface;
227 }else if(IsEqualGUID(riid, &IID_IObjectSafety)) {
228 TRACE("(%p)->(IID_IObjectSafety %p)\n", This, ppv);
229 *ppv = &This->IObjectSafety_iface;
230 }else {
231 FIXME("(%p)->(%s %p)\n", This, debugstr_guid(riid), ppv);
232 *ppv = NULL;
233 return E_NOINTERFACE;
234 }
235
236 IUnknown_AddRef((IUnknown*)*ppv);
237 return S_OK;
238 }
239
240 static ULONG WINAPI VBScript_AddRef(IActiveScript *iface)
241 {
242 VBScript *This = impl_from_IActiveScript(iface);
243 LONG ref = InterlockedIncrement(&This->ref);
244
245 TRACE("(%p) ref=%d\n", This, ref);
246
247 return ref;
248 }
249
250 static ULONG WINAPI VBScript_Release(IActiveScript *iface)
251 {
252 VBScript *This = impl_from_IActiveScript(iface);
253 LONG ref = InterlockedDecrement(&This->ref);
254
255 TRACE("(%p) ref=%d\n", iface, ref);
256
257 if(!ref) {
258 if(This->ctx) {
259 decrease_state(This, SCRIPTSTATE_CLOSED);
260 destroy_script(This->ctx);
261 This->ctx = NULL;
262 }
263 if(This->site)
264 IActiveScriptSite_Release(This->site);
265 heap_free(This);
266 }
267
268 return ref;
269 }
270
271 static HRESULT WINAPI VBScript_SetScriptSite(IActiveScript *iface, IActiveScriptSite *pass)
272 {
273 VBScript *This = impl_from_IActiveScript(iface);
274 LCID lcid;
275 HRESULT hres;
276
277 TRACE("(%p)->(%p)\n", This, pass);
278
279 if(!pass)
280 return E_POINTER;
281
282 if(This->site)
283 return E_UNEXPECTED;
284
285 if(InterlockedCompareExchange(&This->thread_id, GetCurrentThreadId(), 0))
286 return E_UNEXPECTED;
287
288 This->site = pass;
289 IActiveScriptSite_AddRef(This->site);
290
291 hres = IActiveScriptSite_GetLCID(This->site, &lcid);
292 if(hres == S_OK)
293 This->lcid = lcid;
294
295 return This->ctx ? set_ctx_site(This) : S_OK;
296 }
297
298 static HRESULT WINAPI VBScript_GetScriptSite(IActiveScript *iface, REFIID riid,
299 void **ppvObject)
300 {
301 VBScript *This = impl_from_IActiveScript(iface);
302 FIXME("(%p)->()\n", This);
303 return E_NOTIMPL;
304 }
305
306 static HRESULT WINAPI VBScript_SetScriptState(IActiveScript *iface, SCRIPTSTATE ss)
307 {
308 VBScript *This = impl_from_IActiveScript(iface);
309
310 TRACE("(%p)->(%d)\n", This, ss);
311
312 if(This->thread_id && GetCurrentThreadId() != This->thread_id)
313 return E_UNEXPECTED;
314
315 if(ss == SCRIPTSTATE_UNINITIALIZED) {
316 if(This->state == SCRIPTSTATE_CLOSED)
317 return E_UNEXPECTED;
318
319 decrease_state(This, SCRIPTSTATE_UNINITIALIZED);
320 return S_OK;
321 }
322
323 if(!This->ctx)
324 return E_UNEXPECTED;
325
326 switch(ss) {
327 case SCRIPTSTATE_STARTED:
328 case SCRIPTSTATE_CONNECTED: /* FIXME */
329 if(This->state == SCRIPTSTATE_CLOSED)
330 return E_UNEXPECTED;
331
332 exec_queued_code(This->ctx);
333 break;
334 case SCRIPTSTATE_INITIALIZED:
335 FIXME("unimplemented SCRIPTSTATE_INITIALIZED\n");
336 return S_OK;
337 default:
338 FIXME("unimplemented state %d\n", ss);
339 return E_NOTIMPL;
340 }
341
342 change_state(This, ss);
343 return S_OK;
344 }
345
346 static HRESULT WINAPI VBScript_GetScriptState(IActiveScript *iface, SCRIPTSTATE *pssState)
347 {
348 VBScript *This = impl_from_IActiveScript(iface);
349
350 TRACE("(%p)->(%p)\n", This, pssState);
351
352 if(!pssState)
353 return E_POINTER;
354
355 if(This->thread_id && This->thread_id != GetCurrentThreadId())
356 return E_UNEXPECTED;
357
358 *pssState = This->state;
359 return S_OK;
360 }
361
362 static HRESULT WINAPI VBScript_Close(IActiveScript *iface)
363 {
364 VBScript *This = impl_from_IActiveScript(iface);
365
366 TRACE("(%p)->()\n", This);
367
368 if(This->thread_id && This->thread_id != GetCurrentThreadId())
369 return E_UNEXPECTED;
370
371 decrease_state(This, SCRIPTSTATE_CLOSED);
372 return S_OK;
373 }
374
375 static HRESULT WINAPI VBScript_AddNamedItem(IActiveScript *iface, LPCOLESTR pstrName, DWORD dwFlags)
376 {
377 VBScript *This = impl_from_IActiveScript(iface);
378 named_item_t *item;
379 IDispatch *disp = NULL;
380 HRESULT hres;
381
382 TRACE("(%p)->(%s %x)\n", This, debugstr_w(pstrName), dwFlags);
383
384 if(This->thread_id != GetCurrentThreadId() || !This->ctx || This->state == SCRIPTSTATE_CLOSED)
385 return E_UNEXPECTED;
386
387 if(dwFlags & SCRIPTITEM_GLOBALMEMBERS) {
388 IUnknown *unk;
389
390 hres = IActiveScriptSite_GetItemInfo(This->site, pstrName, SCRIPTINFO_IUNKNOWN, &unk, NULL);
391 if(FAILED(hres)) {
392 WARN("GetItemInfo failed: %08x\n", hres);
393 return hres;
394 }
395
396 hres = IUnknown_QueryInterface(unk, &IID_IDispatch, (void**)&disp);
397 IUnknown_Release(unk);
398 if(FAILED(hres)) {
399 WARN("object does not implement IDispatch\n");
400 return hres;
401 }
402
403 if(This->ctx->host_global)
404 IDispatch_Release(This->ctx->host_global);
405 IDispatch_AddRef(disp);
406 This->ctx->host_global = disp;
407 }
408
409 item = heap_alloc(sizeof(*item));
410 if(!item) {
411 if(disp)
412 IDispatch_Release(disp);
413 return E_OUTOFMEMORY;
414 }
415
416 item->disp = disp;
417 item->flags = dwFlags;
418 item->name = heap_strdupW(pstrName);
419 if(!item->name) {
420 if(disp)
421 IDispatch_Release(disp);
422 heap_free(item);
423 return E_OUTOFMEMORY;
424 }
425
426 list_add_tail(&This->ctx->named_items, &item->entry);
427 return S_OK;
428 }
429
430 static HRESULT WINAPI VBScript_AddTypeLib(IActiveScript *iface, REFGUID rguidTypeLib,
431 DWORD dwMajor, DWORD dwMinor, DWORD dwFlags)
432 {
433 VBScript *This = impl_from_IActiveScript(iface);
434 FIXME("(%p)->()\n", This);
435 return E_NOTIMPL;
436 }
437
438 static HRESULT WINAPI VBScript_GetScriptDispatch(IActiveScript *iface, LPCOLESTR pstrItemName, IDispatch **ppdisp)
439 {
440 VBScript *This = impl_from_IActiveScript(iface);
441
442 TRACE("(%p)->(%p)\n", This, ppdisp);
443
444 if(!ppdisp)
445 return E_POINTER;
446
447 if(This->thread_id != GetCurrentThreadId() || !This->ctx || !This->ctx->script_obj) {
448 *ppdisp = NULL;
449 return E_UNEXPECTED;
450 }
451
452 *ppdisp = (IDispatch*)&This->ctx->script_obj->IDispatchEx_iface;
453 IDispatch_AddRef(*ppdisp);
454 return S_OK;
455 }
456
457 static HRESULT WINAPI VBScript_GetCurrentScriptThreadID(IActiveScript *iface,
458 SCRIPTTHREADID *pstridThread)
459 {
460 VBScript *This = impl_from_IActiveScript(iface);
461 FIXME("(%p)->()\n", This);
462 return E_NOTIMPL;
463 }
464
465 static HRESULT WINAPI VBScript_GetScriptThreadID(IActiveScript *iface,
466 DWORD dwWin32ThreadId, SCRIPTTHREADID *pstidThread)
467 {
468 VBScript *This = impl_from_IActiveScript(iface);
469 FIXME("(%p)->()\n", This);
470 return E_NOTIMPL;
471 }
472
473 static HRESULT WINAPI VBScript_GetScriptThreadState(IActiveScript *iface,
474 SCRIPTTHREADID stidThread, SCRIPTTHREADSTATE *pstsState)
475 {
476 VBScript *This = impl_from_IActiveScript(iface);
477 FIXME("(%p)->()\n", This);
478 return E_NOTIMPL;
479 }
480
481 static HRESULT WINAPI VBScript_InterruptScriptThread(IActiveScript *iface,
482 SCRIPTTHREADID stidThread, const EXCEPINFO *pexcepinfo, DWORD dwFlags)
483 {
484 VBScript *This = impl_from_IActiveScript(iface);
485 FIXME("(%p)->()\n", This);
486 return E_NOTIMPL;
487 }
488
489 static HRESULT WINAPI VBScript_Clone(IActiveScript *iface, IActiveScript **ppscript)
490 {
491 VBScript *This = impl_from_IActiveScript(iface);
492 FIXME("(%p)->()\n", This);
493 return E_NOTIMPL;
494 }
495
496 static const IActiveScriptVtbl VBScriptVtbl = {
497 VBScript_QueryInterface,
498 VBScript_AddRef,
499 VBScript_Release,
500 VBScript_SetScriptSite,
501 VBScript_GetScriptSite,
502 VBScript_SetScriptState,
503 VBScript_GetScriptState,
504 VBScript_Close,
505 VBScript_AddNamedItem,
506 VBScript_AddTypeLib,
507 VBScript_GetScriptDispatch,
508 VBScript_GetCurrentScriptThreadID,
509 VBScript_GetScriptThreadID,
510 VBScript_GetScriptThreadState,
511 VBScript_InterruptScriptThread,
512 VBScript_Clone
513 };
514
515 static inline VBScript *impl_from_IActiveScriptParse(IActiveScriptParse *iface)
516 {
517 return CONTAINING_RECORD(iface, VBScript, IActiveScriptParse_iface);
518 }
519
520 static HRESULT WINAPI VBScriptParse_QueryInterface(IActiveScriptParse *iface, REFIID riid, void **ppv)
521 {
522 VBScript *This = impl_from_IActiveScriptParse(iface);
523 return IActiveScript_QueryInterface(&This->IActiveScript_iface, riid, ppv);
524 }
525
526 static ULONG WINAPI VBScriptParse_AddRef(IActiveScriptParse *iface)
527 {
528 VBScript *This = impl_from_IActiveScriptParse(iface);
529 return IActiveScript_AddRef(&This->IActiveScript_iface);
530 }
531
532 static ULONG WINAPI VBScriptParse_Release(IActiveScriptParse *iface)
533 {
534 VBScript *This = impl_from_IActiveScriptParse(iface);
535 return IActiveScript_Release(&This->IActiveScript_iface);
536 }
537
538 static HRESULT WINAPI VBScriptParse_InitNew(IActiveScriptParse *iface)
539 {
540 VBScript *This = impl_from_IActiveScriptParse(iface);
541 script_ctx_t *ctx, *old_ctx;
542
543 TRACE("(%p)\n", This);
544
545 if(This->ctx)
546 return E_UNEXPECTED;
547
548 ctx = heap_alloc_zero(sizeof(script_ctx_t));
549 if(!ctx)
550 return E_OUTOFMEMORY;
551
552 ctx->safeopt = This->safeopt;
553 vbsheap_init(&ctx->heap);
554 list_init(&ctx->objects);
555 list_init(&ctx->code_list);
556 list_init(&ctx->named_items);
557
558 old_ctx = InterlockedCompareExchangePointer((void**)&This->ctx, ctx, NULL);
559 if(old_ctx) {
560 destroy_script(ctx);
561 return E_UNEXPECTED;
562 }
563
564 return This->site ? set_ctx_site(This) : S_OK;
565 }
566
567 static HRESULT WINAPI VBScriptParse_AddScriptlet(IActiveScriptParse *iface,
568 LPCOLESTR pstrDefaultName, LPCOLESTR pstrCode, LPCOLESTR pstrItemName,
569 LPCOLESTR pstrSubItemName, LPCOLESTR pstrEventName, LPCOLESTR pstrDelimiter,
570 CTXARG_T dwSourceContextCookie, ULONG ulStartingLineNumber, DWORD dwFlags,
571 BSTR *pbstrName, EXCEPINFO *pexcepinfo)
572 {
573 VBScript *This = impl_from_IActiveScriptParse(iface);
574 FIXME("(%p)->(%s %s %s %s %s %s %s %u %x %p %p)\n", This, debugstr_w(pstrDefaultName),
575 debugstr_w(pstrCode), debugstr_w(pstrItemName), debugstr_w(pstrSubItemName),
576 debugstr_w(pstrEventName), debugstr_w(pstrDelimiter), wine_dbgstr_longlong(dwSourceContextCookie),
577 ulStartingLineNumber, dwFlags, pbstrName, pexcepinfo);
578 return E_NOTIMPL;
579 }
580
581 static HRESULT WINAPI VBScriptParse_ParseScriptText(IActiveScriptParse *iface,
582 LPCOLESTR pstrCode, LPCOLESTR pstrItemName, IUnknown *punkContext,
583 LPCOLESTR pstrDelimiter, CTXARG_T dwSourceContextCookie, ULONG ulStartingLine,
584 DWORD dwFlags, VARIANT *pvarResult, EXCEPINFO *pexcepinfo)
585 {
586 VBScript *This = impl_from_IActiveScriptParse(iface);
587 vbscode_t *code;
588 HRESULT hres;
589
590 TRACE("(%p)->(%s %s %p %s %s %u %x %p %p)\n", This, debugstr_w(pstrCode),
591 debugstr_w(pstrItemName), punkContext, debugstr_w(pstrDelimiter),
592 wine_dbgstr_longlong(dwSourceContextCookie), ulStartingLine, dwFlags, pvarResult, pexcepinfo);
593
594 if(This->thread_id != GetCurrentThreadId() || This->state == SCRIPTSTATE_CLOSED)
595 return E_UNEXPECTED;
596
597 hres = compile_script(This->ctx, pstrCode, &code);
598 if(FAILED(hres))
599 return hres;
600
601 return is_started(This) ? exec_global_code(This->ctx, code) : S_OK;
602 }
603
604 static const IActiveScriptParseVtbl VBScriptParseVtbl = {
605 VBScriptParse_QueryInterface,
606 VBScriptParse_AddRef,
607 VBScriptParse_Release,
608 VBScriptParse_InitNew,
609 VBScriptParse_AddScriptlet,
610 VBScriptParse_ParseScriptText
611 };
612
613 static inline VBScript *impl_from_IObjectSafety(IObjectSafety *iface)
614 {
615 return CONTAINING_RECORD(iface, VBScript, IObjectSafety_iface);
616 }
617
618 static HRESULT WINAPI VBScriptSafety_QueryInterface(IObjectSafety *iface, REFIID riid, void **ppv)
619 {
620 VBScript *This = impl_from_IObjectSafety(iface);
621 return IActiveScript_QueryInterface(&This->IActiveScript_iface, riid, ppv);
622 }
623
624 static ULONG WINAPI VBScriptSafety_AddRef(IObjectSafety *iface)
625 {
626 VBScript *This = impl_from_IObjectSafety(iface);
627 return IActiveScript_AddRef(&This->IActiveScript_iface);
628 }
629
630 static ULONG WINAPI VBScriptSafety_Release(IObjectSafety *iface)
631 {
632 VBScript *This = impl_from_IObjectSafety(iface);
633 return IActiveScript_Release(&This->IActiveScript_iface);
634 }
635
636 #define SUPPORTED_OPTIONS (INTERFACESAFE_FOR_UNTRUSTED_DATA|INTERFACE_USES_DISPEX|INTERFACE_USES_SECURITY_MANAGER)
637
638 static HRESULT WINAPI VBScriptSafety_GetInterfaceSafetyOptions(IObjectSafety *iface, REFIID riid,
639 DWORD *pdwSupportedOptions, DWORD *pdwEnabledOptions)
640 {
641 VBScript *This = impl_from_IObjectSafety(iface);
642
643 TRACE("(%p)->(%s %p %p)\n", This, debugstr_guid(riid), pdwSupportedOptions, pdwEnabledOptions);
644
645 if(!pdwSupportedOptions || !pdwEnabledOptions)
646 return E_POINTER;
647
648 *pdwSupportedOptions = SUPPORTED_OPTIONS;
649 *pdwEnabledOptions = This->safeopt;
650 return S_OK;
651 }
652
653 static HRESULT WINAPI VBScriptSafety_SetInterfaceSafetyOptions(IObjectSafety *iface, REFIID riid,
654 DWORD dwOptionSetMask, DWORD dwEnabledOptions)
655 {
656 VBScript *This = impl_from_IObjectSafety(iface);
657
658 TRACE("(%p)->(%s %x %x)\n", This, debugstr_guid(riid), dwOptionSetMask, dwEnabledOptions);
659
660 if(dwOptionSetMask & ~SUPPORTED_OPTIONS)
661 return E_FAIL;
662
663 This->safeopt = (dwEnabledOptions & dwOptionSetMask) | (This->safeopt & ~dwOptionSetMask) | INTERFACE_USES_DISPEX;
664 return S_OK;
665 }
666
667 static const IObjectSafetyVtbl VBScriptSafetyVtbl = {
668 VBScriptSafety_QueryInterface,
669 VBScriptSafety_AddRef,
670 VBScriptSafety_Release,
671 VBScriptSafety_GetInterfaceSafetyOptions,
672 VBScriptSafety_SetInterfaceSafetyOptions
673 };
674
675 HRESULT WINAPI VBScriptFactory_CreateInstance(IClassFactory *iface, IUnknown *pUnkOuter, REFIID riid, void **ppv)
676 {
677 VBScript *ret;
678 HRESULT hres;
679
680 TRACE("(%p %s %p)\n", pUnkOuter, debugstr_guid(riid), ppv);
681
682 ret = heap_alloc_zero(sizeof(*ret));
683 if(!ret)
684 return E_OUTOFMEMORY;
685
686 ret->IActiveScript_iface.lpVtbl = &VBScriptVtbl;
687 ret->IActiveScriptParse_iface.lpVtbl = &VBScriptParseVtbl;
688 ret->IObjectSafety_iface.lpVtbl = &VBScriptSafetyVtbl;
689
690 ret->ref = 1;
691 ret->state = SCRIPTSTATE_UNINITIALIZED;
692 ret->safeopt = INTERFACE_USES_DISPEX;
693
694 hres = IActiveScript_QueryInterface(&ret->IActiveScript_iface, riid, ppv);
695 IActiveScript_Release(&ret->IActiveScript_iface);
696 return hres;
697 }
698
699 typedef struct {
700 IServiceProvider IServiceProvider_iface;
701
702 LONG ref;
703
704 IServiceProvider *sp;
705 } AXSite;
706
707 static inline AXSite *impl_from_IServiceProvider(IServiceProvider *iface)
708 {
709 return CONTAINING_RECORD(iface, AXSite, IServiceProvider_iface);
710 }
711
712 static HRESULT WINAPI AXSite_QueryInterface(IServiceProvider *iface, REFIID riid, void **ppv)
713 {
714 AXSite *This = impl_from_IServiceProvider(iface);
715
716 if(IsEqualGUID(&IID_IUnknown, riid)) {
717 TRACE("(%p)->(IID_IUnknown %p)\n", This, ppv);
718 *ppv = &This->IServiceProvider_iface;
719 }else if(IsEqualGUID(&IID_IServiceProvider, riid)) {
720 TRACE("(%p)->(IID_IServiceProvider %p)\n", This, ppv);
721 *ppv = &This->IServiceProvider_iface;
722 }else {
723 TRACE("(%p)->(%s %p)\n", This, debugstr_guid(riid), ppv);
724 *ppv = NULL;
725 return E_NOINTERFACE;
726 }
727
728 IUnknown_AddRef((IUnknown*)*ppv);
729 return S_OK;
730 }
731
732 static ULONG WINAPI AXSite_AddRef(IServiceProvider *iface)
733 {
734 AXSite *This = impl_from_IServiceProvider(iface);
735 LONG ref = InterlockedIncrement(&This->ref);
736
737 TRACE("(%p) ref=%d\n", This, ref);
738
739 return ref;
740 }
741
742 static ULONG WINAPI AXSite_Release(IServiceProvider *iface)
743 {
744 AXSite *This = impl_from_IServiceProvider(iface);
745 LONG ref = InterlockedDecrement(&This->ref);
746
747 TRACE("(%p) ref=%d\n", This, ref);
748
749 if(!ref)
750 heap_free(This);
751
752 return ref;
753 }
754
755 static HRESULT WINAPI AXSite_QueryService(IServiceProvider *iface,
756 REFGUID guidService, REFIID riid, void **ppv)
757 {
758 AXSite *This = impl_from_IServiceProvider(iface);
759
760 TRACE("(%p)->(%s %s %p)\n", This, debugstr_guid(guidService), debugstr_guid(riid), ppv);
761
762 return IServiceProvider_QueryService(This->sp, guidService, riid, ppv);
763 }
764
765 static IServiceProviderVtbl AXSiteVtbl = {
766 AXSite_QueryInterface,
767 AXSite_AddRef,
768 AXSite_Release,
769 AXSite_QueryService
770 };
771
772 IUnknown *create_ax_site(script_ctx_t *ctx)
773 {
774 IServiceProvider *sp;
775 AXSite *ret;
776 HRESULT hres;
777
778 hres = IActiveScriptSite_QueryInterface(ctx->site, &IID_IServiceProvider, (void**)&sp);
779 if(FAILED(hres)) {
780 ERR("Could not get IServiceProvider iface: %08x\n", hres);
781 return NULL;
782 }
783
784 ret = heap_alloc(sizeof(*ret));
785 if(!ret) {
786 IServiceProvider_Release(sp);
787 return NULL;
788 }
789
790 ret->IServiceProvider_iface.lpVtbl = &AXSiteVtbl;
791 ret->ref = 1;
792 ret->sp = sp;
793
794 return (IUnknown*)&ret->IServiceProvider_iface;
795 }
796
This page was automatically generated by the
LXR engine.
Visit the LXR main site for more
information.