devenum: COM cleanup for IParseDisplayName.
[wine] / 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);
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);
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);
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);
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 jsdisp_t *to_jsdisp(IDispatch *disp)
841 {
842     return disp->lpVtbl == (IDispatchVtbl*)&DispatchExVtbl ? impl_from_IDispatchEx((IDispatchEx*)disp) : NULL;
843 }
844
845 HRESULT init_dispex(jsdisp_t *dispex, script_ctx_t *ctx, const builtin_info_t *builtin_info, jsdisp_t *prototype)
846 {
847     TRACE("%p (%p)\n", dispex, prototype);
848
849     dispex->IDispatchEx_iface.lpVtbl = &DispatchExVtbl;
850     dispex->ref = 1;
851     dispex->builtin_info = builtin_info;
852
853     dispex->props = heap_alloc_zero(sizeof(dispex_prop_t)*(dispex->buf_size=4));
854     if(!dispex->props)
855         return E_OUTOFMEMORY;
856
857     dispex->prototype = prototype;
858     if(prototype)
859         jsdisp_addref(prototype);
860
861     dispex->prop_cnt = 1;
862     if(builtin_info->value_prop.invoke) {
863         dispex->props[0].type = PROP_BUILTIN;
864         dispex->props[0].u.p = &builtin_info->value_prop;
865     }else {
866         dispex->props[0].type = PROP_DELETED;
867     }
868
869     script_addref(ctx);
870     dispex->ctx = ctx;
871
872     return S_OK;
873 }
874
875 static const builtin_info_t dispex_info = {
876     JSCLASS_NONE,
877     {NULL, NULL, 0},
878     0, NULL,
879     NULL,
880     NULL
881 };
882
883 HRESULT create_dispex(script_ctx_t *ctx, const builtin_info_t *builtin_info, jsdisp_t *prototype, jsdisp_t **dispex)
884 {
885     jsdisp_t *ret;
886     HRESULT hres;
887
888     ret = heap_alloc_zero(sizeof(jsdisp_t));
889     if(!ret)
890         return E_OUTOFMEMORY;
891
892     hres = init_dispex(ret, ctx, builtin_info ? builtin_info : &dispex_info, prototype);
893     if(FAILED(hres))
894         return hres;
895
896     *dispex = ret;
897     return S_OK;
898 }
899
900 HRESULT init_dispex_from_constr(jsdisp_t *dispex, script_ctx_t *ctx, const builtin_info_t *builtin_info, jsdisp_t *constr)
901 {
902     jsdisp_t *prot = NULL;
903     dispex_prop_t *prop;
904     HRESULT hres;
905
906     static const WCHAR constructorW[] = {'c','o','n','s','t','r','u','c','t','o','r',0};
907     static const WCHAR prototypeW[] = {'p','r','o','t','o','t','y','p','e',0};
908
909     hres = find_prop_name_prot(constr, string_hash(prototypeW), prototypeW, &prop);
910     if(SUCCEEDED(hres) && prop && prop->type!=PROP_DELETED) {
911         jsexcept_t jsexcept;
912         VARIANT var;
913
914         V_VT(&var) = VT_EMPTY;
915         memset(&jsexcept, 0, sizeof(jsexcept));
916         hres = prop_get(constr, prop, NULL, &var, &jsexcept, NULL/*FIXME*/);
917         if(FAILED(hres)) {
918             ERR("Could not get prototype\n");
919             return hres;
920         }
921
922         if(V_VT(&var) == VT_DISPATCH)
923             prot = iface_to_jsdisp((IUnknown*)V_DISPATCH(&var));
924         VariantClear(&var);
925     }
926
927     hres = init_dispex(dispex, ctx, builtin_info, prot);
928
929     if(prot)
930         jsdisp_release(prot);
931     if(FAILED(hres))
932         return hres;
933
934     hres = ensure_prop_name(dispex, constructorW, FALSE, 0, &prop);
935     if(SUCCEEDED(hres)) {
936         jsexcept_t jsexcept;
937         VARIANT var;
938
939         var_set_jsdisp(&var, constr);
940         memset(&jsexcept, 0, sizeof(jsexcept));
941         hres = prop_put(dispex, prop, &var, &jsexcept, NULL/*FIXME*/);
942     }
943     if(FAILED(hres))
944         jsdisp_release(dispex);
945
946     return hres;
947 }
948
949 jsdisp_t *iface_to_jsdisp(IUnknown *iface)
950 {
951     jsdisp_t *ret;
952     HRESULT hres;
953
954     hres = IUnknown_QueryInterface(iface, &IID_IDispatchJS, (void**)&ret);
955     if(FAILED(hres))
956         return NULL;
957
958     return ret;
959 }
960
961 HRESULT jsdisp_get_id(jsdisp_t *jsdisp, const WCHAR *name, DWORD flags, DISPID *id)
962 {
963     dispex_prop_t *prop;
964     HRESULT hres;
965
966     if(flags & fdexNameEnsure)
967         hres = ensure_prop_name(jsdisp, name, TRUE, PROPF_ENUM, &prop);
968     else
969         hres = find_prop_name_prot(jsdisp, string_hash(name), name, &prop);
970     if(FAILED(hres))
971         return hres;
972
973     if(prop && prop->type!=PROP_DELETED) {
974         *id = prop_to_id(jsdisp, prop);
975         return S_OK;
976     }
977
978     TRACE("not found %s\n", debugstr_w(name));
979     return DISP_E_UNKNOWNNAME;
980 }
981
982 HRESULT jsdisp_call_value(jsdisp_t *jsthis, WORD flags, DISPPARAMS *dp, VARIANT *retv, jsexcept_t *ei)
983 {
984     vdisp_t vdisp;
985     HRESULT hres;
986
987     set_jsdisp(&vdisp, jsthis);
988     hres = jsthis->builtin_info->value_prop.invoke(jsthis->ctx, &vdisp, flags, dp, retv, ei);
989     vdisp_release(&vdisp);
990     return hres;
991 }
992
993 HRESULT jsdisp_call(jsdisp_t *disp, DISPID id, WORD flags, DISPPARAMS *dp, VARIANT *retv, jsexcept_t *ei)
994 {
995     dispex_prop_t *prop;
996
997     memset(ei, 0, sizeof(*ei));
998     if(retv)
999         V_VT(retv) = VT_EMPTY;
1000
1001     prop = get_prop(disp, id);
1002     if(!prop)
1003         return DISP_E_MEMBERNOTFOUND;
1004
1005     return invoke_prop_func(disp, disp, prop, flags, dp, retv, ei, NULL);
1006 }
1007
1008 HRESULT jsdisp_call_name(jsdisp_t *disp, const WCHAR *name, WORD flags, DISPPARAMS *dp, VARIANT *retv,
1009         jsexcept_t *ei)
1010 {
1011     dispex_prop_t *prop;
1012     HRESULT hres;
1013
1014     hres = find_prop_name_prot(disp, string_hash(name), name, &prop);
1015     if(FAILED(hres))
1016         return hres;
1017
1018     memset(ei, 0, sizeof(*ei));
1019     if(retv)
1020         V_VT(retv) = VT_EMPTY;
1021
1022     return invoke_prop_func(disp, disp, prop, flags, dp, retv, ei, NULL);
1023 }
1024
1025 HRESULT disp_call(script_ctx_t *ctx, IDispatch *disp, DISPID id, WORD flags, DISPPARAMS *dp, VARIANT *retv, jsexcept_t *ei)
1026 {
1027     jsdisp_t *jsdisp;
1028     IDispatchEx *dispex;
1029     HRESULT hres;
1030
1031     jsdisp = iface_to_jsdisp((IUnknown*)disp);
1032     if(jsdisp) {
1033         hres = jsdisp_call(jsdisp, id, flags, dp, retv, ei);
1034         jsdisp_release(jsdisp);
1035         return hres;
1036     }
1037
1038     memset(ei, 0, sizeof(*ei));
1039     if(retv && arg_cnt(dp))
1040         flags |= DISPATCH_PROPERTYGET;
1041
1042     if(retv)
1043         V_VT(retv) = VT_EMPTY;
1044     hres = IDispatch_QueryInterface(disp, &IID_IDispatchEx, (void**)&dispex);
1045     if(FAILED(hres)) {
1046         UINT err = 0;
1047
1048         if(flags == DISPATCH_CONSTRUCT) {
1049             WARN("IDispatch cannot be constructor\n");
1050             return DISP_E_MEMBERNOTFOUND;
1051         }
1052
1053         TRACE("using IDispatch\n");
1054         return IDispatch_Invoke(disp, id, &IID_NULL, ctx->lcid, flags, dp, retv, &ei->ei, &err);
1055     }
1056
1057     hres = IDispatchEx_InvokeEx(dispex, id, ctx->lcid, flags, dp, retv, &ei->ei, &ctx->jscaller->IServiceProvider_iface);
1058     IDispatchEx_Release(dispex);
1059
1060     return hres;
1061 }
1062
1063 HRESULT jsdisp_propput_name(jsdisp_t *obj, const WCHAR *name, VARIANT *val, jsexcept_t *ei)
1064 {
1065     dispex_prop_t *prop;
1066     HRESULT hres;
1067
1068     hres = ensure_prop_name(obj, name, FALSE, PROPF_ENUM, &prop);
1069     if(FAILED(hres))
1070         return hres;
1071
1072     return prop_put(obj, prop, val, ei, NULL);
1073 }
1074
1075 HRESULT jsdisp_propput_const(jsdisp_t *obj, const WCHAR *name, VARIANT *val)
1076 {
1077     dispex_prop_t *prop;
1078     HRESULT hres;
1079
1080     hres = ensure_prop_name(obj, name, FALSE, PROPF_ENUM|PROPF_CONST, &prop);
1081     if(FAILED(hres))
1082         return hres;
1083
1084     return VariantCopy(&prop->u.var, val);
1085 }
1086
1087 HRESULT jsdisp_propput_idx(jsdisp_t *obj, DWORD idx, VARIANT *val, jsexcept_t *ei)
1088 {
1089     WCHAR buf[12];
1090
1091     static const WCHAR formatW[] = {'%','d',0};
1092
1093     sprintfW(buf, formatW, idx);
1094     return jsdisp_propput_name(obj, buf, val, ei);
1095 }
1096
1097 HRESULT disp_propput(script_ctx_t *ctx, IDispatch *disp, DISPID id, VARIANT *val, jsexcept_t *ei)
1098 {
1099     jsdisp_t *jsdisp;
1100     HRESULT hres;
1101
1102     jsdisp = iface_to_jsdisp((IUnknown*)disp);
1103     if(jsdisp) {
1104         dispex_prop_t *prop;
1105
1106         prop = get_prop(jsdisp, id);
1107         if(prop)
1108             hres = prop_put(jsdisp, prop, val, ei, NULL);
1109         else
1110             hres = DISP_E_MEMBERNOTFOUND;
1111
1112         jsdisp_release(jsdisp);
1113     }else {
1114         DISPID dispid = DISPID_PROPERTYPUT;
1115         DISPPARAMS dp  = {val, &dispid, 1, 1};
1116         IDispatchEx *dispex;
1117
1118         hres = IDispatch_QueryInterface(disp, &IID_IDispatchEx, (void**)&dispex);
1119         if(SUCCEEDED(hres)) {
1120             hres = IDispatchEx_InvokeEx(dispex, id, ctx->lcid, DISPATCH_PROPERTYPUT, &dp, NULL, &ei->ei,
1121                     &ctx->jscaller->IServiceProvider_iface);
1122             IDispatchEx_Release(dispex);
1123         }else {
1124             ULONG err = 0;
1125
1126             TRACE("using IDispatch\n");
1127             hres = IDispatch_Invoke(disp, id, &IID_NULL, ctx->lcid, DISPATCH_PROPERTYPUT, &dp, NULL, &ei->ei, &err);
1128         }
1129     }
1130
1131     return hres;
1132 }
1133
1134 HRESULT jsdisp_propget_name(jsdisp_t *obj, const WCHAR *name, VARIANT *var, jsexcept_t *ei)
1135 {
1136     DISPPARAMS dp = {NULL, NULL, 0, 0};
1137     dispex_prop_t *prop;
1138     HRESULT hres;
1139
1140     hres = find_prop_name_prot(obj, string_hash(name), name, &prop);
1141     if(FAILED(hres))
1142         return hres;
1143
1144     V_VT(var) = VT_EMPTY;
1145     if(!prop || prop->type==PROP_DELETED)
1146         return S_OK;
1147
1148     return prop_get(obj, prop, &dp, var, ei, NULL);
1149 }
1150
1151 HRESULT jsdisp_get_idx(jsdisp_t *obj, DWORD idx, VARIANT *var, jsexcept_t *ei)
1152 {
1153     WCHAR name[12];
1154     DISPPARAMS dp = {NULL, NULL, 0, 0};
1155     dispex_prop_t *prop;
1156     HRESULT hres;
1157
1158     static const WCHAR formatW[] = {'%','d',0};
1159
1160     sprintfW(name, formatW, idx);
1161
1162     hres = find_prop_name_prot(obj, string_hash(name), name, &prop);
1163     if(FAILED(hres))
1164         return hres;
1165
1166     V_VT(var) = VT_EMPTY;
1167     if(!prop || prop->type==PROP_DELETED)
1168         return DISP_E_UNKNOWNNAME;
1169
1170     return prop_get(obj, prop, &dp, var, ei, NULL);
1171 }
1172
1173 HRESULT jsdisp_propget(jsdisp_t *jsdisp, DISPID id, VARIANT *val, jsexcept_t *ei)
1174 {
1175     DISPPARAMS dp  = {NULL,NULL,0,0};
1176     dispex_prop_t *prop;
1177
1178     prop = get_prop(jsdisp, id);
1179     if(!prop)
1180         return DISP_E_MEMBERNOTFOUND;
1181
1182     V_VT(val) = VT_EMPTY;
1183     return prop_get(jsdisp, prop, &dp, val, ei, NULL);
1184 }
1185
1186 HRESULT disp_propget(script_ctx_t *ctx, IDispatch *disp, DISPID id, VARIANT *val, jsexcept_t *ei)
1187 {
1188     DISPPARAMS dp  = {NULL,NULL,0,0};
1189     IDispatchEx *dispex;
1190     jsdisp_t *jsdisp;
1191     HRESULT hres;
1192
1193     jsdisp = iface_to_jsdisp((IUnknown*)disp);
1194     if(jsdisp) {
1195         hres = jsdisp_propget(jsdisp, id, val, ei);
1196         jsdisp_release(jsdisp);
1197         return hres;
1198     }
1199
1200     hres = IDispatch_QueryInterface(disp, &IID_IDispatchEx, (void**)&dispex);
1201     if(FAILED(hres)) {
1202         ULONG err = 0;
1203
1204         TRACE("using IDispatch\n");
1205         return IDispatch_Invoke(disp, id, &IID_NULL, ctx->lcid, INVOKE_PROPERTYGET, &dp, val, &ei->ei, &err);
1206     }
1207
1208     hres = IDispatchEx_InvokeEx(dispex, id, ctx->lcid, INVOKE_PROPERTYGET, &dp, val, &ei->ei,
1209             &ctx->jscaller->IServiceProvider_iface);
1210     IDispatchEx_Release(dispex);
1211
1212     return hres;
1213 }
1214
1215 HRESULT jsdisp_delete_idx(jsdisp_t *obj, DWORD idx)
1216 {
1217     static const WCHAR formatW[] = {'%','d',0};
1218     WCHAR buf[12];
1219     dispex_prop_t *prop;
1220     HRESULT hres;
1221
1222     sprintfW(buf, formatW, idx);
1223
1224     hres = find_prop_name(obj, string_hash(buf), buf, &prop);
1225     if(FAILED(hres) || !prop)
1226         return hres;
1227
1228     return delete_prop(prop);
1229 }
1230
1231 VARIANT_BOOL jsdisp_is_own_prop(jsdisp_t *obj, BSTR name)
1232 {
1233     dispex_prop_t *prop;
1234     HRESULT hres;
1235
1236     hres = find_prop_name(obj, string_hash(name), name, &prop);
1237     if(FAILED(hres))
1238         return VARIANT_FALSE;
1239     else if(!prop)
1240         return VARIANT_FALSE;
1241
1242     return prop->type==PROP_VARIANT ? VARIANT_TRUE : VARIANT_FALSE;
1243 }