File: | tree.c |
Location: | line 21, column 9 |
Description: | Access to field 'left' results in a dereference of a null pointer (loaded from variable 'right_child') |
1 | /* Splay tree implementation. */ | |||
2 | #include <string.h> | |||
3 | #include <stdlib.h> | |||
4 | #include <stdio.h> | |||
5 | #include "lz4.h" | |||
6 | #include "oleg.h" | |||
7 | #include "tree.h" | |||
8 | #include "logging.h" | |||
9 | #include "stack.h" | |||
10 | #include "errhandle.h" | |||
11 | #include "cursor.h" | |||
12 | ||||
13 | static inline void _ols_left_rotate(ol_splay_tree *tree, ol_splay_tree_node *node) { | |||
14 | ol_splay_tree_node *right_child = node->right; | |||
15 | ||||
16 | if (right_child != NULL((void *)0)) { | |||
17 | node->right = right_child->left; | |||
18 | right_child->parent = node->parent; | |||
19 | } | |||
20 | ||||
21 | if (right_child->left != NULL((void *)0)) | |||
| ||||
22 | right_child->left->parent = node; | |||
23 | ||||
24 | if (!node->parent) | |||
25 | tree->root = right_child; | |||
26 | else if (node == node->parent->left) | |||
27 | node->parent->left = right_child; | |||
28 | else | |||
29 | node->parent->right = right_child; | |||
30 | ||||
31 | right_child->left = node; | |||
32 | node->parent = right_child; | |||
33 | } | |||
34 | ||||
35 | static inline void _ols_right_rotate(ol_splay_tree *tree, ol_splay_tree_node *node) { | |||
36 | ol_splay_tree_node *left_child = node->left; | |||
37 | ||||
38 | if (left_child != NULL((void *)0)) { | |||
39 | node->left = left_child->right; | |||
40 | left_child->parent = node->parent; | |||
41 | } | |||
42 | ||||
43 | if (left_child->right != NULL((void *)0)) | |||
44 | left_child->right->parent = node; | |||
45 | ||||
46 | if (!node->parent) | |||
47 | tree->root = left_child; | |||
48 | else if (node == node->parent->right) | |||
49 | node->parent->right = left_child; | |||
50 | else | |||
51 | node->parent->left = left_child; | |||
52 | ||||
53 | left_child->right = node; | |||
54 | node->parent = left_child; | |||
55 | } | |||
56 | ||||
57 | static inline void _ols_splay(ol_splay_tree *tree, ol_splay_tree_node *node) { | |||
58 | while (node->parent) { | |||
59 | if (!node->parent->parent) { | |||
60 | if (node->parent->left == node) _ols_right_rotate(tree, node->parent); | |||
61 | else _ols_left_rotate(tree, node->parent); | |||
62 | } else if (node->parent->left == node && node->parent->parent->left == node->parent) { | |||
63 | _ols_right_rotate(tree, node->parent->parent); | |||
64 | _ols_right_rotate(tree, node->parent); | |||
65 | } else if (node->parent->right == node && node->parent->parent->right == node->parent) { | |||
66 | _ols_left_rotate(tree, node->parent->parent); | |||
67 | _ols_left_rotate(tree, node->parent); | |||
68 | } else if (node->parent->left == node && node->parent->parent->right == node->parent) { | |||
69 | _ols_right_rotate(tree, node->parent); | |||
70 | _ols_left_rotate(tree, node->parent); | |||
71 | } else { | |||
72 | _ols_left_rotate(tree, node->parent); | |||
73 | _ols_right_rotate(tree, node->parent); | |||
74 | } | |||
75 | } | |||
76 | } | |||
77 | ||||
78 | static inline void _ols_replace(ol_splay_tree *tree, | |||
79 | ol_splay_tree_node *node_a, ol_splay_tree_node *node_b) { | |||
80 | ||||
81 | if (!node_a->parent) | |||
82 | tree->root = node_b; | |||
83 | else if (node_a == node_a->parent->left) | |||
84 | node_a->parent->left = node_b; | |||
85 | else | |||
86 | node_a->parent->right = node_b; | |||
87 | ||||
88 | if (node_b) | |||
89 | node_b->parent = node_a->parent; | |||
90 | } | |||
91 | ||||
92 | ol_splay_tree_node *ols_subtree_minimum(ol_splay_tree_node *node) { | |||
93 | while (node->left != NULL((void *)0)) { | |||
94 | node = node->left; | |||
95 | } | |||
96 | return node; | |||
97 | } | |||
98 | ||||
99 | ol_splay_tree_node *ols_subtree_maximum(ol_splay_tree_node *node) { | |||
100 | while (node->right != NULL((void *)0)) { | |||
101 | node = node->right; | |||
102 | } | |||
103 | return node; | |||
104 | } | |||
105 | ||||
106 | ol_splay_tree_node *ols_insert(ol_splay_tree *tree, const char *key, const size_t klen, const void *ref_obj) { | |||
107 | check(klen < KEY_SIZE, "Key is too long.")if(!(klen < 250)){fprintf(__stderrp,"[ERROR] (%s:%d: errno: %s) " "Key is too long." "\n", "src/tree.c",107,((* __error()) == 0 ? "None" : strerror((* __error())))); (* __error())=0; goto error ;}; | |||
108 | check(key != NULL, "Key is null.")if(!(key != ((void *)0))){fprintf(__stderrp,"[ERROR] (%s:%d: errno: %s) " "Key is null." "\n", "src/tree.c",108,((* __error()) == 0 ? "None" : strerror((* __error())))); (* __error())=0; goto error;}; | |||
109 | ol_splay_tree_node *current_node = NULL((void *)0), *previous_node = NULL((void *)0); | |||
110 | current_node = tree->root; | |||
111 | ||||
112 | while (current_node != NULL((void *)0)) { | |||
113 | size_t larger_key = 0; | |||
114 | previous_node = current_node; | |||
115 | larger_key = klen > current_node->klen ? klen : current_node->klen; | |||
116 | if (strncmp(current_node->key, key, larger_key) >= 0) | |||
117 | current_node = current_node->right; | |||
118 | else | |||
119 | current_node = current_node->left; | |||
120 | } | |||
121 | ||||
122 | current_node = malloc(sizeof(ol_splay_tree_node)); | |||
123 | current_node->left = NULL((void *)0); | |||
124 | current_node->right = NULL((void *)0); | |||
125 | current_node->parent = NULL((void *)0); | |||
126 | current_node->klen = 0; | |||
127 | memset(current_node->key, '\0', KEY_SIZE250); | |||
128 | if (strncpy(current_node->key, key, klen) != current_node->key) { | |||
129 | free(current_node); | |||
130 | return NULL((void *)0); | |||
131 | } | |||
132 | current_node->klen = klen; | |||
133 | current_node->ref_obj = ref_obj; | |||
134 | /* Put that shit into the tree */ | |||
135 | current_node->parent = previous_node; | |||
136 | ||||
137 | /* Figure out how current_node relates to previous_node */ | |||
138 | if (previous_node == NULL((void *)0)) | |||
139 | tree->root = current_node; | |||
140 | else { | |||
141 | size_t larger_key = 0; | |||
142 | larger_key = current_node->klen > previous_node->klen ? | |||
143 | current_node->klen : previous_node->klen; | |||
144 | if (strncmp(previous_node->key, current_node->key, larger_key) >= 0) | |||
145 | previous_node->right = current_node; | |||
146 | else | |||
147 | previous_node->left = current_node; | |||
148 | } | |||
149 | ||||
150 | /* Splay the node to the top. */ | |||
151 | _ols_splay(tree, current_node); | |||
152 | tree->rcrd_cnt++; | |||
153 | ||||
154 | return current_node; | |||
155 | ||||
156 | error: | |||
157 | return NULL((void *)0); | |||
158 | } | |||
159 | int ols_find_and_delete(ol_splay_tree *tree, const char *key, const size_t klen) { | |||
160 | ol_splay_tree_node *node = ols_find(tree, key, klen); | |||
161 | return ols_delete(tree, node); | |||
162 | } | |||
163 | ||||
164 | int ols_delete(ol_splay_tree *tree, ol_splay_tree_node *node) { | |||
165 | if (!node) | |||
| ||||
166 | return 1; | |||
167 | ||||
168 | _ols_splay(tree, node); | |||
169 | ||||
170 | if (!node->left) | |||
171 | _ols_replace(tree, node, node->right); | |||
172 | else if (!node->right) | |||
173 | _ols_replace(tree, node, node->left); | |||
174 | else { | |||
175 | ol_splay_tree_node *minimum_node = ols_subtree_minimum(node->right); | |||
176 | if (minimum_node->parent != node) { | |||
177 | _ols_replace(tree, minimum_node, minimum_node->right); | |||
178 | minimum_node->right = node->right; | |||
179 | minimum_node->right->parent = minimum_node; | |||
180 | } | |||
181 | _ols_replace(tree, node, minimum_node); | |||
182 | minimum_node->left = node->left; | |||
183 | minimum_node->left->parent = minimum_node; | |||
184 | } | |||
185 | ||||
186 | node->parent = NULL((void *)0); | |||
187 | node->left = NULL((void *)0); | |||
188 | node->right = NULL((void *)0); | |||
189 | ||||
190 | free(node); | |||
191 | tree->rcrd_cnt--; | |||
192 | return 0; | |||
193 | } | |||
194 | ||||
195 | ol_splay_tree_node *ols_find(ol_splay_tree *tree, const char *key, size_t klen) { | |||
196 | check(klen < KEY_SIZE, "Key is too long.")if(!(klen < 250)){fprintf(__stderrp,"[ERROR] (%s:%d: errno: %s) " "Key is too long." "\n", "src/tree.c",196,((* __error()) == 0 ? "None" : strerror((* __error())))); (* __error())=0; goto error ;}; | |||
197 | check(key != NULL, "Key is null.")if(!(key != ((void *)0))){fprintf(__stderrp,"[ERROR] (%s:%d: errno: %s) " "Key is null." "\n", "src/tree.c",197,((* __error()) == 0 ? "None" : strerror((* __error())))); (* __error())=0; goto error;}; | |||
198 | ol_splay_tree_node *current_node = tree->root; | |||
199 | ||||
200 | while (current_node) { | |||
201 | size_t larger_key = 0; | |||
202 | larger_key = current_node->klen > klen ? | |||
203 | current_node->klen : klen; | |||
204 | const int result = strncmp(current_node->key, key, larger_key); | |||
205 | if (result > 0) | |||
206 | current_node = current_node->right; | |||
207 | else if (result < 0) | |||
208 | current_node = current_node->left; | |||
209 | else | |||
210 | return current_node; | |||
211 | } | |||
212 | ||||
213 | return NULL((void *)0); | |||
214 | ||||
215 | error: | |||
216 | return NULL((void *)0); | |||
217 | } | |||
218 | ||||
219 | static inline void _ols_free_node(ol_splay_tree_node *node) { | |||
220 | ol_stack *stack = NULL((void *)0); | |||
221 | stack = malloc(sizeof(ol_stack)); | |||
222 | check_mem(stack)if(!((stack))){fprintf(__stderrp,"[ERROR] (%s:%d: errno: %s) " "Out of memory." "\n", "src/tree.c",222,((* __error()) == 0 ? "None" : strerror((* __error())))); (* __error())=0; goto error ;}; | |||
223 | ||||
224 | stack->next = NULL((void *)0); | |||
225 | stack->data = NULL((void *)0); | |||
226 | ||||
227 | spush(&stack, (void *)node); | |||
228 | ||||
229 | int iters = 0; | |||
230 | ol_log_msg(LOG_INFO0, "Clearing tree."); | |||
231 | while (stack->next != NULL((void *)0)) { | |||
232 | iters++; | |||
233 | ol_splay_tree_node *cur_node = (ol_splay_tree_node *)spop(&stack); | |||
234 | ||||
235 | if (cur_node->left != NULL((void *)0)) { | |||
236 | spush(&stack, (void *)cur_node->left); | |||
237 | } | |||
238 | if (cur_node->right != NULL((void *)0)) { | |||
239 | spush(&stack, (void *)cur_node->right); | |||
240 | } | |||
241 | free(cur_node); | |||
242 | } | |||
243 | ol_log_msg(LOG_INFO0, "Tree cleared. Iterations: %i", iters); | |||
244 | free(stack); | |||
245 | ||||
246 | error: | |||
247 | return; | |||
248 | } | |||
249 | ||||
250 | void ols_close(ol_splay_tree *tree) { | |||
251 | if (tree->root == NULL((void *)0)) | |||
252 | return; | |||
253 | _ols_free_node(tree->root); | |||
254 | tree->root = NULL((void *)0); | |||
255 | } | |||
256 | ||||
257 | /* Defined in oleg.h */ | |||
258 | int ol_prefix_match(ol_database *db, const char *prefix, size_t plen, ol_val_array *data) { | |||
259 | if (!db->is_enabled(OL_F_SPLAYTREE, &db->feature_set)) | |||
260 | return -1; | |||
261 | if (!prefix) | |||
262 | return -1; | |||
263 | ||||
264 | ol_cursor cursor; | |||
265 | char **to_return = NULL((void *)0); | |||
266 | char *dest = NULL((void *)0); | |||
267 | ol_stack *matches = NULL((void *)0); | |||
268 | ||||
269 | /* Build cursor */ | |||
270 | check(olc_init(db, &cursor), "Could not init cursor.")if(!(olc_init(db, &cursor))){fprintf(__stderrp,"[ERROR] (%s:%d: errno: %s) " "Could not init cursor." "\n", "src/tree.c",270,((* __error( )) == 0 ? "None" : strerror((* __error())))); (* __error())=0 ; goto error;}; | |||
271 | ||||
272 | /* Get current node */ | |||
273 | ol_splay_tree_node *current_node = _olc_get_node(&cursor); | |||
274 | /* Build up our matches stack */ | |||
275 | matches = malloc(sizeof(ol_stack)); | |||
276 | matches->data = NULL((void *)0); | |||
277 | matches->next = NULL((void *)0); | |||
278 | ||||
279 | int imatches = 0; | |||
280 | int saw_bigger_value = 0; | |||
281 | ||||
282 | while (current_node != NULL((void *)0)) { | |||
283 | const int match_result = strncmp(current_node->key, prefix, plen); | |||
284 | if (match_result == 0) { | |||
285 | spush(&matches, current_node); | |||
286 | imatches++; | |||
287 | } else if (saw_bigger_value && match_result > 0) { | |||
288 | /* We've previously seen a bigger value and now we see one | |||
289 | * again. Just quit. */ | |||
290 | break; | |||
291 | } | |||
292 | ||||
293 | if (match_result > 0) { | |||
294 | /* Flip the bit that says we saw a value bigger than the prefix. We | |||
295 | * should now only recurse to the current subtree minimum. */ | |||
296 | saw_bigger_value = 1; | |||
297 | } | |||
298 | ||||
299 | ||||
300 | if (!olc_step(&cursor)) | |||
301 | break; | |||
302 | current_node = _olc_get_node(&cursor); | |||
303 | } | |||
304 | debug("Found %i matches.", imatches); | |||
305 | ||||
306 | /* No pointer in doing anything else if we don't have any matches. */ | |||
307 | check(imatches > 0, "No matched keys.")if(!(imatches > 0)){fprintf(__stderrp,"[ERROR] (%s:%d: errno: %s) " "No matched keys." "\n", "src/tree.c",307,((* __error()) == 0 ? "None" : strerror((* __error())))); (* __error())=0; goto error ;}; | |||
308 | ||||
309 | /* Compute size of everything and malloc it here */ | |||
310 | size_t total_size = sizeof(char *) * imatches; | |||
311 | to_return = malloc(total_size); | |||
312 | check_mem(to_return)if(!((to_return))){fprintf(__stderrp,"[ERROR] (%s:%d: errno: %s) " "Out of memory." "\n", "src/tree.c",312,((* __error()) == 0 ? "None" : strerror((* __error())))); (* __error())=0; goto error ;}; | |||
313 | ||||
314 | int i; | |||
315 | for (i = 0;i < imatches; i++) { | |||
316 | ol_splay_tree_node *cur_node = (ol_splay_tree_node *)spop(&matches); | |||
317 | ol_bucket *deref = (ol_bucket *)cur_node->ref_obj; | |||
318 | ||||
319 | unsigned char *data_ptr = db->values + deref->data_offset; | |||
320 | size_t data_len = 0; | |||
321 | data_len = deref->original_size; | |||
322 | ||||
323 | dest = malloc(data_len); | |||
324 | check_mem(dest)if(!((dest))){fprintf(__stderrp,"[ERROR] (%s:%d: errno: %s) " "Out of memory." "\n", "src/tree.c",324,((* __error()) == 0 ? "None" : strerror((* __error())))); (* __error())=0; goto error ;}; | |||
325 | if (db->is_enabled(OL_F_LZ4, &db->feature_set)) { | |||
326 | int processed = 0; | |||
327 | processed = LZ4_decompress_fast((char *)data_ptr, dest, data_len); | |||
328 | check(processed == deref->data_size, "Could not decompress data.")if(!(processed == deref->data_size)){fprintf(__stderrp,"[ERROR] (%s:%d: errno: %s) " "Could not decompress data." "\n", "src/tree.c",328,((* __error ()) == 0 ? "None" : strerror((* __error())))); (* __error())= 0; goto error;}; | |||
329 | } else { | |||
330 | unsigned char *to_check = memcpy(data_ptr, dest, data_len); | |||
331 | check(to_check == data_ptr, "Could not copy data to msgpuck buffer.")if(!(to_check == data_ptr)){fprintf(__stderrp,"[ERROR] (%s:%d: errno: %s) " "Could not copy data to msgpuck buffer." "\n", "src/tree.c", 331,((* __error()) == 0 ? "None" : strerror((* __error())))); (* __error())=0; goto error;}; | |||
332 | } | |||
333 | ||||
334 | to_return[i] = dest; | |||
335 | } | |||
336 | ||||
337 | *data = to_return; | |||
338 | free(matches); | |||
339 | return imatches; | |||
340 | ||||
341 | error: | |||
342 | if (matches != NULL((void *)0)) | |||
343 | free(matches); | |||
344 | if (*data != NULL((void *)0)) | |||
345 | free(*data); | |||
346 | if (to_return != NULL((void *)0)) | |||
347 | free(to_return); | |||
348 | if (dest != NULL((void *)0)) | |||
349 | free(dest); | |||
350 | ||||
351 | return -1; | |||
352 | } |