1 files changed, 5 insertions, 35 deletions

diff --git a/src/tss2-esys/api/Esys_VerifySignature.c b/src/tss2-esys/api/Esys_VerifySignature.c
index f0ba571a..bbfdd1ce 100644
--- a/src/tss2-esys/api/Esys_VerifySignature.c
+++ b/src/tss2-esys/api/Esys_VerifySignature.c
@@ -15,30 +15,6 @@
 #include "util/log.h"
 #include "util/aux_util.h"
 
-/** Store command parameters inside the ESYS_CONTEXT for use during _Finish */
-static void store_input_parameters (
-    ESYS_CONTEXT *esysContext,
-    ESYS_TR keyHandle,
-    const TPM2B_DIGEST *digest,
-    const TPMT_SIGNATURE *signature)
-{
-    esysContext->in.VerifySignature.keyHandle = keyHandle;
-    if (digest == NULL) {
-        esysContext->in.VerifySignature.digest = NULL;
-    } else {
-        esysContext->in.VerifySignature.digestData = *digest;
-        esysContext->in.VerifySignature.digest =
-            &esysContext->in.VerifySignature.digestData;
-    }
-    if (signature == NULL) {
-        esysContext->in.VerifySignature.signature = NULL;
-    } else {
-        esysContext->in.VerifySignature.signatureData = *signature;
-        esysContext->in.VerifySignature.signature =
-            &esysContext->in.VerifySignature.signatureData;
-    }
-}
-
 /** One-Call function for TPM2_VerifySignature
  *
  * This function invokes the TPM2_VerifySignature command in a one-call
@@ -186,10 +162,9 @@ Esys_VerifySignature_Async(
         return r;
     esysContext->state = _ESYS_STATE_INTERNALERROR;
 
-    /* Check and store input parameters */
+    /* Check input parameters */
     r = check_session_feasibility(shandle1, shandle2, shandle3, 0);
     return_state_if_error(r, _ESYS_STATE_INIT, "Check session usage");
-    store_input_parameters(esysContext, keyHandle, digest, signature);
 
     /* Retrieve the metadata objects for provided handles */
     r = esys_GetResourceObject(esysContext, keyHandle, &keyHandleNode);
@@ -274,7 +249,8 @@ Esys_VerifySignature_Finish(
     }
 
     /* Check for correct sequence and set sequence to irregular for now */
-    if (esysContext->state != _ESYS_STATE_SENT) {
+    if (esysContext->state != _ESYS_STATE_SENT &&
+        esysContext->state != _ESYS_STATE_RESUBMISSION) {
         LOG_ERROR("Esys called in bad sequence.");
         return TSS2_ESYS_RC_BAD_SEQUENCE;
     }
@@ -300,19 +276,13 @@ Esys_VerifySignature_Finish(
     if (r == TPM2_RC_RETRY || r == TPM2_RC_TESTING || r == TPM2_RC_YIELDED) {
         LOG_DEBUG("TPM returned RETRY, TESTING or YIELDED, which triggers a "
             "resubmission: %" PRIx32, r);
-        if (esysContext->submissionCount >= _ESYS_MAX_SUBMISSIONS) {
+        if (esysContext->submissionCount++ >= _ESYS_MAX_SUBMISSIONS) {
             LOG_WARNING("Maximum number of (re)submissions has been reached.");
             esysContext->state = _ESYS_STATE_INIT;
             goto error_cleanup;
         }
         esysContext->state = _ESYS_STATE_RESUBMISSION;
-        r = Esys_VerifySignature_Async(esysContext,
-                                       esysContext->in.VerifySignature.keyHandle,
-                                       esysContext->session_type[0],
-                                       esysContext->session_type[1],
-                                       esysContext->session_type[2],
-                                       esysContext->in.VerifySignature.digest,
-                                       esysContext->in.VerifySignature.signature);
+        r = Tss2_Sys_ExecuteAsync(esysContext->sys);
         if (r != TSS2_RC_SUCCESS) {
             LOG_WARNING("Error attempting to resubmit");
             /* We do not set esysContext->state here but inherit the most recent