From: Rusty Russell <rusty@rustcorp.com.au>
Date: Mon, 10 Jan 2011 05:55:22 +0000 (+1030)
Subject: rbtree: fix allocation failre paths.
X-Git-Url: http://git.ozlabs.org/?p=ccan;a=commitdiff_plain;h=6156d1ab34b93c1e66473120df339fb23b08196a;ds=sidebyside

rbtree: fix allocation failre paths.
---

diff --git a/ccan/rbtree/rbtree.c b/ccan/rbtree/rbtree.c
index e3b4a951..f0a74edb 100644
--- 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;