blkcg: add missing NULL check in ioc_cpd_alloc()

ioc_cpd_alloc() forgot to check NULL return from kzalloc(). Add it. Signed-off-by: Tejun Heo <[email protected]> Reported-by: kbuild test robot <[email protected]> Signed-off-by: Jens Axboe <[email protected]>
CorpDK · Aug 30, 2019 · e916ad2 · e916ad2
1 parent 3532e72
commit e916ad2
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/block/blk-iocost.c b/block/blk-iocost.c
@@ -1888,8 +1888,10 @@ static struct blkcg_policy_data *ioc_cpd_alloc(gfp_t gfp)
 	struct ioc_cgrp *iocc;
 
 	iocc = kzalloc(sizeof(struct ioc_cgrp), gfp);
-	iocc->dfl_weight = CGROUP_WEIGHT_DFL;
+	if (!iocc)
+		return NULL;
 
+	iocc->dfl_weight = CGROUP_WEIGHT_DFL;
 	return &iocc->cpd;
 }