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 
24 using namespace lldb_private;
25 
26 //----------------------------------------------------------------------
27 // SocketAddress constructor
28 //----------------------------------------------------------------------
29 SocketAddress::SocketAddress ()
30 {
31     Clear ();
32 }
33 
34 SocketAddress::SocketAddress (const struct sockaddr &s)
35 {
36     m_socket_addr.sa = s;
37 }
38 
39 
40 SocketAddress::SocketAddress (const struct sockaddr_in &s)
41 {
42     m_socket_addr.sa_ipv4 = s;
43 }
44 
45 
46 SocketAddress::SocketAddress (const struct sockaddr_in6 &s)
47 {
48     m_socket_addr.sa_ipv6 = s;
49 }
50 
51 
52 SocketAddress::SocketAddress (const struct sockaddr_storage &s)
53 {
54     m_socket_addr.sa_storage = s;
55 }
56 
57 //----------------------------------------------------------------------
58 // SocketAddress copy constructor
59 //----------------------------------------------------------------------
60 SocketAddress::SocketAddress (const SocketAddress& rhs) :
61     m_socket_addr (rhs.m_socket_addr)
62 {
63 }
64 
65 //----------------------------------------------------------------------
66 // Destructor
67 //----------------------------------------------------------------------
68 SocketAddress::~SocketAddress()
69 {
70 }
71 
72 void
73 SocketAddress::Clear ()
74 {
75     memset (&m_socket_addr, 0, sizeof(m_socket_addr));
76 }
77 
78 bool
79 SocketAddress::IsValid () const
80 {
81     return GetLength () != 0;
82 }
83 
84 static socklen_t
85 GetFamilyLength (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 
95 socklen_t
96 SocketAddress::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 
105 socklen_t
106 SocketAddress::GetMaxLength ()
107 {
108     return sizeof (sockaddr_t);
109 }
110 
111 sa_family_t
112 SocketAddress::GetFamily () const
113 {
114     return m_socket_addr.sa.sa_family;
115 }
116 
117 void
118 SocketAddress::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 
126 in_port_t
127 SocketAddress::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 
137 bool
138 SocketAddress::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 //----------------------------------------------------------------------
156 const SocketAddress&
157 SocketAddress::operator=(const SocketAddress& rhs)
158 {
159     if (this != &rhs)
160         m_socket_addr = rhs.m_socket_addr;
161     return *this;
162 }
163 
164 const SocketAddress&
165 SocketAddress::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 
180 const SocketAddress&
181 SocketAddress::operator=(const struct sockaddr &s)
182 {
183     m_socket_addr.sa = s;
184     return *this;
185 }
186 
187 const SocketAddress&
188 SocketAddress::operator=(const struct sockaddr_in &s)
189 {
190     m_socket_addr.sa_ipv4 = s;
191     return *this;
192 }
193 
194 const SocketAddress&
195 SocketAddress::operator=(const struct sockaddr_in6 &s)
196 {
197     m_socket_addr.sa_ipv6 = s;
198     return *this;
199 }
200 
201 const SocketAddress&
202 SocketAddress::operator=(const struct sockaddr_storage &s)
203 {
204     m_socket_addr.sa_storage = s;
205     return *this;
206 }
207 
208 bool
209 SocketAddress::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 
237 bool
238 SocketAddress::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