]> git.ipfire.org Git - thirdparty/gcc.git/blame - libgfortran/io/unix.c
2017-07-24 Thomas Koenig <tkoenig@gcc.gnu.org>
[thirdparty/gcc.git] / libgfortran / io / unix.c
CommitLineData
aad93da1 1/* Copyright (C) 2002-2017 Free Software Foundation, Inc.
4ee9c684 2 Contributed by Andy Vaught
84d33b91 3 F2003 I/O support contributed by Jerry DeLisle
4ee9c684 4
5e62a3cc 5This file is part of the GNU Fortran runtime library (libgfortran).
4ee9c684 6
7Libgfortran is free software; you can redistribute it and/or modify
8it under the terms of the GNU General Public License as published by
6bc9506f 9the Free Software Foundation; either version 3, or (at your option)
4ee9c684 10any later version.
11
12Libgfortran is distributed in the hope that it will be useful,
13but WITHOUT ANY WARRANTY; without even the implied warranty of
14MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
15GNU General Public License for more details.
16
6bc9506f 17Under Section 7 of GPL version 3, you are granted additional
18permissions described in the GCC Runtime Library Exception, version
193.1, as published by the Free Software Foundation.
20
21You should have received a copy of the GNU General Public License and
22a copy of the GCC Runtime Library Exception along with this program;
23see the files COPYING3 and COPYING.RUNTIME respectively. If not, see
24<http://www.gnu.org/licenses/>. */
4ee9c684 25
26/* Unix stream I/O module */
27
41f2d5e8 28#include "io.h"
f65f6629 29#include "unix.h"
4ee9c684 30#include <limits.h>
31
ea9ae053 32#ifdef HAVE_UNISTD_H
4ee9c684 33#include <unistd.h>
ea9ae053 34#endif
35
4ee9c684 36#include <sys/stat.h>
37#include <fcntl.h>
38
4ee9c684 39#include <string.h>
40#include <errno.h>
41
c0ecd33c 42
43/* For mingw, we don't identify files by their inode number, but by a
44 64-bit identifier created from a BY_HANDLE_FILE_INFORMATION. */
7a4491a4 45#ifdef __MINGW32__
c0ecd33c 46
47#define WIN32_LEAN_AND_MEAN
48#include <windows.h>
49
f6854450 50#if !defined(_FILE_OFFSET_BITS) || _FILE_OFFSET_BITS != 64
51#undef lseek
4dbc0658 52#define lseek _lseeki64
f6854450 53#undef fstat
7a4491a4 54#define fstat _fstati64
f6854450 55#undef stat
7a4491a4 56#define stat _stati64
f6854450 57#endif
4dbc0658 58
7a4491a4 59#ifndef HAVE_WORKING_STAT
c0ecd33c 60static uint64_t
61id_from_handle (HANDLE hFile)
62{
63 BY_HANDLE_FILE_INFORMATION FileInformation;
64
65 if (hFile == INVALID_HANDLE_VALUE)
66 return 0;
67
68 memset (&FileInformation, 0, sizeof(FileInformation));
69 if (!GetFileInformationByHandle (hFile, &FileInformation))
70 return 0;
71
72 return ((uint64_t) FileInformation.nFileIndexLow)
73 | (((uint64_t) FileInformation.nFileIndexHigh) << 32);
74}
75
76
77static uint64_t
78id_from_path (const char *path)
79{
80 HANDLE hFile;
81 uint64_t res;
82
83 if (!path || !*path || access (path, F_OK))
84 return (uint64_t) -1;
85
86 hFile = CreateFile (path, 0, 0, NULL, OPEN_EXISTING,
87 FILE_FLAG_BACKUP_SEMANTICS | FILE_ATTRIBUTE_READONLY,
88 NULL);
89 res = id_from_handle (hFile);
90 CloseHandle (hFile);
91 return res;
92}
93
94
95static uint64_t
96id_from_fd (const int fd)
97{
98 return id_from_handle ((HANDLE) _get_osfhandle (fd));
99}
100
a8ce53cd 101#endif /* HAVE_WORKING_STAT */
ebb925e3 102
103
104/* On mingw, we don't use umask in tempfile_open(), because it
105 doesn't support the user/group/other-based permissions. */
106#undef HAVE_UMASK
107
a8ce53cd 108#endif /* __MINGW32__ */
109
110
7dfba97b 111/* These flags aren't defined on all targets (mingw32), so provide them
112 here. */
113#ifndef S_IRGRP
114#define S_IRGRP 0
115#endif
116
117#ifndef S_IWGRP
118#define S_IWGRP 0
119#endif
120
121#ifndef S_IROTH
122#define S_IROTH 0
123#endif
124
125#ifndef S_IWOTH
126#define S_IWOTH 0
127#endif
128
76c0a846 129
fb053cd1 130#ifndef HAVE_ACCESS
131
132#ifndef W_OK
133#define W_OK 2
134#endif
135
136#ifndef R_OK
137#define R_OK 4
138#endif
139
140#ifndef F_OK
141#define F_OK 0
142#endif
143
144/* Fallback implementation of access() on systems that don't have it.
145 Only modes R_OK, W_OK and F_OK are used in this file. */
146
147static int
148fallback_access (const char *path, int mode)
149{
398e846b 150 int fd;
151
152 if ((mode & R_OK) && (fd = open (path, O_RDONLY)) < 0)
fb053cd1 153 return -1;
398e846b 154 close (fd);
fb053cd1 155
398e846b 156 if ((mode & W_OK) && (fd = open (path, O_WRONLY)) < 0)
fb053cd1 157 return -1;
398e846b 158 close (fd);
fb053cd1 159
160 if (mode == F_OK)
161 {
f6854450 162 struct stat st;
fb053cd1 163 return stat (path, &st);
164 }
165
166 return 0;
167}
168
169#undef access
170#define access fallback_access
171#endif
172
173
a291e3b6 174/* Fallback directory for creating temporary files. P_tmpdir is
175 defined on many POSIX platforms. */
176#ifndef P_tmpdir
177#ifdef _P_tmpdir
178#define P_tmpdir _P_tmpdir /* MinGW */
179#else
180#define P_tmpdir "/tmp"
181#endif
182#endif
183
184
65f15010 185/* Unix and internal stream I/O module */
76c0a846 186
65f15010 187static const int BUFFER_SIZE = 8192;
76c0a846 188
3479b863 189typedef struct
190{
191 stream st;
192
193 gfc_offset buffer_offset; /* File offset of the start of the buffer */
194 gfc_offset physical_offset; /* Current physical file offset */
195 gfc_offset logical_offset; /* Current logical file offset */
cc65b133 196 gfc_offset file_length; /* Length of the file. */
3479b863 197
198 char *buffer; /* Pointer to the buffer. */
199 int fd; /* The POSIX file descriptor. */
200
201 int active; /* Length of valid bytes in the buffer */
202
203 int ndirty; /* Dirty bytes starting at buffer_offset */
204
01cd2c93 205 /* Cached stat(2) values. */
206 dev_t st_dev;
207 ino_t st_ino;
fe34985d 208
209 bool unbuffered; /* Buffer should be flushed after each I/O statement. */
3479b863 210}
211unix_stream;
212
213
4ee9c684 214/* fix_fd()-- Given a file descriptor, make sure it is not one of the
25a5ce27 215 standard descriptors, returning a non-standard descriptor. If the
216 user specifies that system errors should go to standard output,
217 then closes standard output, we don't want the system errors to a
218 file that has been given file descriptor 1 or 0. We want to send
219 the error to the invalid descriptor. */
4ee9c684 220
221static int
222fix_fd (int fd)
223{
e0582811 224#ifdef HAVE_DUP
4ee9c684 225 int input, output, error;
226
227 input = output = error = 0;
228
7145fd06 229 /* Unix allocates the lowest descriptors first, so a loop is not
230 required, but this order is. */
4ee9c684 231 if (fd == STDIN_FILENO)
232 {
233 fd = dup (fd);
234 input = 1;
235 }
236 if (fd == STDOUT_FILENO)
237 {
238 fd = dup (fd);
239 output = 1;
240 }
241 if (fd == STDERR_FILENO)
242 {
243 fd = dup (fd);
244 error = 1;
245 }
246
247 if (input)
248 close (STDIN_FILENO);
249 if (output)
250 close (STDOUT_FILENO);
251 if (error)
252 close (STDERR_FILENO);
e0582811 253#endif
4ee9c684 254
255 return fd;
256}
257
258
2488b3b6 259/* If the stream corresponds to a preconnected unit, we flush the
260 corresponding C stream. This is bugware for mixed C-Fortran codes
261 where the C code doesn't flush I/O before returning. */
262void
25a5ce27 263flush_if_preconnected (stream *s)
2488b3b6 264{
265 int fd;
266
267 fd = ((unix_stream *) s)->fd;
268 if (fd == STDIN_FILENO)
269 fflush (stdin);
270 else if (fd == STDOUT_FILENO)
271 fflush (stdout);
272 else if (fd == STDERR_FILENO)
273 fflush (stderr);
274}
275
4ee9c684 276
65f15010 277/********************************************************************
278Raw I/O functions (read, write, seek, tell, truncate, close).
279
280These functions wrap the basic POSIX I/O syscalls. Any deviation in
281semantics is a bug, except the following: write restarts in case
282of being interrupted by a signal, and as the first argument the
283functions take the unix_stream struct rather than an integer file
284descriptor. Also, for POSIX read() and write() a nbyte argument larger
285than SSIZE_MAX is undefined; here the type of nbyte is ssize_t rather
286than size_t as for POSIX read/write.
287*********************************************************************/
4ee9c684 288
b2a112ca 289static int
25a5ce27 290raw_flush (unix_stream *s __attribute__ ((unused)))
b2a112ca 291{
65f15010 292 return 0;
4ee9c684 293}
294
65f15010 295static ssize_t
25a5ce27 296raw_read (unix_stream *s, void *buf, ssize_t nbyte)
65f15010 297{
298 /* For read we can't do I/O in a loop like raw_write does, because
714f1fe2 299 that will break applications that wait for interactive I/O. We
300 still can loop around EINTR, though. */
301 while (true)
302 {
303 ssize_t trans = read (s->fd, buf, nbyte);
304 if (trans == -1 && errno == EINTR)
305 continue;
306 return trans;
307 }
65f15010 308}
4ee9c684 309
65f15010 310static ssize_t
25a5ce27 311raw_write (unix_stream *s, const void *buf, ssize_t nbyte)
4ee9c684 312{
65f15010 313 ssize_t trans, bytes_left;
b2a112ca 314 char *buf_st;
b2a112ca 315
65f15010 316 bytes_left = nbyte;
b2a112ca 317 buf_st = (char *) buf;
318
319 /* We must write in a loop since some systems don't restart system
320 calls in case of a signal. */
321 while (bytes_left > 0)
4ee9c684 322 {
65f15010 323 trans = write (s->fd, buf_st, bytes_left);
714f1fe2 324 if (trans == -1)
b2a112ca 325 {
326 if (errno == EINTR)
327 continue;
328 else
65f15010 329 return trans;
b2a112ca 330 }
331 buf_st += trans;
332 bytes_left -= trans;
4ee9c684 333 }
334
65f15010 335 return nbyte - bytes_left;
4ee9c684 336}
4ee9c684 337
4dbc0658 338static gfc_offset
25a5ce27 339raw_seek (unix_stream *s, gfc_offset offset, int whence)
65f15010 340{
714f1fe2 341 while (true)
342 {
343 gfc_offset off = lseek (s->fd, offset, whence);
344 if (off == (gfc_offset) -1 && errno == EINTR)
345 continue;
346 return off;
347 }
65f15010 348}
4ee9c684 349
4dbc0658 350static gfc_offset
25a5ce27 351raw_tell (unix_stream *s)
65f15010 352{
714f1fe2 353 while (true)
354 {
355 gfc_offset off = lseek (s->fd, 0, SEEK_CUR);
356 if (off == (gfc_offset) -1 && errno == EINTR)
357 continue;
358 return off;
359 }
65f15010 360}
4ee9c684 361
41178014 362static gfc_offset
25a5ce27 363raw_size (unix_stream *s)
41178014 364{
365 struct stat statbuf;
714f1fe2 366 if (TEMP_FAILURE_RETRY (fstat (s->fd, &statbuf)) == -1)
367 return -1;
6dfa90cc 368 if (S_ISREG (statbuf.st_mode))
369 return statbuf.st_size;
370 else
371 return 0;
41178014 372}
373
65f15010 374static int
25a5ce27 375raw_truncate (unix_stream *s, gfc_offset length)
4ee9c684 376{
4dbc0658 377#ifdef __MINGW32__
378 HANDLE h;
379 gfc_offset cur;
380
381 if (isatty (s->fd))
382 {
383 errno = EBADF;
384 return -1;
385 }
db23ac43 386 h = (HANDLE) _get_osfhandle (s->fd);
4dbc0658 387 if (h == INVALID_HANDLE_VALUE)
388 {
389 errno = EBADF;
390 return -1;
391 }
392 cur = lseek (s->fd, 0, SEEK_CUR);
393 if (cur == -1)
394 return -1;
395 if (lseek (s->fd, length, SEEK_SET) == -1)
396 goto error;
397 if (!SetEndOfFile (h))
398 {
399 errno = EBADF;
400 goto error;
401 }
402 if (lseek (s->fd, cur, SEEK_SET) == -1)
403 return -1;
404 return 0;
405 error:
406 lseek (s->fd, cur, SEEK_SET);
407 return -1;
408#elif defined HAVE_FTRUNCATE
714f1fe2 409 if (TEMP_FAILURE_RETRY (ftruncate (s->fd, length)) == -1)
410 return -1;
411 return 0;
65f15010 412#elif defined HAVE_CHSIZE
413 return chsize (s->fd, length);
414#else
415 runtime_error ("required ftruncate or chsize support not present");
416 return -1;
417#endif
4ee9c684 418}
419
65f15010 420static int
25a5ce27 421raw_close (unix_stream *s)
65f15010 422{
423 int retval;
424
7b89cd4f 425 if (s->fd == -1)
426 retval = -1;
427 else if (s->fd != STDOUT_FILENO
8b89aa85 428 && s->fd != STDERR_FILENO
429 && s->fd != STDIN_FILENO)
714f1fe2 430 {
431 retval = close (s->fd);
432 /* close() and EINTR is special, as the file descriptor is
433 deallocated before doing anything that might cause the
434 operation to be interrupted. Thus if we get EINTR the best we
435 can do is ignore it and continue (otherwise if we try again
436 the file descriptor may have been allocated again to some
437 other file). */
438 if (retval == -1 && errno == EINTR)
439 retval = errno = 0;
440 }
8b89aa85 441 else
6ffeb312 442 retval = 0;
5e62a3cc 443 free (s);
65f15010 444 return retval;
445}
4ee9c684 446
155fbd31 447static int
25a5ce27 448raw_markeor (unix_stream *s __attribute__ ((unused)))
155fbd31 449{
450 return 0;
451}
452
292d5498 453static const struct stream_vtable raw_vtable = {
454 .read = (void *) raw_read,
455 .write = (void *) raw_write,
456 .seek = (void *) raw_seek,
457 .tell = (void *) raw_tell,
458 .size = (void *) raw_size,
459 .trunc = (void *) raw_truncate,
460 .close = (void *) raw_close,
155fbd31 461 .flush = (void *) raw_flush,
462 .markeor = (void *) raw_markeor
292d5498 463};
464
65f15010 465static int
25a5ce27 466raw_init (unix_stream *s)
65f15010 467{
292d5498 468 s->st.vptr = &raw_vtable;
4ee9c684 469
65f15010 470 s->buffer = NULL;
471 return 0;
472}
b2a112ca 473
4ee9c684 474
65f15010 475/*********************************************************************
476Buffered I/O functions. These functions have the same semantics as the
477raw I/O functions above, except that they are buffered in order to
478improve performance. The buffer must be flushed when switching from
fe34985d 479reading to writing and vice versa.
65f15010 480*********************************************************************/
481
482static int
25a5ce27 483buf_flush (unix_stream *s)
4ee9c684 484{
65f15010 485 int writelen;
486
487 /* Flushing in read mode means discarding read bytes. */
488 s->active = 0;
b2a112ca 489
4ee9c684 490 if (s->ndirty == 0)
65f15010 491 return 0;
352597f9 492
cc65b133 493 if (s->physical_offset != s->buffer_offset
714f1fe2 494 && raw_seek (s, s->buffer_offset, SEEK_SET) < 0)
65f15010 495 return -1;
4ee9c684 496
65f15010 497 writelen = raw_write (s, s->buffer, s->ndirty);
4ee9c684 498
65f15010 499 s->physical_offset = s->buffer_offset + writelen;
5a78b88f 500
cc65b133 501 if (s->physical_offset > s->file_length)
65f15010 502 s->file_length = s->physical_offset;
b2a112ca 503
504 s->ndirty -= writelen;
505 if (s->ndirty != 0)
65f15010 506 return -1;
4ee9c684 507
65f15010 508 return 0;
4ee9c684 509}
510
65f15010 511static ssize_t
25a5ce27 512buf_read (unix_stream *s, void *buf, ssize_t nbyte)
4ee9c684 513{
65f15010 514 if (s->active == 0)
515 s->buffer_offset = s->logical_offset;
5d5f00d5 516
65f15010 517 /* Is the data we want in the buffer? */
518 if (s->logical_offset + nbyte <= s->buffer_offset + s->active
519 && s->buffer_offset <= s->logical_offset)
c60f0c1e 520 {
521 /* When nbyte == 0, buf can be NULL which would lead to undefined
522 behavior if we called memcpy(). */
523 if (nbyte != 0)
524 memcpy (buf, s->buffer + (s->logical_offset - s->buffer_offset),
525 nbyte);
526 }
5d5f00d5 527 else
528 {
65f15010 529 /* First copy the active bytes if applicable, then read the rest
530 either directly or filling the buffer. */
531 char *p;
532 int nread = 0;
533 ssize_t to_read, did_read;
534 gfc_offset new_logical;
535
536 p = (char *) buf;
537 if (s->logical_offset >= s->buffer_offset
538 && s->buffer_offset + s->active >= s->logical_offset)
539 {
540 nread = s->active - (s->logical_offset - s->buffer_offset);
541 memcpy (buf, s->buffer + (s->logical_offset - s->buffer_offset),
542 nread);
543 p += nread;
544 }
545 /* At this point we consider all bytes in the buffer discarded. */
546 to_read = nbyte - nread;
547 new_logical = s->logical_offset + nread;
cc65b133 548 if (s->physical_offset != new_logical
714f1fe2 549 && raw_seek (s, new_logical, SEEK_SET) < 0)
65f15010 550 return -1;
551 s->buffer_offset = s->physical_offset = new_logical;
552 if (to_read <= BUFFER_SIZE/2)
553 {
554 did_read = raw_read (s, s->buffer, BUFFER_SIZE);
3287030a 555 if (likely (did_read >= 0))
556 {
557 s->physical_offset += did_read;
558 s->active = did_read;
559 did_read = (did_read > to_read) ? to_read : did_read;
560 memcpy (p, s->buffer, did_read);
561 }
562 else
563 return did_read;
65f15010 564 }
565 else
566 {
567 did_read = raw_read (s, p, to_read);
3287030a 568 if (likely (did_read >= 0))
569 {
570 s->physical_offset += did_read;
571 s->active = 0;
572 }
573 else
574 return did_read;
65f15010 575 }
576 nbyte = did_read + nread;
4ee9c684 577 }
65f15010 578 s->logical_offset += nbyte;
579 return nbyte;
4ee9c684 580}
581
65f15010 582static ssize_t
25a5ce27 583buf_write (unix_stream *s, const void *buf, ssize_t nbyte)
4ee9c684 584{
65f15010 585 if (s->ndirty == 0)
586 s->buffer_offset = s->logical_offset;
587
588 /* Does the data fit into the buffer? As a special case, if the
589 buffer is empty and the request is bigger than BUFFER_SIZE/2,
590 write directly. This avoids the case where the buffer would have
591 to be flushed at every write. */
592 if (!(s->ndirty == 0 && nbyte > BUFFER_SIZE/2)
593 && s->logical_offset + nbyte <= s->buffer_offset + BUFFER_SIZE
594 && s->buffer_offset <= s->logical_offset
595 && s->buffer_offset + s->ndirty >= s->logical_offset)
fef3501c 596 {
65f15010 597 memcpy (s->buffer + (s->logical_offset - s->buffer_offset), buf, nbyte);
598 int nd = (s->logical_offset - s->buffer_offset) + nbyte;
599 if (nd > s->ndirty)
600 s->ndirty = nd;
fef3501c 601 }
602 else
603 {
65f15010 604 /* Flush, and either fill the buffer with the new data, or if
605 the request is bigger than the buffer size, write directly
606 bypassing the buffer. */
607 buf_flush (s);
608 if (nbyte <= BUFFER_SIZE/2)
609 {
610 memcpy (s->buffer, buf, nbyte);
611 s->buffer_offset = s->logical_offset;
612 s->ndirty += nbyte;
613 }
614 else
35a97371 615 {
cc65b133 616 if (s->physical_offset != s->logical_offset)
35a97371 617 {
714f1fe2 618 if (raw_seek (s, s->logical_offset, SEEK_SET) < 0)
35a97371 619 return -1;
620 s->physical_offset = s->logical_offset;
621 }
622
623 nbyte = raw_write (s, buf, nbyte);
624 s->physical_offset += nbyte;
625 }
4ee9c684 626 }
65f15010 627 s->logical_offset += nbyte;
cc65b133 628 if (s->logical_offset > s->file_length)
65f15010 629 s->file_length = s->logical_offset;
630 return nbyte;
4ee9c684 631}
632
155fbd31 633
634/* "Unbuffered" really means I/O statement buffering. For formatted
635 I/O, the fbuf manages this, and then uses raw I/O. For unformatted
636 I/O, buffered I/O is used, and the buffer is flushed at the end of
637 each I/O statement, where this function is called. Alternatively,
638 the buffer is flushed at the end of the record if the buffer is
639 more than half full; this prevents needless seeking back and forth
640 when writing sequential unformatted. */
641
642static int
25a5ce27 643buf_markeor (unix_stream *s)
155fbd31 644{
645 if (s->unbuffered || s->ndirty >= BUFFER_SIZE / 2)
646 return buf_flush (s);
647 return 0;
648}
649
4dbc0658 650static gfc_offset
25a5ce27 651buf_seek (unix_stream *s, gfc_offset offset, int whence)
4ee9c684 652{
65f15010 653 switch (whence)
352597f9 654 {
65f15010 655 case SEEK_SET:
656 break;
657 case SEEK_CUR:
658 offset += s->logical_offset;
659 break;
660 case SEEK_END:
661 offset += s->file_length;
662 break;
663 default:
664 return -1;
352597f9 665 }
65f15010 666 if (offset < 0)
72909c79 667 {
65f15010 668 errno = EINVAL;
669 return -1;
72909c79 670 }
65f15010 671 s->logical_offset = offset;
672 return offset;
4ee9c684 673}
674
4dbc0658 675static gfc_offset
25a5ce27 676buf_tell (unix_stream *s)
56f281a2 677{
f22f0b38 678 return buf_seek (s, 0, SEEK_CUR);
56f281a2 679}
b2a112ca 680
41178014 681static gfc_offset
25a5ce27 682buf_size (unix_stream *s)
41178014 683{
684 return s->file_length;
685}
686
b2a112ca 687static int
25a5ce27 688buf_truncate (unix_stream *s, gfc_offset length)
b2a112ca 689{
65f15010 690 int r;
b2a112ca 691
65f15010 692 if (buf_flush (s) != 0)
693 return -1;
694 r = raw_truncate (s, length);
695 if (r == 0)
696 s->file_length = length;
697 return r;
b2a112ca 698}
699
b2a112ca 700static int
25a5ce27 701buf_close (unix_stream *s)
4ee9c684 702{
65f15010 703 if (buf_flush (s) != 0)
704 return -1;
5e62a3cc 705 free (s->buffer);
65f15010 706 return raw_close (s);
4ee9c684 707}
708
292d5498 709static const struct stream_vtable buf_vtable = {
710 .read = (void *) buf_read,
711 .write = (void *) buf_write,
712 .seek = (void *) buf_seek,
713 .tell = (void *) buf_tell,
714 .size = (void *) buf_size,
715 .trunc = (void *) buf_truncate,
716 .close = (void *) buf_close,
155fbd31 717 .flush = (void *) buf_flush,
718 .markeor = (void *) buf_markeor
292d5498 719};
720
65f15010 721static int
25a5ce27 722buf_init (unix_stream *s)
4ee9c684 723{
292d5498 724 s->st.vptr = &buf_vtable;
4ee9c684 725
25c067ae 726 s->buffer = xmalloc (BUFFER_SIZE);
65f15010 727 return 0;
4ee9c684 728}
729
730
4ee9c684 731/*********************************************************************
732 memory stream functions - These are used for internal files
733
734 The idea here is that a single stream structure is created and all
735 requests must be satisfied from it. The location and size of the
736 buffer is the character variable supplied to the READ or WRITE
737 statement.
738
739*********************************************************************/
740
65f15010 741char *
25a5ce27 742mem_alloc_r (stream *strm, int *len)
4ee9c684 743{
25a5ce27 744 unix_stream *s = (unix_stream *) strm;
b093181d 745 gfc_offset n;
d875179d 746 gfc_offset where = s->logical_offset;
4ee9c684 747
748 if (where < s->buffer_offset || where > s->buffer_offset + s->active)
749 return NULL;
750
11de4bf9 751 n = s->buffer_offset + s->active - where;
4ee9c684 752 if (*len > n)
753 *len = n;
754
65f15010 755 s->logical_offset = where + *len;
756
4ee9c684 757 return s->buffer + (where - s->buffer_offset);
758}
759
760
e0aaacb7 761char *
25a5ce27 762mem_alloc_r4 (stream *strm, int *len)
e0aaacb7 763{
25a5ce27 764 unix_stream *s = (unix_stream *) strm;
e0aaacb7 765 gfc_offset n;
766 gfc_offset where = s->logical_offset;
767
768 if (where < s->buffer_offset || where > s->buffer_offset + s->active)
769 return NULL;
770
771 n = s->buffer_offset + s->active - where;
772 if (*len > n)
773 *len = n;
774
775 s->logical_offset = where + *len;
776
777 return s->buffer + (where - s->buffer_offset) * 4;
778}
779
780
65f15010 781char *
25a5ce27 782mem_alloc_w (stream *strm, int *len)
4ee9c684 783{
25a5ce27 784 unix_stream *s = (unix_stream *)strm;
b093181d 785 gfc_offset m;
d875179d 786 gfc_offset where = s->logical_offset;
4ee9c684 787
4ee9c684 788 m = where + *len;
789
2639e4cd 790 if (where < s->buffer_offset)
4ee9c684 791 return NULL;
792
2639e4cd 793 if (m > s->file_length)
72231bd6 794 return NULL;
2639e4cd 795
4ee9c684 796 s->logical_offset = m;
797
798 return s->buffer + (where - s->buffer_offset);
799}
800
801
1bb7bffb 802gfc_char4_t *
25a5ce27 803mem_alloc_w4 (stream *strm, int *len)
e0aaacb7 804{
25a5ce27 805 unix_stream *s = (unix_stream *)strm;
e0aaacb7 806 gfc_offset m;
807 gfc_offset where = s->logical_offset;
1bb7bffb 808 gfc_char4_t *result = (gfc_char4_t *) s->buffer;
e0aaacb7 809
810 m = where + *len;
811
812 if (where < s->buffer_offset)
813 return NULL;
814
815 if (m > s->file_length)
816 return NULL;
817
818 s->logical_offset = m;
1bb7bffb 819 return &result[where - s->buffer_offset];
e0aaacb7 820}
821
822
655c7caa 823/* Stream read function for character(kind=1) internal units. */
b2a112ca 824
65f15010 825static ssize_t
25a5ce27 826mem_read (stream *s, void *buf, ssize_t nbytes)
b2a112ca 827{
828 void *p;
65f15010 829 int nb = nbytes;
b2a112ca 830
65f15010 831 p = mem_alloc_r (s, &nb);
b2a112ca 832 if (p)
833 {
65f15010 834 memcpy (buf, p, nb);
835 return (ssize_t) nb;
b2a112ca 836 }
837 else
65f15010 838 return 0;
b2a112ca 839}
840
841
e0aaacb7 842/* Stream read function for chracter(kind=4) internal units. */
843
844static ssize_t
25a5ce27 845mem_read4 (stream *s, void *buf, ssize_t nbytes)
e0aaacb7 846{
847 void *p;
848 int nb = nbytes;
849
fe0171a5 850 p = mem_alloc_r4 (s, &nb);
e0aaacb7 851 if (p)
852 {
fe0171a5 853 memcpy (buf, p, nb * 4);
e0aaacb7 854 return (ssize_t) nb;
855 }
856 else
857 return 0;
858}
859
860
861/* Stream write function for character(kind=1) internal units. */
b2a112ca 862
65f15010 863static ssize_t
25a5ce27 864mem_write (stream *s, const void *buf, ssize_t nbytes)
b2a112ca 865{
866 void *p;
65f15010 867 int nb = nbytes;
b2a112ca 868
65f15010 869 p = mem_alloc_w (s, &nb);
b2a112ca 870 if (p)
871 {
65f15010 872 memcpy (p, buf, nb);
873 return (ssize_t) nb;
b2a112ca 874 }
875 else
65f15010 876 return 0;
b2a112ca 877}
878
879
e0aaacb7 880/* Stream write function for character(kind=4) internal units. */
881
882static ssize_t
25a5ce27 883mem_write4 (stream *s, const void *buf, ssize_t nwords)
e0aaacb7 884{
885 gfc_char4_t *p;
886 int nw = nwords;
887
1bb7bffb 888 p = mem_alloc_w4 (s, &nw);
e0aaacb7 889 if (p)
890 {
891 while (nw--)
892 *p++ = (gfc_char4_t) *((char *) buf);
893 return nwords;
894 }
895 else
896 return 0;
897}
898
899
4dbc0658 900static gfc_offset
25a5ce27 901mem_seek (stream *strm, gfc_offset offset, int whence)
4ee9c684 902{
25a5ce27 903 unix_stream *s = (unix_stream *)strm;
65f15010 904 switch (whence)
905 {
906 case SEEK_SET:
907 break;
908 case SEEK_CUR:
909 offset += s->logical_offset;
910 break;
911 case SEEK_END:
912 offset += s->file_length;
913 break;
914 default:
915 return -1;
916 }
917
918 /* Note that for internal array I/O it's actually possible to have a
919 negative offset, so don't check for that. */
4ee9c684 920 if (offset > s->file_length)
921 {
65f15010 922 errno = EINVAL;
923 return -1;
4ee9c684 924 }
925
926 s->logical_offset = offset;
65f15010 927
928 /* Returning < 0 is the error indicator for sseek(), so return 0 if
929 offset is negative. Thus if the return value is 0, the caller
930 has to use stell() to get the real value of logical_offset. */
931 if (offset >= 0)
932 return offset;
933 return 0;
4ee9c684 934}
935
936
4dbc0658 937static gfc_offset
25a5ce27 938mem_tell (stream *s)
56f281a2 939{
65f15010 940 return ((unix_stream *)s)->logical_offset;
56f281a2 941}
942
943
4ee9c684 944static int
25a5ce27 945mem_truncate (unix_stream *s __attribute__ ((unused)),
4dbc0658 946 gfc_offset length __attribute__ ((unused)))
4ee9c684 947{
65f15010 948 return 0;
4ee9c684 949}
950
951
65f15010 952static int
25a5ce27 953mem_flush (unix_stream *s __attribute__ ((unused)))
4ee9c684 954{
65f15010 955 return 0;
4ee9c684 956}
957
958
65f15010 959static int
25a5ce27 960mem_close (unix_stream *s)
4ee9c684 961{
dd045aee 962 free (s);
4ee9c684 963
65f15010 964 return 0;
965}
4ee9c684 966
292d5498 967static const struct stream_vtable mem_vtable = {
968 .read = (void *) mem_read,
969 .write = (void *) mem_write,
970 .seek = (void *) mem_seek,
971 .tell = (void *) mem_tell,
972 /* buf_size is not a typo, we just reuse an identical
973 implementation. */
974 .size = (void *) buf_size,
975 .trunc = (void *) mem_truncate,
976 .close = (void *) mem_close,
155fbd31 977 .flush = (void *) mem_flush,
978 .markeor = (void *) raw_markeor
292d5498 979};
980
981static const struct stream_vtable mem4_vtable = {
982 .read = (void *) mem_read4,
983 .write = (void *) mem_write4,
984 .seek = (void *) mem_seek,
985 .tell = (void *) mem_tell,
986 /* buf_size is not a typo, we just reuse an identical
987 implementation. */
988 .size = (void *) buf_size,
989 .trunc = (void *) mem_truncate,
990 .close = (void *) mem_close,
155fbd31 991 .flush = (void *) mem_flush,
992 .markeor = (void *) raw_markeor
292d5498 993};
4ee9c684 994
995/*********************************************************************
996 Public functions -- A reimplementation of this module needs to
997 define functional equivalents of the following.
998*********************************************************************/
999
e0aaacb7 1000/* open_internal()-- Returns a stream structure from a character(kind=1)
1001 internal file */
4ee9c684 1002
1003stream *
cf4abc57 1004open_internal (char *base, int length, gfc_offset offset)
4ee9c684 1005{
65f15010 1006 unix_stream *s;
4ee9c684 1007
33123ed7 1008 s = xcalloc (1, sizeof (unix_stream));
4ee9c684 1009
1010 s->buffer = base;
cf4abc57 1011 s->buffer_offset = offset;
4ee9c684 1012
4ee9c684 1013 s->active = s->file_length = length;
1014
292d5498 1015 s->st.vptr = &mem_vtable;
4ee9c684 1016
e0aaacb7 1017 return (stream *) s;
1018}
1019
1020/* open_internal4()-- Returns a stream structure from a character(kind=4)
1021 internal file */
1022
1023stream *
1024open_internal4 (char *base, int length, gfc_offset offset)
1025{
1026 unix_stream *s;
1027
33123ed7 1028 s = xcalloc (1, sizeof (unix_stream));
e0aaacb7 1029
1030 s->buffer = base;
1031 s->buffer_offset = offset;
1032
33ea6150 1033 s->active = s->file_length = length * sizeof (gfc_char4_t);
e0aaacb7 1034
292d5498 1035 s->st.vptr = &mem4_vtable;
e0aaacb7 1036
25a5ce27 1037 return (stream *)s;
4ee9c684 1038}
1039
1040
1041/* fd_to_stream()-- Given an open file descriptor, build a stream
25a5ce27 1042 around it. */
4ee9c684 1043
1044static stream *
fe34985d 1045fd_to_stream (int fd, bool unformatted)
4ee9c684 1046{
f6854450 1047 struct stat statbuf;
4ee9c684 1048 unix_stream *s;
1049
33123ed7 1050 s = xcalloc (1, sizeof (unix_stream));
4ee9c684 1051
1052 s->fd = fd;
4ee9c684 1053
1054 /* Get the current length of the file. */
1055
714f1fe2 1056 if (TEMP_FAILURE_RETRY (fstat (fd, &statbuf)) == -1)
7b89cd4f 1057 {
1058 s->st_dev = s->st_ino = -1;
1059 s->file_length = 0;
1060 if (errno == EBADF)
1061 s->fd = -1;
1062 raw_init (s);
1063 return (stream *) s;
1064 }
352597f9 1065
01cd2c93 1066 s->st_dev = statbuf.st_dev;
1067 s->st_ino = statbuf.st_ino;
cc65b133 1068 s->file_length = statbuf.st_size;
1069
1070 /* Only use buffered IO for regular files. */
1071 if (S_ISREG (statbuf.st_mode)
1072 && !options.all_unbuffered
1073 && !(options.unbuffered_preconnected &&
1074 (s->fd == STDIN_FILENO
1075 || s->fd == STDOUT_FILENO
1076 || s->fd == STDERR_FILENO)))
1077 buf_init (s);
9bfd25a7 1078 else
fe34985d 1079 {
1080 if (unformatted)
1081 {
1082 s->unbuffered = true;
1083 buf_init (s);
1084 }
1085 else
1086 raw_init (s);
1087 }
4ee9c684 1088
1089 return (stream *) s;
1090}
1091
1092
771c1b50 1093/* Given the Fortran unit number, convert it to a C file descriptor. */
1094
1095int
60c514ba 1096unit_to_fd (int unit)
771c1b50 1097{
771c1b50 1098 gfc_unit *us;
60c514ba 1099 int fd;
771c1b50 1100
60c514ba 1101 us = find_unit (unit);
771c1b50 1102 if (us == NULL)
1103 return -1;
1104
60c514ba 1105 fd = ((unix_stream *) us->s)->fd;
1106 unlock_unit (us);
1107 return fd;
771c1b50 1108}
1109
1110
544db64f 1111/* Set the close-on-exec flag for an existing fd, if the system
1112 supports such. */
1113
1114static void __attribute__ ((unused))
1115set_close_on_exec (int fd __attribute__ ((unused)))
1116{
1117 /* Mingw does not define F_SETFD. */
7520a494 1118#if defined(HAVE_FCNTL) && defined(F_SETFD) && defined(FD_CLOEXEC)
544db64f 1119 if (fd >= 0)
1120 fcntl(fd, F_SETFD, FD_CLOEXEC);
1121#endif
1122}
1123
1124
a291e3b6 1125/* Helper function for tempfile(). Tries to open a temporary file in
1126 the directory specified by tempdir. If successful, the file name is
1127 stored in fname and the descriptor returned. Returns -1 on
1128 failure. */
4ee9c684 1129
1130static int
a291e3b6 1131tempfile_open (const char *tempdir, char **fname)
4ee9c684 1132{
4ee9c684 1133 int fd;
a291e3b6 1134 const char *slash = "/";
34f4b81f 1135#if defined(HAVE_UMASK) && defined(HAVE_MKSTEMP)
1136 mode_t mode_mask;
1137#endif
4ee9c684 1138
a291e3b6 1139 if (!tempdir)
1140 return -1;
d06c5aaa 1141
a291e3b6 1142 /* Check for the special case that tempdir ends with a slash or
1143 backslash. */
1144 size_t tempdirlen = strlen (tempdir);
d06c5aaa 1145 if (*tempdir == 0 || tempdir[tempdirlen - 1] == '/'
db23ac43 1146#ifdef __MINGW32__
d06c5aaa 1147 || tempdir[tempdirlen - 1] == '\\'
db23ac43 1148#endif
1149 )
1150 slash = "";
4ee9c684 1151
9f732c4e 1152 /* Take care that the template is longer in the mktemp() branch. */
25a5ce27 1153 char *template = xmalloc (tempdirlen + 23);
4ee9c684 1154
7dfba97b 1155#ifdef HAVE_MKSTEMP
d151c82c 1156 snprintf (template, tempdirlen + 23, "%s%sgfortrantmpXXXXXX",
1157 tempdir, slash);
4ee9c684 1158
34f4b81f 1159#ifdef HAVE_UMASK
1160 /* Temporarily set the umask such that the file has 0600 permissions. */
1161 mode_mask = umask (S_IXUSR | S_IRWXG | S_IRWXO);
1162#endif
1163
544db64f 1164#if defined(HAVE_MKOSTEMP) && defined(O_CLOEXEC)
714f1fe2 1165 TEMP_FAILURE_RETRY (fd = mkostemp (template, O_CLOEXEC));
544db64f 1166#else
714f1fe2 1167 TEMP_FAILURE_RETRY (fd = mkstemp (template));
544db64f 1168 set_close_on_exec (fd);
1169#endif
4ee9c684 1170
34f4b81f 1171#ifdef HAVE_UMASK
1172 (void) umask (mode_mask);
1173#endif
1174
7dfba97b 1175#else /* HAVE_MKSTEMP */
726fd258 1176 fd = -1;
a291e3b6 1177 int count = 0;
1178 size_t slashlen = strlen (slash);
544db64f 1179 int flags = O_RDWR | O_CREAT | O_EXCL;
1180#if defined(HAVE_CRLF) && defined(O_BINARY)
1181 flags |= O_BINARY;
1182#endif
1183#ifdef O_CLOEXEC
1184 flags |= O_CLOEXEC;
1185#endif
726fd258 1186 do
1187 {
d151c82c 1188 snprintf (template, tempdirlen + 23, "%s%sgfortrantmpaaaXXXXXX",
1189 tempdir, slash);
d06c5aaa 1190 if (count > 0)
1191 {
1192 int c = count;
1193 template[tempdirlen + slashlen + 13] = 'a' + (c% 26);
1194 c /= 26;
1195 template[tempdirlen + slashlen + 12] = 'a' + (c % 26);
1196 c /= 26;
1197 template[tempdirlen + slashlen + 11] = 'a' + (c % 26);
1198 if (c >= 26)
1199 break;
1200 }
1201
726fd258 1202 if (!mktemp (template))
d06c5aaa 1203 {
1204 errno = EEXIST;
1205 count++;
1206 continue;
1207 }
1208
714f1fe2 1209 TEMP_FAILURE_RETRY (fd = open (template, flags, S_IRUSR | S_IWUSR));
726fd258 1210 }
1211 while (fd == -1 && errno == EEXIST);
544db64f 1212#ifndef O_CLOEXEC
1213 set_close_on_exec (fd);
1214#endif
7dfba97b 1215#endif /* HAVE_MKSTEMP */
1216
a291e3b6 1217 *fname = template;
1218 return fd;
1219}
1220
1221
1222/* tempfile()-- Generate a temporary filename for a scratch file and
25a5ce27 1223 open it. mkstemp() opens the file for reading and writing, but the
1224 library mode prevents anything that is not allowed. The descriptor
1225 is returned, which is -1 on error. The template is pointed to by
1226 opp->file, which is copied into the unit structure
1227 and freed later. */
a291e3b6 1228
1229static int
1230tempfile (st_parameter_open *opp)
1231{
1232 const char *tempdir;
1233 char *fname;
1234 int fd = -1;
1235
1236 tempdir = secure_getenv ("TMPDIR");
1237 fd = tempfile_open (tempdir, &fname);
1238#ifdef __MINGW32__
1239 if (fd == -1)
1240 {
1241 char buffer[MAX_PATH + 1];
1242 DWORD ret;
1243 ret = GetTempPath (MAX_PATH, buffer);
1244 /* If we are not able to get a temp-directory, we use
1245 current directory. */
1246 if (ret > MAX_PATH || !ret)
1247 buffer[0] = 0;
1248 else
1249 buffer[ret] = 0;
1250 tempdir = strdup (buffer);
1251 fd = tempfile_open (tempdir, &fname);
1252 }
1253#elif defined(__CYGWIN__)
1254 if (fd == -1)
1255 {
1256 tempdir = secure_getenv ("TMP");
1257 fd = tempfile_open (tempdir, &fname);
1258 }
1259 if (fd == -1)
1260 {
1261 tempdir = secure_getenv ("TEMP");
1262 fd = tempfile_open (tempdir, &fname);
1263 }
1264#endif
1265 if (fd == -1)
1266 fd = tempfile_open (P_tmpdir, &fname);
1267
1268 opp->file = fname;
1269 opp->file_len = strlen (fname); /* Don't include trailing nul */
4ee9c684 1270
1271 return fd;
1272}
1273
1274
fc3d374a 1275/* regular_file2()-- Open a regular file.
25a5ce27 1276 Change flags->action if it is ACTION_UNSPECIFIED on entry,
1277 unless an error occurs.
1278 Returns the descriptor, which is less than zero on error. */
4ee9c684 1279
1280static int
fc3d374a 1281regular_file2 (const char *path, st_parameter_open *opp, unit_flags *flags)
4ee9c684 1282{
4ee9c684 1283 int mode;
6d12c489 1284 int rwflag;
327beda5 1285 int crflag, crflag2;
6d12c489 1286 int fd;
4ee9c684 1287
3d17984e 1288#ifdef __CYGWIN__
1289 if (opp->file_len == 7)
1290 {
1291 if (strncmp (path, "CONOUT$", 7) == 0
1292 || strncmp (path, "CONERR$", 7) == 0)
1293 {
1294 fd = open ("/dev/conout", O_WRONLY);
1295 flags->action = ACTION_WRITE;
1296 return fd;
1297 }
1298 }
1299
1300 if (opp->file_len == 6 && strncmp (path, "CONIN$", 6) == 0)
1301 {
1302 fd = open ("/dev/conin", O_RDONLY);
1303 flags->action = ACTION_READ;
1304 return fd;
1305 }
1306#endif
1307
2058bc7e 1308
1309#ifdef __MINGW32__
1310 if (opp->file_len == 7)
1311 {
1312 if (strncmp (path, "CONOUT$", 7) == 0
1313 || strncmp (path, "CONERR$", 7) == 0)
1314 {
1315 fd = open ("CONOUT$", O_WRONLY);
1316 flags->action = ACTION_WRITE;
1317 return fd;
1318 }
1319 }
1320
1321 if (opp->file_len == 6 && strncmp (path, "CONIN$", 6) == 0)
1322 {
1323 fd = open ("CONIN$", O_RDONLY);
1324 flags->action = ACTION_READ;
1325 return fd;
1326 }
1327#endif
1328
6d12c489 1329 switch (flags->action)
4ee9c684 1330 {
1331 case ACTION_READ:
6d12c489 1332 rwflag = O_RDONLY;
4ee9c684 1333 break;
1334
1335 case ACTION_WRITE:
6d12c489 1336 rwflag = O_WRONLY;
4ee9c684 1337 break;
1338
1339 case ACTION_READWRITE:
6d12c489 1340 case ACTION_UNSPECIFIED:
1341 rwflag = O_RDWR;
4ee9c684 1342 break;
1343
1344 default:
60c514ba 1345 internal_error (&opp->common, "regular_file(): Bad action");
4ee9c684 1346 }
1347
6d12c489 1348 switch (flags->status)
4ee9c684 1349 {
1350 case STATUS_NEW:
2d6ba0f9 1351 crflag = O_CREAT | O_EXCL;
4ee9c684 1352 break;
1353
2d6ba0f9 1354 case STATUS_OLD: /* open will fail if the file does not exist*/
1355 crflag = 0;
4ee9c684 1356 break;
1357
1358 case STATUS_UNKNOWN:
327beda5 1359 if (rwflag == O_RDONLY)
1360 crflag = 0;
1361 else
1362 crflag = O_CREAT;
4ee9c684 1363 break;
1364
1365 case STATUS_REPLACE:
a638be8f 1366 crflag = O_CREAT | O_TRUNC;
4ee9c684 1367 break;
1368
1369 default:
327beda5 1370 /* Note: STATUS_SCRATCH is handled by tempfile () and should
1371 never be seen here. */
60c514ba 1372 internal_error (&opp->common, "regular_file(): Bad status");
4ee9c684 1373 }
1374
6d12c489 1375 /* rwflag |= O_LARGEFILE; */
4ee9c684 1376
cf6a3896 1377#if defined(HAVE_CRLF) && defined(O_BINARY)
cf55c3cf 1378 crflag |= O_BINARY;
1379#endif
1380
544db64f 1381#ifdef O_CLOEXEC
1382 crflag |= O_CLOEXEC;
1383#endif
1384
6d12c489 1385 mode = S_IRUSR | S_IWUSR | S_IRGRP | S_IWGRP | S_IROTH | S_IWOTH;
714f1fe2 1386 TEMP_FAILURE_RETRY (fd = open (path, rwflag | crflag, mode));
2d6ba0f9 1387 if (flags->action != ACTION_UNSPECIFIED)
a638be8f 1388 return fd;
2d6ba0f9 1389
1390 if (fd >= 0)
6d12c489 1391 {
2d6ba0f9 1392 flags->action = ACTION_READWRITE;
1393 return fd;
6d12c489 1394 }
771e2b1c 1395 if (errno != EACCES && errno != EPERM && errno != EROFS)
2d6ba0f9 1396 return fd;
1397
1398 /* retry for read-only access */
1399 rwflag = O_RDONLY;
327beda5 1400 if (flags->status == STATUS_UNKNOWN)
1401 crflag2 = crflag & ~(O_CREAT);
1402 else
1403 crflag2 = crflag;
714f1fe2 1404 TEMP_FAILURE_RETRY (fd = open (path, rwflag | crflag2, mode));
2d6ba0f9 1405 if (fd >=0)
1406 {
1407 flags->action = ACTION_READ;
84d33b91 1408 return fd; /* success */
2d6ba0f9 1409 }
1410
771e2b1c 1411 if (errno != EACCES && errno != EPERM && errno != ENOENT)
84d33b91 1412 return fd; /* failure */
2d6ba0f9 1413
1414 /* retry for write-only access */
1415 rwflag = O_WRONLY;
714f1fe2 1416 TEMP_FAILURE_RETRY (fd = open (path, rwflag | crflag, mode));
2d6ba0f9 1417 if (fd >=0)
1418 {
1419 flags->action = ACTION_WRITE;
84d33b91 1420 return fd; /* success */
2d6ba0f9 1421 }
84d33b91 1422 return fd; /* failure */
4ee9c684 1423}
1424
1425
b3db57e8 1426/* Lock the file, if necessary, based on SHARE flags. */
1427
1428#if defined(HAVE_FCNTL) && defined(F_SETLK) && defined(F_UNLCK)
1429static int
1430open_share (st_parameter_open *opp, int fd, unit_flags *flags)
1431{
1432 int r = 0;
1433 struct flock f;
1434 if (fd == STDOUT_FILENO || fd == STDERR_FILENO || fd == STDIN_FILENO)
1435 return 0;
1436
1437 f.l_start = 0;
1438 f.l_len = 0;
1439 f.l_whence = SEEK_SET;
1440
1441 switch (flags->share)
1442 {
1443 case SHARE_DENYNONE:
1444 f.l_type = F_RDLCK;
1445 r = fcntl (fd, F_SETLK, &f);
1446 break;
1447 case SHARE_DENYRW:
1448 /* Must be writable to hold write lock. */
1449 if (flags->action == ACTION_READ)
1450 {
1451 generate_error (&opp->common, LIBERROR_BAD_ACTION,
1452 "Cannot set write lock on file opened for READ");
1453 return -1;
1454 }
1455 f.l_type = F_WRLCK;
1456 r = fcntl (fd, F_SETLK, &f);
1457 break;
1458 case SHARE_UNSPECIFIED:
1459 default:
1460 break;
1461 }
1462
1463 return r;
1464}
1465#else
1466static int
1467open_share (st_parameter_open *opp __attribute__ ((unused)),
1468 int fd __attribute__ ((unused)),
1469 unit_flags *flags __attribute__ ((unused)))
1470{
1471 return 0;
1472}
1473#endif /* defined(HAVE_FCNTL) ... */
1474
1475
fc3d374a 1476/* Wrapper around regular_file2, to make sure we free the path after
1477 we're done. */
1478
1479static int
1480regular_file (st_parameter_open *opp, unit_flags *flags)
1481{
1482 char *path = fc_strdup (opp->file, opp->file_len);
1483 int fd = regular_file2 (path, opp, flags);
1484 free (path);
1485 return fd;
1486}
1487
4ee9c684 1488/* open_external()-- Open an external file, unix specific version.
25a5ce27 1489 Change flags->action if it is ACTION_UNSPECIFIED on entry.
1490 Returns NULL on operating system error. */
4ee9c684 1491
1492stream *
60c514ba 1493open_external (st_parameter_open *opp, unit_flags *flags)
4ee9c684 1494{
aadbd4ae 1495 int fd;
4ee9c684 1496
6d12c489 1497 if (flags->status == STATUS_SCRATCH)
1498 {
60c514ba 1499 fd = tempfile (opp);
6d12c489 1500 if (flags->action == ACTION_UNSPECIFIED)
b3db57e8 1501 flags->action = flags->readonly ? ACTION_READ : ACTION_READWRITE;
1dc95e51 1502
1503#if HAVE_UNLINK_OPEN_FILE
6d12c489 1504 /* We can unlink scratch files now and it will go away when closed. */
60c514ba 1505 if (fd >= 0)
1506 unlink (opp->file);
1dc95e51 1507#endif
6d12c489 1508 }
1509 else
1510 {
2d6ba0f9 1511 /* regular_file resets flags->action if it is ACTION_UNSPECIFIED and
25a5ce27 1512 if it succeeds */
60c514ba 1513 fd = regular_file (opp, flags);
544db64f 1514#ifndef O_CLOEXEC
1515 set_close_on_exec (fd);
1516#endif
6d12c489 1517 }
4ee9c684 1518
1519 if (fd < 0)
1520 return NULL;
1521 fd = fix_fd (fd);
1522
b3db57e8 1523 if (open_share (opp, fd, flags) < 0)
1524 return NULL;
1525
fe34985d 1526 return fd_to_stream (fd, flags->form == FORM_UNFORMATTED);
4ee9c684 1527}
1528
1529
1530/* input_stream()-- Return a stream pointer to the default input stream.
25a5ce27 1531 Called on initialization. */
4ee9c684 1532
1533stream *
1534input_stream (void)
1535{
fe34985d 1536 return fd_to_stream (STDIN_FILENO, false);
4ee9c684 1537}
1538
1539
ff81ee3b 1540/* output_stream()-- Return a stream pointer to the default output stream.
25a5ce27 1541 Called on initialization. */
4ee9c684 1542
1543stream *
1544output_stream (void)
1545{
25a5ce27 1546 stream *s;
3e45a719 1547
e693d7f1 1548#if defined(HAVE_CRLF) && defined(HAVE_SETMODE)
1549 setmode (STDOUT_FILENO, O_BINARY);
1550#endif
3e45a719 1551
fe34985d 1552 s = fd_to_stream (STDOUT_FILENO, false);
3e45a719 1553 return s;
4ee9c684 1554}
1555
1556
ff81ee3b 1557/* error_stream()-- Return a stream pointer to the default error stream.
25a5ce27 1558 Called on initialization. */
ff81ee3b 1559
1560stream *
1561error_stream (void)
1562{
25a5ce27 1563 stream *s;
3e45a719 1564
e693d7f1 1565#if defined(HAVE_CRLF) && defined(HAVE_SETMODE)
1566 setmode (STDERR_FILENO, O_BINARY);
1567#endif
3e45a719 1568
fe34985d 1569 s = fd_to_stream (STDERR_FILENO, false);
3e45a719 1570 return s;
ff81ee3b 1571}
1572
4ee9c684 1573
4ee9c684 1574/* compare_file_filename()-- Given an open stream and a fortran string
25a5ce27 1575 that is a filename, figure out if the file is the same as the
1576 filename. */
4ee9c684 1577
1578int
daad4fd5 1579compare_file_filename (gfc_unit *u, const char *name, int len)
4ee9c684 1580{
f6854450 1581 struct stat st;
fc3d374a 1582 int ret;
daad4fd5 1583#ifdef HAVE_WORKING_STAT
01cd2c93 1584 unix_stream *s;
c0ecd33c 1585#else
1586# ifdef __MINGW32__
1587 uint64_t id1, id2;
1588# endif
daad4fd5 1589#endif
4ee9c684 1590
fc3d374a 1591 char *path = fc_strdup (name, len);
4ee9c684 1592
1593 /* If the filename doesn't exist, then there is no match with the
25a5ce27 1594 existing file. */
4ee9c684 1595
714f1fe2 1596 if (TEMP_FAILURE_RETRY (stat (path, &st)) < 0)
fc3d374a 1597 {
1598 ret = 0;
1599 goto done;
1600 }
4ee9c684 1601
daad4fd5 1602#ifdef HAVE_WORKING_STAT
01cd2c93 1603 s = (unix_stream *) (u->s);
fc3d374a 1604 ret = (st.st_dev == s->st_dev) && (st.st_ino == s->st_ino);
1605 goto done;
daad4fd5 1606#else
c0ecd33c 1607
1608# ifdef __MINGW32__
1609 /* We try to match files by a unique ID. On some filesystems (network
1610 fs and FAT), we can't generate this unique ID, and will simply compare
1611 filenames. */
1612 id1 = id_from_path (path);
1613 id2 = id_from_fd (((unix_stream *) (u->s))->fd);
1614 if (id1 || id2)
fc3d374a 1615 {
1616 ret = (id1 == id2);
1617 goto done;
1618 }
c0ecd33c 1619# endif
98044b92 1620 if (u->filename)
1621 ret = (strcmp(path, u->filename) == 0);
1622 else
1623 ret = 0;
daad4fd5 1624#endif
fc3d374a 1625 done:
1626 free (path);
1627 return ret;
4ee9c684 1628}
1629
1630
60c514ba 1631#ifdef HAVE_WORKING_STAT
f6854450 1632# define FIND_FILE0_DECL struct stat *st
60c514ba 1633# define FIND_FILE0_ARGS st
1634#else
8d832ee4 1635# define FIND_FILE0_DECL uint64_t id, const char *path
1636# define FIND_FILE0_ARGS id, path
60c514ba 1637#endif
1638
4ee9c684 1639/* find_file0()-- Recursive work function for find_file() */
1640
f02dd226 1641static gfc_unit *
60c514ba 1642find_file0 (gfc_unit *u, FIND_FILE0_DECL)
4ee9c684 1643{
f02dd226 1644 gfc_unit *v;
c0ecd33c 1645#if defined(__MINGW32__) && !HAVE_WORKING_STAT
1646 uint64_t id1;
1647#endif
4ee9c684 1648
1649 if (u == NULL)
1650 return NULL;
1651
daad4fd5 1652#ifdef HAVE_WORKING_STAT
01cd2c93 1653 if (u->s != NULL)
1654 {
1655 unix_stream *s = (unix_stream *) (u->s);
1656 if (st[0].st_dev == s->st_dev && st[0].st_ino == s->st_ino)
1657 return u;
1658 }
daad4fd5 1659#else
c0ecd33c 1660# ifdef __MINGW32__
1661 if (u->s && ((id1 = id_from_fd (((unix_stream *) u->s)->fd)) || id1))
1662 {
1663 if (id == id1)
1664 return u;
1665 }
1666 else
1667# endif
98044b92 1668 if (u->filename && strcmp (u->filename, path) == 0)
c0ecd33c 1669 return u;
daad4fd5 1670#endif
4ee9c684 1671
60c514ba 1672 v = find_file0 (u->left, FIND_FILE0_ARGS);
4ee9c684 1673 if (v != NULL)
1674 return v;
1675
60c514ba 1676 v = find_file0 (u->right, FIND_FILE0_ARGS);
4ee9c684 1677 if (v != NULL)
1678 return v;
1679
1680 return NULL;
1681}
1682
1683
1684/* find_file()-- Take the current filename and see if there is a unit
25a5ce27 1685 that has the file already open. Returns a pointer to the unit if so. */
4ee9c684 1686
f02dd226 1687gfc_unit *
60c514ba 1688find_file (const char *file, gfc_charlen_type file_len)
4ee9c684 1689{
f6854450 1690 struct stat st[1];
60c514ba 1691 gfc_unit *u;
c12dc7d7 1692#if defined(__MINGW32__) && !HAVE_WORKING_STAT
1693 uint64_t id = 0ULL;
1694#endif
4ee9c684 1695
fc3d374a 1696 char *path = fc_strdup (file, file_len);
4ee9c684 1697
714f1fe2 1698 if (TEMP_FAILURE_RETRY (stat (path, &st[0])) < 0)
fc3d374a 1699 {
1700 u = NULL;
1701 goto done;
1702 }
4ee9c684 1703
c0ecd33c 1704#if defined(__MINGW32__) && !HAVE_WORKING_STAT
c12dc7d7 1705 id = id_from_path (path);
c0ecd33c 1706#endif
1707
60c514ba 1708 __gthread_mutex_lock (&unit_lock);
1709retry:
1710 u = find_file0 (unit_root, FIND_FILE0_ARGS);
1711 if (u != NULL)
1712 {
1713 /* Fast path. */
1714 if (! __gthread_mutex_trylock (&u->lock))
1715 {
1716 /* assert (u->closed == 0); */
1717 __gthread_mutex_unlock (&unit_lock);
fc3d374a 1718 goto done;
60c514ba 1719 }
1720
1721 inc_waiting_locked (u);
1722 }
1723 __gthread_mutex_unlock (&unit_lock);
1724 if (u != NULL)
1725 {
1726 __gthread_mutex_lock (&u->lock);
1727 if (u->closed)
1728 {
1729 __gthread_mutex_lock (&unit_lock);
1730 __gthread_mutex_unlock (&u->lock);
1731 if (predec_waiting_locked (u) == 0)
5e62a3cc 1732 free (u);
60c514ba 1733 goto retry;
1734 }
1735
1736 dec_waiting_unlocked (u);
1737 }
fc3d374a 1738 done:
1739 free (path);
60c514ba 1740 return u;
1741}
1742
1743static gfc_unit *
1744flush_all_units_1 (gfc_unit *u, int min_unit)
1745{
1746 while (u != NULL)
1747 {
1748 if (u->unit_number > min_unit)
1749 {
1750 gfc_unit *r = flush_all_units_1 (u->left, min_unit);
1751 if (r != NULL)
1752 return r;
1753 }
1754 if (u->unit_number >= min_unit)
1755 {
1756 if (__gthread_mutex_trylock (&u->lock))
1757 return u;
1758 if (u->s)
65f15010 1759 sflush (u->s);
60c514ba 1760 __gthread_mutex_unlock (&u->lock);
1761 }
1762 u = u->right;
1763 }
1764 return NULL;
1765}
1766
1767void
1768flush_all_units (void)
1769{
1770 gfc_unit *u;
1771 int min_unit = 0;
1772
1773 __gthread_mutex_lock (&unit_lock);
1774 do
1775 {
1776 u = flush_all_units_1 (unit_root, min_unit);
1777 if (u != NULL)
1778 inc_waiting_locked (u);
1779 __gthread_mutex_unlock (&unit_lock);
1780 if (u == NULL)
1781 return;
1782
1783 __gthread_mutex_lock (&u->lock);
1784
1785 min_unit = u->unit_number + 1;
1786
1787 if (u->closed == 0)
1788 {
65f15010 1789 sflush (u->s);
60c514ba 1790 __gthread_mutex_lock (&unit_lock);
1791 __gthread_mutex_unlock (&u->lock);
1792 (void) predec_waiting_locked (u);
1793 }
1794 else
1795 {
1796 __gthread_mutex_lock (&unit_lock);
1797 __gthread_mutex_unlock (&u->lock);
1798 if (predec_waiting_locked (u) == 0)
5e62a3cc 1799 free (u);
60c514ba 1800 }
1801 }
1802 while (1);
4ee9c684 1803}
1804
1805
b3db57e8 1806/* Unlock the unit if necessary, based on SHARE flags. */
1807
1808int
1809close_share (gfc_unit *u __attribute__ ((unused)))
1810{
1811 int r = 0;
1812#if defined(HAVE_FCNTL) && defined(F_SETLK) && defined(F_UNLCK)
1813 unix_stream *s = (unix_stream *) u->s;
1814 int fd = s->fd;
1815 struct flock f;
1816
1817 switch (u->flags.share)
1818 {
1819 case SHARE_DENYRW:
1820 case SHARE_DENYNONE:
1821 if (fd != STDOUT_FILENO && fd != STDERR_FILENO && fd != STDIN_FILENO)
1822 {
1823 f.l_start = 0;
1824 f.l_len = 0;
1825 f.l_whence = SEEK_SET;
1826 f.l_type = F_UNLCK;
1827 r = fcntl (fd, F_SETLK, &f);
1828 }
1829 break;
1830 case SHARE_UNSPECIFIED:
1831 default:
1832 break;
1833 }
1834
1835#endif
1836 return r;
1837}
1838
1839
4ee9c684 1840/* file_exists()-- Returns nonzero if the current filename exists on
25a5ce27 1841 the system */
4ee9c684 1842
1843int
60c514ba 1844file_exists (const char *file, gfc_charlen_type file_len)
4ee9c684 1845{
fc3d374a 1846 char *path = fc_strdup (file, file_len);
1847 int res = !(access (path, F_OK));
1848 free (path);
1849 return res;
4ee9c684 1850}
1851
1852
f4e9c676 1853/* file_size()-- Returns the size of the file. */
1854
1855GFC_IO_INT
1856file_size (const char *file, gfc_charlen_type file_len)
1857{
fc3d374a 1858 char *path = fc_strdup (file, file_len);
f6854450 1859 struct stat statbuf;
714f1fe2 1860 int err;
1861 TEMP_FAILURE_RETRY (err = stat (path, &statbuf));
fc3d374a 1862 free (path);
1863 if (err == -1)
f4e9c676 1864 return -1;
f4e9c676 1865 return (GFC_IO_INT) statbuf.st_size;
1866}
4ee9c684 1867
fb35179a 1868static const char yes[] = "YES", no[] = "NO", unknown[] = "UNKNOWN";
4ee9c684 1869
1870/* inquire_sequential()-- Given a fortran string, determine if the
25a5ce27 1871 file is suitable for sequential access. Returns a C-style
1872 string. */
4ee9c684 1873
1874const char *
1875inquire_sequential (const char *string, int len)
1876{
f6854450 1877 struct stat statbuf;
4ee9c684 1878
fc3d374a 1879 if (string == NULL)
1880 return unknown;
1881
1882 char *path = fc_strdup (string, len);
714f1fe2 1883 int err;
1884 TEMP_FAILURE_RETRY (err = stat (path, &statbuf));
fc3d374a 1885 free (path);
1886 if (err == -1)
4ee9c684 1887 return unknown;
1888
1889 if (S_ISREG (statbuf.st_mode) ||
1890 S_ISCHR (statbuf.st_mode) || S_ISFIFO (statbuf.st_mode))
2e1fa727 1891 return unknown;
4ee9c684 1892
1893 if (S_ISDIR (statbuf.st_mode) || S_ISBLK (statbuf.st_mode))
1894 return no;
1895
1896 return unknown;
1897}
1898
1899
1900/* inquire_direct()-- Given a fortran string, determine if the file is
25a5ce27 1901 suitable for direct access. Returns a C-style string. */
4ee9c684 1902
1903const char *
1904inquire_direct (const char *string, int len)
1905{
f6854450 1906 struct stat statbuf;
4ee9c684 1907
fc3d374a 1908 if (string == NULL)
1909 return unknown;
1910
1911 char *path = fc_strdup (string, len);
714f1fe2 1912 int err;
1913 TEMP_FAILURE_RETRY (err = stat (path, &statbuf));
fc3d374a 1914 free (path);
1915 if (err == -1)
4ee9c684 1916 return unknown;
1917
1918 if (S_ISREG (statbuf.st_mode) || S_ISBLK (statbuf.st_mode))
2e1fa727 1919 return unknown;
4ee9c684 1920
1921 if (S_ISDIR (statbuf.st_mode) ||
1922 S_ISCHR (statbuf.st_mode) || S_ISFIFO (statbuf.st_mode))
1923 return no;
1924
1925 return unknown;
1926}
1927
1928
1929/* inquire_formatted()-- Given a fortran string, determine if the file
25a5ce27 1930 is suitable for formatted form. Returns a C-style string. */
4ee9c684 1931
1932const char *
1933inquire_formatted (const char *string, int len)
1934{
f6854450 1935 struct stat statbuf;
4ee9c684 1936
fc3d374a 1937 if (string == NULL)
1938 return unknown;
1939
1940 char *path = fc_strdup (string, len);
714f1fe2 1941 int err;
1942 TEMP_FAILURE_RETRY (err = stat (path, &statbuf));
fc3d374a 1943 free (path);
1944 if (err == -1)
4ee9c684 1945 return unknown;
1946
1947 if (S_ISREG (statbuf.st_mode) ||
1948 S_ISBLK (statbuf.st_mode) ||
1949 S_ISCHR (statbuf.st_mode) || S_ISFIFO (statbuf.st_mode))
2e1fa727 1950 return unknown;
4ee9c684 1951
1952 if (S_ISDIR (statbuf.st_mode))
1953 return no;
1954
1955 return unknown;
1956}
1957
1958
1959/* inquire_unformatted()-- Given a fortran string, determine if the file
25a5ce27 1960 is suitable for unformatted form. Returns a C-style string. */
4ee9c684 1961
1962const char *
1963inquire_unformatted (const char *string, int len)
1964{
4ee9c684 1965 return inquire_formatted (string, len);
1966}
1967
1968
1969/* inquire_access()-- Given a fortran string, determine if the file is
25a5ce27 1970 suitable for access. */
4ee9c684 1971
1972static const char *
1973inquire_access (const char *string, int len, int mode)
1974{
fc3d374a 1975 if (string == NULL)
1976 return no;
1977 char *path = fc_strdup (string, len);
1978 int res = access (path, mode);
1979 free (path);
1980 if (res == -1)
4ee9c684 1981 return no;
1982
1983 return yes;
1984}
1985
1986
1987/* inquire_read()-- Given a fortran string, determine if the file is
25a5ce27 1988 suitable for READ access. */
4ee9c684 1989
1990const char *
1991inquire_read (const char *string, int len)
1992{
4ee9c684 1993 return inquire_access (string, len, R_OK);
1994}
1995
1996
1997/* inquire_write()-- Given a fortran string, determine if the file is
25a5ce27 1998 suitable for READ access. */
4ee9c684 1999
2000const char *
2001inquire_write (const char *string, int len)
2002{
4ee9c684 2003 return inquire_access (string, len, W_OK);
2004}
2005
2006
2007/* inquire_readwrite()-- Given a fortran string, determine if the file is
25a5ce27 2008 suitable for read and write access. */
4ee9c684 2009
2010const char *
2011inquire_readwrite (const char *string, int len)
2012{
4ee9c684 2013 return inquire_access (string, len, R_OK | W_OK);
2014}
2015
2016
60d77e0d 2017int
2018stream_isatty (stream *s)
2019{
2020 return isatty (((unix_stream *) s)->fd);
2021}
2022
57f34837 2023int
2024stream_ttyname (stream *s __attribute__ ((unused)),
25a5ce27 2025 char *buf __attribute__ ((unused)),
57f34837 2026 size_t buflen __attribute__ ((unused)))
2027{
2028#ifdef HAVE_TTYNAME_R
25a5ce27 2029 return ttyname_r (((unix_stream *)s)->fd, buf, buflen);
57f34837 2030#elif defined HAVE_TTYNAME
2031 char *p;
2032 size_t plen;
25a5ce27 2033 p = ttyname (((unix_stream *)s)->fd);
57f34837 2034 if (!p)
2035 return errno;
2036 plen = strlen (p);
2037 if (buflen < plen)
2038 plen = buflen;
2039 memcpy (buf, p, plen);
2040 return 0;
f2c0a16d 2041#else
57f34837 2042 return ENOSYS;
3479b863 2043#endif
57f34837 2044}
2045
3479b863 2046
60d77e0d 2047
4ee9c684 2048
2049/* How files are stored: This is an operating-system specific issue,
2050 and therefore belongs here. There are three cases to consider.
2051
2052 Direct Access:
2053 Records are written as block of bytes corresponding to the record
2054 length of the file. This goes for both formatted and unformatted
2055 records. Positioning is done explicitly for each data transfer,
2056 so positioning is not much of an issue.
2057
2058 Sequential Formatted:
2059 Records are separated by newline characters. The newline character
2060 is prohibited from appearing in a string. If it does, this will be
2061 messed up on the next read. End of file is also the end of a record.
2062
2063 Sequential Unformatted:
2064 In this case, we are merely copying bytes to and from main storage,
2065 yet we need to keep track of varying record lengths. We adopt
2066 the solution used by f2c. Each record contains a pair of length
2067 markers:
2068
84d33b91 2069 Length of record n in bytes
2070 Data of record n
2071 Length of record n in bytes
4ee9c684 2072
84d33b91 2073 Length of record n+1 in bytes
2074 Data of record n+1
2075 Length of record n+1 in bytes
4ee9c684 2076
2077 The length is stored at the end of a record to allow backspacing to the
2078 previous record. Between data transfer statements, the file pointer
2079 is left pointing to the first length of the current record.
2080
2081 ENDFILE records are never explicitly stored.
2082
2083*/