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