Peerplays_Blockchain/peerplays-fc
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Added test for diffie-hellmann key exchange

Browse source
This commit is contained in:
Peter Conrad 2015-07-30 17:42:20 +02:00
parent 1a342f8fae
commit bfa1433cdd
2 changed files with 70 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
CMakeLists.txt
View file

@ -366,6 +366,7 @@ add_executable( all_tests tests/all_tests.cpp
tests/crypto/bigint_test.cpp tests/crypto/bigint_test.cpp
tests/crypto/blind.cpp tests/crypto/blind.cpp
tests/crypto/blowfish_test.cpp tests/crypto/blowfish_test.cpp
tests/crypto/dh_test.cpp
tests/crypto/sha_tests.cpp tests/crypto/sha_tests.cpp
tests/network/ntp_test.cpp tests/network/ntp_test.cpp
tests/thread/task_cancel.cpp tests/thread/task_cancel.cpp

69
tests/crypto/dh_test.cpp Normal file
View file

@ -0,0 +1,69 @@
#include <boost/test/unit_test.hpp>
#include <fc/crypto/dh.hpp>
#include <fc/exception/exception.hpp>
BOOST_AUTO_TEST_SUITE(fc_crypto)
BOOST_AUTO_TEST_CASE(dh_test)
{
fc::diffie_hellman alice;
BOOST_CHECK( alice.generate_params( 128, 5 ) );
BOOST_CHECK( alice.generate_pub_key() );
fc::diffie_hellman bob;
bob.p = alice.p;
BOOST_CHECK( bob.validate() );
BOOST_CHECK( bob.generate_pub_key() );
fc::diffie_hellman charlie;
BOOST_CHECK( !charlie.validate() );
BOOST_CHECK( !charlie.generate_pub_key() );
charlie.p = alice.p;
BOOST_CHECK( charlie.validate() );
BOOST_CHECK( charlie.generate_pub_key() );
BOOST_CHECK( alice.compute_shared_key( bob.pub_key ) );
BOOST_CHECK( bob.compute_shared_key( alice.pub_key ) );
BOOST_CHECK_EQUAL( alice.shared_key.size(), bob.shared_key.size() );
BOOST_CHECK( !memcmp( alice.shared_key.data(), bob.shared_key.data(), alice.shared_key.size() ) );
std::vector<char> alice_bob = alice.shared_key;
BOOST_CHECK( alice.compute_shared_key( charlie.pub_key ) );
BOOST_CHECK( charlie.compute_shared_key( alice.pub_key ) );
BOOST_CHECK_EQUAL( alice.shared_key.size(), charlie.shared_key.size() );
BOOST_CHECK( !memcmp( alice.shared_key.data(), charlie.shared_key.data(), alice.shared_key.size() ) );
std::vector<char> alice_charlie = alice.shared_key;
BOOST_CHECK( charlie.compute_shared_key( bob.pub_key ) );
BOOST_CHECK( bob.compute_shared_key( charlie.pub_key ) );
BOOST_CHECK_EQUAL( charlie.shared_key.size(), bob.shared_key.size() );
BOOST_CHECK( !memcmp( charlie.shared_key.data(), bob.shared_key.data(), bob.shared_key.size() ) );
std::vector<char> bob_charlie = charlie.shared_key;
BOOST_CHECK_EQUAL( alice_bob.size(), alice_charlie.size() );
BOOST_CHECK( memcmp( alice_bob.data(), alice_charlie.data(), alice_bob.size() ) );
BOOST_CHECK_EQUAL( alice_bob.size(), bob_charlie.size() );
BOOST_CHECK( memcmp( alice_bob.data(), bob_charlie.data(), alice_bob.size() ) );
BOOST_CHECK_EQUAL( alice_charlie.size(), bob_charlie.size() );
BOOST_CHECK( memcmp( alice_charlie.data(), bob_charlie.data(), alice_charlie.size() ) );
alice.p.clear(); alice.p.push_back(100); alice.p.push_back(2);
BOOST_CHECK( !alice.validate() );
alice.p = bob.p;
alice.g = 9;
BOOST_CHECK( !alice.validate() );
// It ain't easy...
// alice.g = 2;
// BOOST_CHECK( alice.validate() );
// BOOST_CHECK( alice.generate_pub_key() );
// BOOST_CHECK( alice.compute_shared_key( bob.pub_key ) );
// BOOST_CHECK( bob.compute_shared_key( alice.pub_key ) );
// BOOST_CHECK_EQUAL( alice.shared_key.size(), bob.shared_key.size() );
// BOOST_CHECK( memcmp( alice.shared_key.data(), bob.shared_key.data(), alice.shared_key.size() ) );
}
BOOST_AUTO_TEST_SUITE_END()