// for compilers that support precompilation, includes "wx.h".
#include "wx/wxprec.h"
-#ifdef HAVE_SYS_EPOLL_H
+#if wxUSE_EPOLL_DISPATCHER
#include "wx/unix/private/epolldispatcher.h"
#include "wx/unix/private.h"
-#include "wx/module.h"
+#include "wx/stopwatch.h"
#ifndef WX_PRECOMP
#include "wx/log.h"
#include <sys/epoll.h>
#include <errno.h>
+#include <unistd.h>
#define wxEpollDispatcher_Trace wxT("epolldispatcher")
-static wxEpollDispatcher *gs_epollDispatcher = NULL;
-
// ============================================================================
// implementation
// ============================================================================
// helper: return EPOLLxxx mask corresponding to the given flags (and also log
// debugging messages about it)
-static uint32_t GetEpollMask(int flags, int fd)
+static uint32_t GetEpollMask(int flags, int WXUNUSED_UNLESS_DEBUG(fd))
{
uint32_t ep = 0;
// wxEpollDispatcher
// ----------------------------------------------------------------------------
-wxEpollDispatcher::wxEpollDispatcher()
+/* static */
+wxEpollDispatcher *wxEpollDispatcher::Create()
{
- m_epollDescriptor = epoll_create(1024);
- if ( m_epollDescriptor == -1 )
+ int epollDescriptor = epoll_create(1024);
+ if ( epollDescriptor == -1 )
{
wxLogSysError(_("Failed to create epoll descriptor"));
+ return NULL;
}
+ wxLogTrace(wxEpollDispatcher_Trace,
+ _T("Epoll fd %d created"), epollDescriptor);
+ return new wxEpollDispatcher(epollDescriptor);
}
-bool wxEpollDispatcher::RegisterFD(int fd, wxFDIOHandler* handler, int flags)
+wxEpollDispatcher::wxEpollDispatcher(int epollDescriptor)
{
- if ( !wxFDIODispatcher::RegisterFD(fd, handler, flags) )
- return false;
+ wxASSERT_MSG( epollDescriptor != -1, _T("invalid descriptor") );
+
+ m_epollDescriptor = epollDescriptor;
+}
+wxEpollDispatcher::~wxEpollDispatcher()
+{
+ if ( close(m_epollDescriptor) != 0 )
+ {
+ wxLogSysError(_("Error closing epoll descriptor"));
+ }
+}
+
+bool wxEpollDispatcher::RegisterFD(int fd, wxFDIOHandler* handler, int flags)
+{
epoll_event ev;
ev.events = GetEpollMask(flags, fd);
ev.data.ptr = handler;
return false;
}
+ wxLogTrace(wxEpollDispatcher_Trace,
+ _T("Added fd %d (handler %p) to epoll %d"), fd, handler, m_epollDescriptor);
return true;
}
bool wxEpollDispatcher::ModifyFD(int fd, wxFDIOHandler* handler, int flags)
{
- if ( !wxFDIODispatcher::ModifyFD(fd, handler, flags) )
- return false;
-
epoll_event ev;
ev.events = GetEpollMask(flags, fd);
ev.data.ptr = handler;
return false;
}
+ wxLogTrace(wxEpollDispatcher_Trace,
+ _T("Modified fd %d (handler: %p) on epoll %d"), fd, handler, m_epollDescriptor);
return true;
}
-wxFDIOHandler *wxEpollDispatcher::UnregisterFD(int fd, int flags)
+bool wxEpollDispatcher::UnregisterFD(int fd)
{
- wxFDIOHandler * const handler = wxFDIODispatcher::UnregisterFD(fd, flags);
- if ( !handler )
- return NULL;
-
epoll_event ev;
ev.events = 0;
ev.data.ptr = NULL;
wxLogSysError(_("Failed to unregister descriptor %d from epoll descriptor %d"),
fd, m_epollDescriptor);
}
+ wxLogTrace(wxEpollDispatcher_Trace,
+ _T("removed fd %d from %d"), fd, m_epollDescriptor);
+ return true;
+}
- return handler;
+int
+wxEpollDispatcher::DoPoll(epoll_event *events, int numEvents, int timeout) const
+{
+ // the code below relies on TIMEOUT_INFINITE being -1 so that we can pass
+ // timeout value directly to epoll_wait() which interprets -1 as meaning to
+ // wait forever and would need to be changed if the value of
+ // TIMEOUT_INFINITE ever changes
+ wxCOMPILE_TIME_ASSERT( TIMEOUT_INFINITE == -1, UpdateThisCode );
+
+ wxMilliClock_t timeEnd;
+ if ( timeout > 0 )
+ timeEnd = wxGetLocalTimeMillis();
+
+ int rc;
+ for ( ;; )
+ {
+ rc = epoll_wait(m_epollDescriptor, events, numEvents, timeout);
+ if ( rc != -1 || errno != EINTR )
+ break;
+
+ // we got interrupted, update the timeout and restart
+ if ( timeout > 0 )
+ {
+ timeout = wxMilliClockToLong(timeEnd - wxGetLocalTimeMillis());
+ if ( timeout < 0 )
+ return 0;
+ }
+ }
+
+ return rc;
}
-void wxEpollDispatcher::RunLoop(int timeout)
+bool wxEpollDispatcher::HasPending() const
+{
+ epoll_event event;
+
+ // NB: it's not really clear if epoll_wait() can return a number greater
+ // than the number of events passed to it but just in case it can, use
+ // >= instead of == here, see #10397
+ return DoPoll(&event, 1, 0) >= 1;
+}
+
+int wxEpollDispatcher::Dispatch(int timeout)
{
epoll_event events[16];
- const int e_num = epoll_wait
- (
- m_epollDescriptor,
- events,
- WXSIZEOF(events),
- timeout == TIMEOUT_INFINITE ? -1 : timeout
- );
+ const int rc = DoPoll(events, WXSIZEOF(events), timeout);
- if ( e_num == -1 )
+ if ( rc == -1 )
{
- if ( errno != EINTR )
- {
- wxLogSysError(_("Waiting for IO on epoll descriptor %d failed"),
- m_epollDescriptor);
- return;
- }
+ wxLogSysError(_("Waiting for IO on epoll descriptor %d failed"),
+ m_epollDescriptor);
+ return -1;
}
- for ( epoll_event *p = events; p < events + e_num; p++ )
+ int numEvents = 0;
+ for ( epoll_event *p = events; p < events + rc; p++ )
{
wxFDIOHandler * const handler = (wxFDIOHandler *)(p->data.ptr);
if ( !handler )
continue;
}
- if ( p->events & EPOLLIN )
+ // note that for compatibility with wxSelectDispatcher we call
+ // OnReadWaiting() on EPOLLHUP as this is what epoll_wait() returns
+ // when the write end of a pipe is closed while with select() the
+ // remaining pipe end becomes ready for reading when this happens
+ if ( p->events & (EPOLLIN | EPOLLHUP) )
handler->OnReadWaiting();
-
- if ( p->events & EPOLLOUT )
+ else if ( p->events & EPOLLOUT )
handler->OnWriteWaiting();
-
- if ( p->events & (EPOLLERR | EPOLLHUP) )
+ else if ( p->events & EPOLLERR )
handler->OnExceptionWaiting();
- }
-}
+ else
+ continue;
-/* static */
-wxEpollDispatcher *wxEpollDispatcher::Get()
-{
- if ( !gs_epollDispatcher )
- {
- gs_epollDispatcher = new wxEpollDispatcher;
- if ( !gs_epollDispatcher->IsOk() )
- {
- delete gs_epollDispatcher;
- gs_epollDispatcher = NULL;
- }
+ numEvents++;
}
- return gs_epollDispatcher;
+ return numEvents;
}
-// ----------------------------------------------------------------------------
-// wxEpollDispatcherModule
-// ----------------------------------------------------------------------------
-
-class wxEpollDispatcherModule : public wxModule
-{
-public:
- wxEpollDispatcherModule() { }
-
- virtual bool OnInit() { return true; }
- virtual void OnExit() { wxDELETE(gs_epollDispatcher); }
-
- DECLARE_DYNAMIC_CLASS(wxEpollDispatcherModule)
-};
-
-IMPLEMENT_DYNAMIC_CLASS(wxEpollDispatcherModule, wxModule)
-
-#endif // HAVE_SYS_EPOLL_H
+#endif // wxUSE_EPOLL_DISPATCHER
projects / wxWidgets.git / blobdiff