← Back to branch summary

~serge-hallyn/ubuntu/natty/lxc/lxc-clone

« back to all changes in this revision

Viewing changes to src/lxc/mainloop.c

  • Committer: Bazaar Package Importer
  • Author(s): Guido Trotter
  • Date: 2009-04-29 17:49:13 UTC
  • Revision ID: james.westby@ubuntu.com-20090429174913-jvahs1ykizqtodje
Tags: upstream-0.6.2
Import upstream version 0.6.2

Show diffs side-by-side

added added

removed removed

Lines of Context:
src/lxc/mainloop.c
 
1
/*
 
2
 * lxc: linux Container library
 
3
 *
 
4
 * (C) Copyright IBM Corp. 2007, 2008
 
5
 *
 
6
 * Authors:
 
7
 * Daniel Lezcano <dlezcano at fr.ibm.com>
 
8
 *
 
9
 * This library is free software; you can redistribute it and/or
 
10
 * modify it under the terms of the GNU Lesser General Public
 
11
 * License as published by the Free Software Foundation; either
 
12
 * version 2.1 of the License, or (at your option) any later version.
 
13
 *
 
14
 * This library is distributed in the hope that it will be useful,
 
15
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 
16
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 
17
 * Lesser General Public License for more details.
 
18
 *
 
19
 * You should have received a copy of the GNU Lesser General Public
 
20
 * License along with this library; if not, write to the Free Software
 
21
 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
 
22
 */
 
23
#include <stdio.h>
 
24
#include <string.h>
 
25
#include <stdlib.h>
 
26
#include <errno.h>
 
27
#include <unistd.h>
 
28
#include <sys/epoll.h>
 
29
 
 
30
#include "mainloop.h"
 
31
 
 
32
struct lxc_handler {
 
33
        lxc_mainloop_callback_t callback;
 
34
        int fd;
 
35
        void *data;
 
36
};
 
37
 
 
38
int lxc_mainloop(struct lxc_epoll_descr *descr)
 
39
{
 
40
        int i, nfds, triggered;
 
41
        struct lxc_handler *handler;
 
42
 
 
43
        for (;;) {
 
44
 
 
45
                triggered = 0;
 
46
 
 
47
                nfds = epoll_wait(descr->epfd, descr->ev, descr->nfds, -1);
 
48
                if (nfds < 0) {
 
49
                        if (errno == EINTR)
 
50
                                continue;
 
51
                        return -1;
 
52
                }
 
53
 
 
54
                for (i = 0; i < descr->nfds; i++) {
 
55
 
 
56
                        if (!(descr->ev[i].events & EPOLLIN) &&
 
57
                            !(descr->ev[i].events & EPOLLHUP))
 
58
                                continue;
 
59
 
 
60
                        triggered++;
 
61
                        handler = (struct lxc_handler *)descr->ev[i].data.ptr;
 
62
 
 
63
                        /* If the handler returns a positive value, exit
 
64
                           the mainloop */
 
65
                        if (handler->callback(handler->fd, handler->data, 
 
66
                                              descr) > 0)
 
67
                                return 0;
 
68
 
 
69
                        if (triggered == nfds)
 
70
                                break;
 
71
                }
 
72
 
 
73
                if (!descr->nfds)
 
74
                        return 0;
 
75
        }
 
76
}
 
77
 
 
78
int lxc_mainloop_add_handler(struct lxc_epoll_descr *descr, int fd, 
 
79
                             lxc_mainloop_callback_t callback, void *data)
 
80
{
 
81
        struct epoll_event *ev;
 
82
        struct lxc_handler *handler;
 
83
        int ret = -1;
 
84
 
 
85
        handler = malloc(sizeof(*handler));
 
86
        if (!handler)
 
87
                return -1;
 
88
 
 
89
        handler->callback = callback;
 
90
        handler->fd = fd;
 
91
        handler->data = data;
 
92
 
 
93
        ev = malloc(sizeof(*descr->ev) * (descr->nfds + 1));
 
94
        if (!ev)
 
95
                goto out_free;
 
96
 
 
97
        if (descr->nfds) {
 
98
                memcpy(ev, descr->ev, sizeof(*descr->ev) * (descr->nfds));
 
99
                free(descr->ev);
 
100
        }
 
101
 
 
102
        descr->ev = ev;
 
103
        descr->ev[descr->nfds].events = EPOLLIN;
 
104
        descr->ev[descr->nfds].data.ptr = handler;
 
105
 
 
106
        ret = epoll_ctl(descr->epfd, EPOLL_CTL_ADD, fd, 
 
107
                        &descr->ev[descr->nfds]);
 
108
 
 
109
        descr->nfds++;
 
110
out:
 
111
        return ret;
 
112
 
 
113
out_free:
 
114
        free(handler);
 
115
        goto out;
 
116
}
 
117
 
 
118
int lxc_mainloop_del_handler(struct lxc_epoll_descr *descr, int fd)
 
119
{
 
120
        struct epoll_event *ev;
 
121
        struct lxc_handler *handler;
 
122
        int i, j, idx = 0;
 
123
 
 
124
        for (i = 0; i < descr->nfds; i++) {
 
125
                
 
126
                handler = descr->ev[i].data.ptr;
 
127
 
 
128
                if (handler->fd != fd)
 
129
                        continue;
 
130
 
 
131
                if (epoll_ctl(descr->epfd, EPOLL_CTL_DEL, fd, NULL))
 
132
                        return -1;
 
133
 
 
134
                ev = malloc(sizeof(*ev) * (descr->nfds - 1));
 
135
                if (!ev)
 
136
                        return -1;
 
137
 
 
138
                for (j = 0; j < descr->nfds; j++) {
 
139
                        if (i == j)
 
140
                                continue;
 
141
                        ev[idx] = descr->ev[idx];
 
142
                        idx++;
 
143
                }
 
144
 
 
145
                free(descr->ev[i].data.ptr);
 
146
                free(descr->ev);
 
147
                descr->ev = ev;
 
148
                descr->nfds--;
 
149
                
 
150
                return 0;
 
151
        }
 
152
 
 
153
        return -1;
 
154
}
 
155
 
 
156
int lxc_mainloop_open(int size, struct lxc_epoll_descr *descr)
 
157
{
 
158
        descr->nfds = 0;
 
159
        descr->ev = NULL;
 
160
 
 
161
        descr->epfd = epoll_create(size);
 
162
        if (descr->epfd < 0)
 
163
                return -1;
 
164
 
 
165
        return 0;
 
166
}
 
167
 
 
168
int lxc_mainloop_close(struct lxc_epoll_descr *descr)
 
169
{
 
170
        int i;
 
171
 
 
172
        for (i = 0; i < descr->nfds; i++)
 
173
                free(descr->ev[i].data.ptr);
 
174
        free(descr->ev);
 
175
 
 
176
        return close(descr->epfd);
 
177
}
 
178