rbtree: fix allocation failre paths.

6156d1ab · Rusty Russell · 7467dfe7 · 6156d1ab
Commit 6156d1ab authored Jan 10, 2011 by Rusty Russell
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 0 deletions

ccan/rbtree/rbtree.c ccan/rbtree/rbtree.c +4 -0

No files found.
--- a/ccan/rbtree/rbtree.c
+++ b/ccan/rbtree/rbtree.c
@@ -671,6 +671,8 @@ trbt_insert32(trbt_tree_t *tree, uint32_t key, void *data)
 				trbt_node_t *new_node;

 				new_node = trbt_create_node(tree, node, key, data);
+				if (!new_node)
+					return NULL;
 				node->left=new_node;
 				node=new_node;

@@ -685,6 +687,8 @@ trbt_insert32(trbt_tree_t *tree, uint32_t key, void *data)
 				trbt_node_t *new_node;

 				new_node = trbt_create_node(tree, node, key, data);
+				if (!new_node)
+					return NULL;
 				node->right=new_node;
 				node=new_node;
 				break;