SocketAddress.cpp revision 258054
1//===-- SocketAddress.cpp ---------------------------------------*- C++ -*-===// 2// 3// The LLVM Compiler Infrastructure 4// 5// This file is distributed under the University of Illinois Open Source 6// License. See LICENSE.TXT for details. 7// 8//===----------------------------------------------------------------------===// 9 10#include "lldb/Host/SocketAddress.h" 11#include <stddef.h> 12 13// C Includes 14#if !defined(_MSC_VER) 15#include <arpa/inet.h> 16#endif 17#include <assert.h> 18#include <string.h> 19 20// C++ Includes 21// Other libraries and framework includes 22// Project includes 23 24using namespace lldb_private; 25 26//---------------------------------------------------------------------- 27// SocketAddress constructor 28//---------------------------------------------------------------------- 29SocketAddress::SocketAddress () 30{ 31 Clear (); 32} 33 34SocketAddress::SocketAddress (const struct sockaddr &s) 35{ 36 m_socket_addr.sa = s; 37} 38 39 40SocketAddress::SocketAddress (const struct sockaddr_in &s) 41{ 42 m_socket_addr.sa_ipv4 = s; 43} 44 45 46SocketAddress::SocketAddress (const struct sockaddr_in6 &s) 47{ 48 m_socket_addr.sa_ipv6 = s; 49} 50 51 52SocketAddress::SocketAddress (const struct sockaddr_storage &s) 53{ 54 m_socket_addr.sa_storage = s; 55} 56 57//---------------------------------------------------------------------- 58// SocketAddress copy constructor 59//---------------------------------------------------------------------- 60SocketAddress::SocketAddress (const SocketAddress& rhs) : 61 m_socket_addr (rhs.m_socket_addr) 62{ 63} 64 65//---------------------------------------------------------------------- 66// Destructor 67//---------------------------------------------------------------------- 68SocketAddress::~SocketAddress() 69{ 70} 71 72void 73SocketAddress::Clear () 74{ 75 memset (&m_socket_addr, 0, sizeof(m_socket_addr)); 76} 77 78bool 79SocketAddress::IsValid () const 80{ 81 return GetLength () != 0; 82} 83 84static socklen_t 85GetFamilyLength (sa_family_t family) 86{ 87 switch (family) 88 { 89 case AF_INET: return sizeof(struct sockaddr_in); 90 case AF_INET6: return sizeof(struct sockaddr_in6); 91 } 92 assert(0 && "Unsupported address family"); 93} 94 95socklen_t 96SocketAddress::GetLength () const 97{ 98#if defined(__APPLE__) 99 return m_socket_addr.sa.sa_len; 100#else 101 return GetFamilyLength (GetFamily()); 102#endif 103} 104 105socklen_t 106SocketAddress::GetMaxLength () 107{ 108 return sizeof (sockaddr_t); 109} 110 111sa_family_t 112SocketAddress::GetFamily () const 113{ 114 return m_socket_addr.sa.sa_family; 115} 116 117void 118SocketAddress::SetFamily (sa_family_t family) 119{ 120 m_socket_addr.sa.sa_family = family; 121#if defined(__APPLE__) 122 m_socket_addr.sa.sa_len = GetFamilyLength (family); 123#endif 124} 125 126in_port_t 127SocketAddress::GetPort () const 128{ 129 switch (GetFamily()) 130 { 131 case AF_INET: return m_socket_addr.sa_ipv4.sin_port; 132 case AF_INET6: return m_socket_addr.sa_ipv6.sin6_port; 133 } 134 return 0; 135} 136 137bool 138SocketAddress::SetPort (in_port_t port) 139{ 140 switch (GetFamily()) 141 { 142 case AF_INET: 143 m_socket_addr.sa_ipv4.sin_port = htons(port); 144 return true; 145 146 case AF_INET6: 147 m_socket_addr.sa_ipv6.sin6_port = htons(port); 148 return true; 149 } 150 return false; 151} 152 153//---------------------------------------------------------------------- 154// SocketAddress assignment operator 155//---------------------------------------------------------------------- 156const SocketAddress& 157SocketAddress::operator=(const SocketAddress& rhs) 158{ 159 if (this != &rhs) 160 m_socket_addr = rhs.m_socket_addr; 161 return *this; 162} 163 164const SocketAddress& 165SocketAddress::operator=(const struct addrinfo *addr_info) 166{ 167 Clear(); 168 if (addr_info && 169 addr_info->ai_addr && 170 addr_info->ai_addrlen > 0&& 171 addr_info->ai_addrlen <= sizeof m_socket_addr) 172 { 173 ::memcpy (&m_socket_addr, 174 addr_info->ai_addr, 175 addr_info->ai_addrlen); 176 } 177 return *this; 178} 179 180const SocketAddress& 181SocketAddress::operator=(const struct sockaddr &s) 182{ 183 m_socket_addr.sa = s; 184 return *this; 185} 186 187const SocketAddress& 188SocketAddress::operator=(const struct sockaddr_in &s) 189{ 190 m_socket_addr.sa_ipv4 = s; 191 return *this; 192} 193 194const SocketAddress& 195SocketAddress::operator=(const struct sockaddr_in6 &s) 196{ 197 m_socket_addr.sa_ipv6 = s; 198 return *this; 199} 200 201const SocketAddress& 202SocketAddress::operator=(const struct sockaddr_storage &s) 203{ 204 m_socket_addr.sa_storage = s; 205 return *this; 206} 207 208bool 209SocketAddress::SetAddress (const struct addrinfo *hints_ptr, 210 const char *host, 211 const char *service, 212 struct addrinfo *addr_info_ptr) 213{ 214 struct addrinfo *service_info_list = NULL; 215 int err = ::getaddrinfo (host, service, hints_ptr, &service_info_list); 216 if (err == 0 && service_info_list) 217 { 218 if (addr_info_ptr) 219 *addr_info_ptr = *service_info_list; 220 *this = service_info_list; 221 } 222 else 223 Clear(); 224 225 :: freeaddrinfo (service_info_list); 226 227 const bool is_valid = IsValid(); 228 if (!is_valid) 229 { 230 if (addr_info_ptr) 231 ::memset (addr_info_ptr, 0, sizeof(struct addrinfo)); 232 } 233 return is_valid; 234} 235 236 237bool 238SocketAddress::SetToLocalhost (sa_family_t family, in_port_t port) 239{ 240 switch (family) 241 { 242 case AF_INET: 243 SetFamily (AF_INET); 244 if (SetPort (port)) 245 { 246 m_socket_addr.sa_ipv4.sin_addr.s_addr = htonl (INADDR_ANY); 247 return true; 248 } 249 break; 250 251 case AF_INET6: 252 SetFamily (AF_INET6); 253 if (SetPort (port)) 254 { 255 m_socket_addr.sa_ipv6.sin6_addr = in6addr_any; 256 return true; 257 } 258 break; 259 260 } 261 Clear(); 262 return false; 263} 264