intermediate changes

ref:cde9a383711a11544ce7e107a78147fb96cc4029

1 files changed, 67 insertions, 0 deletions

diff --git a/util/network/endpoint.cpp b/util/network/endpoint.cpp
new file mode 100644
index 0000000000..9acdd06940
--- /dev/null
+++ b/util/network/endpoint.cpp
@@ -0,0 +1,67 @@
+#include "endpoint.h"
+#include "sock.h"
+
+TEndpoint::TEndpoint(const TEndpoint::TAddrRef& addr)
+    : Addr_(addr)
+{
+    const sockaddr* sa = Addr_->Addr();
+
+    if (sa->sa_family != AF_INET && sa->sa_family != AF_INET6 && sa->sa_family != AF_UNIX) {
+        ythrow yexception() << TStringBuf("endpoint can contain only ipv4, ipv6 or unix address");
+    }
+}
+
+TEndpoint::TEndpoint()
+    : Addr_(new NAddr::TIPv4Addr(TIpAddress(TIpHost(0), TIpPort(0))))
+{
+}
+
+void TEndpoint::SetPort(ui16 port) {
+    if (Port() == port || Addr_->Addr()->sa_family == AF_UNIX) {
+        return;
+    }
+
+    NAddr::TOpaqueAddr* oa = new NAddr::TOpaqueAddr(Addr_.Get());
+    Addr_.Reset(oa);
+    sockaddr* sa = oa->MutableAddr();
+
+    if (sa->sa_family == AF_INET) {
+        ((sockaddr_in*)sa)->sin_port = HostToInet(port);
+    } else {
+        ((sockaddr_in6*)sa)->sin6_port = HostToInet(port);
+    }
+}
+
+ui16 TEndpoint::Port() const noexcept {
+    if (Addr_->Addr()->sa_family == AF_UNIX) {
+        return 0;
+    }
+
+    const sockaddr* sa = Addr_->Addr();
+
+    if (sa->sa_family == AF_INET) {
+        return InetToHost(((const sockaddr_in*)sa)->sin_port);
+    } else {
+        return InetToHost(((const sockaddr_in6*)sa)->sin6_port);
+    }
+}
+
+size_t TEndpoint::Hash() const {
+    const sockaddr* sa = Addr_->Addr();
+
+    if (sa->sa_family == AF_INET) {
+        const sockaddr_in* sa4 = (const sockaddr_in*)sa;
+
+        return IntHash((((ui64)sa4->sin_addr.s_addr) << 16) ^ sa4->sin_port);
+    } else if (sa->sa_family == AF_INET6) {
+        const sockaddr_in6* sa6 = (const sockaddr_in6*)sa;
+        const ui64* ptr = (const ui64*)&sa6->sin6_addr;
+
+        return IntHash(ptr[0] ^ ptr[1] ^ sa6->sin6_port);
+    } else {
+        const sockaddr_un* un = (const sockaddr_un*)sa;
+        THash<TString> strHash;
+
+        return strHash(un->sun_path);
+    }
+}