?  search:
summary | log | graphiclog1 | graphiclog2 | commit | commitdiff | tree | refs | edit | fork
blame | history | raw | HEAD
Release 1.1.37.
[wine/gsoc-2012-control.git] / dlls / jscript / string.c
blob63c0fe8133e2096a8605fb9eb2681005809adb0a
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 #include "wine/debug.h"
22
23 WINE_DEFAULT_DEBUG_CHANNEL(jscript);
24
25 typedef struct {
26 DispatchEx dispex;
27
28 WCHAR *str;
29 DWORD length;
30 } StringInstance;
31
32 static const WCHAR lengthW[] = {'l','e','n','g','t','h',0};
33 static const WCHAR toStringW[] = {'t','o','S','t','r','i','n','g',0};
34 static const WCHAR valueOfW[] = {'v','a','l','u','e','O','f',0};
35 static const WCHAR anchorW[] = {'a','n','c','h','o','r',0};
36 static const WCHAR bigW[] = {'b','i','g',0};
37 static const WCHAR blinkW[] = {'b','l','i','n','k',0};
38 static const WCHAR boldW[] = {'b','o','l','d',0};
39 static const WCHAR charAtW[] = {'c','h','a','r','A','t',0};
40 static const WCHAR charCodeAtW[] = {'c','h','a','r','C','o','d','e','A','t',0};
41 static const WCHAR concatW[] = {'c','o','n','c','a','t',0};
42 static const WCHAR fixedW[] = {'f','i','x','e','d',0};
43 static const WCHAR fontcolorW[] = {'f','o','n','t','c','o','l','o','r',0};
44 static const WCHAR fontsizeW[] = {'f','o','n','t','s','i','z','e',0};
45 static const WCHAR indexOfW[] = {'i','n','d','e','x','O','f',0};
46 static const WCHAR italicsW[] = {'i','t','a','l','i','c','s',0};
47 static const WCHAR lastIndexOfW[] = {'l','a','s','t','I','n','d','e','x','O','f',0};
48 static const WCHAR linkW[] = {'l','i','n','k',0};
49 static const WCHAR matchW[] = {'m','a','t','c','h',0};
50 static const WCHAR replaceW[] = {'r','e','p','l','a','c','e',0};
51 static const WCHAR searchW[] = {'s','e','a','r','c','h',0};
52 static const WCHAR sliceW[] = {'s','l','i','c','e',0};
53 static const WCHAR smallW[] = {'s','m','a','l','l',0};
54 static const WCHAR splitW[] = {'s','p','l','i','t',0};
55 static const WCHAR strikeW[] = {'s','t','r','i','k','e',0};
56 static const WCHAR subW[] = {'s','u','b',0};
57 static const WCHAR substringW[] = {'s','u','b','s','t','r','i','n','g',0};
58 static const WCHAR substrW[] = {'s','u','b','s','t','r',0};
59 static const WCHAR supW[] = {'s','u','p',0};
60 static const WCHAR toLowerCaseW[] = {'t','o','L','o','w','e','r','C','a','s','e',0};
61 static const WCHAR toUpperCaseW[] = {'t','o','U','p','p','e','r','C','a','s','e',0};
62 static const WCHAR toLocaleLowerCaseW[] = {'t','o','L','o','c','a','l','e','L','o','w','e','r','C','a','s','e',0};
63 static const WCHAR toLocaleUpperCaseW[] = {'t','o','L','o','c','a','l','e','U','p','p','e','r','C','a','s','e',0};
64 static const WCHAR localeCompareW[] = {'l','o','c','a','l','e','C','o','m','p','a','r','e',0};
65 static const WCHAR fromCharCodeW[] = {'f','r','o','m','C','h','a','r','C','o','d','e',0};
66
67 static inline StringInstance *string_from_vdisp(vdisp_t *vdisp)
68 {
69 return (StringInstance*)vdisp->u.jsdisp;
70 }
71
72 static inline StringInstance *string_this(vdisp_t *jsthis)
73 {
74 return is_vclass(jsthis, JSCLASS_STRING) ? string_from_vdisp(jsthis) : NULL;
75 }
76
77 static HRESULT get_string_val(script_ctx_t *ctx, vdisp_t *jsthis, jsexcept_t *ei,
78 const WCHAR **str, DWORD *len, BSTR *val_str)
79 {
80 StringInstance *string;
81 VARIANT this_var;
82 HRESULT hres;
83
84 if((string = string_this(jsthis))) {
85 *str = string->str;
86 *len = string->length;
87 *val_str = NULL;
88 return S_OK;
89 }
90
91 V_VT(&this_var) = VT_DISPATCH;
92 V_DISPATCH(&this_var) = jsthis->u.disp;
93 hres = to_string(ctx, &this_var, ei, val_str);
94 if(FAILED(hres))
95 return hres;
96
97 *str = *val_str;
98 *len = SysStringLen(*val_str);
99 return S_OK;
100 }
101
102 static HRESULT String_length(script_ctx_t *ctx, vdisp_t *jsthis, WORD flags, DISPPARAMS *dp,
103 VARIANT *retv, jsexcept_t *ei, IServiceProvider *sp)
104 {
105 TRACE("%p\n", jsthis);
106
107 switch(flags) {
108 case DISPATCH_PROPERTYGET: {
109 StringInstance *string = string_from_vdisp(jsthis);
110
111 V_VT(retv) = VT_I4;
112 V_I4(retv) = string->length;
113 break;
114 }
115 default:
116 FIXME("unimplemented flags %x\n", flags);
117 return E_NOTIMPL;
118 }
119
120 return S_OK;
121 }
122
123 static HRESULT stringobj_to_string(vdisp_t *jsthis, VARIANT *retv)
124 {
125 StringInstance *string;
126
127 if(!(string = string_this(jsthis))) {
128 WARN("this is not a string object\n");
129 return E_FAIL;
130 }
131
132 if(retv) {
133 BSTR str = SysAllocString(string->str);
134 if(!str)
135 return E_OUTOFMEMORY;
136
137 V_VT(retv) = VT_BSTR;
138 V_BSTR(retv) = str;
139 }
140 return S_OK;
141 }
142
143 /* ECMA-262 3rd Edition 15.5.4.2 */
144 static HRESULT String_toString(script_ctx_t *ctx, vdisp_t *jsthis, WORD flags, DISPPARAMS *dp,
145 VARIANT *retv, jsexcept_t *ei, IServiceProvider *sp)
146 {
147 TRACE("\n");
148
149 return stringobj_to_string(jsthis, retv);
150 }
151
152 /* ECMA-262 3rd Edition 15.5.4.2 */
153 static HRESULT String_valueOf(script_ctx_t *ctx, vdisp_t *jsthis, WORD flags, DISPPARAMS *dp,
154 VARIANT *retv, jsexcept_t *ei, IServiceProvider *sp)
155 {
156 TRACE("\n");
157
158 return stringobj_to_string(jsthis, retv);
159 }
160
161 static HRESULT do_attributeless_tag_format(script_ctx_t *ctx, vdisp_t *jsthis, WORD flags, DISPPARAMS *dp,
162 VARIANT *retv, jsexcept_t *ei, IServiceProvider *sp, const WCHAR *tagname)
163 {
164 const WCHAR *str;
165 DWORD length;
166 BSTR val_str = NULL;
167 HRESULT hres;
168
169 static const WCHAR tagfmt[] = {'<','%','s','>','%','s','<','/','%','s','>',0};
170
171 hres = get_string_val(ctx, jsthis, ei, &str, &length, &val_str);
172 if(FAILED(hres))
173 return hres;
174
175 if(retv) {
176 BSTR ret = SysAllocStringLen(NULL, length + 2*strlenW(tagname) + 5);
177 if(!ret) {
178 SysFreeString(val_str);
179 return E_OUTOFMEMORY;
180 }
181
182 sprintfW(ret, tagfmt, tagname, str, tagname);
183
184 V_VT(retv) = VT_BSTR;
185 V_BSTR(retv) = ret;
186 }
187
188 SysFreeString(val_str);
189 return S_OK;
190 }
191
192 static HRESULT do_attribute_tag_format(script_ctx_t *ctx, vdisp_t *jsthis, WORD flags,
193 DISPPARAMS *dp, VARIANT *retv, jsexcept_t *ei, IServiceProvider *sp,
194 const WCHAR *tagname, const WCHAR *attr)
195 {
196 static const WCHAR tagfmtW[]
197 = {'<','%','s',' ','%','s','=','\"','%','s','\"','>','%','s','<','/','%','s','>',0};
198 static const WCHAR undefinedW[] = {'u','n','d','e','f','i','n','e','d',0};
199
200 StringInstance *string;
201 const WCHAR *str;
202 DWORD length;
203 BSTR attr_value, val_str = NULL;
204 HRESULT hres;
205
206 if(!(string = string_this(jsthis))) {
207 VARIANT this;
208
209 V_VT(&this) = VT_DISPATCH;
210 V_DISPATCH(&this) = jsthis->u.disp;
211
212 hres = to_string(ctx, &this, ei, &val_str);
213 if(FAILED(hres))
214 return hres;
215
216 str = val_str;
217 length = SysStringLen(val_str);
218 }
219 else {
220 str = string->str;
221 length = string->length;
222 }
223
224 if(arg_cnt(dp)) {
225 hres = to_string(ctx, get_arg(dp, 0), ei, &attr_value);
226 if(FAILED(hres)) {
227 SysFreeString(val_str);
228 return hres;
229 }
230 }
231 else {
232 attr_value = SysAllocString(undefinedW);
233 if(!attr_value) {
234 SysFreeString(val_str);
235 return E_OUTOFMEMORY;
236 }
237 }
238
239 if(retv) {
240 BSTR ret = SysAllocStringLen(NULL, length + 2*strlenW(tagname)
241 + strlenW(attr) + SysStringLen(attr_value) + 9);
242 if(!ret) {
243 SysFreeString(attr_value);
244 SysFreeString(val_str);
245 return E_OUTOFMEMORY;
246 }
247
248 sprintfW(ret, tagfmtW, tagname, attr, attr_value, str, tagname);
249
250 V_VT(retv) = VT_BSTR;
251 V_BSTR(retv) = ret;
252 }
253
254 SysFreeString(attr_value);
255 SysFreeString(val_str);
256 return S_OK;
257 }
258
259 static HRESULT String_anchor(script_ctx_t *ctx, vdisp_t *jsthis, WORD flags, DISPPARAMS *dp,
260 VARIANT *retv, jsexcept_t *ei, IServiceProvider *sp)
261 {
262 static const WCHAR fontW[] = {'A',0};
263 static const WCHAR colorW[] = {'N','A','M','E',0};
264
265 return do_attribute_tag_format(ctx, jsthis, flags, dp, retv, ei, sp, fontW, colorW);
266 }
267
268 static HRESULT String_big(script_ctx_t *ctx, vdisp_t *jsthis, WORD flags, DISPPARAMS *dp,
269 VARIANT *retv, jsexcept_t *ei, IServiceProvider *sp)
270 {
271 static const WCHAR bigtagW[] = {'B','I','G',0};
272 return do_attributeless_tag_format(ctx, jsthis, flags, dp, retv, ei, sp, bigtagW);
273 }
274
275 static HRESULT String_blink(script_ctx_t *ctx, vdisp_t *jsthis, WORD flags, DISPPARAMS *dp,
276 VARIANT *retv, jsexcept_t *ei, IServiceProvider *sp)
277 {
278 static const WCHAR blinktagW[] = {'B','L','I','N','K',0};
279 return do_attributeless_tag_format(ctx, jsthis, flags, dp, retv, ei, sp, blinktagW);
280 }
281
282 static HRESULT String_bold(script_ctx_t *ctx, vdisp_t *jsthis, WORD flags, DISPPARAMS *dp,
283 VARIANT *retv, jsexcept_t *ei, IServiceProvider *sp)
284 {
285 static const WCHAR boldtagW[] = {'B',0};
286 return do_attributeless_tag_format(ctx, jsthis, flags, dp, retv, ei, sp, boldtagW);
287 }
288
289 /* ECMA-262 3rd Edition 15.5.4.5 */
290 static HRESULT String_charAt(script_ctx_t *ctx, vdisp_t *jsthis, WORD flags, DISPPARAMS *dp,
291 VARIANT *retv, jsexcept_t *ei, IServiceProvider *sp)
292 {
293 const WCHAR *str;
294 DWORD length;
295 BSTR ret, val_str;
296 INT pos = 0;
297 HRESULT hres;
298
299 TRACE("\n");
300
301 hres = get_string_val(ctx, jsthis, ei, &str, &length, &val_str);
302 if(FAILED(hres))
303 return hres;
304
305 if(arg_cnt(dp)) {
306 VARIANT num;
307
308 hres = to_integer(ctx, get_arg(dp, 0), ei, &num);
309 if(FAILED(hres)) {
310 SysFreeString(val_str);
311 return hres;
312 }
313
314 if(V_VT(&num) == VT_I4) {
315 pos = V_I4(&num);
316 }else {
317 WARN("pos = %lf\n", V_R8(&num));
318 pos = -1;
319 }
320 }
321
322 if(!retv) {
323 SysFreeString(val_str);
324 return S_OK;
325 }
326
327 if(0 <= pos && pos < length)
328 ret = SysAllocStringLen(str+pos, 1);
329 else
330 ret = SysAllocStringLen(NULL, 0);
331 SysFreeString(val_str);
332 if(!ret) {
333 return E_OUTOFMEMORY;
334 }
335
336 V_VT(retv) = VT_BSTR;
337 V_BSTR(retv) = ret;
338 return S_OK;
339 }
340
341 /* ECMA-262 3rd Edition 15.5.4.5 */
342 static HRESULT String_charCodeAt(script_ctx_t *ctx, vdisp_t *jsthis, WORD flags, DISPPARAMS *dp,
343 VARIANT *retv, jsexcept_t *ei, IServiceProvider *sp)
344 {
345 const WCHAR *str;
346 BSTR val_str;
347 DWORD length, idx = 0;
348 HRESULT hres;
349
350 TRACE("\n");
351
352 hres = get_string_val(ctx, jsthis, ei, &str, &length, &val_str);
353 if(FAILED(hres))
354 return hres;
355
356 if(arg_cnt(dp) > 0) {
357 VARIANT v;
358
359 hres = to_integer(ctx, get_arg(dp, 0), ei, &v);
360 if(FAILED(hres)) {
361 SysFreeString(val_str);
362 return hres;
363 }
364
365 if(V_VT(&v) != VT_I4 || V_I4(&v) < 0 || V_I4(&v) >= length) {
366 if(retv) num_set_nan(&v);
367 SysFreeString(val_str);
368 return S_OK;
369 }
370
371 idx = V_I4(&v);
372 }
373
374 if(retv) {
375 V_VT(retv) = VT_I4;
376 V_I4(retv) = str[idx];
377 }
378
379 SysFreeString(val_str);
380 return S_OK;
381 }
382
383 /* ECMA-262 3rd Edition 15.5.4.6 */
384 static HRESULT String_concat(script_ctx_t *ctx, vdisp_t *jsthis, WORD flags, DISPPARAMS *dp,
385 VARIANT *retv, jsexcept_t *ei, IServiceProvider *sp)
386 {
387 BSTR *strs = NULL, ret = NULL;
388 DWORD len = 0, i, l, str_cnt;
389 VARIANT var;
390 WCHAR *ptr;
391 HRESULT hres;
392
393 TRACE("\n");
394
395 str_cnt = arg_cnt(dp)+1;
396 strs = heap_alloc_zero(str_cnt * sizeof(BSTR));
397 if(!strs)
398 return E_OUTOFMEMORY;
399
400 V_VT(&var) = VT_DISPATCH;
401 V_DISPATCH(&var) = jsthis->u.disp;
402
403 hres = to_string(ctx, &var, ei, strs);
404 if(SUCCEEDED(hres)) {
405 for(i=0; i < arg_cnt(dp); i++) {
406 hres = to_string(ctx, get_arg(dp, i), ei, strs+i+1);
407 if(FAILED(hres))
408 break;
409 }
410 }
411
412 if(SUCCEEDED(hres)) {
413 for(i=0; i < str_cnt; i++)
414 len += SysStringLen(strs[i]);
415
416 ptr = ret = SysAllocStringLen(NULL, len);
417
418 for(i=0; i < str_cnt; i++) {
419 l = SysStringLen(strs[i]);
420 memcpy(ptr, strs[i], l*sizeof(WCHAR));
421 ptr += l;
422 }
423 }
424
425 for(i=0; i < str_cnt; i++)
426 SysFreeString(strs[i]);
427 heap_free(strs);
428
429 if(FAILED(hres))
430 return hres;
431
432 if(retv) {
433 V_VT(retv) = VT_BSTR;
434 V_BSTR(retv) = ret;
435 }else {
436 SysFreeString(ret);
437 }
438 return S_OK;
439 }
440
441 static HRESULT String_fixed(script_ctx_t *ctx, vdisp_t *jsthis, WORD flags, DISPPARAMS *dp,
442 VARIANT *retv, jsexcept_t *ei, IServiceProvider *sp)
443 {
444 static const WCHAR fixedtagW[] = {'T','T',0};
445 return do_attributeless_tag_format(ctx, jsthis, flags, dp, retv, ei, sp, fixedtagW);
446 }
447
448 static HRESULT String_fontcolor(script_ctx_t *ctx, vdisp_t *jsthis, WORD flags, DISPPARAMS *dp,
449 VARIANT *retv, jsexcept_t *ei, IServiceProvider *sp)
450 {
451 static const WCHAR fontW[] = {'F','O','N','T',0};
452 static const WCHAR colorW[] = {'C','O','L','O','R',0};
453
454 return do_attribute_tag_format(ctx, jsthis, flags, dp, retv, ei, sp, fontW, colorW);
455 }
456
457 static HRESULT String_fontsize(script_ctx_t *ctx, vdisp_t *jsthis, WORD flags, DISPPARAMS *dp,
458 VARIANT *retv, jsexcept_t *ei, IServiceProvider *sp)
459 {
460 static const WCHAR fontW[] = {'F','O','N','T',0};
461 static const WCHAR colorW[] = {'S','I','Z','E',0};
462
463 return do_attribute_tag_format(ctx, jsthis, flags, dp, retv, ei, sp, fontW, colorW);
464 }
465
466 static HRESULT String_indexOf(script_ctx_t *ctx, vdisp_t *jsthis, WORD flags, DISPPARAMS *dp,
467 VARIANT *retv, jsexcept_t *ei, IServiceProvider *sp)
468 {
469 DWORD length, pos = 0;
470 const WCHAR *str;
471 BSTR search_str, val_str;
472 INT ret = -1;
473 HRESULT hres;
474
475 TRACE("\n");
476
477 hres = get_string_val(ctx, jsthis, ei, &str, &length, &val_str);
478 if(FAILED(hres))
479 return hres;
480
481 if(!arg_cnt(dp)) {
482 if(retv) {
483 V_VT(retv) = VT_I4;
484 V_I4(retv) = -1;
485 }
486 SysFreeString(val_str);
487 return S_OK;
488 }
489
490 hres = to_string(ctx, get_arg(dp,0), ei, &search_str);
491 if(FAILED(hres)) {
492 SysFreeString(val_str);
493 return hres;
494 }
495
496 if(arg_cnt(dp) >= 2) {
497 VARIANT ival;
498
499 hres = to_integer(ctx, get_arg(dp,1), ei, &ival);
500 if(SUCCEEDED(hres)) {
501 if(V_VT(&ival) == VT_I4)
502 pos = V_VT(&ival) > 0 ? V_I4(&ival) : 0;
503 else
504 pos = V_R8(&ival) > 0.0 ? length : 0;
505 if(pos > length)
506 pos = length;
507 }
508 }
509
510 if(SUCCEEDED(hres)) {
511 const WCHAR *ptr;
512
513 ptr = strstrW(str+pos, search_str);
514 if(ptr)
515 ret = ptr - str;
516 else
517 ret = -1;
518 }
519
520 SysFreeString(search_str);
521 SysFreeString(val_str);
522 if(FAILED(hres))
523 return hres;
524
525 if(retv) {
526 V_VT(retv) = VT_I4;
527 V_I4(retv) = ret;
528 }
529 return S_OK;
530 }
531
532 static HRESULT String_italics(script_ctx_t *ctx, vdisp_t *jsthis, WORD flags, DISPPARAMS *dp,
533 VARIANT *retv, jsexcept_t *ei, IServiceProvider *sp)
534 {
535 static const WCHAR italicstagW[] = {'I',0};
536 return do_attributeless_tag_format(ctx, jsthis, flags, dp, retv, ei, sp, italicstagW);
537 }
538
539 /* ECMA-262 3rd Edition 15.5.4.8 */
540 static HRESULT String_lastIndexOf(script_ctx_t *ctx, vdisp_t *jsthis, WORD flags, DISPPARAMS *dp,
541 VARIANT *retv, jsexcept_t *ei, IServiceProvider *sp)
542 {
543 BSTR search_str, val_str;
544 DWORD length, pos, search_len;
545 const WCHAR *str;
546 INT ret = -1;
547 HRESULT hres;
548
549 TRACE("\n");
550
551 hres = get_string_val(ctx, jsthis, ei, &str, &length, &val_str);
552 if(FAILED(hres))
553 return hres;
554
555 if(!arg_cnt(dp)) {
556 if(retv) {
557 V_VT(retv) = VT_I4;
558 V_I4(retv) = -1;
559 }
560 SysFreeString(val_str);
561 return S_OK;
562 }
563
564 hres = to_string(ctx, get_arg(dp,0), ei, &search_str);
565 if(FAILED(hres)) {
566 SysFreeString(val_str);
567 return hres;
568 }
569
570 search_len = SysStringLen(search_str);
571
572 if(arg_cnt(dp) >= 2) {
573 VARIANT ival;
574
575 hres = to_integer(ctx, get_arg(dp,1), ei, &ival);
576 if(SUCCEEDED(hres)) {
577 if(V_VT(&ival) == VT_I4)
578 pos = V_VT(&ival) > 0 ? V_I4(&ival) : 0;
579 else
580 pos = V_R8(&ival) > 0.0 ? length : 0;
581 if(pos > length)
582 pos = length;
583 }
584 }else {
585 pos = length;
586 }
587
588 if(SUCCEEDED(hres) && length >= search_len) {
589 const WCHAR *ptr;
590
591 for(ptr = str+min(pos, length-search_len); ptr >= str; ptr--) {
592 if(!memcmp(ptr, search_str, search_len*sizeof(WCHAR))) {
593 ret = ptr-str;
594 break;
595 }
596 }
597 }
598
599 SysFreeString(search_str);
600 SysFreeString(val_str);
601 if(FAILED(hres))
602 return hres;
603
604 if(retv) {
605 V_VT(retv) = VT_I4;
606 V_I4(retv) = ret;
607 }
608 return S_OK;
609 }
610
611 static HRESULT String_link(script_ctx_t *ctx, vdisp_t *jsthis, WORD flags, DISPPARAMS *dp,
612 VARIANT *retv, jsexcept_t *ei, IServiceProvider *sp)
613 {
614 static const WCHAR fontW[] = {'A',0};
615 static const WCHAR colorW[] = {'H','R','E','F',0};
616
617 return do_attribute_tag_format(ctx, jsthis, flags, dp, retv, ei, sp, fontW, colorW);
618 }
619
620 /* ECMA-262 3rd Edition 15.5.4.10 */
621 static HRESULT String_match(script_ctx_t *ctx, vdisp_t *jsthis, WORD flags, DISPPARAMS *dp,
622 VARIANT *retv, jsexcept_t *ei, IServiceProvider *sp)
623 {
624 const WCHAR *str;
625 match_result_t *match_result;
626 DispatchEx *regexp;
627 DispatchEx *array;
628 VARIANT var, *arg_var;
629 DWORD length, match_cnt, i;
630 BSTR val_str = NULL;
631 HRESULT hres = S_OK;
632
633 TRACE("\n");
634
635 if(!arg_cnt(dp)) {
636 if(retv) {
637 V_VT(retv) = VT_NULL;
638 }
639
640 return S_OK;
641 }
642
643 arg_var = get_arg(dp, 0);
644 switch(V_VT(arg_var)) {
645 case VT_DISPATCH:
646 regexp = iface_to_jsdisp((IUnknown*)V_DISPATCH(arg_var));
647 if(regexp) {
648 if(regexp->builtin_info->class == JSCLASS_REGEXP)
649 break;
650 jsdisp_release(regexp);
651 }
652 default: {
653 BSTR match_str;
654
655 hres = to_string(ctx, arg_var, ei, &match_str);
656 if(FAILED(hres))
657 return hres;
658
659 hres = create_regexp(ctx, match_str, SysStringLen(match_str), 0, &regexp);
660 SysFreeString(match_str);
661 if(FAILED(hres))
662 return hres;
663 }
664 }
665
666 hres = get_string_val(ctx, jsthis, ei, &str, &length, &val_str);
667 if(SUCCEEDED(hres))
668 hres = regexp_match(ctx, regexp, str, length, FALSE, &match_result, &match_cnt);
669 jsdisp_release(regexp);
670 if(FAILED(hres)) {
671 SysFreeString(val_str);
672 return hres;
673 }
674
675 if(!match_cnt) {
676 TRACE("no match\n");
677
678 if(retv)
679 V_VT(retv) = VT_NULL;
680
681 SysFreeString(val_str);
682 return S_OK;
683 }
684
685 hres = create_array(ctx, match_cnt, &array);
686 if(FAILED(hres)) {
687 SysFreeString(val_str);
688 return hres;
689 }
690
691 V_VT(&var) = VT_BSTR;
692
693 for(i=0; i < match_cnt; i++) {
694 V_BSTR(&var) = SysAllocStringLen(match_result[i].str, match_result[i].len);
695 if(!V_BSTR(&var)) {
696 hres = E_OUTOFMEMORY;
697 break;
698 }
699
700 hres = jsdisp_propput_idx(array, i, &var, ei, NULL/*FIXME*/);
701 SysFreeString(V_BSTR(&var));
702 if(FAILED(hres))
703 break;
704 }
705
706 heap_free(match_result);
707 SysFreeString(val_str);
708
709 if(SUCCEEDED(hres) && retv) {
710 V_VT(retv) = VT_DISPATCH;
711 V_DISPATCH(retv) = (IDispatch*)_IDispatchEx_(array);
712 }else {
713 jsdisp_release(array);
714 }
715 return hres;
716 }
717
718 typedef struct {
719 WCHAR *buf;
720 DWORD size;
721 DWORD len;
722 } strbuf_t;
723
724 static HRESULT strbuf_append(strbuf_t *buf, const WCHAR *str, DWORD len)
725 {
726 if(!len)
727 return S_OK;
728
729 if(len + buf->len > buf->size) {
730 WCHAR *new_buf;
731 DWORD new_size;
732
733 new_size = buf->size ? buf->size<<1 : 16;
734 if(new_size < buf->len+len)
735 new_size = buf->len+len;
736 if(buf->buf)
737 new_buf = heap_realloc(buf->buf, new_size*sizeof(WCHAR));
738 else
739 new_buf = heap_alloc(new_size*sizeof(WCHAR));
740 if(!new_buf)
741 return E_OUTOFMEMORY;
742
743 buf->buf = new_buf;
744 buf->size = new_size;
745 }
746
747 memcpy(buf->buf+buf->len, str, len*sizeof(WCHAR));
748 buf->len += len;
749 return S_OK;
750 }
751
752 static HRESULT rep_call(script_ctx_t *ctx, DispatchEx *func, const WCHAR *str, match_result_t *match,
753 match_result_t *parens, DWORD parens_cnt, BSTR *ret, jsexcept_t *ei, IServiceProvider *caller)
754 {
755 DISPPARAMS dp = {NULL, NULL, 0, 0};
756 VARIANTARG *args, *arg;
757 VARIANT var;
758 DWORD i;
759 HRESULT hres = S_OK;
760
761 dp.cArgs = parens_cnt+3;
762 dp.rgvarg = args = heap_alloc_zero(sizeof(VARIANT)*dp.cArgs);
763 if(!args)
764 return E_OUTOFMEMORY;
765
766 arg = get_arg(&dp,0);
767 V_VT(arg) = VT_BSTR;
768 V_BSTR(arg) = SysAllocStringLen(match->str, match->len);
769 if(!V_BSTR(arg))
770 hres = E_OUTOFMEMORY;
771
772 if(SUCCEEDED(hres)) {
773 for(i=0; i < parens_cnt; i++) {
774 arg = get_arg(&dp,i+1);
775 V_VT(arg) = VT_BSTR;
776 V_BSTR(arg) = SysAllocStringLen(parens[i].str, parens[i].len);
777 if(!V_BSTR(arg)) {
778 hres = E_OUTOFMEMORY;
779 break;
780 }
781 }
782 }
783
784 if(SUCCEEDED(hres)) {
785 arg = get_arg(&dp,parens_cnt+1);
786 V_VT(arg) = VT_I4;
787 V_I4(arg) = match->str - str;
788
789 arg = get_arg(&dp,parens_cnt+2);
790 V_VT(arg) = VT_BSTR;
791 V_BSTR(arg) = SysAllocString(str);
792 if(!V_BSTR(arg))
793 hres = E_OUTOFMEMORY;
794 }
795
796 if(SUCCEEDED(hres))
797 hres = jsdisp_call_value(func, DISPATCH_METHOD, &dp, &var, ei, caller);
798
799 for(i=0; i < parens_cnt+3; i++) {
800 if(i != parens_cnt+1)
801 SysFreeString(V_BSTR(get_arg(&dp,i)));
802 }
803 heap_free(args);
804
805 if(FAILED(hres))
806 return hres;
807
808 hres = to_string(ctx, &var, ei, ret);
809 VariantClear(&var);
810 return hres;
811 }
812
813 /* ECMA-262 3rd Edition 15.5.4.11 */
814 static HRESULT String_replace(script_ctx_t *ctx, vdisp_t *jsthis, WORD flags, DISPPARAMS *dp,
815 VARIANT *retv, jsexcept_t *ei, IServiceProvider *caller)
816 {
817 const WCHAR *str;
818 DWORD parens_cnt = 0, parens_size=0, rep_len=0, length;
819 BSTR rep_str = NULL, match_str = NULL, ret_str, val_str;
820 DispatchEx *rep_func = NULL, *regexp = NULL;
821 match_result_t *parens = NULL, match, **parens_ptr = &parens;
822 strbuf_t ret = {NULL,0,0};
823 BOOL gcheck = FALSE;
824 VARIANT *arg_var;
825 HRESULT hres = S_OK;
826
827 TRACE("\n");
828
829 hres = get_string_val(ctx, jsthis, ei, &str, &length, &val_str);
830 if(FAILED(hres))
831 return hres;
832
833 if(!arg_cnt(dp)) {
834 if(retv) {
835 if(!val_str) {
836 val_str = SysAllocStringLen(str, length);
837 if(!val_str)
838 return E_OUTOFMEMORY;
839 }
840
841 V_VT(retv) = VT_BSTR;
842 V_BSTR(retv) = val_str;
843 }
844 return S_OK;
845 }
846
847 arg_var = get_arg(dp, 0);
848 switch(V_VT(arg_var)) {
849 case VT_DISPATCH:
850 regexp = iface_to_jsdisp((IUnknown*)V_DISPATCH(arg_var));
851 if(regexp) {
852 if(is_class(regexp, JSCLASS_REGEXP)) {
853 break;
854 }else {
855 jsdisp_release(regexp);
856 regexp = NULL;
857 }
858 }
859
860 default:
861 hres = to_string(ctx, arg_var, ei, &match_str);
862 if(FAILED(hres)) {
863 SysFreeString(val_str);
864 return hres;
865 }
866 }
867
868 if(arg_cnt(dp) >= 2) {
869 arg_var = get_arg(dp,1);
870 switch(V_VT(arg_var)) {
871 case VT_DISPATCH:
872 rep_func = iface_to_jsdisp((IUnknown*)V_DISPATCH(arg_var));
873 if(rep_func) {
874 if(is_class(rep_func, JSCLASS_FUNCTION)) {
875 break;
876 }else {
877 jsdisp_release(rep_func);
878 rep_func = NULL;
879 }
880 }
881
882 default:
883 hres = to_string(ctx, arg_var, ei, &rep_str);
884 if(FAILED(hres))
885 break;
886
887 rep_len = SysStringLen(rep_str);
888 if(!strchrW(rep_str, '$'))
889 parens_ptr = NULL;
890 }
891 }
892
893 if(SUCCEEDED(hres)) {
894 const WCHAR *cp, *ecp;
895
896 cp = ecp = str;
897
898 while(1) {
899 if(regexp) {
900 hres = regexp_match_next(ctx, regexp, gcheck, str, length, &cp, parens_ptr,
901 &parens_size, &parens_cnt, &match);
902 gcheck = TRUE;
903
904 if(hres == S_FALSE) {
905 hres = S_OK;
906 break;
907 }
908 if(FAILED(hres))
909 break;
910 }else {
911 match.str = strstrW(cp, match_str);
912 if(!match.str)
913 break;
914 match.len = SysStringLen(match_str);
915 cp = match.str+match.len;
916 }
917
918 hres = strbuf_append(&ret, ecp, match.str-ecp);
919 ecp = match.str+match.len;
920 if(FAILED(hres))
921 break;
922
923 if(rep_func) {
924 BSTR cstr;
925
926 hres = rep_call(ctx, rep_func, str, &match, parens, parens_cnt, &cstr, ei, caller);
927 if(FAILED(hres))
928 break;
929
930 hres = strbuf_append(&ret, cstr, SysStringLen(cstr));
931 SysFreeString(cstr);
932 if(FAILED(hres))
933 break;
934 }else if(rep_str && regexp) {
935 const WCHAR *ptr = rep_str, *ptr2;
936
937 while((ptr2 = strchrW(ptr, '$'))) {
938 hres = strbuf_append(&ret, ptr, ptr2-ptr);
939 if(FAILED(hres))
940 break;
941
942 switch(ptr2[1]) {
943 case '$':
944 hres = strbuf_append(&ret, ptr2, 1);
945 ptr = ptr2+2;
946 break;
947 case '&':
948 hres = strbuf_append(&ret, match.str, match.len);
949 ptr = ptr2+2;
950 break;
951 case '`':
952 hres = strbuf_append(&ret, str, match.str-str);
953 ptr = ptr2+2;
954 break;
955 case '\'':
956 hres = strbuf_append(&ret, ecp, (str+length)-ecp);
957 ptr = ptr2+2;
958 break;
959 default: {
960 DWORD idx;
961
962 if(!isdigitW(ptr2[1])) {
963 hres = strbuf_append(&ret, ptr2, 1);
964 ptr = ptr2+1;
965 break;
966 }
967
968 idx = ptr2[1] - '0';
969 if(isdigitW(ptr2[2]) && idx*10 + (ptr2[2]-'0') <= parens_cnt) {
970 idx = idx*10 + (ptr[2]-'0');
971 ptr = ptr2+3;
972 }else if(idx && idx <= parens_cnt) {
973 ptr = ptr2+2;
974 }else {
975 hres = strbuf_append(&ret, ptr2, 1);
976 ptr = ptr2+1;
977 break;
978 }
979
980 hres = strbuf_append(&ret, parens[idx-1].str, parens[idx-1].len);
981 }
982 }
983
984 if(FAILED(hres))
985 break;
986 }
987
988 if(SUCCEEDED(hres))
989 hres = strbuf_append(&ret, ptr, (rep_str+rep_len)-ptr);
990 if(FAILED(hres))
991 break;
992 }else if(rep_str) {
993 hres = strbuf_append(&ret, rep_str, rep_len);
994 if(FAILED(hres))
995 break;
996 }else {
997 static const WCHAR undefinedW[] = {'u','n','d','e','f','i','n','e','d'};
998
999 hres = strbuf_append(&ret, undefinedW, sizeof(undefinedW)/sizeof(WCHAR));
1000 if(FAILED(hres))
1001 break;
1002 }
1003 }
1004
1005 if(SUCCEEDED(hres))
1006 hres = strbuf_append(&ret, ecp, (str+length)-ecp);
1007 }
1008
1009 if(rep_func)
1010 jsdisp_release(rep_func);
1011 if(regexp)
1012 jsdisp_release(regexp);
1013 SysFreeString(val_str);
1014 SysFreeString(rep_str);
1015 SysFreeString(match_str);
1016 heap_free(parens);
1017
1018 if(SUCCEEDED(hres) && retv) {
1019 ret_str = SysAllocStringLen(ret.buf, ret.len);
1020 if(!ret_str)
1021 return E_OUTOFMEMORY;
1022
1023 V_VT(retv) = VT_BSTR;
1024 V_BSTR(retv) = ret_str;
1025 TRACE("= %s\n", debugstr_w(ret_str));
1026 }
1027
1028 heap_free(ret.buf);
1029 return hres;
1030 }
1031
1032 static HRESULT String_search(script_ctx_t *ctx, vdisp_t *jsthis, WORD flags, DISPPARAMS *dp,
1033 VARIANT *retv, jsexcept_t *ei, IServiceProvider *sp)
1034 {
1035 FIXME("\n");
1036 return E_NOTIMPL;
1037 }
1038
1039 /* ECMA-262 3rd Edition 15.5.4.13 */
1040 static HRESULT String_slice(script_ctx_t *ctx, vdisp_t *jsthis, WORD flags, DISPPARAMS *dp,
1041 VARIANT *retv, jsexcept_t *ei, IServiceProvider *sp)
1042 {
1043 const WCHAR *str;
1044 BSTR val_str;
1045 DWORD length;
1046 INT start=0, end;
1047 VARIANT v;
1048 HRESULT hres;
1049
1050 TRACE("\n");
1051
1052 hres = get_string_val(ctx, jsthis, ei, &str, &length, &val_str);
1053 if(FAILED(hres))
1054 return hres;
1055
1056 if(arg_cnt(dp)) {
1057 hres = to_integer(ctx, get_arg(dp,0), ei, &v);
1058 if(FAILED(hres)) {
1059 SysFreeString(val_str);
1060 return hres;
1061 }
1062
1063 if(V_VT(&v) == VT_I4) {
1064 start = V_I4(&v);
1065 if(start < 0) {
1066 start = length + start;
1067 if(start < 0)
1068 start = 0;
1069 }else if(start > length) {
1070 start = length;
1071 }
1072 }else {
1073 start = V_R8(&v) < 0.0 ? 0 : length;
1074 }
1075 }else {
1076 start = 0;
1077 }
1078
1079 if(arg_cnt(dp) >= 2) {
1080 hres = to_integer(ctx, get_arg(dp,1), ei, &v);
1081 if(FAILED(hres)) {
1082 SysFreeString(val_str);
1083 return hres;
1084 }
1085
1086 if(V_VT(&v) == VT_I4) {
1087 end = V_I4(&v);
1088 if(end < 0) {
1089 end = length + end;
1090 if(end < 0)
1091 end = 0;
1092 }else if(end > length) {
1093 end = length;
1094 }
1095 }else {
1096 end = V_R8(&v) < 0.0 ? 0 : length;
1097 }
1098 }else {
1099 end = length;
1100 }
1101
1102 if(end < start)
1103 end = start;
1104
1105 if(retv) {
1106 BSTR retstr = SysAllocStringLen(str+start, end-start);
1107 if(!retstr) {
1108 SysFreeString(val_str);
1109 return E_OUTOFMEMORY;
1110 }
1111
1112 V_VT(retv) = VT_BSTR;
1113 V_BSTR(retv) = retstr;
1114 }
1115
1116 SysFreeString(val_str);
1117 return S_OK;
1118 }
1119
1120 static HRESULT String_small(script_ctx_t *ctx, vdisp_t *jsthis, WORD flags, DISPPARAMS *dp,
1121 VARIANT *retv, jsexcept_t *ei, IServiceProvider *sp)
1122 {
1123 static const WCHAR smalltagW[] = {'S','M','A','L','L',0};
1124 return do_attributeless_tag_format(ctx, jsthis, flags, dp, retv, ei, sp, smalltagW);
1125 }
1126
1127 static HRESULT String_split(script_ctx_t *ctx, vdisp_t *jsthis, WORD flags, DISPPARAMS *dp,
1128 VARIANT *retv, jsexcept_t *ei, IServiceProvider *sp)
1129 {
1130 match_result_t *match_result = NULL;
1131 DWORD length, match_cnt, i, match_len = 0;
1132 const WCHAR *str, *ptr, *ptr2;
1133 BOOL use_regexp = FALSE;
1134 VARIANT *arg, var;
1135 DispatchEx *array;
1136 BSTR val_str, match_str = NULL;
1137 HRESULT hres;
1138
1139 TRACE("\n");
1140
1141 if(arg_cnt(dp) != 1) {
1142 FIXME("unsupported args\n");
1143 return E_NOTIMPL;
1144 }
1145
1146 hres = get_string_val(ctx, jsthis, ei, &str, &length, &val_str);
1147 if(FAILED(hres))
1148 return hres;
1149
1150 arg = get_arg(dp, 0);
1151 switch(V_VT(arg)) {
1152 case VT_DISPATCH: {
1153 DispatchEx *regexp;
1154
1155 regexp = iface_to_jsdisp((IUnknown*)V_DISPATCH(arg));
1156 if(regexp) {
1157 if(is_class(regexp, JSCLASS_REGEXP)) {
1158 use_regexp = TRUE;
1159 hres = regexp_match(ctx, regexp, str, length, TRUE, &match_result, &match_cnt);
1160 jsdisp_release(regexp);
1161 if(FAILED(hres)) {
1162 SysFreeString(val_str);
1163 return hres;
1164 }
1165 break;
1166 }
1167 jsdisp_release(regexp);
1168 }
1169 }
1170 default:
1171 hres = to_string(ctx, arg, ei, &match_str);
1172 if(FAILED(hres)) {
1173 SysFreeString(val_str);
1174 return hres;
1175 }
1176
1177 match_len = SysStringLen(match_str);
1178 if(!match_len) {
1179 SysFreeString(match_str);
1180 match_str = NULL;
1181 }
1182 }
1183
1184 hres = create_array(ctx, 0, &array);
1185
1186 if(SUCCEEDED(hres)) {
1187 ptr = str;
1188 for(i=0;; i++) {
1189 if(use_regexp) {
1190 if(i == match_cnt)
1191 break;
1192 ptr2 = match_result[i].str;
1193 }else if(match_str) {
1194 ptr2 = strstrW(ptr, match_str);
1195 if(!ptr2)
1196 break;
1197 }else {
1198 if(!*ptr)
1199 break;
1200 ptr2 = ptr+1;
1201 }
1202
1203 V_VT(&var) = VT_BSTR;
1204 V_BSTR(&var) = SysAllocStringLen(ptr, ptr2-ptr);
1205 if(!V_BSTR(&var)) {
1206 hres = E_OUTOFMEMORY;
1207 break;
1208 }
1209
1210 hres = jsdisp_propput_idx(array, i, &var, ei, sp);
1211 SysFreeString(V_BSTR(&var));
1212 if(FAILED(hres))
1213 break;
1214
1215 if(use_regexp)
1216 ptr = match_result[i].str + match_result[i].len;
1217 else if(match_str)
1218 ptr = ptr2 + match_len;
1219 else
1220 ptr++;
1221 }
1222 }
1223
1224 if(SUCCEEDED(hres) && (match_str || use_regexp)) {
1225 DWORD len = (str+length) - ptr;
1226
1227 if(len || match_str) {
1228 V_VT(&var) = VT_BSTR;
1229 V_BSTR(&var) = SysAllocStringLen(ptr, len);
1230
1231 if(V_BSTR(&var)) {
1232 hres = jsdisp_propput_idx(array, i, &var, ei, sp);
1233 SysFreeString(V_BSTR(&var));
1234 }else {
1235 hres = E_OUTOFMEMORY;
1236 }
1237 }
1238 }
1239
1240 SysFreeString(match_str);
1241 SysFreeString(val_str);
1242 heap_free(match_result);
1243
1244 if(SUCCEEDED(hres) && retv) {
1245 V_VT(retv) = VT_DISPATCH;
1246 V_DISPATCH(retv) = (IDispatch*)_IDispatchEx_(array);
1247 }else {
1248 jsdisp_release(array);
1249 }
1250
1251 return hres;
1252 }
1253
1254 static HRESULT String_strike(script_ctx_t *ctx, vdisp_t *jsthis, WORD flags, DISPPARAMS *dp,
1255 VARIANT *retv, jsexcept_t *ei, IServiceProvider *sp)
1256 {
1257 static const WCHAR striketagW[] = {'S','T','R','I','K','E',0};
1258 return do_attributeless_tag_format(ctx, jsthis, flags, dp, retv, ei, sp, striketagW);
1259 }
1260
1261 static HRESULT String_sub(script_ctx_t *ctx, vdisp_t *jsthis, WORD flags, DISPPARAMS *dp,
1262 VARIANT *retv, jsexcept_t *ei, IServiceProvider *sp)
1263 {
1264 static const WCHAR subtagW[] = {'S','U','B',0};
1265 return do_attributeless_tag_format(ctx, jsthis, flags, dp, retv, ei, sp, subtagW);
1266 }
1267
1268 /* ECMA-262 3rd Edition 15.5.4.15 */
1269 static HRESULT String_substring(script_ctx_t *ctx, vdisp_t *jsthis, WORD flags, DISPPARAMS *dp,
1270 VARIANT *retv, jsexcept_t *ei, IServiceProvider *sp)
1271 {
1272 const WCHAR *str;
1273 BSTR val_str;
1274 INT start=0, end;
1275 DWORD length;
1276 VARIANT v;
1277 HRESULT hres;
1278
1279 TRACE("\n");
1280
1281 hres = get_string_val(ctx, jsthis, ei, &str, &length, &val_str);
1282 if(FAILED(hres))
1283 return hres;
1284
1285 if(arg_cnt(dp) >= 1) {
1286 hres = to_integer(ctx, get_arg(dp,0), ei, &v);
1287 if(FAILED(hres)) {
1288 SysFreeString(val_str);
1289 return hres;
1290 }
1291
1292 if(V_VT(&v) == VT_I4) {
1293 start = V_I4(&v);
1294 if(start < 0)
1295 start = 0;
1296 else if(start >= length)
1297 start = length;
1298 }else {
1299 start = V_R8(&v) < 0.0 ? 0 : length;
1300 }
1301 }
1302
1303 if(arg_cnt(dp) >= 2) {
1304 hres = to_integer(ctx, get_arg(dp,1), ei, &v);
1305 if(FAILED(hres)) {
1306 SysFreeString(val_str);
1307 return hres;
1308 }
1309
1310 if(V_VT(&v) == VT_I4) {
1311 end = V_I4(&v);
1312 if(end < 0)
1313 end = 0;
1314 else if(end > length)
1315 end = length;
1316 }else {
1317 end = V_R8(&v) < 0.0 ? 0 : length;
1318 }
1319 }else {
1320 end = length;
1321 }
1322
1323 if(start > end) {
1324 INT tmp = start;
1325 start = end;
1326 end = tmp;
1327 }
1328
1329 if(retv) {
1330 V_VT(retv) = VT_BSTR;
1331 V_BSTR(retv) = SysAllocStringLen(str+start, end-start);
1332 if(!V_BSTR(retv)) {
1333 SysFreeString(val_str);
1334 return E_OUTOFMEMORY;
1335 }
1336 }
1337 SysFreeString(val_str);
1338 return S_OK;
1339 }
1340
1341 /* ECMA-262 3rd Edition B.2.3 */
1342 static HRESULT String_substr(script_ctx_t *ctx, vdisp_t *jsthis, WORD flags, DISPPARAMS *dp,
1343 VARIANT *retv, jsexcept_t *ei, IServiceProvider *sp)
1344 {
1345 BSTR val_str;
1346 const WCHAR *str;
1347 INT start=0, len;
1348 DWORD length;
1349 VARIANT v;
1350 HRESULT hres;
1351
1352 TRACE("\n");
1353
1354 hres = get_string_val(ctx, jsthis, ei, &str, &length, &val_str);
1355 if(FAILED(hres))
1356 return hres;
1357
1358 if(arg_cnt(dp) >= 1) {
1359 hres = to_integer(ctx, get_arg(dp,0), ei, &v);
1360 if(FAILED(hres)) {
1361 SysFreeString(val_str);
1362 return hres;
1363 }
1364
1365 if(V_VT(&v) == VT_I4) {
1366 start = V_I4(&v);
1367 if(start < 0)
1368 start = 0;
1369 else if(start >= length)
1370 start = length;
1371 }else {
1372 start = V_R8(&v) < 0.0 ? 0 : length;
1373 }
1374 }
1375
1376 if(arg_cnt(dp) >= 2) {
1377 hres = to_integer(ctx, get_arg(dp,1), ei, &v);
1378 if(FAILED(hres)) {
1379 SysFreeString(val_str);
1380 return hres;
1381 }
1382
1383 if(V_VT(&v) == VT_I4) {
1384 len = V_I4(&v);
1385 if(len < 0)
1386 len = 0;
1387 else if(len > length-start)
1388 len = length-start;
1389 }else {
1390 len = V_R8(&v) < 0.0 ? 0 : length-start;
1391 }
1392 }else {
1393 len = length-start;
1394 }
1395
1396 hres = S_OK;
1397 if(retv) {
1398 V_VT(retv) = VT_BSTR;
1399 V_BSTR(retv) = SysAllocStringLen(str+start, len);
1400 if(!V_BSTR(retv))
1401 hres = E_OUTOFMEMORY;
1402 }
1403
1404 SysFreeString(val_str);
1405 return hres;
1406 }
1407
1408 static HRESULT String_sup(script_ctx_t *ctx, vdisp_t *jsthis, WORD flags, DISPPARAMS *dp,
1409 VARIANT *retv, jsexcept_t *ei, IServiceProvider *sp)
1410 {
1411 static const WCHAR suptagW[] = {'S','U','P',0};
1412 return do_attributeless_tag_format(ctx, jsthis, flags, dp, retv, ei, sp, suptagW);
1413 }
1414
1415 static HRESULT String_toLowerCase(script_ctx_t *ctx, vdisp_t *jsthis, WORD flags, DISPPARAMS *dp,
1416 VARIANT *retv, jsexcept_t *ei, IServiceProvider *sp)
1417 {
1418 const WCHAR* str;
1419 DWORD length;
1420 BSTR val_str;
1421 HRESULT hres;
1422
1423 TRACE("\n");
1424
1425 hres = get_string_val(ctx, jsthis, ei, &str, &length, &val_str);
1426 if(FAILED(hres))
1427 return hres;
1428
1429 if(retv) {
1430 if(!val_str) {
1431 val_str = SysAllocStringLen(str, length);
1432 if(!val_str)
1433 return E_OUTOFMEMORY;
1434 }
1435
1436 strlwrW(val_str);
1437
1438 V_VT(retv) = VT_BSTR;
1439 V_BSTR(retv) = val_str;
1440 }
1441 else SysFreeString(val_str);
1442 return S_OK;
1443 }
1444
1445 static HRESULT String_toUpperCase(script_ctx_t *ctx, vdisp_t *jsthis, WORD flags, DISPPARAMS *dp,
1446 VARIANT *retv, jsexcept_t *ei, IServiceProvider *sp)
1447 {
1448 const WCHAR* str;
1449 DWORD length;
1450 BSTR val_str;
1451 HRESULT hres;
1452
1453 TRACE("\n");
1454
1455 hres = get_string_val(ctx, jsthis, ei, &str, &length, &val_str);
1456 if(FAILED(hres))
1457 return hres;
1458
1459 if(retv) {
1460 if(!val_str) {
1461 val_str = SysAllocStringLen(str, length);
1462 if(!val_str)
1463 return E_OUTOFMEMORY;
1464 }
1465
1466 struprW(val_str);
1467
1468 V_VT(retv) = VT_BSTR;
1469 V_BSTR(retv) = val_str;
1470 }
1471 else SysFreeString(val_str);
1472 return S_OK;
1473 }
1474
1475 static HRESULT String_toLocaleLowerCase(script_ctx_t *ctx, vdisp_t *jsthis, WORD flags, DISPPARAMS *dp,
1476 VARIANT *retv, jsexcept_t *ei, IServiceProvider *sp)
1477 {
1478 FIXME("\n");
1479 return E_NOTIMPL;
1480 }
1481
1482 static HRESULT String_toLocaleUpperCase(script_ctx_t *ctx, vdisp_t *jsthis, WORD flags, DISPPARAMS *dp,
1483 VARIANT *retv, jsexcept_t *ei, IServiceProvider *sp)
1484 {
1485 FIXME("\n");
1486 return E_NOTIMPL;
1487 }
1488
1489 static HRESULT String_localeCompare(script_ctx_t *ctx, vdisp_t *jsthis, WORD flags, DISPPARAMS *dp,
1490 VARIANT *retv, jsexcept_t *ei, IServiceProvider *sp)
1491 {
1492 FIXME("\n");
1493 return E_NOTIMPL;
1494 }
1495
1496 static HRESULT String_value(script_ctx_t *ctx, vdisp_t *jsthis, WORD flags, DISPPARAMS *dp,
1497 VARIANT *retv, jsexcept_t *ei, IServiceProvider *sp)
1498 {
1499 StringInstance *This = string_from_vdisp(jsthis);
1500
1501 TRACE("\n");
1502
1503 switch(flags) {
1504 case INVOKE_FUNC:
1505 return throw_type_error(ctx, ei, IDS_NOT_FUNC, NULL);
1506 case DISPATCH_PROPERTYGET: {
1507 BSTR str = SysAllocString(This->str);
1508 if(!str)
1509 return E_OUTOFMEMORY;
1510
1511 V_VT(retv) = VT_BSTR;
1512 V_BSTR(retv) = str;
1513 break;
1514 }
1515 default:
1516 FIXME("flags %x\n", flags);
1517 return E_NOTIMPL;
1518 }
1519
1520 return S_OK;
1521 }
1522
1523 static void String_destructor(DispatchEx *dispex)
1524 {
1525 StringInstance *This = (StringInstance*)dispex;
1526
1527 heap_free(This->str);
1528 heap_free(This);
1529 }
1530
1531 static const builtin_prop_t String_props[] = {
1532 {anchorW, String_anchor, PROPF_METHOD|1},
1533 {bigW, String_big, PROPF_METHOD},
1534 {blinkW, String_blink, PROPF_METHOD},
1535 {boldW, String_bold, PROPF_METHOD},
1536 {charAtW, String_charAt, PROPF_METHOD|1},
1537 {charCodeAtW, String_charCodeAt, PROPF_METHOD|1},
1538 {concatW, String_concat, PROPF_METHOD|1},
1539 {fixedW, String_fixed, PROPF_METHOD},
1540 {fontcolorW, String_fontcolor, PROPF_METHOD|1},
1541 {fontsizeW, String_fontsize, PROPF_METHOD|1},
1542 {indexOfW, String_indexOf, PROPF_METHOD|2},
1543 {italicsW, String_italics, PROPF_METHOD},
1544 {lastIndexOfW, String_lastIndexOf, PROPF_METHOD|2},
1545 {lengthW, String_length, 0},
1546 {linkW, String_link, PROPF_METHOD|1},
1547 {localeCompareW, String_localeCompare, PROPF_METHOD|1},
1548 {matchW, String_match, PROPF_METHOD|1},
1549 {replaceW, String_replace, PROPF_METHOD|1},
1550 {searchW, String_search, PROPF_METHOD},
1551 {sliceW, String_slice, PROPF_METHOD},
1552 {smallW, String_small, PROPF_METHOD},
1553 {splitW, String_split, PROPF_METHOD|2},
1554 {strikeW, String_strike, PROPF_METHOD},
1555 {subW, String_sub, PROPF_METHOD},
1556 {substrW, String_substr, PROPF_METHOD|2},
1557 {substringW, String_substring, PROPF_METHOD|2},
1558 {supW, String_sup, PROPF_METHOD},
1559 {toLocaleLowerCaseW, String_toLocaleLowerCase, PROPF_METHOD},
1560 {toLocaleUpperCaseW, String_toLocaleUpperCase, PROPF_METHOD},
1561 {toLowerCaseW, String_toLowerCase, PROPF_METHOD},
1562 {toStringW, String_toString, PROPF_METHOD},
1563 {toUpperCaseW, String_toUpperCase, PROPF_METHOD},
1564 {valueOfW, String_valueOf, PROPF_METHOD}
1565 };
1566
1567 static const builtin_info_t String_info = {
1568 JSCLASS_STRING,
1569 {NULL, String_value, 0},
1570 sizeof(String_props)/sizeof(*String_props),
1571 String_props,
1572 String_destructor,
1573 NULL
1574 };
1575
1576 /* ECMA-262 3rd Edition 15.5.3.2 */
1577 static HRESULT StringConstr_fromCharCode(script_ctx_t *ctx, vdisp_t *jsthis, WORD flags,
1578 DISPPARAMS *dp, VARIANT *retv, jsexcept_t *ei, IServiceProvider *sp)
1579 {
1580 DWORD i, code;
1581 BSTR ret;
1582 HRESULT hres;
1583
1584 ret = SysAllocStringLen(NULL, arg_cnt(dp));
1585 if(!ret)
1586 return E_OUTOFMEMORY;
1587
1588 for(i=0; i<arg_cnt(dp); i++) {
1589 hres = to_uint32(ctx, get_arg(dp, i), ei, &code);
1590 if(FAILED(hres)) {
1591 SysFreeString(ret);
1592 return hres;
1593 }
1594
1595 ret[i] = code;
1596 }
1597
1598 if(retv) {
1599 V_VT(retv) = VT_BSTR;
1600 V_BSTR(retv) = ret;
1601 }
1602 else SysFreeString(ret);
1603
1604 return S_OK;
1605 }
1606
1607 static HRESULT StringConstr_value(script_ctx_t *ctx, vdisp_t *jsthis, WORD flags, DISPPARAMS *dp,
1608 VARIANT *retv, jsexcept_t *ei, IServiceProvider *sp)
1609 {
1610 HRESULT hres;
1611
1612 TRACE("\n");
1613
1614 switch(flags) {
1615 case INVOKE_FUNC: {
1616 BSTR str;
1617
1618 if(arg_cnt(dp)) {
1619 hres = to_string(ctx, get_arg(dp, 0), ei, &str);
1620 if(FAILED(hres))
1621 return hres;
1622 }else {
1623 str = SysAllocStringLen(NULL, 0);
1624 if(!str)
1625 return E_OUTOFMEMORY;
1626 }
1627
1628 V_VT(retv) = VT_BSTR;
1629 V_BSTR(retv) = str;
1630 break;
1631 }
1632 case DISPATCH_CONSTRUCT: {
1633 DispatchEx *ret;
1634
1635 if(arg_cnt(dp)) {
1636 BSTR str;
1637
1638 hres = to_string(ctx, get_arg(dp, 0), ei, &str);
1639 if(FAILED(hres))
1640 return hres;
1641
1642 hres = create_string(ctx, str, SysStringLen(str), &ret);
1643 SysFreeString(str);
1644 }else {
1645 hres = create_string(ctx, NULL, 0, &ret);
1646 }
1647
1648 if(FAILED(hres))
1649 return hres;
1650
1651 V_VT(retv) = VT_DISPATCH;
1652 V_DISPATCH(retv) = (IDispatch*)_IDispatchEx_(ret);
1653 break;
1654 }
1655
1656 default:
1657 FIXME("unimplemented flags: %x\n", flags);
1658 return E_NOTIMPL;
1659 }
1660
1661 return S_OK;
1662 }
1663
1664 static HRESULT string_alloc(script_ctx_t *ctx, DispatchEx *object_prototype, StringInstance **ret)
1665 {
1666 StringInstance *string;
1667 HRESULT hres;
1668
1669 string = heap_alloc_zero(sizeof(StringInstance));
1670 if(!string)
1671 return E_OUTOFMEMORY;
1672
1673 if(object_prototype)
1674 hres = init_dispex(&string->dispex, ctx, &String_info, object_prototype);
1675 else
1676 hres = init_dispex_from_constr(&string->dispex, ctx, &String_info, ctx->string_constr);
1677 if(FAILED(hres)) {
1678 heap_free(string);
1679 return hres;
1680 }
1681
1682 *ret = string;
1683 return S_OK;
1684 }
1685
1686 static const builtin_prop_t StringConstr_props[] = {
1687 {fromCharCodeW, StringConstr_fromCharCode, PROPF_METHOD},
1688 };
1689
1690 static const builtin_info_t StringConstr_info = {
1691 JSCLASS_FUNCTION,
1692 {NULL, Function_value, 0},
1693 sizeof(StringConstr_props)/sizeof(*StringConstr_props),
1694 StringConstr_props,
1695 NULL,
1696 NULL
1697 };
1698
1699 HRESULT create_string_constr(script_ctx_t *ctx, DispatchEx *object_prototype, DispatchEx **ret)
1700 {
1701 StringInstance *string;
1702 HRESULT hres;
1703
1704 static const WCHAR StringW[] = {'S','t','r','i','n','g',0};
1705
1706 hres = string_alloc(ctx, object_prototype, &string);
1707 if(FAILED(hres))
1708 return hres;
1709
1710 hres = create_builtin_function(ctx, StringConstr_value, StringW, &StringConstr_info,
1711 PROPF_CONSTR|1, &string->dispex, ret);
1712
1713 jsdisp_release(&string->dispex);
1714 return hres;
1715 }
1716
1717 HRESULT create_string(script_ctx_t *ctx, const WCHAR *str, DWORD len, DispatchEx **ret)
1718 {
1719 StringInstance *string;
1720 HRESULT hres;
1721
1722 hres = string_alloc(ctx, NULL, &string);
1723 if(FAILED(hres))
1724 return hres;
1725
1726 if(len == -1)
1727 len = strlenW(str);
1728
1729 string->length = len;
1730 string->str = heap_alloc((len+1)*sizeof(WCHAR));
1731 if(!string->str) {
1732 jsdisp_release(&string->dispex);
1733 return E_OUTOFMEMORY;
1734 }
1735
1736 memcpy(string->str, str, len*sizeof(WCHAR));
1737 string->str[len] = 0;
1738
1739 *ret = &string->dispex;
1740 return S_OK;
1741
1742 }
GSOC 2012 project to enhance the control panel