| /* | 
 |  * tar - `tape archiver', actually usable on any medium. | 
 |  *	POSIX "ustar" compliant when extracting, and by default when creating. | 
 |  *	this tar attempts to read and write multiple Tblock-byte blocks | 
 |  *	at once to and from the filesystem, and does not copy blocks | 
 |  *	around internally. | 
 |  */ | 
 |  | 
 | #include <u.h> | 
 | #include <libc.h> | 
 | #include <fcall.h>		/* for %M */ | 
 | #include <libString.h> | 
 |  | 
 | /* | 
 |  * modified versions of those in libc.h; scans only the first arg for | 
 |  * keyletters and options. | 
 |  */ | 
 | #define	TARGBEGIN {\ | 
 | 	if (!argv0) argv0 = *argv; \ | 
 | 	argv++, argc--;\ | 
 | 	if (argv[0]) {\ | 
 | 		char *_args, *_argt;\ | 
 | 		Rune _argc;\ | 
 | 		_args = &argv[0][0];\ | 
 | 		_argc = 0;\ | 
 | 		while(*_args && (_args += chartorune(&_argc, _args)))\ | 
 | 			switch(_argc) | 
 | #define	TARGEND	SET(_argt); USED(_argt);USED(_argc);USED(_args); \ | 
 | 	argc--, argv++; } \ | 
 | 	USED(argv); USED(argc); } | 
 | #define	TARGC() (_argc) | 
 |  | 
 | #define ROUNDUP(a, b)	(((a) + (b) - 1)/(b)) | 
 | #define BYTES2TBLKS(bytes) ROUNDUP(bytes, Tblock) | 
 |  | 
 | /* read big-endian binary integers; args must be (uchar *) */ | 
 | #define	G2BEBYTE(x)	(((x)[0]<<8)  |  (x)[1]) | 
 | #define	G3BEBYTE(x)	(((x)[0]<<16) | ((x)[1]<<8)  |  (x)[2]) | 
 | #define	G4BEBYTE(x)	(((x)[0]<<24) | ((x)[1]<<16) | ((x)[2]<<8) | (x)[3]) | 
 | #define	G8BEBYTE(x)	(((vlong)G4BEBYTE(x)<<32) | (u32int)G4BEBYTE((x)+4)) | 
 |  | 
 | typedef vlong Off; | 
 | typedef char *(*Refill)(int ar, char *bufs, int justhdr); | 
 |  | 
 | enum { Stdin, Stdout, Stderr }; | 
 | enum { Rd, Wr };			/* pipe fd-array indices */ | 
 | enum { Output, Input }; | 
 | enum { None, Toc, Xtract, Replace }; | 
 | enum { Alldata, Justnxthdr }; | 
 | enum { | 
 | 	Tblock = 512, | 
 | 	Namsiz = 100, | 
 | 	Maxpfx = 155,		/* from POSIX */ | 
 | 	Maxname = Namsiz + 1 + Maxpfx, | 
 | 	Binsize = 0x80,		/* flag in size[0], from gnu: positive binary size */ | 
 | 	Binnegsz = 0xff,	/* flag in size[0]: negative binary size */ | 
 |  | 
 | 	Nblock = 40,		/* maximum blocksize */ | 
 | 	Dblock = 20,		/* default blocksize */ | 
 | 	DEBUG = 0 | 
 | }; | 
 |  | 
 | /* POSIX link flags */ | 
 | enum { | 
 | 	LF_PLAIN1 =	'\0', | 
 | 	LF_PLAIN2 =	'0', | 
 | 	LF_LINK =	'1', | 
 | 	LF_SYMLINK1 =	'2', | 
 | 	LF_SYMLINK2 =	's',		/* 4BSD used this */ | 
 | 	LF_CHR =	'3', | 
 | 	LF_BLK =	'4', | 
 | 	LF_DIR =	'5', | 
 | 	LF_FIFO =	'6', | 
 | 	LF_CONTIG =	'7' | 
 | 	/* 'A' - 'Z' are reserved for custom implementations */ | 
 | }; | 
 |  | 
 | #define islink(lf)	(isreallink(lf) || issymlink(lf)) | 
 | #define isreallink(lf)	((lf) == LF_LINK) | 
 | #define issymlink(lf)	((lf) == LF_SYMLINK1 || (lf) == LF_SYMLINK2) | 
 |  | 
 | typedef struct Hdr { | 
 | 	char	name[Namsiz]; | 
 | 	char	mode[8]; | 
 | 	char	uid[8]; | 
 | 	char	gid[8]; | 
 | 	char	size[12]; | 
 | 	char	mtime[12]; | 
 | 	char	chksum[8]; | 
 | 	char	linkflag; | 
 | 	char	linkname[Namsiz]; | 
 |  | 
 | 	/* rest are defined by POSIX's ustar format; see p1003.2b */ | 
 | 	char	magic[6];	/* "ustar" */ | 
 | 	char	version[2]; | 
 | 	char	uname[32]; | 
 | 	char	gname[32]; | 
 | 	char	devmajor[8]; | 
 | 	char	devminor[8]; | 
 | 	char	prefix[Maxpfx]; /* if non-null, path= prefix "/" name */ | 
 | 	 | 
 | 	char pad[12]; | 
 | } Hdr; | 
 |  | 
 | typedef struct Compress { | 
 | 	char	*comp; | 
 | 	char	*decomp; | 
 | 	char	*sfx[4]; | 
 | } Compress; | 
 |  | 
 | static Compress comps[] = { | 
 | 	"gzip",		"gunzip",	{ ".tar.gz", ".tgz" },	/* default */ | 
 | 	"compress",	"uncompress",	{ ".tar.Z",  ".tz" }, | 
 | 	"bzip2",	"bunzip2",	{ ".tar.bz", ".tbz", | 
 | 					  ".tar.bz2",".tbz2" }, | 
 | }; | 
 |  | 
 | typedef struct Pushstate { | 
 | 	int	kid; | 
 | 	int	fd;	/* original fd */ | 
 | 	int	rfd;	/* replacement fd */ | 
 | 	int	input; | 
 | 	int	open; | 
 | } Pushstate; | 
 |  | 
 | #define OTHER(rdwr) (rdwr == Rd? Wr: Rd) | 
 |  | 
 | /* static int debug; */ | 
 | static int verb; | 
 | static int posix = 1; | 
 | static int docreate; | 
 | static int aruid; | 
 | static int argid; | 
 | static int relative = 1; | 
 | static int settime; | 
 | static int verbose; | 
 | static int docompress; | 
 | static int keepexisting; | 
 | static Off blkoff;	/* offset of the current archive block (not Tblock) */ | 
 | static Off nexthdr; | 
 |  | 
 | static int nblock = Dblock; | 
 | static char *usefile; | 
 | static char origdir[Maxname*2]; | 
 | static Hdr *tpblk, *endblk; | 
 | static Hdr *curblk; | 
 |  | 
 | static void | 
 | usage(void) | 
 | { | 
 | 	fprint(2, "usage: %s {crtx}[PRTfgkmpuvz] [archive] file1 file2...\n", | 
 | 		argv0); | 
 | 	exits("usage"); | 
 | } | 
 |  | 
 | /* compression */ | 
 |  | 
 | static Compress * | 
 | compmethod(char *name) | 
 | { | 
 | 	int i, nmlen = strlen(name), sfxlen; | 
 | 	Compress *cp; | 
 |  | 
 | 	for (cp = comps; cp < comps + nelem(comps); cp++) | 
 | 		for (i = 0; i < nelem(cp->sfx) && cp->sfx[i]; i++) { | 
 | 			sfxlen = strlen(cp->sfx[i]); | 
 | 			if (nmlen > sfxlen && | 
 | 			    strcmp(cp->sfx[i], name + nmlen - sfxlen) == 0) | 
 | 				return cp; | 
 | 		} | 
 | 	return docompress? comps: nil; | 
 | } | 
 |  | 
 | /* | 
 |  * push a filter, cmd, onto fd.  if input, it's an input descriptor. | 
 |  * returns a descriptor to replace fd, or -1 on error. | 
 |  */ | 
 | static int | 
 | push(int fd, char *cmd, int input, Pushstate *ps) | 
 | { | 
 | 	int nfd, pifds[2]; | 
 | 	String *s; | 
 |  | 
 | 	ps->open = 0; | 
 | 	ps->fd = fd; | 
 | 	ps->input = input; | 
 | 	if (fd < 0 || pipe(pifds) < 0) | 
 | 		return -1; | 
 | 	ps->kid = fork(); | 
 | 	switch (ps->kid) { | 
 | 	case -1: | 
 | 		return -1; | 
 | 	case 0: | 
 | 		if (input) | 
 | 			dup(pifds[Wr], Stdout); | 
 | 		else | 
 | 			dup(pifds[Rd], Stdin); | 
 | 		close(pifds[input? Rd: Wr]); | 
 | 		dup(fd, (input? Stdin: Stdout)); | 
 | 		s = s_new(); | 
 | 		if (cmd[0] != '/') | 
 | 			s_append(s, "/bin/"); | 
 | 		s_append(s, cmd); | 
 | 		execl(s_to_c(s), cmd, nil); | 
 | 		sysfatal("can't exec %s: %r", cmd); | 
 | 	default: | 
 | 		nfd = pifds[input? Rd: Wr]; | 
 | 		close(pifds[input? Wr: Rd]); | 
 | 		break; | 
 | 	} | 
 | 	ps->rfd = nfd; | 
 | 	ps->open = 1; | 
 | 	return nfd; | 
 | } | 
 |  | 
 | static char * | 
 | pushclose(Pushstate *ps) | 
 | { | 
 | 	Waitmsg *wm; | 
 |  | 
 | 	if (ps->fd < 0 || ps->rfd < 0 || !ps->open) | 
 | 		return "not open"; | 
 | 	close(ps->rfd); | 
 | 	ps->rfd = -1; | 
 | 	ps->open = 0; | 
 | 	while ((wm = wait()) != nil && wm->pid != ps->kid) | 
 | 		continue; | 
 | 	return wm? wm->msg: nil; | 
 | } | 
 |  | 
 | /* | 
 |  * block-buffer management | 
 |  */ | 
 |  | 
 | static void | 
 | initblks(void) | 
 | { | 
 | 	free(tpblk); | 
 | 	tpblk = malloc(Tblock * nblock); | 
 | 	assert(tpblk != nil); | 
 | 	endblk = tpblk + nblock; | 
 | } | 
 |  | 
 | /* | 
 |  * (re)fill block buffers from archive.  `justhdr' means we don't care | 
 |  * about the data before the next header block. | 
 |  */ | 
 | static char * | 
 | refill(int ar, char *bufs, int justhdr) | 
 | { | 
 | 	int i, n; | 
 | 	unsigned bytes = Tblock * nblock; | 
 | 	static int done, first = 1, seekable; | 
 |  | 
 | 	if (done) | 
 | 		return nil; | 
 |  | 
 | 	if (first) | 
 | 		seekable = seek(ar, 0, 1) >= 0; | 
 | 	blkoff = seek(ar, 0, 1);		/* note position for `tar r' */ | 
 | 	/* try to size non-pipe input at first read */ | 
 | 	if (first && usefile) { | 
 | 		n = read(ar, bufs, bytes); | 
 | 		if (n <= 0) | 
 | 			sysfatal("error reading archive: %r"); | 
 | 		i = n; | 
 | 		if (i % Tblock != 0) { | 
 | 			fprint(2, "%s: archive block size (%d) error\n", | 
 | 				argv0, i); | 
 | 			exits("blocksize"); | 
 | 		} | 
 | 		i /= Tblock; | 
 | 		if (i != nblock) { | 
 | 			nblock = i; | 
 | 			fprint(2, "%s: blocking = %d\n", argv0, nblock); | 
 | 			endblk = (Hdr *)bufs + nblock; | 
 | 			bytes = n; | 
 | 		} | 
 | 	} else if (justhdr && seekable && nexthdr - seek(ar, 0, 1) >= bytes) { | 
 | 		/* optimisation for huge archive members on seekable media */ | 
 | 		if (seek(ar, bytes, 1) < 0) | 
 | 			sysfatal("can't seek on archive: %r"); | 
 | 		n = bytes; | 
 | 	} else | 
 | 		n = readn(ar, bufs, bytes); | 
 | 	first = 0; | 
 |  | 
 | 	if (n == 0) | 
 | 		sysfatal("unexpected EOF reading archive"); | 
 | 	else if (n < 0) | 
 | 		sysfatal("error reading archive: %r"); | 
 | 	else if (n%Tblock != 0) | 
 | 		sysfatal("partial block read from archive"); | 
 | 	if (n != bytes) { | 
 | 		done = 1; | 
 | 		memset(bufs + n, 0, bytes - n); | 
 | 	} | 
 | 	return bufs; | 
 | } | 
 |  | 
 | static Hdr * | 
 | getblk(int ar, Refill rfp, int justhdr) | 
 | { | 
 | 	if (curblk == nil || curblk >= endblk) {  /* input block exhausted? */ | 
 | 		if (rfp != nil && (*rfp)(ar, (char *)tpblk, justhdr) == nil) | 
 | 			return nil; | 
 | 		curblk = tpblk; | 
 | 	} | 
 | 	return curblk++; | 
 | } | 
 |  | 
 | static Hdr * | 
 | getblkrd(int ar, int justhdr) | 
 | { | 
 | 	return getblk(ar, refill, justhdr); | 
 | } | 
 |  | 
 | static Hdr * | 
 | getblke(int ar) | 
 | { | 
 | 	return getblk(ar, nil, Alldata); | 
 | } | 
 |  | 
 | static Hdr * | 
 | getblkz(int ar) | 
 | { | 
 | 	Hdr *hp = getblke(ar); | 
 |  | 
 | 	if (hp != nil) | 
 | 		memset(hp, 0, Tblock); | 
 | 	return hp; | 
 | } | 
 |  | 
 | /* | 
 |  * how many block buffers are available, starting at the address | 
 |  * just returned by getblk*? | 
 |  */ | 
 | static int | 
 | gothowmany(int max) | 
 | { | 
 | 	int n = endblk - (curblk - 1); | 
 |  | 
 | 	return n > max? max: n; | 
 | } | 
 |  | 
 | /* | 
 |  * indicate that one is done with the last block obtained from getblke | 
 |  * and it is now available to be written into the archive. | 
 |  */ | 
 | static void | 
 | putlastblk(int ar) | 
 | { | 
 | 	unsigned bytes = Tblock * nblock; | 
 |  | 
 | 	/* if writing end-of-archive, aid compression (good hygiene too) */ | 
 | 	if (curblk < endblk) | 
 | 		memset(curblk, 0, (char *)endblk - (char *)curblk); | 
 | 	if (write(ar, tpblk, bytes) != bytes) | 
 | 		sysfatal("error writing archive: %r"); | 
 | } | 
 |  | 
 | static void | 
 | putblk(int ar) | 
 | { | 
 | 	if (curblk >= endblk) | 
 | 		putlastblk(ar); | 
 | } | 
 |  | 
 | static void | 
 | putbackblk(int ar) | 
 | { | 
 | 	curblk--; | 
 | 	USED(ar); | 
 | } | 
 |  | 
 | static void | 
 | putreadblks(int ar, int blks) | 
 | { | 
 | 	curblk += blks - 1; | 
 | 	USED(ar); | 
 | } | 
 |  | 
 | static void | 
 | putblkmany(int ar, int blks) | 
 | { | 
 | 	curblk += blks - 1; | 
 | 	putblk(ar); | 
 | } | 
 |  | 
 | /* | 
 |  * common routines | 
 |  */ | 
 |  | 
 | /* | 
 |  * modifies hp->chksum but restores it; important for the last block of the | 
 |  * old archive when updating with `tar rf archive' | 
 |  */ | 
 | static long | 
 | chksum(Hdr *hp) | 
 | { | 
 | 	int n = Tblock; | 
 | 	long i = 0; | 
 | 	uchar *cp = (uchar*)hp; | 
 | 	char oldsum[sizeof hp->chksum]; | 
 |  | 
 | 	memmove(oldsum, hp->chksum, sizeof oldsum); | 
 | 	memset(hp->chksum, ' ', sizeof hp->chksum); | 
 | 	while (n-- > 0) | 
 | 		i += *cp++; | 
 | 	memmove(hp->chksum, oldsum, sizeof oldsum); | 
 | 	return i; | 
 | } | 
 |  | 
 | static int | 
 | isustar(Hdr *hp) | 
 | { | 
 | 	return strcmp(hp->magic, "ustar") == 0; | 
 | } | 
 |  | 
 | /* | 
 |  * s is at most n bytes long, but need not be NUL-terminated. | 
 |  * if shorter than n bytes, all bytes after the first NUL must also | 
 |  * be NUL. | 
 |  */ | 
 | static int | 
 | tar_sstrnlen(char *s, int n) | 
 | { | 
 | 	return s[n - 1] != '\0'? n: strlen(s); | 
 | } | 
 |  | 
 | /* set fullname from header */ | 
 | static char * | 
 | name(Hdr *hp) | 
 | { | 
 | 	int pfxlen, namlen; | 
 | 	static char fullnamebuf[2 + Maxname + 1];	/* 2 at beginning for ./ on relative names */ | 
 | 	char *fullname; | 
 |  | 
 | 	fullname = fullnamebuf+2; | 
 | 	namlen = tar_sstrnlen(hp->name, sizeof hp->name); | 
 | 	if (hp->prefix[0] == '\0' || !isustar(hp)) {	/* old-style name? */ | 
 | 		memmove(fullname, hp->name, namlen); | 
 | 		fullname[namlen] = '\0'; | 
 | 		return fullname; | 
 | 	} | 
 |  | 
 | 	/* name is in two pieces */ | 
 | 	pfxlen = tar_sstrnlen(hp->prefix, sizeof hp->prefix); | 
 | 	memmove(fullname, hp->prefix, pfxlen); | 
 | 	fullname[pfxlen] = '/'; | 
 | 	memmove(fullname + pfxlen + 1, hp->name, namlen); | 
 | 	fullname[pfxlen + 1 + namlen] = '\0'; | 
 | 	return fullname; | 
 | } | 
 |  | 
 | static int | 
 | isdir(Hdr *hp) | 
 | { | 
 | 	/* the mode test is ugly but sometimes necessary */ | 
 | 	return hp->linkflag == LF_DIR || | 
 | 		strrchr(name(hp), '\0')[-1] == '/' || | 
 | 		(strtoul(hp->mode, nil, 8)&0170000) == 040000; | 
 | } | 
 |  | 
 | static int | 
 | eotar(Hdr *hp) | 
 | { | 
 | 	return name(hp)[0] == '\0'; | 
 | } | 
 |  | 
 | /* | 
 | static uvlong | 
 | getbe(uchar *src, int size) | 
 | { | 
 | 	uvlong vl = 0; | 
 |  | 
 | 	while (size-- > 0) { | 
 | 		vl <<= 8; | 
 | 		vl |= *src++; | 
 | 	} | 
 | 	return vl; | 
 | } | 
 |  */ | 
 |  | 
 | static void | 
 | putbe(uchar *dest, uvlong vl, int size) | 
 | { | 
 | 	for (dest += size; size-- > 0; vl >>= 8) | 
 | 		*--dest = vl; | 
 | } | 
 |  | 
 | /* | 
 |  * return the nominal size from the header block, which is not always the | 
 |  * size in the archive (the archive size may be zero for some file types | 
 |  * regardless of the nominal size). | 
 |  * | 
 |  * gnu and freebsd tars are now recording vlongs as big-endian binary | 
 |  * with a flag in byte 0 to indicate this, which permits file sizes up to | 
 |  * 2^64-1 (actually 2^80-1 but our file sizes are vlongs) rather than 2^33-1. | 
 |  */ | 
 | static Off | 
 | hdrsize(Hdr *hp) | 
 | { | 
 | 	uchar *p; | 
 |  | 
 | 	if((uchar)hp->size[0] == Binnegsz) { | 
 | 		fprint(2, "%s: %s: negative length, which is insane\n", | 
 | 			argv0, name(hp)); | 
 | 		return 0; | 
 | 	} else if((uchar)hp->size[0] == Binsize) { | 
 | 		p = (uchar *)hp->size + sizeof hp->size - 1 - | 
 | 			sizeof(vlong);		/* -1 for terminating space */ | 
 | 		return G8BEBYTE(p); | 
 | 	} else | 
 | 		return strtoull(hp->size, nil, 8); | 
 | } | 
 |  | 
 | /* | 
 |  * return the number of bytes recorded in the archive. | 
 |  */ | 
 | static Off | 
 | arsize(Hdr *hp) | 
 | { | 
 | 	if(isdir(hp) || islink(hp->linkflag)) | 
 | 		return 0; | 
 | 	return hdrsize(hp); | 
 | } | 
 |  | 
 | static Hdr * | 
 | readhdr(int ar) | 
 | { | 
 | 	long hdrcksum; | 
 | 	Hdr *hp; | 
 |  | 
 | 	hp = getblkrd(ar, Alldata); | 
 | 	if (hp == nil) | 
 | 		sysfatal("unexpected EOF instead of archive header"); | 
 | 	if (eotar(hp))			/* end-of-archive block? */ | 
 | 		return nil; | 
 | 	hdrcksum = strtoul(hp->chksum, nil, 8); | 
 | 	if (chksum(hp) != hdrcksum) | 
 | 		sysfatal("bad archive header checksum: name %.64s... %ld %ld", | 
 | 			hp->name, chksum(hp), hdrcksum); | 
 | 	nexthdr += Tblock*(1 + BYTES2TBLKS(arsize(hp))); | 
 | 	return hp; | 
 | } | 
 |  | 
 | /* | 
 |  * tar r[c] | 
 |  */ | 
 |  | 
 | /* | 
 |  * if name is longer than Namsiz bytes, try to split it at a slash and fit the | 
 |  * pieces into hp->prefix and hp->name. | 
 |  */ | 
 | static int | 
 | putfullname(Hdr *hp, char *name) | 
 | { | 
 | 	int namlen, pfxlen; | 
 | 	char *sl, *osl; | 
 | 	String *slname = nil; | 
 |  | 
 | 	if (isdir(hp)) { | 
 | 		slname = s_new(); | 
 | 		s_append(slname, name); | 
 | 		s_append(slname, "/");		/* posix requires this */ | 
 | 		name = s_to_c(slname); | 
 | 	} | 
 |  | 
 | 	namlen = strlen(name); | 
 | 	if (namlen <= Namsiz) { | 
 | 		strncpy(hp->name, name, Namsiz); | 
 | 		hp->prefix[0] = '\0';		/* ustar paranoia */ | 
 | 		return 0; | 
 | 	} | 
 |  | 
 | 	if (!posix || namlen > Maxname) { | 
 | 		fprint(2, "%s: name too long for tar header: %s\n", | 
 | 			argv0, name); | 
 | 		return -1; | 
 | 	} | 
 | 	/* | 
 | 	 * try various splits until one results in pieces that fit into the | 
 | 	 * appropriate fields of the header.  look for slashes from right | 
 | 	 * to left, in the hopes of putting the largest part of the name into | 
 | 	 * hp->prefix, which is larger than hp->name. | 
 | 	 */ | 
 | 	sl = strrchr(name, '/'); | 
 | 	while (sl != nil) { | 
 | 		pfxlen = sl - name; | 
 | 		if (pfxlen <= sizeof hp->prefix && namlen-1 - pfxlen <= Namsiz) | 
 | 			break; | 
 | 		osl = sl; | 
 | 		*osl = '\0'; | 
 | 		sl = strrchr(name, '/'); | 
 | 		*osl = '/'; | 
 | 	} | 
 | 	if (sl == nil) { | 
 | 		fprint(2, "%s: name can't be split to fit tar header: %s\n", | 
 | 			argv0, name); | 
 | 		return -1; | 
 | 	} | 
 | 	*sl = '\0'; | 
 | 	strncpy(hp->prefix, name, sizeof hp->prefix); | 
 | 	*sl++ = '/'; | 
 | 	strncpy(hp->name, sl, sizeof hp->name); | 
 | 	if (slname) | 
 | 		s_free(slname); | 
 | 	return 0; | 
 | } | 
 |  | 
 | static int | 
 | mkhdr(Hdr *hp, Dir *dir, char *file) | 
 | { | 
 | 	/* | 
 | 	 * these fields run together, so we format them in order and don't use | 
 | 	 * snprint. | 
 | 	 */ | 
 | 	sprint(hp->mode, "%6lo ", dir->mode & 0777); | 
 | 	sprint(hp->uid, "%6o ", aruid); | 
 | 	sprint(hp->gid, "%6o ", argid); | 
 | 	if (dir->length >= (Off)1<<32) { | 
 | 		static int printed; | 
 |  | 
 | 		if (!printed) { | 
 | 			printed = 1; | 
 | 			fprint(2, "%s: storing large sizes in \"base 256\"\n", argv0); | 
 | 		} | 
 | 		hp->size[0] = (char)Binsize; | 
 | 		/* emit so-called `base 256' representation of size */ | 
 | 		putbe((uchar *)hp->size+1, dir->length, sizeof hp->size - 2); | 
 | 		hp->size[sizeof hp->size - 1] = ' '; | 
 | 	} else | 
 | 		sprint(hp->size, "%11lluo ", dir->length); | 
 | 	sprint(hp->mtime, "%11luo ", dir->mtime); | 
 | 	hp->linkflag = (dir->mode&DMDIR? LF_DIR: LF_PLAIN1); | 
 | 	putfullname(hp, file); | 
 | 	if (posix) { | 
 | 		strncpy(hp->magic, "ustar", sizeof hp->magic); | 
 | 		strncpy(hp->version, "00", sizeof hp->version); | 
 | 		strncpy(hp->uname, dir->uid, sizeof hp->uname); | 
 | 		strncpy(hp->gname, dir->gid, sizeof hp->gname); | 
 | 	} | 
 | 	sprint(hp->chksum, "%6luo", chksum(hp)); | 
 | 	return 0; | 
 | } | 
 |  | 
 | static void addtoar(int ar, char *file, char *shortf); | 
 |  | 
 | static void | 
 | addtreetoar(int ar, char *file, char *shortf, int fd) | 
 | { | 
 | 	int n; | 
 | 	Dir *dent, *dirents; | 
 | 	String *name = s_new(); | 
 |  | 
 | 	n = dirreadall(fd, &dirents); | 
 | 	close(fd); | 
 | 	if (n == 0) | 
 | 		return; | 
 |  | 
 | 	if (chdir(shortf) < 0) | 
 | 		sysfatal("chdir %s: %r", file); | 
 | 	if (DEBUG) | 
 | 		fprint(2, "chdir %s\t# %s\n", shortf, file); | 
 |  | 
 | 	for (dent = dirents; dent < dirents + n; dent++) { | 
 | 		s_reset(name); | 
 | 		s_append(name, file); | 
 | 		s_append(name, "/"); | 
 | 		s_append(name, dent->name); | 
 | 		addtoar(ar, s_to_c(name), dent->name); | 
 | 	} | 
 | 	s_free(name); | 
 | 	free(dirents); | 
 |  | 
 | 	/* | 
 | 	 * this assumes that shortf is just one component, which is true | 
 | 	 * during directory descent, but not necessarily true of command-line | 
 | 	 * arguments.  Our caller (or addtoar's) must reset the working | 
 | 	 * directory if necessary. | 
 | 	 */ | 
 | 	if (chdir("..") < 0) | 
 | 		sysfatal("chdir %s/..: %r", file); | 
 | 	if (DEBUG) | 
 | 		fprint(2, "chdir ..\n"); | 
 | } | 
 |  | 
 | static void | 
 | addtoar(int ar, char *file, char *shortf) | 
 | { | 
 | 	int n, fd, isdir; | 
 | 	long bytes; | 
 | 	ulong blksleft, blksread; | 
 | 	Hdr *hbp; | 
 | 	Dir *dir; | 
 | 	String *name = nil; | 
 |  | 
 | 	if (shortf[0] == '#') { | 
 | 		name = s_new(); | 
 | 		s_append(name, "./"); | 
 | 		s_append(name, shortf); | 
 | 		shortf = s_to_c(name); | 
 | 	} | 
 |  | 
 | 	fd = open(shortf, OREAD); | 
 | 	if (fd < 0) { | 
 | 		fprint(2, "%s: can't open %s: %r\n", argv0, file); | 
 | 		if (name) | 
 | 			s_free(name); | 
 | 		return; | 
 | 	} | 
 | 	dir = dirfstat(fd); | 
 | 	if (dir == nil) | 
 | 		sysfatal("can't fstat %s: %r", file); | 
 |  | 
 | 	hbp = getblkz(ar); | 
 | 	isdir = !!(dir->qid.type&QTDIR); | 
 | 	if (mkhdr(hbp, dir, file) < 0) { | 
 | 		putbackblk(ar); | 
 | 		free(dir); | 
 | 		close(fd); | 
 | 		if (name) | 
 | 			s_free(name); | 
 | 		return; | 
 | 	} | 
 | 	putblk(ar); | 
 |  | 
 | 	blksleft = BYTES2TBLKS(dir->length); | 
 | 	free(dir); | 
 |  | 
 | 	if (isdir) | 
 | 		addtreetoar(ar, file, shortf, fd); | 
 | 	else { | 
 | 		for (; blksleft > 0; blksleft -= blksread) { | 
 | 			hbp = getblke(ar); | 
 | 			blksread = gothowmany(blksleft); | 
 | 			bytes = blksread * Tblock; | 
 | 			n = readn(fd, hbp, bytes); | 
 | 			if (n < 0) | 
 | 				sysfatal("error reading %s: %r", file); | 
 | 			/* | 
 | 			 * ignore EOF.  zero any partial block to aid | 
 | 			 * compression and emergency recovery of data. | 
 | 			 */ | 
 | 			if (n < Tblock) | 
 | 				memset((uchar*)hbp + n, 0, bytes - n); | 
 | 			putblkmany(ar, blksread); | 
 | 		} | 
 | 		close(fd); | 
 | 		if (verbose) | 
 | 			fprint(2, "%s\n", file); | 
 | 	} | 
 | 	if (name) | 
 | 		s_free(name); | 
 | } | 
 |  | 
 | static char * | 
 | replace(char **argv) | 
 | { | 
 | 	int i, ar; | 
 | 	ulong blksleft, blksread; | 
 | 	Off bytes; | 
 | 	Hdr *hp; | 
 | 	Compress *comp = nil; | 
 | 	Pushstate ps; | 
 |  | 
 | 	if (usefile && docreate) { | 
 | 		ar = create(usefile, OWRITE, 0666); | 
 | 		if (docompress) | 
 | 			comp = compmethod(usefile); | 
 | 	} else if (usefile) | 
 | 		ar = open(usefile, ORDWR); | 
 | 	else | 
 | 		ar = Stdout; | 
 | 	if (comp) | 
 | 		ar = push(ar, comp->comp, Output, &ps); | 
 | 	if (ar < 0) | 
 | 		sysfatal("can't open archive %s: %r", usefile); | 
 |  | 
 | 	if (usefile && !docreate) { | 
 | 		/* skip quickly to the end */ | 
 | 		while ((hp = readhdr(ar)) != nil) { | 
 | 			bytes = arsize(hp); | 
 | 			for (blksleft = BYTES2TBLKS(bytes); | 
 | 			     blksleft > 0 && getblkrd(ar, Justnxthdr) != nil; | 
 | 			     blksleft -= blksread) { | 
 | 				blksread = gothowmany(blksleft); | 
 | 				putreadblks(ar, blksread); | 
 | 			} | 
 | 		} | 
 | 		/* | 
 | 		 * we have just read the end-of-archive Tblock. | 
 | 		 * now seek back over the (big) archive block containing it, | 
 | 		 * and back up curblk ptr over end-of-archive Tblock in memory. | 
 | 		 */ | 
 | 		if (seek(ar, blkoff, 0) < 0) | 
 | 			sysfatal("can't seek back over end-of-archive: %r"); | 
 | 		curblk--; | 
 | 	} | 
 |  | 
 | 	for (i = 0; argv[i] != nil; i++) { | 
 | 		addtoar(ar, argv[i], argv[i]); | 
 | 		chdir(origdir);		/* for correctness & profiling */ | 
 | 	} | 
 |  | 
 | 	/* write end-of-archive marker */ | 
 | 	getblkz(ar); | 
 | 	putblk(ar); | 
 | 	getblkz(ar); | 
 | 	putlastblk(ar); | 
 |  | 
 | 	if (comp) | 
 | 		return pushclose(&ps); | 
 | 	if (ar > Stderr) | 
 | 		close(ar); | 
 | 	return nil; | 
 | } | 
 |  | 
 | /* | 
 |  * tar [xt] | 
 |  */ | 
 |  | 
 | /* is pfx a file-name prefix of name? */ | 
 | static int | 
 | prefix(char *name, char *pfx) | 
 | { | 
 | 	int pfxlen = strlen(pfx); | 
 | 	char clpfx[Maxname+1]; | 
 |  | 
 | 	if (pfxlen > Maxname) | 
 | 		return 0; | 
 | 	strcpy(clpfx, pfx); | 
 | 	cleanname(clpfx); | 
 | 	return strncmp(pfx, name, pfxlen) == 0 && | 
 | 		(name[pfxlen] == '\0' || name[pfxlen] == '/'); | 
 | } | 
 |  | 
 | static int | 
 | match(char *name, char **argv) | 
 | { | 
 | 	int i; | 
 | 	char clname[Maxname+1]; | 
 |  | 
 | 	if (argv[0] == nil) | 
 | 		return 1; | 
 | 	strcpy(clname, name); | 
 | 	cleanname(clname); | 
 | 	for (i = 0; argv[i] != nil; i++) | 
 | 		if (prefix(clname, argv[i])) | 
 | 			return 1; | 
 | 	return 0; | 
 | } | 
 |  | 
 | static void | 
 | cantcreate(char *s, int mode) | 
 | { | 
 | 	int len; | 
 | 	static char *last; | 
 |  | 
 | 	/* | 
 | 	 * Always print about files.  Only print about directories | 
 | 	 * we haven't printed about.  (Assumes archive is ordered | 
 | 	 * nicely.) | 
 | 	 */ | 
 | 	if(mode&DMDIR){ | 
 | 		if(last){ | 
 | 			/* already printed this directory */ | 
 | 			if(strcmp(s, last) == 0) | 
 | 				return; | 
 | 			/* printed a higher directory, so printed this one */ | 
 | 			len = strlen(s); | 
 | 			if(memcmp(s, last, len) == 0 && last[len] == '/') | 
 | 				return; | 
 | 		} | 
 | 		/* save */ | 
 | 		free(last); | 
 | 		last = strdup(s); | 
 | 	} | 
 | 	fprint(2, "%s: can't create %s: %r\n", argv0, s); | 
 | } | 
 |  | 
 | static int | 
 | makedir(char *s) | 
 | { | 
 | 	int f; | 
 |  | 
 | 	if (access(s, AEXIST) == 0) | 
 | 		return -1; | 
 | 	f = create(s, OREAD, DMDIR | 0777); | 
 | 	if (f >= 0) | 
 | 		close(f); | 
 | 	else | 
 | 		cantcreate(s, DMDIR); | 
 | 	return f; | 
 | } | 
 |  | 
 | static int | 
 | mkpdirs(char *s) | 
 | { | 
 | 	int err; | 
 | 	char *p; | 
 |  | 
 | 	p = s; | 
 | 	err = 0; | 
 | 	while (!err && (p = strchr(p+1, '/')) != nil) { | 
 | 		*p = '\0'; | 
 | 		err = (access(s, AEXIST) < 0 && makedir(s) < 0); | 
 | 		*p = '/'; | 
 | 	} | 
 | 	return -err; | 
 | } | 
 |  | 
 | /* Call access but preserve the error string. */ | 
 | static int | 
 | xaccess(char *name, int mode) | 
 | { | 
 | 	char err[ERRMAX]; | 
 | 	int rv; | 
 |  | 
 | 	err[0] = 0; | 
 | 	errstr(err, sizeof err); | 
 | 	rv = access(name, mode); | 
 | 	errstr(err, sizeof err); | 
 | 	return rv; | 
 | } | 
 |  | 
 | /* copy a file from the archive into the filesystem */ | 
 | /* fname is result of name(), so has two extra bytes at beginning */ | 
 | static void | 
 | extract1(int ar, Hdr *hp, char *fname) | 
 | { | 
 | 	int wrbytes, fd = -1, dir = 0; | 
 | 	long mtime = strtol(hp->mtime, nil, 8); | 
 | 	ulong mode = strtoul(hp->mode, nil, 8) & 0777; | 
 | 	Off bytes = hdrsize(hp);		/* for printing */ | 
 | 	ulong blksread, blksleft = BYTES2TBLKS(arsize(hp)); | 
 | 	Hdr *hbp; | 
 |  | 
 | 	if (isdir(hp)) { | 
 | 		mode |= DMDIR|0700; | 
 | 		dir = 1; | 
 | 	} | 
 | 	switch (hp->linkflag) { | 
 | 	case LF_LINK: | 
 | 	case LF_SYMLINK1: | 
 | 	case LF_SYMLINK2: | 
 | 	case LF_FIFO: | 
 | 		blksleft = 0; | 
 | 		break; | 
 | 	} | 
 | 	if (relative) { | 
 | 		if(fname[0] == '/') | 
 | 			*--fname = '.'; | 
 | 		else if(fname[0] == '#'){ | 
 | 			*--fname = '/'; | 
 | 			*--fname = '.'; | 
 | 		} | 
 | 	} | 
 | 	if (verb == Xtract) { | 
 | 		cleanname(fname); | 
 | 		switch (hp->linkflag) { | 
 | 		case LF_LINK: | 
 | 		case LF_SYMLINK1: | 
 | 		case LF_SYMLINK2: | 
 | 			fprint(2, "%s: can't make (sym)link %s\n", | 
 | 				argv0, fname); | 
 | 			break; | 
 | 		case LF_FIFO: | 
 | 			fprint(2, "%s: can't make fifo %s\n", argv0, fname); | 
 | 			break; | 
 | 		default: | 
 | 			if (!keepexisting || access(fname, AEXIST) < 0) { | 
 | 				int rw = (dir? OREAD: OWRITE); | 
 |  | 
 | 				fd = create(fname, rw, mode); | 
 | 				if (fd < 0) { | 
 | 					mkpdirs(fname); | 
 | 					fd = create(fname, rw, mode); | 
 | 				} | 
 | 				if (fd < 0 && | 
 | 				    (!dir || xaccess(fname, AEXIST) < 0)) | 
 | 				    	cantcreate(fname, mode); | 
 | 			} | 
 | 			if (fd >= 0 && verbose) | 
 | 				fprint(2, "%s\n", fname); | 
 | 			break; | 
 | 		} | 
 | 	} else if (verbose) { | 
 | 		char *cp = ctime(mtime); | 
 |  | 
 | 		print("%M %8lld %-12.12s %-4.4s %s\n", | 
 | 			mode, bytes, cp+4, cp+24, fname); | 
 | 	} else | 
 | 		print("%s\n", fname); | 
 |  | 
 | 	if (blksleft == 0) | 
 | 		bytes = 0; | 
 | 	for (; blksleft > 0; blksleft -= blksread) { | 
 | 		hbp = getblkrd(ar, (fd >= 0? Alldata: Justnxthdr)); | 
 | 		if (hbp == nil) | 
 | 			sysfatal("unexpected EOF on archive extracting %s", | 
 | 				fname); | 
 | 		blksread = gothowmany(blksleft); | 
 | 		if (blksread <= 0) | 
 | 			fprint(2, "%s: got %ld blocks reading %s!\n", | 
 | 				argv0, blksread, fname); | 
 | 		wrbytes = Tblock*blksread; | 
 | 		if(wrbytes > bytes) | 
 | 			wrbytes = bytes; | 
 | 		if (fd >= 0 && write(fd, hbp, wrbytes) != wrbytes) | 
 | 			sysfatal("write error on %s: %r", fname); | 
 | 		putreadblks(ar, blksread); | 
 | 		bytes -= wrbytes; | 
 | 	} | 
 | 	if (bytes > 0) | 
 | 		fprint(2, | 
 | 		    "%s: %lld bytes uncopied at eof; %s not fully extracted\n", | 
 | 			argv0, bytes, fname); | 
 | 	if (fd >= 0) { | 
 | 		/* | 
 | 		 * directories should be wstated after we're done | 
 | 		 * creating files in them. | 
 | 		 */ | 
 | 		if (settime) { | 
 | 			Dir nd; | 
 |  | 
 | 			nulldir(&nd); | 
 | 			nd.mtime = mtime; | 
 | 			dirfwstat(fd, &nd); | 
 | 			if (isustar(hp)) { | 
 | 				nulldir(&nd); | 
 | 				nd.gid = hp->gname; | 
 | 				dirfwstat(fd, &nd); | 
 | 			} | 
 | 		} | 
 | 		close(fd); | 
 | 	} | 
 | } | 
 |  | 
 | static void | 
 | skip(int ar, Hdr *hp, char *fname) | 
 | { | 
 | 	ulong blksleft, blksread; | 
 | 	Hdr *hbp; | 
 |  | 
 | 	for (blksleft = BYTES2TBLKS(arsize(hp)); blksleft > 0; | 
 | 	     blksleft -= blksread) { | 
 | 		hbp = getblkrd(ar, Justnxthdr); | 
 | 		if (hbp == nil) | 
 | 			sysfatal("unexpected EOF on archive extracting %s", | 
 | 				fname); | 
 | 		blksread = gothowmany(blksleft); | 
 | 		putreadblks(ar, blksread); | 
 | 	} | 
 | } | 
 |  | 
 | static char * | 
 | extract(char **argv) | 
 | { | 
 | 	int ar; | 
 | 	char *longname; | 
 | 	Hdr *hp; | 
 | 	Compress *comp = nil; | 
 | 	Pushstate ps; | 
 |  | 
 | 	if (usefile) { | 
 | 		ar = open(usefile, OREAD); | 
 | 		comp = compmethod(usefile); | 
 | 	} else | 
 | 		ar = Stdin; | 
 | 	if (comp) | 
 | 		ar = push(ar, comp->decomp, Input, &ps); | 
 | 	if (ar < 0) | 
 | 		sysfatal("can't open archive %s: %r", usefile); | 
 |  | 
 | 	while ((hp = readhdr(ar)) != nil) { | 
 | 		longname = name(hp); | 
 | 		if (match(longname, argv)) | 
 | 			extract1(ar, hp, longname); | 
 | 		else | 
 | 			skip(ar, hp, longname); | 
 | 	} | 
 |  | 
 | 	if (comp) | 
 | 		return pushclose(&ps); | 
 | 	if (ar > Stderr) | 
 | 		close(ar); | 
 | 	return nil; | 
 | } | 
 |  | 
 | void | 
 | main(int argc, char *argv[]) | 
 | { | 
 | 	int errflg = 0; | 
 | 	char *ret = nil; | 
 |  | 
 | 	fmtinstall('M', dirmodefmt); | 
 |  | 
 | 	if(sizeof(Hdr) != Tblock) | 
 | 		sysfatal("padding in hdr should be %d", Tblock-sizeof(Hdr)+sizeof(curblk->pad)); | 
 | 	TARGBEGIN { | 
 | 	case 'c': | 
 | 		docreate++; | 
 | 		verb = Replace; | 
 | 		break; | 
 | 	case 'f': | 
 | 		usefile = EARGF(usage()); | 
 | 		break; | 
 | 	case 'g': | 
 | 		argid = strtoul(EARGF(usage()), 0, 0); | 
 | 		break; | 
 | 	case 'k': | 
 | 		keepexisting++; | 
 | 		break; | 
 | 	case 'm':	/* compatibility */ | 
 | 		settime = 0; | 
 | 		break; | 
 | 	case 'p': | 
 | 		posix++; | 
 | 		break; | 
 | 	case 'P': | 
 | 		posix = 0; | 
 | 		break; | 
 | 	case 'r': | 
 | 		verb = Replace; | 
 | 		break; | 
 | 	case 'R': | 
 | 		relative = 0; | 
 | 		break; | 
 | 	case 't': | 
 | 		verb = Toc; | 
 | 		break; | 
 | 	case 'T': | 
 | 		settime++; | 
 | 		break; | 
 | 	case 'u': | 
 | 		aruid = strtoul(EARGF(usage()), 0, 0); | 
 | 		break; | 
 | 	case 'v': | 
 | 		verbose++; | 
 | 		break; | 
 | 	case 'x': | 
 | 		verb = Xtract; | 
 | 		break; | 
 | 	case 'z': | 
 | 		docompress++; | 
 | 		break; | 
 | 	case '-': | 
 | 		break; | 
 | 	default: | 
 | 		fprint(2, "tar: unknown letter %C\n", TARGC()); | 
 | 		errflg++; | 
 | 		break; | 
 | 	} TARGEND | 
 |  | 
 | 	if (argc < 0 || errflg) | 
 | 		usage(); | 
 |  | 
 | 	initblks(); | 
 | 	switch (verb) { | 
 | 	case Toc: | 
 | 	case Xtract: | 
 | 		ret = extract(argv); | 
 | 		break; | 
 | 	case Replace: | 
 | 		if (getwd(origdir, sizeof origdir) == nil) | 
 | 			strcpy(origdir, "/tmp"); | 
 | 		ret = replace(argv); | 
 | 		break; | 
 | 	default: | 
 | 		usage(); | 
 | 		break; | 
 | 	} | 
 | 	exits(ret); | 
 | } |