blob: 23603376166dc90ff059b7d6f46c2fbcc2284506 [file] [log] [blame]
William Lallemand48dfbbd2019-04-01 11:29:53 +02001/*
2 * Master Worker
3 *
4 * Copyright HAProxy Technologies 2019 - William Lallemand <wlallemand@haproxy.com>
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
Bertrand Jacquin25439de2021-01-21 01:31:46 +000013#define _GNU_SOURCE
14
William Lallemand3fa724d2019-04-01 11:29:55 +020015#include <errno.h>
16#include <fcntl.h>
17#include <signal.h>
William Lallemand48dfbbd2019-04-01 11:29:53 +020018#include <stdlib.h>
19#include <string.h>
William Lallemande25473c2019-04-01 11:29:56 +020020#include <sys/wait.h>
William Lallemand48dfbbd2019-04-01 11:29:53 +020021
Willy Tarreaub2551052020-06-09 09:07:15 +020022#if defined(USE_SYSTEMD)
23#include <systemd/sd-daemon.h>
24#endif
25
Willy Tarreau4c7e4b72020-05-27 12:58:42 +020026#include <haproxy/api.h>
Willy Tarreau6be78492020-06-05 00:00:29 +020027#include <haproxy/cfgparse.h>
Willy Tarreau83487a82020-06-04 20:19:54 +020028#include <haproxy/cli.h>
Willy Tarreau36979d92020-06-05 17:27:29 +020029#include <haproxy/errors.h>
Willy Tarreaudfd3de82020-06-04 23:46:14 +020030#include <haproxy/fd.h>
31#include <haproxy/global.h>
Willy Tarreau853b2972020-05-27 18:01:47 +020032#include <haproxy/list.h>
Willy Tarreau213e9902020-06-04 14:58:24 +020033#include <haproxy/listener.h>
Willy Tarreaub5abe5b2020-06-04 14:07:37 +020034#include <haproxy/mworker.h>
Willy Tarreau3c2a7c22020-06-04 18:38:21 +020035#include <haproxy/peers.h>
Willy Tarreau7c668572021-05-08 20:21:31 +020036#include <haproxy/proxy.h>
Willy Tarreau3727a8a2020-06-04 17:37:26 +020037#include <haproxy/signal.h>
Willy Tarreaudfd3de82020-06-04 23:46:14 +020038#include <haproxy/stream.h>
Willy Tarreau5e539c92020-06-04 20:45:39 +020039#include <haproxy/stream_interface.h>
Willy Tarreau745e98c2021-05-08 13:58:19 +020040#include <haproxy/tools.h>
Willy Tarreaud6788052020-05-27 15:59:00 +020041#include <haproxy/version.h>
William Lallemand48dfbbd2019-04-01 11:29:53 +020042
William Lallemand48dfbbd2019-04-01 11:29:53 +020043
William Lallemande25473c2019-04-01 11:29:56 +020044static int exitcode = -1;
William Lallemand27edc4b2019-05-07 17:49:33 +020045static int max_reloads = -1; /* number max of reloads a worker can have until they are killed */
Willy Tarreau15f9ac32021-05-08 12:30:50 +020046struct mworker_proc *proc_self = NULL; /* process structure of current process */
William Lallemande25473c2019-04-01 11:29:56 +020047
William Lallemande25473c2019-04-01 11:29:56 +020048/* ----- children processes handling ----- */
William Lallemand48dfbbd2019-04-01 11:29:53 +020049
William Lallemand48dfbbd2019-04-01 11:29:53 +020050/*
William Lallemande25473c2019-04-01 11:29:56 +020051 * Send signal to every known children.
52 */
53
54static void mworker_kill(int sig)
55{
William Lallemand3f128872019-04-01 11:29:59 +020056 struct mworker_proc *child;
William Lallemande25473c2019-04-01 11:29:56 +020057
William Lallemand3f128872019-04-01 11:29:59 +020058 list_for_each_entry(child, &proc_list, list) {
59 /* careful there, we must be sure that the pid > 0, we don't want to emit a kill -1 */
William Lallemand32b69012019-04-16 17:42:42 +020060 if ((child->options & (PROC_O_TYPE_WORKER|PROC_O_TYPE_PROG)) && (child->pid > 0))
William Lallemand3f128872019-04-01 11:29:59 +020061 kill(child->pid, sig);
William Lallemande25473c2019-04-01 11:29:56 +020062 }
63}
64
William Lallemand27edc4b2019-05-07 17:49:33 +020065void mworker_kill_max_reloads(int sig)
66{
67 struct mworker_proc *child;
68
69 list_for_each_entry(child, &proc_list, list) {
70 if (max_reloads != -1 && (child->options & PROC_O_TYPE_WORKER) &&
71 (child->pid > 0) && (child->reloads > max_reloads))
72 kill(child->pid, sig);
73 }
74}
William Lallemande25473c2019-04-01 11:29:56 +020075
76/* return 1 if a pid is a current child otherwise 0 */
William Lallemand3f128872019-04-01 11:29:59 +020077int mworker_current_child(int pid)
William Lallemande25473c2019-04-01 11:29:56 +020078{
William Lallemand3f128872019-04-01 11:29:59 +020079 struct mworker_proc *child;
William Lallemande25473c2019-04-01 11:29:56 +020080
William Lallemand3f128872019-04-01 11:29:59 +020081 list_for_each_entry(child, &proc_list, list) {
William Lallemand8f7069a2019-04-12 16:09:23 +020082 if ((child->options & (PROC_O_TYPE_WORKER|PROC_O_TYPE_PROG)) && (!(child->options & PROC_O_LEAVING)) && (child->pid == pid))
William Lallemande25473c2019-04-01 11:29:56 +020083 return 1;
84 }
85 return 0;
86}
87
William Lallemand3f128872019-04-01 11:29:59 +020088/*
89 * Return the number of new and old children (including workers and external
90 * processes)
91 */
92int mworker_child_nb()
93{
94 struct mworker_proc *child;
95 int ret = 0;
96
97 list_for_each_entry(child, &proc_list, list) {
William Lallemand8f7069a2019-04-12 16:09:23 +020098 if (child->options & (PROC_O_TYPE_WORKER|PROC_O_TYPE_PROG))
William Lallemand3f128872019-04-01 11:29:59 +020099 ret++;
100 }
101
102 return ret;
103}
104
105
William Lallemande25473c2019-04-01 11:29:56 +0200106/*
William Lallemand48dfbbd2019-04-01 11:29:53 +0200107 * serialize the proc list and put it in the environment
108 */
109void mworker_proc_list_to_env()
110{
111 char *msg = NULL;
112 struct mworker_proc *child;
113
114 list_for_each_entry(child, &proc_list, list) {
William Lallemand8f7069a2019-04-12 16:09:23 +0200115 char type = '?';
116
117 if (child->options & PROC_O_TYPE_MASTER)
118 type = 'm';
119 else if (child->options & PROC_O_TYPE_PROG)
120 type = 'e';
121 else if (child->options &= PROC_O_TYPE_WORKER)
122 type = 'w';
123
William Lallemand48dfbbd2019-04-01 11:29:53 +0200124 if (child->pid > -1)
William Lallemand1dc69632019-06-12 19:11:33 +0200125 memprintf(&msg, "%s|type=%c;fd=%d;pid=%d;rpid=%d;reloads=%d;timestamp=%d;id=%s;version=%s", msg ? msg : "", type, child->ipc_fd[0], child->pid, child->relative_pid, child->reloads, child->timestamp, child->id ? child->id : "", child->version);
William Lallemand48dfbbd2019-04-01 11:29:53 +0200126 }
127 if (msg)
128 setenv("HAPROXY_PROCESSES", msg, 1);
129}
130
131/*
132 * unserialize the proc list from the environment
133 */
134void mworker_env_to_proc_list()
135{
136 char *msg, *token = NULL, *s1;
137
138 msg = getenv("HAPROXY_PROCESSES");
139 if (!msg)
140 return;
141
142 while ((token = strtok_r(msg, "|", &s1))) {
143 struct mworker_proc *child;
144 char *subtoken = NULL;
145 char *s2;
146
147 msg = NULL;
148
149 child = calloc(1, sizeof(*child));
150
151 while ((subtoken = strtok_r(token, ";", &s2))) {
152
153 token = NULL;
154
155 if (strncmp(subtoken, "type=", 5) == 0) {
William Lallemand8f7069a2019-04-12 16:09:23 +0200156 char type;
157
158 type = *(subtoken+5);
159 if (type == 'm') { /* we are in the master, assign it */
William Lallemand48dfbbd2019-04-01 11:29:53 +0200160 proc_self = child;
William Lallemand8f7069a2019-04-12 16:09:23 +0200161 child->options |= PROC_O_TYPE_MASTER;
162 } else if (type == 'e') {
163 child->options |= PROC_O_TYPE_PROG;
164 } else if (type == 'w') {
165 child->options |= PROC_O_TYPE_WORKER;
166 }
167
William Lallemand48dfbbd2019-04-01 11:29:53 +0200168 } else if (strncmp(subtoken, "fd=", 3) == 0) {
169 child->ipc_fd[0] = atoi(subtoken+3);
170 } else if (strncmp(subtoken, "pid=", 4) == 0) {
171 child->pid = atoi(subtoken+4);
172 } else if (strncmp(subtoken, "rpid=", 5) == 0) {
173 child->relative_pid = atoi(subtoken+5);
174 } else if (strncmp(subtoken, "reloads=", 8) == 0) {
175 /* we reloaded this process once more */
176 child->reloads = atoi(subtoken+8) + 1;
177 } else if (strncmp(subtoken, "timestamp=", 10) == 0) {
178 child->timestamp = atoi(subtoken+10);
William Lallemand9a1ee7a2019-04-01 11:30:02 +0200179 } else if (strncmp(subtoken, "id=", 3) == 0) {
180 child->id = strdup(subtoken+3);
William Lallemand1dc69632019-06-12 19:11:33 +0200181 } else if (strncmp(subtoken, "version=", 8) == 0) {
182 child->version = strdup(subtoken+8);
William Lallemand48dfbbd2019-04-01 11:29:53 +0200183 }
184 }
William Lallemand9a1ee7a2019-04-01 11:30:02 +0200185 if (child->pid) {
William Lallemand920fc8b2019-05-14 11:15:18 +0200186 /* this is a process inherited from a reload that should be leaving */
187 child->options |= PROC_O_LEAVING;
188
Willy Tarreau2b718102021-04-21 07:32:39 +0200189 LIST_APPEND(&proc_list, &child->list);
William Lallemand9a1ee7a2019-04-01 11:30:02 +0200190 } else {
Tim Duesterhus9b7a9762019-05-16 20:23:22 +0200191 mworker_free_child(child);
William Lallemand9a1ee7a2019-04-01 11:30:02 +0200192 }
William Lallemand48dfbbd2019-04-01 11:29:53 +0200193 }
194
195 unsetenv("HAPROXY_PROCESSES");
196}
William Lallemand3cd95d22019-04-01 11:29:54 +0200197
198/* Signal blocking and unblocking */
199
200void mworker_block_signals()
201{
202 sigset_t set;
203
204 sigemptyset(&set);
205 sigaddset(&set, SIGUSR1);
206 sigaddset(&set, SIGUSR2);
Willy Tarreaud26c9f92019-12-11 14:24:07 +0100207 sigaddset(&set, SIGTTIN);
208 sigaddset(&set, SIGTTOU);
William Lallemand3cd95d22019-04-01 11:29:54 +0200209 sigaddset(&set, SIGHUP);
210 sigaddset(&set, SIGCHLD);
211 ha_sigmask(SIG_SETMASK, &set, NULL);
212}
213
214void mworker_unblock_signals()
215{
216 haproxy_unblock_signals();
217}
William Lallemand3fa724d2019-04-01 11:29:55 +0200218
William Lallemande25473c2019-04-01 11:29:56 +0200219/* ----- mworker signal handlers ----- */
220
Willy Tarreaud26c9f92019-12-11 14:24:07 +0100221/* broadcast the configured signal to the workers */
222void mworker_broadcast_signal(struct sig_handler *sh)
223{
224 mworker_kill(sh->arg);
225}
226
William Lallemande25473c2019-04-01 11:29:56 +0200227/*
228 * When called, this function reexec haproxy with -sf followed by current
229 * children PIDs and possibly old children PIDs if they didn't leave yet.
230 */
231void mworker_catch_sighup(struct sig_handler *sh)
232{
233 mworker_reload();
234}
235
236void mworker_catch_sigterm(struct sig_handler *sh)
237{
238 int sig = sh->arg;
239
240#if defined(USE_SYSTEMD)
241 if (global.tune.options & GTUNE_USE_SYSTEMD) {
242 sd_notify(0, "STOPPING=1");
243 }
244#endif
245 ha_warning("Exiting Master process...\n");
246 mworker_kill(sig);
247}
248
249/*
250 * Wait for every children to exit
251 */
252
253void mworker_catch_sigchld(struct sig_handler *sh)
254{
255 int exitpid = -1;
256 int status = 0;
William Lallemande25473c2019-04-01 11:29:56 +0200257 int childfound;
258
259restart_wait:
260
261 childfound = 0;
262
263 exitpid = waitpid(-1, &status, WNOHANG);
264 if (exitpid > 0) {
Tim Duesterhus9b7a9762019-05-16 20:23:22 +0200265 struct mworker_proc *child, *it;
266
William Lallemande25473c2019-04-01 11:29:56 +0200267 if (WIFEXITED(status))
268 status = WEXITSTATUS(status);
269 else if (WIFSIGNALED(status))
270 status = 128 + WTERMSIG(status);
271 else if (WIFSTOPPED(status))
272 status = 128 + WSTOPSIG(status);
273 else
274 status = 255;
275
William Lallemand3f128872019-04-01 11:29:59 +0200276 /* delete the child from the process list */
William Lallemande25473c2019-04-01 11:29:56 +0200277 list_for_each_entry_safe(child, it, &proc_list, list) {
278 if (child->pid != exitpid)
279 continue;
280
Willy Tarreau2b718102021-04-21 07:32:39 +0200281 LIST_DELETE(&child->list);
William Lallemande25473c2019-04-01 11:29:56 +0200282 close(child->ipc_fd[0]);
283 childfound = 1;
284 break;
285 }
286
William Lallemand3f128872019-04-01 11:29:59 +0200287 if (!childfound) {
288 /* We didn't find the PID in the list, that shouldn't happen but we can emit a warning */
William Lallemand9a1ee7a2019-04-01 11:30:02 +0200289 ha_warning("Process %d exited with code %d (%s)\n", exitpid, status, (status >= 128) ? strsignal(status - 128) : "Exit");
William Lallemande25473c2019-04-01 11:29:56 +0200290 } else {
William Lallemand9a1ee7a2019-04-01 11:30:02 +0200291 /* check if exited child is a current child */
William Lallemand45286112019-04-12 16:09:21 +0200292 if (!(child->options & PROC_O_LEAVING)) {
William Lallemanda655ba42020-05-06 17:27:03 +0200293 if (child->options & PROC_O_TYPE_WORKER) {
294 if (status < 128)
295 ha_warning("Current worker #%d (%d) exited with code %d (%s)\n", child->relative_pid, exitpid, status, "Exit");
296 else
297 ha_alert("Current worker #%d (%d) exited with code %d (%s)\n", child->relative_pid, exitpid, status, strsignal(status - 128));
298 }
William Lallemand8f7069a2019-04-12 16:09:23 +0200299 else if (child->options & PROC_O_TYPE_PROG)
William Lallemand9a1ee7a2019-04-01 11:30:02 +0200300 ha_alert("Current program '%s' (%d) exited with code %d (%s)\n", child->id, exitpid, status, (status >= 128) ? strsignal(status - 128) : "Exit");
William Lallemand3f128872019-04-01 11:29:59 +0200301
William Lallemande25473c2019-04-01 11:29:56 +0200302 if (status != 0 && status != 130 && status != 143
303 && !(global.tune.options & GTUNE_NOEXIT_ONFAILURE)) {
William Lallemand9a1ee7a2019-04-01 11:30:02 +0200304 ha_alert("exit-on-failure: killing every processes with SIGTERM\n");
William Lallemande25473c2019-04-01 11:29:56 +0200305 mworker_kill(SIGTERM);
306 }
William Lallemand74f0ec32019-04-16 17:42:44 +0200307 /* 0 & SIGTERM (143) are normal, but we should report SIGINT (130) and other signals */
308 if (exitcode < 0 && status != 0 && status != 143)
309 exitcode = status;
William Lallemande25473c2019-04-01 11:29:56 +0200310 } else {
William Lallemand8f7069a2019-04-12 16:09:23 +0200311 if (child->options & PROC_O_TYPE_WORKER) {
William Lallemand3f128872019-04-01 11:29:59 +0200312 ha_warning("Former worker #%d (%d) exited with code %d (%s)\n", child->relative_pid, exitpid, status, (status >= 128) ? strsignal(status - 128) : "Exit");
313 delete_oldpid(exitpid);
William Lallemand8f7069a2019-04-12 16:09:23 +0200314 } else if (child->options & PROC_O_TYPE_PROG) {
William Lallemand9a1ee7a2019-04-01 11:30:02 +0200315 ha_warning("Former program '%s' (%d) exited with code %d (%s)\n", child->id, exitpid, status, (status >= 128) ? strsignal(status - 128) : "Exit");
William Lallemand3f128872019-04-01 11:29:59 +0200316 }
William Lallemande25473c2019-04-01 11:29:56 +0200317 }
Tim Duesterhus9b7a9762019-05-16 20:23:22 +0200318 mworker_free_child(child);
319 child = NULL;
William Lallemande25473c2019-04-01 11:29:56 +0200320 }
321
322 /* do it again to check if it was the last worker */
323 goto restart_wait;
324 }
325 /* Better rely on the system than on a list of process to check if it was the last one */
326 else if (exitpid == -1 && errno == ECHILD) {
William Lallemand4cf4b332019-04-16 17:42:43 +0200327 ha_warning("All workers exited. Exiting... (%d)\n", (exitcode > 0) ? exitcode : EXIT_SUCCESS);
William Lallemande25473c2019-04-01 11:29:56 +0200328 atexit_flag = 0;
329 if (exitcode > 0)
William Lallemand4cf4b332019-04-16 17:42:43 +0200330 exit(exitcode); /* parent must leave using the status code that provoked the exit */
331 exit(EXIT_SUCCESS);
William Lallemande25473c2019-04-01 11:29:56 +0200332 }
333
334}
335
William Lallemand3fa724d2019-04-01 11:29:55 +0200336/* ----- IPC FD (sockpair) related ----- */
337
338/* This wrapper is called from the workers. It is registered instead of the
339 * normal listener_accept() so the worker can exit() when it detects that the
340 * master closed the IPC FD. If it's not a close, we just call the regular
Willy Tarreaua74cb382020-10-15 21:29:49 +0200341 * listener_accept() function.
342 */
William Lallemand3fa724d2019-04-01 11:29:55 +0200343void mworker_accept_wrapper(int fd)
344{
345 char c;
346 int ret;
347
348 while (1) {
349 ret = recv(fd, &c, 1, MSG_PEEK);
350 if (ret == -1) {
351 if (errno == EINTR)
352 continue;
353 if (errno == EAGAIN) {
354 fd_cant_recv(fd);
355 return;
356 }
357 break;
358 } else if (ret > 0) {
Willy Tarreaua74cb382020-10-15 21:29:49 +0200359 struct listener *l = fdtab[fd].owner;
360
361 if (l)
362 listener_accept(l);
William Lallemand3fa724d2019-04-01 11:29:55 +0200363 return;
364 } else if (ret == 0) {
365 /* At this step the master is down before
366 * this worker perform a 'normal' exit.
367 * So we want to exit with an error but
368 * other threads could currently process
369 * some stuff so we can't perform a clean
370 * deinit().
371 */
372 exit(EXIT_FAILURE);
373 }
374 }
375 return;
376}
377
378/*
Willy Tarreau619a95f2019-05-20 11:12:15 +0200379 * This function registers the accept wrapper for the sockpair of the master
380 * worker. It's only handled by worker thread #0. Other threads and master do
381 * nothing here. It always returns 1 (success).
William Lallemand3fa724d2019-04-01 11:29:55 +0200382 */
Willy Tarreau619a95f2019-05-20 11:12:15 +0200383static int mworker_pipe_register_per_thread()
William Lallemand3fa724d2019-04-01 11:29:55 +0200384{
Willy Tarreau619a95f2019-05-20 11:12:15 +0200385 if (!(global.mode & MODE_MWORKER) || master)
386 return 1;
387
388 if (tid != 0)
389 return 1;
William Lallemand3fa724d2019-04-01 11:29:55 +0200390
391 fcntl(proc_self->ipc_fd[1], F_SETFL, O_NONBLOCK);
392 /* In multi-tread, we need only one thread to process
393 * events on the pipe with master
394 */
Willy Tarreau619a95f2019-05-20 11:12:15 +0200395 fd_insert(proc_self->ipc_fd[1], fdtab[proc_self->ipc_fd[1]].owner, mworker_accept_wrapper, tid_bit);
William Lallemand3fa724d2019-04-01 11:29:55 +0200396 fd_want_recv(proc_self->ipc_fd[1]);
Willy Tarreau619a95f2019-05-20 11:12:15 +0200397 return 1;
William Lallemand3fa724d2019-04-01 11:29:55 +0200398}
William Lallemand9001ce82019-04-01 11:29:57 +0200399
Willy Tarreau619a95f2019-05-20 11:12:15 +0200400REGISTER_PER_THREAD_INIT(mworker_pipe_register_per_thread);
401
William Lallemand9001ce82019-04-01 11:29:57 +0200402/* ----- proxies ----- */
403/*
404 * Upon a reload, the master worker needs to close all listeners FDs but the mworker_pipe
405 * fd, and the FD provided by fd@
406 */
407void mworker_cleanlisteners()
408{
409 struct listener *l, *l_next;
410 struct proxy *curproxy;
411 struct peers *curpeers;
412
413 /* we might have to unbind some peers sections from some processes */
414 for (curpeers = cfg_peers; curpeers; curpeers = curpeers->next) {
415 if (!curpeers->peers_fe)
416 continue;
417
418 stop_proxy(curpeers->peers_fe);
419 /* disable this peer section so that it kills itself */
420 signal_unregister_handler(curpeers->sighandler);
Olivier Houchard3f795f72019-04-17 22:51:06 +0200421 task_destroy(curpeers->sync_task);
William Lallemand9001ce82019-04-01 11:29:57 +0200422 curpeers->sync_task = NULL;
Olivier Houchard3f795f72019-04-17 22:51:06 +0200423 task_destroy(curpeers->peers_fe->task);
William Lallemand9001ce82019-04-01 11:29:57 +0200424 curpeers->peers_fe->task = NULL;
425 curpeers->peers_fe = NULL;
426 }
427
428 for (curproxy = proxies_list; curproxy; curproxy = curproxy->next) {
429 int listen_in_master = 0;
430
431 list_for_each_entry_safe(l, l_next, &curproxy->conf.listeners, by_fe) {
432 /* remove the listener, but not those we need in the master... */
Willy Tarreau18c20d22020-10-09 16:11:46 +0200433 if (!(l->rx.flags & RX_F_MWORKER)) {
Willy Tarreau75c98d12020-10-09 15:55:23 +0200434 unbind_listener(l);
William Lallemand9001ce82019-04-01 11:29:57 +0200435 delete_listener(l);
436 } else {
437 listen_in_master = 1;
438 }
439 }
440 /* if the proxy shouldn't be in the master, we stop it */
441 if (!listen_in_master)
Willy Tarreauc3914d42020-09-24 08:39:22 +0200442 curproxy->disabled = 1;
William Lallemand9001ce82019-04-01 11:29:57 +0200443 }
William Lallemand88dc7c52019-04-01 11:30:01 +0200444}
445
446/* Displays workers and processes */
447static int cli_io_handler_show_proc(struct appctx *appctx)
448{
449 struct stream_interface *si = appctx->owner;
450 struct mworker_proc *child;
451 int old = 0;
452 int up = now.tv_sec - proc_self->timestamp;
William Lallemande8669fc2019-06-12 18:21:17 +0200453 char *uptime = NULL;
William Lallemand88dc7c52019-04-01 11:30:01 +0200454
455 if (unlikely(si_ic(si)->flags & (CF_WRITE_ERROR|CF_SHUTW)))
456 return 1;
457
458 chunk_reset(&trash);
459
William Lallemand1dc69632019-06-12 19:11:33 +0200460 chunk_printf(&trash, "#%-14s %-15s %-15s %-15s %-15s %-15s\n", "<PID>", "<type>", "<relative PID>", "<reloads>", "<uptime>", "<version>");
William Lallemande8669fc2019-06-12 18:21:17 +0200461 memprintf(&uptime, "%dd%02dh%02dm%02ds", up / 86400, (up % 86400) / 3600, (up % 3600) / 60, (up % 60));
Willy Tarreau76a80c72019-06-22 07:41:38 +0200462 chunk_appendf(&trash, "%-15u %-15s %-15u %-15d %-15s %-15s\n", (unsigned int)getpid(), "master", 0, proc_self->reloads, uptime, haproxy_version);
Willy Tarreau61cfdf42021-02-20 10:46:51 +0100463 ha_free(&uptime);
William Lallemand88dc7c52019-04-01 11:30:01 +0200464
465 /* displays current processes */
466
467 chunk_appendf(&trash, "# workers\n");
468 list_for_each_entry(child, &proc_list, list) {
469 up = now.tv_sec - child->timestamp;
470
William Lallemand8f7069a2019-04-12 16:09:23 +0200471 if (!(child->options & PROC_O_TYPE_WORKER))
William Lallemand88dc7c52019-04-01 11:30:01 +0200472 continue;
473
William Lallemand45286112019-04-12 16:09:21 +0200474 if (child->options & PROC_O_LEAVING) {
William Lallemand88dc7c52019-04-01 11:30:01 +0200475 old++;
476 continue;
477 }
William Lallemande8669fc2019-06-12 18:21:17 +0200478 memprintf(&uptime, "%dd%02dh%02dm%02ds", up / 86400, (up % 86400) / 3600, (up % 3600) / 60, (up % 60));
William Lallemand1dc69632019-06-12 19:11:33 +0200479 chunk_appendf(&trash, "%-15u %-15s %-15u %-15d %-15s %-15s\n", child->pid, "worker", child->relative_pid, child->reloads, uptime, child->version);
Willy Tarreau61cfdf42021-02-20 10:46:51 +0100480 ha_free(&uptime);
William Lallemand88dc7c52019-04-01 11:30:01 +0200481 }
482
483 /* displays old processes */
484
485 if (old) {
486 char *msg = NULL;
487
488 chunk_appendf(&trash, "# old workers\n");
489 list_for_each_entry(child, &proc_list, list) {
490 up = now.tv_sec - child->timestamp;
491
William Lallemand8f7069a2019-04-12 16:09:23 +0200492 if (!(child->options & PROC_O_TYPE_WORKER))
William Lallemand88dc7c52019-04-01 11:30:01 +0200493 continue;
494
William Lallemand45286112019-04-12 16:09:21 +0200495 if (child->options & PROC_O_LEAVING) {
William Lallemand88dc7c52019-04-01 11:30:01 +0200496 memprintf(&msg, "[was: %u]", child->relative_pid);
William Lallemande8669fc2019-06-12 18:21:17 +0200497 memprintf(&uptime, "%dd%02dh%02dm%02ds", up / 86400, (up % 86400) / 3600, (up % 3600) / 60, (up % 60));
William Lallemand1dc69632019-06-12 19:11:33 +0200498 chunk_appendf(&trash, "%-15u %-15s %-15s %-15d %-15s %-15s\n", child->pid, "worker", msg, child->reloads, uptime, child->version);
Willy Tarreau61cfdf42021-02-20 10:46:51 +0100499 ha_free(&uptime);
William Lallemand88dc7c52019-04-01 11:30:01 +0200500 }
501 }
502 free(msg);
503 }
504
William Lallemandad53d6d2019-04-01 11:30:03 +0200505 /* displays external process */
506 chunk_appendf(&trash, "# programs\n");
507 old = 0;
508 list_for_each_entry(child, &proc_list, list) {
509 up = now.tv_sec - child->timestamp;
510
William Lallemand8f7069a2019-04-12 16:09:23 +0200511 if (!(child->options & PROC_O_TYPE_PROG))
William Lallemandad53d6d2019-04-01 11:30:03 +0200512 continue;
513
William Lallemand45286112019-04-12 16:09:21 +0200514 if (child->options & PROC_O_LEAVING) {
William Lallemandad53d6d2019-04-01 11:30:03 +0200515 old++;
516 continue;
517 }
William Lallemande8669fc2019-06-12 18:21:17 +0200518 memprintf(&uptime, "%dd%02dh%02dm%02ds", up / 86400, (up % 86400) / 3600, (up % 3600) / 60, (up % 60));
William Lallemand1dc69632019-06-12 19:11:33 +0200519 chunk_appendf(&trash, "%-15u %-15s %-15s %-15d %-15s %-15s\n", child->pid, child->id, "-", child->reloads, uptime, "-");
Willy Tarreau61cfdf42021-02-20 10:46:51 +0100520 ha_free(&uptime);
William Lallemandad53d6d2019-04-01 11:30:03 +0200521 }
522
523 if (old) {
524 chunk_appendf(&trash, "# old programs\n");
525 list_for_each_entry(child, &proc_list, list) {
526 up = now.tv_sec - child->timestamp;
527
William Lallemand8f7069a2019-04-12 16:09:23 +0200528 if (!(child->options & PROC_O_TYPE_PROG))
William Lallemandad53d6d2019-04-01 11:30:03 +0200529 continue;
530
William Lallemand45286112019-04-12 16:09:21 +0200531 if (child->options & PROC_O_LEAVING) {
William Lallemande8669fc2019-06-12 18:21:17 +0200532 memprintf(&uptime, "%dd%02dh%02dm%02ds", up / 86400, (up % 86400) / 3600, (up % 3600) / 60, (up % 60));
William Lallemand1dc69632019-06-12 19:11:33 +0200533 chunk_appendf(&trash, "%-15u %-15s %-15s %-15d %-15s %-15s\n", child->pid, child->id, "-", child->reloads, uptime, "-");
Willy Tarreau61cfdf42021-02-20 10:46:51 +0100534 ha_free(&uptime);
William Lallemandad53d6d2019-04-01 11:30:03 +0200535 }
536 }
537 }
538
539
540
William Lallemand88dc7c52019-04-01 11:30:01 +0200541 if (ci_putchk(si_ic(si), &trash) == -1) {
542 si_rx_room_blk(si);
543 return 0;
544 }
545
546 /* dump complete */
547 return 1;
William Lallemand9001ce82019-04-01 11:29:57 +0200548}
William Lallemand88dc7c52019-04-01 11:30:01 +0200549
550/* reload the master process */
551static int cli_parse_reload(char **args, char *payload, struct appctx *appctx, void *private)
552{
553 if (!cli_has_level(appctx, ACCESS_LVL_OPER))
554 return 1;
555
556 mworker_reload();
557
558 return 1;
559}
560
561
William Lallemand27edc4b2019-05-07 17:49:33 +0200562static int mworker_parse_global_max_reloads(char **args, int section_type, struct proxy *curpx,
Willy Tarreau01825162021-03-09 09:53:46 +0100563 const struct proxy *defpx, const char *file, int linenum, char **err)
William Lallemand27edc4b2019-05-07 17:49:33 +0200564{
565
566 int err_code = 0;
567
568 if (alertif_too_many_args(1, file, linenum, args, &err_code))
569 goto out;
570
571 if (*(args[1]) == 0) {
572 memprintf(err, "%sparsing [%s:%d] : '%s' expects an integer argument.\n", *err, file, linenum, args[0]);
573 err_code |= ERR_ALERT | ERR_FATAL;
574 goto out;
575 }
576
577 max_reloads = atol(args[1]);
578 if (max_reloads < 0) {
579 memprintf(err, "%sparsing [%s:%d] '%s' : invalid value %d, must be >= 0", *err, file, linenum, args[0], max_reloads);
580 err_code |= ERR_ALERT | ERR_FATAL;
581 goto out;
582 }
583
584out:
585 return err_code;
586}
587
Tim Duesterhus9b7a9762019-05-16 20:23:22 +0200588void mworker_free_child(struct mworker_proc *child)
589{
590 if (child == NULL)
591 return;
592
593 if (child->command) {
594 int i;
595
596 for (i = 0; child->command[i]; i++) {
597 if (child->command[i]) {
Willy Tarreau61cfdf42021-02-20 10:46:51 +0100598 ha_free(&child->command[i]);
Tim Duesterhus9b7a9762019-05-16 20:23:22 +0200599 }
William Lallemande8669fc2019-06-12 18:21:17 +0200600
Tim Duesterhus9b7a9762019-05-16 20:23:22 +0200601 }
Willy Tarreau61cfdf42021-02-20 10:46:51 +0100602 ha_free(&child->command);
Tim Duesterhus9b7a9762019-05-16 20:23:22 +0200603 }
604 if (child->id) {
Willy Tarreau61cfdf42021-02-20 10:46:51 +0100605 ha_free(&child->id);
Tim Duesterhus9b7a9762019-05-16 20:23:22 +0200606 }
William Lallemand1dc69632019-06-12 19:11:33 +0200607 if (child->version) {
Willy Tarreau61cfdf42021-02-20 10:46:51 +0100608 ha_free(&child->version);
William Lallemand1dc69632019-06-12 19:11:33 +0200609 }
Tim Duesterhus9b7a9762019-05-16 20:23:22 +0200610 free(child);
611}
William Lallemand27edc4b2019-05-07 17:49:33 +0200612
613static struct cfg_kw_list mworker_kws = {{ }, {
614 { CFG_GLOBAL, "mworker-max-reloads", mworker_parse_global_max_reloads },
615 { 0, NULL, NULL },
616}};
617
618INITCALL1(STG_REGISTER, cfg_register_keywords, &mworker_kws);
619
620
William Lallemand88dc7c52019-04-01 11:30:01 +0200621/* register cli keywords */
622static struct cli_kw_list cli_kws = {{ },{
Willy Tarreau23c740e2021-05-09 22:49:44 +0200623 { { "@<relative pid>", NULL }, "@<relative pid> : send a command to the <relative pid> process", NULL, cli_io_handler_show_proc, NULL, NULL, ACCESS_MASTER_ONLY},
624 { { "@!<pid>", NULL }, "@!<pid> : send a command to the <pid> process", cli_parse_default, NULL, NULL, NULL, ACCESS_MASTER_ONLY},
625 { { "@master", NULL }, "@master : send a command to the master process", cli_parse_default, NULL, NULL, NULL, ACCESS_MASTER_ONLY},
626 { { "show", "proc", NULL }, "show proc : show processes status", cli_parse_default, cli_io_handler_show_proc, NULL, NULL, ACCESS_MASTER_ONLY},
627 { { "reload", NULL }, "reload : reload haproxy", cli_parse_reload, NULL, NULL, NULL, ACCESS_MASTER_ONLY},
William Lallemand88dc7c52019-04-01 11:30:01 +0200628 {{},}
629}};
630
631INITCALL1(STG_REGISTER, cli_register_kw, &cli_kws);