<commit_before>/*
	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 whisperTopic.cpp
 * @author <NAME> <<EMAIL>>
 * @date 2014
 */
#include <functional>
#include <boost/test/unit_test.hpp>
#include <libp2p/Host.h>
#include <libwhisper/WhisperPeer.h>
#include <libwhisper/WhisperHost.h>
using namespace std;
using namespace dev;
using namespace dev::p2p;
using namespace dev::shh;

BOOST_AUTO_TEST_SUITE(whisper)

BOOST_AUTO_TEST_CASE(topic)
{
	cnote << "Testing Whisper...";
	auto oldLogVerbosity = g_logVerbosity;
	g_logVerbosity = 0;

	Host phOther("Test", NetworkPreferences(30303, "127.0.0.1", false, true));
	auto whOther = phOther.registerCapability(new WhisperHost());
	phOther.start();
	
	bool started = false;
	unsigned result = 0;
	std::thread listener([&]()
	{
		setThreadName("other");
		started = true;

		/// Only interested in odd packets
		auto w = whOther->installWatch(BuildTopicMask("odd"));

		started = true;
		set<unsigned> received;

		for (int iterout = 0, last = 0; iterout < 200 && last < 81; ++iterout)
		{
			for (auto i: whOther->checkWatch(w))
			{
				Message msg = whOther->envelope(i).open(whOther->fullTopic(w));
				last = RLP(msg.payload()).toInt<unsigned>();
				if (received.count(last))
					continue;
				received.insert(last);
				cnote << "New message from:" << msg.from().abridged() << RLP(msg.payload()).toInt<unsigned>();
				result += last;
			}
			this_thread::sleep_for(chrono::milliseconds(50));
		}
		
	});
	
	Host ph("Test", NetworkPreferences(30300, "127.0.0.1", false, true));
	auto wh = ph.registerCapability(new WhisperHost());
	ph.start();
	
	this_thread::sleep_for(chrono::milliseconds(100));
	ph.addNode(phOther.id(), "127.0.0.1", 30303, 30303);
	
	this_thread::sleep_for(chrono::milliseconds(500));

	while (!started)
		this_thread::sleep_for(chrono::milliseconds(2));

	KeyPair us = KeyPair::create();
	for (int i = 0; i < 10; ++i)
	{
		wh->post(us.sec(), RLPStream().append(i * i).out(), BuildTopic(i)(i % 2 ? "odd" : "even"));
		this_thread::sleep_for(chrono::milliseconds(250));
	}

	listener.join();
	g_logVerbosity = oldLogVerbosity;

	BOOST_REQUIRE_EQUAL(result, 1 + 9 + 25 + 49 + 81);
}

