tpm: fix: return rc when devm_add_action() fails

Call put_device() and return error code if devm_add_action() fails.

Signed-off-by: Jarkko Sakkinen <jarkko.sakkinen@linux.intel.com>
Reported-by: Jason Gunthorpe <jgunthorpe@obsidianresearch.com>
Fixes: 8e0ee3c9faed ("tpm: fix the cleanup of struct tpm_chip")

1 files changed, 6 insertions, 1 deletions

diff --git a/drivers/char/tpm/tpm-chip.c b/drivers/char/tpm/tpm-chip.c
index 252142524ff2..274dd0123237 100644
--- a/drivers/char/tpm/tpm-chip.c
+++ b/drivers/char/tpm/tpm-chip.c
@@ -88,6 +88,7 @@ struct tpm_chip *tpmm_chip_alloc(struct device *dev,
                                  const struct tpm_class_ops *ops)
 {
         struct tpm_chip *chip;
+        int rc;
 
         chip = kzalloc(sizeof(*chip), GFP_KERNEL);
         if (chip == NULL)
@@ -136,7 +137,11 @@ struct tpm_chip *tpmm_chip_alloc(struct device *dev,
         chip->cdev.owner = chip->pdev->driver->owner;
         chip->cdev.kobj.parent = &chip->dev.kobj;
 
-        devm_add_action(dev, (void (*)(void *)) put_device, &chip->dev);
+        rc = devm_add_action(dev, (void (*)(void *)) put_device, &chip->dev);
+        if (rc) {
+                put_device(&chip->dev);
+                return ERR_PTR(rc);
+        }
 
         return chip;
 }