Merge branch 'master' into boehm-demers-weiser-gc
[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;
322 int flags = 0;
323 char *file;
324 char *md;
325 char *ptr;
326
327 scm_dynwind_begin (0);
328
329 file = scm_to_locale_string (filename);
330 scm_dynwind_free (file);
331
332 md = scm_to_locale_string (mode);
333 scm_dynwind_free (md);
334
335 switch (*md)
336 {
337 case 'r':
338 flags |= O_RDONLY;
339 break;
340 case 'w':
341 flags |= O_WRONLY | O_CREAT | O_TRUNC;
342 break;
343 case 'a':
344 flags |= O_WRONLY | O_CREAT | O_APPEND;
345 break;
346 default:
347 scm_out_of_range (FUNC_NAME, mode);
348 }
349 ptr = md + 1;
350 while (*ptr != '\0')
351 {
352 switch (*ptr)
353 {
354 case '+':
355 flags = (flags & ~(O_RDONLY | O_WRONLY)) | O_RDWR;
356 break;
357 case 'b':
358 #if defined (O_BINARY)
359 flags |= O_BINARY;
360 #endif
361 break;
362 case '0': /* unbuffered: handled later. */
363 case 'l': /* line buffered: handled during output. */
364 break;
365 default:
366 scm_out_of_range (FUNC_NAME, mode);
367 }
368 ptr++;
369 }
370 SCM_SYSCALL (fdes = open_or_open64 (file, flags, 0666));
371 if (fdes == -1)
372 {
373 int en = errno;
374
375 SCM_SYSERROR_MSG ("~A: ~S",
376 scm_cons (scm_strerror (scm_from_int (en)),
377 scm_cons (filename, SCM_EOL)), en);
378 }
379 port = scm_i_fdes_to_port (fdes, scm_i_mode_bits (mode), filename);
380
381 scm_dynwind_end ();
382
383 return port;
384 }
385 #undef FUNC_NAME
386
387 \f
388 #ifdef __MINGW32__
389 /*
390 * Try getting the appropiate file flags for a given file descriptor
391 * under Windows. This incorporates some fancy operations because Windows
392 * differentiates between file, pipe and socket descriptors.
393 */
394 #ifndef O_ACCMODE
395 # define O_ACCMODE 0x0003
396 #endif
397
398 static int getflags (int fdes)
399 {
400 int flags = 0;
401 struct stat buf;
402 int error, optlen = sizeof (int);
403
404 /* Is this a socket ? */
405 if (getsockopt (fdes, SOL_SOCKET, SO_ERROR, (void *) &error, &optlen) >= 0)
406 flags = O_RDWR;
407 /* Maybe a regular file ? */
408 else if (fstat (fdes, &buf) < 0)
409 flags = -1;
410 else
411 {
412 /* Or an anonymous pipe handle ? */
413 if (buf.st_mode & _S_IFIFO)
414 flags = PeekNamedPipe ((HANDLE) _get_osfhandle (fdes), NULL, 0,
415 NULL, NULL, NULL) ? O_RDONLY : O_WRONLY;
416 /* stdin ? */
417 else if (fdes == fileno (stdin) && isatty (fdes))
418 flags = O_RDONLY;
419 /* stdout / stderr ? */
420 else if ((fdes == fileno (stdout) || fdes == fileno (stderr)) &&
421 isatty (fdes))
422 flags = O_WRONLY;
423 else
424 flags = buf.st_mode;
425 }
426 return flags;
427 }
428 #endif /* __MINGW32__ */
429
430 /* Building Guile ports from a file descriptor. */
431
432 /* Build a Scheme port from an open file descriptor `fdes'.
433 MODE indicates whether FILE is open for reading or writing; it uses
434 the same notation as open-file's second argument.
435 NAME is a string to be used as the port's filename.
436 */
437 SCM
438 scm_i_fdes_to_port (int fdes, long mode_bits, SCM name)
439 #define FUNC_NAME "scm_fdes_to_port"
440 {
441 SCM port;
442 scm_t_port *pt;
443 int flags;
444
445 /* test that fdes is valid. */
446 #ifdef __MINGW32__
447 flags = getflags (fdes);
448 #else
449 flags = fcntl (fdes, F_GETFL, 0);
450 #endif
451 if (flags == -1)
452 SCM_SYSERROR;
453 flags &= O_ACCMODE;
454 if (flags != O_RDWR
455 && ((flags != O_WRONLY && (mode_bits & SCM_WRTNG))
456 || (flags != O_RDONLY && (mode_bits & SCM_RDNG))))
457 {
458 SCM_MISC_ERROR ("requested file mode not available on fdes", SCM_EOL);
459 }
460
461 scm_i_scm_pthread_mutex_lock (&scm_i_port_table_mutex);
462
463 port = scm_new_port_table_entry (scm_tc16_fport);
464 SCM_SET_CELL_TYPE(port, scm_tc16_fport | mode_bits);
465 pt = SCM_PTAB_ENTRY(port);
466 {
467 scm_t_fport *fp
468 = (scm_t_fport *) scm_gc_malloc_pointerless (sizeof (scm_t_fport),
469 "file port");
470
471 fp->fdes = fdes;
472 pt->rw_random = SCM_FDES_RANDOM_P (fdes);
473 SCM_SETSTREAM (port, fp);
474 if (mode_bits & SCM_BUF0)
475 scm_fport_buffer_add (port, 0, 0);
476 else
477 scm_fport_buffer_add (port, -1, -1);
478 }
479 SCM_SET_FILENAME (port, name);
480 scm_i_pthread_mutex_unlock (&scm_i_port_table_mutex);
481 return port;
482 }
483 #undef FUNC_NAME
484
485 SCM
486 scm_fdes_to_port (int fdes, char *mode, SCM name)
487 {
488 return scm_i_fdes_to_port (fdes, scm_mode_bits (mode), name);
489 }
490
491 /* Return a lower bound on the number of bytes available for input. */
492 static int
493 fport_input_waiting (SCM port)
494 {
495 #ifdef HAVE_SELECT
496 int fdes = SCM_FSTREAM (port)->fdes;
497 struct timeval timeout;
498 SELECT_TYPE read_set;
499 SELECT_TYPE write_set;
500 SELECT_TYPE except_set;
501
502 FD_ZERO (&read_set);
503 FD_ZERO (&write_set);
504 FD_ZERO (&except_set);
505
506 FD_SET (fdes, &read_set);
507
508 timeout.tv_sec = 0;
509 timeout.tv_usec = 0;
510
511 if (select (SELECT_SET_SIZE,
512 &read_set, &write_set, &except_set, &timeout)
513 < 0)
514 scm_syserror ("fport_input_waiting");
515 return FD_ISSET (fdes, &read_set) ? 1 : 0;
516
517 #elif HAVE_IOCTL && defined (FIONREAD)
518 /* Note: cannot test just defined(FIONREAD) here, since mingw has FIONREAD
519 (for use with winsock ioctlsocket()) but not ioctl(). */
520 int fdes = SCM_FSTREAM (port)->fdes;
521 int remir;
522 ioctl(fdes, FIONREAD, &remir);
523 return remir;
524
525 #else
526 scm_misc_error ("fport_input_waiting",
527 "Not fully implemented on this platform",
528 SCM_EOL);
529 #endif
530 }
531
532 \f
533 static int
534 fport_print (SCM exp, SCM port, scm_print_state *pstate SCM_UNUSED)
535 {
536 scm_puts ("#<", port);
537 scm_print_port_mode (exp, port);
538 if (SCM_OPFPORTP (exp))
539 {
540 int fdes;
541 SCM name = SCM_FILENAME (exp);
542 if (scm_is_string (name) || scm_is_symbol (name))
543 scm_display (name, port);
544 else
545 scm_puts (SCM_PTOBNAME (SCM_PTOBNUM (exp)), port);
546 scm_putc (' ', port);
547 fdes = (SCM_FSTREAM (exp))->fdes;
548
549 #ifdef HAVE_TTYNAME
550 if (isatty (fdes))
551 scm_display (scm_ttyname (exp), port);
552 else
553 #endif /* HAVE_TTYNAME */
554 scm_intprint (fdes, 10, port);
555 }
556 else
557 {
558 scm_puts (SCM_PTOBNAME (SCM_PTOBNUM (exp)), port);
559 scm_putc (' ', port);
560 scm_uintprint ((scm_t_bits) SCM_PTAB_ENTRY (exp), 16, port);
561 }
562 scm_putc ('>', port);
563 return 1;
564 }
565
566 #ifndef __MINGW32__
567 /* thread-local block for input on fport's fdes. */
568 static void
569 fport_wait_for_input (SCM port)
570 {
571 int fdes = SCM_FSTREAM (port)->fdes;
572
573 if (!fport_input_waiting (port))
574 {
575 int n;
576 SELECT_TYPE readfds;
577 int flags = fcntl (fdes, F_GETFL);
578
579 if (flags == -1)
580 scm_syserror ("scm_fdes_wait_for_input");
581 if (!(flags & O_NONBLOCK))
582 do
583 {
584 FD_ZERO (&readfds);
585 FD_SET (fdes, &readfds);
586 n = scm_std_select (fdes + 1, &readfds, NULL, NULL, NULL);
587 }
588 while (n == -1 && errno == EINTR);
589 }
590 }
591 #endif /* !__MINGW32__ */
592
593 static void fport_flush (SCM port);
594
595 /* fill a port's read-buffer with a single read. returns the first
596 char or EOF if end of file. */
597 static int
598 fport_fill_input (SCM port)
599 {
600 long count;
601 scm_t_port *pt = SCM_PTAB_ENTRY (port);
602 scm_t_fport *fp = SCM_FSTREAM (port);
603
604 #ifndef __MINGW32__
605 fport_wait_for_input (port);
606 #endif /* !__MINGW32__ */
607 SCM_SYSCALL (count = read (fp->fdes, pt->read_buf, pt->read_buf_size));
608 if (count == -1)
609 scm_syserror ("fport_fill_input");
610 if (count == 0)
611 return EOF;
612 else
613 {
614 pt->read_pos = pt->read_buf;
615 pt->read_end = pt->read_buf + count;
616 return *pt->read_buf;
617 }
618 }
619
620 static scm_t_off
621 fport_seek (SCM port, scm_t_off offset, int whence)
622 {
623 scm_t_port *pt = SCM_PTAB_ENTRY (port);
624 scm_t_fport *fp = SCM_FSTREAM (port);
625 off_t_or_off64_t rv;
626 off_t_or_off64_t result;
627
628 if (pt->rw_active == SCM_PORT_WRITE)
629 {
630 if (offset != 0 || whence != SEEK_CUR)
631 {
632 fport_flush (port);
633 result = rv = lseek_or_lseek64 (fp->fdes, offset, whence);
634 }
635 else
636 {
637 /* read current position without disturbing the buffer. */
638 rv = lseek_or_lseek64 (fp->fdes, offset, whence);
639 result = rv + (pt->write_pos - pt->write_buf);
640 }
641 }
642 else if (pt->rw_active == SCM_PORT_READ)
643 {
644 if (offset != 0 || whence != SEEK_CUR)
645 {
646 /* could expand to avoid a second seek. */
647 scm_end_input (port);
648 result = rv = lseek_or_lseek64 (fp->fdes, offset, whence);
649 }
650 else
651 {
652 /* read current position without disturbing the buffer
653 (particularly the unread-char buffer). */
654 rv = lseek_or_lseek64 (fp->fdes, offset, whence);
655 result = rv - (pt->read_end - pt->read_pos);
656
657 if (pt->read_buf == pt->putback_buf)
658 result -= pt->saved_read_end - pt->saved_read_pos;
659 }
660 }
661 else /* SCM_PORT_NEITHER */
662 {
663 result = rv = lseek_or_lseek64 (fp->fdes, offset, whence);
664 }
665
666 if (rv == -1)
667 scm_syserror ("fport_seek");
668
669 return result;
670 }
671
672 static void
673 fport_truncate (SCM port, scm_t_off length)
674 {
675 scm_t_fport *fp = SCM_FSTREAM (port);
676
677 if (ftruncate (fp->fdes, length) == -1)
678 scm_syserror ("ftruncate");
679 }
680
681 /* helper for fport_write: try to write data, using multiple system
682 calls if required. */
683 #define FUNC_NAME "write_all"
684 static void write_all (SCM port, const void *data, size_t remaining)
685 {
686 int fdes = SCM_FSTREAM (port)->fdes;
687
688 while (remaining > 0)
689 {
690 size_t done;
691
692 SCM_SYSCALL (done = write (fdes, data, remaining));
693
694 if (done == -1)
695 SCM_SYSERROR;
696 remaining -= done;
697 data = ((const char *) data) + done;
698 }
699 }
700 #undef FUNC_NAME
701
702 static void
703 fport_write (SCM port, const void *data, size_t size)
704 {
705 /* this procedure tries to minimize the number of writes/flushes. */
706 scm_t_port *pt = SCM_PTAB_ENTRY (port);
707
708 if (pt->write_buf == &pt->shortbuf
709 || (pt->write_pos == pt->write_buf && size >= pt->write_buf_size))
710 {
711 /* "unbuffered" port, or
712 port with empty buffer and data won't fit in buffer. */
713 write_all (port, data, size);
714 return;
715 }
716
717 {
718 scm_t_off space = pt->write_end - pt->write_pos;
719
720 if (size <= space)
721 {
722 /* data fits in buffer. */
723 memcpy (pt->write_pos, data, size);
724 pt->write_pos += size;
725 if (pt->write_pos == pt->write_end)
726 {
727 fport_flush (port);
728 /* we can skip the line-buffering check if nothing's buffered. */
729 return;
730 }
731 }
732 else
733 {
734 memcpy (pt->write_pos, data, space);
735 pt->write_pos = pt->write_end;
736 fport_flush (port);
737 {
738 const void *ptr = ((const char *) data) + space;
739 size_t remaining = size - space;
740
741 if (size >= pt->write_buf_size)
742 {
743 write_all (port, ptr, remaining);
744 return;
745 }
746 else
747 {
748 memcpy (pt->write_pos, ptr, remaining);
749 pt->write_pos += remaining;
750 }
751 }
752 }
753
754 /* handle line buffering. */
755 if ((SCM_CELL_WORD_0 (port) & SCM_BUFLINE) && memchr (data, '\n', size))
756 fport_flush (port);
757 }
758 }
759
760 /* becomes 1 when process is exiting: normal exception handling won't
761 work by this time. */
762 extern int scm_i_terminating;
763
764 static void
765 fport_flush (SCM port)
766 {
767 scm_t_port *pt = SCM_PTAB_ENTRY (port);
768 scm_t_fport *fp = SCM_FSTREAM (port);
769 unsigned char *ptr = pt->write_buf;
770 long init_size = pt->write_pos - pt->write_buf;
771 long remaining = init_size;
772
773 while (remaining > 0)
774 {
775 long count;
776
777 SCM_SYSCALL (count = write (fp->fdes, ptr, remaining));
778 if (count < 0)
779 {
780 /* error. assume nothing was written this call, but
781 fix up the buffer for any previous successful writes. */
782 long done = init_size - remaining;
783
784 if (done > 0)
785 {
786 int i;
787
788 for (i = 0; i < remaining; i++)
789 {
790 *(pt->write_buf + i) = *(pt->write_buf + done + i);
791 }
792 pt->write_pos = pt->write_buf + remaining;
793 }
794 if (scm_i_terminating)
795 {
796 const char *msg = "Error: could not flush file-descriptor ";
797 char buf[11];
798
799 full_write (2, msg, strlen (msg));
800 sprintf (buf, "%d\n", fp->fdes);
801 full_write (2, buf, strlen (buf));
802
803 count = remaining;
804 }
805 else if (scm_gc_running_p)
806 {
807 /* silently ignore the error. scm_error would abort if we
808 called it now. */
809 count = remaining;
810 }
811 else
812 scm_syserror ("fport_flush");
813 }
814 ptr += count;
815 remaining -= count;
816 }
817 pt->write_pos = pt->write_buf;
818 pt->rw_active = SCM_PORT_NEITHER;
819 }
820
821 /* clear the read buffer and adjust the file position for unread bytes. */
822 static void
823 fport_end_input (SCM port, int offset)
824 {
825 scm_t_fport *fp = SCM_FSTREAM (port);
826 scm_t_port *pt = SCM_PTAB_ENTRY (port);
827
828 offset += pt->read_end - pt->read_pos;
829
830 if (offset > 0)
831 {
832 pt->read_pos = pt->read_end;
833 /* will throw error if unread-char used at beginning of file
834 then attempting to write. seems correct. */
835 if (lseek (fp->fdes, -offset, SEEK_CUR) == -1)
836 scm_syserror ("fport_end_input");
837 }
838 pt->rw_active = SCM_PORT_NEITHER;
839 }
840
841 static int
842 fport_close (SCM port)
843 {
844 scm_t_fport *fp = SCM_FSTREAM (port);
845 scm_t_port *pt = SCM_PTAB_ENTRY (port);
846 int rv;
847
848 fport_flush (port);
849 SCM_SYSCALL (rv = close (fp->fdes));
850 if (rv == -1 && errno != EBADF)
851 {
852 if (scm_gc_running_p)
853 /* silently ignore the error. scm_error would abort if we
854 called it now. */
855 ;
856 else
857 scm_syserror ("fport_close");
858 }
859 if (pt->read_buf == pt->putback_buf)
860 pt->read_buf = pt->saved_read_buf;
861 if (pt->read_buf != &pt->shortbuf)
862 scm_gc_free (pt->read_buf, pt->read_buf_size, "port buffer");
863 if (pt->write_buf != &pt->shortbuf)
864 scm_gc_free (pt->write_buf, pt->write_buf_size, "port buffer");
865 scm_gc_free (fp, sizeof (*fp), "file port");
866 return rv;
867 }
868
869 static size_t
870 fport_free (SCM port)
871 {
872 fport_close (port);
873 return 0;
874 }
875
876 static scm_t_bits
877 scm_make_fptob ()
878 {
879 scm_t_bits tc = scm_make_port_type ("file", fport_fill_input, fport_write);
880
881 scm_set_port_free (tc, fport_free);
882 scm_set_port_print (tc, fport_print);
883 scm_set_port_flush (tc, fport_flush);
884 scm_set_port_end_input (tc, fport_end_input);
885 scm_set_port_close (tc, fport_close);
886 scm_set_port_seek (tc, fport_seek);
887 scm_set_port_truncate (tc, fport_truncate);
888 scm_set_port_input_waiting (tc, fport_input_waiting);
889
890 return tc;
891 }
892
893 void
894 scm_init_fports ()
895 {
896 scm_tc16_fport = scm_make_fptob ();
897
898 scm_c_define ("_IOFBF", scm_from_int (_IOFBF));
899 scm_c_define ("_IOLBF", scm_from_int (_IOLBF));
900 scm_c_define ("_IONBF", scm_from_int (_IONBF));
901
902 #include "libguile/fports.x"
903 }
904
905 /*
906 Local Variables:
907 c-file-style: "gnu"
908 End:
909 */