lib/libc/locale/mbsnrtowcs.c

   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 = &(XLOCALE_CTYPE(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 }