libimcv: Allow pts_t.set_peer_public_value() to fail

diff --git a/src/libimcv/plugins/imc_attestation/imc_attestation_process.c b/src/libimcv/plugins/imc_attestation/imc_attestation_process.c
index 6f8e4ea..f24aec8 100644 (file)
--- a/src/libimcv/plugins/imc_attestation/imc_attestation_process.c
+++ b/src/libimcv/plugins/imc_attestation/imc_attestation_process.c
@@ -178,8 +178,10 @@ bool imc_attestation_process(pa_tnc_attr_t *attr, imc_msg_t *msg,
                                return FALSE;
                        }
 
-                       pts->set_peer_public_value(pts, initiator_value, initiator_nonce);
-                       if (!pts->calculate_secret(pts))
+
+                       if (!pts->set_peer_public_value(pts, initiator_value,
+                                                                                       initiator_nonce) ||
+                               !pts->calculate_secret(pts))
                        {
                                return FALSE;
                        }

diff --git a/src/libimcv/plugins/imv_attestation/imv_attestation_process.c b/src/libimcv/plugins/imv_attestation/imv_attestation_process.c
index bad536a..fbeb661 100644 (file)
--- a/src/libimcv/plugins/imv_attestation/imv_attestation_process.c
+++ b/src/libimcv/plugins/imv_attestation/imv_attestation_process.c
@@ -134,11 +134,11 @@ bool imv_attestation_process(pa_tnc_attr_t *attr, imv_msg_t *out_msg,
                        }
 
                        responder_value = attr_cast->get_responder_value(attr_cast);
-                       pts->set_peer_public_value(pts, responder_value,
-                                                                                       responder_nonce);
 
                        /* Calculate secret assessment value */
-                       if (!pts->calculate_secret(pts))
+                       if (!pts->set_peer_public_value(pts, responder_value,
+                                                                                       responder_nonce) ||
+                               !pts->calculate_secret(pts))
                        {
                                return FALSE;
                        }

diff --git a/src/libimcv/pts/pts.c b/src/libimcv/pts/pts.c
index 6deccc8..8d13bfc 100644 (file)
--- a/src/libimcv/pts/pts.c
+++ b/src/libimcv/pts/pts.c
@@ -232,7 +232,7 @@ METHOD(pts_t, get_my_public_value, bool,
        return TRUE;
 }
 
-METHOD(pts_t, set_peer_public_value, void,
+METHOD(pts_t, set_peer_public_value, bool,
        private_pts_t *this, chunk_t value, chunk_t nonce)
 {
        this->dh->set_other_public_value(this->dh, value);
@@ -246,6 +246,7 @@ METHOD(pts_t, set_peer_public_value, void,
        {
                this->responder_nonce = nonce;
        }
+       return TRUE;
 }
 
 METHOD(pts_t, calculate_secret, bool,

diff --git a/src/libimcv/pts/pts.h b/src/libimcv/pts/pts.h
index 4d3284e..d525306 100644 (file)
--- a/src/libimcv/pts/pts.h
+++ b/src/libimcv/pts/pts.h
@@ -152,8 +152,9 @@ struct pts_t {
         *
         * @param value                         Peer public DH value
         * @param nonce                         Peer DH nonce
+        * @return                                      TRUE if public value set successfully
         */
-       void (*set_peer_public_value) (pts_t *this, chunk_t value, chunk_t nonce);
+       bool (*set_peer_public_value) (pts_t *this, chunk_t value, chunk_t nonce);
 
        /**
         * Calculates assessment secret to be used for TPM Quote as ExternalData