1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
|
// Copyright 2011 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#ifndef NET_SOCKET_UDP_CLIENT_SOCKET_H_
#define NET_SOCKET_UDP_CLIENT_SOCKET_H_
#include <stdint.h>
#include "net/base/net_export.h"
#include "net/socket/datagram_client_socket.h"
#include "net/socket/socket_descriptor.h"
#include "net/socket/udp_socket.h"
#include "net/traffic_annotation/network_traffic_annotation.h"
namespace net {
class NetLog;
struct NetLogSource;
// A client socket that uses UDP as the transport layer.
class NET_EXPORT_PRIVATE UDPClientSocket : public DatagramClientSocket {
public:
// If `network` is specified, the socket will be bound to it. All data traffic
// on the socket will be sent and received via `network`. Communication using
// this socket will fail if `network` disconnects.
UDPClientSocket(
DatagramSocket::BindType bind_type,
net::NetLog* net_log,
const net::NetLogSource& source,
handles::NetworkHandle network = handles::kInvalidNetworkHandle);
UDPClientSocket(
DatagramSocket::BindType bind_type,
NetLogWithSource source_net_log,
handles::NetworkHandle network = handles::kInvalidNetworkHandle);
UDPClientSocket(const UDPClientSocket&) = delete;
UDPClientSocket& operator=(const UDPClientSocket&) = delete;
~UDPClientSocket() override;
// DatagramClientSocket implementation.
int Connect(const IPEndPoint& address) override;
int ConnectUsingNetwork(handles::NetworkHandle network,
const IPEndPoint& address) override;
int ConnectUsingDefaultNetwork(const IPEndPoint& address) override;
int ConnectAsync(const IPEndPoint& address,
CompletionOnceCallback callback) override;
int ConnectUsingNetworkAsync(handles::NetworkHandle network,
const IPEndPoint& address,
CompletionOnceCallback callback) override;
int ConnectUsingDefaultNetworkAsync(const IPEndPoint& address,
CompletionOnceCallback callback) override;
DscpAndEcn GetLastTos() const override;
handles::NetworkHandle GetBoundNetwork() const override;
void ApplySocketTag(const SocketTag& tag) override;
int Read(IOBuffer* buf,
int buf_len,
CompletionOnceCallback callback) override;
int Write(IOBuffer* buf,
int buf_len,
CompletionOnceCallback callback,
const NetworkTrafficAnnotationTag& traffic_annotation) override;
void Close() override;
int GetPeerAddress(IPEndPoint* address) const override;
int GetLocalAddress(IPEndPoint* address) const override;
// Switch to use non-blocking IO. Must be called right after construction and
// before other calls.
void UseNonBlockingIO() override;
int SetReceiveBufferSize(int32_t size) override;
int SetSendBufferSize(int32_t size) override;
int SetDoNotFragment() override;
int SetRecvTos() override;
int SetTos(DiffServCodePoint dscp, EcnCodePoint ecn) override;
void SetMsgConfirm(bool confirm) override;
const NetLogWithSource& NetLog() const override;
void EnableRecvOptimization() override;
int SetMulticastInterface(uint32_t interface_index) override;
void SetIOSNetworkServiceType(int ios_network_service_type) override;
void RegisterQuicConnectionClosePayload(base::span<uint8_t> payload) override;
void UnregisterQuicConnectionClosePayload() override;
// Takes ownership of an opened but unconnected and unbound `socket`. This
// method must be called after UseNonBlockingIO, otherwise the adopted socket
// will not have the non-blocking IO flag set.
int AdoptOpenedSocket(AddressFamily address_family, SocketDescriptor socket);
uint32_t get_multicast_interface_for_testing() {
return socket_.get_multicast_interface_for_testing();
}
#if !BUILDFLAG(IS_WIN)
bool get_msg_confirm_for_testing() {
return socket_.get_msg_confirm_for_testing();
}
bool get_recv_optimization_for_testing() {
return socket_.get_experimental_recv_optimization_enabled_for_testing();
}
#endif
#if BUILDFLAG(IS_WIN)
bool get_use_non_blocking_io_for_testing() {
return socket_.get_use_non_blocking_io_for_testing();
}
#endif
private:
NetLogWithSource net_log_;
UDPSocket socket_;
bool adopted_opened_socket_ = false;
bool connect_called_ = false;
// The network the socket is currently bound to.
handles::NetworkHandle network_ = handles::kInvalidNetworkHandle;
handles::NetworkHandle connect_using_network_;
};
} // namespace net
#endif // NET_SOCKET_UDP_CLIENT_SOCKET_H_
|
