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