1 /* Test and measure string and memory functions.
2 Copyright (C) 1999-2022 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 #include <sys/cdefs.h>
20 #include <support/support.h>
21
22 typedef struct
23 {
24 const char *name;
25 void (*fn) (void);
26 long test;
27 } impl_t;
28 extern impl_t __start_impls[], __stop_impls[];
29
30 #define IMPL(name, test) \
31 impl_t tst_ ## name \
32 __attribute__ ((section ("impls"), aligned (sizeof (void *)))) \
33 = { __STRING (name), (void (*) (void))name, test };
34
35 #ifdef TEST_MAIN
36
37 #ifndef _GNU_SOURCE
38 #define _GNU_SOURCE
39 #endif
40
41 #undef __USE_STRING_INLINES
42
43 /* We are compiled under _ISOMAC, so libc-symbols.h does not do this
44 for us. */
45 #include "config.h"
46 #ifdef HAVE_CC_INHIBIT_LOOP_TO_LIBCALL
47 # define inhibit_loop_to_libcall \
48 __attribute__ ((__optimize__ ("-fno-tree-loop-distribute-patterns")))
49 #else
50 # define inhibit_loop_to_libcall
51 #endif
52
53 #include <getopt.h>
54 #include <stdint.h>
55 #include <stdio.h>
56 #include <stdlib.h>
57 #include <string.h>
58 #include <sys/mman.h>
59 #include <sys/param.h>
60 #include <unistd.h>
61 #include <fcntl.h>
62 #include <error.h>
63 #include <errno.h>
64 #include <time.h>
65 #include <ifunc-impl-list.h>
66 #define GL(x) _##x
67 #define GLRO(x) _##x
68
69
70 # define TEST_FUNCTION test_main
71 # ifndef TIMEOUT
72 # define TIMEOUT (4 * 60)
73 # endif
74 # define OPT_ITERATIONS 10000
75 # define OPT_RANDOM 10001
76 # define OPT_SEED 10002
77
78 unsigned char *buf1, *buf2;
79 int ret, do_srandom;
80 unsigned int seed;
81 size_t page_size;
82
83 # ifndef ITERATIONS
84 size_t iterations = 100000;
85 # define ITERATIONS_OPTIONS \
86 { "iterations", required_argument, NULL, OPT_ITERATIONS },
87 # define ITERATIONS_PROCESS \
88 case OPT_ITERATIONS: \
89 iterations = strtoul (optarg, NULL, 0); \
90 break;
91 # define ITERATIONS iterations
92 # else
93 # define ITERATIONS_OPTIONS
94 # define ITERATIONS_PROCESS
95 # endif
96
97 # define CMDLINE_OPTIONS ITERATIONS_OPTIONS \
98 { "random", no_argument, NULL, OPT_RANDOM }, \
99 { "seed", required_argument, NULL, OPT_SEED },
100
101 static void __attribute__ ((used))
cmdline_process_function(int c)102 cmdline_process_function (int c)
103 {
104 switch (c)
105 {
106 ITERATIONS_PROCESS
107 case OPT_RANDOM:
108 {
109 int fdr = open ("/dev/urandom", O_RDONLY);
110 if (fdr < 0 || read (fdr, &seed, sizeof (seed)) != sizeof (seed))
111 seed = time (NULL);
112 if (fdr >= 0)
113 close (fdr);
114 do_srandom = 1;
115 break;
116 }
117
118 case OPT_SEED:
119 seed = strtoul (optarg, NULL, 0);
120 do_srandom = 1;
121 break;
122 }
123 }
124 # define CMDLINE_PROCESS cmdline_process_function
125
126 #define CALL(impl, ...) \
127 (* (proto_t) (impl)->fn) (__VA_ARGS__)
128
129 #ifdef TEST_NAME
130 /* Increase size of FUNC_LIST if assert is triggered at run-time. */
131 static struct libc_ifunc_impl func_list[32];
132 static int func_count;
133 static int impl_count = -1;
134 static impl_t *impl_array;
135
136 # define FOR_EACH_IMPL(impl, notall) \
137 impl_t *impl; \
138 int count; \
139 if (impl_count == -1) \
140 { \
141 impl_count = 0; \
142 if (func_count != 0) \
143 { \
144 int f; \
145 impl_t *skip = NULL, *a; \
146 for (impl = __start_impls; impl < __stop_impls; ++impl) \
147 if (strcmp (impl->name, TEST_NAME) == 0) \
148 skip = impl; \
149 else \
150 impl_count++; \
151 a = impl_array = xmalloc ((impl_count + func_count) * \
152 sizeof (impl_t)); \
153 for (impl = __start_impls; impl < __stop_impls; ++impl) \
154 if (impl != skip) \
155 *a++ = *impl; \
156 for (f = 0; f < func_count; f++) \
157 if (func_list[f].usable) \
158 { \
159 a->name = func_list[f].name; \
160 a->fn = func_list[f].fn; \
161 a->test = 1; \
162 a++; \
163 } \
164 impl_count = a - impl_array; \
165 } \
166 else \
167 { \
168 impl_count = __stop_impls - __start_impls; \
169 impl_array = __start_impls; \
170 } \
171 } \
172 impl = impl_array; \
173 for (count = 0; count < impl_count; ++count, ++impl) \
174 if (!notall || impl->test)
175 #else
176 # define FOR_EACH_IMPL(impl, notall) \
177 for (impl_t *impl = __start_impls; impl < __stop_impls; ++impl) \
178 if (!notall || impl->test)
179 #endif
180
181 #ifndef BUF1PAGES
182 # define BUF1PAGES 1
183 #endif
184
185 static void
test_init(void)186 test_init (void)
187 {
188 #ifdef TEST_NAME
189 func_count = __libc_ifunc_impl_list (TEST_NAME, func_list,
190 (sizeof func_list
191 / sizeof func_list[0]));
192 #endif
193
194 page_size = 2 * getpagesize ();
195 #ifdef MIN_PAGE_SIZE
196 if (page_size < MIN_PAGE_SIZE)
197 page_size = MIN_PAGE_SIZE;
198 #endif
199 buf1 = mmap (0, (BUF1PAGES + 1) * page_size, PROT_READ | PROT_WRITE,
200 MAP_PRIVATE | MAP_ANON, -1, 0);
201 if (buf1 == MAP_FAILED)
202 error (EXIT_FAILURE, errno, "mmap failed");
203 if (mprotect (buf1 + BUF1PAGES * page_size, page_size, PROT_NONE))
204 error (EXIT_FAILURE, errno, "mprotect failed");
205 buf2 = mmap (0, 2 * page_size, PROT_READ | PROT_WRITE,
206 MAP_PRIVATE | MAP_ANON, -1, 0);
207 if (buf2 == MAP_FAILED)
208 error (EXIT_FAILURE, errno, "mmap failed");
209 if (mprotect (buf2 + page_size, page_size, PROT_NONE))
210 error (EXIT_FAILURE, errno, "mprotect failed");
211 if (do_srandom)
212 {
213 printf ("Setting seed to 0x%x\n", seed);
214 srandom (seed);
215 }
216
217 memset (buf1, 0xa5, BUF1PAGES * page_size);
218 memset (buf2, 0x5a, page_size);
219 }
220
221 #endif
222