git.oblomov.eu Git - wine/blob - dlls/mshtml/dispex.c

   1 /*
   2  * Copyright 2008-2009 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 <stdarg.h>
  20
  21 #define COBJMACROS
  22
  23 #include "windef.h"
  24 #include "winbase.h"
  25 #include "winuser.h"
  26 #include "ole2.h"
  27
  28 #include "wine/debug.h"
  29
  30 #include "mshtml_private.h"
  31
  32 WINE_DEFAULT_DEBUG_CHANNEL(mshtml);
  33
  34 static const WCHAR objectW[] = {'[','o','b','j','e','c','t',']',0};
  35
  36 typedef struct {
  37     DISPID id;
  38     BSTR name;
  39     tid_t tid;
  40     int func_disp_idx;
  41 } func_info_t;
  42
  43 struct dispex_data_t {
  44     DWORD func_cnt;
  45     func_info_t *funcs;
  46     func_info_t **name_table;
  47     DWORD func_disp_cnt;
  48
  49     struct list entry;
  50 };
  51
  52 typedef struct {
  53     VARIANT var;
  54     LPWSTR name;
  55 } dynamic_prop_t;
  56
  57 typedef struct {
  58     DispatchEx dispex;
  59     const IUnknownVtbl *lpIUnknownVtbl;
  60     DispatchEx *obj;
  61     func_info_t *info;
  62 } func_disp_t;
  63
  64 #define FUNCUNKNOWN(x)  ((IUnknown*) &(x)->lpIUnknownVtbl)
  65
  66 struct dispex_dynamic_data_t {
  67     DWORD buf_size;
  68     DWORD prop_cnt;
  69     dynamic_prop_t *props;
  70     func_disp_t **func_disps;
  71 };
  72
  73 #define DISPID_DYNPROP_0    0x50000000
  74 #define DISPID_DYNPROP_MAX  0x5fffffff
  75
  76 #define FDEX_VERSION_MASK 0xf0000000
  77
  78 static ITypeLib *typelib;
  79 static ITypeInfo *typeinfos[LAST_tid];
  80 static struct list dispex_data_list = LIST_INIT(dispex_data_list);
  81
  82 static REFIID tid_ids[] = {
  83     &IID_NULL,
  84     &DIID_DispCEventObj,
  85     &DIID_DispDOMChildrenCollection,
  86     &DIID_DispHTMLAnchorElement,
  87     &DIID_DispHTMLBody,
  88     &DIID_DispHTMLCommentElement,
  89     &DIID_DispHTMLCurrentStyle,
  90     &DIID_DispHTMLDocument,
  91     &DIID_DispHTMLDOMTextNode,
  92     &DIID_DispHTMLElementCollection,
  93     &DIID_DispHTMLFormElement,
  94     &DIID_DispHTMLGenericElement,
  95     &DIID_DispHTMLFrameElement,
  96     &DIID_DispHTMLIFrame,
  97     &DIID_DispHTMLImg,
  98     &DIID_DispHTMLInputElement,
  99     &DIID_DispHTMLLocation,
 100     &DIID_DispHTMLNavigator,
 101     &DIID_DispHTMLOptionElement,
 102     &DIID_DispHTMLScreen,
 103     &DIID_DispHTMLScriptElement,
 104     &DIID_DispHTMLSelectElement,
 105     &DIID_DispHTMLStyle,
 106     &DIID_DispHTMLTable,
 107     &DIID_DispHTMLTableRow,
 108     &DIID_DispHTMLUnknownElement,
 109     &DIID_DispHTMLWindow2,
 110     &DIID_HTMLDocumentEvents,
 111     &IID_IHTMLAnchorElement,
 112     &IID_IHTMLBodyElement,
 113     &IID_IHTMLBodyElement2,
 114     &IID_IHTMLCommentElement,
 115     &IID_IHTMLCurrentStyle,
 116     &IID_IHTMLCurrentStyle2,
 117     &IID_IHTMLCurrentStyle3,
 118     &IID_IHTMLCurrentStyle4,
 119     &IID_IHTMLDocument2,
 120     &IID_IHTMLDocument3,
 121     &IID_IHTMLDocument4,
 122     &IID_IHTMLDocument5,
 123     &IID_IHTMLDOMChildrenCollection,
 124     &IID_IHTMLDOMNode,
 125     &IID_IHTMLDOMNode2,
 126     &IID_IHTMLDOMTextNode,
 127     &IID_IHTMLElement,
 128     &IID_IHTMLElement2,
 129     &IID_IHTMLElement3,
 130     &IID_IHTMLElement4,
 131     &IID_IHTMLElementCollection,
 132     &IID_IHTMLEventObj,
 133     &IID_IHTMLFiltersCollection,
 134     &IID_IHTMLFormElement,
 135     &IID_IHTMLFrameBase,
 136     &IID_IHTMLFrameBase2,
 137     &IID_IHTMLGenericElement,
 138     &IID_IHTMLFrameElement3,
 139     &IID_IHTMLIFrameElement,
 140     &IID_IHTMLImageElementFactory,
 141     &IID_IHTMLImgElement,
 142     &IID_IHTMLInputElement,
 143     &IID_IHTMLLocation,
 144     &IID_IHTMLOptionElement,
 145     &IID_IHTMLScreen,
 146     &IID_IHTMLScriptElement,
 147     &IID_IHTMLSelectElement,
 148     &IID_IHTMLStyle,
 149     &IID_IHTMLStyle2,
 150     &IID_IHTMLStyle3,
 151     &IID_IHTMLStyle4,
 152     &IID_IHTMLTable,
 153     &IID_IHTMLTableRow,
 154     &IID_IHTMLTextContainer,
 155     &IID_IHTMLUniqueName,
 156     &IID_IHTMLWindow2,
 157     &IID_IHTMLWindow3,
 158     &IID_IHTMLWindow4,
 159     &IID_IOmNavigator
 160 };
 161
 162 static HRESULT get_typeinfo(tid_t tid, ITypeInfo **typeinfo)
 163 {
 164     HRESULT hres;
 165
 166     if(!typelib) {
 167         ITypeLib *tl;
 168
 169         hres = LoadRegTypeLib(&LIBID_MSHTML, 4, 0, LOCALE_SYSTEM_DEFAULT, &tl);
 170         if(FAILED(hres)) {
 171             ERR("LoadRegTypeLib failed: %08x\n", hres);
 172             return hres;
 173         }
 174
 175         if(InterlockedCompareExchangePointer((void**)&typelib, tl, NULL))
 176             ITypeLib_Release(tl);
 177     }
 178
 179     if(!typeinfos[tid]) {
 180         ITypeInfo *typeinfo;
 181
 182         hres = ITypeLib_GetTypeInfoOfGuid(typelib, tid_ids[tid], &typeinfo);
 183         if(FAILED(hres)) {
 184             ERR("GetTypeInfoOfGuid(%s) failed: %08x\n", debugstr_guid(tid_ids[tid]), hres);
 185             return hres;
 186         }
 187
 188         if(InterlockedCompareExchangePointer((void**)(typeinfos+tid), typeinfo, NULL))
 189             ITypeInfo_Release(typeinfo);
 190     }
 191
 192     *typeinfo = typeinfos[tid];
 193     return S_OK;
 194 }
 195
 196 void release_typelib(void)
 197 {
 198     dispex_data_t *iter;
 199     unsigned i;
 200
 201     while(!list_empty(&dispex_data_list)) {
 202         iter = LIST_ENTRY(list_head(&dispex_data_list), dispex_data_t, entry);
 203         list_remove(&iter->entry);
 204
 205         for(i=0; i < iter->func_cnt; i++)
 206             SysFreeString(iter->funcs[i].name);
 207
 208         heap_free(iter->funcs);
 209         heap_free(iter->name_table);
 210         heap_free(iter);
 211     }
 212
 213     if(!typelib)
 214         return;
 215
 216     for(i=0; i < sizeof(typeinfos)/sizeof(*typeinfos); i++)
 217         if(typeinfos[i])
 218             ITypeInfo_Release(typeinfos[i]);
 219
 220     ITypeLib_Release(typelib);
 221 }
 222
 223 static void add_func_info(dispex_data_t *data, DWORD *size, tid_t tid, const FUNCDESC *desc, ITypeInfo *dti)
 224 {
 225     HRESULT hres;
 226
 227     if(data->func_cnt && data->funcs[data->func_cnt-1].id == desc->memid)
 228         return;
 229
 230     if(data->func_cnt == *size)
 231         data->funcs = heap_realloc(data->funcs, (*size <<= 1)*sizeof(func_info_t));
 232
 233     hres = ITypeInfo_GetDocumentation(dti, desc->memid, &data->funcs[data->func_cnt].name, NULL, NULL, NULL);
 234     if(FAILED(hres))
 235         return;
 236
 237     data->funcs[data->func_cnt].id = desc->memid;
 238     data->funcs[data->func_cnt].tid = tid;
 239     data->funcs[data->func_cnt].func_disp_idx = (desc->invkind & DISPATCH_METHOD) ? data->func_disp_cnt++ : -1;
 240
 241     data->func_cnt++;
 242 }
 243
 244 static int dispid_cmp(const void *p1, const void *p2)
 245 {
 246     return ((func_info_t*)p1)->id - ((func_info_t*)p2)->id;
 247 }
 248
 249 static int func_name_cmp(const void *p1, const void *p2)
 250 {
 251     return strcmpiW((*(func_info_t**)p1)->name, (*(func_info_t**)p2)->name);
 252 }
 253
 254 static dispex_data_t *preprocess_dispex_data(DispatchEx *This)
 255 {
 256     const tid_t *tid = This->data->iface_tids;
 257     FUNCDESC *funcdesc;
 258     dispex_data_t *data;
 259     DWORD size = 16, i;
 260     ITypeInfo *ti, *dti;
 261     HRESULT hres;
 262
 263     TRACE("(%p)\n", This);
 264
 265     if(This->data->disp_tid) {
 266         hres = get_typeinfo(This->data->disp_tid, &dti);
 267         if(FAILED(hres)) {
 268             ERR("Could not get disp type info: %08x\n", hres);
 269             return NULL;
 270         }
 271     }
 272
 273     data = heap_alloc(sizeof(dispex_data_t));
 274     data->func_cnt = 0;
 275     data->func_disp_cnt = 0;
 276     data->funcs = heap_alloc(size*sizeof(func_info_t));
 277     list_add_tail(&dispex_data_list, &data->entry);
 278
 279     while(*tid) {
 280         hres = get_typeinfo(*tid, &ti);
 281         if(FAILED(hres))
 282             break;
 283
 284         i=7;
 285         while(1) {
 286             hres = ITypeInfo_GetFuncDesc(ti, i++, &funcdesc);
 287             if(FAILED(hres))
 288                 break;
 289
 290             add_func_info(data, &size, *tid, funcdesc, dti);
 291             ITypeInfo_ReleaseFuncDesc(ti, funcdesc);
 292         }
 293
 294         tid++;
 295     }
 296
 297     if(!data->func_cnt) {
 298         heap_free(data->funcs);
 299         data->funcs = NULL;
 300     }else if(data->func_cnt != size) {
 301         data->funcs = heap_realloc(data->funcs, data->func_cnt * sizeof(func_info_t));
 302     }
 303
 304     qsort(data->funcs, data->func_cnt, sizeof(func_info_t), dispid_cmp);
 305
 306     if(data->funcs) {
 307         data->name_table = heap_alloc(data->func_cnt * sizeof(func_info_t*));
 308         for(i=0; i < data->func_cnt; i++)
 309             data->name_table[i] = data->funcs+i;
 310         qsort(data->name_table, data->func_cnt, sizeof(func_info_t*), func_name_cmp);
 311     }else {
 312         data->name_table = NULL;
 313     }
 314
 315     return data;
 316 }
 317
 318 static int id_cmp(const void *p1, const void *p2)
 319 {
 320     return *(DISPID*)p1 - *(DISPID*)p2;
 321 }
 322
 323 HRESULT get_dispids(tid_t tid, DWORD *ret_size, DISPID **ret)
 324 {
 325     unsigned i, func_cnt;
 326     FUNCDESC *funcdesc;
 327     ITypeInfo *ti;
 328     TYPEATTR *attr;
 329     DISPID *ids;
 330     HRESULT hres;
 331
 332     hres = get_typeinfo(tid, &ti);
 333     if(FAILED(hres))
 334         return hres;
 335
 336     hres = ITypeInfo_GetTypeAttr(ti, &attr);
 337     if(FAILED(hres)) {
 338         ITypeInfo_Release(ti);
 339         return hres;
 340     }
 341
 342     func_cnt = attr->cFuncs;
 343     ITypeInfo_ReleaseTypeAttr(ti, attr);
 344
 345     ids = heap_alloc(func_cnt*sizeof(DISPID));
 346     if(!ids) {
 347         ITypeInfo_Release(ti);
 348         return E_OUTOFMEMORY;
 349     }
 350
 351     for(i=0; i < func_cnt; i++) {
 352         hres = ITypeInfo_GetFuncDesc(ti, i, &funcdesc);
 353         if(FAILED(hres))
 354             break;
 355
 356         ids[i] = funcdesc->memid;
 357         ITypeInfo_ReleaseFuncDesc(ti, funcdesc);
 358     }
 359
 360     ITypeInfo_Release(ti);
 361     if(FAILED(hres)) {
 362         heap_free(ids);
 363         return hres;
 364     }
 365
 366     qsort(ids, func_cnt, sizeof(DISPID), id_cmp);
 367
 368     *ret_size = func_cnt;
 369     *ret = ids;
 370     return S_OK;
 371 }
 372
 373 static CRITICAL_SECTION cs_dispex_static_data;
 374 static CRITICAL_SECTION_DEBUG cs_dispex_static_data_dbg =
 375 {
 376     0, 0, &cs_dispex_static_data,
 377     { &cs_dispex_static_data_dbg.ProcessLocksList, &cs_dispex_static_data_dbg.ProcessLocksList },
 378       0, 0, { (DWORD_PTR)(__FILE__ ": dispex_static_data") }
 379 };
 380 static CRITICAL_SECTION cs_dispex_static_data = { &cs_dispex_static_data_dbg, -1, 0, 0, 0, 0 };
 381
 382
 383 static dispex_data_t *get_dispex_data(DispatchEx *This)
 384 {
 385     if(This->data->data)
 386         return This->data->data;
 387
 388     EnterCriticalSection(&cs_dispex_static_data);
 389
 390     if(!This->data->data)
 391         This->data->data = preprocess_dispex_data(This);
 392
 393     LeaveCriticalSection(&cs_dispex_static_data);
 394
 395     return This->data->data;
 396 }
 397
 398 HRESULT call_disp_func(IDispatch *disp, DISPPARAMS *dp)
 399 {
 400     EXCEPINFO ei;
 401     IDispatchEx *dispex;
 402     VARIANT res;
 403     HRESULT hres;
 404
 405     VariantInit(&res);
 406     memset(&ei, 0, sizeof(ei));
 407
 408     hres = IDispatch_QueryInterface(disp, &IID_IDispatchEx, (void**)&dispex);
 409     if(SUCCEEDED(hres)) {
 410         hres = IDispatchEx_InvokeEx(dispex, 0, GetUserDefaultLCID(), DISPATCH_METHOD, dp, &res, &ei, NULL);
 411         IDispatchEx_Release(dispex);
 412     }else {
 413         TRACE("Could not get IDispatchEx interface: %08x\n", hres);
 414         hres = IDispatch_Invoke(disp, 0, &IID_NULL, GetUserDefaultLCID(), DISPATCH_METHOD,
 415                 dp, &res, &ei, NULL);
 416     }
 417
 418     VariantClear(&res);
 419     return hres;
 420 }
 421
 422 static inline BOOL is_custom_dispid(DISPID id)
 423 {
 424     return MSHTML_DISPID_CUSTOM_MIN <= id && id <= MSHTML_DISPID_CUSTOM_MAX;
 425 }
 426
 427 static inline BOOL is_dynamic_dispid(DISPID id)
 428 {
 429     return DISPID_DYNPROP_0 <= id && id <= DISPID_DYNPROP_MAX;
 430 }
 431
 432 static inline dispex_dynamic_data_t *get_dynamic_data(DispatchEx *This, BOOL alloc)
 433 {
 434     return !alloc || This->dynamic_data
 435         ? This->dynamic_data
 436         : (This->dynamic_data = heap_alloc_zero(sizeof(dispex_dynamic_data_t)));
 437 }
 438
 439 static HRESULT get_dynamic_prop(DispatchEx *This, const WCHAR *name, DWORD flags, dynamic_prop_t **ret)
 440 {
 441     const BOOL alloc = flags & fdexNameEnsure;
 442     dispex_dynamic_data_t *data;
 443     unsigned i;
 444
 445     data = get_dynamic_data(This, alloc);
 446     if(!data) {
 447         if(alloc)
 448             return E_OUTOFMEMORY;
 449
 450         TRACE("not found %s\n", debugstr_w(name));
 451         return DISP_E_UNKNOWNNAME;
 452     }
 453
 454     for(i=0; i < data->prop_cnt; i++) {
 455         if(flags & fdexNameCaseInsensitive ? !strcmpiW(data->props[i].name, name) : !strcmpW(data->props[i].name, name)) {
 456             *ret = data->props+i;
 457             return S_OK;
 458         }
 459     }
 460
 461     if(!alloc)
 462         return DISP_E_UNKNOWNNAME;
 463
 464     TRACE("creating dynamic prop %s\n", debugstr_w(name));
 465
 466     if(!data->buf_size) {
 467         data->props = heap_alloc(sizeof(dynamic_prop_t)*4);
 468         if(!data->props)
 469             return E_OUTOFMEMORY;
 470         data->buf_size = 4;
 471     }else if(data->buf_size == data->prop_cnt) {
 472         dynamic_prop_t *new_props;
 473
 474         new_props = heap_realloc(data->props, sizeof(dynamic_prop_t)*(data->buf_size<<1));
 475         if(!new_props)
 476             return E_OUTOFMEMORY;
 477
 478         data->props = new_props;
 479         data->buf_size <<= 1;
 480     }
 481
 482     data->props[data->prop_cnt].name = heap_strdupW(name);
 483     VariantInit(&data->props[data->prop_cnt].var);
 484     *ret = data->props + data->prop_cnt++;
 485
 486     return S_OK;
 487 }
 488
 489 HRESULT dispex_get_dprop_ref(DispatchEx *This, const WCHAR *name, BOOL alloc, VARIANT **ret)
 490 {
 491     dynamic_prop_t *prop;
 492     HRESULT hres;
 493
 494     hres = get_dynamic_prop(This, name, alloc ? fdexNameEnsure : 0, &prop);
 495     if(FAILED(hres))
 496         return hres;
 497
 498     *ret = &prop->var;
 499     return S_OK;
 500 }
 501
 502 static HRESULT dispex_value(DispatchEx *This, LCID lcid, WORD flags, DISPPARAMS *params,
 503         VARIANT *res, EXCEPINFO *ei, IServiceProvider *caller)
 504 {
 505     if(This->data->vtbl && This->data->vtbl->value)
 506         return This->data->vtbl->value(This->outer, lcid, flags, params, res, ei, caller);
 507
 508     switch(flags) {
 509     case DISPATCH_PROPERTYGET:
 510         V_VT(res) = VT_BSTR;
 511         V_BSTR(res) = SysAllocString(objectW);
 512         if(!V_BSTR(res))
 513             return E_OUTOFMEMORY;
 514         break;
 515     default:
 516         FIXME("Unimplemented flags %x\n", flags);
 517         return E_NOTIMPL;
 518     }
 519
 520     return S_OK;
 521 }
 522
 523 static HRESULT typeinfo_invoke(DispatchEx *This, func_info_t *func, WORD flags, DISPPARAMS *dp, VARIANT *res,
 524         EXCEPINFO *ei)
 525 {
 526     ITypeInfo *ti;
 527     IUnknown *unk;
 528     UINT argerr=0;
 529     HRESULT hres;
 530
 531     hres = get_typeinfo(func->tid, &ti);
 532     if(FAILED(hres)) {
 533         ERR("Could not get type info: %08x\n", hres);
 534         return hres;
 535     }
 536
 537     hres = IUnknown_QueryInterface(This->outer, tid_ids[func->tid], (void**)&unk);
 538     if(FAILED(hres)) {
 539         ERR("Could not get iface %s: %08x\n", debugstr_guid(tid_ids[func->tid]), hres);
 540         return E_FAIL;
 541     }
 542
 543     hres = ITypeInfo_Invoke(ti, unk, func->id, flags, dp, res, ei, &argerr);
 544
 545     IUnknown_Release(unk);
 546     return hres;
 547 }
 548
 549 #define FUNCTION_THIS(iface) DEFINE_THIS(func_disp_t, IUnknown, iface)
 550
 551 static HRESULT WINAPI Function_QueryInterface(IUnknown *iface, REFIID riid, void **ppv)
 552 {
 553     func_disp_t *This = FUNCTION_THIS(iface);
 554
 555     if(IsEqualGUID(&IID_IUnknown, riid)) {
 556         TRACE("(%p)->(IID_IUnknown %p)\n", This, ppv);
 557         *ppv = FUNCUNKNOWN(This);
 558     }else if(dispex_query_interface(&This->dispex, riid, ppv)) {
 559         return *ppv ? S_OK : E_NOINTERFACE;
 560     }else {
 561         *ppv = NULL;
 562         return E_NOINTERFACE;
 563     }
 564
 565     IUnknown_AddRef((IUnknown*)*ppv);
 566     return S_OK;
 567 }
 568
 569 static ULONG WINAPI Function_AddRef(IUnknown *iface)
 570 {
 571     func_disp_t *This = FUNCTION_THIS(iface);
 572
 573     TRACE("(%p)\n", This);
 574
 575     return IDispatchEx_AddRef(DISPATCHEX(This->obj));
 576 }
 577
 578 static ULONG WINAPI Function_Release(IUnknown *iface)
 579 {
 580     func_disp_t *This = FUNCTION_THIS(iface);
 581
 582     TRACE("(%p)\n", This);
 583
 584     return IDispatchEx_Release(DISPATCHEX(This->obj));
 585 }
 586
 587 static HRESULT function_value(IUnknown *iface, LCID lcid, WORD flags, DISPPARAMS *params,
 588         VARIANT *res, EXCEPINFO *ei, IServiceProvider *caller)
 589 {
 590     func_disp_t *This = FUNCTION_THIS(iface);
 591     HRESULT hres;
 592
 593     switch(flags) {
 594     case DISPATCH_METHOD|DISPATCH_PROPERTYGET:
 595         if(!res)
 596             return E_INVALIDARG;
 597     case DISPATCH_METHOD:
 598         hres = typeinfo_invoke(This->obj, This->info, flags, params, res, ei);
 599         break;
 600     default:
 601         FIXME("Unimplemented flags %x\n", flags);
 602         hres = E_NOTIMPL;
 603     }
 604
 605     return hres;
 606 }
 607
 608 #undef FUNCTION_THIS
 609
 610 static const IUnknownVtbl FunctionUnkVtbl = {
 611     Function_QueryInterface,
 612     Function_AddRef,
 613     Function_Release
 614 };
 615
 616 static const dispex_static_data_vtbl_t function_dispex_vtbl = {
 617     function_value,
 618     NULL,
 619     NULL
 620 };
 621
 622 static const tid_t function_iface_tids[] = {0};
 623
 624 static dispex_static_data_t function_dispex = {
 625     &function_dispex_vtbl,
 626     NULL_tid,
 627     NULL,
 628     function_iface_tids
 629 };
 630
 631 static func_disp_t *create_func_disp(DispatchEx *obj, func_info_t *info)
 632 {
 633     func_disp_t *ret;
 634
 635     ret = heap_alloc_zero(sizeof(func_disp_t));
 636     if(!ret)
 637         return NULL;
 638
 639     ret->lpIUnknownVtbl = &FunctionUnkVtbl;
 640     init_dispex(&ret->dispex, FUNCUNKNOWN(ret),  &function_dispex);
 641     ret->obj = obj;
 642     ret->info = info;
 643
 644     return ret;
 645 }
 646
 647 static HRESULT function_invoke(DispatchEx *This, func_info_t *func, WORD flags, DISPPARAMS *dp, VARIANT *res,
 648         EXCEPINFO *ei)
 649 {
 650     HRESULT hres;
 651
 652     switch(flags) {
 653     case DISPATCH_METHOD|DISPATCH_PROPERTYGET:
 654         if(!res)
 655             return E_INVALIDARG;
 656     case DISPATCH_METHOD:
 657         hres = typeinfo_invoke(This, func, flags, dp, res, ei);
 658         break;
 659     case DISPATCH_PROPERTYGET: {
 660         dispex_dynamic_data_t *dynamic_data;
 661
 662         if(func->id == DISPID_VALUE) {
 663             BSTR ret;
 664
 665             ret = SysAllocString(objectW);
 666             if(!ret)
 667                 return E_OUTOFMEMORY;
 668
 669             V_VT(res) = VT_BSTR;
 670             V_BSTR(res) = ret;
 671             return S_OK;
 672         }
 673
 674         dynamic_data = get_dynamic_data(This, TRUE);
 675         if(!dynamic_data)
 676             return E_OUTOFMEMORY;
 677
 678         if(!dynamic_data->func_disps) {
 679             dynamic_data->func_disps = heap_alloc_zero(This->data->data->func_disp_cnt * sizeof(func_disp_t*));
 680             if(!dynamic_data->func_disps)
 681                 return E_OUTOFMEMORY;
 682         }
 683
 684         if(!dynamic_data->func_disps[func->func_disp_idx]) {
 685             dynamic_data->func_disps[func->func_disp_idx] = create_func_disp(This, func);
 686             if(!dynamic_data->func_disps[func->func_disp_idx])
 687                 return E_OUTOFMEMORY;
 688         }
 689
 690         V_VT(res) = VT_DISPATCH;
 691         V_DISPATCH(res) = (IDispatch*)DISPATCHEX(&dynamic_data->func_disps[func->func_disp_idx]->dispex);
 692         IDispatch_AddRef(V_DISPATCH(res));
 693         hres = S_OK;
 694         break;
 695     }
 696     default:
 697         FIXME("Unimplemented flags %x\n", flags);
 698     case DISPATCH_PROPERTYPUT:
 699         hres = E_NOTIMPL;
 700     }
 701
 702     return hres;
 703 }
 704
 705 static HRESULT get_builtin_func(dispex_data_t *data, DISPID id, func_info_t **ret)
 706 {
 707     int min, max, n;
 708
 709     min = 0;
 710     max = data->func_cnt-1;
 711
 712     while(min <= max) {
 713         n = (min+max)/2;
 714
 715         if(data->funcs[n].id == id) {
 716             *ret = data->funcs+n;
 717             return S_OK;
 718         }
 719
 720         if(data->funcs[n].id < id)
 721             min = n+1;
 722         else
 723             max = n-1;
 724     }
 725
 726     WARN("invalid id %x\n", id);
 727     return DISP_E_UNKNOWNNAME;
 728 }
 729
 730 #define DISPATCHEX_THIS(iface) DEFINE_THIS(DispatchEx, IDispatchEx, iface)
 731
 732 static HRESULT WINAPI DispatchEx_QueryInterface(IDispatchEx *iface, REFIID riid, void **ppv)
 733 {
 734     DispatchEx *This = DISPATCHEX_THIS(iface);
 735
 736     return IUnknown_QueryInterface(This->outer, riid, ppv);
 737 }
 738
 739 static ULONG WINAPI DispatchEx_AddRef(IDispatchEx *iface)
 740 {
 741     DispatchEx *This = DISPATCHEX_THIS(iface);
 742
 743     return IUnknown_AddRef(This->outer);
 744 }
 745
 746 static ULONG WINAPI DispatchEx_Release(IDispatchEx *iface)
 747 {
 748     DispatchEx *This = DISPATCHEX_THIS(iface);
 749
 750     return IUnknown_Release(This->outer);
 751 }
 752
 753 static HRESULT WINAPI DispatchEx_GetTypeInfoCount(IDispatchEx *iface, UINT *pctinfo)
 754 {
 755     DispatchEx *This = DISPATCHEX_THIS(iface);
 756
 757     TRACE("(%p)->(%p)\n", This, pctinfo);
 758
 759     *pctinfo = 1;
 760     return S_OK;
 761 }
 762
 763 static HRESULT WINAPI DispatchEx_GetTypeInfo(IDispatchEx *iface, UINT iTInfo,
 764                                               LCID lcid, ITypeInfo **ppTInfo)
 765 {
 766     DispatchEx *This = DISPATCHEX_THIS(iface);
 767     HRESULT hres;
 768
 769     TRACE("(%p)->(%u %u %p)\n", This, iTInfo, lcid, ppTInfo);
 770
 771     hres = get_typeinfo(This->data->disp_tid, ppTInfo);
 772     if(FAILED(hres))
 773         return hres;
 774
 775     ITypeInfo_AddRef(*ppTInfo);
 776     return S_OK;
 777 }
 778
 779 static HRESULT WINAPI DispatchEx_GetIDsOfNames(IDispatchEx *iface, REFIID riid,
 780                                                 LPOLESTR *rgszNames, UINT cNames,
 781                                                 LCID lcid, DISPID *rgDispId)
 782 {
 783     DispatchEx *This = DISPATCHEX_THIS(iface);
 784     UINT i;
 785     HRESULT hres;
 786
 787     TRACE("(%p)->(%s %p %u %u %p)\n", This, debugstr_guid(riid), rgszNames, cNames,
 788           lcid, rgDispId);
 789
 790     for(i=0; i < cNames; i++) {
 791         hres = IDispatchEx_GetDispID(DISPATCHEX(This), rgszNames[i], 0, rgDispId+i);
 792         if(FAILED(hres))
 793             return hres;
 794     }
 795
 796     return S_OK;
 797 }
 798
 799 static HRESULT WINAPI DispatchEx_Invoke(IDispatchEx *iface, DISPID dispIdMember,
 800                             REFIID riid, LCID lcid, WORD wFlags, DISPPARAMS *pDispParams,
 801                             VARIANT *pVarResult, EXCEPINFO *pExcepInfo, UINT *puArgErr)
 802 {
 803     DispatchEx *This = DISPATCHEX_THIS(iface);
 804
 805     TRACE("(%p)->(%d %s %d %d %p %p %p %p)\n", This, dispIdMember, debugstr_guid(riid),
 806           lcid, wFlags, pDispParams, pVarResult, pExcepInfo, puArgErr);
 807
 808     return IDispatchEx_InvokeEx(DISPATCHEX(This), dispIdMember, lcid, wFlags,
 809                                 pDispParams, pVarResult, pExcepInfo, NULL);
 810 }
 811
 812 static HRESULT WINAPI DispatchEx_GetDispID(IDispatchEx *iface, BSTR bstrName, DWORD grfdex, DISPID *pid)
 813 {
 814     DispatchEx *This = DISPATCHEX_THIS(iface);
 815     dynamic_prop_t *dprop;
 816     dispex_data_t *data;
 817     int min, max, n, c;
 818     HRESULT hres;
 819
 820     TRACE("(%p)->(%s %x %p)\n", This, debugstr_w(bstrName), grfdex, pid);
 821
 822     if(grfdex & ~(fdexNameCaseSensitive|fdexNameCaseInsensitive|fdexNameEnsure|fdexNameImplicit|FDEX_VERSION_MASK))
 823         FIXME("Unsupported grfdex %x\n", grfdex);
 824
 825     data = get_dispex_data(This);
 826     if(!data)
 827         return E_FAIL;
 828
 829     min = 0;
 830     max = data->func_cnt-1;
 831
 832     while(min <= max) {
 833         n = (min+max)/2;
 834
 835         c = strcmpiW(data->name_table[n]->name, bstrName);
 836         if(!c) {
 837             if((grfdex & fdexNameCaseSensitive) && strcmpW(data->name_table[n]->name, bstrName))
 838                 break;
 839
 840             *pid = data->name_table[n]->id;
 841             return S_OK;
 842         }
 843
 844         if(c > 0)
 845             max = n-1;
 846         else
 847             min = n+1;
 848     }
 849
 850     if(This->data->vtbl && This->data->vtbl->get_dispid) {
 851         HRESULT hres;
 852
 853         hres = This->data->vtbl->get_dispid(This->outer, bstrName, grfdex, pid);
 854         if(hres != DISP_E_UNKNOWNNAME)
 855             return hres;
 856     }
 857
 858     hres = get_dynamic_prop(This, bstrName, grfdex, &dprop);
 859     if(FAILED(hres))
 860         return hres;
 861
 862     *pid = DISPID_DYNPROP_0 + (dprop - This->dynamic_data->props);
 863     return S_OK;
 864 }
 865
 866 static HRESULT WINAPI DispatchEx_InvokeEx(IDispatchEx *iface, DISPID id, LCID lcid, WORD wFlags, DISPPARAMS *pdp,
 867         VARIANT *pvarRes, EXCEPINFO *pei, IServiceProvider *pspCaller)
 868 {
 869     DispatchEx *This = DISPATCHEX_THIS(iface);
 870     dispex_data_t *data;
 871     func_info_t *func;
 872     HRESULT hres;
 873
 874     TRACE("(%p)->(%x %x %x %p %p %p %p)\n", This, id, lcid, wFlags, pdp, pvarRes, pei, pspCaller);
 875
 876     if(is_custom_dispid(id) && This->data->vtbl && This->data->vtbl->invoke)
 877         return This->data->vtbl->invoke(This->outer, id, lcid, wFlags, pdp, pvarRes, pei, pspCaller);
 878
 879     if(wFlags == DISPATCH_CONSTRUCT) {
 880         if(id == DISPID_VALUE) {
 881             if(This->data->vtbl && This->data->vtbl->value) {
 882                 return This->data->vtbl->value(This->outer, lcid, wFlags, pdp,
 883                         pvarRes, pei, pspCaller);
 884             }
 885             FIXME("DISPATCH_CONSTRUCT flag but missing value function\n");
 886             return E_FAIL;
 887         }
 888         FIXME("DISPATCH_CONSTRUCT flag without DISPID_VALUE\n");
 889         return E_FAIL;
 890     }
 891
 892     if(is_dynamic_dispid(id)) {
 893         DWORD idx = id - DISPID_DYNPROP_0;
 894         VARIANT *var;
 895
 896         if(!This->dynamic_data || This->dynamic_data->prop_cnt <= idx)
 897             return DISP_E_UNKNOWNNAME;
 898
 899         var = &This->dynamic_data->props[idx].var;
 900
 901         switch(wFlags) {
 902         case DISPATCH_METHOD|DISPATCH_PROPERTYGET:
 903             if(!pvarRes)
 904                 return E_INVALIDARG;
 905         case DISPATCH_METHOD: {
 906             DISPID named_arg = DISPID_THIS;
 907             DISPPARAMS dp = {NULL, &named_arg, 0, 1};
 908             IDispatchEx *dispex;
 909
 910             if(V_VT(var) != VT_DISPATCH) {
 911                 FIXME("invoke vt %d\n", V_VT(var));
 912                 return E_NOTIMPL;
 913             }
 914
 915             if(pdp->cNamedArgs) {
 916                 FIXME("named args not supported\n");
 917                 return E_NOTIMPL;
 918             }
 919
 920             dp.rgvarg = heap_alloc((pdp->cArgs+1)*sizeof(VARIANTARG));
 921             if(!dp.rgvarg)
 922                 return E_OUTOFMEMORY;
 923
 924             dp.cArgs = pdp->cArgs+1;
 925             memcpy(dp.rgvarg+1, pdp->rgvarg, pdp->cArgs*sizeof(VARIANTARG));
 926
 927             V_VT(dp.rgvarg) = VT_DISPATCH;
 928             V_DISPATCH(dp.rgvarg) = (IDispatch*)DISPATCHEX(This);
 929
 930             hres = IDispatch_QueryInterface(V_DISPATCH(var), &IID_IDispatchEx, (void**)&dispex);
 931             TRACE("%s call\n", debugstr_w(This->dynamic_data->props[idx].name));
 932             if(SUCCEEDED(hres)) {
 933                 hres = IDispatchEx_InvokeEx(dispex, DISPID_VALUE, lcid, wFlags, &dp, pvarRes, pei, pspCaller);
 934                 IDispatchEx_Release(dispex);
 935             }else {
 936                 ULONG err = 0;
 937                 hres = IDispatch_Invoke(V_DISPATCH(var), DISPID_VALUE, &IID_NULL, lcid, wFlags, pdp, pvarRes, pei, &err);
 938             }
 939             TRACE("%s ret %08x\n", debugstr_w(This->dynamic_data->props[idx].name), hres);
 940
 941             heap_free(dp.rgvarg);
 942             return hres;
 943         }
 944         case DISPATCH_PROPERTYGET:
 945             return VariantCopy(pvarRes, var);
 946         case DISPATCH_PROPERTYPUT|DISPATCH_PROPERTYPUTREF:
 947         case DISPATCH_PROPERTYPUT:
 948             if(pdp->cArgs != 1 || (pdp->cNamedArgs == 1 && *pdp->rgdispidNamedArgs != DISPID_PROPERTYPUT)
 949                || pdp->cNamedArgs > 1) {
 950                 FIXME("invalid args\n");
 951                 return E_INVALIDARG;
 952             }
 953
 954             TRACE("put %s\n", debugstr_variant(pdp->rgvarg));
 955             VariantClear(var);
 956             return VariantCopy(var, pdp->rgvarg);
 957         default:
 958             FIXME("unhandled wFlags %x\n", wFlags);
 959             return E_NOTIMPL;
 960         }
 961     }
 962
 963     data = get_dispex_data(This);
 964     if(!data)
 965         return E_FAIL;
 966
 967     hres = get_builtin_func(data, id, &func);
 968     if(id == DISPID_VALUE && hres == DISP_E_UNKNOWNNAME)
 969         return dispex_value(This, lcid, wFlags, pdp, pvarRes, pei, pspCaller);
 970     if(FAILED(hres))
 971         return hres;
 972
 973     if(func->func_disp_idx == -1)
 974         hres = typeinfo_invoke(This, func, wFlags, pdp, pvarRes, pei);
 975     else
 976         hres = function_invoke(This, func, wFlags, pdp, pvarRes, pei);
 977
 978     return hres;
 979 }
 980
 981 static HRESULT WINAPI DispatchEx_DeleteMemberByName(IDispatchEx *iface, BSTR bstrName, DWORD grfdex)
 982 {
 983     DispatchEx *This = DISPATCHEX_THIS(iface);
 984     FIXME("(%p)->(%s %x)\n", This, debugstr_w(bstrName), grfdex);
 985     return E_NOTIMPL;
 986 }
 987
 988 static HRESULT WINAPI DispatchEx_DeleteMemberByDispID(IDispatchEx *iface, DISPID id)
 989 {
 990     DispatchEx *This = DISPATCHEX_THIS(iface);
 991     FIXME("(%p)->(%x)\n", This, id);
 992     return E_NOTIMPL;
 993 }
 994
 995 static HRESULT WINAPI DispatchEx_GetMemberProperties(IDispatchEx *iface, DISPID id, DWORD grfdexFetch, DWORD *pgrfdex)
 996 {
 997     DispatchEx *This = DISPATCHEX_THIS(iface);
 998     FIXME("(%p)->(%x %x %p)\n", This, id, grfdexFetch, pgrfdex);
 999     return E_NOTIMPL;
1000 }
1001
1002 static HRESULT WINAPI DispatchEx_GetMemberName(IDispatchEx *iface, DISPID id, BSTR *pbstrName)
1003 {
1004     DispatchEx *This = DISPATCHEX_THIS(iface);
1005     dispex_data_t *data;
1006     func_info_t *func;
1007     HRESULT hres;
1008
1009     TRACE("(%p)->(%x %p)\n", This, id, pbstrName);
1010
1011     if(is_dynamic_dispid(id)) {
1012         DWORD idx = id - DISPID_DYNPROP_0;
1013
1014         if(!This->dynamic_data || This->dynamic_data->prop_cnt <= idx)
1015             return DISP_E_UNKNOWNNAME;
1016
1017         *pbstrName = SysAllocString(This->dynamic_data->props[idx].name);
1018         if(!*pbstrName)
1019             return E_OUTOFMEMORY;
1020
1021         return S_OK;
1022     }
1023
1024     data = get_dispex_data(This);
1025     if(!data)
1026         return E_FAIL;
1027
1028     hres = get_builtin_func(data, id, &func);
1029     if(FAILED(hres))
1030         return hres;
1031
1032     *pbstrName = SysAllocString(func->name);
1033     if(!*pbstrName)
1034         return E_OUTOFMEMORY;
1035     return S_OK;
1036 }
1037
1038 static HRESULT WINAPI DispatchEx_GetNextDispID(IDispatchEx *iface, DWORD grfdex, DISPID id, DISPID *pid)
1039 {
1040     DispatchEx *This = DISPATCHEX_THIS(iface);
1041     dispex_data_t *data;
1042     func_info_t *func;
1043     HRESULT hres;
1044
1045     TRACE("(%p)->(%x %x %p)\n", This, grfdex, id, pid);
1046
1047     if(is_dynamic_dispid(id)) {
1048         DWORD idx = id - DISPID_DYNPROP_0;
1049
1050         if(!This->dynamic_data || This->dynamic_data->prop_cnt <= idx)
1051             return DISP_E_UNKNOWNNAME;
1052
1053         if(idx+1 == This->dynamic_data->prop_cnt) {
1054             *pid = DISPID_STARTENUM;
1055             return S_FALSE;
1056         }
1057
1058         *pid = id+1;
1059         return S_OK;
1060     }
1061
1062     data = get_dispex_data(This);
1063     if(!data)
1064         return E_FAIL;
1065
1066     if(id == DISPID_STARTENUM) {
1067         func = data->funcs;
1068     }else {
1069         hres = get_builtin_func(data, id, &func);
1070         if(FAILED(hres))
1071             return hres;
1072         func++;
1073     }
1074
1075     while(func < data->funcs+data->func_cnt) {
1076         /* FIXME: Skip hidden properties */
1077         if(func->func_disp_idx == -1) {
1078             *pid = func->id;
1079             return S_OK;
1080         }
1081         func++;
1082     }
1083
1084     if(This->dynamic_data && This->dynamic_data->prop_cnt) {
1085         *pid = DISPID_DYNPROP_0;
1086         return S_OK;
1087     }
1088
1089     *pid = DISPID_STARTENUM;
1090     return S_FALSE;
1091 }
1092
1093 static HRESULT WINAPI DispatchEx_GetNameSpaceParent(IDispatchEx *iface, IUnknown **ppunk)
1094 {
1095     DispatchEx *This = DISPATCHEX_THIS(iface);
1096     FIXME("(%p)->(%p)\n", This, ppunk);
1097     return E_NOTIMPL;
1098 }
1099
1100 #undef DISPATCHEX_THIS
1101
1102 static IDispatchExVtbl DispatchExVtbl = {
1103     DispatchEx_QueryInterface,
1104     DispatchEx_AddRef,
1105     DispatchEx_Release,
1106     DispatchEx_GetTypeInfoCount,
1107     DispatchEx_GetTypeInfo,
1108     DispatchEx_GetIDsOfNames,
1109     DispatchEx_Invoke,
1110     DispatchEx_GetDispID,
1111     DispatchEx_InvokeEx,
1112     DispatchEx_DeleteMemberByName,
1113     DispatchEx_DeleteMemberByDispID,
1114     DispatchEx_GetMemberProperties,
1115     DispatchEx_GetMemberName,
1116     DispatchEx_GetNextDispID,
1117     DispatchEx_GetNameSpaceParent
1118 };
1119
1120 BOOL dispex_query_interface(DispatchEx *This, REFIID riid, void **ppv)
1121 {
1122     static const IID IID_UndocumentedScriptIface =
1123         {0x719c3050,0xf9d3,0x11cf,{0xa4,0x93,0x00,0x40,0x05,0x23,0xa8,0xa0}};
1124     static const IID IID_IDispatchJS =
1125         {0x719c3050,0xf9d3,0x11cf,{0xa4,0x93,0x00,0x40,0x05,0x23,0xa8,0xa6}};
1126
1127     if(IsEqualGUID(&IID_IDispatch, riid)) {
1128         TRACE("(%p)->(IID_IDispatch %p)\n", This, ppv);
1129         *ppv = DISPATCHEX(This);
1130     }else if(IsEqualGUID(&IID_IDispatchEx, riid)) {
1131         TRACE("(%p)->(IID_IDispatchEx %p)\n", This, ppv);
1132         *ppv = DISPATCHEX(This);
1133     }else if(IsEqualGUID(&IID_IDispatchJS, riid)) {
1134         TRACE("(%p)->(IID_IDispatchJS %p) returning NULL\n", This, ppv);
1135         *ppv = NULL;
1136     }else if(IsEqualGUID(&IID_UndocumentedScriptIface, riid)) {
1137         TRACE("(%p)->(IID_UndocumentedScriptIface %p) returning NULL\n", This, ppv);
1138         *ppv = NULL;
1139     }else {
1140         return FALSE;
1141     }
1142
1143     if(*ppv)
1144         IUnknown_AddRef((IUnknown*)*ppv);
1145     return TRUE;
1146 }
1147
1148 void release_dispex(DispatchEx *This)
1149 {
1150     dynamic_prop_t *prop;
1151
1152     if(!This->dynamic_data)
1153         return;
1154
1155     for(prop = This->dynamic_data->props; prop < This->dynamic_data->props + This->dynamic_data->prop_cnt; prop++) {
1156         VariantClear(&prop->var);
1157         heap_free(prop->name);
1158     }
1159
1160     heap_free(This->dynamic_data->props);
1161
1162     if(This->dynamic_data->func_disps) {
1163         unsigned i;
1164
1165         for(i=0; i < This->data->data->func_disp_cnt; i++) {
1166             if(This->dynamic_data->func_disps[i]) {
1167                 release_dispex(&This->dynamic_data->func_disps[i]->dispex);
1168                 heap_free(This->dynamic_data->func_disps[i]);
1169             }
1170         }
1171
1172         heap_free(This->dynamic_data->func_disps);
1173     }
1174
1175     heap_free(This->dynamic_data);
1176 }
1177
1178 void init_dispex(DispatchEx *dispex, IUnknown *outer, dispex_static_data_t *data)
1179 {
1180     dispex->lpIDispatchExVtbl = &DispatchExVtbl;
1181     dispex->outer = outer;
1182     dispex->data = data;
1183     dispex->dynamic_data = NULL;
1184 }