blob: d44402e86aa823afbbaaa9f7c3023de139440223 [file] [log] [blame]
Willy Tarreau59f98392012-07-06 14:13:49 +02001/*
2 * Connection management functions
3 *
4 * Copyright 2000-2012 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
Willy Tarreaue1e4a612012-10-05 00:10:55 +020013#include <errno.h>
14
Willy Tarreau4c7e4b72020-05-27 12:58:42 +020015#include <haproxy/api.h>
Willy Tarreau6be78492020-06-05 00:00:29 +020016#include <haproxy/cfgparse.h>
Willy Tarreau7ea393d2020-06-04 18:02:10 +020017#include <haproxy/connection.h>
Willy Tarreaub2551052020-06-09 09:07:15 +020018#include <haproxy/fd.h>
Willy Tarreau762d7a52020-06-04 11:23:07 +020019#include <haproxy/frontend.h>
Willy Tarreaub2551052020-06-09 09:07:15 +020020#include <haproxy/hash.h>
Willy Tarreau36979d92020-06-05 17:27:29 +020021#include <haproxy/log-t.h>
Willy Tarreau7a00efb2020-06-02 17:02:59 +020022#include <haproxy/namespace.h>
Willy Tarreau6131d6a2020-06-02 16:48:09 +020023#include <haproxy/net_helper.h>
Willy Tarreaufc774542020-06-04 17:31:04 +020024#include <haproxy/proto_tcp.h>
Willy Tarreaue6ce10b2020-06-04 15:33:47 +020025#include <haproxy/sample.h>
Willy Tarreau209108d2020-06-04 20:30:20 +020026#include <haproxy/ssl_sock.h>
Willy Tarreau5e539c92020-06-04 20:45:39 +020027#include <haproxy/stream_interface.h>
Emeric Brun46591952012-05-18 15:47:34 +020028
Alexander Liu2a54bb72019-05-22 19:44:48 +080029
Willy Tarreau8ceae722018-11-26 11:58:30 +010030DECLARE_POOL(pool_head_connection, "connection", sizeof(struct connection));
31DECLARE_POOL(pool_head_connstream, "conn_stream", sizeof(struct conn_stream));
Willy Tarreauff5d57b2019-07-17 18:37:02 +020032DECLARE_POOL(pool_head_sockaddr, "sockaddr", sizeof(struct sockaddr_storage));
Geoff Simmons7185b782019-08-27 18:31:16 +020033DECLARE_POOL(pool_head_authority, "authority", PP2_AUTHORITY_MAX);
Willy Tarreau8ceae722018-11-26 11:58:30 +010034
Willy Tarreau4d82bf52020-06-28 00:19:17 +020035struct idle_conns idle_conns[MAX_THREADS] = { };
Willy Tarreau13e14102016-12-22 20:25:26 +010036struct xprt_ops *registered_xprt[XPRT_ENTRIES] = { NULL, };
Willy Tarreauf2943dc2012-10-26 20:10:28 +020037
Christopher Faulet32f61c02018-04-10 14:33:41 +020038/* List head of all known muxes for PROTO */
39struct mux_proto_list mux_proto_list = {
40 .list = LIST_HEAD_INIT(mux_proto_list.list)
Willy Tarreau2386be62017-09-21 19:40:52 +020041};
42
Willy Tarreau119e50e2020-05-22 13:53:29 +020043/* disables sending of proxy-protocol-v2's LOCAL command */
44static int pp2_never_send_local;
45
Olivier Houchard477902b2020-01-22 18:08:48 +010046int conn_create_mux(struct connection *conn)
47{
Olivier Houchard477902b2020-01-22 18:08:48 +010048 if (conn_is_back(conn)) {
49 struct server *srv;
50 struct conn_stream *cs = conn->ctx;
Christopher Faulet14cd3162020-04-16 14:50:06 +020051 struct session *sess = conn->owner;
Olivier Houchard477902b2020-01-22 18:08:48 +010052
53 if (conn->flags & CO_FL_ERROR)
54 goto fail;
Olivier Houcharda8a415d2020-01-23 13:15:14 +010055
Christopher Faulet14cd3162020-04-16 14:50:06 +020056 if (sess && obj_type(sess->origin) == OBJ_TYPE_CHECK) {
Willy Tarreau38b4d2e2020-11-20 17:08:15 +010057 if (conn_install_mux_chk(conn, conn->ctx, sess) < 0)
Christopher Faulet14cd3162020-04-16 14:50:06 +020058 goto fail;
59 }
Willy Tarreau38b4d2e2020-11-20 17:08:15 +010060 else if (conn_install_mux_be(conn, conn->ctx, sess) < 0)
Olivier Houchard477902b2020-01-22 18:08:48 +010061 goto fail;
62 srv = objt_server(conn->target);
Christopher Faulet08016ab2020-07-01 16:10:06 +020063
64 /* If we're doing http-reuse always, and the connection is not
65 * private with available streams (an http2 connection), add it
66 * to the available list, so that others can use it right
67 * away. If the connection is private, add it in the session
68 * server list.
69 */
Christopher Faulet2883fcf2020-07-01 14:59:43 +020070 if (srv && ((srv->proxy->options & PR_O_REUSE_MASK) == PR_O_REUSE_ALWS) &&
Christopher Fauletaa278532020-06-30 14:47:46 +020071 !(conn->flags & CO_FL_PRIVATE) && conn->mux->avail_streams(conn) > 0)
Olivier Houchardf0d4dff2020-03-06 18:12:03 +010072 LIST_ADDQ(&srv->available_conns[tid], mt_list_to_list(&conn->list));
Christopher Faulet08016ab2020-07-01 16:10:06 +020073 else if (conn->flags & CO_FL_PRIVATE) {
Ilya Shipitsin6b79f382020-07-23 00:32:55 +050074 /* If it fail now, the same will be done in mux->detach() callback */
Willy Tarreau38b4d2e2020-11-20 17:08:15 +010075 session_add_conn(sess, conn, conn->target);
Christopher Faulet08016ab2020-07-01 16:10:06 +020076 }
Olivier Houchard477902b2020-01-22 18:08:48 +010077 return 0;
78fail:
79 /* let the upper layer know the connection failed */
80 cs->data_cb->wake(cs);
81 return -1;
82 } else
83 return conn_complete_session(conn);
84
85}
86
Willy Tarreauff3e6482015-03-12 23:56:52 +010087/* Send a message over an established connection. It makes use of send() and
88 * returns the same return code and errno. If the socket layer is not ready yet
89 * then -1 is returned and ENOTSOCK is set into errno. If the fd is not marked
90 * as ready, or if EAGAIN or ENOTCONN is returned, then we return 0. It returns
91 * EMSGSIZE if called with a zero length message. The purpose is to simplify
92 * some rare attempts to directly write on the socket from above the connection
93 * (typically send_proxy). In case of EAGAIN, the fd is marked as "cant_send".
94 * It automatically retries on EINTR. Other errors cause the connection to be
95 * marked as in error state. It takes similar arguments as send() except the
Willy Tarreau827fee72020-12-11 15:26:55 +010096 * first one which is the connection instead of the file descriptor. <flags>
97 * only support CO_SFL_MSG_MORE.
Willy Tarreauff3e6482015-03-12 23:56:52 +010098 */
Willy Tarreau827fee72020-12-11 15:26:55 +010099int conn_ctrl_send(struct connection *conn, const void *buf, int len, int flags)
Willy Tarreauff3e6482015-03-12 23:56:52 +0100100{
Willy Tarreau827fee72020-12-11 15:26:55 +0100101 const struct buffer buffer = b_make((char*)buf, len, 0, len);
102 const struct xprt_ops *xprt = xprt_get(XPRT_RAW);
Willy Tarreauff3e6482015-03-12 23:56:52 +0100103 int ret;
104
105 ret = -1;
106 errno = ENOTSOCK;
107
108 if (conn->flags & CO_FL_SOCK_WR_SH)
109 goto fail;
110
111 if (!conn_ctrl_ready(conn))
112 goto fail;
113
114 errno = EMSGSIZE;
115 if (!len)
116 goto fail;
117
Willy Tarreau827fee72020-12-11 15:26:55 +0100118 /* snd_buf() already takes care of updating conn->flags and handling
119 * the FD polling status.
120 */
121 ret = xprt->snd_buf(conn, NULL, &buffer, buffer.data, flags);
122 if (conn->flags & CO_FL_ERROR)
123 ret = -1;
124 return ret;
Willy Tarreauff3e6482015-03-12 23:56:52 +0100125 fail:
126 conn->flags |= CO_FL_SOCK_RD_SH | CO_FL_SOCK_WR_SH | CO_FL_ERROR;
127 return ret;
128}
129
Willy Tarreauee1a6fc2020-01-17 07:52:13 +0100130/* Called from the upper layer, to subscribe <es> to events <event_type>. The
131 * event subscriber <es> is not allowed to change from a previous call as long
132 * as at least one event is still subscribed. The <event_type> must only be a
133 * combination of SUB_RETRY_RECV and SUB_RETRY_SEND. It always returns 0.
134 */
135int conn_unsubscribe(struct connection *conn, void *xprt_ctx, int event_type, struct wait_event *es)
Olivier Houchard83a0cd82018-09-28 17:57:58 +0200136{
Willy Tarreau7872d1f2020-01-10 07:06:05 +0100137 BUG_ON(event_type & ~(SUB_RETRY_SEND|SUB_RETRY_RECV));
Willy Tarreauee1a6fc2020-01-17 07:52:13 +0100138 BUG_ON(conn->subs && conn->subs != es);
Willy Tarreau7872d1f2020-01-10 07:06:05 +0100139
Willy Tarreauee1a6fc2020-01-17 07:52:13 +0100140 es->events &= ~event_type;
141 if (!es->events)
Willy Tarreau7872d1f2020-01-10 07:06:05 +0100142 conn->subs = NULL;
143
Willy Tarreaud1d14c32020-02-21 10:34:19 +0100144 if (conn_ctrl_ready(conn)) {
145 if (event_type & SUB_RETRY_RECV)
146 fd_stop_recv(conn->handle.fd);
Willy Tarreau7872d1f2020-01-10 07:06:05 +0100147
Willy Tarreaud1d14c32020-02-21 10:34:19 +0100148 if (event_type & SUB_RETRY_SEND)
149 fd_stop_send(conn->handle.fd);
150 }
Olivier Houchard83a0cd82018-09-28 17:57:58 +0200151 return 0;
152}
153
Willy Tarreau7e59c0a2020-02-28 14:24:49 +0100154/* Called from the upper layer, to subscribe <es> to events <event_type>.
155 * The <es> struct is not allowed to differ from the one passed during a
156 * previous call to subscribe(). If the FD is ready, the wait_event is
157 * immediately woken up and the subcription is cancelled. It always
158 * returns zero.
Willy Tarreauee1a6fc2020-01-17 07:52:13 +0100159 */
160int conn_subscribe(struct connection *conn, void *xprt_ctx, int event_type, struct wait_event *es)
Olivier Houchard6ff20392018-07-17 18:46:31 +0200161{
Willy Tarreau7872d1f2020-01-10 07:06:05 +0100162 BUG_ON(event_type & ~(SUB_RETRY_SEND|SUB_RETRY_RECV));
Willy Tarreauee1a6fc2020-01-17 07:52:13 +0100163 BUG_ON(conn->subs && conn->subs != es);
Willy Tarreau7872d1f2020-01-10 07:06:05 +0100164
Willy Tarreau7e59c0a2020-02-28 14:24:49 +0100165 if (conn->subs && (conn->subs->events & event_type) == event_type)
166 return 0;
167
Willy Tarreauee1a6fc2020-01-17 07:52:13 +0100168 conn->subs = es;
169 es->events |= event_type;
Willy Tarreau7872d1f2020-01-10 07:06:05 +0100170
Willy Tarreaud1d14c32020-02-21 10:34:19 +0100171 if (conn_ctrl_ready(conn)) {
Willy Tarreau7e59c0a2020-02-28 14:24:49 +0100172 if (event_type & SUB_RETRY_RECV) {
173 if (fd_recv_ready(conn->handle.fd)) {
174 tasklet_wakeup(es->tasklet);
175 es->events &= ~SUB_RETRY_RECV;
176 if (!es->events)
177 conn->subs = NULL;
178 }
179 else
180 fd_want_recv(conn->handle.fd);
181 }
Willy Tarreau7872d1f2020-01-10 07:06:05 +0100182
Willy Tarreau7e59c0a2020-02-28 14:24:49 +0100183 if (event_type & SUB_RETRY_SEND) {
184 if (fd_send_ready(conn->handle.fd)) {
185 tasklet_wakeup(es->tasklet);
186 es->events &= ~SUB_RETRY_SEND;
187 if (!es->events)
188 conn->subs = NULL;
189 }
190 else
191 fd_want_send(conn->handle.fd);
192 }
Willy Tarreaud1d14c32020-02-21 10:34:19 +0100193 }
Olivier Houchard83a0cd82018-09-28 17:57:58 +0200194 return 0;
Olivier Houchard6ff20392018-07-17 18:46:31 +0200195}
196
Willy Tarreaud85c4852015-03-13 00:40:28 +0100197/* Drains possibly pending incoming data on the file descriptor attached to the
198 * connection and update the connection's flags accordingly. This is used to
199 * know whether we need to disable lingering on close. Returns non-zero if it
200 * is safe to close without disabling lingering, otherwise zero. The SOCK_RD_SH
201 * flag may also be updated if the incoming shutdown was reported by the drain()
202 * function.
203 */
204int conn_sock_drain(struct connection *conn)
205{
Willy Tarreaue215bba2018-08-24 14:31:53 +0200206 int turns = 2;
207 int len;
208
Willy Tarreaud85c4852015-03-13 00:40:28 +0100209 if (!conn_ctrl_ready(conn))
210 return 1;
211
212 if (conn->flags & (CO_FL_ERROR | CO_FL_SOCK_RD_SH))
213 return 1;
214
Willy Tarreaue215bba2018-08-24 14:31:53 +0200215 if (fdtab[conn->handle.fd].ev & (FD_POLL_ERR|FD_POLL_HUP))
216 goto shut;
Willy Tarreaud85c4852015-03-13 00:40:28 +0100217
Willy Tarreaue215bba2018-08-24 14:31:53 +0200218 if (!fd_recv_ready(conn->handle.fd))
219 return 0;
Willy Tarreaud85c4852015-03-13 00:40:28 +0100220
Willy Tarreaue215bba2018-08-24 14:31:53 +0200221 /* no drain function defined, use the generic one */
222
223 while (turns) {
224#ifdef MSG_TRUNC_CLEARS_INPUT
225 len = recv(conn->handle.fd, NULL, INT_MAX, MSG_DONTWAIT | MSG_NOSIGNAL | MSG_TRUNC);
226 if (len == -1 && errno == EFAULT)
227#endif
228 len = recv(conn->handle.fd, trash.area, trash.size,
229 MSG_DONTWAIT | MSG_NOSIGNAL);
230
231 if (len == 0)
232 goto shut;
233
234 if (len < 0) {
235 if (errno == EAGAIN) {
236 /* connection not closed yet */
237 fd_cant_recv(conn->handle.fd);
238 break;
239 }
240 if (errno == EINTR) /* oops, try again */
241 continue;
242 /* other errors indicate a dead connection, fine. */
243 goto shut;
244 }
245 /* OK we read some data, let's try again once */
246 turns--;
Willy Tarreaud85c4852015-03-13 00:40:28 +0100247 }
248
Willy Tarreaue215bba2018-08-24 14:31:53 +0200249 /* some data are still present, give up */
250 return 0;
251
252 shut:
253 /* we're certain the connection was shut down */
254 fdtab[conn->handle.fd].linger_risk = 0;
Willy Tarreaud85c4852015-03-13 00:40:28 +0100255 conn->flags |= CO_FL_SOCK_RD_SH;
256 return 1;
257}
258
KOVACS Krisztianb3e54fe2014-11-17 15:11:45 +0100259/*
260 * Get data length from tlv
261 */
Tim Duesterhusba837ec2020-03-05 23:11:02 +0100262static inline size_t get_tlv_length(const struct tlv *src)
KOVACS Krisztianb3e54fe2014-11-17 15:11:45 +0100263{
264 return (src->length_hi << 8) | src->length_lo;
265}
266
Willy Tarreaue1e4a612012-10-05 00:10:55 +0200267/* This handshake handler waits a PROXY protocol header at the beginning of the
268 * raw data stream. The header looks like this :
269 *
270 * "PROXY" <SP> PROTO <SP> SRC3 <SP> DST3 <SP> SRC4 <SP> <DST4> "\r\n"
271 *
272 * There must be exactly one space between each field. Fields are :
273 * - PROTO : layer 4 protocol, which must be "TCP4" or "TCP6".
274 * - SRC3 : layer 3 (eg: IP) source address in standard text form
275 * - DST3 : layer 3 (eg: IP) destination address in standard text form
276 * - SRC4 : layer 4 (eg: TCP port) source address in standard text form
277 * - DST4 : layer 4 (eg: TCP port) destination address in standard text form
278 *
279 * This line MUST be at the beginning of the buffer and MUST NOT wrap.
280 *
281 * The header line is small and in all cases smaller than the smallest normal
282 * TCP MSS. So it MUST always be delivered as one segment, which ensures we
283 * can safely use MSG_PEEK and avoid buffering.
284 *
285 * Once the data is fetched, the values are set in the connection's address
286 * fields, and data are removed from the socket's buffer. The function returns
287 * zero if it needs to wait for more data or if it fails, or 1 if it completed
288 * and removed itself.
289 */
290int conn_recv_proxy(struct connection *conn, int flag)
291{
292 char *line, *end;
Willy Tarreau77992672014-06-14 11:06:17 +0200293 struct proxy_hdr_v2 *hdr_v2;
294 const char v2sig[] = PP2_SIGNATURE;
Tim Duesterhus488ee7f2020-03-05 22:55:20 +0100295 size_t total_v2_len;
Tim Duesterhusba837ec2020-03-05 23:11:02 +0100296 size_t tlv_offset = 0;
Willy Tarreaub406b872018-08-22 05:20:32 +0200297 int ret;
Willy Tarreaue1e4a612012-10-05 00:10:55 +0200298
Willy Tarreau3c728722014-01-23 13:50:42 +0100299 if (!conn_ctrl_ready(conn))
Willy Tarreauf79c8172013-10-21 16:30:56 +0200300 goto fail;
301
Willy Tarreau9b7587a2020-10-15 07:32:10 +0200302 if (!sockaddr_alloc(&conn->src, NULL, 0) || !sockaddr_alloc(&conn->dst, NULL, 0))
Willy Tarreauca79f592019-07-17 19:04:47 +0200303 goto fail;
304
Willy Tarreau585744b2017-08-24 14:31:19 +0200305 if (!fd_recv_ready(conn->handle.fd))
Willy Tarreau6499b9d2019-06-03 08:17:30 +0200306 goto not_ready;
Willy Tarreaufd803bb2014-01-20 15:13:07 +0100307
Willy Tarreau157788c2020-02-11 10:08:05 +0100308 while (1) {
Willy Tarreaub406b872018-08-22 05:20:32 +0200309 ret = recv(conn->handle.fd, trash.area, trash.size, MSG_PEEK);
310 if (ret < 0) {
Willy Tarreaue1e4a612012-10-05 00:10:55 +0200311 if (errno == EINTR)
312 continue;
313 if (errno == EAGAIN) {
Willy Tarreau585744b2017-08-24 14:31:19 +0200314 fd_cant_recv(conn->handle.fd);
Willy Tarreau6499b9d2019-06-03 08:17:30 +0200315 goto not_ready;
Willy Tarreaue1e4a612012-10-05 00:10:55 +0200316 }
Willy Tarreau8e3bf692012-12-03 15:41:18 +0100317 goto recv_abort;
Willy Tarreaue1e4a612012-10-05 00:10:55 +0200318 }
Willy Tarreaub406b872018-08-22 05:20:32 +0200319 trash.data = ret;
Willy Tarreau157788c2020-02-11 10:08:05 +0100320 break;
321 }
Willy Tarreaue1e4a612012-10-05 00:10:55 +0200322
Willy Tarreau843b7cb2018-07-13 10:54:26 +0200323 if (!trash.data) {
Willy Tarreau8e3bf692012-12-03 15:41:18 +0100324 /* client shutdown */
325 conn->err_code = CO_ER_PRX_EMPTY;
326 goto fail;
327 }
328
Willy Tarreauc192b0a2020-01-23 09:11:58 +0100329 conn->flags &= ~CO_FL_WAIT_L4_CONN;
330
Willy Tarreau843b7cb2018-07-13 10:54:26 +0200331 if (trash.data < 6)
Willy Tarreaue1e4a612012-10-05 00:10:55 +0200332 goto missing;
333
Willy Tarreau843b7cb2018-07-13 10:54:26 +0200334 line = trash.area;
335 end = trash.area + trash.data;
Willy Tarreaue1e4a612012-10-05 00:10:55 +0200336
337 /* Decode a possible proxy request, fail early if it does not match */
Willy Tarreau77992672014-06-14 11:06:17 +0200338 if (strncmp(line, "PROXY ", 6) != 0)
339 goto not_v1;
Willy Tarreaue1e4a612012-10-05 00:10:55 +0200340
341 line += 6;
Willy Tarreau843b7cb2018-07-13 10:54:26 +0200342 if (trash.data < 9) /* shortest possible line */
Willy Tarreaue1e4a612012-10-05 00:10:55 +0200343 goto missing;
344
David CARLIER42ff05e2016-03-24 09:22:36 +0000345 if (memcmp(line, "TCP4 ", 5) == 0) {
Willy Tarreaue1e4a612012-10-05 00:10:55 +0200346 u32 src3, dst3, sport, dport;
347
348 line += 5;
349
350 src3 = inetaddr_host_lim_ret(line, end, &line);
351 if (line == end)
352 goto missing;
353 if (*line++ != ' ')
Willy Tarreau8e3bf692012-12-03 15:41:18 +0100354 goto bad_header;
Willy Tarreaue1e4a612012-10-05 00:10:55 +0200355
356 dst3 = inetaddr_host_lim_ret(line, end, &line);
357 if (line == end)
358 goto missing;
359 if (*line++ != ' ')
Willy Tarreau8e3bf692012-12-03 15:41:18 +0100360 goto bad_header;
Willy Tarreaue1e4a612012-10-05 00:10:55 +0200361
362 sport = read_uint((const char **)&line, end);
363 if (line == end)
364 goto missing;
365 if (*line++ != ' ')
Willy Tarreau8e3bf692012-12-03 15:41:18 +0100366 goto bad_header;
Willy Tarreaue1e4a612012-10-05 00:10:55 +0200367
368 dport = read_uint((const char **)&line, end);
369 if (line > end - 2)
370 goto missing;
371 if (*line++ != '\r')
Willy Tarreau8e3bf692012-12-03 15:41:18 +0100372 goto bad_header;
Willy Tarreaue1e4a612012-10-05 00:10:55 +0200373 if (*line++ != '\n')
Willy Tarreau8e3bf692012-12-03 15:41:18 +0100374 goto bad_header;
Willy Tarreaue1e4a612012-10-05 00:10:55 +0200375
376 /* update the session's addresses and mark them set */
Willy Tarreau226572f2019-07-17 14:46:00 +0200377 ((struct sockaddr_in *)conn->src)->sin_family = AF_INET;
378 ((struct sockaddr_in *)conn->src)->sin_addr.s_addr = htonl(src3);
379 ((struct sockaddr_in *)conn->src)->sin_port = htons(sport);
Willy Tarreaue1e4a612012-10-05 00:10:55 +0200380
Willy Tarreau226572f2019-07-17 14:46:00 +0200381 ((struct sockaddr_in *)conn->dst)->sin_family = AF_INET;
382 ((struct sockaddr_in *)conn->dst)->sin_addr.s_addr = htonl(dst3);
383 ((struct sockaddr_in *)conn->dst)->sin_port = htons(dport);
Willy Tarreaue1e4a612012-10-05 00:10:55 +0200384 conn->flags |= CO_FL_ADDR_FROM_SET | CO_FL_ADDR_TO_SET;
385 }
David CARLIER42ff05e2016-03-24 09:22:36 +0000386 else if (memcmp(line, "TCP6 ", 5) == 0) {
Willy Tarreaue1e4a612012-10-05 00:10:55 +0200387 u32 sport, dport;
388 char *src_s;
389 char *dst_s, *sport_s, *dport_s;
390 struct in6_addr src3, dst3;
391
392 line += 5;
393
394 src_s = line;
395 dst_s = sport_s = dport_s = NULL;
396 while (1) {
397 if (line > end - 2) {
398 goto missing;
399 }
400 else if (*line == '\r') {
401 *line = 0;
402 line++;
403 if (*line++ != '\n')
Willy Tarreau8e3bf692012-12-03 15:41:18 +0100404 goto bad_header;
Willy Tarreaue1e4a612012-10-05 00:10:55 +0200405 break;
406 }
407
408 if (*line == ' ') {
409 *line = 0;
410 if (!dst_s)
411 dst_s = line + 1;
412 else if (!sport_s)
413 sport_s = line + 1;
414 else if (!dport_s)
415 dport_s = line + 1;
416 }
417 line++;
418 }
419
420 if (!dst_s || !sport_s || !dport_s)
Willy Tarreau8e3bf692012-12-03 15:41:18 +0100421 goto bad_header;
Willy Tarreaue1e4a612012-10-05 00:10:55 +0200422
423 sport = read_uint((const char **)&sport_s,dport_s - 1);
424 if (*sport_s != 0)
Willy Tarreau8e3bf692012-12-03 15:41:18 +0100425 goto bad_header;
Willy Tarreaue1e4a612012-10-05 00:10:55 +0200426
427 dport = read_uint((const char **)&dport_s,line - 2);
428 if (*dport_s != 0)
Willy Tarreau8e3bf692012-12-03 15:41:18 +0100429 goto bad_header;
Willy Tarreaue1e4a612012-10-05 00:10:55 +0200430
431 if (inet_pton(AF_INET6, src_s, (void *)&src3) != 1)
Willy Tarreau8e3bf692012-12-03 15:41:18 +0100432 goto bad_header;
Willy Tarreaue1e4a612012-10-05 00:10:55 +0200433
434 if (inet_pton(AF_INET6, dst_s, (void *)&dst3) != 1)
Willy Tarreau8e3bf692012-12-03 15:41:18 +0100435 goto bad_header;
Willy Tarreaue1e4a612012-10-05 00:10:55 +0200436
437 /* update the session's addresses and mark them set */
Willy Tarreau226572f2019-07-17 14:46:00 +0200438 ((struct sockaddr_in6 *)conn->src)->sin6_family = AF_INET6;
439 memcpy(&((struct sockaddr_in6 *)conn->src)->sin6_addr, &src3, sizeof(struct in6_addr));
440 ((struct sockaddr_in6 *)conn->src)->sin6_port = htons(sport);
Willy Tarreaue1e4a612012-10-05 00:10:55 +0200441
Willy Tarreau226572f2019-07-17 14:46:00 +0200442 ((struct sockaddr_in6 *)conn->dst)->sin6_family = AF_INET6;
443 memcpy(&((struct sockaddr_in6 *)conn->dst)->sin6_addr, &dst3, sizeof(struct in6_addr));
444 ((struct sockaddr_in6 *)conn->dst)->sin6_port = htons(dport);
Willy Tarreaue1e4a612012-10-05 00:10:55 +0200445 conn->flags |= CO_FL_ADDR_FROM_SET | CO_FL_ADDR_TO_SET;
446 }
Willy Tarreau4c20d292014-06-14 11:41:36 +0200447 else if (memcmp(line, "UNKNOWN\r\n", 9) == 0) {
448 /* This can be a UNIX socket forwarded by an haproxy upstream */
449 line += 9;
450 }
Willy Tarreaue1e4a612012-10-05 00:10:55 +0200451 else {
Willy Tarreau4c20d292014-06-14 11:41:36 +0200452 /* The protocol does not match something known (TCP4/TCP6/UNKNOWN) */
Willy Tarreau8e3bf692012-12-03 15:41:18 +0100453 conn->err_code = CO_ER_PRX_BAD_PROTO;
Willy Tarreaue1e4a612012-10-05 00:10:55 +0200454 goto fail;
455 }
456
Willy Tarreau843b7cb2018-07-13 10:54:26 +0200457 trash.data = line - trash.area;
Willy Tarreau77992672014-06-14 11:06:17 +0200458 goto eat_header;
459
460 not_v1:
461 /* try PPv2 */
Willy Tarreau843b7cb2018-07-13 10:54:26 +0200462 if (trash.data < PP2_HEADER_LEN)
Willy Tarreau77992672014-06-14 11:06:17 +0200463 goto missing;
464
Willy Tarreau843b7cb2018-07-13 10:54:26 +0200465 hdr_v2 = (struct proxy_hdr_v2 *) trash.area;
Willy Tarreau77992672014-06-14 11:06:17 +0200466
467 if (memcmp(hdr_v2->sig, v2sig, PP2_SIGNATURE_LEN) != 0 ||
468 (hdr_v2->ver_cmd & PP2_VERSION_MASK) != PP2_VERSION) {
469 conn->err_code = CO_ER_PRX_NOT_HDR;
470 goto fail;
471 }
472
Tim Duesterhus488ee7f2020-03-05 22:55:20 +0100473 total_v2_len = PP2_HEADER_LEN + ntohs(hdr_v2->len);
474 if (trash.data < total_v2_len)
Willy Tarreau77992672014-06-14 11:06:17 +0200475 goto missing;
476
477 switch (hdr_v2->ver_cmd & PP2_CMD_MASK) {
478 case 0x01: /* PROXY command */
479 switch (hdr_v2->fam) {
480 case 0x11: /* TCPv4 */
KOVACS Krisztianefd3aa92014-11-19 10:53:20 +0100481 if (ntohs(hdr_v2->len) < PP2_ADDR_LEN_INET)
482 goto bad_header;
483
Willy Tarreau226572f2019-07-17 14:46:00 +0200484 ((struct sockaddr_in *)conn->src)->sin_family = AF_INET;
485 ((struct sockaddr_in *)conn->src)->sin_addr.s_addr = hdr_v2->addr.ip4.src_addr;
486 ((struct sockaddr_in *)conn->src)->sin_port = hdr_v2->addr.ip4.src_port;
487 ((struct sockaddr_in *)conn->dst)->sin_family = AF_INET;
488 ((struct sockaddr_in *)conn->dst)->sin_addr.s_addr = hdr_v2->addr.ip4.dst_addr;
489 ((struct sockaddr_in *)conn->dst)->sin_port = hdr_v2->addr.ip4.dst_port;
Willy Tarreau77992672014-06-14 11:06:17 +0200490 conn->flags |= CO_FL_ADDR_FROM_SET | CO_FL_ADDR_TO_SET;
KOVACS Krisztian7209c202015-07-03 14:09:10 +0200491 tlv_offset = PP2_HEADER_LEN + PP2_ADDR_LEN_INET;
Willy Tarreau77992672014-06-14 11:06:17 +0200492 break;
493 case 0x21: /* TCPv6 */
KOVACS Krisztianefd3aa92014-11-19 10:53:20 +0100494 if (ntohs(hdr_v2->len) < PP2_ADDR_LEN_INET6)
495 goto bad_header;
496
Willy Tarreau226572f2019-07-17 14:46:00 +0200497 ((struct sockaddr_in6 *)conn->src)->sin6_family = AF_INET6;
498 memcpy(&((struct sockaddr_in6 *)conn->src)->sin6_addr, hdr_v2->addr.ip6.src_addr, 16);
499 ((struct sockaddr_in6 *)conn->src)->sin6_port = hdr_v2->addr.ip6.src_port;
500 ((struct sockaddr_in6 *)conn->dst)->sin6_family = AF_INET6;
501 memcpy(&((struct sockaddr_in6 *)conn->dst)->sin6_addr, hdr_v2->addr.ip6.dst_addr, 16);
502 ((struct sockaddr_in6 *)conn->dst)->sin6_port = hdr_v2->addr.ip6.dst_port;
Willy Tarreau77992672014-06-14 11:06:17 +0200503 conn->flags |= CO_FL_ADDR_FROM_SET | CO_FL_ADDR_TO_SET;
KOVACS Krisztian7209c202015-07-03 14:09:10 +0200504 tlv_offset = PP2_HEADER_LEN + PP2_ADDR_LEN_INET6;
Willy Tarreau77992672014-06-14 11:06:17 +0200505 break;
506 }
KOVACS Krisztianb3e54fe2014-11-17 15:11:45 +0100507
508 /* TLV parsing */
Tim Duesterhus488ee7f2020-03-05 22:55:20 +0100509 while (tlv_offset < total_v2_len) {
510 struct tlv *tlv_packet;
Tim Duesterhusba837ec2020-03-05 23:11:02 +0100511 size_t tlv_len;
KOVACS Krisztianb3e54fe2014-11-17 15:11:45 +0100512
Tim Duesterhus488ee7f2020-03-05 22:55:20 +0100513 /* Verify that we have at least TLV_HEADER_SIZE bytes left */
514 if (tlv_offset + TLV_HEADER_SIZE > total_v2_len)
515 goto bad_header;
516
517 tlv_packet = (struct tlv *) &trash.area[tlv_offset];
518 tlv_len = get_tlv_length(tlv_packet);
519 tlv_offset += tlv_len + TLV_HEADER_SIZE;
520
521 /* Verify that the TLV length does not exceed the total PROXYv2 length */
522 if (tlv_offset > total_v2_len)
523 goto bad_header;
524
525 switch (tlv_packet->type) {
526 case PP2_TYPE_CRC32C: {
527 uint32_t n_crc32c;
528
529 /* Verify that this TLV is exactly 4 bytes long */
530 if (tlv_len != 4)
531 goto bad_header;
532
533 n_crc32c = read_n32(tlv_packet->value);
534 write_n32(tlv_packet->value, 0); // compute with CRC==0
535
536 if (hash_crc32c(trash.area, total_v2_len) != n_crc32c)
537 goto bad_header;
538 break;
539 }
Willy Tarreaue5733232019-05-22 19:24:06 +0200540#ifdef USE_NS
Tim Duesterhus488ee7f2020-03-05 22:55:20 +0100541 case PP2_TYPE_NETNS: {
542 const struct netns_entry *ns;
543
544 ns = netns_store_lookup((char*)tlv_packet->value, tlv_len);
545 if (ns)
546 conn->proxy_netns = ns;
547 break;
548 }
KOVACS Krisztianb3e54fe2014-11-17 15:11:45 +0100549#endif
Tim Duesterhus488ee7f2020-03-05 22:55:20 +0100550 case PP2_TYPE_AUTHORITY: {
551 if (tlv_len > PP2_AUTHORITY_MAX)
552 goto bad_header;
553 conn->proxy_authority = pool_alloc(pool_head_authority);
554 if (conn->proxy_authority == NULL)
555 goto fail;
556 memcpy(conn->proxy_authority, (const char *)tlv_packet->value, tlv_len);
557 conn->proxy_authority_len = tlv_len;
558 break;
559 }
Tim Duesterhusd1b15b62020-03-13 12:34:23 +0100560 case PP2_TYPE_UNIQUE_ID: {
561 const struct ist tlv = ist2((const char *)tlv_packet->value, tlv_len);
562
563 if (tlv.len > UNIQUEID_LEN)
564 goto bad_header;
Tim Duesterhus2b7f6c22020-03-14 13:07:05 +0100565 conn->proxy_unique_id = ist2(pool_alloc(pool_head_uniqueid), 0);
Tim Duesterhusd1b15b62020-03-13 12:34:23 +0100566 if (!isttest(conn->proxy_unique_id))
567 goto fail;
568 if (istcpy(&conn->proxy_unique_id, tlv, UNIQUEID_LEN) < 0) {
569 /* This is technically unreachable, because we verified above
570 * that the TLV value fits.
571 */
572 goto fail;
573 }
574 break;
575 }
Tim Duesterhus488ee7f2020-03-05 22:55:20 +0100576 default:
577 break;
KOVACS Krisztianb3e54fe2014-11-17 15:11:45 +0100578 }
579 }
580
Tim Duesterhus488ee7f2020-03-05 22:55:20 +0100581 /* Verify that the PROXYv2 header ends at a TLV boundary.
582 * This is technically unreachable, because the TLV parsing already
583 * verifies that a TLV does not exceed the total length and also
584 * that there is space for a TLV header.
585 */
586 if (tlv_offset != total_v2_len)
587 goto bad_header;
588
Willy Tarreau77992672014-06-14 11:06:17 +0200589 /* unsupported protocol, keep local connection address */
590 break;
591 case 0x00: /* LOCAL command */
592 /* keep local connection address for LOCAL */
593 break;
594 default:
595 goto bad_header; /* not a supported command */
596 }
597
Tim Duesterhus488ee7f2020-03-05 22:55:20 +0100598 trash.data = total_v2_len;
Willy Tarreau77992672014-06-14 11:06:17 +0200599 goto eat_header;
600
601 eat_header:
Willy Tarreaue1e4a612012-10-05 00:10:55 +0200602 /* remove the PROXY line from the request. For this we re-read the
603 * exact line at once. If we don't get the exact same result, we
604 * fail.
605 */
Willy Tarreau157788c2020-02-11 10:08:05 +0100606 while (1) {
Tim Duesterhusa8692f32020-03-13 12:34:25 +0100607 ssize_t len2 = recv(conn->handle.fd, trash.area, trash.data, 0);
608
Willy Tarreaue1e4a612012-10-05 00:10:55 +0200609 if (len2 < 0 && errno == EINTR)
610 continue;
Willy Tarreau843b7cb2018-07-13 10:54:26 +0200611 if (len2 != trash.data)
Willy Tarreau8e3bf692012-12-03 15:41:18 +0100612 goto recv_abort;
Willy Tarreau157788c2020-02-11 10:08:05 +0100613 break;
614 }
Willy Tarreaue1e4a612012-10-05 00:10:55 +0200615
616 conn->flags &= ~flag;
Emeric Brun4f603012017-01-05 15:11:44 +0100617 conn->flags |= CO_FL_RCVD_PROXY;
Willy Tarreaue1e4a612012-10-05 00:10:55 +0200618 return 1;
619
Willy Tarreau6499b9d2019-06-03 08:17:30 +0200620 not_ready:
Willy Tarreau6499b9d2019-06-03 08:17:30 +0200621 return 0;
622
Willy Tarreaue1e4a612012-10-05 00:10:55 +0200623 missing:
624 /* Missing data. Since we're using MSG_PEEK, we can only poll again if
625 * we have not read anything. Otherwise we need to fail because we won't
626 * be able to poll anymore.
627 */
Willy Tarreau8e3bf692012-12-03 15:41:18 +0100628 conn->err_code = CO_ER_PRX_TRUNCATED;
629 goto fail;
630
631 bad_header:
632 /* This is not a valid proxy protocol header */
633 conn->err_code = CO_ER_PRX_BAD_HDR;
634 goto fail;
635
636 recv_abort:
637 conn->err_code = CO_ER_PRX_ABORT;
Willy Tarreau26f4a042013-12-04 23:44:10 +0100638 conn->flags |= CO_FL_SOCK_RD_SH | CO_FL_SOCK_WR_SH;
Willy Tarreau8e3bf692012-12-03 15:41:18 +0100639 goto fail;
640
Willy Tarreaue1e4a612012-10-05 00:10:55 +0200641 fail:
Willy Tarreaue1e4a612012-10-05 00:10:55 +0200642 conn->flags |= CO_FL_ERROR;
Willy Tarreaue1e4a612012-10-05 00:10:55 +0200643 return 0;
644}
645
Bertrand Jacquin93b227d2016-06-04 15:11:10 +0100646/* This handshake handler waits a NetScaler Client IP insertion header
Bertrand Jacquin72fa1ec2017-12-12 01:17:23 +0000647 * at the beginning of the raw data stream. The header format is
648 * described in doc/netscaler-client-ip-insertion-protocol.txt
Bertrand Jacquin93b227d2016-06-04 15:11:10 +0100649 *
650 * This line MUST be at the beginning of the buffer and MUST NOT be
651 * fragmented.
652 *
653 * The header line is small and in all cases smaller than the smallest normal
654 * TCP MSS. So it MUST always be delivered as one segment, which ensures we
655 * can safely use MSG_PEEK and avoid buffering.
656 *
657 * Once the data is fetched, the values are set in the connection's address
658 * fields, and data are removed from the socket's buffer. The function returns
659 * zero if it needs to wait for more data or if it fails, or 1 if it completed
660 * and removed itself.
661 */
662int conn_recv_netscaler_cip(struct connection *conn, int flag)
663{
664 char *line;
Bertrand Jacquin7d668f92017-12-13 01:23:39 +0000665 uint32_t hdr_len;
Willy Tarreau0ca24aa2019-03-29 17:35:32 +0100666 uint8_t ip_ver;
Willy Tarreaub406b872018-08-22 05:20:32 +0200667 int ret;
Bertrand Jacquin93b227d2016-06-04 15:11:10 +0100668
Bertrand Jacquin93b227d2016-06-04 15:11:10 +0100669 if (!conn_ctrl_ready(conn))
670 goto fail;
671
Willy Tarreau9b7587a2020-10-15 07:32:10 +0200672 if (!sockaddr_alloc(&conn->src, NULL, 0) || !sockaddr_alloc(&conn->dst, NULL, 0))
Olivier Houchard1a9dbe52020-01-22 15:31:09 +0100673 goto fail;
674
Willy Tarreau585744b2017-08-24 14:31:19 +0200675 if (!fd_recv_ready(conn->handle.fd))
Willy Tarreau6499b9d2019-06-03 08:17:30 +0200676 goto not_ready;
Bertrand Jacquin93b227d2016-06-04 15:11:10 +0100677
Willy Tarreau157788c2020-02-11 10:08:05 +0100678 while (1) {
Willy Tarreaub406b872018-08-22 05:20:32 +0200679 ret = recv(conn->handle.fd, trash.area, trash.size, MSG_PEEK);
680 if (ret < 0) {
Bertrand Jacquin93b227d2016-06-04 15:11:10 +0100681 if (errno == EINTR)
682 continue;
683 if (errno == EAGAIN) {
Willy Tarreau585744b2017-08-24 14:31:19 +0200684 fd_cant_recv(conn->handle.fd);
Willy Tarreau6499b9d2019-06-03 08:17:30 +0200685 goto not_ready;
Bertrand Jacquin93b227d2016-06-04 15:11:10 +0100686 }
687 goto recv_abort;
688 }
Willy Tarreaub406b872018-08-22 05:20:32 +0200689 trash.data = ret;
Willy Tarreau157788c2020-02-11 10:08:05 +0100690 break;
691 }
Bertrand Jacquin93b227d2016-06-04 15:11:10 +0100692
Willy Tarreauc192b0a2020-01-23 09:11:58 +0100693 conn->flags &= ~CO_FL_WAIT_L4_CONN;
694
Willy Tarreau843b7cb2018-07-13 10:54:26 +0200695 if (!trash.data) {
Bertrand Jacquin93b227d2016-06-04 15:11:10 +0100696 /* client shutdown */
697 conn->err_code = CO_ER_CIP_EMPTY;
698 goto fail;
699 }
700
701 /* Fail if buffer length is not large enough to contain
Bertrand Jacquin72fa1ec2017-12-12 01:17:23 +0000702 * CIP magic, header length or
703 * CIP magic, CIP length, CIP type, header length */
Willy Tarreau843b7cb2018-07-13 10:54:26 +0200704 if (trash.data < 12)
Bertrand Jacquin93b227d2016-06-04 15:11:10 +0100705 goto missing;
706
Willy Tarreau843b7cb2018-07-13 10:54:26 +0200707 line = trash.area;
Bertrand Jacquin93b227d2016-06-04 15:11:10 +0100708
709 /* Decode a possible NetScaler Client IP request, fail early if
710 * it does not match */
Willy Tarreau1ac83af2020-02-25 10:06:49 +0100711 if (ntohl(read_u32(line)) != __objt_listener(conn->target)->bind_conf->ns_cip_magic)
Bertrand Jacquin93b227d2016-06-04 15:11:10 +0100712 goto bad_magic;
713
Bertrand Jacquin72fa1ec2017-12-12 01:17:23 +0000714 /* Legacy CIP protocol */
Willy Tarreau843b7cb2018-07-13 10:54:26 +0200715 if ((trash.area[8] & 0xD0) == 0x40) {
Willy Tarreau1ac83af2020-02-25 10:06:49 +0100716 hdr_len = ntohl(read_u32((line+4)));
Bertrand Jacquin72fa1ec2017-12-12 01:17:23 +0000717 line += 8;
718 }
719 /* Standard CIP protocol */
Willy Tarreau843b7cb2018-07-13 10:54:26 +0200720 else if (trash.area[8] == 0x00) {
Willy Tarreau1ac83af2020-02-25 10:06:49 +0100721 hdr_len = ntohs(read_u32((line+10)));
Bertrand Jacquin72fa1ec2017-12-12 01:17:23 +0000722 line += 12;
723 }
724 /* Unknown CIP protocol */
725 else {
726 conn->err_code = CO_ER_CIP_BAD_PROTO;
727 goto fail;
728 }
729
Bertrand Jacquin93b227d2016-06-04 15:11:10 +0100730 /* Fail if buffer length is not large enough to contain
Bertrand Jacquin72fa1ec2017-12-12 01:17:23 +0000731 * a minimal IP header */
Willy Tarreau843b7cb2018-07-13 10:54:26 +0200732 if (trash.data < 20)
Bertrand Jacquin93b227d2016-06-04 15:11:10 +0100733 goto missing;
734
Bertrand Jacquin93b227d2016-06-04 15:11:10 +0100735 /* Get IP version from the first four bits */
Willy Tarreau0ca24aa2019-03-29 17:35:32 +0100736 ip_ver = (*line & 0xf0) >> 4;
Bertrand Jacquin93b227d2016-06-04 15:11:10 +0100737
Willy Tarreau0ca24aa2019-03-29 17:35:32 +0100738 if (ip_ver == 4) {
Bertrand Jacquin93b227d2016-06-04 15:11:10 +0100739 struct ip *hdr_ip4;
David Carlier3015a2e2016-07-04 22:51:33 +0100740 struct my_tcphdr *hdr_tcp;
Bertrand Jacquin93b227d2016-06-04 15:11:10 +0100741
742 hdr_ip4 = (struct ip *)line;
743
Willy Tarreau843b7cb2018-07-13 10:54:26 +0200744 if (trash.data < 40 || trash.data < hdr_len) {
Bertrand Jacquin93b227d2016-06-04 15:11:10 +0100745 /* Fail if buffer length is not large enough to contain
Bertrand Jacquin67de5a22017-12-13 01:15:05 +0000746 * IPv4 header, TCP header */
Bertrand Jacquin93b227d2016-06-04 15:11:10 +0100747 goto missing;
Bertrand Jacquinb3875912017-12-13 00:58:51 +0000748 }
749 else if (hdr_ip4->ip_p != IPPROTO_TCP) {
Bertrand Jacquin93b227d2016-06-04 15:11:10 +0100750 /* The protocol does not include a TCP header */
751 conn->err_code = CO_ER_CIP_BAD_PROTO;
752 goto fail;
Bertrand Jacquinb3875912017-12-13 00:58:51 +0000753 }
Bertrand Jacquin93b227d2016-06-04 15:11:10 +0100754
David Carlier3015a2e2016-07-04 22:51:33 +0100755 hdr_tcp = (struct my_tcphdr *)(line + (hdr_ip4->ip_hl * 4));
Bertrand Jacquin93b227d2016-06-04 15:11:10 +0100756
757 /* update the session's addresses and mark them set */
Willy Tarreau226572f2019-07-17 14:46:00 +0200758 ((struct sockaddr_in *)conn->src)->sin_family = AF_INET;
759 ((struct sockaddr_in *)conn->src)->sin_addr.s_addr = hdr_ip4->ip_src.s_addr;
760 ((struct sockaddr_in *)conn->src)->sin_port = hdr_tcp->source;
Bertrand Jacquin93b227d2016-06-04 15:11:10 +0100761
Willy Tarreau226572f2019-07-17 14:46:00 +0200762 ((struct sockaddr_in *)conn->dst)->sin_family = AF_INET;
763 ((struct sockaddr_in *)conn->dst)->sin_addr.s_addr = hdr_ip4->ip_dst.s_addr;
764 ((struct sockaddr_in *)conn->dst)->sin_port = hdr_tcp->dest;
Bertrand Jacquin93b227d2016-06-04 15:11:10 +0100765
766 conn->flags |= CO_FL_ADDR_FROM_SET | CO_FL_ADDR_TO_SET;
767 }
Willy Tarreau0ca24aa2019-03-29 17:35:32 +0100768 else if (ip_ver == 6) {
Bertrand Jacquin93b227d2016-06-04 15:11:10 +0100769 struct ip6_hdr *hdr_ip6;
David Carlier3015a2e2016-07-04 22:51:33 +0100770 struct my_tcphdr *hdr_tcp;
Bertrand Jacquin93b227d2016-06-04 15:11:10 +0100771
772 hdr_ip6 = (struct ip6_hdr *)line;
773
Willy Tarreau843b7cb2018-07-13 10:54:26 +0200774 if (trash.data < 60 || trash.data < hdr_len) {
Bertrand Jacquin93b227d2016-06-04 15:11:10 +0100775 /* Fail if buffer length is not large enough to contain
Bertrand Jacquin67de5a22017-12-13 01:15:05 +0000776 * IPv6 header, TCP header */
Bertrand Jacquin93b227d2016-06-04 15:11:10 +0100777 goto missing;
Bertrand Jacquinb3875912017-12-13 00:58:51 +0000778 }
779 else if (hdr_ip6->ip6_nxt != IPPROTO_TCP) {
Bertrand Jacquin93b227d2016-06-04 15:11:10 +0100780 /* The protocol does not include a TCP header */
781 conn->err_code = CO_ER_CIP_BAD_PROTO;
782 goto fail;
Bertrand Jacquinb3875912017-12-13 00:58:51 +0000783 }
Bertrand Jacquin93b227d2016-06-04 15:11:10 +0100784
David Carlier3015a2e2016-07-04 22:51:33 +0100785 hdr_tcp = (struct my_tcphdr *)(line + sizeof(struct ip6_hdr));
Bertrand Jacquin93b227d2016-06-04 15:11:10 +0100786
787 /* update the session's addresses and mark them set */
Willy Tarreau226572f2019-07-17 14:46:00 +0200788 ((struct sockaddr_in6 *)conn->src)->sin6_family = AF_INET6;
789 ((struct sockaddr_in6 *)conn->src)->sin6_addr = hdr_ip6->ip6_src;
790 ((struct sockaddr_in6 *)conn->src)->sin6_port = hdr_tcp->source;
Bertrand Jacquin93b227d2016-06-04 15:11:10 +0100791
Willy Tarreau226572f2019-07-17 14:46:00 +0200792 ((struct sockaddr_in6 *)conn->dst)->sin6_family = AF_INET6;
793 ((struct sockaddr_in6 *)conn->dst)->sin6_addr = hdr_ip6->ip6_dst;
794 ((struct sockaddr_in6 *)conn->dst)->sin6_port = hdr_tcp->dest;
Bertrand Jacquin93b227d2016-06-04 15:11:10 +0100795
796 conn->flags |= CO_FL_ADDR_FROM_SET | CO_FL_ADDR_TO_SET;
797 }
798 else {
799 /* The protocol does not match something known (IPv4/IPv6) */
800 conn->err_code = CO_ER_CIP_BAD_PROTO;
801 goto fail;
802 }
803
Bertrand Jacquin7d668f92017-12-13 01:23:39 +0000804 line += hdr_len;
Willy Tarreau843b7cb2018-07-13 10:54:26 +0200805 trash.data = line - trash.area;
Bertrand Jacquin93b227d2016-06-04 15:11:10 +0100806
807 /* remove the NetScaler Client IP header from the request. For this
808 * we re-read the exact line at once. If we don't get the exact same
809 * result, we fail.
810 */
Willy Tarreau157788c2020-02-11 10:08:05 +0100811 while (1) {
Willy Tarreau843b7cb2018-07-13 10:54:26 +0200812 int len2 = recv(conn->handle.fd, trash.area, trash.data, 0);
Bertrand Jacquin93b227d2016-06-04 15:11:10 +0100813 if (len2 < 0 && errno == EINTR)
814 continue;
Willy Tarreau843b7cb2018-07-13 10:54:26 +0200815 if (len2 != trash.data)
Bertrand Jacquin93b227d2016-06-04 15:11:10 +0100816 goto recv_abort;
Willy Tarreau157788c2020-02-11 10:08:05 +0100817 break;
818 }
Bertrand Jacquin93b227d2016-06-04 15:11:10 +0100819
820 conn->flags &= ~flag;
821 return 1;
822
Willy Tarreau6499b9d2019-06-03 08:17:30 +0200823 not_ready:
Willy Tarreau6499b9d2019-06-03 08:17:30 +0200824 return 0;
825
Bertrand Jacquin93b227d2016-06-04 15:11:10 +0100826 missing:
827 /* Missing data. Since we're using MSG_PEEK, we can only poll again if
828 * we have not read anything. Otherwise we need to fail because we won't
829 * be able to poll anymore.
830 */
831 conn->err_code = CO_ER_CIP_TRUNCATED;
832 goto fail;
833
834 bad_magic:
835 conn->err_code = CO_ER_CIP_BAD_MAGIC;
836 goto fail;
837
838 recv_abort:
839 conn->err_code = CO_ER_CIP_ABORT;
840 conn->flags |= CO_FL_SOCK_RD_SH | CO_FL_SOCK_WR_SH;
841 goto fail;
842
843 fail:
Bertrand Jacquin93b227d2016-06-04 15:11:10 +0100844 conn->flags |= CO_FL_ERROR;
845 return 0;
846}
847
Alexander Liu2a54bb72019-05-22 19:44:48 +0800848
849int conn_send_socks4_proxy_request(struct connection *conn)
850{
851 struct socks4_request req_line;
852
Alexander Liu2a54bb72019-05-22 19:44:48 +0800853 if (!conn_ctrl_ready(conn))
854 goto out_error;
855
Willy Tarreau226572f2019-07-17 14:46:00 +0200856 if (!conn_get_dst(conn))
857 goto out_error;
858
Alexander Liu2a54bb72019-05-22 19:44:48 +0800859 req_line.version = 0x04;
860 req_line.command = 0x01;
Willy Tarreau226572f2019-07-17 14:46:00 +0200861 req_line.port = get_net_port(conn->dst);
862 req_line.ip = is_inet_addr(conn->dst);
Alexander Liu2a54bb72019-05-22 19:44:48 +0800863 memcpy(req_line.user_id, "HAProxy\0", 8);
864
865 if (conn->send_proxy_ofs > 0) {
866 /*
867 * This is the first call to send the request
868 */
869 conn->send_proxy_ofs = -(int)sizeof(req_line);
870 }
871
872 if (conn->send_proxy_ofs < 0) {
873 int ret = 0;
874
875 /* we are sending the socks4_req_line here. If the data layer
876 * has a pending write, we'll also set MSG_MORE.
877 */
Willy Tarreau827fee72020-12-11 15:26:55 +0100878 ret = conn_ctrl_send(
Alexander Liu2a54bb72019-05-22 19:44:48 +0800879 conn,
880 ((char *)(&req_line)) + (sizeof(req_line)+conn->send_proxy_ofs),
881 -conn->send_proxy_ofs,
Willy Tarreau827fee72020-12-11 15:26:55 +0100882 (conn->subs && conn->subs->events & SUB_RETRY_SEND) ? CO_SFL_MSG_MORE : 0);
Alexander Liu2a54bb72019-05-22 19:44:48 +0800883
884 DPRINTF(stderr, "SOCKS PROXY HS FD[%04X]: Before send remain is [%d], sent [%d]\n",
885 conn->handle.fd, -conn->send_proxy_ofs, ret);
886
887 if (ret < 0) {
888 goto out_error;
889 }
890
891 conn->send_proxy_ofs += ret; /* becomes zero once complete */
892 if (conn->send_proxy_ofs != 0) {
893 goto out_wait;
894 }
895 }
896
897 /* OK we've the whole request sent */
898 conn->flags &= ~CO_FL_SOCKS4_SEND;
Alexander Liu2a54bb72019-05-22 19:44:48 +0800899
900 /* The connection is ready now, simply return and let the connection
901 * handler notify upper layers if needed.
902 */
Willy Tarreauc192b0a2020-01-23 09:11:58 +0100903 conn->flags &= ~CO_FL_WAIT_L4_CONN;
Alexander Liu2a54bb72019-05-22 19:44:48 +0800904
905 if (conn->flags & CO_FL_SEND_PROXY) {
906 /*
907 * Get the send_proxy_ofs ready for the send_proxy due to we are
908 * reusing the "send_proxy_ofs", and SOCKS4 handshake should be done
909 * before sending PROXY Protocol.
910 */
911 conn->send_proxy_ofs = 1;
912 }
913 return 1;
914
915 out_error:
916 /* Write error on the file descriptor */
917 conn->flags |= CO_FL_ERROR;
918 if (conn->err_code == CO_ER_NONE) {
919 conn->err_code = CO_ER_SOCKS4_SEND;
920 }
921 return 0;
922
923 out_wait:
Alexander Liu2a54bb72019-05-22 19:44:48 +0800924 return 0;
925}
926
927int conn_recv_socks4_proxy_response(struct connection *conn)
928{
929 char line[SOCKS4_HS_RSP_LEN];
930 int ret;
931
Alexander Liu2a54bb72019-05-22 19:44:48 +0800932 if (!conn_ctrl_ready(conn))
933 goto fail;
934
935 if (!fd_recv_ready(conn->handle.fd))
Willy Tarreau6499b9d2019-06-03 08:17:30 +0200936 goto not_ready;
Alexander Liu2a54bb72019-05-22 19:44:48 +0800937
Willy Tarreau157788c2020-02-11 10:08:05 +0100938 while (1) {
Alexander Liu2a54bb72019-05-22 19:44:48 +0800939 /* SOCKS4 Proxy will response with 8 bytes, 0x00 | 0x5A | 0x00 0x00 | 0x00 0x00 0x00 0x00
940 * Try to peek into it, before all 8 bytes ready.
941 */
942 ret = recv(conn->handle.fd, line, SOCKS4_HS_RSP_LEN, MSG_PEEK);
943
944 if (ret == 0) {
945 /* the socket has been closed or shutdown for send */
946 DPRINTF(stderr, "SOCKS PROXY HS FD[%04X]: Received ret[%d], errno[%d], looks like the socket has been closed or shutdown for send\n",
947 conn->handle.fd, ret, errno);
948 if (conn->err_code == CO_ER_NONE) {
949 conn->err_code = CO_ER_SOCKS4_RECV;
950 }
951 goto fail;
952 }
953
954 if (ret > 0) {
955 if (ret == SOCKS4_HS_RSP_LEN) {
956 DPRINTF(stderr, "SOCKS PROXY HS FD[%04X]: Received 8 bytes, the response is [%02X|%02X|%02X %02X|%02X %02X %02X %02X]\n",
957 conn->handle.fd, line[0], line[1], line[2], line[3], line[4], line[5], line[6], line[7]);
958 }else{
959 DPRINTF(stderr, "SOCKS PROXY HS FD[%04X]: Received ret[%d], first byte is [%02X], last bye is [%02X]\n", conn->handle.fd, ret, line[0], line[ret-1]);
960 }
961 } else {
962 DPRINTF(stderr, "SOCKS PROXY HS FD[%04X]: Received ret[%d], errno[%d]\n", conn->handle.fd, ret, errno);
963 }
964
965 if (ret < 0) {
966 if (errno == EINTR) {
967 continue;
968 }
969 if (errno == EAGAIN) {
970 fd_cant_recv(conn->handle.fd);
Willy Tarreau6499b9d2019-06-03 08:17:30 +0200971 goto not_ready;
Alexander Liu2a54bb72019-05-22 19:44:48 +0800972 }
973 goto recv_abort;
974 }
Willy Tarreau157788c2020-02-11 10:08:05 +0100975 break;
976 }
Alexander Liu2a54bb72019-05-22 19:44:48 +0800977
Willy Tarreauc192b0a2020-01-23 09:11:58 +0100978 conn->flags &= ~CO_FL_WAIT_L4_CONN;
979
Alexander Liu2a54bb72019-05-22 19:44:48 +0800980 if (ret < SOCKS4_HS_RSP_LEN) {
981 /* Missing data. Since we're using MSG_PEEK, we can only poll again if
982 * we are not able to read enough data.
983 */
Willy Tarreau6499b9d2019-06-03 08:17:30 +0200984 goto not_ready;
Alexander Liu2a54bb72019-05-22 19:44:48 +0800985 }
986
987 /*
988 * Base on the SOCSK4 protocol:
989 *
990 * +----+----+----+----+----+----+----+----+
991 * | VN | CD | DSTPORT | DSTIP |
992 * +----+----+----+----+----+----+----+----+
993 * # of bytes: 1 1 2 4
994 * VN is the version of the reply code and should be 0. CD is the result
995 * code with one of the following values:
996 * 90: request granted
997 * 91: request rejected or failed
Ilya Shipitsince7b00f2020-03-23 22:28:40 +0500998 * 92: request rejected because SOCKS server cannot connect to identd on the client
Alexander Liu2a54bb72019-05-22 19:44:48 +0800999 * 93: request rejected because the client program and identd report different user-ids
1000 * The remaining fields are ignored.
1001 */
1002 if (line[1] != 90) {
1003 conn->flags &= ~CO_FL_SOCKS4_RECV;
1004
1005 DPRINTF(stderr, "SOCKS PROXY HS FD[%04X]: FAIL, the response is [%02X|%02X|%02X %02X|%02X %02X %02X %02X]\n",
1006 conn->handle.fd, line[0], line[1], line[2], line[3], line[4], line[5], line[6], line[7]);
1007 if (conn->err_code == CO_ER_NONE) {
1008 conn->err_code = CO_ER_SOCKS4_DENY;
1009 }
1010 goto fail;
1011 }
1012
1013 /* remove the 8 bytes response from the stream */
Willy Tarreau157788c2020-02-11 10:08:05 +01001014 while (1) {
Alexander Liu2a54bb72019-05-22 19:44:48 +08001015 ret = recv(conn->handle.fd, line, SOCKS4_HS_RSP_LEN, 0);
1016 if (ret < 0 && errno == EINTR) {
1017 continue;
1018 }
1019 if (ret != SOCKS4_HS_RSP_LEN) {
1020 if (conn->err_code == CO_ER_NONE) {
1021 conn->err_code = CO_ER_SOCKS4_RECV;
1022 }
1023 goto fail;
1024 }
Willy Tarreau157788c2020-02-11 10:08:05 +01001025 break;
1026 }
Alexander Liu2a54bb72019-05-22 19:44:48 +08001027
1028 conn->flags &= ~CO_FL_SOCKS4_RECV;
1029 return 1;
1030
Willy Tarreau6499b9d2019-06-03 08:17:30 +02001031 not_ready:
Willy Tarreau6499b9d2019-06-03 08:17:30 +02001032 return 0;
1033
Alexander Liu2a54bb72019-05-22 19:44:48 +08001034 recv_abort:
1035 if (conn->err_code == CO_ER_NONE) {
1036 conn->err_code = CO_ER_SOCKS4_ABORT;
1037 }
1038 conn->flags |= (CO_FL_SOCK_RD_SH | CO_FL_SOCK_WR_SH);
1039 goto fail;
1040
1041 fail:
Alexander Liu2a54bb72019-05-22 19:44:48 +08001042 conn->flags |= CO_FL_ERROR;
1043 return 0;
1044}
1045
Ilya Shipitsinca56fce2018-09-15 00:50:05 +05001046/* Note: <remote> is explicitly allowed to be NULL */
Tim Duesterhuscf6e0c82020-03-13 12:34:24 +01001047int make_proxy_line(char *buf, int buf_len, struct server *srv, struct connection *remote, struct stream *strm)
David Safb76832014-05-08 23:42:08 -04001048{
1049 int ret = 0;
1050
1051 if (srv && (srv->pp_opts & SRV_PP_V2)) {
Tim Duesterhuscf6e0c82020-03-13 12:34:24 +01001052 ret = make_proxy_line_v2(buf, buf_len, srv, remote, strm);
David Safb76832014-05-08 23:42:08 -04001053 }
1054 else {
Willy Tarreau226572f2019-07-17 14:46:00 +02001055 if (remote && conn_get_src(remote) && conn_get_dst(remote))
1056 ret = make_proxy_line_v1(buf, buf_len, remote->src, remote->dst);
David Safb76832014-05-08 23:42:08 -04001057 else
1058 ret = make_proxy_line_v1(buf, buf_len, NULL, NULL);
1059 }
1060
1061 return ret;
1062}
1063
Willy Tarreaue1e4a612012-10-05 00:10:55 +02001064/* Makes a PROXY protocol line from the two addresses. The output is sent to
1065 * buffer <buf> for a maximum size of <buf_len> (including the trailing zero).
1066 * It returns the number of bytes composing this line (including the trailing
1067 * LF), or zero in case of failure (eg: not enough space). It supports TCP4,
Willy Tarreau2e1401a2013-10-01 11:41:55 +02001068 * TCP6 and "UNKNOWN" formats. If any of <src> or <dst> is null, UNKNOWN is
1069 * emitted as well.
Willy Tarreaue1e4a612012-10-05 00:10:55 +02001070 */
David Safb76832014-05-08 23:42:08 -04001071int make_proxy_line_v1(char *buf, int buf_len, struct sockaddr_storage *src, struct sockaddr_storage *dst)
Willy Tarreaue1e4a612012-10-05 00:10:55 +02001072{
1073 int ret = 0;
Tim Duesterhus7fec0212018-07-27 18:46:13 +02001074 char * protocol;
1075 char src_str[MAX(INET_ADDRSTRLEN, INET6_ADDRSTRLEN)];
1076 char dst_str[MAX(INET_ADDRSTRLEN, INET6_ADDRSTRLEN)];
1077 in_port_t src_port;
1078 in_port_t dst_port;
Willy Tarreaue1e4a612012-10-05 00:10:55 +02001079
Tim Duesterhus7fec0212018-07-27 18:46:13 +02001080 if ( !src
1081 || !dst
1082 || (src->ss_family != AF_INET && src->ss_family != AF_INET6)
1083 || (dst->ss_family != AF_INET && dst->ss_family != AF_INET6)) {
1084 /* unknown family combination */
1085 ret = snprintf(buf, buf_len, "PROXY UNKNOWN\r\n");
Willy Tarreaue1e4a612012-10-05 00:10:55 +02001086 if (ret >= buf_len)
1087 return 0;
1088
Tim Duesterhus7fec0212018-07-27 18:46:13 +02001089 return ret;
1090 }
Willy Tarreaue1e4a612012-10-05 00:10:55 +02001091
Tim Duesterhus7fec0212018-07-27 18:46:13 +02001092 /* IPv4 for both src and dst */
1093 if (src->ss_family == AF_INET && dst->ss_family == AF_INET) {
1094 protocol = "TCP4";
1095 if (!inet_ntop(AF_INET, &((struct sockaddr_in *)src)->sin_addr, src_str, sizeof(src_str)))
Willy Tarreaue1e4a612012-10-05 00:10:55 +02001096 return 0;
Tim Duesterhus7fec0212018-07-27 18:46:13 +02001097 src_port = ((struct sockaddr_in *)src)->sin_port;
1098 if (!inet_ntop(AF_INET, &((struct sockaddr_in *)dst)->sin_addr, dst_str, sizeof(dst_str)))
Willy Tarreaue1e4a612012-10-05 00:10:55 +02001099 return 0;
Tim Duesterhus7fec0212018-07-27 18:46:13 +02001100 dst_port = ((struct sockaddr_in *)dst)->sin_port;
Willy Tarreaue1e4a612012-10-05 00:10:55 +02001101 }
Tim Duesterhus7fec0212018-07-27 18:46:13 +02001102 /* IPv6 for at least one of src and dst */
1103 else {
1104 struct in6_addr tmp;
Willy Tarreaue1e4a612012-10-05 00:10:55 +02001105
Tim Duesterhus7fec0212018-07-27 18:46:13 +02001106 protocol = "TCP6";
Willy Tarreaue1e4a612012-10-05 00:10:55 +02001107
Tim Duesterhus7fec0212018-07-27 18:46:13 +02001108 if (src->ss_family == AF_INET) {
1109 /* Convert src to IPv6 */
1110 v4tov6(&tmp, &((struct sockaddr_in *)src)->sin_addr);
1111 src_port = ((struct sockaddr_in *)src)->sin_port;
1112 }
1113 else {
1114 tmp = ((struct sockaddr_in6 *)src)->sin6_addr;
1115 src_port = ((struct sockaddr_in6 *)src)->sin6_port;
1116 }
Willy Tarreaue1e4a612012-10-05 00:10:55 +02001117
Tim Duesterhus7fec0212018-07-27 18:46:13 +02001118 if (!inet_ntop(AF_INET6, &tmp, src_str, sizeof(src_str)))
Willy Tarreaue1e4a612012-10-05 00:10:55 +02001119 return 0;
1120
Tim Duesterhus7fec0212018-07-27 18:46:13 +02001121 if (dst->ss_family == AF_INET) {
1122 /* Convert dst to IPv6 */
1123 v4tov6(&tmp, &((struct sockaddr_in *)dst)->sin_addr);
1124 dst_port = ((struct sockaddr_in *)dst)->sin_port;
1125 }
1126 else {
1127 tmp = ((struct sockaddr_in6 *)dst)->sin6_addr;
1128 dst_port = ((struct sockaddr_in6 *)dst)->sin6_port;
1129 }
Willy Tarreaue1e4a612012-10-05 00:10:55 +02001130
Tim Duesterhus7fec0212018-07-27 18:46:13 +02001131 if (!inet_ntop(AF_INET6, &tmp, dst_str, sizeof(dst_str)))
Willy Tarreaue1e4a612012-10-05 00:10:55 +02001132 return 0;
1133 }
Tim Duesterhus7fec0212018-07-27 18:46:13 +02001134
1135 ret = snprintf(buf, buf_len, "PROXY %s %s %s %u %u\r\n", protocol, src_str, dst_str, ntohs(src_port), ntohs(dst_port));
1136 if (ret >= buf_len)
1137 return 0;
1138
Willy Tarreaue1e4a612012-10-05 00:10:55 +02001139 return ret;
1140}
David Safb76832014-05-08 23:42:08 -04001141
KOVACS Krisztianb3e54fe2014-11-17 15:11:45 +01001142static int make_tlv(char *dest, int dest_len, char type, uint16_t length, const char *value)
David Safb76832014-05-08 23:42:08 -04001143{
1144 struct tlv *tlv;
1145
1146 if (!dest || (length + sizeof(*tlv) > dest_len))
1147 return 0;
1148
1149 tlv = (struct tlv *)dest;
1150
1151 tlv->type = type;
1152 tlv->length_hi = length >> 8;
1153 tlv->length_lo = length & 0x00ff;
1154 memcpy(tlv->value, value, length);
1155 return length + sizeof(*tlv);
1156}
David Safb76832014-05-08 23:42:08 -04001157
Ilya Shipitsinca56fce2018-09-15 00:50:05 +05001158/* Note: <remote> is explicitly allowed to be NULL */
Tim Duesterhuscf6e0c82020-03-13 12:34:24 +01001159int make_proxy_line_v2(char *buf, int buf_len, struct server *srv, struct connection *remote, struct stream *strm)
David Safb76832014-05-08 23:42:08 -04001160{
Willy Tarreau8fccfa22014-06-14 08:28:06 +02001161 const char pp2_signature[] = PP2_SIGNATURE;
Emmanuel Hocdet4399c752018-02-05 15:26:43 +01001162 void *tlv_crc32c_p = NULL;
David Safb76832014-05-08 23:42:08 -04001163 int ret = 0;
Willy Tarreau8fccfa22014-06-14 08:28:06 +02001164 struct proxy_hdr_v2 *hdr = (struct proxy_hdr_v2 *)buf;
Vincent Bernat6e615892016-05-18 16:17:44 +02001165 struct sockaddr_storage null_addr = { .ss_family = 0 };
David Safb76832014-05-08 23:42:08 -04001166 struct sockaddr_storage *src = &null_addr;
1167 struct sockaddr_storage *dst = &null_addr;
Emmanuel Hocdet404d9782017-10-24 10:55:14 +02001168 const char *value;
1169 int value_len;
David Safb76832014-05-08 23:42:08 -04001170
1171 if (buf_len < PP2_HEADER_LEN)
1172 return 0;
Willy Tarreau8fccfa22014-06-14 08:28:06 +02001173 memcpy(hdr->sig, pp2_signature, PP2_SIGNATURE_LEN);
David Safb76832014-05-08 23:42:08 -04001174
Willy Tarreau226572f2019-07-17 14:46:00 +02001175 if (remote && conn_get_src(remote) && conn_get_dst(remote)) {
1176 src = remote->src;
1177 dst = remote->dst;
David Safb76832014-05-08 23:42:08 -04001178 }
KOVACS Krisztianb3e54fe2014-11-17 15:11:45 +01001179
Tim Duesterhus7fec0212018-07-27 18:46:13 +02001180 /* At least one of src or dst is not of AF_INET or AF_INET6 */
1181 if ( !src
1182 || !dst
Willy Tarreau119e50e2020-05-22 13:53:29 +02001183 || (!pp2_never_send_local && conn_is_back(remote)) // locally initiated connection
Tim Duesterhus7fec0212018-07-27 18:46:13 +02001184 || (src->ss_family != AF_INET && src->ss_family != AF_INET6)
1185 || (dst->ss_family != AF_INET && dst->ss_family != AF_INET6)) {
David Safb76832014-05-08 23:42:08 -04001186 if (buf_len < PP2_HDR_LEN_UNSPEC)
1187 return 0;
Willy Tarreau8fccfa22014-06-14 08:28:06 +02001188 hdr->ver_cmd = PP2_VERSION | PP2_CMD_LOCAL;
1189 hdr->fam = PP2_FAM_UNSPEC | PP2_TRANS_UNSPEC;
David Safb76832014-05-08 23:42:08 -04001190 ret = PP2_HDR_LEN_UNSPEC;
1191 }
Tim Duesterhus7fec0212018-07-27 18:46:13 +02001192 else {
Willy Tarreau02c88032020-04-14 12:54:10 +02001193 hdr->ver_cmd = PP2_VERSION | PP2_CMD_PROXY;
Tim Duesterhus7fec0212018-07-27 18:46:13 +02001194 /* IPv4 for both src and dst */
1195 if (src->ss_family == AF_INET && dst->ss_family == AF_INET) {
1196 if (buf_len < PP2_HDR_LEN_INET)
1197 return 0;
Tim Duesterhus7fec0212018-07-27 18:46:13 +02001198 hdr->fam = PP2_FAM_INET | PP2_TRANS_STREAM;
1199 hdr->addr.ip4.src_addr = ((struct sockaddr_in *)src)->sin_addr.s_addr;
1200 hdr->addr.ip4.src_port = ((struct sockaddr_in *)src)->sin_port;
1201 hdr->addr.ip4.dst_addr = ((struct sockaddr_in *)dst)->sin_addr.s_addr;
1202 hdr->addr.ip4.dst_port = ((struct sockaddr_in *)dst)->sin_port;
1203 ret = PP2_HDR_LEN_INET;
1204 }
1205 /* IPv6 for at least one of src and dst */
1206 else {
1207 struct in6_addr tmp;
1208
1209 if (buf_len < PP2_HDR_LEN_INET6)
1210 return 0;
Tim Duesterhus7fec0212018-07-27 18:46:13 +02001211 hdr->fam = PP2_FAM_INET6 | PP2_TRANS_STREAM;
1212 if (src->ss_family == AF_INET) {
1213 v4tov6(&tmp, &((struct sockaddr_in *)src)->sin_addr);
1214 memcpy(hdr->addr.ip6.src_addr, &tmp, 16);
1215 hdr->addr.ip6.src_port = ((struct sockaddr_in *)src)->sin_port;
1216 }
1217 else {
1218 memcpy(hdr->addr.ip6.src_addr, &((struct sockaddr_in6 *)src)->sin6_addr, 16);
1219 hdr->addr.ip6.src_port = ((struct sockaddr_in6 *)src)->sin6_port;
1220 }
1221 if (dst->ss_family == AF_INET) {
1222 v4tov6(&tmp, &((struct sockaddr_in *)dst)->sin_addr);
1223 memcpy(hdr->addr.ip6.dst_addr, &tmp, 16);
William Dauchybd8bf672020-01-26 19:06:39 +01001224 hdr->addr.ip6.dst_port = ((struct sockaddr_in *)dst)->sin_port;
Tim Duesterhus7fec0212018-07-27 18:46:13 +02001225 }
1226 else {
1227 memcpy(hdr->addr.ip6.dst_addr, &((struct sockaddr_in6 *)dst)->sin6_addr, 16);
1228 hdr->addr.ip6.dst_port = ((struct sockaddr_in6 *)dst)->sin6_port;
1229 }
1230
1231 ret = PP2_HDR_LEN_INET6;
1232 }
1233 }
David Safb76832014-05-08 23:42:08 -04001234
Emmanuel Hocdet4399c752018-02-05 15:26:43 +01001235 if (srv->pp_opts & SRV_PP_V2_CRC32C) {
1236 uint32_t zero_crc32c = 0;
Tim Duesterhusa8692f32020-03-13 12:34:25 +01001237
Emmanuel Hocdet4399c752018-02-05 15:26:43 +01001238 if ((buf_len - ret) < sizeof(struct tlv))
1239 return 0;
1240 tlv_crc32c_p = (void *)((struct tlv *)&buf[ret])->value;
1241 ret += make_tlv(&buf[ret], (buf_len - ret), PP2_TYPE_CRC32C, sizeof(zero_crc32c), (const char *)&zero_crc32c);
1242 }
1243
Ilya Shipitsinca56fce2018-09-15 00:50:05 +05001244 if (remote && conn_get_alpn(remote, &value, &value_len)) {
Emmanuel Hocdet404d9782017-10-24 10:55:14 +02001245 if ((buf_len - ret) < sizeof(struct tlv))
1246 return 0;
Emmanuel Hocdet571c7ac2017-10-31 18:24:05 +01001247 ret += make_tlv(&buf[ret], (buf_len - ret), PP2_TYPE_ALPN, value_len, value);
Emmanuel Hocdet404d9782017-10-24 10:55:14 +02001248 }
1249
Emmanuel Hocdet253c3b72018-02-01 18:29:59 +01001250 if (srv->pp_opts & SRV_PP_V2_AUTHORITY) {
Emmanuel Hocdet8a4ffa02019-08-29 11:54:51 +02001251 value = NULL;
1252 if (remote && remote->proxy_authority) {
1253 value = remote->proxy_authority;
1254 value_len = remote->proxy_authority_len;
1255 }
1256#ifdef USE_OPENSSL
1257 else {
Jerome Magnin78891c72019-09-02 09:53:41 +02001258 if ((value = ssl_sock_get_sni(remote)))
Emmanuel Hocdet8a4ffa02019-08-29 11:54:51 +02001259 value_len = strlen(value);
1260 }
1261#endif
Emmanuel Hocdet253c3b72018-02-01 18:29:59 +01001262 if (value) {
1263 if ((buf_len - ret) < sizeof(struct tlv))
1264 return 0;
Emmanuel Hocdet8a4ffa02019-08-29 11:54:51 +02001265 ret += make_tlv(&buf[ret], (buf_len - ret), PP2_TYPE_AUTHORITY, value_len, value);
Emmanuel Hocdet253c3b72018-02-01 18:29:59 +01001266 }
1267 }
1268
Christopher Faulet3ab504f2020-05-26 15:16:01 +02001269 if (strm && (srv->pp_opts & SRV_PP_V2_UNIQUE_ID)) {
Tim Duesterhuscf6e0c82020-03-13 12:34:24 +01001270 struct session* sess = strm_sess(strm);
1271 struct ist unique_id = stream_generate_unique_id(strm, &sess->fe->format_unique_id);
1272
1273 value = unique_id.ptr;
1274 value_len = unique_id.len;
1275
1276 if (value_len >= 0) {
1277 if ((buf_len - ret) < sizeof(struct tlv))
1278 return 0;
1279 ret += make_tlv(&buf[ret], (buf_len - ret), PP2_TYPE_UNIQUE_ID, value_len, value);
1280 }
1281 }
1282
Emmanuel Hocdet8a4ffa02019-08-29 11:54:51 +02001283#ifdef USE_OPENSSL
David Safb76832014-05-08 23:42:08 -04001284 if (srv->pp_opts & SRV_PP_V2_SSL) {
Emmanuel Hocdet404d9782017-10-24 10:55:14 +02001285 struct tlv_ssl *tlv;
1286 int ssl_tlv_len = 0;
Tim Duesterhusa8692f32020-03-13 12:34:25 +01001287
David Safb76832014-05-08 23:42:08 -04001288 if ((buf_len - ret) < sizeof(struct tlv_ssl))
1289 return 0;
1290 tlv = (struct tlv_ssl *)&buf[ret];
1291 memset(tlv, 0, sizeof(struct tlv_ssl));
1292 ssl_tlv_len += sizeof(struct tlv_ssl);
1293 tlv->tlv.type = PP2_TYPE_SSL;
1294 if (ssl_sock_is_ssl(remote)) {
1295 tlv->client |= PP2_CLIENT_SSL;
Emmanuel Hocdet01da5712017-10-13 16:59:49 +02001296 value = ssl_sock_get_proto_version(remote);
David Safb76832014-05-08 23:42:08 -04001297 if (value) {
Emmanuel Hocdet8c0c34b2018-02-28 12:02:14 +01001298 ssl_tlv_len += make_tlv(&buf[ret+ssl_tlv_len], (buf_len-ret-ssl_tlv_len), PP2_SUBTYPE_SSL_VERSION, strlen(value), value);
David Safb76832014-05-08 23:42:08 -04001299 }
Dave McCowan328fb582014-07-30 10:39:13 -04001300 if (ssl_sock_get_cert_used_sess(remote)) {
1301 tlv->client |= PP2_CLIENT_CERT_SESS;
David Safb76832014-05-08 23:42:08 -04001302 tlv->verify = htonl(ssl_sock_get_verify_result(remote));
Dave McCowan328fb582014-07-30 10:39:13 -04001303 if (ssl_sock_get_cert_used_conn(remote))
1304 tlv->client |= PP2_CLIENT_CERT_CONN;
David Safb76832014-05-08 23:42:08 -04001305 }
1306 if (srv->pp_opts & SRV_PP_V2_SSL_CN) {
Willy Tarreau83061a82018-07-13 11:56:34 +02001307 struct buffer *cn_trash = get_trash_chunk();
Willy Tarreau3b9a0c92014-07-19 06:37:33 +02001308 if (ssl_sock_get_remote_common_name(remote, cn_trash) > 0) {
Willy Tarreau843b7cb2018-07-13 10:54:26 +02001309 ssl_tlv_len += make_tlv(&buf[ret+ssl_tlv_len], (buf_len - ret - ssl_tlv_len), PP2_SUBTYPE_SSL_CN,
1310 cn_trash->data,
1311 cn_trash->area);
David Safb76832014-05-08 23:42:08 -04001312 }
1313 }
Emmanuel Hocdetfa8d0f12018-02-01 15:53:52 +01001314 if (srv->pp_opts & SRV_PP_V2_SSL_KEY_ALG) {
Willy Tarreau83061a82018-07-13 11:56:34 +02001315 struct buffer *pkey_trash = get_trash_chunk();
Emmanuel Hocdetfa8d0f12018-02-01 15:53:52 +01001316 if (ssl_sock_get_pkey_algo(remote, pkey_trash) > 0) {
Willy Tarreau843b7cb2018-07-13 10:54:26 +02001317 ssl_tlv_len += make_tlv(&buf[ret+ssl_tlv_len], (buf_len - ret - ssl_tlv_len), PP2_SUBTYPE_SSL_KEY_ALG,
1318 pkey_trash->data,
1319 pkey_trash->area);
Emmanuel Hocdetfa8d0f12018-02-01 15:53:52 +01001320 }
1321 }
1322 if (srv->pp_opts & SRV_PP_V2_SSL_SIG_ALG) {
1323 value = ssl_sock_get_cert_sig(remote);
1324 if (value) {
1325 ssl_tlv_len += make_tlv(&buf[ret+ssl_tlv_len], (buf_len - ret - ssl_tlv_len), PP2_SUBTYPE_SSL_SIG_ALG, strlen(value), value);
1326 }
1327 }
1328 if (srv->pp_opts & SRV_PP_V2_SSL_CIPHER) {
1329 value = ssl_sock_get_cipher_name(remote);
1330 if (value) {
1331 ssl_tlv_len += make_tlv(&buf[ret+ssl_tlv_len], (buf_len - ret - ssl_tlv_len), PP2_SUBTYPE_SSL_CIPHER, strlen(value), value);
1332 }
1333 }
David Safb76832014-05-08 23:42:08 -04001334 }
1335 tlv->tlv.length_hi = (uint16_t)(ssl_tlv_len - sizeof(struct tlv)) >> 8;
1336 tlv->tlv.length_lo = (uint16_t)(ssl_tlv_len - sizeof(struct tlv)) & 0x00ff;
1337 ret += ssl_tlv_len;
1338 }
1339#endif
1340
Willy Tarreaue5733232019-05-22 19:24:06 +02001341#ifdef USE_NS
KOVACS Krisztianb3e54fe2014-11-17 15:11:45 +01001342 if (remote && (remote->proxy_netns)) {
1343 if ((buf_len - ret) < sizeof(struct tlv))
1344 return 0;
Emmanuel Hocdet571c7ac2017-10-31 18:24:05 +01001345 ret += make_tlv(&buf[ret], (buf_len - ret), PP2_TYPE_NETNS, remote->proxy_netns->name_len, remote->proxy_netns->node.key);
KOVACS Krisztianb3e54fe2014-11-17 15:11:45 +01001346 }
1347#endif
1348
Willy Tarreau8fccfa22014-06-14 08:28:06 +02001349 hdr->len = htons((uint16_t)(ret - PP2_HEADER_LEN));
David Safb76832014-05-08 23:42:08 -04001350
Emmanuel Hocdet4399c752018-02-05 15:26:43 +01001351 if (tlv_crc32c_p) {
1352 write_u32(tlv_crc32c_p, htonl(hash_crc32c(buf, ret)));
1353 }
1354
David Safb76832014-05-08 23:42:08 -04001355 return ret;
1356}
Emeric Brun4f603012017-01-05 15:11:44 +01001357
Willy Tarreau119e50e2020-05-22 13:53:29 +02001358/* returns 0 on success */
1359static int cfg_parse_pp2_never_send_local(char **args, int section_type, struct proxy *curpx,
1360 struct proxy *defpx, const char *file, int line,
1361 char **err)
1362{
1363 if (too_many_args(0, args, err, NULL))
1364 return -1;
1365 pp2_never_send_local = 1;
1366 return 0;
1367}
1368
Willy Tarreau60ca10a2017-08-18 15:26:54 +02001369/* return the major HTTP version as 1 or 2 depending on how the request arrived
1370 * before being processed.
1371 */
1372static int
1373smp_fetch_fc_http_major(const struct arg *args, struct sample *smp, const char *kw, void *private)
1374{
Jérôme Magnin86577422018-12-07 09:03:11 +01001375 struct connection *conn = (kw[0] != 'b') ? objt_conn(smp->sess->origin) :
1376 smp->strm ? cs_conn(objt_cs(smp->strm->si[1].end)) : NULL;
Willy Tarreau60ca10a2017-08-18 15:26:54 +02001377
1378 smp->data.type = SMP_T_SINT;
1379 smp->data.u.sint = (conn && strcmp(conn_get_mux_name(conn), "H2") == 0) ? 2 : 1;
1380 return 1;
1381}
1382
Emeric Brun4f603012017-01-05 15:11:44 +01001383/* fetch if the received connection used a PROXY protocol header */
1384int smp_fetch_fc_rcvd_proxy(const struct arg *args, struct sample *smp, const char *kw, void *private)
1385{
1386 struct connection *conn;
1387
1388 conn = objt_conn(smp->sess->origin);
1389 if (!conn)
1390 return 0;
1391
Willy Tarreau911db9b2020-01-23 16:27:54 +01001392 if (conn->flags & CO_FL_WAIT_XPRT) {
Emeric Brun4f603012017-01-05 15:11:44 +01001393 smp->flags |= SMP_F_MAY_CHANGE;
1394 return 0;
1395 }
1396
1397 smp->flags = 0;
1398 smp->data.type = SMP_T_BOOL;
1399 smp->data.u.sint = (conn->flags & CO_FL_RCVD_PROXY) ? 1 : 0;
1400
1401 return 1;
1402}
1403
Geoff Simmons7185b782019-08-27 18:31:16 +02001404/* fetch the authority TLV from a PROXY protocol header */
1405int smp_fetch_fc_pp_authority(const struct arg *args, struct sample *smp, const char *kw, void *private)
1406{
1407 struct connection *conn;
1408
1409 conn = objt_conn(smp->sess->origin);
1410 if (!conn)
1411 return 0;
1412
Willy Tarreau911db9b2020-01-23 16:27:54 +01001413 if (conn->flags & CO_FL_WAIT_XPRT) {
Geoff Simmons7185b782019-08-27 18:31:16 +02001414 smp->flags |= SMP_F_MAY_CHANGE;
1415 return 0;
1416 }
1417
1418 if (conn->proxy_authority == NULL)
1419 return 0;
1420
1421 smp->flags = 0;
1422 smp->data.type = SMP_T_STR;
1423 smp->data.u.str.area = conn->proxy_authority;
1424 smp->data.u.str.data = conn->proxy_authority_len;
1425
1426 return 1;
1427}
1428
Tim Duesterhusd1b15b62020-03-13 12:34:23 +01001429/* fetch the unique ID TLV from a PROXY protocol header */
1430int smp_fetch_fc_pp_unique_id(const struct arg *args, struct sample *smp, const char *kw, void *private)
1431{
1432 struct connection *conn;
1433
1434 conn = objt_conn(smp->sess->origin);
1435 if (!conn)
1436 return 0;
1437
1438 if (conn->flags & CO_FL_WAIT_XPRT) {
1439 smp->flags |= SMP_F_MAY_CHANGE;
1440 return 0;
1441 }
1442
1443 if (!isttest(conn->proxy_unique_id))
1444 return 0;
1445
1446 smp->flags = 0;
1447 smp->data.type = SMP_T_STR;
1448 smp->data.u.str.area = conn->proxy_unique_id.ptr;
1449 smp->data.u.str.data = conn->proxy_unique_id.len;
1450
1451 return 1;
1452}
1453
Emeric Brun4f603012017-01-05 15:11:44 +01001454/* Note: must not be declared <const> as its list will be overwritten.
1455 * Note: fetches that may return multiple types must be declared as the lowest
1456 * common denominator, the type that can be casted into all other ones. For
1457 * instance v4/v6 must be declared v4.
1458 */
1459static struct sample_fetch_kw_list sample_fetch_keywords = {ILH, {
Willy Tarreau60ca10a2017-08-18 15:26:54 +02001460 { "fc_http_major", smp_fetch_fc_http_major, 0, NULL, SMP_T_SINT, SMP_USE_L4CLI },
Jérôme Magnin86577422018-12-07 09:03:11 +01001461 { "bc_http_major", smp_fetch_fc_http_major, 0, NULL, SMP_T_SINT, SMP_USE_L4SRV },
Emeric Brun4f603012017-01-05 15:11:44 +01001462 { "fc_rcvd_proxy", smp_fetch_fc_rcvd_proxy, 0, NULL, SMP_T_BOOL, SMP_USE_L4CLI },
Geoff Simmons7185b782019-08-27 18:31:16 +02001463 { "fc_pp_authority", smp_fetch_fc_pp_authority, 0, NULL, SMP_T_STR, SMP_USE_L4CLI },
Tim Duesterhusd1b15b62020-03-13 12:34:23 +01001464 { "fc_pp_unique_id", smp_fetch_fc_pp_unique_id, 0, NULL, SMP_T_STR, SMP_USE_L4CLI },
Emeric Brun4f603012017-01-05 15:11:44 +01001465 { /* END */ },
1466}};
1467
Willy Tarreau0108d902018-11-25 19:14:37 +01001468INITCALL1(STG_REGISTER, sample_register_fetches, &sample_fetch_keywords);
Willy Tarreau119e50e2020-05-22 13:53:29 +02001469
1470static struct cfg_kw_list cfg_kws = {ILH, {
1471 { CFG_GLOBAL, "pp2-never-send-local", cfg_parse_pp2_never_send_local },
1472 { /* END */ },
1473}};
1474
1475INITCALL1(STG_REGISTER, cfg_register_keywords, &cfg_kws);