intermediate changes

ref:cde9a383711a11544ce7e107a78147fb96cc4029

1 files changed, 61 insertions, 0 deletions

diff --git a/util/network/endpoint.h b/util/network/endpoint.h
new file mode 100644
index 0000000000..a3e59b4925
--- /dev/null
+++ b/util/network/endpoint.h
@@ -0,0 +1,61 @@
+#pragma once
+
+#include "address.h"
+
+#include <util/str_stl.h>
+
+//some equivalent boost::asio::ip::endpoint (easy for using pair ip:port)
+class TEndpoint {
+public:
+    using TAddrRef = NAddr::IRemoteAddrRef;
+
+    TEndpoint(const TAddrRef& addr);
+    TEndpoint();
+
+    inline const TAddrRef& Addr() const noexcept {
+        return Addr_;
+    }
+    inline const sockaddr* SockAddr() const {
+        return Addr_->Addr();
+    }
+    inline socklen_t SockAddrLen() const {
+        return Addr_->Len();
+    }
+
+    inline bool IsIpV4() const {
+        return Addr_->Addr()->sa_family == AF_INET;
+    }
+    inline bool IsIpV6() const {
+        return Addr_->Addr()->sa_family == AF_INET6;
+    }
+    inline bool IsUnix() const {
+        return Addr_->Addr()->sa_family == AF_UNIX;
+    }
+
+    inline TString IpToString() const {
+        return NAddr::PrintHost(*Addr_);
+    }
+
+    void SetPort(ui16 port);
+    ui16 Port() const noexcept;
+
+    size_t Hash() const;
+
+private:
+    TAddrRef Addr_;
+};
+
+template <>
+struct THash<TEndpoint> {
+    inline size_t operator()(const TEndpoint& ep) const {
+        return ep.Hash();
+    }
+};
+
+inline bool operator==(const TEndpoint& l, const TEndpoint& r) {
+    try {
+        return NAddr::IsSame(*l.Addr(), *r.Addr()) && l.Port() == r.Port();
+    } catch (...) {
+        return false;
+    }
+}