#ifndef lint
#if 0
static char sccsid[] = "@(#)ar_io.c 8.2 (Berkeley) 4/18/94";
#else
static char rcsid[] __attribute__((__unused__)) = "$OpenBSD: ar_io.c,v 1.17 1997/09/01 18:29:42 deraadt Exp $";
#endif
#endif
#include <sys/types.h>
#include <sys/time.h>
#include <sys/stat.h>
#include <sys/ioctl.h>
#include <sys/mtio.h>
#include <sys/param.h>
#include <signal.h>
#include <string.h>
#include <fcntl.h>
#include <unistd.h>
#include <stdio.h>
#include <errno.h>
#include <stdlib.h>
#include <err.h>
#include "pax.h"
#include "options.h"
#include "extern.h"
#define DMOD 0666
#define EXT_MODE O_RDONLY
#define AR_MODE (O_WRONLY | O_CREAT | O_TRUNC)
#define APP_MODE O_RDWR
#define STDO "<STDOUT>"
#define STDN "<STDIN>"
static int arfd = -1;
static int artyp = ISREG;
static int arvol = 1;
static int lstrval = -1;
static int io_ok;
static int did_io;
static int done;
static struct stat arsb;
static int invld_rec;
static int wr_trail = 1;
static int can_unlnk = 0;
char *arcname;
char *gzip_program;
static int get_phys __P((void));
extern sigset_t s_mask;
static void ar_start_gzip __P((int));
#ifdef __STDC__
int
ar_open(char *name)
#else
int
ar_open(name)
char *name;
#endif
{
struct mtget mb;
if (arfd != -1)
(void)close(arfd);
arfd = -1;
can_unlnk = did_io = io_ok = invld_rec = 0;
artyp = ISREG;
flcnt = 0;
switch (act) {
case LIST:
case EXTRACT:
if (name == NULL) {
arfd = STDIN_FILENO;
arcname = STDN;
} else if ((arfd = open(name, EXT_MODE, DMOD)) < 0) {
syswarn(0, errno, "Failed open to read on %s", name);
exit(1);
}
if (zflag)
ar_start_gzip(arfd);
break;
case ARCHIVE:
if (name == NULL) {
arfd = STDOUT_FILENO;
arcname = STDO;
} else if ((arfd = open(name, AR_MODE, DMOD)) < 0)
syswarn(0, errno, "Failed open to write on %s", name);
else
can_unlnk = 1;
if (zflag)
ar_start_gzip(arfd);
break;
case APPND:
if (zflag)
err(1, "can not gzip while appending");
if (name == NULL) {
arfd = STDOUT_FILENO;
arcname = STDO;
} else if ((arfd = open(name, APP_MODE, DMOD)) < 0)
syswarn(0, errno, "Failed open to read/write on %s",
name);
break;
case COPY:
arcname = "<NONE>";
lstrval = 1;
return(0);
}
if (arfd < 0)
return(-1);
if (chdname != NULL)
if (chdir(chdname) != 0)
syswarn(1, errno, "Failed chdir to %s", chdname);
if (fstat(arfd, &arsb) < 0) {
syswarn(0, errno, "Failed stat on %s", arcname);
(void)close(arfd);
arfd = -1;
can_unlnk = 0;
return(-1);
}
if (S_ISDIR(arsb.st_mode)) {
paxwarn(0, "Cannot write an archive on top of a directory %s",
arcname);
(void)close(arfd);
arfd = -1;
can_unlnk = 0;
return(-1);
}
if (S_ISCHR(arsb.st_mode))
artyp = ioctl(arfd, MTIOCGET, &mb) ? ISCHR : ISTAPE;
else if (S_ISBLK(arsb.st_mode))
artyp = ISBLK;
else if ((lseek(arfd, (off_t)0L, SEEK_CUR) == -1) && (errno == ESPIPE))
artyp = ISPIPE;
else
artyp = ISREG;
if (artyp != ISREG)
can_unlnk = 0;
if (act == ARCHIVE) {
blksz = rdblksz = wrblksz;
lstrval = 1;
return(0);
}
switch(artyp) {
case ISTAPE:
blksz = rdblksz = MAXBLK;
break;
case ISPIPE:
case ISBLK:
case ISCHR:
if ((act == APPND) && wrblksz) {
blksz = rdblksz = wrblksz;
break;
}
if ((arsb.st_blksize > 0) && (arsb.st_blksize < MAXBLK) &&
((arsb.st_blksize % BLKMULT) == 0))
rdblksz = arsb.st_blksize;
else
rdblksz = DEVBLK;
if ((act == APPND) || (artyp == ISCHR))
blksz = rdblksz;
else
blksz = MAXBLK;
break;
case ISREG:
if ((act == APPND) && wrblksz && ((arsb.st_size%wrblksz)==0)){
blksz = rdblksz = wrblksz;
break;
}
for (rdblksz = MAXBLK; rdblksz > 0; rdblksz -= BLKMULT)
if ((arsb.st_size % rdblksz) == 0)
break;
if (rdblksz <= 0)
rdblksz = FILEBLK;
if (act == APPND)
blksz = rdblksz;
else
blksz = MAXBLK;
break;
default:
blksz = rdblksz = BLKMULT;
break;
}
lstrval = 1;
return(0);
}
#ifdef __STDC__
void
ar_close(void)
#else
void
ar_close()
#endif
{
FILE *outf;
if (arfd < 0) {
did_io = io_ok = flcnt = 0;
return;
}
if (act == LIST)
outf = stdout;
else
outf = stderr;
if (vflag && (artyp == ISTAPE)) {
if (vfpart)
(void)putc('\n', outf);
(void)fprintf(outf,
"%s: Waiting for tape drive close to complete...",
argv0);
(void)fflush(outf);
}
if (can_unlnk && (fstat(arfd, &arsb) == 0) && (S_ISREG(arsb.st_mode)) &&
(arsb.st_size == 0)) {
(void)unlink(arcname);
can_unlnk = 0;
}
(void)close(arfd);
if (vflag && (artyp == ISTAPE)) {
(void)fputs("done.\n", outf);
vfpart = 0;
(void)fflush(outf);
}
arfd = -1;
if (!io_ok && !did_io) {
flcnt = 0;
return;
}
did_io = io_ok = 0;
if (frmt != NULL)
++arvol;
if (!vflag) {
flcnt = 0;
return;
}
if (vfpart) {
(void)putc('\n', outf);
vfpart = 0;
}
if (frmt == NULL) {
# ifdef NET2_STAT
(void)fprintf(outf, "%s: unknown format, %lu bytes skipped.\n",
# else
(void)fprintf(outf, "%s: unknown format, %qu bytes skipped.\n",
# endif
argv0, rdcnt);
(void)fflush(outf);
flcnt = 0;
return;
}
if (strcmp(NM_CPIO, argv0) == 0)
(void)fprintf(outf, "%qu blocks\n", (rdcnt ? rdcnt : wrcnt) / 5120);
else if (strcmp(NM_TAR, argv0) != 0)
(void)fprintf(outf,
# ifdef NET2_STAT
"%s: %s vol %d, %lu files, %lu bytes read, %lu bytes written.\n",
# else
"%s: %s vol %d, %lu files, %qu bytes read, %qu bytes written.\n",
# endif
argv0, frmt->name, arvol-1, flcnt, rdcnt, wrcnt);
(void)fflush(outf);
flcnt = 0;
}
#ifdef __STDC__
void
ar_drain(void)
#else
void
ar_drain()
#endif
{
register int res;
char drbuf[MAXBLK];
if ((artyp != ISPIPE) || (lstrval <= 0))
return;
while ((res = read(arfd, drbuf, sizeof(drbuf))) > 0)
;
lstrval = res;
}
#ifdef __STDC__
int
ar_set_wr(void)
#else
int
ar_set_wr()
#endif
{
off_t cpos;
wr_trail = 0;
if (artyp != ISREG)
return(0);
if (((cpos = lseek(arfd, (off_t)0L, SEEK_CUR)) < 0) ||
(ftruncate(arfd, cpos) < 0)) {
syswarn(1, errno, "Unable to truncate archive file");
return(-1);
}
return(0);
}
#ifdef __STDC__
int
ar_app_ok(void)
#else
int
ar_app_ok()
#endif
{
if (artyp == ISPIPE) {
paxwarn(1, "Cannot append to an archive obtained from a pipe.");
return(-1);
}
if (!invld_rec)
return(0);
paxwarn(1,"Cannot append, device record size %d does not support %s spec",
rdblksz, argv0);
return(-1);
}
#ifdef __STDC__
int
ar_read(register char *buf, register int cnt)
#else
int
ar_read(buf, cnt)
register char *buf;
register int cnt;
#endif
{
register int res = 0;
if (lstrval <= 0)
return(lstrval);
switch (artyp) {
case ISTAPE:
if ((res = read(arfd, buf, cnt)) > 0) {
io_ok = 1;
if (res != rdblksz) {
rdblksz = res;
if (rdblksz % BLKMULT)
invld_rec = 1;
}
return(res);
}
break;
case ISREG:
case ISBLK:
case ISCHR:
case ISPIPE:
default:
if ((res = read(arfd, buf, cnt)) > 0) {
io_ok = 1;
return(res);
}
break;
}
lstrval = res;
if (res < 0)
syswarn(1, errno, "Failed read on archive volume %d", arvol);
else
paxwarn(0, "End of archive volume %d reached", arvol);
return(res);
}
#ifdef __STDC__
int
ar_write(register char *buf, register int bsz)
#else
int
ar_write(buf, bsz)
register char *buf;
register int bsz;
#endif
{
register int res;
off_t cpos;
if (lstrval <= 0)
return(lstrval);
if ((res = write(arfd, buf, bsz)) == bsz) {
wr_trail = 1;
io_ok = 1;
return(bsz);
} else if (res < 0 && artyp == ISPIPE && errno == EPIPE) {
wr_trail = 1;
io_ok = 1;
errno = 0;
arfd = open("/dev/null", AR_MODE, DMOD);
artyp = ISREG;
return bsz;
}
if (res < 0)
lstrval = res;
else
lstrval = 0;
switch (artyp) {
case ISREG:
if ((res > 0) && (res % BLKMULT)) {
if ((cpos = lseek(arfd, (off_t)0L, SEEK_CUR)) < 0)
break;
cpos -= (off_t)res;
if (ftruncate(arfd, cpos) < 0)
break;
res = lstrval = 0;
break;
}
if (res >= 0)
break;
if ((errno == ENOSPC) || (errno == EFBIG) || (errno == EDQUOT))
res = lstrval = 0;
break;
case ISTAPE:
case ISCHR:
case ISBLK:
if (res >= 0)
break;
if (errno == EACCES) {
paxwarn(0, "Write failed, archive is write protected.");
res = lstrval = 0;
return(0);
}
if ((errno == ENOSPC) || (errno == EIO) || (errno == ENXIO))
res = lstrval = 0;
break;
case ISPIPE:
default:
break;
}
if (res >= 0) {
if (res > 0)
wr_trail = 1;
io_ok = 1;
}
if (!wr_trail && (res <= 0)) {
paxwarn(1,"Unable to append, trailer re-write failed. Quitting.");
return(res);
}
if (res == 0)
paxwarn(0, "End of archive volume %d reached", arvol);
else if (res < 0)
syswarn(1, errno, "Failed write to archive volume: %d", arvol);
else if (!frmt->blkalgn || ((res % frmt->blkalgn) == 0))
paxwarn(0,"WARNING: partial archive write. Archive MAY BE FLAWED");
else
paxwarn(1,"WARNING: partial archive write. Archive IS FLAWED");
return(res);
}
#ifdef __STDC__
int
ar_rdsync(void)
#else
int
ar_rdsync()
#endif
{
long fsbz;
off_t cpos;
off_t mpos;
struct mtop mb;
if ((done > 0) || (lstrval == 0))
return(-1);
if ((act == APPND) || (act == ARCHIVE)) {
paxwarn(1, "Cannot allow updates to an archive with flaws.");
return(-1);
}
if (io_ok)
did_io = 1;
switch(artyp) {
case ISTAPE:
if (io_ok) {
io_ok = 0;
lstrval = 1;
break;
}
mb.mt_op = MTFSR;
mb.mt_count = 1;
if (ioctl(arfd, MTIOCTOP, &mb) < 0)
break;
lstrval = 1;
break;
case ISREG:
case ISCHR:
case ISBLK:
io_ok = 0;
if (((fsbz = arsb.st_blksize) <= 0) || (artyp != ISREG))
fsbz = BLKMULT;
if ((cpos = lseek(arfd, (off_t)0L, SEEK_CUR)) < 0)
break;
mpos = fsbz - (cpos % (off_t)fsbz);
if (lseek(arfd, mpos, SEEK_CUR) < 0)
break;
lstrval = 1;
break;
case ISPIPE:
default:
io_ok = 0;
break;
}
if (lstrval <= 0) {
paxwarn(1, "Unable to recover from an archive read failure.");
return(-1);
}
paxwarn(0, "Attempting to recover from an archive read failure.");
return(0);
}
#ifdef __STDC__
int
ar_fow(off_t sksz, off_t *skipped)
#else
int
ar_fow(sksz, skipped)
off_t sksz;
off_t *skipped;
#endif
{
off_t cpos;
off_t mpos;
*skipped = 0;
if (sksz <= 0)
return(0);
if (lstrval <= 0)
return(lstrval);
if (artyp != ISREG)
return(0);
if ((cpos = lseek(arfd, (off_t)0L, SEEK_CUR)) >= 0) {
if ((mpos = cpos + sksz) > arsb.st_size) {
*skipped = arsb.st_size - cpos;
mpos = arsb.st_size;
} else
*skipped = sksz;
if (lseek(arfd, mpos, SEEK_SET) >= 0)
return(0);
}
syswarn(1, errno, "Foward positioning operation on archive failed");
lstrval = -1;
return(-1);
}
#ifdef __STDC__
int
ar_rev(off_t sksz)
#else
int
ar_rev(sksz)
off_t sksz;
#endif
{
off_t cpos;
struct mtop mb;
register int phyblk;
if (lstrval < 0)
return(lstrval);
switch(artyp) {
case ISPIPE:
if (sksz <= 0)
break;
paxwarn(1, "Reverse positioning on pipes is not supported.");
lstrval = -1;
return(-1);
case ISREG:
case ISBLK:
case ISCHR:
default:
if (sksz <= 0)
break;
if ((cpos = lseek(arfd, (off_t)0L, SEEK_CUR)) < 0) {
syswarn(1, errno,
"Unable to obtain current archive byte offset");
lstrval = -1;
return(-1);
}
if ((cpos -= sksz) < (off_t)0L) {
if (arvol > 1) {
paxwarn(1,"Reverse position on previous volume.");
lstrval = -1;
return(-1);
}
cpos = (off_t)0L;
}
if (lseek(arfd, cpos, SEEK_SET) < 0) {
syswarn(1, errno, "Unable to seek archive backwards");
lstrval = -1;
return(-1);
}
break;
case ISTAPE:
if ((phyblk = get_phys()) <= 0) {
lstrval = -1;
return(-1);
}
rdblksz = phyblk;
if (sksz <= 0)
break;
if (sksz % phyblk) {
paxwarn(1,
"Tape drive unable to backspace requested amount");
lstrval = -1;
return(-1);
}
mb.mt_op = MTBSR;
mb.mt_count = sksz/phyblk;
if (ioctl(arfd, MTIOCTOP, &mb) < 0) {
syswarn(1,errno, "Unable to backspace tape %d blocks.",
mb.mt_count);
lstrval = -1;
return(-1);
}
break;
}
lstrval = 1;
return(0);
}
#ifdef __STDC__
static int
get_phys(void)
#else
static int
get_phys()
#endif
{
register int padsz = 0;
register int res;
register int phyblk;
struct mtop mb;
char scbuf[MAXBLK];
if (lstrval == 1) {
while ((res = read(arfd, scbuf, sizeof(scbuf))) > 0)
padsz += res;
if (res < 0) {
syswarn(1, errno, "Unable to locate tape filemark.");
return(-1);
}
}
mb.mt_op = MTBSF;
mb.mt_count = 1;
if (ioctl(arfd, MTIOCTOP, &mb) < 0) {
syswarn(1, errno, "Unable to backspace over tape filemark.");
return(-1);
}
mb.mt_op = MTBSR;
mb.mt_count = 1;
if (ioctl(arfd, MTIOCTOP, &mb) < 0) {
syswarn(1, errno, "Unable to backspace over last tape block.");
return(-1);
}
if ((phyblk = read(arfd, scbuf, sizeof(scbuf))) <= 0) {
syswarn(1, errno, "Cannot determine archive tape blocksize.");
return(-1);
}
while ((res = read(arfd, scbuf, sizeof(scbuf))) > 0)
;
if (res < 0) {
syswarn(1, errno, "Unable to locate tape filemark.");
return(-1);
}
mb.mt_op = MTBSF;
mb.mt_count = 1;
if (ioctl(arfd, MTIOCTOP, &mb) < 0) {
syswarn(1, errno, "Unable to backspace over tape filemark.");
return(-1);
}
lstrval = 1;
if (padsz == 0)
return(phyblk);
if (padsz % phyblk) {
paxwarn(1, "Tape drive unable to backspace requested amount");
return(-1);
}
mb.mt_op = MTBSR;
mb.mt_count = padsz/phyblk;
if (ioctl(arfd, MTIOCTOP, &mb) < 0) {
syswarn(1,errno,"Unable to backspace tape over %d pad blocks",
mb.mt_count);
return(-1);
}
return(phyblk);
}
#ifdef __STDC__
int
ar_next(void)
#else
int
ar_next()
#endif
{
char buf[PAXPATHLEN+2];
static int freeit = 0;
sigset_t o_mask;
if (sigprocmask(SIG_BLOCK, &s_mask, &o_mask) < 0)
syswarn(0, errno, "Unable to set signal mask");
ar_close();
if (sigprocmask(SIG_SETMASK, &o_mask, NULL) < 0)
syswarn(0, errno, "Unable to restore signal mask");
if (frmt == NULL || done || !wr_trail || strcmp(NM_TAR, argv0) == 0)
return(-1);
tty_prnt("\nATTENTION! %s archive volume change required.\n", argv0);
if (strcmp(arcname, STDO) && strcmp(arcname, STDN) && (artyp != ISREG)
&& (artyp != ISPIPE)) {
if (artyp == ISTAPE) {
tty_prnt("%s ready for archive tape volume: %d\n",
arcname, arvol);
tty_prnt("Load the NEXT TAPE on the tape drive");
} else {
tty_prnt("%s ready for archive volume: %d\n",
arcname, arvol);
tty_prnt("Load the NEXT STORAGE MEDIA (if required)");
}
if ((act == ARCHIVE) || (act == APPND))
tty_prnt(" and make sure it is WRITE ENABLED.\n");
else
tty_prnt("\n");
for(;;) {
tty_prnt("Type \"y\" to continue, \".\" to quit %s,",
argv0);
tty_prnt(" or \"s\" to switch to new device.\nIf you");
tty_prnt(" cannot change storage media, type \"s\"\n");
tty_prnt("Is the device ready and online? > ");
if ((tty_read(buf,sizeof(buf))<0) || !strcmp(buf,".")){
done = 1;
lstrval = -1;
tty_prnt("Quitting %s!\n", argv0);
vfpart = 0;
return(-1);
}
if ((buf[0] == '\0') || (buf[1] != '\0')) {
tty_prnt("%s unknown command, try again\n",buf);
continue;
}
switch (buf[0]) {
case 'y':
case 'Y':
if (ar_open(arcname) >= 0)
return(0);
tty_prnt("Cannot re-open %s, try again\n",
arcname);
continue;
case 's':
case 'S':
tty_prnt("Switching to a different archive\n");
break;
default:
tty_prnt("%s unknown command, try again\n",buf);
continue;
}
break;
}
} else
tty_prnt("Ready for archive volume: %d\n", arvol);
for (;;) {
tty_prnt("Input archive name or \".\" to quit %s.\n", argv0);
tty_prnt("Archive name > ");
if ((tty_read(buf, sizeof(buf)) < 0) || !strcmp(buf, ".")) {
done = 1;
lstrval = -1;
tty_prnt("Quitting %s!\n", argv0);
vfpart = 0;
return(-1);
}
if (buf[0] == '\0') {
tty_prnt("Empty file name, try again\n");
continue;
}
if (!strcmp(buf, "..")) {
tty_prnt("Illegal file name: .. try again\n");
continue;
}
if (strlen(buf) > PAXPATHLEN) {
tty_prnt("File name too long, try again\n");
continue;
}
if (ar_open(buf) >= 0) {
if (freeit) {
(void)free(arcname);
freeit = 0;
}
if ((arcname = strdup(buf)) == NULL) {
done = 1;
lstrval = -1;
paxwarn(0, "Cannot save archive name.");
return(-1);
}
freeit = 1;
break;
}
tty_prnt("Cannot open %s, try again\n", buf);
continue;
}
return(0);
}
void
#ifdef __STDC__
ar_start_gzip(int fd)
#else
ar_start_gzip(fd)
int fd;
#endif
{
pid_t pid;
int fds[2];
char *gzip_flags=NULL;
if (pipe(fds) < 0)
err(1, "could not pipe");
pid = fork();
if (pid < 0)
err(1, "could not fork");
if (pid) {
switch (act) {
case ARCHIVE:
dup2(fds[1], fd);
break;
case LIST:
case EXTRACT:
dup2(fds[0], fd);
break;
default:
errx(1, "ar_start_gzip: impossible");
}
close(fds[0]);
close(fds[1]);
} else {
switch (act) {
case ARCHIVE:
dup2(fds[0], STDIN_FILENO);
dup2(fd, STDOUT_FILENO);
gzip_flags = "-c";
break;
case LIST:
case EXTRACT:
dup2(fds[1], STDOUT_FILENO);
dup2(fd, STDIN_FILENO);
gzip_flags = "-dc";
break;
default:
errx(1, "ar_start_gzip: impossible");
}
close(fds[0]);
close(fds[1]);
if (execlp(gzip_program, gzip_program, gzip_flags, NULL) < 0)
err(1, "could not exec");
}
}