You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
264 lines
8.0 KiB
264 lines
8.0 KiB
/*
|
|
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/>.
|
|
*/
|
|
/** @file shhrpc.cpp
|
|
* @author Vladislav Gluhovsky <vlad@ethdev.com>
|
|
* @date July 2015
|
|
*/
|
|
|
|
#include <boost/test/unit_test.hpp>
|
|
#include <boost/lexical_cast.hpp>
|
|
#include <libdevcore/Log.h>
|
|
#include <libdevcore/CommonIO.h>
|
|
#include <libethcore/CommonJS.h>
|
|
#include <libwebthree/WebThree.h>
|
|
#include <libweb3jsonrpc/WebThreeStubServer.h>
|
|
#include <jsonrpccpp/server/connectors/httpserver.h>
|
|
#include <jsonrpccpp/client/connectors/httpclient.h>
|
|
#include <test/TestHelper.h>
|
|
#include <test/libweb3jsonrpc/webthreestubclient.h>
|
|
#include <libethcore/KeyManager.h>
|
|
#include <libp2p/Common.h>
|
|
#include <libwhisper/WhisperHost.h>
|
|
|
|
using namespace std;
|
|
using namespace dev;
|
|
using namespace dev::eth;
|
|
using namespace dev::p2p;
|
|
using namespace dev::shh;
|
|
namespace js = json_spirit;
|
|
|
|
WebThreeDirect* web3;
|
|
unique_ptr<WebThreeStubServer> jsonrpcServer;
|
|
unique_ptr<WebThreeStubClient> jsonrpcClient;
|
|
static uint16_t const web3port = 30333;
|
|
|
|
struct Setup
|
|
{
|
|
Setup()
|
|
{
|
|
dev::p2p::NodeIPEndpoint::test_allowLocal = true;
|
|
|
|
static bool setup = false;
|
|
if (!setup)
|
|
{
|
|
setup = true;
|
|
NetworkPreferences nprefs(std::string(), web3port, false);
|
|
web3 = new WebThreeDirect("shhrpc-web3", "", WithExisting::Trust, {"eth", "shh"}, nprefs);
|
|
web3->setIdealPeerCount(1);
|
|
web3->ethereum()->setForceMining(false);
|
|
auto server = new jsonrpc::HttpServer(8080);
|
|
vector<KeyPair> v;
|
|
KeyManager keyMan;
|
|
TrivialGasPricer gp;
|
|
jsonrpcServer = unique_ptr<WebThreeStubServer>(new WebThreeStubServer(*server, *web3, nullptr, v, keyMan, gp));
|
|
jsonrpcServer->setIdentities({});
|
|
jsonrpcServer->StartListening();
|
|
auto client = new jsonrpc::HttpClient("http://localhost:8080");
|
|
jsonrpcClient = unique_ptr<WebThreeStubClient>(new WebThreeStubClient(*client));
|
|
}
|
|
}
|
|
|
|
~Setup()
|
|
{
|
|
dev::p2p::NodeIPEndpoint::test_allowLocal = false;
|
|
}
|
|
};
|
|
|
|
|
|
BOOST_FIXTURE_TEST_SUITE(shhrpc, Setup)
|
|
|
|
BOOST_AUTO_TEST_CASE(basic)
|
|
{
|
|
cnote << "Testing web3 basic functionality...";
|
|
|
|
web3->startNetwork();
|
|
unsigned const step = 10;
|
|
for (unsigned i = 0; i < 3000 && !web3->haveNetwork(); i += step)
|
|
this_thread::sleep_for(chrono::milliseconds(step));
|
|
|
|
BOOST_REQUIRE(web3->haveNetwork());
|
|
|
|
uint16_t const port2 = 30334;
|
|
NetworkPreferences prefs2("127.0.0.1", port2, false);
|
|
string const version2 = "shhrpc-host2";
|
|
Host host2(version2, prefs2);
|
|
auto whost2 = host2.registerCapability(new WhisperHost());
|
|
host2.start();
|
|
|
|
for (unsigned i = 0; i < 3000 && !host2.haveNetwork(); i += step)
|
|
this_thread::sleep_for(chrono::milliseconds(step));
|
|
|
|
BOOST_REQUIRE(host2.haveNetwork());
|
|
|
|
web3->addNode(host2.id(), NodeIPEndpoint(bi::address::from_string("127.0.0.1"), port2, port2));
|
|
|
|
for (unsigned i = 0; i < 3000 && (!web3->peerCount() || !host2.peerCount()); i += step)
|
|
this_thread::sleep_for(chrono::milliseconds(step));
|
|
|
|
BOOST_REQUIRE_EQUAL(host2.peerCount(), 1);
|
|
BOOST_REQUIRE_EQUAL(web3->peerCount(), 1);
|
|
|
|
vector<PeerSessionInfo> vpeers = web3->peers();
|
|
BOOST_REQUIRE(!vpeers.empty());
|
|
PeerSessionInfo const& peer = vpeers.back();
|
|
BOOST_REQUIRE_EQUAL(peer.id, host2.id());
|
|
BOOST_REQUIRE_EQUAL(peer.port, port2);
|
|
BOOST_REQUIRE_EQUAL(peer.clientVersion, version2);
|
|
|
|
web3->stopNetwork();
|
|
|
|
for (unsigned i = 0; i < 3000 && (web3->haveNetwork() || host2.haveNetwork()); i += step)
|
|
this_thread::sleep_for(chrono::milliseconds(step));
|
|
|
|
BOOST_REQUIRE(!web3->peerCount());
|
|
BOOST_REQUIRE(!host2.peerCount());
|
|
}
|
|
|
|
BOOST_AUTO_TEST_CASE(send)
|
|
{
|
|
cnote << "Testing web3 send...";
|
|
|
|
bool sent = false;
|
|
bool ready = false;
|
|
unsigned result = 0;
|
|
unsigned const messageCount = 10;
|
|
unsigned const step = 10;
|
|
uint16_t port2 = 30337;
|
|
|
|
Host host2("shhrpc-host2", NetworkPreferences("127.0.0.1", port2, false));
|
|
host2.setIdealPeerCount(1);
|
|
auto whost2 = host2.registerCapability(new WhisperHost());
|
|
host2.start();
|
|
web3->startNetwork();
|
|
|
|
std::thread listener([&]()
|
|
{
|
|
setThreadName("listener");
|
|
ready = true;
|
|
auto w = whost2->installWatch(BuildTopicMask("odd"));
|
|
set<unsigned> received;
|
|
for (unsigned x = 0; x < 9000 && !sent; x += step)
|
|
this_thread::sleep_for(chrono::milliseconds(step));
|
|
|
|
for (unsigned x = 0, last = 0; x < 100 && received.size() < messageCount; ++x)
|
|
{
|
|
this_thread::sleep_for(chrono::milliseconds(50));
|
|
for (auto i: whost2->checkWatch(w))
|
|
{
|
|
Message msg = whost2->envelope(i).open(whost2->fullTopics(w));
|
|
last = RLP(msg.payload()).toInt<unsigned>();
|
|
if (received.insert(last).second)
|
|
result += last;
|
|
}
|
|
}
|
|
});
|
|
|
|
for (unsigned i = 0; i < 2000 && (!host2.haveNetwork() || !web3->haveNetwork()); i += step)
|
|
this_thread::sleep_for(chrono::milliseconds(step));
|
|
|
|
BOOST_REQUIRE(host2.haveNetwork());
|
|
BOOST_REQUIRE(web3->haveNetwork());
|
|
|
|
web3->requirePeer(host2.id(), NodeIPEndpoint(bi::address::from_string("127.0.0.1"), port2, port2));
|
|
|
|
for (unsigned i = 0; i < 3000 && (!web3->peerCount() || !host2.peerCount()); i += step)
|
|
this_thread::sleep_for(chrono::milliseconds(step));
|
|
|
|
BOOST_REQUIRE_EQUAL(host2.peerCount(), 1);
|
|
BOOST_REQUIRE_EQUAL(web3->peerCount(), 1);
|
|
|
|
KeyPair us = KeyPair::create();
|
|
for (unsigned i = 0; i < messageCount; ++i)
|
|
{
|
|
web3->whisper()->post(us.sec(), RLPStream().append(i * i).out(), BuildTopic(i)(i % 2 ? "odd" : "even"), 777000, 1);
|
|
this_thread::sleep_for(chrono::milliseconds(50));
|
|
}
|
|
|
|
sent = true;
|
|
auto messages = web3->whisper()->all();
|
|
BOOST_REQUIRE_EQUAL(messages.size(), messageCount);
|
|
|
|
listener.join();
|
|
BOOST_REQUIRE_EQUAL(result, 1 + 9 + 25 + 49 + 81);
|
|
}
|
|
|
|
BOOST_AUTO_TEST_CASE(receive)
|
|
{
|
|
cnote << "Testing web3 receive...";
|
|
|
|
bool sent = false;
|
|
bool ready = false;
|
|
unsigned result = 0;
|
|
unsigned const messageCount = 6;
|
|
unsigned const step = 10;
|
|
uint16_t port2 = 30338;
|
|
Host host2("shhrpc-host2", NetworkPreferences("127.0.0.1", port2, false));
|
|
host2.setIdealPeerCount(1);
|
|
auto whost2 = host2.registerCapability(new WhisperHost());
|
|
host2.start();
|
|
web3->startNetwork();
|
|
|
|
std::thread listener([&]()
|
|
{
|
|
setThreadName("listener");
|
|
ready = true;
|
|
auto w = web3->whisper()->installWatch(BuildTopicMask("odd"));
|
|
|
|
set<unsigned> received;
|
|
for (unsigned x = 0; x < 9000 && !sent; x += step)
|
|
this_thread::sleep_for(chrono::milliseconds(step));
|
|
|
|
for (unsigned x = 0, last = 0; x < 100 && received.size() < messageCount; ++x)
|
|
{
|
|
this_thread::sleep_for(chrono::milliseconds(50));
|
|
for (auto i: web3->whisper()->checkWatch(w))
|
|
{
|
|
Message msg = web3->whisper()->envelope(i).open(web3->whisper()->fullTopics(w));
|
|
last = RLP(msg.payload()).toInt<unsigned>();
|
|
if (received.insert(last).second)
|
|
result += last;
|
|
}
|
|
}
|
|
});
|
|
|
|
for (unsigned i = 0; i < 2000 && (!host2.haveNetwork() || !web3->haveNetwork()); i += step)
|
|
this_thread::sleep_for(chrono::milliseconds(step));
|
|
|
|
BOOST_REQUIRE(host2.haveNetwork());
|
|
BOOST_REQUIRE(web3->haveNetwork());
|
|
|
|
host2.addNode(web3->id(), NodeIPEndpoint(bi::address::from_string("127.0.0.1"), web3port, web3port));
|
|
|
|
for (unsigned i = 0; i < 3000 && (!web3->peerCount() || !host2.peerCount()); i += step)
|
|
this_thread::sleep_for(chrono::milliseconds(step));
|
|
|
|
BOOST_REQUIRE_EQUAL(host2.peerCount(), 1);
|
|
BOOST_REQUIRE_EQUAL(web3->peerCount(), 1);
|
|
|
|
KeyPair us = KeyPair::create();
|
|
for (unsigned i = 0; i < messageCount; ++i)
|
|
{
|
|
web3->whisper()->post(us.sec(), RLPStream().append(i * i * i).out(), BuildTopic(i)(i % 2 ? "odd" : "even"), 777000, 1);
|
|
this_thread::sleep_for(chrono::milliseconds(50));
|
|
}
|
|
|
|
sent = true;
|
|
listener.join();
|
|
BOOST_REQUIRE_EQUAL(result, 1 + 27 + 125);
|
|
}
|
|
|
|
BOOST_AUTO_TEST_SUITE_END()
|
|
|