Always initialize a port's encoding name.
[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.")
1033 #define FUNC_NAME s_scm_read_char
1034 {
1035 scm_t_wchar c;
1036 if (SCM_UNBNDP (port))
1037 port = scm_current_input_port ();
1038 SCM_VALIDATE_OPINPORT (1, port);
1039 c = scm_getc (port);
1040 if (EOF == c)
1041 return SCM_EOF_VAL;
1042 return SCM_MAKE_CHAR (c);
1043 }
1044 #undef FUNC_NAME
1045
1046 /* Update the line and column number of PORT after consumption of C. */
1047 static inline void
1048 update_port_lf (scm_t_wchar c, SCM port)
1049 {
1050 switch (c)
1051 {
1052 case '\a':
1053 break;
1054 case '\b':
1055 SCM_DECCOL (port);
1056 break;
1057 case '\n':
1058 SCM_INCLINE (port);
1059 break;
1060 case '\r':
1061 SCM_ZEROCOL (port);
1062 break;
1063 case '\t':
1064 SCM_TABCOL (port);
1065 break;
1066 default:
1067 SCM_INCCOL (port);
1068 break;
1069 }
1070 }
1071
1072 #define SCM_MBCHAR_BUF_SIZE (4)
1073
1074 /* Convert the SIZE-byte UTF-8 sequence in UTF8_BUF to a codepoint.
1075 UTF8_BUF is assumed to contain a valid UTF-8 sequence. */
1076 static scm_t_wchar
1077 utf8_to_codepoint (const scm_t_uint8 *utf8_buf, size_t size)
1078 {
1079 scm_t_wchar codepoint;
1080
1081 if (utf8_buf[0] <= 0x7f)
1082 {
1083 assert (size == 1);
1084 codepoint = utf8_buf[0];
1085 }
1086 else if ((utf8_buf[0] & 0xe0) == 0xc0)
1087 {
1088 assert (size == 2);
1089 codepoint = ((scm_t_wchar) utf8_buf[0] & 0x1f) << 6UL
1090 | (utf8_buf[1] & 0x3f);
1091 }
1092 else if ((utf8_buf[0] & 0xf0) == 0xe0)
1093 {
1094 assert (size == 3);
1095 codepoint = ((scm_t_wchar) utf8_buf[0] & 0x0f) << 12UL
1096 | ((scm_t_wchar) utf8_buf[1] & 0x3f) << 6UL
1097 | (utf8_buf[2] & 0x3f);
1098 }
1099 else
1100 {
1101 assert (size == 4);
1102 codepoint = ((scm_t_wchar) utf8_buf[0] & 0x07) << 18UL
1103 | ((scm_t_wchar) utf8_buf[1] & 0x3f) << 12UL
1104 | ((scm_t_wchar) utf8_buf[2] & 0x3f) << 6UL
1105 | (utf8_buf[3] & 0x3f);
1106 }
1107
1108 return codepoint;
1109 }
1110
1111 /* Read a codepoint from PORT and return it. Fill BUF with the byte
1112 representation of the codepoint in PORT's encoding, and set *LEN to
1113 the length in bytes of that representation. Raise an error on
1114 failure. */
1115 static scm_t_wchar
1116 get_codepoint (SCM port, char buf[SCM_MBCHAR_BUF_SIZE], size_t *len)
1117 {
1118 int err, byte_read;
1119 size_t bytes_consumed, output_size;
1120 scm_t_wchar codepoint;
1121 char *output;
1122 scm_t_uint8 utf8_buf[SCM_MBCHAR_BUF_SIZE];
1123 scm_t_port *pt = SCM_PTAB_ENTRY (port);
1124
1125 if (SCM_UNLIKELY (pt->input_cd == (iconv_t) -1))
1126 /* Initialize the conversion descriptors. */
1127 scm_i_set_port_encoding_x (port, pt->encoding);
1128
1129 for (output_size = 0, output = (char *) utf8_buf,
1130 bytes_consumed = 0, err = 0;
1131 err == 0 && output_size == 0
1132 && (bytes_consumed == 0 || byte_read != EOF);
1133 bytes_consumed++)
1134 {
1135 char *input;
1136 size_t input_left, output_left, done;
1137
1138 byte_read = scm_get_byte_or_eof (port);
1139 if (byte_read == EOF)
1140 {
1141 if (bytes_consumed == 0)
1142 return (scm_t_wchar) EOF;
1143 else
1144 continue;
1145 }
1146
1147 buf[bytes_consumed] = byte_read;
1148
1149 input = buf;
1150 input_left = bytes_consumed + 1;
1151 output_left = sizeof (utf8_buf);
1152
1153 done = iconv (pt->input_cd, &input, &input_left,
1154 &output, &output_left);
1155 if (done == (size_t) -1)
1156 {
1157 err = errno;
1158 if (err == EINVAL)
1159 /* Missing input: keep trying. */
1160 err = 0;
1161 }
1162 else
1163 output_size = sizeof (utf8_buf) - output_left;
1164 }
1165
1166 if (err != 0)
1167 goto failure;
1168
1169 /* Convert the UTF8_BUF sequence to a Unicode code point. */
1170 codepoint = utf8_to_codepoint (utf8_buf, output_size);
1171 update_port_lf (codepoint, port);
1172
1173 *len = bytes_consumed;
1174
1175 return codepoint;
1176
1177 failure:
1178 {
1179 char *err_buf;
1180 SCM err_str = scm_i_make_string (bytes_consumed, &err_buf);
1181 memcpy (err_buf, buf, bytes_consumed);
1182
1183 if (err == EILSEQ)
1184 scm_misc_error (NULL, "input encoding error for ~s: ~s",
1185 scm_list_2 (scm_from_locale_string (scm_i_get_port_encoding (port)),
1186 err_str));
1187 else
1188 scm_misc_error (NULL, "input encoding error (invalid) for ~s: ~s\n",
1189 scm_list_2 (scm_from_locale_string (scm_i_get_port_encoding (port)),
1190 err_str));
1191 }
1192
1193 /* Never gets here. */
1194 return 0;
1195 }
1196
1197 /* Read a codepoint from PORT and return it. */
1198 scm_t_wchar
1199 scm_getc (SCM port)
1200 {
1201 size_t len;
1202 char buf[SCM_MBCHAR_BUF_SIZE];
1203
1204 return get_codepoint (port, buf, &len);
1205 }
1206
1207 /* this should only be called when the read buffer is empty. it
1208 tries to refill the read buffer. it returns the first char from
1209 the port, which is either EOF or *(pt->read_pos). */
1210 int
1211 scm_fill_input (SCM port)
1212 {
1213 scm_t_port *pt = SCM_PTAB_ENTRY (port);
1214
1215 assert (pt->read_pos == pt->read_end);
1216
1217 if (pt->read_buf == pt->putback_buf)
1218 {
1219 /* finished reading put-back chars. */
1220 pt->read_buf = pt->saved_read_buf;
1221 pt->read_pos = pt->saved_read_pos;
1222 pt->read_end = pt->saved_read_end;
1223 pt->read_buf_size = pt->saved_read_buf_size;
1224 if (pt->read_pos < pt->read_end)
1225 return *(pt->read_pos);
1226 }
1227 return scm_ptobs[SCM_PTOBNUM (port)].fill_input (port);
1228 }
1229
1230
1231 /* scm_lfwrite
1232 *
1233 * This function differs from scm_c_write; it updates port line and
1234 * column. */
1235
1236 void
1237 scm_lfwrite (const char *ptr, size_t size, SCM port)
1238 {
1239 scm_t_port *pt = SCM_PTAB_ENTRY (port);
1240 scm_t_ptob_descriptor *ptob = &scm_ptobs[SCM_PTOBNUM (port)];
1241
1242 if (pt->rw_active == SCM_PORT_READ)
1243 scm_end_input (port);
1244
1245 ptob->write (port, ptr, size);
1246
1247 for (; size; ptr++, size--)
1248 update_port_lf ((scm_t_wchar) (unsigned char) *ptr, port);
1249
1250 if (pt->rw_random)
1251 pt->rw_active = SCM_PORT_WRITE;
1252 }
1253
1254 /* Write STR to PORT from START inclusive to END exclusive. */
1255 void
1256 scm_lfwrite_substr (SCM str, size_t start, size_t end, SCM port)
1257 {
1258 scm_t_port *pt = SCM_PTAB_ENTRY (port);
1259
1260 if (pt->rw_active == SCM_PORT_READ)
1261 scm_end_input (port);
1262
1263 if (end == (size_t) -1)
1264 end = scm_i_string_length (str);
1265
1266 scm_display (scm_c_substring (str, start, end), port);
1267
1268 if (pt->rw_random)
1269 pt->rw_active = SCM_PORT_WRITE;
1270 }
1271
1272 /* scm_c_read
1273 *
1274 * Used by an application to read arbitrary number of bytes from an
1275 * SCM port. Same semantics as libc read, except that scm_c_read only
1276 * returns less than SIZE bytes if at end-of-file.
1277 *
1278 * Warning: Doesn't update port line and column counts! */
1279
1280 /* This structure, and the following swap_buffer function, are used
1281 for temporarily swapping a port's own read buffer, and the buffer
1282 that the caller of scm_c_read provides. */
1283 struct port_and_swap_buffer
1284 {
1285 scm_t_port *pt;
1286 unsigned char *buffer;
1287 size_t size;
1288 };
1289
1290 static void
1291 swap_buffer (void *data)
1292 {
1293 struct port_and_swap_buffer *psb = (struct port_and_swap_buffer *) data;
1294 unsigned char *old_buf = psb->pt->read_buf;
1295 size_t old_size = psb->pt->read_buf_size;
1296
1297 /* Make the port use (buffer, size) from the struct. */
1298 psb->pt->read_pos = psb->pt->read_buf = psb->pt->read_end = psb->buffer;
1299 psb->pt->read_buf_size = psb->size;
1300
1301 /* Save the port's old (buffer, size) in the struct. */
1302 psb->buffer = old_buf;
1303 psb->size = old_size;
1304 }
1305
1306 size_t
1307 scm_c_read (SCM port, void *buffer, size_t size)
1308 #define FUNC_NAME "scm_c_read"
1309 {
1310 scm_t_port *pt;
1311 size_t n_read = 0, n_available;
1312 struct port_and_swap_buffer psb;
1313
1314 SCM_VALIDATE_OPINPORT (1, port);
1315
1316 pt = SCM_PTAB_ENTRY (port);
1317 if (pt->rw_active == SCM_PORT_WRITE)
1318 scm_ptobs[SCM_PTOBNUM (port)].flush (port);
1319
1320 if (pt->rw_random)
1321 pt->rw_active = SCM_PORT_READ;
1322
1323 /* Take bytes first from the port's read buffer. */
1324 if (pt->read_pos < pt->read_end)
1325 {
1326 n_available = min (size, pt->read_end - pt->read_pos);
1327 memcpy (buffer, pt->read_pos, n_available);
1328 buffer = (char *) buffer + n_available;
1329 pt->read_pos += n_available;
1330 n_read += n_available;
1331 size -= n_available;
1332 }
1333
1334 /* Avoid the scm_dynwind_* costs if we now have enough data. */
1335 if (size == 0)
1336 return n_read;
1337
1338 /* Now we will call scm_fill_input repeatedly until we have read the
1339 requested number of bytes. (Note that a single scm_fill_input
1340 call does not guarantee to fill the whole of the port's read
1341 buffer.) */
1342 if (pt->read_buf_size <= 1 && pt->encoding == NULL)
1343 {
1344 /* The port that we are reading from is unbuffered - i.e. does
1345 not have its own persistent buffer - but we have a buffer,
1346 provided by our caller, that is the right size for the data
1347 that is wanted. For the following scm_fill_input calls,
1348 therefore, we use the buffer in hand as the port's read
1349 buffer.
1350
1351 We need to make sure that the port's normal (1 byte) buffer
1352 is reinstated in case one of the scm_fill_input () calls
1353 throws an exception; we use the scm_dynwind_* API to achieve
1354 that.
1355
1356 A consequence of this optimization is that the fill_input
1357 functions can't unget characters. That'll push data to the
1358 pushback buffer instead of this psb buffer. */
1359 #if SCM_DEBUG == 1
1360 unsigned char *pback = pt->putback_buf;
1361 #endif
1362 psb.pt = pt;
1363 psb.buffer = buffer;
1364 psb.size = size;
1365 scm_dynwind_begin (SCM_F_DYNWIND_REWINDABLE);
1366 scm_dynwind_rewind_handler (swap_buffer, &psb, SCM_F_WIND_EXPLICITLY);
1367 scm_dynwind_unwind_handler (swap_buffer, &psb, SCM_F_WIND_EXPLICITLY);
1368
1369 /* Call scm_fill_input until we have all the bytes that we need,
1370 or we hit EOF. */
1371 while (pt->read_buf_size && (scm_fill_input (port) != EOF))
1372 {
1373 pt->read_buf_size -= (pt->read_end - pt->read_pos);
1374 pt->read_pos = pt->read_buf = pt->read_end;
1375 }
1376 #if SCM_DEBUG == 1
1377 if (pback != pt->putback_buf
1378 || pt->read_buf - (unsigned char *) buffer < 0)
1379 scm_misc_error (FUNC_NAME,
1380 "scm_c_read must not call a fill function that pushes "
1381 "back characters onto an unbuffered port", SCM_EOL);
1382 #endif
1383 n_read += pt->read_buf - (unsigned char *) buffer;
1384
1385 /* Reinstate the port's normal buffer. */
1386 scm_dynwind_end ();
1387 }
1388 else
1389 {
1390 /* The port has its own buffer. It is important that we use it,
1391 even if it happens to be smaller than our caller's buffer, so
1392 that a custom port implementation's entry points (in
1393 particular, fill_input) can rely on the buffer always being
1394 the same as they first set up. */
1395 while (size && (scm_fill_input (port) != EOF))
1396 {
1397 n_available = min (size, pt->read_end - pt->read_pos);
1398 memcpy (buffer, pt->read_pos, n_available);
1399 buffer = (char *) buffer + n_available;
1400 pt->read_pos += n_available;
1401 n_read += n_available;
1402 size -= n_available;
1403 }
1404 }
1405
1406 return n_read;
1407 }
1408 #undef FUNC_NAME
1409
1410 /* scm_c_write
1411 *
1412 * Used by an application to write arbitrary number of bytes to an SCM
1413 * port. Similar semantics as libc write. However, unlike libc
1414 * write, scm_c_write writes the requested number of bytes and has no
1415 * return value.
1416 *
1417 * Warning: Doesn't update port line and column counts!
1418 */
1419
1420 void
1421 scm_c_write (SCM port, const void *ptr, size_t size)
1422 #define FUNC_NAME "scm_c_write"
1423 {
1424 scm_t_port *pt;
1425 scm_t_ptob_descriptor *ptob;
1426
1427 SCM_VALIDATE_OPOUTPORT (1, port);
1428
1429 pt = SCM_PTAB_ENTRY (port);
1430 ptob = &scm_ptobs[SCM_PTOBNUM (port)];
1431
1432 if (pt->rw_active == SCM_PORT_READ)
1433 scm_end_input (port);
1434
1435 ptob->write (port, ptr, size);
1436
1437 if (pt->rw_random)
1438 pt->rw_active = SCM_PORT_WRITE;
1439 }
1440 #undef FUNC_NAME
1441
1442 void
1443 scm_flush (SCM port)
1444 {
1445 long i = SCM_PTOBNUM (port);
1446 assert (i >= 0);
1447 (scm_ptobs[i].flush) (port);
1448 }
1449
1450 void
1451 scm_end_input (SCM port)
1452 {
1453 long offset;
1454 scm_t_port *pt = SCM_PTAB_ENTRY (port);
1455
1456 if (pt->read_buf == pt->putback_buf)
1457 {
1458 offset = pt->read_end - pt->read_pos;
1459 pt->read_buf = pt->saved_read_buf;
1460 pt->read_pos = pt->saved_read_pos;
1461 pt->read_end = pt->saved_read_end;
1462 pt->read_buf_size = pt->saved_read_buf_size;
1463 }
1464 else
1465 offset = 0;
1466
1467 scm_ptobs[SCM_PTOBNUM (port)].end_input (port, offset);
1468 }
1469
1470 \f
1471
1472
1473 void
1474 scm_unget_byte (int c, SCM port)
1475 #define FUNC_NAME "scm_unget_byte"
1476 {
1477 scm_t_port *pt = SCM_PTAB_ENTRY (port);
1478
1479 if (pt->read_buf == pt->putback_buf)
1480 /* already using the put-back buffer. */
1481 {
1482 /* enlarge putback_buf if necessary. */
1483 if (pt->read_end == pt->read_buf + pt->read_buf_size
1484 && pt->read_buf == pt->read_pos)
1485 {
1486 size_t new_size = pt->read_buf_size * 2;
1487 unsigned char *tmp = (unsigned char *)
1488 scm_gc_realloc (pt->putback_buf, pt->read_buf_size, new_size,
1489 "putback buffer");
1490
1491 pt->read_pos = pt->read_buf = pt->putback_buf = tmp;
1492 pt->read_end = pt->read_buf + pt->read_buf_size;
1493 pt->read_buf_size = pt->putback_buf_size = new_size;
1494 }
1495
1496 /* shift any existing bytes to buffer + 1. */
1497 if (pt->read_pos == pt->read_end)
1498 pt->read_end = pt->read_buf + 1;
1499 else if (pt->read_pos != pt->read_buf + 1)
1500 {
1501 int count = pt->read_end - pt->read_pos;
1502
1503 memmove (pt->read_buf + 1, pt->read_pos, count);
1504 pt->read_end = pt->read_buf + 1 + count;
1505 }
1506
1507 pt->read_pos = pt->read_buf;
1508 }
1509 else
1510 /* switch to the put-back buffer. */
1511 {
1512 if (pt->putback_buf == NULL)
1513 {
1514 pt->putback_buf
1515 = (unsigned char *) scm_gc_malloc_pointerless
1516 (SCM_INITIAL_PUTBACK_BUF_SIZE, "putback buffer");
1517 pt->putback_buf_size = SCM_INITIAL_PUTBACK_BUF_SIZE;
1518 }
1519
1520 pt->saved_read_buf = pt->read_buf;
1521 pt->saved_read_pos = pt->read_pos;
1522 pt->saved_read_end = pt->read_end;
1523 pt->saved_read_buf_size = pt->read_buf_size;
1524
1525 pt->read_pos = pt->read_buf = pt->putback_buf;
1526 pt->read_end = pt->read_buf + 1;
1527 pt->read_buf_size = pt->putback_buf_size;
1528 }
1529
1530 *pt->read_buf = c;
1531
1532 if (pt->rw_random)
1533 pt->rw_active = SCM_PORT_READ;
1534 }
1535 #undef FUNC_NAME
1536
1537 void
1538 scm_ungetc (scm_t_wchar c, SCM port)
1539 #define FUNC_NAME "scm_ungetc"
1540 {
1541 scm_t_port *pt = SCM_PTAB_ENTRY (port);
1542 char *result;
1543 char result_buf[10];
1544 const char *encoding;
1545 size_t len;
1546 int i;
1547
1548 if (pt->encoding != NULL)
1549 encoding = pt->encoding;
1550 else
1551 encoding = "ISO-8859-1";
1552
1553 len = sizeof (result_buf);
1554 result = u32_conv_to_encoding (encoding,
1555 (enum iconv_ilseq_handler) pt->ilseq_handler,
1556 (uint32_t *) &c, 1, NULL,
1557 result_buf, &len);
1558
1559 if (SCM_UNLIKELY (result == NULL || len == 0))
1560 {
1561 SCM chr;
1562
1563 chr = scm_integer_to_char (scm_from_uint32 (c));
1564 scm_encoding_error (FUNC_NAME, errno,
1565 "conversion to port encoding failed",
1566 "UTF-32", encoding,
1567 scm_string (scm_list_1 (chr)));
1568 }
1569
1570 for (i = len - 1; i >= 0; i--)
1571 scm_unget_byte (result[i], port);
1572
1573 if (SCM_UNLIKELY (result != result_buf))
1574 free (result);
1575
1576 if (c == '\n')
1577 {
1578 /* What should col be in this case?
1579 * We'll leave it at -1.
1580 */
1581 SCM_LINUM (port) -= 1;
1582 }
1583 else
1584 SCM_COL(port) -= 1;
1585 }
1586 #undef FUNC_NAME
1587
1588
1589 void
1590 scm_ungets (const char *s, int n, SCM port)
1591 {
1592 /* This is simple minded and inefficient, but unreading strings is
1593 * probably not a common operation, and remember that line and
1594 * column numbers have to be handled...
1595 *
1596 * Please feel free to write an optimized version!
1597 */
1598 while (n--)
1599 scm_ungetc (s[n], port);
1600 }
1601
1602
1603 SCM_DEFINE (scm_peek_char, "peek-char", 0, 1, 0,
1604 (SCM port),
1605 "Return the next character available from @var{port},\n"
1606 "@emph{without} updating @var{port} to point to the following\n"
1607 "character. If no more characters are available, the\n"
1608 "end-of-file object is returned.\n"
1609 "\n"
1610 "The value returned by\n"
1611 "a call to @code{peek-char} is the same as the value that would\n"
1612 "have been returned by a call to @code{read-char} on the same\n"
1613 "port. The only difference is that the very next call to\n"
1614 "@code{read-char} or @code{peek-char} on that @var{port} will\n"
1615 "return the value returned by the preceding call to\n"
1616 "@code{peek-char}. In particular, a call to @code{peek-char} on\n"
1617 "an interactive port will hang waiting for input whenever a call\n"
1618 "to @code{read-char} would have hung.")
1619 #define FUNC_NAME s_scm_peek_char
1620 {
1621 SCM result;
1622 scm_t_wchar c;
1623 char bytes[SCM_MBCHAR_BUF_SIZE];
1624 long column, line;
1625 size_t len;
1626
1627 if (SCM_UNBNDP (port))
1628 port = scm_current_input_port ();
1629 SCM_VALIDATE_OPINPORT (1, port);
1630
1631 column = SCM_COL (port);
1632 line = SCM_LINUM (port);
1633
1634 c = get_codepoint (port, bytes, &len);
1635 if (c == EOF)
1636 result = SCM_EOF_VAL;
1637 else
1638 {
1639 long i;
1640
1641 result = SCM_MAKE_CHAR (c);
1642
1643 for (i = len - 1; i >= 0; i--)
1644 scm_unget_byte (bytes[i], port);
1645
1646 SCM_COL (port) = column;
1647 SCM_LINUM (port) = line;
1648 }
1649
1650 return result;
1651 }
1652 #undef FUNC_NAME
1653
1654 SCM_DEFINE (scm_unread_char, "unread-char", 1, 1, 0,
1655 (SCM cobj, SCM port),
1656 "Place @var{char} in @var{port} so that it will be read by the\n"
1657 "next read operation. If called multiple times, the unread characters\n"
1658 "will be read again in last-in first-out order. If @var{port} is\n"
1659 "not supplied, the current input port is used.")
1660 #define FUNC_NAME s_scm_unread_char
1661 {
1662 int c;
1663
1664 SCM_VALIDATE_CHAR (1, cobj);
1665 if (SCM_UNBNDP (port))
1666 port = scm_current_input_port ();
1667 SCM_VALIDATE_OPINPORT (2, port);
1668
1669 c = SCM_CHAR (cobj);
1670
1671 scm_ungetc (c, port);
1672 return cobj;
1673 }
1674 #undef FUNC_NAME
1675
1676 SCM_DEFINE (scm_unread_string, "unread-string", 2, 0, 0,
1677 (SCM str, SCM port),
1678 "Place the string @var{str} in @var{port} so that its characters will be\n"
1679 "read in subsequent read operations. If called multiple times, the\n"
1680 "unread characters will be read again in last-in first-out order. If\n"
1681 "@var{port} is not supplied, the current-input-port is used.")
1682 #define FUNC_NAME s_scm_unread_string
1683 {
1684 int n;
1685 SCM_VALIDATE_STRING (1, str);
1686 if (SCM_UNBNDP (port))
1687 port = scm_current_input_port ();
1688 SCM_VALIDATE_OPINPORT (2, port);
1689
1690 n = scm_i_string_length (str);
1691
1692 while (n--)
1693 scm_ungetc (scm_i_string_ref (str, n), port);
1694
1695 return str;
1696 }
1697 #undef FUNC_NAME
1698
1699 SCM_DEFINE (scm_seek, "seek", 3, 0, 0,
1700 (SCM fd_port, SCM offset, SCM whence),
1701 "Sets the current position of @var{fd/port} to the integer\n"
1702 "@var{offset}, which is interpreted according to the value of\n"
1703 "@var{whence}.\n"
1704 "\n"
1705 "One of the following variables should be supplied for\n"
1706 "@var{whence}:\n"
1707 "@defvar SEEK_SET\n"
1708 "Seek from the beginning of the file.\n"
1709 "@end defvar\n"
1710 "@defvar SEEK_CUR\n"
1711 "Seek from the current position.\n"
1712 "@end defvar\n"
1713 "@defvar SEEK_END\n"
1714 "Seek from the end of the file.\n"
1715 "@end defvar\n"
1716 "If @var{fd/port} is a file descriptor, the underlying system\n"
1717 "call is @code{lseek}. @var{port} may be a string port.\n"
1718 "\n"
1719 "The value returned is the new position in the file. This means\n"
1720 "that the current position of a port can be obtained using:\n"
1721 "@lisp\n"
1722 "(seek port 0 SEEK_CUR)\n"
1723 "@end lisp")
1724 #define FUNC_NAME s_scm_seek
1725 {
1726 int how;
1727
1728 fd_port = SCM_COERCE_OUTPORT (fd_port);
1729
1730 how = scm_to_int (whence);
1731 if (how != SEEK_SET && how != SEEK_CUR && how != SEEK_END)
1732 SCM_OUT_OF_RANGE (3, whence);
1733
1734 if (SCM_OPPORTP (fd_port))
1735 {
1736 scm_t_ptob_descriptor *ptob = scm_ptobs + SCM_PTOBNUM (fd_port);
1737 off_t_or_off64_t off = scm_to_off_t_or_off64_t (offset);
1738 off_t_or_off64_t rv;
1739
1740 if (!ptob->seek)
1741 SCM_MISC_ERROR ("port is not seekable",
1742 scm_cons (fd_port, SCM_EOL));
1743 else
1744 rv = ptob->seek (fd_port, off, how);
1745 return scm_from_off_t_or_off64_t (rv);
1746 }
1747 else /* file descriptor?. */
1748 {
1749 off_t_or_off64_t off = scm_to_off_t_or_off64_t (offset);
1750 off_t_or_off64_t rv;
1751 rv = lseek_or_lseek64 (scm_to_int (fd_port), off, how);
1752 if (rv == -1)
1753 SCM_SYSERROR;
1754 return scm_from_off_t_or_off64_t (rv);
1755 }
1756 }
1757 #undef FUNC_NAME
1758
1759 #ifndef O_BINARY
1760 #define O_BINARY 0
1761 #endif
1762
1763 /* Mingw has ftruncate(), perhaps implemented above using chsize, but
1764 doesn't have the filename version truncate(), hence this code. */
1765 #if HAVE_FTRUNCATE && ! HAVE_TRUNCATE
1766 static int
1767 truncate (const char *file, off_t length)
1768 {
1769 int ret, fdes;
1770
1771 fdes = open (file, O_BINARY | O_WRONLY);
1772 if (fdes == -1)
1773 return -1;
1774
1775 ret = ftruncate (fdes, length);
1776 if (ret == -1)
1777 {
1778 int save_errno = errno;
1779 close (fdes);
1780 errno = save_errno;
1781 return -1;
1782 }
1783
1784 return close (fdes);
1785 }
1786 #endif /* HAVE_FTRUNCATE && ! HAVE_TRUNCATE */
1787
1788 SCM_DEFINE (scm_truncate_file, "truncate-file", 1, 1, 0,
1789 (SCM object, SCM length),
1790 "Truncate @var{file} to @var{length} bytes. @var{file} can be a\n"
1791 "filename string, a port object, or an integer file descriptor.\n"
1792 "The return value is unspecified.\n"
1793 "\n"
1794 "For a port or file descriptor @var{length} can be omitted, in\n"
1795 "which case the file is truncated at the current position (per\n"
1796 "@code{ftell} above).\n"
1797 "\n"
1798 "On most systems a file can be extended by giving a length\n"
1799 "greater than the current size, but this is not mandatory in the\n"
1800 "POSIX standard.")
1801 #define FUNC_NAME s_scm_truncate_file
1802 {
1803 int rv;
1804
1805 /* "object" can be a port, fdes or filename.
1806
1807 Negative "length" makes no sense, but it's left to truncate() or
1808 ftruncate() to give back an error for that (normally EINVAL).
1809 */
1810
1811 if (SCM_UNBNDP (length))
1812 {
1813 /* must supply length if object is a filename. */
1814 if (scm_is_string (object))
1815 SCM_MISC_ERROR("must supply length if OBJECT is a filename", SCM_EOL);
1816
1817 length = scm_seek (object, SCM_INUM0, scm_from_int (SEEK_CUR));
1818 }
1819
1820 object = SCM_COERCE_OUTPORT (object);
1821 if (scm_is_integer (object))
1822 {
1823 off_t_or_off64_t c_length = scm_to_off_t_or_off64_t (length);
1824 SCM_SYSCALL (rv = ftruncate_or_ftruncate64 (scm_to_int (object),
1825 c_length));
1826 }
1827 else if (SCM_OPOUTPORTP (object))
1828 {
1829 off_t_or_off64_t c_length = scm_to_off_t_or_off64_t (length);
1830 scm_t_port *pt = SCM_PTAB_ENTRY (object);
1831 scm_t_ptob_descriptor *ptob = scm_ptobs + SCM_PTOBNUM (object);
1832
1833 if (!ptob->truncate)
1834 SCM_MISC_ERROR ("port is not truncatable", SCM_EOL);
1835 if (pt->rw_active == SCM_PORT_READ)
1836 scm_end_input (object);
1837 else if (pt->rw_active == SCM_PORT_WRITE)
1838 ptob->flush (object);
1839
1840 ptob->truncate (object, c_length);
1841 rv = 0;
1842 }
1843 else
1844 {
1845 off_t_or_off64_t c_length = scm_to_off_t_or_off64_t (length);
1846 char *str = scm_to_locale_string (object);
1847 int eno;
1848 SCM_SYSCALL (rv = truncate_or_truncate64 (str, c_length));
1849 eno = errno;
1850 free (str);
1851 errno = eno;
1852 }
1853 if (rv == -1)
1854 SCM_SYSERROR;
1855 return SCM_UNSPECIFIED;
1856 }
1857 #undef FUNC_NAME
1858
1859 SCM_DEFINE (scm_port_line, "port-line", 1, 0, 0,
1860 (SCM port),
1861 "Return the current line number for @var{port}.\n"
1862 "\n"
1863 "The first line of a file is 0. But you might want to add 1\n"
1864 "when printing line numbers, since starting from 1 is\n"
1865 "traditional in error messages, and likely to be more natural to\n"
1866 "non-programmers.")
1867 #define FUNC_NAME s_scm_port_line
1868 {
1869 port = SCM_COERCE_OUTPORT (port);
1870 SCM_VALIDATE_OPENPORT (1, port);
1871 return scm_from_long (SCM_LINUM (port));
1872 }
1873 #undef FUNC_NAME
1874
1875 SCM_DEFINE (scm_set_port_line_x, "set-port-line!", 2, 0, 0,
1876 (SCM port, SCM line),
1877 "Set the current line number for @var{port} to @var{line}. The\n"
1878 "first line of a file is 0.")
1879 #define FUNC_NAME s_scm_set_port_line_x
1880 {
1881 port = SCM_COERCE_OUTPORT (port);
1882 SCM_VALIDATE_OPENPORT (1, port);
1883 SCM_PTAB_ENTRY (port)->line_number = scm_to_long (line);
1884 return SCM_UNSPECIFIED;
1885 }
1886 #undef FUNC_NAME
1887
1888 SCM_DEFINE (scm_port_column, "port-column", 1, 0, 0,
1889 (SCM port),
1890 "Return the current column number of @var{port}.\n"
1891 "If the number is\n"
1892 "unknown, the result is #f. Otherwise, the result is a 0-origin integer\n"
1893 "- i.e. the first character of the first line is line 0, column 0.\n"
1894 "(However, when you display a file position, for example in an error\n"
1895 "message, we recommend you add 1 to get 1-origin integers. This is\n"
1896 "because lines and column numbers traditionally start with 1, and that is\n"
1897 "what non-programmers will find most natural.)")
1898 #define FUNC_NAME s_scm_port_column
1899 {
1900 port = SCM_COERCE_OUTPORT (port);
1901 SCM_VALIDATE_OPENPORT (1, port);
1902 return scm_from_int (SCM_COL (port));
1903 }
1904 #undef FUNC_NAME
1905
1906 SCM_DEFINE (scm_set_port_column_x, "set-port-column!", 2, 0, 0,
1907 (SCM port, SCM column),
1908 "Set the current column of @var{port}. Before reading the first\n"
1909 "character on a line the column should be 0.")
1910 #define FUNC_NAME s_scm_set_port_column_x
1911 {
1912 port = SCM_COERCE_OUTPORT (port);
1913 SCM_VALIDATE_OPENPORT (1, port);
1914 SCM_PTAB_ENTRY (port)->column_number = scm_to_int (column);
1915 return SCM_UNSPECIFIED;
1916 }
1917 #undef FUNC_NAME
1918
1919 SCM_DEFINE (scm_port_filename, "port-filename", 1, 0, 0,
1920 (SCM port),
1921 "Return the filename associated with @var{port}. This function returns\n"
1922 "the strings \"standard input\", \"standard output\" and \"standard error\"\n"
1923 "when called on the current input, output and error ports respectively.")
1924 #define FUNC_NAME s_scm_port_filename
1925 {
1926 port = SCM_COERCE_OUTPORT (port);
1927 SCM_VALIDATE_OPENPORT (1, port);
1928 return SCM_FILENAME (port);
1929 }
1930 #undef FUNC_NAME
1931
1932 SCM_DEFINE (scm_set_port_filename_x, "set-port-filename!", 2, 0, 0,
1933 (SCM port, SCM filename),
1934 "Change the filename associated with @var{port}, using the current input\n"
1935 "port if none is specified. Note that this does not change the port's\n"
1936 "source of data, but only the value that is returned by\n"
1937 "@code{port-filename} and reported in diagnostic output.")
1938 #define FUNC_NAME s_scm_set_port_filename_x
1939 {
1940 port = SCM_COERCE_OUTPORT (port);
1941 SCM_VALIDATE_OPENPORT (1, port);
1942 /* We allow the user to set the filename to whatever he likes. */
1943 SCM_SET_FILENAME (port, filename);
1944 return SCM_UNSPECIFIED;
1945 }
1946 #undef FUNC_NAME
1947
1948 /* A fluid specifying the default encoding for newly created ports. If it is
1949 a string, that is the encoding. If it is #f, it is in the "native"
1950 (Latin-1) encoding. */
1951 SCM_VARIABLE (default_port_encoding_var, "%default-port-encoding");
1952
1953 static int scm_port_encoding_init = 0;
1954
1955 /* Return a C string representation of the current encoding. */
1956 const char *
1957 scm_i_get_port_encoding (SCM port)
1958 {
1959 SCM encoding;
1960
1961 if (scm_is_false (port))
1962 {
1963 if (!scm_port_encoding_init)
1964 return NULL;
1965 else if (!scm_is_fluid (SCM_VARIABLE_REF (default_port_encoding_var)))
1966 return NULL;
1967 else
1968 {
1969 encoding = scm_fluid_ref (SCM_VARIABLE_REF (default_port_encoding_var));
1970 if (!scm_is_string (encoding))
1971 return NULL;
1972 else
1973 return scm_i_string_chars (encoding);
1974 }
1975 }
1976 else
1977 {
1978 scm_t_port *pt;
1979 pt = SCM_PTAB_ENTRY (port);
1980 if (pt->encoding)
1981 return pt->encoding;
1982 else
1983 return NULL;
1984 }
1985 }
1986
1987 /* Returns ENC if it is a recognized encoding. If it isn't, it tries
1988 to find an alias of ENC that is valid. Otherwise, it returns
1989 NULL. */
1990 static const char *
1991 find_valid_encoding (const char *enc)
1992 {
1993 int isvalid = 0;
1994 const char str[] = " ";
1995 scm_t_uint32 result_buf;
1996 scm_t_uint32 *u32;
1997 size_t u32len;
1998
1999 u32len = sizeof (result_buf) / sizeof (scm_t_uint32);
2000 u32 = u32_conv_from_encoding (enc, iconveh_error, str, 1,
2001 NULL, &result_buf, &u32len);
2002 isvalid = (u32 != NULL);
2003
2004 if (SCM_UNLIKELY (u32 != &result_buf))
2005 free (u32);
2006
2007 if (isvalid)
2008 return enc;
2009
2010 return NULL;
2011 }
2012
2013 void
2014 scm_i_set_port_encoding_x (SCM port, const char *enc)
2015 {
2016 const char *valid_enc;
2017 scm_t_port *pt;
2018
2019 /* Null is shorthand for the native, Latin-1 encoding. */
2020 if (enc == NULL)
2021 valid_enc = NULL;
2022 else
2023 {
2024 valid_enc = find_valid_encoding (enc);
2025 if (valid_enc == NULL)
2026 goto invalid_encoding;
2027 }
2028
2029 if (scm_is_false (port))
2030 {
2031 /* Set the default encoding for future ports. */
2032 if (!scm_port_encoding_init
2033 || !scm_is_fluid (SCM_VARIABLE_REF (default_port_encoding_var)))
2034 scm_misc_error (NULL, "tried to set port encoding fluid before it is initialized",
2035 SCM_EOL);
2036
2037 if (valid_enc == NULL
2038 || !strcmp (valid_enc, "ASCII")
2039 || !strcmp (valid_enc, "ANSI_X3.4-1968")
2040 || !strcmp (valid_enc, "ISO-8859-1"))
2041 scm_fluid_set_x (SCM_VARIABLE_REF (default_port_encoding_var), SCM_BOOL_F);
2042 else
2043 scm_fluid_set_x (SCM_VARIABLE_REF (default_port_encoding_var),
2044 scm_from_locale_string (valid_enc));
2045 }
2046 else
2047 {
2048 iconv_t new_input_cd, new_output_cd;
2049
2050 new_input_cd = (iconv_t) -1;
2051 new_output_cd = (iconv_t) -1;
2052
2053 /* Set the character encoding for this port. */
2054 pt = SCM_PTAB_ENTRY (port);
2055
2056 if (valid_enc == NULL)
2057 valid_enc = "ISO-8859-1";
2058
2059 pt->encoding = scm_gc_strdup (valid_enc, "port");
2060
2061 if (SCM_CELL_WORD_0 (port) & SCM_RDNG)
2062 {
2063 /* Open an input iconv conversion descriptor, from VALID_ENC
2064 to UTF-8. We choose UTF-8, not UTF-32, because iconv
2065 implementations can typically convert from anything to
2066 UTF-8, but not to UTF-32 (see
2067 <http://lists.gnu.org/archive/html/bug-libunistring/2010-09/msg00007.html>). */
2068 new_input_cd = iconv_open ("UTF-8", valid_enc);
2069 if (new_input_cd == (iconv_t) -1)
2070 goto invalid_encoding;
2071 }
2072
2073 if (SCM_CELL_WORD_0 (port) & SCM_WRTNG)
2074 {
2075 new_output_cd = iconv_open (valid_enc, "UTF-8");
2076 if (new_output_cd == (iconv_t) -1)
2077 {
2078 if (new_input_cd != (iconv_t) -1)
2079 iconv_close (new_input_cd);
2080 goto invalid_encoding;
2081 }
2082 }
2083
2084 if (pt->input_cd != (iconv_t) -1)
2085 iconv_close (pt->input_cd);
2086 if (pt->output_cd != (iconv_t) -1)
2087 iconv_close (pt->output_cd);
2088
2089 pt->input_cd = new_input_cd;
2090 pt->output_cd = new_output_cd;
2091 }
2092
2093 return;
2094
2095 invalid_encoding:
2096 {
2097 SCM err;
2098 err = scm_from_locale_string (enc);
2099 scm_misc_error (NULL, "invalid or unknown character encoding ~s",
2100 scm_list_1 (err));
2101 }
2102 }
2103
2104 SCM_DEFINE (scm_port_encoding, "port-encoding", 1, 0, 0,
2105 (SCM port),
2106 "Returns, as a string, the character encoding that @var{port}\n"
2107 "uses to interpret its input and output.\n")
2108 #define FUNC_NAME s_scm_port_encoding
2109 {
2110 scm_t_port *pt;
2111 const char *enc;
2112
2113 SCM_VALIDATE_PORT (1, port);
2114
2115 pt = SCM_PTAB_ENTRY (port);
2116 enc = scm_i_get_port_encoding (port);
2117 if (enc)
2118 return scm_from_locale_string (pt->encoding);
2119 else
2120 return SCM_BOOL_F;
2121 }
2122 #undef FUNC_NAME
2123
2124 SCM_DEFINE (scm_set_port_encoding_x, "set-port-encoding!", 2, 0, 0,
2125 (SCM port, SCM enc),
2126 "Sets the character encoding that will be used to interpret all\n"
2127 "port I/O. New ports are created with the encoding\n"
2128 "appropriate for the current locale if @code{setlocale} has \n"
2129 "been called or ISO-8859-1 otherwise\n"
2130 "and this procedure can be used to modify that encoding.\n")
2131 #define FUNC_NAME s_scm_set_port_encoding_x
2132 {
2133 char *enc_str;
2134 const char *valid_enc_str;
2135
2136 SCM_VALIDATE_PORT (1, port);
2137 SCM_VALIDATE_STRING (2, enc);
2138
2139 enc_str = scm_to_locale_string (enc);
2140 valid_enc_str = find_valid_encoding (enc_str);
2141 if (valid_enc_str == NULL)
2142 {
2143 free (enc_str);
2144 scm_misc_error (FUNC_NAME, "invalid or unknown character encoding ~s",
2145 scm_list_1 (enc));
2146 }
2147 else
2148 {
2149 scm_i_set_port_encoding_x (port, valid_enc_str);
2150 free (enc_str);
2151 }
2152 return SCM_UNSPECIFIED;
2153 }
2154 #undef FUNC_NAME
2155
2156
2157 /* This determines how conversions handle unconvertible characters. */
2158 SCM_GLOBAL_VARIABLE (scm_conversion_strategy, "%port-conversion-strategy");
2159 static int scm_conversion_strategy_init = 0;
2160
2161 scm_t_string_failed_conversion_handler
2162 scm_i_get_conversion_strategy (SCM port)
2163 {
2164 SCM encoding;
2165
2166 if (scm_is_false (port))
2167 {
2168 if (!scm_conversion_strategy_init
2169 || !scm_is_fluid (SCM_VARIABLE_REF (scm_conversion_strategy)))
2170 return SCM_FAILED_CONVERSION_QUESTION_MARK;
2171 else
2172 {
2173 encoding = scm_fluid_ref (SCM_VARIABLE_REF (scm_conversion_strategy));
2174 if (scm_is_false (encoding))
2175 return SCM_FAILED_CONVERSION_QUESTION_MARK;
2176 else
2177 return (scm_t_string_failed_conversion_handler) scm_to_int (encoding);
2178 }
2179 }
2180 else
2181 {
2182 scm_t_port *pt;
2183 pt = SCM_PTAB_ENTRY (port);
2184 return pt->ilseq_handler;
2185 }
2186
2187 }
2188
2189 void
2190 scm_i_set_conversion_strategy_x (SCM port,
2191 scm_t_string_failed_conversion_handler handler)
2192 {
2193 SCM strategy;
2194 scm_t_port *pt;
2195
2196 strategy = scm_from_int ((int) handler);
2197
2198 if (scm_is_false (port))
2199 {
2200 /* Set the default encoding for future ports. */
2201 if (!scm_conversion_strategy
2202 || !scm_is_fluid (SCM_VARIABLE_REF (scm_conversion_strategy)))
2203 scm_misc_error (NULL, "tried to set conversion strategy fluid before it is initialized",
2204 SCM_EOL);
2205 scm_fluid_set_x (SCM_VARIABLE_REF (scm_conversion_strategy), strategy);
2206 }
2207 else
2208 {
2209 /* Set the character encoding for this port. */
2210 pt = SCM_PTAB_ENTRY (port);
2211 pt->ilseq_handler = handler;
2212 }
2213 }
2214
2215 SCM_DEFINE (scm_port_conversion_strategy, "port-conversion-strategy",
2216 1, 0, 0, (SCM port),
2217 "Returns the behavior of the port when handling a character that\n"
2218 "is not representable in the port's current encoding.\n"
2219 "It returns the symbol @code{error} if unrepresentable characters\n"
2220 "should cause exceptions, @code{substitute} if the port should\n"
2221 "try to replace unrepresentable characters with question marks or\n"
2222 "approximate characters, or @code{escape} if unrepresentable\n"
2223 "characters should be converted to string escapes.\n"
2224 "\n"
2225 "If @var{port} is @code{#f}, then the current default behavior\n"
2226 "will be returned. New ports will have this default behavior\n"
2227 "when they are created.\n")
2228 #define FUNC_NAME s_scm_port_conversion_strategy
2229 {
2230 scm_t_string_failed_conversion_handler h;
2231
2232 SCM_VALIDATE_OPPORT (1, port);
2233
2234 if (!scm_is_false (port))
2235 {
2236 SCM_VALIDATE_OPPORT (1, port);
2237 }
2238
2239 h = scm_i_get_conversion_strategy (port);
2240 if (h == SCM_FAILED_CONVERSION_ERROR)
2241 return scm_from_latin1_symbol ("error");
2242 else if (h == SCM_FAILED_CONVERSION_QUESTION_MARK)
2243 return scm_from_latin1_symbol ("substitute");
2244 else if (h == SCM_FAILED_CONVERSION_ESCAPE_SEQUENCE)
2245 return scm_from_latin1_symbol ("escape");
2246 else
2247 abort ();
2248
2249 /* Never gets here. */
2250 return SCM_UNDEFINED;
2251 }
2252 #undef FUNC_NAME
2253
2254 SCM_DEFINE (scm_set_port_conversion_strategy_x, "set-port-conversion-strategy!",
2255 2, 0, 0,
2256 (SCM port, SCM sym),
2257 "Sets the behavior of the interpreter when outputting a character\n"
2258 "that is not representable in the port's current encoding.\n"
2259 "@var{sym} can be either @code{'error}, @code{'substitute}, or\n"
2260 "@code{'escape}. If it is @code{'error}, an error will be thrown\n"
2261 "when an unconvertible character is encountered. If it is\n"
2262 "@code{'substitute}, then unconvertible characters will \n"
2263 "be replaced with approximate characters, or with question marks\n"
2264 "if no approximately correct character is available.\n"
2265 "If it is @code{'escape},\n"
2266 "it will appear as a hex escape when output.\n"
2267 "\n"
2268 "If @var{port} is an open port, the conversion error behavior\n"
2269 "is set for that port. If it is @code{#f}, it is set as the\n"
2270 "default behavior for any future ports that get created in\n"
2271 "this thread.\n")
2272 #define FUNC_NAME s_scm_set_port_conversion_strategy_x
2273 {
2274 SCM err;
2275 SCM qm;
2276 SCM esc;
2277
2278 if (!scm_is_false (port))
2279 {
2280 SCM_VALIDATE_OPPORT (1, port);
2281 }
2282
2283 err = scm_from_latin1_symbol ("error");
2284 if (scm_is_true (scm_eqv_p (sym, err)))
2285 {
2286 scm_i_set_conversion_strategy_x (port, SCM_FAILED_CONVERSION_ERROR);
2287 return SCM_UNSPECIFIED;
2288 }
2289
2290 qm = scm_from_latin1_symbol ("substitute");
2291 if (scm_is_true (scm_eqv_p (sym, qm)))
2292 {
2293 scm_i_set_conversion_strategy_x (port,
2294 SCM_FAILED_CONVERSION_QUESTION_MARK);
2295 return SCM_UNSPECIFIED;
2296 }
2297
2298 esc = scm_from_latin1_symbol ("escape");
2299 if (scm_is_true (scm_eqv_p (sym, esc)))
2300 {
2301 scm_i_set_conversion_strategy_x (port,
2302 SCM_FAILED_CONVERSION_ESCAPE_SEQUENCE);
2303 return SCM_UNSPECIFIED;
2304 }
2305
2306 SCM_MISC_ERROR ("unknown conversion behavior ~s", scm_list_1 (sym));
2307
2308 return SCM_UNSPECIFIED;
2309 }
2310 #undef FUNC_NAME
2311
2312
2313
2314 void
2315 scm_print_port_mode (SCM exp, SCM port)
2316 {
2317 scm_puts (SCM_CLOSEDP (exp)
2318 ? "closed: "
2319 : (SCM_RDNG & SCM_CELL_WORD_0 (exp)
2320 ? (SCM_WRTNG & SCM_CELL_WORD_0 (exp)
2321 ? "input-output: "
2322 : "input: ")
2323 : (SCM_WRTNG & SCM_CELL_WORD_0 (exp)
2324 ? "output: "
2325 : "bogus: ")),
2326 port);
2327 }
2328
2329 int
2330 scm_port_print (SCM exp, SCM port, scm_print_state *pstate SCM_UNUSED)
2331 {
2332 char *type = SCM_PTOBNAME (SCM_PTOBNUM (exp));
2333 if (!type)
2334 type = "port";
2335 scm_puts ("#<", port);
2336 scm_print_port_mode (exp, port);
2337 scm_puts (type, port);
2338 scm_putc (' ', port);
2339 scm_uintprint (SCM_CELL_WORD_1 (exp), 16, port);
2340 scm_putc ('>', port);
2341 return 1;
2342 }
2343
2344 \f
2345
2346 /* Void ports. */
2347
2348 scm_t_bits scm_tc16_void_port = 0;
2349
2350 static int fill_input_void_port (SCM port SCM_UNUSED)
2351 {
2352 return EOF;
2353 }
2354
2355 static void
2356 write_void_port (SCM port SCM_UNUSED,
2357 const void *data SCM_UNUSED,
2358 size_t size SCM_UNUSED)
2359 {
2360 }
2361
2362 static SCM
2363 scm_i_void_port (long mode_bits)
2364 {
2365 scm_i_scm_pthread_mutex_lock (&scm_i_port_table_mutex);
2366 {
2367 SCM answer = scm_new_port_table_entry (scm_tc16_void_port);
2368 scm_t_port * pt = SCM_PTAB_ENTRY(answer);
2369
2370 scm_port_non_buffer (pt);
2371
2372 SCM_SETSTREAM (answer, 0);
2373 SCM_SET_CELL_TYPE (answer, scm_tc16_void_port | mode_bits);
2374 scm_i_pthread_mutex_unlock (&scm_i_port_table_mutex);
2375 return answer;
2376 }
2377 }
2378
2379 SCM
2380 scm_void_port (char *mode_str)
2381 {
2382 return scm_i_void_port (scm_mode_bits (mode_str));
2383 }
2384
2385 SCM_DEFINE (scm_sys_make_void_port, "%make-void-port", 1, 0, 0,
2386 (SCM mode),
2387 "Create and return a new void port. A void port acts like\n"
2388 "@file{/dev/null}. The @var{mode} argument\n"
2389 "specifies the input/output modes for this port: see the\n"
2390 "documentation for @code{open-file} in @ref{File Ports}.")
2391 #define FUNC_NAME s_scm_sys_make_void_port
2392 {
2393 return scm_i_void_port (scm_i_mode_bits (mode));
2394 }
2395 #undef FUNC_NAME
2396
2397 \f
2398 /* Initialization. */
2399
2400 void
2401 scm_init_ports ()
2402 {
2403 /* lseek() symbols. */
2404 scm_c_define ("SEEK_SET", scm_from_int (SEEK_SET));
2405 scm_c_define ("SEEK_CUR", scm_from_int (SEEK_CUR));
2406 scm_c_define ("SEEK_END", scm_from_int (SEEK_END));
2407
2408 scm_tc16_void_port = scm_make_port_type ("void", fill_input_void_port,
2409 write_void_port);
2410
2411 cur_inport_fluid = scm_make_fluid ();
2412 cur_outport_fluid = scm_make_fluid ();
2413 cur_errport_fluid = scm_make_fluid ();
2414 cur_loadport_fluid = scm_make_fluid ();
2415
2416 scm_i_port_weak_hash = scm_make_weak_key_hash_table (SCM_I_MAKINUM(31));
2417
2418 #include "libguile/ports.x"
2419
2420 /* Use Latin-1 as the default port encoding. */
2421 SCM_VARIABLE_SET (default_port_encoding_var, scm_make_fluid ());
2422 scm_fluid_set_x (SCM_VARIABLE_REF (default_port_encoding_var), SCM_BOOL_F);
2423 scm_port_encoding_init = 1;
2424
2425 SCM_VARIABLE_SET (scm_conversion_strategy, scm_make_fluid ());
2426 scm_fluid_set_x (SCM_VARIABLE_REF (scm_conversion_strategy),
2427 scm_from_int ((int) SCM_FAILED_CONVERSION_QUESTION_MARK));
2428 scm_conversion_strategy_init = 1;
2429
2430 }
2431
2432 /*
2433 Local Variables:
2434 c-file-style: "gnu"
2435 End:
2436 */