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#include <sys/cdefs.h> 28__FBSDID("$FreeBSD: src/lib/libc/locale/mbsnrtowcs.c,v 1.1 2004/07/21 10:54:57 tjr Exp $"); 29 30#include "xlocale_private.h" 31 32#include <errno.h> 33#include <limits.h> 34#include <stdlib.h> 35#include <wchar.h> 36#include "mblocal.h" 37 38size_t 39mbsnrtowcs_l(wchar_t * __restrict dst, const char ** __restrict src, 40 size_t nms, size_t len, mbstate_t * __restrict ps, locale_t loc) 41{ 42 NORMALIZE_LOCALE(loc); 43 if (ps == NULL) 44 ps = &loc->__mbs_mbsnrtowcs; 45 return (loc->__lc_ctype->__mbsnrtowcs(dst, src, nms, len, ps, loc)); 46} 47 48size_t 49mbsnrtowcs(wchar_t * __restrict dst, const char ** __restrict src, 50 size_t nms, size_t len, mbstate_t * __restrict ps) 51{ 52 return mbsnrtowcs_l(dst, src, nms, len, ps, __current_locale()); 53} 54 55__private_extern__ size_t 56__mbsnrtowcs_std(wchar_t * __restrict dst, const char ** __restrict src, 57 size_t nms, size_t len, mbstate_t * __restrict ps, locale_t loc) 58{ 59 const char *s; 60 size_t nchr; 61 wchar_t wc; 62 size_t nb; 63 size_t (*__mbrtowc)(wchar_t * __restrict, const char * __restrict, 64 size_t, mbstate_t * __restrict, locale_t) 65 = loc->__lc_ctype->__mbrtowc; 66 67 s = *src; 68 nchr = 0; 69 70 if (dst == NULL) { 71 for (;;) { 72 if ((nb = __mbrtowc(&wc, s, nms, ps, loc)) == (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 = __mbrtowc(dst, s, nms, ps, loc)) == (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