jscript: Use BSTR also for pure IDispatch call in Object.hasOwnProperty.
[wine] / dlls / jscript / object.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/debug.h"
24
25 WINE_DEFAULT_DEBUG_CHANNEL(jscript);
26
27 static const WCHAR toStringW[] = {'t','o','S','t','r','i','n','g',0};
28 static const WCHAR toLocaleStringW[] = {'t','o','L','o','c','a','l','e','S','t','r','i','n','g',0};
29 static const WCHAR valueOfW[] = {'v','a','l','u','e','O','f',0};
30 static const WCHAR hasOwnPropertyW[] = {'h','a','s','O','w','n','P','r','o','p','e','r','t','y',0};
31 static const WCHAR propertyIsEnumerableW[] =
32     {'p','r','o','p','e','r','t','y','I','s','E','n','u','m','e','r','a','b','l','e',0};
33 static const WCHAR isPrototypeOfW[] = {'i','s','P','r','o','t','o','t','y','p','e','O','f',0};
34
35 static const WCHAR default_valueW[] = {'[','o','b','j','e','c','t',' ','O','b','j','e','c','t',']',0};
36
37 static HRESULT Object_toString(script_ctx_t *ctx, vdisp_t *jsthis, WORD flags, unsigned argc, jsval_t *argv,
38         jsval_t *r)
39 {
40     jsdisp_t *jsdisp;
41     const WCHAR *str;
42
43     static const WCHAR formatW[] = {'[','o','b','j','e','c','t',' ','%','s',']',0};
44
45     static const WCHAR arrayW[] = {'A','r','r','a','y',0};
46     static const WCHAR booleanW[] = {'B','o','o','l','e','a','n',0};
47     static const WCHAR dateW[] = {'D','a','t','e',0};
48     static const WCHAR errorW[] = {'E','r','r','o','r',0};
49     static const WCHAR functionW[] = {'F','u','n','c','t','i','o','n',0};
50     static const WCHAR mathW[] = {'M','a','t','h',0};
51     static const WCHAR numberW[] = {'N','u','m','b','e','r',0};
52     static const WCHAR objectW[] = {'O','b','j','e','c','t',0};
53     static const WCHAR regexpW[] = {'R','e','g','E','x','p',0};
54     static const WCHAR stringW[] = {'S','t','r','i','n','g',0};
55     /* Keep in sync with jsclass_t enum */
56     static const WCHAR *names[] = {NULL, arrayW, booleanW, dateW, errorW,
57         functionW, NULL, mathW, numberW, objectW, regexpW, stringW, objectW, objectW};
58
59     TRACE("\n");
60
61     jsdisp = get_jsdisp(jsthis);
62     if(!jsdisp) {
63         str = objectW;
64     }else if(names[jsdisp->builtin_info->class]) {
65         str = names[jsdisp->builtin_info->class];
66     }else {
67         assert(jsdisp->builtin_info->class != JSCLASS_NONE);
68         FIXME("jdisp->builtin_info->class = %d\n", jsdisp->builtin_info->class);
69         return E_FAIL;
70     }
71
72     if(r) {
73         jsstr_t *ret;
74
75         ret = jsstr_alloc_buf(9+strlenW(str));
76         if(!ret)
77             return E_OUTOFMEMORY;
78
79         sprintfW(ret->str, formatW, str);
80         *r = jsval_string(ret);
81     }
82
83     return S_OK;
84 }
85
86 static HRESULT Object_toLocaleString(script_ctx_t *ctx, vdisp_t *jsthis, WORD flags, unsigned argc, jsval_t *argv,
87         jsval_t *r)
88 {
89     TRACE("\n");
90
91     if(!is_jsdisp(jsthis)) {
92         FIXME("Host object this\n");
93         return E_FAIL;
94     }
95
96     return jsdisp_call_name(jsthis->u.jsdisp, toStringW, DISPATCH_METHOD, 0, NULL, r);
97 }
98
99 static HRESULT Object_valueOf(script_ctx_t *ctx, vdisp_t *jsthis, WORD flags, unsigned argc, jsval_t *argv,
100         jsval_t *r)
101 {
102     TRACE("\n");
103
104     if(r) {
105         IDispatch_AddRef(jsthis->u.disp);
106         *r = jsval_disp(jsthis->u.disp);
107     }
108     return S_OK;
109 }
110
111 static HRESULT Object_hasOwnProperty(script_ctx_t *ctx, vdisp_t *jsthis, WORD flags, unsigned argc, jsval_t *argv,
112         jsval_t *r)
113 {
114     jsstr_t *name;
115     DISPID id;
116     BSTR bstr;
117     HRESULT hres;
118
119     TRACE("\n");
120
121     if(!argc) {
122         if(r)
123             *r = jsval_bool(FALSE);
124         return S_OK;
125     }
126
127     hres = to_string(ctx, argv[0], &name);
128     if(FAILED(hres))
129         return hres;
130
131     if(is_jsdisp(jsthis)) {
132         BOOL result;
133
134         hres = jsdisp_is_own_prop(jsthis->u.jsdisp, name->str, &result);
135         jsstr_release(name);
136         if(FAILED(hres))
137             return hres;
138
139         if(r)
140             *r = jsval_bool(result);
141         return S_OK;
142     }
143
144
145     bstr = SysAllocStringLen(NULL, jsstr_length(name));
146     if(bstr)
147         jsstr_flush(name, bstr);
148     jsstr_release(name);
149     if(!bstr)
150         return E_OUTOFMEMORY;
151
152     if(is_dispex(jsthis))
153         hres = IDispatchEx_GetDispID(jsthis->u.dispex, bstr, make_grfdex(ctx, fdexNameCaseSensitive), &id);
154     else
155         hres = IDispatch_GetIDsOfNames(jsthis->u.disp, &IID_NULL, &bstr, 1, ctx->lcid, &id);
156
157     SysFreeString(bstr);
158     if(r)
159         *r = jsval_bool(SUCCEEDED(hres));
160     return S_OK;
161 }
162
163 static HRESULT Object_propertyIsEnumerable(script_ctx_t *ctx, vdisp_t *jsthis, WORD flags, unsigned argc, jsval_t *argv,
164         jsval_t *r)
165 {
166     jsstr_t *name;
167     BOOL ret;
168     HRESULT hres;
169
170     TRACE("\n");
171
172     if(argc != 1) {
173         FIXME("argc %d not supported\n", argc);
174         return E_NOTIMPL;
175     }
176
177     if(!is_jsdisp(jsthis)) {
178         FIXME("Host object this\n");
179         return E_FAIL;
180     }
181
182     hres = to_string(ctx, argv[0], &name);
183     if(FAILED(hres))
184         return hres;
185
186     hres = jsdisp_is_enumerable(jsthis->u.jsdisp, name->str, &ret);
187     jsstr_release(name);
188     if(FAILED(hres))
189         return hres;
190
191     if(r)
192         *r = jsval_bool(ret);
193     return S_OK;
194 }
195
196 static HRESULT Object_isPrototypeOf(script_ctx_t *ctx, vdisp_t *jsthis, WORD flags, unsigned argc, jsval_t *argv,
197         jsval_t *r)
198 {
199     FIXME("\n");
200     return E_NOTIMPL;
201 }
202
203 static HRESULT Object_value(script_ctx_t *ctx, vdisp_t *jsthis, WORD flags, unsigned argc, jsval_t *argv,
204         jsval_t *r)
205 {
206     TRACE("\n");
207
208     switch(flags) {
209     case INVOKE_FUNC:
210         return throw_type_error(ctx, JS_E_FUNCTION_EXPECTED, NULL);
211     case DISPATCH_PROPERTYGET: {
212         jsstr_t *ret = jsstr_alloc(default_valueW);
213         if(!ret)
214             return E_OUTOFMEMORY;
215         *r = jsval_string(ret);
216         break;
217     }
218     default:
219         FIXME("unimplemented flags %x\n", flags);
220         return E_NOTIMPL;
221     }
222
223     return S_OK;
224 }
225
226 static void Object_destructor(jsdisp_t *dispex)
227 {
228     heap_free(dispex);
229 }
230
231 static const builtin_prop_t Object_props[] = {
232     {hasOwnPropertyW,        Object_hasOwnProperty,        PROPF_METHOD|1},
233     {isPrototypeOfW,         Object_isPrototypeOf,         PROPF_METHOD|1},
234     {propertyIsEnumerableW,  Object_propertyIsEnumerable,  PROPF_METHOD|1},
235     {toLocaleStringW,        Object_toLocaleString,        PROPF_METHOD},
236     {toStringW,              Object_toString,              PROPF_METHOD},
237     {valueOfW,               Object_valueOf,               PROPF_METHOD}
238 };
239
240 static const builtin_info_t Object_info = {
241     JSCLASS_OBJECT,
242     {NULL, Object_value, 0},
243     sizeof(Object_props)/sizeof(*Object_props),
244     Object_props,
245     Object_destructor,
246     NULL
247 };
248
249 static const builtin_info_t ObjectInst_info = {
250     JSCLASS_OBJECT,
251     {NULL, Object_value, 0},
252     0, NULL,
253     Object_destructor,
254     NULL
255 };
256
257 static HRESULT ObjectConstr_value(script_ctx_t *ctx, vdisp_t *jsthis, WORD flags, unsigned argc, jsval_t *argv,
258         jsval_t *r)
259 {
260     HRESULT hres;
261
262     TRACE("\n");
263
264     switch(flags) {
265     case DISPATCH_METHOD:
266         if(argc) {
267             if(!is_undefined(argv[0]) && !is_null(argv[0]) && (!is_object_instance(argv[0]) || get_object(argv[0]))) {
268                 IDispatch *disp;
269
270                 hres = to_object(ctx, argv[0], &disp);
271                 if(FAILED(hres))
272                     return hres;
273
274                 if(r)
275                     *r = jsval_disp(disp);
276                 else
277                     IDispatch_Release(disp);
278                 return S_OK;
279             }
280         }
281         /* fall through */
282     case DISPATCH_CONSTRUCT: {
283         jsdisp_t *obj;
284
285         hres = create_object(ctx, NULL, &obj);
286         if(FAILED(hres))
287             return hres;
288
289         if(r)
290             *r = jsval_obj(obj);
291         else
292             jsdisp_release(obj);
293         break;
294     }
295
296     default:
297         FIXME("unimplemented flags: %x\n", flags);
298         return E_NOTIMPL;
299     }
300
301     return S_OK;
302 }
303
304 HRESULT create_object_constr(script_ctx_t *ctx, jsdisp_t *object_prototype, jsdisp_t **ret)
305 {
306     static const WCHAR ObjectW[] = {'O','b','j','e','c','t',0};
307
308     return create_builtin_constructor(ctx, ObjectConstr_value, ObjectW, NULL, PROPF_CONSTR,
309             object_prototype, ret);
310 }
311
312 HRESULT create_object_prototype(script_ctx_t *ctx, jsdisp_t **ret)
313 {
314     return create_dispex(ctx, &Object_info, NULL, ret);
315 }
316
317 HRESULT create_object(script_ctx_t *ctx, jsdisp_t *constr, jsdisp_t **ret)
318 {
319     jsdisp_t *object;
320     HRESULT hres;
321
322     object = heap_alloc_zero(sizeof(jsdisp_t));
323     if(!object)
324         return E_OUTOFMEMORY;
325
326     hres = init_dispex_from_constr(object, ctx, &ObjectInst_info, constr ? constr : ctx->object_constr);
327     if(FAILED(hres)) {
328         heap_free(object);
329         return hres;
330     }
331
332     *ret = object;
333     return S_OK;
334 }