4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync/** @file
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync Implementation of setvbuf as declared in <stdio.h>.
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync Copyright (c) 2010 - 2011, Intel Corporation. All rights reserved.<BR>
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync This program and the accompanying materials are licensed and made available
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync under the terms and conditions of the BSD License that accompanies this
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync distribution. The full text of the license may be found at
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync http://opensource.org/licenses/bsd-license.
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync Copyright (c) 1990, 1993
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync The Regents of the University of California. All rights reserved.
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync This code is derived from software contributed to Berkeley by
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync Chris Torek.
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync Redistribution and use in source and binary forms, with or without
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync modification, are permitted provided that the following conditions
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync are met:
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync - Redistributions of source code must retain the above copyright
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync notice, this list of conditions and the following disclaimer.
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync - Redistributions in binary form must reproduce the above copyright
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync notice, this list of conditions and the following disclaimer in the
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync documentation and/or other materials provided with the distribution.
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync - Neither the name of the University nor the names of its contributors
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync may be used to endorse or promote products derived from this software
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync without specific prior written permission.
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDERS OR CONTRIBUTORS BE
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync POSSIBILITY OF SUCH DAMAGE.
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync NetBSD: setvbuf.c,v 1.17 2003/08/07 16:43:31 agc Exp
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync setvbuf.c 8.2 (Berkeley) 11/16/93
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync**/
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync#include <LibConfig.h>
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync#include <assert.h>
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync#include <errno.h>
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync#include <stdio.h>
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync#include <stdlib.h>
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync#include <wchar.h>
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync#include "reentrant.h"
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync#include "local.h"
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync#include <MainData.h>
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync/*
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync * Set one of the three kinds of buffering, optionally including
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync * a buffer.
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync */
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsyncint
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsyncsetvbuf(FILE *fp, char *buf, int mode, size_t size)
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync{
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync int ret, flags;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync size_t iosize;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync int ttyflag;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync _DIAGASSERT(fp != NULL);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync /* buf may be NULL */
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync if(fp == NULL) {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync errno = EINVAL;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync return (EOF);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync }
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync /*
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync * Verify arguments. The `int' limit on `size' is due to this
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync * particular implementation. Note, buf and size are ignored
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync * when setting _IONBF.
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync */
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync if (mode != _IONBF)
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync if ((mode != _IOFBF && mode != _IOLBF) || (int)size < 0)
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync return (-1);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync FLOCKFILE(fp);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync /*
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync * Write current buffer, if any. Discard unread input (including
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync * ungetc data), cancel line buffering, and free old buffer if
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync * malloc()ed. We also clear any eof condition, as if this were
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync * a seek.
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync */
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync ret = 0;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync (void)__sflush(fp);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync if (HASUB(fp))
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync FREEUB(fp);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync WCIO_FREE(fp);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync fp->_r = fp->_lbfsize = 0;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync flags = fp->_flags;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync if (flags & __SMBF)
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync free((void *)fp->_bf._base);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync flags &= ~(__SLBF | __SNBF | __SMBF | __SOPT | __SNPT | __SEOF);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync /* If setting unbuffered mode, skip all the hard work. */
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync if (mode == _IONBF)
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync goto nbf;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync /*
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync * Find optimal I/O size for seek optimization. This also returns
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync * a `tty flag' to suggest that we check isatty(fd), but we do not
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync * care since our caller told us how to buffer.
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync */
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync flags |= __swhatbuf(fp, &iosize, &ttyflag);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync if (size == 0) {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync buf = NULL; /* force local allocation */
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync size = iosize;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync }
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync /* Allocate buffer if needed. */
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync if (buf == NULL) {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync if ((buf = malloc(size)) == NULL) {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync /*
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync * Unable to honor user's request. We will return
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync * failure, but try again with file system size.
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync */
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync ret = -1;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync if (size != iosize) {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync size = iosize;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync buf = malloc(size);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync }
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync }
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync if (buf == NULL) {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync /* No luck; switch to unbuffered I/O. */
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsyncnbf:
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync fp->_flags = (unsigned short)(flags | __SNBF);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync fp->_w = 0;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync fp->_bf._base = fp->_p = fp->_nbuf;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync fp->_bf._size = 1;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync FUNLOCKFILE(fp);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync return (ret);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync }
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync flags |= __SMBF;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync }
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync /*
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync * Kill any seek optimization if the buffer is not the
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync * right size.
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync *
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync * SHOULD WE ALLOW MULTIPLES HERE (i.e., ok iff (size % iosize) == 0)?
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync */
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync if (size != iosize)
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync flags |= __SNPT;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync /*
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync * Fix up the FILE fields, and set gMD->cleanup for output flush on
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync * exit (since we are buffered in some way).
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync */
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync if (mode == _IOLBF)
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync flags |= __SLBF;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync fp->_flags = (unsigned short)flags;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync fp->_bf._base = fp->_p = (unsigned char *)buf;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync fp->_bf._size = (int)size;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync /* fp->_lbfsize is still 0 */
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync if (flags & __SWR) {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync /*
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync * Begin or continue writing: see __swsetup(). Note
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync * that __SNBF is impossible (it was handled earlier).
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync */
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync if (flags & __SLBF) {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync fp->_w = 0;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync fp->_lbfsize = -fp->_bf._size;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync } else
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync fp->_w = (int)size;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync } else {
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync /* begin/continue reading, or stay in intermediate state */
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync fp->_w = 0;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync }
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync gMD->cleanup = _cleanup;
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync FUNLOCKFILE(fp);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync return (ret);
4fd606d1f5abe38e1f42c38de1d2e895166bd0f4vboxsync}