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 | f817e9f | 2014-01-10 16:58:45 +0100 | [diff] [blame] | 4 | * Copyright 2000-2014 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 | 4f60f16 | 2007-04-08 16:39:58 +0200 | [diff] [blame] | 28 | |
Willy Tarreau | 58094f2 | 2007-04-10 01:33:20 +0200 | [diff] [blame] | 29 | |
Willy Tarreau | 4f60f16 | 2007-04-08 16:39:58 +0200 | [diff] [blame] | 30 | /* private data */ |
Christopher Faulet | d4604ad | 2017-05-29 10:40:41 +0200 | [diff] [blame] | 31 | static THREAD_LOCAL struct epoll_event *epoll_events = NULL; |
Willy Tarreau | d9e7e36 | 2018-01-18 19:16:02 +0100 | [diff] [blame] | 32 | static int epoll_fd[MAX_THREADS]; // per-thread epoll_fd |
Willy Tarreau | 4f60f16 | 2007-04-08 16:39:58 +0200 | [diff] [blame] | 33 | |
Willy Tarreau | 58094f2 | 2007-04-10 01:33:20 +0200 | [diff] [blame] | 34 | /* This structure may be used for any purpose. Warning! do not use it in |
| 35 | * recursive functions ! |
| 36 | */ |
Christopher Faulet | d4604ad | 2017-05-29 10:40:41 +0200 | [diff] [blame] | 37 | static THREAD_LOCAL struct epoll_event ev; |
Willy Tarreau | 58094f2 | 2007-04-10 01:33:20 +0200 | [diff] [blame] | 38 | |
Willy Tarreau | 1c07b07 | 2013-01-07 16:19:18 +0100 | [diff] [blame] | 39 | #ifndef EPOLLRDHUP |
| 40 | /* EPOLLRDHUP was defined late in libc, and it appeared in kernel 2.6.17 */ |
| 41 | #define EPOLLRDHUP 0x2000 |
| 42 | #endif |
| 43 | |
Willy Tarreau | 4f60f16 | 2007-04-08 16:39:58 +0200 | [diff] [blame] | 44 | /* |
Conrad Hoffmann | 041751c | 2014-05-20 14:28:24 +0200 | [diff] [blame] | 45 | * Immediately remove file descriptor from epoll set upon close. |
| 46 | * Since we forked, some fds share inodes with the other process, and epoll may |
| 47 | * send us events even though this process closed the fd (see man 7 epoll, |
| 48 | * "Questions and answers", Q 6). |
| 49 | */ |
| 50 | REGPRM1 static void __fd_clo(int fd) |
| 51 | { |
Willy Tarreau | d9e7e36 | 2018-01-18 19:16:02 +0100 | [diff] [blame] | 52 | if (unlikely(fdtab[fd].cloned)) { |
Willy Tarreau | 4979592 | 2018-01-31 09:49:29 +0100 | [diff] [blame] | 53 | unsigned long m = fdtab[fd].polled_mask; |
Willy Tarreau | d9e7e36 | 2018-01-18 19:16:02 +0100 | [diff] [blame] | 54 | int i; |
| 55 | |
| 56 | for (i = global.nbthread - 1; i >= 0; i--) |
| 57 | if (m & (1UL << i)) |
| 58 | epoll_ctl(epoll_fd[i], EPOLL_CTL_DEL, fd, &ev); |
| 59 | } |
Conrad Hoffmann | 041751c | 2014-05-20 14:28:24 +0200 | [diff] [blame] | 60 | } |
| 61 | |
| 62 | /* |
Willy Tarreau | f817e9f | 2014-01-10 16:58:45 +0100 | [diff] [blame] | 63 | * Linux epoll() poller |
Willy Tarreau | 4f60f16 | 2007-04-08 16:39:58 +0200 | [diff] [blame] | 64 | */ |
Willy Tarreau | e9f49e7 | 2012-11-11 17:42:00 +0100 | [diff] [blame] | 65 | REGPRM2 static void _do_poll(struct poller *p, int exp) |
Willy Tarreau | 4f60f16 | 2007-04-08 16:39:58 +0200 | [diff] [blame] | 66 | { |
Willy Tarreau | 038e54c | 2018-01-17 21:51:21 +0100 | [diff] [blame] | 67 | int status, en; |
Willy Tarreau | e9f49e7 | 2012-11-11 17:42:00 +0100 | [diff] [blame] | 68 | int fd, opcode; |
| 69 | int count; |
| 70 | int updt_idx; |
| 71 | int wait_time; |
Willy Tarreau | 58094f2 | 2007-04-10 01:33:20 +0200 | [diff] [blame] | 72 | |
Willy Tarreau | 5be2f35 | 2014-11-19 19:43:05 +0100 | [diff] [blame] | 73 | /* first, scan the update list to find polling changes */ |
Willy Tarreau | e9f49e7 | 2012-11-11 17:42:00 +0100 | [diff] [blame] | 74 | for (updt_idx = 0; updt_idx < fd_nbupdt; updt_idx++) { |
| 75 | fd = fd_updt[updt_idx]; |
Willy Tarreau | 58094f2 | 2007-04-10 01:33:20 +0200 | [diff] [blame] | 76 | |
Willy Tarreau | d80cb4e | 2018-01-20 19:30:13 +0100 | [diff] [blame] | 77 | if (!fdtab[fd].owner) { |
| 78 | activity[tid].poll_drop++; |
Willy Tarreau | f817e9f | 2014-01-10 16:58:45 +0100 | [diff] [blame] | 79 | continue; |
Willy Tarreau | d80cb4e | 2018-01-20 19:30:13 +0100 | [diff] [blame] | 80 | } |
Willy Tarreau | 58094f2 | 2007-04-10 01:33:20 +0200 | [diff] [blame] | 81 | |
Christopher Faulet | 2a944ee | 2017-11-07 10:42:54 +0100 | [diff] [blame] | 82 | HA_SPIN_LOCK(FD_LOCK, &fdtab[fd].lock); |
Willy Tarreau | ebc78d7 | 2018-01-20 23:53:50 +0100 | [diff] [blame] | 83 | fdtab[fd].update_mask &= ~tid_bit; |
Willy Tarreau | 038e54c | 2018-01-17 21:51:21 +0100 | [diff] [blame] | 84 | en = fdtab[fd].state; |
Christopher Faulet | 2a944ee | 2017-11-07 10:42:54 +0100 | [diff] [blame] | 85 | HA_SPIN_UNLOCK(FD_LOCK, &fdtab[fd].lock); |
Willy Tarreau | 58094f2 | 2007-04-10 01:33:20 +0200 | [diff] [blame] | 86 | |
Willy Tarreau | d9e7e36 | 2018-01-18 19:16:02 +0100 | [diff] [blame] | 87 | if (fdtab[fd].polled_mask & tid_bit) { |
| 88 | if (!(fdtab[fd].thread_mask & tid_bit) || !(en & FD_EV_POLLED_RW)) { |
Willy Tarreau | f817e9f | 2014-01-10 16:58:45 +0100 | [diff] [blame] | 89 | /* fd removed from poll list */ |
| 90 | opcode = EPOLL_CTL_DEL; |
Willy Tarreau | d9e7e36 | 2018-01-18 19:16:02 +0100 | [diff] [blame] | 91 | HA_ATOMIC_AND(&fdtab[fd].polled_mask, ~tid_bit); |
Willy Tarreau | e9f49e7 | 2012-11-11 17:42:00 +0100 | [diff] [blame] | 92 | } |
Willy Tarreau | f817e9f | 2014-01-10 16:58:45 +0100 | [diff] [blame] | 93 | else { |
| 94 | /* fd status changed */ |
| 95 | opcode = EPOLL_CTL_MOD; |
| 96 | } |
Willy Tarreau | d9e7e36 | 2018-01-18 19:16:02 +0100 | [diff] [blame] | 97 | } |
| 98 | else if ((fdtab[fd].thread_mask & tid_bit) && (en & FD_EV_POLLED_RW)) { |
| 99 | /* new fd in the poll list */ |
| 100 | opcode = EPOLL_CTL_ADD; |
| 101 | HA_ATOMIC_OR(&fdtab[fd].polled_mask, tid_bit); |
| 102 | } |
| 103 | else { |
| 104 | continue; |
| 105 | } |
Willy Tarreau | f817e9f | 2014-01-10 16:58:45 +0100 | [diff] [blame] | 106 | |
Willy Tarreau | d9e7e36 | 2018-01-18 19:16:02 +0100 | [diff] [blame] | 107 | /* construct the epoll events based on new state */ |
| 108 | ev.events = 0; |
| 109 | if (en & FD_EV_POLLED_R) |
| 110 | ev.events |= EPOLLIN | EPOLLRDHUP; |
Willy Tarreau | f817e9f | 2014-01-10 16:58:45 +0100 | [diff] [blame] | 111 | |
Willy Tarreau | d9e7e36 | 2018-01-18 19:16:02 +0100 | [diff] [blame] | 112 | if (en & FD_EV_POLLED_W) |
| 113 | ev.events |= EPOLLOUT; |
Christopher Faulet | d4604ad | 2017-05-29 10:40:41 +0200 | [diff] [blame] | 114 | |
Willy Tarreau | d9e7e36 | 2018-01-18 19:16:02 +0100 | [diff] [blame] | 115 | ev.data.fd = fd; |
| 116 | epoll_ctl(epoll_fd[tid], opcode, fd, &ev); |
Willy Tarreau | 58094f2 | 2007-04-10 01:33:20 +0200 | [diff] [blame] | 117 | } |
Willy Tarreau | e9f49e7 | 2012-11-11 17:42:00 +0100 | [diff] [blame] | 118 | fd_nbupdt = 0; |
Willy Tarreau | 4f60f16 | 2007-04-08 16:39:58 +0200 | [diff] [blame] | 119 | |
Willy Tarreau | e9f49e7 | 2012-11-11 17:42:00 +0100 | [diff] [blame] | 120 | /* compute the epoll_wait() timeout */ |
Willy Tarreau | 10146c9 | 2015-04-13 20:44:19 +0200 | [diff] [blame] | 121 | if (!exp) |
| 122 | wait_time = MAX_DELAY_MS; |
Willy Tarreau | d80cb4e | 2018-01-20 19:30:13 +0100 | [diff] [blame] | 123 | else if (tick_is_expired(exp, now_ms)) { |
| 124 | activity[tid].poll_exp++; |
Willy Tarreau | bdefc51 | 2007-05-14 02:02:04 +0200 | [diff] [blame] | 125 | wait_time = 0; |
Willy Tarreau | d80cb4e | 2018-01-20 19:30:13 +0100 | [diff] [blame] | 126 | } |
Willy Tarreau | b0b37bc | 2008-06-23 14:00:57 +0200 | [diff] [blame] | 127 | else { |
Willy Tarreau | 10146c9 | 2015-04-13 20:44:19 +0200 | [diff] [blame] | 128 | wait_time = TICKS_TO_MS(tick_remain(now_ms, exp)) + 1; |
| 129 | if (wait_time > MAX_DELAY_MS) |
Willy Tarreau | b0b37bc | 2008-06-23 14:00:57 +0200 | [diff] [blame] | 130 | wait_time = MAX_DELAY_MS; |
| 131 | } |
Willy Tarreau | d825eef | 2007-05-12 22:35:00 +0200 | [diff] [blame] | 132 | |
Willy Tarreau | e9f49e7 | 2012-11-11 17:42:00 +0100 | [diff] [blame] | 133 | /* now let's wait for polled events */ |
| 134 | |
Willy Tarreau | 45a1251 | 2011-09-10 16:56:42 +0200 | [diff] [blame] | 135 | gettimeofday(&before_poll, NULL); |
Willy Tarreau | d9e7e36 | 2018-01-18 19:16:02 +0100 | [diff] [blame] | 136 | status = epoll_wait(epoll_fd[tid], epoll_events, global.tune.maxpollevents, wait_time); |
Willy Tarreau | b0b37bc | 2008-06-23 14:00:57 +0200 | [diff] [blame] | 137 | tv_update_date(wait_time, status); |
Willy Tarreau | 45a1251 | 2011-09-10 16:56:42 +0200 | [diff] [blame] | 138 | measure_idle(); |
Willy Tarreau | 4f60f16 | 2007-04-08 16:39:58 +0200 | [diff] [blame] | 139 | |
Willy Tarreau | e9f49e7 | 2012-11-11 17:42:00 +0100 | [diff] [blame] | 140 | /* process polled events */ |
| 141 | |
Willy Tarreau | 4f60f16 | 2007-04-08 16:39:58 +0200 | [diff] [blame] | 142 | for (count = 0; count < status; count++) { |
Willy Tarreau | 1c07b07 | 2013-01-07 16:19:18 +0100 | [diff] [blame] | 143 | unsigned int n; |
| 144 | unsigned int e = epoll_events[count].events; |
Willy Tarreau | 4f60f16 | 2007-04-08 16:39:58 +0200 | [diff] [blame] | 145 | fd = epoll_events[count].data.fd; |
| 146 | |
Willy Tarreau | d80cb4e | 2018-01-20 19:30:13 +0100 | [diff] [blame] | 147 | if (!fdtab[fd].owner) { |
| 148 | activity[tid].poll_dead++; |
| 149 | continue; |
| 150 | } |
| 151 | |
| 152 | if (!(fdtab[fd].thread_mask & tid_bit)) { |
Willy Tarreau | d9e7e36 | 2018-01-18 19:16:02 +0100 | [diff] [blame] | 153 | /* FD has been migrated */ |
Willy Tarreau | d80cb4e | 2018-01-20 19:30:13 +0100 | [diff] [blame] | 154 | activity[tid].poll_skip++; |
Willy Tarreau | d9e7e36 | 2018-01-18 19:16:02 +0100 | [diff] [blame] | 155 | epoll_ctl(epoll_fd[tid], EPOLL_CTL_DEL, fd, &ev); |
| 156 | HA_ATOMIC_AND(&fdtab[fd].polled_mask, ~tid_bit); |
Willy Tarreau | 076be25 | 2012-07-06 16:02:29 +0200 | [diff] [blame] | 157 | continue; |
Willy Tarreau | d80cb4e | 2018-01-20 19:30:13 +0100 | [diff] [blame] | 158 | } |
Willy Tarreau | 076be25 | 2012-07-06 16:02:29 +0200 | [diff] [blame] | 159 | |
Willy Tarreau | 491c498 | 2012-07-06 11:16:01 +0200 | [diff] [blame] | 160 | /* it looks complicated but gcc can optimize it away when constants |
Willy Tarreau | 462c720 | 2012-12-13 22:26:37 +0100 | [diff] [blame] | 161 | * have same values... In fact it depends on gcc :-( |
Willy Tarreau | 491c498 | 2012-07-06 11:16:01 +0200 | [diff] [blame] | 162 | */ |
Willy Tarreau | 462c720 | 2012-12-13 22:26:37 +0100 | [diff] [blame] | 163 | if (EPOLLIN == FD_POLL_IN && EPOLLOUT == FD_POLL_OUT && |
| 164 | EPOLLPRI == FD_POLL_PRI && EPOLLERR == FD_POLL_ERR && |
| 165 | EPOLLHUP == FD_POLL_HUP) { |
Willy Tarreau | 6320c3c | 2012-12-13 23:52:58 +0100 | [diff] [blame] | 166 | n = e & (EPOLLIN|EPOLLOUT|EPOLLPRI|EPOLLERR|EPOLLHUP); |
Willy Tarreau | 462c720 | 2012-12-13 22:26:37 +0100 | [diff] [blame] | 167 | } |
| 168 | else { |
Willy Tarreau | 6320c3c | 2012-12-13 23:52:58 +0100 | [diff] [blame] | 169 | n = ((e & EPOLLIN ) ? FD_POLL_IN : 0) | |
Willy Tarreau | 462c720 | 2012-12-13 22:26:37 +0100 | [diff] [blame] | 170 | ((e & EPOLLPRI) ? FD_POLL_PRI : 0) | |
| 171 | ((e & EPOLLOUT) ? FD_POLL_OUT : 0) | |
| 172 | ((e & EPOLLERR) ? FD_POLL_ERR : 0) | |
| 173 | ((e & EPOLLHUP) ? FD_POLL_HUP : 0); |
| 174 | } |
Willy Tarreau | 491c498 | 2012-07-06 11:16:01 +0200 | [diff] [blame] | 175 | |
Willy Tarreau | 1c07b07 | 2013-01-07 16:19:18 +0100 | [diff] [blame] | 176 | /* always remap RDHUP to HUP as they're used similarly */ |
Willy Tarreau | 5a76769 | 2017-03-13 11:38:28 +0100 | [diff] [blame] | 177 | if (e & EPOLLRDHUP) { |
Christopher Faulet | d4604ad | 2017-05-29 10:40:41 +0200 | [diff] [blame] | 178 | HA_ATOMIC_OR(&cur_poller.flags, HAP_POLL_F_RDHUP); |
Willy Tarreau | 1c07b07 | 2013-01-07 16:19:18 +0100 | [diff] [blame] | 179 | n |= FD_POLL_HUP; |
Willy Tarreau | 5a76769 | 2017-03-13 11:38:28 +0100 | [diff] [blame] | 180 | } |
Christopher Faulet | ab62f51 | 2017-08-30 10:34:36 +0200 | [diff] [blame] | 181 | fd_update_events(fd, n); |
Willy Tarreau | 4f60f16 | 2007-04-08 16:39:58 +0200 | [diff] [blame] | 182 | } |
Willy Tarreau | f817e9f | 2014-01-10 16:58:45 +0100 | [diff] [blame] | 183 | /* the caller will take care of cached events */ |
Willy Tarreau | 4f60f16 | 2007-04-08 16:39:58 +0200 | [diff] [blame] | 184 | } |
| 185 | |
Christopher Faulet | d4604ad | 2017-05-29 10:40:41 +0200 | [diff] [blame] | 186 | static int init_epoll_per_thread() |
| 187 | { |
Willy Tarreau | d9e7e36 | 2018-01-18 19:16:02 +0100 | [diff] [blame] | 188 | int fd; |
| 189 | |
Christopher Faulet | d4604ad | 2017-05-29 10:40:41 +0200 | [diff] [blame] | 190 | epoll_events = calloc(1, sizeof(struct epoll_event) * global.tune.maxpollevents); |
| 191 | if (epoll_events == NULL) |
Willy Tarreau | d9e7e36 | 2018-01-18 19:16:02 +0100 | [diff] [blame] | 192 | goto fail_alloc; |
| 193 | |
Christopher Faulet | 3e805ed | 2018-01-25 16:18:09 +0100 | [diff] [blame] | 194 | if (MAX_THREADS > 1 && tid) { |
Willy Tarreau | d9e7e36 | 2018-01-18 19:16:02 +0100 | [diff] [blame] | 195 | epoll_fd[tid] = epoll_create(global.maxsock + 1); |
| 196 | if (epoll_fd[tid] < 0) |
| 197 | goto fail_fd; |
| 198 | } |
| 199 | |
| 200 | /* we may have to unregister some events initially registered on the |
| 201 | * original fd when it was alone, and/or to register events on the new |
| 202 | * fd for this thread. Let's just mark them as updated, the poller will |
| 203 | * do the rest. |
| 204 | */ |
Willy Tarreau | ce036bc | 2018-01-29 14:58:02 +0100 | [diff] [blame] | 205 | for (fd = 0; fd < global.maxsock; fd++) |
Willy Tarreau | d9e7e36 | 2018-01-18 19:16:02 +0100 | [diff] [blame] | 206 | updt_fd_polling(fd); |
| 207 | |
Christopher Faulet | d4604ad | 2017-05-29 10:40:41 +0200 | [diff] [blame] | 208 | return 1; |
Willy Tarreau | d9e7e36 | 2018-01-18 19:16:02 +0100 | [diff] [blame] | 209 | fail_fd: |
| 210 | free(epoll_events); |
| 211 | fail_alloc: |
| 212 | return 0; |
Christopher Faulet | d4604ad | 2017-05-29 10:40:41 +0200 | [diff] [blame] | 213 | } |
| 214 | |
| 215 | static void deinit_epoll_per_thread() |
| 216 | { |
Christopher Faulet | 3e805ed | 2018-01-25 16:18:09 +0100 | [diff] [blame] | 217 | if (MAX_THREADS > 1 && tid) |
Willy Tarreau | d9e7e36 | 2018-01-18 19:16:02 +0100 | [diff] [blame] | 218 | close(epoll_fd[tid]); |
| 219 | |
Christopher Faulet | d4604ad | 2017-05-29 10:40:41 +0200 | [diff] [blame] | 220 | free(epoll_events); |
Christopher Faulet | cd7879a | 2017-10-27 13:53:47 +0200 | [diff] [blame] | 221 | epoll_events = NULL; |
Christopher Faulet | d4604ad | 2017-05-29 10:40:41 +0200 | [diff] [blame] | 222 | } |
| 223 | |
Willy Tarreau | 4f60f16 | 2007-04-08 16:39:58 +0200 | [diff] [blame] | 224 | /* |
Willy Tarreau | f817e9f | 2014-01-10 16:58:45 +0100 | [diff] [blame] | 225 | * Initialization of the epoll() poller. |
Willy Tarreau | e54e917 | 2007-04-09 09:23:31 +0200 | [diff] [blame] | 226 | * Returns 0 in case of failure, non-zero in case of success. If it fails, it |
| 227 | * disables the poller by setting its pref to 0. |
| 228 | */ |
Willy Tarreau | ef1d1f8 | 2007-04-16 00:25:25 +0200 | [diff] [blame] | 229 | REGPRM1 static int _do_init(struct poller *p) |
Willy Tarreau | e54e917 | 2007-04-09 09:23:31 +0200 | [diff] [blame] | 230 | { |
Willy Tarreau | e54e917 | 2007-04-09 09:23:31 +0200 | [diff] [blame] | 231 | p->private = NULL; |
Willy Tarreau | e54e917 | 2007-04-09 09:23:31 +0200 | [diff] [blame] | 232 | |
Willy Tarreau | d9e7e36 | 2018-01-18 19:16:02 +0100 | [diff] [blame] | 233 | epoll_fd[tid] = epoll_create(global.maxsock + 1); |
| 234 | if (epoll_fd[tid] < 0) |
Willy Tarreau | e54e917 | 2007-04-09 09:23:31 +0200 | [diff] [blame] | 235 | goto fail_fd; |
| 236 | |
Christopher Faulet | cd7879a | 2017-10-27 13:53:47 +0200 | [diff] [blame] | 237 | hap_register_per_thread_init(init_epoll_per_thread); |
| 238 | hap_register_per_thread_deinit(deinit_epoll_per_thread); |
Willy Tarreau | e54e917 | 2007-04-09 09:23:31 +0200 | [diff] [blame] | 239 | |
Willy Tarreau | e54e917 | 2007-04-09 09:23:31 +0200 | [diff] [blame] | 240 | return 1; |
| 241 | |
Willy Tarreau | e54e917 | 2007-04-09 09:23:31 +0200 | [diff] [blame] | 242 | fail_fd: |
| 243 | p->pref = 0; |
| 244 | return 0; |
| 245 | } |
| 246 | |
| 247 | /* |
Willy Tarreau | f817e9f | 2014-01-10 16:58:45 +0100 | [diff] [blame] | 248 | * Termination of the epoll() poller. |
Willy Tarreau | e54e917 | 2007-04-09 09:23:31 +0200 | [diff] [blame] | 249 | * Memory is released and the poller is marked as unselectable. |
| 250 | */ |
Willy Tarreau | ef1d1f8 | 2007-04-16 00:25:25 +0200 | [diff] [blame] | 251 | REGPRM1 static void _do_term(struct poller *p) |
Willy Tarreau | e54e917 | 2007-04-09 09:23:31 +0200 | [diff] [blame] | 252 | { |
Willy Tarreau | d9e7e36 | 2018-01-18 19:16:02 +0100 | [diff] [blame] | 253 | if (epoll_fd[tid] >= 0) { |
| 254 | close(epoll_fd[tid]); |
| 255 | epoll_fd[tid] = -1; |
Willy Tarreau | d79e79b | 2009-05-10 10:18:54 +0200 | [diff] [blame] | 256 | } |
Willy Tarreau | e54e917 | 2007-04-09 09:23:31 +0200 | [diff] [blame] | 257 | |
| 258 | p->private = NULL; |
| 259 | p->pref = 0; |
| 260 | } |
| 261 | |
| 262 | /* |
Willy Tarreau | 2ff7622 | 2007-04-09 19:29:56 +0200 | [diff] [blame] | 263 | * Check that the poller works. |
| 264 | * Returns 1 if OK, otherwise 0. |
| 265 | */ |
Willy Tarreau | ef1d1f8 | 2007-04-16 00:25:25 +0200 | [diff] [blame] | 266 | REGPRM1 static int _do_test(struct poller *p) |
Willy Tarreau | 2ff7622 | 2007-04-09 19:29:56 +0200 | [diff] [blame] | 267 | { |
| 268 | int fd; |
| 269 | |
| 270 | fd = epoll_create(global.maxsock + 1); |
| 271 | if (fd < 0) |
| 272 | return 0; |
| 273 | close(fd); |
| 274 | return 1; |
| 275 | } |
| 276 | |
| 277 | /* |
Willy Tarreau | fb8983f | 2007-06-03 16:40:44 +0200 | [diff] [blame] | 278 | * Recreate the epoll file descriptor after a fork(). Returns 1 if OK, |
| 279 | * otherwise 0. It will ensure that all processes will not share their |
| 280 | * epoll_fd. Some side effects were encountered because of this, such |
| 281 | * as epoll_wait() returning an FD which was previously deleted. |
| 282 | */ |
| 283 | REGPRM1 static int _do_fork(struct poller *p) |
| 284 | { |
Willy Tarreau | d9e7e36 | 2018-01-18 19:16:02 +0100 | [diff] [blame] | 285 | if (epoll_fd[tid] >= 0) |
| 286 | close(epoll_fd[tid]); |
| 287 | epoll_fd[tid] = epoll_create(global.maxsock + 1); |
| 288 | if (epoll_fd[tid] < 0) |
Willy Tarreau | fb8983f | 2007-06-03 16:40:44 +0200 | [diff] [blame] | 289 | return 0; |
| 290 | return 1; |
| 291 | } |
| 292 | |
| 293 | /* |
Willy Tarreau | ef1d1f8 | 2007-04-16 00:25:25 +0200 | [diff] [blame] | 294 | * It is a constructor, which means that it will automatically be called before |
| 295 | * main(). This is GCC-specific but it works at least since 2.95. |
| 296 | * 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] | 297 | */ |
Willy Tarreau | ef1d1f8 | 2007-04-16 00:25:25 +0200 | [diff] [blame] | 298 | __attribute__((constructor)) |
| 299 | static void _do_register(void) |
Willy Tarreau | 4f60f16 | 2007-04-08 16:39:58 +0200 | [diff] [blame] | 300 | { |
Willy Tarreau | ef1d1f8 | 2007-04-16 00:25:25 +0200 | [diff] [blame] | 301 | struct poller *p; |
Willy Tarreau | d9e7e36 | 2018-01-18 19:16:02 +0100 | [diff] [blame] | 302 | int i; |
Willy Tarreau | ef1d1f8 | 2007-04-16 00:25:25 +0200 | [diff] [blame] | 303 | |
| 304 | if (nbpollers >= MAX_POLLERS) |
| 305 | return; |
Willy Tarreau | d79e79b | 2009-05-10 10:18:54 +0200 | [diff] [blame] | 306 | |
Willy Tarreau | d9e7e36 | 2018-01-18 19:16:02 +0100 | [diff] [blame] | 307 | for (i = 0; i < MAX_THREADS; i++) |
| 308 | epoll_fd[i] = -1; |
| 309 | |
Willy Tarreau | ef1d1f8 | 2007-04-16 00:25:25 +0200 | [diff] [blame] | 310 | p = &pollers[nbpollers++]; |
| 311 | |
Willy Tarreau | 4f60f16 | 2007-04-08 16:39:58 +0200 | [diff] [blame] | 312 | p->name = "epoll"; |
| 313 | p->pref = 300; |
Willy Tarreau | 5a76769 | 2017-03-13 11:38:28 +0100 | [diff] [blame] | 314 | p->flags = 0; |
Willy Tarreau | 4f60f16 | 2007-04-08 16:39:58 +0200 | [diff] [blame] | 315 | p->private = NULL; |
| 316 | |
Conrad Hoffmann | 041751c | 2014-05-20 14:28:24 +0200 | [diff] [blame] | 317 | p->clo = __fd_clo; |
Willy Tarreau | ef1d1f8 | 2007-04-16 00:25:25 +0200 | [diff] [blame] | 318 | p->test = _do_test; |
| 319 | p->init = _do_init; |
| 320 | p->term = _do_term; |
| 321 | p->poll = _do_poll; |
Willy Tarreau | fb8983f | 2007-06-03 16:40:44 +0200 | [diff] [blame] | 322 | p->fork = _do_fork; |
Willy Tarreau | 4f60f16 | 2007-04-08 16:39:58 +0200 | [diff] [blame] | 323 | } |
| 324 | |
| 325 | |
| 326 | /* |
| 327 | * Local variables: |
| 328 | * c-indent-level: 8 |
| 329 | * c-basic-offset: 8 |
| 330 | * End: |
| 331 | */ |