rollback due the race condition

8 years ago · 89d69a5d5a
1 changed files with 3 additions and 4 deletions
--- a/Crypto.cpp
+++ b/Crypto.cpp
@ -270,10 +270,9 @@ namespace crypto
 	{
 		if (m_IsUpdated)
 		{	
-			const BIGNUM * priv_key, * pub_key;	
+			// TODO: seems race condition here
-			DH_get0_key (m_DH, &pub_key, &priv_key);
+			bn2buf (m_DH->pub_key, m_PublicKey, 256);
-			bn2buf (pub_key, m_PublicKey, 256);
+			BN_free (m_DH->pub_key); m_DH->pub_key = NULL;
 			DH_set0_key (m_DH, NULL, NULL);
 			m_IsUpdated= false;
 		}	
 		return m_PublicKey;