2018-25-01 Paul Thomas <pault@gcc.gnu.org>
[official-gcc.git] / libgfortran / io / transfer.c
blob8bc828c02146f8183013627347bdc62c83f04c49
1 /* Copyright (C) 2002-2018 Free Software Foundation, Inc.
2 Contributed by Andy Vaught
3 Namelist transfer functions contributed by Paul Thomas
4 F2003 I/O support contributed by Jerry DeLisle
6 This file is part of the GNU Fortran runtime library (libgfortran).
8 Libgfortran is free software; you can redistribute it and/or modify
9 it under the terms of the GNU General Public License as published by
10 the Free Software Foundation; either version 3, or (at your option)
11 any later version.
13 Libgfortran is distributed in the hope that it will be useful,
14 but WITHOUT ANY WARRANTY; without even the implied warranty of
15 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
16 GNU General Public License for more details.
18 Under Section 7 of GPL version 3, you are granted additional
19 permissions described in the GCC Runtime Library Exception, version
20 3.1, as published by the Free Software Foundation.
22 You should have received a copy of the GNU General Public License and
23 a copy of the GCC Runtime Library Exception along with this program;
24 see the files COPYING3 and COPYING.RUNTIME respectively. If not, see
25 <http://www.gnu.org/licenses/>. */
28 /* transfer.c -- Top level handling of data transfer statements. */
30 #include "io.h"
31 #include "fbuf.h"
32 #include "format.h"
33 #include "unix.h"
34 #include <string.h>
35 #include <errno.h>
38 /* Calling conventions: Data transfer statements are unlike other
39 library calls in that they extend over several calls.
41 The first call is always a call to st_read() or st_write(). These
42 subroutines return no status unless a namelist read or write is
43 being done, in which case there is the usual status. No further
44 calls are necessary in this case.
46 For other sorts of data transfer, there are zero or more data
47 transfer statement that depend on the format of the data transfer
48 statement. For READ (and for backwards compatibily: for WRITE), one has
50 transfer_integer
51 transfer_logical
52 transfer_character
53 transfer_character_wide
54 transfer_real
55 transfer_complex
56 transfer_real128
57 transfer_complex128
59 and for WRITE
61 transfer_integer_write
62 transfer_logical_write
63 transfer_character_write
64 transfer_character_wide_write
65 transfer_real_write
66 transfer_complex_write
67 transfer_real128_write
68 transfer_complex128_write
70 These subroutines do not return status. The *128 functions
71 are in the file transfer128.c.
73 The last call is a call to st_[read|write]_done(). While
74 something can easily go wrong with the initial st_read() or
75 st_write(), an error inhibits any data from actually being
76 transferred. */
78 extern void transfer_integer (st_parameter_dt *, void *, int);
79 export_proto(transfer_integer);
81 extern void transfer_integer_write (st_parameter_dt *, void *, int);
82 export_proto(transfer_integer_write);
84 extern void transfer_real (st_parameter_dt *, void *, int);
85 export_proto(transfer_real);
87 extern void transfer_real_write (st_parameter_dt *, void *, int);
88 export_proto(transfer_real_write);
90 extern void transfer_logical (st_parameter_dt *, void *, int);
91 export_proto(transfer_logical);
93 extern void transfer_logical_write (st_parameter_dt *, void *, int);
94 export_proto(transfer_logical_write);
96 extern void transfer_character (st_parameter_dt *, void *, gfc_charlen_type);
97 export_proto(transfer_character);
99 extern void transfer_character_write (st_parameter_dt *, void *, gfc_charlen_type);
100 export_proto(transfer_character_write);
102 extern void transfer_character_wide (st_parameter_dt *, void *, gfc_charlen_type, int);
103 export_proto(transfer_character_wide);
105 extern void transfer_character_wide_write (st_parameter_dt *,
106 void *, gfc_charlen_type, int);
107 export_proto(transfer_character_wide_write);
109 extern void transfer_complex (st_parameter_dt *, void *, int);
110 export_proto(transfer_complex);
112 extern void transfer_complex_write (st_parameter_dt *, void *, int);
113 export_proto(transfer_complex_write);
115 extern void transfer_array (st_parameter_dt *, gfc_array_char *, int,
116 gfc_charlen_type);
117 export_proto(transfer_array);
119 extern void transfer_array_write (st_parameter_dt *, gfc_array_char *, int,
120 gfc_charlen_type);
121 export_proto(transfer_array_write);
123 /* User defined derived type input/output. */
124 extern void
125 transfer_derived (st_parameter_dt *dtp, void *dtio_source, void *dtio_proc);
126 export_proto(transfer_derived);
128 extern void
129 transfer_derived_write (st_parameter_dt *dtp, void *dtio_source, void *dtio_proc);
130 export_proto(transfer_derived_write);
132 static void us_read (st_parameter_dt *, int);
133 static void us_write (st_parameter_dt *, int);
134 static void next_record_r_unf (st_parameter_dt *, int);
135 static void next_record_w_unf (st_parameter_dt *, int);
137 static const st_option advance_opt[] = {
138 {"yes", ADVANCE_YES},
139 {"no", ADVANCE_NO},
140 {NULL, 0}
144 static const st_option decimal_opt[] = {
145 {"point", DECIMAL_POINT},
146 {"comma", DECIMAL_COMMA},
147 {NULL, 0}
150 static const st_option round_opt[] = {
151 {"up", ROUND_UP},
152 {"down", ROUND_DOWN},
153 {"zero", ROUND_ZERO},
154 {"nearest", ROUND_NEAREST},
155 {"compatible", ROUND_COMPATIBLE},
156 {"processor_defined", ROUND_PROCDEFINED},
157 {NULL, 0}
161 static const st_option sign_opt[] = {
162 {"plus", SIGN_SP},
163 {"suppress", SIGN_SS},
164 {"processor_defined", SIGN_S},
165 {NULL, 0}
168 static const st_option blank_opt[] = {
169 {"null", BLANK_NULL},
170 {"zero", BLANK_ZERO},
171 {NULL, 0}
174 static const st_option delim_opt[] = {
175 {"apostrophe", DELIM_APOSTROPHE},
176 {"quote", DELIM_QUOTE},
177 {"none", DELIM_NONE},
178 {NULL, 0}
181 static const st_option pad_opt[] = {
182 {"yes", PAD_YES},
183 {"no", PAD_NO},
184 {NULL, 0}
187 typedef enum
188 { FORMATTED_SEQUENTIAL, UNFORMATTED_SEQUENTIAL,
189 FORMATTED_DIRECT, UNFORMATTED_DIRECT, FORMATTED_STREAM, UNFORMATTED_STREAM
191 file_mode;
194 static file_mode
195 current_mode (st_parameter_dt *dtp)
197 file_mode m;
199 m = FORM_UNSPECIFIED;
201 if (dtp->u.p.current_unit->flags.access == ACCESS_DIRECT)
203 m = dtp->u.p.current_unit->flags.form == FORM_FORMATTED ?
204 FORMATTED_DIRECT : UNFORMATTED_DIRECT;
206 else if (dtp->u.p.current_unit->flags.access == ACCESS_SEQUENTIAL)
208 m = dtp->u.p.current_unit->flags.form == FORM_FORMATTED ?
209 FORMATTED_SEQUENTIAL : UNFORMATTED_SEQUENTIAL;
211 else if (dtp->u.p.current_unit->flags.access == ACCESS_STREAM)
213 m = dtp->u.p.current_unit->flags.form == FORM_FORMATTED ?
214 FORMATTED_STREAM : UNFORMATTED_STREAM;
217 return m;
221 /* Mid level data transfer statements. */
223 /* Read sequential file - internal unit */
225 static char *
226 read_sf_internal (st_parameter_dt *dtp, size_t *length)
228 static char *empty_string[0];
229 char *base = NULL;
230 size_t lorig;
232 /* Zero size array gives internal unit len of 0. Nothing to read. */
233 if (dtp->internal_unit_len == 0
234 && dtp->u.p.current_unit->pad_status == PAD_NO)
235 hit_eof (dtp);
237 /* If we have seen an eor previously, return a length of 0. The
238 caller is responsible for correctly padding the input field. */
239 if (dtp->u.p.sf_seen_eor)
241 *length = 0;
242 /* Just return something that isn't a NULL pointer, otherwise the
243 caller thinks an error occurred. */
244 return (char*) empty_string;
247 /* There are some cases with mixed DTIO where we have read a character
248 and saved it in the last character buffer, so we need to backup. */
249 if (unlikely (dtp->u.p.current_unit->child_dtio > 0 &&
250 dtp->u.p.current_unit->last_char != EOF - 1))
252 dtp->u.p.current_unit->last_char = EOF - 1;
253 sseek (dtp->u.p.current_unit->s, -1, SEEK_CUR);
256 lorig = *length;
257 if (is_char4_unit(dtp))
259 gfc_char4_t *p = (gfc_char4_t *) mem_alloc_r4 (dtp->u.p.current_unit->s,
260 length);
261 base = fbuf_alloc (dtp->u.p.current_unit, lorig);
262 for (size_t i = 0; i < *length; i++, p++)
263 base[i] = *p > 255 ? '?' : (unsigned char) *p;
265 else
266 base = mem_alloc_r (dtp->u.p.current_unit->s, length);
268 if (unlikely (lorig > *length))
270 hit_eof (dtp);
271 return NULL;
274 dtp->u.p.current_unit->bytes_left -= *length;
276 if (((dtp->common.flags & IOPARM_DT_HAS_SIZE) != 0) ||
277 dtp->u.p.current_unit->has_size)
278 dtp->u.p.current_unit->size_used += (GFC_IO_INT) *length;
280 return base;
284 /* When reading sequential formatted records we have a problem. We
285 don't know how long the line is until we read the trailing newline,
286 and we don't want to read too much. If we read too much, we might
287 have to do a physical seek backwards depending on how much data is
288 present, and devices like terminals aren't seekable and would cause
289 an I/O error.
291 Given this, the solution is to read a byte at a time, stopping if
292 we hit the newline. For small allocations, we use a static buffer.
293 For larger allocations, we are forced to allocate memory on the
294 heap. Hopefully this won't happen very often. */
296 /* Read sequential file - external unit */
298 static char *
299 read_sf (st_parameter_dt *dtp, size_t *length)
301 static char *empty_string[0];
302 size_t lorig, n;
303 int q, q2;
304 int seen_comma;
306 /* If we have seen an eor previously, return a length of 0. The
307 caller is responsible for correctly padding the input field. */
308 if (dtp->u.p.sf_seen_eor)
310 *length = 0;
311 /* Just return something that isn't a NULL pointer, otherwise the
312 caller thinks an error occurred. */
313 return (char*) empty_string;
316 /* There are some cases with mixed DTIO where we have read a character
317 and saved it in the last character buffer, so we need to backup. */
318 if (unlikely (dtp->u.p.current_unit->child_dtio > 0 &&
319 dtp->u.p.current_unit->last_char != EOF - 1))
321 dtp->u.p.current_unit->last_char = EOF - 1;
322 fbuf_seek (dtp->u.p.current_unit, -1, SEEK_CUR);
325 n = seen_comma = 0;
327 /* Read data into format buffer and scan through it. */
328 lorig = *length;
330 while (n < *length)
332 q = fbuf_getc (dtp->u.p.current_unit);
333 if (q == EOF)
334 break;
335 else if (dtp->u.p.current_unit->flags.cc != CC_NONE
336 && (q == '\n' || q == '\r'))
338 /* Unexpected end of line. Set the position. */
339 dtp->u.p.sf_seen_eor = 1;
341 /* If we see an EOR during non-advancing I/O, we need to skip
342 the rest of the I/O statement. Set the corresponding flag. */
343 if (dtp->u.p.advance_status == ADVANCE_NO || dtp->u.p.seen_dollar)
344 dtp->u.p.eor_condition = 1;
346 /* If we encounter a CR, it might be a CRLF. */
347 if (q == '\r') /* Probably a CRLF */
349 /* See if there is an LF. */
350 q2 = fbuf_getc (dtp->u.p.current_unit);
351 if (q2 == '\n')
352 dtp->u.p.sf_seen_eor = 2;
353 else if (q2 != EOF) /* Oops, seek back. */
354 fbuf_seek (dtp->u.p.current_unit, -1, SEEK_CUR);
357 /* Without padding, terminate the I/O statement without assigning
358 the value. With padding, the value still needs to be assigned,
359 so we can just continue with a short read. */
360 if (dtp->u.p.current_unit->pad_status == PAD_NO)
362 generate_error (&dtp->common, LIBERROR_EOR, NULL);
363 return NULL;
366 *length = n;
367 goto done;
369 /* Short circuit the read if a comma is found during numeric input.
370 The flag is set to zero during character reads so that commas in
371 strings are not ignored */
372 else if (q == ',')
373 if (dtp->u.p.sf_read_comma == 1)
375 seen_comma = 1;
376 notify_std (&dtp->common, GFC_STD_GNU,
377 "Comma in formatted numeric read.");
378 break;
380 n++;
383 *length = n;
385 /* A short read implies we hit EOF, unless we hit EOR, a comma, or
386 some other stuff. Set the relevant flags. */
387 if (lorig > *length && !dtp->u.p.sf_seen_eor && !seen_comma)
389 if (n > 0)
391 if (dtp->u.p.advance_status == ADVANCE_NO)
393 if (dtp->u.p.current_unit->pad_status == PAD_NO)
395 hit_eof (dtp);
396 return NULL;
398 else
399 dtp->u.p.eor_condition = 1;
401 else
402 dtp->u.p.at_eof = 1;
404 else if (dtp->u.p.advance_status == ADVANCE_NO
405 || dtp->u.p.current_unit->pad_status == PAD_NO
406 || dtp->u.p.current_unit->bytes_left
407 == dtp->u.p.current_unit->recl)
409 hit_eof (dtp);
410 return NULL;
414 done:
416 dtp->u.p.current_unit->bytes_left -= n;
418 if (((dtp->common.flags & IOPARM_DT_HAS_SIZE) != 0) ||
419 dtp->u.p.current_unit->has_size)
420 dtp->u.p.current_unit->size_used += (GFC_IO_INT) n;
422 /* We can't call fbuf_getptr before the loop doing fbuf_getc, because
423 fbuf_getc might reallocate the buffer. So return current pointer
424 minus all the advances, which is n plus up to two characters
425 of newline or comma. */
426 return fbuf_getptr (dtp->u.p.current_unit)
427 - n - dtp->u.p.sf_seen_eor - seen_comma;
431 /* Function for reading the next couple of bytes from the current
432 file, advancing the current position. We return NULL on end of record or
433 end of file. This function is only for formatted I/O, unformatted uses
434 read_block_direct.
436 If the read is short, then it is because the current record does not
437 have enough data to satisfy the read request and the file was
438 opened with PAD=YES. The caller must assume tailing spaces for
439 short reads. */
441 void *
442 read_block_form (st_parameter_dt *dtp, size_t *nbytes)
444 char *source;
445 size_t norig;
447 if (!is_stream_io (dtp))
449 if (dtp->u.p.current_unit->bytes_left < (gfc_offset) *nbytes)
451 /* For preconnected units with default record length, set bytes left
452 to unit record length and proceed, otherwise error. */
453 if (dtp->u.p.current_unit->unit_number == options.stdin_unit
454 && dtp->u.p.current_unit->recl == default_recl)
455 dtp->u.p.current_unit->bytes_left = dtp->u.p.current_unit->recl;
456 else
458 if (unlikely (dtp->u.p.current_unit->pad_status == PAD_NO)
459 && !is_internal_unit (dtp))
461 /* Not enough data left. */
462 generate_error (&dtp->common, LIBERROR_EOR, NULL);
463 return NULL;
467 if (is_internal_unit(dtp))
469 if (*nbytes > 0 && dtp->u.p.current_unit->bytes_left == 0)
471 if (dtp->u.p.advance_status == ADVANCE_NO)
473 generate_error (&dtp->common, LIBERROR_EOR, NULL);
474 return NULL;
478 else
480 if (unlikely (dtp->u.p.current_unit->bytes_left == 0))
482 hit_eof (dtp);
483 return NULL;
487 *nbytes = dtp->u.p.current_unit->bytes_left;
491 if (dtp->u.p.current_unit->flags.form == FORM_FORMATTED &&
492 (dtp->u.p.current_unit->flags.access == ACCESS_SEQUENTIAL ||
493 dtp->u.p.current_unit->flags.access == ACCESS_STREAM))
495 if (is_internal_unit (dtp))
496 source = read_sf_internal (dtp, nbytes);
497 else
498 source = read_sf (dtp, nbytes);
500 dtp->u.p.current_unit->strm_pos +=
501 (gfc_offset) (*nbytes + dtp->u.p.sf_seen_eor);
502 return source;
505 /* If we reach here, we can assume it's direct access. */
507 dtp->u.p.current_unit->bytes_left -= (gfc_offset) *nbytes;
509 norig = *nbytes;
510 source = fbuf_read (dtp->u.p.current_unit, nbytes);
511 fbuf_seek (dtp->u.p.current_unit, *nbytes, SEEK_CUR);
513 if (((dtp->common.flags & IOPARM_DT_HAS_SIZE) != 0) ||
514 dtp->u.p.current_unit->has_size)
515 dtp->u.p.current_unit->size_used += (GFC_IO_INT) *nbytes;
517 if (norig != *nbytes)
519 /* Short read, this shouldn't happen. */
520 if (dtp->u.p.current_unit->pad_status == PAD_NO)
522 generate_error (&dtp->common, LIBERROR_EOR, NULL);
523 source = NULL;
527 dtp->u.p.current_unit->strm_pos += (gfc_offset) *nbytes;
529 return source;
533 /* Read a block from a character(kind=4) internal unit, to be transferred into
534 a character(kind=4) variable. Note: Portions of this code borrowed from
535 read_sf_internal. */
536 void *
537 read_block_form4 (st_parameter_dt *dtp, size_t *nbytes)
539 static gfc_char4_t *empty_string[0];
540 gfc_char4_t *source;
541 size_t lorig;
543 if (dtp->u.p.current_unit->bytes_left < (gfc_offset) *nbytes)
544 *nbytes = dtp->u.p.current_unit->bytes_left;
546 /* Zero size array gives internal unit len of 0. Nothing to read. */
547 if (dtp->internal_unit_len == 0
548 && dtp->u.p.current_unit->pad_status == PAD_NO)
549 hit_eof (dtp);
551 /* If we have seen an eor previously, return a length of 0. The
552 caller is responsible for correctly padding the input field. */
553 if (dtp->u.p.sf_seen_eor)
555 *nbytes = 0;
556 /* Just return something that isn't a NULL pointer, otherwise the
557 caller thinks an error occurred. */
558 return empty_string;
561 lorig = *nbytes;
562 source = (gfc_char4_t *) mem_alloc_r4 (dtp->u.p.current_unit->s, nbytes);
564 if (unlikely (lorig > *nbytes))
566 hit_eof (dtp);
567 return NULL;
570 dtp->u.p.current_unit->bytes_left -= *nbytes;
572 if (((dtp->common.flags & IOPARM_DT_HAS_SIZE) != 0) ||
573 dtp->u.p.current_unit->has_size)
574 dtp->u.p.current_unit->size_used += (GFC_IO_INT) *nbytes;
576 return source;
580 /* Reads a block directly into application data space. This is for
581 unformatted files. */
583 static void
584 read_block_direct (st_parameter_dt *dtp, void *buf, size_t nbytes)
586 ssize_t to_read_record;
587 ssize_t have_read_record;
588 ssize_t to_read_subrecord;
589 ssize_t have_read_subrecord;
590 int short_record;
592 if (is_stream_io (dtp))
594 have_read_record = sread (dtp->u.p.current_unit->s, buf,
595 nbytes);
596 if (unlikely (have_read_record < 0))
598 generate_error (&dtp->common, LIBERROR_OS, NULL);
599 return;
602 dtp->u.p.current_unit->strm_pos += (gfc_offset) have_read_record;
604 if (unlikely ((ssize_t) nbytes != have_read_record))
606 /* Short read, e.g. if we hit EOF. For stream files,
607 we have to set the end-of-file condition. */
608 hit_eof (dtp);
610 return;
613 if (dtp->u.p.current_unit->flags.access == ACCESS_DIRECT)
615 if (dtp->u.p.current_unit->bytes_left < (gfc_offset) nbytes)
617 short_record = 1;
618 to_read_record = dtp->u.p.current_unit->bytes_left;
619 nbytes = to_read_record;
621 else
623 short_record = 0;
624 to_read_record = nbytes;
627 dtp->u.p.current_unit->bytes_left -= to_read_record;
629 to_read_record = sread (dtp->u.p.current_unit->s, buf, to_read_record);
630 if (unlikely (to_read_record < 0))
632 generate_error (&dtp->common, LIBERROR_OS, NULL);
633 return;
636 if (to_read_record != (ssize_t) nbytes)
638 /* Short read, e.g. if we hit EOF. Apparently, we read
639 more than was written to the last record. */
640 return;
643 if (unlikely (short_record))
645 generate_error (&dtp->common, LIBERROR_SHORT_RECORD, NULL);
647 return;
650 /* Unformatted sequential. We loop over the subrecords, reading
651 until the request has been fulfilled or the record has run out
652 of continuation subrecords. */
654 /* Check whether we exceed the total record length. */
656 if (dtp->u.p.current_unit->flags.has_recl
657 && ((gfc_offset) nbytes > dtp->u.p.current_unit->bytes_left))
659 to_read_record = dtp->u.p.current_unit->bytes_left;
660 short_record = 1;
662 else
664 to_read_record = nbytes;
665 short_record = 0;
667 have_read_record = 0;
669 while(1)
671 if (dtp->u.p.current_unit->bytes_left_subrecord
672 < (gfc_offset) to_read_record)
674 to_read_subrecord = dtp->u.p.current_unit->bytes_left_subrecord;
675 to_read_record -= to_read_subrecord;
677 else
679 to_read_subrecord = to_read_record;
680 to_read_record = 0;
683 dtp->u.p.current_unit->bytes_left_subrecord -= to_read_subrecord;
685 have_read_subrecord = sread (dtp->u.p.current_unit->s,
686 buf + have_read_record, to_read_subrecord);
687 if (unlikely (have_read_subrecord < 0))
689 generate_error (&dtp->common, LIBERROR_OS, NULL);
690 return;
693 have_read_record += have_read_subrecord;
695 if (unlikely (to_read_subrecord != have_read_subrecord))
697 /* Short read, e.g. if we hit EOF. This means the record
698 structure has been corrupted, or the trailing record
699 marker would still be present. */
701 generate_error (&dtp->common, LIBERROR_CORRUPT_FILE, NULL);
702 return;
705 if (to_read_record > 0)
707 if (likely (dtp->u.p.current_unit->continued))
709 next_record_r_unf (dtp, 0);
710 us_read (dtp, 1);
712 else
714 /* Let's make sure the file position is correctly pre-positioned
715 for the next read statement. */
717 dtp->u.p.current_unit->current_record = 0;
718 next_record_r_unf (dtp, 0);
719 generate_error (&dtp->common, LIBERROR_SHORT_RECORD, NULL);
720 return;
723 else
725 /* Normal exit, the read request has been fulfilled. */
726 break;
730 dtp->u.p.current_unit->bytes_left -= have_read_record;
731 if (unlikely (short_record))
733 generate_error (&dtp->common, LIBERROR_SHORT_RECORD, NULL);
734 return;
736 return;
740 /* Function for writing a block of bytes to the current file at the
741 current position, advancing the file pointer. We are given a length
742 and return a pointer to a buffer that the caller must (completely)
743 fill in. Returns NULL on error. */
745 void *
746 write_block (st_parameter_dt *dtp, size_t length)
748 char *dest;
750 if (!is_stream_io (dtp))
752 if (dtp->u.p.current_unit->bytes_left < (gfc_offset) length)
754 /* For preconnected units with default record length, set bytes left
755 to unit record length and proceed, otherwise error. */
756 if (likely ((dtp->u.p.current_unit->unit_number
757 == options.stdout_unit
758 || dtp->u.p.current_unit->unit_number
759 == options.stderr_unit)
760 && dtp->u.p.current_unit->recl == default_recl))
761 dtp->u.p.current_unit->bytes_left = dtp->u.p.current_unit->recl;
762 else
764 generate_error (&dtp->common, LIBERROR_EOR, NULL);
765 return NULL;
769 dtp->u.p.current_unit->bytes_left -= (gfc_offset) length;
772 if (is_internal_unit (dtp))
774 if (is_char4_unit(dtp)) /* char4 internel unit. */
776 gfc_char4_t *dest4;
777 dest4 = mem_alloc_w4 (dtp->u.p.current_unit->s, &length);
778 if (dest4 == NULL)
780 generate_error (&dtp->common, LIBERROR_END, NULL);
781 return NULL;
783 return dest4;
785 else
786 dest = mem_alloc_w (dtp->u.p.current_unit->s, &length);
788 if (dest == NULL)
790 generate_error (&dtp->common, LIBERROR_END, NULL);
791 return NULL;
794 if (unlikely (dtp->u.p.current_unit->endfile == AT_ENDFILE))
795 generate_error (&dtp->common, LIBERROR_END, NULL);
797 else
799 dest = fbuf_alloc (dtp->u.p.current_unit, length);
800 if (dest == NULL)
802 generate_error (&dtp->common, LIBERROR_OS, NULL);
803 return NULL;
807 if (((dtp->common.flags & IOPARM_DT_HAS_SIZE) != 0) ||
808 dtp->u.p.current_unit->has_size)
809 dtp->u.p.current_unit->size_used += (GFC_IO_INT) length;
811 dtp->u.p.current_unit->strm_pos += (gfc_offset) length;
813 return dest;
817 /* High level interface to swrite(), taking care of errors. This is only
818 called for unformatted files. There are three cases to consider:
819 Stream I/O, unformatted direct, unformatted sequential. */
821 static bool
822 write_buf (st_parameter_dt *dtp, void *buf, size_t nbytes)
825 ssize_t have_written;
826 ssize_t to_write_subrecord;
827 int short_record;
829 /* Stream I/O. */
831 if (is_stream_io (dtp))
833 have_written = swrite (dtp->u.p.current_unit->s, buf, nbytes);
834 if (unlikely (have_written < 0))
836 generate_error (&dtp->common, LIBERROR_OS, NULL);
837 return false;
840 dtp->u.p.current_unit->strm_pos += (gfc_offset) have_written;
842 return true;
845 /* Unformatted direct access. */
847 if (dtp->u.p.current_unit->flags.access == ACCESS_DIRECT)
849 if (unlikely (dtp->u.p.current_unit->bytes_left < (gfc_offset) nbytes))
851 generate_error (&dtp->common, LIBERROR_DIRECT_EOR, NULL);
852 return false;
855 if (buf == NULL && nbytes == 0)
856 return true;
858 have_written = swrite (dtp->u.p.current_unit->s, buf, nbytes);
859 if (unlikely (have_written < 0))
861 generate_error (&dtp->common, LIBERROR_OS, NULL);
862 return false;
865 dtp->u.p.current_unit->strm_pos += (gfc_offset) have_written;
866 dtp->u.p.current_unit->bytes_left -= (gfc_offset) have_written;
868 return true;
871 /* Unformatted sequential. */
873 have_written = 0;
875 if (dtp->u.p.current_unit->flags.has_recl
876 && (gfc_offset) nbytes > dtp->u.p.current_unit->bytes_left)
878 nbytes = dtp->u.p.current_unit->bytes_left;
879 short_record = 1;
881 else
883 short_record = 0;
886 while (1)
889 to_write_subrecord =
890 (size_t) dtp->u.p.current_unit->bytes_left_subrecord < nbytes ?
891 (size_t) dtp->u.p.current_unit->bytes_left_subrecord : nbytes;
893 dtp->u.p.current_unit->bytes_left_subrecord -=
894 (gfc_offset) to_write_subrecord;
896 to_write_subrecord = swrite (dtp->u.p.current_unit->s,
897 buf + have_written, to_write_subrecord);
898 if (unlikely (to_write_subrecord < 0))
900 generate_error (&dtp->common, LIBERROR_OS, NULL);
901 return false;
904 dtp->u.p.current_unit->strm_pos += (gfc_offset) to_write_subrecord;
905 nbytes -= to_write_subrecord;
906 have_written += to_write_subrecord;
908 if (nbytes == 0)
909 break;
911 next_record_w_unf (dtp, 1);
912 us_write (dtp, 1);
914 dtp->u.p.current_unit->bytes_left -= have_written;
915 if (unlikely (short_record))
917 generate_error (&dtp->common, LIBERROR_SHORT_RECORD, NULL);
918 return false;
920 return true;
924 /* Reverse memcpy - used for byte swapping. */
926 static void
927 reverse_memcpy (void *dest, const void *src, size_t n)
929 char *d, *s;
930 size_t i;
932 d = (char *) dest;
933 s = (char *) src + n - 1;
935 /* Write with ascending order - this is likely faster
936 on modern architectures because of write combining. */
937 for (i=0; i<n; i++)
938 *(d++) = *(s--);
942 /* Utility function for byteswapping an array, using the bswap
943 builtins if possible. dest and src can overlap completely, or then
944 they must point to separate objects; partial overlaps are not
945 allowed. */
947 static void
948 bswap_array (void *dest, const void *src, size_t size, size_t nelems)
950 const char *ps;
951 char *pd;
953 switch (size)
955 case 1:
956 break;
957 case 2:
958 for (size_t i = 0; i < nelems; i++)
959 ((uint16_t*)dest)[i] = __builtin_bswap16 (((uint16_t*)src)[i]);
960 break;
961 case 4:
962 for (size_t i = 0; i < nelems; i++)
963 ((uint32_t*)dest)[i] = __builtin_bswap32 (((uint32_t*)src)[i]);
964 break;
965 case 8:
966 for (size_t i = 0; i < nelems; i++)
967 ((uint64_t*)dest)[i] = __builtin_bswap64 (((uint64_t*)src)[i]);
968 break;
969 case 12:
970 ps = src;
971 pd = dest;
972 for (size_t i = 0; i < nelems; i++)
974 uint32_t tmp;
975 memcpy (&tmp, ps, 4);
976 *(uint32_t*)pd = __builtin_bswap32 (*(uint32_t*)(ps + 8));
977 *(uint32_t*)(pd + 4) = __builtin_bswap32 (*(uint32_t*)(ps + 4));
978 *(uint32_t*)(pd + 8) = __builtin_bswap32 (tmp);
979 ps += size;
980 pd += size;
982 break;
983 case 16:
984 ps = src;
985 pd = dest;
986 for (size_t i = 0; i < nelems; i++)
988 uint64_t tmp;
989 memcpy (&tmp, ps, 8);
990 *(uint64_t*)pd = __builtin_bswap64 (*(uint64_t*)(ps + 8));
991 *(uint64_t*)(pd + 8) = __builtin_bswap64 (tmp);
992 ps += size;
993 pd += size;
995 break;
996 default:
997 pd = dest;
998 if (dest != src)
1000 ps = src;
1001 for (size_t i = 0; i < nelems; i++)
1003 reverse_memcpy (pd, ps, size);
1004 ps += size;
1005 pd += size;
1008 else
1010 /* In-place byte swap. */
1011 for (size_t i = 0; i < nelems; i++)
1013 char tmp, *low = pd, *high = pd + size - 1;
1014 for (size_t j = 0; j < size/2; j++)
1016 tmp = *low;
1017 *low = *high;
1018 *high = tmp;
1019 low++;
1020 high--;
1022 pd += size;
1029 /* Master function for unformatted reads. */
1031 static void
1032 unformatted_read (st_parameter_dt *dtp, bt type,
1033 void *dest, int kind, size_t size, size_t nelems)
1035 if (type == BT_CLASS)
1037 int unit = dtp->u.p.current_unit->unit_number;
1038 char tmp_iomsg[IOMSG_LEN] = "";
1039 char *child_iomsg;
1040 gfc_charlen_type child_iomsg_len;
1041 int noiostat;
1042 int *child_iostat = NULL;
1044 /* Set iostat, intent(out). */
1045 noiostat = 0;
1046 child_iostat = (dtp->common.flags & IOPARM_HAS_IOSTAT) ?
1047 dtp->common.iostat : &noiostat;
1049 /* Set iomsg, intent(inout). */
1050 if (dtp->common.flags & IOPARM_HAS_IOMSG)
1052 child_iomsg = dtp->common.iomsg;
1053 child_iomsg_len = dtp->common.iomsg_len;
1055 else
1057 child_iomsg = tmp_iomsg;
1058 child_iomsg_len = IOMSG_LEN;
1061 /* Call the user defined unformatted READ procedure. */
1062 dtp->u.p.current_unit->child_dtio++;
1063 dtp->u.p.ufdtio_ptr (dest, &unit, child_iostat, child_iomsg,
1064 child_iomsg_len);
1065 dtp->u.p.current_unit->child_dtio--;
1066 return;
1069 if (type == BT_CHARACTER)
1070 size *= GFC_SIZE_OF_CHAR_KIND(kind);
1071 read_block_direct (dtp, dest, size * nelems);
1073 if (unlikely (dtp->u.p.current_unit->flags.convert == GFC_CONVERT_SWAP)
1074 && kind != 1)
1076 /* Handle wide chracters. */
1077 if (type == BT_CHARACTER)
1079 nelems *= size;
1080 size = kind;
1083 /* Break up complex into its constituent reals. */
1084 else if (type == BT_COMPLEX)
1086 nelems *= 2;
1087 size /= 2;
1089 bswap_array (dest, dest, size, nelems);
1094 /* Master function for unformatted writes. NOTE: For kind=10 the size is 16
1095 bytes on 64 bit machines. The unused bytes are not initialized and never
1096 used, which can show an error with memory checking analyzers like
1097 valgrind. We us BT_CLASS to denote a User Defined I/O call. */
1099 static void
1100 unformatted_write (st_parameter_dt *dtp, bt type,
1101 void *source, int kind, size_t size, size_t nelems)
1103 if (type == BT_CLASS)
1105 int unit = dtp->u.p.current_unit->unit_number;
1106 char tmp_iomsg[IOMSG_LEN] = "";
1107 char *child_iomsg;
1108 gfc_charlen_type child_iomsg_len;
1109 int noiostat;
1110 int *child_iostat = NULL;
1112 /* Set iostat, intent(out). */
1113 noiostat = 0;
1114 child_iostat = (dtp->common.flags & IOPARM_HAS_IOSTAT) ?
1115 dtp->common.iostat : &noiostat;
1117 /* Set iomsg, intent(inout). */
1118 if (dtp->common.flags & IOPARM_HAS_IOMSG)
1120 child_iomsg = dtp->common.iomsg;
1121 child_iomsg_len = dtp->common.iomsg_len;
1123 else
1125 child_iomsg = tmp_iomsg;
1126 child_iomsg_len = IOMSG_LEN;
1129 /* Call the user defined unformatted WRITE procedure. */
1130 dtp->u.p.current_unit->child_dtio++;
1131 dtp->u.p.ufdtio_ptr (source, &unit, child_iostat, child_iomsg,
1132 child_iomsg_len);
1133 dtp->u.p.current_unit->child_dtio--;
1134 return;
1137 if (likely (dtp->u.p.current_unit->flags.convert == GFC_CONVERT_NATIVE)
1138 || kind == 1)
1140 size_t stride = type == BT_CHARACTER ?
1141 size * GFC_SIZE_OF_CHAR_KIND(kind) : size;
1143 write_buf (dtp, source, stride * nelems);
1145 else
1147 #define BSWAP_BUFSZ 512
1148 char buffer[BSWAP_BUFSZ];
1149 char *p;
1150 size_t nrem;
1152 p = source;
1154 /* Handle wide chracters. */
1155 if (type == BT_CHARACTER && kind != 1)
1157 nelems *= size;
1158 size = kind;
1161 /* Break up complex into its constituent reals. */
1162 if (type == BT_COMPLEX)
1164 nelems *= 2;
1165 size /= 2;
1168 /* By now, all complex variables have been split into their
1169 constituent reals. */
1171 nrem = nelems;
1174 size_t nc;
1175 if (size * nrem > BSWAP_BUFSZ)
1176 nc = BSWAP_BUFSZ / size;
1177 else
1178 nc = nrem;
1180 bswap_array (buffer, p, size, nc);
1181 write_buf (dtp, buffer, size * nc);
1182 p += size * nc;
1183 nrem -= nc;
1185 while (nrem > 0);
1190 /* Return a pointer to the name of a type. */
1192 const char *
1193 type_name (bt type)
1195 const char *p;
1197 switch (type)
1199 case BT_INTEGER:
1200 p = "INTEGER";
1201 break;
1202 case BT_LOGICAL:
1203 p = "LOGICAL";
1204 break;
1205 case BT_CHARACTER:
1206 p = "CHARACTER";
1207 break;
1208 case BT_REAL:
1209 p = "REAL";
1210 break;
1211 case BT_COMPLEX:
1212 p = "COMPLEX";
1213 break;
1214 case BT_CLASS:
1215 p = "CLASS or DERIVED";
1216 break;
1217 default:
1218 internal_error (NULL, "type_name(): Bad type");
1221 return p;
1225 /* Write a constant string to the output.
1226 This is complicated because the string can have doubled delimiters
1227 in it. The length in the format node is the true length. */
1229 static void
1230 write_constant_string (st_parameter_dt *dtp, const fnode *f)
1232 char c, delimiter, *p, *q;
1233 int length;
1235 length = f->u.string.length;
1236 if (length == 0)
1237 return;
1239 p = write_block (dtp, length);
1240 if (p == NULL)
1241 return;
1243 q = f->u.string.p;
1244 delimiter = q[-1];
1246 for (; length > 0; length--)
1248 c = *p++ = *q++;
1249 if (c == delimiter && c != 'H' && c != 'h')
1250 q++; /* Skip the doubled delimiter. */
1255 /* Given actual and expected types in a formatted data transfer, make
1256 sure they agree. If not, an error message is generated. Returns
1257 nonzero if something went wrong. */
1259 static int
1260 require_type (st_parameter_dt *dtp, bt expected, bt actual, const fnode *f)
1262 #define BUFLEN 100
1263 char buffer[BUFLEN];
1265 if (actual == expected)
1266 return 0;
1268 /* Adjust item_count before emitting error message. */
1269 snprintf (buffer, BUFLEN,
1270 "Expected %s for item %d in formatted transfer, got %s",
1271 type_name (expected), dtp->u.p.item_count - 1, type_name (actual));
1273 format_error (dtp, f, buffer);
1274 return 1;
1278 /* Check that the dtio procedure required for formatted IO is present. */
1280 static int
1281 check_dtio_proc (st_parameter_dt *dtp, const fnode *f)
1283 char buffer[BUFLEN];
1285 if (dtp->u.p.fdtio_ptr != NULL)
1286 return 0;
1288 snprintf (buffer, BUFLEN,
1289 "Missing DTIO procedure or intrinsic type passed for item %d "
1290 "in formatted transfer",
1291 dtp->u.p.item_count - 1);
1293 format_error (dtp, f, buffer);
1294 return 1;
1298 static int
1299 require_numeric_type (st_parameter_dt *dtp, bt actual, const fnode *f)
1301 #define BUFLEN 100
1302 char buffer[BUFLEN];
1304 if (actual == BT_INTEGER || actual == BT_REAL || actual == BT_COMPLEX)
1305 return 0;
1307 /* Adjust item_count before emitting error message. */
1308 snprintf (buffer, BUFLEN,
1309 "Expected numeric type for item %d in formatted transfer, got %s",
1310 dtp->u.p.item_count - 1, type_name (actual));
1312 format_error (dtp, f, buffer);
1313 return 1;
1316 static char *
1317 get_dt_format (char *p, gfc_charlen_type *length)
1319 char delim = p[-1]; /* The delimiter is always the first character back. */
1320 char c, *q, *res;
1321 gfc_charlen_type len = *length; /* This length already correct, less 'DT'. */
1323 res = q = xmalloc (len + 2);
1325 /* Set the beginning of the string to 'DT', length adjusted below. */
1326 *q++ = 'D';
1327 *q++ = 'T';
1329 /* The string may contain doubled quotes so scan and skip as needed. */
1330 for (; len > 0; len--)
1332 c = *q++ = *p++;
1333 if (c == delim)
1334 p++; /* Skip the doubled delimiter. */
1337 /* Adjust the string length by two now that we are done. */
1338 *length += 2;
1340 return res;
1344 /* This function is in the main loop for a formatted data transfer
1345 statement. It would be natural to implement this as a coroutine
1346 with the user program, but C makes that awkward. We loop,
1347 processing format elements. When we actually have to transfer
1348 data instead of just setting flags, we return control to the user
1349 program which calls a function that supplies the address and type
1350 of the next element, then comes back here to process it. */
1352 static void
1353 formatted_transfer_scalar_read (st_parameter_dt *dtp, bt type, void *p, int kind,
1354 size_t size)
1356 int pos, bytes_used;
1357 const fnode *f;
1358 format_token t;
1359 int n;
1360 int consume_data_flag;
1362 /* Change a complex data item into a pair of reals. */
1364 n = (p == NULL) ? 0 : ((type != BT_COMPLEX) ? 1 : 2);
1365 if (type == BT_COMPLEX)
1367 type = BT_REAL;
1368 size /= 2;
1371 /* If there's an EOR condition, we simulate finalizing the transfer
1372 by doing nothing. */
1373 if (dtp->u.p.eor_condition)
1374 return;
1376 /* Set this flag so that commas in reads cause the read to complete before
1377 the entire field has been read. The next read field will start right after
1378 the comma in the stream. (Set to 0 for character reads). */
1379 dtp->u.p.sf_read_comma =
1380 dtp->u.p.current_unit->decimal_status == DECIMAL_COMMA ? 0 : 1;
1382 for (;;)
1384 /* If reversion has occurred and there is another real data item,
1385 then we have to move to the next record. */
1386 if (dtp->u.p.reversion_flag && n > 0)
1388 dtp->u.p.reversion_flag = 0;
1389 next_record (dtp, 0);
1392 consume_data_flag = 1;
1393 if ((dtp->common.flags & IOPARM_LIBRETURN_MASK) != IOPARM_LIBRETURN_OK)
1394 break;
1396 f = next_format (dtp);
1397 if (f == NULL)
1399 /* No data descriptors left. */
1400 if (unlikely (n > 0))
1401 generate_error (&dtp->common, LIBERROR_FORMAT,
1402 "Insufficient data descriptors in format after reversion");
1403 return;
1406 t = f->format;
1408 bytes_used = (int)(dtp->u.p.current_unit->recl
1409 - dtp->u.p.current_unit->bytes_left);
1411 if (is_stream_io(dtp))
1412 bytes_used = 0;
1414 switch (t)
1416 case FMT_I:
1417 if (n == 0)
1418 goto need_read_data;
1419 if (require_type (dtp, BT_INTEGER, type, f))
1420 return;
1421 read_decimal (dtp, f, p, kind);
1422 break;
1424 case FMT_B:
1425 if (n == 0)
1426 goto need_read_data;
1427 if (!(compile_options.allow_std & GFC_STD_GNU)
1428 && require_numeric_type (dtp, type, f))
1429 return;
1430 if (!(compile_options.allow_std & GFC_STD_F2008)
1431 && require_type (dtp, BT_INTEGER, type, f))
1432 return;
1433 read_radix (dtp, f, p, kind, 2);
1434 break;
1436 case FMT_O:
1437 if (n == 0)
1438 goto need_read_data;
1439 if (!(compile_options.allow_std & GFC_STD_GNU)
1440 && require_numeric_type (dtp, type, f))
1441 return;
1442 if (!(compile_options.allow_std & GFC_STD_F2008)
1443 && require_type (dtp, BT_INTEGER, type, f))
1444 return;
1445 read_radix (dtp, f, p, kind, 8);
1446 break;
1448 case FMT_Z:
1449 if (n == 0)
1450 goto need_read_data;
1451 if (!(compile_options.allow_std & GFC_STD_GNU)
1452 && require_numeric_type (dtp, type, f))
1453 return;
1454 if (!(compile_options.allow_std & GFC_STD_F2008)
1455 && require_type (dtp, BT_INTEGER, type, f))
1456 return;
1457 read_radix (dtp, f, p, kind, 16);
1458 break;
1460 case FMT_A:
1461 if (n == 0)
1462 goto need_read_data;
1464 /* It is possible to have FMT_A with something not BT_CHARACTER such
1465 as when writing out hollerith strings, so check both type
1466 and kind before calling wide character routines. */
1467 if (type == BT_CHARACTER && kind == 4)
1468 read_a_char4 (dtp, f, p, size);
1469 else
1470 read_a (dtp, f, p, size);
1471 break;
1473 case FMT_L:
1474 if (n == 0)
1475 goto need_read_data;
1476 read_l (dtp, f, p, kind);
1477 break;
1479 case FMT_D:
1480 if (n == 0)
1481 goto need_read_data;
1482 if (require_type (dtp, BT_REAL, type, f))
1483 return;
1484 read_f (dtp, f, p, kind);
1485 break;
1487 case FMT_DT:
1488 if (n == 0)
1489 goto need_read_data;
1491 if (check_dtio_proc (dtp, f))
1492 return;
1493 if (require_type (dtp, BT_CLASS, type, f))
1494 return;
1495 int unit = dtp->u.p.current_unit->unit_number;
1496 char dt[] = "DT";
1497 char tmp_iomsg[IOMSG_LEN] = "";
1498 char *child_iomsg;
1499 gfc_charlen_type child_iomsg_len;
1500 int noiostat;
1501 int *child_iostat = NULL;
1502 char *iotype;
1503 gfc_charlen_type iotype_len = f->u.udf.string_len;
1505 /* Build the iotype string. */
1506 if (iotype_len == 0)
1508 iotype_len = 2;
1509 iotype = dt;
1511 else
1512 iotype = get_dt_format (f->u.udf.string, &iotype_len);
1514 /* Set iostat, intent(out). */
1515 noiostat = 0;
1516 child_iostat = (dtp->common.flags & IOPARM_HAS_IOSTAT) ?
1517 dtp->common.iostat : &noiostat;
1519 /* Set iomsg, intent(inout). */
1520 if (dtp->common.flags & IOPARM_HAS_IOMSG)
1522 child_iomsg = dtp->common.iomsg;
1523 child_iomsg_len = dtp->common.iomsg_len;
1525 else
1527 child_iomsg = tmp_iomsg;
1528 child_iomsg_len = IOMSG_LEN;
1531 /* Call the user defined formatted READ procedure. */
1532 dtp->u.p.current_unit->child_dtio++;
1533 dtp->u.p.current_unit->last_char = EOF - 1;
1534 dtp->u.p.fdtio_ptr (p, &unit, iotype, f->u.udf.vlist,
1535 child_iostat, child_iomsg,
1536 iotype_len, child_iomsg_len);
1537 dtp->u.p.current_unit->child_dtio--;
1539 if (f->u.udf.string_len != 0)
1540 free (iotype);
1541 /* Note: vlist is freed in free_format_data. */
1542 break;
1544 case FMT_E:
1545 if (n == 0)
1546 goto need_read_data;
1547 if (require_type (dtp, BT_REAL, type, f))
1548 return;
1549 read_f (dtp, f, p, kind);
1550 break;
1552 case FMT_EN:
1553 if (n == 0)
1554 goto need_read_data;
1555 if (require_type (dtp, BT_REAL, type, f))
1556 return;
1557 read_f (dtp, f, p, kind);
1558 break;
1560 case FMT_ES:
1561 if (n == 0)
1562 goto need_read_data;
1563 if (require_type (dtp, BT_REAL, type, f))
1564 return;
1565 read_f (dtp, f, p, kind);
1566 break;
1568 case FMT_F:
1569 if (n == 0)
1570 goto need_read_data;
1571 if (require_type (dtp, BT_REAL, type, f))
1572 return;
1573 read_f (dtp, f, p, kind);
1574 break;
1576 case FMT_G:
1577 if (n == 0)
1578 goto need_read_data;
1579 switch (type)
1581 case BT_INTEGER:
1582 read_decimal (dtp, f, p, kind);
1583 break;
1584 case BT_LOGICAL:
1585 read_l (dtp, f, p, kind);
1586 break;
1587 case BT_CHARACTER:
1588 if (kind == 4)
1589 read_a_char4 (dtp, f, p, size);
1590 else
1591 read_a (dtp, f, p, size);
1592 break;
1593 case BT_REAL:
1594 read_f (dtp, f, p, kind);
1595 break;
1596 default:
1597 internal_error (&dtp->common, "formatted_transfer(): Bad type");
1599 break;
1601 case FMT_STRING:
1602 consume_data_flag = 0;
1603 format_error (dtp, f, "Constant string in input format");
1604 return;
1606 /* Format codes that don't transfer data. */
1607 case FMT_X:
1608 case FMT_TR:
1609 consume_data_flag = 0;
1610 dtp->u.p.skips += f->u.n;
1611 pos = bytes_used + dtp->u.p.skips - 1;
1612 dtp->u.p.pending_spaces = pos - dtp->u.p.max_pos + 1;
1613 read_x (dtp, f->u.n);
1614 break;
1616 case FMT_TL:
1617 case FMT_T:
1618 consume_data_flag = 0;
1620 if (f->format == FMT_TL)
1622 /* Handle the special case when no bytes have been used yet.
1623 Cannot go below zero. */
1624 if (bytes_used == 0)
1626 dtp->u.p.pending_spaces -= f->u.n;
1627 dtp->u.p.skips -= f->u.n;
1628 dtp->u.p.skips = dtp->u.p.skips < 0 ? 0 : dtp->u.p.skips;
1631 pos = bytes_used - f->u.n;
1633 else /* FMT_T */
1634 pos = f->u.n - 1;
1636 /* Standard 10.6.1.1: excessive left tabbing is reset to the
1637 left tab limit. We do not check if the position has gone
1638 beyond the end of record because a subsequent tab could
1639 bring us back again. */
1640 pos = pos < 0 ? 0 : pos;
1642 dtp->u.p.skips = dtp->u.p.skips + pos - bytes_used;
1643 dtp->u.p.pending_spaces = dtp->u.p.pending_spaces
1644 + pos - dtp->u.p.max_pos;
1645 dtp->u.p.pending_spaces = dtp->u.p.pending_spaces < 0
1646 ? 0 : dtp->u.p.pending_spaces;
1647 if (dtp->u.p.skips == 0)
1648 break;
1650 /* Adjust everything for end-of-record condition */
1651 if (dtp->u.p.sf_seen_eor && !is_internal_unit (dtp))
1653 dtp->u.p.current_unit->bytes_left -= dtp->u.p.sf_seen_eor;
1654 dtp->u.p.skips -= dtp->u.p.sf_seen_eor;
1655 bytes_used = pos;
1656 if (dtp->u.p.pending_spaces == 0)
1657 dtp->u.p.sf_seen_eor = 0;
1659 if (dtp->u.p.skips < 0)
1661 if (is_internal_unit (dtp))
1662 sseek (dtp->u.p.current_unit->s, dtp->u.p.skips, SEEK_CUR);
1663 else
1664 fbuf_seek (dtp->u.p.current_unit, dtp->u.p.skips, SEEK_CUR);
1665 dtp->u.p.current_unit->bytes_left -= (gfc_offset) dtp->u.p.skips;
1666 dtp->u.p.skips = dtp->u.p.pending_spaces = 0;
1668 else
1669 read_x (dtp, dtp->u.p.skips);
1670 break;
1672 case FMT_S:
1673 consume_data_flag = 0;
1674 dtp->u.p.sign_status = SIGN_S;
1675 break;
1677 case FMT_SS:
1678 consume_data_flag = 0;
1679 dtp->u.p.sign_status = SIGN_SS;
1680 break;
1682 case FMT_SP:
1683 consume_data_flag = 0;
1684 dtp->u.p.sign_status = SIGN_SP;
1685 break;
1687 case FMT_BN:
1688 consume_data_flag = 0 ;
1689 dtp->u.p.blank_status = BLANK_NULL;
1690 break;
1692 case FMT_BZ:
1693 consume_data_flag = 0;
1694 dtp->u.p.blank_status = BLANK_ZERO;
1695 break;
1697 case FMT_DC:
1698 consume_data_flag = 0;
1699 dtp->u.p.current_unit->decimal_status = DECIMAL_COMMA;
1700 break;
1702 case FMT_DP:
1703 consume_data_flag = 0;
1704 dtp->u.p.current_unit->decimal_status = DECIMAL_POINT;
1705 break;
1707 case FMT_RC:
1708 consume_data_flag = 0;
1709 dtp->u.p.current_unit->round_status = ROUND_COMPATIBLE;
1710 break;
1712 case FMT_RD:
1713 consume_data_flag = 0;
1714 dtp->u.p.current_unit->round_status = ROUND_DOWN;
1715 break;
1717 case FMT_RN:
1718 consume_data_flag = 0;
1719 dtp->u.p.current_unit->round_status = ROUND_NEAREST;
1720 break;
1722 case FMT_RP:
1723 consume_data_flag = 0;
1724 dtp->u.p.current_unit->round_status = ROUND_PROCDEFINED;
1725 break;
1727 case FMT_RU:
1728 consume_data_flag = 0;
1729 dtp->u.p.current_unit->round_status = ROUND_UP;
1730 break;
1732 case FMT_RZ:
1733 consume_data_flag = 0;
1734 dtp->u.p.current_unit->round_status = ROUND_ZERO;
1735 break;
1737 case FMT_P:
1738 consume_data_flag = 0;
1739 dtp->u.p.scale_factor = f->u.k;
1740 break;
1742 case FMT_DOLLAR:
1743 consume_data_flag = 0;
1744 dtp->u.p.seen_dollar = 1;
1745 break;
1747 case FMT_SLASH:
1748 consume_data_flag = 0;
1749 dtp->u.p.skips = dtp->u.p.pending_spaces = 0;
1750 next_record (dtp, 0);
1751 break;
1753 case FMT_COLON:
1754 /* A colon descriptor causes us to exit this loop (in
1755 particular preventing another / descriptor from being
1756 processed) unless there is another data item to be
1757 transferred. */
1758 consume_data_flag = 0;
1759 if (n == 0)
1760 return;
1761 break;
1763 default:
1764 internal_error (&dtp->common, "Bad format node");
1767 /* Adjust the item count and data pointer. */
1769 if ((consume_data_flag > 0) && (n > 0))
1771 n--;
1772 p = ((char *) p) + size;
1775 dtp->u.p.skips = 0;
1777 pos = (int)(dtp->u.p.current_unit->recl - dtp->u.p.current_unit->bytes_left);
1778 dtp->u.p.max_pos = (dtp->u.p.max_pos > pos) ? dtp->u.p.max_pos : pos;
1781 return;
1783 /* Come here when we need a data descriptor but don't have one. We
1784 push the current format node back onto the input, then return and
1785 let the user program call us back with the data. */
1786 need_read_data:
1787 unget_format (dtp, f);
1791 static void
1792 formatted_transfer_scalar_write (st_parameter_dt *dtp, bt type, void *p, int kind,
1793 size_t size)
1795 gfc_offset pos, bytes_used;
1796 const fnode *f;
1797 format_token t;
1798 int n;
1799 int consume_data_flag;
1801 /* Change a complex data item into a pair of reals. */
1803 n = (p == NULL) ? 0 : ((type != BT_COMPLEX) ? 1 : 2);
1804 if (type == BT_COMPLEX)
1806 type = BT_REAL;
1807 size /= 2;
1810 /* If there's an EOR condition, we simulate finalizing the transfer
1811 by doing nothing. */
1812 if (dtp->u.p.eor_condition)
1813 return;
1815 /* Set this flag so that commas in reads cause the read to complete before
1816 the entire field has been read. The next read field will start right after
1817 the comma in the stream. (Set to 0 for character reads). */
1818 dtp->u.p.sf_read_comma =
1819 dtp->u.p.current_unit->decimal_status == DECIMAL_COMMA ? 0 : 1;
1821 for (;;)
1823 /* If reversion has occurred and there is another real data item,
1824 then we have to move to the next record. */
1825 if (dtp->u.p.reversion_flag && n > 0)
1827 dtp->u.p.reversion_flag = 0;
1828 next_record (dtp, 0);
1831 consume_data_flag = 1;
1832 if ((dtp->common.flags & IOPARM_LIBRETURN_MASK) != IOPARM_LIBRETURN_OK)
1833 break;
1835 f = next_format (dtp);
1836 if (f == NULL)
1838 /* No data descriptors left. */
1839 if (unlikely (n > 0))
1840 generate_error (&dtp->common, LIBERROR_FORMAT,
1841 "Insufficient data descriptors in format after reversion");
1842 return;
1845 /* Now discharge T, TR and X movements to the right. This is delayed
1846 until a data producing format to suppress trailing spaces. */
1848 t = f->format;
1849 if (dtp->u.p.mode == WRITING && dtp->u.p.skips != 0
1850 && ((n>0 && ( t == FMT_I || t == FMT_B || t == FMT_O
1851 || t == FMT_Z || t == FMT_F || t == FMT_E
1852 || t == FMT_EN || t == FMT_ES || t == FMT_G
1853 || t == FMT_L || t == FMT_A || t == FMT_D
1854 || t == FMT_DT))
1855 || t == FMT_STRING))
1857 if (dtp->u.p.skips > 0)
1859 gfc_offset tmp;
1860 write_x (dtp, dtp->u.p.skips, dtp->u.p.pending_spaces);
1861 tmp = dtp->u.p.current_unit->recl
1862 - dtp->u.p.current_unit->bytes_left;
1863 dtp->u.p.max_pos =
1864 dtp->u.p.max_pos > tmp ? dtp->u.p.max_pos : tmp;
1865 dtp->u.p.skips = 0;
1867 if (dtp->u.p.skips < 0)
1869 if (is_internal_unit (dtp))
1870 sseek (dtp->u.p.current_unit->s, dtp->u.p.skips, SEEK_CUR);
1871 else
1872 fbuf_seek (dtp->u.p.current_unit, dtp->u.p.skips, SEEK_CUR);
1873 dtp->u.p.current_unit->bytes_left -= (gfc_offset) dtp->u.p.skips;
1875 dtp->u.p.skips = dtp->u.p.pending_spaces = 0;
1878 bytes_used = dtp->u.p.current_unit->recl
1879 - dtp->u.p.current_unit->bytes_left;
1881 if (is_stream_io(dtp))
1882 bytes_used = 0;
1884 switch (t)
1886 case FMT_I:
1887 if (n == 0)
1888 goto need_data;
1889 if (require_type (dtp, BT_INTEGER, type, f))
1890 return;
1891 write_i (dtp, f, p, kind);
1892 break;
1894 case FMT_B:
1895 if (n == 0)
1896 goto need_data;
1897 if (!(compile_options.allow_std & GFC_STD_GNU)
1898 && require_numeric_type (dtp, type, f))
1899 return;
1900 if (!(compile_options.allow_std & GFC_STD_F2008)
1901 && require_type (dtp, BT_INTEGER, type, f))
1902 return;
1903 write_b (dtp, f, p, kind);
1904 break;
1906 case FMT_O:
1907 if (n == 0)
1908 goto need_data;
1909 if (!(compile_options.allow_std & GFC_STD_GNU)
1910 && require_numeric_type (dtp, type, f))
1911 return;
1912 if (!(compile_options.allow_std & GFC_STD_F2008)
1913 && require_type (dtp, BT_INTEGER, type, f))
1914 return;
1915 write_o (dtp, f, p, kind);
1916 break;
1918 case FMT_Z:
1919 if (n == 0)
1920 goto need_data;
1921 if (!(compile_options.allow_std & GFC_STD_GNU)
1922 && require_numeric_type (dtp, type, f))
1923 return;
1924 if (!(compile_options.allow_std & GFC_STD_F2008)
1925 && require_type (dtp, BT_INTEGER, type, f))
1926 return;
1927 write_z (dtp, f, p, kind);
1928 break;
1930 case FMT_A:
1931 if (n == 0)
1932 goto need_data;
1934 /* It is possible to have FMT_A with something not BT_CHARACTER such
1935 as when writing out hollerith strings, so check both type
1936 and kind before calling wide character routines. */
1937 if (type == BT_CHARACTER && kind == 4)
1938 write_a_char4 (dtp, f, p, size);
1939 else
1940 write_a (dtp, f, p, size);
1941 break;
1943 case FMT_L:
1944 if (n == 0)
1945 goto need_data;
1946 write_l (dtp, f, p, kind);
1947 break;
1949 case FMT_D:
1950 if (n == 0)
1951 goto need_data;
1952 if (require_type (dtp, BT_REAL, type, f))
1953 return;
1954 write_d (dtp, f, p, kind);
1955 break;
1957 case FMT_DT:
1958 if (n == 0)
1959 goto need_data;
1960 int unit = dtp->u.p.current_unit->unit_number;
1961 char dt[] = "DT";
1962 char tmp_iomsg[IOMSG_LEN] = "";
1963 char *child_iomsg;
1964 gfc_charlen_type child_iomsg_len;
1965 int noiostat;
1966 int *child_iostat = NULL;
1967 char *iotype;
1968 gfc_charlen_type iotype_len = f->u.udf.string_len;
1970 /* Build the iotype string. */
1971 if (iotype_len == 0)
1973 iotype_len = 2;
1974 iotype = dt;
1976 else
1977 iotype = get_dt_format (f->u.udf.string, &iotype_len);
1979 /* Set iostat, intent(out). */
1980 noiostat = 0;
1981 child_iostat = (dtp->common.flags & IOPARM_HAS_IOSTAT) ?
1982 dtp->common.iostat : &noiostat;
1984 /* Set iomsg, intent(inout). */
1985 if (dtp->common.flags & IOPARM_HAS_IOMSG)
1987 child_iomsg = dtp->common.iomsg;
1988 child_iomsg_len = dtp->common.iomsg_len;
1990 else
1992 child_iomsg = tmp_iomsg;
1993 child_iomsg_len = IOMSG_LEN;
1996 if (check_dtio_proc (dtp, f))
1997 return;
1999 /* Call the user defined formatted WRITE procedure. */
2000 dtp->u.p.current_unit->child_dtio++;
2002 dtp->u.p.fdtio_ptr (p, &unit, iotype, f->u.udf.vlist,
2003 child_iostat, child_iomsg,
2004 iotype_len, child_iomsg_len);
2005 dtp->u.p.current_unit->child_dtio--;
2007 if (f->u.udf.string_len != 0)
2008 free (iotype);
2009 /* Note: vlist is freed in free_format_data. */
2010 break;
2012 case FMT_E:
2013 if (n == 0)
2014 goto need_data;
2015 if (require_type (dtp, BT_REAL, type, f))
2016 return;
2017 write_e (dtp, f, p, kind);
2018 break;
2020 case FMT_EN:
2021 if (n == 0)
2022 goto need_data;
2023 if (require_type (dtp, BT_REAL, type, f))
2024 return;
2025 write_en (dtp, f, p, kind);
2026 break;
2028 case FMT_ES:
2029 if (n == 0)
2030 goto need_data;
2031 if (require_type (dtp, BT_REAL, type, f))
2032 return;
2033 write_es (dtp, f, p, kind);
2034 break;
2036 case FMT_F:
2037 if (n == 0)
2038 goto need_data;
2039 if (require_type (dtp, BT_REAL, type, f))
2040 return;
2041 write_f (dtp, f, p, kind);
2042 break;
2044 case FMT_G:
2045 if (n == 0)
2046 goto need_data;
2047 switch (type)
2049 case BT_INTEGER:
2050 write_i (dtp, f, p, kind);
2051 break;
2052 case BT_LOGICAL:
2053 write_l (dtp, f, p, kind);
2054 break;
2055 case BT_CHARACTER:
2056 if (kind == 4)
2057 write_a_char4 (dtp, f, p, size);
2058 else
2059 write_a (dtp, f, p, size);
2060 break;
2061 case BT_REAL:
2062 if (f->u.real.w == 0)
2063 write_real_g0 (dtp, p, kind, f->u.real.d);
2064 else
2065 write_d (dtp, f, p, kind);
2066 break;
2067 default:
2068 internal_error (&dtp->common,
2069 "formatted_transfer(): Bad type");
2071 break;
2073 case FMT_STRING:
2074 consume_data_flag = 0;
2075 write_constant_string (dtp, f);
2076 break;
2078 /* Format codes that don't transfer data. */
2079 case FMT_X:
2080 case FMT_TR:
2081 consume_data_flag = 0;
2083 dtp->u.p.skips += f->u.n;
2084 pos = bytes_used + dtp->u.p.skips - 1;
2085 dtp->u.p.pending_spaces = pos - dtp->u.p.max_pos + 1;
2086 /* Writes occur just before the switch on f->format, above, so
2087 that trailing blanks are suppressed, unless we are doing a
2088 non-advancing write in which case we want to output the blanks
2089 now. */
2090 if (dtp->u.p.advance_status == ADVANCE_NO)
2092 write_x (dtp, dtp->u.p.skips, dtp->u.p.pending_spaces);
2093 dtp->u.p.skips = dtp->u.p.pending_spaces = 0;
2095 break;
2097 case FMT_TL:
2098 case FMT_T:
2099 consume_data_flag = 0;
2101 if (f->format == FMT_TL)
2104 /* Handle the special case when no bytes have been used yet.
2105 Cannot go below zero. */
2106 if (bytes_used == 0)
2108 dtp->u.p.pending_spaces -= f->u.n;
2109 dtp->u.p.skips -= f->u.n;
2110 dtp->u.p.skips = dtp->u.p.skips < 0 ? 0 : dtp->u.p.skips;
2113 pos = bytes_used - f->u.n;
2115 else /* FMT_T */
2116 pos = f->u.n - dtp->u.p.pending_spaces - 1;
2118 /* Standard 10.6.1.1: excessive left tabbing is reset to the
2119 left tab limit. We do not check if the position has gone
2120 beyond the end of record because a subsequent tab could
2121 bring us back again. */
2122 pos = pos < 0 ? 0 : pos;
2124 dtp->u.p.skips = dtp->u.p.skips + pos - bytes_used;
2125 dtp->u.p.pending_spaces = dtp->u.p.pending_spaces
2126 + pos - dtp->u.p.max_pos;
2127 dtp->u.p.pending_spaces = dtp->u.p.pending_spaces < 0
2128 ? 0 : dtp->u.p.pending_spaces;
2129 break;
2131 case FMT_S:
2132 consume_data_flag = 0;
2133 dtp->u.p.sign_status = SIGN_S;
2134 break;
2136 case FMT_SS:
2137 consume_data_flag = 0;
2138 dtp->u.p.sign_status = SIGN_SS;
2139 break;
2141 case FMT_SP:
2142 consume_data_flag = 0;
2143 dtp->u.p.sign_status = SIGN_SP;
2144 break;
2146 case FMT_BN:
2147 consume_data_flag = 0 ;
2148 dtp->u.p.blank_status = BLANK_NULL;
2149 break;
2151 case FMT_BZ:
2152 consume_data_flag = 0;
2153 dtp->u.p.blank_status = BLANK_ZERO;
2154 break;
2156 case FMT_DC:
2157 consume_data_flag = 0;
2158 dtp->u.p.current_unit->decimal_status = DECIMAL_COMMA;
2159 break;
2161 case FMT_DP:
2162 consume_data_flag = 0;
2163 dtp->u.p.current_unit->decimal_status = DECIMAL_POINT;
2164 break;
2166 case FMT_RC:
2167 consume_data_flag = 0;
2168 dtp->u.p.current_unit->round_status = ROUND_COMPATIBLE;
2169 break;
2171 case FMT_RD:
2172 consume_data_flag = 0;
2173 dtp->u.p.current_unit->round_status = ROUND_DOWN;
2174 break;
2176 case FMT_RN:
2177 consume_data_flag = 0;
2178 dtp->u.p.current_unit->round_status = ROUND_NEAREST;
2179 break;
2181 case FMT_RP:
2182 consume_data_flag = 0;
2183 dtp->u.p.current_unit->round_status = ROUND_PROCDEFINED;
2184 break;
2186 case FMT_RU:
2187 consume_data_flag = 0;
2188 dtp->u.p.current_unit->round_status = ROUND_UP;
2189 break;
2191 case FMT_RZ:
2192 consume_data_flag = 0;
2193 dtp->u.p.current_unit->round_status = ROUND_ZERO;
2194 break;
2196 case FMT_P:
2197 consume_data_flag = 0;
2198 dtp->u.p.scale_factor = f->u.k;
2199 break;
2201 case FMT_DOLLAR:
2202 consume_data_flag = 0;
2203 dtp->u.p.seen_dollar = 1;
2204 break;
2206 case FMT_SLASH:
2207 consume_data_flag = 0;
2208 dtp->u.p.skips = dtp->u.p.pending_spaces = 0;
2209 next_record (dtp, 0);
2210 break;
2212 case FMT_COLON:
2213 /* A colon descriptor causes us to exit this loop (in
2214 particular preventing another / descriptor from being
2215 processed) unless there is another data item to be
2216 transferred. */
2217 consume_data_flag = 0;
2218 if (n == 0)
2219 return;
2220 break;
2222 default:
2223 internal_error (&dtp->common, "Bad format node");
2226 /* Adjust the item count and data pointer. */
2228 if ((consume_data_flag > 0) && (n > 0))
2230 n--;
2231 p = ((char *) p) + size;
2234 pos = dtp->u.p.current_unit->recl - dtp->u.p.current_unit->bytes_left;
2235 dtp->u.p.max_pos = (dtp->u.p.max_pos > pos) ? dtp->u.p.max_pos : pos;
2238 return;
2240 /* Come here when we need a data descriptor but don't have one. We
2241 push the current format node back onto the input, then return and
2242 let the user program call us back with the data. */
2243 need_data:
2244 unget_format (dtp, f);
2247 /* This function is first called from data_init_transfer to initiate the loop
2248 over each item in the format, transferring data as required. Subsequent
2249 calls to this function occur for each data item foound in the READ/WRITE
2250 statement. The item_count is incremented for each call. Since the first
2251 call is from data_transfer_init, the item_count is always one greater than
2252 the actual count number of the item being transferred. */
2254 static void
2255 formatted_transfer (st_parameter_dt *dtp, bt type, void *p, int kind,
2256 size_t size, size_t nelems)
2258 size_t elem;
2259 char *tmp;
2261 tmp = (char *) p;
2262 size_t stride = type == BT_CHARACTER ?
2263 size * GFC_SIZE_OF_CHAR_KIND(kind) : size;
2264 if (dtp->u.p.mode == READING)
2266 /* Big loop over all the elements. */
2267 for (elem = 0; elem < nelems; elem++)
2269 dtp->u.p.item_count++;
2270 formatted_transfer_scalar_read (dtp, type, tmp + stride*elem, kind, size);
2273 else
2275 /* Big loop over all the elements. */
2276 for (elem = 0; elem < nelems; elem++)
2278 dtp->u.p.item_count++;
2279 formatted_transfer_scalar_write (dtp, type, tmp + stride*elem, kind, size);
2285 /* Data transfer entry points. The type of the data entity is
2286 implicit in the subroutine call. This prevents us from having to
2287 share a common enum with the compiler. */
2289 void
2290 transfer_integer (st_parameter_dt *dtp, void *p, int kind)
2292 if ((dtp->common.flags & IOPARM_LIBRETURN_MASK) != IOPARM_LIBRETURN_OK)
2293 return;
2294 dtp->u.p.transfer (dtp, BT_INTEGER, p, kind, kind, 1);
2297 void
2298 transfer_integer_write (st_parameter_dt *dtp, void *p, int kind)
2300 transfer_integer (dtp, p, kind);
2303 void
2304 transfer_real (st_parameter_dt *dtp, void *p, int kind)
2306 size_t size;
2307 if ((dtp->common.flags & IOPARM_LIBRETURN_MASK) != IOPARM_LIBRETURN_OK)
2308 return;
2309 size = size_from_real_kind (kind);
2310 dtp->u.p.transfer (dtp, BT_REAL, p, kind, size, 1);
2313 void
2314 transfer_real_write (st_parameter_dt *dtp, void *p, int kind)
2316 transfer_real (dtp, p, kind);
2319 void
2320 transfer_logical (st_parameter_dt *dtp, void *p, int kind)
2322 if ((dtp->common.flags & IOPARM_LIBRETURN_MASK) != IOPARM_LIBRETURN_OK)
2323 return;
2324 dtp->u.p.transfer (dtp, BT_LOGICAL, p, kind, kind, 1);
2327 void
2328 transfer_logical_write (st_parameter_dt *dtp, void *p, int kind)
2330 transfer_logical (dtp, p, kind);
2333 void
2334 transfer_character (st_parameter_dt *dtp, void *p, gfc_charlen_type len)
2336 static char *empty_string[0];
2338 if ((dtp->common.flags & IOPARM_LIBRETURN_MASK) != IOPARM_LIBRETURN_OK)
2339 return;
2341 /* Strings of zero length can have p == NULL, which confuses the
2342 transfer routines into thinking we need more data elements. To avoid
2343 this, we give them a nice pointer. */
2344 if (len == 0 && p == NULL)
2345 p = empty_string;
2347 /* Set kind here to 1. */
2348 dtp->u.p.transfer (dtp, BT_CHARACTER, p, 1, len, 1);
2351 void
2352 transfer_character_write (st_parameter_dt *dtp, void *p, gfc_charlen_type len)
2354 transfer_character (dtp, p, len);
2357 void
2358 transfer_character_wide (st_parameter_dt *dtp, void *p, gfc_charlen_type len, int kind)
2360 static char *empty_string[0];
2362 if ((dtp->common.flags & IOPARM_LIBRETURN_MASK) != IOPARM_LIBRETURN_OK)
2363 return;
2365 /* Strings of zero length can have p == NULL, which confuses the
2366 transfer routines into thinking we need more data elements. To avoid
2367 this, we give them a nice pointer. */
2368 if (len == 0 && p == NULL)
2369 p = empty_string;
2371 /* Here we pass the actual kind value. */
2372 dtp->u.p.transfer (dtp, BT_CHARACTER, p, kind, len, 1);
2375 void
2376 transfer_character_wide_write (st_parameter_dt *dtp, void *p, gfc_charlen_type len, int kind)
2378 transfer_character_wide (dtp, p, len, kind);
2381 void
2382 transfer_complex (st_parameter_dt *dtp, void *p, int kind)
2384 size_t size;
2385 if ((dtp->common.flags & IOPARM_LIBRETURN_MASK) != IOPARM_LIBRETURN_OK)
2386 return;
2387 size = size_from_complex_kind (kind);
2388 dtp->u.p.transfer (dtp, BT_COMPLEX, p, kind, size, 1);
2391 void
2392 transfer_complex_write (st_parameter_dt *dtp, void *p, int kind)
2394 transfer_complex (dtp, p, kind);
2397 void
2398 transfer_array (st_parameter_dt *dtp, gfc_array_char *desc, int kind,
2399 gfc_charlen_type charlen)
2401 index_type count[GFC_MAX_DIMENSIONS];
2402 index_type extent[GFC_MAX_DIMENSIONS];
2403 index_type stride[GFC_MAX_DIMENSIONS];
2404 index_type stride0, rank, size, n;
2405 size_t tsize;
2406 char *data;
2407 bt iotype;
2409 /* Adjust item_count before emitting error message. */
2411 if ((dtp->common.flags & IOPARM_LIBRETURN_MASK) != IOPARM_LIBRETURN_OK)
2412 return;
2414 iotype = (bt) GFC_DESCRIPTOR_TYPE (desc);
2415 size = iotype == BT_CHARACTER ? (index_type) charlen : GFC_DESCRIPTOR_SIZE (desc);
2417 rank = GFC_DESCRIPTOR_RANK (desc);
2419 for (n = 0; n < rank; n++)
2421 count[n] = 0;
2422 stride[n] = GFC_DESCRIPTOR_STRIDE_BYTES(desc,n);
2423 extent[n] = GFC_DESCRIPTOR_EXTENT(desc,n);
2425 /* If the extent of even one dimension is zero, then the entire
2426 array section contains zero elements, so we return after writing
2427 a zero array record. */
2428 if (extent[n] <= 0)
2430 data = NULL;
2431 tsize = 0;
2432 dtp->u.p.transfer (dtp, iotype, data, kind, size, tsize);
2433 return;
2437 stride0 = stride[0];
2439 /* If the innermost dimension has a stride of 1, we can do the transfer
2440 in contiguous chunks. */
2441 if (stride0 == size)
2442 tsize = extent[0];
2443 else
2444 tsize = 1;
2446 data = GFC_DESCRIPTOR_DATA (desc);
2448 while (data)
2450 dtp->u.p.transfer (dtp, iotype, data, kind, size, tsize);
2451 data += stride0 * tsize;
2452 count[0] += tsize;
2453 n = 0;
2454 while (count[n] == extent[n])
2456 count[n] = 0;
2457 data -= stride[n] * extent[n];
2458 n++;
2459 if (n == rank)
2461 data = NULL;
2462 break;
2464 else
2466 count[n]++;
2467 data += stride[n];
2473 void
2474 transfer_array_write (st_parameter_dt *dtp, gfc_array_char *desc, int kind,
2475 gfc_charlen_type charlen)
2477 transfer_array (dtp, desc, kind, charlen);
2481 /* User defined input/output iomsg. */
2483 #define IOMSG_LEN 256
2485 void
2486 transfer_derived (st_parameter_dt *parent, void *dtio_source, void *dtio_proc)
2488 if (parent->u.p.current_unit)
2490 if (parent->u.p.current_unit->flags.form == FORM_UNFORMATTED)
2491 parent->u.p.ufdtio_ptr = (unformatted_dtio) dtio_proc;
2492 else
2493 parent->u.p.fdtio_ptr = (formatted_dtio) dtio_proc;
2495 parent->u.p.transfer (parent, BT_CLASS, dtio_source, 0, 0, 1);
2499 /* Preposition a sequential unformatted file while reading. */
2501 static void
2502 us_read (st_parameter_dt *dtp, int continued)
2504 ssize_t n, nr;
2505 GFC_INTEGER_4 i4;
2506 GFC_INTEGER_8 i8;
2507 gfc_offset i;
2509 if (compile_options.record_marker == 0)
2510 n = sizeof (GFC_INTEGER_4);
2511 else
2512 n = compile_options.record_marker;
2514 nr = sread (dtp->u.p.current_unit->s, &i, n);
2515 if (unlikely (nr < 0))
2517 generate_error (&dtp->common, LIBERROR_BAD_US, NULL);
2518 return;
2520 else if (nr == 0)
2522 hit_eof (dtp);
2523 return; /* end of file */
2525 else if (unlikely (n != nr))
2527 generate_error (&dtp->common, LIBERROR_BAD_US, NULL);
2528 return;
2531 /* Only GFC_CONVERT_NATIVE and GFC_CONVERT_SWAP are valid here. */
2532 if (likely (dtp->u.p.current_unit->flags.convert == GFC_CONVERT_NATIVE))
2534 switch (nr)
2536 case sizeof(GFC_INTEGER_4):
2537 memcpy (&i4, &i, sizeof (i4));
2538 i = i4;
2539 break;
2541 case sizeof(GFC_INTEGER_8):
2542 memcpy (&i8, &i, sizeof (i8));
2543 i = i8;
2544 break;
2546 default:
2547 runtime_error ("Illegal value for record marker");
2548 break;
2551 else
2553 uint32_t u32;
2554 uint64_t u64;
2555 switch (nr)
2557 case sizeof(GFC_INTEGER_4):
2558 memcpy (&u32, &i, sizeof (u32));
2559 u32 = __builtin_bswap32 (u32);
2560 memcpy (&i4, &u32, sizeof (i4));
2561 i = i4;
2562 break;
2564 case sizeof(GFC_INTEGER_8):
2565 memcpy (&u64, &i, sizeof (u64));
2566 u64 = __builtin_bswap64 (u64);
2567 memcpy (&i8, &u64, sizeof (i8));
2568 i = i8;
2569 break;
2571 default:
2572 runtime_error ("Illegal value for record marker");
2573 break;
2577 if (i >= 0)
2579 dtp->u.p.current_unit->bytes_left_subrecord = i;
2580 dtp->u.p.current_unit->continued = 0;
2582 else
2584 dtp->u.p.current_unit->bytes_left_subrecord = -i;
2585 dtp->u.p.current_unit->continued = 1;
2588 if (! continued)
2589 dtp->u.p.current_unit->bytes_left = dtp->u.p.current_unit->recl;
2593 /* Preposition a sequential unformatted file while writing. This
2594 amount to writing a bogus length that will be filled in later. */
2596 static void
2597 us_write (st_parameter_dt *dtp, int continued)
2599 ssize_t nbytes;
2600 gfc_offset dummy;
2602 dummy = 0;
2604 if (compile_options.record_marker == 0)
2605 nbytes = sizeof (GFC_INTEGER_4);
2606 else
2607 nbytes = compile_options.record_marker ;
2609 if (swrite (dtp->u.p.current_unit->s, &dummy, nbytes) != nbytes)
2610 generate_error (&dtp->common, LIBERROR_OS, NULL);
2612 /* For sequential unformatted, if RECL= was not specified in the OPEN
2613 we write until we have more bytes than can fit in the subrecord
2614 markers, then we write a new subrecord. */
2616 dtp->u.p.current_unit->bytes_left_subrecord =
2617 dtp->u.p.current_unit->recl_subrecord;
2618 dtp->u.p.current_unit->continued = continued;
2622 /* Position to the next record prior to transfer. We are assumed to
2623 be before the next record. We also calculate the bytes in the next
2624 record. */
2626 static void
2627 pre_position (st_parameter_dt *dtp)
2629 if (dtp->u.p.current_unit->current_record)
2630 return; /* Already positioned. */
2632 switch (current_mode (dtp))
2634 case FORMATTED_STREAM:
2635 case UNFORMATTED_STREAM:
2636 /* There are no records with stream I/O. If the position was specified
2637 data_transfer_init has already positioned the file. If no position
2638 was specified, we continue from where we last left off. I.e.
2639 there is nothing to do here. */
2640 break;
2642 case UNFORMATTED_SEQUENTIAL:
2643 if (dtp->u.p.mode == READING)
2644 us_read (dtp, 0);
2645 else
2646 us_write (dtp, 0);
2648 break;
2650 case FORMATTED_SEQUENTIAL:
2651 case FORMATTED_DIRECT:
2652 case UNFORMATTED_DIRECT:
2653 dtp->u.p.current_unit->bytes_left = dtp->u.p.current_unit->recl;
2654 break;
2657 dtp->u.p.current_unit->current_record = 1;
2661 /* Initialize things for a data transfer. This code is common for
2662 both reading and writing. */
2664 static void
2665 data_transfer_init (st_parameter_dt *dtp, int read_flag)
2667 unit_flags u_flags; /* Used for creating a unit if needed. */
2668 GFC_INTEGER_4 cf = dtp->common.flags;
2669 namelist_info *ionml;
2671 ionml = ((cf & IOPARM_DT_IONML_SET) != 0) ? dtp->u.p.ionml : NULL;
2673 memset (&dtp->u.p, 0, sizeof (dtp->u.p));
2675 dtp->u.p.ionml = ionml;
2676 dtp->u.p.mode = read_flag ? READING : WRITING;
2677 dtp->u.p.namelist_mode = 0;
2678 dtp->u.p.cc.len = 0;
2680 if ((dtp->common.flags & IOPARM_LIBRETURN_MASK) != IOPARM_LIBRETURN_OK)
2681 return;
2683 dtp->u.p.current_unit = get_unit (dtp, 1);
2685 if (dtp->u.p.current_unit == NULL)
2687 /* This means we tried to access an external unit < 0 without
2688 having opened it first with NEWUNIT=. */
2689 generate_error (&dtp->common, LIBERROR_BAD_OPTION,
2690 "Unit number is negative and unit was not already "
2691 "opened with OPEN(NEWUNIT=...)");
2692 return;
2694 else if (dtp->u.p.current_unit->s == NULL)
2695 { /* Open the unit with some default flags. */
2696 st_parameter_open opp;
2697 unit_convert conv;
2699 memset (&u_flags, '\0', sizeof (u_flags));
2700 u_flags.access = ACCESS_SEQUENTIAL;
2701 u_flags.action = ACTION_READWRITE;
2703 /* Is it unformatted? */
2704 if (!(cf & (IOPARM_DT_HAS_FORMAT | IOPARM_DT_LIST_FORMAT
2705 | IOPARM_DT_IONML_SET)))
2706 u_flags.form = FORM_UNFORMATTED;
2707 else
2708 u_flags.form = FORM_UNSPECIFIED;
2710 u_flags.delim = DELIM_UNSPECIFIED;
2711 u_flags.blank = BLANK_UNSPECIFIED;
2712 u_flags.pad = PAD_UNSPECIFIED;
2713 u_flags.decimal = DECIMAL_UNSPECIFIED;
2714 u_flags.encoding = ENCODING_UNSPECIFIED;
2715 u_flags.async = ASYNC_UNSPECIFIED;
2716 u_flags.round = ROUND_UNSPECIFIED;
2717 u_flags.sign = SIGN_UNSPECIFIED;
2718 u_flags.share = SHARE_UNSPECIFIED;
2719 u_flags.cc = CC_UNSPECIFIED;
2720 u_flags.readonly = 0;
2722 u_flags.status = STATUS_UNKNOWN;
2724 conv = get_unformatted_convert (dtp->common.unit);
2726 if (conv == GFC_CONVERT_NONE)
2727 conv = compile_options.convert;
2729 switch (conv)
2731 case GFC_CONVERT_NATIVE:
2732 case GFC_CONVERT_SWAP:
2733 break;
2735 case GFC_CONVERT_BIG:
2736 conv = __BYTE_ORDER__ == __ORDER_BIG_ENDIAN__ ? GFC_CONVERT_NATIVE : GFC_CONVERT_SWAP;
2737 break;
2739 case GFC_CONVERT_LITTLE:
2740 conv = __BYTE_ORDER__ == __ORDER_BIG_ENDIAN__ ? GFC_CONVERT_SWAP : GFC_CONVERT_NATIVE;
2741 break;
2743 default:
2744 internal_error (&opp.common, "Illegal value for CONVERT");
2745 break;
2748 u_flags.convert = conv;
2750 opp.common = dtp->common;
2751 opp.common.flags &= IOPARM_COMMON_MASK;
2752 dtp->u.p.current_unit = new_unit (&opp, dtp->u.p.current_unit, &u_flags);
2753 dtp->common.flags &= ~IOPARM_COMMON_MASK;
2754 dtp->common.flags |= (opp.common.flags & IOPARM_COMMON_MASK);
2755 if (dtp->u.p.current_unit == NULL)
2756 return;
2759 if (dtp->u.p.current_unit->child_dtio == 0)
2761 if ((cf & IOPARM_DT_HAS_SIZE) != 0)
2763 dtp->u.p.current_unit->has_size = true;
2764 /* Initialize the count. */
2765 dtp->u.p.current_unit->size_used = 0;
2767 else
2768 dtp->u.p.current_unit->has_size = false;
2770 else if (dtp->u.p.current_unit->internal_unit_kind > 0)
2771 dtp->u.p.unit_is_internal = 1;
2773 /* Check the action. */
2775 if (read_flag && dtp->u.p.current_unit->flags.action == ACTION_WRITE)
2777 generate_error (&dtp->common, LIBERROR_BAD_ACTION,
2778 "Cannot read from file opened for WRITE");
2779 return;
2782 if (!read_flag && dtp->u.p.current_unit->flags.action == ACTION_READ)
2784 generate_error (&dtp->common, LIBERROR_BAD_ACTION,
2785 "Cannot write to file opened for READ");
2786 return;
2789 dtp->u.p.first_item = 1;
2791 /* Check the format. */
2793 if ((cf & IOPARM_DT_HAS_FORMAT) != 0)
2794 parse_format (dtp);
2796 if (dtp->u.p.current_unit->flags.form == FORM_UNFORMATTED
2797 && (cf & (IOPARM_DT_HAS_FORMAT | IOPARM_DT_LIST_FORMAT))
2798 != 0)
2800 generate_error (&dtp->common, LIBERROR_OPTION_CONFLICT,
2801 "Format present for UNFORMATTED data transfer");
2802 return;
2805 if ((cf & IOPARM_DT_HAS_NAMELIST_NAME) != 0 && dtp->u.p.ionml != NULL)
2807 if ((cf & IOPARM_DT_HAS_FORMAT) != 0)
2809 generate_error (&dtp->common, LIBERROR_OPTION_CONFLICT,
2810 "A format cannot be specified with a namelist");
2811 return;
2814 else if (dtp->u.p.current_unit->flags.form == FORM_FORMATTED &&
2815 !(cf & (IOPARM_DT_HAS_FORMAT | IOPARM_DT_LIST_FORMAT)))
2817 generate_error (&dtp->common, LIBERROR_OPTION_CONFLICT,
2818 "Missing format for FORMATTED data transfer");
2819 return;
2822 if (is_internal_unit (dtp)
2823 && dtp->u.p.current_unit->flags.form == FORM_UNFORMATTED)
2825 generate_error (&dtp->common, LIBERROR_OPTION_CONFLICT,
2826 "Internal file cannot be accessed by UNFORMATTED "
2827 "data transfer");
2828 return;
2831 /* Check the record or position number. */
2833 if (dtp->u.p.current_unit->flags.access == ACCESS_DIRECT
2834 && (cf & IOPARM_DT_HAS_REC) == 0)
2836 generate_error (&dtp->common, LIBERROR_MISSING_OPTION,
2837 "Direct access data transfer requires record number");
2838 return;
2841 if (dtp->u.p.current_unit->flags.access == ACCESS_SEQUENTIAL)
2843 if ((cf & IOPARM_DT_HAS_REC) != 0)
2845 generate_error (&dtp->common, LIBERROR_OPTION_CONFLICT,
2846 "Record number not allowed for sequential access "
2847 "data transfer");
2848 return;
2851 if (compile_options.warn_std &&
2852 dtp->u.p.current_unit->endfile == AFTER_ENDFILE)
2854 generate_error (&dtp->common, LIBERROR_OPTION_CONFLICT,
2855 "Sequential READ or WRITE not allowed after "
2856 "EOF marker, possibly use REWIND or BACKSPACE");
2857 return;
2861 /* Process the ADVANCE option. */
2863 dtp->u.p.advance_status
2864 = !(cf & IOPARM_DT_HAS_ADVANCE) ? ADVANCE_UNSPECIFIED :
2865 find_option (&dtp->common, dtp->advance, dtp->advance_len, advance_opt,
2866 "Bad ADVANCE parameter in data transfer statement");
2868 if (dtp->u.p.advance_status != ADVANCE_UNSPECIFIED)
2870 if (dtp->u.p.current_unit->flags.access == ACCESS_DIRECT)
2872 generate_error (&dtp->common, LIBERROR_OPTION_CONFLICT,
2873 "ADVANCE specification conflicts with sequential "
2874 "access");
2875 return;
2878 if (is_internal_unit (dtp))
2880 generate_error (&dtp->common, LIBERROR_OPTION_CONFLICT,
2881 "ADVANCE specification conflicts with internal file");
2882 return;
2885 if ((cf & (IOPARM_DT_HAS_FORMAT | IOPARM_DT_LIST_FORMAT))
2886 != IOPARM_DT_HAS_FORMAT)
2888 generate_error (&dtp->common, LIBERROR_OPTION_CONFLICT,
2889 "ADVANCE specification requires an explicit format");
2890 return;
2894 /* Child IO is non-advancing and any ADVANCE= specifier is ignored.
2895 F2008 9.6.2.4 */
2896 if (dtp->u.p.current_unit->child_dtio > 0)
2897 dtp->u.p.advance_status = ADVANCE_NO;
2899 if (read_flag)
2901 dtp->u.p.current_unit->previous_nonadvancing_write = 0;
2903 if ((cf & IOPARM_EOR) != 0 && dtp->u.p.advance_status != ADVANCE_NO)
2905 generate_error (&dtp->common, LIBERROR_MISSING_OPTION,
2906 "EOR specification requires an ADVANCE specification "
2907 "of NO");
2908 return;
2911 if ((cf & IOPARM_DT_HAS_SIZE) != 0
2912 && dtp->u.p.advance_status != ADVANCE_NO)
2914 generate_error (&dtp->common, LIBERROR_MISSING_OPTION,
2915 "SIZE specification requires an ADVANCE "
2916 "specification of NO");
2917 return;
2920 else
2921 { /* Write constraints. */
2922 if ((cf & IOPARM_END) != 0)
2924 generate_error (&dtp->common, LIBERROR_OPTION_CONFLICT,
2925 "END specification cannot appear in a write "
2926 "statement");
2927 return;
2930 if ((cf & IOPARM_EOR) != 0)
2932 generate_error (&dtp->common, LIBERROR_OPTION_CONFLICT,
2933 "EOR specification cannot appear in a write "
2934 "statement");
2935 return;
2938 if ((cf & IOPARM_DT_HAS_SIZE) != 0)
2940 generate_error (&dtp->common, LIBERROR_OPTION_CONFLICT,
2941 "SIZE specification cannot appear in a write "
2942 "statement");
2943 return;
2947 if (dtp->u.p.advance_status == ADVANCE_UNSPECIFIED)
2948 dtp->u.p.advance_status = ADVANCE_YES;
2950 /* Check the decimal mode. */
2951 dtp->u.p.current_unit->decimal_status
2952 = !(cf & IOPARM_DT_HAS_DECIMAL) ? DECIMAL_UNSPECIFIED :
2953 find_option (&dtp->common, dtp->decimal, dtp->decimal_len,
2954 decimal_opt, "Bad DECIMAL parameter in data transfer "
2955 "statement");
2957 if (dtp->u.p.current_unit->decimal_status == DECIMAL_UNSPECIFIED)
2958 dtp->u.p.current_unit->decimal_status = dtp->u.p.current_unit->flags.decimal;
2960 /* Check the round mode. */
2961 dtp->u.p.current_unit->round_status
2962 = !(cf & IOPARM_DT_HAS_ROUND) ? ROUND_UNSPECIFIED :
2963 find_option (&dtp->common, dtp->round, dtp->round_len,
2964 round_opt, "Bad ROUND parameter in data transfer "
2965 "statement");
2967 if (dtp->u.p.current_unit->round_status == ROUND_UNSPECIFIED)
2968 dtp->u.p.current_unit->round_status = dtp->u.p.current_unit->flags.round;
2970 /* Check the sign mode. */
2971 dtp->u.p.sign_status
2972 = !(cf & IOPARM_DT_HAS_SIGN) ? SIGN_UNSPECIFIED :
2973 find_option (&dtp->common, dtp->sign, dtp->sign_len, sign_opt,
2974 "Bad SIGN parameter in data transfer statement");
2976 if (dtp->u.p.sign_status == SIGN_UNSPECIFIED)
2977 dtp->u.p.sign_status = dtp->u.p.current_unit->flags.sign;
2979 /* Check the blank mode. */
2980 dtp->u.p.blank_status
2981 = !(cf & IOPARM_DT_HAS_BLANK) ? BLANK_UNSPECIFIED :
2982 find_option (&dtp->common, dtp->blank, dtp->blank_len,
2983 blank_opt,
2984 "Bad BLANK parameter in data transfer statement");
2986 if (dtp->u.p.blank_status == BLANK_UNSPECIFIED)
2987 dtp->u.p.blank_status = dtp->u.p.current_unit->flags.blank;
2989 /* Check the delim mode. */
2990 dtp->u.p.current_unit->delim_status
2991 = !(cf & IOPARM_DT_HAS_DELIM) ? DELIM_UNSPECIFIED :
2992 find_option (&dtp->common, dtp->delim, dtp->delim_len,
2993 delim_opt, "Bad DELIM parameter in data transfer statement");
2995 if (dtp->u.p.current_unit->delim_status == DELIM_UNSPECIFIED)
2997 if (ionml && dtp->u.p.current_unit->flags.delim == DELIM_UNSPECIFIED)
2998 dtp->u.p.current_unit->delim_status = DELIM_QUOTE;
2999 else
3000 dtp->u.p.current_unit->delim_status = dtp->u.p.current_unit->flags.delim;
3003 /* Check the pad mode. */
3004 dtp->u.p.current_unit->pad_status
3005 = !(cf & IOPARM_DT_HAS_PAD) ? PAD_UNSPECIFIED :
3006 find_option (&dtp->common, dtp->pad, dtp->pad_len, pad_opt,
3007 "Bad PAD parameter in data transfer statement");
3009 if (dtp->u.p.current_unit->pad_status == PAD_UNSPECIFIED)
3010 dtp->u.p.current_unit->pad_status = dtp->u.p.current_unit->flags.pad;
3012 /* Check to see if we might be reading what we wrote before */
3014 if (dtp->u.p.mode != dtp->u.p.current_unit->mode
3015 && !is_internal_unit (dtp))
3017 int pos = fbuf_reset (dtp->u.p.current_unit);
3018 if (pos != 0)
3019 sseek (dtp->u.p.current_unit->s, pos, SEEK_CUR);
3020 sflush(dtp->u.p.current_unit->s);
3023 /* Check the POS= specifier: that it is in range and that it is used with a
3024 unit that has been connected for STREAM access. F2003 9.5.1.10. */
3026 if (((cf & IOPARM_DT_HAS_POS) != 0))
3028 if (is_stream_io (dtp))
3031 if (dtp->pos <= 0)
3033 generate_error (&dtp->common, LIBERROR_BAD_OPTION,
3034 "POS=specifier must be positive");
3035 return;
3038 if (dtp->pos >= dtp->u.p.current_unit->maxrec)
3040 generate_error (&dtp->common, LIBERROR_BAD_OPTION,
3041 "POS=specifier too large");
3042 return;
3045 dtp->rec = dtp->pos;
3047 if (dtp->u.p.mode == READING)
3049 /* Reset the endfile flag; if we hit EOF during reading
3050 we'll set the flag and generate an error at that point
3051 rather than worrying about it here. */
3052 dtp->u.p.current_unit->endfile = NO_ENDFILE;
3055 if (dtp->pos != dtp->u.p.current_unit->strm_pos)
3057 fbuf_flush (dtp->u.p.current_unit, dtp->u.p.mode);
3058 if (sseek (dtp->u.p.current_unit->s, dtp->pos - 1, SEEK_SET) < 0)
3060 generate_error (&dtp->common, LIBERROR_OS, NULL);
3061 return;
3063 dtp->u.p.current_unit->strm_pos = dtp->pos;
3066 else
3068 generate_error (&dtp->common, LIBERROR_BAD_OPTION,
3069 "POS=specifier not allowed, "
3070 "Try OPEN with ACCESS='stream'");
3071 return;
3076 /* Sanity checks on the record number. */
3077 if ((cf & IOPARM_DT_HAS_REC) != 0)
3079 if (dtp->rec <= 0)
3081 generate_error (&dtp->common, LIBERROR_BAD_OPTION,
3082 "Record number must be positive");
3083 return;
3086 if (dtp->rec >= dtp->u.p.current_unit->maxrec)
3088 generate_error (&dtp->common, LIBERROR_BAD_OPTION,
3089 "Record number too large");
3090 return;
3093 /* Make sure format buffer is reset. */
3094 if (dtp->u.p.current_unit->flags.form == FORM_FORMATTED)
3095 fbuf_reset (dtp->u.p.current_unit);
3098 /* Check whether the record exists to be read. Only
3099 a partial record needs to exist. */
3101 if (dtp->u.p.mode == READING && (dtp->rec - 1)
3102 * dtp->u.p.current_unit->recl >= ssize (dtp->u.p.current_unit->s))
3104 generate_error (&dtp->common, LIBERROR_BAD_OPTION,
3105 "Non-existing record number");
3106 return;
3109 /* Position the file. */
3110 if (sseek (dtp->u.p.current_unit->s, (gfc_offset) (dtp->rec - 1)
3111 * dtp->u.p.current_unit->recl, SEEK_SET) < 0)
3113 generate_error (&dtp->common, LIBERROR_OS, NULL);
3114 return;
3117 if (dtp->u.p.current_unit->flags.access == ACCESS_STREAM)
3119 generate_error (&dtp->common, LIBERROR_OPTION_CONFLICT,
3120 "Record number not allowed for stream access "
3121 "data transfer");
3122 return;
3126 /* Bugware for badly written mixed C-Fortran I/O. */
3127 if (!is_internal_unit (dtp))
3128 flush_if_preconnected(dtp->u.p.current_unit->s);
3130 dtp->u.p.current_unit->mode = dtp->u.p.mode;
3132 /* Set the maximum position reached from the previous I/O operation. This
3133 could be greater than zero from a previous non-advancing write. */
3134 dtp->u.p.max_pos = dtp->u.p.current_unit->saved_pos;
3136 pre_position (dtp);
3139 /* Set up the subroutine that will handle the transfers. */
3141 if (read_flag)
3143 if (dtp->u.p.current_unit->flags.form == FORM_UNFORMATTED)
3144 dtp->u.p.transfer = unformatted_read;
3145 else
3147 if ((cf & IOPARM_DT_LIST_FORMAT) != 0)
3149 if (dtp->u.p.current_unit->child_dtio == 0)
3150 dtp->u.p.current_unit->last_char = EOF - 1;
3151 dtp->u.p.transfer = list_formatted_read;
3153 else
3154 dtp->u.p.transfer = formatted_transfer;
3157 else
3159 if (dtp->u.p.current_unit->flags.form == FORM_UNFORMATTED)
3160 dtp->u.p.transfer = unformatted_write;
3161 else
3163 if ((cf & IOPARM_DT_LIST_FORMAT) != 0)
3164 dtp->u.p.transfer = list_formatted_write;
3165 else
3166 dtp->u.p.transfer = formatted_transfer;
3170 /* Make sure that we don't do a read after a nonadvancing write. */
3172 if (read_flag)
3174 if (dtp->u.p.current_unit->read_bad && !is_stream_io (dtp))
3176 generate_error (&dtp->common, LIBERROR_BAD_OPTION,
3177 "Cannot READ after a nonadvancing WRITE");
3178 return;
3181 else
3183 if (dtp->u.p.advance_status == ADVANCE_YES && !dtp->u.p.seen_dollar)
3184 dtp->u.p.current_unit->read_bad = 1;
3187 if (dtp->u.p.current_unit->flags.form == FORM_FORMATTED)
3189 #ifdef HAVE_USELOCALE
3190 dtp->u.p.old_locale = uselocale (c_locale);
3191 #else
3192 __gthread_mutex_lock (&old_locale_lock);
3193 if (!old_locale_ctr++)
3195 old_locale = setlocale (LC_NUMERIC, NULL);
3196 setlocale (LC_NUMERIC, "C");
3198 __gthread_mutex_unlock (&old_locale_lock);
3199 #endif
3200 /* Start the data transfer if we are doing a formatted transfer. */
3201 if ((cf & (IOPARM_DT_LIST_FORMAT | IOPARM_DT_HAS_NAMELIST_NAME)) == 0
3202 && dtp->u.p.ionml == NULL)
3203 formatted_transfer (dtp, 0, NULL, 0, 0, 1);
3208 /* Initialize an array_loop_spec given the array descriptor. The function
3209 returns the index of the last element of the array, and also returns
3210 starting record, where the first I/O goes to (necessary in case of
3211 negative strides). */
3213 gfc_offset
3214 init_loop_spec (gfc_array_char *desc, array_loop_spec *ls,
3215 gfc_offset *start_record)
3217 int rank = GFC_DESCRIPTOR_RANK(desc);
3218 int i;
3219 gfc_offset index;
3220 int empty;
3222 empty = 0;
3223 index = 1;
3224 *start_record = 0;
3226 for (i=0; i<rank; i++)
3228 ls[i].idx = GFC_DESCRIPTOR_LBOUND(desc,i);
3229 ls[i].start = GFC_DESCRIPTOR_LBOUND(desc,i);
3230 ls[i].end = GFC_DESCRIPTOR_UBOUND(desc,i);
3231 ls[i].step = GFC_DESCRIPTOR_STRIDE(desc,i);
3232 empty = empty || (GFC_DESCRIPTOR_UBOUND(desc,i)
3233 < GFC_DESCRIPTOR_LBOUND(desc,i));
3235 if (GFC_DESCRIPTOR_STRIDE(desc,i) > 0)
3237 index += (GFC_DESCRIPTOR_EXTENT(desc,i) - 1)
3238 * GFC_DESCRIPTOR_STRIDE(desc,i);
3240 else
3242 index -= (GFC_DESCRIPTOR_EXTENT(desc,i) - 1)
3243 * GFC_DESCRIPTOR_STRIDE(desc,i);
3244 *start_record -= (GFC_DESCRIPTOR_EXTENT(desc,i) - 1)
3245 * GFC_DESCRIPTOR_STRIDE(desc,i);
3249 if (empty)
3250 return 0;
3251 else
3252 return index;
3255 /* Determine the index to the next record in an internal unit array by
3256 by incrementing through the array_loop_spec. */
3258 gfc_offset
3259 next_array_record (st_parameter_dt *dtp, array_loop_spec *ls, int *finished)
3261 int i, carry;
3262 gfc_offset index;
3264 carry = 1;
3265 index = 0;
3267 for (i = 0; i < dtp->u.p.current_unit->rank; i++)
3269 if (carry)
3271 ls[i].idx++;
3272 if (ls[i].idx > ls[i].end)
3274 ls[i].idx = ls[i].start;
3275 carry = 1;
3277 else
3278 carry = 0;
3280 index = index + (ls[i].idx - ls[i].start) * ls[i].step;
3283 *finished = carry;
3285 return index;
3290 /* Skip to the end of the current record, taking care of an optional
3291 record marker of size bytes. If the file is not seekable, we
3292 read chunks of size MAX_READ until we get to the right
3293 position. */
3295 static void
3296 skip_record (st_parameter_dt *dtp, gfc_offset bytes)
3298 ssize_t rlength, readb;
3299 #define MAX_READ 4096
3300 char p[MAX_READ];
3302 dtp->u.p.current_unit->bytes_left_subrecord += bytes;
3303 if (dtp->u.p.current_unit->bytes_left_subrecord == 0)
3304 return;
3306 /* Direct access files do not generate END conditions,
3307 only I/O errors. */
3308 if (sseek (dtp->u.p.current_unit->s,
3309 dtp->u.p.current_unit->bytes_left_subrecord, SEEK_CUR) < 0)
3311 /* Seeking failed, fall back to seeking by reading data. */
3312 while (dtp->u.p.current_unit->bytes_left_subrecord > 0)
3314 rlength =
3315 (MAX_READ < dtp->u.p.current_unit->bytes_left_subrecord) ?
3316 MAX_READ : dtp->u.p.current_unit->bytes_left_subrecord;
3318 readb = sread (dtp->u.p.current_unit->s, p, rlength);
3319 if (readb < 0)
3321 generate_error (&dtp->common, LIBERROR_OS, NULL);
3322 return;
3325 dtp->u.p.current_unit->bytes_left_subrecord -= readb;
3327 return;
3329 dtp->u.p.current_unit->bytes_left_subrecord = 0;
3333 /* Advance to the next record reading unformatted files, taking
3334 care of subrecords. If complete_record is nonzero, we loop
3335 until all subrecords are cleared. */
3337 static void
3338 next_record_r_unf (st_parameter_dt *dtp, int complete_record)
3340 size_t bytes;
3342 bytes = compile_options.record_marker == 0 ?
3343 sizeof (GFC_INTEGER_4) : compile_options.record_marker;
3345 while(1)
3348 /* Skip over tail */
3350 skip_record (dtp, bytes);
3352 if ( ! (complete_record && dtp->u.p.current_unit->continued))
3353 return;
3355 us_read (dtp, 1);
3360 static gfc_offset
3361 min_off (gfc_offset a, gfc_offset b)
3363 return (a < b ? a : b);
3367 /* Space to the next record for read mode. */
3369 static void
3370 next_record_r (st_parameter_dt *dtp, int done)
3372 gfc_offset record;
3373 char p;
3374 int cc;
3376 switch (current_mode (dtp))
3378 /* No records in unformatted STREAM I/O. */
3379 case UNFORMATTED_STREAM:
3380 return;
3382 case UNFORMATTED_SEQUENTIAL:
3383 next_record_r_unf (dtp, 1);
3384 dtp->u.p.current_unit->bytes_left = dtp->u.p.current_unit->recl;
3385 break;
3387 case FORMATTED_DIRECT:
3388 case UNFORMATTED_DIRECT:
3389 skip_record (dtp, dtp->u.p.current_unit->bytes_left);
3390 break;
3392 case FORMATTED_STREAM:
3393 case FORMATTED_SEQUENTIAL:
3394 /* read_sf has already terminated input because of an '\n', or
3395 we have hit EOF. */
3396 if (dtp->u.p.sf_seen_eor)
3398 dtp->u.p.sf_seen_eor = 0;
3399 break;
3402 if (is_internal_unit (dtp))
3404 if (is_array_io (dtp))
3406 int finished;
3408 record = next_array_record (dtp, dtp->u.p.current_unit->ls,
3409 &finished);
3410 if (!done && finished)
3411 hit_eof (dtp);
3413 /* Now seek to this record. */
3414 record = record * dtp->u.p.current_unit->recl;
3415 if (sseek (dtp->u.p.current_unit->s, record, SEEK_SET) < 0)
3417 generate_error (&dtp->common, LIBERROR_INTERNAL_UNIT, NULL);
3418 break;
3420 dtp->u.p.current_unit->bytes_left = dtp->u.p.current_unit->recl;
3422 else
3424 gfc_offset bytes_left = dtp->u.p.current_unit->bytes_left;
3425 bytes_left = min_off (bytes_left,
3426 ssize (dtp->u.p.current_unit->s)
3427 - stell (dtp->u.p.current_unit->s));
3428 if (sseek (dtp->u.p.current_unit->s,
3429 bytes_left, SEEK_CUR) < 0)
3431 generate_error (&dtp->common, LIBERROR_INTERNAL_UNIT, NULL);
3432 break;
3434 dtp->u.p.current_unit->bytes_left
3435 = dtp->u.p.current_unit->recl;
3437 break;
3439 else if (dtp->u.p.current_unit->flags.cc != CC_NONE)
3443 errno = 0;
3444 cc = fbuf_getc (dtp->u.p.current_unit);
3445 if (cc == EOF)
3447 if (errno != 0)
3448 generate_error (&dtp->common, LIBERROR_OS, NULL);
3449 else
3451 if (is_stream_io (dtp)
3452 || dtp->u.p.current_unit->pad_status == PAD_NO
3453 || dtp->u.p.current_unit->bytes_left
3454 == dtp->u.p.current_unit->recl)
3455 hit_eof (dtp);
3457 break;
3460 if (is_stream_io (dtp))
3461 dtp->u.p.current_unit->strm_pos++;
3463 p = (char) cc;
3465 while (p != '\n');
3467 break;
3472 /* Small utility function to write a record marker, taking care of
3473 byte swapping and of choosing the correct size. */
3475 static int
3476 write_us_marker (st_parameter_dt *dtp, const gfc_offset buf)
3478 size_t len;
3479 GFC_INTEGER_4 buf4;
3480 GFC_INTEGER_8 buf8;
3482 if (compile_options.record_marker == 0)
3483 len = sizeof (GFC_INTEGER_4);
3484 else
3485 len = compile_options.record_marker;
3487 /* Only GFC_CONVERT_NATIVE and GFC_CONVERT_SWAP are valid here. */
3488 if (likely (dtp->u.p.current_unit->flags.convert == GFC_CONVERT_NATIVE))
3490 switch (len)
3492 case sizeof (GFC_INTEGER_4):
3493 buf4 = buf;
3494 return swrite (dtp->u.p.current_unit->s, &buf4, len);
3495 break;
3497 case sizeof (GFC_INTEGER_8):
3498 buf8 = buf;
3499 return swrite (dtp->u.p.current_unit->s, &buf8, len);
3500 break;
3502 default:
3503 runtime_error ("Illegal value for record marker");
3504 break;
3507 else
3509 uint32_t u32;
3510 uint64_t u64;
3511 switch (len)
3513 case sizeof (GFC_INTEGER_4):
3514 buf4 = buf;
3515 memcpy (&u32, &buf4, sizeof (u32));
3516 u32 = __builtin_bswap32 (u32);
3517 return swrite (dtp->u.p.current_unit->s, &u32, len);
3518 break;
3520 case sizeof (GFC_INTEGER_8):
3521 buf8 = buf;
3522 memcpy (&u64, &buf8, sizeof (u64));
3523 u64 = __builtin_bswap64 (u64);
3524 return swrite (dtp->u.p.current_unit->s, &u64, len);
3525 break;
3527 default:
3528 runtime_error ("Illegal value for record marker");
3529 break;
3535 /* Position to the next (sub)record in write mode for
3536 unformatted sequential files. */
3538 static void
3539 next_record_w_unf (st_parameter_dt *dtp, int next_subrecord)
3541 gfc_offset m, m_write, record_marker;
3543 /* Bytes written. */
3544 m = dtp->u.p.current_unit->recl_subrecord
3545 - dtp->u.p.current_unit->bytes_left_subrecord;
3547 if (compile_options.record_marker == 0)
3548 record_marker = sizeof (GFC_INTEGER_4);
3549 else
3550 record_marker = compile_options.record_marker;
3552 /* Seek to the head and overwrite the bogus length with the real
3553 length. */
3555 if (unlikely (sseek (dtp->u.p.current_unit->s, - m - record_marker,
3556 SEEK_CUR) < 0))
3557 goto io_error;
3559 if (next_subrecord)
3560 m_write = -m;
3561 else
3562 m_write = m;
3564 if (unlikely (write_us_marker (dtp, m_write) < 0))
3565 goto io_error;
3567 /* Seek past the end of the current record. */
3569 if (unlikely (sseek (dtp->u.p.current_unit->s, m, SEEK_CUR) < 0))
3570 goto io_error;
3572 /* Write the length tail. If we finish a record containing
3573 subrecords, we write out the negative length. */
3575 if (dtp->u.p.current_unit->continued)
3576 m_write = -m;
3577 else
3578 m_write = m;
3580 if (unlikely (write_us_marker (dtp, m_write) < 0))
3581 goto io_error;
3583 return;
3585 io_error:
3586 generate_error (&dtp->common, LIBERROR_OS, NULL);
3587 return;
3592 /* Utility function like memset() but operating on streams. Return
3593 value is same as for POSIX write(). */
3595 static gfc_offset
3596 sset (stream *s, int c, gfc_offset nbyte)
3598 #define WRITE_CHUNK 256
3599 char p[WRITE_CHUNK];
3600 gfc_offset bytes_left;
3601 ssize_t trans;
3603 if (nbyte < WRITE_CHUNK)
3604 memset (p, c, nbyte);
3605 else
3606 memset (p, c, WRITE_CHUNK);
3608 bytes_left = nbyte;
3609 while (bytes_left > 0)
3611 trans = (bytes_left < WRITE_CHUNK) ? bytes_left : WRITE_CHUNK;
3612 trans = swrite (s, p, trans);
3613 if (trans <= 0)
3614 return trans;
3615 bytes_left -= trans;
3618 return nbyte - bytes_left;
3622 /* Finish up a record according to the legacy carriagecontrol type, based
3623 on the first character in the record. */
3625 static void
3626 next_record_cc (st_parameter_dt *dtp)
3628 /* Only valid with CARRIAGECONTROL=FORTRAN. */
3629 if (dtp->u.p.current_unit->flags.cc != CC_FORTRAN)
3630 return;
3632 fbuf_seek (dtp->u.p.current_unit, 0, SEEK_END);
3633 if (dtp->u.p.cc.len > 0)
3635 char *p = fbuf_alloc (dtp->u.p.current_unit, dtp->u.p.cc.len);
3636 if (!p)
3637 generate_error (&dtp->common, LIBERROR_OS, NULL);
3639 /* Output CR for the first character with default CC setting. */
3640 *(p++) = dtp->u.p.cc.u.end;
3641 if (dtp->u.p.cc.len > 1)
3642 *p = dtp->u.p.cc.u.end;
3646 /* Position to the next record in write mode. */
3648 static void
3649 next_record_w (st_parameter_dt *dtp, int done)
3651 gfc_offset max_pos_off;
3653 /* Zero counters for X- and T-editing. */
3654 max_pos_off = dtp->u.p.max_pos;
3655 dtp->u.p.max_pos = dtp->u.p.skips = dtp->u.p.pending_spaces = 0;
3657 switch (current_mode (dtp))
3659 /* No records in unformatted STREAM I/O. */
3660 case UNFORMATTED_STREAM:
3661 return;
3663 case FORMATTED_DIRECT:
3664 if (dtp->u.p.current_unit->bytes_left == 0)
3665 break;
3667 fbuf_seek (dtp->u.p.current_unit, 0, SEEK_END);
3668 fbuf_flush (dtp->u.p.current_unit, WRITING);
3669 if (sset (dtp->u.p.current_unit->s, ' ',
3670 dtp->u.p.current_unit->bytes_left)
3671 != dtp->u.p.current_unit->bytes_left)
3672 goto io_error;
3674 break;
3676 case UNFORMATTED_DIRECT:
3677 if (dtp->u.p.current_unit->bytes_left > 0)
3679 gfc_offset length = dtp->u.p.current_unit->bytes_left;
3680 if (sset (dtp->u.p.current_unit->s, 0, length) != length)
3681 goto io_error;
3683 break;
3685 case UNFORMATTED_SEQUENTIAL:
3686 next_record_w_unf (dtp, 0);
3687 dtp->u.p.current_unit->bytes_left = dtp->u.p.current_unit->recl;
3688 break;
3690 case FORMATTED_STREAM:
3691 case FORMATTED_SEQUENTIAL:
3693 if (is_internal_unit (dtp))
3695 char *p;
3696 /* Internal unit, so must fit in memory. */
3697 size_t length, m;
3698 size_t max_pos = max_pos_off;
3699 if (is_array_io (dtp))
3701 int finished;
3703 length = dtp->u.p.current_unit->bytes_left;
3705 /* If the farthest position reached is greater than current
3706 position, adjust the position and set length to pad out
3707 whats left. Otherwise just pad whats left.
3708 (for character array unit) */
3709 m = dtp->u.p.current_unit->recl
3710 - dtp->u.p.current_unit->bytes_left;
3711 if (max_pos > m)
3713 length = (max_pos - m);
3714 if (sseek (dtp->u.p.current_unit->s,
3715 length, SEEK_CUR) < 0)
3717 generate_error (&dtp->common, LIBERROR_INTERNAL_UNIT, NULL);
3718 return;
3720 length = ((size_t) dtp->u.p.current_unit->recl - max_pos);
3723 p = write_block (dtp, length);
3724 if (p == NULL)
3725 return;
3727 if (unlikely (is_char4_unit (dtp)))
3729 gfc_char4_t *p4 = (gfc_char4_t *) p;
3730 memset4 (p4, ' ', length);
3732 else
3733 memset (p, ' ', length);
3735 /* Now that the current record has been padded out,
3736 determine where the next record in the array is.
3737 Note that this can return a negative value, so it
3738 needs to be assigned to a signed value. */
3739 gfc_offset record = next_array_record
3740 (dtp, dtp->u.p.current_unit->ls, &finished);
3741 if (finished)
3742 dtp->u.p.current_unit->endfile = AT_ENDFILE;
3744 /* Now seek to this record */
3745 record = record * dtp->u.p.current_unit->recl;
3747 if (sseek (dtp->u.p.current_unit->s, record, SEEK_SET) < 0)
3749 generate_error (&dtp->common, LIBERROR_INTERNAL_UNIT, NULL);
3750 return;
3753 dtp->u.p.current_unit->bytes_left = dtp->u.p.current_unit->recl;
3755 else
3757 length = 1;
3759 /* If this is the last call to next_record move to the farthest
3760 position reached and set length to pad out the remainder
3761 of the record. (for character scaler unit) */
3762 if (done)
3764 m = dtp->u.p.current_unit->recl
3765 - dtp->u.p.current_unit->bytes_left;
3766 if (max_pos > m)
3768 length = max_pos - m;
3769 if (sseek (dtp->u.p.current_unit->s,
3770 length, SEEK_CUR) < 0)
3772 generate_error (&dtp->common, LIBERROR_INTERNAL_UNIT, NULL);
3773 return;
3775 length = (size_t) dtp->u.p.current_unit->recl
3776 - max_pos;
3778 else
3779 length = dtp->u.p.current_unit->bytes_left;
3781 if (length > 0)
3783 p = write_block (dtp, length);
3784 if (p == NULL)
3785 return;
3787 if (unlikely (is_char4_unit (dtp)))
3789 gfc_char4_t *p4 = (gfc_char4_t *) p;
3790 memset4 (p4, (gfc_char4_t) ' ', length);
3792 else
3793 memset (p, ' ', length);
3797 /* Handle legacy CARRIAGECONTROL line endings. */
3798 else if (dtp->u.p.current_unit->flags.cc == CC_FORTRAN)
3799 next_record_cc (dtp);
3800 else
3802 /* Skip newlines for CC=CC_NONE. */
3803 const int len = (dtp->u.p.current_unit->flags.cc == CC_NONE)
3805 #ifdef HAVE_CRLF
3806 : 2;
3807 #else
3808 : 1;
3809 #endif
3810 fbuf_seek (dtp->u.p.current_unit, 0, SEEK_END);
3811 if (dtp->u.p.current_unit->flags.cc != CC_NONE)
3813 char *p = fbuf_alloc (dtp->u.p.current_unit, len);
3814 if (!p)
3815 goto io_error;
3816 #ifdef HAVE_CRLF
3817 *(p++) = '\r';
3818 #endif
3819 *p = '\n';
3821 if (is_stream_io (dtp))
3823 dtp->u.p.current_unit->strm_pos += len;
3824 if (dtp->u.p.current_unit->strm_pos
3825 < ssize (dtp->u.p.current_unit->s))
3826 unit_truncate (dtp->u.p.current_unit,
3827 dtp->u.p.current_unit->strm_pos - 1,
3828 &dtp->common);
3832 break;
3834 io_error:
3835 generate_error (&dtp->common, LIBERROR_OS, NULL);
3836 break;
3840 /* Position to the next record, which means moving to the end of the
3841 current record. This can happen under several different
3842 conditions. If the done flag is not set, we get ready to process
3843 the next record. */
3845 void
3846 next_record (st_parameter_dt *dtp, int done)
3848 gfc_offset fp; /* File position. */
3850 dtp->u.p.current_unit->read_bad = 0;
3852 if (dtp->u.p.mode == READING)
3853 next_record_r (dtp, done);
3854 else
3855 next_record_w (dtp, done);
3857 fbuf_flush (dtp->u.p.current_unit, dtp->u.p.mode);
3859 if (!is_stream_io (dtp))
3861 /* Since we have changed the position, set it to unspecified so
3862 that INQUIRE(POSITION=) knows it needs to look into it. */
3863 if (done)
3864 dtp->u.p.current_unit->flags.position = POSITION_UNSPECIFIED;
3866 dtp->u.p.current_unit->current_record = 0;
3867 if (dtp->u.p.current_unit->flags.access == ACCESS_DIRECT)
3869 fp = stell (dtp->u.p.current_unit->s);
3870 /* Calculate next record, rounding up partial records. */
3871 dtp->u.p.current_unit->last_record =
3872 (fp + dtp->u.p.current_unit->recl) /
3873 dtp->u.p.current_unit->recl - 1;
3875 else
3876 dtp->u.p.current_unit->last_record++;
3879 if (!done)
3880 pre_position (dtp);
3882 smarkeor (dtp->u.p.current_unit->s);
3886 /* Finalize the current data transfer. For a nonadvancing transfer,
3887 this means advancing to the next record. For internal units close the
3888 stream associated with the unit. */
3890 static void
3891 finalize_transfer (st_parameter_dt *dtp)
3893 GFC_INTEGER_4 cf = dtp->common.flags;
3895 if ((dtp->u.p.ionml != NULL)
3896 && (cf & IOPARM_DT_HAS_NAMELIST_NAME) != 0)
3898 dtp->u.p.namelist_mode = 1;
3899 if ((cf & IOPARM_DT_NAMELIST_READ_MODE) != 0)
3900 namelist_read (dtp);
3901 else
3902 namelist_write (dtp);
3905 if ((dtp->common.flags & IOPARM_DT_HAS_SIZE) != 0)
3906 *dtp->size = dtp->u.p.current_unit->size_used;
3908 if (dtp->u.p.eor_condition)
3910 generate_error (&dtp->common, LIBERROR_EOR, NULL);
3911 goto done;
3914 if (dtp->u.p.current_unit && (dtp->u.p.current_unit->child_dtio > 0))
3916 if (cf & IOPARM_DT_HAS_FORMAT)
3918 free (dtp->u.p.fmt);
3919 free (dtp->format);
3921 return;
3924 if ((dtp->common.flags & IOPARM_LIBRETURN_MASK) != IOPARM_LIBRETURN_OK)
3926 if (dtp->u.p.current_unit && current_mode (dtp) == UNFORMATTED_SEQUENTIAL)
3927 dtp->u.p.current_unit->current_record = 0;
3928 goto done;
3931 dtp->u.p.transfer = NULL;
3932 if (dtp->u.p.current_unit == NULL)
3933 goto done;
3935 if ((cf & IOPARM_DT_LIST_FORMAT) != 0 && dtp->u.p.mode == READING)
3937 finish_list_read (dtp);
3938 goto done;
3941 if (dtp->u.p.mode == WRITING)
3942 dtp->u.p.current_unit->previous_nonadvancing_write
3943 = dtp->u.p.advance_status == ADVANCE_NO;
3945 if (is_stream_io (dtp))
3947 if (dtp->u.p.current_unit->flags.form == FORM_FORMATTED
3948 && dtp->u.p.advance_status != ADVANCE_NO)
3949 next_record (dtp, 1);
3951 goto done;
3954 dtp->u.p.current_unit->current_record = 0;
3956 if (!is_internal_unit (dtp) && dtp->u.p.seen_dollar)
3958 fbuf_flush (dtp->u.p.current_unit, dtp->u.p.mode);
3959 dtp->u.p.seen_dollar = 0;
3960 goto done;
3963 /* For non-advancing I/O, save the current maximum position for use in the
3964 next I/O operation if needed. */
3965 if (dtp->u.p.advance_status == ADVANCE_NO)
3967 if (dtp->u.p.skips > 0)
3969 int tmp;
3970 write_x (dtp, dtp->u.p.skips, dtp->u.p.pending_spaces);
3971 tmp = (int)(dtp->u.p.current_unit->recl
3972 - dtp->u.p.current_unit->bytes_left);
3973 dtp->u.p.max_pos =
3974 dtp->u.p.max_pos > tmp ? dtp->u.p.max_pos : tmp;
3975 dtp->u.p.skips = 0;
3977 int bytes_written = (int) (dtp->u.p.current_unit->recl
3978 - dtp->u.p.current_unit->bytes_left);
3979 dtp->u.p.current_unit->saved_pos =
3980 dtp->u.p.max_pos > 0 ? dtp->u.p.max_pos - bytes_written : 0;
3981 fbuf_flush (dtp->u.p.current_unit, dtp->u.p.mode);
3982 goto done;
3984 else if (dtp->u.p.current_unit->flags.form == FORM_FORMATTED
3985 && dtp->u.p.mode == WRITING && !is_internal_unit (dtp))
3986 fbuf_seek (dtp->u.p.current_unit, 0, SEEK_END);
3988 dtp->u.p.current_unit->saved_pos = 0;
3989 dtp->u.p.current_unit->last_char = EOF - 1;
3990 next_record (dtp, 1);
3992 done:
3994 if (dtp->u.p.unit_is_internal)
3996 fbuf_destroy (dtp->u.p.current_unit);
3997 if (dtp->u.p.current_unit
3998 && (dtp->u.p.current_unit->child_dtio == 0)
3999 && dtp->u.p.current_unit->s)
4001 sclose (dtp->u.p.current_unit->s);
4002 dtp->u.p.current_unit->s = NULL;
4006 #ifdef HAVE_USELOCALE
4007 if (dtp->u.p.old_locale != (locale_t) 0)
4009 uselocale (dtp->u.p.old_locale);
4010 dtp->u.p.old_locale = (locale_t) 0;
4012 #else
4013 __gthread_mutex_lock (&old_locale_lock);
4014 if (!--old_locale_ctr)
4016 setlocale (LC_NUMERIC, old_locale);
4017 old_locale = NULL;
4019 __gthread_mutex_unlock (&old_locale_lock);
4020 #endif
4023 /* Transfer function for IOLENGTH. It doesn't actually do any
4024 data transfer, it just updates the length counter. */
4026 static void
4027 iolength_transfer (st_parameter_dt *dtp, bt type __attribute__((unused)),
4028 void *dest __attribute__ ((unused)),
4029 int kind __attribute__((unused)),
4030 size_t size, size_t nelems)
4032 if ((dtp->common.flags & IOPARM_DT_HAS_IOLENGTH) != 0)
4033 *dtp->iolength += (GFC_IO_INT) (size * nelems);
4037 /* Initialize the IOLENGTH data transfer. This function is in essence
4038 a very much simplified version of data_transfer_init(), because it
4039 doesn't have to deal with units at all. */
4041 static void
4042 iolength_transfer_init (st_parameter_dt *dtp)
4044 if ((dtp->common.flags & IOPARM_DT_HAS_IOLENGTH) != 0)
4045 *dtp->iolength = 0;
4047 memset (&dtp->u.p, 0, sizeof (dtp->u.p));
4049 /* Set up the subroutine that will handle the transfers. */
4051 dtp->u.p.transfer = iolength_transfer;
4055 /* Library entry point for the IOLENGTH form of the INQUIRE
4056 statement. The IOLENGTH form requires no I/O to be performed, but
4057 it must still be a runtime library call so that we can determine
4058 the iolength for dynamic arrays and such. */
4060 extern void st_iolength (st_parameter_dt *);
4061 export_proto(st_iolength);
4063 void
4064 st_iolength (st_parameter_dt *dtp)
4066 library_start (&dtp->common);
4067 iolength_transfer_init (dtp);
4070 extern void st_iolength_done (st_parameter_dt *);
4071 export_proto(st_iolength_done);
4073 void
4074 st_iolength_done (st_parameter_dt *dtp __attribute__((unused)))
4076 free_ionml (dtp);
4077 library_end ();
4081 /* The READ statement. */
4083 extern void st_read (st_parameter_dt *);
4084 export_proto(st_read);
4086 void
4087 st_read (st_parameter_dt *dtp)
4089 library_start (&dtp->common);
4091 data_transfer_init (dtp, 1);
4094 extern void st_read_done (st_parameter_dt *);
4095 export_proto(st_read_done);
4097 void
4098 st_read_done (st_parameter_dt *dtp)
4100 finalize_transfer (dtp);
4102 free_ionml (dtp);
4104 /* If this is a parent READ statement we do not need to retain the
4105 internal unit structure for child use. */
4106 if (dtp->u.p.current_unit != NULL
4107 && dtp->u.p.current_unit->child_dtio == 0)
4109 if (dtp->u.p.unit_is_internal)
4111 if ((dtp->common.flags & IOPARM_DT_HAS_UDTIO) == 0)
4113 free (dtp->u.p.current_unit->filename);
4114 dtp->u.p.current_unit->filename = NULL;
4115 if (dtp->u.p.current_unit->ls)
4116 free (dtp->u.p.current_unit->ls);
4117 dtp->u.p.current_unit->ls = NULL;
4119 newunit_free (dtp->common.unit);
4121 if (dtp->u.p.unit_is_internal || dtp->u.p.format_not_saved)
4123 free_format_data (dtp->u.p.fmt);
4124 free_format (dtp);
4126 unlock_unit (dtp->u.p.current_unit);
4129 library_end ();
4132 extern void st_write (st_parameter_dt *);
4133 export_proto(st_write);
4135 void
4136 st_write (st_parameter_dt *dtp)
4138 library_start (&dtp->common);
4139 data_transfer_init (dtp, 0);
4142 extern void st_write_done (st_parameter_dt *);
4143 export_proto(st_write_done);
4145 void
4146 st_write_done (st_parameter_dt *dtp)
4148 finalize_transfer (dtp);
4150 if (dtp->u.p.current_unit != NULL
4151 && dtp->u.p.current_unit->child_dtio == 0)
4153 /* Deal with endfile conditions associated with sequential files. */
4154 if (dtp->u.p.current_unit->flags.access == ACCESS_SEQUENTIAL)
4155 switch (dtp->u.p.current_unit->endfile)
4157 case AT_ENDFILE: /* Remain at the endfile record. */
4158 break;
4160 case AFTER_ENDFILE:
4161 dtp->u.p.current_unit->endfile = AT_ENDFILE; /* Just at it now. */
4162 break;
4164 case NO_ENDFILE:
4165 /* Get rid of whatever is after this record. */
4166 if (!is_internal_unit (dtp))
4167 unit_truncate (dtp->u.p.current_unit,
4168 stell (dtp->u.p.current_unit->s),
4169 &dtp->common);
4170 dtp->u.p.current_unit->endfile = AT_ENDFILE;
4171 break;
4174 free_ionml (dtp);
4176 /* If this is a parent WRITE statement we do not need to retain the
4177 internal unit structure for child use. */
4178 if (dtp->u.p.unit_is_internal)
4180 if ((dtp->common.flags & IOPARM_DT_HAS_UDTIO) == 0)
4182 free (dtp->u.p.current_unit->filename);
4183 dtp->u.p.current_unit->filename = NULL;
4184 if (dtp->u.p.current_unit->ls)
4185 free (dtp->u.p.current_unit->ls);
4186 dtp->u.p.current_unit->ls = NULL;
4188 newunit_free (dtp->common.unit);
4190 if (dtp->u.p.unit_is_internal || dtp->u.p.format_not_saved)
4192 free_format_data (dtp->u.p.fmt);
4193 free_format (dtp);
4195 unlock_unit (dtp->u.p.current_unit);
4197 library_end ();
4201 /* F2003: This is a stub for the runtime portion of the WAIT statement. */
4202 void
4203 st_wait (st_parameter_wait *wtp __attribute__((unused)))
4208 /* Receives the scalar information for namelist objects and stores it
4209 in a linked list of namelist_info types. */
4211 static void
4212 set_nml_var (st_parameter_dt *dtp, void *var_addr, char *var_name,
4213 GFC_INTEGER_4 len, gfc_charlen_type string_length,
4214 dtype_type dtype, void *dtio_sub, void *vtable)
4216 namelist_info *t1 = NULL;
4217 namelist_info *nml;
4218 size_t var_name_len = strlen (var_name);
4220 nml = (namelist_info*) xmalloc (sizeof (namelist_info));
4222 nml->mem_pos = var_addr;
4223 nml->dtio_sub = dtio_sub;
4224 nml->vtable = vtable;
4226 nml->var_name = (char*) xmalloc (var_name_len + 1);
4227 memcpy (nml->var_name, var_name, var_name_len);
4228 nml->var_name[var_name_len] = '\0';
4230 nml->len = (int) len;
4231 nml->string_length = (index_type) string_length;
4233 nml->var_rank = (int) (dtype.rank);
4234 nml->size = (index_type) (dtype.elem_len);
4235 nml->type = (bt) (dtype.type);
4237 if (nml->var_rank > 0)
4239 nml->dim = (descriptor_dimension*)
4240 xmallocarray (nml->var_rank, sizeof (descriptor_dimension));
4241 nml->ls = (array_loop_spec*)
4242 xmallocarray (nml->var_rank, sizeof (array_loop_spec));
4244 else
4246 nml->dim = NULL;
4247 nml->ls = NULL;
4250 nml->next = NULL;
4252 if ((dtp->common.flags & IOPARM_DT_IONML_SET) == 0)
4254 dtp->common.flags |= IOPARM_DT_IONML_SET;
4255 dtp->u.p.ionml = nml;
4257 else
4259 for (t1 = dtp->u.p.ionml; t1->next; t1 = t1->next);
4260 t1->next = nml;
4264 extern void st_set_nml_var (st_parameter_dt *dtp, void *, char *,
4265 GFC_INTEGER_4, gfc_charlen_type, dtype_type);
4266 export_proto(st_set_nml_var);
4268 void
4269 st_set_nml_var (st_parameter_dt *dtp, void *var_addr, char *var_name,
4270 GFC_INTEGER_4 len, gfc_charlen_type string_length,
4271 dtype_type dtype)
4273 set_nml_var (dtp, var_addr, var_name, len, string_length,
4274 dtype, NULL, NULL);
4278 /* Essentially the same as previous but carrying the dtio procedure
4279 and the vtable as additional arguments. */
4280 extern void st_set_nml_dtio_var (st_parameter_dt *dtp, void *, char *,
4281 GFC_INTEGER_4, gfc_charlen_type, dtype_type,
4282 void *, void *);
4283 export_proto(st_set_nml_dtio_var);
4286 void
4287 st_set_nml_dtio_var (st_parameter_dt *dtp, void *var_addr, char *var_name,
4288 GFC_INTEGER_4 len, gfc_charlen_type string_length,
4289 dtype_type dtype, void *dtio_sub, void *vtable)
4291 set_nml_var (dtp, var_addr, var_name, len, string_length,
4292 dtype, dtio_sub, vtable);
4295 /* Store the dimensional information for the namelist object. */
4296 extern void st_set_nml_var_dim (st_parameter_dt *, GFC_INTEGER_4,
4297 index_type, index_type,
4298 index_type);
4299 export_proto(st_set_nml_var_dim);
4301 void
4302 st_set_nml_var_dim (st_parameter_dt *dtp, GFC_INTEGER_4 n_dim,
4303 index_type stride, index_type lbound,
4304 index_type ubound)
4306 namelist_info *nml;
4307 int n;
4309 n = (int)n_dim;
4311 for (nml = dtp->u.p.ionml; nml->next; nml = nml->next);
4313 GFC_DIMENSION_SET(nml->dim[n],lbound,ubound,stride);
4317 /* Once upon a time, a poor innocent Fortran program was reading a
4318 file, when suddenly it hit the end-of-file (EOF). Unfortunately
4319 the OS doesn't tell whether we're at the EOF or whether we already
4320 went past it. Luckily our hero, libgfortran, keeps track of this.
4321 Call this function when you detect an EOF condition. See Section
4322 9.10.2 in F2003. */
4324 void
4325 hit_eof (st_parameter_dt *dtp)
4327 dtp->u.p.current_unit->flags.position = POSITION_APPEND;
4329 if (dtp->u.p.current_unit->flags.access == ACCESS_SEQUENTIAL)
4330 switch (dtp->u.p.current_unit->endfile)
4332 case NO_ENDFILE:
4333 case AT_ENDFILE:
4334 generate_error (&dtp->common, LIBERROR_END, NULL);
4335 if (!is_internal_unit (dtp) && !dtp->u.p.namelist_mode)
4337 dtp->u.p.current_unit->endfile = AFTER_ENDFILE;
4338 dtp->u.p.current_unit->current_record = 0;
4340 else
4341 dtp->u.p.current_unit->endfile = AT_ENDFILE;
4342 break;
4344 case AFTER_ENDFILE:
4345 generate_error (&dtp->common, LIBERROR_ENDFILE, NULL);
4346 dtp->u.p.current_unit->current_record = 0;
4347 break;
4349 else
4351 /* Non-sequential files don't have an ENDFILE record, so we
4352 can't be at AFTER_ENDFILE. */
4353 dtp->u.p.current_unit->endfile = AT_ENDFILE;
4354 generate_error (&dtp->common, LIBERROR_END, NULL);
4355 dtp->u.p.current_unit->current_record = 0;