X-Git-Url: https://git.gag.com/?a=blobdiff_plain;f=functions%2Fstdio%2Ffputs.c;h=2c5e1fa1e1d78b1ee7e17cd5410c1cf631536ffe;hb=f2c41eedc0c43f9ea65c8e0fac0b642cb7a74b68;hp=748fa5aff414ed8436951c48afed57e52f9cb36b;hpb=fae6eaa5fd1857623b296e7aad3ad97b9923f06e;p=fw%2Fpdclib

diff --git a/functions/stdio/fputs.c b/functions/stdio/fputs.c
index 748fa5a..2c5e1fa 100644
--- a/functions/stdio/fputs.c
+++ b/functions/stdio/fputs.c
@@ -9,52 +9,60 @@
 #include <stdio.h>
 
 #ifndef REGTEST
+#include <_PDCLIB_glue.h>
 
 int fputs( const char * _PDCLIB_restrict s, struct _PDCLIB_file_t * _PDCLIB_restrict stream )
 {
-    /* FIXME: This is devoid of any error checking (file writeable? r/w
-       constraints honored?)
-    */
-    /* FIXME: Proper buffering handling. */
-    while ( stream->bufidx < stream->bufsize )
+    if ( _PDCLIB_prepwrite( stream ) == EOF )
     {
-        if ( ( stream->buffer[stream->bufidx++] = *(s++) ) == '\0' )
-        {
-            break;
-        }
+        return EOF;
     }
-    fflush( stream );
-    if ( *(s-1) != '\0' )
+    while ( *s != '\0' )
     {
-        return fputs( s, stream );
+        /* Unbuffered and line buffered streams get flushed when fputs() does
+           write the terminating end-of-line. All streams get flushed if the
+           buffer runs full.
+        */
+        stream->buffer[ stream->bufidx++ ] = *s;
+        if ( ( stream->bufidx == stream->bufsize ) ||
+             ( ( stream->status & _IOLBF ) && *s == '\n' )
+           )
+        {
+            if ( _PDCLIB_flushbuffer( stream ) == EOF )
+            {
+                return EOF;
+            }
+        }
+        ++s;
     }
-    else
+    if ( stream->status & _IONBF )
     {
-        return 1;
+        if ( _PDCLIB_flushbuffer( stream ) == EOF )
+        {
+            return EOF;
+        }
     }
+    return 0;
 }
 
 #endif
-
 #ifdef TEST
 #include <_PDCLIB_test.h>
 
-#include <string.h>
-
 int main( void )
 {
+    char const * const message = "SUCCESS testing fputs()";
     FILE * fh;
-    char buffer[100];
-    char text[] = "SUCCESS testing fputs().";
-    TESTCASE( ( fh = fopen( "testfile", "w" ) ) != NULL );
-    TESTCASE( fputs( text, fh ) != EOF );
-    TESTCASE( fclose( fh ) == 0 );
-    TESTCASE( ( fh = fopen( "testfile", "r" ) ) != NULL );
-    TESTCASE( fread( buffer, 1, strlen( text ), fh ) == strlen( text ) );
-    TESTCASE( memcmp( buffer, text, strlen( text ) ) == 0 );
+    TESTCASE( ( fh = tmpfile() ) != NULL );
+    TESTCASE( fputs( message, fh ) >= 0 );
+    rewind( fh );
+    for ( size_t i = 0; i < 23; ++i )
+    {
+        TESTCASE( fgetc( fh ) == message[i] );
+    }
     TESTCASE( fclose( fh ) == 0 );
-    TESTCASE( remove( "testfile" ) == 0 );
     return TEST_RESULTS;
 }
 
 #endif
+