lib/mbsnrtowcs-impl.h

   1 /* Convert string to wide string.
   2    Copyright (C) 2008-2024 Free Software Foundation, Inc.
   3    Written by Bruno Haible <bruno@clisp.org>, 2008.
   4
   5    This file is free software: you can redistribute it and/or modify
   6    it under the terms of the GNU Lesser General Public License as
   7    published by the Free Software Foundation, either version 3 of the
   8    License, or (at your option) any later version.
   9
  10    This file is distributed in the hope that it will be useful,
  11    but WITHOUT ANY WARRANTY; without even the implied warranty of
  12    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13    GNU Lesser General Public License for more details.
  14
  15    You should have received a copy of the GNU Lesser General Public License
  16    along with this program.  If not, see <https://www.gnu.org/licenses/>.  */
  17
  18 size_t
  19 FUNC (DCHAR_T *dest, const char **srcp, size_t srclen, size_t len, mbstate_t *ps)
  20 {
  21   if (ps == NULL)
  22     ps = &INTERNAL_STATE;
  23   {
  24     const char *src = *srcp;
  25
  26     if (dest != NULL)
  27       {
  28         DCHAR_T *destptr = dest;
  29
  30         for (; srclen > 0 && len > 0; destptr++, len--)
  31           {
  32             size_t src_avail;
  33             size_t ret;
  34
  35             /* An optimized variant of
  36                src_avail = strnlen1 (src, MIN (srclen, MB_LEN_MAX));  */
  37             if (srclen == 1 || src[0] == '\0')
  38               src_avail = 1;
  39             else if (srclen == 2 || src[1] == '\0')
  40               src_avail = 2;
  41             else if (srclen == 3 || src[2] == '\0')
  42               src_avail = 3;
  43             else if (MB_LEN_MAX <= 4 || srclen == 4 || src[3] == '\0')
  44               src_avail = 4;
  45             else
  46               src_avail = 4 + strnlen1 (src + 4, MIN (srclen, MB_LEN_MAX) - 4);
  47
  48             /* Parse the next multibyte character.  */
  49             ret = MBRTOWC (destptr, src, src_avail, ps);
  50
  51             if (ret == (size_t)(-2))
  52               /* Encountered a multibyte character that extends past a '\0' byte
  53                  or that is longer than MB_LEN_MAX bytes.  Cannot happen.  */
  54               abort ();
  55
  56             if (ret == (size_t)(-1))
  57               goto bad_input;
  58             if (ret == 0)
  59               {
  60                 src = NULL;
  61                 /* Here mbsinit (ps).  */
  62                 break;
  63               }
  64             if (!(USES_C32 && ret == (size_t)(-3)))
  65               {
  66                 src += ret;
  67                 srclen -= ret;
  68               }
  69           }
  70
  71         *srcp = src;
  72         return destptr - dest;
  73       }
  74     else
  75       {
  76         /* Ignore dest and len, don't store *srcp at the end, and
  77            don't clobber *ps.  */
  78         mbstate_t state = *ps;
  79         size_t totalcount = 0;
  80
  81         for (; srclen > 0; totalcount++)
  82           {
  83             size_t src_avail;
  84             size_t ret;
  85
  86             /* An optimized variant of
  87                src_avail = strnlen1 (src, MIN (srclen, MB_LEN_MAX));  */
  88             if (srclen == 1 || src[0] == '\0')
  89               src_avail = 1;
  90             else if (srclen == 2 || src[1] == '\0')
  91               src_avail = 2;
  92             else if (srclen == 3 || src[2] == '\0')
  93               src_avail = 3;
  94             else if (MB_LEN_MAX <= 4 || srclen == 4 || src[3] == '\0')
  95               src_avail = 4;
  96             else
  97               src_avail = 4 + strnlen1 (src + 4, MIN (srclen, MB_LEN_MAX) - 4);
  98
  99             /* Parse the next multibyte character.  */
 100             ret = MBRTOWC (NULL, src, src_avail, &state);
 101
 102             if (ret == (size_t)(-2))
 103               /* Encountered a multibyte character that extends past a '\0' byte
 104                  or that is longer than MB_LEN_MAX bytes.  Cannot happen.  */
 105               abort ();
 106
 107             if (ret == (size_t)(-1))
 108               goto bad_input2;
 109             if (ret == 0)
 110               {
 111                 /* Here mbsinit (&state).  */
 112                 break;
 113               }
 114             src += ret;
 115             srclen -= ret;
 116           }
 117
 118         return totalcount;
 119       }
 120
 121    bad_input:
 122     *srcp = src;
 123    bad_input2:
 124     errno = EILSEQ;
 125     return (size_t)(-1);
 126   }
 127 }