lib-sys/evtloop.c

   1 /*
   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.
   6  *
   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.
  11  *
  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,
  15  *  MA 02110-1301, USA.
  16  *
  17  *  Copyright © 2006 Pierre Habouzit
  18  */
  19
  20 #include <sys/epoll.h>
  21 #include <sys/socket.h>
  22 #ifndef EPOLLRDHUP
  23 #  include <linux/poll.h>
  24 #  ifdef POLLRDHUP
  25 #    define EPOLLRDHUP POLLRDHUP
  26 #  else
  27 #    define EPOLLRDHUP 0
  28 #  endif
  29 #endif
  30 #include "evtloop.h"
  31 #include "mutt.h"
  32
  33 static int epollfd = -1;
  34
  35 static int el_job_setemode(job_t *w, el_mode emode)
  36 {
  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,
  43     };
  44
  45     assert (w->mode == emode || emode == EL_WRITING || emode == EL_READING);
  46
  47     if (emode != w->emode) {
  48         struct epoll_event event = {
  49             .data.ptr = w,
  50             .events   = evtmode_to_epoll[emode],
  51         };
  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);
  55         }
  56     }
  57     w->emode = emode;
  58     return 0;
  59 }
  60
  61 int el_job_setmode(job_t *w, el_mode mode)
  62 {
  63     if (w->mode == w->emode) {
  64         w->mode = mode;
  65         return el_job_setemode(w, mode);
  66     } else {
  67         w->mode = mode;
  68         return 0;
  69     }
  70 }
  71
  72 int el_job_release(job_t *w, el_status reason)
  73 {
  74     w->state = EL_LLP_FINI;
  75     if (w->m && w->m->finalize) {
  76         w->m->finalize(w, reason);
  77     }
  78     if (w->fd >= 0) {
  79         close(w->fd);
  80     }
  81     job_delete(&w);
  82     return -1;
  83 }
  84
  85 static int el_job_connecting(job_t *w)
  86 {
  87     int err = 0;
  88     socklen_t len = sizeof(err);
  89
  90     if (getsockopt(w->fd, SOL_SOCKET, SO_ERROR, (void *)&err, &len) || err)
  91         return el_job_release(w, EL_ERROR);
  92
  93     w->state = EL_LLP_READY;
  94     return w->m->on_event(w, EL_EVT_RUNNING);
  95 }
  96
  97 int el_job_connect(job_t *w, struct sockaddr *addr, socklen_t len,
  98                    int type, int proto)
  99 {
 100     int res, sock = socket(addr->sa_family, type, proto);
 101
 102     if (sock < 0)
 103         goto error;
 104
 105     res = fcntl(sock, F_GETFL);
 106     if (res < 0)
 107         goto error;
 108     if (fcntl(sock, F_SETFL, res | O_NONBLOCK) < 0)
 109         goto error;
 110     if (connect(sock, addr, len) < 0)
 111         goto error;
 112
 113     w->fd  = sock;
 114     w->llp = &el_job_connecting;
 115     return el_job_setmode(w, EL_WRITING);
 116
 117   error:
 118     close(sock);
 119     return el_job_release(w, EL_ERROR);
 120 }
 121
 122 ssize_t el_job_read(job_t *w, buffer_t *buf)
 123 {
 124     ssize_t nr;
 125
 126     buffer_ensure(buf, BUFSIZ);
 127
 128     if (w->session) {
 129         nr = gnutls_record_recv(w->session, buf->data + buf->len, BUFSIZ);
 130         if (nr < 0 && !gnutls_error_is_fatal(nr)) {
 131             int wr = gnutls_record_get_direction(w->session);
 132             return el_job_setemode(w, wr ? EL_WRITING : EL_READING);
 133         }
 134         EL_JOB_CHECK(el_job_setemode(w, w->mode));
 135     } else {
 136         nr = read(w->fd, buf->data + buf->len, BUFSIZ);
 137         if (nr < 0 && (errno == EINTR || errno == EAGAIN))
 138             return 0;
 139     }
 140     if (nr <= 0)
 141         return el_job_release(w, EL_RDHUP);
 142     buffer_extend(buf, nr);
 143     return nr;
 144 }
 145
 146 ssize_t el_job_write(job_t *w, buffer_t *buf)
 147 {
 148     ssize_t nr;
 149
 150     if (buf->len == 0)
 151         return 0;
 152
 153     if (w->session) {
 154         nr = gnutls_record_send(w->session, buf->data, buf->len);
 155         if (nr < 0 && !gnutls_error_is_fatal(nr)) {
 156             int wr = gnutls_record_get_direction(w->session);
 157             return el_job_setemode(w, wr ? EL_WRITING : EL_READING);
 158         }
 159         EL_JOB_CHECK(el_job_setemode(w, w->mode));
 160     } else {
 161         nr = write(w->fd, buf->data, buf->len);
 162         if (nr < 0 && (errno == EINTR || errno == EAGAIN))
 163             return 0;
 164     }
 165     if (nr <= 0)
 166         return el_job_release(w, EL_RDHUP);
 167     buffer_splice(buf, 0, nr, NULL, 0);
 168     return nr;
 169 }
 170
 171 int el_dispatch(int timeout)
 172 {
 173     struct epoll_event events[FD_SETSIZE];
 174     int count = epoll_wait(epollfd, events, countof(events), timeout);
 175
 176     if (count < 0) {
 177         if (errno == EAGAIN || errno == EINTR)
 178             return 0;
 179         mutt_error("epoll_wait");
 180         mutt_exit(EXIT_FAILURE);
 181     }
 182
 183     while (--count >= 0) {
 184         job_t *w  = events[count].data.ptr;
 185         int event = events[count].events;
 186         int evt   = 0;
 187
 188         switch (w->state) {
 189           case EL_LLP_INIT:
 190             w->llp(w);
 191             break;
 192
 193           case EL_LLP_READY:
 194             if (event & EPOLLRDHUP) {
 195                 IGNORE(el_job_release(w, EL_RDHUP));
 196             } else if (w->mode != w->emode) {
 197                     w->m->on_event(w, EL_EVT_INOUT ^ w->emode);
 198             } else {
 199                 if (event & EPOLLIN)
 200                     evt |= EL_EVT_IN;
 201                 if (event & EPOLLOUT)
 202                     evt |= EL_EVT_OUT;
 203                 w->m->on_event(w, evt);
 204             }
 205             break;
 206
 207           default:
 208             IGNORE(el_job_release(w, EL_ERROR));
 209             break;
 210         }
 211     }
 212
 213     return 0;
 214 }