2 * This program is free software; you can redistribute it and/or modify
3 * it under the terms of the GNU General Public License as published by
4 * the Free Software Foundation; either version 2 of the License, or (at
5 * your option) any later version.
7 * This program is distributed in the hope that it will be useful, but
8 * WITHOUT ANY WARRANTY; without even the implied warranty of
9 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
10 * General Public License for more details.
12 * You should have received a copy of the GNU General Public License
13 * along with this program; if not, write to the Free Software
14 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
17 * Copyright © 2006 Pierre Habouzit
20 #include <sys/epoll.h>
21 #include <sys/socket.h>
23 # include <linux/poll.h>
25 # define EPOLLRDHUP POLLRDHUP
33 static int epollfd = -1;
35 static int el_job_setemode(job_t *w, el_mode emode)
37 static int const evtmode_to_epoll[] = {
38 [EL_NEW] = EPOLLRDHUP,
39 [EL_READING] = EPOLLIN,
40 [EL_WRITING] = EPOLLOUT,
41 [EL_RDWR] = EPOLLIN | EPOLLOUT,
42 [EL_IDLE] = EPOLLRDHUP,
45 assert (w->mode == emode || emode == EL_WRITING || emode == EL_READING);
47 if (emode != w->emode) {
48 struct epoll_event event = {
50 .events = evtmode_to_epoll[emode],
52 int action = w->emode == EL_NEW ? EPOLL_CTL_ADD : EPOLL_CTL_MOD;
53 if (epoll_ctl(epollfd, action, w->fd, &event) < 0) {
54 return el_job_release(w, true);
61 int el_job_setmode(job_t *w, el_mode mode)
63 if (w->mode == w->emode) {
65 return el_job_setemode(w, mode);
72 int el_job_release(job_t *w, el_status reason)
74 w->state = EL_LLP_FINI;
75 if (w->m && w->m->finalize) {
76 w->m->finalize(w, reason);
85 static int el_job_connecting(job_t *w)
88 socklen_t len = sizeof(err);
90 if (getsockopt(w->fd, SOL_SOCKET, SO_ERROR, (void *)&err, &len) || err)
91 return el_job_release(w, EL_ERROR);
93 w->state = EL_LLP_READY;
94 return w->m->on_event(w, EL_EVT_RUNNING);
97 int el_job_connect(job_t *w, struct sockaddr *addr, socklen_t len,
100 int res, sock = socket(addr->sa_family, type, proto);
105 res = fcntl(sock, F_GETFL);
108 if (fcntl(sock, F_SETFL, res | O_NONBLOCK) < 0)
110 if (connect(sock, addr, len) < 0)
114 w->llp = &el_job_connecting;
115 return el_job_setmode(w, EL_WRITING);
119 return el_job_release(w, EL_ERROR);
122 ssize_t el_job_read(job_t *w, buffer_t *buf)
126 buffer_ensure(buf, BUFSIZ);
129 nr = gnutls_record_recv(w->session, buf->data + buf->len, BUFSIZ);
130 if (nr < 0 && gnutls_error_is_fatal(nr))
131 return el_job_release(w, EL_RDHUP);
133 int wr = gnutls_record_get_direction(w->session);
134 return el_job_setemode(w, wr ? EL_WRITING : EL_READING);
136 EL_JOB_CHECK(el_job_setemode(w, w->mode));
138 nr = read(w->fd, buf->data + buf->len, BUFSIZ);
140 if (errno == EINTR || errno == EAGAIN)
144 return el_job_release(w, EL_RDHUP);
146 buffer_extend(buf, nr);
150 int el_dispatch(int timeout)
152 struct epoll_event events[FD_SETSIZE];
153 int count = epoll_wait(epollfd, events, countof(events), timeout);
156 if (errno == EAGAIN || errno == EINTR)
158 mutt_error("epoll_wait");
159 mutt_exit(EXIT_FAILURE);
162 while (--count >= 0) {
163 job_t *w = events[count].data.ptr;
164 int event = events[count].events;
173 if (event & EPOLLRDHUP) {
174 IGNORE(el_job_release(w, EL_RDHUP));
175 } else if (w->mode != w->emode) {
176 w->m->on_event(w, EL_EVT_INOUT ^ w->emode);
180 if (event & EPOLLOUT)
182 w->m->on_event(w, evt);
187 IGNORE(el_job_release(w, EL_ERROR));