Sync with trunk r63174.
[reactos.git] / dll / win32 / jscript / array.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 "jscript.h"
20
21 typedef struct {
22 jsdisp_t dispex;
23
24 DWORD length;
25 } ArrayInstance;
26
27 static const WCHAR lengthW[] = {'l','e','n','g','t','h',0};
28 static const WCHAR concatW[] = {'c','o','n','c','a','t',0};
29 static const WCHAR joinW[] = {'j','o','i','n',0};
30 static const WCHAR popW[] = {'p','o','p',0};
31 static const WCHAR pushW[] = {'p','u','s','h',0};
32 static const WCHAR reverseW[] = {'r','e','v','e','r','s','e',0};
33 static const WCHAR shiftW[] = {'s','h','i','f','t',0};
34 static const WCHAR sliceW[] = {'s','l','i','c','e',0};
35 static const WCHAR sortW[] = {'s','o','r','t',0};
36 static const WCHAR spliceW[] = {'s','p','l','i','c','e',0};
37 static const WCHAR toStringW[] = {'t','o','S','t','r','i','n','g',0};
38 static const WCHAR toLocaleStringW[] = {'t','o','L','o','c','a','l','e','S','t','r','i','n','g',0};
39 static const WCHAR unshiftW[] = {'u','n','s','h','i','f','t',0};
40
41 static const WCHAR default_separatorW[] = {',',0};
42
43 static inline ArrayInstance *array_from_vdisp(vdisp_t *vdisp)
44 {
45 return (ArrayInstance*)vdisp->u.jsdisp;
46 }
47
48 static inline ArrayInstance *array_this(vdisp_t *jsthis)
49 {
50 return is_vclass(jsthis, JSCLASS_ARRAY) ? array_from_vdisp(jsthis) : NULL;
51 }
52
53 static HRESULT get_length(script_ctx_t *ctx, vdisp_t *vdisp, jsdisp_t **jsthis, DWORD *ret)
54 {
55 ArrayInstance *array;
56 jsval_t val;
57 HRESULT hres;
58
59 array = array_this(vdisp);
60 if(array) {
61 *jsthis = &array->dispex;
62 *ret = array->length;
63 return S_OK;
64 }
65
66 if(!is_jsdisp(vdisp))
67 return throw_type_error(ctx, JS_E_JSCRIPT_EXPECTED, NULL);
68
69 hres = jsdisp_propget_name(vdisp->u.jsdisp, lengthW, &val);
70 if(FAILED(hres))
71 return hres;
72
73 hres = to_uint32(ctx, val, ret);
74 jsval_release(val);
75 if(FAILED(hres))
76 return hres;
77
78 *jsthis = vdisp->u.jsdisp;
79 return S_OK;
80 }
81
82 static HRESULT set_length(jsdisp_t *obj, DWORD length)
83 {
84 if(is_class(obj, JSCLASS_ARRAY)) {
85 ((ArrayInstance*)obj)->length = length;
86 return S_OK;
87 }
88
89 return jsdisp_propput_name(obj, lengthW, jsval_number(length));
90 }
91
92 static WCHAR *idx_to_str(DWORD idx, WCHAR *ptr)
93 {
94 if(!idx) {
95 *ptr = '0';
96 return ptr;
97 }
98
99 while(idx) {
100 *ptr-- = '0' + (idx%10);
101 idx /= 10;
102 }
103
104 return ptr+1;
105 }
106
107 static HRESULT Array_length(script_ctx_t *ctx, vdisp_t *jsthis, WORD flags, unsigned argc, jsval_t *argv,
108 jsval_t *r)
109 {
110 ArrayInstance *This = array_from_vdisp(jsthis);
111
112 TRACE("%p %d\n", This, This->length);
113
114 switch(flags) {
115 case DISPATCH_PROPERTYGET:
116 *r = jsval_number(This->length);
117 break;
118 case DISPATCH_PROPERTYPUT: {
119 DOUBLE len = -1;
120 DWORD i;
121 HRESULT hres;
122
123 hres = to_number(ctx, argv[0], &len);
124 if(FAILED(hres))
125 return hres;
126
127 len = floor(len);
128 if(len!=(DWORD)len)
129 return throw_range_error(ctx, JS_E_INVALID_LENGTH, NULL);
130
131 for(i=len; i<This->length; i++) {
132 hres = jsdisp_delete_idx(&This->dispex, i);
133 if(FAILED(hres))
134 return hres;
135 }
136
137 This->length = len;
138 break;
139 }
140 default:
141 FIXME("unimplemented flags %x\n", flags);
142 return E_NOTIMPL;
143 }
144
145 return S_OK;
146 }
147
148 static HRESULT concat_array(jsdisp_t *array, ArrayInstance *obj, DWORD *len)
149 {
150 jsval_t val;
151 DWORD i;
152 HRESULT hres;
153
154 for(i=0; i < obj->length; i++) {
155 hres = jsdisp_get_idx(&obj->dispex, i, &val);
156 if(hres == DISP_E_UNKNOWNNAME)
157 continue;
158 if(FAILED(hres))
159 return hres;
160
161 hres = jsdisp_propput_idx(array, *len+i, val);
162 jsval_release(val);
163 if(FAILED(hres))
164 return hres;
165 }
166
167 *len += obj->length;
168 return S_OK;
169 }
170
171 static HRESULT concat_obj(jsdisp_t *array, IDispatch *obj, DWORD *len)
172 {
173 jsdisp_t *jsobj;
174 HRESULT hres;
175
176 jsobj = iface_to_jsdisp((IUnknown*)obj);
177 if(jsobj) {
178 if(is_class(jsobj, JSCLASS_ARRAY)) {
179 hres = concat_array(array, (ArrayInstance*)jsobj, len);
180 jsdisp_release(jsobj);
181 return hres;
182 }
183 jsdisp_release(jsobj);
184 }
185
186 return jsdisp_propput_idx(array, (*len)++, jsval_disp(obj));
187 }
188
189 static HRESULT Array_concat(script_ctx_t *ctx, vdisp_t *jsthis, WORD flags, unsigned argc, jsval_t *argv,
190 jsval_t *r)
191 {
192 jsdisp_t *ret;
193 DWORD len = 0;
194 HRESULT hres;
195
196 TRACE("\n");
197
198 hres = create_array(ctx, 0, &ret);
199 if(FAILED(hres))
200 return hres;
201
202 hres = concat_obj(ret, jsthis->u.disp, &len);
203 if(SUCCEEDED(hres)) {
204 DWORD i;
205
206 for(i=0; i < argc; i++) {
207 if(is_object_instance(argv[i]))
208 hres = concat_obj(ret, get_object(argv[i]), &len);
209 else
210 hres = jsdisp_propput_idx(ret, len++, argv[i]);
211 if(FAILED(hres))
212 break;
213 }
214 }
215
216 if(FAILED(hres))
217 return hres;
218
219 if(r)
220 *r = jsval_obj(ret);
221 else
222 jsdisp_release(ret);
223 return S_OK;
224 }
225
226 static HRESULT array_join(script_ctx_t *ctx, jsdisp_t *array, DWORD length, const WCHAR *sep, jsval_t *r)
227 {
228 jsstr_t **str_tab, *ret;
229 jsval_t val;
230 DWORD i;
231 HRESULT hres = E_FAIL;
232
233 if(!length) {
234 if(r)
235 *r = jsval_string(jsstr_empty());
236 return S_OK;
237 }
238
239 str_tab = heap_alloc_zero(length * sizeof(*str_tab));
240 if(!str_tab)
241 return E_OUTOFMEMORY;
242
243 for(i=0; i < length; i++) {
244 hres = jsdisp_get_idx(array, i, &val);
245 if(hres == DISP_E_UNKNOWNNAME) {
246 hres = S_OK;
247 continue;
248 } else if(FAILED(hres))
249 break;
250
251 if(!is_undefined(val) && !is_null(val)) {
252 hres = to_string(ctx, val, str_tab+i);
253 jsval_release(val);
254 if(FAILED(hres))
255 break;
256 }
257 }
258
259 if(SUCCEEDED(hres)) {
260 DWORD seplen = 0, len = 0;
261
262 seplen = strlenW(sep);
263
264 if(str_tab[0])
265 len = jsstr_length(str_tab[0]);
266 for(i=1; i < length; i++) {
267 len += seplen;
268 if(str_tab[i])
269 len += jsstr_length(str_tab[i]);
270 if(len > JSSTR_MAX_LENGTH) {
271 hres = E_OUTOFMEMORY;
272 break;
273 }
274 }
275
276 if(SUCCEEDED(hres)) {
277 WCHAR *ptr = NULL;
278
279 ptr = jsstr_alloc_buf(len, &ret);
280 if(ptr) {
281 if(str_tab[0])
282 ptr += jsstr_flush(str_tab[0], ptr);
283
284 for(i=1; i < length; i++) {
285 if(seplen) {
286 memcpy(ptr, sep, seplen*sizeof(WCHAR));
287 ptr += seplen;
288 }
289
290 if(str_tab[i])
291 ptr += jsstr_flush(str_tab[i], ptr);
292 }
293 }else {
294 hres = E_OUTOFMEMORY;
295 }
296 }
297 }
298
299 for(i=0; i < length; i++) {
300 if(str_tab[i])
301 jsstr_release(str_tab[i]);
302 }
303 heap_free(str_tab);
304 if(FAILED(hres))
305 return hres;
306
307 TRACE("= %s\n", debugstr_jsstr(ret));
308
309 if(r)
310 *r = jsval_string(ret);
311 else
312 jsstr_release(ret);
313 return S_OK;
314 }
315
316 /* ECMA-262 3rd Edition 15.4.4.5 */
317 static HRESULT Array_join(script_ctx_t *ctx, vdisp_t *vthis, WORD flags, unsigned argc, jsval_t *argv,
318 jsval_t *r)
319 {
320 jsdisp_t *jsthis;
321 DWORD length;
322 HRESULT hres;
323
324 TRACE("\n");
325
326 hres = get_length(ctx, vthis, &jsthis, &length);
327 if(FAILED(hres))
328 return hres;
329
330 if(argc) {
331 const WCHAR *sep;
332 jsstr_t *sep_str;
333
334 hres = to_flat_string(ctx, argv[0], &sep_str, &sep);
335 if(FAILED(hres))
336 return hres;
337
338 hres = array_join(ctx, jsthis, length, sep, r);
339
340 jsstr_release(sep_str);
341 }else {
342 hres = array_join(ctx, jsthis, length, default_separatorW, r);
343 }
344
345 return hres;
346 }
347
348 static HRESULT Array_pop(script_ctx_t *ctx, vdisp_t *vthis, WORD flags, unsigned argc, jsval_t *argv,
349 jsval_t *r)
350 {
351 jsdisp_t *jsthis;
352 jsval_t val;
353 DWORD length;
354 HRESULT hres;
355
356 TRACE("\n");
357
358 hres = get_length(ctx, vthis, &jsthis, &length);
359 if(FAILED(hres))
360 return hres;
361
362 if(!length) {
363 hres = set_length(jsthis, 0);
364 if(FAILED(hres))
365 return hres;
366
367 if(r)
368 *r = jsval_undefined();
369 return S_OK;
370 }
371
372 length--;
373 hres = jsdisp_get_idx(jsthis, length, &val);
374 if(SUCCEEDED(hres))
375 hres = jsdisp_delete_idx(jsthis, length);
376 else if(hres == DISP_E_UNKNOWNNAME)
377 val = jsval_undefined();
378 else
379 return hres;
380
381 if(SUCCEEDED(hres))
382 hres = set_length(jsthis, length);
383
384 if(FAILED(hres)) {
385 jsval_release(val);
386 return hres;
387 }
388
389 if(r)
390 *r = val;
391 else
392 jsval_release(val);
393 return hres;
394 }
395
396 /* ECMA-262 3rd Edition 15.4.4.7 */
397 static HRESULT Array_push(script_ctx_t *ctx, vdisp_t *vthis, WORD flags, unsigned argc, jsval_t *argv,
398 jsval_t *r)
399 {
400 jsdisp_t *jsthis;
401 DWORD length = 0;
402 unsigned i;
403 HRESULT hres;
404
405 TRACE("\n");
406
407 hres = get_length(ctx, vthis, &jsthis, &length);
408 if(FAILED(hres))
409 return hres;
410
411 for(i=0; i < argc; i++) {
412 hres = jsdisp_propput_idx(jsthis, length+i, argv[i]);
413 if(FAILED(hres))
414 return hres;
415 }
416
417 hres = set_length(jsthis, length+argc);
418 if(FAILED(hres))
419 return hres;
420
421 if(r)
422 *r = jsval_number(length+argc);
423 return S_OK;
424 }
425
426 static HRESULT Array_reverse(script_ctx_t *ctx, vdisp_t *vthis, WORD flags, unsigned argc, jsval_t *argv,
427 jsval_t *r)
428 {
429 jsdisp_t *jsthis;
430 DWORD length, k, l;
431 jsval_t v1, v2;
432 HRESULT hres1, hres2;
433
434 TRACE("\n");
435
436 hres1 = get_length(ctx, vthis, &jsthis, &length);
437 if(FAILED(hres1))
438 return hres1;
439
440 for(k=0; k<length/2; k++) {
441 l = length-k-1;
442
443 hres1 = jsdisp_get_idx(jsthis, k, &v1);
444 if(FAILED(hres1) && hres1!=DISP_E_UNKNOWNNAME)
445 return hres1;
446
447 hres2 = jsdisp_get_idx(jsthis, l, &v2);
448 if(FAILED(hres2) && hres2!=DISP_E_UNKNOWNNAME) {
449 jsval_release(v1);
450 return hres2;
451 }
452
453 if(hres1 == DISP_E_UNKNOWNNAME)
454 hres1 = jsdisp_delete_idx(jsthis, l);
455 else
456 hres1 = jsdisp_propput_idx(jsthis, l, v1);
457
458 if(FAILED(hres1)) {
459 jsval_release(v1);
460 jsval_release(v2);
461 return hres1;
462 }
463
464 if(hres2 == DISP_E_UNKNOWNNAME)
465 hres2 = jsdisp_delete_idx(jsthis, k);
466 else
467 hres2 = jsdisp_propput_idx(jsthis, k, v2);
468
469 if(FAILED(hres2)) {
470 jsval_release(v2);
471 return hres2;
472 }
473 }
474
475 if(r)
476 *r = jsval_obj(jsdisp_addref(jsthis));
477 return S_OK;
478 }
479
480 /* ECMA-262 3rd Edition 15.4.4.9 */
481 static HRESULT Array_shift(script_ctx_t *ctx, vdisp_t *vthis, WORD flags, unsigned argc, jsval_t *argv,
482 jsval_t *r)
483 {
484 jsdisp_t *jsthis;
485 DWORD length = 0, i;
486 jsval_t v, ret;
487 HRESULT hres;
488
489 TRACE("\n");
490
491 hres = get_length(ctx, vthis, &jsthis, &length);
492 if(FAILED(hres))
493 return hres;
494
495 if(!length) {
496 hres = set_length(jsthis, 0);
497 if(FAILED(hres))
498 return hres;
499 }
500
501 if(!length) {
502 if(r)
503 *r = jsval_undefined();
504 return S_OK;
505 }
506
507 hres = jsdisp_get_idx(jsthis, 0, &ret);
508 if(hres == DISP_E_UNKNOWNNAME) {
509 ret = jsval_undefined();
510 hres = S_OK;
511 }
512
513 for(i=1; SUCCEEDED(hres) && i<length; i++) {
514 hres = jsdisp_get_idx(jsthis, i, &v);
515 if(hres == DISP_E_UNKNOWNNAME)
516 hres = jsdisp_delete_idx(jsthis, i-1);
517 else if(SUCCEEDED(hres))
518 hres = jsdisp_propput_idx(jsthis, i-1, v);
519 }
520
521 if(SUCCEEDED(hres)) {
522 hres = jsdisp_delete_idx(jsthis, length-1);
523 if(SUCCEEDED(hres))
524 hres = set_length(jsthis, length-1);
525 }
526
527 if(FAILED(hres))
528 return hres;
529
530 if(r)
531 *r = ret;
532 else
533 jsval_release(ret);
534 return hres;
535 }
536
537 /* ECMA-262 3rd Edition 15.4.4.10 */
538 static HRESULT Array_slice(script_ctx_t *ctx, vdisp_t *vthis, WORD flags, unsigned argc, jsval_t *argv, jsval_t *r)
539 {
540 jsdisp_t *arr, *jsthis;
541 DOUBLE range;
542 DWORD length, start, end, idx;
543 HRESULT hres;
544
545 TRACE("\n");
546
547 hres = get_length(ctx, vthis, &jsthis, &length);
548 if(FAILED(hres))
549 return hres;
550
551 if(argc) {
552 hres = to_number(ctx, argv[0], &range);
553 if(FAILED(hres))
554 return hres;
555
556 range = floor(range);
557 if(-range>length || isnan(range)) start = 0;
558 else if(range < 0) start = range+length;
559 else if(range <= length) start = range;
560 else start = length;
561 }
562 else start = 0;
563
564 if(argc > 1) {
565 hres = to_number(ctx, argv[1], &range);
566 if(FAILED(hres))
567 return hres;
568
569 range = floor(range);
570 if(-range>length) end = 0;
571 else if(range < 0) end = range+length;
572 else if(range <= length) end = range;
573 else end = length;
574 }
575 else end = length;
576
577 hres = create_array(ctx, (end>start)?end-start:0, &arr);
578 if(FAILED(hres))
579 return hres;
580
581 for(idx=start; idx<end; idx++) {
582 jsval_t v;
583
584 hres = jsdisp_get_idx(jsthis, idx, &v);
585 if(hres == DISP_E_UNKNOWNNAME)
586 continue;
587
588 if(SUCCEEDED(hres)) {
589 hres = jsdisp_propput_idx(arr, idx-start, v);
590 jsval_release(v);
591 }
592
593 if(FAILED(hres)) {
594 jsdisp_release(arr);
595 return hres;
596 }
597 }
598
599 if(r)
600 *r = jsval_obj(arr);
601 else
602 jsdisp_release(arr);
603
604 return S_OK;
605 }
606
607 static HRESULT sort_cmp(script_ctx_t *ctx, jsdisp_t *cmp_func, jsval_t v1, jsval_t v2, INT *cmp)
608 {
609 HRESULT hres;
610
611 if(cmp_func) {
612 jsval_t args[2];
613 jsval_t res;
614 double n;
615
616 args[0] = v1;
617 args[1] = v2;
618
619 hres = jsdisp_call_value(cmp_func, NULL, DISPATCH_METHOD, 2, args, &res);
620 if(FAILED(hres))
621 return hres;
622
623 hres = to_number(ctx, res, &n);
624 jsval_release(res);
625 if(FAILED(hres))
626 return hres;
627
628 if(n == 0)
629 *cmp = 0;
630 *cmp = n > 0.0 ? 1 : -1;
631 }else if(is_undefined(v1)) {
632 *cmp = is_undefined(v2) ? 0 : 1;
633 }else if(is_undefined(v2)) {
634 *cmp = -1;
635 }else if(is_number(v1) && is_number(v2)) {
636 double d = get_number(v1)-get_number(v2);
637 if(d > 0.0)
638 *cmp = 1;
639 else
640 *cmp = d < -0.0 ? -1 : 0;
641 }else {
642 jsstr_t *x, *y;
643
644 hres = to_string(ctx, v1, &x);
645 if(FAILED(hres))
646 return hres;
647
648 hres = to_string(ctx, v2, &y);
649 if(SUCCEEDED(hres)) {
650 *cmp = jsstr_cmp(x, y);
651 jsstr_release(y);
652 }
653 jsstr_release(x);
654 if(FAILED(hres))
655 return hres;
656 }
657
658 return S_OK;
659 }
660
661 /* ECMA-262 3rd Edition 15.4.4.11 */
662 static HRESULT Array_sort(script_ctx_t *ctx, vdisp_t *vthis, WORD flags, unsigned argc, jsval_t *argv,
663 jsval_t *r)
664 {
665 jsdisp_t *jsthis, *cmp_func = NULL;
666 jsval_t *vtab, **sorttab = NULL;
667 DWORD length;
668 DWORD i;
669 HRESULT hres = S_OK;
670
671 TRACE("\n");
672
673 hres = get_length(ctx, vthis, &jsthis, &length);
674 if(FAILED(hres))
675 return hres;
676
677 if(argc > 1) {
678 WARN("invalid arg_cnt %d\n", argc);
679 return E_FAIL;
680 }
681
682 if(argc == 1) {
683 if(!is_object_instance(argv[0])) {
684 WARN("arg is not dispatch\n");
685 return E_FAIL;
686 }
687
688 cmp_func = iface_to_jsdisp((IUnknown*)get_object(argv[0]));
689 if(!cmp_func || !is_class(cmp_func, JSCLASS_FUNCTION)) {
690 WARN("cmp_func is not a function\n");
691 if(cmp_func)
692 jsdisp_release(cmp_func);
693 return E_FAIL;
694 }
695 }
696
697 if(!length) {
698 if(cmp_func)
699 jsdisp_release(cmp_func);
700 if(r)
701 *r = jsval_obj(jsdisp_addref(jsthis));
702 return S_OK;
703 }
704
705 vtab = heap_alloc_zero(length * sizeof(*vtab));
706 if(vtab) {
707 for(i=0; i<length; i++) {
708 hres = jsdisp_get_idx(jsthis, i, vtab+i);
709 if(hres == DISP_E_UNKNOWNNAME) {
710 vtab[i] = jsval_undefined();
711 hres = S_OK;
712 } else if(FAILED(hres)) {
713 WARN("Could not get elem %d: %08x\n", i, hres);
714 break;
715 }
716 }
717 }else {
718 hres = E_OUTOFMEMORY;
719 }
720
721 if(SUCCEEDED(hres)) {
722 sorttab = heap_alloc(length*2*sizeof(*sorttab));
723 if(!sorttab)
724 hres = E_OUTOFMEMORY;
725 }
726
727 /* merge-sort */
728 if(SUCCEEDED(hres)) {
729 jsval_t *tmpv, **tmpbuf;
730 INT cmp;
731
732 tmpbuf = sorttab + length;
733 for(i=0; i < length; i++)
734 sorttab[i] = vtab+i;
735
736 for(i=0; i < length/2; i++) {
737 hres = sort_cmp(ctx, cmp_func, *sorttab[2*i+1], *sorttab[2*i], &cmp);
738 if(FAILED(hres))
739 break;
740
741 if(cmp < 0) {
742 tmpv = sorttab[2*i];
743 sorttab[2*i] = sorttab[2*i+1];
744 sorttab[2*i+1] = tmpv;
745 }
746 }
747
748 if(SUCCEEDED(hres)) {
749 DWORD k, a, b, bend;
750
751 for(k=2; k < length; k *= 2) {
752 for(i=0; i+k < length; i += 2*k) {
753 a = b = 0;
754 if(i+2*k <= length)
755 bend = k;
756 else
757 bend = length - (i+k);
758
759 memcpy(tmpbuf, sorttab+i, k*sizeof(jsval_t*));
760
761 while(a < k && b < bend) {
762 hres = sort_cmp(ctx, cmp_func, *tmpbuf[a], *sorttab[i+k+b], &cmp);
763 if(FAILED(hres))
764 break;
765
766 if(cmp < 0) {
767 sorttab[i+a+b] = tmpbuf[a];
768 a++;
769 }else {
770 sorttab[i+a+b] = sorttab[i+k+b];
771 b++;
772 }
773 }
774
775 if(FAILED(hres))
776 break;
777
778 if(a < k)
779 memcpy(sorttab+i+a+b, tmpbuf+a, (k-a)*sizeof(jsval_t*));
780 }
781
782 if(FAILED(hres))
783 break;
784 }
785 }
786
787 for(i=0; SUCCEEDED(hres) && i < length; i++)
788 hres = jsdisp_propput_idx(jsthis, i, *sorttab[i]);
789 }
790
791 if(vtab) {
792 for(i=0; i < length; i++)
793 jsval_release(vtab[i]);
794 heap_free(vtab);
795 }
796 heap_free(sorttab);
797 if(cmp_func)
798 jsdisp_release(cmp_func);
799
800 if(FAILED(hres))
801 return hres;
802
803 if(r)
804 *r = jsval_obj(jsdisp_addref(jsthis));
805 return S_OK;
806 }
807
808 /* ECMA-262 3rd Edition 15.4.4.12 */
809 static HRESULT Array_splice(script_ctx_t *ctx, vdisp_t *vthis, WORD flags, unsigned argc, jsval_t *argv,
810 jsval_t *r)
811 {
812 DWORD length, start=0, delete_cnt=0, i, add_args = 0;
813 jsdisp_t *ret_array = NULL, *jsthis;
814 jsval_t val;
815 double d;
816 int n;
817 HRESULT hres = S_OK;
818
819 TRACE("\n");
820
821 hres = get_length(ctx, vthis, &jsthis, &length);
822 if(FAILED(hres))
823 return hres;
824
825 if(argc) {
826 hres = to_integer(ctx, argv[0], &d);
827 if(FAILED(hres))
828 return hres;
829
830 if(is_int32(d)) {
831 if((n = d) >= 0)
832 start = min(n, length);
833 else
834 start = -n > length ? 0 : length + n;
835 }else {
836 start = d < 0.0 ? 0 : length;
837 }
838 }
839
840 if(argc >= 2) {
841 hres = to_integer(ctx, argv[1], &d);
842 if(FAILED(hres))
843 return hres;
844
845 if(is_int32(d)) {
846 if((n = d) > 0)
847 delete_cnt = min(n, length-start);
848 }else if(d > 0.0) {
849 delete_cnt = length-start;
850 }
851
852 add_args = argc-2;
853 }
854
855 if(r) {
856 hres = create_array(ctx, 0, &ret_array);
857 if(FAILED(hres))
858 return hres;
859
860 for(i=0; SUCCEEDED(hres) && i < delete_cnt; i++) {
861 hres = jsdisp_get_idx(jsthis, start+i, &val);
862 if(hres == DISP_E_UNKNOWNNAME) {
863 hres = S_OK;
864 }else if(SUCCEEDED(hres)) {
865 hres = jsdisp_propput_idx(ret_array, i, val);
866 jsval_release(val);
867 }
868 }
869
870 if(SUCCEEDED(hres))
871 hres = jsdisp_propput_name(ret_array, lengthW, jsval_number(delete_cnt));
872 }
873
874 if(add_args < delete_cnt) {
875 for(i = start; SUCCEEDED(hres) && i < length-delete_cnt; i++) {
876 hres = jsdisp_get_idx(jsthis, i+delete_cnt, &val);
877 if(hres == DISP_E_UNKNOWNNAME) {
878 hres = jsdisp_delete_idx(jsthis, i+add_args);
879 }else if(SUCCEEDED(hres)) {
880 hres = jsdisp_propput_idx(jsthis, i+add_args, val);
881 jsval_release(val);
882 }
883 }
884
885 for(i=length; SUCCEEDED(hres) && i != length-delete_cnt+add_args; i--)
886 hres = jsdisp_delete_idx(jsthis, i-1);
887 }else if(add_args > delete_cnt) {
888 for(i=length-delete_cnt; SUCCEEDED(hres) && i != start; i--) {
889 hres = jsdisp_get_idx(jsthis, i+delete_cnt-1, &val);
890 if(hres == DISP_E_UNKNOWNNAME) {
891 hres = jsdisp_delete_idx(jsthis, i+add_args-1);
892 }else if(SUCCEEDED(hres)) {
893 hres = jsdisp_propput_idx(jsthis, i+add_args-1, val);
894 jsval_release(val);
895 }
896 }
897 }
898
899 for(i=0; SUCCEEDED(hres) && i < add_args; i++)
900 hres = jsdisp_propput_idx(jsthis, start+i, argv[i+2]);
901
902 if(SUCCEEDED(hres))
903 hres = jsdisp_propput_name(jsthis, lengthW, jsval_number(length-delete_cnt+add_args));
904
905 if(FAILED(hres)) {
906 if(ret_array)
907 jsdisp_release(ret_array);
908 return hres;
909 }
910
911 if(r)
912 *r = jsval_obj(ret_array);
913 return S_OK;
914 }
915
916 /* ECMA-262 3rd Edition 15.4.4.2 */
917 static HRESULT Array_toString(script_ctx_t *ctx, vdisp_t *jsthis, WORD flags, unsigned argc, jsval_t *argv,
918 jsval_t *r)
919 {
920 ArrayInstance *array;
921
922 TRACE("\n");
923
924 array = array_this(jsthis);
925 if(!array)
926 return throw_type_error(ctx, JS_E_ARRAY_EXPECTED, NULL);
927
928 return array_join(ctx, &array->dispex, array->length, default_separatorW, r);
929 }
930
931 static HRESULT Array_toLocaleString(script_ctx_t *ctx, vdisp_t *vthis, WORD flags, unsigned argc, jsval_t *argv,
932 jsval_t *r)
933 {
934 FIXME("\n");
935 return E_NOTIMPL;
936 }
937
938 /* ECMA-262 3rd Edition 15.4.4.13 */
939 static HRESULT Array_unshift(script_ctx_t *ctx, vdisp_t *vthis, WORD flags, unsigned argc, jsval_t *argv,
940 jsval_t *r)
941 {
942 jsdisp_t *jsthis;
943 WCHAR buf[14], *buf_end, *str;
944 DWORD i, length;
945 jsval_t val;
946 DISPID id;
947 HRESULT hres;
948
949 TRACE("\n");
950
951 hres = get_length(ctx, vthis, &jsthis, &length);
952 if(FAILED(hres))
953 return hres;
954
955 if(argc) {
956 buf_end = buf + sizeof(buf)/sizeof(WCHAR)-1;
957 *buf_end-- = 0;
958 i = length;
959
960 while(i--) {
961 str = idx_to_str(i, buf_end);
962
963 hres = jsdisp_get_id(jsthis, str, 0, &id);
964 if(SUCCEEDED(hres)) {
965 hres = jsdisp_propget(jsthis, id, &val);
966 if(FAILED(hres))
967 return hres;
968
969 hres = jsdisp_propput_idx(jsthis, i+argc, val);
970 jsval_release(val);
971 }else if(hres == DISP_E_UNKNOWNNAME) {
972 hres = IDispatchEx_DeleteMemberByDispID(vthis->u.dispex, id);
973 }
974 }
975
976 if(FAILED(hres))
977 return hres;
978 }
979
980 for(i=0; i<argc; i++) {
981 hres = jsdisp_propput_idx(jsthis, i, argv[i]);
982 if(FAILED(hres))
983 return hres;
984 }
985
986 if(argc) {
987 length += argc;
988 hres = set_length(jsthis, length);
989 if(FAILED(hres))
990 return hres;
991 }
992
993 if(r)
994 *r = ctx->version < 2 ? jsval_undefined() : jsval_number(length);
995 return S_OK;
996 }
997
998 static HRESULT Array_value(script_ctx_t *ctx, vdisp_t *jsthis, WORD flags, unsigned argc, jsval_t *argv,
999 jsval_t *r)
1000 {
1001 TRACE("\n");
1002
1003 switch(flags) {
1004 case INVOKE_FUNC:
1005 return throw_type_error(ctx, JS_E_FUNCTION_EXPECTED, NULL);
1006 case INVOKE_PROPERTYGET:
1007 return array_join(ctx, jsthis->u.jsdisp, array_from_vdisp(jsthis)->length, default_separatorW, r);
1008 default:
1009 FIXME("unimplemented flags %x\n", flags);
1010 return E_NOTIMPL;
1011 }
1012
1013 return S_OK;
1014 }
1015
1016 static void Array_destructor(jsdisp_t *dispex)
1017 {
1018 heap_free(dispex);
1019 }
1020
1021 static void Array_on_put(jsdisp_t *dispex, const WCHAR *name)
1022 {
1023 ArrayInstance *array = (ArrayInstance*)dispex;
1024 const WCHAR *ptr = name;
1025 DWORD id = 0;
1026
1027 if(!isdigitW(*ptr))
1028 return;
1029
1030 while(*ptr && isdigitW(*ptr)) {
1031 id = id*10 + (*ptr-'0');
1032 ptr++;
1033 }
1034
1035 if(*ptr)
1036 return;
1037
1038 if(id >= array->length)
1039 array->length = id+1;
1040 }
1041
1042 static const builtin_prop_t Array_props[] = {
1043 {concatW, Array_concat, PROPF_METHOD|1},
1044 {joinW, Array_join, PROPF_METHOD|1},
1045 {lengthW, Array_length, 0},
1046 {popW, Array_pop, PROPF_METHOD},
1047 {pushW, Array_push, PROPF_METHOD|1},
1048 {reverseW, Array_reverse, PROPF_METHOD},
1049 {shiftW, Array_shift, PROPF_METHOD},
1050 {sliceW, Array_slice, PROPF_METHOD|2},
1051 {sortW, Array_sort, PROPF_METHOD|1},
1052 {spliceW, Array_splice, PROPF_METHOD|2},
1053 {toLocaleStringW, Array_toLocaleString, PROPF_METHOD},
1054 {toStringW, Array_toString, PROPF_METHOD},
1055 {unshiftW, Array_unshift, PROPF_METHOD|1},
1056 };
1057
1058 static const builtin_info_t Array_info = {
1059 JSCLASS_ARRAY,
1060 {NULL, Array_value, 0},
1061 sizeof(Array_props)/sizeof(*Array_props),
1062 Array_props,
1063 Array_destructor,
1064 Array_on_put
1065 };
1066
1067 static const builtin_prop_t ArrayInst_props[] = {
1068 {lengthW, Array_length, 0}
1069 };
1070
1071 static const builtin_info_t ArrayInst_info = {
1072 JSCLASS_ARRAY,
1073 {NULL, Array_value, 0},
1074 sizeof(ArrayInst_props)/sizeof(*ArrayInst_props),
1075 ArrayInst_props,
1076 Array_destructor,
1077 Array_on_put
1078 };
1079
1080 static HRESULT ArrayConstr_value(script_ctx_t *ctx, vdisp_t *vthis, WORD flags, unsigned argc, jsval_t *argv,
1081 jsval_t *r)
1082 {
1083 jsdisp_t *obj;
1084 DWORD i;
1085 HRESULT hres;
1086
1087 TRACE("\n");
1088
1089 switch(flags) {
1090 case DISPATCH_METHOD:
1091 case DISPATCH_CONSTRUCT: {
1092 if(argc == 1 && is_number(argv[0])) {
1093 double n = get_number(argv[0]);
1094
1095 if(n < 0 || !is_int32(n))
1096 return throw_range_error(ctx, JS_E_INVALID_LENGTH, NULL);
1097
1098 hres = create_array(ctx, n, &obj);
1099 if(FAILED(hres))
1100 return hres;
1101
1102 *r = jsval_obj(obj);
1103 return S_OK;
1104 }
1105
1106 hres = create_array(ctx, argc, &obj);
1107 if(FAILED(hres))
1108 return hres;
1109
1110 for(i=0; i < argc; i++) {
1111 hres = jsdisp_propput_idx(obj, i, argv[i]);
1112 if(FAILED(hres))
1113 break;
1114 }
1115 if(FAILED(hres)) {
1116 jsdisp_release(obj);
1117 return hres;
1118 }
1119
1120 *r = jsval_obj(obj);
1121 break;
1122 }
1123 default:
1124 FIXME("unimplemented flags: %x\n", flags);
1125 return E_NOTIMPL;
1126 }
1127
1128 return S_OK;
1129 }
1130
1131 static HRESULT alloc_array(script_ctx_t *ctx, jsdisp_t *object_prototype, ArrayInstance **ret)
1132 {
1133 ArrayInstance *array;
1134 HRESULT hres;
1135
1136 array = heap_alloc_zero(sizeof(ArrayInstance));
1137 if(!array)
1138 return E_OUTOFMEMORY;
1139
1140 if(object_prototype)
1141 hres = init_dispex(&array->dispex, ctx, &Array_info, object_prototype);
1142 else
1143 hres = init_dispex_from_constr(&array->dispex, ctx, &ArrayInst_info, ctx->array_constr);
1144
1145 if(FAILED(hres)) {
1146 heap_free(array);
1147 return hres;
1148 }
1149
1150 *ret = array;
1151 return S_OK;
1152 }
1153
1154 HRESULT create_array_constr(script_ctx_t *ctx, jsdisp_t *object_prototype, jsdisp_t **ret)
1155 {
1156 ArrayInstance *array;
1157 HRESULT hres;
1158
1159 static const WCHAR ArrayW[] = {'A','r','r','a','y',0};
1160
1161 hres = alloc_array(ctx, object_prototype, &array);
1162 if(FAILED(hres))
1163 return hres;
1164
1165 hres = create_builtin_constructor(ctx, ArrayConstr_value, ArrayW, NULL, PROPF_CONSTR|1, &array->dispex, ret);
1166
1167 jsdisp_release(&array->dispex);
1168 return hres;
1169 }
1170
1171 HRESULT create_array(script_ctx_t *ctx, DWORD length, jsdisp_t **ret)
1172 {
1173 ArrayInstance *array;
1174 HRESULT hres;
1175
1176 hres = alloc_array(ctx, NULL, &array);
1177 if(FAILED(hres))
1178 return hres;
1179
1180 array->length = length;
1181
1182 *ret = &array->dispex;
1183 return S_OK;
1184 }