diff options
author | Corinna Vinschen <corinna@vinschen.de> | 2016-03-13 01:41:21 +0300 |
---|---|---|
committer | Corinna Vinschen <corinna@vinschen.de> | 2016-03-13 01:41:21 +0300 |
commit | 1eb6db6efb29b41d696db97a95aaaeee2b84c768 (patch) | |
tree | 89846c53aef688e9e5ce46d834de9063cba8f0e1 /newlib/libc/stdio | |
parent | 99d0e2341d3ff37b98eff9163b195b715ff8bddf (diff) |
Update setvbuf to latest OpenBSD implementation
Newlib's setvbuf function is very old and has two bugs:
- It sets the SRD/SWR flags incorrectly in case of files opened for
reading and writing.
See https://cygwin.com/ml/cygwin/2016-03/msg00180.html
for a desription of the effect.
- It always sets the buffer size to BUFSIZ if it's not provided by
the application, independent of the optimal blocksize for the
underlying IO device.
Update setvbuf to latest code from OpenBSD to fix both problems.
* libc/stdio/setvbuf.c (setvbuf): Import latest OpenBSD
implementation.
Signed-off-by: Corinna Vinschen <corinna@vinschen.de>
Diffstat (limited to 'newlib/libc/stdio')
-rw-r--r-- | newlib/libc/stdio/setvbuf.c | 117 |
1 files changed, 75 insertions, 42 deletions
diff --git a/newlib/libc/stdio/setvbuf.c b/newlib/libc/stdio/setvbuf.c index 41bdff6b6..52dd306e5 100644 --- a/newlib/libc/stdio/setvbuf.c +++ b/newlib/libc/stdio/setvbuf.c @@ -104,21 +104,20 @@ _DEFUN(setvbuf, (fp, buf, mode, size), { int ret = 0; struct _reent *reent = _REENT; + size_t iosize; + int ttyflag; CHECK_INIT (reent, fp); - _newlib_flockfile_start (fp); - /* * Verify arguments. The `int' limit on `size' is due to this - * particular implementation. + * particular implementation. Note, buf and size are ignored + * when setting _IONBF. */ - - if ((mode != _IOFBF && mode != _IOLBF && mode != _IONBF) || (int)(_POINTER_INT) size < 0) - { - _newlib_flockfile_exit (fp); + if (mode != _IONBF) + if ((mode != _IOFBF && mode != _IOLBF) || (int)(_POINTER_INT) size < 0) return (EOF); - } + /* * Write current buffer, if any; drop read count, if any. @@ -126,34 +125,49 @@ _DEFUN(setvbuf, (fp, buf, mode, size), * Free old buffer if it was from malloc(). Clear line and * non buffer flags, and clear malloc flag. */ - + _newlib_flockfile_start (fp); _fflush_r (reent, fp); - fp->_r = 0; - fp->_lbfsize = 0; + if (HASUB(fp)) + FREEUB(reent, fp); + fp->_r = fp->_lbfsize = 0; if (fp->_flags & __SMBF) _free_r (reent, (_PTR) fp->_bf._base); - fp->_flags &= ~(__SLBF | __SNBF | __SMBF); + fp->_flags &= ~(__SLBF | __SNBF | __SMBF | __SOPT | __SNPT | __SEOF); if (mode == _IONBF) goto nbf; /* - * Allocate buffer if needed. */ + * Find optimal I/O size for seek optimization. This also returns + * a `tty flag' to suggest that we check isatty(fd), but we do not + * care since our caller told us how to buffer. + */ + fp->_flags |= __swhatbuf_r (reent, fp, &iosize, &ttyflag); + if (size == 0) + { + buf = NULL; + size = iosize; + } + + /* Allocate buffer if needed. */ if (buf == NULL) { - /* we need this here because malloc() may return a pointer - even if size == 0 */ - if (!size) size = BUFSIZ; if ((buf = malloc (size)) == NULL) { + /* + * Unable to honor user's request. We will return + * failure, but try again with file system size. + */ ret = EOF; - /* Try another size... */ - buf = malloc (BUFSIZ); - size = BUFSIZ; + if (size != iosize) + { + size = iosize; + buf = malloc (size); + } } if (buf == NULL) { - /* Can't allocate it, let's try another approach */ + /* No luck; switch to unbuffered I/O. */ nbf: fp->_flags |= __SNBF; fp->_w = 0; @@ -164,35 +178,54 @@ nbf: } fp->_flags |= __SMBF; } + /* - * Now put back whichever flag is needed, and fix _lbfsize - * if line buffered. Ensure output flush on exit if the - * stream will be buffered at all. - * If buf is NULL then make _lbfsize 0 to force the buffer - * to be flushed and hence malloced on first use + * We're committed to buffering from here, so make sure we've + * registered to flush buffers on exit. */ + if (!reent->__sdidinit) + __sinit(reent); - switch (mode) - { - case _IOLBF: - fp->_flags |= __SLBF; - fp->_lbfsize = buf ? -size : 0; - /* FALLTHROUGH */ - - case _IOFBF: - /* no flag */ - reent->__cleanup = _cleanup_r; - fp->_bf._base = fp->_p = (unsigned char *) buf; - fp->_bf._size = size; - break; - } - +#ifdef _FSEEK_OPTIMIZATION /* - * Patch up write count if necessary. + * Kill any seek optimization if the buffer is not the + * right size. + * + * SHOULD WE ALLOW MULTIPLES HERE (i.e., ok iff (size % iosize) == 0)? */ + if (size != iosize) + fp->_flags |= __SNPT; +#endif + /* + * Fix up the FILE fields, and set __cleanup for output flush on + * exit (since we are buffered in some way). + */ + if (mode == _IOLBF) + fp->_flags |= __SLBF; + reent->__cleanup = _cleanup_r; + fp->_bf._base = fp->_p = (unsigned char *) buf; + fp->_bf._size = size; + /* fp->_lbfsize is still 0 */ if (fp->_flags & __SWR) - fp->_w = fp->_flags & (__SLBF | __SNBF) ? 0 : size; + { + /* + * Begin or continue writing: see __swsetup(). Note + * that __SNBF is impossible (it was handled earlier). + */ + if (fp->_flags & __SLBF) + { + fp->_w = 0; + fp->_lbfsize = -fp->_bf._size; + } + else + fp->_w = size; + } + else + { + /* begin/continue reading, or stay in intermediate state */ + fp->_w = 0; + } _newlib_flockfile_end (fp); return 0; |