lib/wcsrtombs.c

   1 /* Convert wide string to string.
   2    Copyright (C) 2008, 2009, 2010 Free Software Foundation, Inc.
   3    Written by Bruno Haible <bruno@clisp.org>, 2008.
   4
   5    This program is free software: you can redistribute it and/or modify
   6    it under the terms of the GNU General Public License as published by
   7    the Free Software Foundation; either version 3 of the License, or
   8    (at your option) any later version.
   9
  10    This program 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 General Public License for more details.
  14
  15    You should have received a copy of the GNU General Public License
  16    along with this program.  If not, see <http://www.gnu.org/licenses/>.  */
  17
  18 #include <config.h>
  19
  20 /* Specification.  */
  21 #include <wchar.h>
  22
  23 extern mbstate_t _gl_wcsrtombs_state;
  24
  25 #if HAVE_WCSRTOMBS && !WCSRTOMBS_TERMINATION_BUG
  26 /* Override the system's wcsrtombs() function.  */
  27
  28 # undef wcsrtombs
  29
  30 size_t
  31 rpl_wcsrtombs (char *dest, const wchar_t **srcp, size_t len, mbstate_t *ps)
  32 {
  33   if (ps == NULL)
  34     ps = &_gl_wcsrtombs_state;
  35 # if WCSRTOMBS_NULL_ARG_BUG
  36   if (dest == NULL)
  37     {
  38       const wchar_t *temp_src = *srcp;
  39
  40       return wcsrtombs (NULL, &temp_src, len, ps);
  41     }
  42   else
  43 # endif
  44     return wcsrtombs (dest, srcp, len, ps);
  45 }
  46
  47 #else
  48 /* Implement wcsrtombs on top of wcrtomb().  */
  49
  50 # include <errno.h>
  51 # include <stdlib.h>
  52 # include <string.h>
  53
  54 size_t
  55 wcsrtombs (char *dest, const wchar_t **srcp, size_t len, mbstate_t *ps)
  56 {
  57   if (ps == NULL)
  58     ps = &_gl_wcsrtombs_state;
  59   {
  60     const wchar_t *src = *srcp;
  61     size_t cur_max = MB_CUR_MAX;
  62     char buf[64];
  63
  64     if (!(cur_max <= sizeof (buf)))
  65       abort ();
  66
  67     if (dest != NULL)
  68       {
  69         char *destptr = dest;
  70
  71         for (; len > 0; src++)
  72           {
  73             wchar_t wc = *src;
  74             size_t ret = wcrtomb (len >= cur_max ? destptr : buf, wc, ps);
  75
  76             if (ret == (size_t)(-1))
  77               goto bad_input;
  78             if (!(ret <= cur_max))
  79               abort ();
  80             if (len < ret)
  81               break;
  82             if (len < cur_max)
  83               memcpy (destptr, buf, ret);
  84             if (wc == 0)
  85               {
  86                 src = NULL;
  87                 /* Here mbsinit (ps).  */
  88                 break;
  89               }
  90             destptr += ret;
  91             len -= ret;
  92           }
  93         *srcp = src;
  94         return destptr - dest;
  95       }
  96     else
  97       {
  98         /* Ignore dest and len, don't store *srcp at the end, and
  99            don't clobber *ps.  */
 100         mbstate_t state = *ps;
 101         size_t totalcount = 0;
 102
 103         for (;; src++)
 104           {
 105             wchar_t wc = *src;
 106             size_t ret = wcrtomb (buf, wc, &state);
 107
 108             if (ret == (size_t)(-1))
 109               goto bad_input2;
 110             if (wc == 0)
 111               {
 112                 /* Here mbsinit (&state).  */
 113                 break;
 114               }
 115             totalcount += ret;
 116           }
 117         return totalcount;
 118       }
 119
 120    bad_input:
 121     *srcp = src;
 122    bad_input2:
 123     errno = EILSEQ;
 124     return (size_t)(-1);
 125   }
 126 }
 127
 128 #endif