On Wed, Jun 20, 2018 at 12:19:43PM -0400, Stefan Berger wrote:
> Introduce a kref for the tpm_chip that we initialize when the tpm_chip has
> been allocated and release before the tpm_chip is to be freed.
> 
> Signed-off-by: Stefan Berger <stef...@linux.vnet.ibm.com>
>  drivers/char/tpm/tpm-chip.c | 24 +++++++++++++++++++-----
>  drivers/char/tpm/tpm.h      |  1 +
>  2 files changed, 20 insertions(+), 5 deletions(-)
> 
> diff --git a/drivers/char/tpm/tpm-chip.c b/drivers/char/tpm/tpm-chip.c
> index 0a62c19937b6..a933676194a4 100644
> +++ b/drivers/char/tpm/tpm-chip.c
> @@ -119,8 +119,24 @@ struct tpm_chip *tpm_chip_find_get(struct tpm_chip *chip)
>       return res;
>  }
>  
> +static void tpm_chip_free(struct kref *kref)
> +{
> +     struct tpm_chip *chip = container_of(kref, struct tpm_chip, kref);
> +
> +     kfree(chip->log.bios_event_log);
> +     kfree(chip->work_space.context_buf);
> +     kfree(chip->work_space.session_buf);
> +     kfree(chip);
> +}
> +
> +static void tpm_chip_put(struct tpm_chip *chip)
> +{
> +     if (chip)
> +             kref_put(&chip->kref, tpm_chip_free);
> +}
> +
>  /**
> - * tpm_dev_release() - free chip memory and the device number
> + * tpm_dev_release() - free the device number and release reference to chip
>   * @dev: the character device for the TPM chip
>   *
>   * This is used as the release function for the character device.
> @@ -133,10 +149,7 @@ static void tpm_dev_release(struct device *dev)
>       idr_remove(&dev_nums_idr, chip->dev_num);
>       mutex_unlock(&idr_lock);
>  
> -     kfree(chip->log.bios_event_log);
> -     kfree(chip->work_space.context_buf);
> -     kfree(chip->work_space.session_buf);
> -     kfree(chip);
> +     tpm_chip_put(chip);
>  }
>  static void tpm_devs_release(struct device *dev)
> @@ -195,6 +208,7 @@ struct tpm_chip *tpm_chip_alloc(struct device *pdev,
>  
>       mutex_init(&chip->tpm_mutex);
>       init_rwsem(&chip->ops_sem);
> +     kref_init(&chip->kref);
>  
>       chip->ops = ops;
>  
> diff --git a/drivers/char/tpm/tpm.h b/drivers/char/tpm/tpm.h
> index 7f2d0f489e9c..098d7dcc04a4 100644
> +++ b/drivers/char/tpm/tpm.h
> @@ -211,6 +211,7 @@ struct tpm_chip {
>       struct cdev cdev;
>       struct cdev cdevs;
>  
> +     struct kref kref;

NAK, there is already a kref in struct device, that one must be used.

Jason

Reply via email to