pd.if.org Git - pdclib/blob - stdio/ungetc.c

   1 /* $Id$ */
   2
   3 /* ungetc( int, FILE * )
   4
   5    This file is part of the Public Domain C Library (PDCLib).
   6    Permission is granted to use, modify, and / or redistribute at will.
   7 */
   8
   9 #include <stdio.h>
  10
  11 #ifndef REGTEST
  12
  13 int ungetc_unlocked( int c, struct _PDCLIB_file_t * stream )
  14 {
  15     if ( c == EOF || stream->ungetidx == _PDCLIB_UNGETCBUFSIZE )
  16     {
  17         return -1;
  18     }
  19     return stream->ungetbuf[stream->ungetidx++] = (unsigned char) c;
  20 }
  21
  22 int ungetc( int c, struct _PDCLIB_file_t * stream )
  23 {
  24     flockfile( stream );
  25     int r = ungetc_unlocked( c, stream );
  26     funlockfile( stream);
  27     return r;
  28 }
  29
  30 #endif
  31
  32 #ifdef TEST
  33 #include <_PDCLIB_test.h>
  34
  35 const char* hellostr = "Hello, world!";
  36
  37 int main( void )
  38 {
  39     // Also see ftell() for some testing
  40
  41     // PDCLIB-18: fread ignores ungetc
  42     size_t bufsz = strlen( hellostr ) + 1;
  43     char * buf = malloc( bufsz );
  44     FILE * fh;
  45
  46     // Also fgets
  47     TESTCASE( ( fh = tmpfile() ) != NULL );
  48     TESTCASE( fputs(hellostr, fh) == 0 );
  49     rewind(fh);
  50     TESTCASE( fgetc( fh ) == 'H' );
  51     TESTCASE( ungetc( 'H', fh ) == 'H' );
  52     TESTCASE( fgets( buf, bufsz, fh ) != NULL );
  53     TESTCASE( strcmp( buf, hellostr ) == 0 );
  54
  55     // fread
  56     rewind(fh);
  57     TESTCASE( fgetc( fh ) == 'H' );
  58     TESTCASE( ungetc( 'H', fh ) == 'H' );
  59     TESTCASE( fread( buf, bufsz - 1, 1, fh ) == 1 );
  60     TESTCASE( strncmp( buf, hellostr, bufsz - 1 ) == 0 );
  61
  62
  63
  64     return TEST_RESULTS;
  65 }
  66
  67 #endif