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