dlls/vbscript/vbdisp.c

   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 #include <assert.h>
  20
  21 #include "vbscript.h"
  22
  23 #include "wine/debug.h"
  24
  25 WINE_DEFAULT_DEBUG_CHANNEL(vbscript);
  26
  27 #define FDEX_VERSION_MASK 0xf0000000
  28
  29 static inline BOOL is_func_id(vbdisp_t *This, DISPID id)
  30 {
  31     return id < This->desc->func_cnt;
  32 }
  33
  34 static BOOL get_func_id(vbdisp_t *This, const WCHAR *name, vbdisp_invoke_type_t invoke_type, BOOL search_private, DISPID *id)
  35 {
  36     unsigned i;
  37
  38     for(i = invoke_type == VBDISP_ANY ? 0 : 1; i < This->desc->func_cnt; i++) {
  39         if(invoke_type == VBDISP_ANY) {
  40             if(!search_private && !This->desc->funcs[i].is_public)
  41                 continue;
  42             if(!i && !This->desc->funcs[0].name) /* default value may not exist */
  43                 continue;
  44         }else {
  45             if(!This->desc->funcs[i].entries[invoke_type]
  46                 || (!search_private && !This->desc->funcs[i].entries[invoke_type]->is_public))
  47                 continue;
  48         }
  49
  50         if(!strcmpiW(This->desc->funcs[i].name, name)) {
  51             *id = i;
  52             return TRUE;
  53         }
  54     }
  55
  56     return FALSE;
  57 }
  58
  59 HRESULT vbdisp_get_id(vbdisp_t *This, BSTR name, vbdisp_invoke_type_t invoke_type, BOOL search_private, DISPID *id)
  60 {
  61     unsigned i;
  62
  63     if(get_func_id(This, name, invoke_type, search_private, id))
  64         return S_OK;
  65
  66     for(i=0; i < This->desc->prop_cnt; i++) {
  67         if(!search_private && !This->desc->props[i].is_public)
  68             continue;
  69
  70         if(!strcmpiW(This->desc->props[i].name, name)) {
  71             *id = i + This->desc->func_cnt;
  72             return S_OK;
  73         }
  74     }
  75
  76     if(This->desc->typeinfo) {
  77         HRESULT hres;
  78
  79         hres = ITypeInfo_GetIDsOfNames(This->desc->typeinfo, &name, 1, id);
  80         if(SUCCEEDED(hres))
  81             return S_OK;
  82     }
  83
  84     *id = -1;
  85     return DISP_E_UNKNOWNNAME;
  86 }
  87
  88 static VARIANT *get_propput_arg(const DISPPARAMS *dp)
  89 {
  90     unsigned i;
  91
  92     for(i=0; i < dp->cNamedArgs; i++) {
  93         if(dp->rgdispidNamedArgs[i] == DISPID_PROPERTYPUT)
  94             return dp->rgvarg+i;
  95     }
  96
  97     return NULL;
  98 }
  99
 100 static HRESULT invoke_variant_prop(VARIANT *v, WORD flags, DISPPARAMS *dp, VARIANT *res)
 101 {
 102     HRESULT hres;
 103
 104     switch(flags) {
 105     case DISPATCH_PROPERTYGET|DISPATCH_METHOD:
 106     case DISPATCH_PROPERTYGET:
 107         if(dp->cArgs) {
 108             WARN("called with arguments\n");
 109             return DISP_E_MEMBERNOTFOUND; /* That's what tests show */
 110         }
 111
 112         hres = VariantCopy(res, v);
 113         break;
 114
 115     case DISPATCH_PROPERTYPUT: {
 116         VARIANT *put_val;
 117
 118         put_val = get_propput_arg(dp);
 119         if(!put_val) {
 120             WARN("no value to set\n");
 121             return DISP_E_PARAMNOTOPTIONAL;
 122         }
 123
 124         if(res)
 125             V_VT(res) = VT_EMPTY;
 126
 127         hres = VariantCopy(v, put_val);
 128         break;
 129     }
 130
 131     default:
 132         FIXME("unimplemented flags %x\n", flags);
 133         return E_NOTIMPL;
 134     }
 135
 136     return hres;
 137 }
 138
 139 static HRESULT invoke_builtin(vbdisp_t *This, const builtin_prop_t *prop, WORD flags, DISPPARAMS *dp, VARIANT *res)
 140 {
 141     VARIANT *args, arg_buf[8];
 142     unsigned argn;
 143
 144     switch(flags) {
 145     case DISPATCH_PROPERTYGET:
 146         if(!(prop->flags & (BP_GET|BP_GETPUT))) {
 147             FIXME("property does not support DISPATCH_PROPERTYGET\n");
 148             return E_FAIL;
 149         }
 150         break;
 151     case DISPATCH_PROPERTYGET|DISPATCH_METHOD:
 152         break;
 153     case DISPATCH_METHOD:
 154         if(prop->flags & (BP_GET|BP_GETPUT)) {
 155             FIXME("Call on property\n");
 156             return E_FAIL;
 157         }
 158         break;
 159     case DISPATCH_PROPERTYPUT:
 160         if(!(prop->flags & (BP_GET|BP_GETPUT))) {
 161             FIXME("property does not support DISPATCH_PROPERTYPUT\n");
 162             return E_FAIL;
 163         }
 164
 165         FIXME("call put\n");
 166         return E_NOTIMPL;
 167     default:
 168         FIXME("unsupported flags %x\n", flags);
 169         return E_NOTIMPL;
 170     }
 171
 172     argn = arg_cnt(dp);
 173
 174     if(argn < prop->min_args || argn > (prop->max_args ? prop->max_args : prop->min_args)) {
 175         FIXME("invalid number of arguments\n");
 176         return E_FAIL;
 177     }
 178
 179     args = dp->rgvarg;
 180
 181     if(argn == 1) {
 182         if(V_VT(dp->rgvarg) == (VT_BYREF|VT_VARIANT))
 183             args = V_VARIANTREF(dp->rgvarg);
 184     }else {
 185         unsigned i;
 186
 187         assert(argn < sizeof(arg_buf)/sizeof(*arg_buf));
 188
 189         for(i=0; i < argn; i++) {
 190             if(V_VT(dp->rgvarg+i) == (VT_BYREF|VT_VARIANT)) {
 191                 for(i=0; i < argn; i++) {
 192                     if(V_VT(dp->rgvarg+i) == (VT_BYREF|VT_VARIANT))
 193                         arg_buf[i] = *V_VARIANTREF(dp->rgvarg+i);
 194                     else
 195                         arg_buf[i] = dp->rgvarg[i];
 196                 }
 197                 args = arg_buf;
 198                 break;
 199             }
 200         }
 201     }
 202
 203     return prop->proc(This, args, dp->cArgs, res);
 204 }
 205
 206 static BOOL run_terminator(vbdisp_t *This)
 207 {
 208     DISPPARAMS dp = {0};
 209
 210     if(This->terminator_ran)
 211         return TRUE;
 212     This->terminator_ran = TRUE;
 213
 214     if(!This->desc->class_terminate_id)
 215         return TRUE;
 216
 217     This->ref++;
 218     exec_script(This->desc->ctx, This->desc->funcs[This->desc->class_terminate_id].entries[VBDISP_CALLGET],
 219             (IDispatch*)&This->IDispatchEx_iface, &dp, NULL);
 220     return !--This->ref;
 221 }
 222
 223 static void clean_props(vbdisp_t *This)
 224 {
 225     unsigned i;
 226
 227     if(!This->desc)
 228         return;
 229
 230     for(i=0; i < This->desc->prop_cnt; i++)
 231         VariantClear(This->props+i);
 232 }
 233
 234 static inline vbdisp_t *impl_from_IDispatchEx(IDispatchEx *iface)
 235 {
 236     return CONTAINING_RECORD(iface, vbdisp_t, IDispatchEx_iface);
 237 }
 238
 239 static HRESULT WINAPI DispatchEx_QueryInterface(IDispatchEx *iface, REFIID riid, void **ppv)
 240 {
 241     vbdisp_t *This = impl_from_IDispatchEx(iface);
 242
 243     if(IsEqualGUID(&IID_IUnknown, riid)) {
 244         TRACE("(%p)->(IID_IUnknown %p)\n", This, ppv);
 245         *ppv = &This->IDispatchEx_iface;
 246     }else if(IsEqualGUID(&IID_IDispatch, riid)) {
 247         TRACE("(%p)->(IID_IDispatch %p)\n", This, ppv);
 248         *ppv = &This->IDispatchEx_iface;
 249     }else if(IsEqualGUID(&IID_IDispatchEx, riid)) {
 250         TRACE("(%p)->(IID_IDispatchEx %p)\n", This, ppv);
 251         *ppv = &This->IDispatchEx_iface;
 252     }else {
 253         WARN("(%p)->(%s %p)\n", This, debugstr_guid(riid), ppv);
 254         *ppv = NULL;
 255         return E_NOINTERFACE;
 256     }
 257
 258     IUnknown_AddRef((IUnknown*)*ppv);
 259     return S_OK;
 260 }
 261
 262 static ULONG WINAPI DispatchEx_AddRef(IDispatchEx *iface)
 263 {
 264     vbdisp_t *This = impl_from_IDispatchEx(iface);
 265     LONG ref = InterlockedIncrement(&This->ref);
 266
 267     TRACE("(%p) ref=%d\n", This, ref);
 268
 269     return ref;
 270 }
 271
 272 static ULONG WINAPI DispatchEx_Release(IDispatchEx *iface)
 273 {
 274     vbdisp_t *This = impl_from_IDispatchEx(iface);
 275     LONG ref = InterlockedDecrement(&This->ref);
 276
 277     TRACE("(%p) ref=%d\n", This, ref);
 278
 279     if(!ref && run_terminator(This)) {
 280         clean_props(This);
 281         list_remove(&This->entry);
 282         heap_free(This);
 283     }
 284
 285     return ref;
 286 }
 287
 288 static HRESULT WINAPI DispatchEx_GetTypeInfoCount(IDispatchEx *iface, UINT *pctinfo)
 289 {
 290     vbdisp_t *This = impl_from_IDispatchEx(iface);
 291
 292     TRACE("(%p)->(%p)\n", This, pctinfo);
 293
 294     *pctinfo = 1;
 295     return S_OK;
 296 }
 297
 298 static HRESULT WINAPI DispatchEx_GetTypeInfo(IDispatchEx *iface, UINT iTInfo, LCID lcid,
 299                                               ITypeInfo **ppTInfo)
 300 {
 301     vbdisp_t *This = impl_from_IDispatchEx(iface);
 302     FIXME("(%p)->(%u %u %p)\n", This, iTInfo, lcid, ppTInfo);
 303     return E_NOTIMPL;
 304 }
 305
 306 static HRESULT WINAPI DispatchEx_GetIDsOfNames(IDispatchEx *iface, REFIID riid,
 307                                                 LPOLESTR *rgszNames, UINT cNames, LCID lcid,
 308                                                 DISPID *rgDispId)
 309 {
 310     vbdisp_t *This = impl_from_IDispatchEx(iface);
 311     FIXME("(%p)->(%s %p %u %u %p)\n", This, debugstr_guid(riid), rgszNames, cNames,
 312           lcid, rgDispId);
 313     return E_NOTIMPL;
 314 }
 315
 316 static HRESULT WINAPI DispatchEx_Invoke(IDispatchEx *iface, DISPID dispIdMember,
 317                                         REFIID riid, LCID lcid, WORD wFlags, DISPPARAMS *pDispParams,
 318                             VARIANT *pVarResult, EXCEPINFO *pExcepInfo, UINT *puArgErr)
 319 {
 320     vbdisp_t *This = impl_from_IDispatchEx(iface);
 321     FIXME("(%p)->(%d %s %d %d %p %p %p %p)\n", This, dispIdMember, debugstr_guid(riid),
 322           lcid, wFlags, pDispParams, pVarResult, pExcepInfo, puArgErr);
 323     return E_NOTIMPL;
 324 }
 325
 326 static HRESULT WINAPI DispatchEx_GetDispID(IDispatchEx *iface, BSTR bstrName, DWORD grfdex, DISPID *pid)
 327 {
 328     vbdisp_t *This = impl_from_IDispatchEx(iface);
 329
 330     TRACE("(%p)->(%s %x %p)\n", This, debugstr_w(bstrName), grfdex, pid);
 331
 332     grfdex &= ~FDEX_VERSION_MASK;
 333
 334     if(!This->desc)
 335         return E_UNEXPECTED;
 336
 337     /* Tests show that fdexNameCaseSensitive is ignored */
 338
 339     if(grfdex & ~(fdexNameEnsure|fdexNameCaseInsensitive|fdexNameCaseSensitive)) {
 340         FIXME("unsupported flags %x\n", grfdex);
 341         return E_NOTIMPL;
 342     }
 343
 344     return vbdisp_get_id(This, bstrName, VBDISP_ANY, FALSE, pid);
 345 }
 346
 347 static HRESULT WINAPI DispatchEx_InvokeEx(IDispatchEx *iface, DISPID id, LCID lcid, WORD wFlags, DISPPARAMS *pdp,
 348         VARIANT *pvarRes, EXCEPINFO *pei, IServiceProvider *pspCaller)
 349 {
 350     vbdisp_t *This = impl_from_IDispatchEx(iface);
 351
 352     TRACE("(%p)->(%x %x %x %p %p %p %p)\n", This, id, lcid, wFlags, pdp, pvarRes, pei, pspCaller);
 353
 354     if(!This->desc)
 355         return E_UNEXPECTED;
 356
 357     if(pvarRes)
 358         V_VT(pvarRes) = VT_EMPTY;
 359
 360     if(id < 0)
 361         return DISP_E_MEMBERNOTFOUND;
 362
 363     if(is_func_id(This, id)) {
 364         function_t *func;
 365
 366         switch(wFlags) {
 367         case DISPATCH_METHOD:
 368         case DISPATCH_METHOD|DISPATCH_PROPERTYGET:
 369             func = This->desc->funcs[id].entries[VBDISP_CALLGET];
 370             if(!func) {
 371                 FIXME("no invoke/getter\n");
 372                 return DISP_E_MEMBERNOTFOUND;
 373             }
 374
 375             return exec_script(This->desc->ctx, func, (IDispatch*)&This->IDispatchEx_iface, pdp, pvarRes);
 376         case DISPATCH_PROPERTYPUT: {
 377             VARIANT *put_val;
 378             DISPPARAMS dp = {NULL, NULL, 1, 0};
 379
 380             if(arg_cnt(pdp)) {
 381                 FIXME("arguments not implemented\n");
 382                 return E_NOTIMPL;
 383             }
 384
 385             put_val = get_propput_arg(pdp);
 386             if(!put_val) {
 387                 WARN("no value to set\n");
 388                 return DISP_E_PARAMNOTOPTIONAL;
 389             }
 390
 391             dp.rgvarg = put_val;
 392             func = This->desc->funcs[id].entries[V_VT(put_val) == VT_DISPATCH ? VBDISP_SET : VBDISP_LET];
 393             if(!func) {
 394                 FIXME("no letter/setter\n");
 395                 return DISP_E_MEMBERNOTFOUND;
 396             }
 397
 398             return exec_script(This->desc->ctx, func, (IDispatch*)&This->IDispatchEx_iface, &dp, NULL);
 399         }
 400         default:
 401             FIXME("flags %x\n", wFlags);
 402             return DISP_E_MEMBERNOTFOUND;
 403         }
 404     }
 405
 406     if(id < This->desc->prop_cnt + This->desc->func_cnt)
 407         return invoke_variant_prop(This->props+(id-This->desc->func_cnt), wFlags, pdp, pvarRes);
 408
 409     if(This->desc->builtin_prop_cnt) {
 410         unsigned min = 0, max = This->desc->builtin_prop_cnt-1, i;
 411
 412         while(min <= max) {
 413             i = (min+max)/2;
 414             if(This->desc->builtin_props[i].id == id)
 415                 return invoke_builtin(This, This->desc->builtin_props+i, wFlags, pdp, pvarRes);
 416             if(This->desc->builtin_props[i].id < id)
 417                 min = i+1;
 418             else
 419                 max = i-1;
 420         }
 421     }
 422
 423     return DISP_E_MEMBERNOTFOUND;
 424 }
 425
 426 static HRESULT WINAPI DispatchEx_DeleteMemberByName(IDispatchEx *iface, BSTR bstrName, DWORD grfdex)
 427 {
 428     vbdisp_t *This = impl_from_IDispatchEx(iface);
 429     FIXME("(%p)->(%s %x)\n", This, debugstr_w(bstrName), grfdex);
 430     return E_NOTIMPL;
 431 }
 432
 433 static HRESULT WINAPI DispatchEx_DeleteMemberByDispID(IDispatchEx *iface, DISPID id)
 434 {
 435     vbdisp_t *This = impl_from_IDispatchEx(iface);
 436     FIXME("(%p)->(%x)\n", This, id);
 437     return E_NOTIMPL;
 438 }
 439
 440 static HRESULT WINAPI DispatchEx_GetMemberProperties(IDispatchEx *iface, DISPID id, DWORD grfdexFetch, DWORD *pgrfdex)
 441 {
 442     vbdisp_t *This = impl_from_IDispatchEx(iface);
 443     FIXME("(%p)->(%x %x %p)\n", This, id, grfdexFetch, pgrfdex);
 444     return E_NOTIMPL;
 445 }
 446
 447 static HRESULT WINAPI DispatchEx_GetMemberName(IDispatchEx *iface, DISPID id, BSTR *pbstrName)
 448 {
 449     vbdisp_t *This = impl_from_IDispatchEx(iface);
 450     FIXME("(%p)->(%x %p)\n", This, id, pbstrName);
 451     return E_NOTIMPL;
 452 }
 453
 454 static HRESULT WINAPI DispatchEx_GetNextDispID(IDispatchEx *iface, DWORD grfdex, DISPID id, DISPID *pid)
 455 {
 456     vbdisp_t *This = impl_from_IDispatchEx(iface);
 457     FIXME("(%p)->(%x %x %p)\n", This, grfdex, id, pid);
 458     return E_NOTIMPL;
 459 }
 460
 461 static HRESULT WINAPI DispatchEx_GetNameSpaceParent(IDispatchEx *iface, IUnknown **ppunk)
 462 {
 463     vbdisp_t *This = impl_from_IDispatchEx(iface);
 464     FIXME("(%p)->(%p)\n", This, ppunk);
 465     return E_NOTIMPL;
 466 }
 467
 468 static IDispatchExVtbl DispatchExVtbl = {
 469     DispatchEx_QueryInterface,
 470     DispatchEx_AddRef,
 471     DispatchEx_Release,
 472     DispatchEx_GetTypeInfoCount,
 473     DispatchEx_GetTypeInfo,
 474     DispatchEx_GetIDsOfNames,
 475     DispatchEx_Invoke,
 476     DispatchEx_GetDispID,
 477     DispatchEx_InvokeEx,
 478     DispatchEx_DeleteMemberByName,
 479     DispatchEx_DeleteMemberByDispID,
 480     DispatchEx_GetMemberProperties,
 481     DispatchEx_GetMemberName,
 482     DispatchEx_GetNextDispID,
 483     DispatchEx_GetNameSpaceParent
 484 };
 485
 486 static inline vbdisp_t *unsafe_impl_from_IDispatch(IDispatch *iface)
 487 {
 488     return iface->lpVtbl == (IDispatchVtbl*)&DispatchExVtbl
 489         ? CONTAINING_RECORD(iface, vbdisp_t, IDispatchEx_iface)
 490         : NULL;
 491 }
 492
 493 HRESULT create_vbdisp(const class_desc_t *desc, vbdisp_t **ret)
 494 {
 495     vbdisp_t *vbdisp;
 496
 497     vbdisp = heap_alloc_zero( FIELD_OFFSET( vbdisp_t, props[desc->prop_cnt] ));
 498     if(!vbdisp)
 499         return E_OUTOFMEMORY;
 500
 501     vbdisp->IDispatchEx_iface.lpVtbl = &DispatchExVtbl;
 502     vbdisp->ref = 1;
 503     vbdisp->desc = desc;
 504
 505     list_add_tail(&desc->ctx->objects, &vbdisp->entry);
 506
 507     if(desc->class_initialize_id) {
 508         DISPPARAMS dp = {0};
 509         HRESULT hres;
 510
 511         hres = exec_script(desc->ctx, desc->funcs[desc->class_initialize_id].entries[VBDISP_CALLGET],
 512                            (IDispatch*)&vbdisp->IDispatchEx_iface, &dp, NULL);
 513         if(FAILED(hres)) {
 514             IDispatchEx_Release(&vbdisp->IDispatchEx_iface);
 515             return hres;
 516         }
 517     }
 518
 519     *ret = vbdisp;
 520     return S_OK;
 521 }
 522
 523 static HRESULT Procedure_invoke(vbdisp_t *This, VARIANT *args, unsigned args_cnt, VARIANT *res)
 524 {
 525     script_ctx_t *ctx = This->desc->ctx;
 526     HRESULT hres;
 527
 528     TRACE("\n");
 529
 530     IActiveScriptSite_OnEnterScript(ctx->site);
 531     hres = exec_script(ctx, This->desc->value_func, NULL, NULL, NULL);
 532     IActiveScriptSite_OnLeaveScript(ctx->site);
 533
 534     return hres;
 535 }
 536
 537 static const builtin_prop_t procedure_props[] = {
 538     {DISPID_VALUE,  Procedure_invoke, 0}
 539 };
 540
 541 HRESULT create_procedure_disp(script_ctx_t *ctx, vbscode_t *code, IDispatch **ret)
 542 {
 543     class_desc_t *desc;
 544     vbdisp_t *vbdisp;
 545     HRESULT hres;
 546
 547     desc = heap_alloc_zero(sizeof(*desc));
 548     if(!desc)
 549         return E_OUTOFMEMORY;
 550
 551     desc->ctx = ctx;
 552     desc->builtin_prop_cnt = sizeof(procedure_props)/sizeof(*procedure_props);
 553     desc->builtin_props = procedure_props;
 554     desc->value_func = &code->main_code;
 555
 556     hres = create_vbdisp(desc, &vbdisp);
 557     if(FAILED(hres)) {
 558         heap_free(desc);
 559         return hres;
 560     }
 561
 562     desc->next = ctx->procs;
 563     ctx->procs = desc;
 564
 565     *ret = (IDispatch*)&vbdisp->IDispatchEx_iface;
 566     return S_OK;
 567 }
 568
 569 struct _ident_map_t {
 570     const WCHAR *name;
 571     BOOL is_var;
 572     union {
 573         dynamic_var_t *var;
 574         function_t *func;
 575     } u;
 576 };
 577
 578 static inline DISPID ident_to_id(ScriptDisp *This, ident_map_t *ident)
 579 {
 580     return (ident-This->ident_map)+1;
 581 }
 582
 583 static inline ident_map_t *id_to_ident(ScriptDisp *This, DISPID id)
 584 {
 585     return 0 < id && id <= This->ident_map_cnt ? This->ident_map+id-1 : NULL;
 586 }
 587
 588 static ident_map_t *add_ident(ScriptDisp *This, const WCHAR *name)
 589 {
 590     ident_map_t *ret;
 591
 592     if(!This->ident_map_size) {
 593         This->ident_map = heap_alloc(4 * sizeof(*This->ident_map));
 594         if(!This->ident_map)
 595             return NULL;
 596         This->ident_map_size = 4;
 597     }else if(This->ident_map_cnt == This->ident_map_size) {
 598         ident_map_t *new_map;
 599
 600         new_map = heap_realloc(This->ident_map, 2*This->ident_map_size*sizeof(*new_map));
 601         if(!new_map)
 602             return NULL;
 603         This->ident_map = new_map;
 604         This->ident_map_size *= 2;
 605     }
 606
 607     ret = This->ident_map + This->ident_map_cnt++;
 608     ret->name = name;
 609     return ret;
 610 }
 611
 612 static inline ScriptDisp *ScriptDisp_from_IDispatchEx(IDispatchEx *iface)
 613 {
 614     return CONTAINING_RECORD(iface, ScriptDisp, IDispatchEx_iface);
 615 }
 616
 617 static HRESULT WINAPI ScriptDisp_QueryInterface(IDispatchEx *iface, REFIID riid, void **ppv)
 618 {
 619     ScriptDisp *This = ScriptDisp_from_IDispatchEx(iface);
 620
 621     if(IsEqualGUID(&IID_IUnknown, riid)) {
 622         TRACE("(%p)->(IID_IUnknown %p)\n", This, ppv);
 623         *ppv = &This->IDispatchEx_iface;
 624     }else if(IsEqualGUID(&IID_IDispatch, riid)) {
 625         TRACE("(%p)->(IID_IDispatch %p)\n", This, ppv);
 626         *ppv = &This->IDispatchEx_iface;
 627     }else if(IsEqualGUID(&IID_IDispatchEx, riid)) {
 628         TRACE("(%p)->(IID_IDispatchEx %p)\n", This, ppv);
 629         *ppv = &This->IDispatchEx_iface;
 630     }else {
 631         WARN("(%p)->(%s %p)\n", This, debugstr_guid(riid), ppv);
 632         *ppv = NULL;
 633         return E_NOINTERFACE;
 634     }
 635
 636     IUnknown_AddRef((IUnknown*)*ppv);
 637     return S_OK;
 638 }
 639
 640 static ULONG WINAPI ScriptDisp_AddRef(IDispatchEx *iface)
 641 {
 642     ScriptDisp *This = ScriptDisp_from_IDispatchEx(iface);
 643     LONG ref = InterlockedIncrement(&This->ref);
 644
 645     TRACE("(%p) ref=%d\n", This, ref);
 646
 647     return ref;
 648 }
 649
 650 static ULONG WINAPI ScriptDisp_Release(IDispatchEx *iface)
 651 {
 652     ScriptDisp *This = ScriptDisp_from_IDispatchEx(iface);
 653     LONG ref = InterlockedDecrement(&This->ref);
 654
 655     TRACE("(%p) ref=%d\n", This, ref);
 656
 657     if(!ref) {
 658         assert(!This->ctx);
 659         heap_free(This->ident_map);
 660         heap_free(This);
 661     }
 662
 663     return ref;
 664 }
 665
 666 static HRESULT WINAPI ScriptDisp_GetTypeInfoCount(IDispatchEx *iface, UINT *pctinfo)
 667 {
 668     ScriptDisp *This = ScriptDisp_from_IDispatchEx(iface);
 669
 670     TRACE("(%p)->(%p)\n", This, pctinfo);
 671
 672     *pctinfo = 1;
 673     return S_OK;
 674 }
 675
 676 static HRESULT WINAPI ScriptDisp_GetTypeInfo(IDispatchEx *iface, UINT iTInfo, LCID lcid,
 677                                               ITypeInfo **ppTInfo)
 678 {
 679     ScriptDisp *This = ScriptDisp_from_IDispatchEx(iface);
 680     FIXME("(%p)->(%u %u %p)\n", This, iTInfo, lcid, ppTInfo);
 681     return E_NOTIMPL;
 682 }
 683
 684 static HRESULT WINAPI ScriptDisp_GetIDsOfNames(IDispatchEx *iface, REFIID riid,
 685                                                 LPOLESTR *rgszNames, UINT cNames, LCID lcid,
 686                                                 DISPID *rgDispId)
 687 {
 688     ScriptDisp *This = ScriptDisp_from_IDispatchEx(iface);
 689     FIXME("(%p)->(%s %p %u %u %p)\n", This, debugstr_guid(riid), rgszNames, cNames,
 690           lcid, rgDispId);
 691     return E_NOTIMPL;
 692 }
 693
 694 static HRESULT WINAPI ScriptDisp_Invoke(IDispatchEx *iface, DISPID dispIdMember,
 695                                         REFIID riid, LCID lcid, WORD wFlags, DISPPARAMS *pDispParams,
 696                             VARIANT *pVarResult, EXCEPINFO *pExcepInfo, UINT *puArgErr)
 697 {
 698     ScriptDisp *This = ScriptDisp_from_IDispatchEx(iface);
 699     FIXME("(%p)->(%d %s %d %d %p %p %p %p)\n", This, dispIdMember, debugstr_guid(riid),
 700           lcid, wFlags, pDispParams, pVarResult, pExcepInfo, puArgErr);
 701     return E_NOTIMPL;
 702 }
 703
 704 static HRESULT WINAPI ScriptDisp_GetDispID(IDispatchEx *iface, BSTR bstrName, DWORD grfdex, DISPID *pid)
 705 {
 706     ScriptDisp *This = ScriptDisp_from_IDispatchEx(iface);
 707     dynamic_var_t *var;
 708     ident_map_t *ident;
 709     function_t *func;
 710
 711     TRACE("(%p)->(%s %x %p)\n", This, debugstr_w(bstrName), grfdex, pid);
 712
 713     if(!This->ctx)
 714         return E_UNEXPECTED;
 715
 716     for(ident = This->ident_map; ident < This->ident_map+This->ident_map_cnt; ident++) {
 717         if(!strcmpiW(ident->name, bstrName)) {
 718             *pid = ident_to_id(This, ident);
 719             return S_OK;
 720         }
 721     }
 722
 723     for(var = This->ctx->global_vars; var; var = var->next) {
 724         if(!strcmpiW(var->name, bstrName)) {
 725             ident = add_ident(This, var->name);
 726             if(!ident)
 727                 return E_OUTOFMEMORY;
 728
 729             ident->is_var = TRUE;
 730             ident->u.var = var;
 731             *pid = ident_to_id(This, ident);
 732             return S_OK;
 733         }
 734     }
 735
 736     for(func = This->ctx->global_funcs; func; func = func->next) {
 737         if(!strcmpiW(func->name, bstrName)) {
 738             ident = add_ident(This, func->name);
 739             if(!ident)
 740                 return E_OUTOFMEMORY;
 741
 742             ident->is_var = FALSE;
 743             ident->u.func = func;
 744             *pid =  ident_to_id(This, ident);
 745             return S_OK;
 746         }
 747     }
 748
 749     *pid = -1;
 750     return DISP_E_UNKNOWNNAME;
 751 }
 752
 753 static HRESULT WINAPI ScriptDisp_InvokeEx(IDispatchEx *iface, DISPID id, LCID lcid, WORD wFlags, DISPPARAMS *pdp,
 754         VARIANT *pvarRes, EXCEPINFO *pei, IServiceProvider *pspCaller)
 755 {
 756     ScriptDisp *This = ScriptDisp_from_IDispatchEx(iface);
 757     ident_map_t *ident;
 758     HRESULT hres;
 759
 760     TRACE("(%p)->(%x %x %x %p %p %p %p)\n", This, id, lcid, wFlags, pdp, pvarRes, pei, pspCaller);
 761
 762     ident = id_to_ident(This, id);
 763     if(!ident)
 764         return DISP_E_MEMBERNOTFOUND;
 765
 766     if(ident->is_var) {
 767         if(ident->u.var->is_const) {
 768             FIXME("const not supported\n");
 769             return E_NOTIMPL;
 770         }
 771
 772         return invoke_variant_prop(&ident->u.var->v, wFlags, pdp, pvarRes);
 773     }
 774
 775
 776     IActiveScriptSite_OnEnterScript(This->ctx->site);
 777     hres = exec_script(This->ctx, ident->u.func, NULL, pdp, pvarRes);
 778     IActiveScriptSite_OnLeaveScript(This->ctx->site);
 779
 780     return hres;
 781 }
 782
 783 static HRESULT WINAPI ScriptDisp_DeleteMemberByName(IDispatchEx *iface, BSTR bstrName, DWORD grfdex)
 784 {
 785     ScriptDisp *This = ScriptDisp_from_IDispatchEx(iface);
 786     FIXME("(%p)->(%s %x)\n", This, debugstr_w(bstrName), grfdex);
 787     return E_NOTIMPL;
 788 }
 789
 790 static HRESULT WINAPI ScriptDisp_DeleteMemberByDispID(IDispatchEx *iface, DISPID id)
 791 {
 792     ScriptDisp *This = ScriptDisp_from_IDispatchEx(iface);
 793     FIXME("(%p)->(%x)\n", This, id);
 794     return E_NOTIMPL;
 795 }
 796
 797 static HRESULT WINAPI ScriptDisp_GetMemberProperties(IDispatchEx *iface, DISPID id, DWORD grfdexFetch, DWORD *pgrfdex)
 798 {
 799     ScriptDisp *This = ScriptDisp_from_IDispatchEx(iface);
 800     FIXME("(%p)->(%x %x %p)\n", This, id, grfdexFetch, pgrfdex);
 801     return E_NOTIMPL;
 802 }
 803
 804 static HRESULT WINAPI ScriptDisp_GetMemberName(IDispatchEx *iface, DISPID id, BSTR *pbstrName)
 805 {
 806     ScriptDisp *This = ScriptDisp_from_IDispatchEx(iface);
 807     FIXME("(%p)->(%x %p)\n", This, id, pbstrName);
 808     return E_NOTIMPL;
 809 }
 810
 811 static HRESULT WINAPI ScriptDisp_GetNextDispID(IDispatchEx *iface, DWORD grfdex, DISPID id, DISPID *pid)
 812 {
 813     ScriptDisp *This = ScriptDisp_from_IDispatchEx(iface);
 814     FIXME("(%p)->(%x %x %p)\n", This, grfdex, id, pid);
 815     return E_NOTIMPL;
 816 }
 817
 818 static HRESULT WINAPI ScriptDisp_GetNameSpaceParent(IDispatchEx *iface, IUnknown **ppunk)
 819 {
 820     ScriptDisp *This = ScriptDisp_from_IDispatchEx(iface);
 821     FIXME("(%p)->(%p)\n", This, ppunk);
 822     return E_NOTIMPL;
 823 }
 824
 825 static IDispatchExVtbl ScriptDispVtbl = {
 826     ScriptDisp_QueryInterface,
 827     ScriptDisp_AddRef,
 828     ScriptDisp_Release,
 829     ScriptDisp_GetTypeInfoCount,
 830     ScriptDisp_GetTypeInfo,
 831     ScriptDisp_GetIDsOfNames,
 832     ScriptDisp_Invoke,
 833     ScriptDisp_GetDispID,
 834     ScriptDisp_InvokeEx,
 835     ScriptDisp_DeleteMemberByName,
 836     ScriptDisp_DeleteMemberByDispID,
 837     ScriptDisp_GetMemberProperties,
 838     ScriptDisp_GetMemberName,
 839     ScriptDisp_GetNextDispID,
 840     ScriptDisp_GetNameSpaceParent
 841 };
 842
 843 HRESULT create_script_disp(script_ctx_t *ctx, ScriptDisp **ret)
 844 {
 845     ScriptDisp *script_disp;
 846
 847     script_disp = heap_alloc_zero(sizeof(*script_disp));
 848     if(!script_disp)
 849         return E_OUTOFMEMORY;
 850
 851     script_disp->IDispatchEx_iface.lpVtbl = &ScriptDispVtbl;
 852     script_disp->ref = 1;
 853     script_disp->ctx = ctx;
 854
 855     *ret = script_disp;
 856     return S_OK;
 857 }
 858
 859 void collect_objects(script_ctx_t *ctx)
 860 {
 861     vbdisp_t *iter, *iter2;
 862
 863     LIST_FOR_EACH_ENTRY_SAFE(iter, iter2, &ctx->objects, vbdisp_t, entry)
 864         run_terminator(iter);
 865
 866     while(!list_empty(&ctx->objects)) {
 867         iter = LIST_ENTRY(list_head(&ctx->objects), vbdisp_t, entry);
 868
 869         IDispatchEx_AddRef(&iter->IDispatchEx_iface);
 870         clean_props(iter);
 871         iter->desc = NULL;
 872         list_remove(&iter->entry);
 873         list_init(&iter->entry);
 874         IDispatchEx_Release(&iter->IDispatchEx_iface);
 875     }
 876 }
 877
 878 HRESULT disp_get_id(IDispatch *disp, BSTR name, vbdisp_invoke_type_t invoke_type, BOOL search_private, DISPID *id)
 879 {
 880     IDispatchEx *dispex;
 881     vbdisp_t *vbdisp;
 882     HRESULT hres;
 883
 884     vbdisp = unsafe_impl_from_IDispatch(disp);
 885     if(vbdisp)
 886         return vbdisp_get_id(vbdisp, name, invoke_type, search_private, id);
 887
 888     hres = IDispatch_QueryInterface(disp, &IID_IDispatchEx, (void**)&dispex);
 889     if(FAILED(hres)) {
 890         TRACE("using IDispatch\n");
 891         return IDispatch_GetIDsOfNames(disp, &IID_NULL, &name, 1, 0, id);
 892     }
 893
 894     hres = IDispatchEx_GetDispID(dispex, name, fdexNameCaseInsensitive, id);
 895     IDispatchEx_Release(dispex);
 896     return hres;
 897 }
 898
 899 HRESULT disp_call(script_ctx_t *ctx, IDispatch *disp, DISPID id, DISPPARAMS *dp, VARIANT *retv)
 900 {
 901     const WORD flags = DISPATCH_METHOD|(retv ? DISPATCH_PROPERTYGET : 0);
 902     IDispatchEx *dispex;
 903     EXCEPINFO ei;
 904     HRESULT hres;
 905
 906     memset(&ei, 0, sizeof(ei));
 907     if(retv)
 908         V_VT(retv) = VT_EMPTY;
 909
 910     hres = IDispatch_QueryInterface(disp, &IID_IDispatchEx, (void**)&dispex);
 911     if(FAILED(hres)) {
 912         UINT err = 0;
 913
 914         TRACE("using IDispatch\n");
 915         return IDispatch_Invoke(disp, id, &IID_NULL, ctx->lcid, flags, dp, retv, &ei, &err);
 916     }
 917
 918     hres = IDispatchEx_InvokeEx(dispex, id, ctx->lcid, flags, dp, retv, &ei, NULL /* CALLER_FIXME */);
 919     IDispatchEx_Release(dispex);
 920     return hres;
 921 }
 922
 923 HRESULT disp_propput(script_ctx_t *ctx, IDispatch *disp, DISPID id, DISPPARAMS *dp)
 924 {
 925     IDispatchEx *dispex;
 926     EXCEPINFO ei = {0};
 927     HRESULT hres;
 928
 929     hres = IDispatch_QueryInterface(disp, &IID_IDispatchEx, (void**)&dispex);
 930     if(SUCCEEDED(hres)) {
 931         hres = IDispatchEx_InvokeEx(dispex, id, ctx->lcid, DISPATCH_PROPERTYPUT, dp, NULL, &ei, NULL /* FIXME! */);
 932         IDispatchEx_Release(dispex);
 933     }else {
 934         ULONG err = 0;
 935
 936         TRACE("using IDispatch\n");
 937         hres = IDispatch_Invoke(disp, id, &IID_NULL, ctx->lcid, DISPATCH_PROPERTYPUT, dp, NULL, &ei, &err);
 938     }
 939
 940     return hres;
 941 }