Request missing SWID tags in a directed PA-TNC message
[strongswan.git] / src / libimcv / pts / pts.c
index a7def9b..1ca7209 100644 (file)
@@ -224,17 +224,24 @@ METHOD(pts_t, create_dh_nonce, bool,
        return TRUE;
 }
 
-METHOD(pts_t, get_my_public_value, void,
+METHOD(pts_t, get_my_public_value, bool,
        private_pts_t *this, chunk_t *value, chunk_t *nonce)
 {
-       this->dh->get_my_public_value(this->dh, value);
+       if (!this->dh->get_my_public_value(this->dh, value))
+       {
+               return FALSE;
+       }
        *nonce = this->is_imc ? this->responder_nonce : this->initiator_nonce;
+       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);
+       if (!this->dh->set_other_public_value(this->dh, value))
+       {
+               return FALSE;
+       }
 
        nonce = chunk_clone(nonce);
        if (this->is_imc)
@@ -245,6 +252,7 @@ METHOD(pts_t, set_peer_public_value, void,
        {
                this->responder_nonce = nonce;
        }
+       return TRUE;
 }
 
 METHOD(pts_t, calculate_secret, bool,