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 : 74737779 : tree_find_buffer(struct cache_tree *tree, uint64_t offset)
14 : : {
15 : : uint64_t index;
16 : :
17 [ + + ]: 150198026 : while (tree != NULL) {
18 [ - + - + : 146210793 : index = offset / CACHE_TREE_LEVEL_SIZE(tree->level);
# # # # #
# # # ]
19 [ + + # # ]: 146210793 : if (index >= CACHE_TREE_WIDTH) {
20 : 17023 : return NULL;
21 : : }
22 [ + + # # : 146193770 : if (tree->level == 0) {
# # ]
23 [ # # # # : 70733523 : return tree->u.buffer[index];
# # # # #
# ]
24 : : } else {
25 [ - + # # : 75460247 : offset &= CACHE_TREE_LEVEL_MASK(tree->level);
# # # # #
# ]
26 [ # # # # : 75460247 : tree = tree->u.tree[index];
# # # # ]
27 : : }
28 : : }
29 : :
30 : 3987233 : return NULL;
31 : 0 : }
32 : :
33 : : struct cache_buffer *
34 : 27777518 : tree_find_filled_buffer(struct cache_tree *tree, uint64_t offset)
35 : : {
36 : : struct cache_buffer *buf;
37 : :
38 : 27777518 : buf = tree_find_buffer(tree, offset);
39 [ + + + + : 27777518 : if (buf != NULL && buf->bytes_filled > 0) {
# # # # ]
40 : 23553205 : return buf;
41 : : } else {
42 : 4224313 : return NULL;
43 : : }
44 : 0 : }
45 : :
46 : : struct cache_tree *
47 : 1081830 : tree_insert_buffer(struct cache_tree *root, struct cache_buffer *buffer)
48 : : {
49 : : struct cache_tree *tree;
50 : : uint64_t index, offset;
51 : :
52 [ # # # # ]: 1081830 : offset = buffer->offset;
53 [ + + + + : 1087407 : while (offset >= CACHE_TREE_LEVEL_SIZE(root->level + 1)) {
# # # # #
# # # #
# ]
54 [ + + # # : 5577 : if (root->present_mask != 0) {
# # ]
55 : 2495 : tree = calloc(1, sizeof(*tree));
56 [ - + # # ]: 2495 : assert(tree != NULL);
57 [ # # # # : 2495 : tree->level = root->level + 1;
# # # # #
# ]
58 [ # # # # : 2495 : tree->u.tree[0] = root;
# # # # ]
59 : 2495 : root = tree;
60 [ # # # # ]: 2495 : root->present_mask = 0x1ULL;
61 : 0 : } else {
62 [ # # ]: 3082 : root->level++;
63 : : }
64 : : }
65 : :
66 : 1081830 : tree = root;
67 [ + + # # : 2011388 : while (tree->level > 0) {
# # ]
68 [ - + - + : 929558 : index = offset / CACHE_TREE_LEVEL_SIZE(tree->level);
# # # # #
# # # ]
69 [ - + # # : 929558 : assert(index < CACHE_TREE_WIDTH);
# # ]
70 [ - + # # : 929558 : offset &= CACHE_TREE_LEVEL_MASK(tree->level);
# # # # #
# ]
71 [ + + # # : 929558 : if (tree->u.tree[index] == NULL) {
# # # # #
# ]
72 [ # # # # : 21434 : tree->u.tree[index] = calloc(1, sizeof(*tree));
# # # # ]
73 [ - + # # : 21434 : assert(tree->u.tree[index] != NULL);
# # # # #
# # # ]
74 [ # # # # : 21434 : tree->u.tree[index]->level = tree->level - 1;
# # # # #
# # # # #
# # # # ]
75 [ - + # # : 21434 : tree->present_mask |= (1ULL << index);
# # ]
76 : 0 : }
77 [ # # # # : 929558 : tree = tree->u.tree[index];
# # # # ]
78 : : }
79 : :
80 [ # # # # ]: 1081830 : index = offset / CACHE_BUFFER_SIZE;
81 [ - + # # : 1081830 : assert(index < CACHE_TREE_WIDTH);
# # ]
82 [ - + # # : 1081830 : assert(tree->u.buffer[index] == NULL);
# # # # #
# # # #
# ]
83 [ # # # # : 1081830 : tree->u.buffer[index] = buffer;
# # # # #
# ]
84 [ - + # # : 1081830 : tree->present_mask |= (1ULL << index);
# # ]
85 : 1081830 : return root;
86 : : }
87 : :
88 : : void
89 : 740600 : tree_remove_buffer(struct cache_tree *tree, struct cache_buffer *buffer)
90 : : {
91 : : struct cache_tree *child;
92 : : uint64_t index;
93 : :
94 [ - + # # : 740600 : index = CACHE_TREE_INDEX(tree->level, buffer->offset);
# # # # #
# # # # #
# # ]
95 : :
96 [ + + # # : 740600 : if (tree->level == 0) {
# # ]
97 [ - + # # : 366955 : assert(tree->u.buffer[index] != NULL);
# # # # #
# # # #
# ]
98 [ - + # # : 366955 : assert(buffer == tree->u.buffer[index]);
# # # # #
# # # #
# ]
99 [ - + # # : 366955 : tree->present_mask &= ~(1ULL << index);
# # ]
100 [ # # # # : 366955 : tree->u.buffer[index] = NULL;
# # # # #
# ]
101 : 366955 : cache_buffer_free(buffer);
102 : 366955 : return;
103 : : }
104 : :
105 [ # # # # : 373645 : child = tree->u.tree[index];
# # # # ]
106 [ - + # # ]: 373645 : assert(child != NULL);
107 : 373645 : tree_remove_buffer(child, buffer);
108 [ + + # # : 373645 : if (child->present_mask == 0) {
# # ]
109 [ - + # # : 5122 : tree->present_mask &= ~(1ULL << index);
# # ]
110 [ # # # # : 5122 : tree->u.tree[index] = NULL;
# # # # ]
111 : 5122 : free(child);
112 : 0 : }
113 : 0 : }
114 : :
115 : : void
116 : 27674 : tree_free_buffers(struct cache_tree *tree)
117 : : {
118 : : struct cache_buffer *buffer;
119 : : struct cache_tree *child;
120 : : uint32_t i;
121 : :
122 [ - + # # : 27674 : if (tree->present_mask == 0) {
# # ]
123 : 0 : return;
124 : : }
125 : :
126 [ + + # # : 27674 : if (tree->level == 0) {
# # ]
127 [ + + # # ]: 1206400 : for (i = 0; i < CACHE_TREE_WIDTH; i++) {
128 [ # # # # : 1187840 : buffer = tree->u.buffer[i];
# # # # #
# ]
129 [ + + + + : 1187840 : if (buffer != NULL && buffer->in_progress == false &&
+ + # # #
# # # ]
130 [ + + # # : 715353 : buffer->bytes_filled == buffer->bytes_flushed) {
# # # # ]
131 : 714881 : cache_buffer_free(buffer);
132 [ # # # # : 714881 : tree->u.buffer[i] = NULL;
# # # # #
# ]
133 [ - + # # : 714881 : tree->present_mask &= ~(1ULL << i);
# # ]
134 : 0 : }
135 : 0 : }
136 : 0 : } else {
137 [ + + # # ]: 592410 : for (i = 0; i < CACHE_TREE_WIDTH; i++) {
138 [ # # # # : 583296 : child = tree->u.tree[i];
# # # # ]
139 [ + + ]: 583296 : if (child != NULL) {
140 : 18869 : tree_free_buffers(child);
141 [ + + # # : 18869 : if (child->present_mask == 0) {
# # ]
142 : 18813 : free(child);
143 [ # # # # : 18813 : tree->u.tree[i] = NULL;
# # # # ]
144 [ - + # # : 18813 : tree->present_mask &= ~(1ULL << i);
# # ]
145 : 0 : }
146 : 0 : }
147 : 0 : }
148 : : }
149 : 0 : }
|