diff options
Diffstat (limited to 'libbb')
-rw-r--r-- | libbb/copy_file.c | 10 | ||||
-rw-r--r-- | libbb/copyfd.c | 6 | ||||
-rw-r--r-- | libbb/wfopen_input.c | 2 | ||||
-rw-r--r-- | libbb/xfuncs.c | 32 |
4 files changed, 31 insertions, 19 deletions
diff --git a/libbb/copy_file.c b/libbb/copy_file.c index 7d85920..7005642 100644 --- a/libbb/copy_file.c +++ b/libbb/copy_file.c @@ -217,9 +217,8 @@ int copy_file(const char *source, const char *dest, int flags) add_to_ino_dev_hashtable(&source_stat, dest); } - src_fd = open(source, O_RDONLY); - if (src_fd == -1) { - bb_perror_msg("cannot open '%s'", source); + src_fd = open_or_warn(source, O_RDONLY); + if (src_fd < 0) { return -1; } @@ -237,9 +236,8 @@ int copy_file(const char *source, const char *dest, int flags) return ovr; } /* It shouldn't exist. If it exists, do not open (symlink attack?) */ - dst_fd = open(dest, O_WRONLY|O_CREAT|O_EXCL, source_stat.st_mode); - if (dst_fd == -1) { - bb_perror_msg("cannot open '%s'", dest); + dst_fd = open3_or_warn(dest, O_WRONLY|O_CREAT|O_EXCL, source_stat.st_mode); + if (dst_fd < 0) { close(src_fd); return -1; } diff --git a/libbb/copyfd.c b/libbb/copyfd.c index aa8fbb9..3255e42 100644 --- a/libbb/copyfd.c +++ b/libbb/copyfd.c @@ -14,14 +14,13 @@ #define BUFSIZ 4096 #endif -/* Used by NOFORK applets (e.g. cat) - must be very careful - * when calling xfuncs, allocating memory, with signals, termios, etc... */ +/* Used by NOFORK applets (e.g. cat) - must not use xmalloc */ static off_t bb_full_fd_action(int src_fd, int dst_fd, off_t size) { int status = -1; off_t total = 0; - RESERVE_CONFIG_BUFFER(buffer, BUFSIZ); + char buffer[BUFSIZ]; if (src_fd < 0) goto out; @@ -63,7 +62,6 @@ static off_t bb_full_fd_action(int src_fd, int dst_fd, off_t size) } } out: - RELEASE_CONFIG_BUFFER(buffer); return status ? -1 : total; } diff --git a/libbb/wfopen_input.c b/libbb/wfopen_input.c index 7a11dac..1b4928e 100644 --- a/libbb/wfopen_input.c +++ b/libbb/wfopen_input.c @@ -11,7 +11,7 @@ * is a command line arg. Since often that arg is '-' (meaning stdin), * we avoid testing everywhere by consolidating things in this routine. * - * Note: We also consider "" to main stdin (for 'cmp' at least). + * Note: we also consider "" to mean stdin (for 'cmp' at least). */ #include "libbb.h" diff --git a/libbb/xfuncs.c b/libbb/xfuncs.c index dde91a2..870d736 100644 --- a/libbb/xfuncs.c +++ b/libbb/xfuncs.c @@ -106,31 +106,46 @@ FILE *xfopen(const char *path, const char *mode) { FILE *fp = fopen(path, mode); if (fp == NULL) - bb_perror_msg_and_die("%s", path); + bb_perror_msg_and_die("cannot open '%s'", path); return fp; } -// Die if we can't open an existing file and return an fd. -int xopen(const char *pathname, int flags) +// Die if we can't open a file and return a fd. +int xopen3(const char *pathname, int flags, int mode) { - //if (ENABLE_DEBUG && (flags & O_CREAT)) - // bb_error_msg_and_die("xopen() with O_CREAT"); + int ret; + + ret = open(pathname, flags, mode); + if (ret < 0) { + bb_perror_msg_and_die("cannot open '%s'", pathname); + } + return ret; +} +// Die if we can't open an existing file and return a fd. +int xopen(const char *pathname, int flags) +{ return xopen3(pathname, flags, 0666); } -// Die if we can't open a new file and return an fd. -int xopen3(const char *pathname, int flags, int mode) +// Warn if we can't open a file and return a fd. +int open3_or_warn(const char *pathname, int flags, int mode) { int ret; ret = open(pathname, flags, mode); if (ret < 0) { - bb_perror_msg_and_die("%s", pathname); + bb_perror_msg("cannot open '%s'", pathname); } return ret; } +// Warn if we can't open a file and return a fd. +int open_or_warn(const char *pathname, int flags) +{ + return open3_or_warn(pathname, flags, 0666); +} + void xunlink(const char *pathname) { if (unlink(pathname)) @@ -184,6 +199,7 @@ off_t xlseek(int fd, off_t offset, int whence) void die_if_ferror(FILE *fp, const char *fn) { if (ferror(fp)) { + /* doesn't set useful errno */ bb_error_msg_and_die("%s: I/O error", fn); } } |