123456789101112131415161718192021222324252627282930313233343536373839404142434445464748 |
- /*
- * setbuf.c - control buffering of a stream
- */
- /* $Id$ */
- #include <stdio.h>
- #include <stdlib.h>
- #include "loc_incl.h"
- extern void (*_clean)(void);
- int
- setvbuf(register FILE *stream, char *buf, int mode, size_t size)
- {
- int retval = 0;
- _clean = __cleanup;
- if (mode != _IOFBF && mode != _IOLBF && mode != _IONBF)
- return EOF;
- if (stream->_buf && io_testflag(stream,_IOMYBUF) )
- free((void *)stream->_buf);
- stream->_flags &= ~(_IOMYBUF | _IONBF | _IOLBF);
- if (buf && size <= 0) retval = EOF;
- if (!buf && (mode != _IONBF)) {
- if (size <= 0 || (buf = (char *) malloc(size)) == NULL) {
- retval = EOF;
- } else {
- stream->_flags |= _IOMYBUF;
- }
- }
- stream->_buf = (unsigned char *) buf;
- stream->_count = 0;
- stream->_flags |= mode;
- stream->_ptr = stream->_buf;
- if (!buf) {
- stream->_bufsiz = 1;
- } else {
- stream->_bufsiz = size;
- }
- return retval;
- }
|