msvcp90: Added basic_string<char>::erase implementation.
[wine] / dlls / msvcp90 / string.c
1 /*
2  * Copyright 2010 Piotr 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 "config.h"
20
21 #include <stdarg.h>
22
23 #include "msvcp90.h"
24 #include "stdio.h"
25
26 #include "windef.h"
27 #include "winbase.h"
28 #include "wine/debug.h"
29 WINE_DEFAULT_DEBUG_CHANNEL(msvcp90);
30
31 /* char_traits<char> */
32 /* ?assign@?$char_traits@D@std@@SAXAADABD@Z */
33 /* ?assign@?$char_traits@D@std@@SAXAEADAEBD@Z */
34 void CDECL MSVCP_char_traits_char_assign(char *ch, const char *assign)
35 {
36     *ch = *assign;
37 }
38
39 /* ?eq@?$char_traits@D@std@@SA_NABD0@Z */
40 /* ?eq@?$char_traits@D@std@@SA_NAEBD0@Z */
41 MSVCP_BOOL CDECL MSVCP_char_traits_char_eq(const char *ch1, const char *ch2)
42 {
43     return *ch1 == *ch2;
44 }
45
46 /* ?lt@?$char_traits@D@std@@SA_NABD0@Z */
47 /* ?lt@?$char_traits@D@std@@SA_NAEBD0@Z */
48 MSVCP_BOOL CDECL MSVCP_char_traits_lt(const char *ch1, const char *ch2)
49 {
50     return *ch1 < *ch2;
51 }
52
53 /* ?compare@?$char_traits@D@std@@SAHPBD0I@Z */
54 /* ?compare@?$char_traits@D@std@@SAHPEBD0_K@Z */
55 int CDECL MSVCP_char_traits_char_compare(
56         const char *s1, const char *s2, size_t count)
57 {
58     int ret = memcmp(s1, s2, count);
59     return (ret>0 ? 1 : (ret<0 ? -1 : 0));
60 }
61
62 /* ?length@?$char_traits@D@std@@SAIPBD@Z */
63 /* ?length@?$char_traits@D@std@@SA_KPEBD@Z */
64 size_t CDECL MSVCP_char_traits_char_length(const char *str)
65 {
66     return strlen(str);
67 }
68
69 /* ?_Copy_s@?$char_traits@D@std@@SAPADPADIPBDI@Z */
70 /* ?_Copy_s@?$char_traits@D@std@@SAPEADPEAD_KPEBD1@Z */
71 char* CDECL MSVCP_char_traits_char__Copy_s(char *dest,
72         size_t size, const char *src, size_t count)
73 {
74     if(!dest || !src || size<count) {
75         if(dest && size)
76             dest[0] = '\0';
77         _invalid_parameter(NULL, NULL, NULL, 0, 0);
78         return dest;
79     }
80
81     return memcpy(dest, src, count);
82 }
83
84 /* ?copy@?$char_traits@D@std@@SAPADPADPBDI@Z */
85 /* ?copy@?$char_traits@D@std@@SAPEADPEADPEBD_K@Z */
86 char* CDECL MSVCP_char_traits_char_copy(
87         char *dest, const char *src, size_t count)
88 {
89     return MSVCP_char_traits_char__Copy_s(dest, count, src, count);
90 }
91
92 /* ?find@?$char_traits@D@std@@SAPBDPBDIABD@Z */
93 /* ?find@?$char_traits@D@std@@SAPEBDPEBD_KAEBD@Z */
94 const char * CDECL MSVCP_char_traits_char_find(
95         const char *str, size_t range, const char *c)
96 {
97     return memchr(str, *c, range);
98 }
99
100 /* ?_Move_s@?$char_traits@D@std@@SAPADPADIPBDI@Z */
101 /* ?_Move_s@?$char_traits@D@std@@SAPEADPEAD_KPEBD1@Z */
102 char* CDECL MSVCP_char_traits_char__Move_s(char *dest,
103         size_t size, const char *src, size_t count)
104 {
105     if(!dest || !src || size<count) {
106         if(dest && size)
107             dest[0] = '\0';
108         _invalid_parameter(NULL, NULL, NULL, 0, 0);
109         return dest;
110     }
111
112     return memmove(dest, src, count);
113 }
114
115 /* ?move@?$char_traits@D@std@@SAPADPADPBDI@Z */
116 /* ?move@?$char_traits@D@std@@SAPEADPEADPEBD_K@Z */
117 char* CDECL MSVCP_char_traits_char_move(
118         char *dest, const char *src, size_t count)
119 {
120     return MSVCP_char_traits_char__Move_s(dest, count, src, count);
121 }
122
123 /* ?assign@?$char_traits@D@std@@SAPADPADID@Z */
124 /* ?assign@?$char_traits@D@std@@SAPEADPEAD_KD@Z */
125 char* CDECL MSVCP_char_traits_char_assignn(char *str, size_t num, char c)
126 {
127     return memset(str, c, num);
128 }
129
130 /* ?to_char_type@?$char_traits@D@std@@SADABH@Z */
131 /* ?to_char_type@?$char_traits@D@std@@SADAEBH@Z */
132 char CDECL MSVCP_char_traits_char_to_char_type(const int *i)
133 {
134     return (char)*i;
135 }
136
137 /* ?to_int_type@?$char_traits@D@std@@SAHABD@Z */
138 /* ?to_int_type@?$char_traits@D@std@@SAHAEBD@Z */
139 int CDECL MSVCP_char_traits_char_to_int_type(const char *ch)
140 {
141     return (int)*ch;
142 }
143
144 /* ?eq_int_type@?$char_traits@D@std@@SA_NABH0@Z */
145 /* ?eq_int_type@?$char_traits@D@std@@SA_NAEBH0@Z */
146 MSVCP_BOOL CDECL MSVCP_char_traits_char_eq_int_type(const int *i1, const int *i2)
147 {
148     return *i1 == *i2;
149 }
150
151 /* ?eof@?$char_traits@D@std@@SAHXZ */
152 int CDECL MSVCP_char_traits_char_eof(void)
153 {
154     return EOF;
155 }
156
157 /* ?not_eof@?$char_traits@D@std@@SAHABH@Z */
158 /* ?not_eof@?$char_traits@D@std@@SAHAEBH@Z */
159 int CDECL MSVCP_char_traits_char_not_eof(int *in)
160 {
161     return (*in==EOF ? !EOF : *in);
162 }
163
164
165 /* char_traits<wchar_t> */
166 /* ?assign@?$char_traits@_W@std@@SAXAA_WAB_W@Z */
167 /* ?assign@?$char_traits@_W@std@@SAXAEA_WAEB_W@Z */
168 void CDECL MSVCP_char_traits_wchar_assign(wchar_t *ch,
169         const wchar_t *assign)
170 {
171     *ch = *assign;
172 }
173
174 /* ?eq@?$char_traits@_W@std@@SA_NAB_W0@Z */
175 /* ?eq@?$char_traits@_W@std@@SA_NAEB_W0@Z */
176 MSVCP_BOOL CDECL MSVCP_char_traits_wchar_eq(wchar_t *ch1, wchar_t *ch2)
177 {
178     return *ch1 == *ch2;
179 }
180
181 /* ?lt@?$char_traits@_W@std@@SA_NAB_W0@Z */
182 /* ?lt@?$char_traits@_W@std@@SA_NAEB_W0@Z */
183 MSVCP_BOOL CDECL MSVCP_char_traits_wchar_lt(const wchar_t *ch1,
184         const wchar_t *ch2)
185 {
186     return *ch1 < *ch2;
187 }
188
189 /* ?compare@?$char_traits@_W@std@@SAHPB_W0I@Z */
190 /* ?compare@?$char_traits@_W@std@@SAHPEB_W0_K@Z */
191 int CDECL MSVCP_char_traits_wchar_compare(const wchar_t *s1,
192         const wchar_t *s2, size_t count)
193 {
194     int ret = memcmp(s1, s2, sizeof(wchar_t[count]));
195     return (ret>0 ? 1 : (ret<0 ? -1 : 0));
196 }
197
198 /* ?length@?$char_traits@_W@std@@SAIPB_W@Z */
199 /* ?length@?$char_traits@_W@std@@SA_KPEB_W@Z */
200 size_t CDECL MSVCP_char_traits_wchar_length(const wchar_t *str)
201 {
202     return wcslen((WCHAR*)str);
203 }
204
205 /* ?_Copy_s@?$char_traits@_W@std@@SAPA_WPA_WIPB_WI@Z */
206 /* ?_Copy_s@?$char_traits@_W@std@@SAPEA_WPEA_W_KPEB_W1@Z */
207 wchar_t* CDECL MSVCP_char_traits_wchar__Copy_s(wchar_t *dest,
208         size_t size, const wchar_t *src, size_t count)
209 {
210     if(!dest || !src || size<count) {
211         if(dest && size)
212             dest[0] = '\0';
213         _invalid_parameter(NULL, NULL, NULL, 0, 0);
214         return dest;
215     }
216
217     return memcpy(dest, src, sizeof(wchar_t[count]));
218 }
219
220 /* ?copy@?$char_traits@_W@std@@SAPA_WPA_WPB_WI@Z */
221 /* ?copy@?$char_traits@_W@std@@SAPEA_WPEA_WPEB_W_K@Z */
222 wchar_t* CDECL MSVCP_char_traits_wchar_copy(wchar_t *dest,
223         const wchar_t *src, size_t count)
224 {
225     return MSVCP_char_traits_wchar__Copy_s(dest, count, src, count);
226 }
227
228 /* ?find@?$char_traits@_W@std@@SAPB_WPB_WIAB_W@Z */
229 /* ?find@?$char_traits@_W@std@@SAPEB_WPEB_W_KAEB_W@Z */
230 const wchar_t* CDECL MSVCP_char_traits_wchar_find(
231         const wchar_t *str, size_t range, const wchar_t *c)
232 {
233     size_t i=0;
234
235     for(i=0; i<range; i++)
236         if(str[i] == *c)
237             return str+i;
238
239     return NULL;
240 }
241
242 /* ?_Move_s@?$char_traits@_W@std@@SAPA_WPA_WIPB_WI@Z */
243 /* ?_Move_s@?$char_traits@_W@std@@SAPEA_WPEA_W_KPEB_W1@Z */
244 wchar_t* CDECL MSVCP_char_traits_wchar__Move_s(wchar_t *dest,
245         size_t size, const wchar_t *src, size_t count)
246 {
247     if(!dest || !src || size<count) {
248         if(dest && size)
249             dest[0] = '\0';
250         _invalid_parameter(NULL, NULL, NULL, 0, 0);
251         return dest;
252     }
253
254     return memmove(dest, src, sizeof(WCHAR[count]));
255 }
256
257 /* ?move@?$char_traits@_W@std@@SAPA_WPA_WPB_WI@Z */
258 /* ?move@?$char_traits@_W@std@@SAPEA_WPEA_WPEB_W_K@Z */
259 wchar_t* CDECL MSVCP_char_traits_wchar_move(wchar_t *dest,
260         const wchar_t *src, size_t count)
261 {
262     return MSVCP_char_traits_wchar__Move_s(dest, count, src, count);
263 }
264
265 /* ?assign@?$char_traits@_W@std@@SAPA_WPA_WI_W@Z */
266 /* ?assign@?$char_traits@_W@std@@SAPEA_WPEA_W_K_W@Z */
267 wchar_t* CDECL MSVCP_char_traits_wchar_assignn(wchar_t *str,
268         size_t num, wchar_t c)
269 {
270     size_t i;
271
272     for(i=0; i<num; i++)
273         str[i] = c;
274
275     return str;
276 }
277
278 /* ?to_char_type@?$char_traits@_W@std@@SA_WABG@Z */
279 /* ?to_char_type@?$char_traits@_W@std@@SA_WAEBG@Z */
280 wchar_t CDECL MSVCP_char_traits_wchar_to_char_type(const unsigned short *i)
281 {
282     return *i;
283 }
284
285 /* ?to_int_type@?$char_traits@_W@std@@SAGAB_W@Z */
286 /* ?to_int_type@?$char_traits@_W@std@@SAGAEB_W@Z */
287 unsigned short CDECL MSVCP_char_traits_wchar_to_int_type(const wchar_t *ch)
288 {
289     return *ch;
290 }
291
292 /* ?eq_int_type@?$char_traits@_W@std@@SA_NABG0@Z */
293 /* ?eq_int_type@?$char_traits@_W@std@@SA_NAEBG0@Z */
294 MSVCP_BOOL CDECL MSVCP_char_traits_wchar_eq_int_tpe(const unsigned short *i1,
295         const unsigned short *i2)
296 {
297     return *i1 == *i2;
298 }
299
300 /* ?eof@?$char_traits@_W@std@@SAGXZ */
301 unsigned short CDECL MSVCP_char_traits_wchar_eof(void)
302 {
303     return WEOF;
304 }
305
306 /* ?not_eof@?$char_traits@_W@std@@SAGABG@Z */
307 /* ?not_eof@?$char_traits@_W@std@@SAGAEBG@Z */
308 unsigned short CDECL MSVCP_char_traits_wchar_not_eof(const unsigned short *in)
309 {
310     return (*in==WEOF ? !WEOF : *in);
311 }
312
313
314 /* char_traits<unsigned short> */
315 /* ?assign@?$char_traits@G@std@@SAXAAGABG@Z */
316 /* ?assign@?$char_traits@G@std@@SAXAEAGAEBG@Z */
317 void CDECL MSVCP_char_traits_short_assign(unsigned short *ch,
318         const unsigned short *assign)
319 {
320     *ch = *assign;
321 }
322
323 /* ?eq@?$char_traits@G@std@@SA_NABG0@Z */
324 /* ?eq@?$char_traits@G@std@@SA_NAEBG0@Z */
325 MSVCP_BOOL CDECL MSVCP_char_traits_short_eq(const unsigned short *ch1,
326         const unsigned short *ch2)
327 {
328     return *ch1 == *ch2;
329 }
330
331 /* ?lt@?$char_traits@G@std@@SA_NABG0@Z */
332 /* ?lt@?$char_traits@G@std@@SA_NAEBG0@Z */
333 MSVCP_BOOL CDECL MSVCP_char_traits_short_lt(const unsigned short *ch1,
334         const unsigned short *ch2)
335 {
336     return *ch1 < *ch2;
337 }
338
339 /* ?compare@?$char_traits@G@std@@SAHPBG0I@Z */
340 /* ?compare@?$char_traits@G@std@@SAHPEBG0_K@Z */
341 int CDECL MSVCP_char_traits_short_compare(const unsigned short *s1,
342         const unsigned short *s2, size_t count)
343 {
344     size_t i;
345
346     for(i=0; i<count; i++)
347         if(s1[i] != s2[i])
348             return (s1[i] < s2[i] ? -1 : 1);
349
350     return 0;
351 }
352
353 /* ?length@?$char_traits@G@std@@SAIPBG@Z */
354 /* ?length@?$char_traits@G@std@@SA_KPEBG@Z */
355 size_t CDECL MSVCP_char_traits_short_length(const unsigned short *str)
356 {
357     size_t len;
358
359     for(len=0; str[len]; len++);
360
361     return len;
362 }
363
364 /* ?_Copy_s@?$char_traits@G@std@@SAPAGPAGIPBGI@Z */
365 /* ?_Copy_s@?$char_traits@G@std@@SAPEAGPEAG_KPEBG1@Z */
366 unsigned short * CDECL MSVCP_char_traits_short__Copy_s(unsigned short *dest,
367         size_t size, const unsigned short *src, size_t count)
368 {
369     if(size<count) {
370         _invalid_parameter(NULL, NULL, NULL, 0, 0);
371         return dest;
372     }
373
374     return memcpy(dest, src, sizeof(unsigned short[count]));
375 }
376
377 /* ?copy@?$char_traits@G@std@@SAPAGPAGPBGI@Z */
378 /* ?copy@?$char_traits@G@std@@SAPEAGPEAGPEBG_K@Z */
379 unsigned short* CDECL MSVCP_char_traits_short_copy(unsigned short *dest,
380         const unsigned short *src, size_t count)
381 {
382     return MSVCP_char_traits_short__Copy_s(dest, count, src, count);
383 }
384
385 /* ?find@?$char_traits@G@std@@SAPBGPBGIABG@Z */
386 /* ?find@?$char_traits@G@std@@SAPEBGPEBG_KAEBG@Z */
387 const unsigned short* CDECL MSVCP_char_traits_short_find(
388         const unsigned short *str, size_t range, const unsigned short *c)
389 {
390     size_t i;
391
392     for(i=0; i<range; i++)
393         if(str[i] == *c)
394             return str+i;
395
396     return NULL;
397 }
398
399 /* ?_Move_s@?$char_traits@G@std@@SAPAGPAGIPBGI@Z */
400 /* ?_Move_s@?$char_traits@G@std@@SAPEAGPEAG_KPEBG1@Z */
401 unsigned short* CDECL MSVCP_char_traits_short__Move_s(unsigned short *dest,
402         size_t size, const unsigned short *src, size_t count)
403 {
404     if(size<count) {
405         _invalid_parameter(NULL, NULL, NULL, 0, 0);
406         return dest;
407     }
408
409     return memmove(dest, src, sizeof(unsigned short[count]));
410 }
411
412 /* ?move@?$char_traits@G@std@@SAPAGPAGPBGI@Z */
413 /* ?move@?$char_traits@G@std@@SAPEAGPEAGPEBG_K@Z */
414 unsigned short* CDECL MSVCP_char_traits_short_move(unsigned short *dest,
415         const unsigned short *src, size_t count)
416 {
417     return MSVCP_char_traits_short__Move_s(dest, count, src, count);
418 }
419
420 /* ?assign@?$char_traits@G@std@@SAPAGPAGIG@Z */
421 /* ?assign@?$char_traits@G@std@@SAPEAGPEAG_KG@Z */
422 unsigned short* CDECL MSVCP_char_traits_short_assignn(unsigned short *str,
423         size_t num, unsigned short c)
424 {
425     size_t i;
426
427     for(i=0; i<num; i++)
428         str[i] = c;
429
430     return str;
431 }
432
433 /* ?to_char_type@?$char_traits@G@std@@SAGABG@Z */
434 /* ?to_char_type@?$char_traits@G@std@@SAGAEBG@Z */
435 unsigned short CDECL MSVCP_char_traits_short_to_char_type(const unsigned short *i)
436 {
437     return *i;
438 }
439
440 /* ?to_int_type@?$char_traits@G@std@@SAGABG@Z */
441 /* ?to_int_type@?$char_traits@G@std@@SAGAEBG@Z */
442 unsigned short CDECL MSVCP_char_traits_short_to_int_type(const unsigned short *ch)
443 {
444     return *ch;
445 }
446
447 /* ?eq_int_type@?$char_traits@G@std@@SA_NABG0@Z */
448 /* ?eq_int_type@?$char_traits@G@std@@SA_NAEBG0@Z */
449 MSVCP_BOOL CDECL MSVCP_char_traits_short_eq_int_type(unsigned short *i1,
450         unsigned short *i2)
451 {
452     return *i1 == *i2;
453 }
454
455 /* ?eof@?$char_traits@G@std@@SAGXZ */
456 unsigned short CDECL MSVCP_char_traits_short_eof(void)
457 {
458     return -1;
459 }
460
461 /* ?not_eof@?$char_traits@G@std@@SAGABG@Z */
462 /* ?not_eof@?$char_traits@G@std@@SAGAEBG@Z */
463 unsigned short CDECL MSVCP_char_traits_short_not_eof(const unsigned short *in)
464 {
465     return (*in==(unsigned short)-1 ? 0 : *in);
466 }
467
468
469 /* basic_string<char, char_traits<char>, allocator<char>> */
470 /* ?npos@?$basic_string@DU?$char_traits@D@std@@V?$allocator@D@2@@std@@2IB */
471 const size_t MSVCP_basic_string_char_npos = -1;
472
473 /* Internal: basic_string_char_ptr - return pointer to stored string */
474 static char* basic_string_char_ptr(basic_string_char *this)
475 {
476     if(this->res == BUF_SIZE_CHAR-1)
477         return this->data.buf;
478     return this->data.ptr;
479 }
480
481 /* Internal: basic_string_char_eos - sets string length, puts '\0' on the end */
482 static void basic_string_char_eos(basic_string_char *this, size_t len)
483 {
484     static const char nullbyte = '\0';
485
486     this->size = len;
487     MSVCP_char_traits_char_assign(basic_string_char_ptr(this)+len, &nullbyte);
488 }
489
490 /* Internal: basic_string_char_tidy - initialize basic_string buffer, deallocates data */
491 /* Caution: new_size have to be smaller than BUF_SIZE_CHAR */
492 static void basic_string_char_tidy(basic_string_char *this,
493         MSVCP_BOOL built, int new_size)
494 {
495     if(built && BUF_SIZE_CHAR<=this->res) {
496         char *ptr = this->data.ptr;
497
498         if(new_size > 0)
499             MSVCP_char_traits_char__Copy_s(this->data.buf, BUF_SIZE_CHAR, ptr, new_size);
500         MSVCP_allocator_char_deallocate(this->allocator, ptr, this->res+1);
501     }
502
503     this->res = BUF_SIZE_CHAR-1;
504     basic_string_char_eos(this, new_size);
505 }
506
507 /* ?erase@?$basic_string@DU?$char_traits@D@std@@V?$allocator@D@2@@std@@QAEAAV12@II@Z */
508 /* ?erase@?$basic_string@DU?$char_traits@D@std@@V?$allocator@D@2@@std@@QEAAAEAV12@_K0@Z */
509 DEFINE_THISCALL_WRAPPER(MSVCP_basic_string_char_erase, 12)
510 basic_string_char* __stdcall MSVCP_basic_string_char_erase(
511         basic_string_char *this, size_t pos, size_t len)
512 {
513     TRACE("%p %d %d\n", this, pos, len);
514
515     if(pos > this->size) {
516         FIXME("Throw exception (_Xran)\n");
517         return NULL;
518     }
519
520     if(len > this->size-pos)
521         len = this->size-pos;
522
523     if(len) {
524         MSVCP_char_traits_char__Move_s(basic_string_char_ptr(this)+pos,
525                 this->res-pos, basic_string_char_ptr(this)+pos+len,
526                 this->size-pos-len);
527         basic_string_char_eos(this, this->size-len);
528     }
529
530     return this;
531 }
532
533 /* ??0?$basic_string@DU?$char_traits@D@std@@V?$allocator@D@2@@std@@QAE@XZ */
534 /* ??0?$basic_string@DU?$char_traits@D@std@@V?$allocator@D@2@@std@@QEAA@XZ */
535 DEFINE_THISCALL_WRAPPER(MSVCP_basic_string_char_ctor, 4)
536 basic_string_char* __stdcall MSVCP_basic_string_char_ctor(basic_string_char *this)
537 {
538     TRACE("%p\n", this);
539
540     basic_string_char_tidy(this, FALSE, 0);
541     return this;
542 }