]>
Commit | Line | Data |
---|---|---|
b46b1d59 VZ |
1 | /////////////////////////////////////////////////////////////////////////////// |
2 | // Name: src/common/epolldispatcher.cpp | |
3 | // Purpose: implements dispatcher for epoll_wait() call | |
4 | // Author: Lukasz Michalski | |
5 | // Created: April 2007 | |
6 | // RCS-ID: $Id$ | |
7 | // Copyright: (c) 2007 Lukasz Michalski | |
8 | // License: wxWindows licence | |
9 | /////////////////////////////////////////////////////////////////////////////// | |
10 | ||
11 | // ============================================================================ | |
12 | // declarations | |
13 | // ============================================================================ | |
14 | ||
15 | // ---------------------------------------------------------------------------- | |
16 | // headers | |
17 | // ---------------------------------------------------------------------------- | |
18 | ||
19 | // for compilers that support precompilation, includes "wx.h". | |
20 | #include "wx/wxprec.h" | |
21 | ||
9550ee98 | 22 | #if wxUSE_EPOLL_DISPATCHER |
b46b1d59 VZ |
23 | |
24 | #include "wx/unix/private/epolldispatcher.h" | |
25 | #include "wx/unix/private.h" | |
b46b1d59 | 26 | |
b5ef33b2 VS |
27 | #ifndef WX_PRECOMP |
28 | #include "wx/log.h" | |
29 | #include "wx/intl.h" | |
30 | #endif | |
31 | ||
b46b1d59 VZ |
32 | #include <sys/epoll.h> |
33 | #include <errno.h> | |
98883bca | 34 | #include <unistd.h> |
b46b1d59 VZ |
35 | |
36 | #define wxEpollDispatcher_Trace wxT("epolldispatcher") | |
37 | ||
b46b1d59 VZ |
38 | // ============================================================================ |
39 | // implementation | |
40 | // ============================================================================ | |
41 | ||
42 | // helper: return EPOLLxxx mask corresponding to the given flags (and also log | |
43 | // debugging messages about it) | |
e0d1fd7f | 44 | static uint32_t GetEpollMask(int flags, int WXUNUSED_UNLESS_DEBUG(fd)) |
b46b1d59 VZ |
45 | { |
46 | uint32_t ep = 0; | |
47 | ||
48 | if ( flags & wxFDIO_INPUT ) | |
49 | { | |
50 | ep |= EPOLLIN; | |
51 | wxLogTrace(wxEpollDispatcher_Trace, | |
52 | _T("Registered fd %d for input events"), fd); | |
53 | } | |
54 | ||
55 | if ( flags & wxFDIO_OUTPUT ) | |
56 | { | |
57 | ep |= EPOLLOUT; | |
58 | wxLogTrace(wxEpollDispatcher_Trace, | |
59 | _T("Registered fd %d for output events"), fd); | |
60 | } | |
61 | ||
62 | if ( flags & wxFDIO_EXCEPTION ) | |
63 | { | |
64 | ep |= EPOLLERR | EPOLLHUP; | |
65 | wxLogTrace(wxEpollDispatcher_Trace, | |
66 | _T("Registered fd %d for exceptional events"), fd); | |
67 | } | |
68 | ||
69 | return ep; | |
70 | } | |
71 | ||
72 | // ---------------------------------------------------------------------------- | |
73 | // wxEpollDispatcher | |
74 | // ---------------------------------------------------------------------------- | |
75 | ||
5e1eac14 VZ |
76 | /* static */ |
77 | wxEpollDispatcher *wxEpollDispatcher::Create() | |
b46b1d59 | 78 | { |
5e1eac14 VZ |
79 | int epollDescriptor = epoll_create(1024); |
80 | if ( epollDescriptor == -1 ) | |
b46b1d59 VZ |
81 | { |
82 | wxLogSysError(_("Failed to create epoll descriptor")); | |
5e1eac14 | 83 | return NULL; |
b46b1d59 | 84 | } |
2804f77d VZ |
85 | wxLogTrace(wxEpollDispatcher_Trace, |
86 | _T("Epoll fd %d created"), epollDescriptor); | |
5e1eac14 VZ |
87 | return new wxEpollDispatcher(epollDescriptor); |
88 | } | |
89 | ||
90 | wxEpollDispatcher::wxEpollDispatcher(int epollDescriptor) | |
91 | { | |
92 | wxASSERT_MSG( epollDescriptor != -1, _T("invalid descriptor") ); | |
93 | ||
94 | m_epollDescriptor = epollDescriptor; | |
b46b1d59 VZ |
95 | } |
96 | ||
d31a4a84 VZ |
97 | wxEpollDispatcher::~wxEpollDispatcher() |
98 | { | |
99 | if ( close(m_epollDescriptor) != 0 ) | |
100 | { | |
101 | wxLogSysError(_("Error closing epoll descriptor")); | |
102 | } | |
103 | } | |
104 | ||
b46b1d59 VZ |
105 | bool wxEpollDispatcher::RegisterFD(int fd, wxFDIOHandler* handler, int flags) |
106 | { | |
b46b1d59 VZ |
107 | epoll_event ev; |
108 | ev.events = GetEpollMask(flags, fd); | |
109 | ev.data.ptr = handler; | |
110 | ||
111 | const int ret = epoll_ctl(m_epollDescriptor, EPOLL_CTL_ADD, fd, &ev); | |
112 | if ( ret != 0 ) | |
113 | { | |
114 | wxLogSysError(_("Failed to add descriptor %d to epoll descriptor %d"), | |
115 | fd, m_epollDescriptor); | |
116 | ||
117 | return false; | |
118 | } | |
2804f77d VZ |
119 | wxLogTrace(wxEpollDispatcher_Trace, |
120 | _T("Added fd %d (handler %p) to epoll %d"), fd, handler, m_epollDescriptor); | |
b46b1d59 VZ |
121 | |
122 | return true; | |
123 | } | |
124 | ||
125 | bool wxEpollDispatcher::ModifyFD(int fd, wxFDIOHandler* handler, int flags) | |
126 | { | |
b46b1d59 VZ |
127 | epoll_event ev; |
128 | ev.events = GetEpollMask(flags, fd); | |
129 | ev.data.ptr = handler; | |
130 | ||
131 | const int ret = epoll_ctl(m_epollDescriptor, EPOLL_CTL_MOD, fd, &ev); | |
132 | if ( ret != 0 ) | |
133 | { | |
134 | wxLogSysError(_("Failed to modify descriptor %d in epoll descriptor %d"), | |
135 | fd, m_epollDescriptor); | |
136 | ||
137 | return false; | |
138 | } | |
139 | ||
2804f77d VZ |
140 | wxLogTrace(wxEpollDispatcher_Trace, |
141 | _T("Modified fd %d (handler: %p) on epoll %d"), fd, handler, m_epollDescriptor); | |
b46b1d59 VZ |
142 | return true; |
143 | } | |
144 | ||
af57c51a | 145 | bool wxEpollDispatcher::UnregisterFD(int fd) |
b46b1d59 | 146 | { |
b46b1d59 VZ |
147 | epoll_event ev; |
148 | ev.events = 0; | |
149 | ev.data.ptr = NULL; | |
150 | ||
151 | if ( epoll_ctl(m_epollDescriptor, EPOLL_CTL_DEL, fd, &ev) != 0 ) | |
152 | { | |
153 | wxLogSysError(_("Failed to unregister descriptor %d from epoll descriptor %d"), | |
154 | fd, m_epollDescriptor); | |
155 | } | |
2804f77d VZ |
156 | wxLogTrace(wxEpollDispatcher_Trace, |
157 | _T("removed fd %d from %d"), fd, m_epollDescriptor); | |
ad8d42f8 | 158 | return true; |
b46b1d59 VZ |
159 | } |
160 | ||
5a557d1e | 161 | bool wxEpollDispatcher::Dispatch(int timeout) |
b46b1d59 VZ |
162 | { |
163 | epoll_event events[16]; | |
164 | ||
165 | const int e_num = epoll_wait | |
166 | ( | |
167 | m_epollDescriptor, | |
168 | events, | |
169 | WXSIZEOF(events), | |
170 | timeout == TIMEOUT_INFINITE ? -1 : timeout | |
171 | ); | |
172 | ||
173 | if ( e_num == -1 ) | |
174 | { | |
175 | if ( errno != EINTR ) | |
176 | { | |
177 | wxLogSysError(_("Waiting for IO on epoll descriptor %d failed"), | |
178 | m_epollDescriptor); | |
5a557d1e | 179 | return false; |
b46b1d59 VZ |
180 | } |
181 | } | |
182 | ||
5a557d1e | 183 | bool gotEvents = false; |
b46b1d59 VZ |
184 | for ( epoll_event *p = events; p < events + e_num; p++ ) |
185 | { | |
186 | wxFDIOHandler * const handler = (wxFDIOHandler *)(p->data.ptr); | |
187 | if ( !handler ) | |
188 | { | |
189 | wxFAIL_MSG( _T("NULL handler in epoll_event?") ); | |
190 | continue; | |
191 | } | |
192 | ||
1a781247 VZ |
193 | // note that for compatibility with wxSelectDispatcher we call |
194 | // OnReadWaiting() on EPOLLHUP as this is what epoll_wait() returns | |
195 | // when the write end of a pipe is closed while with select() the | |
196 | // remaining pipe end becomes ready for reading when this happens | |
197 | if ( p->events & (EPOLLIN | EPOLLHUP) ) | |
b46b1d59 | 198 | handler->OnReadWaiting(); |
7523de90 | 199 | else if ( p->events & EPOLLOUT ) |
b46b1d59 | 200 | handler->OnWriteWaiting(); |
1a781247 | 201 | else if ( p->events & EPOLLERR ) |
b46b1d59 | 202 | handler->OnExceptionWaiting(); |
5a557d1e VZ |
203 | else |
204 | continue; | |
205 | ||
206 | gotEvents = true; | |
b46b1d59 | 207 | } |
5a557d1e VZ |
208 | |
209 | return gotEvents; | |
b46b1d59 VZ |
210 | } |
211 | ||
a1873279 | 212 | #endif // wxUSE_EPOLL_DISPATCHER |