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