intermediate changes

ref:cde9a383711a11544ce7e107a78147fb96cc4029

1 files changed, 131 insertions, 0 deletions

diff --git a/util/draft/ip.h b/util/draft/ip.h
new file mode 100644
index 0000000000..eb947cd2cd
--- /dev/null
+++ b/util/draft/ip.h
@@ -0,0 +1,131 @@
+#pragma once
+
+#include <util/digest/murmur.h>
+
+#include <util/network/ip.h>
+
+#include <util/str_stl.h>
+#include <util/generic/maybe.h>
+#include <util/generic/variant.h>
+
+#ifdef _unix_
+    #include <sys/types.h>
+    #include <sys/socket.h>
+    #include <arpa/inet.h>
+#endif // _unix_
+
+#include <string.h>
+
+#ifndef INET6_ADDRSTRLEN
+    #define INET6_ADDRSTRLEN 46
+#endif
+
+// Network (big-endian) byte order
+using TIp4 = TIpHost;
+
+// Network (big-endian) byte order
+struct TIp6 {
+    char Data[16];
+
+    bool operator==(const TIp6& rhs) const {
+        return memcmp(Data, rhs.Data, sizeof(Data)) == 0;
+    }
+
+    bool operator<(const TIp6& rhs) const {
+        return memcmp(Data, rhs.Data, sizeof(Data)) < 0;
+    }
+};
+
+template <>
+struct THash<TIp6> {
+    inline size_t operator()(const TIp6& ip) const {
+        return MurmurHash<size_t>((const void*)ip.Data, 16);
+    }
+};
+
+static inline TIp6 Ip6FromIp4(TIp4 addr) {
+    TIp6 res;
+    memset(res.Data, 0, sizeof(res.Data));
+    res.Data[10] = '\xFF';
+    res.Data[11] = '\xFF';
+    memcpy(res.Data + 12, &addr, 4);
+    return res;
+}
+
+static inline TIp6 Ip6FromString(const char* ipStr) {
+    TIp6 res;
+
+    if (inet_pton(AF_INET6, ipStr, &res.Data) == 0) {
+        ythrow TSystemError() << "Failed to convert (" << ipStr << ") to ipv6 address";
+    }
+
+    return res;
+}
+
+static inline TMaybe<TIp6> TryParseIp6FromString(const char* ipStr) {
+    TIp6 res;
+
+    if (inet_pton(AF_INET6, ipStr, &res.Data) == 0) {
+        return Nothing();
+    }
+
+    return res;
+}
+
+static inline char* Ip6ToString(const TIp6& ip, char* buf, size_t len) {
+    if (!inet_ntop(AF_INET6, (void*)&ip.Data, buf, (socklen_t)len)) {
+        ythrow TSystemError() << "Failed to get ipv6 address string";
+    }
+
+    return buf;
+}
+
+static inline TString Ip6ToString(const TIp6& ip) {
+    char buf[INET6_ADDRSTRLEN];
+
+    return TString(Ip6ToString(ip, buf, sizeof(buf)));
+}
+
+template <>
+inline void Out<TIp6>(IOutputStream& os, const TIp6& a) {
+    os << Ip6ToString(a);
+}
+
+using TIp4Or6 = std::variant<TIp4, TIp6>;
+
+static inline TIp4Or6 Ip4Or6FromString(const char* ipStr) {
+    const char* c = ipStr;
+    for (; *c; ++c) {
+        if (*c == '.') {
+            return IpFromString(ipStr);
+        }
+        if (*c == ':') {
+            return Ip6FromString(ipStr);
+        }
+    }
+    ythrow TSystemError() << "Failed to convert (" << ipStr << ") to ipv4 or ipv6 address";
+}
+
+static inline TString Ip4Or6ToString(const TIp4Or6& ip) {
+    if (std::holds_alternative<TIp6>(ip)) {
+        return Ip6ToString(std::get<TIp6>(ip));
+    } else {
+        return IpToString(std::get<TIp4>(ip));
+    }
+}
+
+// for TIp4 or TIp6, not TIp4Or6
+template <class TIp>
+struct TIpCompare {
+    bool Less(const TIp& l, const TIp& r) const {
+        return memcmp(&l, &r, sizeof(TIp)) < 0;
+    }
+
+    bool LessEqual(const TIp& l, const TIp& r) const {
+        return memcmp(&l, &r, sizeof(TIp)) <= 0;
+    }
+
+    bool operator()(const TIp& l, const TIp& r) const {
+        return Less(l, r);
+    }
+};