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 //===----------------------------------------------------------------------===//
11 #define _WINSOCK_DEPRECATED_NO_WARNINGS
14 #include "lldb/Host/SocketAddress.h"
20 #include <arpa/inet.h>
27 // Other libraries and framework includes
29 #include "lldb/Host/PosixApi.h"
31 // WindowsXP needs an inet_ntop implementation
34 #ifndef INET6_ADDRSTRLEN // might not be defined in older Windows SDKs
35 #define INET6_ADDRSTRLEN 46
38 // TODO: implement shortened form "::" for runs of zeros
39 const char *inet_ntop(int af, const void *src, char *dst, socklen_t size) {
47 const char *formatted = inet_ntoa(*static_cast<const in_addr *>(src));
48 if (formatted && strlen(formatted) < static_cast<size_t>(size)) {
49 return ::strcpy(dst, formatted);
54 char tmp[INET6_ADDRSTRLEN] = {0};
55 const uint16_t *src16 = static_cast<const uint16_t *>(src);
56 int full_size = ::snprintf(
57 tmp, sizeof(tmp), "%x:%x:%x:%x:%x:%x:%x:%x", ntohs(src16[0]),
58 ntohs(src16[1]), ntohs(src16[2]), ntohs(src16[3]), ntohs(src16[4]),
59 ntohs(src16[5]), ntohs(src16[6]), ntohs(src16[7]));
60 if (full_size < static_cast<int>(size)) {
61 return ::strcpy(dst, tmp);
71 using namespace lldb_private;
73 //----------------------------------------------------------------------
74 // SocketAddress constructor
75 //----------------------------------------------------------------------
76 SocketAddress::SocketAddress() { Clear(); }
78 SocketAddress::SocketAddress(const struct sockaddr &s) { m_socket_addr.sa = s; }
80 SocketAddress::SocketAddress(const struct sockaddr_in &s) {
81 m_socket_addr.sa_ipv4 = s;
84 SocketAddress::SocketAddress(const struct sockaddr_in6 &s) {
85 m_socket_addr.sa_ipv6 = s;
88 SocketAddress::SocketAddress(const struct sockaddr_storage &s) {
89 m_socket_addr.sa_storage = s;
92 SocketAddress::SocketAddress(const struct addrinfo *addr_info) {
96 //----------------------------------------------------------------------
97 // SocketAddress copy constructor
98 //----------------------------------------------------------------------
99 SocketAddress::SocketAddress(const SocketAddress &rhs)
100 : m_socket_addr(rhs.m_socket_addr) {}
102 //----------------------------------------------------------------------
104 //----------------------------------------------------------------------
105 SocketAddress::~SocketAddress() {}
107 void SocketAddress::Clear() {
108 memset(&m_socket_addr, 0, sizeof(m_socket_addr));
111 bool SocketAddress::IsValid() const { return GetLength() != 0; }
113 static socklen_t GetFamilyLength(sa_family_t family) {
116 return sizeof(struct sockaddr_in);
118 return sizeof(struct sockaddr_in6);
120 assert(0 && "Unsupported address family");
124 socklen_t SocketAddress::GetLength() const {
125 #if defined(__APPLE__) || defined(__FreeBSD__) || defined(__NetBSD__)
126 return m_socket_addr.sa.sa_len;
128 return GetFamilyLength(GetFamily());
132 socklen_t SocketAddress::GetMaxLength() { return sizeof(sockaddr_t); }
134 sa_family_t SocketAddress::GetFamily() const {
135 return m_socket_addr.sa.sa_family;
138 void SocketAddress::SetFamily(sa_family_t family) {
139 m_socket_addr.sa.sa_family = family;
140 #if defined(__APPLE__) || defined(__FreeBSD__) || defined(__NetBSD__)
141 m_socket_addr.sa.sa_len = GetFamilyLength(family);
145 std::string SocketAddress::GetIPAddress() const {
146 char str[INET6_ADDRSTRLEN] = {0};
147 switch (GetFamily()) {
149 if (inet_ntop(GetFamily(), &m_socket_addr.sa_ipv4.sin_addr, str,
154 if (inet_ntop(GetFamily(), &m_socket_addr.sa_ipv6.sin6_addr, str,
162 uint16_t SocketAddress::GetPort() const {
163 switch (GetFamily()) {
165 return ntohs(m_socket_addr.sa_ipv4.sin_port);
167 return ntohs(m_socket_addr.sa_ipv6.sin6_port);
172 bool SocketAddress::SetPort(uint16_t port) {
173 switch (GetFamily()) {
175 m_socket_addr.sa_ipv4.sin_port = htons(port);
179 m_socket_addr.sa_ipv6.sin6_port = htons(port);
185 //----------------------------------------------------------------------
186 // SocketAddress assignment operator
187 //----------------------------------------------------------------------
188 const SocketAddress &SocketAddress::operator=(const SocketAddress &rhs) {
190 m_socket_addr = rhs.m_socket_addr;
194 const SocketAddress &SocketAddress::
195 operator=(const struct addrinfo *addr_info) {
197 if (addr_info && addr_info->ai_addr && addr_info->ai_addrlen > 0 &&
198 addr_info->ai_addrlen <= sizeof m_socket_addr) {
199 ::memcpy(&m_socket_addr, addr_info->ai_addr, addr_info->ai_addrlen);
204 const SocketAddress &SocketAddress::operator=(const struct sockaddr &s) {
205 m_socket_addr.sa = s;
209 const SocketAddress &SocketAddress::operator=(const struct sockaddr_in &s) {
210 m_socket_addr.sa_ipv4 = s;
214 const SocketAddress &SocketAddress::operator=(const struct sockaddr_in6 &s) {
215 m_socket_addr.sa_ipv6 = s;
219 const SocketAddress &SocketAddress::
220 operator=(const struct sockaddr_storage &s) {
221 m_socket_addr.sa_storage = s;
225 bool SocketAddress::getaddrinfo(const char *host, const char *service,
226 int ai_family, int ai_socktype, int ai_protocol,
230 struct addrinfo hints;
231 memset(&hints, 0, sizeof(hints));
232 hints.ai_family = ai_family;
233 hints.ai_socktype = ai_socktype;
234 hints.ai_protocol = ai_protocol;
235 hints.ai_flags = ai_flags;
238 struct addrinfo *service_info_list = NULL;
239 int err = ::getaddrinfo(host, service, &hints, &service_info_list);
240 if (err == 0 && service_info_list) {
241 *this = service_info_list;
245 if (service_info_list)
246 ::freeaddrinfo(service_info_list);
251 std::vector<SocketAddress> SocketAddress::GetAddressInfo(const char *hostname,
252 const char *servname) {
253 std::vector<SocketAddress> addr_list;
255 struct addrinfo *service_info_list = NULL;
256 int err = ::getaddrinfo(hostname, servname, NULL, &service_info_list);
257 if (err == 0 && service_info_list) {
258 for (struct addrinfo *service_ptr = service_info_list; service_ptr != NULL;
259 service_ptr = service_ptr->ai_next) {
260 addr_list.emplace_back(SocketAddress(service_ptr));
264 if (service_info_list)
265 ::freeaddrinfo(service_info_list);
269 bool SocketAddress::SetToLocalhost(sa_family_t family, uint16_t port) {
274 m_socket_addr.sa_ipv4.sin_addr.s_addr = htonl(INADDR_LOOPBACK);
282 m_socket_addr.sa_ipv6.sin6_addr = in6addr_loopback;
291 bool SocketAddress::SetToAnyAddress(sa_family_t family, uint16_t port) {
296 m_socket_addr.sa_ipv4.sin_addr.s_addr = htonl(INADDR_ANY);
304 m_socket_addr.sa_ipv6.sin6_addr = in6addr_any;
313 bool SocketAddress::IsAnyAddr() const {
314 return (GetFamily() == AF_INET)
315 ? m_socket_addr.sa_ipv4.sin_addr.s_addr == htonl(INADDR_ANY)
316 : 0 == memcmp(&m_socket_addr.sa_ipv6.sin6_addr, &in6addr_any, 16);
319 bool SocketAddress::operator==(const SocketAddress &rhs) const {
320 if (GetFamily() != rhs.GetFamily())
322 if (GetLength() != rhs.GetLength())
324 switch (GetFamily()) {
326 return m_socket_addr.sa_ipv4.sin_addr.s_addr ==
327 rhs.m_socket_addr.sa_ipv4.sin_addr.s_addr;
329 return 0 == memcmp(&m_socket_addr.sa_ipv6.sin6_addr,
330 &rhs.m_socket_addr.sa_ipv6.sin6_addr, 16);
335 bool SocketAddress::operator!=(const SocketAddress &rhs) const {
336 return !(*this == rhs);