sortix-mirror/libc/fwrite.cpp

65 lines
1.9 KiB
C++

/*******************************************************************************
Copyright(C) Jonas 'Sortie' Termansen 2011, 2012.
This file is part of the Sortix C Library.
The Sortix C Library is free software: you can redistribute it and/or modify
it under the terms of the GNU Lesser General Public License as published by
the Free Software Foundation, either version 3 of the License, or (at your
option) any later version.
The Sortix C Library is distributed in the hope that it will be useful, but
WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
or FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public
License for more details.
You should have received a copy of the GNU Lesser General Public License
along with the Sortix C Library. If not, see <http://www.gnu.org/licenses/>.
fwrite.cpp
Writes data to a FILE.
*******************************************************************************/
#include <assert.h>
#include <stdio.h>
#include <errno.h>
extern "C" size_t fwrite(const void* ptr, size_t size, size_t nmemb, FILE* fp)
{
if ( !fp->write_func )
return errno = EBADF, 0;
fp->flags &= ~_FILE_LAST_READ; fp->flags |= _FILE_LAST_WRITE;
char* str = (char*) ptr;
size_t total = size * nmemb;
size_t sofar = 0;
while ( sofar < total )
{
size_t left = total - sofar;
if ( fp->flags & _FILE_NO_BUFFER || !fp->buffersize )
{
size_t ret = sofar + fp->write_func(str + sofar, 1, left, fp->user);
return ret;
}
size_t available = fp->buffersize - fp->bufferused;
if ( !available )
{
if ( fflush(fp) == 0 ) continue;
else return sofar;
}
size_t count = available < left ? available : left;
for ( size_t i = 0; i < count; i++ )
{
char c = str[sofar++];
fp->buffer[fp->bufferused++] = c;
assert(fp->bufferused <= fp->buffersize);
if ( c == '\n' || fp->buffersize == fp->bufferused )
break;
}
}
return sofar;
}