diff options
author | Denis Vlasenko <vda.linux@googlemail.com> | 2006-11-24 17:59:45 +0300 |
---|---|---|
committer | Denis Vlasenko <vda.linux@googlemail.com> | 2006-11-24 17:59:45 +0300 |
commit | b8a8e601df9e912faa20857b4b5c07024618f7b3 (patch) | |
tree | 2dde275d11a03e45f7fe4aca444d0195c415a59c | |
parent | c50f370f98e5f5d24d18ca3df444770dacce1824 (diff) |
tar: small fixes:
* size-optimize mapping code
* kill double close
-rw-r--r-- | archival/tar.c | 46 | ||||
-rw-r--r-- | libbb/procps.c | 34 |
2 files changed, 42 insertions, 38 deletions
diff --git a/archival/tar.c b/archival/tar.c index 51d34ea19..be661cc32 100644 --- a/archival/tar.c +++ b/archival/tar.c @@ -150,7 +150,9 @@ static HardLinkInfo *findHardLinkInfo(HardLinkInfo * hlInfo, struct stat *statbu /* Put an octal string into the specified buffer. * The number is zero padded and possibly null terminated. - * Returns TRUE if successful. - DISABLED (no caller ever checked) */ + * Returns TRUE if successful. - DISABLED (no caller ever checked) */ +/* FIXME: we leave field untouched if value doesn't fit. */ +/* This is not good - what will happen at untar time?? */ static void putOctal(char *cp, int len, long long value) { int tempLength; @@ -205,9 +207,7 @@ static int writeTarHeader(struct TarBallInfo *tbInfo, putOctal(header.mode, sizeof(header.mode), statbuf->st_mode & 07777); putOctal(header.uid, sizeof(header.uid), statbuf->st_uid); putOctal(header.gid, sizeof(header.gid), statbuf->st_gid); - if (sizeof(header.size) != sizeof("00000000000")) - BUG_tar_header_size(); - strcpy(header.size, "00000000000"); /* Regular file size is handled later */ + memset(header.size, '0', sizeof(header.size)-1); /* Regular file size is handled later */ putOctal(header.mtime, sizeof(header.mtime), statbuf->st_mtime); strcpy(header.magic, "ustar "); @@ -401,7 +401,7 @@ static int writeFileToTarball(const char *fileName, struct stat *statbuf, /* tar will be corrupted. So bail out. */ /* NB: GNU tar 1.16 warns and pads with zeroes */ /* or even seeks back and updates header */ - bb_error_msg_and_die("short read from %s", fileName); + bb_error_msg_and_die("short read from %s, aborting", fileName); } /* Check that file did not grow in between? */ /* if (safe_read(inputFileFd,1) == 1) warn but continue? */ @@ -461,8 +461,7 @@ static int writeTarFile(const int tar_fd, const int verboseFlag, dup2(gzipDataPipe[0], 0); close(gzipDataPipe[1]); - if (tbInfo.tarFd != 1) - dup2(tbInfo.tarFd, 1); + dup2(tbInfo.tarFd, 1); close(gzipStatusPipe[0]); fcntl(gzipStatusPipe[1], F_SETFD, FD_CLOEXEC); /* close on exec shows success */ @@ -831,7 +830,7 @@ int tar_main(int argc, char **argv) flags = O_RDONLY; } - if ((tar_filename[0] == '-') && (tar_filename[1] == '\0')) { + if (tar_filename[0] == '-' && !tar_filename[1]) { tar_handle->src_fd = fileno(tar_stream); tar_handle->seek = seek_by_read; } else { @@ -860,23 +859,24 @@ int tar_main(int argc, char **argv) writeTarFile(tar_handle->src_fd, verboseFlag, opt & TAR_OPT_DEREFERENCE, tar_handle->accept, tar_handle->reject, zipMode); - } else { - while (get_header_ptr(tar_handle) == EXIT_SUCCESS) - /* nothing */; - - /* Check that every file that should have been extracted was */ - while (tar_handle->accept) { - if (!find_list_entry(tar_handle->reject, tar_handle->accept->data) - && !find_list_entry(tar_handle->passed, tar_handle->accept->data) - ) { - bb_error_msg_and_die("%s: not found in archive", - tar_handle->accept->data); - } - tar_handle->accept = tar_handle->accept->link; - } + /* NB: writeTarFile() closes tar_handle->src_fd */ + return EXIT_SUCCESS; } - if (ENABLE_FEATURE_CLEAN_UP && tar_handle->src_fd != STDIN_FILENO) + while (get_header_ptr(tar_handle) == EXIT_SUCCESS) + /* nothing */; + + /* Check that every file that should have been extracted was */ + while (tar_handle->accept) { + if (!find_list_entry(tar_handle->reject, tar_handle->accept->data) + && !find_list_entry(tar_handle->passed, tar_handle->accept->data) + ) { + bb_error_msg_and_die("%s: not found in archive", + tar_handle->accept->data); + } + tar_handle->accept = tar_handle->accept->link; + } + if (ENABLE_FEATURE_CLEAN_UP /* && tar_handle->src_fd != STDIN_FILENO */) close(tar_handle->src_fd); return EXIT_SUCCESS; diff --git a/libbb/procps.c b/libbb/procps.c index 52203ee9a..2581d03b2 100644 --- a/libbb/procps.c +++ b/libbb/procps.c @@ -35,6 +35,7 @@ void clear_username_cache(void) clear_cache(&groupname); } +#if 0 /* more generic, but we don't need that yet */ /* Returns -N-1 if not found. */ /* cp->cache[N] is allocated and must be filled in this case */ static int get_cached(cache_t *cp, unsigned id) @@ -48,25 +49,28 @@ static int get_cached(cache_t *cp, unsigned id) cp->cache[i++].id = id; return -i; } +#endif + +typedef char* ug_func(char *name, long uid, int bufsize); +static char* get_cached(cache_t *cp, unsigned id, ug_func* fp) +{ + int i; + for (i = 0; i < cp->size; i++) + if (cp->cache[i].id == id) + return cp->cache[i].name; + i = cp->size++; + cp->cache = xrealloc(cp->cache, cp->size * sizeof(*cp->cache)); + cp->cache[i].id = id; + fp(cp->cache[i].name, id, sizeof(cp->cache[i].name)); + return cp->cache[i].name; +} const char* get_cached_username(uid_t uid) { - int i = get_cached(&username, uid); - if (i < 0) { - i = -i - 1; - bb_getpwuid(username.cache[i].name, uid, - sizeof(username.cache[i].name)); - } - return username.cache[i].name; + return get_cached(&username, uid, bb_getpwuid); } -const char* get_cached_groupname(uid_t uid) +const char* get_cached_groupname(gid_t gid) { - int i = get_cached(&groupname, uid); - if (i < 0) { - i = -i - 1; - bb_getgrgid(groupname.cache[i].name, uid, - sizeof(groupname.cache[i].name)); - } - return username.cache[i].name; + return get_cached(&groupname, gid, bb_getgrgid); } |