ViewVC Help
View File | Revision Log | Show Annotations | Download File
/cvs/gvpe/src/iom.h
(Generate patch)

Comparing gvpe/src/iom.h (file contents):
Revision 1.10 by pcg, Wed Apr 2 21:02:25 2003 UTC vs.
Revision 1.33 by pcg, Wed May 31 00:39:31 2006 UTC

1/* 1/*
2 iom.h -- I/O multiplexor 2 iom.h -- generic I/O multiplexer
3 Copyright (C) 2003-2006 Marc Lehmann <gvpe@schmorp.de>
3 4
5 This file is part of GVPE.
6
4 This program is free software; you can redistribute it and/or modify 7 GVPE is free software; you can redistribute it and/or modify
5 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
6 the Free Software Foundation; either version 2 of the License, or 9 the Free Software Foundation; either version 2 of the License, or
7 (at your option) any later version. 10 (at your option) any later version.
8 11
9 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,
10 but WITHOUT ANY WARRANTY; without even the implied warranty of 13 but WITHOUT ANY WARRANTY; without even the implied warranty of
11 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 14 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 GNU General Public License for more details. 15 GNU General Public License for more details.
13 16
14 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
15 along with this program; if not, write to the Free Software 18 along with gvpe; if not, write to the Free Software
16 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
17*/ 20*/
18 21
19#ifndef VPE_IOM_H__ 22#ifndef IOM_H__
20#define VPE_IOM_H__ 23#define IOM_H__
21 24
22#include <vector> 25// required:
26// - a vector template like simplevec or stl's vector
27// - defines for all watcher types required in your app
28// edit iom_conf.h as appropriate.
29#include "iom_conf.h"
23 30
24#include <sys/poll.h> 31#if IOM_CHILD
32# undef IOM_SIG
33# define IOM_SIG 1
34#endif
25 35
26#include "callback.h" 36#include "callback.h"
27#include "slog.h"
28 37
29typedef double tstamp; 38typedef double tstamp;
30
31extern tstamp NOW; 39extern tstamp NOW;
32 40
41// TSTAMP_MAX must still fit into a positive struct timeval
42#define TSTAMP_MAX (double)(1UL<<31)
43
44struct watcher;
45#if IOM_IO
33struct io_watcher; 46struct io_watcher;
47#endif
48#if IOM_TIME
34struct time_watcher; 49struct time_watcher;
50#endif
51#if IOM_CHECK
52struct check_watcher;
53#endif
54#if IOM_IDLE
55struct idle_watcher;
56#endif
57#if IOM_SIG
58struct sig_watcher;
59#endif
60#if IOM_CHILD
61struct child_watcher;
62#endif
35 63
64template<class watcher>
65struct io_manager_vec : vector<watcher *> {
66 void erase_unordered (unsigned int pos)
67 {
68 watcher *w = (*this)[this->size () - 1];
69 this->pop_back ();
70
71 if (!this->empty ())
72 if (((*this)[pos] = w)) // '=' is correct!
73 w->active = pos + 1;
74 }
75};
76
77// only used as a namespace, and for initialisation purposes
36class io_manager { 78class io_manager {
37 vector<pollfd> pfs; 79 template<class watcher>
38 vector<io_watcher *> iow; 80 static void reg (watcher &w, io_manager_vec<watcher> &queue);
39 vector<time_watcher *> tw; // actually a heap
40 81
41 void idle_cb (time_watcher &w); time_watcher *idle; 82 template<class watcher>
83 static void unreg (watcher &w, io_manager_vec<watcher> &queue);
84
42public: 85public:
86#if IOM_TIME
87 // fetch time only
88 static tstamp now ();
43 89
44 void reschedule_time_watchers (); 90 // set NOW
91 static void set_now ();
92#endif
45 93
46 // register a watcher 94 // register a watcher
47 void reg (io_watcher *w); 95#if IOM_IO
48 void unreg (io_watcher *w); 96 static void reg (io_watcher &w); static void unreg (io_watcher &w);
49 void reg (time_watcher *w); 97#endif
50 void unreg (time_watcher *w); 98#if IOM_TIME
99 static void reg (time_watcher &w); static void unreg (time_watcher &w);
100#endif
101#if IOM_CHECK
102 static void reg (check_watcher &w); static void unreg (check_watcher &w);
103#endif
104#if IOM_IDLE
105 static void reg (idle_watcher &w); static void unreg (idle_watcher &w);
106#endif
107#if IOM_SIG
108 static void reg (sig_watcher &w); static void unreg (sig_watcher &w);
109#endif
110#if IOM_CHILD
111 static void reg (child_watcher &w); static void unreg (child_watcher &w);
112#endif
51 113
52 void loop (); 114 static void loop ();
53
54 io_manager ();
55 ~io_manager ();
56}; 115};
57 116
58extern io_manager iom; 117struct watcher {
118 int active; /* 0 == inactive, else index into respective vector */
59 119
120 bool is_active () { return active; }
121
122 watcher () : active (0) { }
123};
124
125#if IOM_IO
126enum { EVENT_UNDEF = -1, EVENT_NONE = 0, EVENT_READ = 1, EVENT_WRITE = 2 };
127
60struct io_watcher : callback2<void, io_watcher &, short> { 128struct io_watcher : watcher, callback<void (io_watcher &, short)> {
61 int fd; 129 int fd;
62 short events; 130 short events;
63 131
132 void set (int fd_, short events_) { fd = fd_; events = events_; }
133
134 void set (short events_) { set (fd, events_); }
135 void start () { io_manager::reg (*this); }
136 void start (int fd_, short events_) { set (fd_, events_); io_manager::reg (*this); }
137 void stop () { io_manager::unreg (*this); }
138
64 template<class O1, class O2> 139 template<class O, class M>
65 io_watcher (O1 *object, void (O2::*method)(io_watcher &, short)) 140 io_watcher (O object, M method)
66 : callback2<void, io_watcher &, short>(object,method) 141 : callback<void (io_watcher &, short)> (object, method)
142 { }
143 ~io_watcher () { stop (); }
144};
145#endif
146
147#if IOM_TIME
148struct time_watcher : watcher, callback<void (time_watcher &)> {
149 tstamp at;
150
151 void trigger ();
152
153 void set (tstamp when) { at = when; }
154 void operator () () { trigger (); }
155 void start () { io_manager::reg (*this); }
156 void start (tstamp when) { set (when); io_manager::reg (*this); }
157 void stop () { io_manager::unreg (*this); }
158
159 template<class O, class M>
160 time_watcher (O object, M method)
161 : callback<void (time_watcher &)> (object, method), at (0)
162 { }
163 ~time_watcher () { stop (); }
164};
165#endif
166
167#if IOM_CHECK
168// run before checking for new events
169struct check_watcher : watcher, callback<void (check_watcher &)> {
170 void start () { io_manager::reg (*this); }
171 void stop () { io_manager::unreg (*this); }
172
173 template<class O, class M>
174 check_watcher (O object, M method)
175 : callback<void (check_watcher &)> (object, method)
176 { }
177 ~check_watcher () { stop (); }
178};
179#endif
180
181#if IOM_IDLE
182// run after checking for any i/o, but before waiting
183struct idle_watcher : watcher, callback<void (idle_watcher &)> {
184 void start () { io_manager::reg (*this); }
185 void stop () { io_manager::unreg (*this); }
186
187 template<class O, class M>
188 idle_watcher (O object, M method)
189 : callback<void (idle_watcher &)> (object, method)
67 { } 190 { }
68 191 ~idle_watcher () { stop (); }
69 ~io_watcher ()
70 {
71 iom.unreg (this);
72 }
73
74 void start (int fd_, short events_)
75 {
76 fd = fd_;
77 events = events_;
78 iom.reg (this);
79 }
80
81 void stop ()
82 {
83 iom.unreg (this);
84 }
85}; 192};
193#endif
86 194
87#define TSTAMP_CANCEL -1. 195#if IOM_SIG
88
89struct time_watcher : callback1<void, time_watcher &> { 196struct sig_watcher : watcher, callback<void (sig_watcher &)> {
90 bool registered; // already registered? 197 int signum;
91 tstamp at;
92 198
199 void start (int signum);
200 void stop () { io_manager::unreg (*this); }
201
93 template<class O1, class O2> 202 template<class O, class M>
94 time_watcher (O1 *object, void (O2::*method)(time_watcher &)) 203 sig_watcher (O object, M method)
95 : callback1<void, time_watcher &>(object,method) 204 : callback<void(sig_watcher &)> (object, method), signum (0)
96 , registered(false)
97 { } 205 { }
206 ~sig_watcher () { stop (); }
207};
208#endif
98 209
99 ~time_watcher () 210#if IOM_CHILD
100 { 211struct child_watcher : watcher, callback<void (child_watcher &, int)> {
101 iom.unreg (this); 212 int /*pid_t*/ pid;
213
214 void start (int pid) { this->pid = pid; io_manager::reg (*this); }
215 void stop () { io_manager::unreg (*this); }
216
217 template<class O, class M>
218 child_watcher (O object, M method)
219 : callback<void (child_watcher &, int)> (object, method), pid (0)
102 } 220 { }
103 221 ~child_watcher () { stop (); }
104 void set (tstamp when);
105 void trigger ();
106
107 void operator ()()
108 {
109 trigger ();
110 }
111
112 void start ();
113 void start (tstamp when)
114 {
115 set (when);
116 }
117
118 void stop ()
119 {
120 iom.unreg (this);
121 }
122
123 void reset (tstamp when = TSTAMP_CANCEL)
124 {
125 stop ();
126 at = when;
127 }
128}; 222};
129
130#endif 223#endif
131 224
225#endif
226

Diff Legend

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