aboutsummaryrefslogtreecommitdiffstats
path: root/whisperTopic.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'whisperTopic.cpp')
-rw-r--r--whisperTopic.cpp301
1 files changed, 301 insertions, 0 deletions
diff --git a/whisperTopic.cpp b/whisperTopic.cpp
new file mode 100644
index 00000000..0ea681b6
--- /dev/null
+++ b/whisperTopic.cpp
@@ -0,0 +1,301 @@
+/*
+ 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 Gav Wood <i@gavwood.com>
+ * @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)
+
+#if ALEX_HASH_FIXED_NETWORKING
+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();
+ wh->post(us.sec(), RLPStream().append(1).out(), BuildTopic("test"));
+ this_thread::sleep_for(chrono::milliseconds(250));
+
+ listener.join();
+ done = true;
+ forwarder.join();
+ g_logVerbosity = oldLogVerbosity;
+
+ BOOST_REQUIRE_EQUAL(result, 1);
+}
+
+BOOST_AUTO_TEST_CASE(asyncforwarding)
+{
+ cnote << "Testing Whisper async forwarding...";
+ auto oldLogVerbosity = g_logVerbosity;
+ g_logVerbosity = 2;
+
+ unsigned result = 0;
+ bool done = false;
+
+ // Host must be configured not to share peers.
+ Host host1("Forwarder", NetworkPreferences(30305, "", false, true));
+ host1.setIdealPeerCount(1);
+ auto whost1 = host1.registerCapability(new WhisperHost());
+ host1.start();
+ while (!host1.isStarted())
+ this_thread::sleep_for(chrono::milliseconds(2));
+
+ bool startedForwarder = false;
+ std::thread forwarder([&]()
+ {
+ setThreadName("forwarder");
+
+ this_thread::sleep_for(chrono::milliseconds(500));
+// ph.addNode("127.0.0.1", 30303, 30303);
+
+ startedForwarder = true;
+
+ /// Only interested in odd packets
+ auto w = whost1->installWatch(BuildTopicMask("test"));
+
+ while (!done)
+ {
+ for (auto i: whost1->checkWatch(w))
+ {
+ Message msg = whost1->envelope(i).open(whost1->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(2));
+
+ {
+ Host host2("Sender", NetworkPreferences(30300, "", false, true));
+ host2.setIdealPeerCount(1);
+ shared_ptr<WhisperHost> whost2 = host2.registerCapability(new WhisperHost());
+ host2.start();
+ while (!host2.isStarted())
+ this_thread::sleep_for(chrono::milliseconds(2));
+ host2.addNode(host1.id(), "127.0.0.1", 30305, 30305);
+
+ while (!host2.peerCount())
+ this_thread::sleep_for(chrono::milliseconds(5));
+
+ KeyPair us = KeyPair::create();
+ whost2->post(us.sec(), RLPStream().append(1).out(), BuildTopic("test"));
+ this_thread::sleep_for(chrono::milliseconds(250));
+ }
+
+ {
+ Host ph("Listener", NetworkPreferences(30300, "", false, true));
+ ph.setIdealPeerCount(1);
+ shared_ptr<WhisperHost> wh = ph.registerCapability(new WhisperHost());
+ ph.start();
+ while (!ph.isStarted())
+ this_thread::sleep_for(chrono::milliseconds(2));
+ ph.addNode(host1.id(), "127.0.0.1", 30305, 30305);
+
+ /// 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));
+ }
+ }
+
+ done = true;
+ forwarder.join();
+ g_logVerbosity = oldLogVerbosity;
+
+ BOOST_REQUIRE_EQUAL(result, 1);
+}
+#endif
+
+BOOST_AUTO_TEST_SUITE_END()
generated by cgit v1.2.3 (git 2.25.1) at 2025-05-04 20:31:17 +0800