sortix-mirror/libc/wchar/mbsnrtowcs.cpp

83 lines
2.5 KiB
C++

/*******************************************************************************
Copyright(C) Jonas 'Sortie' Termansen 2013, 2014.
This file is part of the Sortix C Library.
The Sortix C Library is free software: you can redistribute it and/or modify
it under the terms of the GNU Lesser General Public License as published by
the Free Software Foundation, either version 3 of the License, or (at your
option) any later version.
The Sortix C Library is distributed in the hope that it will be useful, but
WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
or FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public
License for more details.
You should have received a copy of the GNU Lesser General Public License
along with the Sortix C Library. If not, see <http://www.gnu.org/licenses/>.
wchar/mbsnrtowcs.cpp
Convert a multibyte string to a wide-character string.
*******************************************************************************/
#include <assert.h>
#include <wchar.h>
extern "C"
size_t mbsnrtowcs(wchar_t* restrict dst,
const char** restrict src_ptr,
size_t src_len,
size_t dst_len,
mbstate_t* restrict ps)
{
static mbstate_t static_ps;
if ( !ps )
ps = &static_ps;
assert(src_ptr && *src_ptr);
const char* src = *src_ptr;
// Continue to decode wide characters until we have filled the destination
// buffer or if we have exhausted the limit on input multibyte characters.
size_t dst_offset = 0;
size_t src_offset = 0;
while ( (!dst || dst_offset < dst_len) && src_offset < src_len )
{
mbstate_t ps_copy = *ps;
wchar_t wc;
size_t amount = mbrtowc(&wc, src + src_offset, src_len - src_offset, ps);
// Stop in the event a decoding error occured.
if ( amount == (size_t) -1 )
return *src_ptr = src + src_offset, (size_t) -1;
// Stop decoding early in the event we encountered a partial character.
if ( amount == (size_t) -2 )
{
*ps = ps_copy;
break;
}
// Store the decoded wide character in the destination buffer.
if ( dst )
dst[dst_offset] = wc;
// Stop decoding after decoding a null character and return a NULL
// source pointer to the caller, not including the null character in the
// number of characters stored in the destination buffer.
if ( wc == L'\0' )
{
src = NULL;
src_offset = 0;
break;
}
dst_offset++;
src_offset += amount;
}
return *src_ptr = src + src_offset, dst_offset;
}