From f2ce92bc16c4abde7eb98a5abf8c9279770a9e82 Mon Sep 17 00:00:00 2001 From: William Lallemand Date: Wed, 6 Dec 2023 11:15:01 +0100 Subject: [PATCH] BUG/MINOR: mworker/cli: fix set severity-output support "set severity-output" is one of these command that changes the appctx state so the next commands are affected. Unfortunately the master CLI works with pipelining and server close mode, which means the connection between the master and the worker is closed after each response, so for the next command this is a new appctx state. To fix the problem, 2 new flags are added ACCESS_MCLI_SEVERITY_STR and ACCESS_MCLI_SEVERITY_NB which are used to prefix each command sent to the worker with the right "set severity-output" command. This patch fixes issue #2350. It could be backported as far as 2.6. (cherry picked from commit 1c1bb8ef2abeab2dc8e4857ec2fef2b19b472be7) Signed-off-by: Christopher Faulet --- include/haproxy/cli-t.h | 2 ++ src/cli.c | 30 +++++++++++++++++++++++++++++- 2 files changed, 31 insertions(+), 1 deletion(-) diff --git a/include/haproxy/cli-t.h b/include/haproxy/cli-t.h index 0570b69..c155df3 100644 --- a/include/haproxy/cli-t.h +++ b/include/haproxy/cli-t.h @@ -37,6 +37,8 @@ #define ACCESS_EXPERT 0x0020 /* access to dangerous commands reserved to experts */ #define ACCESS_EXPERIMENTAL 0x0040 #define ACCESS_MCLI_DEBUG 0x0080 /* allow the master CLI to use any command without the flag ACCESS_MASTER */ +#define ACCESS_MCLI_SEVERITY_NB 0x0100 /* 'set severity-output number' on master CLI */ +#define ACCESS_MCLI_SEVERITY_STR 0x0200 /* 'set severity-output string' on master CLI */ /* flags for appctx->st1 */ #define APPCTX_CLI_ST1_PROMPT (1 << 0) diff --git a/src/cli.c b/src/cli.c index 069a265..24c3950 100644 --- a/src/cli.c +++ b/src/cli.c @@ -1759,6 +1759,10 @@ static int set_severity_output(int *target, char *argument) /* parse a "set severity-output" command. */ static int cli_parse_set_severity_output(char **args, char *payload, struct appctx *appctx, void *private) { + /* this will ask the applet to not output a \n after the command */ + if (strcmp(args[3], "-") == 0) + appctx->st1 |= APPCTX_CLI_ST1_NOLF; + if (*args[2] && set_severity_output(&appctx->cli_severity_output, args[2])) return 0; @@ -2453,7 +2457,6 @@ static int pcli_prefix_to_pid(const char *prefix) * >= 0 : number of words to escape * = -1 : error */ - int pcli_find_and_exec_kw(struct stream *s, char **args, int argl, char **errmsg, int *next_pid) { if (argl < 1) @@ -2533,6 +2536,21 @@ int pcli_find_and_exec_kw(struct stream *s, char **args, int argl, char **errmsg if ((argl > 1) && (strcmp(args[1], "on") == 0)) s->pcli_flags |= ACCESS_MCLI_DEBUG; return argl; + } else if (strcmp(args[0], "set") == 0) { + if ((argl > 1) && (strcmp(args[1], "severity-output") == 0)) { + if ((argl > 2) &&strcmp(args[2], "none") == 0) { + s->pcli_flags &= ~(ACCESS_MCLI_SEVERITY_NB|ACCESS_MCLI_SEVERITY_STR); + } else if ((argl > 2) && strcmp(args[2], "string") == 0) { + s->pcli_flags |= ACCESS_MCLI_SEVERITY_STR; + } else if ((argl > 2) && strcmp(args[2], "number") == 0) { + s->pcli_flags |= ACCESS_MCLI_SEVERITY_NB; + } else { + memprintf(errmsg, "one of 'none', 'number', 'string' is a required argument\n"); + return -1; + } + /* only skip argl if we have "set severity-output" not only "set" */ + return argl; + } } return 0; @@ -2711,6 +2729,16 @@ int pcli_parse_request(struct stream *s, struct channel *req, char **errmsg, int ci_insert_line2(req, 0, "expert-mode on -", strlen("expert-mode on -")); ret += strlen("expert-mode on -") + 2; } + if (s->pcli_flags & ACCESS_MCLI_SEVERITY_STR) { + const char *cmd = "set severity-output string -"; + ci_insert_line2(req, 0, cmd, strlen(cmd)); + ret += strlen(cmd) + 2; + } + if (s->pcli_flags & ACCESS_MCLI_SEVERITY_NB) { + const char *cmd = "set severity-output number -"; + ci_insert_line2(req, 0, cmd, strlen(cmd)); + ret += strlen(cmd) + 2; + } if (pcli_has_level(s, ACCESS_LVL_ADMIN)) { goto end; -- 1.7.10.4