string/test-memset.c

   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
  54 typedef void (*proto_t) (char *, size_t);
  55 # ifdef TEST_EXPLICIT_BZERO
  56 IMPL (explicit_bzero, 1)
  57 # else
  58 IMPL (bzero, 1)
  59 # endif
  60 #else
  61 typedef CHAR *(*proto_t) (CHAR *, int, size_t);
  62 IMPL (MEMSET, 1)
  63 #endif /* !TEST_BZERO */
  64
  65 /* Naive implementation to verify results.  */
  66 CHAR *
  67 inhibit_loop_to_libcall
  68 SIMPLE_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
  76 static void
  77 do_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, n);
  86 #ifdef TEST_BZERO
  87   SIMPLE_MEMSET (buf, 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 (0, 0, "Wrong result in function %s", impl->name);
 102       ret = 1;
 103       return;
 104     }
 105 }
 106
 107 static void
 108 do_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, (CHAR *) (buf1) + align, c, len, space_below, space_above);
 120 }
 121
 122 #ifndef TEST_BZERO
 123 static void
 124 do_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
 210 int
 211 test_main (void)
 212 {
 213   size_t i;
 214   int c = 0;
 215
 216   test_init ();
 217
 218   printf ("%24s", "");
 219   FOR_EACH_IMPL (impl, 0)
 220     printf ("\t%s", impl->name);
 221   putchar ('\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 (0, c, 1 << i);
 229       for (i = 1; i < 64; ++i)
 230         {
 231           do_test (i, c, i);
 232           do_test (4096 - i, c, i);
 233           do_test (4095, c, i);
 234           if (i & (i - 1))
 235             do_test (0, c, i);
 236         }
 237       do_test (1, c, 14);
 238       do_test (3, c, 1024);
 239       do_test (4, c, 64);
 240       do_test (2, c, 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>