pintos-os.org Git - pspp/blob - tests/test-fpurge.c

   1 /* Test of fpurge() function.
   2    Copyright (C) 2007-2009 Free Software Foundation, Inc.
   3
   4    This program is free software: you can redistribute it and/or modify
   5    it under the terms of the GNU General Public License as published by
   6    the Free Software Foundation; either version 3 of the License, or
   7    (at your option) any later version.
   8
   9    This program is distributed in the hope that it will be useful,
  10    but WITHOUT ANY WARRANTY; without even the implied warranty of
  11    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12    GNU General Public License for more details.
  13
  14    You should have received a copy of the GNU General Public License
  15    along with this program.  If not, see <http://www.gnu.org/licenses/>.  */
  16
  17 /* Written by Bruno Haible <bruno@clisp.org>, 2007.  */
  18
  19 #include <config.h>
  20
  21 #include <stdio.h>
  22
  23 #include <stdlib.h>
  24 #include <string.h>
  25
  26 #define ASSERT(expr) \
  27   do                                                                         \
  28     {                                                                        \
  29       if (!(expr))                                                           \
  30         {                                                                    \
  31           fprintf (stderr, "%s:%d: assertion failed\n", __FILE__, __LINE__); \
  32           fflush (stderr);                                                   \
  33           abort ();                                                          \
  34         }                                                                    \
  35     }                                                                        \
  36   while (0)
  37
  38 #define TESTFILE "t-fpurge.tmp"
  39
  40 int
  41 main ()
  42 {
  43   FILE *fp;
  44
  45   /* Create a file with some contents.  */
  46   fp = fopen (TESTFILE, "w");
  47   if (fp == NULL)
  48     goto skip;
  49   if (fwrite ("foobarsh", 1, 8, fp) < 8)
  50     goto skip;
  51   if (fclose (fp))
  52     goto skip;
  53
  54   /* The file's contents is now "foobarsh".  */
  55
  56   /* Open it in read-write mode.  */
  57   fp = fopen (TESTFILE, "r+");
  58   if (fp == NULL)
  59     goto skip;
  60   if (fseek (fp, 3, SEEK_CUR))
  61     goto skip;
  62   if (fwrite ("g", 1, 1, fp) < 1)
  63     goto skip;
  64   if (fflush (fp))
  65     goto skip;
  66   if (fwrite ("bz", 1, 2, fp) < 2)
  67     goto skip;
  68   /* Discard pending write.  */
  69   ASSERT (fpurge (fp) == 0);
  70   ASSERT (fclose (fp) == 0);
  71
  72   /* Open it in read-only mode.  */
  73   fp = fopen (TESTFILE, "r");
  74   if (fp == NULL)
  75     goto skip;
  76   {
  77     char buf[8];
  78     if (fread (buf, 1, 7, fp) < 7)
  79       goto skip;
  80     ASSERT (memcmp (buf, "foogars", 7) == 0);
  81   }
  82   /* Discard the buffered 'h', leaving position at EOF.  */
  83   ASSERT (fpurge (fp) == 0);
  84   ASSERT (getc (fp) == EOF);
  85   ASSERT (fclose (fp) == 0);
  86
  87   /* The file's contents is now "foogarsh".  */
  88
  89   /* Ensure that purging a read does not corrupt subsequent writes.  */
  90   fp = fopen (TESTFILE, "r+");
  91   if (fp == NULL)
  92     goto skip;
  93   if (fseek (fp, -1, SEEK_END))
  94     goto skip;
  95   ASSERT (getc (fp) == 'h');
  96   ASSERT (getc (fp) == EOF);
  97   ASSERT (fpurge (fp) == 0);
  98   ASSERT (putc ('!', fp) == '!');
  99   ASSERT (fclose (fp) == 0);
 100   fp = fopen (TESTFILE, "r");
 101   if (fp == NULL)
 102     goto skip;
 103   {
 104     char buf[10];
 105     ASSERT (fread (buf, 1, 10, fp) == 9);
 106     ASSERT (memcmp (buf, "foogarsh!", 9) == 0);
 107   }
 108   ASSERT (fclose (fp) == 0);
 109
 110   /* The file's contents is now "foogarsh!".  */
 111
 112   remove (TESTFILE);
 113   return 0;
 114
 115  skip:
 116   fprintf (stderr, "Skipping test: prerequisite file operations failed.\n");
 117   remove (TESTFILE);
 118   return 77;
 119 }