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