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 #define GREETING "Copyright 2002, 2003 Simon Josefsson\n"\
  32         "GNU Libidn comes with NO WARRANTY, to the extent permitted by law.\n"\
  33         "You may redistribute copies of GNU Libidn under the terms of\n"\
  34         "the GNU Lesser General Public License.  For more information\n"\
  35         "about these matters, see the file named COPYING.LIB.\n"\
  36
  37 int
  38 main (int argc, char *argv[])
  39 {
  40   struct gengetopt_args_info args_info;
  41   char readbuf[BUFSIZ];
  42   char *p, *r;
  43   unsigned long *q;
  44   int rc;
  45
  46   if (cmdline_parser(argc, argv, &args_info) != 0)
  47     return 1;
  48
  49   if ((args_info.stringprep_given ? 1 : 0) +
  50       (args_info.punycode_encode_given ? 1 : 0) +
  51       (args_info.punycode_decode_given ? 1 : 0) +
  52       (args_info.idna_to_ascii_given ? 1 : 0) +
  53       (args_info.idna_to_unicode_given ? 1 : 0) != 1)
  54     {
  55       fprintf(stderr, "%s: One of -s, -e, -d, -a or -u must be specified.\n",
  56               argv[0]);
  57       cmdline_parser_print_help();
  58       return 1;
  59     }
  60
  61   if (!args_info.quiet_given)
  62       fprintf(stderr, "%s %s\n" GREETING, PACKAGE, VERSION);
  63
  64   if (args_info.debug_given)
  65     fprintf(stderr, "system locale uses charset `%s'.\n",
  66             stringprep_locale_charset());
  67
  68   do
  69     {
  70       if (fgets(readbuf, BUFSIZ, stdin) == NULL)
  71         {
  72           sprintf(readbuf, "%s: fgets() failed: ", argv[0]);
  73           if (!feof(stdin))
  74             perror(readbuf);
  75           return 1;
  76         }
  77
  78       if (readbuf[strlen(readbuf)-1] == '\n')
  79         readbuf[strlen(readbuf)-1] = '\0';
  80
  81       if (args_info.stringprep_given)
  82         {
  83           p = stringprep_locale_to_utf8 (readbuf);
  84           if (!p)
  85             {
  86               fprintf(stderr, "%s: could not convert from %s to UTF-8.\n",
  87                       argv[0], stringprep_locale_charset());
  88               return 1;
  89             }
  90
  91           if (args_info.debug_given)
  92             {
  93               size_t i;
  94               for (i = 0; p[i]; i++)
  95                 fprintf(stderr, "input[%d] = U+%04x\n", i, p[i] & 0xFFFF);
  96             }
  97
  98           rc = stringprep_profile (p, &r,
  99                                    args_info.profile_given ?
 100                                    args_info.profile_arg :
 101                                    "Nameprep", 0);
 102           free(p);
 103           if (rc != STRINGPREP_OK)
 104             {
 105               fprintf(stderr,
 106                       "%s: stringprep_profile() failed with error %d.\n",
 107                       argv[0], rc);
 108               return 1;
 109             }
 110
 111           if (args_info.debug_given)
 112             {
 113               size_t i;
 114               for (i = 0; r[i]; i++)
 115                 fprintf(stderr, "output[%d] = U+%04x\n", i, r[i] & 0xFFFF);
 116             }
 117
 118           p = stringprep_utf8_to_locale (r);
 119           if (!p)
 120             {
 121               fprintf(stderr, "%s: could not convert from UTF-8 to %s.\n",
 122                       argv[0], stringprep_locale_charset());
 123               return 1;
 124             }
 125
 126           fprintf(stdout, "%s\n", p);
 127
 128           free(p);
 129         }
 130
 131       if (args_info.punycode_encode_given)
 132         {
 133           size_t len, len2;
 134
 135           p = stringprep_locale_to_utf8 (readbuf);
 136           if (!p)
 137             {
 138               fprintf(stderr, "%s: could not convert from %s to UTF-8.\n",
 139                       argv[0], stringprep_locale_charset());
 140               return 1;
 141             }
 142
 143           q = stringprep_utf8_to_ucs4 (p, -1, &len);
 144           free(p);
 145           if (!q)
 146             {
 147               fprintf(stderr, "%s: could not convert from UTF-8 to UCS-4.\n",
 148                       argv[0]);
 149               return 1;
 150             }
 151
 152           if (args_info.debug_given)
 153             {
 154               size_t i;
 155               for (i = 0; i < len; i++)
 156                 fprintf(stderr, "input[%d] = U+%04x\n", i, q[i] & 0xFFFF);
 157             }
 158
 159           len2 = BUFSIZ;
 160           rc = punycode_encode (len, q, NULL, &len2, readbuf);
 161           if (rc != PUNYCODE_SUCCESS)
 162             {
 163               fprintf(stderr, "%s: punycode_encode() failed with error %d.\n",
 164                       argv[0], rc);
 165               return 1;
 166             }
 167
 168           readbuf[len2] = '\0';
 169
 170           p = stringprep_utf8_to_locale (readbuf);
 171           if (!p)
 172             {
 173               fprintf(stderr, "%s: could not convert from UTF-8 to %s.\n",
 174                       argv[0], stringprep_locale_charset());
 175               return 1;
 176             }
 177
 178           fprintf(stdout, "%s\n", p);
 179
 180           free(p);
 181         }
 182
 183       if (args_info.punycode_decode_given)
 184         {
 185           size_t len, len2;
 186
 187           len = BUFSIZ;
 188           q = (unsigned long*) malloc(len * sizeof(q[0]));
 189           if (!q)
 190             {
 191               sprintf(readbuf, "%s: malloc() failed: ", argv[0]);
 192               perror(readbuf);
 193               return 1;
 194
 195             }
 196           rc = punycode_decode (strlen(readbuf), readbuf, &len, q, NULL);
 197           if (rc != PUNYCODE_SUCCESS)
 198             {
 199               free(q);
 200               fprintf(stderr, "%s: punycode_decode() failed with error %d.\n",
 201                       argv[0], rc);
 202               return 1;
 203             }
 204
 205           if (args_info.debug_given)
 206             {
 207               size_t i;
 208               for (i = 0; i < len; i++)
 209                 fprintf(stderr, "output[%d] = U+%04x\n", i, q[i] & 0xFFFF);
 210             }
 211
 212           q[len] = 0;
 213           p = stringprep_ucs4_to_utf8 (q, -1, NULL, NULL);
 214           if (!p)
 215             {
 216               free(q);
 217               fprintf(stderr, "%s: could not convert from UCS-4 to UTF-8.\n",
 218                       argv[0]);
 219               return 1;
 220             }
 221
 222           r = stringprep_utf8_to_locale (p);
 223           free(p);
 224           if (!r)
 225             {
 226               fprintf(stderr, "%s: could not convert from UTF-8 to %s.\n",
 227                       argv[0], stringprep_locale_charset());
 228               return 1;
 229             }
 230
 231           fprintf(stdout, "%s\n", r);
 232
 233           free(r);
 234         }
 235
 236       if (args_info.idna_to_ascii_given)
 237         {
 238           p = stringprep_locale_to_utf8 (readbuf);
 239           if (!p)
 240             {
 241               fprintf(stderr, "%s: could not convert from %s to UTF-8.\n",
 242                       argv[0], stringprep_locale_charset());
 243               return 1;
 244             }
 245
 246           if (args_info.debug_given)
 247             {
 248               size_t i;
 249               for (i = 0; p[i]; i++)
 250                 fprintf(stderr, "input[%d] = U+%04x\n", i, p[i] & 0xFFFF);
 251             }
 252
 253           rc = idna_to_ascii_from_utf8 (p, &r,
 254                                         args_info.allow_unassigned_given,
 255                                         args_info.usestd3asciirules_given);
 256           free(p);
 257           if (rc != IDNA_SUCCESS)
 258             {
 259               fprintf(stderr, "%s: idna_to_ascii_from_locale() failed "
 260                       "with error %d.\n", argv[0], rc);
 261               return 1;
 262             }
 263           fprintf(stdout, "%s\n", r);
 264
 265           free(r);
 266         }
 267
 268       if (args_info.idna_to_unicode_given)
 269         {
 270           p = stringprep_locale_to_utf8 (readbuf);
 271           if (!p)
 272             {
 273               fprintf(stderr, "%s: could not convert from %s to UTF-8.\n",
 274                       argv[0], stringprep_locale_charset());
 275               return 1;
 276             }
 277
 278           if (args_info.debug_given)
 279             {
 280               size_t i;
 281               for (i = 0; p[i]; i++)
 282                 fprintf(stderr, "input[%d] = U+%04x\n", i, p[i] & 0xFFFF);
 283             }
 284
 285           rc = idna_to_unicode_utf8_from_utf8
 286             (p, &r, args_info.allow_unassigned_given,
 287              args_info.usestd3asciirules_given);
 288           free(p);
 289           if (rc != IDNA_SUCCESS)
 290             {
 291               fprintf(stderr, "%s: idna_to_unicode_locale_from_locale() "
 292                       "failed with error %d.\n", argv[0], rc);
 293               return 1;
 294             }
 295
 296           if (args_info.debug_given)
 297             {
 298               size_t i;
 299               for (i = 0; r[i]; i++)
 300                 fprintf(stderr, "output[%d] = U+%04x\n", i, r[i] & 0xFFFF);
 301             }
 302
 303           fprintf(stdout, "%s\n", r);
 304
 305           free(r);
 306         }
 307
 308     }
 309   while (!feof(stdin) && !ferror(stdin));
 310
 311   return 0;
 312 }