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