From 0082c47404ecdeb00e08e1aedbea0f8402cfc906 Mon Sep 17 00:00:00 2001 From: Owen Shepherd Date: Wed, 12 Nov 2014 20:37:29 +0000 Subject: [PATCH] fwrite: output line buffered content as we go along --- functions/stdio/fwrite.c | 67 ++++++++++------------------------------ 1 file changed, 16 insertions(+), 51 deletions(-) diff --git a/functions/stdio/fwrite.c b/functions/stdio/fwrite.c index c9396bf..d0efadb 100644 --- a/functions/stdio/fwrite.c +++ b/functions/stdio/fwrite.c @@ -15,83 +15,48 @@ #include #include -//TODO OS(2012-08-01): Ascertain purpose of lineend & potentially remove - -size_t _PDCLIB_fwrite_unlocked( const void * _PDCLIB_restrict ptr, - size_t size, size_t nmemb, +size_t _PDCLIB_fwrite_unlocked( const void *restrict vptr, + size_t size, size_t nmemb, FILE * _PDCLIB_restrict stream ) { if ( _PDCLIB_prepwrite( stream ) == EOF ) { return 0; } - _PDCLIB_size_t offset = 0; - //bool lineend = false; + + const char *restrict ptr = vptr; size_t nmemb_i; for ( nmemb_i = 0; nmemb_i < nmemb; ++nmemb_i ) { for ( size_t size_i = 0; size_i < size; ++size_i ) { - if ( ( stream->buffer[ stream->bufidx++ ] = ((char*)ptr)[ nmemb_i * size + size_i ] ) == '\n' ) - { - /* Remember last newline, in case we have to do a partial line-buffered flush */ - offset = stream->bufidx; - //lineend = true; - } - if ( stream->bufidx == stream->bufsize ) + char c = ptr[ nmemb_i * size + size_i ]; + stream->buffer[ stream->bufidx++ ] = c; + + if ( stream->bufidx == stream->bufsize || ( c == '\n' && stream->status & _IOLBF ) ) { if ( _PDCLIB_flushbuffer( stream ) == EOF ) { /* Returning number of objects completely buffered */ return nmemb_i; } - //lineend = false; - /* - * The entire buffer has been flushed; this means we have to - * reset our newline position as we have already written - * that part of the stream. - */ - offset = 0; } } - } - /* Fully-buffered streams are OK. Non-buffered streams must be flushed, - line-buffered streams only if there's a newline in the buffer. - */ - switch ( stream->status & ( _IONBF | _IOLBF ) ) - { - case _IONBF: - if ( _PDCLIB_flushbuffer( stream ) == EOF ) - { - /* We are in a pinch here. We have an error, which requires a - return value < nmemb. On the other hand, all objects have - been written to buffer, which means all the caller had to - do was removing the error cause, and re-flush the stream... - Catch 22. We'll return a value one short, to indicate the - error, and can't really do anything about the inconsistency. - */ - return nmemb_i - 1; - } - break; - case _IOLBF: - { - size_t bufidx = stream->bufidx; - stream->bufidx = offset; + + if ( stream->status & _IONBF ) + { if ( _PDCLIB_flushbuffer( stream ) == EOF ) { - /* See comment above. */ - stream->bufidx = bufidx; - return nmemb_i - 1; - } - stream->bufidx = bufidx - offset; - memmove( stream->buffer, stream->buffer + offset, stream->bufidx ); + /* Returning number of objects completely buffered */ + return nmemb_i; } + } } return nmemb_i; } -size_t fwrite( const void * _PDCLIB_restrict ptr, - size_t size, size_t nmemb, +size_t fwrite( const void * _PDCLIB_restrict ptr, + size_t size, size_t nmemb, FILE * _PDCLIB_restrict stream ) { _PDCLIB_flockfile( stream ); -- 2.40.0