From 2ae1f53bd882b296111833eec860ad42305d80df Mon Sep 17 00:00:00 2001
From: clemahieu <clemahieu@gmail.com>
Date: Thu, 5 May 2022 13:32:09 +0100
Subject: [PATCH] Adding unhandled case for message_visitor

---
 nano/node/common.hpp           | 56 +++++++++++++++++++++++++++-------
 nano/node/transport/inproc.cpp | 42 +------------------------
 2 files changed, 46 insertions(+), 52 deletions(-)

diff --git a/nano/node/common.hpp b/nano/node/common.hpp
index 6f7b86d322..298f72feef 100644
--- a/nano/node/common.hpp
+++ b/nano/node/common.hpp
@@ -497,17 +497,51 @@ class node_id_handshake final : public message
 class message_visitor
 {
 public:
-	virtual void keepalive (nano::keepalive const &) = 0;
-	virtual void publish (nano::publish const &) = 0;
-	virtual void confirm_req (nano::confirm_req const &) = 0;
-	virtual void confirm_ack (nano::confirm_ack const &) = 0;
-	virtual void bulk_pull (nano::bulk_pull const &) = 0;
-	virtual void bulk_pull_account (nano::bulk_pull_account const &) = 0;
-	virtual void bulk_push (nano::bulk_push const &) = 0;
-	virtual void frontier_req (nano::frontier_req const &) = 0;
-	virtual void node_id_handshake (nano::node_id_handshake const &) = 0;
-	virtual void telemetry_req (nano::telemetry_req const &) = 0;
-	virtual void telemetry_ack (nano::telemetry_ack const &) = 0;
+	virtual void keepalive (nano::keepalive const & message)
+	{
+		unhandled (message);
+	};
+	virtual void publish (nano::publish const & message)
+	{
+		unhandled (message);
+	}
+	virtual void confirm_req (nano::confirm_req const & message)
+	{
+		unhandled (message);
+	}
+	virtual void confirm_ack (nano::confirm_ack const & message)
+	{
+		unhandled (message);
+	}
+	virtual void bulk_pull (nano::bulk_pull const & message)
+	{
+		unhandled (message);
+	}
+	virtual void bulk_pull_account (nano::bulk_pull_account const & message)
+	{
+		unhandled (message);
+	}
+	virtual void bulk_push (nano::bulk_push const & message)
+	{
+		unhandled (message);
+	}
+	virtual void frontier_req (nano::frontier_req const & message)
+	{
+		unhandled (message);
+	}
+	virtual void node_id_handshake (nano::node_id_handshake const & message)
+	{
+		unhandled (message);
+	}
+	virtual void telemetry_req (nano::telemetry_req const & message)
+	{
+		unhandled (message);
+	}
+	virtual void telemetry_ack (nano::telemetry_ack const & message)
+	{
+		unhandled (message);
+	}
+	virtual void unhandled (nano::message const &) { };
 	virtual ~message_visitor ();
 };
 
diff --git a/nano/node/transport/inproc.cpp b/nano/node/transport/inproc.cpp
index eec7a5e6e2..c7228df5ff 100644
--- a/nano/node/transport/inproc.cpp
+++ b/nano/node/transport/inproc.cpp
@@ -41,47 +41,7 @@ class message_visitor_inbound : public nano::message_visitor
 	// the channel to reply to, if a reply is generated
 	std::shared_ptr<nano::transport::inproc::channel> channel;
 
-	void keepalive (nano::keepalive const & message)
-	{
-		inbound (message, channel);
-	}
-	void publish (nano::publish const & message)
-	{
-		inbound (message, channel);
-	}
-	void confirm_req (nano::confirm_req const & message)
-	{
-		inbound (message, channel);
-	}
-	void confirm_ack (nano::confirm_ack const & message)
-	{
-		inbound (message, channel);
-	}
-	void bulk_pull (nano::bulk_pull const & message)
-	{
-		inbound (message, channel);
-	}
-	void bulk_pull_account (nano::bulk_pull_account const & message)
-	{
-		inbound (message, channel);
-	}
-	void bulk_push (nano::bulk_push const & message)
-	{
-		inbound (message, channel);
-	}
-	void frontier_req (nano::frontier_req const & message)
-	{
-		inbound (message, channel);
-	}
-	void node_id_handshake (nano::node_id_handshake const & message)
-	{
-		inbound (message, channel);
-	}
-	void telemetry_req (nano::telemetry_req const & message)
-	{
-		inbound (message, channel);
-	}
-	void telemetry_ack (nano::telemetry_ack const & message)
+	void unhandled (nano::message const & message) override
 	{
 		inbound (message, channel);
 	}