Willy Tarreau | 4f60f16 | 2007-04-08 16:39:58 +0200 | [diff] [blame] | 1 | /* |
Willy Tarreau | e9f49e7 | 2012-11-11 17:42:00 +0100 | [diff] [blame] | 2 | * FD polling functions for Linux epoll |
Willy Tarreau | 4f60f16 | 2007-04-08 16:39:58 +0200 | [diff] [blame] | 3 | * |
Willy Tarreau | e9f49e7 | 2012-11-11 17:42:00 +0100 | [diff] [blame] | 4 | * Copyright 2000-2012 Willy Tarreau <w@1wt.eu> |
Willy Tarreau | 4f60f16 | 2007-04-08 16:39:58 +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 | 4f60f16 | 2007-04-08 16:39:58 +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 | e9f49e7 | 2012-11-11 17:42:00 +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 | 58094f2 | 2007-04-10 01:33:20 +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 | 4f60f16 | 2007-04-08 16:39:58 +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 | 4f60f16 | 2007-04-08 16:39:58 +0200 | [diff] [blame] | 24 | |
Willy Tarreau | 4f60f16 | 2007-04-08 16:39:58 +0200 | [diff] [blame] | 25 | #include <types/global.h> |
| 26 | |
Willy Tarreau | e9f49e7 | 2012-11-11 17:42:00 +0100 | [diff] [blame] | 27 | #include <proto/fd.h> |
Willy Tarreau | 332740d | 2009-05-10 09:57:21 +0200 | [diff] [blame] | 28 | #include <proto/signal.h> |
Willy Tarreau | 4f60f16 | 2007-04-08 16:39:58 +0200 | [diff] [blame] | 29 | #include <proto/task.h> |
| 30 | |
Willy Tarreau | 58094f2 | 2007-04-10 01:33:20 +0200 | [diff] [blame] | 31 | |
Willy Tarreau | e9f49e7 | 2012-11-11 17:42:00 +0100 | [diff] [blame] | 32 | static int absmaxevents = 0; // absolute maximum amounts of polled events |
Willy Tarreau | 4f60f16 | 2007-04-08 16:39:58 +0200 | [diff] [blame] | 33 | |
| 34 | /* private data */ |
| 35 | static struct epoll_event *epoll_events; |
| 36 | static int epoll_fd; |
| 37 | |
Willy Tarreau | 58094f2 | 2007-04-10 01:33:20 +0200 | [diff] [blame] | 38 | /* This structure may be used for any purpose. Warning! do not use it in |
| 39 | * recursive functions ! |
| 40 | */ |
| 41 | static struct epoll_event ev; |
| 42 | |
Willy Tarreau | 1c07b07 | 2013-01-07 16:19:18 +0100 | [diff] [blame] | 43 | #ifndef EPOLLRDHUP |
| 44 | /* EPOLLRDHUP was defined late in libc, and it appeared in kernel 2.6.17 */ |
| 45 | #define EPOLLRDHUP 0x2000 |
| 46 | #endif |
| 47 | |
Willy Tarreau | 4f60f16 | 2007-04-08 16:39:58 +0200 | [diff] [blame] | 48 | /* |
Willy Tarreau | e9f49e7 | 2012-11-11 17:42:00 +0100 | [diff] [blame] | 49 | * speculative epoll() poller |
Willy Tarreau | 4f60f16 | 2007-04-08 16:39:58 +0200 | [diff] [blame] | 50 | */ |
Willy Tarreau | e9f49e7 | 2012-11-11 17:42:00 +0100 | [diff] [blame] | 51 | REGPRM2 static void _do_poll(struct poller *p, int exp) |
Willy Tarreau | 4f60f16 | 2007-04-08 16:39:58 +0200 | [diff] [blame] | 52 | { |
Willy Tarreau | e9f49e7 | 2012-11-11 17:42:00 +0100 | [diff] [blame] | 53 | int status, eo, en; |
| 54 | int fd, opcode; |
| 55 | int count; |
| 56 | int updt_idx; |
| 57 | int wait_time; |
Willy Tarreau | 58094f2 | 2007-04-10 01:33:20 +0200 | [diff] [blame] | 58 | |
Willy Tarreau | e9f49e7 | 2012-11-11 17:42:00 +0100 | [diff] [blame] | 59 | /* first, scan the update list to find changes */ |
| 60 | for (updt_idx = 0; updt_idx < fd_nbupdt; updt_idx++) { |
| 61 | fd = fd_updt[updt_idx]; |
Willy Tarreau | 69a41fa | 2014-01-20 11:02:59 +0100 | [diff] [blame^] | 62 | en = fdtab[fd].state & 15; /* new events */ |
| 63 | eo = fdtab[fd].state >> 4; /* previous events */ |
Willy Tarreau | 58094f2 | 2007-04-10 01:33:20 +0200 | [diff] [blame] | 64 | |
Willy Tarreau | e9f49e7 | 2012-11-11 17:42:00 +0100 | [diff] [blame] | 65 | if (fdtab[fd].owner && (eo ^ en)) { |
| 66 | if ((eo ^ en) & FD_EV_POLLED_RW) { |
Willy Tarreau | 3ef5af3 | 2013-12-20 16:03:41 +0100 | [diff] [blame] | 67 | /* poll status changed */ |
Willy Tarreau | e9f49e7 | 2012-11-11 17:42:00 +0100 | [diff] [blame] | 68 | if ((en & FD_EV_POLLED_RW) == 0) { |
| 69 | /* fd removed from poll list */ |
| 70 | opcode = EPOLL_CTL_DEL; |
| 71 | } |
| 72 | else if ((eo & FD_EV_POLLED_RW) == 0) { |
| 73 | /* new fd in the poll list */ |
| 74 | opcode = EPOLL_CTL_ADD; |
| 75 | } |
| 76 | else { |
| 77 | /* fd status changed */ |
| 78 | opcode = EPOLL_CTL_MOD; |
| 79 | } |
Willy Tarreau | 58094f2 | 2007-04-10 01:33:20 +0200 | [diff] [blame] | 80 | |
Willy Tarreau | e9f49e7 | 2012-11-11 17:42:00 +0100 | [diff] [blame] | 81 | /* construct the epoll events based on new state */ |
| 82 | ev.events = 0; |
| 83 | if (en & FD_EV_POLLED_R) |
Willy Tarreau | 1c07b07 | 2013-01-07 16:19:18 +0100 | [diff] [blame] | 84 | ev.events |= EPOLLIN | EPOLLRDHUP; |
Willy Tarreau | 58094f2 | 2007-04-10 01:33:20 +0200 | [diff] [blame] | 85 | |
Willy Tarreau | e9f49e7 | 2012-11-11 17:42:00 +0100 | [diff] [blame] | 86 | if (en & FD_EV_POLLED_W) |
| 87 | ev.events |= EPOLLOUT; |
Willy Tarreau | 58094f2 | 2007-04-10 01:33:20 +0200 | [diff] [blame] | 88 | |
Willy Tarreau | e9f49e7 | 2012-11-11 17:42:00 +0100 | [diff] [blame] | 89 | ev.data.fd = fd; |
| 90 | epoll_ctl(epoll_fd, opcode, fd, &ev); |
Willy Tarreau | 58094f2 | 2007-04-10 01:33:20 +0200 | [diff] [blame] | 91 | } |
Willy Tarreau | 58094f2 | 2007-04-10 01:33:20 +0200 | [diff] [blame] | 92 | |
Willy Tarreau | 69a41fa | 2014-01-20 11:02:59 +0100 | [diff] [blame^] | 93 | fdtab[fd].state = (en << 4) + en; /* save new events */ |
Willy Tarreau | f8cfa44 | 2012-10-04 21:54:41 +0200 | [diff] [blame] | 94 | |
Willy Tarreau | e9f49e7 | 2012-11-11 17:42:00 +0100 | [diff] [blame] | 95 | if (!(en & FD_EV_ACTIVE_RW)) { |
| 96 | /* This fd doesn't use any active entry anymore, we can |
| 97 | * kill its entry. |
| 98 | */ |
| 99 | release_spec_entry(fd); |
| 100 | } |
| 101 | else if ((en & ~eo) & FD_EV_ACTIVE_RW) { |
| 102 | /* we need a new spec entry now */ |
| 103 | alloc_spec_entry(fd); |
| 104 | } |
Willy Tarreau | 4f60f16 | 2007-04-08 16:39:58 +0200 | [diff] [blame] | 105 | |
Willy Tarreau | e9f49e7 | 2012-11-11 17:42:00 +0100 | [diff] [blame] | 106 | } |
| 107 | fdtab[fd].updated = 0; |
| 108 | fdtab[fd].new = 0; |
Willy Tarreau | 58094f2 | 2007-04-10 01:33:20 +0200 | [diff] [blame] | 109 | } |
Willy Tarreau | e9f49e7 | 2012-11-11 17:42:00 +0100 | [diff] [blame] | 110 | fd_nbupdt = 0; |
Willy Tarreau | 4f60f16 | 2007-04-08 16:39:58 +0200 | [diff] [blame] | 111 | |
Willy Tarreau | e9f49e7 | 2012-11-11 17:42:00 +0100 | [diff] [blame] | 112 | /* compute the epoll_wait() timeout */ |
Willy Tarreau | 4f60f16 | 2007-04-08 16:39:58 +0200 | [diff] [blame] | 113 | |
Willy Tarreau | e9f49e7 | 2012-11-11 17:42:00 +0100 | [diff] [blame] | 114 | if (fd_nbspec || run_queue || signal_queue_len) { |
| 115 | /* Maybe we still have events in the spec list, or there are |
| 116 | * some tasks left pending in the run_queue, so we must not |
| 117 | * wait in epoll() otherwise we would delay their delivery by |
| 118 | * the next timeout. |
| 119 | */ |
Willy Tarreau | bdefc51 | 2007-05-14 02:02:04 +0200 | [diff] [blame] | 120 | wait_time = 0; |
Willy Tarreau | e9f49e7 | 2012-11-11 17:42:00 +0100 | [diff] [blame] | 121 | } |
Willy Tarreau | b0b37bc | 2008-06-23 14:00:57 +0200 | [diff] [blame] | 122 | else { |
Willy Tarreau | e9f49e7 | 2012-11-11 17:42:00 +0100 | [diff] [blame] | 123 | if (!exp) |
Willy Tarreau | b0b37bc | 2008-06-23 14:00:57 +0200 | [diff] [blame] | 124 | wait_time = MAX_DELAY_MS; |
Willy Tarreau | e9f49e7 | 2012-11-11 17:42:00 +0100 | [diff] [blame] | 125 | else if (tick_is_expired(exp, now_ms)) |
| 126 | wait_time = 0; |
| 127 | else { |
| 128 | wait_time = TICKS_TO_MS(tick_remain(now_ms, exp)) + 1; |
| 129 | if (wait_time > MAX_DELAY_MS) |
| 130 | wait_time = MAX_DELAY_MS; |
| 131 | } |
Willy Tarreau | b0b37bc | 2008-06-23 14:00:57 +0200 | [diff] [blame] | 132 | } |
Willy Tarreau | d825eef | 2007-05-12 22:35:00 +0200 | [diff] [blame] | 133 | |
Willy Tarreau | e9f49e7 | 2012-11-11 17:42:00 +0100 | [diff] [blame] | 134 | /* now let's wait for polled events */ |
| 135 | |
Willy Tarreau | 45a1251 | 2011-09-10 16:56:42 +0200 | [diff] [blame] | 136 | gettimeofday(&before_poll, NULL); |
Willy Tarreau | cf181c9 | 2013-01-18 15:22:41 +0100 | [diff] [blame] | 137 | status = epoll_wait(epoll_fd, epoll_events, global.tune.maxpollevents, wait_time); |
Willy Tarreau | b0b37bc | 2008-06-23 14:00:57 +0200 | [diff] [blame] | 138 | tv_update_date(wait_time, status); |
Willy Tarreau | 45a1251 | 2011-09-10 16:56:42 +0200 | [diff] [blame] | 139 | measure_idle(); |
Willy Tarreau | 4f60f16 | 2007-04-08 16:39:58 +0200 | [diff] [blame] | 140 | |
Willy Tarreau | e9f49e7 | 2012-11-11 17:42:00 +0100 | [diff] [blame] | 141 | /* process polled events */ |
| 142 | |
Willy Tarreau | 4f60f16 | 2007-04-08 16:39:58 +0200 | [diff] [blame] | 143 | for (count = 0; count < status; count++) { |
Willy Tarreau | 1c07b07 | 2013-01-07 16:19:18 +0100 | [diff] [blame] | 144 | unsigned int n; |
| 145 | unsigned int e = epoll_events[count].events; |
Willy Tarreau | 4f60f16 | 2007-04-08 16:39:58 +0200 | [diff] [blame] | 146 | fd = epoll_events[count].data.fd; |
| 147 | |
Willy Tarreau | 076be25 | 2012-07-06 16:02:29 +0200 | [diff] [blame] | 148 | if (!fdtab[fd].owner) |
| 149 | continue; |
| 150 | |
Willy Tarreau | 491c498 | 2012-07-06 11:16:01 +0200 | [diff] [blame] | 151 | /* it looks complicated but gcc can optimize it away when constants |
Willy Tarreau | 462c720 | 2012-12-13 22:26:37 +0100 | [diff] [blame] | 152 | * have same values... In fact it depends on gcc :-( |
Willy Tarreau | 491c498 | 2012-07-06 11:16:01 +0200 | [diff] [blame] | 153 | */ |
| 154 | fdtab[fd].ev &= FD_POLL_STICKY; |
Willy Tarreau | 462c720 | 2012-12-13 22:26:37 +0100 | [diff] [blame] | 155 | if (EPOLLIN == FD_POLL_IN && EPOLLOUT == FD_POLL_OUT && |
| 156 | EPOLLPRI == FD_POLL_PRI && EPOLLERR == FD_POLL_ERR && |
| 157 | EPOLLHUP == FD_POLL_HUP) { |
Willy Tarreau | 6320c3c | 2012-12-13 23:52:58 +0100 | [diff] [blame] | 158 | n = e & (EPOLLIN|EPOLLOUT|EPOLLPRI|EPOLLERR|EPOLLHUP); |
Willy Tarreau | 462c720 | 2012-12-13 22:26:37 +0100 | [diff] [blame] | 159 | } |
| 160 | else { |
Willy Tarreau | 6320c3c | 2012-12-13 23:52:58 +0100 | [diff] [blame] | 161 | n = ((e & EPOLLIN ) ? FD_POLL_IN : 0) | |
Willy Tarreau | 462c720 | 2012-12-13 22:26:37 +0100 | [diff] [blame] | 162 | ((e & EPOLLPRI) ? FD_POLL_PRI : 0) | |
| 163 | ((e & EPOLLOUT) ? FD_POLL_OUT : 0) | |
| 164 | ((e & EPOLLERR) ? FD_POLL_ERR : 0) | |
| 165 | ((e & EPOLLHUP) ? FD_POLL_HUP : 0); |
| 166 | } |
Willy Tarreau | 491c498 | 2012-07-06 11:16:01 +0200 | [diff] [blame] | 167 | |
Willy Tarreau | 1c07b07 | 2013-01-07 16:19:18 +0100 | [diff] [blame] | 168 | /* always remap RDHUP to HUP as they're used similarly */ |
| 169 | if (e & EPOLLRDHUP) |
| 170 | n |= FD_POLL_HUP; |
| 171 | |
Willy Tarreau | 6320c3c | 2012-12-13 23:52:58 +0100 | [diff] [blame] | 172 | if (!n) |
| 173 | continue; |
| 174 | |
| 175 | fdtab[fd].ev |= n; |
| 176 | |
| 177 | if (fdtab[fd].iocb) { |
Willy Tarreau | fb5470d | 2012-12-14 00:02:33 +0100 | [diff] [blame] | 178 | int new_updt, old_updt; |
Willy Tarreau | e9f49e7 | 2012-11-11 17:42:00 +0100 | [diff] [blame] | 179 | |
| 180 | /* Mark the events as speculative before processing |
| 181 | * them so that if nothing can be done we don't need |
| 182 | * to poll again. |
| 183 | */ |
Willy Tarreau | 26d7cfc | 2012-12-07 00:09:43 +0100 | [diff] [blame] | 184 | if (fdtab[fd].ev & FD_POLL_IN) |
Willy Tarreau | e9f49e7 | 2012-11-11 17:42:00 +0100 | [diff] [blame] | 185 | fd_ev_set(fd, DIR_RD); |
| 186 | |
Willy Tarreau | 26d7cfc | 2012-12-07 00:09:43 +0100 | [diff] [blame] | 187 | if (fdtab[fd].ev & FD_POLL_OUT) |
Willy Tarreau | e9f49e7 | 2012-11-11 17:42:00 +0100 | [diff] [blame] | 188 | fd_ev_set(fd, DIR_WR); |
| 189 | |
Willy Tarreau | 39ebef8 | 2012-12-14 00:17:03 +0100 | [diff] [blame] | 190 | if (fdtab[fd].spec_p) { |
| 191 | /* This fd was already scheduled for being called as a speculative I/O */ |
| 192 | continue; |
| 193 | } |
| 194 | |
Willy Tarreau | fb5470d | 2012-12-14 00:02:33 +0100 | [diff] [blame] | 195 | /* Save number of updates to detect creation of new FDs. */ |
| 196 | old_updt = fd_nbupdt; |
Willy Tarreau | 9845e75 | 2012-07-06 11:44:28 +0200 | [diff] [blame] | 197 | fdtab[fd].iocb(fd); |
Willy Tarreau | e9f49e7 | 2012-11-11 17:42:00 +0100 | [diff] [blame] | 198 | |
| 199 | /* One or more fd might have been created during the iocb(). |
| 200 | * This mainly happens with new incoming connections that have |
| 201 | * just been accepted, so we'd like to process them immediately |
| 202 | * for better efficiency. Second benefit, if at the end the fds |
| 203 | * are disabled again, we can safely destroy their update entry |
| 204 | * to reduce the scope of later scans. This is the reason we |
| 205 | * scan the new entries backwards. |
| 206 | */ |
| 207 | |
| 208 | for (new_updt = fd_nbupdt; new_updt > old_updt; new_updt--) { |
| 209 | fd = fd_updt[new_updt - 1]; |
| 210 | if (!fdtab[fd].new) |
| 211 | continue; |
| 212 | |
| 213 | fdtab[fd].new = 0; |
| 214 | fdtab[fd].ev &= FD_POLL_STICKY; |
| 215 | |
Willy Tarreau | 69a41fa | 2014-01-20 11:02:59 +0100 | [diff] [blame^] | 216 | if ((fdtab[fd].state & FD_EV_STATUS_R) == FD_EV_ACTIVE_R) |
Willy Tarreau | e9f49e7 | 2012-11-11 17:42:00 +0100 | [diff] [blame] | 217 | fdtab[fd].ev |= FD_POLL_IN; |
| 218 | |
Willy Tarreau | 69a41fa | 2014-01-20 11:02:59 +0100 | [diff] [blame^] | 219 | if ((fdtab[fd].state & FD_EV_STATUS_W) == FD_EV_ACTIVE_W) |
Willy Tarreau | e9f49e7 | 2012-11-11 17:42:00 +0100 | [diff] [blame] | 220 | fdtab[fd].ev |= FD_POLL_OUT; |
| 221 | |
| 222 | if (fdtab[fd].ev && fdtab[fd].iocb && fdtab[fd].owner) |
| 223 | fdtab[fd].iocb(fd); |
| 224 | |
| 225 | /* we can remove this update entry if it's the last one and is |
| 226 | * unused, otherwise we don't touch anything. |
| 227 | */ |
Willy Tarreau | 69a41fa | 2014-01-20 11:02:59 +0100 | [diff] [blame^] | 228 | if (new_updt == fd_nbupdt && fdtab[fd].state == 0) { |
Willy Tarreau | e9f49e7 | 2012-11-11 17:42:00 +0100 | [diff] [blame] | 229 | fdtab[fd].updated = 0; |
| 230 | fd_nbupdt--; |
| 231 | } |
| 232 | } |
| 233 | } |
Willy Tarreau | 4f60f16 | 2007-04-08 16:39:58 +0200 | [diff] [blame] | 234 | } |
Willy Tarreau | e9f49e7 | 2012-11-11 17:42:00 +0100 | [diff] [blame] | 235 | |
| 236 | /* the caller will take care of speculative events */ |
Willy Tarreau | 4f60f16 | 2007-04-08 16:39:58 +0200 | [diff] [blame] | 237 | } |
| 238 | |
| 239 | /* |
Willy Tarreau | e9f49e7 | 2012-11-11 17:42:00 +0100 | [diff] [blame] | 240 | * Initialization of the speculative epoll() poller. |
Willy Tarreau | e54e917 | 2007-04-09 09:23:31 +0200 | [diff] [blame] | 241 | * Returns 0 in case of failure, non-zero in case of success. If it fails, it |
| 242 | * disables the poller by setting its pref to 0. |
| 243 | */ |
Willy Tarreau | ef1d1f8 | 2007-04-16 00:25:25 +0200 | [diff] [blame] | 244 | REGPRM1 static int _do_init(struct poller *p) |
Willy Tarreau | e54e917 | 2007-04-09 09:23:31 +0200 | [diff] [blame] | 245 | { |
Willy Tarreau | e54e917 | 2007-04-09 09:23:31 +0200 | [diff] [blame] | 246 | p->private = NULL; |
Willy Tarreau | e54e917 | 2007-04-09 09:23:31 +0200 | [diff] [blame] | 247 | |
| 248 | epoll_fd = epoll_create(global.maxsock + 1); |
| 249 | if (epoll_fd < 0) |
| 250 | goto fail_fd; |
| 251 | |
Willy Tarreau | e9f49e7 | 2012-11-11 17:42:00 +0100 | [diff] [blame] | 252 | /* See comments at the top of the file about this formula. */ |
| 253 | absmaxevents = MAX(global.tune.maxpollevents, global.maxsock); |
Willy Tarreau | e54e917 | 2007-04-09 09:23:31 +0200 | [diff] [blame] | 254 | epoll_events = (struct epoll_event*) |
Willy Tarreau | e9f49e7 | 2012-11-11 17:42:00 +0100 | [diff] [blame] | 255 | calloc(1, sizeof(struct epoll_event) * absmaxevents); |
Willy Tarreau | e54e917 | 2007-04-09 09:23:31 +0200 | [diff] [blame] | 256 | |
| 257 | if (epoll_events == NULL) |
| 258 | goto fail_ee; |
| 259 | |
Willy Tarreau | e54e917 | 2007-04-09 09:23:31 +0200 | [diff] [blame] | 260 | return 1; |
| 261 | |
Willy Tarreau | e54e917 | 2007-04-09 09:23:31 +0200 | [diff] [blame] | 262 | fail_ee: |
| 263 | close(epoll_fd); |
Willy Tarreau | d79e79b | 2009-05-10 10:18:54 +0200 | [diff] [blame] | 264 | epoll_fd = -1; |
Willy Tarreau | e54e917 | 2007-04-09 09:23:31 +0200 | [diff] [blame] | 265 | fail_fd: |
| 266 | p->pref = 0; |
| 267 | return 0; |
| 268 | } |
| 269 | |
| 270 | /* |
Willy Tarreau | e9f49e7 | 2012-11-11 17:42:00 +0100 | [diff] [blame] | 271 | * Termination of the speculative epoll() poller. |
Willy Tarreau | e54e917 | 2007-04-09 09:23:31 +0200 | [diff] [blame] | 272 | * Memory is released and the poller is marked as unselectable. |
| 273 | */ |
Willy Tarreau | ef1d1f8 | 2007-04-16 00:25:25 +0200 | [diff] [blame] | 274 | REGPRM1 static void _do_term(struct poller *p) |
Willy Tarreau | e54e917 | 2007-04-09 09:23:31 +0200 | [diff] [blame] | 275 | { |
Willy Tarreau | a534fea | 2008-08-03 12:19:50 +0200 | [diff] [blame] | 276 | free(epoll_events); |
Willy Tarreau | e54e917 | 2007-04-09 09:23:31 +0200 | [diff] [blame] | 277 | |
Willy Tarreau | d79e79b | 2009-05-10 10:18:54 +0200 | [diff] [blame] | 278 | if (epoll_fd >= 0) { |
| 279 | close(epoll_fd); |
| 280 | epoll_fd = -1; |
| 281 | } |
Willy Tarreau | e54e917 | 2007-04-09 09:23:31 +0200 | [diff] [blame] | 282 | |
Willy Tarreau | 58094f2 | 2007-04-10 01:33:20 +0200 | [diff] [blame] | 283 | epoll_events = NULL; |
Willy Tarreau | e54e917 | 2007-04-09 09:23:31 +0200 | [diff] [blame] | 284 | p->private = NULL; |
| 285 | p->pref = 0; |
| 286 | } |
| 287 | |
| 288 | /* |
Willy Tarreau | 2ff7622 | 2007-04-09 19:29:56 +0200 | [diff] [blame] | 289 | * Check that the poller works. |
| 290 | * Returns 1 if OK, otherwise 0. |
| 291 | */ |
Willy Tarreau | ef1d1f8 | 2007-04-16 00:25:25 +0200 | [diff] [blame] | 292 | REGPRM1 static int _do_test(struct poller *p) |
Willy Tarreau | 2ff7622 | 2007-04-09 19:29:56 +0200 | [diff] [blame] | 293 | { |
| 294 | int fd; |
| 295 | |
| 296 | fd = epoll_create(global.maxsock + 1); |
| 297 | if (fd < 0) |
| 298 | return 0; |
| 299 | close(fd); |
| 300 | return 1; |
| 301 | } |
| 302 | |
| 303 | /* |
Willy Tarreau | fb8983f | 2007-06-03 16:40:44 +0200 | [diff] [blame] | 304 | * Recreate the epoll file descriptor after a fork(). Returns 1 if OK, |
| 305 | * otherwise 0. It will ensure that all processes will not share their |
| 306 | * epoll_fd. Some side effects were encountered because of this, such |
| 307 | * as epoll_wait() returning an FD which was previously deleted. |
| 308 | */ |
| 309 | REGPRM1 static int _do_fork(struct poller *p) |
| 310 | { |
Willy Tarreau | d79e79b | 2009-05-10 10:18:54 +0200 | [diff] [blame] | 311 | if (epoll_fd >= 0) |
| 312 | close(epoll_fd); |
Willy Tarreau | fb8983f | 2007-06-03 16:40:44 +0200 | [diff] [blame] | 313 | epoll_fd = epoll_create(global.maxsock + 1); |
| 314 | if (epoll_fd < 0) |
| 315 | return 0; |
| 316 | return 1; |
| 317 | } |
| 318 | |
| 319 | /* |
Willy Tarreau | ef1d1f8 | 2007-04-16 00:25:25 +0200 | [diff] [blame] | 320 | * It is a constructor, which means that it will automatically be called before |
| 321 | * main(). This is GCC-specific but it works at least since 2.95. |
| 322 | * Special care must be taken so that it does not need any uninitialized data. |
Willy Tarreau | 4f60f16 | 2007-04-08 16:39:58 +0200 | [diff] [blame] | 323 | */ |
Willy Tarreau | ef1d1f8 | 2007-04-16 00:25:25 +0200 | [diff] [blame] | 324 | __attribute__((constructor)) |
| 325 | static void _do_register(void) |
Willy Tarreau | 4f60f16 | 2007-04-08 16:39:58 +0200 | [diff] [blame] | 326 | { |
Willy Tarreau | ef1d1f8 | 2007-04-16 00:25:25 +0200 | [diff] [blame] | 327 | struct poller *p; |
| 328 | |
| 329 | if (nbpollers >= MAX_POLLERS) |
| 330 | return; |
Willy Tarreau | d79e79b | 2009-05-10 10:18:54 +0200 | [diff] [blame] | 331 | |
| 332 | epoll_fd = -1; |
Willy Tarreau | ef1d1f8 | 2007-04-16 00:25:25 +0200 | [diff] [blame] | 333 | p = &pollers[nbpollers++]; |
| 334 | |
Willy Tarreau | 4f60f16 | 2007-04-08 16:39:58 +0200 | [diff] [blame] | 335 | p->name = "epoll"; |
| 336 | p->pref = 300; |
| 337 | p->private = NULL; |
| 338 | |
Willy Tarreau | 70c6fd8 | 2012-11-11 21:02:34 +0100 | [diff] [blame] | 339 | p->clo = NULL; |
Willy Tarreau | ef1d1f8 | 2007-04-16 00:25:25 +0200 | [diff] [blame] | 340 | p->test = _do_test; |
| 341 | p->init = _do_init; |
| 342 | p->term = _do_term; |
| 343 | p->poll = _do_poll; |
Willy Tarreau | fb8983f | 2007-06-03 16:40:44 +0200 | [diff] [blame] | 344 | p->fork = _do_fork; |
Willy Tarreau | 4f60f16 | 2007-04-08 16:39:58 +0200 | [diff] [blame] | 345 | } |
| 346 | |
| 347 | |
| 348 | /* |
| 349 | * Local variables: |
| 350 | * c-indent-level: 8 |
| 351 | * c-basic-offset: 8 |
| 352 | * End: |
| 353 | */ |