[PATCH] timers/migration: Ensure proper resource cleanup on error in tmigr_init

[Zhan Xusheng <zhanxusheng1024@gmail.com>]

Free resources allocated by zalloc_cpumask_var and kcalloc on failure.

Signed-off-by: Zhan Xusheng <zhanxusheng@xiaomi.com>
---
 kernel/time/timer_migration.c | 12 +++++++++---
 1 file changed, 9 insertions(+), 3 deletions(-)

diff --git a/kernel/time/timer_migration.c b/kernel/time/timer_migration.c
index 18dda1aa782d..c2b3aa6946c1 100644
--- a/kernel/time/timer_migration.c
+++ b/kernel/time/timer_migration.c
@@ -1994,7 +1994,7 @@ static int __init tmigr_init(void)
 
 	tmigr_level_list = kcalloc(tmigr_hierarchy_levels, sizeof(struct list_head), GFP_KERNEL);
 	if (!tmigr_level_list)
-		goto err;
+		goto err_free_cpumask;
 
 	for (i = 0; i < tmigr_hierarchy_levels; i++)
 		INIT_LIST_HEAD(&tmigr_level_list[i]);
@@ -2007,15 +2007,21 @@ static int __init tmigr_init(void)
 	ret = cpuhp_setup_state(CPUHP_TMIGR_PREPARE, "tmigr:prepare",
 				tmigr_cpu_prepare, NULL);
 	if (ret)
-		goto err;
+		goto err_free_level_list;
 
 	ret = cpuhp_setup_state(CPUHP_AP_TMIGR_ONLINE, "tmigr:online",
 				tmigr_set_cpu_available, tmigr_clear_cpu_available);
 	if (ret)
-		goto err;
+		goto err_free_level_list;
 
 	return 0;
 
+err_free_level_list:
+	kfree(tmigr_level_list);
+
+err_free_cpumask:
+	free_cpumask_var(tmigr_available_cpumask);
+
 err:
 	pr_err("Timer migration setup failed\n");
 	return ret;
-- 
2.43.0