mirror of https://github.com/aria2/aria2
129 lines
3.8 KiB
C++
129 lines
3.8 KiB
C++
#include "MSEHandshake.h"
|
|
|
|
#include <cppunit/extensions/HelperMacros.h>
|
|
|
|
#include "Exception.h"
|
|
#include "Util.h"
|
|
#include "prefs.h"
|
|
#include "Socket.h"
|
|
#include "Option.h"
|
|
#include "MockBtContext.h"
|
|
#include "FileEntry.h"
|
|
|
|
namespace aria2 {
|
|
|
|
class MSEHandshakeTest:public CppUnit::TestFixture {
|
|
|
|
CPPUNIT_TEST_SUITE(MSEHandshakeTest);
|
|
CPPUNIT_TEST(testHandshake);
|
|
CPPUNIT_TEST_SUITE_END();
|
|
private:
|
|
SharedHandle<MockBtContext> _btctx;
|
|
|
|
void doHandshake(const SharedHandle<MSEHandshake>& initiator,
|
|
const SharedHandle<MSEHandshake>& receiver);
|
|
|
|
public:
|
|
void setUp()
|
|
{
|
|
_btctx.reset(new MockBtContext());
|
|
unsigned char infoHash[20];
|
|
memset(infoHash, 0, sizeof(infoHash));
|
|
_btctx->setInfoHash(infoHash);
|
|
}
|
|
|
|
void testHandshake();
|
|
};
|
|
|
|
|
|
CPPUNIT_TEST_SUITE_REGISTRATION(MSEHandshakeTest);
|
|
|
|
static std::pair<SharedHandle<SocketCore>,
|
|
SharedHandle<SocketCore> > createSocketPair()
|
|
{
|
|
SharedHandle<SocketCore> initiatorSock(new SocketCore());
|
|
|
|
SocketCore receiverServerSock;
|
|
receiverServerSock.bind(0);
|
|
receiverServerSock.beginListen();
|
|
|
|
std::pair<std::string, uint16_t> receiverAddrInfo;
|
|
receiverServerSock.getAddrInfo(receiverAddrInfo);
|
|
initiatorSock->establishConnection("localhost", receiverAddrInfo.second);
|
|
initiatorSock->setBlockingMode();
|
|
|
|
SharedHandle<SocketCore> receiverSock(receiverServerSock.acceptConnection());
|
|
receiverSock->setBlockingMode();
|
|
|
|
return std::pair<SharedHandle<SocketCore>,
|
|
SharedHandle<SocketCore> >(initiatorSock, receiverSock);
|
|
}
|
|
|
|
void MSEHandshakeTest::doHandshake(const SharedHandle<MSEHandshake>& initiator, const SharedHandle<MSEHandshake>& receiver)
|
|
{
|
|
initiator->sendPublicKey();
|
|
|
|
while(!receiver->receivePublicKey());
|
|
receiver->sendPublicKey();
|
|
|
|
while(!initiator->receivePublicKey());
|
|
initiator->initCipher(_btctx->getInfoHash());
|
|
initiator->sendInitiatorStep2();
|
|
|
|
while(!receiver->findReceiverHashMarker());
|
|
std::deque<SharedHandle<BtContext> > btContexts;
|
|
btContexts.push_back(_btctx);
|
|
while(!receiver->receiveReceiverHashAndPadCLength(btContexts));
|
|
while(!receiver->receivePad());
|
|
while(!receiver->receiveReceiverIALength());
|
|
while(!receiver->receiveReceiverIA());
|
|
receiver->sendReceiverStep2();
|
|
|
|
while(!initiator->findInitiatorVCMarker());
|
|
while(!initiator->receiveInitiatorCryptoSelectAndPadDLength());
|
|
while(!initiator->receivePad());
|
|
}
|
|
|
|
static SharedHandle<MSEHandshake>
|
|
createMSEHandshake(SharedHandle<SocketCore> socket, bool initiator,
|
|
const Option* option)
|
|
{
|
|
SharedHandle<MSEHandshake> h(new MSEHandshake(1, socket, option));
|
|
h->initEncryptionFacility(initiator);
|
|
return h;
|
|
}
|
|
|
|
void MSEHandshakeTest::testHandshake()
|
|
{
|
|
{
|
|
Option op;
|
|
op.put(PREF_BT_MIN_CRYPTO_LEVEL, V_PLAIN);
|
|
|
|
std::pair<SharedHandle<SocketCore>, SharedHandle<SocketCore> > sockPair =
|
|
createSocketPair();
|
|
SharedHandle<MSEHandshake> initiator = createMSEHandshake(sockPair.first, true, &op);
|
|
SharedHandle<MSEHandshake> receiver = createMSEHandshake(sockPair.second, false, &op);
|
|
|
|
doHandshake(initiator, receiver);
|
|
|
|
CPPUNIT_ASSERT_EQUAL(MSEHandshake::CRYPTO_PLAIN_TEXT, initiator->getNegotiatedCryptoType());
|
|
CPPUNIT_ASSERT_EQUAL(MSEHandshake::CRYPTO_PLAIN_TEXT, receiver->getNegotiatedCryptoType());
|
|
}
|
|
{
|
|
Option op;
|
|
op.put(PREF_BT_MIN_CRYPTO_LEVEL, V_ARC4);
|
|
|
|
std::pair<SharedHandle<SocketCore>, SharedHandle<SocketCore> > sockPair =
|
|
createSocketPair();
|
|
SharedHandle<MSEHandshake> initiator = createMSEHandshake(sockPair.first, true, &op);
|
|
SharedHandle<MSEHandshake> receiver = createMSEHandshake(sockPair.second, false, &op);
|
|
|
|
doHandshake(initiator, receiver);
|
|
|
|
CPPUNIT_ASSERT_EQUAL(MSEHandshake::CRYPTO_ARC4, initiator->getNegotiatedCryptoType());
|
|
CPPUNIT_ASSERT_EQUAL(MSEHandshake::CRYPTO_ARC4, receiver->getNegotiatedCryptoType());
|
|
}
|
|
}
|
|
|
|
} // namespace aria2
|