bpf: fix range_tree_set error handling

`range_tree_set` might fail and return -ENOMEM, causing subsequent `bpf_arena_alloc_pages` to fail. Added the error handling. Signed-off-by: Soma Nakata <[email protected]> Acked-by: Hou Tao <[email protected]>
kernel-patches · Jan 7, 2025 · 303bb99 · 303bb99
1 parent 2410e16
commit 303bb99
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/kernel/bpf/arena.c b/kernel/bpf/arena.c
@@ -138,7 +138,11 @@ static struct bpf_map *arena_map_alloc(union bpf_attr *attr)
 	INIT_LIST_HEAD(&arena->vma_list);
 	bpf_map_init_from_attr(&arena->map, attr);
 	range_tree_init(&arena->rt);
-	range_tree_set(&arena->rt, 0, attr->max_entries);
+	err = range_tree_set(&arena->rt, 0, attr->max_entries);
+	if (err) {
+		bpf_map_area_free(arena);
+		goto err;
+	}
 	mutex_init(&arena->lock);
 
 	return &arena->map;