trusted-keys: additional TSS return code and other error handling

Previously not all TSS return codes were tested, as they were all eventually
caught by the TPM. Now all returns are tested and handled immediately.

This patch also fixes memory leaks in error and non-error paths.

Signed-off-by: David Safford <safford@watson.ibm.com>
Acked-by: Mimi Zohar <zohar@us.ibm.com>
Acked-by: David Howells <dhowells@redhat.com>
Acked-by: Serge E. Hallyn <serge@hallyn.com>
Signed-off-by: James Morris <jmorris@namei.org>

1 files changed, 87 insertions, 62 deletions

diff --git a/security/keys/trusted_defined.c b/security/keys/trusted_defined.c
index aaaa069967a6..3dc3db15b5d9 100644
--- a/security/keys/trusted_defined.c
+++ b/security/keys/trusted_defined.c
@@ -108,7 +108,8 @@ static int TSS_rawhmac(unsigned char *digest, const unsigned char *key,
                         goto out;
         }
         va_end(argp);
-        ret = crypto_shash_final(&sdesc->shash, digest);
+        if (!ret)
+                ret = crypto_shash_final(&sdesc->shash, digest);
 out:
         kfree(sdesc);
         return ret;
@@ -117,9 +118,9 @@ out:
 /*
  * calculate authorization info fields to send to TPM
  */
-static uint32_t TSS_authhmac(unsigned char *digest, const unsigned char *key,
-                             const unsigned int keylen, unsigned char *h1,
-                             unsigned char *h2, unsigned char h3, ...)
+static int TSS_authhmac(unsigned char *digest, const unsigned char *key,
+                        const unsigned int keylen, unsigned char *h1,
+                        unsigned char *h2, unsigned char h3, ...)
 {
         unsigned char paramdigest[SHA1_DIGEST_SIZE];
         struct sdesc *sdesc;
@@ -146,15 +147,17 @@ static uint32_t TSS_authhmac(unsigned char *digest, const unsigned char *key,
                         break;
                 data = va_arg(argp, unsigned char *);
                 ret = crypto_shash_update(&sdesc->shash, data, dlen);
-                if (ret < 0)
+                if (ret < 0) {
+                        va_end(argp);
                         goto out;
+                }
         }
         va_end(argp);
         ret = crypto_shash_final(&sdesc->shash, paramdigest);
         if (!ret)
-                TSS_rawhmac(digest, key, keylen, SHA1_DIGEST_SIZE,
-                            paramdigest, TPM_NONCE_SIZE, h1,
-                            TPM_NONCE_SIZE, h2, 1, &c, 0, 0);
+                ret = TSS_rawhmac(digest, key, keylen, SHA1_DIGEST_SIZE,
+                                  paramdigest, TPM_NONCE_SIZE, h1,
+                                  TPM_NONCE_SIZE, h2, 1, &c, 0, 0);
 out:
         kfree(sdesc);
         return ret;
@@ -163,11 +166,11 @@ out:
 /*
  * verify the AUTH1_COMMAND (Seal) result from TPM
  */
-static uint32_t TSS_checkhmac1(unsigned char *buffer,
-                               const uint32_t command,
-                               const unsigned char *ononce,
-                               const unsigned char *key,
-                               const unsigned int keylen, ...)
+static int TSS_checkhmac1(unsigned char *buffer,
+                          const uint32_t command,
+                          const unsigned char *ononce,
+                          const unsigned char *key,
+                          const unsigned int keylen, ...)
 {
         uint32_t bufsize;
         uint16_t tag;
@@ -219,18 +222,22 @@ static uint32_t TSS_checkhmac1(unsigned char *buffer,
                         break;
                 dpos = va_arg(argp, unsigned int);
                 ret = crypto_shash_update(&sdesc->shash, buffer + dpos, dlen);
-                if (ret < 0)
+                if (ret < 0) {
+                        va_end(argp);
                         goto out;
+                }
         }
         va_end(argp);
         ret = crypto_shash_final(&sdesc->shash, paramdigest);
         if (ret < 0)
                 goto out;
+
         ret = TSS_rawhmac(testhmac, key, keylen, SHA1_DIGEST_SIZE, paramdigest,
                           TPM_NONCE_SIZE, enonce, TPM_NONCE_SIZE, ononce,
                           1, continueflag, 0, 0);
         if (ret < 0)
                 goto out;
+
         if (memcmp(testhmac, authdata, SHA1_DIGEST_SIZE))
                 ret = -EINVAL;
 out:
@@ -241,13 +248,13 @@ out:
 /*
  * verify the AUTH2_COMMAND (unseal) result from TPM
  */
-static uint32_t TSS_checkhmac2(unsigned char *buffer,
-                               const uint32_t command,
-                               const unsigned char *ononce,
-                               const unsigned char *key1,
-                               const unsigned int keylen1,
-                               const unsigned char *key2,
-                               const unsigned int keylen2, ...)
+static int TSS_checkhmac2(unsigned char *buffer,
+                          const uint32_t command,
+                          const unsigned char *ononce,
+                          const unsigned char *key1,
+                          const unsigned int keylen1,
+                          const unsigned char *key2,
+                          const unsigned int keylen2, ...)
 {
         uint32_t bufsize;
         uint16_t tag;
@@ -309,9 +316,12 @@ static uint32_t TSS_checkhmac2(unsigned char *buffer,
                         break;
                 dpos = va_arg(argp, unsigned int);
                 ret = crypto_shash_update(&sdesc->shash, buffer + dpos, dlen);
-                if (ret < 0)
+                if (ret < 0) {
+                        va_end(argp);
                         goto out;
+                }
         }
+        va_end(argp);
         ret = crypto_shash_final(&sdesc->shash, paramdigest);
         if (ret < 0)
                 goto out;
@@ -319,6 +329,8 @@ static uint32_t TSS_checkhmac2(unsigned char *buffer,
         ret = TSS_rawhmac(testhmac1, key1, keylen1, SHA1_DIGEST_SIZE,
                           paramdigest, TPM_NONCE_SIZE, enonce1,
                           TPM_NONCE_SIZE, ononce, 1, continueflag1, 0, 0);
+        if (ret < 0)
+                goto out;
         if (memcmp(testhmac1, authdata1, SHA1_DIGEST_SIZE)) {
                 ret = -EINVAL;
                 goto out;
@@ -326,6 +338,8 @@ static uint32_t TSS_checkhmac2(unsigned char *buffer,
         ret = TSS_rawhmac(testhmac2, key2, keylen2, SHA1_DIGEST_SIZE,
                           paramdigest, TPM_NONCE_SIZE, enonce2,
                           TPM_NONCE_SIZE, ononce, 1, continueflag2, 0, 0);
+        if (ret < 0)
+                goto out;
         if (memcmp(testhmac2, authdata2, SHA1_DIGEST_SIZE))
                 ret = -EINVAL;
 out:
@@ -364,8 +378,8 @@ static int tpm_get_random(struct tpm_buf *tb, unsigned char *buf, uint32_t len)
         store32(tb, TPM_ORD_GETRANDOM);
         store32(tb, len);
         ret = trusted_tpm_send(TPM_ANY_NUM, tb->data, sizeof tb->data);
-        memcpy(buf, tb->data + TPM_GETRANDOM_SIZE, len);
-
+        if (!ret)
+                memcpy(buf, tb->data + TPM_GETRANDOM_SIZE, len);
         return ret;
 }
 
@@ -392,10 +406,13 @@ static int my_get_random(unsigned char *buf, int len)
 static int pcrlock(const int pcrnum)
 {
         unsigned char hash[SHA1_DIGEST_SIZE];