diff --git a/src/daemon/daemon.cpp b/src/daemon/daemon.cpp index e069d1b65..d67a7ff99 100644 --- a/src/daemon/daemon.cpp +++ b/src/daemon/daemon.cpp @@ -92,7 +92,7 @@ struct zmq_internals { explicit zmq_internals(t_core& core, t_p2p& p2p, const bool restricted) : rpc_handler{core.get(), p2p.get(), restricted} - , server{rpc_handler} + , server{rpc_handler, restricted} {} cryptonote::rpc::DaemonHandler rpc_handler; diff --git a/src/rpc/zmq_server.cpp b/src/rpc/zmq_server.cpp index 3aca81101..b9a572f9c 100644 --- a/src/rpc/zmq_server.cpp +++ b/src/rpc/zmq_server.cpp @@ -93,8 +93,9 @@ namespace namespace rpc { -ZmqServer::ZmqServer(RpcHandler& h) : +ZmqServer::ZmqServer(RpcHandler& h, const bool restricted) : handler(h), + restricted(restricted), context(zmq_init(num_zmq_threads)), rep_socket(nullptr), pub_socket(nullptr), @@ -183,7 +184,10 @@ void ZmqServer::serve() } else // no errors { - MDEBUG("Received RPC request: \"" << *message << "\""); + if (restricted) + MDEBUG("Received RPC request"); + else + MDEBUG("Received RPC request: \"" << *message << "\""); epee::byte_slice response = handler.handle(std::move(*message)); const boost::string_ref response_view{reinterpret_cast(response.data()), response.size()}; diff --git a/src/rpc/zmq_server.h b/src/rpc/zmq_server.h index 2d4049baa..b85c3b175 100644 --- a/src/rpc/zmq_server.h +++ b/src/rpc/zmq_server.h @@ -51,7 +51,7 @@ class ZmqServer final { public: - ZmqServer(RpcHandler& h); + ZmqServer(RpcHandler& h, bool restricted); ~ZmqServer(); @@ -68,6 +68,7 @@ class ZmqServer final private: RpcHandler& handler; + const bool restricted; net::zmq::context context; diff --git a/tests/trezor/daemon.cpp b/tests/trezor/daemon.cpp index 8eeb30377..277a68549 100644 --- a/tests/trezor/daemon.cpp +++ b/tests/trezor/daemon.cpp @@ -242,7 +242,7 @@ bool mock_daemon::run_main() { CHECK_AND_ASSERT_THROW_MES(m_rpc_server.run(2, false), "Failed to start RPC"); cryptonote::rpc::DaemonHandler rpc_daemon_handler(*m_core, m_server); - cryptonote::rpc::ZmqServer zmq_server(rpc_daemon_handler); + cryptonote::rpc::ZmqServer zmq_server(rpc_daemon_handler, false); if (m_start_zmq) { diff --git a/tests/unit_tests/zmq_rpc.cpp b/tests/unit_tests/zmq_rpc.cpp index 38b5f083f..33c94bbf9 100644 --- a/tests/unit_tests/zmq_rpc.cpp +++ b/tests/unit_tests/zmq_rpc.cpp @@ -338,7 +338,7 @@ namespace zmq_server() : zmq_base(), handler(), - server(handler), + server(handler, false), pub(), sub() {