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 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 | ||
20 | static off_t bb_full_fd_action(int src_fd, int dst_fd, off_t size) | 19 | static 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. |
114 | int xopen(const char *pathname, int flags) | 114 | int 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. | ||
126 | int 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. |
123 | int xopen3(const char *pathname, int flags, int mode) | 132 | int 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. | ||
144 | int open_or_warn(const char *pathname, int flags) | ||
145 | { | ||
146 | return open3_or_warn(pathname, flags, 0666); | ||
147 | } | ||
148 | |||
134 | void xunlink(const char *pathname) | 149 | void 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) | |||
184 | void die_if_ferror(FILE *fp, const char *fn) | 199 | void 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 | } |