15821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Copyright (c) 2012 The Chromium Authors. All rights reserved. 25821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Use of this source code is governed by a BSD-style license that can be 35821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// found in the LICENSE file. 45821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)#include "net/socket/stream_listen_socket.h" 65821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 75821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#if defined(OS_WIN) 85821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// winsock2.h must be included first in order to ensure it is included before 95821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// windows.h. 105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <winsock2.h> 115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#elif defined(OS_POSIX) 12c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)#include <arpa/inet.h> 135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <errno.h> 145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <netinet/in.h> 15c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)#include <sys/socket.h> 16c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)#include <sys/types.h> 175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "net/base/net_errors.h" 185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#endif 195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/logging.h" 215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/memory/ref_counted.h" 225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/memory/scoped_ptr.h" 232a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#include "base/posix/eintr_wrapper.h" 245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/sys_byteorder.h" 255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/threading/platform_thread.h" 265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "build/build_config.h" 275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "net/base/ip_endpoint.h" 285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "net/base/net_errors.h" 295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "net/base/net_util.h" 305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)using std::string; 325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#if defined(OS_WIN) 345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)typedef int socklen_t; 355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#endif // defined(OS_WIN) 365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace net { 385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace { 405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)const int kReadBufSize = 4096; 425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} // namespace 445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#if defined(OS_WIN) 465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)const SocketDescriptor StreamListenSocket::kInvalidSocket = INVALID_SOCKET; 475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)const int StreamListenSocket::kSocketError = SOCKET_ERROR; 485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#elif defined(OS_POSIX) 495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)const SocketDescriptor StreamListenSocket::kInvalidSocket = -1; 505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)const int StreamListenSocket::kSocketError = -1; 515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#endif 525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)StreamListenSocket::StreamListenSocket(SocketDescriptor s, 545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) StreamListenSocket::Delegate* del) 555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) : socket_delegate_(del), 565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) socket_(s), 575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) reads_paused_(false), 585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) has_pending_reads_(false) { 595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#if defined(OS_WIN) 605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) socket_event_ = WSACreateEvent(); 615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // TODO(ibrar): error handling in case of socket_event_ == WSA_INVALID_EVENT. 625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) WatchSocket(NOT_WAITING); 635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#elif defined(OS_POSIX) 645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) wait_state_ = NOT_WAITING; 655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#endif 665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)StreamListenSocket::~StreamListenSocket() { 695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#if defined(OS_WIN) 705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (socket_event_) { 715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) WSACloseEvent(socket_event_); 725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) socket_event_ = WSA_INVALID_EVENT; 735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#endif 755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) CloseSocket(socket_); 765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void StreamListenSocket::Send(const char* bytes, int len, 795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) bool append_linefeed) { 805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SendInternal(bytes, len); 815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (append_linefeed) 825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SendInternal("\r\n", 2); 835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void StreamListenSocket::Send(const string& str, bool append_linefeed) { 865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) Send(str.data(), static_cast<int>(str.length()), append_linefeed); 875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)int StreamListenSocket::GetLocalAddress(IPEndPoint* address) { 905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SockaddrStorage storage; 915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (getsockname(socket_, storage.addr, &storage.addr_len)) { 925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#if defined(OS_WIN) 935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int err = WSAGetLastError(); 945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#else 955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int err = errno; 965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#endif 975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return MapSystemError(err); 985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!address->FromSockAddr(storage.addr, storage.addr_len)) 1005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return ERR_FAILED; 1015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return OK; 1025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)SocketDescriptor StreamListenSocket::AcceptSocket() { 1055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SocketDescriptor conn = HANDLE_EINTR(accept(socket_, NULL, NULL)); 1065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (conn == kInvalidSocket) 1075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) LOG(ERROR) << "Error accepting connection."; 1085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) else 1095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SetNonBlocking(conn); 1105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return conn; 1115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void StreamListenSocket::SendInternal(const char* bytes, int len) { 1145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) char* send_buf = const_cast<char *>(bytes); 1155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int len_left = len; 1165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) while (true) { 1175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int sent = HANDLE_EINTR(send(socket_, send_buf, len_left, 0)); 1185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (sent == len_left) { // A shortcut to avoid extraneous checks. 1195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) break; 1205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (sent == kSocketError) { 1225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#if defined(OS_WIN) 1235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (WSAGetLastError() != WSAEWOULDBLOCK) { 1245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) LOG(ERROR) << "send failed: WSAGetLastError()==" << WSAGetLastError(); 1255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#elif defined(OS_POSIX) 1265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (errno != EWOULDBLOCK && errno != EAGAIN) { 1275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) LOG(ERROR) << "send failed: errno==" << errno; 1285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#endif 1295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) break; 1305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Otherwise we would block, and now we have to wait for a retry. 1325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Fall through to PlatformThread::YieldCurrentThread() 1335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } else { 1345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // sent != len_left according to the shortcut above. 1355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Shift the buffer start and send the remainder after a short while. 1365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) send_buf += sent; 1375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) len_left -= sent; 1385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::PlatformThread::YieldCurrentThread(); 1405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void StreamListenSocket::Listen() { 1445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int backlog = 10; // TODO(erikkay): maybe don't allow any backlog? 1455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (listen(socket_, backlog) == -1) { 1465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // TODO(erikkay): error handling. 1475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) LOG(ERROR) << "Could not listen on socket."; 1485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return; 1495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#if defined(OS_POSIX) 1515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) WatchSocket(WAITING_ACCEPT); 1525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#endif 1535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void StreamListenSocket::Read() { 1565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) char buf[kReadBufSize + 1]; // +1 for null termination. 1575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int len; 1585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) do { 1595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) len = HANDLE_EINTR(recv(socket_, buf, kReadBufSize, 0)); 1605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (len == kSocketError) { 1615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#if defined(OS_WIN) 1625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int err = WSAGetLastError(); 1635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (err == WSAEWOULDBLOCK) { 1645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#elif defined(OS_POSIX) 1655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (errno == EWOULDBLOCK || errno == EAGAIN) { 1665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#endif 1675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) break; 1685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } else { 1695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // TODO(ibrar): some error handling required here. 1705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) break; 1715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } else if (len == 0) { 1735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // In Windows, Close() is called by OnObjectSignaled. In POSIX, we need 1745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // to call it here. 1755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#if defined(OS_POSIX) 1765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) Close(); 1775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#endif 1785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } else { 1795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // TODO(ibrar): maybe change DidRead to take a length instead. 1805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DCHECK_GT(len, 0); 1815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DCHECK_LE(len, kReadBufSize); 1825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) buf[len] = 0; // Already create a buffer with +1 length. 1835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) socket_delegate_->DidRead(this, buf, len); 1845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } while (len == kReadBufSize); 1865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void StreamListenSocket::Close() { 1895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#if defined(OS_POSIX) 1905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (wait_state_ == NOT_WAITING) 1915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return; 1925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) wait_state_ = NOT_WAITING; 1935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#endif 1945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) UnwatchSocket(); 1955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) socket_delegate_->DidClose(this); 1965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void StreamListenSocket::CloseSocket(SocketDescriptor s) { 1995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (s && s != kInvalidSocket) { 2005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) UnwatchSocket(); 2015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#if defined(OS_WIN) 2025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) closesocket(s); 2035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#elif defined(OS_POSIX) 2045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) close(s); 2055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#endif 2065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 2075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void StreamListenSocket::WatchSocket(WaitState state) { 2105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#if defined(OS_WIN) 2115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) WSAEventSelect(socket_, socket_event_, FD_ACCEPT | FD_CLOSE | FD_READ); 2125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) watcher_.StartWatching(socket_event_, this); 2135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#elif defined(OS_POSIX) 2145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Implicitly calls StartWatchingFileDescriptor(). 21590dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) base::MessageLoopForIO::current()->WatchFileDescriptor( 21690dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) socket_, true, base::MessageLoopForIO::WATCH_READ, &watcher_, this); 2175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) wait_state_ = state; 2185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#endif 2195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void StreamListenSocket::UnwatchSocket() { 2225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#if defined(OS_WIN) 2235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) watcher_.StopWatching(); 2245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#elif defined(OS_POSIX) 2255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) watcher_.StopWatchingFileDescriptor(); 2265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#endif 2275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// TODO(ibrar): We can add these functions into OS dependent files. 2305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#if defined(OS_WIN) 2315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// MessageLoop watcher callback. 2325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void StreamListenSocket::OnObjectSignaled(HANDLE object) { 2335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) WSANETWORKEVENTS ev; 2345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (kSocketError == WSAEnumNetworkEvents(socket_, socket_event_, &ev)) { 2355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // TODO 2365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return; 2375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 2385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 239a36e5920737c6adbddd3e43b760e5de8431db6e0Torne (Richard Coles) if (ev.lNetworkEvents & FD_CLOSE) { 240a36e5920737c6adbddd3e43b760e5de8431db6e0Torne (Richard Coles) Close(); 241a36e5920737c6adbddd3e43b760e5de8431db6e0Torne (Richard Coles) // Close might have deleted this object. We should return immediately. 242a36e5920737c6adbddd3e43b760e5de8431db6e0Torne (Richard Coles) return; 243a36e5920737c6adbddd3e43b760e5de8431db6e0Torne (Richard Coles) } 244a36e5920737c6adbddd3e43b760e5de8431db6e0Torne (Richard Coles) 2455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // The object was reset by WSAEnumNetworkEvents. Watch for the next signal. 2465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) watcher_.StartWatching(object, this); 2475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (ev.lNetworkEvents == 0) { 2495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Occasionally the event is set even though there is no new data. 2505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // The net seems to think that this is ignorable. 2515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return; 2525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 2535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (ev.lNetworkEvents & FD_ACCEPT) { 2545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) Accept(); 2555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 2565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (ev.lNetworkEvents & FD_READ) { 2575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (reads_paused_) { 2585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) has_pending_reads_ = true; 2595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } else { 2605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) Read(); 261a36e5920737c6adbddd3e43b760e5de8431db6e0Torne (Richard Coles) // Read() might call Close() internally and 'this' can be invalid here 262a36e5920737c6adbddd3e43b760e5de8431db6e0Torne (Richard Coles) return; 2635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 2645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 2655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#elif defined(OS_POSIX) 2675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void StreamListenSocket::OnFileCanReadWithoutBlocking(int fd) { 2685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) switch (wait_state_) { 2695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) case WAITING_ACCEPT: 2705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) Accept(); 2715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) break; 2725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) case WAITING_READ: 2735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (reads_paused_) { 2745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) has_pending_reads_ = true; 2755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } else { 2765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) Read(); 2775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 2785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) break; 2795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) default: 2805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Close() is called by Read() in the Linux case. 2815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) NOTREACHED(); 2825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) break; 2835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 2845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void StreamListenSocket::OnFileCanWriteWithoutBlocking(int fd) { 2875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // MessagePumpLibevent callback, we don't listen for write events 2885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // so we shouldn't ever reach here. 2895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) NOTREACHED(); 2905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#endif 2935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void StreamListenSocket::PauseReads() { 2955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DCHECK(!reads_paused_); 2965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) reads_paused_ = true; 2975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void StreamListenSocket::ResumeReads() { 3005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DCHECK(reads_paused_); 3015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) reads_paused_ = false; 3025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (has_pending_reads_) { 3035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) has_pending_reads_ = false; 3045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) Read(); 3055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 3065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 3075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} // namespace net 309