Index: net/tools/epoll_server/epoll_server.h |
diff --git a/net/tools/epoll_server/epoll_server.h b/net/tools/epoll_server/epoll_server.h |
deleted file mode 100644 |
index 92d6555ca6e154f6957275173526128af04d3932..0000000000000000000000000000000000000000 |
--- a/net/tools/epoll_server/epoll_server.h |
+++ /dev/null |
@@ -1,1053 +0,0 @@ |
-// Copyright 2013 The Chromium Authors. All rights reserved. |
-// Use of this source code is governed by a BSD-style license that can be |
-// found in the LICENSE file. |
- |
-#ifndef NET_TOOLS_EPOLL_SERVER_EPOLL_SERVER_H_ |
-#define NET_TOOLS_EPOLL_SERVER_EPOLL_SERVER_H_ |
- |
-#include <fcntl.h> |
-#include <sys/queue.h> |
-#include <map> |
-#include <set> |
-#include <string> |
-#include <utility> |
-#include <vector> |
- |
-// #define EPOLL_SERVER_EVENT_TRACING 1 |
-// |
-// Defining EPOLL_SERVER_EVENT_TRACING |
-// causes code to exist which didn't before. |
-// This code tracks each event generated by the epollserver, |
-// as well as providing a per-fd-registered summary of |
-// events. Note that enabling this code vastly slows |
-// down operations, and uses substantially more |
-// memory. For these reasons, it should only be enabled when doing |
-// developer debugging at his/her workstation. |
-// |
-// A structure called 'EventRecorder' will exist when |
-// the macro is defined. See the EventRecorder class interface |
-// within the EpollServer class for more details. |
-#ifdef EPOLL_SERVER_EVENT_TRACING |
-#include <ostream> |
-#include "base/logging.h" |
-#endif |
- |
-#include "base/basictypes.h" |
-#include "base/compiler_specific.h" |
-#include "base/containers/hash_tables.h" |
-#include "base/memory/scoped_ptr.h" |
-#include <sys/epoll.h> |
- |
-namespace net { |
- |
-class EpollServer; |
-class EpollAlarmCallbackInterface; |
-class ReadPipeCallback; |
- |
-struct EpollEvent { |
- EpollEvent(int events, bool is_epoll_wait) |
- : in_events(events), |
- out_ready_mask(0) { |
- } |
- |
- int in_events; // incoming events |
- int out_ready_mask; // the new event mask for ready list (0 means don't |
- // get on the ready list). This field is always |
- // initialized to 0 when the event is passed to |
- // OnEvent. |
-}; |
- |
-// Callbacks which go into EpollServers are expected to derive from this class. |
-class EpollCallbackInterface { |
- public: |
- // Summary: |
- // Called when the callback is registered into a EpollServer. |
- // Args: |
- // eps - the poll server into which this callback was registered |
- // fd - the file descriptor which was registered |
- // event_mask - the event mask (composed of EPOLLIN, EPOLLOUT, etc) |
- // which was registered (and will initially be used |
- // in the epoll() calls) |
- virtual void OnRegistration(EpollServer* eps, int fd, int event_mask) = 0; |
- |
- // Summary: |
- // Called when the event_mask is modified (for a file-descriptor) |
- // Args: |
- // fd - the file descriptor which was registered |
- // event_mask - the event mask (composed of EPOLLIN, EPOLLOUT, etc) |
- // which was is now curren (and will be used |
- // in subsequent epoll() calls) |
- virtual void OnModification(int fd, int event_mask) = 0; |
- |
- // Summary: |
- // Called whenever an event occurs on the file-descriptor. |
- // This is where the bulk of processing is expected to occur. |
- // Args: |
- // fd - the file descriptor which was registered |
- // event - a struct that contains the event mask (composed of EPOLLIN, |
- // EPOLLOUT, etc), a flag that indicates whether this is a true |
- // epoll_wait event vs one from the ready list, and an output |
- // parameter for OnEvent to inform the EpollServer whether to put |
- // this fd on the ready list. |
- virtual void OnEvent(int fd, EpollEvent* event) = 0; |
- |
- // Summary: |
- // Called when the file-descriptor is unregistered from the poll-server. |
- // Args: |
- // fd - the file descriptor which was registered, and of this call, is now |
- // unregistered. |
- // replaced - If true, this callback is being replaced by another, otherwise |
- // it is simply being removed. |
- virtual void OnUnregistration(int fd, bool replaced) = 0; |
- |
- // Summary: |
- // Called when the epoll server is shutting down. This is different from |
- // OnUnregistration because the subclass may want to clean up memory. |
- // This is called in leiu of OnUnregistration. |
- // Args: |
- // fd - the file descriptor which was registered. |
- virtual void OnShutdown(EpollServer* eps, int fd) = 0; |
- |
- virtual ~EpollCallbackInterface() {} |
- |
- protected: |
- EpollCallbackInterface() {} |
-}; |
- |
-//////////////////////////////////////////////////////////////////////////////// |
-//////////////////////////////////////////////////////////////////////////////// |
- |
-class EpollServer { |
- public: |
- typedef EpollAlarmCallbackInterface AlarmCB; |
- typedef EpollCallbackInterface CB; |
- |
- typedef std::multimap<int64, AlarmCB*> TimeToAlarmCBMap; |
- typedef TimeToAlarmCBMap::iterator AlarmRegToken; |
- |
- // Summary: |
- // Constructor: |
- // By default, we don't wait any amount of time for events, and |
- // we suggest to the epoll-system that we're going to use on-the-order |
- // of 1024 FDs. |
- EpollServer(); |
- |
- //////////////////////////////////////// |
- |
- // Destructor |
- virtual ~EpollServer(); |
- |
- //////////////////////////////////////// |
- |
- // Summary |
- // Register a callback to be called whenever an event contained |
- // in the set of events included in event_mask occurs on the |
- // file-descriptor 'fd' |
- // |
- // Note that only one callback is allowed to be registered for |
- // any specific file-decriptor. |
- // |
- // If a callback is registered for a file-descriptor which has already |
- // been registered, then the previous callback is unregistered with |
- // the 'replaced' flag set to true. I.e. the previous callback's |
- // OnUnregistration() function is called like so: |
- // OnUnregistration(fd, true); |
- // |
- // The epoll server does NOT take on ownership of the callback: the callback |
- // creator is responsible for managing that memory. |
- // |
- // Args: |
- // fd - a valid file-descriptor |
- // cb - an instance of a subclass of EpollCallbackInterface |
- // event_mask - a combination of (EPOLLOUT, EPOLLIN.. etc) indicating |
- // the events for which the callback would like to be |
- // called. |
- virtual void RegisterFD(int fd, CB* cb, int event_mask); |
- |
- //////////////////////////////////////// |
- |
- // Summary: |
- // A shortcut for RegisterFD which sets things up such that the |
- // callback is called when 'fd' is available for writing. |
- // Args: |
- // fd - a valid file-descriptor |
- // cb - an instance of a subclass of EpollCallbackInterface |
- virtual void RegisterFDForWrite(int fd, CB* cb); |
- |
- //////////////////////////////////////// |
- |
- // Summary: |
- // A shortcut for RegisterFD which sets things up such that the |
- // callback is called when 'fd' is available for reading or writing. |
- // Args: |
- // fd - a valid file-descriptor |
- // cb - an instance of a subclass of EpollCallbackInterface |
- virtual void RegisterFDForReadWrite(int fd, CB* cb); |
- |
- //////////////////////////////////////// |
- |
- // Summary: |
- // A shortcut for RegisterFD which sets things up such that the |
- // callback is called when 'fd' is available for reading. |
- // Args: |
- // fd - a valid file-descriptor |
- // cb - an instance of a subclass of EpollCallbackInterface |
- virtual void RegisterFDForRead(int fd, CB* cb); |
- |
- //////////////////////////////////////// |
- |
- // Summary: |
- // Removes the FD and the associated callback from the pollserver. |
- // If the callback is registered with other FDs, they will continue |
- // to be processed using the callback without modification. |
- // If the file-descriptor specified is not registered in the |
- // epoll_server, then nothing happens as a result of this call. |
- // Args: |
- // fd - the file-descriptor which should no-longer be monitored. |
- virtual void UnregisterFD(int fd); |
- |
- //////////////////////////////////////// |
- |
- // Summary: |
- // Modifies the event mask for the file-descriptor, replacing |
- // the old event_mask with the new one specified here. |
- // If the file-descriptor specified is not registered in the |
- // epoll_server, then nothing happens as a result of this call. |
- // Args: |
- // fd - the fd whose event mask should be modified. |
- // event_mask - the new event mask. |
- virtual void ModifyCallback(int fd, int event_mask); |
- |
- //////////////////////////////////////// |
- |
- // Summary: |
- // Modifies the event mask for the file-descriptor such that we |
- // no longer request events when 'fd' is readable. |
- // If the file-descriptor specified is not registered in the |
- // epoll_server, then nothing happens as a result of this call. |
- // Args: |
- // fd - the fd whose event mask should be modified. |
- virtual void StopRead(int fd); |
- |
- //////////////////////////////////////// |
- |
- // Summary: |
- // Modifies the event mask for the file-descriptor such that we |
- // request events when 'fd' is readable. |
- // If the file-descriptor specified is not registered in the |
- // epoll_server, then nothing happens as a result of this call. |
- // Args: |
- // fd - the fd whose event mask should be modified. |
- virtual void StartRead(int fd); |
- |
- //////////////////////////////////////// |
- |
- // Summary: |
- // Modifies the event mask for the file-descriptor such that we |
- // no longer request events when 'fd' is writable. |
- // If the file-descriptor specified is not registered in the |
- // epoll_server, then nothing happens as a result of this call. |
- // Args: |
- // fd - the fd whose event mask should be modified. |
- virtual void StopWrite(int fd); |
- |
- //////////////////////////////////////// |
- |
- // Summary: |
- // Modifies the event mask for the file-descriptor such that we |
- // request events when 'fd' is writable. |
- // If the file-descriptor specified is not registered in the |
- // epoll_server, then nothing happens as a result of this call. |
- // Args: |
- // fd - the fd whose event mask should be modified. |
- virtual void StartWrite(int fd); |
- |
- //////////////////////////////////////// |
- |
- // Summary: |
- // Looks up the callback associated with the file-desriptor 'fd'. |
- // If a callback is associated with this file-descriptor, then |
- // it's OnEvent() method is called with the file-descriptor 'fd', |
- // and event_mask 'event_mask' |
- // |
- // If no callback is registered for this file-descriptor, nothing |
- // will happen as a result of this call. |
- // |
- // This function is used internally by the EpollServer, but is |
- // available publically so that events might be 'faked'. Calling |
- // this function with an fd and event_mask is equivalent (as far |
- // as the callback is concerned) to having a real event generated |
- // by epoll (except, of course, that read(), etc won't necessarily |
- // be able to read anything) |
- // Args: |
- // fd - the file-descriptor on which an event has occured. |
- // event_mask - a bitmask representing the events which have occured |
- // on/for this fd. This bitmask is composed of |
- // POLLIN, POLLOUT, etc. |
- // |
- void HandleEvent(int fd, int event_mask); |
- |
- // Summary: |
- // Call this when you want the pollserver to |
- // wait for events and execute the callbacks associated with |
- // the file-descriptors on which those events have occured. |
- // Depending on the value of timeout_in_us_, this may or may |
- // not return immediately. Please reference the set_timeout() |
- // function for the specific behaviour. |
- virtual void WaitForEventsAndExecuteCallbacks(); |
- |
- // Summary: |
- // When an fd is registered to use edge trigger notification, the ready |
- // list can be used to simulate level trigger semantics. Edge trigger |
- // registration doesn't send an initial event, and only rising edge (going |
- // from blocked to unblocked) events are sent. A callback can put itself on |
- // the ready list by calling SetFDReady() after calling RegisterFD(). The |
- // OnEvent method of all callbacks associated with the fds on the ready |
- // list will be called immediately after processing the events returned by |
- // epoll_wait(). The fd is removed from the ready list before the |
- // callback's OnEvent() method is invoked. To stay on the ready list, the |
- // OnEvent() (or some function in that call chain) must call SetFDReady |
- // again. When a fd is unregistered using UnregisterFD(), the fd is |
- // automatically removed from the ready list. |
- // |
- // When the callback for a edge triggered fd hits the falling edge (about |
- // to block, either because of it got an EAGAIN, or had a short read/write |
- // operation), it should remove itself from the ready list using |
- // SetFDNotReady() (since OnEvent cannot distinguish between invocation |
- // from the ready list vs from a normal epoll event). All four ready list |
- // methods are safe to be called within the context of the callbacks. |
- // |
- // Since the ready list invokes EpollCallbackInterface::OnEvent, only fds |
- // that are registered with the EpollServer will be put on the ready list. |
- // SetFDReady() and SetFDNotReady() will do nothing if the EpollServer |
- // doesn't know about the fd passed in. |
- // |
- // Since the ready list cannot reliably determine proper set of events |
- // which should be sent to the callback, SetFDReady() requests the caller |
- // to provide the ready list with the event mask, which will be used later |
- // when OnEvent() is invoked by the ready list. Hence, the event_mask |
- // passedto SetFDReady() does not affect the actual epoll registration of |
- // the fd with the kernel. If a fd is already put on the ready list, and |
- // SetFDReady() is called again for that fd with a different event_mask, |
- // the event_mask will be updated. |
- virtual void SetFDReady(int fd, int events_to_fake); |
- |
- virtual void SetFDNotReady(int fd); |
- |
- // Summary: |
- // IsFDReady(), ReadyListSize(), and VerifyReadyList are intended as |
- // debugging tools and for writing unit tests. |
- // ISFDReady() returns whether a fd is in the ready list. |
- // ReadyListSize() returns the number of fds on the ready list. |
- // VerifyReadyList() checks the consistency of internal data structure. It |
- // will CHECK if it finds an error. |
- virtual bool IsFDReady(int fd) const; |
- |
- size_t ReadyListSize() const { return ready_list_size_; } |
- |
- void VerifyReadyList() const; |
- |
- //////////////////////////////////////// |
- |
- // Summary: |
- // Registers an alarm 'ac' to go off at time 'timeout_time_in_us'. |
- // If the callback returns a positive number from its OnAlarm() function, |
- // then the callback will be re-registered at that time, else the alarm |
- // owner is responsible for freeing up memory. |
- // |
- // Important: A give AlarmCB* can not be registered again if it is already |
- // registered. If a user wants to register a callback again it should first |
- // unregister the previous callback before calling RegisterAlarm again. |
- // Args: |
- // timeout_time_in_us - the absolute time at which the alarm should go off |
- // ac - the alarm which will be called. |
- virtual void RegisterAlarm(int64 timeout_time_in_us, AlarmCB* ac); |
- |
- // Summary: |
- // Registers an alarm 'ac' to go off at time: (ApproximateNowInUs() + |
- // delta_in_us). While this is somewhat less accurate (see the description |
- // for ApproximateNowInUs() to see how 'approximate'), the error is never |
- // worse than the amount of time it takes to process all events in one |
- // WaitForEvents. As with 'RegisterAlarm()', if the callback returns a |
- // positive number from its OnAlarm() function, then the callback will be |
- // re-registered at that time, else the alarm owner is responsible for |
- // freeing up memory. |
- // Note that this function is purely a convienence. The |
- // same thing may be accomplished by using RegisterAlarm with |
- // ApproximateNowInUs() directly. |
- // |
- // Important: A give AlarmCB* can not be registered again if it is already |
- // registered. If a user wants to register a callback again it should first |
- // unregister the previous callback before calling RegisterAlarm again. |
- // Args: |
- // delta_in_us - the delta in microseconds from the ApproximateTimeInUs() at |
- // which point the alarm should go off. |
- // ac - the alarm which will be called. |
- void RegisterAlarmApproximateDelta(int64 delta_in_us, AlarmCB* ac) { |
- RegisterAlarm(ApproximateNowInUsec() + delta_in_us, ac); |
- } |
- |
- //////////////////////////////////////// |
- |
- // Summary: |
- // Unregister the alarm referred to by iterator_token; Callers should |
- // be warned that a token may have become already invalid when OnAlarm() |
- // is called, was unregistered, or OnShutdown was called on that alarm. |
- // Args: |
- // iterator_token - iterator to the alarm callback to unregister. |
- virtual void UnregisterAlarm( |
- const EpollServer::AlarmRegToken& iterator_token); |
- |
- //////////////////////////////////////// |
- |
- // Summary: |
- // returns the number of file-descriptors registered in this EpollServer. |
- // Returns: |
- // number of FDs registered (discounting the internal pipe used for Wake) |
- virtual int NumFDsRegistered() const; |
- |
- // Summary: |
- // Force the epoll server to wake up (by writing to an internal pipe). |
- virtual void Wake(); |
- |
- // Summary: |
- // Wrapper around WallTimer's NowInUsec. We do this so that we can test |
- // EpollServer without using the system clock (and can avoid the flakiness |
- // that would ensue) |
- // Returns: |
- // the current time as number of microseconds since the Unix epoch. |
- virtual int64 NowInUsec() const; |
- |
- // Summary: |
- // Since calling NowInUsec() many thousands of times per |
- // WaitForEventsAndExecuteCallbacks function call is, to say the least, |
- // inefficient, we allow users to use an approximate time instead. The |
- // time returned from this function is as accurate as NowInUsec() when |
- // WaitForEventsAndExecuteCallbacks is not an ancestor of the caller's |
- // callstack. |
- // However, when WaitForEventsAndExecuteCallbacks -is- an ancestor, then |
- // this function returns the time at which the |
- // WaitForEventsAndExecuteCallbacks function started to process events or |
- // alarms. |
- // |
- // Essentially, this function makes available a fast and mostly accurate |
- // mechanism for getting the time for any function handling an event or |
- // alarm. When functions which are not handling callbacks or alarms call |
- // this function, they get the slow and "absolutely" accurate time. |
- // |
- // Users should be encouraged to use this function. |
- // Returns: |
- // the "approximate" current time as number of microseconds since the Unix |
- // epoch. |
- virtual int64 ApproximateNowInUsec() const; |
- |
- static std::string EventMaskToString(int event_mask); |
- |
- // Summary: |
- // Logs the state of the epoll server with LOG(ERROR). |
- void LogStateOnCrash(); |
- |
- // Summary: |
- // Set the timeout to the value specified. |
- // If the timeout is set to a negative number, |
- // WaitForEventsAndExecuteCallbacks() will only return when an event has |
- // occured |
- // If the timeout is set to zero, |
- // WaitForEventsAndExecuteCallbacks() will return immediately |
- // If the timeout is set to a positive number, |
- // WaitForEventsAndExecuteCallbacks() will return when an event has |
- // occured, or when timeout_in_us microseconds has elapsed, whichever |
- // is first. |
- // Args: |
- // timeout_in_us - value specified depending on behaviour desired. |
- // See above. |
- void set_timeout_in_us(int64 timeout_in_us) { |
- timeout_in_us_ = timeout_in_us; |
- } |
- |
- //////////////////////////////////////// |
- |
- // Summary: |
- // Accessor for the current value of timeout_in_us. |
- int timeout_in_us() const { return timeout_in_us_; } |
- |
- // Summary: |
- // Returns true when the EpollServer() is being destroyed. |
- bool in_shutdown() const { return in_shutdown_; } |
- |
- bool ContainsAlarm(EpollAlarmCallbackInterface* alarm) const { |
- return all_alarms_.find(alarm) != all_alarms_.end(); |
- } |
- |
- // Summary: |
- // A function for implementing the ready list. It invokes OnEvent for each |
- // of the fd in the ready list, and takes care of adding them back to the |
- // ready list if the callback requests it (by checking that out_ready_mask |
- // is non-zero). |
- void CallReadyListCallbacks(); |
- |
- protected: |
- virtual int GetFlags(int fd); |
- inline int SetFlags(int fd, int flags) { |
- return fcntl(fd, F_SETFL, flags | O_NONBLOCK); |
- } |
- |
- virtual void SetNonblocking(int fd); |
- |
- // This exists here so that we can override this function in unittests |
- // in order to make effective mock EpollServer objects. |
- virtual int epoll_wait_impl(int epfd, |
- struct epoll_event* events, |
- int max_events, |
- int timeout_in_ms); |
- |
- // this struct is used internally, and is never used by anything external |
- // to this class. Some of its members are declared mutable to get around the |
- // restriction imposed by hash_set. Since hash_set knows nothing about the |
- // objects it stores, it has to assume that every bit of the object is used |
- // in the hash function and equal_to comparison. Thus hash_set::iterator is a |
- // const iterator. In this case, the only thing that must stay constant is |
- // fd. Everything else are just along for the ride and changing them doesn't |
- // compromise the hash_set integrity. |
- struct CBAndEventMask { |
- CBAndEventMask() |
- : cb(NULL), |
- fd(-1), |
- event_mask(0), |
- events_asserted(0), |
- events_to_fake(0), |
- in_use(false) { |
- entry.le_next = NULL; |
- entry.le_prev = NULL; |
- } |
- |
- CBAndEventMask(EpollCallbackInterface* cb, |
- int event_mask, |
- int fd) |
- : cb(cb), fd(fd), event_mask(event_mask), events_asserted(0), |
- events_to_fake(0), in_use(false) { |
- entry.le_next = NULL; |
- entry.le_prev = NULL; |
- } |
- |
- // Required operator for hash_set. Normally operator== should be a free |
- // standing function. However, since CBAndEventMask is a protected type and |
- // it will never be a base class, it makes no difference. |
- bool operator==(const CBAndEventMask& cb_and_mask) const { |
- return fd == cb_and_mask.fd; |
- } |
- // A callback. If the fd is unregistered inside the callchain of OnEvent, |
- // the cb will be set to NULL. |
- mutable EpollCallbackInterface* cb; |
- |
- mutable LIST_ENTRY(CBAndEventMask) entry; |
- // file descriptor registered with the epoll server. |
- int fd; |
- // the current event_mask registered for this callback. |
- mutable int event_mask; |
- // the event_mask that was returned by epoll |
- mutable int events_asserted; |
- // the event_mask for the ready list to use to call OnEvent. |
- mutable int events_to_fake; |
- // toggle around calls to OnEvent to tell UnregisterFD to not erase the |
- // iterator because HandleEvent is using it. |
- mutable bool in_use; |
- }; |
- |
- // Custom hash function to be used by hash_set. |
- struct CBAndEventMaskHash { |
- size_t operator()(const CBAndEventMask& cb_and_eventmask) const { |
- return static_cast<size_t>(cb_and_eventmask.fd); |
- } |
- }; |
- |
- typedef base::hash_set<CBAndEventMask, CBAndEventMaskHash> FDToCBMap; |
- |
- // the following four functions are OS-specific, and are likely |
- // to be changed in a subclass if the poll/select method is changed |
- // from epoll. |
- |
- // Summary: |
- // Deletes a file-descriptor from the set of FDs that should be |
- // monitored with epoll. |
- // Note that this only deals with modifying data relating -directly- |
- // with the epoll call-- it does not modify any data within the |
- // epoll_server. |
- // Args: |
- // fd - the file descriptor to-be-removed from the monitoring set |
- virtual void DelFD(int fd) const; |
- |
- //////////////////////////////////////// |
- |
- // Summary: |
- // Adds a file-descriptor to the set of FDs that should be |
- // monitored with epoll. |
- // Note that this only deals with modifying data relating -directly- |
- // with the epoll call. |
- // Args: |
- // fd - the file descriptor to-be-added to the monitoring set |
- // event_mask - the event mask (consisting of EPOLLIN, EPOLLOUT, etc |
- // OR'd together) which will be associated with this |
- // FD initially. |
- virtual void AddFD(int fd, int event_mask) const; |
- |
- //////////////////////////////////////// |
- |
- // Summary: |
- // Modifies a file-descriptor in the set of FDs that should be |
- // monitored with epoll. |
- // Note that this only deals with modifying data relating -directly- |
- // with the epoll call. |
- // Args: |
- // fd - the file descriptor to-be-added to the monitoring set |
- // event_mask - the event mask (consisting of EPOLLIN, EPOLLOUT, etc |
- // OR'd together) which will be associated with this |
- // FD after this call. |
- virtual void ModFD(int fd, int event_mask) const; |
- |
- //////////////////////////////////////// |
- |
- // Summary: |
- // Modified the event mask associated with an FD in the set of |
- // data needed by epoll. |
- // Events are removed before they are added, thus, if ~0 is put |
- // in 'remove_event', whatever is put in 'add_event' will be |
- // the new event mask. |
- // If the file-descriptor specified is not registered in the |
- // epoll_server, then nothing happens as a result of this call. |
- // Args: |
- // fd - the file descriptor whose event mask is to be modified |
- // remove_event - the events which are to be removed from the current |
- // event_mask |
- // add_event - the events which are to be added to the current event_mask |
- // |
- // |
- virtual void ModifyFD(int fd, int remove_event, int add_event); |
- |
- //////////////////////////////////////// |
- |
- // Summary: |
- // Waits for events, and calls HandleEvents() for each |
- // fd, event pair discovered to possibly have an event. |
- // Note that a callback (B) may get a spurious event if |
- // another callback (A) has closed a file-descriptor N, and |
- // the callback (B) has a newly opened file-descriptor, which |
- // also happens to be N. |
- virtual void WaitForEventsAndCallHandleEvents(int64 timeout_in_us, |
- struct epoll_event events[], |
- int events_size); |
- |
- |
- |
- // Summary: |
- // An internal function for implementing the ready list. It adds a fd's |
- // CBAndEventMask to the ready list. If the fd is already on the ready |
- // list, it is a no-op. |
- void AddToReadyList(CBAndEventMask* cb_and_mask); |
- |
- // Summary: |
- // An internal function for implementing the ready list. It remove a fd's |
- // CBAndEventMask from the ready list. If the fd is not on the ready list, |
- // it is a no-op. |
- void RemoveFromReadyList(const CBAndEventMask& cb_and_mask); |
- |
- // Summary: |
- // Calls any pending alarms that should go off and reregisters them if they |
- // were recurring. |
- virtual void CallAndReregisterAlarmEvents(); |
- |
- // The file-descriptor created for epolling |
- int epoll_fd_; |
- |
- // The mapping of file-descriptor to CBAndEventMasks |
- FDToCBMap cb_map_; |
- |
- // Custom hash function to be used by hash_set. |
- struct AlarmCBHash { |
- size_t operator()(AlarmCB*const& p) const { |
- return reinterpret_cast<size_t>(p); |
- } |
- }; |
- |
- |
- // TOOD(sushantj): Having this hash_set is avoidable. We currently have it |
- // only so that we can enforce stringent checks that a caller can not register |
- // the same alarm twice. One option is to have an implementation in which |
- // this hash_set is used only in the debug mode. |
- typedef base::hash_set<AlarmCB*, AlarmCBHash> AlarmCBMap; |
- AlarmCBMap all_alarms_; |
- |
- TimeToAlarmCBMap alarm_map_; |
- |
- // The amount of time in microseconds that we'll wait before returning |
- // from the WaitForEventsAndExecuteCallbacks() function. |
- // If this is positive, wait that many microseconds. |
- // If this is negative, wait forever, or for the first event that occurs |
- // If this is zero, never wait for an event. |
- int64 timeout_in_us_; |
- |
- // This is nonzero only after the invocation of epoll_wait_impl within |
- // WaitForEventsAndCallHandleEvents and before the function |
- // WaitForEventsAndExecuteCallbacks returns. At all other times, this is |
- // zero. This enables us to have relatively accurate time returned from the |
- // ApproximateNowInUs() function. See that function for more details. |
- int64 recorded_now_in_us_; |
- |
- // This is used to implement CallAndReregisterAlarmEvents. This stores |
- // all alarms that were reregistered because OnAlarm() returned a |
- // value > 0 and the time at which they should be executed is less that |
- // the current time. By storing such alarms in this map we ensure |
- // that while calling CallAndReregisterAlarmEvents we do not call |
- // OnAlarm on any alarm in this set. This ensures that we do not |
- // go in an infinite loop. |
- AlarmCBMap alarms_reregistered_and_should_be_skipped_; |
- |
- LIST_HEAD(ReadyList, CBAndEventMask) ready_list_; |
- LIST_HEAD(TmpList, CBAndEventMask) tmp_list_; |
- int ready_list_size_; |
- // TODO(alyssar): make this into something that scales up. |
- static const int events_size_ = 256; |
- struct epoll_event events_[256]; |
- |
-#ifdef EPOLL_SERVER_EVENT_TRACING |
- struct EventRecorder { |
- public: |
- EventRecorder() : num_records_(0), record_threshold_(10000) {} |
- |
- ~EventRecorder() { |
- Clear(); |
- } |
- |
- // When a number of events equals the record threshold, |
- // the collected data summary for all FDs will be written |
- // to LOG(INFO). Note that this does not include the |
- // individual events (if you'reinterested in those, you'll |
- // have to get at them programmatically). |
- // After any such flushing to LOG(INFO) all events will |
- // be cleared. |
- // Note that the definition of an 'event' is a bit 'hazy', |
- // as it includes the 'Unregistration' event, and perhaps |
- // others. |
- void set_record_threshold(int64 new_threshold) { |
- record_threshold_ = new_threshold; |
- } |
- |
- void Clear() { |
- for (int i = 0; i < debug_events_.size(); ++i) { |
- delete debug_events_[i]; |
- } |
- debug_events_.clear(); |
- unregistered_fds_.clear(); |
- event_counts_.clear(); |
- } |
- |
- void MaybeRecordAndClear() { |
- ++num_records_; |
- if ((num_records_ > record_threshold_) && |
- (record_threshold_ > 0)) { |
- LOG(INFO) << "\n" << *this; |
- num_records_ = 0; |
- Clear(); |
- } |
- } |
- |
- void RecordFDMaskEvent(int fd, int mask, const char* function) { |
- FDMaskOutput* fdmo = new FDMaskOutput(fd, mask, function); |
- debug_events_.push_back(fdmo); |
- MaybeRecordAndClear(); |
- } |
- |
- void RecordEpollWaitEvent(int timeout_in_ms, |
- int num_events_generated) { |
- EpollWaitOutput* ewo = new EpollWaitOutput(timeout_in_ms, |
- num_events_generated); |
- debug_events_.push_back(ewo); |
- MaybeRecordAndClear(); |
- } |
- |
- void RecordEpollEvent(int fd, int event_mask) { |
- Events& events_for_fd = event_counts_[fd]; |
- events_for_fd.AssignFromMask(event_mask); |
- MaybeRecordAndClear(); |
- } |
- |
- friend ostream& operator<<(ostream& os, const EventRecorder& er) { |
- for (int i = 0; i < er.unregistered_fds_.size(); ++i) { |
- os << "fd: " << er.unregistered_fds_[i] << "\n"; |
- os << er.unregistered_fds_[i]; |
- } |
- for (EventCountsMap::const_iterator i = er.event_counts_.begin(); |
- i != er.event_counts_.end(); |
- ++i) { |
- os << "fd: " << i->first << "\n"; |
- os << i->second; |
- } |
- for (int i = 0; i < er.debug_events_.size(); ++i) { |
- os << *(er.debug_events_[i]) << "\n"; |
- } |
- return os; |
- } |
- |
- void RecordUnregistration(int fd) { |
- EventCountsMap::iterator i = event_counts_.find(fd); |
- if (i != event_counts_.end()) { |
- unregistered_fds_.push_back(i->second); |
- event_counts_.erase(i); |
- } |
- MaybeRecordAndClear(); |
- } |
- |
- protected: |
- class DebugOutput { |
- public: |
- friend ostream& operator<<(ostream& os, const DebugOutput& debug_output) { |
- debug_output.OutputToStream(os); |
- return os; |
- } |
- virtual void OutputToStream(ostream* os) const = 0; |
- virtual ~DebugOutput() {} |
- }; |
- |
- class FDMaskOutput : public DebugOutput { |
- public: |
- FDMaskOutput(int fd, int mask, const char* function) : |
- fd_(fd), mask_(mask), function_(function) {} |
- virtual void OutputToStream(ostream* os) const { |
- (*os) << "func: " << function_ |
- << "\tfd: " << fd_; |
- if (mask_ != 0) { |
- (*os) << "\tmask: " << EventMaskToString(mask_); |
- } |
- } |
- int fd_; |
- int mask_; |
- const char* function_; |
- }; |
- |
- class EpollWaitOutput : public DebugOutput { |
- public: |
- EpollWaitOutput(int timeout_in_ms, |
- int num_events_generated) : |
- timeout_in_ms_(timeout_in_ms), |
- num_events_generated_(num_events_generated) {} |
- virtual void OutputToStream(ostream* os) const { |
- (*os) << "timeout_in_ms: " << timeout_in_ms_ |
- << "\tnum_events_generated: " << num_events_generated_; |
- } |
- protected: |
- int timeout_in_ms_; |
- int num_events_generated_; |
- }; |
- |
- struct Events { |
- Events() : |
- epoll_in(0), |
- epoll_pri(0), |
- epoll_out(0), |
- epoll_rdnorm(0), |
- epoll_rdband(0), |
- epoll_wrnorm(0), |
- epoll_wrband(0), |
- epoll_msg(0), |
- epoll_err(0), |
- epoll_hup(0), |
- epoll_oneshot(0), |
- epoll_et(0) {} |
- |
- void AssignFromMask(int event_mask) { |
- if (event_mask & EPOLLIN) ++epoll_in; |
- if (event_mask & EPOLLPRI) ++epoll_pri; |
- if (event_mask & EPOLLOUT) ++epoll_out; |
- if (event_mask & EPOLLRDNORM) ++epoll_rdnorm; |
- if (event_mask & EPOLLRDBAND) ++epoll_rdband; |
- if (event_mask & EPOLLWRNORM) ++epoll_wrnorm; |
- if (event_mask & EPOLLWRBAND) ++epoll_wrband; |
- if (event_mask & EPOLLMSG) ++epoll_msg; |
- if (event_mask & EPOLLERR) ++epoll_err; |
- if (event_mask & EPOLLHUP) ++epoll_hup; |
- if (event_mask & EPOLLONESHOT) ++epoll_oneshot; |
- if (event_mask & EPOLLET) ++epoll_et; |
- }; |
- |
- friend ostream& operator<<(ostream& os, const Events& ev) { |
- if (ev.epoll_in) { |
- os << "\t EPOLLIN: " << ev.epoll_in << "\n"; |
- } |
- if (ev.epoll_pri) { |
- os << "\t EPOLLPRI: " << ev.epoll_pri << "\n"; |
- } |
- if (ev.epoll_out) { |
- os << "\t EPOLLOUT: " << ev.epoll_out << "\n"; |
- } |
- if (ev.epoll_rdnorm) { |
- os << "\t EPOLLRDNORM: " << ev.epoll_rdnorm << "\n"; |
- } |
- if (ev.epoll_rdband) { |
- os << "\t EPOLLRDBAND: " << ev.epoll_rdband << "\n"; |
- } |
- if (ev.epoll_wrnorm) { |
- os << "\t EPOLLWRNORM: " << ev.epoll_wrnorm << "\n"; |
- } |
- if (ev.epoll_wrband) { |
- os << "\t EPOLLWRBAND: " << ev.epoll_wrband << "\n"; |
- } |
- if (ev.epoll_msg) { |
- os << "\t EPOLLMSG: " << ev.epoll_msg << "\n"; |
- } |
- if (ev.epoll_err) { |
- os << "\t EPOLLERR: " << ev.epoll_err << "\n"; |
- } |
- if (ev.epoll_hup) { |
- os << "\t EPOLLHUP: " << ev.epoll_hup << "\n"; |
- } |
- if (ev.epoll_oneshot) { |
- os << "\t EPOLLONESHOT: " << ev.epoll_oneshot << "\n"; |
- } |
- if (ev.epoll_et) { |
- os << "\t EPOLLET: " << ev.epoll_et << "\n"; |
- } |
- return os; |
- } |
- |
- unsigned int epoll_in; |
- unsigned int epoll_pri; |
- unsigned int epoll_out; |
- unsigned int epoll_rdnorm; |
- unsigned int epoll_rdband; |
- unsigned int epoll_wrnorm; |
- unsigned int epoll_wrband; |
- unsigned int epoll_msg; |
- unsigned int epoll_err; |
- unsigned int epoll_hup; |
- unsigned int epoll_oneshot; |
- unsigned int epoll_et; |
- }; |
- |
- std::vector<DebugOutput*> debug_events_; |
- std::vector<Events> unregistered_fds_; |
- typedef base::hash_map<int, Events> EventCountsMap; |
- EventCountsMap event_counts_; |
- int64 num_records_; |
- int64 record_threshold_; |
- }; |
- |
- void ClearEventRecords() { |
- event_recorder_.Clear(); |
- } |
- void WriteEventRecords(ostream* os) const { |
- (*os) << event_recorder_; |
- } |
- |
- mutable EventRecorder event_recorder_; |
- |
-#endif |
- |
- private: |
- // Helper functions used in the destructor. |
- void CleanupFDToCBMap(); |
- void CleanupTimeToAlarmCBMap(); |
- |
- // The callback registered to the fds below. As the purpose of their |
- // registration is to wake the epoll server it just clears the pipe and |
- // returns. |
- scoped_ptr<ReadPipeCallback> wake_cb_; |
- |
- // A pipe owned by the epoll server. The server will be registered to listen |
- // on read_fd_ and can be woken by Wake() which writes to write_fd_. |
- int read_fd_; |
- int write_fd_; |
- |
- // This boolean is checked to see if it is false at the top of the |
- // WaitForEventsAndExecuteCallbacks function. If not, then it either returns |
- // without doing work, and logs to ERROR, or aborts the program (in |
- // DEBUG mode). If so, then it sets the bool to true, does work, and |
- // sets it back to false when done. This catches unwanted recursion. |
- bool in_wait_for_events_and_execute_callbacks_; |
- |
- // Returns true when the EpollServer() is being destroyed. |
- bool in_shutdown_; |
- |
- DISALLOW_COPY_AND_ASSIGN(EpollServer); |
-}; |
- |
-class EpollAlarmCallbackInterface { |
- public: |
- // Summary: |
- // Called when an alarm times out. Invalidates an AlarmRegToken. |
- // WARNING: If a token was saved to refer to an alarm callback, OnAlarm must |
- // delete it, as the reference is no longer valid. |
- // Returns: |
- // the unix time (in microseconds) at which this alarm should be signaled |
- // again, or 0 if the alarm should be removed. |
- virtual int64 OnAlarm() = 0; |
- |
- // Summary: |
- // Called when the an alarm is registered. Invalidates an AlarmRegToken. |
- // Args: |
- // token: the iterator to the the alarm registered in the alarm map. |
- // WARNING: this token becomes invalid when the alarm fires, is |
- // unregistered, or OnShutdown is called on that alarm. |
- // eps: the epoll server the alarm is registered with. |
- virtual void OnRegistration(const EpollServer::AlarmRegToken& token, |
- EpollServer* eps) = 0; |
- |
- // Summary: |
- // Called when the an alarm is unregistered. |
- // WARNING: It is not valid to unregister a callback and then use the token |
- // that was saved to refer to the callback. |
- virtual void OnUnregistration() = 0; |
- |
- // Summary: |
- // Called when the epoll server is shutting down. |
- // Invalidates the AlarmRegToken that was given when this alarm was |
- // registered. |
- virtual void OnShutdown(EpollServer* eps) = 0; |
- |
- virtual ~EpollAlarmCallbackInterface() {} |
- |
- protected: |
- EpollAlarmCallbackInterface() {} |
-}; |
- |
-// A simple alarm which unregisters itself on destruction. |
-// |
-// PLEASE NOTE: |
-// Any classes overriding these functions must either call the implementation |
-// of the parent class, or is must otherwise make sure that the 'registered_' |
-// boolean and the token, 'token_', are updated appropriately. |
-class EpollAlarm : public EpollAlarmCallbackInterface { |
- public: |
- EpollAlarm(); |
- |
- ~EpollAlarm() override; |
- |
- // Marks the alarm as unregistered and returns 0. The return value may be |
- // safely ignored by subclasses. |
- int64 OnAlarm() override; |
- |
- // Marks the alarm as registered, and stores the token. |
- void OnRegistration(const EpollServer::AlarmRegToken& token, |
- EpollServer* eps) override; |
- |
- // Marks the alarm as unregistered. |
- void OnUnregistration() override; |
- |
- // Marks the alarm as unregistered. |
- void OnShutdown(EpollServer* eps) override; |
- |
- // If the alarm was registered, unregister it. |
- void UnregisterIfRegistered(); |
- |
- bool registered() const { return registered_; } |
- |
- const EpollServer* eps() const { return eps_; } |
- |
- private: |
- EpollServer::AlarmRegToken token_; |
- EpollServer* eps_; |
- bool registered_; |
-}; |
- |
-} // namespace net |
- |
-#endif // NET_TOOLS_EPOLL_SERVER_EPOLL_SERVER_H_ |