blob: 467ce9b248a6e5879df60d453c770b26652e5eee [file] [log] [blame]
William Lallemand9a1ee7a2019-04-01 11:30:02 +02001/*
2 * Master Worker - program
3 *
4 * Copyright HAProxy Technologies - 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
13#define _GNU_SOURCE
14
15#include <sys/types.h>
16#include <errno.h>
17#include <grp.h>
18#include <stdio.h>
19#include <string.h>
20#include <unistd.h>
21
22#include <common/cfgparse.h>
23#include <common/errors.h>
24#include <common/initcall.h>
25
26#include <proto/log.h>
27#include <proto/mworker.h>
28
29static int use_program = 0; /* do we use the program section ? */
30
31/*
32 * Launch every programs
33 */
34int mworker_ext_launch_all()
35{
36 int ret;
37 struct mworker_proc *child;
William Lallemandbd3de3e2019-04-12 16:09:22 +020038 struct mworker_proc *tmp;
39 int reexec = 0;
William Lallemand9a1ee7a2019-04-01 11:30:02 +020040
41 if (!use_program)
42 return 0;
43
William Lallemandbd3de3e2019-04-12 16:09:22 +020044 reexec = getenv("HAPROXY_MWORKER_REEXEC") ? 1 : 0;
45
William Lallemand9a1ee7a2019-04-01 11:30:02 +020046 /* find the right mworker_proc */
William Lallemandbd3de3e2019-04-12 16:09:22 +020047 list_for_each_entry_safe(child, tmp, &proc_list, list) {
William Lallemand8f7069a2019-04-12 16:09:23 +020048 if (child->reloads == 0 && (child->options & PROC_O_TYPE_PROG)) {
William Lallemandbd3de3e2019-04-12 16:09:22 +020049
50 if (reexec && (!(child->options & PROC_O_START_RELOAD))) {
51 struct mworker_proc *old_child;
52
53 /*
54 * This is a reload and we don't want to fork a
55 * new program so have to remove the entry in
56 * the list.
57 *
58 * But before that, we need to mark the
59 * previous program as not leaving, if we find one.
60 */
61
62 list_for_each_entry(old_child, &proc_list, list) {
William Lallemand8f7069a2019-04-12 16:09:23 +020063 if (!(old_child->options & PROC_O_TYPE_PROG) || (!(old_child->options & PROC_O_LEAVING)))
William Lallemandbd3de3e2019-04-12 16:09:22 +020064 continue;
65
66 if (!strcmp(old_child->id, child->id))
67 old_child->options &= ~PROC_O_LEAVING;
68 }
69
70
71 LIST_DEL(&child->list);
Tim Duesterhus9b7a9762019-05-16 20:23:22 +020072 mworker_free_child(child);
William Lallemandbd3de3e2019-04-12 16:09:22 +020073 child = NULL;
74
75 continue;
76 }
77
William Lallemand9a1ee7a2019-04-01 11:30:02 +020078 child->timestamp = now.tv_sec;
79
80 ret = fork();
81 if (ret < 0) {
82 ha_alert("Cannot fork program '%s'.\n", child->id);
83 exit(EXIT_FAILURE); /* there has been an error */
84 } else if (ret > 0) { /* parent */
85 child->pid = ret;
86 ha_notice("New program '%s' (%d) forked\n", child->id, ret);
87 continue;
88 } else if (ret == 0) {
89 /* In child */
90 mworker_unblock_signals();
91 mworker_cleanlisteners();
92 mworker_cleantasks();
93
94 execvp(child->command[0], child->command);
95
96 ha_alert("Cannot execute %s: %s\n", child->command[0], strerror(errno));
97 exit(EXIT_FAILURE);
98 }
99 }
100 }
101
102 return 0;
103
104}
105
106
107/* Configuration */
108
109int cfg_parse_program(const char *file, int linenum, char **args, int kwm)
110{
111 static struct mworker_proc *ext_child = NULL;
112 struct mworker_proc *child;
113 int err_code = 0;
114
115 if (!strcmp(args[0], "program")) {
116 if (alertif_too_many_args(1, file, linenum, args, &err_code)) {
117 err_code |= ERR_ABORT;
118 goto error;
119 }
120
121 if (!*args[1]) {
122 ha_alert("parsing [%s:%d] : '%s' expects an <id> argument\n",
123 file, linenum, args[0]);
124 err_code |= ERR_ALERT | ERR_ABORT;
125 goto error;
126 }
127
128 ext_child = calloc(1, sizeof(*ext_child));
129 if (!ext_child) {
130 ha_alert("parsing [%s:%d] : out of memory.\n", file, linenum);
131 err_code |= ERR_ALERT | ERR_ABORT;
132 goto error;
133 }
134
William Lallemand8f7069a2019-04-12 16:09:23 +0200135 ext_child->options |= PROC_O_TYPE_PROG; /* external process */
William Lallemand9a1ee7a2019-04-01 11:30:02 +0200136 ext_child->command = NULL;
137 ext_child->path = NULL;
138 ext_child->id = NULL;
139 ext_child->pid = -1;
140 ext_child->relative_pid = -1;
141 ext_child->reloads = 0;
142 ext_child->timestamp = -1;
143 ext_child->ipc_fd[0] = -1;
144 ext_child->ipc_fd[1] = -1;
William Lallemandbd3de3e2019-04-12 16:09:22 +0200145 ext_child->options |= PROC_O_START_RELOAD; /* restart the programs by default */
William Lallemand9a1ee7a2019-04-01 11:30:02 +0200146 LIST_INIT(&ext_child->list);
147
148 list_for_each_entry(child, &proc_list, list) {
William Lallemand8f7069a2019-04-12 16:09:23 +0200149 if (child->reloads == 0 && (child->options & PROC_O_TYPE_PROG)) {
William Lallemand9a1ee7a2019-04-01 11:30:02 +0200150 if (!strcmp(args[1], child->id)) {
151 ha_alert("parsing [%s:%d]: '%s' program section already exists in the configuration.\n", file, linenum, args[1]);
152 err_code |= ERR_ALERT | ERR_ABORT;
153 goto error;
154 }
155 }
156 }
157
158 ext_child->id = strdup(args[1]);
159 if (!ext_child->id) {
160 ha_alert("parsing [%s:%d] : out of memory.\n", file, linenum);
161 err_code |= ERR_ALERT | ERR_ABORT;
162 goto error;
163 }
164
165 LIST_ADDQ(&proc_list, &ext_child->list);
166
167 } else if (!strcmp(args[0], "command")) {
168 int arg_nb = 0;
169 int i = 0;
170
171 if (*(args[1]) == 0) {
172 ha_alert("parsing [%s:%d]: '%s' expects a command with optional arguments separated in words.\n", file, linenum, args[0]);
173 err_code |= ERR_ALERT | ERR_FATAL;
174 goto error;
175 }
176
177 while (*args[arg_nb+1])
178 arg_nb++;
179
180 ext_child->command = calloc(arg_nb+1, sizeof(*ext_child->command));
181
182 if (!ext_child->command) {
183 ha_alert("parsing [%s:%d] : out of memory.\n", file, linenum);
184 err_code |= ERR_ALERT | ERR_ABORT;
185 goto error;
186 }
187
188 while (i < arg_nb) {
189 ext_child->command[i] = strdup(args[i+1]);
190 if (!ext_child->command[i]) {
191 ha_alert("parsing [%s:%d] : out of memory.\n", file, linenum);
192 err_code |= ERR_ALERT | ERR_ABORT;
193 goto error;
194 }
195 i++;
196 }
197 ext_child->command[i] = NULL;
198
William Lallemandbd3de3e2019-04-12 16:09:22 +0200199 } else if (!strcmp(args[0], "option")) {
200
201 if (*(args[1]) == '\0') {
202 ha_alert("parsing [%s:%d]: '%s' expects an option name.\n",
203 file, linenum, args[0]);
204 err_code |= ERR_ALERT | ERR_FATAL;
205 goto error;
206 }
207
208 if (strcmp(args[1], "start-on-reload") == 0) {
209 if (alertif_too_many_args_idx(0, 1, file, linenum, args, &err_code))
210 goto error;
211 if (kwm == KWM_STD)
212 ext_child->options |= PROC_O_START_RELOAD;
213 else if (kwm == KWM_NO)
214 ext_child->options &= ~PROC_O_START_RELOAD;
215 goto out;
216
217 } else {
218 ha_alert("parsing [%s:%d] : unknown option '%s'.\n", file, linenum, args[1]);
219 err_code |= ERR_ALERT | ERR_FATAL;
220 goto error;
221 }
William Lallemand9a1ee7a2019-04-01 11:30:02 +0200222 } else {
223 ha_alert("parsing [%s:%d] : unknown keyword '%s' in '%s' section\n", file, linenum, args[0], "program");
224 err_code |= ERR_ALERT | ERR_FATAL;
225 goto error;
226 }
227
228 use_program = 1;
229
230 return err_code;
231
232error:
233 LIST_DEL(&ext_child->list);
234 if (ext_child->command) {
235 int i;
236
237 for (i = 0; ext_child->command[i]; i++) {
238 if (ext_child->command[i]) {
239 free(ext_child->command[i]);
240 ext_child->command[i] = NULL;
241 }
242 }
243 free(ext_child->command);
244 ext_child->command = NULL;
245 }
246 if (ext_child->id) {
247 free(ext_child->id);
248 ext_child->id = NULL;
249 }
250
251 free(ext_child);
252 ext_child = NULL;
253
William Lallemandbd3de3e2019-04-12 16:09:22 +0200254out:
William Lallemand9a1ee7a2019-04-01 11:30:02 +0200255 return err_code;
256
257}
258
259int cfg_program_postparser()
260{
261 int err_code = 0;
262 struct mworker_proc *child;
263
264 list_for_each_entry(child, &proc_list, list) {
William Lallemand8f7069a2019-04-12 16:09:23 +0200265 if (child->reloads == 0 && (child->options & PROC_O_TYPE_PROG)) {
William Lallemand9a1ee7a2019-04-01 11:30:02 +0200266 if (child->command == NULL) {
267 ha_alert("The program section '%s' lacks a command to launch.\n", child->id);
268 err_code |= ERR_ALERT | ERR_FATAL;
269 }
270 }
271 }
272
273 if (use_program && !(global.mode & MODE_MWORKER)) {
274 ha_alert("Can't use a 'program' section without master worker mode.\n");
275 err_code |= ERR_ALERT | ERR_FATAL;
276 }
277
278 return err_code;
279}
280
281
282REGISTER_CONFIG_SECTION("program", cfg_parse_program, NULL);
283REGISTER_CONFIG_POSTPARSER("program", cfg_program_postparser);