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 |