1 /* 2 * Copyright 2013 Garrett D'Amore <garrett@damore.org> 3 * Copyright 2010 Nexenta Systems, Inc. All rights reserved. 4 * Copyright (c) 2002-2004 Tim J. Robbins. 5 * All rights reserved. 6 * 7 * Redistribution and use in source and binary forms, with or without 8 * modification, are permitted provided that the following conditions 9 * are met: 10 * 1. Redistributions of source code must retain the above copyright 11 * notice, this list of conditions and the following disclaimer. 12 * 2. Redistributions in binary form must reproduce the above copyright 13 * notice, this list of conditions and the following disclaimer in the 14 * documentation and/or other materials provided with the distribution. 15 * 16 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND 17 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 18 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 19 * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE 20 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 21 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 22 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 23 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 24 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 25 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 26 * SUCH DAMAGE. 27 */ 28 29 #include "lint.h" 30 #include <limits.h> 31 #include <stdlib.h> 32 #include <string.h> 33 #include <locale.h> 34 #include <wchar.h> 35 #include "mblocal.h" 36 #include "localeimpl.h" 37 #include "lctype.h" 38 39 size_t 40 wcsnrtombs_l(char *_RESTRICT_KYWD dst, const wchar_t **_RESTRICT_KYWD src, 41 size_t nwc, size_t len, mbstate_t *_RESTRICT_KYWD ps, locale_t loc) 42 { 43 static mbstate_t mbs; 44 45 if (ps == NULL) 46 ps = &mbs; 47 return (loc->ctype->lc_wcsnrtombs(dst, src, nwc, len, ps)); 48 } 49 50 size_t 51 wcsnrtombs(char *_RESTRICT_KYWD dst, const wchar_t **_RESTRICT_KYWD src, 52 size_t nwc, size_t len, mbstate_t *_RESTRICT_KYWD ps) 53 { 54 return (wcsnrtombs_l(dst, src, nwc, len, ps, uselocale(NULL))); 55 } 56 57 size_t 58 __wcsnrtombs_std(char *_RESTRICT_KYWD dst, const wchar_t **_RESTRICT_KYWD src, 59 size_t nwc, size_t len, mbstate_t *_RESTRICT_KYWD ps, 60 wcrtomb_pfn_t pwcrtomb) 61 { 62 mbstate_t mbsbak; 63 char buf[MB_LEN_MAX]; 64 const wchar_t *s; 65 size_t nbytes; 66 size_t nb; 67 68 s = *src; 69 nbytes = 0; 70 71 if (dst == NULL) { 72 while (nwc-- > 0) { 73 if ((nb = pwcrtomb(buf, *s, ps)) == (size_t)-1) 74 /* Invalid character - wcrtomb() sets errno. */ 75 return ((size_t)-1); 76 else if (*s == L'\0') 77 return (nbytes + nb - 1); 78 s++; 79 nbytes += nb; 80 } 81 return (nbytes); 82 } 83 84 while (len > 0 && nwc-- > 0) { 85 if (len > (size_t)MB_CUR_MAX) { 86 /* Enough space to translate in-place. */ 87 if ((nb = pwcrtomb(dst, *s, ps)) == (size_t)-1) { 88 *src = s; 89 return ((size_t)-1); 90 } 91 } else { 92 /* 93 * May not be enough space; use temp. buffer. 94 * 95 * We need to save a copy of the conversion state 96 * here so we can restore it if the multibyte 97 * character is too long for the buffer. 98 */ 99 mbsbak = *ps; 100 if ((nb = pwcrtomb(buf, *s, ps)) == (size_t)-1) { 101 *src = s; 102 return ((size_t)-1); 103 } 104 if (nb > (int)len) { 105 /* MB sequence for character won't fit. */ 106 *ps = mbsbak; 107 break; 108 } 109 (void) memcpy(dst, buf, nb); 110 } 111 if (*s == L'\0') { 112 *src = NULL; 113 return (nbytes + nb - 1); 114 } 115 s++; 116 dst += nb; 117 len -= nb; 118 nbytes += nb; 119 } 120 *src = s; 121 return (nbytes); 122 } 123