blob: 2c040492118615d5df69a7b0985b7f258f876476 [file] [log] [blame]
Willy Tarreaudd815982007-10-16 12:25:14 +02001/*
Willy Tarreaud1d54542012-09-12 22:58:11 +02002 * Listener management functions.
Willy Tarreaudd815982007-10-16 12:25:14 +02003 *
Willy Tarreau0ccb7442013-01-07 22:54:17 +01004 * Copyright 2000-2013 Willy Tarreau <w@1wt.eu>
Willy Tarreaudd815982007-10-16 12:25:14 +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 Tarreau6ae1ba62014-05-07 19:01:58 +020013#include <ctype.h>
Willy Tarreaubbebbbf2012-05-07 21:22:09 +020014#include <errno.h>
Willy Tarreaudd815982007-10-16 12:25:14 +020015#include <stdio.h>
16#include <string.h>
Willy Tarreau95ccdde2014-02-01 09:28:36 +010017#include <unistd.h>
Willy Tarreaudd815982007-10-16 12:25:14 +020018
Willy Tarreaudcc048a2020-06-04 19:11:43 +020019#include <haproxy/acl.h>
Willy Tarreau4c7e4b72020-05-27 12:58:42 +020020#include <haproxy/api.h>
Willy Tarreau5d9ddc52021-10-06 19:54:09 +020021#include <haproxy/activity.h>
Willy Tarreau6be78492020-06-05 00:00:29 +020022#include <haproxy/cfgparse.h>
Willy Tarreaudbf78022021-10-06 09:05:08 +020023#include <haproxy/cli-t.h>
Willy Tarreau7ea393d2020-06-04 18:02:10 +020024#include <haproxy/connection.h>
Willy Tarreau8d366972020-05-27 16:10:29 +020025#include <haproxy/errors.h>
Willy Tarreaub2551052020-06-09 09:07:15 +020026#include <haproxy/fd.h>
27#include <haproxy/freq_ctr.h>
Willy Tarreauf268ee82020-06-04 17:05:57 +020028#include <haproxy/global.h>
Willy Tarreau853b2972020-05-27 18:01:47 +020029#include <haproxy/list.h>
Willy Tarreau213e9902020-06-04 14:58:24 +020030#include <haproxy/listener.h>
Willy Tarreauaeed4a82020-06-04 22:01:04 +020031#include <haproxy/log.h>
Willy Tarreaub2551052020-06-09 09:07:15 +020032#include <haproxy/protocol.h>
Willy Tarreau5958c432021-05-08 20:30:37 +020033#include <haproxy/proxy.h>
Willy Tarreaub2551052020-06-09 09:07:15 +020034#include <haproxy/sample.h>
Willy Tarreaudfd3de82020-06-04 23:46:14 +020035#include <haproxy/stream.h>
Willy Tarreaucea0e1b2020-06-04 17:25:40 +020036#include <haproxy/task.h>
Willy Tarreau9310f482021-10-06 16:18:40 +020037#include <haproxy/ticks.h>
Willy Tarreaub2551052020-06-09 09:07:15 +020038#include <haproxy/tools.h>
Willy Tarreaudd815982007-10-16 12:25:14 +020039
Willy Tarreaub648d632007-10-28 22:13:50 +010040
Willy Tarreau26982662012-09-12 23:17:10 +020041/* List head of all known bind keywords */
Willy Tarreauca1acd62022-03-29 15:02:44 +020042struct bind_kw_list bind_keywords = {
Willy Tarreau26982662012-09-12 23:17:10 +020043 .list = LIST_HEAD_INIT(bind_keywords.list)
44};
45
Willy Tarreaua1d97f82019-12-10 11:18:41 +010046/* list of the temporarily limited listeners because of lack of resource */
47static struct mt_list global_listener_queue = MT_LIST_HEAD_INIT(global_listener_queue);
48static struct task *global_listener_queue_task;
Willy Tarreaua1d97f82019-12-10 11:18:41 +010049
William Dauchy3679d0c2021-02-14 23:22:55 +010050/* listener status for stats */
51const char* li_status_st[LI_STATE_COUNT] = {
52 [LI_STATUS_WAITING] = "WAITING",
53 [LI_STATUS_OPEN] = "OPEN",
54 [LI_STATUS_FULL] = "FULL",
55};
Willy Tarreaua1d97f82019-12-10 11:18:41 +010056
Willy Tarreau1efafce2019-01-27 15:37:19 +010057#if defined(USE_THREAD)
58
59struct accept_queue_ring accept_queue_rings[MAX_THREADS] __attribute__((aligned(64))) = { };
60
61/* dequeue and process a pending connection from the local accept queue (single
Willy Tarreau83efc322020-10-14 17:37:17 +020062 * consumer). Returns the accepted connection or NULL if none was found.
Willy Tarreau1efafce2019-01-27 15:37:19 +010063 */
Willy Tarreau83efc322020-10-14 17:37:17 +020064struct connection *accept_queue_pop_sc(struct accept_queue_ring *ring)
Willy Tarreau1efafce2019-01-27 15:37:19 +010065{
Willy Tarreau1efafce2019-01-27 15:37:19 +010066 unsigned int pos, next;
Willy Tarreau83efc322020-10-14 17:37:17 +020067 struct connection *ptr;
68 struct connection **e;
Willy Tarreau1efafce2019-01-27 15:37:19 +010069
70 pos = ring->head;
71
72 if (pos == ring->tail)
Willy Tarreau83efc322020-10-14 17:37:17 +020073 return NULL;
Willy Tarreau1efafce2019-01-27 15:37:19 +010074
75 next = pos + 1;
76 if (next >= ACCEPT_QUEUE_SIZE)
77 next = 0;
78
79 e = &ring->entry[pos];
80
81 /* wait for the producer to update the listener's pointer */
82 while (1) {
Willy Tarreau83efc322020-10-14 17:37:17 +020083 ptr = *e;
Willy Tarreau1efafce2019-01-27 15:37:19 +010084 __ha_barrier_load();
85 if (ptr)
86 break;
87 pl_cpu_relax();
88 }
89
Willy Tarreau1efafce2019-01-27 15:37:19 +010090 /* release the entry */
Willy Tarreau83efc322020-10-14 17:37:17 +020091 *e = NULL;
Willy Tarreau1efafce2019-01-27 15:37:19 +010092
93 __ha_barrier_store();
94 ring->head = next;
Willy Tarreau83efc322020-10-14 17:37:17 +020095 return ptr;
Willy Tarreau1efafce2019-01-27 15:37:19 +010096}
97
98
Willy Tarreau83efc322020-10-14 17:37:17 +020099/* tries to push a new accepted connection <conn> into ring <ring>. Returns
100 * non-zero if it succeeds, or zero if the ring is full. Supports multiple
101 * producers.
Willy Tarreau1efafce2019-01-27 15:37:19 +0100102 */
Willy Tarreau83efc322020-10-14 17:37:17 +0200103int accept_queue_push_mp(struct accept_queue_ring *ring, struct connection *conn)
Willy Tarreau1efafce2019-01-27 15:37:19 +0100104{
Willy Tarreau1efafce2019-01-27 15:37:19 +0100105 unsigned int pos, next;
106
107 pos = ring->tail;
108 do {
109 next = pos + 1;
110 if (next >= ACCEPT_QUEUE_SIZE)
111 next = 0;
112 if (next == ring->head)
113 return 0; // ring full
Olivier Houchard64213e92019-03-08 18:52:57 +0100114 } while (unlikely(!_HA_ATOMIC_CAS(&ring->tail, &pos, next)));
Willy Tarreau1efafce2019-01-27 15:37:19 +0100115
Willy Tarreau83efc322020-10-14 17:37:17 +0200116 ring->entry[pos] = conn;
Willy Tarreau1efafce2019-01-27 15:37:19 +0100117 __ha_barrier_store();
Willy Tarreau1efafce2019-01-27 15:37:19 +0100118 return 1;
119}
120
Willy Tarreaufb5401f2021-01-29 12:25:23 +0100121/* proceed with accepting new connections. Don't mark it static so that it appears
122 * in task dumps.
123 */
Willy Tarreau144f84a2021-03-02 16:09:26 +0100124struct task *accept_queue_process(struct task *t, void *context, unsigned int state)
Willy Tarreau1efafce2019-01-27 15:37:19 +0100125{
126 struct accept_queue_ring *ring = context;
Willy Tarreau83efc322020-10-14 17:37:17 +0200127 struct connection *conn;
Willy Tarreau1efafce2019-01-27 15:37:19 +0100128 struct listener *li;
Christopher Faulet102854c2019-04-30 12:17:13 +0200129 unsigned int max_accept;
Willy Tarreau1efafce2019-01-27 15:37:19 +0100130 int ret;
Willy Tarreau1efafce2019-01-27 15:37:19 +0100131
Christopher Faulet102854c2019-04-30 12:17:13 +0200132 /* if global.tune.maxaccept is -1, then max_accept is UINT_MAX. It
133 * is not really illimited, but it is probably enough.
134 */
Willy Tarreau66161322021-02-19 15:50:27 +0100135 max_accept = global.tune.maxaccept ? global.tune.maxaccept : MAX_ACCEPT;
Christopher Faulet102854c2019-04-30 12:17:13 +0200136 for (; max_accept; max_accept--) {
Willy Tarreau83efc322020-10-14 17:37:17 +0200137 conn = accept_queue_pop_sc(ring);
138 if (!conn)
Willy Tarreau1efafce2019-01-27 15:37:19 +0100139 break;
140
Willy Tarreau83efc322020-10-14 17:37:17 +0200141 li = __objt_listener(conn->target);
Willy Tarreau4781b152021-04-06 13:53:36 +0200142 _HA_ATOMIC_INC(&li->thr_conn[tid]);
Willy Tarreau83efc322020-10-14 17:37:17 +0200143 ret = li->accept(conn);
Willy Tarreau1efafce2019-01-27 15:37:19 +0100144 if (ret <= 0) {
145 /* connection was terminated by the application */
146 continue;
147 }
148
149 /* increase the per-process number of cumulated sessions, this
150 * may only be done once l->accept() has accepted the connection.
151 */
152 if (!(li->options & LI_O_UNLIMITED)) {
153 HA_ATOMIC_UPDATE_MAX(&global.sps_max,
154 update_freq_ctr(&global.sess_per_sec, 1));
155 if (li->bind_conf && li->bind_conf->is_ssl) {
156 HA_ATOMIC_UPDATE_MAX(&global.ssl_max,
157 update_freq_ctr(&global.ssl_per_sec, 1));
158 }
159 }
160 }
161
162 /* ran out of budget ? Let's come here ASAP */
Christopher Faulet102854c2019-04-30 12:17:13 +0200163 if (!max_accept)
Willy Tarreau2bd65a72019-09-24 06:55:18 +0200164 tasklet_wakeup(ring->tasklet);
Willy Tarreau1efafce2019-01-27 15:37:19 +0100165
Willy Tarreau2bd65a72019-09-24 06:55:18 +0200166 return NULL;
Willy Tarreau1efafce2019-01-27 15:37:19 +0100167}
168
169/* Initializes the accept-queues. Returns 0 on success, otherwise ERR_* flags */
170static int accept_queue_init()
171{
Willy Tarreau2bd65a72019-09-24 06:55:18 +0200172 struct tasklet *t;
Willy Tarreau1efafce2019-01-27 15:37:19 +0100173 int i;
174
175 for (i = 0; i < global.nbthread; i++) {
Willy Tarreau2bd65a72019-09-24 06:55:18 +0200176 t = tasklet_new();
Willy Tarreau1efafce2019-01-27 15:37:19 +0100177 if (!t) {
178 ha_alert("Out of memory while initializing accept queue for thread %d\n", i);
179 return ERR_FATAL|ERR_ABORT;
180 }
Willy Tarreau2bd65a72019-09-24 06:55:18 +0200181 t->tid = i;
Willy Tarreau1efafce2019-01-27 15:37:19 +0100182 t->process = accept_queue_process;
183 t->context = &accept_queue_rings[i];
Willy Tarreau2bd65a72019-09-24 06:55:18 +0200184 accept_queue_rings[i].tasklet = t;
Willy Tarreau1efafce2019-01-27 15:37:19 +0100185 }
186 return 0;
187}
188
189REGISTER_CONFIG_POSTPARSER("multi-threaded accept queue", accept_queue_init);
190
Willy Tarreaue01b08d2022-04-27 18:42:47 +0200191static void accept_queue_deinit()
192{
193 int i;
194
195 for (i = 0; i < global.nbthread; i++) {
196 if (accept_queue_rings[i].tasklet)
197 tasklet_free(accept_queue_rings[i].tasklet);
198 }
199}
200
201REGISTER_POST_DEINIT(accept_queue_deinit);
202
Willy Tarreau1efafce2019-01-27 15:37:19 +0100203#endif // USE_THREAD
204
Amaury Denoyellef68b2cb2022-01-25 16:21:47 +0100205/* Memory allocation and initialization of the per_thr field.
206 * Returns 0 if the field has been successfully initialized, -1 on failure.
207 */
208int li_init_per_thr(struct listener *li)
209{
210 int i;
211
212 /* allocate per-thread elements for listener */
213 li->per_thr = calloc(global.nbthread, sizeof(*li->per_thr));
214 if (!li->per_thr)
215 return -1;
216
217 for (i = 0; i < global.nbthread; ++i) {
218 MT_LIST_INIT(&li->per_thr[i].quic_accept.list);
219 MT_LIST_INIT(&li->per_thr[i].quic_accept.conns);
220
221 li->per_thr[i].li = li;
222 }
223
224 return 0;
225}
226
William Dauchy3679d0c2021-02-14 23:22:55 +0100227/* helper to get listener status for stats */
228enum li_status get_li_status(struct listener *l)
229{
230 if (!l->maxconn || l->nbconn < l->maxconn) {
231 if (l->state == LI_LIMITED)
232 return LI_STATUS_WAITING;
233 else
234 return LI_STATUS_OPEN;
235 }
236 return LI_STATUS_FULL;
237}
238
Willy Tarreauefc0eec2020-09-24 07:27:06 +0200239/* adjust the listener's state and its proxy's listener counters if needed.
240 * It must be called under the listener's lock, but uses atomic ops to change
241 * the proxy's counters so that the proxy lock is not needed.
242 */
Willy Tarreaua37b2442020-09-24 07:23:45 +0200243void listener_set_state(struct listener *l, enum li_state st)
244{
Willy Tarreauefc0eec2020-09-24 07:27:06 +0200245 struct proxy *px = l->bind_conf->frontend;
246
247 if (px) {
248 /* from state */
249 switch (l->state) {
250 case LI_NEW: /* first call */
Willy Tarreau4781b152021-04-06 13:53:36 +0200251 _HA_ATOMIC_INC(&px->li_all);
Willy Tarreauefc0eec2020-09-24 07:27:06 +0200252 break;
253 case LI_INIT:
254 case LI_ASSIGNED:
255 break;
256 case LI_PAUSED:
Willy Tarreau4781b152021-04-06 13:53:36 +0200257 _HA_ATOMIC_DEC(&px->li_paused);
Willy Tarreauefc0eec2020-09-24 07:27:06 +0200258 break;
259 case LI_LISTEN:
Willy Tarreau4781b152021-04-06 13:53:36 +0200260 _HA_ATOMIC_DEC(&px->li_bound);
Willy Tarreauefc0eec2020-09-24 07:27:06 +0200261 break;
262 case LI_READY:
263 case LI_FULL:
264 case LI_LIMITED:
Willy Tarreau4781b152021-04-06 13:53:36 +0200265 _HA_ATOMIC_DEC(&px->li_ready);
Willy Tarreauefc0eec2020-09-24 07:27:06 +0200266 break;
267 }
268
269 /* to state */
270 switch (st) {
271 case LI_NEW:
272 case LI_INIT:
273 case LI_ASSIGNED:
274 break;
275 case LI_PAUSED:
Willy Tarreau95a34602020-10-08 15:32:21 +0200276 BUG_ON(l->rx.fd == -1);
Willy Tarreau4781b152021-04-06 13:53:36 +0200277 _HA_ATOMIC_INC(&px->li_paused);
Willy Tarreauefc0eec2020-09-24 07:27:06 +0200278 break;
279 case LI_LISTEN:
Willy Tarreau95a34602020-10-08 15:32:21 +0200280 BUG_ON(l->rx.fd == -1);
Willy Tarreau4781b152021-04-06 13:53:36 +0200281 _HA_ATOMIC_INC(&px->li_bound);
Willy Tarreauefc0eec2020-09-24 07:27:06 +0200282 break;
283 case LI_READY:
284 case LI_FULL:
285 case LI_LIMITED:
Willy Tarreau95a34602020-10-08 15:32:21 +0200286 BUG_ON(l->rx.fd == -1);
Willy Tarreau4781b152021-04-06 13:53:36 +0200287 _HA_ATOMIC_INC(&px->li_ready);
Willy Tarreauefc0eec2020-09-24 07:27:06 +0200288 break;
289 }
290 }
Willy Tarreaua37b2442020-09-24 07:23:45 +0200291 l->state = st;
292}
293
Willy Tarreaudabf2e22007-10-28 21:59:24 +0100294/* This function adds the specified listener's file descriptor to the polling
295 * lists if it is in the LI_LISTEN state. The listener enters LI_READY or
Ilya Shipitsin6fb0f212020-04-02 15:25:26 +0500296 * LI_FULL state depending on its number of connections. In daemon mode, we
Willy Tarreauae302532014-05-07 19:22:24 +0200297 * also support binding only the relevant processes to their respective
298 * listeners. We don't do that in debug mode however.
Willy Tarreaudabf2e22007-10-28 21:59:24 +0100299 */
Willy Tarreau7834a3f2020-09-25 16:40:18 +0200300void enable_listener(struct listener *listener)
Willy Tarreaudabf2e22007-10-28 21:59:24 +0100301{
Willy Tarreau08b6f962022-02-01 16:23:00 +0100302 HA_RWLOCK_WRLOCK(LISTENER_LOCK, &listener->lock);
Willy Tarreaud6afb532020-10-09 10:35:40 +0200303
304 /* If this listener is supposed to be only in the master, close it in
305 * the workers. Conversely, if it's supposed to be only in the workers
306 * close it in the master.
307 */
Willy Tarreau18c20d22020-10-09 16:11:46 +0200308 if (!!master != !!(listener->rx.flags & RX_F_MWORKER))
Willy Tarreau75c98d12020-10-09 15:55:23 +0200309 do_unbind_listener(listener);
Willy Tarreaud6afb532020-10-09 10:35:40 +0200310
Willy Tarreaudabf2e22007-10-28 21:59:24 +0100311 if (listener->state == LI_LISTEN) {
Willy Tarreau95a34602020-10-08 15:32:21 +0200312 BUG_ON(listener->rx.fd == -1);
William Lallemand095ba4c2017-06-01 17:38:50 +0200313 if ((global.mode & (MODE_DAEMON | MODE_MWORKER)) &&
Willy Tarreau72faef32021-06-15 08:36:30 +0200314 (!!master != !!(listener->rx.flags & RX_F_MWORKER))) {
Willy Tarreauae302532014-05-07 19:22:24 +0200315 /* we don't want to enable this listener and don't
316 * want any fd event to reach it.
317 */
Willy Tarreau75c98d12020-10-09 15:55:23 +0200318 do_unbind_listener(listener);
Willy Tarreauae302532014-05-07 19:22:24 +0200319 }
Willy Tarreaua8cf66b2019-02-27 16:49:00 +0100320 else if (!listener->maxconn || listener->nbconn < listener->maxconn) {
Willy Tarreau4b51f422020-09-25 20:32:28 +0200321 listener->rx.proto->enable(listener);
Willy Tarreaua37b2442020-09-24 07:23:45 +0200322 listener_set_state(listener, LI_READY);
Willy Tarreauae302532014-05-07 19:22:24 +0200323 }
324 else {
Willy Tarreaua37b2442020-09-24 07:23:45 +0200325 listener_set_state(listener, LI_FULL);
Willy Tarreaudabf2e22007-10-28 21:59:24 +0100326 }
327 }
Willy Tarreaud6afb532020-10-09 10:35:40 +0200328
Willy Tarreau08b6f962022-02-01 16:23:00 +0100329 HA_RWLOCK_WRUNLOCK(LISTENER_LOCK, &listener->lock);
Willy Tarreaudabf2e22007-10-28 21:59:24 +0100330}
331
Willy Tarreaucaa7df12020-10-07 15:58:50 +0200332/*
333 * This function completely stops a listener. It will need to operate under the
334 * proxy's lock, the protocol's lock, and the listener's lock. The caller is
335 * responsible for indicating in lpx, lpr, lli whether the respective locks are
336 * already held (non-zero) or not (zero) so that the function picks the missing
337 * ones, in this order. The proxy's listeners count is updated and the proxy is
338 * disabled and woken up after the last one is gone.
339 */
340void stop_listener(struct listener *l, int lpx, int lpr, int lli)
341{
342 struct proxy *px = l->bind_conf->frontend;
Willy Tarreaucaa7df12020-10-07 15:58:50 +0200343
344 if (l->options & LI_O_NOSTOP) {
345 /* master-worker sockpairs are never closed but don't count as a
346 * job.
347 */
348 return;
349 }
350
Willy Tarreaucaa7df12020-10-07 15:58:50 +0200351 if (!lpx)
Willy Tarreauac66d6b2020-10-20 17:24:27 +0200352 HA_RWLOCK_WRLOCK(PROXY_LOCK, &px->lock);
Willy Tarreaucaa7df12020-10-07 15:58:50 +0200353
354 if (!lpr)
355 HA_SPIN_LOCK(PROTO_LOCK, &proto_lock);
356
357 if (!lli)
Willy Tarreau08b6f962022-02-01 16:23:00 +0100358 HA_RWLOCK_WRLOCK(LISTENER_LOCK, &l->lock);
Willy Tarreaucaa7df12020-10-07 15:58:50 +0200359
360 if (l->state > LI_INIT) {
Willy Tarreau75c98d12020-10-09 15:55:23 +0200361 do_unbind_listener(l);
Willy Tarreaucaa7df12020-10-07 15:58:50 +0200362
363 if (l->state >= LI_ASSIGNED)
364 __delete_listener(l);
365
Willy Tarreauacde1522020-10-07 16:31:39 +0200366 proxy_cond_disable(px);
Willy Tarreaucaa7df12020-10-07 15:58:50 +0200367 }
368
369 if (!lli)
Willy Tarreau08b6f962022-02-01 16:23:00 +0100370 HA_RWLOCK_WRUNLOCK(LISTENER_LOCK, &l->lock);
Willy Tarreaucaa7df12020-10-07 15:58:50 +0200371
372 if (!lpr)
373 HA_SPIN_UNLOCK(PROTO_LOCK, &proto_lock);
374
375 if (!lpx)
Willy Tarreauac66d6b2020-10-20 17:24:27 +0200376 HA_RWLOCK_WRUNLOCK(PROXY_LOCK, &px->lock);
Willy Tarreaucaa7df12020-10-07 15:58:50 +0200377}
378
Willy Tarreaud1f250f2020-12-04 15:03:36 +0100379/* This function adds the specified <listener> to the protocol <proto>. It
380 * does nothing if the protocol was already added. The listener's state is
381 * automatically updated from LI_INIT to LI_ASSIGNED. The number of listeners
382 * for the protocol is updated. This must be called with the proto lock held.
383 */
384void default_add_listener(struct protocol *proto, struct listener *listener)
385{
386 if (listener->state != LI_INIT)
387 return;
388 listener_set_state(listener, LI_ASSIGNED);
389 listener->rx.proto = proto;
Willy Tarreau2b718102021-04-21 07:32:39 +0200390 LIST_APPEND(&proto->receivers, &listener->rx.proto_list);
Willy Tarreaud1f250f2020-12-04 15:03:36 +0100391 proto->nb_receivers++;
392}
393
Willy Tarreaue03204c2020-10-09 17:02:21 +0200394/* default function called to suspend a listener: it simply passes the call to
395 * the underlying receiver. This is find for most socket-based protocols. This
396 * must be called under the listener's lock. It will return non-zero on success,
397 * 0 on failure. If no receiver-level suspend is provided, the operation is
398 * assumed to succeed.
399 */
400int default_suspend_listener(struct listener *l)
401{
402 int ret = 1;
403
404 if (!l->rx.proto->rx_suspend)
405 return 1;
406
407 ret = l->rx.proto->rx_suspend(&l->rx);
408 return ret > 0 ? ret : 0;
409}
410
411
412/* Tries to resume a suspended listener, and returns non-zero on success or
413 * zero on failure. On certain errors, an alert or a warning might be displayed.
414 * It must be called with the listener's lock held. Depending on the listener's
415 * state and protocol, a listen() call might be used to resume operations, or a
416 * call to the receiver's resume() function might be used as well. This is
417 * suitable as a default function for TCP and UDP. This must be called with the
418 * listener's lock held.
419 */
420int default_resume_listener(struct listener *l)
421{
422 int ret = 1;
423
424 if (l->state == LI_ASSIGNED) {
425 char msg[100];
426 int err;
427
428 err = l->rx.proto->listen(l, msg, sizeof(msg));
429 if (err & ERR_ALERT)
430 ha_alert("Resuming listener: %s\n", msg);
431 else if (err & ERR_WARN)
432 ha_warning("Resuming listener: %s\n", msg);
433
434 if (err & (ERR_FATAL | ERR_ABORT)) {
435 ret = 0;
436 goto end;
437 }
438 }
439
440 if (l->state < LI_PAUSED) {
441 ret = 0;
442 goto end;
443 }
444
445 if (l->state == LI_PAUSED && l->rx.proto->rx_resume &&
446 l->rx.proto->rx_resume(&l->rx) <= 0)
447 ret = 0;
448 end:
449 return ret;
450}
451
452
Willy Tarreaube58c382011-07-24 18:28:10 +0200453/* This function tries to temporarily disable a listener, depending on the OS
454 * capabilities. Linux unbinds the listen socket after a SHUT_RD, and ignores
455 * SHUT_WR. Solaris refuses either shutdown(). OpenBSD ignores SHUT_RD but
456 * closes upon SHUT_WR and refuses to rebind. So a common validation path
457 * involves SHUT_WR && listen && SHUT_RD. In case of success, the FD's polling
458 * is disabled. It normally returns non-zero, unless an error is reported.
459 */
460int pause_listener(struct listener *l)
461{
Willy Tarreau58651b42020-09-24 16:03:29 +0200462 struct proxy *px = l->bind_conf->frontend;
Christopher Faulet8d8aa0d2017-05-30 15:36:50 +0200463 int ret = 1;
464
Willy Tarreau08b6f962022-02-01 16:23:00 +0100465 HA_RWLOCK_WRLOCK(LISTENER_LOCK, &l->lock);
Christopher Faulet8d8aa0d2017-05-30 15:36:50 +0200466
Willy Tarreau9b3a9322020-09-24 14:46:34 +0200467 if (l->state <= LI_PAUSED)
468 goto end;
469
Willy Tarreaue03204c2020-10-09 17:02:21 +0200470 if (l->rx.proto->suspend)
471 ret = l->rx.proto->suspend(l);
Willy Tarreaube58c382011-07-24 18:28:10 +0200472
Willy Tarreau2b718102021-04-21 07:32:39 +0200473 MT_LIST_DELETE(&l->wait_queue);
Willy Tarreaue6ca1fc2011-07-24 22:03:52 +0200474
Willy Tarreaua37b2442020-09-24 07:23:45 +0200475 listener_set_state(l, LI_PAUSED);
Willy Tarreau58651b42020-09-24 16:03:29 +0200476
477 if (px && !px->li_ready) {
478 ha_warning("Paused %s %s.\n", proxy_cap_str(px->cap), px->id);
479 send_log(px, LOG_WARNING, "Paused %s %s.\n", proxy_cap_str(px->cap), px->id);
480 }
Christopher Faulet8d8aa0d2017-05-30 15:36:50 +0200481 end:
Willy Tarreau08b6f962022-02-01 16:23:00 +0100482 HA_RWLOCK_WRUNLOCK(LISTENER_LOCK, &l->lock);
Christopher Faulet8d8aa0d2017-05-30 15:36:50 +0200483 return ret;
Willy Tarreaube58c382011-07-24 18:28:10 +0200484}
485
Willy Tarreaue6ca1fc2011-07-24 22:03:52 +0200486/* This function tries to resume a temporarily disabled listener. Paused, full,
487 * limited and disabled listeners are handled, which means that this function
488 * may replace enable_listener(). The resulting state will either be LI_READY
489 * or LI_FULL. 0 is returned in case of failure to resume (eg: dead socket).
Willy Tarreauae302532014-05-07 19:22:24 +0200490 * Listeners bound to a different process are not woken up unless we're in
Willy Tarreauaf2fd582015-04-14 12:07:16 +0200491 * foreground mode, and are ignored. If the listener was only in the assigned
492 * state, it's totally rebound. This can happen if a pause() has completely
493 * stopped it. If the resume fails, 0 is returned and an error might be
494 * displayed.
Willy Tarreaube58c382011-07-24 18:28:10 +0200495 */
Willy Tarreau01abd022019-02-28 10:27:18 +0100496int resume_listener(struct listener *l)
Willy Tarreaube58c382011-07-24 18:28:10 +0200497{
Willy Tarreau58651b42020-09-24 16:03:29 +0200498 struct proxy *px = l->bind_conf->frontend;
499 int was_paused = px && px->li_paused;
Christopher Faulet8d8aa0d2017-05-30 15:36:50 +0200500 int ret = 1;
501
Willy Tarreau08b6f962022-02-01 16:23:00 +0100502 HA_RWLOCK_WRLOCK(LISTENER_LOCK, &l->lock);
Christopher Faulet8d8aa0d2017-05-30 15:36:50 +0200503
Willy Tarreauf2cb1692019-07-11 10:08:31 +0200504 /* check that another thread didn't to the job in parallel (e.g. at the
505 * end of listen_accept() while we'd come from dequeue_all_listeners().
506 */
Willy Tarreau2b718102021-04-21 07:32:39 +0200507 if (MT_LIST_INLIST(&l->wait_queue))
Willy Tarreauf2cb1692019-07-11 10:08:31 +0200508 goto end;
509
Willy Tarreau5d7f9ce2020-09-24 18:54:11 +0200510 if (l->state == LI_READY)
511 goto end;
Willy Tarreaube58c382011-07-24 18:28:10 +0200512
Willy Tarreaue03204c2020-10-09 17:02:21 +0200513 if (l->rx.proto->resume)
514 ret = l->rx.proto->resume(l);
Willy Tarreaube58c382011-07-24 18:28:10 +0200515
Willy Tarreaua8cf66b2019-02-27 16:49:00 +0100516 if (l->maxconn && l->nbconn >= l->maxconn) {
Willy Tarreau4b51f422020-09-25 20:32:28 +0200517 l->rx.proto->disable(l);
Willy Tarreaua37b2442020-09-24 07:23:45 +0200518 listener_set_state(l, LI_FULL);
Willy Tarreau58651b42020-09-24 16:03:29 +0200519 goto done;
Willy Tarreaube58c382011-07-24 18:28:10 +0200520 }
521
Willy Tarreau4b51f422020-09-25 20:32:28 +0200522 l->rx.proto->enable(l);
Willy Tarreaua37b2442020-09-24 07:23:45 +0200523 listener_set_state(l, LI_READY);
Willy Tarreau58651b42020-09-24 16:03:29 +0200524
525 done:
526 if (was_paused && !px->li_paused) {
527 ha_warning("Resumed %s %s.\n", proxy_cap_str(px->cap), px->id);
528 send_log(px, LOG_WARNING, "Resumed %s %s.\n", proxy_cap_str(px->cap), px->id);
529 }
Christopher Faulet8d8aa0d2017-05-30 15:36:50 +0200530 end:
Willy Tarreau08b6f962022-02-01 16:23:00 +0100531 HA_RWLOCK_WRUNLOCK(LISTENER_LOCK, &l->lock);
Christopher Faulet8d8aa0d2017-05-30 15:36:50 +0200532 return ret;
533}
534
Willy Tarreau87b09662015-04-03 00:22:06 +0200535/* Marks a ready listener as full so that the stream code tries to re-enable
Willy Tarreau62793712011-07-24 19:23:38 +0200536 * it upon next close() using resume_listener().
537 */
Christopher Faulet5580ba22017-08-28 15:29:20 +0200538static void listener_full(struct listener *l)
Willy Tarreau62793712011-07-24 19:23:38 +0200539{
Willy Tarreau08b6f962022-02-01 16:23:00 +0100540 HA_RWLOCK_WRLOCK(LISTENER_LOCK, &l->lock);
Willy Tarreau62793712011-07-24 19:23:38 +0200541 if (l->state >= LI_READY) {
Willy Tarreau2b718102021-04-21 07:32:39 +0200542 MT_LIST_DELETE(&l->wait_queue);
Willy Tarreau3f0d02b2019-02-25 19:23:37 +0100543 if (l->state != LI_FULL) {
Willy Tarreau4b51f422020-09-25 20:32:28 +0200544 l->rx.proto->disable(l);
Willy Tarreaua37b2442020-09-24 07:23:45 +0200545 listener_set_state(l, LI_FULL);
Willy Tarreau3f0d02b2019-02-25 19:23:37 +0100546 }
Willy Tarreau62793712011-07-24 19:23:38 +0200547 }
Willy Tarreau08b6f962022-02-01 16:23:00 +0100548 HA_RWLOCK_WRUNLOCK(LISTENER_LOCK, &l->lock);
Willy Tarreau62793712011-07-24 19:23:38 +0200549}
550
Willy Tarreaue6ca1fc2011-07-24 22:03:52 +0200551/* Marks a ready listener as limited so that we only try to re-enable it when
552 * resources are free again. It will be queued into the specified queue.
553 */
Olivier Houchard859dc802019-08-08 15:47:21 +0200554static void limit_listener(struct listener *l, struct mt_list *list)
Willy Tarreaue6ca1fc2011-07-24 22:03:52 +0200555{
Willy Tarreau08b6f962022-02-01 16:23:00 +0100556 HA_RWLOCK_WRLOCK(LISTENER_LOCK, &l->lock);
Willy Tarreaue6ca1fc2011-07-24 22:03:52 +0200557 if (l->state == LI_READY) {
Willy Tarreau2b718102021-04-21 07:32:39 +0200558 MT_LIST_TRY_APPEND(list, &l->wait_queue);
Willy Tarreau4b51f422020-09-25 20:32:28 +0200559 l->rx.proto->disable(l);
Willy Tarreaua37b2442020-09-24 07:23:45 +0200560 listener_set_state(l, LI_LIMITED);
Willy Tarreaue6ca1fc2011-07-24 22:03:52 +0200561 }
Willy Tarreau08b6f962022-02-01 16:23:00 +0100562 HA_RWLOCK_WRUNLOCK(LISTENER_LOCK, &l->lock);
Willy Tarreaue6ca1fc2011-07-24 22:03:52 +0200563}
564
Willy Tarreau241797a2019-12-10 14:10:52 +0100565/* Dequeues all listeners waiting for a resource the global wait queue */
566void dequeue_all_listeners()
Willy Tarreaue6ca1fc2011-07-24 22:03:52 +0200567{
Willy Tarreau01abd022019-02-28 10:27:18 +0100568 struct listener *listener;
Willy Tarreaue6ca1fc2011-07-24 22:03:52 +0200569
Willy Tarreau241797a2019-12-10 14:10:52 +0100570 while ((listener = MT_LIST_POP(&global_listener_queue, struct listener *, wait_queue))) {
Willy Tarreaue6ca1fc2011-07-24 22:03:52 +0200571 /* This cannot fail because the listeners are by definition in
Willy Tarreau01abd022019-02-28 10:27:18 +0100572 * the LI_LIMITED state.
Willy Tarreaue6ca1fc2011-07-24 22:03:52 +0200573 */
Willy Tarreau01abd022019-02-28 10:27:18 +0100574 resume_listener(listener);
Willy Tarreaue6ca1fc2011-07-24 22:03:52 +0200575 }
576}
577
Willy Tarreau241797a2019-12-10 14:10:52 +0100578/* Dequeues all listeners waiting for a resource in proxy <px>'s queue */
579void dequeue_proxy_listeners(struct proxy *px)
580{
581 struct listener *listener;
582
583 while ((listener = MT_LIST_POP(&px->listener_queue, struct listener *, wait_queue))) {
584 /* This cannot fail because the listeners are by definition in
585 * the LI_LIMITED state.
586 */
587 resume_listener(listener);
588 }
589}
590
Willy Tarreau7b2febd2020-10-09 17:18:29 +0200591
592/* default function used to unbind a listener. This is for use by standard
593 * protocols working on top of accepted sockets. The receiver's rx_unbind()
594 * will automatically be used after the listener is disabled if the socket is
595 * still bound. This must be used under the listener's lock.
Christopher Faulet510c0d62018-03-16 10:04:47 +0100596 */
Willy Tarreau7b2febd2020-10-09 17:18:29 +0200597void default_unbind_listener(struct listener *listener)
Willy Tarreaub648d632007-10-28 22:13:50 +0100598{
Willy Tarreau87acd4e2020-10-08 15:36:46 +0200599 if (listener->state <= LI_ASSIGNED)
600 goto out_close;
601
602 if (listener->rx.fd == -1) {
Willy Tarreaua37b2442020-09-24 07:23:45 +0200603 listener_set_state(listener, LI_ASSIGNED);
Willy Tarreau87acd4e2020-10-08 15:36:46 +0200604 goto out_close;
605 }
606
Willy Tarreauf58b8db2020-10-09 16:32:08 +0200607 if (listener->state >= LI_READY) {
608 listener->rx.proto->disable(listener);
609 if (listener->rx.flags & RX_F_BOUND)
Willy Tarreau87acd4e2020-10-08 15:36:46 +0200610 listener_set_state(listener, LI_LISTEN);
Willy Tarreaub6607bf2020-09-23 16:24:23 +0200611 }
612
Willy Tarreau87acd4e2020-10-08 15:36:46 +0200613 out_close:
Willy Tarreauf58b8db2020-10-09 16:32:08 +0200614 if (listener->rx.flags & RX_F_BOUND)
615 listener->rx.proto->rx_unbind(&listener->rx);
Willy Tarreau7b2febd2020-10-09 17:18:29 +0200616}
617
618/* This function closes the listening socket for the specified listener,
619 * provided that it's already in a listening state. The protocol's unbind()
620 * is called to put the listener into LI_ASSIGNED or LI_LISTEN and handle
621 * the unbinding tasks. The listener enters then the LI_ASSIGNED state if
622 * the receiver is unbound. Must be called with the lock held.
623 */
624void do_unbind_listener(struct listener *listener)
625{
Willy Tarreau2b718102021-04-21 07:32:39 +0200626 MT_LIST_DELETE(&listener->wait_queue);
Willy Tarreau7b2febd2020-10-09 17:18:29 +0200627
628 if (listener->rx.proto->unbind)
629 listener->rx.proto->unbind(listener);
Willy Tarreau374e9af2020-10-09 15:47:17 +0200630
Willy Tarreauf58b8db2020-10-09 16:32:08 +0200631 /* we may have to downgrade the listener if the rx was closed */
632 if (!(listener->rx.flags & RX_F_BOUND) && listener->state > LI_ASSIGNED)
Willy Tarreau374e9af2020-10-09 15:47:17 +0200633 listener_set_state(listener, LI_ASSIGNED);
Willy Tarreaubbd09b92017-11-05 11:38:44 +0100634}
635
Olivier Houchard1fc05162017-04-06 01:05:05 +0200636/* This function closes the listening socket for the specified listener,
637 * provided that it's already in a listening state. The listener enters the
Willy Tarreau75c98d12020-10-09 15:55:23 +0200638 * LI_ASSIGNED state, except if the FD is not closed, in which case it may
639 * remain in LI_LISTEN. This function is intended to be used as a generic
Willy Tarreaubbd09b92017-11-05 11:38:44 +0100640 * function for standard protocols.
Olivier Houchard1fc05162017-04-06 01:05:05 +0200641 */
Willy Tarreaubbd09b92017-11-05 11:38:44 +0100642void unbind_listener(struct listener *listener)
Olivier Houchard1fc05162017-04-06 01:05:05 +0200643{
Willy Tarreau08b6f962022-02-01 16:23:00 +0100644 HA_RWLOCK_WRLOCK(LISTENER_LOCK, &listener->lock);
Willy Tarreau75c98d12020-10-09 15:55:23 +0200645 do_unbind_listener(listener);
Willy Tarreau08b6f962022-02-01 16:23:00 +0100646 HA_RWLOCK_WRUNLOCK(LISTENER_LOCK, &listener->lock);
Olivier Houchard1fc05162017-04-06 01:05:05 +0200647}
648
Willy Tarreau0de59fd2017-09-15 08:10:44 +0200649/* creates one or multiple listeners for bind_conf <bc> on sockaddr <ss> on port
650 * range <portl> to <porth>, and possibly attached to fd <fd> (or -1 for auto
Willy Tarreau9b3178d2020-09-16 17:58:55 +0200651 * allocation). The address family is taken from ss->ss_family, and the protocol
Willy Tarreaud2fb99f2020-10-15 21:22:29 +0200652 * passed in <proto> must be usable on this family. The protocol's default iocb
653 * is automatically preset as the receivers' iocb. The number of jobs and
Willy Tarreau9b3178d2020-09-16 17:58:55 +0200654 * listeners is automatically increased by the number of listeners created. It
655 * returns non-zero on success, zero on error with the error message set in <err>.
Willy Tarreau0de59fd2017-09-15 08:10:44 +0200656 */
657int create_listeners(struct bind_conf *bc, const struct sockaddr_storage *ss,
Willy Tarreau9b3178d2020-09-16 17:58:55 +0200658 int portl, int porth, int fd, struct protocol *proto, char **err)
Willy Tarreau0de59fd2017-09-15 08:10:44 +0200659{
Willy Tarreau0de59fd2017-09-15 08:10:44 +0200660 struct listener *l;
661 int port;
662
Willy Tarreau0de59fd2017-09-15 08:10:44 +0200663 for (port = portl; port <= porth; port++) {
664 l = calloc(1, sizeof(*l));
665 if (!l) {
666 memprintf(err, "out of memory");
667 return 0;
668 }
669 l->obj_type = OBJ_TYPE_LISTENER;
Willy Tarreau2b718102021-04-21 07:32:39 +0200670 LIST_APPEND(&bc->frontend->conf.listeners, &l->by_fe);
671 LIST_APPEND(&bc->listeners, &l->by_bind);
Willy Tarreau0de59fd2017-09-15 08:10:44 +0200672 l->bind_conf = bc;
Willy Tarreau0fce6bc2020-09-03 07:46:06 +0200673 l->rx.settings = &bc->settings;
Willy Tarreaueef45422020-09-03 10:05:03 +0200674 l->rx.owner = l;
Willy Tarreaud2fb99f2020-10-15 21:22:29 +0200675 l->rx.iocb = proto->default_iocb;
Willy Tarreau38ba6472020-08-27 08:16:52 +0200676 l->rx.fd = fd;
Willy Tarreau07400c52020-12-04 14:49:11 +0100677
Willy Tarreau37159062020-08-27 07:48:42 +0200678 memcpy(&l->rx.addr, ss, sizeof(*ss));
Willy Tarreaud1f250f2020-12-04 15:03:36 +0100679 if (proto->fam->set_port)
680 proto->fam->set_port(&l->rx.addr, port);
Willy Tarreau07400c52020-12-04 14:49:11 +0100681
Olivier Houchard859dc802019-08-08 15:47:21 +0200682 MT_LIST_INIT(&l->wait_queue);
Willy Tarreaua37b2442020-09-24 07:23:45 +0200683 listener_set_state(l, LI_INIT);
Willy Tarreau0de59fd2017-09-15 08:10:44 +0200684
Willy Tarreaud1f250f2020-12-04 15:03:36 +0100685 proto->add(proto, l);
Willy Tarreau0de59fd2017-09-15 08:10:44 +0200686
Willy Tarreau909c23b2020-09-15 13:50:58 +0200687 if (fd != -1)
Willy Tarreau43046fa2020-09-01 15:41:59 +0200688 l->rx.flags |= RX_F_INHERITED;
William Lallemand75ea0a02017-11-15 19:02:58 +0100689
Amaury Denoyelle7f8f6cb2020-11-10 14:24:31 +0100690 l->extra_counters = NULL;
691
Willy Tarreau08b6f962022-02-01 16:23:00 +0100692 HA_RWLOCK_INIT(&l->lock);
Willy Tarreau4781b152021-04-06 13:53:36 +0200693 _HA_ATOMIC_INC(&jobs);
694 _HA_ATOMIC_INC(&listeners);
Willy Tarreau0de59fd2017-09-15 08:10:44 +0200695 }
696 return 1;
697}
698
Willy Tarreau59a877d2021-10-12 09:36:10 +0200699/* clones listener <src> and returns the new one. All dynamically allocated
700 * fields are reallocated (name for now). The new listener is inserted before
701 * the original one in the bind_conf and frontend lists. This allows it to be
702 * duplicated while iterating over the current list. The original listener must
703 * only be in the INIT or ASSIGNED states, and the new listener will only be
704 * placed into the INIT state. The counters are always set to NULL. Maxsock is
705 * updated. Returns NULL on allocation error.
706 */
707struct listener *clone_listener(struct listener *src)
708{
709 struct listener *l;
710
711 l = calloc(1, sizeof(*l));
712 if (!l)
713 goto oom1;
714 memcpy(l, src, sizeof(*l));
715
716 if (l->name) {
717 l->name = strdup(l->name);
718 if (!l->name)
719 goto oom2;
720 }
721
722 l->rx.owner = l;
723 l->state = LI_INIT;
724 l->counters = NULL;
725 l->extra_counters = NULL;
726
727 LIST_APPEND(&src->by_fe, &l->by_fe);
728 LIST_APPEND(&src->by_bind, &l->by_bind);
729
730 MT_LIST_INIT(&l->wait_queue);
731
732 l->rx.proto->add(l->rx.proto, l);
733
Willy Tarreau08b6f962022-02-01 16:23:00 +0100734 HA_RWLOCK_INIT(&l->lock);
Willy Tarreau59a877d2021-10-12 09:36:10 +0200735 _HA_ATOMIC_INC(&jobs);
736 _HA_ATOMIC_INC(&listeners);
737 global.maxsock++;
738 return l;
739
Willy Tarreau59a877d2021-10-12 09:36:10 +0200740 oom2:
741 free(l);
742 oom1:
Willy Tarreaua1462892021-10-16 14:45:29 +0200743 return NULL;
Willy Tarreau59a877d2021-10-12 09:36:10 +0200744}
745
Willy Tarreau1a64d162007-10-28 22:26:05 +0100746/* Delete a listener from its protocol's list of listeners. The listener's
747 * state is automatically updated from LI_ASSIGNED to LI_INIT. The protocol's
Willy Tarreau2cc5bae2017-09-15 08:18:11 +0200748 * number of listeners is updated, as well as the global number of listeners
749 * and jobs. Note that the listener must have previously been unbound. This
Willy Tarreaub4c083f2020-10-07 15:36:16 +0200750 * is a low-level function expected to be called with the proto_lock and the
751 * listener's lock held.
Willy Tarreau1a64d162007-10-28 22:26:05 +0100752 */
Willy Tarreaub4c083f2020-10-07 15:36:16 +0200753void __delete_listener(struct listener *listener)
Willy Tarreau1a64d162007-10-28 22:26:05 +0100754{
Willy Tarreau3f0d02b2019-02-25 19:23:37 +0100755 if (listener->state == LI_ASSIGNED) {
Willy Tarreaua37b2442020-09-24 07:23:45 +0200756 listener_set_state(listener, LI_INIT);
Willy Tarreau2b718102021-04-21 07:32:39 +0200757 LIST_DELETE(&listener->rx.proto_list);
Willy Tarreaud7f331c2020-09-25 17:01:43 +0200758 listener->rx.proto->nb_receivers--;
Willy Tarreau4781b152021-04-06 13:53:36 +0200759 _HA_ATOMIC_DEC(&jobs);
760 _HA_ATOMIC_DEC(&listeners);
Willy Tarreau3f0d02b2019-02-25 19:23:37 +0100761 }
Willy Tarreaub4c083f2020-10-07 15:36:16 +0200762}
763
764/* Delete a listener from its protocol's list of listeners (please check
765 * __delete_listener() above). The proto_lock and the listener's lock will
766 * be grabbed in this order.
767 */
768void delete_listener(struct listener *listener)
769{
770 HA_SPIN_LOCK(PROTO_LOCK, &proto_lock);
Willy Tarreau08b6f962022-02-01 16:23:00 +0100771 HA_RWLOCK_WRLOCK(LISTENER_LOCK, &listener->lock);
Willy Tarreaub4c083f2020-10-07 15:36:16 +0200772 __delete_listener(listener);
Willy Tarreau08b6f962022-02-01 16:23:00 +0100773 HA_RWLOCK_WRUNLOCK(LISTENER_LOCK, &listener->lock);
Willy Tarreau6ee9f8d2019-08-26 10:55:52 +0200774 HA_SPIN_UNLOCK(PROTO_LOCK, &proto_lock);
Willy Tarreau1a64d162007-10-28 22:26:05 +0100775}
776
Willy Tarreaue2711c72019-02-27 15:39:41 +0100777/* Returns a suitable value for a listener's backlog. It uses the listener's,
778 * otherwise the frontend's backlog, otherwise the listener's maxconn,
779 * otherwise the frontend's maxconn, otherwise 1024.
780 */
781int listener_backlog(const struct listener *l)
782{
783 if (l->backlog)
784 return l->backlog;
785
786 if (l->bind_conf->frontend->backlog)
787 return l->bind_conf->frontend->backlog;
788
789 if (l->maxconn)
790 return l->maxconn;
791
792 if (l->bind_conf->frontend->maxconn)
793 return l->bind_conf->frontend->maxconn;
794
795 return 1024;
796}
797
Willy Tarreaubbebbbf2012-05-07 21:22:09 +0200798/* This function is called on a read event from a listening socket, corresponding
799 * to an accept. It tries to accept as many connections as possible, and for each
800 * calls the listener's accept handler (generally the frontend's accept handler).
801 */
Willy Tarreaua74cb382020-10-15 21:29:49 +0200802void listener_accept(struct listener *l)
Willy Tarreaubbebbbf2012-05-07 21:22:09 +0200803{
Willy Tarreau83efc322020-10-14 17:37:17 +0200804 struct connection *cli_conn;
Olivier Houchardd16a9df2019-02-25 16:18:16 +0100805 struct proxy *p;
Christopher Faulet102854c2019-04-30 12:17:13 +0200806 unsigned int max_accept;
Willy Tarreau3f0d02b2019-02-25 19:23:37 +0100807 int next_conn = 0;
Willy Tarreau82c97892019-02-27 19:32:32 +0100808 int next_feconn = 0;
809 int next_actconn = 0;
Willy Tarreaubb660302014-05-07 19:47:02 +0200810 int expire;
Willy Tarreaubbebbbf2012-05-07 21:22:09 +0200811 int ret;
812
Olivier Houchardd16a9df2019-02-25 16:18:16 +0100813 p = l->bind_conf->frontend;
Christopher Faulet102854c2019-04-30 12:17:13 +0200814
815 /* if l->maxaccept is -1, then max_accept is UINT_MAX. It is not really
816 * illimited, but it is probably enough.
817 */
Olivier Houchardd16a9df2019-02-25 16:18:16 +0100818 max_accept = l->maxaccept ? l->maxaccept : 1;
Christopher Faulet8d8aa0d2017-05-30 15:36:50 +0200819
Willy Tarreau93e7c002013-10-07 18:51:07 +0200820 if (!(l->options & LI_O_UNLIMITED) && global.sps_lim) {
821 int max = freq_ctr_remain(&global.sess_per_sec, global.sps_lim, 0);
Willy Tarreau93e7c002013-10-07 18:51:07 +0200822
823 if (unlikely(!max)) {
824 /* frontend accept rate limit was reached */
Willy Tarreau93e7c002013-10-07 18:51:07 +0200825 expire = tick_add(now_ms, next_event_delay(&global.sess_per_sec, global.sps_lim, 0));
Willy Tarreau0591bf72019-12-10 12:01:21 +0100826 goto limit_global;
Willy Tarreau93e7c002013-10-07 18:51:07 +0200827 }
828
829 if (max_accept > max)
830 max_accept = max;
831 }
832
833 if (!(l->options & LI_O_UNLIMITED) && global.cps_lim) {
Willy Tarreaubbebbbf2012-05-07 21:22:09 +0200834 int max = freq_ctr_remain(&global.conn_per_sec, global.cps_lim, 0);
835
836 if (unlikely(!max)) {
837 /* frontend accept rate limit was reached */
Willy Tarreau93e7c002013-10-07 18:51:07 +0200838 expire = tick_add(now_ms, next_event_delay(&global.conn_per_sec, global.cps_lim, 0));
Willy Tarreau0591bf72019-12-10 12:01:21 +0100839 goto limit_global;
Willy Tarreaubbebbbf2012-05-07 21:22:09 +0200840 }
841
842 if (max_accept > max)
843 max_accept = max;
844 }
Willy Tarreaue43d5322013-10-07 20:01:52 +0200845#ifdef USE_OPENSSL
846 if (!(l->options & LI_O_UNLIMITED) && global.ssl_lim && l->bind_conf && l->bind_conf->is_ssl) {
847 int max = freq_ctr_remain(&global.ssl_per_sec, global.ssl_lim, 0);
Willy Tarreaubbebbbf2012-05-07 21:22:09 +0200848
Willy Tarreaue43d5322013-10-07 20:01:52 +0200849 if (unlikely(!max)) {
850 /* frontend accept rate limit was reached */
Willy Tarreaue43d5322013-10-07 20:01:52 +0200851 expire = tick_add(now_ms, next_event_delay(&global.ssl_per_sec, global.ssl_lim, 0));
Willy Tarreau0591bf72019-12-10 12:01:21 +0100852 goto limit_global;
Willy Tarreaue43d5322013-10-07 20:01:52 +0200853 }
854
855 if (max_accept > max)
856 max_accept = max;
857 }
858#endif
Willy Tarreaubbebbbf2012-05-07 21:22:09 +0200859 if (p && p->fe_sps_lim) {
860 int max = freq_ctr_remain(&p->fe_sess_per_sec, p->fe_sps_lim, 0);
861
862 if (unlikely(!max)) {
863 /* frontend accept rate limit was reached */
Willy Tarreau0591bf72019-12-10 12:01:21 +0100864 expire = tick_add(now_ms, next_event_delay(&p->fe_sess_per_sec, p->fe_sps_lim, 0));
865 goto limit_proxy;
Willy Tarreaubbebbbf2012-05-07 21:22:09 +0200866 }
867
868 if (max_accept > max)
869 max_accept = max;
870 }
871
872 /* Note: if we fail to allocate a connection because of configured
873 * limits, we'll schedule a new attempt worst 1 second later in the
874 * worst case. If we fail due to system limits or temporary resource
875 * shortage, we try again 100ms later in the worst case.
876 */
Willy Tarreau02757d02021-01-28 18:07:24 +0100877 for (; max_accept; next_conn = next_feconn = next_actconn = 0, max_accept--) {
Christopher Faulet8d8aa0d2017-05-30 15:36:50 +0200878 unsigned int count;
Willy Tarreau9378bbe2020-10-15 10:09:31 +0200879 int status;
Willy Tarreau0aa5a5b2020-10-16 17:43:04 +0200880 __decl_thread(unsigned long mask);
Willy Tarreaubbebbbf2012-05-07 21:22:09 +0200881
Willy Tarreau82c97892019-02-27 19:32:32 +0100882 /* pre-increase the number of connections without going too far.
883 * We process the listener, then the proxy, then the process.
884 * We know which ones to unroll based on the next_xxx value.
885 */
Willy Tarreau3f0d02b2019-02-25 19:23:37 +0100886 do {
887 count = l->nbconn;
Willy Tarreau93604ed2019-11-15 10:20:07 +0100888 if (unlikely(l->maxconn && count >= l->maxconn)) {
Willy Tarreau3f0d02b2019-02-25 19:23:37 +0100889 /* the listener was marked full or another
890 * thread is going to do it.
891 */
892 next_conn = 0;
Willy Tarreau93604ed2019-11-15 10:20:07 +0100893 listener_full(l);
Willy Tarreau3f0d02b2019-02-25 19:23:37 +0100894 goto end;
895 }
896 next_conn = count + 1;
David Carlier56716622019-03-27 16:08:42 +0000897 } while (!_HA_ATOMIC_CAS(&l->nbconn, (int *)(&count), next_conn));
Willy Tarreau3f0d02b2019-02-25 19:23:37 +0100898
Willy Tarreau82c97892019-02-27 19:32:32 +0100899 if (p) {
900 do {
901 count = p->feconn;
Willy Tarreau93604ed2019-11-15 10:20:07 +0100902 if (unlikely(count >= p->maxconn)) {
Willy Tarreau82c97892019-02-27 19:32:32 +0100903 /* the frontend was marked full or another
904 * thread is going to do it.
905 */
906 next_feconn = 0;
Willy Tarreau0591bf72019-12-10 12:01:21 +0100907 expire = TICK_ETERNITY;
908 goto limit_proxy;
Willy Tarreau82c97892019-02-27 19:32:32 +0100909 }
910 next_feconn = count + 1;
Olivier Houchard64213e92019-03-08 18:52:57 +0100911 } while (!_HA_ATOMIC_CAS(&p->feconn, &count, next_feconn));
Willy Tarreaubbebbbf2012-05-07 21:22:09 +0200912 }
913
Willy Tarreau82c97892019-02-27 19:32:32 +0100914 if (!(l->options & LI_O_UNLIMITED)) {
915 do {
916 count = actconn;
Willy Tarreau93604ed2019-11-15 10:20:07 +0100917 if (unlikely(count >= global.maxconn)) {
Willy Tarreau82c97892019-02-27 19:32:32 +0100918 /* the process was marked full or another
919 * thread is going to do it.
920 */
921 next_actconn = 0;
Willy Tarreau0591bf72019-12-10 12:01:21 +0100922 expire = tick_add(now_ms, 1000); /* try again in 1 second */
923 goto limit_global;
Willy Tarreau82c97892019-02-27 19:32:32 +0100924 }
925 next_actconn = count + 1;
David Carlier56716622019-03-27 16:08:42 +0000926 } while (!_HA_ATOMIC_CAS(&actconn, (int *)(&count), next_actconn));
Willy Tarreaubbebbbf2012-05-07 21:22:09 +0200927 }
928
Willy Tarreaufed93d32022-02-01 16:37:00 +0100929 /* be careful below, the listener might be shutting down in
930 * another thread on error and we must not dereference its
931 * FD without a bit of protection.
932 */
933 cli_conn = NULL;
934 status = CO_AC_PERMERR;
935
936 HA_RWLOCK_RDLOCK(LISTENER_LOCK, &l->lock);
937 if (l->rx.flags & RX_F_BOUND)
938 cli_conn = l->rx.proto->accept_conn(l, &status);
939 HA_RWLOCK_RDUNLOCK(LISTENER_LOCK, &l->lock);
940
Willy Tarreau9378bbe2020-10-15 10:09:31 +0200941 if (!cli_conn) {
942 switch (status) {
943 case CO_AC_DONE:
944 goto end;
Willy Tarreau818dca52014-01-31 19:40:19 +0100945
Willy Tarreau9378bbe2020-10-15 10:09:31 +0200946 case CO_AC_RETRY: /* likely a signal */
Willy Tarreau4781b152021-04-06 13:53:36 +0200947 _HA_ATOMIC_DEC(&l->nbconn);
Willy Tarreau82c97892019-02-27 19:32:32 +0100948 if (p)
Willy Tarreau4781b152021-04-06 13:53:36 +0200949 _HA_ATOMIC_DEC(&p->feconn);
Willy Tarreau82c97892019-02-27 19:32:32 +0100950 if (!(l->options & LI_O_UNLIMITED))
Willy Tarreau4781b152021-04-06 13:53:36 +0200951 _HA_ATOMIC_DEC(&actconn);
Willy Tarreaua593ec52014-01-20 21:21:30 +0100952 continue;
Willy Tarreau9378bbe2020-10-15 10:09:31 +0200953
954 case CO_AC_YIELD:
Willy Tarreau92079932019-12-10 09:30:05 +0100955 max_accept = 0;
956 goto end;
William Lallemandd9138002018-11-27 12:02:39 +0100957
Willy Tarreau9378bbe2020-10-15 10:09:31 +0200958 default:
959 goto transient_error;
Willy Tarreau83efc322020-10-14 17:37:17 +0200960 }
961 }
962
Willy Tarreau3f0d02b2019-02-25 19:23:37 +0100963 /* The connection was accepted, it must be counted as such */
964 if (l->counters)
965 HA_ATOMIC_UPDATE_MAX(&l->counters->conn_max, next_conn);
966
Willy Tarreaud8679342022-05-09 20:41:54 +0200967 if (p) {
Willy Tarreau82c97892019-02-27 19:32:32 +0100968 HA_ATOMIC_UPDATE_MAX(&p->fe_counters.conn_max, next_feconn);
Willy Tarreaud8679342022-05-09 20:41:54 +0200969 proxy_inc_fe_conn_ctr(l, p);
970 }
Willy Tarreau82c97892019-02-27 19:32:32 +0100971
Willy Tarreau3f0d02b2019-02-25 19:23:37 +0100972 if (!(l->options & LI_O_UNLIMITED)) {
973 count = update_freq_ctr(&global.conn_per_sec, 1);
974 HA_ATOMIC_UPDATE_MAX(&global.cps_max, count);
Willy Tarreau3f0d02b2019-02-25 19:23:37 +0100975 }
976
Willy Tarreau4781b152021-04-06 13:53:36 +0200977 _HA_ATOMIC_INC(&activity[tid].accepted);
Willy Tarreau64a9c052019-04-12 15:27:17 +0200978
Willy Tarreau3f0d02b2019-02-25 19:23:37 +0100979 /* past this point, l->accept() will automatically decrement
Willy Tarreau82c97892019-02-27 19:32:32 +0100980 * l->nbconn, feconn and actconn once done. Setting next_*conn=0
981 * allows the error path not to rollback on nbconn. It's more
982 * convenient than duplicating all exit labels.
Willy Tarreau3f0d02b2019-02-25 19:23:37 +0100983 */
984 next_conn = 0;
Willy Tarreau82c97892019-02-27 19:32:32 +0100985 next_feconn = 0;
986 next_actconn = 0;
Willy Tarreaubbebbbf2012-05-07 21:22:09 +0200987
Willy Tarreau83efc322020-10-14 17:37:17 +0200988
Willy Tarreaue0e9c482019-01-27 15:37:19 +0100989#if defined(USE_THREAD)
Amaury Denoyelle7f7713d2022-01-19 11:37:50 +0100990 if (l->rx.flags & RX_F_LOCAL_ACCEPT)
991 goto local_accept;
992
Willy Tarreau01cac3f2021-10-12 08:47:54 +0200993 mask = thread_mask(l->rx.bind_thread) & all_threads_mask;
Willy Tarreaua7da5e82020-03-12 17:33:29 +0100994 if (atleast2(mask) && (global.tune.options & GTUNE_LISTENER_MQ) && !stopping) {
Willy Tarreaue0e9c482019-01-27 15:37:19 +0100995 struct accept_queue_ring *ring;
Willy Tarreau0fe703b2019-03-05 08:46:28 +0100996 unsigned int t, t0, t1, t2;
Willy Tarreaufc630bd2019-03-04 19:57:34 +0100997
Willy Tarreau0fe703b2019-03-05 08:46:28 +0100998 /* The principle is that we have two running indexes,
999 * each visiting in turn all threads bound to this
1000 * listener. The connection will be assigned to the one
1001 * with the least connections, and the other one will
1002 * be updated. This provides a good fairness on short
Willy Tarreaufc630bd2019-03-04 19:57:34 +01001003 * connections (round robin) and on long ones (conn
Willy Tarreau0fe703b2019-03-05 08:46:28 +01001004 * count), without ever missing any idle thread.
Willy Tarreaufc630bd2019-03-04 19:57:34 +01001005 */
Willy Tarreau0fe703b2019-03-05 08:46:28 +01001006
1007 /* keep a copy for the final update. thr_idx is composite
1008 * and made of (t2<<16) + t1.
1009 */
Willy Tarreau0cf33172019-03-06 15:26:33 +01001010 t0 = l->thr_idx;
Willy Tarreaufc630bd2019-03-04 19:57:34 +01001011 do {
Willy Tarreau0fe703b2019-03-05 08:46:28 +01001012 unsigned long m1, m2;
1013 int q1, q2;
1014
1015 t2 = t1 = t0;
1016 t2 >>= 16;
1017 t1 &= 0xFFFF;
1018
1019 /* t1 walks low to high bits ;
1020 * t2 walks high to low.
1021 */
1022 m1 = mask >> t1;
1023 m2 = mask & (t2 ? nbits(t2 + 1) : ~0UL);
1024
Willy Tarreau85d04242019-04-16 18:09:13 +02001025 if (unlikely(!(m1 & 1))) {
Willy Tarreau0fe703b2019-03-05 08:46:28 +01001026 m1 &= ~1UL;
1027 if (!m1) {
1028 m1 = mask;
1029 t1 = 0;
1030 }
1031 t1 += my_ffsl(m1) - 1;
1032 }
Willy Tarreaue0e9c482019-01-27 15:37:19 +01001033
Willy Tarreau85d04242019-04-16 18:09:13 +02001034 if (unlikely(!(m2 & (1UL << t2)) || t1 == t2)) {
1035 /* highest bit not set */
1036 if (!m2)
1037 m2 = mask;
1038
1039 t2 = my_flsl(m2) - 1;
1040 }
1041
Willy Tarreau0fe703b2019-03-05 08:46:28 +01001042 /* now we have two distinct thread IDs belonging to the mask */
1043 q1 = accept_queue_rings[t1].tail - accept_queue_rings[t1].head + ACCEPT_QUEUE_SIZE;
1044 if (q1 >= ACCEPT_QUEUE_SIZE)
1045 q1 -= ACCEPT_QUEUE_SIZE;
Willy Tarreaue0e9c482019-01-27 15:37:19 +01001046
Willy Tarreau0fe703b2019-03-05 08:46:28 +01001047 q2 = accept_queue_rings[t2].tail - accept_queue_rings[t2].head + ACCEPT_QUEUE_SIZE;
1048 if (q2 >= ACCEPT_QUEUE_SIZE)
1049 q2 -= ACCEPT_QUEUE_SIZE;
Willy Tarreaue0e9c482019-01-27 15:37:19 +01001050
Willy Tarreau0fe703b2019-03-05 08:46:28 +01001051 /* we have 3 possibilities now :
1052 * q1 < q2 : t1 is less loaded than t2, so we pick it
1053 * and update t2 (since t1 might still be
1054 * lower than another thread)
1055 * q1 > q2 : t2 is less loaded than t1, so we pick it
1056 * and update t1 (since t2 might still be
1057 * lower than another thread)
1058 * q1 = q2 : both are equally loaded, thus we pick t1
1059 * and update t1 as it will become more loaded
1060 * than t2.
1061 */
Willy Tarreaue0e9c482019-01-27 15:37:19 +01001062
Willy Tarreau0fe703b2019-03-05 08:46:28 +01001063 q1 += l->thr_conn[t1];
1064 q2 += l->thr_conn[t2];
Willy Tarreaue0e9c482019-01-27 15:37:19 +01001065
Willy Tarreau0fe703b2019-03-05 08:46:28 +01001066 if (q1 - q2 < 0) {
1067 t = t1;
1068 t2 = t2 ? t2 - 1 : LONGBITS - 1;
1069 }
1070 else if (q1 - q2 > 0) {
1071 t = t2;
1072 t1++;
1073 if (t1 >= LONGBITS)
1074 t1 = 0;
1075 }
1076 else {
1077 t = t1;
1078 t1++;
1079 if (t1 >= LONGBITS)
1080 t1 = 0;
1081 }
Willy Tarreaue0e9c482019-01-27 15:37:19 +01001082
Willy Tarreau0fe703b2019-03-05 08:46:28 +01001083 /* new value for thr_idx */
1084 t1 += (t2 << 16);
Olivier Houchard64213e92019-03-08 18:52:57 +01001085 } while (unlikely(!_HA_ATOMIC_CAS(&l->thr_idx, &t0, t1)));
Willy Tarreaue0e9c482019-01-27 15:37:19 +01001086
Willy Tarreau0fe703b2019-03-05 08:46:28 +01001087 /* We successfully selected the best thread "t" for this
1088 * connection. We use deferred accepts even if it's the
1089 * local thread because tests show that it's the best
1090 * performing model, likely due to better cache locality
1091 * when processing this loop.
Willy Tarreaue0e9c482019-01-27 15:37:19 +01001092 */
Willy Tarreau0fe703b2019-03-05 08:46:28 +01001093 ring = &accept_queue_rings[t];
Willy Tarreau83efc322020-10-14 17:37:17 +02001094 if (accept_queue_push_mp(ring, cli_conn)) {
Willy Tarreau4781b152021-04-06 13:53:36 +02001095 _HA_ATOMIC_INC(&activity[t].accq_pushed);
Willy Tarreau2bd65a72019-09-24 06:55:18 +02001096 tasklet_wakeup(ring->tasklet);
Willy Tarreaue0e9c482019-01-27 15:37:19 +01001097 continue;
1098 }
1099 /* If the ring is full we do a synchronous accept on
1100 * the local thread here.
Willy Tarreaue0e9c482019-01-27 15:37:19 +01001101 */
Willy Tarreau4781b152021-04-06 13:53:36 +02001102 _HA_ATOMIC_INC(&activity[t].accq_full);
Willy Tarreaue0e9c482019-01-27 15:37:19 +01001103 }
1104#endif // USE_THREAD
1105
Amaury Denoyelle7f7713d2022-01-19 11:37:50 +01001106 local_accept:
Willy Tarreau4781b152021-04-06 13:53:36 +02001107 _HA_ATOMIC_INC(&l->thr_conn[tid]);
Willy Tarreau83efc322020-10-14 17:37:17 +02001108 ret = l->accept(cli_conn);
Willy Tarreaubbebbbf2012-05-07 21:22:09 +02001109 if (unlikely(ret <= 0)) {
Willy Tarreau87b09662015-04-03 00:22:06 +02001110 /* The connection was closed by stream_accept(). Either
Willy Tarreaubbebbbf2012-05-07 21:22:09 +02001111 * we just have to ignore it (ret == 0) or it's a critical
1112 * error due to a resource shortage, and we must stop the
1113 * listener (ret < 0).
1114 */
Willy Tarreaubbebbbf2012-05-07 21:22:09 +02001115 if (ret == 0) /* successful termination */
1116 continue;
1117
Willy Tarreaubb660302014-05-07 19:47:02 +02001118 goto transient_error;
Willy Tarreaubbebbbf2012-05-07 21:22:09 +02001119 }
1120
Willy Tarreau3f0d02b2019-02-25 19:23:37 +01001121 /* increase the per-process number of cumulated sessions, this
1122 * may only be done once l->accept() has accepted the connection.
1123 */
Willy Tarreau93e7c002013-10-07 18:51:07 +02001124 if (!(l->options & LI_O_UNLIMITED)) {
Christopher Faulet8d8aa0d2017-05-30 15:36:50 +02001125 count = update_freq_ctr(&global.sess_per_sec, 1);
1126 HA_ATOMIC_UPDATE_MAX(&global.sps_max, count);
Willy Tarreau93e7c002013-10-07 18:51:07 +02001127 }
Willy Tarreaue43d5322013-10-07 20:01:52 +02001128#ifdef USE_OPENSSL
1129 if (!(l->options & LI_O_UNLIMITED) && l->bind_conf && l->bind_conf->is_ssl) {
Christopher Faulet8d8aa0d2017-05-30 15:36:50 +02001130 count = update_freq_ctr(&global.ssl_per_sec, 1);
1131 HA_ATOMIC_UPDATE_MAX(&global.ssl_max, count);
Willy Tarreaue43d5322013-10-07 20:01:52 +02001132 }
1133#endif
Willy Tarreau93e7c002013-10-07 18:51:07 +02001134
Willy Tarreaua0b99532021-09-30 18:48:37 +02001135 th_ctx->flags &= ~TH_FL_STUCK; // this thread is still running
Willy Tarreau3f0d02b2019-02-25 19:23:37 +01001136 } /* end of for (max_accept--) */
Willy Tarreaubbebbbf2012-05-07 21:22:09 +02001137
Christopher Faulet8d8aa0d2017-05-30 15:36:50 +02001138 end:
Willy Tarreau3f0d02b2019-02-25 19:23:37 +01001139 if (next_conn)
Willy Tarreau4781b152021-04-06 13:53:36 +02001140 _HA_ATOMIC_DEC(&l->nbconn);
Willy Tarreau741b4d62019-02-25 15:02:04 +01001141
Willy Tarreau82c97892019-02-27 19:32:32 +01001142 if (p && next_feconn)
Willy Tarreau4781b152021-04-06 13:53:36 +02001143 _HA_ATOMIC_DEC(&p->feconn);
Willy Tarreau82c97892019-02-27 19:32:32 +01001144
1145 if (next_actconn)
Willy Tarreau4781b152021-04-06 13:53:36 +02001146 _HA_ATOMIC_DEC(&actconn);
Willy Tarreau82c97892019-02-27 19:32:32 +01001147
Willy Tarreaua8cf66b2019-02-27 16:49:00 +01001148 if ((l->state == LI_FULL && (!l->maxconn || l->nbconn < l->maxconn)) ||
Willy Tarreau02757d02021-01-28 18:07:24 +01001149 (l->state == LI_LIMITED &&
Willy Tarreaucdcba112019-12-11 15:06:30 +01001150 ((!p || p->feconn < p->maxconn) && (actconn < global.maxconn) &&
1151 (!tick_isset(global_listener_queue_task->expire) ||
1152 tick_is_expired(global_listener_queue_task->expire, now_ms))))) {
Willy Tarreau3f0d02b2019-02-25 19:23:37 +01001153 /* at least one thread has to this when quitting */
1154 resume_listener(l);
1155
Willy Tarreau02757d02021-01-28 18:07:24 +01001156 /* Dequeues all of the listeners waiting for a resource */
Willy Tarreau241797a2019-12-10 14:10:52 +01001157 dequeue_all_listeners();
Willy Tarreau3f0d02b2019-02-25 19:23:37 +01001158
Olivier Houchard859dc802019-08-08 15:47:21 +02001159 if (p && !MT_LIST_ISEMPTY(&p->listener_queue) &&
Willy Tarreau3f0d02b2019-02-25 19:23:37 +01001160 (!p->fe_sps_lim || freq_ctr_remain(&p->fe_sess_per_sec, p->fe_sps_lim, 0) > 0))
Willy Tarreau241797a2019-12-10 14:10:52 +01001161 dequeue_proxy_listeners(p);
Willy Tarreau3f0d02b2019-02-25 19:23:37 +01001162 }
Willy Tarreau0591bf72019-12-10 12:01:21 +01001163 return;
1164
1165 transient_error:
1166 /* pause the listener for up to 100 ms */
1167 expire = tick_add(now_ms, 100);
1168
Willy Tarreau258b3512020-10-13 17:46:05 +02001169 /* This may be a shared socket that was paused by another process.
1170 * Let's put it to pause in this case.
1171 */
1172 if (l->rx.proto && l->rx.proto->rx_listening(&l->rx) == 0) {
1173 pause_listener(l);
1174 goto end;
1175 }
1176
Willy Tarreau0591bf72019-12-10 12:01:21 +01001177 limit_global:
1178 /* (re-)queue the listener to the global queue and set it to expire no
1179 * later than <expire> ahead. The listener turns to LI_LIMITED.
1180 */
1181 limit_listener(l, &global_listener_queue);
1182 task_schedule(global_listener_queue_task, expire);
1183 goto end;
1184
1185 limit_proxy:
1186 /* (re-)queue the listener to the proxy's queue and set it to expire no
1187 * later than <expire> ahead. The listener turns to LI_LIMITED.
1188 */
1189 limit_listener(l, &p->listener_queue);
Willy Tarreaueeea8082020-01-08 19:15:07 +01001190 if (p->task && tick_isset(expire))
1191 task_schedule(p->task, expire);
Willy Tarreau0591bf72019-12-10 12:01:21 +01001192 goto end;
Willy Tarreaubbebbbf2012-05-07 21:22:09 +02001193}
1194
Willy Tarreau05f50472017-09-15 09:19:58 +02001195/* Notify the listener that a connection initiated from it was released. This
1196 * is used to keep the connection count consistent and to possibly re-open
1197 * listening when it was limited.
1198 */
1199void listener_release(struct listener *l)
1200{
1201 struct proxy *fe = l->bind_conf->frontend;
1202
1203 if (!(l->options & LI_O_UNLIMITED))
Willy Tarreau4781b152021-04-06 13:53:36 +02001204 _HA_ATOMIC_DEC(&actconn);
Willy Tarreau82c97892019-02-27 19:32:32 +01001205 if (fe)
Willy Tarreau4781b152021-04-06 13:53:36 +02001206 _HA_ATOMIC_DEC(&fe->feconn);
1207 _HA_ATOMIC_DEC(&l->nbconn);
1208 _HA_ATOMIC_DEC(&l->thr_conn[tid]);
Willy Tarreau82c97892019-02-27 19:32:32 +01001209
1210 if (l->state == LI_FULL || l->state == LI_LIMITED)
Willy Tarreau05f50472017-09-15 09:19:58 +02001211 resume_listener(l);
1212
Willy Tarreau02757d02021-01-28 18:07:24 +01001213 /* Dequeues all of the listeners waiting for a resource */
1214 dequeue_all_listeners();
1215
Olivier Houchard859dc802019-08-08 15:47:21 +02001216 if (!MT_LIST_ISEMPTY(&fe->listener_queue) &&
Willy Tarreau05f50472017-09-15 09:19:58 +02001217 (!fe->fe_sps_lim || freq_ctr_remain(&fe->fe_sess_per_sec, fe->fe_sps_lim, 0) > 0))
Willy Tarreau241797a2019-12-10 14:10:52 +01001218 dequeue_proxy_listeners(fe);
Willy Tarreau05f50472017-09-15 09:19:58 +02001219}
1220
Willy Tarreauf2cb1692019-07-11 10:08:31 +02001221/* Initializes the listener queues. Returns 0 on success, otherwise ERR_* flags */
1222static int listener_queue_init()
1223{
Willy Tarreaubeeabf52021-10-01 18:23:30 +02001224 global_listener_queue_task = task_new_anywhere();
Willy Tarreaua1d97f82019-12-10 11:18:41 +01001225 if (!global_listener_queue_task) {
1226 ha_alert("Out of memory when initializing global listener queue\n");
1227 return ERR_FATAL|ERR_ABORT;
1228 }
1229 /* very simple initialization, users will queue the task if needed */
1230 global_listener_queue_task->context = NULL; /* not even a context! */
1231 global_listener_queue_task->process = manage_global_listener_queue;
1232
Willy Tarreauf2cb1692019-07-11 10:08:31 +02001233 return 0;
1234}
1235
1236static void listener_queue_deinit()
1237{
Willy Tarreaua1d97f82019-12-10 11:18:41 +01001238 task_destroy(global_listener_queue_task);
1239 global_listener_queue_task = NULL;
Willy Tarreauf2cb1692019-07-11 10:08:31 +02001240}
1241
1242REGISTER_CONFIG_POSTPARSER("multi-threaded listener queue", listener_queue_init);
1243REGISTER_POST_DEINIT(listener_queue_deinit);
1244
Willy Tarreaua1d97f82019-12-10 11:18:41 +01001245
1246/* This is the global management task for listeners. It enables listeners waiting
1247 * for global resources when there are enough free resource, or at least once in
Willy Tarreaud597ec22021-01-29 14:29:06 +01001248 * a while. It is designed to be called as a task. It's exported so that it's easy
1249 * to spot in "show tasks" or "show profiling".
Willy Tarreaua1d97f82019-12-10 11:18:41 +01001250 */
Willy Tarreau144f84a2021-03-02 16:09:26 +01001251struct task *manage_global_listener_queue(struct task *t, void *context, unsigned int state)
Willy Tarreaua1d97f82019-12-10 11:18:41 +01001252{
1253 /* If there are still too many concurrent connections, let's wait for
1254 * some of them to go away. We don't need to re-arm the timer because
1255 * each of them will scan the queue anyway.
1256 */
1257 if (unlikely(actconn >= global.maxconn))
1258 goto out;
1259
1260 /* We should periodically try to enable listeners waiting for a global
1261 * resource here, because it is possible, though very unlikely, that
1262 * they have been blocked by a temporary lack of global resource such
1263 * as a file descriptor or memory and that the temporary condition has
1264 * disappeared.
1265 */
1266 dequeue_all_listeners();
1267
1268 out:
1269 t->expire = TICK_ETERNITY;
1270 task_queue(t);
1271 return t;
1272}
1273
Willy Tarreau26982662012-09-12 23:17:10 +02001274/*
1275 * Registers the bind keyword list <kwl> as a list of valid keywords for next
1276 * parsing sessions.
1277 */
1278void bind_register_keywords(struct bind_kw_list *kwl)
1279{
Willy Tarreau2b718102021-04-21 07:32:39 +02001280 LIST_APPEND(&bind_keywords.list, &kwl->list);
Willy Tarreau26982662012-09-12 23:17:10 +02001281}
1282
1283/* Return a pointer to the bind keyword <kw>, or NULL if not found. If the
1284 * keyword is found with a NULL ->parse() function, then an attempt is made to
1285 * find one with a valid ->parse() function. This way it is possible to declare
1286 * platform-dependant, known keywords as NULL, then only declare them as valid
1287 * if some options are met. Note that if the requested keyword contains an
1288 * opening parenthesis, everything from this point is ignored.
1289 */
1290struct bind_kw *bind_find_kw(const char *kw)
1291{
1292 int index;
1293 const char *kwend;
1294 struct bind_kw_list *kwl;
1295 struct bind_kw *ret = NULL;
1296
1297 kwend = strchr(kw, '(');
1298 if (!kwend)
1299 kwend = kw + strlen(kw);
1300
1301 list_for_each_entry(kwl, &bind_keywords.list, list) {
1302 for (index = 0; kwl->kw[index].kw != NULL; index++) {
1303 if ((strncmp(kwl->kw[index].kw, kw, kwend - kw) == 0) &&
1304 kwl->kw[index].kw[kwend-kw] == 0) {
1305 if (kwl->kw[index].parse)
1306 return &kwl->kw[index]; /* found it !*/
1307 else
1308 ret = &kwl->kw[index]; /* may be OK */
1309 }
1310 }
1311 }
1312 return ret;
1313}
1314
Willy Tarreau8638f482012-09-18 18:01:17 +02001315/* Dumps all registered "bind" keywords to the <out> string pointer. The
1316 * unsupported keywords are only dumped if their supported form was not
1317 * found.
1318 */
1319void bind_dump_kws(char **out)
1320{
1321 struct bind_kw_list *kwl;
1322 int index;
1323
Christopher Faulet784063e2020-05-18 12:14:18 +02001324 if (!out)
1325 return;
1326
Willy Tarreau8638f482012-09-18 18:01:17 +02001327 *out = NULL;
1328 list_for_each_entry(kwl, &bind_keywords.list, list) {
1329 for (index = 0; kwl->kw[index].kw != NULL; index++) {
1330 if (kwl->kw[index].parse ||
1331 bind_find_kw(kwl->kw[index].kw) == &kwl->kw[index]) {
Willy Tarreau51fb7652012-09-18 18:24:39 +02001332 memprintf(out, "%s[%4s] %s%s%s\n", *out ? *out : "",
1333 kwl->scope,
Willy Tarreau8638f482012-09-18 18:01:17 +02001334 kwl->kw[index].kw,
Willy Tarreau51fb7652012-09-18 18:24:39 +02001335 kwl->kw[index].skip ? " <arg>" : "",
1336 kwl->kw[index].parse ? "" : " (not supported)");
Willy Tarreau8638f482012-09-18 18:01:17 +02001337 }
1338 }
1339 }
1340}
1341
Willy Tarreau433b05f2021-03-12 10:14:07 +01001342/* Try to find in srv_keyword the word that looks closest to <word> by counting
1343 * transitions between letters, digits and other characters. Will return the
1344 * best matching word if found, otherwise NULL.
1345 */
1346const char *bind_find_best_kw(const char *word)
1347{
1348 uint8_t word_sig[1024];
1349 uint8_t list_sig[1024];
1350 const struct bind_kw_list *kwl;
1351 const char *best_ptr = NULL;
1352 int dist, best_dist = INT_MAX;
1353 int index;
1354
1355 make_word_fingerprint(word_sig, word);
1356 list_for_each_entry(kwl, &bind_keywords.list, list) {
1357 for (index = 0; kwl->kw[index].kw != NULL; index++) {
1358 make_word_fingerprint(list_sig, kwl->kw[index].kw);
1359 dist = word_fingerprint_distance(word_sig, list_sig);
1360 if (dist < best_dist) {
1361 best_dist = dist;
1362 best_ptr = kwl->kw[index].kw;
1363 }
1364 }
1365 }
1366
1367 if (best_dist > 2 * strlen(word) || (best_ptr && best_dist > 2 * strlen(best_ptr)))
1368 best_ptr = NULL;
1369
1370 return best_ptr;
1371}
1372
Willy Tarreaudbf78022021-10-06 09:05:08 +02001373/* allocate an bind_conf struct for a bind line, and chain it to the frontend <fe>.
1374 * If <arg> is not NULL, it is duplicated into ->arg to store useful config
1375 * information for error reporting. NULL is returned on error.
1376 */
1377struct bind_conf *bind_conf_alloc(struct proxy *fe, const char *file,
1378 int line, const char *arg, struct xprt_ops *xprt)
1379{
1380 struct bind_conf *bind_conf = calloc(1, sizeof(*bind_conf));
1381
1382 if (!bind_conf)
1383 goto err;
1384
1385 bind_conf->file = strdup(file);
1386 if (!bind_conf->file)
1387 goto err;
1388 bind_conf->line = line;
1389 if (arg) {
1390 bind_conf->arg = strdup(arg);
1391 if (!bind_conf->arg)
1392 goto err;
1393 }
1394
1395 LIST_APPEND(&fe->conf.bind, &bind_conf->by_fe);
1396 bind_conf->settings.ux.uid = -1;
1397 bind_conf->settings.ux.gid = -1;
1398 bind_conf->settings.ux.mode = 0;
Willy Tarreau6dfbef42021-10-12 15:23:03 +02001399 bind_conf->settings.shards = 1;
Willy Tarreaudbf78022021-10-06 09:05:08 +02001400 bind_conf->xprt = xprt;
1401 bind_conf->frontend = fe;
1402 bind_conf->severity_output = CLI_SEVERITY_NONE;
1403#ifdef USE_OPENSSL
1404 HA_RWLOCK_INIT(&bind_conf->sni_lock);
1405 bind_conf->sni_ctx = EB_ROOT;
1406 bind_conf->sni_w_ctx = EB_ROOT;
1407#endif
1408 LIST_INIT(&bind_conf->listeners);
1409 return bind_conf;
1410
1411 err:
1412 if (bind_conf) {
1413 ha_free(&bind_conf->file);
1414 ha_free(&bind_conf->arg);
1415 }
1416 ha_free(&bind_conf);
1417 return NULL;
1418}
1419
1420const char *listener_state_str(const struct listener *l)
1421{
1422 static const char *states[8] = {
1423 "NEW", "INI", "ASS", "PAU", "LIS", "RDY", "FUL", "LIM",
1424 };
1425 unsigned int st = l->state;
1426
1427 if (st >= sizeof(states) / sizeof(*states))
1428 return "INVALID";
1429 return states[st];
1430}
1431
Willy Tarreau645513a2010-05-24 20:55:15 +02001432/************************************************************************/
Willy Tarreau0ccb7442013-01-07 22:54:17 +01001433/* All supported sample and ACL keywords must be declared here. */
Willy Tarreau645513a2010-05-24 20:55:15 +02001434/************************************************************************/
1435
Willy Tarreaua5e37562011-12-16 17:06:15 +01001436/* set temp integer to the number of connexions to the same listening socket */
Willy Tarreau645513a2010-05-24 20:55:15 +02001437static int
Thierry FOURNIER0786d052015-05-11 15:42:45 +02001438smp_fetch_dconn(const struct arg *args, struct sample *smp, const char *kw, void *private)
Willy Tarreau645513a2010-05-24 20:55:15 +02001439{
Thierry FOURNIER8c542ca2015-08-19 09:00:18 +02001440 smp->data.type = SMP_T_SINT;
Thierry FOURNIER136f9d32015-08-19 09:07:19 +02001441 smp->data.u.sint = smp->sess->listener->nbconn;
Willy Tarreau645513a2010-05-24 20:55:15 +02001442 return 1;
1443}
1444
Willy Tarreaua5e37562011-12-16 17:06:15 +01001445/* set temp integer to the id of the socket (listener) */
Willy Tarreau645513a2010-05-24 20:55:15 +02001446static int
Thierry FOURNIER0786d052015-05-11 15:42:45 +02001447smp_fetch_so_id(const struct arg *args, struct sample *smp, const char *kw, void *private)
Willy Tarreau37406352012-04-23 16:16:37 +02001448{
Thierry FOURNIER8c542ca2015-08-19 09:00:18 +02001449 smp->data.type = SMP_T_SINT;
Thierry FOURNIER136f9d32015-08-19 09:07:19 +02001450 smp->data.u.sint = smp->sess->listener->luid;
Willy Tarreau645513a2010-05-24 20:55:15 +02001451 return 1;
1452}
Jerome Magnineb421b22020-03-27 22:08:40 +01001453static int
1454smp_fetch_so_name(const struct arg *args, struct sample *smp, const char *kw, void *private)
1455{
1456 smp->data.u.str.area = smp->sess->listener->name;
1457 if (!smp->data.u.str.area)
1458 return 0;
1459
1460 smp->data.type = SMP_T_STR;
1461 smp->flags = SMP_F_CONST;
1462 smp->data.u.str.data = strlen(smp->data.u.str.area);
1463 return 1;
1464}
Willy Tarreau645513a2010-05-24 20:55:15 +02001465
Willy Tarreau3dcc3412012-09-18 17:17:28 +02001466/* parse the "accept-proxy" bind keyword */
Willy Tarreau4348fad2012-09-20 16:48:07 +02001467static int bind_parse_accept_proxy(char **args, int cur_arg, struct proxy *px, struct bind_conf *conf, char **err)
Willy Tarreau3dcc3412012-09-18 17:17:28 +02001468{
1469 struct listener *l;
1470
Willy Tarreau4348fad2012-09-20 16:48:07 +02001471 list_for_each_entry(l, &conf->listeners, by_bind)
Willy Tarreau3dcc3412012-09-18 17:17:28 +02001472 l->options |= LI_O_ACC_PROXY;
1473
1474 return 0;
1475}
1476
Bertrand Jacquin93b227d2016-06-04 15:11:10 +01001477/* parse the "accept-netscaler-cip" bind keyword */
1478static int bind_parse_accept_netscaler_cip(char **args, int cur_arg, struct proxy *px, struct bind_conf *conf, char **err)
1479{
1480 struct listener *l;
1481 uint32_t val;
1482
1483 if (!*args[cur_arg + 1]) {
1484 memprintf(err, "'%s' : missing value", args[cur_arg]);
1485 return ERR_ALERT | ERR_FATAL;
1486 }
1487
1488 val = atol(args[cur_arg + 1]);
1489 if (val <= 0) {
Willy Tarreaue2711c72019-02-27 15:39:41 +01001490 memprintf(err, "'%s' : invalid value %d, must be >= 0", args[cur_arg], val);
Bertrand Jacquin93b227d2016-06-04 15:11:10 +01001491 return ERR_ALERT | ERR_FATAL;
1492 }
1493
1494 list_for_each_entry(l, &conf->listeners, by_bind) {
1495 l->options |= LI_O_ACC_CIP;
1496 conf->ns_cip_magic = val;
1497 }
1498
1499 return 0;
1500}
1501
Willy Tarreau3dcc3412012-09-18 17:17:28 +02001502/* parse the "backlog" bind keyword */
Willy Tarreau4348fad2012-09-20 16:48:07 +02001503static int bind_parse_backlog(char **args, int cur_arg, struct proxy *px, struct bind_conf *conf, char **err)
Willy Tarreau3dcc3412012-09-18 17:17:28 +02001504{
1505 struct listener *l;
1506 int val;
1507
1508 if (!*args[cur_arg + 1]) {
Willy Tarreaueb6cead2012-09-20 19:43:14 +02001509 memprintf(err, "'%s' : missing value", args[cur_arg]);
Willy Tarreau3dcc3412012-09-18 17:17:28 +02001510 return ERR_ALERT | ERR_FATAL;
1511 }
1512
1513 val = atol(args[cur_arg + 1]);
Willy Tarreaue2711c72019-02-27 15:39:41 +01001514 if (val < 0) {
Willy Tarreaueb6cead2012-09-20 19:43:14 +02001515 memprintf(err, "'%s' : invalid value %d, must be > 0", args[cur_arg], val);
Willy Tarreau3dcc3412012-09-18 17:17:28 +02001516 return ERR_ALERT | ERR_FATAL;
1517 }
1518
Willy Tarreau4348fad2012-09-20 16:48:07 +02001519 list_for_each_entry(l, &conf->listeners, by_bind)
Willy Tarreau3dcc3412012-09-18 17:17:28 +02001520 l->backlog = val;
1521
1522 return 0;
1523}
1524
1525/* parse the "id" bind keyword */
Willy Tarreau4348fad2012-09-20 16:48:07 +02001526static int bind_parse_id(char **args, int cur_arg, struct proxy *px, struct bind_conf *conf, char **err)
Willy Tarreau3dcc3412012-09-18 17:17:28 +02001527{
1528 struct eb32_node *node;
Willy Tarreau4348fad2012-09-20 16:48:07 +02001529 struct listener *l, *new;
Thierry Fourniere7fe8eb2016-02-26 08:45:58 +01001530 char *error;
Willy Tarreau3dcc3412012-09-18 17:17:28 +02001531
Willy Tarreau4348fad2012-09-20 16:48:07 +02001532 if (conf->listeners.n != conf->listeners.p) {
Willy Tarreaueb6cead2012-09-20 19:43:14 +02001533 memprintf(err, "'%s' can only be used with a single socket", args[cur_arg]);
Willy Tarreau3dcc3412012-09-18 17:17:28 +02001534 return ERR_ALERT | ERR_FATAL;
1535 }
1536
1537 if (!*args[cur_arg + 1]) {
Willy Tarreaueb6cead2012-09-20 19:43:14 +02001538 memprintf(err, "'%s' : expects an integer argument", args[cur_arg]);
Willy Tarreau3dcc3412012-09-18 17:17:28 +02001539 return ERR_ALERT | ERR_FATAL;
1540 }
1541
Willy Tarreau4348fad2012-09-20 16:48:07 +02001542 new = LIST_NEXT(&conf->listeners, struct listener *, by_bind);
Thierry Fourniere7fe8eb2016-02-26 08:45:58 +01001543 new->luid = strtol(args[cur_arg + 1], &error, 10);
1544 if (*error != '\0') {
1545 memprintf(err, "'%s' : expects an integer argument, found '%s'", args[cur_arg], args[cur_arg + 1]);
1546 return ERR_ALERT | ERR_FATAL;
1547 }
Willy Tarreau4348fad2012-09-20 16:48:07 +02001548 new->conf.id.key = new->luid;
Willy Tarreau3dcc3412012-09-18 17:17:28 +02001549
Willy Tarreau4348fad2012-09-20 16:48:07 +02001550 if (new->luid <= 0) {
Willy Tarreaueb6cead2012-09-20 19:43:14 +02001551 memprintf(err, "'%s' : custom id has to be > 0", args[cur_arg]);
Willy Tarreau3dcc3412012-09-18 17:17:28 +02001552 return ERR_ALERT | ERR_FATAL;
1553 }
1554
Willy Tarreau4348fad2012-09-20 16:48:07 +02001555 node = eb32_lookup(&px->conf.used_listener_id, new->luid);
Willy Tarreau3dcc3412012-09-18 17:17:28 +02001556 if (node) {
1557 l = container_of(node, struct listener, conf.id);
Willy Tarreaueb6cead2012-09-20 19:43:14 +02001558 memprintf(err, "'%s' : custom id %d already used at %s:%d ('bind %s')",
1559 args[cur_arg], l->luid, l->bind_conf->file, l->bind_conf->line,
1560 l->bind_conf->arg);
Willy Tarreau3dcc3412012-09-18 17:17:28 +02001561 return ERR_ALERT | ERR_FATAL;
1562 }
1563
Willy Tarreau4348fad2012-09-20 16:48:07 +02001564 eb32_insert(&px->conf.used_listener_id, &new->conf.id);
Willy Tarreau3dcc3412012-09-18 17:17:28 +02001565 return 0;
1566}
1567
1568/* parse the "maxconn" bind keyword */
Willy Tarreau4348fad2012-09-20 16:48:07 +02001569static int bind_parse_maxconn(char **args, int cur_arg, struct proxy *px, struct bind_conf *conf, char **err)
Willy Tarreau3dcc3412012-09-18 17:17:28 +02001570{
1571 struct listener *l;
1572 int val;
1573
1574 if (!*args[cur_arg + 1]) {
Willy Tarreaueb6cead2012-09-20 19:43:14 +02001575 memprintf(err, "'%s' : missing value", args[cur_arg]);
Willy Tarreau3dcc3412012-09-18 17:17:28 +02001576 return ERR_ALERT | ERR_FATAL;
1577 }
1578
1579 val = atol(args[cur_arg + 1]);
Willy Tarreaua8cf66b2019-02-27 16:49:00 +01001580 if (val < 0) {
1581 memprintf(err, "'%s' : invalid value %d, must be >= 0", args[cur_arg], val);
Willy Tarreau3dcc3412012-09-18 17:17:28 +02001582 return ERR_ALERT | ERR_FATAL;
1583 }
1584
Willy Tarreau4348fad2012-09-20 16:48:07 +02001585 list_for_each_entry(l, &conf->listeners, by_bind)
Willy Tarreau3dcc3412012-09-18 17:17:28 +02001586 l->maxconn = val;
1587
1588 return 0;
1589}
1590
1591/* parse the "name" bind keyword */
Willy Tarreau4348fad2012-09-20 16:48:07 +02001592static int bind_parse_name(char **args, int cur_arg, struct proxy *px, struct bind_conf *conf, char **err)
Willy Tarreau3dcc3412012-09-18 17:17:28 +02001593{
1594 struct listener *l;
1595
1596 if (!*args[cur_arg + 1]) {
Willy Tarreaueb6cead2012-09-20 19:43:14 +02001597 memprintf(err, "'%s' : missing name", args[cur_arg]);
Willy Tarreau3dcc3412012-09-18 17:17:28 +02001598 return ERR_ALERT | ERR_FATAL;
1599 }
1600
Willy Tarreau4348fad2012-09-20 16:48:07 +02001601 list_for_each_entry(l, &conf->listeners, by_bind)
Willy Tarreau3dcc3412012-09-18 17:17:28 +02001602 l->name = strdup(args[cur_arg + 1]);
1603
1604 return 0;
1605}
1606
1607/* parse the "nice" bind keyword */
Willy Tarreau4348fad2012-09-20 16:48:07 +02001608static int bind_parse_nice(char **args, int cur_arg, struct proxy *px, struct bind_conf *conf, char **err)
Willy Tarreau3dcc3412012-09-18 17:17:28 +02001609{
1610 struct listener *l;
1611 int val;
1612
1613 if (!*args[cur_arg + 1]) {
Willy Tarreaueb6cead2012-09-20 19:43:14 +02001614 memprintf(err, "'%s' : missing value", args[cur_arg]);
Willy Tarreau3dcc3412012-09-18 17:17:28 +02001615 return ERR_ALERT | ERR_FATAL;
1616 }
1617
1618 val = atol(args[cur_arg + 1]);
1619 if (val < -1024 || val > 1024) {
Willy Tarreaueb6cead2012-09-20 19:43:14 +02001620 memprintf(err, "'%s' : invalid value %d, allowed range is -1024..1024", args[cur_arg], val);
Willy Tarreau3dcc3412012-09-18 17:17:28 +02001621 return ERR_ALERT | ERR_FATAL;
1622 }
1623
Willy Tarreau4348fad2012-09-20 16:48:07 +02001624 list_for_each_entry(l, &conf->listeners, by_bind)
Willy Tarreau3dcc3412012-09-18 17:17:28 +02001625 l->nice = val;
1626
1627 return 0;
1628}
1629
Willy Tarreau6ae1ba62014-05-07 19:01:58 +02001630/* parse the "process" bind keyword */
1631static int bind_parse_process(char **args, int cur_arg, struct proxy *px, struct bind_conf *conf, char **err)
1632{
Christopher Fauletc644fa92017-11-23 22:44:11 +01001633 char *slash;
1634 unsigned long proc = 0, thread = 0;
Willy Tarreau6ae1ba62014-05-07 19:01:58 +02001635
Christopher Fauletc644fa92017-11-23 22:44:11 +01001636 if ((slash = strchr(args[cur_arg + 1], '/')) != NULL)
1637 *slash = 0;
1638
Willy Tarreau72faef32021-06-15 08:36:30 +02001639 if (parse_process_number(args[cur_arg + 1], &proc, 1, NULL, err)) {
Christopher Fauletf1f0c5f2017-11-22 12:06:43 +01001640 memprintf(err, "'%s' : %s", args[cur_arg], *err);
Willy Tarreau6ae1ba62014-05-07 19:01:58 +02001641 return ERR_ALERT | ERR_FATAL;
1642 }
1643
Christopher Fauletc644fa92017-11-23 22:44:11 +01001644 if (slash) {
Willy Tarreauc9a82e42019-01-26 13:25:14 +01001645 if (parse_process_number(slash+1, &thread, MAX_THREADS, NULL, err)) {
Christopher Fauletc644fa92017-11-23 22:44:11 +01001646 memprintf(err, "'%s' : %s", args[cur_arg], *err);
1647 return ERR_ALERT | ERR_FATAL;
1648 }
1649 *slash = '/';
1650 }
1651
Willy Tarreau01cac3f2021-10-12 08:47:54 +02001652 conf->bind_thread |= thread;
Willy Tarreauc8cac042021-09-21 14:31:29 +02001653
1654 memprintf(err, "'process %s' on 'bind' lines is deprecated and will be removed in 2.7.", args[cur_arg+1]);
1655 if (slash)
1656 memprintf(err, "%s Please use 'thread %s' instead.", *err, slash + 1);
1657 return ERR_WARN;
Willy Tarreau6ae1ba62014-05-07 19:01:58 +02001658}
Willy Tarreau3dcc3412012-09-18 17:17:28 +02001659
Christopher Fauleta717b992018-04-10 14:43:00 +02001660/* parse the "proto" bind keyword */
1661static int bind_parse_proto(char **args, int cur_arg, struct proxy *px, struct bind_conf *conf, char **err)
1662{
1663 struct ist proto;
1664
1665 if (!*args[cur_arg + 1]) {
1666 memprintf(err, "'%s' : missing value", args[cur_arg]);
1667 return ERR_ALERT | ERR_FATAL;
1668 }
1669
Tim Duesterhusdcf753a2021-03-04 17:31:47 +01001670 proto = ist(args[cur_arg + 1]);
Christopher Fauleta717b992018-04-10 14:43:00 +02001671 conf->mux_proto = get_mux_proto(proto);
1672 if (!conf->mux_proto) {
1673 memprintf(err, "'%s' : unknown MUX protocol '%s'", args[cur_arg], args[cur_arg+1]);
1674 return ERR_ALERT | ERR_FATAL;
1675 }
Willy Tarreauc8cac042021-09-21 14:31:29 +02001676 return 0;
1677}
1678
Willy Tarreau6dfbef42021-10-12 15:23:03 +02001679/* parse the "shards" bind keyword. Takes an integer or "by-thread" */
1680static int bind_parse_shards(char **args, int cur_arg, struct proxy *px, struct bind_conf *conf, char **err)
1681{
1682 int val;
1683
1684 if (!*args[cur_arg + 1]) {
1685 memprintf(err, "'%s' : missing value", args[cur_arg]);
1686 return ERR_ALERT | ERR_FATAL;
1687 }
1688
1689 if (strcmp(args[cur_arg + 1], "by-thread") == 0) {
1690 val = MAX_THREADS; /* will be trimmed later anyway */
1691 } else {
1692 val = atol(args[cur_arg + 1]);
1693 if (val < 1 || val > MAX_THREADS) {
1694 memprintf(err, "'%s' : invalid value %d, allowed range is %d..%d or 'by-thread'", args[cur_arg], val, 1, MAX_THREADS);
1695 return ERR_ALERT | ERR_FATAL;
1696 }
1697 }
1698
1699 conf->settings.shards = val;
1700 return 0;
1701}
1702
Willy Tarreauc8cac042021-09-21 14:31:29 +02001703/* parse the "thread" bind keyword */
1704static int bind_parse_thread(char **args, int cur_arg, struct proxy *px, struct bind_conf *conf, char **err)
1705{
Willy Tarreaud57b9ff2021-09-29 18:50:31 +02001706 char *sep = NULL;
1707 ulong thread = 0;
1708 long tgroup = 0;
Willy Tarreauc8cac042021-09-21 14:31:29 +02001709
Willy Tarreaud57b9ff2021-09-29 18:50:31 +02001710 tgroup = strtol(args[cur_arg + 1], &sep, 10);
1711 if (*sep == '/') {
1712 /* a thread group was present */
1713 if (tgroup < 1 || tgroup > MAX_TGROUPS) {
1714 memprintf(err, "'%s' thread-group number must be between 1 and %d (was %ld)", args[cur_arg + 1], MAX_TGROUPS, tgroup);
1715 return ERR_ALERT | ERR_FATAL;
1716 }
1717 sep++;
1718 }
1719 else {
1720 /* no thread group */
1721 tgroup = 0;
1722 sep = args[cur_arg + 1];
1723 }
Willy Tarreauc8cac042021-09-21 14:31:29 +02001724
Willy Tarreau01cac3f2021-10-12 08:47:54 +02001725 if ((conf->bind_tgroup || conf->bind_thread) &&
1726 conf->bind_tgroup != tgroup) {
Willy Tarreaud57b9ff2021-09-29 18:50:31 +02001727 memprintf(err, "'%s' multiple thread-groups are not supported", args[cur_arg + 1]);
Willy Tarreauc8cac042021-09-21 14:31:29 +02001728 return ERR_ALERT | ERR_FATAL;
1729 }
Willy Tarreaud57b9ff2021-09-29 18:50:31 +02001730
1731 if (parse_process_number(sep, &thread, MAX_THREADS, NULL, err)) {
1732 memprintf(err, "'%s' : %s", sep, *err);
Willy Tarreauc8cac042021-09-21 14:31:29 +02001733 return ERR_ALERT | ERR_FATAL;
1734 }
1735
Willy Tarreau01cac3f2021-10-12 08:47:54 +02001736 conf->bind_thread |= thread;
1737 conf->bind_tgroup = tgroup;
Christopher Fauleta717b992018-04-10 14:43:00 +02001738 return 0;
1739}
1740
Willy Tarreau7ac908b2019-02-27 12:02:18 +01001741/* config parser for global "tune.listener.multi-queue", accepts "on" or "off" */
1742static int cfg_parse_tune_listener_mq(char **args, int section_type, struct proxy *curpx,
Willy Tarreau01825162021-03-09 09:53:46 +01001743 const struct proxy *defpx, const char *file, int line,
Willy Tarreau7ac908b2019-02-27 12:02:18 +01001744 char **err)
1745{
1746 if (too_many_args(1, args, err, NULL))
1747 return -1;
1748
1749 if (strcmp(args[1], "on") == 0)
1750 global.tune.options |= GTUNE_LISTENER_MQ;
1751 else if (strcmp(args[1], "off") == 0)
1752 global.tune.options &= ~GTUNE_LISTENER_MQ;
1753 else {
1754 memprintf(err, "'%s' expects either 'on' or 'off' but got '%s'.", args[0], args[1]);
1755 return -1;
1756 }
1757 return 0;
1758}
1759
Willy Tarreau61612d42012-04-19 18:42:05 +02001760/* Note: must not be declared <const> as its list will be overwritten.
1761 * Please take care of keeping this list alphabetically sorted.
1762 */
Willy Tarreaudc13c112013-06-21 23:16:39 +02001763static struct sample_fetch_kw_list smp_kws = {ILH, {
Thierry FOURNIER07ee64e2015-07-06 23:43:03 +02001764 { "dst_conn", smp_fetch_dconn, 0, NULL, SMP_T_SINT, SMP_USE_FTEND, },
1765 { "so_id", smp_fetch_so_id, 0, NULL, SMP_T_SINT, SMP_USE_FTEND, },
Jerome Magnineb421b22020-03-27 22:08:40 +01001766 { "so_name", smp_fetch_so_name, 0, NULL, SMP_T_STR, SMP_USE_FTEND, },
Willy Tarreau0ccb7442013-01-07 22:54:17 +01001767 { /* END */ },
1768}};
1769
Willy Tarreau0108d902018-11-25 19:14:37 +01001770INITCALL1(STG_REGISTER, sample_register_fetches, &smp_kws);
1771
Willy Tarreau0ccb7442013-01-07 22:54:17 +01001772/* Note: must not be declared <const> as its list will be overwritten.
1773 * Please take care of keeping this list alphabetically sorted.
1774 */
Willy Tarreaudc13c112013-06-21 23:16:39 +02001775static struct acl_kw_list acl_kws = {ILH, {
Willy Tarreau0ccb7442013-01-07 22:54:17 +01001776 { /* END */ },
Willy Tarreau645513a2010-05-24 20:55:15 +02001777}};
1778
Willy Tarreau0108d902018-11-25 19:14:37 +01001779INITCALL1(STG_REGISTER, acl_register_keywords, &acl_kws);
1780
Willy Tarreau3dcc3412012-09-18 17:17:28 +02001781/* Note: must not be declared <const> as its list will be overwritten.
1782 * Please take care of keeping this list alphabetically sorted, doing so helps
1783 * all code contributors.
1784 * Optional keywords are also declared with a NULL ->parse() function so that
1785 * the config parser can report an appropriate error when a known keyword was
1786 * not enabled.
1787 */
Willy Tarreau51fb7652012-09-18 18:24:39 +02001788static struct bind_kw_list bind_kws = { "ALL", { }, {
Bertrand Jacquin93b227d2016-06-04 15:11:10 +01001789 { "accept-netscaler-cip", bind_parse_accept_netscaler_cip, 1 }, /* enable NetScaler Client IP insertion protocol */
Willy Tarreau3dcc3412012-09-18 17:17:28 +02001790 { "accept-proxy", bind_parse_accept_proxy, 0 }, /* enable PROXY protocol */
1791 { "backlog", bind_parse_backlog, 1 }, /* set backlog of listening socket */
1792 { "id", bind_parse_id, 1 }, /* set id of listening socket */
1793 { "maxconn", bind_parse_maxconn, 1 }, /* set maxconn of listening socket */
1794 { "name", bind_parse_name, 1 }, /* set name of listening socket */
1795 { "nice", bind_parse_nice, 1 }, /* set nice of listening socket */
Willy Tarreau6ae1ba62014-05-07 19:01:58 +02001796 { "process", bind_parse_process, 1 }, /* set list of allowed process for this socket */
Christopher Fauleta717b992018-04-10 14:43:00 +02001797 { "proto", bind_parse_proto, 1 }, /* set the proto to use for all incoming connections */
Willy Tarreau6dfbef42021-10-12 15:23:03 +02001798 { "shards", bind_parse_shards, 1 }, /* set number of shards */
Willy Tarreauc8cac042021-09-21 14:31:29 +02001799 { "thread", bind_parse_thread, 1 }, /* set list of allowed threads for this socket */
Willy Tarreau0ccb7442013-01-07 22:54:17 +01001800 { /* END */ },
Willy Tarreau3dcc3412012-09-18 17:17:28 +02001801}};
1802
Willy Tarreau0108d902018-11-25 19:14:37 +01001803INITCALL1(STG_REGISTER, bind_register_keywords, &bind_kws);
1804
Willy Tarreau7ac908b2019-02-27 12:02:18 +01001805/* config keyword parsers */
1806static struct cfg_kw_list cfg_kws = {ILH, {
1807 { CFG_GLOBAL, "tune.listener.multi-queue", cfg_parse_tune_listener_mq },
1808 { 0, NULL, NULL }
1809}};
1810
1811INITCALL1(STG_REGISTER, cfg_register_keywords, &cfg_kws);
1812
Willy Tarreau645513a2010-05-24 20:55:15 +02001813/*
1814 * Local variables:
1815 * c-indent-level: 8
1816 * c-basic-offset: 8
1817 * End:
1818 */