Support epoll in PhysicalSocketServer.

webrtc/base/physicalsocketserver.h

 /*
  *  Copyright 2004 The WebRTC Project Authors. All rights reserved.
  *
  *  Use of this source code is governed by a BSD-style license
  *  that can be found in the LICENSE file in the root of the source
  *  tree. An additional intellectual property rights grant can be found
  *  in the file PATENTS.  All contributing project authors may
  *  be found in the AUTHORS file in the root of the source tree.
  */
 
 #ifndef WEBRTC_BASE_PHYSICALSOCKETSERVER_H__
 #define WEBRTC_BASE_PHYSICALSOCKETSERVER_H__
 
+#if defined(WEBRTC_POSIX) && defined(WEBRTC_LINUX)
+#include <sys/epoll.h>
+#endif
+
 #include <memory>
+#include <set>
 #include <vector>
 
 #include "webrtc/base/nethelpers.h"
 #include "webrtc/base/socketserver.h"
 #include "webrtc/base/criticalsection.h"
 
 #if defined(WEBRTC_POSIX)
 typedef int SOCKET;
 #endif // WEBRTC_POSIX
 
@@ skipping 44 matching lines @@
 
   // Internal Factory for Accept (virtual so it can be overwritten in tests).
   virtual AsyncSocket* WrapSocket(SOCKET s);
 
   // SocketServer:
   bool Wait(int cms, bool process_io) override;
   void WakeUp() override;
 
   void Add(Dispatcher* dispatcher);
   void Remove(Dispatcher* dispatcher);
+  void Update(Dispatcher* dispatcher);
 
 #if defined(WEBRTC_POSIX)
   // Sets the function to be executed in response to the specified POSIX signal.
   // The function is executed from inside Wait() using the "self-pipe trick"--
   // regardless of which thread receives the signal--and hence can safely
   // manipulate user-level data structures.
   // "handler" may be SIG_IGN, SIG_DFL, or a user-specified function, just like
   // with signal(2).
   // Only one PhysicalSocketServer should have user-level signal handlers.
   // Dispatching signals on multiple PhysicalSocketServers is not reliable.
   // The signal mask is not modified. It is the caller's responsibily to
   // maintain it as desired.
   virtual bool SetPosixSignalHandler(int signum, void (*handler)(int));
 
  protected:
   Dispatcher* signal_dispatcher();
 #endif
 
  private:
   typedef std::vector<Dispatcher*> DispatcherList;
+  typedef std::set<Dispatcher*> DispatcherSet;
   typedef std::vector<size_t*> IteratorList;
 
 #if defined(WEBRTC_POSIX)
+  bool WaitSelect(int cms, bool process_io);
   static bool InstallSignal(int signum, void (*handler)(int));
 
   std::unique_ptr<PosixSignalDispatcher> signal_dispatcher_;
-#endif
+
+#if defined(WEBRTC_LINUX)
+  void AddEpoll(Dispatcher* dispatcher);
+  void RemoveEpoll(Dispatcher* dispatcher);
+  void UpdateEpoll(Dispatcher* dispatcher);
+  bool WaitEpoll(int cms);
+
+  int epoll_fd_ = INVALID_SOCKET;
+  std::vector<struct epoll_event> epoll_events_;
+#endif  // WEBRTC_LINUX
+#endif  // WEBRTC_POSIX
+#if defined(WEBRTC_WIN)
   DispatcherList dispatchers_;
   IteratorList iterators_;
+#else
+  DispatcherSet dispatchers_;
+#endif
   Signaler* signal_wakeup_;
   CriticalSection crit_;
   bool fWait_;
 #if defined(WEBRTC_WIN)
   WSAEVENT socket_ev_;
 #endif
 };
 
 class PhysicalSocket : public AsyncSocket, public sigslot::has_slots<> {
  public:
@@ skipping 48 matching lines @@
 
   // Make virtual so ::sendto can be overwritten in tests.
   virtual int DoSendTo(SOCKET socket, const char* buf, int len, int flags,
                        const struct sockaddr* dest_addr, socklen_t addrlen);
 
   void OnResolveResult(AsyncResolverInterface* resolver);
 
   void UpdateLastError();
   void MaybeRemapSendError();
 
+  virtual void SetEnabledEvents(uint8_t events);
+  virtual void EnableEvents(uint8_t events);
+  virtual void DisableEvents(uint8_t events);
+
   static int TranslateOption(Option opt, int* slevel, int* sopt);
 
   PhysicalSocketServer* ss_;
   SOCKET s_;
   uint8_t enabled_events_;
   bool udp_;
   CriticalSection crit_;
   int error_ GUARDED_BY(crit_);
   ConnState state_;
   AsyncResolver* resolver_;
@@ skipping 22 matching lines @@
   int GetDescriptor() override;
   bool IsDescriptorClosed() override;
 #endif
 
   uint32_t GetRequestedEvents() override;
   void OnPreEvent(uint32_t ff) override;
   void OnEvent(uint32_t ff, int err) override;
 
   int Close() override;
 
+#if defined(WEBRTC_POSIX) && defined(WEBRTC_LINUX)
+ protected:
+  void PushEnabledEvents();
+  void PopEnabledEvents();
+
+  void SetEnabledEvents(uint8_t events) override;
+  void EnableEvents(uint8_t events) override;
+  void DisableEvents(uint8_t events) override;
+#endif
+
+ private:
 #if defined(WEBRTC_WIN)
- private:
   static int next_id_;
   int id_;
   bool signal_close_;
   int signal_err_;
 #endif // WEBRTC_WIN
+#if defined(WEBRTC_POSIX) && defined(WEBRTC_LINUX)
+  void MaybeUpdateDispatcher(uint8_t old_events);
+
+  std::vector<uint8_t> enabled_events_stack_;
+#endif
 };
 
 } // namespace rtc
 
 #endif // WEBRTC_BASE_PHYSICALSOCKETSERVER_H__