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