ViewVC Help
View File | Revision Log | Show Annotations | Download File
/cvs/rxvt-unicode/src/iom.C
(Generate patch)

Comparing rxvt-unicode/src/iom.C (file contents):
Revision 1.8 by pcg, Fri Jan 16 22:11:09 2004 UTC vs.
Revision 1.39 by root, Thu Oct 25 12:42:00 2007 UTC

1/* 1/*
2 iom.C -- generic I/O multiplexor 2 iom.C -- generic I/O multiplexer
3 Copyright (C) 2003 Marc Lehmann <pcg@goof.com> 3 Copyright (C) 2003-2006 Marc Lehmann <gvpe@schmorp.de>
4 4
5 This file is part of GVPE.
6
5 This program is free software; you can redistribute it and/or modify 7 GVPE is free software; you can redistribute it and/or modify
6 it under the terms of the GNU General Public License as published by 8 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 9 the Free Software Foundation; either version 2 of the License, or
8 (at your option) any later version. 10 (at your option) any later version.
9 11
10 This program is distributed in the hope that it will be useful, 12 This program is distributed in the hope that it will be useful,
11 but WITHOUT ANY WARRANTY; without even the implied warranty of 13 but WITHOUT ANY WARRANTY; without even the implied warranty of
12 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 14 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
13 GNU General Public License for more details. 15 GNU General Public License for more details.
14 16
15 You should have received a copy of the GNU General Public License 17 You should have received a copy of the GNU General Public License
16 along with this program; if not, write to the Free Software 18 along with gvpe; if not, write to the Free Software
17 Foundation, Inc. 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA 19 Foundation, Inc. 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
18*/ 20*/
19 21
20#include "../config.h" 22#include "iom.h"
21 23
22#include <cstdio> 24#include <cstdio>
23#include <cstdlib> 25#include <cstdlib>
24#include <cerrno> 26#include <cerrno>
27#include <cassert>
25 28
29#include <sys/types.h>
30#include <sys/time.h>
31
32#if 1 // older unices need these includes for select (2)
33# include <unistd.h>
34# include <time.h>
35#endif
36
37#if IOM_CHILD
38# include <sys/wait.h>
39#endif
40
41#if IOM_SIG
42# include <csignal>
43# include <fcntl.h>
44#endif
45
46// if the BSDs would at least be marginally POSIX-compatible.. *sigh*
47// until that happens, sys/select.h must come last
26#include <sys/select.h> 48#include <sys/select.h>
27#include <sys/time.h>
28 49
29#include "iom.h" 50#define TIMEVAL timeval
51#define TV_FRAC tv_usec
52#define TV_MULT 1000000L
30 53
31// TSTAMP_MAX must still fit into a positive struct timeval 54#if IOM_CHECK
32#define TSTAMP_MAX (double)(1UL<<31) 55static io_manager_vec<check_watcher> cw;
56#endif
57#if IOM_IDLE
58static io_manager_vec<idle_watcher> iw;
59#endif
60
61#if IOM_SIG
62static int sigpipe[2]; // signal signalling pipe
63static sigset_t sigs;
64struct sig_vec : io_manager_vec<sig_watcher> {
65 int pending;
66 sig_vec ()
67 : pending (false)
68 { }
69};
70static vector<sig_vec *> sw;
71#endif
72
73#if IOM_CHILD
74static io_manager_vec<child_watcher> pw;
75#endif
76
77#ifdef IOM_LIBEVENT
78static bool need_set_now; // need to set_now in callback
79#else
80 #if IOM_IO
81 static io_manager_vec<io_watcher> iow;
82 #endif
83 #if IOM_TIME
84 static io_manager_vec<time_watcher> tw;
85 #endif
86#endif
87
88#if IOM_TIME
89tstamp io_manager::now ()
90{
91 struct timeval tv;
92
93 gettimeofday (&tv, 0);
94 return (tstamp)tv.tv_sec + (tstamp)tv.tv_usec / 1000000.;
95}
96
97void io_manager::set_now ()
98{
99 NOW = now ();
100 #ifdef IOM_LIBEVENT
101 need_set_now = false;
102 #endif
103}
104#endif
105
106#ifndef IOM_LIBEVENT
107// this is a dummy time watcher to ensure that the first
108// time watcher is _always_ valid, this gets rid of a lot
109// of null-pointer-checks
110// (must come _before_ iom is being defined)
111static struct tw0 : time_watcher
112{
113 void cb (time_watcher &w)
114 {
115 // should never get called
116 // reached end-of-time, or tstamp has a bogus definition,
117 // or compiler initialisation order broken, or something else :)
118 abort ();
119 }
120
121 tw0 ()
122 : time_watcher (this, &tw0::cb)
123 { }
124} tw0;
125#endif
33 126
34tstamp NOW; 127tstamp NOW;
128
129#if IOM_CHILD
130// sig_watcher for child signal(s)
131static struct sw0 : sig_watcher
132{
133 void cb (sig_watcher &w)
134 {
135 // SIGCHLD, call corresponding watchera
136 pid_t pid;
137 int status;
138
139 while ((pid = waitpid (-1, &status, WNOHANG)) > 0)
140 for (int i = pw.size (); i--; )
141 {
142 child_watcher *w = pw[i];
143
144 if (!w)
145 pw.erase_unordered (i);
146 else if (w->pid == pid)
147 {
148 io_manager::unreg (*w);
149 w->call (*w, status);
150 }
151 }
152 }
153
154 sw0 ()
155 : sig_watcher (this, &sw0::cb)
156 { }
157} sw0;
158#endif
159
35static bool iom_valid; 160static bool iom_valid;
36io_manager iom; 161
162// used for initialisation only
163static struct init {
164 init ()
165 {
166 #ifdef IOM_PREINIT
167 { IOM_PREINIT }
168 #endif
169
170 #ifdef IOM_LIBEVENT
171 event_init ();
172 #endif
173 iom_valid = true;
174
175 #if IOM_SIG
176 sigemptyset (&sigs);
177
178 if (pipe (sigpipe))
179 {
180 perror ("io_manager: unable to create signal pipe, aborting.");
181 abort ();
182 }
183
184 fcntl (sigpipe[0], F_SETFL, O_NONBLOCK); fcntl (sigpipe[0], F_SETFD, FD_CLOEXEC);
185 fcntl (sigpipe[1], F_SETFL, O_NONBLOCK); fcntl (sigpipe[1], F_SETFD, FD_CLOEXEC);
186 #endif
187
188 #if IOM_CHILD
189 sw0.start (SIGCHLD);
190 #endif
191
192 #if IOM_TIME
193 io_manager::set_now ();
194
195 #ifndef IOM_LIBEVENT
196 tw0.start (TSTAMP_MAX);
197 #endif
198 #endif
199
200 #ifdef IOM_POSTINIT
201 { IOM_POSTINIT }
202 #endif
203 }
204
205 ~init ()
206 {
207 iom_valid = false;
208 }
209
210 static void required ();
211} init;
212
213void
214init::required ()
215{
216 if (!iom_valid)
217 {
218 write (2, "io_manager: early registration attempt, aborting.\n",
219 sizeof ("io_manager: early registration attempt, aborting.\n") - 1);
220 abort ();
221 }
222}
37 223
38template<class watcher> 224template<class watcher>
39void io_manager::reg (watcher *w, simplevec<watcher *> &queue) 225void io_manager::reg (watcher &w, io_manager_vec<watcher> &queue)
40{ 226{
41 if (!iom_valid) 227 init::required ();
42 abort ();
43 228
44 if (!w->active) 229 if (!w.active)
45 { 230 {
46 queue.push_back (w); 231 queue.push_back (&w);
47 w->active = queue.size (); 232 w.active = queue.size ();
48 } 233 }
49} 234}
50 235
51template<class watcher> 236template<class watcher>
52void io_manager::unreg (watcher *w, simplevec<watcher *> &queue) 237void io_manager::unreg (watcher &w, io_manager_vec<watcher> &queue)
53{ 238{
54 if (!iom_valid) 239 if (!iom_valid)
55 return; 240 return;
56 241
57 if (w->active) 242 if (w.active)
58 { 243 {
59 queue [w->active - 1] = 0; 244 queue [w.active - 1] = 0;
60 w->active = 0; 245 w.active = 0;
61 } 246 }
62} 247}
63 248
64#if IOM_TIME 249#if IOM_TIME
250 #ifdef IOM_LIBEVENT
251 void iom_time_c_callback (int fd, short events, void *data)
252 {
253 if (need_set_now) io_manager::set_now ();
254 time_watcher *w = static_cast<time_watcher *>(data);
255 w->call (*w);
256 }
257
258 void time_watcher::start ()
259 {
260 stop ();
261 evtimer_set (&ev, iom_time_c_callback, (void *)this);
262 struct timeval tv;
263 tv.tv_sec = (long)at;
264 tv.tv_usec = (long)((at - (tstamp)tv.tv_sec) * 1000000.);
265 evtimer_add (&ev, &tv);
266 active = 1;
267 }
268 #else
269 void io_manager::reg (time_watcher &w) { io_manager::reg (w, tw); }
270 void io_manager::unreg (time_watcher &w) { io_manager::unreg (w, tw); }
271 #endif
272
65void time_watcher::trigger () 273 void time_watcher::trigger ()
66{ 274 {
67 call (*this); 275 call (*this);
68 276 start ();
69 iom.reg (this); 277 }
70}
71
72void io_manager::reg (time_watcher *w) { reg (w, tw); }
73void io_manager::unreg (time_watcher *w) { unreg (w, tw); }
74#endif 278#endif
75 279
76#if IOM_IO 280#if IOM_IO
281 #ifdef IOM_LIBEVENT
282 void iom_io_c_callback (int fd, short events, void *data)
283 {
284 if (need_set_now) io_manager::set_now ();
285 io_watcher *w = static_cast<io_watcher *>(data);
286 w->call (*w, events);
287 }
288
289 void io_watcher::set (int fd_, short events_)
290 {
291 if (active) event_del (&ev);
292 fd = fd_;
293 events = events_;
294 event_set (&ev, fd_, events_ | EV_PERSIST, iom_io_c_callback, (void *)this);
295 if (active) event_add (&ev, 0);
296 }
297 #else
77void io_manager::reg (io_watcher *w) { reg (w, iow); } 298 void io_manager::reg (io_watcher &w) { io_manager::reg (w, iow); }
78void io_manager::unreg (io_watcher *w) { unreg (w, iow); } 299 void io_manager::unreg (io_watcher &w) { io_manager::unreg (w, iow); }
300 #endif
79#endif 301#endif
80 302
81#if IOM_CHECK 303#if IOM_CHECK
82void io_manager::reg (check_watcher *w) { reg (w, cw); } 304void io_manager::reg (check_watcher &w) { io_manager::reg (w, cw); }
83void io_manager::unreg (check_watcher *w) { unreg (w, cw); } 305void io_manager::unreg (check_watcher &w) { io_manager::unreg (w, cw); }
84#endif 306#endif
85 307
86#if IOM_IDLE 308#if IOM_IDLE
87void io_manager::reg (idle_watcher *w) { reg (w, iw); } 309void io_manager::reg (idle_watcher &w) { io_manager::reg (w, iw); }
88void io_manager::unreg (idle_watcher *w) { unreg (w, iw); } 310void io_manager::unreg (idle_watcher &w) { io_manager::unreg (w, iw); }
89#endif 311#endif
90 312
91#if IOM_TIME 313#if IOM_SIG
92inline void set_now (void) 314static void
315sighandler (int signum)
93{ 316{
94 struct timeval tv; 317 sw [signum - 1]->pending = true;
95 318
96 gettimeofday (&tv, 0); 319 // we use a pipe for signal notifications, as most current
97 320 // OSes (Linux...) do not implement pselect correctly. ugh.
98 NOW = (tstamp)tv.tv_sec + (tstamp)tv.tv_usec / 1000000; 321 char ch = signum; // actual content not used
99#endif 322 write (sigpipe[1], &ch, 1);
100} 323}
324
325void io_manager::reg (sig_watcher &w)
326{
327 init::required ();
328
329 assert (0 < w.signum);
330
331 sw.reserve (w.signum);
332
333 while (sw.size () < w.signum) // pathetic
334 sw.push_back (0);
335
336 sig_vec *&sv = sw[w.signum - 1];
337
338 if (!sv)
339 {
340 sv = new sig_vec;
341
342 sigaddset (&sigs, w.signum);
343 sigprocmask (SIG_BLOCK, &sigs, NULL);
344
345 struct sigaction sa;
346 sa.sa_handler = sighandler;
347 sigfillset (&sa.sa_mask);
348 sa.sa_flags = SA_RESTART;
349
350 if (sigaction (w.signum, &sa, 0))
351 {
352 perror ("io_manager: error while installing signal handler, ignoring.");
353 abort ();
354 }
355
356 }
357
358 io_manager::reg (w, *sv);
359}
360
361void io_manager::unreg (sig_watcher &w)
362{
363 if (!w.active || !iom_valid)
364 return;
365
366 assert (0 < w.signum && w.signum <= sw.size ());
367
368 io_manager::unreg (w, *sw[w.signum - 1]);
369}
370
371void sig_watcher::start (int signum)
372{
373 stop ();
374 this->signum = signum;
375 io_manager::reg (*this);
376}
377#endif
378
379#if IOM_CHILD
380void io_manager::reg (child_watcher &w) { io_manager::reg (w, pw); }
381void io_manager::unreg (child_watcher &w) { io_manager::unreg (w, pw); }
382#endif
101 383
102void io_manager::loop () 384void io_manager::loop ()
103{ 385{
386 init::required ();
387
104#if IOM_TIME 388 #if IOM_TIME
105 set_now (); 389 set_now ();
106#endif 390 #endif
107 391
108 for (;;) 392 for (;;)
109 { 393 {
394 #ifndef IOM_LIBEVENT
395 #if IOM_TIME
396 // call pending time watchers
397 {
398 bool activity;
399
400 do
401 {
402 activity = false;
403
404 for (int i = tw.size (); i--; )
405 if (!tw[i])
406 tw.erase_unordered (i);
407 else if (tw[i]->at <= NOW)
408 {
409 time_watcher &w = *tw[i];
410
411 unreg (w);
412 w.call (w);
413
414 activity = true;
415 }
416 }
417 while (activity);
418 }
419 #endif
420 #endif
421
422 #if IOM_CHECK
423 // call all check watchers
424 for (int i = cw.size (); i--; )
425 if (!cw[i])
426 cw.erase_unordered (i);
427 else
428 cw[i]->call (*cw[i]);
429 #endif
430
110 struct timeval *to = 0; 431 struct TIMEVAL *to = 0;
111 struct timeval tval; 432 struct TIMEVAL tval;
112 433
113#if IOM_IDLE 434#if IOM_IDLE
114 if (iw.size ()) 435 if (iw.size ())
115 { 436 {
116 tval.tv_sec = 0; 437 tval.tv_sec = 0;
117 tval.tv_usec = 0; 438 tval.TV_FRAC = 0;
118 to = &tval; 439 to = &tval;
119 } 440 }
120 else 441 else
121#endif 442#endif
122 { 443 {
123#if IOM_TIME 444 #ifndef IOM_LIBEVENT
124 time_watcher *next; 445 #if IOM_TIME
446 // find earliest active watcher
447 time_watcher *next = tw[0]; // the first time-watcher must exist at ALL times
125 448
126 for (;;) 449 for (io_manager_vec<time_watcher>::const_iterator i = tw.end (); i-- > tw.begin (); )
450 if (*i && (*i)->at < next->at)
451 next = *i;
452
453 if (next->at > NOW && next != tw[0])
127 { 454 {
128 next = tw[0]; // the first time-watcher must exist at ALL times 455 double diff = next->at - NOW;
456 tval.tv_sec = (int)diff;
457 tval.TV_FRAC = (int) ((diff - tval.tv_sec) * TV_MULT);
458 to = &tval;
459 }
460 #endif
461 #endif
462 }
129 463
130 for (int i = tw.size (); i--; ) 464 #ifndef IOM_LIBEVENT
465 #if IOM_IO || IOM_SIG
466 fd_set rfd, wfd;
467
468 FD_ZERO (&rfd);
469 FD_ZERO (&wfd);
470
471 int fds = 0;
472
473 #if IOM_IO
474 for (io_manager_vec<io_watcher>::const_iterator i = iow.end (); i-- > iow.begin (); )
131 if (!tw[i]) 475 if (*i)
132 tw.erase_unordered (i);
133 else if (tw[i]->at < next->at)
134 next = tw[i];
135
136 if (next->at > NOW)
137 { 476 {
138 if (next != tw[0]) 477 if ((*i)->events & EVENT_READ ) FD_SET ((*i)->fd, &rfd);
139 { 478 if ((*i)->events & EVENT_WRITE) FD_SET ((*i)->fd, &wfd);
140 double diff = next->at - NOW; 479
141 tval.tv_sec = (int)diff; 480 if ((*i)->fd >= fds) fds = (*i)->fd + 1;
142 tval.tv_usec = (int)((diff - tval.tv_sec) * 1000000);
143 to = &tval;
144 }
145 break;
146 } 481 }
147 else if (next->at >= 0) 482 #endif
483
484 if (!to && !fds) //TODO: also check idle_watchers and check_watchers?
485 break; // no events
486
487 #if IOM_SIG
488 FD_SET (sigpipe[0], &rfd);
489 if (sigpipe[0] >= fds) fds = sigpipe[0] + 1;
490 #endif
491
492 #if IOM_SIG
493 // there is no race, as we use a pipe for signals, so select
494 // will return if a signal is caught.
495 sigprocmask (SIG_UNBLOCK, &sigs, NULL);
496 #endif
497 fds = select (fds, &rfd, &wfd, NULL, to);
498 #if IOM_SIG
499 sigprocmask (SIG_BLOCK, &sigs, NULL);
500 #endif
501 #elif IOM_TIME
502 if (!to)
503 break;
504
505 select (0, 0, 0, 0, to);
506 #endif
507
508 #if IOM_TIME
509 {
510 // update time, try to compensate for gross non-monotonic time changes
511 tstamp diff = NOW;
512 set_now ();
513 diff = NOW - diff;
514
515 if (diff < 0)
516 for (io_manager_vec<time_watcher>::const_iterator i = tw.end (); i-- > tw.begin (); )
517 if (*i)
518 (*i)->at += diff;
519 }
520 #endif
521
522 if (fds > 0)
523 {
524 #if IOM_SIG
525 if (FD_ISSET (sigpipe[0], &rfd))
148 { 526 {
149 unreg (next); 527 char ch;
150 next->call (*next); 528
529 while (read (sigpipe[0], &ch, 1) > 0)
530 ;
531
532 for (vector<sig_vec *>::iterator svp = sw.end (); svp-- > sw.begin (); )
533 if (*svp && (*svp)->pending)
534 {
535 sig_vec &sv = **svp;
536 for (int i = sv.size (); i--; )
537 if (!sv[i])
538 sv.erase_unordered (i);
539 else
540 sv[i]->call (*sv[i]);
541
542 sv.pending = false;
543 }
151 } 544 }
545 #endif
546
547 #if IOM_IO
548 for (int i = iow.size (); i--; )
549 if (!iow[i])
550 iow.erase_unordered (i);
551 else
552 {
553 io_watcher &w = *iow[i];
554 short revents = w.events;
555
556 if (!FD_ISSET (w.fd, &rfd)) revents &= ~EVENT_READ;
557 if (!FD_ISSET (w.fd, &wfd)) revents &= ~EVENT_WRITE;
558
559 if (revents)
560 w.call (w, revents);
561 }
562 #endif
152 } 563 }
153#endif 564 else if (fds < 0 && errno != EINTR)
565 {
566 perror ("io_manager: fatal error while waiting for I/O or time event, aborting.");
567 abort ();
154 } 568 }
155 569#if IOM_IDLE
156#if IOM_CHECK
157 for (int i = cw.size (); i--; )
158 if (!cw[i])
159 cw.erase_unordered (i);
160 else 570 else
161 cw[i]->call (*cw[i]);
162#endif
163
164#if IOM_IO
165 fd_set rfd, wfd, efd;
166
167 FD_ZERO (&rfd);
168 FD_ZERO (&wfd);
169
170 int fds = 0;
171
172 for (io_watcher **i = iow.end (); i-- > iow.begin (); )
173 if (*i)
174 {
175 if ((*i)->events & EVENT_READ ) FD_SET ((*i)->fd, &rfd);
176 if ((*i)->events & EVENT_WRITE) FD_SET ((*i)->fd, &wfd);
177
178 if ((*i)->fd >= fds) fds = (*i)->fd + 1;
179 }
180
181 if (!to && !fds) //TODO: also check idle_watchers and check_watchers
182 break; // no events
183
184 fds = select (fds, &rfd, &wfd, &efd, to);
185# if IOM_TIME
186 set_now ();
187# endif
188
189 if (fds > 0)
190 for (int i = iow.size (); i--; )
191 if (!iow[i])
192 iow.erase_unordered (i);
193 else
194 {
195 short revents = iow[i]->events;
196
197 if (!FD_ISSET (iow[i]->fd, &rfd)) revents &= ~EVENT_READ;
198 if (!FD_ISSET (iow[i]->fd, &wfd)) revents &= ~EVENT_WRITE;
199
200 if (revents)
201 iow[i]->call (*iow[i], revents);
202 }
203 else if (fds < 0 && errno != EINTR)
204 {
205 perror ("Error while waiting for I/O or time event");
206 abort ();
207 }
208#if IOM_IDLE
209 else
210 for (int i = iw.size (); i--; ) 571 for (int i = iw.size (); i--; )
211 if (!iw[i]) 572 if (!iw[i])
212 iw.erase_unordered (i); 573 iw.erase_unordered (i);
213 else 574 else
214 iw[i]->call (*iw[i]); 575 iw[i]->call (*iw[i]);
215#endif 576#endif
216 577
217#elif IOM_TIME 578 #else
579 need_set_now = true;
580
218 if (!to) 581 if (to)
219 break; 582 event_loop (EVLOOP_NONBLOCK);
583 else
584 event_loop (EVLOOP_ONCE);
220 585
221 select (0, 0, 0, 0, &to); 586 if (need_set_now) set_now ();
222 set_now (); 587 #endif
223#else 588 //TODO: IOM_IDLE
224 break;
225#endif
226 }
227}
228
229// this is a dummy time watcher to ensure that the first
230// time watcher is _always_ valid, this gets rid of a lot
231// of null-pointer-checks
232static struct tw0 : time_watcher {
233 void cb (time_watcher &w)
234 {
235 // should never get called
236 // reached end-of-time, or tstamp has a bogus definition :)
237 abort ();
238 }
239
240 tw0()
241 : time_watcher (this, &tw0::cb)
242 { } 589 }
243} tw0;
244
245io_manager::io_manager ()
246{
247 iom_valid = true;
248
249#if IOM_TIME
250 set_now ();
251
252 tw0.start (TSTAMP_MAX);
253#endif
254} 590}
255 591
256io_manager::~io_manager ()
257{
258 iom_valid = false;
259}
260

Diff Legend

– Removed lines
+ Added lines
< Changed lines
> Changed lines