dump-runlisp-image.c, runlisp-base.conf: Make version string more detailed.
[runlisp] / dump-runlisp-image.c
1 /* -*-c-*-
2 *
3 * Dump custom Lisp images for faster script execution
4 *
5 * (c) 2020 Mark Wooding
6 */
7
8 /*----- Licensing notice --------------------------------------------------*
9 *
10 * This file is part of Runlisp, a tool for invoking Common Lisp scripts.
11 *
12 * Runlisp is free software: you can redistribute it and/or modify it
13 * under the terms of the GNU General Public License as published by the
14 * Free Software Foundation; either version 3 of the License, or (at your
15 * option) any later version.
16 *
17 * Runlisp is distributed in the hope that it will be useful, but WITHOUT
18 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
19 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
20 * for more details.
21 *
22 * You should have received a copy of the GNU General Public License
23 * along with Runlisp. If not, see <https://www.gnu.org/licenses/>.
24 */
25
26 /*----- Header files ------------------------------------------------------*/
27
28 #include "config.h"
29
30 #include <assert.h>
31 #include <ctype.h>
32 #include <errno.h>
33 #include <signal.h>
34 #include <stdio.h>
35 #include <stdlib.h>
36 #include <string.h>
37 #include <time.h>
38
39 #include <dirent.h>
40 #include <fcntl.h>
41 #include <unistd.h>
42
43 #include <sys/select.h>
44 #include <sys/stat.h>
45 #include <sys/time.h>
46 #include <sys/uio.h>
47 #include <sys/wait.h>
48
49 #include "common.h"
50 #include "lib.h"
51 #include "mdwopt.h"
52 #include "sha256.h"
53
54 /*----- Static data -------------------------------------------------------*/
55
56 /* The state required to break an output stream from a subprocess into lines
57 * so we can prefix them appropriately. Once our process starts, the `buf'
58 * points to a buffer of `MAXLINE' bytes. This is arranged as a circular
59 * buffer, containing `len' bytes starting at offset `off', and wrapping
60 * around to the start of the buffer if it runs off the end.
61 *
62 * The descriptor `fd' is reset to -1 after it's seen end-of-file.
63 */
64 struct linebuf {
65 int fd; /* our file descriptor (or -1) */
66 char *buf; /* line buffer, or null */
67 unsigned off, len; /* offset */
68 };
69 #define MAXLINE 16384u /* maximum acceptable line length */
70
71 /* Job-state constants. */
72 enum {
73 JST_INTERN, /* not that kind of job */
74 JST_VERSION, /* hashing the Lisp version number */
75 JST_DUMP, /* dumping the custom image */
76 JST_NSTATE
77 };
78
79 /* The state associated with an image-dumping job. */
80 struct job {
81 struct treap_node _node; /* treap intrusion */
82 struct job *next; /* next job in whichever list */
83 unsigned st; /* job state (`JST_...') */
84 struct config_section *sect; /* the system-definition section */
85 struct config_var *dumpvar; /* the `dump-image' variable */
86 struct argv av_version, av_dump; /* argument vectors to execute */
87 char *imgnew, *imghash, *imgnewlink, *imglink; /* link and final outputs */
88 char *oldimg; /* old image name */
89 FILE *log; /* log output file (`stdout'?) */
90 pid_t kid; /* process id of child (or -1) */
91 int exit; /* exit status from child */
92 struct sha256_state h; /* hash context for version */
93 struct linebuf out, err; /* line buffers for stdout, stderr */
94 };
95 #define JOB_NAME(job) TREAP_NODE_KEY(job)
96 #define JOB_NAMELEN(job) TREAP_NODE_KEYLEN(job)
97
98 static struct treap jobs = TREAP_INIT, /* Lisp systems seen so far */
99 good = TREAP_INIT; /* files ok to be in image dir */
100 static struct job /* lists of jobs */
101 *job_ready, **job_ready_tail = &job_ready, /* queue of jobs to start */
102 *job_delete, **job_delete_tail = &job_delete, /* queue of delete jobs */
103 *job_run; /* list of active jobs */
104 static unsigned nrun, maxrun = 1; /* running and maximum job counts */
105 static int rc = 0; /* code that we should return */
106 static int nullfd; /* file descriptor for `/dev/null' */
107 static const char *tmpdir; /* temporary directory path */
108
109 static int sig_pipe[2] = { -1, -1 }; /* pipe for reporting signals */
110 static sigset_t caught, pending; /* signals we catch; have caught */
111 static int sigloss = -1; /* signal that caused us to lose */
112
113 static unsigned flags = 0; /* flags for the application */
114 #define AF_BOGUS 0x0001u /* invalid comand-line syntax */
115 #define AF_SETCONF 0x0002u /* explicit configuration */
116 #define AF_DRYRUN 0x0004u /* don't actually do it */
117 #define AF_ALL 0x0008u /* dump all known Lisps */
118 #define AF_FORCE 0x0010u /* dump even if images exist */
119 #define AF_CHECKINST 0x0020u /* check Lisp exists before dump */
120 #define AF_REMOVE 0x0040u /* remove selected Lisp images */
121 #define AF_CLEAN 0x0080u /* remove other Lisp images */
122 #define AF_JUNK 0x0100u /* remove unrecognized files */
123
124 /*----- Miscellany --------------------------------------------------------*/
125
126 /* Report a (printf(3)-style) message MSG, and remember to fail later. */
127 static PRINTF_LIKE(1, 2) void bad(const char *msg, ...)
128 { va_list ap; va_start(ap, msg); vmoan(msg, ap); va_end(ap); rc = 127; }
129
130 /* Answer whether a string consists entirely of hex digits. */
131 static int hex_digits_p(const char *p, size_t sz)
132 {
133 const char *l;
134
135 for (l = p + sz; p < l; p++) if (!ISXDIGIT(*p)) return (0);
136 return (1);
137 }
138
139 /*----- File utilities ----------------------------------------------------*/
140
141 /* Main recursive subroutine for `recursive_delete'.
142 *
143 * The string DD currently contains the pathname of a directory, without a
144 * trailing `/' (though there is /space/ for a terminating zero or whatever).
145 * Recursively delete all of the files and directories within it. Appending
146 * further text to DD is OK, but clobbering the characters which are there
147 * already isn't allowed.
148 */
149 static void recursive_delete_(struct dstr *dd)
150 {
151 DIR *dir;
152 struct dirent *d;
153 size_t n = dd->len;
154
155 /* Open the directory. */
156 dd->p[n] = 0; dir = opendir(dd->p);
157 if (!dir)
158 lose("failed to open directory `%s' for cleanup: %s",
159 dd->p, strerror(errno));
160
161 /* We'll need to build pathnames for the files inside the directory, so add
162 * the separating `/' character. Remember the length of this prefix
163 * because this is the point we'll be rewinding to for each filename we
164 * find.
165 */
166 dd->p[n++] = '/';
167
168 /* Now go through each file in turn. */
169 for (;;) {
170
171 /* Get a filename. If we've run out then we're done. Skip the special
172 * `.' and `..' entries.
173 */
174 d = readdir(dir); if (!d) break;
175 if (d->d_name[0] == '.' && (!d->d_name[1] ||
176 (d->d_name[1] == '.' && !d->d_name[2])))
177 continue;
178
179 /* Rewind the string offset and append the new filename. */
180 dd->len = n; dstr_puts(dd, d->d_name);
181
182 /* Try to delete it the usual way. If it was actually a directory then
183 * recursively delete it instead. (We could lstat(2) it first, but this
184 * should be at least as quick to identify a directory, and it'll save a
185 * lstat(2) call in the (common) case that it's not a directory.
186 */
187 if (!unlink(dd->p));
188 else if (errno == EISDIR) recursive_delete_(dd);
189 else lose("failed to delete file `%s': %s", dd->p, strerror(errno));
190 }
191
192 /* We're done. Try to delete the directory. (It's possible that there was
193 * some problem with enumerating the directory, but we'll ignore that: if
194 * it matters then the directory won't be empty and the rmdir(2) will
195 * fail.)
196 */
197 closedir(dir);
198 dd->p[--n] = 0;
199 if (rmdir(dd->p))
200 lose("failed to delete directory `%s': %s", dd->p, strerror(errno));
201 }
202
203 /* Recursively delete the thing named PATH. */
204 static void recursive_delete(const char *path)
205 {
206 struct dstr d = DSTR_INIT;
207 dstr_puts(&d, path); recursive_delete_(&d); dstr_release(&d);
208 }
209
210 /* Configure a file descriptor FD.
211 *
212 * Set its nonblocking state to NONBLOCK and close-on-exec state to CLOEXEC.
213 * In both cases, -1 means to leave it alone, zero means to turn it off, and
214 * any other nonzero value means to turn it on.
215 */
216 static int configure_fd(const char *what, int fd, int nonblock, int cloexec)
217 {
218 int fl, nfl;
219
220 if (nonblock != -1) {
221 fl = fcntl(fd, F_GETFL); if (fl < 0) goto fail;
222 if (nonblock) nfl = fl | O_NONBLOCK;
223 else nfl = fl&~O_NONBLOCK;
224 if (fl != nfl && fcntl(fd, F_SETFL, nfl)) goto fail;
225 }
226
227 if (cloexec != -1) {
228 fl = fcntl(fd, F_GETFD); if (fl < 0) goto fail;
229 if (cloexec) nfl = fl | FD_CLOEXEC;
230 else nfl = fl&~FD_CLOEXEC;
231 if (fl != nfl && fcntl(fd, F_SETFD, nfl)) goto fail;
232 }
233
234 return (0);
235
236 fail:
237 bad("failed to configure %s descriptor: %s", what, strerror(errno));
238 return (-1);
239 }
240
241 /* Create a temporary directory and remember where we put it. */
242 static void set_tmpdir(void)
243 {
244 struct dstr d = DSTR_INIT;
245 size_t n;
246 unsigned i;
247
248 /* Start building the path name. Remember the length: we'll rewind to
249 * here and try again if our first attempt doesn't work.
250 */
251 dstr_putf(&d, "%s/runlisp.%d.", my_getenv("TMPDIR", "/tmp"), getpid());
252 i = 0; n = d.len;
253
254 /* Keep trying until it works. */
255 for (;;) {
256
257 /* Build a complete name. */
258 d.len = n; dstr_putf(&d, "%d", rand());
259
260 /* Try to create the directory. If it worked, we're done. If it failed
261 * with `EEXIST' then we'll try again for a while, but give up it it
262 * doesn't look like we're making any progress. If it failed for some
263 * other reason then there's probably not much hope so give up.
264 */
265 if (!mkdir(d.p, 0700)) break;
266 else if (errno != EEXIST)
267 lose("failed to create temporary directory `%s': %s",
268 d.p, strerror(errno));
269 else if (++i >= 32) {
270 d.len = n; dstr_puts(&d, "???");
271 lose("failed to create temporary directory `%s': too many attempts",
272 d.p);
273 }
274 }
275
276 /* Remember the directory name. */
277 tmpdir = xstrndup(d.p, d.len); dstr_release(&d);
278 }
279
280 /*----- Signal handling ---------------------------------------------------*/
281
282 /* Forward reference into job management. */
283 static void reap_children(void);
284
285 /* Clean things up on exit.
286 *
287 * Currently this just means to delete the temporary directory if we've made
288 * one.
289 */
290 static void cleanup(void)
291 { if (tmpdir) { recursive_delete(tmpdir); tmpdir = 0; } }
292
293 /* Check to see whether any signals have arrived, and do the sensible thing
294 * with them.
295 */
296 static void check_signals(void)
297 {
298 sigset_t old, pend;
299 char buf[32];
300 ssize_t n;
301
302 /* Ensure exclusive access to the signal-handling machinery, drain the
303 * signal pipe, and take a copy of the set of caught signals.
304 */
305 sigprocmask(SIG_BLOCK, &caught, &old);
306 pend = pending; sigemptyset(&pending);
307 for (;;) {
308 n = read(sig_pipe[0], buf, sizeof(buf));
309 if (!n) lose("(internal) signal pipe closed!");
310 if (n < 0) break;
311 }
312 if (errno != EAGAIN && errno != EWOULDBLOCK)
313 lose("failed to read signal pipe: %s", strerror(errno));
314 sigprocmask(SIG_SETMASK, &old, 0);
315
316 /* Check for each signal of interest to us.
317 *
318 * Interrupty signals just set `sigloss' -- the `run_jobs' loop will know
319 * to unravel everything if this happens. If `SIGCHLD' happened, then
320 * check on job process status.
321 */
322 if (sigismember(&pend, SIGINT)) sigloss = SIGINT;
323 else if (sigismember(&pend, SIGHUP)) sigloss = SIGHUP;
324 else if (sigismember(&pend, SIGTERM)) sigloss = SIGTERM;
325 if (sigismember(&pend, SIGCHLD)) reap_children();
326 }
327
328 /* The actual signal handler.
329 *
330 * Set the appropriate signal bit in `pending', and a byte (of any value)
331 * down the signal pipe to wake up the select(2) loop.
332 */
333 static void handle_signal(int sig)
334 {
335 sigset_t old;
336 char x = '!';
337
338 /* Ensure exclusive access while we fiddle with the `caught' set. */
339 sigprocmask(SIG_BLOCK, &caught, &old);
340 sigaddset(&pending, sig);
341 sigprocmask(SIG_SETMASK, &old, 0);
342
343 /* Wake up the select(2) loop. If this fails, there's not a lot we can do
344 * about it.
345 */
346 DISCARD(write(sig_pipe[1], &x, 1));
347 }
348
349 /* Install our signal handler to catch SIG.
350 *
351 * If `SIGF_IGNOK' is set in F then don't trap the signal if it's currently
352 * ignored. (This is used for signals like `SIGINT', which usually should
353 * interrupt us; but if the caller wants us to ignore them, we should do as
354 * it wants.)
355 *
356 * WHAT describes the signal, for use in diagnostic messages.
357 */
358 #define SIGF_IGNOK 1u
359 static void set_signal_handler(const char *what, int sig, unsigned f)
360 {
361 struct sigaction sa, sa_old;
362
363 sigaddset(&caught, sig);
364
365 if (f&SIGF_IGNOK) {
366 if (sigaction(sig, 0, &sa_old)) goto fail;
367 if (sa_old.sa_handler == SIG_IGN) return;
368 }
369
370 sa.sa_handler = handle_signal;
371 sigemptyset(&sa.sa_mask);
372 sa.sa_flags = SA_NOCLDSTOP;
373 if (sigaction(sig, &sa, 0)) goto fail;
374
375 return;
376
377 fail:
378 lose("failed to set %s signal handler: %s", what, strerror(errno));
379 }
380
381 /*----- Line buffering ----------------------------------------------------*/
382
383 /* Find the next newline in the line buffer BUF.
384 *
385 * The search starts at `BUF->off', and potentially covers the entire buffer
386 * contents. Set *LINESZ_OUT to the length of the line, in bytes. (Callers
387 * must beware that the text of the line may wrap around the ends of the
388 * buffer.) Return zero if we found a newline, or nonzero if the search
389 * failed.
390 */
391 static int find_newline(struct linebuf *buf, size_t *linesz_out)
392 {
393 char *nl;
394
395 if (buf->off + buf->len <= MAXLINE) {
396 /* The buffer contents is in one piece. Just search it. */
397
398 nl = memchr(buf->buf + buf->off, '\n', buf->len);
399 if (nl) { *linesz_out = (nl - buf->buf) - buf->off; return (0); }
400
401 } else {
402 /* The buffer contents is in two pieces. We must search both of them. */
403
404 nl = memchr(buf->buf + buf->off, '\n', MAXLINE - buf->off);
405 if (nl) { *linesz_out = (nl - buf->buf) - buf->off; return (0); }
406 nl = memchr(buf->buf, '\n', buf->len - (MAXLINE - buf->off));
407 if (nl)
408 { *linesz_out = (nl - buf->buf) + (MAXLINE - buf->off); return (0); }
409 }
410
411 return (-1);
412 }
413
414 /* Write a completed line out to the JOB's log file.
415 *
416 * The line starts at BUF->off, and continues for N bytes, not including the
417 * newline (which, in fact, might not exist at all). Precede the actual text
418 * of the line with the JOB's name, and the MARKER character, and follow it
419 * with the TAIL text (which should include an actual newline character).
420 */
421 static void write_line(struct job *job, struct linebuf *buf,
422 size_t n, char marker, const char *tail)
423 {
424 fprintf(job->log, "%-13s %c ", JOB_NAME(job), marker);
425 if (buf->off + n <= MAXLINE)
426 fwrite(buf->buf + buf->off, 1, n, job->log);
427 else {
428 fwrite(buf->buf + buf->off, 1, MAXLINE - buf->off, job->log);
429 fwrite(buf->buf, 1, n - (MAXLINE - buf->off), job->log);
430 }
431 fputs(tail, job->log);
432 }
433
434 /* Hash N bytes freshly added to the buffer BUF. */
435 static void hash_input(struct linebuf *buf, size_t n, struct sha256_state *h)
436 {
437 size_t start = (buf->off + buf->len)%MAXLINE;
438
439 if (start + n <= MAXLINE)
440 sha256_hash(h, buf->buf + start, n);
441 else {
442 sha256_hash(h, buf->buf + start, MAXLINE - start);
443 sha256_hash(h, buf->buf, n - (MAXLINE - start));
444 }
445 }
446
447 /* Collect output lines from JOB's process and write them to the log.
448 *
449 * Read data from BUF's file descriptor. Output complete (or overlong) lines
450 * usng `write_line'. On end-of-file, output any final incomplete line in
451 * the same way, close the descriptor, and set it to -1.
452 *
453 * As a rather unpleasant quirk, if the hash-state pointer H is not null,
454 * then also feed all the data received into it.
455 */
456 static void prefix_lines(struct job *job, struct linebuf *buf, char marker,
457 struct sha256_state *h)
458 {
459 struct iovec iov[2]; int niov;
460 ssize_t n;
461 size_t linesz;
462
463 /* Read data into the buffer. This fancy dance with readv(2) is probably
464 * overkill.
465 *
466 * We can't have BUF->len = MAXLINE because we'd have flushed out a
467 * maximum-length buffer as an incomplete line last time.
468 */
469 assert(buf->len < MAXLINE);
470 if (!buf->off) {
471 iov[0].iov_base = buf->buf + buf->len;
472 iov[0].iov_len = MAXLINE - buf->len;
473 niov = 1;
474 } else if (buf->off + buf->len >= MAXLINE) {
475 iov[0].iov_base = buf->buf + buf->off + buf->len - MAXLINE;
476 iov[0].iov_len = MAXLINE - buf->len;
477 niov = 1;
478 } else {
479 iov[0].iov_base = buf->buf + buf->off + buf->len;
480 iov[0].iov_len = MAXLINE - (buf->off + buf->len);
481 iov[1].iov_base = buf->buf;
482 iov[1].iov_len = buf->off;
483 niov = 1;
484 }
485 n = readv(buf->fd, iov, niov);
486
487 if (n < 0) {
488 /* An error occurred. If there's no data to read after all then just
489 * move on. Otherwise we have a problem.
490 */
491
492 if (errno == EAGAIN || errno == EWOULDBLOCK) return;
493 lose("failed to read job `%s' output stream: %s",
494 JOB_NAME(job), strerror(errno));
495 } else if (!n) {
496 /* We've hit end-of-file. Close the stream, and write out any
497 * unterminated partial line.
498 */
499
500 close(buf->fd); buf->fd = -1;
501 if (buf->len)
502 write_line(job, buf, buf->len, marker, " [missing final newline]\n");
503 } else {
504 /* We read some fresh data. Output any new complete lines. */
505
506 /* If we're supposed to hash data as it comes in then we should do that
507 * now.
508 */
509 if (h) hash_input(buf, n, h);
510
511 /* Include the new material in the buffer length, and write out any
512 * complete lines we find.
513 */
514 buf->len += n;
515 while (!find_newline(buf, &linesz)) {
516 write_line(job, buf, linesz, marker, "\n");
517 buf->len -= linesz + 1;
518 buf->off += linesz + 1; if (buf->off >= MAXLINE) buf->off -= MAXLINE;
519 }
520
521 if (!buf->len)
522 /* If there's nothing left then we might as well reset the buffer
523 * offset to the start of the buffer.
524 */
525 buf->off = 0;
526 else if (buf->len == MAXLINE) {
527 /* We've filled the buffer with stuff that's not a whole line. Flush
528 * it out anyway.
529 */
530 write_line(job, buf, MAXLINE, marker, " [...]\n");
531 buf->off = buf->len = 0;
532 }
533 }
534 }
535
536 /*----- Job management ----------------------------------------------------*/
537
538 /* Record the SZ-byte leafname at P as being legitimate, so that it doesn't
539 * get junked.
540 */
541 static void notice_filename(const char *p, size_t sz)
542 {
543 struct treap_node *node;
544 struct treap_path path;
545
546 node = treap_probe(&good, p, sz, &path);
547 if (!node) {
548 node = xmalloc(sizeof(*node));
549 treap_insert(&good, &path, node, p, sz);
550 }
551 }
552
553 /* There are basically two kinds of jobs.
554 *
555 * An `internal' job -- state `JST_INTERN' -- can be handled entirely within
556 * this process. Internal jobs have trivial lifecycles: they're created, put
557 * on a queue, executed, and thrown away. Jobs are executed when some code
558 * decides to walk the appropriate queue and do the work. As a result, they
559 * don't need to have distinctive states: `JST_INTERN' only exists to
560 * distinguish internal jobs from active ones if they somehow manage to end
561 * up in the external-job machinery.
562 *
563 * External jobs all work in basically the same way: we fork and exec a
564 * sequence of subprocess to do the work. The majority of handling external
565 * jobs is in the care and feeding of these subprocesses, so they end up on
566 * various lists primarily concerned with the state of the subprocesses, and
567 * the progress of the job through its sequence of subprocesses is recorded
568 * in the job's `st' field.
569 *
570 * External jobs have a comparatively complicated lifecycle.
571 *
572 * * Initially, the job is on the `ready' queue by `add_job'. It has no
573 * child process or log file.
574 *
575 * * At some point, `start_jobs' decides to start this job up: a log file
576 * is created (if the job doesn't have one already), a child process is
577 * forked, and pipes are set up to capture the child's output. It gets
578 * moved to the `run' list (which is not maintained in any particular
579 * order). Jobs on the `run' list participate in the main select(2)
580 * loop.
581 *
582 * * When the job's child process dies and the pipes capturing its output
583 * streams finally dry up, the job is considered finished. What happens
584 * next depends on its state: either it gets updated somehow, and pushed
585 * back onto the end of the `ready' queue so that another child can be
586 * started, or the job is finished and dies.
587 *
588 * The counter `nrun' counts the number of actually running jobs, i.e., those
589 * with living child processes. This doesn't simply count the number of jobs
590 * on the `run' list: remember that the latter also contains jobs whose child
591 * has died, but whose output has not yet been collected.
592 */
593
594 /* Consider a Lisp system description and maybe add a job to the right queue.
595 *
596 * The Lisp system is described by the configuration section SECT. Most of
597 * the function is spent on inspecting this section for suitability and
598 * deciding what to do about it.
599 *
600 * The precise behaviour depends on F, which should be the bitwise-OR of a
601 * `JQ_...' constant and zero or more flags, as follows.
602 *
603 * * The bits covered by `JMASK_QUEUE' identify which queue the job should
604 * be added to if the section defines a cromulent Lisp system:
605 *
606 * -- `JQ_NONE' -- don't actually make a job at all;
607 * -- `JQ_READY' -- add the Lisp to the `job_ready' queue, so we'll; or
608 * -- `JQ_DELETE' -- add the Lisp to the `job_delete' queue.
609 *
610 * * `JF_PICKY': The user identified this Lisp system explicitly, so
611 * complain if the configuration section doesn't look right. This is
612 * clear if the caller is just enumerating all of the configuration
613 * sections: without this feature, we'd be checking everything twice,
614 * which (a) is inefficient, and -- more importantly -- (b) could lead to
615 * problems if the two checks are inconsistent.
616 *
617 * * `JF_CHECKINST': Ignore this Lisp if `AF_CHECKINST' is set and it's not
618 * actually installed. (This is usually set for `JQ_READY' calls, so
619 * that we don't try to dump Lisps which aren't there, but clear for
620 * `JQ_DELETE' calls so that we clear out Lisps which have gone away.)
621 *
622 * * `JF_CHECKEXIST': Ignore this Lisp if its image file already exists.
623 *
624 * * `JF_NOTICE': Record the Lisp's image basename in the `good' treap so
625 * that we can identify everything else we find in the image directory as
626 * junk.
627 */
628 #define JMASK_QUEUE 3u /* which queue to add good Lisp to */
629 #define JQ_NONE 0u /* don't add to any queue */
630 #define JQ_READY 1u /* `job_ready' */
631 #define JQ_DELETE 2u /* `job_delete' */
632 #define JF_PICKY 4u /* lose if section isn't Lisp defn */
633 #define JF_CHECKINST 8u /* maybe check Lisp is installed */
634 #define JF_CHECKEXIST 16u /* skip if image already exists */
635 #define JF_NOTICE 32u /* record Lisp's image basename */
636
637 #define JADD_NAMED (JQ_READY | JF_PICKY | JF_CHECKINST)
638 #define JADD_DEFAULT (JQ_READY | JF_CHECKINST)
639 #define JADD_CLEANUP (JQ_DELETE)
640 #define JADD_NOTICE (JQ_NONE)
641 static void add_job(unsigned f, struct config_section *sect)
642 {
643 const char *name;
644 struct job *job, ***tail;
645 struct treap_path jobpath;
646 struct config_var *dumpvar, *runvar, *imgvar;
647 struct dstr d = DSTR_INIT, dd = DSTR_INIT;
648 struct argv av_version = ARGV_INIT, av_dump = ARGV_INIT;
649 struct stat st;
650 char *imgnewlink = 0, *imglink = 0, *oldimg = 0, *p;
651 unsigned jst;
652 size_t i, len;
653 ssize_t n;
654 unsigned fef;
655
656 /* We'll want the section's name for all sorts of things. */
657 name = CONFIG_SECTION_NAME(sect);
658 len = CONFIG_SECTION_NAMELEN(sect);
659
660 /* Check to see whether this Lisp system is already queued up.
661 *
662 * We'll get around to adding the new job node to the treap right at the
663 * end, so use a separate path object to keep track of where to put it.
664 */
665 job = treap_probe(&jobs, name, len, &jobpath);
666 if (job) {
667 if ((f&JF_PICKY) && verbose >= 1)
668 moan("ignoring duplicate Lisp `%s'", JOB_NAME(job));
669 goto end;
670 }
671
672 /* Check that the section defines a Lisp, and that it can be dumped.
673 *
674 * It's not obvious that this is right. Maybe there should be some
675 * additional flag so that we don't check dumpability if we're planning to
676 * delete the image. But it /is/ right: since the thing which tells us
677 * whether we can dump is that the section tells us the image's name, if
678 * it can't be dumped then we won't know what file to delete! So we have
679 * no choice.
680 */
681 runvar = config_find_var(&config, sect, CF_INHERIT, "run-script");
682 if (!runvar) {
683 if (f&JF_PICKY) lose("unknown Lisp implementation `%s'", name);
684 else if (verbose >= 3) moan("skipping non-Lisp section `%s'", name);
685 goto end;
686 }
687 imgvar = config_find_var(&config, sect, CF_INHERIT, "image-file");
688 if (!imgvar) {
689 if (f&JF_PICKY)
690 lose("Lisp implementation `%s' doesn't use custom images", name);
691 else if (verbose >= 3)
692 moan("skipping Lisp `%s': no custom image support", name);
693 goto end;
694 }
695
696 /* Check that the other necessary variables are present. */
697 dumpvar = config_find_var(&config, sect, CF_INHERIT, "dump-image");
698 if (!dumpvar)
699 lose("variable `dump-image' not defined for Lisp `%s'", name);
700
701 /* Build the job's command lines. */
702 config_subst_split_var(&config, sect, runvar, &av_version);
703 if (!av_version.n)
704 lose("empty `run-script' command for Lisp implementation `%s'", name);
705 argv_append(&av_version,
706 config_subst_string_alloc
707 (&config, sect, "<internal>",
708 "?${lisp-version?(lisp-implementation-version)}"));
709 config_subst_split_var(&config, sect, dumpvar, &av_dump);
710 if (!av_dump.n)
711 lose("empty `dump-image' command for Lisp implementation `%s'", name);
712
713 /* If we're supposed to check that the Lisp exists before proceeding then
714 * do that. There are /two/ commands to check: the basic Lisp command,
715 * /and/ the command to actually do the dumping, which might not be the
716 * same thing. (Be careful not to check the same command twice, though,
717 * because that would cause us to spam the user with redundant
718 * diagnostics.)
719 */
720 if ((f&JF_CHECKINST) && (flags&AF_CHECKINST)) {
721 fef = (verbose >= 3 ? FEF_VERBOSE : 0);
722 if (!found_in_path_p(av_version.v[0], fef)) {
723 if (verbose >= 3)
724 moan("skipping Lisp `%s': can't find Lisp command `%s'",
725 name, av_version.v[0]);
726 goto end;
727 }
728 if (STRCMP(av_version.v[0], !=, av_dump.v[0]) &&
729 !found_in_path_p(av_dump.v[0], fef)) {
730 if (verbose >= 3)
731 moan("skipping Lisp `%s': can't find dump command `%s'",
732 av_dump.v[0], d.p);
733 goto end;
734 }
735 }
736
737 /* Collect the output image file names. */
738 imglink =
739 config_subst_string_alloc(&config, sect, "<internal>", "${@image-link}");
740 imgnewlink =
741 config_subst_string_alloc(&config, sect,
742 "<internal>", "${@image-newlink}");
743
744 /* Determine the image link basename. If necessary, record it so that it
745 * doesn't get junked.
746 */
747 dstr_reset(&dd); config_subst_var(&config, sect, imgvar, &dd);
748 if (f&JF_NOTICE) notice_filename(dd.p, dd.len);
749
750 /* Fill in the directory name for the output image. */
751 dstr_reset(&d);
752 p = strrchr(imglink, '/');
753 if (p) dstr_putm(&d, imglink, p + 1 - imglink);
754
755 /* Inspect the existing image link if there is one, and record its
756 * destination.
757 */
758 for (;;) {
759
760 /* Read the link destination. The `lstat'/`readlink' two-step is
761 * suggested by the POSIX specification.
762 */
763 if (lstat(imglink, &st)) {
764 if (verbose >= (errno == ENOENT ? 3 : 1))
765 moan("failed to read metadata for Lisp `%s' image link `%s': %s",
766 name, imglink, strerror(errno));
767 break;
768 }
769 if (!S_ISLNK(st.st_mode)) {
770 if (verbose >= 1)
771 moan("Lisp `%s' image link `%s' isn't a symbolic link",
772 name, imglink);
773 break;
774 }
775 dstr_ensure(&d, st.st_size + 1);
776 n = readlink(imglink, d.p + d.len, d.sz - d.len);
777 if (n < 0) {
778 moan("failed to read Lisp `%s' image link `%s': %s",
779 name, imglink, strerror(errno));
780 break;
781 }
782 if (n == d.sz - d.len) continue;
783
784 /* Check that the link has the right form. (We don't want to delete the
785 * referent if it's not actually our image.)
786 *
787 * We expect the referent to look like ${image-file} followed by a hyphen
788 * and some hex digits.
789 */
790 if (n <= dd.len ||
791 STRNCMP(d.p + d.len, !=, dd.p, dd.len) ||
792 d.p[d.len + dd.len] != '-' ||
793 !hex_digits_p(d.p + (d.len + dd.len + 1), n - (dd.len + 1))) {
794 if (verbose >= 1)
795 moan("Lisp `%s' image link `%s' has unexpected referent `%s'",
796 name, imglink, d.p);
797 break;
798 }
799
800 /* OK, so it looks legit. Protect it from being junked. */
801 if (f&JF_NOTICE) notice_filename(d.p + d.len, n);
802 d.p[d.len + n] = 0; d.len += n;
803 oldimg = xstrndup(d.p, d.len);
804 break;
805 }
806
807 /* All preflight checks complete. Build the job and hook it onto the end
808 * of the list. (Steal the command-line vector so that we don't try to
809 * free it during cleanup.)
810 */
811 switch (f&JMASK_QUEUE) {
812 case JQ_NONE: jst = JST_INTERN; tail = 0; break;
813 case JQ_READY: jst = JST_VERSION; tail = &job_ready_tail; break;
814 case JQ_DELETE: jst = JST_INTERN; tail = &job_delete_tail; break;
815 default: assert(0);
816 }
817 job = xmalloc(sizeof(*job));
818 job->st = jst; job->sect = sect; job->dumpvar = dumpvar;
819 job->kid = -1; job->log = 0;
820 job->out.fd = -1; job->out.buf = 0;
821 job->err.fd = -1; job->err.buf = 0;
822 job->av_version = av_version; argv_init(&av_version);
823 argv_init(&job->av_dump);
824 job->imgnew = 0; job->imghash = 0;
825 job->imgnewlink = imgnewlink; imgnewlink = 0;
826 job->imglink = imglink; imglink = 0;
827 job->oldimg = oldimg; oldimg = 0;
828 treap_insert(&jobs, &jobpath, &job->_node, name, len);
829 if (tail) { **tail = job; *tail = &job->next; }
830
831 end:
832 /* All done. Cleanup time. */
833 for (i = 0; i < av_version.n; i++) free(av_version.v[i]);
834 for (i = 0; i < av_dump.n; i++) free(av_dump.v[i]);
835 free(imgnewlink); free(imglink); free(oldimg);
836 dstr_release(&d); dstr_release(&dd);
837 argv_release(&av_version); argv_release(&av_dump);
838 }
839
840 /* As `add_job' above, but look the Lisp implementation up by name.
841 *
842 * The flags passed to `add_job' are augmented with `JF_PICKY' because this
843 * is an explicitly-named Lisp implementation.
844 */
845 static void add_named_job(unsigned f, const char *name, size_t len)
846 {
847 struct config_section *sect;
848
849 sect = config_find_section_n(&config, 0, name, len);
850 if (!sect) lose("unknown Lisp implementation `%.*s'", (int)len, name);
851 add_job(f | JF_PICKY, sect);
852 }
853
854 /* Free the JOB and all the resources it holds.
855 *
856 * Close the pipes; kill the child process. Everything must go.
857 */
858 static void release_job(struct job *job)
859 {
860 size_t i;
861 struct job *j;
862
863 if (job->kid > 0) kill(job->kid, SIGKILL); /* ?? */
864 if (job->log && job->log != stdout) fclose(job->log);
865 free(job->imgnew); free(job->imghash);
866 free(job->imglink); free(job->imgnewlink);
867 free(job->oldimg);
868 for (i = 0; i < job->av_version.n; i++) free(job->av_version.v[i]);
869 for (i = 0; i < job->av_dump.n; i++) free(job->av_dump.v[i]);
870 argv_release(&job->av_version); argv_release(&job->av_dump);
871 free(job->out.buf); if (job->out.fd >= 0) close(job->out.fd);
872 free(job->err.buf); if (job->err.fd >= 0) close(job->err.fd);
873 j = treap_remove(&jobs, JOB_NAME(job), JOB_NAMELEN(job)); assert(j == job);
874 free(job);
875 }
876
877 /* Do all the necessary things when JOB finishes (successfully or not).
878 *
879 * Eventually the job is either freed (using `release_job'), or updated and
880 * stuffed back into the `job_run' queue. The caller is expected to have
881 * already unlinked the job from its current list.
882 */
883 static void finish_job(struct job *job)
884 {
885 char buf[16483], *p;
886 unsigned char *hbuf;
887 struct dstr d = DSTR_INIT;
888 size_t i, n;
889 int ok = 0;
890
891 /* Start a final line to the job log describing its eventual fate.
892 *
893 * This is where we actually pick apart the exit status. Set `ok' if it
894 * actually succeeded, because that's all anything else cares about.
895 */
896 fprintf(job->log, "%-13s > ", JOB_NAME(job));
897 if (WIFEXITED(job->exit)) {
898 if (!WEXITSTATUS(job->exit))
899 { fputs("completed successfully\n", job->log); ok = 1; }
900 else
901 fprintf(job->log, "failed with exit status %d\n",
902 WEXITSTATUS(job->exit));
903 } else if (WIFSIGNALED(job->exit))
904 fprintf(job->log, "killed by signal %d (%s%s)", WTERMSIG(job->exit),
905 #if defined(HAVE_STRSIGNAL)
906 strsignal(WTERMSIG(job->exit)),
907 #elif defined(HAVE_DECL_SYS_SIGLIST)
908 sys_siglist[WTERMSIG(job->exit)],
909 #else
910 "unknown signal",
911 #endif
912 #ifdef WCOREDUMP
913 WCOREDUMP(job->exit) ? "; core dumped" :
914 #endif
915 "");
916 else
917 fprintf(job->log, "exited with incomprehensible status %06o\n",
918 job->exit);
919
920 /* What happens next depends on the state of the job. This is the main
921 * place which advanced the job state machine.
922 */
923 if (ok) switch (job->st) {
924
925 case JST_VERSION:
926 /* We've retrieved the Lisp system's version string. */
927
928 /* Complete the hashing and convert to hex. */
929 hbuf = (unsigned char *)buf + 32; sha256_done(&job->h, hbuf);
930 for (i = 0; i < 8; i++) sprintf(buf + 2*i, "%02x", hbuf[i]);
931 if (verbose >= 2)
932 moan("Lisp `%s' version hash = %s", JOB_NAME(job), buf);
933
934 /* Determine the final version-qualified name for the image. */
935 config_set_var(&config, job->sect, CF_LITERAL, "@hash", buf);
936 job->imghash =
937 config_subst_string_alloc(&config, job->sect,
938 "<internal>", "${@image-out}");
939 job->imgnew =
940 config_subst_string_alloc(&config, job->sect,
941 "<internal>", "${@image-new}");
942
943 /* Determine the basename of the final image. */
944 p = strrchr(job->imghash, '/'); if (p) p++; else p = job->imghash;
945
946 /* Inspect the current link pointer to see if we have the right
947 * version.
948 */
949 if (!(flags&AF_FORCE) &&
950 job->oldimg &&
951 STRCMP(job->oldimg, ==, job->imghash) &&
952 !access(job->oldimg, F_OK)) {
953 if (verbose >= 2)
954 moan("Lisp `%s' image `%s' already up-to-date",
955 JOB_NAME(job), job->imghash);
956 break;
957 }
958
959 /* Make sure that there's a clear space for the new image to be
960 * written.
961 */
962 if (!(flags&AF_DRYRUN) && unlink(job->imgnew) && errno != ENOENT) {
963 bad("failed to clear Lisp `%s' image staging path `%s': %s",
964 JOB_NAME(job), job->imgnew, strerror(errno));
965 break;
966 }
967
968 /* If we're still here then we've decided to dump a new image. Update
969 * the job state, and put it back on the run queue.
970 */
971 config_subst_split_var(&config, job->sect,
972 job->dumpvar, &job->av_dump);
973 assert(job->av_dump.n);
974 job->st = JST_DUMP;
975 *job_ready_tail = job; job_ready_tail = &job->next; job->next = 0;
976 job = 0;
977 break;
978
979 case JST_DUMP:
980 /* We've finished dumping a custom image. It's time to apply the
981 * finishing touches.
982 */
983
984 /* Rename the image into place. If this fails, blame it on the dump
985 * job, because the chances are good that it failed to produce the
986 * image properly.
987 */
988 if (verbose >= 3)
989 moan("rename completed Lisp `%s' image `%s' to `%s'",
990 JOB_NAME(job), job->imgnew, job->imghash);
991 if (rename(job->imgnew, job->imghash)) {
992 fprintf(job->log, "%-13s > failed to rename Lisp `%s' "
993 "output image `%s' to `%s': %s",
994 JOB_NAME(job), JOB_NAME(job),
995 job->imgnew, job->imghash, strerror(errno));
996 ok = 0; break;
997 }
998
999 /* Determine the basename of the final image. */
1000 p = strrchr(job->imghash, '/'); if (p) p++; else p = job->imghash;
1001
1002 /* Build the symlink. Start by setting the link in the staging path,
1003 * and then rename, in order to ensure continuity.
1004 */
1005 if (unlink(job->imgnewlink) && errno != ENOENT) {
1006 bad("failed to clear Lisp `%s' link staging path `%s': %s",
1007 JOB_NAME(job), job->imgnewlink, strerror(errno));
1008 break;
1009 }
1010 if (verbose >= 3)
1011 moan("establish Lisp `%s' image link `%s' referring to `%s'",
1012 JOB_NAME(job), job->imglink, job->imghash);
1013 if (symlink(p, job->imgnewlink)) {
1014 bad("failed to create Lisp `%s' image link `%s': %s",
1015 JOB_NAME(job), job->imgnewlink, strerror(errno));
1016 break;
1017 }
1018 if (rename(job->imgnewlink, job->imglink)) {
1019 bad("failed to rename Lisp `%s' image link `%s' to `%s': %s",
1020 JOB_NAME(job), job->imgnewlink, job->imglink, strerror(errno));
1021 break;
1022 }
1023 if (job->oldimg && STRCMP(job->oldimg, !=, job->imghash)) {
1024 if (verbose >= 3)
1025 moan("remove old Lisp `%s' image `%s'",
1026 JOB_NAME(job), job->oldimg);
1027 if (unlink(job->oldimg) && errno != ENOENT) {
1028 if (verbose >= 1)
1029 moan("failed to delete old Lisp `%s' image `%s': %s",
1030 JOB_NAME(job), job->oldimg, strerror(errno));
1031 }
1032 }
1033
1034 /* I think we're all done. */
1035 break;
1036
1037 default:
1038 assert(0);
1039 }
1040
1041 /* If the job failed and we're being quiet then write out the log that we
1042 * made.
1043 */
1044 if (!ok && verbose < 2) {
1045 rewind(job->log);
1046 for (;;) {
1047 n = fread(buf, 1, sizeof(buf), job->log);
1048 if (n) fwrite(buf, 1, n, stdout);
1049 if (n < sizeof(buf)) break;
1050 }
1051 }
1052
1053 /* Also make a node to stderr about what happened. (Just to make sure
1054 * that we've gotten someone's attention.)
1055 */
1056 if (!ok) bad("failed to dump Lisp `%s'", JOB_NAME(job));
1057
1058 /* Finally free the job control block. */
1059 if (job) release_job(job);
1060 dstr_release(&d);
1061 }
1062
1063 /* Called after `SIGCHLD': collect exit statuses and mark jobs as dead. */
1064 static void reap_children(void)
1065 {
1066 struct job *job;
1067 pid_t kid;
1068 int st;
1069
1070 for (;;) {
1071
1072 /* Collect a child exit status. If there aren't any more then we're
1073 * done.
1074 */
1075 kid = waitpid(0, &st, WNOHANG);
1076 if (kid <= 0) break;
1077
1078 /* Try to find a matching job. If we can't, then we should just ignore
1079 * it.
1080 */
1081 for (job = job_run; job; job = job->next)
1082 if (job->kid == kid) goto found;
1083 continue;
1084
1085 found:
1086 /* Mark the job as dead, and save its exit status. */
1087 job->exit = st; job->kid = -1; nrun--;
1088 }
1089
1090 /* If there was a problem with waitpid(2) then report it. */
1091 if (kid < 0 && errno != ECHILD)
1092 lose("failed to collect child process exit status: %s", strerror(errno));
1093 }
1094
1095 /* Execute the handler for some JOB. */
1096 static NORETURN void job_child(struct job *job, struct argv *av)
1097 {
1098 try_exec(av, 0);
1099 moan("failed to run `%s': %s", av->v[0], strerror(errno));
1100 _exit(127);
1101 }
1102
1103 /* Start up jobs while there are (a) jobs to run and (b) slots to run them
1104 * in.
1105 */
1106 static void start_jobs(void)
1107 {
1108 struct dstr d = DSTR_INIT;
1109 int p_out[2], p_err[2];
1110 struct job *job;
1111 struct argv *av;
1112 pid_t kid;
1113
1114 /* Keep going until either we run out of jobs, or we've got enough running
1115 * already.
1116 */
1117 while (job_ready && nrun < maxrun) {
1118
1119 /* Set things up ready. If things go wrong, we need to know what stuff
1120 * needs to be cleaned up.
1121 */
1122 job = job_ready; job_ready = job->next;
1123 if (!job_ready) job_ready_tail = &job_ready;
1124 p_out[0] = p_out[1] = p_err[0] = p_err[1] = -1;
1125
1126 /* Figure out what to do. */
1127 switch (job->st) {
1128 case JST_VERSION: av = &job->av_version; break;
1129 case JST_DUMP: av = &job->av_dump; break;
1130 default: assert(0);
1131 }
1132
1133 /* If we're not actually going to do anything, now is the time to not do
1134 * that. We should do the version-hashing step unconditionally.
1135 */
1136 switch (job->st) {
1137 case JST_VERSION:
1138 break;
1139 case JST_DUMP:
1140 if (flags&AF_DRYRUN) {
1141 if (try_exec(av,
1142 TEF_DRYRUN |
1143 (verbose >= 2 && !(flags&AF_CHECKINST)
1144 ? TEF_VERBOSE : 0)))
1145 rc = 127;
1146 else if (verbose >= 2)
1147 printf("%-13s > not dumping `%s' (dry run)\n",
1148 JOB_NAME(job), JOB_NAME(job));
1149 release_job(job);
1150 continue;
1151 }
1152 break;
1153 default:
1154 assert(0);
1155 }
1156
1157 /* Do one-time setup for external jobs. */
1158 if (!job->log) {
1159
1160 /* Make a temporary subdirectory for this job to use. */
1161 dstr_reset(&d); dstr_putf(&d, "%s/%s", tmpdir, JOB_NAME(job));
1162 if (mkdir(d.p, 0700)) {
1163 bad("failed to create working directory for job `%s': %s",
1164 JOB_NAME(job), strerror(errno));
1165 goto fail;
1166 }
1167
1168 /* Create the job's log file. If we're being verbose then that's just
1169 * our normal standard output -- /not/ stderr: it's likely that users
1170 * will want to pipe this stuff through a pager or something, and
1171 * that'll be easier if we use stdout. Otherwise, make a file in the
1172 * temporary directory.
1173 */
1174 if (verbose >= 2)
1175 job->log = stdout;
1176 else {
1177 dstr_puts(&d, "/log"); job->log = fopen(d.p, "w+");
1178 if (!job->log)
1179 lose("failed to open log file `%s': %s", d.p, strerror(errno));
1180 }
1181 }
1182
1183 /* Make the pipes to capture the child process's standard output and
1184 * error streams.
1185 */
1186 if (pipe(p_out) || pipe(p_err)) {
1187 bad("failed to create pipes for job `%s': %s",
1188 JOB_NAME(job), strerror(errno));
1189 goto fail;
1190 }
1191 if (configure_fd("job stdout pipe", p_out[0], 1, 1) ||
1192 configure_fd("job stdout pipe", p_out[1], 0, 1) ||
1193 configure_fd("job stderr pipe", p_err[0], 1, 1) ||
1194 configure_fd("job stderr pipe", p_err[1], 0, 1) ||
1195 configure_fd("log file", fileno(job->log), 1, 1))
1196 goto fail;
1197
1198 /* Initialize the output-processing structures ready for use. */
1199 if (job->st == JST_VERSION) sha256_init(&job->h);
1200 job->out.buf = xmalloc(MAXLINE); job->out.off = job->out.len = 0;
1201 job->out.fd = p_out[0]; p_out[0] = -1;
1202 job->err.buf = xmalloc(MAXLINE); job->err.off = job->err.len = 0;
1203 job->err.fd = p_err[0]; p_err[0] = -1;
1204
1205 /* Print a note to the top of the log. */
1206 dstr_reset(&d); argv_string(&d, av);
1207 fprintf(job->log, "%-13s > starting %s\n", JOB_NAME(job), d.p);
1208
1209 /* Flush the standard output stream. (Otherwise the child might try to
1210 * flush it too.)
1211 */
1212 fflush(stdout);
1213
1214 /* Spin up the child process. */
1215 kid = fork();
1216 if (kid < 0) {
1217 bad("failed to fork process for job `%s': %s",
1218 JOB_NAME(job), strerror(errno));
1219 goto fail;
1220 }
1221 if (!kid) {
1222 if (dup2(nullfd, 0) < 0 ||
1223 dup2(p_out[1], 1) < 0 ||
1224 dup2(p_err[1], 2) < 0)
1225 lose("failed to juggle job `%s' file descriptors: %s",
1226 JOB_NAME(job), strerror(errno));
1227 job_child(job, av);
1228 }
1229
1230 /* Close the ends of the pipes that we don't need. Move the job into
1231 * the running list.
1232 */
1233 close(p_out[1]); close(p_err[1]);
1234 job->kid = kid; job->next = job_run; job_run = job; nrun++;
1235 continue;
1236
1237 fail:
1238 /* Clean up the wreckage if it didn't work. */
1239 if (p_out[0] >= 0) close(p_out[0]);
1240 if (p_out[1] >= 0) close(p_out[1]);
1241 if (p_err[0] >= 0) close(p_err[0]);
1242 if (p_err[1] >= 0) close(p_err[1]);
1243 release_job(job);
1244 }
1245
1246 /* All done except for some final tidying up. */
1247 dstr_release(&d);
1248 }
1249
1250 /* Take care of all of the jobs until they're all done. */
1251 static void run_jobs(void)
1252 {
1253 struct job *job, *next, **link;
1254 int nfd;
1255 fd_set fd_in;
1256
1257 for (;;) {
1258
1259 /* If there are jobs still to be started and we have slots to spare then
1260 * start some more up.
1261 */
1262 start_jobs();
1263
1264 /* If the queues are now all empty then we're done. (No need to check
1265 * `job_ready' here: `start_jobs' would have started them if `job_run'
1266 * was empty.
1267 */
1268 if (!job_run) break;
1269
1270 /* Prepare for the select(2) call: watch for the signal pipe and all of
1271 * the job pipes.
1272 */
1273 #define SET_FD(dir, fd) do { \
1274 int _fd = (fd); \
1275 FD_SET(_fd, &fd_##dir); \
1276 if (_fd >= nfd) nfd = _fd + 1; \
1277 } while (0)
1278
1279 FD_ZERO(&fd_in); nfd = 0;
1280 SET_FD(in, sig_pipe[0]);
1281 for (job = job_run; job; job = job->next) {
1282 if (job->out.fd >= 0) SET_FD(in, job->out.fd);
1283 if (job->err.fd >= 0) SET_FD(in, job->err.fd);
1284 }
1285
1286 #undef SET_FD
1287
1288 /* Find out what's going on. */
1289 if (select(nfd, &fd_in, 0, 0, 0) < 0) {
1290 if (errno == EINTR) continue;
1291 else lose("select failed: %s", strerror(errno));
1292 }
1293
1294 /* If there were any signals then handle them. */
1295 if (FD_ISSET(sig_pipe[0], &fd_in)) {
1296 check_signals();
1297 if (sigloss >= 0) {
1298 /* We hit a fatal signal. Kill off the remaining jobs and abort. */
1299 for (job = job_ready; job; job = next)
1300 { next = job->next; release_job(job); }
1301 for (job = job_run; job; job = next)
1302 { next = job->next; release_job(job); }
1303 break;
1304 }
1305 }
1306
1307 /* Collect output from running jobs, and clear away any dead jobs once
1308 * we've collected all their output.
1309 */
1310 for (link = &job_run, job = *link; job; job = next) {
1311 if (job->out.fd >= 0 && FD_ISSET(job->out.fd, &fd_in))
1312 prefix_lines(job, &job->out, '|',
1313 job->st == JST_VERSION ? &job->h : 0);
1314 if (job->err.fd >= 0 && FD_ISSET(job->err.fd, &fd_in))
1315 prefix_lines(job, &job->err, '*', 0);
1316 next = job->next;
1317 if (job->kid > 0 || job->out.fd >= 0 || job->err.fd >= 0)
1318 link = &job->next;
1319 else
1320 { *link = next; finish_job(job); }
1321 }
1322 }
1323 }
1324
1325 /*----- Main program ------------------------------------------------------*/
1326
1327 /* Help and related functions. */
1328 static void version(FILE *fp)
1329 { fprintf(fp, "%s, runlisp version %s\n", progname, PACKAGE_VERSION); }
1330
1331 static void usage(FILE *fp)
1332 {
1333 fprintf(fp, "\
1334 usage: %s [-RUafinqrv] [+RUfinr] [-c CONF] [-o [SECT:]VAR=VAL]\n\
1335 [-O FILE|DIR] [-j NJOBS] [LISP ...]\n",
1336 progname);
1337 }
1338
1339 static void help(FILE *fp)
1340 {
1341 version(fp); fputc('\n', fp); usage(fp);
1342 fputs("\n\
1343 Help options:\n\
1344 -h, --help Show this help text and exit successfully.\n\
1345 -V, --version Show version number and exit successfully.\n\
1346 \n\
1347 Diagnostics:\n\
1348 -n, --dry-run Don't run run anything (useful with `-v').\n\
1349 -q, --quiet Don't print warning messages.\n\
1350 -v, --verbose Print informational messages (repeatable).\n\
1351 \n\
1352 Configuration:\n\
1353 -c, --config-file=CONF Read configuration from CONF (repeatable).\n\
1354 -o, --set-option=[SECT:]VAR=VAL Set configuration variable (repeatable).\n\
1355 \n\
1356 Image dumping:\n\
1357 -O, --output=FILE|DIR Store image(s) in FILE or DIR.\n\
1358 -R, --remove-other Delete image files for other Lisp systems.\n\
1359 -U, --remove-unknown Delete unrecognized files in image dir.\n\
1360 -a, --all-configured Select all configured implementations.\n\
1361 -f, --force Dump images even if they already exist.\n\
1362 -i, --check-installed Check Lisp systems exist before dumping.\n\
1363 -j, --jobs=NJOBS Run up to NJOBS jobs in parallel.\n\
1364 -r, --remove-image Delete image files, instead of creating.\n",
1365 fp);
1366 }
1367
1368 static void show_job_list(const char *what, struct job *job)
1369 {
1370 struct dstr d = DSTR_INIT;
1371 int first;
1372
1373 first = 1;
1374 for (; job; job = job->next) {
1375 if (first) first = 0;
1376 else dstr_puts(&d, ", ");
1377 dstr_putf(&d, "`%s'", JOB_NAME(job));
1378 }
1379 if (first) dstr_puts(&d, "(none)");
1380 dstr_putz(&d);
1381 moan("%s: %s", what, d.p);
1382 }
1383
1384 /* Main program. */
1385 int main(int argc, char *argv[])
1386 {
1387 struct config_section_iter si;
1388 struct config_section *sect;
1389 struct config_var *var;
1390 const char *out = 0, *p, *q, *l;
1391 struct job *job;
1392 struct stat st;
1393 struct dstr d = DSTR_INIT;
1394 DIR *dir;
1395 struct dirent *de;
1396 int i, fd;
1397 size_t n, o;
1398 unsigned f;
1399
1400 /* Command-line options. */
1401 static const struct option opts[] = {
1402 { "help", 0, 0, 'h' },
1403 { "version", 0, 0, 'V' },
1404 { "output", OPTF_ARGREQ, 0, 'O' },
1405 { "remove-other", OPTF_NEGATE, 0, 'R' },
1406 { "remove-unknown", OPTF_NEGATE, 0, 'U' },
1407 { "all-configured", 0, 0, 'a' },
1408 { "config-file", OPTF_ARGREQ, 0, 'c' },
1409 { "force", OPTF_NEGATE, 0, 'f' },
1410 { "check-installed", OPTF_NEGATE, 0, 'i' },
1411 { "jobs", OPTF_ARGREQ, 0, 'j' },
1412 { "dry-run", OPTF_NEGATE, 0, 'n' },
1413 { "set-option", OPTF_ARGREQ, 0, 'o' },
1414 { "quiet", 0, 0, 'q' },
1415 { "remove-image", OPTF_NEGATE, 0, 'r' },
1416 { "verbose", 0, 0, 'v' },
1417 { 0, 0, 0, 0 }
1418 };
1419
1420 /* Initial setup. */
1421 set_progname(argv[0]);
1422 init_config();
1423 srand(time(0));
1424
1425 /* Parse the options. */
1426 optprog = (/*unconst*/ char *)progname;
1427
1428 #define FLAGOPT(ch, f) \
1429 case ch: \
1430 flags |= f; \
1431 break; \
1432 case ch | OPTF_NEGATED: \
1433 flags &= ~f; \
1434 break
1435
1436 for (;;) {
1437 i = mdwopt(argc - 1, argv + 1, "hVO:R+U+ac:f+i+j:n+o:qr+v", opts, 0, 0,
1438 OPTF_NEGATION | OPTF_NOPROGNAME);
1439 if (i < 0) break;
1440 switch (i) {
1441 case 'h': help(stdout); exit(0);
1442 case 'V': version(stdout); exit(0);
1443 case 'O': out = optarg; break;
1444 FLAGOPT('R', AF_CLEAN);
1445 FLAGOPT('U', AF_JUNK);
1446 case 'a': flags |= AF_ALL; break;
1447 case 'c': read_config_path(optarg, 0); flags |= AF_SETCONF; break;
1448 FLAGOPT('f', AF_FORCE);
1449 FLAGOPT('i', AF_CHECKINST);
1450 case 'j': maxrun = parse_int("number of jobs", optarg, 1, 65535); break;
1451 FLAGOPT('n', AF_DRYRUN);
1452 case 'o': if (set_config_var(optarg)) flags |= AF_BOGUS; break;
1453 case 'q': if (verbose) verbose--; break;
1454 FLAGOPT('r', AF_REMOVE);
1455 case 'v': verbose++; break;
1456 default: flags |= AF_BOGUS; break;
1457 }
1458 }
1459
1460 #undef FLAGOPT
1461
1462 /* CHeck that everything worked. */
1463 optind++;
1464 if ((flags&AF_ALL) ? optind < argc : optind >= argc) flags |= AF_BOGUS;
1465 if (flags&AF_BOGUS) { usage(stderr); exit(127); }
1466
1467 /* Load default configuration if no explicit files were requested. */
1468 if (!(flags&AF_SETCONF)) load_default_config();
1469
1470 /* OK, so we've probably got some work to do. Let's set things up ready.
1471 * It'll be annoying if our standard descriptors aren't actually set up
1472 * properly, so we'll make sure those slots are populated. We'll need a
1473 * `/dev/null' descriptor anyway (to be stdin for the jobs). We'll also
1474 * need a temporary directory, and it'll be less temporary if we don't
1475 * arrange to delete it when we're done. And finally we'll need to know
1476 * when a child process exits.
1477 */
1478 for (;;) {
1479 fd = open("/dev/null", O_RDWR);
1480 if (fd < 0) lose("failed to open `/dev/null': %s", strerror(errno));
1481 if (fd > 2) { nullfd = fd; break; }
1482 }
1483 configure_fd("null fd", nullfd, 0, 1);
1484 atexit(cleanup);
1485 if (pipe(sig_pipe))
1486 lose("failed to create signal pipe: %s", strerror(errno));
1487 configure_fd("signal pipe (read end)", sig_pipe[0], 1, 1);
1488 configure_fd("signal pipe (write end)", sig_pipe[1], 1, 1);
1489 sigemptyset(&caught); sigemptyset(&pending);
1490 set_signal_handler("SIGTERM", SIGTERM, SIGF_IGNOK);
1491 set_signal_handler("SIGINT", SIGINT, SIGF_IGNOK);
1492 set_signal_handler("SIGHUP", SIGHUP, SIGF_IGNOK);
1493 set_signal_handler("SIGCHLD", SIGCHLD, 0);
1494
1495 /* Create the temporary directory and export it into the configuration. */
1496 set_tmpdir();
1497 config_set_var(&config, builtin, CF_LITERAL, "@%tmp-dir", tmpdir);
1498 config_set_var(&config, builtin, 0,
1499 "@tmp-dir", "${@BUILTIN:@%tmp-dir}/${@name}");
1500
1501 /* Work out where the image files are going to go. If there's no `-O'
1502 * option then we use the main `image-dir'. Otherwise what happens depends
1503 * on whether this is a file or a directory.
1504 */
1505 if (!out) {
1506 config_set_var(&config, builtin, 0,
1507 "@image-link", "${@image-dir}/${image-file}");
1508 var = config_find_var(&config, builtin, CF_INHERIT, "@image-dir");
1509 assert(var); out = config_subst_var_alloc(&config, builtin, var);
1510 } else if (!stat(out, &st) && S_ISDIR(st.st_mode)) {
1511 config_set_var(&config, builtin, CF_LITERAL, "@%out-dir", out);
1512 config_set_var(&config, builtin, 0,
1513 "@image-link", "${@BUILTIN:@%out-dir}/${image-file}");
1514 } else if (argc - optind != 1)
1515 lose("can't dump multiple Lisps to a single output file");
1516 else if (flags&AF_JUNK)
1517 lose("can't clear junk in a single output file");
1518 else if (flags&AF_CLEAN)
1519 lose("can't clean other images with a single output file");
1520 else
1521 config_set_var(&config, builtin, CF_LITERAL, "@image-link", out);
1522
1523 /* Set the staging and versioned filenames. */
1524 config_set_var(&config, builtin, 0,
1525 "@image-out", "${@image-link}-${@hash}");
1526 config_set_var(&config, builtin, 0, "@image-new", "${@image-out}.new");
1527 config_set_var(&config, builtin, 0,
1528 "@image-newlink", "${@image-link}.new");
1529
1530 config_set_var(&config, builtin, 0, "@script",
1531 "${@ENV:RUNLISP_EVAL?"
1532 "${@CONFIG:eval-script?"
1533 "${@data-dir}/eval.lisp}}");
1534
1535 /* Configure an initial value for `@hash'. This is necessary so that
1536 * `add_job' can expand `dump-image' to check that the command exists.
1537 */
1538 config_set_var(&config, builtin, CF_LITERAL, "@hash", "!!!unset!!!");
1539
1540 /* Dump the final configuration if we're being very verbose. */
1541 if (verbose >= 5) dump_config();
1542
1543 /* There are a number of different strategies we might employ, depending on
1544 * the exact request.
1545 *
1546 * queue queue clear
1547 * REMOVE CLEAN JUNK selected others junk?
1548 *
1549 * * nil nil ready/delete -- no
1550 * * nil t ready/delete none yes
1551 * nil t nil ready delete no
1552 * nil t t ready -- yes
1553 * t t nil -- delete no
1554 * t t t -- -- yes
1555 */
1556
1557 /* First step: if `AF_REMOVE' and `AF_CLEAN' are not both set, then scan
1558 * the selected Lisp systems and add them to the appropriate queue.
1559 *
1560 * Bit-hack: if they are not both set, then their complements are not both
1561 * clear.
1562 */
1563 if (~flags&(AF_REMOVE | AF_CLEAN)) {
1564
1565 /* Determine the flags for `add_job' when we select the Lisp systems. If
1566 * we intend to clear junk then we must notice the image names we
1567 * encounter. If we're supposed to check that Lisps exist before dumping
1568 * then do that -- but it doesn't make any sense for deletion.
1569 */
1570 f = flags&AF_REMOVE ? JQ_DELETE : JQ_READY;
1571 if (flags&AF_JUNK) f |= JF_NOTICE;
1572 if (flags&AF_CHECKINST) f |= JF_CHECKINST;
1573 if (!(flags&(AF_FORCE | AF_REMOVE))) f |= JF_CHECKEXIST;
1574
1575 /* If we have named Lisps, then process them. */
1576 if (!(flags&AF_ALL))
1577 for (i = optind; i < argc; i++)
1578 add_named_job(f, argv[i], strlen(argv[i]));
1579
1580 /* Otherwise we're supposed to dump `all' of them. If there's a `dump'
1581 * configuration setting then we need to parse that. Otherwise we just
1582 * try all of them.
1583 */
1584 else {
1585 var = config_find_var(&config, toplevel, CF_INHERIT, "dump");
1586 if (!var) {
1587 /* No setting. Just do all of the Lisps which look available. */
1588
1589 f |= JF_CHECKINST;
1590 for (config_start_section_iter(&config, &si);
1591 (sect = config_next_section(&si)); )
1592 add_job(f, sect);
1593 } else {
1594 /* Parse the `dump' list. */
1595
1596 dstr_reset(&d); config_subst_var(&config, toplevel, var, &d);
1597 p = d.p; l = p + d.len;
1598 for (;;) {
1599 while (p < l && ISSPACE(*p)) p++;
1600 if (p >= l) break;
1601 q = p;
1602 while (p < l && !ISSPACE(*p) && *p != ',') p++;
1603 add_named_job(f, q, p - q);
1604 while (p < l && ISSPACE(*p)) p++;
1605 if (p < l && *p == ',') p++;
1606 }
1607 }
1608 }
1609 }
1610
1611 /* Second step: if exactly one of `AF_CLEAN' and `AF_JUNK' is set, then we
1612 * need to scan all of the remaining Lisps and add them to the `delete'
1613 * queue.
1614 */
1615 if (!(flags&AF_CLEAN) != !(flags&AF_JUNK)) {
1616
1617 /* Determine the flag settings. If we're junking, then we're not
1618 * cleaning -- we just want to mark images belonging to other Lisps as
1619 * off-limits to the junking scan.
1620 */
1621 f = flags&AF_CLEAN ? JQ_DELETE : JQ_NONE | JF_NOTICE;
1622
1623 /* Now scan the Lisp systems. */
1624 for (config_start_section_iter(&config, &si);
1625 (sect = config_next_section(&si)); )
1626 add_job(f, sect);
1627 }
1628
1629 /* Terminate the job queues. */
1630 *job_ready_tail = 0;
1631 *job_delete_tail = 0;
1632
1633 /* Report on what it is we're about to do. */
1634 if (verbose >= 3) {
1635 show_job_list("dumping Lisp images", job_ready);
1636 show_job_list("deleting Lisp images", job_delete);
1637 }
1638
1639 /* If there turns out to be nothing to do, then mention this. */
1640 if (!(flags&AF_REMOVE) && verbose >= 2 && !job_ready)
1641 moan("no Lisp images to dump");
1642
1643 /* Run the dumping jobs. */
1644 run_jobs();
1645
1646 /* Check for any last signals. If we hit any fatal signals then we should
1647 * kill ourselves so that the exit status will be right.
1648 */
1649 check_signals();
1650 if (sigloss) { cleanup(); signal(sigloss, SIG_DFL); raise(sigloss); }
1651
1652 /* Now delete Lisps which need deleting. */
1653 while (job_delete) {
1654 job = job_delete; job_delete = job->next;
1655 if (flags&AF_DRYRUN) {
1656 if (verbose >= 2)
1657 moan("not deleting `%s' image link `%s' (dry run)",
1658 JOB_NAME(job), job->imglink);
1659 if (job->oldimg && verbose >= 2)
1660 moan("not deleting `%s' image `%s' (dry run)",
1661 JOB_NAME(job), job->oldimg);
1662 } else {
1663 if (verbose >= 2)
1664 moan("deleting `%s' image `%s'",
1665 JOB_NAME(job), job->imglink);
1666 if (unlink(job->imglink) && errno != ENOENT)
1667 bad("failed to delete `%s' image link `%s': %s",
1668 JOB_NAME(job), job->imglink, strerror(errno));
1669 if (job->oldimg && unlink(job->oldimg) && errno != ENOENT)
1670 bad("failed to delete `%s' image `%s': %s",
1671 JOB_NAME(job), job->oldimg, strerror(errno));
1672 }
1673 }
1674
1675 /* Finally, maybe delete all of the junk files in the image directory. */
1676 if (flags&AF_JUNK) {
1677 dir = opendir(out);
1678 if (!dir)
1679 lose("failed to open image directory `%s': %s", out, strerror(errno));
1680 dstr_reset(&d);
1681 dstr_puts(&d, out); dstr_putc(&d, '/'); o = d.len;
1682 if (verbose >= 2)
1683 moan("cleaning up junk in image directory `%s'", out);
1684 for (;;) {
1685 de = readdir(dir); if (!de) break;
1686 if (de->d_name[0] == '.' &&
1687 (!de->d_name[1] || (de->d_name[1] == '.' && !de->d_name[2])))
1688 continue;
1689 n = strlen(de->d_name);
1690 d.len = o; dstr_putm(&d, de->d_name, n + 1);
1691 if (!treap_lookup(&good, de->d_name, n)) {
1692 if (flags&AF_DRYRUN) {
1693 if (verbose >= 2)
1694 moan("not deleting junk file `%s' (dry run)", d.p);
1695 } else {
1696 if (verbose >= 2)
1697 moan("deleting junk file `%s'", d.p);
1698 if (unlink(d.p) && errno != ENOENT)
1699 bad("failed to delete junk file `%s': %s", d.p, strerror(errno));
1700 }
1701 }
1702 }
1703 }
1704
1705 /* All done! */
1706 return (rc);
1707 }
1708
1709 /*----- That's all, folks -------------------------------------------------*/