idn.c

   1 /* idn.c        Command line interface to the library
   2  * Copyright (C) 2003  Simon Josefsson
   3  *
   4  * This file is part of GNU Libidn.
   5  *
   6  * GNU Libidn is free software; you can redistribute it and/or modify
   7  * it under the terms of the GNU General Public License as published by
   8  * the Free Software Foundation; either version 2 of the License, or
   9  * (at your option) any later version.
  10  *
  11  * GNU Libidn is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14  * GNU General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU General Public License
  17  * along with GNU Libidn; if not, write to the Free Software
  18  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  19  *
  20  */
  21
  22 #include <stdio.h>
  23 #include <unistd.h>
  24
  25 #include <stringprep.h>
  26 #include <punycode.h>
  27 #include <idna.h>
  28
  29 #include "idn_cmd.h"
  30
  31 int
  32 main (int argc, char *argv[])
  33 {
  34   struct gengetopt_args_info args_info;
  35   char readbuf[BUFSIZ];
  36   char *p, *r;
  37   unsigned long *q;
  38   int rc;
  39
  40   if (cmdline_parser(argc, argv, &args_info) != 0)
  41     return 1;
  42
  43   if ((args_info.punycode_encode_given ? 1 : 0) +
  44       (args_info.punycode_decode_given ? 1 : 0) +
  45       (args_info.idna_to_ascii_given ? 1 : 0) +
  46       (args_info.idna_to_unicode_given ? 1 : 0) != 1)
  47     {
  48       fprintf(stderr, "%s: One of -e, -d, -a or -u must be specified.\n",
  49               argv[0]);
  50       return 1;
  51     }
  52
  53   if (args_info.debug_given)
  54     {
  55       fprintf(stderr, "system locale uses charset `%s'.\n",
  56               stringprep_locale_charset());
  57     }
  58
  59   if (!args_info.quiet_given)
  60     {
  61       fprintf(stderr, "%s %s\n", PACKAGE, VERSION);
  62       fprintf(stderr, "Copyright 2003 Simon Josefsson\n"
  63               "GNU Libidn comes with NO WARRANTY, to the extent permitted "
  64               "by law.\n"
  65               "You may redistribute copies of GNU Libidn under the terms of\n"
  66               "the GNU Lesser General Public License.  For more information\n"
  67               "about these matters, see the file named COPYING.LIB.\n");
  68     }
  69
  70   do
  71     {
  72       if (fgets(readbuf, BUFSIZ, stdin) == NULL)
  73         {
  74           sprintf(readbuf, "%s: fgets() failed: ", argv[0]);
  75           if (!feof(stdin))
  76             perror(readbuf);
  77           return 1;
  78         }
  79
  80       if (readbuf[strlen(readbuf)-1] == '\n')
  81         readbuf[strlen(readbuf)-1] = '\0';
  82
  83       if (args_info.punycode_encode_given)
  84         {
  85           size_t len, len2;
  86
  87           p = stringprep_locale_to_utf8 (readbuf);
  88           if (!p)
  89             {
  90               fprintf(stderr, "%s: could not convert from %s to UTF-8.\n",
  91                       argv[0], stringprep_locale_charset());
  92               return 1;
  93             }
  94
  95           q = stringprep_utf8_to_ucs4 (p, -1, &len);
  96           free(p);
  97           if (!q)
  98             {
  99               fprintf(stderr, "%s: could not convert from UTF-8 to UCS-4.\n",
 100                       argv[0]);
 101               return 1;
 102             }
 103
 104           if (args_info.debug_given)
 105             {
 106               size_t i;
 107               for (i = 0; i < len; i++)
 108                 fprintf(stderr, "input[%d] = U+%04x\n", i, q[i]);
 109             }
 110
 111           len2 = BUFSIZ;
 112           rc = punycode_encode (len, q, NULL, &len2, readbuf);
 113           if (rc != PUNYCODE_SUCCESS)
 114             {
 115               fprintf(stderr, "%s: punycode_encode() failed with error %d.\n",
 116                       argv[0], rc);
 117               return 1;
 118             }
 119
 120           readbuf[len2] = '\0';
 121
 122           p = stringprep_utf8_to_locale (readbuf);
 123           if (!p)
 124             {
 125               fprintf(stderr, "%s: could not convert from UTF-8 to %s.\n",
 126                       argv[0], stringprep_locale_charset());
 127               return 1;
 128             }
 129
 130           fprintf(stdout, "%s\n", p);
 131
 132           free(p);
 133         }
 134
 135       if (args_info.punycode_decode_given)
 136         {
 137           size_t len, len2;
 138
 139           len = BUFSIZ;
 140           q = (unsigned long*) malloc(len * sizeof(q[0]));
 141           if (!q)
 142             {
 143               sprintf(readbuf, "%s: malloc() failed: ", argv[0]);
 144               perror(readbuf);
 145               return 1;
 146
 147             }
 148           rc = punycode_decode (strlen(readbuf), readbuf, &len, q, NULL);
 149           if (rc != PUNYCODE_SUCCESS)
 150             {
 151               free(q);
 152               fprintf(stderr, "%s: punycode_decode() failed with error %d.\n",
 153                       argv[0], rc);
 154               return 1;
 155             }
 156
 157           if (args_info.debug_given)
 158             {
 159               size_t i;
 160               for (i = 0; i < len; i++)
 161                 fprintf(stderr, "output[%d] = U+%04x\n", i, q[i]);
 162             }
 163
 164           q[len] = 0;
 165           p = stringprep_ucs4_to_utf8 (q, -1, NULL, NULL);
 166           if (!p)
 167             {
 168               free(q);
 169               fprintf(stderr, "%s: could not convert from UCS-4 to UTF-8.\n",
 170                       argv[0]);
 171               return 1;
 172             }
 173
 174           r = stringprep_utf8_to_locale (p);
 175           free(p);
 176           if (!r)
 177             {
 178               fprintf(stderr, "%s: could not convert from UTF-8 to %s.\n",
 179                       argv[0], stringprep_locale_charset());
 180               return 1;
 181             }
 182
 183           fprintf(stdout, "%s\n", r);
 184
 185           free(r);
 186         }
 187
 188       if (args_info.idna_to_ascii_given)
 189         {
 190           rc = idna_to_ascii_from_locale (readbuf, &p,
 191                                           args_info.allow_unassigned_given,
 192                                           args_info.usestd3asciirules_given);
 193           if (rc != IDNA_SUCCESS)
 194             {
 195               fprintf(stderr, "%s: idna_to_ascii_from_locale() failed "
 196                       "with error %d.\n", argv[0], rc);
 197               return 1;
 198             }
 199           fprintf(stdout, "%s\n", p);
 200
 201           free(p);
 202         }
 203
 204       if (args_info.idna_to_unicode_given)
 205         {
 206           rc = idna_to_unicode_locale_from_locale
 207             (readbuf, &p, args_info.allow_unassigned_given,
 208              args_info.usestd3asciirules_given);
 209           if (rc != IDNA_SUCCESS)
 210             {
 211               fprintf(stderr, "%s: idna_to_unicode_locale_from_locale() "
 212                       "failed with error %d.\n", argv[0], rc);
 213               return 1;
 214             }
 215           fprintf(stdout, "%s\n", p);
 216
 217           free(p);
 218         }
 219
 220     }
 221   while (!feof(stdin) && !ferror(stdin));
 222
 223   return 0;
 224 }