git.oblomov.eu Git - wine/blob - dlls/jscript/dispex.c

   1 /*
   2  * Copyright 2008 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 <assert.h>
  20
  21 #include "jscript.h"
  22
  23 #include "wine/unicode.h"
  24 #include "wine/debug.h"
  25
  26 WINE_DEFAULT_DEBUG_CHANNEL(jscript);
  27
  28 /*
  29  * This IID is used to get jsdisp_t objecto from interface.
  30  * We might consider using private insteface instead.
  31  */
  32 static const IID IID_IDispatchJS =
  33         {0x719c3050,0xf9d3,0x11cf,{0xa4,0x93,0x00,0x40,0x05,0x23,0xa8,0xa6}};
  34
  35 #define FDEX_VERSION_MASK 0xf0000000
  36 #define GOLDEN_RATIO 0x9E3779B9U
  37
  38 typedef enum {
  39     PROP_VARIANT,
  40     PROP_BUILTIN,
  41     PROP_PROTREF,
  42     PROP_DELETED
  43 } prop_type_t;
  44
  45 struct _dispex_prop_t {
  46     WCHAR *name;
  47     unsigned hash;
  48     prop_type_t type;
  49     DWORD flags;
  50
  51     union {
  52         VARIANT var;
  53         const builtin_prop_t *p;
  54         DWORD ref;
  55     } u;
  56
  57     int bucket_head;
  58     int bucket_next;
  59 };
  60
  61 static inline DISPID prop_to_id(jsdisp_t *This, dispex_prop_t *prop)
  62 {
  63     return prop - This->props;
  64 }
  65
  66 static inline dispex_prop_t *get_prop(jsdisp_t *This, DISPID id)
  67 {
  68     if(id < 0 || id >= This->prop_cnt || This->props[id].type == PROP_DELETED)
  69         return NULL;
  70
  71     return This->props+id;
  72 }
  73
  74 static DWORD get_flags(jsdisp_t *This, dispex_prop_t *prop)
  75 {
  76     if(prop->type == PROP_PROTREF) {
  77         dispex_prop_t *parent = get_prop(This->prototype, prop->u.ref);
  78         if(!parent) {
  79             prop->type = PROP_DELETED;
  80             return 0;
  81         }
  82
  83         return get_flags(This->prototype, parent);
  84     }
  85
  86     return prop->flags;
  87 }
  88
  89 static const builtin_prop_t *find_builtin_prop(jsdisp_t *This, const WCHAR *name)
  90 {
  91     int min = 0, max, i, r;
  92
  93     max = This->builtin_info->props_cnt-1;
  94     while(min <= max) {
  95         i = (min+max)/2;
  96
  97         r = strcmpW(name, This->builtin_info->props[i].name);
  98         if(!r)
  99             return This->builtin_info->props + i;
 100
 101         if(r < 0)
 102             max = i-1;
 103         else
 104             min = i+1;
 105     }
 106
 107     return NULL;
 108 }
 109
 110 static inline unsigned string_hash(const WCHAR *name)
 111 {
 112     unsigned h = 0;
 113     for(; *name; name++)
 114         h = (h>>(sizeof(unsigned)*8-4)) ^ (h<<4) ^ tolowerW(*name);
 115     return h;
 116 }
 117
 118 static inline unsigned get_props_idx(jsdisp_t *This, unsigned hash)
 119 {
 120     return (hash*GOLDEN_RATIO) & (This->buf_size-1);
 121 }
 122
 123 static inline HRESULT resize_props(jsdisp_t *This)
 124 {
 125     dispex_prop_t *props;
 126     int i, bucket;
 127
 128     if(This->buf_size != This->prop_cnt)
 129         return S_FALSE;
 130
 131     props = heap_realloc(This->props, sizeof(dispex_prop_t)*This->buf_size*2);
 132     if(!props)
 133         return E_OUTOFMEMORY;
 134     This->buf_size *= 2;
 135     This->props = props;
 136
 137     for(i=0; i<This->buf_size; i++) {
 138         This->props[i].bucket_head = 0;
 139         This->props[i].bucket_next = 0;
 140     }
 141
 142     for(i=1; i<This->prop_cnt; i++) {
 143         props = This->props+i;
 144
 145         bucket = get_props_idx(This, props->hash);
 146         props->bucket_next = This->props[bucket].bucket_head;
 147         This->props[bucket].bucket_head = i;
 148     }
 149
 150     return S_OK;
 151 }
 152
 153 static inline dispex_prop_t* alloc_prop(jsdisp_t *This, const WCHAR *name, prop_type_t type, DWORD flags)
 154 {
 155     dispex_prop_t *prop;
 156     unsigned bucket;
 157
 158     if(FAILED(resize_props(This)))
 159         return NULL;
 160
 161     prop = &This->props[This->prop_cnt];
 162     prop->name = heap_strdupW(name);
 163     if(!prop->name)
 164         return NULL;
 165     prop->type = type;
 166     prop->flags = flags;
 167     prop->hash = string_hash(name);
 168
 169     bucket = get_props_idx(This, prop->hash);
 170     prop->bucket_next = This->props[bucket].bucket_head;
 171     This->props[bucket].bucket_head = This->prop_cnt++;
 172     return prop;
 173 }
 174
 175 static dispex_prop_t *alloc_protref(jsdisp_t *This, const WCHAR *name, DWORD ref)
 176 {
 177     dispex_prop_t *ret;
 178
 179     ret = alloc_prop(This, name, PROP_PROTREF, 0);
 180     if(!ret)
 181         return NULL;
 182
 183     ret->u.ref = ref;
 184     return ret;
 185 }
 186
 187 static HRESULT find_prop_name(jsdisp_t *This, unsigned hash, const WCHAR *name, dispex_prop_t **ret)
 188 {
 189     const builtin_prop_t *builtin;
 190     unsigned bucket, pos, prev = 0;
 191     dispex_prop_t *prop;
 192
 193     bucket = get_props_idx(This, hash);
 194     pos = This->props[bucket].bucket_head;
 195     while(pos != 0) {
 196         if(!strcmpW(name, This->props[pos].name)) {
 197             if(prev != 0) {
 198                 This->props[prev].bucket_next = This->props[pos].bucket_next;
 199                 This->props[pos].bucket_next = This->props[bucket].bucket_head;
 200                 This->props[bucket].bucket_head = pos;
 201             }
 202
 203             *ret = &This->props[pos];
 204             return S_OK;
 205         }
 206
 207         prev = pos;
 208         pos = This->props[pos].bucket_next;
 209     }
 210
 211     builtin = find_builtin_prop(This, name);
 212     if(builtin) {
 213         prop = alloc_prop(This, name, PROP_BUILTIN, builtin->flags);
 214         if(!prop)
 215             return E_OUTOFMEMORY;
 216
 217         prop->u.p = builtin;
 218         *ret = prop;
 219         return S_OK;
 220     }
 221
 222     *ret = NULL;
 223     return S_OK;
 224 }
 225
 226 static HRESULT find_prop_name_prot(jsdisp_t *This, unsigned hash, const WCHAR *name, dispex_prop_t **ret)
 227 {
 228     dispex_prop_t *prop, *del=NULL;
 229     HRESULT hres;
 230
 231     hres = find_prop_name(This, hash, name, &prop);
 232     if(FAILED(hres))
 233         return hres;
 234     if(prop && prop->type==PROP_DELETED) {
 235         del = prop;
 236     } else if(prop) {
 237         *ret = prop;
 238         return S_OK;
 239     }
 240
 241     if(This->prototype) {
 242         hres = find_prop_name_prot(This->prototype, hash, name, &prop);
 243         if(FAILED(hres))
 244             return hres;
 245         if(prop) {
 246             if(del) {
 247                 del->type = PROP_PROTREF;
 248                 del->flags = 0;
 249                 del->u.ref = prop - This->prototype->props;
 250                 prop = del;
 251             }else {
 252                 prop = alloc_protref(This, prop->name, prop - This->prototype->props);
 253                 if(!prop)
 254                     return E_OUTOFMEMORY;
 255             }
 256
 257             *ret = prop;
 258             return S_OK;
 259         }
 260     }
 261
 262     *ret = del;
 263     return S_OK;
 264 }
 265
 266 static HRESULT ensure_prop_name(jsdisp_t *This, const WCHAR *name, BOOL search_prot, DWORD create_flags, dispex_prop_t **ret)
 267 {
 268     dispex_prop_t *prop;
 269     HRESULT hres;
 270
 271     if(search_prot)
 272         hres = find_prop_name_prot(This, string_hash(name), name, &prop);
 273     else
 274         hres = find_prop_name(This, string_hash(name), name, &prop);
 275     if(SUCCEEDED(hres) && (!prop || prop->type==PROP_DELETED)) {
 276         TRACE("creating prop %s\n", debugstr_w(name));
 277
 278         if(prop) {
 279             prop->type = PROP_VARIANT;
 280             prop->flags = create_flags;
 281         }else {
 282             prop = alloc_prop(This, name, PROP_VARIANT, create_flags);
 283             if(!prop)
 284                 return E_OUTOFMEMORY;
 285         }
 286
 287         VariantInit(&prop->u.var);
 288     }
 289
 290     *ret = prop;
 291     return hres;
 292 }
 293
 294 static HRESULT set_this(DISPPARAMS *dp, DISPPARAMS *olddp, IDispatch *jsthis)
 295 {
 296     VARIANTARG *oldargs;
 297     int i;
 298
 299     static DISPID this_id = DISPID_THIS;
 300
 301     *dp = *olddp;
 302
 303     for(i = 0; i < dp->cNamedArgs; i++) {
 304         if(dp->rgdispidNamedArgs[i] == DISPID_THIS)
 305             return S_OK;
 306     }
 307
 308     oldargs = dp->rgvarg;
 309     dp->rgvarg = heap_alloc((dp->cArgs+1) * sizeof(VARIANTARG));
 310     if(!dp->rgvarg)
 311         return E_OUTOFMEMORY;
 312     memcpy(dp->rgvarg+1, oldargs, dp->cArgs*sizeof(VARIANTARG));
 313     V_VT(dp->rgvarg) = VT_DISPATCH;
 314     V_DISPATCH(dp->rgvarg) = jsthis;
 315     dp->cArgs++;
 316
 317     if(dp->cNamedArgs) {
 318         DISPID *old = dp->rgdispidNamedArgs;
 319         dp->rgdispidNamedArgs = heap_alloc((dp->cNamedArgs+1)*sizeof(DISPID));
 320         if(!dp->rgdispidNamedArgs) {
 321             heap_free(dp->rgvarg);
 322             return E_OUTOFMEMORY;
 323         }
 324
 325         memcpy(dp->rgdispidNamedArgs+1, old, dp->cNamedArgs*sizeof(DISPID));
 326         dp->rgdispidNamedArgs[0] = DISPID_THIS;
 327         dp->cNamedArgs++;
 328     }else {
 329         dp->rgdispidNamedArgs = &this_id;
 330         dp->cNamedArgs = 1;
 331     }
 332
 333     return S_OK;
 334 }
 335
 336 static HRESULT invoke_prop_func(jsdisp_t *This, jsdisp_t *jsthis, dispex_prop_t *prop, WORD flags,
 337         DISPPARAMS *dp, VARIANT *retv, jsexcept_t *ei, IServiceProvider *caller)
 338 {
 339     HRESULT hres;
 340
 341     switch(prop->type) {
 342     case PROP_BUILTIN: {
 343         vdisp_t vthis;
 344
 345         if(flags == DISPATCH_CONSTRUCT && (prop->flags & PROPF_METHOD)) {
 346             WARN("%s is not a constructor\n", debugstr_w(prop->name));
 347             return E_INVALIDARG;
 348         }
 349
 350         set_jsdisp(&vthis, jsthis);
 351         hres = prop->u.p->invoke(This->ctx, &vthis, flags, dp, retv, ei, caller);
 352         vdisp_release(&vthis);
 353         return hres;
 354     }
 355     case PROP_PROTREF:
 356         return invoke_prop_func(This->prototype, jsthis, This->prototype->props+prop->u.ref, flags, dp, retv, ei, caller);
 357     case PROP_VARIANT: {
 358         DISPPARAMS new_dp;
 359
 360         if(V_VT(&prop->u.var) != VT_DISPATCH) {
 361             FIXME("invoke vt %d\n", V_VT(&prop->u.var));
 362             return E_FAIL;
 363         }
 364
 365         TRACE("call %s %p\n", debugstr_w(prop->name), V_DISPATCH(&prop->u.var));
 366
 367         hres = set_this(&new_dp, dp, to_disp(jsthis));
 368         if(FAILED(hres))
 369             return hres;
 370
 371         hres = disp_call(This->ctx, V_DISPATCH(&prop->u.var), DISPID_VALUE, flags, &new_dp, retv, ei, caller);
 372
 373         if(new_dp.rgvarg != dp->rgvarg) {
 374             heap_free(new_dp.rgvarg);
 375             if(new_dp.cNamedArgs > 1)
 376                 heap_free(new_dp.rgdispidNamedArgs);
 377         }
 378
 379         return hres;
 380     }
 381     default:
 382         ERR("type %d\n", prop->type);
 383     }
 384
 385     return E_FAIL;
 386 }
 387
 388 static HRESULT prop_get(jsdisp_t *This, dispex_prop_t *prop, DISPPARAMS *dp,
 389         VARIANT *retv, jsexcept_t *ei, IServiceProvider *caller)
 390 {
 391     HRESULT hres;
 392
 393     switch(prop->type) {
 394     case PROP_BUILTIN:
 395         if(prop->u.p->flags & PROPF_METHOD) {
 396             jsdisp_t *obj;
 397             hres = create_builtin_function(This->ctx, prop->u.p->invoke, prop->u.p->name, NULL,
 398                     prop->u.p->flags, NULL, &obj);
 399             if(FAILED(hres))
 400                 break;
 401
 402             prop->type = PROP_VARIANT;
 403             var_set_jsdisp(&prop->u.var, obj);
 404             hres = VariantCopy(retv, &prop->u.var);
 405         }else {
 406             vdisp_t vthis;
 407
 408             set_jsdisp(&vthis, This);
 409             hres = prop->u.p->invoke(This->ctx, &vthis, DISPATCH_PROPERTYGET, dp, retv, ei, caller);
 410             vdisp_release(&vthis);
 411         }
 412         break;
 413     case PROP_PROTREF:
 414         hres = prop_get(This->prototype, This->prototype->props+prop->u.ref, dp, retv, ei, caller);
 415         break;
 416     case PROP_VARIANT:
 417         hres = VariantCopy(retv, &prop->u.var);
 418         break;
 419     default:
 420         ERR("type %d\n", prop->type);
 421         return E_FAIL;
 422     }
 423
 424     if(FAILED(hres)) {
 425         TRACE("fail %08x\n", hres);
 426         return hres;
 427     }
 428
 429     TRACE("%s ret %s\n", debugstr_w(prop->name), debugstr_variant(retv));
 430     return hres;
 431 }
 432
 433 static HRESULT prop_put(jsdisp_t *This, dispex_prop_t *prop, VARIANT *val,
 434         jsexcept_t *ei, IServiceProvider *caller)
 435 {
 436     HRESULT hres;
 437
 438     if(prop->flags & PROPF_CONST)
 439         return S_OK;
 440
 441     switch(prop->type) {
 442     case PROP_BUILTIN:
 443         if(!(prop->flags & PROPF_METHOD)) {
 444             DISPPARAMS dp = {val, NULL, 1, 0};
 445             vdisp_t vthis;
 446
 447             set_jsdisp(&vthis, This);
 448             hres = prop->u.p->invoke(This->ctx, &vthis, DISPATCH_PROPERTYPUT, &dp, NULL, ei, caller);
 449             vdisp_release(&vthis);
 450             return hres;
 451         }
 452     case PROP_PROTREF:
 453         prop->type = PROP_VARIANT;
 454         prop->flags = PROPF_ENUM;
 455         V_VT(&prop->u.var) = VT_EMPTY;
 456         break;
 457     case PROP_VARIANT:
 458         VariantClear(&prop->u.var);
 459         break;
 460     default:
 461         ERR("type %d\n", prop->type);
 462         return E_FAIL;
 463     }
 464
 465     hres = VariantCopy(&prop->u.var, val);
 466     if(FAILED(hres))
 467         return hres;
 468
 469     if(This->builtin_info->on_put)
 470         This->builtin_info->on_put(This, prop->name);
 471
 472     TRACE("%s = %s\n", debugstr_w(prop->name), debugstr_variant(val));
 473     return S_OK;
 474 }
 475
 476 static HRESULT fill_protrefs(jsdisp_t *This)
 477 {
 478     dispex_prop_t *iter, *prop;
 479     HRESULT hres;
 480
 481     if(!This->prototype)
 482         return S_OK;
 483
 484     fill_protrefs(This->prototype);
 485
 486     for(iter = This->prototype->props; iter < This->prototype->props+This->prototype->prop_cnt; iter++) {
 487         if(!iter->name)
 488             continue;
 489         hres = find_prop_name(This, iter->hash, iter->name, &prop);
 490         if(FAILED(hres))
 491             return hres;
 492         if(!prop || prop->type==PROP_DELETED) {
 493             if(prop) {
 494                 prop->type = PROP_PROTREF;
 495                 prop->flags = 0;
 496                 prop->u.ref = iter - This->prototype->props;
 497             }else {
 498                 prop = alloc_protref(This, iter->name, iter - This->prototype->props);
 499                 if(!prop)
 500                     return E_OUTOFMEMORY;
 501             }
 502         }
 503     }
 504
 505     return S_OK;
 506 }
 507
 508 static inline jsdisp_t *impl_from_IDispatchEx(IDispatchEx *iface)
 509 {
 510     return CONTAINING_RECORD(iface, jsdisp_t, IDispatchEx_iface);
 511 }
 512
 513 static HRESULT WINAPI DispatchEx_QueryInterface(IDispatchEx *iface, REFIID riid, void **ppv)
 514 {
 515     jsdisp_t *This = impl_from_IDispatchEx(iface);
 516
 517     if(IsEqualGUID(&IID_IUnknown, riid)) {
 518         TRACE("(%p)->(IID_IUnknown %p)\n", This, ppv);
 519         *ppv = &This->IDispatchEx_iface;
 520     }else if(IsEqualGUID(&IID_IDispatch, riid)) {
 521         TRACE("(%p)->(IID_IDispatch %p)\n", This, ppv);
 522         *ppv = &This->IDispatchEx_iface;
 523     }else if(IsEqualGUID(&IID_IDispatchEx, riid)) {
 524         TRACE("(%p)->(IID_IDispatchEx %p)\n", This, ppv);
 525         *ppv = &This->IDispatchEx_iface;
 526     }else if(IsEqualGUID(&IID_IDispatchJS, riid)) {
 527         TRACE("(%p)->(IID_IDispatchJS %p)\n", This, ppv);
 528         jsdisp_addref(This);
 529         *ppv = This;
 530         return S_OK;
 531     }else {
 532         WARN("(%p)->(%s %p)\n", This, debugstr_guid(riid), ppv);
 533         *ppv = NULL;
 534         return E_NOINTERFACE;
 535     }
 536
 537     IUnknown_AddRef((IUnknown*)*ppv);
 538     return S_OK;
 539 }
 540
 541 static ULONG WINAPI DispatchEx_AddRef(IDispatchEx *iface)
 542 {
 543     jsdisp_t *This = impl_from_IDispatchEx(iface);
 544     LONG ref = InterlockedIncrement(&This->ref);
 545
 546     TRACE("(%p) ref=%d\n", This, ref);
 547
 548     return ref;
 549 }
 550
 551 static ULONG WINAPI DispatchEx_Release(IDispatchEx *iface)
 552 {
 553     jsdisp_t *This = impl_from_IDispatchEx(iface);
 554     LONG ref = InterlockedDecrement(&This->ref);
 555
 556     TRACE("(%p) ref=%d\n", This, ref);
 557
 558     if(!ref) {
 559         dispex_prop_t *prop;
 560
 561         for(prop = This->props; prop < This->props+This->prop_cnt; prop++) {
 562             if(prop->type == PROP_VARIANT)
 563                 VariantClear(&prop->u.var);
 564             heap_free(prop->name);
 565         }
 566         heap_free(This->props);
 567         script_release(This->ctx);
 568         if(This->prototype)
 569             jsdisp_release(This->prototype);
 570
 571         if(This->builtin_info->destructor)
 572             This->builtin_info->destructor(This);
 573         else
 574             heap_free(This);
 575     }
 576
 577     return ref;
 578 }
 579
 580 static HRESULT WINAPI DispatchEx_GetTypeInfoCount(IDispatchEx *iface, UINT *pctinfo)
 581 {
 582     jsdisp_t *This = impl_from_IDispatchEx(iface);
 583
 584     TRACE("(%p)->(%p)\n", This, pctinfo);
 585
 586     *pctinfo = 1;
 587     return S_OK;
 588 }
 589
 590 static HRESULT WINAPI DispatchEx_GetTypeInfo(IDispatchEx *iface, UINT iTInfo, LCID lcid,
 591                                               ITypeInfo **ppTInfo)
 592 {
 593     jsdisp_t *This = impl_from_IDispatchEx(iface);
 594     FIXME("(%p)->(%u %u %p)\n", This, iTInfo, lcid, ppTInfo);
 595     return E_NOTIMPL;
 596 }
 597
 598 static HRESULT WINAPI DispatchEx_GetIDsOfNames(IDispatchEx *iface, REFIID riid,
 599                                                 LPOLESTR *rgszNames, UINT cNames, LCID lcid,
 600                                                 DISPID *rgDispId)
 601 {
 602     jsdisp_t *This = impl_from_IDispatchEx(iface);
 603     UINT i;
 604     HRESULT hres;
 605
 606     TRACE("(%p)->(%s %p %u %u %p)\n", This, debugstr_guid(riid), rgszNames, cNames,
 607           lcid, rgDispId);
 608
 609     for(i=0; i < cNames; i++) {
 610         hres = IDispatchEx_GetDispID(&This->IDispatchEx_iface, rgszNames[i], 0, rgDispId+i);
 611         if(FAILED(hres))
 612             return hres;
 613     }
 614
 615     return S_OK;
 616 }
 617
 618 static HRESULT WINAPI DispatchEx_Invoke(IDispatchEx *iface, DISPID dispIdMember,
 619                                         REFIID riid, LCID lcid, WORD wFlags, DISPPARAMS *pDispParams,
 620                             VARIANT *pVarResult, EXCEPINFO *pExcepInfo, UINT *puArgErr)
 621 {
 622     jsdisp_t *This = impl_from_IDispatchEx(iface);
 623
 624     TRACE("(%p)->(%d %s %d %d %p %p %p %p)\n", This, dispIdMember, debugstr_guid(riid),
 625           lcid, wFlags, pDispParams, pVarResult, pExcepInfo, puArgErr);
 626
 627     return IDispatchEx_InvokeEx(&This->IDispatchEx_iface, dispIdMember, lcid, wFlags,
 628             pDispParams, pVarResult, pExcepInfo, NULL);
 629 }
 630
 631 static HRESULT WINAPI DispatchEx_GetDispID(IDispatchEx *iface, BSTR bstrName, DWORD grfdex, DISPID *pid)
 632 {
 633     jsdisp_t *This = impl_from_IDispatchEx(iface);
 634
 635     TRACE("(%p)->(%s %x %p)\n", This, debugstr_w(bstrName), grfdex, pid);
 636
 637     if(grfdex & ~(fdexNameCaseSensitive|fdexNameEnsure|fdexNameImplicit|FDEX_VERSION_MASK)) {
 638         FIXME("Unsupported grfdex %x\n", grfdex);
 639         return E_NOTIMPL;
 640     }
 641
 642     return jsdisp_get_id(This, bstrName, grfdex, pid);
 643 }
 644
 645 static HRESULT WINAPI DispatchEx_InvokeEx(IDispatchEx *iface, DISPID id, LCID lcid, WORD wFlags, DISPPARAMS *pdp,
 646         VARIANT *pvarRes, EXCEPINFO *pei, IServiceProvider *pspCaller)
 647 {
 648     jsdisp_t *This = impl_from_IDispatchEx(iface);
 649     dispex_prop_t *prop;
 650     jsexcept_t jsexcept;
 651     HRESULT hres;
 652
 653     TRACE("(%p)->(%x %x %x %p %p %p %p)\n", This, id, lcid, wFlags, pdp, pvarRes, pei, pspCaller);
 654
 655     if(pvarRes)
 656         V_VT(pvarRes) = VT_EMPTY;
 657
 658     prop = get_prop(This, id);
 659     if(!prop || prop->type == PROP_DELETED) {
 660         TRACE("invalid id\n");
 661         return DISP_E_MEMBERNOTFOUND;
 662     }
 663
 664     memset(&jsexcept, 0, sizeof(jsexcept));
 665
 666     switch(wFlags) {
 667     case DISPATCH_METHOD|DISPATCH_PROPERTYGET:
 668         wFlags = DISPATCH_METHOD;
 669         /* fall through */
 670     case DISPATCH_METHOD:
 671     case DISPATCH_CONSTRUCT:
 672         hres = invoke_prop_func(This, This, prop, wFlags, pdp, pvarRes, &jsexcept, pspCaller);
 673         break;
 674     case DISPATCH_PROPERTYGET:
 675         hres = prop_get(This, prop, pdp, pvarRes, &jsexcept, pspCaller);
 676         break;
 677     case DISPATCH_PROPERTYPUT: {
 678         DWORD i;
 679
 680         for(i=0; i < pdp->cNamedArgs; i++) {
 681             if(pdp->rgdispidNamedArgs[i] == DISPID_PROPERTYPUT)
 682                 break;
 683         }
 684
 685         if(i == pdp->cNamedArgs) {
 686             TRACE("no value to set\n");
 687             return DISP_E_PARAMNOTOPTIONAL;
 688         }
 689
 690         hres = prop_put(This, prop, pdp->rgvarg+i, &jsexcept, pspCaller);
 691         break;
 692     }
 693     default:
 694         FIXME("Unimplemented flags %x\n", wFlags);
 695         return E_INVALIDARG;
 696     }
 697
 698     if(pei)
 699         *pei = jsexcept.ei;
 700
 701     return hres;
 702 }
 703
 704 static HRESULT delete_prop(dispex_prop_t *prop)
 705 {
 706     if(prop->type == PROP_VARIANT) {
 707         VariantClear(&prop->u.var);
 708         prop->type = PROP_DELETED;
 709     }
 710     return S_OK;
 711 }
 712
 713 static HRESULT WINAPI DispatchEx_DeleteMemberByName(IDispatchEx *iface, BSTR bstrName, DWORD grfdex)
 714 {
 715     jsdisp_t *This = impl_from_IDispatchEx(iface);
 716     dispex_prop_t *prop;
 717     HRESULT hres;
 718
 719     TRACE("(%p)->(%s %x)\n", This, debugstr_w(bstrName), grfdex);
 720
 721     if(grfdex & ~(fdexNameCaseSensitive|fdexNameEnsure|fdexNameImplicit|FDEX_VERSION_MASK))
 722         FIXME("Unsupported grfdex %x\n", grfdex);
 723
 724     hres = find_prop_name(This, string_hash(bstrName), bstrName, &prop);
 725     if(FAILED(hres))
 726         return hres;
 727     if(!prop) {
 728         TRACE("not found\n");
 729         return S_OK;
 730     }
 731
 732     return delete_prop(prop);
 733 }
 734
 735 static HRESULT WINAPI DispatchEx_DeleteMemberByDispID(IDispatchEx *iface, DISPID id)
 736 {
 737     jsdisp_t *This = impl_from_IDispatchEx(iface);
 738     dispex_prop_t *prop;
 739
 740     TRACE("(%p)->(%x)\n", This, id);
 741
 742     prop = get_prop(This, id);
 743     if(!prop) {
 744         WARN("invalid id\n");
 745         return DISP_E_MEMBERNOTFOUND;
 746     }
 747
 748     return delete_prop(prop);
 749 }
 750
 751 static HRESULT WINAPI DispatchEx_GetMemberProperties(IDispatchEx *iface, DISPID id, DWORD grfdexFetch, DWORD *pgrfdex)
 752 {
 753     jsdisp_t *This = impl_from_IDispatchEx(iface);
 754     FIXME("(%p)->(%x %x %p)\n", This, id, grfdexFetch, pgrfdex);
 755     return E_NOTIMPL;
 756 }
 757
 758 static HRESULT WINAPI DispatchEx_GetMemberName(IDispatchEx *iface, DISPID id, BSTR *pbstrName)
 759 {
 760     jsdisp_t *This = impl_from_IDispatchEx(iface);
 761     dispex_prop_t *prop;
 762
 763     TRACE("(%p)->(%x %p)\n", This, id, pbstrName);
 764
 765     prop = get_prop(This, id);
 766     if(!prop || !prop->name || prop->type == PROP_DELETED)
 767         return DISP_E_MEMBERNOTFOUND;
 768
 769     *pbstrName = SysAllocString(prop->name);
 770     if(!*pbstrName)
 771         return E_OUTOFMEMORY;
 772
 773     return S_OK;
 774 }
 775
 776 static HRESULT WINAPI DispatchEx_GetNextDispID(IDispatchEx *iface, DWORD grfdex, DISPID id, DISPID *pid)
 777 {
 778     jsdisp_t *This = impl_from_IDispatchEx(iface);
 779     dispex_prop_t *iter;
 780     HRESULT hres;
 781
 782     TRACE("(%p)->(%x %x %p)\n", This, grfdex, id, pid);
 783
 784     if(id == DISPID_STARTENUM) {
 785         hres = fill_protrefs(This);
 786         if(FAILED(hres))
 787             return hres;
 788     }
 789
 790     if(id+1>=0 && id+1<This->prop_cnt) {
 791         iter = &This->props[id+1];
 792     }else {
 793         *pid = DISPID_STARTENUM;
 794         return S_FALSE;
 795     }
 796
 797     while(iter < This->props + This->prop_cnt) {
 798         if(iter->name && (get_flags(This, iter) & PROPF_ENUM) && iter->type!=PROP_DELETED) {
 799             *pid = prop_to_id(This, iter);
 800             return S_OK;
 801         }
 802         iter++;
 803     }
 804
 805     *pid = DISPID_STARTENUM;
 806     return S_FALSE;
 807 }
 808
 809 static HRESULT WINAPI DispatchEx_GetNameSpaceParent(IDispatchEx *iface, IUnknown **ppunk)
 810 {
 811     jsdisp_t *This = impl_from_IDispatchEx(iface);
 812     FIXME("(%p)->(%p)\n", This, ppunk);
 813     return E_NOTIMPL;
 814 }
 815
 816 static IDispatchExVtbl DispatchExVtbl = {
 817     DispatchEx_QueryInterface,
 818     DispatchEx_AddRef,
 819     DispatchEx_Release,
 820     DispatchEx_GetTypeInfoCount,
 821     DispatchEx_GetTypeInfo,
 822     DispatchEx_GetIDsOfNames,
 823     DispatchEx_Invoke,
 824     DispatchEx_GetDispID,
 825     DispatchEx_InvokeEx,
 826     DispatchEx_DeleteMemberByName,
 827     DispatchEx_DeleteMemberByDispID,
 828     DispatchEx_GetMemberProperties,
 829     DispatchEx_GetMemberName,
 830     DispatchEx_GetNextDispID,
 831     DispatchEx_GetNameSpaceParent
 832 };
 833
 834 jsdisp_t *as_jsdisp(IDispatch *disp)
 835 {
 836     assert(disp->lpVtbl == (IDispatchVtbl*)&DispatchExVtbl);
 837     return impl_from_IDispatchEx((IDispatchEx*)disp);
 838 }
 839
 840 HRESULT init_dispex(jsdisp_t *dispex, script_ctx_t *ctx, const builtin_info_t *builtin_info, jsdisp_t *prototype)
 841 {
 842     TRACE("%p (%p)\n", dispex, prototype);
 843
 844     dispex->IDispatchEx_iface.lpVtbl = &DispatchExVtbl;
 845     dispex->ref = 1;
 846     dispex->builtin_info = builtin_info;
 847
 848     dispex->props = heap_alloc_zero(sizeof(dispex_prop_t)*(dispex->buf_size=4));
 849     if(!dispex->props)
 850         return E_OUTOFMEMORY;
 851
 852     dispex->prototype = prototype;
 853     if(prototype)
 854         jsdisp_addref(prototype);
 855
 856     dispex->prop_cnt = 1;
 857     if(builtin_info->value_prop.invoke) {
 858         dispex->props[0].type = PROP_BUILTIN;
 859         dispex->props[0].u.p = &builtin_info->value_prop;
 860     }else {
 861         dispex->props[0].type = PROP_DELETED;
 862     }
 863
 864     script_addref(ctx);
 865     dispex->ctx = ctx;
 866
 867     return S_OK;
 868 }
 869
 870 static const builtin_info_t dispex_info = {
 871     JSCLASS_NONE,
 872     {NULL, NULL, 0},
 873     0, NULL,
 874     NULL,
 875     NULL
 876 };
 877
 878 HRESULT create_dispex(script_ctx_t *ctx, const builtin_info_t *builtin_info, jsdisp_t *prototype, jsdisp_t **dispex)
 879 {
 880     jsdisp_t *ret;
 881     HRESULT hres;
 882
 883     ret = heap_alloc_zero(sizeof(jsdisp_t));
 884     if(!ret)
 885         return E_OUTOFMEMORY;
 886
 887     hres = init_dispex(ret, ctx, builtin_info ? builtin_info : &dispex_info, prototype);
 888     if(FAILED(hres))
 889         return hres;
 890
 891     *dispex = ret;
 892     return S_OK;
 893 }
 894
 895 HRESULT init_dispex_from_constr(jsdisp_t *dispex, script_ctx_t *ctx, const builtin_info_t *builtin_info, jsdisp_t *constr)
 896 {
 897     jsdisp_t *prot = NULL;
 898     dispex_prop_t *prop;
 899     HRESULT hres;
 900
 901     static const WCHAR constructorW[] = {'c','o','n','s','t','r','u','c','t','o','r',0};
 902     static const WCHAR prototypeW[] = {'p','r','o','t','o','t','y','p','e',0};
 903
 904     hres = find_prop_name_prot(constr, string_hash(prototypeW), prototypeW, &prop);
 905     if(SUCCEEDED(hres) && prop && prop->type!=PROP_DELETED) {
 906         jsexcept_t jsexcept;
 907         VARIANT var;
 908
 909         V_VT(&var) = VT_EMPTY;
 910         memset(&jsexcept, 0, sizeof(jsexcept));
 911         hres = prop_get(constr, prop, NULL, &var, &jsexcept, NULL/*FIXME*/);
 912         if(FAILED(hres)) {
 913             ERR("Could not get prototype\n");
 914             return hres;
 915         }
 916
 917         if(V_VT(&var) == VT_DISPATCH)
 918             prot = iface_to_jsdisp((IUnknown*)V_DISPATCH(&var));
 919         VariantClear(&var);
 920     }
 921
 922     hres = init_dispex(dispex, ctx, builtin_info, prot);
 923
 924     if(prot)
 925         jsdisp_release(prot);
 926     if(FAILED(hres))
 927         return hres;
 928
 929     hres = ensure_prop_name(dispex, constructorW, FALSE, 0, &prop);
 930     if(SUCCEEDED(hres)) {
 931         jsexcept_t jsexcept;
 932         VARIANT var;
 933
 934         var_set_jsdisp(&var, constr);
 935         memset(&jsexcept, 0, sizeof(jsexcept));
 936         hres = prop_put(dispex, prop, &var, &jsexcept, NULL/*FIXME*/);
 937     }
 938     if(FAILED(hres))
 939         jsdisp_release(dispex);
 940
 941     return hres;
 942 }
 943
 944 jsdisp_t *iface_to_jsdisp(IUnknown *iface)
 945 {
 946     jsdisp_t *ret;
 947     HRESULT hres;
 948
 949     hres = IUnknown_QueryInterface(iface, &IID_IDispatchJS, (void**)&ret);
 950     if(FAILED(hres))
 951         return NULL;
 952
 953     return ret;
 954 }
 955
 956 HRESULT jsdisp_get_id(jsdisp_t *jsdisp, const WCHAR *name, DWORD flags, DISPID *id)
 957 {
 958     dispex_prop_t *prop;
 959     HRESULT hres;
 960
 961     if(flags & fdexNameEnsure)
 962         hres = ensure_prop_name(jsdisp, name, TRUE, PROPF_ENUM, &prop);
 963     else
 964         hres = find_prop_name_prot(jsdisp, string_hash(name), name, &prop);
 965     if(FAILED(hres))
 966         return hres;
 967
 968     if(prop && prop->type!=PROP_DELETED) {
 969         *id = prop_to_id(jsdisp, prop);
 970         return S_OK;
 971     }
 972
 973     TRACE("not found %s\n", debugstr_w(name));
 974     return DISP_E_UNKNOWNNAME;
 975 }
 976
 977 HRESULT jsdisp_call_value(jsdisp_t *jsthis, WORD flags, DISPPARAMS *dp, VARIANT *retv,
 978         jsexcept_t *ei, IServiceProvider *caller)
 979 {
 980     vdisp_t vdisp;
 981     HRESULT hres;
 982
 983     set_jsdisp(&vdisp, jsthis);
 984     hres = jsthis->builtin_info->value_prop.invoke(jsthis->ctx, &vdisp, flags, dp, retv, ei, caller);
 985     vdisp_release(&vdisp);
 986     return hres;
 987 }
 988
 989 HRESULT jsdisp_call(jsdisp_t *disp, DISPID id, WORD flags, DISPPARAMS *dp, VARIANT *retv,
 990         jsexcept_t *ei, IServiceProvider *caller)
 991 {
 992     dispex_prop_t *prop;
 993
 994     memset(ei, 0, sizeof(*ei));
 995     if(retv)
 996         V_VT(retv) = VT_EMPTY;
 997
 998     prop = get_prop(disp, id);
 999     if(!prop)
1000         return DISP_E_MEMBERNOTFOUND;
1001
1002     return invoke_prop_func(disp, disp, prop, flags, dp, retv, ei, caller);
1003 }
1004
1005 HRESULT jsdisp_call_name(jsdisp_t *disp, const WCHAR *name, WORD flags, DISPPARAMS *dp, VARIANT *retv,
1006         jsexcept_t *ei, IServiceProvider *caller)
1007 {
1008     dispex_prop_t *prop;
1009     HRESULT hres;
1010
1011     hres = find_prop_name_prot(disp, string_hash(name), name, &prop);
1012     if(FAILED(hres))
1013         return hres;
1014
1015     memset(ei, 0, sizeof(*ei));
1016     if(retv)
1017         V_VT(retv) = VT_EMPTY;
1018
1019     return invoke_prop_func(disp, disp, prop, flags, dp, retv, ei, caller);
1020 }
1021
1022 HRESULT disp_call(script_ctx_t *ctx, IDispatch *disp, DISPID id, WORD flags, DISPPARAMS *dp, VARIANT *retv,
1023         jsexcept_t *ei, IServiceProvider *caller)
1024 {
1025     jsdisp_t *jsdisp;
1026     IDispatchEx *dispex;
1027     HRESULT hres;
1028
1029     jsdisp = iface_to_jsdisp((IUnknown*)disp);
1030     if(jsdisp) {
1031         hres = jsdisp_call(jsdisp, id, flags, dp, retv, ei, caller);
1032         jsdisp_release(jsdisp);
1033         return hres;
1034     }
1035
1036     memset(ei, 0, sizeof(*ei));
1037     if(retv && arg_cnt(dp))
1038         flags |= DISPATCH_PROPERTYGET;
1039
1040     if(retv)
1041         V_VT(retv) = VT_EMPTY;
1042     hres = IDispatch_QueryInterface(disp, &IID_IDispatchEx, (void**)&dispex);
1043     if(FAILED(hres)) {
1044         UINT err = 0;
1045
1046         if(flags == DISPATCH_CONSTRUCT) {
1047             WARN("IDispatch cannot be constructor\n");
1048             return DISP_E_MEMBERNOTFOUND;
1049         }
1050
1051         TRACE("using IDispatch\n");
1052         return IDispatch_Invoke(disp, id, &IID_NULL, ctx->lcid, flags, dp, retv, &ei->ei, &err);
1053     }
1054
1055     hres = IDispatchEx_InvokeEx(dispex, id, ctx->lcid, flags, dp, retv, &ei->ei, caller);
1056     IDispatchEx_Release(dispex);
1057
1058     return hres;
1059 }
1060
1061 HRESULT jsdisp_propput_name(jsdisp_t *obj, const WCHAR *name, VARIANT *val, jsexcept_t *ei, IServiceProvider *caller)
1062 {
1063     dispex_prop_t *prop;
1064     HRESULT hres;
1065
1066     hres = ensure_prop_name(obj, name, FALSE, PROPF_ENUM, &prop);
1067     if(FAILED(hres))
1068         return hres;
1069
1070     return prop_put(obj, prop, val, ei, caller);
1071 }
1072
1073 HRESULT jsdisp_propput_const(jsdisp_t *obj, const WCHAR *name, VARIANT *val)
1074 {
1075     dispex_prop_t *prop;
1076     HRESULT hres;
1077
1078     hres = ensure_prop_name(obj, name, FALSE, PROPF_ENUM|PROPF_CONST, &prop);
1079     if(FAILED(hres))
1080         return hres;
1081
1082     return VariantCopy(&prop->u.var, val);
1083 }
1084
1085 HRESULT jsdisp_propput_idx(jsdisp_t *obj, DWORD idx, VARIANT *val, jsexcept_t *ei, IServiceProvider *caller)
1086 {
1087     WCHAR buf[12];
1088
1089     static const WCHAR formatW[] = {'%','d',0};
1090
1091     sprintfW(buf, formatW, idx);
1092     return jsdisp_propput_name(obj, buf, val, ei, caller);
1093 }
1094
1095 HRESULT disp_propput(script_ctx_t *ctx, IDispatch *disp, DISPID id, VARIANT *val, jsexcept_t *ei, IServiceProvider *caller)
1096 {
1097     jsdisp_t *jsdisp;
1098     HRESULT hres;
1099
1100     jsdisp = iface_to_jsdisp((IUnknown*)disp);
1101     if(jsdisp) {
1102         dispex_prop_t *prop;
1103
1104         prop = get_prop(jsdisp, id);
1105         if(prop)
1106             hres = prop_put(jsdisp, prop, val, ei, caller);
1107         else
1108             hres = DISP_E_MEMBERNOTFOUND;
1109
1110         jsdisp_release(jsdisp);
1111     }else {
1112         DISPID dispid = DISPID_PROPERTYPUT;
1113         DISPPARAMS dp  = {val, &dispid, 1, 1};
1114         IDispatchEx *dispex;
1115
1116         hres = IDispatch_QueryInterface(disp, &IID_IDispatchEx, (void**)&dispex);
1117         if(SUCCEEDED(hres)) {
1118             hres = IDispatchEx_InvokeEx(dispex, id, ctx->lcid, DISPATCH_PROPERTYPUT, &dp, NULL, &ei->ei, caller);
1119             IDispatchEx_Release(dispex);
1120         }else {
1121             ULONG err = 0;
1122
1123             TRACE("using IDispatch\n");
1124             hres = IDispatch_Invoke(disp, id, &IID_NULL, ctx->lcid, DISPATCH_PROPERTYPUT, &dp, NULL, &ei->ei, &err);
1125         }
1126     }
1127
1128     return hres;
1129 }
1130
1131 HRESULT jsdisp_propget_name(jsdisp_t *obj, const WCHAR *name, VARIANT *var, jsexcept_t *ei, IServiceProvider *caller)
1132 {
1133     DISPPARAMS dp = {NULL, NULL, 0, 0};
1134     dispex_prop_t *prop;
1135     HRESULT hres;
1136
1137     hres = find_prop_name_prot(obj, string_hash(name), name, &prop);
1138     if(FAILED(hres))
1139         return hres;
1140
1141     V_VT(var) = VT_EMPTY;
1142     if(!prop || prop->type==PROP_DELETED)
1143         return S_OK;
1144
1145     return prop_get(obj, prop, &dp, var, ei, caller);
1146 }
1147
1148 HRESULT jsdisp_get_idx(jsdisp_t *obj, DWORD idx, VARIANT *var, jsexcept_t *ei, IServiceProvider *caller)
1149 {
1150     WCHAR name[12];
1151     DISPPARAMS dp = {NULL, NULL, 0, 0};
1152     dispex_prop_t *prop;
1153     HRESULT hres;
1154
1155     static const WCHAR formatW[] = {'%','d',0};
1156
1157     sprintfW(name, formatW, idx);
1158
1159     hres = find_prop_name_prot(obj, string_hash(name), name, &prop);
1160     if(FAILED(hres))
1161         return hres;
1162
1163     V_VT(var) = VT_EMPTY;
1164     if(!prop || prop->type==PROP_DELETED)
1165         return DISP_E_UNKNOWNNAME;
1166
1167     return prop_get(obj, prop, &dp, var, ei, caller);
1168 }
1169
1170 HRESULT jsdisp_propget(jsdisp_t *jsdisp, DISPID id, VARIANT *val, jsexcept_t *ei, IServiceProvider *caller)
1171 {
1172     DISPPARAMS dp  = {NULL,NULL,0,0};
1173     dispex_prop_t *prop;
1174
1175     prop = get_prop(jsdisp, id);
1176     if(!prop)
1177         return DISP_E_MEMBERNOTFOUND;
1178
1179     V_VT(val) = VT_EMPTY;
1180     return prop_get(jsdisp, prop, &dp, val, ei, caller);
1181 }
1182
1183 HRESULT disp_propget(script_ctx_t *ctx, IDispatch *disp, DISPID id, VARIANT *val, jsexcept_t *ei, IServiceProvider *caller)
1184 {
1185     DISPPARAMS dp  = {NULL,NULL,0,0};
1186     IDispatchEx *dispex;
1187     jsdisp_t *jsdisp;
1188     HRESULT hres;
1189
1190     jsdisp = iface_to_jsdisp((IUnknown*)disp);
1191     if(jsdisp) {
1192         hres = jsdisp_propget(jsdisp, id, val, ei, caller);
1193         jsdisp_release(jsdisp);
1194         return hres;
1195     }
1196
1197     hres = IDispatch_QueryInterface(disp, &IID_IDispatchEx, (void**)&dispex);
1198     if(FAILED(hres)) {
1199         ULONG err = 0;
1200
1201         TRACE("using IDispatch\n");
1202         return IDispatch_Invoke(disp, id, &IID_NULL, ctx->lcid, INVOKE_PROPERTYGET, &dp, val, &ei->ei, &err);
1203     }
1204
1205     hres = IDispatchEx_InvokeEx(dispex, id, ctx->lcid, INVOKE_PROPERTYGET, &dp, val, &ei->ei, caller);
1206     IDispatchEx_Release(dispex);
1207
1208     return hres;
1209 }
1210
1211 HRESULT jsdisp_delete_idx(jsdisp_t *obj, DWORD idx)
1212 {
1213     static const WCHAR formatW[] = {'%','d',0};
1214     WCHAR buf[12];
1215     dispex_prop_t *prop;
1216     HRESULT hres;
1217
1218     sprintfW(buf, formatW, idx);
1219
1220     hres = find_prop_name(obj, string_hash(buf), buf, &prop);
1221     if(FAILED(hres) || !prop)
1222         return hres;
1223
1224     return delete_prop(prop);
1225 }
1226
1227 VARIANT_BOOL jsdisp_is_own_prop(jsdisp_t *obj, BSTR name)
1228 {
1229     dispex_prop_t *prop;
1230     HRESULT hres;
1231
1232     hres = find_prop_name(obj, string_hash(name), name, &prop);
1233     if(FAILED(hres))
1234         return VARIANT_FALSE;
1235     else if(!prop)
1236         return VARIANT_FALSE;
1237
1238     return prop->type==PROP_VARIANT ? VARIANT_TRUE : VARIANT_FALSE;
1239 }