ViewVC Help
View File | Revision Log | Show Annotations | Download File
/cvs/rxvt-unicode/src/iom.C
Revision: 1.19
Committed: Thu Sep 2 07:44:40 2004 UTC (19 years, 8 months ago) by root
Content type: text/plain
Branch: MAIN
CVS Tags: rel-4_0
Changes since 1.17: +201 -100 lines
Log Message:
*** empty log message ***

File Contents

# User Rev Content
1 pcg 1.1 /*
2     iom.C -- generic I/O multiplexor
3 pcg 1.9 Copyright (C) 2003, 2004 Marc Lehmann <pcg@goof.com>
4 pcg 1.1
5     This program is free software; you can redistribute it and/or modify
6     it under the terms of the GNU General Public License as published by
7     the Free Software Foundation; either version 2 of the License, or
8     (at your option) any later version.
9    
10     This program is distributed in the hope that it will be useful,
11     but WITHOUT ANY WARRANTY; without even the implied warranty of
12     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
13     GNU General Public License for more details.
14    
15     You should have received a copy of the GNU General Public License
16     along with this program; if not, write to the Free Software
17     Foundation, Inc. 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
18     */
19    
20 root 1.19 #include "iom.h"
21    
22 pcg 1.1 #include <cstdio>
23 pcg 1.8 #include <cstdlib>
24     #include <cerrno>
25 pcg 1.1
26 pcg 1.11 #include <sys/time.h>
27    
28 pcg 1.16 #include <assert.h>
29    
30 pcg 1.14 #if 1 // older unices need these includes for select (2)
31 pcg 1.11 # include <unistd.h>
32     # include <sys/types.h>
33 root 1.19 # include <time.h>
34     #endif
35    
36     // for IOM_SIG
37     #if IOM_SIG
38     # include <signal.h>
39     # include <fcntl.h>
40 pcg 1.11 #endif
41    
42     // if the BSDs would at least be marginally POSIX-compatible.. *sigh*
43     // until that happens, sys/select.h must come last
44 pcg 1.1 #include <sys/select.h>
45    
46 root 1.19 // TSTAMP_MAX must still fit into a positive struct timeval
47     #define TSTAMP_MAX (double)(1UL<<31)
48 pcg 1.16
49 root 1.19 #define TIMEVAL timeval
50     #define TV_FRAC tv_usec
51     #define TV_MULT 1000000L
52 pcg 1.1
53 root 1.19 #if IOM_IO
54     static io_manager_vec<io_watcher> iow;
55     #endif
56     #if IOM_CHECK
57     static io_manager_vec<check_watcher> cw;
58     #endif
59     #if IOM_TIME
60     static io_manager_vec<time_watcher> tw;
61     #endif
62     #if IOM_IDLE
63     static io_manager_vec<idle_watcher> iw;
64     #endif
65     #if IOM_SIG
66     static int sigpipe[2]; // signal signalling pipe
67     static sigset_t sigs;
68     struct sig_vec : io_manager_vec<sig_watcher> {
69     int pending;
70     sig_vec ()
71     : pending (false)
72     { }
73     };
74     static vector<sig_vec *> sw;
75     #endif
76 pcg 1.8
77 pcg 1.10 // this is a dummy time watcher to ensure that the first
78     // time watcher is _always_ valid, this gets rid of a lot
79     // of null-pointer-checks
80     // (must come _before_ iom is being defined)
81 pcg 1.13 static struct tw0 : time_watcher
82 pcg 1.10 {
83 pcg 1.13 void cb (time_watcher &w)
84     {
85     // should never get called
86     // reached end-of-time, or tstamp has a bogus definition,
87 root 1.17 // or compiler initialisation order broken, or something else :)
88 pcg 1.13 abort ();
89     }
90 pcg 1.10
91 pcg 1.14 tw0 ()
92 root 1.19 : time_watcher (this, &tw0::cb)
93     { }
94 pcg 1.16 } tw0;
95 pcg 1.10
96 pcg 1.1 tstamp NOW;
97 root 1.19
98     #if IOM_TIME
99     inline void set_now (void)
100     {
101     struct timeval tv;
102    
103     gettimeofday (&tv, 0);
104     NOW = (tstamp)tv.tv_sec + (tstamp)tv.tv_usec / 1000000.;
105     }
106     #endif
107    
108 pcg 1.8 static bool iom_valid;
109 root 1.19
110     // used for initialisation only
111     static struct init {
112     init ()
113     {
114     #if IOM_SIG
115     sigemptyset (&sigs);
116    
117     if (pipe (sigpipe))
118     {
119     perror ("io_manager: unable to create signal pipe, aborting.");
120     abort ();
121     }
122    
123     fcntl (sigpipe[0], F_SETFL, O_NONBLOCK);
124     fcntl (sigpipe[1], F_SETFL, O_NONBLOCK);
125     #endif
126    
127     iom_valid = true;
128    
129     #if IOM_TIME
130     set_now ();
131    
132     tw0.start (TSTAMP_MAX);
133     #endif
134     }
135    
136     static void required ();
137     } init;
138    
139     void
140     init::required ()
141     {
142     if (!iom_valid)
143     {
144     write (2, "io_manager: early registration attempt, aborting.\n",
145     sizeof ("io_manager: early registration attempt, aborting.\n") - 1);
146     abort ();
147     }
148     }
149 pcg 1.1
150 pcg 1.4 template<class watcher>
151 root 1.19 void io_manager::reg (watcher &w, io_manager_vec<watcher> &queue)
152 pcg 1.4 {
153 root 1.19 init::required ();
154 pcg 1.8
155 root 1.19 if (!w.active)
156 pcg 1.8 {
157 pcg 1.12 #if IOM_CHECK
158     queue.activity = true;
159     #endif
160 root 1.19 queue.push_back (&w);
161     w.active = queue.size ();
162 pcg 1.8 }
163 pcg 1.4 }
164    
165     template<class watcher>
166 root 1.19 void io_manager::unreg (watcher &w, io_manager_vec<watcher> &queue)
167 pcg 1.4 {
168 pcg 1.8 if (!iom_valid)
169     return;
170 pcg 1.4
171 root 1.19 if (w.active)
172 pcg 1.8 {
173 root 1.19 queue [w.active - 1] = 0;
174     w.active = 0;
175 pcg 1.8 }
176 pcg 1.4 }
177    
178     #if IOM_TIME
179 pcg 1.1 void time_watcher::trigger ()
180     {
181     call (*this);
182 root 1.19 io_manager::reg (*this);
183 pcg 1.1 }
184    
185 root 1.19 void io_manager::reg (time_watcher &w) { io_manager::reg (w, tw); }
186     void io_manager::unreg (time_watcher &w) { io_manager::unreg (w, tw); }
187 pcg 1.8 #endif
188 pcg 1.3
189 pcg 1.8 #if IOM_IO
190 root 1.19 void io_manager::reg (io_watcher &w) { io_manager::reg (w, iow); }
191     void io_manager::unreg (io_watcher &w) { io_manager::unreg (w, iow); }
192 pcg 1.4 #endif
193 pcg 1.1
194 pcg 1.4 #if IOM_CHECK
195 root 1.19 void io_manager::reg (check_watcher &w) { io_manager::reg (w, cw); }
196     void io_manager::unreg (check_watcher &w) { io_manager::unreg (w, cw); }
197 pcg 1.4 #endif
198 pcg 1.1
199 pcg 1.7 #if IOM_IDLE
200 root 1.19 void io_manager::reg (idle_watcher &w) { io_manager::reg (w, iw); }
201     void io_manager::unreg (idle_watcher &w) { io_manager::unreg (w, iw); }
202 pcg 1.7 #endif
203    
204 root 1.19 #if IOM_SIG
205     static void
206     sighandler (int signum)
207 pcg 1.1 {
208 root 1.19 sw [signum - 1]->pending = true;
209 pcg 1.1
210 root 1.19 // we use a pipe for signal notifications, as most current
211     // OSes (Linux...) do not implement pselect correctly. ugh.
212     char ch = signum; // actual content not used
213     write (sigpipe[1], &ch, 1);
214 pcg 1.16 }
215    
216 root 1.19 void io_manager::reg (sig_watcher &w)
217 pcg 1.16 {
218 root 1.19 assert (0 < w.signum);
219 pcg 1.16
220 root 1.19 sw.reserve (w.signum);
221 pcg 1.16
222 root 1.19 while (sw.size () < w.signum) // pathetic
223     sw.push_back (0);
224 pcg 1.16
225 root 1.19 sig_vec *&sv = sw[w.signum - 1];
226 pcg 1.16
227     if (!sv)
228     {
229     sv = new sig_vec;
230    
231 root 1.19 sigaddset (&sigs, w.signum);
232     sigprocmask (SIG_BLOCK, &sigs, NULL);
233    
234 pcg 1.16 struct sigaction sa;
235 root 1.19 sa.sa_handler = sighandler;
236 pcg 1.16 sigfillset (&sa.sa_mask);
237 root 1.19 sa.sa_flags = SA_RESTART;
238 pcg 1.16
239 root 1.19 if (sigaction (w.signum, &sa, 0))
240 pcg 1.16 {
241 root 1.19 perror ("io_manager: error while installing signal handler, ignoring.");
242 pcg 1.16 abort ();
243     }
244 root 1.19
245 pcg 1.16 }
246    
247 root 1.19 io_manager::reg (w, *sv);
248 pcg 1.16 }
249    
250 root 1.19 void io_manager::unreg (sig_watcher &w)
251 pcg 1.16 {
252 root 1.19 if (!w.active)
253     return;
254    
255     assert (0 < w.signum && w.signum <= sw.size ());
256 pcg 1.16
257 root 1.19 io_manager::unreg (w, *sw[w.signum - 1]);
258 pcg 1.16 }
259    
260     void sig_watcher::start (int signum)
261     {
262     stop ();
263     this->signum = signum;
264 root 1.19 io_manager::reg (*this);
265 pcg 1.16 }
266     #endif
267 root 1.19
268 pcg 1.1 void io_manager::loop ()
269     {
270 root 1.19 init::required ();
271    
272 pcg 1.4 #if IOM_TIME
273 pcg 1.1 set_now ();
274 pcg 1.4 #endif
275 pcg 1.1
276     for (;;)
277     {
278 root 1.19 struct TIMEVAL *to = 0;
279     struct TIMEVAL tval;
280 pcg 1.1
281 pcg 1.7 #if IOM_IDLE
282     if (iw.size ())
283     {
284     tval.tv_sec = 0;
285 root 1.19 tval.TV_FRAC = 0;
286 pcg 1.7 to = &tval;
287     }
288     else
289     #endif
290 pcg 1.1 {
291 pcg 1.7 #if IOM_TIME
292 pcg 1.8 time_watcher *next;
293 pcg 1.1
294 pcg 1.8 for (;;)
295 pcg 1.7 {
296 pcg 1.8 next = tw[0]; // the first time-watcher must exist at ALL times
297 pcg 1.1
298 pcg 1.8 for (int i = tw.size (); i--; )
299     if (!tw[i])
300     tw.erase_unordered (i);
301     else if (tw[i]->at < next->at)
302     next = tw[i];
303 pcg 1.7
304 pcg 1.8 if (next->at > NOW)
305 pcg 1.7 {
306 pcg 1.8 if (next != tw[0])
307     {
308     double diff = next->at - NOW;
309     tval.tv_sec = (int)diff;
310 root 1.19 tval.TV_FRAC = (int) ((diff - tval.tv_sec) * TV_MULT);
311 pcg 1.8 to = &tval;
312     }
313 pcg 1.7 break;
314     }
315 pcg 1.9 else
316 pcg 1.8 {
317 root 1.19 unreg (*next);
318 pcg 1.8 next->call (*next);
319     }
320 pcg 1.5 }
321 pcg 1.7 #endif
322 pcg 1.1 }
323 pcg 1.6
324     #if IOM_CHECK
325 pcg 1.12 tw.activity = false;
326    
327 pcg 1.8 for (int i = cw.size (); i--; )
328     if (!cw[i])
329     cw.erase_unordered (i);
330     else
331     cw[i]->call (*cw[i]);
332 pcg 1.12
333     if (tw.activity)
334     {
335     tval.tv_sec = 0;
336 root 1.19 tval.TV_FRAC = 0;
337 pcg 1.12 to = &tval;
338     }
339 pcg 1.4 #endif
340 pcg 1.1
341 root 1.19 #if IOM_IO || IOM_SIG
342 pcg 1.15 fd_set rfd, wfd;
343 pcg 1.1
344     FD_ZERO (&rfd);
345     FD_ZERO (&wfd);
346    
347     int fds = 0;
348    
349 root 1.19 # if IOM_IO
350 pcg 1.9 for (io_manager_vec<io_watcher>::iterator i = iow.end (); i-- > iow.begin (); )
351 pcg 1.8 if (*i)
352     {
353     if ((*i)->events & EVENT_READ ) FD_SET ((*i)->fd, &rfd);
354     if ((*i)->events & EVENT_WRITE) FD_SET ((*i)->fd, &wfd);
355 pcg 1.1
356 pcg 1.8 if ((*i)->fd >= fds) fds = (*i)->fd + 1;
357     }
358 root 1.19 # endif
359 pcg 1.1
360 pcg 1.8 if (!to && !fds) //TODO: also check idle_watchers and check_watchers
361 pcg 1.5 break; // no events
362    
363 root 1.19 # if IOM_SIG
364     FD_SET (sigpipe[0], &rfd);
365     if (sigpipe[0] >= fds) fds = sigpipe[0] + 1;
366     # endif
367    
368     # if IOM_SIG
369     // there is no race, as we use a pipe for signals, so select
370     // will return if a signal is caught.
371     sigprocmask (SIG_UNBLOCK, &sigs, NULL);
372     # endif
373 pcg 1.15 fds = select (fds, &rfd, &wfd, NULL, to);
374 root 1.19 # if IOM_SIG
375     sigprocmask (SIG_BLOCK, &sigs, NULL);
376     # endif
377    
378 pcg 1.4 # if IOM_TIME
379 pcg 1.1 set_now ();
380 pcg 1.4 # endif
381 pcg 1.1
382     if (fds > 0)
383 root 1.19 {
384     # if IOM_SIG
385     if (FD_ISSET (sigpipe[0], &rfd))
386 pcg 1.8 {
387 root 1.19 char ch;
388    
389     while (read (sigpipe[0], &ch, 1) > 0)
390     ;
391 pcg 1.1
392 root 1.19 for (sig_vec **svp = sw.end (); svp-- > sw.begin (); )
393     if (*svp && (*svp)->pending)
394     {
395     sig_vec &sv = **svp;
396     for (int i = sv.size (); i--; )
397     if (!sv[i])
398     sv.erase_unordered (i);
399     else
400     sv[i]->call (*sv[i]);
401 pcg 1.4
402 root 1.19 sv.pending = false;
403     }
404 pcg 1.8 }
405 root 1.19 # endif
406    
407     # if IOM_IO
408     for (int i = iow.size (); i--; )
409     if (!iow[i])
410     iow.erase_unordered (i);
411     else
412     {
413     short revents = iow[i]->events;
414    
415     if (!FD_ISSET (iow[i]->fd, &rfd)) revents &= ~EVENT_READ;
416     if (!FD_ISSET (iow[i]->fd, &wfd)) revents &= ~EVENT_WRITE;
417    
418     if (revents)
419     iow[i]->call (*iow[i], revents);
420     }
421     #endif
422     }
423 pcg 1.8 else if (fds < 0 && errno != EINTR)
424     {
425 root 1.19 perror ("io_manager: fatal error while waiting for I/O or time event, aborting.");
426 pcg 1.8 abort ();
427     }
428 pcg 1.7 #if IOM_IDLE
429 pcg 1.8 else
430     for (int i = iw.size (); i--; )
431     if (!iw[i])
432     iw.erase_unordered (i);
433     else
434     iw[i]->call (*iw[i]);
435 pcg 1.7 #endif
436    
437 pcg 1.4 #elif IOM_TIME
438 pcg 1.5 if (!to)
439     break;
440    
441 pcg 1.4 select (0, 0, 0, 0, &to);
442     set_now ();
443 pcg 1.5 #else
444     break;
445 pcg 1.4 #endif
446 pcg 1.5 }
447 pcg 1.1 }
448