regulator: tps65218: Add NULL test for devm_kzalloc call

Signed-off-by: Axel Lin <axel.lin@ingics.com>
Reviewed-by: Keerthy <j-keerthy@ti.com>
Signed-off-by: Mark Brown <broonie@kernel.org>

1 files changed, 2 insertions, 0 deletions

diff --git a/drivers/regulator/tps65218-regulator.c b/drivers/regulator/tps65218-regulator.c
index b6f705292aef..1827185beacc 100644
--- a/drivers/regulator/tps65218-regulator.c
+++ b/drivers/regulator/tps65218-regulator.c
@@ -326,6 +326,8 @@ static int tps65218_regulator_probe(struct platform_device *pdev)
         /* Allocate memory for strobes */
         tps->strobes = devm_kzalloc(&pdev->dev, sizeof(u8) *
                                     TPS65218_NUM_REGULATOR, GFP_KERNEL);
+        if (!tps->strobes)
+                return -ENOMEM;
 
         for (i = 0; i < ARRAY_SIZE(regulators); i++) {
                 rdev = devm_regulator_register(&pdev->dev, &regulators[i],