Merge commit 'd364a8971828e38e8f9112b711066f4962bb400e'
[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, 2012, 2013 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 #define _GNU_SOURCE /* ask for LONG_LONG_MAX/LONG_LONG_MIN */
24
25 #ifdef HAVE_CONFIG_H
26 # include <config.h>
27 #endif
28
29 #include <stdio.h>
30 #include <fcntl.h>
31
32 #ifdef HAVE_STRING_H
33 #include <string.h>
34 #endif
35 #ifdef HAVE_UNISTD_H
36 #include <unistd.h>
37 #endif
38 #ifdef HAVE_IO_H
39 #include <io.h>
40 #endif
41 #ifdef HAVE_STRUCT_STAT_ST_BLKSIZE
42 #include <sys/stat.h>
43 #endif
44 #include <poll.h>
45 #include <errno.h>
46 #include <sys/types.h>
47 #include <sys/stat.h>
48 #include <sys/select.h>
49
50 #include <full-write.h>
51
52 #include "libguile/_scm.h"
53 #include "libguile/strings.h"
54 #include "libguile/validate.h"
55 #include "libguile/gc.h"
56 #include "libguile/posix.h"
57 #include "libguile/dynwind.h"
58 #include "libguile/hashtab.h"
59
60 #include "libguile/fports.h"
61
62 #if SIZEOF_OFF_T == SIZEOF_INT
63 #define OFF_T_MAX INT_MAX
64 #define OFF_T_MIN INT_MIN
65 #elif SIZEOF_OFF_T == SIZEOF_LONG
66 #define OFF_T_MAX LONG_MAX
67 #define OFF_T_MIN LONG_MIN
68 #elif SIZEOF_OFF_T == SIZEOF_LONG_LONG
69 #define OFF_T_MAX LONG_LONG_MAX
70 #define OFF_T_MIN LONG_LONG_MIN
71 #else
72 #error Oops, unknown OFF_T size
73 #endif
74
75 scm_t_bits scm_tc16_fport;
76
77
78 /* default buffer size, used if the O/S won't supply a value. */
79 static const size_t default_buffer_size = 1024;
80
81 /* create FPORT buffer with specified sizes (or -1 to use default size or
82 0 for no buffer. */
83 static void
84 scm_fport_buffer_add (SCM port, long read_size, int write_size)
85 #define FUNC_NAME "scm_fport_buffer_add"
86 {
87 scm_t_port *pt = SCM_PTAB_ENTRY (port);
88
89 if (read_size == -1 || write_size == -1)
90 {
91 size_t default_size;
92 #ifdef HAVE_STRUCT_STAT_ST_BLKSIZE
93 struct stat st;
94 scm_t_fport *fp = SCM_FSTREAM (port);
95
96 default_size = (fstat (fp->fdes, &st) == -1) ? default_buffer_size
97 : st.st_blksize;
98 #else
99 default_size = default_buffer_size;
100 #endif
101 if (read_size == -1)
102 read_size = default_size;
103 if (write_size == -1)
104 write_size = default_size;
105 }
106
107 if (SCM_INPUT_PORT_P (port) && read_size > 0)
108 {
109 pt->read_buf = scm_gc_malloc_pointerless (read_size, "port buffer");
110 pt->read_pos = pt->read_end = pt->read_buf;
111 pt->read_buf_size = read_size;
112 }
113 else
114 {
115 pt->read_pos = pt->read_buf = pt->read_end = &pt->shortbuf;
116 pt->read_buf_size = 1;
117 }
118
119 if (SCM_OUTPUT_PORT_P (port) && write_size > 0)
120 {
121 pt->write_buf = scm_gc_malloc_pointerless (write_size, "port buffer");
122 pt->write_pos = pt->write_buf;
123 pt->write_buf_size = write_size;
124 }
125 else
126 {
127 pt->write_buf = pt->write_pos = &pt->shortbuf;
128 pt->write_buf_size = 1;
129 }
130
131 pt->write_end = pt->write_buf + pt->write_buf_size;
132 if (read_size > 0 || write_size > 0)
133 SCM_SET_CELL_WORD_0 (port, SCM_CELL_WORD_0 (port) & ~SCM_BUF0);
134 else
135 SCM_SET_CELL_WORD_0 (port, SCM_CELL_WORD_0 (port) | SCM_BUF0);
136 }
137 #undef FUNC_NAME
138
139 SCM_DEFINE (scm_setvbuf, "setvbuf", 2, 1, 0,
140 (SCM port, SCM mode, SCM size),
141 "Set the buffering mode for @var{port}. @var{mode} can be:\n"
142 "@table @code\n"
143 "@item _IONBF\n"
144 "non-buffered\n"
145 "@item _IOLBF\n"
146 "line buffered\n"
147 "@item _IOFBF\n"
148 "block buffered, using a newly allocated buffer of @var{size} bytes.\n"
149 "If @var{size} is omitted, a default size will be used.\n"
150 "@end table")
151 #define FUNC_NAME s_scm_setvbuf
152 {
153 int cmode;
154 long csize;
155 size_t ndrained;
156 char *drained;
157 scm_t_port *pt;
158
159 port = SCM_COERCE_OUTPORT (port);
160
161 SCM_VALIDATE_OPFPORT (1,port);
162 cmode = scm_to_int (mode);
163 if (cmode != _IONBF && cmode != _IOFBF && cmode != _IOLBF)
164 scm_out_of_range (FUNC_NAME, mode);
165
166 if (cmode == _IOLBF)
167 {
168 SCM_SET_CELL_WORD_0 (port, SCM_CELL_WORD_0 (port) | SCM_BUFLINE);
169 cmode = _IOFBF;
170 }
171 else
172 {
173 SCM_SET_CELL_WORD_0 (port, SCM_CELL_WORD_0 (port) & ~(scm_t_bits)SCM_BUFLINE);
174 }
175
176 if (SCM_UNBNDP (size))
177 {
178 if (cmode == _IOFBF)
179 csize = -1;
180 else
181 csize = 0;
182 }
183 else
184 {
185 csize = scm_to_int (size);
186 if (csize < 0 || (cmode == _IONBF && csize > 0))
187 scm_out_of_range (FUNC_NAME, size);
188 }
189
190 pt = SCM_PTAB_ENTRY (port);
191
192 if (SCM_INPUT_PORT_P (port))
193 {
194 /* Drain pending input from PORT. Don't use `scm_drain_input' since
195 it returns a string, whereas we want binary input here. */
196 ndrained = pt->read_end - pt->read_pos;
197 if (pt->read_buf == pt->putback_buf)
198 ndrained += pt->saved_read_end - pt->saved_read_pos;
199
200 if (ndrained > 0)
201 {
202 drained = scm_gc_malloc_pointerless (ndrained, "file port");
203 scm_take_from_input_buffers (port, drained, ndrained);
204 }
205 }
206 else
207 ndrained = 0;
208
209 if (SCM_OUTPUT_PORT_P (port))
210 scm_flush_unlocked (port);
211
212 if (pt->read_buf == pt->putback_buf)
213 {
214 pt->read_buf = pt->saved_read_buf;
215 pt->read_pos = pt->saved_read_pos;
216 pt->read_end = pt->saved_read_end;
217 pt->read_buf_size = pt->saved_read_buf_size;
218 }
219 if (pt->read_buf != &pt->shortbuf)
220 scm_gc_free (pt->read_buf, pt->read_buf_size, "port buffer");
221 if (pt->write_buf != &pt->shortbuf)
222 scm_gc_free (pt->write_buf, pt->write_buf_size, "port buffer");
223
224 scm_fport_buffer_add (port, csize, csize);
225
226 if (ndrained > 0)
227 /* Put DRAINED back to PORT. */
228 scm_unget_bytes ((unsigned char *) drained, ndrained, port);
229
230 return SCM_UNSPECIFIED;
231 }
232 #undef FUNC_NAME
233
234 /* Move ports with the specified file descriptor to new descriptors,
235 * resetting the revealed count to 0.
236 */
237 static void
238 scm_i_evict_port (void *closure, SCM port)
239 {
240 int fd = * (int*) closure;
241
242 if (SCM_FPORTP (port))
243 {
244 scm_t_port *p;
245 scm_t_fport *fp;
246
247 /* XXX: In some cases, we can encounter a port with no associated ptab
248 entry. */
249 p = SCM_PTAB_ENTRY (port);
250 fp = (p != NULL) ? (scm_t_fport *) p->stream : NULL;
251
252 if ((fp != NULL) && (fp->fdes == fd))
253 {
254 fp->fdes = dup (fd);
255 if (fp->fdes == -1)
256 scm_syserror ("scm_evict_ports");
257 scm_set_port_revealed_x (port, scm_from_int (0));
258 }
259 }
260 }
261
262 void
263 scm_evict_ports (int fd)
264 {
265 scm_c_port_for_each (scm_i_evict_port, (void *) &fd);
266 }
267
268
269 SCM_DEFINE (scm_file_port_p, "file-port?", 1, 0, 0,
270 (SCM obj),
271 "Determine whether @var{obj} is a port that is related to a file.")
272 #define FUNC_NAME s_scm_file_port_p
273 {
274 return scm_from_bool (SCM_FPORTP (obj));
275 }
276 #undef FUNC_NAME
277
278
279 static SCM sys_file_port_name_canonicalization;
280 SCM_SYMBOL (sym_relative, "relative");
281 SCM_SYMBOL (sym_absolute, "absolute");
282
283 static SCM
284 fport_canonicalize_filename (SCM filename)
285 {
286 SCM mode = scm_fluid_ref (sys_file_port_name_canonicalization);
287
288 if (!scm_is_string (filename))
289 {
290 return filename;
291 }
292 else if (scm_is_eq (mode, sym_relative))
293 {
294 SCM path, rel;
295
296 path = scm_variable_ref (scm_c_module_lookup (scm_the_root_module (),
297 "%load-path"));
298 rel = scm_i_relativize_path (filename, path);
299
300 return scm_is_true (rel) ? rel : filename;
301 }
302 else if (scm_is_eq (mode, sym_absolute))
303 {
304 char *str, *canon;
305
306 str = scm_to_locale_string (filename);
307 canon = canonicalize_file_name (str);
308 free (str);
309
310 return canon ? scm_take_locale_string (canon) : filename;
311 }
312 else
313 {
314 return filename;
315 }
316 }
317
318 /* scm_open_file_with_encoding
319 Return a new port open on a given file.
320
321 The mode string must match the pattern: [rwa+]** which
322 is interpreted in the usual unix way.
323
324 Unless binary mode is requested, the character encoding of the new
325 port is determined as follows: First, if GUESS_ENCODING is true,
326 'file-encoding' is used to guess the encoding of the file. If
327 GUESS_ENCODING is false or if 'file-encoding' fails, ENCODING is used
328 unless it is also false. As a last resort, the default port encoding
329 is used. It is an error to pass a non-false GUESS_ENCODING or
330 ENCODING if binary mode is requested.
331
332 Return the new port. */
333 SCM
334 scm_open_file_with_encoding (SCM filename, SCM mode,
335 SCM guess_encoding, SCM encoding)
336 #define FUNC_NAME "open-file"
337 {
338 SCM port;
339 int fdes, flags = 0, binary = 0;
340 unsigned int retries;
341 char *file, *md, *ptr;
342
343 if (SCM_UNLIKELY (!(scm_is_false (encoding) || scm_is_string (encoding))))
344 scm_wrong_type_arg_msg (FUNC_NAME, 0, encoding,
345 "encoding to be string or false");
346
347 scm_dynwind_begin (0);
348
349 file = scm_to_locale_string (filename);
350 scm_dynwind_free (file);
351
352 md = scm_to_locale_string (mode);
353 scm_dynwind_free (md);
354
355 switch (*md)
356 {
357 case 'r':
358 flags |= O_RDONLY;
359 break;
360 case 'w':
361 flags |= O_WRONLY | O_CREAT | O_TRUNC;
362 break;
363 case 'a':
364 flags |= O_WRONLY | O_CREAT | O_APPEND;
365 break;
366 default:
367 scm_out_of_range (FUNC_NAME, mode);
368 }
369 ptr = md + 1;
370 while (*ptr != '\0')
371 {
372 switch (*ptr)
373 {
374 case '+':
375 flags = (flags & ~(O_RDONLY | O_WRONLY)) | O_RDWR;
376 break;
377 case 'b':
378 binary = 1;
379 #if defined (O_BINARY)
380 flags |= O_BINARY;
381 #endif
382 break;
383 case '0': /* unbuffered: handled later. */
384 case 'l': /* line buffered: handled during output. */
385 break;
386 default:
387 scm_out_of_range (FUNC_NAME, mode);
388 }
389 ptr++;
390 }
391
392 for (retries = 0, fdes = -1;
393 fdes < 0 && retries < 2;
394 retries++)
395 {
396 SCM_SYSCALL (fdes = open_or_open64 (file, flags, 0666));
397 if (fdes == -1)
398 {
399 int en = errno;
400
401 if (en == EMFILE && retries == 0)
402 /* Run the GC in case it collects open file ports that are no
403 longer referenced. */
404 scm_i_gc (FUNC_NAME);
405 else
406 SCM_SYSERROR_MSG ("~A: ~S",
407 scm_cons (scm_strerror (scm_from_int (en)),
408 scm_cons (filename, SCM_EOL)), en);
409 }
410 }
411
412 /* Create a port from this file descriptor. The port's encoding is initially
413 %default-port-encoding. */
414 port = scm_i_fdes_to_port (fdes, scm_i_mode_bits (mode),
415 fport_canonicalize_filename (filename));
416
417 if (binary)
418 {
419 if (scm_is_true (encoding))
420 scm_misc_error (FUNC_NAME,
421 "Encoding specified on a binary port",
422 scm_list_1 (encoding));
423 if (scm_is_true (guess_encoding))
424 scm_misc_error (FUNC_NAME,
425 "Request to guess encoding on a binary port",
426 SCM_EOL);
427
428 /* Use the binary-friendly ISO-8859-1 encoding. */
429 scm_i_set_port_encoding_x (port, NULL);
430 }
431 else
432 {
433 char *enc = NULL;
434
435 if (scm_is_true (guess_encoding))
436 {
437 if (SCM_INPUT_PORT_P (port))
438 enc = scm_i_scan_for_encoding (port);
439 else
440 scm_misc_error (FUNC_NAME,
441 "Request to guess encoding on an output-only port",
442 SCM_EOL);
443 }
444
445 if (!enc && scm_is_true (encoding))
446 {
447 char *buf = scm_to_latin1_string (encoding);
448 enc = scm_gc_strdup (buf, "encoding");
449 free (buf);
450 }
451
452 if (enc)
453 scm_i_set_port_encoding_x (port, enc);
454 }
455
456 scm_dynwind_end ();
457
458 return port;
459 }
460 #undef FUNC_NAME
461
462 SCM
463 scm_open_file (SCM filename, SCM mode)
464 {
465 return scm_open_file_with_encoding (filename, mode, SCM_BOOL_F, SCM_BOOL_F);
466 }
467
468 /* We can't define these using SCM_KEYWORD, because keywords have not
469 yet been initialized when scm_init_fports is called. */
470 static SCM k_guess_encoding = SCM_UNDEFINED;
471 static SCM k_encoding = SCM_UNDEFINED;
472
473 SCM_INTERNAL SCM scm_i_open_file (SCM, SCM, SCM);
474
475 SCM_DEFINE (scm_i_open_file, "open-file", 2, 0, 1,
476 (SCM filename, SCM mode, SCM keyword_args),
477 "Open the file whose name is @var{filename}, and return a port\n"
478 "representing that file. The attributes of the port are\n"
479 "determined by the @var{mode} string. The way in which this is\n"
480 "interpreted is similar to C stdio. The first character must be\n"
481 "one of the following:\n"
482 "@table @samp\n"
483 "@item r\n"
484 "Open an existing file for input.\n"
485 "@item w\n"
486 "Open a file for output, creating it if it doesn't already exist\n"
487 "or removing its contents if it does.\n"
488 "@item a\n"
489 "Open a file for output, creating it if it doesn't already\n"
490 "exist. All writes to the port will go to the end of the file.\n"
491 "The \"append mode\" can be turned off while the port is in use\n"
492 "@pxref{Ports and File Descriptors, fcntl}\n"
493 "@end table\n"
494 "The following additional characters can be appended:\n"
495 "@table @samp\n"
496 "@item b\n"
497 "Open the underlying file in binary mode, if supported by the system.\n"
498 "Also, open the file using the binary-compatible character encoding\n"
499 "\"ISO-8859-1\", ignoring the default port encoding.\n"
500 "@item +\n"
501 "Open the port for both input and output. E.g., @code{r+}: open\n"
502 "an existing file for both input and output.\n"
503 "@item 0\n"
504 "Create an \"unbuffered\" port. In this case input and output\n"
505 "operations are passed directly to the underlying port\n"
506 "implementation without additional buffering. This is likely to\n"
507 "slow down I/O operations. The buffering mode can be changed\n"
508 "while a port is in use @pxref{Ports and File Descriptors,\n"
509 "setvbuf}\n"
510 "@item l\n"
511 "Add line-buffering to the port. The port output buffer will be\n"
512 "automatically flushed whenever a newline character is written.\n"
513 "@end table\n"
514 "In theory we could create read/write ports which were buffered\n"
515 "in one direction only. However this isn't included in the\n"
516 "current interfaces. If a file cannot be opened with the access\n"
517 "requested, @code{open-file} throws an exception.")
518 #define FUNC_NAME s_scm_i_open_file
519 {
520 SCM encoding = SCM_BOOL_F;
521 SCM guess_encoding = SCM_BOOL_F;
522
523 scm_c_bind_keyword_arguments (FUNC_NAME, keyword_args, 0,
524 k_guess_encoding, &guess_encoding,
525 k_encoding, &encoding,
526 SCM_UNDEFINED);
527
528 return scm_open_file_with_encoding (filename, mode,
529 guess_encoding, encoding);
530 }
531 #undef FUNC_NAME
532
533 \f
534 /* Building Guile ports from a file descriptor. */
535
536 /* Build a Scheme port from an open file descriptor `fdes'.
537 MODE indicates whether FILE is open for reading or writing; it uses
538 the same notation as open-file's second argument.
539 NAME is a string to be used as the port's filename.
540 */
541 SCM
542 scm_i_fdes_to_port (int fdes, long mode_bits, SCM name)
543 #define FUNC_NAME "scm_fdes_to_port"
544 {
545 SCM port;
546 scm_t_fport *fp;
547
548 /* Test that fdes is valid. */
549 #ifdef F_GETFL
550 int flags = fcntl (fdes, F_GETFL, 0);
551 if (flags == -1)
552 SCM_SYSERROR;
553 flags &= O_ACCMODE;
554 if (flags != O_RDWR
555 && ((flags != O_WRONLY && (mode_bits & SCM_WRTNG))
556 || (flags != O_RDONLY && (mode_bits & SCM_RDNG))))
557 {
558 SCM_MISC_ERROR ("requested file mode not available on fdes", SCM_EOL);
559 }
560 #else
561 /* If we don't have F_GETFL, as on mingw, at least we can test that
562 it is a valid file descriptor. */
563 struct stat st;
564 if (fstat (fdes, &st) != 0)
565 SCM_SYSERROR;
566 #endif
567
568 fp = (scm_t_fport *) scm_gc_malloc_pointerless (sizeof (scm_t_fport),
569 "file port");
570 fp->fdes = fdes;
571
572 port = scm_c_make_port (scm_tc16_fport, mode_bits, (scm_t_bits)fp);
573
574 SCM_PTAB_ENTRY (port)->rw_random = SCM_FDES_RANDOM_P (fdes);
575
576 if (mode_bits & SCM_BUF0)
577 scm_fport_buffer_add (port, 0, 0);
578 else
579 scm_fport_buffer_add (port, -1, -1);
580
581 SCM_SET_FILENAME (port, name);
582
583 return port;
584 }
585 #undef FUNC_NAME
586
587 SCM
588 scm_fdes_to_port (int fdes, char *mode, SCM name)
589 {
590 return scm_i_fdes_to_port (fdes, scm_mode_bits (mode), name);
591 }
592
593 /* Return a lower bound on the number of bytes available for input. */
594 static int
595 fport_input_waiting (SCM port)
596 {
597 int fdes = SCM_FSTREAM (port)->fdes;
598
599 struct pollfd pollfd = { fdes, POLLIN, 0 };
600
601 if (poll (&pollfd, 1, 0) < 0)
602 scm_syserror ("fport_input_waiting");
603
604 return pollfd.revents & POLLIN ? 1 : 0;
605 }
606
607
608 \f
609
610 /* Revealed counts --- an oddity inherited from SCSH. */
611
612 #define SCM_REVEALED(x) (SCM_FSTREAM(x)->revealed)
613
614 static SCM revealed_ports = SCM_EOL;
615 static scm_i_pthread_mutex_t revealed_lock = SCM_I_PTHREAD_MUTEX_INITIALIZER;
616
617 /* Find a port in the table and return its revealed count.
618 Also used by the garbage collector.
619 */
620 int
621 scm_revealed_count (SCM port)
622 {
623 int ret;
624
625 scm_i_pthread_mutex_lock (&revealed_lock);
626 ret = SCM_REVEALED (port);
627 scm_i_pthread_mutex_unlock (&revealed_lock);
628
629 return ret;
630 }
631
632 SCM_DEFINE (scm_port_revealed, "port-revealed", 1, 0, 0,
633 (SCM port),
634 "Return the revealed count for @var{port}.")
635 #define FUNC_NAME s_scm_port_revealed
636 {
637 port = SCM_COERCE_OUTPORT (port);
638 SCM_VALIDATE_OPFPORT (1, port);
639 return scm_from_int (scm_revealed_count (port));
640 }
641 #undef FUNC_NAME
642
643 /* Set the revealed count for a port. */
644 SCM_DEFINE (scm_set_port_revealed_x, "set-port-revealed!", 2, 0, 0,
645 (SCM port, SCM rcount),
646 "Sets the revealed count for a port to a given value.\n"
647 "The return value is unspecified.")
648 #define FUNC_NAME s_scm_set_port_revealed_x
649 {
650 int r, prev;
651
652 port = SCM_COERCE_OUTPORT (port);
653 SCM_VALIDATE_OPFPORT (1, port);
654
655 r = scm_to_int (rcount);
656
657 scm_i_pthread_mutex_lock (&revealed_lock);
658
659 prev = SCM_REVEALED (port);
660 SCM_REVEALED (port) = r;
661
662 if (r && !prev)
663 revealed_ports = scm_cons (port, revealed_ports);
664 else if (prev && !r)
665 revealed_ports = scm_delq_x (port, revealed_ports);
666
667 scm_i_pthread_mutex_unlock (&revealed_lock);
668
669 return SCM_UNSPECIFIED;
670 }
671 #undef FUNC_NAME
672
673 /* Set the revealed count for a port. */
674 SCM_DEFINE (scm_adjust_port_revealed_x, "adjust-port-revealed!", 2, 0, 0,
675 (SCM port, SCM addend),
676 "Add @var{addend} to the revealed count of @var{port}.\n"
677 "The return value is unspecified.")
678 #define FUNC_NAME s_scm_adjust_port_revealed_x
679 {
680 int a;
681
682 port = SCM_COERCE_OUTPORT (port);
683 SCM_VALIDATE_OPFPORT (1, port);
684
685 a = scm_to_int (addend);
686 if (!a)
687 return SCM_UNSPECIFIED;
688
689 scm_i_pthread_mutex_lock (&revealed_lock);
690
691 SCM_REVEALED (port) += a;
692 if (SCM_REVEALED (port) == a)
693 revealed_ports = scm_cons (port, revealed_ports);
694 else if (!SCM_REVEALED (port))
695 revealed_ports = scm_delq_x (port, revealed_ports);
696
697 scm_i_pthread_mutex_unlock (&revealed_lock);
698
699 return SCM_UNSPECIFIED;
700 }
701 #undef FUNC_NAME
702
703
704 \f
705 static int
706 fport_print (SCM exp, SCM port, scm_print_state *pstate SCM_UNUSED)
707 {
708 scm_puts_unlocked ("#<", port);
709 scm_print_port_mode (exp, port);
710 if (SCM_OPFPORTP (exp))
711 {
712 int fdes;
713 SCM name = SCM_FILENAME (exp);
714 if (scm_is_string (name) || scm_is_symbol (name))
715 scm_display (name, port);
716 else
717 scm_puts_unlocked (SCM_PTOBNAME (SCM_PTOBNUM (exp)), port);
718 scm_putc_unlocked (' ', port);
719 fdes = (SCM_FSTREAM (exp))->fdes;
720
721 #if (defined HAVE_TTYNAME) && (defined HAVE_POSIX)
722 if (isatty (fdes))
723 scm_display (scm_ttyname (exp), port);
724 else
725 #endif /* HAVE_TTYNAME */
726 scm_intprint (fdes, 10, port);
727 }
728 else
729 {
730 scm_puts_unlocked (SCM_PTOBNAME (SCM_PTOBNUM (exp)), port);
731 scm_putc_unlocked (' ', port);
732 scm_uintprint ((scm_t_bits) SCM_PTAB_ENTRY (exp), 16, port);
733 }
734 scm_putc_unlocked ('>', port);
735 return 1;
736 }
737
738 static void fport_flush (SCM port);
739
740 /* fill a port's read-buffer with a single read. returns the first
741 char or EOF if end of file. */
742 static scm_t_wchar
743 fport_fill_input (SCM port)
744 {
745 long count;
746 scm_t_port *pt = SCM_PTAB_ENTRY (port);
747 scm_t_fport *fp = SCM_FSTREAM (port);
748
749 SCM_SYSCALL (count = read (fp->fdes, pt->read_buf, pt->read_buf_size));
750 if (count == -1)
751 scm_syserror ("fport_fill_input");
752 if (count == 0)
753 return (scm_t_wchar) EOF;
754 else
755 {
756 pt->read_pos = pt->read_buf;
757 pt->read_end = pt->read_buf + count;
758 return *pt->read_buf;
759 }
760 }
761
762 static scm_t_off
763 fport_seek (SCM port, scm_t_off offset, int whence)
764 {
765 scm_t_port *pt = SCM_PTAB_ENTRY (port);
766 scm_t_fport *fp = SCM_FSTREAM (port);
767 off_t_or_off64_t rv;
768 off_t_or_off64_t result;
769
770 if (pt->rw_active == SCM_PORT_WRITE)
771 {
772 if (offset != 0 || whence != SEEK_CUR)
773 {
774 fport_flush (port);
775 result = rv = lseek_or_lseek64 (fp->fdes, offset, whence);
776 }
777 else
778 {
779 /* read current position without disturbing the buffer. */
780 rv = lseek_or_lseek64 (fp->fdes, offset, whence);
781 result = rv + (pt->write_pos - pt->write_buf);
782 }
783 }
784 else if (pt->rw_active == SCM_PORT_READ)
785 {
786 if (offset != 0 || whence != SEEK_CUR)
787 {
788 /* could expand to avoid a second seek. */
789 scm_end_input_unlocked (port);
790 result = rv = lseek_or_lseek64 (fp->fdes, offset, whence);
791 }
792 else
793 {
794 /* read current position without disturbing the buffer
795 (particularly the unread-char buffer). */
796 rv = lseek_or_lseek64 (fp->fdes, offset, whence);
797 result = rv - (pt->read_end - pt->read_pos);
798
799 if (pt->read_buf == pt->putback_buf)
800 result -= pt->saved_read_end - pt->saved_read_pos;
801 }
802 }
803 else /* SCM_PORT_NEITHER */
804 {
805 result = rv = lseek_or_lseek64 (fp->fdes, offset, whence);
806 }
807
808 if (rv == -1)
809 scm_syserror ("fport_seek");
810
811 return result;
812 }
813
814 static void
815 fport_truncate (SCM port, scm_t_off length)
816 {
817 scm_t_fport *fp = SCM_FSTREAM (port);
818
819 if (ftruncate (fp->fdes, length) == -1)
820 scm_syserror ("ftruncate");
821 }
822
823 static void
824 fport_write (SCM port, const void *data, size_t size)
825 #define FUNC_NAME "fport_write"
826 {
827 /* this procedure tries to minimize the number of writes/flushes. */
828 scm_t_port *pt = SCM_PTAB_ENTRY (port);
829
830 if (pt->write_buf == &pt->shortbuf
831 || (pt->write_pos == pt->write_buf && size >= pt->write_buf_size))
832 {
833 /* Unbuffered port, or port with empty buffer and data won't fit in
834 buffer. */
835 if (full_write (SCM_FPORT_FDES (port), data, size) < size)
836 SCM_SYSERROR;
837
838 return;
839 }
840
841 {
842 scm_t_off space = pt->write_end - pt->write_pos;
843
844 if (size <= space)
845 {
846 /* data fits in buffer. */
847 memcpy (pt->write_pos, data, size);
848 pt->write_pos += size;
849 if (pt->write_pos == pt->write_end)
850 {
851 fport_flush (port);
852 /* we can skip the line-buffering check if nothing's buffered. */
853 return;
854 }
855 }
856 else
857 {
858 memcpy (pt->write_pos, data, space);
859 pt->write_pos = pt->write_end;
860 fport_flush (port);
861 {
862 const void *ptr = ((const char *) data) + space;
863 size_t remaining = size - space;
864
865 if (size >= pt->write_buf_size)
866 {
867 if (full_write (SCM_FPORT_FDES (port), ptr, remaining)
868 < remaining)
869 SCM_SYSERROR;
870 return;
871 }
872 else
873 {
874 memcpy (pt->write_pos, ptr, remaining);
875 pt->write_pos += remaining;
876 }
877 }
878 }
879
880 /* handle line buffering. */
881 if ((SCM_CELL_WORD_0 (port) & SCM_BUFLINE) && memchr (data, '\n', size))
882 fport_flush (port);
883 }
884 }
885 #undef FUNC_NAME
886
887 static void
888 fport_flush (SCM port)
889 {
890 size_t written;
891 scm_t_port *pt = SCM_PTAB_ENTRY (port);
892 scm_t_fport *fp = SCM_FSTREAM (port);
893 size_t count = pt->write_pos - pt->write_buf;
894
895 written = full_write (fp->fdes, pt->write_buf, count);
896 if (written < count)
897 scm_syserror ("scm_flush");
898
899 pt->write_pos = pt->write_buf;
900 pt->rw_active = SCM_PORT_NEITHER;
901 }
902
903 /* clear the read buffer and adjust the file position for unread bytes. */
904 static void
905 fport_end_input (SCM port, int offset)
906 {
907 scm_t_fport *fp = SCM_FSTREAM (port);
908 scm_t_port *pt = SCM_PTAB_ENTRY (port);
909
910 offset += pt->read_end - pt->read_pos;
911
912 if (offset > 0)
913 {
914 pt->read_pos = pt->read_end;
915 /* will throw error if unread-char used at beginning of file
916 then attempting to write. seems correct. */
917 if (lseek (fp->fdes, -offset, SEEK_CUR) == -1)
918 scm_syserror ("fport_end_input");
919 }
920 pt->rw_active = SCM_PORT_NEITHER;
921 }
922
923 static void
924 close_the_fd (void *data)
925 {
926 scm_t_fport *fp = data;
927
928 close (fp->fdes);
929 /* There's already one exception. That's probably enough! */
930 errno = 0;
931 }
932
933 static int
934 fport_close (SCM port)
935 {
936 scm_t_fport *fp = SCM_FSTREAM (port);
937 int rv;
938
939 scm_dynwind_begin (0);
940 scm_dynwind_unwind_handler (close_the_fd, fp, 0);
941 fport_flush (port);
942 scm_dynwind_end ();
943
944 scm_port_non_buffer (SCM_PTAB_ENTRY (port));
945
946 rv = close (fp->fdes);
947 if (rv)
948 /* It's not useful to retry after EINTR, as the file descriptor is
949 in an undefined state. See http://lwn.net/Articles/365294/.
950 Instead just throw an error if close fails, trusting that the fd
951 was cleaned up. */
952 scm_syserror ("fport_close");
953
954 return 0;
955 }
956
957 static size_t
958 fport_free (SCM port)
959 {
960 fport_close (port);
961 return 0;
962 }
963
964 static scm_t_bits
965 scm_make_fptob ()
966 {
967 scm_t_bits tc = scm_make_port_type ("file", fport_fill_input, fport_write);
968
969 scm_set_port_free (tc, fport_free);
970 scm_set_port_print (tc, fport_print);
971 scm_set_port_flush (tc, fport_flush);
972 scm_set_port_end_input (tc, fport_end_input);
973 scm_set_port_close (tc, fport_close);
974 scm_set_port_seek (tc, fport_seek);
975 scm_set_port_truncate (tc, fport_truncate);
976 scm_set_port_input_waiting (tc, fport_input_waiting);
977
978 return tc;
979 }
980
981 /* We can't initialize the keywords from 'scm_init_fports', because
982 keywords haven't yet been initialized at that point. */
983 void
984 scm_init_fports_keywords ()
985 {
986 k_guess_encoding = scm_from_latin1_keyword ("guess-encoding");
987 k_encoding = scm_from_latin1_keyword ("encoding");
988 }
989
990 void
991 scm_init_fports ()
992 {
993 scm_tc16_fport = scm_make_fptob ();
994
995 scm_c_define ("_IOFBF", scm_from_int (_IOFBF));
996 scm_c_define ("_IOLBF", scm_from_int (_IOLBF));
997 scm_c_define ("_IONBF", scm_from_int (_IONBF));
998
999 sys_file_port_name_canonicalization = scm_make_fluid ();
1000 scm_c_define ("%file-port-name-canonicalization",
1001 sys_file_port_name_canonicalization);
1002
1003 #include "libguile/fports.x"
1004 }
1005
1006 /*
1007 Local Variables:
1008 c-file-style: "gnu"
1009 End:
1010 */