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.6 by pcg, Fri Mar 28 04:05:10 2003 UTC vs.
Revision 1.29 by pcg, Tue Apr 26 00:55:56 2005 UTC

1/* 1/*
2 iom.h -- I/O multiplexor 2 iom.h -- generic I/O multiplexer
3 Copyright (C) 2003, 2004 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#include "callback.h"
25 32
26#include "slog.h" 33#ifndef IOM_IO
34# define IOM_IO 0
35#endif
36#ifndef IOM_TIME
37# define IOM_TIME 0
38#endif
39#ifndef IOM_CHECK
40# define IOM_CHECK 0
41#endif
42#ifndef IOM_IDLE
43# define IOM_IDLE 0
44#endif
45#ifndef IOM_SIG
46# define IOM_SIG 0
47#endif
27 48
28typedef double tstamp; 49typedef double tstamp;
29
30extern tstamp NOW; 50extern tstamp NOW;
31 51
32template<class R, class A> class callback; 52// TSTAMP_MAX must still fit into a positive struct timeval
53#define TSTAMP_MAX (double)(1UL<<31)
54
55struct watcher;
56#if IOM_IO
33struct io_watcher; 57struct io_watcher;
58#endif
59#if IOM_TIME
34struct time_watcher; 60struct time_watcher;
61#endif
62#if IOM_CHECK
63struct check_watcher;
64#endif
65#if IOM_IDLE
66struct idle_watcher;
67#endif
68#if IOM_SIG
69struct sig_watcher;
70#endif
35 71
72template<class watcher>
73struct io_manager_vec : vector<watcher *> {
74 void erase_unordered (unsigned int pos)
75 {
76 watcher *w = (*this)[this->size () - 1];
77 this->pop_back ();
78
79 if (!this->empty ())
80 if (((*this)[pos] = w)) // '=' is correct!
81 w->active = pos + 1;
82 }
83};
84
85// only used as a namespace, and for initialisation purposes
36class io_manager { 86class io_manager {
37 vector<pollfd> pfs; 87 template<class watcher>
38 vector<const io_watcher *> iow; 88 static void reg (watcher &w, io_manager_vec<watcher> &queue);
39 vector<time_watcher *> tw; // actually a heap
40 89
41 void idle_cb (tstamp &ts); time_watcher *idle; 90 template<class watcher>
91 static void unreg (watcher &w, io_manager_vec<watcher> &queue);
92
42public: 93public:
94#if IOM_TIME
95 // fetch time only
96 static tstamp now ();
43 97
44 void reschedule_time_watchers (); 98 // set NOW
99 static void set_now ();
100#endif
45 101
46 // register a watcher 102 // register a watcher
47 void reg (int fd, short events, io_watcher *w); 103#if IOM_IO
48 void unreg (const io_watcher *w); 104 static void reg (io_watcher &w); static void unreg (io_watcher &w);
49 void reg (time_watcher *w); 105#endif
50 void unreg (const time_watcher *w); 106#if IOM_TIME
107 static void reg (time_watcher &w); static void unreg (time_watcher &w);
108#endif
109#if IOM_CHECK
110 static void reg (check_watcher &w); static void unreg (check_watcher &w);
111#endif
112#if IOM_IDLE
113 static void reg (idle_watcher &w); static void unreg (idle_watcher &w);
114#endif
115#if IOM_SIG
116 static void reg (sig_watcher &w); static void unreg (sig_watcher &w);
117#endif
51 118
52 void loop (); 119 static void loop ();
53
54 io_manager ();
55 ~io_manager ();
56}; 120};
57 121
58extern io_manager iom; 122struct watcher {
123 int active; /* 0 == inactive, else index into respective vector */
59 124
125 watcher () : active (0) { }
126};
127
128#if IOM_IO
129enum { EVENT_READ = 1, EVENT_WRITE = 2 };
130
131struct io_watcher : watcher, callback2<void, io_watcher &, short> {
132 int fd;
133 short events;
134
135 void set (int fd_, short events_) { fd = fd_; events = events_; }
136
137 void set (short events_) { set (fd, events_); }
138 void start () { io_manager::reg (*this); }
139 void start (int fd_, short events_) { set (fd_, events_); io_manager::reg (*this); }
140 void stop () { io_manager::unreg (*this); }
141
60template<class R, class A> 142 template<class O1, class O2>
61class callback {
62 struct object { };
63
64 void *obj;
65 R (object::*meth)(A arg);
66
67 // a proxy is a kind of recipe on how to call a specific class method
68 struct proxy_base {
69 virtual R call (void *obj, R (object::*meth)(A), A arg) = 0;
70 };
71 template<class O1, class O2>
72 struct proxy : proxy_base {
73 virtual R call (void *obj, R (object::*meth)(A), A arg)
74 {
75 ((reinterpret_cast<O1 *>(obj)) ->* (reinterpret_cast<R (O2::*)(A)>(meth)))
76 (arg);
77 }
78 };
79
80 proxy_base *prxy;
81
82public:
83 template<class O1, class O2>
84 callback (O1 *object, R (O2::*method)(A))
85 {
86 static proxy<O1,O2> p;
87 obj = reinterpret_cast<void *>(object);
88 meth = reinterpret_cast<R (object::*)(A)>(method);
89 prxy = &p;
90 }
91
92 R call(A arg) const
93 {
94 return prxy->call (obj, meth, arg);
95 }
96
97 R operator ()(A arg) const
98 {
99 return call (arg);
100 }
101};
102
103struct io_watcher : callback<void, short> {
104 template<class O1, class O2>
105 io_watcher (O1 *object, void (O2::*method)(short revents)) 143 io_watcher (O1 *object, void (O2::*method) (io_watcher &, short))
144 : callback2<void, io_watcher &, short> (object,method)
145 { }
146 ~io_watcher () { stop (); }
147};
148#endif
149
150#if IOM_TIME
151struct time_watcher : watcher, callback1<void, time_watcher &> {
152 tstamp at;
153
154 void trigger ();
155
156 void set (tstamp when) { at = when; }
157 void operator () () { trigger (); }
158 void start () { io_manager::reg (*this); }
159 void start (tstamp when) { set (when); io_manager::reg (*this); }
160 void stop () { io_manager::unreg (*this); }
161
162 template<class O1, class O2>
163 time_watcher (O1 *object, void (O2::*method) (time_watcher &))
164 : callback1<void, time_watcher &> (object,method), at (0)
165 { }
166 ~time_watcher () { stop (); }
167};
168#endif
169
170#if IOM_CHECK
171// run before checking for new events
172struct check_watcher : watcher, callback1<void, check_watcher &> {
173 void start () { io_manager::reg (*this); }
174 void stop () { io_manager::unreg (*this); }
175
176 template<class O1, class O2>
177 check_watcher (O1 *object, void (O2::*method) (check_watcher &))
178 : callback1<void, check_watcher &> (object,method)
179 { }
180 ~check_watcher () { stop (); }
181};
182#endif
183
184#if IOM_IDLE
185// run after checking for any i/o, but before waiting
186struct idle_watcher : watcher, callback1<void, idle_watcher &> {
187 void start () { io_manager::reg (*this); }
188 void stop () { io_manager::unreg (*this); }
189
190 template<class O1, class O2>
191 idle_watcher (O1 *object, void (O2::*method) (idle_watcher &))
106 : callback<void, short>(object,method) 192 : callback1<void, idle_watcher &> (object,method)
107 { } 193 { }
108 194 ~idle_watcher () { stop (); }
109 ~io_watcher ()
110 {
111 iom.unreg (this);
112 }
113
114 void start (int fd, short events)
115 {
116 iom.reg (fd, events, this);
117 }
118
119 void stop () const
120 {
121 iom.unreg (this);
122 }
123}; 195};
196#endif
124 197
125#define TSTAMP_CANCEL -1. 198#if IOM_SIG
199struct sig_watcher : watcher, callback1<void, sig_watcher &> {
200 int signum;
126 201
127struct time_watcher : callback<void, tstamp &> { 202 void start (int signum);
128 bool registered; // already registered? 203 void stop () { io_manager::unreg (*this); }
129 tstamp at;
130 204
131 template<class O1, class O2> 205 template<class O1, class O2>
132 time_watcher (O1 *object, void (O2::*method)(tstamp &)) 206 sig_watcher (O1 *object, void (O2::*method) (sig_watcher &))
133 : callback<void, tstamp &>(object,method) 207 : callback1<void, sig_watcher &> (object,method), signum (-1)
134 , registered(false)
135 { } 208 { }
136 209 ~sig_watcher () { stop (); }
137 ~time_watcher ()
138 {
139 iom.unreg (this);
140 }
141
142 void set (tstamp when);
143 void trigger ();
144
145 void operator ()()
146 {
147 trigger ();
148 }
149
150 void start ();
151 void start (tstamp when)
152 {
153 set (when);
154 }
155
156 void stop () const
157 {
158 iom.unreg (this);
159 }
160
161 void reset (tstamp when = TSTAMP_CANCEL)
162 {
163 stop ();
164 at = when;
165 }
166}; 210};
167
168#endif 211#endif
169 212
213#endif
214

Diff Legend

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