Branch data Line data Source code
1 : : /* SPDX-License-Identifier: BSD-3-Clause
2 : : * Copyright (C) 2017 Intel Corporation.
3 : : * All rights reserved.
4 : : */
5 : :
6 : : #include "spdk/stdinc.h"
7 : :
8 : : #include "cache_tree.h"
9 : :
10 : : #include "spdk/assert.h"
11 : :
12 : : struct cache_buffer *
13 : 59888415 : tree_find_buffer(struct cache_tree *tree, uint64_t offset)
14 : : {
15 : : uint64_t index;
16 : :
17 [ + + ]: 120430100 : while (tree != NULL) {
18 [ - + - + : 116624377 : index = offset / CACHE_TREE_LEVEL_SIZE(tree->level);
# # # # #
# # # ]
19 [ + + # # ]: 116624377 : if (index >= CACHE_TREE_WIDTH) {
20 : 31990 : return NULL;
21 : : }
22 [ + + # # : 116592387 : if (tree->level == 0) {
# # ]
23 [ # # # # : 56050702 : return tree->u.buffer[index];
# # # # #
# ]
24 : : } else {
25 [ - + # # : 60541685 : offset &= CACHE_TREE_LEVEL_MASK(tree->level);
# # # # #
# ]
26 [ # # # # : 60541685 : tree = tree->u.tree[index];
# # # # ]
27 : : }
28 : : }
29 : :
30 : 3805723 : return NULL;
31 : 0 : }
32 : :
33 : : struct cache_buffer *
34 : 23069734 : tree_find_filled_buffer(struct cache_tree *tree, uint64_t offset)
35 : : {
36 : : struct cache_buffer *buf;
37 : :
38 : 23069734 : buf = tree_find_buffer(tree, offset);
39 [ + + + + : 23069734 : if (buf != NULL && buf->bytes_filled > 0) {
# # # # ]
40 : 18657100 : return buf;
41 : : } else {
42 : 4412634 : return NULL;
43 : : }
44 : 0 : }
45 : :
46 : : struct cache_tree *
47 : 942602 : tree_insert_buffer(struct cache_tree *root, struct cache_buffer *buffer)
48 : : {
49 : : struct cache_tree *tree;
50 : : uint64_t index, offset;
51 : :
52 [ # # # # ]: 942602 : offset = buffer->offset;
53 [ + + + + : 947834 : while (offset >= CACHE_TREE_LEVEL_SIZE(root->level + 1)) {
# # # # #
# # # #
# ]
54 [ + + # # : 5232 : if (root->present_mask != 0) {
# # ]
55 : 2233 : tree = calloc(1, sizeof(*tree));
56 [ - + # # ]: 2233 : assert(tree != NULL);
57 [ # # # # : 2233 : tree->level = root->level + 1;
# # # # #
# ]
58 [ # # # # : 2233 : tree->u.tree[0] = root;
# # # # ]
59 : 2233 : root = tree;
60 [ # # # # ]: 2233 : root->present_mask = 0x1ULL;
61 : 0 : } else {
62 [ # # ]: 2999 : root->level++;
63 : : }
64 : : }
65 : :
66 : 942602 : tree = root;
67 [ + + # # : 1750597 : while (tree->level > 0) {
# # ]
68 [ - + - + : 807995 : index = offset / CACHE_TREE_LEVEL_SIZE(tree->level);
# # # # #
# # # ]
69 [ - + # # : 807995 : assert(index < CACHE_TREE_WIDTH);
# # ]
70 [ - + # # : 807995 : offset &= CACHE_TREE_LEVEL_MASK(tree->level);
# # # # #
# ]
71 [ + + # # : 807995 : if (tree->u.tree[index] == NULL) {
# # # # #
# ]
72 [ # # # # : 19994 : tree->u.tree[index] = calloc(1, sizeof(*tree));
# # # # ]
73 [ - + # # : 19994 : assert(tree->u.tree[index] != NULL);
# # # # #
# # # ]
74 [ # # # # : 19994 : tree->u.tree[index]->level = tree->level - 1;
# # # # #
# # # # #
# # # # ]
75 [ - + # # : 19994 : tree->present_mask |= (1ULL << index);
# # ]
76 : 0 : }
77 [ # # # # : 807995 : tree = tree->u.tree[index];
# # # # ]
78 : : }
79 : :
80 [ # # # # ]: 942602 : index = offset / CACHE_BUFFER_SIZE;
81 [ - + # # : 942602 : assert(index < CACHE_TREE_WIDTH);
# # ]
82 [ - + # # : 942602 : assert(tree->u.buffer[index] == NULL);
# # # # #
# # # #
# ]
83 [ # # # # : 942602 : tree->u.buffer[index] = buffer;
# # # # #
# ]
84 [ - + # # : 942602 : tree->present_mask |= (1ULL << index);
# # ]
85 : 942602 : return root;
86 : : }
87 : :
88 : : void
89 : 587565 : tree_remove_buffer(struct cache_tree *tree, struct cache_buffer *buffer)
90 : : {
91 : : struct cache_tree *child;
92 : : uint64_t index;
93 : :
94 [ - + # # : 587565 : index = CACHE_TREE_INDEX(tree->level, buffer->offset);
# # # # #
# # # # #
# # ]
95 : :
96 [ + + # # : 587565 : if (tree->level == 0) {
# # ]
97 [ - + # # : 290694 : assert(tree->u.buffer[index] != NULL);
# # # # #
# # # #
# ]
98 [ - + # # : 290694 : assert(buffer == tree->u.buffer[index]);
# # # # #
# # # #
# ]
99 [ - + # # : 290694 : tree->present_mask &= ~(1ULL << index);
# # ]
100 [ # # # # : 290694 : tree->u.buffer[index] = NULL;
# # # # #
# ]
101 : 290694 : cache_buffer_free(buffer);
102 : 290694 : return;
103 : : }
104 : :
105 [ # # # # : 296871 : child = tree->u.tree[index];
# # # # ]
106 [ - + # # ]: 296871 : assert(child != NULL);
107 : 296871 : tree_remove_buffer(child, buffer);
108 [ + + # # : 296871 : if (child->present_mask == 0) {
# # ]
109 [ - + # # : 4094 : tree->present_mask &= ~(1ULL << index);
# # ]
110 [ # # # # : 4094 : tree->u.tree[index] = NULL;
# # # # ]
111 : 4094 : free(child);
112 : 0 : }
113 : 0 : }
114 : :
115 : : void
116 : 27494 : tree_free_buffers(struct cache_tree *tree)
117 : : {
118 : : struct cache_buffer *buffer;
119 : : struct cache_tree *child;
120 : : uint32_t i;
121 : :
122 [ - + # # : 27494 : if (tree->present_mask == 0) {
# # ]
123 : 0 : return;
124 : : }
125 : :
126 [ + + # # : 27494 : if (tree->level == 0) {
# # ]
127 [ + + # # ]: 1168245 : for (i = 0; i < CACHE_TREE_WIDTH; i++) {
128 [ # # # # : 1150272 : buffer = tree->u.buffer[i];
# # # # #
# ]
129 [ + + + + : 1150272 : if (buffer != NULL && buffer->in_progress == false &&
+ + # # #
# # # ]
130 [ + + # # : 651915 : buffer->bytes_filled == buffer->bytes_flushed) {
# # # # ]
131 : 651914 : cache_buffer_free(buffer);
132 [ # # # # : 651914 : tree->u.buffer[i] = NULL;
# # # # #
# ]
133 [ - + # # : 651914 : tree->present_mask &= ~(1ULL << i);
# # ]
134 : 0 : }
135 : 0 : }
136 : 0 : } else {
137 [ + + # # ]: 618865 : for (i = 0; i < CACHE_TREE_WIDTH; i++) {
138 [ # # # # : 609344 : child = tree->u.tree[i];
# # # # ]
139 [ + + ]: 609344 : if (child != NULL) {
140 : 18169 : tree_free_buffers(child);
141 [ + + # # : 18169 : if (child->present_mask == 0) {
# # ]
142 : 18139 : free(child);
143 [ # # # # : 18139 : tree->u.tree[i] = NULL;
# # # # ]
144 [ - + # # : 18139 : tree->present_mask &= ~(1ULL << i);
# # ]
145 : 0 : }
146 : 0 : }
147 : 0 : }
148 : : }
149 : 0 : }
|