| OLD | NEW |
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 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 | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "net/socket/tcp_listen_socket.h" | 5 #include "net/socket/tcp_listen_socket.h" |
| 6 | 6 |
| 7 #if defined(OS_WIN) | 7 #if defined(OS_WIN) |
| 8 // winsock2.h must be included first in order to ensure it is included before | 8 // winsock2.h must be included first in order to ensure it is included before |
| 9 // windows.h. | 9 // windows.h. |
| 10 #include <winsock2.h> | 10 #include <winsock2.h> |
| (...skipping 13 matching lines...) Expand all Loading... |
| 24 #include "net/base/net_util.h" | 24 #include "net/base/net_util.h" |
| 25 #include "net/base/winsock_init.h" | 25 #include "net/base/winsock_init.h" |
| 26 #include "net/socket/socket_descriptor.h" | 26 #include "net/socket/socket_descriptor.h" |
| 27 | 27 |
| 28 using std::string; | 28 using std::string; |
| 29 | 29 |
| 30 namespace net { | 30 namespace net { |
| 31 | 31 |
| 32 // static | 32 // static |
| 33 scoped_ptr<TCPListenSocket> TCPListenSocket::CreateAndListen( | 33 scoped_ptr<TCPListenSocket> TCPListenSocket::CreateAndListen( |
| 34 const string& ip, int port, StreamListenSocket::Delegate* del) { | 34 const string& ip, |
| 35 uint16 port, |
| 36 StreamListenSocket::Delegate* del) { |
| 35 SocketDescriptor s = CreateAndBind(ip, port); | 37 SocketDescriptor s = CreateAndBind(ip, port); |
| 36 if (s == kInvalidSocket) | 38 if (s == kInvalidSocket) |
| 37 return scoped_ptr<TCPListenSocket>(); | 39 return scoped_ptr<TCPListenSocket>(); |
| 38 scoped_ptr<TCPListenSocket> sock(new TCPListenSocket(s, del)); | 40 scoped_ptr<TCPListenSocket> sock(new TCPListenSocket(s, del)); |
| 39 sock->Listen(); | 41 sock->Listen(); |
| 40 return sock.Pass(); | 42 return sock.Pass(); |
| 41 } | 43 } |
| 42 | 44 |
| 43 TCPListenSocket::TCPListenSocket(SocketDescriptor s, | 45 TCPListenSocket::TCPListenSocket(SocketDescriptor s, |
| 44 StreamListenSocket::Delegate* del) | 46 StreamListenSocket::Delegate* del) |
| 45 : StreamListenSocket(s, del) { | 47 : StreamListenSocket(s, del) { |
| 46 } | 48 } |
| 47 | 49 |
| 48 TCPListenSocket::~TCPListenSocket() {} | 50 TCPListenSocket::~TCPListenSocket() {} |
| 49 | 51 |
| 50 SocketDescriptor TCPListenSocket::CreateAndBind(const string& ip, int port) { | 52 SocketDescriptor TCPListenSocket::CreateAndBind(const string& ip, uint16 port) { |
| 51 SocketDescriptor s = CreatePlatformSocket(AF_INET, SOCK_STREAM, IPPROTO_TCP); | 53 SocketDescriptor s = CreatePlatformSocket(AF_INET, SOCK_STREAM, IPPROTO_TCP); |
| 52 if (s != kInvalidSocket) { | 54 if (s != kInvalidSocket) { |
| 53 #if defined(OS_POSIX) | 55 #if defined(OS_POSIX) |
| 54 // Allow rapid reuse. | 56 // Allow rapid reuse. |
| 55 static const int kOn = 1; | 57 static const int kOn = 1; |
| 56 setsockopt(s, SOL_SOCKET, SO_REUSEADDR, &kOn, sizeof(kOn)); | 58 setsockopt(s, SOL_SOCKET, SO_REUSEADDR, &kOn, sizeof(kOn)); |
| 57 #endif | 59 #endif |
| 58 sockaddr_in addr; | 60 sockaddr_in addr; |
| 59 memset(&addr, 0, sizeof(addr)); | 61 memset(&addr, 0, sizeof(addr)); |
| 60 addr.sin_family = AF_INET; | 62 addr.sin_family = AF_INET; |
| 61 addr.sin_addr.s_addr = inet_addr(ip.c_str()); | 63 addr.sin_addr.s_addr = inet_addr(ip.c_str()); |
| 62 addr.sin_port = base::HostToNet16(port); | 64 addr.sin_port = base::HostToNet16(port); |
| 63 if (bind(s, reinterpret_cast<sockaddr*>(&addr), sizeof(addr))) { | 65 if (bind(s, reinterpret_cast<sockaddr*>(&addr), sizeof(addr))) { |
| 64 #if defined(OS_WIN) | 66 #if defined(OS_WIN) |
| 65 closesocket(s); | 67 closesocket(s); |
| 66 #elif defined(OS_POSIX) | 68 #elif defined(OS_POSIX) |
| 67 close(s); | 69 close(s); |
| 68 #endif | 70 #endif |
| 69 LOG(ERROR) << "Could not bind socket to " << ip << ":" << port; | 71 LOG(ERROR) << "Could not bind socket to " << ip << ":" << port; |
| 70 s = kInvalidSocket; | 72 s = kInvalidSocket; |
| 71 } | 73 } |
| 72 } | 74 } |
| 73 return s; | 75 return s; |
| 74 } | 76 } |
| 75 | 77 |
| 76 SocketDescriptor TCPListenSocket::CreateAndBindAnyPort(const string& ip, | 78 SocketDescriptor TCPListenSocket::CreateAndBindAnyPort(const string& ip, |
| 77 int* port) { | 79 uint16* port) { |
| 78 SocketDescriptor s = CreateAndBind(ip, 0); | 80 SocketDescriptor s = CreateAndBind(ip, 0); |
| 79 if (s == kInvalidSocket) | 81 if (s == kInvalidSocket) |
| 80 return kInvalidSocket; | 82 return kInvalidSocket; |
| 81 sockaddr_in addr; | 83 sockaddr_in addr; |
| 82 socklen_t addr_size = sizeof(addr); | 84 socklen_t addr_size = sizeof(addr); |
| 83 bool failed = getsockname(s, reinterpret_cast<struct sockaddr*>(&addr), | 85 bool failed = getsockname(s, reinterpret_cast<struct sockaddr*>(&addr), |
| 84 &addr_size) != 0; | 86 &addr_size) != 0; |
| 85 if (addr_size != sizeof(addr)) | 87 if (addr_size != sizeof(addr)) |
| 86 failed = true; | 88 failed = true; |
| 87 if (failed) { | 89 if (failed) { |
| (...skipping 15 matching lines...) Expand all Loading... |
| 103 return; | 105 return; |
| 104 scoped_ptr<TCPListenSocket> sock( | 106 scoped_ptr<TCPListenSocket> sock( |
| 105 new TCPListenSocket(conn, socket_delegate_)); | 107 new TCPListenSocket(conn, socket_delegate_)); |
| 106 // It's up to the delegate to AddRef if it wants to keep it around. | 108 // It's up to the delegate to AddRef if it wants to keep it around. |
| 107 #if defined(OS_POSIX) | 109 #if defined(OS_POSIX) |
| 108 sock->WatchSocket(WAITING_READ); | 110 sock->WatchSocket(WAITING_READ); |
| 109 #endif | 111 #endif |
| 110 socket_delegate_->DidAccept(this, sock.Pass()); | 112 socket_delegate_->DidAccept(this, sock.Pass()); |
| 111 } | 113 } |
| 112 | 114 |
| 113 TCPListenSocketFactory::TCPListenSocketFactory(const string& ip, int port) | |
| 114 : ip_(ip), | |
| 115 port_(port) { | |
| 116 } | |
| 117 | |
| 118 TCPListenSocketFactory::~TCPListenSocketFactory() {} | |
| 119 | |
| 120 scoped_ptr<StreamListenSocket> TCPListenSocketFactory::CreateAndListen( | |
| 121 StreamListenSocket::Delegate* delegate) const { | |
| 122 return TCPListenSocket::CreateAndListen(ip_, port_, delegate); | |
| 123 } | |
| 124 | |
| 125 } // namespace net | 115 } // namespace net |
| OLD | NEW |