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