OLD | NEW |
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright 2014 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 #ifndef NET_UDP_UDP_SOCKET_WIN_H_ | 5 #ifndef NET_UDP_UDP_SOCKET_NACL_H_ |
6 #define NET_UDP_UDP_SOCKET_WIN_H_ | 6 #define NET_UDP_UDP_SOCKET_NACL_H_ |
7 | |
8 #include <winsock2.h> | |
9 | 7 |
10 #include "base/memory/ref_counted.h" | 8 #include "base/memory/ref_counted.h" |
11 #include "base/memory/scoped_ptr.h" | 9 #include "base/memory/scoped_ptr.h" |
| 10 #include "base/message_loop/message_loop.h" |
12 #include "base/threading/non_thread_safe.h" | 11 #include "base/threading/non_thread_safe.h" |
13 #include "base/win/object_watcher.h" | |
14 #include "net/base/completion_callback.h" | 12 #include "net/base/completion_callback.h" |
| 13 #include "net/base/io_buffer.h" |
| 14 #include "net/base/ip_endpoint.h" |
15 #include "net/base/net_export.h" | 15 #include "net/base/net_export.h" |
| 16 #include "net/base/net_log.h" |
16 #include "net/base/rand_callback.h" | 17 #include "net/base/rand_callback.h" |
17 #include "net/base/ip_endpoint.h" | 18 #include "net/socket/socket_descriptor.h" |
18 #include "net/base/io_buffer.h" | |
19 #include "net/base/net_log.h" | |
20 #include "net/udp/datagram_socket.h" | 19 #include "net/udp/datagram_socket.h" |
21 | 20 |
22 namespace net { | 21 namespace net { |
23 | 22 |
24 class NET_EXPORT UDPSocketWin : NON_EXPORTED_BASE(public base::NonThreadSafe) { | 23 class NET_EXPORT UDPSocketNacl : public base::NonThreadSafe { |
25 public: | 24 public: |
26 UDPSocketWin(DatagramSocket::BindType bind_type, | 25 UDPSocketNacl(DatagramSocket::BindType bind_type, |
27 const RandIntCallback& rand_int_cb, | 26 const RandIntCallback& rand_int_cb, |
28 net::NetLog* net_log, | 27 net::NetLog* net_log, |
29 const net::NetLog::Source& source); | 28 const net::NetLog::Source& source); |
30 virtual ~UDPSocketWin(); | 29 virtual ~UDPSocketNacl(); |
31 | 30 |
32 // Connect the socket to connect with a certain |address|. | 31 // Connect the socket to connect with a certain |address|. |
33 // Returns a net error code. | 32 // Returns a net error code. |
34 int Connect(const IPEndPoint& address); | 33 int Connect(const IPEndPoint& address); |
35 | 34 |
36 // Bind the address/port for this socket to |address|. This is generally | 35 // Bind the address/port for this socket to |address|. This is generally |
37 // only used on a server. | 36 // only used on a server. |
38 // Returns a net error code. | 37 // Returns a net error code. |
39 int Bind(const IPEndPoint& address); | 38 int Bind(const IPEndPoint& address); |
40 | 39 |
(...skipping 53 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
94 const IPEndPoint& address, | 93 const IPEndPoint& address, |
95 const CompletionCallback& callback); | 94 const CompletionCallback& callback); |
96 | 95 |
97 // Set the receive buffer size (in bytes) for the socket. | 96 // Set the receive buffer size (in bytes) for the socket. |
98 bool SetReceiveBufferSize(int32 size); | 97 bool SetReceiveBufferSize(int32 size); |
99 | 98 |
100 // Set the send buffer size (in bytes) for the socket. | 99 // Set the send buffer size (in bytes) for the socket. |
101 bool SetSendBufferSize(int32 size); | 100 bool SetSendBufferSize(int32 size); |
102 | 101 |
103 // Returns true if the socket is already connected or bound. | 102 // Returns true if the socket is already connected or bound. |
104 bool is_connected() const { return socket_ != INVALID_SOCKET; } | 103 bool is_connected() const { return false; } |
105 | 104 |
106 const BoundNetLog& NetLog() const { return net_log_; } | 105 const BoundNetLog& NetLog() const { return net_log_; } |
107 | 106 |
108 // Sets corresponding flags in |socket_options_| to allow the socket | 107 // Sets corresponding flags in |socket_options_| to allow the socket |
109 // to share the local address to which the socket will be bound with | 108 // to share the local address to which the socket will be bound with |
110 // other processes. Should be called before Bind(). | 109 // other processes. Should be called before Bind(). |
111 void AllowAddressReuse(); | 110 void AllowAddressReuse(); |
112 | 111 |
113 // Sets corresponding flags in |socket_options_| to allow sending | 112 // Sets corresponding flags in |socket_options_| to allow sending |
114 // and receiving packets to and from broadcast addresses. Should be | 113 // and receiving packets to and from broadcast addresses. Should be |
(...skipping 18 matching lines...) Expand all Loading... |
133 // Set interface to use for multicast. If |interface_index| set to 0, default | 132 // Set interface to use for multicast. If |interface_index| set to 0, default |
134 // interface is used. | 133 // interface is used. |
135 // Should be called before Bind(). | 134 // Should be called before Bind(). |
136 // Returns a network error code. | 135 // Returns a network error code. |
137 int SetMulticastInterface(uint32 interface_index); | 136 int SetMulticastInterface(uint32 interface_index); |
138 | 137 |
139 // Set the time-to-live option for UDP packets sent to the multicast | 138 // Set the time-to-live option for UDP packets sent to the multicast |
140 // group address. The default value of this option is 1. | 139 // group address. The default value of this option is 1. |
141 // Cannot be negative or more than 255. | 140 // Cannot be negative or more than 255. |
142 // Should be called before Bind(). | 141 // Should be called before Bind(). |
| 142 // Return a network error code. |
143 int SetMulticastTimeToLive(int time_to_live); | 143 int SetMulticastTimeToLive(int time_to_live); |
144 | 144 |
145 // Set the loopback flag for UDP socket. If this flag is true, the host | 145 // Set the loopback flag for UDP socket. If this flag is true, the host |
146 // will receive packets sent to the joined group from itself. | 146 // will receive packets sent to the joined group from itself. |
147 // The default value of this option is true. | 147 // The default value of this option is true. |
148 // Should be called before Bind(). | 148 // Should be called before Bind(). |
| 149 // Return a network error code. |
149 // | 150 // |
150 // Note: the behavior of |SetMulticastLoopbackMode| is slightly | 151 // Note: the behavior of |SetMulticastLoopbackMode| is slightly |
151 // different between Windows and Unix-like systems. The inconsistency only | 152 // different between Windows and Unix-like systems. The inconsistency only |
152 // happens when there are more than one applications on the same host | 153 // happens when there are more than one applications on the same host |
153 // joined to the same multicast group while having different settings on | 154 // joined to the same multicast group while having different settings on |
154 // multicast loopback mode. On Windows, the applications with loopback off | 155 // multicast loopback mode. On Windows, the applications with loopback off |
155 // will not RECEIVE the loopback packets; while on Unix-like systems, the | 156 // will not RECEIVE the loopback packets; while on Unix-like systems, the |
156 // applications with loopback off will not SEND the loopback packets to | 157 // applications with loopback off will not SEND the loopback packets to |
157 // other applications on the same host. See MSDN: http://goo.gl/6vqbj | 158 // other applications on the same host. See MSDN: http://goo.gl/6vqbj |
158 int SetMulticastLoopbackMode(bool loopback); | 159 int SetMulticastLoopbackMode(bool loopback); |
159 | 160 |
160 // Set the differentiated services flags on outgoing packets. May not | 161 // Set the differentiated services flags on outgoing packets. May not |
161 // do anything on some platforms. | 162 // do anything on some platforms. |
| 163 // Return a network error code. |
162 int SetDiffServCodePoint(DiffServCodePoint dscp); | 164 int SetDiffServCodePoint(DiffServCodePoint dscp); |
163 | 165 |
164 private: | 166 private: |
165 enum SocketOptions { | |
166 SOCKET_OPTION_REUSE_ADDRESS = 1 << 0, | |
167 SOCKET_OPTION_BROADCAST = 1 << 1, | |
168 SOCKET_OPTION_MULTICAST_LOOP = 1 << 2 | |
169 }; | |
170 | |
171 class Core; | |
172 | |
173 void DoReadCallback(int rv); | |
174 void DoWriteCallback(int rv); | |
175 void DidCompleteRead(); | |
176 void DidCompleteWrite(); | |
177 | |
178 // Handles stats and logging. |result| is the number of bytes transferred, on | |
179 // success, or the net error code on failure. LogRead retrieves the address | |
180 // from |recv_addr_storage_|, while LogWrite takes it as an optional argument. | |
181 void LogRead(int result, const char* bytes) const; | |
182 void LogWrite(int result, const char* bytes, const IPEndPoint* address) const; | |
183 | |
184 // Returns the OS error code (or 0 on success). | |
185 int CreateSocket(int addr_family); | |
186 | |
187 // Same as SendTo(), except that address is passed by pointer | |
188 // instead of by reference. It is called from Write() with |address| | |
189 // set to NULL. | |
190 int SendToOrWrite(IOBuffer* buf, | |
191 int buf_len, | |
192 const IPEndPoint* address, | |
193 const CompletionCallback& callback); | |
194 | |
195 int InternalConnect(const IPEndPoint& address); | |
196 int InternalRecvFrom(IOBuffer* buf, int buf_len, IPEndPoint* address); | |
197 int InternalSendTo(IOBuffer* buf, int buf_len, const IPEndPoint* address); | |
198 | |
199 // Applies |socket_options_| to |socket_|. Should be called before | |
200 // Bind(). | |
201 int SetSocketOptions(); | |
202 int DoBind(const IPEndPoint& address); | |
203 // Binds to a random port on |address|. | |
204 int RandomBind(const IPAddressNumber& address); | |
205 | |
206 // Attempts to convert the data in |recv_addr_storage_| and |recv_addr_len_| | |
207 // to an IPEndPoint and writes it to |address|. Returns true on success. | |
208 bool ReceiveAddressToIPEndpoint(IPEndPoint* address) const; | |
209 | |
210 SOCKET socket_; | |
211 int addr_family_; | |
212 | |
213 // Bitwise-or'd combination of SocketOptions. Specifies the set of | |
214 // options that should be applied to |socket_| before Bind(). | |
215 int socket_options_; | |
216 | |
217 // Multicast interface. | |
218 uint32 multicast_interface_; | |
219 | |
220 // Multicast socket options cached for SetSocketOption. | |
221 // Cannot be used after Bind(). | |
222 int multicast_time_to_live_; | |
223 | |
224 // How to do source port binding, used only when UDPSocket is part of | |
225 // UDPClientSocket, since UDPServerSocket provides Bind. | |
226 DatagramSocket::BindType bind_type_; | |
227 | |
228 // PRNG function for generating port numbers. | |
229 RandIntCallback rand_int_cb_; | |
230 | |
231 // These are mutable since they're just cached copies to make | |
232 // GetPeerAddress/GetLocalAddress smarter. | |
233 mutable scoped_ptr<IPEndPoint> local_address_; | |
234 mutable scoped_ptr<IPEndPoint> remote_address_; | |
235 | |
236 // The core of the socket that can live longer than the socket itself. We pass | |
237 // resources to the Windows async IO functions and we have to make sure that | |
238 // they are not destroyed while the OS still references them. | |
239 scoped_refptr<Core> core_; | |
240 | |
241 IPEndPoint* recv_from_address_; | |
242 | |
243 // Cached copy of the current address we're sending to, if any. Used for | |
244 // logging. | |
245 scoped_ptr<IPEndPoint> send_to_address_; | |
246 | |
247 // External callback; called when read is complete. | |
248 CompletionCallback read_callback_; | |
249 | |
250 // External callback; called when write is complete. | |
251 CompletionCallback write_callback_; | |
252 | |
253 BoundNetLog net_log_; | 167 BoundNetLog net_log_; |
254 | 168 |
255 DISALLOW_COPY_AND_ASSIGN(UDPSocketWin); | 169 DISALLOW_COPY_AND_ASSIGN(UDPSocketNacl); |
256 }; | 170 }; |
257 | 171 |
258 } // namespace net | 172 } // namespace net |
259 | 173 |
260 #endif // NET_UDP_UDP_SOCKET_WIN_H_ | 174 #endif // NET_UDP_UDP_SOCKET_NACL_H_ |
OLD | NEW |