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_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, DISPID *rgDispId)
 686 {
 687     ScriptDisp *This = ScriptDisp_from_IDispatchEx(iface);
 688     UINT i;
 689     HRESULT hres;
 690
 691     TRACE("(%p)->(%s %p %u %u %p)\n", This, debugstr_guid(riid), rgszNames, cNames,
 692           lcid, rgDispId);
 693
 694     for(i=0; i < cNames; i++) {
 695         hres = IDispatchEx_GetDispID(&This->IDispatchEx_iface, rgszNames[i], 0, rgDispId+i);
 696         if(FAILED(hres))
 697             return hres;
 698     }
 699
 700     return S_OK;
 701 }
 702
 703 static HRESULT WINAPI ScriptDisp_Invoke(IDispatchEx *iface, DISPID dispIdMember, REFIID riid, LCID lcid,
 704          WORD wFlags, DISPPARAMS *pDispParams, VARIANT *pVarResult, EXCEPINFO *pExcepInfo, UINT *puArgErr)
 705 {
 706     ScriptDisp *This = ScriptDisp_from_IDispatchEx(iface);
 707
 708     TRACE("(%p)->(%d %s %d %d %p %p %p %p)\n", This, dispIdMember, debugstr_guid(riid),
 709           lcid, wFlags, pDispParams, pVarResult, pExcepInfo, puArgErr);
 710
 711     return IDispatchEx_InvokeEx(&This->IDispatchEx_iface, dispIdMember, lcid, wFlags,
 712             pDispParams, pVarResult, pExcepInfo, NULL);
 713 }
 714
 715 static HRESULT WINAPI ScriptDisp_GetDispID(IDispatchEx *iface, BSTR bstrName, DWORD grfdex, DISPID *pid)
 716 {
 717     ScriptDisp *This = ScriptDisp_from_IDispatchEx(iface);
 718     dynamic_var_t *var;
 719     ident_map_t *ident;
 720     function_t *func;
 721
 722     TRACE("(%p)->(%s %x %p)\n", This, debugstr_w(bstrName), grfdex, pid);
 723
 724     if(!This->ctx)
 725         return E_UNEXPECTED;
 726
 727     for(ident = This->ident_map; ident < This->ident_map+This->ident_map_cnt; ident++) {
 728         if(!strcmpiW(ident->name, bstrName)) {
 729             *pid = ident_to_id(This, ident);
 730             return S_OK;
 731         }
 732     }
 733
 734     for(var = This->ctx->global_vars; var; var = var->next) {
 735         if(!strcmpiW(var->name, bstrName)) {
 736             ident = add_ident(This, var->name);
 737             if(!ident)
 738                 return E_OUTOFMEMORY;
 739
 740             ident->is_var = TRUE;
 741             ident->u.var = var;
 742             *pid = ident_to_id(This, ident);
 743             return S_OK;
 744         }
 745     }
 746
 747     for(func = This->ctx->global_funcs; func; func = func->next) {
 748         if(!strcmpiW(func->name, bstrName)) {
 749             ident = add_ident(This, func->name);
 750             if(!ident)
 751                 return E_OUTOFMEMORY;
 752
 753             ident->is_var = FALSE;
 754             ident->u.func = func;
 755             *pid =  ident_to_id(This, ident);
 756             return S_OK;
 757         }
 758     }
 759
 760     *pid = -1;
 761     return DISP_E_UNKNOWNNAME;
 762 }
 763
 764 static HRESULT WINAPI ScriptDisp_InvokeEx(IDispatchEx *iface, DISPID id, LCID lcid, WORD wFlags, DISPPARAMS *pdp,
 765         VARIANT *pvarRes, EXCEPINFO *pei, IServiceProvider *pspCaller)
 766 {
 767     ScriptDisp *This = ScriptDisp_from_IDispatchEx(iface);
 768     ident_map_t *ident;
 769     HRESULT hres;
 770
 771     TRACE("(%p)->(%x %x %x %p %p %p %p)\n", This, id, lcid, wFlags, pdp, pvarRes, pei, pspCaller);
 772
 773     ident = id_to_ident(This, id);
 774     if(!ident)
 775         return DISP_E_MEMBERNOTFOUND;
 776
 777     if(ident->is_var) {
 778         if(ident->u.var->is_const) {
 779             FIXME("const not supported\n");
 780             return E_NOTIMPL;
 781         }
 782
 783         return invoke_variant_prop(&ident->u.var->v, wFlags, pdp, pvarRes);
 784     }
 785
 786     switch(wFlags) {
 787     case DISPATCH_METHOD:
 788     case DISPATCH_METHOD|DISPATCH_PROPERTYGET:
 789         IActiveScriptSite_OnEnterScript(This->ctx->site);
 790         hres = exec_script(This->ctx, ident->u.func, NULL, pdp, pvarRes);
 791         IActiveScriptSite_OnLeaveScript(This->ctx->site);
 792         break;
 793     default:
 794         FIXME("Unsupported flags %x\n", wFlags);
 795         hres = E_NOTIMPL;
 796     }
 797
 798     return hres;
 799 }
 800
 801 static HRESULT WINAPI ScriptDisp_DeleteMemberByName(IDispatchEx *iface, BSTR bstrName, DWORD grfdex)
 802 {
 803     ScriptDisp *This = ScriptDisp_from_IDispatchEx(iface);
 804     FIXME("(%p)->(%s %x)\n", This, debugstr_w(bstrName), grfdex);
 805     return E_NOTIMPL;
 806 }
 807
 808 static HRESULT WINAPI ScriptDisp_DeleteMemberByDispID(IDispatchEx *iface, DISPID id)
 809 {
 810     ScriptDisp *This = ScriptDisp_from_IDispatchEx(iface);
 811     FIXME("(%p)->(%x)\n", This, id);
 812     return E_NOTIMPL;
 813 }
 814
 815 static HRESULT WINAPI ScriptDisp_GetMemberProperties(IDispatchEx *iface, DISPID id, DWORD grfdexFetch, DWORD *pgrfdex)
 816 {
 817     ScriptDisp *This = ScriptDisp_from_IDispatchEx(iface);
 818     FIXME("(%p)->(%x %x %p)\n", This, id, grfdexFetch, pgrfdex);
 819     return E_NOTIMPL;
 820 }
 821
 822 static HRESULT WINAPI ScriptDisp_GetMemberName(IDispatchEx *iface, DISPID id, BSTR *pbstrName)
 823 {
 824     ScriptDisp *This = ScriptDisp_from_IDispatchEx(iface);
 825     FIXME("(%p)->(%x %p)\n", This, id, pbstrName);
 826     return E_NOTIMPL;
 827 }
 828
 829 static HRESULT WINAPI ScriptDisp_GetNextDispID(IDispatchEx *iface, DWORD grfdex, DISPID id, DISPID *pid)
 830 {
 831     ScriptDisp *This = ScriptDisp_from_IDispatchEx(iface);
 832     FIXME("(%p)->(%x %x %p)\n", This, grfdex, id, pid);
 833     return E_NOTIMPL;
 834 }
 835
 836 static HRESULT WINAPI ScriptDisp_GetNameSpaceParent(IDispatchEx *iface, IUnknown **ppunk)
 837 {
 838     ScriptDisp *This = ScriptDisp_from_IDispatchEx(iface);
 839     FIXME("(%p)->(%p)\n", This, ppunk);
 840     return E_NOTIMPL;
 841 }
 842
 843 static IDispatchExVtbl ScriptDispVtbl = {
 844     ScriptDisp_QueryInterface,
 845     ScriptDisp_AddRef,
 846     ScriptDisp_Release,
 847     ScriptDisp_GetTypeInfoCount,
 848     ScriptDisp_GetTypeInfo,
 849     ScriptDisp_GetIDsOfNames,
 850     ScriptDisp_Invoke,
 851     ScriptDisp_GetDispID,
 852     ScriptDisp_InvokeEx,
 853     ScriptDisp_DeleteMemberByName,
 854     ScriptDisp_DeleteMemberByDispID,
 855     ScriptDisp_GetMemberProperties,
 856     ScriptDisp_GetMemberName,
 857     ScriptDisp_GetNextDispID,
 858     ScriptDisp_GetNameSpaceParent
 859 };
 860
 861 HRESULT create_script_disp(script_ctx_t *ctx, ScriptDisp **ret)
 862 {
 863     ScriptDisp *script_disp;
 864
 865     script_disp = heap_alloc_zero(sizeof(*script_disp));
 866     if(!script_disp)
 867         return E_OUTOFMEMORY;
 868
 869     script_disp->IDispatchEx_iface.lpVtbl = &ScriptDispVtbl;
 870     script_disp->ref = 1;
 871     script_disp->ctx = ctx;
 872
 873     *ret = script_disp;
 874     return S_OK;
 875 }
 876
 877 void collect_objects(script_ctx_t *ctx)
 878 {
 879     vbdisp_t *iter, *iter2;
 880
 881     LIST_FOR_EACH_ENTRY_SAFE(iter, iter2, &ctx->objects, vbdisp_t, entry)
 882         run_terminator(iter);
 883
 884     while(!list_empty(&ctx->objects)) {
 885         iter = LIST_ENTRY(list_head(&ctx->objects), vbdisp_t, entry);
 886
 887         IDispatchEx_AddRef(&iter->IDispatchEx_iface);
 888         clean_props(iter);
 889         iter->desc = NULL;
 890         list_remove(&iter->entry);
 891         list_init(&iter->entry);
 892         IDispatchEx_Release(&iter->IDispatchEx_iface);
 893     }
 894 }
 895
 896 HRESULT disp_get_id(IDispatch *disp, BSTR name, vbdisp_invoke_type_t invoke_type, BOOL search_private, DISPID *id)
 897 {
 898     IDispatchEx *dispex;
 899     vbdisp_t *vbdisp;
 900     HRESULT hres;
 901
 902     vbdisp = unsafe_impl_from_IDispatch(disp);
 903     if(vbdisp)
 904         return vbdisp_get_id(vbdisp, name, invoke_type, search_private, id);
 905
 906     hres = IDispatch_QueryInterface(disp, &IID_IDispatchEx, (void**)&dispex);
 907     if(FAILED(hres)) {
 908         TRACE("using IDispatch\n");
 909         return IDispatch_GetIDsOfNames(disp, &IID_NULL, &name, 1, 0, id);
 910     }
 911
 912     hres = IDispatchEx_GetDispID(dispex, name, fdexNameCaseInsensitive, id);
 913     IDispatchEx_Release(dispex);
 914     return hres;
 915 }
 916
 917 HRESULT disp_call(script_ctx_t *ctx, IDispatch *disp, DISPID id, DISPPARAMS *dp, VARIANT *retv)
 918 {
 919     const WORD flags = DISPATCH_METHOD|(retv ? DISPATCH_PROPERTYGET : 0);
 920     IDispatchEx *dispex;
 921     EXCEPINFO ei;
 922     HRESULT hres;
 923
 924     memset(&ei, 0, sizeof(ei));
 925     if(retv)
 926         V_VT(retv) = VT_EMPTY;
 927
 928     hres = IDispatch_QueryInterface(disp, &IID_IDispatchEx, (void**)&dispex);
 929     if(FAILED(hres)) {
 930         UINT err = 0;
 931
 932         TRACE("using IDispatch\n");
 933         return IDispatch_Invoke(disp, id, &IID_NULL, ctx->lcid, flags, dp, retv, &ei, &err);
 934     }
 935
 936     hres = IDispatchEx_InvokeEx(dispex, id, ctx->lcid, flags, dp, retv, &ei, NULL /* CALLER_FIXME */);
 937     IDispatchEx_Release(dispex);
 938     return hres;
 939 }
 940
 941 HRESULT disp_propput(script_ctx_t *ctx, IDispatch *disp, DISPID id, DISPPARAMS *dp)
 942 {
 943     IDispatchEx *dispex;
 944     EXCEPINFO ei = {0};
 945     HRESULT hres;
 946
 947     hres = IDispatch_QueryInterface(disp, &IID_IDispatchEx, (void**)&dispex);
 948     if(SUCCEEDED(hres)) {
 949         hres = IDispatchEx_InvokeEx(dispex, id, ctx->lcid, DISPATCH_PROPERTYPUT, dp, NULL, &ei, NULL /* FIXME! */);
 950         IDispatchEx_Release(dispex);
 951     }else {
 952         ULONG err = 0;
 953
 954         TRACE("using IDispatch\n");
 955         hres = IDispatch_Invoke(disp, id, &IID_NULL, ctx->lcid, DISPATCH_PROPERTYPUT, dp, NULL, &ei, &err);
 956     }
 957
 958     return hres;
 959 }