diff options
Diffstat (limited to 'progress.c')
| -rw-r--r-- | progress.c | 291 |
1 files changed, 291 insertions, 0 deletions
diff --git a/progress.c b/progress.c new file mode 100644 index 0000000..a921ac9 --- /dev/null +++ b/progress.c @@ -0,0 +1,291 @@ +/* $NetBSD: progress.c,v 1.23 2021/01/07 12:02:52 lukem Exp $ */ + +/*- + * Copyright (c) 2003 The NetBSD Foundation, Inc. + * All rights reserved. + * + * This code is derived from software contributed to The NetBSD Foundation + * by John Hawkinson. + * + * Redistribution and use in source and binary forms, with or without + * modification, are permitted provided that the following conditions + * are met: + * 1. Redistributions of source code must retain the above copyright + * notice, this list of conditions and the following disclaimer. + * 2. Redistributions in binary form must reproduce the above copyright + * notice, this list of conditions and the following disclaimer in the + * documentation and/or other materials provided with the distribution. + * + * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS + * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED + * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR + * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS + * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR + * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF + * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS + * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN + * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) + * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE + * POSSIBILITY OF SUCH DAMAGE. + */ + +#define _GNU_SOURCE + +#include <sys/cdefs.h> +#ifndef lint +__RCSID("$NetBSD: progress.c,v 1.23 2021/01/07 12:02:52 lukem Exp $"); +#endif /* not lint */ + +#include <sys/types.h> +#include <sys/ioctl.h> +#include <sys/stat.h> +#include <sys/wait.h> + +#include <err.h> +#include <errno.h> +#include <fcntl.h> +#include <inttypes.h> +#include <limits.h> +#include <signal.h> +#include <stdio.h> +#include <stdlib.h> +#include <string.h> +#include <unistd.h> + + /* LONGLONG */ +long long strsuftoll(const char *, const char *, long long, long long); + /* LONGLONG */ +long long strsuftollx(const char *, const char *, long long, long long, + char *, size_t); + +#define GLOBAL /* force GLOBAL decls in progressbar.h to be + * declared */ +#include "progressbar.h" + +static void broken_pipe(int unused); +static void usage(void); + +static void +broken_pipe(int unused) +{ + signal(SIGPIPE, SIG_DFL); + progressmeter(1); + kill(getpid(), SIGPIPE); +} + +static void +usage(void) +{ + fprintf(stderr, + "usage: %s [-ez] [-b buffersize] [-f file] [-l length]\n" + " %*.s [-p prefix] cmd [args...]\n", + getprogname(), (int) strlen(getprogname()), ""); + exit(EXIT_FAILURE); +} + +int +main(int argc, char *argv[]) +{ + char *fb_buf; + char *infile = NULL; + pid_t pid = 0, gzippid = 0, deadpid; + int ch, fd, outpipe[2]; + int ws, gzipstat, cmdstat; + int eflag = 0, lflag = 0, zflag = 0; + ssize_t nr, nw, off; + size_t buffersize; + struct stat statb; + struct winsize wins; + + setprogname(argv[0]); + + /* defaults: Read from stdin, 0 filesize (no completion estimate) */ + fd = STDIN_FILENO; + filesize = 0; + buffersize = 64 * 1024; + prefix = NULL; + + while ((ch = getopt(argc, argv, "b:ef:l:p:z")) != -1) + switch (ch) { + case 'b': + buffersize = (size_t) strsuftoll("buffer size", optarg, + 0, SSIZE_MAX); + break; + case 'e': + eflag++; + break; + case 'f': + infile = optarg; + break; + case 'l': + lflag++; + filesize = strsuftoll("input size", optarg, 0, + LLONG_MAX); + break; + case 'p': + prefix = optarg; + break; + case 'z': + zflag++; + break; + case '?': + default: + usage(); + /* NOTREACHED */ + } + argc -= optind; + argv += optind; + + if (argc < 1) + usage(); + + if (infile && (fd = open(infile, O_RDONLY, 0)) < 0) + err(1, "%s", infile); + + /* stat() to get the filesize unless overridden, or -z */ + if (!zflag && !lflag && (fstat(fd, &statb) == 0)) { + if (S_ISFIFO(statb.st_mode)) { + /* stat(2) on pipe may return only the + * first few bytes with more coming. + * Don't trust! + */ + } else { + filesize = statb.st_size; + } + } + + /* gzip -l the file if we have the name and -z is given */ + if (zflag && !lflag && infile != NULL) { + FILE *gzipsizepipe; + char buf[256], *cp, *cmd; + + /* + * Read second word of last line of gzip -l output. Looks like: + * % gzip -l ../etc.tgz + * compressed uncompressed ratio uncompressed_name + * 119737 696320 82.8% ../etc.tar + */ + + asprintf(&cmd, "gzip -l %s", infile); + if ((gzipsizepipe = popen(cmd, "r")) == NULL) + err(1, "reading compressed file length"); + for (; fgets(buf, 256, gzipsizepipe) != NULL;) + continue; + strtoimax(buf, &cp, 10); + filesize = strtoimax(cp, NULL, 10); + if (pclose(gzipsizepipe) < 0) + err(1, "closing compressed file length pipe"); + free(cmd); + } + /* Pipe input through gzip -dc if -z is given */ + if (zflag) { + int gzippipe[2]; + + if (pipe(gzippipe) < 0) + err(1, "gzip pipe"); + gzippid = fork(); + if (gzippid < 0) + err(1, "fork for gzip"); + + if (gzippid) { + /* parent */ + dup2(gzippipe[0], fd); + close(gzippipe[0]); + close(gzippipe[1]); + } else { + dup2(gzippipe[1], STDOUT_FILENO); + dup2(fd, STDIN_FILENO); + close(gzippipe[0]); + close(gzippipe[1]); + if (execlp("gzip", "gzip", "-dc", NULL)) + err(1, "exec()ing gzip"); + } + } + + /* Initialize progressbar.c's global state */ + bytes = 0; + progress = 1; + ttyout = eflag ? stderr : stdout; + + if (ioctl(fileno(ttyout), TIOCGWINSZ, &wins) == -1) + ttywidth = 80; + else + ttywidth = wins.ws_col; + + fb_buf = malloc(buffersize); + if (fb_buf == NULL) + err(1, "malloc for buffersize"); + + if (pipe(outpipe) < 0) + err(1, "output pipe"); + pid = fork(); + if (pid < 0) + err(1, "fork for output pipe"); + + if (pid == 0) { + /* child */ + dup2(outpipe[0], STDIN_FILENO); + close(outpipe[0]); + close(outpipe[1]); + execvp(argv[0], argv); + err(1, "could not exec %s", argv[0]); + } + close(outpipe[0]); + + signal(SIGPIPE, broken_pipe); + progressmeter(-1); + + while (1) { + do { + nr = read(fd, fb_buf, buffersize); + } while (nr < 0 && errno == EINTR); + if (nr <= 0) + break; + for (off = 0; nr; nr -= nw, off += nw, bytes += nw) + if ((nw = write(outpipe[1], fb_buf + off, + (size_t) nr)) < 0) { + if (errno == EINTR) { + nw = 0; + continue; + } + progressmeter(1); + err(1, "writing %u bytes to output pipe", + (unsigned) nr); + } + } + close(outpipe[1]); + + gzipstat = 0; + cmdstat = 0; + while (pid || gzippid) { + deadpid = wait(&ws); + /* + * We need to exit with an error if the command (or gzip) + * exited abnormally. + * Unfortunately we can't generate a true 'exited by signal' + * error without sending the signal to ourselves :-( + */ + ws = WIFSIGNALED(ws) ? WTERMSIG(ws) : WEXITSTATUS(ws); + + if (deadpid != -1 && errno == EINTR) + continue; + if (deadpid == pid) { + pid = 0; + cmdstat = ws; + continue; + } + if (deadpid == gzippid) { + gzippid = 0; + gzipstat = ws; + continue; + } + break; + } + + progressmeter(1); + signal(SIGPIPE, SIG_DFL); + + free(fb_buf); + + exit(cmdstat ? cmdstat : gzipstat); +} |