1 /**********************************************************************
6 created at: Fri Oct 15 18:08:59 JST 1993
8 Copyright (C) 1993-2007 Yukihiro Matsumoto
9 Copyright (C) 2000 Network Applied Communication Laboratory, Inc.
10 Copyright (C) 2000 Information-technology Promotion Agency, Japan
12 **********************************************************************/
14 #include "ruby/ruby.h"
16 #include "ruby/signal.h"
21 #define free(x) xfree(x)
23 #if defined(DOSISH) || defined(__CYGWIN__)
27 #include <sys/types.h>
28 #if defined HAVE_NET_SOCKET_H
29 # include <net/socket.h>
30 #elif defined HAVE_SYS_SOCKET_H
31 # include <sys/socket.h>
34 #if defined(MSDOS) || defined(__BOW__) || defined(__CYGWIN__) || defined(_WIN32) || defined(__human68k__) || defined(__EMX__) || defined(__BEOS__)
35 # define NO_SAFE_RENAME
38 #if defined(MSDOS) || defined(__CYGWIN__) || defined(_WIN32)
39 # define NO_LONG_FNAME
42 #if defined(__FreeBSD__) || defined(__NetBSD__) || defined(__OpenBSD__) || defined(__DragonFly__) || defined(sun) || defined(_nec_ews)
50 #include <sys/types.h>
51 #if defined(HAVE_SYS_IOCTL_H) && !defined(DJGPP) && !defined(_WIN32) && !defined(__human68k__)
52 #include <sys/ioctl.h>
54 #if defined(HAVE_FCNTL_H) || defined(_WIN32)
56 #elif defined(HAVE_SYS_FCNTL_H)
57 #include <sys/fcntl.h>
60 #if !HAVE_OFF_T && !defined(off_t)
66 /* EMX has sys/param.h, but.. */
67 #if defined(HAVE_SYS_PARAM_H) && !(defined(__EMX__) || defined(__HIUX_MPP__))
68 # include <sys/param.h>
81 #elif defined HAVE_SYS_SYSCALL_H
82 #include <sys/syscall.h>
85 extern void Init_File(void);
89 # define NOFILE (OPEN_MAX)
93 #include "ruby/util.h"
96 #define O_ACCMODE (O_RDONLY | O_WRONLY | O_RDWR)
99 #if SIZEOF_OFF_T > SIZEOF_LONG && !defined(HAVE_LONG_LONG)
100 # error off_t is bigger than long, but you have no long long...
104 # ifdef _POSIX_PIPE_BUF
105 # define PIPE_BUF _POSIX_PIPE_BUF
107 # define PIPE_BUF 512 /* is this ok? */
115 VALUE rb_stdin
, rb_stdout
, rb_stderr
;
116 VALUE rb_deferr
; /* rescue VIM plugin */
117 static VALUE orig_stdout
, orig_stderr
;
126 static ID id_write
, id_read
, id_getc
, id_flush
, id_encode
, id_readpartial
;
127 static VALUE sym_mode
, sym_perm
, sym_extenc
, sym_intenc
, sym_encoding
, sym_open_args
;
129 struct timeval
rb_time_interval(VALUE
);
132 VALUE filename
, current_file
;
139 rb_encoding
*enc
, *enc2
;
142 static int max_file_descriptor
= NOFILE
;
143 #define UPDATE_MAXFD(fd) \
145 if (max_file_descriptor < (fd)) max_file_descriptor = (fd); \
148 #define argf_of(obj) (*(struct argf *)DATA_PTR(obj))
149 #define ARGF argf_of(argf)
151 #ifdef _STDIO_USES_IOSTREAM /* GNU libc */
153 # define STDIO_READ_DATA_PENDING(fp) ((fp)->_IO_read_ptr != (fp)->_IO_read_end)
155 # define STDIO_READ_DATA_PENDING(fp) ((fp)->_gptr < (fp)->_egptr)
157 #elif defined(FILE_COUNT)
158 # define STDIO_READ_DATA_PENDING(fp) ((fp)->FILE_COUNT > 0)
159 #elif defined(FILE_READEND)
160 # define STDIO_READ_DATA_PENDING(fp) ((fp)->FILE_READPTR < (fp)->FILE_READEND)
161 #elif defined(__BEOS__)
162 # define STDIO_READ_DATA_PENDING(fp) (fp->_state._eof == 0)
164 # define STDIO_READ_DATA_PENDING(fp) (((unsigned int)(*(fp))->_cnt) > 0)
166 # define STDIO_READ_DATA_PENDING(fp) (!feof(fp))
170 #define fopen(file_spec, mode) fopen(file_spec, mode, "rfm=stmlf")
171 #define open(file_spec, flags, mode) open(file_spec, flags, mode, "rfm=stmlf")
174 #define GetWriteIO(io) rb_io_get_write_io(io)
176 #define READ_DATA_PENDING(fptr) ((fptr)->rbuf_len)
177 #define READ_DATA_PENDING_COUNT(fptr) ((fptr)->rbuf_len)
178 #define READ_DATA_PENDING_PTR(fptr) ((fptr)->rbuf+(fptr)->rbuf_off)
179 #define READ_DATA_BUFFERED(fptr) READ_DATA_PENDING(fptr)
181 #define READ_CHECK(fptr) do {\
182 if (!READ_DATA_PENDING(fptr)) {\
183 rb_thread_wait_fd((fptr)->fd);\
184 rb_io_check_closed(fptr);\
190 # define S_ISSOCK(m) _S_ISSOCK(m)
193 # define S_ISSOCK(m) ((m & S_IFMT) == _S_IFSOCK)
196 # define S_ISSOCK(m) ((m & S_IFMT) == S_IFSOCK)
202 #if !defined HAVE_SHUTDOWN && !defined shutdown
203 #define shutdown(a,b) 0
207 #define is_socket(fd, path) rb_w32_is_socket(fd)
208 #elif !defined(S_ISSOCK)
209 #define is_socket(fd, path) 0
212 is_socket(int fd
, const char *path
)
215 if (fstat(fd
, &sbuf
) < 0)
217 return S_ISSOCK(sbuf
.st_mode
);
224 rb_raise(rb_eEOFError
, "end of file reached");
228 rb_io_taint_check(VALUE io
)
230 if (!OBJ_UNTRUSTED(io
) && rb_safe_level() >= 4)
231 rb_raise(rb_eSecurityError
, "Insecure: operation on trusted IO");
237 rb_io_check_initialized(rb_io_t
*fptr
)
240 rb_raise(rb_eIOError
, "uninitialized stream");
245 rb_io_check_closed(rb_io_t
*fptr
)
247 rb_io_check_initialized(fptr
);
249 rb_raise(rb_eIOError
, "closed stream");
253 static int io_fflush(rb_io_t
*);
256 rb_io_get_io(VALUE io
)
258 return rb_convert_type(io
, T_FILE
, "IO", "to_io");
262 rb_io_check_io(VALUE io
)
264 return rb_check_convert_type(io
, T_FILE
, "IO", "to_io");
268 rb_io_get_write_io(VALUE io
)
271 rb_io_check_initialized(RFILE(io
)->fptr
);
272 write_io
= RFILE(io
)->fptr
->tied_io_for_writing
;
281 * IO.try_convert(obj) -> io or nil
283 * Try to convert <i>obj</i> into an IO, using to_io method.
284 * Returns converted IO or nil if <i>obj</i> cannot be converted
287 * IO.try_convert(STDOUT) # => STDOUT
288 * IO.try_convert("STDOUT") # => nil
291 rb_io_s_try_convert(VALUE dummy
, VALUE io
)
293 return rb_io_check_io(io
);
297 io_unread(rb_io_t
*fptr
)
300 rb_io_check_closed(fptr
);
301 if (fptr
->rbuf_len
== 0 || fptr
->mode
& FMODE_DUPLEX
)
303 /* xxx: target position may be negative if buffer is filled by ungetc */
304 #if defined(_WIN32) || defined(DJGPP) || defined(__CYGWIN__) || defined(__human68k__) || defined(__EMX__)
305 if (!(fptr
->mode
& FMODE_BINMODE
)) {
306 int len
= fptr
->rbuf_len
;
307 while (fptr
->rbuf_len
-- > 0) {
308 if (fptr
->rbuf
[fptr
->rbuf_len
] == '\n')
311 r
= lseek(fptr
->fd
, -len
, SEEK_CUR
);
315 r
= lseek(fptr
->fd
, -fptr
->rbuf_len
, SEEK_CUR
);
318 fptr
->mode
|= FMODE_DUPLEX
;
326 static rb_encoding
*io_input_encoding(rb_io_t
*fptr
);
329 io_ungetc(VALUE str
, rb_io_t
*fptr
)
331 int len
= RSTRING_LEN(str
);
333 if (rb_enc_dummy_p(io_input_encoding(fptr
))) {
334 rb_raise(rb_eNotImpError
, "ungetc against dummy encoding is not currently supported");
337 if (fptr
->rbuf
== NULL
) {
341 fptr
->rbuf_capa
= len
;
343 fptr
->rbuf_capa
= 8192;
344 fptr
->rbuf
= ALLOC_N(char, fptr
->rbuf_capa
);
346 if (fptr
->rbuf_capa
< len
+ fptr
->rbuf_len
) {
347 rb_raise(rb_eIOError
, "ungetc failed");
349 if (fptr
->rbuf_off
< len
) {
350 MEMMOVE(fptr
->rbuf
+fptr
->rbuf_capa
-fptr
->rbuf_len
,
351 fptr
->rbuf
+fptr
->rbuf_off
,
352 char, fptr
->rbuf_len
);
353 fptr
->rbuf_off
= fptr
->rbuf_capa
-fptr
->rbuf_len
;
357 MEMMOVE(fptr
->rbuf
+fptr
->rbuf_off
, RSTRING_PTR(str
), char, len
);
361 flush_before_seek(rb_io_t
*fptr
)
369 #define io_seek(fptr, ofs, whence) lseek(flush_before_seek(fptr)->fd, ofs, whence)
370 #define io_tell(fptr) lseek(flush_before_seek(fptr)->fd, 0, SEEK_CUR)
378 #define FMODE_SYNCWRITE (FMODE_SYNC|FMODE_WRITABLE)
381 rb_io_check_readable(rb_io_t
*fptr
)
383 rb_io_check_closed(fptr
);
384 if (!(fptr
->mode
& FMODE_READABLE
)) {
385 rb_raise(rb_eIOError
, "not opened for reading");
387 if (fptr
->wbuf_len
) {
390 if (fptr
->tied_io_for_writing
) {
392 GetOpenFile(fptr
->tied_io_for_writing
, wfptr
);
395 if (!fptr
->enc
&& fptr
->fd
== 0) {
396 fptr
->enc
= rb_default_external_encoding();
401 io_read_encoding(rb_io_t
*fptr
)
406 return rb_default_external_encoding();
410 io_input_encoding(rb_io_t
*fptr
)
415 return io_read_encoding(fptr
);
419 rb_io_check_writable(rb_io_t
*fptr
)
421 rb_io_check_closed(fptr
);
422 if (!(fptr
->mode
& FMODE_WRITABLE
)) {
423 rb_raise(rb_eIOError
, "not opened for writing");
425 if (fptr
->rbuf_len
) {
431 rb_read_pending(FILE *fp
)
433 return STDIO_READ_DATA_PENDING(fp
);
437 rb_io_read_pending(rb_io_t
*fptr
)
439 return READ_DATA_PENDING(fptr
);
443 rb_read_check(FILE *fp
)
445 if (!STDIO_READ_DATA_PENDING(fp
)) {
446 rb_thread_wait_fd(fileno(fp
));
451 rb_io_read_check(rb_io_t
*fptr
)
453 if (!READ_DATA_PENDING(fptr
)) {
454 rb_thread_wait_fd(fptr
->fd
);
466 if (errno
== EMFILE
|| errno
== ENFILE
|| errno
== ENOMEM
) {
478 io_alloc(VALUE klass
)
480 NEWOBJ(io
, struct RFile
);
481 OBJSETUP(io
, klass
, T_FILE
);
489 # define S_ISREG(m) ((m & S_IFMT) == S_IFREG)
493 wsplit_p(rb_io_t
*fptr
)
495 #if defined(HAVE_FCNTL) && defined(F_GETFL) && defined(O_NONBLOCK)
499 if (!(fptr
->mode
& FMODE_WSPLIT_INITIALIZED
)) {
501 if (fstat(fptr
->fd
, &buf
) == 0 &&
502 !S_ISREG(buf
.st_mode
)
503 #if defined(HAVE_FCNTL) && defined(F_GETFL) && defined(O_NONBLOCK)
504 && (r
= fcntl(fptr
->fd
, F_GETFL
)) != -1 &&
508 fptr
->mode
|= FMODE_WSPLIT
;
510 fptr
->mode
|= FMODE_WSPLIT_INITIALIZED
;
512 return fptr
->mode
& FMODE_WSPLIT
;
515 struct io_internal_struct
{
522 internal_read_func(void *ptr
)
524 struct io_internal_struct
*iis
= (struct io_internal_struct
*)ptr
;
525 return read(iis
->fd
, iis
->buf
, iis
->capa
);
529 internal_write_func(void *ptr
)
531 struct io_internal_struct
*iis
= (struct io_internal_struct
*)ptr
;
532 return write(iis
->fd
, iis
->buf
, iis
->capa
);
536 rb_read_internal(int fd
, void *buf
, size_t count
)
538 struct io_internal_struct iis
;
543 return rb_thread_blocking_region(internal_read_func
, &iis
, RB_UBF_DFL
, 0);
547 rb_write_internal(int fd
, void *buf
, size_t count
)
549 struct io_internal_struct iis
;
554 return rb_thread_blocking_region(internal_write_func
, &iis
, RB_UBF_DFL
, 0);
558 io_fflush(rb_io_t
*fptr
)
561 int wbuf_off
, wbuf_len
;
563 rb_io_check_closed(fptr
);
564 if (fptr
->wbuf_len
== 0)
566 if (!rb_thread_fd_writable(fptr
->fd
)) {
567 rb_io_check_closed(fptr
);
570 if (fptr
->wbuf_len
== 0)
572 wbuf_off
= fptr
->wbuf_off
;
573 wbuf_len
= fptr
->wbuf_len
;
576 !rb_thread_critical
&&
577 !rb_thread_alone() &&
581 r
= rb_write_internal(fptr
->fd
, fptr
->wbuf
+wbuf_off
, l
);
582 /* xxx: Other threads may modify wbuf.
583 * A lock is required, definitely. */
584 rb_io_check_closed(fptr
);
585 if (fptr
->wbuf_len
<= r
) {
595 if (rb_io_wait_writable(fptr
->fd
)) {
596 rb_io_check_closed(fptr
);
602 #ifdef HAVE_RB_FD_INIT
604 wait_readable(VALUE p
)
606 rb_fdset_t
*rfds
= (rb_fdset_t
*)p
;
608 return rb_thread_select(rb_fd_max(rfds
), rb_fd_ptr(rfds
), NULL
, NULL
, NULL
);
613 rb_io_wait_readable(int f
)
618 rb_raise(rb_eIOError
, "closed stream");
622 #if defined(ERESTART)
625 rb_thread_wait_fd(f
);
629 #if defined(EWOULDBLOCK) && EWOULDBLOCK != EAGAIN
634 #ifdef HAVE_RB_FD_INIT
635 rb_ensure(wait_readable
, (VALUE
)&rfds
,
636 (VALUE (*)(VALUE
))rb_fd_term
, (VALUE
)&rfds
);
638 rb_thread_select(f
+ 1, &rfds
, NULL
, NULL
, NULL
);
647 #ifdef HAVE_RB_FD_INIT
649 wait_writable(VALUE p
)
651 rb_fdset_t
*wfds
= (rb_fdset_t
*)p
;
653 return rb_thread_select(rb_fd_max(wfds
), NULL
, rb_fd_ptr(wfds
), NULL
, NULL
);
658 rb_io_wait_writable(int f
)
663 rb_raise(rb_eIOError
, "closed stream");
667 #if defined(ERESTART)
670 rb_thread_fd_writable(f
);
674 #if defined(EWOULDBLOCK) && EWOULDBLOCK != EAGAIN
679 #ifdef HAVE_RB_FD_INIT
680 rb_ensure(wait_writable
, (VALUE
)&wfds
,
681 (VALUE (*)(VALUE
))rb_fd_term
, (VALUE
)&wfds
);
683 rb_thread_select(f
+ 1, NULL
, &wfds
, NULL
, NULL
);
692 /* writing functions */
694 io_fwrite(VALUE str
, rb_io_t
*fptr
)
696 long len
, n
, r
, l
, offset
= 0;
699 * If an external encoding was specified and it differs from
700 * the strings encoding then we must transcode before writing.
701 * We must also transcode if two encodings were specified
704 /* transcode str before output */
705 /* the methods in transcode.c are static, so call indirectly */
706 /* Can't use encode! because puts writes a frozen newline */
708 str
= rb_funcall(str
, id_encode
, 2,
709 rb_enc_from_encoding(fptr
->enc2
),
710 rb_enc_from_encoding(fptr
->enc
));
713 str
= rb_funcall(str
, id_encode
, 1,
714 rb_enc_from_encoding(fptr
->enc
));
718 len
= RSTRING_LEN(str
);
719 if ((n
= len
) <= 0) return n
;
720 if (fptr
->wbuf
== NULL
&& !(fptr
->mode
& FMODE_SYNC
)) {
723 fptr
->wbuf_capa
= 8192;
724 fptr
->wbuf
= ALLOC_N(char, fptr
->wbuf_capa
);
726 if ((fptr
->mode
& (FMODE_SYNC
|FMODE_TTY
)) ||
727 (fptr
->wbuf
&& fptr
->wbuf_capa
<= fptr
->wbuf_len
+ len
)) {
728 /* xxx: use writev to avoid double write if available */
729 if (fptr
->wbuf_len
&& fptr
->wbuf_len
+len
<= fptr
->wbuf_capa
) {
730 if (fptr
->wbuf_capa
< fptr
->wbuf_off
+fptr
->wbuf_len
+len
) {
731 MEMMOVE(fptr
->wbuf
, fptr
->wbuf
+fptr
->wbuf_off
, char, fptr
->wbuf_len
);
734 MEMMOVE(fptr
->wbuf
+fptr
->wbuf_off
+fptr
->wbuf_len
, RSTRING_PTR(str
)+offset
, char, len
);
735 fptr
->wbuf_len
+= len
;
738 if (io_fflush(fptr
) < 0)
742 /* avoid context switch between "a" and "\n" in STDERR.puts "a".
744 if (fptr
->stdio_file
!= stderr
&& !rb_thread_fd_writable(fptr
->fd
)) {
745 rb_io_check_closed(fptr
);
750 !rb_thread_critical
&&
751 !rb_thread_alone() &&
755 r
= rb_write_internal(fptr
->fd
, RSTRING_PTR(str
)+offset
, l
);
756 /* xxx: other threads may modify given string. */
757 if (r
== n
) return len
;
763 if (rb_io_wait_writable(fptr
->fd
)) {
764 rb_io_check_closed(fptr
);
765 if (offset
< RSTRING_LEN(str
))
771 if (fptr
->wbuf_off
) {
773 MEMMOVE(fptr
->wbuf
, fptr
->wbuf
+fptr
->wbuf_off
, char, fptr
->wbuf_len
);
776 MEMMOVE(fptr
->wbuf
+fptr
->wbuf_off
+fptr
->wbuf_len
, RSTRING_PTR(str
)+offset
, char, len
);
777 fptr
->wbuf_len
+= len
;
782 rb_io_fwrite(const char *ptr
, long len
, FILE *f
)
788 of
.mode
= FMODE_WRITABLE
;
790 return io_fwrite(rb_str_new(ptr
, len
), &of
);
795 * ios.write(string) => integer
797 * Writes the given string to <em>ios</em>. The stream must be opened
798 * for writing. If the argument is not a string, it will be converted
799 * to a string using <code>to_s</code>. Returns the number of bytes
802 * count = $stdout.write( "This is a test\n" )
803 * puts "That was #{count} bytes of data"
808 * That was 15 bytes of data
812 io_write(VALUE io
, VALUE str
)
820 str
= rb_obj_as_string(str
);
821 tmp
= rb_io_check_io(io
);
823 /* port is not IO, call write method for it. */
824 return rb_funcall(io
, id_write
, 1, str
);
827 if (RSTRING_LEN(str
) == 0) return INT2FIX(0);
829 GetOpenFile(io
, fptr
);
830 rb_io_check_writable(fptr
);
832 n
= io_fwrite(str
, fptr
);
833 if (n
== -1L) rb_sys_fail(fptr
->path
);
839 rb_io_write(VALUE io
, VALUE str
)
841 return rb_funcall(io
, id_write
, 1, str
);
848 * String Output---Writes <i>obj</i> to <em>ios</em>.
849 * <i>obj</i> will be converted to a string using
852 * $stdout << "Hello " << "world!\n"
861 rb_io_addstr(VALUE io
, VALUE str
)
863 rb_io_write(io
, str
);
871 * Flushes any buffered data within <em>ios</em> to the underlying
872 * operating system (note that this is Ruby internal buffering only;
873 * the OS may buffer the data as well).
875 * $stdout.print "no newline"
884 rb_io_flush(VALUE io
)
888 if (TYPE(io
) != T_FILE
) {
889 return rb_funcall(io
, id_flush
, 0);
893 GetOpenFile(io
, fptr
);
895 if (fptr
->mode
& FMODE_WRITABLE
) {
898 if (fptr
->mode
& FMODE_READABLE
) {
908 * ios.tell => integer
910 * Returns the current offset (in bytes) of <em>ios</em>.
912 * f = File.new("testfile")
914 * f.gets #=> "This is line one\n"
924 GetOpenFile(io
, fptr
);
926 if (pos
< 0 && errno
) rb_sys_fail(fptr
->path
);
927 return OFFT2NUM(pos
);
931 rb_io_seek(VALUE io
, VALUE offset
, int whence
)
936 pos
= NUM2OFFT(offset
);
937 GetOpenFile(io
, fptr
);
938 pos
= io_seek(fptr
, pos
, whence
);
939 if (pos
< 0 && errno
) rb_sys_fail(fptr
->path
);
946 * ios.seek(amount, whence=SEEK_SET) -> 0
948 * Seeks to a given offset <i>anInteger</i> in the stream according to
949 * the value of <i>whence</i>:
951 * IO::SEEK_CUR | Seeks to _amount_ plus current position
952 * --------------+----------------------------------------------------
953 * IO::SEEK_END | Seeks to _amount_ plus end of stream (you probably
954 * | want a negative value for _amount_)
955 * --------------+----------------------------------------------------
956 * IO::SEEK_SET | Seeks to the absolute location given by _amount_
960 * f = File.new("testfile")
961 * f.seek(-13, IO::SEEK_END) #=> 0
962 * f.readline #=> "And so on...\n"
966 rb_io_seek_m(int argc
, VALUE
*argv
, VALUE io
)
968 VALUE offset
, ptrname
;
969 int whence
= SEEK_SET
;
971 if (rb_scan_args(argc
, argv
, "11", &offset
, &ptrname
) == 2) {
972 whence
= NUM2INT(ptrname
);
975 return rb_io_seek(io
, offset
, whence
);
980 * ios.pos = integer => integer
982 * Seeks to the given position (in bytes) in <em>ios</em>.
984 * f = File.new("testfile")
986 * f.gets #=> "This is line two\n"
990 rb_io_set_pos(VALUE io
, VALUE offset
)
995 pos
= NUM2OFFT(offset
);
996 GetOpenFile(io
, fptr
);
997 pos
= io_seek(fptr
, pos
, SEEK_SET
);
998 if (pos
< 0) rb_sys_fail(fptr
->path
);
1000 return OFFT2NUM(pos
);
1007 * Positions <em>ios</em> to the beginning of input, resetting
1008 * <code>lineno</code> to zero.
1010 * f = File.new("testfile")
1011 * f.readline #=> "This is line one\n"
1014 * f.readline #=> "This is line one\n"
1018 rb_io_rewind(VALUE io
)
1022 GetOpenFile(io
, fptr
);
1023 if (io_seek(fptr
, 0L, 0) < 0) rb_sys_fail(fptr
->path
);
1024 if (io
== ARGF
.current_file
) {
1025 ARGF
.gets_lineno
-= fptr
->lineno
;
1033 io_fillbuf(rb_io_t
*fptr
)
1037 if (fptr
->rbuf
== NULL
) {
1040 fptr
->rbuf_capa
= 8192;
1041 fptr
->rbuf
= ALLOC_N(char, fptr
->rbuf_capa
);
1043 if (fptr
->rbuf_len
== 0) {
1046 r
= rb_read_internal(fptr
->fd
, fptr
->rbuf
, fptr
->rbuf_capa
);
1049 if (rb_io_wait_readable(fptr
->fd
))
1051 rb_sys_fail(fptr
->path
);
1056 return -1; /* EOF */
1063 * ios.eof => true or false
1064 * ios.eof? => true or false
1066 * Returns true if <em>ios</em> is at end of file that means
1067 * there are no more data to read.
1068 * The stream must be opened for reading or an <code>IOError</code> will be
1071 * f = File.new("testfile")
1072 * dummy = f.readlines
1075 * If <em>ios</em> is a stream such as pipe or socket, <code>IO#eof?</code>
1076 * blocks until the other end sends some data or closes it.
1079 * Thread.new { sleep 1; w.close }
1080 * r.eof? #=> true after 1 second blocking
1083 * Thread.new { sleep 1; w.puts "a" }
1084 * r.eof? #=> false after 1 second blocking
1087 * r.eof? # blocks forever
1089 * Note that <code>IO#eof?</code> reads data to a input buffer.
1090 * So <code>IO#sysread</code> doesn't work with <code>IO#eof?</code>.
1098 GetOpenFile(io
, fptr
);
1099 rb_io_check_readable(fptr
);
1101 if (READ_DATA_PENDING(fptr
)) return Qfalse
;
1103 if (io_fillbuf(fptr
) < 0) {
1111 * ios.sync => true or false
1113 * Returns the current ``sync mode'' of <em>ios</em>. When sync mode is
1114 * true, all output is immediately flushed to the underlying operating
1115 * system and is not buffered by Ruby internally. See also
1116 * <code>IO#fsync</code>.
1118 * f = File.new("testfile")
1123 rb_io_sync(VALUE io
)
1127 io
= GetWriteIO(io
);
1128 GetOpenFile(io
, fptr
);
1129 return (fptr
->mode
& FMODE_SYNC
) ? Qtrue
: Qfalse
;
1134 * ios.sync = boolean => boolean
1136 * Sets the ``sync mode'' to <code>true</code> or <code>false</code>.
1137 * When sync mode is true, all output is immediately flushed to the
1138 * underlying operating system and is not buffered internally. Returns
1139 * the new state. See also <code>IO#fsync</code>.
1141 * f = File.new("testfile")
1144 * <em>(produces no output)</em>
1148 rb_io_set_sync(VALUE io
, VALUE mode
)
1152 io
= GetWriteIO(io
);
1153 GetOpenFile(io
, fptr
);
1155 fptr
->mode
|= FMODE_SYNC
;
1158 fptr
->mode
&= ~FMODE_SYNC
;
1165 * ios.fsync => 0 or nil
1167 * Immediately writes all buffered data in <em>ios</em> to disk.
1168 * Returns <code>nil</code> if the underlying operating system does not
1169 * support <em>fsync(2)</em>. Note that <code>fsync</code> differs from
1170 * using <code>IO#sync=</code>. The latter ensures that data is flushed
1171 * from Ruby's buffers, but doesn't not guarantee that the underlying
1172 * operating system actually writes it to disk.
1176 rb_io_fsync(VALUE io
)
1181 io
= GetWriteIO(io
);
1182 GetOpenFile(io
, fptr
);
1185 if (fsync(fptr
->fd
) < 0)
1186 rb_sys_fail(fptr
->path
);
1190 return Qnil
; /* not reached */
1196 * ios.fileno => fixnum
1197 * ios.to_i => fixnum
1199 * Returns an integer representing the numeric file descriptor for
1202 * $stdin.fileno #=> 0
1203 * $stdout.fileno #=> 1
1207 rb_io_fileno(VALUE io
)
1212 GetOpenFile(io
, fptr
);
1222 * Returns the process ID of a child process associated with
1223 * <em>ios</em>. This will be set by <code>IO::popen</code>.
1225 * pipe = IO.popen("-")
1227 * $stderr.puts "In parent, child pid is #{pipe.pid}"
1229 * $stderr.puts "In child, pid is #{$$}"
1232 * <em>produces:</em>
1234 * In child, pid is 26209
1235 * In parent, child pid is 26209
1243 GetOpenFile(io
, fptr
);
1246 return INT2FIX(fptr
->pid
);
1252 * ios.inspect => string
1254 * Return a string describing this IO object.
1258 rb_io_inspect(VALUE obj
)
1262 const char *st
= "";
1264 fptr
= RFILE(rb_io_taint_check(obj
))->fptr
;
1265 if (!fptr
|| !fptr
->path
) return rb_any_to_s(obj
);
1266 cname
= rb_obj_classname(obj
);
1270 return rb_sprintf("#<%s:%s%s>", cname
, fptr
->path
, st
);
1277 * Returns <em>ios</em>.
1281 rb_io_to_io(VALUE io
)
1286 /* reading functions */
1288 read_buffered_data(char *ptr
, long len
, rb_io_t
*fptr
)
1292 n
= READ_DATA_PENDING_COUNT(fptr
);
1293 if (n
<= 0) return 0;
1294 if (n
> len
) n
= len
;
1295 MEMMOVE(ptr
, fptr
->rbuf
+fptr
->rbuf_off
, char, n
);
1296 fptr
->rbuf_off
+= n
;
1297 fptr
->rbuf_len
-= n
;
1302 io_fread(VALUE str
, long offset
, rb_io_t
*fptr
)
1304 long len
= RSTRING_LEN(str
) - offset
;
1308 if (READ_DATA_PENDING(fptr
) == 0) {
1310 c
= rb_read_internal(fptr
->fd
, RSTRING_PTR(str
)+offset
, n
);
1313 rb_sys_fail(fptr
->path
);
1316 if ((n
-= c
) <= 0) break;
1317 rb_thread_wait_fd(fptr
->fd
);
1323 c
= read_buffered_data(RSTRING_PTR(str
)+offset
, n
, fptr
);
1326 if ((n
-= c
) <= 0) break;
1328 rb_thread_wait_fd(fptr
->fd
);
1329 rb_io_check_closed(fptr
);
1330 if (io_fillbuf(fptr
) < 0) {
1338 rb_io_fread(char *ptr
, long len
, FILE *f
)
1346 of
.mode
= FMODE_READABLE
;
1347 str
= rb_str_new(ptr
, len
);
1348 n
= io_fread(str
, 0, &of
);
1349 MEMCPY(ptr
, RSTRING_PTR(str
), char, n
);
1353 #define SMALLBUF 100
1356 remain_size(rb_io_t
*fptr
)
1359 off_t siz
= READ_DATA_PENDING_COUNT(fptr
);
1362 if (fstat(fptr
->fd
, &st
) == 0 && S_ISREG(st
.st_mode
)
1369 pos
= lseek(fptr
->fd
, 0, SEEK_CUR
);
1370 if (st
.st_size
>= pos
&& pos
>= 0) {
1371 siz
+= st
.st_size
- pos
;
1372 if (siz
> LONG_MAX
) {
1373 rb_raise(rb_eIOError
, "file too big for single read");
1384 io_enc_str(VALUE str
, rb_io_t
*fptr
)
1387 rb_enc_associate(str
, io_read_encoding(fptr
));
1392 make_readconv(rb_io_t
*fptr
)
1394 if (!fptr
->readconv
) {
1395 fptr
->readconv
= rb_econv_open(fptr
->enc2
->name
, fptr
->enc
->name
, 0);
1396 if (!fptr
->readconv
)
1397 rb_raise(rb_eIOError
, "code converter open failed (%s to %s)", fptr
->enc
->name
, fptr
->enc2
->name
);
1398 fptr
->crbuf_off
= 0;
1399 fptr
->crbuf_len
= 0;
1400 fptr
->crbuf_capa
= 1024;
1401 fptr
->crbuf
= ALLOC_N(char, fptr
->crbuf_capa
);
1406 more_char(rb_io_t
*fptr
)
1408 const unsigned char *ss
, *sp
, *se
;
1409 unsigned char *ds
, *dp
, *de
;
1410 rb_econv_result_t res
;
1414 if (fptr
->crbuf_len
== fptr
->crbuf_capa
)
1415 return 0; /* crbuf full */
1416 if (fptr
->crbuf_len
== 0)
1417 fptr
->crbuf_off
= 0;
1418 else if (fptr
->crbuf_off
+ fptr
->crbuf_len
== fptr
->crbuf_capa
) {
1419 memmove(fptr
->crbuf
, fptr
->crbuf
+fptr
->crbuf_off
, fptr
->crbuf_len
);
1420 fptr
->crbuf_off
= 0;
1423 crbuf_len0
= fptr
->crbuf_len
;
1426 ss
= sp
= (const unsigned char *)fptr
->rbuf
+ fptr
->rbuf_off
;
1427 se
= sp
+ fptr
->rbuf_len
;
1428 ds
= dp
= (unsigned char *)fptr
->crbuf
+ fptr
->crbuf_off
+ fptr
->crbuf_len
;
1429 de
= (unsigned char *)fptr
->crbuf
+ fptr
->crbuf_capa
;
1430 res
= rb_econv_convert(fptr
->readconv
, &sp
, se
, &dp
, de
, ECONV_PARTIAL_INPUT
|ECONV_OUTPUT_FOLLOWED_BY_INPUT
);
1431 fptr
->rbuf_off
+= sp
- ss
;
1432 fptr
->rbuf_len
-= sp
- ss
;
1433 fptr
->crbuf_len
+= dp
- ds
;
1435 putbackable
= rb_econv_putbackable(fptr
->readconv
);
1437 rb_econv_putback(fptr
->readconv
, (unsigned char *)fptr
->rbuf
+ fptr
->rbuf_off
- putbackable
, putbackable
);
1438 fptr
->rbuf_off
-= putbackable
;
1439 fptr
->rbuf_len
+= putbackable
;
1442 rb_econv_check_error(fptr
->readconv
);
1444 if (crbuf_len0
!= fptr
->crbuf_len
)
1447 if (res
== econv_finished
)
1450 if (res
== econv_source_buffer_empty
) {
1451 if (fptr
->rbuf_len
== 0) {
1452 rb_thread_wait_fd(fptr
->fd
);
1453 rb_io_check_closed(fptr
);
1454 if (io_fillbuf(fptr
) == -1) {
1455 ds
= dp
= (unsigned char *)fptr
->crbuf
+ fptr
->crbuf_off
+ fptr
->crbuf_len
;
1456 de
= (unsigned char *)fptr
->crbuf
+ fptr
->crbuf_capa
;
1457 res
= rb_econv_convert(fptr
->readconv
, NULL
, NULL
, &dp
, de
, 0);
1458 fptr
->crbuf_len
+= dp
- ds
;
1459 rb_econv_check_error(fptr
->readconv
);
1467 io_shift_crbuf(rb_io_t
*fptr
, int len
, VALUE
*strp
)
1471 *strp
= str
= rb_str_new(fptr
->crbuf
+fptr
->crbuf_off
, len
);
1476 slen
= RSTRING_LEN(str
);
1477 rb_str_resize(str
, RSTRING_LEN(str
) + len
);
1478 memcpy(RSTRING_PTR(str
)+slen
, fptr
->crbuf
+fptr
->crbuf_off
, len
);
1480 fptr
->crbuf_off
+= len
;
1481 fptr
->crbuf_len
-= len
;
1483 rb_enc_associate(str
, fptr
->enc
);
1484 /* xxx: set coderange */
1485 if (fptr
->crbuf_len
== 0)
1486 fptr
->crbuf_off
= 0;
1487 if (fptr
->crbuf_off
< fptr
->crbuf_capa
/2) {
1488 memmove(fptr
->crbuf
, fptr
->crbuf
+fptr
->crbuf_off
, fptr
->crbuf_len
);
1489 fptr
->crbuf_off
= 0;
1495 read_all(rb_io_t
*fptr
, long siz
, VALUE str
)
1504 VALUE str
= rb_str_new(NULL
, 0);
1505 make_readconv(fptr
);
1507 if (fptr
->crbuf_len
) {
1508 io_shift_crbuf(fptr
, fptr
->crbuf_len
, &str
);
1510 if (more_char(fptr
) == -1) {
1511 return io_enc_str(str
, fptr
);
1519 enc
= io_read_encoding(fptr
);
1520 cr
= fptr
->enc2
? ENC_CODERANGE_BROKEN
: 0;
1522 if (siz
== 0) siz
= BUFSIZ
;
1524 str
= rb_str_new(0, siz
);
1527 rb_str_resize(str
, siz
);
1531 n
= io_fread(str
, bytes
, fptr
);
1532 if (n
== 0 && bytes
== 0) {
1536 if (cr
!= ENC_CODERANGE_BROKEN
)
1537 pos
= rb_str_coderange_scan_restartable(RSTRING_PTR(str
) + pos
, RSTRING_PTR(str
) + bytes
, enc
, &cr
);
1538 if (bytes
< siz
) break;
1540 rb_str_resize(str
, siz
);
1542 if (bytes
!= siz
) rb_str_resize(str
, bytes
);
1543 str
= io_enc_str(str
, fptr
);
1545 ENC_CODERANGE_SET(str
, cr
);
1551 rb_io_set_nonblock(rb_io_t
*fptr
)
1555 flags
= fcntl(fptr
->fd
, F_GETFL
);
1557 rb_sys_fail(fptr
->path
);
1562 if ((flags
& O_NONBLOCK
) == 0) {
1563 flags
|= O_NONBLOCK
;
1564 if (fcntl(fptr
->fd
, F_SETFL
, flags
) == -1) {
1565 rb_sys_fail(fptr
->path
);
1571 io_getpartial(int argc
, VALUE
*argv
, VALUE io
, int nonblock
)
1577 rb_scan_args(argc
, argv
, "11", &length
, &str
);
1579 if ((len
= NUM2LONG(length
)) < 0) {
1580 rb_raise(rb_eArgError
, "negative length %ld given", len
);
1584 str
= rb_str_new(0, len
);
1589 rb_str_resize(str
, len
);
1593 GetOpenFile(io
, fptr
);
1594 rb_io_check_readable(fptr
);
1601 if (RSTRING_LEN(str
) != len
) {
1603 rb_raise(rb_eRuntimeError
, "buffer string modified");
1605 n
= read_buffered_data(RSTRING_PTR(str
), len
, fptr
);
1608 if (RSTRING_LEN(str
) != len
) goto modified
;
1610 rb_io_set_nonblock(fptr
);
1612 n
= rb_read_internal(fptr
->fd
, RSTRING_PTR(str
), len
);
1614 if (!nonblock
&& rb_io_wait_readable(fptr
->fd
))
1616 rb_sys_fail(fptr
->path
);
1619 rb_str_resize(str
, n
);
1629 * ios.readpartial(maxlen) => string
1630 * ios.readpartial(maxlen, outbuf) => outbuf
1632 * Reads at most <i>maxlen</i> bytes from the I/O stream.
1633 * It blocks only if <em>ios</em> has no data immediately available.
1634 * It doesn't block if some data available.
1635 * If the optional <i>outbuf</i> argument is present,
1636 * it must reference a String, which will receive the data.
1637 * It raises <code>EOFError</code> on end of file.
1639 * readpartial is designed for streams such as pipe, socket, tty, etc.
1640 * It blocks only when no data immediately available.
1641 * This means that it blocks only when following all conditions hold.
1642 * * the buffer in the IO object is empty.
1643 * * the content of the stream is empty.
1644 * * the stream is not reached to EOF.
1646 * When readpartial blocks, it waits data or EOF on the stream.
1647 * If some data is reached, readpartial returns with the data.
1648 * If EOF is reached, readpartial raises EOFError.
1650 * When readpartial doesn't blocks, it returns or raises immediately.
1651 * If the buffer is not empty, it returns the data in the buffer.
1652 * Otherwise if the stream has some content,
1653 * it returns the data in the stream.
1654 * Otherwise if the stream is reached to EOF, it raises EOFError.
1656 * r, w = IO.pipe # buffer pipe content
1657 * w << "abc" # "" "abc".
1658 * r.readpartial(4096) #=> "abc" "" ""
1659 * r.readpartial(4096) # blocks because buffer and pipe is empty.
1661 * r, w = IO.pipe # buffer pipe content
1662 * w << "abc" # "" "abc"
1663 * w.close # "" "abc" EOF
1664 * r.readpartial(4096) #=> "abc" "" EOF
1665 * r.readpartial(4096) # raises EOFError
1667 * r, w = IO.pipe # buffer pipe content
1668 * w << "abc\ndef\n" # "" "abc\ndef\n"
1669 * r.gets #=> "abc\n" "def\n" ""
1670 * w << "ghi\n" # "def\n" "ghi\n"
1671 * r.readpartial(4096) #=> "def\n" "" "ghi\n"
1672 * r.readpartial(4096) #=> "ghi\n" "" ""
1674 * Note that readpartial behaves similar to sysread.
1675 * The differences are:
1676 * * If the buffer is not empty, read from the buffer instead of "sysread for buffered IO (IOError)".
1677 * * It doesn't cause Errno::EAGAIN and Errno::EINTR. When readpartial meets EAGAIN and EINTR by read system call, readpartial retry the system call.
1679 * The later means that readpartial is nonblocking-flag insensitive.
1680 * It blocks on the situation IO#sysread causes Errno::EAGAIN as if the fd is blocking mode.
1685 io_readpartial(int argc
, VALUE
*argv
, VALUE io
)
1689 ret
= io_getpartial(argc
, argv
, io
, 0);
1698 * ios.read_nonblock(maxlen) => string
1699 * ios.read_nonblock(maxlen, outbuf) => outbuf
1701 * Reads at most <i>maxlen</i> bytes from <em>ios</em> using
1702 * read(2) system call after O_NONBLOCK is set for
1703 * the underlying file descriptor.
1705 * If the optional <i>outbuf</i> argument is present,
1706 * it must reference a String, which will receive the data.
1708 * read_nonblock just calls read(2).
1709 * It causes all errors read(2) causes: EAGAIN, EINTR, etc.
1710 * The caller should care such errors.
1712 * read_nonblock causes EOFError on EOF.
1714 * If the read buffer is not empty,
1715 * read_nonblock reads from the buffer like readpartial.
1716 * In this case, read(2) is not called.
1721 io_read_nonblock(int argc
, VALUE
*argv
, VALUE io
)
1725 ret
= io_getpartial(argc
, argv
, io
, 1);
1734 * ios.write_nonblock(string) => integer
1736 * Writes the given string to <em>ios</em> using
1737 * write(2) system call after O_NONBLOCK is set for
1738 * the underlying file descriptor.
1740 * write_nonblock just calls write(2).
1741 * It causes all errors write(2) causes: EAGAIN, EINTR, etc.
1742 * The result may also be smaller than string.length (partial write).
1743 * The caller should care such errors and partial write.
1745 * If the write buffer is not empty, it is flushed at first.
1750 rb_io_write_nonblock(VALUE io
, VALUE str
)
1756 if (TYPE(str
) != T_STRING
)
1757 str
= rb_obj_as_string(str
);
1759 io
= GetWriteIO(io
);
1760 GetOpenFile(io
, fptr
);
1761 rb_io_check_writable(fptr
);
1765 rb_io_set_nonblock(fptr
);
1766 n
= write(fptr
->fd
, RSTRING_PTR(str
), RSTRING_LEN(str
));
1768 if (n
== -1) rb_sys_fail(fptr
->path
);
1775 * ios.read([length [, buffer]]) => string, buffer, or nil
1777 * Reads at most <i>length</i> bytes from the I/O stream, or to the
1778 * end of file if <i>length</i> is omitted or is <code>nil</code>.
1779 * <i>length</i> must be a non-negative integer or nil.
1780 * If the optional <i>buffer</i> argument is present, it must reference
1781 * a String, which will receive the data.
1783 * At end of file, it returns <code>nil</code> or <code>""</code>
1784 * depend on <i>length</i>.
1785 * <code><i>ios</i>.read()</code> and
1786 * <code><i>ios</i>.read(nil)</code> returns <code>""</code>.
1787 * <code><i>ios</i>.read(<i>positive-integer</i>)</code> returns nil.
1789 * <code><i>ios</i>.read(0)</code> returns <code>""</code>.
1791 * f = File.new("testfile")
1792 * f.read(16) #=> "This is line one"
1796 io_read(int argc
, VALUE
*argv
, VALUE io
)
1802 rb_scan_args(argc
, argv
, "02", &length
, &str
);
1804 if (NIL_P(length
)) {
1805 if (!NIL_P(str
)) StringValue(str
);
1806 GetOpenFile(io
, fptr
);
1807 rb_io_check_readable(fptr
);
1808 return read_all(fptr
, remain_size(fptr
), str
);
1810 len
= NUM2LONG(length
);
1812 rb_raise(rb_eArgError
, "negative length %ld given", len
);
1816 str
= rb_str_new(0, len
);
1821 rb_str_resize(str
,len
);
1824 GetOpenFile(io
, fptr
);
1825 rb_io_check_readable(fptr
);
1826 if (len
== 0) return str
;
1829 if (RSTRING_LEN(str
) != len
) {
1830 rb_raise(rb_eRuntimeError
, "buffer string modified");
1832 n
= io_fread(str
, 0, fptr
);
1834 if (fptr
->fd
< 0) return Qnil
;
1835 rb_str_resize(str
, 0);
1838 rb_str_resize(str
, n
);
1844 rscheck(const char *rsptr
, long rslen
, VALUE rs
)
1847 if (RSTRING_PTR(rs
) != rsptr
&& RSTRING_LEN(rs
) != rslen
)
1848 rb_raise(rb_eRuntimeError
, "rs modified");
1852 appendline(rb_io_t
*fptr
, int delim
, VALUE
*strp
, long *lp
)
1858 make_readconv(fptr
);
1862 if (fptr
->crbuf_len
) {
1863 p
= fptr
->crbuf
+fptr
->crbuf_off
;
1864 searchlen
= fptr
->crbuf_len
;
1865 if (0 < limit
&& limit
< searchlen
)
1867 e
= memchr(p
, delim
, searchlen
);
1870 *strp
= str
= rb_str_new(p
, e
-p
+1);
1872 rb_str_buf_cat(str
, p
, e
-p
+1);
1873 fptr
->crbuf_off
+= e
-p
+1;
1874 fptr
->crbuf_len
-= e
-p
+1;
1881 *strp
= str
= rb_str_new(p
, searchlen
);
1883 rb_str_buf_cat(str
, p
, searchlen
);
1884 fptr
->crbuf_off
+= searchlen
;
1885 fptr
->crbuf_len
-= searchlen
;
1890 return (unsigned char)RSTRING_PTR(str
)[RSTRING_LEN(str
)-1];
1894 if (more_char(fptr
) == -1) {
1902 long pending
= READ_DATA_PENDING_COUNT(fptr
);
1904 const char *p
= READ_DATA_PENDING_PTR(fptr
);
1908 if (limit
> 0 && pending
> limit
) pending
= limit
;
1909 e
= memchr(p
, delim
, pending
);
1910 if (e
) pending
= e
- p
+ 1;
1912 last
= RSTRING_LEN(str
);
1913 rb_str_resize(str
, last
+ pending
);
1917 *strp
= str
= rb_str_buf_new(pending
);
1918 rb_str_set_len(str
, pending
);
1920 read_buffered_data(RSTRING_PTR(str
) + last
, pending
, fptr
); /* must not fail */
1923 if (e
) return delim
;
1925 return (unsigned char)RSTRING_PTR(str
)[RSTRING_LEN(str
)-1];
1927 rb_thread_wait_fd(fptr
->fd
);
1928 rb_io_check_closed(fptr
);
1929 if (io_fillbuf(fptr
) < 0) {
1937 swallow(rb_io_t
*fptr
, int term
)
1941 while ((cnt
= READ_DATA_PENDING_COUNT(fptr
)) > 0) {
1943 const char *p
= READ_DATA_PENDING_PTR(fptr
);
1945 if (cnt
> sizeof buf
) cnt
= sizeof buf
;
1946 if (*p
!= term
) return Qtrue
;
1948 while (--i
&& *++p
== term
);
1949 if (!read_buffered_data(buf
, cnt
- i
, fptr
)) /* must not fail */
1950 rb_sys_fail(fptr
->path
);
1952 rb_thread_wait_fd(fptr
->fd
);
1953 rb_io_check_closed(fptr
);
1954 } while (io_fillbuf(fptr
) == 0);
1959 rb_io_getline_fast(rb_io_t
*fptr
, rb_encoding
*enc
)
1964 int cr
= fptr
->enc2
? ENC_CODERANGE_BROKEN
: 0;
1967 long pending
= READ_DATA_PENDING_COUNT(fptr
);
1970 const char *p
= READ_DATA_PENDING_PTR(fptr
);
1973 e
= memchr(p
, '\n', pending
);
1975 pending
= e
- p
+ 1;
1978 str
= rb_str_new(p
, pending
);
1979 fptr
->rbuf_off
+= pending
;
1980 fptr
->rbuf_len
-= pending
;
1983 rb_str_resize(str
, len
+ pending
);
1984 read_buffered_data(RSTRING_PTR(str
)+len
, pending
, fptr
);
1987 if (cr
!= ENC_CODERANGE_BROKEN
)
1988 pos
= rb_str_coderange_scan_restartable(RSTRING_PTR(str
) + pos
, RSTRING_PTR(str
) + len
, enc
, &cr
);
1991 rb_thread_wait_fd(fptr
->fd
);
1992 rb_io_check_closed(fptr
);
1993 if (io_fillbuf(fptr
) < 0) {
1994 if (NIL_P(str
)) return Qnil
;
1999 str
= io_enc_str(str
, fptr
);
2000 if (!fptr
->enc2
) ENC_CODERANGE_SET(str
, cr
);
2002 ARGF
.lineno
= INT2FIX(fptr
->lineno
);
2007 prepare_getline_args(int argc
, VALUE
*argv
, VALUE
*rsp
, long *limit
, VALUE io
)
2009 VALUE rs
= rb_rs
, lim
= Qnil
;
2015 if (NIL_P(argv
[0]) || !NIL_P(tmp
= rb_check_string_type(argv
[0]))) {
2022 else if (2 <= argc
) {
2023 rb_scan_args(argc
, argv
, "2", &rs
, &lim
);
2028 rb_encoding
*enc_rs
, *enc_io
;
2030 GetOpenFile(io
, fptr
);
2031 enc_rs
= rb_enc_get(rs
);
2032 enc_io
= io_read_encoding(fptr
);
2033 if (enc_io
!= enc_rs
&&
2034 (rb_enc_str_coderange(rs
) != ENC_CODERANGE_7BIT
||
2035 !rb_enc_asciicompat(enc_io
))) {
2036 if (rs
== rb_default_rs
) {
2037 rs
= rb_enc_str_new(0, 0, enc_io
);
2038 rb_str_buf_cat_ascii(rs
, "\n");
2041 rb_raise(rb_eArgError
, "encoding mismatch: %s IO with %s RS",
2042 rb_enc_name(enc_io
),
2043 rb_enc_name(enc_rs
));
2048 *limit
= NIL_P(lim
) ? -1L : NUM2LONG(lim
);
2052 rb_io_getline_1(VALUE rs
, long limit
, VALUE io
)
2059 GetOpenFile(io
, fptr
);
2060 rb_io_check_readable(fptr
);
2062 str
= read_all(fptr
, 0, Qnil
);
2063 if (RSTRING_LEN(str
) == 0) return Qnil
;
2065 else if (limit
== 0) {
2066 return rb_enc_str_new(0, 0, io_read_encoding(fptr
));
2068 else if (rs
== rb_default_rs
&& limit
< 0 && !fptr
->enc2
&&
2069 rb_enc_asciicompat(enc
= io_read_encoding(fptr
))) {
2070 return rb_io_getline_fast(fptr
, enc
);
2077 int extra_limit
= 16;
2079 rslen
= RSTRING_LEN(rs
);
2084 swallow(fptr
, '\n');
2088 rsptr
= RSTRING_PTR(rs
);
2090 newline
= (unsigned char)rsptr
[rslen
- 1];
2095 enc
= io_input_encoding(fptr
);
2096 while ((c
= appendline(fptr
, newline
, &str
, &limit
)) != EOF
) {
2097 const char *s
, *p
, *pp
;
2100 if (RSTRING_LEN(str
) < rslen
) continue;
2101 s
= RSTRING_PTR(str
);
2102 p
= s
+ RSTRING_LEN(str
) - rslen
;
2103 pp
= rb_enc_left_char_head(s
, p
, enc
);
2104 if (pp
!= p
) continue;
2105 if (!rspara
) rscheck(rsptr
, rslen
, rs
);
2106 if (memcmp(p
, rsptr
, rslen
) == 0) break;
2109 s
= RSTRING_PTR(str
);
2110 p
= s
+ RSTRING_LEN(str
);
2111 pp
= rb_enc_left_char_head(s
, p
-1, enc
);
2113 MBCLEN_NEEDMORE_P(rb_enc_precise_mbclen(pp
, p
, enc
))) {
2114 /* relax the limit while incomplete character.
2115 * extra_limit limits the relax length */
2128 swallow(fptr
, '\n');
2132 str
= io_enc_str(str
, fptr
);
2138 ARGF
.lineno
= INT2FIX(fptr
->lineno
);
2146 rb_io_getline(int argc
, VALUE
*argv
, VALUE io
)
2151 prepare_getline_args(argc
, argv
, &rs
, &limit
, io
);
2152 return rb_io_getline_1(rs
, limit
, io
);
2156 rb_io_gets(VALUE io
)
2160 GetOpenFile(io
, fptr
);
2161 rb_io_check_readable(fptr
);
2162 return rb_io_getline_fast(fptr
, io_read_encoding(fptr
));
2167 * ios.gets(sep=$/) => string or nil
2168 * ios.gets(limit) => string or nil
2169 * ios.gets(sep, limit) => string or nil
2171 * Reads the next ``line'' from the I/O stream; lines are separated by
2172 * <i>sep</i>. A separator of <code>nil</code> reads the entire
2173 * contents, and a zero-length separator reads the input a paragraph at
2174 * a time (two successive newlines in the input separate paragraphs).
2175 * The stream must be opened for reading or an <code>IOError</code>
2176 * will be raised. The line read in will be returned and also assigned
2177 * to <code>$_</code>. Returns <code>nil</code> if called at end of
2178 * file. If the first argument is an integer, or optional second
2179 * argument is given, the returning string would not be longer than the
2182 * File.new("testfile").gets #=> "This is line one\n"
2183 * $_ #=> "This is line one\n"
2187 rb_io_gets_m(int argc
, VALUE
*argv
, VALUE io
)
2191 str
= rb_io_getline(argc
, argv
, io
);
2192 rb_lastline_set(str
);
2199 * ios.lineno => integer
2201 * Returns the current line number in <em>ios</em>. The stream must be
2202 * opened for reading. <code>lineno</code> counts the number of times
2203 * <code>gets</code> is called, rather than the number of newlines
2204 * encountered. The two values will differ if <code>gets</code> is
2205 * called with a separator other than newline. See also the
2206 * <code>$.</code> variable.
2208 * f = File.new("testfile")
2210 * f.gets #=> "This is line one\n"
2212 * f.gets #=> "This is line two\n"
2217 rb_io_lineno(VALUE io
)
2221 GetOpenFile(io
, fptr
);
2222 rb_io_check_readable(fptr
);
2223 return INT2NUM(fptr
->lineno
);
2228 * ios.lineno = integer => integer
2230 * Manually sets the current line number to the given value.
2231 * <code>$.</code> is updated only on the next read.
2233 * f = File.new("testfile")
2234 * f.gets #=> "This is line one\n"
2238 * $. #=> 1 # lineno of last read
2239 * f.gets #=> "This is line two\n"
2240 * $. #=> 1001 # lineno of last read
2244 rb_io_set_lineno(VALUE io
, VALUE lineno
)
2248 GetOpenFile(io
, fptr
);
2249 rb_io_check_readable(fptr
);
2250 fptr
->lineno
= NUM2INT(lineno
);
2256 * ios.readline(sep=$/) => string
2257 * ios.readline(limit) => string
2258 * ios.readline(sep, limit) => string
2260 * Reads a line as with <code>IO#gets</code>, but raises an
2261 * <code>EOFError</code> on end of file.
2265 rb_io_readline(int argc
, VALUE
*argv
, VALUE io
)
2267 VALUE line
= rb_io_gets_m(argc
, argv
, io
);
2277 * ios.readlines(sep=$/) => array
2278 * ios.readlines(limit) => array
2279 * ios.readlines(sep, limit) => array
2281 * Reads all of the lines in <em>ios</em>, and returns them in
2282 * <i>anArray</i>. Lines are separated by the optional <i>sep</i>. If
2283 * <i>sep</i> is <code>nil</code>, the rest of the stream is returned
2284 * as a single record. If the first argument is an integer, or
2285 * optional second argument is given, the returning string would not be
2286 * longer than the given value. The stream must be opened for reading
2287 * or an <code>IOError</code> will be raised.
2289 * f = File.new("testfile")
2290 * f.readlines[0] #=> "This is line one\n"
2294 rb_io_readlines(int argc
, VALUE
*argv
, VALUE io
)
2296 VALUE line
, ary
, rs
;
2299 prepare_getline_args(argc
, argv
, &rs
, &limit
, io
);
2301 while (!NIL_P(line
= rb_io_getline_1(rs
, limit
, io
))) {
2302 rb_ary_push(ary
, line
);
2309 * ios.each(sep=$/) {|line| block } => ios
2310 * ios.each(limit) {|line| block } => ios
2311 * ios.each(sep,limit) {|line| block } => ios
2312 * ios.each_line(sep=$/) {|line| block } => ios
2313 * ios.each_line(limit) {|line| block } => ios
2314 * ios.each_line(sep,limit) {|line| block } => ios
2316 * Executes the block for every line in <em>ios</em>, where lines are
2317 * separated by <i>sep</i>. <em>ios</em> must be opened for
2318 * reading or an <code>IOError</code> will be raised.
2320 * f = File.new("testfile")
2321 * f.each {|line| puts "#{f.lineno}: #{line}" }
2323 * <em>produces:</em>
2325 * 1: This is line one
2326 * 2: This is line two
2327 * 3: This is line three
2332 rb_io_each_line(int argc
, VALUE
*argv
, VALUE io
)
2337 RETURN_ENUMERATOR(io
, argc
, argv
);
2338 prepare_getline_args(argc
, argv
, &rs
, &limit
, io
);
2339 while (!NIL_P(str
= rb_io_getline_1(rs
, limit
, io
))) {
2347 * ios.each_byte {|byte| block } => ios
2349 * Calls the given block once for each byte (0..255) in <em>ios</em>,
2350 * passing the byte as an argument. The stream must be opened for
2351 * reading or an <code>IOError</code> will be raised.
2353 * f = File.new("testfile")
2355 * f.each_byte {|x| checksum ^= x } #=> #<File:testfile>
2360 rb_io_each_byte(VALUE io
)
2365 RETURN_ENUMERATOR(io
, 0, 0);
2366 GetOpenFile(io
, fptr
);
2369 p
= fptr
->rbuf
+fptr
->rbuf_off
;
2370 e
= p
+ fptr
->rbuf_len
;
2374 rb_yield(INT2FIX(*p
& 0xff));
2378 rb_io_check_readable(fptr
);
2380 if (io_fillbuf(fptr
) < 0) {
2388 io_getc(rb_io_t
*fptr
, rb_encoding
*enc
)
2396 if (!fptr
->readconv
) {
2397 make_readconv(fptr
);
2401 if (fptr
->crbuf_len
) {
2402 r
= rb_enc_precise_mbclen(fptr
->crbuf
+fptr
->crbuf_off
,
2403 fptr
->crbuf
+fptr
->crbuf_off
+fptr
->crbuf_len
,
2405 if (!MBCLEN_NEEDMORE_P(r
))
2407 if (fptr
->crbuf_len
== fptr
->crbuf_capa
) {
2408 rb_raise(rb_eIOError
, "too long character");
2412 if (more_char(fptr
) == -1) {
2413 if (fptr
->crbuf_len
== 0)
2415 /* return an incomplete character just before EOF */
2416 return io_shift_crbuf(fptr
, fptr
->crbuf_len
, &str
);
2419 if (MBCLEN_INVALID_P(r
)) {
2420 r
= rb_enc_mbclen(fptr
->crbuf
+fptr
->crbuf_off
,
2421 fptr
->crbuf
+fptr
->crbuf_off
+fptr
->crbuf_len
,
2423 return io_shift_crbuf(fptr
, r
, &str
);
2425 return io_shift_crbuf(fptr
, MBCLEN_CHARFOUND_LEN(r
), &str
);
2428 if (io_fillbuf(fptr
) < 0) {
2431 if (rb_enc_asciicompat(enc
) && ISASCII(fptr
->rbuf
[fptr
->rbuf_off
])) {
2432 str
= rb_str_new(fptr
->rbuf
+fptr
->rbuf_off
, 1);
2433 fptr
->rbuf_off
+= 1;
2434 fptr
->rbuf_len
-= 1;
2435 cr
= ENC_CODERANGE_7BIT
;
2438 r
= rb_enc_precise_mbclen(fptr
->rbuf
+fptr
->rbuf_off
, fptr
->rbuf
+fptr
->rbuf_off
+fptr
->rbuf_len
, enc
);
2439 if (MBCLEN_CHARFOUND_P(r
) &&
2440 (n
= MBCLEN_CHARFOUND_LEN(r
)) <= fptr
->rbuf_len
) {
2441 str
= rb_str_new(fptr
->rbuf
+fptr
->rbuf_off
, n
);
2442 fptr
->rbuf_off
+= n
;
2443 fptr
->rbuf_len
-= n
;
2444 cr
= ENC_CODERANGE_VALID
;
2446 else if (MBCLEN_NEEDMORE_P(r
)) {
2447 str
= rb_str_new(fptr
->rbuf
+fptr
->rbuf_off
, fptr
->rbuf_len
);
2450 if (io_fillbuf(fptr
) != -1) {
2451 rb_str_cat(str
, fptr
->rbuf
+fptr
->rbuf_off
, 1);
2454 r
= rb_enc_precise_mbclen(RSTRING_PTR(str
), RSTRING_PTR(str
)+RSTRING_LEN(str
), enc
);
2455 if (MBCLEN_NEEDMORE_P(r
)) {
2458 else if (MBCLEN_CHARFOUND_P(r
)) {
2459 cr
= ENC_CODERANGE_VALID
;
2464 str
= rb_str_new(fptr
->rbuf
+fptr
->rbuf_off
, 1);
2469 if (!cr
) cr
= ENC_CODERANGE_BROKEN
;
2470 str
= io_enc_str(str
, fptr
);
2472 ENC_CODERANGE_SET(str
, cr
);
2479 * ios.each_char {|c| block } => ios
2481 * Calls the given block once for each character in <em>ios</em>,
2482 * passing the character as an argument. The stream must be opened for
2483 * reading or an <code>IOError</code> will be raised.
2485 * f = File.new("testfile")
2486 * f.each_char {|c| print c, ' ' } #=> #<File:testfile>
2490 rb_io_each_char(VALUE io
)
2496 RETURN_ENUMERATOR(io
, 0, 0);
2497 GetOpenFile(io
, fptr
);
2498 rb_io_check_readable(fptr
);
2500 enc
= io_input_encoding(fptr
);
2502 while (!NIL_P(c
= io_getc(fptr
, enc
))) {
2512 * ios.lines(sep=$/) => anEnumerator
2513 * ios.lines(limit) => anEnumerator
2514 * ios.lines(sep, limit) => anEnumerator
2516 * Returns an enumerator that gives each line in <em>ios</em>.
2517 * The stream must be opened for reading or an <code>IOError</code>
2520 * f = File.new("testfile")
2521 * f.lines.to_a #=> ["foo\n", "bar\n"]
2523 * f.lines.sort #=> ["bar\n", "foo\n"]
2527 rb_io_lines(int argc
, VALUE
*argv
, VALUE io
)
2529 return rb_enumeratorize(io
, ID2SYM(rb_intern("each_line")), argc
, argv
);
2534 * ios.bytes => anEnumerator
2536 * Returns an enumerator that gives each byte (0..255) in <em>ios</em>.
2537 * The stream must be opened for reading or an <code>IOError</code>
2540 * f = File.new("testfile")
2541 * f.bytes.to_a #=> [104, 101, 108, 108, 111]
2543 * f.bytes.sort #=> [101, 104, 108, 108, 111]
2547 rb_io_bytes(VALUE io
)
2549 return rb_enumeratorize(io
, ID2SYM(rb_intern("each_byte")), 0, 0);
2554 * ios.chars => anEnumerator
2556 * Returns an enumerator that gives each character in <em>ios</em>.
2557 * The stream must be opened for reading or an <code>IOError</code>
2560 * f = File.new("testfile")
2561 * f.chars.to_a #=> ["h", "e", "l", "l", "o"]
2563 * f.chars.sort #=> ["e", "h", "l", "l", "o"]
2567 rb_io_chars(VALUE io
)
2569 return rb_enumeratorize(io
, ID2SYM(rb_intern("each_char")), 0, 0);
2574 * ios.getc => fixnum or nil
2576 * Reads a one-character string from <em>ios</em>. Returns
2577 * <code>nil</code> if called at end of file.
2579 * f = File.new("testfile")
2585 rb_io_getc(VALUE io
)
2590 GetOpenFile(io
, fptr
);
2591 rb_io_check_readable(fptr
);
2593 enc
= io_input_encoding(fptr
);
2595 return io_getc(fptr
, enc
);
2612 * ios.readchar => string
2614 * Reads a one-character string from <em>ios</em>. Raises an
2615 * <code>EOFError</code> on end of file.
2617 * f = File.new("testfile")
2618 * f.readchar #=> "8"
2619 * f.readchar #=> "1"
2623 rb_io_readchar(VALUE io
)
2625 VALUE c
= rb_io_getc(io
);
2635 * ios.getbyte => fixnum or nil
2637 * Gets the next 8-bit byte (0..255) from <em>ios</em>. Returns
2638 * <code>nil</code> if called at end of file.
2640 * f = File.new("testfile")
2646 rb_io_getbyte(VALUE io
)
2651 GetOpenFile(io
, fptr
);
2652 rb_io_check_readable(fptr
);
2654 if (fptr
->fd
== 0 && (fptr
->mode
& FMODE_TTY
) && TYPE(rb_stdout
) == T_FILE
) {
2656 GetOpenFile(rb_stdout
, ofp
);
2657 if (ofp
->mode
& FMODE_TTY
) {
2658 rb_io_flush(rb_stdout
);
2661 if (io_fillbuf(fptr
) < 0) {
2666 c
= (unsigned char)fptr
->rbuf
[fptr
->rbuf_off
-1];
2667 return INT2FIX(c
& 0xff);
2672 * ios.readbyte => fixnum
2674 * Reads a character as with <code>IO#getc</code>, but raises an
2675 * <code>EOFError</code> on end of file.
2679 rb_io_readbyte(VALUE io
)
2681 VALUE c
= rb_io_getbyte(io
);
2691 * ios.ungetc(string) => nil
2693 * Pushes back one character (passed as a parameter) onto <em>ios</em>,
2694 * such that a subsequent buffered read will return it. Only one character
2695 * may be pushed back before a subsequent read operation (that is,
2696 * you will be able to read only the last of several characters that have been pushed
2697 * back). Has no effect with unbuffered reads (such as <code>IO#sysread</code>).
2699 * f = File.new("testfile") #=> #<File:testfile>
2700 * c = f.getc #=> "8"
2701 * f.ungetc(c) #=> nil
2706 rb_io_ungetc(VALUE io
, VALUE c
)
2710 GetOpenFile(io
, fptr
);
2711 rb_io_check_readable(fptr
);
2712 if (NIL_P(c
)) return Qnil
;
2714 int cc
= FIX2INT(c
);
2715 rb_encoding
*enc
= io_read_encoding(fptr
);
2718 c
= rb_str_new(buf
, rb_enc_mbcput(cc
, buf
, enc
));
2729 * ios.isatty => true or false
2730 * ios.tty? => true or false
2732 * Returns <code>true</code> if <em>ios</em> is associated with a
2733 * terminal device (tty), <code>false</code> otherwise.
2735 * File.new("testfile").isatty #=> false
2736 * File.new("/dev/tty").isatty #=> true
2740 rb_io_isatty(VALUE io
)
2744 GetOpenFile(io
, fptr
);
2745 if (isatty(fptr
->fd
) == 0)
2752 * ios.close_on_exec? => true or false
2754 * Returns <code>true</code> if <em>ios</em> will be closed on exec.
2756 * f = open("/dev/null")
2757 * f.close_on_exec? #=> false
2758 * f.close_on_exec = true
2759 * f.close_on_exec? #=> true
2760 * f.close_on_exec = false
2761 * f.close_on_exec? #=> false
2765 rb_io_close_on_exec_p(VALUE io
)
2767 #if defined(HAVE_FCNTL) && defined(F_GETFD) && defined(F_SETFD) && defined(FD_CLOEXEC)
2772 write_io
= GetWriteIO(io
);
2773 if (io
!= write_io
) {
2774 GetOpenFile(write_io
, fptr
);
2775 if (fptr
&& 0 <= (fd
= fptr
->fd
)) {
2776 if ((ret
= fcntl(fd
, F_GETFD
)) == -1) rb_sys_fail(fptr
->path
);
2777 if (!(ret
& FD_CLOEXEC
)) return Qfalse
;
2781 GetOpenFile(io
, fptr
);
2782 if (fptr
&& 0 <= (fd
= fptr
->fd
)) {
2783 if ((ret
= fcntl(fd
, F_GETFD
)) == -1) rb_sys_fail(fptr
->path
);
2784 if (!(ret
& FD_CLOEXEC
)) return Qfalse
;
2789 return Qnil
; /* not reached */
2795 * ios.close_on_exec = bool => true or false
2797 * Sets a close-on-exec flag.
2799 * f = open("/dev/null")
2800 * f.close_on_exec = true
2801 * system("cat", "/proc/self/fd/#{f.fileno}") # cat: /proc/self/fd/3: No such file or directory
2802 * f.closed? #=> false
2806 rb_io_set_close_on_exec(VALUE io
, VALUE arg
)
2808 #if defined(HAVE_FCNTL) && defined(F_GETFD) && defined(F_SETFD) && defined(FD_CLOEXEC)
2809 int flag
= RTEST(arg
) ? FD_CLOEXEC
: 0;
2814 write_io
= GetWriteIO(io
);
2815 if (io
!= write_io
) {
2816 GetOpenFile(write_io
, fptr
);
2817 if (fptr
&& 0 <= (fd
= fptr
->fd
)) {
2818 if ((ret
= fcntl(fptr
->fd
, F_GETFD
)) == -1) rb_sys_fail(fptr
->path
);
2819 if ((ret
& FD_CLOEXEC
) != flag
) {
2820 ret
= (ret
& ~FD_CLOEXEC
) | flag
;
2821 ret
= fcntl(fd
, F_SETFD
, ret
);
2822 if (ret
== -1) rb_sys_fail(fptr
->path
);
2828 GetOpenFile(io
, fptr
);
2829 if (fptr
&& 0 <= (fd
= fptr
->fd
)) {
2830 if ((ret
= fcntl(fd
, F_GETFD
)) == -1) rb_sys_fail(fptr
->path
);
2831 if ((ret
& FD_CLOEXEC
) != flag
) {
2832 ret
= (ret
& ~FD_CLOEXEC
) | flag
;
2833 ret
= fcntl(fd
, F_SETFD
, ret
);
2834 if (ret
== -1) rb_sys_fail(fptr
->path
);
2843 #define FMODE_PREP (1<<16)
2844 #define IS_PREP_STDIO(f) ((f)->mode & FMODE_PREP)
2845 #define PREP_STDIO_NAME(f) ((f)->path)
2848 fptr_finalize(rb_io_t
*fptr
, int noraise
)
2851 if (fptr
->wbuf_len
) {
2854 if (IS_PREP_STDIO(fptr
) ||
2858 if (fptr
->stdio_file
) {
2859 if (fclose(fptr
->stdio_file
) < 0 && !noraise
) {
2860 /* fptr->stdio_file is deallocated anyway */
2861 fptr
->stdio_file
= 0;
2863 rb_sys_fail(fptr
->path
);
2866 else if (0 <= fptr
->fd
) {
2867 if (close(fptr
->fd
) < 0 && !noraise
) {
2868 if (errno
!= EBADF
) {
2869 /* fptr->fd is still not closed */
2870 rb_sys_fail(fptr
->path
);
2873 /* fptr->fd is already closed. */
2879 fptr
->stdio_file
= 0;
2880 fptr
->mode
&= ~(FMODE_READABLE
|FMODE_WRITABLE
);
2882 rb_sys_fail(fptr
->path
);
2887 rb_io_fptr_cleanup(rb_io_t
*fptr
, int noraise
)
2889 if (fptr
->finalize
) {
2890 (*fptr
->finalize
)(fptr
, noraise
);
2893 fptr_finalize(fptr
, noraise
);
2898 rb_io_fptr_finalize(rb_io_t
*fptr
)
2900 if (!fptr
) return 0;
2901 if (fptr
->refcnt
<= 0 || --fptr
->refcnt
) return 0;
2907 rb_io_fptr_cleanup(fptr
, Qtrue
);
2916 if (fptr
->readconv
) {
2917 rb_econv_close(fptr
->readconv
);
2918 fptr
->readconv
= NULL
;
2929 rb_io_close(VALUE io
)
2934 rb_io_t
*write_fptr
;
2936 write_io
= GetWriteIO(io
);
2937 if (io
!= write_io
) {
2938 write_fptr
= RFILE(write_io
)->fptr
;
2939 if (write_fptr
&& 0 <= write_fptr
->fd
) {
2940 rb_io_fptr_cleanup(write_fptr
, Qtrue
);
2944 fptr
= RFILE(io
)->fptr
;
2945 if (!fptr
) return Qnil
;
2946 if (fptr
->fd
< 0) return Qnil
;
2949 rb_io_fptr_cleanup(fptr
, Qfalse
);
2950 rb_thread_fd_close(fd
);
2953 rb_syswait(fptr
->pid
);
2964 * Closes <em>ios</em> and flushes any pending writes to the operating
2965 * system. The stream is unavailable for any further data operations;
2966 * an <code>IOError</code> is raised if such an attempt is made. I/O
2967 * streams are automatically closed when they are claimed by the
2968 * garbage collector.
2970 * If <em>ios</em> is opened by <code>IO.popen</code>,
2971 * <code>close</code> sets <code>$?</code>.
2975 rb_io_close_m(VALUE io
)
2977 if (rb_safe_level() >= 4 && !OBJ_UNTRUSTED(io
)) {
2978 rb_raise(rb_eSecurityError
, "Insecure: can't close");
2980 rb_io_check_closed(RFILE(io
)->fptr
);
2986 io_call_close(VALUE io
)
2988 return rb_funcall(io
, rb_intern("close"), 0, 0);
2994 return rb_rescue(io_call_close
, io
, 0, 0);
2999 * ios.closed? => true or false
3001 * Returns <code>true</code> if <em>ios</em> is completely closed (for
3002 * duplex streams, both reader and writer), <code>false</code>
3005 * f = File.new("testfile")
3007 * f.closed? #=> true
3008 * f = IO.popen("/bin/sh","r+")
3009 * f.close_write #=> nil
3010 * f.closed? #=> false
3011 * f.close_read #=> nil
3012 * f.closed? #=> true
3017 rb_io_closed(VALUE io
)
3021 rb_io_t
*write_fptr
;
3023 write_io
= GetWriteIO(io
);
3024 if (io
!= write_io
) {
3025 write_fptr
= RFILE(write_io
)->fptr
;
3026 if (write_fptr
&& 0 <= write_fptr
->fd
) {
3031 fptr
= RFILE(io
)->fptr
;
3032 rb_io_check_initialized(fptr
);
3033 return 0 <= fptr
->fd
? Qfalse
: Qtrue
;
3038 * ios.close_read => nil
3040 * Closes the read end of a duplex I/O stream (i.e., one that contains
3041 * both a read and a write stream, such as a pipe). Will raise an
3042 * <code>IOError</code> if the stream is not duplexed.
3044 * f = IO.popen("/bin/sh","r+")
3048 * <em>produces:</em>
3050 * prog.rb:3:in `readlines': not opened for reading (IOError)
3055 rb_io_close_read(VALUE io
)
3060 if (rb_safe_level() >= 4 && !OBJ_UNTRUSTED(io
)) {
3061 rb_raise(rb_eSecurityError
, "Insecure: can't close");
3063 GetOpenFile(io
, fptr
);
3064 if (is_socket(fptr
->fd
, fptr
->path
)) {
3068 if (shutdown(fptr
->fd
, SHUT_RD
) < 0)
3069 rb_sys_fail(fptr
->path
);
3070 fptr
->mode
&= ~FMODE_READABLE
;
3071 if (!(fptr
->mode
& FMODE_WRITABLE
))
3072 return rb_io_close(io
);
3076 write_io
= GetWriteIO(io
);
3077 if (io
!= write_io
) {
3079 fptr_finalize(fptr
, Qfalse
);
3080 GetOpenFile(write_io
, wfptr
);
3081 if (fptr
->refcnt
< LONG_MAX
) {
3083 RFILE(io
)->fptr
= wfptr
;
3084 rb_io_fptr_finalize(fptr
);
3089 if (fptr
->mode
& FMODE_WRITABLE
) {
3090 rb_raise(rb_eIOError
, "closing non-duplex IO for reading");
3092 return rb_io_close(io
);
3097 * ios.close_write => nil
3099 * Closes the write end of a duplex I/O stream (i.e., one that contains
3100 * both a read and a write stream, such as a pipe). Will raise an
3101 * <code>IOError</code> if the stream is not duplexed.
3103 * f = IO.popen("/bin/sh","r+")
3107 * <em>produces:</em>
3109 * prog.rb:3:in `write': not opened for writing (IOError)
3110 * from prog.rb:3:in `print'
3115 rb_io_close_write(VALUE io
)
3120 if (rb_safe_level() >= 4 && !OBJ_UNTRUSTED(io
)) {
3121 rb_raise(rb_eSecurityError
, "Insecure: can't close");
3123 write_io
= GetWriteIO(io
);
3124 GetOpenFile(write_io
, fptr
);
3125 if (is_socket(fptr
->fd
, fptr
->path
)) {
3129 if (shutdown(fptr
->fd
, SHUT_WR
) < 0)
3130 rb_sys_fail(fptr
->path
);
3131 fptr
->mode
&= ~FMODE_WRITABLE
;
3132 if (!(fptr
->mode
& FMODE_READABLE
))
3133 return rb_io_close(write_io
);
3137 if (fptr
->mode
& FMODE_READABLE
) {
3138 rb_raise(rb_eIOError
, "closing non-duplex IO for writing");
3141 rb_io_close(write_io
);
3142 if (io
!= write_io
) {
3143 GetOpenFile(io
, fptr
);
3144 fptr
->tied_io_for_writing
= 0;
3145 fptr
->mode
&= ~FMODE_DUPLEX
;
3152 * ios.sysseek(offset, whence=SEEK_SET) => integer
3154 * Seeks to a given <i>offset</i> in the stream according to the value
3155 * of <i>whence</i> (see <code>IO#seek</code> for values of
3156 * <i>whence</i>). Returns the new offset into the file.
3158 * f = File.new("testfile")
3159 * f.sysseek(-13, IO::SEEK_END) #=> 53
3160 * f.sysread(10) #=> "And so on."
3164 rb_io_sysseek(int argc
, VALUE
*argv
, VALUE io
)
3166 VALUE offset
, ptrname
;
3167 int whence
= SEEK_SET
;
3171 if (rb_scan_args(argc
, argv
, "11", &offset
, &ptrname
) == 2) {
3172 whence
= NUM2INT(ptrname
);
3174 pos
= NUM2OFFT(offset
);
3175 GetOpenFile(io
, fptr
);
3176 if ((fptr
->mode
& FMODE_READABLE
) && READ_DATA_BUFFERED(fptr
)) {
3177 rb_raise(rb_eIOError
, "sysseek for buffered IO");
3179 if ((fptr
->mode
& FMODE_WRITABLE
) && fptr
->wbuf_len
) {
3180 rb_warn("sysseek for buffered IO");
3182 pos
= lseek(fptr
->fd
, pos
, whence
);
3183 if (pos
== -1) rb_sys_fail(fptr
->path
);
3185 return OFFT2NUM(pos
);
3190 * ios.syswrite(string) => integer
3192 * Writes the given string to <em>ios</em> using a low-level write.
3193 * Returns the number of bytes written. Do not mix with other methods
3194 * that write to <em>ios</em> or you may get unpredictable results.
3195 * Raises <code>SystemCallError</code> on error.
3197 * f = File.new("out", "w")
3198 * f.syswrite("ABCDEF") #=> 6
3202 rb_io_syswrite(VALUE io
, VALUE str
)
3208 if (TYPE(str
) != T_STRING
)
3209 str
= rb_obj_as_string(str
);
3211 io
= GetWriteIO(io
);
3212 GetOpenFile(io
, fptr
);
3213 rb_io_check_writable(fptr
);
3215 if (fptr
->wbuf_len
) {
3216 rb_warn("syswrite for buffered IO");
3218 if (!rb_thread_fd_writable(fptr
->fd
)) {
3219 rb_io_check_closed(fptr
);
3222 n
= write(fptr
->fd
, RSTRING_PTR(str
), RSTRING_LEN(str
));
3225 if (n
== -1) rb_sys_fail(fptr
->path
);
3232 * ios.sysread(integer[, outbuf]) => string
3234 * Reads <i>integer</i> bytes from <em>ios</em> using a low-level
3235 * read and returns them as a string. Do not mix with other methods
3236 * that read from <em>ios</em> or you may get unpredictable results.
3237 * If the optional <i>outbuf</i> argument is present, it must reference
3238 * a String, which will receive the data.
3239 * Raises <code>SystemCallError</code> on error and
3240 * <code>EOFError</code> at end of file.
3242 * f = File.new("testfile")
3243 * f.sysread(16) #=> "This is line one"
3247 rb_io_sysread(int argc
, VALUE
*argv
, VALUE io
)
3253 rb_scan_args(argc
, argv
, "11", &len
, &str
);
3254 ilen
= NUM2LONG(len
);
3257 str
= rb_str_new(0, ilen
);
3262 rb_str_resize(str
, ilen
);
3264 if (ilen
== 0) return str
;
3266 GetOpenFile(io
, fptr
);
3267 rb_io_check_readable(fptr
);
3269 if (READ_DATA_BUFFERED(fptr
)) {
3270 rb_raise(rb_eIOError
, "sysread for buffered IO");
3274 rb_thread_wait_fd(fptr
->fd
);
3275 rb_io_check_closed(fptr
);
3276 if (RSTRING_LEN(str
) != ilen
) {
3277 rb_raise(rb_eRuntimeError
, "buffer string modified");
3280 n
= rb_read_internal(fptr
->fd
, RSTRING_PTR(str
), ilen
);
3283 rb_sys_fail(fptr
->path
);
3285 rb_str_set_len(str
, n
);
3286 if (n
== 0 && ilen
> 0) {
3289 rb_str_resize(str
, n
);
3296 rb_io_binmode(VALUE io
)
3300 GetOpenFile(io
, fptr
);
3301 #if defined(_WIN32) || defined(DJGPP) || defined(__CYGWIN__) || defined(__human68k__) || defined(__EMX__)
3302 if (!(fptr
->mode
& FMODE_BINMODE
) && READ_DATA_BUFFERED(fptr
)) {
3303 rb_raise(rb_eIOError
, "buffer already filled with text-mode content");
3305 if (0 <= fptr
->fd
&& setmode(fptr
->fd
, O_BINARY
) == -1)
3306 rb_sys_fail(fptr
->path
);
3308 fptr
->mode
|= FMODE_BINMODE
;
3314 * ios.binmode => ios
3316 * Puts <em>ios</em> into binary mode. This is useful only in
3317 * MS-DOS/Windows environments. Once a stream is in binary mode, it
3318 * cannot be reset to nonbinary mode.
3322 rb_io_binmode_m(VALUE io
)
3324 #if defined(_WIN32) || defined(DJGPP) || defined(__CYGWIN__) || defined(__human68k__) || defined(__EMX__)
3330 #if defined(_WIN32) || defined(DJGPP) || defined(__CYGWIN__) || defined(__human68k__) || defined(__EMX__)
3331 write_io
= GetWriteIO(io
);
3333 rb_io_binmode(write_io
);
3340 * ios.binmode? => true or false
3342 * Returns <code>true</code> if <em>ios</em> is binmode.
3345 rb_io_binmode_p(VALUE io
)
3348 GetOpenFile(io
, fptr
);
3349 return fptr
->mode
& FMODE_BINMODE
? Qtrue
: Qfalse
;
3353 rb_io_flags_mode(int flags
)
3356 # define MODE_BINMODE(a,b) ((flags & FMODE_BINMODE) ? (b) : (a))
3358 # define MODE_BINMODE(a,b) (a)
3360 if (flags
& FMODE_APPEND
) {
3361 if ((flags
& FMODE_READWRITE
) == FMODE_READWRITE
) {
3362 return MODE_BINMODE("a+", "ab+");
3364 return MODE_BINMODE("a", "ab");
3366 switch (flags
& FMODE_READWRITE
) {
3367 case FMODE_READABLE
:
3368 return MODE_BINMODE("r", "rb");
3369 case FMODE_WRITABLE
:
3370 return MODE_BINMODE("w", "wb");
3371 case FMODE_READWRITE
:
3372 if (flags
& FMODE_CREATE
) {
3373 return MODE_BINMODE("w+", "wb+");
3375 return MODE_BINMODE("r+", "rb+");
3377 rb_raise(rb_eArgError
, "invalid access modenum %o", flags
);
3378 return NULL
; /* not reached */
3382 rb_io_mode_flags(const char *mode
)
3385 const char *m
= mode
;
3389 flags
|= FMODE_READABLE
;
3392 flags
|= FMODE_WRITABLE
| FMODE_CREATE
;
3395 flags
|= FMODE_WRITABLE
| FMODE_APPEND
| FMODE_CREATE
;
3399 rb_raise(rb_eArgError
, "invalid access mode %s", mode
);
3405 flags
|= FMODE_BINMODE
;
3408 flags
|= FMODE_READWRITE
;
3421 rb_io_modenum_flags(int mode
)
3425 switch (mode
& (O_RDONLY
|O_WRONLY
|O_RDWR
)) {
3427 flags
= FMODE_READABLE
;
3430 flags
= FMODE_WRITABLE
;
3433 flags
= FMODE_READWRITE
;
3437 if (mode
& O_APPEND
) {
3438 flags
|= FMODE_APPEND
;
3440 if (mode
& O_CREAT
) {
3441 flags
|= FMODE_CREATE
;
3444 if (mode
& O_BINARY
) {
3445 flags
|= FMODE_BINMODE
;
3453 rb_io_mode_modenum(const char *mode
)
3456 const char *m
= mode
;
3463 flags
|= O_WRONLY
| O_CREAT
| O_TRUNC
;
3466 flags
|= O_WRONLY
| O_CREAT
| O_APPEND
;
3470 rb_raise(rb_eArgError
, "invalid access mode %s", mode
);
3481 flags
= (flags
& ~O_ACCMODE
) | O_RDWR
;
3493 #define MODENUM_MAX 4
3496 rb_io_modenum_mode(int flags
)
3499 # define MODE_BINARY(a,b) ((flags & O_BINARY) ? (b) : (a))
3501 # define MODE_BINARY(a,b) (a)
3503 if (flags
& O_APPEND
) {
3504 if ((flags
& O_RDWR
) == O_RDWR
) {
3505 return MODE_BINARY("a+", "ab+");
3507 return MODE_BINARY("a", "ab");
3509 switch (flags
& (O_RDONLY
|O_WRONLY
|O_RDWR
)) {
3511 return MODE_BINARY("r", "rb");
3513 return MODE_BINARY("w", "wb");
3515 return MODE_BINARY("r+", "rb+");
3517 rb_raise(rb_eArgError
, "invalid access modenum %o", flags
);
3518 return NULL
; /* not reached */
3522 mode_enc(rb_io_t
*fptr
, const char *estr
)
3524 const char *p0
, *p1
;
3528 /* parse estr as "enc" or "enc2:enc" */
3530 p0
= strrchr(estr
, ':');
3533 idx
= rb_enc_find_index(p1
);
3535 fptr
->enc
= rb_enc_from_index(idx
);
3538 rb_warn("Unsupported encoding %s ignored", p1
);
3543 if (n
> ENCODING_MAXNAMELEN
) {
3547 enc2name
= ALLOCA_N(char, n
+1);
3548 memcpy(enc2name
, estr
, n
);
3551 idx2
= rb_enc_find_index(enc2name
);
3554 rb_warn("Unsupported encoding %.*s ignored", n
, estr
);
3556 else if (idx2
== idx
) {
3557 rb_warn("Ignoring internal encoding %.*s: it is identical to external encoding %s",
3561 fptr
->enc2
= rb_enc_from_index(idx2
);
3567 rb_io_mode_enc(rb_io_t
*fptr
, const char *mode
)
3569 const char *p
= strchr(mode
, ':');
3571 mode_enc(fptr
, p
+1);
3575 struct sysopen_struct
{
3582 sysopen_func(void *ptr
)
3584 struct sysopen_struct
*data
= ptr
;
3585 return (VALUE
)open(data
->fname
, data
->flag
, data
->mode
);
3589 rb_sysopen_internal(char *fname
, int flags
, unsigned int mode
)
3591 struct sysopen_struct data
;
3595 return (int)rb_thread_blocking_region(sysopen_func
, &data
, RB_UBF_DFL
, 0);
3599 rb_sysopen(char *fname
, int flags
, unsigned int mode
)
3603 fd
= rb_sysopen_internal(fname
, flags
, mode
);
3605 if (errno
== EMFILE
|| errno
== ENFILE
) {
3607 fd
= rb_sysopen_internal(fname
, flags
, mode
);
3618 rb_fopen(const char *fname
, const char *mode
)
3622 file
= fopen(fname
, mode
);
3624 if (errno
== EMFILE
|| errno
== ENFILE
) {
3626 file
= fopen(fname
, mode
);
3633 if (setvbuf(file
, NULL
, _IOFBF
, 0) != 0)
3634 rb_warn("setvbuf() can't be honoured for %s", fname
);
3637 setmode(fileno(file
), O_TEXT
);
3643 rb_fdopen(int fd
, const char *mode
)
3650 file
= fdopen(fd
, mode
);
3656 errno
== EMFILE
|| errno
== ENFILE
) {
3661 file
= fdopen(fd
, mode
);
3665 if (errno
== 0) errno
= EINVAL
;
3667 if (errno
== 0) errno
= EMFILE
;
3673 /* xxx: should be _IONBF? A buffer in FILE may have trouble. */
3675 if (setvbuf(file
, NULL
, _IOFBF
, 0) != 0)
3676 rb_warn("setvbuf() can't be honoured (fd=%d)", fd
);
3682 io_check_tty(rb_io_t
*fptr
)
3684 if (isatty(fptr
->fd
))
3685 fptr
->mode
|= FMODE_TTY
|FMODE_DUPLEX
;
3689 rb_file_open_internal(VALUE io
, const char *fname
, const char *mode
)
3693 MakeOpenFile(io
, fptr
);
3694 fptr
->mode
= rb_io_mode_flags(mode
);
3695 rb_io_mode_enc(fptr
, mode
);
3696 fptr
->path
= strdup(fname
);
3697 fptr
->fd
= rb_sysopen(fptr
->path
, rb_io_mode_modenum(rb_io_flags_mode(fptr
->mode
)), 0666);
3704 rb_file_open(const char *fname
, const char *mode
)
3706 return rb_file_open_internal(io_alloc(rb_cFile
), fname
, mode
);
3710 rb_file_sysopen_internal(VALUE io
, const char *fname
, int flags
, int mode
)
3714 MakeOpenFile(io
, fptr
);
3716 fptr
->path
= strdup(fname
);
3717 fptr
->mode
= rb_io_modenum_flags(flags
);
3718 fptr
->fd
= rb_sysopen(fptr
->path
, flags
, mode
);
3725 rb_file_sysopen(const char *fname
, int flags
, int mode
)
3727 return rb_file_sysopen_internal(io_alloc(rb_cFile
), fname
, flags
, mode
);
3730 #if defined(__CYGWIN__) || !defined(HAVE_FORK)
3731 static struct pipe_list
{
3733 struct pipe_list
*next
;
3737 pipe_add_fptr(rb_io_t
*fptr
)
3739 struct pipe_list
*list
;
3741 list
= ALLOC(struct pipe_list
);
3743 list
->next
= pipe_list
;
3748 pipe_del_fptr(rb_io_t
*fptr
)
3750 struct pipe_list
*list
= pipe_list
;
3751 struct pipe_list
*tmp
;
3753 if (list
->fptr
== fptr
) {
3754 pipe_list
= list
->next
;
3759 while (list
->next
) {
3760 if (list
->next
->fptr
== fptr
) {
3762 list
->next
= list
->next
->next
;
3773 struct pipe_list
*list
= pipe_list
;
3774 struct pipe_list
*tmp
;
3778 rb_io_fptr_finalize(list
->fptr
);
3784 pipe_finalize(rb_io_t
*fptr
, int noraise
)
3786 #if !defined(HAVE_FORK) && !defined(_WIN32)
3788 if (fptr
->stdio_file
) {
3789 status
= pclose(fptr
->stdio_file
);
3792 fptr
->stdio_file
= 0;
3796 rb_last_status_set(status
, fptr
->pid
);
3798 fptr_finalize(fptr
, noraise
);
3800 pipe_del_fptr(fptr
);
3805 rb_io_synchronized(rb_io_t
*fptr
)
3807 rb_io_check_initialized(fptr
);
3808 fptr
->mode
|= FMODE_SYNC
;
3812 rb_io_unbuffered(rb_io_t
*fptr
)
3814 rb_io_synchronized(fptr
);
3823 if (errno
== EMFILE
|| errno
== ENFILE
) {
3829 UPDATE_MAXFD(pipes
[0]);
3830 UPDATE_MAXFD(pipes
[1]);
3837 struct rb_exec_arg
*execp
;
3844 popen_redirect(struct popen_arg
*p
)
3846 if ((p
->modef
& FMODE_READABLE
) && (p
->modef
& FMODE_WRITABLE
)) {
3847 close(p
->write_pair
[1]);
3848 if (p
->write_pair
[0] != 0) {
3849 dup2(p
->write_pair
[0], 0);
3850 close(p
->write_pair
[0]);
3853 if (p
->pair
[1] != 1) {
3854 dup2(p
->pair
[1], 1);
3858 else if (p
->modef
& FMODE_READABLE
) {
3860 if (p
->pair
[1] != 1) {
3861 dup2(p
->pair
[1], 1);
3867 if (p
->pair
[0] != 0) {
3868 dup2(p
->pair
[0], 0);
3875 rb_close_before_exec(int lowfd
, int maxhint
, VALUE noclose_fds
)
3878 int max
= max_file_descriptor
;
3881 for (fd
= lowfd
; fd
<= max
; fd
++) {
3882 if (!NIL_P(noclose_fds
) &&
3883 RTEST(rb_hash_lookup(noclose_fds
, INT2FIX(fd
))))
3886 ret
= fcntl(fd
, F_GETFD
);
3887 if (ret
!= -1 && !(ret
& FD_CLOEXEC
)) {
3888 fcntl(fd
, F_SETFD
, ret
|FD_CLOEXEC
);
3897 popen_exec(void *pp
)
3899 struct popen_arg
*p
= (struct popen_arg
*)pp
;
3901 rb_thread_atfork_before_exec();
3902 return rb_exec(p
->execp
);
3907 pipe_open(struct rb_exec_arg
*eargp
, VALUE prog
, const char *mode
)
3909 int modef
= rb_io_mode_flags(mode
);
3913 rb_io_t
*write_fptr
;
3915 #if defined(HAVE_FORK)
3917 struct popen_arg arg
;
3918 #elif defined(_WIN32)
3919 int openmode
= rb_io_mode_modenum(mode
);
3920 const char *exename
= NULL
;
3921 volatile VALUE cmdbuf
;
3922 struct rb_exec_arg sarg
;
3927 const char *cmd
= 0;
3932 cmd
= StringValueCStr(prog
);
3935 /* fork : IO.popen("-") */
3939 else if (eargp
->argc
) {
3940 /* no shell : IO.popen([prog, arg0], arg1, ...) */
3945 /* with shell : IO.popen(prog) */
3950 #if defined(HAVE_FORK)
3953 arg
.pair
[0] = arg
.pair
[1] = -1;
3954 arg
.write_pair
[0] = arg
.write_pair
[1] = -1;
3955 switch (modef
& (FMODE_READABLE
|FMODE_WRITABLE
)) {
3956 case FMODE_READABLE
|FMODE_WRITABLE
:
3957 if (rb_pipe(arg
.write_pair
) < 0)
3959 if (rb_pipe(arg
.pair
) < 0) {
3961 close(arg
.write_pair
[0]);
3962 close(arg
.write_pair
[1]);
3967 rb_exec_arg_addopt(eargp
, INT2FIX(0), INT2FIX(arg
.write_pair
[0]));
3968 rb_exec_arg_addopt(eargp
, INT2FIX(1), INT2FIX(arg
.pair
[1]));
3971 case FMODE_READABLE
:
3972 if (rb_pipe(arg
.pair
) < 0)
3975 rb_exec_arg_addopt(eargp
, INT2FIX(1), INT2FIX(arg
.pair
[1]));
3977 case FMODE_WRITABLE
:
3978 if (rb_pipe(arg
.pair
) < 0)
3981 rb_exec_arg_addopt(eargp
, INT2FIX(0), INT2FIX(arg
.pair
[0]));
3987 rb_exec_arg_fixup(arg
.execp
);
3988 pid
= rb_fork(&status
, popen_exec
, &arg
, arg
.execp
->redirect_fds
);
3991 fflush(stdin
); /* is it really needed? */
3992 rb_io_flush(rb_stdout
);
3993 rb_io_flush(rb_stderr
);
3994 pid
= rb_fork(&status
, 0, 0, Qnil
);
3995 if (pid
== 0) { /* child */
3996 popen_redirect(&arg
);
3997 rb_io_synchronized(RFILE(orig_stdout
)->fptr
);
3998 rb_io_synchronized(RFILE(orig_stderr
)->fptr
);
4008 if ((modef
& (FMODE_READABLE
|FMODE_WRITABLE
)) == (FMODE_READABLE
|FMODE_WRITABLE
)) {
4009 close(arg
.write_pair
[0]);
4010 close(arg
.write_pair
[1]);
4015 if ((modef
& FMODE_READABLE
) && (modef
& FMODE_WRITABLE
)) {
4018 close(arg
.write_pair
[0]);
4019 write_fd
= arg
.write_pair
[1];
4021 else if (modef
& FMODE_READABLE
) {
4029 #elif defined(_WIN32)
4031 volatile VALUE argbuf
;
4035 if (argc
>= FIXNUM_MAX
/ sizeof(char *)) {
4036 rb_raise(rb_eArgError
, "too many arguments");
4038 argbuf
= rb_str_tmp_new((argc
+1) * sizeof(char *));
4039 args
= (void *)RSTRING_PTR(argbuf
);
4040 for (i
= 0; i
< argc
; ++i
) {
4041 args
[i
] = StringValueCStr(argv
[i
]);
4045 cmdbuf
= rb_str_tmp_new(rb_w32_argv_size(args
));
4046 cmd
= rb_w32_join_argv(RSTRING_PTR(cmdbuf
), args
);
4047 rb_str_resize(argbuf
, 0);
4050 rb_exec_arg_fixup(eargp
);
4051 rb_run_exec_options(eargp
, &sarg
);
4053 while ((pid
= rb_w32_pipe_exec(cmd
, exename
, openmode
, &fd
, &write_fd
)) == -1) {
4057 #if defined(EWOULDBLOCK) && EWOULDBLOCK != EAGAIN
4064 rb_run_exec_options(&sarg
, NULL
);
4070 rb_run_exec_options(&sarg
, NULL
);
4073 prog
= rb_ary_join(rb_ary_new4(argc
, argv
), rb_str_new2(" "));
4074 cmd
= StringValueCStr(prog
);
4077 rb_exec_arg_fixup(eargp
);
4078 rb_run_exec_options(eargp
, &sarg
);
4080 fp
= popen(cmd
, mode
);
4082 rb_run_exec_options(&sarg
, NULL
);
4083 if (!fp
) rb_sys_fail(RSTRING_PTR(prog
));
4087 port
= io_alloc(rb_cIO
);
4088 MakeOpenFile(port
, fptr
);
4090 fptr
->stdio_file
= fp
;
4091 fptr
->mode
= modef
| FMODE_SYNC
|FMODE_DUPLEX
;
4092 rb_io_mode_enc(fptr
, mode
);
4095 if (0 <= write_fd
) {
4096 write_port
= io_alloc(rb_cIO
);
4097 MakeOpenFile(write_port
, write_fptr
);
4098 write_fptr
->fd
= write_fd
;
4099 write_fptr
->mode
= (modef
& ~FMODE_READABLE
)| FMODE_SYNC
|FMODE_DUPLEX
;
4100 fptr
->mode
&= ~FMODE_WRITABLE
;
4101 fptr
->tied_io_for_writing
= write_port
;
4102 rb_ivar_set(port
, rb_intern("@tied_io_for_writing"), write_port
);
4105 #if defined (__CYGWIN__) || !defined(HAVE_FORK)
4106 fptr
->finalize
= pipe_finalize
;
4107 pipe_add_fptr(fptr
);
4113 pipe_open_v(int argc
, VALUE
*argv
, const char *mode
)
4116 struct rb_exec_arg earg
;
4117 prog
= rb_exec_arg_init(argc
, argv
, Qfalse
, &earg
);
4118 return pipe_open(&earg
, prog
, mode
);
4122 pipe_open_s(VALUE prog
, const char *mode
)
4124 const char *cmd
= RSTRING_PTR(prog
);
4126 VALUE
*argv
= &prog
;
4127 struct rb_exec_arg earg
;
4129 if (RSTRING_LEN(prog
) == 1 && cmd
[0] == '-') {
4130 #if !defined(HAVE_FORK)
4131 rb_raise(rb_eNotImpError
,
4132 "fork() function is unimplemented on this machine");
4134 return pipe_open(0, 0, mode
);
4137 rb_exec_arg_init(argc
, argv
, Qtrue
, &earg
);
4138 return pipe_open(&earg
, prog
, mode
);
4143 * IO.popen(cmd, mode="r") => io
4144 * IO.popen(cmd, mode="r") {|io| block } => obj
4146 * Runs the specified command as a subprocess; the subprocess's
4147 * standard input and output will be connected to the returned
4148 * <code>IO</code> object. If _cmd_ is a +String+
4149 * ``<code>-</code>'', then a new instance of Ruby is started as the
4150 * subprocess. If <i>cmd</i> is an +Array+ of +String+, then it will
4151 * be used as the subprocess's +argv+ bypassing a shell.
4152 * The array can contains a hash at first for environments and
4153 * a hash at last for options similar to <code>spawn</code>. The default
4154 * mode for the new file object is ``r'', but <i>mode</i> may be set
4155 * to any of the modes listed in the description for class IO.
4157 * Raises exceptions which <code>IO::pipe</code> and
4158 * <code>Kernel::system</code> raise.
4160 * If a block is given, Ruby will run the command as a child connected
4161 * to Ruby with a pipe. Ruby's end of the pipe will be passed as a
4162 * parameter to the block.
4163 * At the end of block, Ruby close the pipe and sets <code>$?</code>.
4164 * In this case <code>IO::popen</code> returns
4165 * the value of the block.
4167 * If a block is given with a _cmd_ of ``<code>-</code>'',
4168 * the block will be run in two separate processes: once in the parent,
4169 * and once in a child. The parent process will be passed the pipe
4170 * object as a parameter to the block, the child version of the block
4171 * will be passed <code>nil</code>, and the child's standard in and
4172 * standard out will be connected to the parent through the pipe. Not
4173 * available on all platforms.
4175 * f = IO.popen("uname")
4177 * puts "Parent is #{Process.pid}"
4178 * IO.popen("date") { |f| puts f.gets }
4179 * IO.popen("-") {|f| $stderr.puts "#{Process.pid} is here, f is #{f}"}
4181 * IO.popen(%w"sed -e s|^|<foo>| -e s&$&;zot;&", "r+") {|f|
4182 * f.puts "bar"; f.close_write; puts f.gets
4185 * <em>produces:</em>
4189 * Wed Apr 9 08:53:52 CDT 2003
4190 * 26169 is here, f is
4191 * 26166 is here, f is #<IO:0x401b3d44>
4192 * #<Process::Status: pid=26166,exited(0)>
4197 rb_io_s_popen(int argc
, VALUE
*argv
, VALUE klass
)
4200 VALUE pname
, pmode
, port
, tmp
;
4202 if (rb_scan_args(argc
, argv
, "11", &pname
, &pmode
) == 1) {
4205 else if (FIXNUM_P(pmode
)) {
4206 mode
= rb_io_modenum_mode(FIX2INT(pmode
));
4209 mode
= StringValueCStr(pmode
);
4211 tmp
= rb_check_array_type(pname
);
4213 tmp
= rb_ary_dup(tmp
);
4214 RBASIC(tmp
)->klass
= 0;
4215 port
= pipe_open_v(RARRAY_LEN(tmp
), RARRAY_PTR(tmp
), mode
);
4219 SafeStringValue(pname
);
4220 port
= pipe_open_s(pname
, mode
);
4224 if (rb_block_given_p()) {
4226 rb_io_flush(rb_stdout
);
4227 rb_io_flush(rb_stderr
);
4232 RBASIC(port
)->klass
= klass
;
4233 if (rb_block_given_p()) {
4234 return rb_ensure(rb_yield
, port
, io_close
, port
);
4240 io_set_encoding(VALUE io
, VALUE opt
)
4243 VALUE encoding
=Qnil
, extenc
=Qnil
, intenc
=Qnil
;
4246 v
= rb_hash_aref(opt
, sym_encoding
);
4247 if (!NIL_P(v
)) encoding
= v
;
4248 v
= rb_hash_aref(opt
, sym_extenc
);
4249 if (!NIL_P(v
)) extenc
= v
;
4250 v
= rb_hash_aref(opt
, sym_intenc
);
4251 if (!NIL_P(v
)) intenc
= v
;
4253 if (!NIL_P(extenc
)) {
4254 rb_encoding
*extencoding
= rb_to_encoding(extenc
);
4255 GetOpenFile(io
, fptr
);
4256 if (!NIL_P(encoding
)) {
4257 rb_warn("Ignoring encoding parameter '%s': external_encoding is used",
4258 RSTRING_PTR(encoding
));
4260 if (!NIL_P(intenc
)) {
4261 rb_encoding
*intencoding
= rb_to_encoding(intenc
);
4262 if (extencoding
== intencoding
) {
4263 rb_warn("Ignoring internal encoding '%s': it is identical to external encoding '%s'",
4264 RSTRING_PTR(rb_inspect(intenc
)),
4265 RSTRING_PTR(rb_inspect(extenc
)));
4268 fptr
->enc2
= intencoding
;
4271 fptr
->enc
= extencoding
;
4274 if (!NIL_P(intenc
)) {
4275 rb_raise(rb_eArgError
, "External encoding must be specified when internal encoding is given");
4277 if (!NIL_P(encoding
)) {
4278 GetOpenFile(io
, fptr
);
4279 mode_enc(fptr
, StringValueCStr(encoding
));
4285 rb_open_file(int argc
, VALUE
*argv
, VALUE io
)
4287 VALUE opt
, fname
, vmode
, perm
;
4292 opt
= rb_check_convert_type(argv
[argc
-1], T_HASH
, "Hash", "to_hash");
4295 v
= rb_hash_aref(opt
, sym_mode
);
4296 if (!NIL_P(v
)) vmode
= v
;
4297 v
= rb_hash_aref(opt
, sym_perm
);
4298 if (!NIL_P(v
)) perm
= v
;
4302 rb_scan_args(argc
, argv
, "12", &fname
, &vmode
, &perm
);
4303 #if defined _WIN32 || defined __APPLE__
4305 static rb_encoding
*fs_encoding
;
4306 rb_encoding
*fname_encoding
= rb_enc_get(fname
);
4308 fs_encoding
= rb_filesystem_encoding();
4309 if (rb_usascii_encoding() != fname_encoding
4310 && rb_ascii8bit_encoding() != fname_encoding
4311 #if defined __APPLE__
4312 && rb_utf8_encoding() != fname_encoding
4314 && fs_encoding
!= fname_encoding
) {
4315 static VALUE fs_enc
;
4317 fs_enc
= rb_enc_from_encoding(fs_encoding
);
4318 fname
= rb_str_transcode(fname
, fs_enc
);
4322 FilePathValue(fname
);
4324 if (FIXNUM_P(vmode
) || !NIL_P(perm
)) {
4325 if (FIXNUM_P(vmode
)) {
4326 flags
= FIX2INT(vmode
);
4329 SafeStringValue(vmode
);
4330 flags
= rb_io_mode_modenum(StringValueCStr(vmode
));
4332 fmode
= NIL_P(perm
) ? 0666 : NUM2UINT(perm
);
4334 rb_file_sysopen_internal(io
, RSTRING_PTR(fname
), flags
, fmode
);
4337 mode
= NIL_P(vmode
) ? "r" : StringValueCStr(vmode
);
4338 rb_file_open_internal(io
, RSTRING_PTR(fname
), mode
);
4341 io_set_encoding(io
, opt
);
4347 * IO.open(fd, mode_string="r" ) => io
4348 * IO.open(fd, mode_string="r" ) {|io| block } => obj
4350 * With no associated block, <code>open</code> is a synonym for
4351 * <code>IO::new</code>. If the optional code block is given, it will
4352 * be passed <i>io</i> as an argument, and the IO object will
4353 * automatically be closed when the block terminates. In this instance,
4354 * <code>IO::open</code> returns the value of the block.
4359 rb_io_s_open(int argc
, VALUE
*argv
, VALUE klass
)
4361 VALUE io
= rb_class_new_instance(argc
, argv
, klass
);
4363 if (rb_block_given_p()) {
4364 return rb_ensure(rb_yield
, io
, io_close
, io
);
4372 * IO.sysopen(path, [mode, [perm]]) => fixnum
4374 * Opens the given path, returning the underlying file descriptor as a
4375 * <code>Fixnum</code>.
4377 * IO.sysopen("testfile") #=> 3
4382 rb_io_s_sysopen(int argc
, VALUE
*argv
)
4384 VALUE fname
, vmode
, perm
;
4389 rb_scan_args(argc
, argv
, "12", &fname
, &vmode
, &perm
);
4390 FilePathValue(fname
);
4392 if (NIL_P(vmode
)) flags
= O_RDONLY
;
4393 else if (FIXNUM_P(vmode
)) flags
= FIX2INT(vmode
);
4395 SafeStringValue(vmode
);
4396 flags
= rb_io_mode_modenum(StringValueCStr(vmode
));
4398 if (NIL_P(perm
)) fmode
= 0666;
4399 else fmode
= NUM2UINT(perm
);
4401 RB_GC_GUARD(fname
) = rb_str_new4(fname
);
4402 path
= RSTRING_PTR(fname
);
4403 fd
= rb_sysopen(path
, flags
, fmode
);
4409 * open(path [, mode_enc [, perm]] ) => io or nil
4410 * open(path [, mode_enc [, perm]] ) {|io| block } => obj
4412 * Creates an <code>IO</code> object connected to the given stream,
4413 * file, or subprocess.
4415 * If <i>path</i> does not start with a pipe character
4416 * (``<code>|</code>''), treat it as the name of a file to open using
4417 * the specified mode (defaulting to ``<code>r</code>'').
4420 * either a string or an integer. If it is an integer, it must be
4421 * bitwise-or of open(2) flags, such as File::RDWR or File::EXCL.
4422 * If it is a string, it is either "mode", "mode:ext_enc", or
4423 * "mode:ext_enc:int_enc".
4424 * The mode is one of the following:
4430 * The mode can be followed by "b" (means binary-mode), or "+"
4431 * (means both reading and writing allowed) or both.
4432 * If ext_enc (external encoding) is specified,
4433 * read string will be tagged by the encoding in reading,
4434 * and output string will be converted
4435 * to the specified encoding in writing.
4436 * If two encoding names,
4437 * ext_enc and int_enc (external encoding and internal encoding),
4438 * are specified, the read string is converted from ext_enc
4439 * to int_enc then tagged with the int_enc in read mode,
4440 * and in write mode, the output string will be
4441 * converted from int_enc to ext_enc before writing.
4443 * If a file is being created, its initial permissions may be
4444 * set using the integer third parameter.
4446 * If a block is specified, it will be invoked with the
4447 * <code>File</code> object as a parameter, and the file will be
4448 * automatically closed when the block terminates. The call
4449 * returns the value of the block.
4451 * If <i>path</i> starts with a pipe character, a subprocess is
4452 * created, connected to the caller by a pair of pipes. The returned
4453 * <code>IO</code> object may be used to write to the standard input
4454 * and read from the standard output of this subprocess. If the command
4455 * following the ``<code>|</code>'' is a single minus sign, Ruby forks,
4456 * and this subprocess is connected to the parent. In the subprocess,
4457 * the <code>open</code> call returns <code>nil</code>. If the command
4458 * is not ``<code>-</code>'', the subprocess runs the command. If a
4459 * block is associated with an <code>open("|-")</code> call, that block
4460 * will be run twice---once in the parent and once in the child. The
4461 * block parameter will be an <code>IO</code> object in the parent and
4462 * <code>nil</code> in the child. The parent's <code>IO</code> object
4463 * will be connected to the child's <code>$stdin</code> and
4464 * <code>$stdout</code>. The subprocess will be terminated at the end
4467 * open("testfile") do |f|
4471 * <em>produces:</em>
4475 * Open a subprocess and read its output:
4477 * cmd = open("|date")
4481 * <em>produces:</em>
4483 * Wed Apr 9 08:56:31 CDT 2003
4485 * Open a subprocess running the same Ruby program:
4487 * f = open("|-", "w+")
4492 * puts "Got: #{f.gets}"
4495 * <em>produces:</em>
4499 * Open a subprocess using a block to receive the I/O object:
4505 * puts "Got: #{f.gets}"
4509 * <em>produces:</em>
4515 rb_f_open(int argc
, VALUE
*argv
)
4518 int redirect
= Qfalse
;
4521 CONST_ID(to_open
, "to_open");
4522 if (rb_respond_to(argv
[0], to_open
)) {
4526 VALUE tmp
= argv
[0];
4532 char *str
= StringValuePtr(tmp
);
4533 if (str
&& str
[0] == '|') {
4534 argv
[0] = rb_str_new(str
+1, RSTRING_LEN(tmp
)-1);
4535 OBJ_INFECT(argv
[0], tmp
);
4536 return rb_io_s_popen(argc
, argv
, rb_cIO
);
4542 VALUE io
= rb_funcall2(argv
[0], to_open
, argc
-1, argv
+1);
4544 if (rb_block_given_p()) {
4545 return rb_ensure(rb_yield
, io
, io_close
, io
);
4549 return rb_io_s_open(argc
, argv
, rb_cFile
);
4553 rb_io_open(const char *fname
, const char *mode
)
4555 if (fname
[0] == '|') {
4556 VALUE cmd
= rb_str_new2(fname
+1);
4557 return pipe_open_s(cmd
, mode
);
4560 return rb_file_open(fname
, mode
);
4565 rb_io_open_with_args(int argc
, VALUE
*argv
)
4570 if (rb_scan_args(argc
, argv
, "11", &pname
, &pmode
) == 1) {
4573 else if (FIXNUM_P(pmode
)) {
4574 mode
= rb_io_modenum_mode(FIX2INT(pmode
));
4577 mode
= StringValueCStr(pmode
);
4579 return rb_io_open(StringValueCStr(pname
), mode
);
4583 io_reopen(VALUE io
, VALUE nfile
)
4585 rb_io_t
*fptr
, *orig
;
4589 nfile
= rb_io_get_io(nfile
);
4590 if (rb_safe_level() >= 4 &&
4591 (!OBJ_UNTRUSTED(io
) || !OBJ_UNTRUSTED(nfile
))) {
4592 rb_raise(rb_eSecurityError
, "Insecure: can't reopen");
4594 GetOpenFile(io
, fptr
);
4595 GetOpenFile(nfile
, orig
);
4597 if (fptr
== orig
) return io
;
4598 if (IS_PREP_STDIO(fptr
)) {
4599 if ((fptr
->stdio_file
== stdin
&& !(orig
->mode
& FMODE_READABLE
)) ||
4600 (fptr
->stdio_file
== stdout
&& !(orig
->mode
& FMODE_WRITABLE
)) ||
4601 (fptr
->stdio_file
== stderr
&& !(orig
->mode
& FMODE_WRITABLE
))) {
4602 rb_raise(rb_eArgError
,
4603 "%s can't change access mode from \"%s\" to \"%s\"",
4604 PREP_STDIO_NAME(fptr
), rb_io_flags_mode(fptr
->mode
),
4605 rb_io_flags_mode(orig
->mode
));
4608 if (orig
->mode
& FMODE_READABLE
) {
4609 pos
= io_tell(orig
);
4611 if (orig
->mode
& FMODE_WRITABLE
) {
4614 if (fptr
->mode
& FMODE_WRITABLE
) {
4618 /* copy rb_io_t structure */
4619 fptr
->mode
= orig
->mode
| (fptr
->mode
& FMODE_PREP
);
4620 fptr
->pid
= orig
->pid
;
4621 fptr
->lineno
= orig
->lineno
;
4622 if (fptr
->path
) free(fptr
->path
);
4623 if (orig
->path
) fptr
->path
= strdup(orig
->path
);
4624 else fptr
->path
= 0;
4625 fptr
->finalize
= orig
->finalize
;
4630 if (IS_PREP_STDIO(fptr
)) {
4631 /* need to keep stdio objects */
4632 if (dup2(fd2
, fd
) < 0)
4633 rb_sys_fail(orig
->path
);
4636 if (fptr
->stdio_file
)
4637 fclose(fptr
->stdio_file
);
4640 fptr
->stdio_file
= 0;
4642 if (dup2(fd2
, fd
) < 0)
4643 rb_sys_fail(orig
->path
);
4646 rb_thread_fd_close(fd
);
4647 if ((orig
->mode
& FMODE_READABLE
) && pos
>= 0) {
4648 if (io_seek(fptr
, pos
, SEEK_SET
) < 0) {
4649 rb_sys_fail(fptr
->path
);
4651 if (io_seek(orig
, pos
, SEEK_SET
) < 0) {
4652 rb_sys_fail(orig
->path
);
4657 if (fptr
->mode
& FMODE_BINMODE
) {
4661 RBASIC(io
)->klass
= rb_obj_class(nfile
);
4667 * ios.reopen(other_IO) => ios
4668 * ios.reopen(path, mode_str) => ios
4670 * Reassociates <em>ios</em> with the I/O stream given in
4671 * <i>other_IO</i> or to a new stream opened on <i>path</i>. This may
4672 * dynamically change the actual class of this stream.
4674 * f1 = File.new("testfile")
4675 * f2 = File.new("testfile")
4676 * f2.readlines[0] #=> "This is line one\n"
4677 * f2.reopen(f1) #=> #<File:testfile>
4678 * f2.readlines[0] #=> "This is line one\n"
4682 rb_io_reopen(int argc
, VALUE
*argv
, VALUE file
)
4689 if (rb_scan_args(argc
, argv
, "11", &fname
, &nmode
) == 1) {
4690 VALUE tmp
= rb_io_check_io(fname
);
4692 return io_reopen(file
, tmp
);
4696 FilePathValue(fname
);
4697 rb_io_taint_check(file
);
4698 fptr
= RFILE(file
)->fptr
;
4700 fptr
= RFILE(file
)->fptr
= ALLOC(rb_io_t
);
4701 MEMZERO(fptr
, rb_io_t
, 1);
4704 if (!NIL_P(nmode
)) {
4705 int flags
= rb_io_mode_flags(StringValueCStr(nmode
));
4706 if (IS_PREP_STDIO(fptr
) &&
4707 ((fptr
->mode
& FMODE_READWRITE
) & (flags
& FMODE_READWRITE
)) !=
4708 (fptr
->mode
& FMODE_READWRITE
)) {
4709 rb_raise(rb_eArgError
,
4710 "%s can't change access mode from \"%s\" to \"%s\"",
4711 PREP_STDIO_NAME(fptr
), rb_io_flags_mode(fptr
->mode
),
4712 rb_io_flags_mode(flags
));
4715 rb_io_mode_enc(fptr
, StringValueCStr(nmode
));
4723 fptr
->path
= strdup(StringValueCStr(fname
));
4724 mode
= rb_io_flags_mode(fptr
->mode
);
4726 fptr
->fd
= rb_sysopen(fptr
->path
, rb_io_mode_modenum(mode
), 0666);
4727 fptr
->stdio_file
= 0;
4731 if (fptr
->mode
& FMODE_WRITABLE
) {
4734 fptr
->rbuf_off
= fptr
->rbuf_len
= 0;
4736 if (fptr
->stdio_file
) {
4737 if (freopen(fptr
->path
, mode
, fptr
->stdio_file
) == 0) {
4738 rb_sys_fail(fptr
->path
);
4740 fptr
->fd
= fileno(fptr
->stdio_file
);
4742 if (setvbuf(fptr
->stdio_file
, NULL
, _IOFBF
, 0) != 0)
4743 rb_warn("setvbuf() can't be honoured for %s", fptr
->path
);
4747 if (close(fptr
->fd
) < 0)
4748 rb_sys_fail(fptr
->path
);
4750 fptr
->fd
= rb_sysopen(fptr
->path
, rb_io_mode_modenum(mode
), 0666);
4758 rb_io_init_copy(VALUE dest
, VALUE io
)
4760 rb_io_t
*fptr
, *orig
;
4764 io
= rb_io_get_io(io
);
4765 if (dest
== io
) return dest
;
4766 GetOpenFile(io
, orig
);
4767 MakeOpenFile(dest
, fptr
);
4771 /* copy rb_io_t structure */
4772 fptr
->mode
= orig
->mode
& ~FMODE_PREP
;
4773 fptr
->pid
= orig
->pid
;
4774 fptr
->lineno
= orig
->lineno
;
4775 if (orig
->path
) fptr
->path
= strdup(orig
->path
);
4776 fptr
->finalize
= orig
->finalize
;
4778 fd
= ruby_dup(orig
->fd
);
4780 io_seek(fptr
, io_tell(orig
), SEEK_SET
);
4781 if (fptr
->mode
& FMODE_BINMODE
) {
4782 rb_io_binmode(dest
);
4785 write_io
= GetWriteIO(io
);
4786 if (io
!= write_io
) {
4787 write_io
= rb_obj_dup(write_io
);
4788 fptr
->tied_io_for_writing
= write_io
;
4789 rb_ivar_set(dest
, rb_intern("@tied_io_for_writing"), write_io
);
4797 * ios.printf(format_string [, obj, ...] ) => nil
4799 * Formats and writes to <em>ios</em>, converting parameters under
4800 * control of the format string. See <code>Kernel#sprintf</code>
4805 rb_io_printf(int argc
, VALUE
*argv
, VALUE out
)
4807 rb_io_write(out
, rb_f_sprintf(argc
, argv
));
4813 * printf(io, string [, obj ... ] ) => nil
4814 * printf(string [, obj ... ] ) => nil
4817 * io.write(sprintf(string, obj, ...)
4819 * $stdout.write(sprintf(string, obj, ...)
4823 rb_f_printf(int argc
, VALUE
*argv
)
4827 if (argc
== 0) return Qnil
;
4828 if (TYPE(argv
[0]) == T_STRING
) {
4836 rb_io_write(out
, rb_f_sprintf(argc
, argv
));
4843 * ios.print() => nil
4844 * ios.print(obj, ...) => nil
4846 * Writes the given object(s) to <em>ios</em>. The stream must be
4847 * opened for writing. If the output record separator (<code>$\\</code>)
4848 * is not <code>nil</code>, it will be appended to the output. If no
4849 * arguments are given, prints <code>$_</code>. Objects that aren't
4850 * strings will be converted by calling their <code>to_s</code> method.
4851 * With no argument, prints the contents of the variable <code>$_</code>.
4852 * Returns <code>nil</code>.
4854 * $stdout.print("This is ", 100, " percent.\n")
4856 * <em>produces:</em>
4858 * This is 100 percent.
4862 rb_io_print(int argc
, VALUE
*argv
, VALUE out
)
4867 /* if no argument given, print `$_' */
4870 line
= rb_lastline_get();
4873 for (i
=0; i
<argc
; i
++) {
4874 rb_io_write(out
, argv
[i
]);
4875 if (!NIL_P(rb_output_fs
)) {
4876 rb_io_write(out
, rb_output_fs
);
4879 if (argc
> 0 && !NIL_P(rb_output_rs
)) {
4880 rb_io_write(out
, rb_output_rs
);
4888 * print(obj, ...) => nil
4890 * Prints each object in turn to <code>$stdout</code>. If the output
4891 * field separator (<code>$,</code>) is not +nil+, its
4892 * contents will appear between each field. If the output record
4893 * separator (<code>$\\</code>) is not +nil+, it will be
4894 * appended to the output. If no arguments are given, prints
4895 * <code>$_</code>. Objects that aren't strings will be converted by
4896 * calling their <code>to_s</code> method.
4898 * print "cat", [1,2,3], 99, "\n"
4901 * print "cat", [1,2,3], 99
4903 * <em>produces:</em>
4910 rb_f_print(int argc
, VALUE
*argv
)
4912 rb_io_print(argc
, argv
, rb_stdout
);
4918 * ios.putc(obj) => obj
4920 * If <i>obj</i> is <code>Numeric</code>, write the character whose
4921 * code is <i>obj</i>, otherwise write the first character of the
4922 * string representation of <i>obj</i> to <em>ios</em>.
4927 * <em>produces:</em>
4933 rb_io_putc(VALUE io
, VALUE ch
)
4935 char c
= NUM2CHR(ch
);
4937 rb_io_write(io
, rb_str_new(&c
, 1));
4951 rb_f_putc(VALUE recv
, VALUE ch
)
4953 if (recv
== rb_stdout
) {
4954 return rb_io_putc(recv
, ch
);
4956 return rb_funcall2(rb_stdout
, rb_intern("putc"), 1, &ch
);
4960 io_puts_ary(VALUE ary
, VALUE out
, int recur
)
4966 tmp
= rb_str_new2("[...]");
4967 rb_io_puts(1, &tmp
, out
);
4970 for (i
=0; i
<RARRAY_LEN(ary
); i
++) {
4971 tmp
= RARRAY_PTR(ary
)[i
];
4972 rb_io_puts(1, &tmp
, out
);
4979 * ios.puts(obj, ...) => nil
4981 * Writes the given objects to <em>ios</em> as with
4982 * <code>IO#print</code>. Writes a record separator (typically a
4983 * newline) after any that do not already end with a newline sequence.
4984 * If called with an array argument, writes each element on a new line.
4985 * If called without arguments, outputs a single record separator.
4987 * $stdout.puts("this", "is", "a", "test")
4989 * <em>produces:</em>
4998 rb_io_puts(int argc
, VALUE
*argv
, VALUE out
)
5003 /* if no argument given, print newline. */
5005 rb_io_write(out
, rb_default_rs
);
5008 for (i
=0; i
<argc
; i
++) {
5009 line
= rb_check_array_type(argv
[i
]);
5011 rb_exec_recursive(io_puts_ary
, line
, out
);
5014 line
= rb_obj_as_string(argv
[i
]);
5015 rb_io_write(out
, line
);
5016 if (RSTRING_LEN(line
) == 0 ||
5017 RSTRING_PTR(line
)[RSTRING_LEN(line
)-1] != '\n') {
5018 rb_io_write(out
, rb_default_rs
);
5027 * puts(obj, ...) => nil
5031 * $stdout.puts(obj, ...)
5035 rb_f_puts(int argc
, VALUE
*argv
, VALUE recv
)
5037 if (recv
== rb_stdout
) {
5038 return rb_io_puts(argc
, argv
, recv
);
5040 return rb_funcall2(rb_stdout
, rb_intern("puts"), argc
, argv
);
5044 rb_p(VALUE obj
) /* for debug print within C code */
5046 VALUE str
= rb_obj_as_string(rb_inspect(obj
));
5047 rb_str_buf_append(str
, rb_default_rs
);
5048 rb_io_write(rb_stdout
, str
);
5054 * p(obj1, obj2, ...) => [obj, ...]
5057 * For each object, directly writes
5058 * _obj_.+inspect+ followed by the current output
5059 * record separator to the program's standard output.
5061 * S = Struct.new(:name, :state)
5062 * s = S['dave', 'TX']
5065 * <em>produces:</em>
5067 * #<S name="dave", state="TX">
5071 rb_f_p(int argc
, VALUE
*argv
, VALUE self
)
5076 for (i
=0; i
<argc
; i
++) {
5082 else if (argc
> 1) {
5083 ret
= rb_ary_new4(argc
, argv
);
5085 if (TYPE(rb_stdout
) == T_FILE
) {
5086 rb_io_flush(rb_stdout
);
5093 * obj.display(port=$>) => nil
5095 * Prints <i>obj</i> on the given port (default <code>$></code>).
5098 * def display(port=$>)
5106 * [ 4, 5, 6 ].display
5109 * <em>produces:</em>
5115 rb_obj_display(int argc
, VALUE
*argv
, VALUE self
)
5123 rb_scan_args(argc
, argv
, "01", &out
);
5125 rb_io_write(out
, self
);
5131 rb_write_error2(const char *mesg
, long len
)
5133 if (rb_stderr
== orig_stderr
|| RFILE(orig_stderr
)->fptr
->fd
< 0) {
5134 fwrite(mesg
, sizeof(char), len
, stderr
);
5137 rb_io_write(rb_stderr
, rb_str_new(mesg
, len
));
5142 rb_write_error(const char *mesg
)
5144 rb_write_error2(mesg
, strlen(mesg
));
5148 must_respond_to(ID mid
, VALUE val
, ID id
)
5150 if (!rb_respond_to(val
, mid
)) {
5151 rb_raise(rb_eTypeError
, "%s must have %s method, %s given",
5152 rb_id2name(id
), rb_id2name(mid
),
5153 rb_obj_classname(val
));
5158 stdout_setter(VALUE val
, ID id
, VALUE
*variable
)
5160 must_respond_to(id_write
, val
, id
);
5165 prep_io(int fd
, int mode
, VALUE klass
, const char *path
)
5168 VALUE io
= io_alloc(klass
);
5170 MakeOpenFile(io
, fp
);
5175 setmode(fd
, O_BINARY
);
5180 if (path
) fp
->path
= strdup(path
);
5186 rb_io_fdopen(int fd
, int mode
, const char *path
)
5188 VALUE klass
= rb_cIO
;
5190 if (path
&& strcmp(path
, "-")) klass
= rb_cFile
;
5191 return prep_io(fd
, rb_io_modenum_flags(mode
), klass
, path
);
5195 prep_stdio(FILE *f
, int mode
, VALUE klass
, const char *path
)
5198 VALUE io
= prep_io(fileno(f
), mode
|FMODE_PREP
, klass
, path
);
5200 GetOpenFile(io
, fptr
);
5201 fptr
->stdio_file
= f
;
5207 rb_io_stdio_file(rb_io_t
*fptr
)
5209 if (!fptr
->stdio_file
) {
5210 fptr
->stdio_file
= rb_fdopen(fptr
->fd
, rb_io_flags_mode(fptr
->mode
));
5212 return fptr
->stdio_file
;
5217 * IO.new(fd, mode) => io
5219 * Returns a new <code>IO</code> object (a stream) for the given
5220 * <code>IO</code> object or integer file descriptor and mode
5221 * string. See also <code>IO#fileno</code> and
5222 * <code>IO::for_fd</code>.
5224 * puts IO.new($stdout).fileno # => 1
5226 * a = IO.new(2,"w") # '2' is standard error
5227 * $stderr.puts "Hello"
5230 * <em>produces:</em>
5237 rb_io_initialize(int argc
, VALUE
*argv
, VALUE io
)
5239 VALUE fnum
, mode
, orig
;
5240 rb_io_t
*fp
, *ofp
= NULL
;
5241 int fd
, fmode
, flags
= O_RDONLY
;
5244 rb_scan_args(argc
, argv
, "11", &fnum
, &mode
);
5246 if (FIXNUM_P(mode
)) {
5247 flags
= FIX2LONG(mode
);
5250 SafeStringValue(mode
);
5251 flags
= rb_io_mode_modenum(StringValueCStr(mode
));
5254 orig
= rb_io_check_io(fnum
);
5259 #if defined(HAVE_FCNTL) && defined(F_GETFL)
5260 flags
= fcntl(fd
, F_GETFL
);
5261 if (flags
== -1) rb_sys_fail(0);
5264 MakeOpenFile(io
, fp
);
5266 fp
->mode
= rb_io_modenum_flags(flags
);
5269 else if (RFILE(io
)->fptr
) {
5270 rb_raise(rb_eRuntimeError
, "reinitializing IO");
5273 GetOpenFile(orig
, ofp
);
5274 if (ofp
->refcnt
== LONG_MAX
) {
5275 VALUE s
= rb_inspect(orig
);
5276 rb_raise(rb_eIOError
, "too many shared IO for %s", StringValueCStr(s
));
5279 fmode
= rb_io_modenum_flags(flags
);
5280 if ((ofp
->mode
^ fmode
) & (FMODE_READWRITE
|FMODE_BINMODE
)) {
5281 if (FIXNUM_P(mode
)) {
5282 rb_raise(rb_eArgError
, "incompatible mode 0%o", flags
);
5285 rb_raise(rb_eArgError
, "incompatible mode \"%s\"", RSTRING_PTR(mode
));
5290 RFILE(io
)->fptr
= ofp
;
5299 * File.new(filename, mode="r") => file
5300 * File.new(filename [, mode [, perm]]) => file
5303 * Opens the file named by _filename_ according to
5304 * _mode_ (default is ``r'') and returns a new
5305 * <code>File</code> object. See the description of class +IO+ for
5306 * a description of _mode_. The file mode may optionally be
5307 * specified as a +Fixnum+ by _or_-ing together the
5308 * flags (O_RDONLY etc, again described under +IO+). Optional
5309 * permission bits may be given in _perm_. These mode and permission
5310 * bits are platform dependent; on Unix systems, see
5311 * <code>open(2)</code> for details.
5313 * f = File.new("testfile", "r")
5314 * f = File.new("newfile", "w+")
5315 * f = File.new("newfile", File::CREAT|File::TRUNC|File::RDWR, 0644)
5319 rb_file_initialize(int argc
, VALUE
*argv
, VALUE io
)
5321 if (RFILE(io
)->fptr
) {
5322 rb_raise(rb_eRuntimeError
, "reinitializing File");
5324 if (0 < argc
&& argc
< 3) {
5325 VALUE fd
= rb_check_convert_type(argv
[0], T_FIXNUM
, "Fixnum", "to_int");
5329 return rb_io_initialize(argc
, argv
, io
);
5332 rb_open_file(argc
, argv
, io
);
5339 * IO.new(fd, mode_string) => io
5341 * Returns a new <code>IO</code> object (a stream) for the given
5342 * integer file descriptor and mode string. See also
5343 * <code>IO#fileno</code> and <code>IO::for_fd</code>.
5345 * a = IO.new(2,"w") # '2' is standard error
5346 * $stderr.puts "Hello"
5349 * <em>produces:</em>
5356 rb_io_s_new(int argc
, VALUE
*argv
, VALUE klass
)
5358 if (rb_block_given_p()) {
5359 const char *cname
= rb_class2name(klass
);
5361 rb_warn("%s::new() does not take block; use %s::open() instead",
5364 return rb_class_new_instance(argc
, argv
, klass
);
5370 * IO.for_fd(fd, mode) => io
5372 * Synonym for <code>IO::new</code>.
5377 rb_io_s_for_fd(int argc
, VALUE
*argv
, VALUE klass
)
5379 VALUE io
= rb_obj_alloc(klass
);
5380 rb_io_initialize(argc
, argv
, io
);
5385 argf_mark(void *ptr
)
5387 struct argf
*p
= ptr
;
5388 rb_gc_mark(p
->filename
);
5389 rb_gc_mark(p
->current_file
);
5390 rb_gc_mark(p
->lineno
);
5391 rb_gc_mark(p
->argv
);
5395 argf_free(void *ptr
)
5397 struct argf
*p
= ptr
;
5402 argf_init(struct argf
*p
, VALUE v
)
5405 p
->current_file
= Qnil
;
5411 argf_alloc(VALUE klass
)
5414 VALUE argf
= Data_Make_Struct(klass
, struct argf
, argf_mark
, argf_free
, p
);
5421 #define filename ARGF.filename
5422 #define current_file ARGF.current_file
5423 #define gets_lineno ARGF.gets_lineno
5424 #define init_p ARGF.init_p
5425 #define next_p ARGF.next_p
5426 #define lineno ARGF.lineno
5427 #define ruby_inplace_mode ARGF.inplace
5428 #define argf_binmode ARGF.binmode
5429 #define argf_enc ARGF.enc
5430 #define argf_enc2 ARGF.enc2
5431 #define rb_argv ARGF.argv
5434 argf_initialize(VALUE argf
, VALUE argv
)
5436 memset(&ARGF
, 0, sizeof(ARGF
));
5437 argf_init(&ARGF
, argv
);
5443 argf_initialize_copy(VALUE argf
, VALUE orig
)
5445 ARGF
= argf_of(orig
);
5446 rb_argv
= rb_obj_dup(rb_argv
);
5448 const char *inplace
= ARGF
.inplace
;
5450 ARGF
.inplace
= ruby_strdup(inplace
);
5456 argf_set_lineno(VALUE argf
, VALUE val
)
5458 gets_lineno
= NUM2INT(val
);
5459 lineno
= INT2FIX(gets_lineno
);
5464 argf_lineno(VALUE argf
)
5470 argf_forward(int argc
, VALUE
*argv
, VALUE argf
)
5472 return rb_funcall3(current_file
, rb_frame_this_func(), argc
, argv
);
5475 #define next_argv() argf_next_argv(argf)
5476 #define ARGF_GENERIC_INPUT_P() \
5477 (current_file == rb_stdin && TYPE(current_file) != T_FILE)
5478 #define ARGF_FORWARD(argc, argv) do {\
5479 if (ARGF_GENERIC_INPUT_P())\
5480 return argf_forward(argc, argv, argf);\
5482 #define NEXT_ARGF_FORWARD(argc, argv) do {\
5483 if (!next_argv()) return Qnil;\
5484 ARGF_FORWARD(argc, argv);\
5488 argf_close(VALUE file
)
5490 rb_funcall3(file
, rb_intern("close"), 0, 0);
5494 argf_next_argv(VALUE argf
)
5498 int stdout_binmode
= 0;
5500 if (TYPE(rb_stdout
) == T_FILE
) {
5501 GetOpenFile(rb_stdout
, fptr
);
5502 if (fptr
->mode
& FMODE_BINMODE
)
5507 if (!NIL_P(rb_argv
) && RARRAY_LEN(rb_argv
) > 0) {
5520 if (RARRAY_LEN(rb_argv
) > 0) {
5521 filename
= rb_ary_shift(rb_argv
);
5522 fn
= StringValueCStr(filename
);
5523 if (strlen(fn
) == 1 && fn
[0] == '-') {
5524 current_file
= rb_stdin
;
5525 if (ruby_inplace_mode
) {
5526 rb_warn("Can't do inplace edit for stdio; skipping");
5531 int fr
= rb_sysopen(fn
, O_RDONLY
, 0);
5533 if (ruby_inplace_mode
) {
5535 #ifndef NO_SAFE_RENAME
5541 if (TYPE(rb_stdout
) == T_FILE
&& rb_stdout
!= orig_stdout
) {
5542 rb_io_close(rb_stdout
);
5545 if (*ruby_inplace_mode
) {
5546 str
= rb_str_new2(fn
);
5547 #ifdef NO_LONG_FNAME
5548 ruby_add_suffix(str
, ruby_inplace_mode
);
5550 rb_str_cat2(str
, ruby_inplace_mode
);
5552 #ifdef NO_SAFE_RENAME
5554 (void)unlink(RSTRING_PTR(str
));
5555 (void)rename(fn
, RSTRING_PTR(str
));
5556 fr
= rb_sysopen(RSTRING_PTR(str
), O_RDONLY
, 0);
5558 if (rename(fn
, RSTRING_PTR(str
)) < 0) {
5559 rb_warn("Can't rename %s to %s: %s, skipping file",
5560 fn
, RSTRING_PTR(str
), strerror(errno
));
5567 #ifdef NO_SAFE_RENAME
5568 rb_fatal("Can't do inplace edit without backup");
5570 if (unlink(fn
) < 0) {
5571 rb_warn("Can't remove %s: %s, skipping file",
5572 fn
, strerror(errno
));
5578 fw
= rb_sysopen(fn
, O_WRONLY
|O_CREAT
|O_TRUNC
, 0666);
5579 #ifndef NO_SAFE_RENAME
5582 fchmod(fw
, st
.st_mode
);
5584 chmod(fn
, st
.st_mode
);
5586 if (st
.st_uid
!=st2
.st_uid
|| st
.st_gid
!=st2
.st_gid
) {
5587 fchown(fw
, st
.st_uid
, st
.st_gid
);
5590 rb_stdout
= prep_io(fw
, FMODE_WRITABLE
, rb_cFile
, fn
);
5591 if (stdout_binmode
) rb_io_binmode(rb_stdout
);
5593 current_file
= prep_io(fr
, FMODE_READABLE
, rb_cFile
, fn
);
5595 if (argf_binmode
) rb_io_binmode(current_file
);
5599 GetOpenFile(current_file
, fptr
);
5600 fptr
->enc
= argf_enc
;
5601 fptr
->enc2
= argf_enc2
;
5609 else if (next_p
== -1) {
5610 current_file
= rb_stdin
;
5611 filename
= rb_str_new2("-");
5612 if (ruby_inplace_mode
) {
5613 rb_warn("Can't do inplace edit for stdio");
5614 rb_stdout
= orig_stdout
;
5621 argf_getline(int argc
, VALUE
*argv
, VALUE argf
)
5626 if (!next_argv()) return Qnil
;
5627 if (ARGF_GENERIC_INPUT_P()) {
5628 line
= rb_funcall3(current_file
, rb_intern("gets"), argc
, argv
);
5631 if (argc
== 0 && rb_rs
== rb_default_rs
) {
5632 line
= rb_io_gets(current_file
);
5635 line
= rb_io_getline(argc
, argv
, current_file
);
5637 if (NIL_P(line
) && next_p
!= -1) {
5638 argf_close(current_file
);
5645 lineno
= INT2FIX(gets_lineno
);
5651 argf_lineno_getter(ID id
, VALUE
*var
)
5658 argf_lineno_setter(VALUE val
, ID id
, VALUE
*var
)
5661 int n
= NUM2INT(val
);
5663 lineno
= INT2FIX(n
);
5666 static VALUE
argf_gets(int, VALUE
*, VALUE
);
5670 * gets(sep=$/) => string or nil
5671 * gets(limit) => string or nil
5672 * gets(sep,limit) => string or nil
5674 * Returns (and assigns to <code>$_</code>) the next line from the list
5675 * of files in +ARGV+ (or <code>$*</code>), or from standard input if
5676 * no files are present on the command line. Returns +nil+ at end of
5677 * file. The optional argument specifies the record separator. The
5678 * separator is included with the contents of each record. A separator
5679 * of +nil+ reads the entire contents, and a zero-length separator
5680 * reads the input one paragraph at a time, where paragraphs are
5681 * divided by two consecutive newlines. If the first argument is an
5682 * integer, or optional second argument is given, the returning string
5683 * would not be longer than the given value. If multiple filenames are
5684 * present in +ARGV+, +gets(nil)+ will read the contents one file at a
5687 * ARGV << "testfile"
5690 * <em>produces:</em>
5694 * This is line three
5697 * The style of programming using <code>$_</code> as an implicit
5698 * parameter is gradually losing favor in the Ruby community.
5702 rb_f_gets(int argc
, VALUE
*argv
, VALUE recv
)
5705 return argf_gets(argc
, argv
, argf
);
5707 return rb_funcall2(argf
, rb_intern("gets"), argc
, argv
);
5711 argf_gets(int argc
, VALUE
*argv
, VALUE argf
)
5715 line
= argf_getline(argc
, argv
, argf
);
5716 rb_lastline_set(line
);
5725 if (rb_rs
!= rb_default_rs
) {
5726 return rb_f_gets(0, 0, argf
);
5730 if (!next_argv()) return Qnil
;
5731 line
= rb_io_gets(current_file
);
5732 if (NIL_P(line
) && next_p
!= -1) {
5733 rb_io_close(current_file
);
5737 rb_lastline_set(line
);
5740 lineno
= INT2FIX(gets_lineno
);
5746 static VALUE
argf_readline(int, VALUE
*, VALUE
);
5750 * readline(sep=$/) => string
5751 * readline(limit) => string
5752 * readline(sep, limit) => string
5754 * Equivalent to <code>Kernel::gets</code>, except
5755 * +readline+ raises +EOFError+ at end of file.
5759 rb_f_readline(int argc
, VALUE
*argv
, VALUE recv
)
5762 return argf_readline(argc
, argv
, argf
);
5764 return rb_funcall2(argf
, rb_intern("readline"), argc
, argv
);
5768 argf_readline(int argc
, VALUE
*argv
, VALUE argf
)
5772 if (!next_argv()) rb_eof_error();
5773 ARGF_FORWARD(argc
, argv
);
5774 line
= argf_gets(argc
, argv
, argf
);
5782 static VALUE
argf_readlines(int, VALUE
*, VALUE
);
5786 * readlines(sep=$/) => array
5787 * readlines(limit) => array
5788 * readlines(sep,limit) => array
5790 * Returns an array containing the lines returned by calling
5791 * <code>Kernel.gets(<i>sep</i>)</code> until the end of file.
5795 rb_f_readlines(int argc
, VALUE
*argv
, VALUE recv
)
5798 return argf_readlines(argc
, argv
, argf
);
5800 return rb_funcall2(argf
, rb_intern("readlines"), argc
, argv
);
5804 argf_readlines(int argc
, VALUE
*argv
, VALUE argf
)
5809 while (!NIL_P(line
= argf_getline(argc
, argv
, argf
))) {
5810 rb_ary_push(ary
, line
);
5820 * Returns the standard output of running _cmd_ in a subshell.
5821 * The built-in syntax <code>%x{...}</code> uses
5822 * this method. Sets <code>$?</code> to the process status.
5824 * `date` #=> "Wed Apr 9 08:56:30 CDT 2003\n"
5825 * `ls testdir`.split[1] #=> "main.rb"
5826 * `echo oops && exit 99` #=> "oops\n"
5827 * $?.exitstatus #=> 99
5831 rb_f_backquote(VALUE obj
, VALUE str
)
5833 volatile VALUE port
;
5837 SafeStringValue(str
);
5838 port
= pipe_open_s(str
, "r");
5839 if (NIL_P(port
)) return rb_str_new(0,0);
5841 GetOpenFile(port
, fptr
);
5842 result
= read_all(fptr
, remain_size(fptr
), Qnil
);
5848 #ifdef HAVE_SYS_SELECT_H
5849 #include <sys/select.h>
5853 select_internal(VALUE read
, VALUE write
, VALUE except
, struct timeval
*tp
, rb_fdset_t
*fds
)
5856 fd_set
*rp
, *wp
, *ep
;
5860 int interrupt_flag
= 0;
5862 struct timeval timerec
;
5865 Check_Type(read
, T_ARRAY
);
5866 for (i
=0; i
<RARRAY_LEN(read
); i
++) {
5867 GetOpenFile(rb_io_get_io(RARRAY_PTR(read
)[i
]), fptr
);
5868 rb_fd_set(fptr
->fd
, &fds
[0]);
5869 if (READ_DATA_PENDING(fptr
)) { /* check for buffered data */
5871 rb_fd_set(fptr
->fd
, &fds
[3]);
5873 if (max
< fptr
->fd
) max
= fptr
->fd
;
5875 if (pending
) { /* no blocking if there's buffered data */
5876 timerec
.tv_sec
= timerec
.tv_usec
= 0;
5879 rp
= rb_fd_ptr(&fds
[0]);
5884 if (!NIL_P(write
)) {
5885 Check_Type(write
, T_ARRAY
);
5886 for (i
=0; i
<RARRAY_LEN(write
); i
++) {
5887 VALUE write_io
= GetWriteIO(rb_io_get_io(RARRAY_PTR(write
)[i
]));
5888 GetOpenFile(write_io
, fptr
);
5889 rb_fd_set(fptr
->fd
, &fds
[1]);
5890 if (max
< fptr
->fd
) max
= fptr
->fd
;
5892 wp
= rb_fd_ptr(&fds
[1]);
5897 if (!NIL_P(except
)) {
5898 Check_Type(except
, T_ARRAY
);
5899 for (i
=0; i
<RARRAY_LEN(except
); i
++) {
5900 VALUE io
= rb_io_get_io(RARRAY_PTR(except
)[i
]);
5901 VALUE write_io
= GetWriteIO(io
);
5902 GetOpenFile(io
, fptr
);
5903 rb_fd_set(fptr
->fd
, &fds
[2]);
5904 if (max
< fptr
->fd
) max
= fptr
->fd
;
5905 if (io
!= write_io
) {
5906 GetOpenFile(write_io
, fptr
);
5907 rb_fd_set(fptr
->fd
, &fds
[2]);
5908 if (max
< fptr
->fd
) max
= fptr
->fd
;
5911 ep
= rb_fd_ptr(&fds
[2]);
5919 n
= rb_thread_select(max
, rp
, wp
, ep
, tp
);
5923 if (!pending
&& n
== 0) return Qnil
; /* returns nil on timeout */
5925 res
= rb_ary_new2(3);
5926 rb_ary_push(res
, rp
?rb_ary_new():rb_ary_new2(0));
5927 rb_ary_push(res
, wp
?rb_ary_new():rb_ary_new2(0));
5928 rb_ary_push(res
, ep
?rb_ary_new():rb_ary_new2(0));
5930 if (interrupt_flag
== 0) {
5932 list
= RARRAY_PTR(res
)[0];
5933 for (i
=0; i
< RARRAY_LEN(read
); i
++) {
5934 VALUE obj
= rb_ary_entry(read
, i
);
5935 VALUE io
= rb_io_get_io(obj
);
5936 GetOpenFile(io
, fptr
);
5937 if (rb_fd_isset(fptr
->fd
, &fds
[0]) ||
5938 rb_fd_isset(fptr
->fd
, &fds
[3])) {
5939 rb_ary_push(list
, obj
);
5945 list
= RARRAY_PTR(res
)[1];
5946 for (i
=0; i
< RARRAY_LEN(write
); i
++) {
5947 VALUE obj
= rb_ary_entry(write
, i
);
5948 VALUE io
= rb_io_get_io(obj
);
5949 VALUE write_io
= GetWriteIO(io
);
5950 GetOpenFile(write_io
, fptr
);
5951 if (rb_fd_isset(fptr
->fd
, &fds
[1])) {
5952 rb_ary_push(list
, obj
);
5958 list
= RARRAY_PTR(res
)[2];
5959 for (i
=0; i
< RARRAY_LEN(except
); i
++) {
5960 VALUE obj
= rb_ary_entry(except
, i
);
5961 VALUE io
= rb_io_get_io(obj
);
5962 VALUE write_io
= GetWriteIO(io
);
5963 GetOpenFile(io
, fptr
);
5964 if (rb_fd_isset(fptr
->fd
, &fds
[2])) {
5965 rb_ary_push(list
, obj
);
5967 else if (io
!= write_io
) {
5968 GetOpenFile(write_io
, fptr
);
5969 if (rb_fd_isset(fptr
->fd
, &fds
[2])) {
5970 rb_ary_push(list
, obj
);
5977 return res
; /* returns an empty array on interrupt */
5980 struct select_args
{
5981 VALUE read
, write
, except
;
5982 struct timeval
*timeout
;
5983 rb_fdset_t fdsets
[4];
5986 #ifdef HAVE_RB_FD_INIT
5988 select_call(VALUE arg
)
5990 struct select_args
*p
= (struct select_args
*)arg
;
5992 return select_internal(p
->read
, p
->write
, p
->except
, p
->timeout
, p
->fdsets
);
5996 select_end(VALUE arg
)
5998 struct select_args
*p
= (struct select_args
*)arg
;
6001 for (i
= 0; i
< sizeof(p
->fdsets
) / sizeof(p
->fdsets
[0]); ++i
)
6002 rb_fd_term(&p
->fdsets
[i
]);
6009 * IO.select(read_array
6012 * [, timeout]]] ) => array or nil
6014 * See <code>Kernel#select</code>.
6018 rb_f_select(int argc
, VALUE
*argv
, VALUE obj
)
6021 struct select_args args
;
6022 struct timeval timerec
;
6025 rb_scan_args(argc
, argv
, "13", &args
.read
, &args
.write
, &args
.except
, &timeout
);
6026 if (NIL_P(timeout
)) {
6030 timerec
= rb_time_interval(timeout
);
6031 args
.timeout
= &timerec
;
6034 for (i
= 0; i
< sizeof(args
.fdsets
) / sizeof(args
.fdsets
[0]); ++i
)
6035 rb_fd_init(&args
.fdsets
[i
]);
6037 #ifdef HAVE_RB_FD_INIT
6038 return rb_ensure(select_call
, (VALUE
)&args
, select_end
, (VALUE
)&args
);
6040 return select_internal(args
.read
, args
.write
, args
.except
,
6041 args
.timeout
, args
.fdsets
);
6046 #if !defined(MSDOS) && !defined(__human68k__)
6048 io_cntl(int fd
, int cmd
, long narg
, int io_p
)
6054 # if defined(__CYGWIN__)
6055 retval
= io_p
?ioctl(fd
, cmd
, (void*)narg
):fcntl(fd
, cmd
, narg
);
6057 retval
= io_p
?ioctl(fd
, cmd
, narg
):fcntl(fd
, cmd
, narg
);
6065 retval
= ioctl(fd
, cmd
, narg
);
6073 rb_io_ctl(VALUE io
, VALUE req
, VALUE arg
, int io_p
)
6075 #if !defined(MSDOS) && !defined(__human68k__)
6076 int cmd
= NUM2ULONG(req
);
6084 if (NIL_P(arg
) || arg
== Qfalse
) {
6087 else if (FIXNUM_P(arg
)) {
6088 narg
= FIX2LONG(arg
);
6090 else if (arg
== Qtrue
) {
6094 VALUE tmp
= rb_check_string_type(arg
);
6097 narg
= NUM2LONG(arg
);
6103 #define IOCPARM_LEN(x) (((x) >> 16) & IOCPARM_MASK)
6107 len
= IOCPARM_LEN(cmd
); /* on BSDish systems we're safe */
6109 len
= 256; /* otherwise guess at what's safe */
6113 if (len
<= RSTRING_LEN(arg
)) {
6114 len
= RSTRING_LEN(arg
);
6116 if (RSTRING_LEN(arg
) < len
) {
6117 rb_str_resize(arg
, len
+1);
6119 RSTRING_PTR(arg
)[len
] = 17; /* a little sanity check here */
6120 narg
= (long)RSTRING_PTR(arg
);
6123 GetOpenFile(io
, fptr
);
6124 retval
= io_cntl(fptr
->fd
, cmd
, narg
, io_p
);
6125 if (retval
< 0) rb_sys_fail(fptr
->path
);
6126 if (TYPE(arg
) == T_STRING
&& RSTRING_PTR(arg
)[len
] != 17) {
6127 rb_raise(rb_eArgError
, "return value overflowed string");
6130 if (!io_p
&& cmd
== F_SETFL
) {
6131 if (narg
& O_NONBLOCK
) {
6132 fptr
->mode
|= FMODE_WSPLIT_INITIALIZED
;
6133 fptr
->mode
&= ~FMODE_WSPLIT
;
6136 fptr
->mode
&= ~(FMODE_WSPLIT_INITIALIZED
|FMODE_WSPLIT
);
6140 return INT2NUM(retval
);
6143 return Qnil
; /* not reached */
6150 * ios.ioctl(integer_cmd, arg) => integer
6152 * Provides a mechanism for issuing low-level commands to control or
6153 * query I/O devices. Arguments and results are platform dependent. If
6154 * <i>arg</i> is a number, its value is passed directly. If it is a
6155 * string, it is interpreted as a binary sequence of bytes. On Unix
6156 * platforms, see <code>ioctl(2)</code> for details. Not implemented on
6161 rb_io_ioctl(int argc
, VALUE
*argv
, VALUE io
)
6165 rb_scan_args(argc
, argv
, "11", &req
, &arg
);
6166 return rb_io_ctl(io
, req
, arg
, 1);
6171 * ios.fcntl(integer_cmd, arg) => integer
6173 * Provides a mechanism for issuing low-level commands to control or
6174 * query file-oriented I/O streams. Arguments and results are platform
6175 * dependent. If <i>arg</i> is a number, its value is passed
6176 * directly. If it is a string, it is interpreted as a binary sequence
6177 * of bytes (<code>Array#pack</code> might be a useful way to build this
6178 * string). On Unix platforms, see <code>fcntl(2)</code> for details.
6179 * Not implemented on all platforms.
6183 rb_io_fcntl(int argc
, VALUE
*argv
, VALUE io
)
6188 rb_scan_args(argc
, argv
, "11", &req
, &arg
);
6189 return rb_io_ctl(io
, req
, arg
, 0);
6192 return Qnil
; /* not reached */
6198 * syscall(fixnum [, args...]) => integer
6200 * Calls the operating system function identified by _fixnum_,
6201 * passing in the arguments, which must be either +String+
6202 * objects, or +Integer+ objects that ultimately fit within
6203 * a native +long+. Up to nine parameters may be passed (14
6204 * on the Atari-ST). The function identified by _fixnum_ is system
6205 * dependent. On some Unix systems, the numbers may be obtained from a
6206 * header file called <code>syscall.h</code>.
6208 * syscall 4, 1, "hello\n", 6 # '4' is write(2) on our box
6210 * <em>produces:</em>
6216 rb_f_syscall(int argc
, VALUE
*argv
)
6218 #if defined(HAVE_SYSCALL) && !defined(__CHECKER__)
6220 unsigned long arg
[14]; /* yes, we really need that many ! */
6222 unsigned long arg
[8];
6226 int items
= argc
- 1;
6228 /* This probably won't work on machines where sizeof(long) != sizeof(int)
6229 * or where sizeof(long) != sizeof(char*). But such machines will
6230 * not likely have syscall implemented either, so who cares?
6235 rb_raise(rb_eArgError
, "too few arguments for syscall");
6236 if (argc
> sizeof(arg
) / sizeof(arg
[0]))
6237 rb_raise(rb_eArgError
, "too many arguments for syscall");
6238 arg
[0] = NUM2LONG(argv
[0]); argv
++;
6240 VALUE v
= rb_check_string_type(*argv
);
6245 arg
[i
] = (unsigned long)StringValueCStr(v
);
6248 arg
[i
] = (unsigned long)NUM2LONG(*argv
);
6256 retval
= syscall(arg
[0]);
6259 retval
= syscall(arg
[0],arg
[1]);
6262 retval
= syscall(arg
[0],arg
[1],arg
[2]);
6265 retval
= syscall(arg
[0],arg
[1],arg
[2],arg
[3]);
6268 retval
= syscall(arg
[0],arg
[1],arg
[2],arg
[3],arg
[4]);
6271 retval
= syscall(arg
[0],arg
[1],arg
[2],arg
[3],arg
[4],arg
[5]);
6274 retval
= syscall(arg
[0],arg
[1],arg
[2],arg
[3],arg
[4],arg
[5],arg
[6]);
6277 retval
= syscall(arg
[0],arg
[1],arg
[2],arg
[3],arg
[4],arg
[5],arg
[6],
6282 retval
= syscall(arg
[0],arg
[1],arg
[2],arg
[3],arg
[4],arg
[5],arg
[6],
6286 retval
= syscall(arg
[0],arg
[1],arg
[2],arg
[3],arg
[4],arg
[5],arg
[6],
6287 arg
[7], arg
[8], arg
[9]);
6290 retval
= syscall(arg
[0],arg
[1],arg
[2],arg
[3],arg
[4],arg
[5],arg
[6],
6291 arg
[7], arg
[8], arg
[9], arg
[10]);
6294 retval
= syscall(arg
[0],arg
[1],arg
[2],arg
[3],arg
[4],arg
[5],arg
[6],
6295 arg
[7], arg
[8], arg
[9], arg
[10], arg
[11]);
6298 retval
= syscall(arg
[0],arg
[1],arg
[2],arg
[3],arg
[4],arg
[5],arg
[6],
6299 arg
[7], arg
[8], arg
[9], arg
[10], arg
[11], arg
[12]);
6302 retval
= syscall(arg
[0],arg
[1],arg
[2],arg
[3],arg
[4],arg
[5],arg
[6],
6303 arg
[7], arg
[8], arg
[9], arg
[10], arg
[11], arg
[12], arg
[13]);
6305 #endif /* atarist */
6308 if (retval
< 0) rb_sys_fail(0);
6309 return INT2NUM(retval
);
6312 return Qnil
; /* not reached */
6317 io_new_instance(VALUE args
)
6319 return rb_class_new_instance(2, (VALUE
*)args
+1, *(VALUE
*)args
);
6323 io_encoding_set(rb_io_t
*fptr
, int argc
, VALUE v1
, VALUE v2
)
6325 if (NIL_P(v2
)) argc
= 1;
6327 fptr
->enc2
= rb_to_encoding(v1
);
6328 fptr
->enc
= rb_to_encoding(v2
);
6330 else if (argc
== 1) {
6335 VALUE tmp
= rb_check_string_type(v1
);
6337 mode_enc(fptr
, StringValueCStr(tmp
));
6340 fptr
->enc
= rb_to_encoding(v1
);
6348 * IO.pipe -> [read_io, write_io]
6349 * IO.pipe(ext_enc) -> [read_io, write_io]
6350 * IO.pipe("ext_enc:int_enc") -> [read_io, write_io]
6351 * IO.pipe(ext_enc, int_enc) -> [read_io, write_io]
6353 * Creates a pair of pipe endpoints (connected to each other) and
6354 * returns them as a two-element array of <code>IO</code> objects:
6355 * <code>[</code> <i>read_io</i>, <i>write_io</i> <code>]</code>. Not
6356 * available on all platforms.
6358 * If an encoding (encoding name or encoding object) is specified as an optional argument,
6359 * read string from pipe is tagged with the encoding specified.
6360 * If the argument is a colon separated two encoding names "A:B",
6361 * the read string is converted from encoding A (external encoding)
6362 * to encoding B (internal encoding), then tagged with B.
6363 * If two optional arguments are specified, those must be
6364 * encoding objects or encoding names,
6365 * and the first one is the external encoding,
6366 * and the second one is the internal encoding.
6368 * In the example below, the two processes close the ends of the pipe
6369 * that they are not using. This is not just a cosmetic nicety. The
6370 * read end of a pipe will not generate an end of file condition if
6371 * there are any writers with the pipe still open. In the case of the
6372 * parent process, the <code>rd.read</code> will never return if it
6373 * does not first issue a <code>wr.close</code>.
6379 * puts "Parent got: <#{rd.read}>"
6384 * puts "Sending message to parent"
6389 * <em>produces:</em>
6391 * Sending message to parent
6392 * Parent got: <Hi Dad>
6396 rb_io_s_pipe(int argc
, VALUE
*argv
, VALUE klass
)
6400 return Qnil
; /* not reached */
6402 int pipes
[2], state
;
6403 VALUE r
, w
, args
[3], v1
, v2
;
6406 rb_scan_args(argc
, argv
, "02", &v1
, &v2
);
6407 if (rb_pipe(pipes
) == -1)
6411 args
[1] = INT2NUM(pipes
[0]);
6412 args
[2] = INT2FIX(O_RDONLY
);
6413 r
= rb_protect(io_new_instance
, (VALUE
)args
, &state
);
6419 GetOpenFile(r
, fptr
);
6420 io_encoding_set(fptr
, argc
, v1
, v2
);
6421 args
[1] = INT2NUM(pipes
[1]);
6422 args
[2] = INT2FIX(O_WRONLY
);
6423 w
= rb_protect(io_new_instance
, (VALUE
)args
, &state
);
6426 if (!NIL_P(r
)) rb_io_close(r
);
6429 rb_io_synchronized(RFILE(w
)->fptr
);
6431 return rb_assoc_new(r
, w
);
6435 struct foreach_arg
{
6442 open_key_args(int argc
, VALUE
*argv
, struct foreach_arg
*arg
)
6446 FilePathValue(argv
[0]);
6448 arg
->argc
= argc
> 1 ? 1 : 0;
6449 arg
->argv
= argv
+ 1;
6452 arg
->io
= rb_io_open(RSTRING_PTR(argv
[0]), "r");
6455 opt
= rb_check_convert_type(argv
[argc
-1], T_HASH
, "Hash", "to_hash");
6456 if (NIL_P(opt
)) goto no_key
;
6457 if (argc
> 2) arg
->argc
= 1;
6460 v
= rb_hash_aref(opt
, sym_open_args
);
6464 v
= rb_convert_type(v
, T_ARRAY
, "Array", "to_ary");
6465 args
= rb_ary_new2(RARRAY_LEN(v
)+1);
6466 rb_ary_push(args
, argv
[0]);
6467 rb_ary_concat(args
, v
);
6468 MEMCPY(RARRAY_PTR(args
)+1, RARRAY_PTR(v
), VALUE
, RARRAY_LEN(v
));
6470 arg
->io
= rb_io_open_with_args(RARRAY_LEN(args
), RARRAY_PTR(args
));
6473 v
= rb_hash_aref(opt
, sym_mode
);
6475 arg
->io
= rb_io_open(RSTRING_PTR(argv
[0]), StringValueCStr(v
));
6478 arg
->io
= rb_io_open(RSTRING_PTR(argv
[0]), "r");
6481 io_set_encoding(arg
->io
, opt
);
6485 io_s_foreach(struct foreach_arg
*arg
)
6489 while (!NIL_P(str
= rb_io_gets_m(arg
->argc
, arg
->argv
, arg
->io
))) {
6497 * IO.foreach(name, sep=$/) {|line| block } => nil
6498 * IO.foreach(name, limit) {|line| block } => nil
6499 * IO.foreach(name, sep, limit) {|line| block } => nil
6501 * Executes the block for every line in the named I/O port, where lines
6502 * are separated by <em>sep</em>.
6504 * IO.foreach("testfile") {|x| print "GOT ", x }
6506 * <em>produces:</em>
6508 * GOT This is line one
6509 * GOT This is line two
6510 * GOT This is line three
6513 * If the last argument is a hash, it's the keyword argument to open.
6514 * See <code>IO.read</code> for detail.
6519 rb_io_s_foreach(int argc
, VALUE
*argv
, VALUE self
)
6521 struct foreach_arg arg
;
6523 rb_scan_args(argc
, argv
, "13", NULL
, NULL
, NULL
, NULL
);
6524 RETURN_ENUMERATOR(self
, argc
, argv
);
6525 open_key_args(argc
, argv
, &arg
);
6526 if (NIL_P(arg
.io
)) return Qnil
;
6527 return rb_ensure(io_s_foreach
, (VALUE
)&arg
, rb_io_close
, arg
.io
);
6531 io_s_readlines(struct foreach_arg
*arg
)
6533 return rb_io_readlines(arg
->argc
, arg
->argv
, arg
->io
);
6538 * IO.readlines(name, sep=$/) => array
6539 * IO.readlines(name, limit) => array
6540 * IO.readlines(name, sep, limit) => array
6542 * Reads the entire file specified by <i>name</i> as individual
6543 * lines, and returns those lines in an array. Lines are separated by
6546 * a = IO.readlines("testfile")
6547 * a[0] #=> "This is line one\n"
6549 * If the last argument is a hash, it's the keyword argument to open.
6550 * See <code>IO.read</code> for detail.
6555 rb_io_s_readlines(int argc
, VALUE
*argv
, VALUE io
)
6557 struct foreach_arg arg
;
6559 rb_scan_args(argc
, argv
, "13", NULL
, NULL
, NULL
, NULL
);
6560 open_key_args(argc
, argv
, &arg
);
6561 if (NIL_P(arg
.io
)) return Qnil
;
6562 return rb_ensure(io_s_readlines
, (VALUE
)&arg
, rb_io_close
, arg
.io
);
6566 io_s_read(struct foreach_arg
*arg
)
6568 return io_read(arg
->argc
, arg
->argv
, arg
->io
);
6573 * IO.read(name, [length [, offset]] ) => string
6574 * IO.read(name, [length [, offset]], opt) => string
6576 * Opens the file, optionally seeks to the given offset, then returns
6577 * <i>length</i> bytes (defaulting to the rest of the file).
6578 * <code>read</code> ensures the file is closed before returning.
6580 * If the last argument is a hash, it specifies option for internal
6581 * open(). The key would be the following. open_args: is exclusive
6584 * encoding: string or encoding
6586 * specifies encoding of the read string. encoding will be ignored
6587 * if length is specified.
6591 * specifies mode argument for open(). it should start with "r"
6592 * otherwise it would cause error.
6594 * open_args: array of strings
6596 * specifies arguments for open() as an array.
6598 * IO.read("testfile") #=> "This is line one\nThis is line two\nThis is line three\nAnd so on...\n"
6599 * IO.read("testfile", 20) #=> "This is line one\nThi"
6600 * IO.read("testfile", 20, 10) #=> "ne one\nThis is line "
6604 rb_io_s_read(int argc
, VALUE
*argv
, VALUE io
)
6607 struct foreach_arg arg
;
6609 rb_scan_args(argc
, argv
, "13", NULL
, NULL
, &offset
, NULL
);
6610 open_key_args(argc
, argv
, &arg
);
6611 if (NIL_P(arg
.io
)) return Qnil
;
6612 if (!NIL_P(offset
)) {
6613 rb_io_binmode(arg
.io
);
6614 rb_io_seek(arg
.io
, offset
, SEEK_SET
);
6615 if (arg
.argc
== 2) arg
.argc
= 1;
6617 return rb_ensure(io_s_read
, (VALUE
)&arg
, rb_io_close
, arg
.io
);
6620 struct copy_stream_struct
{
6623 off_t copy_length
; /* (off_t)-1 if not specified */
6624 off_t src_offset
; /* (off_t)-1 if not specified */
6639 copy_stream_wait_read(struct copy_stream_struct
*stp
)
6642 rb_fd_zero(&stp
->fds
);
6643 rb_fd_set(stp
->src_fd
, &stp
->fds
);
6644 ret
= rb_fd_select(rb_fd_max(&stp
->fds
), &stp
->fds
, NULL
, NULL
, NULL
);
6646 stp
->syserr
= "select";
6647 stp
->error_no
= errno
;
6654 copy_stream_wait_write(struct copy_stream_struct
*stp
)
6657 rb_fd_zero(&stp
->fds
);
6658 rb_fd_set(stp
->dst_fd
, &stp
->fds
);
6659 ret
= rb_fd_select(rb_fd_max(&stp
->fds
), NULL
, &stp
->fds
, NULL
, NULL
);
6661 stp
->syserr
= "select";
6662 stp
->error_no
= errno
;
6668 #ifdef HAVE_SENDFILE
6671 #define USE_SENDFILE
6673 #ifdef HAVE_SYS_SENDFILE_H
6674 #include <sys/sendfile.h>
6678 simple_sendfile(int out_fd
, int in_fd
, off_t
*offset
, size_t count
)
6680 return sendfile(out_fd
, in_fd
, offset
, count
);
6689 copy_stream_sendfile(struct copy_stream_struct
*stp
)
6691 struct stat src_stat
, dst_stat
;
6699 ret
= fstat(stp
->src_fd
, &src_stat
);
6701 stp
->syserr
= "fstat";
6702 stp
->error_no
= errno
;
6705 if (!S_ISREG(src_stat
.st_mode
))
6708 ret
= fstat(stp
->dst_fd
, &dst_stat
);
6710 stp
->syserr
= "fstat";
6711 stp
->error_no
= errno
;
6714 if ((dst_stat
.st_mode
& S_IFMT
) != S_IFSOCK
)
6717 src_offset
= stp
->src_offset
;
6718 use_pread
= src_offset
!= (off_t
)-1;
6720 copy_length
= stp
->copy_length
;
6721 if (copy_length
== (off_t
)-1) {
6723 copy_length
= src_stat
.st_size
- src_offset
;
6725 off_t cur
= lseek(stp
->src_fd
, 0, SEEK_CUR
);
6726 if (cur
== (off_t
)-1) {
6727 stp
->syserr
= "lseek";
6728 stp
->error_no
= errno
;
6731 copy_length
= src_stat
.st_size
- cur
;
6737 ss
= simple_sendfile(stp
->dst_fd
, stp
->src_fd
, &src_offset
, copy_length
);
6740 ss
= simple_sendfile(stp
->dst_fd
, stp
->src_fd
, NULL
, copy_length
);
6745 if (0 < copy_length
) {
6758 #if defined(EWOULDBLOCK) && EWOULDBLOCK != EAGAIN
6761 if (copy_stream_wait_write(stp
) == -1)
6763 if (RUBY_VM_INTERRUPTED(stp
->th
))
6765 goto retry_sendfile
;
6767 stp
->syserr
= "sendfile";
6768 stp
->error_no
= errno
;
6776 copy_stream_read(struct copy_stream_struct
*stp
, char *buf
, int len
, off_t offset
)
6780 if (offset
== (off_t
)-1)
6781 ss
= read(stp
->src_fd
, buf
, len
);
6784 ss
= pread(stp
->src_fd
, buf
, len
, offset
);
6786 stp
->notimp
= "pread";
6796 #if defined(EWOULDBLOCK) && EWOULDBLOCK != EAGAIN
6799 if (copy_stream_wait_read(stp
) == -1)
6805 stp
->notimp
= "pread";
6808 stp
->syserr
= offset
== (off_t
)-1 ? "read" : "pread";
6809 stp
->error_no
= errno
;
6816 copy_stream_write(struct copy_stream_struct
*stp
, char *buf
, int len
)
6821 ss
= write(stp
->dst_fd
, buf
+off
, len
);
6823 if (errno
== EAGAIN
|| errno
== EWOULDBLOCK
) {
6824 if (copy_stream_wait_write(stp
) == -1)
6828 stp
->syserr
= "write";
6829 stp
->error_no
= errno
;
6840 copy_stream_read_write(struct copy_stream_struct
*stp
)
6851 copy_length
= stp
->copy_length
;
6852 use_eof
= copy_length
== (off_t
)-1;
6853 src_offset
= stp
->src_offset
;
6854 use_pread
= src_offset
!= (off_t
)-1;
6856 if (use_pread
&& stp
->close_src
) {
6858 r
= lseek(stp
->src_fd
, src_offset
, SEEK_SET
);
6859 if (r
== (off_t
)-1) {
6860 stp
->syserr
= "lseek";
6861 stp
->error_no
= errno
;
6864 src_offset
= (off_t
)-1;
6868 while (use_eof
|| 0 < copy_length
) {
6869 if (!use_eof
&& copy_length
< sizeof(buf
)) {
6876 ss
= copy_stream_read(stp
, buf
, len
, src_offset
);
6881 ss
= copy_stream_read(stp
, buf
, len
, (off_t
)-1);
6883 if (ss
<= 0) /* EOF or error */
6886 ret
= copy_stream_write(stp
, buf
, ss
);
6893 if (RUBY_VM_INTERRUPTED(stp
->th
))
6899 copy_stream_func(void *arg
)
6901 struct copy_stream_struct
*stp
= (struct copy_stream_struct
*)arg
;
6907 ret
= copy_stream_sendfile(stp
);
6909 goto finish
; /* error or success */
6912 copy_stream_read_write(stp
);
6921 copy_stream_fallback_body(VALUE arg
)
6923 struct copy_stream_struct
*stp
= (struct copy_stream_struct
*)arg
;
6924 const int buflen
= 16*1024;
6926 VALUE buf
= rb_str_buf_new(buflen
);
6927 long rest
= stp
->copy_length
;
6928 off_t off
= stp
->src_offset
;
6933 if (stp
->copy_length
== (off_t
)-1) {
6939 l
= buflen
< rest
? buflen
: rest
;
6941 if (stp
->src_fd
== -1) {
6942 rb_funcall(stp
->src
, id_readpartial
, 2, INT2FIX(l
), buf
);
6946 rb_thread_wait_fd(stp
->src_fd
);
6947 rb_str_resize(buf
, buflen
);
6948 ss
= copy_stream_read(stp
, RSTRING_PTR(buf
), l
, off
);
6953 rb_str_resize(buf
, ss
);
6954 if (off
!= (off_t
)-1)
6957 n
= rb_io_write(stp
->dst
, buf
);
6958 numwrote
= NUM2LONG(n
);
6959 stp
->total
+= numwrote
;
6967 copy_stream_fallback(struct copy_stream_struct
*stp
)
6969 if (stp
->src_fd
== -1 && stp
->src_offset
!= (off_t
)-1) {
6970 rb_raise(rb_eArgError
, "cannot specify src_offset for non-IO");
6972 rb_rescue2(copy_stream_fallback_body
, (VALUE
)stp
,
6973 (VALUE (*) (ANYARGS
))0, (VALUE
)0,
6974 rb_eEOFError
, (VALUE
)0);
6979 copy_stream_body(VALUE arg
)
6981 struct copy_stream_struct
*stp
= (struct copy_stream_struct
*)arg
;
6982 VALUE src_io
, dst_io
;
6983 rb_io_t
*src_fptr
= 0, *dst_fptr
= 0;
6986 stp
->th
= GET_THREAD();
6990 if (stp
->src
== argf
||
6991 !(TYPE(stp
->src
) == T_FILE
||
6992 rb_respond_to(stp
->src
, rb_intern("to_io")) ||
6993 TYPE(stp
->src
) == T_STRING
||
6994 rb_respond_to(stp
->src
, rb_intern("to_path")))) {
6998 src_io
= rb_check_convert_type(stp
->src
, T_FILE
, "IO", "to_io");
6999 if (NIL_P(src_io
)) {
7001 int flags
= O_RDONLY
;
7005 FilePathValue(stp
->src
);
7007 args
[1] = INT2NUM(flags
);
7008 src_io
= rb_class_new_instance(2, args
, rb_cFile
);
7012 GetOpenFile(src_io
, src_fptr
);
7013 rb_io_check_readable(src_fptr
);
7014 src_fd
= src_fptr
->fd
;
7016 stp
->src_fd
= src_fd
;
7018 if (stp
->dst
== argf
||
7019 !(TYPE(stp
->dst
) == T_FILE
||
7020 rb_respond_to(stp
->dst
, rb_intern("to_io")) ||
7021 TYPE(stp
->dst
) == T_STRING
||
7022 rb_respond_to(stp
->dst
, rb_intern("to_path")))) {
7026 dst_io
= rb_check_convert_type(stp
->dst
, T_FILE
, "IO", "to_io");
7027 if (NIL_P(dst_io
)) {
7029 int flags
= O_WRONLY
|O_CREAT
|O_TRUNC
;
7033 FilePathValue(stp
->dst
);
7035 args
[1] = INT2NUM(flags
);
7036 args
[2] = INT2FIX(0600);
7037 dst_io
= rb_class_new_instance(3, args
, rb_cFile
);
7042 dst_io
= GetWriteIO(dst_io
);
7045 GetOpenFile(dst_io
, dst_fptr
);
7046 rb_io_check_writable(dst_fptr
);
7047 dst_fd
= dst_fptr
->fd
;
7049 stp
->dst_fd
= dst_fd
;
7051 if (stp
->src_offset
== (off_t
)-1 && src_fptr
&& src_fptr
->rbuf_len
) {
7052 long len
= src_fptr
->rbuf_len
;
7054 if (stp
->copy_length
!= (off_t
)-1 && stp
->copy_length
< len
) {
7055 len
= stp
->copy_length
;
7057 str
= rb_str_buf_new(len
);
7058 rb_str_resize(str
,len
);
7059 read_buffered_data(RSTRING_PTR(str
), len
, src_fptr
);
7060 if (dst_fptr
) /* IO or filename */
7061 io_fwrite(str
, dst_fptr
);
7062 else /* others such as StringIO */
7063 rb_io_write(stp
->dst
, str
);
7065 if (stp
->copy_length
!= (off_t
)-1)
7066 stp
->copy_length
-= len
;
7069 if (dst_fptr
&& io_fflush(dst_fptr
) < 0) {
7070 rb_raise(rb_eIOError
, "flush failed");
7073 if (stp
->copy_length
== 0)
7076 if (src_fd
== -1 || dst_fd
== -1) {
7077 return copy_stream_fallback(stp
);
7080 rb_fd_init(&stp
->fds
);
7081 rb_fd_set(src_fd
, &stp
->fds
);
7082 rb_fd_set(dst_fd
, &stp
->fds
);
7084 return rb_thread_blocking_region(copy_stream_func
, (void*)stp
, RB_UBF_DFL
, 0);
7088 copy_stream_finalize(VALUE arg
)
7090 struct copy_stream_struct
*stp
= (struct copy_stream_struct
*)arg
;
7091 if (stp
->close_src
) {
7092 rb_io_close_m(stp
->src
);
7094 if (stp
->close_dst
) {
7095 rb_io_close_m(stp
->dst
);
7097 rb_fd_term(&stp
->fds
);
7099 errno
= stp
->error_no
;
7100 rb_sys_fail(stp
->syserr
);
7103 rb_raise(rb_eNotImpError
, "%s() not implemented", stp
->notimp
);
7110 * IO.copy_stream(src, dst)
7111 * IO.copy_stream(src, dst, copy_length)
7112 * IO.copy_stream(src, dst, copy_length, src_offset)
7114 * IO.copy_stream copies <i>src</i> to <i>dst</i>.
7115 * <i>src</i> and <i>dst</i> is either a filename or an IO.
7117 * This method returns the number of bytes copied.
7119 * If optional arguments are not given,
7120 * the start position of the copy is
7121 * the beginning of the filename or
7122 * the current file offset of the IO.
7123 * The end position of the copy is the end of file.
7125 * If <i>copy_length</i> is given,
7126 * No more than <i>copy_length</i> bytes are copied.
7128 * If <i>src_offset</i> is given,
7129 * it specifies the start position of the copy.
7131 * When <i>src_offset</i> is specified and
7132 * <i>src</i> is an IO,
7133 * IO.copy_stream doesn't move the current file offset.
7137 rb_io_s_copy_stream(int argc
, VALUE
*argv
, VALUE io
)
7139 VALUE src
, dst
, length
, src_offset
;
7140 struct copy_stream_struct st
;
7142 MEMZERO(&st
, struct copy_stream_struct
, 1);
7144 rb_scan_args(argc
, argv
, "22", &src
, &dst
, &length
, &src_offset
);
7150 st
.copy_length
= (off_t
)-1;
7152 st
.copy_length
= NUM2OFFT(length
);
7154 if (NIL_P(src_offset
))
7155 st
.src_offset
= (off_t
)-1;
7157 st
.src_offset
= NUM2OFFT(src_offset
);
7159 rb_ensure(copy_stream_body
, (VALUE
)&st
, copy_stream_finalize
, (VALUE
)&st
);
7161 return OFFT2NUM(st
.total
);
7166 * io.external_encoding => encoding
7168 * Returns the Encoding object that represents the encoding of the file.
7169 * If io is write mode and no encoding is specified, returns <code>nil</code>.
7173 rb_io_external_encoding(VALUE io
)
7177 GetOpenFile(io
, fptr
);
7179 return rb_enc_from_encoding(fptr
->enc2
);
7181 if (!fptr
->enc
&& fptr
->fd
== 0) {
7182 fptr
->enc
= rb_default_external_encoding();
7184 if (fptr
->mode
& FMODE_WRITABLE
) {
7186 return rb_enc_from_encoding(fptr
->enc
);
7189 return rb_enc_from_encoding(io_read_encoding(fptr
));
7194 * io.internal_encoding => encoding
7196 * Returns the Encoding of the internal string if conversion is
7197 * specified. Otherwise returns nil.
7201 rb_io_internal_encoding(VALUE io
)
7205 GetOpenFile(io
, fptr
);
7206 if (!fptr
->enc2
) return Qnil
;
7207 return rb_enc_from_encoding(io_read_encoding(fptr
));
7212 * io.set_encoding(ext_enc) => io
7213 * io.set_encoding("ext_enc:int_enc") => io
7214 * io.set_encoding(ext_enc, int_enc) => io
7216 * If single argument is specified, read string from io is tagged
7217 * with the encoding specified. If encoding is a colon separated two
7218 * encoding names "A:B", the read string is converted from encoding A
7219 * (external encoding) to encoding B (internal encoding), then tagged
7220 * with B. If two arguments are specified, those must be encoding
7221 * objects or encoding names, and the first one is the external encoding, and the
7222 * second one is the internal encoding.
7226 rb_io_set_encoding(int argc
, VALUE
*argv
, VALUE io
)
7231 rb_scan_args(argc
, argv
, "11", &v1
, &v2
);
7232 GetOpenFile(io
, fptr
);
7233 io_encoding_set(fptr
, argc
, v1
, v2
);
7238 argf_external_encoding(VALUE argf
)
7240 if (!RTEST(current_file
)) {
7241 return rb_enc_from_encoding(rb_default_external_encoding());
7243 return rb_io_external_encoding(rb_io_check_io(current_file
));
7247 argf_internal_encoding(VALUE argf
)
7249 if (!RTEST(current_file
)) {
7250 return rb_enc_from_encoding(rb_default_external_encoding());
7252 return rb_io_internal_encoding(rb_io_check_io(current_file
));
7256 argf_set_encoding(int argc
, VALUE
*argv
, VALUE argf
)
7261 rb_raise(rb_eArgError
, "no stream to set encoding");
7263 rb_io_set_encoding(argc
, argv
, current_file
);
7264 GetOpenFile(current_file
, fptr
);
7265 argf_enc
= fptr
->enc
;
7266 argf_enc2
= fptr
->enc2
;
7271 argf_tell(VALUE argf
)
7274 rb_raise(rb_eArgError
, "no stream to tell");
7277 return rb_io_tell(current_file
);
7281 argf_seek_m(int argc
, VALUE
*argv
, VALUE argf
)
7284 rb_raise(rb_eArgError
, "no stream to seek");
7286 ARGF_FORWARD(argc
, argv
);
7287 return rb_io_seek_m(argc
, argv
, current_file
);
7291 argf_set_pos(VALUE argf
, VALUE offset
)
7294 rb_raise(rb_eArgError
, "no stream to set position");
7296 ARGF_FORWARD(1, &offset
);
7297 return rb_io_set_pos(current_file
, offset
);
7301 argf_rewind(VALUE argf
)
7304 rb_raise(rb_eArgError
, "no stream to rewind");
7307 return rb_io_rewind(current_file
);
7311 argf_fileno(VALUE argf
)
7314 rb_raise(rb_eArgError
, "no stream");
7317 return rb_io_fileno(current_file
);
7321 argf_to_io(VALUE argf
)
7325 return current_file
;
7329 argf_eof(VALUE argf
)
7332 if (init_p
== 0) return Qtrue
;
7334 if (rb_io_eof(current_file
)) {
7342 argf_read(int argc
, VALUE
*argv
, VALUE argf
)
7344 VALUE tmp
, str
, length
;
7347 rb_scan_args(argc
, argv
, "02", &length
, &str
);
7348 if (!NIL_P(length
)) {
7349 len
= NUM2LONG(argv
[0]);
7353 rb_str_resize(str
,0);
7361 if (ARGF_GENERIC_INPUT_P()) {
7362 tmp
= argf_forward(argc
, argv
, argf
);
7365 tmp
= io_read(argc
, argv
, current_file
);
7367 if (NIL_P(str
)) str
= tmp
;
7368 else if (!NIL_P(tmp
)) rb_str_append(str
, tmp
);
7369 if (NIL_P(tmp
) || NIL_P(length
)) {
7371 argf_close(current_file
);
7376 else if (argc
>= 1) {
7377 if (RSTRING_LEN(str
) < len
) {
7378 len
-= RSTRING_LEN(str
);
7379 argv
[0] = INT2NUM(len
);
7386 struct argf_call_arg
{
7393 argf_forward_call(VALUE arg
)
7395 struct argf_call_arg
*p
= (struct argf_call_arg
*)arg
;
7396 argf_forward(p
->argc
, p
->argv
, p
->argf
);
7401 argf_readpartial(int argc
, VALUE
*argv
, VALUE argf
)
7403 VALUE tmp
, str
, length
;
7405 rb_scan_args(argc
, argv
, "11", &length
, &str
);
7412 rb_str_resize(str
, 0);
7415 if (ARGF_GENERIC_INPUT_P()) {
7416 struct argf_call_arg arg
;
7420 tmp
= rb_rescue2(argf_forward_call
, (VALUE
)&arg
,
7421 RUBY_METHOD_FUNC(0), Qnil
, rb_eEOFError
, (VALUE
)0);
7424 tmp
= io_getpartial(argc
, argv
, current_file
, 0);
7430 argf_close(current_file
);
7432 if (RARRAY_LEN(rb_argv
) == 0)
7435 str
= rb_str_new(NULL
, 0);
7442 argf_getc(VALUE argf
)
7447 if (!next_argv()) return Qnil
;
7448 if (ARGF_GENERIC_INPUT_P()) {
7449 ch
= rb_funcall3(current_file
, rb_intern("getc"), 0, 0);
7452 ch
= rb_io_getc(current_file
);
7454 if (NIL_P(ch
) && next_p
!= -1) {
7455 argf_close(current_file
);
7464 argf_getbyte(VALUE argf
)
7469 if (!next_argv()) return Qnil
;
7470 if (TYPE(current_file
) != T_FILE
) {
7471 ch
= rb_funcall3(current_file
, rb_intern("getbyte"), 0, 0);
7474 ch
= rb_io_getbyte(current_file
);
7476 if (NIL_P(ch
) && next_p
!= -1) {
7477 argf_close(current_file
);
7486 argf_readchar(VALUE argf
)
7491 if (!next_argv()) rb_eof_error();
7492 if (TYPE(current_file
) != T_FILE
) {
7493 ch
= rb_funcall3(current_file
, rb_intern("getc"), 0, 0);
7496 ch
= rb_io_getc(current_file
);
7498 if (NIL_P(ch
) && next_p
!= -1) {
7499 argf_close(current_file
);
7508 argf_readbyte(VALUE argf
)
7512 NEXT_ARGF_FORWARD(0, 0);
7513 c
= argf_getbyte(argf
);
7521 argf_each_line(int argc
, VALUE
*argv
, VALUE argf
)
7523 RETURN_ENUMERATOR(argf
, argc
, argv
);
7525 if (!next_argv()) return Qnil
;
7526 rb_block_call(current_file
, rb_intern("each_line"), argc
, argv
, rb_yield
, 0);
7533 argf_each_byte(VALUE argf
)
7535 RETURN_ENUMERATOR(argf
, 0, 0);
7537 if (!next_argv()) return Qnil
;
7538 rb_block_call(current_file
, rb_intern("each_byte"), 0, 0, rb_yield
, 0);
7544 argf_each_char(VALUE argf
)
7546 RETURN_ENUMERATOR(argf
, 0, 0);
7548 if (!next_argv()) return Qnil
;
7549 rb_block_call(current_file
, rb_intern("each_char"), 0, 0, rb_yield
, 0);
7555 argf_filename(VALUE argf
)
7562 argf_filename_getter(ID id
, VALUE
*var
)
7564 return argf_filename(*var
);
7568 argf_file(VALUE argf
)
7571 return current_file
;
7575 argf_binmode_m(VALUE argf
)
7580 rb_io_binmode(current_file
);
7585 argf_binmode_p(VALUE argf
)
7587 return argf_binmode
? Qtrue
: Qfalse
;
7591 argf_skip(VALUE argf
)
7594 argf_close(current_file
);
7601 argf_close_m(VALUE argf
)
7604 argf_close(current_file
);
7613 argf_closed(VALUE argf
)
7617 return rb_io_closed(current_file
);
7621 argf_to_s(VALUE argf
)
7623 return rb_str_new2("ARGF");
7627 argf_inplace_mode_get(VALUE argf
)
7629 if (!ruby_inplace_mode
) return Qnil
;
7630 return rb_str_new2(ruby_inplace_mode
);
7634 opt_i_get(ID id
, VALUE
*var
)
7636 return argf_inplace_mode_get(*var
);
7640 argf_inplace_mode_set(VALUE argf
, VALUE val
)
7643 if (ruby_inplace_mode
) free(ruby_inplace_mode
);
7644 ruby_inplace_mode
= 0;
7648 if (ruby_inplace_mode
) free(ruby_inplace_mode
);
7649 ruby_inplace_mode
= 0;
7650 ruby_inplace_mode
= strdup(RSTRING_PTR(val
));
7656 opt_i_set(VALUE val
, ID id
, VALUE
*var
)
7658 argf_inplace_mode_set(*var
, val
);
7662 ruby_get_inplace_mode(void)
7664 return ruby_inplace_mode
;
7668 ruby_set_inplace_mode(const char *suffix
)
7670 if (ruby_inplace_mode
) free(ruby_inplace_mode
);
7671 ruby_inplace_mode
= 0;
7672 if (suffix
) ruby_inplace_mode
= strdup(suffix
);
7676 argf_argv(VALUE argf
)
7682 argf_argv_getter(ID id
, VALUE
*var
)
7684 return argf_argv(*var
);
7694 * Class <code>IO</code> is the basis for all input and output in Ruby.
7695 * An I/O stream may be <em>duplexed</em> (that is, bidirectional), and
7696 * so may use more than one native operating system stream.
7698 * Many of the examples in this section use class <code>File</code>,
7699 * the only standard subclass of <code>IO</code>. The two classes are
7700 * closely associated.
7702 * As used in this section, <em>portname</em> may take any of the
7705 * * A plain string represents a filename suitable for the underlying
7708 * * A string starting with ``<code>|</code>'' indicates a subprocess.
7709 * The remainder of the string following the ``<code>|</code>'' is
7710 * invoked as a process with appropriate input/output channels
7713 * * A string equal to ``<code>|-</code>'' will create another Ruby
7714 * instance as a subprocess.
7716 * Ruby will convert pathnames between different operating system
7717 * conventions if possible. For instance, on a Windows system the
7718 * filename ``<code>/gumby/ruby/test.rb</code>'' will be opened as
7719 * ``<code>\gumby\ruby\test.rb</code>''. When specifying a
7720 * Windows-style filename in a Ruby string, remember to escape the
7723 * "c:\\gumby\\ruby\\test.rb"
7725 * Our examples here will use the Unix-style forward slashes;
7726 * <code>File::SEPARATOR</code> can be used to get the
7727 * platform-specific separator character.
7729 * I/O ports may be opened in any one of several different modes, which
7730 * are shown in this section as <em>mode</em>. The mode may
7731 * either be a Fixnum or a String. If numeric, it should be
7732 * one of the operating system specific constants (O_RDONLY,
7733 * O_WRONLY, O_RDWR, O_APPEND and so on). See man open(2) for
7736 * If the mode is given as a String, it must be one of the
7737 * values listed in the following table.
7740 * -----+--------------------------------------------------------
7741 * "r" | Read-only, starts at beginning of file (default mode).
7742 * -----+--------------------------------------------------------
7743 * "r+" | Read-write, starts at beginning of file.
7744 * -----+--------------------------------------------------------
7745 * "w" | Write-only, truncates existing file
7746 * | to zero length or creates a new file for writing.
7747 * -----+--------------------------------------------------------
7748 * "w+" | Read-write, truncates existing file to zero length
7749 * | or creates a new file for reading and writing.
7750 * -----+--------------------------------------------------------
7751 * "a" | Write-only, starts at end of file if file exists,
7752 * | otherwise creates a new file for writing.
7753 * -----+--------------------------------------------------------
7754 * "a+" | Read-write, starts at end of file if file exists,
7755 * | otherwise creates a new file for reading and
7757 * -----+--------------------------------------------------------
7758 * "b" | (DOS/Windows only) Binary file mode (may appear with
7759 * | any of the key letters listed above).
7762 * The global constant ARGF (also accessible as $<) provides an
7763 * IO-like stream which allows access to all files mentioned on the
7764 * command line (or STDIN if no files are mentioned). ARGF provides
7765 * the methods <code>#path</code> and <code>#filename</code> to access
7766 * the name of the file currently being read.
7773 #define rb_intern(str) rb_intern_const(str)
7777 #include <sys/cygwin.h>
7778 static struct __cygwin_perfile pf
[] =
7780 {"", O_RDONLY
| O_BINARY
},
7781 {"", O_WRONLY
| O_BINARY
},
7782 {"", O_RDWR
| O_BINARY
},
7783 {"", O_APPEND
| O_BINARY
},
7786 cygwin_internal(CW_PERFILE
, pf
);
7789 rb_eIOError
= rb_define_class("IOError", rb_eStandardError
);
7790 rb_eEOFError
= rb_define_class("EOFError", rb_eIOError
);
7792 id_write
= rb_intern("write");
7793 id_read
= rb_intern("read");
7794 id_getc
= rb_intern("getc");
7795 id_flush
= rb_intern("flush");
7796 id_encode
= rb_intern("encode");
7797 id_readpartial
= rb_intern("readpartial");
7799 rb_define_global_function("syscall", rb_f_syscall
, -1);
7801 rb_define_global_function("open", rb_f_open
, -1);
7802 rb_define_global_function("printf", rb_f_printf
, -1);
7803 rb_define_global_function("print", rb_f_print
, -1);
7804 rb_define_global_function("putc", rb_f_putc
, 1);
7805 rb_define_global_function("puts", rb_f_puts
, -1);
7806 rb_define_global_function("gets", rb_f_gets
, -1);
7807 rb_define_global_function("readline", rb_f_readline
, -1);
7808 rb_define_global_function("select", rb_f_select
, -1);
7810 rb_define_global_function("readlines", rb_f_readlines
, -1);
7812 rb_define_global_function("`", rb_f_backquote
, 1);
7814 rb_define_global_function("p", rb_f_p
, -1);
7815 rb_define_method(rb_mKernel
, "display", rb_obj_display
, -1);
7817 rb_cIO
= rb_define_class("IO", rb_cObject
);
7818 rb_include_module(rb_cIO
, rb_mEnumerable
);
7820 rb_define_alloc_func(rb_cIO
, io_alloc
);
7821 rb_define_singleton_method(rb_cIO
, "new", rb_io_s_new
, -1);
7822 rb_define_singleton_method(rb_cIO
, "open", rb_io_s_open
, -1);
7823 rb_define_singleton_method(rb_cIO
, "sysopen", rb_io_s_sysopen
, -1);
7824 rb_define_singleton_method(rb_cIO
, "for_fd", rb_io_s_for_fd
, -1);
7825 rb_define_singleton_method(rb_cIO
, "popen", rb_io_s_popen
, -1);
7826 rb_define_singleton_method(rb_cIO
, "foreach", rb_io_s_foreach
, -1);
7827 rb_define_singleton_method(rb_cIO
, "readlines", rb_io_s_readlines
, -1);
7828 rb_define_singleton_method(rb_cIO
, "read", rb_io_s_read
, -1);
7829 rb_define_singleton_method(rb_cIO
, "select", rb_f_select
, -1);
7830 rb_define_singleton_method(rb_cIO
, "pipe", rb_io_s_pipe
, -1);
7831 rb_define_singleton_method(rb_cIO
, "try_convert", rb_io_s_try_convert
, 1);
7832 rb_define_singleton_method(rb_cIO
, "copy_stream", rb_io_s_copy_stream
, -1);
7834 rb_define_method(rb_cIO
, "initialize", rb_io_initialize
, -1);
7836 rb_output_fs
= Qnil
;
7837 rb_define_hooked_variable("$,", &rb_output_fs
, 0, rb_str_setter
);
7839 rb_global_variable(&rb_default_rs
);
7840 rb_rs
= rb_default_rs
= rb_str_new2("\n");
7841 rb_output_rs
= Qnil
;
7842 OBJ_FREEZE(rb_default_rs
); /* avoid modifying RS_default */
7843 rb_define_hooked_variable("$/", &rb_rs
, 0, rb_str_setter
);
7844 rb_define_hooked_variable("$-0", &rb_rs
, 0, rb_str_setter
);
7845 rb_define_hooked_variable("$\\", &rb_output_rs
, 0, rb_str_setter
);
7847 rb_define_virtual_variable("$_", rb_lastline_get
, rb_lastline_set
);
7849 rb_define_method(rb_cIO
, "initialize_copy", rb_io_init_copy
, 1);
7850 rb_define_method(rb_cIO
, "reopen", rb_io_reopen
, -1);
7852 rb_define_method(rb_cIO
, "print", rb_io_print
, -1);
7853 rb_define_method(rb_cIO
, "putc", rb_io_putc
, 1);
7854 rb_define_method(rb_cIO
, "puts", rb_io_puts
, -1);
7855 rb_define_method(rb_cIO
, "printf", rb_io_printf
, -1);
7857 rb_define_method(rb_cIO
, "each", rb_io_each_line
, -1);
7858 rb_define_method(rb_cIO
, "each_line", rb_io_each_line
, -1);
7859 rb_define_method(rb_cIO
, "each_byte", rb_io_each_byte
, 0);
7860 rb_define_method(rb_cIO
, "each_char", rb_io_each_char
, 0);
7861 rb_define_method(rb_cIO
, "lines", rb_io_lines
, -1);
7862 rb_define_method(rb_cIO
, "bytes", rb_io_bytes
, 0);
7863 rb_define_method(rb_cIO
, "chars", rb_io_chars
, 0);
7865 rb_define_method(rb_cIO
, "syswrite", rb_io_syswrite
, 1);
7866 rb_define_method(rb_cIO
, "sysread", rb_io_sysread
, -1);
7868 rb_define_method(rb_cIO
, "fileno", rb_io_fileno
, 0);
7869 rb_define_alias(rb_cIO
, "to_i", "fileno");
7870 rb_define_method(rb_cIO
, "to_io", rb_io_to_io
, 0);
7872 rb_define_method(rb_cIO
, "fsync", rb_io_fsync
, 0);
7873 rb_define_method(rb_cIO
, "sync", rb_io_sync
, 0);
7874 rb_define_method(rb_cIO
, "sync=", rb_io_set_sync
, 1);
7876 rb_define_method(rb_cIO
, "lineno", rb_io_lineno
, 0);
7877 rb_define_method(rb_cIO
, "lineno=", rb_io_set_lineno
, 1);
7879 rb_define_method(rb_cIO
, "readlines", rb_io_readlines
, -1);
7881 rb_define_method(rb_cIO
, "read_nonblock", io_read_nonblock
, -1);
7882 rb_define_method(rb_cIO
, "write_nonblock", rb_io_write_nonblock
, 1);
7883 rb_define_method(rb_cIO
, "readpartial", io_readpartial
, -1);
7884 rb_define_method(rb_cIO
, "read", io_read
, -1);
7885 rb_define_method(rb_cIO
, "write", io_write
, 1);
7886 rb_define_method(rb_cIO
, "gets", rb_io_gets_m
, -1);
7887 rb_define_method(rb_cIO
, "readline", rb_io_readline
, -1);
7888 rb_define_method(rb_cIO
, "getc", rb_io_getc
, 0);
7889 rb_define_method(rb_cIO
, "getbyte", rb_io_getbyte
, 0);
7890 rb_define_method(rb_cIO
, "readchar", rb_io_readchar
, 0);
7891 rb_define_method(rb_cIO
, "readbyte", rb_io_readbyte
, 0);
7892 rb_define_method(rb_cIO
, "ungetc",rb_io_ungetc
, 1);
7893 rb_define_method(rb_cIO
, "<<", rb_io_addstr
, 1);
7894 rb_define_method(rb_cIO
, "flush", rb_io_flush
, 0);
7895 rb_define_method(rb_cIO
, "tell", rb_io_tell
, 0);
7896 rb_define_method(rb_cIO
, "seek", rb_io_seek_m
, -1);
7897 rb_define_const(rb_cIO
, "SEEK_SET", INT2FIX(SEEK_SET
));
7898 rb_define_const(rb_cIO
, "SEEK_CUR", INT2FIX(SEEK_CUR
));
7899 rb_define_const(rb_cIO
, "SEEK_END", INT2FIX(SEEK_END
));
7900 rb_define_method(rb_cIO
, "rewind", rb_io_rewind
, 0);
7901 rb_define_method(rb_cIO
, "pos", rb_io_tell
, 0);
7902 rb_define_method(rb_cIO
, "pos=", rb_io_set_pos
, 1);
7903 rb_define_method(rb_cIO
, "eof", rb_io_eof
, 0);
7904 rb_define_method(rb_cIO
, "eof?", rb_io_eof
, 0);
7906 rb_define_method(rb_cIO
, "close_on_exec?", rb_io_close_on_exec_p
, 0);
7907 rb_define_method(rb_cIO
, "close_on_exec=", rb_io_set_close_on_exec
, 1);
7909 rb_define_method(rb_cIO
, "close", rb_io_close_m
, 0);
7910 rb_define_method(rb_cIO
, "closed?", rb_io_closed
, 0);
7911 rb_define_method(rb_cIO
, "close_read", rb_io_close_read
, 0);
7912 rb_define_method(rb_cIO
, "close_write", rb_io_close_write
, 0);
7914 rb_define_method(rb_cIO
, "isatty", rb_io_isatty
, 0);
7915 rb_define_method(rb_cIO
, "tty?", rb_io_isatty
, 0);
7916 rb_define_method(rb_cIO
, "binmode", rb_io_binmode_m
, 0);
7917 rb_define_method(rb_cIO
, "binmode?", rb_io_binmode_p
, 0);
7918 rb_define_method(rb_cIO
, "sysseek", rb_io_sysseek
, -1);
7920 rb_define_method(rb_cIO
, "ioctl", rb_io_ioctl
, -1);
7921 rb_define_method(rb_cIO
, "fcntl", rb_io_fcntl
, -1);
7922 rb_define_method(rb_cIO
, "pid", rb_io_pid
, 0);
7923 rb_define_method(rb_cIO
, "inspect", rb_io_inspect
, 0);
7925 rb_define_method(rb_cIO
, "external_encoding", rb_io_external_encoding
, 0);
7926 rb_define_method(rb_cIO
, "internal_encoding", rb_io_internal_encoding
, 0);
7927 rb_define_method(rb_cIO
, "set_encoding", rb_io_set_encoding
, -1);
7929 rb_define_variable("$stdin", &rb_stdin
);
7930 rb_stdin
= prep_stdio(stdin
, FMODE_READABLE
, rb_cIO
, "<STDIN>");
7931 rb_define_hooked_variable("$stdout", &rb_stdout
, 0, stdout_setter
);
7932 rb_stdout
= prep_stdio(stdout
, FMODE_WRITABLE
, rb_cIO
, "<STDOUT>");
7933 rb_define_hooked_variable("$stderr", &rb_stderr
, 0, stdout_setter
);
7934 rb_stderr
= prep_stdio(stderr
, FMODE_WRITABLE
|FMODE_SYNC
, rb_cIO
, "<STDERR>");
7935 rb_define_hooked_variable("$>", &rb_stdout
, 0, stdout_setter
);
7936 orig_stdout
= rb_stdout
;
7937 rb_deferr
= orig_stderr
= rb_stderr
;
7939 /* constants to hold original stdin/stdout/stderr */
7940 rb_define_global_const("STDIN", rb_stdin
);
7941 rb_define_global_const("STDOUT", rb_stdout
);
7942 rb_define_global_const("STDERR", rb_stderr
);
7944 rb_cARGF
= rb_class_new(rb_cObject
);
7945 rb_set_class_path(rb_cARGF
, rb_cObject
, "ARGF.class");
7946 rb_define_alloc_func(rb_cARGF
, argf_alloc
);
7948 rb_include_module(rb_cARGF
, rb_mEnumerable
);
7950 rb_define_method(rb_cARGF
, "initialize", argf_initialize
, -2);
7951 rb_define_method(rb_cARGF
, "initialize_copy", argf_initialize_copy
, 1);
7952 rb_define_method(rb_cARGF
, "to_s", argf_to_s
, 0);
7953 rb_define_method(rb_cARGF
, "argv", argf_argv
, 0);
7955 rb_define_method(rb_cARGF
, "fileno", argf_fileno
, 0);
7956 rb_define_method(rb_cARGF
, "to_i", argf_fileno
, 0);
7957 rb_define_method(rb_cARGF
, "to_io", argf_to_io
, 0);
7958 rb_define_method(rb_cARGF
, "each", argf_each_line
, -1);
7959 rb_define_method(rb_cARGF
, "each_line", argf_each_line
, -1);
7960 rb_define_method(rb_cARGF
, "each_byte", argf_each_byte
, 0);
7961 rb_define_method(rb_cARGF
, "each_char", argf_each_char
, 0);
7962 rb_define_method(rb_cARGF
, "lines", argf_each_line
, -1);
7963 rb_define_method(rb_cARGF
, "bytes", argf_each_byte
, 0);
7964 rb_define_method(rb_cARGF
, "chars", argf_each_char
, 0);
7966 rb_define_method(rb_cARGF
, "read", argf_read
, -1);
7967 rb_define_method(rb_cARGF
, "readpartial", argf_readpartial
, -1);
7968 rb_define_method(rb_cARGF
, "readlines", argf_readlines
, -1);
7969 rb_define_method(rb_cARGF
, "to_a", argf_readlines
, -1);
7970 rb_define_method(rb_cARGF
, "gets", argf_gets
, -1);
7971 rb_define_method(rb_cARGF
, "readline", argf_readline
, -1);
7972 rb_define_method(rb_cARGF
, "getc", argf_getc
, 0);
7973 rb_define_method(rb_cARGF
, "getbyte", argf_getbyte
, 0);
7974 rb_define_method(rb_cARGF
, "readchar", argf_readchar
, 0);
7975 rb_define_method(rb_cARGF
, "readbyte", argf_readbyte
, 0);
7976 rb_define_method(rb_cARGF
, "tell", argf_tell
, 0);
7977 rb_define_method(rb_cARGF
, "seek", argf_seek_m
, -1);
7978 rb_define_method(rb_cARGF
, "rewind", argf_rewind
, 0);
7979 rb_define_method(rb_cARGF
, "pos", argf_tell
, 0);
7980 rb_define_method(rb_cARGF
, "pos=", argf_set_pos
, 1);
7981 rb_define_method(rb_cARGF
, "eof", argf_eof
, 0);
7982 rb_define_method(rb_cARGF
, "eof?", argf_eof
, 0);
7983 rb_define_method(rb_cARGF
, "binmode", argf_binmode_m
, 0);
7984 rb_define_method(rb_cARGF
, "binmode?", argf_binmode_p
, 0);
7986 rb_define_method(rb_cARGF
, "filename", argf_filename
, 0);
7987 rb_define_method(rb_cARGF
, "path", argf_filename
, 0);
7988 rb_define_method(rb_cARGF
, "file", argf_file
, 0);
7989 rb_define_method(rb_cARGF
, "skip", argf_skip
, 0);
7990 rb_define_method(rb_cARGF
, "close", argf_close_m
, 0);
7991 rb_define_method(rb_cARGF
, "closed?", argf_closed
, 0);
7993 rb_define_method(rb_cARGF
, "lineno", argf_lineno
, 0);
7994 rb_define_method(rb_cARGF
, "lineno=", argf_set_lineno
, 1);
7996 rb_define_method(rb_cARGF
, "inplace_mode", argf_inplace_mode_get
, 0);
7997 rb_define_method(rb_cARGF
, "inplace_mode=", argf_inplace_mode_set
, 1);
7999 rb_define_method(rb_cARGF
, "external_encoding", argf_external_encoding
, 0);
8000 rb_define_method(rb_cARGF
, "internal_encoding", argf_internal_encoding
, 0);
8001 rb_define_method(rb_cARGF
, "set_encoding", argf_set_encoding
, -1);
8003 argf
= rb_class_new_instance(0, 0, rb_cARGF
);
8005 rb_define_readonly_variable("$<", &argf
);
8006 rb_define_global_const("ARGF", argf
);
8008 rb_define_hooked_variable("$.", &argf
, argf_lineno_getter
, argf_lineno_setter
);
8009 rb_define_hooked_variable("$FILENAME", &argf
, argf_filename_getter
, 0);
8010 filename
= rb_str_new2("-");
8012 rb_define_hooked_variable("$-i", &argf
, opt_i_get
, opt_i_set
);
8013 rb_define_hooked_variable("$*", &argf
, argf_argv_getter
, 0);
8015 #if defined (_WIN32) || defined(DJGPP) || defined(__CYGWIN__) || defined(__human68k__)
8016 atexit(pipe_atexit
);
8021 rb_define_method(rb_cFile
, "initialize", rb_file_initialize
, -1);
8023 rb_file_const("RDONLY", INT2FIX(O_RDONLY
));
8024 rb_file_const("WRONLY", INT2FIX(O_WRONLY
));
8025 rb_file_const("RDWR", INT2FIX(O_RDWR
));
8026 rb_file_const("APPEND", INT2FIX(O_APPEND
));
8027 rb_file_const("CREAT", INT2FIX(O_CREAT
));
8028 rb_file_const("EXCL", INT2FIX(O_EXCL
));
8029 #if defined(O_NDELAY) || defined(O_NONBLOCK)
8031 rb_file_const("NONBLOCK", INT2FIX(O_NONBLOCK
));
8033 rb_file_const("NONBLOCK", INT2FIX(O_NDELAY
));
8036 rb_file_const("TRUNC", INT2FIX(O_TRUNC
));
8038 rb_file_const("NOCTTY", INT2FIX(O_NOCTTY
));
8041 rb_file_const("BINARY", INT2FIX(O_BINARY
));
8043 rb_file_const("BINARY", INT2FIX(0));
8046 rb_file_const("SYNC", INT2FIX(O_SYNC
));
8049 sym_mode
= ID2SYM(rb_intern("mode"));
8050 sym_perm
= ID2SYM(rb_intern("perm"));
8051 sym_extenc
= ID2SYM(rb_intern("external_encoding"));
8052 sym_intenc
= ID2SYM(rb_intern("internal_encoding"));
8053 sym_encoding
= ID2SYM(rb_intern("encoding"));
8054 sym_open_args
= ID2SYM(rb_intern("open_args"));