11590Srgrimes/*- 21590Srgrimes * Copyright (c) 2002-2004 Tim J. Robbins. 31590Srgrimes * 41590Srgrimes * Copyright (c) 2011 The FreeBSD Foundation 51590Srgrimes * All rights reserved. 61590Srgrimes * Portions of this software were developed by David Chisnall 71590Srgrimes * under sponsorship from the FreeBSD Foundation. 81590Srgrimes * All rights reserved. 91590Srgrimes * 101590Srgrimes * Redistribution and use in source and binary forms, with or without 111590Srgrimes * modification, are permitted provided that the following conditions 121590Srgrimes * are met: 131590Srgrimes * 1. Redistributions of source code must retain the above copyright 141590Srgrimes * notice, this list of conditions and the following disclaimer. 151590Srgrimes * 2. Redistributions in binary form must reproduce the above copyright 161590Srgrimes * notice, this list of conditions and the following disclaimer in the 171590Srgrimes * documentation and/or other materials provided with the distribution. 181590Srgrimes * 191590Srgrimes * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND 201590Srgrimes * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 211590Srgrimes * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 221590Srgrimes * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE 231590Srgrimes * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 241590Srgrimes * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 251590Srgrimes * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 261590Srgrimes * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 271590Srgrimes * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 281590Srgrimes * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 291590Srgrimes * SUCH DAMAGE. 301590Srgrimes */ 311590Srgrimes 321590Srgrimes#include <sys/cdefs.h> 331590Srgrimes__FBSDID("$FreeBSD$"); 341590Srgrimes 351590Srgrimes#include <errno.h> 361590Srgrimes#include <limits.h> 371590Srgrimes#include <stdlib.h> 381590Srgrimes#include <wchar.h> 391590Srgrimes#include "mblocal.h" 401590Srgrimes 41size_t 42mbsnrtowcs_l(wchar_t * __restrict dst, const char ** __restrict src, 43 size_t nms, size_t len, mbstate_t * __restrict ps, locale_t locale) 44{ 45 FIX_LOCALE(locale); 46 if (ps == NULL) 47 ps = &locale->mbsnrtowcs; 48 return (XLOCALE_CTYPE(locale)->__mbsnrtowcs(dst, src, nms, len, ps)); 49} 50size_t 51mbsnrtowcs(wchar_t * __restrict dst, const char ** __restrict src, 52 size_t nms, size_t len, mbstate_t * __restrict ps) 53{ 54 return mbsnrtowcs_l(dst, src, nms, len, ps, __get_locale()); 55} 56 57size_t 58__mbsnrtowcs_std(wchar_t * __restrict dst, const char ** __restrict src, 59 size_t nms, size_t len, mbstate_t * __restrict ps) 60{ 61 const char *s; 62 size_t nchr; 63 wchar_t wc; 64 size_t nb; 65 struct xlocale_ctype *ct = XLOCALE_CTYPE(__get_locale()); 66 67 s = *src; 68 nchr = 0; 69 70 if (dst == NULL) { 71 for (;;) { 72 if ((nb = ct->__mbrtowc(&wc, s, nms, ps)) == (size_t)-1) 73 /* Invalid sequence - mbrtowc() sets errno. */ 74 return ((size_t)-1); 75 else if (nb == 0 || nb == (size_t)-2) 76 return (nchr); 77 s += nb; 78 nms -= nb; 79 nchr++; 80 } 81 /*NOTREACHED*/ 82 } 83 84 while (len-- > 0) { 85 if ((nb = ct->__mbrtowc(dst, s, nms, ps)) == (size_t)-1) { 86 *src = s; 87 return ((size_t)-1); 88 } else if (nb == (size_t)-2) { 89 *src = s + nms; 90 return (nchr); 91 } else if (nb == 0) { 92 *src = NULL; 93 return (nchr); 94 } 95 s += nb; 96 nms -= nb; 97 nchr++; 98 dst++; 99 } 100 *src = s; 101 return (nchr); 102} 103