1 //===-- SocketAddress.cpp ---------------------------------------*- C++ -*-===//
3 // The LLVM Compiler Infrastructure
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
8 //===----------------------------------------------------------------------===//
10 // Note: This file is used on Darwin by debugserver, so it needs to remain as
11 // self contained as possible, and devoid of references to LLVM unless
12 // there is compelling reason.
14 //===----------------------------------------------------------------------===//
17 #define _WINSOCK_DEPRECATED_NO_WARNINGS
20 #include "lldb/Host/SocketAddress.h"
26 #include <arpa/inet.h>
33 // Other libraries and framework includes
35 #include "lldb/Host/PosixApi.h"
37 // WindowsXP needs an inet_ntop implementation
40 #ifndef INET6_ADDRSTRLEN // might not be defined in older Windows SDKs
41 #define INET6_ADDRSTRLEN 46
44 // TODO: implement shortened form "::" for runs of zeros
45 const char *inet_ntop(int af, const void *src, char *dst, socklen_t size) {
53 const char *formatted = inet_ntoa(*static_cast<const in_addr *>(src));
54 if (formatted && strlen(formatted) < static_cast<size_t>(size)) {
55 return ::strcpy(dst, formatted);
60 char tmp[INET6_ADDRSTRLEN] = {0};
61 const uint16_t *src16 = static_cast<const uint16_t *>(src);
62 int full_size = ::snprintf(
63 tmp, sizeof(tmp), "%x:%x:%x:%x:%x:%x:%x:%x", ntohs(src16[0]),
64 ntohs(src16[1]), ntohs(src16[2]), ntohs(src16[3]), ntohs(src16[4]),
65 ntohs(src16[5]), ntohs(src16[6]), ntohs(src16[7]));
66 if (full_size < static_cast<int>(size)) {
67 return ::strcpy(dst, tmp);
77 using namespace lldb_private;
79 //----------------------------------------------------------------------
80 // SocketAddress constructor
81 //----------------------------------------------------------------------
82 SocketAddress::SocketAddress() { Clear(); }
84 SocketAddress::SocketAddress(const struct sockaddr &s) { m_socket_addr.sa = s; }
86 SocketAddress::SocketAddress(const struct sockaddr_in &s) {
87 m_socket_addr.sa_ipv4 = s;
90 SocketAddress::SocketAddress(const struct sockaddr_in6 &s) {
91 m_socket_addr.sa_ipv6 = s;
94 SocketAddress::SocketAddress(const struct sockaddr_storage &s) {
95 m_socket_addr.sa_storage = s;
98 SocketAddress::SocketAddress(const struct addrinfo *addr_info) {
102 //----------------------------------------------------------------------
103 // SocketAddress copy constructor
104 //----------------------------------------------------------------------
105 SocketAddress::SocketAddress(const SocketAddress &rhs)
106 : m_socket_addr(rhs.m_socket_addr) {}
108 //----------------------------------------------------------------------
110 //----------------------------------------------------------------------
111 SocketAddress::~SocketAddress() {}
113 void SocketAddress::Clear() {
114 memset(&m_socket_addr, 0, sizeof(m_socket_addr));
117 bool SocketAddress::IsValid() const { return GetLength() != 0; }
119 static socklen_t GetFamilyLength(sa_family_t family) {
122 return sizeof(struct sockaddr_in);
124 return sizeof(struct sockaddr_in6);
126 assert(0 && "Unsupported address family");
130 socklen_t SocketAddress::GetLength() const {
131 #if defined(__APPLE__) || defined(__FreeBSD__) || defined(__NetBSD__)
132 return m_socket_addr.sa.sa_len;
134 return GetFamilyLength(GetFamily());
138 socklen_t SocketAddress::GetMaxLength() { return sizeof(sockaddr_t); }
140 sa_family_t SocketAddress::GetFamily() const {
141 return m_socket_addr.sa.sa_family;
144 void SocketAddress::SetFamily(sa_family_t family) {
145 m_socket_addr.sa.sa_family = family;
146 #if defined(__APPLE__) || defined(__FreeBSD__) || defined(__NetBSD__)
147 m_socket_addr.sa.sa_len = GetFamilyLength(family);
151 std::string SocketAddress::GetIPAddress() const {
152 char str[INET6_ADDRSTRLEN] = {0};
153 switch (GetFamily()) {
155 if (inet_ntop(GetFamily(), &m_socket_addr.sa_ipv4.sin_addr, str,
160 if (inet_ntop(GetFamily(), &m_socket_addr.sa_ipv6.sin6_addr, str,
168 uint16_t SocketAddress::GetPort() const {
169 switch (GetFamily()) {
171 return ntohs(m_socket_addr.sa_ipv4.sin_port);
173 return ntohs(m_socket_addr.sa_ipv6.sin6_port);
178 bool SocketAddress::SetPort(uint16_t port) {
179 switch (GetFamily()) {
181 m_socket_addr.sa_ipv4.sin_port = htons(port);
185 m_socket_addr.sa_ipv6.sin6_port = htons(port);
191 //----------------------------------------------------------------------
192 // SocketAddress assignment operator
193 //----------------------------------------------------------------------
194 const SocketAddress &SocketAddress::operator=(const SocketAddress &rhs) {
196 m_socket_addr = rhs.m_socket_addr;
200 const SocketAddress &SocketAddress::
201 operator=(const struct addrinfo *addr_info) {
203 if (addr_info && addr_info->ai_addr && addr_info->ai_addrlen > 0 &&
204 size_t(addr_info->ai_addrlen) <= sizeof m_socket_addr) {
205 ::memcpy(&m_socket_addr, addr_info->ai_addr, addr_info->ai_addrlen);
210 const SocketAddress &SocketAddress::operator=(const struct sockaddr &s) {
211 m_socket_addr.sa = s;
215 const SocketAddress &SocketAddress::operator=(const struct sockaddr_in &s) {
216 m_socket_addr.sa_ipv4 = s;
220 const SocketAddress &SocketAddress::operator=(const struct sockaddr_in6 &s) {
221 m_socket_addr.sa_ipv6 = s;
225 const SocketAddress &SocketAddress::
226 operator=(const struct sockaddr_storage &s) {
227 m_socket_addr.sa_storage = s;
231 bool SocketAddress::getaddrinfo(const char *host, const char *service,
232 int ai_family, int ai_socktype, int ai_protocol,
236 auto addresses = GetAddressInfo(host, service, ai_family, ai_socktype,
237 ai_protocol, ai_flags);
238 if (!addresses.empty())
239 *this = addresses[0];
243 std::vector<SocketAddress>
244 SocketAddress::GetAddressInfo(const char *hostname, const char *servname,
245 int ai_family, int ai_socktype, int ai_protocol,
247 std::vector<SocketAddress> addr_list;
249 struct addrinfo hints;
250 memset(&hints, 0, sizeof(hints));
251 hints.ai_family = ai_family;
252 hints.ai_socktype = ai_socktype;
253 hints.ai_protocol = ai_protocol;
254 hints.ai_flags = ai_flags;
256 struct addrinfo *service_info_list = NULL;
257 int err = ::getaddrinfo(hostname, servname, &hints, &service_info_list);
258 if (err == 0 && service_info_list) {
259 for (struct addrinfo *service_ptr = service_info_list; service_ptr != NULL;
260 service_ptr = service_ptr->ai_next) {
261 addr_list.emplace_back(SocketAddress(service_ptr));
265 if (service_info_list)
266 ::freeaddrinfo(service_info_list);
270 bool SocketAddress::SetToLocalhost(sa_family_t family, uint16_t port) {
275 m_socket_addr.sa_ipv4.sin_addr.s_addr = htonl(INADDR_LOOPBACK);
283 m_socket_addr.sa_ipv6.sin6_addr = in6addr_loopback;
292 bool SocketAddress::SetToAnyAddress(sa_family_t family, uint16_t port) {
297 m_socket_addr.sa_ipv4.sin_addr.s_addr = htonl(INADDR_ANY);
305 m_socket_addr.sa_ipv6.sin6_addr = in6addr_any;
314 bool SocketAddress::IsAnyAddr() const {
315 return (GetFamily() == AF_INET)
316 ? m_socket_addr.sa_ipv4.sin_addr.s_addr == htonl(INADDR_ANY)
317 : 0 == memcmp(&m_socket_addr.sa_ipv6.sin6_addr, &in6addr_any, 16);
320 bool SocketAddress::IsLocalhost() const {
321 return (GetFamily() == AF_INET)
322 ? m_socket_addr.sa_ipv4.sin_addr.s_addr == htonl(INADDR_LOOPBACK)
323 : 0 == memcmp(&m_socket_addr.sa_ipv6.sin6_addr, &in6addr_loopback,
327 bool SocketAddress::operator==(const SocketAddress &rhs) const {
328 if (GetFamily() != rhs.GetFamily())
330 if (GetLength() != rhs.GetLength())
332 switch (GetFamily()) {
334 return m_socket_addr.sa_ipv4.sin_addr.s_addr ==
335 rhs.m_socket_addr.sa_ipv4.sin_addr.s_addr;
337 return 0 == memcmp(&m_socket_addr.sa_ipv6.sin6_addr,
338 &rhs.m_socket_addr.sa_ipv6.sin6_addr, 16);
343 bool SocketAddress::operator!=(const SocketAddress &rhs) const {
344 return !(*this == rhs);