source: trunk/minix/lib/stdio/setvbuf.c@ 9

/*
 * setbuf.c - control buffering of a stream
 */
/* $Id: setvbuf.c,v 1.1.1.1 2005/04/21 14:56:36 beng Exp $ */

#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;
}