drivers/virt/nitro_enclaves/ne_misc_dev_test.c

   1 // SPDX-License-Identifier: GPL-2.0
   2
   3 #include <kunit/test.h>
   4
   5 #define MAX_PHYS_REGIONS        16
   6 #define INVALID_VALUE           (~0ull)
   7
   8 struct ne_phys_regions_test {
   9         u64           paddr;
  10         u64           size;
  11         int           expect_rc;
  12         unsigned long expect_num;
  13         u64           expect_last_paddr;
  14         u64           expect_last_size;
  15 } phys_regions_test_cases[] = {
  16         /*
  17          * Add the region from 0x1000 to (0x1000 + 0x200000 - 1):
  18          *   Expected result:
  19          *       Failed, start address is not 2M-aligned
  20          *
  21          * Now the instance of struct ne_phys_contig_mem_regions is:
  22          *   num = 0
  23          *   regions = {}
  24          */
  25         {0x1000, 0x200000, -EINVAL, 0, INVALID_VALUE, INVALID_VALUE},
  26
  27         /*
  28          * Add the region from 0x200000 to (0x200000 + 0x1000 - 1):
  29          *   Expected result:
  30          *       Failed, size is not 2M-aligned
  31          *
  32          * Now the instance of struct ne_phys_contig_mem_regions is:
  33          *   num = 0
  34          *   regions = {}
  35          */
  36         {0x200000, 0x1000, -EINVAL, 0, INVALID_VALUE, INVALID_VALUE},
  37
  38         /*
  39          * Add the region from 0x200000 to (0x200000 + 0x200000 - 1):
  40          *   Expected result:
  41          *       Successful
  42          *
  43          * Now the instance of struct ne_phys_contig_mem_regions is:
  44          *   num = 1
  45          *   regions = {
  46          *       {start=0x200000, end=0x3fffff}, // len=0x200000
  47          *   }
  48          */
  49         {0x200000, 0x200000, 0, 1, 0x200000, 0x200000},
  50
  51         /*
  52          * Add the region from 0x0 to (0x0 + 0x200000 - 1):
  53          *   Expected result:
  54          *       Successful
  55          *
  56          * Now the instance of struct ne_phys_contig_mem_regions is:
  57          *   num = 2
  58          *   regions = {
  59          *       {start=0x200000, end=0x3fffff}, // len=0x200000
  60          *       {start=0x0,      end=0x1fffff}, // len=0x200000
  61          *   }
  62          */
  63         {0x0, 0x200000, 0, 2, 0x0, 0x200000},
  64
  65         /*
  66          * Add the region from 0x600000 to (0x600000 + 0x400000 - 1):
  67          *   Expected result:
  68          *       Successful
  69          *
  70          * Now the instance of struct ne_phys_contig_mem_regions is:
  71          *   num = 3
  72          *   regions = {
  73          *       {start=0x200000, end=0x3fffff}, // len=0x200000
  74          *       {start=0x0,      end=0x1fffff}, // len=0x200000
  75          *       {start=0x600000, end=0x9fffff}, // len=0x400000
  76          *   }
  77          */
  78         {0x600000, 0x400000, 0, 3, 0x600000, 0x400000},
  79
  80         /*
  81          * Add the region from 0xa00000 to (0xa00000 + 0x400000 - 1):
  82          *   Expected result:
  83          *       Successful, merging case!
  84          *
  85          * Now the instance of struct ne_phys_contig_mem_regions is:
  86          *   num = 3
  87          *   regions = {
  88          *       {start=0x200000, end=0x3fffff}, // len=0x200000
  89          *       {start=0x0,      end=0x1fffff}, // len=0x200000
  90          *       {start=0x600000, end=0xdfffff}, // len=0x800000
  91          *   }
  92          */
  93         {0xa00000, 0x400000, 0, 3, 0x600000, 0x800000},
  94
  95         /*
  96          * Add the region from 0x1000 to (0x1000 + 0x200000 - 1):
  97          *   Expected result:
  98          *       Failed, start address is not 2M-aligned
  99          *
 100          * Now the instance of struct ne_phys_contig_mem_regions is:
 101          *   num = 3
 102          *   regions = {
 103          *       {start=0x200000, end=0x3fffff}, // len=0x200000
 104          *       {start=0x0,      end=0x1fffff}, // len=0x200000
 105          *       {start=0x600000, end=0xdfffff}, // len=0x800000
 106          *   }
 107          */
 108         {0x1000, 0x200000, -EINVAL, 3, 0x600000, 0x800000},
 109 };
 110
 111 static void ne_misc_dev_test_merge_phys_contig_memory_regions(struct kunit *test)
 112 {
 113         struct ne_phys_contig_mem_regions phys_contig_mem_regions = {};
 114         int rc = 0;
 115         int i = 0;
 116
 117         phys_contig_mem_regions.regions = kunit_kcalloc(test, MAX_PHYS_REGIONS,
 118                                                         sizeof(*phys_contig_mem_regions.regions),
 119                                                         GFP_KERNEL);
 120         KUNIT_ASSERT_TRUE(test, phys_contig_mem_regions.regions);
 121
 122         for (i = 0; i < ARRAY_SIZE(phys_regions_test_cases); i++) {
 123                 struct ne_phys_regions_test *test_case = &phys_regions_test_cases[i];
 124                 unsigned long num = 0;
 125
 126                 rc = ne_merge_phys_contig_memory_regions(&phys_contig_mem_regions,
 127                                                          test_case->paddr, test_case->size);
 128                 KUNIT_EXPECT_EQ(test, rc, test_case->expect_rc);
 129                 KUNIT_EXPECT_EQ(test, phys_contig_mem_regions.num, test_case->expect_num);
 130
 131                 if (test_case->expect_last_paddr == INVALID_VALUE)
 132                         continue;
 133
 134                 num = phys_contig_mem_regions.num;
 135                 KUNIT_EXPECT_EQ(test, phys_contig_mem_regions.regions[num - 1].start,
 136                                 test_case->expect_last_paddr);
 137                 KUNIT_EXPECT_EQ(test, range_len(&phys_contig_mem_regions.regions[num - 1]),
 138                                 test_case->expect_last_size);
 139         }
 140
 141         kunit_kfree(test, phys_contig_mem_regions.regions);
 142 }
 143
 144 static struct kunit_case ne_misc_dev_test_cases[] = {
 145         KUNIT_CASE(ne_misc_dev_test_merge_phys_contig_memory_regions),
 146         {}
 147 };
 148
 149 static struct kunit_suite ne_misc_dev_test_suite = {
 150         .name = "ne_misc_dev_test",
 151         .test_cases = ne_misc_dev_test_cases,
 152 };
 153
 154 kunit_test_suite(ne_misc_dev_test_suite);