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