intermediate changes

ref:cde9a383711a11544ce7e107a78147fb96cc4029

1 files changed, 86 insertions, 0 deletions

diff --git a/library/cpp/messagebus/config/netaddr.h b/library/cpp/messagebus/config/netaddr.h
new file mode 100644
index 00000000000..b79c0cc3551
--- /dev/null
+++ b/library/cpp/messagebus/config/netaddr.h
@@ -0,0 +1,86 @@
+#pragma once
+
+#include <util/digest/numeric.h>
+#include <util/generic/hash.h>
+#include <util/generic/ptr.h>
+#include <util/generic/strbuf.h>
+#include <util/generic/vector.h>
+#include <util/network/address.h>
+
+namespace NBus {
+    using namespace NAddr;
+
+    /// IP protocol version.
+    enum EIpVersion {
+        EIP_VERSION_4 = 1,
+        EIP_VERSION_6 = 2,
+        EIP_VERSION_ANY = EIP_VERSION_4 | EIP_VERSION_6,
+    };
+
+    inline bool IsFamilyAllowed(ui16 sa_family, EIpVersion ipVersion) {
+        if (ipVersion == EIP_VERSION_4 && sa_family != AF_INET) {
+            return false;
+        }
+        if (ipVersion == EIP_VERSION_6 && sa_family != AF_INET6) {
+            return false;
+        }
+        return true;
+    }
+
+    const char* ToCString(EIpVersion);
+    int ToAddrFamily(EIpVersion);
+
+    /// Hold referenced pointer to address description structure (ex. sockaddr_storage)
+    /// It's make possible to work with IPv4 / IPv6 addresses simultaneously
+    class TNetAddr: public IRemoteAddr {
+    public:
+        class TError: public yexception {
+        };
+
+        TNetAddr();
+        TNetAddr(TAutoPtr<IRemoteAddr> addr);
+        TNetAddr(const char* hostPort, EIpVersion requireVersion = EIP_VERSION_ANY, EIpVersion preferVersion = EIP_VERSION_ANY);
+        TNetAddr(TStringBuf host, int port, EIpVersion requireVersion = EIP_VERSION_ANY, EIpVersion preferVersion = EIP_VERSION_ANY);
+        TNetAddr(const TNetworkAddress& na, EIpVersion requireVersion = EIP_VERSION_ANY, EIpVersion preferVersion = EIP_VERSION_ANY);
+        TNetAddr(const TNetworkAddress& na, const TAddrInfo& ai);
+
+        bool operator==(const TNetAddr&) const;
+        bool operator!=(const TNetAddr& other) const {
+            return !(*this == other);
+        }
+        inline explicit operator bool() const noexcept {
+            return !!Ptr;
+        }
+
+        const sockaddr* Addr() const override;
+        socklen_t Len() const override;
+
+        bool IsIpv4() const;
+        bool IsIpv6() const;
+        int GetPort() const;
+
+    private:
+        TAtomicSharedPtr<IRemoteAddr> Ptr;
+    };
+
+    using TSockAddrInVector = TVector<TNetAddr>;
+
+    struct TNetAddrHostPortHash {
+        inline size_t operator()(const TNetAddr& a) const {
+            const sockaddr* s = a.Addr();
+            const sockaddr_in* const sa = reinterpret_cast<const sockaddr_in*>(s);
+            const sockaddr_in6* const sa6 = reinterpret_cast<const sockaddr_in6*>(s);
+
+            switch (s->sa_family) {
+                case AF_INET:
+                    return CombineHashes<size_t>(ComputeHash(TStringBuf(reinterpret_cast<const char*>(&sa->sin_addr), sizeof(sa->sin_addr))), IntHashImpl(sa->sin_port));
+
+                case AF_INET6:
+                    return CombineHashes<size_t>(ComputeHash(TStringBuf(reinterpret_cast<const char*>(&sa6->sin6_addr), sizeof(sa6->sin6_addr))), IntHashImpl(sa6->sin6_port));
+            }
+
+            return ComputeHash(TStringBuf(reinterpret_cast<const char*>(s), a.Len()));
+        }
+    };
+
+}