blob: 777f47b9d1ff8677d4ef5ed5dcc96ae132e92fd8 [file] [log] [blame]
Christopher Faulet1329f2a2021-12-16 17:32:56 +01001/*
Willy Tarreau4596fe22022-05-17 19:07:51 +02002 * stream connector management functions
Christopher Faulet1329f2a2021-12-16 17:32:56 +01003 *
4 * Copyright 2021 Christopher Faulet <cfaulet@haproxy.com>
5 *
6 * This program is free software; you can redistribute it and/or
7 * modify it under the terms of the GNU General Public License
8 * as published by the Free Software Foundation; either version
9 * 2 of the License, or (at your option) any later version.
10 *
11 */
12
13#include <haproxy/api.h>
Christopher Faulet37046632022-04-01 11:36:58 +020014#include <haproxy/applet.h>
Christopher Faulet1329f2a2021-12-16 17:32:56 +010015#include <haproxy/connection.h>
Christopher Faulet5e29b762022-04-04 08:58:34 +020016#include <haproxy/check.h>
Christopher Faulet439a41b2024-04-15 19:09:01 +020017#include <haproxy/filters.h>
Christopher Faulet5e29b762022-04-04 08:58:34 +020018#include <haproxy/http_ana.h>
19#include <haproxy/pipe.h>
Christopher Faulet1329f2a2021-12-16 17:32:56 +010020#include <haproxy/pool.h>
Willy Tarreau5edca2f2022-05-27 09:25:10 +020021#include <haproxy/sc_strm.h>
Willy Tarreaucb086c62022-05-27 09:47:12 +020022#include <haproxy/stconn.h>
Christopher Faulet7542fb42023-05-11 14:40:27 +020023#include <haproxy/xref.h>
Christopher Faulet1329f2a2021-12-16 17:32:56 +010024
Willy Tarreau4596fe22022-05-17 19:07:51 +020025DECLARE_POOL(pool_head_connstream, "stconn", sizeof(struct stconn));
Willy Tarreauea59b022022-05-17 17:53:22 +020026DECLARE_POOL(pool_head_sedesc, "sedesc", sizeof(struct sedesc));
Christopher Faulet1329f2a2021-12-16 17:32:56 +010027
Willy Tarreau3a3f4802022-05-17 18:28:19 +020028/* functions used by default on a detached stream connector */
Christopher Fauletcfc11c02023-04-13 16:10:23 +020029static void sc_app_abort(struct stconn *sc);
Christopher Fauletb2b1c3a2023-04-13 16:23:48 +020030static void sc_app_shut(struct stconn *sc);
Willy Tarreau0adb2812022-05-27 10:02:48 +020031static void sc_app_chk_rcv(struct stconn *sc);
32static void sc_app_chk_snd(struct stconn *sc);
Christopher Faulet9ffddd52022-04-01 14:04:29 +020033
Willy Tarreau3a3f4802022-05-17 18:28:19 +020034/* functions used on a mux-based stream connector */
Christopher Fauletcfc11c02023-04-13 16:10:23 +020035static void sc_app_abort_conn(struct stconn *sc);
Christopher Fauletb2b1c3a2023-04-13 16:23:48 +020036static void sc_app_shut_conn(struct stconn *sc);
Willy Tarreau0adb2812022-05-27 10:02:48 +020037static void sc_app_chk_rcv_conn(struct stconn *sc);
38static void sc_app_chk_snd_conn(struct stconn *sc);
Christopher Faulet9ffddd52022-04-01 14:04:29 +020039
Willy Tarreau3a3f4802022-05-17 18:28:19 +020040/* functions used on an applet-based stream connector */
Christopher Fauletcfc11c02023-04-13 16:10:23 +020041static void sc_app_abort_applet(struct stconn *sc);
Christopher Fauletb2b1c3a2023-04-13 16:23:48 +020042static void sc_app_shut_applet(struct stconn *sc);
Willy Tarreau0adb2812022-05-27 10:02:48 +020043static void sc_app_chk_rcv_applet(struct stconn *sc);
44static void sc_app_chk_snd_applet(struct stconn *sc);
Christopher Faulet9ffddd52022-04-01 14:04:29 +020045
Willy Tarreau0adb2812022-05-27 10:02:48 +020046static int sc_conn_process(struct stconn *sc);
47static int sc_conn_recv(struct stconn *sc);
48static int sc_conn_send(struct stconn *sc);
49static int sc_applet_process(struct stconn *sc);
Willy Tarreau2f2318d2022-05-18 10:17:16 +020050
Willy Tarreau3a3f4802022-05-17 18:28:19 +020051/* stream connector operations for connections */
52struct sc_app_ops sc_app_conn_ops = {
53 .chk_rcv = sc_app_chk_rcv_conn,
54 .chk_snd = sc_app_chk_snd_conn,
Christopher Fauletcfc11c02023-04-13 16:10:23 +020055 .abort = sc_app_abort_conn,
Christopher Fauletb2b1c3a2023-04-13 16:23:48 +020056 .shutdown= sc_app_shut_conn,
Willy Tarreau462b9892022-05-18 18:06:53 +020057 .wake = sc_conn_process,
Willy Tarreau2f2318d2022-05-18 10:17:16 +020058 .name = "STRM",
Christopher Faulet9ffddd52022-04-01 14:04:29 +020059};
60
Willy Tarreau3a3f4802022-05-17 18:28:19 +020061/* stream connector operations for embedded tasks */
62struct sc_app_ops sc_app_embedded_ops = {
63 .chk_rcv = sc_app_chk_rcv,
64 .chk_snd = sc_app_chk_snd,
Christopher Fauletcfc11c02023-04-13 16:10:23 +020065 .abort = sc_app_abort,
Christopher Fauletb2b1c3a2023-04-13 16:23:48 +020066 .shutdown= sc_app_shut,
Willy Tarreau2f2318d2022-05-18 10:17:16 +020067 .wake = NULL, /* may never be used */
68 .name = "NONE", /* may never be used */
Christopher Faulet9ffddd52022-04-01 14:04:29 +020069};
70
Willy Tarreau2f2318d2022-05-18 10:17:16 +020071/* stream connector operations for applets */
Willy Tarreau3a3f4802022-05-17 18:28:19 +020072struct sc_app_ops sc_app_applet_ops = {
73 .chk_rcv = sc_app_chk_rcv_applet,
74 .chk_snd = sc_app_chk_snd_applet,
Christopher Fauletcfc11c02023-04-13 16:10:23 +020075 .abort = sc_app_abort_applet,
Christopher Fauletb2b1c3a2023-04-13 16:23:48 +020076 .shutdown= sc_app_shut_applet,
Willy Tarreau19c65a92022-05-27 08:49:24 +020077 .wake = sc_applet_process,
Christopher Faulet5e29b762022-04-04 08:58:34 +020078 .name = "STRM",
79};
80
Willy Tarreau2f2318d2022-05-18 10:17:16 +020081/* stream connector for health checks on connections */
82struct sc_app_ops sc_app_check_ops = {
83 .chk_rcv = NULL,
84 .chk_snd = NULL,
Christopher Fauletcfc11c02023-04-13 16:10:23 +020085 .abort = NULL,
Christopher Fauletb2b1c3a2023-04-13 16:23:48 +020086 .shutdown= NULL,
Willy Tarreau2f2318d2022-05-18 10:17:16 +020087 .wake = wake_srv_chk,
88 .name = "CHCK",
89};
Christopher Faulet5e29b762022-04-04 08:58:34 +020090
Christopher Faulet9ed77422022-04-12 08:51:15 +020091/* Initializes an endpoint */
Willy Tarreauea59b022022-05-17 17:53:22 +020092void sedesc_init(struct sedesc *sedesc)
Christopher Fauletdb90f2a2022-03-22 16:06:25 +010093{
Willy Tarreauea59b022022-05-17 17:53:22 +020094 sedesc->se = NULL;
95 sedesc->conn = NULL;
Willy Tarreauc1054922022-05-18 07:43:52 +020096 sedesc->sc = NULL;
Christopher Faulet4c135682023-02-16 11:09:31 +010097 sedesc->lra = TICK_ETERNITY;
98 sedesc->fsb = TICK_ETERNITY;
Christopher Faulet7542fb42023-05-11 14:40:27 +020099 sedesc->xref.peer = NULL;
Willy Tarreauea59b022022-05-17 17:53:22 +0200100 se_fl_setall(sedesc, SE_FL_NONE);
Christopher Fauletdb90f2a2022-03-22 16:06:25 +0100101}
102
Christopher Faulet9ed77422022-04-12 08:51:15 +0200103/* Tries to alloc an endpoint and initialize it. Returns NULL on failure. */
Willy Tarreauea59b022022-05-17 17:53:22 +0200104struct sedesc *sedesc_new()
Christopher Fauletdb90f2a2022-03-22 16:06:25 +0100105{
Willy Tarreauea59b022022-05-17 17:53:22 +0200106 struct sedesc *sedesc;
Christopher Fauletdb90f2a2022-03-22 16:06:25 +0100107
Willy Tarreauea59b022022-05-17 17:53:22 +0200108 sedesc = pool_alloc(pool_head_sedesc);
109 if (unlikely(!sedesc))
Christopher Fauletdb90f2a2022-03-22 16:06:25 +0100110 return NULL;
111
Willy Tarreauea59b022022-05-17 17:53:22 +0200112 sedesc_init(sedesc);
113 return sedesc;
Christopher Fauletdb90f2a2022-03-22 16:06:25 +0100114}
115
Christopher Faulet9ed77422022-04-12 08:51:15 +0200116/* Releases an endpoint. It is the caller responsibility to be sure it is safe
117 * and it is not shared with another entity
118 */
Willy Tarreauea59b022022-05-17 17:53:22 +0200119void sedesc_free(struct sedesc *sedesc)
Christopher Fauletdb90f2a2022-03-22 16:06:25 +0100120{
Willy Tarreauea59b022022-05-17 17:53:22 +0200121 pool_free(pool_head_sedesc, sedesc);
Christopher Fauletdb90f2a2022-03-22 16:06:25 +0100122}
Christopher Faulet1329f2a2021-12-16 17:32:56 +0100123
Willy Tarreau4596fe22022-05-17 19:07:51 +0200124/* Tries to allocate a new stconn and initialize its main fields. On
Christopher Faulet9ed77422022-04-12 08:51:15 +0200125 * failure, nothing is allocated and NULL is returned. It is an internal
Willy Tarreaub605c422022-05-17 17:04:55 +0200126 * function. The caller must, at least, set the SE_FL_ORPHAN or SE_FL_DETACHED
Christopher Faulet9ed77422022-04-12 08:51:15 +0200127 * flag.
Christopher Faulet1329f2a2021-12-16 17:32:56 +0100128 */
Willy Tarreaua0b58b52022-05-27 08:33:53 +0200129static struct stconn *sc_new(struct sedesc *sedesc)
Christopher Faulet1329f2a2021-12-16 17:32:56 +0100130{
Willy Tarreau0adb2812022-05-27 10:02:48 +0200131 struct stconn *sc;
Christopher Faulet1329f2a2021-12-16 17:32:56 +0100132
Willy Tarreau0adb2812022-05-27 10:02:48 +0200133 sc = pool_alloc(pool_head_connstream);
Christopher Fauletdb90f2a2022-03-22 16:06:25 +0100134
Willy Tarreau0adb2812022-05-27 10:02:48 +0200135 if (unlikely(!sc))
Christopher Fauletdb90f2a2022-03-22 16:06:25 +0100136 goto alloc_error;
Christopher Fauletbb772d02022-03-22 15:28:36 +0100137
Willy Tarreau1d2c79a2022-05-27 11:15:19 +0200138 sc->obj_type = OBJ_TYPE_SC;
Willy Tarreau0adb2812022-05-27 10:02:48 +0200139 sc->flags = SC_FL_NONE;
140 sc->state = SC_ST_INI;
Christopher Fauletbe5cc762023-02-20 08:41:55 +0100141 sc->ioto = TICK_ETERNITY;
Christopher Faulet9aed1122023-05-05 11:25:19 +0200142 sc->room_needed = 0;
Willy Tarreau0adb2812022-05-27 10:02:48 +0200143 sc->app = NULL;
144 sc->app_ops = NULL;
145 sc->src = NULL;
146 sc->dst = NULL;
147 sc->wait_event.tasklet = NULL;
148 sc->wait_event.events = 0;
Christopher Faulet2f35e7b2022-03-31 11:09:28 +0200149
Christopher Faulet9ed77422022-04-12 08:51:15 +0200150 /* If there is no endpoint, allocate a new one now */
Willy Tarreauea59b022022-05-17 17:53:22 +0200151 if (!sedesc) {
152 sedesc = sedesc_new();
153 if (unlikely(!sedesc))
Christopher Fauletb669d682022-03-22 18:37:19 +0100154 goto alloc_error;
155 }
Willy Tarreau0adb2812022-05-27 10:02:48 +0200156 sc->sedesc = sedesc;
157 sedesc->sc = sc;
Christopher Fauletdb90f2a2022-03-22 16:06:25 +0100158
Willy Tarreau0adb2812022-05-27 10:02:48 +0200159 return sc;
Christopher Fauletdb90f2a2022-03-22 16:06:25 +0100160
161 alloc_error:
Willy Tarreau0adb2812022-05-27 10:02:48 +0200162 pool_free(pool_head_connstream, sc);
Christopher Fauletdb90f2a2022-03-22 16:06:25 +0100163 return NULL;
Christopher Faulet1329f2a2021-12-16 17:32:56 +0100164}
165
Willy Tarreau31219282022-05-27 16:21:33 +0200166/* Creates a new stream connector and its associated stream from a mux. <sd> must
167 * be defined. It returns NULL on error. On success, the new stream connector is
Willy Tarreaub605c422022-05-17 17:04:55 +0200168 * returned. In this case, SE_FL_ORPHAN flag is removed.
Christopher Faulet9ed77422022-04-12 08:51:15 +0200169 */
Willy Tarreau31219282022-05-27 16:21:33 +0200170struct stconn *sc_new_from_endp(struct sedesc *sd, struct session *sess, struct buffer *input)
Christopher Fauleta9e8b392022-03-23 11:01:09 +0100171{
Willy Tarreau0adb2812022-05-27 10:02:48 +0200172 struct stconn *sc;
Christopher Fauleta9e8b392022-03-23 11:01:09 +0100173
Willy Tarreau31219282022-05-27 16:21:33 +0200174 sc = sc_new(sd);
Willy Tarreau0adb2812022-05-27 10:02:48 +0200175 if (unlikely(!sc))
Christopher Fauleta9e8b392022-03-23 11:01:09 +0100176 return NULL;
Willy Tarreau0adb2812022-05-27 10:02:48 +0200177 if (unlikely(!stream_new(sess, sc, input))) {
Christopher Faulet3ab72c62022-09-27 09:18:20 +0200178 sd->sc = NULL;
Willy Tarreau7a8ca0a2023-03-20 19:53:14 +0100179 if (sc->sedesc != sd) {
180 /* none was provided so sc_new() allocated one */
181 sedesc_free(sc->sedesc);
182 }
183 pool_free(pool_head_connstream, sc);
Christopher Faulet3ab72c62022-09-27 09:18:20 +0200184 se_fl_set(sd, SE_FL_ORPHAN);
185 return NULL;
Christopher Fauleta9e8b392022-03-23 11:01:09 +0100186 }
Willy Tarreau31219282022-05-27 16:21:33 +0200187 se_fl_clr(sd, SE_FL_ORPHAN);
Willy Tarreau0adb2812022-05-27 10:02:48 +0200188 return sc;
Christopher Fauleta9e8b392022-03-23 11:01:09 +0100189}
190
Willy Tarreau4596fe22022-05-17 19:07:51 +0200191/* Creates a new stream connector from an stream. There is no endpoint here, thus it
Willy Tarreaua0b58b52022-05-27 08:33:53 +0200192 * will be created by sc_new(). So the SE_FL_DETACHED flag is set. It returns
Willy Tarreau4596fe22022-05-17 19:07:51 +0200193 * NULL on error. On success, the new stream connector is returned.
Christopher Faulet9ed77422022-04-12 08:51:15 +0200194 */
Willy Tarreaua0b58b52022-05-27 08:33:53 +0200195struct stconn *sc_new_from_strm(struct stream *strm, unsigned int flags)
Christopher Fauleta9e8b392022-03-23 11:01:09 +0100196{
Willy Tarreau0adb2812022-05-27 10:02:48 +0200197 struct stconn *sc;
Christopher Fauleta9e8b392022-03-23 11:01:09 +0100198
Willy Tarreau0adb2812022-05-27 10:02:48 +0200199 sc = sc_new(NULL);
200 if (unlikely(!sc))
Christopher Fauleta9e8b392022-03-23 11:01:09 +0100201 return NULL;
Willy Tarreau0adb2812022-05-27 10:02:48 +0200202 sc->flags |= flags;
203 sc_ep_set(sc, SE_FL_DETACHED);
204 sc->app = &strm->obj_type;
205 sc->app_ops = &sc_app_embedded_ops;
206 return sc;
Christopher Fauleta9e8b392022-03-23 11:01:09 +0100207}
208
Willy Tarreau4596fe22022-05-17 19:07:51 +0200209/* Creates a new stream connector from an health-check. There is no endpoint here,
Willy Tarreaua0b58b52022-05-27 08:33:53 +0200210 * thus it will be created by sc_new(). So the SE_FL_DETACHED flag is set. It
Willy Tarreau4596fe22022-05-17 19:07:51 +0200211 * returns NULL on error. On success, the new stream connector is returned.
Christopher Faulet9ed77422022-04-12 08:51:15 +0200212 */
Willy Tarreaua0b58b52022-05-27 08:33:53 +0200213struct stconn *sc_new_from_check(struct check *check, unsigned int flags)
Christopher Fauleta9e8b392022-03-23 11:01:09 +0100214{
Willy Tarreau0adb2812022-05-27 10:02:48 +0200215 struct stconn *sc;
Christopher Fauleta9e8b392022-03-23 11:01:09 +0100216
Willy Tarreau0adb2812022-05-27 10:02:48 +0200217 sc = sc_new(NULL);
218 if (unlikely(!sc))
Christopher Fauleta9e8b392022-03-23 11:01:09 +0100219 return NULL;
Willy Tarreau0adb2812022-05-27 10:02:48 +0200220 sc->flags |= flags;
221 sc_ep_set(sc, SE_FL_DETACHED);
222 sc->app = &check->obj_type;
223 sc->app_ops = &sc_app_check_ops;
224 return sc;
Christopher Fauleta9e8b392022-03-23 11:01:09 +0100225}
226
Willy Tarreaua0b58b52022-05-27 08:33:53 +0200227/* Releases a stconn previously allocated by sc_new(), as well as its
Christopher Faulet9ed77422022-04-12 08:51:15 +0200228 * endpoint, if it exists. This function is called internally or on error path.
Christopher Faulet1329f2a2021-12-16 17:32:56 +0100229 */
Willy Tarreau0adb2812022-05-27 10:02:48 +0200230void sc_free(struct stconn *sc)
Christopher Faulet1329f2a2021-12-16 17:32:56 +0100231{
Willy Tarreau0adb2812022-05-27 10:02:48 +0200232 sockaddr_free(&sc->src);
233 sockaddr_free(&sc->dst);
234 if (sc->sedesc) {
235 BUG_ON(!sc_ep_test(sc, SE_FL_DETACHED));
236 sedesc_free(sc->sedesc);
Christopher Fauletdb90f2a2022-03-22 16:06:25 +0100237 }
Tim Duesterhusb1ec21d2023-04-22 17:47:32 +0200238 tasklet_free(sc->wait_event.tasklet);
Willy Tarreau0adb2812022-05-27 10:02:48 +0200239 pool_free(pool_head_connstream, sc);
Christopher Faulet1329f2a2021-12-16 17:32:56 +0100240}
Christopher Fauletcda94ac2021-12-23 17:28:17 +0100241
Willy Tarreau4596fe22022-05-17 19:07:51 +0200242/* Conditionally removes a stream connector if it is detached and if there is no app
Christopher Fauleteb50c012022-04-21 14:22:53 +0200243 * layer defined. Except on error path, this one must be used. if release, the
Willy Tarreaue68bc612022-05-27 11:23:05 +0200244 * pointer on the SC is set to NULL.
Christopher Fauletaa69d8f2022-04-12 18:09:48 +0200245 */
Willy Tarreaue68bc612022-05-27 11:23:05 +0200246static void sc_free_cond(struct stconn **scp)
Christopher Fauletaa69d8f2022-04-12 18:09:48 +0200247{
Willy Tarreaue68bc612022-05-27 11:23:05 +0200248 struct stconn *sc = *scp;
Christopher Fauleteb50c012022-04-21 14:22:53 +0200249
Willy Tarreau0adb2812022-05-27 10:02:48 +0200250 if (!sc->app && (!sc->sedesc || sc_ep_test(sc, SE_FL_DETACHED))) {
251 sc_free(sc);
Willy Tarreaue68bc612022-05-27 11:23:05 +0200252 *scp = NULL;
Christopher Fauleteb50c012022-04-21 14:22:53 +0200253 }
Christopher Fauletaa69d8f2022-04-12 18:09:48 +0200254}
255
Christopher Fauletcda94ac2021-12-23 17:28:17 +0100256
Willy Tarreau4596fe22022-05-17 19:07:51 +0200257/* Attaches a stconn to a mux endpoint and sets the endpoint ctx. Returns
Ilya Shipitsin3b64a282022-07-29 22:26:53 +0500258 * -1 on error and 0 on success. SE_FL_DETACHED flag is removed. This function is
Christopher Faulet9ed77422022-04-12 08:51:15 +0200259 * called from a mux when it is attached to a stream or a health-check.
260 */
Willy Tarreau31219282022-05-27 16:21:33 +0200261int sc_attach_mux(struct stconn *sc, void *sd, void *ctx)
Christopher Fauletcda94ac2021-12-23 17:28:17 +0100262{
Christopher Faulet93882042022-01-19 14:56:50 +0100263 struct connection *conn = ctx;
Willy Tarreau0adb2812022-05-27 10:02:48 +0200264 struct sedesc *sedesc = sc->sedesc;
Christopher Fauletcda94ac2021-12-23 17:28:17 +0100265
Willy Tarreau0adb2812022-05-27 10:02:48 +0200266 if (sc_strm(sc)) {
267 if (!sc->wait_event.tasklet) {
268 sc->wait_event.tasklet = tasklet_new();
269 if (!sc->wait_event.tasklet)
Christopher Faulet2f35e7b2022-03-31 11:09:28 +0200270 return -1;
Willy Tarreau0adb2812022-05-27 10:02:48 +0200271 sc->wait_event.tasklet->process = sc_conn_io_cb;
272 sc->wait_event.tasklet->context = sc;
273 sc->wait_event.events = 0;
Christopher Faulet2f35e7b2022-03-31 11:09:28 +0200274 }
275
Willy Tarreau0adb2812022-05-27 10:02:48 +0200276 sc->app_ops = &sc_app_conn_ops;
Christopher Faulet7542fb42023-05-11 14:40:27 +0200277 xref_create(&sc->sedesc->xref, &sc_opposite(sc)->sedesc->xref);
Christopher Fauletcda94ac2021-12-23 17:28:17 +0100278 }
Willy Tarreau0adb2812022-05-27 10:02:48 +0200279 else if (sc_check(sc)) {
280 if (!sc->wait_event.tasklet) {
281 sc->wait_event.tasklet = tasklet_new();
282 if (!sc->wait_event.tasklet)
Christopher Fauletc95eaef2022-05-18 15:57:15 +0200283 return -1;
Willy Tarreau0adb2812022-05-27 10:02:48 +0200284 sc->wait_event.tasklet->process = srv_chk_io_cb;
285 sc->wait_event.tasklet->context = sc;
286 sc->wait_event.events = 0;
Christopher Fauletc95eaef2022-05-18 15:57:15 +0200287 }
288
Willy Tarreau0adb2812022-05-27 10:02:48 +0200289 sc->app_ops = &sc_app_check_ops;
Christopher Fauletc95eaef2022-05-18 15:57:15 +0200290 }
Willy Tarreaue2f79462023-03-20 19:45:41 +0100291
292 sedesc->se = sd;
293 sedesc->conn = ctx;
294 se_fl_set(sedesc, SE_FL_T_MUX);
295 se_fl_clr(sedesc, SE_FL_DETACHED);
296 if (!conn->ctx)
297 conn->ctx = sc;
Christopher Faulet070b91b2022-03-31 19:27:18 +0200298 return 0;
Christopher Faulet93882042022-01-19 14:56:50 +0100299}
300
Willy Tarreau4596fe22022-05-17 19:07:51 +0200301/* Attaches a stconn to an applet endpoint and sets the endpoint
Ilya Shipitsin3b64a282022-07-29 22:26:53 +0500302 * ctx. Returns -1 on error and 0 on success. SE_FL_DETACHED flag is
Christopher Faulet9ed77422022-04-12 08:51:15 +0200303 * removed. This function is called by a stream when a backend applet is
304 * registered.
305 */
Willy Tarreau31219282022-05-27 16:21:33 +0200306static void sc_attach_applet(struct stconn *sc, void *sd)
Christopher Faulet93882042022-01-19 14:56:50 +0100307{
Willy Tarreau31219282022-05-27 16:21:33 +0200308 sc->sedesc->se = sd;
Willy Tarreau0adb2812022-05-27 10:02:48 +0200309 sc_ep_set(sc, SE_FL_T_APPLET);
310 sc_ep_clr(sc, SE_FL_DETACHED);
Christopher Faulet7542fb42023-05-11 14:40:27 +0200311 if (sc_strm(sc)) {
Willy Tarreau0adb2812022-05-27 10:02:48 +0200312 sc->app_ops = &sc_app_applet_ops;
Christopher Faulet7542fb42023-05-11 14:40:27 +0200313 xref_create(&sc->sedesc->xref, &sc_opposite(sc)->sedesc->xref);
314 }
Christopher Fauletcda94ac2021-12-23 17:28:17 +0100315}
316
Willy Tarreau4596fe22022-05-17 19:07:51 +0200317/* Attaches a stconn to a app layer and sets the relevant
Willy Tarreaub605c422022-05-17 17:04:55 +0200318 * callbacks. Returns -1 on error and 0 on success. SE_FL_ORPHAN flag is
Christopher Faulet9ed77422022-04-12 08:51:15 +0200319 * removed. This function is called by a stream when it is created to attach it
Willy Tarreau4596fe22022-05-17 19:07:51 +0200320 * on the stream connector on the client side.
Christopher Faulet9ed77422022-04-12 08:51:15 +0200321 */
Willy Tarreau0adb2812022-05-27 10:02:48 +0200322int sc_attach_strm(struct stconn *sc, struct stream *strm)
Christopher Fauletcda94ac2021-12-23 17:28:17 +0100323{
Willy Tarreau0adb2812022-05-27 10:02:48 +0200324 sc->app = &strm->obj_type;
325 sc_ep_clr(sc, SE_FL_ORPHAN);
Christopher Fauletf4ef3d92023-09-06 08:46:15 +0200326 sc_ep_report_read_activity(sc);
Willy Tarreau0adb2812022-05-27 10:02:48 +0200327 if (sc_ep_test(sc, SE_FL_T_MUX)) {
328 sc->wait_event.tasklet = tasklet_new();
329 if (!sc->wait_event.tasklet)
Christopher Faulet2f35e7b2022-03-31 11:09:28 +0200330 return -1;
Willy Tarreau0adb2812022-05-27 10:02:48 +0200331 sc->wait_event.tasklet->process = sc_conn_io_cb;
332 sc->wait_event.tasklet->context = sc;
333 sc->wait_event.events = 0;
Christopher Faulet2f35e7b2022-03-31 11:09:28 +0200334
Willy Tarreau0adb2812022-05-27 10:02:48 +0200335 sc->app_ops = &sc_app_conn_ops;
Christopher Fauletcda94ac2021-12-23 17:28:17 +0100336 }
Willy Tarreau0adb2812022-05-27 10:02:48 +0200337 else if (sc_ep_test(sc, SE_FL_T_APPLET)) {
338 sc->app_ops = &sc_app_applet_ops;
Christopher Fauleta9e8b392022-03-23 11:01:09 +0100339 }
340 else {
Willy Tarreau0adb2812022-05-27 10:02:48 +0200341 sc->app_ops = &sc_app_embedded_ops;
Christopher Fauleta9e8b392022-03-23 11:01:09 +0100342 }
Christopher Fauletcda94ac2021-12-23 17:28:17 +0100343 return 0;
344}
345
Willy Tarreau4596fe22022-05-17 19:07:51 +0200346/* Detaches the stconn from the endpoint, if any. For a connecrion, if a
Christopher Faulet9ed77422022-04-12 08:51:15 +0200347 * mux owns the connection ->detach() callback is called. Otherwise, it means
Willy Tarreau4596fe22022-05-17 19:07:51 +0200348 * the stream connector owns the connection. In this case the connection is closed
Christopher Faulet9ed77422022-04-12 08:51:15 +0200349 * and released. For an applet, the appctx is released. If still allocated, the
350 * endpoint is reset and flag as detached. If the app layer is also detached,
Willy Tarreau4596fe22022-05-17 19:07:51 +0200351 * the stream connector is released.
Christopher Fauletcda94ac2021-12-23 17:28:17 +0100352 */
Willy Tarreaue68bc612022-05-27 11:23:05 +0200353static void sc_detach_endp(struct stconn **scp)
Christopher Fauletcda94ac2021-12-23 17:28:17 +0100354{
Willy Tarreaue68bc612022-05-27 11:23:05 +0200355 struct stconn *sc = *scp;
Christopher Faulet6eb53b12023-05-15 09:53:29 +0200356 struct xref *peer;
Christopher Fauleteb50c012022-04-21 14:22:53 +0200357
Willy Tarreau0adb2812022-05-27 10:02:48 +0200358 if (!sc)
Christopher Fauleteb50c012022-04-21 14:22:53 +0200359 return;
360
Christopher Faulet7542fb42023-05-11 14:40:27 +0200361
Christopher Faulet6eb53b12023-05-15 09:53:29 +0200362 /* Remove my link in the original objects. */
363 peer = xref_get_peer_and_lock(&sc->sedesc->xref);
364 if (peer)
365 xref_disconnect(&sc->sedesc->xref, peer);
Christopher Faulet7542fb42023-05-11 14:40:27 +0200366
Willy Tarreau0adb2812022-05-27 10:02:48 +0200367 if (sc_ep_test(sc, SE_FL_T_MUX)) {
368 struct connection *conn = __sc_conn(sc);
369 struct sedesc *sedesc = sc->sedesc;
Christopher Fauletcda94ac2021-12-23 17:28:17 +0100370
Christopher Fauletcda94ac2021-12-23 17:28:17 +0100371 if (conn->mux) {
Willy Tarreau0adb2812022-05-27 10:02:48 +0200372 if (sc->wait_event.events != 0)
373 conn->mux->unsubscribe(sc, sc->wait_event.events, &sc->wait_event);
Willy Tarreau798465b2022-05-17 18:20:02 +0200374 se_fl_set(sedesc, SE_FL_ORPHAN);
Willy Tarreauc1054922022-05-18 07:43:52 +0200375 sedesc->sc = NULL;
Willy Tarreau0adb2812022-05-27 10:02:48 +0200376 sc->sedesc = NULL;
Willy Tarreau798465b2022-05-17 18:20:02 +0200377 conn->mux->detach(sedesc);
Christopher Fauletcda94ac2021-12-23 17:28:17 +0100378 }
379 else {
380 /* It's too early to have a mux, let's just destroy
381 * the connection
382 */
383 conn_stop_tracking(conn);
384 conn_full_close(conn);
385 if (conn->destroy_cb)
386 conn->destroy_cb(conn);
387 conn_free(conn);
388 }
389 }
Willy Tarreau0adb2812022-05-27 10:02:48 +0200390 else if (sc_ep_test(sc, SE_FL_T_APPLET)) {
391 struct appctx *appctx = __sc_appctx(sc);
Christopher Fauletdb90f2a2022-03-22 16:06:25 +0100392
Willy Tarreau0adb2812022-05-27 10:02:48 +0200393 sc_ep_set(sc, SE_FL_ORPHAN);
394 sc->sedesc->sc = NULL;
395 sc->sedesc = NULL;
Willy Tarreau1c3ead42022-05-10 19:42:22 +0200396 appctx_shut(appctx);
397 appctx_free(appctx);
Christopher Fauletcda94ac2021-12-23 17:28:17 +0100398 }
399
Willy Tarreau0adb2812022-05-27 10:02:48 +0200400 if (sc->sedesc) {
Willy Tarreauda59c892022-05-27 17:03:34 +0200401 /* the SD wasn't used and can be recycled */
Willy Tarreau0adb2812022-05-27 10:02:48 +0200402 sc->sedesc->se = NULL;
403 sc->sedesc->conn = NULL;
Willy Tarreauda59c892022-05-27 17:03:34 +0200404 sc->sedesc->flags = 0;
Willy Tarreau0adb2812022-05-27 10:02:48 +0200405 sc_ep_set(sc, SE_FL_DETACHED);
Christopher Fauletdb90f2a2022-03-22 16:06:25 +0100406 }
407
Willy Tarreaue68bc612022-05-27 11:23:05 +0200408 /* FIXME: Rest SC for now but must be reviewed. SC flags are only
Christopher Fauletc36de9d2022-01-06 08:44:58 +0100409 * connection related for now but this will evolved
410 */
Willy Tarreau0adb2812022-05-27 10:02:48 +0200411 sc->flags &= SC_FL_ISBACK;
412 if (sc_strm(sc))
413 sc->app_ops = &sc_app_embedded_ops;
Willy Tarreau2f2318d2022-05-18 10:17:16 +0200414 else
Willy Tarreau0adb2812022-05-27 10:02:48 +0200415 sc->app_ops = NULL;
Willy Tarreaue68bc612022-05-27 11:23:05 +0200416 sc_free_cond(scp);
Christopher Fauletc36de9d2022-01-06 08:44:58 +0100417}
418
Willy Tarreau4596fe22022-05-17 19:07:51 +0200419/* Detaches the stconn from the app layer. If there is no endpoint attached
420 * to the stconn
Christopher Faulet9ed77422022-04-12 08:51:15 +0200421 */
Willy Tarreaue68bc612022-05-27 11:23:05 +0200422static void sc_detach_app(struct stconn **scp)
Christopher Fauletc36de9d2022-01-06 08:44:58 +0100423{
Willy Tarreaue68bc612022-05-27 11:23:05 +0200424 struct stconn *sc = *scp;
Christopher Fauleteb50c012022-04-21 14:22:53 +0200425
Willy Tarreau0adb2812022-05-27 10:02:48 +0200426 if (!sc)
Christopher Fauleteb50c012022-04-21 14:22:53 +0200427 return;
428
Willy Tarreau0adb2812022-05-27 10:02:48 +0200429 sc->app = NULL;
430 sc->app_ops = NULL;
431 sockaddr_free(&sc->src);
432 sockaddr_free(&sc->dst);
Christopher Faulet2f35e7b2022-03-31 11:09:28 +0200433
Tim Duesterhusb1ec21d2023-04-22 17:47:32 +0200434 tasklet_free(sc->wait_event.tasklet);
Willy Tarreau0adb2812022-05-27 10:02:48 +0200435 sc->wait_event.tasklet = NULL;
436 sc->wait_event.events = 0;
Willy Tarreaue68bc612022-05-27 11:23:05 +0200437 sc_free_cond(scp);
Christopher Fauleteb50c012022-04-21 14:22:53 +0200438}
439
Willy Tarreau4596fe22022-05-17 19:07:51 +0200440/* Destroy the stconn. It is detached from its endpoint and its
441 * application. After this call, the stconn must be considered as released.
Christopher Fauleteb50c012022-04-21 14:22:53 +0200442 */
Willy Tarreau0adb2812022-05-27 10:02:48 +0200443void sc_destroy(struct stconn *sc)
Christopher Fauleteb50c012022-04-21 14:22:53 +0200444{
Willy Tarreau0adb2812022-05-27 10:02:48 +0200445 sc_detach_endp(&sc);
446 sc_detach_app(&sc);
447 BUG_ON_HOT(sc);
Christopher Fauletcda94ac2021-12-23 17:28:17 +0100448}
Christopher Faulet9ec2f4d2022-03-23 15:15:29 +0100449
Willy Tarreau4596fe22022-05-17 19:07:51 +0200450/* Resets the stream connector endpoint. It happens when the app layer want to renew
Christopher Faulet9ed77422022-04-12 08:51:15 +0200451 * its endpoint. For a connection retry for instance. If a mux or an applet is
Ilya Shipitsin3b64a282022-07-29 22:26:53 +0500452 * attached, a new endpoint is created. Returns -1 on error and 0 on success.
Christopher Faulet9ed77422022-04-12 08:51:15 +0200453 */
Willy Tarreau0adb2812022-05-27 10:02:48 +0200454int sc_reset_endp(struct stconn *sc)
Christopher Faulet9ec2f4d2022-03-23 15:15:29 +0100455{
Willy Tarreau31219282022-05-27 16:21:33 +0200456 struct sedesc *new_sd;
Christopher Fauletb041b232022-03-24 10:27:02 +0100457
Willy Tarreau0adb2812022-05-27 10:02:48 +0200458 BUG_ON(!sc->app);
Christopher Fauleta6c4a482022-04-28 18:25:24 +0200459
Willy Tarreau0adb2812022-05-27 10:02:48 +0200460 if (!__sc_endp(sc)) {
Christopher Fauletb041b232022-03-24 10:27:02 +0100461 /* endpoint not attached or attached to a mux with no
462 * target. Thus the endpoint will not be release but just
Willy Tarreau0adb2812022-05-27 10:02:48 +0200463 * reset. The app is still attached, the sc will not be
Christopher Fauleteb50c012022-04-21 14:22:53 +0200464 * released.
Christopher Fauletb041b232022-03-24 10:27:02 +0100465 */
Willy Tarreau0adb2812022-05-27 10:02:48 +0200466 sc_detach_endp(&sc);
Christopher Fauletb041b232022-03-24 10:27:02 +0100467 return 0;
Christopher Faulet9ec2f4d2022-03-23 15:15:29 +0100468 }
Christopher Fauletb041b232022-03-24 10:27:02 +0100469
470 /* allocate the new endpoint first to be able to set error if it
471 * fails */
Willy Tarreau31219282022-05-27 16:21:33 +0200472 new_sd = sedesc_new();
Christopher Faulet638fe6a2023-04-14 10:28:28 +0200473 if (!unlikely(new_sd))
Christopher Fauletb041b232022-03-24 10:27:02 +0100474 return -1;
Christopher Fauletb041b232022-03-24 10:27:02 +0100475
Willy Tarreau0adb2812022-05-27 10:02:48 +0200476 /* The app is still attached, the sc will not be released */
477 sc_detach_endp(&sc);
Willy Tarreau6a378d12022-08-11 13:56:42 +0200478 BUG_ON(!sc);
Willy Tarreau0adb2812022-05-27 10:02:48 +0200479 BUG_ON(sc->sedesc);
Willy Tarreau31219282022-05-27 16:21:33 +0200480 sc->sedesc = new_sd;
Willy Tarreau0adb2812022-05-27 10:02:48 +0200481 sc->sedesc->sc = sc;
482 sc_ep_set(sc, SE_FL_DETACHED);
Christopher Faulet9ec2f4d2022-03-23 15:15:29 +0100483 return 0;
484}
Christopher Faulet37046632022-04-01 11:36:58 +0200485
486
Willy Tarreaue68bc612022-05-27 11:23:05 +0200487/* Create an applet to handle a stream connector as a new appctx. The SC will
Christopher Faulet37046632022-04-01 11:36:58 +0200488 * wake it up every time it is solicited. The appctx must be deleted by the task
Willy Tarreau19c65a92022-05-27 08:49:24 +0200489 * handler using sc_detach_endp(), possibly from within the function itself.
Christopher Faulet37046632022-04-01 11:36:58 +0200490 * It also pre-initializes the applet's context and returns it (or NULL in case
491 * it could not be allocated).
492 */
Willy Tarreau0adb2812022-05-27 10:02:48 +0200493struct appctx *sc_applet_create(struct stconn *sc, struct applet *app)
Christopher Faulet37046632022-04-01 11:36:58 +0200494{
495 struct appctx *appctx;
496
Willy Tarreau0adb2812022-05-27 10:02:48 +0200497 appctx = appctx_new_here(app, sc->sedesc);
Christopher Faulet37046632022-04-01 11:36:58 +0200498 if (!appctx)
499 return NULL;
Willy Tarreau0adb2812022-05-27 10:02:48 +0200500 sc_attach_applet(sc, appctx);
501 appctx->t->nice = __sc_strm(sc)->task->nice;
Willy Tarreau90e8b452022-05-25 18:21:43 +0200502 applet_need_more_data(appctx);
Christopher Faulet37046632022-04-01 11:36:58 +0200503 appctx_wakeup(appctx);
Christopher Fauleta33ff7a2022-04-21 11:52:07 +0200504
Willy Tarreau0adb2812022-05-27 10:02:48 +0200505 sc->state = SC_ST_RDY;
Christopher Faulet37046632022-04-01 11:36:58 +0200506 return appctx;
507}
508
Ilya Shipitsin07be66d2023-04-01 12:26:42 +0200509/* Conditionally forward the close to the write side. It return 1 if it can be
Christopher Fauleteb3f26d2023-02-08 16:18:48 +0100510 * forwarded. It is the caller responsibility to forward the close to the write
Christopher Faulete38534c2023-04-13 15:45:24 +0200511 * side. Otherwise, 0 is returned. In this case, SC_FL_SHUT_WANTED flag may be set on
Christopher Faulet87633c32023-04-03 18:32:50 +0200512 * the consumer SC if we are only waiting for the outgoing data to be flushed.
Christopher Fauleteb3f26d2023-02-08 16:18:48 +0100513 */
Christopher Fauletb2b1c3a2023-04-13 16:23:48 +0200514static inline int sc_cond_forward_shut(struct stconn *sc)
Christopher Fauleteb3f26d2023-02-08 16:18:48 +0100515{
Christopher Faulet406b81c2023-09-06 08:59:33 +0200516
Christopher Fauleteb3f26d2023-02-08 16:18:48 +0100517 /* The close must not be forwarded */
Christopher Fauletca5309a2023-04-17 16:17:32 +0200518 if (!(sc->flags & (SC_FL_EOS|SC_FL_ABRT_DONE)) || !(sc->flags & SC_FL_NOHALF))
Christopher Fauleteb3f26d2023-02-08 16:18:48 +0100519 return 0;
520
Christopher Fauletc9783272024-01-05 16:48:40 +0100521 if (!channel_is_empty(sc_ic(sc)) && !(sc_ic(sc)->flags & CF_WRITE_TIMEOUT)) {
Christopher Fauletdf7cd712023-04-13 15:56:26 +0200522 /* the shutdown cannot be forwarded now because
Christopher Fauleteb3f26d2023-02-08 16:18:48 +0100523 * we should flush outgoing data first. But instruct the output
524 * channel it should be done ASAP.
525 */
Christopher Fauletdf7cd712023-04-13 15:56:26 +0200526 sc_schedule_shutdown(sc);
Christopher Fauleteb3f26d2023-02-08 16:18:48 +0100527 return 0;
528 }
529
530 /* the close can be immediately forwarded to the write side */
531 return 1;
532}
533
Christopher Faulet9ffddd52022-04-01 14:04:29 +0200534/*
Willy Tarreau4596fe22022-05-17 19:07:51 +0200535 * This function performs a shutdown-read on a detached stream connector in a
Christopher Faulet9ffddd52022-04-01 14:04:29 +0200536 * connected or init state (it does nothing for other states). It either shuts
537 * the read side or marks itself as closed. The buffer flags are updated to
Willy Tarreaucb041662022-05-17 19:44:42 +0200538 * reflect the new state. If the stream connector has SC_FL_NOHALF, we also
Christopher Faulet9ffddd52022-04-01 14:04:29 +0200539 * forward the close to the write side. The owner task is woken up if it exists.
540 */
Christopher Fauletcfc11c02023-04-13 16:10:23 +0200541static void sc_app_abort(struct stconn *sc)
Christopher Faulet9ffddd52022-04-01 14:04:29 +0200542{
Willy Tarreau0adb2812022-05-27 10:02:48 +0200543 struct channel *ic = sc_ic(sc);
Christopher Faulet9ffddd52022-04-01 14:04:29 +0200544
Christopher Fauletca5309a2023-04-17 16:17:32 +0200545 if (sc->flags & (SC_FL_EOS|SC_FL_ABRT_DONE))
Christopher Fauletc665bb52023-04-04 10:06:57 +0200546 return;
Christopher Faulet87633c32023-04-03 18:32:50 +0200547
Christopher Faulet0c370ee2023-04-13 16:05:13 +0200548 sc->flags |= SC_FL_ABRT_DONE;
Christopher Faulet87633c32023-04-03 18:32:50 +0200549 ic->flags |= CF_READ_EVENT;
Christopher Faulet9ffddd52022-04-01 14:04:29 +0200550
Willy Tarreau0adb2812022-05-27 10:02:48 +0200551 if (!sc_state_in(sc->state, SC_SB_CON|SC_SB_RDY|SC_SB_EST))
Christopher Faulet9ffddd52022-04-01 14:04:29 +0200552 return;
553
Christopher Faulet208c7122023-04-13 16:16:15 +0200554 if (sc->flags & SC_FL_SHUT_DONE) {
Willy Tarreau0adb2812022-05-27 10:02:48 +0200555 sc->state = SC_ST_DIS;
Christopher Fauletca679922022-07-20 13:24:04 +0200556 if (sc->flags & SC_FL_ISBACK)
557 __sc_strm(sc)->conn_exp = TICK_ETERNITY;
Christopher Faulet9ffddd52022-04-01 14:04:29 +0200558 }
Christopher Fauletb2b1c3a2023-04-13 16:23:48 +0200559 else if (sc_cond_forward_shut(sc))
560 return sc_app_shut(sc);
Christopher Faulet9ffddd52022-04-01 14:04:29 +0200561
562 /* note that if the task exists, it must unregister itself once it runs */
Willy Tarreau0adb2812022-05-27 10:02:48 +0200563 if (!(sc->flags & SC_FL_DONT_WAKE))
564 task_wakeup(sc_strm_task(sc), TASK_WOKEN_IO);
Christopher Faulet9ffddd52022-04-01 14:04:29 +0200565}
566
567/*
Willy Tarreau4596fe22022-05-17 19:07:51 +0200568 * This function performs a shutdown-write on a detached stream connector in a
Christopher Faulet9ffddd52022-04-01 14:04:29 +0200569 * connected or init state (it does nothing for other states). It either shuts
570 * the write side or marks itself as closed. The buffer flags are updated to
Willy Tarreaue68bc612022-05-27 11:23:05 +0200571 * reflect the new state. It does also close everything if the SC was marked as
Christopher Faulet9ffddd52022-04-01 14:04:29 +0200572 * being in error state. The owner task is woken up if it exists.
573 */
Christopher Fauletb2b1c3a2023-04-13 16:23:48 +0200574static void sc_app_shut(struct stconn *sc)
Christopher Faulet9ffddd52022-04-01 14:04:29 +0200575{
Willy Tarreau0adb2812022-05-27 10:02:48 +0200576 struct channel *ic = sc_ic(sc);
577 struct channel *oc = sc_oc(sc);
Christopher Faulet9ffddd52022-04-01 14:04:29 +0200578
Christopher Faulete38534c2023-04-13 15:45:24 +0200579 sc->flags &= ~SC_FL_SHUT_WANTED;
Christopher Faulet208c7122023-04-13 16:16:15 +0200580 if (sc->flags & SC_FL_SHUT_DONE)
Christopher Faulet9ffddd52022-04-01 14:04:29 +0200581 return;
Christopher Faulet208c7122023-04-13 16:16:15 +0200582 sc->flags |= SC_FL_SHUT_DONE;
Christopher Faulet87633c32023-04-03 18:32:50 +0200583 oc->flags |= CF_WRITE_EVENT;
Christopher Fauletbcdcfad2023-02-20 08:36:53 +0100584 sc_set_hcto(sc);
Christopher Faulet9ffddd52022-04-01 14:04:29 +0200585
Willy Tarreau0adb2812022-05-27 10:02:48 +0200586 switch (sc->state) {
Willy Tarreau026e8fb2022-05-17 19:47:17 +0200587 case SC_ST_RDY:
588 case SC_ST_EST:
Christopher Faulet9ffddd52022-04-01 14:04:29 +0200589 /* we have to shut before closing, otherwise some short messages
590 * may never leave the system, especially when there are remaining
591 * unread data in the socket input buffer, or when nolinger is set.
Willy Tarreaucb041662022-05-17 19:44:42 +0200592 * However, if SC_FL_NOLINGER is explicitly set, we know there is
Christopher Faulet9ffddd52022-04-01 14:04:29 +0200593 * no risk so we close both sides immediately.
594 */
Christopher Fauletca5309a2023-04-17 16:17:32 +0200595 if (!(sc->flags & (SC_FL_ERROR|SC_FL_NOLINGER|SC_FL_EOS|SC_FL_ABRT_DONE)) &&
Christopher Fauletad46e522023-04-14 11:59:15 +0200596 !(ic->flags & CF_DONT_READ))
Christopher Faulet9ffddd52022-04-01 14:04:29 +0200597 return;
598
Willy Tarreau476c2802022-11-14 07:36:42 +0100599 __fallthrough;
Willy Tarreau026e8fb2022-05-17 19:47:17 +0200600 case SC_ST_CON:
601 case SC_ST_CER:
602 case SC_ST_QUE:
603 case SC_ST_TAR:
Christopher Faulet9ffddd52022-04-01 14:04:29 +0200604 /* Note that none of these states may happen with applets */
Willy Tarreau0adb2812022-05-27 10:02:48 +0200605 sc->state = SC_ST_DIS;
Willy Tarreau476c2802022-11-14 07:36:42 +0100606 __fallthrough;
Christopher Faulet9ffddd52022-04-01 14:04:29 +0200607 default:
Willy Tarreau0adb2812022-05-27 10:02:48 +0200608 sc->flags &= ~SC_FL_NOLINGER;
Christopher Faulet0c370ee2023-04-13 16:05:13 +0200609 sc->flags |= SC_FL_ABRT_DONE;
Christopher Fauletca679922022-07-20 13:24:04 +0200610 if (sc->flags & SC_FL_ISBACK)
611 __sc_strm(sc)->conn_exp = TICK_ETERNITY;
Christopher Faulet9ffddd52022-04-01 14:04:29 +0200612 }
613
614 /* note that if the task exists, it must unregister itself once it runs */
Willy Tarreau0adb2812022-05-27 10:02:48 +0200615 if (!(sc->flags & SC_FL_DONT_WAKE))
616 task_wakeup(sc_strm_task(sc), TASK_WOKEN_IO);
Christopher Faulet9ffddd52022-04-01 14:04:29 +0200617}
618
619/* default chk_rcv function for scheduled tasks */
Willy Tarreau0adb2812022-05-27 10:02:48 +0200620static void sc_app_chk_rcv(struct stconn *sc)
Christopher Faulet9ffddd52022-04-01 14:04:29 +0200621{
Willy Tarreau0adb2812022-05-27 10:02:48 +0200622 struct channel *ic = sc_ic(sc);
Christopher Faulet9ffddd52022-04-01 14:04:29 +0200623
Christopher Faulet9ffddd52022-04-01 14:04:29 +0200624 if (ic->pipe) {
625 /* stop reading */
Christopher Faulet7b3d38a2023-05-05 11:28:45 +0200626 sc_need_room(sc, -1);
Christopher Faulet9ffddd52022-04-01 14:04:29 +0200627 }
628 else {
629 /* (re)start reading */
Willy Tarreau0adb2812022-05-27 10:02:48 +0200630 if (!(sc->flags & SC_FL_DONT_WAKE))
631 task_wakeup(sc_strm_task(sc), TASK_WOKEN_IO);
Christopher Faulet9ffddd52022-04-01 14:04:29 +0200632 }
633}
634
635/* default chk_snd function for scheduled tasks */
Willy Tarreau0adb2812022-05-27 10:02:48 +0200636static void sc_app_chk_snd(struct stconn *sc)
Christopher Faulet9ffddd52022-04-01 14:04:29 +0200637{
Willy Tarreau0adb2812022-05-27 10:02:48 +0200638 struct channel *oc = sc_oc(sc);
Christopher Faulet9ffddd52022-04-01 14:04:29 +0200639
Christopher Faulet208c7122023-04-13 16:16:15 +0200640 if (unlikely(sc->state != SC_ST_EST || (sc->flags & SC_FL_SHUT_DONE)))
Christopher Faulet9ffddd52022-04-01 14:04:29 +0200641 return;
642
Willy Tarreau0adb2812022-05-27 10:02:48 +0200643 if (!sc_ep_test(sc, SE_FL_WAIT_DATA) || /* not waiting for data */
Christopher Faulet9ffddd52022-04-01 14:04:29 +0200644 channel_is_empty(oc)) /* called with nothing to send ! */
645 return;
646
647 /* Otherwise there are remaining data to be sent in the buffer,
648 * so we tell the handler.
649 */
Willy Tarreau0adb2812022-05-27 10:02:48 +0200650 sc_ep_clr(sc, SE_FL_WAIT_DATA);
Willy Tarreau0adb2812022-05-27 10:02:48 +0200651 if (!(sc->flags & SC_FL_DONT_WAKE))
652 task_wakeup(sc_strm_task(sc), TASK_WOKEN_IO);
Christopher Faulet9ffddd52022-04-01 14:04:29 +0200653}
654
655/*
Willy Tarreau3a3f4802022-05-17 18:28:19 +0200656 * This function performs a shutdown-read on a stream connector attached to
Christopher Faulet9ffddd52022-04-01 14:04:29 +0200657 * a connection in a connected or init state (it does nothing for other
658 * states). It either shuts the read side or marks itself as closed. The buffer
Willy Tarreau3a3f4802022-05-17 18:28:19 +0200659 * flags are updated to reflect the new state. If the stream connector has
Willy Tarreaucb041662022-05-17 19:44:42 +0200660 * SC_FL_NOHALF, we also forward the close to the write side. If a control
Christopher Faulet9ffddd52022-04-01 14:04:29 +0200661 * layer is defined, then it is supposed to be a socket layer and file
662 * descriptors are then shutdown or closed accordingly. The function
663 * automatically disables polling if needed.
664 */
Christopher Fauletcfc11c02023-04-13 16:10:23 +0200665static void sc_app_abort_conn(struct stconn *sc)
Christopher Faulet9ffddd52022-04-01 14:04:29 +0200666{
Willy Tarreau0adb2812022-05-27 10:02:48 +0200667 struct channel *ic = sc_ic(sc);
Christopher Faulet9ffddd52022-04-01 14:04:29 +0200668
Willy Tarreau0adb2812022-05-27 10:02:48 +0200669 BUG_ON(!sc_conn(sc));
Christopher Faulet9ffddd52022-04-01 14:04:29 +0200670
Christopher Fauletca5309a2023-04-17 16:17:32 +0200671 if (sc->flags & (SC_FL_EOS|SC_FL_ABRT_DONE))
Christopher Faulet9ffddd52022-04-01 14:04:29 +0200672 return;
Christopher Faulet0c370ee2023-04-13 16:05:13 +0200673 sc->flags |= SC_FL_ABRT_DONE;
Christopher Faulet87633c32023-04-03 18:32:50 +0200674 ic->flags |= CF_READ_EVENT;
Christopher Faulet9ffddd52022-04-01 14:04:29 +0200675
Willy Tarreau0adb2812022-05-27 10:02:48 +0200676 if (!sc_state_in(sc->state, SC_SB_CON|SC_SB_RDY|SC_SB_EST))
Christopher Faulet9ffddd52022-04-01 14:04:29 +0200677 return;
678
Christopher Faulet208c7122023-04-13 16:16:15 +0200679 if (sc->flags & SC_FL_SHUT_DONE) {
Willy Tarreau0adb2812022-05-27 10:02:48 +0200680 sc_conn_shut(sc);
681 sc->state = SC_ST_DIS;
Christopher Fauletca679922022-07-20 13:24:04 +0200682 if (sc->flags & SC_FL_ISBACK)
683 __sc_strm(sc)->conn_exp = TICK_ETERNITY;
Christopher Faulet9ffddd52022-04-01 14:04:29 +0200684 }
Christopher Fauletb2b1c3a2023-04-13 16:23:48 +0200685 else if (sc_cond_forward_shut(sc))
686 return sc_app_shut_conn(sc);
Christopher Faulet9ffddd52022-04-01 14:04:29 +0200687}
688
689/*
Willy Tarreau3a3f4802022-05-17 18:28:19 +0200690 * This function performs a shutdown-write on a stream connector attached to
Christopher Faulet9ffddd52022-04-01 14:04:29 +0200691 * a connection in a connected or init state (it does nothing for other
692 * states). It either shuts the write side or marks itself as closed. The
693 * buffer flags are updated to reflect the new state. It does also close
Willy Tarreaue68bc612022-05-27 11:23:05 +0200694 * everything if the SC was marked as being in error state. If there is a
Christopher Faulet9ffddd52022-04-01 14:04:29 +0200695 * data-layer shutdown, it is called.
696 */
Christopher Fauletb2b1c3a2023-04-13 16:23:48 +0200697static void sc_app_shut_conn(struct stconn *sc)
Christopher Faulet9ffddd52022-04-01 14:04:29 +0200698{
Willy Tarreau0adb2812022-05-27 10:02:48 +0200699 struct channel *ic = sc_ic(sc);
700 struct channel *oc = sc_oc(sc);
Christopher Faulet9ffddd52022-04-01 14:04:29 +0200701
Willy Tarreau0adb2812022-05-27 10:02:48 +0200702 BUG_ON(!sc_conn(sc));
Christopher Faulet9ffddd52022-04-01 14:04:29 +0200703
Christopher Faulete38534c2023-04-13 15:45:24 +0200704 sc->flags &= ~SC_FL_SHUT_WANTED;
Christopher Faulet208c7122023-04-13 16:16:15 +0200705 if (sc->flags & SC_FL_SHUT_DONE)
Christopher Faulet9ffddd52022-04-01 14:04:29 +0200706 return;
Christopher Faulet208c7122023-04-13 16:16:15 +0200707 sc->flags |= SC_FL_SHUT_DONE;
Christopher Faulet87633c32023-04-03 18:32:50 +0200708 oc->flags |= CF_WRITE_EVENT;
Christopher Fauletbcdcfad2023-02-20 08:36:53 +0100709 sc_set_hcto(sc);
Christopher Faulet9ffddd52022-04-01 14:04:29 +0200710
Willy Tarreau0adb2812022-05-27 10:02:48 +0200711 switch (sc->state) {
Willy Tarreau026e8fb2022-05-17 19:47:17 +0200712 case SC_ST_RDY:
713 case SC_ST_EST:
Christopher Faulet9ffddd52022-04-01 14:04:29 +0200714 /* we have to shut before closing, otherwise some short messages
715 * may never leave the system, especially when there are remaining
716 * unread data in the socket input buffer, or when nolinger is set.
Willy Tarreaucb041662022-05-17 19:44:42 +0200717 * However, if SC_FL_NOLINGER is explicitly set, we know there is
Christopher Faulet9ffddd52022-04-01 14:04:29 +0200718 * no risk so we close both sides immediately.
719 */
720
Christopher Faulet25d9fe52023-04-14 12:09:35 +0200721 if (sc->flags & SC_FL_ERROR) {
Christopher Faulet9ffddd52022-04-01 14:04:29 +0200722 /* quick close, the socket is already shut anyway */
723 }
Willy Tarreau0adb2812022-05-27 10:02:48 +0200724 else if (sc->flags & SC_FL_NOLINGER) {
Christopher Faulet9ffddd52022-04-01 14:04:29 +0200725 /* unclean data-layer shutdown, typically an aborted request
726 * or a forwarded shutdown from a client to a server due to
727 * option abortonclose. No need for the TLS layer to try to
728 * emit a shutdown message.
729 */
Willy Tarreau0adb2812022-05-27 10:02:48 +0200730 sc_conn_shutw(sc, CO_SHW_SILENT);
Christopher Faulet9ffddd52022-04-01 14:04:29 +0200731 }
732 else {
733 /* clean data-layer shutdown. This only happens on the
734 * frontend side, or on the backend side when forwarding
735 * a client close in TCP mode or in HTTP TUNNEL mode
736 * while option abortonclose is set. We want the TLS
737 * layer to try to signal it to the peer before we close.
738 */
Willy Tarreau0adb2812022-05-27 10:02:48 +0200739 sc_conn_shutw(sc, CO_SHW_NORMAL);
Christopher Faulet9ffddd52022-04-01 14:04:29 +0200740
Christopher Fauletca5309a2023-04-17 16:17:32 +0200741 if (!(sc->flags & (SC_FL_EOS|SC_FL_ABRT_DONE)) && !(ic->flags & CF_DONT_READ))
Christopher Faulet9ffddd52022-04-01 14:04:29 +0200742 return;
743 }
744
Willy Tarreau476c2802022-11-14 07:36:42 +0100745 __fallthrough;
Willy Tarreau026e8fb2022-05-17 19:47:17 +0200746 case SC_ST_CON:
Christopher Faulet9ffddd52022-04-01 14:04:29 +0200747 /* we may have to close a pending connection, and mark the
Christopher Fauletcfc11c02023-04-13 16:10:23 +0200748 * response buffer as abort
Christopher Faulet9ffddd52022-04-01 14:04:29 +0200749 */
Willy Tarreau0adb2812022-05-27 10:02:48 +0200750 sc_conn_shut(sc);
Willy Tarreau476c2802022-11-14 07:36:42 +0100751 __fallthrough;
Willy Tarreau026e8fb2022-05-17 19:47:17 +0200752 case SC_ST_CER:
753 case SC_ST_QUE:
754 case SC_ST_TAR:
Willy Tarreau0adb2812022-05-27 10:02:48 +0200755 sc->state = SC_ST_DIS;
Willy Tarreau476c2802022-11-14 07:36:42 +0100756 __fallthrough;
Christopher Faulet9ffddd52022-04-01 14:04:29 +0200757 default:
Willy Tarreau0adb2812022-05-27 10:02:48 +0200758 sc->flags &= ~SC_FL_NOLINGER;
Christopher Faulet0c370ee2023-04-13 16:05:13 +0200759 sc->flags |= SC_FL_ABRT_DONE;
Christopher Fauletca679922022-07-20 13:24:04 +0200760 if (sc->flags & SC_FL_ISBACK)
761 __sc_strm(sc)->conn_exp = TICK_ETERNITY;
Christopher Faulet9ffddd52022-04-01 14:04:29 +0200762 }
763}
764
Willy Tarreau3a3f4802022-05-17 18:28:19 +0200765/* This function is used for inter-stream connector calls. It is called by the
Christopher Faulet9ffddd52022-04-01 14:04:29 +0200766 * consumer to inform the producer side that it may be interested in checking
767 * for free space in the buffer. Note that it intentionally does not update
768 * timeouts, so that we can still check them later at wake-up. This function is
Willy Tarreau3a3f4802022-05-17 18:28:19 +0200769 * dedicated to connection-based stream connectors.
Christopher Faulet9ffddd52022-04-01 14:04:29 +0200770 */
Willy Tarreau0adb2812022-05-27 10:02:48 +0200771static void sc_app_chk_rcv_conn(struct stconn *sc)
Christopher Faulet9ffddd52022-04-01 14:04:29 +0200772{
Willy Tarreau0adb2812022-05-27 10:02:48 +0200773 BUG_ON(!sc_conn(sc));
Christopher Faulet9ffddd52022-04-01 14:04:29 +0200774
775 /* (re)start reading */
Willy Tarreau0adb2812022-05-27 10:02:48 +0200776 if (sc_state_in(sc->state, SC_SB_CON|SC_SB_RDY|SC_SB_EST))
777 tasklet_wakeup(sc->wait_event.tasklet);
Christopher Faulet9ffddd52022-04-01 14:04:29 +0200778}
779
780
Willy Tarreau3a3f4802022-05-17 18:28:19 +0200781/* This function is used for inter-stream connector calls. It is called by the
Christopher Faulet9ffddd52022-04-01 14:04:29 +0200782 * producer to inform the consumer side that it may be interested in checking
783 * for data in the buffer. Note that it intentionally does not update timeouts,
784 * so that we can still check them later at wake-up.
785 */
Willy Tarreau0adb2812022-05-27 10:02:48 +0200786static void sc_app_chk_snd_conn(struct stconn *sc)
Christopher Faulet9ffddd52022-04-01 14:04:29 +0200787{
Willy Tarreau0adb2812022-05-27 10:02:48 +0200788 struct channel *oc = sc_oc(sc);
Christopher Faulet9ffddd52022-04-01 14:04:29 +0200789
Willy Tarreau0adb2812022-05-27 10:02:48 +0200790 BUG_ON(!sc_conn(sc));
Christopher Faulet9ffddd52022-04-01 14:04:29 +0200791
Willy Tarreau0adb2812022-05-27 10:02:48 +0200792 if (unlikely(!sc_state_in(sc->state, SC_SB_RDY|SC_SB_EST) ||
Christopher Faulet208c7122023-04-13 16:16:15 +0200793 (sc->flags & SC_FL_SHUT_DONE)))
Christopher Faulet9ffddd52022-04-01 14:04:29 +0200794 return;
795
796 if (unlikely(channel_is_empty(oc))) /* called with nothing to send ! */
797 return;
798
799 if (!oc->pipe && /* spliced data wants to be forwarded ASAP */
Willy Tarreau0adb2812022-05-27 10:02:48 +0200800 !sc_ep_test(sc, SE_FL_WAIT_DATA)) /* not waiting for data */
Christopher Faulet9ffddd52022-04-01 14:04:29 +0200801 return;
802
Willy Tarreau0adb2812022-05-27 10:02:48 +0200803 if (!(sc->wait_event.events & SUB_RETRY_SEND) && !channel_is_empty(sc_oc(sc)))
804 sc_conn_send(sc);
Christopher Faulet9ffddd52022-04-01 14:04:29 +0200805
Willy Tarreau0adb2812022-05-27 10:02:48 +0200806 if (sc_ep_test(sc, SE_FL_ERROR | SE_FL_ERR_PENDING) || sc_is_conn_error(sc)) {
Christopher Faulet9ffddd52022-04-01 14:04:29 +0200807 /* Write error on the file descriptor */
Christopher Faulet56a2b602023-04-14 09:42:59 +0200808 BUG_ON(sc_ep_test(sc, SE_FL_EOS|SE_FL_ERROR|SE_FL_ERR_PENDING) == (SE_FL_EOS|SE_FL_ERR_PENDING));
Christopher Faulet9ffddd52022-04-01 14:04:29 +0200809 goto out_wakeup;
810 }
811
812 /* OK, so now we know that some data might have been sent, and that we may
813 * have to poll first. We have to do that too if the buffer is not empty.
814 */
815 if (channel_is_empty(oc)) {
816 /* the connection is established but we can't write. Either the
817 * buffer is empty, or we just refrain from sending because the
818 * ->o limit was reached. Maybe we just wrote the last
819 * chunk and need to close.
820 */
Christopher Faulet87633c32023-04-03 18:32:50 +0200821 if ((oc->flags & CF_AUTO_CLOSE) &&
Christopher Faulet208c7122023-04-13 16:16:15 +0200822 ((sc->flags & (SC_FL_SHUT_DONE|SC_FL_SHUT_WANTED)) == SC_FL_SHUT_WANTED) &&
Willy Tarreau0adb2812022-05-27 10:02:48 +0200823 sc_state_in(sc->state, SC_SB_RDY|SC_SB_EST)) {
Christopher Fauletb2b1c3a2023-04-13 16:23:48 +0200824 sc_shutdown(sc);
Christopher Faulet9ffddd52022-04-01 14:04:29 +0200825 goto out_wakeup;
826 }
827
Christopher Faulet208c7122023-04-13 16:16:15 +0200828 if ((sc->flags & (SC_FL_SHUT_DONE|SC_FL_SHUT_WANTED)) == 0)
Willy Tarreau0adb2812022-05-27 10:02:48 +0200829 sc_ep_set(sc, SE_FL_WAIT_DATA);
Christopher Faulet9ffddd52022-04-01 14:04:29 +0200830 }
831 else {
832 /* Otherwise there are remaining data to be sent in the buffer,
833 * which means we have to poll before doing so.
834 */
Willy Tarreau0adb2812022-05-27 10:02:48 +0200835 sc_ep_clr(sc, SE_FL_WAIT_DATA);
Christopher Faulet9ffddd52022-04-01 14:04:29 +0200836 }
837
Christopher Faulet9ffddd52022-04-01 14:04:29 +0200838 /* in case of special condition (error, shutdown, end of write...), we
839 * have to notify the task.
840 */
Christopher Faulet208c7122023-04-13 16:16:15 +0200841 if (likely((sc->flags & SC_FL_SHUT_DONE) ||
Christopher Faulet71c486b2023-02-09 14:14:38 +0100842 ((oc->flags & CF_WRITE_EVENT) && sc->state < SC_ST_EST) ||
843 ((oc->flags & CF_WAKE_WRITE) &&
844 ((channel_is_empty(oc) && !oc->to_forward) ||
845 !sc_state_in(sc->state, SC_SB_EST))))) {
Christopher Faulet9ffddd52022-04-01 14:04:29 +0200846 out_wakeup:
Willy Tarreau0adb2812022-05-27 10:02:48 +0200847 if (!(sc->flags & SC_FL_DONT_WAKE))
848 task_wakeup(sc_strm_task(sc), TASK_WOKEN_IO);
Christopher Faulet9ffddd52022-04-01 14:04:29 +0200849 }
850}
851
852/*
Willy Tarreau3a3f4802022-05-17 18:28:19 +0200853 * This function performs a shutdown-read on a stream connector attached to an
Christopher Faulet9ffddd52022-04-01 14:04:29 +0200854 * applet in a connected or init state (it does nothing for other states). It
855 * either shuts the read side or marks itself as closed. The buffer flags are
Willy Tarreaucb041662022-05-17 19:44:42 +0200856 * updated to reflect the new state. If the stream connector has SC_FL_NOHALF,
Christopher Faulet9ffddd52022-04-01 14:04:29 +0200857 * we also forward the close to the write side. The owner task is woken up if
858 * it exists.
859 */
Christopher Fauletcfc11c02023-04-13 16:10:23 +0200860static void sc_app_abort_applet(struct stconn *sc)
Christopher Faulet9ffddd52022-04-01 14:04:29 +0200861{
Willy Tarreau0adb2812022-05-27 10:02:48 +0200862 struct channel *ic = sc_ic(sc);
Christopher Faulet9ffddd52022-04-01 14:04:29 +0200863
Willy Tarreau0adb2812022-05-27 10:02:48 +0200864 BUG_ON(!sc_appctx(sc));
Christopher Faulet9ffddd52022-04-01 14:04:29 +0200865
Christopher Fauletca5309a2023-04-17 16:17:32 +0200866 if (sc->flags & (SC_FL_EOS|SC_FL_ABRT_DONE))
Christopher Faulet9ffddd52022-04-01 14:04:29 +0200867 return;
Christopher Faulet0c370ee2023-04-13 16:05:13 +0200868 sc->flags |= SC_FL_ABRT_DONE;
Christopher Faulet87633c32023-04-03 18:32:50 +0200869 ic->flags |= CF_READ_EVENT;
Christopher Faulet9ffddd52022-04-01 14:04:29 +0200870
Christopher Fauletcfc11c02023-04-13 16:10:23 +0200871 /* Note: on abort, we don't call the applet */
Christopher Faulet9ffddd52022-04-01 14:04:29 +0200872
Willy Tarreau0adb2812022-05-27 10:02:48 +0200873 if (!sc_state_in(sc->state, SC_SB_CON|SC_SB_RDY|SC_SB_EST))
Christopher Faulet9ffddd52022-04-01 14:04:29 +0200874 return;
875
Christopher Faulet208c7122023-04-13 16:16:15 +0200876 if (sc->flags & SC_FL_SHUT_DONE) {
Willy Tarreau0adb2812022-05-27 10:02:48 +0200877 appctx_shut(__sc_appctx(sc));
878 sc->state = SC_ST_DIS;
Christopher Fauletca679922022-07-20 13:24:04 +0200879 if (sc->flags & SC_FL_ISBACK)
880 __sc_strm(sc)->conn_exp = TICK_ETERNITY;
Christopher Faulet9ffddd52022-04-01 14:04:29 +0200881 }
Christopher Fauletb2b1c3a2023-04-13 16:23:48 +0200882 else if (sc_cond_forward_shut(sc))
883 return sc_app_shut_applet(sc);
Christopher Faulet9ffddd52022-04-01 14:04:29 +0200884}
885
886/*
Willy Tarreau3a3f4802022-05-17 18:28:19 +0200887 * This function performs a shutdown-write on a stream connector attached to an
Christopher Faulet9ffddd52022-04-01 14:04:29 +0200888 * applet in a connected or init state (it does nothing for other states). It
889 * either shuts the write side or marks itself as closed. The buffer flags are
890 * updated to reflect the new state. It does also close everything if the SI
891 * was marked as being in error state. The owner task is woken up if it exists.
892 */
Christopher Fauletb2b1c3a2023-04-13 16:23:48 +0200893static void sc_app_shut_applet(struct stconn *sc)
Christopher Faulet9ffddd52022-04-01 14:04:29 +0200894{
Willy Tarreau0adb2812022-05-27 10:02:48 +0200895 struct channel *ic = sc_ic(sc);
896 struct channel *oc = sc_oc(sc);
Christopher Faulet9ffddd52022-04-01 14:04:29 +0200897
Willy Tarreau0adb2812022-05-27 10:02:48 +0200898 BUG_ON(!sc_appctx(sc));
Christopher Faulet9ffddd52022-04-01 14:04:29 +0200899
Christopher Faulete38534c2023-04-13 15:45:24 +0200900 sc->flags &= ~SC_FL_SHUT_WANTED;
Christopher Faulet208c7122023-04-13 16:16:15 +0200901 if (sc->flags & SC_FL_SHUT_DONE)
Christopher Faulet9ffddd52022-04-01 14:04:29 +0200902 return;
Christopher Faulet208c7122023-04-13 16:16:15 +0200903 sc->flags |= SC_FL_SHUT_DONE;
Christopher Faulet87633c32023-04-03 18:32:50 +0200904 oc->flags |= CF_WRITE_EVENT;
Christopher Fauletbcdcfad2023-02-20 08:36:53 +0100905 sc_set_hcto(sc);
Christopher Faulet9ffddd52022-04-01 14:04:29 +0200906
907 /* on shutw we always wake the applet up */
Willy Tarreau0adb2812022-05-27 10:02:48 +0200908 appctx_wakeup(__sc_appctx(sc));
Christopher Faulet9ffddd52022-04-01 14:04:29 +0200909
Willy Tarreau0adb2812022-05-27 10:02:48 +0200910 switch (sc->state) {
Willy Tarreau026e8fb2022-05-17 19:47:17 +0200911 case SC_ST_RDY:
912 case SC_ST_EST:
Christopher Faulet9ffddd52022-04-01 14:04:29 +0200913 /* we have to shut before closing, otherwise some short messages
914 * may never leave the system, especially when there are remaining
915 * unread data in the socket input buffer, or when nolinger is set.
Willy Tarreaucb041662022-05-17 19:44:42 +0200916 * However, if SC_FL_NOLINGER is explicitly set, we know there is
Christopher Faulet9ffddd52022-04-01 14:04:29 +0200917 * no risk so we close both sides immediately.
918 */
Christopher Fauletca5309a2023-04-17 16:17:32 +0200919 if (!(sc->flags & (SC_FL_ERROR|SC_FL_NOLINGER|SC_FL_EOS|SC_FL_ABRT_DONE)) &&
Christopher Faulet87633c32023-04-03 18:32:50 +0200920 !(ic->flags & CF_DONT_READ))
Christopher Faulet9ffddd52022-04-01 14:04:29 +0200921 return;
922
Willy Tarreau476c2802022-11-14 07:36:42 +0100923 __fallthrough;
Willy Tarreau026e8fb2022-05-17 19:47:17 +0200924 case SC_ST_CON:
925 case SC_ST_CER:
926 case SC_ST_QUE:
927 case SC_ST_TAR:
Christopher Faulet9ffddd52022-04-01 14:04:29 +0200928 /* Note that none of these states may happen with applets */
Willy Tarreau0adb2812022-05-27 10:02:48 +0200929 appctx_shut(__sc_appctx(sc));
930 sc->state = SC_ST_DIS;
Willy Tarreau476c2802022-11-14 07:36:42 +0100931 __fallthrough;
Christopher Faulet9ffddd52022-04-01 14:04:29 +0200932 default:
Willy Tarreau0adb2812022-05-27 10:02:48 +0200933 sc->flags &= ~SC_FL_NOLINGER;
Christopher Faulet0c370ee2023-04-13 16:05:13 +0200934 sc->flags |= SC_FL_ABRT_DONE;
Christopher Fauletca679922022-07-20 13:24:04 +0200935 if (sc->flags & SC_FL_ISBACK)
936 __sc_strm(sc)->conn_exp = TICK_ETERNITY;
Christopher Faulet9ffddd52022-04-01 14:04:29 +0200937 }
938}
939
940/* chk_rcv function for applets */
Willy Tarreau0adb2812022-05-27 10:02:48 +0200941static void sc_app_chk_rcv_applet(struct stconn *sc)
Christopher Faulet9ffddd52022-04-01 14:04:29 +0200942{
Willy Tarreau0adb2812022-05-27 10:02:48 +0200943 struct channel *ic = sc_ic(sc);
Christopher Faulet9ffddd52022-04-01 14:04:29 +0200944
Willy Tarreau0adb2812022-05-27 10:02:48 +0200945 BUG_ON(!sc_appctx(sc));
Christopher Faulet9ffddd52022-04-01 14:04:29 +0200946
Christopher Faulet9ffddd52022-04-01 14:04:29 +0200947 if (!ic->pipe) {
948 /* (re)start reading */
Willy Tarreau0adb2812022-05-27 10:02:48 +0200949 appctx_wakeup(__sc_appctx(sc));
Christopher Faulet9ffddd52022-04-01 14:04:29 +0200950 }
951}
952
953/* chk_snd function for applets */
Willy Tarreau0adb2812022-05-27 10:02:48 +0200954static void sc_app_chk_snd_applet(struct stconn *sc)
Christopher Faulet9ffddd52022-04-01 14:04:29 +0200955{
Willy Tarreau0adb2812022-05-27 10:02:48 +0200956 struct channel *oc = sc_oc(sc);
Christopher Faulet9ffddd52022-04-01 14:04:29 +0200957
Willy Tarreau0adb2812022-05-27 10:02:48 +0200958 BUG_ON(!sc_appctx(sc));
Christopher Faulet9ffddd52022-04-01 14:04:29 +0200959
Christopher Faulet208c7122023-04-13 16:16:15 +0200960 if (unlikely(sc->state != SC_ST_EST || (sc->flags & SC_FL_SHUT_DONE)))
Christopher Faulet9ffddd52022-04-01 14:04:29 +0200961 return;
962
Christopher Fauletdf8c1492024-02-12 18:30:33 +0100963 /* we only wake the applet up if it was waiting for some data and is ready to consume it */
964 if (!sc_ep_test(sc, SE_FL_WAIT_DATA|SE_FL_WONT_CONSUME))
Christopher Faulet9ffddd52022-04-01 14:04:29 +0200965 return;
966
Christopher Faulet9ffddd52022-04-01 14:04:29 +0200967 if (!channel_is_empty(oc)) {
968 /* (re)start sending */
Willy Tarreau0adb2812022-05-27 10:02:48 +0200969 appctx_wakeup(__sc_appctx(sc));
Christopher Faulet9ffddd52022-04-01 14:04:29 +0200970 }
971}
Christopher Faulet13045f02022-04-01 14:23:38 +0200972
973
974/* This function is designed to be called from within the stream handler to
Willy Tarreau4596fe22022-05-17 19:07:51 +0200975 * update the input channel's expiration timer and the stream connector's
Christopher Faulet13045f02022-04-01 14:23:38 +0200976 * Rx flags based on the channel's flags. It needs to be called only once
977 * after the channel's flags have settled down, and before they are cleared,
978 * though it doesn't harm to call it as often as desired (it just slightly
979 * hurts performance). It must not be called from outside of the stream
980 * handler, as what it does will be used to compute the stream task's
981 * expiration.
982 */
Willy Tarreau0adb2812022-05-27 10:02:48 +0200983void sc_update_rx(struct stconn *sc)
Christopher Faulet13045f02022-04-01 14:23:38 +0200984{
Willy Tarreau0adb2812022-05-27 10:02:48 +0200985 struct channel *ic = sc_ic(sc);
Christopher Faulet13045f02022-04-01 14:23:38 +0200986
Christopher Fauletca5309a2023-04-17 16:17:32 +0200987 if (sc->flags & (SC_FL_EOS|SC_FL_ABRT_DONE))
Christopher Faulet13045f02022-04-01 14:23:38 +0200988 return;
Christopher Faulet13045f02022-04-01 14:23:38 +0200989
Christopher Fauletfab82bf2023-05-05 11:30:16 +0200990 /* Unblock the SC if it needs room and the free space is large enough (0
991 * means it can always be unblocked). Do not unblock it if -1 was
992 * specified.
993 */
994 if (!sc->room_needed || (sc->room_needed > 0 && channel_recv_max(ic) >= sc->room_needed))
995 sc_have_room(sc);
996
Christopher Faulet13045f02022-04-01 14:23:38 +0200997 /* Read not closed, update FD status and timeout for reads */
998 if (ic->flags & CF_DONT_READ)
Willy Tarreau0adb2812022-05-27 10:02:48 +0200999 sc_wont_read(sc);
Christopher Faulet13045f02022-04-01 14:23:38 +02001000 else
Willy Tarreau0adb2812022-05-27 10:02:48 +02001001 sc_will_read(sc);
Christopher Faulet13045f02022-04-01 14:23:38 +02001002
Willy Tarreau0adb2812022-05-27 10:02:48 +02001003 sc_chk_rcv(sc);
Christopher Faulet13045f02022-04-01 14:23:38 +02001004}
1005
1006/* This function is designed to be called from within the stream handler to
Willy Tarreau4596fe22022-05-17 19:07:51 +02001007 * update the output channel's expiration timer and the stream connector's
Christopher Faulet13045f02022-04-01 14:23:38 +02001008 * Tx flags based on the channel's flags. It needs to be called only once
1009 * after the channel's flags have settled down, and before they are cleared,
1010 * though it doesn't harm to call it as often as desired (it just slightly
1011 * hurts performance). It must not be called from outside of the stream
1012 * handler, as what it does will be used to compute the stream task's
1013 * expiration.
1014 */
Willy Tarreau0adb2812022-05-27 10:02:48 +02001015void sc_update_tx(struct stconn *sc)
Christopher Faulet13045f02022-04-01 14:23:38 +02001016{
Willy Tarreau0adb2812022-05-27 10:02:48 +02001017 struct channel *oc = sc_oc(sc);
Christopher Faulet13045f02022-04-01 14:23:38 +02001018
Christopher Faulet208c7122023-04-13 16:16:15 +02001019 if (sc->flags & SC_FL_SHUT_DONE)
Christopher Faulet13045f02022-04-01 14:23:38 +02001020 return;
1021
1022 /* Write not closed, update FD status and timeout for writes */
1023 if (channel_is_empty(oc)) {
1024 /* stop writing */
Willy Tarreau0adb2812022-05-27 10:02:48 +02001025 if (!sc_ep_test(sc, SE_FL_WAIT_DATA)) {
Christopher Faulete38534c2023-04-13 15:45:24 +02001026 if ((sc->flags & SC_FL_SHUT_WANTED) == 0)
Willy Tarreau0adb2812022-05-27 10:02:48 +02001027 sc_ep_set(sc, SE_FL_WAIT_DATA);
Christopher Faulet13045f02022-04-01 14:23:38 +02001028 }
1029 return;
1030 }
1031
Christopher Faulet15315d62023-02-20 08:23:51 +01001032 /* (re)start writing */
Willy Tarreau0adb2812022-05-27 10:02:48 +02001033 sc_ep_clr(sc, SE_FL_WAIT_DATA);
Christopher Faulet5e29b762022-04-04 08:58:34 +02001034}
1035
Willy Tarreau19c65a92022-05-27 08:49:24 +02001036/* This function is the equivalent to sc_update() except that it's
Christopher Faulet5e29b762022-04-04 08:58:34 +02001037 * designed to be called from outside the stream handlers, typically the lower
1038 * layers (applets, connections) after I/O completion. After updating the stream
1039 * interface and timeouts, it will try to forward what can be forwarded, then to
1040 * wake the associated task up if an important event requires special handling.
Willy Tarreau15252cd2022-05-25 16:36:21 +02001041 * It may update SE_FL_WAIT_DATA and/or SC_FL_NEED_ROOM, that the callers are
Christopher Faulet5e29b762022-04-04 08:58:34 +02001042 * encouraged to watch to take appropriate action.
Willy Tarreau19c65a92022-05-27 08:49:24 +02001043 * It should not be called from within the stream itself, sc_update()
Christopher Faulet5e29b762022-04-04 08:58:34 +02001044 * is designed for this.
1045 */
Willy Tarreau0adb2812022-05-27 10:02:48 +02001046static void sc_notify(struct stconn *sc)
Christopher Faulet5e29b762022-04-04 08:58:34 +02001047{
Willy Tarreau0adb2812022-05-27 10:02:48 +02001048 struct channel *ic = sc_ic(sc);
1049 struct channel *oc = sc_oc(sc);
Willy Tarreaue68bc612022-05-27 11:23:05 +02001050 struct stconn *sco = sc_opposite(sc);
Willy Tarreau0adb2812022-05-27 10:02:48 +02001051 struct task *task = sc_strm_task(sc);
Christopher Faulet5e29b762022-04-04 08:58:34 +02001052
1053 /* process consumer side */
1054 if (channel_is_empty(oc)) {
Willy Tarreau0adb2812022-05-27 10:02:48 +02001055 struct connection *conn = sc_conn(sc);
Christopher Faulet5e29b762022-04-04 08:58:34 +02001056
Christopher Faulet208c7122023-04-13 16:16:15 +02001057 if (((sc->flags & (SC_FL_SHUT_DONE|SC_FL_SHUT_WANTED)) == SC_FL_SHUT_WANTED) &&
Willy Tarreau0adb2812022-05-27 10:02:48 +02001058 (sc->state == SC_ST_EST) && (!conn || !(conn->flags & (CO_FL_WAIT_XPRT | CO_FL_EARLY_SSL_HS))))
Christopher Fauletb2b1c3a2023-04-13 16:23:48 +02001059 sc_shutdown(sc);
Christopher Faulet5e29b762022-04-04 08:58:34 +02001060 }
1061
1062 /* indicate that we may be waiting for data from the output channel or
Christopher Faulete38534c2023-04-13 15:45:24 +02001063 * we're about to close and can't expect more data if SC_FL_SHUT_WANTED is there.
Christopher Faulet5e29b762022-04-04 08:58:34 +02001064 */
Christopher Faulet208c7122023-04-13 16:16:15 +02001065 if (!(sc->flags & (SC_FL_SHUT_DONE|SC_FL_SHUT_WANTED)))
Willy Tarreau0adb2812022-05-27 10:02:48 +02001066 sc_ep_set(sc, SE_FL_WAIT_DATA);
Christopher Faulet208c7122023-04-13 16:16:15 +02001067 else if ((sc->flags & (SC_FL_SHUT_DONE|SC_FL_SHUT_WANTED)) == SC_FL_SHUT_WANTED)
Willy Tarreau0adb2812022-05-27 10:02:48 +02001068 sc_ep_clr(sc, SE_FL_WAIT_DATA);
Christopher Faulet5e29b762022-04-04 08:58:34 +02001069
Christopher Faulet5e29b762022-04-04 08:58:34 +02001070 if (oc->flags & CF_DONT_READ)
Willy Tarreaue68bc612022-05-27 11:23:05 +02001071 sc_wont_read(sco);
Christopher Faulet5e29b762022-04-04 08:58:34 +02001072 else
Willy Tarreaue68bc612022-05-27 11:23:05 +02001073 sc_will_read(sco);
Christopher Faulet5e29b762022-04-04 08:58:34 +02001074
1075 /* Notify the other side when we've injected data into the IC that
1076 * needs to be forwarded. We can do fast-forwarding as soon as there
1077 * are output data, but we avoid doing this if some of the data are
1078 * not yet scheduled for being forwarded, because it is very likely
1079 * that it will be done again immediately afterwards once the following
Willy Tarreau15252cd2022-05-25 16:36:21 +02001080 * data are parsed (eg: HTTP chunking). We only clear SC_FL_NEED_ROOM
1081 * once we've emptied *some* of the output buffer, and not just when
1082 * there is available room, because applets are often forced to stop
1083 * before the buffer is full. We must not stop based on input data
1084 * alone because an HTTP parser might need more data to complete the
1085 * parsing.
Christopher Faulet5e29b762022-04-04 08:58:34 +02001086 */
1087 if (!channel_is_empty(ic) &&
Willy Tarreaue68bc612022-05-27 11:23:05 +02001088 sc_ep_test(sco, SE_FL_WAIT_DATA) &&
Christopher Faulet439a41b2024-04-15 19:09:01 +02001089 (!HAS_DATA_FILTERS(__sc_strm(sc), ic) || channel_input_data(ic) == 0) &&
Christopher Fauletd1142082023-11-17 11:59:33 +01001090 (!(sc->flags & SC_FL_SND_EXP_MORE) || channel_full(ic, co_data(ic)) || channel_input_data(ic) == 0)) {
Christopher Faulet5e29b762022-04-04 08:58:34 +02001091 int new_len, last_len;
1092
1093 last_len = co_data(ic);
1094 if (ic->pipe)
1095 last_len += ic->pipe->data;
1096
Willy Tarreaue68bc612022-05-27 11:23:05 +02001097 sc_chk_snd(sco);
Christopher Faulet5e29b762022-04-04 08:58:34 +02001098
1099 new_len = co_data(ic);
1100 if (ic->pipe)
1101 new_len += ic->pipe->data;
1102
1103 /* check if the consumer has freed some space either in the
1104 * buffer or in the pipe.
1105 */
Christopher Faulet18b33092023-05-05 11:40:07 +02001106 if (!sc->room_needed || (new_len < last_len && (sc->room_needed < 0 || channel_recv_max(ic) >= sc->room_needed)))
Willy Tarreau0adb2812022-05-27 10:02:48 +02001107 sc_have_room(sc);
Christopher Faulet5e29b762022-04-04 08:58:34 +02001108 }
1109
1110 if (!(ic->flags & CF_DONT_READ))
Willy Tarreau0adb2812022-05-27 10:02:48 +02001111 sc_will_read(sc);
Christopher Faulet5e29b762022-04-04 08:58:34 +02001112
Willy Tarreau0adb2812022-05-27 10:02:48 +02001113 sc_chk_rcv(sc);
Willy Tarreaue68bc612022-05-27 11:23:05 +02001114 sc_chk_rcv(sco);
Christopher Faulet5e29b762022-04-04 08:58:34 +02001115
Christopher Faulet5e29b762022-04-04 08:58:34 +02001116 /* wake the task up only when needed */
Christopher Faulet285f7612022-12-12 08:28:55 +01001117 if (/* changes on the production side that must be handled:
Christopher Fauletad46e522023-04-14 11:59:15 +02001118 * - An error on receipt: SC_FL_ERROR
Christopher Fauletca5309a2023-04-17 16:17:32 +02001119 * - A read event: shutdown for reads (CF_READ_EVENT + EOS/ABRT_DONE)
Christopher Faulet904763f2023-03-22 14:53:11 +01001120 * end of input (CF_READ_EVENT + SC_FL_EOI)
Christopher Faulet285f7612022-12-12 08:28:55 +01001121 * data received and no fast-forwarding (CF_READ_EVENT + !to_forward)
1122 * read event while consumer side is not established (CF_READ_EVENT + sco->state != SC_ST_EST)
1123 */
Christopher Fauletca5309a2023-04-17 16:17:32 +02001124 ((ic->flags & CF_READ_EVENT) && ((sc->flags & SC_FL_EOI) || (sc->flags & (SC_FL_EOS|SC_FL_ABRT_DONE)) || !ic->to_forward || sco->state != SC_ST_EST)) ||
Christopher Faulet25d9fe52023-04-14 12:09:35 +02001125 (sc->flags & SC_FL_ERROR) ||
Christopher Faulet5e29b762022-04-04 08:58:34 +02001126
1127 /* changes on the consumption side */
Christopher Faulet2e56a732023-01-26 16:18:09 +01001128 sc_ep_test(sc, SE_FL_ERR_PENDING) ||
Christopher Fauletd8988412022-12-20 18:10:04 +01001129 ((oc->flags & CF_WRITE_EVENT) &&
1130 ((sc->state < SC_ST_EST) ||
Christopher Faulet208c7122023-04-13 16:16:15 +02001131 (sc->flags & SC_FL_SHUT_DONE) ||
Christopher Faulet5e29b762022-04-04 08:58:34 +02001132 (((oc->flags & CF_WAKE_WRITE) ||
Christopher Faulet87633c32023-04-03 18:32:50 +02001133 (!(oc->flags & CF_AUTO_CLOSE) &&
Christopher Faulet208c7122023-04-13 16:16:15 +02001134 !(sc->flags & (SC_FL_SHUT_WANTED|SC_FL_SHUT_DONE)))) &&
Christopher Faulet87633c32023-04-03 18:32:50 +02001135 (sco->state != SC_ST_EST ||
1136 (channel_is_empty(oc) && !oc->to_forward)))))) {
Christopher Faulet5e29b762022-04-04 08:58:34 +02001137 task_wakeup(task, TASK_WOKEN_IO);
1138 }
Christopher Faulet13a8a592023-09-06 09:09:10 +02001139 else {
Christopher Fauleta93d8d22023-11-09 09:54:51 +01001140 /* Update expiration date for the task and requeue it if not already expired */
1141 if (!tick_is_expired(task->expire, now_ms)) {
1142 task->expire = tick_first(task->expire, sc_ep_rcv_ex(sc));
1143 task->expire = tick_first(task->expire, sc_ep_snd_ex(sc));
1144 task->expire = tick_first(task->expire, sc_ep_rcv_ex(sco));
1145 task->expire = tick_first(task->expire, sc_ep_snd_ex(sco));
1146 task->expire = tick_first(task->expire, ic->analyse_exp);
1147 task->expire = tick_first(task->expire, oc->analyse_exp);
1148 task->expire = tick_first(task->expire, __sc_strm(sc)->conn_exp);
Christopher Faulet13a8a592023-09-06 09:09:10 +02001149
Christopher Fauleta93d8d22023-11-09 09:54:51 +01001150 task_queue(task);
1151 }
Christopher Faulet13a8a592023-09-06 09:09:10 +02001152 }
1153
Christopher Faulet2e56a732023-01-26 16:18:09 +01001154 if (ic->flags & CF_READ_EVENT)
Christopher Faulet9a790f62023-03-16 14:40:03 +01001155 sc->flags &= ~SC_FL_RCV_ONCE;
Christopher Faulet5e29b762022-04-04 08:58:34 +02001156}
1157
1158/*
Christopher Faulet1aec6c92023-04-17 17:29:29 +02001159 * This function propagates an end-of-stream received on a socket-based connection.
Willy Tarreaucb041662022-05-17 19:44:42 +02001160 * It updates the stream connector. If the stream connector has SC_FL_NOHALF,
Christopher Faulet5e29b762022-04-04 08:58:34 +02001161 * the close is also forwarded to the write side as an abort.
1162 */
Christopher Faulet1aec6c92023-04-17 17:29:29 +02001163static void sc_conn_eos(struct stconn *sc)
Christopher Faulet5e29b762022-04-04 08:58:34 +02001164{
Willy Tarreau0adb2812022-05-27 10:02:48 +02001165 struct channel *ic = sc_ic(sc);
Christopher Faulet5e29b762022-04-04 08:58:34 +02001166
Willy Tarreau0adb2812022-05-27 10:02:48 +02001167 BUG_ON(!sc_conn(sc));
Christopher Faulet5e29b762022-04-04 08:58:34 +02001168
Christopher Fauletca5309a2023-04-17 16:17:32 +02001169 if (sc->flags & (SC_FL_EOS|SC_FL_ABRT_DONE))
Christopher Faulet5e29b762022-04-04 08:58:34 +02001170 return;
Christopher Faulet1aec6c92023-04-17 17:29:29 +02001171 sc->flags |= SC_FL_EOS;
Christopher Faulet87633c32023-04-03 18:32:50 +02001172 ic->flags |= CF_READ_EVENT;
Christopher Faulet4c135682023-02-16 11:09:31 +01001173 sc_ep_report_read_activity(sc);
Christopher Faulet5e29b762022-04-04 08:58:34 +02001174
Willy Tarreau0adb2812022-05-27 10:02:48 +02001175 if (!sc_state_in(sc->state, SC_SB_CON|SC_SB_RDY|SC_SB_EST))
Christopher Faulet5e29b762022-04-04 08:58:34 +02001176 return;
1177
Christopher Faulet208c7122023-04-13 16:16:15 +02001178 if (sc->flags & SC_FL_SHUT_DONE)
Christopher Faulet5e29b762022-04-04 08:58:34 +02001179 goto do_close;
1180
Christopher Fauletb2b1c3a2023-04-13 16:23:48 +02001181 if (sc_cond_forward_shut(sc)) {
Christopher Faulet5e29b762022-04-04 08:58:34 +02001182 /* we want to immediately forward this close to the write side */
1183 /* force flag on ssl to keep stream in cache */
Willy Tarreau0adb2812022-05-27 10:02:48 +02001184 sc_conn_shutw(sc, CO_SHW_SILENT);
Christopher Faulet5e29b762022-04-04 08:58:34 +02001185 goto do_close;
1186 }
1187
1188 /* otherwise that's just a normal read shutdown */
1189 return;
1190
1191 do_close:
Willy Tarreauf61dd192022-05-27 09:00:19 +02001192 /* OK we completely close the socket here just as if we went through sc_shut[rw]() */
Willy Tarreau0adb2812022-05-27 10:02:48 +02001193 sc_conn_shut(sc);
Christopher Faulet5e29b762022-04-04 08:58:34 +02001194
Christopher Faulete38534c2023-04-13 15:45:24 +02001195 sc->flags &= ~SC_FL_SHUT_WANTED;
Christopher Faulet208c7122023-04-13 16:16:15 +02001196 sc->flags |= SC_FL_SHUT_DONE;
Christopher Faulet5e29b762022-04-04 08:58:34 +02001197
Willy Tarreau0adb2812022-05-27 10:02:48 +02001198 sc->state = SC_ST_DIS;
Christopher Fauletca679922022-07-20 13:24:04 +02001199 if (sc->flags & SC_FL_ISBACK)
1200 __sc_strm(sc)->conn_exp = TICK_ETERNITY;
Christopher Faulet5e29b762022-04-04 08:58:34 +02001201 return;
1202}
1203
1204/*
1205 * This is the callback which is called by the connection layer to receive data
1206 * into the buffer from the connection. It iterates over the mux layer's
1207 * rcv_buf function.
1208 */
Willy Tarreau0adb2812022-05-27 10:02:48 +02001209static int sc_conn_recv(struct stconn *sc)
Christopher Faulet5e29b762022-04-04 08:58:34 +02001210{
Willy Tarreau0adb2812022-05-27 10:02:48 +02001211 struct connection *conn = __sc_conn(sc);
1212 struct channel *ic = sc_ic(sc);
Christopher Faulet5e29b762022-04-04 08:58:34 +02001213 int ret, max, cur_read = 0;
1214 int read_poll = MAX_READ_POLL_LOOPS;
1215 int flags = 0;
1216
1217 /* If not established yet, do nothing. */
Willy Tarreau0adb2812022-05-27 10:02:48 +02001218 if (sc->state != SC_ST_EST)
Christopher Faulet5e29b762022-04-04 08:58:34 +02001219 return 0;
1220
Willy Tarreau462b9892022-05-18 18:06:53 +02001221 /* If another call to sc_conn_recv() failed, and we subscribed to
Christopher Faulet5e29b762022-04-04 08:58:34 +02001222 * recv events already, give up now.
1223 */
Christopher Faulet95125882023-04-12 18:35:18 +02001224 if ((sc->wait_event.events & SUB_RETRY_RECV) || sc_waiting_room(sc))
Christopher Faulet5e29b762022-04-04 08:58:34 +02001225 return 0;
1226
Christopher Fauletcfc11c02023-04-13 16:10:23 +02001227 /* maybe we were called immediately after an asynchronous abort */
Christopher Fauletca5309a2023-04-17 16:17:32 +02001228 if (sc->flags & (SC_FL_EOS|SC_FL_ABRT_DONE))
Christopher Faulet5e29b762022-04-04 08:58:34 +02001229 return 1;
1230
1231 /* we must wait because the mux is not installed yet */
1232 if (!conn->mux)
1233 return 0;
1234
Christopher Faulet5e29b762022-04-04 08:58:34 +02001235 /* stop immediately on errors. Note that we DON'T want to stop on
1236 * POLL_ERR, as the poller might report a write error while there
1237 * are still data available in the recv buffer. This typically
1238 * happens when we send too large a request to a backend server
1239 * which rejects it before reading it all.
1240 */
Willy Tarreau0adb2812022-05-27 10:02:48 +02001241 if (!sc_ep_test(sc, SE_FL_RCV_MORE)) {
Christopher Faulet5e29b762022-04-04 08:58:34 +02001242 if (!conn_xprt_ready(conn))
1243 return 0;
Willy Tarreau0adb2812022-05-27 10:02:48 +02001244 if (sc_ep_test(sc, SE_FL_ERROR))
Christopher Faulet5e29b762022-04-04 08:58:34 +02001245 goto end_recv;
1246 }
1247
1248 /* prepare to detect if the mux needs more room */
Willy Tarreau0adb2812022-05-27 10:02:48 +02001249 sc_ep_clr(sc, SE_FL_WANT_ROOM);
Christopher Faulet5e29b762022-04-04 08:58:34 +02001250
1251 if ((ic->flags & (CF_STREAMER | CF_STREAMER_FAST)) && !co_data(ic) &&
1252 global.tune.idle_timer &&
1253 (unsigned short)(now_ms - ic->last_read) >= global.tune.idle_timer) {
1254 /* The buffer was empty and nothing was transferred for more
1255 * than one second. This was caused by a pause and not by
1256 * congestion. Reset any streaming mode to reduce latency.
1257 */
1258 ic->xfer_small = 0;
1259 ic->xfer_large = 0;
1260 ic->flags &= ~(CF_STREAMER | CF_STREAMER_FAST);
1261 }
1262
1263 /* First, let's see if we may splice data across the channel without
1264 * using a buffer.
1265 */
Willy Tarreau0adb2812022-05-27 10:02:48 +02001266 if (sc_ep_test(sc, SE_FL_MAY_SPLICE) &&
Christopher Faulet5e29b762022-04-04 08:58:34 +02001267 (ic->pipe || ic->to_forward >= MIN_SPLICE_FORWARD) &&
1268 ic->flags & CF_KERN_SPLICING) {
Christopher Fauletd1142082023-11-17 11:59:33 +01001269 if (channel_data(ic)) {
Christopher Faulet5e29b762022-04-04 08:58:34 +02001270 /* We're embarrassed, there are already data pending in
1271 * the buffer and we don't want to have them at two
1272 * locations at a time. Let's indicate we need some
1273 * place and ask the consumer to hurry.
1274 */
1275 flags |= CO_RFL_BUF_FLUSH;
1276 goto abort_splice;
1277 }
1278
1279 if (unlikely(ic->pipe == NULL)) {
1280 if (pipes_used >= global.maxpipes || !(ic->pipe = get_pipe())) {
1281 ic->flags &= ~CF_KERN_SPLICING;
1282 goto abort_splice;
1283 }
1284 }
1285
Willy Tarreau0adb2812022-05-27 10:02:48 +02001286 ret = conn->mux->rcv_pipe(sc, ic->pipe, ic->to_forward);
Christopher Faulet5e29b762022-04-04 08:58:34 +02001287 if (ret < 0) {
1288 /* splice not supported on this end, let's disable it */
1289 ic->flags &= ~CF_KERN_SPLICING;
1290 goto abort_splice;
1291 }
1292
1293 if (ret > 0) {
1294 if (ic->to_forward != CHN_INFINITE_FORWARD)
1295 ic->to_forward -= ret;
1296 ic->total += ret;
1297 cur_read += ret;
Christopher Faulet285f7612022-12-12 08:28:55 +01001298 ic->flags |= CF_READ_EVENT;
Christopher Faulet5e29b762022-04-04 08:58:34 +02001299 }
1300
Willy Tarreau0adb2812022-05-27 10:02:48 +02001301 if (sc_ep_test(sc, SE_FL_EOS | SE_FL_ERROR))
Christopher Faulet5e29b762022-04-04 08:58:34 +02001302 goto end_recv;
1303
1304 if (conn->flags & CO_FL_WAIT_ROOM) {
1305 /* the pipe is full or we have read enough data that it
1306 * could soon be full. Let's stop before needing to poll.
1307 */
Christopher Faulet9b890672024-07-30 09:28:23 +02001308 sc_need_room(sc, -1);
Christopher Faulet5e29b762022-04-04 08:58:34 +02001309 goto done_recv;
1310 }
1311
1312 /* splice not possible (anymore), let's go on on standard copy */
1313 }
1314
1315 abort_splice:
1316 if (ic->pipe && unlikely(!ic->pipe->data)) {
1317 put_pipe(ic->pipe);
1318 ic->pipe = NULL;
1319 }
1320
Willy Tarreau0adb2812022-05-27 10:02:48 +02001321 if (ic->pipe && ic->to_forward && !(flags & CO_RFL_BUF_FLUSH) && sc_ep_test(sc, SE_FL_MAY_SPLICE)) {
Christopher Faulet5e29b762022-04-04 08:58:34 +02001322 /* don't break splicing by reading, but still call rcv_buf()
1323 * to pass the flag.
1324 */
1325 goto done_recv;
1326 }
1327
1328 /* now we'll need a input buffer for the stream */
Willy Tarreau0adb2812022-05-27 10:02:48 +02001329 if (!sc_alloc_ibuf(sc, &(__sc_strm(sc)->buffer_wait)))
Christopher Faulet5e29b762022-04-04 08:58:34 +02001330 goto end_recv;
1331
1332 /* For an HTX stream, if the buffer is stuck (no output data with some
1333 * input data) and if the HTX message is fragmented or if its free space
1334 * wraps, we force an HTX deframentation. It is a way to have a
1335 * contiguous free space nad to let the mux to copy as much data as
1336 * possible.
1337 *
1338 * NOTE: A possible optim may be to let the mux decides if defrag is
1339 * required or not, depending on amount of data to be xferred.
1340 */
Willy Tarreau0adb2812022-05-27 10:02:48 +02001341 if (IS_HTX_STRM(__sc_strm(sc)) && !co_data(ic)) {
Christopher Faulet5e29b762022-04-04 08:58:34 +02001342 struct htx *htx = htxbuf(&ic->buf);
1343
1344 if (htx_is_not_empty(htx) && ((htx->flags & HTX_FL_FRAGMENTED) || htx_space_wraps(htx)))
1345 htx_defrag(htx, NULL, 0);
1346 }
1347
1348 /* Instruct the mux it must subscribed for read events */
Christopher Fauletd24b51a2023-11-13 19:16:09 +01001349 if (!conn_is_back(conn) && /* for frontend conns only */
1350 (sc_opposite(sc)->state != SC_ST_INI) && /* before backend connection setup */
1351 (__sc_strm(sc)->be->options & PR_O_ABRT_CLOSE)) /* if abortonclose option is set for the current backend */
1352 flags |= CO_RFL_KEEP_RECV;
Christopher Faulet5e29b762022-04-04 08:58:34 +02001353
1354 /* Important note : if we're called with POLL_IN|POLL_HUP, it means the read polling
1355 * was enabled, which implies that the recv buffer was not full. So we have a guarantee
1356 * that if such an event is not handled above in splice, it will be handled here by
1357 * recv().
1358 */
Willy Tarreau0adb2812022-05-27 10:02:48 +02001359 while (sc_ep_test(sc, SE_FL_RCV_MORE) ||
Christopher Faulet5e29b762022-04-04 08:58:34 +02001360 (!(conn->flags & CO_FL_HANDSHAKE) &&
Christopher Fauletca5309a2023-04-17 16:17:32 +02001361 (!sc_ep_test(sc, SE_FL_ERROR | SE_FL_EOS)) && !(sc->flags & (SC_FL_EOS|SC_FL_ABRT_DONE)))) {
Christopher Faulet5e29b762022-04-04 08:58:34 +02001362 int cur_flags = flags;
1363
1364 /* Compute transient CO_RFL_* flags */
1365 if (co_data(ic)) {
1366 cur_flags |= (CO_RFL_BUF_WET | CO_RFL_BUF_NOT_STUCK);
1367 }
1368
1369 /* <max> may be null. This is the mux responsibility to set
Willy Tarreaue68bc612022-05-27 11:23:05 +02001370 * SE_FL_RCV_MORE on the SC if more space is needed.
Christopher Faulet5e29b762022-04-04 08:58:34 +02001371 */
1372 max = channel_recv_max(ic);
Willy Tarreau0adb2812022-05-27 10:02:48 +02001373 ret = conn->mux->rcv_buf(sc, &ic->buf, max, cur_flags);
Christopher Faulet5e29b762022-04-04 08:58:34 +02001374
Willy Tarreau0adb2812022-05-27 10:02:48 +02001375 if (sc_ep_test(sc, SE_FL_WANT_ROOM)) {
Willy Tarreaub605c422022-05-17 17:04:55 +02001376 /* SE_FL_WANT_ROOM must not be reported if the channel's
Christopher Faulet5e29b762022-04-04 08:58:34 +02001377 * buffer is empty.
1378 */
1379 BUG_ON(c_empty(ic));
1380
Christopher Faulet7b3d38a2023-05-05 11:28:45 +02001381 sc_need_room(sc, channel_recv_max(ic) + 1);
Christopher Faulet5e29b762022-04-04 08:58:34 +02001382 /* Add READ_PARTIAL because some data are pending but
1383 * cannot be xferred to the channel
1384 */
Christopher Faulet285f7612022-12-12 08:28:55 +01001385 ic->flags |= CF_READ_EVENT;
Christopher Faulet2bb2ee22023-11-07 07:45:43 +01001386 sc_ep_report_read_activity(sc);
Christopher Faulet5e29b762022-04-04 08:58:34 +02001387 }
1388
1389 if (ret <= 0) {
1390 /* if we refrained from reading because we asked for a
1391 * flush to satisfy rcv_pipe(), we must not subscribe
1392 * and instead report that there's not enough room
1393 * here to proceed.
1394 */
1395 if (flags & CO_RFL_BUF_FLUSH)
Christopher Faulet7b3d38a2023-05-05 11:28:45 +02001396 sc_need_room(sc, -1);
Christopher Faulet5e29b762022-04-04 08:58:34 +02001397 break;
1398 }
1399
1400 cur_read += ret;
1401
1402 /* if we're allowed to directly forward data, we must update ->o */
Christopher Faulet64350bb2023-04-13 16:37:37 +02001403 if (ic->to_forward && !(sc_opposite(sc)->flags & (SC_FL_SHUT_DONE|SC_FL_SHUT_WANTED))) {
Christopher Faulet5e29b762022-04-04 08:58:34 +02001404 unsigned long fwd = ret;
1405 if (ic->to_forward != CHN_INFINITE_FORWARD) {
1406 if (fwd > ic->to_forward)
1407 fwd = ic->to_forward;
1408 ic->to_forward -= fwd;
1409 }
1410 c_adv(ic, fwd);
1411 }
1412
Christopher Faulet285f7612022-12-12 08:28:55 +01001413 ic->flags |= CF_READ_EVENT;
Christopher Faulet5e29b762022-04-04 08:58:34 +02001414 ic->total += ret;
1415
1416 /* End-of-input reached, we can leave. In this case, it is
Willy Tarreaue68bc612022-05-27 11:23:05 +02001417 * important to break the loop to not block the SC because of
Christopher Faulet5e29b762022-04-04 08:58:34 +02001418 * the channel's policies.This way, we are still able to receive
1419 * shutdowns.
1420 */
Willy Tarreau0adb2812022-05-27 10:02:48 +02001421 if (sc_ep_test(sc, SE_FL_EOI))
Christopher Faulet5e29b762022-04-04 08:58:34 +02001422 break;
1423
Christopher Faulet9a790f62023-03-16 14:40:03 +01001424 if ((sc->flags & SC_FL_RCV_ONCE) || --read_poll <= 0) {
1425 /* we don't expect to read more data */
Willy Tarreau0adb2812022-05-27 10:02:48 +02001426 sc_wont_read(sc);
Christopher Faulet5e29b762022-04-04 08:58:34 +02001427 break;
1428 }
1429
1430 /* if too many bytes were missing from last read, it means that
1431 * it's pointless trying to read again because the system does
1432 * not have them in buffers.
1433 */
1434 if (ret < max) {
1435 /* if a streamer has read few data, it may be because we
1436 * have exhausted system buffers. It's not worth trying
1437 * again.
1438 */
1439 if (ic->flags & CF_STREAMER) {
1440 /* we're stopped by the channel's policy */
Willy Tarreau0adb2812022-05-27 10:02:48 +02001441 sc_wont_read(sc);
Christopher Faulet5e29b762022-04-04 08:58:34 +02001442 break;
1443 }
1444
1445 /* if we read a large block smaller than what we requested,
1446 * it's almost certain we'll never get anything more.
1447 */
1448 if (ret >= global.tune.recv_enough) {
1449 /* we're stopped by the channel's policy */
Willy Tarreau0adb2812022-05-27 10:02:48 +02001450 sc_wont_read(sc);
Christopher Faulet5e29b762022-04-04 08:58:34 +02001451 break;
1452 }
1453 }
1454
1455 /* if we are waiting for more space, don't try to read more data
1456 * right now.
1457 */
Willy Tarreau0adb2812022-05-27 10:02:48 +02001458 if (sc->flags & (SC_FL_WONT_READ|SC_FL_NEED_BUFF|SC_FL_NEED_ROOM))
Christopher Faulet5e29b762022-04-04 08:58:34 +02001459 break;
1460 } /* while !flags */
1461
1462 done_recv:
1463 if (cur_read) {
1464 if ((ic->flags & (CF_STREAMER | CF_STREAMER_FAST)) &&
1465 (cur_read <= ic->buf.size / 2)) {
1466 ic->xfer_large = 0;
1467 ic->xfer_small++;
1468 if (ic->xfer_small >= 3) {
1469 /* we have read less than half of the buffer in
1470 * one pass, and this happened at least 3 times.
1471 * This is definitely not a streamer.
1472 */
1473 ic->flags &= ~(CF_STREAMER | CF_STREAMER_FAST);
1474 }
1475 else if (ic->xfer_small >= 2) {
1476 /* if the buffer has been at least half full twice,
1477 * we receive faster than we send, so at least it
1478 * is not a "fast streamer".
1479 */
1480 ic->flags &= ~CF_STREAMER_FAST;
1481 }
1482 }
Christopher Faulet0b25b6a2023-11-17 11:23:11 +01001483 else if (!(ic->flags & CF_STREAMER_FAST) && (cur_read >= channel_data_limit(ic))) {
Christopher Faulet5e29b762022-04-04 08:58:34 +02001484 /* we read a full buffer at once */
1485 ic->xfer_small = 0;
1486 ic->xfer_large++;
1487 if (ic->xfer_large >= 3) {
1488 /* we call this buffer a fast streamer if it manages
1489 * to be filled in one call 3 consecutive times.
1490 */
1491 ic->flags |= (CF_STREAMER | CF_STREAMER_FAST);
1492 }
1493 }
1494 else {
1495 ic->xfer_small = 0;
1496 ic->xfer_large = 0;
1497 }
1498 ic->last_read = now_ms;
Christopher Faulet4c135682023-02-16 11:09:31 +01001499 sc_ep_report_read_activity(sc);
Christopher Faulet5e29b762022-04-04 08:58:34 +02001500 }
1501
1502 end_recv:
1503 ret = (cur_read != 0);
1504
1505 /* Report EOI on the channel if it was reached from the mux point of
1506 * view. */
Christopher Faulet904763f2023-03-22 14:53:11 +01001507 if (sc_ep_test(sc, SE_FL_EOI) && !(sc->flags & SC_FL_EOI)) {
Christopher Faulet4c135682023-02-16 11:09:31 +01001508 sc_ep_report_read_activity(sc);
Christopher Faulet904763f2023-03-22 14:53:11 +01001509 sc->flags |= SC_FL_EOI;
1510 ic->flags |= CF_READ_EVENT;
Christopher Faulet5e29b762022-04-04 08:58:34 +02001511 ret = 1;
1512 }
1513
Christopher Fauletb208d8c2023-03-21 11:25:21 +01001514 if (sc_ep_test(sc, SE_FL_EOS)) {
Christopher Faulet5e29b762022-04-04 08:58:34 +02001515 /* we received a shutdown */
Christopher Faulet5e29b762022-04-04 08:58:34 +02001516 if (ic->flags & CF_AUTO_CLOSE)
Christopher Fauletdf7cd712023-04-13 15:56:26 +02001517 sc_schedule_shutdown(sc_opposite(sc));
Christopher Faulet1aec6c92023-04-17 17:29:29 +02001518 sc_conn_eos(sc);
Christopher Faulet5e29b762022-04-04 08:58:34 +02001519 ret = 1;
1520 }
Christopher Fauletb208d8c2023-03-21 11:25:21 +01001521
Christopher Fauleta1d14a72023-04-14 10:42:08 +02001522 if (sc_ep_test(sc, SE_FL_ERROR)) {
1523 sc->flags |= SC_FL_ERROR;
Christopher Fauletb208d8c2023-03-21 11:25:21 +01001524 ret = 1;
Christopher Fauleta1d14a72023-04-14 10:42:08 +02001525 }
Willy Tarreau0adb2812022-05-27 10:02:48 +02001526 else if (!(sc->flags & (SC_FL_WONT_READ|SC_FL_NEED_BUFF|SC_FL_NEED_ROOM)) &&
Christopher Fauletca5309a2023-04-17 16:17:32 +02001527 !(sc->flags & (SC_FL_EOS|SC_FL_ABRT_DONE))) {
Christopher Faulet5e29b762022-04-04 08:58:34 +02001528 /* Subscribe to receive events if we're blocking on I/O */
Willy Tarreau0adb2812022-05-27 10:02:48 +02001529 conn->mux->subscribe(sc, SUB_RETRY_RECV, &sc->wait_event);
1530 se_have_no_more_data(sc->sedesc);
Christopher Fauletb208d8c2023-03-21 11:25:21 +01001531 }
1532 else {
Willy Tarreau0adb2812022-05-27 10:02:48 +02001533 se_have_more_data(sc->sedesc);
Christopher Faulet5e29b762022-04-04 08:58:34 +02001534 ret = 1;
1535 }
Christopher Faulet8019f782023-03-23 17:30:29 +01001536
Christopher Faulet5e29b762022-04-04 08:58:34 +02001537 return ret;
1538}
1539
Willy Tarreau4596fe22022-05-17 19:07:51 +02001540/* This tries to perform a synchronous receive on the stream connector to
Christopher Faulet5e29b762022-04-04 08:58:34 +02001541 * try to collect last arrived data. In practice it's only implemented on
Willy Tarreau4596fe22022-05-17 19:07:51 +02001542 * stconns. Returns 0 if nothing was done, non-zero if new data or a
Christopher Faulet5e29b762022-04-04 08:58:34 +02001543 * shutdown were collected. This may result on some delayed receive calls
1544 * to be programmed and performed later, though it doesn't provide any
1545 * such guarantee.
1546 */
Willy Tarreau0adb2812022-05-27 10:02:48 +02001547int sc_conn_sync_recv(struct stconn *sc)
Christopher Faulet5e29b762022-04-04 08:58:34 +02001548{
Willy Tarreau0adb2812022-05-27 10:02:48 +02001549 if (!sc_state_in(sc->state, SC_SB_RDY|SC_SB_EST))
Christopher Faulet5e29b762022-04-04 08:58:34 +02001550 return 0;
1551
Willy Tarreau0adb2812022-05-27 10:02:48 +02001552 if (!sc_mux_ops(sc))
Willy Tarreau4596fe22022-05-17 19:07:51 +02001553 return 0; // only stconns are supported
Christopher Faulet5e29b762022-04-04 08:58:34 +02001554
Willy Tarreau0adb2812022-05-27 10:02:48 +02001555 if (sc->wait_event.events & SUB_RETRY_RECV)
Christopher Faulet5e29b762022-04-04 08:58:34 +02001556 return 0; // already subscribed
1557
Willy Tarreau0adb2812022-05-27 10:02:48 +02001558 if (!sc_is_recv_allowed(sc))
Christopher Faulet5e29b762022-04-04 08:58:34 +02001559 return 0; // already failed
1560
Willy Tarreau0adb2812022-05-27 10:02:48 +02001561 return sc_conn_recv(sc);
Christopher Faulet5e29b762022-04-04 08:58:34 +02001562}
1563
1564/*
1565 * This function is called to send buffer data to a stream socket.
1566 * It calls the mux layer's snd_buf function. It relies on the
1567 * caller to commit polling changes. The caller should check conn->flags
1568 * for errors.
1569 */
Willy Tarreau0adb2812022-05-27 10:02:48 +02001570static int sc_conn_send(struct stconn *sc)
Christopher Faulet5e29b762022-04-04 08:58:34 +02001571{
Willy Tarreau0adb2812022-05-27 10:02:48 +02001572 struct connection *conn = __sc_conn(sc);
Christopher Faulet904763f2023-03-22 14:53:11 +01001573 struct stconn *sco = sc_opposite(sc);
Willy Tarreau0adb2812022-05-27 10:02:48 +02001574 struct stream *s = __sc_strm(sc);
1575 struct channel *oc = sc_oc(sc);
Christopher Faulet5e29b762022-04-04 08:58:34 +02001576 int ret;
1577 int did_send = 0;
1578
Willy Tarreau0adb2812022-05-27 10:02:48 +02001579 if (sc_ep_test(sc, SE_FL_ERROR | SE_FL_ERR_PENDING) || sc_is_conn_error(sc)) {
Christopher Faulet5e29b762022-04-04 08:58:34 +02001580 /* We're probably there because the tasklet was woken up,
1581 * but process_stream() ran before, detected there were an
Willy Tarreaue68bc612022-05-27 11:23:05 +02001582 * error and put the SC back to SC_ST_TAR. There's still
Christopher Faulet5e29b762022-04-04 08:58:34 +02001583 * CO_FL_ERROR on the connection but we don't want to add
Willy Tarreaub605c422022-05-17 17:04:55 +02001584 * SE_FL_ERROR back, so give up
Christopher Faulet5e29b762022-04-04 08:58:34 +02001585 */
Willy Tarreau0adb2812022-05-27 10:02:48 +02001586 if (sc->state < SC_ST_CON)
Christopher Faulet5e29b762022-04-04 08:58:34 +02001587 return 0;
Christopher Faulet56a2b602023-04-14 09:42:59 +02001588 BUG_ON(sc_ep_test(sc, SE_FL_EOS|SE_FL_ERROR|SE_FL_ERR_PENDING) == (SE_FL_EOS|SE_FL_ERR_PENDING));
Christopher Faulet326ec912024-07-29 17:48:16 +02001589 if (sc_ep_test(sc, SE_FL_ERROR))
1590 sc->flags |= SC_FL_ERROR;
Christopher Faulet5e29b762022-04-04 08:58:34 +02001591 return 1;
1592 }
1593
1594 /* We're already waiting to be able to send, give up */
Willy Tarreau0adb2812022-05-27 10:02:48 +02001595 if (sc->wait_event.events & SUB_RETRY_SEND)
Christopher Faulet5e29b762022-04-04 08:58:34 +02001596 return 0;
1597
1598 /* we might have been called just after an asynchronous shutw */
Christopher Faulet208c7122023-04-13 16:16:15 +02001599 if (sc->flags & SC_FL_SHUT_DONE)
Christopher Faulet5e29b762022-04-04 08:58:34 +02001600 return 1;
1601
1602 /* we must wait because the mux is not installed yet */
1603 if (!conn->mux)
1604 return 0;
1605
1606 if (oc->pipe && conn->xprt->snd_pipe && conn->mux->snd_pipe) {
Willy Tarreau0adb2812022-05-27 10:02:48 +02001607 ret = conn->mux->snd_pipe(sc, oc->pipe);
Christopher Faulet5e29b762022-04-04 08:58:34 +02001608 if (ret > 0)
1609 did_send = 1;
1610
Christopher Faulet4c492102024-07-30 08:41:03 +02001611 if (!oc->pipe->data || sc_ep_test(sc, SE_FL_ERROR | SE_FL_ERR_PENDING)) {
Christopher Faulet5e29b762022-04-04 08:58:34 +02001612 put_pipe(oc->pipe);
1613 oc->pipe = NULL;
1614 }
1615
1616 if (oc->pipe)
1617 goto end;
1618 }
1619
1620 /* At this point, the pipe is empty, but we may still have data pending
1621 * in the normal buffer.
1622 */
1623 if (co_data(oc)) {
1624 /* when we're here, we already know that there is no spliced
1625 * data left, and that there are sendable buffered data.
1626 */
1627
1628 /* check if we want to inform the kernel that we're interested in
1629 * sending more data after this call. We want this if :
1630 * - we're about to close after this last send and want to merge
1631 * the ongoing FIN with the last segment.
1632 * - we know we can't send everything at once and must get back
1633 * here because of unaligned data
1634 * - there is still a finite amount of data to forward
1635 * The test is arranged so that the most common case does only 2
1636 * tests.
1637 */
1638 unsigned int send_flag = 0;
1639
Christopher Faulet68ef2182023-03-17 15:38:18 +01001640 if ((!(sc->flags & (SC_FL_SND_ASAP|SC_FL_SND_NEVERWAIT)) &&
Christopher Faulet5e29b762022-04-04 08:58:34 +02001641 ((oc->to_forward && oc->to_forward != CHN_INFINITE_FORWARD) ||
Christopher Faulet84d3ef92023-03-17 15:45:58 +01001642 (sc->flags & SC_FL_SND_EXP_MORE) ||
Christopher Faulet5e29b762022-04-04 08:58:34 +02001643 (IS_HTX_STRM(s) &&
Christopher Fauletca5309a2023-04-17 16:17:32 +02001644 (!(sco->flags & (SC_FL_EOI|SC_FL_EOS|SC_FL_ABRT_DONE)) && htx_expect_more(htxbuf(&oc->buf)))))) ||
Christopher Faulet5e29b762022-04-04 08:58:34 +02001645 ((oc->flags & CF_ISRESP) &&
Christopher Faulet87633c32023-04-03 18:32:50 +02001646 (oc->flags & CF_AUTO_CLOSE) &&
Christopher Faulete38534c2023-04-13 15:45:24 +02001647 (sc->flags & SC_FL_SHUT_WANTED)))
Christopher Faulet5e29b762022-04-04 08:58:34 +02001648 send_flag |= CO_SFL_MSG_MORE;
1649
1650 if (oc->flags & CF_STREAMER)
1651 send_flag |= CO_SFL_STREAMER;
1652
1653 if (s->txn && s->txn->flags & TX_L7_RETRY && !b_data(&s->txn->l7_buffer)) {
1654 /* If we want to be able to do L7 retries, copy
1655 * the data we're about to send, so that we are able
1656 * to resend them if needed
1657 */
1658 /* Try to allocate a buffer if we had none.
1659 * If it fails, the next test will just
1660 * disable the l7 retries by setting
1661 * l7_conn_retries to 0.
1662 */
1663 if (s->txn->req.msg_state != HTTP_MSG_DONE)
1664 s->txn->flags &= ~TX_L7_RETRY;
1665 else {
1666 if (b_alloc(&s->txn->l7_buffer) == NULL)
1667 s->txn->flags &= ~TX_L7_RETRY;
1668 else {
1669 memcpy(b_orig(&s->txn->l7_buffer),
1670 b_orig(&oc->buf),
1671 b_size(&oc->buf));
1672 s->txn->l7_buffer.head = co_data(oc);
1673 b_add(&s->txn->l7_buffer, co_data(oc));
1674 }
1675
1676 }
1677 }
1678
Willy Tarreau0adb2812022-05-27 10:02:48 +02001679 ret = conn->mux->snd_buf(sc, &oc->buf, co_data(oc), send_flag);
Christopher Faulet5e29b762022-04-04 08:58:34 +02001680 if (ret > 0) {
1681 did_send = 1;
1682 c_rew(oc, ret);
1683 c_realign_if_empty(oc);
1684
1685 if (!co_data(oc)) {
1686 /* Always clear both flags once everything has been sent, they're one-shot */
Christopher Faulet84d3ef92023-03-17 15:45:58 +01001687 sc->flags &= ~(SC_FL_SND_ASAP|SC_FL_SND_EXP_MORE);
Christopher Faulet5e29b762022-04-04 08:58:34 +02001688 }
1689 /* if some data remain in the buffer, it's only because the
1690 * system buffers are full, we will try next time.
1691 */
Christopher Faulet13045f02022-04-01 14:23:38 +02001692 }
1693 }
Christopher Faulet5e29b762022-04-04 08:58:34 +02001694
1695 end:
1696 if (did_send) {
Christopher Fauletd8988412022-12-20 18:10:04 +01001697 oc->flags |= CF_WRITE_EVENT | CF_WROTE_DATA;
Willy Tarreau0adb2812022-05-27 10:02:48 +02001698 if (sc->state == SC_ST_CON)
1699 sc->state = SC_ST_RDY;
Christopher Faulet5e29b762022-04-04 08:58:34 +02001700 }
1701
Christopher Faulete7405d42023-05-05 11:40:30 +02001702 if (!sco->room_needed || (did_send && (sco->room_needed < 0 || channel_recv_max(sc_oc(sc)) >= sco->room_needed)))
1703 sc_have_room(sco);
1704
Willy Tarreau0adb2812022-05-27 10:02:48 +02001705 if (sc_ep_test(sc, SE_FL_ERROR | SE_FL_ERR_PENDING)) {
Christopher Faulet2e56a732023-01-26 16:18:09 +01001706 oc->flags |= CF_WRITE_EVENT;
Christopher Faulet56a2b602023-04-14 09:42:59 +02001707 BUG_ON(sc_ep_test(sc, SE_FL_EOS|SE_FL_ERROR|SE_FL_ERR_PENDING) == (SE_FL_EOS|SE_FL_ERR_PENDING));
Christopher Fauletd0c57d32023-04-18 18:38:32 +02001708 if (sc_ep_test(sc, SE_FL_ERROR))
1709 sc->flags |= SC_FL_ERROR;
Christopher Faulet5e29b762022-04-04 08:58:34 +02001710 return 1;
1711 }
1712
Christopher Faulete65daba2023-11-15 17:33:06 +01001713 if (channel_is_empty(oc)) {
Christopher Faulet7fb330b2023-11-17 15:26:57 +01001714 if (did_send)
1715 sc_ep_report_send_activity(sc);
Christopher Faulete65daba2023-11-15 17:33:06 +01001716 }
1717 else {
Christopher Faulet59b240c2023-02-27 16:38:12 +01001718 /* We couldn't send all of our data, let the mux know we'd like to send more */
Willy Tarreau0adb2812022-05-27 10:02:48 +02001719 conn->mux->subscribe(sc, SUB_RETRY_SEND, &sc->wait_event);
Christopher Faulete65daba2023-11-15 17:33:06 +01001720 if (sc_state_in(sc->state, SC_SB_EST|SC_SB_DIS|SC_SB_CLO))
1721 sc_ep_report_blocked_send(sc, did_send);
Christopher Faulet59b240c2023-02-27 16:38:12 +01001722 }
1723
Christopher Faulet5e29b762022-04-04 08:58:34 +02001724 return did_send;
1725}
1726
Christopher Fauletd8988412022-12-20 18:10:04 +01001727/* perform a synchronous send() for the stream connector. The CF_WRITE_EVENT
1728 * flag are cleared prior to the attempt, and will possibly be updated in case
1729 * of success.
Christopher Faulet5e29b762022-04-04 08:58:34 +02001730 */
Willy Tarreau0adb2812022-05-27 10:02:48 +02001731void sc_conn_sync_send(struct stconn *sc)
Christopher Faulet5e29b762022-04-04 08:58:34 +02001732{
Willy Tarreau0adb2812022-05-27 10:02:48 +02001733 struct channel *oc = sc_oc(sc);
Christopher Faulet5e29b762022-04-04 08:58:34 +02001734
Christopher Fauletd8988412022-12-20 18:10:04 +01001735 oc->flags &= ~CF_WRITE_EVENT;
Christopher Faulet5e29b762022-04-04 08:58:34 +02001736
Christopher Faulet208c7122023-04-13 16:16:15 +02001737 if (sc->flags & SC_FL_SHUT_DONE)
Christopher Faulet5e29b762022-04-04 08:58:34 +02001738 return;
1739
1740 if (channel_is_empty(oc))
1741 return;
1742
Willy Tarreau0adb2812022-05-27 10:02:48 +02001743 if (!sc_state_in(sc->state, SC_SB_CON|SC_SB_RDY|SC_SB_EST))
Christopher Faulet5e29b762022-04-04 08:58:34 +02001744 return;
1745
Willy Tarreau0adb2812022-05-27 10:02:48 +02001746 if (!sc_mux_ops(sc))
Christopher Faulet5e29b762022-04-04 08:58:34 +02001747 return;
1748
Willy Tarreau0adb2812022-05-27 10:02:48 +02001749 sc_conn_send(sc);
Christopher Faulet5e29b762022-04-04 08:58:34 +02001750}
1751
1752/* Called by I/O handlers after completion.. It propagates
Willy Tarreau4596fe22022-05-17 19:07:51 +02001753 * connection flags to the stream connector, updates the stream (which may or
Christopher Faulet5e29b762022-04-04 08:58:34 +02001754 * may not take this opportunity to try to forward data), then update the
Willy Tarreau4596fe22022-05-17 19:07:51 +02001755 * connection's polling based on the channels and stream connector's final
Christopher Faulet5e29b762022-04-04 08:58:34 +02001756 * states. The function always returns 0.
1757 */
Willy Tarreau0adb2812022-05-27 10:02:48 +02001758static int sc_conn_process(struct stconn *sc)
Christopher Faulet5e29b762022-04-04 08:58:34 +02001759{
Willy Tarreau0adb2812022-05-27 10:02:48 +02001760 struct connection *conn = __sc_conn(sc);
1761 struct channel *ic = sc_ic(sc);
1762 struct channel *oc = sc_oc(sc);
Christopher Faulet5e29b762022-04-04 08:58:34 +02001763
1764 BUG_ON(!conn);
1765
1766 /* If we have data to send, try it now */
Willy Tarreau0adb2812022-05-27 10:02:48 +02001767 if (!channel_is_empty(oc) && !(sc->wait_event.events & SUB_RETRY_SEND))
1768 sc_conn_send(sc);
Christopher Faulet5e29b762022-04-04 08:58:34 +02001769
Willy Tarreau4596fe22022-05-17 19:07:51 +02001770 /* First step, report to the stream connector what was detected at the
Christopher Faulet5e29b762022-04-04 08:58:34 +02001771 * connection layer : errors and connection establishment.
Christopher Faulet88d05a02023-04-14 12:03:50 +02001772 * Only add SC_FL_ERROR if we're connected, or we're attempting to
Christopher Faulet5e29b762022-04-04 08:58:34 +02001773 * connect, we may get there because we got woken up, but only run
1774 * after process_stream() noticed there were an error, and decided
1775 * to retry to connect, the connection may still have CO_FL_ERROR,
Christopher Faulet88d05a02023-04-14 12:03:50 +02001776 * and we don't want to add SC_FL_ERROR back
Christopher Faulet5e29b762022-04-04 08:58:34 +02001777 *
Willy Tarreau462b9892022-05-18 18:06:53 +02001778 * Note: This test is only required because sc_conn_process is also the SI
1779 * wake callback. Otherwise sc_conn_recv()/sc_conn_send() already take
Christopher Faulet5e29b762022-04-04 08:58:34 +02001780 * care of it.
1781 */
1782
Willy Tarreau0adb2812022-05-27 10:02:48 +02001783 if (sc->state >= SC_ST_CON) {
Christopher Faulet88d05a02023-04-14 12:03:50 +02001784 if (sc_is_conn_error(sc))
Christopher Fauleta1d14a72023-04-14 10:42:08 +02001785 sc->flags |= SC_FL_ERROR;
Christopher Faulet5e29b762022-04-04 08:58:34 +02001786 }
1787
1788 /* If we had early data, and the handshake ended, then
1789 * we can remove the flag, and attempt to wake the task up,
1790 * in the event there's an analyser waiting for the end of
1791 * the handshake.
1792 */
1793 if (!(conn->flags & (CO_FL_WAIT_XPRT | CO_FL_EARLY_SSL_HS)) &&
Willy Tarreau0adb2812022-05-27 10:02:48 +02001794 sc_ep_test(sc, SE_FL_WAIT_FOR_HS)) {
1795 sc_ep_clr(sc, SE_FL_WAIT_FOR_HS);
1796 task_wakeup(sc_strm_task(sc), TASK_WOKEN_MSG);
Christopher Faulet5e29b762022-04-04 08:58:34 +02001797 }
1798
Willy Tarreau0adb2812022-05-27 10:02:48 +02001799 if (!sc_state_in(sc->state, SC_SB_EST|SC_SB_DIS|SC_SB_CLO) &&
Christopher Faulet5e29b762022-04-04 08:58:34 +02001800 (conn->flags & CO_FL_WAIT_XPRT) == 0) {
Christopher Fauletca679922022-07-20 13:24:04 +02001801 if (sc->flags & SC_FL_ISBACK)
1802 __sc_strm(sc)->conn_exp = TICK_ETERNITY;
Christopher Fauletb96f2aa2022-12-12 08:11:36 +01001803 oc->flags |= CF_WRITE_EVENT;
Willy Tarreau0adb2812022-05-27 10:02:48 +02001804 if (sc->state == SC_ST_CON)
1805 sc->state = SC_ST_RDY;
Christopher Faulet5e29b762022-04-04 08:58:34 +02001806 }
1807
1808 /* Report EOS on the channel if it was reached from the mux point of
1809 * view.
1810 *
Willy Tarreau462b9892022-05-18 18:06:53 +02001811 * Note: This test is only required because sc_conn_process is also the SI
1812 * wake callback. Otherwise sc_conn_recv()/sc_conn_send() already take
Christopher Faulet5e29b762022-04-04 08:58:34 +02001813 * care of it.
1814 */
Christopher Faulet1aec6c92023-04-17 17:29:29 +02001815 if (sc_ep_test(sc, SE_FL_EOS) && !(sc->flags & SC_FL_EOS)) {
Christopher Faulet5e29b762022-04-04 08:58:34 +02001816 /* we received a shutdown */
Christopher Faulet5e29b762022-04-04 08:58:34 +02001817 if (ic->flags & CF_AUTO_CLOSE)
Christopher Fauletdf7cd712023-04-13 15:56:26 +02001818 sc_schedule_shutdown(sc_opposite(sc));
Christopher Faulet1aec6c92023-04-17 17:29:29 +02001819 sc_conn_eos(sc);
Christopher Faulet5e29b762022-04-04 08:58:34 +02001820 }
1821
1822 /* Report EOI on the channel if it was reached from the mux point of
1823 * view.
1824 *
Willy Tarreau462b9892022-05-18 18:06:53 +02001825 * Note: This test is only required because sc_conn_process is also the SI
1826 * wake callback. Otherwise sc_conn_recv()/sc_conn_send() already take
Christopher Faulet5e29b762022-04-04 08:58:34 +02001827 * care of it.
1828 */
Christopher Faulet904763f2023-03-22 14:53:11 +01001829 if (sc_ep_test(sc, SE_FL_EOI) && !(sc->flags & SC_FL_EOI)) {
1830 sc->flags |= SC_FL_EOI;
1831 ic->flags |= CF_READ_EVENT;
Christopher Faulet2bb2ee22023-11-07 07:45:43 +01001832 sc_ep_report_read_activity(sc);
Christopher Faulet904763f2023-03-22 14:53:11 +01001833 }
Christopher Faulet5e29b762022-04-04 08:58:34 +02001834
Christopher Fauleta1d14a72023-04-14 10:42:08 +02001835 if (sc_ep_test(sc, SE_FL_ERROR))
1836 sc->flags |= SC_FL_ERROR;
1837
Willy Tarreau4596fe22022-05-17 19:07:51 +02001838 /* Second step : update the stream connector and channels, try to forward any
Christopher Faulet5e29b762022-04-04 08:58:34 +02001839 * pending data, then possibly wake the stream up based on the new
Willy Tarreau4596fe22022-05-17 19:07:51 +02001840 * stream connector status.
Christopher Faulet5e29b762022-04-04 08:58:34 +02001841 */
Willy Tarreau0adb2812022-05-27 10:02:48 +02001842 sc_notify(sc);
1843 stream_release_buffers(__sc_strm(sc));
Christopher Faulet5e29b762022-04-04 08:58:34 +02001844 return 0;
1845}
1846
Willy Tarreau4596fe22022-05-17 19:07:51 +02001847/* This is the ->process() function for any stream connector's wait_event task.
1848 * It's assigned during the stream connector's initialization, for any type of
1849 * stream connector. Thus it is always safe to perform a tasklet_wakeup() on a
Willy Tarreaue68bc612022-05-27 11:23:05 +02001850 * stream connector, as the presence of the SC is checked there.
Christopher Faulet5e29b762022-04-04 08:58:34 +02001851 */
Willy Tarreau462b9892022-05-18 18:06:53 +02001852struct task *sc_conn_io_cb(struct task *t, void *ctx, unsigned int state)
Christopher Faulet5e29b762022-04-04 08:58:34 +02001853{
Willy Tarreau0adb2812022-05-27 10:02:48 +02001854 struct stconn *sc = ctx;
Christopher Faulet5e29b762022-04-04 08:58:34 +02001855 int ret = 0;
1856
Willy Tarreau0adb2812022-05-27 10:02:48 +02001857 if (!sc_conn(sc))
Christopher Faulet5e29b762022-04-04 08:58:34 +02001858 return t;
1859
Willy Tarreau0adb2812022-05-27 10:02:48 +02001860 if (!(sc->wait_event.events & SUB_RETRY_SEND) && !channel_is_empty(sc_oc(sc)))
1861 ret = sc_conn_send(sc);
1862 if (!(sc->wait_event.events & SUB_RETRY_RECV))
1863 ret |= sc_conn_recv(sc);
Christopher Faulet5e29b762022-04-04 08:58:34 +02001864 if (ret != 0)
Willy Tarreau0adb2812022-05-27 10:02:48 +02001865 sc_conn_process(sc);
Christopher Faulet5e29b762022-04-04 08:58:34 +02001866
Willy Tarreau0adb2812022-05-27 10:02:48 +02001867 stream_release_buffers(__sc_strm(sc));
Christopher Faulet5e29b762022-04-04 08:58:34 +02001868 return t;
1869}
1870
Christopher Fauletb36e5122023-04-17 17:32:43 +02001871/*
1872 * This function propagates an end-of-stream received from an applet. It
1873 * updates the stream connector. If it is is already shut, the applet is
1874 * released. Otherwise, we try to forward the shutdown, immediately or ASAP.
1875 */
1876static void sc_applet_eos(struct stconn *sc)
1877{
1878 struct channel *ic = sc_ic(sc);
1879
1880 BUG_ON(!sc_appctx(sc));
1881
1882 if (sc->flags & (SC_FL_EOS|SC_FL_ABRT_DONE))
1883 return;
1884 sc->flags |= SC_FL_EOS;
1885 ic->flags |= CF_READ_EVENT;
Christopher Faulet2bb2ee22023-11-07 07:45:43 +01001886 sc_ep_report_read_activity(sc);
Christopher Fauletb36e5122023-04-17 17:32:43 +02001887
1888 /* Note: on abort, we don't call the applet */
1889
1890 if (!sc_state_in(sc->state, SC_SB_CON|SC_SB_RDY|SC_SB_EST))
1891 return;
1892
1893 if (sc->flags & SC_FL_SHUT_DONE) {
1894 appctx_shut(__sc_appctx(sc));
1895 sc->state = SC_ST_DIS;
1896 if (sc->flags & SC_FL_ISBACK)
1897 __sc_strm(sc)->conn_exp = TICK_ETERNITY;
1898 }
1899 else if (sc_cond_forward_shut(sc))
1900 return sc_app_shut_applet(sc);
1901}
1902
Christopher Faulet5e29b762022-04-04 08:58:34 +02001903/* Callback to be used by applet handlers upon completion. It updates the stream
1904 * (which may or may not take this opportunity to try to forward data), then
Willy Tarreau4596fe22022-05-17 19:07:51 +02001905 * may re-enable the applet's based on the channels and stream connector's final
Christopher Faulet5e29b762022-04-04 08:58:34 +02001906 * states.
1907 */
Willy Tarreau0adb2812022-05-27 10:02:48 +02001908static int sc_applet_process(struct stconn *sc)
Christopher Faulet5e29b762022-04-04 08:58:34 +02001909{
Willy Tarreau0adb2812022-05-27 10:02:48 +02001910 struct channel *ic = sc_ic(sc);
Christopher Faulet5e29b762022-04-04 08:58:34 +02001911
Willy Tarreau0adb2812022-05-27 10:02:48 +02001912 BUG_ON(!sc_appctx(sc));
Christopher Faulet5e29b762022-04-04 08:58:34 +02001913
Christopher Fauletf8fbb6d2023-03-21 11:49:21 +01001914 /* Report EOI on the channel if it was reached from the applet point of
1915 * view. */
Christopher Faulet904763f2023-03-22 14:53:11 +01001916 if (sc_ep_test(sc, SE_FL_EOI) && !(sc->flags & SC_FL_EOI)) {
Christopher Fauletf8fbb6d2023-03-21 11:49:21 +01001917 sc_ep_report_read_activity(sc);
Christopher Faulet904763f2023-03-22 14:53:11 +01001918 sc->flags |= SC_FL_EOI;
1919 ic->flags |= CF_READ_EVENT;
Christopher Fauletf8fbb6d2023-03-21 11:49:21 +01001920 }
1921
Christopher Fauleta1d14a72023-04-14 10:42:08 +02001922 if (sc_ep_test(sc, SE_FL_ERROR))
1923 sc->flags |= SC_FL_ERROR;
1924
Christopher Faulet0ffc9d72023-03-21 14:19:08 +01001925 if (sc_ep_test(sc, SE_FL_EOS)) {
1926 /* we received a shutdown */
Christopher Fauletb36e5122023-04-17 17:32:43 +02001927 sc_applet_eos(sc);
Christopher Faulet0ffc9d72023-03-21 14:19:08 +01001928 }
1929
Christopher Faulete8bcef52023-04-14 09:45:41 +02001930 BUG_ON(sc_ep_test(sc, SE_FL_HAVE_NO_DATA|SE_FL_EOI) == SE_FL_EOI);
1931
Christopher Faulet5e29b762022-04-04 08:58:34 +02001932 /* If the applet wants to write and the channel is closed, it's a
1933 * broken pipe and it must be reported.
1934 */
Christopher Fauletca5309a2023-04-17 16:17:32 +02001935 if (!sc_ep_test(sc, SE_FL_HAVE_NO_DATA) && (sc->flags & (SC_FL_EOS|SC_FL_ABRT_DONE)))
Willy Tarreau0adb2812022-05-27 10:02:48 +02001936 sc_ep_set(sc, SE_FL_ERROR);
Christopher Faulet5e29b762022-04-04 08:58:34 +02001937
1938 /* automatically mark the applet having data available if it reported
1939 * begin blocked by the channel.
1940 */
Willy Tarreau0adb2812022-05-27 10:02:48 +02001941 if ((sc->flags & (SC_FL_WONT_READ|SC_FL_NEED_BUFF|SC_FL_NEED_ROOM)) ||
1942 sc_ep_test(sc, SE_FL_APPLET_NEED_CONN))
1943 applet_have_more_data(__sc_appctx(sc));
Christopher Faulet5e29b762022-04-04 08:58:34 +02001944
Willy Tarreau4596fe22022-05-17 19:07:51 +02001945 /* update the stream connector, channels, and possibly wake the stream up */
Willy Tarreau0adb2812022-05-27 10:02:48 +02001946 sc_notify(sc);
1947 stream_release_buffers(__sc_strm(sc));
Christopher Faulet5e29b762022-04-04 08:58:34 +02001948
Willy Tarreau19c65a92022-05-27 08:49:24 +02001949 /* sc_notify may have passed through chk_snd and released some blocking
Willy Tarreau15252cd2022-05-25 16:36:21 +02001950 * flags. Process_stream will consider those flags to wake up the
Christopher Faulet5e29b762022-04-04 08:58:34 +02001951 * appctx but in the case the task is not in runqueue we may have to
1952 * wakeup the appctx immediately.
1953 */
Willy Tarreau0adb2812022-05-27 10:02:48 +02001954 if (sc_is_recv_allowed(sc) || sc_is_send_allowed(sc))
1955 appctx_wakeup(__sc_appctx(sc));
Christopher Faulet5e29b762022-04-04 08:58:34 +02001956 return 0;
Christopher Faulet13045f02022-04-01 14:23:38 +02001957}
Christopher Fauletb68f77d2022-06-16 16:24:16 +02001958
1959
1960/* Prepares an endpoint upgrade. We don't now at this stage if the upgrade will
1961 * succeed or not and if the stconn will be reused by the new endpoint. Thus,
1962 * for now, only pretend the stconn is detached.
1963 */
1964void sc_conn_prepare_endp_upgrade(struct stconn *sc)
1965{
1966 BUG_ON(!sc_conn(sc) || !sc->app);
1967 sc_ep_clr(sc, SE_FL_T_MUX);
1968 sc_ep_set(sc, SE_FL_DETACHED);
1969}
1970
Ilya Shipitsin3b64a282022-07-29 22:26:53 +05001971/* Endpoint upgrade failed. Restore the stconn state. */
Christopher Fauletb68f77d2022-06-16 16:24:16 +02001972void sc_conn_abort_endp_upgrade(struct stconn *sc)
1973{
1974 sc_ep_set(sc, SE_FL_T_MUX);
1975 sc_ep_clr(sc, SE_FL_DETACHED);
1976}
1977
1978/* Commit the endpoint upgrade. If stconn is attached, it means the new endpoint
1979 * use it. So we do nothing. Otherwise, the stconn will be destroy with the
1980 * overlying stream. So, it means we must commit the detach.
1981*/
1982void sc_conn_commit_endp_upgrade(struct stconn *sc)
1983{
1984 if (!sc_ep_test(sc, SE_FL_DETACHED))
1985 return;
1986 sc_detach_endp(&sc);
1987 /* Because it was already set as detached, the sedesc must be preserved */
Willy Tarreau6a378d12022-08-11 13:56:42 +02001988 BUG_ON(!sc);
Christopher Fauletb68f77d2022-06-16 16:24:16 +02001989 BUG_ON(!sc->sedesc);
1990}