1  /* Tests for <support/blob_repeat.h>
       2     Copyright (C) 2018-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 <stdio.h>
      20  #include <string.h>
      21  #include <support/blob_repeat.h>
      22  #include <support/check.h>
      23  
      24  static int
      25  do_test (void)
      26  {
      27    struct support_blob_repeat repeat
      28      = support_blob_repeat_allocate ("5", 1, 5);
      29    TEST_COMPARE_BLOB (repeat.start, repeat.size, "55555", 5);
      30    support_blob_repeat_free (&repeat);
      31  
      32    repeat = support_blob_repeat_allocate ("ABC", 3, 3);
      33    TEST_COMPARE_BLOB (repeat.start, repeat.size, "ABCABCABC", 9);
      34    support_blob_repeat_free (&repeat);
      35  
      36    repeat = support_blob_repeat_allocate ("abc", 4, 3);
      37    TEST_COMPARE_BLOB (repeat.start, repeat.size, "abc\0abc\0abc", 12);
      38    support_blob_repeat_free (&repeat);
      39  
      40    size_t gigabyte = 1U << 30;
      41    repeat = support_blob_repeat_allocate ("X", 1, gigabyte + 1);
      42    if (repeat.start == NULL)
      43      puts ("warning: not enough memory for 1 GiB mapping");
      44    else
      45      {
      46        TEST_COMPARE (repeat.size, gigabyte + 1);
      47        {
      48          unsigned char *p = repeat.start;
      49          for (size_t i = 0; i < gigabyte + 1; ++i)
      50            if (p[i] != 'X')
      51              FAIL_EXIT1 ("invalid byte 0x%02x at %zu", p[i], i);
      52  
      53          /* Check that there is no sharing across the mapping.  */
      54          p[0] = 'Y';
      55          p[1U << 24] = 'Z';
      56          for (size_t i = 0; i < gigabyte + 1; ++i)
      57            if (i == 0)
      58              TEST_COMPARE (p[i], 'Y');
      59            else if (i == 1U << 24)
      60              TEST_COMPARE (p[i], 'Z');
      61            else if (p[i] != 'X')
      62              FAIL_EXIT1 ("invalid byte 0x%02x at %zu", p[i], i);
      63        }
      64      }
      65    support_blob_repeat_free (&repeat);
      66  
      67    for (int do_shared = 0; do_shared < 2; ++do_shared)
      68      {
      69        if (do_shared)
      70          repeat = support_blob_repeat_allocate_shared ("012345678", 9,
      71                                                        10 * 1000 * 1000);
      72        else
      73          repeat = support_blob_repeat_allocate ("012345678", 9,
      74                                                 10 * 1000 * 1000);
      75        if (repeat.start == NULL)
      76          puts ("warning: not enough memory for large mapping");
      77        else
      78          {
      79            unsigned char *p = repeat.start;
      80            for (int i = 0; i < 10 * 1000 * 1000; ++i)
      81              for (int j = 0; j <= 8; ++j)
      82                if (p[i * 9 + j] != '0' + j)
      83                  {
      84                    printf ("error: element %d index %d\n", i, j);
      85                    TEST_COMPARE (p[i * 9 + j], '0' + j);
      86                  }
      87  
      88            enum { total_size = 9 * 10 * 1000 * 1000 };
      89            p[total_size - 1] = '\0';
      90            asm ("" ::: "memory");
      91            if (do_shared)
      92              /* The write is repeated in multiple places earlier in the
      93                 string due to page sharing.  */
      94              TEST_VERIFY (strlen (repeat.start) < total_size - 1);
      95            else
      96              TEST_COMPARE (strlen (repeat.start), total_size - 1);
      97          }
      98        support_blob_repeat_free (&repeat);
      99      }
     100  
     101    return 0;
     102  }
     103  
     104  #include <support/test-driver.c>