nsz Git - libc-test/blob - src/stdio/ungetc.c

   1 #include <stdio.h>
   2 #include <string.h>
   3 #include <errno.h>
   4 #include <limits.h>
   5 #include <unistd.h>
   6 #include "test.h"
   7
   8 #define TEST(r, f, x, m) ( \
   9         errno = 0, ((r) = (f)) == (x) || \
  10         (error("%s failed (" m ")\n", #f, r, x, strerror(errno)), 0) )
  11
  12 #define TEST_S(s, x, m) ( \
  13         !strcmp((s),(x)) || \
  14         (error("[%s] != [%s] (%s)\n", s, x, m), 0) )
  15
  16 void test_ungetc(void) {
  17         int i;
  18         char a[100];
  19         FILE *f;
  20
  21         TEST(i, !(f = tmpfile()), 0, "failed to create temp file %d!=%d (%s)");
  22
  23         if (!f) return;
  24
  25         TEST(i, fprintf(f, "hello, world\n"), 13, "%d != %d (%m)");
  26         TEST(i, fseek(f, 0, SEEK_SET), 0, "%d != %d (%m)");
  27
  28         TEST(i, feof(f), 0, "%d != %d");
  29         TEST(i, fgetc(f), 'h', "'%c' != '%c'");
  30         TEST(i, ftell(f), 1, "%d != %d");
  31         TEST(i, ungetc('x', f), 'x', "%d != %d");
  32         TEST(i, ftell(f), 0, "%d != %d");
  33         TEST(i, fscanf(f, "%[h]", a), 0, "got %d fields, expected %d");
  34         TEST(i, ftell(f), 0, "%d != %d");
  35         TEST(i, fgetc(f), 'x', "'%c' != '%c'");
  36         TEST(i, ftell(f), 1, "%d != %d");
  37
  38         TEST(i, fseek(f, 0, SEEK_SET), 0, "%d != %d");
  39         TEST(i, ungetc('x', f), 'x', "%d != %d");
  40         TEST(i, fread(a, 1, sizeof a, f), 14, "read %d, expected %d");
  41         a[14] = 0;
  42         TEST_S(a, "xhello, world\n", "mismatch reading ungot character");
  43
  44         TEST(i, fseek(f, 0, SEEK_SET), 0, "%d != %d");
  45         TEST(i, fscanf(f, "%[x]", a), 0, "got %d fields, expected %d");
  46         TEST(i, ungetc('x', f), 'x', "unget failed after fscanf: %d != %d");
  47         TEST(i, fgetc(f), 'x', "'%c' != '%c'");
  48         TEST(i, fgetc(f), 'h', "'%c' != '%c'");
  49
  50         fclose(f);
  51 }