1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
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()));
}
};
}
|
