tools/testing/radix-tree/test.c

   1 #include <stdlib.h>
   2 #include <assert.h>
   3 #include <stdio.h>
   4 #include <linux/types.h>
   5 #include <linux/kernel.h>
   6 #include <linux/bitops.h>
   7
   8 #include "test.h"
   9
  10 struct item *
  11 item_tag_set(struct radix_tree_root *root, unsigned long index, int tag)
  12 {
  13         return radix_tree_tag_set(root, index, tag);
  14 }
  15
  16 struct item *
  17 item_tag_clear(struct radix_tree_root *root, unsigned long index, int tag)
  18 {
  19         return radix_tree_tag_clear(root, index, tag);
  20 }
  21
  22 int item_tag_get(struct radix_tree_root *root, unsigned long index, int tag)
  23 {
  24         return radix_tree_tag_get(root, index, tag);
  25 }
  26
  27 int __item_insert(struct radix_tree_root *root, struct item *item)
  28 {
  29         return radix_tree_insert(root, item->index, item);
  30 }
  31
  32 int item_insert(struct radix_tree_root *root, unsigned long index)
  33 {
  34         return __item_insert(root, item_create(index));
  35 }
  36
  37 int item_delete(struct radix_tree_root *root, unsigned long index)
  38 {
  39         struct item *item = radix_tree_delete(root, index);
  40
  41         if (item) {
  42                 assert(item->index == index);
  43                 free(item);
  44                 return 1;
  45         }
  46         return 0;
  47 }
  48
  49 struct item *item_create(unsigned long index)
  50 {
  51         struct item *ret = malloc(sizeof(*ret));
  52
  53         ret->index = index;
  54         return ret;
  55 }
  56
  57 void item_check_present(struct radix_tree_root *root, unsigned long index)
  58 {
  59         struct item *item;
  60
  61         item = radix_tree_lookup(root, index);
  62         assert(item != 0);
  63         assert(item->index == index);
  64 }
  65
  66 struct item *item_lookup(struct radix_tree_root *root, unsigned long index)
  67 {
  68         return radix_tree_lookup(root, index);
  69 }
  70
  71 void item_check_absent(struct radix_tree_root *root, unsigned long index)
  72 {
  73         struct item *item;
  74
  75         item = radix_tree_lookup(root, index);
  76         assert(item == 0);
  77 }
  78
  79 /*
  80  * Scan only the passed (start, start+nr] for present items
  81  */
  82 void item_gang_check_present(struct radix_tree_root *root,
  83                         unsigned long start, unsigned long nr,
  84                         int chunk, int hop)
  85 {
  86         struct item *items[chunk];
  87         unsigned long into;
  88
  89         for (into = 0; into < nr; ) {
  90                 int nfound;
  91                 int nr_to_find = chunk;
  92                 int i;
  93
  94                 if (nr_to_find > (nr - into))
  95                         nr_to_find = nr - into;
  96
  97                 nfound = radix_tree_gang_lookup(root, (void **)items,
  98                                                 start + into, nr_to_find);
  99                 assert(nfound == nr_to_find);
 100                 for (i = 0; i < nfound; i++)
 101                         assert(items[i]->index == start + into + i);
 102                 into += hop;
 103         }
 104 }
 105
 106 /*
 107  * Scan the entire tree, only expecting present items (start, start+nr]
 108  */
 109 void item_full_scan(struct radix_tree_root *root, unsigned long start,
 110                         unsigned long nr, int chunk)
 111 {
 112         struct item *items[chunk];
 113         unsigned long into = 0;
 114         unsigned long this_index = start;
 115         int nfound;
 116         int i;
 117
 118 //      printf("%s(0x%08lx, 0x%08lx, %d)\n", __FUNCTION__, start, nr, chunk);
 119
 120         while ((nfound = radix_tree_gang_lookup(root, (void **)items, into,
 121                                         chunk))) {
 122 //              printf("At 0x%08lx, nfound=%d\n", into, nfound);
 123                 for (i = 0; i < nfound; i++) {
 124                         assert(items[i]->index == this_index);
 125                         this_index++;
 126                 }
 127 //              printf("Found 0x%08lx->0x%08lx\n",
 128 //                      items[0]->index, items[nfound-1]->index);
 129                 into = this_index;
 130         }
 131         if (chunk)
 132                 assert(this_index == start + nr);
 133         nfound = radix_tree_gang_lookup(root, (void **)items,
 134                                         this_index, chunk);
 135         assert(nfound == 0);
 136 }
 137
 138 static int verify_node(struct radix_tree_node *slot, unsigned int tag,
 139                         unsigned int height, int tagged)
 140 {
 141         int anyset = 0;
 142         int i;
 143         int j;
 144
 145         /* Verify consistency at this level */
 146         for (i = 0; i < RADIX_TREE_TAG_LONGS; i++) {
 147                 if (slot->tags[tag][i]) {
 148                         anyset = 1;
 149                         break;
 150                 }
 151         }
 152         if (tagged != anyset) {
 153                 printf("tag: %u, height %u, tagged: %d, anyset: %d\n", tag, height, tagged, anyset);
 154                 for (j = 0; j < RADIX_TREE_MAX_TAGS; j++) {
 155                         printf("tag %d: ", j);
 156                         for (i = 0; i < RADIX_TREE_TAG_LONGS; i++)
 157                                 printf("%016lx ", slot->tags[j][i]);
 158                         printf("\n");
 159                 }
 160                 return 1;
 161         }
 162         assert(tagged == anyset);
 163
 164         /* Go for next level */
 165         if (height > 1) {
 166                 for (i = 0; i < RADIX_TREE_MAP_SIZE; i++)
 167                         if (slot->slots[i])
 168                                 if (verify_node(slot->slots[i], tag, height - 1,
 169                                             !!test_bit(i, slot->tags[tag]))) {
 170                                         printf("Failure at off %d\n", i);
 171                                         for (j = 0; j < RADIX_TREE_MAX_TAGS; j++) {
 172                                                 printf("tag %d: ", j);
 173                                                 for (i = 0; i < RADIX_TREE_TAG_LONGS; i++)
 174                                                         printf("%016lx ", slot->tags[j][i]);
 175                                                 printf("\n");
 176                                         }
 177                                         return 1;
 178                                 }
 179         }
 180         return 0;
 181 }
 182
 183 void verify_tag_consistency(struct radix_tree_root *root, unsigned int tag)
 184 {
 185         if (!root->height)
 186                 return;
 187         verify_node(indirect_to_ptr(root->rnode),
 188                         tag, root->height, !!root_tag_get(root, tag));
 189 }
 190
 191 void item_kill_tree(struct radix_tree_root *root)
 192 {
 193         struct item *items[32];
 194         int nfound;
 195
 196         while ((nfound = radix_tree_gang_lookup(root, (void **)items, 0, 32))) {
 197                 int i;
 198
 199                 for (i = 0; i < nfound; i++) {
 200                         void *ret;
 201
 202                         ret = radix_tree_delete(root, items[i]->index);
 203                         assert(ret == items[i]);
 204                         free(items[i]);
 205                 }
 206         }
 207         assert(radix_tree_gang_lookup(root, (void **)items, 0, 32) == 0);
 208         assert(root->rnode == NULL);
 209 }
 210
 211 void tree_verify_min_height(struct radix_tree_root *root, int maxindex)
 212 {
 213         assert(radix_tree_maxindex(root->height) >= maxindex);
 214         if (root->height > 1)
 215                 assert(radix_tree_maxindex(root->height-1) < maxindex);
 216         else if (root->height == 1)
 217                 assert(radix_tree_maxindex(root->height-1) <= maxindex);
 218 }