diff options
author | shirosaki <shirosaki@b2dd03c8-39d4-4d8f-98ff-823fe69b080e> | 2011-12-26 13:49:31 +0000 |
---|---|---|
committer | shirosaki <shirosaki@b2dd03c8-39d4-4d8f-98ff-823fe69b080e> | 2011-12-26 13:49:31 +0000 |
commit | 68fde4449ac3a085a6fb47b44f983d547b9fece2 (patch) | |
tree | 1e698c5be445b53c691b6787993d90648c979440 /io.c | |
parent | fb37a14aa258d817391d7ff6314c2baefce4d81c (diff) | |
download | ruby-68fde4449ac3a085a6fb47b44f983d547b9fece2.tar.gz |
* io.c (rb_sys_fail_path): move the definition.
Move above for using it in set_binary_mode_with_seek_cur().
* io.c (set_binary_mode_with_seek_cur): fix improper seek cursor.
Seeking file cursor with setting binary mode has possibility to
cause infinite loop. Fixed the bug and refined error handling.
Introduced at r34043.
And cleanups as below.
Remove unnecessary parentheses of `fptr`.
Use return value of setmode().
* test/ruby/test_io_m17n.rb
(TestIO_M17N#test_seek_with_setting_binmode): add a test for abobe.
[ruby-core:41671] [Bug #5714]
git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@34132 b2dd03c8-39d4-4d8f-98ff-823fe69b080e
Diffstat (limited to 'io.c')
-rw-r--r-- | io.c | 58 |
1 files changed, 32 insertions, 26 deletions
@@ -374,6 +374,8 @@ rb_cloexec_fcntl_dupfd(int fd, int minfd) # endif #endif +#define rb_sys_fail_path(path) rb_sys_fail(NIL_P(path) ? 0 : RSTRING_PTR(path)) + static int io_fflush(rb_io_t *); #define NEED_NEWLINE_DECORATOR_ON_READ(fptr) ((fptr)->mode & FMODE_TEXTMODE) @@ -423,49 +425,55 @@ inline static int set_binary_mode_with_seek_cur(rb_io_t *fptr) { ssize_t read_size; long i; long newlines = 0; + long extra_max; char *p; - if (!rb_w32_fd_is_text((fptr)->fd)) return O_BINARY; + if (!rb_w32_fd_is_text(fptr->fd)) return O_BINARY; - if ((fptr)->rbuf.len == 0 || (fptr)->mode & FMODE_DUPLEX) { - setmode((fptr)->fd, O_BINARY); - return O_TEXT; + if (fptr->rbuf.len == 0 || fptr->mode & FMODE_DUPLEX) { + return setmode(fptr->fd, O_BINARY); } if (io_fflush(fptr) < 0) { rb_sys_fail(0); } errno = 0; - pos = lseek((fptr)->fd, 0, SEEK_CUR); + pos = lseek(fptr->fd, 0, SEEK_CUR); if (pos < 0 && errno) { if (errno == ESPIPE) - (fptr)->mode |= FMODE_DUPLEX; - setmode((fptr)->fd, O_BINARY); - return O_TEXT; + fptr->mode |= FMODE_DUPLEX; + return setmode(fptr->fd, O_BINARY); } - /* add extra offset for '\r' */ - p = (fptr)->rbuf.ptr+(fptr)->rbuf.off; - for (i = 0; i < (fptr)->rbuf.len; i++) { + /* add extra offset for removed '\r' in rbuf */ + extra_max = pos - fptr->rbuf.len; + p = fptr->rbuf.ptr + fptr->rbuf.off; + for (i = 0; i < fptr->rbuf.len; i++) { if (*p == '\n') newlines++; + if (extra_max == newlines) break; p++; } while (newlines >= 0) { - r = lseek((fptr)->fd, pos - (fptr)->rbuf.len - newlines, SEEK_SET); + r = lseek(fptr->fd, pos - fptr->rbuf.len - newlines, SEEK_SET); if (newlines == 0) break; - if (read_size = _read((fptr)->fd, (fptr)->rbuf.ptr, (fptr)->rbuf.len + newlines)) { - if (read_size == (fptr)->rbuf.len) { - lseek((fptr)->fd, r, SEEK_SET); - break; - } - else { - newlines--; - } + if (r < 0) { + newlines--; + continue; + } + read_size = _read(fptr->fd, fptr->rbuf.ptr, fptr->rbuf.len + newlines); + if (read_size < 0) { + rb_sys_fail_path(fptr->pathv); + } + if (read_size == fptr->rbuf.len) { + lseek(fptr->fd, r, SEEK_SET); + break; + } + else { + newlines--; } } - (fptr)->rbuf.off = 0; - (fptr)->rbuf.len = 0; - setmode((fptr)->fd, O_BINARY); - return O_TEXT; + fptr->rbuf.off = 0; + fptr->rbuf.len = 0; + return setmode(fptr->fd, O_BINARY); } #define SET_BINARY_MODE_WITH_SEEK_CUR(fptr) set_binary_mode_with_seek_cur(fptr) @@ -484,8 +492,6 @@ inline static int set_binary_mode_with_seek_cur(rb_io_t *fptr) { #define shutdown(a,b) 0 #endif -#define rb_sys_fail_path(path) rb_sys_fail(NIL_P(path) ? 0 : RSTRING_PTR(path)) - #if defined(_WIN32) #define is_socket(fd, path) rb_w32_is_socket(fd) #elif !defined(S_ISSOCK) |