Skip to content

Commit 4ac1ce1

Browse files
jasnellMylesBorins
jasnell
authored and
MylesBorins
committed
src: introduce node_sockaddr
Introduce the SocketAddress utility class. The QUIC implementation makes extensive use of this for handling of socket addresses. It was separated out to make it generically reusable throughout core Signed-off-by: James M Snell <[email protected]> PR-URL: #32070 Reviewed-By: David Carlier <[email protected]> Reviewed-By: Richard Lau <[email protected]> Reviewed-By: Anna Henningsen <[email protected]> Reviewed-By: Denys Otrishko <[email protected]>
1 parent a727b13 commit 4ac1ce1

File tree

7 files changed

+460
-11
lines changed

7 files changed

+460
-11
lines changed

‎node.gyp‎

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -597,6 +597,7 @@
597597
'src/node_process_methods.cc',
598598
'src/node_process_object.cc',
599599
'src/node_serdes.cc',
600+
'src/node_sockaddr.cc',
600601
'src/node_stat_watcher.cc',
601602
'src/node_symbols.cc',
602603
'src/node_task_queue.cc',
@@ -684,6 +685,8 @@
684685
'src/node_process.h',
685686
'src/node_revert.h',
686687
'src/node_root_certs.h',
688+
'src/node_sockaddr.h',
689+
'src/node_sockaddr-inl.h',
687690
'src/node_stat_watcher.h',
688691
'src/node_union_bytes.h',
689692
'src/node_url.h',
@@ -1150,6 +1153,7 @@
11501153
'test/cctest/test_linked_binding.cc',
11511154
'test/cctest/test_per_process.cc',
11521155
'test/cctest/test_platform.cc',
1156+
'test/cctest/test_sockaddr.cc',
11531157
'test/cctest/test_traced_value.cc',
11541158
'test/cctest/test_util.cc',
11551159
'test/cctest/test_url.cc',

‎src/node_sockaddr-inl.h‎

