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