summaryrefslogtreecommitdiff
path: root/stdio-common/tst-ungetc.c
blob: 1344b2b591e3d6b12418d4c27bf930f783c7fcf6 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
/* Test for ungetc bugs.  */

#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>

#undef assert
#define assert(x) \
  if (!(x)) \
    { \
      fputs ("test failed: " #x "\n", stderr); \
      retval = 1; \
      goto the_end; \
    }

int
main (int argc, char *argv[])
{
  char name[] = "/tmp/tst-ungetc.XXXXXX";
  FILE *fp = NULL;
  int retval = 0;
  int c;
  char buffer[64];

  int fd = mkstemp (name);
  if (fd == -1)
    {
      printf ("mkstemp failed: %m\n");
      return 1;
    }
  close (fd);
  fp = fopen (name, "w");
  assert (fp != NULL)
  fputs ("bla", fp);
  fclose (fp);
  fp = NULL;

  fp = fopen (name, "r");
  assert (fp != NULL);
  assert (ungetc ('z', fp) == 'z');
  assert (getc (fp) == 'z');
  assert (getc (fp) == 'b');
  assert (getc (fp) == 'l');
  assert (ungetc ('m', fp) == 'm');
  assert (getc (fp) == 'm');
  assert ((c = getc (fp)) == 'a');
  assert (getc (fp) == EOF);
  assert (ungetc (c, fp) == c);
  assert (feof (fp) == 0);
  assert (getc (fp) == c);
  assert (getc (fp) == EOF);
  fclose (fp);
  fp = NULL;

  fp = fopen (name, "r");
  assert (fp != NULL);
  assert (getc (fp) == 'b');
  assert (getc (fp) == 'l');
  assert (ungetc ('b', fp) == 'b');
  assert (fread (buffer, 1, 64, fp) == 2);
  assert (buffer[0] == 'b');
  assert (buffer[1] == 'a');

the_end:
  if (fp != NULL)
    fclose (fp);
  unlink (name);

  return retval;
}