diff --git a/locator/token_metadata.cc b/locator/token_metadata.cc
index 1fb61c939a..923910a876 100644
--- a/locator/token_metadata.cc
+++ b/locator/token_metadata.cc
@@ -27,6 +27,15 @@ const std::vector<token>& token_metadata::sorted_tokens() const {
     return _sorted_tokens;
 }
 
+std::vector<token> token_metadata::get_tokens(const inet_address& addr) const {
+    std::vector<token> res;
+    for (auto&& i : _token_to_endpoint_map) {
+        if (i.second == addr) {
+            res.push_back(i.first);
+        }
+    }
+    return res;
+}
 /**
  * Update token map with a single token/endpoint pair in normal state.
  */
diff --git a/locator/token_metadata.hh b/locator/token_metadata.hh
index 49f766b403..e86539305d 100644
--- a/locator/token_metadata.hh
+++ b/locator/token_metadata.hh
@@ -49,6 +49,10 @@ public:
     const token& first_token(const token& start);
     size_t first_token_index(const token& start);
     std::experimental::optional<inet_address> get_endpoint(const token& token) const;
+    std::vector<token> get_tokens(const inet_address& addr) const;
+    const std::map<token, inet_address>& get_token_to_endpoint() const {
+        return _token_to_endpoint_map;
+    }
 };
 
 }