net/socket/tcp_listen_socket.cc

   1 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "net/socket/tcp_listen_socket.h"
   6
   7 #if defined(OS_WIN)
   8 // winsock2.h must be included first in order to ensure it is included before
   9 // windows.h.
  10 #include <winsock2.h>
  11 #elif defined(OS_POSIX)
  12 #include <arpa/inet.h>
  13 #include <errno.h>
  14 #include <netinet/in.h>
  15 #include <sys/socket.h>
  16 #include <sys/types.h>
  17 #include "net/base/net_errors.h"
  18 #endif
  19
  20 #include "base/logging.h"
  21 #include "base/sys_byteorder.h"
  22 #include "base/threading/platform_thread.h"
  23 #include "build/build_config.h"
  24 #include "net/base/net_util.h"
  25 #include "net/base/winsock_init.h"
  26 #include "net/socket/socket_descriptor.h"
  27
  28 using std::string;
  29
  30 namespace net {
  31
  32 // static
  33 scoped_ptr<TCPListenSocket> TCPListenSocket::CreateAndListen(
  34     const string& ip,
  35     uint16 port,
  36     StreamListenSocket::Delegate* del) {
  37   SocketDescriptor s = CreateAndBind(ip, port);
  38   if (s == kInvalidSocket)
  39     return scoped_ptr<TCPListenSocket>();
  40   scoped_ptr<TCPListenSocket> sock(new TCPListenSocket(s, del));
  41   sock->Listen();
  42   return sock.Pass();
  43 }
  44
  45 TCPListenSocket::TCPListenSocket(SocketDescriptor s,
  46                                  StreamListenSocket::Delegate* del)
  47     : StreamListenSocket(s, del) {
  48 }
  49
  50 TCPListenSocket::~TCPListenSocket() {}
  51
  52 SocketDescriptor TCPListenSocket::CreateAndBind(const string& ip, uint16 port) {
  53   SocketDescriptor s = CreatePlatformSocket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
  54   if (s != kInvalidSocket) {
  55 #if defined(OS_POSIX)
  56     // Allow rapid reuse.
  57     static const int kOn = 1;
  58     setsockopt(s, SOL_SOCKET, SO_REUSEADDR, &kOn, sizeof(kOn));
  59 #endif
  60     sockaddr_in addr;
  61     memset(&addr, 0, sizeof(addr));
  62     addr.sin_family = AF_INET;
  63     addr.sin_addr.s_addr = inet_addr(ip.c_str());
  64     addr.sin_port = base::HostToNet16(port);
  65     if (bind(s, reinterpret_cast<sockaddr*>(&addr), sizeof(addr))) {
  66 #if defined(OS_WIN)
  67       closesocket(s);
  68 #elif defined(OS_POSIX)
  69       close(s);
  70 #endif
  71       LOG(ERROR) << "Could not bind socket to " << ip << ":" << port;
  72       s = kInvalidSocket;
  73     }
  74   }
  75   return s;
  76 }
  77
  78 SocketDescriptor TCPListenSocket::CreateAndBindAnyPort(const string& ip,
  79                                                        uint16* port) {
  80   SocketDescriptor s = CreateAndBind(ip, 0);
  81   if (s == kInvalidSocket)
  82     return kInvalidSocket;
  83   sockaddr_in addr;
  84   socklen_t addr_size = sizeof(addr);
  85   bool failed = getsockname(s, reinterpret_cast<struct sockaddr*>(&addr),
  86                             &addr_size) != 0;
  87   if (addr_size != sizeof(addr))
  88     failed = true;
  89   if (failed) {
  90     LOG(ERROR) << "Could not determine bound port, getsockname() failed";
  91 #if defined(OS_WIN)
  92     closesocket(s);
  93 #elif defined(OS_POSIX)
  94     close(s);
  95 #endif
  96     return kInvalidSocket;
  97   }
  98   *port = base::NetToHost16(addr.sin_port);
  99   return s;
 100 }
 101
 102 void TCPListenSocket::Accept() {
 103   SocketDescriptor conn = AcceptSocket();
 104   if (conn == kInvalidSocket)
 105     return;
 106   scoped_ptr<TCPListenSocket> sock(
 107       new TCPListenSocket(conn, socket_delegate_));
 108   // It's up to the delegate to AddRef if it wants to keep it around.
 109 #if defined(OS_POSIX)
 110   sock->WatchSocket(WAITING_READ);
 111 #endif
 112   socket_delegate_->DidAccept(this, sock.Pass());
 113 }
 114
 115 }  // namespace net