X-Git-Url: https://pd.if.org/git/?a=blobdiff_plain;f=functions%2Fstdio%2Fsetbuf.c;h=6c32072876efd1afe09ae7b14ec2aece96c564c5;hb=d865c4403fc91d1f1ac95ba76febcee9f429bb97;hp=d9b722b88f46ba37f82fd342bd2a9eefa85eef2f;hpb=d02f38605b53cdff5460cc6b9e1b2a80c3a2ba4c;p=pdclib

diff --git a/functions/stdio/setbuf.c b/functions/stdio/setbuf.c
index d9b722b..6c32072 100644
--- a/functions/stdio/setbuf.c
+++ b/functions/stdio/setbuf.c
@@ -1,7 +1,3 @@
-/* $Id$ */
-
-/* Release $Name$ */
-
 /* setbuf( FILE *, char * )
 
    This file is part of the Public Domain C Library (PDCLib).
@@ -12,9 +8,8 @@
 
 #ifndef REGTEST
 
-void setbuf( FILE * _PDCLIB_restrict stream, char * _PDCLIB_restrict buf )
+void setbuf( struct _PDCLIB_file_t * _PDCLIB_restrict stream, char * _PDCLIB_restrict buf )
 {
-    /* TODO: Only allowed on a "virgin" stream; add check. */
     if ( buf == NULL )
     {
         setvbuf( stream, buf, _IONBF, BUFSIZ );
@@ -28,11 +23,32 @@ void setbuf( FILE * _PDCLIB_restrict stream, char * _PDCLIB_restrict buf )
 #endif
 
 #ifdef TEST
-#include <_PDCLIB_test.h>
+
+#include "_PDCLIB_test.h"
+
+#include <stdlib.h>
 
 int main( void )
 {
-    TESTCASE( NO_TESTDRIVER );
+    /* TODO: Extend testing once setvbuf() is finished. */
+#ifndef REGTEST
+    char buffer[ BUFSIZ + 1 ];
+    FILE * fh;
+    /* full buffered */
+    TESTCASE( ( fh = tmpfile() ) != NULL );
+    setbuf( fh, buffer );
+    TESTCASE( fh->buffer == buffer );
+    TESTCASE( fh->bufsize == BUFSIZ );
+    TESTCASE( ( fh->status & ( _IOFBF | _IONBF | _IOLBF ) ) == _IOFBF );
+    TESTCASE( fclose( fh ) == 0 );
+    /* not buffered */
+    TESTCASE( ( fh = tmpfile() ) != NULL );
+    setbuf( fh, NULL );
+    TESTCASE( ( fh->status & ( _IOFBF | _IONBF | _IOLBF ) ) == _IONBF );
+    TESTCASE( fclose( fh ) == 0 );
+#else
+    puts( " NOTEST setbuf() test driver is PDCLib-specific." );
+#endif
     return TEST_RESULTS;
 }