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