1/* Measure __strcpy_chk functions.
2 Copyright (C) 2013-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#ifndef STRCPY_RESULT
20# define STRCPY_RESULT(dst, len) dst
21# define TEST_MAIN
22# define TEST_NAME "strcpy_chk"
23# include "bench-string.h"
24
25/* This test case implicitly tests the availability of the __chk_fail
26 symbol, which is part of the public ABI and may be used
27 externally. */
28extern void __attribute__ ((noreturn)) __chk_fail (void);
29extern char *normal_strcpy (char *, const char *, size_t)
30 __asm ("strcpy");
31extern char *__strcpy_chk (char *, const char *, size_t);
32
33IMPL (normal_strcpy, 1)
34IMPL (__strcpy_chk, 2)
35
36#endif
37
38#include <fcntl.h>
39#include <paths.h>
40#include <setjmp.h>
41#include <signal.h>
42
43#include <support/support.h>
44
45volatile int chk_fail_ok;
46jmp_buf chk_fail_buf;
47
48static void
49handler (int sig)
50{
51 if (chk_fail_ok)
52 {
53 chk_fail_ok = 0;
54 longjmp (chk_fail_buf, 1);
55 }
56 else
57 _exit (127);
58}
59
60typedef char *(*proto_t) (char *, const char *, size_t);
61
62static void
63do_one_test (impl_t *impl, char *dst, const char *src,
64 size_t len, size_t dlen)
65{
66 char *res;
67 size_t i, iters = INNER_LOOP_ITERS_LARGE;
68 timing_t start, stop, cur;
69
70 if (dlen <= len)
71 {
72 if (impl->test == 1)
73 return;
74
75 chk_fail_ok = 1;
76 if (setjmp (chk_fail_buf) == 0)
77 {
78 res = CALL (impl, dst, src, dlen);
79 printf (format: "*** Function %s (%zd; %zd) did not __chk_fail\n",
80 impl->name, len, dlen);
81 chk_fail_ok = 0;
82 ret = 1;
83 }
84 return;
85 }
86 else
87 res = CALL (impl, dst, src, dlen);
88
89 if (res != STRCPY_RESULT (dst, len))
90 {
91 printf (format: "Wrong result in function %s %p %p\n", impl->name,
92 res, STRCPY_RESULT (dst, len));
93 ret = 1;
94 return;
95 }
96
97 if (strcmp (dst, src) != 0)
98 {
99 printf (format: "Wrong result in function %s dst \"%s\" src \"%s\"\n",
100 impl->name, dst, src);
101 ret = 1;
102 return;
103 }
104
105 TIMING_NOW (start);
106 for (i = 0; i < iters; ++i)
107 {
108 CALL (impl, dst, src, dlen);
109 }
110 TIMING_NOW (stop);
111
112 TIMING_DIFF (cur, start, stop);
113
114 TIMING_PRINT_MEAN ((double) cur, (double) iters);
115}
116
117static void
118do_test (size_t align1, size_t align2, size_t len, size_t dlen, int max_char)
119{
120 size_t i;
121 char *s1, *s2;
122
123 align1 &= 7;
124 if (align1 + len >= page_size)
125 return;
126
127 align2 &= 7;
128 if (align2 + len >= page_size)
129 return;
130
131 s1 = (char *) buf1 + align1;
132 s2 = (char *) buf2 + align2;
133
134 for (i = 0; i < len; i++)
135 s1[i] = 32 + 23 * i % (max_char - 32);
136 s1[len] = 0;
137
138 if (dlen > len)
139 printf (format: "Length %4zd, alignment %2zd/%2zd:", len, align1, align2);
140
141 FOR_EACH_IMPL (impl, 0)
142 do_one_test (impl, dst: s2, src: s1, len, dlen);
143
144 if (dlen > len)
145 putchar (c: '\n');
146}
147
148static int
149test_main (void)
150{
151 size_t i;
152
153 set_fortify_handler (handler);
154
155 test_init ();
156
157 printf (format: "%23s", "");
158 FOR_EACH_IMPL (impl, 0)
159 printf (format: "\t%s", impl->name);
160 putchar (c: '\n');
161
162 for (i = 0; i < 16; ++i)
163 {
164 do_test (align1: 0, align2: 0, len: i, dlen: i + 1, max_char: 127);
165 do_test (align1: 0, align2: 0, len: i, dlen: i + 1, max_char: 255);
166 do_test (align1: 0, align2: i, len: i, dlen: i + 1, max_char: 127);
167 do_test (align1: i, align2: 0, len: i, dlen: i + 1, max_char: 255);
168 }
169
170 for (i = 1; i < 8; ++i)
171 {
172 do_test (align1: 0, align2: 0, len: 8 << i, dlen: (8 << i) + 1, max_char: 127);
173 do_test (align1: 8 - i, align2: 2 * i, len: (8 << i), dlen: (8 << i) + 1, max_char: 127);
174 }
175
176 for (i = 1; i < 8; ++i)
177 {
178 do_test (align1: i, align2: 2 * i, len: (8 << i), dlen: (8 << i) + 1, max_char: 127);
179 do_test (align1: 2 * i, align2: i, len: (8 << i), dlen: (8 << i) + 1, max_char: 255);
180 do_test (align1: i, align2: i, len: (8 << i), dlen: (8 << i) + 1, max_char: 127);
181 do_test (align1: i, align2: i, len: (8 << i), dlen: (8 << i) + 1, max_char: 255);
182 }
183
184 for (i = 0; i < 16; ++i)
185 {
186 do_test (align1: 0, align2: 0, len: i, dlen: i + 256, max_char: 127);
187 do_test (align1: 0, align2: 0, len: i, dlen: i + 256, max_char: 255);
188 do_test (align1: 0, align2: i, len: i, dlen: i + 256, max_char: 127);
189 do_test (align1: i, align2: 0, len: i, dlen: i + 256, max_char: 255);
190 }
191
192 for (i = 1; i < 8; ++i)
193 {
194 do_test (align1: 0, align2: 0, len: 8 << i, dlen: (8 << i) + 256, max_char: 127);
195 do_test (align1: 8 - i, align2: 2 * i, len: (8 << i), dlen: (8 << i) + 256, max_char: 127);
196 }
197
198 for (i = 1; i < 8; ++i)
199 {
200 do_test (align1: i, align2: 2 * i, len: (8 << i), dlen: (8 << i) + 256, max_char: 127);
201 do_test (align1: 2 * i, align2: i, len: (8 << i), dlen: (8 << i) + 256, max_char: 255);
202 do_test (align1: i, align2: i, len: (8 << i), dlen: (8 << i) + 256, max_char: 127);
203 do_test (align1: i, align2: i, len: (8 << i), dlen: (8 << i) + 256, max_char: 255);
204 }
205
206 for (i = 0; i < 16; ++i)
207 {
208 do_test (align1: 0, align2: 0, len: i, dlen: i, max_char: 127);
209 do_test (align1: 0, align2: 0, len: i, dlen: i + 2, max_char: 255);
210 do_test (align1: 0, align2: i, len: i, dlen: i + 3, max_char: 127);
211 do_test (align1: i, align2: 0, len: i, dlen: i + 4, max_char: 255);
212 }
213
214 for (i = 1; i < 8; ++i)
215 {
216 do_test (align1: 0, align2: 0, len: 8 << i, dlen: (8 << i) - 15, max_char: 127);
217 do_test (align1: 8 - i, align2: 2 * i, len: (8 << i), dlen: (8 << i) + 5, max_char: 127);
218 }
219
220 for (i = 1; i < 8; ++i)
221 {
222 do_test (align1: i, align2: 2 * i, len: (8 << i), dlen: (8 << i) + i, max_char: 127);
223 do_test (align1: 2 * i, align2: i, len: (8 << i), dlen: (8 << i) + (i - 1), max_char: 255);
224 do_test (align1: i, align2: i, len: (8 << i), dlen: (8 << i) + i + 2, max_char: 127);
225 do_test (align1: i, align2: i, len: (8 << i), dlen: (8 << i) + i + 3, max_char: 255);
226 }
227
228 return 0;
229}
230
231#include <support/test-driver.c>
232

source code of glibc/benchtests/bench-strcpy_chk.c