1/* Test memset functions.
2 Copyright (C) 1999-2024 Free Software Foundation, Inc.
3 This file is part of the GNU C Library.
4
5 The GNU C Library is free software; you can redistribute it and/or
6 modify it under the terms of the GNU Lesser General Public
7 License as published by the Free Software Foundation; either
8 version 2.1 of the License, or (at your option) any later version.
9
10 The GNU C Library is distributed in the hope that it will be useful,
11 but WITHOUT ANY WARRANTY; without even the implied warranty of
12 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
13 Lesser General Public License for more details.
14
15 You should have received a copy of the GNU Lesser General Public
16 License along with the GNU C Library; if not, see
17 <https://www.gnu.org/licenses/>. */
18
19#define TEST_MAIN
20#ifdef TEST_BZERO
21# ifdef TEST_EXPLICIT_BZERO
22# define TEST_NAME "explicit_bzero"
23# else
24# define TEST_NAME "bzero"
25# endif
26#else
27# ifndef WIDE
28# define TEST_NAME "memset"
29# else
30# define TEST_NAME "wmemset"
31# endif /* WIDE */
32#endif /* !TEST_BZERO */
33#define MIN_PAGE_SIZE 131072
34#include "test-string.h"
35
36#ifndef WIDE
37# define MEMSET memset
38# define CHAR char
39# define UCHAR unsigned char
40# define SIMPLE_MEMSET simple_memset
41# define MEMCMP memcmp
42# define BIG_CHAR CHAR_MAX
43#else
44# include <wchar.h>
45# define MEMSET wmemset
46# define CHAR wchar_t
47# define UCHAR wchar_t
48# define SIMPLE_MEMSET simple_wmemset
49# define MEMCMP wmemcmp
50# define BIG_CHAR WCHAR_MAX
51#endif /* WIDE */
52
53#ifdef TEST_BZERO
54typedef void (*proto_t) (char *, size_t);
55# ifdef TEST_EXPLICIT_BZERO
56IMPL (explicit_bzero, 1)
57# else
58IMPL (bzero, 1)
59# endif
60#else
61typedef CHAR *(*proto_t) (CHAR *, int, size_t);
62IMPL (MEMSET, 1)
63#endif /* !TEST_BZERO */
64
65/* Naive implementation to verify results. */
66CHAR *
67inhibit_loop_to_libcall
68SIMPLE_MEMSET (CHAR *s, int c, size_t n)
69{
70 CHAR *r = s, *end = s + n;
71 while (r < end)
72 *r++ = c;
73 return s;
74}
75
76static void
77do_one_test (impl_t *impl, CHAR *s, int c __attribute ((unused)), size_t n, int space_below, int space_above)
78{
79 CHAR buf[n];
80 CHAR sentinel = ~c;
81 if (space_below)
82 s[-1] = sentinel;
83 if (space_above)
84 s[n] = sentinel;
85 SIMPLE_MEMSET(s, c: ~c, n);
86#ifdef TEST_BZERO
87 SIMPLE_MEMSET (s: buf, c: 0, n);
88 CALL (impl, s, n);
89 if (memcmp (s, buf, n) != 0
90 || (space_below && s[-1] != sentinel)
91 || (space_above && s[n] != sentinel))
92#else
93 CHAR *res = CALL (impl, s, c, n);
94 if (res != s
95 || SIMPLE_MEMSET (buf, c, n) != buf
96 || MEMCMP (s, buf, n) != 0
97 || (space_below && s[-1] != sentinel)
98 || (space_above && s[n] != sentinel))
99#endif /* !TEST_BZERO */
100 {
101 error (status: 0, errnum: 0, format: "Wrong result in function %s", impl->name);
102 ret = 1;
103 return;
104 }
105}
106
107static void
108do_test (size_t align, int c, size_t len)
109{
110 int space_below, space_above;
111 align &= 4095;
112 if ((align + len) * sizeof (CHAR) > page_size)
113 return;
114
115 space_below = !!align;
116 space_above = !((align + len + 1) * sizeof (CHAR) > page_size);
117
118 FOR_EACH_IMPL (impl, 0)
119 do_one_test (impl, s: (CHAR *) (buf1) + align, c, n: len, space_below, space_above);
120}
121
122#ifndef TEST_BZERO
123static void
124do_random_tests (void)
125{
126 size_t i, j, k, n, align, len, size;
127 int c, o;
128 UCHAR *p, *res;
129 UCHAR *p2 = (UCHAR *) buf2;
130
131 for (i = 0; i < 65536 / sizeof (CHAR); ++i)
132 p2[i] = random () & BIG_CHAR;
133
134 for (n = 0; n < ITERATIONS; n++)
135 {
136 if ((random () & 31) == 0)
137 size = 65536 / sizeof (CHAR);
138 else
139 size = 512;
140 p = (UCHAR *) (buf1 + page_size) - size;
141 len = random () & (size - 1);
142 align = size - len - (random () & 31);
143 if (align > size)
144 align = size - len;
145 if ((random () & 7) == 0)
146 align &= ~63;
147 if ((random () & 7) == 0)
148 c = 0;
149 else
150 c = random () & BIG_CHAR;
151 o = random () & BIG_CHAR;
152 if (o == c)
153 o = (c + 1) & BIG_CHAR;
154 j = len + align + 128;
155 if (j > size)
156 j = size;
157 if (align >= 128)
158 k = align - 128;
159 else
160 k = 0;
161 for (i = k; i < align; ++i)
162 p[i] = o;
163 for (i = align + len; i < j; ++i)
164 p[i] = o;
165
166 FOR_EACH_IMPL (impl, 1)
167 {
168 for (i = 0; i < len; ++i)
169 {
170 p[i + align] = p2[i];
171 if (p[i + align] == c)
172 p[i + align] = o;
173 }
174 res = (UCHAR *) CALL (impl, (CHAR *) p + align, c, len);
175 if (res != p + align)
176 {
177 error (0, 0, "Iteration %zd - wrong result in function %s (%zd, %d, %zd) %p != %p",
178 n, impl->name, align, c, len, res, p + align);
179 ret = 1;
180 }
181 for (i = k; i < align; ++i)
182 if (p[i] != o)
183 {
184 error (0, 0, "Iteration %zd - garbage before %s (%zd, %d, %zd)",
185 n, impl->name, align, c, len);
186 ret = 1;
187 break;
188 }
189 for (; i < align + len; ++i)
190 if (p[i] != c)
191 {
192 error (0, 0, "Iteration %zd - not cleared correctly %s (%zd, %d, %zd)",
193 n, impl->name, align, c, len);
194 ret = 1;
195 break;
196 }
197 for (; i < j; ++i)
198 if (p[i] != o)
199 {
200 error (0, 0, "Iteration %zd - garbage after %s (%zd, %d, %zd)",
201 n, impl->name, align, c, len);
202 ret = 1;
203 break;
204 }
205 }
206 }
207}
208#endif /* !TEST_BZERO */
209
210int
211test_main (void)
212{
213 size_t i;
214 int c = 0;
215
216 test_init ();
217
218 printf (format: "%24s", "");
219 FOR_EACH_IMPL (impl, 0)
220 printf (format: "\t%s", impl->name);
221 putchar (c: '\n');
222
223#ifndef TEST_BZERO
224 for (c = -65; c <= 130; c += 65)
225#endif
226 {
227 for (i = 0; i < 18; ++i)
228 do_test (align: 0, c, len: 1 << i);
229 for (i = 1; i < 64; ++i)
230 {
231 do_test (align: i, c, len: i);
232 do_test (align: 4096 - i, c, len: i);
233 do_test (align: 4095, c, len: i);
234 if (i & (i - 1))
235 do_test (align: 0, c, len: i);
236 }
237 do_test (align: 1, c, len: 14);
238 do_test (align: 3, c, len: 1024);
239 do_test (align: 4, c, len: 64);
240 do_test (align: 2, c, len: 25);
241 }
242
243#ifndef TEST_BZERO
244 do_random_tests ();
245#endif
246
247 return ret;
248}
249
250#include <support/test-driver.c>
251

source code of glibc/string/test-memset.c