From c8de117f1c0c9cd77333c88b0182aab63d381154 Mon Sep 17 00:00:00 2001 From: Rob Landley Date: Mon, 1 Apr 2019 12:52:35 -0500 Subject: More tar cleanup. --- toys/pending/tar.c | 114 ++++++++++++++++++++++++++++++----------------------- 1 file changed, 65 insertions(+), 49 deletions(-) (limited to 'toys') diff --git a/toys/pending/tar.c b/toys/pending/tar.c index 77c03540..7d688241 100644 --- a/toys/pending/tar.c +++ b/toys/pending/tar.c @@ -18,7 +18,7 @@ * Extract into dir same as filename, --restrict? "Tarball is splodey" * -USE_TAR(NEWTOY(tar, "&(no-recursion)(numeric-owner)(no-same-permissions)(overwrite)(exclude)*(mtime):(group):(owner):(to-command):o(no-same-owner)p(same-permissions)k(keep-old)c(create)|h(dereference)x(extract)|t(list)|v(verbose)j(bzip2)z(gzip)O(to-stdout)m(touch)X(exclude-from)*T(files-from)*C(directory):f(file):[!txc][!jz]", TOYFLAG_USR|TOYFLAG_BIN)) +USE_TAR(NEWTOY(tar, "&(full-time)(no-recursion)(numeric-owner)(no-same-permissions)(overwrite)(exclude)*(mtime):(group):(owner):(to-command):o(no-same-owner)p(same-permissions)k(keep-old)c(create)|h(dereference)x(extract)|t(list)|v(verbose)j(bzip2)z(gzip)O(to-stdout)m(touch)X(exclude-from)*T(files-from)*C(directory):f(file):[!txc][!jz]", TOYFLAG_USR|TOYFLAG_BIN)) config TAR bool "tar" @@ -49,7 +49,7 @@ GLOBALS( struct double_list *incl, *excl, *seen; struct string_list *dirs; char *cwd; - int fd, ouid, ggid, hlc, warn; + int fd, ouid, ggid, hlc, warn, adev, aino; time_t mtt; // hardlinks seen so far (hlc many) @@ -60,7 +60,7 @@ GLOBALS( } *hlx; // Parsed information about a tar header. - struct { + struct tar_header { char *name, *link_target, *uname, *gname; long long size; uid_t uid; @@ -129,8 +129,8 @@ static void write_longname(char *name, char type) tmp.type = type; strcpy(tmp.magic, "ustar "); - // Calculate checksum. Since 0777777 is twice 512*255 it can never use more - // than 6 digits, last byte is ' ' or historical reasons. + // Calculate checksum. Since 512*255 = 0377000 in octal, this can never + // use more than 6 digits. The last byte is ' ' or historical reasons. itoo(tmp.chksum, sizeof(tmp.chksum)-1, cksum(&tmp)); tmp.chksum[7] = ' '; @@ -168,34 +168,42 @@ static void alloread(void *buf, int len) b[len] = 0; } -static void add_file(char **nam, struct stat *st) +static int add_to_tar(struct dirtree *node) { + struct stat *st = &(node->st); struct tar_hdr hdr; struct passwd *pw = pw; struct group *gr = gr; int i, fd =-1; - char *c, *p, *name = *nam, *lnk, *hname; + char *c, *p, *name, *lnk, *hname; + + if (!dirtree_notdotdot(node)) return 0; + if (TT.adev == st->st_dev && TT.aino == st->st_ino) { + error_msg("'%s' file is the archive; not dumped", node->name); + return 0; + } + + i = 1; + name = dirtree_path(node, &i); // exclusion defaults to --no-anchored and --wildcards-match-slash for (p = name; *p; p++) - if ((p == name || p[-1] == '/') && *p != '/' && filter(TT.excl, p)) return; + if ((p == name || p[-1] == '/') && *p != '/' && filter(TT.excl, p)) + goto done; - if (S_ISDIR(st->st_mode) && name[strlen(name)-1] != '/') { - lnk = xmprintf("%s/", name); - free(name); - *nam = name = lnk; - } + // The 1 extra byte from dirtree_path() + if (S_ISDIR(st->st_mode) && name[i-1] != '/') strcat(name, "/"); // remove leading / and any .. entries from saved name for (hname = name; *hname == '/'; hname++); for (c = hname;;) { if (!(c = strstr(c, ".."))) break; if (c == hname || c[-1] == '/') { - if (!c[2]) return; + if (!c[2]) goto done; if (c[2]=='/') c = hname = c+3; } else c+= 2; } - if (!*hname) return; + if (!*hname) goto done; if (TT.warn && hname != name) { fprintf(stderr, "removing leading '%.*s' from member names\n", @@ -214,6 +222,7 @@ static void add_file(char **nam, struct stat *st) ITOO(hdr.gid, st->st_gid); ITOO(hdr.size, 0); //set size later ITOO(hdr.mtime, st->st_mtime); + strcpy(hdr.magic, "ustar "); // Hard link or symlink? i=0 neither, i=1 hardlink, i=2 symlink @@ -228,7 +237,7 @@ static void add_file(char **nam, struct stat *st) lnk = TT.hlx[i].arg; i = 1; } else { - // first time we've seen it, store as normal file. + // first time we've seen it. Store as normal file, but remember it. if (!(TT.hlc&255)) TT.hlx = xrealloc(TT.hlx, TT.hlc+256); TT.hlx[TT.hlc].arg = xstrdup(hname); TT.hlx[TT.hlc].ino = st->st_ino; @@ -247,7 +256,10 @@ static void add_file(char **nam, struct stat *st) // Handle file types if (i) { hdr.type = '0'+i; - if (i==2 && !(lnk = xreadlink(name))) return perror_msg("readlink"); + if (i==2 && !(lnk = xreadlink(name))) { + perror_msg("readlink"); + goto done; + } if (strlen(lnk) > sizeof(hdr.link)) write_longname(lnk, 'K'); strncpy(hdr.link, lnk, sizeof(hdr.link)); if (i) free(lnk); @@ -260,10 +272,12 @@ static void add_file(char **nam, struct stat *st) hdr.type = (S_ISCHR(st->st_mode))?'3':'4'; ITOO(hdr.major, dev_major(st->st_rdev)); ITOO(hdr.minor, dev_minor(st->st_rdev)); - } else return error_msg("unknown file type '%o'", st->st_mode & S_IFMT); + } else { + error_msg("unknown file type '%o'", st->st_mode & S_IFMT); + goto done; + } if (strlen(hname) > sizeof(hdr.name)) write_longname(hname, 'L'); - strcpy(hdr.magic, "ustar "); if (!FLAG(numeric_owner)) { if (TT.owner || (pw = bufgetpwuid(st->st_uid))) strncpy(hdr.uname, TT.owner ? TT.owner : pw->pw_name, sizeof(hdr.uname)); @@ -278,29 +292,16 @@ static void add_file(char **nam, struct stat *st) // Write header and data to archive xwrite(TT.fd, &hdr, 512); - if (hdr.type != '0') return; //nothing to write - if ((fd = open(name, O_RDONLY)) < 0) - return perror_msg("can't open '%s'", name); - xsendfile_pad(fd, TT.fd, st->st_size); - if (st->st_size%512) writeall(TT.fd, toybuf, (512-(st->st_size%512))); - close(fd); -} - -static int add_to_tar(struct dirtree *node) -{ - struct stat st; - char *path; - - if (!dirtree_notdotdot(node)) return 0; - if (!fstat(TT.fd, &st) && st.st_dev == node->st.st_dev - && st.st_ino == node->st.st_ino) { - error_msg("'%s' file is the archive; not dumped", TT.f); - return 0; + if (hdr.type == '0') { + if ((fd = open(name, O_RDONLY)) < 0) perror_msg("can't open '%s'", name); + else { + xsendfile_pad(fd, TT.fd, st->st_size); + if (st->st_size%512) writeall(TT.fd, toybuf, (512-(st->st_size%512))); + close(fd); + } } - - path = dirtree_path(node, 0); - add_file(&path, &(node->st)); //path may be modified - free(path); +done: + free(name); return (DIRTREE_RECURSE|(FLAG(h)?DIRTREE_SYMFOLLOW:0))*!FLAG(no_recursion); } @@ -425,18 +426,18 @@ static void extract_to_disk(void) return perror_msg("can't create '%s'", name); + // Set ownership if (!FLAG(o) && !geteuid()) { - //set ownership..., --no-same-owner, --numeric-owner int u = TT.hdr.uid, g = TT.hdr.gid; if (TT.owner) TT.hdr.uid = TT.ouid; - else if (!FLAG(numeric_owner)) { + else if (!FLAG(numeric_owner) && *TT.hdr.uname) { struct passwd *pw = getpwnam(TT.hdr.uname); if (pw && (TT.owner || !FLAG(numeric_owner))) TT.hdr.uid = pw->pw_uid; } if (TT.group) TT.hdr.gid = TT.ggid; - else if (!FLAG(numeric_owner)) { + else if (!FLAG(numeric_owner) && *TT.hdr.uname) { struct group *gr = getgrnam(TT.hdr.gname); if (gr) TT.hdr.gid = gr->gr_gid; } @@ -598,16 +599,21 @@ static void unpack_tar(void) else if (FLAG(t)) { if (FLAG(v)) { struct tm *lc = localtime(TT.mtime ? &TT.mtt : &TT.hdr.mtime); - char perm[11]; + char perm[12], gname[12]; mode_to_string(TT.hdr.mode, perm); - printf("%s %s/%s ", perm, TT.hdr.uname, TT.hdr.gname); + printf("%s", perm); + sprintf(perm, "%u", TT.hdr.uid); + sprintf(gname, "%u", TT.hdr.gid); + printf(" %s/%s ", *TT.hdr.uname ? TT.hdr.uname : perm, + *TT.hdr.gname ? TT.hdr.gname : gname); if (tar.type=='3' || tar.type=='4') printf("%u,%u", maj, min); else printf("%9lld", (long long)TT.hdr.size); - printf(" %d-%02d-%02d %02d:%02d:%02d ", 1900+lc->tm_year, 1+lc->tm_mon, - lc->tm_mday, lc->tm_hour, lc->tm_min, lc->tm_sec); + sprintf(perm, ":%02d", lc->tm_sec); + printf(" %d-%02d-%02d %02d:%02d%s ", 1900+lc->tm_year, 1+lc->tm_mon, + lc->tm_mday, lc->tm_hour, lc->tm_min, FLAG(full_time) ? perm : ""); } - printf("%s", TT.hdr.name); + printf(" %s", TT.hdr.name); if (TT.hdr.link_target) printf(" -> %s", TT.hdr.link_target); xputc('\n'); skippy(TT.hdr.size); @@ -663,6 +669,16 @@ void tar_main(void) if (TT.f && strcmp(TT.f, "-")) TT.fd = xcreate(TT.f, TT.fd*(O_WRONLY|O_CREAT|O_TRUNC), 0666); + // grab archive inode + { + struct stat st; + + if (!fstat(TT.fd, &st)) { + TT.aino = st.st_ino; + TT.adev = st.st_dev; + } + } + // Get destination directory if (TT.C) xchdir(TT.C); TT.cwd = xabspath(s = xgetcwd(), 1); -- cgit v1.2.3