| Index: src/platform/socket.cc
|
| diff --git a/src/platform/socket.cc b/src/platform/socket.cc
|
| new file mode 100644
|
| index 0000000000000000000000000000000000000000..e82c38c32c77c08ea46cc91a44717e865835124d
|
| --- /dev/null
|
| +++ b/src/platform/socket.cc
|
| @@ -0,0 +1,223 @@
|
| +// Copyright 2013 the V8 project authors. All rights reserved.
|
| +// Redistribution and use in source and binary forms, with or without
|
| +// modification, are permitted provided that the following conditions are
|
| +// met:
|
| +//
|
| +// * Redistributions of source code must retain the above copyright
|
| +// notice, this list of conditions and the following disclaimer.
|
| +// * Redistributions in binary form must reproduce the above
|
| +// copyright notice, this list of conditions and the following
|
| +// disclaimer in the documentation and/or other materials provided
|
| +// with the distribution.
|
| +// * Neither the name of Google Inc. nor the names of its
|
| +// contributors may be used to endorse or promote products derived
|
| +// from this software without specific prior written permission.
|
| +//
|
| +// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
|
| +// "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
|
| +// LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
|
| +// A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
|
| +// OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
|
| +// SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
|
| +// LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
|
| +// DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
|
| +// THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
| +// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
|
| +// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
| +
|
| +#include "platform/socket.h"
|
| +
|
| +#if V8_OS_POSIX
|
| +#include <sys/types.h>
|
| +#include <sys/socket.h>
|
| +
|
| +#include <netinet/in.h>
|
| +#include <netdb.h>
|
| +
|
| +#include <unistd.h>
|
| +#endif
|
| +
|
| +#include <cerrno>
|
| +
|
| +#include "checks.h"
|
| +#include "once.h"
|
| +
|
| +namespace v8 {
|
| +namespace internal {
|
| +
|
| +#if V8_OS_WIN
|
| +
|
| +static V8_DECLARE_ONCE(initialize_winsock) = V8_ONCE_INIT;
|
| +
|
| +
|
| +static void InitializeWinsock() {
|
| + WSADATA wsa_data;
|
| + int result = WSAStartup(MAKEWORD(1, 0), &wsa_data);
|
| + CHECK_EQ(0, result);
|
| +}
|
| +
|
| +#endif // V8_OS_WIN
|
| +
|
| +
|
| +Socket::Socket() {
|
| +#if V8_OS_WIN
|
| + // Be sure to initialize the WinSock DLL first.
|
| + CallOnce(&initialize_winsock, &InitializeWinsock);
|
| +#endif // V8_OS_WIN
|
| +
|
| + // Create the native socket handle.
|
| + native_handle_ = ::socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
|
| +}
|
| +
|
| +
|
| +bool Socket::Bind(int port) {
|
| + ASSERT_GE(port, 0);
|
| + ASSERT_LT(port, 65536);
|
| + if (!IsValid()) return false;
|
| + struct sockaddr_in sin;
|
| + memset(&sin, 0, sizeof(sin));
|
| + sin.sin_family = AF_INET;
|
| + sin.sin_addr.s_addr = htonl(INADDR_LOOPBACK);
|
| + sin.sin_port = htons(static_cast<uint16_t>(port));
|
| + int result = ::bind(
|
| + native_handle_, reinterpret_cast<struct sockaddr*>(&sin), sizeof(sin));
|
| + return result == 0;
|
| +}
|
| +
|
| +
|
| +bool Socket::Listen(int backlog) {
|
| + if (!IsValid()) return false;
|
| + int result = ::listen(native_handle_, backlog);
|
| + return result == 0;
|
| +}
|
| +
|
| +
|
| +Socket* Socket::Accept() {
|
| + if (!IsValid()) return NULL;
|
| + while (true) {
|
| + NativeHandle native_handle = ::accept(native_handle_, NULL, NULL);
|
| + if (native_handle == kInvalidNativeHandle) {
|
| +#if V8_OS_POSIX
|
| + if (errno == EINTR) continue; // Retry after signal.
|
| +#endif
|
| + return NULL;
|
| + }
|
| + return new Socket(native_handle);
|
| + }
|
| +}
|
| +
|
| +
|
| +bool Socket::Connect(const char* host, const char* port) {
|
| + ASSERT_NE(NULL, host);
|
| + ASSERT_NE(NULL, port);
|
| + if (!IsValid()) return false;
|
| +
|
| + // Lookup host and port.
|
| + struct addrinfo* info = NULL;
|
| + struct addrinfo hint;
|
| + memset(&hint, 0, sizeof(hint));
|
| + hint.ai_family = AF_INET;
|
| + hint.ai_socktype = SOCK_STREAM;
|
| + hint.ai_protocol = IPPROTO_TCP;
|
| + int result = ::getaddrinfo(host, port, &hint, &info);
|
| + if (result != 0) {
|
| + return false;
|
| + }
|
| +
|
| + // Connect to the host on the given port.
|
| + for (struct addrinfo* ai = info; ai != NULL; ai = ai->ai_next) {
|
| + // Try to connect using this addr info.
|
| + while (true) {
|
| + result = ::connect(native_handle_, ai->ai_addr, ai->ai_addrlen);
|
| + if (result == 0) {
|
| + freeaddrinfo(info);
|
| + return true;
|
| + }
|
| +#if V8_OS_POSIX
|
| + if (errno == EINTR) continue; // Retry after signal.
|
| +#endif
|
| + break;
|
| + }
|
| + }
|
| + freeaddrinfo(info);
|
| + return false;
|
| +}
|
| +
|
| +
|
| +bool Socket::Shutdown() {
|
| + if (!IsValid()) return false;
|
| + // Shutdown socket for both read and write.
|
| +#if V8_OS_POSIX
|
| + int result = ::shutdown(native_handle_, SHUT_RDWR);
|
| + ::close(native_handle_);
|
| +#elif V8_OS_WIN
|
| + int result = ::shutdown(native_handle_, SD_BOTH);
|
| + ::closesocket(native_handle_);
|
| +#endif
|
| + native_handle_ = kInvalidNativeHandle;
|
| + return result == 0;
|
| +}
|
| +
|
| +
|
| +int Socket::Send(const char* buffer, int length) {
|
| + ASSERT(length <= 0 || buffer != NULL);
|
| + if (!IsValid()) return 0;
|
| + int offset = 0;
|
| + while (offset < length) {
|
| + int result = ::send(native_handle_, buffer + offset, length - offset, 0);
|
| + if (result == 0) {
|
| + break;
|
| + } else if (result > 0) {
|
| + ASSERT(result <= length - offset);
|
| + offset += result;
|
| + } else {
|
| +#if V8_OS_POSIX
|
| + if (errno == EINTR) continue; // Retry after signal.
|
| +#endif
|
| + return 0;
|
| + }
|
| + }
|
| + return offset;
|
| +}
|
| +
|
| +
|
| +int Socket::Receive(char* buffer, int length) {
|
| + if (!IsValid()) return 0;
|
| + if (length <= 0) return 0;
|
| + ASSERT_NE(NULL, buffer);
|
| + while (true) {
|
| + int result = ::recv(native_handle_, buffer, length, 0);
|
| + if (result < 0) {
|
| +#if V8_OS_POSIX
|
| + if (errno == EINTR) continue; // Retry after signal.
|
| +#endif
|
| + return 0;
|
| + }
|
| + return result;
|
| + }
|
| +}
|
| +
|
| +
|
| +bool Socket::SetReuseAddress(bool reuse_address) {
|
| + if (!IsValid()) return 0;
|
| + int v = reuse_address ? 1 : 0;
|
| + int result = ::setsockopt(native_handle_, SOL_SOCKET, SO_REUSEADDR,
|
| + reinterpret_cast<char*>(&v), sizeof(v));
|
| + return result == 0;
|
| +}
|
| +
|
| +
|
| +// static
|
| +int Socket::GetLastError() {
|
| +#if V8_OS_POSIX
|
| + return errno;
|
| +#elif V8_OS_WIN
|
| + // Be sure to initialize the WinSock DLL first.
|
| + CallOnce(&initialize_winsock, &InitializeWinsock);
|
| +
|
| + // Now we can safely perform WSA calls.
|
| + return ::WSAGetLastError();
|
| +#endif
|
| +}
|
| +
|
| +} } // namespace v8::internal
|
|
|