lib/iconvme.c

   1 /* Recode strings between character sets, using iconv.
   2    Copyright (C) 2002, 2003, 2004, 2005 Free Software Foundation, Inc.
   3
   4    This program is free software; you can redistribute it and/or modify
   5    it under the terms of the GNU General Public License as published by
   6    the Free Software Foundation; either version 2, or (at your option)
   7    any later version.
   8
   9    This program is distributed in the hope that it will be useful,
  10    but WITHOUT ANY WARRANTY; without even the implied warranty of
  11    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12    GNU General Public License for more details.
  13
  14    You should have received a copy of the GNU General Public License along
  15    with this program; if not, write to the Free Software Foundation,
  16    Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.  */
  17
  18 #ifdef HAVE_CONFIG_H
  19 # include <config.h>
  20 #endif
  21
  22 /* Get prototype. */
  23 #include "iconvme.h"
  24
  25 /* Get malloc. */
  26 #include <stdlib.h>
  27
  28 /* Get strcmp. */
  29 #include <string.h>
  30
  31 /* Get errno. */
  32 #include <errno.h>
  33
  34 #ifdef _LIBC
  35 # define HAVE_ICONV 1
  36 #else
  37 /* Get strdup. */
  38 # include "strdup.h"
  39 #endif
  40
  41 #if HAVE_ICONV
  42 /* Get iconv etc. */
  43 # include <iconv.h>
  44 /* Get MB_LEN_MAX. */
  45 # include <limits.h>
  46 #endif
  47
  48 /* Convert a zero-terminated string STR from the FROM_CODSET code set
  49    to the TO_CODESET code set.  The returned string is allocated using
  50    malloc, and must be dellocated by the caller using free.  On
  51    failure, NULL is returned and errno holds the error reason.  Note
  52    that if TO_CODESET uses \0 for anything but to terminate the
  53    string, the caller of this function may have difficulties finding
  54    out the length of the output string.  */
  55 char *
  56 iconv_string (const char *str, const char *from_codeset,
  57               const char *to_codeset)
  58 {
  59   char *dest = NULL;
  60 #if HAVE_ICONV
  61   iconv_t cd;
  62   char *outp;
  63   char *p = (char *) str;
  64   size_t inbytes_remaining = strlen (p);
  65   /* Guess the maximum length the output string can have.  */
  66   size_t outbuf_size = (inbytes_remaining + 1) * MB_LEN_MAX;
  67   size_t outbytes_remaining = outbuf_size - 1; /* -1 for NUL */
  68   size_t err;
  69   int have_error = 0;
  70 #endif
  71
  72   if (strcmp (to_codeset, from_codeset) == 0)
  73     return strdup (str);
  74
  75 #if HAVE_ICONV
  76   cd = iconv_open (to_codeset, from_codeset);
  77   if (cd == (iconv_t) -1)
  78     return NULL;
  79
  80   outp = dest = (char *) malloc (outbuf_size);
  81   if (dest == NULL)
  82     goto out;
  83
  84 again:
  85   err = iconv (cd, &p, &inbytes_remaining, &outp, &outbytes_remaining);
  86
  87   if (err == (size_t) - 1)
  88     {
  89       switch (errno)
  90         {
  91         case EINVAL:
  92           /* Incomplete text, do not report an error */
  93           break;
  94
  95         case E2BIG:
  96           {
  97             size_t used = outp - dest;
  98             size_t newsize = outbuf_size * 2;
  99             char *newdest;
 100
 101             if (newsize <= outbuf_size)
 102               {
 103                 errno = ENOMEM;
 104                 have_error = 1;
 105                 goto out;
 106               }
 107             newdest = (char *) realloc (dest, newsize);
 108             if (newdest == NULL)
 109               {
 110                 have_error = 1;
 111                 goto out;
 112               }
 113             dest = newdest;
 114             outbuf_size = newsize;
 115
 116             outp = dest + used;
 117             outbytes_remaining = outbuf_size - used - 1;        /* -1 for NUL */
 118
 119             goto again;
 120           }
 121           break;
 122
 123         case EILSEQ:
 124           have_error = 1;
 125           break;
 126
 127         default:
 128           have_error = 1;
 129           break;
 130         }
 131     }
 132
 133   *outp = '\0';
 134
 135 out:
 136   {
 137     int save_errno = errno;
 138
 139     if (iconv_close (cd) < 0 && !have_error)
 140       {
 141         /* If we didn't have a real error before, make sure we restore
 142            the iconv_close error below. */
 143         save_errno = errno;
 144         have_error = 1;
 145       }
 146
 147     if (have_error && dest)
 148       {
 149         free (dest);
 150         dest = NULL;
 151         errno = save_errno;
 152       }
 153   }
 154 #else
 155   errno = ENOSYS;
 156 #endif
 157
 158   return dest;
 159 }