string/test-memset.c

   1 /* Test memset 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 #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 CHAR *SIMPLE_MEMSET (CHAR *, int, size_t);
  54
  55 #ifdef TEST_BZERO
  56 typedef void (*proto_t) (char *, size_t);
  57 void simple_bzero (char *, size_t);
  58 void builtin_bzero (char *, size_t);
  59
  60 IMPL (simple_bzero, 0)
  61 IMPL (builtin_bzero, 0)
  62 #ifdef TEST_EXPLICIT_BZERO
  63 IMPL (explicit_bzero, 1)
  64 #else
  65 IMPL (bzero, 1)
  66 #endif
  67
  68 void
  69 simple_bzero (char *s, size_t n)
  70 {
  71   SIMPLE_MEMSET (s, 0, n);
  72 }
  73
  74 void
  75 builtin_bzero (char *s, size_t n)
  76 {
  77   __builtin_bzero (s, n);
  78 }
  79 #else
  80 typedef CHAR *(*proto_t) (CHAR *, int, size_t);
  81
  82 IMPL (SIMPLE_MEMSET, 0)
  83 # ifndef WIDE
  84 char *builtin_memset (char *, int, size_t);
  85 IMPL (builtin_memset, 0)
  86 # endif /* !WIDE */
  87 IMPL (MEMSET, 1)
  88
  89 # ifndef WIDE
  90 char *
  91 builtin_memset (char *s, int c, size_t n)
  92 {
  93   return __builtin_memset (s, c, n);
  94 }
  95 # endif /* !WIDE */
  96 #endif /* !TEST_BZERO */
  97
  98 CHAR *
  99 inhibit_loop_to_libcall
 100 SIMPLE_MEMSET (CHAR *s, int c, size_t n)
 101 {
 102   CHAR *r = s, *end = s + n;
 103   while (r < end)
 104     *r++ = c;
 105   return s;
 106 }
 107
 108 static void
 109 do_one_test (impl_t *impl, CHAR *s, int c __attribute ((unused)), size_t n, int space_below, int space_above)
 110 {
 111   CHAR buf[n];
 112   CHAR sentinel = ~c;
 113   if (space_below)
 114       s[-1] = sentinel;
 115   if (space_above)
 116       s[n] = sentinel;
 117   SIMPLE_MEMSET(s, ~c, n);
 118 #ifdef TEST_BZERO
 119   simple_bzero (buf, n);
 120   CALL (impl, s, n);
 121   if (memcmp (s, buf, n) != 0
 122       || (space_below && s[-1] != sentinel)
 123       || (space_above && s[n] != sentinel))
 124 #else
 125   CHAR *res = CALL (impl, s, c, n);
 126   if (res != s
 127       || SIMPLE_MEMSET (buf, c, n) != buf
 128       || MEMCMP (s, buf, n) != 0
 129       || (space_below && s[-1] != sentinel)
 130       || (space_above && s[n] != sentinel))
 131 #endif /* !TEST_BZERO */
 132     {
 133       error (0, 0, "Wrong result in function %s", impl->name);
 134       ret = 1;
 135       return;
 136     }
 137 }
 138
 139 static void
 140 do_test (size_t align, int c, size_t len)
 141 {
 142   int space_below, space_above;
 143   align &= 4095;
 144   if ((align + len) * sizeof (CHAR) > page_size)
 145     return;
 146
 147   space_below = !!align;
 148   space_above = !((align + len + 1) * sizeof (CHAR) > page_size);
 149
 150   FOR_EACH_IMPL (impl, 0)
 151     do_one_test (impl, (CHAR *) (buf1) + align, c, len, space_below, space_above);
 152 }
 153
 154 #ifndef TEST_BZERO
 155 static void
 156 do_random_tests (void)
 157 {
 158   size_t i, j, k, n, align, len, size;
 159   int c, o;
 160   UCHAR *p, *res;
 161   UCHAR *p2 = (UCHAR *) buf2;
 162
 163   for (i = 0; i < 65536 / sizeof (CHAR); ++i)
 164     p2[i] = random () & BIG_CHAR;
 165
 166   for (n = 0; n < ITERATIONS; n++)
 167     {
 168       if ((random () & 31) == 0)
 169         size = 65536 / sizeof (CHAR);
 170       else
 171         size = 512;
 172       p = (UCHAR *) (buf1 + page_size) - size;
 173       len = random () & (size - 1);
 174       align = size - len - (random () & 31);
 175       if (align > size)
 176         align = size - len;
 177       if ((random () & 7) == 0)
 178         align &= ~63;
 179       if ((random () & 7) == 0)
 180         c = 0;
 181       else
 182         c = random () & BIG_CHAR;
 183       o = random () & BIG_CHAR;
 184       if (o == c)
 185         o = (c + 1) & BIG_CHAR;
 186       j = len + align + 128;
 187       if (j > size)
 188         j = size;
 189       if (align >= 128)
 190         k = align - 128;
 191       else
 192         k = 0;
 193       for (i = k; i < align; ++i)
 194         p[i] = o;
 195       for (i = align + len; i < j; ++i)
 196         p[i] = o;
 197
 198       FOR_EACH_IMPL (impl, 1)
 199         {
 200           for (i = 0; i < len; ++i)
 201             {
 202               p[i + align] = p2[i];
 203               if (p[i + align] == c)
 204                 p[i + align] = o;
 205             }
 206           res = (UCHAR *) CALL (impl, (CHAR *) p + align, c, len);
 207           if (res != p + align)
 208             {
 209               error (0, 0, "Iteration %zd - wrong result in function %s (%zd, %d, %zd) %p != %p",
 210                      n, impl->name, align, c, len, res, p + align);
 211               ret = 1;
 212             }
 213           for (i = k; i < align; ++i)
 214             if (p[i] != o)
 215               {
 216                 error (0, 0, "Iteration %zd - garbage before %s (%zd, %d, %zd)",
 217                        n, impl->name, align, c, len);
 218                 ret = 1;
 219                 break;
 220               }
 221           for (; i < align + len; ++i)
 222             if (p[i] != c)
 223               {
 224                 error (0, 0, "Iteration %zd - not cleared correctly %s (%zd, %d, %zd)",
 225                        n, impl->name, align, c, len);
 226                 ret = 1;
 227                 break;
 228               }
 229           for (; i < j; ++i)
 230             if (p[i] != o)
 231               {
 232                 error (0, 0, "Iteration %zd - garbage after %s (%zd, %d, %zd)",
 233                        n, impl->name, align, c, len);
 234                 ret = 1;
 235                 break;
 236               }
 237         }
 238     }
 239 }
 240 #endif /* !TEST_BZERO */
 241
 242 int
 243 test_main (void)
 244 {
 245   size_t i;
 246   int c = 0;
 247
 248   test_init ();
 249
 250   printf ("%24s", "");
 251   FOR_EACH_IMPL (impl, 0)
 252     printf ("\t%s", impl->name);
 253   putchar ('\n');
 254
 255 #ifndef TEST_BZERO
 256   for (c = -65; c <= 130; c += 65)
 257 #endif
 258     {
 259       for (i = 0; i < 18; ++i)
 260         do_test (0, c, 1 << i);
 261       for (i = 1; i < 64; ++i)
 262         {
 263           do_test (i, c, i);
 264           do_test (4096 - i, c, i);
 265           do_test (4095, c, i);
 266           if (i & (i - 1))
 267             do_test (0, c, i);
 268         }
 269       do_test (1, c, 14);
 270       do_test (3, c, 1024);
 271       do_test (4, c, 64);
 272       do_test (2, c, 25);
 273     }
 274
 275 #ifndef TEST_BZERO
 276   do_random_tests ();
 277 #endif
 278
 279   return ret;
 280 }
 281
 282 #include <support/test-driver.c>