3 * $Id: sel.c,v 1.8 2000/03/23 20:42:08 mdw Exp $
5 * I/O multiplexing support
7 * (c) 1999 Straylight/Edgeware
10 /*----- Licensing notice --------------------------------------------------*
12 * This file is part of the mLib utilities library.
14 * mLib is free software; you can redistribute it and/or modify
15 * it under the terms of the GNU Library General Public License as
16 * published by the Free Software Foundation; either version 2 of the
17 * License, or (at your option) any later version.
19 * mLib is distributed in the hope that it will be useful,
20 * but WITHOUT ANY WARRANTY; without even the implied warranty of
21 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
22 * GNU Library General Public License for more details.
24 * You should have received a copy of the GNU Library General Public
25 * License along with mLib; if not, write to the Free
26 * Software Foundation, Inc., 59 Temple Place - Suite 330, Boston,
30 /*----- Revision history --------------------------------------------------*
33 * Revision 1.8 2000/03/23 20:42:08 mdw
34 * Rearrange timeout handling to avoid list corruptions.
36 * Revision 1.7 1999/12/11 11:12:17 mdw
37 * Fix comment formatting error.
39 * Revision 1.6 1999/09/26 14:28:11 mdw
40 * (sel_select): Almost pointless efficiency tweak.
42 * Revision 1.5 1999/08/31 17:42:22 mdw
43 * New function `sel_force' to force a descriptor to be `selected'.
45 * Revision 1.4 1999/08/19 18:30:26 mdw
46 * Implement hooks for foreign select-using systems (currently not well
49 * Revision 1.3 1999/05/21 22:13:59 mdw
50 * Use new `tv' macros. Fix ordering bug for timeout selectors.
52 * Revision 1.2 1999/05/15 10:33:32 mdw
53 * Fix copyright notices.
55 * Revision 1.1 1999/05/14 21:01:14 mdw
56 * Integrated `select' handling bits from the background resolver project.
60 /*----- Header files ------------------------------------------------------*/
66 #include <sys/types.h>
73 /*----- Main code ---------------------------------------------------------*/
75 /* --- @sel_init@ --- *
77 * Arguments: @sel_state *s@ = pointer to a state block to initialize
81 * Use: Initializes a select state block.
84 void sel_init(sel_state
*s
)
88 for (i
= 0; i
< SEL_MODES
; i
++) {
97 /* --- @sel_initfile@ --- *
99 * Arguments: @sel_state *s@ = select state to attach to
100 * @sel_file *f@ = pointer to a file block to initialize
101 * @int fd@ = the file descriptor to listen to
102 * @unsigned mode@ = what to listen for
103 * @void (*func)(int fd, unsigned mode, void *p)@ = handler
104 * @void *p@ = argument to pass to handler
108 * Use: Initializes a file block ready for use. The file block
109 * isn't added to the list of things to do until a call to
113 void sel_initfile(sel_state
*s
, sel_file
*f
,
114 int fd
, unsigned mode
,
115 void (*func
)(int /*fd*/, unsigned /*mode*/, void */
*p*/
),
125 /* --- @sel_addfile@ --- *
127 * Arguments: @sel_file *f@ = pointer to a file block
131 * Use: Adds a file block into the list of things to listen to.
134 void sel_addfile(sel_file
*f
)
136 sel_file
**ff
= &f
->s
->files
[f
->mode
];
138 /* --- This little dance looks like line-noise, but it does the job --- */
140 while (*ff
&& (*ff
)->fd
> f
->fd
)
143 f
->prev
= (sel_file
*)ff
;
147 FD_SET(f
->fd
, f
->s
->fd
+ f
->mode
);
150 /* --- @sel_force@ --- *
152 * Arguments: @sel_file *f@ = pointer to file selector
156 * Use: Forces a file selector to be considered ready. This is only
157 * useful during a call to @sel_select@. Of particular use is
158 * forcing a write selector when there's something interesting
162 void sel_force(sel_file
*f
)
165 FD_SET(f
->fd
, &f
->s
->args
->fd
[f
->mode
]);
168 /* --- @sel_rmfile@ --- *
170 * Arguments: @sel_file *f@ = pointer to a file block
174 * Use: Removes a file block from the list of things to listen to.
177 void sel_rmfile(sel_file
*f
)
179 f
->prev
->next
= f
->next
;
181 f
->next
->prev
= f
->prev
;
182 FD_CLR(f
->fd
, f
->s
->fd
+ f
->mode
);
185 /* --- @sel_addtimer@ --- *
187 * Arguments: @sel_state *s@ = pointer to a state block
188 * @sel_timer *t@ = pointer to a timer block
189 * @struct timeval *tv@ = pointer to time to activate
190 * @void (*func)(struct timeval *tv, void *p)@ = handler
191 * @void *p@ = argument for handler function
195 * Use: Registers and sets up a timer.
198 void sel_addtimer(sel_state
*s
, sel_timer
*t
,
200 void (*func
)(struct timeval */
*tv*/
, void */
*p*/
),
203 sel_timer
**tt
= &s
->timers
;
205 /* --- Set up the timer block --- */
211 /* --- More line noise --- */
213 while (*tt
&& TV_CMP(&(*tt
)->tv
, <, tv
))
216 t
->prev
= (sel_timer
*)tt
;
222 /* --- @sel_rmtimer@ --- *
224 * Arguments: @sel_timer *t@ = pointer to timer block
228 * Use: Removes a timer from the list of timers.
231 void sel_rmtimer(sel_timer
*t
)
233 t
->prev
->next
= t
->next
;
235 t
->next
->prev
= t
->prev
;
238 /* --- @sel_addhook@ --- *
240 * Arguments: @sel_state *s@ = pointer to state block
241 * @sel_hook *h@ = pointer to hook block
242 * @sel_hookfn before, after@ = hook functions
243 * @void *p@ = pointer argument to pass to hook functions
247 * Use: Registers hook functions to be called on each select call.
250 void sel_addhook(sel_state
*s
, sel_hook
*h
,
251 sel_hookfn before
, sel_hookfn after
,
258 h
->prev
= (sel_hook
*)&s
->hooks
;
264 /* --- @sel_rmhook@ --- *
266 * Arguments: @sel_hook *h@ = pointer to hook block
270 * Use: Removes hook functions.
273 void sel_rmhook(sel_hook
*h
)
276 h
->next
->prev
= h
->prev
;
277 h
->prev
->next
= h
->next
;
280 /* --- @sel_fdmerge@ --- *
282 * Arguments: @fd_set *dest@ = destination FD set
283 * @fd_set *fd@ = pointer to set to merge
284 * @int maxfd@ = highest numbered descriptor in @fd@ + 1
286 * Returns: Actual highest numbered descriptor.
288 * Use: Merges file descriptor sets, and returns an accurate @maxfd@
292 int sel_fdmerge(fd_set
*dest
, fd_set
*fd
, int maxfd
)
296 for (i
= 0; i
< maxfd
; i
++) {
297 if (FD_ISSET(i
, fd
)) {
306 /* --- @sel_select@ --- *
308 * Arguments: @sel_state *s@ = pointer to state block
310 * Returns: Zero if all OK, -1 on error.
312 * Use: Does a @select@ call (or equivalent @poll@).
315 int sel_select(sel_state
*s
)
320 /* --- Initialize the argument block --- */
325 for (i
= 0; i
< SEL_MODES
; i
++) {
326 if (s
->files
[i
] && s
->files
[i
]->fd
>= a
.maxfd
)
327 a
.maxfd
= s
->files
[i
]->fd
+ 1;
331 memcpy(a
.fd
, s
->fd
, sizeof(a
.fd
));
332 if (s
->timers
|| s
->hooks
)
333 gettimeofday(&a
.now
, 0);
337 TV_SUB(&a
.tv
, &s
->timers
->tv
, &a
.now
);
342 /* --- Grind through the pre hooks --- */
345 sel_hook
*h
= s
->hooks
;
350 hh
->before(s
, &a
, hh
->p
);
354 /* --- Run the @select@ call --- */
356 if ((err
= select(a
.maxfd
,
357 &a
.fd
[SEL_READ
], &a
.fd
[SEL_WRITE
], &a
.fd
[SEL_EXC
],
364 gettimeofday(&a
.now
, 0);
366 /* --- Run through the hooks again --- */
369 sel_hook
*h
= s
->hooks
;
374 hh
->after(s
, &a
, hh
->p
);
378 /* --- Run through the timers --- */
380 if (s
->timers
&& TV_CMP(&s
->timers
->tv
, <=, &a
.now
)) {
383 for (t
= tt
; t
&& TV_CMP(&t
->tv
, <=, &a
.now
); t
= t
->next
)
387 t
->prev
= (sel_timer
*)&s
->timers
;
390 for (t
= tt
; t
; t
= tt
) {
392 t
->func(&a
.now
, t
->p
);
396 /* --- And finally run through the files --- *
398 * Do reads first. It's quite possible that a read might prompt a write,
399 * but the other way around is less likely. Fortunately, the modes are
400 * in the right order for this.
406 for (i
= 0; i
< SEL_MODES
; i
++) {
408 for (f
= s
->files
[i
]; f
; f
= ff
) {
410 if (FD_ISSET(f
->fd
, a
.fd
+ i
))
411 f
->func(f
->fd
, i
, f
->p
);
420 /*----- That's all, folks -------------------------------------------------*/