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