Sage Weil | 5ecc0a0 | 2009-10-06 11:31:11 -0700 | [diff] [blame] | 1 | #ifdef __KERNEL__ |
| 2 | # include <linux/slab.h> |
Ilya Dryomov | b459be7 | 2015-06-12 13:21:07 +0300 | [diff] [blame] | 3 | # include <linux/crush/crush.h> |
Ilya Dryomov | 5cf9c4a | 2017-06-22 19:44:05 +0200 | [diff] [blame^] | 4 | void clear_choose_args(struct crush_map *c); |
Sage Weil | 5ecc0a0 | 2009-10-06 11:31:11 -0700 | [diff] [blame] | 5 | #else |
Ilya Dryomov | b459be7 | 2015-06-12 13:21:07 +0300 | [diff] [blame] | 6 | # include "crush_compat.h" |
| 7 | # include "crush.h" |
Sage Weil | 5ecc0a0 | 2009-10-06 11:31:11 -0700 | [diff] [blame] | 8 | #endif |
| 9 | |
Sage Weil | c6cf726 | 2009-11-06 16:39:26 -0800 | [diff] [blame] | 10 | const char *crush_bucket_alg_name(int alg) |
| 11 | { |
| 12 | switch (alg) { |
| 13 | case CRUSH_BUCKET_UNIFORM: return "uniform"; |
| 14 | case CRUSH_BUCKET_LIST: return "list"; |
| 15 | case CRUSH_BUCKET_TREE: return "tree"; |
| 16 | case CRUSH_BUCKET_STRAW: return "straw"; |
Ilya Dryomov | 958a276 | 2015-04-14 16:54:52 +0300 | [diff] [blame] | 17 | case CRUSH_BUCKET_STRAW2: return "straw2"; |
Sage Weil | c6cf726 | 2009-11-06 16:39:26 -0800 | [diff] [blame] | 18 | default: return "unknown"; |
| 19 | } |
| 20 | } |
| 21 | |
Sage Weil | 5ecc0a0 | 2009-10-06 11:31:11 -0700 | [diff] [blame] | 22 | /** |
| 23 | * crush_get_bucket_item_weight - Get weight of an item in given bucket |
| 24 | * @b: bucket pointer |
| 25 | * @p: item index in bucket |
| 26 | */ |
Sage Weil | 8b12d47 | 2012-05-07 15:38:35 -0700 | [diff] [blame] | 27 | int crush_get_bucket_item_weight(const struct crush_bucket *b, int p) |
Sage Weil | 5ecc0a0 | 2009-10-06 11:31:11 -0700 | [diff] [blame] | 28 | { |
Sage Weil | 8b12d47 | 2012-05-07 15:38:35 -0700 | [diff] [blame] | 29 | if ((__u32)p >= b->size) |
Sage Weil | 5ecc0a0 | 2009-10-06 11:31:11 -0700 | [diff] [blame] | 30 | return 0; |
| 31 | |
| 32 | switch (b->alg) { |
| 33 | case CRUSH_BUCKET_UNIFORM: |
| 34 | return ((struct crush_bucket_uniform *)b)->item_weight; |
| 35 | case CRUSH_BUCKET_LIST: |
| 36 | return ((struct crush_bucket_list *)b)->item_weights[p]; |
| 37 | case CRUSH_BUCKET_TREE: |
Sage Weil | f671d4c | 2012-05-07 15:36:49 -0700 | [diff] [blame] | 38 | return ((struct crush_bucket_tree *)b)->node_weights[crush_calc_tree_node(p)]; |
Sage Weil | 5ecc0a0 | 2009-10-06 11:31:11 -0700 | [diff] [blame] | 39 | case CRUSH_BUCKET_STRAW: |
| 40 | return ((struct crush_bucket_straw *)b)->item_weights[p]; |
Ilya Dryomov | 958a276 | 2015-04-14 16:54:52 +0300 | [diff] [blame] | 41 | case CRUSH_BUCKET_STRAW2: |
| 42 | return ((struct crush_bucket_straw2 *)b)->item_weights[p]; |
Sage Weil | 5ecc0a0 | 2009-10-06 11:31:11 -0700 | [diff] [blame] | 43 | } |
| 44 | return 0; |
| 45 | } |
| 46 | |
Sage Weil | 5ecc0a0 | 2009-10-06 11:31:11 -0700 | [diff] [blame] | 47 | void crush_destroy_bucket_uniform(struct crush_bucket_uniform *b) |
| 48 | { |
Sage Weil | 5ecc0a0 | 2009-10-06 11:31:11 -0700 | [diff] [blame] | 49 | kfree(b->h.items); |
| 50 | kfree(b); |
| 51 | } |
| 52 | |
| 53 | void crush_destroy_bucket_list(struct crush_bucket_list *b) |
| 54 | { |
| 55 | kfree(b->item_weights); |
| 56 | kfree(b->sum_weights); |
Sage Weil | 5ecc0a0 | 2009-10-06 11:31:11 -0700 | [diff] [blame] | 57 | kfree(b->h.items); |
| 58 | kfree(b); |
| 59 | } |
| 60 | |
| 61 | void crush_destroy_bucket_tree(struct crush_bucket_tree *b) |
| 62 | { |
Sage Weil | 6eb43f4 | 2012-05-07 15:37:05 -0700 | [diff] [blame] | 63 | kfree(b->h.items); |
Sage Weil | 5ecc0a0 | 2009-10-06 11:31:11 -0700 | [diff] [blame] | 64 | kfree(b->node_weights); |
| 65 | kfree(b); |
| 66 | } |
| 67 | |
| 68 | void crush_destroy_bucket_straw(struct crush_bucket_straw *b) |
| 69 | { |
| 70 | kfree(b->straws); |
| 71 | kfree(b->item_weights); |
Sage Weil | 5ecc0a0 | 2009-10-06 11:31:11 -0700 | [diff] [blame] | 72 | kfree(b->h.items); |
| 73 | kfree(b); |
| 74 | } |
| 75 | |
Ilya Dryomov | 958a276 | 2015-04-14 16:54:52 +0300 | [diff] [blame] | 76 | void crush_destroy_bucket_straw2(struct crush_bucket_straw2 *b) |
| 77 | { |
| 78 | kfree(b->item_weights); |
Ilya Dryomov | 958a276 | 2015-04-14 16:54:52 +0300 | [diff] [blame] | 79 | kfree(b->h.items); |
| 80 | kfree(b); |
| 81 | } |
| 82 | |
Sage Weil | 5ecc0a0 | 2009-10-06 11:31:11 -0700 | [diff] [blame] | 83 | void crush_destroy_bucket(struct crush_bucket *b) |
| 84 | { |
| 85 | switch (b->alg) { |
| 86 | case CRUSH_BUCKET_UNIFORM: |
| 87 | crush_destroy_bucket_uniform((struct crush_bucket_uniform *)b); |
| 88 | break; |
| 89 | case CRUSH_BUCKET_LIST: |
| 90 | crush_destroy_bucket_list((struct crush_bucket_list *)b); |
| 91 | break; |
| 92 | case CRUSH_BUCKET_TREE: |
| 93 | crush_destroy_bucket_tree((struct crush_bucket_tree *)b); |
| 94 | break; |
| 95 | case CRUSH_BUCKET_STRAW: |
| 96 | crush_destroy_bucket_straw((struct crush_bucket_straw *)b); |
| 97 | break; |
Ilya Dryomov | 958a276 | 2015-04-14 16:54:52 +0300 | [diff] [blame] | 98 | case CRUSH_BUCKET_STRAW2: |
| 99 | crush_destroy_bucket_straw2((struct crush_bucket_straw2 *)b); |
| 100 | break; |
Sage Weil | 5ecc0a0 | 2009-10-06 11:31:11 -0700 | [diff] [blame] | 101 | } |
| 102 | } |
| 103 | |
| 104 | /** |
| 105 | * crush_destroy - Destroy a crush_map |
| 106 | * @map: crush_map pointer |
| 107 | */ |
| 108 | void crush_destroy(struct crush_map *map) |
| 109 | { |
Sage Weil | 5ecc0a0 | 2009-10-06 11:31:11 -0700 | [diff] [blame] | 110 | /* buckets */ |
| 111 | if (map->buckets) { |
Sage Weil | 8b12d47 | 2012-05-07 15:38:35 -0700 | [diff] [blame] | 112 | __s32 b; |
Sage Weil | 5ecc0a0 | 2009-10-06 11:31:11 -0700 | [diff] [blame] | 113 | for (b = 0; b < map->max_buckets; b++) { |
| 114 | if (map->buckets[b] == NULL) |
| 115 | continue; |
| 116 | crush_destroy_bucket(map->buckets[b]); |
| 117 | } |
| 118 | kfree(map->buckets); |
| 119 | } |
| 120 | |
| 121 | /* rules */ |
| 122 | if (map->rules) { |
Sage Weil | 8b12d47 | 2012-05-07 15:38:35 -0700 | [diff] [blame] | 123 | __u32 b; |
Sage Weil | 5ecc0a0 | 2009-10-06 11:31:11 -0700 | [diff] [blame] | 124 | for (b = 0; b < map->max_rules; b++) |
Ilya Dryomov | bfb16d7 | 2013-12-24 21:19:24 +0200 | [diff] [blame] | 125 | crush_destroy_rule(map->rules[b]); |
Sage Weil | 5ecc0a0 | 2009-10-06 11:31:11 -0700 | [diff] [blame] | 126 | kfree(map->rules); |
| 127 | } |
| 128 | |
Ilya Dryomov | b459be7 | 2015-06-12 13:21:07 +0300 | [diff] [blame] | 129 | #ifndef __KERNEL__ |
| 130 | kfree(map->choose_tries); |
Ilya Dryomov | 5cf9c4a | 2017-06-22 19:44:05 +0200 | [diff] [blame^] | 131 | #else |
| 132 | clear_choose_args(map); |
Ilya Dryomov | b459be7 | 2015-06-12 13:21:07 +0300 | [diff] [blame] | 133 | #endif |
Sage Weil | 5ecc0a0 | 2009-10-06 11:31:11 -0700 | [diff] [blame] | 134 | kfree(map); |
| 135 | } |
| 136 | |
Ilya Dryomov | bfb16d7 | 2013-12-24 21:19:24 +0200 | [diff] [blame] | 137 | void crush_destroy_rule(struct crush_rule *rule) |
| 138 | { |
| 139 | kfree(rule); |
| 140 | } |