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