Willy Tarreau | de99e99 | 2007-04-16 00:53:59 +0200 | [diff] [blame] | 1 | /* |
| 2 | * FD polling functions for Speculative I/O combined with Linux epoll() |
| 3 | * |
Willy Tarreau | 037d2c1 | 2012-11-06 02:34:46 +0100 | [diff] [blame] | 4 | * Copyright 2000-2012 Willy Tarreau <w@1wt.eu> |
Willy Tarreau | de99e99 | 2007-04-16 00:53:59 +0200 | [diff] [blame] | 5 | * |
| 6 | * This program is free software; you can redistribute it and/or |
| 7 | * modify it under the terms of the GNU General Public License |
| 8 | * as published by the Free Software Foundation; either version |
| 9 | * 2 of the License, or (at your option) any later version. |
Willy Tarreau | de99e99 | 2007-04-16 00:53:59 +0200 | [diff] [blame] | 10 | */ |
| 11 | |
| 12 | #include <unistd.h> |
| 13 | #include <sys/time.h> |
| 14 | #include <sys/types.h> |
| 15 | |
| 16 | #include <common/compat.h> |
| 17 | #include <common/config.h> |
Willy Tarreau | d6f087e | 2008-01-18 17:20:13 +0100 | [diff] [blame] | 18 | #include <common/debug.h> |
Willy Tarreau | 43d8fb2 | 2011-08-22 17:12:02 +0200 | [diff] [blame] | 19 | #include <common/epoll.h> |
Willy Tarreau | de99e99 | 2007-04-16 00:53:59 +0200 | [diff] [blame] | 20 | #include <common/standard.h> |
Willy Tarreau | 0c303ee | 2008-07-07 00:09:58 +0200 | [diff] [blame] | 21 | #include <common/ticks.h> |
Willy Tarreau | de99e99 | 2007-04-16 00:53:59 +0200 | [diff] [blame] | 22 | #include <common/time.h> |
Willy Tarreau | 1db3771 | 2007-06-03 17:16:49 +0200 | [diff] [blame] | 23 | #include <common/tools.h> |
Willy Tarreau | de99e99 | 2007-04-16 00:53:59 +0200 | [diff] [blame] | 24 | |
Willy Tarreau | de99e99 | 2007-04-16 00:53:59 +0200 | [diff] [blame] | 25 | #include <types/global.h> |
| 26 | |
| 27 | #include <proto/fd.h> |
Willy Tarreau | 332740d | 2009-05-10 09:57:21 +0200 | [diff] [blame] | 28 | #include <proto/signal.h> |
Willy Tarreau | de99e99 | 2007-04-16 00:53:59 +0200 | [diff] [blame] | 29 | #include <proto/task.h> |
| 30 | |
Willy Tarreau | de99e99 | 2007-04-16 00:53:59 +0200 | [diff] [blame] | 31 | |
Willy Tarreau | f2e8ee2 | 2008-05-25 10:39:02 +0200 | [diff] [blame] | 32 | static int absmaxevents = 0; // absolute maximum amounts of polled events |
Willy Tarreau | 037d2c1 | 2012-11-06 02:34:46 +0100 | [diff] [blame] | 33 | static int in_poll_loop = 0; // non-null if polled events are being processed |
Willy Tarreau | de99e99 | 2007-04-16 00:53:59 +0200 | [diff] [blame] | 34 | |
Willy Tarreau | de99e99 | 2007-04-16 00:53:59 +0200 | [diff] [blame] | 35 | /* private data */ |
| 36 | static struct epoll_event *epoll_events; |
| 37 | static int epoll_fd; |
| 38 | |
| 39 | /* This structure may be used for any purpose. Warning! do not use it in |
| 40 | * recursive functions ! |
| 41 | */ |
| 42 | static struct epoll_event ev; |
| 43 | |
Willy Tarreau | de99e99 | 2007-04-16 00:53:59 +0200 | [diff] [blame] | 44 | /* |
| 45 | * Returns non-zero if <fd> is already monitored for events in direction <dir>. |
| 46 | */ |
| 47 | REGPRM2 static int __fd_is_set(const int fd, int dir) |
| 48 | { |
Willy Tarreau | 7a52a5c | 2008-08-16 16:06:02 +0200 | [diff] [blame] | 49 | #if DEBUG_DEV |
Willy Tarreau | db3b326 | 2012-07-05 23:19:22 +0200 | [diff] [blame] | 50 | if (!fdtab[fd].owner) { |
Willy Tarreau | 7a52a5c | 2008-08-16 16:06:02 +0200 | [diff] [blame] | 51 | fprintf(stderr, "sepoll.fd_isset called on closed fd #%d.\n", fd); |
| 52 | ABORT_NOW(); |
| 53 | } |
| 54 | #endif |
Willy Tarreau | 037d2c1 | 2012-11-06 02:34:46 +0100 | [diff] [blame] | 55 | return ((unsigned)fdtab[fd].spec_e >> dir) & FD_EV_STATUS; |
Willy Tarreau | de99e99 | 2007-04-16 00:53:59 +0200 | [diff] [blame] | 56 | } |
| 57 | |
| 58 | /* |
| 59 | * Don't worry about the strange constructs in __fd_set/__fd_clr, they are |
| 60 | * designed like this in order to reduce the number of jumps (verified). |
| 61 | */ |
Willy Tarreau | babd05a | 2012-08-09 12:14:03 +0200 | [diff] [blame] | 62 | REGPRM2 static void __fd_wai(const int fd, int dir) |
| 63 | { |
| 64 | unsigned int i; |
| 65 | |
| 66 | #if DEBUG_DEV |
| 67 | if (!fdtab[fd].owner) { |
| 68 | fprintf(stderr, "sepoll.fd_wai called on closed fd #%d.\n", fd); |
| 69 | ABORT_NOW(); |
| 70 | } |
| 71 | #endif |
Willy Tarreau | 037d2c1 | 2012-11-06 02:34:46 +0100 | [diff] [blame] | 72 | i = ((unsigned)fdtab[fd].spec_e >> dir) & FD_EV_STATUS; |
Willy Tarreau | babd05a | 2012-08-09 12:14:03 +0200 | [diff] [blame] | 73 | |
Willy Tarreau | 037d2c1 | 2012-11-06 02:34:46 +0100 | [diff] [blame] | 74 | if (i == FD_EV_POLLED) |
| 75 | return; /* already in desired state */ |
| 76 | updt_fd(fd); /* need an update entry to change the state */ |
| 77 | fdtab[fd].spec_e ^= (i ^ (unsigned int)FD_EV_POLLED) << dir; |
Willy Tarreau | babd05a | 2012-08-09 12:14:03 +0200 | [diff] [blame] | 78 | } |
| 79 | |
Willy Tarreau | 3788e4c | 2012-07-30 14:29:35 +0200 | [diff] [blame] | 80 | REGPRM2 static void __fd_set(const int fd, int dir) |
Willy Tarreau | de99e99 | 2007-04-16 00:53:59 +0200 | [diff] [blame] | 81 | { |
Willy Tarreau | de99e99 | 2007-04-16 00:53:59 +0200 | [diff] [blame] | 82 | unsigned int i; |
| 83 | |
Willy Tarreau | 7a52a5c | 2008-08-16 16:06:02 +0200 | [diff] [blame] | 84 | #if DEBUG_DEV |
Willy Tarreau | db3b326 | 2012-07-05 23:19:22 +0200 | [diff] [blame] | 85 | if (!fdtab[fd].owner) { |
Willy Tarreau | 7a52a5c | 2008-08-16 16:06:02 +0200 | [diff] [blame] | 86 | fprintf(stderr, "sepoll.fd_set called on closed fd #%d.\n", fd); |
| 87 | ABORT_NOW(); |
| 88 | } |
| 89 | #endif |
Willy Tarreau | 037d2c1 | 2012-11-06 02:34:46 +0100 | [diff] [blame] | 90 | i = ((unsigned)fdtab[fd].spec_e >> dir) & FD_EV_STATUS; |
Willy Tarreau | de99e99 | 2007-04-16 00:53:59 +0200 | [diff] [blame] | 91 | |
Willy Tarreau | 037d2c1 | 2012-11-06 02:34:46 +0100 | [diff] [blame] | 92 | /* note that we don't care about disabling the polled state when |
| 93 | * enabling the active state, since it brings no benefit but costs |
| 94 | * some syscalls. |
| 95 | */ |
| 96 | if (i & FD_EV_ACTIVE) |
| 97 | return; /* already in desired state */ |
| 98 | updt_fd(fd); /* need an update entry to change the state */ |
| 99 | fdtab[fd].spec_e |= ((unsigned int)FD_EV_ACTIVE) << dir; |
Willy Tarreau | de99e99 | 2007-04-16 00:53:59 +0200 | [diff] [blame] | 100 | } |
| 101 | |
Willy Tarreau | 3788e4c | 2012-07-30 14:29:35 +0200 | [diff] [blame] | 102 | REGPRM2 static void __fd_clr(const int fd, int dir) |
Willy Tarreau | de99e99 | 2007-04-16 00:53:59 +0200 | [diff] [blame] | 103 | { |
Willy Tarreau | de99e99 | 2007-04-16 00:53:59 +0200 | [diff] [blame] | 104 | unsigned int i; |
| 105 | |
Willy Tarreau | 7a52a5c | 2008-08-16 16:06:02 +0200 | [diff] [blame] | 106 | #if DEBUG_DEV |
Willy Tarreau | db3b326 | 2012-07-05 23:19:22 +0200 | [diff] [blame] | 107 | if (!fdtab[fd].owner) { |
Willy Tarreau | 7a52a5c | 2008-08-16 16:06:02 +0200 | [diff] [blame] | 108 | fprintf(stderr, "sepoll.fd_clr called on closed fd #%d.\n", fd); |
| 109 | ABORT_NOW(); |
| 110 | } |
| 111 | #endif |
Willy Tarreau | 037d2c1 | 2012-11-06 02:34:46 +0100 | [diff] [blame] | 112 | i = ((unsigned)fdtab[fd].spec_e >> dir) & FD_EV_STATUS; |
Willy Tarreau | de99e99 | 2007-04-16 00:53:59 +0200 | [diff] [blame] | 113 | |
Willy Tarreau | 037d2c1 | 2012-11-06 02:34:46 +0100 | [diff] [blame] | 114 | if (i == 0) |
| 115 | return /* already disabled */; |
| 116 | updt_fd(fd); /* need an update entry to change the state */ |
| 117 | fdtab[fd].spec_e ^= i << dir; |
Willy Tarreau | de99e99 | 2007-04-16 00:53:59 +0200 | [diff] [blame] | 118 | } |
| 119 | |
Willy Tarreau | 6653d17 | 2007-05-13 01:52:05 +0200 | [diff] [blame] | 120 | /* normally unused */ |
Willy Tarreau | de99e99 | 2007-04-16 00:53:59 +0200 | [diff] [blame] | 121 | REGPRM1 static void __fd_rem(int fd) |
| 122 | { |
| 123 | __fd_clr(fd, DIR_RD); |
| 124 | __fd_clr(fd, DIR_WR); |
| 125 | } |
| 126 | |
| 127 | /* |
| 128 | * On valid epoll() implementations, a call to close() automatically removes |
| 129 | * the fds. This means that the FD will appear as previously unset. |
| 130 | */ |
| 131 | REGPRM1 static void __fd_clo(int fd) |
| 132 | { |
Willy Tarreau | 7a52a5c | 2008-08-16 16:06:02 +0200 | [diff] [blame] | 133 | release_spec_entry(fd); |
Willy Tarreau | 037d2c1 | 2012-11-06 02:34:46 +0100 | [diff] [blame] | 134 | fdtab[fd].spec_e &= ~(FD_EV_CURR_MASK | FD_EV_PREV_MASK); |
Willy Tarreau | de99e99 | 2007-04-16 00:53:59 +0200 | [diff] [blame] | 135 | } |
| 136 | |
Willy Tarreau | dc246a7 | 2007-05-09 21:57:51 +0200 | [diff] [blame] | 137 | /* |
Willy Tarreau | de99e99 | 2007-04-16 00:53:59 +0200 | [diff] [blame] | 138 | * speculative epoll() poller |
| 139 | */ |
Willy Tarreau | 0c303ee | 2008-07-07 00:09:58 +0200 | [diff] [blame] | 140 | REGPRM2 static void _do_poll(struct poller *p, int exp) |
Willy Tarreau | de99e99 | 2007-04-16 00:53:59 +0200 | [diff] [blame] | 141 | { |
Willy Tarreau | dbcd47e | 2012-05-13 09:42:26 +0200 | [diff] [blame] | 142 | int status, eo, en; |
Willy Tarreau | de99e99 | 2007-04-16 00:53:59 +0200 | [diff] [blame] | 143 | int fd, opcode; |
| 144 | int count; |
| 145 | int spec_idx; |
Willy Tarreau | 037d2c1 | 2012-11-06 02:34:46 +0100 | [diff] [blame] | 146 | int updt_idx; |
Willy Tarreau | d825eef | 2007-05-12 22:35:00 +0200 | [diff] [blame] | 147 | int wait_time; |
Willy Tarreau | de99e99 | 2007-04-16 00:53:59 +0200 | [diff] [blame] | 148 | |
Willy Tarreau | 037d2c1 | 2012-11-06 02:34:46 +0100 | [diff] [blame] | 149 | /* first, scan the update list to find changes */ |
Willy Tarreau | 7be79a4 | 2012-11-11 15:02:54 +0100 | [diff] [blame^] | 150 | for (updt_idx = 0; updt_idx < fd_nbupdt; updt_idx++) { |
| 151 | fd = fd_updt[updt_idx]; |
Willy Tarreau | 45dab73 | 2012-09-02 22:19:18 +0200 | [diff] [blame] | 152 | en = fdtab[fd].spec_e & 15; /* new events */ |
| 153 | eo = fdtab[fd].spec_e >> 4; /* previous events */ |
Willy Tarreau | de99e99 | 2007-04-16 00:53:59 +0200 | [diff] [blame] | 154 | |
Willy Tarreau | 037d2c1 | 2012-11-06 02:34:46 +0100 | [diff] [blame] | 155 | if (fdtab[fd].owner && (eo ^ en)) { |
| 156 | if ((eo ^ en) & FD_EV_POLLED_RW) { |
| 157 | /* poll status changed */ |
| 158 | if ((en & FD_EV_POLLED_RW) == 0) { |
| 159 | /* fd removed from poll list */ |
| 160 | opcode = EPOLL_CTL_DEL; |
| 161 | } |
| 162 | else if ((eo & FD_EV_POLLED_RW) == 0) { |
| 163 | /* new fd in the poll list */ |
| 164 | opcode = EPOLL_CTL_ADD; |
| 165 | } |
| 166 | else { |
| 167 | /* fd status changed */ |
| 168 | opcode = EPOLL_CTL_MOD; |
| 169 | } |
Willy Tarreau | de99e99 | 2007-04-16 00:53:59 +0200 | [diff] [blame] | 170 | |
Willy Tarreau | 037d2c1 | 2012-11-06 02:34:46 +0100 | [diff] [blame] | 171 | /* construct the epoll events based on new state */ |
| 172 | ev.events = 0; |
| 173 | if (en & FD_EV_POLLED_R) |
| 174 | ev.events |= EPOLLIN; |
Willy Tarreau | dbcd47e | 2012-05-13 09:42:26 +0200 | [diff] [blame] | 175 | |
Willy Tarreau | 037d2c1 | 2012-11-06 02:34:46 +0100 | [diff] [blame] | 176 | if (en & FD_EV_POLLED_W) |
| 177 | ev.events |= EPOLLOUT; |
Willy Tarreau | dbcd47e | 2012-05-13 09:42:26 +0200 | [diff] [blame] | 178 | |
Willy Tarreau | 037d2c1 | 2012-11-06 02:34:46 +0100 | [diff] [blame] | 179 | ev.data.fd = fd; |
| 180 | epoll_ctl(epoll_fd, opcode, fd, &ev); |
| 181 | } |
Willy Tarreau | dbcd47e | 2012-05-13 09:42:26 +0200 | [diff] [blame] | 182 | |
Willy Tarreau | 037d2c1 | 2012-11-06 02:34:46 +0100 | [diff] [blame] | 183 | fdtab[fd].spec_e = (en << 4) + en; /* save new events */ |
Willy Tarreau | de99e99 | 2007-04-16 00:53:59 +0200 | [diff] [blame] | 184 | |
Willy Tarreau | 037d2c1 | 2012-11-06 02:34:46 +0100 | [diff] [blame] | 185 | if (!(en & FD_EV_ACTIVE_RW)) { |
| 186 | /* This fd doesn't use any active entry anymore, we can |
| 187 | * kill its entry. |
| 188 | */ |
| 189 | release_spec_entry(fd); |
Willy Tarreau | 6653d17 | 2007-05-13 01:52:05 +0200 | [diff] [blame] | 190 | } |
Willy Tarreau | 037d2c1 | 2012-11-06 02:34:46 +0100 | [diff] [blame] | 191 | else if ((en & ~eo) & FD_EV_ACTIVE_RW) { |
| 192 | /* we need a new spec entry now */ |
| 193 | alloc_spec_entry(fd); |
Willy Tarreau | de99e99 | 2007-04-16 00:53:59 +0200 | [diff] [blame] | 194 | } |
Willy Tarreau | 6653d17 | 2007-05-13 01:52:05 +0200 | [diff] [blame] | 195 | |
Willy Tarreau | 6653d17 | 2007-05-13 01:52:05 +0200 | [diff] [blame] | 196 | } |
Willy Tarreau | 037d2c1 | 2012-11-06 02:34:46 +0100 | [diff] [blame] | 197 | fdtab[fd].updated = 0; |
| 198 | fdtab[fd].new = 0; |
Willy Tarreau | de99e99 | 2007-04-16 00:53:59 +0200 | [diff] [blame] | 199 | } |
Willy Tarreau | 7be79a4 | 2012-11-11 15:02:54 +0100 | [diff] [blame^] | 200 | fd_nbupdt = 0; |
Willy Tarreau | de99e99 | 2007-04-16 00:53:59 +0200 | [diff] [blame] | 201 | |
Willy Tarreau | dbcd47e | 2012-05-13 09:42:26 +0200 | [diff] [blame] | 202 | /* compute the epoll_wait() timeout */ |
Willy Tarreau | cb65125 | 2008-08-29 13:57:30 +0200 | [diff] [blame] | 203 | |
Willy Tarreau | 7be79a4 | 2012-11-11 15:02:54 +0100 | [diff] [blame^] | 204 | if (fd_nbspec || run_queue || signal_queue_len) { |
Willy Tarreau | dbcd47e | 2012-05-13 09:42:26 +0200 | [diff] [blame] | 205 | /* Maybe we still have events in the spec list, or there are |
Willy Tarreau | 3a62811 | 2008-06-13 21:06:56 +0200 | [diff] [blame] | 206 | * some tasks left pending in the run_queue, so we must not |
Willy Tarreau | dbcd47e | 2012-05-13 09:42:26 +0200 | [diff] [blame] | 207 | * wait in epoll() otherwise we would delay their delivery by |
Willy Tarreau | 6653d17 | 2007-05-13 01:52:05 +0200 | [diff] [blame] | 208 | * the next timeout. |
| 209 | */ |
Willy Tarreau | de99e99 | 2007-04-16 00:53:59 +0200 | [diff] [blame] | 210 | wait_time = 0; |
| 211 | } |
Willy Tarreau | d825eef | 2007-05-12 22:35:00 +0200 | [diff] [blame] | 212 | else { |
Willy Tarreau | 0c303ee | 2008-07-07 00:09:58 +0200 | [diff] [blame] | 213 | if (!exp) |
Willy Tarreau | b0b37bc | 2008-06-23 14:00:57 +0200 | [diff] [blame] | 214 | wait_time = MAX_DELAY_MS; |
Willy Tarreau | 0c303ee | 2008-07-07 00:09:58 +0200 | [diff] [blame] | 215 | else if (tick_is_expired(exp, now_ms)) |
Willy Tarreau | bdefc51 | 2007-05-14 02:02:04 +0200 | [diff] [blame] | 216 | wait_time = 0; |
Willy Tarreau | b0b37bc | 2008-06-23 14:00:57 +0200 | [diff] [blame] | 217 | else { |
Willy Tarreau | 0c303ee | 2008-07-07 00:09:58 +0200 | [diff] [blame] | 218 | wait_time = TICKS_TO_MS(tick_remain(now_ms, exp)) + 1; |
Willy Tarreau | b0b37bc | 2008-06-23 14:00:57 +0200 | [diff] [blame] | 219 | if (wait_time > MAX_DELAY_MS) |
| 220 | wait_time = MAX_DELAY_MS; |
| 221 | } |
Willy Tarreau | d825eef | 2007-05-12 22:35:00 +0200 | [diff] [blame] | 222 | } |
Willy Tarreau | de99e99 | 2007-04-16 00:53:59 +0200 | [diff] [blame] | 223 | |
Willy Tarreau | 037d2c1 | 2012-11-06 02:34:46 +0100 | [diff] [blame] | 224 | /* now let's wait for polled events */ |
| 225 | |
Willy Tarreau | dbcd47e | 2012-05-13 09:42:26 +0200 | [diff] [blame] | 226 | fd = MIN(maxfd, global.tune.maxpollevents); |
Willy Tarreau | 45a1251 | 2011-09-10 16:56:42 +0200 | [diff] [blame] | 227 | gettimeofday(&before_poll, NULL); |
Willy Tarreau | 1db3771 | 2007-06-03 17:16:49 +0200 | [diff] [blame] | 228 | status = epoll_wait(epoll_fd, epoll_events, fd, wait_time); |
Willy Tarreau | b0b37bc | 2008-06-23 14:00:57 +0200 | [diff] [blame] | 229 | tv_update_date(wait_time, status); |
Willy Tarreau | 45a1251 | 2011-09-10 16:56:42 +0200 | [diff] [blame] | 230 | measure_idle(); |
Willy Tarreau | de99e99 | 2007-04-16 00:53:59 +0200 | [diff] [blame] | 231 | |
Willy Tarreau | 037d2c1 | 2012-11-06 02:34:46 +0100 | [diff] [blame] | 232 | in_poll_loop = 1; |
| 233 | |
| 234 | /* process polled events */ |
| 235 | |
Willy Tarreau | de99e99 | 2007-04-16 00:53:59 +0200 | [diff] [blame] | 236 | for (count = 0; count < status; count++) { |
| 237 | int e = epoll_events[count].events; |
| 238 | fd = epoll_events[count].data.fd; |
| 239 | |
Willy Tarreau | 076be25 | 2012-07-06 16:02:29 +0200 | [diff] [blame] | 240 | if (!fdtab[fd].owner) |
| 241 | continue; |
| 242 | |
Willy Tarreau | de99e99 | 2007-04-16 00:53:59 +0200 | [diff] [blame] | 243 | /* it looks complicated but gcc can optimize it away when constants |
| 244 | * have same values. |
| 245 | */ |
Willy Tarreau | d6f087e | 2008-01-18 17:20:13 +0100 | [diff] [blame] | 246 | fdtab[fd].ev &= FD_POLL_STICKY; |
Willy Tarreau | 491c498 | 2012-07-06 11:16:01 +0200 | [diff] [blame] | 247 | fdtab[fd].ev |= |
Willy Tarreau | de99e99 | 2007-04-16 00:53:59 +0200 | [diff] [blame] | 248 | ((e & EPOLLIN ) ? FD_POLL_IN : 0) | |
| 249 | ((e & EPOLLPRI) ? FD_POLL_PRI : 0) | |
| 250 | ((e & EPOLLOUT) ? FD_POLL_OUT : 0) | |
| 251 | ((e & EPOLLERR) ? FD_POLL_ERR : 0) | |
| 252 | ((e & EPOLLHUP) ? FD_POLL_HUP : 0); |
Willy Tarreau | 491c498 | 2012-07-06 11:16:01 +0200 | [diff] [blame] | 253 | |
Willy Tarreau | 037d2c1 | 2012-11-06 02:34:46 +0100 | [diff] [blame] | 254 | if (fdtab[fd].iocb && fdtab[fd].owner && fdtab[fd].ev) { |
Willy Tarreau | 7be79a4 | 2012-11-11 15:02:54 +0100 | [diff] [blame^] | 255 | int new_updt, old_updt = fd_nbupdt; /* Save number of updates to detect creation of new FDs. */ |
Willy Tarreau | 037d2c1 | 2012-11-06 02:34:46 +0100 | [diff] [blame] | 256 | |
| 257 | /* Mark the events as speculative before processing |
| 258 | * them so that if nothing can be done we don't need |
| 259 | * to poll again. |
| 260 | */ |
| 261 | if (fdtab[fd].ev & (FD_POLL_IN|FD_POLL_HUP|FD_POLL_ERR)) |
| 262 | __fd_set(fd, DIR_RD); |
| 263 | |
| 264 | if (fdtab[fd].ev & (FD_POLL_OUT|FD_POLL_ERR)) |
| 265 | __fd_set(fd, DIR_WR); |
| 266 | |
Willy Tarreau | 9845e75 | 2012-07-06 11:44:28 +0200 | [diff] [blame] | 267 | fdtab[fd].iocb(fd); |
Willy Tarreau | 037d2c1 | 2012-11-06 02:34:46 +0100 | [diff] [blame] | 268 | |
| 269 | /* One or more fd might have been created during the iocb(). |
| 270 | * This mainly happens with new incoming connections that have |
| 271 | * just been accepted, so we'd like to process them immediately |
| 272 | * for better efficiency. Second benefit, if at the end the fds |
| 273 | * are disabled again, we can safely destroy their update entry |
| 274 | * to reduce the scope of later scans. This is the reason we |
| 275 | * scan the new entries backwards. |
| 276 | */ |
| 277 | |
Willy Tarreau | 7be79a4 | 2012-11-11 15:02:54 +0100 | [diff] [blame^] | 278 | for (new_updt = fd_nbupdt; new_updt > old_updt; new_updt--) { |
| 279 | fd = fd_updt[new_updt - 1]; |
Willy Tarreau | 037d2c1 | 2012-11-06 02:34:46 +0100 | [diff] [blame] | 280 | if (!fdtab[fd].new) |
| 281 | continue; |
| 282 | |
| 283 | fdtab[fd].new = 0; |
| 284 | fdtab[fd].ev &= FD_POLL_STICKY; |
| 285 | |
| 286 | if ((fdtab[fd].spec_e & FD_EV_STATUS_R) == FD_EV_ACTIVE_R) |
| 287 | fdtab[fd].ev |= FD_POLL_IN; |
| 288 | |
| 289 | if ((fdtab[fd].spec_e & FD_EV_STATUS_W) == FD_EV_ACTIVE_W) |
| 290 | fdtab[fd].ev |= FD_POLL_OUT; |
| 291 | |
| 292 | if (fdtab[fd].ev && fdtab[fd].iocb && fdtab[fd].owner) |
| 293 | fdtab[fd].iocb(fd); |
| 294 | |
| 295 | /* we can remove this update entry if it's the last one and is |
| 296 | * unused, otherwise we don't touch anything. |
| 297 | */ |
Willy Tarreau | 7be79a4 | 2012-11-11 15:02:54 +0100 | [diff] [blame^] | 298 | if (new_updt == fd_nbupdt && fdtab[fd].spec_e == 0) { |
Willy Tarreau | 037d2c1 | 2012-11-06 02:34:46 +0100 | [diff] [blame] | 299 | fdtab[fd].updated = 0; |
Willy Tarreau | 7be79a4 | 2012-11-11 15:02:54 +0100 | [diff] [blame^] | 300 | fd_nbupdt--; |
Willy Tarreau | 037d2c1 | 2012-11-06 02:34:46 +0100 | [diff] [blame] | 301 | } |
| 302 | } |
| 303 | } |
Willy Tarreau | de99e99 | 2007-04-16 00:53:59 +0200 | [diff] [blame] | 304 | } |
Willy Tarreau | cb65125 | 2008-08-29 13:57:30 +0200 | [diff] [blame] | 305 | |
Willy Tarreau | dbcd47e | 2012-05-13 09:42:26 +0200 | [diff] [blame] | 306 | /* now process speculative events if any */ |
| 307 | |
Willy Tarreau | 7be79a4 | 2012-11-11 15:02:54 +0100 | [diff] [blame^] | 308 | for (spec_idx = 0; spec_idx < fd_nbspec; ) { |
| 309 | fd = fd_spec[spec_idx]; |
Willy Tarreau | 037d2c1 | 2012-11-06 02:34:46 +0100 | [diff] [blame] | 310 | eo = fdtab[fd].spec_e; |
Willy Tarreau | dbcd47e | 2012-05-13 09:42:26 +0200 | [diff] [blame] | 311 | |
| 312 | /* |
| 313 | * Process the speculative events. |
| 314 | * |
Willy Tarreau | 037d2c1 | 2012-11-06 02:34:46 +0100 | [diff] [blame] | 315 | * Principle: events which are marked FD_EV_ACTIVE are processed |
| 316 | * with their usual I/O callback. The callback may remove the |
| 317 | * events from the list or tag them for polling. Changes will be |
| 318 | * applied on next round. |
Willy Tarreau | cb65125 | 2008-08-29 13:57:30 +0200 | [diff] [blame] | 319 | */ |
Willy Tarreau | dbcd47e | 2012-05-13 09:42:26 +0200 | [diff] [blame] | 320 | |
| 321 | fdtab[fd].ev &= FD_POLL_STICKY; |
Willy Tarreau | 037d2c1 | 2012-11-06 02:34:46 +0100 | [diff] [blame] | 322 | |
| 323 | if ((eo & FD_EV_STATUS_R) == FD_EV_ACTIVE_R) |
Willy Tarreau | 5d526b7 | 2012-07-05 23:33:51 +0200 | [diff] [blame] | 324 | fdtab[fd].ev |= FD_POLL_IN; |
Willy Tarreau | dbcd47e | 2012-05-13 09:42:26 +0200 | [diff] [blame] | 325 | |
Willy Tarreau | 037d2c1 | 2012-11-06 02:34:46 +0100 | [diff] [blame] | 326 | if ((eo & FD_EV_STATUS_W) == FD_EV_ACTIVE_W) |
Willy Tarreau | 5d526b7 | 2012-07-05 23:33:51 +0200 | [diff] [blame] | 327 | fdtab[fd].ev |= FD_POLL_OUT; |
Willy Tarreau | 9845e75 | 2012-07-06 11:44:28 +0200 | [diff] [blame] | 328 | |
Willy Tarreau | 26f44d1 | 2012-08-17 23:55:05 +0200 | [diff] [blame] | 329 | if (fdtab[fd].iocb && fdtab[fd].owner && fdtab[fd].ev) |
| 330 | fdtab[fd].iocb(fd); |
Willy Tarreau | dbcd47e | 2012-05-13 09:42:26 +0200 | [diff] [blame] | 331 | |
Willy Tarreau | 037d2c1 | 2012-11-06 02:34:46 +0100 | [diff] [blame] | 332 | /* if the fd was removed from the spec list, it has been |
| 333 | * replaced by the next one that we don't want to skip ! |
| 334 | */ |
Willy Tarreau | 7be79a4 | 2012-11-11 15:02:54 +0100 | [diff] [blame^] | 335 | if (spec_idx < fd_nbspec && fd_spec[spec_idx] != fd) |
Willy Tarreau | dbcd47e | 2012-05-13 09:42:26 +0200 | [diff] [blame] | 336 | continue; |
| 337 | |
Willy Tarreau | 037d2c1 | 2012-11-06 02:34:46 +0100 | [diff] [blame] | 338 | spec_idx++; |
Willy Tarreau | cb65125 | 2008-08-29 13:57:30 +0200 | [diff] [blame] | 339 | } |
Willy Tarreau | dbcd47e | 2012-05-13 09:42:26 +0200 | [diff] [blame] | 340 | |
Willy Tarreau | 037d2c1 | 2012-11-06 02:34:46 +0100 | [diff] [blame] | 341 | in_poll_loop = 0; |
Willy Tarreau | dbcd47e | 2012-05-13 09:42:26 +0200 | [diff] [blame] | 342 | /* in the end, we have processed status + spec_processed FDs */ |
Willy Tarreau | de99e99 | 2007-04-16 00:53:59 +0200 | [diff] [blame] | 343 | } |
| 344 | |
| 345 | /* |
| 346 | * Initialization of the speculative epoll() poller. |
| 347 | * Returns 0 in case of failure, non-zero in case of success. If it fails, it |
| 348 | * disables the poller by setting its pref to 0. |
| 349 | */ |
| 350 | REGPRM1 static int _do_init(struct poller *p) |
| 351 | { |
Willy Tarreau | de99e99 | 2007-04-16 00:53:59 +0200 | [diff] [blame] | 352 | p->private = NULL; |
| 353 | |
| 354 | epoll_fd = epoll_create(global.maxsock + 1); |
| 355 | if (epoll_fd < 0) |
| 356 | goto fail_fd; |
| 357 | |
Willy Tarreau | f2e8ee2 | 2008-05-25 10:39:02 +0200 | [diff] [blame] | 358 | /* See comments at the top of the file about this formula. */ |
Willy Tarreau | 037d2c1 | 2012-11-06 02:34:46 +0100 | [diff] [blame] | 359 | absmaxevents = MAX(global.tune.maxpollevents, global.maxsock); |
Willy Tarreau | de99e99 | 2007-04-16 00:53:59 +0200 | [diff] [blame] | 360 | epoll_events = (struct epoll_event*) |
Willy Tarreau | f2e8ee2 | 2008-05-25 10:39:02 +0200 | [diff] [blame] | 361 | calloc(1, sizeof(struct epoll_event) * absmaxevents); |
Willy Tarreau | de99e99 | 2007-04-16 00:53:59 +0200 | [diff] [blame] | 362 | |
| 363 | if (epoll_events == NULL) |
| 364 | goto fail_ee; |
| 365 | |
Willy Tarreau | de99e99 | 2007-04-16 00:53:59 +0200 | [diff] [blame] | 366 | return 1; |
| 367 | |
Willy Tarreau | de99e99 | 2007-04-16 00:53:59 +0200 | [diff] [blame] | 368 | fail_ee: |
| 369 | close(epoll_fd); |
Willy Tarreau | d79e79b | 2009-05-10 10:18:54 +0200 | [diff] [blame] | 370 | epoll_fd = -1; |
Willy Tarreau | de99e99 | 2007-04-16 00:53:59 +0200 | [diff] [blame] | 371 | fail_fd: |
| 372 | p->pref = 0; |
| 373 | return 0; |
| 374 | } |
| 375 | |
| 376 | /* |
| 377 | * Termination of the speculative epoll() poller. |
| 378 | * Memory is released and the poller is marked as unselectable. |
| 379 | */ |
| 380 | REGPRM1 static void _do_term(struct poller *p) |
| 381 | { |
Willy Tarreau | a534fea | 2008-08-03 12:19:50 +0200 | [diff] [blame] | 382 | free(epoll_events); |
Willy Tarreau | de99e99 | 2007-04-16 00:53:59 +0200 | [diff] [blame] | 383 | |
Willy Tarreau | d79e79b | 2009-05-10 10:18:54 +0200 | [diff] [blame] | 384 | if (epoll_fd >= 0) { |
| 385 | close(epoll_fd); |
| 386 | epoll_fd = -1; |
| 387 | } |
Willy Tarreau | de99e99 | 2007-04-16 00:53:59 +0200 | [diff] [blame] | 388 | |
Willy Tarreau | de99e99 | 2007-04-16 00:53:59 +0200 | [diff] [blame] | 389 | epoll_events = NULL; |
Willy Tarreau | de99e99 | 2007-04-16 00:53:59 +0200 | [diff] [blame] | 390 | p->private = NULL; |
| 391 | p->pref = 0; |
| 392 | } |
| 393 | |
| 394 | /* |
| 395 | * Check that the poller works. |
| 396 | * Returns 1 if OK, otherwise 0. |
| 397 | */ |
| 398 | REGPRM1 static int _do_test(struct poller *p) |
| 399 | { |
| 400 | int fd; |
| 401 | |
| 402 | fd = epoll_create(global.maxsock + 1); |
| 403 | if (fd < 0) |
| 404 | return 0; |
| 405 | close(fd); |
| 406 | return 1; |
| 407 | } |
| 408 | |
| 409 | /* |
Willy Tarreau | fb8983f | 2007-06-03 16:40:44 +0200 | [diff] [blame] | 410 | * Recreate the epoll file descriptor after a fork(). Returns 1 if OK, |
| 411 | * otherwise 0. It will ensure that all processes will not share their |
| 412 | * epoll_fd. Some side effects were encountered because of this, such |
| 413 | * as epoll_wait() returning an FD which was previously deleted. |
| 414 | */ |
| 415 | REGPRM1 static int _do_fork(struct poller *p) |
| 416 | { |
Willy Tarreau | d79e79b | 2009-05-10 10:18:54 +0200 | [diff] [blame] | 417 | if (epoll_fd >= 0) |
| 418 | close(epoll_fd); |
Willy Tarreau | fb8983f | 2007-06-03 16:40:44 +0200 | [diff] [blame] | 419 | epoll_fd = epoll_create(global.maxsock + 1); |
| 420 | if (epoll_fd < 0) |
| 421 | return 0; |
| 422 | return 1; |
| 423 | } |
| 424 | |
| 425 | /* |
Willy Tarreau | de99e99 | 2007-04-16 00:53:59 +0200 | [diff] [blame] | 426 | * It is a constructor, which means that it will automatically be called before |
| 427 | * main(). This is GCC-specific but it works at least since 2.95. |
| 428 | * Special care must be taken so that it does not need any uninitialized data. |
| 429 | */ |
| 430 | __attribute__((constructor)) |
| 431 | static void _do_register(void) |
| 432 | { |
| 433 | struct poller *p; |
| 434 | |
| 435 | if (nbpollers >= MAX_POLLERS) |
| 436 | return; |
Willy Tarreau | d79e79b | 2009-05-10 10:18:54 +0200 | [diff] [blame] | 437 | |
| 438 | epoll_fd = -1; |
Willy Tarreau | de99e99 | 2007-04-16 00:53:59 +0200 | [diff] [blame] | 439 | p = &pollers[nbpollers++]; |
| 440 | |
| 441 | p->name = "sepoll"; |
| 442 | p->pref = 400; |
| 443 | p->private = NULL; |
| 444 | |
| 445 | p->test = _do_test; |
| 446 | p->init = _do_init; |
| 447 | p->term = _do_term; |
| 448 | p->poll = _do_poll; |
Willy Tarreau | fb8983f | 2007-06-03 16:40:44 +0200 | [diff] [blame] | 449 | p->fork = _do_fork; |
Willy Tarreau | de99e99 | 2007-04-16 00:53:59 +0200 | [diff] [blame] | 450 | |
| 451 | p->is_set = __fd_is_set; |
Willy Tarreau | 3788e4c | 2012-07-30 14:29:35 +0200 | [diff] [blame] | 452 | p->set = __fd_set; |
Willy Tarreau | babd05a | 2012-08-09 12:14:03 +0200 | [diff] [blame] | 453 | p->wai = __fd_wai; |
Willy Tarreau | 3788e4c | 2012-07-30 14:29:35 +0200 | [diff] [blame] | 454 | p->clr = __fd_clr; |
Willy Tarreau | de99e99 | 2007-04-16 00:53:59 +0200 | [diff] [blame] | 455 | p->rem = __fd_rem; |
| 456 | p->clo = __fd_clo; |
| 457 | } |
| 458 | |
| 459 | |
| 460 | /* |
| 461 | * Local variables: |
| 462 | * c-indent-level: 8 |
| 463 | * c-basic-offset: 8 |
| 464 | * End: |
| 465 | */ |