Lines changed: 170 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,170 @@
1+
#ifndef SRC_NODE_SOCKADDR_INL_H_
2+
#defineSRC_NODE_SOCKADDR_INL_H_
3+
4+
#if defined(NODE_WANT_INTERNALS) && NODE_WANT_INTERNALS
5+
6+
#include"node.h"
7+
#include"node_internals.h"
8+
#include"node_sockaddr.h"
9+
#include"util-inl.h"
10+
11+
#include<string>
12+
13+
namespacenode{
14+
15+
staticconstexpruint32_tkLabelMask = 0xFFFFF;
16+
17+
inlinevoidhash_combine(size_t* seed){}
18+
19+
template <typename T, typename... Args>
20+
inlinevoidhash_combine(size_t* seed, const T& value, Args... rest){
21+
*seed ^= std::hash<T>{}(value) + 0x9e3779b9 + (*seed << 6) + (*seed >> 2);
22+
hash_combine(seed, rest...);
23+
}
24+
25+
boolSocketAddress::is_numeric_host(constchar* hostname){
26+
returnis_numeric_host(hostname, AF_INET) ||
27+
is_numeric_host(hostname, AF_INET6);
28+
}
29+
30+
boolSocketAddress::is_numeric_host(constchar* hostname, int family){
31+
in6_addr dst;
32+
returninet_pton(family, hostname, &dst) == 1;
33+
}
34+
35+
intSocketAddress::GetPort(const sockaddr* addr){
36+
CHECK(addr->sa_family == AF_INET || addr->sa_family == AF_INET6);
37+
returnntohs(addr->sa_family == AF_INET ?
38+
reinterpret_cast<const sockaddr_in*>(addr)->sin_port :
39+
reinterpret_cast<const sockaddr_in6*>(addr)->sin6_port);
40+
}
41+
42+
intSocketAddress::GetPort(const sockaddr_storage* addr){
43+
returnGetPort(reinterpret_cast<const sockaddr*>(addr));
44+
}
45+
46+
std::string SocketAddress::GetAddress(const sockaddr* addr){
47+
CHECK(addr->sa_family == AF_INET || addr->sa_family == AF_INET6);
48+
char host[INET6_ADDRSTRLEN];
49+
constvoid* src = addr->sa_family == AF_INET ?
50+
static_cast<constvoid*>(
51+
&(reinterpret_cast<const sockaddr_in*>(addr)->sin_addr)) :
52+
static_cast<constvoid*>(
53+
&(reinterpret_cast<const sockaddr_in6*>(addr)->sin6_addr));
54+
uv_inet_ntop(addr->sa_family, src, host, INET6_ADDRSTRLEN);
55+
returnstd::string(host);
56+
}
57+
58+
std::string SocketAddress::GetAddress(const sockaddr_storage* addr){
59+
returnGetAddress(reinterpret_cast<const sockaddr*>(addr));
60+
}
61+
62+
size_tSocketAddress::GetLength(const sockaddr* addr){
63+
return addr->sa_family == AF_INET ?
64+
sizeof(sockaddr_in) : sizeof(sockaddr_in6);
65+
}
66+
67+
size_tSocketAddress::GetLength(const sockaddr_storage* addr){
68+
returnGetLength(reinterpret_cast<const sockaddr*>(addr));
69+
}
70+
71+
SocketAddress::SocketAddress(const sockaddr* addr){
72+
memcpy(&address_, addr, GetLength(addr));
73+
}
74+
75+
SocketAddress::SocketAddress(const SocketAddress& addr){
76+
memcpy(&address_, &addr.address_, addr.length());
77+
}
78+
79+
SocketAddress& SocketAddress::operator=(const sockaddr* addr){
80+
memcpy(&address_, addr, GetLength(addr));
81+
return *this;
82+
}
83+
84+
SocketAddress& SocketAddress::operator=(const SocketAddress& addr){
85+
memcpy(&address_, &addr.address_, addr.length());
86+
return *this;
87+
}
88+
89+
const sockaddr& SocketAddress::operator*() const{
90+
return *this->data();
91+
}
92+
93+
const sockaddr* SocketAddress::operator->() const{
94+
returnthis->data();
95+
}
96+
97+
size_tSocketAddress::length() const{
98+
returnGetLength(&address_);
99+
}
100+
101+
const sockaddr* SocketAddress::data() const{
102+
returnreinterpret_cast<const sockaddr*>(&address_);
103+
}
104+
105+
constuint8_t* SocketAddress::raw() const{
106+
returnreinterpret_cast<constuint8_t*>(&address_);
107+
}
108+
109+
sockaddr* SocketAddress::storage(){
110+
returnreinterpret_cast<sockaddr*>(&address_);
111+
}
112+
113+
intSocketAddress::family() const{
114+
return address_.ss_family;
115+
}
116+
117+
std::string SocketAddress::address() const{
118+
returnGetAddress(&address_);
119+
}
120+
121+
intSocketAddress::port() const{
122+
returnGetPort(&address_);
123+
}
124+
125+
uint32_tSocketAddress::flow_label() const{
126+
if (family() != AF_INET6)
127+
return0;
128+
const sockaddr_in6* in = reinterpret_cast<const sockaddr_in6*>(data());
129+
return in->sin6_flowinfo;
130+
}
131+
132+
voidSocketAddress::set_flow_label(uint32_t label){
133+
if (family() != AF_INET6)
134+
return;
135+
CHECK_LE(label, kLabelMask);
136+
sockaddr_in6* in = reinterpret_cast<sockaddr_in6*>(&address_);
137+
in->sin6_flowinfo = label;
138+
}
139+
140+
std::string SocketAddress::ToString() const{
141+
if (family() != AF_INET && family() != AF_INET6) return"";
142+
return (family() == AF_INET6 ?
143+
std::string("[") + address() + "]:" :
144+
address() + ":") +
145+
std::to_string(port());
146+
}
147+
148+
voidSocketAddress::Update(uint8_t* data, size_t len){
149+
CHECK_LE(len, sizeof(address_));
150+
memcpy(&address_, data, len);
151+
}
152+
153+
v8::Local<v8::Object> SocketAddress::ToJS(
154+
Environment* env,
155+
v8::Local<v8::Object> info) const{
156+
returnAddressToJS(env, data(), info);
157+
}
158+
159+
bool SocketAddress::operator==(const SocketAddress& other) const{
160+
if (family() != other.family()) returnfalse;
161+
returnmemcmp(raw(), other.raw(), length()) == 0;
162+
}
163+
164+
bool SocketAddress::operator!=(const SocketAddress& other) const{
165+
return !(*this == other);
166+
}
167+
} // namespace node
168+
169+
#endif// NODE_WANT_INTERNALS
170+
#endif// SRC_NODE_SOCKADDR_INL_H_

