blob: dbdd78b0949c4fc94f33d5039886fdcf2ede2e74 [file] [log] [blame]
Willy Tarreau53a47662017-08-28 10:53:00 +02001/*
2 * Pass-through mux-demux for connections
3 *
4 * Copyright 2017 Willy Tarreau <w@1wt.eu>
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 <common/config.h>
14#include <proto/connection.h>
15#include <proto/stream.h>
16
Olivier Houchard9aaf7782017-09-13 18:30:23 +020017/* Initialize the mux once it's attached. It is expected that conn->mux_ctx
18 * points to the existing conn_stream (for outgoing connections) or NULL (for
19 * incoming ones, in which case one will be allocated and a new stream will be
20 * instanciated). Returns < 0 on error.
Willy Tarreau53a47662017-08-28 10:53:00 +020021 */
22static int mux_pt_init(struct connection *conn)
23{
Olivier Houchard9aaf7782017-09-13 18:30:23 +020024 struct conn_stream *cs = conn->mux_ctx;
25
26 if (!cs) {
27 cs = cs_new(conn);
28 if (!cs)
29 goto fail;
30
31 if (stream_create_from_cs(cs) < 0)
32 goto fail_free;
33
34 conn->mux_ctx = cs;
35 }
Willy Tarreau53a47662017-08-28 10:53:00 +020036 return 0;
Olivier Houchard9aaf7782017-09-13 18:30:23 +020037
38 fail_free:
39 cs_free(cs);
40 fail:
41 return -1;
Willy Tarreau53a47662017-08-28 10:53:00 +020042}
43
44/* callback to be used by default for the pass-through mux. It calls the data
45 * layer wake() callback if it is set otherwise returns 0.
46 */
47static int mux_pt_wake(struct connection *conn)
48{
Olivier Houchard9aaf7782017-09-13 18:30:23 +020049 struct conn_stream *cs = conn->mux_ctx;
50 int ret;
51
52 ret = cs->data_cb->wake ? cs->data_cb->wake(cs) : 0;
53
Olivier Houchard7fc96d52017-11-23 18:25:47 +010054 /* If we had early data, and we're done with the handshake
55 * then whe know the data are safe, and we can remove the flag.
56 */
57 if ((conn->flags & (CO_FL_EARLY_DATA | CO_FL_EARLY_SSL_HS | CO_FL_HANDSHAKE)) ==
58 CO_FL_EARLY_DATA)
59 conn->flags &= ~CO_FL_EARLY_DATA;
Willy Tarreaued339a32017-11-03 15:55:24 +010060 if (ret >= 0)
61 cs_update_mux_polling(cs);
62 return ret;
Willy Tarreau53a47662017-08-28 10:53:00 +020063}
64
Olivier Houchard7a3f0df2017-09-13 18:30:23 +020065/* callback used to update the mux's polling flags after changing a cs' status.
66 * The caller (cs_mux_update_poll) will take care of propagating any changes to
67 * the transport layer.
68 */
69static void mux_pt_update_poll(struct conn_stream *cs)
70{
71 struct connection *conn = cs->conn;
72 int flags = 0;
73
74 conn_refresh_polling_flags(conn);
75
76 if (cs->flags & CS_FL_DATA_RD_ENA)
77 flags |= CO_FL_XPRT_RD_ENA;
78 if (cs->flags & CS_FL_DATA_WR_ENA)
79 flags |= CO_FL_XPRT_WR_ENA;
80
81 conn->flags = (conn->flags & ~(CO_FL_XPRT_RD_ENA | CO_FL_XPRT_WR_ENA)) | flags;
82 conn_cond_update_xprt_polling(conn);
83}
84
Willy Tarreau53a47662017-08-28 10:53:00 +020085/* callback to be used by default for the pass-through mux. It simply calls the
86 * data layer recv() callback much must be set.
87 */
88static void mux_pt_recv(struct connection *conn)
89{
Olivier Houchard9aaf7782017-09-13 18:30:23 +020090 struct conn_stream *cs = conn->mux_ctx;
91
Willy Tarreau4ff3b892017-10-16 15:17:17 +020092 if (conn->flags & CO_FL_ERROR)
93 cs->flags |= CS_FL_ERROR;
Olivier Houchard9aaf7782017-09-13 18:30:23 +020094 if (conn_xprt_read0_pending(conn))
95 cs->flags |= CS_FL_EOS;
96 cs->data_cb->recv(cs);
97 cs_update_mux_polling(cs);
Willy Tarreau53a47662017-08-28 10:53:00 +020098}
99
Olivier Houchard7a3f0df2017-09-13 18:30:23 +0200100/*
101 * Attach a new stream to a connection
102 * (Used for outgoing connections)
103 */
104static struct conn_stream *mux_pt_attach(struct connection *conn)
105{
106 return NULL;
107}
108
109/*
Willy Tarreau2c52a2b2017-10-08 11:00:17 +0200110 * Detach the stream from the connection and possibly release the connection.
Olivier Houchard7a3f0df2017-09-13 18:30:23 +0200111 */
112static void mux_pt_detach(struct conn_stream *cs)
113{
Willy Tarreau2c52a2b2017-10-08 11:00:17 +0200114 struct connection *conn = cs->conn;
115
116 LIST_DEL(&conn->list);
117 conn_stop_tracking(conn);
118 conn_full_close(conn);
Willy Tarreau436d3332017-10-08 11:16:46 +0200119 if (conn->destroy_cb)
120 conn->destroy_cb(conn);
Willy Tarreau2c52a2b2017-10-08 11:00:17 +0200121 conn_free(conn);
Olivier Houchard7a3f0df2017-09-13 18:30:23 +0200122}
123
Willy Tarreauecdb3fe2017-10-05 15:25:48 +0200124static void mux_pt_shutr(struct conn_stream *cs, enum cs_shr_mode mode)
Olivier Houchard7a3f0df2017-09-13 18:30:23 +0200125{
Willy Tarreau4b795242017-10-05 18:47:38 +0200126 if (cs->flags & CS_FL_SHR)
127 return;
Olivier Houchard7a3f0df2017-09-13 18:30:23 +0200128 if (conn_xprt_ready(cs->conn) && cs->conn->xprt->shutr)
Willy Tarreauecdb3fe2017-10-05 15:25:48 +0200129 cs->conn->xprt->shutr(cs->conn, (mode == CS_SHR_DRAIN));
Willy Tarreau4b795242017-10-05 18:47:38 +0200130 if (cs->flags & CS_FL_SHW)
131 conn_full_close(cs->conn);
Olivier Houchard7a3f0df2017-09-13 18:30:23 +0200132}
133
Willy Tarreauecdb3fe2017-10-05 15:25:48 +0200134static void mux_pt_shutw(struct conn_stream *cs, enum cs_shw_mode mode)
Olivier Houchard7a3f0df2017-09-13 18:30:23 +0200135{
Willy Tarreau4b795242017-10-05 18:47:38 +0200136 if (cs->flags & CS_FL_SHW)
137 return;
Olivier Houchard7a3f0df2017-09-13 18:30:23 +0200138 if (conn_xprt_ready(cs->conn) && cs->conn->xprt->shutw)
Willy Tarreauecdb3fe2017-10-05 15:25:48 +0200139 cs->conn->xprt->shutw(cs->conn, (mode == CS_SHW_NORMAL));
Willy Tarreau4b795242017-10-05 18:47:38 +0200140 if (!(cs->flags & CS_FL_SHR))
Willy Tarreaua48c1412017-12-22 18:46:33 +0100141 conn_sock_shutw(cs->conn, (mode == CS_SHW_NORMAL));
Willy Tarreau4b795242017-10-05 18:47:38 +0200142 else
143 conn_full_close(cs->conn);
Olivier Houchard7a3f0df2017-09-13 18:30:23 +0200144}
145
146/*
147 * Called from the upper layer, to get more data
148 */
Willy Tarreau7f3225f2018-06-19 06:15:17 +0200149static size_t mux_pt_rcv_buf(struct conn_stream *cs, struct buffer *buf, size_t count, int flags)
Olivier Houchard7a3f0df2017-09-13 18:30:23 +0200150{
Willy Tarreaud9cf5402018-07-18 11:29:06 +0200151 size_t ret;
Olivier Houchard7a3f0df2017-09-13 18:30:23 +0200152
Willy Tarreau7f3225f2018-06-19 06:15:17 +0200153 ret = cs->conn->xprt->rcv_buf(cs->conn, buf, count, flags);
Olivier Houchard7a3f0df2017-09-13 18:30:23 +0200154 if (conn_xprt_read0_pending(cs->conn))
155 cs->flags |= CS_FL_EOS;
Willy Tarreau4ff3b892017-10-16 15:17:17 +0200156 if (cs->conn->flags & CO_FL_ERROR)
157 cs->flags |= CS_FL_ERROR;
Willy Tarreaud9cf5402018-07-18 11:29:06 +0200158 return ret;
Olivier Houchard7a3f0df2017-09-13 18:30:23 +0200159}
160
161/* Called from the upper layer, to send data */
Christopher Fauletd44a9b32018-07-27 11:59:41 +0200162static size_t mux_pt_snd_buf(struct conn_stream *cs, struct buffer *buf, size_t count, int flags)
Olivier Houchard7a3f0df2017-09-13 18:30:23 +0200163{
Christopher Fauletd44a9b32018-07-27 11:59:41 +0200164 size_t ret = cs->conn->xprt->snd_buf(cs->conn, buf, count, flags);
165
166 if (ret > 0)
167 b_del(buf, ret);
168 return ret;
Olivier Houchard7a3f0df2017-09-13 18:30:23 +0200169}
170
Olivier Houchard6ff20392018-07-17 18:46:31 +0200171/* Called from the upper layer, to subscribe to events */
172static int mux_pt_subscribe(struct conn_stream *cs, int event_type, void *param)
173{
174 return (cs->conn->xprt->subscribe(cs->conn, event_type, param));
175}
176
Olivier Houchard7da120b2017-11-01 13:55:10 +0100177#if defined(CONFIG_HAP_LINUX_SPLICE)
Olivier Houchard7a3f0df2017-09-13 18:30:23 +0200178/* Send and get, using splicing */
179static int mux_pt_rcv_pipe(struct conn_stream *cs, struct pipe *pipe, unsigned int count)
180{
181 int ret;
182
183 ret = cs->conn->xprt->rcv_pipe(cs->conn, pipe, count);
184 if (conn_xprt_read0_pending(cs->conn))
185 cs->flags |= CS_FL_EOS;
Willy Tarreau4ff3b892017-10-16 15:17:17 +0200186 if (cs->conn->flags & CO_FL_ERROR)
187 cs->flags |= CS_FL_ERROR;
Olivier Houchard7a3f0df2017-09-13 18:30:23 +0200188 return (ret);
189}
190
191static int mux_pt_snd_pipe(struct conn_stream *cs, struct pipe *pipe)
192{
193 return (cs->conn->xprt->snd_pipe(cs->conn, pipe));
194}
Olivier Houchard7da120b2017-11-01 13:55:10 +0100195#endif
Olivier Houchard7a3f0df2017-09-13 18:30:23 +0200196
Willy Tarreau53a47662017-08-28 10:53:00 +0200197/* The mux operations */
198const struct mux_ops mux_pt_ops = {
199 .init = mux_pt_init,
200 .recv = mux_pt_recv,
Willy Tarreau53a47662017-08-28 10:53:00 +0200201 .wake = mux_pt_wake,
Olivier Houchard7a3f0df2017-09-13 18:30:23 +0200202 .update_poll = mux_pt_update_poll,
203 .rcv_buf = mux_pt_rcv_buf,
204 .snd_buf = mux_pt_snd_buf,
Olivier Houchard6ff20392018-07-17 18:46:31 +0200205 .subscribe = mux_pt_subscribe,
Olivier Houchard7a3f0df2017-09-13 18:30:23 +0200206#if defined(CONFIG_HAP_LINUX_SPLICE)
207 .rcv_pipe = mux_pt_rcv_pipe,
208 .snd_pipe = mux_pt_snd_pipe,
209#endif
210 .attach = mux_pt_attach,
211 .detach = mux_pt_detach,
212 .shutr = mux_pt_shutr,
213 .shutw = mux_pt_shutw,
Willy Tarreau28f1cb92017-12-20 16:14:44 +0100214 .flags = MX_FL_NONE,
Willy Tarreau53a47662017-08-28 10:53:00 +0200215 .name = "PASS",
216};
Willy Tarreauf6490822017-09-21 19:43:21 +0200217
Christopher Faulet32f61c02018-04-10 14:33:41 +0200218/* PROT selection : default mux has empty name */
219static struct mux_proto_list mux_proto_pt =
220 { .token = IST(""), .mode = PROTO_MODE_ANY, .side = PROTO_SIDE_BOTH, .mux = &mux_pt_ops };
Willy Tarreauf6490822017-09-21 19:43:21 +0200221
222__attribute__((constructor))
223static void __mux_pt_init(void)
224{
Christopher Faulet32f61c02018-04-10 14:33:41 +0200225 register_mux_proto(&mux_proto_pt);
Willy Tarreauf6490822017-09-21 19:43:21 +0200226}