blob: f9e445113c371290453befa02de690b4a55688bd [file] [log] [blame]
Willy Tarreau4f60f162007-04-08 16:39:58 +02001/*
2 * FD polling functions for generic poll()
3 *
Willy Tarreauf817e9f2014-01-10 16:58:45 +01004 * Copyright 2000-2014 Willy Tarreau <w@1wt.eu>
Willy Tarreau4f60f162007-04-08 16:39:58 +02005 *
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
Willy Tarreau3c8a8962017-03-13 17:14:51 +010013#define _GNU_SOURCE // for POLLRDHUP on Linux
14
Willy Tarreau4f60f162007-04-08 16:39:58 +020015#include <unistd.h>
Willy Tarreau3c8a8962017-03-13 17:14:51 +010016#include <poll.h>
Willy Tarreau4f60f162007-04-08 16:39:58 +020017#include <sys/time.h>
18#include <sys/types.h>
19
20#include <common/compat.h>
21#include <common/config.h>
Willy Tarreau0c303ee2008-07-07 00:09:58 +020022#include <common/ticks.h>
Willy Tarreau4f60f162007-04-08 16:39:58 +020023#include <common/time.h>
24
Willy Tarreau4f60f162007-04-08 16:39:58 +020025#include <types/global.h>
26
27#include <proto/fd.h>
Willy Tarreau4f60f162007-04-08 16:39:58 +020028
Willy Tarreau3c8a8962017-03-13 17:14:51 +010029
30#ifndef POLLRDHUP
31/* POLLRDHUP was defined late in libc, and it appeared in kernel 2.6.17 */
32#define POLLRDHUP 0
33#endif
Willy Tarreau4f60f162007-04-08 16:39:58 +020034
Willy Tarreau80da05a2013-03-31 14:06:57 +020035static unsigned int *fd_evts[2];
Willy Tarreau4f60f162007-04-08 16:39:58 +020036
37/* private data */
Christopher Fauletd4604ad2017-05-29 10:40:41 +020038static THREAD_LOCAL int nbfd = 0;
39static THREAD_LOCAL struct pollfd *poll_events = NULL;
Willy Tarreau4f60f162007-04-08 16:39:58 +020040
Willy Tarreau80da05a2013-03-31 14:06:57 +020041static 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
46static 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
51REGPRM1 static void __fd_clo(int fd)
52{
Christopher Faulet2a944ee2017-11-07 10:42:54 +010053 HA_SPIN_LOCK(POLL_LOCK, &poll_lock);
Willy Tarreau80da05a2013-03-31 14:06:57 +020054 hap_fd_clr(fd, fd_evts[DIR_RD]);
55 hap_fd_clr(fd, fd_evts[DIR_WR]);
Christopher Faulet2a944ee2017-11-07 10:42:54 +010056 HA_SPIN_UNLOCK(POLL_LOCK, &poll_lock);
Willy Tarreau80da05a2013-03-31 14:06:57 +020057}
58
Willy Tarreau4f60f162007-04-08 16:39:58 +020059/*
60 * Poll() poller
61 */
Willy Tarreau0c303ee2008-07-07 00:09:58 +020062REGPRM2 static void _do_poll(struct poller *p, int exp)
Willy Tarreau4f60f162007-04-08 16:39:58 +020063{
64 int status;
Christopher Fauletd4604ad2017-05-29 10:40:41 +020065 int fd;
Willy Tarreaud825eef2007-05-12 22:35:00 +020066 int wait_time;
Willy Tarreaucc7e3f72012-11-11 17:25:15 +010067 int updt_idx, en, eo;
Willy Tarreau4f60f162007-04-08 16:39:58 +020068 int fds, count;
69 int sr, sw;
70 unsigned rn, wn; /* read new, write new */
71
Willy Tarreaucc7e3f72012-11-11 17:25:15 +010072 /* first, scan the update list to find changes */
73 for (updt_idx = 0; updt_idx < fd_nbupdt; updt_idx++) {
74 fd = fd_updt[updt_idx];
Willy Tarreaucc7e3f72012-11-11 17:25:15 +010075
Willy Tarreauf817e9f2014-01-10 16:58:45 +010076 if (!fdtab[fd].owner)
77 continue;
Willy Tarreaucc7e3f72012-11-11 17:25:15 +010078
Christopher Faulet2a944ee2017-11-07 10:42:54 +010079 HA_SPIN_LOCK(FD_LOCK, &fdtab[fd].lock);
Christopher Fauletd4604ad2017-05-29 10:40:41 +020080 fdtab[fd].updated = 0;
81 fdtab[fd].new = 0;
82
Willy Tarreau25002d22014-01-25 10:32:56 +010083 eo = fdtab[fd].state;
84 en = fd_compute_new_polled_status(eo);
Christopher Fauletd4604ad2017-05-29 10:40:41 +020085 fdtab[fd].state = en;
Christopher Faulet2a944ee2017-11-07 10:42:54 +010086 HA_SPIN_UNLOCK(FD_LOCK, &fdtab[fd].lock);
Willy Tarreauf817e9f2014-01-10 16:58:45 +010087
88 if ((eo ^ en) & FD_EV_POLLED_RW) {
89 /* poll status changed, update the lists */
Christopher Faulet2a944ee2017-11-07 10:42:54 +010090 HA_SPIN_LOCK(POLL_LOCK, &poll_lock);
Willy Tarreauf817e9f2014-01-10 16:58:45 +010091 if ((eo & ~en) & FD_EV_POLLED_R)
92 hap_fd_clr(fd, fd_evts[DIR_RD]);
93 else if ((en & ~eo) & FD_EV_POLLED_R)
94 hap_fd_set(fd, fd_evts[DIR_RD]);
95
96 if ((eo & ~en) & FD_EV_POLLED_W)
97 hap_fd_clr(fd, fd_evts[DIR_WR]);
98 else if ((en & ~eo) & FD_EV_POLLED_W)
99 hap_fd_set(fd, fd_evts[DIR_WR]);
Christopher Faulet2a944ee2017-11-07 10:42:54 +0100100 HA_SPIN_UNLOCK(POLL_LOCK, &poll_lock);
Willy Tarreauf817e9f2014-01-10 16:58:45 +0100101 }
Willy Tarreaucc7e3f72012-11-11 17:25:15 +0100102 }
103 fd_nbupdt = 0;
104
Willy Tarreau4f60f162007-04-08 16:39:58 +0200105 nbfd = 0;
Willy Tarreau80da05a2013-03-31 14:06:57 +0200106 for (fds = 0; (fds * 8*sizeof(**fd_evts)) < maxfd; fds++) {
107 rn = fd_evts[DIR_RD][fds];
108 wn = fd_evts[DIR_WR][fds];
Christopher Fauletd4604ad2017-05-29 10:40:41 +0200109
Willy Tarreau80da05a2013-03-31 14:06:57 +0200110 if (!(rn|wn))
111 continue;
Willy Tarreau4f60f162007-04-08 16:39:58 +0200112
Willy Tarreau80da05a2013-03-31 14:06:57 +0200113 for (count = 0, fd = fds * 8*sizeof(**fd_evts); count < 8*sizeof(**fd_evts) && fd < maxfd; count++, fd++) {
Christopher Faulet63e2ce62017-06-02 14:36:39 +0200114
Willy Tarreauf65610a2017-10-31 16:06:06 +0100115 if (!fdtab[fd].owner || !(fdtab[fd].thread_mask & tid_bit))
Christopher Faulet63e2ce62017-06-02 14:36:39 +0200116 continue;
117
Willy Tarreau80da05a2013-03-31 14:06:57 +0200118 sr = (rn >> count) & 1;
119 sw = (wn >> count) & 1;
120 if ((sr|sw)) {
121 poll_events[nbfd].fd = fd;
Willy Tarreau3c8a8962017-03-13 17:14:51 +0100122 poll_events[nbfd].events = (sr ? (POLLIN | POLLRDHUP) : 0) | (sw ? POLLOUT : 0);
Willy Tarreau80da05a2013-03-31 14:06:57 +0200123 nbfd++;
Willy Tarreau4f60f162007-04-08 16:39:58 +0200124 }
Christopher Fauletd4604ad2017-05-29 10:40:41 +0200125 }
Willy Tarreau4f60f162007-04-08 16:39:58 +0200126 }
Christopher Fauletd4604ad2017-05-29 10:40:41 +0200127
Willy Tarreau4f60f162007-04-08 16:39:58 +0200128 /* now let's wait for events */
Willy Tarreau10146c92015-04-13 20:44:19 +0200129 if (!exp)
Willy Tarreaub0b37bc2008-06-23 14:00:57 +0200130 wait_time = MAX_DELAY_MS;
Willy Tarreau0c303ee2008-07-07 00:09:58 +0200131 else if (tick_is_expired(exp, now_ms))
Willy Tarreaubdefc512007-05-14 02:02:04 +0200132 wait_time = 0;
Willy Tarreaub0b37bc2008-06-23 14:00:57 +0200133 else {
Willy Tarreau0c303ee2008-07-07 00:09:58 +0200134 wait_time = TICKS_TO_MS(tick_remain(now_ms, exp)) + 1;
Willy Tarreaub0b37bc2008-06-23 14:00:57 +0200135 if (wait_time > MAX_DELAY_MS)
136 wait_time = MAX_DELAY_MS;
137 }
Willy Tarreaud825eef2007-05-12 22:35:00 +0200138
Willy Tarreau45a12512011-09-10 16:56:42 +0200139 gettimeofday(&before_poll, NULL);
Willy Tarreau4f60f162007-04-08 16:39:58 +0200140 status = poll(poll_events, nbfd, wait_time);
Willy Tarreaub0b37bc2008-06-23 14:00:57 +0200141 tv_update_date(wait_time, status);
Willy Tarreau45a12512011-09-10 16:56:42 +0200142 measure_idle();
Willy Tarreau4f60f162007-04-08 16:39:58 +0200143
144 for (count = 0; status > 0 && count < nbfd; count++) {
Christopher Fauletab62f512017-08-30 10:34:36 +0200145 unsigned int n;
Willy Tarreau491c4982012-07-06 11:16:01 +0200146 int e = poll_events[count].revents;
Willy Tarreau4f60f162007-04-08 16:39:58 +0200147 fd = poll_events[count].fd;
Christopher Fauletd4604ad2017-05-29 10:40:41 +0200148
Willy Tarreau3c8a8962017-03-13 17:14:51 +0100149 if (!(e & ( POLLOUT | POLLIN | POLLERR | POLLHUP | POLLRDHUP )))
Willy Tarreau4f60f162007-04-08 16:39:58 +0200150 continue;
151
Willy Tarreau076be252012-07-06 16:02:29 +0200152 /* ok, we found one active fd */
153 status--;
154
155 if (!fdtab[fd].owner)
156 continue;
157
Willy Tarreau462c7202012-12-13 22:26:37 +0100158 /* it looks complicated but gcc can optimize it away when constants
159 * have same values... In fact it depends on gcc :-(
160 */
Willy Tarreau462c7202012-12-13 22:26:37 +0100161 if (POLLIN == FD_POLL_IN && POLLOUT == FD_POLL_OUT &&
162 POLLERR == FD_POLL_ERR && POLLHUP == FD_POLL_HUP) {
Christopher Fauletab62f512017-08-30 10:34:36 +0200163 n = e & (POLLIN|POLLOUT|POLLERR|POLLHUP);
Willy Tarreau462c7202012-12-13 22:26:37 +0100164 }
165 else {
Christopher Fauletab62f512017-08-30 10:34:36 +0200166 n = ((e & POLLIN ) ? FD_POLL_IN : 0) |
Willy Tarreau462c7202012-12-13 22:26:37 +0100167 ((e & POLLOUT) ? FD_POLL_OUT : 0) |
168 ((e & POLLERR) ? FD_POLL_ERR : 0) |
169 ((e & POLLHUP) ? FD_POLL_HUP : 0);
170 }
Willy Tarreau491c4982012-07-06 11:16:01 +0200171
Willy Tarreau3c8a8962017-03-13 17:14:51 +0100172 /* always remap RDHUP to HUP as they're used similarly */
173 if (e & POLLRDHUP) {
Christopher Fauletd4604ad2017-05-29 10:40:41 +0200174 HA_ATOMIC_OR(&cur_poller.flags, HAP_POLL_F_RDHUP);
Christopher Fauletab62f512017-08-30 10:34:36 +0200175 n |= FD_POLL_HUP;
Willy Tarreau3c8a8962017-03-13 17:14:51 +0100176 }
Christopher Fauletab62f512017-08-30 10:34:36 +0200177 fd_update_events(fd, n);
Willy Tarreau4f60f162007-04-08 16:39:58 +0200178 }
179
Willy Tarreaue54e9172007-04-09 09:23:31 +0200180}
181
Christopher Fauletd4604ad2017-05-29 10:40:41 +0200182
183static int init_poll_per_thread()
184{
185 poll_events = calloc(1, sizeof(struct pollfd) * global.maxsock);
186 if (poll_events == NULL)
187 return 0;
188 return 1;
189}
190
191static void deinit_poll_per_thread()
192{
193 free(poll_events);
Christopher Fauletcd7879a2017-10-27 13:53:47 +0200194 poll_events = NULL;
Christopher Fauletd4604ad2017-05-29 10:40:41 +0200195}
196
Willy Tarreaue54e9172007-04-09 09:23:31 +0200197/*
198 * Initialization of the poll() poller.
199 * Returns 0 in case of failure, non-zero in case of success. If it fails, it
200 * disables the poller by setting its pref to 0.
201 */
Willy Tarreauef1d1f82007-04-16 00:25:25 +0200202REGPRM1 static int _do_init(struct poller *p)
Willy Tarreaue54e9172007-04-09 09:23:31 +0200203{
Christopher Fauletcd7879a2017-10-27 13:53:47 +0200204 __label__ fail_swevt, fail_srevt;
Willy Tarreau80da05a2013-03-31 14:06:57 +0200205 int fd_evts_bytes;
Willy Tarreaue54e9172007-04-09 09:23:31 +0200206
207 p->private = NULL;
Willy Tarreau80da05a2013-03-31 14:06:57 +0200208 fd_evts_bytes = (global.maxsock + sizeof(**fd_evts) - 1) / sizeof(**fd_evts) * sizeof(**fd_evts);
Willy Tarreaue54e9172007-04-09 09:23:31 +0200209
Willy Tarreau80da05a2013-03-31 14:06:57 +0200210 if ((fd_evts[DIR_RD] = calloc(1, fd_evts_bytes)) == NULL)
Willy Tarreaue54e9172007-04-09 09:23:31 +0200211 goto fail_srevt;
Willy Tarreau80da05a2013-03-31 14:06:57 +0200212 if ((fd_evts[DIR_WR] = calloc(1, fd_evts_bytes)) == NULL)
Willy Tarreaue54e9172007-04-09 09:23:31 +0200213 goto fail_swevt;
214
Christopher Fauletcd7879a2017-10-27 13:53:47 +0200215 hap_register_per_thread_init(init_poll_per_thread);
216 hap_register_per_thread_deinit(deinit_poll_per_thread);
217
Willy Tarreaue54e9172007-04-09 09:23:31 +0200218 return 1;
219
220 fail_swevt:
221 free(fd_evts[DIR_RD]);
222 fail_srevt:
223 free(poll_events);
Willy Tarreaue54e9172007-04-09 09:23:31 +0200224 p->pref = 0;
225 return 0;
226}
227
228/*
229 * Termination of the poll() poller.
230 * Memory is released and the poller is marked as unselectable.
231 */
Willy Tarreauef1d1f82007-04-16 00:25:25 +0200232REGPRM1 static void _do_term(struct poller *p)
Willy Tarreaue54e9172007-04-09 09:23:31 +0200233{
Willy Tarreaua534fea2008-08-03 12:19:50 +0200234 free(fd_evts[DIR_WR]);
235 free(fd_evts[DIR_RD]);
Willy Tarreaue54e9172007-04-09 09:23:31 +0200236 p->private = NULL;
237 p->pref = 0;
Willy Tarreau4f60f162007-04-08 16:39:58 +0200238}
239
240/*
Willy Tarreau2ff76222007-04-09 19:29:56 +0200241 * Check that the poller works.
242 * Returns 1 if OK, otherwise 0.
243 */
Willy Tarreauef1d1f82007-04-16 00:25:25 +0200244REGPRM1 static int _do_test(struct poller *p)
Willy Tarreau2ff76222007-04-09 19:29:56 +0200245{
246 return 1;
247}
248
249/*
Willy Tarreauef1d1f82007-04-16 00:25:25 +0200250 * It is a constructor, which means that it will automatically be called before
251 * main(). This is GCC-specific but it works at least since 2.95.
252 * Special care must be taken so that it does not need any uninitialized data.
Willy Tarreau4f60f162007-04-08 16:39:58 +0200253 */
Willy Tarreauef1d1f82007-04-16 00:25:25 +0200254__attribute__((constructor))
255static void _do_register(void)
Willy Tarreau4f60f162007-04-08 16:39:58 +0200256{
Willy Tarreauef1d1f82007-04-16 00:25:25 +0200257 struct poller *p;
258
259 if (nbpollers >= MAX_POLLERS)
260 return;
261 p = &pollers[nbpollers++];
262
Willy Tarreau4f60f162007-04-08 16:39:58 +0200263 p->name = "poll";
264 p->pref = 200;
Willy Tarreau5a767692017-03-13 11:38:28 +0100265 p->flags = 0;
Willy Tarreau4f60f162007-04-08 16:39:58 +0200266 p->private = NULL;
267
Willy Tarreau70c6fd82012-11-11 21:02:34 +0100268 p->clo = __fd_clo;
Willy Tarreauef1d1f82007-04-16 00:25:25 +0200269 p->test = _do_test;
270 p->init = _do_init;
271 p->term = _do_term;
272 p->poll = _do_poll;
Willy Tarreau4f60f162007-04-08 16:39:58 +0200273}
274
275
276/*
277 * Local variables:
278 * c-indent-level: 8
279 * c-basic-offset: 8
280 * End:
281 */