intermediate changes

ref:cde9a383711a11544ce7e107a78147fb96cc4029

1 files changed, 113 insertions, 0 deletions

diff --git a/library/cpp/messagebus/socket_addr.h b/library/cpp/messagebus/socket_addr.h
new file mode 100644
index 0000000000..959eafe689
--- /dev/null
+++ b/library/cpp/messagebus/socket_addr.h
@@ -0,0 +1,113 @@
+#pragma once
+
+#include "hash.h"
+
+#include <util/generic/hash.h>
+#include <util/generic/utility.h>
+#include <util/network/address.h>
+#include <util/network/init.h>
+
+#include <string.h>
+
+namespace NBus {
+    class TNetAddr;
+}
+
+namespace NBus {
+    namespace NPrivate {
+        enum EAddrFamily {
+            ADDR_UNSPEC = AF_UNSPEC,
+            ADDR_IPV4 = AF_INET,
+            ADDR_IPV6 = AF_INET6,
+        };
+
+        class TBusIpAddr {
+        private:
+            EAddrFamily Af;
+
+            union {
+                in_addr In4;
+                in6_addr In6;
+            };
+
+        public:
+            TBusIpAddr() {
+                Clear();
+            }
+
+            EAddrFamily GetAddrFamily() const {
+                return Af;
+            }
+
+            void Clear() {
+                Zero(*this);
+            }
+
+            in_addr GetInAddr() const {
+                Y_ASSERT(Af == ADDR_IPV4);
+                return In4;
+            }
+
+            void SetInAddr(const in_addr& in4) {
+                Clear();
+                Af = ADDR_IPV4;
+                In4 = in4;
+            }
+
+            in6_addr GetIn6Addr() const {
+                Y_ASSERT(Af == ADDR_IPV6);
+                return In6;
+            }
+
+            void SetIn6Addr(const in6_addr& in6) {
+                Clear();
+                Af = ADDR_IPV6;
+                In6 = in6;
+            }
+
+            bool operator==(const TBusIpAddr& that) const {
+                return memcmp(this, &that, sizeof(that)) == 0;
+            }
+        };
+
+        class TBusSocketAddr {
+        public:
+            TBusIpAddr IpAddr;
+            ui16 Port;
+
+            //Only makes sense for IPv6 link-local addresses
+            ui32 IPv6ScopeID;
+
+            TBusSocketAddr()
+                : Port(0)
+                , IPv6ScopeID(0)
+            {
+            }
+
+            TBusSocketAddr(const NAddr::IRemoteAddr*);
+            TBusSocketAddr(const TNetAddr&);
+            TBusSocketAddr(TStringBuf host, unsigned port);
+
+            TNetAddr ToNetAddr() const;
+
+            bool operator==(const TBusSocketAddr& that) const {
+                return IpAddr == that.IpAddr && Port == that.Port;
+            }
+        };
+
+    }
+}
+
+template <>
+struct THash<NBus::NPrivate::TBusIpAddr> {
+    inline size_t operator()(const NBus::NPrivate::TBusIpAddr& a) const {
+        return ComputeHash(TStringBuf((const char*)&a, sizeof(a)));
+    }
+};
+
+template <>
+struct THash<NBus::NPrivate::TBusSocketAddr> {
+    inline size_t operator()(const NBus::NPrivate::TBusSocketAddr& a) const {
+        return HashValues(a.IpAddr, a.Port);
+    }
+};