11dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project/*- 21dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project * Copyright (c) 1990, 1993 31dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project * The Regents of the University of California. All rights reserved. 41dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project * 51dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project * This code is derived from software contributed to Berkeley by 61dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project * Chris Torek. 71dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project * 81dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project * Redistribution and use in source and binary forms, with or without 91dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project * modification, are permitted provided that the following conditions 101dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project * are met: 111dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project * 1. Redistributions of source code must retain the above copyright 121dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project * notice, this list of conditions and the following disclaimer. 131dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project * 2. Redistributions in binary form must reproduce the above copyright 141dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project * notice, this list of conditions and the following disclaimer in the 151dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project * documentation and/or other materials provided with the distribution. 161dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project * 3. Neither the name of the University nor the names of its contributors 171dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project * may be used to endorse or promote products derived from this software 181dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project * without specific prior written permission. 191dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project * 201dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND 211dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 221dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 231dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE 241dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 251dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 261dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 271dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 281dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 291dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 301dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project * SUCH DAMAGE. 311dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project */ 321dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project 33677ee5647721df22f13909375d5d0e770a1a80bfElliott Hughes#if defined(LIBC_SCCS) && !defined(lint) 34677ee5647721df22f13909375d5d0e770a1a80bfElliott Hughesstatic char sccsid[] = "@(#)setvbuf.c 8.2 (Berkeley) 11/16/93"; 35677ee5647721df22f13909375d5d0e770a1a80bfElliott Hughes#endif /* LIBC_SCCS and not lint */ 36677ee5647721df22f13909375d5d0e770a1a80bfElliott Hughes#include <sys/cdefs.h> 37677ee5647721df22f13909375d5d0e770a1a80bfElliott Hughes__FBSDID("$FreeBSD$"); 38677ee5647721df22f13909375d5d0e770a1a80bfElliott Hughes 39677ee5647721df22f13909375d5d0e770a1a80bfElliott Hughes#include "namespace.h" 401dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project#include <stdio.h> 411dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project#include <stdlib.h> 42677ee5647721df22f13909375d5d0e770a1a80bfElliott Hughes#include "un-namespace.h" 431dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project#include "local.h" 44677ee5647721df22f13909375d5d0e770a1a80bfElliott Hughes#include "libc_private.h" 451dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project 461dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project/* 471dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project * Set one of the three kinds of buffering, optionally including 481dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project * a buffer. 491dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project */ 501dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Projectint 51677ee5647721df22f13909375d5d0e770a1a80bfElliott Hughessetvbuf(FILE * __restrict fp, char * __restrict buf, int mode, size_t size) 521dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project{ 531dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project int ret, flags; 541dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project size_t iosize; 551dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project int ttyflag; 561dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project 571dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project /* 581dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project * Verify arguments. The `int' limit on `size' is due to this 591dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project * particular implementation. Note, buf and size are ignored 601dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project * when setting _IONBF. 611dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project */ 621dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project if (mode != _IONBF) 631dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project if ((mode != _IOFBF && mode != _IOLBF) || (int)size < 0) 641dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project return (EOF); 651dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project 66677ee5647721df22f13909375d5d0e770a1a80bfElliott Hughes FLOCKFILE(fp); 671dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project /* 681dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project * Write current buffer, if any. Discard unread input (including 691dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project * ungetc data), cancel line buffering, and free old buffer if 701dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project * malloc()ed. We also clear any eof condition, as if this were 711dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project * a seek. 721dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project */ 731dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project ret = 0; 741dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project (void)__sflush(fp); 751dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project if (HASUB(fp)) 761dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project FREEUB(fp); 771dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project fp->_r = fp->_lbfsize = 0; 781dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project flags = fp->_flags; 791dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project if (flags & __SMBF) 801dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project free((void *)fp->_bf._base); 81677ee5647721df22f13909375d5d0e770a1a80bfElliott Hughes flags &= ~(__SLBF | __SNBF | __SMBF | __SOPT | __SOFF | __SNPT | __SEOF); 821dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project 831dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project /* If setting unbuffered mode, skip all the hard work. */ 841dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project if (mode == _IONBF) 851dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project goto nbf; 861dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project 871dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project /* 881dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project * Find optimal I/O size for seek optimization. This also returns 891dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project * a `tty flag' to suggest that we check isatty(fd), but we do not 901dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project * care since our caller told us how to buffer. 911dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project */ 921dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project flags |= __swhatbuf(fp, &iosize, &ttyflag); 931dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project if (size == 0) { 941dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project buf = NULL; /* force local allocation */ 951dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project size = iosize; 961dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project } 971dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project 981dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project /* Allocate buffer if needed. */ 991dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project if (buf == NULL) { 1001dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project if ((buf = malloc(size)) == NULL) { 1011dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project /* 1021dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project * Unable to honor user's request. We will return 1031dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project * failure, but try again with file system size. 1041dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project */ 1051dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project ret = EOF; 1061dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project if (size != iosize) { 1071dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project size = iosize; 1081dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project buf = malloc(size); 1091dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project } 1101dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project } 1111dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project if (buf == NULL) { 1121dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project /* No luck; switch to unbuffered I/O. */ 1131dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Projectnbf: 1141dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project fp->_flags = flags | __SNBF; 1151dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project fp->_w = 0; 1161dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project fp->_bf._base = fp->_p = fp->_nbuf; 1171dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project fp->_bf._size = 1; 118f582340a6a48588aa50da17e1620e8f91b146941Kenny Root FUNLOCKFILE(fp); 1191dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project return (ret); 1201dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project } 1211dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project flags |= __SMBF; 1221dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project } 1231dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project 1241dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project /* 1251dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project * Kill any seek optimization if the buffer is not the 1261dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project * right size. 1271dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project * 1281dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project * SHOULD WE ALLOW MULTIPLES HERE (i.e., ok iff (size % iosize) == 0)? 1291dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project */ 1301dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project if (size != iosize) 1311dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project flags |= __SNPT; 1321dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project 1331dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project /* 134677ee5647721df22f13909375d5d0e770a1a80bfElliott Hughes * Fix up the FILE fields, and set __cleanup for output flush on 135677ee5647721df22f13909375d5d0e770a1a80bfElliott Hughes * exit (since we are buffered in some way). 1361dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project */ 1371dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project if (mode == _IOLBF) 1381dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project flags |= __SLBF; 1391dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project fp->_flags = flags; 1401dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project fp->_bf._base = fp->_p = (unsigned char *)buf; 1411dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project fp->_bf._size = size; 1421dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project /* fp->_lbfsize is still 0 */ 1431dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project if (flags & __SWR) { 1441dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project /* 1451dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project * Begin or continue writing: see __swsetup(). Note 1461dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project * that __SNBF is impossible (it was handled earlier). 1471dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project */ 1481dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project if (flags & __SLBF) { 1491dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project fp->_w = 0; 1501dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project fp->_lbfsize = -fp->_bf._size; 1511dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project } else 1521dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project fp->_w = size; 1531dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project } else { 1541dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project /* begin/continue reading, or stay in intermediate state */ 1551dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project fp->_w = 0; 1561dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project } 157677ee5647721df22f13909375d5d0e770a1a80bfElliott Hughes __cleanup = _cleanup; 1581dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project 159677ee5647721df22f13909375d5d0e770a1a80bfElliott Hughes FUNLOCKFILE(fp); 1601dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project return (ret); 1611dc9e472e19acfe6dc7f41e429236e7eef7ceda1The Android Open Source Project} 162