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           q = stringprep_utf8_to_ucs4 (p, -1, NULL);
 247           if (!q)
 248             {
 249               free(p);
 250               fprintf(stderr, "%s: could not convert from UCS-4 to UTF-8.\n",
 251                       argv[0]);
 252               return 1;
 253             }
 254           if (args_info.debug_given)
 255             {
 256               size_t i;
 257               for (i = 0; q[i]; i++)
 258                 fprintf(stderr, "input[%d] = U+%04x\n", i, q[i] & 0xFFFF);
 259             }
 260
 261           rc = idna_to_ascii_from_ucs4 (q, &r,
 262                                         args_info.allow_unassigned_given,
 263                                         args_info.usestd3asciirules_given);
 264           free(q);
 265           if (rc != IDNA_SUCCESS)
 266             {
 267               fprintf(stderr, "%s: idna_to_ascii_from_locale() failed "
 268                       "with error %d.\n", argv[0], rc);
 269               return 1;
 270             }
 271           fprintf(stdout, "%s\n", r);
 272
 273           free(r);
 274         }
 275
 276       if (args_info.idna_to_unicode_given)
 277         {
 278           p = stringprep_locale_to_utf8 (readbuf);
 279           if (!p)
 280             {
 281               fprintf(stderr, "%s: could not convert from %s to UTF-8.\n",
 282                       argv[0], stringprep_locale_charset());
 283               return 1;
 284             }
 285
 286           q = stringprep_utf8_to_ucs4 (p, -1, NULL);
 287           if (!q)
 288             {
 289               free(p);
 290               fprintf(stderr, "%s: could not convert from UCS-4 to UTF-8.\n",
 291                       argv[0]);
 292               return 1;
 293             }
 294           if (args_info.debug_given)
 295             {
 296               size_t i;
 297               for (i = 0; q[i]; i++)
 298                 fprintf(stderr, "input[%d] = U+%04x\n", i, q[i] & 0xFFFF);
 299             }
 300           free(q);
 301
 302           rc = idna_to_unicode_ucs4_from_utf8
 303             (p, &q, args_info.allow_unassigned_given,
 304              args_info.usestd3asciirules_given);
 305           free(p);
 306           if (rc != IDNA_SUCCESS)
 307             {
 308               fprintf(stderr, "%s: idna_to_unicode_locale_from_locale() "
 309                       "failed with error %d.\n", argv[0], rc);
 310               return 1;
 311             }
 312
 313           if (args_info.debug_given)
 314             {
 315               size_t i;
 316               for (i = 0; q[i]; i++)
 317                 fprintf(stderr, "output[%d] = U+%04x\n", i, q[i] & 0xFFFF);
 318             }
 319
 320           p = stringprep_ucs4_to_utf8 (q, -1, NULL, NULL);
 321           if (!p)
 322             {
 323               free(q);
 324               fprintf(stderr, "%s: could not convert from UCS-4 to UTF-8.\n",
 325                       argv[0]);
 326               return 1;
 327             }
 328
 329           fprintf(stdout, "%s\n", p);
 330
 331           free(p);
 332         }
 333
 334     }
 335   while (!feof(stdin) && !ferror(stdin));
 336
 337   return 0;
 338 }