56 lines
886 B
C
56 lines
886 B
C
/*
|
|
* 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;
|
|
}
|