tools/tests/darwintests/jumbo_va_spaces_28530648.c

   1 #include <stdio.h>
   2 #include <errno.h>
   3 #include <string.h>
   4 #include <unistd.h>
   5 #include <sys/mman.h>
   6
   7 #include <darwintest.h>
   8 #include <darwintest_utils.h>
   9
  10
  11 #define GB (1ULL * 1024 * 1024 * 1024)
  12
  13 /*
  14  * This test expects the entitlement to be the enabling factor for a process to
  15  * allocate at least this many GB of VA space. i.e. with the entitlement, n GB
  16  * must be allocatable; whereas without it, it must be less.
  17  */
  18 #define ALLOC_TEST_GB 12
  19
  20 T_DECL(jumbo_va_spaces_28530648,
  21         "Verify that the \"dynamic-codesigning\" entitlement is required to utilize an extra-large "
  22         "VA space on arm64",
  23         T_META_NAMESPACE("xnu.vm"),
  24         T_META_CHECK_LEAKS(false))
  25 {
  26         int     i;
  27         void    *res;
  28
  29         if (!dt_64_bit_kernel()) {
  30                 T_SKIP("This test is only applicable to arm64");
  31         }
  32
  33         T_LOG("Attemping to allocate VA space in 1 GB chunks.");
  34
  35         for (i = 0; i < (ALLOC_TEST_GB * 2); i++) {
  36                 res = mmap(NULL, 1 * GB, PROT_NONE, MAP_PRIVATE | MAP_ANON, 0, 0);
  37                 if (res == MAP_FAILED) {
  38                         if (errno != ENOMEM) {
  39                                 T_WITH_ERRNO;
  40                                 T_LOG("mmap failed: stopped at %d of %d GB allocated", i, ALLOC_TEST_GB);
  41                         }
  42                         break;
  43                 } else {
  44                         T_LOG("%d: %p\n", i, res);
  45                 }
  46         }
  47
  48 #if defined(ENTITLED)
  49         T_EXPECT_GE_INT(i, ALLOC_TEST_GB, "Allocate at least %d GB of VA space", ALLOC_TEST_GB);
  50 #else
  51         T_EXPECT_LT_INT(i, ALLOC_TEST_GB, "Not permitted to allocate %d GB of VA space", ALLOC_TEST_GB);
  52 #endif
  53 }