On Wed, May 24, 2017 at 03:22:11PM -0700, Jarkko Sakkinen wrote:
> On Wed, May 24, 2017 at 05:39:41PM -0400, Stefan Berger wrote:
> > To prevent userspace from sending the TPM driver command to set
> > the locality, we need to check every command that is sent from
> > user space. To distinguish user space commands from internally
> > sent commands we introduce an additional state flag
> > STATE_DRIVER_COMMAND that is set while the driver sends this
> > command. Similar to the TPM 2 space commands we return an error
> > code when this command is detected.
> > 
> > Signed-off-by: Stefan Berger <stef...@linux.vnet.ibm.com>
> > ---
> >  drivers/char/tpm/tpm_vtpm_proxy.c | 33 +++++++++++++++++++++++++++++++++
> >  1 file changed, 33 insertions(+)
> > 
> > diff --git a/drivers/char/tpm/tpm_vtpm_proxy.c 
> > b/drivers/char/tpm/tpm_vtpm_proxy.c
> > index 66024bf..1d877cc 100644
> > --- a/drivers/char/tpm/tpm_vtpm_proxy.c
> > +++ b/drivers/char/tpm/tpm_vtpm_proxy.c
> > @@ -43,6 +43,7 @@ struct proxy_dev {
> >  #define STATE_OPENED_FLAG        BIT(0)
> >  #define STATE_WAIT_RESPONSE_FLAG BIT(1)  /* waiting for emulator response 
> > */
> >  #define STATE_REGISTERED_FLAG       BIT(2)
> > +#define STATE_DRIVER_COMMAND     BIT(3)  /* sending a driver specific 
> > command */
> >  
> >     size_t req_len;              /* length of queued TPM request */
> >     size_t resp_len;             /* length of queued TPM response */
> > @@ -299,6 +300,28 @@ static int vtpm_proxy_tpm_op_recv(struct tpm_chip 
> > *chip, u8 *buf, size_t count)
> >     return len;
> >  }
> >  
> > +static int vtpm_proxy_is_driver_command(struct tpm_chip *chip,
> > +                                   u8 *buf, size_t count)
> > +{
> > +   struct tpm_input_header *hdr = (struct tpm_input_header *)buf;
> > +
> > +   if (count < sizeof(struct tpm_input_header))
> > +           return 0;
> > +
> > +   if (chip->flags & TPM_CHIP_FLAG_TPM2) {
> > +           switch (be32_to_cpu(hdr->ordinal)) {
> > +           case TPM2_CC_SET_LOCALITY:
> > +                   return 1;
> > +           }
> > +   } else {
> > +           switch (be32_to_cpu(hdr->ordinal)) {
> > +           case TPM_ORD_SET_LOCALITY:
> > +                   return 1;
> > +           }
> > +   }
> > +   return 0;
> > +}
> > +
> >  /*
> >   * Called when core TPM driver forwards TPM requests to 'server side'.
> >   *
> > @@ -321,6 +344,10 @@ static int vtpm_proxy_tpm_op_send(struct tpm_chip 
> > *chip, u8 *buf, size_t count)
> >             return -EIO;
> >     }
> >  
> > +   if (!(proxy_dev->state & STATE_DRIVER_COMMAND) &&
> > +       vtpm_proxy_is_driver_command(chip, buf, count))
> > +           return -EFAULT;
> > +
> >     mutex_lock(&proxy_dev->buf_lock);
> >  
> >     if (!(proxy_dev->state & STATE_OPENED_FLAG)) {
> > @@ -376,6 +403,7 @@ static int vtpm_proxy_request_locality(struct tpm_chip 
> > *chip, int locality)
> >     struct tpm_buf buf;
> >     int rc;
> >     const struct tpm_output_header *header;
> > +   struct proxy_dev *proxy_dev = dev_get_drvdata(&chip->dev);
> >  
> >     if (chip->flags & TPM_CHIP_FLAG_TPM2)
> >             rc = tpm_buf_init(&buf, TPM2_ST_SESSIONS,
> > @@ -387,9 +415,14 @@ static int vtpm_proxy_request_locality(struct tpm_chip 
> > *chip, int locality)
> >             return rc;
> >     tpm_buf_append_u8(&buf, locality);
> >  
> > +   proxy_dev->state |= STATE_DRIVER_COMMAND;
> > +
> >     rc = tpm_transmit_cmd(chip, NULL, buf.data, tpm_buf_length(&buf), 0,
> >                           TPM_TRANSMIT_UNLOCKED | TPM_TRANSMIT_RAW,
> >                           "attempting to set locality");
> > +
> > +   proxy_dev->state &= ~STATE_DRIVER_COMMAND;
> > +
> >     if (rc < 0) {
> >             locality = rc;
> >             goto out;
> > -- 
> > 2.4.3
> > 
> 
> Otherwise fine except for the redundant code.
> 
> /Jarkko

Reviewed-by: Jarkko Sakkinen <jarkko.sakki...@linux.intel.com>

Probably not yet merging and testing this this week though as I'm
travelling in US right now.

/Jarkko

Reply via email to