tests/test-fseeko.c

   1 /* Test of fseeko() function.
   2    Copyright (C) 2007, 2008, 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 /* None of the files accessed by this test are large, so disable the
  22    fseek link warning if we are not using the gnulib fseek module.  */
  23 #if !GNULIB_FSEEK
  24 # undef GL_LINK_WARNING
  25 # define GL_LINK_WARNING(ignored) ((void) 0)
  26 #endif
  27
  28 #include <stdio.h>
  29
  30 #include "signature.h"
  31 SIGNATURE_CHECK (fseeko, int, (FILE *, off_t, int));
  32
  33 #include <stdlib.h>
  34
  35 #define ASSERT(expr) \
  36   do                                                                         \
  37     {                                                                        \
  38       if (!(expr))                                                           \
  39         {                                                                    \
  40           fprintf (stderr, "%s:%d: assertion failed\n", __FILE__, __LINE__); \
  41           fflush (stderr);                                                   \
  42           abort ();                                                          \
  43         }                                                                    \
  44     }                                                                        \
  45   while (0)
  46
  47 #ifndef FUNC_UNGETC_BROKEN
  48 # define FUNC_UNGETC_BROKEN 0
  49 #endif
  50
  51 int
  52 main (int argc, char **argv _GL_UNUSED)
  53 {
  54   /* Assume stdin is non-empty, seekable, and starts with '#!/bin/sh'
  55      iff argc > 1.  */
  56   int expected = argc > 1 ? 0 : -1;
  57   /* Exit with success only if fseek/fseeko agree.  */
  58   int r1 = fseeko (stdin, 0, SEEK_CUR);
  59   int r2 = fseek (stdin, 0, SEEK_CUR);
  60   ASSERT (r1 == r2 && r1 == expected);
  61   if (argc > 1)
  62     {
  63       /* Test that fseek discards previously read ungetc data.  */
  64       int ch = fgetc (stdin);
  65       ASSERT (ch == '#');
  66       ASSERT (ungetc (ch, stdin) == ch);
  67       ASSERT (fseeko (stdin, 2, SEEK_SET) == 0);
  68       ch = fgetc (stdin);
  69       ASSERT (ch == '/');
  70       if (2 < argc)
  71         {
  72           if (FUNC_UNGETC_BROKEN)
  73             {
  74               fputs ("Skipping test: ungetc cannot handle arbitrary bytes\n",
  75                      stderr);
  76               return 77;
  77             }
  78           /* Test that fseek discards random ungetc data.  */
  79           ASSERT (ungetc (ch ^ 0xff, stdin) == (ch ^ 0xff));
  80         }
  81       ASSERT (fseeko (stdin, 0, SEEK_END) == 0);
  82       ASSERT (fgetc (stdin) == EOF);
  83       /* Test that fseek resets end-of-file marker.  */
  84       ASSERT (feof (stdin));
  85       ASSERT (fseeko (stdin, 0, SEEK_END) == 0);
  86       ASSERT (!feof (stdin));
  87     }
  88   return 0;
  89 }