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