dlls/ntdll/wcstring.c

   1 /*
   2  * NTDLL wide-char functions
   3  *
   4  * Copyright 2000 Alexandre Julliard
   5  * Copyright 2000 Jon Griffiths
   6  *
   7  * This library is free software; you can redistribute it and/or
   8  * modify it under the terms of the GNU Lesser General Public
   9  * License as published by the Free Software Foundation; either
  10  * version 2.1 of the License, or (at your option) any later version.
  11  *
  12  * This library is distributed in the hope that it will be useful,
  13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  15  * Lesser General Public License for more details.
  16  *
  17  * You should have received a copy of the GNU Lesser General Public
  18  * License along with this library; if not, write to the Free Software
  19  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  20  */
  21
  22 #include "config.h"
  23
  24 #include <ctype.h>
  25 #include <limits.h>
  26 #include <stdlib.h>
  27 #include <string.h>
  28 #include <stdio.h>
  29
  30 #include "ntddk.h"
  31 #include "wine/unicode.h"
  32 #include "wine/debug.h"
  33
  34 WINE_DEFAULT_DEBUG_CHANNEL(ntdll);
  35
  36
  37 /*********************************************************************
  38  *           _wcsicmp    (NTDLL.@)
  39  */
  40 INT __cdecl NTDLL__wcsicmp( LPCWSTR str1, LPCWSTR str2 )
  41 {
  42     return strcmpiW( str1, str2 );
  43 }
  44
  45
  46 /*********************************************************************
  47  *           _wcslwr    (NTDLL.@)
  48  */
  49 LPWSTR __cdecl NTDLL__wcslwr( LPWSTR str )
  50 {
  51     return strlwrW( str );
  52 }
  53
  54
  55 /*********************************************************************
  56  *           _wcsnicmp    (NTDLL.@)
  57  */
  58 INT __cdecl NTDLL__wcsnicmp( LPCWSTR str1, LPCWSTR str2, INT n )
  59 {
  60     return strncmpiW( str1, str2, n );
  61 }
  62
  63
  64 /*********************************************************************
  65  *           _wcsupr    (NTDLL.@)
  66  */
  67 LPWSTR __cdecl NTDLL__wcsupr( LPWSTR str )
  68 {
  69     return struprW( str );
  70 }
  71
  72
  73 /*********************************************************************
  74  *           towlower    (NTDLL.@)
  75  */
  76 WCHAR __cdecl NTDLL_towlower( WCHAR ch )
  77 {
  78     return tolowerW(ch);
  79 }
  80
  81
  82 /*********************************************************************
  83  *           towupper    (NTDLL.@)
  84  */
  85 WCHAR __cdecl NTDLL_towupper( WCHAR ch )
  86 {
  87     return toupperW(ch);
  88 }
  89
  90
  91 /***********************************************************************
  92  *           wcscat    (NTDLL.@)
  93  */
  94 LPWSTR __cdecl NTDLL_wcscat( LPWSTR dst, LPCWSTR src )
  95 {
  96     return strcatW( dst, src );
  97 }
  98
  99
 100 /*********************************************************************
 101  *           wcschr    (NTDLL.@)
 102  */
 103 LPWSTR __cdecl NTDLL_wcschr( LPCWSTR str, WCHAR ch )
 104 {
 105     return strchrW( str, ch );
 106 }
 107
 108
 109 /*********************************************************************
 110  *           wcscmp    (NTDLL.@)
 111  */
 112 INT __cdecl NTDLL_wcscmp( LPCWSTR str1, LPCWSTR str2 )
 113 {
 114     return strcmpW( str1, str2 );
 115 }
 116
 117
 118 /***********************************************************************
 119  *           wcscpy    (NTDLL.@)
 120  */
 121 LPWSTR __cdecl NTDLL_wcscpy( LPWSTR dst, LPCWSTR src )
 122 {
 123     return strcpyW( dst, src );
 124 }
 125
 126
 127 /*********************************************************************
 128  *           wcscspn    (NTDLL.@)
 129  */
 130 INT __cdecl NTDLL_wcscspn( LPCWSTR str, LPCWSTR reject )
 131 {
 132     LPCWSTR start = str;
 133     while (*str)
 134     {
 135         LPCWSTR p = reject;
 136         while (*p && (*p != *str)) p++;
 137         if (*p) break;
 138         str++;
 139     }
 140     return str - start;
 141 }
 142
 143
 144 /***********************************************************************
 145  *           wcslen    (NTDLL.@)
 146  */
 147 INT __cdecl NTDLL_wcslen( LPCWSTR str )
 148 {
 149     return strlenW( str );
 150 }
 151
 152
 153 /*********************************************************************
 154  *           wcsncat    (NTDLL.@)
 155  */
 156 LPWSTR __cdecl NTDLL_wcsncat( LPWSTR s1, LPCWSTR s2, INT n )
 157 {
 158     LPWSTR ret = s1;
 159     while (*s1) s1++;
 160     while (n-- > 0) if (!(*s1++ = *s2++)) return ret;
 161     *s1 = 0;
 162     return ret;
 163 }
 164
 165
 166 /*********************************************************************
 167  *           wcsncmp    (NTDLL.@)
 168  */
 169 INT __cdecl NTDLL_wcsncmp( LPCWSTR str1, LPCWSTR str2, INT n )
 170 {
 171     return strncmpW( str1, str2, n );
 172 }
 173
 174
 175 /*********************************************************************
 176  *           wcsncpy    (NTDLL.@)
 177  */
 178 LPWSTR __cdecl NTDLL_wcsncpy( LPWSTR s1, LPCWSTR s2, INT n )
 179 {
 180     return strncpyW( s1, s2, n );
 181 }
 182
 183
 184 /*********************************************************************
 185  *           wcspbrk    (NTDLL.@)
 186  */
 187 LPWSTR __cdecl NTDLL_wcspbrk( LPCWSTR str, LPCWSTR accept )
 188 {
 189     LPCWSTR p;
 190     while (*str)
 191     {
 192         for (p = accept; *p; p++) if (*p == *str) return (LPWSTR)str;
 193         str++;
 194     }
 195     return NULL;
 196 }
 197
 198
 199 /*********************************************************************
 200  *           wcsrchr    (NTDLL.@)
 201  */
 202 LPWSTR __cdecl NTDLL_wcsrchr( LPWSTR str, WCHAR ch )
 203 {
 204     LPWSTR last = NULL;
 205     while (*str)
 206     {
 207         if (*str == ch) last = str;
 208         str++;
 209     }
 210     return last;
 211 }
 212
 213
 214 /*********************************************************************
 215  *           wcsspn    (NTDLL.@)
 216  */
 217 INT __cdecl NTDLL_wcsspn( LPCWSTR str, LPCWSTR accept )
 218 {
 219     LPCWSTR start = str;
 220     while (*str)
 221     {
 222         LPCWSTR p = accept;
 223         while (*p && (*p != *str)) p++;
 224         if (!*p) break;
 225         str++;
 226     }
 227     return str - start;
 228 }
 229
 230
 231 /*********************************************************************
 232  *           wcsstr    (NTDLL.@)
 233  */
 234 LPWSTR __cdecl NTDLL_wcsstr( LPCWSTR str, LPCWSTR sub )
 235 {
 236     return strstrW( str, sub );
 237 }
 238
 239
 240 /*********************************************************************
 241  *           wcstok    (NTDLL.@)
 242  */
 243 LPWSTR __cdecl NTDLL_wcstok( LPWSTR str, LPCWSTR delim )
 244 {
 245     static LPWSTR next = NULL;
 246     LPWSTR ret;
 247
 248     if (!str)
 249         if (!(str = next)) return NULL;
 250
 251     while (*str && NTDLL_wcschr( delim, *str )) str++;
 252     if (!*str) return NULL;
 253     ret = str++;
 254     while (*str && !NTDLL_wcschr( delim, *str )) str++;
 255     if (*str) *str++ = 0;
 256     next = str;
 257     return ret;
 258 }
 259
 260
 261 /*********************************************************************
 262  *           wcstombs    (NTDLL.@)
 263  */
 264 INT __cdecl NTDLL_wcstombs( LPSTR dst, LPCWSTR src, INT n )
 265 {
 266     DWORD len;
 267
 268     if (!dst)
 269     {
 270         RtlUnicodeToMultiByteSize( &len, src, strlenW(src)*sizeof(WCHAR) );
 271         return len;
 272     }
 273     else
 274     {
 275         if (n <= 0) return 0;
 276         RtlUnicodeToMultiByteN( dst, n, &len, src, strlenW(src)*sizeof(WCHAR) );
 277         if (len < n) dst[len] = 0;
 278     }
 279     return len;
 280 }
 281
 282
 283 /*********************************************************************
 284  *           mbstowcs    (NTDLL.@)
 285  */
 286 INT __cdecl NTDLL_mbstowcs( LPWSTR dst, LPCSTR src, INT n )
 287 {
 288     DWORD len;
 289
 290     if (!dst)
 291     {
 292         RtlMultiByteToUnicodeSize( &len, src, strlen(src) );
 293     }
 294     else
 295     {
 296         if (n <= 0) return 0;
 297         RtlMultiByteToUnicodeN( dst, n*sizeof(WCHAR), &len, src, strlen(src) );
 298         if (len / sizeof(WCHAR) < n) dst[len / sizeof(WCHAR)] = 0;
 299     }
 300     return len / sizeof(WCHAR);
 301 }
 302
 303
 304 /*********************************************************************
 305  *                  wcstol  (NTDLL.@)
 306  */
 307 long __cdecl NTDLL_wcstol(LPCWSTR s,LPWSTR *end,INT base)
 308 {
 309     return strtolW( s, end, base );
 310 }
 311
 312
 313 /*********************************************************************
 314  *                  wcstoul  (NTDLL.@)
 315  */
 316 unsigned long __cdecl NTDLL_wcstoul(LPCWSTR s,LPWSTR *end,INT base)
 317 {
 318     return strtoulW( s, end, base );
 319 }
 320
 321
 322 /*********************************************************************
 323  *           iswctype    (NTDLL.@)
 324  */
 325 INT __cdecl NTDLL_iswctype( WCHAR wc, WCHAR wct )
 326 {
 327     return (get_char_typeW(wc) & 0xfff) & wct;
 328 }
 329
 330
 331 /*********************************************************************
 332  *           iswalpha    (NTDLL.@)
 333  */
 334 INT __cdecl NTDLL_iswalpha( WCHAR wc )
 335 {
 336     return get_char_typeW(wc) & C1_ALPHA;
 337 }
 338
 339
 340 /*********************************************************************
 341  *           _ultow    (NTDLL.@)
 342  * Like _ultoa, but for wide character strings.
 343  */
 344 LPWSTR __cdecl _ultow(ULONG value, LPWSTR string, INT radix)
 345 {
 346     WCHAR tmp[33];
 347     LPWSTR tp = tmp;
 348     LPWSTR sp;
 349     LONG i;
 350     ULONG v = value;
 351
 352     if (radix > 36 || radix <= 1)
 353         return 0;
 354
 355     while (v || tp == tmp)
 356     {
 357         i = v % radix;
 358         v = v / radix;
 359         if (i < 10)
 360             *tp++ = i + '0';
 361         else
 362             *tp++ = i + 'a' - 10;
 363     }
 364
 365     sp = string;
 366     while (tp > tmp)
 367         *sp++ = *--tp;
 368     *sp = 0;
 369     return string;
 370 }
 371
 372 /*********************************************************************
 373  *           _wtol    (NTDLL.@)
 374  * Like atol, but for wide character strings.
 375  */
 376 LONG __cdecl NTDLL__wtol(LPWSTR string)
 377 {
 378     return strtolW( string, NULL, 10 );
 379 }
 380
 381 /*********************************************************************
 382  *           _wtoi    (NTDLL.@)
 383  */
 384 INT __cdecl NTDLL__wtoi(LPWSTR string)
 385 {
 386     return NTDLL__wtol(string);
 387 }
 388
 389 /* INTERNAL: Wide char snprintf
 390  * If you fix a bug in this function, fix it in msvcrt/wcs.c also!
 391  */
 392 static int __cdecl NTDLL_vsnwprintf(WCHAR *str, unsigned int len,
 393                                     const WCHAR *format, va_list valist)
 394 {
 395   unsigned int written = 0;
 396   const WCHAR *iter = format;
 397   char bufa[256], fmtbufa[64], *fmta;
 398
 399   TRACE("(%d,%s)\n",len,debugstr_w(format));
 400
 401   while (*iter)
 402   {
 403     while (*iter && *iter != (WCHAR)L'%')
 404     {
 405      if (written++ >= len)
 406        return -1;
 407      *str++ = *iter++;
 408     }
 409     if (*iter == (WCHAR)L'%')
 410     {
 411       fmta = fmtbufa;
 412       *fmta++ = *iter++;
 413       while (*iter == (WCHAR)L'0' ||
 414              *iter == (WCHAR)L'+' ||
 415              *iter == (WCHAR)L'-' ||
 416              *iter == (WCHAR)L' ' ||
 417              *iter == (WCHAR)L'0' ||
 418              *iter == (WCHAR)L'*' ||
 419              *iter == (WCHAR)L'#')
 420       {
 421         if (*iter == (WCHAR)L'*')
 422         {
 423           char *buffiter = bufa;
 424           int fieldlen = va_arg(valist, int);
 425           sprintf(buffiter, "%d", fieldlen);
 426           while (*buffiter)
 427             *fmta++ = *buffiter++;
 428         }
 429         else
 430           *fmta++ = *iter;
 431         iter++;
 432       }
 433
 434       while (isdigit(*iter))
 435         *fmta++ = *iter++;
 436
 437       if (*iter == (WCHAR)L'.')
 438       {
 439         *fmta++ = *iter++;
 440         if (*iter == (WCHAR)L'*')
 441         {
 442           char *buffiter = bufa;
 443           int fieldlen = va_arg(valist, int);
 444           sprintf(buffiter, "%d", fieldlen);
 445           while (*buffiter)
 446             *fmta++ = *buffiter++;
 447         }
 448         else
 449           while (isdigit(*iter))
 450             *fmta++ = *iter++;
 451       }
 452       if (*iter == (WCHAR)L'h' ||
 453           *iter == (WCHAR)L'l')
 454           *fmta++ = *iter++;
 455
 456       switch (*iter)
 457       {
 458       case (WCHAR)L's':
 459         {
 460           static const WCHAR none[] = { '(', 'n', 'u', 'l', 'l', ')', 0 };
 461           const WCHAR *wstr = va_arg(valist, const WCHAR *);
 462           const WCHAR *striter = wstr ? wstr : none;
 463           while (*striter)
 464           {
 465             if (written++ >= len)
 466               return -1;
 467             *str++ = *striter++;
 468           }
 469           iter++;
 470           break;
 471         }
 472
 473       case (WCHAR)L'c':
 474         if (written++ >= len)
 475           return -1;
 476         *str++ = (WCHAR)va_arg(valist, int);
 477         iter++;
 478         break;
 479
 480       default:
 481         {
 482           /* For non wc types, use system sprintf and append to wide char output */
 483           /* FIXME: for unrecognised types, should ignore % when printing */
 484           char *bufaiter = bufa;
 485           if (*iter == (WCHAR)L'p')
 486             sprintf(bufaiter, "%08lX", va_arg(valist, long));
 487           else
 488           {
 489             *fmta++ = *iter;
 490             *fmta = '\0';
 491             if (*iter == (WCHAR)L'f')
 492               sprintf(bufaiter, fmtbufa, va_arg(valist, double));
 493             else
 494               sprintf(bufaiter, fmtbufa, va_arg(valist, void *));
 495           }
 496           while (*bufaiter)
 497           {
 498             if (written++ >= len)
 499               return -1;
 500             *str++ = *bufaiter++;
 501           }
 502           iter++;
 503           break;
 504         }
 505       }
 506     }
 507   }
 508   if (written >= len)
 509     return -1;
 510   *str++ = (WCHAR)L'\0';
 511   return (int)written;
 512 }
 513
 514
 515 /***********************************************************************
 516  *        _snwprintf (NTDLL.@)
 517  */
 518 int __cdecl _snwprintf(WCHAR *str, unsigned int len, const WCHAR *format, ...)
 519 {
 520   int retval;
 521   va_list valist;
 522   va_start(valist, format);
 523   retval = NTDLL_vsnwprintf(str, len, format, valist);
 524   va_end(valist);
 525   return retval;
 526 }
 527
 528
 529 /***********************************************************************
 530  *        swprintf (NTDLL.@)
 531  */
 532 int __cdecl NTDLL_swprintf(WCHAR *str, const WCHAR *format, ...)
 533 {
 534   int retval;
 535   va_list valist;
 536   va_start(valist, format);
 537   retval = NTDLL_vsnwprintf(str, INT_MAX, format, valist);
 538   va_end(valist);
 539   return retval;
 540 }