1 //===-- SocketAddress.cpp ---------------------------------------*- C++ -*-===//
3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4 // See https://llvm.org/LICENSE.txt for license information.
5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
7 //===----------------------------------------------------------------------===//
9 // Note: This file is used on Darwin by debugserver, so it needs to remain as
10 // self contained as possible, and devoid of references to LLVM unless
11 // there is compelling reason.
13 //===----------------------------------------------------------------------===//
16 #define _WINSOCK_DEPRECATED_NO_WARNINGS
19 #include "lldb/Host/SocketAddress.h"
24 #include <arpa/inet.h>
30 #include "lldb/Host/PosixApi.h"
32 // WindowsXP needs an inet_ntop implementation
35 #ifndef INET6_ADDRSTRLEN // might not be defined in older Windows SDKs
36 #define INET6_ADDRSTRLEN 46
39 // TODO: implement shortened form "::" for runs of zeros
40 const char *inet_ntop(int af, const void *src, char *dst, socklen_t size) {
48 const char *formatted = inet_ntoa(*static_cast<const in_addr *>(src));
49 if (formatted && strlen(formatted) < static_cast<size_t>(size)) {
50 return ::strcpy(dst, formatted);
55 char tmp[INET6_ADDRSTRLEN] = {0};
56 const uint16_t *src16 = static_cast<const uint16_t *>(src);
57 int full_size = ::snprintf(
58 tmp, sizeof(tmp), "%x:%x:%x:%x:%x:%x:%x:%x", ntohs(src16[0]),
59 ntohs(src16[1]), ntohs(src16[2]), ntohs(src16[3]), ntohs(src16[4]),
60 ntohs(src16[5]), ntohs(src16[6]), ntohs(src16[7]));
61 if (full_size < static_cast<int>(size)) {
62 return ::strcpy(dst, tmp);
72 using namespace lldb_private;
74 // SocketAddress constructor
75 SocketAddress::SocketAddress() { Clear(); }
77 SocketAddress::SocketAddress(const struct sockaddr &s) { m_socket_addr.sa = s; }
79 SocketAddress::SocketAddress(const struct sockaddr_in &s) {
80 m_socket_addr.sa_ipv4 = s;
83 SocketAddress::SocketAddress(const struct sockaddr_in6 &s) {
84 m_socket_addr.sa_ipv6 = s;
87 SocketAddress::SocketAddress(const struct sockaddr_storage &s) {
88 m_socket_addr.sa_storage = s;
91 SocketAddress::SocketAddress(const struct addrinfo *addr_info) {
96 SocketAddress::~SocketAddress() {}
98 void SocketAddress::Clear() {
99 memset(&m_socket_addr, 0, sizeof(m_socket_addr));
102 bool SocketAddress::IsValid() const { return GetLength() != 0; }
104 static socklen_t GetFamilyLength(sa_family_t family) {
107 return sizeof(struct sockaddr_in);
109 return sizeof(struct sockaddr_in6);
111 assert(0 && "Unsupported address family");
115 socklen_t SocketAddress::GetLength() const {
116 #if defined(__APPLE__) || defined(__FreeBSD__) || defined(__NetBSD__)
117 return m_socket_addr.sa.sa_len;
119 return GetFamilyLength(GetFamily());
123 socklen_t SocketAddress::GetMaxLength() { return sizeof(sockaddr_t); }
125 sa_family_t SocketAddress::GetFamily() const {
126 return m_socket_addr.sa.sa_family;
129 void SocketAddress::SetFamily(sa_family_t family) {
130 m_socket_addr.sa.sa_family = family;
131 #if defined(__APPLE__) || defined(__FreeBSD__) || defined(__NetBSD__)
132 m_socket_addr.sa.sa_len = GetFamilyLength(family);
136 std::string SocketAddress::GetIPAddress() const {
137 char str[INET6_ADDRSTRLEN] = {0};
138 switch (GetFamily()) {
140 if (inet_ntop(GetFamily(), &m_socket_addr.sa_ipv4.sin_addr, str,
145 if (inet_ntop(GetFamily(), &m_socket_addr.sa_ipv6.sin6_addr, str,
153 uint16_t SocketAddress::GetPort() const {
154 switch (GetFamily()) {
156 return ntohs(m_socket_addr.sa_ipv4.sin_port);
158 return ntohs(m_socket_addr.sa_ipv6.sin6_port);
163 bool SocketAddress::SetPort(uint16_t port) {
164 switch (GetFamily()) {
166 m_socket_addr.sa_ipv4.sin_port = htons(port);
170 m_socket_addr.sa_ipv6.sin6_port = htons(port);
176 // SocketAddress assignment operator
177 const SocketAddress &SocketAddress::operator=(const SocketAddress &rhs) {
179 m_socket_addr = rhs.m_socket_addr;
183 const SocketAddress &SocketAddress::
184 operator=(const struct addrinfo *addr_info) {
186 if (addr_info && addr_info->ai_addr && addr_info->ai_addrlen > 0 &&
187 size_t(addr_info->ai_addrlen) <= sizeof m_socket_addr) {
188 ::memcpy(&m_socket_addr, addr_info->ai_addr, addr_info->ai_addrlen);
193 const SocketAddress &SocketAddress::operator=(const struct sockaddr &s) {
194 m_socket_addr.sa = s;
198 const SocketAddress &SocketAddress::operator=(const struct sockaddr_in &s) {
199 m_socket_addr.sa_ipv4 = s;
203 const SocketAddress &SocketAddress::operator=(const struct sockaddr_in6 &s) {
204 m_socket_addr.sa_ipv6 = s;
208 const SocketAddress &SocketAddress::
209 operator=(const struct sockaddr_storage &s) {
210 m_socket_addr.sa_storage = s;
214 bool SocketAddress::getaddrinfo(const char *host, const char *service,
215 int ai_family, int ai_socktype, int ai_protocol,
219 auto addresses = GetAddressInfo(host, service, ai_family, ai_socktype,
220 ai_protocol, ai_flags);
221 if (!addresses.empty())
222 *this = addresses[0];
226 std::vector<SocketAddress>
227 SocketAddress::GetAddressInfo(const char *hostname, const char *servname,
228 int ai_family, int ai_socktype, int ai_protocol,
230 std::vector<SocketAddress> addr_list;
232 struct addrinfo hints;
233 memset(&hints, 0, sizeof(hints));
234 hints.ai_family = ai_family;
235 hints.ai_socktype = ai_socktype;
236 hints.ai_protocol = ai_protocol;
237 hints.ai_flags = ai_flags;
239 struct addrinfo *service_info_list = nullptr;
240 int err = ::getaddrinfo(hostname, servname, &hints, &service_info_list);
241 if (err == 0 && service_info_list) {
242 for (struct addrinfo *service_ptr = service_info_list;
243 service_ptr != nullptr; service_ptr = service_ptr->ai_next) {
244 addr_list.emplace_back(SocketAddress(service_ptr));
248 if (service_info_list)
249 ::freeaddrinfo(service_info_list);
253 bool SocketAddress::SetToLocalhost(sa_family_t family, uint16_t port) {
258 m_socket_addr.sa_ipv4.sin_addr.s_addr = htonl(INADDR_LOOPBACK);
266 m_socket_addr.sa_ipv6.sin6_addr = in6addr_loopback;
275 bool SocketAddress::SetToAnyAddress(sa_family_t family, uint16_t port) {
280 m_socket_addr.sa_ipv4.sin_addr.s_addr = htonl(INADDR_ANY);
288 m_socket_addr.sa_ipv6.sin6_addr = in6addr_any;
297 bool SocketAddress::IsAnyAddr() const {
298 return (GetFamily() == AF_INET)
299 ? m_socket_addr.sa_ipv4.sin_addr.s_addr == htonl(INADDR_ANY)
300 : 0 == memcmp(&m_socket_addr.sa_ipv6.sin6_addr, &in6addr_any, 16);
303 bool SocketAddress::IsLocalhost() const {
304 return (GetFamily() == AF_INET)
305 ? m_socket_addr.sa_ipv4.sin_addr.s_addr == htonl(INADDR_LOOPBACK)
306 : 0 == memcmp(&m_socket_addr.sa_ipv6.sin6_addr, &in6addr_loopback,
310 bool SocketAddress::operator==(const SocketAddress &rhs) const {
311 if (GetFamily() != rhs.GetFamily())
313 if (GetLength() != rhs.GetLength())
315 switch (GetFamily()) {
317 return m_socket_addr.sa_ipv4.sin_addr.s_addr ==
318 rhs.m_socket_addr.sa_ipv4.sin_addr.s_addr;
320 return 0 == memcmp(&m_socket_addr.sa_ipv6.sin6_addr,
321 &rhs.m_socket_addr.sa_ipv6.sin6_addr, 16);
326 bool SocketAddress::operator!=(const SocketAddress &rhs) const {
327 return !(*this == rhs);