X-Git-Url: https://pd.if.org/git/?a=blobdiff_plain;f=functions%2Fstdio%2Fungetc.c;h=4e9388b40d7f3e83094a00331630bf8c2d832f6f;hb=d1aaf8b626ae8f729bc81dd7f15a16d8ccc0b227;hp=1bd7323ba80eb7345f26586de79d48912e386439;hpb=1d9d92ba957a0b8307c9a65c35867fde68e6533b;p=pdclib

diff --git a/functions/stdio/ungetc.c b/functions/stdio/ungetc.c
index 1bd7323..4e9388b 100644
--- a/functions/stdio/ungetc.c
+++ b/functions/stdio/ungetc.c
@@ -1,20 +1,68 @@
-/* ----------------------------------------------------------------------------
- * $Id$
- * ----------------------------------------------------------------------------
- * Public Domain C Library - http://pdclib.sourceforge.net
- * This code is Public Domain. Use, modify, and redistribute at will.
- * --------------------------------------------------------------------------*/
+/* $Id$ */
 
-int ungetc( int c, FILE * stream ) { /* TODO */ };
+/* ungetc( int, FILE * )
 
-/* PDPC code - unreviewed
+   This file is part of the Public Domain C Library (PDCLib).
+   Permission is granted to use, modify, and / or redistribute at will.
+*/
+
+#include <stdio.h>
+
+#ifndef REGTEST
+
+int ungetc_unlocked( int c, struct _PDCLIB_file_t * stream )
 {
-    if ((stream->ungetCh != -1) || (c == EOF))
+    if ( c == EOF || stream->ungetidx == _PDCLIB_UNGETCBUFSIZE )
     {
-        return (EOF);
+        return -1;
     }
-    stream->ungetCh = (unsigned char)c;
-    stream->quickText = 0;
-    return ((unsigned char)c);
+    return stream->ungetbuf[stream->ungetidx++] = (unsigned char) c;
 }
-*/
+
+int ungetc( int c, struct _PDCLIB_file_t * stream )
+{
+    flockfile( stream );
+    int r = ungetc_unlocked( c, stream );
+    funlockfile( stream);
+    return r;
+}
+
+#endif
+
+#ifdef TEST
+#include <_PDCLIB_test.h>
+#include <stdlib.h>
+
+const char* hellostr = "Hello, world!";
+
+int main( void )
+{
+    // Also see ftell() for some testing
+
+    // PDCLIB-18: fread ignores ungetc
+    size_t bufsz = strlen( hellostr ) + 1;
+    char * buf = malloc( bufsz );
+    FILE * fh;
+
+    // Also fgets
+    TESTCASE( ( fh = tmpfile() ) != NULL );
+    TESTCASE( fputs(hellostr, fh) == 0 );
+    rewind(fh);
+    TESTCASE( fgetc( fh ) == 'H' );
+    TESTCASE( ungetc( 'H', fh ) == 'H' );
+    TESTCASE( fgets( buf, bufsz, fh ) != NULL );
+    TESTCASE( strcmp( buf, hellostr ) == 0 );
+
+    // fread
+    rewind(fh);
+    TESTCASE( fgetc( fh ) == 'H' );
+    TESTCASE( ungetc( 'H', fh ) == 'H' );
+    TESTCASE( fread( buf, bufsz - 1, 1, fh ) == 1 );
+    TESTCASE( strncmp( buf, hellostr, bufsz - 1 ) == 0 );
+
+
+
+    return TEST_RESULTS;
+}
+
+#endif