Flush the standard error stream before aborting.
[gnulib.git] / tests / test-mbsstr2.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 <locale.h>
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           fflush (stderr);                                                   \
34           abort ();                                                          \
35         }                                                                    \
36     }                                                                        \
37   while (0)
38
39 int
40 main ()
41 {
42   /* configure should already have checked that the locale is supported.  */
43   if (setlocale (LC_ALL, "") == NULL)
44     return 1;
45
46   {
47     const char input[] = "f\303\266\303\266";
48     const char *result = mbsstr (input, "");
49     ASSERT (result == input);
50   }
51
52   {
53     const char input[] = "f\303\266\303\266";
54     const char *result = mbsstr (input, "\303\266");
55     ASSERT (result == input + 1);
56   }
57
58   {
59     const char input[] = "f\303\266\303\266";
60     const char *result = mbsstr (input, "\266\303");
61     ASSERT (result == NULL);
62   }
63
64   {
65     const char input[] = "\303\204BC \303\204BCD\303\204B \303\204BCD\303\204BCD\303\204BDE"; /* "ÄBC ÄBCDÄB ÄBCDÄBCDÄBDE" */
66     const char *result = mbsstr (input, "\303\204BCD\303\204BD"); /* "ÄBCDÄBD" */
67     ASSERT (result == input + 19);
68   }
69
70   {
71     const char input[] = "\303\204BC \303\204BCD\303\204B \303\204BCD\303\204BCD\303\204BDE"; /* "ÄBC ÄBCDÄB ÄBCDÄBCDÄBDE" */
72     const char *result = mbsstr (input, "\303\204BCD\303\204BE"); /* "ÄBCDÄBE" */
73     ASSERT (result == NULL);
74   }
75
76   /* Check that a very long haystack is handled quickly if the needle is
77      short and occurs near the beginning.  */
78   {
79     size_t repeat = 10000;
80     size_t m = 1000000;
81     char *needle =
82       "AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA"
83       "AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA";
84     char *haystack = (char *) malloc (m + 1);
85     if (haystack != NULL)
86       {
87         memset (haystack, 'A', m);
88         haystack[0] = '\303'; haystack[1] = '\204';
89         haystack[m] = '\0';
90
91         for (; repeat > 0; repeat--)
92           {
93             ASSERT (mbsstr (haystack, needle) == haystack + 2);
94           }
95
96         free (haystack);
97       }
98   }
99
100   /* Check that a very long needle is discarded quickly if the haystack is
101      short.  */
102   {
103     size_t repeat = 10000;
104     size_t m = 1000000;
105     char *haystack =
106       "AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA"
107       "A\303\207A\303\207A\303\207A\303\207A\303\207A\303\207A\303\207"
108       "A\303\207A\303\207A\303\207A\303\207A\303\207A\303\207A\303\207"
109       "A\303\207A\303\207A\303\207A\303\207A\303\207A\303\207A\303\207"
110       "A\303\207A\303\207A\303\207A\303\207A\303\207A\303\207A\303\207"
111       "A\303\207A\303\207A\303\207A\303\207A\303\207A\303\207";
112     char *needle = (char *) malloc (m + 1);
113     if (needle != NULL)
114       {
115         memset (needle, 'A', m);
116         needle[m] = '\0';
117
118         for (; repeat > 0; repeat--)
119           {
120             ASSERT (mbsstr (haystack, needle) == NULL);
121           }
122
123         free (needle);
124       }
125   }
126
127   /* Check that the asymptotic worst-case complexity is not quadratic.  */
128   {
129     size_t m = 1000000;
130     char *haystack = (char *) malloc (2 * m + 3);
131     char *needle = (char *) malloc (m + 3);
132     if (haystack != NULL && needle != NULL)
133       {
134         const char *result;
135
136         memset (haystack, 'A', 2 * m);
137         haystack[2 * m] = '\303'; haystack[2 * m + 1] = '\207';
138         haystack[2 * m + 2] = '\0';
139
140         memset (needle, 'A', m);
141         needle[m] = '\303'; needle[m + 1] = '\207';
142         needle[m + 2] = '\0';
143
144         result = mbsstr (haystack, needle);
145         ASSERT (result == haystack + m);
146       }
147     free (needle);
148     free (haystack);
149   }
150
151   return 0;
152 }