2014-04-08 05:32:00 +00:00
|
|
|
/*
|
|
|
|
This file is part of cpp-ethereum.
|
|
|
|
|
|
|
|
cpp-ethereum is free software: you can redistribute it and/or modify
|
|
|
|
it under the terms of the GNU General Public License as published by
|
|
|
|
the Free Software Foundation, either version 3 of the License, or
|
|
|
|
(at your option) any later version.
|
|
|
|
|
|
|
|
cpp-ethereum is distributed in the hope that it will be useful,
|
|
|
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
|
|
GNU General Public License for more details.
|
|
|
|
|
|
|
|
You should have received a copy of the GNU General Public License
|
|
|
|
along with cpp-ethereum. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
*/
|
2014-12-18 07:35:12 +00:00
|
|
|
/** @file net.cpp
|
|
|
|
* @author Alex Leverington <nessence@gmail.com>
|
2014-04-08 05:32:00 +00:00
|
|
|
* @date 2014
|
|
|
|
*/
|
|
|
|
|
|
|
|
#include <boost/test/unit_test.hpp>
|
2014-12-18 07:35:12 +00:00
|
|
|
#include <libdevcore/Worker.h>
|
2014-12-22 00:51:19 +00:00
|
|
|
#include <libdevcrypto/Common.h>
|
2014-12-18 07:35:12 +00:00
|
|
|
#include <libp2p/UDP.h>
|
2014-12-22 02:56:07 +00:00
|
|
|
#include <libp2p/NodeTable.h>
|
2014-04-08 05:32:00 +00:00
|
|
|
using namespace std;
|
2014-09-05 15:09:58 +00:00
|
|
|
using namespace dev;
|
2014-12-18 07:35:12 +00:00
|
|
|
using namespace dev::p2p;
|
|
|
|
namespace ba = boost::asio;
|
|
|
|
namespace bi = ba::ip;
|
2014-04-08 05:32:00 +00:00
|
|
|
|
2014-12-23 07:56:56 +00:00
|
|
|
BOOST_AUTO_TEST_SUITE(p2p)
|
|
|
|
|
2014-12-22 00:51:19 +00:00
|
|
|
/**
|
|
|
|
* Only used for testing. Not useful beyond tests.
|
|
|
|
*/
|
|
|
|
class TestHost: public Worker
|
|
|
|
{
|
|
|
|
public:
|
|
|
|
TestHost(): Worker("test",0), m_io() {};
|
2014-12-23 07:56:56 +00:00
|
|
|
virtual ~TestHost() { m_io.stop(); stopWorking(); }
|
2014-12-22 00:51:19 +00:00
|
|
|
void start() { startWorking(); }
|
|
|
|
void doWork() { m_io.run(); }
|
|
|
|
|
|
|
|
protected:
|
|
|
|
ba::io_service m_io;
|
|
|
|
};
|
|
|
|
|
2014-12-23 07:56:56 +00:00
|
|
|
struct TestNodeTable: public NodeTable
|
|
|
|
{
|
|
|
|
/// Constructor
|
|
|
|
using NodeTable::NodeTable;
|
|
|
|
|
2014-12-23 08:25:36 +00:00
|
|
|
void setup(std::vector<std::pair<KeyPair,unsigned>> const& _testNodes)
|
2014-12-23 07:56:56 +00:00
|
|
|
{
|
|
|
|
/// Phase 2 test: pre-populate *expected* ping-responses, send pings
|
|
|
|
|
|
|
|
bi::address ourIp = bi::address::from_string("127.0.0.1");
|
|
|
|
uint16_t ourPort = 30300;
|
|
|
|
bi::udp::endpoint ourEndpoint(ourIp, ourPort);
|
|
|
|
|
2014-12-23 08:25:36 +00:00
|
|
|
for (auto& n: _testNodes)
|
2014-12-23 07:56:56 +00:00
|
|
|
ping(bi::udp::endpoint(ourIp, n.second));
|
|
|
|
}
|
|
|
|
|
|
|
|
void reset()
|
|
|
|
{
|
|
|
|
Guard l(x_state);
|
|
|
|
for (auto& n: m_state) n.nodes.clear();
|
|
|
|
}
|
|
|
|
};
|
|
|
|
|
2014-12-22 00:51:19 +00:00
|
|
|
/**
|
|
|
|
* Only used for testing. Not useful beyond tests.
|
|
|
|
*/
|
2014-12-23 07:56:56 +00:00
|
|
|
struct TestNodeTableHost: public TestHost
|
2014-12-22 00:51:19 +00:00
|
|
|
{
|
2014-12-23 08:25:36 +00:00
|
|
|
TestNodeTableHost(): m_alias(KeyPair::create()), nodeTable(new TestNodeTable(m_io, m_alias)) {};
|
|
|
|
|
|
|
|
void generateTestNodes(int _count = 10)
|
|
|
|
{
|
|
|
|
asserts(_count < 1000);
|
|
|
|
static uint16_t s_basePort = 30500;
|
|
|
|
|
|
|
|
m_testNodes.clear();
|
|
|
|
for (auto i = 0; i < _count; i++)
|
|
|
|
{
|
|
|
|
KeyPair k = KeyPair::create();
|
|
|
|
m_testNodes.push_back(make_pair(k,s_basePort+i));
|
|
|
|
testNodes.push_back(make_shared<TestNodeTable>(m_io,k,s_basePort+i));
|
|
|
|
}
|
|
|
|
}
|
|
|
|
std::vector<std::pair<KeyPair,unsigned>> m_testNodes; // keypair and port
|
|
|
|
|
|
|
|
void setup()
|
|
|
|
{
|
|
|
|
generateTestNodes();
|
|
|
|
nodeTable->setup(m_testNodes);
|
|
|
|
}
|
|
|
|
|
|
|
|
KeyPair m_alias;
|
2014-12-23 07:56:56 +00:00
|
|
|
shared_ptr<TestNodeTable> nodeTable;
|
2014-12-23 08:25:36 +00:00
|
|
|
std::vector<shared_ptr<TestNodeTable>> testNodes;
|
2014-12-22 00:51:19 +00:00
|
|
|
};
|
|
|
|
|
|
|
|
class TestUDPSocket: UDPSocketEvents, public TestHost
|
|
|
|
{
|
|
|
|
public:
|
|
|
|
TestUDPSocket(): m_socket(new UDPSocket<TestUDPSocket, 1024>(m_io, *this, 30300)) {}
|
2014-12-23 07:56:56 +00:00
|
|
|
|
2014-12-18 07:35:12 +00:00
|
|
|
void onDisconnected(UDPSocketFace*) {};
|
2014-12-22 00:51:19 +00:00
|
|
|
void onReceived(UDPSocketFace*, bi::udp::endpoint const&, bytesConstRef _packet) { if (_packet.toString() == "AAAA") success = true; }
|
2014-04-08 05:32:00 +00:00
|
|
|
|
2014-12-22 00:51:19 +00:00
|
|
|
shared_ptr<UDPSocket<TestUDPSocket, 1024>> m_socket;
|
2014-12-18 07:35:12 +00:00
|
|
|
|
|
|
|
bool success = false;
|
|
|
|
};
|
2014-04-08 05:32:00 +00:00
|
|
|
|
2014-12-22 00:51:19 +00:00
|
|
|
BOOST_AUTO_TEST_CASE(kademlia)
|
2014-12-18 07:35:12 +00:00
|
|
|
{
|
2014-12-23 08:25:36 +00:00
|
|
|
TestNodeTableHost node;
|
|
|
|
node.start();
|
|
|
|
node.nodeTable->join(); // ideally, joining with empty node table logs warning we can check for
|
|
|
|
node.setup();
|
|
|
|
sleep(1);
|
2014-12-22 00:51:19 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
BOOST_AUTO_TEST_CASE(test_txrx_one)
|
|
|
|
{
|
|
|
|
UDPDatagram d(bi::udp::endpoint(boost::asio::ip::address::from_string("127.0.0.1"), 30300), bytes({65,65,65,65}));
|
|
|
|
TestUDPSocket a; a.m_socket->connect(); a.start();
|
2014-12-18 07:35:12 +00:00
|
|
|
a.m_socket->send(d);
|
|
|
|
sleep(1);
|
|
|
|
BOOST_REQUIRE_EQUAL(true, a.success);
|
2014-04-08 05:32:00 +00:00
|
|
|
}
|
2014-12-18 07:35:12 +00:00
|
|
|
|
|
|
|
BOOST_AUTO_TEST_SUITE_END()
|
|
|
|
|