libitm/testsuite/libitm.c/memset-1.c

   1 /* This program is free software; you can redistribute it and/or modify
   2    it under the terms of the GNU General Public License as published by
   3    the Free Software Foundation; either version 2 of the License, or
   4    (at your option) any later version.
   5
   6    This program is distributed in the hope that it will be useful, but
   7    WITHOUT ANY WARRANTY; without even the implied warranty of
   8    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
   9    General Public License for more details.
  10
  11    You should have received a copy of the GNU General Public License
  12    along with this program; if not, write to the Free Software
  13    Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
  14    02110-1301, USA.  */
  15
  16 /* Verify memcpy operation.  */
  17
  18 #include <stdio.h>
  19 #include <stdlib.h>
  20 #include <string.h>
  21 #include <sys/mman.h>
  22 #include <libitm.h>
  23
  24 #define BEG_TRANSACTION \
  25   _ITM_beginTransaction (pr_instrumentedCode | pr_hasNoAbort \
  26                          | pr_hasNoIrrevocable)
  27 #define END_TRANSACTION \
  28   _ITM_commitTransaction ()
  29
  30 #define MEMSET _ITM_memsetW
  31
  32 static unsigned char *buf;
  33 static size_t bufsize, page_size;
  34 static int fail;
  35
  36 static void
  37 do_test (size_t align, size_t len)
  38 {
  39   size_t i, j;
  40   unsigned char c1, c2;
  41
  42   if (align + len >= bufsize)
  43     return;
  44
  45   c1 = random () >> 8;
  46   c2 = random () >> 8;
  47   if (c1 == c2)
  48     c1++;
  49   memset (buf, c1, bufsize);
  50
  51   BEG_TRANSACTION;
  52   MEMSET (buf + align, c2, len);
  53   END_TRANSACTION;
  54
  55   i = (align > 64 ? align - 64 : 0);
  56   for (; i < align; ++i)
  57     if (buf[i] != c1)
  58       {
  59         printf ("Garbage before: ofs %zd\n", i);
  60         fail = 1;
  61         break;
  62       }
  63   for (; i < align + len; ++i)
  64     if (buf[i] != c2)
  65       {
  66         printf ("Wrong result: ofs %zd\n", i);
  67         fail = 1;
  68         break;
  69       }
  70   for (j = i + 64 < bufsize ? i + 64 : bufsize; i < j; ++i)
  71     if (buf[i] != c1)
  72       {
  73         printf ("Garbage after: ofs %zd\n", i);
  74         fail = 1;
  75         break;
  76       }
  77 }
  78
  79 int main()
  80 {
  81   size_t i, j;
  82
  83   page_size = getpagesize ();
  84   bufsize = 2 * page_size;
  85
  86   buf = mmap (NULL, bufsize + 2*page_size, PROT_READ | PROT_WRITE,
  87               MAP_PRIVATE | MAP_ANONYMOUS, -1, 0);
  88   if (buf == MAP_FAILED)
  89     return 1;
  90
  91   if (mprotect (buf, page_size, PROT_NONE))
  92     return 1;
  93   buf += page_size;
  94   if (mprotect (buf + bufsize, page_size, PROT_NONE))
  95     return 1;
  96
  97   for (i = 0; i < 18; ++i)
  98     {
  99       size_t len = 1 << i;
 100       do_test (0, len);
 101       do_test (bufsize - len, len);
 102     }
 103
 104   for (i = 0; i < 32; ++i)
 105     for (j = 0; j < 32; ++j)
 106       do_test (j, i);
 107
 108   for (i = 3; i < 32; ++i)
 109     {
 110       if ((i & (i - 1)) == 0)
 111         continue;
 112       do_test (0, 16 * i);
 113       do_test (i, 16 * i);
 114     }
 115
 116   return fail;
 117 }