Tests for module 'c-strstr'.
[gnulib.git] / tests / test-c-strstr.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 #ifdef HAVE_CONFIG_H
21 # include <config.h>
22 #endif
23
24 #include "c-strstr.h"
25
26 #include <stdlib.h>
27
28 #define ASSERT(expr) if (!(expr)) abort ();
29
30 int
31 main ()
32 {
33   {
34     const char input[] = "foo";
35     const char *result = c_strstr (input, "");
36     ASSERT (result == input);
37   }
38
39   {
40     const char input[] = "foo";
41     const char *result = c_strstr (input, "o");
42     ASSERT (result == input + 1);
43   }
44
45   {
46     const char input[] = "ABC ABCDAB ABCDABCDABDE";
47     const char *result = c_strstr (input, "ABCDABD");
48     ASSERT (result == input + 15);
49   }
50
51   {
52     const char input[] = "ABC ABCDAB ABCDABCDABDE";
53     const char *result = c_strstr (input, "ABCDABE");
54     ASSERT (result == NULL);
55   }
56
57   /* Check that a very long haystack is handled quickly if the needle is
58      short and occurs near the beginning.  */
59   {
60     size_t repeat = 10000;
61     size_t m = 1000000;
62     char *needle =
63       "AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA"
64       "AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA";
65     char *haystack = (char *) malloc (m + 1);
66     if (haystack != NULL)
67       {
68         memset (haystack, 'A', m);
69         haystack[0] = 'B';
70         haystack[m] = '\0';
71
72         for (; repeat > 0; repeat--)
73           {
74             ASSERT (c_strstr (haystack, needle) == haystack + 1);
75           }
76
77         free (haystack);
78       }
79   }
80
81   /* Check that a very long needle is discarded quickly if the haystack is
82      short.  */
83   {
84     size_t repeat = 10000;
85     size_t m = 1000000;
86     char *haystack =
87       "AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA"
88       "ABABABABABABABABABABABABABABABABABABABABABABABABABABABABABABABABABAB";
89     char *needle = (char *) malloc (m + 1);
90     if (needle != NULL)
91       {
92         memset (needle, 'A', m);
93         needle[m] = '\0';
94
95         for (; repeat > 0; repeat--)
96           {
97             ASSERT (c_strstr (haystack, needle) == NULL);
98           }
99
100         free (needle);
101       }
102   }
103
104   /* Check that the asymptotic worst-case complexity is not quadratic.  */
105   {
106     size_t m = 1000000;
107     char *haystack = (char *) malloc (2 * m + 2);
108     char *needle = (char *) malloc (m + 2);
109     if (haystack != NULL && needle != NULL)
110       {
111         const char *result;
112
113         memset (haystack, 'A', 2 * m);
114         haystack[2 * m] = 'B';
115         haystack[2 * m + 1] = '\0';
116
117         memset (needle, 'A', m);
118         needle[m] = 'B';
119         needle[m + 1] = '\0';
120
121         result = c_strstr (haystack, needle);
122         ASSERT (result == haystack + m);
123       }
124     if (needle != NULL)
125       free (needle);
126     if (haystack != NULL)
127       free (haystack);
128   }
129
130   return 0;
131 }