From cdca20775f02ae87bbdea3cd5f9330527967cc2e Mon Sep 17 00:00:00 2001
From: Tomasz Grabiec <tgrabiec@scylladb.com>
Date: Mon, 4 Jan 2016 22:48:40 +0100
Subject: [PATCH] messaging_service: Introduce get_source()

---
 message/messaging_service.cc | 7 +++++++
 message/messaging_service.hh | 1 +
 2 files changed, 8 insertions(+)

diff --git a/message/messaging_service.cc b/message/messaging_service.cc
index dc02a2bd63..e669edd141 100644
--- a/message/messaging_service.cc
+++ b/message/messaging_service.cc
@@ -195,6 +195,13 @@ messaging_service::messaging_service(gms::inet_address ip
     });
 }
 
+msg_addr messaging_service::get_source(const rpc::client_info& cinfo) {
+    return msg_addr{
+        cinfo.retrieve_auxiliary<gms::inet_address>("baddr"),
+        cinfo.retrieve_auxiliary<uint32_t>("src_cpu_id")
+    };
+}
+
 messaging_service::~messaging_service() = default;
 
 uint16_t messaging_service::port() {
diff --git a/message/messaging_service.hh b/message/messaging_service.hh
index 479223f009..247a2b105d 100644
--- a/message/messaging_service.hh
+++ b/message/messaging_service.hh
@@ -524,6 +524,7 @@ public:
     void remove_error_rpc_client(messaging_verb verb, msg_addr id);
     void remove_rpc_client(msg_addr id);
     std::unique_ptr<rpc_protocol_wrapper>& rpc();
+    static msg_addr get_source(const rpc::client_info& client);
 };
 
 extern distributed<messaging_service> _the_messaging_service;