1 /*- 2 * SPDX-License-Identifier: BSD-2-Clause-FreeBSD 3 * 4 * Copyright 2013 Garrett D'Amore <garrett@damore.org> 5 * Copyright 2010 Nexenta Systems, Inc. All rights reserved. 6 * Copyright (c) 2002-2004 Tim J. Robbins. 7 * 8 * Copyright (c) 2011 The FreeBSD Foundation 9 * 10 * Portions of this software were developed by David Chisnall 11 * under sponsorship from the FreeBSD Foundation. 12 * 13 * Redistribution and use in source and binary forms, with or without 14 * modification, are permitted provided that the following conditions 15 * are met: 16 * 1. Redistributions of source code must retain the above copyright 17 * notice, this list of conditions and the following disclaimer. 18 * 2. Redistributions in binary form must reproduce the above copyright 19 * notice, this list of conditions and the following disclaimer in the 20 * documentation and/or other materials provided with the distribution. 21 * 22 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND 23 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 24 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 25 * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE 26 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 27 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 28 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 29 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 30 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 31 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 32 * SUCH DAMAGE. 33 */ 34 35 #include <sys/cdefs.h> 36 __FBSDID("$FreeBSD$"); 37 38 #include <errno.h> 39 #include <limits.h> 40 #include <stdlib.h> 41 #include <wchar.h> 42 #include "mblocal.h" 43 44 size_t 45 mbsnrtowcs_l(wchar_t * __restrict dst, const char ** __restrict src, 46 size_t nms, size_t len, mbstate_t * __restrict ps, locale_t locale) 47 { 48 FIX_LOCALE(locale); 49 if (ps == NULL) 50 ps = &(XLOCALE_CTYPE(locale)->mbsnrtowcs); 51 return (XLOCALE_CTYPE(locale)->__mbsnrtowcs(dst, src, nms, len, ps)); 52 } 53 size_t 54 mbsnrtowcs(wchar_t * __restrict dst, const char ** __restrict src, 55 size_t nms, size_t len, mbstate_t * __restrict ps) 56 { 57 return mbsnrtowcs_l(dst, src, nms, len, ps, __get_locale()); 58 } 59 60 size_t 61 __mbsnrtowcs_std(wchar_t * __restrict dst, const char ** __restrict src, 62 size_t nms, size_t len, mbstate_t * __restrict ps, 63 mbrtowc_pfn_t pmbrtowc) 64 { 65 const char *s; 66 size_t nchr; 67 wchar_t wc; 68 size_t nb; 69 70 s = *src; 71 nchr = 0; 72 73 if (dst == NULL) { 74 for (;;) { 75 if ((nb = pmbrtowc(&wc, s, nms, ps)) == (size_t)-1) 76 /* Invalid sequence - mbrtowc() sets errno. */ 77 return ((size_t)-1); 78 else if (nb == 0 || nb == (size_t)-2) 79 return (nchr); 80 s += nb; 81 nms -= nb; 82 nchr++; 83 } 84 /*NOTREACHED*/ 85 } 86 87 while (len-- > 0) { 88 if ((nb = pmbrtowc(dst, s, nms, ps)) == (size_t)-1) { 89 *src = s; 90 return ((size_t)-1); 91 } else if (nb == (size_t)-2) { 92 *src = s + nms; 93 return (nchr); 94 } else if (nb == 0) { 95 *src = NULL; 96 return (nchr); 97 } 98 s += nb; 99 nms -= nb; 100 nchr++; 101 dst++; 102 } 103 *src = s; 104 return (nchr); 105 } 106