2 7d59ed71 2005-02-11 devnull * tar - `tape archiver', actually usable on any medium.
3 7d59ed71 2005-02-11 devnull * POSIX "ustar" compliant when extracting, and by default when creating.
4 7d59ed71 2005-02-11 devnull * this tar attempts to read and write multiple Tblock-byte blocks
5 7d59ed71 2005-02-11 devnull * at once to and from the filesystem, and does not copy blocks
6 7d59ed71 2005-02-11 devnull * around internally.
9 bc7cb1a1 2003-11-23 devnull #include <u.h>
10 bc7cb1a1 2003-11-23 devnull #include <libc.h>
11 7d59ed71 2005-02-11 devnull #include <fcall.h> /* for %M */
12 7d59ed71 2005-02-11 devnull #include <libString.h>
14 7d59ed71 2005-02-11 devnull #define creat tar_creat
17 7d59ed71 2005-02-11 devnull * modified versions of those in libc.h; scans only the first arg for
18 7d59ed71 2005-02-11 devnull * keyletters and options.
20 7d59ed71 2005-02-11 devnull #define TARGBEGIN {\
21 f93c7b2d 2005-02-11 devnull __fixargv0(); if(!argv0) argv0=*argv; \
22 f93c7b2d 2005-02-11 devnull argv++, argc--; \
23 7d59ed71 2005-02-11 devnull if (argv[0]) {\
24 7d59ed71 2005-02-11 devnull char *_args, *_argt;\
25 7d59ed71 2005-02-11 devnull Rune _argc;\
26 7d59ed71 2005-02-11 devnull _args = &argv[0][0];\
27 7d59ed71 2005-02-11 devnull _argc = 0;\
28 7d59ed71 2005-02-11 devnull while(*_args && (_args += chartorune(&_argc, _args)))\
29 7d59ed71 2005-02-11 devnull switch(_argc)
30 7d59ed71 2005-02-11 devnull #define TARGEND SET(_argt); USED(_argt); USED(_argc); USED(_args); argc--, argv++; } \
31 7d59ed71 2005-02-11 devnull USED(argv);USED(argc); }
33 7d59ed71 2005-02-11 devnull #define ROUNDUP(a, b) (((a) + (b) - 1)/(b))
34 7d59ed71 2005-02-11 devnull #define BYTES2TBLKS(bytes) ROUNDUP(bytes, Tblock)
36 7d59ed71 2005-02-11 devnull typedef vlong Off;
37 7d59ed71 2005-02-11 devnull typedef char *(*Refill)(int ar, char *bufs);
39 7d59ed71 2005-02-11 devnull enum { Stdin, Stdout, Stderr };
40 7d59ed71 2005-02-11 devnull enum { None, Toc, Xtract, Replace };
42 7d59ed71 2005-02-11 devnull Tblock = 512,
43 7d59ed71 2005-02-11 devnull Nblock = 40, /* maximum blocksize */
44 7d59ed71 2005-02-11 devnull Dblock = 20, /* default blocksize */
45 7d59ed71 2005-02-11 devnull Namsiz = 100,
46 7d59ed71 2005-02-11 devnull Maxpfx = 155, /* from POSIX */
47 7d59ed71 2005-02-11 devnull Maxname = Namsiz + 1 + Maxpfx,
48 7d59ed71 2005-02-11 devnull DEBUG = 0,
51 7d59ed71 2005-02-11 devnull /* POSIX link flags */
53 7d59ed71 2005-02-11 devnull LF_PLAIN1 = '\0',
54 7d59ed71 2005-02-11 devnull LF_PLAIN2 = '0',
55 7d59ed71 2005-02-11 devnull LF_LINK = '1',
56 7d59ed71 2005-02-11 devnull LF_SYMLINK1 = '2',
57 7d59ed71 2005-02-11 devnull LF_SYMLINK2 = 's',
58 7d59ed71 2005-02-11 devnull LF_CHR = '3',
59 7d59ed71 2005-02-11 devnull LF_BLK = '4',
60 7d59ed71 2005-02-11 devnull LF_DIR = '5',
61 7d59ed71 2005-02-11 devnull LF_FIFO = '6',
62 7d59ed71 2005-02-11 devnull LF_CONTIG = '7',
63 7d59ed71 2005-02-11 devnull /* 'A' - 'Z' are reserved for custom implementations */
66 7d59ed71 2005-02-11 devnull #define islink(lf) (isreallink(lf) || issymlink(lf))
67 7d59ed71 2005-02-11 devnull #define isreallink(lf) ((lf) == LF_LINK)
68 7d59ed71 2005-02-11 devnull #define issymlink(lf) ((lf) == LF_SYMLINK1 || (lf) == LF_SYMLINK2)
70 7d59ed71 2005-02-11 devnull typedef union {
71 510f0dbf 2005-02-11 devnull char name[Namsiz];
72 510f0dbf 2005-02-11 devnull char mode[8];
73 510f0dbf 2005-02-11 devnull char uid[8];
74 510f0dbf 2005-02-11 devnull char gid[8];
75 510f0dbf 2005-02-11 devnull char size[12];
76 510f0dbf 2005-02-11 devnull char mtime[12];
77 510f0dbf 2005-02-11 devnull char chksum[8];
78 510f0dbf 2005-02-11 devnull char linkflag;
79 510f0dbf 2005-02-11 devnull char linkname[Namsiz];
81 510f0dbf 2005-02-11 devnull /* rest are defined by POSIX's ustar format; see p1003.2b */
82 510f0dbf 2005-02-11 devnull char magic[6]; /* "ustar" */
83 510f0dbf 2005-02-11 devnull char version[2];
84 510f0dbf 2005-02-11 devnull char uname[32];
85 510f0dbf 2005-02-11 devnull char gname[32];
86 510f0dbf 2005-02-11 devnull char devmajor[8];
87 510f0dbf 2005-02-11 devnull char devminor[8];
88 510f0dbf 2005-02-11 devnull char prefix[Maxpfx]; /* if non-null, path= prefix "/" name */
91 6c0209f6 2005-02-11 devnull int debug;
92 7d59ed71 2005-02-11 devnull static int verb;
93 7d59ed71 2005-02-11 devnull static int posix = 1;
94 7d59ed71 2005-02-11 devnull static int creat;
95 7d59ed71 2005-02-11 devnull static int aruid;
96 7d59ed71 2005-02-11 devnull static int argid;
97 7d59ed71 2005-02-11 devnull static int relative;
98 7d59ed71 2005-02-11 devnull static int settime;
99 7d59ed71 2005-02-11 devnull static int verbose;
101 7d59ed71 2005-02-11 devnull static int nblock = Dblock;
102 7d59ed71 2005-02-11 devnull static char *usefile;
103 7d59ed71 2005-02-11 devnull static char origdir[Maxname*2];
104 7d59ed71 2005-02-11 devnull static Hdr *tpblk, *endblk;
105 7d59ed71 2005-02-11 devnull static Hdr *curblk;
107 7d59ed71 2005-02-11 devnull static void
108 7d59ed71 2005-02-11 devnull usage(void)
110 7d59ed71 2005-02-11 devnull fprint(2, "usage: %s {txrc}[pPvRTugf] [archive] file1 file2...\n",
112 7d59ed71 2005-02-11 devnull exits("usage");
116 7d59ed71 2005-02-11 devnull * block-buffer management
119 7d59ed71 2005-02-11 devnull static void
120 7d59ed71 2005-02-11 devnull initblks(void)
122 7d59ed71 2005-02-11 devnull free(tpblk);
123 7d59ed71 2005-02-11 devnull tpblk = malloc(Tblock * nblock);
124 7d59ed71 2005-02-11 devnull assert(tpblk != nil);
125 7d59ed71 2005-02-11 devnull endblk = tpblk + nblock;
128 7d59ed71 2005-02-11 devnull static char *
129 7d59ed71 2005-02-11 devnull refill(int ar, char *bufs)
131 7d59ed71 2005-02-11 devnull int i, n;
132 7d59ed71 2005-02-11 devnull unsigned bytes = Tblock * nblock;
133 7d59ed71 2005-02-11 devnull static int done, first = 1;
135 7d59ed71 2005-02-11 devnull if (done)
136 7d59ed71 2005-02-11 devnull return nil;
138 7d59ed71 2005-02-11 devnull /* try to size non-pipe input at first read */
139 7d59ed71 2005-02-11 devnull if (first && usefile) {
140 7d59ed71 2005-02-11 devnull first = 0;
141 7d59ed71 2005-02-11 devnull n = read(ar, bufs, bytes);
142 7d59ed71 2005-02-11 devnull if (n <= 0)
143 7d59ed71 2005-02-11 devnull sysfatal("error reading archive: %r");
145 7d59ed71 2005-02-11 devnull if (i % Tblock != 0) {
146 7d59ed71 2005-02-11 devnull fprint(2, "%s: archive block size (%d) error\n",
147 7d59ed71 2005-02-11 devnull argv0, i);
148 7d59ed71 2005-02-11 devnull exits("blocksize");
150 7d59ed71 2005-02-11 devnull i /= Tblock;
151 7d59ed71 2005-02-11 devnull if (i != nblock) {
152 7d59ed71 2005-02-11 devnull nblock = i;
153 7d59ed71 2005-02-11 devnull fprint(2, "%s: blocking = %d\n", argv0, nblock);
154 7d59ed71 2005-02-11 devnull endblk = (Hdr *)bufs + nblock;
155 7d59ed71 2005-02-11 devnull bytes = n;
158 7d59ed71 2005-02-11 devnull n = readn(ar, bufs, bytes);
159 7d59ed71 2005-02-11 devnull if (n == 0)
160 7d59ed71 2005-02-11 devnull sysfatal("unexpected EOF reading archive");
161 7d59ed71 2005-02-11 devnull else if (n < 0)
162 7d59ed71 2005-02-11 devnull sysfatal("error reading archive: %r");
163 7d59ed71 2005-02-11 devnull else if (n%Tblock != 0)
164 7d59ed71 2005-02-11 devnull sysfatal("partial block read from archive");
165 7d59ed71 2005-02-11 devnull if (n != bytes) {
166 7d59ed71 2005-02-11 devnull done = 1;
167 7d59ed71 2005-02-11 devnull memset(bufs + n, 0, bytes - n);
169 7d59ed71 2005-02-11 devnull return bufs;
172 7d59ed71 2005-02-11 devnull static Hdr *
173 7d59ed71 2005-02-11 devnull getblk(int ar, Refill rfp)
175 7d59ed71 2005-02-11 devnull if (curblk == nil || curblk >= endblk) { /* input block exhausted? */
176 7d59ed71 2005-02-11 devnull if (rfp != nil && (*rfp)(ar, (char *)tpblk) == nil)
177 7d59ed71 2005-02-11 devnull return nil;
178 7d59ed71 2005-02-11 devnull curblk = tpblk;
180 7d59ed71 2005-02-11 devnull return curblk++;
183 7d59ed71 2005-02-11 devnull static Hdr *
184 7d59ed71 2005-02-11 devnull getblkrd(int ar)
186 7d59ed71 2005-02-11 devnull return getblk(ar, refill);
189 7d59ed71 2005-02-11 devnull static Hdr *
190 7d59ed71 2005-02-11 devnull getblke(int ar)
192 7d59ed71 2005-02-11 devnull return getblk(ar, nil);
195 7d59ed71 2005-02-11 devnull static Hdr *
196 7d59ed71 2005-02-11 devnull getblkz(int ar)
198 7d59ed71 2005-02-11 devnull Hdr *hp = getblke(ar);
200 7d59ed71 2005-02-11 devnull if (hp != nil)
201 510f0dbf 2005-02-11 devnull memset(hp, 0, Tblock);
202 7d59ed71 2005-02-11 devnull return hp;
206 7d59ed71 2005-02-11 devnull * how many block buffers are available, starting at the address
207 7d59ed71 2005-02-11 devnull * just returned by getblk*?
209 7d59ed71 2005-02-11 devnull static int
210 7d59ed71 2005-02-11 devnull gothowmany(int max)
212 7d59ed71 2005-02-11 devnull int n = endblk - (curblk - 1);
214 7d59ed71 2005-02-11 devnull return n > max? max: n;
218 7d59ed71 2005-02-11 devnull * indicate that one is done with the last block obtained from getblke
219 7d59ed71 2005-02-11 devnull * and it is now available to be written into the archive.
221 7d59ed71 2005-02-11 devnull static void
222 7d59ed71 2005-02-11 devnull putlastblk(int ar)
224 7d59ed71 2005-02-11 devnull unsigned bytes = Tblock * nblock;
226 7d59ed71 2005-02-11 devnull /* if writing end-of-archive, aid compression (good hygiene too) */
227 7d59ed71 2005-02-11 devnull if (curblk < endblk)
228 7d59ed71 2005-02-11 devnull memset(curblk, 0, (char *)endblk - (char *)curblk);
229 7d59ed71 2005-02-11 devnull if (write(ar, tpblk, bytes) != bytes)
230 7d59ed71 2005-02-11 devnull sysfatal("error writing archive: %r");
233 7d59ed71 2005-02-11 devnull static void
234 7d59ed71 2005-02-11 devnull putblk(int ar)
236 7d59ed71 2005-02-11 devnull if (curblk >= endblk)
237 7d59ed71 2005-02-11 devnull putlastblk(ar);
240 7d59ed71 2005-02-11 devnull static void
241 7d59ed71 2005-02-11 devnull putbackblk(int ar)
243 7d59ed71 2005-02-11 devnull curblk--;
244 7d59ed71 2005-02-11 devnull USED(ar);
247 7d59ed71 2005-02-11 devnull static void
248 7d59ed71 2005-02-11 devnull putreadblks(int ar, int blks)
250 7d59ed71 2005-02-11 devnull curblk += blks - 1;
251 7d59ed71 2005-02-11 devnull USED(ar);
254 7d59ed71 2005-02-11 devnull static void
255 7d59ed71 2005-02-11 devnull putblkmany(int ar, int blks)
257 7d59ed71 2005-02-11 devnull curblk += blks - 1;
258 7d59ed71 2005-02-11 devnull putblk(ar);
262 7d59ed71 2005-02-11 devnull * common routines
265 7d59ed71 2005-02-11 devnull /* modifies hp->chksum */
267 7d59ed71 2005-02-11 devnull chksum(Hdr *hp)
269 7d59ed71 2005-02-11 devnull int n = Tblock;
270 7d59ed71 2005-02-11 devnull long i = 0;
271 510f0dbf 2005-02-11 devnull uchar *cp = (uchar*)hp;
273 7d59ed71 2005-02-11 devnull memset(hp->chksum, ' ', sizeof hp->chksum);
274 7d59ed71 2005-02-11 devnull while (n-- > 0)
275 7d59ed71 2005-02-11 devnull i += *cp++;
276 7d59ed71 2005-02-11 devnull return i;
279 7d59ed71 2005-02-11 devnull static int
280 7d59ed71 2005-02-11 devnull isustar(Hdr *hp)
282 7d59ed71 2005-02-11 devnull return strcmp(hp->magic, "ustar") == 0;
286 7d59ed71 2005-02-11 devnull * s is at most n bytes long, but need not be NUL-terminated.
287 7d59ed71 2005-02-11 devnull * if shorter than n bytes, all bytes after the first NUL must also
288 7d59ed71 2005-02-11 devnull * be NUL.
290 7d59ed71 2005-02-11 devnull static int
291 7d59ed71 2005-02-11 devnull strnlen(char *s, int n)
293 7d59ed71 2005-02-11 devnull return s[n - 1] != '\0'? n: strlen(s);
296 7d59ed71 2005-02-11 devnull /* set fullname from header */
297 7d59ed71 2005-02-11 devnull static char *
298 7d59ed71 2005-02-11 devnull name(Hdr *hp)
300 7d59ed71 2005-02-11 devnull int pfxlen, namlen;
301 7d59ed71 2005-02-11 devnull static char fullname[Maxname + 1];
303 7d59ed71 2005-02-11 devnull namlen = strnlen(hp->name, sizeof hp->name);
304 7d59ed71 2005-02-11 devnull if (hp->prefix[0] == '\0' || !isustar(hp)) { /* old-style name? */
305 7d59ed71 2005-02-11 devnull memmove(fullname, hp->name, namlen);
306 7d59ed71 2005-02-11 devnull fullname[namlen] = '\0';
307 7d59ed71 2005-02-11 devnull return fullname;
310 7d59ed71 2005-02-11 devnull /* name is in two pieces */
311 7d59ed71 2005-02-11 devnull pfxlen = strnlen(hp->prefix, sizeof hp->prefix);
312 7d59ed71 2005-02-11 devnull memmove(fullname, hp->prefix, pfxlen);
313 7d59ed71 2005-02-11 devnull fullname[pfxlen] = '/';
314 7d59ed71 2005-02-11 devnull memmove(fullname + pfxlen + 1, hp->name, namlen);
315 7d59ed71 2005-02-11 devnull fullname[pfxlen + 1 + namlen] = '\0';
316 7d59ed71 2005-02-11 devnull return fullname;
319 7d59ed71 2005-02-11 devnull static int
320 7d59ed71 2005-02-11 devnull isdir(Hdr *hp)
322 7d59ed71 2005-02-11 devnull /* the mode test is ugly but sometimes necessary */
323 7d59ed71 2005-02-11 devnull return hp->linkflag == LF_DIR ||
324 7d59ed71 2005-02-11 devnull strrchr(name(hp), '\0')[-1] == '/' ||
325 7d59ed71 2005-02-11 devnull (strtoul(hp->mode, nil, 8)&0170000) == 040000;
328 7d59ed71 2005-02-11 devnull static int
329 7d59ed71 2005-02-11 devnull eotar(Hdr *hp)
331 7d59ed71 2005-02-11 devnull return name(hp)[0] == '\0';
334 7d59ed71 2005-02-11 devnull static Hdr *
335 7d59ed71 2005-02-11 devnull readhdr(int ar)
337 7d59ed71 2005-02-11 devnull long hdrcksum;
338 7d59ed71 2005-02-11 devnull Hdr *hp;
340 7d59ed71 2005-02-11 devnull hp = getblkrd(ar);
341 7d59ed71 2005-02-11 devnull if (hp == nil)
342 7d59ed71 2005-02-11 devnull sysfatal("unexpected EOF instead of archive header");
343 7d59ed71 2005-02-11 devnull if (eotar(hp)) /* end-of-archive block? */
344 7d59ed71 2005-02-11 devnull return nil;
345 7d59ed71 2005-02-11 devnull hdrcksum = strtoul(hp->chksum, nil, 8);
346 7d59ed71 2005-02-11 devnull if (chksum(hp) != hdrcksum)
347 7d59ed71 2005-02-11 devnull sysfatal("bad archive header checksum: name %.64s...",
348 7d59ed71 2005-02-11 devnull hp->name);
349 7d59ed71 2005-02-11 devnull return hp;
353 7d59ed71 2005-02-11 devnull * tar r[c]
357 7d59ed71 2005-02-11 devnull * if name is longer than Namsiz bytes, try to split it at a slash and fit the
358 7d59ed71 2005-02-11 devnull * pieces into hp->prefix and hp->name.
360 7d59ed71 2005-02-11 devnull static int
361 7d59ed71 2005-02-11 devnull putfullname(Hdr *hp, char *name)
363 7d59ed71 2005-02-11 devnull int namlen, pfxlen;
364 7d59ed71 2005-02-11 devnull char *sl, *osl;
365 7d59ed71 2005-02-11 devnull String *slname = nil;
367 7d59ed71 2005-02-11 devnull if (isdir(hp)) {
368 7d59ed71 2005-02-11 devnull slname = s_new();
369 7d59ed71 2005-02-11 devnull s_append(slname, name);
370 7d59ed71 2005-02-11 devnull s_append(slname, "/"); /* posix requires this */
371 7d59ed71 2005-02-11 devnull name = s_to_c(slname);
374 7d59ed71 2005-02-11 devnull namlen = strlen(name);
375 7d59ed71 2005-02-11 devnull if (namlen <= Namsiz) {
376 7d59ed71 2005-02-11 devnull strncpy(hp->name, name, Namsiz);
377 7d59ed71 2005-02-11 devnull hp->prefix[0] = '\0'; /* ustar paranoia */
378 7d59ed71 2005-02-11 devnull return 0;
381 7d59ed71 2005-02-11 devnull if (!posix || namlen > Maxname) {
382 7d59ed71 2005-02-11 devnull fprint(2, "%s: name too long for tar header: %s\n",
383 7d59ed71 2005-02-11 devnull argv0, name);
384 7d59ed71 2005-02-11 devnull return -1;
387 7d59ed71 2005-02-11 devnull * try various splits until one results in pieces that fit into the
388 7d59ed71 2005-02-11 devnull * appropriate fields of the header. look for slashes from right
389 7d59ed71 2005-02-11 devnull * to left, in the hopes of putting the largest part of the name into
390 7d59ed71 2005-02-11 devnull * hp->prefix, which is larger than hp->name.
392 7d59ed71 2005-02-11 devnull sl = strrchr(name, '/');
393 7d59ed71 2005-02-11 devnull while (sl != nil) {
394 7d59ed71 2005-02-11 devnull pfxlen = sl - name;
395 7d59ed71 2005-02-11 devnull if (pfxlen <= sizeof hp->prefix && namlen-1 - pfxlen <= Namsiz)
397 7d59ed71 2005-02-11 devnull osl = sl;
398 7d59ed71 2005-02-11 devnull *osl = '\0';
399 7d59ed71 2005-02-11 devnull sl = strrchr(name, '/');
400 7d59ed71 2005-02-11 devnull *osl = '/';
402 7d59ed71 2005-02-11 devnull if (sl == nil) {
403 7d59ed71 2005-02-11 devnull fprint(2, "%s: name can't be split to fit tar header: %s\n",
404 7d59ed71 2005-02-11 devnull argv0, name);
405 7d59ed71 2005-02-11 devnull return -1;
407 7d59ed71 2005-02-11 devnull *sl = '\0';
408 7d59ed71 2005-02-11 devnull strncpy(hp->prefix, name, sizeof hp->prefix);
409 7d59ed71 2005-02-11 devnull *sl++ = '/';
410 7d59ed71 2005-02-11 devnull strncpy(hp->name, sl, sizeof hp->name);
411 7d59ed71 2005-02-11 devnull if (slname)
412 7d59ed71 2005-02-11 devnull s_free(slname);
413 7d59ed71 2005-02-11 devnull return 0;
416 7d59ed71 2005-02-11 devnull static int
417 7d59ed71 2005-02-11 devnull mkhdr(Hdr *hp, Dir *dir, char *file)
420 7d59ed71 2005-02-11 devnull * these fields run together, so we format them in order and don't use
421 7d59ed71 2005-02-11 devnull * snprint.
423 7d59ed71 2005-02-11 devnull sprint(hp->mode, "%6lo ", dir->mode & 0777);
424 7d59ed71 2005-02-11 devnull sprint(hp->uid, "%6o ", aruid);
425 7d59ed71 2005-02-11 devnull sprint(hp->gid, "%6o ", argid);
427 7d59ed71 2005-02-11 devnull * files > 2 bytes can't be described
428 7d59ed71 2005-02-11 devnull * (unless we resort to xustar or exustar formats).
430 7d59ed71 2005-02-11 devnull if (dir->length >= (Off)1<<33) {
431 7d59ed71 2005-02-11 devnull fprint(2, "%s: %s: too large for tar header format\n",
432 7d59ed71 2005-02-11 devnull argv0, file);
433 7d59ed71 2005-02-11 devnull return -1;
435 7d59ed71 2005-02-11 devnull sprint(hp->size, "%11lluo ", dir->length);
436 7d59ed71 2005-02-11 devnull sprint(hp->mtime, "%11luo ", dir->mtime);
437 7d59ed71 2005-02-11 devnull hp->linkflag = (dir->mode&DMDIR? LF_DIR: LF_PLAIN1);
438 7d59ed71 2005-02-11 devnull putfullname(hp, file);
439 7d59ed71 2005-02-11 devnull if (posix) {
440 7d59ed71 2005-02-11 devnull strncpy(hp->magic, "ustar", sizeof hp->magic);
441 7d59ed71 2005-02-11 devnull strncpy(hp->version, "00", sizeof hp->version);
442 7d59ed71 2005-02-11 devnull strncpy(hp->uname, dir->uid, sizeof hp->uname);
443 7d59ed71 2005-02-11 devnull strncpy(hp->gname, dir->gid, sizeof hp->gname);
445 7d59ed71 2005-02-11 devnull sprint(hp->chksum, "%6luo", chksum(hp));
446 7d59ed71 2005-02-11 devnull return 0;
449 7d59ed71 2005-02-11 devnull static void addtoar(int ar, char *file, char *shortf);
451 7d59ed71 2005-02-11 devnull static void
452 7d59ed71 2005-02-11 devnull addtreetoar(int ar, char *file, char *shortf, int fd)
455 7d59ed71 2005-02-11 devnull Dir *dent, *dirents;
456 7d59ed71 2005-02-11 devnull String *name = s_new();
458 7d59ed71 2005-02-11 devnull n = dirreadall(fd, &dirents);
459 7d59ed71 2005-02-11 devnull close(fd);
460 7d59ed71 2005-02-11 devnull if (n == 0)
463 7d59ed71 2005-02-11 devnull if (chdir(shortf) < 0)
464 7d59ed71 2005-02-11 devnull sysfatal("chdir %s: %r", file);
465 7d59ed71 2005-02-11 devnull if (DEBUG)
466 7d59ed71 2005-02-11 devnull fprint(2, "chdir %s\t# %s\n", shortf, file);
468 7d59ed71 2005-02-11 devnull for (dent = dirents; dent < dirents + n; dent++) {
469 7d59ed71 2005-02-11 devnull s_reset(name);
470 7d59ed71 2005-02-11 devnull s_append(name, file);
471 7d59ed71 2005-02-11 devnull s_append(name, "/");
472 7d59ed71 2005-02-11 devnull s_append(name, dent->name);
473 7d59ed71 2005-02-11 devnull addtoar(ar, s_to_c(name), dent->name);
475 7d59ed71 2005-02-11 devnull s_free(name);
476 7d59ed71 2005-02-11 devnull free(dirents);
478 7d59ed71 2005-02-11 devnull if (chdir("..") < 0)
479 7d59ed71 2005-02-11 devnull sysfatal("chdir %s: %r", file);
480 7d59ed71 2005-02-11 devnull if (DEBUG)
481 7d59ed71 2005-02-11 devnull fprint(2, "chdir ..\n");
484 7d59ed71 2005-02-11 devnull static void
485 7d59ed71 2005-02-11 devnull addtoar(int ar, char *file, char *shortf)
487 7d59ed71 2005-02-11 devnull int n, fd, isdir, r;
488 7d59ed71 2005-02-11 devnull long bytes;
489 7d59ed71 2005-02-11 devnull ulong blksleft, blksread;
490 7d59ed71 2005-02-11 devnull Hdr *hbp;
491 7d59ed71 2005-02-11 devnull Dir *dir;
493 7d59ed71 2005-02-11 devnull fd = open(shortf, OREAD);
494 7d59ed71 2005-02-11 devnull if (fd < 0) {
495 7d59ed71 2005-02-11 devnull fprint(2, "%s: can't open %s: %r\n", argv0, file);
498 7d59ed71 2005-02-11 devnull dir = dirfstat(fd);
499 7d59ed71 2005-02-11 devnull if (dir == nil)
500 7d59ed71 2005-02-11 devnull sysfatal("can't fstat %s: %r", file);
502 7d59ed71 2005-02-11 devnull hbp = getblkz(ar);
503 7d59ed71 2005-02-11 devnull isdir = !!(dir->qid.type&QTDIR);
504 7d59ed71 2005-02-11 devnull r = mkhdr(hbp, dir, file);
505 7d59ed71 2005-02-11 devnull if (r < 0) {
506 7d59ed71 2005-02-11 devnull putbackblk(ar);
507 7d59ed71 2005-02-11 devnull free(dir);
508 7d59ed71 2005-02-11 devnull close(fd);
511 7d59ed71 2005-02-11 devnull putblk(ar);
513 7d59ed71 2005-02-11 devnull blksleft = BYTES2TBLKS(dir->length);
514 7d59ed71 2005-02-11 devnull free(dir);
516 7d59ed71 2005-02-11 devnull if (isdir)
517 7d59ed71 2005-02-11 devnull addtreetoar(ar, file, shortf, fd);
519 7d59ed71 2005-02-11 devnull for (; blksleft > 0; blksleft -= blksread) {
520 7d59ed71 2005-02-11 devnull hbp = getblke(ar);
521 7d59ed71 2005-02-11 devnull blksread = gothowmany(blksleft);
522 7d59ed71 2005-02-11 devnull bytes = blksread * Tblock;
523 510f0dbf 2005-02-11 devnull n = read(fd, hbp, bytes);
524 7d59ed71 2005-02-11 devnull if (n < 0)
525 7d59ed71 2005-02-11 devnull sysfatal("error reading %s: %r", file);
527 7d59ed71 2005-02-11 devnull * ignore EOF. zero any partial block to aid
528 7d59ed71 2005-02-11 devnull * compression and emergency recovery of data.
530 7d59ed71 2005-02-11 devnull if (n < Tblock)
531 510f0dbf 2005-02-11 devnull memset((char*)hbp + n, 0, bytes - n);
532 7d59ed71 2005-02-11 devnull putblkmany(ar, blksread);
534 7d59ed71 2005-02-11 devnull close(fd);
535 7d59ed71 2005-02-11 devnull if (verbose)
536 7d59ed71 2005-02-11 devnull fprint(2, "%s: a %s\n", argv0, file);
540 7d59ed71 2005-02-11 devnull static void
541 7d59ed71 2005-02-11 devnull replace(char **argv)
543 7d59ed71 2005-02-11 devnull int i, ar;
544 7d59ed71 2005-02-11 devnull ulong blksleft, blksread;
545 7d59ed71 2005-02-11 devnull Off bytes;
546 7d59ed71 2005-02-11 devnull Hdr *hp;
548 7d59ed71 2005-02-11 devnull if (usefile && creat)
549 7d59ed71 2005-02-11 devnull ar = create(usefile, ORDWR, 0666);
550 7d59ed71 2005-02-11 devnull else if (usefile)
551 7d59ed71 2005-02-11 devnull ar = open(usefile, ORDWR);
553 7d59ed71 2005-02-11 devnull ar = Stdout;
554 7d59ed71 2005-02-11 devnull if (ar < 0)
555 7d59ed71 2005-02-11 devnull sysfatal("can't open archive %s: %r", usefile);
557 7d59ed71 2005-02-11 devnull if (usefile && !creat) {
558 7d59ed71 2005-02-11 devnull /* skip quickly to the end */
559 7d59ed71 2005-02-11 devnull while ((hp = readhdr(ar)) != nil) {
560 7d59ed71 2005-02-11 devnull bytes = (isdir(hp)? 0: strtoull(hp->size, nil, 8));
561 7d59ed71 2005-02-11 devnull for (blksleft = BYTES2TBLKS(bytes);
562 7d59ed71 2005-02-11 devnull blksleft > 0 && getblkrd(ar) != nil;
563 7d59ed71 2005-02-11 devnull blksleft -= blksread) {
564 7d59ed71 2005-02-11 devnull blksread = gothowmany(blksleft);
565 7d59ed71 2005-02-11 devnull putreadblks(ar, blksread);
569 7d59ed71 2005-02-11 devnull * we have just read the end-of-archive Tblock.
570 7d59ed71 2005-02-11 devnull * now seek back over the (big) archive block containing it,
571 7d59ed71 2005-02-11 devnull * and back up curblk ptr over end-of-archive Tblock in memory.
573 7d59ed71 2005-02-11 devnull if (seek(ar, -Tblock*nblock, 1) < 0)
574 7d59ed71 2005-02-11 devnull sysfatal("can't seek back over end-of-archive: %r");
575 7d59ed71 2005-02-11 devnull curblk--;
578 7d59ed71 2005-02-11 devnull for (i = 0; argv[i] != nil; i++)
579 7d59ed71 2005-02-11 devnull addtoar(ar, argv[i], argv[i]);
581 7d59ed71 2005-02-11 devnull /* write end-of-archive marker */
582 7d59ed71 2005-02-11 devnull getblkz(ar);
583 7d59ed71 2005-02-11 devnull putblk(ar);
584 7d59ed71 2005-02-11 devnull getblkz(ar);
585 7d59ed71 2005-02-11 devnull putlastblk(ar);
587 7d59ed71 2005-02-11 devnull if (ar > Stderr)
588 7d59ed71 2005-02-11 devnull close(ar);
592 7d59ed71 2005-02-11 devnull * tar [xt]
595 7d59ed71 2005-02-11 devnull /* is pfx a file-name prefix of name? */
596 7d59ed71 2005-02-11 devnull static int
597 7d59ed71 2005-02-11 devnull prefix(char *name, char *pfx)
599 7d59ed71 2005-02-11 devnull int pfxlen = strlen(pfx);
600 7d59ed71 2005-02-11 devnull char clpfx[Maxname+1];
602 7d59ed71 2005-02-11 devnull if (pfxlen > Maxname)
603 7d59ed71 2005-02-11 devnull return 0;
604 7d59ed71 2005-02-11 devnull strcpy(clpfx, pfx);
605 7d59ed71 2005-02-11 devnull cleanname(clpfx);
606 7d59ed71 2005-02-11 devnull return strncmp(pfx, name, pfxlen) == 0 &&
607 7d59ed71 2005-02-11 devnull (name[pfxlen] == '\0' || name[pfxlen] == '/');
610 7d59ed71 2005-02-11 devnull static int
611 7d59ed71 2005-02-11 devnull match(char *name, char **argv)
614 7d59ed71 2005-02-11 devnull char clname[Maxname+1];
616 7d59ed71 2005-02-11 devnull if (argv[0] == nil)
617 7d59ed71 2005-02-11 devnull return 1;
618 7d59ed71 2005-02-11 devnull strcpy(clname, name);
619 7d59ed71 2005-02-11 devnull cleanname(clname);
620 7d59ed71 2005-02-11 devnull for (i = 0; argv[i] != nil; i++)
621 7d59ed71 2005-02-11 devnull if (prefix(clname, argv[i]))
622 7d59ed71 2005-02-11 devnull return 1;
623 7d59ed71 2005-02-11 devnull return 0;
626 7d59ed71 2005-02-11 devnull static int
627 7d59ed71 2005-02-11 devnull makedir(char *s)
631 7d59ed71 2005-02-11 devnull if (access(s, AEXIST) == 0)
632 7d59ed71 2005-02-11 devnull return -1;
633 7d59ed71 2005-02-11 devnull f = create(s, OREAD, DMDIR | 0777);
634 7d59ed71 2005-02-11 devnull if (f >= 0)
635 7d59ed71 2005-02-11 devnull close(f);
636 7d59ed71 2005-02-11 devnull return f;
639 7d59ed71 2005-02-11 devnull static void
640 7d59ed71 2005-02-11 devnull mkpdirs(char *s)
642 7d59ed71 2005-02-11 devnull int done = 0;
643 7d59ed71 2005-02-11 devnull char *p = s;
645 7d59ed71 2005-02-11 devnull while (!done && (p = strchr(p + 1, '/')) != nil) {
646 7d59ed71 2005-02-11 devnull *p = '\0';
647 7d59ed71 2005-02-11 devnull done = (access(s, AEXIST) < 0 && makedir(s) < 0);
648 7d59ed71 2005-02-11 devnull *p = '/';
652 7d59ed71 2005-02-11 devnull /* copy a file from the archive into the filesystem */
653 7d59ed71 2005-02-11 devnull static void
654 7d59ed71 2005-02-11 devnull extract1(int ar, Hdr *hp, char *fname)
656 7d59ed71 2005-02-11 devnull int wrbytes, fd = -1, dir = 0, okcreate = 1;
657 7d59ed71 2005-02-11 devnull long mtime = strtol(hp->mtime, nil, 8);
658 7d59ed71 2005-02-11 devnull ulong mode = strtoul(hp->mode, nil, 8) & 0777;
659 7d59ed71 2005-02-11 devnull Off bytes = strtoll(hp->size, nil, 8);
660 7d59ed71 2005-02-11 devnull ulong blksread, blksleft = BYTES2TBLKS(bytes);
661 7d59ed71 2005-02-11 devnull Hdr *hbp;
663 7d59ed71 2005-02-11 devnull if (isdir(hp)) {
664 7d59ed71 2005-02-11 devnull mode |= DMDIR|0700;
665 7d59ed71 2005-02-11 devnull blksleft = 0;
666 7d59ed71 2005-02-11 devnull dir = 1;
668 7d59ed71 2005-02-11 devnull switch (hp->linkflag) {
669 7d59ed71 2005-02-11 devnull case LF_LINK:
670 7d59ed71 2005-02-11 devnull case LF_SYMLINK1:
671 7d59ed71 2005-02-11 devnull case LF_SYMLINK2:
672 7d59ed71 2005-02-11 devnull case LF_FIFO:
673 7d59ed71 2005-02-11 devnull blksleft = okcreate = 0;
676 7d59ed71 2005-02-11 devnull if (relative && fname[0] == '/')
677 7d59ed71 2005-02-11 devnull fname++;
678 7d59ed71 2005-02-11 devnull if (verb == Xtract) {
679 7d59ed71 2005-02-11 devnull cleanname(fname);
680 7d59ed71 2005-02-11 devnull switch (hp->linkflag) {
681 7d59ed71 2005-02-11 devnull case LF_LINK:
682 7d59ed71 2005-02-11 devnull case LF_SYMLINK1:
683 7d59ed71 2005-02-11 devnull case LF_SYMLINK2:
684 7d59ed71 2005-02-11 devnull fprint(2, "%s: can't make (sym)link %s\n",
685 7d59ed71 2005-02-11 devnull argv0, fname);
687 7d59ed71 2005-02-11 devnull case LF_FIFO:
688 7d59ed71 2005-02-11 devnull fprint(2, "%s: can't make fifo %s\n", argv0, fname);
691 7d59ed71 2005-02-11 devnull if (okcreate)
692 7d59ed71 2005-02-11 devnull fd = create(fname, (dir? OREAD: OWRITE), mode);
693 7d59ed71 2005-02-11 devnull if (fd < 0) {
694 7d59ed71 2005-02-11 devnull mkpdirs(fname);
695 7d59ed71 2005-02-11 devnull fd = create(fname, (dir? OREAD: OWRITE), mode);
696 7d59ed71 2005-02-11 devnull if (fd < 0 && (!dir || access(fname, AEXIST) < 0))
697 7d59ed71 2005-02-11 devnull fprint(2, "%s: can't create %s: %r\n",
698 7d59ed71 2005-02-11 devnull argv0, fname);
700 7d59ed71 2005-02-11 devnull if (fd >= 0 && verbose)
701 7d59ed71 2005-02-11 devnull fprint(2, "%s: x %s\n", argv0, fname);
702 7d59ed71 2005-02-11 devnull } else if (verbose) {
703 7d59ed71 2005-02-11 devnull char *cp = ctime(mtime);
705 7d59ed71 2005-02-11 devnull print("%M %8lld %-12.12s %-4.4s %s\n",
706 7d59ed71 2005-02-11 devnull mode, bytes, cp+4, cp+24, fname);
708 7d59ed71 2005-02-11 devnull print("%s\n", fname);
710 7d59ed71 2005-02-11 devnull for (; blksleft > 0; blksleft -= blksread) {
711 7d59ed71 2005-02-11 devnull hbp = getblkrd(ar);
712 7d59ed71 2005-02-11 devnull if (hbp == nil)
713 7d59ed71 2005-02-11 devnull sysfatal("unexpected EOF on archive extracting %s",
715 7d59ed71 2005-02-11 devnull blksread = gothowmany(blksleft);
716 7d59ed71 2005-02-11 devnull wrbytes = (bytes >= Tblock*blksread? Tblock*blksread: bytes);
717 510f0dbf 2005-02-11 devnull if (fd >= 0 && write(fd, (char*)hbp, wrbytes) != wrbytes)
718 7d59ed71 2005-02-11 devnull sysfatal("write error on %s: %r", fname);
719 7d59ed71 2005-02-11 devnull putreadblks(ar, blksread);
720 7d59ed71 2005-02-11 devnull bytes -= wrbytes;
722 7d59ed71 2005-02-11 devnull if (fd >= 0) {
724 7d59ed71 2005-02-11 devnull * directories should be wstated after we're done
725 7d59ed71 2005-02-11 devnull * creating files in them.
727 7d59ed71 2005-02-11 devnull if (settime) {
730 7d59ed71 2005-02-11 devnull nulldir(&nd);
731 7d59ed71 2005-02-11 devnull nd.mtime = mtime;
732 7d59ed71 2005-02-11 devnull if (isustar(hp))
733 7d59ed71 2005-02-11 devnull nd.gid = hp->gname;
734 7d59ed71 2005-02-11 devnull dirfwstat(fd, &nd);
736 7d59ed71 2005-02-11 devnull close(fd);
740 7d59ed71 2005-02-11 devnull static void
741 7d59ed71 2005-02-11 devnull skip(int ar, Hdr *hp, char *fname)
743 7d59ed71 2005-02-11 devnull Off bytes;
744 7d59ed71 2005-02-11 devnull ulong blksleft, blksread;
745 7d59ed71 2005-02-11 devnull Hdr *hbp;
747 7d59ed71 2005-02-11 devnull if (isdir(hp))
749 7d59ed71 2005-02-11 devnull bytes = strtoull(hp->size, nil, 8);
750 7d59ed71 2005-02-11 devnull blksleft = BYTES2TBLKS(bytes);
751 7d59ed71 2005-02-11 devnull for (; blksleft > 0; blksleft -= blksread) {
752 7d59ed71 2005-02-11 devnull hbp = getblkrd(ar);
753 7d59ed71 2005-02-11 devnull if (hbp == nil)
754 7d59ed71 2005-02-11 devnull sysfatal("unexpected EOF on archive extracting %s",
756 7d59ed71 2005-02-11 devnull blksread = gothowmany(blksleft);
757 7d59ed71 2005-02-11 devnull putreadblks(ar, blksread);
761 7d59ed71 2005-02-11 devnull static void
762 7d59ed71 2005-02-11 devnull extract(char **argv)
765 7d59ed71 2005-02-11 devnull char *longname;
766 7d59ed71 2005-02-11 devnull Hdr *hp;
768 7d59ed71 2005-02-11 devnull if (usefile)
769 7d59ed71 2005-02-11 devnull ar = open(usefile, OREAD);
771 7d59ed71 2005-02-11 devnull ar = Stdin;
772 7d59ed71 2005-02-11 devnull if (ar < 0)
773 7d59ed71 2005-02-11 devnull sysfatal("can't open archive %s: %r", usefile);
774 7d59ed71 2005-02-11 devnull while ((hp = readhdr(ar)) != nil) {
775 7d59ed71 2005-02-11 devnull longname = name(hp);
776 7d59ed71 2005-02-11 devnull if (match(longname, argv))
777 7d59ed71 2005-02-11 devnull extract1(ar, hp, longname);
779 7d59ed71 2005-02-11 devnull skip(ar, hp, longname);
781 7d59ed71 2005-02-11 devnull if (ar > Stderr)
782 7d59ed71 2005-02-11 devnull close(ar);
786 7d59ed71 2005-02-11 devnull main(int argc, char *argv[])
788 7d59ed71 2005-02-11 devnull int errflg = 0;
789 7d59ed71 2005-02-11 devnull char *ret = nil;
791 7d59ed71 2005-02-11 devnull quotefmtinstall();
792 7d59ed71 2005-02-11 devnull fmtinstall('M', dirmodefmt);
794 7d59ed71 2005-02-11 devnull TARGBEGIN {
795 7d59ed71 2005-02-11 devnull case 'c':
796 7d59ed71 2005-02-11 devnull creat++;
797 7d59ed71 2005-02-11 devnull verb = Replace;
799 7d59ed71 2005-02-11 devnull case 'f':
800 7d59ed71 2005-02-11 devnull usefile = EARGF(usage());
802 7d59ed71 2005-02-11 devnull case 'g':
803 7d59ed71 2005-02-11 devnull argid = strtoul(EARGF(usage()), 0, 0);
805 7d59ed71 2005-02-11 devnull case 'p':
806 7d59ed71 2005-02-11 devnull posix++;
808 7d59ed71 2005-02-11 devnull case 'P':
809 7d59ed71 2005-02-11 devnull posix = 0;
811 7d59ed71 2005-02-11 devnull case 'r':
812 7d59ed71 2005-02-11 devnull verb = Replace;
814 7d59ed71 2005-02-11 devnull case 'R':
815 7d59ed71 2005-02-11 devnull relative++;
817 7d59ed71 2005-02-11 devnull case 't':
818 7d59ed71 2005-02-11 devnull verb = Toc;
820 7d59ed71 2005-02-11 devnull case 'T':
821 7d59ed71 2005-02-11 devnull settime++;
823 7d59ed71 2005-02-11 devnull case 'u':
824 7d59ed71 2005-02-11 devnull aruid = strtoul(EARGF(usage()), 0, 0);
826 7d59ed71 2005-02-11 devnull case 'v':
827 7d59ed71 2005-02-11 devnull verbose++;
829 7d59ed71 2005-02-11 devnull case 'x':
830 7d59ed71 2005-02-11 devnull verb = Xtract;
832 7d59ed71 2005-02-11 devnull case '-':
834 7d59ed71 2005-02-11 devnull default:
835 7d59ed71 2005-02-11 devnull errflg++;
837 7d59ed71 2005-02-11 devnull } TARGEND
839 7d59ed71 2005-02-11 devnull if (argc < 0 || errflg)
840 7d59ed71 2005-02-11 devnull usage();
842 7d59ed71 2005-02-11 devnull initblks();
843 7d59ed71 2005-02-11 devnull switch (verb) {
844 7d59ed71 2005-02-11 devnull case Toc:
845 7d59ed71 2005-02-11 devnull case Xtract:
846 7d59ed71 2005-02-11 devnull extract(argv);
848 7d59ed71 2005-02-11 devnull case Replace:
849 7d59ed71 2005-02-11 devnull if (getwd(origdir, sizeof origdir) == nil)
850 7d59ed71 2005-02-11 devnull strcpy(origdir, "/tmp");
851 7d59ed71 2005-02-11 devnull replace(argv);
852 7d59ed71 2005-02-11 devnull chdir(origdir); /* for profiling */
854 7d59ed71 2005-02-11 devnull default:
855 7d59ed71 2005-02-11 devnull usage();
858 7d59ed71 2005-02-11 devnull exits(ret);