1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
|
#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));
SharedHandle<TorrentAttribute> torrentAttrs(new TorrentAttribute());
torrentAttrs->infoHash = std::string(vbegin(infoHash), vend(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::vector<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
|
