| OLD | NEW |
| 1 // Copyright (c) 2013, the Dart project authors. Please see the AUTHORS file | 1 // Copyright (c) 2013, the Dart project authors. Please see the AUTHORS file |
| 2 // for details. All rights reserved. Use of this source code is governed by a | 2 // for details. All rights reserved. Use of this source code is governed by a |
| 3 // BSD-style license that can be found in the LICENSE file. | 3 // BSD-style license that can be found in the LICENSE file. |
| 4 | 4 |
| 5 #ifndef BIN_SOCKET_H_ | 5 #ifndef BIN_SOCKET_H_ |
| 6 #define BIN_SOCKET_H_ | 6 #define BIN_SOCKET_H_ |
| 7 | 7 |
| 8 #include <map> | 8 #include <map> |
| 9 | 9 |
| 10 #include "platform/globals.h" | 10 #include "platform/globals.h" |
| 11 | 11 |
| 12 #include "bin/builtin.h" | 12 #include "bin/builtin.h" |
| 13 #include "bin/dartutils.h" | 13 #include "bin/dartutils.h" |
| 14 // Declare the OS-specific types ahead of defining the generic class. | 14 // Declare the OS-specific types ahead of defining the generic class. |
| 15 #if defined(TARGET_OS_ANDROID) | 15 #if defined(TARGET_OS_ANDROID) |
| 16 #include "bin/socket_android.h" | 16 #include "bin/socket_android.h" |
| 17 #elif defined(TARGET_OS_LINUX) | 17 #elif defined(TARGET_OS_LINUX) |
| 18 #include "bin/socket_linux.h" | 18 #include "bin/socket_linux.h" |
| 19 #elif defined(TARGET_OS_MACOS) | 19 #elif defined(TARGET_OS_MACOS) |
| 20 #include "bin/socket_macos.h" | 20 #include "bin/socket_macos.h" |
| 21 #elif defined(TARGET_OS_WINDOWS) | 21 #elif defined(TARGET_OS_WINDOWS) |
| 22 #include "bin/socket_win.h" | 22 #include "bin/socket_win.h" |
| 23 #else | 23 #else |
| 24 #error Unknown target os. | 24 #error Unknown target os. |
| 25 #endif | 25 #endif |
| 26 #include "bin/thread.h" | 26 #include "bin/thread.h" |
| 27 #include "bin/utils.h" | 27 #include "bin/utils.h" |
| 28 | 28 |
| 29 | |
| 30 namespace dart { | 29 namespace dart { |
| 31 namespace bin { | 30 namespace bin { |
| 32 | 31 |
| 33 union RawAddr { | 32 union RawAddr { |
| 34 struct sockaddr_in in; | 33 struct sockaddr_in in; |
| 35 struct sockaddr_in6 in6; | 34 struct sockaddr_in6 in6; |
| 36 struct sockaddr_storage ss; | 35 struct sockaddr_storage ss; |
| 37 struct sockaddr addr; | 36 struct sockaddr addr; |
| 38 }; | 37 }; |
| 39 | 38 |
| (...skipping 12 matching lines...) Expand all Loading... |
| 52 ADDRESS_ANY_IP_V6, | 51 ADDRESS_ANY_IP_V6, |
| 53 ADDRESS_FIRST = ADDRESS_LOOPBACK_IP_V4, | 52 ADDRESS_FIRST = ADDRESS_LOOPBACK_IP_V4, |
| 54 ADDRESS_LAST = ADDRESS_ANY_IP_V6, | 53 ADDRESS_LAST = ADDRESS_ANY_IP_V6, |
| 55 }; | 54 }; |
| 56 | 55 |
| 57 explicit SocketAddress(struct sockaddr* sa); | 56 explicit SocketAddress(struct sockaddr* sa); |
| 58 | 57 |
| 59 ~SocketAddress() {} | 58 ~SocketAddress() {} |
| 60 | 59 |
| 61 int GetType() { | 60 int GetType() { |
| 62 if (addr_.ss.ss_family == AF_INET6) return TYPE_IPV6; | 61 if (addr_.ss.ss_family == AF_INET6) { |
| 62 return TYPE_IPV6; |
| 63 } |
| 63 return TYPE_IPV4; | 64 return TYPE_IPV4; |
| 64 } | 65 } |
| 65 | 66 |
| 66 const char* as_string() const { return as_string_; } | 67 const char* as_string() const { return as_string_; } |
| 67 const RawAddr& addr() const { return addr_; } | 68 const RawAddr& addr() const { return addr_; } |
| 68 | 69 |
| 69 static intptr_t GetAddrLength(const RawAddr& addr) { | 70 static intptr_t GetAddrLength(const RawAddr& addr) { |
| 70 ASSERT(addr.ss.ss_family == AF_INET || addr.ss.ss_family == AF_INET6); | 71 ASSERT((addr.ss.ss_family == AF_INET) || (addr.ss.ss_family == AF_INET6)); |
| 71 return addr.ss.ss_family == AF_INET6 ? | 72 return (addr.ss.ss_family == AF_INET6) ? |
| 72 sizeof(struct sockaddr_in6) : sizeof(struct sockaddr_in); | 73 sizeof(struct sockaddr_in6) : sizeof(struct sockaddr_in); |
| 73 } | 74 } |
| 74 | 75 |
| 75 static intptr_t GetInAddrLength(const RawAddr& addr) { | 76 static intptr_t GetInAddrLength(const RawAddr& addr) { |
| 76 ASSERT(addr.ss.ss_family == AF_INET || addr.ss.ss_family == AF_INET6); | 77 ASSERT((addr.ss.ss_family == AF_INET) || (addr.ss.ss_family == AF_INET6)); |
| 77 return addr.ss.ss_family == AF_INET6 ? | 78 return (addr.ss.ss_family == AF_INET6) ? |
| 78 sizeof(struct in6_addr) : sizeof(struct in_addr); | 79 sizeof(struct in6_addr) : sizeof(struct in_addr); |
| 79 } | 80 } |
| 80 | 81 |
| 81 static bool AreAddressesEqual(const RawAddr& a, const RawAddr& b) { | 82 static bool AreAddressesEqual(const RawAddr& a, const RawAddr& b) { |
| 82 if (a.ss.ss_family == AF_INET) { | 83 if (a.ss.ss_family == AF_INET) { |
| 83 if (b.ss.ss_family != AF_INET) return false; | 84 if (b.ss.ss_family != AF_INET) { |
| 85 return false; |
| 86 } |
| 84 return memcmp(&a.in.sin_addr, &b.in.sin_addr, sizeof(a.in.sin_addr)) == 0; | 87 return memcmp(&a.in.sin_addr, &b.in.sin_addr, sizeof(a.in.sin_addr)) == 0; |
| 85 } else if (a.ss.ss_family == AF_INET6) { | 88 } else if (a.ss.ss_family == AF_INET6) { |
| 86 if (b.ss.ss_family != AF_INET6) return false; | 89 if (b.ss.ss_family != AF_INET6) { |
| 90 return false; |
| 91 } |
| 87 return memcmp(&a.in6.sin6_addr, | 92 return memcmp(&a.in6.sin6_addr, |
| 88 &b.in6.sin6_addr, | 93 &b.in6.sin6_addr, |
| 89 sizeof(a.in6.sin6_addr)) == 0; | 94 sizeof(a.in6.sin6_addr)) == 0; |
| 90 } else { | 95 } else { |
| 91 UNREACHABLE(); | 96 UNREACHABLE(); |
| 92 return false; | 97 return false; |
| 93 } | 98 } |
| 94 } | 99 } |
| 95 | 100 |
| 96 static void GetSockAddr(Dart_Handle obj, RawAddr* addr) { | 101 static void GetSockAddr(Dart_Handle obj, RawAddr* addr) { |
| 97 Dart_TypedData_Type data_type; | 102 Dart_TypedData_Type data_type; |
| 98 uint8_t* data = NULL; | 103 uint8_t* data = NULL; |
| 99 intptr_t len; | 104 intptr_t len; |
| 100 Dart_Handle result = Dart_TypedDataAcquireData( | 105 Dart_Handle result = Dart_TypedDataAcquireData( |
| 101 obj, &data_type, reinterpret_cast<void**>(&data), &len); | 106 obj, &data_type, reinterpret_cast<void**>(&data), &len); |
| 102 if (Dart_IsError(result)) Dart_PropagateError(result); | 107 if (Dart_IsError(result)) { |
| 103 if (data_type != Dart_TypedData_kUint8 || | 108 Dart_PropagateError(result); |
| 104 (len != sizeof(in_addr) && len != sizeof(in6_addr))) { | 109 } |
| 110 if ((data_type != Dart_TypedData_kUint8) || |
| 111 ((len != sizeof(in_addr)) && (len != sizeof(in6_addr)))) { |
| 105 Dart_PropagateError( | 112 Dart_PropagateError( |
| 106 Dart_NewApiError("Unexpected type for socket address")); | 113 Dart_NewApiError("Unexpected type for socket address")); |
| 107 } | 114 } |
| 108 memset(reinterpret_cast<void*>(addr), 0, sizeof(RawAddr)); | 115 memset(reinterpret_cast<void*>(addr), 0, sizeof(RawAddr)); |
| 109 if (len == sizeof(in_addr)) { | 116 if (len == sizeof(in_addr)) { |
| 110 addr->in.sin_family = AF_INET; | 117 addr->in.sin_family = AF_INET; |
| 111 memmove(reinterpret_cast<void *>(&addr->in.sin_addr), data, len); | 118 memmove(reinterpret_cast<void *>(&addr->in.sin_addr), data, len); |
| 112 } else { | 119 } else { |
| 113 ASSERT(len == sizeof(in6_addr)); | 120 ASSERT(len == sizeof(in6_addr)); |
| 114 addr->in6.sin6_family = AF_INET6; | 121 addr->in6.sin6_family = AF_INET6; |
| 115 memmove(reinterpret_cast<void*>(&addr->in6.sin6_addr), data, len); | 122 memmove(reinterpret_cast<void*>(&addr->in6.sin6_addr), data, len); |
| 116 } | 123 } |
| 117 Dart_TypedDataReleaseData(obj); | 124 Dart_TypedDataReleaseData(obj); |
| 118 } | 125 } |
| 119 | 126 |
| 120 static int16_t FromType(int type) { | 127 static int16_t FromType(int type) { |
| 121 if (type == TYPE_ANY) return AF_UNSPEC; | 128 if (type == TYPE_ANY) { |
| 122 if (type == TYPE_IPV4) return AF_INET; | 129 return AF_UNSPEC; |
| 123 ASSERT(type == TYPE_IPV6 && "Invalid type"); | 130 } |
| 131 if (type == TYPE_IPV4) { |
| 132 return AF_INET; |
| 133 } |
| 134 ASSERT((type == TYPE_IPV6) && "Invalid type"); |
| 124 return AF_INET6; | 135 return AF_INET6; |
| 125 } | 136 } |
| 126 | 137 |
| 127 static void SetAddrPort(RawAddr* addr, intptr_t port) { | 138 static void SetAddrPort(RawAddr* addr, intptr_t port) { |
| 128 if (addr->ss.ss_family == AF_INET) { | 139 if (addr->ss.ss_family == AF_INET) { |
| 129 addr->in.sin_port = htons(port); | 140 addr->in.sin_port = htons(port); |
| 130 } else { | 141 } else { |
| 131 addr->in6.sin6_port = htons(port); | 142 addr->in6.sin6_port = htons(port); |
| 132 } | 143 } |
| 133 } | 144 } |
| 134 | 145 |
| 135 static intptr_t GetAddrPort(const RawAddr& addr) { | 146 static intptr_t GetAddrPort(const RawAddr& addr) { |
| 136 if (addr.ss.ss_family == AF_INET) { | 147 if (addr.ss.ss_family == AF_INET) { |
| 137 return ntohs(addr.in.sin_port); | 148 return ntohs(addr.in.sin_port); |
| 138 } else { | 149 } else { |
| 139 return ntohs(addr.in6.sin6_port); | 150 return ntohs(addr.in6.sin6_port); |
| 140 } | 151 } |
| 141 } | 152 } |
| 142 | 153 |
| 143 static Dart_Handle ToTypedData(const RawAddr& addr) { | 154 static Dart_Handle ToTypedData(const RawAddr& addr) { |
| 144 int len = GetInAddrLength(addr); | 155 int len = GetInAddrLength(addr); |
| 145 Dart_Handle result = Dart_NewTypedData(Dart_TypedData_kUint8, len); | 156 Dart_Handle result = Dart_NewTypedData(Dart_TypedData_kUint8, len); |
| 146 if (Dart_IsError(result)) Dart_PropagateError(result); | 157 if (Dart_IsError(result)) { |
| 158 Dart_PropagateError(result); |
| 159 } |
| 147 Dart_Handle err; | 160 Dart_Handle err; |
| 148 if (addr.addr.sa_family == AF_INET6) { | 161 if (addr.addr.sa_family == AF_INET6) { |
| 149 err = Dart_ListSetAsBytes( | 162 err = Dart_ListSetAsBytes( |
| 150 result, 0, | 163 result, 0, |
| 151 reinterpret_cast<const uint8_t*>(&addr.in6.sin6_addr), len); | 164 reinterpret_cast<const uint8_t*>(&addr.in6.sin6_addr), len); |
| 152 } else { | 165 } else { |
| 153 err = Dart_ListSetAsBytes( | 166 err = Dart_ListSetAsBytes( |
| 154 result, 0, reinterpret_cast<const uint8_t*>(&addr.in.sin_addr), len); | 167 result, 0, reinterpret_cast<const uint8_t*>(&addr.in.sin_addr), len); |
| 155 } | 168 } |
| 156 if (Dart_IsError(err)) Dart_PropagateError(err); | 169 if (Dart_IsError(err)) { |
| 170 Dart_PropagateError(err); |
| 171 } |
| 157 return result; | 172 return result; |
| 158 } | 173 } |
| 159 | 174 |
| 160 static CObjectUint8Array* ToCObject(const RawAddr& addr) { | 175 static CObjectUint8Array* ToCObject(const RawAddr& addr) { |
| 161 int in_addr_len = SocketAddress::GetInAddrLength(addr); | 176 int in_addr_len = SocketAddress::GetInAddrLength(addr); |
| 162 const void* in_addr; | 177 const void* in_addr; |
| 163 CObjectUint8Array* data = | 178 CObjectUint8Array* data = |
| 164 new CObjectUint8Array(CObject::NewUint8Array(in_addr_len)); | 179 new CObjectUint8Array(CObject::NewUint8Array(in_addr_len)); |
| 165 if (addr.addr.sa_family == AF_INET6) { | 180 if (addr.addr.sa_family == AF_INET6) { |
| 166 in_addr = reinterpret_cast<const void*>(&addr.in6.sin6_addr); | 181 in_addr = reinterpret_cast<const void*>(&addr.in6.sin6_addr); |
| 167 } else { | 182 } else { |
| 168 in_addr = reinterpret_cast<const void*>(&addr.in.sin_addr); | 183 in_addr = reinterpret_cast<const void*>(&addr.in.sin_addr); |
| 169 } | 184 } |
| 170 memmove(data->Buffer(), in_addr, in_addr_len); | 185 memmove(data->Buffer(), in_addr, in_addr_len); |
| 171 return data; | 186 return data; |
| 172 } | 187 } |
| 173 | 188 |
| 174 private: | 189 private: |
| 175 char as_string_[INET6_ADDRSTRLEN]; | 190 char as_string_[INET6_ADDRSTRLEN]; |
| 176 RawAddr addr_; | 191 RawAddr addr_; |
| 177 | 192 |
| 178 DISALLOW_COPY_AND_ASSIGN(SocketAddress); | 193 DISALLOW_COPY_AND_ASSIGN(SocketAddress); |
| 179 }; | 194 }; |
| 180 | 195 |
| 196 |
| 181 class InterfaceSocketAddress { | 197 class InterfaceSocketAddress { |
| 182 public: | 198 public: |
| 183 explicit InterfaceSocketAddress(struct sockaddr* sa, | 199 InterfaceSocketAddress(struct sockaddr* sa, |
| 184 const char* interface_name, | 200 const char* interface_name, |
| 185 intptr_t interface_index) | 201 intptr_t interface_index) |
| 186 : socket_address_(new SocketAddress(sa)), | 202 : socket_address_(new SocketAddress(sa)), |
| 187 interface_name_(interface_name), | 203 interface_name_(interface_name), |
| 188 interface_index_(interface_index) {} | 204 interface_index_(interface_index) {} |
| 189 | 205 |
| 190 ~InterfaceSocketAddress() { | 206 ~InterfaceSocketAddress() { |
| 191 delete socket_address_; | 207 delete socket_address_; |
| 192 } | 208 } |
| 193 | 209 |
| 194 SocketAddress* socket_address() const { return socket_address_; } | 210 SocketAddress* socket_address() const { return socket_address_; } |
| 195 const char* interface_name() const { return interface_name_; } | 211 const char* interface_name() const { return interface_name_; } |
| 196 int interface_index() const { return interface_index_; } | 212 int interface_index() const { return interface_index_; } |
| 197 | 213 |
| 198 private: | 214 private: |
| 199 SocketAddress* socket_address_; | 215 SocketAddress* socket_address_; |
| 200 const char* interface_name_; | 216 const char* interface_name_; |
| 201 intptr_t interface_index_; | 217 intptr_t interface_index_; |
| 202 | 218 |
| 203 DISALLOW_COPY_AND_ASSIGN(InterfaceSocketAddress); | 219 DISALLOW_COPY_AND_ASSIGN(InterfaceSocketAddress); |
| 204 }; | 220 }; |
| 205 | 221 |
| 222 |
| 206 template<typename T> | 223 template<typename T> |
| 207 class AddressList { | 224 class AddressList { |
| 208 public: | 225 public: |
| 209 explicit AddressList(intptr_t count) | 226 explicit AddressList(intptr_t count) |
| 210 : count_(count), | 227 : count_(count), |
| 211 addresses_(new T*[count_]) {} | 228 addresses_(new T*[count_]) {} |
| 212 | 229 |
| 213 ~AddressList() { | 230 ~AddressList() { |
| 214 for (intptr_t i = 0; i < count_; i++) { | 231 for (intptr_t i = 0; i < count_; i++) { |
| 215 delete addresses_[i]; | 232 delete addresses_[i]; |
| 216 } | 233 } |
| 217 delete[] addresses_; | 234 delete[] addresses_; |
| 218 } | 235 } |
| 219 | 236 |
| 220 intptr_t count() const { return count_; } | 237 intptr_t count() const { return count_; } |
| 221 T* GetAt(intptr_t i) const { return addresses_[i]; } | 238 T* GetAt(intptr_t i) const { return addresses_[i]; } |
| 222 void SetAt(intptr_t i, T* addr) { addresses_[i] = addr; } | 239 void SetAt(intptr_t i, T* addr) { addresses_[i] = addr; } |
| 223 | 240 |
| 224 private: | 241 private: |
| 225 const intptr_t count_; | 242 const intptr_t count_; |
| 226 T** addresses_; | 243 T** addresses_; |
| 227 | 244 |
| 228 DISALLOW_COPY_AND_ASSIGN(AddressList); | 245 DISALLOW_COPY_AND_ASSIGN(AddressList); |
| 229 }; | 246 }; |
| 230 | 247 |
| 248 |
| 231 class Socket { | 249 class Socket { |
| 232 public: | 250 public: |
| 233 enum SocketRequest { | 251 enum SocketRequest { |
| 234 kLookupRequest = 0, | 252 kLookupRequest = 0, |
| 235 kListInterfacesRequest = 1, | 253 kListInterfacesRequest = 1, |
| 236 kReverseLookupRequest = 2, | 254 kReverseLookupRequest = 2, |
| 237 }; | 255 }; |
| 238 | 256 |
| 239 static bool Initialize(); | 257 static bool Initialize(); |
| 240 static intptr_t Available(intptr_t fd); | 258 static intptr_t Available(intptr_t fd); |
| (...skipping 92 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 333 | 351 |
| 334 // Start accepting on a newly created listening socket. If it was unable to | 352 // Start accepting on a newly created listening socket. If it was unable to |
| 335 // start accepting incoming sockets, the fd is invalidated. | 353 // start accepting incoming sockets, the fd is invalidated. |
| 336 static bool StartAccept(intptr_t fd); | 354 static bool StartAccept(intptr_t fd); |
| 337 | 355 |
| 338 private: | 356 private: |
| 339 DISALLOW_ALLOCATION(); | 357 DISALLOW_ALLOCATION(); |
| 340 DISALLOW_IMPLICIT_CONSTRUCTORS(ServerSocket); | 358 DISALLOW_IMPLICIT_CONSTRUCTORS(ServerSocket); |
| 341 }; | 359 }; |
| 342 | 360 |
| 361 |
| 343 class ListeningSocketRegistry { | 362 class ListeningSocketRegistry { |
| 344 private: | 363 private: |
| 345 struct OSSocket { | 364 struct OSSocket { |
| 346 RawAddr address; | 365 RawAddr address; |
| 347 int port; | 366 int port; |
| 348 bool v6_only; | 367 bool v6_only; |
| 349 bool shared; | 368 bool shared; |
| 350 int ref_count; | 369 int ref_count; |
| 351 intptr_t socketfd; | 370 intptr_t socketfd; |
| 352 | 371 |
| (...skipping 56 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 409 std::map<intptr_t, OSSocket*> sockets_by_port_; | 428 std::map<intptr_t, OSSocket*> sockets_by_port_; |
| 410 std::map<intptr_t, OSSocket*> sockets_by_fd_; | 429 std::map<intptr_t, OSSocket*> sockets_by_fd_; |
| 411 Mutex *mutex_; | 430 Mutex *mutex_; |
| 412 | 431 |
| 413 typedef std::map<intptr_t, OSSocket*>::iterator SocketsIterator; | 432 typedef std::map<intptr_t, OSSocket*>::iterator SocketsIterator; |
| 414 | 433 |
| 415 private: | 434 private: |
| 416 DISALLOW_COPY_AND_ASSIGN(ListeningSocketRegistry); | 435 DISALLOW_COPY_AND_ASSIGN(ListeningSocketRegistry); |
| 417 }; | 436 }; |
| 418 | 437 |
| 419 | |
| 420 } // namespace bin | 438 } // namespace bin |
| 421 } // namespace dart | 439 } // namespace dart |
| 422 | 440 |
| 423 #endif // BIN_SOCKET_H_ | 441 #endif // BIN_SOCKET_H_ |
| OLD | NEW |