* global.c (global_alloc): Make it static.
[official-gcc.git] / libgfortran / io / unix.c
blob237f09e99308aa47585cbd7618086a32358a979f
1 /* Copyright (C) 2002, 2003, 2004, 2005
2 Free Software Foundation, Inc.
3 Contributed by Andy Vaught
5 This file is part of the GNU Fortran 95 runtime library (libgfortran).
7 Libgfortran is free software; you can redistribute it and/or modify
8 it under the terms of the GNU General Public License as published by
9 the Free Software Foundation; either version 2, or (at your option)
10 any later version.
12 In addition to the permissions in the GNU General Public License, the
13 Free Software Foundation gives you unlimited permission to link the
14 compiled version of this file into combinations with other programs,
15 and to distribute those combinations without any restriction coming
16 from the use of this file. (The General Public License restrictions
17 do apply in other respects; for example, they cover modification of
18 the file, and distribution when not linked into a combine
19 executable.)
21 Libgfortran is distributed in the hope that it will be useful,
22 but WITHOUT ANY WARRANTY; without even the implied warranty of
23 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
24 GNU General Public License for more details.
26 You should have received a copy of the GNU General Public License
27 along with Libgfortran; see the file COPYING. If not, write to
28 the Free Software Foundation, 51 Franklin Street, Fifth Floor,
29 Boston, MA 02110-1301, USA. */
31 /* Unix stream I/O module */
33 #include "config.h"
34 #include <stdlib.h>
35 #include <limits.h>
37 #include <unistd.h>
38 #include <stdio.h>
39 #include <sys/stat.h>
40 #include <fcntl.h>
41 #include <assert.h>
43 #include <string.h>
44 #include <errno.h>
46 #include "libgfortran.h"
47 #include "io.h"
48 #include "unix.h"
50 #ifndef SSIZE_MAX
51 #define SSIZE_MAX SHRT_MAX
52 #endif
54 #ifndef PATH_MAX
55 #define PATH_MAX 1024
56 #endif
58 #ifndef PROT_READ
59 #define PROT_READ 1
60 #endif
62 #ifndef PROT_WRITE
63 #define PROT_WRITE 2
64 #endif
66 /* These flags aren't defined on all targets (mingw32), so provide them
67 here. */
68 #ifndef S_IRGRP
69 #define S_IRGRP 0
70 #endif
72 #ifndef S_IWGRP
73 #define S_IWGRP 0
74 #endif
76 #ifndef S_IROTH
77 #define S_IROTH 0
78 #endif
80 #ifndef S_IWOTH
81 #define S_IWOTH 0
82 #endif
84 /* This implementation of stream I/O is based on the paper:
86 * "Exploiting the advantages of mapped files for stream I/O",
87 * O. Krieger, M. Stumm and R. Umrau, "Proceedings of the 1992 Winter
88 * USENIX conference", p. 27-42.
90 * It differs in a number of ways from the version described in the
91 * paper. First of all, threads are not an issue during I/O and we
92 * also don't have to worry about having multiple regions, since
93 * fortran's I/O model only allows you to be one place at a time.
95 * On the other hand, we have to be able to writing at the end of a
96 * stream, read from the start of a stream or read and write blocks of
97 * bytes from an arbitrary position. After opening a file, a pointer
98 * to a stream structure is returned, which is used to handle file
99 * accesses until the file is closed.
101 * salloc_at_r(stream, len, where)-- Given a stream pointer, return a
102 * pointer to a block of memory that mirror the file at position
103 * 'where' that is 'len' bytes long. The len integer is updated to
104 * reflect how many bytes were actually read. The only reason for a
105 * short read is end of file. The file pointer is updated. The
106 * pointer is valid until the next call to salloc_*.
108 * salloc_at_w(stream, len, where)-- Given the stream pointer, returns
109 * a pointer to a block of memory that is updated to reflect the state
110 * of the file. The length of the buffer is always equal to that
111 * requested. The buffer must be completely set by the caller. When
112 * data has been written, the sfree() function must be called to
113 * indicate that the caller is done writing data to the buffer. This
114 * may or may not cause a physical write.
116 * Short forms of these are salloc_r() and salloc_w() which drop the
117 * 'where' parameter and use the current file pointer. */
120 /*move_pos_offset()-- Move the record pointer right or left
121 *relative to current position */
124 move_pos_offset (stream* st, int pos_off)
126 unix_stream * str = (unix_stream*)st;
127 if (pos_off < 0)
129 str->logical_offset += pos_off;
131 if (str->dirty_offset + str->ndirty > str->logical_offset)
133 if (str->ndirty + pos_off > 0)
134 str->ndirty += pos_off;
135 else
137 str->dirty_offset += pos_off + pos_off;
138 str->ndirty = 0;
142 return pos_off;
144 return 0;
148 /* fix_fd()-- Given a file descriptor, make sure it is not one of the
149 * standard descriptors, returning a non-standard descriptor. If the
150 * user specifies that system errors should go to standard output,
151 * then closes standard output, we don't want the system errors to a
152 * file that has been given file descriptor 1 or 0. We want to send
153 * the error to the invalid descriptor. */
155 static int
156 fix_fd (int fd)
158 int input, output, error;
160 input = output = error = 0;
162 /* Unix allocates the lowest descriptors first, so a loop is not
163 required, but this order is. */
165 if (fd == STDIN_FILENO)
167 fd = dup (fd);
168 input = 1;
170 if (fd == STDOUT_FILENO)
172 fd = dup (fd);
173 output = 1;
175 if (fd == STDERR_FILENO)
177 fd = dup (fd);
178 error = 1;
181 if (input)
182 close (STDIN_FILENO);
183 if (output)
184 close (STDOUT_FILENO);
185 if (error)
186 close (STDERR_FILENO);
188 return fd;
192 is_preconnected (stream * s)
194 int fd;
196 fd = ((unix_stream *) s)->fd;
197 if (fd == STDIN_FILENO || fd == STDOUT_FILENO || fd == STDERR_FILENO)
198 return 1;
199 else
200 return 0;
203 /* If the stream corresponds to a preconnected unit, we flush the
204 corresponding C stream. This is bugware for mixed C-Fortran codes
205 where the C code doesn't flush I/O before returning. */
206 void
207 flush_if_preconnected (stream * s)
209 int fd;
211 fd = ((unix_stream *) s)->fd;
212 if (fd == STDIN_FILENO)
213 fflush (stdin);
214 else if (fd == STDOUT_FILENO)
215 fflush (stdout);
216 else if (fd == STDERR_FILENO)
217 fflush (stderr);
221 /* Reset a stream after reading/writing. Assumes that the buffers have
222 been flushed. */
224 inline static void
225 reset_stream (unix_stream * s, size_t bytes_rw)
227 s->physical_offset += bytes_rw;
228 s->logical_offset = s->physical_offset;
229 if (s->file_length != -1 && s->physical_offset > s->file_length)
230 s->file_length = s->physical_offset;
234 /* Read bytes into a buffer, allowing for short reads. If the nbytes
235 * argument is less on return than on entry, it is because we've hit
236 * the end of file. */
238 static int
239 do_read (unix_stream * s, void * buf, size_t * nbytes)
241 ssize_t trans;
242 size_t bytes_left;
243 char *buf_st;
244 int status;
246 status = 0;
247 bytes_left = *nbytes;
248 buf_st = (char *) buf;
250 /* We must read in a loop since some systems don't restart system
251 calls in case of a signal. */
252 while (bytes_left > 0)
254 /* Requests between SSIZE_MAX and SIZE_MAX are undefined by SUSv3,
255 so we must read in chunks smaller than SSIZE_MAX. */
256 trans = (bytes_left < SSIZE_MAX) ? bytes_left : SSIZE_MAX;
257 trans = read (s->fd, buf_st, trans);
258 if (trans < 0)
260 if (errno == EINTR)
261 continue;
262 else
264 status = errno;
265 break;
268 else if (trans == 0) /* We hit EOF. */
269 break;
270 buf_st += trans;
271 bytes_left -= trans;
274 *nbytes -= bytes_left;
275 return status;
279 /* Write a buffer to a stream, allowing for short writes. */
281 static int
282 do_write (unix_stream * s, const void * buf, size_t * nbytes)
284 ssize_t trans;
285 size_t bytes_left;
286 char *buf_st;
287 int status;
289 status = 0;
290 bytes_left = *nbytes;
291 buf_st = (char *) buf;
293 /* We must write in a loop since some systems don't restart system
294 calls in case of a signal. */
295 while (bytes_left > 0)
297 /* Requests between SSIZE_MAX and SIZE_MAX are undefined by SUSv3,
298 so we must write in chunks smaller than SSIZE_MAX. */
299 trans = (bytes_left < SSIZE_MAX) ? bytes_left : SSIZE_MAX;
300 trans = write (s->fd, buf_st, trans);
301 if (trans < 0)
303 if (errno == EINTR)
304 continue;
305 else
307 status = errno;
308 break;
311 buf_st += trans;
312 bytes_left -= trans;
315 *nbytes -= bytes_left;
316 return status;
320 /* get_oserror()-- Get the most recent operating system error. For
321 * unix, this is errno. */
323 const char *
324 get_oserror (void)
326 return strerror (errno);
330 /* sys_exit()-- Terminate the program with an exit code */
332 void
333 sys_exit (int code)
335 exit (code);
339 /*********************************************************************
340 File descriptor stream functions
341 *********************************************************************/
344 /* fd_flush()-- Write bytes that need to be written */
346 static try
347 fd_flush (unix_stream * s)
349 size_t writelen;
351 if (s->ndirty == 0)
352 return SUCCESS;;
354 if (s->physical_offset != s->dirty_offset &&
355 lseek (s->fd, s->dirty_offset, SEEK_SET) < 0)
356 return FAILURE;
358 writelen = s->ndirty;
359 if (do_write (s, s->buffer + (s->dirty_offset - s->buffer_offset),
360 &writelen) != 0)
361 return FAILURE;
363 s->physical_offset = s->dirty_offset + writelen;
365 /* don't increment file_length if the file is non-seekable */
366 if (s->file_length != -1 && s->physical_offset > s->file_length)
367 s->file_length = s->physical_offset;
369 s->ndirty -= writelen;
370 if (s->ndirty != 0)
371 return FAILURE;
373 return SUCCESS;
377 /* fd_alloc()-- Arrange a buffer such that the salloc() request can be
378 * satisfied. This subroutine gets the buffer ready for whatever is
379 * to come next. */
381 static void
382 fd_alloc (unix_stream * s, gfc_offset where,
383 int *len __attribute__ ((unused)))
385 char *new_buffer;
386 int n, read_len;
388 if (*len <= BUFFER_SIZE)
390 new_buffer = s->small_buffer;
391 read_len = BUFFER_SIZE;
393 else
395 new_buffer = get_mem (*len);
396 read_len = *len;
399 /* Salvage bytes currently within the buffer. This is important for
400 * devices that cannot seek. */
402 if (s->buffer != NULL && s->buffer_offset <= where &&
403 where <= s->buffer_offset + s->active)
406 n = s->active - (where - s->buffer_offset);
407 memmove (new_buffer, s->buffer + (where - s->buffer_offset), n);
409 s->active = n;
411 else
412 { /* new buffer starts off empty */
413 s->active = 0;
416 s->buffer_offset = where;
418 /* free the old buffer if necessary */
420 if (s->buffer != NULL && s->buffer != s->small_buffer)
421 free_mem (s->buffer);
423 s->buffer = new_buffer;
424 s->len = read_len;
428 /* fd_alloc_r_at()-- Allocate a stream buffer for reading. Either
429 * we've already buffered the data or we need to load it. Returns
430 * NULL on I/O error. */
432 static char *
433 fd_alloc_r_at (unix_stream * s, int *len, gfc_offset where)
435 gfc_offset m;
437 if (where == -1)
438 where = s->logical_offset;
440 if (s->buffer != NULL && s->buffer_offset <= where &&
441 where + *len <= s->buffer_offset + s->active)
444 /* Return a position within the current buffer */
446 s->logical_offset = where + *len;
447 return s->buffer + where - s->buffer_offset;
450 fd_alloc (s, where, len);
452 m = where + s->active;
454 if (s->physical_offset != m && lseek (s->fd, m, SEEK_SET) < 0)
455 return NULL;
457 /* do_read() hangs on read from terminals for *BSD-systems. Only
458 use read() in that case. */
460 if (s->special_file)
462 ssize_t n;
464 n = read (s->fd, s->buffer + s->active, s->len - s->active);
465 if (n < 0)
466 return NULL;
468 s->physical_offset = where + n;
469 s->active += n;
471 else
473 size_t n;
475 n = s->len - s->active;
476 if (do_read (s, s->buffer + s->active, &n) != 0)
477 return NULL;
479 s->physical_offset = where + n;
480 s->active += n;
483 if (s->active < *len)
484 *len = s->active; /* Bytes actually available */
486 s->logical_offset = where + *len;
488 return s->buffer;
492 /* fd_alloc_w_at()-- Allocate a stream buffer for writing. Either
493 * we've already buffered the data or we need to load it. */
495 static char *
496 fd_alloc_w_at (unix_stream * s, int *len, gfc_offset where)
498 gfc_offset n;
500 if (where == -1)
501 where = s->logical_offset;
503 if (s->buffer == NULL || s->buffer_offset > where ||
504 where + *len > s->buffer_offset + s->len)
507 if (fd_flush (s) == FAILURE)
508 return NULL;
509 fd_alloc (s, where, len);
512 /* Return a position within the current buffer */
513 if (s->ndirty == 0
514 || where > s->dirty_offset + s->ndirty
515 || s->dirty_offset > where + *len)
516 { /* Discontiguous blocks, start with a clean buffer. */
517 /* Flush the buffer. */
518 if (s->ndirty != 0)
519 fd_flush (s);
520 s->dirty_offset = where;
521 s->ndirty = *len;
523 else
525 gfc_offset start; /* Merge with the existing data. */
526 if (where < s->dirty_offset)
527 start = where;
528 else
529 start = s->dirty_offset;
530 if (where + *len > s->dirty_offset + s->ndirty)
531 s->ndirty = where + *len - start;
532 else
533 s->ndirty = s->dirty_offset + s->ndirty - start;
534 s->dirty_offset = start;
537 s->logical_offset = where + *len;
539 if (where + *len > s->file_length)
540 s->file_length = where + *len;
542 n = s->logical_offset - s->buffer_offset;
543 if (n > s->active)
544 s->active = n;
546 return s->buffer + where - s->buffer_offset;
550 static try
551 fd_sfree (unix_stream * s)
553 if (s->ndirty != 0 &&
554 (s->buffer != s->small_buffer || options.all_unbuffered ||
555 s->unbuffered))
556 return fd_flush (s);
558 return SUCCESS;
562 static try
563 fd_seek (unix_stream * s, gfc_offset offset)
565 if (s->physical_offset == offset) /* Are we lucky and avoid syscall? */
567 s->logical_offset = offset;
568 return SUCCESS;
571 s->physical_offset = s->logical_offset = offset;
573 return (lseek (s->fd, offset, SEEK_SET) < 0) ? FAILURE : SUCCESS;
577 /* truncate_file()-- Given a unit, truncate the file at the current
578 * position. Sets the physical location to the new end of the file.
579 * Returns nonzero on error. */
581 static try
582 fd_truncate (unix_stream * s)
584 if (lseek (s->fd, s->logical_offset, SEEK_SET) == -1)
585 return FAILURE;
587 /* non-seekable files, like terminals and fifo's fail the lseek.
588 Using ftruncate on a seekable special file (like /dev/null)
589 is undefined, so we treat it as if the ftruncate failed.
591 #ifdef HAVE_FTRUNCATE
592 if (s->special_file || ftruncate (s->fd, s->logical_offset))
593 #else
594 #ifdef HAVE_CHSIZE
595 if (s->special_file || chsize (s->fd, s->logical_offset))
596 #endif
597 #endif
599 s->physical_offset = s->file_length = 0;
600 return FAILURE;
603 s->physical_offset = s->file_length = s->logical_offset;
604 s->active = 0;
605 return SUCCESS;
611 /* Stream read function. Avoids using a buffer for big reads. The
612 interface is like POSIX read(), but the nbytes argument is a
613 pointer; on return it contains the number of bytes written. The
614 function return value is the status indicator (0 for success). */
616 static int
617 fd_read (unix_stream * s, void * buf, size_t * nbytes)
619 void *p;
620 int tmp, status;
622 if (*nbytes < BUFFER_SIZE && !s->unbuffered)
624 tmp = *nbytes;
625 p = fd_alloc_r_at (s, &tmp, -1);
626 if (p)
628 *nbytes = tmp;
629 memcpy (buf, p, *nbytes);
630 return 0;
632 else
634 *nbytes = 0;
635 return errno;
639 /* If the request is bigger than BUFFER_SIZE we flush the buffers
640 and read directly. */
641 if (fd_flush (s) == FAILURE)
643 *nbytes = 0;
644 return errno;
647 if (is_seekable ((stream *) s) && fd_seek (s, s->logical_offset) == FAILURE)
649 *nbytes = 0;
650 return errno;
653 status = do_read (s, buf, nbytes);
654 reset_stream (s, *nbytes);
655 return status;
659 /* Stream write function. Avoids using a buffer for big writes. The
660 interface is like POSIX write(), but the nbytes argument is a
661 pointer; on return it contains the number of bytes written. The
662 function return value is the status indicator (0 for success). */
664 static int
665 fd_write (unix_stream * s, const void * buf, size_t * nbytes)
667 void *p;
668 int tmp, status;
670 if (*nbytes < BUFFER_SIZE && !s->unbuffered)
672 tmp = *nbytes;
673 p = fd_alloc_w_at (s, &tmp, -1);
674 if (p)
676 *nbytes = tmp;
677 memcpy (p, buf, *nbytes);
678 return 0;
680 else
682 *nbytes = 0;
683 return errno;
687 /* If the request is bigger than BUFFER_SIZE we flush the buffers
688 and write directly. */
689 if (fd_flush (s) == FAILURE)
691 *nbytes = 0;
692 return errno;
695 if (is_seekable ((stream *) s) && fd_seek (s, s->logical_offset) == FAILURE)
697 *nbytes = 0;
698 return errno;
701 status = do_write (s, buf, nbytes);
702 reset_stream (s, *nbytes);
703 return status;
707 static try
708 fd_close (unix_stream * s)
710 if (fd_flush (s) == FAILURE)
711 return FAILURE;
713 if (s->buffer != NULL && s->buffer != s->small_buffer)
714 free_mem (s->buffer);
716 if (s->fd != STDOUT_FILENO && s->fd != STDERR_FILENO)
718 if (close (s->fd) < 0)
719 return FAILURE;
722 free_mem (s);
724 return SUCCESS;
728 static void
729 fd_open (unix_stream * s)
731 if (isatty (s->fd))
732 s->unbuffered = 1;
734 s->st.alloc_r_at = (void *) fd_alloc_r_at;
735 s->st.alloc_w_at = (void *) fd_alloc_w_at;
736 s->st.sfree = (void *) fd_sfree;
737 s->st.close = (void *) fd_close;
738 s->st.seek = (void *) fd_seek;
739 s->st.truncate = (void *) fd_truncate;
740 s->st.read = (void *) fd_read;
741 s->st.write = (void *) fd_write;
743 s->buffer = NULL;
749 /*********************************************************************
750 memory stream functions - These are used for internal files
752 The idea here is that a single stream structure is created and all
753 requests must be satisfied from it. The location and size of the
754 buffer is the character variable supplied to the READ or WRITE
755 statement.
757 *********************************************************************/
760 static char *
761 mem_alloc_r_at (unix_stream * s, int *len, gfc_offset where)
763 gfc_offset n;
765 if (where == -1)
766 where = s->logical_offset;
768 if (where < s->buffer_offset || where > s->buffer_offset + s->active)
769 return NULL;
771 s->logical_offset = where + *len;
773 n = s->buffer_offset + s->active - where;
774 if (*len > n)
775 *len = n;
777 return s->buffer + (where - s->buffer_offset);
781 static char *
782 mem_alloc_w_at (unix_stream * s, int *len, gfc_offset where)
784 gfc_offset m;
786 assert (*len >= 0); /* Negative values not allowed. */
788 if (where == -1)
789 where = s->logical_offset;
791 m = where + *len;
793 if (where < s->buffer_offset)
794 return NULL;
796 if (m > s->file_length)
797 return NULL;
799 s->logical_offset = m;
801 return s->buffer + (where - s->buffer_offset);
805 /* Stream read function for internal units. This is not actually used
806 at the moment, as all internal IO is formatted and the formatted IO
807 routines use mem_alloc_r_at. */
809 static int
810 mem_read (unix_stream * s, void * buf, size_t * nbytes)
812 void *p;
813 int tmp;
815 tmp = *nbytes;
816 p = mem_alloc_r_at (s, &tmp, -1);
817 if (p)
819 *nbytes = tmp;
820 memcpy (buf, p, *nbytes);
821 return 0;
823 else
825 *nbytes = 0;
826 return errno;
831 /* Stream write function for internal units. This is not actually used
832 at the moment, as all internal IO is formatted and the formatted IO
833 routines use mem_alloc_w_at. */
835 static int
836 mem_write (unix_stream * s, const void * buf, size_t * nbytes)
838 void *p;
839 int tmp;
841 errno = 0;
843 tmp = *nbytes;
844 p = mem_alloc_w_at (s, &tmp, -1);
845 if (p)
847 *nbytes = tmp;
848 memcpy (p, buf, *nbytes);
849 return 0;
851 else
853 *nbytes = 0;
854 return errno;
859 static int
860 mem_seek (unix_stream * s, gfc_offset offset)
862 if (offset > s->file_length)
864 errno = ESPIPE;
865 return FAILURE;
868 s->logical_offset = offset;
869 return SUCCESS;
873 static int
874 mem_truncate (unix_stream * s __attribute__ ((unused)))
876 return SUCCESS;
880 static try
881 mem_close (unix_stream * s)
883 free_mem (s);
885 return SUCCESS;
889 static try
890 mem_sfree (unix_stream * s __attribute__ ((unused)))
892 return SUCCESS;
897 /*********************************************************************
898 Public functions -- A reimplementation of this module needs to
899 define functional equivalents of the following.
900 *********************************************************************/
902 /* empty_internal_buffer()-- Zero the buffer of Internal file */
904 void
905 empty_internal_buffer(stream *strm)
907 unix_stream * s = (unix_stream *) strm;
908 memset(s->buffer, ' ', s->file_length);
911 /* open_internal()-- Returns a stream structure from an internal file */
913 stream *
914 open_internal (char *base, int length)
916 unix_stream *s;
918 s = get_mem (sizeof (unix_stream));
919 memset (s, '\0', sizeof (unix_stream));
921 s->buffer = base;
922 s->buffer_offset = 0;
924 s->logical_offset = 0;
925 s->active = s->file_length = length;
927 s->st.alloc_r_at = (void *) mem_alloc_r_at;
928 s->st.alloc_w_at = (void *) mem_alloc_w_at;
929 s->st.sfree = (void *) mem_sfree;
930 s->st.close = (void *) mem_close;
931 s->st.seek = (void *) mem_seek;
932 s->st.truncate = (void *) mem_truncate;
933 s->st.read = (void *) mem_read;
934 s->st.write = (void *) mem_write;
936 return (stream *) s;
940 /* fd_to_stream()-- Given an open file descriptor, build a stream
941 * around it. */
943 static stream *
944 fd_to_stream (int fd, int prot)
946 struct stat statbuf;
947 unix_stream *s;
949 s = get_mem (sizeof (unix_stream));
950 memset (s, '\0', sizeof (unix_stream));
952 s->fd = fd;
953 s->buffer_offset = 0;
954 s->physical_offset = 0;
955 s->logical_offset = 0;
956 s->prot = prot;
958 /* Get the current length of the file. */
960 fstat (fd, &statbuf);
961 s->file_length = S_ISREG (statbuf.st_mode) ? statbuf.st_size : -1;
962 s->special_file = !S_ISREG (statbuf.st_mode);
964 fd_open (s);
966 return (stream *) s;
970 /* Given the Fortran unit number, convert it to a C file descriptor. */
973 unit_to_fd (int unit)
975 gfc_unit *us;
976 int fd;
978 us = find_unit (unit);
979 if (us == NULL)
980 return -1;
982 fd = ((unix_stream *) us->s)->fd;
983 unlock_unit (us);
984 return fd;
988 /* unpack_filename()-- Given a fortran string and a pointer to a
989 * buffer that is PATH_MAX characters, convert the fortran string to a
990 * C string in the buffer. Returns nonzero if this is not possible. */
993 unpack_filename (char *cstring, const char *fstring, int len)
995 len = fstrlen (fstring, len);
996 if (len >= PATH_MAX)
997 return 1;
999 memmove (cstring, fstring, len);
1000 cstring[len] = '\0';
1002 return 0;
1006 /* tempfile()-- Generate a temporary filename for a scratch file and
1007 * open it. mkstemp() opens the file for reading and writing, but the
1008 * library mode prevents anything that is not allowed. The descriptor
1009 * is returned, which is -1 on error. The template is pointed to by
1010 * opp->file, which is copied into the unit structure
1011 * and freed later. */
1013 static int
1014 tempfile (st_parameter_open *opp)
1016 const char *tempdir;
1017 char *template;
1018 int fd;
1020 tempdir = getenv ("GFORTRAN_TMPDIR");
1021 if (tempdir == NULL)
1022 tempdir = getenv ("TMP");
1023 if (tempdir == NULL)
1024 tempdir = getenv ("TEMP");
1025 if (tempdir == NULL)
1026 tempdir = DEFAULT_TEMPDIR;
1028 template = get_mem (strlen (tempdir) + 20);
1030 st_sprintf (template, "%s/gfortrantmpXXXXXX", tempdir);
1032 #ifdef HAVE_MKSTEMP
1034 fd = mkstemp (template);
1036 #else /* HAVE_MKSTEMP */
1038 if (mktemp (template))
1040 #if defined(HAVE_CRLF) && defined(O_BINARY)
1041 fd = open (template, O_RDWR | O_CREAT | O_EXCL | O_BINARY,
1042 S_IREAD | S_IWRITE);
1043 #else
1044 fd = open (template, O_RDWR | O_CREAT | O_EXCL, S_IREAD | S_IWRITE);
1045 #endif
1046 while (!(fd == -1 && errno == EEXIST) && mktemp (template));
1047 else
1048 fd = -1;
1050 #endif /* HAVE_MKSTEMP */
1052 if (fd < 0)
1053 free_mem (template);
1054 else
1056 opp->file = template;
1057 opp->file_len = strlen (template); /* Don't include trailing nul */
1060 return fd;
1064 /* regular_file()-- Open a regular file.
1065 * Change flags->action if it is ACTION_UNSPECIFIED on entry,
1066 * unless an error occurs.
1067 * Returns the descriptor, which is less than zero on error. */
1069 static int
1070 regular_file (st_parameter_open *opp, unit_flags *flags)
1072 char path[PATH_MAX + 1];
1073 int mode;
1074 int rwflag;
1075 int crflag;
1076 int fd;
1078 if (unpack_filename (path, opp->file, opp->file_len))
1080 errno = ENOENT; /* Fake an OS error */
1081 return -1;
1084 rwflag = 0;
1086 switch (flags->action)
1088 case ACTION_READ:
1089 rwflag = O_RDONLY;
1090 break;
1092 case ACTION_WRITE:
1093 rwflag = O_WRONLY;
1094 break;
1096 case ACTION_READWRITE:
1097 case ACTION_UNSPECIFIED:
1098 rwflag = O_RDWR;
1099 break;
1101 default:
1102 internal_error (&opp->common, "regular_file(): Bad action");
1105 switch (flags->status)
1107 case STATUS_NEW:
1108 crflag = O_CREAT | O_EXCL;
1109 break;
1111 case STATUS_OLD: /* open will fail if the file does not exist*/
1112 crflag = 0;
1113 break;
1115 case STATUS_UNKNOWN:
1116 case STATUS_SCRATCH:
1117 crflag = O_CREAT;
1118 break;
1120 case STATUS_REPLACE:
1121 crflag = O_CREAT | O_TRUNC;
1122 break;
1124 default:
1125 internal_error (&opp->common, "regular_file(): Bad status");
1128 /* rwflag |= O_LARGEFILE; */
1130 #if defined(HAVE_CRLF) && defined(O_BINARY)
1131 crflag |= O_BINARY;
1132 #endif
1134 mode = S_IRUSR | S_IWUSR | S_IRGRP | S_IWGRP | S_IROTH | S_IWOTH;
1135 fd = open (path, rwflag | crflag, mode);
1136 if (flags->action != ACTION_UNSPECIFIED)
1137 return fd;
1139 if (fd >= 0)
1141 flags->action = ACTION_READWRITE;
1142 return fd;
1144 if (errno != EACCES)
1145 return fd;
1147 /* retry for read-only access */
1148 rwflag = O_RDONLY;
1149 fd = open (path, rwflag | crflag, mode);
1150 if (fd >=0)
1152 flags->action = ACTION_READ;
1153 return fd; /* success */
1156 if (errno != EACCES)
1157 return fd; /* failure */
1159 /* retry for write-only access */
1160 rwflag = O_WRONLY;
1161 fd = open (path, rwflag | crflag, mode);
1162 if (fd >=0)
1164 flags->action = ACTION_WRITE;
1165 return fd; /* success */
1167 return fd; /* failure */
1171 /* open_external()-- Open an external file, unix specific version.
1172 * Change flags->action if it is ACTION_UNSPECIFIED on entry.
1173 * Returns NULL on operating system error. */
1175 stream *
1176 open_external (st_parameter_open *opp, unit_flags *flags)
1178 int fd, prot;
1180 if (flags->status == STATUS_SCRATCH)
1182 fd = tempfile (opp);
1183 if (flags->action == ACTION_UNSPECIFIED)
1184 flags->action = ACTION_READWRITE;
1186 #if HAVE_UNLINK_OPEN_FILE
1187 /* We can unlink scratch files now and it will go away when closed. */
1188 if (fd >= 0)
1189 unlink (opp->file);
1190 #endif
1192 else
1194 /* regular_file resets flags->action if it is ACTION_UNSPECIFIED and
1195 * if it succeeds */
1196 fd = regular_file (opp, flags);
1199 if (fd < 0)
1200 return NULL;
1201 fd = fix_fd (fd);
1203 switch (flags->action)
1205 case ACTION_READ:
1206 prot = PROT_READ;
1207 break;
1209 case ACTION_WRITE:
1210 prot = PROT_WRITE;
1211 break;
1213 case ACTION_READWRITE:
1214 prot = PROT_READ | PROT_WRITE;
1215 break;
1217 default:
1218 internal_error (&opp->common, "open_external(): Bad action");
1221 return fd_to_stream (fd, prot);
1225 /* input_stream()-- Return a stream pointer to the default input stream.
1226 * Called on initialization. */
1228 stream *
1229 input_stream (void)
1231 return fd_to_stream (STDIN_FILENO, PROT_READ);
1235 /* output_stream()-- Return a stream pointer to the default output stream.
1236 * Called on initialization. */
1238 stream *
1239 output_stream (void)
1241 return fd_to_stream (STDOUT_FILENO, PROT_WRITE);
1245 /* error_stream()-- Return a stream pointer to the default error stream.
1246 * Called on initialization. */
1248 stream *
1249 error_stream (void)
1251 return fd_to_stream (STDERR_FILENO, PROT_WRITE);
1254 /* init_error_stream()-- Return a pointer to the error stream. This
1255 * subroutine is called when the stream is needed, rather than at
1256 * initialization. We want to work even if memory has been seriously
1257 * corrupted. */
1259 stream *
1260 init_error_stream (unix_stream *error)
1262 memset (error, '\0', sizeof (*error));
1264 error->fd = options.use_stderr ? STDERR_FILENO : STDOUT_FILENO;
1266 error->st.alloc_w_at = (void *) fd_alloc_w_at;
1267 error->st.sfree = (void *) fd_sfree;
1269 error->unbuffered = 1;
1270 error->buffer = error->small_buffer;
1272 return (stream *) error;
1276 /* compare_file_filename()-- Given an open stream and a fortran string
1277 * that is a filename, figure out if the file is the same as the
1278 * filename. */
1281 compare_file_filename (gfc_unit *u, const char *name, int len)
1283 char path[PATH_MAX + 1];
1284 struct stat st1;
1285 #ifdef HAVE_WORKING_STAT
1286 struct stat st2;
1287 #endif
1289 if (unpack_filename (path, name, len))
1290 return 0; /* Can't be the same */
1292 /* If the filename doesn't exist, then there is no match with the
1293 * existing file. */
1295 if (stat (path, &st1) < 0)
1296 return 0;
1298 #ifdef HAVE_WORKING_STAT
1299 fstat (((unix_stream *) (u->s))->fd, &st2);
1300 return (st1.st_dev == st2.st_dev) && (st1.st_ino == st2.st_ino);
1301 #else
1302 if (len != u->file_len)
1303 return 0;
1304 return (memcmp(path, u->file, len) == 0);
1305 #endif
1309 #ifdef HAVE_WORKING_STAT
1310 # define FIND_FILE0_DECL struct stat *st
1311 # define FIND_FILE0_ARGS st
1312 #else
1313 # define FIND_FILE0_DECL const char *file, gfc_charlen_type file_len
1314 # define FIND_FILE0_ARGS file, file_len
1315 #endif
1317 /* find_file0()-- Recursive work function for find_file() */
1319 static gfc_unit *
1320 find_file0 (gfc_unit *u, FIND_FILE0_DECL)
1322 gfc_unit *v;
1324 if (u == NULL)
1325 return NULL;
1327 #ifdef HAVE_WORKING_STAT
1328 if (u->s != NULL
1329 && fstat (((unix_stream *) u->s)->fd, &st[1]) >= 0 &&
1330 st[0].st_dev == st[1].st_dev && st[0].st_ino == st[1].st_ino)
1331 return u;
1332 #else
1333 if (compare_string (u->file_len, u->file, file_len, file) == 0)
1334 return u;
1335 #endif
1337 v = find_file0 (u->left, FIND_FILE0_ARGS);
1338 if (v != NULL)
1339 return v;
1341 v = find_file0 (u->right, FIND_FILE0_ARGS);
1342 if (v != NULL)
1343 return v;
1345 return NULL;
1349 /* find_file()-- Take the current filename and see if there is a unit
1350 * that has the file already open. Returns a pointer to the unit if so. */
1352 gfc_unit *
1353 find_file (const char *file, gfc_charlen_type file_len)
1355 char path[PATH_MAX + 1];
1356 struct stat st[2];
1357 gfc_unit *u;
1359 if (unpack_filename (path, file, file_len))
1360 return NULL;
1362 if (stat (path, &st[0]) < 0)
1363 return NULL;
1365 __gthread_mutex_lock (&unit_lock);
1366 retry:
1367 u = find_file0 (unit_root, FIND_FILE0_ARGS);
1368 if (u != NULL)
1370 /* Fast path. */
1371 if (! __gthread_mutex_trylock (&u->lock))
1373 /* assert (u->closed == 0); */
1374 __gthread_mutex_unlock (&unit_lock);
1375 return u;
1378 inc_waiting_locked (u);
1380 __gthread_mutex_unlock (&unit_lock);
1381 if (u != NULL)
1383 __gthread_mutex_lock (&u->lock);
1384 if (u->closed)
1386 __gthread_mutex_lock (&unit_lock);
1387 __gthread_mutex_unlock (&u->lock);
1388 if (predec_waiting_locked (u) == 0)
1389 free_mem (u);
1390 goto retry;
1393 dec_waiting_unlocked (u);
1395 return u;
1398 static gfc_unit *
1399 flush_all_units_1 (gfc_unit *u, int min_unit)
1401 while (u != NULL)
1403 if (u->unit_number > min_unit)
1405 gfc_unit *r = flush_all_units_1 (u->left, min_unit);
1406 if (r != NULL)
1407 return r;
1409 if (u->unit_number >= min_unit)
1411 if (__gthread_mutex_trylock (&u->lock))
1412 return u;
1413 if (u->s)
1414 flush (u->s);
1415 __gthread_mutex_unlock (&u->lock);
1417 u = u->right;
1419 return NULL;
1422 void
1423 flush_all_units (void)
1425 gfc_unit *u;
1426 int min_unit = 0;
1428 __gthread_mutex_lock (&unit_lock);
1431 u = flush_all_units_1 (unit_root, min_unit);
1432 if (u != NULL)
1433 inc_waiting_locked (u);
1434 __gthread_mutex_unlock (&unit_lock);
1435 if (u == NULL)
1436 return;
1438 __gthread_mutex_lock (&u->lock);
1440 min_unit = u->unit_number + 1;
1442 if (u->closed == 0)
1444 flush (u->s);
1445 __gthread_mutex_lock (&unit_lock);
1446 __gthread_mutex_unlock (&u->lock);
1447 (void) predec_waiting_locked (u);
1449 else
1451 __gthread_mutex_lock (&unit_lock);
1452 __gthread_mutex_unlock (&u->lock);
1453 if (predec_waiting_locked (u) == 0)
1454 free_mem (u);
1457 while (1);
1461 /* stream_at_bof()-- Returns nonzero if the stream is at the beginning
1462 * of the file. */
1465 stream_at_bof (stream * s)
1467 unix_stream *us;
1469 if (!is_seekable (s))
1470 return 0;
1472 us = (unix_stream *) s;
1474 return us->logical_offset == 0;
1478 /* stream_at_eof()-- Returns nonzero if the stream is at the end
1479 * of the file. */
1482 stream_at_eof (stream * s)
1484 unix_stream *us;
1486 if (!is_seekable (s))
1487 return 0;
1489 us = (unix_stream *) s;
1491 return us->logical_offset == us->dirty_offset;
1495 /* delete_file()-- Given a unit structure, delete the file associated
1496 * with the unit. Returns nonzero if something went wrong. */
1499 delete_file (gfc_unit * u)
1501 char path[PATH_MAX + 1];
1503 if (unpack_filename (path, u->file, u->file_len))
1504 { /* Shouldn't be possible */
1505 errno = ENOENT;
1506 return 1;
1509 return unlink (path);
1513 /* file_exists()-- Returns nonzero if the current filename exists on
1514 * the system */
1517 file_exists (const char *file, gfc_charlen_type file_len)
1519 char path[PATH_MAX + 1];
1520 struct stat statbuf;
1522 if (unpack_filename (path, file, file_len))
1523 return 0;
1525 if (stat (path, &statbuf) < 0)
1526 return 0;
1528 return 1;
1533 static const char yes[] = "YES", no[] = "NO", unknown[] = "UNKNOWN";
1535 /* inquire_sequential()-- Given a fortran string, determine if the
1536 * file is suitable for sequential access. Returns a C-style
1537 * string. */
1539 const char *
1540 inquire_sequential (const char *string, int len)
1542 char path[PATH_MAX + 1];
1543 struct stat statbuf;
1545 if (string == NULL ||
1546 unpack_filename (path, string, len) || stat (path, &statbuf) < 0)
1547 return unknown;
1549 if (S_ISREG (statbuf.st_mode) ||
1550 S_ISCHR (statbuf.st_mode) || S_ISFIFO (statbuf.st_mode))
1551 return yes;
1553 if (S_ISDIR (statbuf.st_mode) || S_ISBLK (statbuf.st_mode))
1554 return no;
1556 return unknown;
1560 /* inquire_direct()-- Given a fortran string, determine if the file is
1561 * suitable for direct access. Returns a C-style string. */
1563 const char *
1564 inquire_direct (const char *string, int len)
1566 char path[PATH_MAX + 1];
1567 struct stat statbuf;
1569 if (string == NULL ||
1570 unpack_filename (path, string, len) || stat (path, &statbuf) < 0)
1571 return unknown;
1573 if (S_ISREG (statbuf.st_mode) || S_ISBLK (statbuf.st_mode))
1574 return yes;
1576 if (S_ISDIR (statbuf.st_mode) ||
1577 S_ISCHR (statbuf.st_mode) || S_ISFIFO (statbuf.st_mode))
1578 return no;
1580 return unknown;
1584 /* inquire_formatted()-- Given a fortran string, determine if the file
1585 * is suitable for formatted form. Returns a C-style string. */
1587 const char *
1588 inquire_formatted (const char *string, int len)
1590 char path[PATH_MAX + 1];
1591 struct stat statbuf;
1593 if (string == NULL ||
1594 unpack_filename (path, string, len) || stat (path, &statbuf) < 0)
1595 return unknown;
1597 if (S_ISREG (statbuf.st_mode) ||
1598 S_ISBLK (statbuf.st_mode) ||
1599 S_ISCHR (statbuf.st_mode) || S_ISFIFO (statbuf.st_mode))
1600 return yes;
1602 if (S_ISDIR (statbuf.st_mode))
1603 return no;
1605 return unknown;
1609 /* inquire_unformatted()-- Given a fortran string, determine if the file
1610 * is suitable for unformatted form. Returns a C-style string. */
1612 const char *
1613 inquire_unformatted (const char *string, int len)
1615 return inquire_formatted (string, len);
1619 /* inquire_access()-- Given a fortran string, determine if the file is
1620 * suitable for access. */
1622 static const char *
1623 inquire_access (const char *string, int len, int mode)
1625 char path[PATH_MAX + 1];
1627 if (string == NULL || unpack_filename (path, string, len) ||
1628 access (path, mode) < 0)
1629 return no;
1631 return yes;
1635 /* inquire_read()-- Given a fortran string, determine if the file is
1636 * suitable for READ access. */
1638 const char *
1639 inquire_read (const char *string, int len)
1641 return inquire_access (string, len, R_OK);
1645 /* inquire_write()-- Given a fortran string, determine if the file is
1646 * suitable for READ access. */
1648 const char *
1649 inquire_write (const char *string, int len)
1651 return inquire_access (string, len, W_OK);
1655 /* inquire_readwrite()-- Given a fortran string, determine if the file is
1656 * suitable for read and write access. */
1658 const char *
1659 inquire_readwrite (const char *string, int len)
1661 return inquire_access (string, len, R_OK | W_OK);
1665 /* file_length()-- Return the file length in bytes, -1 if unknown */
1667 gfc_offset
1668 file_length (stream * s)
1670 return ((unix_stream *) s)->file_length;
1674 /* file_position()-- Return the current position of the file */
1676 gfc_offset
1677 file_position (stream * s)
1679 return ((unix_stream *) s)->logical_offset;
1683 /* is_seekable()-- Return nonzero if the stream is seekable, zero if
1684 * it is not */
1687 is_seekable (stream * s)
1689 /* By convention, if file_length == -1, the file is not
1690 seekable. */
1691 return ((unix_stream *) s)->file_length!=-1;
1695 flush (stream *s)
1697 return fd_flush( (unix_stream *) s);
1701 stream_isatty (stream *s)
1703 return isatty (((unix_stream *) s)->fd);
1706 char *
1707 stream_ttyname (stream *s)
1709 #ifdef HAVE_TTYNAME
1710 return ttyname (((unix_stream *) s)->fd);
1711 #else
1712 return NULL;
1713 #endif
1716 gfc_offset
1717 stream_offset (stream *s)
1719 return (((unix_stream *) s)->logical_offset);
1723 /* How files are stored: This is an operating-system specific issue,
1724 and therefore belongs here. There are three cases to consider.
1726 Direct Access:
1727 Records are written as block of bytes corresponding to the record
1728 length of the file. This goes for both formatted and unformatted
1729 records. Positioning is done explicitly for each data transfer,
1730 so positioning is not much of an issue.
1732 Sequential Formatted:
1733 Records are separated by newline characters. The newline character
1734 is prohibited from appearing in a string. If it does, this will be
1735 messed up on the next read. End of file is also the end of a record.
1737 Sequential Unformatted:
1738 In this case, we are merely copying bytes to and from main storage,
1739 yet we need to keep track of varying record lengths. We adopt
1740 the solution used by f2c. Each record contains a pair of length
1741 markers:
1743 Length of record n in bytes
1744 Data of record n
1745 Length of record n in bytes
1747 Length of record n+1 in bytes
1748 Data of record n+1
1749 Length of record n+1 in bytes
1751 The length is stored at the end of a record to allow backspacing to the
1752 previous record. Between data transfer statements, the file pointer
1753 is left pointing to the first length of the current record.
1755 ENDFILE records are never explicitly stored.