1*d9f75844SAndroid Build Coastguard Worker /*
2*d9f75844SAndroid Build Coastguard Worker * Copyright 2004 The WebRTC Project Authors. All rights reserved.
3*d9f75844SAndroid Build Coastguard Worker *
4*d9f75844SAndroid Build Coastguard Worker * Use of this source code is governed by a BSD-style license
5*d9f75844SAndroid Build Coastguard Worker * that can be found in the LICENSE file in the root of the source
6*d9f75844SAndroid Build Coastguard Worker * tree. An additional intellectual property rights grant can be found
7*d9f75844SAndroid Build Coastguard Worker * in the file PATENTS. All contributing project authors may
8*d9f75844SAndroid Build Coastguard Worker * be found in the AUTHORS file in the root of the source tree.
9*d9f75844SAndroid Build Coastguard Worker */
10*d9f75844SAndroid Build Coastguard Worker #include "rtc_base/physical_socket_server.h"
11*d9f75844SAndroid Build Coastguard Worker
12*d9f75844SAndroid Build Coastguard Worker #include <cstdint>
13*d9f75844SAndroid Build Coastguard Worker
14*d9f75844SAndroid Build Coastguard Worker #if defined(_MSC_VER) && _MSC_VER < 1300
15*d9f75844SAndroid Build Coastguard Worker #pragma warning(disable : 4786)
16*d9f75844SAndroid Build Coastguard Worker #endif
17*d9f75844SAndroid Build Coastguard Worker
18*d9f75844SAndroid Build Coastguard Worker #ifdef MEMORY_SANITIZER
19*d9f75844SAndroid Build Coastguard Worker #include <sanitizer/msan_interface.h>
20*d9f75844SAndroid Build Coastguard Worker #endif
21*d9f75844SAndroid Build Coastguard Worker
22*d9f75844SAndroid Build Coastguard Worker #if defined(WEBRTC_POSIX)
23*d9f75844SAndroid Build Coastguard Worker #include <fcntl.h>
24*d9f75844SAndroid Build Coastguard Worker #include <string.h>
25*d9f75844SAndroid Build Coastguard Worker #if defined(WEBRTC_USE_EPOLL)
26*d9f75844SAndroid Build Coastguard Worker // "poll" will be used to wait for the signal dispatcher.
27*d9f75844SAndroid Build Coastguard Worker #include <poll.h>
28*d9f75844SAndroid Build Coastguard Worker #endif
29*d9f75844SAndroid Build Coastguard Worker #include <sys/ioctl.h>
30*d9f75844SAndroid Build Coastguard Worker #include <sys/select.h>
31*d9f75844SAndroid Build Coastguard Worker #include <sys/time.h>
32*d9f75844SAndroid Build Coastguard Worker #include <unistd.h>
33*d9f75844SAndroid Build Coastguard Worker #endif
34*d9f75844SAndroid Build Coastguard Worker
35*d9f75844SAndroid Build Coastguard Worker #if defined(WEBRTC_WIN)
36*d9f75844SAndroid Build Coastguard Worker #include <windows.h>
37*d9f75844SAndroid Build Coastguard Worker #include <winsock2.h>
38*d9f75844SAndroid Build Coastguard Worker #include <ws2tcpip.h>
39*d9f75844SAndroid Build Coastguard Worker #undef SetPort
40*d9f75844SAndroid Build Coastguard Worker #endif
41*d9f75844SAndroid Build Coastguard Worker
42*d9f75844SAndroid Build Coastguard Worker #include <errno.h>
43*d9f75844SAndroid Build Coastguard Worker
44*d9f75844SAndroid Build Coastguard Worker #include <algorithm>
45*d9f75844SAndroid Build Coastguard Worker #include <map>
46*d9f75844SAndroid Build Coastguard Worker
47*d9f75844SAndroid Build Coastguard Worker #include "rtc_base/arraysize.h"
48*d9f75844SAndroid Build Coastguard Worker #include "rtc_base/byte_order.h"
49*d9f75844SAndroid Build Coastguard Worker #include "rtc_base/checks.h"
50*d9f75844SAndroid Build Coastguard Worker #include "rtc_base/logging.h"
51*d9f75844SAndroid Build Coastguard Worker #include "rtc_base/network_monitor.h"
52*d9f75844SAndroid Build Coastguard Worker #include "rtc_base/null_socket_server.h"
53*d9f75844SAndroid Build Coastguard Worker #include "rtc_base/synchronization/mutex.h"
54*d9f75844SAndroid Build Coastguard Worker #include "rtc_base/time_utils.h"
55*d9f75844SAndroid Build Coastguard Worker #include "system_wrappers/include/field_trial.h"
56*d9f75844SAndroid Build Coastguard Worker
57*d9f75844SAndroid Build Coastguard Worker #if defined(WEBRTC_LINUX)
58*d9f75844SAndroid Build Coastguard Worker #include <linux/sockios.h>
59*d9f75844SAndroid Build Coastguard Worker #endif
60*d9f75844SAndroid Build Coastguard Worker
61*d9f75844SAndroid Build Coastguard Worker #if defined(WEBRTC_WIN)
62*d9f75844SAndroid Build Coastguard Worker #define LAST_SYSTEM_ERROR (::GetLastError())
63*d9f75844SAndroid Build Coastguard Worker #elif defined(__native_client__) && __native_client__
64*d9f75844SAndroid Build Coastguard Worker #define LAST_SYSTEM_ERROR (0)
65*d9f75844SAndroid Build Coastguard Worker #elif defined(WEBRTC_POSIX)
66*d9f75844SAndroid Build Coastguard Worker #define LAST_SYSTEM_ERROR (errno)
67*d9f75844SAndroid Build Coastguard Worker #endif // WEBRTC_WIN
68*d9f75844SAndroid Build Coastguard Worker
69*d9f75844SAndroid Build Coastguard Worker #if defined(WEBRTC_POSIX)
70*d9f75844SAndroid Build Coastguard Worker #include <netinet/tcp.h> // for TCP_NODELAY
71*d9f75844SAndroid Build Coastguard Worker #define IP_MTU 14 // Until this is integrated from linux/in.h to netinet/in.h
72*d9f75844SAndroid Build Coastguard Worker typedef void* SockOptArg;
73*d9f75844SAndroid Build Coastguard Worker
74*d9f75844SAndroid Build Coastguard Worker #endif // WEBRTC_POSIX
75*d9f75844SAndroid Build Coastguard Worker
76*d9f75844SAndroid Build Coastguard Worker #if defined(WEBRTC_POSIX) && !defined(WEBRTC_MAC) && !defined(__native_client__)
77*d9f75844SAndroid Build Coastguard Worker
GetSocketRecvTimestamp(int socket)78*d9f75844SAndroid Build Coastguard Worker int64_t GetSocketRecvTimestamp(int socket) {
79*d9f75844SAndroid Build Coastguard Worker struct timeval tv_ioctl;
80*d9f75844SAndroid Build Coastguard Worker int ret = ioctl(socket, SIOCGSTAMP, &tv_ioctl);
81*d9f75844SAndroid Build Coastguard Worker if (ret != 0)
82*d9f75844SAndroid Build Coastguard Worker return -1;
83*d9f75844SAndroid Build Coastguard Worker int64_t timestamp =
84*d9f75844SAndroid Build Coastguard Worker rtc::kNumMicrosecsPerSec * static_cast<int64_t>(tv_ioctl.tv_sec) +
85*d9f75844SAndroid Build Coastguard Worker static_cast<int64_t>(tv_ioctl.tv_usec);
86*d9f75844SAndroid Build Coastguard Worker return timestamp;
87*d9f75844SAndroid Build Coastguard Worker }
88*d9f75844SAndroid Build Coastguard Worker
89*d9f75844SAndroid Build Coastguard Worker #else
90*d9f75844SAndroid Build Coastguard Worker
GetSocketRecvTimestamp(int socket)91*d9f75844SAndroid Build Coastguard Worker int64_t GetSocketRecvTimestamp(int socket) {
92*d9f75844SAndroid Build Coastguard Worker return -1;
93*d9f75844SAndroid Build Coastguard Worker }
94*d9f75844SAndroid Build Coastguard Worker #endif
95*d9f75844SAndroid Build Coastguard Worker
96*d9f75844SAndroid Build Coastguard Worker #if defined(WEBRTC_WIN)
97*d9f75844SAndroid Build Coastguard Worker typedef char* SockOptArg;
98*d9f75844SAndroid Build Coastguard Worker #endif
99*d9f75844SAndroid Build Coastguard Worker
100*d9f75844SAndroid Build Coastguard Worker #if defined(WEBRTC_USE_EPOLL)
101*d9f75844SAndroid Build Coastguard Worker // POLLRDHUP / EPOLLRDHUP are only defined starting with Linux 2.6.17.
102*d9f75844SAndroid Build Coastguard Worker #if !defined(POLLRDHUP)
103*d9f75844SAndroid Build Coastguard Worker #define POLLRDHUP 0x2000
104*d9f75844SAndroid Build Coastguard Worker #endif
105*d9f75844SAndroid Build Coastguard Worker #if !defined(EPOLLRDHUP)
106*d9f75844SAndroid Build Coastguard Worker #define EPOLLRDHUP 0x2000
107*d9f75844SAndroid Build Coastguard Worker #endif
108*d9f75844SAndroid Build Coastguard Worker #endif
109*d9f75844SAndroid Build Coastguard Worker
110*d9f75844SAndroid Build Coastguard Worker namespace {
111*d9f75844SAndroid Build Coastguard Worker class ScopedSetTrue {
112*d9f75844SAndroid Build Coastguard Worker public:
ScopedSetTrue(bool * value)113*d9f75844SAndroid Build Coastguard Worker ScopedSetTrue(bool* value) : value_(value) {
114*d9f75844SAndroid Build Coastguard Worker RTC_DCHECK(!*value_);
115*d9f75844SAndroid Build Coastguard Worker *value_ = true;
116*d9f75844SAndroid Build Coastguard Worker }
~ScopedSetTrue()117*d9f75844SAndroid Build Coastguard Worker ~ScopedSetTrue() { *value_ = false; }
118*d9f75844SAndroid Build Coastguard Worker
119*d9f75844SAndroid Build Coastguard Worker private:
120*d9f75844SAndroid Build Coastguard Worker bool* value_;
121*d9f75844SAndroid Build Coastguard Worker };
122*d9f75844SAndroid Build Coastguard Worker
123*d9f75844SAndroid Build Coastguard Worker // Returns true if the the client is in the experiment to get timestamps
124*d9f75844SAndroid Build Coastguard Worker // from the socket implementation.
IsScmTimeStampExperimentEnabled()125*d9f75844SAndroid Build Coastguard Worker bool IsScmTimeStampExperimentEnabled() {
126*d9f75844SAndroid Build Coastguard Worker return webrtc::field_trial::IsEnabled("WebRTC-SCM-Timestamp");
127*d9f75844SAndroid Build Coastguard Worker }
128*d9f75844SAndroid Build Coastguard Worker } // namespace
129*d9f75844SAndroid Build Coastguard Worker
130*d9f75844SAndroid Build Coastguard Worker namespace rtc {
131*d9f75844SAndroid Build Coastguard Worker
PhysicalSocket(PhysicalSocketServer * ss,SOCKET s)132*d9f75844SAndroid Build Coastguard Worker PhysicalSocket::PhysicalSocket(PhysicalSocketServer* ss, SOCKET s)
133*d9f75844SAndroid Build Coastguard Worker : ss_(ss),
134*d9f75844SAndroid Build Coastguard Worker s_(s),
135*d9f75844SAndroid Build Coastguard Worker error_(0),
136*d9f75844SAndroid Build Coastguard Worker state_((s == INVALID_SOCKET) ? CS_CLOSED : CS_CONNECTED),
137*d9f75844SAndroid Build Coastguard Worker resolver_(nullptr),
138*d9f75844SAndroid Build Coastguard Worker read_scm_timestamp_experiment_(IsScmTimeStampExperimentEnabled()) {
139*d9f75844SAndroid Build Coastguard Worker if (s_ != INVALID_SOCKET) {
140*d9f75844SAndroid Build Coastguard Worker SetEnabledEvents(DE_READ | DE_WRITE);
141*d9f75844SAndroid Build Coastguard Worker
142*d9f75844SAndroid Build Coastguard Worker int type = SOCK_STREAM;
143*d9f75844SAndroid Build Coastguard Worker socklen_t len = sizeof(type);
144*d9f75844SAndroid Build Coastguard Worker const int res =
145*d9f75844SAndroid Build Coastguard Worker getsockopt(s_, SOL_SOCKET, SO_TYPE, (SockOptArg)&type, &len);
146*d9f75844SAndroid Build Coastguard Worker RTC_DCHECK_EQ(0, res);
147*d9f75844SAndroid Build Coastguard Worker udp_ = (SOCK_DGRAM == type);
148*d9f75844SAndroid Build Coastguard Worker }
149*d9f75844SAndroid Build Coastguard Worker }
150*d9f75844SAndroid Build Coastguard Worker
~PhysicalSocket()151*d9f75844SAndroid Build Coastguard Worker PhysicalSocket::~PhysicalSocket() {
152*d9f75844SAndroid Build Coastguard Worker Close();
153*d9f75844SAndroid Build Coastguard Worker }
154*d9f75844SAndroid Build Coastguard Worker
Create(int family,int type)155*d9f75844SAndroid Build Coastguard Worker bool PhysicalSocket::Create(int family, int type) {
156*d9f75844SAndroid Build Coastguard Worker Close();
157*d9f75844SAndroid Build Coastguard Worker s_ = ::socket(family, type, 0);
158*d9f75844SAndroid Build Coastguard Worker udp_ = (SOCK_DGRAM == type);
159*d9f75844SAndroid Build Coastguard Worker family_ = family;
160*d9f75844SAndroid Build Coastguard Worker UpdateLastError();
161*d9f75844SAndroid Build Coastguard Worker if (udp_) {
162*d9f75844SAndroid Build Coastguard Worker SetEnabledEvents(DE_READ | DE_WRITE);
163*d9f75844SAndroid Build Coastguard Worker }
164*d9f75844SAndroid Build Coastguard Worker return s_ != INVALID_SOCKET;
165*d9f75844SAndroid Build Coastguard Worker }
166*d9f75844SAndroid Build Coastguard Worker
GetLocalAddress() const167*d9f75844SAndroid Build Coastguard Worker SocketAddress PhysicalSocket::GetLocalAddress() const {
168*d9f75844SAndroid Build Coastguard Worker sockaddr_storage addr_storage = {};
169*d9f75844SAndroid Build Coastguard Worker socklen_t addrlen = sizeof(addr_storage);
170*d9f75844SAndroid Build Coastguard Worker sockaddr* addr = reinterpret_cast<sockaddr*>(&addr_storage);
171*d9f75844SAndroid Build Coastguard Worker int result = ::getsockname(s_, addr, &addrlen);
172*d9f75844SAndroid Build Coastguard Worker SocketAddress address;
173*d9f75844SAndroid Build Coastguard Worker if (result >= 0) {
174*d9f75844SAndroid Build Coastguard Worker SocketAddressFromSockAddrStorage(addr_storage, &address);
175*d9f75844SAndroid Build Coastguard Worker } else {
176*d9f75844SAndroid Build Coastguard Worker RTC_LOG(LS_WARNING) << "GetLocalAddress: unable to get local addr, socket="
177*d9f75844SAndroid Build Coastguard Worker << s_;
178*d9f75844SAndroid Build Coastguard Worker }
179*d9f75844SAndroid Build Coastguard Worker return address;
180*d9f75844SAndroid Build Coastguard Worker }
181*d9f75844SAndroid Build Coastguard Worker
GetRemoteAddress() const182*d9f75844SAndroid Build Coastguard Worker SocketAddress PhysicalSocket::GetRemoteAddress() const {
183*d9f75844SAndroid Build Coastguard Worker sockaddr_storage addr_storage = {};
184*d9f75844SAndroid Build Coastguard Worker socklen_t addrlen = sizeof(addr_storage);
185*d9f75844SAndroid Build Coastguard Worker sockaddr* addr = reinterpret_cast<sockaddr*>(&addr_storage);
186*d9f75844SAndroid Build Coastguard Worker int result = ::getpeername(s_, addr, &addrlen);
187*d9f75844SAndroid Build Coastguard Worker SocketAddress address;
188*d9f75844SAndroid Build Coastguard Worker if (result >= 0) {
189*d9f75844SAndroid Build Coastguard Worker SocketAddressFromSockAddrStorage(addr_storage, &address);
190*d9f75844SAndroid Build Coastguard Worker } else {
191*d9f75844SAndroid Build Coastguard Worker RTC_LOG(LS_WARNING)
192*d9f75844SAndroid Build Coastguard Worker << "GetRemoteAddress: unable to get remote addr, socket=" << s_;
193*d9f75844SAndroid Build Coastguard Worker }
194*d9f75844SAndroid Build Coastguard Worker return address;
195*d9f75844SAndroid Build Coastguard Worker }
196*d9f75844SAndroid Build Coastguard Worker
Bind(const SocketAddress & bind_addr)197*d9f75844SAndroid Build Coastguard Worker int PhysicalSocket::Bind(const SocketAddress& bind_addr) {
198*d9f75844SAndroid Build Coastguard Worker SocketAddress copied_bind_addr = bind_addr;
199*d9f75844SAndroid Build Coastguard Worker // If a network binder is available, use it to bind a socket to an interface
200*d9f75844SAndroid Build Coastguard Worker // instead of bind(), since this is more reliable on an OS with a weak host
201*d9f75844SAndroid Build Coastguard Worker // model.
202*d9f75844SAndroid Build Coastguard Worker if (ss_->network_binder() && !bind_addr.IsAnyIP()) {
203*d9f75844SAndroid Build Coastguard Worker NetworkBindingResult result =
204*d9f75844SAndroid Build Coastguard Worker ss_->network_binder()->BindSocketToNetwork(s_, bind_addr.ipaddr());
205*d9f75844SAndroid Build Coastguard Worker if (result == NetworkBindingResult::SUCCESS) {
206*d9f75844SAndroid Build Coastguard Worker // Since the network binder handled binding the socket to the desired
207*d9f75844SAndroid Build Coastguard Worker // network interface, we don't need to (and shouldn't) include an IP in
208*d9f75844SAndroid Build Coastguard Worker // the bind() call; bind() just needs to assign a port.
209*d9f75844SAndroid Build Coastguard Worker copied_bind_addr.SetIP(GetAnyIP(copied_bind_addr.ipaddr().family()));
210*d9f75844SAndroid Build Coastguard Worker } else if (result == NetworkBindingResult::NOT_IMPLEMENTED) {
211*d9f75844SAndroid Build Coastguard Worker RTC_LOG(LS_INFO) << "Can't bind socket to network because "
212*d9f75844SAndroid Build Coastguard Worker "network binding is not implemented for this OS.";
213*d9f75844SAndroid Build Coastguard Worker } else {
214*d9f75844SAndroid Build Coastguard Worker if (bind_addr.IsLoopbackIP()) {
215*d9f75844SAndroid Build Coastguard Worker // If we couldn't bind to a loopback IP (which should only happen in
216*d9f75844SAndroid Build Coastguard Worker // test scenarios), continue on. This may be expected behavior.
217*d9f75844SAndroid Build Coastguard Worker RTC_LOG(LS_VERBOSE) << "Binding socket to loopback address"
218*d9f75844SAndroid Build Coastguard Worker << " failed; result: " << static_cast<int>(result);
219*d9f75844SAndroid Build Coastguard Worker } else {
220*d9f75844SAndroid Build Coastguard Worker RTC_LOG(LS_WARNING) << "Binding socket to network address"
221*d9f75844SAndroid Build Coastguard Worker << " failed; result: " << static_cast<int>(result);
222*d9f75844SAndroid Build Coastguard Worker // If a network binding was attempted and failed, we should stop here
223*d9f75844SAndroid Build Coastguard Worker // and not try to use the socket. Otherwise, we may end up sending
224*d9f75844SAndroid Build Coastguard Worker // packets with an invalid source address.
225*d9f75844SAndroid Build Coastguard Worker // See: https://bugs.chromium.org/p/webrtc/issues/detail?id=7026
226*d9f75844SAndroid Build Coastguard Worker return -1;
227*d9f75844SAndroid Build Coastguard Worker }
228*d9f75844SAndroid Build Coastguard Worker }
229*d9f75844SAndroid Build Coastguard Worker }
230*d9f75844SAndroid Build Coastguard Worker sockaddr_storage addr_storage;
231*d9f75844SAndroid Build Coastguard Worker size_t len = copied_bind_addr.ToSockAddrStorage(&addr_storage);
232*d9f75844SAndroid Build Coastguard Worker sockaddr* addr = reinterpret_cast<sockaddr*>(&addr_storage);
233*d9f75844SAndroid Build Coastguard Worker int err = ::bind(s_, addr, static_cast<int>(len));
234*d9f75844SAndroid Build Coastguard Worker UpdateLastError();
235*d9f75844SAndroid Build Coastguard Worker #if !defined(NDEBUG)
236*d9f75844SAndroid Build Coastguard Worker if (0 == err) {
237*d9f75844SAndroid Build Coastguard Worker dbg_addr_ = "Bound @ ";
238*d9f75844SAndroid Build Coastguard Worker dbg_addr_.append(GetLocalAddress().ToString());
239*d9f75844SAndroid Build Coastguard Worker }
240*d9f75844SAndroid Build Coastguard Worker #endif
241*d9f75844SAndroid Build Coastguard Worker return err;
242*d9f75844SAndroid Build Coastguard Worker }
243*d9f75844SAndroid Build Coastguard Worker
Connect(const SocketAddress & addr)244*d9f75844SAndroid Build Coastguard Worker int PhysicalSocket::Connect(const SocketAddress& addr) {
245*d9f75844SAndroid Build Coastguard Worker // TODO(pthatcher): Implicit creation is required to reconnect...
246*d9f75844SAndroid Build Coastguard Worker // ...but should we make it more explicit?
247*d9f75844SAndroid Build Coastguard Worker if (state_ != CS_CLOSED) {
248*d9f75844SAndroid Build Coastguard Worker SetError(EALREADY);
249*d9f75844SAndroid Build Coastguard Worker return SOCKET_ERROR;
250*d9f75844SAndroid Build Coastguard Worker }
251*d9f75844SAndroid Build Coastguard Worker if (addr.IsUnresolvedIP()) {
252*d9f75844SAndroid Build Coastguard Worker RTC_LOG(LS_VERBOSE) << "Resolving addr in PhysicalSocket::Connect";
253*d9f75844SAndroid Build Coastguard Worker resolver_ = new AsyncResolver();
254*d9f75844SAndroid Build Coastguard Worker resolver_->SignalDone.connect(this, &PhysicalSocket::OnResolveResult);
255*d9f75844SAndroid Build Coastguard Worker resolver_->Start(addr);
256*d9f75844SAndroid Build Coastguard Worker state_ = CS_CONNECTING;
257*d9f75844SAndroid Build Coastguard Worker return 0;
258*d9f75844SAndroid Build Coastguard Worker }
259*d9f75844SAndroid Build Coastguard Worker
260*d9f75844SAndroid Build Coastguard Worker return DoConnect(addr);
261*d9f75844SAndroid Build Coastguard Worker }
262*d9f75844SAndroid Build Coastguard Worker
DoConnect(const SocketAddress & connect_addr)263*d9f75844SAndroid Build Coastguard Worker int PhysicalSocket::DoConnect(const SocketAddress& connect_addr) {
264*d9f75844SAndroid Build Coastguard Worker if ((s_ == INVALID_SOCKET) && !Create(connect_addr.family(), SOCK_STREAM)) {
265*d9f75844SAndroid Build Coastguard Worker return SOCKET_ERROR;
266*d9f75844SAndroid Build Coastguard Worker }
267*d9f75844SAndroid Build Coastguard Worker sockaddr_storage addr_storage;
268*d9f75844SAndroid Build Coastguard Worker size_t len = connect_addr.ToSockAddrStorage(&addr_storage);
269*d9f75844SAndroid Build Coastguard Worker sockaddr* addr = reinterpret_cast<sockaddr*>(&addr_storage);
270*d9f75844SAndroid Build Coastguard Worker int err = ::connect(s_, addr, static_cast<int>(len));
271*d9f75844SAndroid Build Coastguard Worker UpdateLastError();
272*d9f75844SAndroid Build Coastguard Worker uint8_t events = DE_READ | DE_WRITE;
273*d9f75844SAndroid Build Coastguard Worker if (err == 0) {
274*d9f75844SAndroid Build Coastguard Worker state_ = CS_CONNECTED;
275*d9f75844SAndroid Build Coastguard Worker } else if (IsBlockingError(GetError())) {
276*d9f75844SAndroid Build Coastguard Worker state_ = CS_CONNECTING;
277*d9f75844SAndroid Build Coastguard Worker events |= DE_CONNECT;
278*d9f75844SAndroid Build Coastguard Worker } else {
279*d9f75844SAndroid Build Coastguard Worker return SOCKET_ERROR;
280*d9f75844SAndroid Build Coastguard Worker }
281*d9f75844SAndroid Build Coastguard Worker
282*d9f75844SAndroid Build Coastguard Worker EnableEvents(events);
283*d9f75844SAndroid Build Coastguard Worker return 0;
284*d9f75844SAndroid Build Coastguard Worker }
285*d9f75844SAndroid Build Coastguard Worker
GetError() const286*d9f75844SAndroid Build Coastguard Worker int PhysicalSocket::GetError() const {
287*d9f75844SAndroid Build Coastguard Worker webrtc::MutexLock lock(&mutex_);
288*d9f75844SAndroid Build Coastguard Worker return error_;
289*d9f75844SAndroid Build Coastguard Worker }
290*d9f75844SAndroid Build Coastguard Worker
SetError(int error)291*d9f75844SAndroid Build Coastguard Worker void PhysicalSocket::SetError(int error) {
292*d9f75844SAndroid Build Coastguard Worker webrtc::MutexLock lock(&mutex_);
293*d9f75844SAndroid Build Coastguard Worker error_ = error;
294*d9f75844SAndroid Build Coastguard Worker }
295*d9f75844SAndroid Build Coastguard Worker
GetState() const296*d9f75844SAndroid Build Coastguard Worker Socket::ConnState PhysicalSocket::GetState() const {
297*d9f75844SAndroid Build Coastguard Worker return state_;
298*d9f75844SAndroid Build Coastguard Worker }
299*d9f75844SAndroid Build Coastguard Worker
GetOption(Option opt,int * value)300*d9f75844SAndroid Build Coastguard Worker int PhysicalSocket::GetOption(Option opt, int* value) {
301*d9f75844SAndroid Build Coastguard Worker int slevel;
302*d9f75844SAndroid Build Coastguard Worker int sopt;
303*d9f75844SAndroid Build Coastguard Worker if (TranslateOption(opt, &slevel, &sopt) == -1)
304*d9f75844SAndroid Build Coastguard Worker return -1;
305*d9f75844SAndroid Build Coastguard Worker socklen_t optlen = sizeof(*value);
306*d9f75844SAndroid Build Coastguard Worker int ret = ::getsockopt(s_, slevel, sopt, (SockOptArg)value, &optlen);
307*d9f75844SAndroid Build Coastguard Worker if (ret == -1) {
308*d9f75844SAndroid Build Coastguard Worker return -1;
309*d9f75844SAndroid Build Coastguard Worker }
310*d9f75844SAndroid Build Coastguard Worker if (opt == OPT_DONTFRAGMENT) {
311*d9f75844SAndroid Build Coastguard Worker #if defined(WEBRTC_LINUX) && !defined(WEBRTC_ANDROID)
312*d9f75844SAndroid Build Coastguard Worker *value = (*value != IP_PMTUDISC_DONT) ? 1 : 0;
313*d9f75844SAndroid Build Coastguard Worker #endif
314*d9f75844SAndroid Build Coastguard Worker } else if (opt == OPT_DSCP) {
315*d9f75844SAndroid Build Coastguard Worker #if defined(WEBRTC_POSIX)
316*d9f75844SAndroid Build Coastguard Worker // unshift DSCP value to get six most significant bits of IP DiffServ field
317*d9f75844SAndroid Build Coastguard Worker *value >>= 2;
318*d9f75844SAndroid Build Coastguard Worker #endif
319*d9f75844SAndroid Build Coastguard Worker }
320*d9f75844SAndroid Build Coastguard Worker return ret;
321*d9f75844SAndroid Build Coastguard Worker }
322*d9f75844SAndroid Build Coastguard Worker
SetOption(Option opt,int value)323*d9f75844SAndroid Build Coastguard Worker int PhysicalSocket::SetOption(Option opt, int value) {
324*d9f75844SAndroid Build Coastguard Worker int slevel;
325*d9f75844SAndroid Build Coastguard Worker int sopt;
326*d9f75844SAndroid Build Coastguard Worker if (TranslateOption(opt, &slevel, &sopt) == -1)
327*d9f75844SAndroid Build Coastguard Worker return -1;
328*d9f75844SAndroid Build Coastguard Worker if (opt == OPT_DONTFRAGMENT) {
329*d9f75844SAndroid Build Coastguard Worker #if defined(WEBRTC_LINUX) && !defined(WEBRTC_ANDROID)
330*d9f75844SAndroid Build Coastguard Worker value = (value) ? IP_PMTUDISC_DO : IP_PMTUDISC_DONT;
331*d9f75844SAndroid Build Coastguard Worker #endif
332*d9f75844SAndroid Build Coastguard Worker } else if (opt == OPT_DSCP) {
333*d9f75844SAndroid Build Coastguard Worker #if defined(WEBRTC_POSIX)
334*d9f75844SAndroid Build Coastguard Worker // shift DSCP value to fit six most significant bits of IP DiffServ field
335*d9f75844SAndroid Build Coastguard Worker value <<= 2;
336*d9f75844SAndroid Build Coastguard Worker #endif
337*d9f75844SAndroid Build Coastguard Worker }
338*d9f75844SAndroid Build Coastguard Worker #if defined(WEBRTC_POSIX)
339*d9f75844SAndroid Build Coastguard Worker if (sopt == IPV6_TCLASS) {
340*d9f75844SAndroid Build Coastguard Worker // Set the IPv4 option in all cases to support dual-stack sockets.
341*d9f75844SAndroid Build Coastguard Worker // Don't bother checking the return code, as this is expected to fail if
342*d9f75844SAndroid Build Coastguard Worker // it's not actually dual-stack.
343*d9f75844SAndroid Build Coastguard Worker ::setsockopt(s_, IPPROTO_IP, IP_TOS, (SockOptArg)&value, sizeof(value));
344*d9f75844SAndroid Build Coastguard Worker }
345*d9f75844SAndroid Build Coastguard Worker #endif
346*d9f75844SAndroid Build Coastguard Worker int result =
347*d9f75844SAndroid Build Coastguard Worker ::setsockopt(s_, slevel, sopt, (SockOptArg)&value, sizeof(value));
348*d9f75844SAndroid Build Coastguard Worker if (result != 0) {
349*d9f75844SAndroid Build Coastguard Worker UpdateLastError();
350*d9f75844SAndroid Build Coastguard Worker }
351*d9f75844SAndroid Build Coastguard Worker return result;
352*d9f75844SAndroid Build Coastguard Worker }
353*d9f75844SAndroid Build Coastguard Worker
Send(const void * pv,size_t cb)354*d9f75844SAndroid Build Coastguard Worker int PhysicalSocket::Send(const void* pv, size_t cb) {
355*d9f75844SAndroid Build Coastguard Worker int sent = DoSend(
356*d9f75844SAndroid Build Coastguard Worker s_, reinterpret_cast<const char*>(pv), static_cast<int>(cb),
357*d9f75844SAndroid Build Coastguard Worker #if defined(WEBRTC_LINUX) && !defined(WEBRTC_ANDROID)
358*d9f75844SAndroid Build Coastguard Worker // Suppress SIGPIPE. Without this, attempting to send on a socket whose
359*d9f75844SAndroid Build Coastguard Worker // other end is closed will result in a SIGPIPE signal being raised to
360*d9f75844SAndroid Build Coastguard Worker // our process, which by default will terminate the process, which we
361*d9f75844SAndroid Build Coastguard Worker // don't want. By specifying this flag, we'll just get the error EPIPE
362*d9f75844SAndroid Build Coastguard Worker // instead and can handle the error gracefully.
363*d9f75844SAndroid Build Coastguard Worker MSG_NOSIGNAL
364*d9f75844SAndroid Build Coastguard Worker #else
365*d9f75844SAndroid Build Coastguard Worker 0
366*d9f75844SAndroid Build Coastguard Worker #endif
367*d9f75844SAndroid Build Coastguard Worker );
368*d9f75844SAndroid Build Coastguard Worker UpdateLastError();
369*d9f75844SAndroid Build Coastguard Worker MaybeRemapSendError();
370*d9f75844SAndroid Build Coastguard Worker // We have seen minidumps where this may be false.
371*d9f75844SAndroid Build Coastguard Worker RTC_DCHECK(sent <= static_cast<int>(cb));
372*d9f75844SAndroid Build Coastguard Worker if ((sent > 0 && sent < static_cast<int>(cb)) ||
373*d9f75844SAndroid Build Coastguard Worker (sent < 0 && IsBlockingError(GetError()))) {
374*d9f75844SAndroid Build Coastguard Worker EnableEvents(DE_WRITE);
375*d9f75844SAndroid Build Coastguard Worker }
376*d9f75844SAndroid Build Coastguard Worker return sent;
377*d9f75844SAndroid Build Coastguard Worker }
378*d9f75844SAndroid Build Coastguard Worker
SendTo(const void * buffer,size_t length,const SocketAddress & addr)379*d9f75844SAndroid Build Coastguard Worker int PhysicalSocket::SendTo(const void* buffer,
380*d9f75844SAndroid Build Coastguard Worker size_t length,
381*d9f75844SAndroid Build Coastguard Worker const SocketAddress& addr) {
382*d9f75844SAndroid Build Coastguard Worker sockaddr_storage saddr;
383*d9f75844SAndroid Build Coastguard Worker size_t len = addr.ToSockAddrStorage(&saddr);
384*d9f75844SAndroid Build Coastguard Worker int sent =
385*d9f75844SAndroid Build Coastguard Worker DoSendTo(s_, static_cast<const char*>(buffer), static_cast<int>(length),
386*d9f75844SAndroid Build Coastguard Worker #if defined(WEBRTC_LINUX) && !defined(WEBRTC_ANDROID)
387*d9f75844SAndroid Build Coastguard Worker // Suppress SIGPIPE. See above for explanation.
388*d9f75844SAndroid Build Coastguard Worker MSG_NOSIGNAL,
389*d9f75844SAndroid Build Coastguard Worker #else
390*d9f75844SAndroid Build Coastguard Worker 0,
391*d9f75844SAndroid Build Coastguard Worker #endif
392*d9f75844SAndroid Build Coastguard Worker reinterpret_cast<sockaddr*>(&saddr), static_cast<int>(len));
393*d9f75844SAndroid Build Coastguard Worker UpdateLastError();
394*d9f75844SAndroid Build Coastguard Worker MaybeRemapSendError();
395*d9f75844SAndroid Build Coastguard Worker // We have seen minidumps where this may be false.
396*d9f75844SAndroid Build Coastguard Worker RTC_DCHECK(sent <= static_cast<int>(length));
397*d9f75844SAndroid Build Coastguard Worker if ((sent > 0 && sent < static_cast<int>(length)) ||
398*d9f75844SAndroid Build Coastguard Worker (sent < 0 && IsBlockingError(GetError()))) {
399*d9f75844SAndroid Build Coastguard Worker EnableEvents(DE_WRITE);
400*d9f75844SAndroid Build Coastguard Worker }
401*d9f75844SAndroid Build Coastguard Worker return sent;
402*d9f75844SAndroid Build Coastguard Worker }
403*d9f75844SAndroid Build Coastguard Worker
Recv(void * buffer,size_t length,int64_t * timestamp)404*d9f75844SAndroid Build Coastguard Worker int PhysicalSocket::Recv(void* buffer, size_t length, int64_t* timestamp) {
405*d9f75844SAndroid Build Coastguard Worker int received =
406*d9f75844SAndroid Build Coastguard Worker DoReadFromSocket(buffer, length, /*out_addr*/ nullptr, timestamp);
407*d9f75844SAndroid Build Coastguard Worker if ((received == 0) && (length != 0)) {
408*d9f75844SAndroid Build Coastguard Worker // Note: on graceful shutdown, recv can return 0. In this case, we
409*d9f75844SAndroid Build Coastguard Worker // pretend it is blocking, and then signal close, so that simplifying
410*d9f75844SAndroid Build Coastguard Worker // assumptions can be made about Recv.
411*d9f75844SAndroid Build Coastguard Worker RTC_LOG(LS_WARNING) << "EOF from socket; deferring close event";
412*d9f75844SAndroid Build Coastguard Worker // Must turn this back on so that the select() loop will notice the close
413*d9f75844SAndroid Build Coastguard Worker // event.
414*d9f75844SAndroid Build Coastguard Worker EnableEvents(DE_READ);
415*d9f75844SAndroid Build Coastguard Worker SetError(EWOULDBLOCK);
416*d9f75844SAndroid Build Coastguard Worker return SOCKET_ERROR;
417*d9f75844SAndroid Build Coastguard Worker }
418*d9f75844SAndroid Build Coastguard Worker
419*d9f75844SAndroid Build Coastguard Worker UpdateLastError();
420*d9f75844SAndroid Build Coastguard Worker int error = GetError();
421*d9f75844SAndroid Build Coastguard Worker bool success = (received >= 0) || IsBlockingError(error);
422*d9f75844SAndroid Build Coastguard Worker if (udp_ || success) {
423*d9f75844SAndroid Build Coastguard Worker EnableEvents(DE_READ);
424*d9f75844SAndroid Build Coastguard Worker }
425*d9f75844SAndroid Build Coastguard Worker if (!success) {
426*d9f75844SAndroid Build Coastguard Worker RTC_LOG_F(LS_VERBOSE) << "Error = " << error;
427*d9f75844SAndroid Build Coastguard Worker }
428*d9f75844SAndroid Build Coastguard Worker return received;
429*d9f75844SAndroid Build Coastguard Worker }
430*d9f75844SAndroid Build Coastguard Worker
RecvFrom(void * buffer,size_t length,SocketAddress * out_addr,int64_t * timestamp)431*d9f75844SAndroid Build Coastguard Worker int PhysicalSocket::RecvFrom(void* buffer,
432*d9f75844SAndroid Build Coastguard Worker size_t length,
433*d9f75844SAndroid Build Coastguard Worker SocketAddress* out_addr,
434*d9f75844SAndroid Build Coastguard Worker int64_t* timestamp) {
435*d9f75844SAndroid Build Coastguard Worker int received = DoReadFromSocket(buffer, length, out_addr, timestamp);
436*d9f75844SAndroid Build Coastguard Worker UpdateLastError();
437*d9f75844SAndroid Build Coastguard Worker int error = GetError();
438*d9f75844SAndroid Build Coastguard Worker bool success = (received >= 0) || IsBlockingError(error);
439*d9f75844SAndroid Build Coastguard Worker if (udp_ || success) {
440*d9f75844SAndroid Build Coastguard Worker EnableEvents(DE_READ);
441*d9f75844SAndroid Build Coastguard Worker }
442*d9f75844SAndroid Build Coastguard Worker if (!success) {
443*d9f75844SAndroid Build Coastguard Worker RTC_LOG_F(LS_VERBOSE) << "Error = " << error;
444*d9f75844SAndroid Build Coastguard Worker }
445*d9f75844SAndroid Build Coastguard Worker return received;
446*d9f75844SAndroid Build Coastguard Worker }
447*d9f75844SAndroid Build Coastguard Worker
DoReadFromSocket(void * buffer,size_t length,SocketAddress * out_addr,int64_t * timestamp)448*d9f75844SAndroid Build Coastguard Worker int PhysicalSocket::DoReadFromSocket(void* buffer,
449*d9f75844SAndroid Build Coastguard Worker size_t length,
450*d9f75844SAndroid Build Coastguard Worker SocketAddress* out_addr,
451*d9f75844SAndroid Build Coastguard Worker int64_t* timestamp) {
452*d9f75844SAndroid Build Coastguard Worker sockaddr_storage addr_storage;
453*d9f75844SAndroid Build Coastguard Worker socklen_t addr_len = sizeof(addr_storage);
454*d9f75844SAndroid Build Coastguard Worker sockaddr* addr = reinterpret_cast<sockaddr*>(&addr_storage);
455*d9f75844SAndroid Build Coastguard Worker
456*d9f75844SAndroid Build Coastguard Worker #if defined(WEBRTC_POSIX)
457*d9f75844SAndroid Build Coastguard Worker int received = 0;
458*d9f75844SAndroid Build Coastguard Worker if (read_scm_timestamp_experiment_) {
459*d9f75844SAndroid Build Coastguard Worker iovec iov = {.iov_base = buffer, .iov_len = length};
460*d9f75844SAndroid Build Coastguard Worker msghdr msg = {.msg_iov = &iov, .msg_iovlen = 1};
461*d9f75844SAndroid Build Coastguard Worker if (out_addr) {
462*d9f75844SAndroid Build Coastguard Worker out_addr->Clear();
463*d9f75844SAndroid Build Coastguard Worker msg.msg_name = addr;
464*d9f75844SAndroid Build Coastguard Worker msg.msg_namelen = addr_len;
465*d9f75844SAndroid Build Coastguard Worker }
466*d9f75844SAndroid Build Coastguard Worker char control[CMSG_SPACE(sizeof(struct timeval))] = {};
467*d9f75844SAndroid Build Coastguard Worker if (timestamp) {
468*d9f75844SAndroid Build Coastguard Worker *timestamp = -1;
469*d9f75844SAndroid Build Coastguard Worker msg.msg_control = &control;
470*d9f75844SAndroid Build Coastguard Worker msg.msg_controllen = sizeof(control);
471*d9f75844SAndroid Build Coastguard Worker }
472*d9f75844SAndroid Build Coastguard Worker received = ::recvmsg(s_, &msg, 0);
473*d9f75844SAndroid Build Coastguard Worker if (received <= 0) {
474*d9f75844SAndroid Build Coastguard Worker // An error occured or shut down.
475*d9f75844SAndroid Build Coastguard Worker return received;
476*d9f75844SAndroid Build Coastguard Worker }
477*d9f75844SAndroid Build Coastguard Worker if (timestamp) {
478*d9f75844SAndroid Build Coastguard Worker struct cmsghdr* cmsg;
479*d9f75844SAndroid Build Coastguard Worker for (cmsg = CMSG_FIRSTHDR(&msg); cmsg; cmsg = CMSG_NXTHDR(&msg, cmsg)) {
480*d9f75844SAndroid Build Coastguard Worker if (cmsg->cmsg_level != SOL_SOCKET)
481*d9f75844SAndroid Build Coastguard Worker continue;
482*d9f75844SAndroid Build Coastguard Worker if (cmsg->cmsg_type == SCM_TIMESTAMP) {
483*d9f75844SAndroid Build Coastguard Worker timeval* ts = reinterpret_cast<timeval*>(CMSG_DATA(cmsg));
484*d9f75844SAndroid Build Coastguard Worker *timestamp =
485*d9f75844SAndroid Build Coastguard Worker rtc::kNumMicrosecsPerSec * static_cast<int64_t>(ts->tv_sec) +
486*d9f75844SAndroid Build Coastguard Worker static_cast<int64_t>(ts->tv_usec);
487*d9f75844SAndroid Build Coastguard Worker break;
488*d9f75844SAndroid Build Coastguard Worker }
489*d9f75844SAndroid Build Coastguard Worker }
490*d9f75844SAndroid Build Coastguard Worker }
491*d9f75844SAndroid Build Coastguard Worker if (out_addr) {
492*d9f75844SAndroid Build Coastguard Worker SocketAddressFromSockAddrStorage(addr_storage, out_addr);
493*d9f75844SAndroid Build Coastguard Worker }
494*d9f75844SAndroid Build Coastguard Worker } else { // !read_scm_timestamp_experiment_
495*d9f75844SAndroid Build Coastguard Worker if (out_addr) {
496*d9f75844SAndroid Build Coastguard Worker received = ::recvfrom(s_, static_cast<char*>(buffer),
497*d9f75844SAndroid Build Coastguard Worker static_cast<int>(length), 0, addr, &addr_len);
498*d9f75844SAndroid Build Coastguard Worker SocketAddressFromSockAddrStorage(addr_storage, out_addr);
499*d9f75844SAndroid Build Coastguard Worker } else {
500*d9f75844SAndroid Build Coastguard Worker received =
501*d9f75844SAndroid Build Coastguard Worker ::recv(s_, static_cast<char*>(buffer), static_cast<int>(length), 0);
502*d9f75844SAndroid Build Coastguard Worker }
503*d9f75844SAndroid Build Coastguard Worker if (timestamp) {
504*d9f75844SAndroid Build Coastguard Worker *timestamp = GetSocketRecvTimestamp(s_);
505*d9f75844SAndroid Build Coastguard Worker }
506*d9f75844SAndroid Build Coastguard Worker }
507*d9f75844SAndroid Build Coastguard Worker return received;
508*d9f75844SAndroid Build Coastguard Worker
509*d9f75844SAndroid Build Coastguard Worker #else
510*d9f75844SAndroid Build Coastguard Worker int received = 0;
511*d9f75844SAndroid Build Coastguard Worker if (out_addr) {
512*d9f75844SAndroid Build Coastguard Worker received = ::recvfrom(s_, static_cast<char*>(buffer),
513*d9f75844SAndroid Build Coastguard Worker static_cast<int>(length), 0, addr, &addr_len);
514*d9f75844SAndroid Build Coastguard Worker SocketAddressFromSockAddrStorage(addr_storage, out_addr);
515*d9f75844SAndroid Build Coastguard Worker } else {
516*d9f75844SAndroid Build Coastguard Worker received =
517*d9f75844SAndroid Build Coastguard Worker ::recv(s_, static_cast<char*>(buffer), static_cast<int>(length), 0);
518*d9f75844SAndroid Build Coastguard Worker }
519*d9f75844SAndroid Build Coastguard Worker if (timestamp) {
520*d9f75844SAndroid Build Coastguard Worker *timestamp = -1;
521*d9f75844SAndroid Build Coastguard Worker }
522*d9f75844SAndroid Build Coastguard Worker return received;
523*d9f75844SAndroid Build Coastguard Worker #endif
524*d9f75844SAndroid Build Coastguard Worker }
525*d9f75844SAndroid Build Coastguard Worker
Listen(int backlog)526*d9f75844SAndroid Build Coastguard Worker int PhysicalSocket::Listen(int backlog) {
527*d9f75844SAndroid Build Coastguard Worker int err = ::listen(s_, backlog);
528*d9f75844SAndroid Build Coastguard Worker UpdateLastError();
529*d9f75844SAndroid Build Coastguard Worker if (err == 0) {
530*d9f75844SAndroid Build Coastguard Worker state_ = CS_CONNECTING;
531*d9f75844SAndroid Build Coastguard Worker EnableEvents(DE_ACCEPT);
532*d9f75844SAndroid Build Coastguard Worker #if !defined(NDEBUG)
533*d9f75844SAndroid Build Coastguard Worker dbg_addr_ = "Listening @ ";
534*d9f75844SAndroid Build Coastguard Worker dbg_addr_.append(GetLocalAddress().ToString());
535*d9f75844SAndroid Build Coastguard Worker #endif
536*d9f75844SAndroid Build Coastguard Worker }
537*d9f75844SAndroid Build Coastguard Worker return err;
538*d9f75844SAndroid Build Coastguard Worker }
539*d9f75844SAndroid Build Coastguard Worker
Accept(SocketAddress * out_addr)540*d9f75844SAndroid Build Coastguard Worker Socket* PhysicalSocket::Accept(SocketAddress* out_addr) {
541*d9f75844SAndroid Build Coastguard Worker // Always re-subscribe DE_ACCEPT to make sure new incoming connections will
542*d9f75844SAndroid Build Coastguard Worker // trigger an event even if DoAccept returns an error here.
543*d9f75844SAndroid Build Coastguard Worker EnableEvents(DE_ACCEPT);
544*d9f75844SAndroid Build Coastguard Worker sockaddr_storage addr_storage;
545*d9f75844SAndroid Build Coastguard Worker socklen_t addr_len = sizeof(addr_storage);
546*d9f75844SAndroid Build Coastguard Worker sockaddr* addr = reinterpret_cast<sockaddr*>(&addr_storage);
547*d9f75844SAndroid Build Coastguard Worker SOCKET s = DoAccept(s_, addr, &addr_len);
548*d9f75844SAndroid Build Coastguard Worker UpdateLastError();
549*d9f75844SAndroid Build Coastguard Worker if (s == INVALID_SOCKET)
550*d9f75844SAndroid Build Coastguard Worker return nullptr;
551*d9f75844SAndroid Build Coastguard Worker if (out_addr != nullptr)
552*d9f75844SAndroid Build Coastguard Worker SocketAddressFromSockAddrStorage(addr_storage, out_addr);
553*d9f75844SAndroid Build Coastguard Worker return ss_->WrapSocket(s);
554*d9f75844SAndroid Build Coastguard Worker }
555*d9f75844SAndroid Build Coastguard Worker
Close()556*d9f75844SAndroid Build Coastguard Worker int PhysicalSocket::Close() {
557*d9f75844SAndroid Build Coastguard Worker if (s_ == INVALID_SOCKET)
558*d9f75844SAndroid Build Coastguard Worker return 0;
559*d9f75844SAndroid Build Coastguard Worker int err = ::closesocket(s_);
560*d9f75844SAndroid Build Coastguard Worker UpdateLastError();
561*d9f75844SAndroid Build Coastguard Worker s_ = INVALID_SOCKET;
562*d9f75844SAndroid Build Coastguard Worker state_ = CS_CLOSED;
563*d9f75844SAndroid Build Coastguard Worker SetEnabledEvents(0);
564*d9f75844SAndroid Build Coastguard Worker if (resolver_) {
565*d9f75844SAndroid Build Coastguard Worker resolver_->Destroy(false);
566*d9f75844SAndroid Build Coastguard Worker resolver_ = nullptr;
567*d9f75844SAndroid Build Coastguard Worker }
568*d9f75844SAndroid Build Coastguard Worker return err;
569*d9f75844SAndroid Build Coastguard Worker }
570*d9f75844SAndroid Build Coastguard Worker
DoAccept(SOCKET socket,sockaddr * addr,socklen_t * addrlen)571*d9f75844SAndroid Build Coastguard Worker SOCKET PhysicalSocket::DoAccept(SOCKET socket,
572*d9f75844SAndroid Build Coastguard Worker sockaddr* addr,
573*d9f75844SAndroid Build Coastguard Worker socklen_t* addrlen) {
574*d9f75844SAndroid Build Coastguard Worker return ::accept(socket, addr, addrlen);
575*d9f75844SAndroid Build Coastguard Worker }
576*d9f75844SAndroid Build Coastguard Worker
DoSend(SOCKET socket,const char * buf,int len,int flags)577*d9f75844SAndroid Build Coastguard Worker int PhysicalSocket::DoSend(SOCKET socket, const char* buf, int len, int flags) {
578*d9f75844SAndroid Build Coastguard Worker return ::send(socket, buf, len, flags);
579*d9f75844SAndroid Build Coastguard Worker }
580*d9f75844SAndroid Build Coastguard Worker
DoSendTo(SOCKET socket,const char * buf,int len,int flags,const struct sockaddr * dest_addr,socklen_t addrlen)581*d9f75844SAndroid Build Coastguard Worker int PhysicalSocket::DoSendTo(SOCKET socket,
582*d9f75844SAndroid Build Coastguard Worker const char* buf,
583*d9f75844SAndroid Build Coastguard Worker int len,
584*d9f75844SAndroid Build Coastguard Worker int flags,
585*d9f75844SAndroid Build Coastguard Worker const struct sockaddr* dest_addr,
586*d9f75844SAndroid Build Coastguard Worker socklen_t addrlen) {
587*d9f75844SAndroid Build Coastguard Worker return ::sendto(socket, buf, len, flags, dest_addr, addrlen);
588*d9f75844SAndroid Build Coastguard Worker }
589*d9f75844SAndroid Build Coastguard Worker
OnResolveResult(AsyncResolverInterface * resolver)590*d9f75844SAndroid Build Coastguard Worker void PhysicalSocket::OnResolveResult(AsyncResolverInterface* resolver) {
591*d9f75844SAndroid Build Coastguard Worker if (resolver != resolver_) {
592*d9f75844SAndroid Build Coastguard Worker return;
593*d9f75844SAndroid Build Coastguard Worker }
594*d9f75844SAndroid Build Coastguard Worker
595*d9f75844SAndroid Build Coastguard Worker int error = resolver_->GetError();
596*d9f75844SAndroid Build Coastguard Worker if (error == 0) {
597*d9f75844SAndroid Build Coastguard Worker error = DoConnect(resolver_->address());
598*d9f75844SAndroid Build Coastguard Worker } else {
599*d9f75844SAndroid Build Coastguard Worker Close();
600*d9f75844SAndroid Build Coastguard Worker }
601*d9f75844SAndroid Build Coastguard Worker
602*d9f75844SAndroid Build Coastguard Worker if (error) {
603*d9f75844SAndroid Build Coastguard Worker SetError(error);
604*d9f75844SAndroid Build Coastguard Worker SignalCloseEvent(this, error);
605*d9f75844SAndroid Build Coastguard Worker }
606*d9f75844SAndroid Build Coastguard Worker }
607*d9f75844SAndroid Build Coastguard Worker
UpdateLastError()608*d9f75844SAndroid Build Coastguard Worker void PhysicalSocket::UpdateLastError() {
609*d9f75844SAndroid Build Coastguard Worker SetError(LAST_SYSTEM_ERROR);
610*d9f75844SAndroid Build Coastguard Worker }
611*d9f75844SAndroid Build Coastguard Worker
MaybeRemapSendError()612*d9f75844SAndroid Build Coastguard Worker void PhysicalSocket::MaybeRemapSendError() {
613*d9f75844SAndroid Build Coastguard Worker #if defined(WEBRTC_MAC)
614*d9f75844SAndroid Build Coastguard Worker // https://developer.apple.com/library/mac/documentation/Darwin/
615*d9f75844SAndroid Build Coastguard Worker // Reference/ManPages/man2/sendto.2.html
616*d9f75844SAndroid Build Coastguard Worker // ENOBUFS - The output queue for a network interface is full.
617*d9f75844SAndroid Build Coastguard Worker // This generally indicates that the interface has stopped sending,
618*d9f75844SAndroid Build Coastguard Worker // but may be caused by transient congestion.
619*d9f75844SAndroid Build Coastguard Worker if (GetError() == ENOBUFS) {
620*d9f75844SAndroid Build Coastguard Worker SetError(EWOULDBLOCK);
621*d9f75844SAndroid Build Coastguard Worker }
622*d9f75844SAndroid Build Coastguard Worker #endif
623*d9f75844SAndroid Build Coastguard Worker }
624*d9f75844SAndroid Build Coastguard Worker
SetEnabledEvents(uint8_t events)625*d9f75844SAndroid Build Coastguard Worker void PhysicalSocket::SetEnabledEvents(uint8_t events) {
626*d9f75844SAndroid Build Coastguard Worker enabled_events_ = events;
627*d9f75844SAndroid Build Coastguard Worker }
628*d9f75844SAndroid Build Coastguard Worker
EnableEvents(uint8_t events)629*d9f75844SAndroid Build Coastguard Worker void PhysicalSocket::EnableEvents(uint8_t events) {
630*d9f75844SAndroid Build Coastguard Worker enabled_events_ |= events;
631*d9f75844SAndroid Build Coastguard Worker }
632*d9f75844SAndroid Build Coastguard Worker
DisableEvents(uint8_t events)633*d9f75844SAndroid Build Coastguard Worker void PhysicalSocket::DisableEvents(uint8_t events) {
634*d9f75844SAndroid Build Coastguard Worker enabled_events_ &= ~events;
635*d9f75844SAndroid Build Coastguard Worker }
636*d9f75844SAndroid Build Coastguard Worker
TranslateOption(Option opt,int * slevel,int * sopt)637*d9f75844SAndroid Build Coastguard Worker int PhysicalSocket::TranslateOption(Option opt, int* slevel, int* sopt) {
638*d9f75844SAndroid Build Coastguard Worker switch (opt) {
639*d9f75844SAndroid Build Coastguard Worker case OPT_DONTFRAGMENT:
640*d9f75844SAndroid Build Coastguard Worker #if defined(WEBRTC_WIN)
641*d9f75844SAndroid Build Coastguard Worker *slevel = IPPROTO_IP;
642*d9f75844SAndroid Build Coastguard Worker *sopt = IP_DONTFRAGMENT;
643*d9f75844SAndroid Build Coastguard Worker break;
644*d9f75844SAndroid Build Coastguard Worker #elif defined(WEBRTC_MAC) || defined(BSD) || defined(__native_client__)
645*d9f75844SAndroid Build Coastguard Worker RTC_LOG(LS_WARNING) << "Socket::OPT_DONTFRAGMENT not supported.";
646*d9f75844SAndroid Build Coastguard Worker return -1;
647*d9f75844SAndroid Build Coastguard Worker #elif defined(WEBRTC_POSIX)
648*d9f75844SAndroid Build Coastguard Worker *slevel = IPPROTO_IP;
649*d9f75844SAndroid Build Coastguard Worker *sopt = IP_MTU_DISCOVER;
650*d9f75844SAndroid Build Coastguard Worker break;
651*d9f75844SAndroid Build Coastguard Worker #endif
652*d9f75844SAndroid Build Coastguard Worker case OPT_RCVBUF:
653*d9f75844SAndroid Build Coastguard Worker *slevel = SOL_SOCKET;
654*d9f75844SAndroid Build Coastguard Worker *sopt = SO_RCVBUF;
655*d9f75844SAndroid Build Coastguard Worker break;
656*d9f75844SAndroid Build Coastguard Worker case OPT_SNDBUF:
657*d9f75844SAndroid Build Coastguard Worker *slevel = SOL_SOCKET;
658*d9f75844SAndroid Build Coastguard Worker *sopt = SO_SNDBUF;
659*d9f75844SAndroid Build Coastguard Worker break;
660*d9f75844SAndroid Build Coastguard Worker case OPT_NODELAY:
661*d9f75844SAndroid Build Coastguard Worker *slevel = IPPROTO_TCP;
662*d9f75844SAndroid Build Coastguard Worker *sopt = TCP_NODELAY;
663*d9f75844SAndroid Build Coastguard Worker break;
664*d9f75844SAndroid Build Coastguard Worker case OPT_DSCP:
665*d9f75844SAndroid Build Coastguard Worker #if defined(WEBRTC_POSIX)
666*d9f75844SAndroid Build Coastguard Worker if (family_ == AF_INET6) {
667*d9f75844SAndroid Build Coastguard Worker *slevel = IPPROTO_IPV6;
668*d9f75844SAndroid Build Coastguard Worker *sopt = IPV6_TCLASS;
669*d9f75844SAndroid Build Coastguard Worker } else {
670*d9f75844SAndroid Build Coastguard Worker *slevel = IPPROTO_IP;
671*d9f75844SAndroid Build Coastguard Worker *sopt = IP_TOS;
672*d9f75844SAndroid Build Coastguard Worker }
673*d9f75844SAndroid Build Coastguard Worker break;
674*d9f75844SAndroid Build Coastguard Worker #else
675*d9f75844SAndroid Build Coastguard Worker RTC_LOG(LS_WARNING) << "Socket::OPT_DSCP not supported.";
676*d9f75844SAndroid Build Coastguard Worker return -1;
677*d9f75844SAndroid Build Coastguard Worker #endif
678*d9f75844SAndroid Build Coastguard Worker case OPT_RTP_SENDTIME_EXTN_ID:
679*d9f75844SAndroid Build Coastguard Worker return -1; // No logging is necessary as this not a OS socket option.
680*d9f75844SAndroid Build Coastguard Worker default:
681*d9f75844SAndroid Build Coastguard Worker RTC_DCHECK_NOTREACHED();
682*d9f75844SAndroid Build Coastguard Worker return -1;
683*d9f75844SAndroid Build Coastguard Worker }
684*d9f75844SAndroid Build Coastguard Worker return 0;
685*d9f75844SAndroid Build Coastguard Worker }
686*d9f75844SAndroid Build Coastguard Worker
SocketDispatcher(PhysicalSocketServer * ss)687*d9f75844SAndroid Build Coastguard Worker SocketDispatcher::SocketDispatcher(PhysicalSocketServer* ss)
688*d9f75844SAndroid Build Coastguard Worker #if defined(WEBRTC_WIN)
689*d9f75844SAndroid Build Coastguard Worker : PhysicalSocket(ss),
690*d9f75844SAndroid Build Coastguard Worker id_(0),
691*d9f75844SAndroid Build Coastguard Worker signal_close_(false)
692*d9f75844SAndroid Build Coastguard Worker #else
693*d9f75844SAndroid Build Coastguard Worker : PhysicalSocket(ss)
694*d9f75844SAndroid Build Coastguard Worker #endif
695*d9f75844SAndroid Build Coastguard Worker {
696*d9f75844SAndroid Build Coastguard Worker }
697*d9f75844SAndroid Build Coastguard Worker
SocketDispatcher(SOCKET s,PhysicalSocketServer * ss)698*d9f75844SAndroid Build Coastguard Worker SocketDispatcher::SocketDispatcher(SOCKET s, PhysicalSocketServer* ss)
699*d9f75844SAndroid Build Coastguard Worker #if defined(WEBRTC_WIN)
700*d9f75844SAndroid Build Coastguard Worker : PhysicalSocket(ss, s),
701*d9f75844SAndroid Build Coastguard Worker id_(0),
702*d9f75844SAndroid Build Coastguard Worker signal_close_(false)
703*d9f75844SAndroid Build Coastguard Worker #else
704*d9f75844SAndroid Build Coastguard Worker : PhysicalSocket(ss, s)
705*d9f75844SAndroid Build Coastguard Worker #endif
706*d9f75844SAndroid Build Coastguard Worker {
707*d9f75844SAndroid Build Coastguard Worker }
708*d9f75844SAndroid Build Coastguard Worker
~SocketDispatcher()709*d9f75844SAndroid Build Coastguard Worker SocketDispatcher::~SocketDispatcher() {
710*d9f75844SAndroid Build Coastguard Worker Close();
711*d9f75844SAndroid Build Coastguard Worker }
712*d9f75844SAndroid Build Coastguard Worker
Initialize()713*d9f75844SAndroid Build Coastguard Worker bool SocketDispatcher::Initialize() {
714*d9f75844SAndroid Build Coastguard Worker RTC_DCHECK(s_ != INVALID_SOCKET);
715*d9f75844SAndroid Build Coastguard Worker // Must be a non-blocking
716*d9f75844SAndroid Build Coastguard Worker #if defined(WEBRTC_WIN)
717*d9f75844SAndroid Build Coastguard Worker u_long argp = 1;
718*d9f75844SAndroid Build Coastguard Worker ioctlsocket(s_, FIONBIO, &argp);
719*d9f75844SAndroid Build Coastguard Worker #elif defined(WEBRTC_POSIX)
720*d9f75844SAndroid Build Coastguard Worker fcntl(s_, F_SETFL, fcntl(s_, F_GETFL, 0) | O_NONBLOCK);
721*d9f75844SAndroid Build Coastguard Worker if (IsScmTimeStampExperimentEnabled()) {
722*d9f75844SAndroid Build Coastguard Worker int value = 1;
723*d9f75844SAndroid Build Coastguard Worker // Attempt to get receive packet timestamp from the socket.
724*d9f75844SAndroid Build Coastguard Worker if (::setsockopt(s_, SOL_SOCKET, SO_TIMESTAMP, &value, sizeof(value)) !=
725*d9f75844SAndroid Build Coastguard Worker 0) {
726*d9f75844SAndroid Build Coastguard Worker RTC_DLOG(LS_ERROR) << "::setsockopt failed. errno: " << LAST_SYSTEM_ERROR;
727*d9f75844SAndroid Build Coastguard Worker }
728*d9f75844SAndroid Build Coastguard Worker }
729*d9f75844SAndroid Build Coastguard Worker #endif
730*d9f75844SAndroid Build Coastguard Worker
731*d9f75844SAndroid Build Coastguard Worker #if defined(WEBRTC_IOS)
732*d9f75844SAndroid Build Coastguard Worker // iOS may kill sockets when the app is moved to the background
733*d9f75844SAndroid Build Coastguard Worker // (specifically, if the app doesn't use the "voip" UIBackgroundMode). When
734*d9f75844SAndroid Build Coastguard Worker // we attempt to write to such a socket, SIGPIPE will be raised, which by
735*d9f75844SAndroid Build Coastguard Worker // default will terminate the process, which we don't want. By specifying
736*d9f75844SAndroid Build Coastguard Worker // this socket option, SIGPIPE will be disabled for the socket.
737*d9f75844SAndroid Build Coastguard Worker int value = 1;
738*d9f75844SAndroid Build Coastguard Worker if (::setsockopt(s_, SOL_SOCKET, SO_NOSIGPIPE, &value, sizeof(value)) != 0) {
739*d9f75844SAndroid Build Coastguard Worker RTC_DLOG(LS_ERROR) << "::setsockopt failed. errno: " << LAST_SYSTEM_ERROR;
740*d9f75844SAndroid Build Coastguard Worker }
741*d9f75844SAndroid Build Coastguard Worker #endif
742*d9f75844SAndroid Build Coastguard Worker ss_->Add(this);
743*d9f75844SAndroid Build Coastguard Worker return true;
744*d9f75844SAndroid Build Coastguard Worker }
745*d9f75844SAndroid Build Coastguard Worker
Create(int type)746*d9f75844SAndroid Build Coastguard Worker bool SocketDispatcher::Create(int type) {
747*d9f75844SAndroid Build Coastguard Worker return Create(AF_INET, type);
748*d9f75844SAndroid Build Coastguard Worker }
749*d9f75844SAndroid Build Coastguard Worker
Create(int family,int type)750*d9f75844SAndroid Build Coastguard Worker bool SocketDispatcher::Create(int family, int type) {
751*d9f75844SAndroid Build Coastguard Worker // Change the socket to be non-blocking.
752*d9f75844SAndroid Build Coastguard Worker if (!PhysicalSocket::Create(family, type))
753*d9f75844SAndroid Build Coastguard Worker return false;
754*d9f75844SAndroid Build Coastguard Worker
755*d9f75844SAndroid Build Coastguard Worker if (!Initialize())
756*d9f75844SAndroid Build Coastguard Worker return false;
757*d9f75844SAndroid Build Coastguard Worker
758*d9f75844SAndroid Build Coastguard Worker #if defined(WEBRTC_WIN)
759*d9f75844SAndroid Build Coastguard Worker do {
760*d9f75844SAndroid Build Coastguard Worker id_ = ++next_id_;
761*d9f75844SAndroid Build Coastguard Worker } while (id_ == 0);
762*d9f75844SAndroid Build Coastguard Worker #endif
763*d9f75844SAndroid Build Coastguard Worker return true;
764*d9f75844SAndroid Build Coastguard Worker }
765*d9f75844SAndroid Build Coastguard Worker
766*d9f75844SAndroid Build Coastguard Worker #if defined(WEBRTC_WIN)
767*d9f75844SAndroid Build Coastguard Worker
GetWSAEvent()768*d9f75844SAndroid Build Coastguard Worker WSAEVENT SocketDispatcher::GetWSAEvent() {
769*d9f75844SAndroid Build Coastguard Worker return WSA_INVALID_EVENT;
770*d9f75844SAndroid Build Coastguard Worker }
771*d9f75844SAndroid Build Coastguard Worker
GetSocket()772*d9f75844SAndroid Build Coastguard Worker SOCKET SocketDispatcher::GetSocket() {
773*d9f75844SAndroid Build Coastguard Worker return s_;
774*d9f75844SAndroid Build Coastguard Worker }
775*d9f75844SAndroid Build Coastguard Worker
CheckSignalClose()776*d9f75844SAndroid Build Coastguard Worker bool SocketDispatcher::CheckSignalClose() {
777*d9f75844SAndroid Build Coastguard Worker if (!signal_close_)
778*d9f75844SAndroid Build Coastguard Worker return false;
779*d9f75844SAndroid Build Coastguard Worker
780*d9f75844SAndroid Build Coastguard Worker char ch;
781*d9f75844SAndroid Build Coastguard Worker if (recv(s_, &ch, 1, MSG_PEEK) > 0)
782*d9f75844SAndroid Build Coastguard Worker return false;
783*d9f75844SAndroid Build Coastguard Worker
784*d9f75844SAndroid Build Coastguard Worker state_ = CS_CLOSED;
785*d9f75844SAndroid Build Coastguard Worker signal_close_ = false;
786*d9f75844SAndroid Build Coastguard Worker SignalCloseEvent(this, signal_err_);
787*d9f75844SAndroid Build Coastguard Worker return true;
788*d9f75844SAndroid Build Coastguard Worker }
789*d9f75844SAndroid Build Coastguard Worker
790*d9f75844SAndroid Build Coastguard Worker int SocketDispatcher::next_id_ = 0;
791*d9f75844SAndroid Build Coastguard Worker
792*d9f75844SAndroid Build Coastguard Worker #elif defined(WEBRTC_POSIX)
793*d9f75844SAndroid Build Coastguard Worker
GetDescriptor()794*d9f75844SAndroid Build Coastguard Worker int SocketDispatcher::GetDescriptor() {
795*d9f75844SAndroid Build Coastguard Worker return s_;
796*d9f75844SAndroid Build Coastguard Worker }
797*d9f75844SAndroid Build Coastguard Worker
IsDescriptorClosed()798*d9f75844SAndroid Build Coastguard Worker bool SocketDispatcher::IsDescriptorClosed() {
799*d9f75844SAndroid Build Coastguard Worker if (udp_) {
800*d9f75844SAndroid Build Coastguard Worker // The MSG_PEEK trick doesn't work for UDP, since (at least in some
801*d9f75844SAndroid Build Coastguard Worker // circumstances) it requires reading an entire UDP packet, which would be
802*d9f75844SAndroid Build Coastguard Worker // bad for performance here. So, just check whether `s_` has been closed,
803*d9f75844SAndroid Build Coastguard Worker // which should be sufficient.
804*d9f75844SAndroid Build Coastguard Worker return s_ == INVALID_SOCKET;
805*d9f75844SAndroid Build Coastguard Worker }
806*d9f75844SAndroid Build Coastguard Worker // We don't have a reliable way of distinguishing end-of-stream
807*d9f75844SAndroid Build Coastguard Worker // from readability. So test on each readable call. Is this
808*d9f75844SAndroid Build Coastguard Worker // inefficient? Probably.
809*d9f75844SAndroid Build Coastguard Worker char ch;
810*d9f75844SAndroid Build Coastguard Worker ssize_t res;
811*d9f75844SAndroid Build Coastguard Worker // Retry if the system call was interrupted.
812*d9f75844SAndroid Build Coastguard Worker do {
813*d9f75844SAndroid Build Coastguard Worker res = ::recv(s_, &ch, 1, MSG_PEEK);
814*d9f75844SAndroid Build Coastguard Worker } while (res < 0 && errno == EINTR);
815*d9f75844SAndroid Build Coastguard Worker if (res > 0) {
816*d9f75844SAndroid Build Coastguard Worker // Data available, so not closed.
817*d9f75844SAndroid Build Coastguard Worker return false;
818*d9f75844SAndroid Build Coastguard Worker } else if (res == 0) {
819*d9f75844SAndroid Build Coastguard Worker // EOF, so closed.
820*d9f75844SAndroid Build Coastguard Worker return true;
821*d9f75844SAndroid Build Coastguard Worker } else { // error
822*d9f75844SAndroid Build Coastguard Worker switch (errno) {
823*d9f75844SAndroid Build Coastguard Worker // Returned if we've already closed s_.
824*d9f75844SAndroid Build Coastguard Worker case EBADF:
825*d9f75844SAndroid Build Coastguard Worker // This is dangerous: if we keep attempting to access a FD after close,
826*d9f75844SAndroid Build Coastguard Worker // it could be reopened by something else making us think it's still
827*d9f75844SAndroid Build Coastguard Worker // open. Note that this is only a DCHECK.
828*d9f75844SAndroid Build Coastguard Worker RTC_DCHECK_NOTREACHED();
829*d9f75844SAndroid Build Coastguard Worker return true;
830*d9f75844SAndroid Build Coastguard Worker // Returned during ungraceful peer shutdown.
831*d9f75844SAndroid Build Coastguard Worker case ECONNRESET:
832*d9f75844SAndroid Build Coastguard Worker return true;
833*d9f75844SAndroid Build Coastguard Worker case ECONNABORTED:
834*d9f75844SAndroid Build Coastguard Worker return true;
835*d9f75844SAndroid Build Coastguard Worker case EPIPE:
836*d9f75844SAndroid Build Coastguard Worker return true;
837*d9f75844SAndroid Build Coastguard Worker // The normal blocking error; don't log anything.
838*d9f75844SAndroid Build Coastguard Worker case EWOULDBLOCK:
839*d9f75844SAndroid Build Coastguard Worker return false;
840*d9f75844SAndroid Build Coastguard Worker default:
841*d9f75844SAndroid Build Coastguard Worker // Assume that all other errors are just blocking errors, meaning the
842*d9f75844SAndroid Build Coastguard Worker // connection is still good but we just can't read from it right now.
843*d9f75844SAndroid Build Coastguard Worker // This should only happen when connecting (and at most once), because
844*d9f75844SAndroid Build Coastguard Worker // in all other cases this function is only called if the file
845*d9f75844SAndroid Build Coastguard Worker // descriptor is already known to be in the readable state. However,
846*d9f75844SAndroid Build Coastguard Worker // it's not necessary a problem if we spuriously interpret a
847*d9f75844SAndroid Build Coastguard Worker // "connection lost"-type error as a blocking error, because typically
848*d9f75844SAndroid Build Coastguard Worker // the next recv() will get EOF, so we'll still eventually notice that
849*d9f75844SAndroid Build Coastguard Worker // the socket is closed.
850*d9f75844SAndroid Build Coastguard Worker RTC_LOG_ERR(LS_WARNING) << "Assuming benign blocking error";
851*d9f75844SAndroid Build Coastguard Worker return false;
852*d9f75844SAndroid Build Coastguard Worker }
853*d9f75844SAndroid Build Coastguard Worker }
854*d9f75844SAndroid Build Coastguard Worker }
855*d9f75844SAndroid Build Coastguard Worker
856*d9f75844SAndroid Build Coastguard Worker #endif // WEBRTC_POSIX
857*d9f75844SAndroid Build Coastguard Worker
GetRequestedEvents()858*d9f75844SAndroid Build Coastguard Worker uint32_t SocketDispatcher::GetRequestedEvents() {
859*d9f75844SAndroid Build Coastguard Worker return enabled_events();
860*d9f75844SAndroid Build Coastguard Worker }
861*d9f75844SAndroid Build Coastguard Worker
862*d9f75844SAndroid Build Coastguard Worker #if defined(WEBRTC_WIN)
863*d9f75844SAndroid Build Coastguard Worker
OnEvent(uint32_t ff,int err)864*d9f75844SAndroid Build Coastguard Worker void SocketDispatcher::OnEvent(uint32_t ff, int err) {
865*d9f75844SAndroid Build Coastguard Worker if ((ff & DE_CONNECT) != 0)
866*d9f75844SAndroid Build Coastguard Worker state_ = CS_CONNECTED;
867*d9f75844SAndroid Build Coastguard Worker
868*d9f75844SAndroid Build Coastguard Worker // We set CS_CLOSED from CheckSignalClose.
869*d9f75844SAndroid Build Coastguard Worker
870*d9f75844SAndroid Build Coastguard Worker int cache_id = id_;
871*d9f75844SAndroid Build Coastguard Worker // Make sure we deliver connect/accept first. Otherwise, consumers may see
872*d9f75844SAndroid Build Coastguard Worker // something like a READ followed by a CONNECT, which would be odd.
873*d9f75844SAndroid Build Coastguard Worker if (((ff & DE_CONNECT) != 0) && (id_ == cache_id)) {
874*d9f75844SAndroid Build Coastguard Worker if (ff != DE_CONNECT)
875*d9f75844SAndroid Build Coastguard Worker RTC_LOG(LS_VERBOSE) << "Signalled with DE_CONNECT: " << ff;
876*d9f75844SAndroid Build Coastguard Worker DisableEvents(DE_CONNECT);
877*d9f75844SAndroid Build Coastguard Worker #if !defined(NDEBUG)
878*d9f75844SAndroid Build Coastguard Worker dbg_addr_ = "Connected @ ";
879*d9f75844SAndroid Build Coastguard Worker dbg_addr_.append(GetRemoteAddress().ToString());
880*d9f75844SAndroid Build Coastguard Worker #endif
881*d9f75844SAndroid Build Coastguard Worker SignalConnectEvent(this);
882*d9f75844SAndroid Build Coastguard Worker }
883*d9f75844SAndroid Build Coastguard Worker if (((ff & DE_ACCEPT) != 0) && (id_ == cache_id)) {
884*d9f75844SAndroid Build Coastguard Worker DisableEvents(DE_ACCEPT);
885*d9f75844SAndroid Build Coastguard Worker SignalReadEvent(this);
886*d9f75844SAndroid Build Coastguard Worker }
887*d9f75844SAndroid Build Coastguard Worker if ((ff & DE_READ) != 0) {
888*d9f75844SAndroid Build Coastguard Worker DisableEvents(DE_READ);
889*d9f75844SAndroid Build Coastguard Worker SignalReadEvent(this);
890*d9f75844SAndroid Build Coastguard Worker }
891*d9f75844SAndroid Build Coastguard Worker if (((ff & DE_WRITE) != 0) && (id_ == cache_id)) {
892*d9f75844SAndroid Build Coastguard Worker DisableEvents(DE_WRITE);
893*d9f75844SAndroid Build Coastguard Worker SignalWriteEvent(this);
894*d9f75844SAndroid Build Coastguard Worker }
895*d9f75844SAndroid Build Coastguard Worker if (((ff & DE_CLOSE) != 0) && (id_ == cache_id)) {
896*d9f75844SAndroid Build Coastguard Worker signal_close_ = true;
897*d9f75844SAndroid Build Coastguard Worker signal_err_ = err;
898*d9f75844SAndroid Build Coastguard Worker }
899*d9f75844SAndroid Build Coastguard Worker }
900*d9f75844SAndroid Build Coastguard Worker
901*d9f75844SAndroid Build Coastguard Worker #elif defined(WEBRTC_POSIX)
902*d9f75844SAndroid Build Coastguard Worker
OnEvent(uint32_t ff,int err)903*d9f75844SAndroid Build Coastguard Worker void SocketDispatcher::OnEvent(uint32_t ff, int err) {
904*d9f75844SAndroid Build Coastguard Worker if ((ff & DE_CONNECT) != 0)
905*d9f75844SAndroid Build Coastguard Worker state_ = CS_CONNECTED;
906*d9f75844SAndroid Build Coastguard Worker
907*d9f75844SAndroid Build Coastguard Worker if ((ff & DE_CLOSE) != 0)
908*d9f75844SAndroid Build Coastguard Worker state_ = CS_CLOSED;
909*d9f75844SAndroid Build Coastguard Worker
910*d9f75844SAndroid Build Coastguard Worker #if defined(WEBRTC_USE_EPOLL)
911*d9f75844SAndroid Build Coastguard Worker // Remember currently enabled events so we can combine multiple changes
912*d9f75844SAndroid Build Coastguard Worker // into one update call later.
913*d9f75844SAndroid Build Coastguard Worker // The signal handlers might re-enable events disabled here, so we can't
914*d9f75844SAndroid Build Coastguard Worker // keep a list of events to disable at the end of the method. This list
915*d9f75844SAndroid Build Coastguard Worker // would not be updated with the events enabled by the signal handlers.
916*d9f75844SAndroid Build Coastguard Worker StartBatchedEventUpdates();
917*d9f75844SAndroid Build Coastguard Worker #endif
918*d9f75844SAndroid Build Coastguard Worker // Make sure we deliver connect/accept first. Otherwise, consumers may see
919*d9f75844SAndroid Build Coastguard Worker // something like a READ followed by a CONNECT, which would be odd.
920*d9f75844SAndroid Build Coastguard Worker if ((ff & DE_CONNECT) != 0) {
921*d9f75844SAndroid Build Coastguard Worker DisableEvents(DE_CONNECT);
922*d9f75844SAndroid Build Coastguard Worker SignalConnectEvent(this);
923*d9f75844SAndroid Build Coastguard Worker }
924*d9f75844SAndroid Build Coastguard Worker if ((ff & DE_ACCEPT) != 0) {
925*d9f75844SAndroid Build Coastguard Worker DisableEvents(DE_ACCEPT);
926*d9f75844SAndroid Build Coastguard Worker SignalReadEvent(this);
927*d9f75844SAndroid Build Coastguard Worker }
928*d9f75844SAndroid Build Coastguard Worker if ((ff & DE_READ) != 0) {
929*d9f75844SAndroid Build Coastguard Worker DisableEvents(DE_READ);
930*d9f75844SAndroid Build Coastguard Worker SignalReadEvent(this);
931*d9f75844SAndroid Build Coastguard Worker }
932*d9f75844SAndroid Build Coastguard Worker if ((ff & DE_WRITE) != 0) {
933*d9f75844SAndroid Build Coastguard Worker DisableEvents(DE_WRITE);
934*d9f75844SAndroid Build Coastguard Worker SignalWriteEvent(this);
935*d9f75844SAndroid Build Coastguard Worker }
936*d9f75844SAndroid Build Coastguard Worker if ((ff & DE_CLOSE) != 0) {
937*d9f75844SAndroid Build Coastguard Worker // The socket is now dead to us, so stop checking it.
938*d9f75844SAndroid Build Coastguard Worker SetEnabledEvents(0);
939*d9f75844SAndroid Build Coastguard Worker SignalCloseEvent(this, err);
940*d9f75844SAndroid Build Coastguard Worker }
941*d9f75844SAndroid Build Coastguard Worker #if defined(WEBRTC_USE_EPOLL)
942*d9f75844SAndroid Build Coastguard Worker FinishBatchedEventUpdates();
943*d9f75844SAndroid Build Coastguard Worker #endif
944*d9f75844SAndroid Build Coastguard Worker }
945*d9f75844SAndroid Build Coastguard Worker
946*d9f75844SAndroid Build Coastguard Worker #endif // WEBRTC_POSIX
947*d9f75844SAndroid Build Coastguard Worker
948*d9f75844SAndroid Build Coastguard Worker #if defined(WEBRTC_USE_EPOLL)
949*d9f75844SAndroid Build Coastguard Worker
GetEpollEvents(uint32_t ff)950*d9f75844SAndroid Build Coastguard Worker inline static int GetEpollEvents(uint32_t ff) {
951*d9f75844SAndroid Build Coastguard Worker int events = 0;
952*d9f75844SAndroid Build Coastguard Worker if (ff & (DE_READ | DE_ACCEPT)) {
953*d9f75844SAndroid Build Coastguard Worker events |= EPOLLIN;
954*d9f75844SAndroid Build Coastguard Worker }
955*d9f75844SAndroid Build Coastguard Worker if (ff & (DE_WRITE | DE_CONNECT)) {
956*d9f75844SAndroid Build Coastguard Worker events |= EPOLLOUT;
957*d9f75844SAndroid Build Coastguard Worker }
958*d9f75844SAndroid Build Coastguard Worker return events;
959*d9f75844SAndroid Build Coastguard Worker }
960*d9f75844SAndroid Build Coastguard Worker
StartBatchedEventUpdates()961*d9f75844SAndroid Build Coastguard Worker void SocketDispatcher::StartBatchedEventUpdates() {
962*d9f75844SAndroid Build Coastguard Worker RTC_DCHECK_EQ(saved_enabled_events_, -1);
963*d9f75844SAndroid Build Coastguard Worker saved_enabled_events_ = enabled_events();
964*d9f75844SAndroid Build Coastguard Worker }
965*d9f75844SAndroid Build Coastguard Worker
FinishBatchedEventUpdates()966*d9f75844SAndroid Build Coastguard Worker void SocketDispatcher::FinishBatchedEventUpdates() {
967*d9f75844SAndroid Build Coastguard Worker RTC_DCHECK_NE(saved_enabled_events_, -1);
968*d9f75844SAndroid Build Coastguard Worker uint8_t old_events = static_cast<uint8_t>(saved_enabled_events_);
969*d9f75844SAndroid Build Coastguard Worker saved_enabled_events_ = -1;
970*d9f75844SAndroid Build Coastguard Worker MaybeUpdateDispatcher(old_events);
971*d9f75844SAndroid Build Coastguard Worker }
972*d9f75844SAndroid Build Coastguard Worker
MaybeUpdateDispatcher(uint8_t old_events)973*d9f75844SAndroid Build Coastguard Worker void SocketDispatcher::MaybeUpdateDispatcher(uint8_t old_events) {
974*d9f75844SAndroid Build Coastguard Worker if (GetEpollEvents(enabled_events()) != GetEpollEvents(old_events) &&
975*d9f75844SAndroid Build Coastguard Worker saved_enabled_events_ == -1) {
976*d9f75844SAndroid Build Coastguard Worker ss_->Update(this);
977*d9f75844SAndroid Build Coastguard Worker }
978*d9f75844SAndroid Build Coastguard Worker }
979*d9f75844SAndroid Build Coastguard Worker
SetEnabledEvents(uint8_t events)980*d9f75844SAndroid Build Coastguard Worker void SocketDispatcher::SetEnabledEvents(uint8_t events) {
981*d9f75844SAndroid Build Coastguard Worker uint8_t old_events = enabled_events();
982*d9f75844SAndroid Build Coastguard Worker PhysicalSocket::SetEnabledEvents(events);
983*d9f75844SAndroid Build Coastguard Worker MaybeUpdateDispatcher(old_events);
984*d9f75844SAndroid Build Coastguard Worker }
985*d9f75844SAndroid Build Coastguard Worker
EnableEvents(uint8_t events)986*d9f75844SAndroid Build Coastguard Worker void SocketDispatcher::EnableEvents(uint8_t events) {
987*d9f75844SAndroid Build Coastguard Worker uint8_t old_events = enabled_events();
988*d9f75844SAndroid Build Coastguard Worker PhysicalSocket::EnableEvents(events);
989*d9f75844SAndroid Build Coastguard Worker MaybeUpdateDispatcher(old_events);
990*d9f75844SAndroid Build Coastguard Worker }
991*d9f75844SAndroid Build Coastguard Worker
DisableEvents(uint8_t events)992*d9f75844SAndroid Build Coastguard Worker void SocketDispatcher::DisableEvents(uint8_t events) {
993*d9f75844SAndroid Build Coastguard Worker uint8_t old_events = enabled_events();
994*d9f75844SAndroid Build Coastguard Worker PhysicalSocket::DisableEvents(events);
995*d9f75844SAndroid Build Coastguard Worker MaybeUpdateDispatcher(old_events);
996*d9f75844SAndroid Build Coastguard Worker }
997*d9f75844SAndroid Build Coastguard Worker
998*d9f75844SAndroid Build Coastguard Worker #endif // WEBRTC_USE_EPOLL
999*d9f75844SAndroid Build Coastguard Worker
Close()1000*d9f75844SAndroid Build Coastguard Worker int SocketDispatcher::Close() {
1001*d9f75844SAndroid Build Coastguard Worker if (s_ == INVALID_SOCKET)
1002*d9f75844SAndroid Build Coastguard Worker return 0;
1003*d9f75844SAndroid Build Coastguard Worker
1004*d9f75844SAndroid Build Coastguard Worker #if defined(WEBRTC_WIN)
1005*d9f75844SAndroid Build Coastguard Worker id_ = 0;
1006*d9f75844SAndroid Build Coastguard Worker signal_close_ = false;
1007*d9f75844SAndroid Build Coastguard Worker #endif
1008*d9f75844SAndroid Build Coastguard Worker #if defined(WEBRTC_USE_EPOLL)
1009*d9f75844SAndroid Build Coastguard Worker // If we're batching events, the socket can be closed and reopened
1010*d9f75844SAndroid Build Coastguard Worker // during the batch. Set saved_enabled_events_ to 0 here so the new
1011*d9f75844SAndroid Build Coastguard Worker // socket, if any, has the correct old events bitfield
1012*d9f75844SAndroid Build Coastguard Worker if (saved_enabled_events_ != -1) {
1013*d9f75844SAndroid Build Coastguard Worker saved_enabled_events_ = 0;
1014*d9f75844SAndroid Build Coastguard Worker }
1015*d9f75844SAndroid Build Coastguard Worker #endif
1016*d9f75844SAndroid Build Coastguard Worker ss_->Remove(this);
1017*d9f75844SAndroid Build Coastguard Worker return PhysicalSocket::Close();
1018*d9f75844SAndroid Build Coastguard Worker }
1019*d9f75844SAndroid Build Coastguard Worker
1020*d9f75844SAndroid Build Coastguard Worker #if defined(WEBRTC_POSIX)
1021*d9f75844SAndroid Build Coastguard Worker // Sets the value of a boolean value to false when signaled.
1022*d9f75844SAndroid Build Coastguard Worker class Signaler : public Dispatcher {
1023*d9f75844SAndroid Build Coastguard Worker public:
Signaler(PhysicalSocketServer * ss,bool & flag_to_clear)1024*d9f75844SAndroid Build Coastguard Worker Signaler(PhysicalSocketServer* ss, bool& flag_to_clear)
1025*d9f75844SAndroid Build Coastguard Worker : ss_(ss),
1026*d9f75844SAndroid Build Coastguard Worker afd_([] {
1027*d9f75844SAndroid Build Coastguard Worker std::array<int, 2> afd = {-1, -1};
1028*d9f75844SAndroid Build Coastguard Worker
1029*d9f75844SAndroid Build Coastguard Worker if (pipe(afd.data()) < 0) {
1030*d9f75844SAndroid Build Coastguard Worker RTC_LOG(LS_ERROR) << "pipe failed";
1031*d9f75844SAndroid Build Coastguard Worker }
1032*d9f75844SAndroid Build Coastguard Worker return afd;
1033*d9f75844SAndroid Build Coastguard Worker }()),
1034*d9f75844SAndroid Build Coastguard Worker fSignaled_(false),
1035*d9f75844SAndroid Build Coastguard Worker flag_to_clear_(flag_to_clear) {
1036*d9f75844SAndroid Build Coastguard Worker ss_->Add(this);
1037*d9f75844SAndroid Build Coastguard Worker }
1038*d9f75844SAndroid Build Coastguard Worker
~Signaler()1039*d9f75844SAndroid Build Coastguard Worker ~Signaler() override {
1040*d9f75844SAndroid Build Coastguard Worker ss_->Remove(this);
1041*d9f75844SAndroid Build Coastguard Worker close(afd_[0]);
1042*d9f75844SAndroid Build Coastguard Worker close(afd_[1]);
1043*d9f75844SAndroid Build Coastguard Worker }
1044*d9f75844SAndroid Build Coastguard Worker
Signal()1045*d9f75844SAndroid Build Coastguard Worker virtual void Signal() {
1046*d9f75844SAndroid Build Coastguard Worker webrtc::MutexLock lock(&mutex_);
1047*d9f75844SAndroid Build Coastguard Worker if (!fSignaled_) {
1048*d9f75844SAndroid Build Coastguard Worker const uint8_t b[1] = {0};
1049*d9f75844SAndroid Build Coastguard Worker const ssize_t res = write(afd_[1], b, sizeof(b));
1050*d9f75844SAndroid Build Coastguard Worker RTC_DCHECK_EQ(1, res);
1051*d9f75844SAndroid Build Coastguard Worker fSignaled_ = true;
1052*d9f75844SAndroid Build Coastguard Worker }
1053*d9f75844SAndroid Build Coastguard Worker }
1054*d9f75844SAndroid Build Coastguard Worker
GetRequestedEvents()1055*d9f75844SAndroid Build Coastguard Worker uint32_t GetRequestedEvents() override { return DE_READ; }
1056*d9f75844SAndroid Build Coastguard Worker
OnEvent(uint32_t ff,int err)1057*d9f75844SAndroid Build Coastguard Worker void OnEvent(uint32_t ff, int err) override {
1058*d9f75844SAndroid Build Coastguard Worker // It is not possible to perfectly emulate an auto-resetting event with
1059*d9f75844SAndroid Build Coastguard Worker // pipes. This simulates it by resetting before the event is handled.
1060*d9f75844SAndroid Build Coastguard Worker
1061*d9f75844SAndroid Build Coastguard Worker webrtc::MutexLock lock(&mutex_);
1062*d9f75844SAndroid Build Coastguard Worker if (fSignaled_) {
1063*d9f75844SAndroid Build Coastguard Worker uint8_t b[4]; // Allow for reading more than 1 byte, but expect 1.
1064*d9f75844SAndroid Build Coastguard Worker const ssize_t res = read(afd_[0], b, sizeof(b));
1065*d9f75844SAndroid Build Coastguard Worker RTC_DCHECK_EQ(1, res);
1066*d9f75844SAndroid Build Coastguard Worker fSignaled_ = false;
1067*d9f75844SAndroid Build Coastguard Worker }
1068*d9f75844SAndroid Build Coastguard Worker flag_to_clear_ = false;
1069*d9f75844SAndroid Build Coastguard Worker }
1070*d9f75844SAndroid Build Coastguard Worker
GetDescriptor()1071*d9f75844SAndroid Build Coastguard Worker int GetDescriptor() override { return afd_[0]; }
1072*d9f75844SAndroid Build Coastguard Worker
IsDescriptorClosed()1073*d9f75844SAndroid Build Coastguard Worker bool IsDescriptorClosed() override { return false; }
1074*d9f75844SAndroid Build Coastguard Worker
1075*d9f75844SAndroid Build Coastguard Worker private:
1076*d9f75844SAndroid Build Coastguard Worker PhysicalSocketServer* const ss_;
1077*d9f75844SAndroid Build Coastguard Worker const std::array<int, 2> afd_;
1078*d9f75844SAndroid Build Coastguard Worker bool fSignaled_ RTC_GUARDED_BY(mutex_);
1079*d9f75844SAndroid Build Coastguard Worker webrtc::Mutex mutex_;
1080*d9f75844SAndroid Build Coastguard Worker bool& flag_to_clear_;
1081*d9f75844SAndroid Build Coastguard Worker };
1082*d9f75844SAndroid Build Coastguard Worker
1083*d9f75844SAndroid Build Coastguard Worker #endif // WEBRTC_POSIX
1084*d9f75844SAndroid Build Coastguard Worker
1085*d9f75844SAndroid Build Coastguard Worker #if defined(WEBRTC_WIN)
FlagsToEvents(uint32_t events)1086*d9f75844SAndroid Build Coastguard Worker static uint32_t FlagsToEvents(uint32_t events) {
1087*d9f75844SAndroid Build Coastguard Worker uint32_t ffFD = FD_CLOSE;
1088*d9f75844SAndroid Build Coastguard Worker if (events & DE_READ)
1089*d9f75844SAndroid Build Coastguard Worker ffFD |= FD_READ;
1090*d9f75844SAndroid Build Coastguard Worker if (events & DE_WRITE)
1091*d9f75844SAndroid Build Coastguard Worker ffFD |= FD_WRITE;
1092*d9f75844SAndroid Build Coastguard Worker if (events & DE_CONNECT)
1093*d9f75844SAndroid Build Coastguard Worker ffFD |= FD_CONNECT;
1094*d9f75844SAndroid Build Coastguard Worker if (events & DE_ACCEPT)
1095*d9f75844SAndroid Build Coastguard Worker ffFD |= FD_ACCEPT;
1096*d9f75844SAndroid Build Coastguard Worker return ffFD;
1097*d9f75844SAndroid Build Coastguard Worker }
1098*d9f75844SAndroid Build Coastguard Worker
1099*d9f75844SAndroid Build Coastguard Worker // Sets the value of a boolean value to false when signaled.
1100*d9f75844SAndroid Build Coastguard Worker class Signaler : public Dispatcher {
1101*d9f75844SAndroid Build Coastguard Worker public:
Signaler(PhysicalSocketServer * ss,bool & flag_to_clear)1102*d9f75844SAndroid Build Coastguard Worker Signaler(PhysicalSocketServer* ss, bool& flag_to_clear)
1103*d9f75844SAndroid Build Coastguard Worker : ss_(ss), flag_to_clear_(flag_to_clear) {
1104*d9f75844SAndroid Build Coastguard Worker hev_ = WSACreateEvent();
1105*d9f75844SAndroid Build Coastguard Worker if (hev_) {
1106*d9f75844SAndroid Build Coastguard Worker ss_->Add(this);
1107*d9f75844SAndroid Build Coastguard Worker }
1108*d9f75844SAndroid Build Coastguard Worker }
1109*d9f75844SAndroid Build Coastguard Worker
~Signaler()1110*d9f75844SAndroid Build Coastguard Worker ~Signaler() override {
1111*d9f75844SAndroid Build Coastguard Worker if (hev_ != nullptr) {
1112*d9f75844SAndroid Build Coastguard Worker ss_->Remove(this);
1113*d9f75844SAndroid Build Coastguard Worker WSACloseEvent(hev_);
1114*d9f75844SAndroid Build Coastguard Worker hev_ = nullptr;
1115*d9f75844SAndroid Build Coastguard Worker }
1116*d9f75844SAndroid Build Coastguard Worker }
1117*d9f75844SAndroid Build Coastguard Worker
Signal()1118*d9f75844SAndroid Build Coastguard Worker virtual void Signal() {
1119*d9f75844SAndroid Build Coastguard Worker if (hev_ != nullptr)
1120*d9f75844SAndroid Build Coastguard Worker WSASetEvent(hev_);
1121*d9f75844SAndroid Build Coastguard Worker }
1122*d9f75844SAndroid Build Coastguard Worker
GetRequestedEvents()1123*d9f75844SAndroid Build Coastguard Worker uint32_t GetRequestedEvents() override { return 0; }
1124*d9f75844SAndroid Build Coastguard Worker
OnEvent(uint32_t ff,int err)1125*d9f75844SAndroid Build Coastguard Worker void OnEvent(uint32_t ff, int err) override {
1126*d9f75844SAndroid Build Coastguard Worker WSAResetEvent(hev_);
1127*d9f75844SAndroid Build Coastguard Worker flag_to_clear_ = false;
1128*d9f75844SAndroid Build Coastguard Worker }
1129*d9f75844SAndroid Build Coastguard Worker
GetWSAEvent()1130*d9f75844SAndroid Build Coastguard Worker WSAEVENT GetWSAEvent() override { return hev_; }
1131*d9f75844SAndroid Build Coastguard Worker
GetSocket()1132*d9f75844SAndroid Build Coastguard Worker SOCKET GetSocket() override { return INVALID_SOCKET; }
1133*d9f75844SAndroid Build Coastguard Worker
CheckSignalClose()1134*d9f75844SAndroid Build Coastguard Worker bool CheckSignalClose() override { return false; }
1135*d9f75844SAndroid Build Coastguard Worker
1136*d9f75844SAndroid Build Coastguard Worker private:
1137*d9f75844SAndroid Build Coastguard Worker PhysicalSocketServer* ss_;
1138*d9f75844SAndroid Build Coastguard Worker WSAEVENT hev_;
1139*d9f75844SAndroid Build Coastguard Worker bool& flag_to_clear_;
1140*d9f75844SAndroid Build Coastguard Worker };
1141*d9f75844SAndroid Build Coastguard Worker #endif // WEBRTC_WIN
1142*d9f75844SAndroid Build Coastguard Worker
PhysicalSocketServer()1143*d9f75844SAndroid Build Coastguard Worker PhysicalSocketServer::PhysicalSocketServer()
1144*d9f75844SAndroid Build Coastguard Worker :
1145*d9f75844SAndroid Build Coastguard Worker #if defined(WEBRTC_USE_EPOLL)
1146*d9f75844SAndroid Build Coastguard Worker // Since Linux 2.6.8, the size argument is ignored, but must be greater
1147*d9f75844SAndroid Build Coastguard Worker // than zero. Before that the size served as hint to the kernel for the
1148*d9f75844SAndroid Build Coastguard Worker // amount of space to initially allocate in internal data structures.
1149*d9f75844SAndroid Build Coastguard Worker epoll_fd_(epoll_create(FD_SETSIZE)),
1150*d9f75844SAndroid Build Coastguard Worker #endif
1151*d9f75844SAndroid Build Coastguard Worker #if defined(WEBRTC_WIN)
1152*d9f75844SAndroid Build Coastguard Worker socket_ev_(WSACreateEvent()),
1153*d9f75844SAndroid Build Coastguard Worker #endif
1154*d9f75844SAndroid Build Coastguard Worker fWait_(false) {
1155*d9f75844SAndroid Build Coastguard Worker #if defined(WEBRTC_USE_EPOLL)
1156*d9f75844SAndroid Build Coastguard Worker if (epoll_fd_ == -1) {
1157*d9f75844SAndroid Build Coastguard Worker // Not an error, will fall back to "select" below.
1158*d9f75844SAndroid Build Coastguard Worker RTC_LOG_E(LS_WARNING, EN, errno) << "epoll_create";
1159*d9f75844SAndroid Build Coastguard Worker // Note that -1 == INVALID_SOCKET, the alias used by later checks.
1160*d9f75844SAndroid Build Coastguard Worker }
1161*d9f75844SAndroid Build Coastguard Worker #endif
1162*d9f75844SAndroid Build Coastguard Worker // The `fWait_` flag to be cleared by the Signaler.
1163*d9f75844SAndroid Build Coastguard Worker signal_wakeup_ = new Signaler(this, fWait_);
1164*d9f75844SAndroid Build Coastguard Worker }
1165*d9f75844SAndroid Build Coastguard Worker
~PhysicalSocketServer()1166*d9f75844SAndroid Build Coastguard Worker PhysicalSocketServer::~PhysicalSocketServer() {
1167*d9f75844SAndroid Build Coastguard Worker #if defined(WEBRTC_WIN)
1168*d9f75844SAndroid Build Coastguard Worker WSACloseEvent(socket_ev_);
1169*d9f75844SAndroid Build Coastguard Worker #endif
1170*d9f75844SAndroid Build Coastguard Worker delete signal_wakeup_;
1171*d9f75844SAndroid Build Coastguard Worker #if defined(WEBRTC_USE_EPOLL)
1172*d9f75844SAndroid Build Coastguard Worker if (epoll_fd_ != INVALID_SOCKET) {
1173*d9f75844SAndroid Build Coastguard Worker close(epoll_fd_);
1174*d9f75844SAndroid Build Coastguard Worker }
1175*d9f75844SAndroid Build Coastguard Worker #endif
1176*d9f75844SAndroid Build Coastguard Worker RTC_DCHECK(dispatcher_by_key_.empty());
1177*d9f75844SAndroid Build Coastguard Worker RTC_DCHECK(key_by_dispatcher_.empty());
1178*d9f75844SAndroid Build Coastguard Worker }
1179*d9f75844SAndroid Build Coastguard Worker
WakeUp()1180*d9f75844SAndroid Build Coastguard Worker void PhysicalSocketServer::WakeUp() {
1181*d9f75844SAndroid Build Coastguard Worker signal_wakeup_->Signal();
1182*d9f75844SAndroid Build Coastguard Worker }
1183*d9f75844SAndroid Build Coastguard Worker
CreateSocket(int family,int type)1184*d9f75844SAndroid Build Coastguard Worker Socket* PhysicalSocketServer::CreateSocket(int family, int type) {
1185*d9f75844SAndroid Build Coastguard Worker SocketDispatcher* dispatcher = new SocketDispatcher(this);
1186*d9f75844SAndroid Build Coastguard Worker if (dispatcher->Create(family, type)) {
1187*d9f75844SAndroid Build Coastguard Worker return dispatcher;
1188*d9f75844SAndroid Build Coastguard Worker } else {
1189*d9f75844SAndroid Build Coastguard Worker delete dispatcher;
1190*d9f75844SAndroid Build Coastguard Worker return nullptr;
1191*d9f75844SAndroid Build Coastguard Worker }
1192*d9f75844SAndroid Build Coastguard Worker }
1193*d9f75844SAndroid Build Coastguard Worker
WrapSocket(SOCKET s)1194*d9f75844SAndroid Build Coastguard Worker Socket* PhysicalSocketServer::WrapSocket(SOCKET s) {
1195*d9f75844SAndroid Build Coastguard Worker SocketDispatcher* dispatcher = new SocketDispatcher(s, this);
1196*d9f75844SAndroid Build Coastguard Worker if (dispatcher->Initialize()) {
1197*d9f75844SAndroid Build Coastguard Worker return dispatcher;
1198*d9f75844SAndroid Build Coastguard Worker } else {
1199*d9f75844SAndroid Build Coastguard Worker delete dispatcher;
1200*d9f75844SAndroid Build Coastguard Worker return nullptr;
1201*d9f75844SAndroid Build Coastguard Worker }
1202*d9f75844SAndroid Build Coastguard Worker }
1203*d9f75844SAndroid Build Coastguard Worker
Add(Dispatcher * pdispatcher)1204*d9f75844SAndroid Build Coastguard Worker void PhysicalSocketServer::Add(Dispatcher* pdispatcher) {
1205*d9f75844SAndroid Build Coastguard Worker CritScope cs(&crit_);
1206*d9f75844SAndroid Build Coastguard Worker if (key_by_dispatcher_.count(pdispatcher)) {
1207*d9f75844SAndroid Build Coastguard Worker RTC_LOG(LS_WARNING)
1208*d9f75844SAndroid Build Coastguard Worker << "PhysicalSocketServer asked to add a duplicate dispatcher.";
1209*d9f75844SAndroid Build Coastguard Worker return;
1210*d9f75844SAndroid Build Coastguard Worker }
1211*d9f75844SAndroid Build Coastguard Worker uint64_t key = next_dispatcher_key_++;
1212*d9f75844SAndroid Build Coastguard Worker dispatcher_by_key_.emplace(key, pdispatcher);
1213*d9f75844SAndroid Build Coastguard Worker key_by_dispatcher_.emplace(pdispatcher, key);
1214*d9f75844SAndroid Build Coastguard Worker #if defined(WEBRTC_USE_EPOLL)
1215*d9f75844SAndroid Build Coastguard Worker if (epoll_fd_ != INVALID_SOCKET) {
1216*d9f75844SAndroid Build Coastguard Worker AddEpoll(pdispatcher, key);
1217*d9f75844SAndroid Build Coastguard Worker }
1218*d9f75844SAndroid Build Coastguard Worker #endif // WEBRTC_USE_EPOLL
1219*d9f75844SAndroid Build Coastguard Worker }
1220*d9f75844SAndroid Build Coastguard Worker
Remove(Dispatcher * pdispatcher)1221*d9f75844SAndroid Build Coastguard Worker void PhysicalSocketServer::Remove(Dispatcher* pdispatcher) {
1222*d9f75844SAndroid Build Coastguard Worker CritScope cs(&crit_);
1223*d9f75844SAndroid Build Coastguard Worker if (!key_by_dispatcher_.count(pdispatcher)) {
1224*d9f75844SAndroid Build Coastguard Worker RTC_LOG(LS_WARNING)
1225*d9f75844SAndroid Build Coastguard Worker << "PhysicalSocketServer asked to remove a unknown "
1226*d9f75844SAndroid Build Coastguard Worker "dispatcher, potentially from a duplicate call to Add.";
1227*d9f75844SAndroid Build Coastguard Worker return;
1228*d9f75844SAndroid Build Coastguard Worker }
1229*d9f75844SAndroid Build Coastguard Worker uint64_t key = key_by_dispatcher_.at(pdispatcher);
1230*d9f75844SAndroid Build Coastguard Worker key_by_dispatcher_.erase(pdispatcher);
1231*d9f75844SAndroid Build Coastguard Worker dispatcher_by_key_.erase(key);
1232*d9f75844SAndroid Build Coastguard Worker #if defined(WEBRTC_USE_EPOLL)
1233*d9f75844SAndroid Build Coastguard Worker if (epoll_fd_ != INVALID_SOCKET) {
1234*d9f75844SAndroid Build Coastguard Worker RemoveEpoll(pdispatcher);
1235*d9f75844SAndroid Build Coastguard Worker }
1236*d9f75844SAndroid Build Coastguard Worker #endif // WEBRTC_USE_EPOLL
1237*d9f75844SAndroid Build Coastguard Worker }
1238*d9f75844SAndroid Build Coastguard Worker
Update(Dispatcher * pdispatcher)1239*d9f75844SAndroid Build Coastguard Worker void PhysicalSocketServer::Update(Dispatcher* pdispatcher) {
1240*d9f75844SAndroid Build Coastguard Worker #if defined(WEBRTC_USE_EPOLL)
1241*d9f75844SAndroid Build Coastguard Worker if (epoll_fd_ == INVALID_SOCKET) {
1242*d9f75844SAndroid Build Coastguard Worker return;
1243*d9f75844SAndroid Build Coastguard Worker }
1244*d9f75844SAndroid Build Coastguard Worker
1245*d9f75844SAndroid Build Coastguard Worker // Don't update dispatchers that haven't yet been added.
1246*d9f75844SAndroid Build Coastguard Worker CritScope cs(&crit_);
1247*d9f75844SAndroid Build Coastguard Worker if (!key_by_dispatcher_.count(pdispatcher)) {
1248*d9f75844SAndroid Build Coastguard Worker return;
1249*d9f75844SAndroid Build Coastguard Worker }
1250*d9f75844SAndroid Build Coastguard Worker
1251*d9f75844SAndroid Build Coastguard Worker UpdateEpoll(pdispatcher, key_by_dispatcher_.at(pdispatcher));
1252*d9f75844SAndroid Build Coastguard Worker #endif
1253*d9f75844SAndroid Build Coastguard Worker }
1254*d9f75844SAndroid Build Coastguard Worker
ToCmsWait(webrtc::TimeDelta max_wait_duration)1255*d9f75844SAndroid Build Coastguard Worker int PhysicalSocketServer::ToCmsWait(webrtc::TimeDelta max_wait_duration) {
1256*d9f75844SAndroid Build Coastguard Worker return max_wait_duration == Event::kForever
1257*d9f75844SAndroid Build Coastguard Worker ? kForeverMs
1258*d9f75844SAndroid Build Coastguard Worker : max_wait_duration.RoundUpTo(webrtc::TimeDelta::Millis(1)).ms();
1259*d9f75844SAndroid Build Coastguard Worker }
1260*d9f75844SAndroid Build Coastguard Worker
1261*d9f75844SAndroid Build Coastguard Worker #if defined(WEBRTC_POSIX)
1262*d9f75844SAndroid Build Coastguard Worker
Wait(webrtc::TimeDelta max_wait_duration,bool process_io)1263*d9f75844SAndroid Build Coastguard Worker bool PhysicalSocketServer::Wait(webrtc::TimeDelta max_wait_duration,
1264*d9f75844SAndroid Build Coastguard Worker bool process_io) {
1265*d9f75844SAndroid Build Coastguard Worker // We don't support reentrant waiting.
1266*d9f75844SAndroid Build Coastguard Worker RTC_DCHECK(!waiting_);
1267*d9f75844SAndroid Build Coastguard Worker ScopedSetTrue s(&waiting_);
1268*d9f75844SAndroid Build Coastguard Worker const int cmsWait = ToCmsWait(max_wait_duration);
1269*d9f75844SAndroid Build Coastguard Worker #if defined(WEBRTC_USE_EPOLL)
1270*d9f75844SAndroid Build Coastguard Worker // We don't keep a dedicated "epoll" descriptor containing only the non-IO
1271*d9f75844SAndroid Build Coastguard Worker // (i.e. signaling) dispatcher, so "poll" will be used instead of the default
1272*d9f75844SAndroid Build Coastguard Worker // "select" to support sockets larger than FD_SETSIZE.
1273*d9f75844SAndroid Build Coastguard Worker if (!process_io) {
1274*d9f75844SAndroid Build Coastguard Worker return WaitPoll(cmsWait, signal_wakeup_);
1275*d9f75844SAndroid Build Coastguard Worker } else if (epoll_fd_ != INVALID_SOCKET) {
1276*d9f75844SAndroid Build Coastguard Worker return WaitEpoll(cmsWait);
1277*d9f75844SAndroid Build Coastguard Worker }
1278*d9f75844SAndroid Build Coastguard Worker #endif
1279*d9f75844SAndroid Build Coastguard Worker return WaitSelect(cmsWait, process_io);
1280*d9f75844SAndroid Build Coastguard Worker }
1281*d9f75844SAndroid Build Coastguard Worker
1282*d9f75844SAndroid Build Coastguard Worker // `error_event` is true if we are responding to an event where we know an
1283*d9f75844SAndroid Build Coastguard Worker // error has occurred, which is possible with the poll/epoll implementations
1284*d9f75844SAndroid Build Coastguard Worker // but not the select implementation.
1285*d9f75844SAndroid Build Coastguard Worker //
1286*d9f75844SAndroid Build Coastguard Worker // `check_error` is true if there is the possibility of an error.
ProcessEvents(Dispatcher * dispatcher,bool readable,bool writable,bool error_event,bool check_error)1287*d9f75844SAndroid Build Coastguard Worker static void ProcessEvents(Dispatcher* dispatcher,
1288*d9f75844SAndroid Build Coastguard Worker bool readable,
1289*d9f75844SAndroid Build Coastguard Worker bool writable,
1290*d9f75844SAndroid Build Coastguard Worker bool error_event,
1291*d9f75844SAndroid Build Coastguard Worker bool check_error) {
1292*d9f75844SAndroid Build Coastguard Worker RTC_DCHECK(!(error_event && !check_error));
1293*d9f75844SAndroid Build Coastguard Worker int errcode = 0;
1294*d9f75844SAndroid Build Coastguard Worker if (check_error) {
1295*d9f75844SAndroid Build Coastguard Worker socklen_t len = sizeof(errcode);
1296*d9f75844SAndroid Build Coastguard Worker int res = ::getsockopt(dispatcher->GetDescriptor(), SOL_SOCKET, SO_ERROR,
1297*d9f75844SAndroid Build Coastguard Worker &errcode, &len);
1298*d9f75844SAndroid Build Coastguard Worker if (res < 0) {
1299*d9f75844SAndroid Build Coastguard Worker // If we are sure an error has occurred, or if getsockopt failed for a
1300*d9f75844SAndroid Build Coastguard Worker // socket descriptor, make sure we set the error code to a nonzero value.
1301*d9f75844SAndroid Build Coastguard Worker if (error_event || errno != ENOTSOCK) {
1302*d9f75844SAndroid Build Coastguard Worker errcode = EBADF;
1303*d9f75844SAndroid Build Coastguard Worker }
1304*d9f75844SAndroid Build Coastguard Worker }
1305*d9f75844SAndroid Build Coastguard Worker }
1306*d9f75844SAndroid Build Coastguard Worker
1307*d9f75844SAndroid Build Coastguard Worker // Most often the socket is writable or readable or both, so make a single
1308*d9f75844SAndroid Build Coastguard Worker // virtual call to get requested events
1309*d9f75844SAndroid Build Coastguard Worker const uint32_t requested_events = dispatcher->GetRequestedEvents();
1310*d9f75844SAndroid Build Coastguard Worker uint32_t ff = 0;
1311*d9f75844SAndroid Build Coastguard Worker
1312*d9f75844SAndroid Build Coastguard Worker // Check readable descriptors. If we're waiting on an accept, signal
1313*d9f75844SAndroid Build Coastguard Worker // that. Otherwise we're waiting for data, check to see if we're
1314*d9f75844SAndroid Build Coastguard Worker // readable or really closed.
1315*d9f75844SAndroid Build Coastguard Worker // TODO(pthatcher): Only peek at TCP descriptors.
1316*d9f75844SAndroid Build Coastguard Worker if (readable) {
1317*d9f75844SAndroid Build Coastguard Worker if (errcode || dispatcher->IsDescriptorClosed()) {
1318*d9f75844SAndroid Build Coastguard Worker ff |= DE_CLOSE;
1319*d9f75844SAndroid Build Coastguard Worker } else if (requested_events & DE_ACCEPT) {
1320*d9f75844SAndroid Build Coastguard Worker ff |= DE_ACCEPT;
1321*d9f75844SAndroid Build Coastguard Worker } else {
1322*d9f75844SAndroid Build Coastguard Worker ff |= DE_READ;
1323*d9f75844SAndroid Build Coastguard Worker }
1324*d9f75844SAndroid Build Coastguard Worker }
1325*d9f75844SAndroid Build Coastguard Worker
1326*d9f75844SAndroid Build Coastguard Worker // Check writable descriptors. If we're waiting on a connect, detect
1327*d9f75844SAndroid Build Coastguard Worker // success versus failure by the reaped error code.
1328*d9f75844SAndroid Build Coastguard Worker if (writable) {
1329*d9f75844SAndroid Build Coastguard Worker if (requested_events & DE_CONNECT) {
1330*d9f75844SAndroid Build Coastguard Worker if (!errcode) {
1331*d9f75844SAndroid Build Coastguard Worker ff |= DE_CONNECT;
1332*d9f75844SAndroid Build Coastguard Worker }
1333*d9f75844SAndroid Build Coastguard Worker } else {
1334*d9f75844SAndroid Build Coastguard Worker ff |= DE_WRITE;
1335*d9f75844SAndroid Build Coastguard Worker }
1336*d9f75844SAndroid Build Coastguard Worker }
1337*d9f75844SAndroid Build Coastguard Worker
1338*d9f75844SAndroid Build Coastguard Worker // Make sure we report any errors regardless of whether readable or writable.
1339*d9f75844SAndroid Build Coastguard Worker if (errcode) {
1340*d9f75844SAndroid Build Coastguard Worker ff |= DE_CLOSE;
1341*d9f75844SAndroid Build Coastguard Worker }
1342*d9f75844SAndroid Build Coastguard Worker
1343*d9f75844SAndroid Build Coastguard Worker // Tell the descriptor about the event.
1344*d9f75844SAndroid Build Coastguard Worker if (ff != 0) {
1345*d9f75844SAndroid Build Coastguard Worker dispatcher->OnEvent(ff, errcode);
1346*d9f75844SAndroid Build Coastguard Worker }
1347*d9f75844SAndroid Build Coastguard Worker }
1348*d9f75844SAndroid Build Coastguard Worker
WaitSelect(int cmsWait,bool process_io)1349*d9f75844SAndroid Build Coastguard Worker bool PhysicalSocketServer::WaitSelect(int cmsWait, bool process_io) {
1350*d9f75844SAndroid Build Coastguard Worker // Calculate timing information
1351*d9f75844SAndroid Build Coastguard Worker
1352*d9f75844SAndroid Build Coastguard Worker struct timeval* ptvWait = nullptr;
1353*d9f75844SAndroid Build Coastguard Worker struct timeval tvWait;
1354*d9f75844SAndroid Build Coastguard Worker int64_t stop_us;
1355*d9f75844SAndroid Build Coastguard Worker if (cmsWait != kForeverMs) {
1356*d9f75844SAndroid Build Coastguard Worker // Calculate wait timeval
1357*d9f75844SAndroid Build Coastguard Worker tvWait.tv_sec = cmsWait / 1000;
1358*d9f75844SAndroid Build Coastguard Worker tvWait.tv_usec = (cmsWait % 1000) * 1000;
1359*d9f75844SAndroid Build Coastguard Worker ptvWait = &tvWait;
1360*d9f75844SAndroid Build Coastguard Worker
1361*d9f75844SAndroid Build Coastguard Worker // Calculate when to return
1362*d9f75844SAndroid Build Coastguard Worker stop_us = rtc::TimeMicros() + cmsWait * 1000;
1363*d9f75844SAndroid Build Coastguard Worker }
1364*d9f75844SAndroid Build Coastguard Worker
1365*d9f75844SAndroid Build Coastguard Worker fd_set fdsRead;
1366*d9f75844SAndroid Build Coastguard Worker fd_set fdsWrite;
1367*d9f75844SAndroid Build Coastguard Worker // Explicitly unpoison these FDs on MemorySanitizer which doesn't handle the
1368*d9f75844SAndroid Build Coastguard Worker // inline assembly in FD_ZERO.
1369*d9f75844SAndroid Build Coastguard Worker // http://crbug.com/344505
1370*d9f75844SAndroid Build Coastguard Worker #ifdef MEMORY_SANITIZER
1371*d9f75844SAndroid Build Coastguard Worker __msan_unpoison(&fdsRead, sizeof(fdsRead));
1372*d9f75844SAndroid Build Coastguard Worker __msan_unpoison(&fdsWrite, sizeof(fdsWrite));
1373*d9f75844SAndroid Build Coastguard Worker #endif
1374*d9f75844SAndroid Build Coastguard Worker
1375*d9f75844SAndroid Build Coastguard Worker fWait_ = true;
1376*d9f75844SAndroid Build Coastguard Worker
1377*d9f75844SAndroid Build Coastguard Worker while (fWait_) {
1378*d9f75844SAndroid Build Coastguard Worker // Zero all fd_sets. Although select() zeros the descriptors not signaled,
1379*d9f75844SAndroid Build Coastguard Worker // we may need to do this for dispatchers that were deleted while
1380*d9f75844SAndroid Build Coastguard Worker // iterating.
1381*d9f75844SAndroid Build Coastguard Worker FD_ZERO(&fdsRead);
1382*d9f75844SAndroid Build Coastguard Worker FD_ZERO(&fdsWrite);
1383*d9f75844SAndroid Build Coastguard Worker int fdmax = -1;
1384*d9f75844SAndroid Build Coastguard Worker {
1385*d9f75844SAndroid Build Coastguard Worker CritScope cr(&crit_);
1386*d9f75844SAndroid Build Coastguard Worker current_dispatcher_keys_.clear();
1387*d9f75844SAndroid Build Coastguard Worker for (auto const& kv : dispatcher_by_key_) {
1388*d9f75844SAndroid Build Coastguard Worker uint64_t key = kv.first;
1389*d9f75844SAndroid Build Coastguard Worker Dispatcher* pdispatcher = kv.second;
1390*d9f75844SAndroid Build Coastguard Worker // Query dispatchers for read and write wait state
1391*d9f75844SAndroid Build Coastguard Worker if (!process_io && (pdispatcher != signal_wakeup_))
1392*d9f75844SAndroid Build Coastguard Worker continue;
1393*d9f75844SAndroid Build Coastguard Worker current_dispatcher_keys_.push_back(key);
1394*d9f75844SAndroid Build Coastguard Worker int fd = pdispatcher->GetDescriptor();
1395*d9f75844SAndroid Build Coastguard Worker // "select"ing a file descriptor that is equal to or larger than
1396*d9f75844SAndroid Build Coastguard Worker // FD_SETSIZE will result in undefined behavior.
1397*d9f75844SAndroid Build Coastguard Worker RTC_DCHECK_LT(fd, FD_SETSIZE);
1398*d9f75844SAndroid Build Coastguard Worker if (fd > fdmax)
1399*d9f75844SAndroid Build Coastguard Worker fdmax = fd;
1400*d9f75844SAndroid Build Coastguard Worker
1401*d9f75844SAndroid Build Coastguard Worker uint32_t ff = pdispatcher->GetRequestedEvents();
1402*d9f75844SAndroid Build Coastguard Worker if (ff & (DE_READ | DE_ACCEPT))
1403*d9f75844SAndroid Build Coastguard Worker FD_SET(fd, &fdsRead);
1404*d9f75844SAndroid Build Coastguard Worker if (ff & (DE_WRITE | DE_CONNECT))
1405*d9f75844SAndroid Build Coastguard Worker FD_SET(fd, &fdsWrite);
1406*d9f75844SAndroid Build Coastguard Worker }
1407*d9f75844SAndroid Build Coastguard Worker }
1408*d9f75844SAndroid Build Coastguard Worker
1409*d9f75844SAndroid Build Coastguard Worker // Wait then call handlers as appropriate
1410*d9f75844SAndroid Build Coastguard Worker // < 0 means error
1411*d9f75844SAndroid Build Coastguard Worker // 0 means timeout
1412*d9f75844SAndroid Build Coastguard Worker // > 0 means count of descriptors ready
1413*d9f75844SAndroid Build Coastguard Worker int n = select(fdmax + 1, &fdsRead, &fdsWrite, nullptr, ptvWait);
1414*d9f75844SAndroid Build Coastguard Worker
1415*d9f75844SAndroid Build Coastguard Worker // If error, return error.
1416*d9f75844SAndroid Build Coastguard Worker if (n < 0) {
1417*d9f75844SAndroid Build Coastguard Worker if (errno != EINTR) {
1418*d9f75844SAndroid Build Coastguard Worker RTC_LOG_E(LS_ERROR, EN, errno) << "select";
1419*d9f75844SAndroid Build Coastguard Worker return false;
1420*d9f75844SAndroid Build Coastguard Worker }
1421*d9f75844SAndroid Build Coastguard Worker // Else ignore the error and keep going. If this EINTR was for one of the
1422*d9f75844SAndroid Build Coastguard Worker // signals managed by this PhysicalSocketServer, the
1423*d9f75844SAndroid Build Coastguard Worker // PosixSignalDeliveryDispatcher will be in the signaled state in the next
1424*d9f75844SAndroid Build Coastguard Worker // iteration.
1425*d9f75844SAndroid Build Coastguard Worker } else if (n == 0) {
1426*d9f75844SAndroid Build Coastguard Worker // If timeout, return success
1427*d9f75844SAndroid Build Coastguard Worker return true;
1428*d9f75844SAndroid Build Coastguard Worker } else {
1429*d9f75844SAndroid Build Coastguard Worker // We have signaled descriptors
1430*d9f75844SAndroid Build Coastguard Worker CritScope cr(&crit_);
1431*d9f75844SAndroid Build Coastguard Worker // Iterate only on the dispatchers whose sockets were passed into
1432*d9f75844SAndroid Build Coastguard Worker // WSAEventSelect; this avoids the ABA problem (a socket being
1433*d9f75844SAndroid Build Coastguard Worker // destroyed and a new one created with the same file descriptor).
1434*d9f75844SAndroid Build Coastguard Worker for (uint64_t key : current_dispatcher_keys_) {
1435*d9f75844SAndroid Build Coastguard Worker if (!dispatcher_by_key_.count(key))
1436*d9f75844SAndroid Build Coastguard Worker continue;
1437*d9f75844SAndroid Build Coastguard Worker Dispatcher* pdispatcher = dispatcher_by_key_.at(key);
1438*d9f75844SAndroid Build Coastguard Worker
1439*d9f75844SAndroid Build Coastguard Worker int fd = pdispatcher->GetDescriptor();
1440*d9f75844SAndroid Build Coastguard Worker
1441*d9f75844SAndroid Build Coastguard Worker bool readable = FD_ISSET(fd, &fdsRead);
1442*d9f75844SAndroid Build Coastguard Worker if (readable) {
1443*d9f75844SAndroid Build Coastguard Worker FD_CLR(fd, &fdsRead);
1444*d9f75844SAndroid Build Coastguard Worker }
1445*d9f75844SAndroid Build Coastguard Worker
1446*d9f75844SAndroid Build Coastguard Worker bool writable = FD_ISSET(fd, &fdsWrite);
1447*d9f75844SAndroid Build Coastguard Worker if (writable) {
1448*d9f75844SAndroid Build Coastguard Worker FD_CLR(fd, &fdsWrite);
1449*d9f75844SAndroid Build Coastguard Worker }
1450*d9f75844SAndroid Build Coastguard Worker
1451*d9f75844SAndroid Build Coastguard Worker // The error code can be signaled through reads or writes.
1452*d9f75844SAndroid Build Coastguard Worker ProcessEvents(pdispatcher, readable, writable, /*error_event=*/false,
1453*d9f75844SAndroid Build Coastguard Worker readable || writable);
1454*d9f75844SAndroid Build Coastguard Worker }
1455*d9f75844SAndroid Build Coastguard Worker }
1456*d9f75844SAndroid Build Coastguard Worker
1457*d9f75844SAndroid Build Coastguard Worker // Recalc the time remaining to wait. Doing it here means it doesn't get
1458*d9f75844SAndroid Build Coastguard Worker // calced twice the first time through the loop
1459*d9f75844SAndroid Build Coastguard Worker if (ptvWait) {
1460*d9f75844SAndroid Build Coastguard Worker ptvWait->tv_sec = 0;
1461*d9f75844SAndroid Build Coastguard Worker ptvWait->tv_usec = 0;
1462*d9f75844SAndroid Build Coastguard Worker int64_t time_left_us = stop_us - rtc::TimeMicros();
1463*d9f75844SAndroid Build Coastguard Worker if (time_left_us > 0) {
1464*d9f75844SAndroid Build Coastguard Worker ptvWait->tv_sec = time_left_us / rtc::kNumMicrosecsPerSec;
1465*d9f75844SAndroid Build Coastguard Worker ptvWait->tv_usec = time_left_us % rtc::kNumMicrosecsPerSec;
1466*d9f75844SAndroid Build Coastguard Worker }
1467*d9f75844SAndroid Build Coastguard Worker }
1468*d9f75844SAndroid Build Coastguard Worker }
1469*d9f75844SAndroid Build Coastguard Worker
1470*d9f75844SAndroid Build Coastguard Worker return true;
1471*d9f75844SAndroid Build Coastguard Worker }
1472*d9f75844SAndroid Build Coastguard Worker
1473*d9f75844SAndroid Build Coastguard Worker #if defined(WEBRTC_USE_EPOLL)
1474*d9f75844SAndroid Build Coastguard Worker
AddEpoll(Dispatcher * pdispatcher,uint64_t key)1475*d9f75844SAndroid Build Coastguard Worker void PhysicalSocketServer::AddEpoll(Dispatcher* pdispatcher, uint64_t key) {
1476*d9f75844SAndroid Build Coastguard Worker RTC_DCHECK(epoll_fd_ != INVALID_SOCKET);
1477*d9f75844SAndroid Build Coastguard Worker int fd = pdispatcher->GetDescriptor();
1478*d9f75844SAndroid Build Coastguard Worker RTC_DCHECK(fd != INVALID_SOCKET);
1479*d9f75844SAndroid Build Coastguard Worker if (fd == INVALID_SOCKET) {
1480*d9f75844SAndroid Build Coastguard Worker return;
1481*d9f75844SAndroid Build Coastguard Worker }
1482*d9f75844SAndroid Build Coastguard Worker
1483*d9f75844SAndroid Build Coastguard Worker struct epoll_event event = {0};
1484*d9f75844SAndroid Build Coastguard Worker event.events = GetEpollEvents(pdispatcher->GetRequestedEvents());
1485*d9f75844SAndroid Build Coastguard Worker if (event.events == 0u) {
1486*d9f75844SAndroid Build Coastguard Worker // Don't add at all if we don't have any requested events. Could indicate a
1487*d9f75844SAndroid Build Coastguard Worker // closed socket.
1488*d9f75844SAndroid Build Coastguard Worker return;
1489*d9f75844SAndroid Build Coastguard Worker }
1490*d9f75844SAndroid Build Coastguard Worker event.data.u64 = key;
1491*d9f75844SAndroid Build Coastguard Worker int err = epoll_ctl(epoll_fd_, EPOLL_CTL_ADD, fd, &event);
1492*d9f75844SAndroid Build Coastguard Worker RTC_DCHECK_EQ(err, 0);
1493*d9f75844SAndroid Build Coastguard Worker if (err == -1) {
1494*d9f75844SAndroid Build Coastguard Worker RTC_LOG_E(LS_ERROR, EN, errno) << "epoll_ctl EPOLL_CTL_ADD";
1495*d9f75844SAndroid Build Coastguard Worker }
1496*d9f75844SAndroid Build Coastguard Worker }
1497*d9f75844SAndroid Build Coastguard Worker
RemoveEpoll(Dispatcher * pdispatcher)1498*d9f75844SAndroid Build Coastguard Worker void PhysicalSocketServer::RemoveEpoll(Dispatcher* pdispatcher) {
1499*d9f75844SAndroid Build Coastguard Worker RTC_DCHECK(epoll_fd_ != INVALID_SOCKET);
1500*d9f75844SAndroid Build Coastguard Worker int fd = pdispatcher->GetDescriptor();
1501*d9f75844SAndroid Build Coastguard Worker RTC_DCHECK(fd != INVALID_SOCKET);
1502*d9f75844SAndroid Build Coastguard Worker if (fd == INVALID_SOCKET) {
1503*d9f75844SAndroid Build Coastguard Worker return;
1504*d9f75844SAndroid Build Coastguard Worker }
1505*d9f75844SAndroid Build Coastguard Worker
1506*d9f75844SAndroid Build Coastguard Worker struct epoll_event event = {0};
1507*d9f75844SAndroid Build Coastguard Worker int err = epoll_ctl(epoll_fd_, EPOLL_CTL_DEL, fd, &event);
1508*d9f75844SAndroid Build Coastguard Worker RTC_DCHECK(err == 0 || errno == ENOENT);
1509*d9f75844SAndroid Build Coastguard Worker // Ignore ENOENT, which could occur if this descriptor wasn't added due to
1510*d9f75844SAndroid Build Coastguard Worker // having no requested events.
1511*d9f75844SAndroid Build Coastguard Worker if (err == -1 && errno != ENOENT) {
1512*d9f75844SAndroid Build Coastguard Worker RTC_LOG_E(LS_ERROR, EN, errno) << "epoll_ctl EPOLL_CTL_DEL";
1513*d9f75844SAndroid Build Coastguard Worker }
1514*d9f75844SAndroid Build Coastguard Worker }
1515*d9f75844SAndroid Build Coastguard Worker
UpdateEpoll(Dispatcher * pdispatcher,uint64_t key)1516*d9f75844SAndroid Build Coastguard Worker void PhysicalSocketServer::UpdateEpoll(Dispatcher* pdispatcher, uint64_t key) {
1517*d9f75844SAndroid Build Coastguard Worker RTC_DCHECK(epoll_fd_ != INVALID_SOCKET);
1518*d9f75844SAndroid Build Coastguard Worker int fd = pdispatcher->GetDescriptor();
1519*d9f75844SAndroid Build Coastguard Worker RTC_DCHECK(fd != INVALID_SOCKET);
1520*d9f75844SAndroid Build Coastguard Worker if (fd == INVALID_SOCKET) {
1521*d9f75844SAndroid Build Coastguard Worker return;
1522*d9f75844SAndroid Build Coastguard Worker }
1523*d9f75844SAndroid Build Coastguard Worker
1524*d9f75844SAndroid Build Coastguard Worker struct epoll_event event = {0};
1525*d9f75844SAndroid Build Coastguard Worker event.events = GetEpollEvents(pdispatcher->GetRequestedEvents());
1526*d9f75844SAndroid Build Coastguard Worker event.data.u64 = key;
1527*d9f75844SAndroid Build Coastguard Worker // Remove if we don't have any requested events. Could indicate a closed
1528*d9f75844SAndroid Build Coastguard Worker // socket.
1529*d9f75844SAndroid Build Coastguard Worker if (event.events == 0u) {
1530*d9f75844SAndroid Build Coastguard Worker epoll_ctl(epoll_fd_, EPOLL_CTL_DEL, fd, &event);
1531*d9f75844SAndroid Build Coastguard Worker } else {
1532*d9f75844SAndroid Build Coastguard Worker int err = epoll_ctl(epoll_fd_, EPOLL_CTL_MOD, fd, &event);
1533*d9f75844SAndroid Build Coastguard Worker RTC_DCHECK(err == 0 || errno == ENOENT);
1534*d9f75844SAndroid Build Coastguard Worker if (err == -1) {
1535*d9f75844SAndroid Build Coastguard Worker // Could have been removed earlier due to no requested events.
1536*d9f75844SAndroid Build Coastguard Worker if (errno == ENOENT) {
1537*d9f75844SAndroid Build Coastguard Worker err = epoll_ctl(epoll_fd_, EPOLL_CTL_ADD, fd, &event);
1538*d9f75844SAndroid Build Coastguard Worker if (err == -1) {
1539*d9f75844SAndroid Build Coastguard Worker RTC_LOG_E(LS_ERROR, EN, errno) << "epoll_ctl EPOLL_CTL_ADD";
1540*d9f75844SAndroid Build Coastguard Worker }
1541*d9f75844SAndroid Build Coastguard Worker } else {
1542*d9f75844SAndroid Build Coastguard Worker RTC_LOG_E(LS_ERROR, EN, errno) << "epoll_ctl EPOLL_CTL_MOD";
1543*d9f75844SAndroid Build Coastguard Worker }
1544*d9f75844SAndroid Build Coastguard Worker }
1545*d9f75844SAndroid Build Coastguard Worker }
1546*d9f75844SAndroid Build Coastguard Worker }
1547*d9f75844SAndroid Build Coastguard Worker
WaitEpoll(int cmsWait)1548*d9f75844SAndroid Build Coastguard Worker bool PhysicalSocketServer::WaitEpoll(int cmsWait) {
1549*d9f75844SAndroid Build Coastguard Worker RTC_DCHECK(epoll_fd_ != INVALID_SOCKET);
1550*d9f75844SAndroid Build Coastguard Worker int64_t tvWait = -1;
1551*d9f75844SAndroid Build Coastguard Worker int64_t tvStop = -1;
1552*d9f75844SAndroid Build Coastguard Worker if (cmsWait != kForeverMs) {
1553*d9f75844SAndroid Build Coastguard Worker tvWait = cmsWait;
1554*d9f75844SAndroid Build Coastguard Worker tvStop = TimeAfter(cmsWait);
1555*d9f75844SAndroid Build Coastguard Worker }
1556*d9f75844SAndroid Build Coastguard Worker
1557*d9f75844SAndroid Build Coastguard Worker fWait_ = true;
1558*d9f75844SAndroid Build Coastguard Worker while (fWait_) {
1559*d9f75844SAndroid Build Coastguard Worker // Wait then call handlers as appropriate
1560*d9f75844SAndroid Build Coastguard Worker // < 0 means error
1561*d9f75844SAndroid Build Coastguard Worker // 0 means timeout
1562*d9f75844SAndroid Build Coastguard Worker // > 0 means count of descriptors ready
1563*d9f75844SAndroid Build Coastguard Worker int n = epoll_wait(epoll_fd_, epoll_events_.data(), epoll_events_.size(),
1564*d9f75844SAndroid Build Coastguard Worker static_cast<int>(tvWait));
1565*d9f75844SAndroid Build Coastguard Worker if (n < 0) {
1566*d9f75844SAndroid Build Coastguard Worker if (errno != EINTR) {
1567*d9f75844SAndroid Build Coastguard Worker RTC_LOG_E(LS_ERROR, EN, errno) << "epoll";
1568*d9f75844SAndroid Build Coastguard Worker return false;
1569*d9f75844SAndroid Build Coastguard Worker }
1570*d9f75844SAndroid Build Coastguard Worker // Else ignore the error and keep going. If this EINTR was for one of the
1571*d9f75844SAndroid Build Coastguard Worker // signals managed by this PhysicalSocketServer, the
1572*d9f75844SAndroid Build Coastguard Worker // PosixSignalDeliveryDispatcher will be in the signaled state in the next
1573*d9f75844SAndroid Build Coastguard Worker // iteration.
1574*d9f75844SAndroid Build Coastguard Worker } else if (n == 0) {
1575*d9f75844SAndroid Build Coastguard Worker // If timeout, return success
1576*d9f75844SAndroid Build Coastguard Worker return true;
1577*d9f75844SAndroid Build Coastguard Worker } else {
1578*d9f75844SAndroid Build Coastguard Worker // We have signaled descriptors
1579*d9f75844SAndroid Build Coastguard Worker CritScope cr(&crit_);
1580*d9f75844SAndroid Build Coastguard Worker for (int i = 0; i < n; ++i) {
1581*d9f75844SAndroid Build Coastguard Worker const epoll_event& event = epoll_events_[i];
1582*d9f75844SAndroid Build Coastguard Worker uint64_t key = event.data.u64;
1583*d9f75844SAndroid Build Coastguard Worker if (!dispatcher_by_key_.count(key)) {
1584*d9f75844SAndroid Build Coastguard Worker // The dispatcher for this socket no longer exists.
1585*d9f75844SAndroid Build Coastguard Worker continue;
1586*d9f75844SAndroid Build Coastguard Worker }
1587*d9f75844SAndroid Build Coastguard Worker Dispatcher* pdispatcher = dispatcher_by_key_.at(key);
1588*d9f75844SAndroid Build Coastguard Worker
1589*d9f75844SAndroid Build Coastguard Worker bool readable = (event.events & (EPOLLIN | EPOLLPRI));
1590*d9f75844SAndroid Build Coastguard Worker bool writable = (event.events & EPOLLOUT);
1591*d9f75844SAndroid Build Coastguard Worker bool error = (event.events & (EPOLLRDHUP | EPOLLERR | EPOLLHUP));
1592*d9f75844SAndroid Build Coastguard Worker
1593*d9f75844SAndroid Build Coastguard Worker ProcessEvents(pdispatcher, readable, writable, error, error);
1594*d9f75844SAndroid Build Coastguard Worker }
1595*d9f75844SAndroid Build Coastguard Worker }
1596*d9f75844SAndroid Build Coastguard Worker
1597*d9f75844SAndroid Build Coastguard Worker if (cmsWait != kForeverMs) {
1598*d9f75844SAndroid Build Coastguard Worker tvWait = TimeDiff(tvStop, TimeMillis());
1599*d9f75844SAndroid Build Coastguard Worker if (tvWait <= 0) {
1600*d9f75844SAndroid Build Coastguard Worker // Return success on timeout.
1601*d9f75844SAndroid Build Coastguard Worker return true;
1602*d9f75844SAndroid Build Coastguard Worker }
1603*d9f75844SAndroid Build Coastguard Worker }
1604*d9f75844SAndroid Build Coastguard Worker }
1605*d9f75844SAndroid Build Coastguard Worker
1606*d9f75844SAndroid Build Coastguard Worker return true;
1607*d9f75844SAndroid Build Coastguard Worker }
1608*d9f75844SAndroid Build Coastguard Worker
WaitPoll(int cmsWait,Dispatcher * dispatcher)1609*d9f75844SAndroid Build Coastguard Worker bool PhysicalSocketServer::WaitPoll(int cmsWait, Dispatcher* dispatcher) {
1610*d9f75844SAndroid Build Coastguard Worker RTC_DCHECK(dispatcher);
1611*d9f75844SAndroid Build Coastguard Worker int64_t tvWait = -1;
1612*d9f75844SAndroid Build Coastguard Worker int64_t tvStop = -1;
1613*d9f75844SAndroid Build Coastguard Worker if (cmsWait != kForeverMs) {
1614*d9f75844SAndroid Build Coastguard Worker tvWait = cmsWait;
1615*d9f75844SAndroid Build Coastguard Worker tvStop = TimeAfter(cmsWait);
1616*d9f75844SAndroid Build Coastguard Worker }
1617*d9f75844SAndroid Build Coastguard Worker
1618*d9f75844SAndroid Build Coastguard Worker fWait_ = true;
1619*d9f75844SAndroid Build Coastguard Worker
1620*d9f75844SAndroid Build Coastguard Worker struct pollfd fds = {0};
1621*d9f75844SAndroid Build Coastguard Worker int fd = dispatcher->GetDescriptor();
1622*d9f75844SAndroid Build Coastguard Worker fds.fd = fd;
1623*d9f75844SAndroid Build Coastguard Worker
1624*d9f75844SAndroid Build Coastguard Worker while (fWait_) {
1625*d9f75844SAndroid Build Coastguard Worker uint32_t ff = dispatcher->GetRequestedEvents();
1626*d9f75844SAndroid Build Coastguard Worker fds.events = 0;
1627*d9f75844SAndroid Build Coastguard Worker if (ff & (DE_READ | DE_ACCEPT)) {
1628*d9f75844SAndroid Build Coastguard Worker fds.events |= POLLIN;
1629*d9f75844SAndroid Build Coastguard Worker }
1630*d9f75844SAndroid Build Coastguard Worker if (ff & (DE_WRITE | DE_CONNECT)) {
1631*d9f75844SAndroid Build Coastguard Worker fds.events |= POLLOUT;
1632*d9f75844SAndroid Build Coastguard Worker }
1633*d9f75844SAndroid Build Coastguard Worker fds.revents = 0;
1634*d9f75844SAndroid Build Coastguard Worker
1635*d9f75844SAndroid Build Coastguard Worker // Wait then call handlers as appropriate
1636*d9f75844SAndroid Build Coastguard Worker // < 0 means error
1637*d9f75844SAndroid Build Coastguard Worker // 0 means timeout
1638*d9f75844SAndroid Build Coastguard Worker // > 0 means count of descriptors ready
1639*d9f75844SAndroid Build Coastguard Worker int n = poll(&fds, 1, static_cast<int>(tvWait));
1640*d9f75844SAndroid Build Coastguard Worker if (n < 0) {
1641*d9f75844SAndroid Build Coastguard Worker if (errno != EINTR) {
1642*d9f75844SAndroid Build Coastguard Worker RTC_LOG_E(LS_ERROR, EN, errno) << "poll";
1643*d9f75844SAndroid Build Coastguard Worker return false;
1644*d9f75844SAndroid Build Coastguard Worker }
1645*d9f75844SAndroid Build Coastguard Worker // Else ignore the error and keep going. If this EINTR was for one of the
1646*d9f75844SAndroid Build Coastguard Worker // signals managed by this PhysicalSocketServer, the
1647*d9f75844SAndroid Build Coastguard Worker // PosixSignalDeliveryDispatcher will be in the signaled state in the next
1648*d9f75844SAndroid Build Coastguard Worker // iteration.
1649*d9f75844SAndroid Build Coastguard Worker } else if (n == 0) {
1650*d9f75844SAndroid Build Coastguard Worker // If timeout, return success
1651*d9f75844SAndroid Build Coastguard Worker return true;
1652*d9f75844SAndroid Build Coastguard Worker } else {
1653*d9f75844SAndroid Build Coastguard Worker // We have signaled descriptors (should only be the passed dispatcher).
1654*d9f75844SAndroid Build Coastguard Worker RTC_DCHECK_EQ(n, 1);
1655*d9f75844SAndroid Build Coastguard Worker RTC_DCHECK_EQ(fds.fd, fd);
1656*d9f75844SAndroid Build Coastguard Worker
1657*d9f75844SAndroid Build Coastguard Worker bool readable = (fds.revents & (POLLIN | POLLPRI));
1658*d9f75844SAndroid Build Coastguard Worker bool writable = (fds.revents & POLLOUT);
1659*d9f75844SAndroid Build Coastguard Worker bool error = (fds.revents & (POLLRDHUP | POLLERR | POLLHUP));
1660*d9f75844SAndroid Build Coastguard Worker
1661*d9f75844SAndroid Build Coastguard Worker ProcessEvents(dispatcher, readable, writable, error, error);
1662*d9f75844SAndroid Build Coastguard Worker }
1663*d9f75844SAndroid Build Coastguard Worker
1664*d9f75844SAndroid Build Coastguard Worker if (cmsWait != kForeverMs) {
1665*d9f75844SAndroid Build Coastguard Worker tvWait = TimeDiff(tvStop, TimeMillis());
1666*d9f75844SAndroid Build Coastguard Worker if (tvWait < 0) {
1667*d9f75844SAndroid Build Coastguard Worker // Return success on timeout.
1668*d9f75844SAndroid Build Coastguard Worker return true;
1669*d9f75844SAndroid Build Coastguard Worker }
1670*d9f75844SAndroid Build Coastguard Worker }
1671*d9f75844SAndroid Build Coastguard Worker }
1672*d9f75844SAndroid Build Coastguard Worker
1673*d9f75844SAndroid Build Coastguard Worker return true;
1674*d9f75844SAndroid Build Coastguard Worker }
1675*d9f75844SAndroid Build Coastguard Worker
1676*d9f75844SAndroid Build Coastguard Worker #endif // WEBRTC_USE_EPOLL
1677*d9f75844SAndroid Build Coastguard Worker
1678*d9f75844SAndroid Build Coastguard Worker #endif // WEBRTC_POSIX
1679*d9f75844SAndroid Build Coastguard Worker
1680*d9f75844SAndroid Build Coastguard Worker #if defined(WEBRTC_WIN)
Wait(webrtc::TimeDelta max_wait_duration,bool process_io)1681*d9f75844SAndroid Build Coastguard Worker bool PhysicalSocketServer::Wait(webrtc::TimeDelta max_wait_duration,
1682*d9f75844SAndroid Build Coastguard Worker bool process_io) {
1683*d9f75844SAndroid Build Coastguard Worker // We don't support reentrant waiting.
1684*d9f75844SAndroid Build Coastguard Worker RTC_DCHECK(!waiting_);
1685*d9f75844SAndroid Build Coastguard Worker ScopedSetTrue s(&waiting_);
1686*d9f75844SAndroid Build Coastguard Worker
1687*d9f75844SAndroid Build Coastguard Worker int cmsWait = ToCmsWait(max_wait_duration);
1688*d9f75844SAndroid Build Coastguard Worker int64_t cmsTotal = cmsWait;
1689*d9f75844SAndroid Build Coastguard Worker int64_t cmsElapsed = 0;
1690*d9f75844SAndroid Build Coastguard Worker int64_t msStart = Time();
1691*d9f75844SAndroid Build Coastguard Worker
1692*d9f75844SAndroid Build Coastguard Worker fWait_ = true;
1693*d9f75844SAndroid Build Coastguard Worker while (fWait_) {
1694*d9f75844SAndroid Build Coastguard Worker std::vector<WSAEVENT> events;
1695*d9f75844SAndroid Build Coastguard Worker std::vector<uint64_t> event_owners;
1696*d9f75844SAndroid Build Coastguard Worker
1697*d9f75844SAndroid Build Coastguard Worker events.push_back(socket_ev_);
1698*d9f75844SAndroid Build Coastguard Worker
1699*d9f75844SAndroid Build Coastguard Worker {
1700*d9f75844SAndroid Build Coastguard Worker CritScope cr(&crit_);
1701*d9f75844SAndroid Build Coastguard Worker // Get a snapshot of all current dispatchers; this is used to avoid the
1702*d9f75844SAndroid Build Coastguard Worker // ABA problem (see later comment) and avoids the dispatcher_by_key_
1703*d9f75844SAndroid Build Coastguard Worker // iterator being invalidated by calling CheckSignalClose, which may
1704*d9f75844SAndroid Build Coastguard Worker // remove the dispatcher from the list.
1705*d9f75844SAndroid Build Coastguard Worker current_dispatcher_keys_.clear();
1706*d9f75844SAndroid Build Coastguard Worker for (auto const& kv : dispatcher_by_key_) {
1707*d9f75844SAndroid Build Coastguard Worker current_dispatcher_keys_.push_back(kv.first);
1708*d9f75844SAndroid Build Coastguard Worker }
1709*d9f75844SAndroid Build Coastguard Worker for (uint64_t key : current_dispatcher_keys_) {
1710*d9f75844SAndroid Build Coastguard Worker if (!dispatcher_by_key_.count(key)) {
1711*d9f75844SAndroid Build Coastguard Worker continue;
1712*d9f75844SAndroid Build Coastguard Worker }
1713*d9f75844SAndroid Build Coastguard Worker Dispatcher* disp = dispatcher_by_key_.at(key);
1714*d9f75844SAndroid Build Coastguard Worker if (!disp)
1715*d9f75844SAndroid Build Coastguard Worker continue;
1716*d9f75844SAndroid Build Coastguard Worker if (!process_io && (disp != signal_wakeup_))
1717*d9f75844SAndroid Build Coastguard Worker continue;
1718*d9f75844SAndroid Build Coastguard Worker SOCKET s = disp->GetSocket();
1719*d9f75844SAndroid Build Coastguard Worker if (disp->CheckSignalClose()) {
1720*d9f75844SAndroid Build Coastguard Worker // We just signalled close, don't poll this socket.
1721*d9f75844SAndroid Build Coastguard Worker } else if (s != INVALID_SOCKET) {
1722*d9f75844SAndroid Build Coastguard Worker WSAEventSelect(s, events[0],
1723*d9f75844SAndroid Build Coastguard Worker FlagsToEvents(disp->GetRequestedEvents()));
1724*d9f75844SAndroid Build Coastguard Worker } else {
1725*d9f75844SAndroid Build Coastguard Worker events.push_back(disp->GetWSAEvent());
1726*d9f75844SAndroid Build Coastguard Worker event_owners.push_back(key);
1727*d9f75844SAndroid Build Coastguard Worker }
1728*d9f75844SAndroid Build Coastguard Worker }
1729*d9f75844SAndroid Build Coastguard Worker }
1730*d9f75844SAndroid Build Coastguard Worker
1731*d9f75844SAndroid Build Coastguard Worker // Which is shorter, the delay wait or the asked wait?
1732*d9f75844SAndroid Build Coastguard Worker
1733*d9f75844SAndroid Build Coastguard Worker int64_t cmsNext;
1734*d9f75844SAndroid Build Coastguard Worker if (cmsWait == kForeverMs) {
1735*d9f75844SAndroid Build Coastguard Worker cmsNext = cmsWait;
1736*d9f75844SAndroid Build Coastguard Worker } else {
1737*d9f75844SAndroid Build Coastguard Worker cmsNext = std::max<int64_t>(0, cmsTotal - cmsElapsed);
1738*d9f75844SAndroid Build Coastguard Worker }
1739*d9f75844SAndroid Build Coastguard Worker
1740*d9f75844SAndroid Build Coastguard Worker // Wait for one of the events to signal
1741*d9f75844SAndroid Build Coastguard Worker DWORD dw =
1742*d9f75844SAndroid Build Coastguard Worker WSAWaitForMultipleEvents(static_cast<DWORD>(events.size()), &events[0],
1743*d9f75844SAndroid Build Coastguard Worker false, static_cast<DWORD>(cmsNext), false);
1744*d9f75844SAndroid Build Coastguard Worker
1745*d9f75844SAndroid Build Coastguard Worker if (dw == WSA_WAIT_FAILED) {
1746*d9f75844SAndroid Build Coastguard Worker // Failed?
1747*d9f75844SAndroid Build Coastguard Worker // TODO(pthatcher): need a better strategy than this!
1748*d9f75844SAndroid Build Coastguard Worker WSAGetLastError();
1749*d9f75844SAndroid Build Coastguard Worker RTC_DCHECK_NOTREACHED();
1750*d9f75844SAndroid Build Coastguard Worker return false;
1751*d9f75844SAndroid Build Coastguard Worker } else if (dw == WSA_WAIT_TIMEOUT) {
1752*d9f75844SAndroid Build Coastguard Worker // Timeout?
1753*d9f75844SAndroid Build Coastguard Worker return true;
1754*d9f75844SAndroid Build Coastguard Worker } else {
1755*d9f75844SAndroid Build Coastguard Worker // Figure out which one it is and call it
1756*d9f75844SAndroid Build Coastguard Worker CritScope cr(&crit_);
1757*d9f75844SAndroid Build Coastguard Worker int index = dw - WSA_WAIT_EVENT_0;
1758*d9f75844SAndroid Build Coastguard Worker if (index > 0) {
1759*d9f75844SAndroid Build Coastguard Worker --index; // The first event is the socket event
1760*d9f75844SAndroid Build Coastguard Worker uint64_t key = event_owners[index];
1761*d9f75844SAndroid Build Coastguard Worker if (!dispatcher_by_key_.count(key)) {
1762*d9f75844SAndroid Build Coastguard Worker // The dispatcher could have been removed while waiting for events.
1763*d9f75844SAndroid Build Coastguard Worker continue;
1764*d9f75844SAndroid Build Coastguard Worker }
1765*d9f75844SAndroid Build Coastguard Worker Dispatcher* disp = dispatcher_by_key_.at(key);
1766*d9f75844SAndroid Build Coastguard Worker disp->OnEvent(0, 0);
1767*d9f75844SAndroid Build Coastguard Worker } else if (process_io) {
1768*d9f75844SAndroid Build Coastguard Worker // Iterate only on the dispatchers whose sockets were passed into
1769*d9f75844SAndroid Build Coastguard Worker // WSAEventSelect; this avoids the ABA problem (a socket being
1770*d9f75844SAndroid Build Coastguard Worker // destroyed and a new one created with the same SOCKET handle).
1771*d9f75844SAndroid Build Coastguard Worker for (uint64_t key : current_dispatcher_keys_) {
1772*d9f75844SAndroid Build Coastguard Worker if (!dispatcher_by_key_.count(key)) {
1773*d9f75844SAndroid Build Coastguard Worker continue;
1774*d9f75844SAndroid Build Coastguard Worker }
1775*d9f75844SAndroid Build Coastguard Worker Dispatcher* disp = dispatcher_by_key_.at(key);
1776*d9f75844SAndroid Build Coastguard Worker SOCKET s = disp->GetSocket();
1777*d9f75844SAndroid Build Coastguard Worker if (s == INVALID_SOCKET)
1778*d9f75844SAndroid Build Coastguard Worker continue;
1779*d9f75844SAndroid Build Coastguard Worker
1780*d9f75844SAndroid Build Coastguard Worker WSANETWORKEVENTS wsaEvents;
1781*d9f75844SAndroid Build Coastguard Worker int err = WSAEnumNetworkEvents(s, events[0], &wsaEvents);
1782*d9f75844SAndroid Build Coastguard Worker if (err == 0) {
1783*d9f75844SAndroid Build Coastguard Worker {
1784*d9f75844SAndroid Build Coastguard Worker if ((wsaEvents.lNetworkEvents & FD_READ) &&
1785*d9f75844SAndroid Build Coastguard Worker wsaEvents.iErrorCode[FD_READ_BIT] != 0) {
1786*d9f75844SAndroid Build Coastguard Worker RTC_LOG(LS_WARNING)
1787*d9f75844SAndroid Build Coastguard Worker << "PhysicalSocketServer got FD_READ_BIT error "
1788*d9f75844SAndroid Build Coastguard Worker << wsaEvents.iErrorCode[FD_READ_BIT];
1789*d9f75844SAndroid Build Coastguard Worker }
1790*d9f75844SAndroid Build Coastguard Worker if ((wsaEvents.lNetworkEvents & FD_WRITE) &&
1791*d9f75844SAndroid Build Coastguard Worker wsaEvents.iErrorCode[FD_WRITE_BIT] != 0) {
1792*d9f75844SAndroid Build Coastguard Worker RTC_LOG(LS_WARNING)
1793*d9f75844SAndroid Build Coastguard Worker << "PhysicalSocketServer got FD_WRITE_BIT error "
1794*d9f75844SAndroid Build Coastguard Worker << wsaEvents.iErrorCode[FD_WRITE_BIT];
1795*d9f75844SAndroid Build Coastguard Worker }
1796*d9f75844SAndroid Build Coastguard Worker if ((wsaEvents.lNetworkEvents & FD_CONNECT) &&
1797*d9f75844SAndroid Build Coastguard Worker wsaEvents.iErrorCode[FD_CONNECT_BIT] != 0) {
1798*d9f75844SAndroid Build Coastguard Worker RTC_LOG(LS_WARNING)
1799*d9f75844SAndroid Build Coastguard Worker << "PhysicalSocketServer got FD_CONNECT_BIT error "
1800*d9f75844SAndroid Build Coastguard Worker << wsaEvents.iErrorCode[FD_CONNECT_BIT];
1801*d9f75844SAndroid Build Coastguard Worker }
1802*d9f75844SAndroid Build Coastguard Worker if ((wsaEvents.lNetworkEvents & FD_ACCEPT) &&
1803*d9f75844SAndroid Build Coastguard Worker wsaEvents.iErrorCode[FD_ACCEPT_BIT] != 0) {
1804*d9f75844SAndroid Build Coastguard Worker RTC_LOG(LS_WARNING)
1805*d9f75844SAndroid Build Coastguard Worker << "PhysicalSocketServer got FD_ACCEPT_BIT error "
1806*d9f75844SAndroid Build Coastguard Worker << wsaEvents.iErrorCode[FD_ACCEPT_BIT];
1807*d9f75844SAndroid Build Coastguard Worker }
1808*d9f75844SAndroid Build Coastguard Worker if ((wsaEvents.lNetworkEvents & FD_CLOSE) &&
1809*d9f75844SAndroid Build Coastguard Worker wsaEvents.iErrorCode[FD_CLOSE_BIT] != 0) {
1810*d9f75844SAndroid Build Coastguard Worker RTC_LOG(LS_WARNING)
1811*d9f75844SAndroid Build Coastguard Worker << "PhysicalSocketServer got FD_CLOSE_BIT error "
1812*d9f75844SAndroid Build Coastguard Worker << wsaEvents.iErrorCode[FD_CLOSE_BIT];
1813*d9f75844SAndroid Build Coastguard Worker }
1814*d9f75844SAndroid Build Coastguard Worker }
1815*d9f75844SAndroid Build Coastguard Worker uint32_t ff = 0;
1816*d9f75844SAndroid Build Coastguard Worker int errcode = 0;
1817*d9f75844SAndroid Build Coastguard Worker if (wsaEvents.lNetworkEvents & FD_READ)
1818*d9f75844SAndroid Build Coastguard Worker ff |= DE_READ;
1819*d9f75844SAndroid Build Coastguard Worker if (wsaEvents.lNetworkEvents & FD_WRITE)
1820*d9f75844SAndroid Build Coastguard Worker ff |= DE_WRITE;
1821*d9f75844SAndroid Build Coastguard Worker if (wsaEvents.lNetworkEvents & FD_CONNECT) {
1822*d9f75844SAndroid Build Coastguard Worker if (wsaEvents.iErrorCode[FD_CONNECT_BIT] == 0) {
1823*d9f75844SAndroid Build Coastguard Worker ff |= DE_CONNECT;
1824*d9f75844SAndroid Build Coastguard Worker } else {
1825*d9f75844SAndroid Build Coastguard Worker ff |= DE_CLOSE;
1826*d9f75844SAndroid Build Coastguard Worker errcode = wsaEvents.iErrorCode[FD_CONNECT_BIT];
1827*d9f75844SAndroid Build Coastguard Worker }
1828*d9f75844SAndroid Build Coastguard Worker }
1829*d9f75844SAndroid Build Coastguard Worker if (wsaEvents.lNetworkEvents & FD_ACCEPT)
1830*d9f75844SAndroid Build Coastguard Worker ff |= DE_ACCEPT;
1831*d9f75844SAndroid Build Coastguard Worker if (wsaEvents.lNetworkEvents & FD_CLOSE) {
1832*d9f75844SAndroid Build Coastguard Worker ff |= DE_CLOSE;
1833*d9f75844SAndroid Build Coastguard Worker errcode = wsaEvents.iErrorCode[FD_CLOSE_BIT];
1834*d9f75844SAndroid Build Coastguard Worker }
1835*d9f75844SAndroid Build Coastguard Worker if (ff != 0) {
1836*d9f75844SAndroid Build Coastguard Worker disp->OnEvent(ff, errcode);
1837*d9f75844SAndroid Build Coastguard Worker }
1838*d9f75844SAndroid Build Coastguard Worker }
1839*d9f75844SAndroid Build Coastguard Worker }
1840*d9f75844SAndroid Build Coastguard Worker }
1841*d9f75844SAndroid Build Coastguard Worker
1842*d9f75844SAndroid Build Coastguard Worker // Reset the network event until new activity occurs
1843*d9f75844SAndroid Build Coastguard Worker WSAResetEvent(socket_ev_);
1844*d9f75844SAndroid Build Coastguard Worker }
1845*d9f75844SAndroid Build Coastguard Worker
1846*d9f75844SAndroid Build Coastguard Worker // Break?
1847*d9f75844SAndroid Build Coastguard Worker if (!fWait_)
1848*d9f75844SAndroid Build Coastguard Worker break;
1849*d9f75844SAndroid Build Coastguard Worker cmsElapsed = TimeSince(msStart);
1850*d9f75844SAndroid Build Coastguard Worker if ((cmsWait != kForeverMs) && (cmsElapsed >= cmsWait)) {
1851*d9f75844SAndroid Build Coastguard Worker break;
1852*d9f75844SAndroid Build Coastguard Worker }
1853*d9f75844SAndroid Build Coastguard Worker }
1854*d9f75844SAndroid Build Coastguard Worker
1855*d9f75844SAndroid Build Coastguard Worker // Done
1856*d9f75844SAndroid Build Coastguard Worker return true;
1857*d9f75844SAndroid Build Coastguard Worker }
1858*d9f75844SAndroid Build Coastguard Worker #endif // WEBRTC_WIN
1859*d9f75844SAndroid Build Coastguard Worker
1860*d9f75844SAndroid Build Coastguard Worker } // namespace rtc
1861