Willy Tarreau | 4f60f16 | 2007-04-08 16:39:58 +0200 | [diff] [blame] | 1 | /* |
| 2 | * FD polling functions for generic poll() |
| 3 | * |
Willy Tarreau | cc7e3f7 | 2012-11-11 17:25:15 +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. |
| 10 | * |
| 11 | */ |
| 12 | |
| 13 | #include <unistd.h> |
Willy Tarreau | 69801b8 | 2007-04-09 15:28:51 +0200 | [diff] [blame] | 14 | #include <sys/poll.h> |
Willy Tarreau | 4f60f16 | 2007-04-08 16:39:58 +0200 | [diff] [blame] | 15 | #include <sys/time.h> |
| 16 | #include <sys/types.h> |
| 17 | |
| 18 | #include <common/compat.h> |
| 19 | #include <common/config.h> |
Willy Tarreau | 0c303ee | 2008-07-07 00:09:58 +0200 | [diff] [blame] | 20 | #include <common/ticks.h> |
Willy Tarreau | 4f60f16 | 2007-04-08 16:39:58 +0200 | [diff] [blame] | 21 | #include <common/time.h> |
| 22 | |
Willy Tarreau | 4f60f16 | 2007-04-08 16:39:58 +0200 | [diff] [blame] | 23 | #include <types/global.h> |
| 24 | |
| 25 | #include <proto/fd.h> |
Willy Tarreau | 332740d | 2009-05-10 09:57:21 +0200 | [diff] [blame] | 26 | #include <proto/signal.h> |
Willy Tarreau | 4f60f16 | 2007-04-08 16:39:58 +0200 | [diff] [blame] | 27 | #include <proto/task.h> |
| 28 | |
| 29 | |
Willy Tarreau | 80da05a | 2013-03-31 14:06:57 +0200 | [diff] [blame] | 30 | static unsigned int *fd_evts[2]; |
Willy Tarreau | 4f60f16 | 2007-04-08 16:39:58 +0200 | [diff] [blame] | 31 | |
| 32 | /* private data */ |
| 33 | static struct pollfd *poll_events = NULL; |
| 34 | |
| 35 | |
Willy Tarreau | 80da05a | 2013-03-31 14:06:57 +0200 | [diff] [blame] | 36 | static inline unsigned int hap_fd_isset(int fd, unsigned int *evts) |
Willy Tarreau | 4f60f16 | 2007-04-08 16:39:58 +0200 | [diff] [blame] | 37 | { |
Willy Tarreau | 80da05a | 2013-03-31 14:06:57 +0200 | [diff] [blame] | 38 | return evts[fd / (8*sizeof(*evts))] & (1U << (fd & (8*sizeof(*evts) - 1))); |
Willy Tarreau | 4f60f16 | 2007-04-08 16:39:58 +0200 | [diff] [blame] | 39 | } |
| 40 | |
Willy Tarreau | 80da05a | 2013-03-31 14:06:57 +0200 | [diff] [blame] | 41 | static inline void hap_fd_set(int fd, unsigned int *evts) |
| 42 | { |
| 43 | evts[fd / (8*sizeof(*evts))] |= 1U << (fd & (8*sizeof(*evts) - 1)); |
| 44 | } |
| 45 | |
| 46 | static inline void hap_fd_clr(int fd, unsigned int *evts) |
| 47 | { |
| 48 | evts[fd / (8*sizeof(*evts))] &= ~(1U << (fd & (8*sizeof(*evts) - 1))); |
| 49 | } |
| 50 | |
| 51 | REGPRM1 static void __fd_clo(int fd) |
| 52 | { |
| 53 | hap_fd_clr(fd, fd_evts[DIR_RD]); |
| 54 | hap_fd_clr(fd, fd_evts[DIR_WR]); |
| 55 | } |
| 56 | |
Willy Tarreau | 4f60f16 | 2007-04-08 16:39:58 +0200 | [diff] [blame] | 57 | /* |
| 58 | * Poll() poller |
| 59 | */ |
Willy Tarreau | 0c303ee | 2008-07-07 00:09:58 +0200 | [diff] [blame] | 60 | REGPRM2 static void _do_poll(struct poller *p, int exp) |
Willy Tarreau | 4f60f16 | 2007-04-08 16:39:58 +0200 | [diff] [blame] | 61 | { |
| 62 | int status; |
| 63 | int fd, nbfd; |
Willy Tarreau | d825eef | 2007-05-12 22:35:00 +0200 | [diff] [blame] | 64 | int wait_time; |
Willy Tarreau | cc7e3f7 | 2012-11-11 17:25:15 +0100 | [diff] [blame] | 65 | int updt_idx, en, eo; |
Willy Tarreau | 4f60f16 | 2007-04-08 16:39:58 +0200 | [diff] [blame] | 66 | int fds, count; |
| 67 | int sr, sw; |
| 68 | unsigned rn, wn; /* read new, write new */ |
| 69 | |
Willy Tarreau | cc7e3f7 | 2012-11-11 17:25:15 +0100 | [diff] [blame] | 70 | /* first, scan the update list to find changes */ |
| 71 | for (updt_idx = 0; updt_idx < fd_nbupdt; updt_idx++) { |
| 72 | fd = fd_updt[updt_idx]; |
Willy Tarreau | 69a41fa | 2014-01-20 11:02:59 +0100 | [diff] [blame^] | 73 | en = fdtab[fd].state & 15; /* new events */ |
| 74 | eo = fdtab[fd].state >> 4; /* previous events */ |
Willy Tarreau | cc7e3f7 | 2012-11-11 17:25:15 +0100 | [diff] [blame] | 75 | |
| 76 | if (fdtab[fd].owner && (eo ^ en)) { |
| 77 | if ((eo ^ en) & FD_EV_POLLED_RW) { |
| 78 | /* poll status changed, update the lists */ |
| 79 | if ((eo & ~en) & FD_EV_POLLED_R) |
Willy Tarreau | 80da05a | 2013-03-31 14:06:57 +0200 | [diff] [blame] | 80 | hap_fd_clr(fd, fd_evts[DIR_RD]); |
Willy Tarreau | cc7e3f7 | 2012-11-11 17:25:15 +0100 | [diff] [blame] | 81 | else if ((en & ~eo) & FD_EV_POLLED_R) |
Willy Tarreau | 80da05a | 2013-03-31 14:06:57 +0200 | [diff] [blame] | 82 | hap_fd_set(fd, fd_evts[DIR_RD]); |
Willy Tarreau | cc7e3f7 | 2012-11-11 17:25:15 +0100 | [diff] [blame] | 83 | |
| 84 | if ((eo & ~en) & FD_EV_POLLED_W) |
Willy Tarreau | 80da05a | 2013-03-31 14:06:57 +0200 | [diff] [blame] | 85 | hap_fd_clr(fd, fd_evts[DIR_WR]); |
Willy Tarreau | cc7e3f7 | 2012-11-11 17:25:15 +0100 | [diff] [blame] | 86 | else if ((en & ~eo) & FD_EV_POLLED_W) |
Willy Tarreau | 80da05a | 2013-03-31 14:06:57 +0200 | [diff] [blame] | 87 | hap_fd_set(fd, fd_evts[DIR_WR]); |
Willy Tarreau | cc7e3f7 | 2012-11-11 17:25:15 +0100 | [diff] [blame] | 88 | } |
| 89 | |
Willy Tarreau | 69a41fa | 2014-01-20 11:02:59 +0100 | [diff] [blame^] | 90 | fdtab[fd].state = (en << 4) + en; /* save new events */ |
Willy Tarreau | cc7e3f7 | 2012-11-11 17:25:15 +0100 | [diff] [blame] | 91 | |
| 92 | if (!(en & FD_EV_ACTIVE_RW)) { |
| 93 | /* This fd doesn't use any active entry anymore, we can |
| 94 | * kill its entry. |
| 95 | */ |
| 96 | release_spec_entry(fd); |
| 97 | } |
| 98 | else if ((en & ~eo) & FD_EV_ACTIVE_RW) { |
| 99 | /* we need a new spec entry now */ |
| 100 | alloc_spec_entry(fd); |
| 101 | } |
| 102 | |
| 103 | } |
| 104 | fdtab[fd].updated = 0; |
| 105 | fdtab[fd].new = 0; |
| 106 | } |
| 107 | fd_nbupdt = 0; |
| 108 | |
Willy Tarreau | 4f60f16 | 2007-04-08 16:39:58 +0200 | [diff] [blame] | 109 | nbfd = 0; |
Willy Tarreau | 80da05a | 2013-03-31 14:06:57 +0200 | [diff] [blame] | 110 | for (fds = 0; (fds * 8*sizeof(**fd_evts)) < maxfd; fds++) { |
| 111 | rn = fd_evts[DIR_RD][fds]; |
| 112 | wn = fd_evts[DIR_WR][fds]; |
Willy Tarreau | 4f60f16 | 2007-04-08 16:39:58 +0200 | [diff] [blame] | 113 | |
Willy Tarreau | 80da05a | 2013-03-31 14:06:57 +0200 | [diff] [blame] | 114 | if (!(rn|wn)) |
| 115 | continue; |
Willy Tarreau | 4f60f16 | 2007-04-08 16:39:58 +0200 | [diff] [blame] | 116 | |
Willy Tarreau | 80da05a | 2013-03-31 14:06:57 +0200 | [diff] [blame] | 117 | for (count = 0, fd = fds * 8*sizeof(**fd_evts); count < 8*sizeof(**fd_evts) && fd < maxfd; count++, fd++) { |
| 118 | sr = (rn >> count) & 1; |
| 119 | sw = (wn >> count) & 1; |
| 120 | if ((sr|sw)) { |
| 121 | poll_events[nbfd].fd = fd; |
| 122 | poll_events[nbfd].events = (sr ? POLLIN : 0) | (sw ? POLLOUT : 0); |
| 123 | nbfd++; |
Willy Tarreau | 4f60f16 | 2007-04-08 16:39:58 +0200 | [diff] [blame] | 124 | } |
| 125 | } |
| 126 | } |
| 127 | |
| 128 | /* now let's wait for events */ |
Willy Tarreau | cc7e3f7 | 2012-11-11 17:25:15 +0100 | [diff] [blame] | 129 | if (fd_nbspec || run_queue || signal_queue_len) |
Willy Tarreau | 3a62811 | 2008-06-13 21:06:56 +0200 | [diff] [blame] | 130 | wait_time = 0; |
Willy Tarreau | 0c303ee | 2008-07-07 00:09:58 +0200 | [diff] [blame] | 131 | else if (!exp) |
Willy Tarreau | b0b37bc | 2008-06-23 14:00:57 +0200 | [diff] [blame] | 132 | wait_time = MAX_DELAY_MS; |
Willy Tarreau | 0c303ee | 2008-07-07 00:09:58 +0200 | [diff] [blame] | 133 | else if (tick_is_expired(exp, now_ms)) |
Willy Tarreau | bdefc51 | 2007-05-14 02:02:04 +0200 | [diff] [blame] | 134 | wait_time = 0; |
Willy Tarreau | b0b37bc | 2008-06-23 14:00:57 +0200 | [diff] [blame] | 135 | else { |
Willy Tarreau | 0c303ee | 2008-07-07 00:09:58 +0200 | [diff] [blame] | 136 | wait_time = TICKS_TO_MS(tick_remain(now_ms, exp)) + 1; |
Willy Tarreau | b0b37bc | 2008-06-23 14:00:57 +0200 | [diff] [blame] | 137 | if (wait_time > MAX_DELAY_MS) |
| 138 | wait_time = MAX_DELAY_MS; |
| 139 | } |
Willy Tarreau | d825eef | 2007-05-12 22:35:00 +0200 | [diff] [blame] | 140 | |
Willy Tarreau | 45a1251 | 2011-09-10 16:56:42 +0200 | [diff] [blame] | 141 | gettimeofday(&before_poll, NULL); |
Willy Tarreau | 4f60f16 | 2007-04-08 16:39:58 +0200 | [diff] [blame] | 142 | status = poll(poll_events, nbfd, wait_time); |
Willy Tarreau | b0b37bc | 2008-06-23 14:00:57 +0200 | [diff] [blame] | 143 | tv_update_date(wait_time, status); |
Willy Tarreau | 45a1251 | 2011-09-10 16:56:42 +0200 | [diff] [blame] | 144 | measure_idle(); |
Willy Tarreau | 4f60f16 | 2007-04-08 16:39:58 +0200 | [diff] [blame] | 145 | |
| 146 | for (count = 0; status > 0 && count < nbfd; count++) { |
Willy Tarreau | 491c498 | 2012-07-06 11:16:01 +0200 | [diff] [blame] | 147 | int e = poll_events[count].revents; |
Willy Tarreau | 4f60f16 | 2007-04-08 16:39:58 +0200 | [diff] [blame] | 148 | fd = poll_events[count].fd; |
| 149 | |
Willy Tarreau | 491c498 | 2012-07-06 11:16:01 +0200 | [diff] [blame] | 150 | if (!(e & ( POLLOUT | POLLIN | POLLERR | POLLHUP ))) |
Willy Tarreau | 4f60f16 | 2007-04-08 16:39:58 +0200 | [diff] [blame] | 151 | continue; |
| 152 | |
Willy Tarreau | 076be25 | 2012-07-06 16:02:29 +0200 | [diff] [blame] | 153 | /* ok, we found one active fd */ |
| 154 | status--; |
| 155 | |
| 156 | if (!fdtab[fd].owner) |
| 157 | continue; |
| 158 | |
Willy Tarreau | 462c720 | 2012-12-13 22:26:37 +0100 | [diff] [blame] | 159 | /* it looks complicated but gcc can optimize it away when constants |
| 160 | * have same values... In fact it depends on gcc :-( |
| 161 | */ |
Willy Tarreau | 491c498 | 2012-07-06 11:16:01 +0200 | [diff] [blame] | 162 | fdtab[fd].ev &= FD_POLL_STICKY; |
Willy Tarreau | 462c720 | 2012-12-13 22:26:37 +0100 | [diff] [blame] | 163 | if (POLLIN == FD_POLL_IN && POLLOUT == FD_POLL_OUT && |
| 164 | POLLERR == FD_POLL_ERR && POLLHUP == FD_POLL_HUP) { |
| 165 | fdtab[fd].ev |= e & (POLLIN|POLLOUT|POLLERR|POLLHUP); |
| 166 | } |
| 167 | else { |
| 168 | fdtab[fd].ev |= |
| 169 | ((e & POLLIN ) ? FD_POLL_IN : 0) | |
| 170 | ((e & POLLOUT) ? FD_POLL_OUT : 0) | |
| 171 | ((e & POLLERR) ? FD_POLL_ERR : 0) | |
| 172 | ((e & POLLHUP) ? FD_POLL_HUP : 0); |
| 173 | } |
Willy Tarreau | 491c498 | 2012-07-06 11:16:01 +0200 | [diff] [blame] | 174 | |
Willy Tarreau | db9cb0b | 2012-12-13 23:41:12 +0100 | [diff] [blame] | 175 | if (fdtab[fd].iocb && fdtab[fd].ev) { |
Willy Tarreau | cc7e3f7 | 2012-11-11 17:25:15 +0100 | [diff] [blame] | 176 | /* Mark the events as speculative before processing |
| 177 | * them so that if nothing can be done we don't need |
| 178 | * to poll again. |
| 179 | */ |
Willy Tarreau | 26d7cfc | 2012-12-07 00:09:43 +0100 | [diff] [blame] | 180 | if (fdtab[fd].ev & FD_POLL_IN) |
Willy Tarreau | cc7e3f7 | 2012-11-11 17:25:15 +0100 | [diff] [blame] | 181 | fd_ev_set(fd, DIR_RD); |
| 182 | |
Willy Tarreau | 26d7cfc | 2012-12-07 00:09:43 +0100 | [diff] [blame] | 183 | if (fdtab[fd].ev & FD_POLL_OUT) |
Willy Tarreau | cc7e3f7 | 2012-11-11 17:25:15 +0100 | [diff] [blame] | 184 | fd_ev_set(fd, DIR_WR); |
| 185 | |
Willy Tarreau | 39ebef8 | 2012-12-14 00:17:03 +0100 | [diff] [blame] | 186 | if (fdtab[fd].spec_p) { |
| 187 | /* This fd was already scheduled for being |
| 188 | * called as a speculative I/O |
| 189 | */ |
| 190 | continue; |
| 191 | } |
| 192 | |
Willy Tarreau | 9845e75 | 2012-07-06 11:44:28 +0200 | [diff] [blame] | 193 | fdtab[fd].iocb(fd); |
Willy Tarreau | cc7e3f7 | 2012-11-11 17:25:15 +0100 | [diff] [blame] | 194 | } |
Willy Tarreau | 4f60f16 | 2007-04-08 16:39:58 +0200 | [diff] [blame] | 195 | } |
| 196 | |
Willy Tarreau | e54e917 | 2007-04-09 09:23:31 +0200 | [diff] [blame] | 197 | } |
| 198 | |
| 199 | /* |
| 200 | * Initialization of the poll() poller. |
| 201 | * Returns 0 in case of failure, non-zero in case of success. If it fails, it |
| 202 | * disables the poller by setting its pref to 0. |
| 203 | */ |
Willy Tarreau | ef1d1f8 | 2007-04-16 00:25:25 +0200 | [diff] [blame] | 204 | REGPRM1 static int _do_init(struct poller *p) |
Willy Tarreau | e54e917 | 2007-04-09 09:23:31 +0200 | [diff] [blame] | 205 | { |
| 206 | __label__ fail_swevt, fail_srevt, fail_pe; |
Willy Tarreau | 80da05a | 2013-03-31 14:06:57 +0200 | [diff] [blame] | 207 | int fd_evts_bytes; |
Willy Tarreau | e54e917 | 2007-04-09 09:23:31 +0200 | [diff] [blame] | 208 | |
| 209 | p->private = NULL; |
Willy Tarreau | 80da05a | 2013-03-31 14:06:57 +0200 | [diff] [blame] | 210 | fd_evts_bytes = (global.maxsock + sizeof(**fd_evts) - 1) / sizeof(**fd_evts) * sizeof(**fd_evts); |
Willy Tarreau | e54e917 | 2007-04-09 09:23:31 +0200 | [diff] [blame] | 211 | |
Willy Tarreau | 80da05a | 2013-03-31 14:06:57 +0200 | [diff] [blame] | 212 | poll_events = calloc(1, sizeof(struct pollfd) * global.maxsock); |
Willy Tarreau | e54e917 | 2007-04-09 09:23:31 +0200 | [diff] [blame] | 213 | |
| 214 | if (poll_events == NULL) |
| 215 | goto fail_pe; |
| 216 | |
Willy Tarreau | 80da05a | 2013-03-31 14:06:57 +0200 | [diff] [blame] | 217 | if ((fd_evts[DIR_RD] = calloc(1, fd_evts_bytes)) == NULL) |
Willy Tarreau | e54e917 | 2007-04-09 09:23:31 +0200 | [diff] [blame] | 218 | goto fail_srevt; |
| 219 | |
Willy Tarreau | 80da05a | 2013-03-31 14:06:57 +0200 | [diff] [blame] | 220 | if ((fd_evts[DIR_WR] = calloc(1, fd_evts_bytes)) == NULL) |
Willy Tarreau | e54e917 | 2007-04-09 09:23:31 +0200 | [diff] [blame] | 221 | goto fail_swevt; |
| 222 | |
| 223 | return 1; |
| 224 | |
| 225 | fail_swevt: |
| 226 | free(fd_evts[DIR_RD]); |
| 227 | fail_srevt: |
| 228 | free(poll_events); |
| 229 | fail_pe: |
| 230 | p->pref = 0; |
| 231 | return 0; |
| 232 | } |
| 233 | |
| 234 | /* |
| 235 | * Termination of the poll() poller. |
| 236 | * Memory is released and the poller is marked as unselectable. |
| 237 | */ |
Willy Tarreau | ef1d1f8 | 2007-04-16 00:25:25 +0200 | [diff] [blame] | 238 | REGPRM1 static void _do_term(struct poller *p) |
Willy Tarreau | e54e917 | 2007-04-09 09:23:31 +0200 | [diff] [blame] | 239 | { |
Willy Tarreau | a534fea | 2008-08-03 12:19:50 +0200 | [diff] [blame] | 240 | free(fd_evts[DIR_WR]); |
| 241 | free(fd_evts[DIR_RD]); |
| 242 | free(poll_events); |
Willy Tarreau | e54e917 | 2007-04-09 09:23:31 +0200 | [diff] [blame] | 243 | p->private = NULL; |
| 244 | p->pref = 0; |
Willy Tarreau | 4f60f16 | 2007-04-08 16:39:58 +0200 | [diff] [blame] | 245 | } |
| 246 | |
| 247 | /* |
Willy Tarreau | 2ff7622 | 2007-04-09 19:29:56 +0200 | [diff] [blame] | 248 | * Check that the poller works. |
| 249 | * Returns 1 if OK, otherwise 0. |
| 250 | */ |
Willy Tarreau | ef1d1f8 | 2007-04-16 00:25:25 +0200 | [diff] [blame] | 251 | REGPRM1 static int _do_test(struct poller *p) |
Willy Tarreau | 2ff7622 | 2007-04-09 19:29:56 +0200 | [diff] [blame] | 252 | { |
| 253 | return 1; |
| 254 | } |
| 255 | |
| 256 | /* |
Willy Tarreau | ef1d1f8 | 2007-04-16 00:25:25 +0200 | [diff] [blame] | 257 | * It is a constructor, which means that it will automatically be called before |
| 258 | * main(). This is GCC-specific but it works at least since 2.95. |
| 259 | * 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] | 260 | */ |
Willy Tarreau | ef1d1f8 | 2007-04-16 00:25:25 +0200 | [diff] [blame] | 261 | __attribute__((constructor)) |
| 262 | static void _do_register(void) |
Willy Tarreau | 4f60f16 | 2007-04-08 16:39:58 +0200 | [diff] [blame] | 263 | { |
Willy Tarreau | ef1d1f8 | 2007-04-16 00:25:25 +0200 | [diff] [blame] | 264 | struct poller *p; |
| 265 | |
| 266 | if (nbpollers >= MAX_POLLERS) |
| 267 | return; |
| 268 | p = &pollers[nbpollers++]; |
| 269 | |
Willy Tarreau | 4f60f16 | 2007-04-08 16:39:58 +0200 | [diff] [blame] | 270 | p->name = "poll"; |
| 271 | p->pref = 200; |
| 272 | p->private = NULL; |
| 273 | |
Willy Tarreau | 70c6fd8 | 2012-11-11 21:02:34 +0100 | [diff] [blame] | 274 | p->clo = __fd_clo; |
Willy Tarreau | ef1d1f8 | 2007-04-16 00:25:25 +0200 | [diff] [blame] | 275 | p->test = _do_test; |
| 276 | p->init = _do_init; |
| 277 | p->term = _do_term; |
| 278 | p->poll = _do_poll; |
Willy Tarreau | 4f60f16 | 2007-04-08 16:39:58 +0200 | [diff] [blame] | 279 | } |
| 280 | |
| 281 | |
| 282 | /* |
| 283 | * Local variables: |
| 284 | * c-indent-level: 8 |
| 285 | * c-basic-offset: 8 |
| 286 | * End: |
| 287 | */ |