‎src/node_sockaddr.cc‎

Lines changed: 95 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,95 @@
1+
#include"node_sockaddr-inl.h"// NOLINT(build/include)
2+
#include"uv.h"
3+
4+
namespacenode{
5+
6+
namespace{
7+
template <typename T, typename F>
8+
SocketAddress FromUVHandle(F fn, const T& handle){
9+
SocketAddress addr;
10+
int len = sizeof(sockaddr_storage);
11+
if (fn(&handle, addr.storage(), &len) == 0)
12+
CHECK_EQ(static_cast<size_t>(len), addr.length());
13+
else
14+
addr.storage()->sa_family = 0;
15+
return addr;
16+
}
17+
} // namespace
18+
19+
boolSocketAddress::ToSockAddr(
20+
int32_t family,
21+
constchar* host,
22+
uint32_t port,
23+
sockaddr_storage* addr){
24+
switch (family){
25+
case AF_INET:
26+
returnuv_ip4_addr(
27+
host,
28+
port,
29+
reinterpret_cast<sockaddr_in*>(addr)) == 0;
30+
case AF_INET6:
31+
returnuv_ip6_addr(
32+
host,
33+
port,
34+
reinterpret_cast<sockaddr_in6*>(addr)) == 0;
35+
default:
36+
UNREACHABLE();
37+
}
38+
}
39+
40+
boolSocketAddress::New(
41+
constchar* host,
42+
uint32_t port,
43+
SocketAddress* addr){
44+
returnNew(AF_INET, host, port, addr) || New(AF_INET6, host, port, addr);
45+
}
46+
47+
boolSocketAddress::New(
48+
int32_t family,
49+
constchar* host,
50+
uint32_t port,
51+
SocketAddress* addr){
52+
returnToSockAddr(family, host, port,
53+
reinterpret_cast<sockaddr_storage*>(addr->storage()));
54+
}
55+
56+
size_tSocketAddress::Hash::operator()(const SocketAddress& addr) const{
57+
size_t hash = 0;
58+
switch (addr.family()){
59+
case AF_INET:{
60+
const sockaddr_in* ipv4 =
61+
reinterpret_cast<const sockaddr_in*>(addr.raw());
62+
hash_combine(&hash, ipv4->sin_port, ipv4->sin_addr.s_addr);
63+
break;
64+
}
65+
case AF_INET6:{
66+
const sockaddr_in6* ipv6 =
67+
reinterpret_cast<const sockaddr_in6*>(addr.raw());
68+
constuint64_t* a =
69+
reinterpret_cast<constuint64_t*>(&ipv6->sin6_addr);
70+
hash_combine(&hash, ipv6->sin6_port, a[0], a[1]);
71+
break;
72+
}
73+
default:
74+
UNREACHABLE();
75+
}
76+
return hash;
77+
}
78+
79+
SocketAddress SocketAddress::FromSockName(constuv_tcp_t& handle){
80+
returnFromUVHandle(uv_tcp_getsockname, handle);
81+
}
82+
83+
SocketAddress SocketAddress::FromSockName(constuv_udp_t& handle){
84+
returnFromUVHandle(uv_udp_getsockname, handle);
85+
}
86+
87+
SocketAddress SocketAddress::FromPeerName(constuv_tcp_t& handle){
88+
returnFromUVHandle(uv_tcp_getpeername, handle);
89+
}
90+
91+
SocketAddress SocketAddress::FromPeerName(constuv_udp_t& handle){
92+
returnFromUVHandle(uv_udp_getpeername, handle);
93+
}
94+
95+
} // namespace node

0 commit comments

Comments
(0)