BOOST_AUTO_TEST_CASE(forwarding)
{
	cnote << "Testing Whisper forwarding...";
	auto oldLogVerbosity = g_logVerbosity;
	g_logVerbosity = 0;

	unsigned result = 0;
	bool done = false;

	bool startedListener = false;
	Public phid;
	std::thread listener([&]()
	{
		setThreadName("listener");

		// Host must be configured not to share peers.
		Host ph("Listner", NetworkPreferences(30303, "", false, true));
		ph.setIdealPeerCount(0);
		auto wh = ph.registerCapability(new WhisperHost());
		ph.start();
		phid = ph.id();

		startedListener = true;

		/// Only interested in odd packets
		auto w = wh->installWatch(BuildTopicMask("test"));

		for (int i = 0; i < 200 && !result; ++i)
		{
			for (auto i: wh->checkWatch(w))
			{
				Message msg = wh->envelope(i).open(wh->fullTopic(w));
				unsigned last = RLP(msg.payload()).toInt<unsigned>();
				cnote << "New message from:" << msg.from().abridged() << RLP(msg.payload()).toInt<unsigned>();
				result = last;
			}
			this_thread::sleep_for(chrono::milliseconds(50));
		}
	});

	bool startedForwarder = false;
	Public fwderid;
	std::thread forwarder([&]()
	{
		setThreadName("forwarder");

		while (!startedListener)
			this_thread::sleep_for(chrono::milliseconds(50));

		// Host must be configured not to share peers.
		Host ph("Forwarder", NetworkPreferences(30305, "", false, true));
		ph.setIdealPeerCount(0);
		auto wh = ph.registerCapability(new WhisperHost());
		this_thread::sleep_for(chrono::milliseconds(500));
		ph.start();
		fwderid = ph.id();

		this_thread::sleep_for(chrono::milliseconds(500));
		ph.addNode(phid, "127.0.0.1", 30303, 30303);

		startedForwarder = true;

		/// Only interested in odd packets
		auto w = wh->installWatch(BuildTopicMask("test"));

		while (!done)
		{
			for (auto i: wh->checkWatch(w))
			{
				Message msg = wh->envelope(i).open(wh->fullTopic(w));
				cnote << "New message from:" << msg.from().abridged() << RLP(msg.payload()).toInt<unsigned>();
			}
			this_thread::sleep_for(chrono::milliseconds(50));
		}
	});

	while (!startedForwarder)
		this_thread::sleep_for(chrono::milliseconds(50));

	Host ph("Sender", NetworkPreferences(30300, "", false, true));
	ph.setIdealPeerCount(0);
	shared_ptr<WhisperHost> wh = ph.registerCapability(new WhisperHost());
	this_thread::sleep_for(chrono::milliseconds(500));
	ph.start();
	this_thread::sleep_for(chrono::milliseconds(500));
	ph.addNode(fwderid, "127.0.0.1", 30305, 30305);

	KeyPair us = KeyPair::create();<commit_msg>update whisper topic and forwarding tests for node-discovery<commit_after>/*
	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 whisperTopic.cpp
 * @author <NAME> <<EMAIL>>
 * @date 2014
 */
#include <functional>
#include <boost/test/unit_test.hpp>
#include <libp2p/Host.h>
#include <libwhisper/WhisperPeer.h>
#include <libwhisper/WhisperHost.h>
using namespace std;
using namespace dev;
using namespace dev::p2p;
using namespace dev::shh;

BOOST_AUTO_TEST_SUITE(whisper)

BOOST_AUTO_TEST_CASE(topic)
{
	cnote << "Testing Whisper...";
	auto oldLogVerbosity = g_logVerbosity;
	g_logVerbosity = 0;

	Host host1("Test", NetworkPreferences(30303, "127.0.0.1", false, true));
	auto whost1 = host1.registerCapability(new WhisperHost());
	host1.start();
	
	while (!host1.isStarted())
		this_thread::sleep_for(chrono::milliseconds(2));
	
	bool started = false;
	unsigned result = 0;
	std::thread listener([&]()
	{
		setThreadName("other");
		started = true;

		/// Only interested in odd packets
		auto w = whost1->installWatch(BuildTopicMask("odd"));

		started = true;
		set<unsigned> received;

		for (int iterout = 0, last = 0; iterout < 200 && last < 81; ++iterout)
		{
			for (auto i: whost1->checkWatch(w))
			{
				Message msg = whost1->envelope(i).open(whost1->fullTopic(w));
				last = RLP(msg.payload()).toInt<unsigned>();
				if (received.count(last))
					continue;
				received.insert(last);
				cnote << "New message from:" << msg.from().abridged() << RLP(msg.payload()).toInt<unsigned>();
				result += last;
			}
			this_thread::sleep_for(chrono::milliseconds(50));
		}
		
	});
	
	Host host2("Test", NetworkPreferences(30300, "127.0.0.1", false, true));
	auto whost2 = host2.registerCapability(new WhisperHost());
	host2.start();
	
	while (!host2.isStarted())
		this_thread::sleep_for(chrono::milliseconds(2));
	
	this_thread::sleep_for(chrono::milliseconds(100));
	host2.addNode(host1.id(), "127.0.0.1", 30303, 30303);
	
	this_thread::sleep_for(chrono::milliseconds(500));

	while (!started)
		this_thread::sleep_for(chrono::milliseconds(2));

	KeyPair us = KeyPair::create();
	for (int i = 0; i < 10; ++i)
	{
		whost2->post(us.sec(), RLPStream().append(i * i).out(), BuildTopic(i)(i % 2 ? "odd" : "even"));
		this_thread::sleep_for(chrono::milliseconds(250));
	}

	listener.join();
	g_logVerbosity = oldLogVerbosity;

	BOOST_REQUIRE_EQUAL(result, 1 + 9 + 25 + 49 + 81);
}

BOOST_AUTO_TEST_CASE(forwarding)
{
	cnote << "Testing Whisper forwarding...";
	auto oldLogVerbosity = g_logVerbosity;
	g_logVerbosity = 0;
	
	// Host must be configured not to share peers.
	Host host1("Listner", NetworkPreferences(30303, "", false, true));
	host1.setIdealPeerCount(0);
	auto whost1 = host1.registerCapability(new WhisperHost());
	host1.start();
	while (!host1.isStarted())
		this_thread::sleep_for(chrono::milliseconds(2));
	
	unsigned result = 0;
	bool done = false;

	bool startedListener = false;
	std::thread listener([&]()
	{
		setThreadName("listener");

		startedListener = true;

		/// Only interested in odd packets
		auto w = whost1->installWatch(BuildTopicMask("test"));

		for (int i = 0; i < 200 && !result; ++i)
		{
			for (auto i: whost1->checkWatch(w))
			{
				Message msg = whost1->envelope(i).open(whost1->fullTopic(w));
				unsigned last = RLP(msg.payload()).toInt<unsigned>();
				cnote << "New message from:" << msg.from().abridged() << RLP(msg.payload()).toInt<unsigned>();
				result = last;
			}
			this_thread::sleep_for(chrono::milliseconds(50));
		}
	});

	
	// Host must be configured not to share peers.
	Host host2("Forwarder", NetworkPreferences(30305, "", false, true));
	host2.setIdealPeerCount(1);
	auto whost2 = host2.registerCapability(new WhisperHost());
	host2.start();
	while (!host2.isStarted())
		this_thread::sleep_for(chrono::milliseconds(2));
	
	Public fwderid;
	bool startedForwarder = false;
	std::thread forwarder([&]()
	{
		setThreadName("forwarder");

		while (!startedListener)
			this_thread::sleep_for(chrono::milliseconds(50));

		this_thread::sleep_for(chrono::milliseconds(500));
		host2.addNode(host1.id(), "127.0.0.1", 30303, 30303);

		startedForwarder = true;

		/// Only interested in odd packets
		auto w = whost2->installWatch(BuildTopicMask("test"));

		while (!done)
		{
			for (auto i: whost2->checkWatch(w))
			{
				Message msg = whost2->envelope(i).open(whost2->fullTopic(w));
				cnote << "New message from:" << msg.from().abridged() << RLP(msg.payload()).toInt<unsigned>();
			}
			this_thread::sleep_for(chrono::milliseconds(50));
		}
	});

	while (!startedForwarder)
		this_thread::sleep_for(chrono::milliseconds(50));

	Host ph("Sender", NetworkPreferences(30300, "", false, true));
	ph.setIdealPeerCount(1);
	shared_ptr<WhisperHost> wh = ph.registerCapability(new WhisperHost());
	ph.start();
	ph.addNode(host2.id(), "127.0.0.1", 30305, 30305);
	while (!ph.isStarted())
		this_thread::sleep_for(chrono::milliseconds(10));

	KeyPair us = KeyPair::create();
{'fraction_non_alphanumeric': 0.15882134914751667, 'fraction_numerical': 0.03196812453669385, 'mean_word_length': 4.725729442970822, 'pattern_counts': {'":': 0, '<': 87, '<?xml version=': 0, '>': 61, 'https://': 0, 'lorem ipsum': 0, 'www.': 2, 'xml': 0}, 'pii_count': 10, 'substrings_counts': 0, 'word_list_counts': {'cursed_substrings.json': 0, 'profanity_word_list.json': 0, 'sexual_word_list.json': 0, 'zh_pornsignals.json': 0}}
{'dir': 'git-commits-cleaned', 'id': '5128007', 'n_tokens_mistral': 4244, 'n_tokens_neox': 3720, 'n_words': 862}