diff options
author | Denis Vlasenko | 2007-11-13 17:51:40 +0000 |
---|---|---|
committer | Denis Vlasenko | 2007-11-13 17:51:40 +0000 |
commit | a89d50fb9addfe0060f697a7e1d31a7b3010b2bb (patch) | |
tree | 81b252e0f0a931b1485eaefd89e900859ec8fcfb /coreutils/tr.c | |
parent | 130f559f4dcb64fc7dbf55aebaf0def4f601ac2f (diff) | |
download | busybox-a89d50fb9addfe0060f697a7e1d31a7b3010b2bb.zip busybox-a89d50fb9addfe0060f697a7e1d31a7b3010b2bb.tar.gz |
tr: properly detect read errors (Fernando Silveira <fsilveira@gmail.com>)
readprofile: use xwrite
function old new delta
tr_main 679 677 -2
readprofile_main 1964 1944 -20
Diffstat (limited to 'coreutils/tr.c')
-rw-r--r-- | coreutils/tr.c | 8 |
1 files changed, 5 insertions, 3 deletions
diff --git a/coreutils/tr.c b/coreutils/tr.c index 9521bbc..e60e08b 100644 --- a/coreutils/tr.c +++ b/coreutils/tr.c @@ -173,7 +173,8 @@ int tr_main(int argc, char **argv) int idx = 1; int i; smalluint flags = 0; - size_t read_chars = 0, in_index = 0, out_index = 0, c, coded, last = -1; + ssize_t read_chars = 0; + size_t in_index = 0, out_index = 0, c, coded, last = -1; RESERVE_CONFIG_UBUFFER(output, BUFSIZ); RESERVE_CONFIG_BUFFER(vector, ASCII+1); RESERVE_CONFIG_BUFFER(invec, ASCII+1); @@ -223,8 +224,9 @@ int tr_main(int argc, char **argv) } read_chars = read(STDIN_FILENO, tr_buf, BUFSIZ); if (read_chars <= 0) { - if (write(STDOUT_FILENO, (char *)output, out_index) != out_index) - bb_perror_msg(bb_msg_write_error); + xwrite(STDOUT_FILENO, (char *)output, out_index); + if (read_chars < 0) + bb_perror_msg_and_die(bb_msg_read_error); exit(EXIT_SUCCESS); } in_index = 0; |