[ViewVC] Diff of: cvs/gvpe/src/iom.C

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

Diff Legend

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

Comparing gvpe/src/iom.C (file contents): Revision 1.12 by pcg, Thu Oct 16 02:41:21 2003 UTC vs. Revision 1.27 by pcg, Sun Dec 19 18:44:33 2004 UTC

Diff Legend

Comparing gvpe/src/iom.C (file contents):
Revision 1.12 by pcg, Thu Oct 16 02:41:21 2003 UTC vs.
Revision 1.27 by pcg, Sun Dec 19 18:44:33 2004 UTC