fix segfaults when closing the current input port
[bpt/guile.git] / libguile / ports.c
1 /* Copyright (C) 1995,1996,1997,1998,1999,2000,2001, 2003, 2004, 2006, 2007, 2008, 2009, 2010 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 /* Headers. */
22
23 #define _LARGEFILE64_SOURCE /* ask for stat64 etc */
24
25 #ifdef HAVE_CONFIG_H
26 # include <config.h>
27 #endif
28
29 #include <stdio.h>
30 #include <errno.h>
31 #include <fcntl.h> /* for chsize on mingw */
32 #include <assert.h>
33 #include <uniconv.h>
34 #include <unistr.h>
35 #include <striconveh.h>
36
37 #include <assert.h>
38
39 #include "libguile/_scm.h"
40 #include "libguile/async.h"
41 #include "libguile/eval.h"
42 #include "libguile/fports.h" /* direct access for seek and truncate */
43 #include "libguile/goops.h"
44 #include "libguile/smob.h"
45 #include "libguile/chars.h"
46 #include "libguile/dynwind.h"
47
48 #include "libguile/keywords.h"
49 #include "libguile/hashtab.h"
50 #include "libguile/root.h"
51 #include "libguile/strings.h"
52 #include "libguile/mallocs.h"
53 #include "libguile/validate.h"
54 #include "libguile/ports.h"
55 #include "libguile/vectors.h"
56 #include "libguile/weaks.h"
57 #include "libguile/fluids.h"
58 #include "libguile/eq.h"
59
60 #ifdef HAVE_STRING_H
61 #include <string.h>
62 #endif
63
64 #ifdef HAVE_IO_H
65 #include <io.h>
66 #endif
67
68 #ifdef HAVE_UNISTD_H
69 #include <unistd.h>
70 #endif
71
72 #ifdef HAVE_SYS_IOCTL_H
73 #include <sys/ioctl.h>
74 #endif
75
76 /* Mingw (version 3.4.5, circa 2006) has ftruncate as an alias for chsize
77 already, but have this code here in case that wasn't so in past versions,
78 or perhaps to help other minimal DOS environments.
79
80 gnulib ftruncate.c has code using fcntl F_CHSIZE and F_FREESP, which
81 might be possibilities if we've got other systems without ftruncate. */
82
83 #if defined HAVE_CHSIZE && ! defined HAVE_FTRUNCATE
84 #define ftruncate(fd, size) chsize (fd, size)
85 #undef HAVE_FTRUNCATE
86 #define HAVE_FTRUNCATE 1
87 #endif
88
89 \f
90 /* The port kind table --- a dynamically resized array of port types. */
91
92
93 /* scm_ptobs scm_numptob
94 * implement a dynamically resized array of ptob records.
95 * Indexes into this table are used when generating type
96 * tags for smobjects (if you know a tag you can get an index and conversely).
97 */
98 scm_t_ptob_descriptor *scm_ptobs = NULL;
99 long scm_numptob = 0;
100
101 /* GC marker for a port with stream of SCM type. */
102 SCM
103 scm_markstream (SCM ptr)
104 {
105 int openp;
106 openp = SCM_CELL_WORD_0 (ptr) & SCM_OPN;
107 if (openp)
108 return SCM_PACK (SCM_STREAM (ptr));
109 else
110 return SCM_BOOL_F;
111 }
112
113 /*
114 * We choose to use an interface similar to the smob interface with
115 * fill_input and write as standard fields, passed to the port
116 * type constructor, and optional fields set by setters.
117 */
118
119 static void
120 flush_port_default (SCM port SCM_UNUSED)
121 {
122 }
123
124 static void
125 end_input_default (SCM port SCM_UNUSED, int offset SCM_UNUSED)
126 {
127 }
128
129 scm_t_bits
130 scm_make_port_type (char *name,
131 int (*fill_input) (SCM port),
132 void (*write) (SCM port, const void *data, size_t size))
133 {
134 char *tmp;
135 if (SCM_I_MAX_PORT_TYPE_COUNT - 1 <= scm_numptob)
136 goto ptoberr;
137 SCM_CRITICAL_SECTION_START;
138 tmp = (char *) scm_gc_realloc ((char *) scm_ptobs,
139 scm_numptob * sizeof (scm_t_ptob_descriptor),
140 (1 + scm_numptob)
141 * sizeof (scm_t_ptob_descriptor),
142 "port-type");
143 if (tmp)
144 {
145 scm_ptobs = (scm_t_ptob_descriptor *) tmp;
146
147 scm_ptobs[scm_numptob].name = name;
148 scm_ptobs[scm_numptob].mark = 0;
149 scm_ptobs[scm_numptob].free = NULL;
150 scm_ptobs[scm_numptob].print = scm_port_print;
151 scm_ptobs[scm_numptob].equalp = 0;
152 scm_ptobs[scm_numptob].close = 0;
153
154 scm_ptobs[scm_numptob].write = write;
155 scm_ptobs[scm_numptob].flush = flush_port_default;
156
157 scm_ptobs[scm_numptob].end_input = end_input_default;
158 scm_ptobs[scm_numptob].fill_input = fill_input;
159 scm_ptobs[scm_numptob].input_waiting = 0;
160
161 scm_ptobs[scm_numptob].seek = 0;
162 scm_ptobs[scm_numptob].truncate = 0;
163
164 scm_numptob++;
165 }
166 SCM_CRITICAL_SECTION_END;
167 if (!tmp)
168 {
169 ptoberr:
170 scm_memory_error ("scm_make_port_type");
171 }
172 /* Make a class object if Goops is present */
173 if (SCM_UNPACK (scm_port_class[0]) != 0)
174 scm_make_port_classes (scm_numptob - 1, SCM_PTOBNAME (scm_numptob - 1));
175 return scm_tc7_port + (scm_numptob - 1) * 256;
176 }
177
178 void
179 scm_set_port_mark (scm_t_bits tc, SCM (*mark) (SCM))
180 {
181 scm_ptobs[SCM_TC2PTOBNUM (tc)].mark = mark;
182 }
183
184 void
185 scm_set_port_free (scm_t_bits tc, size_t (*free) (SCM))
186 {
187 scm_ptobs[SCM_TC2PTOBNUM (tc)].free = free;
188 }
189
190 void
191 scm_set_port_print (scm_t_bits tc, int (*print) (SCM exp, SCM port,
192 scm_print_state *pstate))
193 {
194 scm_ptobs[SCM_TC2PTOBNUM (tc)].print = print;
195 }
196
197 void
198 scm_set_port_equalp (scm_t_bits tc, SCM (*equalp) (SCM, SCM))
199 {
200 scm_ptobs[SCM_TC2PTOBNUM (tc)].equalp = equalp;
201 }
202
203 void
204 scm_set_port_flush (scm_t_bits tc, void (*flush) (SCM port))
205 {
206 scm_ptobs[SCM_TC2PTOBNUM (tc)].flush = flush;
207 }
208
209 void
210 scm_set_port_end_input (scm_t_bits tc, void (*end_input) (SCM port, int offset))
211 {
212 scm_ptobs[SCM_TC2PTOBNUM (tc)].end_input = end_input;
213 }
214
215 void
216 scm_set_port_close (scm_t_bits tc, int (*close) (SCM))
217 {
218 scm_ptobs[SCM_TC2PTOBNUM (tc)].close = close;
219 }
220
221 void
222 scm_set_port_seek (scm_t_bits tc,
223 scm_t_off (*seek) (SCM, scm_t_off, int))
224 {
225 scm_ptobs[SCM_TC2PTOBNUM (tc)].seek = seek;
226 }
227
228 void
229 scm_set_port_truncate (scm_t_bits tc, void (*truncate) (SCM, scm_t_off))
230 {
231 scm_ptobs[SCM_TC2PTOBNUM (tc)].truncate = truncate;
232 }
233
234 void
235 scm_set_port_input_waiting (scm_t_bits tc, int (*input_waiting) (SCM))
236 {
237 scm_ptobs[SCM_TC2PTOBNUM (tc)].input_waiting = input_waiting;
238 }
239
240 \f
241
242 SCM_DEFINE (scm_char_ready_p, "char-ready?", 0, 1, 0,
243 (SCM port),
244 "Return @code{#t} if a character is ready on input @var{port}\n"
245 "and return @code{#f} otherwise. If @code{char-ready?} returns\n"
246 "@code{#t} then the next @code{read-char} operation on\n"
247 "@var{port} is guaranteed not to hang. If @var{port} is a file\n"
248 "port at end of file then @code{char-ready?} returns @code{#t}.\n"
249 "\n"
250 "@code{char-ready?} exists to make it possible for a\n"
251 "program to accept characters from interactive ports without\n"
252 "getting stuck waiting for input. Any input editors associated\n"
253 "with such ports must make sure that characters whose existence\n"
254 "has been asserted by @code{char-ready?} cannot be rubbed out.\n"
255 "If @code{char-ready?} were to return @code{#f} at end of file,\n"
256 "a port at end of file would be indistinguishable from an\n"
257 "interactive port that has no ready characters.")
258 #define FUNC_NAME s_scm_char_ready_p
259 {
260 scm_t_port *pt;
261
262 if (SCM_UNBNDP (port))
263 port = scm_current_input_port ();
264 /* It's possible to close the current input port, so validate even in
265 this case. */
266 SCM_VALIDATE_OPINPORT (1, port);
267
268 pt = SCM_PTAB_ENTRY (port);
269
270 /* if the current read buffer is filled, or the
271 last pushed-back char has been read and the saved buffer is
272 filled, result is true. */
273 if (pt->read_pos < pt->read_end
274 || (pt->read_buf == pt->putback_buf
275 && pt->saved_read_pos < pt->saved_read_end))
276 return SCM_BOOL_T;
277 else
278 {
279 scm_t_ptob_descriptor *ptob = &scm_ptobs[SCM_PTOBNUM (port)];
280
281 if (ptob->input_waiting)
282 return scm_from_bool(ptob->input_waiting (port));
283 else
284 return SCM_BOOL_T;
285 }
286 }
287 #undef FUNC_NAME
288
289 /* move up to read_len chars from port's putback and/or read buffers
290 into memory starting at dest. returns the number of chars moved. */
291 size_t scm_take_from_input_buffers (SCM port, char *dest, size_t read_len)
292 {
293 scm_t_port *pt = SCM_PTAB_ENTRY (port);
294 size_t chars_read = 0;
295 size_t from_buf = min (pt->read_end - pt->read_pos, read_len);
296
297 if (from_buf > 0)
298 {
299 memcpy (dest, pt->read_pos, from_buf);
300 pt->read_pos += from_buf;
301 chars_read += from_buf;
302 read_len -= from_buf;
303 dest += from_buf;
304 }
305
306 /* if putback was active, try the real input buffer too. */
307 if (pt->read_buf == pt->putback_buf)
308 {
309 from_buf = min (pt->saved_read_end - pt->saved_read_pos, read_len);
310 if (from_buf > 0)
311 {
312 memcpy (dest, pt->saved_read_pos, from_buf);
313 pt->saved_read_pos += from_buf;
314 chars_read += from_buf;
315 }
316 }
317 return chars_read;
318 }
319
320 /* Clear a port's read buffers, returning the contents. */
321 SCM_DEFINE (scm_drain_input, "drain-input", 1, 0, 0,
322 (SCM port),
323 "This procedure clears a port's input buffers, similar\n"
324 "to the way that force-output clears the output buffer. The\n"
325 "contents of the buffers are returned as a single string, e.g.,\n"
326 "\n"
327 "@lisp\n"
328 "(define p (open-input-file ...))\n"
329 "(drain-input p) => empty string, nothing buffered yet.\n"
330 "(unread-char (read-char p) p)\n"
331 "(drain-input p) => initial chars from p, up to the buffer size.\n"
332 "@end lisp\n\n"
333 "Draining the buffers may be useful for cleanly finishing\n"
334 "buffered I/O so that the file descriptor can be used directly\n"
335 "for further input.")
336 #define FUNC_NAME s_scm_drain_input
337 {
338 SCM result;
339 char *data;
340 scm_t_port *pt;
341 long count;
342
343 SCM_VALIDATE_OPINPORT (1, port);
344 pt = SCM_PTAB_ENTRY (port);
345
346 count = pt->read_end - pt->read_pos;
347 if (pt->read_buf == pt->putback_buf)
348 count += pt->saved_read_end - pt->saved_read_pos;
349
350 result = scm_i_make_string (count, &data);
351 scm_take_from_input_buffers (port, data, count);
352 return result;
353 }
354 #undef FUNC_NAME
355
356 \f
357 /* Standard ports --- current input, output, error, and more(!). */
358
359 static SCM cur_inport_fluid = 0;
360 static SCM cur_outport_fluid = 0;
361 static SCM cur_errport_fluid = 0;
362 static SCM cur_loadport_fluid = 0;
363
364 SCM_DEFINE (scm_current_input_port, "current-input-port", 0, 0, 0,
365 (),
366 "Return the current input port. This is the default port used\n"
367 "by many input procedures. Initially, @code{current-input-port}\n"
368 "returns the @dfn{standard input} in Unix and C terminology.")
369 #define FUNC_NAME s_scm_current_input_port
370 {
371 if (cur_inport_fluid)
372 return scm_fluid_ref (cur_inport_fluid);
373 else
374 return SCM_BOOL_F;
375 }
376 #undef FUNC_NAME
377
378 SCM_DEFINE (scm_current_output_port, "current-output-port", 0, 0, 0,
379 (),
380 "Return the current output port. This is the default port used\n"
381 "by many output procedures. Initially,\n"
382 "@code{current-output-port} returns the @dfn{standard output} in\n"
383 "Unix and C terminology.")
384 #define FUNC_NAME s_scm_current_output_port
385 {
386 if (cur_outport_fluid)
387 return scm_fluid_ref (cur_outport_fluid);
388 else
389 return SCM_BOOL_F;
390 }
391 #undef FUNC_NAME
392
393 SCM_DEFINE (scm_current_error_port, "current-error-port", 0, 0, 0,
394 (),
395 "Return the port to which errors and warnings should be sent (the\n"
396 "@dfn{standard error} in Unix and C terminology).")
397 #define FUNC_NAME s_scm_current_error_port
398 {
399 if (cur_errport_fluid)
400 return scm_fluid_ref (cur_errport_fluid);
401 else
402 return SCM_BOOL_F;
403 }
404 #undef FUNC_NAME
405
406 SCM_DEFINE (scm_current_load_port, "current-load-port", 0, 0, 0,
407 (),
408 "Return the current-load-port.\n"
409 "The load port is used internally by @code{primitive-load}.")
410 #define FUNC_NAME s_scm_current_load_port
411 {
412 return scm_fluid_ref (cur_loadport_fluid);
413 }
414 #undef FUNC_NAME
415
416 SCM_DEFINE (scm_set_current_input_port, "set-current-input-port", 1, 0, 0,
417 (SCM port),
418 "@deffnx {Scheme Procedure} set-current-output-port port\n"
419 "@deffnx {Scheme Procedure} set-current-error-port port\n"
420 "Change the ports returned by @code{current-input-port},\n"
421 "@code{current-output-port} and @code{current-error-port}, respectively,\n"
422 "so that they use the supplied @var{port} for input or output.")
423 #define FUNC_NAME s_scm_set_current_input_port
424 {
425 SCM oinp = scm_fluid_ref (cur_inport_fluid);
426 SCM_VALIDATE_OPINPORT (1, port);
427 scm_fluid_set_x (cur_inport_fluid, port);
428 return oinp;
429 }
430 #undef FUNC_NAME
431
432
433 SCM_DEFINE (scm_set_current_output_port, "set-current-output-port", 1, 0, 0,
434 (SCM port),
435 "Set the current default output port to @var{port}.")
436 #define FUNC_NAME s_scm_set_current_output_port
437 {
438 SCM ooutp = scm_fluid_ref (cur_outport_fluid);
439 port = SCM_COERCE_OUTPORT (port);
440 SCM_VALIDATE_OPOUTPORT (1, port);
441 scm_fluid_set_x (cur_outport_fluid, port);
442 return ooutp;
443 }
444 #undef FUNC_NAME
445
446
447 SCM_DEFINE (scm_set_current_error_port, "set-current-error-port", 1, 0, 0,
448 (SCM port),
449 "Set the current default error port to @var{port}.")
450 #define FUNC_NAME s_scm_set_current_error_port
451 {
452 SCM oerrp = scm_fluid_ref (cur_errport_fluid);
453 port = SCM_COERCE_OUTPORT (port);
454 SCM_VALIDATE_OPOUTPORT (1, port);
455 scm_fluid_set_x (cur_errport_fluid, port);
456 return oerrp;
457 }
458 #undef FUNC_NAME
459
460 void
461 scm_dynwind_current_input_port (SCM port)
462 #define FUNC_NAME NULL
463 {
464 SCM_VALIDATE_OPINPORT (1, port);
465 scm_dynwind_fluid (cur_inport_fluid, port);
466 }
467 #undef FUNC_NAME
468
469 void
470 scm_dynwind_current_output_port (SCM port)
471 #define FUNC_NAME NULL
472 {
473 port = SCM_COERCE_OUTPORT (port);
474 SCM_VALIDATE_OPOUTPORT (1, port);
475 scm_dynwind_fluid (cur_outport_fluid, port);
476 }
477 #undef FUNC_NAME
478
479 void
480 scm_dynwind_current_error_port (SCM port)
481 #define FUNC_NAME NULL
482 {
483 port = SCM_COERCE_OUTPORT (port);
484 SCM_VALIDATE_OPOUTPORT (1, port);
485 scm_dynwind_fluid (cur_errport_fluid, port);
486 }
487 #undef FUNC_NAME
488
489 void
490 scm_i_dynwind_current_load_port (SCM port)
491 {
492 scm_dynwind_fluid (cur_loadport_fluid, port);
493 }
494
495 \f
496 /* The port table --- an array of pointers to ports. */
497
498 /*
499 We need a global registry of ports to flush them all at exit, and to
500 get all the ports matching a file descriptor.
501 */
502 SCM scm_i_port_weak_hash;
503
504 scm_i_pthread_mutex_t scm_i_port_table_mutex = SCM_I_PTHREAD_MUTEX_INITIALIZER;
505
506 \f
507 /* Port finalization. */
508
509
510 static void finalize_port (GC_PTR, GC_PTR);
511
512 /* Register a finalizer for PORT, if needed by its port type. */
513 static SCM_C_INLINE_KEYWORD void
514 register_finalizer_for_port (SCM port)
515 {
516 long port_type;
517
518 port_type = SCM_TC2PTOBNUM (SCM_CELL_TYPE (port));
519 if (scm_ptobs[port_type].free)
520 {
521 GC_finalization_proc prev_finalizer;
522 GC_PTR prev_finalization_data;
523
524 GC_REGISTER_FINALIZER_NO_ORDER (SCM2PTR (port), finalize_port, 0,
525 &prev_finalizer,
526 &prev_finalization_data);
527 }
528 }
529
530 /* Finalize the object (a port) pointed to by PTR. */
531 static void
532 finalize_port (GC_PTR ptr, GC_PTR data)
533 {
534 long port_type;
535 SCM port = PTR2SCM (ptr);
536
537 if (!SCM_PORTP (port))
538 abort ();
539
540 if (SCM_OPENP (port))
541 {
542 if (SCM_REVEALED (port) > 0)
543 /* Keep "revealed" ports alive and re-register a finalizer. */
544 register_finalizer_for_port (port);
545 else
546 {
547 port_type = SCM_TC2PTOBNUM (SCM_CELL_TYPE (port));
548 if (port_type >= scm_numptob)
549 abort ();
550
551 if (scm_ptobs[port_type].free)
552 /* Yes, I really do mean `.free' rather than `.close'. `.close'
553 is for explicit `close-port' by user. */
554 scm_ptobs[port_type].free (port);
555
556 SCM_SETSTREAM (port, 0);
557 SCM_CLR_PORT_OPEN_FLAG (port);
558
559 scm_gc_ports_collected++;
560 }
561 }
562 }
563
564
565
566 \f
567
568 /* This function is not and should not be thread safe. */
569 SCM
570 scm_new_port_table_entry (scm_t_bits tag)
571 #define FUNC_NAME "scm_new_port_table_entry"
572 {
573 /*
574 We initialize the cell to empty, this is in case scm_gc_calloc
575 triggers GC ; we don't want the GC to scan a half-finished Z.
576 */
577
578 SCM z = scm_cons (SCM_EOL, SCM_EOL);
579 scm_t_port *entry = (scm_t_port *) scm_gc_calloc (sizeof (scm_t_port), "port");
580 const char *enc;
581
582 entry->file_name = SCM_BOOL_F;
583 entry->rw_active = SCM_PORT_NEITHER;
584 entry->port = z;
585 /* Initialize this port with the thread's current default
586 encoding. */
587 if ((enc = scm_i_get_port_encoding (SCM_BOOL_F)) == NULL)
588 entry->encoding = NULL;
589 else
590 entry->encoding = scm_gc_strdup (enc, "port");
591 entry->ilseq_handler = scm_i_get_conversion_strategy (SCM_BOOL_F);
592
593 SCM_SET_CELL_TYPE (z, tag);
594 SCM_SETPTAB_ENTRY (z, entry);
595
596 scm_hashq_set_x (scm_i_port_weak_hash, z, SCM_BOOL_F);
597
598 /* For each new port, register a finalizer so that it port type's free
599 function can be invoked eventually. */
600 register_finalizer_for_port (z);
601
602 return z;
603 }
604 #undef FUNC_NAME
605
606 #if SCM_ENABLE_DEPRECATED==1
607 SCM_API scm_t_port *
608 scm_add_to_port_table (SCM port)
609 {
610 SCM z = scm_new_port_table_entry (scm_tc7_port);
611 scm_t_port * pt = SCM_PTAB_ENTRY(z);
612
613 pt->port = port;
614 SCM_SETCAR (z, SCM_EOL);
615 SCM_SETCDR (z, SCM_EOL);
616 SCM_SETPTAB_ENTRY (port, pt);
617 return pt;
618 }
619 #endif
620
621
622 /* Remove a port from the table and destroy it. */
623
624 /* This function is not and should not be thread safe. */
625 void
626 scm_i_remove_port (SCM port)
627 #define FUNC_NAME "scm_remove_port"
628 {
629 scm_t_port *p = SCM_PTAB_ENTRY (port);
630
631 scm_port_non_buffer (p);
632
633 p->putback_buf = NULL;
634 p->putback_buf_size = 0;
635
636 SCM_SETPTAB_ENTRY (port, 0);
637 scm_hashq_remove_x (scm_i_port_weak_hash, port);
638 }
639 #undef FUNC_NAME
640
641
642 /* Functions for debugging. */
643 #ifdef GUILE_DEBUG
644 SCM_DEFINE (scm_pt_size, "pt-size", 0, 0, 0,
645 (),
646 "Return the number of ports in the port table. @code{pt-size}\n"
647 "is only included in @code{--enable-guile-debug} builds.")
648 #define FUNC_NAME s_scm_pt_size
649 {
650 return scm_from_int (SCM_HASHTABLE_N_ITEMS (scm_i_port_weak_hash));
651 }
652 #undef FUNC_NAME
653 #endif
654
655 void
656 scm_port_non_buffer (scm_t_port *pt)
657 {
658 pt->read_pos = pt->read_buf = pt->read_end = &pt->shortbuf;
659 pt->write_buf = pt->write_pos = &pt->shortbuf;
660 pt->read_buf_size = pt->write_buf_size = 1;
661 pt->write_end = pt->write_buf + pt->write_buf_size;
662 }
663
664 \f
665 /* Revealed counts --- an oddity inherited from SCSH. */
666
667 /* Find a port in the table and return its revealed count.
668 Also used by the garbage collector.
669 */
670
671 int
672 scm_revealed_count (SCM port)
673 {
674 return SCM_REVEALED(port);
675 }
676
677
678
679 /* Return the revealed count for a port. */
680
681 SCM_DEFINE (scm_port_revealed, "port-revealed", 1, 0, 0,
682 (SCM port),
683 "Return the revealed count for @var{port}.")
684 #define FUNC_NAME s_scm_port_revealed
685 {
686 port = SCM_COERCE_OUTPORT (port);
687 SCM_VALIDATE_OPENPORT (1, port);
688 return scm_from_int (scm_revealed_count (port));
689 }
690 #undef FUNC_NAME
691
692 /* Set the revealed count for a port. */
693 SCM_DEFINE (scm_set_port_revealed_x, "set-port-revealed!", 2, 0, 0,
694 (SCM port, SCM rcount),
695 "Sets the revealed count for a port to a given value.\n"
696 "The return value is unspecified.")
697 #define FUNC_NAME s_scm_set_port_revealed_x
698 {
699 port = SCM_COERCE_OUTPORT (port);
700 SCM_VALIDATE_OPENPORT (1, port);
701 SCM_REVEALED (port) = scm_to_int (rcount);
702 return SCM_UNSPECIFIED;
703 }
704 #undef FUNC_NAME
705
706
707 \f
708 /* Retrieving a port's mode. */
709
710 /* Return the flags that characterize a port based on the mode
711 * string used to open a file for that port.
712 *
713 * See PORT FLAGS in scm.h
714 */
715
716 static long
717 scm_i_mode_bits_n (SCM modes)
718 {
719 return (SCM_OPN
720 | (scm_i_string_contains_char (modes, 'r')
721 || scm_i_string_contains_char (modes, '+') ? SCM_RDNG : 0)
722 | (scm_i_string_contains_char (modes, 'w')
723 || scm_i_string_contains_char (modes, 'a')
724 || scm_i_string_contains_char (modes, '+') ? SCM_WRTNG : 0)
725 | (scm_i_string_contains_char (modes, '0') ? SCM_BUF0 : 0)
726 | (scm_i_string_contains_char (modes, 'l') ? SCM_BUFLINE : 0));
727 }
728
729 long
730 scm_mode_bits (char *modes)
731 {
732 return scm_i_mode_bits (scm_from_locale_string (modes));
733 }
734
735 long
736 scm_i_mode_bits (SCM modes)
737 {
738 long bits;
739
740 if (!scm_is_string (modes))
741 scm_wrong_type_arg_msg (NULL, 0, modes, "string");
742
743 bits = scm_i_mode_bits_n (modes);
744 scm_remember_upto_here_1 (modes);
745 return bits;
746 }
747
748 /* Return the mode flags from an open port.
749 * Some modes such as "append" are only used when opening
750 * a file and are not returned here. */
751
752 SCM_DEFINE (scm_port_mode, "port-mode", 1, 0, 0,
753 (SCM port),
754 "Return the port modes associated with the open port @var{port}.\n"
755 "These will not necessarily be identical to the modes used when\n"
756 "the port was opened, since modes such as \"append\" which are\n"
757 "used only during port creation are not retained.")
758 #define FUNC_NAME s_scm_port_mode
759 {
760 char modes[4];
761 modes[0] = '\0';
762
763 port = SCM_COERCE_OUTPORT (port);
764 SCM_VALIDATE_OPPORT (1, port);
765 if (SCM_CELL_WORD_0 (port) & SCM_RDNG) {
766 if (SCM_CELL_WORD_0 (port) & SCM_WRTNG)
767 strcpy (modes, "r+");
768 else
769 strcpy (modes, "r");
770 }
771 else if (SCM_CELL_WORD_0 (port) & SCM_WRTNG)
772 strcpy (modes, "w");
773 if (SCM_CELL_WORD_0 (port) & SCM_BUF0)
774 strcat (modes, "0");
775 return scm_from_locale_string (modes);
776 }
777 #undef FUNC_NAME
778
779
780 \f
781 /* Closing ports. */
782
783 /* scm_close_port
784 * Call the close operation on a port object.
785 * see also scm_close.
786 */
787 SCM_DEFINE (scm_close_port, "close-port", 1, 0, 0,
788 (SCM port),
789 "Close the specified port object. Return @code{#t} if it\n"
790 "successfully closes a port or @code{#f} if it was already\n"
791 "closed. An exception may be raised if an error occurs, for\n"
792 "example when flushing buffered output. See also @ref{Ports and\n"
793 "File Descriptors, close}, for a procedure which can close file\n"
794 "descriptors.")
795 #define FUNC_NAME s_scm_close_port
796 {
797 size_t i;
798 int rv;
799
800 port = SCM_COERCE_OUTPORT (port);
801
802 SCM_VALIDATE_PORT (1, port);
803 if (SCM_CLOSEDP (port))
804 return SCM_BOOL_F;
805 i = SCM_PTOBNUM (port);
806 if (scm_ptobs[i].close)
807 rv = (scm_ptobs[i].close) (port);
808 else
809 rv = 0;
810 scm_i_scm_pthread_mutex_lock (&scm_i_port_table_mutex);
811 scm_i_remove_port (port);
812 scm_i_pthread_mutex_unlock (&scm_i_port_table_mutex);
813 SCM_CLR_PORT_OPEN_FLAG (port);
814 return scm_from_bool (rv >= 0);
815 }
816 #undef FUNC_NAME
817
818 SCM_DEFINE (scm_close_input_port, "close-input-port", 1, 0, 0,
819 (SCM port),
820 "Close the specified input port object. The routine has no effect if\n"
821 "the file has already been closed. An exception may be raised if an\n"
822 "error occurs. The value returned is unspecified.\n\n"
823 "See also @ref{Ports and File Descriptors, close}, for a procedure\n"
824 "which can close file descriptors.")
825 #define FUNC_NAME s_scm_close_input_port
826 {
827 SCM_VALIDATE_INPUT_PORT (1, port);
828 scm_close_port (port);
829 return SCM_UNSPECIFIED;
830 }
831 #undef FUNC_NAME
832
833 SCM_DEFINE (scm_close_output_port, "close-output-port", 1, 0, 0,
834 (SCM port),
835 "Close the specified output port object. The routine has no effect if\n"
836 "the file has already been closed. An exception may be raised if an\n"
837 "error occurs. The value returned is unspecified.\n\n"
838 "See also @ref{Ports and File Descriptors, close}, for a procedure\n"
839 "which can close file descriptors.")
840 #define FUNC_NAME s_scm_close_output_port
841 {
842 port = SCM_COERCE_OUTPORT (port);
843 SCM_VALIDATE_OUTPUT_PORT (1, port);
844 scm_close_port (port);
845 return SCM_UNSPECIFIED;
846 }
847 #undef FUNC_NAME
848
849 static SCM
850 scm_i_collect_keys_in_vector (void *closure, SCM key, SCM value, SCM result)
851 {
852 int *i = (int*) closure;
853 scm_c_vector_set_x (result, *i, key);
854 (*i)++;
855
856 return result;
857 }
858
859 void
860 scm_c_port_for_each (void (*proc)(void *data, SCM p), void *data)
861 {
862 int i = 0;
863 size_t n;
864 SCM ports;
865
866 /* Even without pre-emptive multithreading, running arbitrary code
867 while scanning the port table is unsafe because the port table
868 can change arbitrarily (from a GC, for example). So we first
869 collect the ports into a vector. -mvo */
870
871 scm_i_scm_pthread_mutex_lock (&scm_i_port_table_mutex);
872 n = SCM_HASHTABLE_N_ITEMS (scm_i_port_weak_hash);
873 scm_i_pthread_mutex_unlock (&scm_i_port_table_mutex);
874 ports = scm_c_make_vector (n, SCM_BOOL_F);
875
876 scm_i_pthread_mutex_lock (&scm_i_port_table_mutex);
877 ports = scm_internal_hash_fold (scm_i_collect_keys_in_vector, &i,
878 ports, scm_i_port_weak_hash);
879 scm_i_pthread_mutex_unlock (&scm_i_port_table_mutex);
880
881 for (i = 0; i < n; i++) {
882 SCM p = SCM_SIMPLE_VECTOR_REF (ports, i);
883 if (SCM_PORTP (p))
884 proc (data, p);
885 }
886
887 scm_remember_upto_here_1 (ports);
888 }
889
890 SCM_DEFINE (scm_port_for_each, "port-for-each", 1, 0, 0,
891 (SCM proc),
892 "Apply @var{proc} to each port in the Guile port table\n"
893 "in turn. The return value is unspecified. More specifically,\n"
894 "@var{proc} is applied exactly once to every port that exists\n"
895 "in the system at the time @var{port-for-each} is invoked.\n"
896 "Changes to the port table while @var{port-for-each} is running\n"
897 "have no effect as far as @var{port-for-each} is concerned.")
898 #define FUNC_NAME s_scm_port_for_each
899 {
900 SCM_VALIDATE_PROC (1, proc);
901
902 scm_c_port_for_each ((void (*)(void*,SCM))scm_call_1, proc);
903 return SCM_UNSPECIFIED;
904 }
905 #undef FUNC_NAME
906
907
908 \f
909 /* Utter miscellany. Gosh, we should clean this up some time. */
910
911 SCM_DEFINE (scm_input_port_p, "input-port?", 1, 0, 0,
912 (SCM x),
913 "Return @code{#t} if @var{x} is an input port, otherwise return\n"
914 "@code{#f}. Any object satisfying this predicate also satisfies\n"
915 "@code{port?}.")
916 #define FUNC_NAME s_scm_input_port_p
917 {
918 return scm_from_bool (SCM_INPUT_PORT_P (x));
919 }
920 #undef FUNC_NAME
921
922 SCM_DEFINE (scm_output_port_p, "output-port?", 1, 0, 0,
923 (SCM x),
924 "Return @code{#t} if @var{x} is an output port, otherwise return\n"
925 "@code{#f}. Any object satisfying this predicate also satisfies\n"
926 "@code{port?}.")
927 #define FUNC_NAME s_scm_output_port_p
928 {
929 x = SCM_COERCE_OUTPORT (x);
930 return scm_from_bool (SCM_OUTPUT_PORT_P (x));
931 }
932 #undef FUNC_NAME
933
934 SCM_DEFINE (scm_port_p, "port?", 1, 0, 0,
935 (SCM x),
936 "Return a boolean indicating whether @var{x} is a port.\n"
937 "Equivalent to @code{(or (input-port? @var{x}) (output-port?\n"
938 "@var{x}))}.")
939 #define FUNC_NAME s_scm_port_p
940 {
941 return scm_from_bool (SCM_PORTP (x));
942 }
943 #undef FUNC_NAME
944
945 SCM_DEFINE (scm_port_closed_p, "port-closed?", 1, 0, 0,
946 (SCM port),
947 "Return @code{#t} if @var{port} is closed or @code{#f} if it is\n"
948 "open.")
949 #define FUNC_NAME s_scm_port_closed_p
950 {
951 SCM_VALIDATE_PORT (1, port);
952 return scm_from_bool (!SCM_OPPORTP (port));
953 }
954 #undef FUNC_NAME
955
956 SCM_DEFINE (scm_eof_object_p, "eof-object?", 1, 0, 0,
957 (SCM x),
958 "Return @code{#t} if @var{x} is an end-of-file object; otherwise\n"
959 "return @code{#f}.")
960 #define FUNC_NAME s_scm_eof_object_p
961 {
962 return scm_from_bool(SCM_EOF_OBJECT_P (x));
963 }
964 #undef FUNC_NAME
965
966 SCM_DEFINE (scm_force_output, "force-output", 0, 1, 0,
967 (SCM port),
968 "Flush the specified output port, or the current output port if @var{port}\n"
969 "is omitted. The current output buffer contents are passed to the\n"
970 "underlying port implementation (e.g., in the case of fports, the\n"
971 "data will be written to the file and the output buffer will be cleared.)\n"
972 "It has no effect on an unbuffered port.\n\n"
973 "The return value is unspecified.")
974 #define FUNC_NAME s_scm_force_output
975 {
976 if (SCM_UNBNDP (port))
977 port = scm_current_output_port ();
978 else
979 {
980 port = SCM_COERCE_OUTPORT (port);
981 SCM_VALIDATE_OPOUTPORT (1, port);
982 }
983 scm_flush (port);
984 return SCM_UNSPECIFIED;
985 }
986 #undef FUNC_NAME
987
988
989 static void
990 flush_output_port (void *closure, SCM port)
991 {
992 if (SCM_OPOUTPORTP (port))
993 scm_flush (port);
994 }
995
996 SCM_DEFINE (scm_flush_all_ports, "flush-all-ports", 0, 0, 0,
997 (),
998 "Equivalent to calling @code{force-output} on\n"
999 "all open output ports. The return value is unspecified.")
1000 #define FUNC_NAME s_scm_flush_all_ports
1001 {
1002 scm_c_port_for_each (&flush_output_port, NULL);
1003 return SCM_UNSPECIFIED;
1004 }
1005 #undef FUNC_NAME
1006
1007 SCM_DEFINE (scm_read_char, "read-char", 0, 1, 0,
1008 (SCM port),
1009 "Return the next character available from @var{port}, updating\n"
1010 "@var{port} to point to the following character. If no more\n"
1011 "characters are available, the end-of-file object is returned.")
1012 #define FUNC_NAME s_scm_read_char
1013 {
1014 scm_t_wchar c;
1015 if (SCM_UNBNDP (port))
1016 port = scm_current_input_port ();
1017 SCM_VALIDATE_OPINPORT (1, port);
1018 c = scm_getc (port);
1019 if (EOF == c)
1020 return SCM_EOF_VAL;
1021 return SCM_MAKE_CHAR (c);
1022 }
1023 #undef FUNC_NAME
1024
1025 #define SCM_MBCHAR_BUF_SIZE (4)
1026
1027 /* Read a codepoint from PORT and return it. Fill BUF with the byte
1028 representation of the codepoint in PORT's encoding, and set *LEN to
1029 the length in bytes of that representation. Raise an error on
1030 failure. */
1031 static scm_t_wchar
1032 get_codepoint (SCM port, char buf[SCM_MBCHAR_BUF_SIZE], size_t *len)
1033 {
1034 int c;
1035 size_t bufcount = 0;
1036 scm_t_uint32 result_buf;
1037 scm_t_wchar codepoint = 0;
1038 scm_t_uint32 *u32;
1039 size_t u32len;
1040 scm_t_port *pt = SCM_PTAB_ENTRY (port);
1041
1042 c = scm_get_byte_or_eof (port);
1043 if (c == EOF)
1044 return (scm_t_wchar) EOF;
1045
1046 buf[0] = c;
1047 bufcount++;
1048
1049 if (pt->encoding == NULL)
1050 {
1051 /* The encoding is Latin-1: bytes are characters. */
1052 codepoint = (unsigned char) buf[0];
1053 goto success;
1054 }
1055
1056 for (;;)
1057 {
1058 u32len = sizeof (result_buf) / sizeof (scm_t_uint32);
1059 u32 = u32_conv_from_encoding (pt->encoding,
1060 (enum iconv_ilseq_handler) pt->ilseq_handler,
1061 buf, bufcount, NULL, &result_buf, &u32len);
1062 if (u32 == NULL || u32len == 0)
1063 {
1064 if (errno == ENOMEM)
1065 scm_memory_error ("Input decoding");
1066
1067 /* Otherwise errno is EILSEQ or EINVAL, so perhaps more
1068 bytes are needed. Keep looping. */
1069 }
1070 else
1071 {
1072 /* Complete codepoint found. */
1073 codepoint = u32[0];
1074
1075 if (SCM_UNLIKELY (u32 != &result_buf))
1076 /* libunistring up to 0.9.3 (included) would always heap-allocate
1077 the result even when a large-enough RESULT_BUF is supplied, see
1078 <http://lists.gnu.org/archive/html/bug-libunistring/2010-07/msg00003.html>. */
1079 free (u32);
1080
1081 goto success;
1082 }
1083
1084 if (bufcount == SCM_MBCHAR_BUF_SIZE)
1085 {
1086 /* We've read several bytes and didn't find a good
1087 codepoint. Give up. */
1088 goto failure;
1089 }
1090
1091 c = scm_get_byte_or_eof (port);
1092
1093 if (c == EOF)
1094 {
1095 /* EOF before a complete character was read. Push it all
1096 back and return EOF. */
1097 while (bufcount > 0)
1098 {
1099 /* FIXME: this will probably cause errors in the port column. */
1100 scm_unget_byte (buf[bufcount-1], port);
1101 bufcount --;
1102 }
1103 return EOF;
1104 }
1105
1106 if (c == '\n')
1107 {
1108 /* It is always invalid to have EOL in the middle of a
1109 multibyte character. */
1110 scm_unget_byte ('\n', port);
1111 goto failure;
1112 }
1113
1114 buf[bufcount++] = c;
1115 }
1116
1117 success:
1118 switch (codepoint)
1119 {
1120 case '\a':
1121 break;
1122 case '\b':
1123 SCM_DECCOL (port);
1124 break;
1125 case '\n':
1126 SCM_INCLINE (port);
1127 break;
1128 case '\r':
1129 SCM_ZEROCOL (port);
1130 break;
1131 case '\t':
1132 SCM_TABCOL (port);
1133 break;
1134 default:
1135 SCM_INCCOL (port);
1136 break;
1137 }
1138
1139 *len = bufcount;
1140
1141 return codepoint;
1142
1143 failure:
1144 {
1145 char *err_buf;
1146 SCM err_str = scm_i_make_string (bufcount, &err_buf);
1147 memcpy (err_buf, buf, bufcount);
1148
1149 if (errno == EILSEQ)
1150 scm_misc_error (NULL, "input encoding error for ~s: ~s",
1151 scm_list_2 (scm_from_locale_string (scm_i_get_port_encoding (port)),
1152 err_str));
1153 else
1154 scm_misc_error (NULL, "input encoding error (invalid) for ~s: ~s\n",
1155 scm_list_2 (scm_from_locale_string (scm_i_get_port_encoding (port)),
1156 err_str));
1157 }
1158
1159 /* Never gets here. */
1160 return 0;
1161 }
1162
1163 /* Read a codepoint from PORT and return it. */
1164 scm_t_wchar
1165 scm_getc (SCM port)
1166 {
1167 size_t len;
1168 char buf[SCM_MBCHAR_BUF_SIZE];
1169
1170 return get_codepoint (port, buf, &len);
1171 }
1172
1173 /* this should only be called when the read buffer is empty. it
1174 tries to refill the read buffer. it returns the first char from
1175 the port, which is either EOF or *(pt->read_pos). */
1176 int
1177 scm_fill_input (SCM port)
1178 {
1179 scm_t_port *pt = SCM_PTAB_ENTRY (port);
1180
1181 assert (pt->read_pos == pt->read_end);
1182
1183 if (pt->read_buf == pt->putback_buf)
1184 {
1185 /* finished reading put-back chars. */
1186 pt->read_buf = pt->saved_read_buf;
1187 pt->read_pos = pt->saved_read_pos;
1188 pt->read_end = pt->saved_read_end;
1189 pt->read_buf_size = pt->saved_read_buf_size;
1190 if (pt->read_pos < pt->read_end)
1191 return *(pt->read_pos);
1192 }
1193 return scm_ptobs[SCM_PTOBNUM (port)].fill_input (port);
1194 }
1195
1196
1197 /* scm_lfwrite
1198 *
1199 * This function differs from scm_c_write; it updates port line and
1200 * column. */
1201
1202 static void
1203 update_port_lf (scm_t_wchar c, SCM port)
1204 {
1205 if (c == '\a')
1206 ; /* Do nothing. */
1207 else if (c == '\b')
1208 SCM_DECCOL (port);
1209 else if (c == '\n')
1210 SCM_INCLINE (port);
1211 else if (c == '\r')
1212 SCM_ZEROCOL (port);
1213 else if (c == '\t')
1214 SCM_TABCOL (port);
1215 else
1216 SCM_INCCOL (port);
1217 }
1218
1219 void
1220 scm_lfwrite (const char *ptr, size_t size, SCM port)
1221 {
1222 scm_t_port *pt = SCM_PTAB_ENTRY (port);
1223 scm_t_ptob_descriptor *ptob = &scm_ptobs[SCM_PTOBNUM (port)];
1224
1225 if (pt->rw_active == SCM_PORT_READ)
1226 scm_end_input (port);
1227
1228 ptob->write (port, ptr, size);
1229
1230 for (; size; ptr++, size--)
1231 update_port_lf ((scm_t_wchar) (unsigned char) *ptr, port);
1232
1233 if (pt->rw_random)
1234 pt->rw_active = SCM_PORT_WRITE;
1235 }
1236
1237 /* Write a scheme string STR to PORT from START inclusive to END
1238 exclusive. */
1239 void
1240 scm_lfwrite_substr (SCM str, size_t start, size_t end, SCM port)
1241 {
1242 size_t i, size = scm_i_string_length (str);
1243 scm_t_port *pt = SCM_PTAB_ENTRY (port);
1244 scm_t_ptob_descriptor *ptob = &scm_ptobs[SCM_PTOBNUM (port)];
1245 scm_t_wchar p;
1246 char *buf;
1247 size_t len;
1248
1249 if (pt->rw_active == SCM_PORT_READ)
1250 scm_end_input (port);
1251
1252 if (end == (size_t) (-1))
1253 end = size;
1254 size = end - start;
1255
1256 /* Note that making a substring will likely take the
1257 stringbuf_write_mutex. So, one shouldn't use scm_lfwrite_substr
1258 if the stringbuf write mutex may still be held elsewhere. */
1259 buf = scm_to_stringn (scm_c_substring (str, start, end), &len,
1260 pt->encoding, pt->ilseq_handler);
1261 ptob->write (port, buf, len);
1262 free (buf);
1263
1264 for (i = 0; i < size; i++)
1265 {
1266 p = scm_i_string_ref (str, i + start);
1267 update_port_lf (p, port);
1268 }
1269
1270 if (pt->rw_random)
1271 pt->rw_active = SCM_PORT_WRITE;
1272 }
1273
1274 /* Write a scheme string STR to PORT. */
1275 void
1276 scm_lfwrite_str (SCM str, SCM port)
1277 {
1278 size_t i, size = scm_i_string_length (str);
1279 scm_t_port *pt = SCM_PTAB_ENTRY (port);
1280 scm_t_ptob_descriptor *ptob = &scm_ptobs[SCM_PTOBNUM (port)];
1281 scm_t_wchar p;
1282 char *buf;
1283 size_t len;
1284
1285 if (pt->rw_active == SCM_PORT_READ)
1286 scm_end_input (port);
1287
1288 buf = scm_to_stringn (str, &len,
1289 pt->encoding, pt->ilseq_handler);
1290 ptob->write (port, buf, len);
1291 free (buf);
1292
1293 for (i = 0; i < size; i++)
1294 {
1295 p = scm_i_string_ref (str, i);
1296 update_port_lf (p, port);
1297 }
1298
1299 if (pt->rw_random)
1300 pt->rw_active = SCM_PORT_WRITE;
1301 }
1302
1303 /* scm_c_read
1304 *
1305 * Used by an application to read arbitrary number of bytes from an
1306 * SCM port. Same semantics as libc read, except that scm_c_read only
1307 * returns less than SIZE bytes if at end-of-file.
1308 *
1309 * Warning: Doesn't update port line and column counts! */
1310
1311 /* This structure, and the following swap_buffer function, are used
1312 for temporarily swapping a port's own read buffer, and the buffer
1313 that the caller of scm_c_read provides. */
1314 struct port_and_swap_buffer
1315 {
1316 scm_t_port *pt;
1317 unsigned char *buffer;
1318 size_t size;
1319 };
1320
1321 static void
1322 swap_buffer (void *data)
1323 {
1324 struct port_and_swap_buffer *psb = (struct port_and_swap_buffer *) data;
1325 unsigned char *old_buf = psb->pt->read_buf;
1326 size_t old_size = psb->pt->read_buf_size;
1327
1328 /* Make the port use (buffer, size) from the struct. */
1329 psb->pt->read_pos = psb->pt->read_buf = psb->pt->read_end = psb->buffer;
1330 psb->pt->read_buf_size = psb->size;
1331
1332 /* Save the port's old (buffer, size) in the struct. */
1333 psb->buffer = old_buf;
1334 psb->size = old_size;
1335 }
1336
1337 size_t
1338 scm_c_read (SCM port, void *buffer, size_t size)
1339 #define FUNC_NAME "scm_c_read"
1340 {
1341 scm_t_port *pt;
1342 size_t n_read = 0, n_available;
1343 struct port_and_swap_buffer psb;
1344
1345 SCM_VALIDATE_OPINPORT (1, port);
1346
1347 pt = SCM_PTAB_ENTRY (port);
1348 if (pt->rw_active == SCM_PORT_WRITE)
1349 scm_ptobs[SCM_PTOBNUM (port)].flush (port);
1350
1351 if (pt->rw_random)
1352 pt->rw_active = SCM_PORT_READ;
1353
1354 /* Take bytes first from the port's read buffer. */
1355 if (pt->read_pos < pt->read_end)
1356 {
1357 n_available = min (size, pt->read_end - pt->read_pos);
1358 memcpy (buffer, pt->read_pos, n_available);
1359 buffer = (char *) buffer + n_available;
1360 pt->read_pos += n_available;
1361 n_read += n_available;
1362 size -= n_available;
1363 }
1364
1365 /* Avoid the scm_dynwind_* costs if we now have enough data. */
1366 if (size == 0)
1367 return n_read;
1368
1369 /* Now we will call scm_fill_input repeatedly until we have read the
1370 requested number of bytes. (Note that a single scm_fill_input
1371 call does not guarantee to fill the whole of the port's read
1372 buffer.) */
1373 if (pt->read_buf_size <= 1 && pt->encoding == NULL)
1374 {
1375 /* The port that we are reading from is unbuffered - i.e. does
1376 not have its own persistent buffer - but we have a buffer,
1377 provided by our caller, that is the right size for the data
1378 that is wanted. For the following scm_fill_input calls,
1379 therefore, we use the buffer in hand as the port's read
1380 buffer.
1381
1382 We need to make sure that the port's normal (1 byte) buffer
1383 is reinstated in case one of the scm_fill_input () calls
1384 throws an exception; we use the scm_dynwind_* API to achieve
1385 that.
1386
1387 A consequence of this optimization is that the fill_input
1388 functions can't unget characters. That'll push data to the
1389 pushback buffer instead of this psb buffer. */
1390 #if SCM_DEBUG == 1
1391 unsigned char *pback = pt->putback_buf;
1392 #endif
1393 psb.pt = pt;
1394 psb.buffer = buffer;
1395 psb.size = size;
1396 scm_dynwind_begin (SCM_F_DYNWIND_REWINDABLE);
1397 scm_dynwind_rewind_handler (swap_buffer, &psb, SCM_F_WIND_EXPLICITLY);
1398 scm_dynwind_unwind_handler (swap_buffer, &psb, SCM_F_WIND_EXPLICITLY);
1399
1400 /* Call scm_fill_input until we have all the bytes that we need,
1401 or we hit EOF. */
1402 while (pt->read_buf_size && (scm_fill_input (port) != EOF))
1403 {
1404 pt->read_buf_size -= (pt->read_end - pt->read_pos);
1405 pt->read_pos = pt->read_buf = pt->read_end;
1406 }
1407 #if SCM_DEBUG == 1
1408 if (pback != pt->putback_buf
1409 || pt->read_buf - (unsigned char *) buffer < 0)
1410 scm_misc_error (FUNC_NAME,
1411 "scm_c_read must not call a fill function that pushes "
1412 "back characters onto an unbuffered port", SCM_EOL);
1413 #endif
1414 n_read += pt->read_buf - (unsigned char *) buffer;
1415
1416 /* Reinstate the port's normal buffer. */
1417 scm_dynwind_end ();
1418 }
1419 else
1420 {
1421 /* The port has its own buffer. It is important that we use it,
1422 even if it happens to be smaller than our caller's buffer, so
1423 that a custom port implementation's entry points (in
1424 particular, fill_input) can rely on the buffer always being
1425 the same as they first set up. */
1426 while (size && (scm_fill_input (port) != EOF))
1427 {
1428 n_available = min (size, pt->read_end - pt->read_pos);
1429 memcpy (buffer, pt->read_pos, n_available);
1430 buffer = (char *) buffer + n_available;
1431 pt->read_pos += n_available;
1432 n_read += n_available;
1433 size -= n_available;
1434 }
1435 }
1436
1437 return n_read;
1438 }
1439 #undef FUNC_NAME
1440
1441 /* scm_c_write
1442 *
1443 * Used by an application to write arbitrary number of bytes to an SCM
1444 * port. Similar semantics as libc write. However, unlike libc
1445 * write, scm_c_write writes the requested number of bytes and has no
1446 * return value.
1447 *
1448 * Warning: Doesn't update port line and column counts!
1449 */
1450
1451 void
1452 scm_c_write (SCM port, const void *ptr, size_t size)
1453 #define FUNC_NAME "scm_c_write"
1454 {
1455 scm_t_port *pt;
1456 scm_t_ptob_descriptor *ptob;
1457
1458 SCM_VALIDATE_OPOUTPORT (1, port);
1459
1460 pt = SCM_PTAB_ENTRY (port);
1461 ptob = &scm_ptobs[SCM_PTOBNUM (port)];
1462
1463 if (pt->rw_active == SCM_PORT_READ)
1464 scm_end_input (port);
1465
1466 ptob->write (port, ptr, size);
1467
1468 if (pt->rw_random)
1469 pt->rw_active = SCM_PORT_WRITE;
1470 }
1471 #undef FUNC_NAME
1472
1473 void
1474 scm_flush (SCM port)
1475 {
1476 long i = SCM_PTOBNUM (port);
1477 assert (i >= 0);
1478 (scm_ptobs[i].flush) (port);
1479 }
1480
1481 void
1482 scm_end_input (SCM port)
1483 {
1484 long offset;
1485 scm_t_port *pt = SCM_PTAB_ENTRY (port);
1486
1487 if (pt->read_buf == pt->putback_buf)
1488 {
1489 offset = pt->read_end - pt->read_pos;
1490 pt->read_buf = pt->saved_read_buf;
1491 pt->read_pos = pt->saved_read_pos;
1492 pt->read_end = pt->saved_read_end;
1493 pt->read_buf_size = pt->saved_read_buf_size;
1494 }
1495 else
1496 offset = 0;
1497
1498 scm_ptobs[SCM_PTOBNUM (port)].end_input (port, offset);
1499 }
1500
1501 \f
1502
1503
1504 void
1505 scm_unget_byte (int c, SCM port)
1506 #define FUNC_NAME "scm_unget_byte"
1507 {
1508 scm_t_port *pt = SCM_PTAB_ENTRY (port);
1509
1510 if (pt->read_buf == pt->putback_buf)
1511 /* already using the put-back buffer. */
1512 {
1513 /* enlarge putback_buf if necessary. */
1514 if (pt->read_end == pt->read_buf + pt->read_buf_size
1515 && pt->read_buf == pt->read_pos)
1516 {
1517 size_t new_size = pt->read_buf_size * 2;
1518 unsigned char *tmp = (unsigned char *)
1519 scm_gc_realloc (pt->putback_buf, pt->read_buf_size, new_size,
1520 "putback buffer");
1521
1522 pt->read_pos = pt->read_buf = pt->putback_buf = tmp;
1523 pt->read_end = pt->read_buf + pt->read_buf_size;
1524 pt->read_buf_size = pt->putback_buf_size = new_size;
1525 }
1526
1527 /* shift any existing bytes to buffer + 1. */
1528 if (pt->read_pos == pt->read_end)
1529 pt->read_end = pt->read_buf + 1;
1530 else if (pt->read_pos != pt->read_buf + 1)
1531 {
1532 int count = pt->read_end - pt->read_pos;
1533
1534 memmove (pt->read_buf + 1, pt->read_pos, count);
1535 pt->read_end = pt->read_buf + 1 + count;
1536 }
1537
1538 pt->read_pos = pt->read_buf;
1539 }
1540 else
1541 /* switch to the put-back buffer. */
1542 {
1543 if (pt->putback_buf == NULL)
1544 {
1545 pt->putback_buf
1546 = (unsigned char *) scm_gc_malloc_pointerless
1547 (SCM_INITIAL_PUTBACK_BUF_SIZE, "putback buffer");
1548 pt->putback_buf_size = SCM_INITIAL_PUTBACK_BUF_SIZE;
1549 }
1550
1551 pt->saved_read_buf = pt->read_buf;
1552 pt->saved_read_pos = pt->read_pos;
1553 pt->saved_read_end = pt->read_end;
1554 pt->saved_read_buf_size = pt->read_buf_size;
1555
1556 pt->read_pos = pt->read_buf = pt->putback_buf;
1557 pt->read_end = pt->read_buf + 1;
1558 pt->read_buf_size = pt->putback_buf_size;
1559 }
1560
1561 *pt->read_buf = c;
1562
1563 if (pt->rw_random)
1564 pt->rw_active = SCM_PORT_READ;
1565 }
1566 #undef FUNC_NAME
1567
1568 void
1569 scm_ungetc (scm_t_wchar c, SCM port)
1570 #define FUNC_NAME "scm_ungetc"
1571 {
1572 scm_t_port *pt = SCM_PTAB_ENTRY (port);
1573 char *result;
1574 char result_buf[10];
1575 const char *encoding;
1576 size_t len;
1577 int i;
1578
1579 if (pt->encoding != NULL)
1580 encoding = pt->encoding;
1581 else
1582 encoding = "ISO-8859-1";
1583
1584 len = sizeof (result_buf);
1585 result = u32_conv_to_encoding (encoding,
1586 (enum iconv_ilseq_handler) pt->ilseq_handler,
1587 (uint32_t *) &c, 1, NULL,
1588 result_buf, &len);
1589
1590 if (SCM_UNLIKELY (result == NULL || len == 0))
1591 {
1592 SCM chr;
1593
1594 chr = scm_integer_to_char (scm_from_uint32 (c));
1595 scm_encoding_error (FUNC_NAME, errno,
1596 "conversion to port encoding failed",
1597 "UTF-32", encoding,
1598 scm_string (scm_list_1 (chr)));
1599 }
1600
1601 for (i = len - 1; i >= 0; i--)
1602 scm_unget_byte (result[i], port);
1603
1604 if (SCM_UNLIKELY (result != result_buf))
1605 free (result);
1606
1607 if (c == '\n')
1608 {
1609 /* What should col be in this case?
1610 * We'll leave it at -1.
1611 */
1612 SCM_LINUM (port) -= 1;
1613 }
1614 else
1615 SCM_COL(port) -= 1;
1616 }
1617 #undef FUNC_NAME
1618
1619
1620 void
1621 scm_ungets (const char *s, int n, SCM port)
1622 {
1623 /* This is simple minded and inefficient, but unreading strings is
1624 * probably not a common operation, and remember that line and
1625 * column numbers have to be handled...
1626 *
1627 * Please feel free to write an optimized version!
1628 */
1629 while (n--)
1630 scm_ungetc (s[n], port);
1631 }
1632
1633
1634 SCM_DEFINE (scm_peek_char, "peek-char", 0, 1, 0,
1635 (SCM port),
1636 "Return the next character available from @var{port},\n"
1637 "@emph{without} updating @var{port} to point to the following\n"
1638 "character. If no more characters are available, the\n"
1639 "end-of-file object is returned.\n"
1640 "\n"
1641 "The value returned by\n"
1642 "a call to @code{peek-char} is the same as the value that would\n"
1643 "have been returned by a call to @code{read-char} on the same\n"
1644 "port. The only difference is that the very next call to\n"
1645 "@code{read-char} or @code{peek-char} on that @var{port} will\n"
1646 "return the value returned by the preceding call to\n"
1647 "@code{peek-char}. In particular, a call to @code{peek-char} on\n"
1648 "an interactive port will hang waiting for input whenever a call\n"
1649 "to @code{read-char} would have hung.")
1650 #define FUNC_NAME s_scm_peek_char
1651 {
1652 SCM result;
1653 scm_t_wchar c;
1654 char bytes[SCM_MBCHAR_BUF_SIZE];
1655 long column, line;
1656 size_t len;
1657
1658 if (SCM_UNBNDP (port))
1659 port = scm_current_input_port ();
1660 SCM_VALIDATE_OPINPORT (1, port);
1661
1662 column = SCM_COL (port);
1663 line = SCM_LINUM (port);
1664
1665 c = get_codepoint (port, bytes, &len);
1666 if (c == EOF)
1667 result = SCM_EOF_VAL;
1668 else
1669 {
1670 long i;
1671
1672 result = SCM_MAKE_CHAR (c);
1673
1674 for (i = len - 1; i >= 0; i--)
1675 scm_unget_byte (bytes[i], port);
1676
1677 SCM_COL (port) = column;
1678 SCM_LINUM (port) = line;
1679 }
1680
1681 return result;
1682 }
1683 #undef FUNC_NAME
1684
1685 SCM_DEFINE (scm_unread_char, "unread-char", 1, 1, 0,
1686 (SCM cobj, SCM port),
1687 "Place @var{char} in @var{port} so that it will be read by the\n"
1688 "next read operation. If called multiple times, the unread characters\n"
1689 "will be read again in last-in first-out order. If @var{port} is\n"
1690 "not supplied, the current input port is used.")
1691 #define FUNC_NAME s_scm_unread_char
1692 {
1693 int c;
1694
1695 SCM_VALIDATE_CHAR (1, cobj);
1696 if (SCM_UNBNDP (port))
1697 port = scm_current_input_port ();
1698 SCM_VALIDATE_OPINPORT (2, port);
1699
1700 c = SCM_CHAR (cobj);
1701
1702 scm_ungetc (c, port);
1703 return cobj;
1704 }
1705 #undef FUNC_NAME
1706
1707 SCM_DEFINE (scm_unread_string, "unread-string", 2, 0, 0,
1708 (SCM str, SCM port),
1709 "Place the string @var{str} in @var{port} so that its characters will be\n"
1710 "read in subsequent read operations. If called multiple times, the\n"
1711 "unread characters will be read again in last-in first-out order. If\n"
1712 "@var{port} is not supplied, the current-input-port is used.")
1713 #define FUNC_NAME s_scm_unread_string
1714 {
1715 int n;
1716 SCM_VALIDATE_STRING (1, str);
1717 if (SCM_UNBNDP (port))
1718 port = scm_current_input_port ();
1719 SCM_VALIDATE_OPINPORT (2, port);
1720
1721 n = scm_i_string_length (str);
1722
1723 while (n--)
1724 scm_ungetc (scm_i_string_ref (str, n), port);
1725
1726 return str;
1727 }
1728 #undef FUNC_NAME
1729
1730 SCM_DEFINE (scm_seek, "seek", 3, 0, 0,
1731 (SCM fd_port, SCM offset, SCM whence),
1732 "Sets the current position of @var{fd/port} to the integer\n"
1733 "@var{offset}, which is interpreted according to the value of\n"
1734 "@var{whence}.\n"
1735 "\n"
1736 "One of the following variables should be supplied for\n"
1737 "@var{whence}:\n"
1738 "@defvar SEEK_SET\n"
1739 "Seek from the beginning of the file.\n"
1740 "@end defvar\n"
1741 "@defvar SEEK_CUR\n"
1742 "Seek from the current position.\n"
1743 "@end defvar\n"
1744 "@defvar SEEK_END\n"
1745 "Seek from the end of the file.\n"
1746 "@end defvar\n"
1747 "If @var{fd/port} is a file descriptor, the underlying system\n"
1748 "call is @code{lseek}. @var{port} may be a string port.\n"
1749 "\n"
1750 "The value returned is the new position in the file. This means\n"
1751 "that the current position of a port can be obtained using:\n"
1752 "@lisp\n"
1753 "(seek port 0 SEEK_CUR)\n"
1754 "@end lisp")
1755 #define FUNC_NAME s_scm_seek
1756 {
1757 int how;
1758
1759 fd_port = SCM_COERCE_OUTPORT (fd_port);
1760
1761 how = scm_to_int (whence);
1762 if (how != SEEK_SET && how != SEEK_CUR && how != SEEK_END)
1763 SCM_OUT_OF_RANGE (3, whence);
1764
1765 if (SCM_OPPORTP (fd_port))
1766 {
1767 scm_t_ptob_descriptor *ptob = scm_ptobs + SCM_PTOBNUM (fd_port);
1768 off_t_or_off64_t off = scm_to_off_t_or_off64_t (offset);
1769 off_t_or_off64_t rv;
1770
1771 if (!ptob->seek)
1772 SCM_MISC_ERROR ("port is not seekable",
1773 scm_cons (fd_port, SCM_EOL));
1774 else
1775 rv = ptob->seek (fd_port, off, how);
1776 return scm_from_off_t_or_off64_t (rv);
1777 }
1778 else /* file descriptor?. */
1779 {
1780 off_t_or_off64_t off = scm_to_off_t_or_off64_t (offset);
1781 off_t_or_off64_t rv;
1782 rv = lseek_or_lseek64 (scm_to_int (fd_port), off, how);
1783 if (rv == -1)
1784 SCM_SYSERROR;
1785 return scm_from_off_t_or_off64_t (rv);
1786 }
1787 }
1788 #undef FUNC_NAME
1789
1790 #ifndef O_BINARY
1791 #define O_BINARY 0
1792 #endif
1793
1794 /* Mingw has ftruncate(), perhaps implemented above using chsize, but
1795 doesn't have the filename version truncate(), hence this code. */
1796 #if HAVE_FTRUNCATE && ! HAVE_TRUNCATE
1797 static int
1798 truncate (const char *file, off_t length)
1799 {
1800 int ret, fdes;
1801
1802 fdes = open (file, O_BINARY | O_WRONLY);
1803 if (fdes == -1)
1804 return -1;
1805
1806 ret = ftruncate (fdes, length);
1807 if (ret == -1)
1808 {
1809 int save_errno = errno;
1810 close (fdes);
1811 errno = save_errno;
1812 return -1;
1813 }
1814
1815 return close (fdes);
1816 }
1817 #endif /* HAVE_FTRUNCATE && ! HAVE_TRUNCATE */
1818
1819 SCM_DEFINE (scm_truncate_file, "truncate-file", 1, 1, 0,
1820 (SCM object, SCM length),
1821 "Truncate @var{file} to @var{length} bytes. @var{file} can be a\n"
1822 "filename string, a port object, or an integer file descriptor.\n"
1823 "The return value is unspecified.\n"
1824 "\n"
1825 "For a port or file descriptor @var{length} can be omitted, in\n"
1826 "which case the file is truncated at the current position (per\n"
1827 "@code{ftell} above).\n"
1828 "\n"
1829 "On most systems a file can be extended by giving a length\n"
1830 "greater than the current size, but this is not mandatory in the\n"
1831 "POSIX standard.")
1832 #define FUNC_NAME s_scm_truncate_file
1833 {
1834 int rv;
1835
1836 /* "object" can be a port, fdes or filename.
1837
1838 Negative "length" makes no sense, but it's left to truncate() or
1839 ftruncate() to give back an error for that (normally EINVAL).
1840 */
1841
1842 if (SCM_UNBNDP (length))
1843 {
1844 /* must supply length if object is a filename. */
1845 if (scm_is_string (object))
1846 SCM_MISC_ERROR("must supply length if OBJECT is a filename", SCM_EOL);
1847
1848 length = scm_seek (object, SCM_INUM0, scm_from_int (SEEK_CUR));
1849 }
1850
1851 object = SCM_COERCE_OUTPORT (object);
1852 if (scm_is_integer (object))
1853 {
1854 off_t_or_off64_t c_length = scm_to_off_t_or_off64_t (length);
1855 SCM_SYSCALL (rv = ftruncate_or_ftruncate64 (scm_to_int (object),
1856 c_length));
1857 }
1858 else if (SCM_OPOUTPORTP (object))
1859 {
1860 off_t_or_off64_t c_length = scm_to_off_t_or_off64_t (length);
1861 scm_t_port *pt = SCM_PTAB_ENTRY (object);
1862 scm_t_ptob_descriptor *ptob = scm_ptobs + SCM_PTOBNUM (object);
1863
1864 if (!ptob->truncate)
1865 SCM_MISC_ERROR ("port is not truncatable", SCM_EOL);
1866 if (pt->rw_active == SCM_PORT_READ)
1867 scm_end_input (object);
1868 else if (pt->rw_active == SCM_PORT_WRITE)
1869 ptob->flush (object);
1870
1871 ptob->truncate (object, c_length);
1872 rv = 0;
1873 }
1874 else
1875 {
1876 off_t_or_off64_t c_length = scm_to_off_t_or_off64_t (length);
1877 char *str = scm_to_locale_string (object);
1878 int eno;
1879 SCM_SYSCALL (rv = truncate_or_truncate64 (str, c_length));
1880 eno = errno;
1881 free (str);
1882 errno = eno;
1883 }
1884 if (rv == -1)
1885 SCM_SYSERROR;
1886 return SCM_UNSPECIFIED;
1887 }
1888 #undef FUNC_NAME
1889
1890 SCM_DEFINE (scm_port_line, "port-line", 1, 0, 0,
1891 (SCM port),
1892 "Return the current line number for @var{port}.\n"
1893 "\n"
1894 "The first line of a file is 0. But you might want to add 1\n"
1895 "when printing line numbers, since starting from 1 is\n"
1896 "traditional in error messages, and likely to be more natural to\n"
1897 "non-programmers.")
1898 #define FUNC_NAME s_scm_port_line
1899 {
1900 port = SCM_COERCE_OUTPORT (port);
1901 SCM_VALIDATE_OPENPORT (1, port);
1902 return scm_from_long (SCM_LINUM (port));
1903 }
1904 #undef FUNC_NAME
1905
1906 SCM_DEFINE (scm_set_port_line_x, "set-port-line!", 2, 0, 0,
1907 (SCM port, SCM line),
1908 "Set the current line number for @var{port} to @var{line}. The\n"
1909 "first line of a file is 0.")
1910 #define FUNC_NAME s_scm_set_port_line_x
1911 {
1912 port = SCM_COERCE_OUTPORT (port);
1913 SCM_VALIDATE_OPENPORT (1, port);
1914 SCM_PTAB_ENTRY (port)->line_number = scm_to_long (line);
1915 return SCM_UNSPECIFIED;
1916 }
1917 #undef FUNC_NAME
1918
1919 SCM_DEFINE (scm_port_column, "port-column", 1, 0, 0,
1920 (SCM port),
1921 "Return the current column number of @var{port}.\n"
1922 "If the number is\n"
1923 "unknown, the result is #f. Otherwise, the result is a 0-origin integer\n"
1924 "- i.e. the first character of the first line is line 0, column 0.\n"
1925 "(However, when you display a file position, for example in an error\n"
1926 "message, we recommend you add 1 to get 1-origin integers. This is\n"
1927 "because lines and column numbers traditionally start with 1, and that is\n"
1928 "what non-programmers will find most natural.)")
1929 #define FUNC_NAME s_scm_port_column
1930 {
1931 port = SCM_COERCE_OUTPORT (port);
1932 SCM_VALIDATE_OPENPORT (1, port);
1933 return scm_from_int (SCM_COL (port));
1934 }
1935 #undef FUNC_NAME
1936
1937 SCM_DEFINE (scm_set_port_column_x, "set-port-column!", 2, 0, 0,
1938 (SCM port, SCM column),
1939 "Set the current column of @var{port}. Before reading the first\n"
1940 "character on a line the column should be 0.")
1941 #define FUNC_NAME s_scm_set_port_column_x
1942 {
1943 port = SCM_COERCE_OUTPORT (port);
1944 SCM_VALIDATE_OPENPORT (1, port);
1945 SCM_PTAB_ENTRY (port)->column_number = scm_to_int (column);
1946 return SCM_UNSPECIFIED;
1947 }
1948 #undef FUNC_NAME
1949
1950 SCM_DEFINE (scm_port_filename, "port-filename", 1, 0, 0,
1951 (SCM port),
1952 "Return the filename associated with @var{port}. This function returns\n"
1953 "the strings \"standard input\", \"standard output\" and \"standard error\"\n"
1954 "when called on the current input, output and error ports respectively.")
1955 #define FUNC_NAME s_scm_port_filename
1956 {
1957 port = SCM_COERCE_OUTPORT (port);
1958 SCM_VALIDATE_OPENPORT (1, port);
1959 return SCM_FILENAME (port);
1960 }
1961 #undef FUNC_NAME
1962
1963 SCM_DEFINE (scm_set_port_filename_x, "set-port-filename!", 2, 0, 0,
1964 (SCM port, SCM filename),
1965 "Change the filename associated with @var{port}, using the current input\n"
1966 "port if none is specified. Note that this does not change the port's\n"
1967 "source of data, but only the value that is returned by\n"
1968 "@code{port-filename} and reported in diagnostic output.")
1969 #define FUNC_NAME s_scm_set_port_filename_x
1970 {
1971 port = SCM_COERCE_OUTPORT (port);
1972 SCM_VALIDATE_OPENPORT (1, port);
1973 /* We allow the user to set the filename to whatever he likes. */
1974 SCM_SET_FILENAME (port, filename);
1975 return SCM_UNSPECIFIED;
1976 }
1977 #undef FUNC_NAME
1978
1979 /* A fluid specifying the default encoding for newly created ports. If it is
1980 a string, that is the encoding. If it is #f, it is in the "native"
1981 (Latin-1) encoding. */
1982 SCM_VARIABLE (default_port_encoding_var, "%default-port-encoding");
1983
1984 static int scm_port_encoding_init = 0;
1985
1986 /* Return a C string representation of the current encoding. */
1987 const char *
1988 scm_i_get_port_encoding (SCM port)
1989 {
1990 SCM encoding;
1991
1992 if (scm_is_false (port))
1993 {
1994 if (!scm_port_encoding_init)
1995 return NULL;
1996 else if (!scm_is_fluid (SCM_VARIABLE_REF (default_port_encoding_var)))
1997 return NULL;
1998 else
1999 {
2000 encoding = scm_fluid_ref (SCM_VARIABLE_REF (default_port_encoding_var));
2001 if (!scm_is_string (encoding))
2002 return NULL;
2003 else
2004 return scm_i_string_chars (encoding);
2005 }
2006 }
2007 else
2008 {
2009 scm_t_port *pt;
2010 pt = SCM_PTAB_ENTRY (port);
2011 if (pt->encoding)
2012 return pt->encoding;
2013 else
2014 return NULL;
2015 }
2016 }
2017
2018 /* Returns ENC if it is a recognized encoding. If it isn't, it tries
2019 to find an alias of ENC that is valid. Otherwise, it returns
2020 NULL. */
2021 static const char *
2022 find_valid_encoding (const char *enc)
2023 {
2024 int isvalid = 0;
2025 const char str[] = " ";
2026 scm_t_uint32 result_buf;
2027 scm_t_uint32 *u32;
2028 size_t u32len;
2029
2030 u32len = sizeof (result_buf) / sizeof (scm_t_uint32);
2031 u32 = u32_conv_from_encoding (enc, iconveh_error, str, 1,
2032 NULL, &result_buf, &u32len);
2033 isvalid = (u32 != NULL);
2034
2035 if (SCM_UNLIKELY (u32 != &result_buf))
2036 free (u32);
2037
2038 if (isvalid)
2039 return enc;
2040
2041 return NULL;
2042 }
2043
2044 void
2045 scm_i_set_port_encoding_x (SCM port, const char *enc)
2046 {
2047 const char *valid_enc;
2048 scm_t_port *pt;
2049
2050 /* Null is shorthand for the native, Latin-1 encoding. */
2051 if (enc == NULL)
2052 valid_enc = NULL;
2053 else
2054 {
2055 valid_enc = find_valid_encoding (enc);
2056 if (valid_enc == NULL)
2057 {
2058 SCM err;
2059 err = scm_from_locale_string (enc);
2060 scm_misc_error (NULL, "invalid or unknown character encoding ~s",
2061 scm_list_1 (err));
2062 }
2063 }
2064
2065 if (scm_is_false (port))
2066 {
2067 /* Set the default encoding for future ports. */
2068 if (!scm_port_encoding_init
2069 || !scm_is_fluid (SCM_VARIABLE_REF (default_port_encoding_var)))
2070 scm_misc_error (NULL, "tried to set port encoding fluid before it is initialized",
2071 SCM_EOL);
2072
2073 if (valid_enc == NULL
2074 || !strcmp (valid_enc, "ASCII")
2075 || !strcmp (valid_enc, "ANSI_X3.4-1968")
2076 || !strcmp (valid_enc, "ISO-8859-1"))
2077 scm_fluid_set_x (SCM_VARIABLE_REF (default_port_encoding_var), SCM_BOOL_F);
2078 else
2079 scm_fluid_set_x (SCM_VARIABLE_REF (default_port_encoding_var),
2080 scm_from_locale_string (valid_enc));
2081 }
2082 else
2083 {
2084 /* Set the character encoding for this port. */
2085 pt = SCM_PTAB_ENTRY (port);
2086 if (valid_enc == NULL)
2087 pt->encoding = NULL;
2088 else
2089 pt->encoding = scm_gc_strdup (valid_enc, "port");
2090 }
2091 }
2092
2093 SCM_DEFINE (scm_port_encoding, "port-encoding", 1, 0, 0,
2094 (SCM port),
2095 "Returns, as a string, the character encoding that @var{port}\n"
2096 "uses to interpret its input and output.\n")
2097 #define FUNC_NAME s_scm_port_encoding
2098 {
2099 scm_t_port *pt;
2100 const char *enc;
2101
2102 SCM_VALIDATE_PORT (1, port);
2103
2104 pt = SCM_PTAB_ENTRY (port);
2105 enc = scm_i_get_port_encoding (port);
2106 if (enc)
2107 return scm_from_locale_string (pt->encoding);
2108 else
2109 return SCM_BOOL_F;
2110 }
2111 #undef FUNC_NAME
2112
2113 SCM_DEFINE (scm_set_port_encoding_x, "set-port-encoding!", 2, 0, 0,
2114 (SCM port, SCM enc),
2115 "Sets the character encoding that will be used to interpret all\n"
2116 "port I/O. New ports are created with the encoding\n"
2117 "appropriate for the current locale if @code{setlocale} has \n"
2118 "been called or ISO-8859-1 otherwise\n"
2119 "and this procedure can be used to modify that encoding.\n")
2120 #define FUNC_NAME s_scm_set_port_encoding_x
2121 {
2122 char *enc_str;
2123 const char *valid_enc_str;
2124
2125 SCM_VALIDATE_PORT (1, port);
2126 SCM_VALIDATE_STRING (2, enc);
2127
2128 enc_str = scm_to_locale_string (enc);
2129 valid_enc_str = find_valid_encoding (enc_str);
2130 if (valid_enc_str == NULL)
2131 {
2132 free (enc_str);
2133 scm_misc_error (FUNC_NAME, "invalid or unknown character encoding ~s",
2134 scm_list_1 (enc));
2135 }
2136 else
2137 {
2138 scm_i_set_port_encoding_x (port, valid_enc_str);
2139 free (enc_str);
2140 }
2141 return SCM_UNSPECIFIED;
2142 }
2143 #undef FUNC_NAME
2144
2145
2146 /* This determines how conversions handle unconvertible characters. */
2147 SCM_GLOBAL_VARIABLE (scm_conversion_strategy, "%port-conversion-strategy");
2148 static int scm_conversion_strategy_init = 0;
2149
2150 scm_t_string_failed_conversion_handler
2151 scm_i_get_conversion_strategy (SCM port)
2152 {
2153 SCM encoding;
2154
2155 if (scm_is_false (port))
2156 {
2157 if (!scm_conversion_strategy_init
2158 || !scm_is_fluid (SCM_VARIABLE_REF (scm_conversion_strategy)))
2159 return SCM_FAILED_CONVERSION_QUESTION_MARK;
2160 else
2161 {
2162 encoding = scm_fluid_ref (SCM_VARIABLE_REF (scm_conversion_strategy));
2163 if (scm_is_false (encoding))
2164 return SCM_FAILED_CONVERSION_QUESTION_MARK;
2165 else
2166 return (scm_t_string_failed_conversion_handler) scm_to_int (encoding);
2167 }
2168 }
2169 else
2170 {
2171 scm_t_port *pt;
2172 pt = SCM_PTAB_ENTRY (port);
2173 return pt->ilseq_handler;
2174 }
2175
2176 }
2177
2178 void
2179 scm_i_set_conversion_strategy_x (SCM port,
2180 scm_t_string_failed_conversion_handler handler)
2181 {
2182 SCM strategy;
2183 scm_t_port *pt;
2184
2185 strategy = scm_from_int ((int) handler);
2186
2187 if (scm_is_false (port))
2188 {
2189 /* Set the default encoding for future ports. */
2190 if (!scm_conversion_strategy
2191 || !scm_is_fluid (SCM_VARIABLE_REF (scm_conversion_strategy)))
2192 scm_misc_error (NULL, "tried to set conversion strategy fluid before it is initialized",
2193 SCM_EOL);
2194 scm_fluid_set_x (SCM_VARIABLE_REF (scm_conversion_strategy), strategy);
2195 }
2196 else
2197 {
2198 /* Set the character encoding for this port. */
2199 pt = SCM_PTAB_ENTRY (port);
2200 pt->ilseq_handler = handler;
2201 }
2202 }
2203
2204 SCM_DEFINE (scm_port_conversion_strategy, "port-conversion-strategy",
2205 1, 0, 0, (SCM port),
2206 "Returns the behavior of the port when handling a character that\n"
2207 "is not representable in the port's current encoding.\n"
2208 "It returns the symbol @code{error} if unrepresentable characters\n"
2209 "should cause exceptions, @code{substitute} if the port should\n"
2210 "try to replace unrepresentable characters with question marks or\n"
2211 "approximate characters, or @code{escape} if unrepresentable\n"
2212 "characters should be converted to string escapes.\n"
2213 "\n"
2214 "If @var{port} is @code{#f}, then the current default behavior\n"
2215 "will be returned. New ports will have this default behavior\n"
2216 "when they are created.\n")
2217 #define FUNC_NAME s_scm_port_conversion_strategy
2218 {
2219 scm_t_string_failed_conversion_handler h;
2220
2221 SCM_VALIDATE_OPPORT (1, port);
2222
2223 if (!scm_is_false (port))
2224 {
2225 SCM_VALIDATE_OPPORT (1, port);
2226 }
2227
2228 h = scm_i_get_conversion_strategy (port);
2229 if (h == SCM_FAILED_CONVERSION_ERROR)
2230 return scm_from_locale_symbol ("error");
2231 else if (h == SCM_FAILED_CONVERSION_QUESTION_MARK)
2232 return scm_from_locale_symbol ("substitute");
2233 else if (h == SCM_FAILED_CONVERSION_ESCAPE_SEQUENCE)
2234 return scm_from_locale_symbol ("escape");
2235 else
2236 abort ();
2237
2238 /* Never gets here. */
2239 return SCM_UNDEFINED;
2240 }
2241 #undef FUNC_NAME
2242
2243 SCM_DEFINE (scm_set_port_conversion_strategy_x, "set-port-conversion-strategy!",
2244 2, 0, 0,
2245 (SCM port, SCM sym),
2246 "Sets the behavior of the interpreter when outputting a character\n"
2247 "that is not representable in the port's current encoding.\n"
2248 "@var{sym} can be either @code{'error}, @code{'substitute}, or\n"
2249 "@code{'escape}. If it is @code{'error}, an error will be thrown\n"
2250 "when an unconvertible character is encountered. If it is\n"
2251 "@code{'substitute}, then unconvertible characters will \n"
2252 "be replaced with approximate characters, or with question marks\n"
2253 "if no approximately correct character is available.\n"
2254 "If it is @code{'escape},\n"
2255 "it will appear as a hex escape when output.\n"
2256 "\n"
2257 "If @var{port} is an open port, the conversion error behavior\n"
2258 "is set for that port. If it is @code{#f}, it is set as the\n"
2259 "default behavior for any future ports that get created in\n"
2260 "this thread.\n")
2261 #define FUNC_NAME s_scm_set_port_conversion_strategy_x
2262 {
2263 SCM err;
2264 SCM qm;
2265 SCM esc;
2266
2267 if (!scm_is_false (port))
2268 {
2269 SCM_VALIDATE_OPPORT (1, port);
2270 }
2271
2272 err = scm_from_locale_symbol ("error");
2273 if (scm_is_true (scm_eqv_p (sym, err)))
2274 {
2275 scm_i_set_conversion_strategy_x (port, SCM_FAILED_CONVERSION_ERROR);
2276 return SCM_UNSPECIFIED;
2277 }
2278
2279 qm = scm_from_locale_symbol ("substitute");
2280 if (scm_is_true (scm_eqv_p (sym, qm)))
2281 {
2282 scm_i_set_conversion_strategy_x (port,
2283 SCM_FAILED_CONVERSION_QUESTION_MARK);
2284 return SCM_UNSPECIFIED;
2285 }
2286
2287 esc = scm_from_locale_symbol ("escape");
2288 if (scm_is_true (scm_eqv_p (sym, esc)))
2289 {
2290 scm_i_set_conversion_strategy_x (port,
2291 SCM_FAILED_CONVERSION_ESCAPE_SEQUENCE);
2292 return SCM_UNSPECIFIED;
2293 }
2294
2295 SCM_MISC_ERROR ("unknown conversion behavior ~s", scm_list_1 (sym));
2296
2297 return SCM_UNSPECIFIED;
2298 }
2299 #undef FUNC_NAME
2300
2301
2302
2303 void
2304 scm_print_port_mode (SCM exp, SCM port)
2305 {
2306 scm_puts (SCM_CLOSEDP (exp)
2307 ? "closed: "
2308 : (SCM_RDNG & SCM_CELL_WORD_0 (exp)
2309 ? (SCM_WRTNG & SCM_CELL_WORD_0 (exp)
2310 ? "input-output: "
2311 : "input: ")
2312 : (SCM_WRTNG & SCM_CELL_WORD_0 (exp)
2313 ? "output: "
2314 : "bogus: ")),
2315 port);
2316 }
2317
2318 int
2319 scm_port_print (SCM exp, SCM port, scm_print_state *pstate SCM_UNUSED)
2320 {
2321 char *type = SCM_PTOBNAME (SCM_PTOBNUM (exp));
2322 if (!type)
2323 type = "port";
2324 scm_puts ("#<", port);
2325 scm_print_port_mode (exp, port);
2326 scm_puts (type, port);
2327 scm_putc (' ', port);
2328 scm_uintprint (SCM_CELL_WORD_1 (exp), 16, port);
2329 scm_putc ('>', port);
2330 return 1;
2331 }
2332
2333 \f
2334
2335 /* Void ports. */
2336
2337 scm_t_bits scm_tc16_void_port = 0;
2338
2339 static int fill_input_void_port (SCM port SCM_UNUSED)
2340 {
2341 return EOF;
2342 }
2343
2344 static void
2345 write_void_port (SCM port SCM_UNUSED,
2346 const void *data SCM_UNUSED,
2347 size_t size SCM_UNUSED)
2348 {
2349 }
2350
2351 static SCM
2352 scm_i_void_port (long mode_bits)
2353 {
2354 scm_i_scm_pthread_mutex_lock (&scm_i_port_table_mutex);
2355 {
2356 SCM answer = scm_new_port_table_entry (scm_tc16_void_port);
2357 scm_t_port * pt = SCM_PTAB_ENTRY(answer);
2358
2359 scm_port_non_buffer (pt);
2360
2361 SCM_SETSTREAM (answer, 0);
2362 SCM_SET_CELL_TYPE (answer, scm_tc16_void_port | mode_bits);
2363 scm_i_pthread_mutex_unlock (&scm_i_port_table_mutex);
2364 return answer;
2365 }
2366 }
2367
2368 SCM
2369 scm_void_port (char *mode_str)
2370 {
2371 return scm_i_void_port (scm_mode_bits (mode_str));
2372 }
2373
2374 SCM_DEFINE (scm_sys_make_void_port, "%make-void-port", 1, 0, 0,
2375 (SCM mode),
2376 "Create and return a new void port. A void port acts like\n"
2377 "@file{/dev/null}. The @var{mode} argument\n"
2378 "specifies the input/output modes for this port: see the\n"
2379 "documentation for @code{open-file} in @ref{File Ports}.")
2380 #define FUNC_NAME s_scm_sys_make_void_port
2381 {
2382 return scm_i_void_port (scm_i_mode_bits (mode));
2383 }
2384 #undef FUNC_NAME
2385
2386 \f
2387 /* Initialization. */
2388
2389 void
2390 scm_init_ports ()
2391 {
2392 /* lseek() symbols. */
2393 scm_c_define ("SEEK_SET", scm_from_int (SEEK_SET));
2394 scm_c_define ("SEEK_CUR", scm_from_int (SEEK_CUR));
2395 scm_c_define ("SEEK_END", scm_from_int (SEEK_END));
2396
2397 scm_tc16_void_port = scm_make_port_type ("void", fill_input_void_port,
2398 write_void_port);
2399
2400 cur_inport_fluid = scm_make_fluid ();
2401 cur_outport_fluid = scm_make_fluid ();
2402 cur_errport_fluid = scm_make_fluid ();
2403 cur_loadport_fluid = scm_make_fluid ();
2404
2405 scm_i_port_weak_hash = scm_make_weak_key_hash_table (SCM_I_MAKINUM(31));
2406
2407 #include "libguile/ports.x"
2408
2409 /* Use Latin-1 as the default port encoding. */
2410 SCM_VARIABLE_SET (default_port_encoding_var, scm_make_fluid ());
2411 scm_fluid_set_x (SCM_VARIABLE_REF (default_port_encoding_var), SCM_BOOL_F);
2412 scm_port_encoding_init = 1;
2413
2414 SCM_VARIABLE_SET (scm_conversion_strategy, scm_make_fluid ());
2415 scm_fluid_set_x (SCM_VARIABLE_REF (scm_conversion_strategy),
2416 scm_from_int ((int) SCM_FAILED_CONVERSION_QUESTION_MARK));
2417 scm_conversion_strategy_init = 1;
2418
2419 }
2420
2421 /*
2422 Local Variables:
2423 c-file-style: "gnu"
2424 End:
2425 */