tst_stringprep.c

   1 /* tst_stringprep.c     libstringprep self tests for stringprep_stringprep()
   2  * Copyright (C) 2002  Simon Josefsson
   3  *
   4  * This file is part of libstringprep.
   5  *
   6  * Libstringprep 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  * Libstringprep 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 libstringprep; if not, write to the Free Software
  18  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  19  *
  20  */
  21
  22 #ifdef HAVE_CONFIG_H
  23 #include "config.h"
  24 #endif
  25
  26 #include <stdio.h>
  27 #include <stdarg.h>
  28 #include <stdlib.h>
  29 #if HAVE_STRING_H
  30 # if !STDC_HEADERS && HAVE_MEMORY_H
  31 #  include <memory.h>
  32 # endif
  33 # include <string.h>
  34 #endif
  35 #if HAVE_STRINGS_H
  36 # include <strings.h>
  37 #endif
  38 #if HAVE_INTTYPES_H
  39 # include <inttypes.h>
  40 #else
  41 # if HAVE_STDINT_H
  42 #  include <stdint.h>
  43 # endif
  44 #endif
  45 #if HAVE_SYS_TYPES_H
  46 #include <sys/types.h>
  47 #endif
  48
  49 #include <stringprep.h>
  50 #include <stringprep_generic.h>
  51
  52 static int debug = 0;
  53 static int error_count = 0;
  54 static int break_on_error = 0;
  55
  56 static void
  57 fail (const char *format, ...)
  58 {
  59   va_list arg_ptr;
  60
  61   va_start (arg_ptr, format);
  62   vfprintf (stderr, format, arg_ptr);
  63   va_end (arg_ptr);
  64   error_count++;
  65   if (break_on_error)
  66     exit (1);
  67 }
  68
  69 static void
  70 escapeprint (char *str, int len)
  71 {
  72   int i;
  73
  74   printf ("\t ;; `");
  75   for (i = 0; i < len; i++)
  76     {
  77       str[i] = str[i] & 0xFF;
  78       if ((str[i] >= 'A' && str[i] <= 'Z') ||
  79           (str[i] >= 'a' && str[i] <= 'z') ||
  80           (str[i] >= '0' && str[i] <= '9') || str[i] == '.')
  81         printf ("%c", str[i]);
  82       else
  83         printf ("\\x%02x", str[i]);
  84     }
  85   printf ("' (length %d bytes)\n", len);
  86 }
  87
  88 static void
  89 hexprint (char *str, int len)
  90 {
  91   int i;
  92
  93   printf ("\t ;; ");
  94   for (i = 0; i < len; i++)
  95     {
  96       str[i] = str[i] & 0xFF;
  97       printf ("%02x ", str[i]);
  98       if ((i + 1) % 8 == 0)
  99         printf (" ");
 100       if ((i + 1) % 16 == 0 && i + 1 < len)
 101         printf ("\n\t ;; ");
 102     }
 103 }
 104
 105 static void
 106 binprint (char *str, int len)
 107 {
 108   int i;
 109
 110   printf ("\t ;; ");
 111   for (i = 0; i < len; i++)
 112     {
 113       str[i] = str[i] & 0xFF;
 114       printf ("%d%d%d%d%d%d%d%d ",
 115               str[i] & 0x80 ? 1 : 0,
 116               str[i] & 0x40 ? 1 : 0,
 117               str[i] & 0x20 ? 1 : 0,
 118               str[i] & 0x10 ? 1 : 0,
 119               str[i] & 0x08 ? 1 : 0,
 120               str[i] & 0x04 ? 1 : 0,
 121               str[i] & 0x02 ? 1 : 0, str[i] & 0x01 ? 1 : 0);
 122       if ((i + 1) % 3 == 0)
 123         printf (" ");
 124       if ((i + 1) % 6 == 0 && i + 1 < len)
 125         printf ("\n\t ;; ");
 126     }
 127 }
 128
 129 struct stringprep
 130 {
 131   char *in;
 132   int flags;
 133   char *out;
 134   stringprep_profile *profile;
 135 }
 136 strprep[] =
 137 {
 138   /* nothing */
 139   { "foo\xC2\xAD" "bar", 0, "foobar",  stringprep_generic },
 140   /* case_nfkc + normalization: */
 141   { "\xC2\xB5", 0, "\xCE\xBC",  stringprep_generic },
 142   /* case_nonfkc: */
 143   { "\xC2\xB5", STRINGPREP_NO_NFKC, "\xCE\xBC",  stringprep_generic },
 144   { "\xC2\xAA", 0, "\x61", stringprep_generic }
 145 };
 146
 147 int
 148 main (int argc, char *argv[])
 149 {
 150   char *p;
 151   int rc, i;
 152
 153   do
 154     if (strcmp (argv[argc - 1], "-v") == 0 ||
 155         strcmp (argv[argc - 1], "--verbose") == 0)
 156       debug = 1;
 157     else if (strcmp (argv[argc - 1], "-b") == 0 ||
 158              strcmp (argv[argc - 1], "--break-on-error") == 0)
 159       break_on_error = 1;
 160     else if (strcmp (argv[argc - 1], "-h") == 0 ||
 161              strcmp (argv[argc - 1], "-?") == 0 ||
 162              strcmp (argv[argc - 1], "--help") == 0)
 163       {
 164         printf ("Usage: %s [-vbh?] [--verbose] [--break-on-error] [--help]\n",
 165                 argv[0]);
 166         return 1;
 167       }
 168   while (argc-- > 1);
 169
 170   p = malloc(BUFSIZ);
 171   if (p == NULL)
 172     fail ("malloc() returned NULL\n");
 173
 174   for (i = 0; i < sizeof (strprep) / sizeof (strprep[0]); i++)
 175     {
 176       if (debug)
 177         printf ("STRINGPREP entry %d\n", i);
 178
 179       strcpy(p, strprep[i].in);
 180
 181       rc = stringprep (p, BUFSIZ, strprep[i].flags, strprep[i].profile);
 182       if (rc != STRINGPREP_OK)
 183         {
 184           fail ("stringprep() entry %d failed: %d\n", i, rc);
 185           continue;
 186         }
 187
 188       if (debug)
 189         {
 190           printf ("in:\n");
 191           escapeprint (strprep[i].in, strlen (strprep[i].in));
 192           hexprint (strprep[i].in, strlen (strprep[i].in));
 193           puts ("");
 194           binprint (strprep[i].in, strlen (strprep[i].in));
 195           puts ("");
 196
 197           printf ("out:\n");
 198           escapeprint (p, strlen (p));
 199           hexprint (p, strlen (p));
 200           puts ("");
 201           binprint (p, strlen (p));
 202           puts ("");
 203
 204           printf ("expected out:\n");
 205           escapeprint (strprep[i].out, strlen (strprep[i].out));
 206           hexprint (strprep[i].out, strlen (strprep[i].out));
 207           puts ("");
 208           binprint (strprep[i].out, strlen (strprep[i].out));
 209           puts ("");
 210         }
 211
 212       if (strlen (strprep[i].out) != strlen (p) ||
 213           memcmp (strprep[i].out, p, strlen (p)) != 0)
 214         {
 215           fail ("stringprep() entry %d failed\n", i);
 216           if (debug)
 217             printf ("ERROR\n");
 218         }
 219       else if (debug)
 220         printf ("OK\n");
 221     }
 222
 223   free(p);
 224
 225   if (debug)
 226     printf ("Stringprep self tests done with %d errors\n", error_count);
 227
 228   return error_count ? 1 : 0;
 229 }