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