string/test-string.h

   1 /* Test and measure string and memory functions.
   2    Copyright (C) 1999-2023 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))
 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
 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