source3/torture/smbiconv.c

   1 /*
   2    Unix SMB/CIFS implementation.
   3    Charset module tester
   4
   5    Copyright (C) Jelmer Vernooij 2003
   6    Based on iconv/icon_prog.c from the GNU C Library,
   7       Contributed by Ulrich Drepper <drepper@cygnus.com>, 1998.
   8
   9    This program is free software; you can redistribute it and/or modify
  10    it under the terms of the GNU General Public License as published by
  11    the Free Software Foundation; either version 3 of the License, or
  12    (at your option) any later version.
  13
  14    This program is distributed in the hope that it will be useful,
  15    but WITHOUT ANY WARRANTY; without even the implied warranty of
  16    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  17    GNU General Public License for more details.
  18
  19    You should have received a copy of the GNU General Public License
  20    along with this program.  If not, see <http://www.gnu.org/licenses/>.
  21 */
  22
  23 #include "includes.h"
  24 #undef realloc
  25
  26 static int
  27 process_block (smb_iconv_t cd, const char *addr, size_t len, FILE *output)
  28 {
  29 #define OUTBUF_SIZE     32768
  30   const char *start = addr;
  31   char outbuf[OUTBUF_SIZE];
  32   char *outptr;
  33   size_t outlen;
  34   size_t n;
  35
  36   while (len > 0)
  37     {
  38       outptr = outbuf;
  39       outlen = OUTBUF_SIZE;
  40       n = smb_iconv (cd,  &addr, &len, &outptr, &outlen);
  41
  42       if (outptr != outbuf)
  43         {
  44           /* We have something to write out.  */
  45           int errno_save = errno;
  46
  47           if (fwrite (outbuf, 1, outptr - outbuf, output)
  48               < (size_t) (outptr - outbuf)
  49               || ferror (output))
  50             {
  51               /* Error occurred while printing the result.  */
  52               DEBUG (0, ("conversion stopped due to problem in writing the output"));
  53               return -1;
  54             }
  55
  56           errno = errno_save;
  57         }
  58
  59       if (errno != E2BIG)
  60         {
  61           /* iconv() ran into a problem.  */
  62           switch (errno)
  63             {
  64             case EILSEQ:
  65               DEBUG(0,("illegal input sequence at position %ld",
  66                      (long) (addr - start)));
  67               break;
  68             case EINVAL:
  69               DEBUG(0, ("\
  70 incomplete character or shift sequence at end of buffer"));
  71               break;
  72             case EBADF:
  73               DEBUG(0, ("internal error (illegal descriptor)"));
  74               break;
  75             default:
  76               DEBUG(0, ("unknown iconv() error %d", errno));
  77               break;
  78             }
  79
  80           return -1;
  81         }
  82     }
  83
  84   return 0;
  85 }
  86
  87
  88 static int
  89 process_fd (smb_iconv_t cd, int fd, FILE *output)
  90 {
  91   /* we have a problem with reading from a descriptor since we must not
  92      provide the iconv() function an incomplete character or shift
  93      sequence at the end of the buffer.  Since we have to deal with
  94      arbitrary encodings we must read the whole text in a buffer and
  95      process it in one step.  */
  96   static char *inbuf = NULL;
  97   static size_t maxlen = 0;
  98   char *inptr = NULL;
  99   size_t actlen = 0;
 100
 101   while (actlen < maxlen)
 102     {
 103       ssize_t n = read (fd, inptr, maxlen - actlen);
 104
 105       if (n == 0)
 106         /* No more text to read.  */
 107         break;
 108
 109       if (n == -1)
 110         {
 111           /* Error while reading.  */
 112           DEBUG(0, ("error while reading the input"));
 113           return -1;
 114         }
 115
 116       inptr += n;
 117       actlen += n;
 118     }
 119
 120   if (actlen == maxlen)
 121     while (1)
 122       {
 123         ssize_t n;
 124         char *new_inbuf;
 125
 126         /* Increase the buffer.  */
 127         new_inbuf = (char *) realloc (inbuf, maxlen + 32768);
 128         if (new_inbuf == NULL)
 129           {
 130             DEBUG(0, ("unable to allocate buffer for input"));
 131             return -1;
 132           }
 133         inbuf = new_inbuf;
 134         maxlen += 32768;
 135         inptr = inbuf + actlen;
 136
 137         do
 138           {
 139             n = read (fd, inptr, maxlen - actlen);
 140
 141             if (n == 0)
 142               /* No more text to read.  */
 143               break;
 144
 145             if (n == -1)
 146               {
 147                 /* Error while reading.  */
 148                 DEBUG(0, ("error while reading the input"));
 149                 return -1;
 150               }
 151
 152             inptr += n;
 153             actlen += n;
 154           }
 155         while (actlen < maxlen);
 156
 157         if (n == 0)
 158           /* Break again so we leave both loops.  */
 159           break;
 160       }
 161
 162   /* Now we have all the input in the buffer.  Process it in one run.  */
 163   return process_block (cd, inbuf, actlen, output);
 164 }
 165
 166 /* Main function */
 167
 168 int main(int argc, char *argv[])
 169 {
 170         const char *file = NULL;
 171         const char *from = "";
 172         const char *to = "";
 173         char *output = NULL;
 174         const char *preload_modules[] = {NULL, NULL};
 175         FILE *out = stdout;
 176         int fd;
 177         smb_iconv_t cd;
 178
 179         /* make sure the vars that get altered (4th field) are in
 180            a fixed location or certain compilers complain */
 181         poptContext pc;
 182         struct poptOption long_options[] = {
 183                 POPT_AUTOHELP
 184                 { "from-code", 'f', POPT_ARG_STRING, &from, 0, "Encoding of original text" },
 185                 { "to-code", 't', POPT_ARG_STRING, &to, 0, "Encoding for output" },
 186                 { "output", 'o', POPT_ARG_STRING, &output, 0, "Write output to this file" },
 187                 { "preload-modules", 'p', POPT_ARG_STRING, &preload_modules[0], 0, "Modules to load" },
 188                 POPT_COMMON_SAMBA
 189                 POPT_TABLEEND
 190         };
 191
 192         setlinebuf(stdout);
 193
 194         pc = poptGetContext("smbiconv", argc, (const char **) argv,
 195                             long_options, 0);
 196
 197         poptSetOtherOptionHelp(pc, "[FILE] ...");
 198
 199         while(poptGetNextOpt(pc) != -1);
 200
 201         /* the following functions are part of the Samba debugging
 202            facilities.  See lib/debug.c */
 203         setup_logging("smbiconv", True);
 204
 205         if (preload_modules[0]) smb_load_modules(preload_modules);
 206
 207         if(output) {
 208                 out = fopen(output, "w");
 209
 210                 if(!out) {
 211                         DEBUG(0, ("Can't open output file '%s': %s, exiting...\n", output, strerror(errno)));
 212                         return 1;
 213                 }
 214         }
 215
 216         cd = smb_iconv_open(to, from);
 217         if (cd == (smb_iconv_t)-1) {
 218                 DEBUG(0,("unable to find from or to encoding, exiting...\n"));
 219                 if (out != stdout) fclose(out);
 220                 return 1;
 221         }
 222
 223         while((file = poptGetArg(pc))) {
 224                 if(strcmp(file, "-") == 0) fd = 0;
 225                 else {
 226                         fd = open(file, O_RDONLY);
 227
 228                         if(!fd) {
 229                                 DEBUG(0, ("Can't open input file '%s': %s, ignoring...\n", file, strerror(errno)));
 230                                 continue;
 231                         }
 232                 }
 233
 234                 /* Loop thru all arguments */
 235                 process_fd(cd, fd, out);
 236
 237                 close(fd);
 238         }
 239         poptFreeContext(pc);
 240
 241         fclose(out);
 242
 243         return 0;
 244 }