Minecon724/dncurrency
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages 1 Wiki Activity Actions 1

Rename to fake_work_peer

Browse source
This commit is contained in:
Piotr Wójcik 2024-05-12 17:17:47 +02:00
commit d8c6a79129
2 changed files with 16 additions and 16 deletions
Download patch file Download diff file Expand all files Collapse all files

12
nano/core_test/distributed_work.cpp
View file

@ -132,7 +132,7 @@ TEST (distributed_work, peer)
work = work_a;
done = true;
};
auto work_peer (std::make_shared<fake_work_peer> (node->work, node->io_ctx, system.get_available_port (), work_peer_type::good));
auto work_peer (std::make_shared<fake_work_peer> (node->work, node->io_ctx, system.get_available_port (), fake_work_peer_type::good));
work_peer->start ();
decltype (node->config.work_peers) peers;
peers.emplace_back ("::ffff:127.0.0.1", work_peer->port ());
@ -158,7 +158,7 @@ TEST (distributed_work, peer_malicious)
work = work_a;
done = true;
};
auto malicious_peer (std::make_shared<fake_work_peer> (node->work, node->io_ctx, system.get_available_port (), work_peer_type::malicious));
auto malicious_peer (std::make_shared<fake_work_peer> (node->work, node->io_ctx, system.get_available_port (), fake_work_peer_type::malicious));
malicious_peer->start ();
decltype (node->config.work_peers) peers;
peers.emplace_back ("::ffff:127.0.0.1", malicious_peer->port ());
@ -176,7 +176,7 @@ TEST (distributed_work, peer_malicious)
// Test again with no local work generation enabled to make sure the malicious peer is sent more than one request
node->config.work_threads = 0;
ASSERT_FALSE (node->local_work_generation_enabled ());
auto malicious_peer2 (std::make_shared<fake_work_peer> (node->work, node->io_ctx, system.get_available_port (), work_peer_type::malicious));
auto malicious_peer2 (std::make_shared<fake_work_peer> (node->work, node->io_ctx, system.get_available_port (), fake_work_peer_type::malicious));
malicious_peer2->start ();
peers[0].second = malicious_peer2->port ();
ASSERT_FALSE (node->distributed_work.make (nano::work_version::work_1, hash, peers, node->network_params.work.base, {}, nano::account ()));
@ -201,9 +201,9 @@ TEST (distributed_work, DISABLED_peer_multi)
work = work_a;
done = true;
};
auto good_peer (std::make_shared<fake_work_peer> (node->work, node->io_ctx, system.get_available_port (), work_peer_type::good));
auto malicious_peer (std::make_shared<fake_work_peer> (node->work, node->io_ctx, system.get_available_port (), work_peer_type::malicious));
auto slow_peer (std::make_shared<fake_work_peer> (node->work, node->io_ctx, system.get_available_port (), work_peer_type::slow));
auto good_peer (std::make_shared<fake_work_peer> (node->work, node->io_ctx, system.get_available_port (), fake_work_peer_type::good));
auto malicious_peer (std::make_shared<fake_work_peer> (node->work, node->io_ctx, system.get_available_port (), fake_work_peer_type::malicious));
auto slow_peer (std::make_shared<fake_work_peer> (node->work, node->io_ctx, system.get_available_port (), fake_work_peer_type::slow));
good_peer->start ();
malicious_peer->start ();
slow_peer->start ();

20
nano/core_test/fakes/work_peer.hpp
View file

@ -23,19 +23,19 @@ using tcp = boost::asio::ip::tcp;
namespace
{
enum class work_peer_type
enum class fake_work_peer_type
{
good,
malicious,
slow
};
class work_peer_connection : public std::enable_shared_from_this<work_peer_connection>
class fake_work_peer_connection : public std::enable_shared_from_this<fake_work_peer_connection>
{
std::string const generic_error = "Unable to parse JSON";
public:
work_peer_connection (asio::io_context & ioc_a, work_peer_type const type_a, nano::work_version const version_a, nano::work_pool & pool_a, std::function<void (bool const)> on_generation_a, std::function<void ()> on_cancel_a) :
fake_work_peer_connection (asio::io_context & ioc_a, fake_work_peer_type const type_a, nano::work_version const version_a, nano::work_pool & pool_a, std::function<void (bool const)> on_generation_a, std::function<void ()> on_cancel_a) :
socket (ioc_a),
type (type_a),
version (version_a),
@ -52,7 +52,7 @@ public:
tcp::socket socket;
private:
work_peer_type type;
fake_work_peer_type type;
nano::work_version version;
nano::work_pool & work_pool;
beast::flat_buffer buffer{ 8192 };
@ -138,7 +138,7 @@ private:
void handle_generate (nano::block_hash const & hash_a)
{
if (type == work_peer_type::good)
if (type == fake_work_peer_type::good)
{
auto hash = hash_a;
auto request_difficulty = work_pool.network_constants.work.threshold_base (version);
@ -154,7 +154,7 @@ private:
ptree::write_json (ostream, message_l);
beast::ostream (this_l->response.body ()) << ostream.str ();
// Delay response by 500ms as a slow peer, immediate async call for a good peer
this_l->timer.expires_from_now (boost::posix_time::milliseconds (this_l->type == work_peer_type::slow ? 500 : 0));
this_l->timer.expires_from_now (boost::posix_time::milliseconds (this_l->type == fake_work_peer_type::slow ? 500 : 0));
this_l->timer.async_wait ([this_l, result] (boost::system::error_code const & ec) {
if (this_l->on_generation)
{
@ -164,7 +164,7 @@ private:
});
});
}
else if (type == work_peer_type::malicious)
else if (type == fake_work_peer_type::malicious)
{
// Respond immediately with no work
on_generation (false);
@ -197,7 +197,7 @@ class fake_work_peer : public std::enable_shared_from_this<fake_work_peer>
{
public:
fake_work_peer () = delete;
fake_work_peer (nano::work_pool & pool_a, asio::io_context & ioc_a, unsigned short port_a, work_peer_type const type_a, nano::work_version const version_a = nano::work_version::work_1) :
fake_work_peer (nano::work_pool & pool_a, asio::io_context & ioc_a, unsigned short port_a, fake_work_peer_type const type_a, nano::work_version const version_a = nano::work_version::work_1) :
pool (pool_a),
ioc (ioc_a),
acceptor (ioc_a, tcp::endpoint{ tcp::v4 (), port_a }),
@ -221,7 +221,7 @@ private:
void listen ()
{
std::weak_ptr<fake_work_peer> this_w (shared_from_this ());
auto connection (std::make_shared<work_peer_connection> (
auto connection (std::make_shared<fake_work_peer_connection> (
ioc, type, version, pool,
[this_w] (bool const good_generation) {
if (auto this_l = this_w.lock ())
@ -257,7 +257,7 @@ private:
nano::work_pool & pool;
asio::io_context & ioc;
tcp::acceptor acceptor;
work_peer_type const type;
fake_work_peer_type const type;
nano::work_version version;
};
}