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