Run the GC and retry `open-file' when getting `EMFILE'.
[bpt/guile.git] / libguile / fports.c
1 /* Copyright (C) 1995,1996,1997,1998,1999,2000,2001, 2002, 2003, 2004, 2006, 2007, 2008, 2009 Free Software Foundation, Inc.
2 *
3 * This library is free software; you can redistribute it and/or
4 * modify it under the terms of the GNU Lesser General Public License
5 * as published by the Free Software Foundation; either version 3 of
6 * the License, or (at your option) any later version.
7 *
8 * This library is distributed in the hope that it will be useful, but
9 * WITHOUT ANY WARRANTY; without even the implied warranty of
10 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
11 * Lesser General Public License for more details.
12 *
13 * You should have received a copy of the GNU Lesser General Public
14 * License along with this library; if not, write to the Free Software
15 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
16 * 02110-1301 USA
17 */
18
19
20 \f
21 #define _LARGEFILE64_SOURCE /* ask for stat64 etc */
22
23 #ifdef HAVE_CONFIG_H
24 # include <config.h>
25 #endif
26
27 #include <stdio.h>
28 #include <fcntl.h>
29 #include "libguile/_scm.h"
30 #include "libguile/strings.h"
31 #include "libguile/validate.h"
32 #include "libguile/gc.h"
33 #include "libguile/posix.h"
34 #include "libguile/dynwind.h"
35 #include "libguile/hashtab.h"
36
37 #include "libguile/fports.h"
38
39 #ifdef HAVE_STRING_H
40 #include <string.h>
41 #endif
42 #ifdef HAVE_UNISTD_H
43 #include <unistd.h>
44 #endif
45 #ifdef HAVE_IO_H
46 #include <io.h>
47 #endif
48 #ifdef HAVE_STRUCT_STAT_ST_BLKSIZE
49 #include <sys/stat.h>
50 #endif
51
52 #include <errno.h>
53 #include <sys/types.h>
54
55 #include "libguile/iselect.h"
56
57 /* Some defines for Windows (native port, not Cygwin). */
58 #ifdef __MINGW32__
59 # include <sys/stat.h>
60 # include <winsock2.h>
61 #endif /* __MINGW32__ */
62
63 #include <full-write.h>
64
65 /* Mingw (version 3.4.5, circa 2006) has ftruncate as an alias for chsize
66 already, but have this code here in case that wasn't so in past versions,
67 or perhaps to help other minimal DOS environments.
68
69 gnulib ftruncate.c has code using fcntl F_CHSIZE and F_FREESP, which
70 might be possibilities if we've got other systems without ftruncate. */
71
72 #if HAVE_CHSIZE && ! HAVE_FTRUNCATE
73 # define ftruncate(fd, size) chsize (fd, size)
74 #undef HAVE_FTRUNCATE
75 #define HAVE_FTRUNCATE 1
76 #endif
77
78 #if SIZEOF_OFF_T == SIZEOF_INT
79 #define OFF_T_MAX INT_MAX
80 #define OFF_T_MIN INT_MIN
81 #elif SIZEOF_OFF_T == SIZEOF_LONG
82 #define OFF_T_MAX LONG_MAX
83 #define OFF_T_MIN LONG_MIN
84 #elif SIZEOF_OFF_T == SIZEOF_LONG_LONG
85 #define OFF_T_MAX LONG_LONG_MAX
86 #define OFF_T_MIN LONG_LONG_MIN
87 #else
88 #error Oops, unknown OFF_T size
89 #endif
90
91 scm_t_bits scm_tc16_fport;
92
93
94 /* default buffer size, used if the O/S won't supply a value. */
95 static const size_t default_buffer_size = 1024;
96
97 /* create FPORT buffer with specified sizes (or -1 to use default size or
98 0 for no buffer. */
99 static void
100 scm_fport_buffer_add (SCM port, long read_size, int write_size)
101 #define FUNC_NAME "scm_fport_buffer_add"
102 {
103 scm_t_port *pt = SCM_PTAB_ENTRY (port);
104
105 if (read_size == -1 || write_size == -1)
106 {
107 size_t default_size;
108 #ifdef HAVE_STRUCT_STAT_ST_BLKSIZE
109 struct stat st;
110 scm_t_fport *fp = SCM_FSTREAM (port);
111
112 default_size = (fstat (fp->fdes, &st) == -1) ? default_buffer_size
113 : st.st_blksize;
114 #else
115 default_size = default_buffer_size;
116 #endif
117 if (read_size == -1)
118 read_size = default_size;
119 if (write_size == -1)
120 write_size = default_size;
121 }
122
123 if (SCM_INPUT_PORT_P (port) && read_size > 0)
124 {
125 pt->read_buf = scm_gc_malloc_pointerless (read_size, "port buffer");
126 pt->read_pos = pt->read_end = pt->read_buf;
127 pt->read_buf_size = read_size;
128 }
129 else
130 {
131 pt->read_pos = pt->read_buf = pt->read_end = &pt->shortbuf;
132 pt->read_buf_size = 1;
133 }
134
135 if (SCM_OUTPUT_PORT_P (port) && write_size > 0)
136 {
137 pt->write_buf = scm_gc_malloc_pointerless (write_size, "port buffer");
138 pt->write_pos = pt->write_buf;
139 pt->write_buf_size = write_size;
140 }
141 else
142 {
143 pt->write_buf = pt->write_pos = &pt->shortbuf;
144 pt->write_buf_size = 1;
145 }
146
147 pt->write_end = pt->write_buf + pt->write_buf_size;
148 if (read_size > 0 || write_size > 0)
149 SCM_SET_CELL_WORD_0 (port, SCM_CELL_WORD_0 (port) & ~SCM_BUF0);
150 else
151 SCM_SET_CELL_WORD_0 (port, SCM_CELL_WORD_0 (port) | SCM_BUF0);
152 }
153 #undef FUNC_NAME
154
155 SCM_DEFINE (scm_setvbuf, "setvbuf", 2, 1, 0,
156 (SCM port, SCM mode, SCM size),
157 "Set the buffering mode for @var{port}. @var{mode} can be:\n"
158 "@table @code\n"
159 "@item _IONBF\n"
160 "non-buffered\n"
161 "@item _IOLBF\n"
162 "line buffered\n"
163 "@item _IOFBF\n"
164 "block buffered, using a newly allocated buffer of @var{size} bytes.\n"
165 "If @var{size} is omitted, a default size will be used.\n"
166 "@end table")
167 #define FUNC_NAME s_scm_setvbuf
168 {
169 int cmode;
170 long csize;
171 scm_t_port *pt;
172
173 port = SCM_COERCE_OUTPORT (port);
174
175 SCM_VALIDATE_OPFPORT (1,port);
176 cmode = scm_to_int (mode);
177 if (cmode != _IONBF && cmode != _IOFBF && cmode != _IOLBF)
178 scm_out_of_range (FUNC_NAME, mode);
179
180 if (cmode == _IOLBF)
181 {
182 SCM_SET_CELL_WORD_0 (port, SCM_CELL_WORD_0 (port) | SCM_BUFLINE);
183 cmode = _IOFBF;
184 }
185 else
186 {
187 SCM_SET_CELL_WORD_0 (port, SCM_CELL_WORD_0 (port) & ~(scm_t_bits)SCM_BUFLINE);
188 }
189
190 if (SCM_UNBNDP (size))
191 {
192 if (cmode == _IOFBF)
193 csize = -1;
194 else
195 csize = 0;
196 }
197 else
198 {
199 csize = scm_to_int (size);
200 if (csize < 0 || (cmode == _IONBF && csize > 0))
201 scm_out_of_range (FUNC_NAME, size);
202 }
203
204 pt = SCM_PTAB_ENTRY (port);
205
206 /* silently discards buffered and put-back chars. */
207 if (pt->read_buf == pt->putback_buf)
208 {
209 pt->read_buf = pt->saved_read_buf;
210 pt->read_pos = pt->saved_read_pos;
211 pt->read_end = pt->saved_read_end;
212 pt->read_buf_size = pt->saved_read_buf_size;
213 }
214 if (pt->read_buf != &pt->shortbuf)
215 scm_gc_free (pt->read_buf, pt->read_buf_size, "port buffer");
216 if (pt->write_buf != &pt->shortbuf)
217 scm_gc_free (pt->write_buf, pt->write_buf_size, "port buffer");
218
219 scm_fport_buffer_add (port, csize, csize);
220 return SCM_UNSPECIFIED;
221 }
222 #undef FUNC_NAME
223
224 /* Move ports with the specified file descriptor to new descriptors,
225 * resetting the revealed count to 0.
226 */
227 static void
228 scm_i_evict_port (void *closure, SCM port)
229 {
230 int fd = * (int*) closure;
231
232 if (SCM_FPORTP (port))
233 {
234 scm_t_port *p;
235 scm_t_fport *fp;
236
237 /* XXX: In some cases, we can encounter a port with no associated ptab
238 entry. */
239 p = SCM_PTAB_ENTRY (port);
240 fp = (p != NULL) ? (scm_t_fport *) p->stream : NULL;
241
242 if ((fp != NULL) && (fp->fdes == fd))
243 {
244 fp->fdes = dup (fd);
245 if (fp->fdes == -1)
246 scm_syserror ("scm_evict_ports");
247 scm_set_port_revealed_x (port, scm_from_int (0));
248 }
249 }
250 }
251
252 void
253 scm_evict_ports (int fd)
254 {
255 scm_c_port_for_each (scm_i_evict_port, (void *) &fd);
256 }
257
258
259 SCM_DEFINE (scm_file_port_p, "file-port?", 1, 0, 0,
260 (SCM obj),
261 "Determine whether @var{obj} is a port that is related to a file.")
262 #define FUNC_NAME s_scm_file_port_p
263 {
264 return scm_from_bool (SCM_FPORTP (obj));
265 }
266 #undef FUNC_NAME
267
268
269 /* scm_open_file
270 * Return a new port open on a given file.
271 *
272 * The mode string must match the pattern: [rwa+]** which
273 * is interpreted in the usual unix way.
274 *
275 * Return the new port.
276 */
277 SCM_DEFINE (scm_open_file, "open-file", 2, 0, 0,
278 (SCM filename, SCM mode),
279 "Open the file whose name is @var{filename}, and return a port\n"
280 "representing that file. The attributes of the port are\n"
281 "determined by the @var{mode} string. The way in which this is\n"
282 "interpreted is similar to C stdio. The first character must be\n"
283 "one of the following:\n"
284 "@table @samp\n"
285 "@item r\n"
286 "Open an existing file for input.\n"
287 "@item w\n"
288 "Open a file for output, creating it if it doesn't already exist\n"
289 "or removing its contents if it does.\n"
290 "@item a\n"
291 "Open a file for output, creating it if it doesn't already\n"
292 "exist. All writes to the port will go to the end of the file.\n"
293 "The \"append mode\" can be turned off while the port is in use\n"
294 "@pxref{Ports and File Descriptors, fcntl}\n"
295 "@end table\n"
296 "The following additional characters can be appended:\n"
297 "@table @samp\n"
298 "@item b\n"
299 "Open the underlying file in binary mode, if supported by the operating system. "
300 "@item +\n"
301 "Open the port for both input and output. E.g., @code{r+}: open\n"
302 "an existing file for both input and output.\n"
303 "@item 0\n"
304 "Create an \"unbuffered\" port. In this case input and output\n"
305 "operations are passed directly to the underlying port\n"
306 "implementation without additional buffering. This is likely to\n"
307 "slow down I/O operations. The buffering mode can be changed\n"
308 "while a port is in use @pxref{Ports and File Descriptors,\n"
309 "setvbuf}\n"
310 "@item l\n"
311 "Add line-buffering to the port. The port output buffer will be\n"
312 "automatically flushed whenever a newline character is written.\n"
313 "@end table\n"
314 "In theory we could create read/write ports which were buffered\n"
315 "in one direction only. However this isn't included in the\n"
316 "current interfaces. If a file cannot be opened with the access\n"
317 "requested, @code{open-file} throws an exception.")
318 #define FUNC_NAME s_scm_open_file
319 {
320 SCM port;
321 int fdes, flags = 0;
322 unsigned int retries;
323 char *file, *md, *ptr;
324
325 scm_dynwind_begin (0);
326
327 file = scm_to_locale_string (filename);
328 scm_dynwind_free (file);
329
330 md = scm_to_locale_string (mode);
331 scm_dynwind_free (md);
332
333 switch (*md)
334 {
335 case 'r':
336 flags |= O_RDONLY;
337 break;
338 case 'w':
339 flags |= O_WRONLY | O_CREAT | O_TRUNC;
340 break;
341 case 'a':
342 flags |= O_WRONLY | O_CREAT | O_APPEND;
343 break;
344 default:
345 scm_out_of_range (FUNC_NAME, mode);
346 }
347 ptr = md + 1;
348 while (*ptr != '\0')
349 {
350 switch (*ptr)
351 {
352 case '+':
353 flags = (flags & ~(O_RDONLY | O_WRONLY)) | O_RDWR;
354 break;
355 case 'b':
356 #if defined (O_BINARY)
357 flags |= O_BINARY;
358 #endif
359 break;
360 case '0': /* unbuffered: handled later. */
361 case 'l': /* line buffered: handled during output. */
362 break;
363 default:
364 scm_out_of_range (FUNC_NAME, mode);
365 }
366 ptr++;
367 }
368
369 for (retries = 0, fdes = -1;
370 fdes < 0 && retries < 2;
371 retries++)
372 {
373 SCM_SYSCALL (fdes = open_or_open64 (file, flags, 0666));
374 if (fdes == -1)
375 {
376 int en = errno;
377
378 if (en == EMFILE && retries == 0)
379 /* Run the GC in case it collects open file ports that are no
380 longer referenced. */
381 scm_i_gc (FUNC_NAME);
382 else
383 SCM_SYSERROR_MSG ("~A: ~S",
384 scm_cons (scm_strerror (scm_from_int (en)),
385 scm_cons (filename, SCM_EOL)), en);
386 }
387 }
388
389 port = scm_i_fdes_to_port (fdes, scm_i_mode_bits (mode), filename);
390
391 scm_dynwind_end ();
392
393 return port;
394 }
395 #undef FUNC_NAME
396
397 \f
398 #ifdef __MINGW32__
399 /*
400 * Try getting the appropiate file flags for a given file descriptor
401 * under Windows. This incorporates some fancy operations because Windows
402 * differentiates between file, pipe and socket descriptors.
403 */
404 #ifndef O_ACCMODE
405 # define O_ACCMODE 0x0003
406 #endif
407
408 static int getflags (int fdes)
409 {
410 int flags = 0;
411 struct stat buf;
412 int error, optlen = sizeof (int);
413
414 /* Is this a socket ? */
415 if (getsockopt (fdes, SOL_SOCKET, SO_ERROR, (void *) &error, &optlen) >= 0)
416 flags = O_RDWR;
417 /* Maybe a regular file ? */
418 else if (fstat (fdes, &buf) < 0)
419 flags = -1;
420 else
421 {
422 /* Or an anonymous pipe handle ? */
423 if (buf.st_mode & _S_IFIFO)
424 flags = PeekNamedPipe ((HANDLE) _get_osfhandle (fdes), NULL, 0,
425 NULL, NULL, NULL) ? O_RDONLY : O_WRONLY;
426 /* stdin ? */
427 else if (fdes == fileno (stdin) && isatty (fdes))
428 flags = O_RDONLY;
429 /* stdout / stderr ? */
430 else if ((fdes == fileno (stdout) || fdes == fileno (stderr)) &&
431 isatty (fdes))
432 flags = O_WRONLY;
433 else
434 flags = buf.st_mode;
435 }
436 return flags;
437 }
438 #endif /* __MINGW32__ */
439
440 /* Building Guile ports from a file descriptor. */
441
442 /* Build a Scheme port from an open file descriptor `fdes'.
443 MODE indicates whether FILE is open for reading or writing; it uses
444 the same notation as open-file's second argument.
445 NAME is a string to be used as the port's filename.
446 */
447 SCM
448 scm_i_fdes_to_port (int fdes, long mode_bits, SCM name)
449 #define FUNC_NAME "scm_fdes_to_port"
450 {
451 SCM port;
452 scm_t_port *pt;
453 int flags;
454
455 /* test that fdes is valid. */
456 #ifdef __MINGW32__
457 flags = getflags (fdes);
458 #else
459 flags = fcntl (fdes, F_GETFL, 0);
460 #endif
461 if (flags == -1)
462 SCM_SYSERROR;
463 flags &= O_ACCMODE;
464 if (flags != O_RDWR
465 && ((flags != O_WRONLY && (mode_bits & SCM_WRTNG))
466 || (flags != O_RDONLY && (mode_bits & SCM_RDNG))))
467 {
468 SCM_MISC_ERROR ("requested file mode not available on fdes", SCM_EOL);
469 }
470
471 scm_i_scm_pthread_mutex_lock (&scm_i_port_table_mutex);
472
473 port = scm_new_port_table_entry (scm_tc16_fport);
474 SCM_SET_CELL_TYPE(port, scm_tc16_fport | mode_bits);
475 pt = SCM_PTAB_ENTRY(port);
476 {
477 scm_t_fport *fp
478 = (scm_t_fport *) scm_gc_malloc_pointerless (sizeof (scm_t_fport),
479 "file port");
480
481 fp->fdes = fdes;
482 pt->rw_random = SCM_FDES_RANDOM_P (fdes);
483 SCM_SETSTREAM (port, fp);
484 if (mode_bits & SCM_BUF0)
485 scm_fport_buffer_add (port, 0, 0);
486 else
487 scm_fport_buffer_add (port, -1, -1);
488 }
489 SCM_SET_FILENAME (port, name);
490 scm_i_pthread_mutex_unlock (&scm_i_port_table_mutex);
491 return port;
492 }
493 #undef FUNC_NAME
494
495 SCM
496 scm_fdes_to_port (int fdes, char *mode, SCM name)
497 {
498 return scm_i_fdes_to_port (fdes, scm_mode_bits (mode), name);
499 }
500
501 /* Return a lower bound on the number of bytes available for input. */
502 static int
503 fport_input_waiting (SCM port)
504 {
505 #ifdef HAVE_SELECT
506 int fdes = SCM_FSTREAM (port)->fdes;
507 struct timeval timeout;
508 SELECT_TYPE read_set;
509 SELECT_TYPE write_set;
510 SELECT_TYPE except_set;
511
512 FD_ZERO (&read_set);
513 FD_ZERO (&write_set);
514 FD_ZERO (&except_set);
515
516 FD_SET (fdes, &read_set);
517
518 timeout.tv_sec = 0;
519 timeout.tv_usec = 0;
520
521 if (select (SELECT_SET_SIZE,
522 &read_set, &write_set, &except_set, &timeout)
523 < 0)
524 scm_syserror ("fport_input_waiting");
525 return FD_ISSET (fdes, &read_set) ? 1 : 0;
526
527 #elif HAVE_IOCTL && defined (FIONREAD)
528 /* Note: cannot test just defined(FIONREAD) here, since mingw has FIONREAD
529 (for use with winsock ioctlsocket()) but not ioctl(). */
530 int fdes = SCM_FSTREAM (port)->fdes;
531 int remir;
532 ioctl(fdes, FIONREAD, &remir);
533 return remir;
534
535 #else
536 scm_misc_error ("fport_input_waiting",
537 "Not fully implemented on this platform",
538 SCM_EOL);
539 #endif
540 }
541
542 \f
543 static int
544 fport_print (SCM exp, SCM port, scm_print_state *pstate SCM_UNUSED)
545 {
546 scm_puts ("#<", port);
547 scm_print_port_mode (exp, port);
548 if (SCM_OPFPORTP (exp))
549 {
550 int fdes;
551 SCM name = SCM_FILENAME (exp);
552 if (scm_is_string (name) || scm_is_symbol (name))
553 scm_display (name, port);
554 else
555 scm_puts (SCM_PTOBNAME (SCM_PTOBNUM (exp)), port);
556 scm_putc (' ', port);
557 fdes = (SCM_FSTREAM (exp))->fdes;
558
559 #ifdef HAVE_TTYNAME
560 if (isatty (fdes))
561 scm_display (scm_ttyname (exp), port);
562 else
563 #endif /* HAVE_TTYNAME */
564 scm_intprint (fdes, 10, port);
565 }
566 else
567 {
568 scm_puts (SCM_PTOBNAME (SCM_PTOBNUM (exp)), port);
569 scm_putc (' ', port);
570 scm_uintprint ((scm_t_bits) SCM_PTAB_ENTRY (exp), 16, port);
571 }
572 scm_putc ('>', port);
573 return 1;
574 }
575
576 #ifndef __MINGW32__
577 /* thread-local block for input on fport's fdes. */
578 static void
579 fport_wait_for_input (SCM port)
580 {
581 int fdes = SCM_FSTREAM (port)->fdes;
582
583 if (!fport_input_waiting (port))
584 {
585 int n;
586 SELECT_TYPE readfds;
587 int flags = fcntl (fdes, F_GETFL);
588
589 if (flags == -1)
590 scm_syserror ("scm_fdes_wait_for_input");
591 if (!(flags & O_NONBLOCK))
592 do
593 {
594 FD_ZERO (&readfds);
595 FD_SET (fdes, &readfds);
596 n = scm_std_select (fdes + 1, &readfds, NULL, NULL, NULL);
597 }
598 while (n == -1 && errno == EINTR);
599 }
600 }
601 #endif /* !__MINGW32__ */
602
603 static void fport_flush (SCM port);
604
605 /* fill a port's read-buffer with a single read. returns the first
606 char or EOF if end of file. */
607 static scm_t_wchar
608 fport_fill_input (SCM port)
609 {
610 long count;
611 scm_t_port *pt = SCM_PTAB_ENTRY (port);
612 scm_t_fport *fp = SCM_FSTREAM (port);
613
614 #ifndef __MINGW32__
615 fport_wait_for_input (port);
616 #endif /* !__MINGW32__ */
617 SCM_SYSCALL (count = read (fp->fdes, pt->read_buf, pt->read_buf_size));
618 if (count == -1)
619 scm_syserror ("fport_fill_input");
620 if (count == 0)
621 return (scm_t_wchar) EOF;
622 else
623 {
624 pt->read_pos = pt->read_buf;
625 pt->read_end = pt->read_buf + count;
626 return *pt->read_buf;
627 }
628 }
629
630 static scm_t_off
631 fport_seek (SCM port, scm_t_off offset, int whence)
632 {
633 scm_t_port *pt = SCM_PTAB_ENTRY (port);
634 scm_t_fport *fp = SCM_FSTREAM (port);
635 off_t_or_off64_t rv;
636 off_t_or_off64_t result;
637
638 if (pt->rw_active == SCM_PORT_WRITE)
639 {
640 if (offset != 0 || whence != SEEK_CUR)
641 {
642 fport_flush (port);
643 result = rv = lseek_or_lseek64 (fp->fdes, offset, whence);
644 }
645 else
646 {
647 /* read current position without disturbing the buffer. */
648 rv = lseek_or_lseek64 (fp->fdes, offset, whence);
649 result = rv + (pt->write_pos - pt->write_buf);
650 }
651 }
652 else if (pt->rw_active == SCM_PORT_READ)
653 {
654 if (offset != 0 || whence != SEEK_CUR)
655 {
656 /* could expand to avoid a second seek. */
657 scm_end_input (port);
658 result = rv = lseek_or_lseek64 (fp->fdes, offset, whence);
659 }
660 else
661 {
662 /* read current position without disturbing the buffer
663 (particularly the unread-char buffer). */
664 rv = lseek_or_lseek64 (fp->fdes, offset, whence);
665 result = rv - (pt->read_end - pt->read_pos);
666
667 if (pt->read_buf == pt->putback_buf)
668 result -= pt->saved_read_end - pt->saved_read_pos;
669 }
670 }
671 else /* SCM_PORT_NEITHER */
672 {
673 result = rv = lseek_or_lseek64 (fp->fdes, offset, whence);
674 }
675
676 if (rv == -1)
677 scm_syserror ("fport_seek");
678
679 return result;
680 }
681
682 static void
683 fport_truncate (SCM port, scm_t_off length)
684 {
685 scm_t_fport *fp = SCM_FSTREAM (port);
686
687 if (ftruncate (fp->fdes, length) == -1)
688 scm_syserror ("ftruncate");
689 }
690
691 /* helper for fport_write: try to write data, using multiple system
692 calls if required. */
693 #define FUNC_NAME "write_all"
694 static void write_all (SCM port, const void *data, size_t remaining)
695 {
696 int fdes = SCM_FSTREAM (port)->fdes;
697
698 while (remaining > 0)
699 {
700 size_t done;
701
702 SCM_SYSCALL (done = write (fdes, data, remaining));
703
704 if (done == -1)
705 SCM_SYSERROR;
706 remaining -= done;
707 data = ((const char *) data) + done;
708 }
709 }
710 #undef FUNC_NAME
711
712 static void
713 fport_write (SCM port, const void *data, size_t size)
714 {
715 /* this procedure tries to minimize the number of writes/flushes. */
716 scm_t_port *pt = SCM_PTAB_ENTRY (port);
717
718 if (pt->write_buf == &pt->shortbuf
719 || (pt->write_pos == pt->write_buf && size >= pt->write_buf_size))
720 {
721 /* "unbuffered" port, or
722 port with empty buffer and data won't fit in buffer. */
723 write_all (port, data, size);
724 return;
725 }
726
727 {
728 scm_t_off space = pt->write_end - pt->write_pos;
729
730 if (size <= space)
731 {
732 /* data fits in buffer. */
733 memcpy (pt->write_pos, data, size);
734 pt->write_pos += size;
735 if (pt->write_pos == pt->write_end)
736 {
737 fport_flush (port);
738 /* we can skip the line-buffering check if nothing's buffered. */
739 return;
740 }
741 }
742 else
743 {
744 memcpy (pt->write_pos, data, space);
745 pt->write_pos = pt->write_end;
746 fport_flush (port);
747 {
748 const void *ptr = ((const char *) data) + space;
749 size_t remaining = size - space;
750
751 if (size >= pt->write_buf_size)
752 {
753 write_all (port, ptr, remaining);
754 return;
755 }
756 else
757 {
758 memcpy (pt->write_pos, ptr, remaining);
759 pt->write_pos += remaining;
760 }
761 }
762 }
763
764 /* handle line buffering. */
765 if ((SCM_CELL_WORD_0 (port) & SCM_BUFLINE) && memchr (data, '\n', size))
766 fport_flush (port);
767 }
768 }
769
770 /* becomes 1 when process is exiting: normal exception handling won't
771 work by this time. */
772 extern int scm_i_terminating;
773
774 static void
775 fport_flush (SCM port)
776 {
777 scm_t_port *pt = SCM_PTAB_ENTRY (port);
778 scm_t_fport *fp = SCM_FSTREAM (port);
779 unsigned char *ptr = pt->write_buf;
780 long init_size = pt->write_pos - pt->write_buf;
781 long remaining = init_size;
782
783 while (remaining > 0)
784 {
785 long count;
786
787 SCM_SYSCALL (count = write (fp->fdes, ptr, remaining));
788 if (count < 0)
789 {
790 /* error. assume nothing was written this call, but
791 fix up the buffer for any previous successful writes. */
792 long done = init_size - remaining;
793
794 if (done > 0)
795 {
796 int i;
797
798 for (i = 0; i < remaining; i++)
799 {
800 *(pt->write_buf + i) = *(pt->write_buf + done + i);
801 }
802 pt->write_pos = pt->write_buf + remaining;
803 }
804 if (scm_i_terminating)
805 {
806 const char *msg = "Error: could not flush file-descriptor ";
807 char buf[11];
808
809 full_write (2, msg, strlen (msg));
810 sprintf (buf, "%d\n", fp->fdes);
811 full_write (2, buf, strlen (buf));
812
813 count = remaining;
814 }
815 else if (scm_gc_running_p)
816 {
817 /* silently ignore the error. scm_error would abort if we
818 called it now. */
819 count = remaining;
820 }
821 else
822 scm_syserror ("fport_flush");
823 }
824 ptr += count;
825 remaining -= count;
826 }
827 pt->write_pos = pt->write_buf;
828 pt->rw_active = SCM_PORT_NEITHER;
829 }
830
831 /* clear the read buffer and adjust the file position for unread bytes. */
832 static void
833 fport_end_input (SCM port, int offset)
834 {
835 scm_t_fport *fp = SCM_FSTREAM (port);
836 scm_t_port *pt = SCM_PTAB_ENTRY (port);
837
838 offset += pt->read_end - pt->read_pos;
839
840 if (offset > 0)
841 {
842 pt->read_pos = pt->read_end;
843 /* will throw error if unread-char used at beginning of file
844 then attempting to write. seems correct. */
845 if (lseek (fp->fdes, -offset, SEEK_CUR) == -1)
846 scm_syserror ("fport_end_input");
847 }
848 pt->rw_active = SCM_PORT_NEITHER;
849 }
850
851 static int
852 fport_close (SCM port)
853 {
854 scm_t_fport *fp = SCM_FSTREAM (port);
855 scm_t_port *pt = SCM_PTAB_ENTRY (port);
856 int rv;
857
858 fport_flush (port);
859 SCM_SYSCALL (rv = close (fp->fdes));
860 if (rv == -1 && errno != EBADF)
861 {
862 if (scm_gc_running_p)
863 /* silently ignore the error. scm_error would abort if we
864 called it now. */
865 ;
866 else
867 scm_syserror ("fport_close");
868 }
869 if (pt->read_buf == pt->putback_buf)
870 pt->read_buf = pt->saved_read_buf;
871 if (pt->read_buf != &pt->shortbuf)
872 scm_gc_free (pt->read_buf, pt->read_buf_size, "port buffer");
873 if (pt->write_buf != &pt->shortbuf)
874 scm_gc_free (pt->write_buf, pt->write_buf_size, "port buffer");
875 scm_gc_free (fp, sizeof (*fp), "file port");
876 return rv;
877 }
878
879 static size_t
880 fport_free (SCM port)
881 {
882 fport_close (port);
883 return 0;
884 }
885
886 static scm_t_bits
887 scm_make_fptob ()
888 {
889 scm_t_bits tc = scm_make_port_type ("file", fport_fill_input, fport_write);
890
891 scm_set_port_free (tc, fport_free);
892 scm_set_port_print (tc, fport_print);
893 scm_set_port_flush (tc, fport_flush);
894 scm_set_port_end_input (tc, fport_end_input);
895 scm_set_port_close (tc, fport_close);
896 scm_set_port_seek (tc, fport_seek);
897 scm_set_port_truncate (tc, fport_truncate);
898 scm_set_port_input_waiting (tc, fport_input_waiting);
899
900 return tc;
901 }
902
903 void
904 scm_init_fports ()
905 {
906 scm_tc16_fport = scm_make_fptob ();
907
908 scm_c_define ("_IOFBF", scm_from_int (_IOFBF));
909 scm_c_define ("_IOLBF", scm_from_int (_IOLBF));
910 scm_c_define ("_IONBF", scm_from_int (_IONBF));
911
912 #include "libguile/fports.x"
913 }
914
915 /*
916 Local Variables:
917 c-file-style: "gnu"
918 End:
919 */