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