gas/input-file.c

   1 /* input_file.c - Deal with Input Files -
   2    Copyright 1987, 1990, 1991, 1992, 1993, 1994, 1995, 1999, 2000, 2001,
   3    2002, 2003, 2005, 2006
   4    Free Software Foundation, Inc.
   5
   6    This file is part of GAS, the GNU Assembler.
   7
   8    GAS is free software; you can redistribute it and/or modify
   9    it under the terms of the GNU General Public License as published by
  10    the Free Software Foundation; either version 2, or (at your option)
  11    any later version.
  12
  13    GAS is distributed in the hope that it will be useful,
  14    but WITHOUT ANY WARRANTY; without even the implied warranty of
  15    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  16    GNU General Public License for more details.
  17
  18    You should have received a copy of the GNU General Public License
  19    along with GAS; see the file COPYING.  If not, write to the Free
  20    Software Foundation, 51 Franklin Street - Fifth Floor, Boston, MA
  21    02110-1301, USA.  */
  22
  23 /* Confines all details of reading source bytes to this module.
  24    All O/S specific crocks should live here.
  25    What we lose in "efficiency" we gain in modularity.
  26    Note we don't need to #include the "as.h" file. No common coupling!  */
  27
  28 #include "as.h"
  29 #include "input-file.h"
  30 #include "safe-ctype.h"
  31
  32 static int input_file_get (char *, int);
  33
  34 /* This variable is non-zero if the file currently being read should be
  35    preprocessed by app.  It is zero if the file can be read straight in.  */
  36 int preprocess = 0;
  37
  38 /* This code opens a file, then delivers BUFFER_SIZE character
  39    chunks of the file on demand.
  40    BUFFER_SIZE is supposed to be a number chosen for speed.
  41    The caller only asks once what BUFFER_SIZE is, and asks before
  42    the nature of the input files (if any) is known.  */
  43
  44 #define BUFFER_SIZE (32 * 1024)
  45
  46 /* We use static data: the data area is not sharable.  */
  47
  48 static FILE *f_in;
  49 static char *file_name;
  50
  51 /* Struct for saving the state of this module for file includes.  */
  52 struct saved_file
  53   {
  54     FILE * f_in;
  55     char * file_name;
  56     int    preprocess;
  57     char * app_save;
  58   };
  59 \f
  60 /* These hooks accommodate most operating systems.  */
  61
  62 void
  63 input_file_begin (void)
  64 {
  65   f_in = (FILE *) 0;
  66 }
  67
  68 void
  69 input_file_end (void)
  70 {
  71 }
  72
  73 /* Return BUFFER_SIZE.  */
  74 unsigned int
  75 input_file_buffer_size (void)
  76 {
  77   return (BUFFER_SIZE);
  78 }
  79
  80 /* Push the state of our input, returning a pointer to saved info that
  81    can be restored with input_file_pop ().  */
  82
  83 char *
  84 input_file_push (void)
  85 {
  86   register struct saved_file *saved;
  87
  88   saved = (struct saved_file *) xmalloc (sizeof *saved);
  89
  90   saved->f_in = f_in;
  91   saved->file_name = file_name;
  92   saved->preprocess = preprocess;
  93   if (preprocess)
  94     saved->app_save = app_push ();
  95
  96   /* Initialize for new file.  */
  97   input_file_begin ();
  98
  99   return (char *) saved;
 100 }
 101
 102 void
 103 input_file_pop (char *arg)
 104 {
 105   register struct saved_file *saved = (struct saved_file *) arg;
 106
 107   input_file_end ();            /* Close out old file.  */
 108
 109   f_in = saved->f_in;
 110   file_name = saved->file_name;
 111   preprocess = saved->preprocess;
 112   if (preprocess)
 113     app_pop (saved->app_save);
 114
 115   free (arg);
 116 }
 117 \f
 118 void
 119 input_file_open (char *filename, /* "" means use stdin. Must not be 0.  */
 120                  int pre)
 121 {
 122   int c;
 123   char buf[80];
 124
 125   preprocess = pre;
 126
 127   assert (filename != 0);       /* Filename may not be NULL.  */
 128   if (filename[0])
 129     {
 130       f_in = fopen (filename, FOPEN_RT);
 131       file_name = filename;
 132     }
 133   else
 134     {
 135       /* Use stdin for the input file.  */
 136       f_in = stdin;
 137       /* For error messages.  */
 138       file_name = _("{standard input}");
 139     }
 140
 141   if (f_in == NULL)
 142     {
 143       bfd_set_error (bfd_error_system_call);
 144       as_perror (_("Can't open %s for reading"), file_name);
 145       return;
 146     }
 147
 148   c = getc (f_in);
 149
 150   if (ferror (f_in))
 151     {
 152       bfd_set_error (bfd_error_system_call);
 153       as_perror (_("Can't open %s for reading"), file_name);
 154
 155       fclose (f_in);
 156       f_in = NULL;
 157       return;
 158     }
 159
 160   if (c == '#')
 161     {
 162       /* Begins with comment, may not want to preprocess.  */
 163       c = getc (f_in);
 164       if (c == 'N')
 165         {
 166           fgets (buf, 80, f_in);
 167           if (!strncmp (buf, "O_APP", 5) && ISSPACE (buf[5]))
 168             preprocess = 0;
 169           if (!strchr (buf, '\n'))
 170             ungetc ('#', f_in); /* It was longer.  */
 171           else
 172             ungetc ('\n', f_in);
 173         }
 174       else if (c == 'A')
 175         {
 176           fgets (buf, 80, f_in);
 177           if (!strncmp (buf, "PP", 2) && ISSPACE (buf[2]))
 178             preprocess = 1;
 179           if (!strchr (buf, '\n'))
 180             ungetc ('#', f_in);
 181           else
 182             ungetc ('\n', f_in);
 183         }
 184       else if (c == '\n')
 185         ungetc ('\n', f_in);
 186       else
 187         ungetc ('#', f_in);
 188     }
 189   else
 190     ungetc (c, f_in);
 191 }
 192
 193 /* Close input file.  */
 194
 195 void
 196 input_file_close (void)
 197 {
 198   /* Don't close a null file pointer.  */
 199   if (f_in != NULL)
 200     fclose (f_in);
 201
 202   f_in = 0;
 203 }
 204
 205 /* This function is passed to do_scrub_chars.  */
 206
 207 static int
 208 input_file_get (char *buf, int buflen)
 209 {
 210   int size;
 211
 212   size = fread (buf, sizeof (char), buflen, f_in);
 213   if (size < 0)
 214     {
 215       bfd_set_error (bfd_error_system_call);
 216       as_perror (_("Can't read from %s"), file_name);
 217       size = 0;
 218     }
 219   return size;
 220 }
 221
 222 /* Read a buffer from the input file.  */
 223
 224 char *
 225 input_file_give_next_buffer (char *where /* Where to place 1st character of new buffer.  */)
 226 {
 227   char *return_value;           /* -> Last char of what we read, + 1.  */
 228   register int size;
 229
 230   if (f_in == (FILE *) 0)
 231     return 0;
 232   /* fflush (stdin); could be done here if you want to synchronise
 233      stdin and stdout, for the case where our input file is stdin.
 234      Since the assembler shouldn't do any output to stdout, we
 235      don't bother to synch output and input.  */
 236   if (preprocess)
 237     size = do_scrub_chars (input_file_get, where, BUFFER_SIZE);
 238   else
 239     size = fread (where, sizeof (char), BUFFER_SIZE, f_in);
 240   if (size < 0)
 241     {
 242       bfd_set_error (bfd_error_system_call);
 243       as_perror (_("Can't read from %s"), file_name);
 244       size = 0;
 245     }
 246   if (size)
 247     return_value = where + size;
 248   else
 249     {
 250       if (fclose (f_in))
 251         {
 252           bfd_set_error (bfd_error_system_call);
 253           as_perror (_("Can't close %s"), file_name);
 254         }
 255       f_in = (FILE *) 0;
 256       return_value = 0;
 257     }
 258
 259   return return_value;
 260 }