Updated.
[wine] / dlls / ntdll / wcstring.c
1 /*
2  * NTDLL wide-char functions
3  *
4  * Copyright 2000 Alexandre Julliard
5  * Copyright 2000 Jon Griffiths
6  */
7
8 #include "config.h"
9
10 #include <ctype.h>
11 #include <limits.h>
12 #include <stdlib.h>
13 #include <string.h>
14 #include <stdio.h>
15
16 #include "windef.h"
17 #include "winbase.h"
18 #include "winnls.h"
19 #include "wine/unicode.h"
20 #include "heap.h"
21 #include "debugtools.h"
22
23 DEFAULT_DEBUG_CHANNEL(ntdll);
24
25
26 /*********************************************************************
27  *           NTDLL__wcsicmp    (NTDLL)
28  */
29 INT __cdecl NTDLL__wcsicmp( LPCWSTR str1, LPCWSTR str2 )
30 {
31     return strcmpiW( str1, str2 );
32 }
33
34
35 /*********************************************************************
36  *           NTDLL__wcslwr    (NTDLL)
37  */
38 LPWSTR __cdecl NTDLL__wcslwr( LPWSTR str )
39 {
40     return strlwrW( str );
41 }
42
43
44 /*********************************************************************
45  *           NTDLL__wcsnicmp    (NTDLL)
46  */
47 INT __cdecl NTDLL__wcsnicmp( LPCWSTR str1, LPCWSTR str2, INT n )
48 {
49     return strncmpiW( str1, str2, n );
50 }
51
52
53 /*********************************************************************
54  *           NTDLL__wcsupr    (NTDLL)
55  */
56 LPWSTR __cdecl NTDLL__wcsupr( LPWSTR str )
57 {
58     return struprW( str );
59 }
60
61
62 /*********************************************************************
63  *           NTDLL_towlower    (NTDLL)
64  */
65 WCHAR __cdecl NTDLL_towlower( WCHAR ch )
66 {
67     return tolowerW(ch);
68 }
69
70
71 /*********************************************************************
72  *           NTDLL_towupper    (NTDLL)
73  */
74 WCHAR __cdecl NTDLL_towupper( WCHAR ch )
75 {
76     return toupperW(ch);
77 }
78
79
80 /***********************************************************************
81  *           NTDLL_wcscat    (NTDLL)
82  */
83 LPWSTR __cdecl NTDLL_wcscat( LPWSTR dst, LPCWSTR src )
84 {
85     return strcatW( dst, src );
86 }
87
88
89 /*********************************************************************
90  *           NTDLL_wcschr    (NTDLL)
91  */
92 LPWSTR __cdecl NTDLL_wcschr( LPCWSTR str, WCHAR ch )
93 {
94     return strchrW( str, ch );
95 }
96
97
98 /*********************************************************************
99  *           NTDLL_wcscmp    (NTDLL)
100  */
101 INT __cdecl NTDLL_wcscmp( LPCWSTR str1, LPCWSTR str2 )
102 {
103     return strcmpW( str1, str2 );
104 }
105
106
107 /***********************************************************************
108  *           NTDLL_wcscpy    (NTDLL)
109  */
110 LPWSTR __cdecl NTDLL_wcscpy( LPWSTR dst, LPCWSTR src )
111 {
112     return strcpyW( dst, src );
113 }
114
115
116 /*********************************************************************
117  *           NTDLL_wcscspn    (NTDLL)
118  */
119 INT __cdecl NTDLL_wcscspn( LPCWSTR str, LPCWSTR reject )
120 {
121     LPCWSTR start = str;
122     while (*str)
123     {
124         LPCWSTR p = reject;
125         while (*p && (*p != *str)) p++;
126         if (*p) break;
127         str++;
128     }
129     return str - start;
130 }
131
132
133 /***********************************************************************
134  *           NTDLL_wcslen    (NTDLL)
135  */
136 INT __cdecl NTDLL_wcslen( LPCWSTR str )
137 {
138     return strlenW( str );
139 }
140
141
142 /*********************************************************************
143  *           NTDLL_wcsncat    (NTDLL)
144  */
145 LPWSTR __cdecl NTDLL_wcsncat( LPWSTR s1, LPCWSTR s2, INT n )
146 {
147     LPWSTR ret = s1;
148     while (*s1) s1++;
149     while (n-- > 0) if (!(*s1++ = *s2++)) return ret;
150     *s1 = 0;
151     return ret;
152 }
153
154
155 /*********************************************************************
156  *           NTDLL_wcsncmp    (NTDLL)
157  */
158 INT __cdecl NTDLL_wcsncmp( LPCWSTR str1, LPCWSTR str2, INT n )
159 {
160     return strncmpW( str1, str2, n );
161 }
162
163
164 /*********************************************************************
165  *           NTDLL_wcsncpy    (NTDLL)
166  */
167 LPWSTR __cdecl NTDLL_wcsncpy( LPWSTR s1, LPCWSTR s2, INT n )
168 {
169     return strncpyW( s1, s2, n );
170 }
171
172
173 /*********************************************************************
174  *           NTDLL_wcspbrk    (NTDLL)
175  */
176 LPWSTR __cdecl NTDLL_wcspbrk( LPCWSTR str, LPCWSTR accept )
177 {
178     LPCWSTR p;
179     while (*str)
180     {
181         for (p = accept; *p; p++) if (*p == *str) return (LPWSTR)str;
182         str++;
183     }
184     return NULL;
185 }
186
187
188 /*********************************************************************
189  *           NTDLL_wcsrchr    (NTDLL)
190  */
191 LPWSTR __cdecl NTDLL_wcsrchr( LPWSTR str, WCHAR ch )
192 {
193     LPWSTR last = NULL;
194     while (*str)
195     {
196         if (*str == ch) last = str;
197         str++;
198     }
199     return last;
200 }
201
202
203 /*********************************************************************
204  *           NTDLL_wcsspn    (NTDLL)
205  */
206 INT __cdecl NTDLL_wcsspn( LPCWSTR str, LPCWSTR accept )
207 {
208     LPCWSTR start = str;
209     while (*str)
210     {
211         LPCWSTR p = accept;
212         while (*p && (*p != *str)) p++;
213         if (!*p) break;
214         str++;
215     }
216     return str - start;
217 }
218
219
220 /*********************************************************************
221  *           NTDLL_wcsstr    (NTDLL)
222  */
223 LPWSTR __cdecl NTDLL_wcsstr( LPCWSTR str, LPCWSTR sub )
224 {
225     return strstrW( str, sub );
226 }
227
228
229 /*********************************************************************
230  *           NTDLL_wcstok    (NTDLL)
231  */
232 LPWSTR __cdecl NTDLL_wcstok( LPWSTR str, LPCWSTR delim )
233 {
234     static LPWSTR next = NULL;
235     LPWSTR ret;
236
237     if (!str)
238         if (!(str = next)) return NULL;
239
240     while (*str && NTDLL_wcschr( delim, *str )) str++;
241     if (!*str) return NULL;
242     ret = str++;
243     while (*str && !NTDLL_wcschr( delim, *str )) str++;
244     if (*str) *str++ = 0;
245     next = str;
246     return ret;
247 }
248
249
250 /*********************************************************************
251  *           NTDLL_wcstombs    (NTDLL)
252  */
253 INT __cdecl NTDLL_wcstombs( LPSTR dst, LPCWSTR src, INT n )
254 {
255     INT ret;
256     if (n <= 0) return 0;
257     ret = WideCharToMultiByte( CP_ACP, 0, src, -1, dst, dst ? n : 0, NULL, NULL );
258     if (!ret) return n;  /* overflow */
259     return ret - 1;  /* do not count terminating NULL */
260 }
261
262
263 /*********************************************************************
264  *           NTDLL_mbstowcs    (NTDLL)
265  */
266 INT __cdecl NTDLL_mbstowcs( LPWSTR dst, LPCSTR src, INT n )
267 {
268     INT ret;
269     if (n <= 0) return 0;
270     ret = MultiByteToWideChar( CP_ACP, 0, src, -1, dst, dst ? n : 0 );
271     if (!ret) return n;  /* overflow */
272     return ret - 1;  /* do not count terminating NULL */
273 }
274
275
276 /*********************************************************************
277  *                  wcstol  (NTDLL)
278  * Like strtol, but for wide character strings.
279  */
280 INT __cdecl NTDLL_wcstol(LPWSTR s,LPWSTR *end,INT base)
281 {
282     LPSTR sA = HEAP_strdupWtoA(GetProcessHeap(),0,s),endA;
283     INT ret = strtol(sA,&endA,base);
284
285     HeapFree(GetProcessHeap(),0,sA);
286     if (end) *end = s+(endA-sA); /* pointer magic checked. */
287     return ret;
288 }
289
290
291 /*********************************************************************
292  *           NTDLL_iswctype    (NTDLL)
293  */
294 INT __cdecl NTDLL_iswctype( WCHAR wc, WCHAR wct )
295 {
296     return (get_char_typeW(wc) & 0xfff) & wct;
297 }
298
299
300 /*********************************************************************
301  *           NTDLL_iswalpha    (NTDLL)
302  */
303 INT __cdecl NTDLL_iswalpha( WCHAR wc )
304 {
305     return get_char_typeW(wc) & C1_ALPHA;
306 }
307
308
309 /*********************************************************************
310  *           _ultow    (NTDLL)
311  * Like _ultoa, but for wide character strings.
312  */
313 LPWSTR __cdecl _ultow(ULONG value, LPWSTR string, INT radix)
314 {
315     WCHAR tmp[33];
316     LPWSTR tp = tmp;
317     LPWSTR sp;
318     LONG i;
319     ULONG v = value;
320
321     if (radix > 36 || radix <= 1)
322         return 0;
323
324     while (v || tp == tmp)
325     {
326         i = v % radix;
327         v = v / radix;
328         if (i < 10)
329             *tp++ = i + '0';
330         else
331             *tp++ = i + 'a' - 10;
332     }
333
334     sp = string;
335     while (tp > tmp)
336         *sp++ = *--tp;
337     *sp = 0;
338     return string;
339 }
340
341
342 /* INTERNAL: Wide char snprintf
343  * If you fix a bug in this function, fix it in msvcrt/wcs.c also!
344  */
345 static int __cdecl NTDLL_vsnwprintf(WCHAR *str, unsigned int len,
346                                     const WCHAR *format, va_list valist)
347 {
348   unsigned int written = 0;
349   const WCHAR *iter = format;
350   char bufa[256], fmtbufa[64], *fmta;
351
352   TRACE("(%d,%s)\n",len,debugstr_w(format));
353
354   while (*iter)
355   {
356     while (*iter && *iter != (WCHAR)L'%')
357     {
358      if (written++ >= len)
359        return -1;
360      *str++ = *iter++;
361     }
362     if (*iter == (WCHAR)L'%')
363     {
364       fmta = fmtbufa;
365       *fmta++ = *iter++;
366       while (*iter == (WCHAR)L'0' ||
367              *iter == (WCHAR)L'+' ||
368              *iter == (WCHAR)L'-' ||
369              *iter == (WCHAR)L' ' ||
370              *iter == (WCHAR)L'0' ||
371              *iter == (WCHAR)L'*' ||
372              *iter == (WCHAR)L'#')
373       {
374         if (*iter == (WCHAR)L'*')
375         {
376           char *buffiter = bufa;
377           int fieldlen = va_arg(valist, int);
378           sprintf(buffiter, "%d", fieldlen);
379           while (*buffiter)
380             *fmta++ = *buffiter++;
381         }
382         else
383           *fmta++ = *iter;
384         iter++;
385       }
386
387       while (isdigit(*iter))
388         *fmta++ = *iter++;
389
390       if (*iter == (WCHAR)L'.')
391       {
392         *fmta++ = *iter++;
393         if (*iter == (WCHAR)L'*')
394         {
395           char *buffiter = bufa;
396           int fieldlen = va_arg(valist, int);
397           sprintf(buffiter, "%d", fieldlen);
398           while (*buffiter)
399             *fmta++ = *buffiter++;
400         }
401         else
402           while (isdigit(*iter))
403             *fmta++ = *iter++;
404       }
405       if (*iter == (WCHAR)L'h' ||
406           *iter == (WCHAR)L'l')
407       {
408           *fmta++ = *iter++;
409           *fmta++ = *iter++;
410       }
411
412       switch (*iter)
413       {
414       case (WCHAR)L's':
415         {
416           static const WCHAR none[] = { '(', 'n', 'u', 'l', 'l', ')', 0 };
417           const WCHAR *wstr = va_arg(valist, const WCHAR *);
418           const WCHAR *striter = wstr ? wstr : none;
419           while (*striter)
420           {
421             if (written++ >= len)
422               return -1;
423             *str++ = *striter++;
424           }
425           iter++;
426           break;
427         }
428
429       case (WCHAR)L'c':
430         if (written++ >= len)
431           return -1;
432         *str++ = (WCHAR)va_arg(valist, int);
433         iter++;
434         break;
435
436       default:
437         {
438           /* For non wc types, use system sprintf and append to wide char output */
439           /* FIXME: for unrecognised types, should ignore % when printing */
440           char *bufaiter = bufa;
441           if (*iter == (WCHAR)L'p')
442             sprintf(bufaiter, "%08lX", va_arg(valist, long));
443           else
444           {
445             *fmta++ = *iter;
446             *fmta = '\0';
447             sprintf(bufaiter, fmtbufa, va_arg(valist, void *));
448           }
449           while (*bufaiter)
450           {
451             if (written++ >= len)
452               return -1;
453             *str++ = *bufaiter++;
454           }
455           iter++;
456           break;
457         }
458       }
459     }
460   }
461   if (written >= len)
462     return -1;
463   *str++ = (WCHAR)L'\0';
464   return (int)written;
465 }
466
467
468 /***********************************************************************
469  *        _snwprintf (NTDLL)
470  */
471 int __cdecl _snwprintf(WCHAR *str, unsigned int len, const WCHAR *format, ...)
472 {
473   int retval;
474   va_list valist;
475   va_start(valist, format);
476   retval = NTDLL_vsnwprintf(str, len, format, valist);
477   va_end(valist);
478   return retval;
479 }
480
481
482 /***********************************************************************
483  *        NTDLL_swprintf (NTDLL)
484  */
485 int __cdecl NTDLL_swprintf(WCHAR *str, const WCHAR *format, ...)
486 {
487   int retval;
488   va_list valist;
489   va_start(valist, format);
490   retval = NTDLL_vsnwprintf(str, INT_MAX, format, valist);
491   va_end(valist);
492   return retval;
493 }