job.c revision 1.507 1 1.507 rillig /* $NetBSD: job.c,v 1.507 2025/05/17 20:59:27 rillig Exp $ */
2 1.10 christos
3 1.1 cgd /*
4 1.1 cgd * Copyright (c) 1988, 1989, 1990 The Regents of the University of California.
5 1.82 agc * All rights reserved.
6 1.82 agc *
7 1.82 agc * This code is derived from software contributed to Berkeley by
8 1.82 agc * Adam de Boor.
9 1.82 agc *
10 1.82 agc * Redistribution and use in source and binary forms, with or without
11 1.82 agc * modification, are permitted provided that the following conditions
12 1.82 agc * are met:
13 1.82 agc * 1. Redistributions of source code must retain the above copyright
14 1.82 agc * notice, this list of conditions and the following disclaimer.
15 1.82 agc * 2. Redistributions in binary form must reproduce the above copyright
16 1.82 agc * notice, this list of conditions and the following disclaimer in the
17 1.82 agc * documentation and/or other materials provided with the distribution.
18 1.82 agc * 3. Neither the name of the University nor the names of its contributors
19 1.82 agc * may be used to endorse or promote products derived from this software
20 1.82 agc * without specific prior written permission.
21 1.82 agc *
22 1.82 agc * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
23 1.82 agc * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
24 1.82 agc * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
25 1.82 agc * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
26 1.82 agc * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
27 1.82 agc * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
28 1.82 agc * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
29 1.82 agc * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
30 1.82 agc * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
31 1.82 agc * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
32 1.82 agc * SUCH DAMAGE.
33 1.82 agc */
34 1.82 agc
35 1.82 agc /*
36 1.1 cgd * Copyright (c) 1988, 1989 by Adam de Boor
37 1.1 cgd * Copyright (c) 1989 by Berkeley Softworks
38 1.1 cgd * All rights reserved.
39 1.1 cgd *
40 1.1 cgd * This code is derived from software contributed to Berkeley by
41 1.1 cgd * Adam de Boor.
42 1.1 cgd *
43 1.1 cgd * Redistribution and use in source and binary forms, with or without
44 1.1 cgd * modification, are permitted provided that the following conditions
45 1.1 cgd * are met:
46 1.1 cgd * 1. Redistributions of source code must retain the above copyright
47 1.1 cgd * notice, this list of conditions and the following disclaimer.
48 1.1 cgd * 2. Redistributions in binary form must reproduce the above copyright
49 1.1 cgd * notice, this list of conditions and the following disclaimer in the
50 1.1 cgd * documentation and/or other materials provided with the distribution.
51 1.1 cgd * 3. All advertising materials mentioning features or use of this software
52 1.1 cgd * must display the following acknowledgement:
53 1.1 cgd * This product includes software developed by the University of
54 1.1 cgd * California, Berkeley and its contributors.
55 1.1 cgd * 4. Neither the name of the University nor the names of its contributors
56 1.1 cgd * may be used to endorse or promote products derived from this software
57 1.1 cgd * without specific prior written permission.
58 1.1 cgd *
59 1.1 cgd * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
60 1.1 cgd * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
61 1.1 cgd * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
62 1.1 cgd * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
63 1.1 cgd * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
64 1.1 cgd * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
65 1.1 cgd * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
66 1.1 cgd * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
67 1.1 cgd * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
68 1.1 cgd * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
69 1.1 cgd * SUCH DAMAGE.
70 1.1 cgd */
71 1.1 cgd
72 1.398 rillig /*
73 1.461 rillig * Create child processes and collect their output.
74 1.1 cgd *
75 1.1 cgd * Interface:
76 1.278 rillig * Job_Init Called to initialize this module. In addition,
77 1.461 rillig * the .BEGIN target is made, including all of its
78 1.407 rillig * dependencies before this function returns.
79 1.278 rillig * Hence, the makefiles must have been parsed
80 1.278 rillig * before this function is called.
81 1.278 rillig *
82 1.278 rillig * Job_End Clean up any memory used.
83 1.278 rillig *
84 1.246 rillig * Job_Make Start the creation of the given target.
85 1.1 cgd *
86 1.246 rillig * Job_CatchChildren
87 1.500 rillig * Check for and handle the termination of any children.
88 1.246 rillig *
89 1.246 rillig * Job_CatchOutput
90 1.500 rillig * Print any output the child processes have produced.
91 1.246 rillig *
92 1.407 rillig * Job_ParseShell Given a special dependency line with target '.SHELL',
93 1.407 rillig * define the shell that is used for the creation
94 1.407 rillig * commands in jobs mode.
95 1.246 rillig *
96 1.487 rillig * Job_Finish Make the .END target. Should only be called when the
97 1.246 rillig * job table is empty.
98 1.246 rillig *
99 1.407 rillig * Job_AbortAll Abort all currently running jobs. Do not handle
100 1.407 rillig * output or do anything for the jobs, just kill them.
101 1.407 rillig * Should only be called in an emergency.
102 1.246 rillig *
103 1.246 rillig * Job_CheckCommands
104 1.246 rillig * Verify that the commands for a target are
105 1.246 rillig * ok. Provide them if necessary and possible.
106 1.1 cgd *
107 1.246 rillig * Job_Touch Update a target without really updating it.
108 1.1 cgd *
109 1.246 rillig * Job_Wait Wait for all currently-running jobs to finish.
110 1.1 cgd */
111 1.1 cgd
112 1.5 cgd #include <sys/types.h>
113 1.1 cgd #include <sys/stat.h>
114 1.1 cgd #include <sys/time.h>
115 1.1 cgd #include <sys/wait.h>
116 1.72 wiz
117 1.72 wiz #include <errno.h>
118 1.35 christos #ifndef USE_SELECT
119 1.35 christos #include <poll.h>
120 1.35 christos #endif
121 1.72 wiz #include <signal.h>
122 1.72 wiz #include <utime.h>
123 1.72 wiz
124 1.5 cgd #include "make.h"
125 1.5 cgd #include "dir.h"
126 1.1 cgd #include "job.h"
127 1.493 rillig #ifdef USE_META
128 1.493 rillig # include "meta.h"
129 1.493 rillig #endif
130 1.1 cgd #include "pathnames.h"
131 1.40 sommerfe #include "trace.h"
132 1.232 rillig
133 1.232 rillig /* "@(#)job.c 8.2 (Berkeley) 3/19/94" */
134 1.507 rillig MAKE_RCSID("$NetBSD: job.c,v 1.507 2025/05/17 20:59:27 rillig Exp $");
135 1.493 rillig
136 1.493 rillig
137 1.493 rillig #ifdef USE_SELECT
138 1.493 rillig /*
139 1.493 rillig * Emulate poll() in terms of select(). This is not a complete
140 1.493 rillig * emulation but it is sufficient for make's purposes.
141 1.493 rillig */
142 1.493 rillig
143 1.493 rillig #define poll emul_poll
144 1.493 rillig #define pollfd emul_pollfd
145 1.493 rillig
146 1.493 rillig struct emul_pollfd {
147 1.493 rillig int fd;
148 1.493 rillig short events;
149 1.493 rillig short revents;
150 1.493 rillig };
151 1.493 rillig
152 1.493 rillig #define POLLIN 0x0001
153 1.493 rillig #define POLLOUT 0x0004
154 1.493 rillig
155 1.493 rillig int emul_poll(struct pollfd *, int, int);
156 1.493 rillig #endif
157 1.493 rillig
158 1.493 rillig struct pollfd;
159 1.493 rillig
160 1.493 rillig
161 1.501 rillig enum JobStatus {
162 1.501 rillig JOB_ST_FREE, /* Job is available */
163 1.501 rillig JOB_ST_SET_UP, /* Job is allocated but otherwise invalid */
164 1.501 rillig JOB_ST_RUNNING, /* Job is running, pid valid */
165 1.501 rillig JOB_ST_FINISHED /* Job is done (i.e. after SIGCHLD) */
166 1.501 rillig };
167 1.501 rillig
168 1.501 rillig static const char JobStatus_Name[][9] = {
169 1.501 rillig "free",
170 1.501 rillig "set-up",
171 1.501 rillig "running",
172 1.501 rillig "finished",
173 1.501 rillig };
174 1.493 rillig
175 1.493 rillig /*
176 1.493 rillig * A Job manages the shell commands that are run to create a single target.
177 1.493 rillig * Each job is run in a separate subprocess by a shell. Several jobs can run
178 1.493 rillig * in parallel.
179 1.493 rillig *
180 1.493 rillig * The shell commands for the target are written to a temporary file,
181 1.493 rillig * then the shell is run with the temporary file as stdin, and the output
182 1.493 rillig * of that shell is captured via a pipe.
183 1.493 rillig *
184 1.493 rillig * When a job is finished, Make_Update updates all parents of the node
185 1.493 rillig * that was just remade, marking them as ready to be made next if all
186 1.493 rillig * other dependencies are finished as well.
187 1.493 rillig */
188 1.493 rillig struct Job {
189 1.493 rillig /* The process ID of the shell running the commands */
190 1.493 rillig int pid;
191 1.493 rillig
192 1.493 rillig /* The target the child is making */
193 1.493 rillig GNode *node;
194 1.493 rillig
195 1.493 rillig /*
196 1.493 rillig * If one of the shell commands is "...", all following commands are
197 1.493 rillig * delayed until the .END node is made. This list node points to the
198 1.493 rillig * first of these commands, if any.
199 1.493 rillig */
200 1.493 rillig StringListNode *tailCmds;
201 1.493 rillig
202 1.493 rillig /* This is where the shell commands go. */
203 1.493 rillig FILE *cmdFILE;
204 1.493 rillig
205 1.493 rillig int exit_status; /* from wait4() in signal handler */
206 1.493 rillig
207 1.501 rillig enum JobStatus status;
208 1.493 rillig
209 1.493 rillig bool suspended;
210 1.493 rillig
211 1.493 rillig /* Ignore non-zero exits */
212 1.493 rillig bool ignerr;
213 1.493 rillig /* Output the command before or instead of running it. */
214 1.493 rillig bool echo;
215 1.493 rillig /* Target is a special one. */
216 1.493 rillig bool special;
217 1.493 rillig
218 1.493 rillig int inPipe; /* Pipe for reading output from job */
219 1.493 rillig int outPipe; /* Pipe for writing control commands */
220 1.493 rillig struct pollfd *inPollfd; /* pollfd associated with inPipe */
221 1.493 rillig
222 1.493 rillig #define JOB_BUFSIZE 1024
223 1.493 rillig /* Buffer for storing the output of the job, line by line. */
224 1.493 rillig char outBuf[JOB_BUFSIZE + 1];
225 1.493 rillig size_t curPos; /* Current position in outBuf. */
226 1.493 rillig
227 1.493 rillig #ifdef USE_META
228 1.493 rillig struct BuildMon bm;
229 1.493 rillig #endif
230 1.493 rillig };
231 1.493 rillig
232 1.1 cgd
233 1.341 rillig /*
234 1.341 rillig * A shell defines how the commands are run. All commands for a target are
235 1.275 rillig * written into a single file, which is then given to the shell to execute
236 1.275 rillig * the commands from it. The commands are written to the file using a few
237 1.275 rillig * templates for echo control and error control.
238 1.275 rillig *
239 1.275 rillig * The name of the shell is the basename for the predefined shells, such as
240 1.275 rillig * "sh", "csh", "bash". For custom shells, it is the full pathname, and its
241 1.275 rillig * basename is used to select the type of shell; the longest match wins.
242 1.275 rillig * So /usr/pkg/bin/bash has type sh, /usr/local/bin/tcsh has type csh.
243 1.275 rillig *
244 1.275 rillig * The echoing of command lines is controlled using hasEchoCtl, echoOff,
245 1.275 rillig * echoOn, noPrint and noPrintLen. When echoOff is executed by the shell, it
246 1.275 rillig * still outputs something, but this something is not interesting, therefore
247 1.275 rillig * it is filtered out using noPrint and noPrintLen.
248 1.275 rillig *
249 1.275 rillig * The error checking for individual commands is controlled using hasErrCtl,
250 1.364 rillig * errOn, errOff and runChkTmpl.
251 1.275 rillig *
252 1.407 rillig * In case a shell doesn't have error control, echoTmpl is a printf template
253 1.407 rillig * for echoing the command, should echoing be on; runIgnTmpl is another
254 1.407 rillig * printf template for executing the command while ignoring the return
255 1.364 rillig * status. Finally runChkTmpl is a printf template for running the command and
256 1.275 rillig * causing the shell to exit on error. If any of these strings are empty when
257 1.421 rillig * hasErrCtl is false, the command will be executed anyway as is, and if it
258 1.319 rillig * causes an error, so be it. Any templates set up to echo the command will
259 1.407 rillig * escape any '$ ` \ "' characters in the command string to avoid unwanted
260 1.407 rillig * shell code injection, the escaped command is safe to use in double quotes.
261 1.275 rillig *
262 1.275 rillig * The command-line flags "echo" and "exit" also control the behavior. The
263 1.275 rillig * "echo" flag causes the shell to start echoing commands right away. The
264 1.275 rillig * "exit" flag causes the shell to exit when an error is detected in one of
265 1.275 rillig * the commands.
266 1.266 rillig */
267 1.266 rillig typedef struct Shell {
268 1.272 rillig
269 1.341 rillig /*
270 1.341 rillig * The name of the shell. For Bourne and C shells, this is used only
271 1.341 rillig * to find the shell description when used as the single source of a
272 1.341 rillig * .SHELL target. For user-defined shells, this is the full path of
273 1.341 rillig * the shell.
274 1.341 rillig */
275 1.341 rillig const char *name;
276 1.341 rillig
277 1.421 rillig bool hasEchoCtl; /* whether both echoOff and echoOn are there */
278 1.365 rillig const char *echoOff; /* command to turn echoing off */
279 1.365 rillig const char *echoOn; /* command to turn echoing back on */
280 1.375 rillig const char *noPrint; /* text to skip when printing output from the
281 1.272 rillig * shell. This is usually the same as echoOff */
282 1.341 rillig size_t noPrintLen; /* length of noPrint command */
283 1.272 rillig
284 1.421 rillig bool hasErrCtl; /* whether error checking can be controlled
285 1.375 rillig * for individual commands */
286 1.359 rillig const char *errOn; /* command to turn on error checking */
287 1.359 rillig const char *errOff; /* command to turn off error checking */
288 1.364 rillig
289 1.364 rillig const char *echoTmpl; /* template to echo a command */
290 1.443 rillig const char *runIgnTmpl; /* template to run a command without error
291 1.443 rillig * checking */
292 1.443 rillig const char *runChkTmpl; /* template to run a command with error
293 1.443 rillig * checking */
294 1.266 rillig
295 1.443 rillig /*
296 1.443 rillig * A string literal that results in a newline character when it
297 1.443 rillig * occurs outside of any 'quote' or "quote" characters.
298 1.443 rillig */
299 1.341 rillig const char *newline;
300 1.341 rillig char commentChar; /* character used by shell for comment lines */
301 1.341 rillig
302 1.366 rillig const char *echoFlag; /* shell flag to echo commands */
303 1.375 rillig const char *errFlag; /* shell flag to exit on error */
304 1.266 rillig } Shell;
305 1.266 rillig
306 1.360 rillig typedef struct CommandFlags {
307 1.407 rillig /* Whether to echo the command before or instead of running it. */
308 1.421 rillig bool echo;
309 1.348 rillig
310 1.360 rillig /* Run the command even in -n or -N mode. */
311 1.421 rillig bool always;
312 1.349 rillig
313 1.349 rillig /*
314 1.429 rillig * true if we turned error checking off before writing the command to
315 1.429 rillig * the commands file and need to turn it back on
316 1.349 rillig */
317 1.421 rillig bool ignerr;
318 1.360 rillig } CommandFlags;
319 1.347 rillig
320 1.1 cgd /*
321 1.371 rillig * Write shell commands to a file.
322 1.371 rillig *
323 1.371 rillig * TODO: keep track of whether commands are echoed.
324 1.371 rillig * TODO: keep track of whether error checking is active.
325 1.371 rillig */
326 1.371 rillig typedef struct ShellWriter {
327 1.371 rillig FILE *f;
328 1.374 rillig
329 1.374 rillig /* we've sent 'set -x' */
330 1.421 rillig bool xtraced;
331 1.374 rillig
332 1.371 rillig } ShellWriter;
333 1.371 rillig
334 1.461 rillig /* error handling variables */
335 1.331 rillig static int job_errors = 0; /* number of errors reported */
336 1.427 rillig static enum { /* Why is the make aborting? */
337 1.341 rillig ABORT_NONE,
338 1.427 rillig ABORT_ERROR, /* Aborted because of an error */
339 1.427 rillig ABORT_INTERRUPT, /* Aborted because it was interrupted */
340 1.341 rillig ABORT_WAIT /* Waiting for jobs to finish */
341 1.427 rillig } aborting = ABORT_NONE;
342 1.341 rillig #define JOB_TOKENS "+EI+" /* Token to requeue for each abort state */
343 1.1 cgd
344 1.498 rillig static const char aborting_name[][6] = { "NONE", "ERROR", "INTR", "WAIT" };
345 1.498 rillig
346 1.461 rillig /* Tracks the number of tokens currently "out" to build jobs. */
347 1.103 dsl int jobTokensRunning = 0;
348 1.103 dsl
349 1.1 cgd /*
350 1.1 cgd * Descriptions for various shells.
351 1.121 apb *
352 1.133 apb * The build environment may set DEFSHELL_INDEX to one of
353 1.133 apb * DEFSHELL_INDEX_SH, DEFSHELL_INDEX_KSH, or DEFSHELL_INDEX_CSH, to
354 1.243 rillig * select one of the predefined shells as the default shell.
355 1.133 apb *
356 1.133 apb * Alternatively, the build environment may set DEFSHELL_CUSTOM to the
357 1.133 apb * name or the full path of a sh-compatible shell, which will be used as
358 1.133 apb * the default shell.
359 1.133 apb *
360 1.133 apb * ".SHELL" lines in Makefiles can choose the default shell from the
361 1.289 rillig * set defined here, or add additional shells.
362 1.133 apb */
363 1.133 apb
364 1.133 apb #ifdef DEFSHELL_CUSTOM
365 1.133 apb #define DEFSHELL_INDEX_CUSTOM 0
366 1.133 apb #define DEFSHELL_INDEX_SH 1
367 1.133 apb #define DEFSHELL_INDEX_KSH 2
368 1.133 apb #define DEFSHELL_INDEX_CSH 3
369 1.461 rillig #else
370 1.133 apb #define DEFSHELL_INDEX_SH 0
371 1.133 apb #define DEFSHELL_INDEX_KSH 1
372 1.133 apb #define DEFSHELL_INDEX_CSH 2
373 1.461 rillig #endif
374 1.133 apb
375 1.133 apb #ifndef DEFSHELL_INDEX
376 1.133 apb #define DEFSHELL_INDEX 0 /* DEFSHELL_INDEX_CUSTOM or DEFSHELL_INDEX_SH */
377 1.461 rillig #endif
378 1.133 apb
379 1.341 rillig static Shell shells[] = {
380 1.133 apb #ifdef DEFSHELL_CUSTOM
381 1.1 cgd /*
382 1.121 apb * An sh-compatible shell with a non-standard name.
383 1.121 apb *
384 1.121 apb * Keep this in sync with the "sh" description below, but avoid
385 1.121 apb * non-portable features that might not be supplied by all
386 1.121 apb * sh-compatible shells.
387 1.1 cgd */
388 1.341 rillig {
389 1.341 rillig DEFSHELL_CUSTOM, /* .name */
390 1.421 rillig false, /* .hasEchoCtl */
391 1.341 rillig "", /* .echoOff */
392 1.341 rillig "", /* .echoOn */
393 1.341 rillig "", /* .noPrint */
394 1.341 rillig 0, /* .noPrintLen */
395 1.421 rillig false, /* .hasErrCtl */
396 1.358 rillig "", /* .errOn */
397 1.359 rillig "", /* .errOff */
398 1.364 rillig "echo \"%s\"\n", /* .echoTmpl */
399 1.364 rillig "%s\n", /* .runIgnTmpl */
400 1.364 rillig "{ %s \n} || exit $?\n", /* .runChkTmpl */
401 1.341 rillig "'\n'", /* .newline */
402 1.341 rillig '#', /* .commentChar */
403 1.375 rillig "", /* .echoFlag */
404 1.375 rillig "", /* .errFlag */
405 1.341 rillig },
406 1.133 apb #endif /* DEFSHELL_CUSTOM */
407 1.1 cgd /*
408 1.1 cgd * SH description. Echo control is also possible and, under
409 1.1 cgd * sun UNIX anyway, one can even control error checking.
410 1.1 cgd */
411 1.341 rillig {
412 1.341 rillig "sh", /* .name */
413 1.421 rillig false, /* .hasEchoCtl */
414 1.341 rillig "", /* .echoOff */
415 1.341 rillig "", /* .echoOn */
416 1.341 rillig "", /* .noPrint */
417 1.341 rillig 0, /* .noPrintLen */
418 1.421 rillig false, /* .hasErrCtl */
419 1.358 rillig "", /* .errOn */
420 1.359 rillig "", /* .errOff */
421 1.364 rillig "echo \"%s\"\n", /* .echoTmpl */
422 1.364 rillig "%s\n", /* .runIgnTmpl */
423 1.364 rillig "{ %s \n} || exit $?\n", /* .runChkTmpl */
424 1.341 rillig "'\n'", /* .newline */
425 1.341 rillig '#', /* .commentChar*/
426 1.121 apb #if defined(MAKE_NATIVE) && defined(__NetBSD__)
427 1.375 rillig /* XXX: -q is not really echoFlag, it's more like noEchoInSysFlag. */
428 1.375 rillig "q", /* .echoFlag */
429 1.27 christos #else
430 1.375 rillig "", /* .echoFlag */
431 1.27 christos #endif
432 1.375 rillig "", /* .errFlag */
433 1.341 rillig },
434 1.1 cgd /*
435 1.201 rillig * KSH description.
436 1.81 sjg */
437 1.341 rillig {
438 1.341 rillig "ksh", /* .name */
439 1.421 rillig true, /* .hasEchoCtl */
440 1.341 rillig "set +v", /* .echoOff */
441 1.341 rillig "set -v", /* .echoOn */
442 1.341 rillig "set +v", /* .noPrint */
443 1.341 rillig 6, /* .noPrintLen */
444 1.421 rillig false, /* .hasErrCtl */
445 1.358 rillig "", /* .errOn */
446 1.359 rillig "", /* .errOff */
447 1.364 rillig "echo \"%s\"\n", /* .echoTmpl */
448 1.364 rillig "%s\n", /* .runIgnTmpl */
449 1.364 rillig "{ %s \n} || exit $?\n", /* .runChkTmpl */
450 1.341 rillig "'\n'", /* .newline */
451 1.341 rillig '#', /* .commentChar */
452 1.375 rillig "v", /* .echoFlag */
453 1.375 rillig "", /* .errFlag */
454 1.341 rillig },
455 1.81 sjg /*
456 1.121 apb * CSH description. The csh can do echo control by playing
457 1.121 apb * with the setting of the 'echo' shell variable. Sadly,
458 1.121 apb * however, it is unable to do error control nicely.
459 1.121 apb */
460 1.341 rillig {
461 1.341 rillig "csh", /* .name */
462 1.421 rillig true, /* .hasEchoCtl */
463 1.341 rillig "unset verbose", /* .echoOff */
464 1.341 rillig "set verbose", /* .echoOn */
465 1.341 rillig "unset verbose", /* .noPrint */
466 1.341 rillig 13, /* .noPrintLen */
467 1.421 rillig false, /* .hasErrCtl */
468 1.358 rillig "", /* .errOn */
469 1.359 rillig "", /* .errOff */
470 1.364 rillig "echo \"%s\"\n", /* .echoTmpl */
471 1.364 rillig "csh -c \"%s || exit 0\"\n", /* .runIgnTmpl */
472 1.364 rillig "", /* .runChkTmpl */
473 1.341 rillig "'\\\n'", /* .newline */
474 1.341 rillig '#', /* .commentChar */
475 1.375 rillig "v", /* .echoFlag */
476 1.375 rillig "e", /* .errFlag */
477 1.341 rillig }
478 1.1 cgd };
479 1.243 rillig
480 1.391 rillig /*
481 1.391 rillig * This is the shell to which we pass all commands in the Makefile.
482 1.391 rillig * It is set by the Job_ParseShell function.
483 1.391 rillig */
484 1.355 rillig static Shell *shell = &shells[DEFSHELL_INDEX];
485 1.478 rillig char *shellPath; /* full pathname of executable image */
486 1.243 rillig const char *shellName = NULL; /* last component of shellPath */
487 1.174 sjg char *shellErrFlag = NULL;
488 1.407 rillig static char *shell_freeIt = NULL; /* Allocated memory for custom .SHELL */
489 1.1 cgd
490 1.1 cgd
491 1.285 rillig static Job *job_table; /* The structures that describe them */
492 1.285 rillig static Job *job_table_end; /* job_table + maxJobs */
493 1.492 rillig static bool wantToken;
494 1.421 rillig static bool lurking_children = false;
495 1.421 rillig static bool make_suspended = false; /* Whether we've seen a SIGTSTP (etc) */
496 1.40 sommerfe
497 1.35 christos /*
498 1.35 christos * Set of descriptors of pipes connected to
499 1.35 christos * the output channels of children
500 1.35 christos */
501 1.35 christos static struct pollfd *fds = NULL;
502 1.408 rillig static Job **jobByFdIndex = NULL;
503 1.408 rillig static nfds_t fdsLen = 0;
504 1.72 wiz static void watchfd(Job *);
505 1.72 wiz static void clearfd(Job *);
506 1.1 cgd
507 1.443 rillig static char *targPrefix = NULL; /* To identify a job change in the output. */
508 1.495 rillig static Job tokenPoolJob; /* token wait pseudo-job */
509 1.12 christos
510 1.73 gson static Job childExitJob; /* child exit pseudo-job */
511 1.502 rillig #define CEJ_CHILD_EXITED '.'
512 1.502 rillig #define CEJ_RESUME_JOBS 'R'
513 1.73 gson
514 1.341 rillig enum {
515 1.341 rillig npseudojobs = 2 /* number of pseudo-jobs */
516 1.341 rillig };
517 1.196 riastrad
518 1.104 dsl static sigset_t caught_signals; /* Set of signals we handle */
519 1.406 sjg static volatile sig_atomic_t caught_sigchld;
520 1.1 cgd
521 1.424 rillig static void CollectOutput(Job *, bool);
522 1.421 rillig static void JobInterrupt(bool, int) MAKE_ATTR_DEAD;
523 1.69 pk static void JobSigReset(void);
524 1.69 pk
525 1.339 rillig static void
526 1.340 rillig SwitchOutputTo(GNode *gn)
527 1.339 rillig {
528 1.340 rillig /* The node for which output was most recently produced. */
529 1.340 rillig static GNode *lastNode = NULL;
530 1.340 rillig
531 1.340 rillig if (gn == lastNode)
532 1.340 rillig return;
533 1.340 rillig lastNode = gn;
534 1.340 rillig
535 1.339 rillig if (opts.maxJobs != 1 && targPrefix != NULL && targPrefix[0] != '\0')
536 1.339 rillig (void)fprintf(stdout, "%s %s ---\n", targPrefix, gn->name);
537 1.339 rillig }
538 1.339 rillig
539 1.196 riastrad static unsigned
540 1.196 riastrad nfds_per_job(void)
541 1.196 riastrad {
542 1.197 sjg #if defined(USE_FILEMON) && !defined(USE_FILEMON_DEV)
543 1.341 rillig if (useMeta)
544 1.341 rillig return 2;
545 1.196 riastrad #endif
546 1.341 rillig return 1;
547 1.196 riastrad }
548 1.196 riastrad
549 1.354 rillig void
550 1.356 rillig Job_FlagsToString(const Job *job, char *buf, size_t bufsize)
551 1.354 rillig {
552 1.374 rillig snprintf(buf, bufsize, "%c%c%c",
553 1.356 rillig job->ignerr ? 'i' : '-',
554 1.357 rillig !job->echo ? 's' : '-',
555 1.374 rillig job->special ? 'S' : '-');
556 1.354 rillig }
557 1.354 rillig
558 1.493 rillig #ifdef USE_META
559 1.493 rillig struct BuildMon *
560 1.493 rillig Job_BuildMon(Job *job)
561 1.493 rillig {
562 1.493 rillig return &job->bm;
563 1.493 rillig }
564 1.493 rillig #endif
565 1.493 rillig
566 1.493 rillig GNode *
567 1.493 rillig Job_Node(Job *job)
568 1.493 rillig {
569 1.493 rillig return job->node;
570 1.493 rillig }
571 1.493 rillig
572 1.493 rillig int
573 1.493 rillig Job_Pid(Job *job)
574 1.493 rillig {
575 1.493 rillig return job->pid;
576 1.493 rillig }
577 1.493 rillig
578 1.114 dsl static void
579 1.428 rillig DumpJobs(const char *where)
580 1.114 dsl {
581 1.501 rillig const Job *job;
582 1.374 rillig char flags[4];
583 1.114 dsl
584 1.505 rillig debug_printf("%s, job table:\n", where);
585 1.341 rillig for (job = job_table; job < job_table_end; job++) {
586 1.356 rillig Job_FlagsToString(job, flags, sizeof flags);
587 1.501 rillig debug_printf("job %d, status %s, flags %s, pid %d\n",
588 1.501 rillig (int)(job - job_table), JobStatus_Name[job->status],
589 1.501 rillig flags, job->pid);
590 1.341 rillig }
591 1.114 dsl }
592 1.69 pk
593 1.69 pk /*
594 1.188 dholland * Delete the target of a failed, interrupted, or otherwise
595 1.188 dholland * unsuccessful job unless inhibited by .PRECIOUS.
596 1.188 dholland */
597 1.188 dholland static void
598 1.188 dholland JobDeleteTarget(GNode *gn)
599 1.188 dholland {
600 1.341 rillig const char *file;
601 1.341 rillig
602 1.341 rillig if (gn->type & OP_JOIN)
603 1.341 rillig return;
604 1.341 rillig if (gn->type & OP_PHONY)
605 1.341 rillig return;
606 1.447 rillig if (GNode_IsPrecious(gn))
607 1.341 rillig return;
608 1.341 rillig if (opts.noExecute)
609 1.341 rillig return;
610 1.287 rillig
611 1.341 rillig file = GNode_Path(gn);
612 1.456 rillig if (unlink_file(file) == 0)
613 1.341 rillig Error("*** %s removed", file);
614 1.188 dholland }
615 1.188 dholland
616 1.500 rillig /* Lock the jobs table and the jobs therein. */
617 1.388 rillig static void
618 1.500 rillig JobsTable_Lock(sigset_t *omaskp)
619 1.69 pk {
620 1.500 rillig if (sigprocmask(SIG_BLOCK, &caught_signals, omaskp) != 0)
621 1.505 rillig Punt("JobsTable_Lock: %s", strerror(errno));
622 1.69 pk }
623 1.69 pk
624 1.500 rillig /* Unlock the jobs table and the jobs therein. */
625 1.388 rillig static void
626 1.500 rillig JobsTable_Unlock(sigset_t *omaskp)
627 1.69 pk {
628 1.96 christos (void)sigprocmask(SIG_SETMASK, omaskp, NULL);
629 1.69 pk }
630 1.1 cgd
631 1.122 dsl static void
632 1.497 rillig SetNonblocking(int fd)
633 1.497 rillig {
634 1.497 rillig int flags = fcntl(fd, F_GETFL, 0);
635 1.497 rillig if (flags == -1)
636 1.497 rillig Punt("SetNonblocking.get: %s", strerror(errno));
637 1.497 rillig flags |= O_NONBLOCK;
638 1.497 rillig if (fcntl(fd, F_SETFL, flags) == -1)
639 1.497 rillig Punt("SetNonblocking.set: %s", strerror(errno));
640 1.497 rillig }
641 1.497 rillig
642 1.497 rillig static void
643 1.122 dsl JobCreatePipe(Job *job, int minfd)
644 1.122 dsl {
645 1.497 rillig int i;
646 1.341 rillig int pipe_fds[2];
647 1.122 dsl
648 1.341 rillig if (pipe(pipe_fds) == -1)
649 1.505 rillig Punt("JobCreatePipe: %s", strerror(errno));
650 1.122 dsl
651 1.341 rillig for (i = 0; i < 2; i++) {
652 1.461 rillig /* Avoid using low-numbered fds */
653 1.497 rillig int fd = fcntl(pipe_fds[i], F_DUPFD, minfd);
654 1.341 rillig if (fd != -1) {
655 1.341 rillig close(pipe_fds[i]);
656 1.341 rillig pipe_fds[i] = fd;
657 1.341 rillig }
658 1.247 rillig }
659 1.201 rillig
660 1.341 rillig job->inPipe = pipe_fds[0];
661 1.341 rillig job->outPipe = pipe_fds[1];
662 1.249 rillig
663 1.341 rillig if (fcntl(job->inPipe, F_SETFD, FD_CLOEXEC) == -1)
664 1.505 rillig Punt("SetCloseOnExec: %s", strerror(errno));
665 1.341 rillig if (fcntl(job->outPipe, F_SETFD, FD_CLOEXEC) == -1)
666 1.505 rillig Punt("SetCloseOnExec: %s", strerror(errno));
667 1.341 rillig
668 1.341 rillig /*
669 1.341 rillig * We mark the input side of the pipe non-blocking; we poll(2) the
670 1.341 rillig * pipe when we're waiting for a job token, but we might lose the
671 1.341 rillig * race for the token when a new one becomes available, so the read
672 1.341 rillig * from the pipe should not block.
673 1.341 rillig */
674 1.497 rillig SetNonblocking(job->inPipe);
675 1.122 dsl }
676 1.122 dsl
677 1.243 rillig /* Pass the signal to each running job. */
678 1.114 dsl static void
679 1.114 dsl JobCondPassSig(int signo)
680 1.1 cgd {
681 1.341 rillig Job *job;
682 1.106 dsl
683 1.505 rillig DEBUG1(JOB, "JobCondPassSig: signal %d\n", signo);
684 1.116 dsl
685 1.341 rillig for (job = job_table; job < job_table_end; job++) {
686 1.341 rillig if (job->status != JOB_ST_RUNNING)
687 1.341 rillig continue;
688 1.505 rillig DEBUG2(JOB, "JobCondPassSig passing signal %d to pid %d\n",
689 1.341 rillig signo, job->pid);
690 1.341 rillig KILLPG(job->pid, signo);
691 1.341 rillig }
692 1.1 cgd }
693 1.1 cgd
694 1.37 sommerfe static void
695 1.503 rillig WriteOrDie(int fd, char ch)
696 1.37 sommerfe {
697 1.503 rillig if (write(fd, &ch, 1) != 1)
698 1.503 rillig execDie("write", "child");
699 1.37 sommerfe }
700 1.37 sommerfe
701 1.502 rillig static void
702 1.502 rillig HandleSIGCHLD(int signo MAKE_ATTR_UNUSED)
703 1.502 rillig {
704 1.502 rillig caught_sigchld = 1;
705 1.502 rillig /* Wake up from poll(). */
706 1.503 rillig WriteOrDie(childExitJob.outPipe, CEJ_CHILD_EXITED);
707 1.502 rillig }
708 1.37 sommerfe
709 1.47 sommerfe static void
710 1.502 rillig HandleSIGCONT(int signo MAKE_ATTR_UNUSED)
711 1.47 sommerfe {
712 1.503 rillig WriteOrDie(childExitJob.outPipe, CEJ_RESUME_JOBS);
713 1.47 sommerfe }
714 1.37 sommerfe
715 1.341 rillig /*
716 1.341 rillig * Pass a signal on to all jobs, then resend to ourselves.
717 1.341 rillig * We die by the same signal.
718 1.341 rillig */
719 1.162 joerg MAKE_ATTR_DEAD static void
720 1.116 dsl JobPassSig_int(int signo)
721 1.116 dsl {
722 1.341 rillig /* Run .INTERRUPT target then exit */
723 1.421 rillig JobInterrupt(true, signo);
724 1.116 dsl }
725 1.116 dsl
726 1.341 rillig /*
727 1.341 rillig * Pass a signal on to all jobs, then resend to ourselves.
728 1.341 rillig * We die by the same signal.
729 1.341 rillig */
730 1.162 joerg MAKE_ATTR_DEAD static void
731 1.116 dsl JobPassSig_term(int signo)
732 1.116 dsl {
733 1.341 rillig /* Dont run .INTERRUPT target then exit */
734 1.421 rillig JobInterrupt(false, signo);
735 1.116 dsl }
736 1.116 dsl
737 1.116 dsl static void
738 1.116 dsl JobPassSig_suspend(int signo)
739 1.1 cgd {
740 1.341 rillig sigset_t nmask, omask;
741 1.341 rillig struct sigaction act;
742 1.118 dsl
743 1.341 rillig /* Suppress job started/continued messages */
744 1.421 rillig make_suspended = true;
745 1.16 christos
746 1.341 rillig /* Pass the signal onto every job */
747 1.341 rillig JobCondPassSig(signo);
748 1.1 cgd
749 1.341 rillig /*
750 1.341 rillig * Send ourselves the signal now we've given the message to everyone
751 1.341 rillig * else. Note we block everything else possible while we're getting
752 1.341 rillig * the signal. This ensures that all our jobs get continued when we
753 1.341 rillig * wake up before we take any other signal.
754 1.341 rillig */
755 1.341 rillig sigfillset(&nmask);
756 1.341 rillig sigdelset(&nmask, signo);
757 1.341 rillig (void)sigprocmask(SIG_SETMASK, &nmask, &omask);
758 1.341 rillig
759 1.341 rillig act.sa_handler = SIG_DFL;
760 1.341 rillig sigemptyset(&act.sa_mask);
761 1.341 rillig act.sa_flags = 0;
762 1.341 rillig (void)sigaction(signo, &act, NULL);
763 1.13 christos
764 1.505 rillig DEBUG1(JOB, "JobPassSig_suspend passing signal %d to self\n", signo);
765 1.1 cgd
766 1.341 rillig (void)kill(getpid(), signo);
767 1.115 dsl
768 1.341 rillig /*
769 1.341 rillig * We've been continued.
770 1.341 rillig *
771 1.461 rillig * A whole host of signals is going to happen!
772 1.341 rillig * SIGCHLD for any processes that actually suspended themselves.
773 1.461 rillig * SIGCHLD for any processes that exited while we were asleep.
774 1.461 rillig * The SIGCONT that actually caused us to wake up.
775 1.341 rillig *
776 1.341 rillig * Since we defer passing the SIGCONT on to our children until
777 1.341 rillig * the main processing loop, we can be sure that all the SIGCHLD
778 1.341 rillig * events will have happened by then - and that the waitpid() will
779 1.341 rillig * collect the child 'suspended' events.
780 1.341 rillig * For correct sequencing we just need to ensure we process the
781 1.341 rillig * waitpid() before passing on the SIGCONT.
782 1.341 rillig *
783 1.341 rillig * In any case nothing else is needed here.
784 1.341 rillig */
785 1.1 cgd
786 1.341 rillig /* Restore handler and signal mask */
787 1.341 rillig act.sa_handler = JobPassSig_suspend;
788 1.341 rillig (void)sigaction(signo, &act, NULL);
789 1.341 rillig (void)sigprocmask(SIG_SETMASK, &omask, NULL);
790 1.1 cgd }
791 1.1 cgd
792 1.114 dsl static Job *
793 1.501 rillig JobFindPid(int pid, enum JobStatus status, bool isJobs)
794 1.1 cgd {
795 1.341 rillig Job *job;
796 1.114 dsl
797 1.341 rillig for (job = job_table; job < job_table_end; job++) {
798 1.341 rillig if (job->status == status && job->pid == pid)
799 1.341 rillig return job;
800 1.341 rillig }
801 1.341 rillig if (DEBUG(JOB) && isJobs)
802 1.428 rillig DumpJobs("no pid");
803 1.341 rillig return NULL;
804 1.12 christos }
805 1.12 christos
806 1.298 rillig /* Parse leading '@', '-' and '+', which control the exact execution mode. */
807 1.298 rillig static void
808 1.383 rillig ParseCommandFlags(char **pp, CommandFlags *out_cmdFlags)
809 1.341 rillig {
810 1.341 rillig char *p = *pp;
811 1.421 rillig out_cmdFlags->echo = true;
812 1.421 rillig out_cmdFlags->ignerr = false;
813 1.421 rillig out_cmdFlags->always = false;
814 1.341 rillig
815 1.341 rillig for (;;) {
816 1.341 rillig if (*p == '@')
817 1.360 rillig out_cmdFlags->echo = DEBUG(LOUD);
818 1.341 rillig else if (*p == '-')
819 1.421 rillig out_cmdFlags->ignerr = true;
820 1.341 rillig else if (*p == '+')
821 1.421 rillig out_cmdFlags->always = true;
822 1.457 christos else if (!ch_isspace(*p))
823 1.461 rillig /* Ignore whitespace for compatibility with GNU make */
824 1.341 rillig break;
825 1.341 rillig p++;
826 1.341 rillig }
827 1.298 rillig
828 1.341 rillig *pp = p;
829 1.298 rillig }
830 1.298 rillig
831 1.299 rillig /* Escape a string for a double-quoted string literal in sh, csh and ksh. */
832 1.299 rillig static char *
833 1.299 rillig EscapeShellDblQuot(const char *cmd)
834 1.299 rillig {
835 1.341 rillig size_t i, j;
836 1.299 rillig
837 1.341 rillig /* Worst that could happen is every char needs escaping. */
838 1.341 rillig char *esc = bmake_malloc(strlen(cmd) * 2 + 1);
839 1.341 rillig for (i = 0, j = 0; cmd[i] != '\0'; i++, j++) {
840 1.341 rillig if (cmd[i] == '$' || cmd[i] == '`' || cmd[i] == '\\' ||
841 1.341 rillig cmd[i] == '"')
842 1.341 rillig esc[j++] = '\\';
843 1.341 rillig esc[j] = cmd[i];
844 1.341 rillig }
845 1.341 rillig esc[j] = '\0';
846 1.299 rillig
847 1.341 rillig return esc;
848 1.299 rillig }
849 1.299 rillig
850 1.318 rillig static void
851 1.429 rillig ShellWriter_WriteFmt(ShellWriter *wr, const char *fmt, const char *arg)
852 1.318 rillig {
853 1.343 rillig DEBUG1(JOB, fmt, arg);
854 1.318 rillig
855 1.371 rillig (void)fprintf(wr->f, fmt, arg);
856 1.436 sjg if (wr->f == stdout)
857 1.436 sjg (void)fflush(wr->f);
858 1.371 rillig }
859 1.371 rillig
860 1.371 rillig static void
861 1.429 rillig ShellWriter_WriteLine(ShellWriter *wr, const char *line)
862 1.318 rillig {
863 1.429 rillig ShellWriter_WriteFmt(wr, "%s\n", line);
864 1.318 rillig }
865 1.318 rillig
866 1.373 rillig static void
867 1.373 rillig ShellWriter_EchoOff(ShellWriter *wr)
868 1.373 rillig {
869 1.373 rillig if (shell->hasEchoCtl)
870 1.429 rillig ShellWriter_WriteLine(wr, shell->echoOff);
871 1.373 rillig }
872 1.373 rillig
873 1.373 rillig static void
874 1.373 rillig ShellWriter_EchoCmd(ShellWriter *wr, const char *escCmd)
875 1.373 rillig {
876 1.429 rillig ShellWriter_WriteFmt(wr, shell->echoTmpl, escCmd);
877 1.373 rillig }
878 1.373 rillig
879 1.373 rillig static void
880 1.373 rillig ShellWriter_EchoOn(ShellWriter *wr)
881 1.373 rillig {
882 1.373 rillig if (shell->hasEchoCtl)
883 1.429 rillig ShellWriter_WriteLine(wr, shell->echoOn);
884 1.373 rillig }
885 1.373 rillig
886 1.374 rillig static void
887 1.374 rillig ShellWriter_TraceOn(ShellWriter *wr)
888 1.374 rillig {
889 1.374 rillig if (!wr->xtraced) {
890 1.429 rillig ShellWriter_WriteLine(wr, "set -x");
891 1.421 rillig wr->xtraced = true;
892 1.374 rillig }
893 1.374 rillig }
894 1.374 rillig
895 1.346 rillig static void
896 1.421 rillig ShellWriter_ErrOff(ShellWriter *wr, bool echo)
897 1.346 rillig {
898 1.382 rillig if (echo)
899 1.373 rillig ShellWriter_EchoOff(wr);
900 1.429 rillig ShellWriter_WriteLine(wr, shell->errOff);
901 1.382 rillig if (echo)
902 1.382 rillig ShellWriter_EchoOn(wr);
903 1.382 rillig }
904 1.382 rillig
905 1.382 rillig static void
906 1.421 rillig ShellWriter_ErrOn(ShellWriter *wr, bool echo)
907 1.382 rillig {
908 1.382 rillig if (echo)
909 1.382 rillig ShellWriter_EchoOff(wr);
910 1.429 rillig ShellWriter_WriteLine(wr, shell->errOn);
911 1.382 rillig if (echo)
912 1.373 rillig ShellWriter_EchoOn(wr);
913 1.346 rillig }
914 1.346 rillig
915 1.346 rillig /*
916 1.384 rillig * The shell has no built-in error control, so emulate error control by
917 1.384 rillig * enclosing each shell command in a template like "{ %s \n } || exit $?"
918 1.384 rillig * (configurable per shell).
919 1.346 rillig */
920 1.346 rillig static void
921 1.429 rillig JobWriteSpecialsEchoCtl(Job *job, ShellWriter *wr, CommandFlags *inout_cmdFlags,
922 1.360 rillig const char *escCmd, const char **inout_cmdTemplate)
923 1.346 rillig {
924 1.452 rillig /* XXX: Why is the whole job modified at this point? */
925 1.421 rillig job->ignerr = true;
926 1.346 rillig
927 1.360 rillig if (job->echo && inout_cmdFlags->echo) {
928 1.373 rillig ShellWriter_EchoOff(wr);
929 1.373 rillig ShellWriter_EchoCmd(wr, escCmd);
930 1.384 rillig
931 1.384 rillig /*
932 1.384 rillig * Leave echoing off so the user doesn't see the commands
933 1.384 rillig * for toggling the error checking.
934 1.384 rillig */
935 1.421 rillig inout_cmdFlags->echo = false;
936 1.346 rillig }
937 1.364 rillig *inout_cmdTemplate = shell->runIgnTmpl;
938 1.346 rillig
939 1.346 rillig /*
940 1.364 rillig * The template runIgnTmpl already takes care of ignoring errors,
941 1.364 rillig * so pretend error checking is still on.
942 1.364 rillig * XXX: What effects does this have, and why is it necessary?
943 1.346 rillig */
944 1.421 rillig inout_cmdFlags->ignerr = false;
945 1.346 rillig }
946 1.346 rillig
947 1.345 rillig static void
948 1.429 rillig JobWriteSpecials(Job *job, ShellWriter *wr, const char *escCmd, bool run,
949 1.370 rillig CommandFlags *inout_cmdFlags, const char **inout_cmdTemplate)
950 1.345 rillig {
951 1.462 rillig if (!run)
952 1.421 rillig inout_cmdFlags->ignerr = false;
953 1.462 rillig else if (shell->hasErrCtl)
954 1.373 rillig ShellWriter_ErrOff(wr, job->echo && inout_cmdFlags->echo);
955 1.364 rillig else if (shell->runIgnTmpl != NULL && shell->runIgnTmpl[0] != '\0') {
956 1.429 rillig JobWriteSpecialsEchoCtl(job, wr, inout_cmdFlags, escCmd,
957 1.349 rillig inout_cmdTemplate);
958 1.346 rillig } else
959 1.421 rillig inout_cmdFlags->ignerr = false;
960 1.345 rillig }
961 1.345 rillig
962 1.338 rillig /*
963 1.429 rillig * Write a shell command to the job's commands file, to be run later.
964 1.72 wiz *
965 1.358 rillig * If the command starts with '@' and neither the -s nor the -n flag was
966 1.429 rillig * given to make, stick a shell-specific echoOff command in the script.
967 1.358 rillig *
968 1.358 rillig * If the command starts with '-' and the shell has no error control (none
969 1.429 rillig * of the predefined shells has that), ignore errors for the entire job.
970 1.358 rillig *
971 1.429 rillig * XXX: Why ignore errors for the entire job? This is even documented in the
972 1.429 rillig * manual page, but without any rationale since there is no known rationale.
973 1.429 rillig *
974 1.429 rillig * XXX: The manual page says the '-' "affects the entire job", but that's not
975 1.429 rillig * accurate. The '-' does not affect the commands before the '-'.
976 1.429 rillig *
977 1.429 rillig * If the command is just "...", skip all further commands of this job. These
978 1.429 rillig * commands are attached to the .END node instead and will be run by
979 1.429 rillig * Job_Finish after all other targets have been made.
980 1.1 cgd */
981 1.270 rillig static void
982 1.429 rillig JobWriteCommand(Job *job, ShellWriter *wr, StringListNode *ln, const char *ucmd)
983 1.1 cgd {
984 1.421 rillig bool run;
985 1.347 rillig
986 1.360 rillig CommandFlags cmdFlags;
987 1.429 rillig /* Template for writing a command to the shell file */
988 1.341 rillig const char *cmdTemplate;
989 1.363 rillig char *xcmd; /* The expanded command */
990 1.363 rillig char *xcmdStart;
991 1.363 rillig char *escCmd; /* xcmd escaped to be used in double quotes */
992 1.1 cgd
993 1.351 rillig run = GNode_ShouldExecute(job->node);
994 1.298 rillig
995 1.476 rillig xcmd = Var_SubstInTarget(ucmd, job->node);
996 1.341 rillig /* TODO: handle errors */
997 1.363 rillig xcmdStart = xcmd;
998 1.1 cgd
999 1.341 rillig cmdTemplate = "%s\n";
1000 1.83 jmc
1001 1.383 rillig ParseCommandFlags(&xcmd, &cmdFlags);
1002 1.83 jmc
1003 1.351 rillig /* The '+' command flag overrides the -n or -N options. */
1004 1.360 rillig if (cmdFlags.always && !run) {
1005 1.341 rillig /*
1006 1.341 rillig * We're not actually executing anything...
1007 1.341 rillig * but this one needs to be - use compat mode just for it.
1008 1.341 rillig */
1009 1.441 rillig (void)Compat_RunCommand(ucmd, job->node, ln);
1010 1.363 rillig free(xcmdStart);
1011 1.341 rillig return;
1012 1.1 cgd }
1013 1.1 cgd
1014 1.341 rillig /*
1015 1.363 rillig * If the shell doesn't have error control, the alternate echoing
1016 1.363 rillig * will be done (to avoid showing additional error checking code)
1017 1.363 rillig * and this needs some characters escaped.
1018 1.341 rillig */
1019 1.363 rillig escCmd = shell->hasErrCtl ? NULL : EscapeShellDblQuot(xcmd);
1020 1.341 rillig
1021 1.360 rillig if (!cmdFlags.echo) {
1022 1.463 rillig if (job->echo && run && shell->hasEchoCtl)
1023 1.373 rillig ShellWriter_EchoOff(wr);
1024 1.463 rillig else if (shell->hasErrCtl)
1025 1.463 rillig cmdFlags.echo = true;
1026 1.341 rillig }
1027 1.341 rillig
1028 1.360 rillig if (cmdFlags.ignerr) {
1029 1.429 rillig JobWriteSpecials(job, wr, escCmd, run, &cmdFlags, &cmdTemplate);
1030 1.341 rillig } else {
1031 1.341 rillig
1032 1.1 cgd /*
1033 1.341 rillig * If errors are being checked and the shell doesn't have
1034 1.364 rillig * error control but does supply an runChkTmpl template, then
1035 1.341 rillig * set up commands to run through it.
1036 1.341 rillig */
1037 1.341 rillig
1038 1.396 rillig if (!shell->hasErrCtl && shell->runChkTmpl != NULL &&
1039 1.364 rillig shell->runChkTmpl[0] != '\0') {
1040 1.360 rillig if (job->echo && cmdFlags.echo) {
1041 1.373 rillig ShellWriter_EchoOff(wr);
1042 1.373 rillig ShellWriter_EchoCmd(wr, escCmd);
1043 1.421 rillig cmdFlags.echo = false;
1044 1.341 rillig }
1045 1.341 rillig /*
1046 1.367 rillig * If it's a comment line or blank, avoid the possible
1047 1.381 rillig * syntax error generated by "{\n} || exit $?".
1048 1.341 rillig */
1049 1.367 rillig cmdTemplate = escCmd[0] == shell->commentChar ||
1050 1.367 rillig escCmd[0] == '\0'
1051 1.367 rillig ? shell->runIgnTmpl
1052 1.367 rillig : shell->runChkTmpl;
1053 1.421 rillig cmdFlags.ignerr = false;
1054 1.341 rillig }
1055 1.1 cgd }
1056 1.83 jmc
1057 1.374 rillig if (DEBUG(SHELL) && strcmp(shellName, "sh") == 0)
1058 1.374 rillig ShellWriter_TraceOn(wr);
1059 1.16 christos
1060 1.429 rillig ShellWriter_WriteFmt(wr, cmdTemplate, xcmd);
1061 1.363 rillig free(xcmdStart);
1062 1.341 rillig free(escCmd);
1063 1.382 rillig
1064 1.382 rillig if (cmdFlags.ignerr)
1065 1.382 rillig ShellWriter_ErrOn(wr, cmdFlags.echo && job->echo);
1066 1.382 rillig
1067 1.373 rillig if (!cmdFlags.echo)
1068 1.373 rillig ShellWriter_EchoOn(wr);
1069 1.269 rillig }
1070 1.269 rillig
1071 1.341 rillig /*
1072 1.429 rillig * Write all commands to the shell file that is later executed.
1073 1.270 rillig *
1074 1.429 rillig * The special command "..." stops writing and saves the remaining commands
1075 1.409 rillig * to be executed later, when the target '.END' is made.
1076 1.362 rillig *
1077 1.362 rillig * Return whether at least one command was written to the shell file.
1078 1.341 rillig */
1079 1.421 rillig static bool
1080 1.429 rillig JobWriteCommands(Job *job)
1081 1.269 rillig {
1082 1.341 rillig StringListNode *ln;
1083 1.421 rillig bool seen = false;
1084 1.422 rillig ShellWriter wr;
1085 1.422 rillig
1086 1.422 rillig wr.f = job->cmdFILE;
1087 1.422 rillig wr.xtraced = false;
1088 1.341 rillig
1089 1.341 rillig for (ln = job->node->commands.first; ln != NULL; ln = ln->next) {
1090 1.341 rillig const char *cmd = ln->datum;
1091 1.269 rillig
1092 1.341 rillig if (strcmp(cmd, "...") == 0) {
1093 1.341 rillig job->node->type |= OP_SAVE_CMDS;
1094 1.341 rillig job->tailCmds = ln->next;
1095 1.341 rillig break;
1096 1.341 rillig }
1097 1.270 rillig
1098 1.429 rillig JobWriteCommand(job, &wr, ln, ln->datum);
1099 1.421 rillig seen = true;
1100 1.314 rillig }
1101 1.362 rillig
1102 1.362 rillig return seen;
1103 1.1 cgd }
1104 1.1 cgd
1105 1.403 rillig /*
1106 1.403 rillig * Save the delayed commands (those after '...'), to be executed later in
1107 1.403 rillig * the '.END' node, when everything else is done.
1108 1.403 rillig */
1109 1.236 rillig static void
1110 1.236 rillig JobSaveCommands(Job *job)
1111 1.1 cgd {
1112 1.341 rillig StringListNode *ln;
1113 1.236 rillig
1114 1.341 rillig for (ln = job->tailCmds; ln != NULL; ln = ln->next) {
1115 1.341 rillig const char *cmd = ln->datum;
1116 1.341 rillig char *expanded_cmd;
1117 1.403 rillig /*
1118 1.403 rillig * XXX: This Var_Subst is only intended to expand the dynamic
1119 1.341 rillig * variables such as .TARGET, .IMPSRC. It is not intended to
1120 1.403 rillig * expand the other variables as well; see deptgt-end.mk.
1121 1.403 rillig */
1122 1.476 rillig expanded_cmd = Var_SubstInTarget(cmd, job->node);
1123 1.341 rillig /* TODO: handle errors */
1124 1.341 rillig Lst_Append(&Targ_GetEndNode()->commands, expanded_cmd);
1125 1.473 rillig Parse_RegisterCommand(expanded_cmd);
1126 1.341 rillig }
1127 1.12 christos }
1128 1.12 christos
1129 1.12 christos
1130 1.243 rillig /* Called to close both input and output pipes when a job is finished. */
1131 1.12 christos static void
1132 1.320 rillig JobClosePipes(Job *job)
1133 1.12 christos {
1134 1.341 rillig clearfd(job);
1135 1.341 rillig (void)close(job->outPipe);
1136 1.341 rillig job->outPipe = -1;
1137 1.341 rillig
1138 1.424 rillig CollectOutput(job, true);
1139 1.341 rillig (void)close(job->inPipe);
1140 1.341 rillig job->inPipe = -1;
1141 1.1 cgd }
1142 1.1 cgd
1143 1.378 rillig static void
1144 1.430 christos DebugFailedJob(const Job *job)
1145 1.430 christos {
1146 1.432 rillig const StringListNode *ln;
1147 1.430 christos
1148 1.430 christos if (!DEBUG(ERROR))
1149 1.430 christos return;
1150 1.430 christos
1151 1.437 rillig debug_printf("\n");
1152 1.437 rillig debug_printf("*** Failed target: %s\n", job->node->name);
1153 1.469 rillig debug_printf("*** In directory: %s\n", curdir);
1154 1.437 rillig debug_printf("*** Failed commands:\n");
1155 1.437 rillig for (ln = job->node->commands.first; ln != NULL; ln = ln->next) {
1156 1.439 rillig const char *cmd = ln->datum;
1157 1.439 rillig debug_printf("\t%s\n", cmd);
1158 1.437 rillig
1159 1.439 rillig if (strchr(cmd, '$') != NULL) {
1160 1.475 rillig char *xcmd = Var_Subst(cmd, job->node, VARE_EVAL);
1161 1.439 rillig debug_printf("\t=> %s\n", xcmd);
1162 1.439 rillig free(xcmd);
1163 1.437 rillig }
1164 1.437 rillig }
1165 1.430 christos }
1166 1.430 christos
1167 1.430 christos static void
1168 1.379 rillig JobFinishDoneExitedError(Job *job, int *inout_status)
1169 1.378 rillig {
1170 1.379 rillig SwitchOutputTo(job->node);
1171 1.378 rillig #ifdef USE_META
1172 1.379 rillig if (useMeta) {
1173 1.379 rillig meta_job_error(job, job->node,
1174 1.379 rillig job->ignerr, WEXITSTATUS(*inout_status));
1175 1.379 rillig }
1176 1.378 rillig #endif
1177 1.379 rillig if (!shouldDieQuietly(job->node, -1)) {
1178 1.430 christos DebugFailedJob(job);
1179 1.379 rillig (void)printf("*** [%s] Error code %d%s\n",
1180 1.379 rillig job->node->name, WEXITSTATUS(*inout_status),
1181 1.379 rillig job->ignerr ? " (ignored)" : "");
1182 1.379 rillig }
1183 1.379 rillig
1184 1.379 rillig if (job->ignerr)
1185 1.379 rillig *inout_status = 0;
1186 1.379 rillig else {
1187 1.379 rillig if (deleteOnError)
1188 1.379 rillig JobDeleteTarget(job->node);
1189 1.449 rillig PrintOnError(job->node, "\n");
1190 1.379 rillig }
1191 1.379 rillig }
1192 1.379 rillig
1193 1.379 rillig static void
1194 1.379 rillig JobFinishDoneExited(Job *job, int *inout_status)
1195 1.379 rillig {
1196 1.505 rillig DEBUG2(JOB, "Target %s, pid %d exited\n",
1197 1.505 rillig job->node->name, job->pid);
1198 1.379 rillig
1199 1.379 rillig if (WEXITSTATUS(*inout_status) != 0)
1200 1.379 rillig JobFinishDoneExitedError(job, inout_status);
1201 1.379 rillig else if (DEBUG(JOB)) {
1202 1.378 rillig SwitchOutputTo(job->node);
1203 1.505 rillig (void)printf("Target %s, pid %d exited successfully\n",
1204 1.505 rillig job->node->name, job->pid);
1205 1.378 rillig }
1206 1.379 rillig }
1207 1.379 rillig
1208 1.379 rillig static void
1209 1.379 rillig JobFinishDoneSignaled(Job *job, int status)
1210 1.379 rillig {
1211 1.379 rillig SwitchOutputTo(job->node);
1212 1.430 christos DebugFailedJob(job);
1213 1.379 rillig (void)printf("*** [%s] Signal %d\n", job->node->name, WTERMSIG(status));
1214 1.379 rillig if (deleteOnError)
1215 1.379 rillig JobDeleteTarget(job->node);
1216 1.379 rillig }
1217 1.379 rillig
1218 1.379 rillig static void
1219 1.379 rillig JobFinishDone(Job *job, int *inout_status)
1220 1.379 rillig {
1221 1.379 rillig if (WIFEXITED(*inout_status))
1222 1.379 rillig JobFinishDoneExited(job, inout_status);
1223 1.379 rillig else
1224 1.379 rillig JobFinishDoneSignaled(job, *inout_status);
1225 1.378 rillig
1226 1.378 rillig (void)fflush(stdout);
1227 1.378 rillig }
1228 1.378 rillig
1229 1.341 rillig /*
1230 1.341 rillig * Do final processing for the given job including updating parent nodes and
1231 1.320 rillig * starting new jobs as available/necessary.
1232 1.320 rillig *
1233 1.320 rillig * Deferred commands for the job are placed on the .END node.
1234 1.320 rillig *
1235 1.331 rillig * If there was a serious error (job_errors != 0; not an ignored one), no more
1236 1.320 rillig * jobs will be started.
1237 1.1 cgd *
1238 1.72 wiz * Input:
1239 1.72 wiz * job job to finish
1240 1.72 wiz * status sub-why job went away
1241 1.1 cgd */
1242 1.5 cgd static void
1243 1.116 dsl JobFinish(Job *job, int status)
1244 1.1 cgd {
1245 1.421 rillig bool done, return_job_token;
1246 1.1 cgd
1247 1.505 rillig DEBUG3(JOB, "JobFinish: target %s, pid %d, status %#x\n",
1248 1.505 rillig job->node->name, job->pid, status);
1249 1.116 dsl
1250 1.341 rillig if ((WIFEXITED(status) &&
1251 1.356 rillig ((WEXITSTATUS(status) != 0 && !job->ignerr))) ||
1252 1.341 rillig WIFSIGNALED(status)) {
1253 1.377 rillig /* Finished because of an error. */
1254 1.377 rillig
1255 1.341 rillig JobClosePipes(job);
1256 1.341 rillig if (job->cmdFILE != NULL && job->cmdFILE != stdout) {
1257 1.436 sjg if (fclose(job->cmdFILE) != 0)
1258 1.505 rillig Punt("Cannot write shell script for \"%s\": %s",
1259 1.436 sjg job->node->name, strerror(errno));
1260 1.341 rillig job->cmdFILE = NULL;
1261 1.341 rillig }
1262 1.421 rillig done = true;
1263 1.377 rillig
1264 1.341 rillig } else if (WIFEXITED(status)) {
1265 1.341 rillig /*
1266 1.341 rillig * Deal with ignored errors in -B mode. We need to print a
1267 1.341 rillig * message telling of the ignored error as well as to run
1268 1.341 rillig * the next command.
1269 1.341 rillig */
1270 1.341 rillig done = WEXITSTATUS(status) != 0;
1271 1.377 rillig
1272 1.341 rillig JobClosePipes(job);
1273 1.377 rillig
1274 1.341 rillig } else {
1275 1.377 rillig /* No need to close things down or anything. */
1276 1.421 rillig done = false;
1277 1.1 cgd }
1278 1.16 christos
1279 1.378 rillig if (done)
1280 1.378 rillig JobFinishDone(job, &status);
1281 1.16 christos
1282 1.156 sjg #ifdef USE_META
1283 1.341 rillig if (useMeta) {
1284 1.341 rillig int meta_status = meta_job_finish(job);
1285 1.341 rillig if (meta_status != 0 && status == 0)
1286 1.341 rillig status = meta_status;
1287 1.341 rillig }
1288 1.156 sjg #endif
1289 1.201 rillig
1290 1.421 rillig return_job_token = false;
1291 1.102 dsl
1292 1.341 rillig Trace_Log(JOBEND, job);
1293 1.356 rillig if (!job->special) {
1294 1.341 rillig if (status != 0 ||
1295 1.341 rillig (aborting == ABORT_ERROR) || aborting == ABORT_INTERRUPT)
1296 1.421 rillig return_job_token = true;
1297 1.341 rillig }
1298 1.1 cgd
1299 1.341 rillig if (aborting != ABORT_ERROR && aborting != ABORT_INTERRUPT &&
1300 1.500 rillig status == 0) {
1301 1.341 rillig /*
1302 1.341 rillig * As long as we aren't aborting and the job didn't return a
1303 1.341 rillig * non-zero status that we shouldn't ignore, we call
1304 1.341 rillig * Make_Update to update the parents.
1305 1.341 rillig */
1306 1.341 rillig JobSaveCommands(job);
1307 1.341 rillig job->node->made = MADE;
1308 1.356 rillig if (!job->special)
1309 1.421 rillig return_job_token = true;
1310 1.341 rillig Make_Update(job->node);
1311 1.341 rillig job->status = JOB_ST_FREE;
1312 1.341 rillig } else if (status != 0) {
1313 1.341 rillig job_errors++;
1314 1.341 rillig job->status = JOB_ST_FREE;
1315 1.341 rillig }
1316 1.341 rillig
1317 1.341 rillig if (job_errors > 0 && !opts.keepgoing && aborting != ABORT_INTERRUPT) {
1318 1.341 rillig /* Prevent more jobs from getting started. */
1319 1.341 rillig aborting = ABORT_ERROR;
1320 1.341 rillig }
1321 1.1 cgd
1322 1.341 rillig if (return_job_token)
1323 1.495 rillig TokenPool_Return();
1324 1.16 christos
1325 1.487 rillig if (aborting == ABORT_ERROR && jobTokensRunning == 0) {
1326 1.494 rillig if (shouldDieQuietly(NULL, -1))
1327 1.487 rillig exit(2);
1328 1.487 rillig Fatal("%d error%s", job_errors, job_errors == 1 ? "" : "s");
1329 1.487 rillig }
1330 1.321 rillig }
1331 1.321 rillig
1332 1.321 rillig static void
1333 1.321 rillig TouchRegular(GNode *gn)
1334 1.321 rillig {
1335 1.341 rillig const char *file = GNode_Path(gn);
1336 1.422 rillig struct utimbuf times;
1337 1.341 rillig int fd;
1338 1.341 rillig char c;
1339 1.341 rillig
1340 1.422 rillig times.actime = now;
1341 1.422 rillig times.modtime = now;
1342 1.341 rillig if (utime(file, ×) >= 0)
1343 1.341 rillig return;
1344 1.321 rillig
1345 1.341 rillig fd = open(file, O_RDWR | O_CREAT, 0666);
1346 1.341 rillig if (fd < 0) {
1347 1.341 rillig (void)fprintf(stderr, "*** couldn't touch %s: %s\n",
1348 1.341 rillig file, strerror(errno));
1349 1.341 rillig (void)fflush(stderr);
1350 1.341 rillig return; /* XXX: What about propagating the error? */
1351 1.341 rillig }
1352 1.341 rillig
1353 1.443 rillig /*
1354 1.443 rillig * Last resort: update the file's time stamps in the traditional way.
1355 1.341 rillig * XXX: This doesn't work for empty files, which are sometimes used
1356 1.443 rillig * as marker files.
1357 1.443 rillig */
1358 1.341 rillig if (read(fd, &c, 1) == 1) {
1359 1.341 rillig (void)lseek(fd, 0, SEEK_SET);
1360 1.341 rillig while (write(fd, &c, 1) == -1 && errno == EAGAIN)
1361 1.341 rillig continue;
1362 1.341 rillig }
1363 1.341 rillig (void)close(fd); /* XXX: What about propagating the error? */
1364 1.1 cgd }
1365 1.1 cgd
1366 1.391 rillig /*
1367 1.484 rillig * Touch the given target. Called by Job_Make when the -t flag was given.
1368 1.1 cgd *
1369 1.243 rillig * The modification date of the file is changed.
1370 1.391 rillig * If the file did not exist, it is created.
1371 1.391 rillig */
1372 1.1 cgd void
1373 1.421 rillig Job_Touch(GNode *gn, bool echo)
1374 1.1 cgd {
1375 1.341 rillig if (gn->type &
1376 1.341 rillig (OP_JOIN | OP_USE | OP_USEBEFORE | OP_EXEC | OP_OPTIONAL |
1377 1.341 rillig OP_SPECIAL | OP_PHONY)) {
1378 1.341 rillig /*
1379 1.341 rillig * These are "virtual" targets and should not really be
1380 1.341 rillig * created.
1381 1.341 rillig */
1382 1.341 rillig return;
1383 1.341 rillig }
1384 1.16 christos
1385 1.357 rillig if (echo || !GNode_ShouldExecute(gn)) {
1386 1.341 rillig (void)fprintf(stdout, "touch %s\n", gn->name);
1387 1.341 rillig (void)fflush(stdout);
1388 1.341 rillig }
1389 1.1 cgd
1390 1.341 rillig if (!GNode_ShouldExecute(gn))
1391 1.341 rillig return;
1392 1.1 cgd
1393 1.380 rillig if (gn->type & OP_ARCHV)
1394 1.341 rillig Arch_Touch(gn);
1395 1.380 rillig else if (gn->type & OP_LIB)
1396 1.341 rillig Arch_TouchLib(gn);
1397 1.380 rillig else
1398 1.380 rillig TouchRegular(gn);
1399 1.1 cgd }
1400 1.1 cgd
1401 1.391 rillig /*
1402 1.391 rillig * Make sure the given node has all the commands it needs.
1403 1.243 rillig *
1404 1.243 rillig * The node will have commands from the .DEFAULT rule added to it if it
1405 1.243 rillig * needs them.
1406 1.1 cgd *
1407 1.72 wiz * Input:
1408 1.72 wiz * gn The target whose commands need verifying
1409 1.72 wiz * abortProc Function to abort with message
1410 1.72 wiz *
1411 1.1 cgd * Results:
1412 1.500 rillig * true if the commands are ok.
1413 1.1 cgd */
1414 1.421 rillig bool
1415 1.77 christos Job_CheckCommands(GNode *gn, void (*abortProc)(const char *, ...))
1416 1.1 cgd {
1417 1.341 rillig if (GNode_IsTarget(gn))
1418 1.421 rillig return true;
1419 1.341 rillig if (!Lst_IsEmpty(&gn->commands))
1420 1.421 rillig return true;
1421 1.341 rillig if ((gn->type & OP_LIB) && !Lst_IsEmpty(&gn->children))
1422 1.421 rillig return true;
1423 1.280 rillig
1424 1.281 rillig /*
1425 1.341 rillig * No commands. Look for .DEFAULT rule from which we might infer
1426 1.341 rillig * commands.
1427 1.281 rillig */
1428 1.341 rillig if (defaultNode != NULL && !Lst_IsEmpty(&defaultNode->commands) &&
1429 1.341 rillig !(gn->type & OP_SPECIAL)) {
1430 1.341 rillig /*
1431 1.341 rillig * The traditional Make only looks for a .DEFAULT if the node
1432 1.341 rillig * was never the target of an operator, so that's what we do
1433 1.341 rillig * too.
1434 1.341 rillig *
1435 1.341 rillig * The .DEFAULT node acts like a transformation rule, in that
1436 1.341 rillig * gn also inherits any attributes or sources attached to
1437 1.341 rillig * .DEFAULT itself.
1438 1.341 rillig */
1439 1.341 rillig Make_HandleUse(defaultNode, gn);
1440 1.417 rillig Var_Set(gn, IMPSRC, GNode_VarTarget(gn));
1441 1.421 rillig return true;
1442 1.341 rillig }
1443 1.341 rillig
1444 1.421 rillig Dir_UpdateMTime(gn, false);
1445 1.341 rillig if (gn->mtime != 0 || (gn->type & OP_SPECIAL))
1446 1.421 rillig return true;
1447 1.281 rillig
1448 1.341 rillig /*
1449 1.341 rillig * The node wasn't the target of an operator. We have no .DEFAULT
1450 1.341 rillig * rule to go on and the target doesn't already exist. There's
1451 1.341 rillig * nothing more we can do for this branch. If the -k flag wasn't
1452 1.341 rillig * given, we stop in our tracks, otherwise we just don't update
1453 1.341 rillig * this node's parents so they never get examined.
1454 1.341 rillig */
1455 1.282 rillig
1456 1.440 rillig if (gn->flags.fromDepend) {
1457 1.341 rillig if (!Job_RunTarget(".STALE", gn->fname))
1458 1.341 rillig fprintf(stdout,
1459 1.490 rillig "%s: %s:%u: ignoring stale %s for %s\n",
1460 1.341 rillig progname, gn->fname, gn->lineno, makeDependfile,
1461 1.341 rillig gn->name);
1462 1.421 rillig return true;
1463 1.341 rillig }
1464 1.282 rillig
1465 1.341 rillig if (gn->type & OP_OPTIONAL) {
1466 1.341 rillig (void)fprintf(stdout, "%s: don't know how to make %s (%s)\n",
1467 1.341 rillig progname, gn->name, "ignored");
1468 1.341 rillig (void)fflush(stdout);
1469 1.421 rillig return true;
1470 1.341 rillig }
1471 1.282 rillig
1472 1.341 rillig if (opts.keepgoing) {
1473 1.341 rillig (void)fprintf(stdout, "%s: don't know how to make %s (%s)\n",
1474 1.341 rillig progname, gn->name, "continuing");
1475 1.341 rillig (void)fflush(stdout);
1476 1.421 rillig return false;
1477 1.341 rillig }
1478 1.282 rillig
1479 1.464 sjg abortProc("don't know how to make %s. Stop", gn->name);
1480 1.421 rillig return false;
1481 1.1 cgd }
1482 1.1 cgd
1483 1.391 rillig /*
1484 1.391 rillig * Execute the shell for the given job.
1485 1.1 cgd *
1486 1.391 rillig * See Job_CatchOutput for handling the output of the shell.
1487 1.391 rillig */
1488 1.1 cgd static void
1489 1.72 wiz JobExec(Job *job, char **argv)
1490 1.1 cgd {
1491 1.341 rillig int cpid; /* ID of new child */
1492 1.341 rillig sigset_t mask;
1493 1.16 christos
1494 1.341 rillig if (DEBUG(JOB)) {
1495 1.341 rillig int i;
1496 1.16 christos
1497 1.341 rillig debug_printf("Running %s\n", job->node->name);
1498 1.505 rillig debug_printf("\tCommand:");
1499 1.341 rillig for (i = 0; argv[i] != NULL; i++) {
1500 1.505 rillig debug_printf(" %s", argv[i]);
1501 1.341 rillig }
1502 1.341 rillig debug_printf("\n");
1503 1.1 cgd }
1504 1.16 christos
1505 1.341 rillig /*
1506 1.461 rillig * Some jobs produce no output, and it's disconcerting to have
1507 1.341 rillig * no feedback of their running (since they produce no output, the
1508 1.341 rillig * banner with their name in it never appears). This is an attempt to
1509 1.341 rillig * provide that feedback, even if nothing follows it.
1510 1.341 rillig */
1511 1.357 rillig if (job->echo)
1512 1.341 rillig SwitchOutputTo(job->node);
1513 1.16 christos
1514 1.500 rillig /* No interruptions until this job is in the jobs table. */
1515 1.500 rillig JobsTable_Lock(&mask);
1516 1.69 pk
1517 1.341 rillig /* Pre-emptively mark job running, pid still zero though */
1518 1.341 rillig job->status = JOB_ST_RUNNING;
1519 1.116 dsl
1520 1.466 sjg Var_ReexportVars(job->node);
1521 1.390 rillig
1522 1.485 rillig cpid = FORK_FUNCTION();
1523 1.341 rillig if (cpid == -1)
1524 1.485 rillig Punt("Cannot fork: %s", strerror(errno));
1525 1.341 rillig
1526 1.341 rillig if (cpid == 0) {
1527 1.341 rillig /* Child */
1528 1.341 rillig sigset_t tmask;
1529 1.1 cgd
1530 1.155 sjg #ifdef USE_META
1531 1.448 rillig if (useMeta)
1532 1.341 rillig meta_job_child(job);
1533 1.155 sjg #endif
1534 1.341 rillig /*
1535 1.341 rillig * Reset all signal handlers; this is necessary because we
1536 1.341 rillig * also need to unblock signals before we exec(2).
1537 1.341 rillig */
1538 1.341 rillig JobSigReset();
1539 1.69 pk
1540 1.341 rillig sigemptyset(&tmask);
1541 1.500 rillig JobsTable_Unlock(&tmask);
1542 1.69 pk
1543 1.341 rillig /*
1544 1.341 rillig * Must duplicate the input stream down to the child's input
1545 1.341 rillig * and reset it to the beginning (again). Since the stream
1546 1.341 rillig * was marked close-on-exec, we must clear that bit in the
1547 1.341 rillig * new input.
1548 1.341 rillig */
1549 1.474 rillig if (dup2(fileno(job->cmdFILE), STDIN_FILENO) == -1)
1550 1.341 rillig execDie("dup2", "job->cmdFILE");
1551 1.474 rillig if (fcntl(STDIN_FILENO, F_SETFD, 0) == -1)
1552 1.341 rillig execDie("fcntl clear close-on-exec", "stdin");
1553 1.474 rillig if (lseek(STDIN_FILENO, 0, SEEK_SET) == -1)
1554 1.341 rillig execDie("lseek to 0", "stdin");
1555 1.341 rillig
1556 1.341 rillig if (job->node->type & (OP_MAKE | OP_SUBMAKE)) {
1557 1.461 rillig /* Pass job token pipe to submakes. */
1558 1.495 rillig if (fcntl(tokenPoolJob.inPipe, F_SETFD, 0) == -1)
1559 1.341 rillig execDie("clear close-on-exec",
1560 1.495 rillig "tokenPoolJob.inPipe");
1561 1.495 rillig if (fcntl(tokenPoolJob.outPipe, F_SETFD, 0) == -1)
1562 1.341 rillig execDie("clear close-on-exec",
1563 1.495 rillig "tokenPoolJob.outPipe");
1564 1.341 rillig }
1565 1.16 christos
1566 1.341 rillig /*
1567 1.341 rillig * Set up the child's output to be routed through the pipe
1568 1.341 rillig * we've created for it.
1569 1.341 rillig */
1570 1.474 rillig if (dup2(job->outPipe, STDOUT_FILENO) == -1)
1571 1.341 rillig execDie("dup2", "job->outPipe");
1572 1.201 rillig
1573 1.341 rillig /*
1574 1.341 rillig * The output channels are marked close on exec. This bit
1575 1.474 rillig * was duplicated by dup2 (on some systems), so we have
1576 1.341 rillig * to clear it before routing the shell's error output to
1577 1.341 rillig * the same place as its standard output.
1578 1.341 rillig */
1579 1.474 rillig if (fcntl(STDOUT_FILENO, F_SETFD, 0) == -1)
1580 1.341 rillig execDie("clear close-on-exec", "stdout");
1581 1.474 rillig if (dup2(STDOUT_FILENO, STDERR_FILENO) == -1)
1582 1.341 rillig execDie("dup2", "1, 2");
1583 1.264 rillig
1584 1.341 rillig /*
1585 1.341 rillig * We want to switch the child into a different process
1586 1.341 rillig * family so we can kill it and all its descendants in
1587 1.341 rillig * one fell swoop, by killing its process family, but not
1588 1.341 rillig * commit suicide.
1589 1.341 rillig */
1590 1.170 christos #if defined(MAKE_NATIVE) || defined(HAVE_SETPGID)
1591 1.412 rillig # if defined(SYSV)
1592 1.341 rillig /* XXX: dsl - I'm sure this should be setpgrp()... */
1593 1.341 rillig (void)setsid();
1594 1.412 rillig # else
1595 1.341 rillig (void)setpgid(0, getpid());
1596 1.412 rillig # endif
1597 1.170 christos #endif
1598 1.1 cgd
1599 1.341 rillig (void)execv(shellPath, argv);
1600 1.341 rillig execDie("exec", shellPath);
1601 1.341 rillig }
1602 1.116 dsl
1603 1.341 rillig /* Parent, continuing after the child exec */
1604 1.341 rillig job->pid = cpid;
1605 1.1 cgd
1606 1.341 rillig Trace_Log(JOBSTART, job);
1607 1.40 sommerfe
1608 1.196 riastrad #ifdef USE_META
1609 1.448 rillig if (useMeta)
1610 1.341 rillig meta_job_parent(job, cpid);
1611 1.196 riastrad #endif
1612 1.196 riastrad
1613 1.341 rillig /*
1614 1.341 rillig * Set the current position in the buffer to the beginning
1615 1.341 rillig * and mark another stream to watch in the outputs mask
1616 1.341 rillig */
1617 1.341 rillig job->curPos = 0;
1618 1.16 christos
1619 1.341 rillig watchfd(job);
1620 1.1 cgd
1621 1.341 rillig if (job->cmdFILE != NULL && job->cmdFILE != stdout) {
1622 1.436 sjg if (fclose(job->cmdFILE) != 0)
1623 1.505 rillig Punt("Cannot write shell script for \"%s\": %s",
1624 1.436 sjg job->node->name, strerror(errno));
1625 1.341 rillig job->cmdFILE = NULL;
1626 1.341 rillig }
1627 1.1 cgd
1628 1.409 rillig /* Now that the job is actually running, add it to the table. */
1629 1.341 rillig if (DEBUG(JOB)) {
1630 1.505 rillig debug_printf(
1631 1.505 rillig "JobExec: target %s, pid %d added to jobs table\n",
1632 1.341 rillig job->node->name, job->pid);
1633 1.428 rillig DumpJobs("job started");
1634 1.341 rillig }
1635 1.500 rillig JobsTable_Unlock(&mask);
1636 1.1 cgd }
1637 1.1 cgd
1638 1.228 rillig /* Create the argv needed to execute the shell for a given job. */
1639 1.1 cgd static void
1640 1.72 wiz JobMakeArgv(Job *job, char **argv)
1641 1.1 cgd {
1642 1.341 rillig int argc;
1643 1.341 rillig static char args[10]; /* For merged arguments */
1644 1.16 christos
1645 1.341 rillig argv[0] = UNCONST(shellName);
1646 1.341 rillig argc = 1;
1647 1.1 cgd
1648 1.375 rillig if ((shell->errFlag != NULL && shell->errFlag[0] != '-') ||
1649 1.366 rillig (shell->echoFlag != NULL && shell->echoFlag[0] != '-')) {
1650 1.341 rillig /*
1651 1.341 rillig * At least one of the flags doesn't have a minus before it,
1652 1.341 rillig * so merge them together. Have to do this because the Bourne
1653 1.341 rillig * shell thinks its second argument is a file to source.
1654 1.341 rillig * Grrrr. Note the ten-character limitation on the combined
1655 1.341 rillig * arguments.
1656 1.341 rillig *
1657 1.341 rillig * TODO: Research until when the above comments were
1658 1.341 rillig * practically relevant.
1659 1.341 rillig */
1660 1.341 rillig (void)snprintf(args, sizeof args, "-%s%s",
1661 1.356 rillig (job->ignerr ? "" :
1662 1.375 rillig (shell->errFlag != NULL ? shell->errFlag : "")),
1663 1.357 rillig (!job->echo ? "" :
1664 1.366 rillig (shell->echoFlag != NULL ? shell->echoFlag : "")));
1665 1.341 rillig
1666 1.397 rillig if (args[1] != '\0') {
1667 1.341 rillig argv[argc] = args;
1668 1.341 rillig argc++;
1669 1.341 rillig }
1670 1.341 rillig } else {
1671 1.396 rillig if (!job->ignerr && shell->errFlag != NULL) {
1672 1.375 rillig argv[argc] = UNCONST(shell->errFlag);
1673 1.341 rillig argc++;
1674 1.341 rillig }
1675 1.396 rillig if (job->echo && shell->echoFlag != NULL) {
1676 1.366 rillig argv[argc] = UNCONST(shell->echoFlag);
1677 1.341 rillig argc++;
1678 1.341 rillig }
1679 1.1 cgd }
1680 1.341 rillig argv[argc] = NULL;
1681 1.1 cgd }
1682 1.1 cgd
1683 1.361 rillig static void
1684 1.434 rillig JobWriteShellCommands(Job *job, GNode *gn, bool *out_run)
1685 1.361 rillig {
1686 1.361 rillig /*
1687 1.361 rillig * tfile is the name of a file into which all shell commands
1688 1.361 rillig * are put. It is removed before the child shell is executed,
1689 1.361 rillig * unless DEBUG(SCRIPT) is set.
1690 1.361 rillig */
1691 1.419 sjg char tfile[MAXPATHLEN];
1692 1.361 rillig int tfd; /* File descriptor to the temp file */
1693 1.361 rillig
1694 1.496 rillig tfd = Job_TempFile(NULL, tfile, sizeof tfile);
1695 1.361 rillig
1696 1.361 rillig job->cmdFILE = fdopen(tfd, "w+");
1697 1.361 rillig if (job->cmdFILE == NULL)
1698 1.361 rillig Punt("Could not fdopen %s", tfile);
1699 1.361 rillig
1700 1.361 rillig (void)fcntl(fileno(job->cmdFILE), F_SETFD, FD_CLOEXEC);
1701 1.361 rillig
1702 1.361 rillig #ifdef USE_META
1703 1.361 rillig if (useMeta) {
1704 1.361 rillig meta_job_start(job, gn);
1705 1.443 rillig if (gn->type & OP_SILENT) /* might have changed */
1706 1.421 rillig job->echo = false;
1707 1.361 rillig }
1708 1.361 rillig #endif
1709 1.362 rillig
1710 1.429 rillig *out_run = JobWriteCommands(job);
1711 1.361 rillig }
1712 1.361 rillig
1713 1.484 rillig void
1714 1.484 rillig Job_Make(GNode *gn)
1715 1.1 cgd {
1716 1.341 rillig Job *job; /* new job descriptor */
1717 1.341 rillig char *argv[10]; /* Argument vector to shell */
1718 1.421 rillig bool cmdsOK; /* true if the nodes commands were all right */
1719 1.421 rillig bool run;
1720 1.341 rillig
1721 1.341 rillig for (job = job_table; job < job_table_end; job++) {
1722 1.341 rillig if (job->status == JOB_ST_FREE)
1723 1.341 rillig break;
1724 1.341 rillig }
1725 1.341 rillig if (job >= job_table_end)
1726 1.484 rillig Punt("Job_Make no job slots vacant");
1727 1.122 dsl
1728 1.341 rillig memset(job, 0, sizeof *job);
1729 1.341 rillig job->node = gn;
1730 1.341 rillig job->tailCmds = NULL;
1731 1.341 rillig job->status = JOB_ST_SET_UP;
1732 1.1 cgd
1733 1.484 rillig job->special = (gn->type & OP_SPECIAL) != OP_NONE;
1734 1.369 rillig job->ignerr = opts.ignoreErrors || gn->type & OP_IGNORE;
1735 1.445 rillig job->echo = !(opts.silent || gn->type & OP_SILENT);
1736 1.16 christos
1737 1.1 cgd /*
1738 1.341 rillig * Check the commands now so any attributes from .DEFAULT have a
1739 1.341 rillig * chance to migrate to the node.
1740 1.34 christos */
1741 1.341 rillig cmdsOK = Job_CheckCommands(gn, Error);
1742 1.341 rillig
1743 1.341 rillig job->inPollfd = NULL;
1744 1.403 rillig
1745 1.399 rillig if (Lst_IsEmpty(&gn->commands)) {
1746 1.399 rillig job->cmdFILE = stdout;
1747 1.421 rillig run = false;
1748 1.435 rillig
1749 1.435 rillig /*
1750 1.435 rillig * We're serious here, but if the commands were bogus, we're
1751 1.435 rillig * also dead...
1752 1.435 rillig */
1753 1.435 rillig if (!cmdsOK) {
1754 1.449 rillig PrintOnError(gn, "\n"); /* provide some clue */
1755 1.435 rillig DieHorribly();
1756 1.435 rillig }
1757 1.399 rillig } else if (((gn->type & OP_MAKE) && !opts.noRecursiveExecute) ||
1758 1.445 rillig (!opts.noExecute && !opts.touch)) {
1759 1.401 rillig /*
1760 1.403 rillig * The above condition looks very similar to
1761 1.418 rillig * GNode_ShouldExecute but is subtly different. It prevents
1762 1.418 rillig * that .MAKE targets are touched since these are usually
1763 1.418 rillig * virtual targets.
1764 1.401 rillig */
1765 1.402 rillig
1766 1.481 rillig int parseErrorsBefore;
1767 1.481 rillig
1768 1.434 rillig /*
1769 1.434 rillig * We're serious here, but if the commands were bogus, we're
1770 1.434 rillig * also dead...
1771 1.434 rillig */
1772 1.434 rillig if (!cmdsOK) {
1773 1.449 rillig PrintOnError(gn, "\n"); /* provide some clue */
1774 1.434 rillig DieHorribly();
1775 1.434 rillig }
1776 1.434 rillig
1777 1.481 rillig parseErrorsBefore = parseErrors;
1778 1.434 rillig JobWriteShellCommands(job, gn, &run);
1779 1.481 rillig if (parseErrors != parseErrorsBefore)
1780 1.481 rillig run = false;
1781 1.401 rillig (void)fflush(job->cmdFILE);
1782 1.341 rillig } else if (!GNode_ShouldExecute(gn)) {
1783 1.341 rillig /*
1784 1.429 rillig * Just write all the commands to stdout in one fell swoop.
1785 1.418 rillig * This still sets up job->tailCmds correctly.
1786 1.341 rillig */
1787 1.341 rillig SwitchOutputTo(gn);
1788 1.341 rillig job->cmdFILE = stdout;
1789 1.341 rillig if (cmdsOK)
1790 1.429 rillig JobWriteCommands(job);
1791 1.421 rillig run = false;
1792 1.401 rillig (void)fflush(job->cmdFILE);
1793 1.341 rillig } else {
1794 1.357 rillig Job_Touch(gn, job->echo);
1795 1.421 rillig run = false;
1796 1.341 rillig }
1797 1.341 rillig
1798 1.341 rillig /* If we're not supposed to execute a shell, don't. */
1799 1.353 rillig if (!run) {
1800 1.356 rillig if (!job->special)
1801 1.495 rillig TokenPool_Return();
1802 1.341 rillig /* Unlink and close the command file if we opened one */
1803 1.341 rillig if (job->cmdFILE != NULL && job->cmdFILE != stdout) {
1804 1.341 rillig (void)fclose(job->cmdFILE);
1805 1.341 rillig job->cmdFILE = NULL;
1806 1.341 rillig }
1807 1.1 cgd
1808 1.341 rillig /*
1809 1.341 rillig * We only want to work our way up the graph if we aren't
1810 1.341 rillig * here because the commands for the job were no good.
1811 1.341 rillig */
1812 1.341 rillig if (cmdsOK && aborting == ABORT_NONE) {
1813 1.341 rillig JobSaveCommands(job);
1814 1.341 rillig job->node->made = MADE;
1815 1.341 rillig Make_Update(job->node);
1816 1.341 rillig }
1817 1.341 rillig job->status = JOB_ST_FREE;
1818 1.484 rillig return;
1819 1.1 cgd }
1820 1.1 cgd
1821 1.1 cgd /*
1822 1.341 rillig * Set up the control arguments to the shell. This is based on the
1823 1.341 rillig * flags set earlier for this job.
1824 1.1 cgd */
1825 1.341 rillig JobMakeArgv(job, argv);
1826 1.1 cgd
1827 1.341 rillig /* Create the pipe by which we'll get the shell's output. */
1828 1.341 rillig JobCreatePipe(job, 3);
1829 1.1 cgd
1830 1.341 rillig JobExec(job, argv);
1831 1.1 cgd }
1832 1.1 cgd
1833 1.341 rillig /*
1834 1.426 rillig * If the shell has an output filter (which only csh and ksh have by default),
1835 1.426 rillig * print the output of the child process, skipping the noPrint text of the
1836 1.426 rillig * shell.
1837 1.426 rillig *
1838 1.426 rillig * Return the part of the output that the calling function needs to output by
1839 1.426 rillig * itself.
1840 1.341 rillig */
1841 1.16 christos static char *
1842 1.470 rillig PrintFilteredOutput(char *p, const char *endp) /* XXX: p should be const */
1843 1.12 christos {
1844 1.460 rillig char *ep; /* XXX: should be const */
1845 1.12 christos
1846 1.355 rillig if (shell->noPrint == NULL || shell->noPrint[0] == '\0')
1847 1.460 rillig return p;
1848 1.324 rillig
1849 1.392 rillig /*
1850 1.392 rillig * XXX: What happens if shell->noPrint occurs on the boundary of
1851 1.392 rillig * the buffer? To work correctly in all cases, this should rather
1852 1.392 rillig * be a proper stream filter instead of doing string matching on
1853 1.392 rillig * selected chunks of the output.
1854 1.392 rillig */
1855 1.460 rillig while ((ep = strstr(p, shell->noPrint)) != NULL) {
1856 1.460 rillig if (ep != p) {
1857 1.460 rillig *ep = '\0'; /* XXX: avoid writing to the buffer */
1858 1.341 rillig /*
1859 1.341 rillig * The only way there wouldn't be a newline after
1860 1.341 rillig * this line is if it were the last in the buffer.
1861 1.392 rillig * however, since the noPrint output comes after it,
1862 1.341 rillig * there must be a newline, so we don't print one.
1863 1.341 rillig */
1864 1.392 rillig /* XXX: What about null bytes in the output? */
1865 1.460 rillig (void)fprintf(stdout, "%s", p);
1866 1.341 rillig (void)fflush(stdout);
1867 1.341 rillig }
1868 1.460 rillig p = ep + shell->noPrintLen;
1869 1.460 rillig if (p == endp)
1870 1.392 rillig break;
1871 1.460 rillig p++; /* skip over the (XXX: assumed) newline */
1872 1.460 rillig pp_skip_whitespace(&p);
1873 1.12 christos }
1874 1.460 rillig return p;
1875 1.12 christos }
1876 1.12 christos
1877 1.325 rillig /*
1878 1.325 rillig * This function is called whenever there is something to read on the pipe.
1879 1.325 rillig * We collect more output from the given job and store it in the job's
1880 1.325 rillig * outBuf. If this makes up a line, we print it tagged by the job's
1881 1.325 rillig * identifier, as necessary.
1882 1.325 rillig *
1883 1.325 rillig * In the output of the shell, the 'noPrint' lines are removed. If the
1884 1.325 rillig * command is not alone on the line (the character after it is not \0 or
1885 1.325 rillig * \n), we do print whatever follows it.
1886 1.1 cgd *
1887 1.72 wiz * Input:
1888 1.72 wiz * job the job whose output needs printing
1889 1.421 rillig * finish true if this is the last time we'll be called
1890 1.72 wiz * for this job
1891 1.1 cgd */
1892 1.285 rillig static void
1893 1.424 rillig CollectOutput(Job *job, bool finish)
1894 1.1 cgd {
1895 1.421 rillig bool gotNL; /* true if got a newline */
1896 1.421 rillig bool fbuf; /* true if our buffer filled up */
1897 1.341 rillig size_t nr; /* number of bytes read */
1898 1.341 rillig size_t i; /* auxiliary index into outBuf */
1899 1.341 rillig size_t max; /* limit for i (end of current data) */
1900 1.341 rillig ssize_t nRead; /* (Temporary) number of bytes read */
1901 1.1 cgd
1902 1.341 rillig /* Read as many bytes as will fit in the buffer. */
1903 1.325 rillig again:
1904 1.421 rillig gotNL = false;
1905 1.421 rillig fbuf = false;
1906 1.16 christos
1907 1.341 rillig nRead = read(job->inPipe, &job->outBuf[job->curPos],
1908 1.341 rillig JOB_BUFSIZE - job->curPos);
1909 1.341 rillig if (nRead < 0) {
1910 1.341 rillig if (errno == EAGAIN)
1911 1.341 rillig return;
1912 1.455 rillig if (DEBUG(JOB))
1913 1.424 rillig perror("CollectOutput(piperead)");
1914 1.341 rillig nr = 0;
1915 1.455 rillig } else
1916 1.341 rillig nr = (size_t)nRead;
1917 1.455 rillig
1918 1.455 rillig if (nr == 0)
1919 1.455 rillig finish = false; /* stop looping */
1920 1.1 cgd
1921 1.341 rillig /*
1922 1.341 rillig * If we hit the end-of-file (the job is dead), we must flush its
1923 1.341 rillig * remaining output, so pretend we read a newline if there's any
1924 1.341 rillig * output remaining in the buffer.
1925 1.341 rillig */
1926 1.341 rillig if (nr == 0 && job->curPos != 0) {
1927 1.341 rillig job->outBuf[job->curPos] = '\n';
1928 1.341 rillig nr = 1;
1929 1.341 rillig }
1930 1.16 christos
1931 1.341 rillig max = job->curPos + nr;
1932 1.454 rillig for (i = job->curPos; i < max; i++)
1933 1.454 rillig if (job->outBuf[i] == '\0')
1934 1.454 rillig job->outBuf[i] = ' ';
1935 1.455 rillig
1936 1.455 rillig /* Look for the last newline in the bytes we just got. */
1937 1.341 rillig for (i = job->curPos + nr - 1;
1938 1.341 rillig i >= job->curPos && i != (size_t)-1; i--) {
1939 1.341 rillig if (job->outBuf[i] == '\n') {
1940 1.421 rillig gotNL = true;
1941 1.341 rillig break;
1942 1.341 rillig }
1943 1.1 cgd }
1944 1.16 christos
1945 1.341 rillig if (!gotNL) {
1946 1.341 rillig job->curPos += nr;
1947 1.341 rillig if (job->curPos == JOB_BUFSIZE) {
1948 1.341 rillig /*
1949 1.341 rillig * If we've run out of buffer space, we have no choice
1950 1.341 rillig * but to print the stuff. sigh.
1951 1.341 rillig */
1952 1.421 rillig fbuf = true;
1953 1.341 rillig i = job->curPos;
1954 1.341 rillig }
1955 1.120 dsl }
1956 1.341 rillig if (gotNL || fbuf) {
1957 1.341 rillig /*
1958 1.341 rillig * Need to send the output to the screen. Null terminate it
1959 1.341 rillig * first, overwriting the newline character if there was one.
1960 1.341 rillig * So long as the line isn't one we should filter (according
1961 1.341 rillig * to the shell description), we print the line, preceded
1962 1.341 rillig * by a target banner if this target isn't the same as the
1963 1.341 rillig * one for which we last printed something.
1964 1.341 rillig * The rest of the data in the buffer are then shifted down
1965 1.341 rillig * to the start of the buffer and curPos is set accordingly.
1966 1.341 rillig */
1967 1.341 rillig job->outBuf[i] = '\0';
1968 1.341 rillig if (i >= job->curPos) {
1969 1.460 rillig char *p;
1970 1.341 rillig
1971 1.425 rillig /*
1972 1.425 rillig * FIXME: SwitchOutputTo should be here, according to
1973 1.425 rillig * the comment above. But since PrintOutput does not
1974 1.425 rillig * do anything in the default shell, this bug has gone
1975 1.425 rillig * unnoticed until now.
1976 1.425 rillig */
1977 1.460 rillig p = PrintFilteredOutput(job->outBuf, &job->outBuf[i]);
1978 1.341 rillig
1979 1.341 rillig /*
1980 1.426 rillig * There's still more in the output buffer. This time,
1981 1.341 rillig * though, we know there's no newline at the end, so
1982 1.341 rillig * we add one of our own free will.
1983 1.341 rillig */
1984 1.460 rillig if (*p != '\0') {
1985 1.445 rillig if (!opts.silent)
1986 1.341 rillig SwitchOutputTo(job->node);
1987 1.155 sjg #ifdef USE_META
1988 1.341 rillig if (useMeta) {
1989 1.460 rillig meta_job_output(job, p,
1990 1.341 rillig gotNL ? "\n" : "");
1991 1.341 rillig }
1992 1.341 rillig #endif
1993 1.460 rillig (void)fprintf(stdout, "%s%s", p,
1994 1.341 rillig gotNL ? "\n" : "");
1995 1.341 rillig (void)fflush(stdout);
1996 1.341 rillig }
1997 1.341 rillig }
1998 1.341 rillig /*
1999 1.341 rillig * max is the last offset still in the buffer. Move any
2000 1.341 rillig * remaining characters to the start of the buffer and
2001 1.341 rillig * update the end marker curPos.
2002 1.341 rillig */
2003 1.341 rillig if (i < max) {
2004 1.341 rillig (void)memmove(job->outBuf, &job->outBuf[i + 1],
2005 1.341 rillig max - (i + 1));
2006 1.341 rillig job->curPos = max - (i + 1);
2007 1.341 rillig } else {
2008 1.341 rillig assert(i == max);
2009 1.341 rillig job->curPos = 0;
2010 1.155 sjg }
2011 1.120 dsl }
2012 1.341 rillig if (finish) {
2013 1.341 rillig /*
2014 1.341 rillig * If the finish flag is true, we must loop until we hit
2015 1.341 rillig * end-of-file on the pipe. This is guaranteed to happen
2016 1.341 rillig * eventually since the other end of the pipe is now closed
2017 1.341 rillig * (we closed it explicitly and the child has exited). When
2018 1.421 rillig * we do get an EOF, finish will be set false and we'll fall
2019 1.341 rillig * through and out.
2020 1.341 rillig */
2021 1.341 rillig goto again;
2022 1.180 joerg }
2023 1.1 cgd }
2024 1.1 cgd
2025 1.91 christos static void
2026 1.91 christos JobRun(GNode *targ)
2027 1.91 christos {
2028 1.483 rillig /* Don't let these special jobs overlap with other unrelated jobs. */
2029 1.341 rillig Compat_Make(targ, targ);
2030 1.482 rillig if (GNode_IsError(targ)) {
2031 1.449 rillig PrintOnError(targ, "\n\nStop.\n");
2032 1.341 rillig exit(1);
2033 1.341 rillig }
2034 1.91 christos }
2035 1.91 christos
2036 1.1 cgd void
2037 1.120 dsl Job_CatchChildren(void)
2038 1.1 cgd {
2039 1.487 rillig int pid;
2040 1.487 rillig int status;
2041 1.1 cgd
2042 1.341 rillig if (jobTokensRunning == 0)
2043 1.341 rillig return;
2044 1.405 sjg if (caught_sigchld == 0)
2045 1.405 sjg return;
2046 1.405 sjg caught_sigchld = 0;
2047 1.405 sjg
2048 1.341 rillig while ((pid = waitpid((pid_t)-1, &status, WNOHANG | WUNTRACED)) > 0) {
2049 1.505 rillig DEBUG2(JOB,
2050 1.505 rillig "Process with pid %d exited/stopped with status %#x.\n",
2051 1.341 rillig pid, status);
2052 1.421 rillig JobReapChild(pid, status, true);
2053 1.341 rillig }
2054 1.153 sjg }
2055 1.153 sjg
2056 1.153 sjg /*
2057 1.153 sjg * It is possible that wait[pid]() was called from elsewhere,
2058 1.153 sjg * this lets us reap jobs regardless.
2059 1.153 sjg */
2060 1.153 sjg void
2061 1.421 rillig JobReapChild(pid_t pid, int status, bool isJobs)
2062 1.153 sjg {
2063 1.487 rillig Job *job;
2064 1.153 sjg
2065 1.341 rillig if (jobTokensRunning == 0)
2066 1.341 rillig return;
2067 1.16 christos
2068 1.341 rillig job = JobFindPid(pid, JOB_ST_RUNNING, isJobs);
2069 1.341 rillig if (job == NULL) {
2070 1.487 rillig if (isJobs && !lurking_children)
2071 1.505 rillig Error("Child with pid %d and status %#x not in table?",
2072 1.487 rillig pid, status);
2073 1.487 rillig return;
2074 1.116 dsl }
2075 1.487 rillig
2076 1.341 rillig if (WIFSTOPPED(status)) {
2077 1.505 rillig DEBUG2(JOB, "Process for target %s, pid %d stopped\n",
2078 1.505 rillig job->node->name, job->pid);
2079 1.341 rillig if (!make_suspended) {
2080 1.341 rillig switch (WSTOPSIG(status)) {
2081 1.341 rillig case SIGTSTP:
2082 1.341 rillig (void)printf("*** [%s] Suspended\n",
2083 1.341 rillig job->node->name);
2084 1.341 rillig break;
2085 1.341 rillig case SIGSTOP:
2086 1.341 rillig (void)printf("*** [%s] Stopped\n",
2087 1.341 rillig job->node->name);
2088 1.341 rillig break;
2089 1.341 rillig default:
2090 1.341 rillig (void)printf("*** [%s] Stopped -- signal %d\n",
2091 1.341 rillig job->node->name, WSTOPSIG(status));
2092 1.341 rillig }
2093 1.421 rillig job->suspended = true;
2094 1.341 rillig }
2095 1.341 rillig (void)fflush(stdout);
2096 1.341 rillig return;
2097 1.1 cgd }
2098 1.1 cgd
2099 1.341 rillig job->status = JOB_ST_FINISHED;
2100 1.341 rillig job->exit_status = status;
2101 1.467 sjg if (WIFEXITED(status))
2102 1.467 sjg job->node->exit_status = WEXITSTATUS(status);
2103 1.116 dsl
2104 1.341 rillig JobFinish(job, status);
2105 1.1 cgd }
2106 1.1 cgd
2107 1.504 rillig static void
2108 1.504 rillig Job_Continue(Job *job)
2109 1.504 rillig {
2110 1.505 rillig DEBUG1(JOB, "Continuing pid %d\n", job->pid);
2111 1.504 rillig if (job->suspended) {
2112 1.504 rillig (void)printf("*** [%s] Continued\n", job->node->name);
2113 1.504 rillig (void)fflush(stdout);
2114 1.504 rillig job->suspended = false;
2115 1.504 rillig }
2116 1.504 rillig if (KILLPG(job->pid, SIGCONT) != 0)
2117 1.505 rillig DEBUG1(JOB, "Failed to send SIGCONT to pid %d\n", job->pid);
2118 1.504 rillig }
2119 1.504 rillig
2120 1.504 rillig static void
2121 1.504 rillig ContinueJobs(void)
2122 1.504 rillig {
2123 1.504 rillig Job *job;
2124 1.504 rillig
2125 1.504 rillig for (job = job_table; job < job_table_end; job++) {
2126 1.504 rillig if (job->status == JOB_ST_RUNNING &&
2127 1.504 rillig (make_suspended || job->suspended))
2128 1.504 rillig Job_Continue(job);
2129 1.504 rillig else if (job->status == JOB_ST_FINISHED)
2130 1.504 rillig JobFinish(job, job->exit_status);
2131 1.504 rillig }
2132 1.504 rillig make_suspended = false;
2133 1.504 rillig }
2134 1.504 rillig
2135 1.1 cgd void
2136 1.72 wiz Job_CatchOutput(void)
2137 1.1 cgd {
2138 1.341 rillig int nready;
2139 1.341 rillig Job *job;
2140 1.496 rillig unsigned i;
2141 1.169 christos
2142 1.341 rillig (void)fflush(stdout);
2143 1.120 dsl
2144 1.341 rillig do {
2145 1.492 rillig /* Maybe skip the job token pipe. */
2146 1.492 rillig nfds_t skip = wantToken ? 0 : 1;
2147 1.507 rillig nready = poll(fds + skip, fdsLen - skip, -1);
2148 1.341 rillig } while (nready < 0 && errno == EINTR);
2149 1.341 rillig
2150 1.341 rillig if (nready < 0)
2151 1.341 rillig Punt("poll: %s", strerror(errno));
2152 1.341 rillig
2153 1.492 rillig if (nready > 0 && childExitJob.inPollfd->revents & POLLIN) {
2154 1.492 rillig char token;
2155 1.487 rillig ssize_t count = read(childExitJob.inPipe, &token, 1);
2156 1.507 rillig if (count != 1)
2157 1.507 rillig Punt("childExitJob.read: %s",
2158 1.507 rillig count == 0 ? "EOF" : strerror(errno));
2159 1.507 rillig if (token == CEJ_RESUME_JOBS)
2160 1.507 rillig ContinueJobs();
2161 1.341 rillig nready--;
2162 1.169 christos }
2163 1.120 dsl
2164 1.341 rillig Job_CatchChildren();
2165 1.341 rillig if (nready == 0)
2166 1.341 rillig return;
2167 1.341 rillig
2168 1.408 rillig for (i = npseudojobs * nfds_per_job(); i < fdsLen; i++) {
2169 1.396 rillig if (fds[i].revents == 0)
2170 1.341 rillig continue;
2171 1.408 rillig job = jobByFdIndex[i];
2172 1.341 rillig if (job->status == JOB_ST_RUNNING)
2173 1.424 rillig CollectOutput(job, false);
2174 1.197 sjg #if defined(USE_FILEMON) && !defined(USE_FILEMON_DEV)
2175 1.341 rillig /*
2176 1.341 rillig * With meta mode, we may have activity on the job's filemon
2177 1.341 rillig * descriptor too, which at the moment is any pollfd other
2178 1.341 rillig * than job->inPollfd.
2179 1.341 rillig */
2180 1.341 rillig if (useMeta && job->inPollfd != &fds[i]) {
2181 1.448 rillig if (meta_job_event(job) <= 0)
2182 1.443 rillig fds[i].events = 0; /* never mind */
2183 1.341 rillig }
2184 1.341 rillig #endif
2185 1.341 rillig if (--nready == 0)
2186 1.341 rillig return;
2187 1.196 riastrad }
2188 1.1 cgd }
2189 1.1 cgd
2190 1.344 rillig static void
2191 1.344 rillig InitShellNameAndPath(void)
2192 1.344 rillig {
2193 1.355 rillig shellName = shell->name;
2194 1.344 rillig
2195 1.344 rillig #ifdef DEFSHELL_CUSTOM
2196 1.344 rillig if (shellName[0] == '/') {
2197 1.477 rillig shellPath = bmake_strdup(shellName);
2198 1.387 rillig shellName = str_basename(shellPath);
2199 1.344 rillig return;
2200 1.344 rillig }
2201 1.344 rillig #endif
2202 1.344 rillig
2203 1.344 rillig shellPath = str_concat3(_PATH_DEFSHELLDIR, "/", shellName);
2204 1.344 rillig }
2205 1.344 rillig
2206 1.81 sjg void
2207 1.120 dsl Shell_Init(void)
2208 1.81 sjg {
2209 1.344 rillig if (shellPath == NULL)
2210 1.344 rillig InitShellNameAndPath();
2211 1.344 rillig
2212 1.471 sjg Var_SetWithFlags(SCOPE_CMDLINE, ".SHELL", shellPath,
2213 1.471 sjg VAR_SET_INTERNAL|VAR_SET_READONLY);
2214 1.375 rillig if (shell->errFlag == NULL)
2215 1.375 rillig shell->errFlag = "";
2216 1.366 rillig if (shell->echoFlag == NULL)
2217 1.366 rillig shell->echoFlag = "";
2218 1.375 rillig if (shell->hasErrCtl && shell->errFlag[0] != '\0') {
2219 1.396 rillig if (shellErrFlag != NULL &&
2220 1.375 rillig strcmp(shell->errFlag, &shellErrFlag[1]) != 0) {
2221 1.341 rillig free(shellErrFlag);
2222 1.341 rillig shellErrFlag = NULL;
2223 1.341 rillig }
2224 1.446 rillig if (shellErrFlag == NULL)
2225 1.446 rillig shellErrFlag = str_concat2("-", shell->errFlag);
2226 1.341 rillig } else if (shellErrFlag != NULL) {
2227 1.341 rillig free(shellErrFlag);
2228 1.341 rillig shellErrFlag = NULL;
2229 1.341 rillig }
2230 1.81 sjg }
2231 1.81 sjg
2232 1.487 rillig /* Return the shell string literal that results in a newline character. */
2233 1.112 rillig const char *
2234 1.112 rillig Shell_GetNewline(void)
2235 1.112 rillig {
2236 1.355 rillig return shell->newline;
2237 1.112 rillig }
2238 1.112 rillig
2239 1.125 sjg void
2240 1.125 sjg Job_SetPrefix(void)
2241 1.125 sjg {
2242 1.463 rillig if (targPrefix != NULL)
2243 1.341 rillig free(targPrefix);
2244 1.463 rillig else if (!Var_Exists(SCOPE_GLOBAL, ".MAKE.JOB.PREFIX"))
2245 1.459 rillig Global_Set(".MAKE.JOB.PREFIX", "---");
2246 1.128 dsl
2247 1.459 rillig targPrefix = Var_Subst("${.MAKE.JOB.PREFIX}",
2248 1.475 rillig SCOPE_GLOBAL, VARE_EVAL);
2249 1.341 rillig /* TODO: handle errors */
2250 1.125 sjg }
2251 1.125 sjg
2252 1.342 rillig static void
2253 1.342 rillig AddSig(int sig, SignalProc handler)
2254 1.342 rillig {
2255 1.342 rillig if (bmake_signal(sig, SIG_IGN) != SIG_IGN) {
2256 1.342 rillig sigaddset(&caught_signals, sig);
2257 1.342 rillig (void)bmake_signal(sig, handler);
2258 1.342 rillig }
2259 1.342 rillig }
2260 1.342 rillig
2261 1.1 cgd void
2262 1.111 dsl Job_Init(void)
2263 1.1 cgd {
2264 1.341 rillig Job_SetPrefix();
2265 1.487 rillig
2266 1.341 rillig job_table = bmake_malloc((size_t)opts.maxJobs * sizeof *job_table);
2267 1.341 rillig memset(job_table, 0, (size_t)opts.maxJobs * sizeof *job_table);
2268 1.341 rillig job_table_end = job_table + opts.maxJobs;
2269 1.492 rillig wantToken = false;
2270 1.405 sjg caught_sigchld = 0;
2271 1.341 rillig
2272 1.341 rillig aborting = ABORT_NONE;
2273 1.341 rillig job_errors = 0;
2274 1.341 rillig
2275 1.341 rillig /*
2276 1.487 rillig * There is a non-zero chance that we already have children,
2277 1.487 rillig * e.g. after 'make -f- <<EOF'.
2278 1.341 rillig * Since their termination causes a 'Child (pid) not in table'
2279 1.341 rillig * message, Collect the status of any that are already dead, and
2280 1.341 rillig * suppress the error message if there are any undead ones.
2281 1.341 rillig */
2282 1.341 rillig for (;;) {
2283 1.341 rillig int rval, status;
2284 1.341 rillig rval = waitpid((pid_t)-1, &status, WNOHANG);
2285 1.341 rillig if (rval > 0)
2286 1.341 rillig continue;
2287 1.341 rillig if (rval == 0)
2288 1.421 rillig lurking_children = true;
2289 1.341 rillig break;
2290 1.341 rillig }
2291 1.1 cgd
2292 1.341 rillig Shell_Init();
2293 1.1 cgd
2294 1.341 rillig JobCreatePipe(&childExitJob, 3);
2295 1.117 dsl
2296 1.408 rillig {
2297 1.408 rillig size_t nfds = (npseudojobs + (size_t)opts.maxJobs) *
2298 1.408 rillig nfds_per_job();
2299 1.408 rillig fds = bmake_malloc(sizeof *fds * nfds);
2300 1.408 rillig jobByFdIndex = bmake_malloc(sizeof *jobByFdIndex * nfds);
2301 1.408 rillig }
2302 1.1 cgd
2303 1.341 rillig /* These are permanent entries and take slots 0 and 1 */
2304 1.495 rillig watchfd(&tokenPoolJob);
2305 1.341 rillig watchfd(&childExitJob);
2306 1.73 gson
2307 1.341 rillig sigemptyset(&caught_signals);
2308 1.502 rillig (void)bmake_signal(SIGCHLD, HandleSIGCHLD);
2309 1.341 rillig sigaddset(&caught_signals, SIGCHLD);
2310 1.122 dsl
2311 1.487 rillig /* Handle the signals specified by POSIX. */
2312 1.342 rillig AddSig(SIGINT, JobPassSig_int);
2313 1.342 rillig AddSig(SIGHUP, JobPassSig_term);
2314 1.342 rillig AddSig(SIGTERM, JobPassSig_term);
2315 1.342 rillig AddSig(SIGQUIT, JobPassSig_term);
2316 1.69 pk
2317 1.341 rillig /*
2318 1.487 rillig * These signals need to be passed to the jobs, as each job has its
2319 1.487 rillig * own process group and thus the terminal driver doesn't forward the
2320 1.487 rillig * signals itself.
2321 1.341 rillig */
2322 1.342 rillig AddSig(SIGTSTP, JobPassSig_suspend);
2323 1.342 rillig AddSig(SIGTTOU, JobPassSig_suspend);
2324 1.342 rillig AddSig(SIGTTIN, JobPassSig_suspend);
2325 1.342 rillig AddSig(SIGWINCH, JobCondPassSig);
2326 1.502 rillig AddSig(SIGCONT, HandleSIGCONT);
2327 1.16 christos
2328 1.341 rillig (void)Job_RunTarget(".BEGIN", NULL);
2329 1.487 rillig /* Create the .END node, see Targ_GetEndNode in Compat_MakeAll. */
2330 1.341 rillig (void)Targ_GetEndNode();
2331 1.1 cgd }
2332 1.1 cgd
2333 1.342 rillig static void
2334 1.342 rillig DelSig(int sig)
2335 1.342 rillig {
2336 1.397 rillig if (sigismember(&caught_signals, sig) != 0)
2337 1.342 rillig (void)bmake_signal(sig, SIG_DFL);
2338 1.342 rillig }
2339 1.342 rillig
2340 1.388 rillig static void
2341 1.388 rillig JobSigReset(void)
2342 1.69 pk {
2343 1.342 rillig DelSig(SIGINT);
2344 1.342 rillig DelSig(SIGHUP);
2345 1.342 rillig DelSig(SIGQUIT);
2346 1.342 rillig DelSig(SIGTERM);
2347 1.342 rillig DelSig(SIGTSTP);
2348 1.342 rillig DelSig(SIGTTOU);
2349 1.342 rillig DelSig(SIGTTIN);
2350 1.342 rillig DelSig(SIGWINCH);
2351 1.342 rillig DelSig(SIGCONT);
2352 1.341 rillig (void)bmake_signal(SIGCHLD, SIG_DFL);
2353 1.69 pk }
2354 1.69 pk
2355 1.1 cgd static Shell *
2356 1.301 rillig FindShellByName(const char *name)
2357 1.1 cgd {
2358 1.341 rillig Shell *sh = shells;
2359 1.341 rillig const Shell *shellsEnd = sh + sizeof shells / sizeof shells[0];
2360 1.1 cgd
2361 1.341 rillig for (sh = shells; sh < shellsEnd; sh++) {
2362 1.341 rillig if (strcmp(name, sh->name) == 0)
2363 1.341 rillig return sh;
2364 1.341 rillig }
2365 1.341 rillig return NULL;
2366 1.1 cgd }
2367 1.1 cgd
2368 1.338 rillig /*
2369 1.355 rillig * Parse a shell specification and set up 'shell', shellPath and
2370 1.338 rillig * shellName appropriately.
2371 1.1 cgd *
2372 1.72 wiz * Input:
2373 1.72 wiz * line The shell spec
2374 1.72 wiz *
2375 1.1 cgd * Results:
2376 1.487 rillig * Returns false if the specification was incorrect.
2377 1.487 rillig * If successful, 'shell' is usable, shellPath is the full path of the
2378 1.487 rillig * shell described by 'shell', and shellName is the final component of
2379 1.487 rillig * shellPath.
2380 1.1 cgd *
2381 1.1 cgd * Notes:
2382 1.487 rillig * A shell specification has the form ".SHELL: keyword=value...". Double
2383 1.487 rillig * quotes can be used to enclose blanks in words. A backslash escapes
2384 1.1 cgd * anything (most notably a double-quote and a space) and
2385 1.487 rillig * provides the usual escape sequences from C. There should be no
2386 1.487 rillig * unnecessary spaces in the word. The keywords are:
2387 1.246 rillig * name Name of shell.
2388 1.246 rillig * path Location of shell.
2389 1.246 rillig * quiet Command to turn off echoing.
2390 1.246 rillig * echo Command to turn echoing on
2391 1.487 rillig * filter The output from the shell command that turns off
2392 1.487 rillig * echoing, to be filtered from the final output.
2393 1.487 rillig * echoFlag Flag to turn echoing on at the start.
2394 1.487 rillig * errFlag Flag to turn error checking on at the start.
2395 1.487 rillig * hasErrCtl True if the shell has error checking control.
2396 1.487 rillig * newline String literal to represent a newline character.
2397 1.487 rillig * check If hasErrCtl is true: The command to turn on error
2398 1.487 rillig * checking. If hasErrCtl is false: The template for a
2399 1.487 rillig * shell command that echoes a command for which error
2400 1.487 rillig * checking is off.
2401 1.487 rillig * ignore If hasErrCtl is true: The command to turn off error
2402 1.487 rillig * checking. If hasErrCtl is false: The template for a
2403 1.487 rillig * shell command that executes a command so as to ignore
2404 1.487 rillig * any errors it returns.
2405 1.1 cgd */
2406 1.421 rillig bool
2407 1.72 wiz Job_ParseShell(char *line)
2408 1.1 cgd {
2409 1.341 rillig Words wordsList;
2410 1.341 rillig char **words;
2411 1.341 rillig char **argv;
2412 1.341 rillig size_t argc;
2413 1.341 rillig char *path;
2414 1.341 rillig Shell newShell;
2415 1.421 rillig bool fullSpec = false;
2416 1.341 rillig Shell *sh;
2417 1.341 rillig
2418 1.386 rillig /* XXX: don't use line as an iterator variable */
2419 1.341 rillig pp_skip_whitespace(&line);
2420 1.341 rillig
2421 1.407 rillig free(shell_freeIt);
2422 1.341 rillig
2423 1.341 rillig memset(&newShell, 0, sizeof newShell);
2424 1.341 rillig
2425 1.421 rillig wordsList = Str_Words(line, true);
2426 1.341 rillig words = wordsList.words;
2427 1.341 rillig argc = wordsList.len;
2428 1.341 rillig path = wordsList.freeIt;
2429 1.341 rillig if (words == NULL) {
2430 1.341 rillig Error("Unterminated quoted string [%s]", line);
2431 1.421 rillig return false;
2432 1.244 rillig }
2433 1.407 rillig shell_freeIt = path;
2434 1.1 cgd
2435 1.341 rillig for (path = NULL, argv = words; argc != 0; argc--, argv++) {
2436 1.341 rillig char *arg = *argv;
2437 1.341 rillig if (strncmp(arg, "path=", 5) == 0) {
2438 1.341 rillig path = arg + 5;
2439 1.341 rillig } else if (strncmp(arg, "name=", 5) == 0) {
2440 1.341 rillig newShell.name = arg + 5;
2441 1.341 rillig } else {
2442 1.341 rillig if (strncmp(arg, "quiet=", 6) == 0) {
2443 1.341 rillig newShell.echoOff = arg + 6;
2444 1.341 rillig } else if (strncmp(arg, "echo=", 5) == 0) {
2445 1.341 rillig newShell.echoOn = arg + 5;
2446 1.341 rillig } else if (strncmp(arg, "filter=", 7) == 0) {
2447 1.341 rillig newShell.noPrint = arg + 7;
2448 1.341 rillig newShell.noPrintLen = strlen(newShell.noPrint);
2449 1.341 rillig } else if (strncmp(arg, "echoFlag=", 9) == 0) {
2450 1.366 rillig newShell.echoFlag = arg + 9;
2451 1.341 rillig } else if (strncmp(arg, "errFlag=", 8) == 0) {
2452 1.375 rillig newShell.errFlag = arg + 8;
2453 1.341 rillig } else if (strncmp(arg, "hasErrCtl=", 10) == 0) {
2454 1.341 rillig char c = arg[10];
2455 1.341 rillig newShell.hasErrCtl = c == 'Y' || c == 'y' ||
2456 1.341 rillig c == 'T' || c == 't';
2457 1.341 rillig } else if (strncmp(arg, "newline=", 8) == 0) {
2458 1.341 rillig newShell.newline = arg + 8;
2459 1.341 rillig } else if (strncmp(arg, "check=", 6) == 0) {
2460 1.443 rillig /*
2461 1.443 rillig * Before 2020-12-10, these two variables had
2462 1.443 rillig * been a single variable.
2463 1.443 rillig */
2464 1.358 rillig newShell.errOn = arg + 6;
2465 1.364 rillig newShell.echoTmpl = arg + 6;
2466 1.341 rillig } else if (strncmp(arg, "ignore=", 7) == 0) {
2467 1.443 rillig /*
2468 1.443 rillig * Before 2020-12-10, these two variables had
2469 1.443 rillig * been a single variable.
2470 1.443 rillig */
2471 1.359 rillig newShell.errOff = arg + 7;
2472 1.364 rillig newShell.runIgnTmpl = arg + 7;
2473 1.341 rillig } else if (strncmp(arg, "errout=", 7) == 0) {
2474 1.364 rillig newShell.runChkTmpl = arg + 7;
2475 1.341 rillig } else if (strncmp(arg, "comment=", 8) == 0) {
2476 1.341 rillig newShell.commentChar = arg[8];
2477 1.341 rillig } else {
2478 1.341 rillig Parse_Error(PARSE_FATAL,
2479 1.341 rillig "Unknown keyword \"%s\"", arg);
2480 1.341 rillig free(words);
2481 1.421 rillig return false;
2482 1.341 rillig }
2483 1.421 rillig fullSpec = true;
2484 1.341 rillig }
2485 1.1 cgd }
2486 1.341 rillig
2487 1.12 christos if (path == NULL) {
2488 1.341 rillig if (newShell.name == NULL) {
2489 1.341 rillig Parse_Error(PARSE_FATAL,
2490 1.341 rillig "Neither path nor name specified");
2491 1.341 rillig free(words);
2492 1.421 rillig return false;
2493 1.341 rillig } else {
2494 1.341 rillig if ((sh = FindShellByName(newShell.name)) == NULL) {
2495 1.341 rillig Parse_Error(PARSE_WARNING,
2496 1.341 rillig "%s: No matching shell", newShell.name);
2497 1.341 rillig free(words);
2498 1.421 rillig return false;
2499 1.341 rillig }
2500 1.355 rillig shell = sh;
2501 1.341 rillig shellName = newShell.name;
2502 1.341 rillig if (shellPath != NULL) {
2503 1.478 rillig free(shellPath);
2504 1.341 rillig shellPath = NULL;
2505 1.341 rillig Shell_Init();
2506 1.341 rillig }
2507 1.341 rillig }
2508 1.1 cgd } else {
2509 1.478 rillig free(shellPath);
2510 1.477 rillig shellPath = bmake_strdup(path);
2511 1.465 rillig shellName = newShell.name != NULL ? newShell.name
2512 1.465 rillig : str_basename(path);
2513 1.341 rillig if (!fullSpec) {
2514 1.341 rillig if ((sh = FindShellByName(shellName)) == NULL) {
2515 1.341 rillig Parse_Error(PARSE_WARNING,
2516 1.341 rillig "%s: No matching shell", shellName);
2517 1.341 rillig free(words);
2518 1.421 rillig return false;
2519 1.341 rillig }
2520 1.355 rillig shell = sh;
2521 1.341 rillig } else {
2522 1.355 rillig shell = bmake_malloc(sizeof *shell);
2523 1.355 rillig *shell = newShell;
2524 1.341 rillig }
2525 1.487 rillig /* This will take care of shellErrFlag. */
2526 1.341 rillig Shell_Init();
2527 1.1 cgd }
2528 1.341 rillig
2529 1.396 rillig if (shell->echoOn != NULL && shell->echoOff != NULL)
2530 1.421 rillig shell->hasEchoCtl = true;
2531 1.1 cgd
2532 1.355 rillig if (!shell->hasErrCtl) {
2533 1.364 rillig if (shell->echoTmpl == NULL)
2534 1.364 rillig shell->echoTmpl = "";
2535 1.364 rillig if (shell->runIgnTmpl == NULL)
2536 1.364 rillig shell->runIgnTmpl = "%s\n";
2537 1.1 cgd }
2538 1.16 christos
2539 1.341 rillig /*
2540 1.487 rillig * Do not free up the words themselves, since they may be in use
2541 1.341 rillig * by the shell specification.
2542 1.341 rillig */
2543 1.341 rillig free(words);
2544 1.421 rillig return true;
2545 1.1 cgd }
2546 1.1 cgd
2547 1.341 rillig /*
2548 1.487 rillig * After receiving an interrupt signal, terminate all child processes and if
2549 1.487 rillig * necessary make the .INTERRUPT target.
2550 1.1 cgd */
2551 1.1 cgd static void
2552 1.421 rillig JobInterrupt(bool runINTERRUPT, int signo)
2553 1.69 pk {
2554 1.487 rillig Job *job;
2555 1.487 rillig GNode *interrupt;
2556 1.341 rillig sigset_t mask;
2557 1.341 rillig
2558 1.341 rillig aborting = ABORT_INTERRUPT;
2559 1.341 rillig
2560 1.500 rillig JobsTable_Lock(&mask);
2561 1.341 rillig
2562 1.341 rillig for (job = job_table; job < job_table_end; job++) {
2563 1.486 rillig if (job->status == JOB_ST_RUNNING && job->pid != 0) {
2564 1.341 rillig DEBUG2(JOB,
2565 1.341 rillig "JobInterrupt passing signal %d to child %d.\n",
2566 1.341 rillig signo, job->pid);
2567 1.341 rillig KILLPG(job->pid, signo);
2568 1.341 rillig }
2569 1.12 christos }
2570 1.12 christos
2571 1.486 rillig for (job = job_table; job < job_table_end; job++) {
2572 1.486 rillig if (job->status == JOB_ST_RUNNING && job->pid != 0) {
2573 1.486 rillig int status;
2574 1.486 rillig (void)waitpid(job->pid, &status, 0);
2575 1.486 rillig JobDeleteTarget(job->node);
2576 1.486 rillig }
2577 1.486 rillig }
2578 1.486 rillig
2579 1.500 rillig JobsTable_Unlock(&mask);
2580 1.69 pk
2581 1.445 rillig if (runINTERRUPT && !opts.touch) {
2582 1.341 rillig interrupt = Targ_FindNode(".INTERRUPT");
2583 1.341 rillig if (interrupt != NULL) {
2584 1.421 rillig opts.ignoreErrors = false;
2585 1.341 rillig JobRun(interrupt);
2586 1.341 rillig }
2587 1.1 cgd }
2588 1.341 rillig Trace_Log(MAKEINTR, NULL);
2589 1.368 rillig exit(signo); /* XXX: why signo? */
2590 1.1 cgd }
2591 1.1 cgd
2592 1.487 rillig /* Make the .END target, returning the number of job-related errors. */
2593 1.1 cgd int
2594 1.72 wiz Job_Finish(void)
2595 1.1 cgd {
2596 1.341 rillig GNode *endNode = Targ_GetEndNode();
2597 1.341 rillig if (!Lst_IsEmpty(&endNode->commands) ||
2598 1.341 rillig !Lst_IsEmpty(&endNode->children)) {
2599 1.463 rillig if (job_errors != 0)
2600 1.341 rillig Error("Errors reported so .END ignored");
2601 1.463 rillig else
2602 1.341 rillig JobRun(endNode);
2603 1.1 cgd }
2604 1.341 rillig return job_errors;
2605 1.22 christos }
2606 1.22 christos
2607 1.480 rillig #ifdef CLEANUP
2608 1.22 christos void
2609 1.72 wiz Job_End(void)
2610 1.22 christos {
2611 1.407 rillig free(shell_freeIt);
2612 1.480 rillig }
2613 1.30 mycroft #endif
2614 1.1 cgd
2615 1.487 rillig /* Waits for all running jobs to finish. */
2616 1.1 cgd void
2617 1.72 wiz Job_Wait(void)
2618 1.1 cgd {
2619 1.487 rillig aborting = ABORT_WAIT; /* Prevent other jobs from starting. */
2620 1.500 rillig while (jobTokensRunning != 0)
2621 1.341 rillig Job_CatchOutput();
2622 1.341 rillig aborting = ABORT_NONE;
2623 1.1 cgd }
2624 1.1 cgd
2625 1.341 rillig /*
2626 1.341 rillig * Abort all currently running jobs without handling output or anything.
2627 1.243 rillig * This function is to be called only in the event of a major error.
2628 1.243 rillig * Most definitely NOT to be called from JobInterrupt.
2629 1.341 rillig */
2630 1.1 cgd void
2631 1.72 wiz Job_AbortAll(void)
2632 1.1 cgd {
2633 1.487 rillig Job *job;
2634 1.487 rillig int status;
2635 1.16 christos
2636 1.341 rillig aborting = ABORT_ERROR;
2637 1.16 christos
2638 1.341 rillig if (jobTokensRunning != 0) {
2639 1.341 rillig for (job = job_table; job < job_table_end; job++) {
2640 1.341 rillig if (job->status != JOB_ST_RUNNING)
2641 1.341 rillig continue;
2642 1.341 rillig KILLPG(job->pid, SIGINT);
2643 1.341 rillig KILLPG(job->pid, SIGKILL);
2644 1.341 rillig }
2645 1.1 cgd }
2646 1.16 christos
2647 1.489 rillig while (waitpid((pid_t)-1, &status, WNOHANG) > 0)
2648 1.341 rillig continue;
2649 1.12 christos }
2650 1.12 christos
2651 1.35 christos static void
2652 1.72 wiz watchfd(Job *job)
2653 1.35 christos {
2654 1.341 rillig if (job->inPollfd != NULL)
2655 1.341 rillig Punt("Watching watched job");
2656 1.35 christos
2657 1.408 rillig fds[fdsLen].fd = job->inPipe;
2658 1.408 rillig fds[fdsLen].events = POLLIN;
2659 1.408 rillig jobByFdIndex[fdsLen] = job;
2660 1.408 rillig job->inPollfd = &fds[fdsLen];
2661 1.408 rillig fdsLen++;
2662 1.341 rillig #if defined(USE_FILEMON) && !defined(USE_FILEMON_DEV)
2663 1.341 rillig if (useMeta) {
2664 1.408 rillig fds[fdsLen].fd = meta_job_fd(job);
2665 1.408 rillig fds[fdsLen].events = fds[fdsLen].fd == -1 ? 0 : POLLIN;
2666 1.408 rillig jobByFdIndex[fdsLen] = job;
2667 1.408 rillig fdsLen++;
2668 1.341 rillig }
2669 1.196 riastrad #endif
2670 1.35 christos }
2671 1.35 christos
2672 1.35 christos static void
2673 1.72 wiz clearfd(Job *job)
2674 1.35 christos {
2675 1.341 rillig size_t i;
2676 1.341 rillig if (job->inPollfd == NULL)
2677 1.341 rillig Punt("Unwatching unwatched job");
2678 1.341 rillig i = (size_t)(job->inPollfd - fds);
2679 1.408 rillig fdsLen--;
2680 1.197 sjg #if defined(USE_FILEMON) && !defined(USE_FILEMON_DEV)
2681 1.341 rillig if (useMeta) {
2682 1.341 rillig assert(nfds_per_job() == 2);
2683 1.397 rillig if (i % 2 != 0)
2684 1.341 rillig Punt("odd-numbered fd with meta");
2685 1.408 rillig fdsLen--;
2686 1.341 rillig }
2687 1.341 rillig #endif
2688 1.461 rillig /* Move last job in table into hole made by dead job. */
2689 1.408 rillig if (fdsLen != i) {
2690 1.408 rillig fds[i] = fds[fdsLen];
2691 1.408 rillig jobByFdIndex[i] = jobByFdIndex[fdsLen];
2692 1.408 rillig jobByFdIndex[i]->inPollfd = &fds[i];
2693 1.341 rillig #if defined(USE_FILEMON) && !defined(USE_FILEMON_DEV)
2694 1.341 rillig if (useMeta) {
2695 1.408 rillig fds[i + 1] = fds[fdsLen + 1];
2696 1.408 rillig jobByFdIndex[i + 1] = jobByFdIndex[fdsLen + 1];
2697 1.341 rillig }
2698 1.196 riastrad #endif
2699 1.196 riastrad }
2700 1.341 rillig job->inPollfd = NULL;
2701 1.35 christos }
2702 1.35 christos
2703 1.495 rillig int
2704 1.495 rillig Job_TempFile(const char *pattern, char *tfile, size_t tfile_sz)
2705 1.495 rillig {
2706 1.495 rillig int fd;
2707 1.495 rillig sigset_t mask;
2708 1.495 rillig
2709 1.500 rillig JobsTable_Lock(&mask);
2710 1.495 rillig fd = mkTempFile(pattern, tfile, tfile_sz);
2711 1.495 rillig if (tfile != NULL && !DEBUG(SCRIPT))
2712 1.495 rillig unlink(tfile);
2713 1.500 rillig JobsTable_Unlock(&mask);
2714 1.495 rillig
2715 1.495 rillig return fd;
2716 1.495 rillig }
2717 1.495 rillig
2718 1.497 rillig static void
2719 1.497 rillig TokenPool_Write(char tok)
2720 1.497 rillig {
2721 1.497 rillig if (write(tokenPoolJob.outPipe, &tok, 1) != 1)
2722 1.500 rillig Punt("Cannot write \"%c\" to the token pool: %s",
2723 1.500 rillig tok, strerror(errno));
2724 1.497 rillig }
2725 1.497 rillig
2726 1.391 rillig /*
2727 1.495 rillig * Put a token (back) into the job token pool.
2728 1.391 rillig * This allows a make process to start a build job.
2729 1.391 rillig */
2730 1.40 sommerfe static void
2731 1.495 rillig TokenPool_Add(void)
2732 1.40 sommerfe {
2733 1.341 rillig char tok = JOB_TOKENS[aborting], tok1;
2734 1.101 dsl
2735 1.487 rillig /* If we are depositing an error token, flush everything else. */
2736 1.495 rillig while (tok != '+' && read(tokenPoolJob.inPipe, &tok1, 1) == 1)
2737 1.341 rillig continue;
2738 1.341 rillig
2739 1.499 rillig DEBUG3(JOB, "TokenPool_Add: pid %d, aborting %s, token %c\n",
2740 1.498 rillig getpid(), aborting_name[aborting], tok);
2741 1.497 rillig TokenPool_Write(tok);
2742 1.40 sommerfe }
2743 1.40 sommerfe
2744 1.495 rillig static void
2745 1.506 rillig TokenPool_InitClient(int tokenPoolReader, int tokenPoolWriter)
2746 1.419 sjg {
2747 1.506 rillig tokenPoolJob.inPipe = tokenPoolReader;
2748 1.506 rillig tokenPoolJob.outPipe = tokenPoolWriter;
2749 1.506 rillig (void)fcntl(tokenPoolReader, F_SETFD, FD_CLOEXEC);
2750 1.506 rillig (void)fcntl(tokenPoolWriter, F_SETFD, FD_CLOEXEC);
2751 1.419 sjg }
2752 1.419 sjg
2753 1.487 rillig /* Prepare the job token pipe in the root make process. */
2754 1.495 rillig static void
2755 1.506 rillig TokenPool_InitServer(int maxJobTokens)
2756 1.40 sommerfe {
2757 1.341 rillig int i;
2758 1.341 rillig char jobarg[64];
2759 1.201 rillig
2760 1.495 rillig JobCreatePipe(&tokenPoolJob, 15);
2761 1.40 sommerfe
2762 1.341 rillig snprintf(jobarg, sizeof jobarg, "%d,%d",
2763 1.495 rillig tokenPoolJob.inPipe, tokenPoolJob.outPipe);
2764 1.40 sommerfe
2765 1.415 rillig Global_Append(MAKEFLAGS, "-J");
2766 1.415 rillig Global_Append(MAKEFLAGS, jobarg);
2767 1.101 dsl
2768 1.341 rillig /*
2769 1.341 rillig * Preload the job pipe with one token per job, save the one
2770 1.341 rillig * "extra" token for the primary job.
2771 1.341 rillig */
2772 1.497 rillig SetNonblocking(tokenPoolJob.outPipe);
2773 1.506 rillig for (i = 1; i < maxJobTokens; i++)
2774 1.495 rillig TokenPool_Add();
2775 1.495 rillig }
2776 1.495 rillig
2777 1.495 rillig void
2778 1.506 rillig TokenPool_Init(int maxJobTokens, int tokenPoolReader, int tokenPoolWriter)
2779 1.495 rillig {
2780 1.506 rillig if (tokenPoolReader >= 0 && tokenPoolWriter >= 0)
2781 1.506 rillig TokenPool_InitClient(tokenPoolReader, tokenPoolWriter);
2782 1.495 rillig else
2783 1.506 rillig TokenPool_InitServer(maxJobTokens);
2784 1.40 sommerfe }
2785 1.40 sommerfe
2786 1.495 rillig /* Return a taken token to the pool. */
2787 1.40 sommerfe void
2788 1.495 rillig TokenPool_Return(void)
2789 1.40 sommerfe {
2790 1.341 rillig jobTokensRunning--;
2791 1.341 rillig if (jobTokensRunning < 0)
2792 1.341 rillig Punt("token botch");
2793 1.396 rillig if (jobTokensRunning != 0 || JOB_TOKENS[aborting] != '+')
2794 1.495 rillig TokenPool_Add();
2795 1.40 sommerfe }
2796 1.40 sommerfe
2797 1.341 rillig /*
2798 1.495 rillig * Attempt to take a token from the pool.
2799 1.40 sommerfe *
2800 1.487 rillig * If the pool is empty, set wantToken so that we wake up when a token is
2801 1.243 rillig * released.
2802 1.40 sommerfe *
2803 1.495 rillig * Returns true if a token was taken, and false if the pool is currently
2804 1.341 rillig * empty.
2805 1.341 rillig */
2806 1.421 rillig bool
2807 1.495 rillig TokenPool_Take(void)
2808 1.40 sommerfe {
2809 1.341 rillig char tok, tok1;
2810 1.341 rillig ssize_t count;
2811 1.40 sommerfe
2812 1.492 rillig wantToken = false;
2813 1.499 rillig DEBUG3(JOB, "TokenPool_Take: pid %d, aborting %s, running %d\n",
2814 1.498 rillig getpid(), aborting_name[aborting], jobTokensRunning);
2815 1.60 enami
2816 1.500 rillig if (aborting != ABORT_NONE || jobTokensRunning >= opts.maxJobs)
2817 1.421 rillig return false;
2818 1.40 sommerfe
2819 1.495 rillig count = read(tokenPoolJob.inPipe, &tok, 1);
2820 1.341 rillig if (count == 0)
2821 1.500 rillig Fatal("eof on job pipe");
2822 1.341 rillig if (count < 0 && jobTokensRunning != 0) {
2823 1.463 rillig if (errno != EAGAIN)
2824 1.341 rillig Fatal("job pipe read: %s", strerror(errno));
2825 1.499 rillig DEBUG1(JOB, "TokenPool_Take: pid %d blocked for token\n",
2826 1.499 rillig getpid());
2827 1.492 rillig wantToken = true;
2828 1.421 rillig return false;
2829 1.40 sommerfe }
2830 1.102 dsl
2831 1.341 rillig if (count == 1 && tok != '+') {
2832 1.341 rillig /* make being aborted - remove any other job tokens */
2833 1.499 rillig DEBUG2(JOB, "TokenPool_Take: pid %d aborted by token %c\n",
2834 1.499 rillig getpid(), tok);
2835 1.495 rillig while (read(tokenPoolJob.inPipe, &tok1, 1) == 1)
2836 1.341 rillig continue;
2837 1.341 rillig /* And put the stopper back */
2838 1.497 rillig TokenPool_Write(tok);
2839 1.487 rillig if (shouldDieQuietly(NULL, 1)) {
2840 1.488 rillig Job_Wait();
2841 1.487 rillig exit(6);
2842 1.487 rillig }
2843 1.341 rillig Fatal("A failure has been detected "
2844 1.341 rillig "in another branch of the parallel make");
2845 1.341 rillig }
2846 1.341 rillig
2847 1.341 rillig if (count == 1 && jobTokensRunning == 0)
2848 1.341 rillig /* We didn't want the token really */
2849 1.497 rillig TokenPool_Write(tok);
2850 1.102 dsl
2851 1.341 rillig jobTokensRunning++;
2852 1.499 rillig DEBUG1(JOB, "TokenPool_Take: pid %d took a token\n", getpid());
2853 1.421 rillig return true;
2854 1.41 sommerfe }
2855 1.41 sommerfe
2856 1.487 rillig /* Make the named target if found, exit if the target fails. */
2857 1.421 rillig bool
2858 1.341 rillig Job_RunTarget(const char *target, const char *fname)
2859 1.341 rillig {
2860 1.341 rillig GNode *gn = Targ_FindNode(target);
2861 1.341 rillig if (gn == NULL)
2862 1.421 rillig return false;
2863 1.172 christos
2864 1.341 rillig if (fname != NULL)
2865 1.417 rillig Var_Set(gn, ALLSRC, fname);
2866 1.172 christos
2867 1.341 rillig JobRun(gn);
2868 1.421 rillig return true;
2869 1.172 christos }
2870 1.172 christos
2871 1.73 gson #ifdef USE_SELECT
2872 1.73 gson int
2873 1.73 gson emul_poll(struct pollfd *fd, int nfd, int timeout)
2874 1.73 gson {
2875 1.341 rillig fd_set rfds, wfds;
2876 1.341 rillig int i, maxfd, nselect, npoll;
2877 1.341 rillig struct timeval tv, *tvp;
2878 1.341 rillig long usecs;
2879 1.73 gson
2880 1.341 rillig FD_ZERO(&rfds);
2881 1.341 rillig FD_ZERO(&wfds);
2882 1.73 gson
2883 1.341 rillig maxfd = -1;
2884 1.341 rillig for (i = 0; i < nfd; i++) {
2885 1.341 rillig fd[i].revents = 0;
2886 1.73 gson
2887 1.341 rillig if (fd[i].events & POLLIN)
2888 1.341 rillig FD_SET(fd[i].fd, &rfds);
2889 1.341 rillig
2890 1.341 rillig if (fd[i].events & POLLOUT)
2891 1.341 rillig FD_SET(fd[i].fd, &wfds);
2892 1.341 rillig
2893 1.341 rillig if (fd[i].fd > maxfd)
2894 1.341 rillig maxfd = fd[i].fd;
2895 1.341 rillig }
2896 1.201 rillig
2897 1.341 rillig if (maxfd >= FD_SETSIZE) {
2898 1.341 rillig Punt("Ran out of fd_set slots; "
2899 1.341 rillig "recompile with a larger FD_SETSIZE.");
2900 1.341 rillig }
2901 1.73 gson
2902 1.341 rillig if (timeout < 0) {
2903 1.341 rillig tvp = NULL;
2904 1.341 rillig } else {
2905 1.341 rillig usecs = timeout * 1000;
2906 1.341 rillig tv.tv_sec = usecs / 1000000;
2907 1.341 rillig tv.tv_usec = usecs % 1000000;
2908 1.341 rillig tvp = &tv;
2909 1.341 rillig }
2910 1.73 gson
2911 1.341 rillig nselect = select(maxfd + 1, &rfds, &wfds, NULL, tvp);
2912 1.73 gson
2913 1.341 rillig if (nselect <= 0)
2914 1.341 rillig return nselect;
2915 1.73 gson
2916 1.341 rillig npoll = 0;
2917 1.341 rillig for (i = 0; i < nfd; i++) {
2918 1.341 rillig if (FD_ISSET(fd[i].fd, &rfds))
2919 1.341 rillig fd[i].revents |= POLLIN;
2920 1.73 gson
2921 1.341 rillig if (FD_ISSET(fd[i].fd, &wfds))
2922 1.341 rillig fd[i].revents |= POLLOUT;
2923 1.73 gson
2924 1.341 rillig if (fd[i].revents)
2925 1.341 rillig npoll++;
2926 1.341 rillig }
2927 1.73 gson
2928 1.341 rillig return npoll;
2929 1.73 gson }
2930 1.443 rillig #endif /* USE_SELECT */
2931