File: fdset.c

package info (click to toggle)
alevt 1:1.6.2-4
  • links: PTS
  • area: main
  • in suites: squeeze
  • size: 956 kB
  • ctags: 662
  • sloc: ansic: 6,284; makefile: 119; perl: 104; sh: 15
file content (99 lines) | stat: -rw-r--r-- 1,894 bytes parent folder | download | duplicates (4)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
#include <stdlib.h>
#include <unistd.h>
#include <sys/time.h>

#include "dllist.h"
#include "misc.h"
#include "fdset.h"

struct fdset fds[1];		/* global fd list */

int
fdset_init(struct fdset *fds)
{
    dl_init(fds->list);
    fds->del_count = 0;
    return 0;
}

int
fdset_add_fd(struct fdset *fds, int fd, void *handler, void *data)
{
    struct fdset_node *fn;

    if (fd < 0)
	return -1;
    if (handler == 0)
	return -1;

    for (fn = PTR fds->list->first; fn->node->next; fn = PTR fn->node->next)
	if (fn->fd == fd)
	    return -1;

    if (not(fn = malloc(sizeof(*fn))))
	return -1;
    fn->fd = fd;
    fn->handler = handler;
    fn->data = data;
    dl_insert_last(fds->list, fn->node);
    return 0;
}

int
fdset_del_fd(struct fdset *fds, int fd)
{
    struct fdset_node *fn;

    for (fn = PTR fds->list->first; fn->node->next; fn = PTR fn->node->next)
	if (fn->fd == fd)
	{
	    dl_remove(fn->node);
	    free(fn);
	    fds->del_count++;
	    return 0;
	}
    return -1;
}

int
fdset_select(struct fdset *fds, int timeout)
{
    struct fdset_node *fn;
    fd_set rfds[1];
    struct timeval tv[1], *tvp = 0;
    int max_fd, x, del_count;

    FD_ZERO(rfds);
    max_fd = 0;
    for (fn = PTR fds->list->first; fn->node->next; fn = PTR fn->node->next)
    {
	FD_SET(fn->fd, rfds);
	if (fn->fd >= max_fd)
	    max_fd = fn->fd + 1;
    }

    if (timeout >= 0)
    {
	tv->tv_sec = timeout/1000;
	tv->tv_usec = timeout%1000*1000;
	tvp = tv;
    }

    x = select(max_fd, rfds, 0, 0, tvp);
    if (x <= 0)
	return x;

    /* A little bit complicated.  A called handler may modify the fdset... */
restart:
    del_count = fds->del_count;
    for (fn = PTR fds->list->first; fn->node->next; fn = PTR fn->node->next)
	if (FD_ISSET(fn->fd, rfds))
	{
	    FD_CLR(fn->fd, rfds);
	    fn->handler(fn->data, fn->fd);
	    if (fds->del_count != del_count)
		goto restart;
	}

    return 1;
}