/***
*mbstowcs.c - Convert multibyte char string to wide char string.
*
* Copyright (c) 1990-2001, Microsoft Corporation. All rights reserved.
*
*Purpose:
* Convert a multibyte char string into the equivalent wide char string.
*
*Revision History:
* 08-24-90 KRS Module created.
* 03-20-91 KRS Ported from 16-bit tree.
* 10-16-91 ETC Locale support under _INTL switch.
* 12-09-91 ETC Updated nlsapi; added multithread.
* 08-20-92 KRS Activated NLSAPI support.
* 08-31-92 SRW Allow INTL definition to be conditional for building ntcrt.lib
* 09-02-92 SRW Get _INTL definition via ..\crt32.def
* 02-09-93 CFW Always stuff WC 0 at end of output string of room (non _INTL).
* 04-06-93 SKS Replace _CRTAPI* with _cdecl
* 05-03-93 CFW Return pointer == NULL, return size, plus massive cleanup.
* 06-01-93 CFW Minor optimization and beautify.
* 06-02-93 SRW ignore _INTL if _NTSUBSET_ defined.
* 09-15-93 CFW Use ANSI conformant "__" names.
* 09-21-93 CFW Avoid cast bug.
* 09-27-93 GJF Merged NT SDK and Cuda.
* 10-22-93 CFW Test for invalid MB chars using global preset flag.
* 01-14-94 SRW if _NTSUBSET_ defined call Rtl functions
* 02-03-94 GJF Merged in Steve Wood's latest change (affects
* _NTSUBSET_ build only).
* 02-07-94 CFW POSIXify.
* 08-03-94 CFW Bug #15300; fix MBToWC workaround for small buffer.
* 09-06-94 CFW Remove _INTL switch.
* 10-18-94 BWT Fix build warning for call to RtlMultiByteToUnicodeN
* 12-21-94 CFW Remove invalid MB chars NT 3.1 hack.
* 01-07-95 CFW Mac merge cleanup.
* 02-06-95 CFW assert -> _ASSERTE.
* 04-19-95 CFW Rearrange & fix non-Win32 version.
* 09-26-95 GJF New locking macro, and scheme, for functions which
* reference the locale.
* 04-01-96 BWT POSIX work.
* 05-26-96 BWT Return the word count, not the byte count for
* _NTSUBSET_/POSIX case.
* 06-25-96 GJF Removed DLL_FOR_WIN32S. Replaced defined(_WIN32) with
* !defined(_MAC). Polished the format a bit.
* 07-27-98 GJF Revised multithread support based on threadlocinfo
* struct.
* 12-15-98 GJF Changes for 64-bit size_t.
* 05-17-99 PML Remove all Macintosh support.
*
*******************************************************************************/
#if defined(_NTSUBSET_) || defined(_POSIX_)
#include <nt.h>
#include <ntrtl.h>
#include <nturtl.h>
#endif
#include <internal.h>
#include <locale.h>
#include <setlocal.h>
#include <errno.h>
#include <cruntime.h>
#include <stdlib.h>
#include <string.h>
#include <mtdll.h>
#include <dbgint.h>
#include <stdio.h>
/***
*size_t mbstowcs() - Convert multibyte char string to wide char string.
*
*Purpose:
* Convert a multi-byte char string into the equivalent wide char string,
* according to the LC_CTYPE category of the current locale.
* [ANSI].
*
*Entry:
* wchar_t *pwcs = pointer to destination wide character string buffer
* const char *s = pointer to source multibyte character string
* size_t n = maximum number of wide characters to store
*
*Exit:
* If s != NULL, returns: number of words modified (<=n)
* (size_t)-1 (if invalid mbcs)
*
*Exceptions:
* Returns (size_t)-1 if s is NULL or invalid mbcs character encountered
*
*******************************************************************************/
size_t __cdecl mbstowcs
(
wchar_t *pwcs,
const char *s,
size_t n
)
{
#ifdef _MT
pthreadlocinfo ptloci = _getptd()->ptlocinfo;
if ( ptloci != __ptlocinfo )
ptloci = __updatetlocinfo();
return __mbstowcs_mt(ptloci, pwcs, s, n);
}
size_t __cdecl __mbstowcs_mt (
pthreadlocinfo ptloci,
wchar_t *pwcs,
const char *s,
size_t n
)
{
#endif
size_t count = 0;
if (pwcs && n == 0)
/* dest string exists, but 0 bytes converted */
return (size_t) 0;
_ASSERTE(s != NULL);
#ifdef _WIN64
/* n must fit into an int for MultiByteToWideChar */
if ( n > INT_MAX )
return (size_t)-1;
#endif
#if !defined(_NTSUBSET_) && !defined(_POSIX_)
/* if destination string exists, fill it in */
if (pwcs)
{
#ifdef _MT
if (ptloci->lc_handle[LC_CTYPE] == _CLOCALEHANDLE)
#else
if (__lc_handle[LC_CTYPE] == _CLOCALEHANDLE)
#endif
{
/* C locale: easy and fast */
while (count < n)
{
*pwcs = (wchar_t) ((unsigned char)s[count]);
if (!s[count])
return count;
count++;
pwcs++;
}
return count;
} else {
int bytecnt, charcnt;
unsigned char *p;
/* Assume that the buffer is large enough */
#ifdef _MT
if ( (count = MultiByteToWideChar( ptloci->lc_codepage,
#else
if ( (count = MultiByteToWideChar( __lc_codepage,
#endif
MB_PRECOMPOSED |
MB_ERR_INVALID_CHARS,
s,
-1,
pwcs,
(int)n )) != 0 )
return count - 1; /* don't count NUL */
if (GetLastError() != ERROR_INSUFFICIENT_BUFFER)
{
errno = EILSEQ;
return (size_t)-1;
}
/* User-supplied buffer not large enough. */
/* How many bytes are in n characters of the string? */
charcnt = (int)n;
for (p = (unsigned char *)s; (charcnt-- && *p); p++)
{
#ifdef _MT
if (__isleadbyte_mt(ptloci, *p))
#else
if (isleadbyte(*p))
#endif
p++;
}
bytecnt = ((int) ((char *)p - (char *)s));
#ifdef _MT
if ( (count = MultiByteToWideChar( ptloci->lc_codepage,
#else
if ( (count = MultiByteToWideChar( __lc_codepage,
#endif
MB_PRECOMPOSED,
s,
bytecnt,
pwcs,
(int)n )) == 0 )
{
errno = EILSEQ;
return (size_t)-1;
}
return count; /* no NUL in string */
}
}
else { /* pwcs == NULL, get size only, s must be NUL-terminated */
#ifdef _MT
if (ptloci->lc_handle[LC_CTYPE] == _CLOCALEHANDLE)
#else
if (__lc_handle[LC_CTYPE] == _CLOCALEHANDLE)
#endif
return strlen(s);
else {
#ifdef _MT
if ( (count = MultiByteToWideChar( ptloci->lc_codepage,
#else
if ( (count = MultiByteToWideChar( __lc_codepage,
#endif
MB_PRECOMPOSED |
MB_ERR_INVALID_CHARS,
s,
-1,
NULL,
0 )) == 0 )
{
errno = EILSEQ;
return (size_t)-1;
}
return count - 1;
}
}
#else /* _NTSUBSET_/_POSIX_ */
if (pwcs) {
NTSTATUS Status;
int size;
size = _mbstrlen(s);
Status = RtlMultiByteToUnicodeN(pwcs,
(ULONG) ( n * sizeof( *pwcs ) ),
(PULONG)&size,
(char *)s,
size+1 );
if (!NT_SUCCESS(Status)) {
errno = EILSEQ;
size = -1;
} else {
size = size / sizeof(*pwcs);
if (pwcs[size-1] == L'\0') {
size -= 1;
}
}
return size;
} else { /* pwcs == NULL, get size only, s must be NUL-terminated */
return strlen(s);
}
#endif /* _NTSUBSET_/_POSIX_ */
}