mirror of
https://github.com/aria2/aria2.git
synced 2024-12-14 09:28:55 +00:00
27ab4b1579
Fixed the bug that DH key exchange sometimes fails due to bad handling of the number of bytes required for storing public key and shared secret. * src/LibgcryptDHKeyExchange.h * src/LibsslDHKeyExchange.h: Also added function name to handleError. * src/MSEHandshake.cc * test/DHKeyExchangeTest.cc
56 lines
1.5 KiB
C++
56 lines
1.5 KiB
C++
#include "DHKeyExchange.h"
|
|
#include "Exception.h"
|
|
#include "Util.h"
|
|
#include <cppunit/extensions/HelperMacros.h>
|
|
|
|
namespace aria2 {
|
|
|
|
class DHKeyExchangeTest:public CppUnit::TestFixture {
|
|
|
|
CPPUNIT_TEST_SUITE(DHKeyExchangeTest);
|
|
CPPUNIT_TEST(testHandshake);
|
|
CPPUNIT_TEST_SUITE_END();
|
|
public:
|
|
void setUp() {}
|
|
|
|
void tearDown() {}
|
|
|
|
void testHandshake();
|
|
};
|
|
|
|
|
|
CPPUNIT_TEST_SUITE_REGISTRATION(DHKeyExchangeTest);
|
|
|
|
void DHKeyExchangeTest::testHandshake()
|
|
{
|
|
DHKeyExchange dhA;
|
|
DHKeyExchange dhB;
|
|
|
|
const unsigned char* PRIME = reinterpret_cast<const unsigned char*>("FFFFFFFFFFFFFFFFC90FDAA22168C234C4C6628B80DC1CD129024E088A67CC74020BBEA63B139B22514A08798E3404DDEF9519B3CD3A431B302B0A6DF25F14374FE1356D6D51C245E485B576625E7EC6F44C42E9A63A36210000000000090563");
|
|
|
|
const size_t PRIME_BITS = 768;
|
|
|
|
const unsigned char* GENERATOR = reinterpret_cast<const unsigned char*>("2");
|
|
|
|
dhA.init(PRIME, PRIME_BITS, GENERATOR, 160);
|
|
dhB.init(PRIME, PRIME_BITS, GENERATOR, 160);
|
|
|
|
dhA.generatePublicKey();
|
|
dhB.generatePublicKey();
|
|
|
|
unsigned char publicKeyA[96];
|
|
unsigned char publicKeyB[96];
|
|
dhA.getPublicKey(publicKeyA, sizeof(publicKeyA));
|
|
dhB.getPublicKey(publicKeyB, sizeof(publicKeyB));
|
|
|
|
unsigned char secretA[96];
|
|
unsigned char secretB[96];
|
|
dhA.computeSecret(secretA, sizeof(secretA), publicKeyB, sizeof(publicKeyB));
|
|
dhB.computeSecret(secretB, sizeof(secretB), publicKeyA, sizeof(publicKeyA));
|
|
|
|
CPPUNIT_ASSERT_EQUAL(Util::toHex(secretA, sizeof(secretA)),
|
|
Util::toHex(secretB, sizeof(secretB)));
|
|
}
|
|
|
|
} // namespace aria2
|