tests/test-mbsstr1.c

   1 /* Test of searching in a string.
   2    Copyright (C) 2007-2008 Free Software Foundation, Inc.
   3
   4    This program is free software: you can redistribute it and/or modify
   5    it under the terms of the GNU General Public License as published by
   6    the Free Software Foundation; either version 3 of the License, or
   7    (at your option) any later version.
   8
   9    This program is distributed in the hope that it will be useful,
  10    but WITHOUT ANY WARRANTY; without even the implied warranty of
  11    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12    GNU General Public License for more details.
  13
  14    You should have received a copy of the GNU General Public License
  15    along with this program.  If not, see <http://www.gnu.org/licenses/>.  */
  16
  17 /* Written by Bruno Haible <bruno@clisp.org>, 2007.  */
  18
  19 #include <config.h>
  20
  21 #include <string.h>
  22
  23 #include <stdio.h>
  24 #include <stdlib.h>
  25
  26 #define ASSERT(expr) \
  27   do                                                                         \
  28     {                                                                        \
  29       if (!(expr))                                                           \
  30         {                                                                    \
  31           fprintf (stderr, "%s:%d: assertion failed\n", __FILE__, __LINE__); \
  32           fflush (stderr);                                                   \
  33           abort ();                                                          \
  34         }                                                                    \
  35     }                                                                        \
  36   while (0)
  37
  38 int
  39 main ()
  40 {
  41   /* This test is executed in the C locale.  */
  42
  43   {
  44     const char input[] = "foo";
  45     const char *result = mbsstr (input, "");
  46     ASSERT (result == input);
  47   }
  48
  49   {
  50     const char input[] = "foo";
  51     const char *result = mbsstr (input, "o");
  52     ASSERT (result == input + 1);
  53   }
  54
  55   {
  56     const char input[] = "ABC ABCDAB ABCDABCDABDE";
  57     const char *result = mbsstr (input, "ABCDABD");
  58     ASSERT (result == input + 15);
  59   }
  60
  61   {
  62     const char input[] = "ABC ABCDAB ABCDABCDABDE";
  63     const char *result = mbsstr (input, "ABCDABE");
  64     ASSERT (result == NULL);
  65   }
  66
  67   /* Check that a very long haystack is handled quickly if the needle is
  68      short and occurs near the beginning.  */
  69   {
  70     size_t repeat = 10000;
  71     size_t m = 1000000;
  72     char *needle =
  73       "AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA"
  74       "AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA";
  75     char *haystack = (char *) malloc (m + 1);
  76     if (haystack != NULL)
  77       {
  78         memset (haystack, 'A', m);
  79         haystack[0] = 'B';
  80         haystack[m] = '\0';
  81
  82         for (; repeat > 0; repeat--)
  83           {
  84             ASSERT (mbsstr (haystack, needle) == haystack + 1);
  85           }
  86
  87         free (haystack);
  88       }
  89   }
  90
  91   /* Check that a very long needle is discarded quickly if the haystack is
  92      short.  */
  93   {
  94     size_t repeat = 10000;
  95     size_t m = 1000000;
  96     char *haystack =
  97       "AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA"
  98       "ABABABABABABABABABABABABABABABABABABABABABABABABABABABABABABABABABAB";
  99     char *needle = (char *) malloc (m + 1);
 100     if (needle != NULL)
 101       {
 102         memset (needle, 'A', m);
 103         needle[m] = '\0';
 104
 105         for (; repeat > 0; repeat--)
 106           {
 107             ASSERT (mbsstr (haystack, needle) == NULL);
 108           }
 109
 110         free (needle);
 111       }
 112   }
 113
 114   /* Check that the asymptotic worst-case complexity is not quadratic.  */
 115   {
 116     size_t m = 1000000;
 117     char *haystack = (char *) malloc (2 * m + 2);
 118     char *needle = (char *) malloc (m + 2);
 119     if (haystack != NULL && needle != NULL)
 120       {
 121         const char *result;
 122
 123         memset (haystack, 'A', 2 * m);
 124         haystack[2 * m] = 'B';
 125         haystack[2 * m + 1] = '\0';
 126
 127         memset (needle, 'A', m);
 128         needle[m] = 'B';
 129         needle[m + 1] = '\0';
 130
 131         result = mbsstr (haystack, needle);
 132         ASSERT (result == haystack + m);
 133       }
 134     free (needle);
 135     free (haystack);
 136   }
 137
 138   return 0;
 139 }