2 031ddeb2 2005-07-22 devnull * tar - `tape archiver', actually usable on any medium.
3 031ddeb2 2005-07-22 devnull * POSIX "ustar" compliant when extracting, and by default when creating.
4 031ddeb2 2005-07-22 devnull * this tar attempts to read and write multiple Tblock-byte blocks
5 031ddeb2 2005-07-22 devnull * at once to and from the filesystem, and does not copy blocks
6 031ddeb2 2005-07-22 devnull * around internally.
9 031ddeb2 2005-07-22 devnull #include <u.h>
10 031ddeb2 2005-07-22 devnull #include <libc.h>
11 031ddeb2 2005-07-22 devnull #include <fcall.h> /* for %M */
12 67e944db 2005-07-22 devnull #include <libString.h>
15 031ddeb2 2005-07-22 devnull * modified versions of those in libc.h; scans only the first arg for
16 031ddeb2 2005-07-22 devnull * keyletters and options.
18 031ddeb2 2005-07-22 devnull #define TARGBEGIN {\
19 4515de8f 2006-04-20 devnull if (!argv0) argv0 = *argv; \
20 4515de8f 2006-04-20 devnull argv++, argc--;\
21 031ddeb2 2005-07-22 devnull if (argv[0]) {\
22 031ddeb2 2005-07-22 devnull char *_args, *_argt;\
23 031ddeb2 2005-07-22 devnull Rune _argc;\
24 031ddeb2 2005-07-22 devnull _args = &argv[0][0];\
25 031ddeb2 2005-07-22 devnull _argc = 0;\
26 031ddeb2 2005-07-22 devnull while(*_args && (_args += chartorune(&_argc, _args)))\
27 031ddeb2 2005-07-22 devnull switch(_argc)
28 031ddeb2 2005-07-22 devnull #define TARGEND SET(_argt); USED(_argt);USED(_argc);USED(_args); \
29 031ddeb2 2005-07-22 devnull argc--, argv++; } \
30 031ddeb2 2005-07-22 devnull USED(argv); USED(argc); }
31 031ddeb2 2005-07-22 devnull #define TARGC() (_argc)
33 031ddeb2 2005-07-22 devnull #define ROUNDUP(a, b) (((a) + (b) - 1)/(b))
34 031ddeb2 2005-07-22 devnull #define BYTES2TBLKS(bytes) ROUNDUP(bytes, Tblock)
36 6019a96c 2006-03-30 devnull /* read big-endian binary integers; args must be (uchar *) */
37 6019a96c 2006-03-30 devnull #define G2BEBYTE(x) (((x)[0]<<8) | (x)[1])
38 6019a96c 2006-03-30 devnull #define G3BEBYTE(x) (((x)[0]<<16) | ((x)[1]<<8) | (x)[2])
39 6019a96c 2006-03-30 devnull #define G4BEBYTE(x) (((x)[0]<<24) | ((x)[1]<<16) | ((x)[2]<<8) | (x)[3])
40 6019a96c 2006-03-30 devnull #define G8BEBYTE(x) (((vlong)G4BEBYTE(x)<<32) | (u32int)G4BEBYTE((x)+4))
42 031ddeb2 2005-07-22 devnull typedef vlong Off;
43 031ddeb2 2005-07-22 devnull typedef char *(*Refill)(int ar, char *bufs, int justhdr);
45 031ddeb2 2005-07-22 devnull enum { Stdin, Stdout, Stderr };
46 031ddeb2 2005-07-22 devnull enum { Rd, Wr }; /* pipe fd-array indices */
47 031ddeb2 2005-07-22 devnull enum { Output, Input };
48 031ddeb2 2005-07-22 devnull enum { None, Toc, Xtract, Replace };
49 031ddeb2 2005-07-22 devnull enum { Alldata, Justnxthdr };
51 031ddeb2 2005-07-22 devnull Tblock = 512,
52 031ddeb2 2005-07-22 devnull Namsiz = 100,
53 031ddeb2 2005-07-22 devnull Maxpfx = 155, /* from POSIX */
54 031ddeb2 2005-07-22 devnull Maxname = Namsiz + 1 + Maxpfx,
55 6019a96c 2006-03-30 devnull Binsize = 0x80, /* flag in size[0], from gnu: positive binary size */
56 6019a96c 2006-03-30 devnull Binnegsz = 0xff, /* flag in size[0]: negative binary size */
58 6019a96c 2006-03-30 devnull Nblock = 40, /* maximum blocksize */
59 6019a96c 2006-03-30 devnull Dblock = 20, /* default blocksize */
60 cbeb0b26 2006-04-01 devnull DEBUG = 0
63 031ddeb2 2005-07-22 devnull /* POSIX link flags */
65 031ddeb2 2005-07-22 devnull LF_PLAIN1 = '\0',
66 031ddeb2 2005-07-22 devnull LF_PLAIN2 = '0',
67 031ddeb2 2005-07-22 devnull LF_LINK = '1',
68 031ddeb2 2005-07-22 devnull LF_SYMLINK1 = '2',
69 6019a96c 2006-03-30 devnull LF_SYMLINK2 = 's', /* 4BSD used this */
70 031ddeb2 2005-07-22 devnull LF_CHR = '3',
71 031ddeb2 2005-07-22 devnull LF_BLK = '4',
72 031ddeb2 2005-07-22 devnull LF_DIR = '5',
73 031ddeb2 2005-07-22 devnull LF_FIFO = '6',
74 cbeb0b26 2006-04-01 devnull LF_CONTIG = '7'
75 031ddeb2 2005-07-22 devnull /* 'A' - 'Z' are reserved for custom implementations */
78 031ddeb2 2005-07-22 devnull #define islink(lf) (isreallink(lf) || issymlink(lf))
79 031ddeb2 2005-07-22 devnull #define isreallink(lf) ((lf) == LF_LINK)
80 031ddeb2 2005-07-22 devnull #define issymlink(lf) ((lf) == LF_SYMLINK1 || (lf) == LF_SYMLINK2)
82 61309328 2006-03-31 devnull typedef struct Hdr {
83 61309328 2006-03-31 devnull char name[Namsiz];
84 61309328 2006-03-31 devnull char mode[8];
85 61309328 2006-03-31 devnull char uid[8];
86 61309328 2006-03-31 devnull char gid[8];
87 61309328 2006-03-31 devnull char size[12];
88 61309328 2006-03-31 devnull char mtime[12];
89 61309328 2006-03-31 devnull char chksum[8];
90 61309328 2006-03-31 devnull char linkflag;
91 61309328 2006-03-31 devnull char linkname[Namsiz];
93 61309328 2006-03-31 devnull /* rest are defined by POSIX's ustar format; see p1003.2b */
94 61309328 2006-03-31 devnull char magic[6]; /* "ustar" */
95 61309328 2006-03-31 devnull char version[2];
96 61309328 2006-03-31 devnull char uname[32];
97 61309328 2006-03-31 devnull char gname[32];
98 61309328 2006-03-31 devnull char devmajor[8];
99 61309328 2006-03-31 devnull char devminor[8];
100 61309328 2006-03-31 devnull char prefix[Maxpfx]; /* if non-null, path= prefix "/" name */
102 61309328 2006-03-31 devnull char pad[12];
105 61309328 2006-03-31 devnull typedef struct Compress {
106 031ddeb2 2005-07-22 devnull char *comp;
107 031ddeb2 2005-07-22 devnull char *decomp;
108 031ddeb2 2005-07-22 devnull char *sfx[4];
109 031ddeb2 2005-07-22 devnull } Compress;
111 031ddeb2 2005-07-22 devnull static Compress comps[] = {
112 031ddeb2 2005-07-22 devnull "gzip", "gunzip", { ".tar.gz", ".tgz" }, /* default */
113 031ddeb2 2005-07-22 devnull "compress", "uncompress", { ".tar.Z", ".tz" },
114 031ddeb2 2005-07-22 devnull "bzip2", "bunzip2", { ".tar.bz", ".tbz",
115 031ddeb2 2005-07-22 devnull ".tar.bz2",".tbz2" },
118 61309328 2006-03-31 devnull typedef struct Pushstate {
119 031ddeb2 2005-07-22 devnull int kid;
120 031ddeb2 2005-07-22 devnull int fd; /* original fd */
121 031ddeb2 2005-07-22 devnull int rfd; /* replacement fd */
122 031ddeb2 2005-07-22 devnull int input;
123 031ddeb2 2005-07-22 devnull int open;
124 031ddeb2 2005-07-22 devnull } Pushstate;
126 031ddeb2 2005-07-22 devnull #define OTHER(rdwr) (rdwr == Rd? Wr: Rd)
128 67e944db 2005-07-22 devnull /* static int debug; */
129 031ddeb2 2005-07-22 devnull static int verb;
130 031ddeb2 2005-07-22 devnull static int posix = 1;
131 031ddeb2 2005-07-22 devnull static int docreate;
132 031ddeb2 2005-07-22 devnull static int aruid;
133 031ddeb2 2005-07-22 devnull static int argid;
134 031ddeb2 2005-07-22 devnull static int relative = 1;
135 031ddeb2 2005-07-22 devnull static int settime;
136 031ddeb2 2005-07-22 devnull static int verbose;
137 031ddeb2 2005-07-22 devnull static int docompress;
138 031ddeb2 2005-07-22 devnull static int keepexisting;
139 031ddeb2 2005-07-22 devnull static Off blkoff; /* offset of the current archive block (not Tblock) */
140 031ddeb2 2005-07-22 devnull static Off nexthdr;
142 031ddeb2 2005-07-22 devnull static int nblock = Dblock;
143 031ddeb2 2005-07-22 devnull static char *usefile;
144 031ddeb2 2005-07-22 devnull static char origdir[Maxname*2];
145 031ddeb2 2005-07-22 devnull static Hdr *tpblk, *endblk;
146 031ddeb2 2005-07-22 devnull static Hdr *curblk;
148 031ddeb2 2005-07-22 devnull static void
149 031ddeb2 2005-07-22 devnull usage(void)
151 031ddeb2 2005-07-22 devnull fprint(2, "usage: %s {crtx}[PRTfgkmpuvz] [archive] file1 file2...\n",
153 031ddeb2 2005-07-22 devnull exits("usage");
156 031ddeb2 2005-07-22 devnull /* compression */
158 031ddeb2 2005-07-22 devnull static Compress *
159 031ddeb2 2005-07-22 devnull compmethod(char *name)
161 031ddeb2 2005-07-22 devnull int i, nmlen = strlen(name), sfxlen;
162 031ddeb2 2005-07-22 devnull Compress *cp;
164 031ddeb2 2005-07-22 devnull for (cp = comps; cp < comps + nelem(comps); cp++)
165 031ddeb2 2005-07-22 devnull for (i = 0; i < nelem(cp->sfx) && cp->sfx[i]; i++) {
166 031ddeb2 2005-07-22 devnull sfxlen = strlen(cp->sfx[i]);
167 031ddeb2 2005-07-22 devnull if (nmlen > sfxlen &&
168 031ddeb2 2005-07-22 devnull strcmp(cp->sfx[i], name + nmlen - sfxlen) == 0)
169 031ddeb2 2005-07-22 devnull return cp;
171 031ddeb2 2005-07-22 devnull return docompress? comps: nil;
175 031ddeb2 2005-07-22 devnull * push a filter, cmd, onto fd. if input, it's an input descriptor.
176 031ddeb2 2005-07-22 devnull * returns a descriptor to replace fd, or -1 on error.
178 031ddeb2 2005-07-22 devnull static int
179 031ddeb2 2005-07-22 devnull push(int fd, char *cmd, int input, Pushstate *ps)
181 031ddeb2 2005-07-22 devnull int nfd, pifds[2];
182 031ddeb2 2005-07-22 devnull String *s;
184 031ddeb2 2005-07-22 devnull ps->open = 0;
185 031ddeb2 2005-07-22 devnull ps->fd = fd;
186 031ddeb2 2005-07-22 devnull ps->input = input;
187 031ddeb2 2005-07-22 devnull if (fd < 0 || pipe(pifds) < 0)
188 031ddeb2 2005-07-22 devnull return -1;
189 031ddeb2 2005-07-22 devnull ps->kid = fork();
190 031ddeb2 2005-07-22 devnull switch (ps->kid) {
191 031ddeb2 2005-07-22 devnull case -1:
192 031ddeb2 2005-07-22 devnull return -1;
194 031ddeb2 2005-07-22 devnull if (input)
195 031ddeb2 2005-07-22 devnull dup(pifds[Wr], Stdout);
197 031ddeb2 2005-07-22 devnull dup(pifds[Rd], Stdin);
198 031ddeb2 2005-07-22 devnull close(pifds[input? Rd: Wr]);
199 031ddeb2 2005-07-22 devnull dup(fd, (input? Stdin: Stdout));
200 031ddeb2 2005-07-22 devnull s = s_new();
201 031ddeb2 2005-07-22 devnull if (cmd[0] != '/')
202 031ddeb2 2005-07-22 devnull s_append(s, "/bin/");
203 031ddeb2 2005-07-22 devnull s_append(s, cmd);
204 031ddeb2 2005-07-22 devnull execl(s_to_c(s), cmd, nil);
205 031ddeb2 2005-07-22 devnull sysfatal("can't exec %s: %r", cmd);
206 031ddeb2 2005-07-22 devnull default:
207 031ddeb2 2005-07-22 devnull nfd = pifds[input? Rd: Wr];
208 031ddeb2 2005-07-22 devnull close(pifds[input? Wr: Rd]);
211 031ddeb2 2005-07-22 devnull ps->rfd = nfd;
212 031ddeb2 2005-07-22 devnull ps->open = 1;
213 031ddeb2 2005-07-22 devnull return nfd;
216 031ddeb2 2005-07-22 devnull static char *
217 031ddeb2 2005-07-22 devnull pushclose(Pushstate *ps)
219 031ddeb2 2005-07-22 devnull Waitmsg *wm;
221 031ddeb2 2005-07-22 devnull if (ps->fd < 0 || ps->rfd < 0 || !ps->open)
222 031ddeb2 2005-07-22 devnull return "not open";
223 031ddeb2 2005-07-22 devnull close(ps->rfd);
224 031ddeb2 2005-07-22 devnull ps->rfd = -1;
225 031ddeb2 2005-07-22 devnull ps->open = 0;
226 031ddeb2 2005-07-22 devnull while ((wm = wait()) != nil && wm->pid != ps->kid)
227 031ddeb2 2005-07-22 devnull continue;
228 031ddeb2 2005-07-22 devnull return wm? wm->msg: nil;
232 031ddeb2 2005-07-22 devnull * block-buffer management
235 031ddeb2 2005-07-22 devnull static void
236 031ddeb2 2005-07-22 devnull initblks(void)
238 031ddeb2 2005-07-22 devnull free(tpblk);
239 031ddeb2 2005-07-22 devnull tpblk = malloc(Tblock * nblock);
240 031ddeb2 2005-07-22 devnull assert(tpblk != nil);
241 031ddeb2 2005-07-22 devnull endblk = tpblk + nblock;
245 031ddeb2 2005-07-22 devnull * (re)fill block buffers from archive. `justhdr' means we don't care
246 031ddeb2 2005-07-22 devnull * about the data before the next header block.
248 031ddeb2 2005-07-22 devnull static char *
249 031ddeb2 2005-07-22 devnull refill(int ar, char *bufs, int justhdr)
251 031ddeb2 2005-07-22 devnull int i, n;
252 031ddeb2 2005-07-22 devnull unsigned bytes = Tblock * nblock;
253 031ddeb2 2005-07-22 devnull static int done, first = 1, seekable;
255 031ddeb2 2005-07-22 devnull if (done)
256 031ddeb2 2005-07-22 devnull return nil;
258 031ddeb2 2005-07-22 devnull if (first)
259 031ddeb2 2005-07-22 devnull seekable = seek(ar, 0, 1) >= 0;
260 6019a96c 2006-03-30 devnull blkoff = seek(ar, 0, 1); /* note position for `tar r' */
261 031ddeb2 2005-07-22 devnull /* try to size non-pipe input at first read */
262 031ddeb2 2005-07-22 devnull if (first && usefile) {
263 031ddeb2 2005-07-22 devnull n = read(ar, bufs, bytes);
264 031ddeb2 2005-07-22 devnull if (n <= 0)
265 031ddeb2 2005-07-22 devnull sysfatal("error reading archive: %r");
267 031ddeb2 2005-07-22 devnull if (i % Tblock != 0) {
268 031ddeb2 2005-07-22 devnull fprint(2, "%s: archive block size (%d) error\n",
269 031ddeb2 2005-07-22 devnull argv0, i);
270 031ddeb2 2005-07-22 devnull exits("blocksize");
272 031ddeb2 2005-07-22 devnull i /= Tblock;
273 031ddeb2 2005-07-22 devnull if (i != nblock) {
274 031ddeb2 2005-07-22 devnull nblock = i;
275 031ddeb2 2005-07-22 devnull fprint(2, "%s: blocking = %d\n", argv0, nblock);
276 031ddeb2 2005-07-22 devnull endblk = (Hdr *)bufs + nblock;
277 031ddeb2 2005-07-22 devnull bytes = n;
279 031ddeb2 2005-07-22 devnull } else if (justhdr && seekable && nexthdr - seek(ar, 0, 1) >= bytes) {
280 031ddeb2 2005-07-22 devnull /* optimisation for huge archive members on seekable media */
281 031ddeb2 2005-07-22 devnull if (seek(ar, bytes, 1) < 0)
282 031ddeb2 2005-07-22 devnull sysfatal("can't seek on archive: %r");
283 031ddeb2 2005-07-22 devnull n = bytes;
285 031ddeb2 2005-07-22 devnull n = readn(ar, bufs, bytes);
286 031ddeb2 2005-07-22 devnull first = 0;
288 031ddeb2 2005-07-22 devnull if (n == 0)
289 031ddeb2 2005-07-22 devnull sysfatal("unexpected EOF reading archive");
290 031ddeb2 2005-07-22 devnull else if (n < 0)
291 031ddeb2 2005-07-22 devnull sysfatal("error reading archive: %r");
292 031ddeb2 2005-07-22 devnull else if (n%Tblock != 0)
293 031ddeb2 2005-07-22 devnull sysfatal("partial block read from archive");
294 031ddeb2 2005-07-22 devnull if (n != bytes) {
295 031ddeb2 2005-07-22 devnull done = 1;
296 031ddeb2 2005-07-22 devnull memset(bufs + n, 0, bytes - n);
298 031ddeb2 2005-07-22 devnull return bufs;
301 031ddeb2 2005-07-22 devnull static Hdr *
302 031ddeb2 2005-07-22 devnull getblk(int ar, Refill rfp, int justhdr)
304 031ddeb2 2005-07-22 devnull if (curblk == nil || curblk >= endblk) { /* input block exhausted? */
305 031ddeb2 2005-07-22 devnull if (rfp != nil && (*rfp)(ar, (char *)tpblk, justhdr) == nil)
306 031ddeb2 2005-07-22 devnull return nil;
307 031ddeb2 2005-07-22 devnull curblk = tpblk;
309 031ddeb2 2005-07-22 devnull return curblk++;
312 031ddeb2 2005-07-22 devnull static Hdr *
313 031ddeb2 2005-07-22 devnull getblkrd(int ar, int justhdr)
315 031ddeb2 2005-07-22 devnull return getblk(ar, refill, justhdr);
318 031ddeb2 2005-07-22 devnull static Hdr *
319 031ddeb2 2005-07-22 devnull getblke(int ar)
321 031ddeb2 2005-07-22 devnull return getblk(ar, nil, Alldata);
324 031ddeb2 2005-07-22 devnull static Hdr *
325 031ddeb2 2005-07-22 devnull getblkz(int ar)
327 031ddeb2 2005-07-22 devnull Hdr *hp = getblke(ar);
329 031ddeb2 2005-07-22 devnull if (hp != nil)
330 61309328 2006-03-31 devnull memset(hp, 0, Tblock);
331 031ddeb2 2005-07-22 devnull return hp;
335 031ddeb2 2005-07-22 devnull * how many block buffers are available, starting at the address
336 031ddeb2 2005-07-22 devnull * just returned by getblk*?
338 031ddeb2 2005-07-22 devnull static int
339 031ddeb2 2005-07-22 devnull gothowmany(int max)
341 031ddeb2 2005-07-22 devnull int n = endblk - (curblk - 1);
343 031ddeb2 2005-07-22 devnull return n > max? max: n;
347 031ddeb2 2005-07-22 devnull * indicate that one is done with the last block obtained from getblke
348 031ddeb2 2005-07-22 devnull * and it is now available to be written into the archive.
350 031ddeb2 2005-07-22 devnull static void
351 031ddeb2 2005-07-22 devnull putlastblk(int ar)
353 031ddeb2 2005-07-22 devnull unsigned bytes = Tblock * nblock;
355 031ddeb2 2005-07-22 devnull /* if writing end-of-archive, aid compression (good hygiene too) */
356 031ddeb2 2005-07-22 devnull if (curblk < endblk)
357 031ddeb2 2005-07-22 devnull memset(curblk, 0, (char *)endblk - (char *)curblk);
358 031ddeb2 2005-07-22 devnull if (write(ar, tpblk, bytes) != bytes)
359 031ddeb2 2005-07-22 devnull sysfatal("error writing archive: %r");
362 031ddeb2 2005-07-22 devnull static void
363 031ddeb2 2005-07-22 devnull putblk(int ar)
365 031ddeb2 2005-07-22 devnull if (curblk >= endblk)
366 031ddeb2 2005-07-22 devnull putlastblk(ar);
369 031ddeb2 2005-07-22 devnull static void
370 031ddeb2 2005-07-22 devnull putbackblk(int ar)
372 031ddeb2 2005-07-22 devnull curblk--;
373 031ddeb2 2005-07-22 devnull USED(ar);
376 031ddeb2 2005-07-22 devnull static void
377 031ddeb2 2005-07-22 devnull putreadblks(int ar, int blks)
379 031ddeb2 2005-07-22 devnull curblk += blks - 1;
380 031ddeb2 2005-07-22 devnull USED(ar);
383 031ddeb2 2005-07-22 devnull static void
384 031ddeb2 2005-07-22 devnull putblkmany(int ar, int blks)
386 031ddeb2 2005-07-22 devnull curblk += blks - 1;
387 031ddeb2 2005-07-22 devnull putblk(ar);
391 031ddeb2 2005-07-22 devnull * common routines
395 031ddeb2 2005-07-22 devnull * modifies hp->chksum but restores it; important for the last block of the
396 031ddeb2 2005-07-22 devnull * old archive when updating with `tar rf archive'
398 6019a96c 2006-03-30 devnull static long
399 031ddeb2 2005-07-22 devnull chksum(Hdr *hp)
401 031ddeb2 2005-07-22 devnull int n = Tblock;
402 031ddeb2 2005-07-22 devnull long i = 0;
403 61309328 2006-03-31 devnull uchar *cp = (uchar*)hp;
404 031ddeb2 2005-07-22 devnull char oldsum[sizeof hp->chksum];
406 031ddeb2 2005-07-22 devnull memmove(oldsum, hp->chksum, sizeof oldsum);
407 031ddeb2 2005-07-22 devnull memset(hp->chksum, ' ', sizeof hp->chksum);
408 031ddeb2 2005-07-22 devnull while (n-- > 0)
409 031ddeb2 2005-07-22 devnull i += *cp++;
410 031ddeb2 2005-07-22 devnull memmove(hp->chksum, oldsum, sizeof oldsum);
411 031ddeb2 2005-07-22 devnull return i;
414 031ddeb2 2005-07-22 devnull static int
415 031ddeb2 2005-07-22 devnull isustar(Hdr *hp)
417 031ddeb2 2005-07-22 devnull return strcmp(hp->magic, "ustar") == 0;
421 031ddeb2 2005-07-22 devnull * s is at most n bytes long, but need not be NUL-terminated.
422 031ddeb2 2005-07-22 devnull * if shorter than n bytes, all bytes after the first NUL must also
423 031ddeb2 2005-07-22 devnull * be NUL.
425 031ddeb2 2005-07-22 devnull static int
426 031ddeb2 2005-07-22 devnull strnlen(char *s, int n)
428 031ddeb2 2005-07-22 devnull return s[n - 1] != '\0'? n: strlen(s);
431 031ddeb2 2005-07-22 devnull /* set fullname from header */
432 031ddeb2 2005-07-22 devnull static char *
433 031ddeb2 2005-07-22 devnull name(Hdr *hp)
435 031ddeb2 2005-07-22 devnull int pfxlen, namlen;
436 031ddeb2 2005-07-22 devnull static char fullnamebuf[2 + Maxname + 1]; /* 2 at beginning for ./ on relative names */
437 031ddeb2 2005-07-22 devnull char *fullname;
439 031ddeb2 2005-07-22 devnull fullname = fullnamebuf+2;
440 031ddeb2 2005-07-22 devnull namlen = strnlen(hp->name, sizeof hp->name);
441 031ddeb2 2005-07-22 devnull if (hp->prefix[0] == '\0' || !isustar(hp)) { /* old-style name? */
442 031ddeb2 2005-07-22 devnull memmove(fullname, hp->name, namlen);
443 031ddeb2 2005-07-22 devnull fullname[namlen] = '\0';
444 031ddeb2 2005-07-22 devnull return fullname;
447 031ddeb2 2005-07-22 devnull /* name is in two pieces */
448 031ddeb2 2005-07-22 devnull pfxlen = strnlen(hp->prefix, sizeof hp->prefix);
449 031ddeb2 2005-07-22 devnull memmove(fullname, hp->prefix, pfxlen);
450 031ddeb2 2005-07-22 devnull fullname[pfxlen] = '/';
451 031ddeb2 2005-07-22 devnull memmove(fullname + pfxlen + 1, hp->name, namlen);
452 031ddeb2 2005-07-22 devnull fullname[pfxlen + 1 + namlen] = '\0';
453 031ddeb2 2005-07-22 devnull return fullname;
456 031ddeb2 2005-07-22 devnull static int
457 031ddeb2 2005-07-22 devnull isdir(Hdr *hp)
459 031ddeb2 2005-07-22 devnull /* the mode test is ugly but sometimes necessary */
460 031ddeb2 2005-07-22 devnull return hp->linkflag == LF_DIR ||
461 031ddeb2 2005-07-22 devnull strrchr(name(hp), '\0')[-1] == '/' ||
462 031ddeb2 2005-07-22 devnull (strtoul(hp->mode, nil, 8)&0170000) == 040000;
465 031ddeb2 2005-07-22 devnull static int
466 031ddeb2 2005-07-22 devnull eotar(Hdr *hp)
468 031ddeb2 2005-07-22 devnull return name(hp)[0] == '\0';
472 6019a96c 2006-03-30 devnull static uvlong
473 6019a96c 2006-03-30 devnull getbe(uchar *src, int size)
475 6019a96c 2006-03-30 devnull uvlong vl = 0;
477 6019a96c 2006-03-30 devnull while (size-- > 0) {
478 6019a96c 2006-03-30 devnull vl <<= 8;
479 6019a96c 2006-03-30 devnull vl |= *src++;
481 6019a96c 2006-03-30 devnull return vl;
485 6019a96c 2006-03-30 devnull static void
486 6019a96c 2006-03-30 devnull putbe(uchar *dest, uvlong vl, int size)
488 6019a96c 2006-03-30 devnull for (dest += size; size-- > 0; vl >>= 8)
489 6019a96c 2006-03-30 devnull *--dest = vl;
493 6019a96c 2006-03-30 devnull * return the nominal size from the header block, which is not always the
494 6019a96c 2006-03-30 devnull * size in the archive (the archive size may be zero for some file types
495 6019a96c 2006-03-30 devnull * regardless of the nominal size).
497 6019a96c 2006-03-30 devnull * gnu and freebsd tars are now recording vlongs as big-endian binary
498 6019a96c 2006-03-30 devnull * with a flag in byte 0 to indicate this, which permits file sizes up to
499 6019a96c 2006-03-30 devnull * 2^64-1 (actually 2^80-1 but our file sizes are vlongs) rather than 2^33-1.
501 6019a96c 2006-03-30 devnull static Off
502 031ddeb2 2005-07-22 devnull hdrsize(Hdr *hp)
504 6019a96c 2006-03-30 devnull uchar *p;
506 6019a96c 2006-03-30 devnull if((uchar)hp->size[0] == Binnegsz) {
507 6019a96c 2006-03-30 devnull fprint(2, "%s: %s: negative length, which is insane\n",
508 6019a96c 2006-03-30 devnull argv0, name(hp));
509 6019a96c 2006-03-30 devnull return 0;
510 6019a96c 2006-03-30 devnull } else if((uchar)hp->size[0] == Binsize) {
511 6019a96c 2006-03-30 devnull p = (uchar *)hp->size + sizeof hp->size - 1 -
512 6019a96c 2006-03-30 devnull sizeof(vlong); /* -1 for terminating space */
513 6019a96c 2006-03-30 devnull return G8BEBYTE(p);
515 6019a96c 2006-03-30 devnull return strtoull(hp->size, nil, 8);
519 6019a96c 2006-03-30 devnull * return the number of bytes recorded in the archive.
521 6019a96c 2006-03-30 devnull static Off
522 6019a96c 2006-03-30 devnull arsize(Hdr *hp)
524 6019a96c 2006-03-30 devnull if(isdir(hp) || islink(hp->linkflag))
525 6019a96c 2006-03-30 devnull return 0;
526 6019a96c 2006-03-30 devnull return hdrsize(hp);
529 031ddeb2 2005-07-22 devnull static Hdr *
530 031ddeb2 2005-07-22 devnull readhdr(int ar)
532 031ddeb2 2005-07-22 devnull long hdrcksum;
533 031ddeb2 2005-07-22 devnull Hdr *hp;
535 031ddeb2 2005-07-22 devnull hp = getblkrd(ar, Alldata);
536 031ddeb2 2005-07-22 devnull if (hp == nil)
537 031ddeb2 2005-07-22 devnull sysfatal("unexpected EOF instead of archive header");
538 031ddeb2 2005-07-22 devnull if (eotar(hp)) /* end-of-archive block? */
539 031ddeb2 2005-07-22 devnull return nil;
540 031ddeb2 2005-07-22 devnull hdrcksum = strtoul(hp->chksum, nil, 8);
541 031ddeb2 2005-07-22 devnull if (chksum(hp) != hdrcksum)
542 031ddeb2 2005-07-22 devnull sysfatal("bad archive header checksum: name %.64s...",
543 031ddeb2 2005-07-22 devnull hp->name);
544 6019a96c 2006-03-30 devnull nexthdr += Tblock*(1 + BYTES2TBLKS(arsize(hp)));
545 031ddeb2 2005-07-22 devnull return hp;
549 031ddeb2 2005-07-22 devnull * tar r[c]
553 031ddeb2 2005-07-22 devnull * if name is longer than Namsiz bytes, try to split it at a slash and fit the
554 031ddeb2 2005-07-22 devnull * pieces into hp->prefix and hp->name.
556 031ddeb2 2005-07-22 devnull static int
557 031ddeb2 2005-07-22 devnull putfullname(Hdr *hp, char *name)
559 031ddeb2 2005-07-22 devnull int namlen, pfxlen;
560 031ddeb2 2005-07-22 devnull char *sl, *osl;
561 031ddeb2 2005-07-22 devnull String *slname = nil;
563 031ddeb2 2005-07-22 devnull if (isdir(hp)) {
564 031ddeb2 2005-07-22 devnull slname = s_new();
565 031ddeb2 2005-07-22 devnull s_append(slname, name);
566 031ddeb2 2005-07-22 devnull s_append(slname, "/"); /* posix requires this */
567 031ddeb2 2005-07-22 devnull name = s_to_c(slname);
570 031ddeb2 2005-07-22 devnull namlen = strlen(name);
571 031ddeb2 2005-07-22 devnull if (namlen <= Namsiz) {
572 031ddeb2 2005-07-22 devnull strncpy(hp->name, name, Namsiz);
573 031ddeb2 2005-07-22 devnull hp->prefix[0] = '\0'; /* ustar paranoia */
574 031ddeb2 2005-07-22 devnull return 0;
577 031ddeb2 2005-07-22 devnull if (!posix || namlen > Maxname) {
578 031ddeb2 2005-07-22 devnull fprint(2, "%s: name too long for tar header: %s\n",
579 031ddeb2 2005-07-22 devnull argv0, name);
580 031ddeb2 2005-07-22 devnull return -1;
583 031ddeb2 2005-07-22 devnull * try various splits until one results in pieces that fit into the
584 031ddeb2 2005-07-22 devnull * appropriate fields of the header. look for slashes from right
585 031ddeb2 2005-07-22 devnull * to left, in the hopes of putting the largest part of the name into
586 031ddeb2 2005-07-22 devnull * hp->prefix, which is larger than hp->name.
588 031ddeb2 2005-07-22 devnull sl = strrchr(name, '/');
589 031ddeb2 2005-07-22 devnull while (sl != nil) {
590 031ddeb2 2005-07-22 devnull pfxlen = sl - name;
591 031ddeb2 2005-07-22 devnull if (pfxlen <= sizeof hp->prefix && namlen-1 - pfxlen <= Namsiz)
593 031ddeb2 2005-07-22 devnull osl = sl;
594 031ddeb2 2005-07-22 devnull *osl = '\0';
595 031ddeb2 2005-07-22 devnull sl = strrchr(name, '/');
596 031ddeb2 2005-07-22 devnull *osl = '/';
598 031ddeb2 2005-07-22 devnull if (sl == nil) {
599 031ddeb2 2005-07-22 devnull fprint(2, "%s: name can't be split to fit tar header: %s\n",
600 031ddeb2 2005-07-22 devnull argv0, name);
601 031ddeb2 2005-07-22 devnull return -1;
603 031ddeb2 2005-07-22 devnull *sl = '\0';
604 031ddeb2 2005-07-22 devnull strncpy(hp->prefix, name, sizeof hp->prefix);
605 031ddeb2 2005-07-22 devnull *sl++ = '/';
606 031ddeb2 2005-07-22 devnull strncpy(hp->name, sl, sizeof hp->name);
607 031ddeb2 2005-07-22 devnull if (slname)
608 031ddeb2 2005-07-22 devnull s_free(slname);
609 031ddeb2 2005-07-22 devnull return 0;
612 031ddeb2 2005-07-22 devnull static int
613 031ddeb2 2005-07-22 devnull mkhdr(Hdr *hp, Dir *dir, char *file)
616 031ddeb2 2005-07-22 devnull * these fields run together, so we format them in order and don't use
617 031ddeb2 2005-07-22 devnull * snprint.
619 031ddeb2 2005-07-22 devnull sprint(hp->mode, "%6lo ", dir->mode & 0777);
620 031ddeb2 2005-07-22 devnull sprint(hp->uid, "%6o ", aruid);
621 031ddeb2 2005-07-22 devnull sprint(hp->gid, "%6o ", argid);
622 6019a96c 2006-03-30 devnull if (dir->length >= (Off)1<<32) {
623 6019a96c 2006-03-30 devnull static int printed;
625 6019a96c 2006-03-30 devnull if (!printed) {
626 6019a96c 2006-03-30 devnull printed = 1;
627 6019a96c 2006-03-30 devnull fprint(2, "%s: storing large sizes in \"base 256\"\n", argv0);
629 cbeb0b26 2006-04-01 devnull hp->size[0] = (char)Binsize;
630 6019a96c 2006-03-30 devnull /* emit so-called `base 256' representation of size */
631 6019a96c 2006-03-30 devnull putbe((uchar *)hp->size+1, dir->length, sizeof hp->size - 2);
632 6019a96c 2006-03-30 devnull hp->size[sizeof hp->size - 1] = ' ';
634 6019a96c 2006-03-30 devnull sprint(hp->size, "%11lluo ", dir->length);
635 031ddeb2 2005-07-22 devnull sprint(hp->mtime, "%11luo ", dir->mtime);
636 031ddeb2 2005-07-22 devnull hp->linkflag = (dir->mode&DMDIR? LF_DIR: LF_PLAIN1);
637 031ddeb2 2005-07-22 devnull putfullname(hp, file);
638 031ddeb2 2005-07-22 devnull if (posix) {
639 031ddeb2 2005-07-22 devnull strncpy(hp->magic, "ustar", sizeof hp->magic);
640 031ddeb2 2005-07-22 devnull strncpy(hp->version, "00", sizeof hp->version);
641 031ddeb2 2005-07-22 devnull strncpy(hp->uname, dir->uid, sizeof hp->uname);
642 031ddeb2 2005-07-22 devnull strncpy(hp->gname, dir->gid, sizeof hp->gname);
644 031ddeb2 2005-07-22 devnull sprint(hp->chksum, "%6luo", chksum(hp));
645 031ddeb2 2005-07-22 devnull return 0;
648 031ddeb2 2005-07-22 devnull static void addtoar(int ar, char *file, char *shortf);
650 031ddeb2 2005-07-22 devnull static void
651 031ddeb2 2005-07-22 devnull addtreetoar(int ar, char *file, char *shortf, int fd)
654 031ddeb2 2005-07-22 devnull Dir *dent, *dirents;
655 031ddeb2 2005-07-22 devnull String *name = s_new();
657 031ddeb2 2005-07-22 devnull n = dirreadall(fd, &dirents);
658 031ddeb2 2005-07-22 devnull close(fd);
659 031ddeb2 2005-07-22 devnull if (n == 0)
662 031ddeb2 2005-07-22 devnull if (chdir(shortf) < 0)
663 031ddeb2 2005-07-22 devnull sysfatal("chdir %s: %r", file);
664 031ddeb2 2005-07-22 devnull if (DEBUG)
665 031ddeb2 2005-07-22 devnull fprint(2, "chdir %s\t# %s\n", shortf, file);
667 031ddeb2 2005-07-22 devnull for (dent = dirents; dent < dirents + n; dent++) {
668 031ddeb2 2005-07-22 devnull s_reset(name);
669 031ddeb2 2005-07-22 devnull s_append(name, file);
670 031ddeb2 2005-07-22 devnull s_append(name, "/");
671 031ddeb2 2005-07-22 devnull s_append(name, dent->name);
672 031ddeb2 2005-07-22 devnull addtoar(ar, s_to_c(name), dent->name);
674 031ddeb2 2005-07-22 devnull s_free(name);
675 031ddeb2 2005-07-22 devnull free(dirents);
678 031ddeb2 2005-07-22 devnull * this assumes that shortf is just one component, which is true
679 031ddeb2 2005-07-22 devnull * during directory descent, but not necessarily true of command-line
680 031ddeb2 2005-07-22 devnull * arguments. Our caller (or addtoar's) must reset the working
681 031ddeb2 2005-07-22 devnull * directory if necessary.
683 031ddeb2 2005-07-22 devnull if (chdir("..") < 0)
684 031ddeb2 2005-07-22 devnull sysfatal("chdir %s/..: %r", file);
685 031ddeb2 2005-07-22 devnull if (DEBUG)
686 031ddeb2 2005-07-22 devnull fprint(2, "chdir ..\n");
689 031ddeb2 2005-07-22 devnull static void
690 031ddeb2 2005-07-22 devnull addtoar(int ar, char *file, char *shortf)
692 031ddeb2 2005-07-22 devnull int n, fd, isdir;
693 031ddeb2 2005-07-22 devnull long bytes;
694 031ddeb2 2005-07-22 devnull ulong blksleft, blksread;
695 031ddeb2 2005-07-22 devnull Hdr *hbp;
696 031ddeb2 2005-07-22 devnull Dir *dir;
697 6019a96c 2006-03-30 devnull String *name = nil;
699 6019a96c 2006-03-30 devnull if (shortf[0] == '#') {
700 6019a96c 2006-03-30 devnull name = s_new();
701 6019a96c 2006-03-30 devnull s_append(name, "./");
702 6019a96c 2006-03-30 devnull s_append(name, shortf);
703 6019a96c 2006-03-30 devnull shortf = s_to_c(name);
706 031ddeb2 2005-07-22 devnull fd = open(shortf, OREAD);
707 031ddeb2 2005-07-22 devnull if (fd < 0) {
708 031ddeb2 2005-07-22 devnull fprint(2, "%s: can't open %s: %r\n", argv0, file);
709 6019a96c 2006-03-30 devnull if (name)
710 6019a96c 2006-03-30 devnull s_free(name);
713 031ddeb2 2005-07-22 devnull dir = dirfstat(fd);
714 031ddeb2 2005-07-22 devnull if (dir == nil)
715 031ddeb2 2005-07-22 devnull sysfatal("can't fstat %s: %r", file);
717 031ddeb2 2005-07-22 devnull hbp = getblkz(ar);
718 031ddeb2 2005-07-22 devnull isdir = !!(dir->qid.type&QTDIR);
719 031ddeb2 2005-07-22 devnull if (mkhdr(hbp, dir, file) < 0) {
720 031ddeb2 2005-07-22 devnull putbackblk(ar);
721 031ddeb2 2005-07-22 devnull free(dir);
722 031ddeb2 2005-07-22 devnull close(fd);
723 6019a96c 2006-03-30 devnull if (name)
724 6019a96c 2006-03-30 devnull s_free(name);
727 031ddeb2 2005-07-22 devnull putblk(ar);
729 031ddeb2 2005-07-22 devnull blksleft = BYTES2TBLKS(dir->length);
730 031ddeb2 2005-07-22 devnull free(dir);
732 031ddeb2 2005-07-22 devnull if (isdir)
733 031ddeb2 2005-07-22 devnull addtreetoar(ar, file, shortf, fd);
735 031ddeb2 2005-07-22 devnull for (; blksleft > 0; blksleft -= blksread) {
736 031ddeb2 2005-07-22 devnull hbp = getblke(ar);
737 031ddeb2 2005-07-22 devnull blksread = gothowmany(blksleft);
738 031ddeb2 2005-07-22 devnull bytes = blksread * Tblock;
739 61309328 2006-03-31 devnull n = readn(fd, hbp, bytes);
740 031ddeb2 2005-07-22 devnull if (n < 0)
741 031ddeb2 2005-07-22 devnull sysfatal("error reading %s: %r", file);
743 031ddeb2 2005-07-22 devnull * ignore EOF. zero any partial block to aid
744 031ddeb2 2005-07-22 devnull * compression and emergency recovery of data.
746 031ddeb2 2005-07-22 devnull if (n < Tblock)
747 61309328 2006-03-31 devnull memset((uchar*)hbp + n, 0, bytes - n);
748 031ddeb2 2005-07-22 devnull putblkmany(ar, blksread);
750 031ddeb2 2005-07-22 devnull close(fd);
751 031ddeb2 2005-07-22 devnull if (verbose)
752 031ddeb2 2005-07-22 devnull fprint(2, "%s\n", file);
754 6019a96c 2006-03-30 devnull if (name)
755 6019a96c 2006-03-30 devnull s_free(name);
758 031ddeb2 2005-07-22 devnull static char *
759 031ddeb2 2005-07-22 devnull replace(char **argv)
761 031ddeb2 2005-07-22 devnull int i, ar;
762 031ddeb2 2005-07-22 devnull ulong blksleft, blksread;
763 031ddeb2 2005-07-22 devnull Off bytes;
764 031ddeb2 2005-07-22 devnull Hdr *hp;
765 031ddeb2 2005-07-22 devnull Compress *comp = nil;
766 031ddeb2 2005-07-22 devnull Pushstate ps;
768 031ddeb2 2005-07-22 devnull if (usefile && docreate) {
769 031ddeb2 2005-07-22 devnull ar = create(usefile, OWRITE, 0666);
770 031ddeb2 2005-07-22 devnull if (docompress)
771 031ddeb2 2005-07-22 devnull comp = compmethod(usefile);
772 031ddeb2 2005-07-22 devnull } else if (usefile)
773 031ddeb2 2005-07-22 devnull ar = open(usefile, ORDWR);
775 031ddeb2 2005-07-22 devnull ar = Stdout;
776 031ddeb2 2005-07-22 devnull if (comp)
777 031ddeb2 2005-07-22 devnull ar = push(ar, comp->comp, Output, &ps);
778 031ddeb2 2005-07-22 devnull if (ar < 0)
779 031ddeb2 2005-07-22 devnull sysfatal("can't open archive %s: %r", usefile);
781 031ddeb2 2005-07-22 devnull if (usefile && !docreate) {
782 031ddeb2 2005-07-22 devnull /* skip quickly to the end */
783 031ddeb2 2005-07-22 devnull while ((hp = readhdr(ar)) != nil) {
784 6019a96c 2006-03-30 devnull bytes = arsize(hp);
785 031ddeb2 2005-07-22 devnull for (blksleft = BYTES2TBLKS(bytes);
786 031ddeb2 2005-07-22 devnull blksleft > 0 && getblkrd(ar, Justnxthdr) != nil;
787 031ddeb2 2005-07-22 devnull blksleft -= blksread) {
788 031ddeb2 2005-07-22 devnull blksread = gothowmany(blksleft);
789 031ddeb2 2005-07-22 devnull putreadblks(ar, blksread);
793 031ddeb2 2005-07-22 devnull * we have just read the end-of-archive Tblock.
794 031ddeb2 2005-07-22 devnull * now seek back over the (big) archive block containing it,
795 031ddeb2 2005-07-22 devnull * and back up curblk ptr over end-of-archive Tblock in memory.
797 031ddeb2 2005-07-22 devnull if (seek(ar, blkoff, 0) < 0)
798 031ddeb2 2005-07-22 devnull sysfatal("can't seek back over end-of-archive: %r");
799 031ddeb2 2005-07-22 devnull curblk--;
802 031ddeb2 2005-07-22 devnull for (i = 0; argv[i] != nil; i++) {
803 031ddeb2 2005-07-22 devnull addtoar(ar, argv[i], argv[i]);
804 031ddeb2 2005-07-22 devnull chdir(origdir); /* for correctness & profiling */
807 031ddeb2 2005-07-22 devnull /* write end-of-archive marker */
808 031ddeb2 2005-07-22 devnull getblkz(ar);
809 031ddeb2 2005-07-22 devnull putblk(ar);
810 031ddeb2 2005-07-22 devnull getblkz(ar);
811 031ddeb2 2005-07-22 devnull putlastblk(ar);
813 031ddeb2 2005-07-22 devnull if (comp)
814 031ddeb2 2005-07-22 devnull return pushclose(&ps);
815 031ddeb2 2005-07-22 devnull if (ar > Stderr)
816 031ddeb2 2005-07-22 devnull close(ar);
817 031ddeb2 2005-07-22 devnull return nil;
821 031ddeb2 2005-07-22 devnull * tar [xt]
824 031ddeb2 2005-07-22 devnull /* is pfx a file-name prefix of name? */
825 031ddeb2 2005-07-22 devnull static int
826 031ddeb2 2005-07-22 devnull prefix(char *name, char *pfx)
828 031ddeb2 2005-07-22 devnull int pfxlen = strlen(pfx);
829 031ddeb2 2005-07-22 devnull char clpfx[Maxname+1];
831 031ddeb2 2005-07-22 devnull if (pfxlen > Maxname)
832 031ddeb2 2005-07-22 devnull return 0;
833 031ddeb2 2005-07-22 devnull strcpy(clpfx, pfx);
834 031ddeb2 2005-07-22 devnull cleanname(clpfx);
835 031ddeb2 2005-07-22 devnull return strncmp(pfx, name, pfxlen) == 0 &&
836 031ddeb2 2005-07-22 devnull (name[pfxlen] == '\0' || name[pfxlen] == '/');
839 031ddeb2 2005-07-22 devnull static int
840 031ddeb2 2005-07-22 devnull match(char *name, char **argv)
843 031ddeb2 2005-07-22 devnull char clname[Maxname+1];
845 031ddeb2 2005-07-22 devnull if (argv[0] == nil)
846 031ddeb2 2005-07-22 devnull return 1;
847 031ddeb2 2005-07-22 devnull strcpy(clname, name);
848 031ddeb2 2005-07-22 devnull cleanname(clname);
849 031ddeb2 2005-07-22 devnull for (i = 0; argv[i] != nil; i++)
850 031ddeb2 2005-07-22 devnull if (prefix(clname, argv[i]))
851 031ddeb2 2005-07-22 devnull return 1;
852 031ddeb2 2005-07-22 devnull return 0;
855 6019a96c 2006-03-30 devnull static void
856 6019a96c 2006-03-30 devnull cantcreate(char *s, int mode)
858 6019a96c 2006-03-30 devnull int len;
859 6019a96c 2006-03-30 devnull static char *last;
862 6019a96c 2006-03-30 devnull * Always print about files. Only print about directories
863 6019a96c 2006-03-30 devnull * we haven't printed about. (Assumes archive is ordered
864 6019a96c 2006-03-30 devnull * nicely.)
866 6019a96c 2006-03-30 devnull if(mode&DMDIR){
867 6019a96c 2006-03-30 devnull if(last){
868 6019a96c 2006-03-30 devnull /* already printed this directory */
869 6019a96c 2006-03-30 devnull if(strcmp(s, last) == 0)
871 6019a96c 2006-03-30 devnull /* printed a higher directory, so printed this one */
872 6019a96c 2006-03-30 devnull len = strlen(s);
873 6019a96c 2006-03-30 devnull if(memcmp(s, last, len) == 0 && last[len] == '/')
876 6019a96c 2006-03-30 devnull /* save */
877 6019a96c 2006-03-30 devnull free(last);
878 6019a96c 2006-03-30 devnull last = strdup(s);
880 6019a96c 2006-03-30 devnull fprint(2, "%s: can't create %s: %r\n", argv0, s);
883 031ddeb2 2005-07-22 devnull static int
884 031ddeb2 2005-07-22 devnull makedir(char *s)
888 031ddeb2 2005-07-22 devnull if (access(s, AEXIST) == 0)
889 031ddeb2 2005-07-22 devnull return -1;
890 031ddeb2 2005-07-22 devnull f = create(s, OREAD, DMDIR | 0777);
891 031ddeb2 2005-07-22 devnull if (f >= 0)
892 031ddeb2 2005-07-22 devnull close(f);
894 6019a96c 2006-03-30 devnull cantcreate(s, DMDIR);
895 031ddeb2 2005-07-22 devnull return f;
898 6019a96c 2006-03-30 devnull static int
899 031ddeb2 2005-07-22 devnull mkpdirs(char *s)
901 6019a96c 2006-03-30 devnull int err;
902 6019a96c 2006-03-30 devnull char *p;
905 6019a96c 2006-03-30 devnull err = 0;
906 6019a96c 2006-03-30 devnull while (!err && (p = strchr(p+1, '/')) != nil) {
907 031ddeb2 2005-07-22 devnull *p = '\0';
908 6019a96c 2006-03-30 devnull err = (access(s, AEXIST) < 0 && makedir(s) < 0);
909 031ddeb2 2005-07-22 devnull *p = '/';
911 6019a96c 2006-03-30 devnull return -err;
914 6019a96c 2006-03-30 devnull /* Call access but preserve the error string. */
915 6019a96c 2006-03-30 devnull static int
916 6019a96c 2006-03-30 devnull xaccess(char *name, int mode)
918 6019a96c 2006-03-30 devnull char err[ERRMAX];
921 6019a96c 2006-03-30 devnull err[0] = 0;
922 6019a96c 2006-03-30 devnull errstr(err, sizeof err);
923 6019a96c 2006-03-30 devnull rv = access(name, mode);
924 6019a96c 2006-03-30 devnull errstr(err, sizeof err);
925 6019a96c 2006-03-30 devnull return rv;
928 031ddeb2 2005-07-22 devnull /* copy a file from the archive into the filesystem */
929 031ddeb2 2005-07-22 devnull /* fname is result of name(), so has two extra bytes at beginning */
930 031ddeb2 2005-07-22 devnull static void
931 031ddeb2 2005-07-22 devnull extract1(int ar, Hdr *hp, char *fname)
933 031ddeb2 2005-07-22 devnull int wrbytes, fd = -1, dir = 0;
934 031ddeb2 2005-07-22 devnull long mtime = strtol(hp->mtime, nil, 8);
935 031ddeb2 2005-07-22 devnull ulong mode = strtoul(hp->mode, nil, 8) & 0777;
936 6019a96c 2006-03-30 devnull Off bytes = hdrsize(hp); /* for printing */
937 6019a96c 2006-03-30 devnull ulong blksread, blksleft = BYTES2TBLKS(arsize(hp));
938 031ddeb2 2005-07-22 devnull Hdr *hbp;
940 031ddeb2 2005-07-22 devnull if (isdir(hp)) {
941 031ddeb2 2005-07-22 devnull mode |= DMDIR|0700;
942 031ddeb2 2005-07-22 devnull dir = 1;
944 031ddeb2 2005-07-22 devnull switch (hp->linkflag) {
945 031ddeb2 2005-07-22 devnull case LF_LINK:
946 031ddeb2 2005-07-22 devnull case LF_SYMLINK1:
947 031ddeb2 2005-07-22 devnull case LF_SYMLINK2:
948 031ddeb2 2005-07-22 devnull case LF_FIFO:
949 031ddeb2 2005-07-22 devnull blksleft = 0;
952 031ddeb2 2005-07-22 devnull if (relative) {
953 031ddeb2 2005-07-22 devnull if(fname[0] == '/')
954 031ddeb2 2005-07-22 devnull *--fname = '.';
955 031ddeb2 2005-07-22 devnull else if(fname[0] == '#'){
956 031ddeb2 2005-07-22 devnull *--fname = '/';
957 031ddeb2 2005-07-22 devnull *--fname = '.';
960 031ddeb2 2005-07-22 devnull if (verb == Xtract) {
961 031ddeb2 2005-07-22 devnull cleanname(fname);
962 031ddeb2 2005-07-22 devnull switch (hp->linkflag) {
963 031ddeb2 2005-07-22 devnull case LF_LINK:
964 031ddeb2 2005-07-22 devnull case LF_SYMLINK1:
965 031ddeb2 2005-07-22 devnull case LF_SYMLINK2:
966 031ddeb2 2005-07-22 devnull fprint(2, "%s: can't make (sym)link %s\n",
967 031ddeb2 2005-07-22 devnull argv0, fname);
969 031ddeb2 2005-07-22 devnull case LF_FIFO:
970 031ddeb2 2005-07-22 devnull fprint(2, "%s: can't make fifo %s\n", argv0, fname);
972 031ddeb2 2005-07-22 devnull default:
973 031ddeb2 2005-07-22 devnull if (!keepexisting || access(fname, AEXIST) < 0) {
974 031ddeb2 2005-07-22 devnull int rw = (dir? OREAD: OWRITE);
976 031ddeb2 2005-07-22 devnull fd = create(fname, rw, mode);
977 031ddeb2 2005-07-22 devnull if (fd < 0) {
978 031ddeb2 2005-07-22 devnull mkpdirs(fname);
979 031ddeb2 2005-07-22 devnull fd = create(fname, rw, mode);
981 031ddeb2 2005-07-22 devnull if (fd < 0 &&
982 6019a96c 2006-03-30 devnull (!dir || xaccess(fname, AEXIST) < 0))
983 6019a96c 2006-03-30 devnull cantcreate(fname, mode);
985 031ddeb2 2005-07-22 devnull if (fd >= 0 && verbose)
986 031ddeb2 2005-07-22 devnull fprint(2, "%s\n", fname);
989 031ddeb2 2005-07-22 devnull } else if (verbose) {
990 031ddeb2 2005-07-22 devnull char *cp = ctime(mtime);
992 031ddeb2 2005-07-22 devnull print("%M %8lld %-12.12s %-4.4s %s\n",
993 031ddeb2 2005-07-22 devnull mode, bytes, cp+4, cp+24, fname);
995 031ddeb2 2005-07-22 devnull print("%s\n", fname);
997 6019a96c 2006-03-30 devnull if (blksleft == 0)
998 6019a96c 2006-03-30 devnull bytes = 0;
999 031ddeb2 2005-07-22 devnull for (; blksleft > 0; blksleft -= blksread) {
1000 031ddeb2 2005-07-22 devnull hbp = getblkrd(ar, (fd >= 0? Alldata: Justnxthdr));
1001 031ddeb2 2005-07-22 devnull if (hbp == nil)
1002 031ddeb2 2005-07-22 devnull sysfatal("unexpected EOF on archive extracting %s",
1003 031ddeb2 2005-07-22 devnull fname);
1004 031ddeb2 2005-07-22 devnull blksread = gothowmany(blksleft);
1005 6019a96c 2006-03-30 devnull if (blksread <= 0)
1006 6019a96c 2006-03-30 devnull fprint(2, "%s: got %ld blocks reading %s!\n",
1007 6019a96c 2006-03-30 devnull argv0, blksread, fname);
1008 031ddeb2 2005-07-22 devnull wrbytes = Tblock*blksread;
1009 031ddeb2 2005-07-22 devnull if(wrbytes > bytes)
1010 031ddeb2 2005-07-22 devnull wrbytes = bytes;
1011 61309328 2006-03-31 devnull if (fd >= 0 && write(fd, hbp, wrbytes) != wrbytes)
1012 031ddeb2 2005-07-22 devnull sysfatal("write error on %s: %r", fname);
1013 031ddeb2 2005-07-22 devnull putreadblks(ar, blksread);
1014 031ddeb2 2005-07-22 devnull bytes -= wrbytes;
1016 6019a96c 2006-03-30 devnull if (bytes > 0)
1017 6019a96c 2006-03-30 devnull fprint(2,
1018 6019a96c 2006-03-30 devnull "%s: %lld bytes uncopied at eof; %s not fully extracted\n",
1019 6019a96c 2006-03-30 devnull argv0, bytes, fname);
1020 031ddeb2 2005-07-22 devnull if (fd >= 0) {
1022 031ddeb2 2005-07-22 devnull * directories should be wstated after we're done
1023 031ddeb2 2005-07-22 devnull * creating files in them.
1025 031ddeb2 2005-07-22 devnull if (settime) {
1026 031ddeb2 2005-07-22 devnull Dir nd;
1028 031ddeb2 2005-07-22 devnull nulldir(&nd);
1029 031ddeb2 2005-07-22 devnull nd.mtime = mtime;
1030 031ddeb2 2005-07-22 devnull dirfwstat(fd, &nd);
1031 6019a96c 2006-03-30 devnull if (isustar(hp)) {
1032 6019a96c 2006-03-30 devnull nulldir(&nd);
1033 6019a96c 2006-03-30 devnull nd.gid = hp->gname;
1034 6019a96c 2006-03-30 devnull dirfwstat(fd, &nd);
1037 031ddeb2 2005-07-22 devnull close(fd);
1041 031ddeb2 2005-07-22 devnull static void
1042 031ddeb2 2005-07-22 devnull skip(int ar, Hdr *hp, char *fname)
1044 031ddeb2 2005-07-22 devnull ulong blksleft, blksread;
1045 031ddeb2 2005-07-22 devnull Hdr *hbp;
1047 6019a96c 2006-03-30 devnull for (blksleft = BYTES2TBLKS(arsize(hp)); blksleft > 0;
1048 031ddeb2 2005-07-22 devnull blksleft -= blksread) {
1049 031ddeb2 2005-07-22 devnull hbp = getblkrd(ar, Justnxthdr);
1050 031ddeb2 2005-07-22 devnull if (hbp == nil)
1051 031ddeb2 2005-07-22 devnull sysfatal("unexpected EOF on archive extracting %s",
1052 031ddeb2 2005-07-22 devnull fname);
1053 031ddeb2 2005-07-22 devnull blksread = gothowmany(blksleft);
1054 031ddeb2 2005-07-22 devnull putreadblks(ar, blksread);
1058 031ddeb2 2005-07-22 devnull static char *
1059 031ddeb2 2005-07-22 devnull extract(char **argv)
1061 031ddeb2 2005-07-22 devnull int ar;
1062 031ddeb2 2005-07-22 devnull char *longname;
1063 031ddeb2 2005-07-22 devnull Hdr *hp;
1064 031ddeb2 2005-07-22 devnull Compress *comp = nil;
1065 031ddeb2 2005-07-22 devnull Pushstate ps;
1067 031ddeb2 2005-07-22 devnull if (usefile) {
1068 031ddeb2 2005-07-22 devnull ar = open(usefile, OREAD);
1069 031ddeb2 2005-07-22 devnull comp = compmethod(usefile);
1071 031ddeb2 2005-07-22 devnull ar = Stdin;
1072 031ddeb2 2005-07-22 devnull if (comp)
1073 031ddeb2 2005-07-22 devnull ar = push(ar, comp->decomp, Input, &ps);
1074 031ddeb2 2005-07-22 devnull if (ar < 0)
1075 031ddeb2 2005-07-22 devnull sysfatal("can't open archive %s: %r", usefile);
1077 031ddeb2 2005-07-22 devnull while ((hp = readhdr(ar)) != nil) {
1078 031ddeb2 2005-07-22 devnull longname = name(hp);
1079 031ddeb2 2005-07-22 devnull if (match(longname, argv))
1080 031ddeb2 2005-07-22 devnull extract1(ar, hp, longname);
1082 031ddeb2 2005-07-22 devnull skip(ar, hp, longname);
1085 031ddeb2 2005-07-22 devnull if (comp)
1086 031ddeb2 2005-07-22 devnull return pushclose(&ps);
1087 031ddeb2 2005-07-22 devnull if (ar > Stderr)
1088 031ddeb2 2005-07-22 devnull close(ar);
1089 031ddeb2 2005-07-22 devnull return nil;
1093 031ddeb2 2005-07-22 devnull main(int argc, char *argv[])
1095 031ddeb2 2005-07-22 devnull int errflg = 0;
1096 031ddeb2 2005-07-22 devnull char *ret = nil;
1098 031ddeb2 2005-07-22 devnull fmtinstall('M', dirmodefmt);
1100 61309328 2006-03-31 devnull if(sizeof(Hdr) != Tblock)
1101 61309328 2006-03-31 devnull sysfatal("padding in hdr should be %d", Tblock-sizeof(Hdr)+sizeof(curblk->pad));
1102 031ddeb2 2005-07-22 devnull TARGBEGIN {
1103 031ddeb2 2005-07-22 devnull case 'c':
1104 031ddeb2 2005-07-22 devnull docreate++;
1105 031ddeb2 2005-07-22 devnull verb = Replace;
1107 031ddeb2 2005-07-22 devnull case 'f':
1108 031ddeb2 2005-07-22 devnull usefile = EARGF(usage());
1110 031ddeb2 2005-07-22 devnull case 'g':
1111 031ddeb2 2005-07-22 devnull argid = strtoul(EARGF(usage()), 0, 0);
1113 031ddeb2 2005-07-22 devnull case 'k':
1114 031ddeb2 2005-07-22 devnull keepexisting++;
1116 031ddeb2 2005-07-22 devnull case 'm': /* compatibility */
1117 031ddeb2 2005-07-22 devnull settime = 0;
1119 031ddeb2 2005-07-22 devnull case 'p':
1120 031ddeb2 2005-07-22 devnull posix++;
1122 031ddeb2 2005-07-22 devnull case 'P':
1123 031ddeb2 2005-07-22 devnull posix = 0;
1125 031ddeb2 2005-07-22 devnull case 'r':
1126 031ddeb2 2005-07-22 devnull verb = Replace;
1128 031ddeb2 2005-07-22 devnull case 'R':
1129 031ddeb2 2005-07-22 devnull relative = 0;
1131 031ddeb2 2005-07-22 devnull case 't':
1132 031ddeb2 2005-07-22 devnull verb = Toc;
1134 031ddeb2 2005-07-22 devnull case 'T':
1135 031ddeb2 2005-07-22 devnull settime++;
1137 031ddeb2 2005-07-22 devnull case 'u':
1138 031ddeb2 2005-07-22 devnull aruid = strtoul(EARGF(usage()), 0, 0);
1140 031ddeb2 2005-07-22 devnull case 'v':
1141 031ddeb2 2005-07-22 devnull verbose++;
1143 031ddeb2 2005-07-22 devnull case 'x':
1144 031ddeb2 2005-07-22 devnull verb = Xtract;
1146 031ddeb2 2005-07-22 devnull case 'z':
1147 031ddeb2 2005-07-22 devnull docompress++;
1149 031ddeb2 2005-07-22 devnull case '-':
1151 031ddeb2 2005-07-22 devnull default:
1152 031ddeb2 2005-07-22 devnull fprint(2, "tar: unknown letter %C\n", TARGC());
1153 031ddeb2 2005-07-22 devnull errflg++;
1155 031ddeb2 2005-07-22 devnull } TARGEND
1157 031ddeb2 2005-07-22 devnull if (argc < 0 || errflg)
1158 031ddeb2 2005-07-22 devnull usage();
1160 031ddeb2 2005-07-22 devnull initblks();
1161 031ddeb2 2005-07-22 devnull switch (verb) {
1162 031ddeb2 2005-07-22 devnull case Toc:
1163 031ddeb2 2005-07-22 devnull case Xtract:
1164 031ddeb2 2005-07-22 devnull ret = extract(argv);
1166 031ddeb2 2005-07-22 devnull case Replace:
1167 031ddeb2 2005-07-22 devnull if (getwd(origdir, sizeof origdir) == nil)
1168 031ddeb2 2005-07-22 devnull strcpy(origdir, "/tmp");
1169 031ddeb2 2005-07-22 devnull ret = replace(argv);
1171 031ddeb2 2005-07-22 devnull default:
1172 031ddeb2 2005-07-22 devnull usage();
1175 031ddeb2 2005-07-22 devnull exits(ret);
