NetBSD-5.0.2/usr.bin/progress/progress.c
/* $NetBSD: progress.c,v 1.17 2008/05/26 04:53:11 dholland 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.
*/
#include <sys/cdefs.h>
#ifndef lint
__RCSID("$NetBSD: progress.c,v 1.17 2008/05/26 04:53:11 dholland Exp $");
#endif /* not lint */
#include <sys/types.h>
#include <sys/param.h>
#include <sys/socket.h>
#include <sys/ioctl.h>
#include <sys/time.h>
#include <sys/wait.h>
#include <netinet/in.h>
#include <arpa/ftp.h>
#include <err.h>
#include <errno.h>
#include <fcntl.h>
#include <glob.h>
#include <signal.h>
#include <inttypes.h>
#include <limits.h>
#include <netdb.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <termios.h>
#include <time.h>
#include <tzfile.h>
#include <unistd.h>
#define GLOBAL /* force GLOBAL decls in progressbar.h to be
* declared */
#include "progressbar.h"
static void broken_pipe(int unused);
static void usage(void);
int main(int, char *[]);
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 ttysize ts;
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, SIZE_T_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), TIOCGSIZE, &ts) == -1)
ttywidth = 80;
else
ttywidth = ts.ts_cols;
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 ((nr = read(fd, fb_buf, buffersize)) > 0)
for (off = 0; nr; nr -= nw, off += nw, bytes += nw)
if ((nw = write(outpipe[1], fb_buf + off,
(size_t) nr)) < 0) {
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);
}