Unconditionally include <config.h> in unit tests.
[gnulib.git] / tests / test-mbsstr1.c
1 /* Test of searching in a string.
2    Copyright (C) 2007 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 2, or (at your option)
7    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, write to the Free Software Foundation,
16    Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.  */
17
18 /* Written by Bruno Haible <bruno@clisp.org>, 2007.  */
19
20 #include <config.h>
21
22 #include <string.h>
23
24 #include <stdio.h>
25 #include <stdlib.h>
26
27 #define ASSERT(expr) \
28   do                                                                         \
29     {                                                                        \
30       if (!(expr))                                                           \
31         {                                                                    \
32           fprintf (stderr, "%s:%d: assertion failed\n", __FILE__, __LINE__); \
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     if (needle != NULL)
135       free (needle);
136     if (haystack != NULL)
137       free (haystack);
138   }
139
140   return 0;
141 }