SocketAddress.cpp revision 258054
150276Speter//===-- SocketAddress.cpp ---------------------------------------*- C++ -*-===// 2166124Srafan// 350276Speter// The LLVM Compiler Infrastructure 450276Speter// 550276Speter// This file is distributed under the University of Illinois Open Source 650276Speter// License. See LICENSE.TXT for details. 750276Speter// 850276Speter//===----------------------------------------------------------------------===// 950276Speter 1050276Speter#include "lldb/Host/SocketAddress.h" 1150276Speter#include <stddef.h> 1250276Speter 1350276Speter// C Includes 1450276Speter#if !defined(_MSC_VER) 1550276Speter#include <arpa/inet.h> 1650276Speter#endif 1750276Speter#include <assert.h> 1850276Speter#include <string.h> 1950276Speter 2050276Speter// C++ Includes 2150276Speter// Other libraries and framework includes 2250276Speter// Project includes 2350276Speter 2450276Speterusing namespace lldb_private; 2550276Speter 2650276Speter//---------------------------------------------------------------------- 2750276Speter// SocketAddress constructor 2850276Speter//---------------------------------------------------------------------- 2950276SpeterSocketAddress::SocketAddress () 3050276Speter{ 3150276Speter Clear (); 32166124Srafan} 3350276Speter 3450276SpeterSocketAddress::SocketAddress (const struct sockaddr &s) 3550276Speter{ 3650276Speter m_socket_addr.sa = s; 3750276Speter} 3850276Speter 3950276Speter 4050276SpeterSocketAddress::SocketAddress (const struct sockaddr_in &s) 4150276Speter{ 4250276Speter m_socket_addr.sa_ipv4 = s; 4350276Speter} 4450276Speter 45166124Srafan 4650276SpeterSocketAddress::SocketAddress (const struct sockaddr_in6 &s) 4776726Speter{ 48166124Srafan m_socket_addr.sa_ipv6 = s; 4950276Speter} 5076726Speter 51166124Srafan 52166124SrafanSocketAddress::SocketAddress (const struct sockaddr_storage &s) 53166124Srafan{ 54166124Srafan m_socket_addr.sa_storage = s; 55166124Srafan} 56166124Srafan 57166124Srafan//---------------------------------------------------------------------- 58166124Srafan// SocketAddress copy constructor 59166124Srafan//---------------------------------------------------------------------- 60166124SrafanSocketAddress::SocketAddress (const SocketAddress& rhs) : 6176726Speter m_socket_addr (rhs.m_socket_addr) 6276726Speter{ 6376726Speter} 6450276Speter 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