driver core: node: use put_device() if device_register fail

if device_register() returned an error! Always use put_device() to give up the reference initialized. Signed-off-by: Arvind Yadav <[email protected]> Signed-off-by: Greg Kroah-Hartman <[email protected]>
Kleist · Mar 15, 2018 · c1cc0d5 · c1cc0d5
1 parent c8ae167
commit c1cc0d5
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/drivers/base/node.c b/drivers/base/node.c
@@ -315,7 +315,9 @@ static int register_node(struct node *node, int num)
 	node->dev.groups = node_dev_groups;
 	error = device_register(&node->dev);
 
-	if (!error){
+	if (error)
+		put_device(&node->dev);
+	else {
 		hugetlb_register_node(node);
 
 		compaction_register_node(node);