blob: 039a625a1fb43bb6f215c276ed9f7641067d5330 [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>
14#include <fcntl.h>
15#include <stdio.h>
16#include <stdlib.h>
17#include <string.h>
18
19#include <sys/socket.h>
20#include <sys/stat.h>
21#include <sys/types.h>
22
23#include <common/compat.h>
24#include <common/config.h>
25#include <common/time.h>
Frédéric Lécaille523cc9e2016-10-12 17:30:30 +020026#include <common/standard.h>
Christopher Faulet8d8aa0d2017-05-30 15:36:50 +020027#include <common/hathreads.h>
Emeric Brun2b920a12010-09-23 18:30:22 +020028
29#include <types/global.h>
Willy Tarreau3fdb3662012-11-12 00:42:33 +010030#include <types/listener.h>
31#include <types/obj_type.h>
Emeric Brun2b920a12010-09-23 18:30:22 +020032#include <types/peers.h>
Frédéric Lécaille95679dc2019-04-15 10:25:27 +020033#include <types/stats.h>
Emeric Brun2b920a12010-09-23 18:30:22 +020034
35#include <proto/acl.h>
Willy Tarreau8a8d83b2015-04-13 13:24:54 +020036#include <proto/applet.h>
Willy Tarreauc7e42382012-08-24 19:22:53 +020037#include <proto/channel.h>
Frédéric Lécaille95679dc2019-04-15 10:25:27 +020038#include <proto/cli.h>
Frédéric Lécaille8d78fa72019-05-20 18:22:52 +020039#include <proto/dict.h>
Emeric Brun2b920a12010-09-23 18:30:22 +020040#include <proto/fd.h>
Willy Tarreaud1d48d42015-03-13 16:15:46 +010041#include <proto/frontend.h>
Emeric Brun2b920a12010-09-23 18:30:22 +020042#include <proto/log.h>
43#include <proto/hdr_idx.h>
Willy Tarreau53a47662017-08-28 10:53:00 +020044#include <proto/mux_pt.h>
Frédéric Lécaille1055e682018-04-26 14:35:21 +020045#include <proto/peers.h>
Emeric Brun2b920a12010-09-23 18:30:22 +020046#include <proto/proxy.h>
Willy Tarreaufeb76402015-04-03 14:10:06 +020047#include <proto/session.h>
Willy Tarreau87b09662015-04-03 00:22:06 +020048#include <proto/stream.h>
Willy Tarreau22ec1ea2014-11-27 20:45:39 +010049#include <proto/signal.h>
50#include <proto/stick_table.h>
Emeric Brun2b920a12010-09-23 18:30:22 +020051#include <proto/stream_interface.h>
Emeric Brun2b920a12010-09-23 18:30:22 +020052#include <proto/task.h>
Emeric Brun2b920a12010-09-23 18:30:22 +020053
54
55/*******************************/
56/* Current peer learning state */
57/*******************************/
58
59/******************************/
Emeric Brunb3971ab2015-05-12 18:49:09 +020060/* Current peers section resync state */
Emeric Brun2b920a12010-09-23 18:30:22 +020061/******************************/
Frédéric Lécailleaba44a22019-03-26 10:18:07 +010062#define PEERS_F_RESYNC_LOCAL 0x00000001 /* Learn from local finished or no more needed */
63#define PEERS_F_RESYNC_REMOTE 0x00000002 /* Learn from remote finished or no more needed */
64#define PEERS_F_RESYNC_ASSIGN 0x00000004 /* A peer was assigned to learn our lesson */
65#define PEERS_F_RESYNC_PROCESS 0x00000008 /* The assigned peer was requested for resync */
66#define PEERS_F_DONOTSTOP 0x00010000 /* Main table sync task block process during soft stop
67 to push data to new process */
Emeric Brun2b920a12010-09-23 18:30:22 +020068
Frédéric Lécailleaba44a22019-03-26 10:18:07 +010069#define PEERS_RESYNC_STATEMASK (PEERS_F_RESYNC_LOCAL|PEERS_F_RESYNC_REMOTE)
70#define PEERS_RESYNC_FROMLOCAL 0x00000000
71#define PEERS_RESYNC_FROMREMOTE PEERS_F_RESYNC_LOCAL
72#define PEERS_RESYNC_FINISHED (PEERS_F_RESYNC_LOCAL|PEERS_F_RESYNC_REMOTE)
Emeric Brunb3971ab2015-05-12 18:49:09 +020073
74/***********************************/
75/* Current shared table sync state */
76/***********************************/
Frédéric Lécailleaba44a22019-03-26 10:18:07 +010077#define SHTABLE_F_TEACH_STAGE1 0x00000001 /* Teach state 1 complete */
78#define SHTABLE_F_TEACH_STAGE2 0x00000002 /* Teach state 2 complete */
Emeric Brun2b920a12010-09-23 18:30:22 +020079
80/******************************/
81/* Remote peer teaching state */
82/******************************/
Frédéric Lécailleaba44a22019-03-26 10:18:07 +010083#define PEER_F_TEACH_PROCESS 0x00000001 /* Teach a lesson to current peer */
84#define PEER_F_TEACH_FINISHED 0x00000008 /* Teach conclude, (wait for confirm) */
85#define PEER_F_TEACH_COMPLETE 0x00000010 /* All that we know already taught to current peer, used only for a local peer */
86#define PEER_F_LEARN_ASSIGN 0x00000100 /* Current peer was assigned for a lesson */
87#define PEER_F_LEARN_NOTUP2DATE 0x00000200 /* Learn from peer finished but peer is not up to date */
88#define PEER_F_ALIVE 0x20000000 /* Used to flag a peer a alive. */
89#define PEER_F_HEARTBEAT 0x40000000 /* Heartbeat message to send. */
90#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 +020091
Frédéric Lécailleaba44a22019-03-26 10:18:07 +010092#define PEER_TEACH_RESET ~(PEER_F_TEACH_PROCESS|PEER_F_TEACH_FINISHED) /* PEER_F_TEACH_COMPLETE should never be reset */
93#define PEER_LEARN_RESET ~(PEER_F_LEARN_ASSIGN|PEER_F_LEARN_NOTUP2DATE)
Emeric Brun2b920a12010-09-23 18:30:22 +020094
Frédéric Lécaille54bff832019-03-26 10:25:20 +010095#define PEER_RESYNC_TIMEOUT 5000 /* 5 seconds */
96#define PEER_RECONNECT_TIMEOUT 5000 /* 5 seconds */
Frédéric Lécaille645635d2019-02-11 17:49:39 +010097#define PEER_HEARTBEAT_TIMEOUT 3000 /* 3 seconds */
98
Emeric Brunb3971ab2015-05-12 18:49:09 +020099/*****************************/
100/* Sync message class */
101/*****************************/
102enum {
103 PEER_MSG_CLASS_CONTROL = 0,
104 PEER_MSG_CLASS_ERROR,
105 PEER_MSG_CLASS_STICKTABLE = 10,
106 PEER_MSG_CLASS_RESERVED = 255,
107};
108
109/*****************************/
110/* control message types */
111/*****************************/
112enum {
113 PEER_MSG_CTRL_RESYNCREQ = 0,
114 PEER_MSG_CTRL_RESYNCFINISHED,
115 PEER_MSG_CTRL_RESYNCPARTIAL,
116 PEER_MSG_CTRL_RESYNCCONFIRM,
Frédéric Lécaille645635d2019-02-11 17:49:39 +0100117 PEER_MSG_CTRL_HEARTBEAT,
Emeric Brunb3971ab2015-05-12 18:49:09 +0200118};
119
120/*****************************/
121/* error message types */
122/*****************************/
123enum {
124 PEER_MSG_ERR_PROTOCOL = 0,
125 PEER_MSG_ERR_SIZELIMIT,
126};
127
Frédéric Lécailled5fe14b2019-01-24 10:33:40 +0100128/*
129 * Parameters used by functions to build peer protocol messages. */
130struct peer_prep_params {
131 struct {
Frédéric Lécaille7d0ceee2019-01-24 14:24:05 +0100132 struct peer *peer;
133 } hello;
134 struct {
135 unsigned int st1;
136 } error_status;
137 struct {
Frédéric Lécailled5fe14b2019-01-24 10:33:40 +0100138 struct stksess *stksess;
139 struct shared_table *shared_table;
140 unsigned int updateid;
141 int use_identifier;
142 int use_timed;
Frédéric Lécaille8d78fa72019-05-20 18:22:52 +0200143 struct peer *peer;
Frédéric Lécailled5fe14b2019-01-24 10:33:40 +0100144 } updt;
145 struct {
146 struct shared_table *shared_table;
147 } swtch;
148 struct {
149 struct shared_table *shared_table;
150 } ack;
Frédéric Lécaille7d0ceee2019-01-24 14:24:05 +0100151 struct {
152 unsigned char head[2];
153 } control;
154 struct {
155 unsigned char head[2];
156 } error;
Frédéric Lécailled5fe14b2019-01-24 10:33:40 +0100157};
Emeric Brunb3971ab2015-05-12 18:49:09 +0200158
159/*******************************/
160/* stick table sync mesg types */
161/* Note: ids >= 128 contains */
162/* id message cotains data */
163/*******************************/
Olivier Houchard33992262018-10-16 18:49:26 +0200164#define PEER_MSG_STKT_UPDATE 0x80
165#define PEER_MSG_STKT_INCUPDATE 0x81
166#define PEER_MSG_STKT_DEFINE 0x82
167#define PEER_MSG_STKT_SWITCH 0x83
168#define PEER_MSG_STKT_ACK 0x84
169#define PEER_MSG_STKT_UPDATE_TIMED 0x85
170#define PEER_MSG_STKT_INCUPDATE_TIMED 0x86
Emeric Brun2b920a12010-09-23 18:30:22 +0200171
Frédéric Lécaille39143342019-05-24 14:32:27 +0200172/* The maximum length of an encoded data length. */
173#define PEER_MSG_ENC_LENGTH_MAXLEN 5
174
Frédéric Lécaille32b55732019-06-03 18:29:51 +0200175/* Minimum 64-bits value encoded with 2 bytes */
176#define PEER_ENC_2BYTES_MIN 0xf0 /* 0xf0 (or 240) */
177/* 3 bytes */
178#define PEER_ENC_3BYTES_MIN ((1ULL << 11) | PEER_ENC_2BYTES_MIN) /* 0x8f0 (or 2288) */
179/* 4 bytes */
180#define PEER_ENC_4BYTES_MIN ((1ULL << 18) | PEER_ENC_3BYTES_MIN) /* 0x408f0 (or 264432) */
181/* 5 bytes */
182#define PEER_ENC_5BYTES_MIN ((1ULL << 25) | PEER_ENC_4BYTES_MIN) /* 0x20408f0 (or 33818864) */
183/* 6 bytes */
184#define PEER_ENC_6BYTES_MIN ((1ULL << 32) | PEER_ENC_5BYTES_MIN) /* 0x1020408f0 (or 4328786160) */
185/* 7 bytes */
186#define PEER_ENC_7BYTES_MIN ((1ULL << 39) | PEER_ENC_6BYTES_MIN) /* 0x81020408f0 (or 554084600048) */
187/* 8 bytes */
188#define PEER_ENC_8BYTES_MIN ((1ULL << 46) | PEER_ENC_7BYTES_MIN) /* 0x4081020408f0 (or 70922828777712) */
189/* 9 bytes */
190#define PEER_ENC_9BYTES_MIN ((1ULL << 53) | PEER_ENC_8BYTES_MIN) /* 0x204081020408f0 (or 9078122083518704) */
191/* 10 bytes */
192#define PEER_ENC_10BYTES_MIN ((1ULL << 60) | PEER_ENC_9BYTES_MIN) /* 0x10204081020408f0 (or 1161999626690365680) */
193
194/* #7 bit used to detect the last byte to be encoded */
195#define PEER_ENC_STOP_BIT 7
196/* The byte minimum value with #7 bit set */
197#define PEER_ENC_STOP_BYTE (1 << PEER_ENC_STOP_BIT)
198/* The left most number of bits set for PEER_ENC_2BYTES_MIN */
199#define PEER_ENC_2BYTES_MIN_BITS 4
200
Frédéric Lécaille39143342019-05-24 14:32:27 +0200201#define PEER_MSG_HEADER_LEN 2
202
Frédéric Lécaille8d78fa72019-05-20 18:22:52 +0200203#define PEER_STKT_CACHE_MAX_ENTRIES 128
204
Emeric Brun2b920a12010-09-23 18:30:22 +0200205/**********************************/
206/* Peer Session IO handler states */
207/**********************************/
208
Willy Tarreaue4d927a2013-12-01 12:47:35 +0100209enum {
210 PEER_SESS_ST_ACCEPT = 0, /* Initial state for session create by an accept, must be zero! */
211 PEER_SESS_ST_GETVERSION, /* Validate supported protocol version */
212 PEER_SESS_ST_GETHOST, /* Validate host ID correspond to local host id */
213 PEER_SESS_ST_GETPEER, /* Validate peer ID correspond to a known remote peer id */
Willy Tarreaue4d927a2013-12-01 12:47:35 +0100214 /* after this point, data were possibly exchanged */
215 PEER_SESS_ST_SENDSUCCESS, /* Send ret code 200 (success) and wait for message */
216 PEER_SESS_ST_CONNECT, /* Initial state for session create on a connect, push presentation into buffer */
217 PEER_SESS_ST_GETSTATUS, /* Wait for the welcome message */
218 PEER_SESS_ST_WAITMSG, /* Wait for data messages */
219 PEER_SESS_ST_EXIT, /* Exit with status code */
Emeric Brunb3971ab2015-05-12 18:49:09 +0200220 PEER_SESS_ST_ERRPROTO, /* Send error proto message before exit */
221 PEER_SESS_ST_ERRSIZE, /* Send error size message before exit */
Willy Tarreaue4d927a2013-12-01 12:47:35 +0100222 PEER_SESS_ST_END, /* Killed session */
223};
Emeric Brun2b920a12010-09-23 18:30:22 +0200224
Willy Tarreaue4d927a2013-12-01 12:47:35 +0100225/***************************************************/
226/* Peer Session status code - part of the protocol */
227/***************************************************/
Emeric Brun2b920a12010-09-23 18:30:22 +0200228
Frédéric Lécailleaba44a22019-03-26 10:18:07 +0100229#define PEER_SESS_SC_CONNECTCODE 100 /* connect in progress */
230#define PEER_SESS_SC_CONNECTEDCODE 110 /* tcp connect success */
Emeric Brun2b920a12010-09-23 18:30:22 +0200231
Frédéric Lécailleaba44a22019-03-26 10:18:07 +0100232#define PEER_SESS_SC_SUCCESSCODE 200 /* accept or connect successful */
Emeric Brun2b920a12010-09-23 18:30:22 +0200233
Frédéric Lécailleaba44a22019-03-26 10:18:07 +0100234#define PEER_SESS_SC_TRYAGAIN 300 /* try again later */
Emeric Brun2b920a12010-09-23 18:30:22 +0200235
Frédéric Lécailleaba44a22019-03-26 10:18:07 +0100236#define PEER_SESS_SC_ERRPROTO 501 /* error protocol */
237#define PEER_SESS_SC_ERRVERSION 502 /* unknown protocol version */
238#define PEER_SESS_SC_ERRHOST 503 /* bad host name */
239#define PEER_SESS_SC_ERRPEER 504 /* unknown peer */
Emeric Brun2b920a12010-09-23 18:30:22 +0200240
241#define PEER_SESSION_PROTO_NAME "HAProxyS"
Frédéric Lécaille523cc9e2016-10-12 17:30:30 +0200242#define PEER_MAJOR_VER 2
243#define PEER_MINOR_VER 1
244#define PEER_DWNGRD_MINOR_VER 0
Emeric Brun2b920a12010-09-23 18:30:22 +0200245
Willy Tarreau6254a922019-01-29 17:45:23 +0100246static size_t proto_len = sizeof(PEER_SESSION_PROTO_NAME) - 1;
Frédéric Lécailleed2b4a62017-07-13 09:07:09 +0200247struct peers *cfg_peers = NULL;
Emeric Brun9ef2ad72019-04-02 17:22:01 +0200248static void peer_session_forceshutdown(struct peer *peer);
Emeric Brun2b920a12010-09-23 18:30:22 +0200249
Frédéric Lécaille8d78fa72019-05-20 18:22:52 +0200250struct dcache_tx_entry *dcache_tx_insert(struct dcache *dc,
251 struct dcache_tx_entry *i);
252static inline void flush_dcache(struct peer *peer);
253
Frédéric Lécaille95679dc2019-04-15 10:25:27 +0200254static const char *statuscode_str(int statuscode)
255{
256 switch (statuscode) {
257 case PEER_SESS_SC_CONNECTCODE:
258 return "CONN";
259 case PEER_SESS_SC_CONNECTEDCODE:
260 return "HSHK";
261 case PEER_SESS_SC_SUCCESSCODE:
262 return "ESTA";
263 case PEER_SESS_SC_TRYAGAIN:
264 return "RETR";
265 case PEER_SESS_SC_ERRPROTO:
266 return "PROT";
267 case PEER_SESS_SC_ERRVERSION:
268 return "VERS";
269 case PEER_SESS_SC_ERRHOST:
270 return "NAME";
271 case PEER_SESS_SC_ERRPEER:
272 return "UNKN";
273 default:
274 return "NONE";
275 }
276}
277
Emeric Brun18928af2017-03-29 16:32:53 +0200278/* This function encode an uint64 to 'dynamic' length format.
279 The encoded value is written at address *str, and the
280 caller must assure that size after *str is large enought.
281 At return, the *str is set at the next Byte after then
282 encoded integer. The function returns then length of the
283 encoded integer in Bytes */
Emeric Brunb3971ab2015-05-12 18:49:09 +0200284int intencode(uint64_t i, char **str) {
285 int idx = 0;
286 unsigned char *msg;
287
Emeric Brunb3971ab2015-05-12 18:49:09 +0200288 msg = (unsigned char *)*str;
Frédéric Lécaille32b55732019-06-03 18:29:51 +0200289 if (i < PEER_ENC_2BYTES_MIN) {
Emeric Brunb3971ab2015-05-12 18:49:09 +0200290 msg[0] = (unsigned char)i;
291 *str = (char *)&msg[idx+1];
292 return (idx+1);
293 }
294
Frédéric Lécaille32b55732019-06-03 18:29:51 +0200295 msg[idx] =(unsigned char)i | PEER_ENC_2BYTES_MIN;
296 i = (i - PEER_ENC_2BYTES_MIN) >> PEER_ENC_2BYTES_MIN_BITS;
297 while (i >= PEER_ENC_STOP_BYTE) {
298 msg[++idx] = (unsigned char)i | PEER_ENC_STOP_BYTE;
299 i = (i - PEER_ENC_STOP_BYTE) >> PEER_ENC_STOP_BIT;
Emeric Brunb3971ab2015-05-12 18:49:09 +0200300 }
301 msg[++idx] = (unsigned char)i;
302 *str = (char *)&msg[idx+1];
303 return (idx+1);
304}
305
306
307/* This function returns the decoded integer or 0
308 if decode failed
309 *str point on the beginning of the integer to decode
310 at the end of decoding *str point on the end of the
311 encoded integer or to null if end is reached */
Emeric Brun18928af2017-03-29 16:32:53 +0200312uint64_t intdecode(char **str, char *end)
313{
Emeric Brunb3971ab2015-05-12 18:49:09 +0200314 unsigned char *msg;
Emeric Brun18928af2017-03-29 16:32:53 +0200315 uint64_t i;
316 int shift;
Emeric Brunb3971ab2015-05-12 18:49:09 +0200317
318 if (!*str)
319 return 0;
320
321 msg = (unsigned char *)*str;
Emeric Brun18928af2017-03-29 16:32:53 +0200322 if (msg >= (unsigned char *)end)
323 goto fail;
Emeric Brunb3971ab2015-05-12 18:49:09 +0200324
Emeric Brun18928af2017-03-29 16:32:53 +0200325 i = *(msg++);
Frédéric Lécaille32b55732019-06-03 18:29:51 +0200326 if (i >= PEER_ENC_2BYTES_MIN) {
327 shift = PEER_ENC_2BYTES_MIN_BITS;
Emeric Brun18928af2017-03-29 16:32:53 +0200328 do {
329 if (msg >= (unsigned char *)end)
330 goto fail;
331 i += (uint64_t)*msg << shift;
Frédéric Lécaille32b55732019-06-03 18:29:51 +0200332 shift += PEER_ENC_STOP_BIT;
333 } while (*(msg++) >= PEER_ENC_STOP_BYTE);
Emeric Brunb3971ab2015-05-12 18:49:09 +0200334 }
Frédéric Lécaillea8725ec2019-01-22 10:31:39 +0100335 *str = (char *)msg;
336 return i;
Emeric Brun18928af2017-03-29 16:32:53 +0200337
Frédéric Lécaillea8725ec2019-01-22 10:31:39 +0100338 fail:
339 *str = NULL;
340 return 0;
Emeric Brunb3971ab2015-05-12 18:49:09 +0200341}
Emeric Brun2b920a12010-09-23 18:30:22 +0200342
Frédéric Lécaille7d0ceee2019-01-24 14:24:05 +0100343/*
344 * Build a "hello" peer protocol message.
345 * Return the number of written bytes written to build this messages if succeeded,
346 * 0 if not.
347 */
348static int peer_prepare_hellomsg(char *msg, size_t size, struct peer_prep_params *p)
349{
350 int min_ver, ret;
351 struct peer *peer;
352
353 peer = p->hello.peer;
354 min_ver = (peer->flags & PEER_F_DWNGRD) ? PEER_DWNGRD_MINOR_VER : PEER_MINOR_VER;
355 /* Prepare headers */
356 ret = snprintf(msg, size, PEER_SESSION_PROTO_NAME " %u.%u\n%s\n%s %d %d\n",
357 PEER_MAJOR_VER, min_ver, peer->id, localpeer, (int)getpid(), relative_pid);
358 if (ret >= size)
359 return 0;
360
361 return ret;
362}
363
364/*
365 * Build a "handshake succeeded" status message.
366 * Return the number of written bytes written to build this messages if succeeded,
367 * 0 if not.
368 */
369static int peer_prepare_status_successmsg(char *msg, size_t size, struct peer_prep_params *p)
370{
371 int ret;
372
373 ret = snprintf(msg, size, "%d\n", PEER_SESS_SC_SUCCESSCODE);
374 if (ret >= size)
375 return 0;
376
377 return ret;
378}
379
380/*
381 * Build an error status message.
382 * Return the number of written bytes written to build this messages if succeeded,
383 * 0 if not.
384 */
385static int peer_prepare_status_errormsg(char *msg, size_t size, struct peer_prep_params *p)
386{
387 int ret;
388 unsigned int st1;
389
390 st1 = p->error_status.st1;
391 ret = snprintf(msg, size, "%d\n", st1);
392 if (ret >= size)
393 return 0;
394
395 return ret;
396}
397
Frédéric Lécaille523cc9e2016-10-12 17:30:30 +0200398/* Set the stick-table UPDATE message type byte at <msg_type> address,
399 * depending on <use_identifier> and <use_timed> boolean parameters.
400 * Always successful.
401 */
402static inline void peer_set_update_msg_type(char *msg_type, int use_identifier, int use_timed)
403{
404 if (use_timed) {
405 if (use_identifier)
406 *msg_type = PEER_MSG_STKT_UPDATE_TIMED;
407 else
408 *msg_type = PEER_MSG_STKT_INCUPDATE_TIMED;
409 }
410 else {
411 if (use_identifier)
412 *msg_type = PEER_MSG_STKT_UPDATE;
413 else
414 *msg_type = PEER_MSG_STKT_INCUPDATE;
415 }
Frédéric Lécaille523cc9e2016-10-12 17:30:30 +0200416}
Emeric Brun2b920a12010-09-23 18:30:22 +0200417/*
Emeric Brunb3971ab2015-05-12 18:49:09 +0200418 * This prepare the data update message on the stick session <ts>, <st> is the considered
419 * stick table.
Joseph Herlant82b2f542018-11-15 12:19:14 -0800420 * <msg> is a buffer of <size> to receive data message content
Emeric Brunb3971ab2015-05-12 18:49:09 +0200421 * If function returns 0, the caller should consider we were unable to encode this message (TODO:
422 * check size)
Emeric Brun2b920a12010-09-23 18:30:22 +0200423 */
Frédéric Lécailled5fe14b2019-01-24 10:33:40 +0100424static int peer_prepare_updatemsg(char *msg, size_t size, struct peer_prep_params *p)
Emeric Brun2b920a12010-09-23 18:30:22 +0200425{
426 uint32_t netinteger;
Emeric Brunb3971ab2015-05-12 18:49:09 +0200427 unsigned short datalen;
428 char *cursor, *datamsg;
Emeric Brun94900952015-06-11 18:25:54 +0200429 unsigned int data_type;
430 void *data_ptr;
Frédéric Lécailled5fe14b2019-01-24 10:33:40 +0100431 struct stksess *ts;
432 struct shared_table *st;
433 unsigned int updateid;
434 int use_identifier;
435 int use_timed;
Frédéric Lécaille8d78fa72019-05-20 18:22:52 +0200436 struct peer *peer;
Frédéric Lécailled5fe14b2019-01-24 10:33:40 +0100437
438 ts = p->updt.stksess;
439 st = p->updt.shared_table;
440 updateid = p->updt.updateid;
441 use_identifier = p->updt.use_identifier;
442 use_timed = p->updt.use_timed;
Frédéric Lécaille8d78fa72019-05-20 18:22:52 +0200443 peer = p->updt.peer;
Emeric Brunb3971ab2015-05-12 18:49:09 +0200444
Frédéric Lécaille0e8db972019-05-24 14:34:34 +0200445 cursor = datamsg = msg + PEER_MSG_HEADER_LEN + PEER_MSG_ENC_LENGTH_MAXLEN;
Emeric Brunb3971ab2015-05-12 18:49:09 +0200446
Emeric Brun2b920a12010-09-23 18:30:22 +0200447 /* construct message */
Emeric Brunb3971ab2015-05-12 18:49:09 +0200448
449 /* check if we need to send the update identifer */
Emeric Brun819fc6f2017-06-13 19:37:32 +0200450 if (!st->last_pushed || updateid < st->last_pushed || ((updateid - st->last_pushed) != 1)) {
Emeric Bruna6a09982015-09-22 15:34:19 +0200451 use_identifier = 1;
Emeric Brun2b920a12010-09-23 18:30:22 +0200452 }
Emeric Brunb3971ab2015-05-12 18:49:09 +0200453
454 /* encode update identifier if needed */
455 if (use_identifier) {
Emeric Brun819fc6f2017-06-13 19:37:32 +0200456 netinteger = htonl(updateid);
Emeric Brunb3971ab2015-05-12 18:49:09 +0200457 memcpy(cursor, &netinteger, sizeof(netinteger));
458 cursor += sizeof(netinteger);
Emeric Brun2b920a12010-09-23 18:30:22 +0200459 }
460
Frédéric Lécaille523cc9e2016-10-12 17:30:30 +0200461 if (use_timed) {
462 netinteger = htonl(tick_remain(now_ms, ts->expire));
463 memcpy(cursor, &netinteger, sizeof(netinteger));
464 cursor += sizeof(netinteger);
465 }
466
Emeric Brunb3971ab2015-05-12 18:49:09 +0200467 /* encode the key */
Thierry FOURNIER5d24ebc2015-07-24 08:46:42 +0200468 if (st->table->type == SMP_T_STR) {
Emeric Brun2b920a12010-09-23 18:30:22 +0200469 int stlen = strlen((char *)ts->key.key);
470
Emeric Brunb3971ab2015-05-12 18:49:09 +0200471 intencode(stlen, &cursor);
472 memcpy(cursor, ts->key.key, stlen);
473 cursor += stlen;
Emeric Brun2b920a12010-09-23 18:30:22 +0200474 }
Thierry FOURNIER5d24ebc2015-07-24 08:46:42 +0200475 else if (st->table->type == SMP_T_SINT) {
Emeric Brun2b920a12010-09-23 18:30:22 +0200476 netinteger = htonl(*((uint32_t *)ts->key.key));
Emeric Brunb3971ab2015-05-12 18:49:09 +0200477 memcpy(cursor, &netinteger, sizeof(netinteger));
478 cursor += sizeof(netinteger);
Emeric Brun2b920a12010-09-23 18:30:22 +0200479 }
480 else {
Emeric Brunb3971ab2015-05-12 18:49:09 +0200481 memcpy(cursor, ts->key.key, st->table->key_size);
482 cursor += st->table->key_size;
Emeric Brun2b920a12010-09-23 18:30:22 +0200483 }
484
Christopher Faulet2a944ee2017-11-07 10:42:54 +0100485 HA_RWLOCK_RDLOCK(STK_SESS_LOCK, &ts->lock);
Emeric Brunb3971ab2015-05-12 18:49:09 +0200486 /* encode values */
Emeric Brun94900952015-06-11 18:25:54 +0200487 for (data_type = 0 ; data_type < STKTABLE_DATA_TYPES ; data_type++) {
Emeric Brunb3971ab2015-05-12 18:49:09 +0200488
Emeric Brun94900952015-06-11 18:25:54 +0200489 data_ptr = stktable_data_ptr(st->table, ts, data_type);
490 if (data_ptr) {
491 switch (stktable_data_types[data_type].std_type) {
492 case STD_T_SINT: {
493 int data;
494
495 data = stktable_data_cast(data_ptr, std_t_sint);
496 intencode(data, &cursor);
497 break;
498 }
499 case STD_T_UINT: {
500 unsigned int data;
501
502 data = stktable_data_cast(data_ptr, std_t_uint);
503 intencode(data, &cursor);
504 break;
505 }
506 case STD_T_ULL: {
507 unsigned long long data;
508
509 data = stktable_data_cast(data_ptr, std_t_ull);
510 intencode(data, &cursor);
511 break;
512 }
513 case STD_T_FRQP: {
514 struct freq_ctr_period *frqp;
515
516 frqp = &stktable_data_cast(data_ptr, std_t_frqp);
517 intencode((unsigned int)(now_ms - frqp->curr_tick), &cursor);
518 intencode(frqp->curr_ctr, &cursor);
519 intencode(frqp->prev_ctr, &cursor);
520 break;
521 }
Frédéric Lécaille8d78fa72019-05-20 18:22:52 +0200522 case STD_T_DICT: {
523 struct dict_entry *de;
524 struct dcache_tx_entry *cached_de;
525 struct dcache_tx_entry cde = {0};
526 char *beg, *end;
527 size_t value_len, data_len;
528 struct dcache *dc;
529
530 de = stktable_data_cast(data_ptr, std_t_dict);
531 if (!de)
532 break;
533
534 dc = peer->dcache;
535 cde.value.key = de;
536 cached_de = dcache_tx_insert(dc, &cde);
537 if (!cached_de)
538 break;
539
540 /* Leave enough room to encode the remaining data length. */
541 end = beg = cursor + PEER_MSG_ENC_LENGTH_MAXLEN;
542 /* Encode the dictionary entry key */
543 intencode(cached_de->key.key + 1, &end);
544 if (cached_de != &cde) {
545 /* Encode the length of the dictionary entry data */
546 value_len = strlen(de->value.key);
547 intencode(value_len, &end);
548 /* Copy the data */
549 memcpy(end, de->value.key, value_len);
550 end += value_len;
551 }
552 /* Encode the length of the data */
553 data_len = end - beg;
554 intencode(data_len, &cursor);
555 memmove(cursor, beg, data_len);
556 cursor += data_len;
557 break;
558 }
Emeric Brun94900952015-06-11 18:25:54 +0200559 }
560 }
Emeric Brunb3971ab2015-05-12 18:49:09 +0200561 }
Christopher Faulet2a944ee2017-11-07 10:42:54 +0100562 HA_RWLOCK_RDUNLOCK(STK_SESS_LOCK, &ts->lock);
Emeric Brunb3971ab2015-05-12 18:49:09 +0200563
564 /* Compute datalen */
565 datalen = (cursor - datamsg);
566
567 /* prepare message header */
568 msg[0] = PEER_MSG_CLASS_STICKTABLE;
Frédéric Lécaille523cc9e2016-10-12 17:30:30 +0200569 peer_set_update_msg_type(&msg[1], use_identifier, use_timed);
Emeric Brunb3971ab2015-05-12 18:49:09 +0200570 cursor = &msg[2];
571 intencode(datalen, &cursor);
572
573 /* move data after header */
574 memmove(cursor, datamsg, datalen);
575
576 /* return header size + data_len */
577 return (cursor - msg) + datalen;
578}
579
580/*
581 * This prepare the switch table message to targeted share table <st>.
Joseph Herlant82b2f542018-11-15 12:19:14 -0800582 * <msg> is a buffer of <size> to receive data message content
Emeric Brunb3971ab2015-05-12 18:49:09 +0200583 * If function returns 0, the caller should consider we were unable to encode this message (TODO:
584 * check size)
585 */
Frédéric Lécailled5fe14b2019-01-24 10:33:40 +0100586static int peer_prepare_switchmsg(char *msg, size_t size, struct peer_prep_params *params)
Emeric Brunb3971ab2015-05-12 18:49:09 +0200587{
588 int len;
589 unsigned short datalen;
Willy Tarreau83061a82018-07-13 11:56:34 +0200590 struct buffer *chunk;
Frédéric Lécaille37a72542017-07-06 15:02:16 +0200591 char *cursor, *datamsg, *chunkp, *chunkq;
Emeric Brunb3971ab2015-05-12 18:49:09 +0200592 uint64_t data = 0;
Emeric Brun94900952015-06-11 18:25:54 +0200593 unsigned int data_type;
Frédéric Lécailled5fe14b2019-01-24 10:33:40 +0100594 struct shared_table *st;
Emeric Brunb3971ab2015-05-12 18:49:09 +0200595
Frédéric Lécailled5fe14b2019-01-24 10:33:40 +0100596 st = params->swtch.shared_table;
Frédéric Lécaille39143342019-05-24 14:32:27 +0200597 cursor = datamsg = msg + PEER_MSG_HEADER_LEN + PEER_MSG_ENC_LENGTH_MAXLEN;
Emeric Brunb3971ab2015-05-12 18:49:09 +0200598
599 /* Encode data */
600
601 /* encode local id */
602 intencode(st->local_id, &cursor);
603
604 /* encode table name */
Frédéric Lécaille7fcc24d2019-03-20 15:09:45 +0100605 len = strlen(st->table->nid);
Emeric Brunb3971ab2015-05-12 18:49:09 +0200606 intencode(len, &cursor);
Frédéric Lécaille7fcc24d2019-03-20 15:09:45 +0100607 memcpy(cursor, st->table->nid, len);
Emeric Brunb3971ab2015-05-12 18:49:09 +0200608 cursor += len;
609
610 /* encode table type */
611
612 intencode(st->table->type, &cursor);
613
614 /* encode table key size */
615 intencode(st->table->key_size, &cursor);
616
Frédéric Lécaille37a72542017-07-06 15:02:16 +0200617 chunk = get_trash_chunk();
Willy Tarreau843b7cb2018-07-13 10:54:26 +0200618 chunkp = chunkq = chunk->area;
Emeric Brun94900952015-06-11 18:25:54 +0200619 /* encode available known data types in table */
620 for (data_type = 0 ; data_type < STKTABLE_DATA_TYPES ; data_type++) {
621 if (st->table->data_ofs[data_type]) {
622 switch (stktable_data_types[data_type].std_type) {
623 case STD_T_SINT:
624 case STD_T_UINT:
625 case STD_T_ULL:
Frédéric Lécaille8d78fa72019-05-20 18:22:52 +0200626 case STD_T_DICT:
Frédéric Lécaille37a72542017-07-06 15:02:16 +0200627 data |= 1 << data_type;
628 break;
Emeric Brun94900952015-06-11 18:25:54 +0200629 case STD_T_FRQP:
630 data |= 1 << data_type;
Frédéric Lécaille37a72542017-07-06 15:02:16 +0200631 intencode(data_type, &chunkq);
632 intencode(st->table->data_arg[data_type].u, &chunkq);
Emeric Brun94900952015-06-11 18:25:54 +0200633 break;
634 }
635 }
Emeric Brunb3971ab2015-05-12 18:49:09 +0200636 }
637 intencode(data, &cursor);
638
Frédéric Lécaille37a72542017-07-06 15:02:16 +0200639 /* Encode stick-table entries duration. */
640 intencode(st->table->expire, &cursor);
641
642 if (chunkq > chunkp) {
Willy Tarreau843b7cb2018-07-13 10:54:26 +0200643 chunk->data = chunkq - chunkp;
644 memcpy(cursor, chunk->area, chunk->data);
645 cursor += chunk->data;
Frédéric Lécaille37a72542017-07-06 15:02:16 +0200646 }
647
Emeric Brunb3971ab2015-05-12 18:49:09 +0200648 /* Compute datalen */
649 datalen = (cursor - datamsg);
Emeric Brun2b920a12010-09-23 18:30:22 +0200650
Emeric Brunb3971ab2015-05-12 18:49:09 +0200651 /* prepare message header */
652 msg[0] = PEER_MSG_CLASS_STICKTABLE;
653 msg[1] = PEER_MSG_STKT_DEFINE;
654 cursor = &msg[2];
655 intencode(datalen, &cursor);
Emeric Brun2b920a12010-09-23 18:30:22 +0200656
Emeric Brunb3971ab2015-05-12 18:49:09 +0200657 /* move data after header */
658 memmove(cursor, datamsg, datalen);
659
660 /* return header size + data_len */
661 return (cursor - msg) + datalen;
Emeric Brun2b920a12010-09-23 18:30:22 +0200662}
663
Emeric Brunb3971ab2015-05-12 18:49:09 +0200664/*
665 * This prepare the acknowledge message on the stick session <ts>, <st> is the considered
666 * stick table.
Joseph Herlant82b2f542018-11-15 12:19:14 -0800667 * <msg> is a buffer of <size> to receive data message content
Emeric Brunb3971ab2015-05-12 18:49:09 +0200668 * If function returns 0, the caller should consider we were unable to encode this message (TODO:
669 * check size)
670 */
Frédéric Lécailled5fe14b2019-01-24 10:33:40 +0100671static int peer_prepare_ackmsg(char *msg, size_t size, struct peer_prep_params *p)
Emeric Brunb3971ab2015-05-12 18:49:09 +0200672{
673 unsigned short datalen;
674 char *cursor, *datamsg;
675 uint32_t netinteger;
Frédéric Lécailled5fe14b2019-01-24 10:33:40 +0100676 struct shared_table *st;
Emeric Brunb3971ab2015-05-12 18:49:09 +0200677
Frédéric Lécaille39143342019-05-24 14:32:27 +0200678 cursor = datamsg = msg + PEER_MSG_HEADER_LEN + PEER_MSG_ENC_LENGTH_MAXLEN;
Emeric Brunb3971ab2015-05-12 18:49:09 +0200679
Frédéric Lécailled5fe14b2019-01-24 10:33:40 +0100680 st = p->ack.shared_table;
Emeric Brunb3971ab2015-05-12 18:49:09 +0200681 intencode(st->remote_id, &cursor);
682 netinteger = htonl(st->last_get);
683 memcpy(cursor, &netinteger, sizeof(netinteger));
684 cursor += sizeof(netinteger);
685
686 /* Compute datalen */
687 datalen = (cursor - datamsg);
688
689 /* prepare message header */
690 msg[0] = PEER_MSG_CLASS_STICKTABLE;
Emeric Brune1ab8082015-08-21 11:48:54 +0200691 msg[1] = PEER_MSG_STKT_ACK;
Emeric Brunb3971ab2015-05-12 18:49:09 +0200692 cursor = &msg[2];
693 intencode(datalen, &cursor);
694
695 /* move data after header */
696 memmove(cursor, datamsg, datalen);
697
698 /* return header size + data_len */
699 return (cursor - msg) + datalen;
700}
Emeric Brun2b920a12010-09-23 18:30:22 +0200701
702/*
Emeric Brun9ef2ad72019-04-02 17:22:01 +0200703 * Function to deinit connected peer
704 */
705void __peer_session_deinit(struct peer *peer)
706{
707 struct stream_interface *si;
708 struct stream *s;
709 struct peers *peers;
710
711 if (!peer->appctx)
712 return;
713
714 si = peer->appctx->owner;
715 if (!si)
716 return;
717
718 s = si_strm(si);
719 if (!s)
720 return;
721
722 peers = strm_fe(s)->parent;
723 if (!peers)
724 return;
725
726 if (peer->appctx->st0 == PEER_SESS_ST_WAITMSG)
727 HA_ATOMIC_SUB(&connected_peers, 1);
728
729 HA_ATOMIC_SUB(&active_peers, 1);
730
Frédéric Lécaille8d78fa72019-05-20 18:22:52 +0200731 flush_dcache(peer);
732
Emeric Brun9ef2ad72019-04-02 17:22:01 +0200733 /* Re-init current table pointers to force announcement on re-connect */
734 peer->remote_table = peer->last_local_table = NULL;
735 peer->appctx = NULL;
736 if (peer->flags & PEER_F_LEARN_ASSIGN) {
737 /* unassign current peer for learning */
738 peer->flags &= ~(PEER_F_LEARN_ASSIGN);
739 peers->flags &= ~(PEERS_F_RESYNC_ASSIGN|PEERS_F_RESYNC_PROCESS);
740
741 /* reschedule a resync */
742 peers->resync_timeout = tick_add(now_ms, MS_TO_TICKS(5000));
743 }
744 /* reset teaching and learning flags to 0 */
745 peer->flags &= PEER_TEACH_RESET;
746 peer->flags &= PEER_LEARN_RESET;
747 task_wakeup(peers->sync_task, TASK_WOKEN_MSG);
748}
749
750/*
Emeric Brun2b920a12010-09-23 18:30:22 +0200751 * Callback to release a session with a peer
752 */
Willy Tarreau00a37f02015-04-13 12:05:19 +0200753static void peer_session_release(struct appctx *appctx)
Emeric Brun2b920a12010-09-23 18:30:22 +0200754{
Vincent Bernat3c2f2f22016-04-03 13:48:42 +0200755 struct peer *peer = appctx->ctx.peers.ptr;
Emeric Brun2b920a12010-09-23 18:30:22 +0200756
Willy Tarreau7b4b4992013-12-01 09:15:12 +0100757 /* appctx->ctx.peers.ptr is not a peer session */
Willy Tarreaue4d927a2013-12-01 12:47:35 +0100758 if (appctx->st0 < PEER_SESS_ST_SENDSUCCESS)
Emeric Brun2b920a12010-09-23 18:30:22 +0200759 return;
760
761 /* peer session identified */
Emeric Brunb3971ab2015-05-12 18:49:09 +0200762 if (peer) {
Christopher Faulet2a944ee2017-11-07 10:42:54 +0100763 HA_SPIN_LOCK(PEER_LOCK, &peer->lock);
Emeric Brun9ef2ad72019-04-02 17:22:01 +0200764 if (peer->appctx == appctx)
765 __peer_session_deinit(peer);
Christopher Faulet2a944ee2017-11-07 10:42:54 +0100766 HA_SPIN_UNLOCK(PEER_LOCK, &peer->lock);
Emeric Brun2b920a12010-09-23 18:30:22 +0200767 }
768}
769
Frédéric Lécaille523cc9e2016-10-12 17:30:30 +0200770/* Retrieve the major and minor versions of peers protocol
771 * announced by a remote peer. <str> is a null-terminated
772 * string with the following format: "<maj_ver>.<min_ver>".
773 */
774static int peer_get_version(const char *str,
775 unsigned int *maj_ver, unsigned int *min_ver)
776{
777 unsigned int majv, minv;
778 const char *pos, *saved;
779 const char *end;
780
781 saved = pos = str;
782 end = str + strlen(str);
783
784 majv = read_uint(&pos, end);
785 if (saved == pos || *pos++ != '.')
786 return -1;
787
788 saved = pos;
789 minv = read_uint(&pos, end);
790 if (saved == pos || pos != end)
791 return -1;
792
793 *maj_ver = majv;
794 *min_ver = minv;
795
796 return 0;
797}
Emeric Brun2b920a12010-09-23 18:30:22 +0200798
799/*
Frédéric Lécaillece025572019-01-21 13:38:06 +0100800 * Parse a line terminated by an optional '\r' character, followed by a mandatory
801 * '\n' character.
802 * Returns 1 if succeeded or 0 if a '\n' character could not be found, and -1 if
803 * a line could not be read because the communication channel is closed.
804 */
805static inline int peer_getline(struct appctx *appctx)
806{
807 int n;
808 struct stream_interface *si = appctx->owner;
809
810 n = co_getline(si_oc(si), trash.area, trash.size);
811 if (!n)
812 return 0;
813
814 if (n < 0 || trash.area[n - 1] != '\n') {
815 appctx->st0 = PEER_SESS_ST_END;
816 return -1;
817 }
818
819 if (n > 1 && (trash.area[n - 2] == '\r'))
820 trash.area[n - 2] = 0;
821 else
822 trash.area[n - 1] = 0;
823
824 co_skip(si_oc(si), n);
825
826 return n;
827}
828
829/*
Frédéric Lécailleec44ea82019-01-22 15:54:53 +0100830 * Send a message after having called <peer_prepare_msg> to build it.
831 * Return 0 if the message could not be built modifying the appcxt st0 to PEER_SESS_ST_END value.
832 * Returns -1 if there was not enough room left to send the message,
833 * any other negative returned value must be considered as an error with an appcxt st0
834 * returned value equal to PEER_SESS_ST_END.
835 */
Frédéric Lécailled5fe14b2019-01-24 10:33:40 +0100836static inline int peer_send_msg(struct appctx *appctx,
837 int (*peer_prepare_msg)(char *, size_t, struct peer_prep_params *),
838 struct peer_prep_params *params)
Frédéric Lécailleec44ea82019-01-22 15:54:53 +0100839{
840 int ret, msglen;
841 struct stream_interface *si = appctx->owner;
842
Frédéric Lécailled5fe14b2019-01-24 10:33:40 +0100843 msglen = peer_prepare_msg(trash.area, trash.size, params);
Frédéric Lécailleec44ea82019-01-22 15:54:53 +0100844 if (!msglen) {
845 /* internal error: message does not fit in trash */
846 appctx->st0 = PEER_SESS_ST_END;
847 return 0;
848 }
849
850 /* message to buffer */
851 ret = ci_putblk(si_ic(si), trash.area, msglen);
852 if (ret <= 0) {
853 if (ret == -1) {
854 /* No more write possible */
855 si_rx_room_blk(si);
856 return -1;
857 }
858 appctx->st0 = PEER_SESS_ST_END;
859 }
860
861 return ret;
862}
863
864/*
Frédéric Lécaille7d0ceee2019-01-24 14:24:05 +0100865 * Send a hello message.
866 * Return 0 if the message could not be built modifying the appcxt st0 to PEER_SESS_ST_END value.
867 * Returns -1 if there was not enough room left to send the message,
868 * any other negative returned value must be considered as an error with an appcxt st0
869 * returned value equal to PEER_SESS_ST_END.
870 */
871static inline int peer_send_hellomsg(struct appctx *appctx, struct peer *peer)
872{
873 struct peer_prep_params p = {
874 .hello.peer = peer,
875 };
876
877 return peer_send_msg(appctx, peer_prepare_hellomsg, &p);
878}
879
880/*
881 * Send a success peer handshake status message.
882 * Return 0 if the message could not be built modifying the appcxt st0 to PEER_SESS_ST_END value.
883 * Returns -1 if there was not enough room left to send the message,
884 * any other negative returned value must be considered as an error with an appcxt st0
885 * returned value equal to PEER_SESS_ST_END.
886 */
887static inline int peer_send_status_successmsg(struct appctx *appctx)
888{
889 return peer_send_msg(appctx, peer_prepare_status_successmsg, NULL);
890}
891
892/*
893 * Send a peer handshake status error message.
894 * Return 0 if the message could not be built modifying the appcxt st0 to PEER_SESS_ST_END value.
895 * Returns -1 if there was not enough room left to send the message,
896 * any other negative returned value must be considered as an error with an appcxt st0
897 * returned value equal to PEER_SESS_ST_END.
898 */
899static inline int peer_send_status_errormsg(struct appctx *appctx)
900{
901 struct peer_prep_params p = {
902 .error_status.st1 = appctx->st1,
903 };
904
905 return peer_send_msg(appctx, peer_prepare_status_errormsg, &p);
906}
907
908/*
Frédéric Lécailleec44ea82019-01-22 15:54:53 +0100909 * Send a stick-table switch message.
910 * Return 0 if the message could not be built modifying the appcxt st0 to PEER_SESS_ST_END value.
911 * Returns -1 if there was not enough room left to send the message,
912 * any other negative returned value must be considered as an error with an appcxt st0
913 * returned value equal to PEER_SESS_ST_END.
914 */
915static inline int peer_send_switchmsg(struct shared_table *st, struct appctx *appctx)
916{
Frédéric Lécailled5fe14b2019-01-24 10:33:40 +0100917 struct peer_prep_params p = {
918 .swtch.shared_table = st,
919 };
920
921 return peer_send_msg(appctx, peer_prepare_switchmsg, &p);
Frédéric Lécailleec44ea82019-01-22 15:54:53 +0100922}
923
924/*
925 * Send a stick-table update acknowledgement message.
926 * Return 0 if the message could not be built modifying the appcxt st0 to PEER_SESS_ST_END value.
927 * Returns -1 if there was not enough room left to send the message,
928 * any other negative returned value must be considered as an error with an appcxt st0
929 * returned value equal to PEER_SESS_ST_END.
930 */
931static inline int peer_send_ackmsg(struct shared_table *st, struct appctx *appctx)
932{
Frédéric Lécailled5fe14b2019-01-24 10:33:40 +0100933 struct peer_prep_params p = {
934 .ack.shared_table = st,
935 };
936
937 return peer_send_msg(appctx, peer_prepare_ackmsg, &p);
Frédéric Lécailleec44ea82019-01-22 15:54:53 +0100938}
939
940/*
Frédéric Lécaille87f554c2019-01-22 17:26:50 +0100941 * Send a stick-table update message.
942 * Return 0 if the message could not be built modifying the appcxt st0 to PEER_SESS_ST_END value.
943 * Returns -1 if there was not enough room left to send the message,
944 * any other negative returned value must be considered as an error with an appcxt st0
945 * returned value equal to PEER_SESS_ST_END.
946 */
947static inline int peer_send_updatemsg(struct shared_table *st, struct appctx *appctx, struct stksess *ts,
948 unsigned int updateid, int use_identifier, int use_timed)
949{
Frédéric Lécailled5fe14b2019-01-24 10:33:40 +0100950 struct peer_prep_params p = {
951 .updt.stksess = ts,
952 .updt.shared_table = st,
953 .updt.updateid = updateid,
954 .updt.use_identifier = use_identifier,
955 .updt.use_timed = use_timed,
Frédéric Lécaille8d78fa72019-05-20 18:22:52 +0200956 .updt.peer = appctx->ctx.peers.ptr,
Frédéric Lécailled5fe14b2019-01-24 10:33:40 +0100957 };
958
959 return peer_send_msg(appctx, peer_prepare_updatemsg, &p);
Frédéric Lécaille87f554c2019-01-22 17:26:50 +0100960}
961
Frédéric Lécaille7d0ceee2019-01-24 14:24:05 +0100962/*
963 * Build a peer protocol control class message.
964 * Returns the number of written bytes used to build the message if succeeded,
965 * 0 if not.
966 */
967static int peer_prepare_control_msg(char *msg, size_t size, struct peer_prep_params *p)
968{
969 if (size < sizeof p->control.head)
970 return 0;
971
972 msg[0] = p->control.head[0];
973 msg[1] = p->control.head[1];
974
975 return 2;
976}
Frédéric Lécaille6a8303d2019-01-22 22:25:17 +0100977
978/*
Frédéric Lécaille7d0ceee2019-01-24 14:24:05 +0100979 * Send a stick-table synchronization request message.
980 * Return 0 if the message could not be built modifying the appcxt st0 to PEER_SESS_ST_END value.
981 * Returns -1 if there was not enough room left to send the message,
982 * any other negative returned value must be considered as an error with an appctx st0
983 * returned value equal to PEER_SESS_ST_END.
984 */
985static inline int peer_send_resync_reqmsg(struct appctx *appctx)
986{
987 struct peer_prep_params p = {
988 .control.head = { PEER_MSG_CLASS_CONTROL, PEER_MSG_CTRL_RESYNCREQ, },
989 };
990
991 return peer_send_msg(appctx, peer_prepare_control_msg, &p);
992}
993
994/*
995 * Send a stick-table synchronization confirmation message.
996 * Return 0 if the message could not be built modifying the appcxt st0 to PEER_SESS_ST_END value.
997 * Returns -1 if there was not enough room left to send the message,
998 * any other negative returned value must be considered as an error with an appctx st0
999 * returned value equal to PEER_SESS_ST_END.
1000 */
1001static inline int peer_send_resync_confirmsg(struct appctx *appctx)
1002{
1003 struct peer_prep_params p = {
1004 .control.head = { PEER_MSG_CLASS_CONTROL, PEER_MSG_CTRL_RESYNCCONFIRM, },
1005 };
1006
1007 return peer_send_msg(appctx, peer_prepare_control_msg, &p);
1008}
1009
1010/*
1011 * Send a stick-table synchronization finished message.
1012 * Return 0 if the message could not be built modifying the appcxt st0 to PEER_SESS_ST_END value.
1013 * Returns -1 if there was not enough room left to send the message,
1014 * any other negative returned value must be considered as an error with an appctx st0
1015 * returned value equal to PEER_SESS_ST_END.
1016 */
1017static inline int peer_send_resync_finishedmsg(struct appctx *appctx, struct peer *peer)
1018{
1019 struct peer_prep_params p = {
1020 .control.head = { PEER_MSG_CLASS_CONTROL, },
1021 };
1022
1023 p.control.head[1] = (peer->flags & PEERS_RESYNC_STATEMASK) == PEERS_RESYNC_FINISHED ?
1024 PEER_MSG_CTRL_RESYNCFINISHED : PEER_MSG_CTRL_RESYNCPARTIAL;
1025
1026 return peer_send_msg(appctx, peer_prepare_control_msg, &p);
1027}
1028
1029/*
Frédéric Lécaille645635d2019-02-11 17:49:39 +01001030 * Send a heartbeat message.
1031 * Return 0 if the message could not be built modifying the appctx st0 to PEER_SESS_ST_END value.
1032 * Returns -1 if there was not enough room left to send the message,
1033 * any other negative returned value must be considered as an error with an appctx st0
1034 * returned value equal to PEER_SESS_ST_END.
1035 */
1036static inline int peer_send_heartbeatmsg(struct appctx *appctx)
1037{
1038 struct peer_prep_params p = {
1039 .control.head = { PEER_MSG_CLASS_CONTROL, PEER_MSG_CTRL_HEARTBEAT, },
1040 };
1041
1042 return peer_send_msg(appctx, peer_prepare_control_msg, &p);
1043}
1044
1045/*
Frédéric Lécaille7d0ceee2019-01-24 14:24:05 +01001046 * Build a peer protocol error class message.
1047 * Returns the number of written bytes used to build the message if succeeded,
1048 * 0 if not.
1049 */
1050static int peer_prepare_error_msg(char *msg, size_t size, struct peer_prep_params *p)
1051{
1052 if (size < sizeof p->error.head)
1053 return 0;
1054
1055 msg[0] = p->error.head[0];
1056 msg[1] = p->error.head[1];
1057
1058 return 2;
1059}
1060
1061/*
1062 * Send a "size limit reached" error message.
1063 * Return 0 if the message could not be built modifying the appcxt st0 to PEER_SESS_ST_END value.
1064 * Returns -1 if there was not enough room left to send the message,
1065 * any other negative returned value must be considered as an error with an appctx st0
1066 * returned value equal to PEER_SESS_ST_END.
1067 */
1068static inline int peer_send_error_size_limitmsg(struct appctx *appctx)
1069{
1070 struct peer_prep_params p = {
1071 .error.head = { PEER_MSG_CLASS_ERROR, PEER_MSG_ERR_SIZELIMIT, },
1072 };
1073
1074 return peer_send_msg(appctx, peer_prepare_error_msg, &p);
1075}
1076
1077/*
1078 * Send a "peer protocol" error message.
1079 * Return 0 if the message could not be built modifying the appcxt st0 to PEER_SESS_ST_END value.
1080 * Returns -1 if there was not enough room left to send the message,
1081 * any other negative returned value must be considered as an error with an appctx st0
1082 * returned value equal to PEER_SESS_ST_END.
1083 */
1084static inline int peer_send_error_protomsg(struct appctx *appctx)
1085{
1086 struct peer_prep_params p = {
1087 .error.head = { PEER_MSG_CLASS_ERROR, PEER_MSG_ERR_PROTOCOL, },
1088 };
1089
1090 return peer_send_msg(appctx, peer_prepare_error_msg, &p);
1091}
1092
1093/*
Frédéric Lécaille6a8303d2019-01-22 22:25:17 +01001094 * Function used to lookup for recent stick-table updates associated with
1095 * <st> shared stick-table when a lesson must be taught a peer (PEER_F_LEARN_ASSIGN flag set).
1096 */
1097static inline struct stksess *peer_teach_process_stksess_lookup(struct shared_table *st)
1098{
1099 struct eb32_node *eb;
1100
1101 eb = eb32_lookup_ge(&st->table->updates, st->last_pushed+1);
1102 if (!eb) {
1103 eb = eb32_first(&st->table->updates);
1104 if (!eb || ((int)(eb->key - st->last_pushed) <= 0)) {
1105 st->table->commitupdate = st->last_pushed = st->table->localupdate;
1106 return NULL;
1107 }
1108 }
1109
1110 if ((int)(eb->key - st->table->localupdate) > 0) {
1111 st->table->commitupdate = st->last_pushed = st->table->localupdate;
1112 return NULL;
1113 }
1114
1115 return eb32_entry(eb, struct stksess, upd);
1116}
1117
1118/*
1119 * Function used to lookup for recent stick-table updates associated with
1120 * <st> shared stick-table during teach state 1 step.
1121 */
1122static inline struct stksess *peer_teach_stage1_stksess_lookup(struct shared_table *st)
1123{
1124 struct eb32_node *eb;
1125
1126 eb = eb32_lookup_ge(&st->table->updates, st->last_pushed+1);
1127 if (!eb) {
1128 st->flags |= SHTABLE_F_TEACH_STAGE1;
1129 eb = eb32_first(&st->table->updates);
1130 if (eb)
1131 st->last_pushed = eb->key - 1;
1132 return NULL;
1133 }
1134
1135 return eb32_entry(eb, struct stksess, upd);
1136}
1137
1138/*
1139 * Function used to lookup for recent stick-table updates associated with
1140 * <st> shared stick-table during teach state 2 step.
1141 */
1142static inline struct stksess *peer_teach_stage2_stksess_lookup(struct shared_table *st)
1143{
1144 struct eb32_node *eb;
1145
1146 eb = eb32_lookup_ge(&st->table->updates, st->last_pushed+1);
1147 if (!eb || eb->key > st->teaching_origin) {
1148 st->flags |= SHTABLE_F_TEACH_STAGE2;
1149 return NULL;
1150 }
1151
1152 return eb32_entry(eb, struct stksess, upd);
1153}
1154
1155/*
1156 * Generic function to emit update messages for <st> stick-table when a lesson must
1157 * be taught to the peer <p>.
1158 * <locked> must be set to 1 if the shared table <st> is already locked when entering
1159 * this function, 0 if not.
1160 *
1161 * This function temporary unlock/lock <st> when it sends stick-table updates or
1162 * when decrementing its refcount in case of any error when it sends this updates.
1163 *
1164 * Return 0 if any message could not be built modifying the appcxt st0 to PEER_SESS_ST_END value.
1165 * Returns -1 if there was not enough room left to send the message,
1166 * any other negative returned value must be considered as an error with an appcxt st0
1167 * returned value equal to PEER_SESS_ST_END.
1168 * If it returns 0 or -1, this function leave <st> locked if already locked when entering this function
1169 * unlocked if not already locked when entering this function.
1170 */
1171static inline int peer_send_teachmsgs(struct appctx *appctx, struct peer *p,
1172 struct stksess *(*peer_stksess_lookup)(struct shared_table *),
1173 struct shared_table *st, int locked)
1174{
1175 int ret, new_pushed, use_timed;
1176
1177 ret = 1;
1178 use_timed = 0;
1179 if (st != p->last_local_table) {
1180 ret = peer_send_switchmsg(st, appctx);
1181 if (ret <= 0)
1182 return ret;
1183
1184 p->last_local_table = st;
1185 }
1186
1187 if (peer_stksess_lookup != peer_teach_process_stksess_lookup)
1188 use_timed = !(p->flags & PEER_F_DWNGRD);
1189
1190 /* We force new pushed to 1 to force identifier in update message */
1191 new_pushed = 1;
1192
1193 if (!locked)
1194 HA_SPIN_LOCK(STK_TABLE_LOCK, &st->table->lock);
1195
1196 while (1) {
1197 struct stksess *ts;
1198 unsigned updateid;
1199
1200 /* push local updates */
1201 ts = peer_stksess_lookup(st);
1202 if (!ts)
1203 break;
1204
1205 updateid = ts->upd.key;
1206 ts->ref_cnt++;
1207 HA_SPIN_UNLOCK(STK_TABLE_LOCK, &st->table->lock);
1208
1209 ret = peer_send_updatemsg(st, appctx, ts, updateid, new_pushed, use_timed);
1210 if (ret <= 0) {
1211 HA_SPIN_LOCK(STK_TABLE_LOCK, &st->table->lock);
1212 ts->ref_cnt--;
1213 if (!locked)
1214 HA_SPIN_UNLOCK(STK_TABLE_LOCK, &st->table->lock);
1215 return ret;
1216 }
1217
1218 HA_SPIN_LOCK(STK_TABLE_LOCK, &st->table->lock);
1219 ts->ref_cnt--;
1220 st->last_pushed = updateid;
1221
1222 if (peer_stksess_lookup == peer_teach_process_stksess_lookup &&
1223 (int)(st->last_pushed - st->table->commitupdate) > 0)
1224 st->table->commitupdate = st->last_pushed;
1225
1226 /* identifier may not needed in next update message */
1227 new_pushed = 0;
1228 }
1229
1230 out:
1231 if (!locked)
1232 HA_SPIN_UNLOCK(STK_TABLE_LOCK, &st->table->lock);
1233 return 1;
1234}
1235
1236/*
1237 * Function to emit update messages for <st> stick-table when a lesson must
1238 * be taught to the peer <p> (PEER_F_LEARN_ASSIGN flag set).
1239 *
1240 * Note that <st> shared stick-table is locked when calling this function.
1241 *
1242 * Return 0 if any 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_teach_process_msgs(struct appctx *appctx, struct peer *p,
1248 struct shared_table *st)
1249{
1250 return peer_send_teachmsgs(appctx, p, peer_teach_process_stksess_lookup, st, 1);
1251}
1252
1253/*
1254 * Function to emit update messages for <st> stick-table when a lesson must
1255 * be taught to the peer <p> during teach state 1 step.
1256 *
1257 * Return 0 if any message could not be built modifying the appcxt st0 to PEER_SESS_ST_END value.
1258 * Returns -1 if there was not enough room left to send the message,
1259 * any other negative returned value must be considered as an error with an appcxt st0
1260 * returned value equal to PEER_SESS_ST_END.
1261 */
1262static inline int peer_send_teach_stage1_msgs(struct appctx *appctx, struct peer *p,
1263 struct shared_table *st)
1264{
1265 return peer_send_teachmsgs(appctx, p, peer_teach_stage1_stksess_lookup, st, 0);
1266}
1267
1268/*
1269 * Function to emit update messages for <st> stick-table when a lesson must
1270 * be taught to the peer <p> during teach state 1 step.
1271 *
1272 * Return 0 if any message could not be built modifying the appcxt st0 to PEER_SESS_ST_END value.
1273 * Returns -1 if there was not enough room left to send the message,
1274 * any other negative returned value must be considered as an error with an appcxt st0
1275 * returned value equal to PEER_SESS_ST_END.
1276 */
1277static inline int peer_send_teach_stage2_msgs(struct appctx *appctx, struct peer *p,
1278 struct shared_table *st)
1279{
1280 return peer_send_teachmsgs(appctx, p, peer_teach_stage2_stksess_lookup, st, 0);
1281}
1282
Frédéric Lécaille168a34b2019-01-23 11:16:57 +01001283
1284/*
1285 * Function used to parse a stick-table update message after it has been received
1286 * by <p> peer with <msg_cur> as address of the pointer to the position in the
1287 * receipt buffer with <msg_end> being position of the end of the stick-table message.
1288 * Update <msg_curr> accordingly to the peer protocol specs if no peer protocol error
1289 * was encountered.
1290 * <exp> must be set if the stick-table entry expires.
1291 * <updt> must be set for PEER_MSG_STKT_UPDATE or PEER_MSG_STKT_UPDATE_TIMED stick-table
1292 * messages, in this case the stick-table udpate message is received with a stick-table
1293 * update ID.
1294 * <totl> is the length of the stick-table update message computed upon receipt.
1295 */
Frédéric Lécaille444243c2019-01-24 15:40:11 +01001296static int peer_treat_updatemsg(struct appctx *appctx, struct peer *p, int updt, int exp,
1297 char **msg_cur, char *msg_end, int msg_len, int totl)
Frédéric Lécaille168a34b2019-01-23 11:16:57 +01001298{
1299 struct stream_interface *si = appctx->owner;
1300 struct shared_table *st = p->remote_table;
1301 struct stksess *ts, *newts;
1302 uint32_t update;
1303 int expire;
1304 unsigned int data_type;
1305 void *data_ptr;
1306
1307 /* Here we have data message */
1308 if (!st)
1309 goto ignore_msg;
1310
1311 expire = MS_TO_TICKS(st->table->expire);
1312
1313 if (updt) {
Willy Tarreau1e82a142019-01-29 11:08:06 +01001314 if (msg_len < sizeof(update))
1315 goto malformed_exit;
1316
Frédéric Lécaille168a34b2019-01-23 11:16:57 +01001317 memcpy(&update, *msg_cur, sizeof(update));
1318 *msg_cur += sizeof(update);
1319 st->last_get = htonl(update);
1320 }
1321 else {
1322 st->last_get++;
1323 }
1324
1325 if (exp) {
1326 size_t expire_sz = sizeof expire;
1327
Willy Tarreau1e82a142019-01-29 11:08:06 +01001328 if (*msg_cur + expire_sz > msg_end)
1329 goto malformed_exit;
1330
Frédéric Lécaille168a34b2019-01-23 11:16:57 +01001331 memcpy(&expire, *msg_cur, expire_sz);
1332 *msg_cur += expire_sz;
1333 expire = ntohl(expire);
1334 }
1335
1336 newts = stksess_new(st->table, NULL);
1337 if (!newts)
1338 goto ignore_msg;
1339
1340 if (st->table->type == SMP_T_STR) {
1341 unsigned int to_read, to_store;
1342
1343 to_read = intdecode(msg_cur, msg_end);
Willy Tarreau1e82a142019-01-29 11:08:06 +01001344 if (!*msg_cur)
1345 goto malformed_free_newts;
1346
Frédéric Lécaille168a34b2019-01-23 11:16:57 +01001347 to_store = MIN(to_read, st->table->key_size - 1);
Willy Tarreau1e82a142019-01-29 11:08:06 +01001348 if (*msg_cur + to_store > msg_end)
1349 goto malformed_free_newts;
Frédéric Lécaille168a34b2019-01-23 11:16:57 +01001350
1351 memcpy(newts->key.key, *msg_cur, to_store);
1352 newts->key.key[to_store] = 0;
1353 *msg_cur += to_read;
1354 }
1355 else if (st->table->type == SMP_T_SINT) {
1356 unsigned int netinteger;
1357
Willy Tarreau1e82a142019-01-29 11:08:06 +01001358 if (*msg_cur + sizeof(netinteger) > msg_end)
1359 goto malformed_free_newts;
1360
Frédéric Lécaille168a34b2019-01-23 11:16:57 +01001361 memcpy(&netinteger, *msg_cur, sizeof(netinteger));
1362 netinteger = ntohl(netinteger);
1363 memcpy(newts->key.key, &netinteger, sizeof(netinteger));
1364 *msg_cur += sizeof(netinteger);
1365 }
1366 else {
Willy Tarreau1e82a142019-01-29 11:08:06 +01001367 if (*msg_cur + st->table->key_size > msg_end)
1368 goto malformed_free_newts;
1369
Frédéric Lécaille168a34b2019-01-23 11:16:57 +01001370 memcpy(newts->key.key, *msg_cur, st->table->key_size);
1371 *msg_cur += st->table->key_size;
1372 }
1373
1374 /* lookup for existing entry */
1375 ts = stktable_set_entry(st->table, newts);
1376 if (ts != newts) {
1377 stksess_free(st->table, newts);
1378 newts = NULL;
1379 }
1380
1381 HA_RWLOCK_WRLOCK(STK_SESS_LOCK, &ts->lock);
1382
1383 for (data_type = 0 ; data_type < STKTABLE_DATA_TYPES ; data_type++) {
Willy Tarreau1e82a142019-01-29 11:08:06 +01001384 uint64_t decoded_int;
Frédéric Lécaille168a34b2019-01-23 11:16:57 +01001385
1386 if (!((1 << data_type) & st->remote_data))
1387 continue;
1388
Willy Tarreau1e82a142019-01-29 11:08:06 +01001389 decoded_int = intdecode(msg_cur, msg_end);
1390 if (!*msg_cur)
1391 goto malformed_unlock;
Frédéric Lécaille168a34b2019-01-23 11:16:57 +01001392
Willy Tarreau1e82a142019-01-29 11:08:06 +01001393 switch (stktable_data_types[data_type].std_type) {
1394 case STD_T_SINT:
Frédéric Lécaille168a34b2019-01-23 11:16:57 +01001395 data_ptr = stktable_data_ptr(st->table, ts, data_type);
1396 if (data_ptr)
Willy Tarreau1e82a142019-01-29 11:08:06 +01001397 stktable_data_cast(data_ptr, std_t_sint) = decoded_int;
Frédéric Lécaille168a34b2019-01-23 11:16:57 +01001398 break;
Frédéric Lécaille168a34b2019-01-23 11:16:57 +01001399
Willy Tarreau1e82a142019-01-29 11:08:06 +01001400 case STD_T_UINT:
Frédéric Lécaille168a34b2019-01-23 11:16:57 +01001401 data_ptr = stktable_data_ptr(st->table, ts, data_type);
1402 if (data_ptr)
Willy Tarreau1e82a142019-01-29 11:08:06 +01001403 stktable_data_cast(data_ptr, std_t_uint) = decoded_int;
Frédéric Lécaille168a34b2019-01-23 11:16:57 +01001404 break;
Frédéric Lécaille168a34b2019-01-23 11:16:57 +01001405
Willy Tarreau1e82a142019-01-29 11:08:06 +01001406 case STD_T_ULL:
Frédéric Lécaille168a34b2019-01-23 11:16:57 +01001407 data_ptr = stktable_data_ptr(st->table, ts, data_type);
1408 if (data_ptr)
Willy Tarreau1e82a142019-01-29 11:08:06 +01001409 stktable_data_cast(data_ptr, std_t_ull) = decoded_int;
Frédéric Lécaille168a34b2019-01-23 11:16:57 +01001410 break;
Willy Tarreau1e82a142019-01-29 11:08:06 +01001411
Frédéric Lécaille168a34b2019-01-23 11:16:57 +01001412 case STD_T_FRQP: {
1413 struct freq_ctr_period data;
1414
1415 /* First bit is reserved for the freq_ctr_period lock
1416 Note: here we're still protected by the stksess lock
1417 so we don't need to update the update the freq_ctr_period
1418 using its internal lock */
1419
Willy Tarreau1e82a142019-01-29 11:08:06 +01001420 data.curr_tick = tick_add(now_ms, -decoded_int) & ~0x1;
Frédéric Lécaille168a34b2019-01-23 11:16:57 +01001421 data.curr_ctr = intdecode(msg_cur, msg_end);
Willy Tarreau1e82a142019-01-29 11:08:06 +01001422 if (!*msg_cur)
1423 goto malformed_unlock;
1424
Frédéric Lécaille168a34b2019-01-23 11:16:57 +01001425 data.prev_ctr = intdecode(msg_cur, msg_end);
Willy Tarreau1e82a142019-01-29 11:08:06 +01001426 if (!*msg_cur)
1427 goto malformed_unlock;
Frédéric Lécaille168a34b2019-01-23 11:16:57 +01001428
1429 data_ptr = stktable_data_ptr(st->table, ts, data_type);
1430 if (data_ptr)
1431 stktable_data_cast(data_ptr, std_t_frqp) = data;
1432 break;
1433 }
Frédéric Lécaille8d78fa72019-05-20 18:22:52 +02001434 case STD_T_DICT: {
1435 struct buffer *chunk;
1436 size_t data_len, value_len;
1437 unsigned int id;
1438 struct dict_entry *de;
1439 struct dcache *dc;
1440
1441 data_len = decoded_int;
1442 if (*msg_cur + data_len > msg_end)
1443 goto malformed_unlock;
1444
1445 id = intdecode(msg_cur, msg_end);
1446 if (!*msg_cur || !id)
1447 goto malformed_unlock;
1448
1449 dc = p->dcache;
1450 if (*msg_cur == msg_end) {
1451 /* Dictionary entry key without value. */
1452 if (id > dc->max_entries)
1453 break;
1454 /* IDs sent over the network are numbered from 1. */
1455 de = dc->rx[id - 1].de;
1456 }
1457 else {
1458 chunk = get_trash_chunk();
1459 value_len = intdecode(msg_cur, msg_end);
1460 if (!*msg_cur || *msg_cur + value_len > msg_end ||
1461 unlikely(value_len + 1 >= chunk->size))
1462 goto malformed_unlock;
1463
1464 chunk_memcpy(chunk, *msg_cur, value_len);
1465 chunk->area[chunk->data] = '\0';
1466 de = dict_insert(&server_name_dict, chunk->area);
1467 dc->rx[id - 1].de = de;
1468 }
1469 if (de) {
1470 data_ptr = stktable_data_ptr(st->table, ts, data_type);
1471 if (data_ptr)
1472 stktable_data_cast(data_ptr, std_t_dict) = de;
1473 }
1474 break;
1475 }
Frédéric Lécaille168a34b2019-01-23 11:16:57 +01001476 }
1477 }
1478 /* Force new expiration */
1479 ts->expire = tick_add(now_ms, expire);
1480
1481 HA_RWLOCK_WRUNLOCK(STK_SESS_LOCK, &ts->lock);
1482 stktable_touch_remote(st->table, ts, 1);
1483 return 1;
1484
1485 ignore_msg:
1486 /* skip consumed message */
1487 co_skip(si_oc(si), totl);
1488 return 0;
Willy Tarreau1e82a142019-01-29 11:08:06 +01001489
1490 malformed_unlock:
1491 /* malformed message */
1492 HA_RWLOCK_WRUNLOCK(STK_SESS_LOCK, &ts->lock);
1493 stktable_touch_remote(st->table, ts, 1);
1494 appctx->st0 = PEER_SESS_ST_ERRPROTO;
1495 return 0;
1496
1497 malformed_free_newts:
1498 /* malformed message */
1499 stksess_free(st->table, newts);
1500 malformed_exit:
1501 appctx->st0 = PEER_SESS_ST_ERRPROTO;
1502 return 0;
Frédéric Lécaille168a34b2019-01-23 11:16:57 +01001503}
1504
Frédéric Lécaille87f554c2019-01-22 17:26:50 +01001505/*
Frédéric Lécailled27b0942019-01-23 17:31:37 +01001506 * Function used to parse a stick-table update acknowledgement message after it
1507 * has been received by <p> peer with <msg_cur> as address of the pointer to the position in the
1508 * receipt buffer with <msg_end> being the position of the end of the stick-table message.
1509 * Update <msg_curr> accordingly to the peer protocol specs if no peer protocol error
1510 * was encountered.
1511 * Return 1 if succeeded, 0 if not with the appctx state st0 set to PEER_SESS_ST_ERRPROTO.
1512 */
1513static inline int peer_treat_ackmsg(struct appctx *appctx, struct peer *p,
1514 char **msg_cur, char *msg_end)
1515{
1516 /* ack message */
1517 uint32_t table_id ;
1518 uint32_t update;
1519 struct shared_table *st;
1520
1521 table_id = intdecode(msg_cur, msg_end);
1522 if (!*msg_cur || (*msg_cur + sizeof(update) > msg_end)) {
1523 /* malformed message */
1524 appctx->st0 = PEER_SESS_ST_ERRPROTO;
1525 return 0;
1526 }
1527
1528 memcpy(&update, *msg_cur, sizeof(update));
1529 update = ntohl(update);
1530
1531 for (st = p->tables; st; st = st->next) {
1532 if (st->local_id == table_id) {
1533 st->update = update;
1534 break;
1535 }
1536 }
1537
1538 return 1;
1539}
1540
1541/*
1542 * Function used to parse a stick-table switch message after it has been received
1543 * by <p> peer with <msg_cur> as address of the pointer to the position in the
1544 * receipt buffer with <msg_end> being the position of the end of the stick-table message.
1545 * Update <msg_curr> accordingly to the peer protocol specs if no peer protocol error
1546 * was encountered.
1547 * Return 1 if succeeded, 0 if not with the appctx state st0 set to PEER_SESS_ST_ERRPROTO.
1548 */
1549static inline int peer_treat_switchmsg(struct appctx *appctx, struct peer *p,
1550 char **msg_cur, char *msg_end)
1551{
1552 struct shared_table *st;
1553 int table_id;
1554
1555 table_id = intdecode(msg_cur, msg_end);
1556 if (!*msg_cur) {
1557 /* malformed message */
1558 appctx->st0 = PEER_SESS_ST_ERRPROTO;
1559 return 0;
1560 }
1561
1562 p->remote_table = NULL;
1563 for (st = p->tables; st; st = st->next) {
1564 if (st->remote_id == table_id) {
1565 p->remote_table = st;
1566 break;
1567 }
1568 }
1569
1570 return 1;
1571}
1572
1573/*
1574 * Function used to parse a stick-table definition message after it has been received
1575 * by <p> peer with <msg_cur> as address of the pointer to the position in the
1576 * receipt buffer with <msg_end> being the position of the end of the stick-table message.
1577 * Update <msg_curr> accordingly to the peer protocol specs if no peer protocol error
1578 * was encountered.
1579 * <totl> is the length of the stick-table update message computed upon receipt.
1580 * Return 1 if succeeded, 0 if not with the appctx state st0 set to PEER_SESS_ST_ERRPROTO.
1581 */
1582static inline int peer_treat_definemsg(struct appctx *appctx, struct peer *p,
1583 char **msg_cur, char *msg_end, int totl)
1584{
1585 struct stream_interface *si = appctx->owner;
1586 int table_id_len;
1587 struct shared_table *st;
1588 int table_type;
1589 int table_keylen;
1590 int table_id;
1591 uint64_t table_data;
1592
1593 table_id = intdecode(msg_cur, msg_end);
Willy Tarreau6f731f32019-01-29 11:11:23 +01001594 if (!*msg_cur)
1595 goto malformed_exit;
Frédéric Lécailled27b0942019-01-23 17:31:37 +01001596
1597 table_id_len = intdecode(msg_cur, msg_end);
Willy Tarreau6f731f32019-01-29 11:11:23 +01001598 if (!*msg_cur)
1599 goto malformed_exit;
Frédéric Lécailled27b0942019-01-23 17:31:37 +01001600
1601 p->remote_table = NULL;
Willy Tarreau6f731f32019-01-29 11:11:23 +01001602 if (!table_id_len || (*msg_cur + table_id_len) >= msg_end)
1603 goto malformed_exit;
Frédéric Lécailled27b0942019-01-23 17:31:37 +01001604
1605 for (st = p->tables; st; st = st->next) {
1606 /* Reset IDs */
1607 if (st->remote_id == table_id)
1608 st->remote_id = 0;
1609
Frédéric Lécaille7fcc24d2019-03-20 15:09:45 +01001610 if (!p->remote_table && (table_id_len == strlen(st->table->nid)) &&
1611 (memcmp(st->table->nid, *msg_cur, table_id_len) == 0))
Frédéric Lécailled27b0942019-01-23 17:31:37 +01001612 p->remote_table = st;
1613 }
1614
1615 if (!p->remote_table)
1616 goto ignore_msg;
1617
1618 *msg_cur += table_id_len;
Willy Tarreau6f731f32019-01-29 11:11:23 +01001619 if (*msg_cur >= msg_end)
1620 goto malformed_exit;
Frédéric Lécailled27b0942019-01-23 17:31:37 +01001621
1622 table_type = intdecode(msg_cur, msg_end);
Willy Tarreau6f731f32019-01-29 11:11:23 +01001623 if (!*msg_cur)
1624 goto malformed_exit;
Frédéric Lécailled27b0942019-01-23 17:31:37 +01001625
1626 table_keylen = intdecode(msg_cur, msg_end);
Willy Tarreau6f731f32019-01-29 11:11:23 +01001627 if (!*msg_cur)
1628 goto malformed_exit;
Frédéric Lécailled27b0942019-01-23 17:31:37 +01001629
1630 table_data = intdecode(msg_cur, msg_end);
Willy Tarreau6f731f32019-01-29 11:11:23 +01001631 if (!*msg_cur)
1632 goto malformed_exit;
Frédéric Lécailled27b0942019-01-23 17:31:37 +01001633
1634 if (p->remote_table->table->type != table_type
1635 || p->remote_table->table->key_size != table_keylen) {
1636 p->remote_table = NULL;
1637 goto ignore_msg;
1638 }
1639
1640 p->remote_table->remote_data = table_data;
1641 p->remote_table->remote_id = table_id;
1642 return 1;
1643
1644 ignore_msg:
1645 co_skip(si_oc(si), totl);
1646 return 0;
Willy Tarreau6f731f32019-01-29 11:11:23 +01001647
1648 malformed_exit:
1649 /* malformed message */
1650 appctx->st0 = PEER_SESS_ST_ERRPROTO;
1651 return 0;
Frédéric Lécailled27b0942019-01-23 17:31:37 +01001652}
1653
1654/*
Frédéric Lécaille95203f22019-01-23 19:38:11 +01001655 * Receive a stick-table message.
1656 * Returns 1 if there was no error, if not, returns 0 if not enough data were available,
1657 * -1 if there was an error updating the appctx state st0 accordingly.
1658 */
1659static inline int peer_recv_msg(struct appctx *appctx, char *msg_head, size_t msg_head_sz,
1660 uint32_t *msg_len, int *totl)
1661{
1662 int reql;
1663 struct stream_interface *si = appctx->owner;
1664
1665 reql = co_getblk(si_oc(si), msg_head, 2 * sizeof(char), *totl);
1666 if (reql <= 0) /* closed or EOL not found */
1667 goto incomplete;
1668
1669 *totl += reql;
1670
1671 if ((unsigned int)msg_head[1] < 128)
1672 return 1;
1673
1674 /* Read and Decode message length */
1675 reql = co_getblk(si_oc(si), &msg_head[2], sizeof(char), *totl);
1676 if (reql <= 0) /* closed */
1677 goto incomplete;
1678
1679 *totl += reql;
1680
Frédéric Lécaille32b55732019-06-03 18:29:51 +02001681 if ((unsigned int)msg_head[2] < PEER_ENC_2BYTES_MIN) {
Frédéric Lécaille95203f22019-01-23 19:38:11 +01001682 *msg_len = msg_head[2];
1683 }
1684 else {
1685 int i;
1686 char *cur;
1687 char *end;
1688
1689 for (i = 3 ; i < msg_head_sz ; i++) {
1690 reql = co_getblk(si_oc(si), &msg_head[i], sizeof(char), *totl);
1691 if (reql <= 0) /* closed */
1692 goto incomplete;
1693
1694 *totl += reql;
1695
1696 if (!(msg_head[i] & 0x80))
1697 break;
1698 }
1699
1700 if (i == msg_head_sz) {
1701 /* malformed message */
1702 appctx->st0 = PEER_SESS_ST_ERRPROTO;
1703 return -1;
1704 }
1705 end = msg_head + msg_head_sz;
1706 cur = &msg_head[2];
1707 *msg_len = intdecode(&cur, end);
1708 if (!cur) {
1709 /* malformed message */
1710 appctx->st0 = PEER_SESS_ST_ERRPROTO;
1711 return -1;
1712 }
1713 }
1714
1715 /* Read message content */
1716 if (*msg_len) {
1717 if (*msg_len > trash.size) {
1718 /* Status code is not success, abort */
1719 appctx->st0 = PEER_SESS_ST_ERRSIZE;
1720 return -1;
1721 }
1722
1723 reql = co_getblk(si_oc(si), trash.area, *msg_len, *totl);
1724 if (reql <= 0) /* closed */
1725 goto incomplete;
1726 *totl += reql;
1727 }
1728
1729 return 1;
1730
1731 incomplete:
1732 if (reql < 0) {
1733 /* there was an error */
1734 appctx->st0 = PEER_SESS_ST_END;
1735 return -1;
1736 }
1737
1738 return 0;
1739}
Frédéric Lécaille444243c2019-01-24 15:40:11 +01001740
1741/*
1742 * Treat the awaited message with <msg_head> as header.*
1743 * Return 1 if succeeded, 0 if not.
1744 */
1745static inline int peer_treat_awaited_msg(struct appctx *appctx, struct peer *peer, unsigned char *msg_head,
1746 char **msg_cur, char *msg_end, int msg_len, int totl)
1747{
1748 struct stream_interface *si = appctx->owner;
1749 struct stream *s = si_strm(si);
1750 struct peers *peers = strm_fe(s)->parent;
1751
1752 if (msg_head[0] == PEER_MSG_CLASS_CONTROL) {
1753 if (msg_head[1] == PEER_MSG_CTRL_RESYNCREQ) {
1754 struct shared_table *st;
1755 /* Reset message: remote need resync */
1756
1757 /* prepare tables fot a global push */
1758 for (st = peer->tables; st; st = st->next) {
1759 st->teaching_origin = st->last_pushed = st->table->update;
1760 st->flags = 0;
1761 }
1762
1763 /* reset teaching flags to 0 */
1764 peer->flags &= PEER_TEACH_RESET;
1765
1766 /* flag to start to teach lesson */
1767 peer->flags |= PEER_F_TEACH_PROCESS;
1768 }
1769 else if (msg_head[1] == PEER_MSG_CTRL_RESYNCFINISHED) {
1770 if (peer->flags & PEER_F_LEARN_ASSIGN) {
1771 peer->flags &= ~PEER_F_LEARN_ASSIGN;
1772 peers->flags &= ~(PEERS_F_RESYNC_ASSIGN|PEERS_F_RESYNC_PROCESS);
1773 peers->flags |= (PEERS_F_RESYNC_LOCAL|PEERS_F_RESYNC_REMOTE);
1774 }
1775 peer->confirm++;
1776 }
1777 else if (msg_head[1] == PEER_MSG_CTRL_RESYNCPARTIAL) {
1778 if (peer->flags & PEER_F_LEARN_ASSIGN) {
1779 peer->flags &= ~PEER_F_LEARN_ASSIGN;
1780 peers->flags &= ~(PEERS_F_RESYNC_ASSIGN|PEERS_F_RESYNC_PROCESS);
1781
1782 peer->flags |= PEER_F_LEARN_NOTUP2DATE;
Frédéric Lécaille54bff832019-03-26 10:25:20 +01001783 peers->resync_timeout = tick_add(now_ms, MS_TO_TICKS(PEER_RESYNC_TIMEOUT));
Frédéric Lécaille444243c2019-01-24 15:40:11 +01001784 task_wakeup(peers->sync_task, TASK_WOKEN_MSG);
1785 }
1786 peer->confirm++;
1787 }
1788 else if (msg_head[1] == PEER_MSG_CTRL_RESYNCCONFIRM) {
1789 struct shared_table *st;
1790
1791 /* If stopping state */
1792 if (stopping) {
1793 /* Close session, push resync no more needed */
1794 peer->flags |= PEER_F_TEACH_COMPLETE;
1795 appctx->st0 = PEER_SESS_ST_END;
1796 return 0;
1797 }
1798 for (st = peer->tables; st; st = st->next) {
1799 st->update = st->last_pushed = st->teaching_origin;
1800 st->flags = 0;
1801 }
1802
1803 /* reset teaching flags to 0 */
1804 peer->flags &= PEER_TEACH_RESET;
1805 }
Frédéric Lécaille645635d2019-02-11 17:49:39 +01001806 else if (msg_head[1] == PEER_MSG_CTRL_HEARTBEAT) {
Frédéric Lécaille54bff832019-03-26 10:25:20 +01001807 peer->reconnect = tick_add(now_ms, MS_TO_TICKS(PEER_RECONNECT_TIMEOUT));
Frédéric Lécaille645635d2019-02-11 17:49:39 +01001808 }
Frédéric Lécaille444243c2019-01-24 15:40:11 +01001809 }
1810 else if (msg_head[0] == PEER_MSG_CLASS_STICKTABLE) {
1811 if (msg_head[1] == PEER_MSG_STKT_DEFINE) {
1812 if (!peer_treat_definemsg(appctx, peer, msg_cur, msg_end, totl))
1813 return 0;
1814 }
1815 else if (msg_head[1] == PEER_MSG_STKT_SWITCH) {
1816 if (!peer_treat_switchmsg(appctx, peer, msg_cur, msg_end))
1817 return 0;
1818 }
1819 else if (msg_head[1] == PEER_MSG_STKT_UPDATE ||
1820 msg_head[1] == PEER_MSG_STKT_INCUPDATE ||
1821 msg_head[1] == PEER_MSG_STKT_UPDATE_TIMED ||
1822 msg_head[1] == PEER_MSG_STKT_INCUPDATE_TIMED) {
1823 int update, expire;
1824
1825 update = msg_head[1] == PEER_MSG_STKT_UPDATE || msg_head[1] == PEER_MSG_STKT_UPDATE_TIMED;
1826 expire = msg_head[1] == PEER_MSG_STKT_UPDATE_TIMED || msg_head[1] == PEER_MSG_STKT_INCUPDATE_TIMED;
1827 if (!peer_treat_updatemsg(appctx, peer, update, expire,
1828 msg_cur, msg_end, msg_len, totl))
1829 return 0;
1830
1831 }
1832 else if (msg_head[1] == PEER_MSG_STKT_ACK) {
1833 if (!peer_treat_ackmsg(appctx, peer, msg_cur, msg_end))
1834 return 0;
1835 }
1836 }
1837 else if (msg_head[0] == PEER_MSG_CLASS_RESERVED) {
1838 appctx->st0 = PEER_SESS_ST_ERRPROTO;
1839 return 0;
1840 }
1841
1842 return 1;
1843}
Frédéric Lécaille25e1d5e2019-01-24 17:33:48 +01001844
1845
1846/*
1847 * Send any message to <peer> peer.
1848 * Returns 1 if succeeded, or -1 or 0 if failed.
1849 * -1 means an internal error occured, 0 is for a peer protocol error leading
1850 * to a peer state change (from the peer I/O handler point of view).
1851 */
1852static inline int peer_send_msgs(struct appctx *appctx, struct peer *peer)
1853{
1854 int repl;
1855 struct stream_interface *si = appctx->owner;
1856 struct stream *s = si_strm(si);
1857 struct peers *peers = strm_fe(s)->parent;
1858
1859 /* Need to request a resync */
1860 if ((peer->flags & PEER_F_LEARN_ASSIGN) &&
1861 (peers->flags & PEERS_F_RESYNC_ASSIGN) &&
1862 !(peers->flags & PEERS_F_RESYNC_PROCESS)) {
1863
1864 repl = peer_send_resync_reqmsg(appctx);
1865 if (repl <= 0)
1866 return repl;
1867
1868 peers->flags |= PEERS_F_RESYNC_PROCESS;
1869 }
1870
1871 /* Nothing to read, now we start to write */
1872 if (peer->tables) {
1873 struct shared_table *st;
1874 struct shared_table *last_local_table;
1875
1876 last_local_table = peer->last_local_table;
1877 if (!last_local_table)
1878 last_local_table = peer->tables;
1879 st = last_local_table->next;
1880
1881 while (1) {
1882 if (!st)
1883 st = peer->tables;
1884
1885 /* It remains some updates to ack */
1886 if (st->last_get != st->last_acked) {
1887 repl = peer_send_ackmsg(st, appctx);
1888 if (repl <= 0)
1889 return repl;
1890
1891 st->last_acked = st->last_get;
1892 }
1893
1894 if (!(peer->flags & PEER_F_TEACH_PROCESS)) {
1895 HA_SPIN_LOCK(STK_TABLE_LOCK, &st->table->lock);
1896 if (!(peer->flags & PEER_F_LEARN_ASSIGN) &&
1897 ((int)(st->last_pushed - st->table->localupdate) < 0)) {
1898
1899 repl = peer_send_teach_process_msgs(appctx, peer, st);
1900 if (repl <= 0) {
1901 HA_SPIN_UNLOCK(STK_TABLE_LOCK, &st->table->lock);
1902 return repl;
1903 }
1904 }
1905 HA_SPIN_UNLOCK(STK_TABLE_LOCK, &st->table->lock);
1906 }
1907 else {
1908 if (!(st->flags & SHTABLE_F_TEACH_STAGE1)) {
1909 repl = peer_send_teach_stage1_msgs(appctx, peer, st);
1910 if (repl <= 0)
1911 return repl;
1912 }
1913
1914 if (!(st->flags & SHTABLE_F_TEACH_STAGE2)) {
1915 repl = peer_send_teach_stage2_msgs(appctx, peer, st);
1916 if (repl <= 0)
1917 return repl;
1918 }
1919 }
1920
1921 if (st == last_local_table)
1922 break;
1923 st = st->next;
1924 }
1925 }
1926
1927 if ((peer->flags & PEER_F_TEACH_PROCESS) && !(peer->flags & PEER_F_TEACH_FINISHED)) {
1928 repl = peer_send_resync_finishedmsg(appctx, peer);
1929 if (repl <= 0)
1930 return repl;
1931
1932 /* flag finished message sent */
1933 peer->flags |= PEER_F_TEACH_FINISHED;
1934 }
1935
1936 /* Confirm finished or partial messages */
1937 while (peer->confirm) {
1938 repl = peer_send_resync_confirmsg(appctx);
1939 if (repl <= 0)
1940 return repl;
1941
1942 peer->confirm--;
1943 }
1944
1945 return 1;
1946}
1947
Frédéric Lécaille95203f22019-01-23 19:38:11 +01001948/*
Frédéric Lécaille3f0fb9d2019-01-25 08:30:29 +01001949 * Read and parse a first line of a "hello" peer protocol message.
1950 * Returns 0 if could not read a line, -1 if there was a read error or
1951 * the line is malformed, 1 if succeeded.
1952 */
1953static inline int peer_getline_version(struct appctx *appctx,
1954 unsigned int *maj_ver, unsigned int *min_ver)
1955{
1956 int reql;
1957
1958 reql = peer_getline(appctx);
1959 if (!reql)
1960 return 0;
1961
1962 if (reql < 0)
1963 return -1;
1964
1965 /* test protocol */
1966 if (strncmp(PEER_SESSION_PROTO_NAME " ", trash.area, proto_len + 1) != 0) {
1967 appctx->st0 = PEER_SESS_ST_EXIT;
1968 appctx->st1 = PEER_SESS_SC_ERRPROTO;
1969 return -1;
1970 }
1971 if (peer_get_version(trash.area + proto_len + 1, maj_ver, min_ver) == -1 ||
1972 *maj_ver != PEER_MAJOR_VER || *min_ver > PEER_MINOR_VER) {
1973 appctx->st0 = PEER_SESS_ST_EXIT;
1974 appctx->st1 = PEER_SESS_SC_ERRVERSION;
1975 return -1;
1976 }
1977
1978 return 1;
1979}
1980
1981/*
1982 * Read and parse a second line of a "hello" peer protocol message.
1983 * Returns 0 if could not read a line, -1 if there was a read error or
1984 * the line is malformed, 1 if succeeded.
1985 */
1986static inline int peer_getline_host(struct appctx *appctx)
1987{
1988 int reql;
1989
1990 reql = peer_getline(appctx);
1991 if (!reql)
1992 return 0;
1993
1994 if (reql < 0)
1995 return -1;
1996
1997 /* test hostname match */
1998 if (strcmp(localpeer, trash.area) != 0) {
1999 appctx->st0 = PEER_SESS_ST_EXIT;
2000 appctx->st1 = PEER_SESS_SC_ERRHOST;
2001 return -1;
2002 }
2003
2004 return 1;
2005}
2006
2007/*
2008 * Read and parse a last line of a "hello" peer protocol message.
2009 * Returns 0 if could not read a character, -1 if there was a read error or
2010 * the line is malformed, 1 if succeeded.
2011 * Set <curpeer> accordingly (the remote peer sending the "hello" message).
2012 */
2013static inline int peer_getline_last(struct appctx *appctx, struct peer **curpeer)
2014{
2015 char *p;
2016 int reql;
2017 struct peer *peer;
2018 struct stream_interface *si = appctx->owner;
2019 struct stream *s = si_strm(si);
2020 struct peers *peers = strm_fe(s)->parent;
2021
2022 reql = peer_getline(appctx);
2023 if (!reql)
2024 return 0;
2025
2026 if (reql < 0)
2027 return -1;
2028
2029 /* parse line "<peer name> <pid> <relative_pid>" */
2030 p = strchr(trash.area, ' ');
2031 if (!p) {
2032 appctx->st0 = PEER_SESS_ST_EXIT;
2033 appctx->st1 = PEER_SESS_SC_ERRPROTO;
2034 return -1;
2035 }
2036 *p = 0;
2037
2038 /* lookup known peer */
2039 for (peer = peers->remote; peer; peer = peer->next) {
2040 if (strcmp(peer->id, trash.area) == 0)
2041 break;
2042 }
2043
2044 /* if unknown peer */
2045 if (!peer) {
2046 appctx->st0 = PEER_SESS_ST_EXIT;
2047 appctx->st1 = PEER_SESS_SC_ERRPEER;
2048 return -1;
2049 }
2050 *curpeer = peer;
2051
2052 return 1;
2053}
2054
2055/*
Frédéric Lécaille4b2fd9b2019-01-25 08:58:41 +01002056 * Init <peer> peer after having accepted it at peer protocol level.
2057 */
2058static inline void init_accepted_peer(struct peer *peer, struct peers *peers)
2059{
2060 struct shared_table *st;
2061
2062 /* Register status code */
2063 peer->statuscode = PEER_SESS_SC_SUCCESSCODE;
2064
2065 /* Awake main task */
2066 task_wakeup(peers->sync_task, TASK_WOKEN_MSG);
2067
2068 /* Init confirm counter */
2069 peer->confirm = 0;
2070
2071 /* Init cursors */
2072 for (st = peer->tables; st ; st = st->next) {
2073 st->last_get = st->last_acked = 0;
2074 st->teaching_origin = st->last_pushed = st->update;
2075 }
2076
2077 /* reset teaching and learning flags to 0 */
2078 peer->flags &= PEER_TEACH_RESET;
2079 peer->flags &= PEER_LEARN_RESET;
2080
2081 /* if current peer is local */
2082 if (peer->local) {
2083 /* if current host need resyncfrom local and no process assined */
2084 if ((peers->flags & PEERS_RESYNC_STATEMASK) == PEERS_RESYNC_FROMLOCAL &&
2085 !(peers->flags & PEERS_F_RESYNC_ASSIGN)) {
2086 /* assign local peer for a lesson, consider lesson already requested */
2087 peer->flags |= PEER_F_LEARN_ASSIGN;
2088 peers->flags |= (PEERS_F_RESYNC_ASSIGN|PEERS_F_RESYNC_PROCESS);
2089 }
2090
2091 }
2092 else if ((peers->flags & PEERS_RESYNC_STATEMASK) == PEERS_RESYNC_FROMREMOTE &&
2093 !(peers->flags & PEERS_F_RESYNC_ASSIGN)) {
2094 /* assign peer for a lesson */
2095 peer->flags |= PEER_F_LEARN_ASSIGN;
2096 peers->flags |= PEERS_F_RESYNC_ASSIGN;
2097 }
2098}
2099
2100/*
2101 * Init <peer> peer after having connected it at peer protocol level.
2102 */
2103static inline void init_connected_peer(struct peer *peer, struct peers *peers)
2104{
2105 struct shared_table *st;
2106
2107 /* Init cursors */
2108 for (st = peer->tables; st ; st = st->next) {
2109 st->last_get = st->last_acked = 0;
2110 st->teaching_origin = st->last_pushed = st->update;
2111 }
2112
2113 /* Init confirm counter */
2114 peer->confirm = 0;
2115
2116 /* reset teaching and learning flags to 0 */
2117 peer->flags &= PEER_TEACH_RESET;
2118 peer->flags &= PEER_LEARN_RESET;
2119
2120 /* If current peer is local */
2121 if (peer->local) {
2122 /* flag to start to teach lesson */
2123 peer->flags |= PEER_F_TEACH_PROCESS;
2124 }
2125 else if ((peers->flags & PEERS_RESYNC_STATEMASK) == PEERS_RESYNC_FROMREMOTE &&
2126 !(peers->flags & PEERS_F_RESYNC_ASSIGN)) {
2127 /* If peer is remote and resync from remote is needed,
2128 and no peer currently assigned */
2129
2130 /* assign peer for a lesson */
2131 peer->flags |= PEER_F_LEARN_ASSIGN;
2132 peers->flags |= PEERS_F_RESYNC_ASSIGN;
2133 }
2134}
2135
2136/*
Emeric Brun2b920a12010-09-23 18:30:22 +02002137 * IO Handler to handle message exchance with a peer
2138 */
Willy Tarreau00a37f02015-04-13 12:05:19 +02002139static void peer_io_handler(struct appctx *appctx)
Emeric Brun2b920a12010-09-23 18:30:22 +02002140{
Willy Tarreau00a37f02015-04-13 12:05:19 +02002141 struct stream_interface *si = appctx->owner;
Willy Tarreau87b09662015-04-03 00:22:06 +02002142 struct stream *s = si_strm(si);
Vincent Bernat3c2f2f22016-04-03 13:48:42 +02002143 struct peers *curpeers = strm_fe(s)->parent;
Emeric Brun80527f52017-06-19 17:46:37 +02002144 struct peer *curpeer = NULL;
Emeric Brun2b920a12010-09-23 18:30:22 +02002145 int reql = 0;
2146 int repl = 0;
Frédéric Lécaille523cc9e2016-10-12 17:30:30 +02002147 unsigned int maj_ver, min_ver;
Willy Tarreau2d372c22018-11-05 17:12:27 +01002148 int prev_state;
Emeric Brun2b920a12010-09-23 18:30:22 +02002149
Joseph Herlant82b2f542018-11-15 12:19:14 -08002150 /* Check if the input buffer is available. */
Frédéric Lécaille7d0ceee2019-01-24 14:24:05 +01002151 if (si_ic(si)->buf.size == 0) {
2152 si_rx_room_blk(si);
2153 goto out;
2154 }
Christopher Fauleta73e59b2016-12-09 17:30:18 +01002155
Emeric Brun2b920a12010-09-23 18:30:22 +02002156 while (1) {
Willy Tarreau2d372c22018-11-05 17:12:27 +01002157 prev_state = appctx->st0;
Emeric Brun2b920a12010-09-23 18:30:22 +02002158switchstate:
Frédéric Lécaille523cc9e2016-10-12 17:30:30 +02002159 maj_ver = min_ver = (unsigned int)-1;
Willy Tarreau7b4b4992013-12-01 09:15:12 +01002160 switch(appctx->st0) {
Willy Tarreaue4d927a2013-12-01 12:47:35 +01002161 case PEER_SESS_ST_ACCEPT:
Willy Tarreau2d372c22018-11-05 17:12:27 +01002162 prev_state = appctx->st0;
Willy Tarreau7b4b4992013-12-01 09:15:12 +01002163 appctx->ctx.peers.ptr = NULL;
Willy Tarreaue4d927a2013-12-01 12:47:35 +01002164 appctx->st0 = PEER_SESS_ST_GETVERSION;
Emeric Brun2b920a12010-09-23 18:30:22 +02002165 /* fall through */
Willy Tarreaue4d927a2013-12-01 12:47:35 +01002166 case PEER_SESS_ST_GETVERSION:
Willy Tarreau2d372c22018-11-05 17:12:27 +01002167 prev_state = appctx->st0;
Frédéric Lécaille3f0fb9d2019-01-25 08:30:29 +01002168 reql = peer_getline_version(appctx, &maj_ver, &min_ver);
2169 if (reql <= 0) {
2170 if (!reql)
2171 goto out;
Emeric Brun2b920a12010-09-23 18:30:22 +02002172 goto switchstate;
2173 }
2174
Willy Tarreaue4d927a2013-12-01 12:47:35 +01002175 appctx->st0 = PEER_SESS_ST_GETHOST;
Emeric Brun2b920a12010-09-23 18:30:22 +02002176 /* fall through */
Willy Tarreaue4d927a2013-12-01 12:47:35 +01002177 case PEER_SESS_ST_GETHOST:
Willy Tarreau2d372c22018-11-05 17:12:27 +01002178 prev_state = appctx->st0;
Frédéric Lécaille3f0fb9d2019-01-25 08:30:29 +01002179 reql = peer_getline_host(appctx);
2180 if (reql <= 0) {
2181 if (!reql)
2182 goto out;
Emeric Brun2b920a12010-09-23 18:30:22 +02002183 goto switchstate;
2184 }
2185
Willy Tarreaue4d927a2013-12-01 12:47:35 +01002186 appctx->st0 = PEER_SESS_ST_GETPEER;
Emeric Brun2b920a12010-09-23 18:30:22 +02002187 /* fall through */
Willy Tarreaue4d927a2013-12-01 12:47:35 +01002188 case PEER_SESS_ST_GETPEER: {
Willy Tarreau2d372c22018-11-05 17:12:27 +01002189 prev_state = appctx->st0;
Frédéric Lécaille3f0fb9d2019-01-25 08:30:29 +01002190 reql = peer_getline_last(appctx, &curpeer);
2191 if (reql <= 0) {
2192 if (!reql)
2193 goto out;
Emeric Brun2b920a12010-09-23 18:30:22 +02002194 goto switchstate;
2195 }
Emeric Brun2b920a12010-09-23 18:30:22 +02002196
Christopher Faulet2a944ee2017-11-07 10:42:54 +01002197 HA_SPIN_LOCK(PEER_LOCK, &curpeer->lock);
Willy Tarreau9df94c22016-10-31 18:42:52 +01002198 if (curpeer->appctx && curpeer->appctx != appctx) {
Emeric Brunb3971ab2015-05-12 18:49:09 +02002199 if (curpeer->local) {
2200 /* Local connection, reply a retry */
2201 appctx->st0 = PEER_SESS_ST_EXIT;
2202 appctx->st1 = PEER_SESS_SC_TRYAGAIN;
2203 goto switchstate;
Emeric Brun2b920a12010-09-23 18:30:22 +02002204 }
Emeric Brun80527f52017-06-19 17:46:37 +02002205
2206 /* we're killing a connection, we must apply a random delay before
2207 * retrying otherwise the other end will do the same and we can loop
2208 * for a while.
2209 */
2210 curpeer->reconnect = tick_add(now_ms, MS_TO_TICKS(50 + random() % 2000));
Emeric Brun9ef2ad72019-04-02 17:22:01 +02002211 peer_session_forceshutdown(curpeer);
Emeric Brun2b920a12010-09-23 18:30:22 +02002212 }
Frédéric Lécaille523cc9e2016-10-12 17:30:30 +02002213 if (maj_ver != (unsigned int)-1 && min_ver != (unsigned int)-1) {
2214 if (min_ver == PEER_DWNGRD_MINOR_VER) {
2215 curpeer->flags |= PEER_F_DWNGRD;
2216 }
2217 else {
2218 curpeer->flags &= ~PEER_F_DWNGRD;
2219 }
2220 }
Emeric Brunb3971ab2015-05-12 18:49:09 +02002221 curpeer->appctx = appctx;
2222 appctx->ctx.peers.ptr = curpeer;
Willy Tarreaue4d927a2013-12-01 12:47:35 +01002223 appctx->st0 = PEER_SESS_ST_SENDSUCCESS;
Olivier Houcharded879892019-03-08 18:53:43 +01002224 _HA_ATOMIC_ADD(&active_peers, 1);
Emeric Brun2b920a12010-09-23 18:30:22 +02002225 /* fall through */
2226 }
Willy Tarreaue4d927a2013-12-01 12:47:35 +01002227 case PEER_SESS_ST_SENDSUCCESS: {
Willy Tarreau2d372c22018-11-05 17:12:27 +01002228 prev_state = appctx->st0;
Emeric Brun80527f52017-06-19 17:46:37 +02002229 if (!curpeer) {
2230 curpeer = appctx->ctx.peers.ptr;
Christopher Faulet2a944ee2017-11-07 10:42:54 +01002231 HA_SPIN_LOCK(PEER_LOCK, &curpeer->lock);
Emeric Brun80527f52017-06-19 17:46:37 +02002232 if (curpeer->appctx != appctx) {
2233 appctx->st0 = PEER_SESS_ST_END;
2234 goto switchstate;
2235 }
2236 }
Frédéric Lécaille7d0ceee2019-01-24 14:24:05 +01002237
2238 repl = peer_send_status_successmsg(appctx);
Emeric Brun2b920a12010-09-23 18:30:22 +02002239 if (repl <= 0) {
2240 if (repl == -1)
Frédéric Lécaille7d0ceee2019-01-24 14:24:05 +01002241 goto out;
Emeric Brun2b920a12010-09-23 18:30:22 +02002242 goto switchstate;
2243 }
2244
Frédéric Lécaille4b2fd9b2019-01-25 08:58:41 +01002245 init_accepted_peer(curpeer, curpeers);
Emeric Brunb3971ab2015-05-12 18:49:09 +02002246
Emeric Brun2b920a12010-09-23 18:30:22 +02002247 /* switch to waiting message state */
Olivier Houcharded879892019-03-08 18:53:43 +01002248 _HA_ATOMIC_ADD(&connected_peers, 1);
Willy Tarreaue4d927a2013-12-01 12:47:35 +01002249 appctx->st0 = PEER_SESS_ST_WAITMSG;
Emeric Brun2b920a12010-09-23 18:30:22 +02002250 goto switchstate;
2251 }
Willy Tarreaue4d927a2013-12-01 12:47:35 +01002252 case PEER_SESS_ST_CONNECT: {
Willy Tarreau2d372c22018-11-05 17:12:27 +01002253 prev_state = appctx->st0;
Emeric Brun80527f52017-06-19 17:46:37 +02002254 if (!curpeer) {
2255 curpeer = appctx->ctx.peers.ptr;
Christopher Faulet2a944ee2017-11-07 10:42:54 +01002256 HA_SPIN_LOCK(PEER_LOCK, &curpeer->lock);
Emeric Brun80527f52017-06-19 17:46:37 +02002257 if (curpeer->appctx != appctx) {
2258 appctx->st0 = PEER_SESS_ST_END;
2259 goto switchstate;
2260 }
2261 }
Emeric Brun2b920a12010-09-23 18:30:22 +02002262
Frédéric Lécaille7d0ceee2019-01-24 14:24:05 +01002263 repl = peer_send_hellomsg(appctx, curpeer);
Emeric Brun2b920a12010-09-23 18:30:22 +02002264 if (repl <= 0) {
2265 if (repl == -1)
Frédéric Lécaille7d0ceee2019-01-24 14:24:05 +01002266 goto out;
Emeric Brun2b920a12010-09-23 18:30:22 +02002267 goto switchstate;
2268 }
2269
2270 /* switch to the waiting statuscode state */
Willy Tarreaue4d927a2013-12-01 12:47:35 +01002271 appctx->st0 = PEER_SESS_ST_GETSTATUS;
Emeric Brun2b920a12010-09-23 18:30:22 +02002272 /* fall through */
2273 }
Willy Tarreaue4d927a2013-12-01 12:47:35 +01002274 case PEER_SESS_ST_GETSTATUS: {
Willy Tarreau2d372c22018-11-05 17:12:27 +01002275 prev_state = appctx->st0;
Emeric Brun80527f52017-06-19 17:46:37 +02002276 if (!curpeer) {
2277 curpeer = appctx->ctx.peers.ptr;
Christopher Faulet2a944ee2017-11-07 10:42:54 +01002278 HA_SPIN_LOCK(PEER_LOCK, &curpeer->lock);
Emeric Brun80527f52017-06-19 17:46:37 +02002279 if (curpeer->appctx != appctx) {
2280 appctx->st0 = PEER_SESS_ST_END;
2281 goto switchstate;
2282 }
2283 }
2284
Willy Tarreau2bb4a962014-11-28 11:11:05 +01002285 if (si_ic(si)->flags & CF_WRITE_PARTIAL)
Emeric Brunb3971ab2015-05-12 18:49:09 +02002286 curpeer->statuscode = PEER_SESS_SC_CONNECTEDCODE;
Emeric Brun2b920a12010-09-23 18:30:22 +02002287
Frédéric Lécaillece025572019-01-21 13:38:06 +01002288 reql = peer_getline(appctx);
2289 if (!reql)
2290 goto out;
Emeric Brun2b920a12010-09-23 18:30:22 +02002291
Frédéric Lécaillece025572019-01-21 13:38:06 +01002292 if (reql < 0)
2293 goto switchstate;
Emeric Brun2b920a12010-09-23 18:30:22 +02002294
2295 /* Register status code */
Willy Tarreau843b7cb2018-07-13 10:54:26 +02002296 curpeer->statuscode = atoi(trash.area);
Emeric Brun2b920a12010-09-23 18:30:22 +02002297
2298 /* Awake main task */
Frédéric Lécailleed2b4a62017-07-13 09:07:09 +02002299 task_wakeup(curpeers->sync_task, TASK_WOKEN_MSG);
Emeric Brun2b920a12010-09-23 18:30:22 +02002300
2301 /* If status code is success */
Emeric Brunb3971ab2015-05-12 18:49:09 +02002302 if (curpeer->statuscode == PEER_SESS_SC_SUCCESSCODE) {
Frédéric Lécaille4b2fd9b2019-01-25 08:58:41 +01002303 init_connected_peer(curpeer, curpeers);
Emeric Brun2b920a12010-09-23 18:30:22 +02002304 }
2305 else {
Frédéric Lécaille523cc9e2016-10-12 17:30:30 +02002306 if (curpeer->statuscode == PEER_SESS_SC_ERRVERSION)
2307 curpeer->flags |= PEER_F_DWNGRD;
Emeric Brun2b920a12010-09-23 18:30:22 +02002308 /* Status code is not success, abort */
Willy Tarreaue4d927a2013-12-01 12:47:35 +01002309 appctx->st0 = PEER_SESS_ST_END;
Emeric Brun2b920a12010-09-23 18:30:22 +02002310 goto switchstate;
2311 }
Olivier Houcharded879892019-03-08 18:53:43 +01002312 _HA_ATOMIC_ADD(&connected_peers, 1);
Willy Tarreaue4d927a2013-12-01 12:47:35 +01002313 appctx->st0 = PEER_SESS_ST_WAITMSG;
Emeric Brun2b920a12010-09-23 18:30:22 +02002314 /* fall through */
2315 }
Willy Tarreaue4d927a2013-12-01 12:47:35 +01002316 case PEER_SESS_ST_WAITMSG: {
Emeric Brunb3971ab2015-05-12 18:49:09 +02002317 uint32_t msg_len = 0;
Willy Tarreau843b7cb2018-07-13 10:54:26 +02002318 char *msg_cur = trash.area;
2319 char *msg_end = trash.area;
Emeric Brunb3971ab2015-05-12 18:49:09 +02002320 unsigned char msg_head[7];
Emeric Brun2b920a12010-09-23 18:30:22 +02002321 int totl = 0;
2322
Willy Tarreau2d372c22018-11-05 17:12:27 +01002323 prev_state = appctx->st0;
Emeric Brun80527f52017-06-19 17:46:37 +02002324 if (!curpeer) {
2325 curpeer = appctx->ctx.peers.ptr;
Christopher Faulet2a944ee2017-11-07 10:42:54 +01002326 HA_SPIN_LOCK(PEER_LOCK, &curpeer->lock);
Emeric Brun80527f52017-06-19 17:46:37 +02002327 if (curpeer->appctx != appctx) {
2328 appctx->st0 = PEER_SESS_ST_END;
2329 goto switchstate;
2330 }
2331 }
2332
Frédéric Lécaille95203f22019-01-23 19:38:11 +01002333 reql = peer_recv_msg(appctx, (char *)msg_head, sizeof msg_head, &msg_len, &totl);
2334 if (reql <= 0) {
2335 if (reql == -1)
2336 goto switchstate;
Frédéric Lécaillebe825e52019-01-24 18:28:44 +01002337 goto send_msgs;
Emeric Brunb3971ab2015-05-12 18:49:09 +02002338 }
Willy Tarreau86a446e2013-11-25 23:02:37 +01002339
Frédéric Lécaille95203f22019-01-23 19:38:11 +01002340 msg_end += msg_len;
Frédéric Lécaille444243c2019-01-24 15:40:11 +01002341 if (!peer_treat_awaited_msg(appctx, curpeer, msg_head, &msg_cur, msg_end, msg_len, totl))
Emeric Brun2b920a12010-09-23 18:30:22 +02002342 goto switchstate;
Frédéric Lécaillebe825e52019-01-24 18:28:44 +01002343
Frédéric Lécaille045e0d42019-03-21 11:12:32 +01002344 curpeer->flags |= PEER_F_ALIVE;
2345
Emeric Brun2b920a12010-09-23 18:30:22 +02002346 /* skip consumed message */
Willy Tarreau06d80a92017-10-19 14:32:15 +02002347 co_skip(si_oc(si), totl);
Emeric Brun2b920a12010-09-23 18:30:22 +02002348 /* loop on that state to peek next message */
Willy Tarreau72d6c162013-04-11 16:14:13 +02002349 goto switchstate;
2350
Frédéric Lécaillebe825e52019-01-24 18:28:44 +01002351send_msgs:
Frédéric Lécaille645635d2019-02-11 17:49:39 +01002352 if (curpeer->flags & PEER_F_HEARTBEAT) {
2353 curpeer->flags &= ~PEER_F_HEARTBEAT;
2354 repl = peer_send_heartbeatmsg(appctx);
2355 if (repl <= 0) {
2356 if (repl == -1)
2357 goto out;
2358 goto switchstate;
2359 }
2360 }
Frédéric Lécaillebe825e52019-01-24 18:28:44 +01002361 /* we get here when a peer_recv_msg() returns 0 in reql */
Frédéric Lécaille25e1d5e2019-01-24 17:33:48 +01002362 repl = peer_send_msgs(appctx, curpeer);
2363 if (repl <= 0) {
2364 if (repl == -1)
2365 goto out;
2366 goto switchstate;
Emeric Brun597b26e2016-08-12 11:23:31 +02002367 }
2368
Emeric Brun2b920a12010-09-23 18:30:22 +02002369 /* noting more to do */
2370 goto out;
2371 }
Willy Tarreaue4d927a2013-12-01 12:47:35 +01002372 case PEER_SESS_ST_EXIT:
Willy Tarreau2d372c22018-11-05 17:12:27 +01002373 if (prev_state == PEER_SESS_ST_WAITMSG)
Olivier Houcharded879892019-03-08 18:53:43 +01002374 _HA_ATOMIC_SUB(&connected_peers, 1);
Willy Tarreau2d372c22018-11-05 17:12:27 +01002375 prev_state = appctx->st0;
Frédéric Lécaille7d0ceee2019-01-24 14:24:05 +01002376 if (peer_send_status_errormsg(appctx) == -1)
2377 goto out;
Willy Tarreaue4d927a2013-12-01 12:47:35 +01002378 appctx->st0 = PEER_SESS_ST_END;
Emeric Brunb3971ab2015-05-12 18:49:09 +02002379 goto switchstate;
2380 case PEER_SESS_ST_ERRSIZE: {
Willy Tarreau2d372c22018-11-05 17:12:27 +01002381 if (prev_state == PEER_SESS_ST_WAITMSG)
Olivier Houcharded879892019-03-08 18:53:43 +01002382 _HA_ATOMIC_SUB(&connected_peers, 1);
Willy Tarreau2d372c22018-11-05 17:12:27 +01002383 prev_state = appctx->st0;
Frédéric Lécaille7d0ceee2019-01-24 14:24:05 +01002384 if (peer_send_error_size_limitmsg(appctx) == -1)
2385 goto out;
Emeric Brunb3971ab2015-05-12 18:49:09 +02002386 appctx->st0 = PEER_SESS_ST_END;
2387 goto switchstate;
2388 }
2389 case PEER_SESS_ST_ERRPROTO: {
Willy Tarreau2d372c22018-11-05 17:12:27 +01002390 if (prev_state == PEER_SESS_ST_WAITMSG)
Olivier Houcharded879892019-03-08 18:53:43 +01002391 _HA_ATOMIC_SUB(&connected_peers, 1);
Willy Tarreau2d372c22018-11-05 17:12:27 +01002392 prev_state = appctx->st0;
Frédéric Lécaille7d0ceee2019-01-24 14:24:05 +01002393 if (peer_send_error_protomsg(appctx) == -1)
2394 goto out;
Emeric Brunb3971ab2015-05-12 18:49:09 +02002395 appctx->st0 = PEER_SESS_ST_END;
Willy Tarreau2d372c22018-11-05 17:12:27 +01002396 prev_state = appctx->st0;
Emeric Brun2b920a12010-09-23 18:30:22 +02002397 /* fall through */
Emeric Brunb3971ab2015-05-12 18:49:09 +02002398 }
Willy Tarreaue4d927a2013-12-01 12:47:35 +01002399 case PEER_SESS_ST_END: {
Willy Tarreau2d372c22018-11-05 17:12:27 +01002400 if (prev_state == PEER_SESS_ST_WAITMSG)
Olivier Houcharded879892019-03-08 18:53:43 +01002401 _HA_ATOMIC_SUB(&connected_peers, 1);
Willy Tarreau2d372c22018-11-05 17:12:27 +01002402 prev_state = appctx->st0;
Emeric Brun80527f52017-06-19 17:46:37 +02002403 if (curpeer) {
Christopher Faulet2a944ee2017-11-07 10:42:54 +01002404 HA_SPIN_UNLOCK(PEER_LOCK, &curpeer->lock);
Emeric Brun80527f52017-06-19 17:46:37 +02002405 curpeer = NULL;
2406 }
Willy Tarreau73b013b2012-05-21 16:31:45 +02002407 si_shutw(si);
2408 si_shutr(si);
Willy Tarreau2bb4a962014-11-28 11:11:05 +01002409 si_ic(si)->flags |= CF_READ_NULL;
Willy Tarreau828824a2015-04-19 17:20:03 +02002410 goto out;
Emeric Brun2b920a12010-09-23 18:30:22 +02002411 }
2412 }
2413 }
2414out:
Willy Tarreau2bb4a962014-11-28 11:11:05 +01002415 si_oc(si)->flags |= CF_READ_DONTWAIT;
Emeric Brun80527f52017-06-19 17:46:37 +02002416
2417 if (curpeer)
Christopher Faulet2a944ee2017-11-07 10:42:54 +01002418 HA_SPIN_UNLOCK(PEER_LOCK, &curpeer->lock);
Emeric Brun2b920a12010-09-23 18:30:22 +02002419 return;
2420}
2421
Willy Tarreau30576452015-04-13 13:50:30 +02002422static struct applet peer_applet = {
Willy Tarreau3fdb3662012-11-12 00:42:33 +01002423 .obj_type = OBJ_TYPE_APPLET,
Willy Tarreaub24281b2011-02-13 13:16:36 +01002424 .name = "<PEER>", /* used for logging */
2425 .fct = peer_io_handler,
Aman Gupta9a13e842012-04-02 18:57:53 -07002426 .release = peer_session_release,
Willy Tarreaub24281b2011-02-13 13:16:36 +01002427};
Emeric Brun2b920a12010-09-23 18:30:22 +02002428
Emeric Brun9ef2ad72019-04-02 17:22:01 +02002429
Emeric Brun2b920a12010-09-23 18:30:22 +02002430/*
2431 * Use this function to force a close of a peer session
2432 */
Emeric Brun9ef2ad72019-04-02 17:22:01 +02002433static void peer_session_forceshutdown(struct peer *peer)
Emeric Brun2b920a12010-09-23 18:30:22 +02002434{
Emeric Brun9ef2ad72019-04-02 17:22:01 +02002435 struct appctx *appctx = peer->appctx;
2436
Frédéric Lécaille5df11902017-06-13 16:39:57 +02002437 /* Note that the peer sessions which have just been created
2438 * (->st0 == PEER_SESS_ST_CONNECT) must not
2439 * be shutdown, if not, the TCP session will never be closed
2440 * and stay in CLOSE_WAIT state after having been closed by
2441 * the remote side.
2442 */
2443 if (!appctx || appctx->st0 == PEER_SESS_ST_CONNECT)
Willy Tarreau7b4b4992013-12-01 09:15:12 +01002444 return;
2445
Willy Tarreau81bc3b02016-10-31 17:37:39 +01002446 if (appctx->applet != &peer_applet)
2447 return;
2448
Emeric Brun9ef2ad72019-04-02 17:22:01 +02002449 __peer_session_deinit(peer);
2450
Willy Tarreaue4d927a2013-12-01 12:47:35 +01002451 appctx->st0 = PEER_SESS_ST_END;
Willy Tarreau78c0c502016-10-31 17:32:20 +01002452 appctx_wakeup(appctx);
Emeric Brun2b920a12010-09-23 18:30:22 +02002453}
2454
Willy Tarreau91d96282015-03-13 15:47:26 +01002455/* Pre-configures a peers frontend to accept incoming connections */
2456void peers_setup_frontend(struct proxy *fe)
2457{
2458 fe->last_change = now.tv_sec;
Frédéric Lécaillec06b5d42018-04-26 10:06:41 +02002459 fe->cap = PR_CAP_FE | PR_CAP_BE;
Willy Tarreau91d96282015-03-13 15:47:26 +01002460 fe->maxconn = 0;
2461 fe->conn_retries = CONN_RETRIES;
2462 fe->timeout.client = MS_TO_TICKS(5000);
Willy Tarreaud1d48d42015-03-13 16:15:46 +01002463 fe->accept = frontend_accept;
Willy Tarreauf87ab942015-03-13 15:55:16 +01002464 fe->default_target = &peer_applet.obj_type;
Willy Tarreau91d96282015-03-13 15:47:26 +01002465 fe->options2 |= PR_O2_INDEPSTR | PR_O2_SMARTCON | PR_O2_SMARTACC;
Willy Tarreau0fca4832015-05-01 19:12:05 +02002466 fe->bind_proc = 0; /* will be filled by users */
Willy Tarreau91d96282015-03-13 15:47:26 +01002467}
2468
Emeric Brun2b920a12010-09-23 18:30:22 +02002469/*
Willy Tarreaubd55e312010-11-11 10:55:09 +01002470 * Create a new peer session in assigned state (connect will start automatically)
Emeric Brun2b920a12010-09-23 18:30:22 +02002471 */
Willy Tarreau9df94c22016-10-31 18:42:52 +01002472static struct appctx *peer_session_create(struct peers *peers, struct peer *peer)
Emeric Brun2b920a12010-09-23 18:30:22 +02002473{
Willy Tarreau04b92862017-09-15 11:01:04 +02002474 struct proxy *p = peers->peers_fe; /* attached frontend */
Willy Tarreau7b4b4992013-12-01 09:15:12 +01002475 struct appctx *appctx;
Willy Tarreau15b5e142015-04-04 14:38:25 +02002476 struct session *sess;
Willy Tarreau87b09662015-04-03 00:22:06 +02002477 struct stream *s;
Willy Tarreau32e3c6a2013-10-11 19:34:20 +02002478 struct connection *conn;
Olivier Houchard9aaf7782017-09-13 18:30:23 +02002479 struct conn_stream *cs;
Emeric Brun2b920a12010-09-23 18:30:22 +02002480
Frédéric Lécaille54bff832019-03-26 10:25:20 +01002481 peer->reconnect = tick_add(now_ms, MS_TO_TICKS(PEER_RECONNECT_TIMEOUT));
Frédéric Lécaille645635d2019-02-11 17:49:39 +01002482 peer->heartbeat = tick_add(now_ms, MS_TO_TICKS(PEER_HEARTBEAT_TIMEOUT));
Emeric Brunb3971ab2015-05-12 18:49:09 +02002483 peer->statuscode = PEER_SESS_SC_CONNECTCODE;
Willy Tarreaud990baf2015-04-05 00:32:03 +02002484 s = NULL;
2485
Emeric Brun1138fd02017-06-19 12:38:55 +02002486 appctx = appctx_new(&peer_applet, tid_bit);
Willy Tarreaud990baf2015-04-05 00:32:03 +02002487 if (!appctx)
2488 goto out_close;
2489
2490 appctx->st0 = PEER_SESS_ST_CONNECT;
Emeric Brunb3971ab2015-05-12 18:49:09 +02002491 appctx->ctx.peers.ptr = (void *)peer;
Willy Tarreaud990baf2015-04-05 00:32:03 +02002492
Willy Tarreau04b92862017-09-15 11:01:04 +02002493 sess = session_new(p, NULL, &appctx->obj_type);
Willy Tarreau15b5e142015-04-04 14:38:25 +02002494 if (!sess) {
Christopher Faulet767a84b2017-11-24 16:50:31 +01002495 ha_alert("out of memory in peer_session_create().\n");
Willy Tarreaud990baf2015-04-05 00:32:03 +02002496 goto out_free_appctx;
Emeric Brun2b920a12010-09-23 18:30:22 +02002497 }
2498
Willy Tarreau87787ac2017-08-28 16:22:54 +02002499 if ((s = stream_new(sess, &appctx->obj_type)) == NULL) {
Christopher Faulet767a84b2017-11-24 16:50:31 +01002500 ha_alert("Failed to initialize stream in peer_session_create().\n");
Willy Tarreau87787ac2017-08-28 16:22:54 +02002501 goto out_free_sess;
Willy Tarreau8baf9062015-04-05 00:46:36 +02002502 }
2503
Willy Tarreau342bfb12015-04-05 01:35:34 +02002504 /* The tasks below are normally what is supposed to be done by
2505 * fe->accept().
2506 */
Willy Tarreaue7dff022015-04-03 01:14:29 +02002507 s->flags = SF_ASSIGNED|SF_ADDR_SET;
Emeric Brun2b920a12010-09-23 18:30:22 +02002508
Willy Tarreau6e2979c2015-04-27 13:21:15 +02002509 /* applet is waiting for data */
Willy Tarreau0cd3bd62018-11-06 18:46:37 +01002510 si_cant_get(&s->si[0]);
Willy Tarreau6e2979c2015-04-27 13:21:15 +02002511 appctx_wakeup(appctx);
2512
Willy Tarreau3ed35ef2013-10-24 11:51:38 +02002513 /* initiate an outgoing connection */
Willy Tarreaudbd02672017-12-06 17:39:53 +01002514 s->si[1].flags |= SI_FL_NOLINGER;
Willy Tarreau3ed35ef2013-10-24 11:51:38 +02002515 si_set_state(&s->si[1], SI_ST_ASS);
Willy Tarreau3ed35ef2013-10-24 11:51:38 +02002516
Willy Tarreau32e3c6a2013-10-11 19:34:20 +02002517 /* automatically prepare the stream interface to connect to the
Willy Tarreaub363a1f2013-10-01 10:45:07 +02002518 * pre-initialized connection in si->conn.
2519 */
Willy Tarreau32e3c6a2013-10-11 19:34:20 +02002520 if (unlikely((conn = conn_new()) == NULL))
Willy Tarreau8baf9062015-04-05 00:46:36 +02002521 goto out_free_strm;
Willy Tarreau32e3c6a2013-10-11 19:34:20 +02002522
Olivier Houchard9aaf7782017-09-13 18:30:23 +02002523 if (unlikely((cs = cs_new(conn)) == NULL))
2524 goto out_free_conn;
2525
Frédéric Lécaille1055e682018-04-26 14:35:21 +02002526 conn->target = s->target = peer_session_target(peer, s);
Willy Tarreaube373152018-09-06 11:45:30 +02002527 memcpy(&conn->addr.to, &peer->addr, sizeof(conn->addr.to));
2528
Frédéric Lécaille1055e682018-04-26 14:35:21 +02002529 conn_prepare(conn, peer->proto, peer_xprt(peer));
Olivier Houchardef60ff32019-01-29 19:00:33 +01002530 if (conn_install_mux(conn, &mux_pt_ops, cs, s->be, NULL) < 0)
2531 goto out_free_cs;
Olivier Houchard9aaf7782017-09-13 18:30:23 +02002532 si_attach_cs(&s->si[1], cs);
Willy Tarreau32e3c6a2013-10-11 19:34:20 +02002533
Emeric Brun2b920a12010-09-23 18:30:22 +02002534 s->do_log = NULL;
Emeric Brun2b920a12010-09-23 18:30:22 +02002535 s->uniq_id = 0;
Emeric Brun2b920a12010-09-23 18:30:22 +02002536
Willy Tarreau22ec1ea2014-11-27 20:45:39 +01002537 s->res.flags |= CF_READ_DONTWAIT;
Willy Tarreau696a2912014-11-24 11:36:57 +01002538
Emeric Brunb3971ab2015-05-12 18:49:09 +02002539 peer->appctx = appctx;
Willy Tarreau87787ac2017-08-28 16:22:54 +02002540 task_wakeup(s->task, TASK_WOKEN_INIT);
Olivier Houcharded879892019-03-08 18:53:43 +01002541 _HA_ATOMIC_ADD(&active_peers, 1);
Willy Tarreau9df94c22016-10-31 18:42:52 +01002542 return appctx;
Emeric Brun2b920a12010-09-23 18:30:22 +02002543
2544 /* Error unrolling */
Olivier Houchardef60ff32019-01-29 19:00:33 +01002545out_free_cs:
2546 cs_free(cs);
Olivier Houchard9aaf7782017-09-13 18:30:23 +02002547 out_free_conn:
2548 conn_free(conn);
Willy Tarreau15b5e142015-04-04 14:38:25 +02002549 out_free_strm:
Emeric Brun2b920a12010-09-23 18:30:22 +02002550 LIST_DEL(&s->list);
Willy Tarreaubafbe012017-11-24 17:34:44 +01002551 pool_free(pool_head_stream, s);
Willy Tarreau15b5e142015-04-04 14:38:25 +02002552 out_free_sess:
Willy Tarreau11c36242015-04-04 15:54:03 +02002553 session_free(sess);
Willy Tarreaud990baf2015-04-05 00:32:03 +02002554 out_free_appctx:
2555 appctx_free(appctx);
Emeric Brun2b920a12010-09-23 18:30:22 +02002556 out_close:
Willy Tarreaub21d08e2016-10-31 17:46:57 +01002557 return NULL;
Emeric Brun2b920a12010-09-23 18:30:22 +02002558}
2559
2560/*
Frédéric Lécaille645635d2019-02-11 17:49:39 +01002561 * Task processing function to manage re-connect, peer session
2562 * tasks wakeup on local update and heartbeat.
Emeric Brun2b920a12010-09-23 18:30:22 +02002563 */
Olivier Houchard9f6af332018-05-25 14:04:04 +02002564static struct task *process_peer_sync(struct task * task, void *context, unsigned short state)
Emeric Brun2b920a12010-09-23 18:30:22 +02002565{
Olivier Houchard9f6af332018-05-25 14:04:04 +02002566 struct peers *peers = context;
Emeric Brunb3971ab2015-05-12 18:49:09 +02002567 struct peer *ps;
2568 struct shared_table *st;
Emeric Brun2b920a12010-09-23 18:30:22 +02002569
2570 task->expire = TICK_ETERNITY;
2571
Emeric Brunb3971ab2015-05-12 18:49:09 +02002572 if (!peers->peers_fe) {
Willy Tarreau46dc1ca2015-05-01 18:32:13 +02002573 /* this one was never started, kill it */
Emeric Brunb3971ab2015-05-12 18:49:09 +02002574 signal_unregister_handler(peers->sighandler);
Olivier Houchard3f795f72019-04-17 22:51:06 +02002575 task_destroy(peers->sync_task);
Willy Tarreau37bb7be2015-09-21 15:24:58 +02002576 peers->sync_task = NULL;
Willy Tarreau46dc1ca2015-05-01 18:32:13 +02002577 return NULL;
2578 }
2579
Emeric Brun80527f52017-06-19 17:46:37 +02002580 /* Acquire lock for all peers of the section */
2581 for (ps = peers->remote; ps; ps = ps->next)
Christopher Faulet2a944ee2017-11-07 10:42:54 +01002582 HA_SPIN_LOCK(PEER_LOCK, &ps->lock);
Emeric Brun80527f52017-06-19 17:46:37 +02002583
Emeric Brun2b920a12010-09-23 18:30:22 +02002584 if (!stopping) {
2585 /* Normal case (not soft stop)*/
Emeric Brunb3971ab2015-05-12 18:49:09 +02002586
2587 if (((peers->flags & PEERS_RESYNC_STATEMASK) == PEERS_RESYNC_FROMLOCAL) &&
2588 (!nb_oldpids || tick_is_expired(peers->resync_timeout, now_ms)) &&
2589 !(peers->flags & PEERS_F_RESYNC_ASSIGN)) {
Emeric Brun2b920a12010-09-23 18:30:22 +02002590 /* Resync from local peer needed
2591 no peer was assigned for the lesson
2592 and no old local peer found
2593 or resync timeout expire */
2594
2595 /* flag no more resync from local, to try resync from remotes */
Emeric Brunb3971ab2015-05-12 18:49:09 +02002596 peers->flags |= PEERS_F_RESYNC_LOCAL;
Emeric Brun2b920a12010-09-23 18:30:22 +02002597
2598 /* reschedule a resync */
Frédéric Lécaille54bff832019-03-26 10:25:20 +01002599 peers->resync_timeout = tick_add(now_ms, MS_TO_TICKS(PEER_RESYNC_TIMEOUT));
Emeric Brun2b920a12010-09-23 18:30:22 +02002600 }
2601
2602 /* For each session */
Emeric Brunb3971ab2015-05-12 18:49:09 +02002603 for (ps = peers->remote; ps; ps = ps->next) {
Emeric Brun2b920a12010-09-23 18:30:22 +02002604 /* For each remote peers */
Emeric Brunb3971ab2015-05-12 18:49:09 +02002605 if (!ps->local) {
Willy Tarreau9df94c22016-10-31 18:42:52 +01002606 if (!ps->appctx) {
2607 /* no active peer connection */
Emeric Brun2b920a12010-09-23 18:30:22 +02002608 if (ps->statuscode == 0 ||
Willy Tarreaue4d927a2013-12-01 12:47:35 +01002609 ((ps->statuscode == PEER_SESS_SC_CONNECTCODE ||
Willy Tarreaub4e34da2015-05-20 10:39:04 +02002610 ps->statuscode == PEER_SESS_SC_SUCCESSCODE ||
Willy Tarreaue4d927a2013-12-01 12:47:35 +01002611 ps->statuscode == PEER_SESS_SC_CONNECTEDCODE) &&
Emeric Brun2b920a12010-09-23 18:30:22 +02002612 tick_is_expired(ps->reconnect, now_ms))) {
2613 /* connection never tried
Willy Tarreau9df94c22016-10-31 18:42:52 +01002614 * or previous peer connection established with success
2615 * or previous peer connection failed while connecting
Emeric Brun2b920a12010-09-23 18:30:22 +02002616 * and reconnection timer is expired */
2617
2618 /* retry a connect */
Willy Tarreau9df94c22016-10-31 18:42:52 +01002619 ps->appctx = peer_session_create(peers, ps);
Emeric Brun2b920a12010-09-23 18:30:22 +02002620 }
Willy Tarreaub4e34da2015-05-20 10:39:04 +02002621 else if (!tick_is_expired(ps->reconnect, now_ms)) {
Emeric Brun2b920a12010-09-23 18:30:22 +02002622 /* If previous session failed during connection
2623 * but reconnection timer is not expired */
2624
2625 /* reschedule task for reconnect */
2626 task->expire = tick_first(task->expire, ps->reconnect);
2627 }
2628 /* else do nothing */
Willy Tarreau9df94c22016-10-31 18:42:52 +01002629 } /* !ps->appctx */
Willy Tarreaue4d927a2013-12-01 12:47:35 +01002630 else if (ps->statuscode == PEER_SESS_SC_SUCCESSCODE) {
Willy Tarreau9df94c22016-10-31 18:42:52 +01002631 /* current peer connection is active and established */
Emeric Brunb3971ab2015-05-12 18:49:09 +02002632 if (((peers->flags & PEERS_RESYNC_STATEMASK) == PEERS_RESYNC_FROMREMOTE) &&
2633 !(peers->flags & PEERS_F_RESYNC_ASSIGN) &&
Emeric Brun2b920a12010-09-23 18:30:22 +02002634 !(ps->flags & PEER_F_LEARN_NOTUP2DATE)) {
2635 /* Resync from a remote is needed
2636 * and no peer was assigned for lesson
2637 * and current peer may be up2date */
2638
2639 /* assign peer for the lesson */
2640 ps->flags |= PEER_F_LEARN_ASSIGN;
Emeric Brunb3971ab2015-05-12 18:49:09 +02002641 peers->flags |= PEERS_F_RESYNC_ASSIGN;
Emeric Brun2b920a12010-09-23 18:30:22 +02002642
Willy Tarreau9df94c22016-10-31 18:42:52 +01002643 /* wake up peer handler to handle a request of resync */
Willy Tarreaue5843b32015-04-27 18:40:14 +02002644 appctx_wakeup(ps->appctx);
Emeric Brun2b920a12010-09-23 18:30:22 +02002645 }
Emeric Brunb3971ab2015-05-12 18:49:09 +02002646 else {
Frédéric Lécaille645635d2019-02-11 17:49:39 +01002647 int update_to_push = 0;
2648
Emeric Brunb3971ab2015-05-12 18:49:09 +02002649 /* Awake session if there is data to push */
2650 for (st = ps->tables; st ; st = st->next) {
2651 if ((int)(st->last_pushed - st->table->localupdate) < 0) {
Willy Tarreau9df94c22016-10-31 18:42:52 +01002652 /* wake up the peer handler to push local updates */
Frédéric Lécaille645635d2019-02-11 17:49:39 +01002653 update_to_push = 1;
Frédéric Lécaille045e0d42019-03-21 11:12:32 +01002654 /* There is no need to send a heartbeat message
2655 * when some updates must be pushed. The remote
2656 * peer will consider <ps> peer as alive when it will
2657 * receive these updates.
2658 */
Frédéric Lécaille645635d2019-02-11 17:49:39 +01002659 ps->flags &= ~PEER_F_HEARTBEAT;
Frédéric Lécaille045e0d42019-03-21 11:12:32 +01002660 /* Re-schedule another one later. */
Frédéric Lécaille645635d2019-02-11 17:49:39 +01002661 ps->heartbeat = tick_add(now_ms, MS_TO_TICKS(PEER_HEARTBEAT_TIMEOUT));
Frédéric Lécaille045e0d42019-03-21 11:12:32 +01002662 /* We are going to send updates, let's ensure we will
2663 * come back to send heartbeat messages or to reconnect.
2664 */
2665 task->expire = tick_first(ps->reconnect, ps->heartbeat);
Emeric Brunb3971ab2015-05-12 18:49:09 +02002666 appctx_wakeup(ps->appctx);
2667 break;
2668 }
2669 }
Frédéric Lécaille045e0d42019-03-21 11:12:32 +01002670 /* When there are updates to send we do not reconnect
2671 * and do not send heartbeat message either.
2672 */
2673 if (!update_to_push) {
2674 if (tick_is_expired(ps->reconnect, now_ms)) {
2675 if (ps->flags & PEER_F_ALIVE) {
2676 /* This peer was alive during a 'reconnect' period.
2677 * Flag it as not alive again for the next period.
2678 */
2679 ps->flags &= ~PEER_F_ALIVE;
Frédéric Lécaille54bff832019-03-26 10:25:20 +01002680 ps->reconnect = tick_add(now_ms, MS_TO_TICKS(PEER_RECONNECT_TIMEOUT));
Frédéric Lécaille045e0d42019-03-21 11:12:32 +01002681 }
2682 else {
2683 ps->reconnect = tick_add(now_ms, MS_TO_TICKS(50 + random() % 2000));
Emeric Brun9ef2ad72019-04-02 17:22:01 +02002684 peer_session_forceshutdown(ps);
Frédéric Lécaille045e0d42019-03-21 11:12:32 +01002685 }
2686 }
2687 else if (tick_is_expired(ps->heartbeat, now_ms)) {
2688 ps->heartbeat = tick_add(now_ms, MS_TO_TICKS(PEER_HEARTBEAT_TIMEOUT));
2689 ps->flags |= PEER_F_HEARTBEAT;
2690 appctx_wakeup(ps->appctx);
2691 }
Frédéric Lécailleb7405c12019-03-27 14:32:39 +01002692 task->expire = tick_first(ps->reconnect, ps->heartbeat);
Frédéric Lécaille645635d2019-02-11 17:49:39 +01002693 }
Emeric Brun2b920a12010-09-23 18:30:22 +02002694 }
2695 /* else do nothing */
2696 } /* SUCCESSCODE */
2697 } /* !ps->peer->local */
2698 } /* for */
2699
2700 /* Resync from remotes expired: consider resync is finished */
Emeric Brunb3971ab2015-05-12 18:49:09 +02002701 if (((peers->flags & PEERS_RESYNC_STATEMASK) == PEERS_RESYNC_FROMREMOTE) &&
2702 !(peers->flags & PEERS_F_RESYNC_ASSIGN) &&
2703 tick_is_expired(peers->resync_timeout, now_ms)) {
Emeric Brun2b920a12010-09-23 18:30:22 +02002704 /* Resync from remote peer needed
2705 * no peer was assigned for the lesson
2706 * and resync timeout expire */
2707
2708 /* flag no more resync from remote, consider resync is finished */
Emeric Brunb3971ab2015-05-12 18:49:09 +02002709 peers->flags |= PEERS_F_RESYNC_REMOTE;
Emeric Brun2b920a12010-09-23 18:30:22 +02002710 }
2711
Emeric Brunb3971ab2015-05-12 18:49:09 +02002712 if ((peers->flags & PEERS_RESYNC_STATEMASK) != PEERS_RESYNC_FINISHED) {
Emeric Brun2b920a12010-09-23 18:30:22 +02002713 /* Resync not finished*/
Frédéric Lécaille5d6e5f82017-05-29 13:47:16 +02002714 /* reschedule task to resync timeout if not expired, to ended resync if needed */
2715 if (!tick_is_expired(peers->resync_timeout, now_ms))
2716 task->expire = tick_first(task->expire, peers->resync_timeout);
Emeric Brun2b920a12010-09-23 18:30:22 +02002717 }
2718 } /* !stopping */
2719 else {
2720 /* soft stop case */
Willy Tarreau086735a2018-11-05 15:09:47 +01002721 if (state & TASK_WOKEN_SIGNAL) {
Joseph Herlant82b2f542018-11-15 12:19:14 -08002722 /* We've just received the signal */
Emeric Brunb3971ab2015-05-12 18:49:09 +02002723 if (!(peers->flags & PEERS_F_DONOTSTOP)) {
Emeric Brun2b920a12010-09-23 18:30:22 +02002724 /* add DO NOT STOP flag if not present */
Olivier Houcharded879892019-03-08 18:53:43 +01002725 _HA_ATOMIC_ADD(&jobs, 1);
Emeric Brunb3971ab2015-05-12 18:49:09 +02002726 peers->flags |= PEERS_F_DONOTSTOP;
2727 ps = peers->local;
2728 for (st = ps->tables; st ; st = st->next)
2729 st->table->syncing++;
Emeric Brun2b920a12010-09-23 18:30:22 +02002730 }
2731
2732 /* disconnect all connected peers */
Emeric Brunb3971ab2015-05-12 18:49:09 +02002733 for (ps = peers->remote; ps; ps = ps->next) {
Emeric Brun80527f52017-06-19 17:46:37 +02002734 /* we're killing a connection, we must apply a random delay before
2735 * retrying otherwise the other end will do the same and we can loop
2736 * for a while.
2737 */
2738 ps->reconnect = tick_add(now_ms, MS_TO_TICKS(50 + random() % 2000));
Willy Tarreau9df94c22016-10-31 18:42:52 +01002739 if (ps->appctx) {
Emeric Brun9ef2ad72019-04-02 17:22:01 +02002740 peer_session_forceshutdown(ps);
Emeric Brun2b920a12010-09-23 18:30:22 +02002741 }
2742 }
2743 }
Emeric Brun2b920a12010-09-23 18:30:22 +02002744
Emeric Brunb3971ab2015-05-12 18:49:09 +02002745 ps = peers->local;
Emeric Brun2b920a12010-09-23 18:30:22 +02002746 if (ps->flags & PEER_F_TEACH_COMPLETE) {
Emeric Brunb3971ab2015-05-12 18:49:09 +02002747 if (peers->flags & PEERS_F_DONOTSTOP) {
Emeric Brun2b920a12010-09-23 18:30:22 +02002748 /* resync of new process was complete, current process can die now */
Olivier Houcharded879892019-03-08 18:53:43 +01002749 _HA_ATOMIC_SUB(&jobs, 1);
Emeric Brunb3971ab2015-05-12 18:49:09 +02002750 peers->flags &= ~PEERS_F_DONOTSTOP;
2751 for (st = ps->tables; st ; st = st->next)
2752 st->table->syncing--;
Emeric Brun2b920a12010-09-23 18:30:22 +02002753 }
2754 }
Willy Tarreau9df94c22016-10-31 18:42:52 +01002755 else if (!ps->appctx) {
2756 /* If there's no active peer connection */
Emeric Brun2b920a12010-09-23 18:30:22 +02002757 if (ps->statuscode == 0 ||
Willy Tarreaue4d927a2013-12-01 12:47:35 +01002758 ps->statuscode == PEER_SESS_SC_SUCCESSCODE ||
2759 ps->statuscode == PEER_SESS_SC_CONNECTEDCODE ||
2760 ps->statuscode == PEER_SESS_SC_TRYAGAIN) {
Emeric Brun2b920a12010-09-23 18:30:22 +02002761 /* connection never tried
Willy Tarreau9df94c22016-10-31 18:42:52 +01002762 * or previous peer connection was successfully established
2763 * or previous tcp connect succeeded but init state incomplete
Emeric Brun2b920a12010-09-23 18:30:22 +02002764 * or during previous connect, peer replies a try again statuscode */
2765
2766 /* connect to the peer */
Emeric Brunb3971ab2015-05-12 18:49:09 +02002767 peer_session_create(peers, ps);
Emeric Brun2b920a12010-09-23 18:30:22 +02002768 }
2769 else {
2770 /* Other error cases */
Emeric Brunb3971ab2015-05-12 18:49:09 +02002771 if (peers->flags & PEERS_F_DONOTSTOP) {
Emeric Brun2b920a12010-09-23 18:30:22 +02002772 /* unable to resync new process, current process can die now */
Olivier Houcharded879892019-03-08 18:53:43 +01002773 _HA_ATOMIC_SUB(&jobs, 1);
Emeric Brunb3971ab2015-05-12 18:49:09 +02002774 peers->flags &= ~PEERS_F_DONOTSTOP;
2775 for (st = ps->tables; st ; st = st->next)
2776 st->table->syncing--;
Emeric Brun2b920a12010-09-23 18:30:22 +02002777 }
2778 }
2779 }
Emeric Brunb3971ab2015-05-12 18:49:09 +02002780 else if (ps->statuscode == PEER_SESS_SC_SUCCESSCODE ) {
Willy Tarreau9df94c22016-10-31 18:42:52 +01002781 /* current peer connection is active and established
2782 * wake up all peer handlers to push remaining local updates */
Emeric Brunb3971ab2015-05-12 18:49:09 +02002783 for (st = ps->tables; st ; st = st->next) {
2784 if ((int)(st->last_pushed - st->table->localupdate) < 0) {
Emeric Brunb3971ab2015-05-12 18:49:09 +02002785 appctx_wakeup(ps->appctx);
2786 break;
2787 }
2788 }
Emeric Brun2b920a12010-09-23 18:30:22 +02002789 }
2790 } /* stopping */
Emeric Brun80527f52017-06-19 17:46:37 +02002791
2792 /* Release lock for all peers of the section */
2793 for (ps = peers->remote; ps; ps = ps->next)
Christopher Faulet2a944ee2017-11-07 10:42:54 +01002794 HA_SPIN_UNLOCK(PEER_LOCK, &ps->lock);
Emeric Brun80527f52017-06-19 17:46:37 +02002795
Emeric Brun2b920a12010-09-23 18:30:22 +02002796 /* Wakeup for re-connect */
2797 return task;
2798}
2799
Emeric Brunb3971ab2015-05-12 18:49:09 +02002800
Emeric Brun2b920a12010-09-23 18:30:22 +02002801/*
Willy Tarreaud9443442018-10-15 11:18:03 +02002802 * returns 0 in case of error.
Emeric Brun2b920a12010-09-23 18:30:22 +02002803 */
Willy Tarreaud9443442018-10-15 11:18:03 +02002804int peers_init_sync(struct peers *peers)
Emeric Brun2b920a12010-09-23 18:30:22 +02002805{
Emeric Brun2b920a12010-09-23 18:30:22 +02002806 struct peer * curpeer;
Emeric Brun2b920a12010-09-23 18:30:22 +02002807
Emeric Brun2b920a12010-09-23 18:30:22 +02002808 for (curpeer = peers->remote; curpeer; curpeer = curpeer->next) {
Emeric Brun2b920a12010-09-23 18:30:22 +02002809 peers->peers_fe->maxconn += 3;
2810 }
2811
Emeric Brunc60def82017-09-27 14:59:38 +02002812 peers->sync_task = task_new(MAX_THREADS_MASK);
Willy Tarreaud9443442018-10-15 11:18:03 +02002813 if (!peers->sync_task)
2814 return 0;
2815
Emeric Brunb3971ab2015-05-12 18:49:09 +02002816 peers->sync_task->process = process_peer_sync;
Emeric Brunb3971ab2015-05-12 18:49:09 +02002817 peers->sync_task->context = (void *)peers;
2818 peers->sighandler = signal_register_task(0, peers->sync_task, 0);
2819 task_wakeup(peers->sync_task, TASK_WOKEN_INIT);
Willy Tarreaud9443442018-10-15 11:18:03 +02002820 return 1;
Emeric Brunb3971ab2015-05-12 18:49:09 +02002821}
2822
Frédéric Lécaille74167b22019-05-28 19:02:42 +02002823/*
2824 * Allocate a cache a dictionary entries used upon transmission.
2825 */
2826struct dcache_tx *new_dcache_tx(size_t max_entries)
2827{
2828 struct dcache_tx *d;
2829
2830 d = malloc(sizeof *d);
2831 if (!d)
2832 return NULL;
2833
2834 d->lru_key = 0;
2835 d->keys = EB_ROOT_UNIQUE;
2836 d->values = EB_ROOT_UNIQUE;
2837 d->max_entries = max_entries;
2838
2839 return d;
2840}
2841
2842/*
2843 * Allocate a cache of dictionary entries with <name> as name and <max_entries>
2844 * as maximum of entries.
2845 * Return the dictionay cache if succeeded, NULL if not.
2846 * Must be deallocated calling free_dcache().
2847 */
2848struct dcache *new_dcache(size_t max_entries)
2849{
2850 struct dcache_tx *dc_tx;
2851 struct dcache *dc;
2852 struct dcache_rx *dc_rx;
2853
2854 dc = calloc(1, sizeof *dc);
2855 dc_tx = new_dcache_tx(max_entries);
2856 dc_rx = calloc(max_entries, sizeof *dc_rx);
2857 if (!dc || !dc_tx || !dc_rx)
2858 goto err;
2859
2860 dc->tx = dc_tx;
2861 dc->rx = dc_rx;
2862 dc->max_entries = max_entries;
2863
2864 return dc;
2865
2866 err:
2867 free(dc);
2868 free(dc_tx);
2869 free(dc_rx);
2870 return NULL;
2871}
2872
2873/*
2874 * Deallocate a cache of dictionary entries.
2875 */
2876void free_dcache(struct dcache *dc)
2877{
2878 free(dc->tx); dc->tx = NULL;
2879 free(dc->rx); dc->rx = NULL;
2880 free(dc);
2881}
2882
2883/*
2884 * Look for the dictionary entry with <k> as key in <d> cache of dictionary entries
2885 * used upon transmission.
2886 * Return the entry if found, NULL if not.
2887 */
2888struct dcache_tx_entry *dcache_tx_lookup_key(struct dcache_tx *d, unsigned int k)
2889{
2890 struct dcache_tx_entry *de;
2891 struct eb32_node *node;
2892
2893 de = NULL;
2894 node = eb32_lookup(&d->keys, k);
2895 if (node)
2896 de = container_of(node, struct dcache_tx_entry, key);
2897
2898 return de;
2899}
2900
2901/*
2902 * Look for the dictionary entry with the value of <i> in <d> cache of dictionary
2903 * entries used upon transmission.
2904 * Return the entry if found, NULL if not.
2905 */
2906struct dcache_tx_entry *dcache_tx_lookup_value(struct dcache_tx *d,
2907 struct dcache_tx_entry *i)
2908{
2909 struct dcache_tx_entry *de;
2910 struct ebpt_node *node;
2911
2912 de = NULL;
2913 node = ebpt_lookup(&d->values, i->value.key);
2914 if (node)
2915 de = container_of(node, struct dcache_tx_entry, value);
2916
2917 return de;
2918}
2919
2920/*
2921 * Release the memory allocated for the cached dictionary entry <dce>.
2922 */
2923
2924static inline void free_dcache_tx_entry(struct dcache_tx_entry *dce)
2925{
2926 dce->value.key = NULL;
2927 dce->key.key = 0;
2928 free(dce);
2929}
2930
2931/*
2932 * Allocate a new dictionary entry with <s> as string value which is strdup()'ed.
2933 * Returns the new allocated entry if succeeded, NULL if not.
2934 */
2935struct dcache_tx_entry *new_dcache_tx_entry(unsigned int k, struct dict_entry *de)
2936{
2937 struct dcache_tx_entry *dce;
2938
2939 dce = calloc(1, sizeof *dce);
2940 if (!dce)
2941 return NULL;
2942
2943 dce->value.key = de;
2944 dce->key.key = k;
2945
2946 return dce;
2947}
2948
2949/*
2950 * Flush <dc> cache.
2951 * Always succeeds.
2952 */
2953static inline void flush_dcache(struct peer *peer)
2954{
2955 int i;
Frédéric Lécaille8d78fa72019-05-20 18:22:52 +02002956 struct dcache *dc = peer->dcache;
Frédéric Lécaille74167b22019-05-28 19:02:42 +02002957
2958 if (!eb_is_empty(&dc->tx->keys)) {
2959 struct eb32_node *node, *next;
2960
2961 for (node = eb32_first(&dc->tx->keys); node; node = next) {
2962 next = eb32_next(node);
2963 eb32_delete(node);
2964 }
2965 }
2966
2967 if (!eb_is_empty(&dc->tx->values)) {
2968 struct ebpt_node *node, *next;
2969
2970 for (node = ebpt_first(&dc->tx->values); node; node = next) {
2971 struct dcache_tx_entry *de;
2972
2973 next = ebpt_next(node);
2974 ebpt_delete(node);
2975 de = container_of(node, struct dcache_tx_entry, value);
2976 free_dcache_tx_entry(de);
2977 }
2978 }
2979
2980 for (i = 0; i < dc->max_entries; i++) {
2981 dc->rx[i].id = 0;
2982 dc->rx[i].de = NULL;
2983 }
2984}
2985
2986/*
2987 * Insert a dictionary entry in <dc> cache part used upon transmission (->tx)
2988 * with information provided by <i> dictionary cache entry (especially the value
2989 * to be inserted if not already). Return <i> if already present in the cache
2990 * or something different of <i> if not already in the cache, or NULL if something
2991 * failed.
2992 */
2993struct dcache_tx_entry *dcache_tx_insert(struct dcache *dc,
2994 struct dcache_tx_entry *i)
2995{
2996 struct dcache_tx *dc_tx;
2997 struct dcache_tx_entry *o;
2998
2999 dc_tx = dc->tx;
3000 o = dcache_tx_lookup_value(dc_tx, i);
3001 if (o) {
3002 /* Copy the key. */
3003 i->key.key = o->key.key;
3004 return i;
3005 }
3006
3007 o = dcache_tx_lookup_key(dc_tx, dc_tx->lru_key);
3008 if (o) {
3009 o->value.key = i->value.key;
3010 return o;
3011 }
3012
3013 o = new_dcache_tx_entry(dc_tx->lru_key, i->value.key);
3014 if (!o)
3015 return NULL;
3016
3017 dc_tx->lru_key = (dc_tx->lru_key + 1) & (dc_tx->max_entries - 1);
3018 eb32_insert(&dc_tx->keys, &o->key);
3019 ebpt_insert(&dc_tx->values, &o->value);
Emeric Brunb3971ab2015-05-12 18:49:09 +02003020
Frédéric Lécaille74167b22019-05-28 19:02:42 +02003021 return o;
3022}
Emeric Brunb3971ab2015-05-12 18:49:09 +02003023
3024/*
Frédéric Lécaille8d78fa72019-05-20 18:22:52 +02003025 * Allocate a dictionary cache for each peer of <peers> section.
3026 * Return 1 if succeeded, 0 if not.
3027 */
3028int peers_alloc_dcache(struct peers *peers)
3029{
3030 struct peer *p;
3031
3032 for (p = peers->remote; p; p = p->next) {
3033 p->dcache = new_dcache(PEER_STKT_CACHE_MAX_ENTRIES);
3034 if (!p->dcache)
3035 return 0;
3036 }
3037
3038 return 1;
3039}
3040
3041/*
Emeric Brunb3971ab2015-05-12 18:49:09 +02003042 * Function used to register a table for sync on a group of peers
3043 *
3044 */
3045void peers_register_table(struct peers *peers, struct stktable *table)
3046{
3047 struct shared_table *st;
3048 struct peer * curpeer;
3049 int id = 0;
3050
3051 for (curpeer = peers->remote; curpeer; curpeer = curpeer->next) {
Vincent Bernat02779b62016-04-03 13:48:43 +02003052 st = calloc(1,sizeof(*st));
Emeric Brunb3971ab2015-05-12 18:49:09 +02003053 st->table = table;
3054 st->next = curpeer->tables;
3055 if (curpeer->tables)
3056 id = curpeer->tables->local_id;
3057 st->local_id = id + 1;
3058
3059 curpeer->tables = st;
3060 }
3061
3062 table->sync_task = peers->sync_task;
Emeric Brun2b920a12010-09-23 18:30:22 +02003063}
3064
Frédéric Lécaille95679dc2019-04-15 10:25:27 +02003065/*
3066 * Parse the "show peers" command arguments.
3067 * Returns 0 if succeeded, 1 if not with the ->msg of the appctx set as
3068 * error message.
3069 */
3070static int cli_parse_show_peers(char **args, char *payload, struct appctx *appctx, void *private)
3071{
3072 appctx->ctx.cfgpeers.target = NULL;
3073
3074 if (*args[2]) {
3075 struct peers *p;
3076
3077 for (p = cfg_peers; p; p = p->next) {
3078 if (!strcmp(p->id, args[2])) {
3079 appctx->ctx.cfgpeers.target = p;
3080 break;
3081 }
3082 }
3083
3084 if (!p) {
3085 appctx->ctx.cli.severity = LOG_ERR;
3086 appctx->ctx.cli.msg = "No such peers\n";
3087 appctx->st0 = CLI_ST_PRINT;
3088 return 1;
3089 }
3090 }
3091
3092 return 0;
3093}
3094
3095/*
3096 * This function dumps the peer state information of <peers> "peers" section.
3097 * Returns 0 if the output buffer is full and needs to be called again, non-zero if not.
3098 * Dedicated to be called by cli_io_handler_show_peers() cli I/O handler.
3099 */
3100static int peers_dump_head(struct buffer *msg, struct stream_interface *si, struct peers *peers)
3101{
3102 struct tm tm;
3103
3104 get_localtime(peers->last_change, &tm);
Emeric Brun0bbec0f2019-04-18 11:39:43 +02003105 chunk_appendf(msg, "%p: [%02d/%s/%04d:%02d:%02d:%02d] id=%s state=%d flags=0x%x resync_timeout=%s task_calls=%u\n",
Frédéric Lécaille95679dc2019-04-15 10:25:27 +02003106 peers,
3107 tm.tm_mday, monthname[tm.tm_mon], tm.tm_year+1900,
3108 tm.tm_hour, tm.tm_min, tm.tm_sec,
3109 peers->id, peers->state, peers->flags,
3110 peers->resync_timeout ?
3111 tick_is_expired(peers->resync_timeout, now_ms) ? "<PAST>" :
3112 human_time(TICKS_TO_MS(peers->resync_timeout - now_ms),
Emeric Brun0bbec0f2019-04-18 11:39:43 +02003113 TICKS_TO_MS(1000)) : "<NEVER>",
3114 peers->sync_task ? peers->sync_task->calls : 0);
Frédéric Lécaille95679dc2019-04-15 10:25:27 +02003115
3116 if (ci_putchk(si_ic(si), msg) == -1) {
3117 si_rx_room_blk(si);
3118 return 0;
3119 }
3120
3121 return 1;
3122}
3123
3124/*
3125 * This function dumps <peer> state information.
3126 * Returns 0 if the output buffer is full and needs to be called again, non-zero
3127 * if not. Dedicated to be called by cli_io_handler_show_peers() cli I/O handler.
3128 */
3129static int peers_dump_peer(struct buffer *msg, struct stream_interface *si, struct peer *peer)
3130{
3131 struct connection *conn;
3132 char pn[INET6_ADDRSTRLEN];
3133 struct stream_interface *peer_si;
3134 struct stream *peer_s;
3135 struct appctx *appctx;
3136 struct shared_table *st;
3137
3138 addr_to_str(&peer->addr, pn, sizeof pn);
3139 chunk_appendf(msg, " %p: id=%s(%s) addr=%s:%d status=%s reconnect=%s confirm=%u\n",
3140 peer, peer->id,
3141 peer->local ? "local" : "remote",
3142 pn, get_host_port(&peer->addr),
3143 statuscode_str(peer->statuscode),
3144 peer->reconnect ?
3145 tick_is_expired(peer->reconnect, now_ms) ? "<PAST>" :
3146 human_time(TICKS_TO_MS(peer->reconnect - now_ms),
3147 TICKS_TO_MS(1000)) : "<NEVER>",
3148 peer->confirm);
3149
3150 chunk_appendf(&trash, " flags=0x%x", peer->flags);
3151
3152 appctx = peer->appctx;
3153 if (!appctx)
3154 goto end;
3155
Emeric Brun0bbec0f2019-04-18 11:39:43 +02003156 chunk_appendf(&trash, " appctx:%p st0=%d st1=%d task_calls=%u", appctx, appctx->st0, appctx->st1,
3157 appctx->t ? appctx->t->calls : 0);
Frédéric Lécaille95679dc2019-04-15 10:25:27 +02003158
3159 peer_si = peer->appctx->owner;
3160 if (!peer_si)
3161 goto end;
3162
3163 peer_s = si_strm(peer_si);
3164 if (!peer_s)
3165 goto end;
3166
3167 chunk_appendf(&trash, " state=%s", si_state_str(si_opposite(peer_si)->state));
3168
3169 conn = objt_conn(strm_orig(peer_s));
3170 if (conn)
3171 chunk_appendf(&trash, "\n xprt=%s", conn_get_xprt_name(conn));
3172
3173 switch (conn ? addr_to_str(&conn->addr.from, pn, sizeof(pn)) : AF_UNSPEC) {
3174 case AF_INET:
3175 case AF_INET6:
3176 chunk_appendf(&trash, " src=%s:%d", pn, get_host_port(&conn->addr.from));
3177 break;
3178 case AF_UNIX:
3179 chunk_appendf(&trash, " src=unix:%d", strm_li(peer_s)->luid);
3180 break;
3181 }
3182
3183 if (conn)
3184 conn_get_to_addr(conn);
3185
3186 switch (conn ? addr_to_str(&conn->addr.to, pn, sizeof(pn)) : AF_UNSPEC) {
3187 case AF_INET:
3188 case AF_INET6:
3189 chunk_appendf(&trash, " addr=%s:%d", pn, get_host_port(&conn->addr.to));
3190 break;
3191 case AF_UNIX:
3192 chunk_appendf(&trash, " addr=unix:%d", strm_li(peer_s)->luid);
3193 break;
3194 }
3195
3196 if (peer->remote_table)
3197 chunk_appendf(&trash, "\n remote_table:%p id=%s local_id=%d remote_id=%d",
3198 peer->remote_table,
3199 peer->remote_table->table->id,
3200 peer->remote_table->local_id,
3201 peer->remote_table->remote_id);
3202
3203 if (peer->last_local_table)
3204 chunk_appendf(&trash, "\n last_local_table:%p id=%s local_id=%d remote_id=%d",
3205 peer->last_local_table,
3206 peer->last_local_table->table->id,
3207 peer->last_local_table->local_id,
3208 peer->last_local_table->remote_id);
3209
3210 if (peer->tables) {
3211 chunk_appendf(&trash, "\n shared tables:");
3212 for (st = peer->tables; st; st = st->next) {
Frédéric Lécaille62b0b0b2019-05-29 16:20:41 +02003213 int i, count;
Frédéric Lécaille95679dc2019-04-15 10:25:27 +02003214 struct stktable *t;
Frédéric Lécaille62b0b0b2019-05-29 16:20:41 +02003215 struct dcache *dcache;
3216
Frédéric Lécaille95679dc2019-04-15 10:25:27 +02003217 t = st->table;
Frédéric Lécaille62b0b0b2019-05-29 16:20:41 +02003218 dcache = peer->dcache;
3219
Frédéric Lécaille95679dc2019-04-15 10:25:27 +02003220 chunk_appendf(&trash, "\n %p local_id=%d remote_id=%d "
3221 "flags=0x%x remote_data=0x%llx",
3222 st, st->local_id, st->remote_id,
3223 st->flags, (unsigned long long)st->remote_data);
3224 chunk_appendf(&trash, "\n last_acked=%u last_pushed=%u last_get=%u"
3225 " teaching_origin=%u update=%u",
3226 st->last_acked, st->last_pushed, st->last_get,
3227 st->teaching_origin, st->update);
3228 chunk_appendf(&trash, "\n table:%p id=%s update=%u localupdate=%u"
3229 " commitupdate=%u syncing=%u",
3230 t, t->id, t->update, t->localupdate, t->commitupdate, t->syncing);
Frédéric Lécaille62b0b0b2019-05-29 16:20:41 +02003231 if (!eb_is_empty(&dcache->tx->keys)) {
3232 struct eb32_node *node;
3233 struct dcache_tx_entry *de;
3234
3235 chunk_appendf(&trash, "\n TX dictionary cache:");
3236 count = 0;
3237 for (node = eb32_first(&dcache->tx->keys); node; node = eb32_next(node)) {
3238 if (!count++)
3239 chunk_appendf(&trash, "\n ");
3240 de = container_of(node, struct dcache_tx_entry, key);
3241 chunk_appendf(&trash, " %3u -> %s", node->key,
3242 (char *)((struct dict_entry *)(de->value.key))->value.key);
3243 count &= 0x3;
3244 }
3245 }
3246 chunk_appendf(&trash, "\n RX dictionary cache:");
3247 count = 0;
3248 for (i = 0; i < dcache->max_entries; i++) {
3249 if (!count++)
3250 chunk_appendf(&trash, "\n ");
3251 chunk_appendf(&trash, " %3u -> %s", i,
3252 dcache->rx[i].de ?
3253 (char *)dcache->rx[i].de->value.key : "-");
3254 count &= 0x3;
3255 }
Frédéric Lécaille95679dc2019-04-15 10:25:27 +02003256 }
3257 }
3258
3259 end:
3260 chunk_appendf(&trash, "\n");
3261 if (ci_putchk(si_ic(si), msg) == -1) {
3262 si_rx_room_blk(si);
3263 return 0;
3264 }
3265
3266 return 1;
3267}
3268
3269/*
3270 * This function dumps all the peers of "peers" section.
3271 * Returns 0 if the output buffer is full and needs to be called
3272 * again, non-zero if not. It proceeds in an isolated thread, so
3273 * there is no thread safety issue here.
3274 */
3275static int cli_io_handler_show_peers(struct appctx *appctx)
3276{
3277 int show_all;
3278 int ret = 0, first_peers = 1;
3279 struct stream_interface *si = appctx->owner;
3280
3281 thread_isolate();
3282
3283 show_all = !appctx->ctx.cfgpeers.target;
3284
3285 chunk_reset(&trash);
3286
3287 while (appctx->st2 != STAT_ST_FIN) {
3288 switch (appctx->st2) {
3289 case STAT_ST_INIT:
3290 if (show_all)
3291 appctx->ctx.cfgpeers.peers = cfg_peers;
3292 else
3293 appctx->ctx.cfgpeers.peers = appctx->ctx.cfgpeers.target;
3294
3295 appctx->st2 = STAT_ST_LIST;
3296 /* fall through */
3297
3298 case STAT_ST_LIST:
3299 if (!appctx->ctx.cfgpeers.peers) {
3300 /* No more peers list. */
3301 appctx->st2 = STAT_ST_END;
3302 }
3303 else {
3304 if (!first_peers)
3305 chunk_appendf(&trash, "\n");
3306 else
3307 first_peers = 0;
3308 if (!peers_dump_head(&trash, si, appctx->ctx.cfgpeers.peers))
3309 goto out;
3310
3311 appctx->ctx.cfgpeers.peer = appctx->ctx.cfgpeers.peers->remote;
3312 appctx->ctx.cfgpeers.peers = appctx->ctx.cfgpeers.peers->next;
3313 appctx->st2 = STAT_ST_INFO;
3314 }
3315 break;
3316
3317 case STAT_ST_INFO:
3318 if (!appctx->ctx.cfgpeers.peer) {
3319 /* End of peer list */
3320 if (show_all)
3321 appctx->st2 = STAT_ST_LIST;
3322 else
3323 appctx->st2 = STAT_ST_END;
3324 }
3325 else {
3326 if (!peers_dump_peer(&trash, si, appctx->ctx.cfgpeers.peer))
3327 goto out;
3328
3329 appctx->ctx.cfgpeers.peer = appctx->ctx.cfgpeers.peer->next;
3330 }
3331 break;
3332
3333 case STAT_ST_END:
3334 appctx->st2 = STAT_ST_FIN;
3335 break;
3336 }
3337 }
3338 ret = 1;
3339 out:
3340 thread_release();
3341 return ret;
3342}
3343
3344/*
3345 * CLI keywords.
3346 */
3347static struct cli_kw_list cli_kws = {{ }, {
3348 { { "show", "peers", NULL }, "show peers [peers section]: dump some information about all the peers or this peers section", cli_parse_show_peers, cli_io_handler_show_peers, },
3349 {},
3350}};
3351
3352/* Register cli keywords */
3353INITCALL1(STG_REGISTER, cli_register_kw, &cli_kws);
3354