blob: 3c396958a39bbeca3f71d725d7c26dbec66c8e5f [file] [log] [blame]
Emeric Brun2b920a12010-09-23 18:30:22 +02001/*
Emeric Brunb3971ab2015-05-12 18:49:09 +02002 * Peer synchro management.
Emeric Brun2b920a12010-09-23 18:30:22 +02003 *
4 * Copyright 2010 EXCELIANCE, Emeric Brun <ebrun@exceliance.fr>
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 <errno.h>
Emeric Brun2b920a12010-09-23 18:30:22 +020014#include <stdio.h>
15#include <stdlib.h>
16#include <string.h>
17
18#include <sys/socket.h>
19#include <sys/stat.h>
20#include <sys/types.h>
21
Willy Tarreau8db34cc2021-10-06 17:53:19 +020022#include <import/eb32tree.h>
23#include <import/ebmbtree.h>
24#include <import/ebpttree.h>
25
Willy Tarreau4c7e4b72020-05-27 12:58:42 +020026#include <haproxy/api.h>
Willy Tarreaub2551052020-06-09 09:07:15 +020027#include <haproxy/applet.h>
Willy Tarreauf1d32c42020-06-04 21:07:02 +020028#include <haproxy/channel.h>
Willy Tarreau83487a82020-06-04 20:19:54 +020029#include <haproxy/cli.h>
Christopher Faulet908628c2022-03-25 16:43:49 +010030#include <haproxy/conn_stream.h>
31#include <haproxy/cs_utils.h>
Willy Tarreau3afc4c42020-06-03 18:23:19 +020032#include <haproxy/dict.h>
Willy Tarreau36979d92020-06-05 17:27:29 +020033#include <haproxy/errors.h>
Willy Tarreaudfd3de82020-06-04 23:46:14 +020034#include <haproxy/fd.h>
Willy Tarreau762d7a52020-06-04 11:23:07 +020035#include <haproxy/frontend.h>
Willy Tarreau6131d6a2020-06-02 16:48:09 +020036#include <haproxy/net_helper.h>
Willy Tarreau8efbdfb2020-06-04 11:29:21 +020037#include <haproxy/obj_type-t.h>
Willy Tarreau3c2a7c22020-06-04 18:38:21 +020038#include <haproxy/peers.h>
Willy Tarreaua264d962020-06-04 22:29:18 +020039#include <haproxy/proxy.h>
Willy Tarreau48d25b32020-06-04 18:58:52 +020040#include <haproxy/session-t.h>
Willy Tarreau3727a8a2020-06-04 17:37:26 +020041#include <haproxy/signal.h>
Willy Tarreau2eec9b52020-06-04 19:58:55 +020042#include <haproxy/stats-t.h>
Willy Tarreaudfd3de82020-06-04 23:46:14 +020043#include <haproxy/stick_table.h>
44#include <haproxy/stream.h>
Willy Tarreaub2551052020-06-09 09:07:15 +020045#include <haproxy/task.h>
46#include <haproxy/thread.h>
Willy Tarreau92b4f132020-06-01 11:05:15 +020047#include <haproxy/time.h>
Willy Tarreau48fbcae2020-06-03 18:09:46 +020048#include <haproxy/tools.h>
Frédéric Lécailled8659352020-11-10 16:18:03 +010049#include <haproxy/trace.h>
Emeric Brun2b920a12010-09-23 18:30:22 +020050
Emeric Brun2b920a12010-09-23 18:30:22 +020051
52/*******************************/
53/* Current peer learning state */
54/*******************************/
55
56/******************************/
Emeric Brunb3971ab2015-05-12 18:49:09 +020057/* Current peers section resync state */
Emeric Brun2b920a12010-09-23 18:30:22 +020058/******************************/
Emeric Brunccdfbae2021-04-28 12:59:35 +020059#define PEERS_F_RESYNC_LOCAL 0x00000001 /* Learn from local finished or no more needed */
60#define PEERS_F_RESYNC_REMOTE 0x00000002 /* Learn from remote finished or no more needed */
61#define PEERS_F_RESYNC_ASSIGN 0x00000004 /* A peer was assigned to learn our lesson */
62#define PEERS_F_RESYNC_PROCESS 0x00000008 /* The assigned peer was requested for resync */
63#define PEERS_F_RESYNC_LOCALTIMEOUT 0x00000010 /* Timeout waiting for a full resync from a local node */
64#define PEERS_F_RESYNC_REMOTETIMEOUT 0x00000020 /* Timeout waiting for a full resync from a remote node */
65#define PEERS_F_RESYNC_LOCALABORT 0x00000040 /* Session aborted learning from a local node */
66#define PEERS_F_RESYNC_REMOTEABORT 0x00000080 /* Session aborted learning from a remote node */
67#define PEERS_F_RESYNC_LOCALFINISHED 0x00000100 /* A local node teach us and was fully up to date */
68#define PEERS_F_RESYNC_REMOTEFINISHED 0x00000200 /* A remote node teach us and was fully up to date */
69#define PEERS_F_RESYNC_LOCALPARTIAL 0x00000400 /* A local node teach us but was partially up to date */
70#define PEERS_F_RESYNC_REMOTEPARTIAL 0x00000800 /* A remote node teach us but was partially up to date */
71#define PEERS_F_RESYNC_LOCALASSIGN 0x00001000 /* A local node was assigned for a full resync */
72#define PEERS_F_RESYNC_REMOTEASSIGN 0x00002000 /* A remote node was assigned for a full resync */
73#define PEERS_F_RESYNC_REQUESTED 0x00004000 /* A resync was explicitly requested */
74#define PEERS_F_DONOTSTOP 0x00010000 /* Main table sync task block process during soft stop
75 to push data to new process */
Emeric Brun2b920a12010-09-23 18:30:22 +020076
Frédéric Lécailleaba44a22019-03-26 10:18:07 +010077#define PEERS_RESYNC_STATEMASK (PEERS_F_RESYNC_LOCAL|PEERS_F_RESYNC_REMOTE)
78#define PEERS_RESYNC_FROMLOCAL 0x00000000
79#define PEERS_RESYNC_FROMREMOTE PEERS_F_RESYNC_LOCAL
80#define PEERS_RESYNC_FINISHED (PEERS_F_RESYNC_LOCAL|PEERS_F_RESYNC_REMOTE)
Emeric Brunb3971ab2015-05-12 18:49:09 +020081
82/***********************************/
83/* Current shared table sync state */
84/***********************************/
Frédéric Lécailleaba44a22019-03-26 10:18:07 +010085#define SHTABLE_F_TEACH_STAGE1 0x00000001 /* Teach state 1 complete */
86#define SHTABLE_F_TEACH_STAGE2 0x00000002 /* Teach state 2 complete */
Emeric Brun2b920a12010-09-23 18:30:22 +020087
88/******************************/
89/* Remote peer teaching state */
90/******************************/
Frédéric Lécailleaba44a22019-03-26 10:18:07 +010091#define PEER_F_TEACH_PROCESS 0x00000001 /* Teach a lesson to current peer */
92#define PEER_F_TEACH_FINISHED 0x00000008 /* Teach conclude, (wait for confirm) */
93#define PEER_F_TEACH_COMPLETE 0x00000010 /* All that we know already taught to current peer, used only for a local peer */
94#define PEER_F_LEARN_ASSIGN 0x00000100 /* Current peer was assigned for a lesson */
95#define PEER_F_LEARN_NOTUP2DATE 0x00000200 /* Learn from peer finished but peer is not up to date */
96#define PEER_F_ALIVE 0x20000000 /* Used to flag a peer a alive. */
97#define PEER_F_HEARTBEAT 0x40000000 /* Heartbeat message to send. */
98#define PEER_F_DWNGRD 0x80000000 /* When this flag is enabled, we must downgrade the supported version announced during peer sessions. */
Emeric Brun2b920a12010-09-23 18:30:22 +020099
Frédéric Lécailleaba44a22019-03-26 10:18:07 +0100100#define PEER_TEACH_RESET ~(PEER_F_TEACH_PROCESS|PEER_F_TEACH_FINISHED) /* PEER_F_TEACH_COMPLETE should never be reset */
101#define PEER_LEARN_RESET ~(PEER_F_LEARN_ASSIGN|PEER_F_LEARN_NOTUP2DATE)
Emeric Brun2b920a12010-09-23 18:30:22 +0200102
Frédéric Lécaille54bff832019-03-26 10:25:20 +0100103#define PEER_RESYNC_TIMEOUT 5000 /* 5 seconds */
104#define PEER_RECONNECT_TIMEOUT 5000 /* 5 seconds */
Frédéric Lécaille645635d2019-02-11 17:49:39 +0100105#define PEER_HEARTBEAT_TIMEOUT 3000 /* 3 seconds */
106
Willy Tarreau49962b52021-02-12 16:56:22 +0100107/* flags for "show peers" */
108#define PEERS_SHOW_F_DICT 0x00000001 /* also show the contents of the dictionary */
109
Emeric Brunb3971ab2015-05-12 18:49:09 +0200110/*****************************/
111/* Sync message class */
112/*****************************/
113enum {
114 PEER_MSG_CLASS_CONTROL = 0,
115 PEER_MSG_CLASS_ERROR,
116 PEER_MSG_CLASS_STICKTABLE = 10,
117 PEER_MSG_CLASS_RESERVED = 255,
118};
119
120/*****************************/
121/* control message types */
122/*****************************/
123enum {
124 PEER_MSG_CTRL_RESYNCREQ = 0,
125 PEER_MSG_CTRL_RESYNCFINISHED,
126 PEER_MSG_CTRL_RESYNCPARTIAL,
127 PEER_MSG_CTRL_RESYNCCONFIRM,
Frédéric Lécaille645635d2019-02-11 17:49:39 +0100128 PEER_MSG_CTRL_HEARTBEAT,
Emeric Brunb3971ab2015-05-12 18:49:09 +0200129};
130
131/*****************************/
132/* error message types */
133/*****************************/
134enum {
135 PEER_MSG_ERR_PROTOCOL = 0,
136 PEER_MSG_ERR_SIZELIMIT,
137};
138
Emeric Brun530ba382020-06-02 11:17:42 +0200139/* network key types;
140 * network types were directly and mistakenly
141 * mapped on sample types, to keep backward
142 * compatiblitiy we keep those values but
143 * we now use a internal/network mapping
144 * to avoid further mistakes adding or
145 * modifying internals types
146 */
147enum {
148 PEER_KT_ANY = 0, /* any type */
149 PEER_KT_RESV1, /* UNUSED */
150 PEER_KT_SINT, /* signed 64bits integer type */
151 PEER_KT_RESV3, /* UNUSED */
152 PEER_KT_IPV4, /* ipv4 type */
153 PEER_KT_IPV6, /* ipv6 type */
154 PEER_KT_STR, /* char string type */
155 PEER_KT_BIN, /* buffer type */
156 PEER_KT_TYPES /* number of types, must always be last */
157};
158
159/* Map used to retrieve network type from internal type
160 * Note: Undeclared mapping maps entry to PEER_KT_ANY == 0
161 */
162static int peer_net_key_type[SMP_TYPES] = {
163 [SMP_T_SINT] = PEER_KT_SINT,
164 [SMP_T_IPV4] = PEER_KT_IPV4,
165 [SMP_T_IPV6] = PEER_KT_IPV6,
166 [SMP_T_STR] = PEER_KT_STR,
167 [SMP_T_BIN] = PEER_KT_BIN,
168};
169
170/* Map used to retrieve internal type from external type
171 * Note: Undeclared mapping maps entry to SMP_T_ANY == 0
172 */
173static int peer_int_key_type[PEER_KT_TYPES] = {
174 [PEER_KT_SINT] = SMP_T_SINT,
175 [PEER_KT_IPV4] = SMP_T_IPV4,
176 [PEER_KT_IPV6] = SMP_T_IPV6,
177 [PEER_KT_STR] = SMP_T_STR,
178 [PEER_KT_BIN] = SMP_T_BIN,
179};
180
Frédéric Lécailled5fe14b2019-01-24 10:33:40 +0100181/*
182 * Parameters used by functions to build peer protocol messages. */
183struct peer_prep_params {
184 struct {
Frédéric Lécaille7d0ceee2019-01-24 14:24:05 +0100185 struct peer *peer;
186 } hello;
187 struct {
188 unsigned int st1;
189 } error_status;
190 struct {
Frédéric Lécailled5fe14b2019-01-24 10:33:40 +0100191 struct stksess *stksess;
192 struct shared_table *shared_table;
193 unsigned int updateid;
194 int use_identifier;
195 int use_timed;
Frédéric Lécaille8d78fa72019-05-20 18:22:52 +0200196 struct peer *peer;
Frédéric Lécailled5fe14b2019-01-24 10:33:40 +0100197 } updt;
198 struct {
199 struct shared_table *shared_table;
200 } swtch;
201 struct {
202 struct shared_table *shared_table;
203 } ack;
Frédéric Lécaille7d0ceee2019-01-24 14:24:05 +0100204 struct {
205 unsigned char head[2];
206 } control;
207 struct {
208 unsigned char head[2];
209 } error;
Frédéric Lécailled5fe14b2019-01-24 10:33:40 +0100210};
Emeric Brunb3971ab2015-05-12 18:49:09 +0200211
212/*******************************/
213/* stick table sync mesg types */
214/* Note: ids >= 128 contains */
Ilya Shipitsinc02a23f2020-05-06 00:53:22 +0500215/* id message contains data */
Emeric Brunb3971ab2015-05-12 18:49:09 +0200216/*******************************/
Olivier Houchard33992262018-10-16 18:49:26 +0200217#define PEER_MSG_STKT_UPDATE 0x80
218#define PEER_MSG_STKT_INCUPDATE 0x81
219#define PEER_MSG_STKT_DEFINE 0x82
220#define PEER_MSG_STKT_SWITCH 0x83
221#define PEER_MSG_STKT_ACK 0x84
222#define PEER_MSG_STKT_UPDATE_TIMED 0x85
223#define PEER_MSG_STKT_INCUPDATE_TIMED 0x86
Frédéric Lécaille36fb77e2019-06-04 08:28:19 +0200224/* All the stick-table message identifiers abova have the #7 bit set */
225#define PEER_MSG_STKT_BIT 7
226#define PEER_MSG_STKT_BIT_MASK (1 << PEER_MSG_STKT_BIT)
Emeric Brun2b920a12010-09-23 18:30:22 +0200227
Frédéric Lécaille39143342019-05-24 14:32:27 +0200228/* The maximum length of an encoded data length. */
229#define PEER_MSG_ENC_LENGTH_MAXLEN 5
230
Frédéric Lécaille32b55732019-06-03 18:29:51 +0200231/* Minimum 64-bits value encoded with 2 bytes */
232#define PEER_ENC_2BYTES_MIN 0xf0 /* 0xf0 (or 240) */
233/* 3 bytes */
234#define PEER_ENC_3BYTES_MIN ((1ULL << 11) | PEER_ENC_2BYTES_MIN) /* 0x8f0 (or 2288) */
235/* 4 bytes */
236#define PEER_ENC_4BYTES_MIN ((1ULL << 18) | PEER_ENC_3BYTES_MIN) /* 0x408f0 (or 264432) */
237/* 5 bytes */
238#define PEER_ENC_5BYTES_MIN ((1ULL << 25) | PEER_ENC_4BYTES_MIN) /* 0x20408f0 (or 33818864) */
239/* 6 bytes */
240#define PEER_ENC_6BYTES_MIN ((1ULL << 32) | PEER_ENC_5BYTES_MIN) /* 0x1020408f0 (or 4328786160) */
241/* 7 bytes */
242#define PEER_ENC_7BYTES_MIN ((1ULL << 39) | PEER_ENC_6BYTES_MIN) /* 0x81020408f0 (or 554084600048) */
243/* 8 bytes */
244#define PEER_ENC_8BYTES_MIN ((1ULL << 46) | PEER_ENC_7BYTES_MIN) /* 0x4081020408f0 (or 70922828777712) */
245/* 9 bytes */
246#define PEER_ENC_9BYTES_MIN ((1ULL << 53) | PEER_ENC_8BYTES_MIN) /* 0x204081020408f0 (or 9078122083518704) */
247/* 10 bytes */
248#define PEER_ENC_10BYTES_MIN ((1ULL << 60) | PEER_ENC_9BYTES_MIN) /* 0x10204081020408f0 (or 1161999626690365680) */
249
250/* #7 bit used to detect the last byte to be encoded */
251#define PEER_ENC_STOP_BIT 7
252/* The byte minimum value with #7 bit set */
253#define PEER_ENC_STOP_BYTE (1 << PEER_ENC_STOP_BIT)
254/* The left most number of bits set for PEER_ENC_2BYTES_MIN */
255#define PEER_ENC_2BYTES_MIN_BITS 4
256
Frédéric Lécaille39143342019-05-24 14:32:27 +0200257#define PEER_MSG_HEADER_LEN 2
258
Frédéric Lécaille8d78fa72019-05-20 18:22:52 +0200259#define PEER_STKT_CACHE_MAX_ENTRIES 128
260
Emeric Brun2b920a12010-09-23 18:30:22 +0200261/**********************************/
262/* Peer Session IO handler states */
263/**********************************/
264
Willy Tarreaue4d927a2013-12-01 12:47:35 +0100265enum {
266 PEER_SESS_ST_ACCEPT = 0, /* Initial state for session create by an accept, must be zero! */
267 PEER_SESS_ST_GETVERSION, /* Validate supported protocol version */
268 PEER_SESS_ST_GETHOST, /* Validate host ID correspond to local host id */
269 PEER_SESS_ST_GETPEER, /* Validate peer ID correspond to a known remote peer id */
Willy Tarreaue4d927a2013-12-01 12:47:35 +0100270 /* after this point, data were possibly exchanged */
271 PEER_SESS_ST_SENDSUCCESS, /* Send ret code 200 (success) and wait for message */
272 PEER_SESS_ST_CONNECT, /* Initial state for session create on a connect, push presentation into buffer */
273 PEER_SESS_ST_GETSTATUS, /* Wait for the welcome message */
274 PEER_SESS_ST_WAITMSG, /* Wait for data messages */
275 PEER_SESS_ST_EXIT, /* Exit with status code */
Emeric Brunb3971ab2015-05-12 18:49:09 +0200276 PEER_SESS_ST_ERRPROTO, /* Send error proto message before exit */
277 PEER_SESS_ST_ERRSIZE, /* Send error size message before exit */
Willy Tarreaue4d927a2013-12-01 12:47:35 +0100278 PEER_SESS_ST_END, /* Killed session */
279};
Emeric Brun2b920a12010-09-23 18:30:22 +0200280
Willy Tarreaue4d927a2013-12-01 12:47:35 +0100281/***************************************************/
282/* Peer Session status code - part of the protocol */
283/***************************************************/
Emeric Brun2b920a12010-09-23 18:30:22 +0200284
Frédéric Lécailleaba44a22019-03-26 10:18:07 +0100285#define PEER_SESS_SC_CONNECTCODE 100 /* connect in progress */
286#define PEER_SESS_SC_CONNECTEDCODE 110 /* tcp connect success */
Emeric Brun2b920a12010-09-23 18:30:22 +0200287
Frédéric Lécailleaba44a22019-03-26 10:18:07 +0100288#define PEER_SESS_SC_SUCCESSCODE 200 /* accept or connect successful */
Emeric Brun2b920a12010-09-23 18:30:22 +0200289
Frédéric Lécailleaba44a22019-03-26 10:18:07 +0100290#define PEER_SESS_SC_TRYAGAIN 300 /* try again later */
Emeric Brun2b920a12010-09-23 18:30:22 +0200291
Frédéric Lécailleaba44a22019-03-26 10:18:07 +0100292#define PEER_SESS_SC_ERRPROTO 501 /* error protocol */
293#define PEER_SESS_SC_ERRVERSION 502 /* unknown protocol version */
294#define PEER_SESS_SC_ERRHOST 503 /* bad host name */
295#define PEER_SESS_SC_ERRPEER 504 /* unknown peer */
Emeric Brun2b920a12010-09-23 18:30:22 +0200296
297#define PEER_SESSION_PROTO_NAME "HAProxyS"
Frédéric Lécaille523cc9e2016-10-12 17:30:30 +0200298#define PEER_MAJOR_VER 2
299#define PEER_MINOR_VER 1
300#define PEER_DWNGRD_MINOR_VER 0
Emeric Brun2b920a12010-09-23 18:30:22 +0200301
Willy Tarreau6254a922019-01-29 17:45:23 +0100302static size_t proto_len = sizeof(PEER_SESSION_PROTO_NAME) - 1;
Frédéric Lécailleed2b4a62017-07-13 09:07:09 +0200303struct peers *cfg_peers = NULL;
Emeric Brun9ef2ad72019-04-02 17:22:01 +0200304static void peer_session_forceshutdown(struct peer *peer);
Emeric Brun2b920a12010-09-23 18:30:22 +0200305
Frédéric Lécaille6c391982019-06-06 11:34:03 +0200306static struct ebpt_node *dcache_tx_insert(struct dcache *dc,
307 struct dcache_tx_entry *i);
Frédéric Lécaille8d78fa72019-05-20 18:22:52 +0200308static inline void flush_dcache(struct peer *peer);
309
Frédéric Lécailled8659352020-11-10 16:18:03 +0100310/* trace source and events */
311static void peers_trace(enum trace_level level, uint64_t mask,
312 const struct trace_source *src,
313 const struct ist where, const struct ist func,
314 const void *a1, const void *a2, const void *a3, const void *a4);
315
316static const struct trace_event peers_trace_events[] = {
317#define PEERS_EV_UPDTMSG (1 << 0)
318 { .mask = PEERS_EV_UPDTMSG, .name = "updtmsg", .desc = "update message received" },
Frédéric Lécailleda2b0842021-01-15 16:21:28 +0100319#define PEERS_EV_ACKMSG (1 << 1)
320 { .mask = PEERS_EV_ACKMSG, .name = "ackmsg", .desc = "ack message received" },
321#define PEERS_EV_SWTCMSG (1 << 2)
322 { .mask = PEERS_EV_SWTCMSG, .name = "swtcmsg", .desc = "switch message received" },
323#define PEERS_EV_DEFMSG (1 << 3)
324 { .mask = PEERS_EV_DEFMSG, .name = "defmsg", .desc = "definition message received" },
325#define PEERS_EV_CTRLMSG (1 << 4)
326 { .mask = PEERS_EV_CTRLMSG, .name = "ctrlmsg", .desc = "control message sent/received" },
327#define PEERS_EV_SESSREL (1 << 5)
328 { .mask = PEERS_EV_SESSREL, .name = "sessrl", .desc = "peer session releasing" },
329#define PEERS_EV_PROTOERR (1 << 6)
330 { .mask = PEERS_EV_PROTOERR, .name = "protoerr", .desc = "protocol error" },
Frédéric Lécailled8659352020-11-10 16:18:03 +0100331};
332
333static const struct name_desc peers_trace_lockon_args[4] = {
334 /* arg1 */ { /* already used by the connection */ },
335 /* arg2 */ { .name="peers", .desc="Peers protocol" },
336 /* arg3 */ { },
337 /* arg4 */ { }
338};
339
340static const struct name_desc peers_trace_decoding[] = {
341#define PEERS_VERB_CLEAN 1
342 { .name="clean", .desc="only user-friendly stuff, generally suitable for level \"user\"" },
343 { /* end */ }
344};
345
346
347struct trace_source trace_peers = {
348 .name = IST("peers"),
349 .desc = "Peers protocol",
350 .arg_def = TRC_ARG1_CONN, /* TRACE()'s first argument is always a connection */
351 .default_cb = peers_trace,
352 .known_events = peers_trace_events,
353 .lockon_args = peers_trace_lockon_args,
354 .decoding = peers_trace_decoding,
355 .report_events = ~0, /* report everything by default */
356};
357
Frédéric Lécailleda2b0842021-01-15 16:21:28 +0100358/* Return peer control message types as strings (only for debugging purpose). */
359static inline char *ctrl_msg_type_str(unsigned int type)
360{
361 switch (type) {
362 case PEER_MSG_CTRL_RESYNCREQ:
363 return "RESYNCREQ";
364 case PEER_MSG_CTRL_RESYNCFINISHED:
365 return "RESYNCFINISHED";
366 case PEER_MSG_CTRL_RESYNCPARTIAL:
367 return "RESYNCPARTIAL";
368 case PEER_MSG_CTRL_RESYNCCONFIRM:
369 return "RESYNCCONFIRM";
370 case PEER_MSG_CTRL_HEARTBEAT:
371 return "HEARTBEAT";
372 default:
373 return "???";
374 }
375}
376
Frédéric Lécailled8659352020-11-10 16:18:03 +0100377#define TRACE_SOURCE &trace_peers
378INITCALL1(STG_REGISTER, trace_register_source, TRACE_SOURCE);
379
380static void peers_trace(enum trace_level level, uint64_t mask,
381 const struct trace_source *src,
382 const struct ist where, const struct ist func,
383 const void *a1, const void *a2, const void *a3, const void *a4)
384{
Frédéric Lécailleda2b0842021-01-15 16:21:28 +0100385 if (mask & (PEERS_EV_UPDTMSG|PEERS_EV_ACKMSG|PEERS_EV_SWTCMSG)) {
Frédéric Lécailled8659352020-11-10 16:18:03 +0100386 if (a2) {
387 const struct peer *peer = a2;
388
389 chunk_appendf(&trace_buf, " peer=%s", peer->id);
390 }
391 if (a3) {
392 const char *p = a3;
393
394 chunk_appendf(&trace_buf, " @%p", p);
395 }
396 if (a4) {
397 const size_t *val = a4;
398
399 chunk_appendf(&trace_buf, " %llu", (unsigned long long)*val);
400 }
401 }
Frédéric Lécailleda2b0842021-01-15 16:21:28 +0100402
403 if (mask & PEERS_EV_DEFMSG) {
404 if (a2) {
405 const struct peer *peer = a2;
406
407 chunk_appendf(&trace_buf, " peer=%s", peer->id);
408 }
409 if (a3) {
410 const char *p = a3;
411
412 chunk_appendf(&trace_buf, " @%p", p);
413 }
414 if (a4) {
415 const int *val = a4;
416
417 chunk_appendf(&trace_buf, " %d", *val);
418 }
419 }
420
421 if (mask & PEERS_EV_CTRLMSG) {
422 if (a2) {
423 const unsigned char *ctrl_msg_type = a2;
424
425 chunk_appendf(&trace_buf, " %s", ctrl_msg_type_str(*ctrl_msg_type));
426
427 }
428 if (a3) {
429 const char *local_peer = a3;
430
431 chunk_appendf(&trace_buf, " %s", local_peer);
432 }
433
434 if (a4) {
435 const char *remote_peer = a4;
436
437 chunk_appendf(&trace_buf, " -> %s", remote_peer);
438 }
439 }
440
441 if (mask & (PEERS_EV_SESSREL|PEERS_EV_PROTOERR)) {
442 if (a2) {
443 const struct peer *peer = a2;
444 struct peers *peers = NULL;
445
Christopher Fauletd9e6b352021-11-15 09:40:57 +0100446 if (peer->appctx) {
Christopher Faulet908628c2022-03-25 16:43:49 +0100447 struct stream *s = __cs_strm(peer->appctx->owner);
Frédéric Lécailleda2b0842021-01-15 16:21:28 +0100448
Christopher Faulet908628c2022-03-25 16:43:49 +0100449 peers = strm_fe(s)->parent;
Frédéric Lécailleda2b0842021-01-15 16:21:28 +0100450 }
451
452 if (peers)
453 chunk_appendf(&trace_buf, " %s", peers->local->id);
Christopher Fauletd9e6b352021-11-15 09:40:57 +0100454 chunk_appendf(&trace_buf, " -> %s", peer->id);
Frédéric Lécailleda2b0842021-01-15 16:21:28 +0100455 }
456
457 if (a3) {
458 const int *prev_state = a3;
459
460 chunk_appendf(&trace_buf, " prev_state=%d\n", *prev_state);
461 }
462 }
Frédéric Lécailled8659352020-11-10 16:18:03 +0100463}
464
Frédéric Lécaille95679dc2019-04-15 10:25:27 +0200465static const char *statuscode_str(int statuscode)
466{
467 switch (statuscode) {
468 case PEER_SESS_SC_CONNECTCODE:
469 return "CONN";
470 case PEER_SESS_SC_CONNECTEDCODE:
471 return "HSHK";
472 case PEER_SESS_SC_SUCCESSCODE:
473 return "ESTA";
474 case PEER_SESS_SC_TRYAGAIN:
475 return "RETR";
476 case PEER_SESS_SC_ERRPROTO:
477 return "PROT";
478 case PEER_SESS_SC_ERRVERSION:
479 return "VERS";
480 case PEER_SESS_SC_ERRHOST:
481 return "NAME";
482 case PEER_SESS_SC_ERRPEER:
483 return "UNKN";
484 default:
485 return "NONE";
486 }
487}
488
Emeric Brun18928af2017-03-29 16:32:53 +0200489/* This function encode an uint64 to 'dynamic' length format.
490 The encoded value is written at address *str, and the
Ilya Shipitsinc02a23f2020-05-06 00:53:22 +0500491 caller must assure that size after *str is large enough.
Emeric Brun18928af2017-03-29 16:32:53 +0200492 At return, the *str is set at the next Byte after then
493 encoded integer. The function returns then length of the
494 encoded integer in Bytes */
Emeric Brunb3971ab2015-05-12 18:49:09 +0200495int intencode(uint64_t i, char **str) {
496 int idx = 0;
497 unsigned char *msg;
498
Emeric Brunb3971ab2015-05-12 18:49:09 +0200499 msg = (unsigned char *)*str;
Frédéric Lécaille32b55732019-06-03 18:29:51 +0200500 if (i < PEER_ENC_2BYTES_MIN) {
Emeric Brunb3971ab2015-05-12 18:49:09 +0200501 msg[0] = (unsigned char)i;
502 *str = (char *)&msg[idx+1];
503 return (idx+1);
504 }
505
Frédéric Lécaille32b55732019-06-03 18:29:51 +0200506 msg[idx] =(unsigned char)i | PEER_ENC_2BYTES_MIN;
507 i = (i - PEER_ENC_2BYTES_MIN) >> PEER_ENC_2BYTES_MIN_BITS;
508 while (i >= PEER_ENC_STOP_BYTE) {
509 msg[++idx] = (unsigned char)i | PEER_ENC_STOP_BYTE;
510 i = (i - PEER_ENC_STOP_BYTE) >> PEER_ENC_STOP_BIT;
Emeric Brunb3971ab2015-05-12 18:49:09 +0200511 }
512 msg[++idx] = (unsigned char)i;
513 *str = (char *)&msg[idx+1];
514 return (idx+1);
515}
516
517
Emeric Brun5ea07d92021-06-30 13:21:58 +0200518/* This function returns a decoded 64bits unsigned integer
519 * from a varint
520 *
521 * Calling:
522 * - *str must point on the first byte of the buffer to decode.
523 * - end must point on the next byte after the end of the buffer
524 * we are authorized to parse (buf + buflen)
525 *
526 * At return:
527 *
528 * On success *str will point at the byte following
529 * the fully decoded integer into the buffer. and
530 * the decoded value is returned.
531 *
532 * If end is reached before the integer was fully decoded,
533 * *str is set to NULL and the caller have to check this
534 * to know there is a decoding error. In this case
535 * the returned integer is also forced to 0
536 */
Emeric Brun18928af2017-03-29 16:32:53 +0200537uint64_t intdecode(char **str, char *end)
538{
Emeric Brunb3971ab2015-05-12 18:49:09 +0200539 unsigned char *msg;
Emeric Brun18928af2017-03-29 16:32:53 +0200540 uint64_t i;
541 int shift;
Emeric Brunb3971ab2015-05-12 18:49:09 +0200542
543 if (!*str)
544 return 0;
545
546 msg = (unsigned char *)*str;
Emeric Brun18928af2017-03-29 16:32:53 +0200547 if (msg >= (unsigned char *)end)
548 goto fail;
Emeric Brunb3971ab2015-05-12 18:49:09 +0200549
Emeric Brun18928af2017-03-29 16:32:53 +0200550 i = *(msg++);
Frédéric Lécaille32b55732019-06-03 18:29:51 +0200551 if (i >= PEER_ENC_2BYTES_MIN) {
552 shift = PEER_ENC_2BYTES_MIN_BITS;
Emeric Brun18928af2017-03-29 16:32:53 +0200553 do {
554 if (msg >= (unsigned char *)end)
555 goto fail;
556 i += (uint64_t)*msg << shift;
Frédéric Lécaille32b55732019-06-03 18:29:51 +0200557 shift += PEER_ENC_STOP_BIT;
558 } while (*(msg++) >= PEER_ENC_STOP_BYTE);
Emeric Brunb3971ab2015-05-12 18:49:09 +0200559 }
Frédéric Lécaillea8725ec2019-01-22 10:31:39 +0100560 *str = (char *)msg;
561 return i;
Emeric Brun18928af2017-03-29 16:32:53 +0200562
Frédéric Lécaillea8725ec2019-01-22 10:31:39 +0100563 fail:
564 *str = NULL;
565 return 0;
Emeric Brunb3971ab2015-05-12 18:49:09 +0200566}
Emeric Brun2b920a12010-09-23 18:30:22 +0200567
Frédéric Lécaille7d0ceee2019-01-24 14:24:05 +0100568/*
569 * Build a "hello" peer protocol message.
570 * Return the number of written bytes written to build this messages if succeeded,
571 * 0 if not.
572 */
573static int peer_prepare_hellomsg(char *msg, size_t size, struct peer_prep_params *p)
574{
575 int min_ver, ret;
576 struct peer *peer;
577
578 peer = p->hello.peer;
579 min_ver = (peer->flags & PEER_F_DWNGRD) ? PEER_DWNGRD_MINOR_VER : PEER_MINOR_VER;
580 /* Prepare headers */
Willy Tarreau2645b342022-04-12 08:28:18 +0200581 ret = snprintf(msg, size, PEER_SESSION_PROTO_NAME " %d.%d\n%s\n%s %d %d\n",
582 (int)PEER_MAJOR_VER, min_ver, peer->id, localpeer, (int)getpid(), (int)1);
Frédéric Lécaille7d0ceee2019-01-24 14:24:05 +0100583 if (ret >= size)
584 return 0;
585
586 return ret;
587}
588
589/*
590 * Build a "handshake succeeded" status message.
591 * Return the number of written bytes written to build this messages if succeeded,
592 * 0 if not.
593 */
594static int peer_prepare_status_successmsg(char *msg, size_t size, struct peer_prep_params *p)
595{
596 int ret;
597
Willy Tarreau2645b342022-04-12 08:28:18 +0200598 ret = snprintf(msg, size, "%d\n", (int)PEER_SESS_SC_SUCCESSCODE);
Frédéric Lécaille7d0ceee2019-01-24 14:24:05 +0100599 if (ret >= size)
600 return 0;
601
602 return ret;
603}
604
605/*
606 * Build an error status message.
607 * Return the number of written bytes written to build this messages if succeeded,
608 * 0 if not.
609 */
610static int peer_prepare_status_errormsg(char *msg, size_t size, struct peer_prep_params *p)
611{
612 int ret;
613 unsigned int st1;
614
615 st1 = p->error_status.st1;
616 ret = snprintf(msg, size, "%d\n", st1);
617 if (ret >= size)
618 return 0;
619
620 return ret;
621}
622
Frédéric Lécaille523cc9e2016-10-12 17:30:30 +0200623/* Set the stick-table UPDATE message type byte at <msg_type> address,
624 * depending on <use_identifier> and <use_timed> boolean parameters.
625 * Always successful.
626 */
627static inline void peer_set_update_msg_type(char *msg_type, int use_identifier, int use_timed)
628{
629 if (use_timed) {
630 if (use_identifier)
631 *msg_type = PEER_MSG_STKT_UPDATE_TIMED;
632 else
633 *msg_type = PEER_MSG_STKT_INCUPDATE_TIMED;
634 }
635 else {
636 if (use_identifier)
637 *msg_type = PEER_MSG_STKT_UPDATE;
638 else
639 *msg_type = PEER_MSG_STKT_INCUPDATE;
640 }
Frédéric Lécaille523cc9e2016-10-12 17:30:30 +0200641}
Emeric Brun2b920a12010-09-23 18:30:22 +0200642/*
Emeric Brunb3971ab2015-05-12 18:49:09 +0200643 * This prepare the data update message on the stick session <ts>, <st> is the considered
644 * stick table.
Joseph Herlant82b2f542018-11-15 12:19:14 -0800645 * <msg> is a buffer of <size> to receive data message content
Emeric Brunb3971ab2015-05-12 18:49:09 +0200646 * If function returns 0, the caller should consider we were unable to encode this message (TODO:
647 * check size)
Emeric Brun2b920a12010-09-23 18:30:22 +0200648 */
Frédéric Lécailled5fe14b2019-01-24 10:33:40 +0100649static int peer_prepare_updatemsg(char *msg, size_t size, struct peer_prep_params *p)
Emeric Brun2b920a12010-09-23 18:30:22 +0200650{
651 uint32_t netinteger;
Emeric Brunb3971ab2015-05-12 18:49:09 +0200652 unsigned short datalen;
653 char *cursor, *datamsg;
Emeric Brun94900952015-06-11 18:25:54 +0200654 unsigned int data_type;
655 void *data_ptr;
Frédéric Lécailled5fe14b2019-01-24 10:33:40 +0100656 struct stksess *ts;
657 struct shared_table *st;
658 unsigned int updateid;
659 int use_identifier;
660 int use_timed;
Frédéric Lécaille8d78fa72019-05-20 18:22:52 +0200661 struct peer *peer;
Frédéric Lécailled5fe14b2019-01-24 10:33:40 +0100662
663 ts = p->updt.stksess;
664 st = p->updt.shared_table;
665 updateid = p->updt.updateid;
666 use_identifier = p->updt.use_identifier;
667 use_timed = p->updt.use_timed;
Frédéric Lécaille8d78fa72019-05-20 18:22:52 +0200668 peer = p->updt.peer;
Emeric Brunb3971ab2015-05-12 18:49:09 +0200669
Frédéric Lécaille0e8db972019-05-24 14:34:34 +0200670 cursor = datamsg = msg + PEER_MSG_HEADER_LEN + PEER_MSG_ENC_LENGTH_MAXLEN;
Emeric Brunb3971ab2015-05-12 18:49:09 +0200671
Emeric Brun2b920a12010-09-23 18:30:22 +0200672 /* construct message */
Emeric Brunb3971ab2015-05-12 18:49:09 +0200673
Ilya Shipitsinc02a23f2020-05-06 00:53:22 +0500674 /* check if we need to send the update identifier */
Emeric Brun819fc6f2017-06-13 19:37:32 +0200675 if (!st->last_pushed || updateid < st->last_pushed || ((updateid - st->last_pushed) != 1)) {
Emeric Bruna6a09982015-09-22 15:34:19 +0200676 use_identifier = 1;
Emeric Brun2b920a12010-09-23 18:30:22 +0200677 }
Emeric Brunb3971ab2015-05-12 18:49:09 +0200678
679 /* encode update identifier if needed */
680 if (use_identifier) {
Emeric Brun819fc6f2017-06-13 19:37:32 +0200681 netinteger = htonl(updateid);
Emeric Brunb3971ab2015-05-12 18:49:09 +0200682 memcpy(cursor, &netinteger, sizeof(netinteger));
683 cursor += sizeof(netinteger);
Emeric Brun2b920a12010-09-23 18:30:22 +0200684 }
685
Frédéric Lécaille523cc9e2016-10-12 17:30:30 +0200686 if (use_timed) {
687 netinteger = htonl(tick_remain(now_ms, ts->expire));
688 memcpy(cursor, &netinteger, sizeof(netinteger));
689 cursor += sizeof(netinteger);
690 }
691
Emeric Brunb3971ab2015-05-12 18:49:09 +0200692 /* encode the key */
Thierry FOURNIER5d24ebc2015-07-24 08:46:42 +0200693 if (st->table->type == SMP_T_STR) {
Emeric Brun2b920a12010-09-23 18:30:22 +0200694 int stlen = strlen((char *)ts->key.key);
695
Emeric Brunb3971ab2015-05-12 18:49:09 +0200696 intencode(stlen, &cursor);
697 memcpy(cursor, ts->key.key, stlen);
698 cursor += stlen;
Emeric Brun2b920a12010-09-23 18:30:22 +0200699 }
Thierry FOURNIER5d24ebc2015-07-24 08:46:42 +0200700 else if (st->table->type == SMP_T_SINT) {
Willy Tarreau6cde5d82020-02-25 09:41:22 +0100701 netinteger = htonl(read_u32(ts->key.key));
Emeric Brunb3971ab2015-05-12 18:49:09 +0200702 memcpy(cursor, &netinteger, sizeof(netinteger));
703 cursor += sizeof(netinteger);
Emeric Brun2b920a12010-09-23 18:30:22 +0200704 }
705 else {
Emeric Brunb3971ab2015-05-12 18:49:09 +0200706 memcpy(cursor, ts->key.key, st->table->key_size);
707 cursor += st->table->key_size;
Emeric Brun2b920a12010-09-23 18:30:22 +0200708 }
709
Christopher Faulet2a944ee2017-11-07 10:42:54 +0100710 HA_RWLOCK_RDLOCK(STK_SESS_LOCK, &ts->lock);
Emeric Brunb3971ab2015-05-12 18:49:09 +0200711 /* encode values */
Emeric Brun94900952015-06-11 18:25:54 +0200712 for (data_type = 0 ; data_type < STKTABLE_DATA_TYPES ; data_type++) {
Emeric Brunb3971ab2015-05-12 18:49:09 +0200713
Emeric Brun94900952015-06-11 18:25:54 +0200714 data_ptr = stktable_data_ptr(st->table, ts, data_type);
715 if (data_ptr) {
Emeric Brun90a9b672021-06-22 16:09:55 +0200716 /* in case of array all elements use
717 * the same std_type and they are linearly
718 * encoded.
719 */
720 if (stktable_data_types[data_type].is_array) {
721 unsigned int idx = 0;
722
723 switch (stktable_data_types[data_type].std_type) {
724 case STD_T_SINT: {
725 int data;
726
727 do {
728 data = stktable_data_cast(data_ptr, std_t_sint);
729 intencode(data, &cursor);
730
731 data_ptr = stktable_data_ptr_idx(st->table, ts, data_type, ++idx);
732 } while(data_ptr);
733 break;
734 }
735 case STD_T_UINT: {
736 unsigned int data;
737
738 do {
739 data = stktable_data_cast(data_ptr, std_t_uint);
740 intencode(data, &cursor);
741
742 data_ptr = stktable_data_ptr_idx(st->table, ts, data_type, ++idx);
743 } while(data_ptr);
744 break;
745 }
746 case STD_T_ULL: {
747 unsigned long long data;
748
749 do {
750 data = stktable_data_cast(data_ptr, std_t_ull);
751 intencode(data, &cursor);
752
753 data_ptr = stktable_data_ptr_idx(st->table, ts, data_type, ++idx);
754 } while(data_ptr);
755 break;
756 }
757 case STD_T_FRQP: {
758 struct freq_ctr *frqp;
759
760 do {
761 frqp = &stktable_data_cast(data_ptr, std_t_frqp);
762 intencode((unsigned int)(now_ms - frqp->curr_tick), &cursor);
763 intencode(frqp->curr_ctr, &cursor);
764 intencode(frqp->prev_ctr, &cursor);
765
766 data_ptr = stktable_data_ptr_idx(st->table, ts, data_type, ++idx);
767 } while(data_ptr);
768 break;
769 }
770 }
771
772 /* array elements fully encoded
773 * proceed next data_type.
774 */
775 continue;
776 }
Emeric Brun94900952015-06-11 18:25:54 +0200777 switch (stktable_data_types[data_type].std_type) {
778 case STD_T_SINT: {
779 int data;
780
781 data = stktable_data_cast(data_ptr, std_t_sint);
782 intencode(data, &cursor);
783 break;
784 }
785 case STD_T_UINT: {
786 unsigned int data;
787
788 data = stktable_data_cast(data_ptr, std_t_uint);
789 intencode(data, &cursor);
790 break;
791 }
792 case STD_T_ULL: {
793 unsigned long long data;
794
795 data = stktable_data_cast(data_ptr, std_t_ull);
796 intencode(data, &cursor);
797 break;
798 }
799 case STD_T_FRQP: {
Willy Tarreaufa1258f2021-04-10 23:00:53 +0200800 struct freq_ctr *frqp;
Emeric Brun94900952015-06-11 18:25:54 +0200801
802 frqp = &stktable_data_cast(data_ptr, std_t_frqp);
803 intencode((unsigned int)(now_ms - frqp->curr_tick), &cursor);
804 intencode(frqp->curr_ctr, &cursor);
805 intencode(frqp->prev_ctr, &cursor);
806 break;
807 }
Frédéric Lécaille8d78fa72019-05-20 18:22:52 +0200808 case STD_T_DICT: {
809 struct dict_entry *de;
Frédéric Lécaille6c391982019-06-06 11:34:03 +0200810 struct ebpt_node *cached_de;
Willy Tarreau237f8ae2019-06-06 16:40:43 +0200811 struct dcache_tx_entry cde = { };
Frédéric Lécaille8d78fa72019-05-20 18:22:52 +0200812 char *beg, *end;
813 size_t value_len, data_len;
814 struct dcache *dc;
815
816 de = stktable_data_cast(data_ptr, std_t_dict);
Frédéric Lécailleaf9990f2019-11-13 17:50:34 +0100817 if (!de) {
818 /* No entry */
819 intencode(0, &cursor);
Frédéric Lécaille8d78fa72019-05-20 18:22:52 +0200820 break;
Frédéric Lécailleaf9990f2019-11-13 17:50:34 +0100821 }
Frédéric Lécaille8d78fa72019-05-20 18:22:52 +0200822
823 dc = peer->dcache;
Frédéric Lécaille6c391982019-06-06 11:34:03 +0200824 cde.entry.key = de;
Frédéric Lécaille8d78fa72019-05-20 18:22:52 +0200825 cached_de = dcache_tx_insert(dc, &cde);
Frédéric Lécaillefd827932019-06-07 10:34:04 +0200826 if (cached_de == &cde.entry) {
827 if (cde.id + 1 >= PEER_ENC_2BYTES_MIN)
828 break;
829 /* Encode the length of the remaining data -> 1 */
830 intencode(1, &cursor);
831 /* Encode the cache entry ID */
832 intencode(cde.id + 1, &cursor);
833 }
834 else {
835 /* Leave enough room to encode the remaining data length. */
836 end = beg = cursor + PEER_MSG_ENC_LENGTH_MAXLEN;
837 /* Encode the dictionary entry key */
838 intencode(cde.id + 1, &end);
Frédéric Lécaille8d78fa72019-05-20 18:22:52 +0200839 /* Encode the length of the dictionary entry data */
Frédéric Lécaillefd827932019-06-07 10:34:04 +0200840 value_len = de->len;
Frédéric Lécaille8d78fa72019-05-20 18:22:52 +0200841 intencode(value_len, &end);
842 /* Copy the data */
843 memcpy(end, de->value.key, value_len);
844 end += value_len;
Frédéric Lécaillefd827932019-06-07 10:34:04 +0200845 /* Encode the length of the data */
846 data_len = end - beg;
847 intencode(data_len, &cursor);
848 memmove(cursor, beg, data_len);
849 cursor += data_len;
Frédéric Lécaille8d78fa72019-05-20 18:22:52 +0200850 }
Frédéric Lécaille8d78fa72019-05-20 18:22:52 +0200851 break;
852 }
Emeric Brun94900952015-06-11 18:25:54 +0200853 }
854 }
Emeric Brunb3971ab2015-05-12 18:49:09 +0200855 }
Christopher Faulet2a944ee2017-11-07 10:42:54 +0100856 HA_RWLOCK_RDUNLOCK(STK_SESS_LOCK, &ts->lock);
Emeric Brunb3971ab2015-05-12 18:49:09 +0200857
858 /* Compute datalen */
859 datalen = (cursor - datamsg);
860
861 /* prepare message header */
862 msg[0] = PEER_MSG_CLASS_STICKTABLE;
Frédéric Lécaille523cc9e2016-10-12 17:30:30 +0200863 peer_set_update_msg_type(&msg[1], use_identifier, use_timed);
Emeric Brunb3971ab2015-05-12 18:49:09 +0200864 cursor = &msg[2];
865 intencode(datalen, &cursor);
866
867 /* move data after header */
868 memmove(cursor, datamsg, datalen);
869
870 /* return header size + data_len */
871 return (cursor - msg) + datalen;
872}
873
874/*
875 * This prepare the switch table message to targeted share table <st>.
Joseph Herlant82b2f542018-11-15 12:19:14 -0800876 * <msg> is a buffer of <size> to receive data message content
Emeric Brunb3971ab2015-05-12 18:49:09 +0200877 * If function returns 0, the caller should consider we were unable to encode this message (TODO:
878 * check size)
879 */
Frédéric Lécailled5fe14b2019-01-24 10:33:40 +0100880static int peer_prepare_switchmsg(char *msg, size_t size, struct peer_prep_params *params)
Emeric Brunb3971ab2015-05-12 18:49:09 +0200881{
882 int len;
883 unsigned short datalen;
Willy Tarreau83061a82018-07-13 11:56:34 +0200884 struct buffer *chunk;
Frédéric Lécaille37a72542017-07-06 15:02:16 +0200885 char *cursor, *datamsg, *chunkp, *chunkq;
Emeric Brunb3971ab2015-05-12 18:49:09 +0200886 uint64_t data = 0;
Emeric Brun94900952015-06-11 18:25:54 +0200887 unsigned int data_type;
Frédéric Lécailled5fe14b2019-01-24 10:33:40 +0100888 struct shared_table *st;
Emeric Brunb3971ab2015-05-12 18:49:09 +0200889
Frédéric Lécailled5fe14b2019-01-24 10:33:40 +0100890 st = params->swtch.shared_table;
Frédéric Lécaille39143342019-05-24 14:32:27 +0200891 cursor = datamsg = msg + PEER_MSG_HEADER_LEN + PEER_MSG_ENC_LENGTH_MAXLEN;
Emeric Brunb3971ab2015-05-12 18:49:09 +0200892
893 /* Encode data */
894
895 /* encode local id */
896 intencode(st->local_id, &cursor);
897
898 /* encode table name */
Frédéric Lécaille7fcc24d2019-03-20 15:09:45 +0100899 len = strlen(st->table->nid);
Emeric Brunb3971ab2015-05-12 18:49:09 +0200900 intencode(len, &cursor);
Frédéric Lécaille7fcc24d2019-03-20 15:09:45 +0100901 memcpy(cursor, st->table->nid, len);
Emeric Brunb3971ab2015-05-12 18:49:09 +0200902 cursor += len;
903
904 /* encode table type */
905
Emeric Brun530ba382020-06-02 11:17:42 +0200906 intencode(peer_net_key_type[st->table->type], &cursor);
Emeric Brunb3971ab2015-05-12 18:49:09 +0200907
908 /* encode table key size */
909 intencode(st->table->key_size, &cursor);
910
Frédéric Lécaille37a72542017-07-06 15:02:16 +0200911 chunk = get_trash_chunk();
Willy Tarreau843b7cb2018-07-13 10:54:26 +0200912 chunkp = chunkq = chunk->area;
Emeric Brun94900952015-06-11 18:25:54 +0200913 /* encode available known data types in table */
914 for (data_type = 0 ; data_type < STKTABLE_DATA_TYPES ; data_type++) {
915 if (st->table->data_ofs[data_type]) {
Emeric Brun90a9b672021-06-22 16:09:55 +0200916 /* stored data types parameters are all linearly encoded
917 * at the end of the 'table definition' message.
918 *
919 * Currently only array data_types and and data_types
920 * using freq_counter base type have parameters:
921 *
922 * - array has always at least one parameter set to the
923 * number of elements.
924 *
925 * - array of base-type freq_counters has an additional
926 * parameter set to the period used to compute those
927 * freq_counters.
928 *
929 * - simple freq counter has a parameter set to the period
930 * used to compute
931 *
932 * A set of parameter for a datatype MUST BE prefixed
933 * by the data-type id itself:
934 * This is useless because the data_types are ordered and
935 * the data_type bitfield already gives the information of
936 * stored types, but it was designed this way when the
937 * push of period parameter was added for freq counters
938 * and we don't want to break the compatibility.
939 *
940 */
941 if (stktable_data_types[data_type].is_array) {
942 /* This is an array type so we first encode
943 * the data_type itself to prefix parameters
944 */
945 intencode(data_type, &chunkq);
946
947 /* We encode the first parameter which is
948 * the number of elements of this array
949 */
950 intencode(st->table->data_nbelem[data_type], &chunkq);
951
Ilya Shipitsin01881082021-08-07 14:41:56 +0500952 /* for array of freq counters, there is an additional
Emeric Brun90a9b672021-06-22 16:09:55 +0200953 * period parameter to encode
954 */
955 if (stktable_data_types[data_type].std_type == STD_T_FRQP)
Frédéric Lécaille37a72542017-07-06 15:02:16 +0200956 intencode(st->table->data_arg[data_type].u, &chunkq);
Emeric Brun94900952015-06-11 18:25:54 +0200957 }
Emeric Brun90a9b672021-06-22 16:09:55 +0200958 else if (stktable_data_types[data_type].std_type == STD_T_FRQP) {
959 /* this datatype is a simple freq counter not part
960 * of an array. We encode the data_type itself
961 * to prefix the 'period' parameter
962 */
963 intencode(data_type, &chunkq);
964 intencode(st->table->data_arg[data_type].u, &chunkq);
965 }
966 /* set the bit corresponding to stored data type */
967 data |= 1ULL << data_type;
Emeric Brun94900952015-06-11 18:25:54 +0200968 }
Emeric Brunb3971ab2015-05-12 18:49:09 +0200969 }
970 intencode(data, &cursor);
971
Frédéric Lécaille37a72542017-07-06 15:02:16 +0200972 /* Encode stick-table entries duration. */
973 intencode(st->table->expire, &cursor);
974
975 if (chunkq > chunkp) {
Willy Tarreau843b7cb2018-07-13 10:54:26 +0200976 chunk->data = chunkq - chunkp;
977 memcpy(cursor, chunk->area, chunk->data);
978 cursor += chunk->data;
Frédéric Lécaille37a72542017-07-06 15:02:16 +0200979 }
980
Emeric Brunb3971ab2015-05-12 18:49:09 +0200981 /* Compute datalen */
982 datalen = (cursor - datamsg);
Emeric Brun2b920a12010-09-23 18:30:22 +0200983
Emeric Brunb3971ab2015-05-12 18:49:09 +0200984 /* prepare message header */
985 msg[0] = PEER_MSG_CLASS_STICKTABLE;
986 msg[1] = PEER_MSG_STKT_DEFINE;
987 cursor = &msg[2];
988 intencode(datalen, &cursor);
Emeric Brun2b920a12010-09-23 18:30:22 +0200989
Emeric Brunb3971ab2015-05-12 18:49:09 +0200990 /* move data after header */
991 memmove(cursor, datamsg, datalen);
992
993 /* return header size + data_len */
994 return (cursor - msg) + datalen;
Emeric Brun2b920a12010-09-23 18:30:22 +0200995}
996
Emeric Brunb3971ab2015-05-12 18:49:09 +0200997/*
998 * This prepare the acknowledge message on the stick session <ts>, <st> is the considered
999 * stick table.
Joseph Herlant82b2f542018-11-15 12:19:14 -08001000 * <msg> is a buffer of <size> to receive data message content
Emeric Brunb3971ab2015-05-12 18:49:09 +02001001 * If function returns 0, the caller should consider we were unable to encode this message (TODO:
1002 * check size)
1003 */
Frédéric Lécailled5fe14b2019-01-24 10:33:40 +01001004static int peer_prepare_ackmsg(char *msg, size_t size, struct peer_prep_params *p)
Emeric Brunb3971ab2015-05-12 18:49:09 +02001005{
1006 unsigned short datalen;
1007 char *cursor, *datamsg;
1008 uint32_t netinteger;
Frédéric Lécailled5fe14b2019-01-24 10:33:40 +01001009 struct shared_table *st;
Emeric Brunb3971ab2015-05-12 18:49:09 +02001010
Frédéric Lécaille39143342019-05-24 14:32:27 +02001011 cursor = datamsg = msg + PEER_MSG_HEADER_LEN + PEER_MSG_ENC_LENGTH_MAXLEN;
Emeric Brunb3971ab2015-05-12 18:49:09 +02001012
Frédéric Lécailled5fe14b2019-01-24 10:33:40 +01001013 st = p->ack.shared_table;
Emeric Brunb3971ab2015-05-12 18:49:09 +02001014 intencode(st->remote_id, &cursor);
1015 netinteger = htonl(st->last_get);
1016 memcpy(cursor, &netinteger, sizeof(netinteger));
1017 cursor += sizeof(netinteger);
1018
1019 /* Compute datalen */
1020 datalen = (cursor - datamsg);
1021
1022 /* prepare message header */
1023 msg[0] = PEER_MSG_CLASS_STICKTABLE;
Emeric Brune1ab8082015-08-21 11:48:54 +02001024 msg[1] = PEER_MSG_STKT_ACK;
Emeric Brunb3971ab2015-05-12 18:49:09 +02001025 cursor = &msg[2];
1026 intencode(datalen, &cursor);
1027
1028 /* move data after header */
1029 memmove(cursor, datamsg, datalen);
1030
1031 /* return header size + data_len */
1032 return (cursor - msg) + datalen;
1033}
Emeric Brun2b920a12010-09-23 18:30:22 +02001034
1035/*
Emeric Brun9ef2ad72019-04-02 17:22:01 +02001036 * Function to deinit connected peer
1037 */
1038void __peer_session_deinit(struct peer *peer)
1039{
Emeric Brun9ef2ad72019-04-02 17:22:01 +02001040 struct stream *s;
1041 struct peers *peers;
1042
1043 if (!peer->appctx)
1044 return;
1045
Christopher Faulet908628c2022-03-25 16:43:49 +01001046 s = __cs_strm(peer->appctx->owner);
Emeric Brun9ef2ad72019-04-02 17:22:01 +02001047
1048 peers = strm_fe(s)->parent;
1049 if (!peers)
1050 return;
1051
1052 if (peer->appctx->st0 == PEER_SESS_ST_WAITMSG)
Willy Tarreau4781b152021-04-06 13:53:36 +02001053 HA_ATOMIC_DEC(&connected_peers);
Emeric Brun9ef2ad72019-04-02 17:22:01 +02001054
Willy Tarreau4781b152021-04-06 13:53:36 +02001055 HA_ATOMIC_DEC(&active_peers);
Emeric Brun9ef2ad72019-04-02 17:22:01 +02001056
Frédéric Lécaille8d78fa72019-05-20 18:22:52 +02001057 flush_dcache(peer);
1058
Emeric Brun9ef2ad72019-04-02 17:22:01 +02001059 /* Re-init current table pointers to force announcement on re-connect */
1060 peer->remote_table = peer->last_local_table = NULL;
1061 peer->appctx = NULL;
1062 if (peer->flags & PEER_F_LEARN_ASSIGN) {
1063 /* unassign current peer for learning */
1064 peer->flags &= ~(PEER_F_LEARN_ASSIGN);
1065 peers->flags &= ~(PEERS_F_RESYNC_ASSIGN|PEERS_F_RESYNC_PROCESS);
1066
Emeric Brunccdfbae2021-04-28 12:59:35 +02001067 if (peer->local)
1068 peers->flags |= PEERS_F_RESYNC_LOCALABORT;
1069 else
1070 peers->flags |= PEERS_F_RESYNC_REMOTEABORT;
Emeric Brun9ef2ad72019-04-02 17:22:01 +02001071 /* reschedule a resync */
1072 peers->resync_timeout = tick_add(now_ms, MS_TO_TICKS(5000));
1073 }
1074 /* reset teaching and learning flags to 0 */
1075 peer->flags &= PEER_TEACH_RESET;
1076 peer->flags &= PEER_LEARN_RESET;
1077 task_wakeup(peers->sync_task, TASK_WOKEN_MSG);
1078}
1079
1080/*
Emeric Brun2b920a12010-09-23 18:30:22 +02001081 * Callback to release a session with a peer
1082 */
Willy Tarreau00a37f02015-04-13 12:05:19 +02001083static void peer_session_release(struct appctx *appctx)
Emeric Brun2b920a12010-09-23 18:30:22 +02001084{
Vincent Bernat3c2f2f22016-04-03 13:48:42 +02001085 struct peer *peer = appctx->ctx.peers.ptr;
Emeric Brun2b920a12010-09-23 18:30:22 +02001086
Frédéric Lécailleda2b0842021-01-15 16:21:28 +01001087 TRACE_PROTO("releasing peer session", PEERS_EV_SESSREL, NULL, peer);
Willy Tarreau7b4b4992013-12-01 09:15:12 +01001088 /* appctx->ctx.peers.ptr is not a peer session */
Willy Tarreaue4d927a2013-12-01 12:47:35 +01001089 if (appctx->st0 < PEER_SESS_ST_SENDSUCCESS)
Emeric Brun2b920a12010-09-23 18:30:22 +02001090 return;
1091
1092 /* peer session identified */
Emeric Brunb3971ab2015-05-12 18:49:09 +02001093 if (peer) {
Christopher Faulet2a944ee2017-11-07 10:42:54 +01001094 HA_SPIN_LOCK(PEER_LOCK, &peer->lock);
Emeric Brun9ef2ad72019-04-02 17:22:01 +02001095 if (peer->appctx == appctx)
1096 __peer_session_deinit(peer);
Frédéric Lécaillebaeb9192020-10-14 11:50:26 +02001097 peer->flags &= ~PEER_F_ALIVE;
Christopher Faulet2a944ee2017-11-07 10:42:54 +01001098 HA_SPIN_UNLOCK(PEER_LOCK, &peer->lock);
Emeric Brun2b920a12010-09-23 18:30:22 +02001099 }
1100}
1101
Frédéric Lécaille523cc9e2016-10-12 17:30:30 +02001102/* Retrieve the major and minor versions of peers protocol
1103 * announced by a remote peer. <str> is a null-terminated
1104 * string with the following format: "<maj_ver>.<min_ver>".
1105 */
1106static int peer_get_version(const char *str,
1107 unsigned int *maj_ver, unsigned int *min_ver)
1108{
1109 unsigned int majv, minv;
1110 const char *pos, *saved;
1111 const char *end;
1112
1113 saved = pos = str;
1114 end = str + strlen(str);
1115
1116 majv = read_uint(&pos, end);
1117 if (saved == pos || *pos++ != '.')
1118 return -1;
1119
1120 saved = pos;
1121 minv = read_uint(&pos, end);
1122 if (saved == pos || pos != end)
1123 return -1;
1124
1125 *maj_ver = majv;
1126 *min_ver = minv;
1127
1128 return 0;
1129}
Emeric Brun2b920a12010-09-23 18:30:22 +02001130
1131/*
Frédéric Lécaillece025572019-01-21 13:38:06 +01001132 * Parse a line terminated by an optional '\r' character, followed by a mandatory
1133 * '\n' character.
1134 * Returns 1 if succeeded or 0 if a '\n' character could not be found, and -1 if
1135 * a line could not be read because the communication channel is closed.
1136 */
1137static inline int peer_getline(struct appctx *appctx)
1138{
Christopher Faulet908628c2022-03-25 16:43:49 +01001139 struct conn_stream *cs = appctx->owner;
Frédéric Lécaillece025572019-01-21 13:38:06 +01001140 int n;
Frédéric Lécaillece025572019-01-21 13:38:06 +01001141
Christopher Faulet908628c2022-03-25 16:43:49 +01001142 n = co_getline(cs_oc(cs), trash.area, trash.size);
Frédéric Lécaillece025572019-01-21 13:38:06 +01001143 if (!n)
1144 return 0;
1145
1146 if (n < 0 || trash.area[n - 1] != '\n') {
1147 appctx->st0 = PEER_SESS_ST_END;
1148 return -1;
1149 }
1150
1151 if (n > 1 && (trash.area[n - 2] == '\r'))
1152 trash.area[n - 2] = 0;
1153 else
1154 trash.area[n - 1] = 0;
1155
Christopher Faulet908628c2022-03-25 16:43:49 +01001156 co_skip(cs_oc(cs), n);
Frédéric Lécaillece025572019-01-21 13:38:06 +01001157
1158 return n;
1159}
1160
1161/*
Frédéric Lécailleec44ea82019-01-22 15:54:53 +01001162 * Send a message after having called <peer_prepare_msg> to build it.
1163 * Return 0 if the message could not be built modifying the appcxt st0 to PEER_SESS_ST_END value.
1164 * Returns -1 if there was not enough room left to send the message,
1165 * any other negative returned value must be considered as an error with an appcxt st0
1166 * returned value equal to PEER_SESS_ST_END.
1167 */
Frédéric Lécailled5fe14b2019-01-24 10:33:40 +01001168static inline int peer_send_msg(struct appctx *appctx,
1169 int (*peer_prepare_msg)(char *, size_t, struct peer_prep_params *),
1170 struct peer_prep_params *params)
Frédéric Lécailleec44ea82019-01-22 15:54:53 +01001171{
1172 int ret, msglen;
Christopher Faulet908628c2022-03-25 16:43:49 +01001173 struct conn_stream *cs = appctx->owner;
Frédéric Lécailleec44ea82019-01-22 15:54:53 +01001174
Frédéric Lécailled5fe14b2019-01-24 10:33:40 +01001175 msglen = peer_prepare_msg(trash.area, trash.size, params);
Frédéric Lécailleec44ea82019-01-22 15:54:53 +01001176 if (!msglen) {
1177 /* internal error: message does not fit in trash */
1178 appctx->st0 = PEER_SESS_ST_END;
1179 return 0;
1180 }
1181
1182 /* message to buffer */
Christopher Faulet908628c2022-03-25 16:43:49 +01001183 ret = ci_putblk(cs_ic(cs), trash.area, msglen);
Frédéric Lécailleec44ea82019-01-22 15:54:53 +01001184 if (ret <= 0) {
1185 if (ret == -1) {
1186 /* No more write possible */
Christopher Fauleta0bdec32022-04-04 07:51:21 +02001187 cs_rx_room_blk(cs);
Frédéric Lécailleec44ea82019-01-22 15:54:53 +01001188 return -1;
1189 }
1190 appctx->st0 = PEER_SESS_ST_END;
1191 }
1192
1193 return ret;
1194}
1195
1196/*
Frédéric Lécaille7d0ceee2019-01-24 14:24:05 +01001197 * Send a hello message.
1198 * Return 0 if the message could not be built modifying the appcxt st0 to PEER_SESS_ST_END value.
1199 * Returns -1 if there was not enough room left to send the message,
1200 * any other negative returned value must be considered as an error with an appcxt st0
1201 * returned value equal to PEER_SESS_ST_END.
1202 */
1203static inline int peer_send_hellomsg(struct appctx *appctx, struct peer *peer)
1204{
1205 struct peer_prep_params p = {
1206 .hello.peer = peer,
1207 };
1208
1209 return peer_send_msg(appctx, peer_prepare_hellomsg, &p);
1210}
1211
1212/*
1213 * Send a success peer handshake status message.
1214 * Return 0 if the message could not be built modifying the appcxt st0 to PEER_SESS_ST_END value.
1215 * Returns -1 if there was not enough room left to send the message,
1216 * any other negative returned value must be considered as an error with an appcxt st0
1217 * returned value equal to PEER_SESS_ST_END.
1218 */
1219static inline int peer_send_status_successmsg(struct appctx *appctx)
1220{
1221 return peer_send_msg(appctx, peer_prepare_status_successmsg, NULL);
1222}
1223
1224/*
1225 * Send a peer handshake status error message.
1226 * Return 0 if the message could not be built modifying the appcxt st0 to PEER_SESS_ST_END value.
1227 * Returns -1 if there was not enough room left to send the message,
1228 * any other negative returned value must be considered as an error with an appcxt st0
1229 * returned value equal to PEER_SESS_ST_END.
1230 */
1231static inline int peer_send_status_errormsg(struct appctx *appctx)
1232{
1233 struct peer_prep_params p = {
1234 .error_status.st1 = appctx->st1,
1235 };
1236
1237 return peer_send_msg(appctx, peer_prepare_status_errormsg, &p);
1238}
1239
1240/*
Frédéric Lécailleec44ea82019-01-22 15:54:53 +01001241 * Send a stick-table switch message.
1242 * Return 0 if the message could not be built modifying the appcxt st0 to PEER_SESS_ST_END value.
1243 * Returns -1 if there was not enough room left to send the message,
1244 * any other negative returned value must be considered as an error with an appcxt st0
1245 * returned value equal to PEER_SESS_ST_END.
1246 */
1247static inline int peer_send_switchmsg(struct shared_table *st, struct appctx *appctx)
1248{
Frédéric Lécailled5fe14b2019-01-24 10:33:40 +01001249 struct peer_prep_params p = {
1250 .swtch.shared_table = st,
1251 };
1252
1253 return peer_send_msg(appctx, peer_prepare_switchmsg, &p);
Frédéric Lécailleec44ea82019-01-22 15:54:53 +01001254}
1255
1256/*
1257 * Send a stick-table update acknowledgement message.
1258 * Return 0 if the message could not be built modifying the appcxt st0 to PEER_SESS_ST_END value.
1259 * Returns -1 if there was not enough room left to send the message,
1260 * any other negative returned value must be considered as an error with an appcxt st0
1261 * returned value equal to PEER_SESS_ST_END.
1262 */
1263static inline int peer_send_ackmsg(struct shared_table *st, struct appctx *appctx)
1264{
Frédéric Lécailled5fe14b2019-01-24 10:33:40 +01001265 struct peer_prep_params p = {
1266 .ack.shared_table = st,
1267 };
1268
1269 return peer_send_msg(appctx, peer_prepare_ackmsg, &p);
Frédéric Lécailleec44ea82019-01-22 15:54:53 +01001270}
1271
1272/*
Frédéric Lécaille87f554c2019-01-22 17:26:50 +01001273 * Send a stick-table update message.
1274 * Return 0 if the message could not be built modifying the appcxt st0 to PEER_SESS_ST_END value.
1275 * Returns -1 if there was not enough room left to send the message,
1276 * any other negative returned value must be considered as an error with an appcxt st0
1277 * returned value equal to PEER_SESS_ST_END.
1278 */
1279static inline int peer_send_updatemsg(struct shared_table *st, struct appctx *appctx, struct stksess *ts,
1280 unsigned int updateid, int use_identifier, int use_timed)
1281{
Frédéric Lécailled5fe14b2019-01-24 10:33:40 +01001282 struct peer_prep_params p = {
Willy Tarreaua898f0c2020-07-03 19:09:29 +02001283 .updt = {
1284 .stksess = ts,
1285 .shared_table = st,
1286 .updateid = updateid,
1287 .use_identifier = use_identifier,
1288 .use_timed = use_timed,
1289 .peer = appctx->ctx.peers.ptr,
1290 },
Frédéric Lécailled5fe14b2019-01-24 10:33:40 +01001291 };
1292
1293 return peer_send_msg(appctx, peer_prepare_updatemsg, &p);
Frédéric Lécaille87f554c2019-01-22 17:26:50 +01001294}
1295
Frédéric Lécaille7d0ceee2019-01-24 14:24:05 +01001296/*
1297 * Build a peer protocol control class message.
1298 * Returns the number of written bytes used to build the message if succeeded,
1299 * 0 if not.
1300 */
1301static int peer_prepare_control_msg(char *msg, size_t size, struct peer_prep_params *p)
1302{
1303 if (size < sizeof p->control.head)
1304 return 0;
1305
1306 msg[0] = p->control.head[0];
1307 msg[1] = p->control.head[1];
1308
1309 return 2;
1310}
Frédéric Lécaille6a8303d2019-01-22 22:25:17 +01001311
1312/*
Frédéric Lécaille7d0ceee2019-01-24 14:24:05 +01001313 * Send a stick-table synchronization request message.
1314 * Return 0 if the message could not be built modifying the appcxt st0 to PEER_SESS_ST_END value.
1315 * Returns -1 if there was not enough room left to send the message,
1316 * any other negative returned value must be considered as an error with an appctx st0
1317 * returned value equal to PEER_SESS_ST_END.
1318 */
Frédéric Lécailleda2b0842021-01-15 16:21:28 +01001319static inline int peer_send_resync_reqmsg(struct appctx *appctx,
1320 struct peer *peer, struct peers *peers)
Frédéric Lécaille7d0ceee2019-01-24 14:24:05 +01001321{
1322 struct peer_prep_params p = {
1323 .control.head = { PEER_MSG_CLASS_CONTROL, PEER_MSG_CTRL_RESYNCREQ, },
1324 };
1325
Frédéric Lécailleda2b0842021-01-15 16:21:28 +01001326 TRACE_PROTO("send control message", PEERS_EV_CTRLMSG,
1327 NULL, &p.control.head[1], peers->local->id, peer->id);
1328
Frédéric Lécaille7d0ceee2019-01-24 14:24:05 +01001329 return peer_send_msg(appctx, peer_prepare_control_msg, &p);
1330}
1331
1332/*
1333 * Send a stick-table synchronization confirmation message.
1334 * Return 0 if the message could not be built modifying the appcxt st0 to PEER_SESS_ST_END value.
1335 * Returns -1 if there was not enough room left to send the message,
1336 * any other negative returned value must be considered as an error with an appctx st0
1337 * returned value equal to PEER_SESS_ST_END.
1338 */
Frédéric Lécailleda2b0842021-01-15 16:21:28 +01001339static inline int peer_send_resync_confirmsg(struct appctx *appctx,
1340 struct peer *peer, struct peers *peers)
Frédéric Lécaille7d0ceee2019-01-24 14:24:05 +01001341{
1342 struct peer_prep_params p = {
1343 .control.head = { PEER_MSG_CLASS_CONTROL, PEER_MSG_CTRL_RESYNCCONFIRM, },
1344 };
1345
Frédéric Lécailleda2b0842021-01-15 16:21:28 +01001346 TRACE_PROTO("send control message", PEERS_EV_CTRLMSG,
1347 NULL, &p.control.head[1], peers->local->id, peer->id);
1348
Frédéric Lécaille7d0ceee2019-01-24 14:24:05 +01001349 return peer_send_msg(appctx, peer_prepare_control_msg, &p);
1350}
1351
1352/*
1353 * Send a stick-table synchronization finished message.
1354 * Return 0 if the message could not be built modifying the appcxt st0 to PEER_SESS_ST_END value.
1355 * Returns -1 if there was not enough room left to send the message,
1356 * any other negative returned value must be considered as an error with an appctx st0
1357 * returned value equal to PEER_SESS_ST_END.
1358 */
Frédéric Lécailleda2b0842021-01-15 16:21:28 +01001359static inline int peer_send_resync_finishedmsg(struct appctx *appctx,
1360 struct peer *peer, struct peers *peers)
Frédéric Lécaille7d0ceee2019-01-24 14:24:05 +01001361{
1362 struct peer_prep_params p = {
1363 .control.head = { PEER_MSG_CLASS_CONTROL, },
1364 };
1365
Emeric Brun70de43b2020-03-16 10:51:01 +01001366 p.control.head[1] = (peers->flags & PEERS_RESYNC_STATEMASK) == PEERS_RESYNC_FINISHED ?
Frédéric Lécaille7d0ceee2019-01-24 14:24:05 +01001367 PEER_MSG_CTRL_RESYNCFINISHED : PEER_MSG_CTRL_RESYNCPARTIAL;
1368
Frédéric Lécailleda2b0842021-01-15 16:21:28 +01001369 TRACE_PROTO("send control message", PEERS_EV_CTRLMSG,
1370 NULL, &p.control.head[1], peers->local->id, peer->id);
1371
Frédéric Lécaille7d0ceee2019-01-24 14:24:05 +01001372 return peer_send_msg(appctx, peer_prepare_control_msg, &p);
1373}
1374
1375/*
Frédéric Lécaille645635d2019-02-11 17:49:39 +01001376 * Send a heartbeat message.
1377 * Return 0 if the message could not be built modifying the appctx st0 to PEER_SESS_ST_END value.
1378 * Returns -1 if there was not enough room left to send the message,
1379 * any other negative returned value must be considered as an error with an appctx st0
1380 * returned value equal to PEER_SESS_ST_END.
1381 */
Frédéric Lécailleda2b0842021-01-15 16:21:28 +01001382static inline int peer_send_heartbeatmsg(struct appctx *appctx,
1383 struct peer *peer, struct peers *peers)
Frédéric Lécaille645635d2019-02-11 17:49:39 +01001384{
1385 struct peer_prep_params p = {
1386 .control.head = { PEER_MSG_CLASS_CONTROL, PEER_MSG_CTRL_HEARTBEAT, },
1387 };
1388
Frédéric Lécailleda2b0842021-01-15 16:21:28 +01001389 TRACE_PROTO("send control message", PEERS_EV_CTRLMSG,
1390 NULL, &p.control.head[1], peers->local->id, peer->id);
1391
Frédéric Lécaille645635d2019-02-11 17:49:39 +01001392 return peer_send_msg(appctx, peer_prepare_control_msg, &p);
1393}
1394
1395/*
Frédéric Lécaille7d0ceee2019-01-24 14:24:05 +01001396 * Build a peer protocol error class message.
1397 * Returns the number of written bytes used to build the message if succeeded,
1398 * 0 if not.
1399 */
1400static int peer_prepare_error_msg(char *msg, size_t size, struct peer_prep_params *p)
1401{
1402 if (size < sizeof p->error.head)
1403 return 0;
1404
1405 msg[0] = p->error.head[0];
1406 msg[1] = p->error.head[1];
1407
1408 return 2;
1409}
1410
1411/*
1412 * Send a "size limit reached" error message.
1413 * Return 0 if the message could not be built modifying the appcxt st0 to PEER_SESS_ST_END value.
1414 * Returns -1 if there was not enough room left to send the message,
1415 * any other negative returned value must be considered as an error with an appctx st0
1416 * returned value equal to PEER_SESS_ST_END.
1417 */
1418static inline int peer_send_error_size_limitmsg(struct appctx *appctx)
1419{
1420 struct peer_prep_params p = {
1421 .error.head = { PEER_MSG_CLASS_ERROR, PEER_MSG_ERR_SIZELIMIT, },
1422 };
1423
1424 return peer_send_msg(appctx, peer_prepare_error_msg, &p);
1425}
1426
1427/*
1428 * Send a "peer protocol" error message.
1429 * Return 0 if the message could not be built modifying the appcxt st0 to PEER_SESS_ST_END value.
1430 * Returns -1 if there was not enough room left to send the message,
1431 * any other negative returned value must be considered as an error with an appctx st0
1432 * returned value equal to PEER_SESS_ST_END.
1433 */
1434static inline int peer_send_error_protomsg(struct appctx *appctx)
1435{
1436 struct peer_prep_params p = {
1437 .error.head = { PEER_MSG_CLASS_ERROR, PEER_MSG_ERR_PROTOCOL, },
1438 };
1439
1440 return peer_send_msg(appctx, peer_prepare_error_msg, &p);
1441}
1442
1443/*
Frédéric Lécaille6a8303d2019-01-22 22:25:17 +01001444 * Function used to lookup for recent stick-table updates associated with
1445 * <st> shared stick-table when a lesson must be taught a peer (PEER_F_LEARN_ASSIGN flag set).
1446 */
1447static inline struct stksess *peer_teach_process_stksess_lookup(struct shared_table *st)
1448{
1449 struct eb32_node *eb;
1450
1451 eb = eb32_lookup_ge(&st->table->updates, st->last_pushed+1);
1452 if (!eb) {
1453 eb = eb32_first(&st->table->updates);
Emeric Brun8e7a13e2021-04-28 11:48:15 +02001454 if (!eb || (eb->key == st->last_pushed)) {
Frédéric Lécaille6a8303d2019-01-22 22:25:17 +01001455 st->table->commitupdate = st->last_pushed = st->table->localupdate;
1456 return NULL;
1457 }
1458 }
1459
Emeric Brun8e7a13e2021-04-28 11:48:15 +02001460 /* if distance between the last pushed and the retrieved key
1461 * is greater than the distance last_pushed and the local_update
1462 * this means we are beyond localupdate.
1463 */
1464 if ((eb->key - st->last_pushed) > (st->table->localupdate - st->last_pushed)) {
Frédéric Lécaille6a8303d2019-01-22 22:25:17 +01001465 st->table->commitupdate = st->last_pushed = st->table->localupdate;
1466 return NULL;
1467 }
1468
1469 return eb32_entry(eb, struct stksess, upd);
1470}
1471
1472/*
1473 * Function used to lookup for recent stick-table updates associated with
1474 * <st> shared stick-table during teach state 1 step.
1475 */
1476static inline struct stksess *peer_teach_stage1_stksess_lookup(struct shared_table *st)
1477{
1478 struct eb32_node *eb;
1479
1480 eb = eb32_lookup_ge(&st->table->updates, st->last_pushed+1);
1481 if (!eb) {
1482 st->flags |= SHTABLE_F_TEACH_STAGE1;
1483 eb = eb32_first(&st->table->updates);
1484 if (eb)
1485 st->last_pushed = eb->key - 1;
1486 return NULL;
1487 }
1488
1489 return eb32_entry(eb, struct stksess, upd);
1490}
1491
1492/*
1493 * Function used to lookup for recent stick-table updates associated with
1494 * <st> shared stick-table during teach state 2 step.
1495 */
1496static inline struct stksess *peer_teach_stage2_stksess_lookup(struct shared_table *st)
1497{
1498 struct eb32_node *eb;
1499
1500 eb = eb32_lookup_ge(&st->table->updates, st->last_pushed+1);
1501 if (!eb || eb->key > st->teaching_origin) {
1502 st->flags |= SHTABLE_F_TEACH_STAGE2;
1503 return NULL;
1504 }
1505
1506 return eb32_entry(eb, struct stksess, upd);
1507}
1508
1509/*
1510 * Generic function to emit update messages for <st> stick-table when a lesson must
1511 * be taught to the peer <p>.
1512 * <locked> must be set to 1 if the shared table <st> is already locked when entering
1513 * this function, 0 if not.
1514 *
1515 * This function temporary unlock/lock <st> when it sends stick-table updates or
1516 * when decrementing its refcount in case of any error when it sends this updates.
1517 *
1518 * Return 0 if any message could not be built modifying the appcxt st0 to PEER_SESS_ST_END value.
1519 * Returns -1 if there was not enough room left to send the message,
1520 * any other negative returned value must be considered as an error with an appcxt st0
1521 * returned value equal to PEER_SESS_ST_END.
1522 * If it returns 0 or -1, this function leave <st> locked if already locked when entering this function
1523 * unlocked if not already locked when entering this function.
1524 */
1525static inline int peer_send_teachmsgs(struct appctx *appctx, struct peer *p,
1526 struct stksess *(*peer_stksess_lookup)(struct shared_table *),
1527 struct shared_table *st, int locked)
1528{
1529 int ret, new_pushed, use_timed;
1530
1531 ret = 1;
1532 use_timed = 0;
1533 if (st != p->last_local_table) {
1534 ret = peer_send_switchmsg(st, appctx);
1535 if (ret <= 0)
1536 return ret;
1537
1538 p->last_local_table = st;
1539 }
1540
1541 if (peer_stksess_lookup != peer_teach_process_stksess_lookup)
1542 use_timed = !(p->flags & PEER_F_DWNGRD);
1543
1544 /* We force new pushed to 1 to force identifier in update message */
1545 new_pushed = 1;
1546
1547 if (!locked)
1548 HA_SPIN_LOCK(STK_TABLE_LOCK, &st->table->lock);
1549
1550 while (1) {
1551 struct stksess *ts;
1552 unsigned updateid;
1553
1554 /* push local updates */
1555 ts = peer_stksess_lookup(st);
1556 if (!ts)
1557 break;
1558
1559 updateid = ts->upd.key;
1560 ts->ref_cnt++;
1561 HA_SPIN_UNLOCK(STK_TABLE_LOCK, &st->table->lock);
1562
1563 ret = peer_send_updatemsg(st, appctx, ts, updateid, new_pushed, use_timed);
1564 if (ret <= 0) {
1565 HA_SPIN_LOCK(STK_TABLE_LOCK, &st->table->lock);
1566 ts->ref_cnt--;
1567 if (!locked)
1568 HA_SPIN_UNLOCK(STK_TABLE_LOCK, &st->table->lock);
1569 return ret;
1570 }
1571
1572 HA_SPIN_LOCK(STK_TABLE_LOCK, &st->table->lock);
1573 ts->ref_cnt--;
1574 st->last_pushed = updateid;
1575
1576 if (peer_stksess_lookup == peer_teach_process_stksess_lookup &&
1577 (int)(st->last_pushed - st->table->commitupdate) > 0)
1578 st->table->commitupdate = st->last_pushed;
1579
1580 /* identifier may not needed in next update message */
1581 new_pushed = 0;
1582 }
1583
1584 out:
1585 if (!locked)
1586 HA_SPIN_UNLOCK(STK_TABLE_LOCK, &st->table->lock);
1587 return 1;
1588}
1589
1590/*
1591 * Function to emit update messages for <st> stick-table when a lesson must
1592 * be taught to the peer <p> (PEER_F_LEARN_ASSIGN flag set).
1593 *
1594 * Note that <st> shared stick-table is locked when calling this function.
1595 *
1596 * Return 0 if any message could not be built modifying the appcxt st0 to PEER_SESS_ST_END value.
1597 * Returns -1 if there was not enough room left to send the message,
1598 * any other negative returned value must be considered as an error with an appcxt st0
1599 * returned value equal to PEER_SESS_ST_END.
1600 */
1601static inline int peer_send_teach_process_msgs(struct appctx *appctx, struct peer *p,
1602 struct shared_table *st)
1603{
1604 return peer_send_teachmsgs(appctx, p, peer_teach_process_stksess_lookup, st, 1);
1605}
1606
1607/*
1608 * Function to emit update messages for <st> stick-table when a lesson must
1609 * be taught to the peer <p> during teach state 1 step.
1610 *
1611 * Return 0 if any message could not be built modifying the appcxt st0 to PEER_SESS_ST_END value.
1612 * Returns -1 if there was not enough room left to send the message,
1613 * any other negative returned value must be considered as an error with an appcxt st0
1614 * returned value equal to PEER_SESS_ST_END.
1615 */
1616static inline int peer_send_teach_stage1_msgs(struct appctx *appctx, struct peer *p,
1617 struct shared_table *st)
1618{
1619 return peer_send_teachmsgs(appctx, p, peer_teach_stage1_stksess_lookup, st, 0);
1620}
1621
1622/*
1623 * Function to emit update messages for <st> stick-table when a lesson must
1624 * be taught to the peer <p> during teach state 1 step.
1625 *
1626 * Return 0 if any message could not be built modifying the appcxt st0 to PEER_SESS_ST_END value.
1627 * Returns -1 if there was not enough room left to send the message,
1628 * any other negative returned value must be considered as an error with an appcxt st0
1629 * returned value equal to PEER_SESS_ST_END.
1630 */
1631static inline int peer_send_teach_stage2_msgs(struct appctx *appctx, struct peer *p,
1632 struct shared_table *st)
1633{
1634 return peer_send_teachmsgs(appctx, p, peer_teach_stage2_stksess_lookup, st, 0);
1635}
1636
Frédéric Lécaille168a34b2019-01-23 11:16:57 +01001637
1638/*
1639 * Function used to parse a stick-table update message after it has been received
1640 * by <p> peer with <msg_cur> as address of the pointer to the position in the
1641 * receipt buffer with <msg_end> being position of the end of the stick-table message.
1642 * Update <msg_curr> accordingly to the peer protocol specs if no peer protocol error
1643 * was encountered.
1644 * <exp> must be set if the stick-table entry expires.
1645 * <updt> must be set for PEER_MSG_STKT_UPDATE or PEER_MSG_STKT_UPDATE_TIMED stick-table
Ilya Shipitsinc02a23f2020-05-06 00:53:22 +05001646 * messages, in this case the stick-table update message is received with a stick-table
Frédéric Lécaille168a34b2019-01-23 11:16:57 +01001647 * update ID.
1648 * <totl> is the length of the stick-table update message computed upon receipt.
1649 */
Frédéric Lécaille444243c2019-01-24 15:40:11 +01001650static int peer_treat_updatemsg(struct appctx *appctx, struct peer *p, int updt, int exp,
1651 char **msg_cur, char *msg_end, int msg_len, int totl)
Frédéric Lécaille168a34b2019-01-23 11:16:57 +01001652{
Christopher Faulet908628c2022-03-25 16:43:49 +01001653 struct conn_stream *cs = appctx->owner;
Frédéric Lécaille168a34b2019-01-23 11:16:57 +01001654 struct shared_table *st = p->remote_table;
1655 struct stksess *ts, *newts;
1656 uint32_t update;
1657 int expire;
1658 unsigned int data_type;
1659 void *data_ptr;
1660
Frédéric Lécailled8659352020-11-10 16:18:03 +01001661 TRACE_ENTER(PEERS_EV_UPDTMSG, NULL, p);
Frédéric Lécaille168a34b2019-01-23 11:16:57 +01001662 /* Here we have data message */
1663 if (!st)
1664 goto ignore_msg;
1665
1666 expire = MS_TO_TICKS(st->table->expire);
1667
1668 if (updt) {
Frédéric Lécailled8659352020-11-10 16:18:03 +01001669 if (msg_len < sizeof(update)) {
1670 TRACE_PROTO("malformed message", PEERS_EV_UPDTMSG, NULL, p);
Willy Tarreau1e82a142019-01-29 11:08:06 +01001671 goto malformed_exit;
Frédéric Lécailled8659352020-11-10 16:18:03 +01001672 }
Willy Tarreau1e82a142019-01-29 11:08:06 +01001673
Frédéric Lécaille168a34b2019-01-23 11:16:57 +01001674 memcpy(&update, *msg_cur, sizeof(update));
1675 *msg_cur += sizeof(update);
1676 st->last_get = htonl(update);
1677 }
1678 else {
1679 st->last_get++;
1680 }
1681
1682 if (exp) {
1683 size_t expire_sz = sizeof expire;
1684
Frédéric Lécailled8659352020-11-10 16:18:03 +01001685 if (*msg_cur + expire_sz > msg_end) {
1686 TRACE_PROTO("malformed message", PEERS_EV_UPDTMSG,
1687 NULL, p, *msg_cur);
1688 TRACE_PROTO("malformed message", PEERS_EV_UPDTMSG,
1689 NULL, p, msg_end, &expire_sz);
Willy Tarreau1e82a142019-01-29 11:08:06 +01001690 goto malformed_exit;
Frédéric Lécailled8659352020-11-10 16:18:03 +01001691 }
Willy Tarreau1e82a142019-01-29 11:08:06 +01001692
Frédéric Lécaille168a34b2019-01-23 11:16:57 +01001693 memcpy(&expire, *msg_cur, expire_sz);
1694 *msg_cur += expire_sz;
1695 expire = ntohl(expire);
1696 }
1697
1698 newts = stksess_new(st->table, NULL);
1699 if (!newts)
1700 goto ignore_msg;
1701
1702 if (st->table->type == SMP_T_STR) {
1703 unsigned int to_read, to_store;
1704
1705 to_read = intdecode(msg_cur, msg_end);
Frédéric Lécailled8659352020-11-10 16:18:03 +01001706 if (!*msg_cur) {
1707 TRACE_PROTO("malformed message", PEERS_EV_UPDTMSG, NULL, p);
Willy Tarreau1e82a142019-01-29 11:08:06 +01001708 goto malformed_free_newts;
Frédéric Lécailled8659352020-11-10 16:18:03 +01001709 }
Willy Tarreau1e82a142019-01-29 11:08:06 +01001710
Frédéric Lécaille168a34b2019-01-23 11:16:57 +01001711 to_store = MIN(to_read, st->table->key_size - 1);
Frédéric Lécailled8659352020-11-10 16:18:03 +01001712 if (*msg_cur + to_store > msg_end) {
1713 TRACE_PROTO("malformed message", PEERS_EV_UPDTMSG,
1714 NULL, p, *msg_cur);
1715 TRACE_PROTO("malformed message", PEERS_EV_UPDTMSG,
1716 NULL, p, msg_end, &to_store);
Willy Tarreau1e82a142019-01-29 11:08:06 +01001717 goto malformed_free_newts;
Frédéric Lécailled8659352020-11-10 16:18:03 +01001718 }
Frédéric Lécaille168a34b2019-01-23 11:16:57 +01001719
1720 memcpy(newts->key.key, *msg_cur, to_store);
1721 newts->key.key[to_store] = 0;
1722 *msg_cur += to_read;
1723 }
1724 else if (st->table->type == SMP_T_SINT) {
1725 unsigned int netinteger;
1726
Frédéric Lécailled8659352020-11-10 16:18:03 +01001727 if (*msg_cur + sizeof(netinteger) > msg_end) {
1728 TRACE_PROTO("malformed message", PEERS_EV_UPDTMSG,
1729 NULL, p, *msg_cur);
1730 TRACE_PROTO("malformed message", PEERS_EV_UPDTMSG,
1731 NULL, p, msg_end);
Willy Tarreau1e82a142019-01-29 11:08:06 +01001732 goto malformed_free_newts;
Frédéric Lécailled8659352020-11-10 16:18:03 +01001733 }
Willy Tarreau1e82a142019-01-29 11:08:06 +01001734
Frédéric Lécaille168a34b2019-01-23 11:16:57 +01001735 memcpy(&netinteger, *msg_cur, sizeof(netinteger));
1736 netinteger = ntohl(netinteger);
1737 memcpy(newts->key.key, &netinteger, sizeof(netinteger));
1738 *msg_cur += sizeof(netinteger);
1739 }
1740 else {
Frédéric Lécailled8659352020-11-10 16:18:03 +01001741 if (*msg_cur + st->table->key_size > msg_end) {
1742 TRACE_PROTO("malformed message", PEERS_EV_UPDTMSG,
1743 NULL, p, *msg_cur);
1744 TRACE_PROTO("malformed message", PEERS_EV_UPDTMSG,
1745 NULL, p, msg_end, &st->table->key_size);
Willy Tarreau1e82a142019-01-29 11:08:06 +01001746 goto malformed_free_newts;
Frédéric Lécailled8659352020-11-10 16:18:03 +01001747 }
Willy Tarreau1e82a142019-01-29 11:08:06 +01001748
Frédéric Lécaille168a34b2019-01-23 11:16:57 +01001749 memcpy(newts->key.key, *msg_cur, st->table->key_size);
1750 *msg_cur += st->table->key_size;
1751 }
1752
1753 /* lookup for existing entry */
1754 ts = stktable_set_entry(st->table, newts);
1755 if (ts != newts) {
1756 stksess_free(st->table, newts);
1757 newts = NULL;
1758 }
1759
1760 HA_RWLOCK_WRLOCK(STK_SESS_LOCK, &ts->lock);
1761
1762 for (data_type = 0 ; data_type < STKTABLE_DATA_TYPES ; data_type++) {
Willy Tarreau1e82a142019-01-29 11:08:06 +01001763 uint64_t decoded_int;
Emeric Brun90a9b672021-06-22 16:09:55 +02001764 unsigned int idx;
Willy Tarreaub4ff6f42021-12-24 13:38:49 +01001765 int ignore;
Frédéric Lécaille168a34b2019-01-23 11:16:57 +01001766
Emeric Brun08b0f672021-07-01 18:54:05 +02001767 if (!((1ULL << data_type) & st->remote_data))
Frédéric Lécaille168a34b2019-01-23 11:16:57 +01001768 continue;
Willy Tarreaudb2ab822021-10-08 17:53:12 +02001769
Willy Tarreaub4ff6f42021-12-24 13:38:49 +01001770 ignore = stktable_data_types[data_type].is_local;
Willy Tarreaudb2ab822021-10-08 17:53:12 +02001771
Emeric Brun90a9b672021-06-22 16:09:55 +02001772 if (stktable_data_types[data_type].is_array) {
1773 /* in case of array all elements
1774 * use the same std_type and they
1775 * are linearly encoded.
1776 * The number of elements was provided
1777 * by table definition message
1778 */
1779 switch (stktable_data_types[data_type].std_type) {
1780 case STD_T_SINT:
1781 for (idx = 0; idx < st->remote_data_nbelem[data_type]; idx++) {
1782 decoded_int = intdecode(msg_cur, msg_end);
1783 if (!*msg_cur) {
1784 TRACE_PROTO("malformed message", PEERS_EV_UPDTMSG, NULL, p);
1785 goto malformed_unlock;
1786 }
Frédéric Lécaille168a34b2019-01-23 11:16:57 +01001787
Emeric Brun90a9b672021-06-22 16:09:55 +02001788 data_ptr = stktable_data_ptr_idx(st->table, ts, data_type, idx);
Willy Tarreaub4ff6f42021-12-24 13:38:49 +01001789 if (data_ptr && !ignore)
Emeric Brun90a9b672021-06-22 16:09:55 +02001790 stktable_data_cast(data_ptr, std_t_sint) = decoded_int;
1791 }
1792 break;
1793 case STD_T_UINT:
1794 for (idx = 0; idx < st->remote_data_nbelem[data_type]; idx++) {
1795 decoded_int = intdecode(msg_cur, msg_end);
1796 if (!*msg_cur) {
1797 TRACE_PROTO("malformed message", PEERS_EV_UPDTMSG, NULL, p);
1798 goto malformed_unlock;
1799 }
1800
1801 data_ptr = stktable_data_ptr_idx(st->table, ts, data_type, idx);
Willy Tarreaub4ff6f42021-12-24 13:38:49 +01001802 if (data_ptr && !ignore)
Emeric Brun90a9b672021-06-22 16:09:55 +02001803 stktable_data_cast(data_ptr, std_t_uint) = decoded_int;
1804 }
1805 break;
1806 case STD_T_ULL:
1807 for (idx = 0; idx < st->remote_data_nbelem[data_type]; idx++) {
1808 decoded_int = intdecode(msg_cur, msg_end);
1809 if (!*msg_cur) {
1810 TRACE_PROTO("malformed message", PEERS_EV_UPDTMSG, NULL, p);
1811 goto malformed_unlock;
1812 }
1813
1814 data_ptr = stktable_data_ptr_idx(st->table, ts, data_type, idx);
Willy Tarreaub4ff6f42021-12-24 13:38:49 +01001815 if (data_ptr && !ignore)
Emeric Brun90a9b672021-06-22 16:09:55 +02001816 stktable_data_cast(data_ptr, std_t_ull) = decoded_int;
1817 }
1818 break;
1819 case STD_T_FRQP:
1820 for (idx = 0; idx < st->remote_data_nbelem[data_type]; idx++) {
1821 struct freq_ctr data;
1822
1823 /* First bit is reserved for the freq_ctr lock
1824 * Note: here we're still protected by the stksess lock
1825 * so we don't need to update the update the freq_ctr
1826 * using its internal lock.
1827 */
1828
1829 decoded_int = intdecode(msg_cur, msg_end);
1830 if (!*msg_cur) {
1831 TRACE_PROTO("malformed message", PEERS_EV_UPDTMSG, NULL, p);
1832 goto malformed_unlock;
1833 }
1834
1835 data.curr_tick = tick_add(now_ms, -decoded_int) & ~0x1;
1836 data.curr_ctr = intdecode(msg_cur, msg_end);
1837 if (!*msg_cur) {
1838 TRACE_PROTO("malformed message", PEERS_EV_UPDTMSG, NULL, p);
1839 goto malformed_unlock;
1840 }
1841
1842 data.prev_ctr = intdecode(msg_cur, msg_end);
1843 if (!*msg_cur) {
1844 TRACE_PROTO("malformed message", PEERS_EV_UPDTMSG, NULL, p);
1845 goto malformed_unlock;
1846 }
1847
1848 data_ptr = stktable_data_ptr_idx(st->table, ts, data_type, idx);
Willy Tarreaub4ff6f42021-12-24 13:38:49 +01001849 if (data_ptr && !ignore)
Emeric Brun90a9b672021-06-22 16:09:55 +02001850 stktable_data_cast(data_ptr, std_t_frqp) = data;
1851 }
1852 break;
1853 }
1854
1855 /* array is fully decoded
1856 * proceed next data_type.
1857 */
1858 continue;
1859 }
Willy Tarreau1e82a142019-01-29 11:08:06 +01001860 decoded_int = intdecode(msg_cur, msg_end);
Frédéric Lécailled8659352020-11-10 16:18:03 +01001861 if (!*msg_cur) {
1862 TRACE_PROTO("malformed message", PEERS_EV_UPDTMSG, NULL, p);
Willy Tarreau1e82a142019-01-29 11:08:06 +01001863 goto malformed_unlock;
Frédéric Lécailled8659352020-11-10 16:18:03 +01001864 }
Frédéric Lécaille168a34b2019-01-23 11:16:57 +01001865
Willy Tarreau1e82a142019-01-29 11:08:06 +01001866 switch (stktable_data_types[data_type].std_type) {
1867 case STD_T_SINT:
Frédéric Lécaille168a34b2019-01-23 11:16:57 +01001868 data_ptr = stktable_data_ptr(st->table, ts, data_type);
Willy Tarreaub4ff6f42021-12-24 13:38:49 +01001869 if (data_ptr && !ignore)
Willy Tarreau1e82a142019-01-29 11:08:06 +01001870 stktable_data_cast(data_ptr, std_t_sint) = decoded_int;
Frédéric Lécaille168a34b2019-01-23 11:16:57 +01001871 break;
Frédéric Lécaille168a34b2019-01-23 11:16:57 +01001872
Willy Tarreau1e82a142019-01-29 11:08:06 +01001873 case STD_T_UINT:
Frédéric Lécaille168a34b2019-01-23 11:16:57 +01001874 data_ptr = stktable_data_ptr(st->table, ts, data_type);
Willy Tarreaub4ff6f42021-12-24 13:38:49 +01001875 if (data_ptr && !ignore)
Willy Tarreau1e82a142019-01-29 11:08:06 +01001876 stktable_data_cast(data_ptr, std_t_uint) = decoded_int;
Frédéric Lécaille168a34b2019-01-23 11:16:57 +01001877 break;
Frédéric Lécaille168a34b2019-01-23 11:16:57 +01001878
Willy Tarreau1e82a142019-01-29 11:08:06 +01001879 case STD_T_ULL:
Frédéric Lécaille168a34b2019-01-23 11:16:57 +01001880 data_ptr = stktable_data_ptr(st->table, ts, data_type);
Willy Tarreaub4ff6f42021-12-24 13:38:49 +01001881 if (data_ptr && !ignore)
Willy Tarreau1e82a142019-01-29 11:08:06 +01001882 stktable_data_cast(data_ptr, std_t_ull) = decoded_int;
Frédéric Lécaille168a34b2019-01-23 11:16:57 +01001883 break;
Willy Tarreau1e82a142019-01-29 11:08:06 +01001884
Frédéric Lécaille168a34b2019-01-23 11:16:57 +01001885 case STD_T_FRQP: {
Willy Tarreaufa1258f2021-04-10 23:00:53 +02001886 struct freq_ctr data;
Frédéric Lécaille168a34b2019-01-23 11:16:57 +01001887
Willy Tarreaufa1258f2021-04-10 23:00:53 +02001888 /* First bit is reserved for the freq_ctr lock
Frédéric Lécaille168a34b2019-01-23 11:16:57 +01001889 Note: here we're still protected by the stksess lock
Willy Tarreaufa1258f2021-04-10 23:00:53 +02001890 so we don't need to update the update the freq_ctr
Emeric Brun90a9b672021-06-22 16:09:55 +02001891 using its internal lock.
1892 */
Frédéric Lécaille168a34b2019-01-23 11:16:57 +01001893
Willy Tarreau1e82a142019-01-29 11:08:06 +01001894 data.curr_tick = tick_add(now_ms, -decoded_int) & ~0x1;
Frédéric Lécaille168a34b2019-01-23 11:16:57 +01001895 data.curr_ctr = intdecode(msg_cur, msg_end);
Frédéric Lécailled8659352020-11-10 16:18:03 +01001896 if (!*msg_cur) {
1897 TRACE_PROTO("malformed message", PEERS_EV_UPDTMSG, NULL, p);
Willy Tarreau1e82a142019-01-29 11:08:06 +01001898 goto malformed_unlock;
Frédéric Lécailled8659352020-11-10 16:18:03 +01001899 }
Willy Tarreau1e82a142019-01-29 11:08:06 +01001900
Frédéric Lécaille168a34b2019-01-23 11:16:57 +01001901 data.prev_ctr = intdecode(msg_cur, msg_end);
Frédéric Lécailled8659352020-11-10 16:18:03 +01001902 if (!*msg_cur) {
1903 TRACE_PROTO("malformed message", PEERS_EV_UPDTMSG, NULL, p);
Willy Tarreau1e82a142019-01-29 11:08:06 +01001904 goto malformed_unlock;
Frédéric Lécailled8659352020-11-10 16:18:03 +01001905 }
Frédéric Lécaille168a34b2019-01-23 11:16:57 +01001906
1907 data_ptr = stktable_data_ptr(st->table, ts, data_type);
Willy Tarreaub4ff6f42021-12-24 13:38:49 +01001908 if (data_ptr && !ignore)
Frédéric Lécaille168a34b2019-01-23 11:16:57 +01001909 stktable_data_cast(data_ptr, std_t_frqp) = data;
1910 break;
1911 }
Frédéric Lécaille8d78fa72019-05-20 18:22:52 +02001912 case STD_T_DICT: {
1913 struct buffer *chunk;
1914 size_t data_len, value_len;
1915 unsigned int id;
1916 struct dict_entry *de;
1917 struct dcache *dc;
Frédéric Lécaille344e9482019-06-05 10:20:09 +02001918 char *end;
Frédéric Lécaille8d78fa72019-05-20 18:22:52 +02001919
Frédéric Lécailleaf9990f2019-11-13 17:50:34 +01001920 if (!decoded_int) {
1921 /* No entry. */
1922 break;
1923 }
Frédéric Lécaille8d78fa72019-05-20 18:22:52 +02001924 data_len = decoded_int;
Frédéric Lécailled8659352020-11-10 16:18:03 +01001925 if (*msg_cur + data_len > msg_end) {
1926 TRACE_PROTO("malformed message", PEERS_EV_UPDTMSG,
1927 NULL, p, *msg_cur);
1928 TRACE_PROTO("malformed message", PEERS_EV_UPDTMSG,
1929 NULL, p, msg_end, &data_len);
Frédéric Lécaille8d78fa72019-05-20 18:22:52 +02001930 goto malformed_unlock;
Frédéric Lécailled8659352020-11-10 16:18:03 +01001931 }
Frédéric Lécaille8d78fa72019-05-20 18:22:52 +02001932
Frédéric Lécaille344e9482019-06-05 10:20:09 +02001933 /* Compute the end of the current data, <msg_end> being at the end of
1934 * the entire message.
1935 */
1936 end = *msg_cur + data_len;
1937 id = intdecode(msg_cur, end);
Frédéric Lécailled8659352020-11-10 16:18:03 +01001938 if (!*msg_cur || !id) {
1939 TRACE_PROTO("malformed message", PEERS_EV_UPDTMSG,
1940 NULL, p, *msg_cur, &id);
Frédéric Lécaille8d78fa72019-05-20 18:22:52 +02001941 goto malformed_unlock;
Frédéric Lécailled8659352020-11-10 16:18:03 +01001942 }
Frédéric Lécaille8d78fa72019-05-20 18:22:52 +02001943
1944 dc = p->dcache;
Frédéric Lécaille344e9482019-06-05 10:20:09 +02001945 if (*msg_cur == end) {
Frédéric Lécaille8d78fa72019-05-20 18:22:52 +02001946 /* Dictionary entry key without value. */
Frédéric Lécaillef9e51be2020-11-12 19:53:11 +01001947 if (id > dc->max_entries) {
1948 TRACE_PROTO("malformed message", PEERS_EV_UPDTMSG,
1949 NULL, p, NULL, &id);
1950 goto malformed_unlock;
1951 }
Frédéric Lécaille8d78fa72019-05-20 18:22:52 +02001952 /* IDs sent over the network are numbered from 1. */
1953 de = dc->rx[id - 1].de;
1954 }
1955 else {
1956 chunk = get_trash_chunk();
Frédéric Lécaille344e9482019-06-05 10:20:09 +02001957 value_len = intdecode(msg_cur, end);
1958 if (!*msg_cur || *msg_cur + value_len > end ||
Frédéric Lécailled8659352020-11-10 16:18:03 +01001959 unlikely(value_len + 1 >= chunk->size)) {
1960 TRACE_PROTO("malformed message", PEERS_EV_UPDTMSG,
1961 NULL, p, *msg_cur, &value_len);
1962 TRACE_PROTO("malformed message", PEERS_EV_UPDTMSG,
1963 NULL, p, end, &chunk->size);
Frédéric Lécaille8d78fa72019-05-20 18:22:52 +02001964 goto malformed_unlock;
Frédéric Lécailled8659352020-11-10 16:18:03 +01001965 }
Frédéric Lécaille8d78fa72019-05-20 18:22:52 +02001966
1967 chunk_memcpy(chunk, *msg_cur, value_len);
1968 chunk->area[chunk->data] = '\0';
Frédéric Lécaille56aec0d2019-06-06 14:14:15 +02001969 *msg_cur += value_len;
1970
Thayne McCombs92149f92020-11-20 01:28:26 -07001971 de = dict_insert(&server_key_dict, chunk->area);
1972 dict_entry_unref(&server_key_dict, dc->rx[id - 1].de);
Frédéric Lécaille8d78fa72019-05-20 18:22:52 +02001973 dc->rx[id - 1].de = de;
1974 }
1975 if (de) {
1976 data_ptr = stktable_data_ptr(st->table, ts, data_type);
Willy Tarreaub4ff6f42021-12-24 13:38:49 +01001977 if (data_ptr && !ignore) {
Willy Tarreau4781b152021-04-06 13:53:36 +02001978 HA_ATOMIC_INC(&de->refcount);
Frédéric Lécaille8d78fa72019-05-20 18:22:52 +02001979 stktable_data_cast(data_ptr, std_t_dict) = de;
Thayne McCombs92149f92020-11-20 01:28:26 -07001980 }
Frédéric Lécaille8d78fa72019-05-20 18:22:52 +02001981 }
1982 break;
1983 }
Frédéric Lécaille168a34b2019-01-23 11:16:57 +01001984 }
1985 }
1986 /* Force new expiration */
1987 ts->expire = tick_add(now_ms, expire);
1988
1989 HA_RWLOCK_WRUNLOCK(STK_SESS_LOCK, &ts->lock);
1990 stktable_touch_remote(st->table, ts, 1);
Frédéric Lécailled8659352020-11-10 16:18:03 +01001991 TRACE_LEAVE(PEERS_EV_UPDTMSG, NULL, p);
Frédéric Lécaille168a34b2019-01-23 11:16:57 +01001992 return 1;
1993
1994 ignore_msg:
1995 /* skip consumed message */
Christopher Faulet908628c2022-03-25 16:43:49 +01001996 co_skip(cs_oc(cs), totl);
Frédéric Lécailled8659352020-11-10 16:18:03 +01001997 TRACE_DEVEL("leaving in error", PEERS_EV_UPDTMSG);
Frédéric Lécaille168a34b2019-01-23 11:16:57 +01001998 return 0;
Willy Tarreau1e82a142019-01-29 11:08:06 +01001999
2000 malformed_unlock:
2001 /* malformed message */
2002 HA_RWLOCK_WRUNLOCK(STK_SESS_LOCK, &ts->lock);
2003 stktable_touch_remote(st->table, ts, 1);
2004 appctx->st0 = PEER_SESS_ST_ERRPROTO;
Frédéric Lécailled8659352020-11-10 16:18:03 +01002005 TRACE_DEVEL("leaving in error", PEERS_EV_UPDTMSG);
Willy Tarreau1e82a142019-01-29 11:08:06 +01002006 return 0;
2007
2008 malformed_free_newts:
2009 /* malformed message */
2010 stksess_free(st->table, newts);
2011 malformed_exit:
2012 appctx->st0 = PEER_SESS_ST_ERRPROTO;
Frédéric Lécailled8659352020-11-10 16:18:03 +01002013 TRACE_DEVEL("leaving in error", PEERS_EV_UPDTMSG);
Willy Tarreau1e82a142019-01-29 11:08:06 +01002014 return 0;
Frédéric Lécaille168a34b2019-01-23 11:16:57 +01002015}
2016
Frédéric Lécaille87f554c2019-01-22 17:26:50 +01002017/*
Frédéric Lécailled27b0942019-01-23 17:31:37 +01002018 * Function used to parse a stick-table update acknowledgement message after it
2019 * has been received by <p> peer with <msg_cur> as address of the pointer to the position in the
2020 * receipt buffer with <msg_end> being the position of the end of the stick-table message.
2021 * Update <msg_curr> accordingly to the peer protocol specs if no peer protocol error
2022 * was encountered.
2023 * Return 1 if succeeded, 0 if not with the appctx state st0 set to PEER_SESS_ST_ERRPROTO.
2024 */
2025static inline int peer_treat_ackmsg(struct appctx *appctx, struct peer *p,
2026 char **msg_cur, char *msg_end)
2027{
2028 /* ack message */
2029 uint32_t table_id ;
2030 uint32_t update;
2031 struct shared_table *st;
2032
Emeric Brunb0d60be2021-03-04 10:27:10 +01002033 /* ignore ack during teaching process */
2034 if (p->flags & PEER_F_TEACH_PROCESS)
2035 return 1;
2036
Frédéric Lécailled27b0942019-01-23 17:31:37 +01002037 table_id = intdecode(msg_cur, msg_end);
2038 if (!*msg_cur || (*msg_cur + sizeof(update) > msg_end)) {
2039 /* malformed message */
Frédéric Lécailleda2b0842021-01-15 16:21:28 +01002040
2041 TRACE_PROTO("malformed message", PEERS_EV_ACKMSG,
2042 NULL, p, *msg_cur);
Frédéric Lécailled27b0942019-01-23 17:31:37 +01002043 appctx->st0 = PEER_SESS_ST_ERRPROTO;
2044 return 0;
2045 }
2046
2047 memcpy(&update, *msg_cur, sizeof(update));
2048 update = ntohl(update);
2049
2050 for (st = p->tables; st; st = st->next) {
2051 if (st->local_id == table_id) {
2052 st->update = update;
2053 break;
2054 }
2055 }
2056
2057 return 1;
2058}
2059
2060/*
2061 * Function used to parse a stick-table switch message after it has been received
2062 * by <p> peer with <msg_cur> as address of the pointer to the position in the
2063 * receipt buffer with <msg_end> being the position of the end of the stick-table message.
2064 * Update <msg_curr> accordingly to the peer protocol specs if no peer protocol error
2065 * was encountered.
2066 * Return 1 if succeeded, 0 if not with the appctx state st0 set to PEER_SESS_ST_ERRPROTO.
2067 */
2068static inline int peer_treat_switchmsg(struct appctx *appctx, struct peer *p,
2069 char **msg_cur, char *msg_end)
2070{
2071 struct shared_table *st;
2072 int table_id;
2073
2074 table_id = intdecode(msg_cur, msg_end);
2075 if (!*msg_cur) {
Frédéric Lécailleda2b0842021-01-15 16:21:28 +01002076 TRACE_PROTO("malformed message", PEERS_EV_SWTCMSG, NULL, p);
Frédéric Lécailled27b0942019-01-23 17:31:37 +01002077 /* malformed message */
2078 appctx->st0 = PEER_SESS_ST_ERRPROTO;
2079 return 0;
2080 }
2081
2082 p->remote_table = NULL;
2083 for (st = p->tables; st; st = st->next) {
2084 if (st->remote_id == table_id) {
2085 p->remote_table = st;
2086 break;
2087 }
2088 }
2089
2090 return 1;
2091}
2092
2093/*
2094 * Function used to parse a stick-table definition message after it has been received
2095 * by <p> peer with <msg_cur> as address of the pointer to the position in the
2096 * receipt buffer with <msg_end> being the position of the end of the stick-table message.
2097 * Update <msg_curr> accordingly to the peer protocol specs if no peer protocol error
2098 * was encountered.
2099 * <totl> is the length of the stick-table update message computed upon receipt.
2100 * Return 1 if succeeded, 0 if not with the appctx state st0 set to PEER_SESS_ST_ERRPROTO.
2101 */
2102static inline int peer_treat_definemsg(struct appctx *appctx, struct peer *p,
2103 char **msg_cur, char *msg_end, int totl)
2104{
Christopher Faulet908628c2022-03-25 16:43:49 +01002105 struct conn_stream *cs = appctx->owner;
Frédéric Lécailled27b0942019-01-23 17:31:37 +01002106 int table_id_len;
2107 struct shared_table *st;
2108 int table_type;
2109 int table_keylen;
2110 int table_id;
2111 uint64_t table_data;
2112
2113 table_id = intdecode(msg_cur, msg_end);
Frédéric Lécailleda2b0842021-01-15 16:21:28 +01002114 if (!*msg_cur) {
2115 TRACE_PROTO("malformed message", PEERS_EV_DEFMSG, NULL, p);
Willy Tarreau6f731f32019-01-29 11:11:23 +01002116 goto malformed_exit;
Frédéric Lécailleda2b0842021-01-15 16:21:28 +01002117 }
Frédéric Lécailled27b0942019-01-23 17:31:37 +01002118
2119 table_id_len = intdecode(msg_cur, msg_end);
Frédéric Lécailleda2b0842021-01-15 16:21:28 +01002120 if (!*msg_cur) {
2121 TRACE_PROTO("malformed message", PEERS_EV_DEFMSG, NULL, p, *msg_cur);
Willy Tarreau6f731f32019-01-29 11:11:23 +01002122 goto malformed_exit;
Frédéric Lécailleda2b0842021-01-15 16:21:28 +01002123 }
Frédéric Lécailled27b0942019-01-23 17:31:37 +01002124
2125 p->remote_table = NULL;
Frédéric Lécailleda2b0842021-01-15 16:21:28 +01002126 if (!table_id_len || (*msg_cur + table_id_len) >= msg_end) {
2127 TRACE_PROTO("malformed message", PEERS_EV_DEFMSG, NULL, p, *msg_cur, &table_id_len);
Willy Tarreau6f731f32019-01-29 11:11:23 +01002128 goto malformed_exit;
Frédéric Lécailleda2b0842021-01-15 16:21:28 +01002129 }
Frédéric Lécailled27b0942019-01-23 17:31:37 +01002130
2131 for (st = p->tables; st; st = st->next) {
2132 /* Reset IDs */
2133 if (st->remote_id == table_id)
2134 st->remote_id = 0;
2135
Frédéric Lécaille7fcc24d2019-03-20 15:09:45 +01002136 if (!p->remote_table && (table_id_len == strlen(st->table->nid)) &&
2137 (memcmp(st->table->nid, *msg_cur, table_id_len) == 0))
Frédéric Lécailled27b0942019-01-23 17:31:37 +01002138 p->remote_table = st;
2139 }
2140
Frédéric Lécailleda2b0842021-01-15 16:21:28 +01002141 if (!p->remote_table) {
2142 TRACE_PROTO("ignored message", PEERS_EV_DEFMSG, NULL, p);
Frédéric Lécailled27b0942019-01-23 17:31:37 +01002143 goto ignore_msg;
Frédéric Lécailleda2b0842021-01-15 16:21:28 +01002144 }
Frédéric Lécailled27b0942019-01-23 17:31:37 +01002145
2146 *msg_cur += table_id_len;
Frédéric Lécailleda2b0842021-01-15 16:21:28 +01002147 if (*msg_cur >= msg_end) {
2148 TRACE_PROTO("malformed message", PEERS_EV_DEFMSG, NULL, p);
Willy Tarreau6f731f32019-01-29 11:11:23 +01002149 goto malformed_exit;
Frédéric Lécailleda2b0842021-01-15 16:21:28 +01002150 }
Frédéric Lécailled27b0942019-01-23 17:31:37 +01002151
2152 table_type = intdecode(msg_cur, msg_end);
Frédéric Lécailleda2b0842021-01-15 16:21:28 +01002153 if (!*msg_cur) {
2154 TRACE_PROTO("malformed message", PEERS_EV_DEFMSG, NULL, p);
Willy Tarreau6f731f32019-01-29 11:11:23 +01002155 goto malformed_exit;
Frédéric Lécailleda2b0842021-01-15 16:21:28 +01002156 }
Frédéric Lécailled27b0942019-01-23 17:31:37 +01002157
2158 table_keylen = intdecode(msg_cur, msg_end);
Frédéric Lécailleda2b0842021-01-15 16:21:28 +01002159 if (!*msg_cur) {
2160 TRACE_PROTO("malformed message", PEERS_EV_DEFMSG, NULL, p);
Willy Tarreau6f731f32019-01-29 11:11:23 +01002161 goto malformed_exit;
Frédéric Lécailleda2b0842021-01-15 16:21:28 +01002162 }
Frédéric Lécailled27b0942019-01-23 17:31:37 +01002163
2164 table_data = intdecode(msg_cur, msg_end);
Frédéric Lécailleda2b0842021-01-15 16:21:28 +01002165 if (!*msg_cur) {
2166 TRACE_PROTO("malformed message", PEERS_EV_DEFMSG, NULL, p);
Willy Tarreau6f731f32019-01-29 11:11:23 +01002167 goto malformed_exit;
Frédéric Lécailleda2b0842021-01-15 16:21:28 +01002168 }
Frédéric Lécailled27b0942019-01-23 17:31:37 +01002169
Emeric Brun530ba382020-06-02 11:17:42 +02002170 if (p->remote_table->table->type != peer_int_key_type[table_type]
Frédéric Lécailled27b0942019-01-23 17:31:37 +01002171 || p->remote_table->table->key_size != table_keylen) {
2172 p->remote_table = NULL;
Frédéric Lécailleda2b0842021-01-15 16:21:28 +01002173 TRACE_PROTO("ignored message", PEERS_EV_DEFMSG, NULL, p);
Frédéric Lécailled27b0942019-01-23 17:31:37 +01002174 goto ignore_msg;
2175 }
2176
Ilya Shipitsin01881082021-08-07 14:41:56 +05002177 /* Check if there there is the additional expire data */
Emeric Brun90a9b672021-06-22 16:09:55 +02002178 intdecode(msg_cur, msg_end);
2179 if (*msg_cur) {
2180 uint64_t data_type;
2181 uint64_t type;
2182
2183 /* This define contains the expire data so we consider
2184 * it also contain all data_types parameters.
2185 */
2186 for (data_type = 0; data_type < STKTABLE_DATA_TYPES; data_type++) {
2187 if (table_data & (1ULL << data_type)) {
2188 if (stktable_data_types[data_type].is_array) {
2189 /* This should be an array
2190 * so we parse the data_type prefix
2191 * because we must have parameters.
2192 */
2193 type = intdecode(msg_cur, msg_end);
2194 if (!*msg_cur) {
2195 p->remote_table = NULL;
2196 TRACE_PROTO("missing meta data for array", PEERS_EV_DEFMSG, NULL, p);
2197 goto ignore_msg;
2198 }
2199
2200 /* check if the data_type match the current from the bitfield */
2201 if (type != data_type) {
2202 p->remote_table = NULL;
Ilya Shipitsin01881082021-08-07 14:41:56 +05002203 TRACE_PROTO("meta data mismatch type", PEERS_EV_DEFMSG, NULL, p);
Emeric Brun90a9b672021-06-22 16:09:55 +02002204 goto ignore_msg;
2205 }
2206
2207 /* decode the nbelem of the array */
2208 p->remote_table->remote_data_nbelem[type] = intdecode(msg_cur, msg_end);
2209 if (!*msg_cur) {
2210 p->remote_table = NULL;
2211 TRACE_PROTO("missing array size meta data for array", PEERS_EV_DEFMSG, NULL, p);
2212 goto ignore_msg;
2213 }
2214
2215 /* if it is an array of frqp, we must also have the period to decode */
2216 if (stktable_data_types[data_type].std_type == STD_T_FRQP) {
2217 intdecode(msg_cur, msg_end);
2218 if (!*msg_cur) {
2219 p->remote_table = NULL;
2220 TRACE_PROTO("missing period for frqp", PEERS_EV_DEFMSG, NULL, p);
2221 goto ignore_msg;
2222 }
2223 }
2224 }
2225 else if (stktable_data_types[data_type].std_type == STD_T_FRQP) {
2226 /* This should be a std freq counter data_type
2227 * so we parse the data_type prefix
2228 * because we must have parameters.
2229 */
2230 type = intdecode(msg_cur, msg_end);
2231 if (!*msg_cur) {
2232 p->remote_table = NULL;
2233 TRACE_PROTO("missing meta data for frqp", PEERS_EV_DEFMSG, NULL, p);
2234 goto ignore_msg;
2235 }
2236
2237 /* check if the data_type match the current from the bitfield */
2238 if (type != data_type) {
2239 p->remote_table = NULL;
Ilya Shipitsin01881082021-08-07 14:41:56 +05002240 TRACE_PROTO("meta data mismatch type", PEERS_EV_DEFMSG, NULL, p);
Emeric Brun90a9b672021-06-22 16:09:55 +02002241 goto ignore_msg;
2242 }
2243
2244 /* decode the period */
2245 intdecode(msg_cur, msg_end);
2246 if (!*msg_cur) {
2247 p->remote_table = NULL;
2248 TRACE_PROTO("missing period for frqp", PEERS_EV_DEFMSG, NULL, p);
2249 goto ignore_msg;
2250 }
2251 }
2252 }
2253 }
2254 }
2255 else {
2256 uint64_t data_type;
2257
2258 /* There is not additional data but
2259 * array size parameter is mandatory to parse array
2260 * so we consider an error if an array data_type is define
2261 * but there is no additional data.
2262 */
2263 for (data_type = 0; data_type < STKTABLE_DATA_TYPES; data_type++) {
2264 if (table_data & (1ULL << data_type)) {
2265 if (stktable_data_types[data_type].is_array) {
2266 p->remote_table = NULL;
2267 TRACE_PROTO("missing array size meta data for array", PEERS_EV_DEFMSG, NULL, p);
2268 goto ignore_msg;
2269 }
2270 }
2271 }
2272 }
2273
Frédéric Lécailled27b0942019-01-23 17:31:37 +01002274 p->remote_table->remote_data = table_data;
2275 p->remote_table->remote_id = table_id;
2276 return 1;
2277
2278 ignore_msg:
Christopher Faulet908628c2022-03-25 16:43:49 +01002279 co_skip(cs_oc(cs), totl);
Frédéric Lécailled27b0942019-01-23 17:31:37 +01002280 return 0;
Willy Tarreau6f731f32019-01-29 11:11:23 +01002281
2282 malformed_exit:
2283 /* malformed message */
2284 appctx->st0 = PEER_SESS_ST_ERRPROTO;
2285 return 0;
Frédéric Lécailled27b0942019-01-23 17:31:37 +01002286}
2287
2288/*
Willy Tarreau1dfd4f12020-11-13 14:10:20 +01002289 * Receive a stick-table message or pre-parse any other message.
2290 * The message's header will be sent into <msg_head> which must be at least
2291 * <msg_head_sz> bytes long (at least 7 to store 32-bit variable lengths).
2292 * The first two bytes are always read, and the rest is only read if the
2293 * first bytes indicate a stick-table message. If the message is a stick-table
2294 * message, the varint is decoded and the equivalent number of bytes will be
2295 * copied into the trash at trash.area. <totl> is incremented by the number of
2296 * bytes read EVEN IN CASE OF INCOMPLETE MESSAGES.
Frédéric Lécaille95203f22019-01-23 19:38:11 +01002297 * Returns 1 if there was no error, if not, returns 0 if not enough data were available,
2298 * -1 if there was an error updating the appctx state st0 accordingly.
2299 */
2300static inline int peer_recv_msg(struct appctx *appctx, char *msg_head, size_t msg_head_sz,
2301 uint32_t *msg_len, int *totl)
2302{
2303 int reql;
Christopher Faulet908628c2022-03-25 16:43:49 +01002304 struct conn_stream *cs = appctx->owner;
Willy Tarreau1dfd4f12020-11-13 14:10:20 +01002305 char *cur;
Frédéric Lécaille95203f22019-01-23 19:38:11 +01002306
Christopher Faulet908628c2022-03-25 16:43:49 +01002307 reql = co_getblk(cs_oc(cs), msg_head, 2 * sizeof(char), *totl);
Frédéric Lécaille95203f22019-01-23 19:38:11 +01002308 if (reql <= 0) /* closed or EOL not found */
2309 goto incomplete;
2310
2311 *totl += reql;
2312
Frédéric Lécaille36fb77e2019-06-04 08:28:19 +02002313 if (!(msg_head[1] & PEER_MSG_STKT_BIT_MASK))
Frédéric Lécaille95203f22019-01-23 19:38:11 +01002314 return 1;
2315
Willy Tarreau1dfd4f12020-11-13 14:10:20 +01002316 /* This is a stick-table message, let's go on */
2317
Frédéric Lécaille95203f22019-01-23 19:38:11 +01002318 /* Read and Decode message length */
Willy Tarreau1dfd4f12020-11-13 14:10:20 +01002319 msg_head += *totl;
2320 msg_head_sz -= *totl;
Christopher Faulet908628c2022-03-25 16:43:49 +01002321 reql = co_data(cs_oc(cs)) - *totl;
Willy Tarreau1dfd4f12020-11-13 14:10:20 +01002322 if (reql > msg_head_sz)
2323 reql = msg_head_sz;
2324
Christopher Faulet908628c2022-03-25 16:43:49 +01002325 reql = co_getblk(cs_oc(cs), msg_head, reql, *totl);
Frédéric Lécaille95203f22019-01-23 19:38:11 +01002326 if (reql <= 0) /* closed */
2327 goto incomplete;
2328
Willy Tarreau1dfd4f12020-11-13 14:10:20 +01002329 cur = msg_head;
2330 *msg_len = intdecode(&cur, cur + reql);
2331 if (!cur) {
2332 /* the number is truncated, did we read enough ? */
2333 if (reql < msg_head_sz)
2334 goto incomplete;
Frédéric Lécaille95203f22019-01-23 19:38:11 +01002335
Willy Tarreau1dfd4f12020-11-13 14:10:20 +01002336 /* malformed message */
2337 TRACE_PROTO("malformed message: too large length encoding", PEERS_EV_UPDTMSG);
2338 appctx->st0 = PEER_SESS_ST_ERRPROTO;
2339 return -1;
Frédéric Lécaille95203f22019-01-23 19:38:11 +01002340 }
Willy Tarreau1dfd4f12020-11-13 14:10:20 +01002341 *totl += cur - msg_head;
Frédéric Lécaille95203f22019-01-23 19:38:11 +01002342
2343 /* Read message content */
2344 if (*msg_len) {
2345 if (*msg_len > trash.size) {
2346 /* Status code is not success, abort */
2347 appctx->st0 = PEER_SESS_ST_ERRSIZE;
2348 return -1;
2349 }
2350
Christopher Faulet908628c2022-03-25 16:43:49 +01002351 reql = co_getblk(cs_oc(cs), trash.area, *msg_len, *totl);
Frédéric Lécaille95203f22019-01-23 19:38:11 +01002352 if (reql <= 0) /* closed */
2353 goto incomplete;
2354 *totl += reql;
2355 }
2356
2357 return 1;
2358
2359 incomplete:
Christopher Faulet908628c2022-03-25 16:43:49 +01002360 if (reql < 0 || (cs_oc(cs)->flags & (CF_SHUTW|CF_SHUTW_NOW))) {
Willy Tarreau345ebcf2020-11-26 17:06:04 +01002361 /* there was an error or the message was truncated */
Frédéric Lécaille95203f22019-01-23 19:38:11 +01002362 appctx->st0 = PEER_SESS_ST_END;
2363 return -1;
2364 }
2365
2366 return 0;
2367}
Frédéric Lécaille444243c2019-01-24 15:40:11 +01002368
2369/*
2370 * Treat the awaited message with <msg_head> as header.*
2371 * Return 1 if succeeded, 0 if not.
2372 */
2373static inline int peer_treat_awaited_msg(struct appctx *appctx, struct peer *peer, unsigned char *msg_head,
2374 char **msg_cur, char *msg_end, int msg_len, int totl)
2375{
Christopher Faulet908628c2022-03-25 16:43:49 +01002376 struct stream *s = __cs_strm(appctx->owner);
Frédéric Lécaille444243c2019-01-24 15:40:11 +01002377 struct peers *peers = strm_fe(s)->parent;
2378
2379 if (msg_head[0] == PEER_MSG_CLASS_CONTROL) {
2380 if (msg_head[1] == PEER_MSG_CTRL_RESYNCREQ) {
2381 struct shared_table *st;
2382 /* Reset message: remote need resync */
2383
Frédéric Lécailleda2b0842021-01-15 16:21:28 +01002384 TRACE_PROTO("received control message", PEERS_EV_CTRLMSG,
2385 NULL, &msg_head[1], peers->local->id, peer->id);
Ilya Shipitsinc02a23f2020-05-06 00:53:22 +05002386 /* prepare tables for a global push */
Frédéric Lécaille444243c2019-01-24 15:40:11 +01002387 for (st = peer->tables; st; st = st->next) {
Emeric Brun437e48a2021-04-28 09:49:33 +02002388 st->teaching_origin = st->last_pushed = st->update;
Frédéric Lécaille444243c2019-01-24 15:40:11 +01002389 st->flags = 0;
2390 }
2391
2392 /* reset teaching flags to 0 */
2393 peer->flags &= PEER_TEACH_RESET;
2394
2395 /* flag to start to teach lesson */
2396 peer->flags |= PEER_F_TEACH_PROCESS;
Emeric Brunccdfbae2021-04-28 12:59:35 +02002397 peers->flags |= PEERS_F_RESYNC_REQUESTED;
Frédéric Lécaille444243c2019-01-24 15:40:11 +01002398 }
2399 else if (msg_head[1] == PEER_MSG_CTRL_RESYNCFINISHED) {
Frédéric Lécailleda2b0842021-01-15 16:21:28 +01002400 TRACE_PROTO("received control message", PEERS_EV_CTRLMSG,
2401 NULL, &msg_head[1], peers->local->id, peer->id);
Frédéric Lécaille444243c2019-01-24 15:40:11 +01002402 if (peer->flags & PEER_F_LEARN_ASSIGN) {
2403 peer->flags &= ~PEER_F_LEARN_ASSIGN;
2404 peers->flags &= ~(PEERS_F_RESYNC_ASSIGN|PEERS_F_RESYNC_PROCESS);
2405 peers->flags |= (PEERS_F_RESYNC_LOCAL|PEERS_F_RESYNC_REMOTE);
Emeric Brunccdfbae2021-04-28 12:59:35 +02002406 if (peer->local)
2407 peers->flags |= PEERS_F_RESYNC_LOCALFINISHED;
2408 else
2409 peers->flags |= PEERS_F_RESYNC_REMOTEFINISHED;
Frédéric Lécaille444243c2019-01-24 15:40:11 +01002410 }
2411 peer->confirm++;
2412 }
2413 else if (msg_head[1] == PEER_MSG_CTRL_RESYNCPARTIAL) {
Frédéric Lécailleda2b0842021-01-15 16:21:28 +01002414 TRACE_PROTO("received control message", PEERS_EV_CTRLMSG,
2415 NULL, &msg_head[1], peers->local->id, peer->id);
Frédéric Lécaille444243c2019-01-24 15:40:11 +01002416 if (peer->flags & PEER_F_LEARN_ASSIGN) {
2417 peer->flags &= ~PEER_F_LEARN_ASSIGN;
2418 peers->flags &= ~(PEERS_F_RESYNC_ASSIGN|PEERS_F_RESYNC_PROCESS);
2419
Emeric Brunccdfbae2021-04-28 12:59:35 +02002420 if (peer->local)
2421 peers->flags |= PEERS_F_RESYNC_LOCALPARTIAL;
2422 else
2423 peers->flags |= PEERS_F_RESYNC_REMOTEPARTIAL;
Frédéric Lécaille444243c2019-01-24 15:40:11 +01002424 peer->flags |= PEER_F_LEARN_NOTUP2DATE;
Frédéric Lécaille54bff832019-03-26 10:25:20 +01002425 peers->resync_timeout = tick_add(now_ms, MS_TO_TICKS(PEER_RESYNC_TIMEOUT));
Frédéric Lécaille444243c2019-01-24 15:40:11 +01002426 task_wakeup(peers->sync_task, TASK_WOKEN_MSG);
2427 }
2428 peer->confirm++;
2429 }
2430 else if (msg_head[1] == PEER_MSG_CTRL_RESYNCCONFIRM) {
2431 struct shared_table *st;
2432
Frédéric Lécailleda2b0842021-01-15 16:21:28 +01002433 TRACE_PROTO("received control message", PEERS_EV_CTRLMSG,
2434 NULL, &msg_head[1], peers->local->id, peer->id);
Frédéric Lécaille444243c2019-01-24 15:40:11 +01002435 /* If stopping state */
2436 if (stopping) {
2437 /* Close session, push resync no more needed */
2438 peer->flags |= PEER_F_TEACH_COMPLETE;
2439 appctx->st0 = PEER_SESS_ST_END;
2440 return 0;
2441 }
2442 for (st = peer->tables; st; st = st->next) {
2443 st->update = st->last_pushed = st->teaching_origin;
2444 st->flags = 0;
2445 }
2446
2447 /* reset teaching flags to 0 */
2448 peer->flags &= PEER_TEACH_RESET;
2449 }
Frédéric Lécaille645635d2019-02-11 17:49:39 +01002450 else if (msg_head[1] == PEER_MSG_CTRL_HEARTBEAT) {
Frédéric Lécailleda2b0842021-01-15 16:21:28 +01002451 TRACE_PROTO("received control message", PEERS_EV_CTRLMSG,
2452 NULL, &msg_head[1], peers->local->id, peer->id);
Frédéric Lécaille54bff832019-03-26 10:25:20 +01002453 peer->reconnect = tick_add(now_ms, MS_TO_TICKS(PEER_RECONNECT_TIMEOUT));
Frédéric Lécaille33cab3c2019-11-06 11:51:26 +01002454 peer->rx_hbt++;
Frédéric Lécaille645635d2019-02-11 17:49:39 +01002455 }
Frédéric Lécaille444243c2019-01-24 15:40:11 +01002456 }
2457 else if (msg_head[0] == PEER_MSG_CLASS_STICKTABLE) {
2458 if (msg_head[1] == PEER_MSG_STKT_DEFINE) {
2459 if (!peer_treat_definemsg(appctx, peer, msg_cur, msg_end, totl))
2460 return 0;
2461 }
2462 else if (msg_head[1] == PEER_MSG_STKT_SWITCH) {
2463 if (!peer_treat_switchmsg(appctx, peer, msg_cur, msg_end))
2464 return 0;
2465 }
2466 else if (msg_head[1] == PEER_MSG_STKT_UPDATE ||
2467 msg_head[1] == PEER_MSG_STKT_INCUPDATE ||
2468 msg_head[1] == PEER_MSG_STKT_UPDATE_TIMED ||
2469 msg_head[1] == PEER_MSG_STKT_INCUPDATE_TIMED) {
2470 int update, expire;
2471
2472 update = msg_head[1] == PEER_MSG_STKT_UPDATE || msg_head[1] == PEER_MSG_STKT_UPDATE_TIMED;
2473 expire = msg_head[1] == PEER_MSG_STKT_UPDATE_TIMED || msg_head[1] == PEER_MSG_STKT_INCUPDATE_TIMED;
2474 if (!peer_treat_updatemsg(appctx, peer, update, expire,
2475 msg_cur, msg_end, msg_len, totl))
2476 return 0;
2477
2478 }
2479 else if (msg_head[1] == PEER_MSG_STKT_ACK) {
2480 if (!peer_treat_ackmsg(appctx, peer, msg_cur, msg_end))
2481 return 0;
2482 }
2483 }
2484 else if (msg_head[0] == PEER_MSG_CLASS_RESERVED) {
2485 appctx->st0 = PEER_SESS_ST_ERRPROTO;
2486 return 0;
2487 }
2488
2489 return 1;
2490}
Frédéric Lécaille25e1d5e2019-01-24 17:33:48 +01002491
2492
2493/*
2494 * Send any message to <peer> peer.
2495 * Returns 1 if succeeded, or -1 or 0 if failed.
Ilya Shipitsinc02a23f2020-05-06 00:53:22 +05002496 * -1 means an internal error occurred, 0 is for a peer protocol error leading
Frédéric Lécaille25e1d5e2019-01-24 17:33:48 +01002497 * to a peer state change (from the peer I/O handler point of view).
2498 */
Frédéric Lécailleda2b0842021-01-15 16:21:28 +01002499static inline int peer_send_msgs(struct appctx *appctx,
2500 struct peer *peer, struct peers *peers)
Frédéric Lécaille25e1d5e2019-01-24 17:33:48 +01002501{
2502 int repl;
Frédéric Lécaille25e1d5e2019-01-24 17:33:48 +01002503
2504 /* Need to request a resync */
2505 if ((peer->flags & PEER_F_LEARN_ASSIGN) &&
2506 (peers->flags & PEERS_F_RESYNC_ASSIGN) &&
2507 !(peers->flags & PEERS_F_RESYNC_PROCESS)) {
2508
Frédéric Lécailleda2b0842021-01-15 16:21:28 +01002509 repl = peer_send_resync_reqmsg(appctx, peer, peers);
Frédéric Lécaille25e1d5e2019-01-24 17:33:48 +01002510 if (repl <= 0)
2511 return repl;
2512
2513 peers->flags |= PEERS_F_RESYNC_PROCESS;
2514 }
2515
2516 /* Nothing to read, now we start to write */
2517 if (peer->tables) {
2518 struct shared_table *st;
2519 struct shared_table *last_local_table;
2520
2521 last_local_table = peer->last_local_table;
2522 if (!last_local_table)
2523 last_local_table = peer->tables;
2524 st = last_local_table->next;
2525
2526 while (1) {
2527 if (!st)
2528 st = peer->tables;
2529
2530 /* It remains some updates to ack */
2531 if (st->last_get != st->last_acked) {
2532 repl = peer_send_ackmsg(st, appctx);
2533 if (repl <= 0)
2534 return repl;
2535
2536 st->last_acked = st->last_get;
2537 }
2538
2539 if (!(peer->flags & PEER_F_TEACH_PROCESS)) {
2540 HA_SPIN_LOCK(STK_TABLE_LOCK, &st->table->lock);
2541 if (!(peer->flags & PEER_F_LEARN_ASSIGN) &&
Emeric Brun8e7a13e2021-04-28 11:48:15 +02002542 (st->last_pushed != st->table->localupdate)) {
Frédéric Lécaille25e1d5e2019-01-24 17:33:48 +01002543
2544 repl = peer_send_teach_process_msgs(appctx, peer, st);
2545 if (repl <= 0) {
2546 HA_SPIN_UNLOCK(STK_TABLE_LOCK, &st->table->lock);
2547 return repl;
2548 }
2549 }
2550 HA_SPIN_UNLOCK(STK_TABLE_LOCK, &st->table->lock);
2551 }
Emeric Brun1675ada2021-04-22 18:13:13 +02002552 else if (!(peer->flags & PEER_F_TEACH_FINISHED)) {
Frédéric Lécaille25e1d5e2019-01-24 17:33:48 +01002553 if (!(st->flags & SHTABLE_F_TEACH_STAGE1)) {
2554 repl = peer_send_teach_stage1_msgs(appctx, peer, st);
2555 if (repl <= 0)
2556 return repl;
2557 }
2558
2559 if (!(st->flags & SHTABLE_F_TEACH_STAGE2)) {
2560 repl = peer_send_teach_stage2_msgs(appctx, peer, st);
2561 if (repl <= 0)
2562 return repl;
2563 }
2564 }
2565
2566 if (st == last_local_table)
2567 break;
2568 st = st->next;
2569 }
2570 }
2571
2572 if ((peer->flags & PEER_F_TEACH_PROCESS) && !(peer->flags & PEER_F_TEACH_FINISHED)) {
Frédéric Lécailleda2b0842021-01-15 16:21:28 +01002573 repl = peer_send_resync_finishedmsg(appctx, peer, peers);
Frédéric Lécaille25e1d5e2019-01-24 17:33:48 +01002574 if (repl <= 0)
2575 return repl;
2576
2577 /* flag finished message sent */
2578 peer->flags |= PEER_F_TEACH_FINISHED;
2579 }
2580
2581 /* Confirm finished or partial messages */
2582 while (peer->confirm) {
Frédéric Lécailleda2b0842021-01-15 16:21:28 +01002583 repl = peer_send_resync_confirmsg(appctx, peer, peers);
Frédéric Lécaille25e1d5e2019-01-24 17:33:48 +01002584 if (repl <= 0)
2585 return repl;
2586
2587 peer->confirm--;
2588 }
2589
2590 return 1;
2591}
2592
Frédéric Lécaille95203f22019-01-23 19:38:11 +01002593/*
Frédéric Lécaille3f0fb9d2019-01-25 08:30:29 +01002594 * Read and parse a first line of a "hello" peer protocol message.
2595 * Returns 0 if could not read a line, -1 if there was a read error or
2596 * the line is malformed, 1 if succeeded.
2597 */
2598static inline int peer_getline_version(struct appctx *appctx,
2599 unsigned int *maj_ver, unsigned int *min_ver)
2600{
2601 int reql;
2602
2603 reql = peer_getline(appctx);
2604 if (!reql)
2605 return 0;
2606
2607 if (reql < 0)
2608 return -1;
2609
2610 /* test protocol */
2611 if (strncmp(PEER_SESSION_PROTO_NAME " ", trash.area, proto_len + 1) != 0) {
2612 appctx->st0 = PEER_SESS_ST_EXIT;
2613 appctx->st1 = PEER_SESS_SC_ERRPROTO;
2614 return -1;
2615 }
2616 if (peer_get_version(trash.area + proto_len + 1, maj_ver, min_ver) == -1 ||
2617 *maj_ver != PEER_MAJOR_VER || *min_ver > PEER_MINOR_VER) {
2618 appctx->st0 = PEER_SESS_ST_EXIT;
2619 appctx->st1 = PEER_SESS_SC_ERRVERSION;
2620 return -1;
2621 }
2622
2623 return 1;
2624}
2625
2626/*
2627 * Read and parse a second line of a "hello" peer protocol message.
2628 * Returns 0 if could not read a line, -1 if there was a read error or
2629 * the line is malformed, 1 if succeeded.
2630 */
2631static inline int peer_getline_host(struct appctx *appctx)
2632{
2633 int reql;
2634
2635 reql = peer_getline(appctx);
2636 if (!reql)
2637 return 0;
2638
2639 if (reql < 0)
2640 return -1;
2641
2642 /* test hostname match */
2643 if (strcmp(localpeer, trash.area) != 0) {
2644 appctx->st0 = PEER_SESS_ST_EXIT;
2645 appctx->st1 = PEER_SESS_SC_ERRHOST;
2646 return -1;
2647 }
2648
2649 return 1;
2650}
2651
2652/*
2653 * Read and parse a last line of a "hello" peer protocol message.
2654 * Returns 0 if could not read a character, -1 if there was a read error or
2655 * the line is malformed, 1 if succeeded.
2656 * Set <curpeer> accordingly (the remote peer sending the "hello" message).
2657 */
2658static inline int peer_getline_last(struct appctx *appctx, struct peer **curpeer)
2659{
2660 char *p;
2661 int reql;
2662 struct peer *peer;
Christopher Faulet908628c2022-03-25 16:43:49 +01002663 struct stream *s = __cs_strm(appctx->owner);
Frédéric Lécaille3f0fb9d2019-01-25 08:30:29 +01002664 struct peers *peers = strm_fe(s)->parent;
2665
2666 reql = peer_getline(appctx);
2667 if (!reql)
2668 return 0;
2669
2670 if (reql < 0)
2671 return -1;
2672
2673 /* parse line "<peer name> <pid> <relative_pid>" */
2674 p = strchr(trash.area, ' ');
2675 if (!p) {
2676 appctx->st0 = PEER_SESS_ST_EXIT;
2677 appctx->st1 = PEER_SESS_SC_ERRPROTO;
2678 return -1;
2679 }
2680 *p = 0;
2681
2682 /* lookup known peer */
2683 for (peer = peers->remote; peer; peer = peer->next) {
2684 if (strcmp(peer->id, trash.area) == 0)
2685 break;
2686 }
2687
2688 /* if unknown peer */
2689 if (!peer) {
2690 appctx->st0 = PEER_SESS_ST_EXIT;
2691 appctx->st1 = PEER_SESS_SC_ERRPEER;
2692 return -1;
2693 }
2694 *curpeer = peer;
2695
2696 return 1;
2697}
2698
2699/*
Frédéric Lécaille4b2fd9b2019-01-25 08:58:41 +01002700 * Init <peer> peer after having accepted it at peer protocol level.
2701 */
2702static inline void init_accepted_peer(struct peer *peer, struct peers *peers)
2703{
2704 struct shared_table *st;
2705
Frédéric Lécaillebaeb9192020-10-14 11:50:26 +02002706 peer->heartbeat = tick_add(now_ms, MS_TO_TICKS(PEER_HEARTBEAT_TIMEOUT));
Frédéric Lécaille4b2fd9b2019-01-25 08:58:41 +01002707 /* Register status code */
2708 peer->statuscode = PEER_SESS_SC_SUCCESSCODE;
Frédéric Lécaille3fc0fe02020-10-08 09:46:24 +02002709 peer->last_hdshk = now_ms;
Frédéric Lécaille4b2fd9b2019-01-25 08:58:41 +01002710
2711 /* Awake main task */
2712 task_wakeup(peers->sync_task, TASK_WOKEN_MSG);
2713
2714 /* Init confirm counter */
2715 peer->confirm = 0;
2716
2717 /* Init cursors */
2718 for (st = peer->tables; st ; st = st->next) {
2719 st->last_get = st->last_acked = 0;
Emeric Brund9729da2021-02-23 16:50:53 +01002720 HA_SPIN_LOCK(STK_TABLE_LOCK, &st->table->lock);
2721 /* if st->update appears to be in future it means
2722 * that the last acked value is very old and we
2723 * remain unconnected a too long time to use this
2724 * acknowlegement as a reset.
2725 * We should update the protocol to be able to
2726 * signal the remote peer that it needs a full resync.
2727 * Here a partial fix consist to set st->update at
2728 * the max past value
2729 */
2730 if ((int)(st->table->localupdate - st->update) < 0)
2731 st->update = st->table->localupdate + (2147483648U);
Frédéric Lécaille4b2fd9b2019-01-25 08:58:41 +01002732 st->teaching_origin = st->last_pushed = st->update;
Emeric Brun1a6b43e2021-04-20 14:43:46 +02002733 st->flags = 0;
Emeric Bruncc9cce92021-02-23 17:08:08 +01002734 if ((int)(st->last_pushed - st->table->commitupdate) > 0)
2735 st->table->commitupdate = st->last_pushed;
Emeric Brund9729da2021-02-23 16:50:53 +01002736 HA_SPIN_UNLOCK(STK_TABLE_LOCK, &st->table->lock);
Frédéric Lécaille4b2fd9b2019-01-25 08:58:41 +01002737 }
2738
2739 /* reset teaching and learning flags to 0 */
2740 peer->flags &= PEER_TEACH_RESET;
2741 peer->flags &= PEER_LEARN_RESET;
2742
2743 /* if current peer is local */
2744 if (peer->local) {
Ilya Shipitsin46a030c2020-07-05 16:36:08 +05002745 /* if current host need resyncfrom local and no process assigned */
Frédéric Lécaille4b2fd9b2019-01-25 08:58:41 +01002746 if ((peers->flags & PEERS_RESYNC_STATEMASK) == PEERS_RESYNC_FROMLOCAL &&
2747 !(peers->flags & PEERS_F_RESYNC_ASSIGN)) {
2748 /* assign local peer for a lesson, consider lesson already requested */
2749 peer->flags |= PEER_F_LEARN_ASSIGN;
2750 peers->flags |= (PEERS_F_RESYNC_ASSIGN|PEERS_F_RESYNC_PROCESS);
Emeric Brunccdfbae2021-04-28 12:59:35 +02002751 peers->flags |= PEERS_F_RESYNC_LOCALASSIGN;
Frédéric Lécaille4b2fd9b2019-01-25 08:58:41 +01002752 }
2753
2754 }
2755 else if ((peers->flags & PEERS_RESYNC_STATEMASK) == PEERS_RESYNC_FROMREMOTE &&
2756 !(peers->flags & PEERS_F_RESYNC_ASSIGN)) {
2757 /* assign peer for a lesson */
2758 peer->flags |= PEER_F_LEARN_ASSIGN;
2759 peers->flags |= PEERS_F_RESYNC_ASSIGN;
Emeric Brunccdfbae2021-04-28 12:59:35 +02002760 peers->flags |= PEERS_F_RESYNC_REMOTEASSIGN;
Frédéric Lécaille4b2fd9b2019-01-25 08:58:41 +01002761 }
2762}
2763
2764/*
2765 * Init <peer> peer after having connected it at peer protocol level.
2766 */
2767static inline void init_connected_peer(struct peer *peer, struct peers *peers)
2768{
2769 struct shared_table *st;
2770
Frédéric Lécaillebaeb9192020-10-14 11:50:26 +02002771 peer->heartbeat = tick_add(now_ms, MS_TO_TICKS(PEER_HEARTBEAT_TIMEOUT));
Frédéric Lécaille4b2fd9b2019-01-25 08:58:41 +01002772 /* Init cursors */
2773 for (st = peer->tables; st ; st = st->next) {
2774 st->last_get = st->last_acked = 0;
Emeric Brund9729da2021-02-23 16:50:53 +01002775 HA_SPIN_LOCK(STK_TABLE_LOCK, &st->table->lock);
2776 /* if st->update appears to be in future it means
2777 * that the last acked value is very old and we
2778 * remain unconnected a too long time to use this
2779 * acknowlegement as a reset.
2780 * We should update the protocol to be able to
2781 * signal the remote peer that it needs a full resync.
2782 * Here a partial fix consist to set st->update at
2783 * the max past value.
2784 */
2785 if ((int)(st->table->localupdate - st->update) < 0)
2786 st->update = st->table->localupdate + (2147483648U);
Frédéric Lécaille4b2fd9b2019-01-25 08:58:41 +01002787 st->teaching_origin = st->last_pushed = st->update;
Emeric Brun1a6b43e2021-04-20 14:43:46 +02002788 st->flags = 0;
Emeric Bruncc9cce92021-02-23 17:08:08 +01002789 if ((int)(st->last_pushed - st->table->commitupdate) > 0)
2790 st->table->commitupdate = st->last_pushed;
Emeric Brund9729da2021-02-23 16:50:53 +01002791 HA_SPIN_UNLOCK(STK_TABLE_LOCK, &st->table->lock);
Frédéric Lécaille4b2fd9b2019-01-25 08:58:41 +01002792 }
2793
2794 /* Init confirm counter */
2795 peer->confirm = 0;
2796
2797 /* reset teaching and learning flags to 0 */
2798 peer->flags &= PEER_TEACH_RESET;
2799 peer->flags &= PEER_LEARN_RESET;
2800
2801 /* If current peer is local */
2802 if (peer->local) {
2803 /* flag to start to teach lesson */
2804 peer->flags |= PEER_F_TEACH_PROCESS;
2805 }
2806 else if ((peers->flags & PEERS_RESYNC_STATEMASK) == PEERS_RESYNC_FROMREMOTE &&
2807 !(peers->flags & PEERS_F_RESYNC_ASSIGN)) {
2808 /* If peer is remote and resync from remote is needed,
2809 and no peer currently assigned */
2810
2811 /* assign peer for a lesson */
2812 peer->flags |= PEER_F_LEARN_ASSIGN;
2813 peers->flags |= PEERS_F_RESYNC_ASSIGN;
Emeric Brunccdfbae2021-04-28 12:59:35 +02002814 peers->flags |= PEERS_F_RESYNC_REMOTEASSIGN;
Frédéric Lécaille4b2fd9b2019-01-25 08:58:41 +01002815 }
2816}
2817
2818/*
Ilya Shipitsin46a030c2020-07-05 16:36:08 +05002819 * IO Handler to handle message exchange with a peer
Emeric Brun2b920a12010-09-23 18:30:22 +02002820 */
Willy Tarreau00a37f02015-04-13 12:05:19 +02002821static void peer_io_handler(struct appctx *appctx)
Emeric Brun2b920a12010-09-23 18:30:22 +02002822{
Christopher Faulet908628c2022-03-25 16:43:49 +01002823 struct conn_stream *cs = appctx->owner;
2824 struct stream *s = __cs_strm(cs);
Vincent Bernat3c2f2f22016-04-03 13:48:42 +02002825 struct peers *curpeers = strm_fe(s)->parent;
Emeric Brun80527f52017-06-19 17:46:37 +02002826 struct peer *curpeer = NULL;
Emeric Brun2b920a12010-09-23 18:30:22 +02002827 int reql = 0;
2828 int repl = 0;
Frédéric Lécaille523cc9e2016-10-12 17:30:30 +02002829 unsigned int maj_ver, min_ver;
Willy Tarreau2d372c22018-11-05 17:12:27 +01002830 int prev_state;
Emeric Brun2b920a12010-09-23 18:30:22 +02002831
Joseph Herlant82b2f542018-11-15 12:19:14 -08002832 /* Check if the input buffer is available. */
Christopher Faulet908628c2022-03-25 16:43:49 +01002833 if (cs_ib(cs)->size == 0) {
Christopher Fauleta0bdec32022-04-04 07:51:21 +02002834 cs_rx_room_blk(cs);
Frédéric Lécaille7d0ceee2019-01-24 14:24:05 +01002835 goto out;
2836 }
Christopher Fauleta73e59b2016-12-09 17:30:18 +01002837
Emeric Brun2b920a12010-09-23 18:30:22 +02002838 while (1) {
Willy Tarreau2d372c22018-11-05 17:12:27 +01002839 prev_state = appctx->st0;
Emeric Brun2b920a12010-09-23 18:30:22 +02002840switchstate:
Frédéric Lécaille523cc9e2016-10-12 17:30:30 +02002841 maj_ver = min_ver = (unsigned int)-1;
Willy Tarreau7b4b4992013-12-01 09:15:12 +01002842 switch(appctx->st0) {
Willy Tarreaue4d927a2013-12-01 12:47:35 +01002843 case PEER_SESS_ST_ACCEPT:
Willy Tarreau2d372c22018-11-05 17:12:27 +01002844 prev_state = appctx->st0;
Willy Tarreau7b4b4992013-12-01 09:15:12 +01002845 appctx->ctx.peers.ptr = NULL;
Willy Tarreaue4d927a2013-12-01 12:47:35 +01002846 appctx->st0 = PEER_SESS_ST_GETVERSION;
Emeric Brun2b920a12010-09-23 18:30:22 +02002847 /* fall through */
Willy Tarreaue4d927a2013-12-01 12:47:35 +01002848 case PEER_SESS_ST_GETVERSION:
Willy Tarreau2d372c22018-11-05 17:12:27 +01002849 prev_state = appctx->st0;
Frédéric Lécaille3f0fb9d2019-01-25 08:30:29 +01002850 reql = peer_getline_version(appctx, &maj_ver, &min_ver);
2851 if (reql <= 0) {
2852 if (!reql)
2853 goto out;
Emeric Brun2b920a12010-09-23 18:30:22 +02002854 goto switchstate;
2855 }
2856
Willy Tarreaue4d927a2013-12-01 12:47:35 +01002857 appctx->st0 = PEER_SESS_ST_GETHOST;
Emeric Brun2b920a12010-09-23 18:30:22 +02002858 /* fall through */
Willy Tarreaue4d927a2013-12-01 12:47:35 +01002859 case PEER_SESS_ST_GETHOST:
Willy Tarreau2d372c22018-11-05 17:12:27 +01002860 prev_state = appctx->st0;
Frédéric Lécaille3f0fb9d2019-01-25 08:30:29 +01002861 reql = peer_getline_host(appctx);
2862 if (reql <= 0) {
2863 if (!reql)
2864 goto out;
Emeric Brun2b920a12010-09-23 18:30:22 +02002865 goto switchstate;
2866 }
2867
Willy Tarreaue4d927a2013-12-01 12:47:35 +01002868 appctx->st0 = PEER_SESS_ST_GETPEER;
Emeric Brun2b920a12010-09-23 18:30:22 +02002869 /* fall through */
Willy Tarreaue4d927a2013-12-01 12:47:35 +01002870 case PEER_SESS_ST_GETPEER: {
Willy Tarreau2d372c22018-11-05 17:12:27 +01002871 prev_state = appctx->st0;
Frédéric Lécaille3f0fb9d2019-01-25 08:30:29 +01002872 reql = peer_getline_last(appctx, &curpeer);
2873 if (reql <= 0) {
2874 if (!reql)
2875 goto out;
Emeric Brun2b920a12010-09-23 18:30:22 +02002876 goto switchstate;
2877 }
Emeric Brun2b920a12010-09-23 18:30:22 +02002878
Christopher Faulet2a944ee2017-11-07 10:42:54 +01002879 HA_SPIN_LOCK(PEER_LOCK, &curpeer->lock);
Willy Tarreau9df94c22016-10-31 18:42:52 +01002880 if (curpeer->appctx && curpeer->appctx != appctx) {
Emeric Brunb3971ab2015-05-12 18:49:09 +02002881 if (curpeer->local) {
2882 /* Local connection, reply a retry */
2883 appctx->st0 = PEER_SESS_ST_EXIT;
2884 appctx->st1 = PEER_SESS_SC_TRYAGAIN;
2885 goto switchstate;
Emeric Brun2b920a12010-09-23 18:30:22 +02002886 }
Emeric Brun80527f52017-06-19 17:46:37 +02002887
2888 /* we're killing a connection, we must apply a random delay before
2889 * retrying otherwise the other end will do the same and we can loop
2890 * for a while.
2891 */
Willy Tarreau52bf8392020-03-08 00:42:37 +01002892 curpeer->reconnect = tick_add(now_ms, MS_TO_TICKS(50 + ha_random() % 2000));
Emeric Brun9ef2ad72019-04-02 17:22:01 +02002893 peer_session_forceshutdown(curpeer);
Frédéric Lécaillebaeb9192020-10-14 11:50:26 +02002894 curpeer->heartbeat = TICK_ETERNITY;
Frédéric Lécaille3fc0fe02020-10-08 09:46:24 +02002895 curpeer->coll++;
Emeric Brun2b920a12010-09-23 18:30:22 +02002896 }
Frédéric Lécaille523cc9e2016-10-12 17:30:30 +02002897 if (maj_ver != (unsigned int)-1 && min_ver != (unsigned int)-1) {
2898 if (min_ver == PEER_DWNGRD_MINOR_VER) {
2899 curpeer->flags |= PEER_F_DWNGRD;
2900 }
2901 else {
2902 curpeer->flags &= ~PEER_F_DWNGRD;
2903 }
2904 }
Emeric Brunb3971ab2015-05-12 18:49:09 +02002905 curpeer->appctx = appctx;
Frédéric Lécaillebaeb9192020-10-14 11:50:26 +02002906 curpeer->flags |= PEER_F_ALIVE;
Emeric Brunb3971ab2015-05-12 18:49:09 +02002907 appctx->ctx.peers.ptr = curpeer;
Willy Tarreaue4d927a2013-12-01 12:47:35 +01002908 appctx->st0 = PEER_SESS_ST_SENDSUCCESS;
Willy Tarreau4781b152021-04-06 13:53:36 +02002909 _HA_ATOMIC_INC(&active_peers);
Emeric Brun2b920a12010-09-23 18:30:22 +02002910 }
Tim Duesterhus588b3142020-05-29 14:35:51 +02002911 /* fall through */
Willy Tarreaue4d927a2013-12-01 12:47:35 +01002912 case PEER_SESS_ST_SENDSUCCESS: {
Willy Tarreau2d372c22018-11-05 17:12:27 +01002913 prev_state = appctx->st0;
Emeric Brun80527f52017-06-19 17:46:37 +02002914 if (!curpeer) {
2915 curpeer = appctx->ctx.peers.ptr;
Christopher Faulet2a944ee2017-11-07 10:42:54 +01002916 HA_SPIN_LOCK(PEER_LOCK, &curpeer->lock);
Emeric Brun80527f52017-06-19 17:46:37 +02002917 if (curpeer->appctx != appctx) {
2918 appctx->st0 = PEER_SESS_ST_END;
2919 goto switchstate;
2920 }
2921 }
Frédéric Lécaille7d0ceee2019-01-24 14:24:05 +01002922
2923 repl = peer_send_status_successmsg(appctx);
Emeric Brun2b920a12010-09-23 18:30:22 +02002924 if (repl <= 0) {
2925 if (repl == -1)
Frédéric Lécaille7d0ceee2019-01-24 14:24:05 +01002926 goto out;
Emeric Brun2b920a12010-09-23 18:30:22 +02002927 goto switchstate;
2928 }
2929
Frédéric Lécaille4b2fd9b2019-01-25 08:58:41 +01002930 init_accepted_peer(curpeer, curpeers);
Emeric Brunb3971ab2015-05-12 18:49:09 +02002931
Emeric Brun2b920a12010-09-23 18:30:22 +02002932 /* switch to waiting message state */
Willy Tarreau4781b152021-04-06 13:53:36 +02002933 _HA_ATOMIC_INC(&connected_peers);
Willy Tarreaue4d927a2013-12-01 12:47:35 +01002934 appctx->st0 = PEER_SESS_ST_WAITMSG;
Emeric Brun2b920a12010-09-23 18:30:22 +02002935 goto switchstate;
2936 }
Willy Tarreaue4d927a2013-12-01 12:47:35 +01002937 case PEER_SESS_ST_CONNECT: {
Willy Tarreau2d372c22018-11-05 17:12:27 +01002938 prev_state = appctx->st0;
Emeric Brun80527f52017-06-19 17:46:37 +02002939 if (!curpeer) {
2940 curpeer = appctx->ctx.peers.ptr;
Christopher Faulet2a944ee2017-11-07 10:42:54 +01002941 HA_SPIN_LOCK(PEER_LOCK, &curpeer->lock);
Emeric Brun80527f52017-06-19 17:46:37 +02002942 if (curpeer->appctx != appctx) {
2943 appctx->st0 = PEER_SESS_ST_END;
2944 goto switchstate;
2945 }
2946 }
Emeric Brun2b920a12010-09-23 18:30:22 +02002947
Frédéric Lécaille7d0ceee2019-01-24 14:24:05 +01002948 repl = peer_send_hellomsg(appctx, curpeer);
Emeric Brun2b920a12010-09-23 18:30:22 +02002949 if (repl <= 0) {
2950 if (repl == -1)
Frédéric Lécaille7d0ceee2019-01-24 14:24:05 +01002951 goto out;
Emeric Brun2b920a12010-09-23 18:30:22 +02002952 goto switchstate;
2953 }
2954
2955 /* switch to the waiting statuscode state */
Willy Tarreaue4d927a2013-12-01 12:47:35 +01002956 appctx->st0 = PEER_SESS_ST_GETSTATUS;
Emeric Brun2b920a12010-09-23 18:30:22 +02002957 }
Tim Duesterhus588b3142020-05-29 14:35:51 +02002958 /* fall through */
Willy Tarreaue4d927a2013-12-01 12:47:35 +01002959 case PEER_SESS_ST_GETSTATUS: {
Willy Tarreau2d372c22018-11-05 17:12:27 +01002960 prev_state = appctx->st0;
Emeric Brun80527f52017-06-19 17:46:37 +02002961 if (!curpeer) {
2962 curpeer = appctx->ctx.peers.ptr;
Christopher Faulet2a944ee2017-11-07 10:42:54 +01002963 HA_SPIN_LOCK(PEER_LOCK, &curpeer->lock);
Emeric Brun80527f52017-06-19 17:46:37 +02002964 if (curpeer->appctx != appctx) {
2965 appctx->st0 = PEER_SESS_ST_END;
2966 goto switchstate;
2967 }
2968 }
2969
Christopher Faulet908628c2022-03-25 16:43:49 +01002970 if (cs_ic(cs)->flags & CF_WRITE_PARTIAL)
Emeric Brunb3971ab2015-05-12 18:49:09 +02002971 curpeer->statuscode = PEER_SESS_SC_CONNECTEDCODE;
Emeric Brun2b920a12010-09-23 18:30:22 +02002972
Frédéric Lécaillece025572019-01-21 13:38:06 +01002973 reql = peer_getline(appctx);
2974 if (!reql)
2975 goto out;
Emeric Brun2b920a12010-09-23 18:30:22 +02002976
Frédéric Lécaillece025572019-01-21 13:38:06 +01002977 if (reql < 0)
2978 goto switchstate;
Emeric Brun2b920a12010-09-23 18:30:22 +02002979
2980 /* Register status code */
Willy Tarreau843b7cb2018-07-13 10:54:26 +02002981 curpeer->statuscode = atoi(trash.area);
Frédéric Lécaille3fc0fe02020-10-08 09:46:24 +02002982 curpeer->last_hdshk = now_ms;
Emeric Brun2b920a12010-09-23 18:30:22 +02002983
2984 /* Awake main task */
Frédéric Lécailleed2b4a62017-07-13 09:07:09 +02002985 task_wakeup(curpeers->sync_task, TASK_WOKEN_MSG);
Emeric Brun2b920a12010-09-23 18:30:22 +02002986
2987 /* If status code is success */
Emeric Brunb3971ab2015-05-12 18:49:09 +02002988 if (curpeer->statuscode == PEER_SESS_SC_SUCCESSCODE) {
Frédéric Lécaille4b2fd9b2019-01-25 08:58:41 +01002989 init_connected_peer(curpeer, curpeers);
Emeric Brun2b920a12010-09-23 18:30:22 +02002990 }
2991 else {
Frédéric Lécaille523cc9e2016-10-12 17:30:30 +02002992 if (curpeer->statuscode == PEER_SESS_SC_ERRVERSION)
2993 curpeer->flags |= PEER_F_DWNGRD;
Emeric Brun2b920a12010-09-23 18:30:22 +02002994 /* Status code is not success, abort */
Willy Tarreaue4d927a2013-12-01 12:47:35 +01002995 appctx->st0 = PEER_SESS_ST_END;
Emeric Brun2b920a12010-09-23 18:30:22 +02002996 goto switchstate;
2997 }
Willy Tarreau4781b152021-04-06 13:53:36 +02002998 _HA_ATOMIC_INC(&connected_peers);
Willy Tarreaue4d927a2013-12-01 12:47:35 +01002999 appctx->st0 = PEER_SESS_ST_WAITMSG;
Emeric Brun2b920a12010-09-23 18:30:22 +02003000 }
Tim Duesterhus588b3142020-05-29 14:35:51 +02003001 /* fall through */
Willy Tarreaue4d927a2013-12-01 12:47:35 +01003002 case PEER_SESS_ST_WAITMSG: {
Emeric Brunb3971ab2015-05-12 18:49:09 +02003003 uint32_t msg_len = 0;
Willy Tarreau843b7cb2018-07-13 10:54:26 +02003004 char *msg_cur = trash.area;
3005 char *msg_end = trash.area;
Willy Tarreau1dfd4f12020-11-13 14:10:20 +01003006 unsigned char msg_head[7]; // 2 + 5 for varint32
Emeric Brun2b920a12010-09-23 18:30:22 +02003007 int totl = 0;
3008
Willy Tarreau2d372c22018-11-05 17:12:27 +01003009 prev_state = appctx->st0;
Emeric Brun80527f52017-06-19 17:46:37 +02003010 if (!curpeer) {
3011 curpeer = appctx->ctx.peers.ptr;
Christopher Faulet2a944ee2017-11-07 10:42:54 +01003012 HA_SPIN_LOCK(PEER_LOCK, &curpeer->lock);
Emeric Brun80527f52017-06-19 17:46:37 +02003013 if (curpeer->appctx != appctx) {
3014 appctx->st0 = PEER_SESS_ST_END;
3015 goto switchstate;
3016 }
3017 }
3018
Frédéric Lécaille95203f22019-01-23 19:38:11 +01003019 reql = peer_recv_msg(appctx, (char *)msg_head, sizeof msg_head, &msg_len, &totl);
3020 if (reql <= 0) {
3021 if (reql == -1)
3022 goto switchstate;
Frédéric Lécaillebe825e52019-01-24 18:28:44 +01003023 goto send_msgs;
Emeric Brunb3971ab2015-05-12 18:49:09 +02003024 }
Willy Tarreau86a446e2013-11-25 23:02:37 +01003025
Frédéric Lécaille95203f22019-01-23 19:38:11 +01003026 msg_end += msg_len;
Frédéric Lécaille444243c2019-01-24 15:40:11 +01003027 if (!peer_treat_awaited_msg(appctx, curpeer, msg_head, &msg_cur, msg_end, msg_len, totl))
Emeric Brun2b920a12010-09-23 18:30:22 +02003028 goto switchstate;
Frédéric Lécaillebe825e52019-01-24 18:28:44 +01003029
Frédéric Lécaille045e0d42019-03-21 11:12:32 +01003030 curpeer->flags |= PEER_F_ALIVE;
3031
Emeric Brun2b920a12010-09-23 18:30:22 +02003032 /* skip consumed message */
Christopher Faulet908628c2022-03-25 16:43:49 +01003033 co_skip(cs_oc(cs), totl);
Emeric Brun2b920a12010-09-23 18:30:22 +02003034 /* loop on that state to peek next message */
Willy Tarreau72d6c162013-04-11 16:14:13 +02003035 goto switchstate;
3036
Frédéric Lécaillebe825e52019-01-24 18:28:44 +01003037send_msgs:
Frédéric Lécaille645635d2019-02-11 17:49:39 +01003038 if (curpeer->flags & PEER_F_HEARTBEAT) {
3039 curpeer->flags &= ~PEER_F_HEARTBEAT;
Frédéric Lécailleda2b0842021-01-15 16:21:28 +01003040 repl = peer_send_heartbeatmsg(appctx, curpeer, curpeers);
Frédéric Lécaille645635d2019-02-11 17:49:39 +01003041 if (repl <= 0) {
3042 if (repl == -1)
3043 goto out;
3044 goto switchstate;
3045 }
Frédéric Lécaille33cab3c2019-11-06 11:51:26 +01003046 curpeer->tx_hbt++;
Frédéric Lécaille645635d2019-02-11 17:49:39 +01003047 }
Frédéric Lécaillebe825e52019-01-24 18:28:44 +01003048 /* we get here when a peer_recv_msg() returns 0 in reql */
Frédéric Lécailleda2b0842021-01-15 16:21:28 +01003049 repl = peer_send_msgs(appctx, curpeer, curpeers);
Frédéric Lécaille25e1d5e2019-01-24 17:33:48 +01003050 if (repl <= 0) {
3051 if (repl == -1)
3052 goto out;
3053 goto switchstate;
Emeric Brun597b26e2016-08-12 11:23:31 +02003054 }
3055
Emeric Brun2b920a12010-09-23 18:30:22 +02003056 /* noting more to do */
3057 goto out;
3058 }
Willy Tarreaue4d927a2013-12-01 12:47:35 +01003059 case PEER_SESS_ST_EXIT:
Willy Tarreau2d372c22018-11-05 17:12:27 +01003060 if (prev_state == PEER_SESS_ST_WAITMSG)
Willy Tarreau4781b152021-04-06 13:53:36 +02003061 _HA_ATOMIC_DEC(&connected_peers);
Willy Tarreau2d372c22018-11-05 17:12:27 +01003062 prev_state = appctx->st0;
Frédéric Lécaille7d0ceee2019-01-24 14:24:05 +01003063 if (peer_send_status_errormsg(appctx) == -1)
3064 goto out;
Willy Tarreaue4d927a2013-12-01 12:47:35 +01003065 appctx->st0 = PEER_SESS_ST_END;
Emeric Brunb3971ab2015-05-12 18:49:09 +02003066 goto switchstate;
3067 case PEER_SESS_ST_ERRSIZE: {
Willy Tarreau2d372c22018-11-05 17:12:27 +01003068 if (prev_state == PEER_SESS_ST_WAITMSG)
Willy Tarreau4781b152021-04-06 13:53:36 +02003069 _HA_ATOMIC_DEC(&connected_peers);
Willy Tarreau2d372c22018-11-05 17:12:27 +01003070 prev_state = appctx->st0;
Frédéric Lécaille7d0ceee2019-01-24 14:24:05 +01003071 if (peer_send_error_size_limitmsg(appctx) == -1)
3072 goto out;
Emeric Brunb3971ab2015-05-12 18:49:09 +02003073 appctx->st0 = PEER_SESS_ST_END;
3074 goto switchstate;
3075 }
3076 case PEER_SESS_ST_ERRPROTO: {
Frédéric Lécailleda2b0842021-01-15 16:21:28 +01003077 TRACE_PROTO("protocol error", PEERS_EV_PROTOERR,
3078 NULL, curpeer, &prev_state);
Frédéric Lécailleec1c10b2019-11-07 15:22:33 +01003079 if (curpeer)
3080 curpeer->proto_err++;
Willy Tarreau2d372c22018-11-05 17:12:27 +01003081 if (prev_state == PEER_SESS_ST_WAITMSG)
Willy Tarreau4781b152021-04-06 13:53:36 +02003082 _HA_ATOMIC_DEC(&connected_peers);
Willy Tarreau2d372c22018-11-05 17:12:27 +01003083 prev_state = appctx->st0;
Frédéric Lécailleda2b0842021-01-15 16:21:28 +01003084 if (peer_send_error_protomsg(appctx) == -1) {
3085 TRACE_PROTO("could not send error message", PEERS_EV_PROTOERR);
Frédéric Lécaille7d0ceee2019-01-24 14:24:05 +01003086 goto out;
Frédéric Lécailleda2b0842021-01-15 16:21:28 +01003087 }
Emeric Brunb3971ab2015-05-12 18:49:09 +02003088 appctx->st0 = PEER_SESS_ST_END;
Willy Tarreau2d372c22018-11-05 17:12:27 +01003089 prev_state = appctx->st0;
Emeric Brunb3971ab2015-05-12 18:49:09 +02003090 }
Tim Duesterhus588b3142020-05-29 14:35:51 +02003091 /* fall through */
Willy Tarreaue4d927a2013-12-01 12:47:35 +01003092 case PEER_SESS_ST_END: {
Willy Tarreau2d372c22018-11-05 17:12:27 +01003093 if (prev_state == PEER_SESS_ST_WAITMSG)
Willy Tarreau4781b152021-04-06 13:53:36 +02003094 _HA_ATOMIC_DEC(&connected_peers);
Willy Tarreau2d372c22018-11-05 17:12:27 +01003095 prev_state = appctx->st0;
Emeric Brun80527f52017-06-19 17:46:37 +02003096 if (curpeer) {
Christopher Faulet2a944ee2017-11-07 10:42:54 +01003097 HA_SPIN_UNLOCK(PEER_LOCK, &curpeer->lock);
Emeric Brun80527f52017-06-19 17:46:37 +02003098 curpeer = NULL;
3099 }
Christopher Fauletda098e62022-03-31 17:44:45 +02003100 cs_shutw(cs);
3101 cs_shutr(cs);
Christopher Faulet908628c2022-03-25 16:43:49 +01003102 cs_ic(cs)->flags |= CF_READ_NULL;
Willy Tarreau828824a2015-04-19 17:20:03 +02003103 goto out;
Emeric Brun2b920a12010-09-23 18:30:22 +02003104 }
3105 }
3106 }
3107out:
Christopher Faulet908628c2022-03-25 16:43:49 +01003108 cs_oc(cs)->flags |= CF_READ_DONTWAIT;
Emeric Brun80527f52017-06-19 17:46:37 +02003109
3110 if (curpeer)
Christopher Faulet2a944ee2017-11-07 10:42:54 +01003111 HA_SPIN_UNLOCK(PEER_LOCK, &curpeer->lock);
Emeric Brun2b920a12010-09-23 18:30:22 +02003112 return;
3113}
3114
Willy Tarreau30576452015-04-13 13:50:30 +02003115static struct applet peer_applet = {
Willy Tarreau3fdb3662012-11-12 00:42:33 +01003116 .obj_type = OBJ_TYPE_APPLET,
Willy Tarreaub24281b2011-02-13 13:16:36 +01003117 .name = "<PEER>", /* used for logging */
3118 .fct = peer_io_handler,
Aman Gupta9a13e842012-04-02 18:57:53 -07003119 .release = peer_session_release,
Willy Tarreaub24281b2011-02-13 13:16:36 +01003120};
Emeric Brun2b920a12010-09-23 18:30:22 +02003121
Emeric Brun9ef2ad72019-04-02 17:22:01 +02003122
Emeric Brun2b920a12010-09-23 18:30:22 +02003123/*
3124 * Use this function to force a close of a peer session
3125 */
Emeric Brun9ef2ad72019-04-02 17:22:01 +02003126static void peer_session_forceshutdown(struct peer *peer)
Emeric Brun2b920a12010-09-23 18:30:22 +02003127{
Emeric Brun9ef2ad72019-04-02 17:22:01 +02003128 struct appctx *appctx = peer->appctx;
3129
Frédéric Lécaille5df11902017-06-13 16:39:57 +02003130 /* Note that the peer sessions which have just been created
3131 * (->st0 == PEER_SESS_ST_CONNECT) must not
3132 * be shutdown, if not, the TCP session will never be closed
3133 * and stay in CLOSE_WAIT state after having been closed by
3134 * the remote side.
3135 */
3136 if (!appctx || appctx->st0 == PEER_SESS_ST_CONNECT)
Willy Tarreau7b4b4992013-12-01 09:15:12 +01003137 return;
3138
Willy Tarreau81bc3b02016-10-31 17:37:39 +01003139 if (appctx->applet != &peer_applet)
3140 return;
3141
Emeric Brun9ef2ad72019-04-02 17:22:01 +02003142 __peer_session_deinit(peer);
3143
Willy Tarreaue4d927a2013-12-01 12:47:35 +01003144 appctx->st0 = PEER_SESS_ST_END;
Willy Tarreau78c0c502016-10-31 17:32:20 +01003145 appctx_wakeup(appctx);
Emeric Brun2b920a12010-09-23 18:30:22 +02003146}
3147
Willy Tarreau91d96282015-03-13 15:47:26 +01003148/* Pre-configures a peers frontend to accept incoming connections */
3149void peers_setup_frontend(struct proxy *fe)
3150{
3151 fe->last_change = now.tv_sec;
Frédéric Lécaillec06b5d42018-04-26 10:06:41 +02003152 fe->cap = PR_CAP_FE | PR_CAP_BE;
Willy Tarreaua389c9e2020-10-07 17:49:42 +02003153 fe->mode = PR_MODE_PEERS;
Willy Tarreau91d96282015-03-13 15:47:26 +01003154 fe->maxconn = 0;
3155 fe->conn_retries = CONN_RETRIES;
3156 fe->timeout.client = MS_TO_TICKS(5000);
Willy Tarreaud1d48d42015-03-13 16:15:46 +01003157 fe->accept = frontend_accept;
Willy Tarreauf87ab942015-03-13 15:55:16 +01003158 fe->default_target = &peer_applet.obj_type;
Willy Tarreau91d96282015-03-13 15:47:26 +01003159 fe->options2 |= PR_O2_INDEPSTR | PR_O2_SMARTCON | PR_O2_SMARTACC;
3160}
3161
Emeric Brun2b920a12010-09-23 18:30:22 +02003162/*
Willy Tarreaubd55e312010-11-11 10:55:09 +01003163 * Create a new peer session in assigned state (connect will start automatically)
Emeric Brun2b920a12010-09-23 18:30:22 +02003164 */
Willy Tarreau9df94c22016-10-31 18:42:52 +01003165static struct appctx *peer_session_create(struct peers *peers, struct peer *peer)
Emeric Brun2b920a12010-09-23 18:30:22 +02003166{
Willy Tarreau04b92862017-09-15 11:01:04 +02003167 struct proxy *p = peers->peers_fe; /* attached frontend */
Willy Tarreau7b4b4992013-12-01 09:15:12 +01003168 struct appctx *appctx;
Willy Tarreau15b5e142015-04-04 14:38:25 +02003169 struct session *sess;
Christopher Faulet13a35e52021-12-20 15:34:16 +01003170 struct conn_stream *cs;
Willy Tarreau87b09662015-04-03 00:22:06 +02003171 struct stream *s;
Christopher Fauleta9e8b392022-03-23 11:01:09 +01003172 struct sockaddr_storage *addr = NULL;
Emeric Brun2b920a12010-09-23 18:30:22 +02003173
Frédéric Lécaille2b0ba542021-01-18 15:14:39 +01003174 peer->new_conn++;
Frédéric Lécaille54bff832019-03-26 10:25:20 +01003175 peer->reconnect = tick_add(now_ms, MS_TO_TICKS(PEER_RECONNECT_TIMEOUT));
Frédéric Lécaillebaeb9192020-10-14 11:50:26 +02003176 peer->heartbeat = TICK_ETERNITY;
Emeric Brunb3971ab2015-05-12 18:49:09 +02003177 peer->statuscode = PEER_SESS_SC_CONNECTCODE;
Frédéric Lécaille3fc0fe02020-10-08 09:46:24 +02003178 peer->last_hdshk = now_ms;
Willy Tarreaud990baf2015-04-05 00:32:03 +02003179 s = NULL;
3180
Christopher Faulet9ec2f4d2022-03-23 15:15:29 +01003181 appctx = appctx_new(&peer_applet, NULL);
Christopher Faulet2479e5f2022-01-19 14:50:11 +01003182 if (!appctx)
Christopher Fauleta9e8b392022-03-23 11:01:09 +01003183 goto out_close;
Willy Tarreaud990baf2015-04-05 00:32:03 +02003184
3185 appctx->st0 = PEER_SESS_ST_CONNECT;
Emeric Brunb3971ab2015-05-12 18:49:09 +02003186 appctx->ctx.peers.ptr = (void *)peer;
Willy Tarreaud990baf2015-04-05 00:32:03 +02003187
Willy Tarreau04b92862017-09-15 11:01:04 +02003188 sess = session_new(p, NULL, &appctx->obj_type);
Willy Tarreau15b5e142015-04-04 14:38:25 +02003189 if (!sess) {
Christopher Faulet767a84b2017-11-24 16:50:31 +01003190 ha_alert("out of memory in peer_session_create().\n");
Willy Tarreaud990baf2015-04-05 00:32:03 +02003191 goto out_free_appctx;
Emeric Brun2b920a12010-09-23 18:30:22 +02003192 }
3193
Christopher Fauleta9e8b392022-03-23 11:01:09 +01003194 if (!sockaddr_alloc(&addr, &peer->addr, sizeof(peer->addr)))
Christopher Faulet2479e5f2022-01-19 14:50:11 +01003195 goto out_free_sess;
Christopher Fauleta9e8b392022-03-23 11:01:09 +01003196
Christopher Faulet9ec2f4d2022-03-23 15:15:29 +01003197 cs = cs_new_from_applet(appctx->endp, sess, &BUF_NULL);
Christopher Fauleta9e8b392022-03-23 11:01:09 +01003198 if (!cs) {
3199 ha_alert("Failed to initialize stream in peer_session_create().\n");
Christopher Fauleta9e8b392022-03-23 11:01:09 +01003200 goto out_free_addr;
Christopher Faulet13a35e52021-12-20 15:34:16 +01003201 }
3202
Christopher Fauleta9e8b392022-03-23 11:01:09 +01003203 s = DISGUISE(cs_strm(cs));
3204
Willy Tarreau6e2979c2015-04-27 13:21:15 +02003205 /* applet is waiting for data */
Christopher Fauleta0bdec32022-04-04 07:51:21 +02003206 cs_cant_get(s->csf);
Willy Tarreau6e2979c2015-04-27 13:21:15 +02003207 appctx_wakeup(appctx);
3208
Willy Tarreau3ed35ef2013-10-24 11:51:38 +02003209 /* initiate an outgoing connection */
Christopher Faulet8da67aa2022-03-29 17:53:09 +02003210 s->csb->dst = addr;
Christopher Faulet8abe7122022-03-30 15:10:18 +02003211 s->csb->flags |= CS_FL_NOLINGER;
Christopher Fauleta9e8b392022-03-23 11:01:09 +01003212 s->flags = SF_ASSIGNED|SF_ADDR_SET;
3213 s->target = peer_session_target(peer, s);
Willy Tarreau32e3c6a2013-10-11 19:34:20 +02003214
Emeric Brun2b920a12010-09-23 18:30:22 +02003215 s->do_log = NULL;
Emeric Brun2b920a12010-09-23 18:30:22 +02003216 s->uniq_id = 0;
Emeric Brun2b920a12010-09-23 18:30:22 +02003217
Willy Tarreau22ec1ea2014-11-27 20:45:39 +01003218 s->res.flags |= CF_READ_DONTWAIT;
Willy Tarreau696a2912014-11-24 11:36:57 +01003219
Emeric Brunb3971ab2015-05-12 18:49:09 +02003220 peer->appctx = appctx;
Willy Tarreau4781b152021-04-06 13:53:36 +02003221 _HA_ATOMIC_INC(&active_peers);
Willy Tarreau9df94c22016-10-31 18:42:52 +01003222 return appctx;
Emeric Brun2b920a12010-09-23 18:30:22 +02003223
3224 /* Error unrolling */
Christopher Fauleta9e8b392022-03-23 11:01:09 +01003225 out_free_addr:
3226 sockaddr_free(&addr);
Willy Tarreau15b5e142015-04-04 14:38:25 +02003227 out_free_sess:
Willy Tarreau11c36242015-04-04 15:54:03 +02003228 session_free(sess);
Willy Tarreaud990baf2015-04-05 00:32:03 +02003229 out_free_appctx:
3230 appctx_free(appctx);
Emeric Brun2b920a12010-09-23 18:30:22 +02003231 out_close:
Willy Tarreaub21d08e2016-10-31 17:46:57 +01003232 return NULL;
Emeric Brun2b920a12010-09-23 18:30:22 +02003233}
3234
3235/*
Frédéric Lécaille645635d2019-02-11 17:49:39 +01003236 * Task processing function to manage re-connect, peer session
Willy Tarreauf6c88422021-01-29 12:38:42 +01003237 * tasks wakeup on local update and heartbeat. Let's keep it exported so that it
3238 * resolves in stack traces and "show tasks".
Emeric Brun2b920a12010-09-23 18:30:22 +02003239 */
Willy Tarreau144f84a2021-03-02 16:09:26 +01003240struct task *process_peer_sync(struct task * task, void *context, unsigned int state)
Emeric Brun2b920a12010-09-23 18:30:22 +02003241{
Olivier Houchard9f6af332018-05-25 14:04:04 +02003242 struct peers *peers = context;
Emeric Brunb3971ab2015-05-12 18:49:09 +02003243 struct peer *ps;
3244 struct shared_table *st;
Emeric Brun2b920a12010-09-23 18:30:22 +02003245
3246 task->expire = TICK_ETERNITY;
3247
Emeric Brunb3971ab2015-05-12 18:49:09 +02003248 if (!peers->peers_fe) {
Willy Tarreau46dc1ca2015-05-01 18:32:13 +02003249 /* this one was never started, kill it */
Emeric Brunb3971ab2015-05-12 18:49:09 +02003250 signal_unregister_handler(peers->sighandler);
Olivier Houchard3f795f72019-04-17 22:51:06 +02003251 task_destroy(peers->sync_task);
Willy Tarreau37bb7be2015-09-21 15:24:58 +02003252 peers->sync_task = NULL;
Willy Tarreau46dc1ca2015-05-01 18:32:13 +02003253 return NULL;
3254 }
3255
Emeric Brun80527f52017-06-19 17:46:37 +02003256 /* Acquire lock for all peers of the section */
3257 for (ps = peers->remote; ps; ps = ps->next)
Christopher Faulet2a944ee2017-11-07 10:42:54 +01003258 HA_SPIN_LOCK(PEER_LOCK, &ps->lock);
Emeric Brun80527f52017-06-19 17:46:37 +02003259
Emeric Brun2b920a12010-09-23 18:30:22 +02003260 if (!stopping) {
3261 /* Normal case (not soft stop)*/
Emeric Brunb3971ab2015-05-12 18:49:09 +02003262
Emeric Brun2c4ab412021-04-21 16:06:35 +02003263 /* resync timeout set to TICK_ETERNITY means we just start
3264 * a new process and timer was not initialized.
3265 * We must arm this timer to switch to a request to a remote
3266 * node if incoming connection from old local process never
3267 * comes.
3268 */
3269 if (peers->resync_timeout == TICK_ETERNITY)
3270 peers->resync_timeout = tick_add(now_ms, MS_TO_TICKS(PEER_RESYNC_TIMEOUT));
3271
Emeric Brunb3971ab2015-05-12 18:49:09 +02003272 if (((peers->flags & PEERS_RESYNC_STATEMASK) == PEERS_RESYNC_FROMLOCAL) &&
3273 (!nb_oldpids || tick_is_expired(peers->resync_timeout, now_ms)) &&
3274 !(peers->flags & PEERS_F_RESYNC_ASSIGN)) {
Emeric Brun2b920a12010-09-23 18:30:22 +02003275 /* Resync from local peer needed
3276 no peer was assigned for the lesson
3277 and no old local peer found
3278 or resync timeout expire */
3279
3280 /* flag no more resync from local, to try resync from remotes */
Emeric Brunb3971ab2015-05-12 18:49:09 +02003281 peers->flags |= PEERS_F_RESYNC_LOCAL;
Emeric Brunccdfbae2021-04-28 12:59:35 +02003282 peers->flags |= PEERS_F_RESYNC_LOCALTIMEOUT;
Emeric Brun2b920a12010-09-23 18:30:22 +02003283
3284 /* reschedule a resync */
Frédéric Lécaille54bff832019-03-26 10:25:20 +01003285 peers->resync_timeout = tick_add(now_ms, MS_TO_TICKS(PEER_RESYNC_TIMEOUT));
Emeric Brun2b920a12010-09-23 18:30:22 +02003286 }
3287
3288 /* For each session */
Emeric Brunb3971ab2015-05-12 18:49:09 +02003289 for (ps = peers->remote; ps; ps = ps->next) {
Emeric Brun2b920a12010-09-23 18:30:22 +02003290 /* For each remote peers */
Emeric Brunb3971ab2015-05-12 18:49:09 +02003291 if (!ps->local) {
Willy Tarreau9df94c22016-10-31 18:42:52 +01003292 if (!ps->appctx) {
3293 /* no active peer connection */
Emeric Brun2b920a12010-09-23 18:30:22 +02003294 if (ps->statuscode == 0 ||
Willy Tarreaue4d927a2013-12-01 12:47:35 +01003295 ((ps->statuscode == PEER_SESS_SC_CONNECTCODE ||
Willy Tarreaub4e34da2015-05-20 10:39:04 +02003296 ps->statuscode == PEER_SESS_SC_SUCCESSCODE ||
Willy Tarreaue4d927a2013-12-01 12:47:35 +01003297 ps->statuscode == PEER_SESS_SC_CONNECTEDCODE) &&
Emeric Brun2b920a12010-09-23 18:30:22 +02003298 tick_is_expired(ps->reconnect, now_ms))) {
3299 /* connection never tried
Willy Tarreau9df94c22016-10-31 18:42:52 +01003300 * or previous peer connection established with success
3301 * or previous peer connection failed while connecting
Emeric Brun2b920a12010-09-23 18:30:22 +02003302 * and reconnection timer is expired */
3303
3304 /* retry a connect */
Willy Tarreau9df94c22016-10-31 18:42:52 +01003305 ps->appctx = peer_session_create(peers, ps);
Emeric Brun2b920a12010-09-23 18:30:22 +02003306 }
Willy Tarreaub4e34da2015-05-20 10:39:04 +02003307 else if (!tick_is_expired(ps->reconnect, now_ms)) {
Emeric Brun2b920a12010-09-23 18:30:22 +02003308 /* If previous session failed during connection
3309 * but reconnection timer is not expired */
3310
3311 /* reschedule task for reconnect */
3312 task->expire = tick_first(task->expire, ps->reconnect);
3313 }
3314 /* else do nothing */
Willy Tarreau9df94c22016-10-31 18:42:52 +01003315 } /* !ps->appctx */
Willy Tarreaue4d927a2013-12-01 12:47:35 +01003316 else if (ps->statuscode == PEER_SESS_SC_SUCCESSCODE) {
Willy Tarreau9df94c22016-10-31 18:42:52 +01003317 /* current peer connection is active and established */
Emeric Brunb3971ab2015-05-12 18:49:09 +02003318 if (((peers->flags & PEERS_RESYNC_STATEMASK) == PEERS_RESYNC_FROMREMOTE) &&
3319 !(peers->flags & PEERS_F_RESYNC_ASSIGN) &&
Emeric Brun2b920a12010-09-23 18:30:22 +02003320 !(ps->flags & PEER_F_LEARN_NOTUP2DATE)) {
3321 /* Resync from a remote is needed
3322 * and no peer was assigned for lesson
3323 * and current peer may be up2date */
3324
3325 /* assign peer for the lesson */
3326 ps->flags |= PEER_F_LEARN_ASSIGN;
Emeric Brunb3971ab2015-05-12 18:49:09 +02003327 peers->flags |= PEERS_F_RESYNC_ASSIGN;
Emeric Brunccdfbae2021-04-28 12:59:35 +02003328 peers->flags |= PEERS_F_RESYNC_REMOTEASSIGN;
Emeric Brun2b920a12010-09-23 18:30:22 +02003329
Willy Tarreau9df94c22016-10-31 18:42:52 +01003330 /* wake up peer handler to handle a request of resync */
Willy Tarreaue5843b32015-04-27 18:40:14 +02003331 appctx_wakeup(ps->appctx);
Emeric Brun2b920a12010-09-23 18:30:22 +02003332 }
Emeric Brunb3971ab2015-05-12 18:49:09 +02003333 else {
Frédéric Lécaille645635d2019-02-11 17:49:39 +01003334 int update_to_push = 0;
3335
Emeric Brunb3971ab2015-05-12 18:49:09 +02003336 /* Awake session if there is data to push */
3337 for (st = ps->tables; st ; st = st->next) {
Emeric Brun8e7a13e2021-04-28 11:48:15 +02003338 if (st->last_pushed != st->table->localupdate) {
Willy Tarreau9df94c22016-10-31 18:42:52 +01003339 /* wake up the peer handler to push local updates */
Frédéric Lécaille645635d2019-02-11 17:49:39 +01003340 update_to_push = 1;
Frédéric Lécaille045e0d42019-03-21 11:12:32 +01003341 /* There is no need to send a heartbeat message
3342 * when some updates must be pushed. The remote
3343 * peer will consider <ps> peer as alive when it will
3344 * receive these updates.
3345 */
Frédéric Lécaille645635d2019-02-11 17:49:39 +01003346 ps->flags &= ~PEER_F_HEARTBEAT;
Frédéric Lécaille045e0d42019-03-21 11:12:32 +01003347 /* Re-schedule another one later. */
Frédéric Lécaille645635d2019-02-11 17:49:39 +01003348 ps->heartbeat = tick_add(now_ms, MS_TO_TICKS(PEER_HEARTBEAT_TIMEOUT));
Frédéric Lécaille045e0d42019-03-21 11:12:32 +01003349 /* We are going to send updates, let's ensure we will
3350 * come back to send heartbeat messages or to reconnect.
3351 */
3352 task->expire = tick_first(ps->reconnect, ps->heartbeat);
Emeric Brunb3971ab2015-05-12 18:49:09 +02003353 appctx_wakeup(ps->appctx);
3354 break;
3355 }
3356 }
Frédéric Lécaille045e0d42019-03-21 11:12:32 +01003357 /* When there are updates to send we do not reconnect
3358 * and do not send heartbeat message either.
3359 */
3360 if (!update_to_push) {
3361 if (tick_is_expired(ps->reconnect, now_ms)) {
3362 if (ps->flags & PEER_F_ALIVE) {
3363 /* This peer was alive during a 'reconnect' period.
3364 * Flag it as not alive again for the next period.
3365 */
3366 ps->flags &= ~PEER_F_ALIVE;
Frédéric Lécaille54bff832019-03-26 10:25:20 +01003367 ps->reconnect = tick_add(now_ms, MS_TO_TICKS(PEER_RECONNECT_TIMEOUT));
Frédéric Lécaille045e0d42019-03-21 11:12:32 +01003368 }
3369 else {
Willy Tarreau52bf8392020-03-08 00:42:37 +01003370 ps->reconnect = tick_add(now_ms, MS_TO_TICKS(50 + ha_random() % 2000));
Frédéric Lécaillebaeb9192020-10-14 11:50:26 +02003371 ps->heartbeat = TICK_ETERNITY;
Emeric Brun9ef2ad72019-04-02 17:22:01 +02003372 peer_session_forceshutdown(ps);
Frédéric Lécailleec1c10b2019-11-07 15:22:33 +01003373 ps->no_hbt++;
Frédéric Lécaille045e0d42019-03-21 11:12:32 +01003374 }
3375 }
3376 else if (tick_is_expired(ps->heartbeat, now_ms)) {
3377 ps->heartbeat = tick_add(now_ms, MS_TO_TICKS(PEER_HEARTBEAT_TIMEOUT));
3378 ps->flags |= PEER_F_HEARTBEAT;
3379 appctx_wakeup(ps->appctx);
3380 }
Frédéric Lécailleb7405c12019-03-27 14:32:39 +01003381 task->expire = tick_first(ps->reconnect, ps->heartbeat);
Frédéric Lécaille645635d2019-02-11 17:49:39 +01003382 }
Emeric Brun2b920a12010-09-23 18:30:22 +02003383 }
3384 /* else do nothing */
3385 } /* SUCCESSCODE */
3386 } /* !ps->peer->local */
3387 } /* for */
3388
3389 /* Resync from remotes expired: consider resync is finished */
Emeric Brunb3971ab2015-05-12 18:49:09 +02003390 if (((peers->flags & PEERS_RESYNC_STATEMASK) == PEERS_RESYNC_FROMREMOTE) &&
3391 !(peers->flags & PEERS_F_RESYNC_ASSIGN) &&
3392 tick_is_expired(peers->resync_timeout, now_ms)) {
Emeric Brun2b920a12010-09-23 18:30:22 +02003393 /* Resync from remote peer needed
3394 * no peer was assigned for the lesson
3395 * and resync timeout expire */
3396
3397 /* flag no more resync from remote, consider resync is finished */
Emeric Brunb3971ab2015-05-12 18:49:09 +02003398 peers->flags |= PEERS_F_RESYNC_REMOTE;
Emeric Brunccdfbae2021-04-28 12:59:35 +02003399 peers->flags |= PEERS_F_RESYNC_REMOTETIMEOUT;
Emeric Brun2b920a12010-09-23 18:30:22 +02003400 }
3401
Emeric Brunb3971ab2015-05-12 18:49:09 +02003402 if ((peers->flags & PEERS_RESYNC_STATEMASK) != PEERS_RESYNC_FINISHED) {
Emeric Brun2b920a12010-09-23 18:30:22 +02003403 /* Resync not finished*/
Frédéric Lécaille5d6e5f82017-05-29 13:47:16 +02003404 /* reschedule task to resync timeout if not expired, to ended resync if needed */
3405 if (!tick_is_expired(peers->resync_timeout, now_ms))
3406 task->expire = tick_first(task->expire, peers->resync_timeout);
Emeric Brun2b920a12010-09-23 18:30:22 +02003407 }
3408 } /* !stopping */
3409 else {
3410 /* soft stop case */
Willy Tarreau086735a2018-11-05 15:09:47 +01003411 if (state & TASK_WOKEN_SIGNAL) {
Joseph Herlant82b2f542018-11-15 12:19:14 -08003412 /* We've just received the signal */
Emeric Brunb3971ab2015-05-12 18:49:09 +02003413 if (!(peers->flags & PEERS_F_DONOTSTOP)) {
Emeric Brun2b920a12010-09-23 18:30:22 +02003414 /* add DO NOT STOP flag if not present */
Willy Tarreau4781b152021-04-06 13:53:36 +02003415 _HA_ATOMIC_INC(&jobs);
Emeric Brunb3971ab2015-05-12 18:49:09 +02003416 peers->flags |= PEERS_F_DONOTSTOP;
Emeric Brun2b920a12010-09-23 18:30:22 +02003417
Emeric Bruncbfe5eb2021-04-22 18:20:37 +02003418 /* disconnect all connected peers to process a local sync
3419 * this must be done only the first time we are switching
3420 * in stopping state
Emeric Brun80527f52017-06-19 17:46:37 +02003421 */
Emeric Bruncbfe5eb2021-04-22 18:20:37 +02003422 for (ps = peers->remote; ps; ps = ps->next) {
3423 /* we're killing a connection, we must apply a random delay before
3424 * retrying otherwise the other end will do the same and we can loop
3425 * for a while.
3426 */
3427 ps->reconnect = tick_add(now_ms, MS_TO_TICKS(50 + ha_random() % 2000));
3428 if (ps->appctx) {
3429 peer_session_forceshutdown(ps);
3430 }
Emeric Brun2b920a12010-09-23 18:30:22 +02003431 }
3432 }
3433 }
Emeric Brun2b920a12010-09-23 18:30:22 +02003434
Emeric Brunb3971ab2015-05-12 18:49:09 +02003435 ps = peers->local;
Emeric Brun2b920a12010-09-23 18:30:22 +02003436 if (ps->flags & PEER_F_TEACH_COMPLETE) {
Emeric Brunb3971ab2015-05-12 18:49:09 +02003437 if (peers->flags & PEERS_F_DONOTSTOP) {
Emeric Brun2b920a12010-09-23 18:30:22 +02003438 /* resync of new process was complete, current process can die now */
Willy Tarreau4781b152021-04-06 13:53:36 +02003439 _HA_ATOMIC_DEC(&jobs);
Emeric Brunb3971ab2015-05-12 18:49:09 +02003440 peers->flags &= ~PEERS_F_DONOTSTOP;
3441 for (st = ps->tables; st ; st = st->next)
Emeric Brun2cc201f2021-04-23 12:21:26 +02003442 HA_ATOMIC_DEC(&st->table->refcnt);
Emeric Brun2b920a12010-09-23 18:30:22 +02003443 }
3444 }
Willy Tarreau9df94c22016-10-31 18:42:52 +01003445 else if (!ps->appctx) {
3446 /* If there's no active peer connection */
Emeric Brun2b920a12010-09-23 18:30:22 +02003447 if (ps->statuscode == 0 ||
Willy Tarreaue4d927a2013-12-01 12:47:35 +01003448 ps->statuscode == PEER_SESS_SC_SUCCESSCODE ||
3449 ps->statuscode == PEER_SESS_SC_CONNECTEDCODE ||
3450 ps->statuscode == PEER_SESS_SC_TRYAGAIN) {
Emeric Brun2b920a12010-09-23 18:30:22 +02003451 /* connection never tried
Willy Tarreau9df94c22016-10-31 18:42:52 +01003452 * or previous peer connection was successfully established
3453 * or previous tcp connect succeeded but init state incomplete
Emeric Brun2b920a12010-09-23 18:30:22 +02003454 * or during previous connect, peer replies a try again statuscode */
3455
Emeric Bruncbfe5eb2021-04-22 18:20:37 +02003456 /* connect to the local peer if we must push a local sync */
3457 if (peers->flags & PEERS_F_DONOTSTOP) {
3458 peer_session_create(peers, ps);
3459 }
Emeric Brun2b920a12010-09-23 18:30:22 +02003460 }
3461 else {
3462 /* Other error cases */
Emeric Brunb3971ab2015-05-12 18:49:09 +02003463 if (peers->flags & PEERS_F_DONOTSTOP) {
Emeric Brun2b920a12010-09-23 18:30:22 +02003464 /* unable to resync new process, current process can die now */
Willy Tarreau4781b152021-04-06 13:53:36 +02003465 _HA_ATOMIC_DEC(&jobs);
Emeric Brunb3971ab2015-05-12 18:49:09 +02003466 peers->flags &= ~PEERS_F_DONOTSTOP;
3467 for (st = ps->tables; st ; st = st->next)
Emeric Brun2cc201f2021-04-23 12:21:26 +02003468 HA_ATOMIC_DEC(&st->table->refcnt);
Emeric Brun2b920a12010-09-23 18:30:22 +02003469 }
3470 }
3471 }
Emeric Brunb3971ab2015-05-12 18:49:09 +02003472 else if (ps->statuscode == PEER_SESS_SC_SUCCESSCODE ) {
Willy Tarreau9df94c22016-10-31 18:42:52 +01003473 /* current peer connection is active and established
3474 * wake up all peer handlers to push remaining local updates */
Emeric Brunb3971ab2015-05-12 18:49:09 +02003475 for (st = ps->tables; st ; st = st->next) {
Emeric Brun8e7a13e2021-04-28 11:48:15 +02003476 if (st->last_pushed != st->table->localupdate) {
Emeric Brunb3971ab2015-05-12 18:49:09 +02003477 appctx_wakeup(ps->appctx);
3478 break;
3479 }
3480 }
Emeric Brun2b920a12010-09-23 18:30:22 +02003481 }
3482 } /* stopping */
Emeric Brun80527f52017-06-19 17:46:37 +02003483
3484 /* Release lock for all peers of the section */
3485 for (ps = peers->remote; ps; ps = ps->next)
Christopher Faulet2a944ee2017-11-07 10:42:54 +01003486 HA_SPIN_UNLOCK(PEER_LOCK, &ps->lock);
Emeric Brun80527f52017-06-19 17:46:37 +02003487
Emeric Brun2b920a12010-09-23 18:30:22 +02003488 /* Wakeup for re-connect */
3489 return task;
3490}
3491
Emeric Brunb3971ab2015-05-12 18:49:09 +02003492
Emeric Brun2b920a12010-09-23 18:30:22 +02003493/*
Willy Tarreaud9443442018-10-15 11:18:03 +02003494 * returns 0 in case of error.
Emeric Brun2b920a12010-09-23 18:30:22 +02003495 */
Willy Tarreaud9443442018-10-15 11:18:03 +02003496int peers_init_sync(struct peers *peers)
Emeric Brun2b920a12010-09-23 18:30:22 +02003497{
Emeric Brun2b920a12010-09-23 18:30:22 +02003498 struct peer * curpeer;
Emeric Brun2b920a12010-09-23 18:30:22 +02003499
Emeric Brun2b920a12010-09-23 18:30:22 +02003500 for (curpeer = peers->remote; curpeer; curpeer = curpeer->next) {
Emeric Brun2b920a12010-09-23 18:30:22 +02003501 peers->peers_fe->maxconn += 3;
3502 }
3503
Willy Tarreaubeeabf52021-10-01 18:23:30 +02003504 peers->sync_task = task_new_anywhere();
Willy Tarreaud9443442018-10-15 11:18:03 +02003505 if (!peers->sync_task)
3506 return 0;
3507
Emeric Brunb3971ab2015-05-12 18:49:09 +02003508 peers->sync_task->process = process_peer_sync;
Emeric Brunb3971ab2015-05-12 18:49:09 +02003509 peers->sync_task->context = (void *)peers;
3510 peers->sighandler = signal_register_task(0, peers->sync_task, 0);
3511 task_wakeup(peers->sync_task, TASK_WOKEN_INIT);
Willy Tarreaud9443442018-10-15 11:18:03 +02003512 return 1;
Emeric Brunb3971ab2015-05-12 18:49:09 +02003513}
3514
Frédéric Lécaille74167b22019-05-28 19:02:42 +02003515/*
3516 * Allocate a cache a dictionary entries used upon transmission.
3517 */
Frédéric Lécaille6c391982019-06-06 11:34:03 +02003518static struct dcache_tx *new_dcache_tx(size_t max_entries)
Frédéric Lécaille74167b22019-05-28 19:02:42 +02003519{
3520 struct dcache_tx *d;
Frédéric Lécaille6c391982019-06-06 11:34:03 +02003521 struct ebpt_node *entries;
Frédéric Lécaille74167b22019-05-28 19:02:42 +02003522
3523 d = malloc(sizeof *d);
Frédéric Lécaille6c391982019-06-06 11:34:03 +02003524 entries = calloc(max_entries, sizeof *entries);
3525 if (!d || !entries)
3526 goto err;
Frédéric Lécaille74167b22019-05-28 19:02:42 +02003527
3528 d->lru_key = 0;
Frédéric Lécailleb65717f2019-06-07 14:25:25 +02003529 d->prev_lookup = NULL;
Frédéric Lécaille6c391982019-06-06 11:34:03 +02003530 d->cached_entries = EB_ROOT_UNIQUE;
3531 d->entries = entries;
Frédéric Lécaille74167b22019-05-28 19:02:42 +02003532
3533 return d;
Frédéric Lécaille6c391982019-06-06 11:34:03 +02003534
3535 err:
3536 free(d);
3537 free(entries);
3538 return NULL;
Frédéric Lécaille74167b22019-05-28 19:02:42 +02003539}
3540
3541/*
3542 * Allocate a cache of dictionary entries with <name> as name and <max_entries>
3543 * as maximum of entries.
Ilya Shipitsinc02a23f2020-05-06 00:53:22 +05003544 * Return the dictionary cache if succeeded, NULL if not.
Frédéric Lécaille74167b22019-05-28 19:02:42 +02003545 * Must be deallocated calling free_dcache().
3546 */
Frédéric Lécaille6c391982019-06-06 11:34:03 +02003547static struct dcache *new_dcache(size_t max_entries)
Frédéric Lécaille74167b22019-05-28 19:02:42 +02003548{
3549 struct dcache_tx *dc_tx;
3550 struct dcache *dc;
3551 struct dcache_rx *dc_rx;
3552
3553 dc = calloc(1, sizeof *dc);
3554 dc_tx = new_dcache_tx(max_entries);
3555 dc_rx = calloc(max_entries, sizeof *dc_rx);
3556 if (!dc || !dc_tx || !dc_rx)
3557 goto err;
3558
3559 dc->tx = dc_tx;
3560 dc->rx = dc_rx;
3561 dc->max_entries = max_entries;
3562
3563 return dc;
3564
3565 err:
3566 free(dc);
3567 free(dc_tx);
3568 free(dc_rx);
3569 return NULL;
3570}
3571
3572/*
Frédéric Lécaille74167b22019-05-28 19:02:42 +02003573 * Look for the dictionary entry with the value of <i> in <d> cache of dictionary
3574 * entries used upon transmission.
3575 * Return the entry if found, NULL if not.
3576 */
Frédéric Lécaille6c391982019-06-06 11:34:03 +02003577static struct ebpt_node *dcache_tx_lookup_value(struct dcache_tx *d,
3578 struct dcache_tx_entry *i)
Frédéric Lécaille74167b22019-05-28 19:02:42 +02003579{
Frédéric Lécaille6c391982019-06-06 11:34:03 +02003580 return ebpt_lookup(&d->cached_entries, i->entry.key);
Frédéric Lécaille74167b22019-05-28 19:02:42 +02003581}
3582
3583/*
Frédéric Lécaille74167b22019-05-28 19:02:42 +02003584 * Flush <dc> cache.
3585 * Always succeeds.
3586 */
3587static inline void flush_dcache(struct peer *peer)
3588{
3589 int i;
Frédéric Lécaille8d78fa72019-05-20 18:22:52 +02003590 struct dcache *dc = peer->dcache;
Frédéric Lécaille74167b22019-05-28 19:02:42 +02003591
Frédéric Lécailleea875e62020-11-12 21:01:54 +01003592 for (i = 0; i < dc->max_entries; i++) {
Frédéric Lécaille6c391982019-06-06 11:34:03 +02003593 ebpt_delete(&dc->tx->entries[i]);
Frédéric Lécailleea875e62020-11-12 21:01:54 +01003594 dc->tx->entries[i].key = NULL;
Thayne McCombs92149f92020-11-20 01:28:26 -07003595 dict_entry_unref(&server_key_dict, dc->rx[i].de);
3596 dc->rx[i].de = NULL;
Frédéric Lécailleea875e62020-11-12 21:01:54 +01003597 }
3598 dc->tx->prev_lookup = NULL;
3599 dc->tx->lru_key = 0;
Frédéric Lécaille74167b22019-05-28 19:02:42 +02003600
Frédéric Lécaille6c391982019-06-06 11:34:03 +02003601 memset(dc->rx, 0, dc->max_entries * sizeof *dc->rx);
Frédéric Lécaille74167b22019-05-28 19:02:42 +02003602}
3603
3604/*
3605 * Insert a dictionary entry in <dc> cache part used upon transmission (->tx)
3606 * with information provided by <i> dictionary cache entry (especially the value
3607 * to be inserted if not already). Return <i> if already present in the cache
Frédéric Lécaille6c391982019-06-06 11:34:03 +02003608 * or something different of <i> if not.
Frédéric Lécaille74167b22019-05-28 19:02:42 +02003609 */
Frédéric Lécaille6c391982019-06-06 11:34:03 +02003610static struct ebpt_node *dcache_tx_insert(struct dcache *dc, struct dcache_tx_entry *i)
Frédéric Lécaille74167b22019-05-28 19:02:42 +02003611{
3612 struct dcache_tx *dc_tx;
Frédéric Lécaille6c391982019-06-06 11:34:03 +02003613 struct ebpt_node *o;
Frédéric Lécaille74167b22019-05-28 19:02:42 +02003614
3615 dc_tx = dc->tx;
Frédéric Lécailleb65717f2019-06-07 14:25:25 +02003616
3617 if (dc_tx->prev_lookup && dc_tx->prev_lookup->key == i->entry.key) {
3618 o = dc_tx->prev_lookup;
3619 } else {
3620 o = dcache_tx_lookup_value(dc_tx, i);
3621 if (o) {
3622 /* Save it */
3623 dc_tx->prev_lookup = o;
3624 }
3625 }
3626
Frédéric Lécaille74167b22019-05-28 19:02:42 +02003627 if (o) {
Frédéric Lécaille6c391982019-06-06 11:34:03 +02003628 /* Copy the ID. */
3629 i->id = o - dc->tx->entries;
3630 return &i->entry;
Frédéric Lécaille74167b22019-05-28 19:02:42 +02003631 }
3632
Frédéric Lécaille6c391982019-06-06 11:34:03 +02003633 /* The new entry to put in cache */
Frédéric Lécailleb65717f2019-06-07 14:25:25 +02003634 dc_tx->prev_lookup = o = &dc_tx->entries[dc_tx->lru_key];
Frédéric Lécaille74167b22019-05-28 19:02:42 +02003635
Frédéric Lécaille6c391982019-06-06 11:34:03 +02003636 ebpt_delete(o);
3637 o->key = i->entry.key;
3638 ebpt_insert(&dc_tx->cached_entries, o);
3639 i->id = dc_tx->lru_key;
Frédéric Lécaille74167b22019-05-28 19:02:42 +02003640
Frédéric Lécaille6c391982019-06-06 11:34:03 +02003641 /* Update the index for the next entry to put in cache */
3642 dc_tx->lru_key = (dc_tx->lru_key + 1) & (dc->max_entries - 1);
Emeric Brunb3971ab2015-05-12 18:49:09 +02003643
Frédéric Lécaille74167b22019-05-28 19:02:42 +02003644 return o;
3645}
Emeric Brunb3971ab2015-05-12 18:49:09 +02003646
3647/*
Frédéric Lécaille8d78fa72019-05-20 18:22:52 +02003648 * Allocate a dictionary cache for each peer of <peers> section.
3649 * Return 1 if succeeded, 0 if not.
3650 */
3651int peers_alloc_dcache(struct peers *peers)
3652{
3653 struct peer *p;
3654
3655 for (p = peers->remote; p; p = p->next) {
3656 p->dcache = new_dcache(PEER_STKT_CACHE_MAX_ENTRIES);
3657 if (!p->dcache)
3658 return 0;
3659 }
3660
3661 return 1;
3662}
3663
3664/*
Emeric Brunb3971ab2015-05-12 18:49:09 +02003665 * Function used to register a table for sync on a group of peers
Remi Tricot-Le Breton208ff012021-05-12 17:39:04 +02003666 * Returns 0 in case of success.
Emeric Brunb3971ab2015-05-12 18:49:09 +02003667 */
Remi Tricot-Le Breton208ff012021-05-12 17:39:04 +02003668int peers_register_table(struct peers *peers, struct stktable *table)
Emeric Brunb3971ab2015-05-12 18:49:09 +02003669{
3670 struct shared_table *st;
3671 struct peer * curpeer;
3672 int id = 0;
Remi Tricot-Le Breton208ff012021-05-12 17:39:04 +02003673 int retval = 0;
Emeric Brunb3971ab2015-05-12 18:49:09 +02003674
3675 for (curpeer = peers->remote; curpeer; curpeer = curpeer->next) {
Vincent Bernat02779b62016-04-03 13:48:43 +02003676 st = calloc(1,sizeof(*st));
Remi Tricot-Le Breton208ff012021-05-12 17:39:04 +02003677 if (!st) {
3678 retval = 1;
3679 break;
3680 }
Emeric Brunb3971ab2015-05-12 18:49:09 +02003681 st->table = table;
3682 st->next = curpeer->tables;
3683 if (curpeer->tables)
3684 id = curpeer->tables->local_id;
3685 st->local_id = id + 1;
3686
Emeric Brun2cc201f2021-04-23 12:21:26 +02003687 /* If peer is local we inc table
3688 * refcnt to protect against flush
3689 * until this process pushed all
3690 * table content to the new one
3691 */
3692 if (curpeer->local)
3693 HA_ATOMIC_INC(&st->table->refcnt);
Emeric Brunb3971ab2015-05-12 18:49:09 +02003694 curpeer->tables = st;
3695 }
3696
3697 table->sync_task = peers->sync_task;
Remi Tricot-Le Breton208ff012021-05-12 17:39:04 +02003698
3699 return retval;
Emeric Brun2b920a12010-09-23 18:30:22 +02003700}
3701
Frédéric Lécaille95679dc2019-04-15 10:25:27 +02003702/*
3703 * Parse the "show peers" command arguments.
3704 * Returns 0 if succeeded, 1 if not with the ->msg of the appctx set as
3705 * error message.
3706 */
3707static int cli_parse_show_peers(char **args, char *payload, struct appctx *appctx, void *private)
3708{
3709 appctx->ctx.cfgpeers.target = NULL;
3710
Willy Tarreau49962b52021-02-12 16:56:22 +01003711 if (strcmp(args[2], "dict") == 0) {
3712 /* show the dictionaries (large dump) */
3713 appctx->ctx.cfgpeers.flags |= PEERS_SHOW_F_DICT;
3714 args++;
3715 } else if (strcmp(args[2], "-") == 0)
3716 args++; // allows to show a section called "dict"
3717
Frédéric Lécaille95679dc2019-04-15 10:25:27 +02003718 if (*args[2]) {
3719 struct peers *p;
3720
3721 for (p = cfg_peers; p; p = p->next) {
Tim Duesterhuse5ff1412021-01-02 22:31:53 +01003722 if (strcmp(p->id, args[2]) == 0) {
Frédéric Lécaille95679dc2019-04-15 10:25:27 +02003723 appctx->ctx.cfgpeers.target = p;
3724 break;
3725 }
3726 }
3727
Willy Tarreau9d008692019-08-09 11:21:01 +02003728 if (!p)
3729 return cli_err(appctx, "No such peers\n");
Frédéric Lécaille95679dc2019-04-15 10:25:27 +02003730 }
3731
3732 return 0;
3733}
3734
3735/*
3736 * This function dumps the peer state information of <peers> "peers" section.
3737 * Returns 0 if the output buffer is full and needs to be called again, non-zero if not.
3738 * Dedicated to be called by cli_io_handler_show_peers() cli I/O handler.
3739 */
Christopher Faulet908628c2022-03-25 16:43:49 +01003740static int peers_dump_head(struct buffer *msg, struct conn_stream *cs, struct peers *peers)
Frédéric Lécaille95679dc2019-04-15 10:25:27 +02003741{
3742 struct tm tm;
3743
3744 get_localtime(peers->last_change, &tm);
Willy Tarreau1ad64ac2020-09-24 08:48:08 +02003745 chunk_appendf(msg, "%p: [%02d/%s/%04d:%02d:%02d:%02d] id=%s disabled=%d flags=0x%x resync_timeout=%s task_calls=%u\n",
Frédéric Lécaille95679dc2019-04-15 10:25:27 +02003746 peers,
3747 tm.tm_mday, monthname[tm.tm_mon], tm.tm_year+1900,
3748 tm.tm_hour, tm.tm_min, tm.tm_sec,
Willy Tarreau1ad64ac2020-09-24 08:48:08 +02003749 peers->id, peers->disabled, peers->flags,
Frédéric Lécaille95679dc2019-04-15 10:25:27 +02003750 peers->resync_timeout ?
3751 tick_is_expired(peers->resync_timeout, now_ms) ? "<PAST>" :
3752 human_time(TICKS_TO_MS(peers->resync_timeout - now_ms),
Emeric Brun0bbec0f2019-04-18 11:39:43 +02003753 TICKS_TO_MS(1000)) : "<NEVER>",
3754 peers->sync_task ? peers->sync_task->calls : 0);
Frédéric Lécaille95679dc2019-04-15 10:25:27 +02003755
Christopher Faulet908628c2022-03-25 16:43:49 +01003756 if (ci_putchk(cs_ic(cs), msg) == -1) {
Christopher Fauleta0bdec32022-04-04 07:51:21 +02003757 cs_rx_room_blk(cs);
Frédéric Lécaille95679dc2019-04-15 10:25:27 +02003758 return 0;
3759 }
3760
3761 return 1;
3762}
3763
3764/*
3765 * This function dumps <peer> state information.
3766 * Returns 0 if the output buffer is full and needs to be called again, non-zero
3767 * if not. Dedicated to be called by cli_io_handler_show_peers() cli I/O handler.
3768 */
Christopher Faulet908628c2022-03-25 16:43:49 +01003769static int peers_dump_peer(struct buffer *msg, struct conn_stream *cs, struct peer *peer, int flags)
Frédéric Lécaille95679dc2019-04-15 10:25:27 +02003770{
3771 struct connection *conn;
3772 char pn[INET6_ADDRSTRLEN];
Christopher Faulet908628c2022-03-25 16:43:49 +01003773 struct conn_stream *peer_cs;
Frédéric Lécaille95679dc2019-04-15 10:25:27 +02003774 struct stream *peer_s;
3775 struct appctx *appctx;
3776 struct shared_table *st;
3777
3778 addr_to_str(&peer->addr, pn, sizeof pn);
Frédéric Lécaille3fc0fe02020-10-08 09:46:24 +02003779 chunk_appendf(msg, " %p: id=%s(%s,%s) addr=%s:%d last_status=%s",
Frédéric Lécaille95679dc2019-04-15 10:25:27 +02003780 peer, peer->id,
3781 peer->local ? "local" : "remote",
Frédéric Lécaillee7e2b212020-10-05 12:33:07 +02003782 peer->appctx ? "active" : "inactive",
Frédéric Lécaille95679dc2019-04-15 10:25:27 +02003783 pn, get_host_port(&peer->addr),
Frédéric Lécaille3fc0fe02020-10-08 09:46:24 +02003784 statuscode_str(peer->statuscode));
3785
3786 chunk_appendf(msg, " last_hdshk=%s\n",
3787 peer->last_hdshk ? human_time(TICKS_TO_MS(now_ms - peer->last_hdshk),
3788 TICKS_TO_MS(1000)) : "<NEVER>");
3789
3790 chunk_appendf(msg, " reconnect=%s",
Frédéric Lécaille95679dc2019-04-15 10:25:27 +02003791 peer->reconnect ?
3792 tick_is_expired(peer->reconnect, now_ms) ? "<PAST>" :
3793 human_time(TICKS_TO_MS(peer->reconnect - now_ms),
Frédéric Lécaille3fc0fe02020-10-08 09:46:24 +02003794 TICKS_TO_MS(1000)) : "<NEVER>");
3795
3796 chunk_appendf(msg, " heartbeat=%s",
3797 peer->heartbeat ?
3798 tick_is_expired(peer->heartbeat, now_ms) ? "<PAST>" :
3799 human_time(TICKS_TO_MS(peer->heartbeat - now_ms),
3800 TICKS_TO_MS(1000)) : "<NEVER>");
3801
3802 chunk_appendf(msg, " confirm=%u tx_hbt=%u rx_hbt=%u no_hbt=%u new_conn=%u proto_err=%u coll=%u\n",
Frédéric Lécailleec1c10b2019-11-07 15:22:33 +01003803 peer->confirm, peer->tx_hbt, peer->rx_hbt,
Frédéric Lécaille3fc0fe02020-10-08 09:46:24 +02003804 peer->no_hbt, peer->new_conn, peer->proto_err, peer->coll);
Frédéric Lécaille95679dc2019-04-15 10:25:27 +02003805
3806 chunk_appendf(&trash, " flags=0x%x", peer->flags);
3807
3808 appctx = peer->appctx;
3809 if (!appctx)
Frédéric Lécaille470502b2019-11-06 10:41:03 +01003810 goto table_info;
Frédéric Lécaille95679dc2019-04-15 10:25:27 +02003811
Emeric Brun0bbec0f2019-04-18 11:39:43 +02003812 chunk_appendf(&trash, " appctx:%p st0=%d st1=%d task_calls=%u", appctx, appctx->st0, appctx->st1,
3813 appctx->t ? appctx->t->calls : 0);
Frédéric Lécaille95679dc2019-04-15 10:25:27 +02003814
Christopher Faulet908628c2022-03-25 16:43:49 +01003815 peer_cs = peer->appctx->owner;
3816 peer_s = __cs_strm(peer_cs);
Frédéric Lécaille95679dc2019-04-15 10:25:27 +02003817
Christopher Faulet62e75742022-03-31 09:16:34 +02003818 chunk_appendf(&trash, " state=%s", cs_state_str(cs_opposite(peer_cs)->state));
Frédéric Lécaille95679dc2019-04-15 10:25:27 +02003819
3820 conn = objt_conn(strm_orig(peer_s));
3821 if (conn)
3822 chunk_appendf(&trash, "\n xprt=%s", conn_get_xprt_name(conn));
3823
Willy Tarreau3ca14902019-07-17 14:53:15 +02003824 switch (conn && conn_get_src(conn) ? addr_to_str(conn->src, pn, sizeof(pn)) : AF_UNSPEC) {
Frédéric Lécaille95679dc2019-04-15 10:25:27 +02003825 case AF_INET:
3826 case AF_INET6:
Willy Tarreau3ca14902019-07-17 14:53:15 +02003827 chunk_appendf(&trash, " src=%s:%d", pn, get_host_port(conn->src));
Frédéric Lécaille95679dc2019-04-15 10:25:27 +02003828 break;
3829 case AF_UNIX:
3830 chunk_appendf(&trash, " src=unix:%d", strm_li(peer_s)->luid);
3831 break;
3832 }
3833
Willy Tarreau3ca14902019-07-17 14:53:15 +02003834 switch (conn && conn_get_dst(conn) ? addr_to_str(conn->dst, pn, sizeof(pn)) : AF_UNSPEC) {
Frédéric Lécaille95679dc2019-04-15 10:25:27 +02003835 case AF_INET:
3836 case AF_INET6:
Willy Tarreau3ca14902019-07-17 14:53:15 +02003837 chunk_appendf(&trash, " addr=%s:%d", pn, get_host_port(conn->dst));
Frédéric Lécaille95679dc2019-04-15 10:25:27 +02003838 break;
3839 case AF_UNIX:
3840 chunk_appendf(&trash, " addr=unix:%d", strm_li(peer_s)->luid);
3841 break;
3842 }
3843
Frédéric Lécaille470502b2019-11-06 10:41:03 +01003844 table_info:
Frédéric Lécaille95679dc2019-04-15 10:25:27 +02003845 if (peer->remote_table)
3846 chunk_appendf(&trash, "\n remote_table:%p id=%s local_id=%d remote_id=%d",
3847 peer->remote_table,
3848 peer->remote_table->table->id,
3849 peer->remote_table->local_id,
3850 peer->remote_table->remote_id);
3851
3852 if (peer->last_local_table)
3853 chunk_appendf(&trash, "\n last_local_table:%p id=%s local_id=%d remote_id=%d",
3854 peer->last_local_table,
3855 peer->last_local_table->table->id,
3856 peer->last_local_table->local_id,
3857 peer->last_local_table->remote_id);
3858
3859 if (peer->tables) {
3860 chunk_appendf(&trash, "\n shared tables:");
3861 for (st = peer->tables; st; st = st->next) {
Frédéric Lécaille62b0b0b2019-05-29 16:20:41 +02003862 int i, count;
Frédéric Lécaille95679dc2019-04-15 10:25:27 +02003863 struct stktable *t;
Frédéric Lécaille62b0b0b2019-05-29 16:20:41 +02003864 struct dcache *dcache;
3865
Frédéric Lécaille95679dc2019-04-15 10:25:27 +02003866 t = st->table;
Frédéric Lécaille62b0b0b2019-05-29 16:20:41 +02003867 dcache = peer->dcache;
3868
Frédéric Lécaille95679dc2019-04-15 10:25:27 +02003869 chunk_appendf(&trash, "\n %p local_id=%d remote_id=%d "
3870 "flags=0x%x remote_data=0x%llx",
3871 st, st->local_id, st->remote_id,
3872 st->flags, (unsigned long long)st->remote_data);
3873 chunk_appendf(&trash, "\n last_acked=%u last_pushed=%u last_get=%u"
3874 " teaching_origin=%u update=%u",
3875 st->last_acked, st->last_pushed, st->last_get,
3876 st->teaching_origin, st->update);
3877 chunk_appendf(&trash, "\n table:%p id=%s update=%u localupdate=%u"
Emeric Brun2cc201f2021-04-23 12:21:26 +02003878 " commitupdate=%u refcnt=%u",
3879 t, t->id, t->update, t->localupdate, t->commitupdate, t->refcnt);
Willy Tarreau49962b52021-02-12 16:56:22 +01003880 if (flags & PEERS_SHOW_F_DICT) {
3881 chunk_appendf(&trash, "\n TX dictionary cache:");
3882 count = 0;
3883 for (i = 0; i < dcache->max_entries; i++) {
3884 struct ebpt_node *node;
3885 struct dict_entry *de;
Frédéric Lécaille62b0b0b2019-05-29 16:20:41 +02003886
Willy Tarreau49962b52021-02-12 16:56:22 +01003887 node = &dcache->tx->entries[i];
3888 if (!node->key)
3889 break;
Frédéric Lécaille6c391982019-06-06 11:34:03 +02003890
Willy Tarreau49962b52021-02-12 16:56:22 +01003891 if (!count++)
3892 chunk_appendf(&trash, "\n ");
3893 de = node->key;
3894 chunk_appendf(&trash, " %3u -> %s", i, (char *)de->value.key);
3895 count &= 0x3;
3896 }
3897 chunk_appendf(&trash, "\n RX dictionary cache:");
3898 count = 0;
3899 for (i = 0; i < dcache->max_entries; i++) {
3900 if (!count++)
3901 chunk_appendf(&trash, "\n ");
3902 chunk_appendf(&trash, " %3u -> %s", i,
3903 dcache->rx[i].de ?
3904 (char *)dcache->rx[i].de->value.key : "-");
3905 count &= 0x3;
3906 }
3907 } else {
3908 chunk_appendf(&trash, "\n Dictionary cache not dumped (use \"show peers dict\")");
Frédéric Lécaille62b0b0b2019-05-29 16:20:41 +02003909 }
Frédéric Lécaille95679dc2019-04-15 10:25:27 +02003910 }
3911 }
3912
3913 end:
3914 chunk_appendf(&trash, "\n");
Christopher Faulet908628c2022-03-25 16:43:49 +01003915 if (ci_putchk(cs_ic(cs), msg) == -1) {
Christopher Fauleta0bdec32022-04-04 07:51:21 +02003916 cs_rx_room_blk(cs);
Frédéric Lécaille95679dc2019-04-15 10:25:27 +02003917 return 0;
3918 }
3919
3920 return 1;
3921}
3922
3923/*
3924 * This function dumps all the peers of "peers" section.
3925 * Returns 0 if the output buffer is full and needs to be called
3926 * again, non-zero if not. It proceeds in an isolated thread, so
3927 * there is no thread safety issue here.
3928 */
3929static int cli_io_handler_show_peers(struct appctx *appctx)
3930{
3931 int show_all;
3932 int ret = 0, first_peers = 1;
Frédéric Lécaille95679dc2019-04-15 10:25:27 +02003933
3934 thread_isolate();
3935
3936 show_all = !appctx->ctx.cfgpeers.target;
3937
3938 chunk_reset(&trash);
3939
3940 while (appctx->st2 != STAT_ST_FIN) {
3941 switch (appctx->st2) {
3942 case STAT_ST_INIT:
3943 if (show_all)
3944 appctx->ctx.cfgpeers.peers = cfg_peers;
3945 else
3946 appctx->ctx.cfgpeers.peers = appctx->ctx.cfgpeers.target;
3947
3948 appctx->st2 = STAT_ST_LIST;
3949 /* fall through */
3950
3951 case STAT_ST_LIST:
3952 if (!appctx->ctx.cfgpeers.peers) {
3953 /* No more peers list. */
3954 appctx->st2 = STAT_ST_END;
3955 }
3956 else {
3957 if (!first_peers)
3958 chunk_appendf(&trash, "\n");
3959 else
3960 first_peers = 0;
Christopher Faulet908628c2022-03-25 16:43:49 +01003961 if (!peers_dump_head(&trash, appctx->owner, appctx->ctx.cfgpeers.peers))
Frédéric Lécaille95679dc2019-04-15 10:25:27 +02003962 goto out;
3963
3964 appctx->ctx.cfgpeers.peer = appctx->ctx.cfgpeers.peers->remote;
3965 appctx->ctx.cfgpeers.peers = appctx->ctx.cfgpeers.peers->next;
3966 appctx->st2 = STAT_ST_INFO;
3967 }
3968 break;
3969
3970 case STAT_ST_INFO:
3971 if (!appctx->ctx.cfgpeers.peer) {
3972 /* End of peer list */
3973 if (show_all)
3974 appctx->st2 = STAT_ST_LIST;
3975 else
3976 appctx->st2 = STAT_ST_END;
3977 }
3978 else {
Christopher Faulet908628c2022-03-25 16:43:49 +01003979 if (!peers_dump_peer(&trash, appctx->owner, appctx->ctx.cfgpeers.peer, appctx->ctx.cfgpeers.flags))
Frédéric Lécaille95679dc2019-04-15 10:25:27 +02003980 goto out;
3981
3982 appctx->ctx.cfgpeers.peer = appctx->ctx.cfgpeers.peer->next;
3983 }
3984 break;
3985
3986 case STAT_ST_END:
3987 appctx->st2 = STAT_ST_FIN;
3988 break;
3989 }
3990 }
3991 ret = 1;
3992 out:
3993 thread_release();
3994 return ret;
3995}
3996
3997/*
3998 * CLI keywords.
3999 */
4000static struct cli_kw_list cli_kws = {{ }, {
Willy Tarreaub205bfd2021-05-07 11:38:37 +02004001 { { "show", "peers", NULL }, "show peers [dict|-] [section] : dump some information about all the peers or this peers section", cli_parse_show_peers, cli_io_handler_show_peers, },
Frédéric Lécaille95679dc2019-04-15 10:25:27 +02004002 {},
4003}};
4004
4005/* Register cli keywords */
4006INITCALL1(STG_REGISTER, cli_register_kw, &cli_kws);