]> pd.if.org Git - pdclib/blob - stdio/ungetc.c
Cosmetic comment fixes.
[pdclib] / 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