mirror of https://github.com/aria2/aria2
136 lines
4.1 KiB
C++
136 lines
4.1 KiB
C++
#include "MSEHandshake.h"
|
|
|
|
#include <cstring>
|
|
|
|
#include <cppunit/extensions/HelperMacros.h>
|
|
|
|
#include "Exception.h"
|
|
#include "util.h"
|
|
#include "prefs.h"
|
|
#include "Socket.h"
|
|
#include "Option.h"
|
|
#include "DownloadContext.h"
|
|
#include "FileEntry.h"
|
|
#include "array_fun.h"
|
|
#include "bittorrent_helper.h"
|
|
|
|
namespace aria2 {
|
|
|
|
class MSEHandshakeTest:public CppUnit::TestFixture {
|
|
|
|
CPPUNIT_TEST_SUITE(MSEHandshakeTest);
|
|
CPPUNIT_TEST(testHandshake);
|
|
CPPUNIT_TEST_SUITE_END();
|
|
private:
|
|
SharedHandle<DownloadContext> _dctx;
|
|
|
|
void doHandshake(const SharedHandle<MSEHandshake>& initiator,
|
|
const SharedHandle<MSEHandshake>& receiver);
|
|
|
|
public:
|
|
void setUp()
|
|
{
|
|
_dctx.reset(new DownloadContext());
|
|
unsigned char infoHash[20];
|
|
memset(infoHash, 0, sizeof(infoHash));
|
|
BDE torrentAttrs = BDE::dict();
|
|
torrentAttrs[bittorrent::INFO_HASH] =
|
|
std::string(&infoHash[0], &infoHash[arrayLength(infoHash)]);
|
|
_dctx->setAttribute(bittorrent::BITTORRENT, torrentAttrs);
|
|
}
|
|
|
|
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(bittorrent::getInfoHash(_dctx));
|
|
initiator->sendInitiatorStep2();
|
|
|
|
while(!receiver->findReceiverHashMarker());
|
|
std::deque<SharedHandle<DownloadContext> > contexts;
|
|
contexts.push_back(_dctx);
|
|
while(!receiver->receiveReceiverHashAndPadCLength(contexts));
|
|
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
|