X-Git-Url: https://pd.if.org/git/?a=blobdiff_plain;f=functions%2Fstdio%2Fputs.c;h=aa759dc17839d539d4bd936267c0e6ff09967970;hb=d865c4403fc91d1f1ac95ba76febcee9f429bb97;hp=a90e07863fbdb26185b9ae2827da22eed895b054;hpb=393020b6e48719d27699dea6b29e53025bbd5123;p=pdclib

diff --git a/functions/stdio/puts.c b/functions/stdio/puts.c
index a90e078..aa759dc 100644
--- a/functions/stdio/puts.c
+++ b/functions/stdio/puts.c
@@ -1,5 +1,3 @@
-/* $Id$ */
-
 /* puts( const char * )
 
    This file is part of the Public Domain C Library (PDCLib).
@@ -9,7 +7,8 @@
 #include <stdio.h>
 
 #ifndef REGTEST
-#include <_PDCLIB_glue.h>
+
+#include "_PDCLIB_glue.h"
 
 extern char * _PDCLIB_eol;
 
@@ -30,35 +29,39 @@ int puts( const char * s )
             }
         }
     }
-    s = _PDCLIB_eol;
-    while ( *s != '\0' )
+    stdout->buffer[ stdout->bufidx++ ] = '\n';
+    if ( ( stdout->bufidx == stdout->bufsize ) ||
+         ( stdout->status & ( _IOLBF | _IONBF ) ) )
     {
-        stdout->buffer[ stdout->bufidx++ ] = *s++;
-        if ( stdout->bufidx == stdout->bufsize )
-        {
-            if ( _PDCLIB_flushbuffer( stdout ) == EOF )
-            {
-                return EOF;
-            }
-        }
+        return _PDCLIB_flushbuffer( stdout );
     }
-    if ( stdout->status & ( _IOLBF | _IONBF ) )
+    else
     {
-        return _PDCLIB_flushbuffer( stdout );
+        return 0;
     }
-    return 0;
 }
 
 #endif
 
 #ifdef TEST
-#include <_PDCLIB_test.h>
+
+#include "_PDCLIB_test.h"
 
 int main( void )
 {
-    TESTCASE( puts( "SUCCESS testing puts()" ) >= 0 );
+    FILE * fh;
+    char const * message = "SUCCESS testing puts()";
+    char buffer[23];
+    buffer[22] = 'x';
+    TESTCASE( ( fh = freopen( testfile, "wb+", stdout ) ) != NULL );
+    TESTCASE( puts( message ) >= 0 );
+    rewind( fh );
+    TESTCASE( fread( buffer, 1, 22, fh ) == 22 );
+    TESTCASE( memcmp( buffer, message, 22 ) == 0 );
+    TESTCASE( buffer[22] == 'x' );
+    TESTCASE( fclose( fh ) == 0 );
+    TESTCASE( remove( testfile ) == 0 );
     return TEST_RESULTS;
 }
 
 #endif
-