Index: net/socket/tcp_server_socket_libevent.cc |
diff --git a/net/socket/tcp_server_socket_libevent.cc b/net/socket/tcp_server_socket_libevent.cc |
new file mode 100644 |
index 0000000000000000000000000000000000000000..40a01f040e746f5ac950db18b2c8dcda040adf04 |
--- /dev/null |
+++ b/net/socket/tcp_server_socket_libevent.cc |
@@ -0,0 +1,194 @@ |
+// Copyright (c) 2011 The Chromium Authors. All rights reserved. |
+// Use of this source code is governed by a BSD-style license that can be |
+// found in the LICENSE file. |
+ |
+#include "net/socket/tcp_server_socket_libevent.h" |
+ |
+#include <errno.h> |
+#include <fcntl.h> |
+#include <netdb.h> |
+#include <sys/socket.h> |
+ |
+#include "build/build_config.h" |
+ |
+#if defined(OS_POSIX) |
+#include <netinet/in.h> |
+#endif |
+#if defined(USE_SYSTEM_LIBEVENT) |
+#include <event.h> |
+#else |
+#include "third_party/libevent/event.h" |
+#endif |
+ |
+#include "base/eintr_wrapper.h" |
+#include "net/base/ip_endpoint.h" |
+#include "net/base/net_errors.h" |
+#include "net/base/net_util.h" |
+#include "net/socket/tcp_client_socket.h" |
+ |
+namespace net { |
+ |
+namespace { |
+ |
+const int kInvalidSocket = -1; |
+ |
+} // namespace |
+ |
+TCPServerSocketLibevent::TCPServerSocketLibevent( |
+ net::NetLog* net_log, |
+ const net::NetLog::Source& source) |
+ : socket_(kInvalidSocket), |
+ accept_socket_(NULL), |
+ accept_callback_(NULL), |
+ net_log_(BoundNetLog::Make(net_log, NetLog::SOURCE_SOCKET)) { |
+ scoped_refptr<NetLog::EventParameters> params; |
+ if (source.is_valid()) |
+ params = new NetLogSourceParameter("source_dependency", source); |
+ net_log_.BeginEvent(NetLog::TYPE_SOCKET_ALIVE, params); |
+} |
+ |
+TCPServerSocketLibevent::~TCPServerSocketLibevent() { |
+ if (socket_ != kInvalidSocket) |
+ Close(); |
+ net_log_.EndEvent(NetLog::TYPE_SOCKET_ALIVE, NULL); |
+} |
+ |
+int TCPServerSocketLibevent::Listen(const IPEndPoint& address, int backlog) { |
+ DCHECK(CalledOnValidThread()); |
+ DCHECK_GT(backlog, 0); |
+ DCHECK_EQ(socket_, kInvalidSocket); |
+ |
+ socket_ = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP); |
+ if (socket_ < 0) { |
+ PLOG(ERROR) << "socket() returned an error"; |
+ return MapSystemError(errno); |
+ } |
+ |
+ if (SetNonBlocking(socket_)) { |
+ int result = MapSystemError(errno); |
+ Close(); |
+ return result; |
+ } |
+ |
+ struct sockaddr_storage addr_storage; |
+ size_t addr_len = sizeof(addr_storage); |
+ struct sockaddr* addr = reinterpret_cast<struct sockaddr*>(&addr_storage); |
+ if (!address.ToSockAddr(addr, &addr_len)) |
+ return ERR_INVALID_ARGUMENT; |
+ |
+ int result = bind(socket_, addr, addr_len); |
+ if (result < 0) { |
+ PLOG(ERROR) << "bind() returned an error"; |
+ result = MapSystemError(errno); |
+ Close(); |
+ return result; |
+ } |
+ |
+ result = listen(socket_, backlog); |
+ if (result < 0) { |
+ PLOG(ERROR) << "listen() returned an error"; |
+ result = MapSystemError(errno); |
+ Close(); |
+ return result; |
+ } |
+ |
+ return OK; |
+} |
+ |
+int TCPServerSocketLibevent::GetLocalAddress(IPEndPoint* address) const { |
+ DCHECK(CalledOnValidThread()); |
+ DCHECK(address); |
+ |
+ struct sockaddr_storage addr_storage; |
+ socklen_t addr_len = sizeof(addr_storage); |
+ struct sockaddr* addr = reinterpret_cast<struct sockaddr*>(&addr_storage); |
+ if (getsockname(socket_, addr, &addr_len) < 0) |
+ return MapSystemError(errno); |
+ if (!address->FromSockAddr(addr, addr_len)) |
+ return ERR_FAILED; |
+ |
+ return OK; |
+} |
+ |
+int TCPServerSocketLibevent::Accept( |
+ scoped_ptr<ClientSocket>* socket, CompletionCallback* callback) { |
+ DCHECK(CalledOnValidThread()); |
+ DCHECK(socket); |
+ DCHECK(callback); |
+ DCHECK(!accept_callback_); |
+ |
+ net_log_.BeginEvent(NetLog::TYPE_TCP_ACCEPT, NULL); |
+ |
+ int result = AcceptInternal(socket); |
+ |
+ if (result == ERR_IO_PENDING) { |
+ if (!MessageLoopForIO::current()->WatchFileDescriptor( |
+ socket_, true, MessageLoopForIO::WATCH_READ, |
+ &accept_socket_watcher_, this)) { |
+ PLOG(ERROR) << "WatchFileDescriptor failed on read"; |
+ return MapSystemError(errno); |
+ } |
+ |
+ accept_socket_ = socket; |
+ accept_callback_ = callback; |
+ } |
+ |
+ return result; |
+} |
+ |
+int TCPServerSocketLibevent::AcceptInternal( |
+ scoped_ptr<ClientSocket>* socket) { |
+ struct sockaddr_storage addr_storage; |
+ socklen_t addr_len = sizeof(addr_storage); |
+ struct sockaddr* addr = reinterpret_cast<struct sockaddr*>(&addr_storage); |
+ |
+ int result = HANDLE_EINTR(accept(socket_, addr, &addr_len)); |
+ if (result < 0) { |
+ int net_error = MapSystemError(errno); |
+ if (net_error != ERR_IO_PENDING) |
+ net_log_.EndEventWithNetErrorCode(NetLog::TYPE_TCP_ACCEPT, net_error); |
+ return net_error; |
+ } |
+ |
+ IPEndPoint address; |
+ if (!address.FromSockAddr(addr, addr_len)) { |
+ NOTREACHED(); |
+ HANDLE_EINTR(close(result)); |
+ net_log_.EndEventWithNetErrorCode(NetLog::TYPE_TCP_ACCEPT, ERR_FAILED); |
+ return ERR_FAILED; |
+ } |
+ TCPClientSocket* tcp_socket = new TCPClientSocket( |
+ AddressList(address.address(), address.port(), false), |
+ net_log_.net_log(), net_log_.source()); |
+ tcp_socket->AdoptSocket(result); |
+ socket->reset(tcp_socket); |
+ net_log_.EndEvent(NetLog::TYPE_TCP_ACCEPT, |
+ make_scoped_refptr(new NetLogStringParameter( |
+ "address", address.ToString()))); |
+ return OK; |
+} |
+ |
+void TCPServerSocketLibevent::Close() { |
+ if (socket_ != kInvalidSocket) { |
+ HANDLE_EINTR(close(socket_)); |
+ socket_ = kInvalidSocket; |
+ } |
+} |
+ |
+void TCPServerSocketLibevent::OnFileCanReadWithoutBlocking(int fd) { |
+ DCHECK(CalledOnValidThread()); |
+ |
+ int result = AcceptInternal(accept_socket_); |
+ if (result != ERR_IO_PENDING) { |
+ CompletionCallback* c = accept_callback_; |
+ accept_callback_ = NULL; |
+ accept_socket_ = NULL; |
+ c->Run(result); |
+ } |
+} |
+ |
+void TCPServerSocketLibevent::OnFileCanWriteWithoutBlocking(int fd) { |
+ NOTREACHED(); |
+} |
+ |
+} // namespace net |