11949a347STim J. Robbins /*- 21949a347STim J. Robbins * Copyright (c) 2002-2004 Tim J. Robbins. 31949a347STim J. Robbins * All rights reserved. 41949a347STim J. Robbins * 51949a347STim J. Robbins * Redistribution and use in source and binary forms, with or without 61949a347STim J. Robbins * modification, are permitted provided that the following conditions 71949a347STim J. Robbins * are met: 81949a347STim J. Robbins * 1. Redistributions of source code must retain the above copyright 91949a347STim J. Robbins * notice, this list of conditions and the following disclaimer. 101949a347STim J. Robbins * 2. Redistributions in binary form must reproduce the above copyright 111949a347STim J. Robbins * notice, this list of conditions and the following disclaimer in the 121949a347STim J. Robbins * documentation and/or other materials provided with the distribution. 131949a347STim J. Robbins * 141949a347STim J. Robbins * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND 151949a347STim J. Robbins * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 161949a347STim J. Robbins * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 171949a347STim J. Robbins * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE 181949a347STim J. Robbins * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 191949a347STim J. Robbins * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 201949a347STim J. Robbins * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 211949a347STim J. Robbins * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 221949a347STim J. Robbins * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 231949a347STim J. Robbins * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 241949a347STim J. Robbins * SUCH DAMAGE. 251949a347STim J. Robbins */ 261949a347STim J. Robbins 271949a347STim J. Robbins #include <sys/cdefs.h> 281949a347STim J. Robbins __FBSDID("$FreeBSD$"); 291949a347STim J. Robbins 301949a347STim J. Robbins #include <errno.h> 311949a347STim J. Robbins #include <limits.h> 321949a347STim J. Robbins #include <stdlib.h> 331949a347STim J. Robbins #include <wchar.h> 341949a347STim J. Robbins #include "mblocal.h" 351949a347STim J. Robbins 361949a347STim J. Robbins size_t 371949a347STim J. Robbins mbsnrtowcs(wchar_t * __restrict dst, const char ** __restrict src, 381949a347STim J. Robbins size_t nms, size_t len, mbstate_t * __restrict ps) 391949a347STim J. Robbins { 401949a347STim J. Robbins static mbstate_t mbs; 411949a347STim J. Robbins 421949a347STim J. Robbins if (ps == NULL) 431949a347STim J. Robbins ps = &mbs; 441949a347STim J. Robbins return (__mbsnrtowcs(dst, src, nms, len, ps)); 451949a347STim J. Robbins } 461949a347STim J. Robbins 471949a347STim J. Robbins size_t 481949a347STim J. Robbins __mbsnrtowcs_std(wchar_t * __restrict dst, const char ** __restrict src, 491949a347STim J. Robbins size_t nms, size_t len, mbstate_t * __restrict ps) 501949a347STim J. Robbins { 511949a347STim J. Robbins const char *s; 521949a347STim J. Robbins size_t nchr; 531949a347STim J. Robbins wchar_t wc; 541949a347STim J. Robbins size_t nb; 551949a347STim J. Robbins 561949a347STim J. Robbins s = *src; 571949a347STim J. Robbins nchr = 0; 581949a347STim J. Robbins 591949a347STim J. Robbins if (dst == NULL) { 601949a347STim J. Robbins for (;;) { 611949a347STim J. Robbins if ((nb = __mbrtowc(&wc, s, nms, ps)) == (size_t)-1) 621949a347STim J. Robbins /* Invalid sequence - mbrtowc() sets errno. */ 631949a347STim J. Robbins return ((size_t)-1); 641949a347STim J. Robbins else if (nb == 0 || nb == (size_t)-2) 651949a347STim J. Robbins return (nchr); 661949a347STim J. Robbins s += nb; 671949a347STim J. Robbins nms -= nb; 681949a347STim J. Robbins nchr++; 691949a347STim J. Robbins } 701949a347STim J. Robbins /*NOTREACHED*/ 711949a347STim J. Robbins } 721949a347STim J. Robbins 731949a347STim J. Robbins while (len-- > 0) { 741949a347STim J. Robbins if ((nb = __mbrtowc(dst, s, nms, ps)) == (size_t)-1) { 751949a347STim J. Robbins *src = s; 761949a347STim J. Robbins return ((size_t)-1); 771949a347STim J. Robbins } else if (nb == (size_t)-2) { 781949a347STim J. Robbins *src = s + nms; 791949a347STim J. Robbins return (nchr); 801949a347STim J. Robbins } else if (nb == 0) { 811949a347STim J. Robbins *src = NULL; 821949a347STim J. Robbins return (nchr); 831949a347STim J. Robbins } 841949a347STim J. Robbins s += nb; 851949a347STim J. Robbins nms -= nb; 861949a347STim J. Robbins nchr++; 871949a347STim J. Robbins dst++; 881949a347STim J. Robbins } 891949a347STim J. Robbins *src = s; 901949a347STim J. Robbins return (nchr); 911949a347STim J. Robbins } 92