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