aboutsummaryrefslogtreecommitdiff
path: root/libbb
diff options
context:
space:
mode:
Diffstat (limited to 'libbb')
-rw-r--r--libbb/copy_file.c10
-rw-r--r--libbb/copyfd.c6
-rw-r--r--libbb/wfopen_input.c2
-rw-r--r--libbb/xfuncs.c32
4 files changed, 31 insertions, 19 deletions
diff --git a/libbb/copy_file.c b/libbb/copy_file.c
index 7d85920d9..700564212 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)
217 add_to_ino_dev_hashtable(&source_stat, dest); 217 add_to_ino_dev_hashtable(&source_stat, dest);
218 } 218 }
219 219
220 src_fd = open(source, O_RDONLY); 220 src_fd = open_or_warn(source, O_RDONLY);
221 if (src_fd == -1) { 221 if (src_fd < 0) {
222 bb_perror_msg("cannot open '%s'", source);
223 return -1; 222 return -1;
224 } 223 }
225 224
@@ -237,9 +236,8 @@ int copy_file(const char *source, const char *dest, int flags)
237 return ovr; 236 return ovr;
238 } 237 }
239 /* It shouldn't exist. If it exists, do not open (symlink attack?) */ 238 /* It shouldn't exist. If it exists, do not open (symlink attack?) */
240 dst_fd = open(dest, O_WRONLY|O_CREAT|O_EXCL, source_stat.st_mode); 239 dst_fd = open3_or_warn(dest, O_WRONLY|O_CREAT|O_EXCL, source_stat.st_mode);
241 if (dst_fd == -1) { 240 if (dst_fd < 0) {
242 bb_perror_msg("cannot open '%s'", dest);
243 close(src_fd); 241 close(src_fd);
244 return -1; 242 return -1;
245 } 243 }
diff --git a/libbb/copyfd.c b/libbb/copyfd.c
index aa8fbb967..3255e424a 100644
--- a/libbb/copyfd.c
+++ b/libbb/copyfd.c
@@ -14,14 +14,13 @@
14#define BUFSIZ 4096 14#define BUFSIZ 4096
15#endif 15#endif
16 16
17/* Used by NOFORK applets (e.g. cat) - must be very careful 17/* Used by NOFORK applets (e.g. cat) - must not use xmalloc */
18 * when calling xfuncs, allocating memory, with signals, termios, etc... */
19 18
20static off_t bb_full_fd_action(int src_fd, int dst_fd, off_t size) 19static off_t bb_full_fd_action(int src_fd, int dst_fd, off_t size)
21{ 20{
22 int status = -1; 21 int status = -1;
23 off_t total = 0; 22 off_t total = 0;
24 RESERVE_CONFIG_BUFFER(buffer, BUFSIZ); 23 char buffer[BUFSIZ];
25 24
26 if (src_fd < 0) 25 if (src_fd < 0)
27 goto out; 26 goto out;
@@ -63,7 +62,6 @@ static off_t bb_full_fd_action(int src_fd, int dst_fd, off_t size)
63 } 62 }
64 } 63 }
65 out: 64 out:
66 RELEASE_CONFIG_BUFFER(buffer);
67 return status ? -1 : total; 65 return status ? -1 : total;
68} 66}
69 67
diff --git a/libbb/wfopen_input.c b/libbb/wfopen_input.c
index 7a11dacd7..1b4928e1f 100644
--- a/libbb/wfopen_input.c
+++ b/libbb/wfopen_input.c
@@ -11,7 +11,7 @@
11 * is a command line arg. Since often that arg is '-' (meaning stdin), 11 * is a command line arg. Since often that arg is '-' (meaning stdin),
12 * we avoid testing everywhere by consolidating things in this routine. 12 * we avoid testing everywhere by consolidating things in this routine.
13 * 13 *
14 * Note: We also consider "" to main stdin (for 'cmp' at least). 14 * Note: we also consider "" to mean stdin (for 'cmp' at least).
15 */ 15 */
16 16
17#include "libbb.h" 17#include "libbb.h"
diff --git a/libbb/xfuncs.c b/libbb/xfuncs.c
index dde91a2ba..870d736b3 100644
--- a/libbb/xfuncs.c
+++ b/libbb/xfuncs.c
@@ -106,31 +106,46 @@ FILE *xfopen(const char *path, const char *mode)
106{ 106{
107 FILE *fp = fopen(path, mode); 107 FILE *fp = fopen(path, mode);
108 if (fp == NULL) 108 if (fp == NULL)
109 bb_perror_msg_and_die("%s", path); 109 bb_perror_msg_and_die("cannot open '%s'", path);
110 return fp; 110 return fp;
111} 111}
112 112
113// Die if we can't open an existing file and return an fd. 113// Die if we can't open a file and return a fd.
114int xopen(const char *pathname, int flags) 114int xopen3(const char *pathname, int flags, int mode)
115{ 115{
116 //if (ENABLE_DEBUG && (flags & O_CREAT)) 116 int ret;
117 // bb_error_msg_and_die("xopen() with O_CREAT"); 117
118 ret = open(pathname, flags, mode);
119 if (ret < 0) {
120 bb_perror_msg_and_die("cannot open '%s'", pathname);
121 }
122 return ret;
123}
118 124
125// Die if we can't open an existing file and return a fd.
126int xopen(const char *pathname, int flags)
127{
119 return xopen3(pathname, flags, 0666); 128 return xopen3(pathname, flags, 0666);
120} 129}
121 130
122// Die if we can't open a new file and return an fd. 131// Warn if we can't open a file and return a fd.
123int xopen3(const char *pathname, int flags, int mode) 132int open3_or_warn(const char *pathname, int flags, int mode)
124{ 133{
125 int ret; 134 int ret;
126 135
127 ret = open(pathname, flags, mode); 136 ret = open(pathname, flags, mode);
128 if (ret < 0) { 137 if (ret < 0) {
129 bb_perror_msg_and_die("%s", pathname); 138 bb_perror_msg("cannot open '%s'", pathname);
130 } 139 }
131 return ret; 140 return ret;
132} 141}
133 142
143// Warn if we can't open a file and return a fd.
144int open_or_warn(const char *pathname, int flags)
145{
146 return open3_or_warn(pathname, flags, 0666);
147}
148
134void xunlink(const char *pathname) 149void xunlink(const char *pathname)
135{ 150{
136 if (unlink(pathname)) 151 if (unlink(pathname))
@@ -184,6 +199,7 @@ off_t xlseek(int fd, off_t offset, int whence)
184void die_if_ferror(FILE *fp, const char *fn) 199void die_if_ferror(FILE *fp, const char *fn)
185{ 200{
186 if (ferror(fp)) { 201 if (ferror(fp)) {
202 /* doesn't set useful errno */
187 bb_error_msg_and_die("%s: I/O error", fn); 203 bb_error_msg_and_die("%s: I/O error", fn);
188 } 204 }
189} 205}