mirror of
https://github.com/ruby/ruby.git
synced 2022-11-09 12:17:21 -05:00
80b5a0ff2a
Cfuncs that use rb_scan_args with the : entry suffer similar keyword argument separation issues that Ruby methods suffer if the cfuncs accept optional or variable arguments. This makes the following changes to : handling. * Treats as **kw, prompting keyword argument separation warnings if called with a positional hash. * Do not look for an option hash if empty keywords are provided. For backwards compatibility, treat an empty keyword splat as a empty mandatory positional hash argument, but emit a a warning, as this behavior will be removed in Ruby 3. The argument number check needs to be moved lower so it can correctly handle an empty positional argument being added. * If the last argument is nil and it is necessary to treat it as an option hash in order to make sure all arguments are processed, continue to treat the last argument as the option hash. Emit a warning in this case, as this behavior will be removed in Ruby 3. * If splitting the keyword hash into two hashes, issue a warning, as we will not be splitting hashes in Ruby 3. * If the keyword argument is required to fill a mandatory positional argument, continue to do so, but emit a warning as this behavior will be going away in Ruby 3. * If keyword arguments are provided and the last argument is not a hash, that indicates something wrong. This can happen if a cfunc is calling rb_scan_args multiple times, and providing arguments that were not passed to it from Ruby. Callers need to switch to the new rb_scan_args_kw function, which allows passing of whether keywords were provided. This commit fixes all warnings caused by the changes above. It switches some function calls to *_kw versions with appropriate kw_splat flags. If delegating arguments, RB_PASS_CALLED_KEYWORDS is used. If creating new arguments, RB_PASS_KEYWORDS is used if the last argument is a hash to be treated as keywords. In open_key_args in io.c, use rb_scan_args_kw. In this case, the arguments provided come from another C function, not Ruby. The last argument may or may not be a hash, so we can't set keyword argument mode. However, if it is a hash, we don't want to warn when treating it as keywords. In Ruby files, make sure to appropriately use keyword splats or literal keywords when calling Cfuncs that now issue keyword argument separation warnings through rb_scan_args. Also, make sure not to pass nil in place of an option hash. Work around Kernel#warn warnings due to problems in the Rubygems override of the method. There is an open pull request to fix these issues in Rubygems, but part of the Rubygems tests for their override fail on ruby-head due to rb_scan_args not recognizing empty keyword splats, which this commit fixes. Implementation wise, adding rb_scan_args_kw is kind of a pain, because rb_scan_args takes a variable number of arguments. In order to not duplicate all the code, the function internals need to be split into two functions taking a va_list, and to avoid passing in a ton of arguments, a single struct argument is used to handle the variables previously local to the function.
13527 lines
342 KiB
C
13527 lines
342 KiB
C
/**********************************************************************
|
|
|
|
io.c -
|
|
|
|
$Author$
|
|
created at: Fri Oct 15 18:08:59 JST 1993
|
|
|
|
Copyright (C) 1993-2007 Yukihiro Matsumoto
|
|
Copyright (C) 2000 Network Applied Communication Laboratory, Inc.
|
|
Copyright (C) 2000 Information-technology Promotion Agency, Japan
|
|
|
|
**********************************************************************/
|
|
|
|
#include "ruby/encoding.h"
|
|
#include "ruby/io.h"
|
|
#include "ruby/thread.h"
|
|
#include "internal.h"
|
|
#include "dln.h"
|
|
#include "encindex.h"
|
|
#include "id.h"
|
|
#include <ctype.h>
|
|
#include <errno.h>
|
|
#include "ruby_atomic.h"
|
|
#include "ccan/list/list.h"
|
|
|
|
#undef free
|
|
#define free(x) xfree(x)
|
|
|
|
#if defined(DOSISH) || defined(__CYGWIN__)
|
|
#include <io.h>
|
|
#endif
|
|
|
|
#include <sys/types.h>
|
|
#if defined HAVE_NET_SOCKET_H
|
|
# include <net/socket.h>
|
|
#elif defined HAVE_SYS_SOCKET_H
|
|
# include <sys/socket.h>
|
|
#endif
|
|
|
|
#if defined(__BOW__) || defined(__CYGWIN__) || defined(_WIN32)
|
|
# define NO_SAFE_RENAME
|
|
#endif
|
|
|
|
#if defined(__FreeBSD__) || defined(__NetBSD__) || defined(__OpenBSD__) || defined(__DragonFly__) || defined(__sun) || defined(_nec_ews)
|
|
# define USE_SETVBUF
|
|
#endif
|
|
|
|
#ifdef __QNXNTO__
|
|
#include <unix.h>
|
|
#endif
|
|
|
|
#include <sys/types.h>
|
|
#if defined(HAVE_SYS_IOCTL_H) && !defined(_WIN32)
|
|
#include <sys/ioctl.h>
|
|
#endif
|
|
#if defined(HAVE_FCNTL_H) || defined(_WIN32)
|
|
#include <fcntl.h>
|
|
#elif defined(HAVE_SYS_FCNTL_H)
|
|
#include <sys/fcntl.h>
|
|
#endif
|
|
|
|
#if !HAVE_OFF_T && !defined(off_t)
|
|
# define off_t long
|
|
#endif
|
|
|
|
#ifdef HAVE_SYS_TIME_H
|
|
# include <sys/time.h>
|
|
#endif
|
|
|
|
#include <sys/stat.h>
|
|
|
|
#if defined(HAVE_SYS_PARAM_H) || defined(__HIUX_MPP__)
|
|
# include <sys/param.h>
|
|
#endif
|
|
|
|
#if !defined NOFILE
|
|
# define NOFILE 64
|
|
#endif
|
|
|
|
#ifdef HAVE_UNISTD_H
|
|
#include <unistd.h>
|
|
#endif
|
|
|
|
#ifdef HAVE_SYSCALL_H
|
|
#include <syscall.h>
|
|
#elif defined HAVE_SYS_SYSCALL_H
|
|
#include <sys/syscall.h>
|
|
#endif
|
|
|
|
#ifdef HAVE_SYS_UIO_H
|
|
#include <sys/uio.h>
|
|
#endif
|
|
|
|
#ifdef HAVE_SYS_WAIT_H
|
|
# include <sys/wait.h> /* for WNOHANG on BSD */
|
|
#endif
|
|
|
|
#ifdef HAVE_COPYFILE_H
|
|
# include <copyfile.h>
|
|
#endif
|
|
|
|
#include "ruby/util.h"
|
|
|
|
#ifndef O_ACCMODE
|
|
#define O_ACCMODE (O_RDONLY | O_WRONLY | O_RDWR)
|
|
#endif
|
|
|
|
#if SIZEOF_OFF_T > SIZEOF_LONG && !defined(HAVE_LONG_LONG)
|
|
# error off_t is bigger than long, but you have no long long...
|
|
#endif
|
|
|
|
#ifndef PIPE_BUF
|
|
# ifdef _POSIX_PIPE_BUF
|
|
# define PIPE_BUF _POSIX_PIPE_BUF
|
|
# else
|
|
# define PIPE_BUF 512 /* is this ok? */
|
|
# endif
|
|
#endif
|
|
|
|
#ifndef EWOULDBLOCK
|
|
# define EWOULDBLOCK EAGAIN
|
|
#endif
|
|
|
|
#if defined(HAVE___SYSCALL) && (defined(__APPLE__) || defined(__OpenBSD__))
|
|
/* Mac OS X and OpenBSD have __syscall but don't define it in headers */
|
|
off_t __syscall(quad_t number, ...);
|
|
#endif
|
|
|
|
#define IO_RBUF_CAPA_MIN 8192
|
|
#define IO_CBUF_CAPA_MIN (128*1024)
|
|
#define IO_RBUF_CAPA_FOR(fptr) (NEED_READCONV(fptr) ? IO_CBUF_CAPA_MIN : IO_RBUF_CAPA_MIN)
|
|
#define IO_WBUF_CAPA_MIN 8192
|
|
|
|
/* define system APIs */
|
|
#ifdef _WIN32
|
|
#undef open
|
|
#define open rb_w32_uopen
|
|
#undef rename
|
|
#define rename(f, t) rb_w32_urename((f), (t))
|
|
#endif
|
|
|
|
#if defined(_WIN32)
|
|
# define RUBY_PIPE_NONBLOCK_DEFAULT (0)
|
|
#elif defined(O_NONBLOCK)
|
|
/* disabled for [Bug #15356] (Rack::Deflater + rails) failure: */
|
|
# define RUBY_PIPE_NONBLOCK_DEFAULT (0)
|
|
#else /* any platforms where O_NONBLOCK does not exist? */
|
|
# define RUBY_PIPE_NONBLOCK_DEFAULT (0)
|
|
#endif
|
|
|
|
VALUE rb_cIO;
|
|
VALUE rb_eEOFError;
|
|
VALUE rb_eIOError;
|
|
VALUE rb_mWaitReadable;
|
|
VALUE rb_mWaitWritable;
|
|
|
|
static VALUE rb_eEAGAINWaitReadable;
|
|
static VALUE rb_eEAGAINWaitWritable;
|
|
static VALUE rb_eEWOULDBLOCKWaitReadable;
|
|
static VALUE rb_eEWOULDBLOCKWaitWritable;
|
|
static VALUE rb_eEINPROGRESSWaitWritable;
|
|
static VALUE rb_eEINPROGRESSWaitReadable;
|
|
|
|
VALUE rb_stdin, rb_stdout, rb_stderr;
|
|
static VALUE orig_stdout, orig_stderr;
|
|
|
|
VALUE rb_output_fs;
|
|
VALUE rb_rs;
|
|
VALUE rb_output_rs;
|
|
VALUE rb_default_rs;
|
|
|
|
static VALUE argf;
|
|
|
|
static ID id_write, id_read, id_getc, id_flush, id_readpartial, id_set_encoding;
|
|
static VALUE sym_mode, sym_perm, sym_flags, sym_extenc, sym_intenc, sym_encoding, sym_open_args;
|
|
static VALUE sym_textmode, sym_binmode, sym_autoclose;
|
|
static VALUE sym_SET, sym_CUR, sym_END;
|
|
static VALUE sym_wait_readable, sym_wait_writable;
|
|
#ifdef SEEK_DATA
|
|
static VALUE sym_DATA;
|
|
#endif
|
|
#ifdef SEEK_HOLE
|
|
static VALUE sym_HOLE;
|
|
#endif
|
|
|
|
struct argf {
|
|
VALUE filename, current_file;
|
|
long last_lineno; /* $. */
|
|
long lineno;
|
|
VALUE argv;
|
|
VALUE inplace;
|
|
struct rb_io_enc_t encs;
|
|
int8_t init_p, next_p, binmode;
|
|
};
|
|
|
|
static rb_atomic_t max_file_descriptor = NOFILE;
|
|
void
|
|
rb_update_max_fd(int fd)
|
|
{
|
|
rb_atomic_t afd = (rb_atomic_t)fd;
|
|
rb_atomic_t max_fd = max_file_descriptor;
|
|
int err;
|
|
|
|
if (afd <= max_fd)
|
|
return;
|
|
|
|
#if defined(HAVE_FCNTL) && defined(F_GETFL)
|
|
err = fcntl(fd, F_GETFL) == -1;
|
|
#else
|
|
{
|
|
struct stat buf;
|
|
err = fstat(fd, &buf) != 0;
|
|
}
|
|
#endif
|
|
if (err && errno == EBADF) {
|
|
rb_bug("rb_update_max_fd: invalid fd (%d) given.", fd);
|
|
}
|
|
|
|
while (max_fd < afd) {
|
|
max_fd = ATOMIC_CAS(max_file_descriptor, max_fd, afd);
|
|
}
|
|
}
|
|
|
|
void
|
|
rb_maygvl_fd_fix_cloexec(int fd)
|
|
{
|
|
/* MinGW don't have F_GETFD and FD_CLOEXEC. [ruby-core:40281] */
|
|
#if defined(HAVE_FCNTL) && defined(F_GETFD) && defined(F_SETFD) && defined(FD_CLOEXEC)
|
|
int flags, flags2, ret;
|
|
flags = fcntl(fd, F_GETFD); /* should not fail except EBADF. */
|
|
if (flags == -1) {
|
|
rb_bug("rb_maygvl_fd_fix_cloexec: fcntl(%d, F_GETFD) failed: %s", fd, strerror(errno));
|
|
}
|
|
if (fd <= 2)
|
|
flags2 = flags & ~FD_CLOEXEC; /* Clear CLOEXEC for standard file descriptors: 0, 1, 2. */
|
|
else
|
|
flags2 = flags | FD_CLOEXEC; /* Set CLOEXEC for non-standard file descriptors: 3, 4, 5, ... */
|
|
if (flags != flags2) {
|
|
ret = fcntl(fd, F_SETFD, flags2);
|
|
if (ret != 0) {
|
|
rb_bug("rb_maygvl_fd_fix_cloexec: fcntl(%d, F_SETFD, %d) failed: %s", fd, flags2, strerror(errno));
|
|
}
|
|
}
|
|
#endif
|
|
}
|
|
|
|
void
|
|
rb_fd_fix_cloexec(int fd)
|
|
{
|
|
rb_maygvl_fd_fix_cloexec(fd);
|
|
rb_update_max_fd(fd);
|
|
}
|
|
|
|
/* this is only called once */
|
|
static int
|
|
rb_fix_detect_o_cloexec(int fd)
|
|
{
|
|
#if defined(O_CLOEXEC) && defined(F_GETFD)
|
|
int flags = fcntl(fd, F_GETFD);
|
|
|
|
if (flags == -1)
|
|
rb_bug("rb_fix_detect_o_cloexec: fcntl(%d, F_GETFD) failed: %s", fd, strerror(errno));
|
|
|
|
if (flags & FD_CLOEXEC)
|
|
return 1;
|
|
#endif /* fall through if O_CLOEXEC does not work: */
|
|
rb_maygvl_fd_fix_cloexec(fd);
|
|
return 0;
|
|
}
|
|
|
|
int
|
|
rb_cloexec_open(const char *pathname, int flags, mode_t mode)
|
|
{
|
|
int ret;
|
|
static int o_cloexec_state = -1; /* <0: unknown, 0: ignored, >0: working */
|
|
|
|
#ifdef O_CLOEXEC
|
|
/* O_CLOEXEC is available since Linux 2.6.23. Linux 2.6.18 silently ignore it. */
|
|
flags |= O_CLOEXEC;
|
|
#elif defined O_NOINHERIT
|
|
flags |= O_NOINHERIT;
|
|
#endif
|
|
ret = open(pathname, flags, mode);
|
|
if (ret < 0) return ret;
|
|
if (ret <= 2 || o_cloexec_state == 0) {
|
|
rb_maygvl_fd_fix_cloexec(ret);
|
|
}
|
|
else if (o_cloexec_state > 0) {
|
|
return ret;
|
|
}
|
|
else {
|
|
o_cloexec_state = rb_fix_detect_o_cloexec(ret);
|
|
}
|
|
return ret;
|
|
}
|
|
|
|
int
|
|
rb_cloexec_dup(int oldfd)
|
|
{
|
|
/* Don't allocate standard file descriptors: 0, 1, 2 */
|
|
return rb_cloexec_fcntl_dupfd(oldfd, 3);
|
|
}
|
|
|
|
int
|
|
rb_cloexec_dup2(int oldfd, int newfd)
|
|
{
|
|
int ret;
|
|
|
|
/* When oldfd == newfd, dup2 succeeds but dup3 fails with EINVAL.
|
|
* rb_cloexec_dup2 succeeds as dup2. */
|
|
if (oldfd == newfd) {
|
|
ret = newfd;
|
|
}
|
|
else {
|
|
#if defined(HAVE_DUP3) && defined(O_CLOEXEC)
|
|
static int try_dup3 = 1;
|
|
if (2 < newfd && try_dup3) {
|
|
ret = dup3(oldfd, newfd, O_CLOEXEC);
|
|
if (ret != -1)
|
|
return ret;
|
|
/* dup3 is available since Linux 2.6.27, glibc 2.9. */
|
|
if (errno == ENOSYS) {
|
|
try_dup3 = 0;
|
|
ret = dup2(oldfd, newfd);
|
|
}
|
|
}
|
|
else {
|
|
ret = dup2(oldfd, newfd);
|
|
}
|
|
#else
|
|
ret = dup2(oldfd, newfd);
|
|
#endif
|
|
if (ret < 0) return ret;
|
|
}
|
|
rb_maygvl_fd_fix_cloexec(ret);
|
|
return ret;
|
|
}
|
|
|
|
static int
|
|
rb_fd_set_nonblock(int fd)
|
|
{
|
|
#ifdef _WIN32
|
|
return rb_w32_set_nonblock(fd);
|
|
#elif defined(F_GETFL)
|
|
int oflags = fcntl(fd, F_GETFL);
|
|
|
|
if (oflags == -1)
|
|
return -1;
|
|
if (oflags & O_NONBLOCK)
|
|
return 0;
|
|
oflags |= O_NONBLOCK;
|
|
return fcntl(fd, F_SETFL, oflags);
|
|
#endif
|
|
return 0;
|
|
}
|
|
|
|
int
|
|
rb_cloexec_pipe(int fildes[2])
|
|
{
|
|
int ret;
|
|
|
|
#if defined(HAVE_PIPE2)
|
|
static int try_pipe2 = 1;
|
|
if (try_pipe2) {
|
|
ret = pipe2(fildes, O_CLOEXEC | RUBY_PIPE_NONBLOCK_DEFAULT);
|
|
if (ret != -1)
|
|
return ret;
|
|
/* pipe2 is available since Linux 2.6.27, glibc 2.9. */
|
|
if (errno == ENOSYS) {
|
|
try_pipe2 = 0;
|
|
ret = pipe(fildes);
|
|
}
|
|
}
|
|
else {
|
|
ret = pipe(fildes);
|
|
}
|
|
#else
|
|
ret = pipe(fildes);
|
|
#endif
|
|
if (ret < 0) return ret;
|
|
#ifdef __CYGWIN__
|
|
if (ret == 0 && fildes[1] == -1) {
|
|
close(fildes[0]);
|
|
fildes[0] = -1;
|
|
errno = ENFILE;
|
|
return -1;
|
|
}
|
|
#endif
|
|
rb_maygvl_fd_fix_cloexec(fildes[0]);
|
|
rb_maygvl_fd_fix_cloexec(fildes[1]);
|
|
if (RUBY_PIPE_NONBLOCK_DEFAULT) {
|
|
rb_fd_set_nonblock(fildes[0]);
|
|
rb_fd_set_nonblock(fildes[1]);
|
|
}
|
|
return ret;
|
|
}
|
|
|
|
int
|
|
rb_cloexec_fcntl_dupfd(int fd, int minfd)
|
|
{
|
|
int ret;
|
|
|
|
#if defined(HAVE_FCNTL) && defined(F_DUPFD_CLOEXEC) && defined(F_DUPFD)
|
|
static int try_dupfd_cloexec = 1;
|
|
if (try_dupfd_cloexec) {
|
|
ret = fcntl(fd, F_DUPFD_CLOEXEC, minfd);
|
|
if (ret != -1) {
|
|
if (ret <= 2)
|
|
rb_maygvl_fd_fix_cloexec(ret);
|
|
return ret;
|
|
}
|
|
/* F_DUPFD_CLOEXEC is available since Linux 2.6.24. Linux 2.6.18 fails with EINVAL */
|
|
if (errno == EINVAL) {
|
|
ret = fcntl(fd, F_DUPFD, minfd);
|
|
if (ret != -1) {
|
|
try_dupfd_cloexec = 0;
|
|
}
|
|
}
|
|
}
|
|
else {
|
|
ret = fcntl(fd, F_DUPFD, minfd);
|
|
}
|
|
#elif defined(HAVE_FCNTL) && defined(F_DUPFD)
|
|
ret = fcntl(fd, F_DUPFD, minfd);
|
|
#elif defined(HAVE_DUP)
|
|
ret = dup(fd);
|
|
if (ret >= 0 && ret < minfd) {
|
|
const int prev_fd = ret;
|
|
ret = rb_cloexec_fcntl_dupfd(fd, minfd);
|
|
close(prev_fd);
|
|
}
|
|
return ret;
|
|
#else
|
|
# error "dup() or fcntl(F_DUPFD) must be supported."
|
|
#endif
|
|
if (ret < 0) return ret;
|
|
rb_maygvl_fd_fix_cloexec(ret);
|
|
return ret;
|
|
}
|
|
|
|
#define argf_of(obj) (*(struct argf *)DATA_PTR(obj))
|
|
#define ARGF argf_of(argf)
|
|
|
|
#define GetWriteIO(io) rb_io_get_write_io(io)
|
|
|
|
#define READ_DATA_PENDING(fptr) ((fptr)->rbuf.len)
|
|
#define READ_DATA_PENDING_COUNT(fptr) ((fptr)->rbuf.len)
|
|
#define READ_DATA_PENDING_PTR(fptr) ((fptr)->rbuf.ptr+(fptr)->rbuf.off)
|
|
#define READ_DATA_BUFFERED(fptr) READ_DATA_PENDING(fptr)
|
|
|
|
#define READ_CHAR_PENDING(fptr) ((fptr)->cbuf.len)
|
|
#define READ_CHAR_PENDING_COUNT(fptr) ((fptr)->cbuf.len)
|
|
#define READ_CHAR_PENDING_PTR(fptr) ((fptr)->cbuf.ptr+(fptr)->cbuf.off)
|
|
|
|
#if defined(_WIN32)
|
|
#define WAIT_FD_IN_WIN32(fptr) \
|
|
(rb_w32_io_cancelable_p((fptr)->fd) ? 0 : rb_thread_wait_fd((fptr)->fd))
|
|
#else
|
|
#define WAIT_FD_IN_WIN32(fptr)
|
|
#endif
|
|
|
|
#define READ_CHECK(fptr) do {\
|
|
if (!READ_DATA_PENDING(fptr)) {\
|
|
WAIT_FD_IN_WIN32(fptr);\
|
|
rb_io_check_closed(fptr);\
|
|
}\
|
|
} while(0)
|
|
|
|
#ifndef S_ISSOCK
|
|
# ifdef _S_ISSOCK
|
|
# define S_ISSOCK(m) _S_ISSOCK(m)
|
|
# else
|
|
# ifdef _S_IFSOCK
|
|
# define S_ISSOCK(m) (((m) & S_IFMT) == _S_IFSOCK)
|
|
# else
|
|
# ifdef S_IFSOCK
|
|
# define S_ISSOCK(m) (((m) & S_IFMT) == S_IFSOCK)
|
|
# endif
|
|
# endif
|
|
# endif
|
|
#endif
|
|
|
|
static int io_fflush(rb_io_t *);
|
|
static rb_io_t *flush_before_seek(rb_io_t *fptr);
|
|
|
|
#define NEED_NEWLINE_DECORATOR_ON_READ(fptr) ((fptr)->mode & FMODE_TEXTMODE)
|
|
#define NEED_NEWLINE_DECORATOR_ON_WRITE(fptr) ((fptr)->mode & FMODE_TEXTMODE)
|
|
#if defined(RUBY_TEST_CRLF_ENVIRONMENT) || defined(_WIN32)
|
|
/* Windows */
|
|
# define DEFAULT_TEXTMODE FMODE_TEXTMODE
|
|
# define TEXTMODE_NEWLINE_DECORATOR_ON_WRITE ECONV_CRLF_NEWLINE_DECORATOR
|
|
/*
|
|
* CRLF newline is set as default newline decorator.
|
|
* If only CRLF newline conversion is needed, we use binary IO process
|
|
* with OS's text mode for IO performance improvement.
|
|
* If encoding conversion is needed or a user sets text mode, we use encoding
|
|
* conversion IO process and universal newline decorator by default.
|
|
*/
|
|
#define NEED_READCONV(fptr) ((fptr)->encs.enc2 != NULL || (fptr)->encs.ecflags & ~ECONV_CRLF_NEWLINE_DECORATOR)
|
|
#define WRITECONV_MASK ( \
|
|
(ECONV_DECORATOR_MASK & ~ECONV_CRLF_NEWLINE_DECORATOR)|\
|
|
ECONV_STATEFUL_DECORATOR_MASK|\
|
|
0)
|
|
#define NEED_WRITECONV(fptr) ( \
|
|
((fptr)->encs.enc != NULL && (fptr)->encs.enc != rb_ascii8bit_encoding()) || \
|
|
((fptr)->encs.ecflags & WRITECONV_MASK) || \
|
|
0)
|
|
#define SET_BINARY_MODE(fptr) setmode((fptr)->fd, O_BINARY)
|
|
|
|
#define NEED_NEWLINE_DECORATOR_ON_READ_CHECK(fptr) do {\
|
|
if (NEED_NEWLINE_DECORATOR_ON_READ(fptr)) {\
|
|
if (((fptr)->mode & FMODE_READABLE) &&\
|
|
!((fptr)->encs.ecflags & ECONV_NEWLINE_DECORATOR_MASK)) {\
|
|
setmode((fptr)->fd, O_BINARY);\
|
|
}\
|
|
else {\
|
|
setmode((fptr)->fd, O_TEXT);\
|
|
}\
|
|
}\
|
|
} while(0)
|
|
|
|
#define SET_UNIVERSAL_NEWLINE_DECORATOR_IF_ENC2(enc2, ecflags) do {\
|
|
if ((enc2) && ((ecflags) & ECONV_DEFAULT_NEWLINE_DECORATOR)) {\
|
|
(ecflags) |= ECONV_UNIVERSAL_NEWLINE_DECORATOR;\
|
|
}\
|
|
} while(0)
|
|
|
|
/*
|
|
* IO unread with taking care of removed '\r' in text mode.
|
|
*/
|
|
static void
|
|
io_unread(rb_io_t *fptr)
|
|
{
|
|
off_t r, pos;
|
|
ssize_t read_size;
|
|
long i;
|
|
long newlines = 0;
|
|
long extra_max;
|
|
char *p;
|
|
char *buf;
|
|
|
|
rb_io_check_closed(fptr);
|
|
if (fptr->rbuf.len == 0 || fptr->mode & FMODE_DUPLEX) {
|
|
return;
|
|
}
|
|
|
|
errno = 0;
|
|
if (!rb_w32_fd_is_text(fptr->fd)) {
|
|
r = lseek(fptr->fd, -fptr->rbuf.len, SEEK_CUR);
|
|
if (r < 0 && errno) {
|
|
if (errno == ESPIPE)
|
|
fptr->mode |= FMODE_DUPLEX;
|
|
return;
|
|
}
|
|
|
|
fptr->rbuf.off = 0;
|
|
fptr->rbuf.len = 0;
|
|
return;
|
|
}
|
|
|
|
pos = lseek(fptr->fd, 0, SEEK_CUR);
|
|
if (pos < 0 && errno) {
|
|
if (errno == ESPIPE)
|
|
fptr->mode |= FMODE_DUPLEX;
|
|
return;
|
|
}
|
|
|
|
/* add extra offset for removed '\r' in rbuf */
|
|
extra_max = (long)(pos - fptr->rbuf.len);
|
|
p = fptr->rbuf.ptr + fptr->rbuf.off;
|
|
|
|
/* if the end of rbuf is '\r', rbuf doesn't have '\r' within rbuf.len */
|
|
if (*(fptr->rbuf.ptr + fptr->rbuf.capa - 1) == '\r') {
|
|
newlines++;
|
|
}
|
|
|
|
for (i = 0; i < fptr->rbuf.len; i++) {
|
|
if (*p == '\n') newlines++;
|
|
if (extra_max == newlines) break;
|
|
p++;
|
|
}
|
|
|
|
buf = ALLOC_N(char, fptr->rbuf.len + newlines);
|
|
while (newlines >= 0) {
|
|
r = lseek(fptr->fd, pos - fptr->rbuf.len - newlines, SEEK_SET);
|
|
if (newlines == 0) break;
|
|
if (r < 0) {
|
|
newlines--;
|
|
continue;
|
|
}
|
|
read_size = _read(fptr->fd, buf, fptr->rbuf.len + newlines);
|
|
if (read_size < 0) {
|
|
int e = errno;
|
|
free(buf);
|
|
rb_syserr_fail_path(e, fptr->pathv);
|
|
}
|
|
if (read_size == fptr->rbuf.len) {
|
|
lseek(fptr->fd, r, SEEK_SET);
|
|
break;
|
|
}
|
|
else {
|
|
newlines--;
|
|
}
|
|
}
|
|
free(buf);
|
|
fptr->rbuf.off = 0;
|
|
fptr->rbuf.len = 0;
|
|
return;
|
|
}
|
|
|
|
/*
|
|
* We use io_seek to back cursor position when changing mode from text to binary,
|
|
* but stdin and pipe cannot seek back. Stdin and pipe read should use encoding
|
|
* conversion for working properly with mode change.
|
|
*
|
|
* Return previous translation mode.
|
|
*/
|
|
static inline int
|
|
set_binary_mode_with_seek_cur(rb_io_t *fptr)
|
|
{
|
|
if (!rb_w32_fd_is_text(fptr->fd)) return O_BINARY;
|
|
|
|
if (fptr->rbuf.len == 0 || fptr->mode & FMODE_DUPLEX) {
|
|
return setmode(fptr->fd, O_BINARY);
|
|
}
|
|
flush_before_seek(fptr);
|
|
return setmode(fptr->fd, O_BINARY);
|
|
}
|
|
#define SET_BINARY_MODE_WITH_SEEK_CUR(fptr) set_binary_mode_with_seek_cur(fptr)
|
|
|
|
#else
|
|
/* Unix */
|
|
# define DEFAULT_TEXTMODE 0
|
|
#define NEED_READCONV(fptr) ((fptr)->encs.enc2 != NULL || NEED_NEWLINE_DECORATOR_ON_READ(fptr))
|
|
#define NEED_WRITECONV(fptr) ( \
|
|
((fptr)->encs.enc != NULL && (fptr)->encs.enc != rb_ascii8bit_encoding()) || \
|
|
NEED_NEWLINE_DECORATOR_ON_WRITE(fptr) || \
|
|
((fptr)->encs.ecflags & (ECONV_DECORATOR_MASK|ECONV_STATEFUL_DECORATOR_MASK)) || \
|
|
0)
|
|
#define SET_BINARY_MODE(fptr) (void)(fptr)
|
|
#define NEED_NEWLINE_DECORATOR_ON_READ_CHECK(fptr) (void)(fptr)
|
|
#define SET_UNIVERSAL_NEWLINE_DECORATOR_IF_ENC2(enc2, ecflags) ((void)(enc2), (void)(ecflags))
|
|
#define SET_BINARY_MODE_WITH_SEEK_CUR(fptr) (void)(fptr)
|
|
#endif
|
|
|
|
#if !defined HAVE_SHUTDOWN && !defined shutdown
|
|
#define shutdown(a,b) 0
|
|
#endif
|
|
|
|
#if defined(_WIN32)
|
|
#define is_socket(fd, path) rb_w32_is_socket(fd)
|
|
#elif !defined(S_ISSOCK)
|
|
#define is_socket(fd, path) 0
|
|
#else
|
|
static int
|
|
is_socket(int fd, VALUE path)
|
|
{
|
|
struct stat sbuf;
|
|
if (fstat(fd, &sbuf) < 0)
|
|
rb_sys_fail_path(path);
|
|
return S_ISSOCK(sbuf.st_mode);
|
|
}
|
|
#endif
|
|
|
|
static const char closed_stream[] = "closed stream";
|
|
|
|
static void
|
|
io_fd_check_closed(int fd)
|
|
{
|
|
if (fd < 0) {
|
|
rb_thread_check_ints(); /* check for ruby_error_stream_closed */
|
|
rb_raise(rb_eIOError, closed_stream);
|
|
}
|
|
}
|
|
|
|
void
|
|
rb_eof_error(void)
|
|
{
|
|
rb_raise(rb_eEOFError, "end of file reached");
|
|
}
|
|
|
|
VALUE
|
|
rb_io_taint_check(VALUE io)
|
|
{
|
|
rb_check_frozen(io);
|
|
return io;
|
|
}
|
|
|
|
void
|
|
rb_io_check_initialized(rb_io_t *fptr)
|
|
{
|
|
if (!fptr) {
|
|
rb_raise(rb_eIOError, "uninitialized stream");
|
|
}
|
|
}
|
|
|
|
void
|
|
rb_io_check_closed(rb_io_t *fptr)
|
|
{
|
|
rb_io_check_initialized(fptr);
|
|
io_fd_check_closed(fptr->fd);
|
|
}
|
|
|
|
static rb_io_t *
|
|
rb_io_get_fptr(VALUE io)
|
|
{
|
|
rb_io_t *fptr = RFILE(io)->fptr;
|
|
rb_io_check_initialized(fptr);
|
|
return fptr;
|
|
}
|
|
|
|
VALUE
|
|
rb_io_get_io(VALUE io)
|
|
{
|
|
return rb_convert_type_with_id(io, T_FILE, "IO", idTo_io);
|
|
}
|
|
|
|
VALUE
|
|
rb_io_check_io(VALUE io)
|
|
{
|
|
return rb_check_convert_type_with_id(io, T_FILE, "IO", idTo_io);
|
|
}
|
|
|
|
VALUE
|
|
rb_io_get_write_io(VALUE io)
|
|
{
|
|
VALUE write_io;
|
|
write_io = rb_io_get_fptr(io)->tied_io_for_writing;
|
|
if (write_io) {
|
|
return write_io;
|
|
}
|
|
return io;
|
|
}
|
|
|
|
VALUE
|
|
rb_io_set_write_io(VALUE io, VALUE w)
|
|
{
|
|
VALUE write_io;
|
|
rb_io_t *fptr = rb_io_get_fptr(io);
|
|
if (!RTEST(w)) {
|
|
w = 0;
|
|
}
|
|
else {
|
|
GetWriteIO(w);
|
|
}
|
|
write_io = fptr->tied_io_for_writing;
|
|
fptr->tied_io_for_writing = w;
|
|
return write_io ? write_io : Qnil;
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* IO.try_convert(obj) -> io or nil
|
|
*
|
|
* Try to convert <i>obj</i> into an IO, using to_io method.
|
|
* Returns converted IO or +nil+ if <i>obj</i> cannot be converted
|
|
* for any reason.
|
|
*
|
|
* IO.try_convert(STDOUT) #=> STDOUT
|
|
* IO.try_convert("STDOUT") #=> nil
|
|
*
|
|
* require 'zlib'
|
|
* f = open("/tmp/zz.gz") #=> #<File:/tmp/zz.gz>
|
|
* z = Zlib::GzipReader.open(f) #=> #<Zlib::GzipReader:0x81d8744>
|
|
* IO.try_convert(z) #=> #<File:/tmp/zz.gz>
|
|
*
|
|
*/
|
|
static VALUE
|
|
rb_io_s_try_convert(VALUE dummy, VALUE io)
|
|
{
|
|
return rb_io_check_io(io);
|
|
}
|
|
|
|
#if !(defined(RUBY_TEST_CRLF_ENVIRONMENT) || defined(_WIN32))
|
|
static void
|
|
io_unread(rb_io_t *fptr)
|
|
{
|
|
off_t r;
|
|
rb_io_check_closed(fptr);
|
|
if (fptr->rbuf.len == 0 || fptr->mode & FMODE_DUPLEX)
|
|
return;
|
|
/* xxx: target position may be negative if buffer is filled by ungetc */
|
|
errno = 0;
|
|
r = lseek(fptr->fd, -fptr->rbuf.len, SEEK_CUR);
|
|
if (r < 0 && errno) {
|
|
if (errno == ESPIPE)
|
|
fptr->mode |= FMODE_DUPLEX;
|
|
return;
|
|
}
|
|
fptr->rbuf.off = 0;
|
|
fptr->rbuf.len = 0;
|
|
return;
|
|
}
|
|
#endif
|
|
|
|
static rb_encoding *io_input_encoding(rb_io_t *fptr);
|
|
|
|
static void
|
|
io_ungetbyte(VALUE str, rb_io_t *fptr)
|
|
{
|
|
long len = RSTRING_LEN(str);
|
|
|
|
if (fptr->rbuf.ptr == NULL) {
|
|
const int min_capa = IO_RBUF_CAPA_FOR(fptr);
|
|
fptr->rbuf.off = 0;
|
|
fptr->rbuf.len = 0;
|
|
#if SIZEOF_LONG > SIZEOF_INT
|
|
if (len > INT_MAX)
|
|
rb_raise(rb_eIOError, "ungetbyte failed");
|
|
#endif
|
|
if (len > min_capa)
|
|
fptr->rbuf.capa = (int)len;
|
|
else
|
|
fptr->rbuf.capa = min_capa;
|
|
fptr->rbuf.ptr = ALLOC_N(char, fptr->rbuf.capa);
|
|
}
|
|
if (fptr->rbuf.capa < len + fptr->rbuf.len) {
|
|
rb_raise(rb_eIOError, "ungetbyte failed");
|
|
}
|
|
if (fptr->rbuf.off < len) {
|
|
MEMMOVE(fptr->rbuf.ptr+fptr->rbuf.capa-fptr->rbuf.len,
|
|
fptr->rbuf.ptr+fptr->rbuf.off,
|
|
char, fptr->rbuf.len);
|
|
fptr->rbuf.off = fptr->rbuf.capa-fptr->rbuf.len;
|
|
}
|
|
fptr->rbuf.off-=(int)len;
|
|
fptr->rbuf.len+=(int)len;
|
|
MEMMOVE(fptr->rbuf.ptr+fptr->rbuf.off, RSTRING_PTR(str), char, len);
|
|
}
|
|
|
|
static rb_io_t *
|
|
flush_before_seek(rb_io_t *fptr)
|
|
{
|
|
if (io_fflush(fptr) < 0)
|
|
rb_sys_fail(0);
|
|
io_unread(fptr);
|
|
errno = 0;
|
|
return fptr;
|
|
}
|
|
|
|
#define io_seek(fptr, ofs, whence) (errno = 0, lseek(flush_before_seek(fptr)->fd, (ofs), (whence)))
|
|
#define io_tell(fptr) lseek(flush_before_seek(fptr)->fd, 0, SEEK_CUR)
|
|
|
|
#ifndef SEEK_CUR
|
|
# define SEEK_SET 0
|
|
# define SEEK_CUR 1
|
|
# define SEEK_END 2
|
|
#endif
|
|
|
|
void
|
|
rb_io_check_char_readable(rb_io_t *fptr)
|
|
{
|
|
rb_io_check_closed(fptr);
|
|
if (!(fptr->mode & FMODE_READABLE)) {
|
|
rb_raise(rb_eIOError, "not opened for reading");
|
|
}
|
|
if (fptr->wbuf.len) {
|
|
if (io_fflush(fptr) < 0)
|
|
rb_sys_fail(0);
|
|
}
|
|
if (fptr->tied_io_for_writing) {
|
|
rb_io_t *wfptr;
|
|
GetOpenFile(fptr->tied_io_for_writing, wfptr);
|
|
if (io_fflush(wfptr) < 0)
|
|
rb_sys_fail(0);
|
|
}
|
|
}
|
|
|
|
void
|
|
rb_io_check_byte_readable(rb_io_t *fptr)
|
|
{
|
|
rb_io_check_char_readable(fptr);
|
|
if (READ_CHAR_PENDING(fptr)) {
|
|
rb_raise(rb_eIOError, "byte oriented read for character buffered IO");
|
|
}
|
|
}
|
|
|
|
void
|
|
rb_io_check_readable(rb_io_t *fptr)
|
|
{
|
|
rb_io_check_byte_readable(fptr);
|
|
}
|
|
|
|
static rb_encoding*
|
|
io_read_encoding(rb_io_t *fptr)
|
|
{
|
|
if (fptr->encs.enc) {
|
|
return fptr->encs.enc;
|
|
}
|
|
return rb_default_external_encoding();
|
|
}
|
|
|
|
static rb_encoding*
|
|
io_input_encoding(rb_io_t *fptr)
|
|
{
|
|
if (fptr->encs.enc2) {
|
|
return fptr->encs.enc2;
|
|
}
|
|
return io_read_encoding(fptr);
|
|
}
|
|
|
|
void
|
|
rb_io_check_writable(rb_io_t *fptr)
|
|
{
|
|
rb_io_check_closed(fptr);
|
|
if (!(fptr->mode & FMODE_WRITABLE)) {
|
|
rb_raise(rb_eIOError, "not opened for writing");
|
|
}
|
|
if (fptr->rbuf.len) {
|
|
io_unread(fptr);
|
|
}
|
|
}
|
|
|
|
int
|
|
rb_io_read_pending(rb_io_t *fptr)
|
|
{
|
|
/* This function is used for bytes and chars. Confusing. */
|
|
if (READ_CHAR_PENDING(fptr))
|
|
return 1; /* should raise? */
|
|
return READ_DATA_PENDING(fptr);
|
|
}
|
|
|
|
void
|
|
rb_io_read_check(rb_io_t *fptr)
|
|
{
|
|
if (!READ_DATA_PENDING(fptr)) {
|
|
rb_thread_wait_fd(fptr->fd);
|
|
}
|
|
return;
|
|
}
|
|
|
|
int
|
|
rb_gc_for_fd(int err)
|
|
{
|
|
if (err == EMFILE || err == ENFILE || err == ENOMEM) {
|
|
rb_gc();
|
|
return 1;
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
static int
|
|
ruby_dup(int orig)
|
|
{
|
|
int fd;
|
|
|
|
fd = rb_cloexec_dup(orig);
|
|
if (fd < 0) {
|
|
int e = errno;
|
|
if (rb_gc_for_fd(e)) {
|
|
fd = rb_cloexec_dup(orig);
|
|
}
|
|
if (fd < 0) {
|
|
rb_syserr_fail(e, 0);
|
|
}
|
|
}
|
|
rb_update_max_fd(fd);
|
|
return fd;
|
|
}
|
|
|
|
static VALUE
|
|
io_alloc(VALUE klass)
|
|
{
|
|
NEWOBJ_OF(io, struct RFile, klass, T_FILE);
|
|
|
|
io->fptr = 0;
|
|
|
|
return (VALUE)io;
|
|
}
|
|
|
|
#ifndef S_ISREG
|
|
# define S_ISREG(m) (((m) & S_IFMT) == S_IFREG)
|
|
#endif
|
|
|
|
struct io_internal_read_struct {
|
|
int fd;
|
|
int nonblock;
|
|
void *buf;
|
|
size_t capa;
|
|
};
|
|
|
|
struct io_internal_write_struct {
|
|
int fd;
|
|
const void *buf;
|
|
size_t capa;
|
|
};
|
|
|
|
#ifdef HAVE_WRITEV
|
|
struct io_internal_writev_struct {
|
|
int fd;
|
|
int iovcnt;
|
|
const struct iovec *iov;
|
|
};
|
|
#endif
|
|
|
|
static int nogvl_wait_for_single_fd(int fd, short events);
|
|
static VALUE
|
|
internal_read_func(void *ptr)
|
|
{
|
|
struct io_internal_read_struct *iis = ptr;
|
|
ssize_t r;
|
|
retry:
|
|
r = read(iis->fd, iis->buf, iis->capa);
|
|
if (r < 0 && !iis->nonblock) {
|
|
int e = errno;
|
|
if (e == EAGAIN || e == EWOULDBLOCK) {
|
|
if (nogvl_wait_for_single_fd(iis->fd, RB_WAITFD_IN) != -1) {
|
|
goto retry;
|
|
}
|
|
errno = e;
|
|
}
|
|
}
|
|
return r;
|
|
}
|
|
|
|
#if defined __APPLE__
|
|
# define do_write_retry(code) do {ret = code;} while (ret == -1 && errno == EPROTOTYPE)
|
|
#else
|
|
# define do_write_retry(code) ret = code
|
|
#endif
|
|
static VALUE
|
|
internal_write_func(void *ptr)
|
|
{
|
|
struct io_internal_write_struct *iis = ptr;
|
|
ssize_t ret;
|
|
do_write_retry(write(iis->fd, iis->buf, iis->capa));
|
|
return (VALUE)ret;
|
|
}
|
|
|
|
static void*
|
|
internal_write_func2(void *ptr)
|
|
{
|
|
return (void*)internal_write_func(ptr);
|
|
}
|
|
|
|
#ifdef HAVE_WRITEV
|
|
static VALUE
|
|
internal_writev_func(void *ptr)
|
|
{
|
|
struct io_internal_writev_struct *iis = ptr;
|
|
ssize_t ret;
|
|
do_write_retry(writev(iis->fd, iis->iov, iis->iovcnt));
|
|
return (VALUE)ret;
|
|
}
|
|
#endif
|
|
|
|
static ssize_t
|
|
rb_read_internal(int fd, void *buf, size_t count)
|
|
{
|
|
struct io_internal_read_struct iis;
|
|
|
|
iis.fd = fd;
|
|
iis.nonblock = 0;
|
|
iis.buf = buf;
|
|
iis.capa = count;
|
|
|
|
return (ssize_t)rb_thread_io_blocking_region(internal_read_func, &iis, fd);
|
|
}
|
|
|
|
static ssize_t
|
|
rb_write_internal(int fd, const void *buf, size_t count)
|
|
{
|
|
struct io_internal_write_struct iis;
|
|
iis.fd = fd;
|
|
iis.buf = buf;
|
|
iis.capa = count;
|
|
|
|
return (ssize_t)rb_thread_io_blocking_region(internal_write_func, &iis, fd);
|
|
}
|
|
|
|
static ssize_t
|
|
rb_write_internal2(int fd, const void *buf, size_t count)
|
|
{
|
|
struct io_internal_write_struct iis;
|
|
iis.fd = fd;
|
|
iis.buf = buf;
|
|
iis.capa = count;
|
|
|
|
return (ssize_t)rb_thread_call_without_gvl2(internal_write_func2, &iis,
|
|
RUBY_UBF_IO, NULL);
|
|
}
|
|
|
|
#ifdef HAVE_WRITEV
|
|
static ssize_t
|
|
rb_writev_internal(int fd, const struct iovec *iov, int iovcnt)
|
|
{
|
|
struct io_internal_writev_struct iis;
|
|
iis.fd = fd;
|
|
iis.iov = iov;
|
|
iis.iovcnt = iovcnt;
|
|
|
|
return (ssize_t)rb_thread_io_blocking_region(internal_writev_func, &iis, fd);
|
|
}
|
|
#endif
|
|
|
|
static VALUE
|
|
io_flush_buffer_sync(void *arg)
|
|
{
|
|
rb_io_t *fptr = arg;
|
|
long l = fptr->wbuf.len;
|
|
ssize_t r = write(fptr->fd, fptr->wbuf.ptr+fptr->wbuf.off, (size_t)l);
|
|
|
|
if (fptr->wbuf.len <= r) {
|
|
fptr->wbuf.off = 0;
|
|
fptr->wbuf.len = 0;
|
|
return 0;
|
|
}
|
|
if (0 <= r) {
|
|
fptr->wbuf.off += (int)r;
|
|
fptr->wbuf.len -= (int)r;
|
|
errno = EAGAIN;
|
|
}
|
|
return (VALUE)-1;
|
|
}
|
|
|
|
static void*
|
|
io_flush_buffer_sync2(void *arg)
|
|
{
|
|
VALUE result = io_flush_buffer_sync(arg);
|
|
|
|
/*
|
|
* rb_thread_call_without_gvl2 uses 0 as interrupted.
|
|
* So, we need to avoid to use 0.
|
|
*/
|
|
return !result ? (void*)1 : (void*)result;
|
|
}
|
|
|
|
static VALUE
|
|
io_flush_buffer_async(VALUE arg)
|
|
{
|
|
rb_io_t *fptr = (rb_io_t *)arg;
|
|
return rb_thread_io_blocking_region(io_flush_buffer_sync, fptr, fptr->fd);
|
|
}
|
|
|
|
static VALUE
|
|
io_flush_buffer_async2(VALUE arg)
|
|
{
|
|
rb_io_t *fptr = (rb_io_t *)arg;
|
|
VALUE ret;
|
|
|
|
ret = (VALUE)rb_thread_call_without_gvl2(io_flush_buffer_sync2, fptr,
|
|
RUBY_UBF_IO, NULL);
|
|
|
|
if (!ret) {
|
|
/* pending async interrupt is there. */
|
|
errno = EAGAIN;
|
|
return -1;
|
|
}
|
|
else if (ret == 1) {
|
|
return 0;
|
|
}
|
|
return ret;
|
|
}
|
|
|
|
static inline int
|
|
io_flush_buffer(rb_io_t *fptr)
|
|
{
|
|
if (fptr->write_lock) {
|
|
if (rb_mutex_owned_p(fptr->write_lock))
|
|
return (int)io_flush_buffer_async2((VALUE)fptr);
|
|
else
|
|
return (int)rb_mutex_synchronize(fptr->write_lock, io_flush_buffer_async2, (VALUE)fptr);
|
|
}
|
|
else {
|
|
return (int)io_flush_buffer_async((VALUE)fptr);
|
|
}
|
|
}
|
|
|
|
static int
|
|
io_fflush(rb_io_t *fptr)
|
|
{
|
|
rb_io_check_closed(fptr);
|
|
if (fptr->wbuf.len == 0)
|
|
return 0;
|
|
while (fptr->wbuf.len > 0 && io_flush_buffer(fptr) != 0) {
|
|
if (!rb_io_wait_writable(fptr->fd))
|
|
return -1;
|
|
rb_io_check_closed(fptr);
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
int
|
|
rb_io_wait_readable(int f)
|
|
{
|
|
io_fd_check_closed(f);
|
|
switch (errno) {
|
|
case EINTR:
|
|
#if defined(ERESTART)
|
|
case ERESTART:
|
|
#endif
|
|
rb_thread_check_ints();
|
|
return TRUE;
|
|
|
|
case EAGAIN:
|
|
#if defined(EWOULDBLOCK) && EWOULDBLOCK != EAGAIN
|
|
case EWOULDBLOCK:
|
|
#endif
|
|
rb_thread_wait_fd(f);
|
|
return TRUE;
|
|
|
|
default:
|
|
return FALSE;
|
|
}
|
|
}
|
|
|
|
int
|
|
rb_io_wait_writable(int f)
|
|
{
|
|
io_fd_check_closed(f);
|
|
switch (errno) {
|
|
case EINTR:
|
|
#if defined(ERESTART)
|
|
case ERESTART:
|
|
#endif
|
|
/*
|
|
* In old Linux, several special files under /proc and /sys don't handle
|
|
* select properly. Thus we need avoid to call if don't use O_NONBLOCK.
|
|
* Otherwise, we face nasty hang up. Sigh.
|
|
* e.g. http://git.kernel.org/?p=linux/kernel/git/torvalds/linux-2.6.git;a=commit;h=31b07093c44a7a442394d44423e21d783f5523b8
|
|
* http://git.kernel.org/?p=linux/kernel/git/torvalds/linux-2.6.git;a=commit;h=31b07093c44a7a442394d44423e21d783f5523b8
|
|
* In EINTR case, we only need to call RUBY_VM_CHECK_INTS_BLOCKING().
|
|
* Then rb_thread_check_ints() is enough.
|
|
*/
|
|
rb_thread_check_ints();
|
|
return TRUE;
|
|
|
|
case EAGAIN:
|
|
#if defined(EWOULDBLOCK) && EWOULDBLOCK != EAGAIN
|
|
case EWOULDBLOCK:
|
|
#endif
|
|
rb_thread_fd_writable(f);
|
|
return TRUE;
|
|
|
|
default:
|
|
return FALSE;
|
|
}
|
|
}
|
|
|
|
static void
|
|
make_writeconv(rb_io_t *fptr)
|
|
{
|
|
if (!fptr->writeconv_initialized) {
|
|
const char *senc, *denc;
|
|
rb_encoding *enc;
|
|
int ecflags;
|
|
VALUE ecopts;
|
|
|
|
fptr->writeconv_initialized = 1;
|
|
|
|
ecflags = fptr->encs.ecflags & ~ECONV_NEWLINE_DECORATOR_READ_MASK;
|
|
ecopts = fptr->encs.ecopts;
|
|
|
|
if (!fptr->encs.enc || (fptr->encs.enc == rb_ascii8bit_encoding() && !fptr->encs.enc2)) {
|
|
/* no encoding conversion */
|
|
fptr->writeconv_pre_ecflags = 0;
|
|
fptr->writeconv_pre_ecopts = Qnil;
|
|
fptr->writeconv = rb_econv_open_opts("", "", ecflags, ecopts);
|
|
if (!fptr->writeconv)
|
|
rb_exc_raise(rb_econv_open_exc("", "", ecflags));
|
|
fptr->writeconv_asciicompat = Qnil;
|
|
}
|
|
else {
|
|
enc = fptr->encs.enc2 ? fptr->encs.enc2 : fptr->encs.enc;
|
|
senc = rb_econv_asciicompat_encoding(rb_enc_name(enc));
|
|
if (!senc && !(fptr->encs.ecflags & ECONV_STATEFUL_DECORATOR_MASK)) {
|
|
/* single conversion */
|
|
fptr->writeconv_pre_ecflags = ecflags;
|
|
fptr->writeconv_pre_ecopts = ecopts;
|
|
fptr->writeconv = NULL;
|
|
fptr->writeconv_asciicompat = Qnil;
|
|
}
|
|
else {
|
|
/* double conversion */
|
|
fptr->writeconv_pre_ecflags = ecflags & ~ECONV_STATEFUL_DECORATOR_MASK;
|
|
fptr->writeconv_pre_ecopts = ecopts;
|
|
if (senc) {
|
|
denc = rb_enc_name(enc);
|
|
fptr->writeconv_asciicompat = rb_str_new2(senc);
|
|
}
|
|
else {
|
|
senc = denc = "";
|
|
fptr->writeconv_asciicompat = rb_str_new2(rb_enc_name(enc));
|
|
}
|
|
ecflags = fptr->encs.ecflags & (ECONV_ERROR_HANDLER_MASK|ECONV_STATEFUL_DECORATOR_MASK);
|
|
ecopts = fptr->encs.ecopts;
|
|
fptr->writeconv = rb_econv_open_opts(senc, denc, ecflags, ecopts);
|
|
if (!fptr->writeconv)
|
|
rb_exc_raise(rb_econv_open_exc(senc, denc, ecflags));
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
/* writing functions */
|
|
struct binwrite_arg {
|
|
rb_io_t *fptr;
|
|
VALUE str;
|
|
const char *ptr;
|
|
long length;
|
|
};
|
|
|
|
struct write_arg {
|
|
VALUE io;
|
|
VALUE str;
|
|
int nosync;
|
|
};
|
|
|
|
#ifdef HAVE_WRITEV
|
|
static VALUE
|
|
io_binwrite_string(VALUE arg)
|
|
{
|
|
struct binwrite_arg *p = (struct binwrite_arg *)arg;
|
|
rb_io_t *fptr = p->fptr;
|
|
long r;
|
|
|
|
if (fptr->wbuf.len) {
|
|
struct iovec iov[2];
|
|
|
|
iov[0].iov_base = fptr->wbuf.ptr+fptr->wbuf.off;
|
|
iov[0].iov_len = fptr->wbuf.len;
|
|
iov[1].iov_base = (char *)p->ptr;
|
|
iov[1].iov_len = p->length;
|
|
|
|
r = rb_writev_internal(fptr->fd, iov, 2);
|
|
|
|
if (r < 0)
|
|
return r;
|
|
|
|
if (fptr->wbuf.len <= r) {
|
|
r -= fptr->wbuf.len;
|
|
fptr->wbuf.off = 0;
|
|
fptr->wbuf.len = 0;
|
|
}
|
|
else {
|
|
fptr->wbuf.off += (int)r;
|
|
fptr->wbuf.len -= (int)r;
|
|
r = 0L;
|
|
}
|
|
}
|
|
else {
|
|
r = rb_write_internal(fptr->fd, p->ptr, p->length);
|
|
}
|
|
|
|
return r;
|
|
}
|
|
#else
|
|
static VALUE
|
|
io_binwrite_string(VALUE arg)
|
|
{
|
|
struct binwrite_arg *p = (struct binwrite_arg *)arg;
|
|
rb_io_t *fptr = p->fptr;
|
|
long l, len;
|
|
|
|
l = len = p->length;
|
|
|
|
if (fptr->wbuf.len) {
|
|
if (fptr->wbuf.len+len <= fptr->wbuf.capa) {
|
|
if (fptr->wbuf.capa < fptr->wbuf.off+fptr->wbuf.len+len) {
|
|
MEMMOVE(fptr->wbuf.ptr, fptr->wbuf.ptr+fptr->wbuf.off, char, fptr->wbuf.len);
|
|
fptr->wbuf.off = 0;
|
|
}
|
|
MEMMOVE(fptr->wbuf.ptr+fptr->wbuf.off+fptr->wbuf.len, p->ptr, char, len);
|
|
fptr->wbuf.len += (int)len;
|
|
l = 0;
|
|
}
|
|
if (io_fflush(fptr) < 0)
|
|
return -2L; /* fail in fflush */
|
|
if (l == 0)
|
|
return len;
|
|
}
|
|
|
|
if (fptr->stdio_file != stderr && !rb_thread_fd_writable(fptr->fd))
|
|
rb_io_check_closed(fptr);
|
|
|
|
return rb_write_internal(p->fptr->fd, p->ptr, p->length);
|
|
}
|
|
#endif
|
|
|
|
static long
|
|
io_binwrite(VALUE str, const char *ptr, long len, rb_io_t *fptr, int nosync)
|
|
{
|
|
long n, r, offset = 0;
|
|
|
|
/* don't write anything if current thread has a pending interrupt. */
|
|
rb_thread_check_ints();
|
|
|
|
if ((n = len) <= 0) return n;
|
|
if (fptr->wbuf.ptr == NULL && !(!nosync && (fptr->mode & FMODE_SYNC))) {
|
|
fptr->wbuf.off = 0;
|
|
fptr->wbuf.len = 0;
|
|
fptr->wbuf.capa = IO_WBUF_CAPA_MIN;
|
|
fptr->wbuf.ptr = ALLOC_N(char, fptr->wbuf.capa);
|
|
fptr->write_lock = rb_mutex_new();
|
|
rb_mutex_allow_trap(fptr->write_lock, 1);
|
|
}
|
|
if ((!nosync && (fptr->mode & (FMODE_SYNC|FMODE_TTY))) ||
|
|
(fptr->wbuf.ptr && fptr->wbuf.capa <= fptr->wbuf.len + len)) {
|
|
struct binwrite_arg arg;
|
|
|
|
arg.fptr = fptr;
|
|
arg.str = str;
|
|
retry:
|
|
arg.ptr = ptr + offset;
|
|
arg.length = n;
|
|
if (fptr->write_lock) {
|
|
r = rb_mutex_synchronize(fptr->write_lock, io_binwrite_string, (VALUE)&arg);
|
|
}
|
|
else {
|
|
r = io_binwrite_string((VALUE)&arg);
|
|
}
|
|
/* xxx: other threads may modify given string. */
|
|
if (r == n) return len;
|
|
if (0 <= r) {
|
|
offset += r;
|
|
n -= r;
|
|
errno = EAGAIN;
|
|
}
|
|
if (r == -2L)
|
|
return -1L;
|
|
if (rb_io_wait_writable(fptr->fd)) {
|
|
rb_io_check_closed(fptr);
|
|
if (offset < len)
|
|
goto retry;
|
|
}
|
|
return -1L;
|
|
}
|
|
|
|
if (fptr->wbuf.off) {
|
|
if (fptr->wbuf.len)
|
|
MEMMOVE(fptr->wbuf.ptr, fptr->wbuf.ptr+fptr->wbuf.off, char, fptr->wbuf.len);
|
|
fptr->wbuf.off = 0;
|
|
}
|
|
MEMMOVE(fptr->wbuf.ptr+fptr->wbuf.off+fptr->wbuf.len, ptr+offset, char, len);
|
|
fptr->wbuf.len += (int)len;
|
|
return len;
|
|
}
|
|
|
|
# define MODE_BTMODE(a,b,c) ((fmode & FMODE_BINMODE) ? (b) : \
|
|
(fmode & FMODE_TEXTMODE) ? (c) : (a))
|
|
|
|
#define MODE_BTXMODE(a, b, c, d, e, f) ((fmode & FMODE_EXCL) ? \
|
|
MODE_BTMODE(d, e, f) : \
|
|
MODE_BTMODE(a, b, c))
|
|
|
|
static VALUE
|
|
do_writeconv(VALUE str, rb_io_t *fptr, int *converted)
|
|
{
|
|
if (NEED_WRITECONV(fptr)) {
|
|
VALUE common_encoding = Qnil;
|
|
SET_BINARY_MODE(fptr);
|
|
|
|
make_writeconv(fptr);
|
|
|
|
if (fptr->writeconv) {
|
|
#define fmode (fptr->mode)
|
|
if (!NIL_P(fptr->writeconv_asciicompat))
|
|
common_encoding = fptr->writeconv_asciicompat;
|
|
else if (MODE_BTMODE(DEFAULT_TEXTMODE,0,1) && !rb_enc_asciicompat(rb_enc_get(str))) {
|
|
rb_raise(rb_eArgError, "ASCII incompatible string written for text mode IO without encoding conversion: %s",
|
|
rb_enc_name(rb_enc_get(str)));
|
|
}
|
|
#undef fmode
|
|
}
|
|
else {
|
|
if (fptr->encs.enc2)
|
|
common_encoding = rb_enc_from_encoding(fptr->encs.enc2);
|
|
else if (fptr->encs.enc != rb_ascii8bit_encoding())
|
|
common_encoding = rb_enc_from_encoding(fptr->encs.enc);
|
|
}
|
|
|
|
if (!NIL_P(common_encoding)) {
|
|
str = rb_str_encode(str, common_encoding,
|
|
fptr->writeconv_pre_ecflags, fptr->writeconv_pre_ecopts);
|
|
*converted = 1;
|
|
}
|
|
|
|
if (fptr->writeconv) {
|
|
str = rb_econv_str_convert(fptr->writeconv, str, ECONV_PARTIAL_INPUT);
|
|
*converted = 1;
|
|
}
|
|
}
|
|
#if defined(RUBY_TEST_CRLF_ENVIRONMENT) || defined(_WIN32)
|
|
#define fmode (fptr->mode)
|
|
else if (MODE_BTMODE(DEFAULT_TEXTMODE,0,1)) {
|
|
if ((fptr->mode & FMODE_READABLE) &&
|
|
!(fptr->encs.ecflags & ECONV_NEWLINE_DECORATOR_MASK)) {
|
|
setmode(fptr->fd, O_BINARY);
|
|
}
|
|
else {
|
|
setmode(fptr->fd, O_TEXT);
|
|
}
|
|
if (!rb_enc_asciicompat(rb_enc_get(str))) {
|
|
rb_raise(rb_eArgError, "ASCII incompatible string written for text mode IO without encoding conversion: %s",
|
|
rb_enc_name(rb_enc_get(str)));
|
|
}
|
|
}
|
|
#undef fmode
|
|
#endif
|
|
return str;
|
|
}
|
|
|
|
static long
|
|
io_fwrite(VALUE str, rb_io_t *fptr, int nosync)
|
|
{
|
|
int converted = 0;
|
|
VALUE tmp;
|
|
long n, len;
|
|
const char *ptr;
|
|
#ifdef _WIN32
|
|
if (fptr->mode & FMODE_TTY) {
|
|
long len = rb_w32_write_console(str, fptr->fd);
|
|
if (len > 0) return len;
|
|
}
|
|
#endif
|
|
str = do_writeconv(str, fptr, &converted);
|
|
if (converted)
|
|
OBJ_FREEZE(str);
|
|
|
|
tmp = rb_str_tmp_frozen_acquire(str);
|
|
RSTRING_GETMEM(tmp, ptr, len);
|
|
n = io_binwrite(tmp, ptr, len, fptr, nosync);
|
|
rb_str_tmp_frozen_release(str, tmp);
|
|
|
|
return n;
|
|
}
|
|
|
|
ssize_t
|
|
rb_io_bufwrite(VALUE io, const void *buf, size_t size)
|
|
{
|
|
rb_io_t *fptr;
|
|
|
|
GetOpenFile(io, fptr);
|
|
rb_io_check_writable(fptr);
|
|
return (ssize_t)io_binwrite(0, buf, (long)size, fptr, 0);
|
|
}
|
|
|
|
static VALUE
|
|
io_write(VALUE io, VALUE str, int nosync)
|
|
{
|
|
rb_io_t *fptr;
|
|
long n;
|
|
VALUE tmp;
|
|
|
|
io = GetWriteIO(io);
|
|
str = rb_obj_as_string(str);
|
|
tmp = rb_io_check_io(io);
|
|
if (NIL_P(tmp)) {
|
|
/* port is not IO, call write method for it. */
|
|
return rb_funcall(io, id_write, 1, str);
|
|
}
|
|
io = tmp;
|
|
if (RSTRING_LEN(str) == 0) return INT2FIX(0);
|
|
|
|
GetOpenFile(io, fptr);
|
|
rb_io_check_writable(fptr);
|
|
|
|
n = io_fwrite(str, fptr, nosync);
|
|
if (n < 0L) rb_sys_fail_path(fptr->pathv);
|
|
|
|
return LONG2FIX(n);
|
|
}
|
|
|
|
#ifdef HAVE_WRITEV
|
|
struct binwritev_arg {
|
|
rb_io_t *fptr;
|
|
const struct iovec *iov;
|
|
int iovcnt;
|
|
};
|
|
|
|
static VALUE
|
|
call_writev_internal(VALUE arg)
|
|
{
|
|
struct binwritev_arg *p = (struct binwritev_arg *)arg;
|
|
return rb_writev_internal(p->fptr->fd, p->iov, p->iovcnt);
|
|
}
|
|
|
|
static long
|
|
io_binwritev(struct iovec *iov, int iovcnt, rb_io_t *fptr)
|
|
{
|
|
int i;
|
|
long r, total = 0, written_len = 0;
|
|
|
|
/* don't write anything if current thread has a pending interrupt. */
|
|
rb_thread_check_ints();
|
|
|
|
if (iovcnt == 0) return 0;
|
|
for (i = 1; i < iovcnt; i++) total += iov[i].iov_len;
|
|
|
|
if (fptr->wbuf.ptr == NULL && !(fptr->mode & FMODE_SYNC)) {
|
|
fptr->wbuf.off = 0;
|
|
fptr->wbuf.len = 0;
|
|
fptr->wbuf.capa = IO_WBUF_CAPA_MIN;
|
|
fptr->wbuf.ptr = ALLOC_N(char, fptr->wbuf.capa);
|
|
fptr->write_lock = rb_mutex_new();
|
|
rb_mutex_allow_trap(fptr->write_lock, 1);
|
|
}
|
|
|
|
if (fptr->wbuf.ptr && fptr->wbuf.len) {
|
|
long offset = fptr->wbuf.off + fptr->wbuf.len;
|
|
if (offset + total <= fptr->wbuf.capa) {
|
|
for (i = 1; i < iovcnt; i++) {
|
|
memcpy(fptr->wbuf.ptr+offset, iov[i].iov_base, iov[i].iov_len);
|
|
offset += iov[i].iov_len;
|
|
}
|
|
fptr->wbuf.len += total;
|
|
return total;
|
|
}
|
|
else {
|
|
iov[0].iov_base = fptr->wbuf.ptr + fptr->wbuf.off;
|
|
iov[0].iov_len = fptr->wbuf.len;
|
|
}
|
|
}
|
|
else {
|
|
iov++;
|
|
if (!--iovcnt) return 0;
|
|
}
|
|
|
|
retry:
|
|
if (fptr->write_lock) {
|
|
struct binwritev_arg arg;
|
|
arg.fptr = fptr;
|
|
arg.iov = iov;
|
|
arg.iovcnt = iovcnt;
|
|
r = rb_mutex_synchronize(fptr->write_lock, call_writev_internal, (VALUE)&arg);
|
|
}
|
|
else {
|
|
r = rb_writev_internal(fptr->fd, iov, iovcnt);
|
|
}
|
|
|
|
if (r >= 0) {
|
|
written_len += r;
|
|
if (fptr->wbuf.ptr && fptr->wbuf.len) {
|
|
if (written_len < fptr->wbuf.len) {
|
|
fptr->wbuf.off += r;
|
|
fptr->wbuf.len -= r;
|
|
}
|
|
else {
|
|
written_len -= fptr->wbuf.len;
|
|
fptr->wbuf.off = 0;
|
|
fptr->wbuf.len = 0;
|
|
}
|
|
}
|
|
if (written_len == total) return total;
|
|
|
|
while (r >= (ssize_t)iov->iov_len) {
|
|
/* iovcnt > 0 */
|
|
r -= iov->iov_len;
|
|
iov->iov_len = 0;
|
|
iov++;
|
|
if (!--iovcnt) return total;
|
|
/* defensive check: written_len should == total */
|
|
}
|
|
iov->iov_base = (char *)iov->iov_base + r;
|
|
iov->iov_len -= r;
|
|
|
|
errno = EAGAIN;
|
|
}
|
|
if (rb_io_wait_writable(fptr->fd)) {
|
|
rb_io_check_closed(fptr);
|
|
goto retry;
|
|
}
|
|
|
|
return -1L;
|
|
}
|
|
|
|
static long
|
|
io_fwritev(int argc, VALUE *argv, rb_io_t *fptr)
|
|
{
|
|
int i, converted, iovcnt = argc + 1;
|
|
long n;
|
|
VALUE v1, v2, str, tmp, *tmp_array;
|
|
struct iovec *iov;
|
|
|
|
iov = ALLOCV_N(struct iovec, v1, iovcnt);
|
|
tmp_array = ALLOCV_N(VALUE, v2, argc);
|
|
|
|
for (i = 0; i < argc; i++) {
|
|
str = rb_obj_as_string(argv[i]);
|
|
converted = 0;
|
|
str = do_writeconv(str, fptr, &converted);
|
|
if (converted)
|
|
OBJ_FREEZE(str);
|
|
|
|
tmp = rb_str_tmp_frozen_acquire(str);
|
|
tmp_array[i] = tmp;
|
|
/* iov[0] is reserved for buffer of fptr */
|
|
iov[i+1].iov_base = RSTRING_PTR(tmp);
|
|
iov[i+1].iov_len = RSTRING_LEN(tmp);
|
|
}
|
|
|
|
n = io_binwritev(iov, iovcnt, fptr);
|
|
if (v1) ALLOCV_END(v1);
|
|
|
|
for (i = 0; i < argc; i++) {
|
|
rb_str_tmp_frozen_release(argv[i], tmp_array[i]);
|
|
}
|
|
|
|
if (v2) ALLOCV_END(v2);
|
|
|
|
return n;
|
|
}
|
|
|
|
static int
|
|
iovcnt_ok(int iovcnt)
|
|
{
|
|
#ifdef IOV_MAX
|
|
return iovcnt < IOV_MAX;
|
|
#else /* GNU/Hurd has writev, but no IOV_MAX */
|
|
return 1;
|
|
#endif
|
|
}
|
|
#endif /* HAVE_WRITEV */
|
|
|
|
static VALUE
|
|
io_writev(int argc, VALUE *argv, VALUE io)
|
|
{
|
|
rb_io_t *fptr;
|
|
long n;
|
|
VALUE tmp, total = INT2FIX(0);
|
|
int i, cnt = 1;
|
|
|
|
io = GetWriteIO(io);
|
|
tmp = rb_io_check_io(io);
|
|
if (NIL_P(tmp)) {
|
|
/* port is not IO, call write method for it. */
|
|
return rb_funcallv(io, id_write, argc, argv);
|
|
}
|
|
io = tmp;
|
|
|
|
GetOpenFile(io, fptr);
|
|
rb_io_check_writable(fptr);
|
|
|
|
for (i = 0; i < argc; i += cnt) {
|
|
#ifdef HAVE_WRITEV
|
|
if ((fptr->mode & (FMODE_SYNC|FMODE_TTY)) && iovcnt_ok(cnt = argc - i)) {
|
|
n = io_fwritev(cnt, &argv[i], fptr);
|
|
}
|
|
else
|
|
#endif
|
|
{
|
|
cnt = 1;
|
|
/* sync at last item */
|
|
n = io_fwrite(rb_obj_as_string(argv[i]), fptr, (i < argc-1));
|
|
}
|
|
if (n < 0L) rb_sys_fail_path(fptr->pathv);
|
|
total = rb_fix_plus(LONG2FIX(n), total);
|
|
}
|
|
|
|
return total;
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* ios.write(string, ...) -> integer
|
|
*
|
|
* Writes the given strings to <em>ios</em>. The stream must be opened
|
|
* for writing. Arguments that are not a string will be converted
|
|
* to a string using <code>to_s</code>. Returns the number of bytes
|
|
* written in total.
|
|
*
|
|
* count = $stdout.write("This is", " a test\n")
|
|
* puts "That was #{count} bytes of data"
|
|
*
|
|
* <em>produces:</em>
|
|
*
|
|
* This is a test
|
|
* That was 15 bytes of data
|
|
*/
|
|
|
|
static VALUE
|
|
io_write_m(int argc, VALUE *argv, VALUE io)
|
|
{
|
|
if (argc != 1) {
|
|
return io_writev(argc, argv, io);
|
|
}
|
|
else {
|
|
VALUE str = argv[0];
|
|
return io_write(io, str, 0);
|
|
}
|
|
}
|
|
|
|
VALUE
|
|
rb_io_write(VALUE io, VALUE str)
|
|
{
|
|
return rb_funcallv(io, id_write, 1, &str);
|
|
}
|
|
|
|
static VALUE
|
|
rb_io_writev(VALUE io, int argc, VALUE *argv)
|
|
{
|
|
if (argc > 1 && rb_obj_method_arity(io, id_write) == 1) {
|
|
if (io != rb_stderr && RTEST(ruby_verbose)) {
|
|
VALUE klass = CLASS_OF(io);
|
|
char sep = FL_TEST(klass, FL_SINGLETON) ? (klass = io, '.') : '#';
|
|
rb_warning("%+"PRIsVALUE"%c""write is outdated interface"
|
|
" which accepts just one argument",
|
|
klass, sep);
|
|
}
|
|
do rb_io_write(io, *argv++); while (--argc);
|
|
return argv[0]; /* unused right now */
|
|
}
|
|
return rb_funcallv(io, id_write, argc, argv);
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* ios << obj -> ios
|
|
*
|
|
* String Output---Writes <i>obj</i> to <em>ios</em>.
|
|
* <i>obj</i> will be converted to a string using
|
|
* <code>to_s</code>.
|
|
*
|
|
* $stdout << "Hello " << "world!\n"
|
|
*
|
|
* <em>produces:</em>
|
|
*
|
|
* Hello world!
|
|
*/
|
|
|
|
|
|
VALUE
|
|
rb_io_addstr(VALUE io, VALUE str)
|
|
{
|
|
rb_io_write(io, str);
|
|
return io;
|
|
}
|
|
|
|
#ifdef HAVE_FSYNC
|
|
static VALUE
|
|
nogvl_fsync(void *ptr)
|
|
{
|
|
rb_io_t *fptr = ptr;
|
|
|
|
#ifdef _WIN32
|
|
if (GetFileType((HANDLE)rb_w32_get_osfhandle(fptr->fd)) != FILE_TYPE_DISK)
|
|
return 0;
|
|
#endif
|
|
return (VALUE)fsync(fptr->fd);
|
|
}
|
|
#endif
|
|
|
|
VALUE
|
|
rb_io_flush_raw(VALUE io, int sync)
|
|
{
|
|
rb_io_t *fptr;
|
|
|
|
if (!RB_TYPE_P(io, T_FILE)) {
|
|
return rb_funcall(io, id_flush, 0);
|
|
}
|
|
|
|
io = GetWriteIO(io);
|
|
GetOpenFile(io, fptr);
|
|
|
|
if (fptr->mode & FMODE_WRITABLE) {
|
|
if (io_fflush(fptr) < 0)
|
|
rb_sys_fail(0);
|
|
}
|
|
if (fptr->mode & FMODE_READABLE) {
|
|
io_unread(fptr);
|
|
}
|
|
|
|
return io;
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* ios.flush -> ios
|
|
*
|
|
* Flushes any buffered data within <em>ios</em> to the underlying
|
|
* operating system (note that this is Ruby internal buffering only;
|
|
* the OS may buffer the data as well).
|
|
*
|
|
* $stdout.print "no newline"
|
|
* $stdout.flush
|
|
*
|
|
* <em>produces:</em>
|
|
*
|
|
* no newline
|
|
*/
|
|
|
|
VALUE
|
|
rb_io_flush(VALUE io)
|
|
{
|
|
return rb_io_flush_raw(io, 1);
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* ios.pos -> integer
|
|
* ios.tell -> integer
|
|
*
|
|
* Returns the current offset (in bytes) of <em>ios</em>.
|
|
*
|
|
* f = File.new("testfile")
|
|
* f.pos #=> 0
|
|
* f.gets #=> "This is line one\n"
|
|
* f.pos #=> 17
|
|
*/
|
|
|
|
static VALUE
|
|
rb_io_tell(VALUE io)
|
|
{
|
|
rb_io_t *fptr;
|
|
off_t pos;
|
|
|
|
GetOpenFile(io, fptr);
|
|
pos = io_tell(fptr);
|
|
if (pos < 0 && errno) rb_sys_fail_path(fptr->pathv);
|
|
pos -= fptr->rbuf.len;
|
|
return OFFT2NUM(pos);
|
|
}
|
|
|
|
static VALUE
|
|
rb_io_seek(VALUE io, VALUE offset, int whence)
|
|
{
|
|
rb_io_t *fptr;
|
|
off_t pos;
|
|
|
|
pos = NUM2OFFT(offset);
|
|
GetOpenFile(io, fptr);
|
|
pos = io_seek(fptr, pos, whence);
|
|
if (pos < 0 && errno) rb_sys_fail_path(fptr->pathv);
|
|
|
|
return INT2FIX(0);
|
|
}
|
|
|
|
static int
|
|
interpret_seek_whence(VALUE vwhence)
|
|
{
|
|
if (vwhence == sym_SET)
|
|
return SEEK_SET;
|
|
if (vwhence == sym_CUR)
|
|
return SEEK_CUR;
|
|
if (vwhence == sym_END)
|
|
return SEEK_END;
|
|
#ifdef SEEK_DATA
|
|
if (vwhence == sym_DATA)
|
|
return SEEK_DATA;
|
|
#endif
|
|
#ifdef SEEK_HOLE
|
|
if (vwhence == sym_HOLE)
|
|
return SEEK_HOLE;
|
|
#endif
|
|
return NUM2INT(vwhence);
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* ios.seek(amount, whence=IO::SEEK_SET) -> 0
|
|
*
|
|
* Seeks to a given offset <i>anInteger</i> in the stream according to
|
|
* the value of <i>whence</i>:
|
|
*
|
|
* :CUR or IO::SEEK_CUR | Seeks to _amount_ plus current position
|
|
* ----------------------+--------------------------------------------------
|
|
* :END or IO::SEEK_END | Seeks to _amount_ plus end of stream (you
|
|
* | probably want a negative value for _amount_)
|
|
* ----------------------+--------------------------------------------------
|
|
* :SET or IO::SEEK_SET | Seeks to the absolute location given by _amount_
|
|
*
|
|
* Example:
|
|
*
|
|
* f = File.new("testfile")
|
|
* f.seek(-13, IO::SEEK_END) #=> 0
|
|
* f.readline #=> "And so on...\n"
|
|
*/
|
|
|
|
static VALUE
|
|
rb_io_seek_m(int argc, VALUE *argv, VALUE io)
|
|
{
|
|
VALUE offset, ptrname;
|
|
int whence = SEEK_SET;
|
|
|
|
if (rb_scan_args(argc, argv, "11", &offset, &ptrname) == 2) {
|
|
whence = interpret_seek_whence(ptrname);
|
|
}
|
|
|
|
return rb_io_seek(io, offset, whence);
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* ios.pos = integer -> integer
|
|
*
|
|
* Seeks to the given position (in bytes) in <em>ios</em>.
|
|
* It is not guaranteed that seeking to the right position when <em>ios</em>
|
|
* is textmode.
|
|
*
|
|
* f = File.new("testfile")
|
|
* f.pos = 17
|
|
* f.gets #=> "This is line two\n"
|
|
*/
|
|
|
|
static VALUE
|
|
rb_io_set_pos(VALUE io, VALUE offset)
|
|
{
|
|
rb_io_t *fptr;
|
|
off_t pos;
|
|
|
|
pos = NUM2OFFT(offset);
|
|
GetOpenFile(io, fptr);
|
|
pos = io_seek(fptr, pos, SEEK_SET);
|
|
if (pos < 0 && errno) rb_sys_fail_path(fptr->pathv);
|
|
|
|
return OFFT2NUM(pos);
|
|
}
|
|
|
|
static void clear_readconv(rb_io_t *fptr);
|
|
|
|
/*
|
|
* call-seq:
|
|
* ios.rewind -> 0
|
|
*
|
|
* Positions <em>ios</em> to the beginning of input, resetting
|
|
* #lineno to zero.
|
|
*
|
|
* f = File.new("testfile")
|
|
* f.readline #=> "This is line one\n"
|
|
* f.rewind #=> 0
|
|
* f.lineno #=> 0
|
|
* f.readline #=> "This is line one\n"
|
|
*
|
|
* Note that it cannot be used with streams such as pipes, ttys, and sockets.
|
|
*/
|
|
|
|
static VALUE
|
|
rb_io_rewind(VALUE io)
|
|
{
|
|
rb_io_t *fptr;
|
|
|
|
GetOpenFile(io, fptr);
|
|
if (io_seek(fptr, 0L, 0) < 0 && errno) rb_sys_fail_path(fptr->pathv);
|
|
if (io == ARGF.current_file) {
|
|
ARGF.lineno -= fptr->lineno;
|
|
}
|
|
fptr->lineno = 0;
|
|
if (fptr->readconv) {
|
|
clear_readconv(fptr);
|
|
}
|
|
|
|
return INT2FIX(0);
|
|
}
|
|
|
|
static int
|
|
fptr_wait_readable(rb_io_t *fptr)
|
|
{
|
|
int ret = rb_io_wait_readable(fptr->fd);
|
|
|
|
if (ret)
|
|
rb_io_check_closed(fptr);
|
|
return ret;
|
|
}
|
|
|
|
static int
|
|
io_fillbuf(rb_io_t *fptr)
|
|
{
|
|
ssize_t r;
|
|
|
|
if (fptr->rbuf.ptr == NULL) {
|
|
fptr->rbuf.off = 0;
|
|
fptr->rbuf.len = 0;
|
|
fptr->rbuf.capa = IO_RBUF_CAPA_FOR(fptr);
|
|
fptr->rbuf.ptr = ALLOC_N(char, fptr->rbuf.capa);
|
|
#ifdef _WIN32
|
|
fptr->rbuf.capa--;
|
|
#endif
|
|
}
|
|
if (fptr->rbuf.len == 0) {
|
|
retry:
|
|
{
|
|
r = rb_read_internal(fptr->fd, fptr->rbuf.ptr, fptr->rbuf.capa);
|
|
}
|
|
if (r < 0) {
|
|
if (fptr_wait_readable(fptr))
|
|
goto retry;
|
|
{
|
|
int e = errno;
|
|
VALUE path = rb_sprintf("fd:%d ", fptr->fd);
|
|
if (!NIL_P(fptr->pathv)) {
|
|
rb_str_append(path, fptr->pathv);
|
|
}
|
|
rb_syserr_fail_path(e, path);
|
|
}
|
|
}
|
|
if (r > 0) rb_io_check_closed(fptr);
|
|
fptr->rbuf.off = 0;
|
|
fptr->rbuf.len = (int)r; /* r should be <= rbuf_capa */
|
|
if (r == 0)
|
|
return -1; /* EOF */
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* ios.eof -> true or false
|
|
* ios.eof? -> true or false
|
|
*
|
|
* Returns true if <em>ios</em> is at end of file that means
|
|
* there are no more data to read.
|
|
* The stream must be opened for reading or an IOError will be
|
|
* raised.
|
|
*
|
|
* f = File.new("testfile")
|
|
* dummy = f.readlines
|
|
* f.eof #=> true
|
|
*
|
|
* If <em>ios</em> is a stream such as pipe or socket, IO#eof?
|
|
* blocks until the other end sends some data or closes it.
|
|
*
|
|
* r, w = IO.pipe
|
|
* Thread.new { sleep 1; w.close }
|
|
* r.eof? #=> true after 1 second blocking
|
|
*
|
|
* r, w = IO.pipe
|
|
* Thread.new { sleep 1; w.puts "a" }
|
|
* r.eof? #=> false after 1 second blocking
|
|
*
|
|
* r, w = IO.pipe
|
|
* r.eof? # blocks forever
|
|
*
|
|
* Note that IO#eof? reads data to the input byte buffer. So
|
|
* IO#sysread may not behave as you intend with IO#eof?, unless you
|
|
* call IO#rewind first (which is not available for some streams).
|
|
*/
|
|
|
|
VALUE
|
|
rb_io_eof(VALUE io)
|
|
{
|
|
rb_io_t *fptr;
|
|
|
|
GetOpenFile(io, fptr);
|
|
rb_io_check_char_readable(fptr);
|
|
|
|
if (READ_CHAR_PENDING(fptr)) return Qfalse;
|
|
if (READ_DATA_PENDING(fptr)) return Qfalse;
|
|
READ_CHECK(fptr);
|
|
#if defined(RUBY_TEST_CRLF_ENVIRONMENT) || defined(_WIN32)
|
|
if (!NEED_READCONV(fptr) && NEED_NEWLINE_DECORATOR_ON_READ(fptr)) {
|
|
return eof(fptr->fd) ? Qtrue : Qfalse;
|
|
}
|
|
#endif
|
|
if (io_fillbuf(fptr) < 0) {
|
|
return Qtrue;
|
|
}
|
|
return Qfalse;
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* ios.sync -> true or false
|
|
*
|
|
* Returns the current ``sync mode'' of <em>ios</em>. When sync mode is
|
|
* true, all output is immediately flushed to the underlying operating
|
|
* system and is not buffered by Ruby internally. See also
|
|
* IO#fsync.
|
|
*
|
|
* f = File.new("testfile")
|
|
* f.sync #=> false
|
|
*/
|
|
|
|
static VALUE
|
|
rb_io_sync(VALUE io)
|
|
{
|
|
rb_io_t *fptr;
|
|
|
|
io = GetWriteIO(io);
|
|
GetOpenFile(io, fptr);
|
|
return (fptr->mode & FMODE_SYNC) ? Qtrue : Qfalse;
|
|
}
|
|
|
|
#ifdef HAVE_FSYNC
|
|
|
|
/*
|
|
* call-seq:
|
|
* ios.sync = boolean -> boolean
|
|
*
|
|
* Sets the ``sync mode'' to <code>true</code> or <code>false</code>.
|
|
* When sync mode is true, all output is immediately flushed to the
|
|
* underlying operating system and is not buffered internally. Returns
|
|
* the new state. See also IO#fsync.
|
|
*
|
|
* f = File.new("testfile")
|
|
* f.sync = true
|
|
*/
|
|
|
|
static VALUE
|
|
rb_io_set_sync(VALUE io, VALUE sync)
|
|
{
|
|
rb_io_t *fptr;
|
|
|
|
io = GetWriteIO(io);
|
|
GetOpenFile(io, fptr);
|
|
if (RTEST(sync)) {
|
|
fptr->mode |= FMODE_SYNC;
|
|
}
|
|
else {
|
|
fptr->mode &= ~FMODE_SYNC;
|
|
}
|
|
return sync;
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* ios.fsync -> 0 or nil
|
|
*
|
|
* Immediately writes all buffered data in <em>ios</em> to disk.
|
|
* Note that #fsync differs from using IO#sync=. The latter ensures
|
|
* that data is flushed from Ruby's buffers, but does not guarantee
|
|
* that the underlying operating system actually writes it to disk.
|
|
*
|
|
* NotImplementedError is raised
|
|
* if the underlying operating system does not support <em>fsync(2)</em>.
|
|
*/
|
|
|
|
static VALUE
|
|
rb_io_fsync(VALUE io)
|
|
{
|
|
rb_io_t *fptr;
|
|
|
|
io = GetWriteIO(io);
|
|
GetOpenFile(io, fptr);
|
|
|
|
if (io_fflush(fptr) < 0)
|
|
rb_sys_fail(0);
|
|
if ((int)rb_thread_io_blocking_region(nogvl_fsync, fptr, fptr->fd) < 0)
|
|
rb_sys_fail_path(fptr->pathv);
|
|
return INT2FIX(0);
|
|
}
|
|
#else
|
|
# define rb_io_fsync rb_f_notimplement
|
|
# define rb_io_sync rb_f_notimplement
|
|
static VALUE
|
|
rb_io_set_sync(VALUE io, VALUE sync)
|
|
{
|
|
rb_notimplement();
|
|
UNREACHABLE;
|
|
}
|
|
#endif
|
|
|
|
#ifdef HAVE_FDATASYNC
|
|
static VALUE
|
|
nogvl_fdatasync(void *ptr)
|
|
{
|
|
rb_io_t *fptr = ptr;
|
|
|
|
#ifdef _WIN32
|
|
if (GetFileType((HANDLE)rb_w32_get_osfhandle(fptr->fd)) != FILE_TYPE_DISK)
|
|
return 0;
|
|
#endif
|
|
return (VALUE)fdatasync(fptr->fd);
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* ios.fdatasync -> 0 or nil
|
|
*
|
|
* Immediately writes all buffered data in <em>ios</em> to disk.
|
|
*
|
|
* If the underlying operating system does not support <em>fdatasync(2)</em>,
|
|
* IO#fsync is called instead (which might raise a
|
|
* NotImplementedError).
|
|
*/
|
|
|
|
static VALUE
|
|
rb_io_fdatasync(VALUE io)
|
|
{
|
|
rb_io_t *fptr;
|
|
|
|
io = GetWriteIO(io);
|
|
GetOpenFile(io, fptr);
|
|
|
|
if (io_fflush(fptr) < 0)
|
|
rb_sys_fail(0);
|
|
|
|
if ((int)rb_thread_io_blocking_region(nogvl_fdatasync, fptr, fptr->fd) == 0)
|
|
return INT2FIX(0);
|
|
|
|
/* fall back */
|
|
return rb_io_fsync(io);
|
|
}
|
|
#else
|
|
#define rb_io_fdatasync rb_io_fsync
|
|
#endif
|
|
|
|
/*
|
|
* call-seq:
|
|
* ios.fileno -> integer
|
|
* ios.to_i -> integer
|
|
*
|
|
* Returns an integer representing the numeric file descriptor for
|
|
* <em>ios</em>.
|
|
*
|
|
* $stdin.fileno #=> 0
|
|
* $stdout.fileno #=> 1
|
|
*/
|
|
|
|
static VALUE
|
|
rb_io_fileno(VALUE io)
|
|
{
|
|
rb_io_t *fptr = RFILE(io)->fptr;
|
|
int fd;
|
|
|
|
rb_io_check_closed(fptr);
|
|
fd = fptr->fd;
|
|
return INT2FIX(fd);
|
|
}
|
|
|
|
|
|
/*
|
|
* call-seq:
|
|
* ios.pid -> integer
|
|
*
|
|
* Returns the process ID of a child process associated with
|
|
* <em>ios</em>. This will be set by IO.popen.
|
|
*
|
|
* pipe = IO.popen("-")
|
|
* if pipe
|
|
* $stderr.puts "In parent, child pid is #{pipe.pid}"
|
|
* else
|
|
* $stderr.puts "In child, pid is #{$$}"
|
|
* end
|
|
*
|
|
* <em>produces:</em>
|
|
*
|
|
* In child, pid is 26209
|
|
* In parent, child pid is 26209
|
|
*/
|
|
|
|
static VALUE
|
|
rb_io_pid(VALUE io)
|
|
{
|
|
rb_io_t *fptr;
|
|
|
|
GetOpenFile(io, fptr);
|
|
if (!fptr->pid)
|
|
return Qnil;
|
|
return PIDT2NUM(fptr->pid);
|
|
}
|
|
|
|
|
|
/*
|
|
* call-seq:
|
|
* ios.inspect -> string
|
|
*
|
|
* Return a string describing this IO object.
|
|
*/
|
|
|
|
static VALUE
|
|
rb_io_inspect(VALUE obj)
|
|
{
|
|
rb_io_t *fptr;
|
|
VALUE result;
|
|
static const char closed[] = " (closed)";
|
|
|
|
fptr = RFILE(obj)->fptr;
|
|
if (!fptr) return rb_any_to_s(obj);
|
|
result = rb_str_new_cstr("#<");
|
|
rb_str_append(result, rb_class_name(CLASS_OF(obj)));
|
|
rb_str_cat2(result, ":");
|
|
if (NIL_P(fptr->pathv)) {
|
|
if (fptr->fd < 0) {
|
|
rb_str_cat(result, closed+1, strlen(closed)-1);
|
|
}
|
|
else {
|
|
rb_str_catf(result, "fd %d", fptr->fd);
|
|
}
|
|
}
|
|
else {
|
|
rb_str_append(result, fptr->pathv);
|
|
if (fptr->fd < 0) {
|
|
rb_str_cat(result, closed, strlen(closed));
|
|
}
|
|
}
|
|
return rb_str_cat2(result, ">");
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* ios.to_io -> ios
|
|
*
|
|
* Returns <em>ios</em>.
|
|
*/
|
|
|
|
static VALUE
|
|
rb_io_to_io(VALUE io)
|
|
{
|
|
return io;
|
|
}
|
|
|
|
/* reading functions */
|
|
static long
|
|
read_buffered_data(char *ptr, long len, rb_io_t *fptr)
|
|
{
|
|
int n;
|
|
|
|
n = READ_DATA_PENDING_COUNT(fptr);
|
|
if (n <= 0) return 0;
|
|
if (n > len) n = (int)len;
|
|
MEMMOVE(ptr, fptr->rbuf.ptr+fptr->rbuf.off, char, n);
|
|
fptr->rbuf.off += n;
|
|
fptr->rbuf.len -= n;
|
|
return n;
|
|
}
|
|
|
|
static long
|
|
io_bufread(char *ptr, long len, rb_io_t *fptr)
|
|
{
|
|
long offset = 0;
|
|
long n = len;
|
|
long c;
|
|
|
|
if (READ_DATA_PENDING(fptr) == 0) {
|
|
while (n > 0) {
|
|
again:
|
|
c = rb_read_internal(fptr->fd, ptr+offset, n);
|
|
if (c == 0) break;
|
|
if (c < 0) {
|
|
if (fptr_wait_readable(fptr))
|
|
goto again;
|
|
return -1;
|
|
}
|
|
offset += c;
|
|
if ((n -= c) <= 0) break;
|
|
}
|
|
return len - n;
|
|
}
|
|
|
|
while (n > 0) {
|
|
c = read_buffered_data(ptr+offset, n, fptr);
|
|
if (c > 0) {
|
|
offset += c;
|
|
if ((n -= c) <= 0) break;
|
|
}
|
|
rb_io_check_closed(fptr);
|
|
if (io_fillbuf(fptr) < 0) {
|
|
break;
|
|
}
|
|
}
|
|
return len - n;
|
|
}
|
|
|
|
static int io_setstrbuf(VALUE *str, long len);
|
|
|
|
struct bufread_arg {
|
|
char *str_ptr;
|
|
long len;
|
|
rb_io_t *fptr;
|
|
};
|
|
|
|
static VALUE
|
|
bufread_call(VALUE arg)
|
|
{
|
|
struct bufread_arg *p = (struct bufread_arg *)arg;
|
|
p->len = io_bufread(p->str_ptr, p->len, p->fptr);
|
|
return Qundef;
|
|
}
|
|
|
|
static long
|
|
io_fread(VALUE str, long offset, long size, rb_io_t *fptr)
|
|
{
|
|
long len;
|
|
struct bufread_arg arg;
|
|
|
|
io_setstrbuf(&str, offset + size);
|
|
arg.str_ptr = RSTRING_PTR(str) + offset;
|
|
arg.len = size;
|
|
arg.fptr = fptr;
|
|
rb_str_locktmp_ensure(str, bufread_call, (VALUE)&arg);
|
|
len = arg.len;
|
|
if (len < 0) rb_sys_fail_path(fptr->pathv);
|
|
return len;
|
|
}
|
|
|
|
ssize_t
|
|
rb_io_bufread(VALUE io, void *buf, size_t size)
|
|
{
|
|
rb_io_t *fptr;
|
|
|
|
GetOpenFile(io, fptr);
|
|
rb_io_check_readable(fptr);
|
|
return (ssize_t)io_bufread(buf, (long)size, fptr);
|
|
}
|
|
|
|
static long
|
|
remain_size(rb_io_t *fptr)
|
|
{
|
|
struct stat st;
|
|
off_t siz = READ_DATA_PENDING_COUNT(fptr);
|
|
off_t pos;
|
|
|
|
if (fstat(fptr->fd, &st) == 0 && S_ISREG(st.st_mode)
|
|
#if defined(__HAIKU__)
|
|
&& (st.st_dev > 3)
|
|
#endif
|
|
)
|
|
{
|
|
if (io_fflush(fptr) < 0)
|
|
rb_sys_fail(0);
|
|
pos = lseek(fptr->fd, 0, SEEK_CUR);
|
|
if (st.st_size >= pos && pos >= 0) {
|
|
siz += st.st_size - pos;
|
|
if (siz > LONG_MAX) {
|
|
rb_raise(rb_eIOError, "file too big for single read");
|
|
}
|
|
}
|
|
}
|
|
else {
|
|
siz += BUFSIZ;
|
|
}
|
|
return (long)siz;
|
|
}
|
|
|
|
static VALUE
|
|
io_enc_str(VALUE str, rb_io_t *fptr)
|
|
{
|
|
OBJ_TAINT(str);
|
|
rb_enc_associate(str, io_read_encoding(fptr));
|
|
return str;
|
|
}
|
|
|
|
static void
|
|
make_readconv(rb_io_t *fptr, int size)
|
|
{
|
|
if (!fptr->readconv) {
|
|
int ecflags;
|
|
VALUE ecopts;
|
|
const char *sname, *dname;
|
|
ecflags = fptr->encs.ecflags & ~ECONV_NEWLINE_DECORATOR_WRITE_MASK;
|
|
ecopts = fptr->encs.ecopts;
|
|
if (fptr->encs.enc2) {
|
|
sname = rb_enc_name(fptr->encs.enc2);
|
|
dname = rb_enc_name(fptr->encs.enc);
|
|
}
|
|
else {
|
|
sname = dname = "";
|
|
}
|
|
fptr->readconv = rb_econv_open_opts(sname, dname, ecflags, ecopts);
|
|
if (!fptr->readconv)
|
|
rb_exc_raise(rb_econv_open_exc(sname, dname, ecflags));
|
|
fptr->cbuf.off = 0;
|
|
fptr->cbuf.len = 0;
|
|
if (size < IO_CBUF_CAPA_MIN) size = IO_CBUF_CAPA_MIN;
|
|
fptr->cbuf.capa = size;
|
|
fptr->cbuf.ptr = ALLOC_N(char, fptr->cbuf.capa);
|
|
}
|
|
}
|
|
|
|
#define MORE_CHAR_SUSPENDED Qtrue
|
|
#define MORE_CHAR_FINISHED Qnil
|
|
static VALUE
|
|
fill_cbuf(rb_io_t *fptr, int ec_flags)
|
|
{
|
|
const unsigned char *ss, *sp, *se;
|
|
unsigned char *ds, *dp, *de;
|
|
rb_econv_result_t res;
|
|
int putbackable;
|
|
int cbuf_len0;
|
|
VALUE exc;
|
|
|
|
ec_flags |= ECONV_PARTIAL_INPUT;
|
|
|
|
if (fptr->cbuf.len == fptr->cbuf.capa)
|
|
return MORE_CHAR_SUSPENDED; /* cbuf full */
|
|
if (fptr->cbuf.len == 0)
|
|
fptr->cbuf.off = 0;
|
|
else if (fptr->cbuf.off + fptr->cbuf.len == fptr->cbuf.capa) {
|
|
memmove(fptr->cbuf.ptr, fptr->cbuf.ptr+fptr->cbuf.off, fptr->cbuf.len);
|
|
fptr->cbuf.off = 0;
|
|
}
|
|
|
|
cbuf_len0 = fptr->cbuf.len;
|
|
|
|
while (1) {
|
|
ss = sp = (const unsigned char *)fptr->rbuf.ptr + fptr->rbuf.off;
|
|
se = sp + fptr->rbuf.len;
|
|
ds = dp = (unsigned char *)fptr->cbuf.ptr + fptr->cbuf.off + fptr->cbuf.len;
|
|
de = (unsigned char *)fptr->cbuf.ptr + fptr->cbuf.capa;
|
|
res = rb_econv_convert(fptr->readconv, &sp, se, &dp, de, ec_flags);
|
|
fptr->rbuf.off += (int)(sp - ss);
|
|
fptr->rbuf.len -= (int)(sp - ss);
|
|
fptr->cbuf.len += (int)(dp - ds);
|
|
|
|
putbackable = rb_econv_putbackable(fptr->readconv);
|
|
if (putbackable) {
|
|
rb_econv_putback(fptr->readconv, (unsigned char *)fptr->rbuf.ptr + fptr->rbuf.off - putbackable, putbackable);
|
|
fptr->rbuf.off -= putbackable;
|
|
fptr->rbuf.len += putbackable;
|
|
}
|
|
|
|
exc = rb_econv_make_exception(fptr->readconv);
|
|
if (!NIL_P(exc))
|
|
return exc;
|
|
|
|
if (cbuf_len0 != fptr->cbuf.len)
|
|
return MORE_CHAR_SUSPENDED;
|
|
|
|
if (res == econv_finished) {
|
|
return MORE_CHAR_FINISHED;
|
|
}
|
|
|
|
if (res == econv_source_buffer_empty) {
|
|
if (fptr->rbuf.len == 0) {
|
|
READ_CHECK(fptr);
|
|
if (io_fillbuf(fptr) < 0) {
|
|
if (!fptr->readconv) {
|
|
return MORE_CHAR_FINISHED;
|
|
}
|
|
ds = dp = (unsigned char *)fptr->cbuf.ptr + fptr->cbuf.off + fptr->cbuf.len;
|
|
de = (unsigned char *)fptr->cbuf.ptr + fptr->cbuf.capa;
|
|
res = rb_econv_convert(fptr->readconv, NULL, NULL, &dp, de, 0);
|
|
fptr->cbuf.len += (int)(dp - ds);
|
|
rb_econv_check_error(fptr->readconv);
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
if (cbuf_len0 != fptr->cbuf.len)
|
|
return MORE_CHAR_SUSPENDED;
|
|
|
|
return MORE_CHAR_FINISHED;
|
|
}
|
|
|
|
static VALUE
|
|
more_char(rb_io_t *fptr)
|
|
{
|
|
VALUE v;
|
|
v = fill_cbuf(fptr, ECONV_AFTER_OUTPUT);
|
|
if (v != MORE_CHAR_SUSPENDED && v != MORE_CHAR_FINISHED)
|
|
rb_exc_raise(v);
|
|
return v;
|
|
}
|
|
|
|
static VALUE
|
|
io_shift_cbuf(rb_io_t *fptr, int len, VALUE *strp)
|
|
{
|
|
VALUE str = Qnil;
|
|
if (strp) {
|
|
str = *strp;
|
|
if (NIL_P(str)) {
|
|
*strp = str = rb_str_new(fptr->cbuf.ptr+fptr->cbuf.off, len);
|
|
}
|
|
else {
|
|
rb_str_cat(str, fptr->cbuf.ptr+fptr->cbuf.off, len);
|
|
}
|
|
OBJ_TAINT(str);
|
|
rb_enc_associate(str, fptr->encs.enc);
|
|
}
|
|
fptr->cbuf.off += len;
|
|
fptr->cbuf.len -= len;
|
|
/* xxx: set coderange */
|
|
if (fptr->cbuf.len == 0)
|
|
fptr->cbuf.off = 0;
|
|
else if (fptr->cbuf.capa/2 < fptr->cbuf.off) {
|
|
memmove(fptr->cbuf.ptr, fptr->cbuf.ptr+fptr->cbuf.off, fptr->cbuf.len);
|
|
fptr->cbuf.off = 0;
|
|
}
|
|
return str;
|
|
}
|
|
|
|
static int
|
|
io_setstrbuf(VALUE *str, long len)
|
|
{
|
|
#ifdef _WIN32
|
|
len = (len + 1) & ~1L; /* round up for wide char */
|
|
#endif
|
|
if (NIL_P(*str)) {
|
|
*str = rb_str_new(0, len);
|
|
return TRUE;
|
|
}
|
|
else {
|
|
VALUE s = StringValue(*str);
|
|
long clen = RSTRING_LEN(s);
|
|
if (clen >= len) {
|
|
rb_str_modify(s);
|
|
return FALSE;
|
|
}
|
|
len -= clen;
|
|
}
|
|
rb_str_modify_expand(*str, len);
|
|
return FALSE;
|
|
}
|
|
|
|
#define MAX_REALLOC_GAP 4096
|
|
static void
|
|
io_shrink_read_string(VALUE str, long n)
|
|
{
|
|
if (rb_str_capacity(str) - n > MAX_REALLOC_GAP) {
|
|
rb_str_resize(str, n);
|
|
}
|
|
}
|
|
|
|
static void
|
|
io_set_read_length(VALUE str, long n, int shrinkable)
|
|
{
|
|
if (RSTRING_LEN(str) != n) {
|
|
rb_str_modify(str);
|
|
rb_str_set_len(str, n);
|
|
if (shrinkable) io_shrink_read_string(str, n);
|
|
}
|
|
}
|
|
|
|
static VALUE
|
|
read_all(rb_io_t *fptr, long siz, VALUE str)
|
|
{
|
|
long bytes;
|
|
long n;
|
|
long pos;
|
|
rb_encoding *enc;
|
|
int cr;
|
|
int shrinkable;
|
|
|
|
if (NEED_READCONV(fptr)) {
|
|
int first = !NIL_P(str);
|
|
SET_BINARY_MODE(fptr);
|
|
shrinkable = io_setstrbuf(&str,0);
|
|
make_readconv(fptr, 0);
|
|
while (1) {
|
|
VALUE v;
|
|
if (fptr->cbuf.len) {
|
|
if (first) rb_str_set_len(str, first = 0);
|
|
io_shift_cbuf(fptr, fptr->cbuf.len, &str);
|
|
}
|
|
v = fill_cbuf(fptr, 0);
|
|
if (v != MORE_CHAR_SUSPENDED && v != MORE_CHAR_FINISHED) {
|
|
if (fptr->cbuf.len) {
|
|
if (first) rb_str_set_len(str, first = 0);
|
|
io_shift_cbuf(fptr, fptr->cbuf.len, &str);
|
|
}
|
|
rb_exc_raise(v);
|
|
}
|
|
if (v == MORE_CHAR_FINISHED) {
|
|
clear_readconv(fptr);
|
|
if (first) rb_str_set_len(str, first = 0);
|
|
if (shrinkable) io_shrink_read_string(str, RSTRING_LEN(str));
|
|
return io_enc_str(str, fptr);
|
|
}
|
|
}
|
|
}
|
|
|
|
NEED_NEWLINE_DECORATOR_ON_READ_CHECK(fptr);
|
|
bytes = 0;
|
|
pos = 0;
|
|
|
|
enc = io_read_encoding(fptr);
|
|
cr = 0;
|
|
|
|
if (siz == 0) siz = BUFSIZ;
|
|
shrinkable = io_setstrbuf(&str, siz);
|
|
for (;;) {
|
|
READ_CHECK(fptr);
|
|
n = io_fread(str, bytes, siz - bytes, fptr);
|
|
if (n == 0 && bytes == 0) {
|
|
rb_str_set_len(str, 0);
|
|
break;
|
|
}
|
|
bytes += n;
|
|
rb_str_set_len(str, bytes);
|
|
if (cr != ENC_CODERANGE_BROKEN)
|
|
pos += rb_str_coderange_scan_restartable(RSTRING_PTR(str) + pos, RSTRING_PTR(str) + bytes, enc, &cr);
|
|
if (bytes < siz) break;
|
|
siz += BUFSIZ;
|
|
rb_str_modify_expand(str, BUFSIZ);
|
|
}
|
|
if (shrinkable) io_shrink_read_string(str, RSTRING_LEN(str));
|
|
str = io_enc_str(str, fptr);
|
|
ENC_CODERANGE_SET(str, cr);
|
|
return str;
|
|
}
|
|
|
|
void
|
|
rb_io_set_nonblock(rb_io_t *fptr)
|
|
{
|
|
if (rb_fd_set_nonblock(fptr->fd) != 0) {
|
|
rb_sys_fail_path(fptr->pathv);
|
|
}
|
|
}
|
|
|
|
static VALUE
|
|
read_internal_call(VALUE arg)
|
|
{
|
|
struct io_internal_read_struct *iis = (struct io_internal_read_struct *)arg;
|
|
|
|
return rb_thread_io_blocking_region(internal_read_func, iis, iis->fd);
|
|
}
|
|
|
|
static long
|
|
read_internal_locktmp(VALUE str, struct io_internal_read_struct *iis)
|
|
{
|
|
return (long)rb_str_locktmp_ensure(str, read_internal_call, (VALUE)iis);
|
|
}
|
|
|
|
#define no_exception_p(opts) !rb_opts_exception_p((opts), TRUE)
|
|
|
|
static VALUE
|
|
io_getpartial(int argc, VALUE *argv, VALUE io, int no_exception, int nonblock)
|
|
{
|
|
rb_io_t *fptr;
|
|
VALUE length, str;
|
|
long n, len;
|
|
struct io_internal_read_struct iis;
|
|
int shrinkable;
|
|
|
|
rb_scan_args(argc, argv, "11", &length, &str);
|
|
|
|
if ((len = NUM2LONG(length)) < 0) {
|
|
rb_raise(rb_eArgError, "negative length %ld given", len);
|
|
}
|
|
|
|
shrinkable = io_setstrbuf(&str, len);
|
|
OBJ_TAINT(str);
|
|
|
|
GetOpenFile(io, fptr);
|
|
rb_io_check_byte_readable(fptr);
|
|
|
|
if (len == 0)
|
|
return str;
|
|
|
|
if (!nonblock)
|
|
READ_CHECK(fptr);
|
|
n = read_buffered_data(RSTRING_PTR(str), len, fptr);
|
|
if (n <= 0) {
|
|
again:
|
|
if (nonblock) {
|
|
rb_io_set_nonblock(fptr);
|
|
}
|
|
io_setstrbuf(&str, len);
|
|
iis.fd = fptr->fd;
|
|
iis.nonblock = nonblock;
|
|
iis.buf = RSTRING_PTR(str);
|
|
iis.capa = len;
|
|
n = read_internal_locktmp(str, &iis);
|
|
if (n < 0) {
|
|
int e = errno;
|
|
if (!nonblock && fptr_wait_readable(fptr))
|
|
goto again;
|
|
if (nonblock && (e == EWOULDBLOCK || e == EAGAIN)) {
|
|
if (no_exception)
|
|
return sym_wait_readable;
|
|
else
|
|
rb_readwrite_syserr_fail(RB_IO_WAIT_READABLE,
|
|
e, "read would block");
|
|
}
|
|
rb_syserr_fail_path(e, fptr->pathv);
|
|
}
|
|
}
|
|
io_set_read_length(str, n, shrinkable);
|
|
|
|
if (n == 0)
|
|
return Qnil;
|
|
else
|
|
return str;
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* ios.readpartial(maxlen) -> string
|
|
* ios.readpartial(maxlen, outbuf) -> outbuf
|
|
*
|
|
* Reads at most <i>maxlen</i> bytes from the I/O stream.
|
|
* It blocks only if <em>ios</em> has no data immediately available.
|
|
* It doesn't block if some data available.
|
|
*
|
|
* If the optional _outbuf_ argument is present,
|
|
* it must reference a String, which will receive the data.
|
|
* The _outbuf_ will contain only the received data after the method call
|
|
* even if it is not empty at the beginning.
|
|
*
|
|
* It raises EOFError on end of file.
|
|
*
|
|
* readpartial is designed for streams such as pipe, socket, tty, etc.
|
|
* It blocks only when no data immediately available.
|
|
* This means that it blocks only when following all conditions hold.
|
|
* * the byte buffer in the IO object is empty.
|
|
* * the content of the stream is empty.
|
|
* * the stream is not reached to EOF.
|
|
*
|
|
* When readpartial blocks, it waits data or EOF on the stream.
|
|
* If some data is reached, readpartial returns with the data.
|
|
* If EOF is reached, readpartial raises EOFError.
|
|
*
|
|
* When readpartial doesn't blocks, it returns or raises immediately.
|
|
* If the byte buffer is not empty, it returns the data in the buffer.
|
|
* Otherwise if the stream has some content,
|
|
* it returns the data in the stream.
|
|
* Otherwise if the stream is reached to EOF, it raises EOFError.
|
|
*
|
|
* r, w = IO.pipe # buffer pipe content
|
|
* w << "abc" # "" "abc".
|
|
* r.readpartial(4096) #=> "abc" "" ""
|
|
* r.readpartial(4096) # blocks because buffer and pipe is empty.
|
|
*
|
|
* r, w = IO.pipe # buffer pipe content
|
|
* w << "abc" # "" "abc"
|
|
* w.close # "" "abc" EOF
|
|
* r.readpartial(4096) #=> "abc" "" EOF
|
|
* r.readpartial(4096) # raises EOFError
|
|
*
|
|
* r, w = IO.pipe # buffer pipe content
|
|
* w << "abc\ndef\n" # "" "abc\ndef\n"
|
|
* r.gets #=> "abc\n" "def\n" ""
|
|
* w << "ghi\n" # "def\n" "ghi\n"
|
|
* r.readpartial(4096) #=> "def\n" "" "ghi\n"
|
|
* r.readpartial(4096) #=> "ghi\n" "" ""
|
|
*
|
|
* Note that readpartial behaves similar to sysread.
|
|
* The differences are:
|
|
* * If the byte buffer is not empty, read from the byte buffer
|
|
* instead of "sysread for buffered IO (IOError)".
|
|
* * It doesn't cause Errno::EWOULDBLOCK and Errno::EINTR. When
|
|
* readpartial meets EWOULDBLOCK and EINTR by read system call,
|
|
* readpartial retry the system call.
|
|
*
|
|
* The latter means that readpartial is nonblocking-flag insensitive.
|
|
* It blocks on the situation IO#sysread causes Errno::EWOULDBLOCK as
|
|
* if the fd is blocking mode.
|
|
*
|
|
*/
|
|
|
|
static VALUE
|
|
io_readpartial(int argc, VALUE *argv, VALUE io)
|
|
{
|
|
VALUE ret;
|
|
|
|
ret = io_getpartial(argc, argv, io, Qnil, 0);
|
|
if (NIL_P(ret))
|
|
rb_eof_error();
|
|
return ret;
|
|
}
|
|
|
|
static VALUE
|
|
io_nonblock_eof(int no_exception)
|
|
{
|
|
if (!no_exception) {
|
|
rb_eof_error();
|
|
}
|
|
return Qnil;
|
|
}
|
|
|
|
/* :nodoc: */
|
|
static VALUE
|
|
io_read_nonblock(VALUE io, VALUE length, VALUE str, VALUE ex)
|
|
{
|
|
rb_io_t *fptr;
|
|
long n, len;
|
|
struct io_internal_read_struct iis;
|
|
int shrinkable;
|
|
|
|
if ((len = NUM2LONG(length)) < 0) {
|
|
rb_raise(rb_eArgError, "negative length %ld given", len);
|
|
}
|
|
|
|
shrinkable = io_setstrbuf(&str, len);
|
|
OBJ_TAINT(str);
|
|
rb_bool_expected(ex, "exception");
|
|
|
|
GetOpenFile(io, fptr);
|
|
rb_io_check_byte_readable(fptr);
|
|
|
|
if (len == 0)
|
|
return str;
|
|
|
|
n = read_buffered_data(RSTRING_PTR(str), len, fptr);
|
|
if (n <= 0) {
|
|
rb_io_set_nonblock(fptr);
|
|
shrinkable |= io_setstrbuf(&str, len);
|
|
iis.fd = fptr->fd;
|
|
iis.nonblock = 1;
|
|
iis.buf = RSTRING_PTR(str);
|
|
iis.capa = len;
|
|
n = read_internal_locktmp(str, &iis);
|
|
if (n < 0) {
|
|
int e = errno;
|
|
if ((e == EWOULDBLOCK || e == EAGAIN)) {
|
|
if (!ex) return sym_wait_readable;
|
|
rb_readwrite_syserr_fail(RB_IO_WAIT_READABLE,
|
|
e, "read would block");
|
|
}
|
|
rb_syserr_fail_path(e, fptr->pathv);
|
|
}
|
|
}
|
|
io_set_read_length(str, n, shrinkable);
|
|
|
|
if (n == 0) {
|
|
if (!ex) return Qnil;
|
|
rb_eof_error();
|
|
}
|
|
|
|
return str;
|
|
}
|
|
|
|
/* :nodoc: */
|
|
static VALUE
|
|
io_write_nonblock(VALUE io, VALUE str, VALUE ex)
|
|
{
|
|
rb_io_t *fptr;
|
|
long n;
|
|
|
|
if (!RB_TYPE_P(str, T_STRING))
|
|
str = rb_obj_as_string(str);
|
|
rb_bool_expected(ex, "exception");
|
|
|
|
io = GetWriteIO(io);
|
|
GetOpenFile(io, fptr);
|
|
rb_io_check_writable(fptr);
|
|
|
|
if (io_fflush(fptr) < 0)
|
|
rb_sys_fail(0);
|
|
|
|
rb_io_set_nonblock(fptr);
|
|
n = write(fptr->fd, RSTRING_PTR(str), RSTRING_LEN(str));
|
|
RB_GC_GUARD(str);
|
|
|
|
if (n < 0) {
|
|
int e = errno;
|
|
if (e == EWOULDBLOCK || e == EAGAIN) {
|
|
if (!ex) {
|
|
return sym_wait_writable;
|
|
}
|
|
else {
|
|
rb_readwrite_syserr_fail(RB_IO_WAIT_WRITABLE, e, "write would block");
|
|
}
|
|
}
|
|
rb_syserr_fail_path(e, fptr->pathv);
|
|
}
|
|
|
|
return LONG2FIX(n);
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* ios.read([length [, outbuf]]) -> string, outbuf, or nil
|
|
*
|
|
* Reads _length_ bytes from the I/O stream.
|
|
*
|
|
* _length_ must be a non-negative integer or +nil+.
|
|
*
|
|
* If _length_ is a positive integer, +read+ tries to read
|
|
* _length_ bytes without any conversion (binary mode).
|
|
* It returns +nil+ if an EOF is encountered before anything can be read.
|
|
* Fewer than _length_ bytes are returned if an EOF is encountered during
|
|
* the read.
|
|
* In the case of an integer _length_, the resulting string is always
|
|
* in ASCII-8BIT encoding.
|
|
*
|
|
* If _length_ is omitted or is +nil+, it reads until EOF
|
|
* and the encoding conversion is applied, if applicable.
|
|
* A string is returned even if EOF is encountered before any data is read.
|
|
*
|
|
* If _length_ is zero, it returns an empty string (<code>""</code>).
|
|
*
|
|
* If the optional _outbuf_ argument is present,
|
|
* it must reference a String, which will receive the data.
|
|
* The _outbuf_ will contain only the received data after the method call
|
|
* even if it is not empty at the beginning.
|
|
*
|
|
* When this method is called at end of file, it returns +nil+
|
|
* or <code>""</code>, depending on _length_:
|
|
* +read+, <code>read(nil)</code>, and <code>read(0)</code> return
|
|
* <code>""</code>,
|
|
* <code>read(<i>positive_integer</i>)</code> returns +nil+.
|
|
*
|
|
* f = File.new("testfile")
|
|
* f.read(16) #=> "This is line one"
|
|
*
|
|
* # read whole file
|
|
* open("file") do |f|
|
|
* data = f.read # This returns a string even if the file is empty.
|
|
* # ...
|
|
* end
|
|
*
|
|
* # iterate over fixed length records
|
|
* open("fixed-record-file") do |f|
|
|
* while record = f.read(256)
|
|
* # ...
|
|
* end
|
|
* end
|
|
*
|
|
* # iterate over variable length records,
|
|
* # each record is prefixed by its 32-bit length
|
|
* open("variable-record-file") do |f|
|
|
* while len = f.read(4)
|
|
* len = len.unpack("N")[0] # 32-bit length
|
|
* record = f.read(len) # This returns a string even if len is 0.
|
|
* end
|
|
* end
|
|
*
|
|
* Note that this method behaves like the fread() function in C.
|
|
* This means it retries to invoke read(2) system calls to read data
|
|
* with the specified length (or until EOF).
|
|
* This behavior is preserved even if <i>ios</i> is in non-blocking mode.
|
|
* (This method is non-blocking flag insensitive as other methods.)
|
|
* If you need the behavior like a single read(2) system call,
|
|
* consider #readpartial, #read_nonblock, and #sysread.
|
|
*/
|
|
|
|
static VALUE
|
|
io_read(int argc, VALUE *argv, VALUE io)
|
|
{
|
|
rb_io_t *fptr;
|
|
long n, len;
|
|
VALUE length, str;
|
|
int shrinkable;
|
|
#if defined(RUBY_TEST_CRLF_ENVIRONMENT) || defined(_WIN32)
|
|
int previous_mode;
|
|
#endif
|
|
|
|
rb_scan_args(argc, argv, "02", &length, &str);
|
|
|
|
if (NIL_P(length)) {
|
|
GetOpenFile(io, fptr);
|
|
rb_io_check_char_readable(fptr);
|
|
return read_all(fptr, remain_size(fptr), str);
|
|
}
|
|
len = NUM2LONG(length);
|
|
if (len < 0) {
|
|
rb_raise(rb_eArgError, "negative length %ld given", len);
|
|
}
|
|
|
|
shrinkable = io_setstrbuf(&str,len);
|
|
|
|
GetOpenFile(io, fptr);
|
|
rb_io_check_byte_readable(fptr);
|
|
if (len == 0) {
|
|
io_set_read_length(str, 0, shrinkable);
|
|
return str;
|
|
}
|
|
|
|
READ_CHECK(fptr);
|
|
#if defined(RUBY_TEST_CRLF_ENVIRONMENT) || defined(_WIN32)
|
|
previous_mode = set_binary_mode_with_seek_cur(fptr);
|
|
#endif
|
|
n = io_fread(str, 0, len, fptr);
|
|
io_set_read_length(str, n, shrinkable);
|
|
#if defined(RUBY_TEST_CRLF_ENVIRONMENT) || defined(_WIN32)
|
|
if (previous_mode == O_TEXT) {
|
|
setmode(fptr->fd, O_TEXT);
|
|
}
|
|
#endif
|
|
if (n == 0) return Qnil;
|
|
OBJ_TAINT(str);
|
|
|
|
return str;
|
|
}
|
|
|
|
static void
|
|
rscheck(const char *rsptr, long rslen, VALUE rs)
|
|
{
|
|
if (!rs) return;
|
|
if (RSTRING_PTR(rs) != rsptr && RSTRING_LEN(rs) != rslen)
|
|
rb_raise(rb_eRuntimeError, "rs modified");
|
|
}
|
|
|
|
static int
|
|
appendline(rb_io_t *fptr, int delim, VALUE *strp, long *lp)
|
|
{
|
|
VALUE str = *strp;
|
|
long limit = *lp;
|
|
|
|
if (NEED_READCONV(fptr)) {
|
|
SET_BINARY_MODE(fptr);
|
|
make_readconv(fptr, 0);
|
|
do {
|
|
const char *p, *e;
|
|
int searchlen = READ_CHAR_PENDING_COUNT(fptr);
|
|
if (searchlen) {
|
|
p = READ_CHAR_PENDING_PTR(fptr);
|
|
if (0 < limit && limit < searchlen)
|
|
searchlen = (int)limit;
|
|
e = memchr(p, delim, searchlen);
|
|
if (e) {
|
|
int len = (int)(e-p+1);
|
|
if (NIL_P(str))
|
|
*strp = str = rb_str_new(p, len);
|
|
else
|
|
rb_str_buf_cat(str, p, len);
|
|
fptr->cbuf.off += len;
|
|
fptr->cbuf.len -= len;
|
|
limit -= len;
|
|
*lp = limit;
|
|
return delim;
|
|
}
|
|
|
|
if (NIL_P(str))
|
|
*strp = str = rb_str_new(p, searchlen);
|
|
else
|
|
rb_str_buf_cat(str, p, searchlen);
|
|
fptr->cbuf.off += searchlen;
|
|
fptr->cbuf.len -= searchlen;
|
|
limit -= searchlen;
|
|
|
|
if (limit == 0) {
|
|
*lp = limit;
|
|
return (unsigned char)RSTRING_PTR(str)[RSTRING_LEN(str)-1];
|
|
}
|
|
}
|
|
} while (more_char(fptr) != MORE_CHAR_FINISHED);
|
|
clear_readconv(fptr);
|
|
*lp = limit;
|
|
return EOF;
|
|
}
|
|
|
|
NEED_NEWLINE_DECORATOR_ON_READ_CHECK(fptr);
|
|
do {
|
|
long pending = READ_DATA_PENDING_COUNT(fptr);
|
|
if (pending > 0) {
|
|
const char *p = READ_DATA_PENDING_PTR(fptr);
|
|
const char *e;
|
|
long last;
|
|
|
|
if (limit > 0 && pending > limit) pending = limit;
|
|
e = memchr(p, delim, pending);
|
|
if (e) pending = e - p + 1;
|
|
if (!NIL_P(str)) {
|
|
last = RSTRING_LEN(str);
|
|
rb_str_resize(str, last + pending);
|
|
}
|
|
else {
|
|
last = 0;
|
|
*strp = str = rb_str_buf_new(pending);
|
|
rb_str_set_len(str, pending);
|
|
}
|
|
read_buffered_data(RSTRING_PTR(str) + last, pending, fptr); /* must not fail */
|
|
limit -= pending;
|
|
*lp = limit;
|
|
if (e) return delim;
|
|
if (limit == 0)
|
|
return (unsigned char)RSTRING_PTR(str)[RSTRING_LEN(str)-1];
|
|
}
|
|
READ_CHECK(fptr);
|
|
} while (io_fillbuf(fptr) >= 0);
|
|
*lp = limit;
|
|
return EOF;
|
|
}
|
|
|
|
static inline int
|
|
swallow(rb_io_t *fptr, int term)
|
|
{
|
|
if (NEED_READCONV(fptr)) {
|
|
rb_encoding *enc = io_read_encoding(fptr);
|
|
int needconv = rb_enc_mbminlen(enc) != 1;
|
|
SET_BINARY_MODE(fptr);
|
|
make_readconv(fptr, 0);
|
|
do {
|
|
size_t cnt;
|
|
while ((cnt = READ_CHAR_PENDING_COUNT(fptr)) > 0) {
|
|
const char *p = READ_CHAR_PENDING_PTR(fptr);
|
|
int i;
|
|
if (!needconv) {
|
|
if (*p != term) return TRUE;
|
|
i = (int)cnt;
|
|
while (--i && *++p == term);
|
|
}
|
|
else {
|
|
const char *e = p + cnt;
|
|
if (rb_enc_ascget(p, e, &i, enc) != term) return TRUE;
|
|
while ((p += i) < e && rb_enc_ascget(p, e, &i, enc) == term);
|
|
i = (int)(e - p);
|
|
}
|
|
io_shift_cbuf(fptr, (int)cnt - i, NULL);
|
|
}
|
|
} while (more_char(fptr) != MORE_CHAR_FINISHED);
|
|
return FALSE;
|
|
}
|
|
|
|
NEED_NEWLINE_DECORATOR_ON_READ_CHECK(fptr);
|
|
do {
|
|
size_t cnt;
|
|
while ((cnt = READ_DATA_PENDING_COUNT(fptr)) > 0) {
|
|
char buf[1024];
|
|
const char *p = READ_DATA_PENDING_PTR(fptr);
|
|
int i;
|
|
if (cnt > sizeof buf) cnt = sizeof buf;
|
|
if (*p != term) return TRUE;
|
|
i = (int)cnt;
|
|
while (--i && *++p == term);
|
|
if (!read_buffered_data(buf, cnt - i, fptr)) /* must not fail */
|
|
rb_sys_fail_path(fptr->pathv);
|
|
}
|
|
READ_CHECK(fptr);
|
|
} while (io_fillbuf(fptr) == 0);
|
|
return FALSE;
|
|
}
|
|
|
|
static VALUE
|
|
rb_io_getline_fast(rb_io_t *fptr, rb_encoding *enc, int chomp)
|
|
{
|
|
VALUE str = Qnil;
|
|
int len = 0;
|
|
long pos = 0;
|
|
int cr = 0;
|
|
|
|
do {
|
|
int pending = READ_DATA_PENDING_COUNT(fptr);
|
|
|
|
if (pending > 0) {
|
|
const char *p = READ_DATA_PENDING_PTR(fptr);
|
|
const char *e;
|
|
int chomplen = 0;
|
|
|
|
e = memchr(p, '\n', pending);
|
|
if (e) {
|
|
pending = (int)(e - p + 1);
|
|
if (chomp) {
|
|
chomplen = (pending > 1 && *(e-1) == '\r') + 1;
|
|
}
|
|
}
|
|
if (NIL_P(str)) {
|
|
str = rb_str_new(p, pending - chomplen);
|
|
fptr->rbuf.off += pending;
|
|
fptr->rbuf.len -= pending;
|
|
}
|
|
else {
|
|
rb_str_resize(str, len + pending - chomplen);
|
|
read_buffered_data(RSTRING_PTR(str)+len, pending - chomplen, fptr);
|
|
fptr->rbuf.off += chomplen;
|
|
fptr->rbuf.len -= chomplen;
|
|
if (pending == 1 && chomplen == 1 && len > 0) {
|
|
if (RSTRING_PTR(str)[len-1] == '\r') {
|
|
rb_str_resize(str, --len);
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
len += pending - chomplen;
|
|
if (cr != ENC_CODERANGE_BROKEN)
|
|
pos += rb_str_coderange_scan_restartable(RSTRING_PTR(str) + pos, RSTRING_PTR(str) + len, enc, &cr);
|
|
if (e) break;
|
|
}
|
|
READ_CHECK(fptr);
|
|
} while (io_fillbuf(fptr) >= 0);
|
|
if (NIL_P(str)) return Qnil;
|
|
|
|
str = io_enc_str(str, fptr);
|
|
ENC_CODERANGE_SET(str, cr);
|
|
fptr->lineno++;
|
|
|
|
return str;
|
|
}
|
|
|
|
struct getline_arg {
|
|
VALUE io;
|
|
VALUE rs;
|
|
long limit;
|
|
unsigned int chomp: 1;
|
|
};
|
|
|
|
static void
|
|
extract_getline_opts(VALUE opts, struct getline_arg *args)
|
|
{
|
|
int chomp = FALSE;
|
|
if (!NIL_P(opts)) {
|
|
static ID kwds[1];
|
|
VALUE vchomp;
|
|
if (!kwds[0]) {
|
|
kwds[0] = rb_intern_const("chomp");
|
|
}
|
|
rb_get_kwargs(opts, kwds, 0, -2, &vchomp);
|
|
chomp = (vchomp != Qundef) && RTEST(vchomp);
|
|
}
|
|
args->chomp = chomp;
|
|
}
|
|
|
|
static void
|
|
extract_getline_args(int argc, VALUE *argv, struct getline_arg *args)
|
|
{
|
|
VALUE rs = rb_rs, lim = Qnil;
|
|
|
|
if (argc == 1) {
|
|
VALUE tmp = Qnil;
|
|
|
|
if (NIL_P(argv[0]) || !NIL_P(tmp = rb_check_string_type(argv[0]))) {
|
|
rs = tmp;
|
|
}
|
|
else {
|
|
lim = argv[0];
|
|
}
|
|
}
|
|
else if (2 <= argc) {
|
|
rs = argv[0], lim = argv[1];
|
|
if (!NIL_P(rs))
|
|
StringValue(rs);
|
|
}
|
|
args->rs = rs;
|
|
args->limit = NIL_P(lim) ? -1L : NUM2LONG(lim);
|
|
}
|
|
|
|
static void
|
|
check_getline_args(VALUE *rsp, long *limit, VALUE io)
|
|
{
|
|
rb_io_t *fptr;
|
|
VALUE rs = *rsp;
|
|
|
|
if (!NIL_P(rs)) {
|
|
rb_encoding *enc_rs, *enc_io;
|
|
|
|
GetOpenFile(io, fptr);
|
|
enc_rs = rb_enc_get(rs);
|
|
enc_io = io_read_encoding(fptr);
|
|
if (enc_io != enc_rs &&
|
|
(rb_enc_str_coderange(rs) != ENC_CODERANGE_7BIT ||
|
|
(RSTRING_LEN(rs) > 0 && !rb_enc_asciicompat(enc_io)))) {
|
|
if (rs == rb_default_rs) {
|
|
rs = rb_enc_str_new(0, 0, enc_io);
|
|
rb_str_buf_cat_ascii(rs, "\n");
|
|
*rsp = rs;
|
|
}
|
|
else {
|
|
rb_raise(rb_eArgError, "encoding mismatch: %s IO with %s RS",
|
|
rb_enc_name(enc_io),
|
|
rb_enc_name(enc_rs));
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
static void
|
|
prepare_getline_args(int argc, VALUE *argv, struct getline_arg *args, VALUE io)
|
|
{
|
|
VALUE opts;
|
|
argc = rb_scan_args(argc, argv, "02:", NULL, NULL, &opts);
|
|
extract_getline_args(argc, argv, args);
|
|
extract_getline_opts(opts, args);
|
|
check_getline_args(&args->rs, &args->limit, io);
|
|
}
|
|
|
|
static VALUE
|
|
rb_io_getline_0(VALUE rs, long limit, int chomp, rb_io_t *fptr)
|
|
{
|
|
VALUE str = Qnil;
|
|
int nolimit = 0;
|
|
rb_encoding *enc;
|
|
|
|
rb_io_check_char_readable(fptr);
|
|
if (NIL_P(rs) && limit < 0) {
|
|
str = read_all(fptr, 0, Qnil);
|
|
if (RSTRING_LEN(str) == 0) return Qnil;
|
|
if (chomp) rb_str_chomp_string(str, rb_default_rs);
|
|
}
|
|
else if (limit == 0) {
|
|
return rb_enc_str_new(0, 0, io_read_encoding(fptr));
|
|
}
|
|
else if (rs == rb_default_rs && limit < 0 && !NEED_READCONV(fptr) &&
|
|
rb_enc_asciicompat(enc = io_read_encoding(fptr))) {
|
|
NEED_NEWLINE_DECORATOR_ON_READ_CHECK(fptr);
|
|
return rb_io_getline_fast(fptr, enc, chomp);
|
|
}
|
|
else {
|
|
int c, newline = -1;
|
|
const char *rsptr = 0;
|
|
long rslen = 0;
|
|
int rspara = 0;
|
|
int extra_limit = 16;
|
|
int chomp_cr = chomp;
|
|
|
|
SET_BINARY_MODE(fptr);
|
|
enc = io_read_encoding(fptr);
|
|
|
|
if (!NIL_P(rs)) {
|
|
rslen = RSTRING_LEN(rs);
|
|
if (rslen == 0) {
|
|
rsptr = "\n\n";
|
|
rslen = 2;
|
|
rspara = 1;
|
|
swallow(fptr, '\n');
|
|
rs = 0;
|
|
if (!rb_enc_asciicompat(enc)) {
|
|
rs = rb_usascii_str_new(rsptr, rslen);
|
|
rs = rb_str_encode(rs, rb_enc_from_encoding(enc), 0, Qnil);
|
|
OBJ_FREEZE(rs);
|
|
rsptr = RSTRING_PTR(rs);
|
|
rslen = RSTRING_LEN(rs);
|
|
}
|
|
}
|
|
else {
|
|
rsptr = RSTRING_PTR(rs);
|
|
}
|
|
newline = (unsigned char)rsptr[rslen - 1];
|
|
chomp_cr = chomp && rslen == 1 && newline == '\n';
|
|
}
|
|
|
|
/* MS - Optimization */
|
|
while ((c = appendline(fptr, newline, &str, &limit)) != EOF) {
|
|
const char *s, *p, *pp, *e;
|
|
|
|
if (c == newline) {
|
|
if (RSTRING_LEN(str) < rslen) continue;
|
|
s = RSTRING_PTR(str);
|
|
e = RSTRING_END(str);
|
|
p = e - rslen;
|
|
pp = rb_enc_left_char_head(s, p, e, enc);
|
|
if (pp != p) continue;
|
|
if (!rspara) rscheck(rsptr, rslen, rs);
|
|
if (memcmp(p, rsptr, rslen) == 0) {
|
|
if (chomp) {
|
|
if (chomp_cr && p > s && *(p-1) == '\r') --p;
|
|
rb_str_set_len(str, p - s);
|
|
}
|
|
break;
|
|
}
|
|
}
|
|
if (limit == 0) {
|
|
s = RSTRING_PTR(str);
|
|
p = RSTRING_END(str);
|
|
pp = rb_enc_left_char_head(s, p-1, p, enc);
|
|
if (extra_limit &&
|
|
MBCLEN_NEEDMORE_P(rb_enc_precise_mbclen(pp, p, enc))) {
|
|
/* relax the limit while incomplete character.
|
|
* extra_limit limits the relax length */
|
|
limit = 1;
|
|
extra_limit--;
|
|
}
|
|
else {
|
|
nolimit = 1;
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
if (rspara && c != EOF)
|
|
swallow(fptr, '\n');
|
|
if (!NIL_P(str))
|
|
str = io_enc_str(str, fptr);
|
|
}
|
|
|
|
if (!NIL_P(str) && !nolimit) {
|
|
fptr->lineno++;
|
|
}
|
|
|
|
return str;
|
|
}
|
|
|
|
static VALUE
|
|
rb_io_getline_1(VALUE rs, long limit, int chomp, VALUE io)
|
|
{
|
|
rb_io_t *fptr;
|
|
int old_lineno, new_lineno;
|
|
VALUE str;
|
|
|
|
GetOpenFile(io, fptr);
|
|
old_lineno = fptr->lineno;
|
|
str = rb_io_getline_0(rs, limit, chomp, fptr);
|
|
if (!NIL_P(str) && (new_lineno = fptr->lineno) != old_lineno) {
|
|
if (io == ARGF.current_file) {
|
|
ARGF.lineno += new_lineno - old_lineno;
|
|
ARGF.last_lineno = ARGF.lineno;
|
|
}
|
|
else {
|
|
ARGF.last_lineno = new_lineno;
|
|
}
|
|
}
|
|
|
|
return str;
|
|
}
|
|
|
|
static VALUE
|
|
rb_io_getline(int argc, VALUE *argv, VALUE io)
|
|
{
|
|
struct getline_arg args;
|
|
|
|
prepare_getline_args(argc, argv, &args, io);
|
|
return rb_io_getline_1(args.rs, args.limit, args.chomp, io);
|
|
}
|
|
|
|
VALUE
|
|
rb_io_gets(VALUE io)
|
|
{
|
|
return rb_io_getline_1(rb_default_rs, -1, FALSE, io);
|
|
}
|
|
|
|
VALUE
|
|
rb_io_gets_internal(VALUE io)
|
|
{
|
|
rb_io_t *fptr;
|
|
GetOpenFile(io, fptr);
|
|
return rb_io_getline_0(rb_default_rs, -1, FALSE, fptr);
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* ios.gets(sep=$/ [, getline_args]) -> string or nil
|
|
* ios.gets(limit [, getline_args]) -> string or nil
|
|
* ios.gets(sep, limit [, getline_args]) -> string or nil
|
|
*
|
|
* Reads the next ``line'' from the I/O stream; lines are separated by
|
|
* <i>sep</i>. A separator of +nil+ reads the entire
|
|
* contents, and a zero-length separator reads the input a paragraph at
|
|
* a time (two successive newlines in the input separate paragraphs).
|
|
* The stream must be opened for reading or an IOError will be raised.
|
|
* The line read in will be returned and also assigned to
|
|
* <code>$_</code>. Returns +nil+ if called at end of file. If the
|
|
* first argument is an integer, or optional second argument is given,
|
|
* the returning string would not be longer than the given value in
|
|
* bytes.
|
|
*
|
|
* File.new("testfile").gets #=> "This is line one\n"
|
|
* $_ #=> "This is line one\n"
|
|
*
|
|
* File.new("testfile").gets(4)#=> "This"
|
|
*
|
|
* If IO contains multibyte characters byte then <code>gets(1)</code>
|
|
* returns character entirely:
|
|
*
|
|
* # Russian characters take 2 bytes
|
|
* File.write("testfile", "\u{442 435 441 442}")
|
|
* File.open("testfile") {|f|f.gets(1)} #=> "\u0442"
|
|
* File.open("testfile") {|f|f.gets(2)} #=> "\u0442"
|
|
* File.open("testfile") {|f|f.gets(3)} #=> "\u0442\u0435"
|
|
* File.open("testfile") {|f|f.gets(4)} #=> "\u0442\u0435"
|
|
*/
|
|
|
|
static VALUE
|
|
rb_io_gets_m(int argc, VALUE *argv, VALUE io)
|
|
{
|
|
VALUE str;
|
|
|
|
str = rb_io_getline(argc, argv, io);
|
|
rb_lastline_set(str);
|
|
|
|
return str;
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* ios.lineno -> integer
|
|
*
|
|
* Returns the current line number in <em>ios</em>. The stream must be
|
|
* opened for reading. #lineno counts the number of times #gets is called
|
|
* rather than the number of newlines encountered. The two values will
|
|
* differ if #gets is called with a separator other than newline.
|
|
*
|
|
* Methods that use <code>$/</code> like #each, #lines and #readline will
|
|
* also increment #lineno.
|
|
*
|
|
* See also the <code>$.</code> variable.
|
|
*
|
|
* f = File.new("testfile")
|
|
* f.lineno #=> 0
|
|
* f.gets #=> "This is line one\n"
|
|
* f.lineno #=> 1
|
|
* f.gets #=> "This is line two\n"
|
|
* f.lineno #=> 2
|
|
*/
|
|
|
|
static VALUE
|
|
rb_io_lineno(VALUE io)
|
|
{
|
|
rb_io_t *fptr;
|
|
|
|
GetOpenFile(io, fptr);
|
|
rb_io_check_char_readable(fptr);
|
|
return INT2NUM(fptr->lineno);
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* ios.lineno = integer -> integer
|
|
*
|
|
* Manually sets the current line number to the given value.
|
|
* <code>$.</code> is updated only on the next read.
|
|
*
|
|
* f = File.new("testfile")
|
|
* f.gets #=> "This is line one\n"
|
|
* $. #=> 1
|
|
* f.lineno = 1000
|
|
* f.lineno #=> 1000
|
|
* $. #=> 1 # lineno of last read
|
|
* f.gets #=> "This is line two\n"
|
|
* $. #=> 1001 # lineno of last read
|
|
*/
|
|
|
|
static VALUE
|
|
rb_io_set_lineno(VALUE io, VALUE lineno)
|
|
{
|
|
rb_io_t *fptr;
|
|
|
|
GetOpenFile(io, fptr);
|
|
rb_io_check_char_readable(fptr);
|
|
fptr->lineno = NUM2INT(lineno);
|
|
return lineno;
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* ios.readline(sep=$/ [, getline_args]) -> string
|
|
* ios.readline(limit [, getline_args]) -> string
|
|
* ios.readline(sep, limit [, getline_args]) -> string
|
|
*
|
|
* Reads a line as with IO#gets, but raises an EOFError on end of file.
|
|
*/
|
|
|
|
static VALUE
|
|
rb_io_readline(int argc, VALUE *argv, VALUE io)
|
|
{
|
|
VALUE line = rb_io_gets_m(argc, argv, io);
|
|
|
|
if (NIL_P(line)) {
|
|
rb_eof_error();
|
|
}
|
|
return line;
|
|
}
|
|
|
|
static VALUE io_readlines(const struct getline_arg *arg, VALUE io);
|
|
|
|
/*
|
|
* call-seq:
|
|
* ios.readlines(sep=$/ [, getline_args]) -> array
|
|
* ios.readlines(limit [, getline_args]) -> array
|
|
* ios.readlines(sep, limit [, getline_args]) -> array
|
|
*
|
|
* Reads all of the lines in <em>ios</em>, and returns them in
|
|
* an array. Lines are separated by the optional <i>sep</i>. If
|
|
* <i>sep</i> is +nil+, the rest of the stream is returned
|
|
* as a single record.
|
|
* If the first argument is an integer, or an
|
|
* optional second argument is given, the returning string would not be
|
|
* longer than the given value in bytes. The stream must be opened for
|
|
* reading or an IOError will be raised.
|
|
*
|
|
* f = File.new("testfile")
|
|
* f.readlines[0] #=> "This is line one\n"
|
|
*
|
|
* f = File.new("testfile", chomp: true)
|
|
* f.readlines[0] #=> "This is line one"
|
|
*
|
|
* See IO.readlines for details about getline_args.
|
|
*/
|
|
|
|
static VALUE
|
|
rb_io_readlines(int argc, VALUE *argv, VALUE io)
|
|
{
|
|
struct getline_arg args;
|
|
|
|
prepare_getline_args(argc, argv, &args, io);
|
|
return io_readlines(&args, io);
|
|
}
|
|
|
|
static VALUE
|
|
io_readlines(const struct getline_arg *arg, VALUE io)
|
|
{
|
|
VALUE line, ary;
|
|
|
|
if (arg->limit == 0)
|
|
rb_raise(rb_eArgError, "invalid limit: 0 for readlines");
|
|
ary = rb_ary_new();
|
|
while (!NIL_P(line = rb_io_getline_1(arg->rs, arg->limit, arg->chomp, io))) {
|
|
rb_ary_push(ary, line);
|
|
}
|
|
return ary;
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* ios.each(sep=$/ [, getline_args]) {|line| block } -> ios
|
|
* ios.each(limit [, getline_args]) {|line| block } -> ios
|
|
* ios.each(sep, limit [, getline_args]) {|line| block } -> ios
|
|
* ios.each(...) -> an_enumerator
|
|
*
|
|
* ios.each_line(sep=$/ [, getline_args]) {|line| block } -> ios
|
|
* ios.each_line(limit [, getline_args]) {|line| block } -> ios
|
|
* ios.each_line(sep, limit [, getline_args]) {|line| block } -> ios
|
|
* ios.each_line(...) -> an_enumerator
|
|
*
|
|
* Executes the block for every line in <em>ios</em>, where lines are
|
|
* separated by <i>sep</i>. <em>ios</em> must be opened for
|
|
* reading or an IOError will be raised.
|
|
*
|
|
* If no block is given, an enumerator is returned instead.
|
|
*
|
|
* f = File.new("testfile")
|
|
* f.each {|line| puts "#{f.lineno}: #{line}" }
|
|
*
|
|
* <em>produces:</em>
|
|
*
|
|
* 1: This is line one
|
|
* 2: This is line two
|
|
* 3: This is line three
|
|
* 4: And so on...
|
|
*
|
|
* See IO.readlines for details about getline_args.
|
|
*/
|
|
|
|
static VALUE
|
|
rb_io_each_line(int argc, VALUE *argv, VALUE io)
|
|
{
|
|
VALUE str;
|
|
struct getline_arg args;
|
|
|
|
RETURN_ENUMERATOR(io, argc, argv);
|
|
prepare_getline_args(argc, argv, &args, io);
|
|
if (args.limit == 0)
|
|
rb_raise(rb_eArgError, "invalid limit: 0 for each_line");
|
|
while (!NIL_P(str = rb_io_getline_1(args.rs, args.limit, args.chomp, io))) {
|
|
rb_yield(str);
|
|
}
|
|
return io;
|
|
}
|
|
|
|
/*
|
|
* This is a deprecated alias for #each_line.
|
|
*/
|
|
|
|
static VALUE
|
|
rb_io_lines(int argc, VALUE *argv, VALUE io)
|
|
{
|
|
rb_warn("IO#lines is deprecated; use #each_line instead");
|
|
if (!rb_block_given_p())
|
|
return rb_enumeratorize(io, ID2SYM(rb_intern("each_line")), argc, argv);
|
|
return rb_io_each_line(argc, argv, io);
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* ios.each_byte {|byte| block } -> ios
|
|
* ios.each_byte -> an_enumerator
|
|
*
|
|
* Calls the given block once for each byte (0..255) in <em>ios</em>,
|
|
* passing the byte as an argument. The stream must be opened for
|
|
* reading or an IOError will be raised.
|
|
*
|
|
* If no block is given, an enumerator is returned instead.
|
|
*
|
|
* f = File.new("testfile")
|
|
* checksum = 0
|
|
* f.each_byte {|x| checksum ^= x } #=> #<File:testfile>
|
|
* checksum #=> 12
|
|
*/
|
|
|
|
static VALUE
|
|
rb_io_each_byte(VALUE io)
|
|
{
|
|
rb_io_t *fptr;
|
|
|
|
RETURN_ENUMERATOR(io, 0, 0);
|
|
GetOpenFile(io, fptr);
|
|
|
|
do {
|
|
while (fptr->rbuf.len > 0) {
|
|
char *p = fptr->rbuf.ptr + fptr->rbuf.off++;
|
|
fptr->rbuf.len--;
|
|
rb_yield(INT2FIX(*p & 0xff));
|
|
errno = 0;
|
|
}
|
|
rb_io_check_byte_readable(fptr);
|
|
READ_CHECK(fptr);
|
|
} while (io_fillbuf(fptr) >= 0);
|
|
return io;
|
|
}
|
|
|
|
/*
|
|
* This is a deprecated alias for #each_byte.
|
|
*/
|
|
|
|
static VALUE
|
|
rb_io_bytes(VALUE io)
|
|
{
|
|
rb_warn("IO#bytes is deprecated; use #each_byte instead");
|
|
if (!rb_block_given_p())
|
|
return rb_enumeratorize(io, ID2SYM(rb_intern("each_byte")), 0, 0);
|
|
return rb_io_each_byte(io);
|
|
}
|
|
|
|
static VALUE
|
|
io_getc(rb_io_t *fptr, rb_encoding *enc)
|
|
{
|
|
int r, n, cr = 0;
|
|
VALUE str;
|
|
|
|
if (NEED_READCONV(fptr)) {
|
|
rb_encoding *read_enc = io_read_encoding(fptr);
|
|
|
|
str = Qnil;
|
|
SET_BINARY_MODE(fptr);
|
|
make_readconv(fptr, 0);
|
|
|
|
while (1) {
|
|
if (fptr->cbuf.len) {
|
|
r = rb_enc_precise_mbclen(fptr->cbuf.ptr+fptr->cbuf.off,
|
|
fptr->cbuf.ptr+fptr->cbuf.off+fptr->cbuf.len,
|
|
read_enc);
|
|
if (!MBCLEN_NEEDMORE_P(r))
|
|
break;
|
|
if (fptr->cbuf.len == fptr->cbuf.capa) {
|
|
rb_raise(rb_eIOError, "too long character");
|
|
}
|
|
}
|
|
|
|
if (more_char(fptr) == MORE_CHAR_FINISHED) {
|
|
if (fptr->cbuf.len == 0) {
|
|
clear_readconv(fptr);
|
|
return Qnil;
|
|
}
|
|
/* return an unit of an incomplete character just before EOF */
|
|
str = rb_enc_str_new(fptr->cbuf.ptr+fptr->cbuf.off, 1, read_enc);
|
|
fptr->cbuf.off += 1;
|
|
fptr->cbuf.len -= 1;
|
|
if (fptr->cbuf.len == 0) clear_readconv(fptr);
|
|
ENC_CODERANGE_SET(str, ENC_CODERANGE_BROKEN);
|
|
return str;
|
|
}
|
|
}
|
|
if (MBCLEN_INVALID_P(r)) {
|
|
r = rb_enc_mbclen(fptr->cbuf.ptr+fptr->cbuf.off,
|
|
fptr->cbuf.ptr+fptr->cbuf.off+fptr->cbuf.len,
|
|
read_enc);
|
|
io_shift_cbuf(fptr, r, &str);
|
|
cr = ENC_CODERANGE_BROKEN;
|
|
}
|
|
else {
|
|
io_shift_cbuf(fptr, MBCLEN_CHARFOUND_LEN(r), &str);
|
|
cr = ENC_CODERANGE_VALID;
|
|
if (MBCLEN_CHARFOUND_LEN(r) == 1 && rb_enc_asciicompat(read_enc) &&
|
|
ISASCII(RSTRING_PTR(str)[0])) {
|
|
cr = ENC_CODERANGE_7BIT;
|
|
}
|
|
}
|
|
str = io_enc_str(str, fptr);
|
|
ENC_CODERANGE_SET(str, cr);
|
|
return str;
|
|
}
|
|
|
|
NEED_NEWLINE_DECORATOR_ON_READ_CHECK(fptr);
|
|
if (io_fillbuf(fptr) < 0) {
|
|
return Qnil;
|
|
}
|
|
if (rb_enc_asciicompat(enc) && ISASCII(fptr->rbuf.ptr[fptr->rbuf.off])) {
|
|
str = rb_str_new(fptr->rbuf.ptr+fptr->rbuf.off, 1);
|
|
fptr->rbuf.off += 1;
|
|
fptr->rbuf.len -= 1;
|
|
cr = ENC_CODERANGE_7BIT;
|
|
}
|
|
else {
|
|
r = rb_enc_precise_mbclen(fptr->rbuf.ptr+fptr->rbuf.off, fptr->rbuf.ptr+fptr->rbuf.off+fptr->rbuf.len, enc);
|
|
if (MBCLEN_CHARFOUND_P(r) &&
|
|
(n = MBCLEN_CHARFOUND_LEN(r)) <= fptr->rbuf.len) {
|
|
str = rb_str_new(fptr->rbuf.ptr+fptr->rbuf.off, n);
|
|
fptr->rbuf.off += n;
|
|
fptr->rbuf.len -= n;
|
|
cr = ENC_CODERANGE_VALID;
|
|
}
|
|
else if (MBCLEN_NEEDMORE_P(r)) {
|
|
str = rb_str_new(fptr->rbuf.ptr+fptr->rbuf.off, fptr->rbuf.len);
|
|
fptr->rbuf.len = 0;
|
|
getc_needmore:
|
|
if (io_fillbuf(fptr) != -1) {
|
|
rb_str_cat(str, fptr->rbuf.ptr+fptr->rbuf.off, 1);
|
|
fptr->rbuf.off++;
|
|
fptr->rbuf.len--;
|
|
r = rb_enc_precise_mbclen(RSTRING_PTR(str), RSTRING_PTR(str)+RSTRING_LEN(str), enc);
|
|
if (MBCLEN_NEEDMORE_P(r)) {
|
|
goto getc_needmore;
|
|
}
|
|
else if (MBCLEN_CHARFOUND_P(r)) {
|
|
cr = ENC_CODERANGE_VALID;
|
|
}
|
|
}
|
|
}
|
|
else {
|
|
str = rb_str_new(fptr->rbuf.ptr+fptr->rbuf.off, 1);
|
|
fptr->rbuf.off++;
|
|
fptr->rbuf.len--;
|
|
}
|
|
}
|
|
if (!cr) cr = ENC_CODERANGE_BROKEN;
|
|
str = io_enc_str(str, fptr);
|
|
ENC_CODERANGE_SET(str, cr);
|
|
return str;
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* ios.each_char {|c| block } -> ios
|
|
* ios.each_char -> an_enumerator
|
|
*
|
|
* Calls the given block once for each character in <em>ios</em>,
|
|
* passing the character as an argument. The stream must be opened for
|
|
* reading or an IOError will be raised.
|
|
*
|
|
* If no block is given, an enumerator is returned instead.
|
|
*
|
|
* f = File.new("testfile")
|
|
* f.each_char {|c| print c, ' ' } #=> #<File:testfile>
|
|
*/
|
|
|
|
static VALUE
|
|
rb_io_each_char(VALUE io)
|
|
{
|
|
rb_io_t *fptr;
|
|
rb_encoding *enc;
|
|
VALUE c;
|
|
|
|
RETURN_ENUMERATOR(io, 0, 0);
|
|
GetOpenFile(io, fptr);
|
|
rb_io_check_char_readable(fptr);
|
|
|
|
enc = io_input_encoding(fptr);
|
|
READ_CHECK(fptr);
|
|
while (!NIL_P(c = io_getc(fptr, enc))) {
|
|
rb_yield(c);
|
|
}
|
|
return io;
|
|
}
|
|
|
|
/*
|
|
* This is a deprecated alias for #each_char.
|
|
*/
|
|
|
|
static VALUE
|
|
rb_io_chars(VALUE io)
|
|
{
|
|
rb_warn("IO#chars is deprecated; use #each_char instead");
|
|
if (!rb_block_given_p())
|
|
return rb_enumeratorize(io, ID2SYM(rb_intern("each_char")), 0, 0);
|
|
return rb_io_each_char(io);
|
|
}
|
|
|
|
|
|
/*
|
|
* call-seq:
|
|
* ios.each_codepoint {|c| block } -> ios
|
|
* ios.codepoints {|c| block } -> ios
|
|
* ios.each_codepoint -> an_enumerator
|
|
* ios.codepoints -> an_enumerator
|
|
*
|
|
* Passes the Integer ordinal of each character in <i>ios</i>,
|
|
* passing the codepoint as an argument. The stream must be opened for
|
|
* reading or an IOError will be raised.
|
|
*
|
|
* If no block is given, an enumerator is returned instead.
|
|
*
|
|
*/
|
|
|
|
static VALUE
|
|
rb_io_each_codepoint(VALUE io)
|
|
{
|
|
rb_io_t *fptr;
|
|
rb_encoding *enc;
|
|
unsigned int c;
|
|
int r, n;
|
|
|
|
RETURN_ENUMERATOR(io, 0, 0);
|
|
GetOpenFile(io, fptr);
|
|
rb_io_check_char_readable(fptr);
|
|
|
|
READ_CHECK(fptr);
|
|
if (NEED_READCONV(fptr)) {
|
|
SET_BINARY_MODE(fptr);
|
|
r = 1; /* no invalid char yet */
|
|
for (;;) {
|
|
make_readconv(fptr, 0);
|
|
for (;;) {
|
|
if (fptr->cbuf.len) {
|
|
if (fptr->encs.enc)
|
|
r = rb_enc_precise_mbclen(fptr->cbuf.ptr+fptr->cbuf.off,
|
|
fptr->cbuf.ptr+fptr->cbuf.off+fptr->cbuf.len,
|
|
fptr->encs.enc);
|
|
else
|
|
r = ONIGENC_CONSTRUCT_MBCLEN_CHARFOUND(1);
|
|
if (!MBCLEN_NEEDMORE_P(r))
|
|
break;
|
|
if (fptr->cbuf.len == fptr->cbuf.capa) {
|
|
rb_raise(rb_eIOError, "too long character");
|
|
}
|
|
}
|
|
if (more_char(fptr) == MORE_CHAR_FINISHED) {
|
|
clear_readconv(fptr);
|
|
if (!MBCLEN_CHARFOUND_P(r)) {
|
|
enc = fptr->encs.enc;
|
|
goto invalid;
|
|
}
|
|
return io;
|
|
}
|
|
}
|
|
if (MBCLEN_INVALID_P(r)) {
|
|
enc = fptr->encs.enc;
|
|
goto invalid;
|
|
}
|
|
n = MBCLEN_CHARFOUND_LEN(r);
|
|
if (fptr->encs.enc) {
|
|
c = rb_enc_codepoint(fptr->cbuf.ptr+fptr->cbuf.off,
|
|
fptr->cbuf.ptr+fptr->cbuf.off+fptr->cbuf.len,
|
|
fptr->encs.enc);
|
|
}
|
|
else {
|
|
c = (unsigned char)fptr->cbuf.ptr[fptr->cbuf.off];
|
|
}
|
|
fptr->cbuf.off += n;
|
|
fptr->cbuf.len -= n;
|
|
rb_yield(UINT2NUM(c));
|
|
}
|
|
}
|
|
NEED_NEWLINE_DECORATOR_ON_READ_CHECK(fptr);
|
|
enc = io_input_encoding(fptr);
|
|
while (io_fillbuf(fptr) >= 0) {
|
|
r = rb_enc_precise_mbclen(fptr->rbuf.ptr+fptr->rbuf.off,
|
|
fptr->rbuf.ptr+fptr->rbuf.off+fptr->rbuf.len, enc);
|
|
if (MBCLEN_CHARFOUND_P(r) &&
|
|
(n = MBCLEN_CHARFOUND_LEN(r)) <= fptr->rbuf.len) {
|
|
c = rb_enc_codepoint(fptr->rbuf.ptr+fptr->rbuf.off,
|
|
fptr->rbuf.ptr+fptr->rbuf.off+fptr->rbuf.len, enc);
|
|
fptr->rbuf.off += n;
|
|
fptr->rbuf.len -= n;
|
|
rb_yield(UINT2NUM(c));
|
|
}
|
|
else if (MBCLEN_INVALID_P(r)) {
|
|
invalid:
|
|
rb_raise(rb_eArgError, "invalid byte sequence in %s", rb_enc_name(enc));
|
|
}
|
|
else if (MBCLEN_NEEDMORE_P(r)) {
|
|
char cbuf[8], *p = cbuf;
|
|
int more = MBCLEN_NEEDMORE_LEN(r);
|
|
if (more > numberof(cbuf)) goto invalid;
|
|
more += n = fptr->rbuf.len;
|
|
if (more > numberof(cbuf)) goto invalid;
|
|
while ((n = (int)read_buffered_data(p, more, fptr)) > 0 &&
|
|
(p += n, (more -= n) > 0)) {
|
|
if (io_fillbuf(fptr) < 0) goto invalid;
|
|
if ((n = fptr->rbuf.len) > more) n = more;
|
|
}
|
|
r = rb_enc_precise_mbclen(cbuf, p, enc);
|
|
if (!MBCLEN_CHARFOUND_P(r)) goto invalid;
|
|
c = rb_enc_codepoint(cbuf, p, enc);
|
|
rb_yield(UINT2NUM(c));
|
|
}
|
|
else {
|
|
continue;
|
|
}
|
|
}
|
|
return io;
|
|
}
|
|
|
|
/*
|
|
* This is a deprecated alias for #each_codepoint.
|
|
*/
|
|
|
|
static VALUE
|
|
rb_io_codepoints(VALUE io)
|
|
{
|
|
rb_warn("IO#codepoints is deprecated; use #each_codepoint instead");
|
|
if (!rb_block_given_p())
|
|
return rb_enumeratorize(io, ID2SYM(rb_intern("each_codepoint")), 0, 0);
|
|
return rb_io_each_codepoint(io);
|
|
}
|
|
|
|
|
|
/*
|
|
* call-seq:
|
|
* ios.getc -> string or nil
|
|
*
|
|
* Reads a one-character string from <em>ios</em>. Returns
|
|
* +nil+ if called at end of file.
|
|
*
|
|
* f = File.new("testfile")
|
|
* f.getc #=> "h"
|
|
* f.getc #=> "e"
|
|
*/
|
|
|
|
static VALUE
|
|
rb_io_getc(VALUE io)
|
|
{
|
|
rb_io_t *fptr;
|
|
rb_encoding *enc;
|
|
|
|
GetOpenFile(io, fptr);
|
|
rb_io_check_char_readable(fptr);
|
|
|
|
enc = io_input_encoding(fptr);
|
|
READ_CHECK(fptr);
|
|
return io_getc(fptr, enc);
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* ios.readchar -> string
|
|
*
|
|
* Reads a one-character string from <em>ios</em>. Raises an
|
|
* EOFError on end of file.
|
|
*
|
|
* f = File.new("testfile")
|
|
* f.readchar #=> "h"
|
|
* f.readchar #=> "e"
|
|
*/
|
|
|
|
static VALUE
|
|
rb_io_readchar(VALUE io)
|
|
{
|
|
VALUE c = rb_io_getc(io);
|
|
|
|
if (NIL_P(c)) {
|
|
rb_eof_error();
|
|
}
|
|
return c;
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* ios.getbyte -> integer or nil
|
|
*
|
|
* Gets the next 8-bit byte (0..255) from <em>ios</em>. Returns
|
|
* +nil+ if called at end of file.
|
|
*
|
|
* f = File.new("testfile")
|
|
* f.getbyte #=> 84
|
|
* f.getbyte #=> 104
|
|
*/
|
|
|
|
VALUE
|
|
rb_io_getbyte(VALUE io)
|
|
{
|
|
rb_io_t *fptr;
|
|
int c;
|
|
|
|
GetOpenFile(io, fptr);
|
|
rb_io_check_byte_readable(fptr);
|
|
READ_CHECK(fptr);
|
|
if (fptr->fd == 0 && (fptr->mode & FMODE_TTY) && RB_TYPE_P(rb_stdout, T_FILE)) {
|
|
rb_io_t *ofp;
|
|
GetOpenFile(rb_stdout, ofp);
|
|
if (ofp->mode & FMODE_TTY) {
|
|
rb_io_flush(rb_stdout);
|
|
}
|
|
}
|
|
if (io_fillbuf(fptr) < 0) {
|
|
return Qnil;
|
|
}
|
|
fptr->rbuf.off++;
|
|
fptr->rbuf.len--;
|
|
c = (unsigned char)fptr->rbuf.ptr[fptr->rbuf.off-1];
|
|
return INT2FIX(c & 0xff);
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* ios.readbyte -> integer
|
|
*
|
|
* Reads a byte as with IO#getbyte, but raises an EOFError on end of
|
|
* file.
|
|
*/
|
|
|
|
static VALUE
|
|
rb_io_readbyte(VALUE io)
|
|
{
|
|
VALUE c = rb_io_getbyte(io);
|
|
|
|
if (NIL_P(c)) {
|
|
rb_eof_error();
|
|
}
|
|
return c;
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* ios.ungetbyte(string) -> nil
|
|
* ios.ungetbyte(integer) -> nil
|
|
*
|
|
* Pushes back bytes (passed as a parameter) onto <em>ios</em>,
|
|
* such that a subsequent buffered read will return it. Only one byte
|
|
* may be pushed back before a subsequent read operation (that is,
|
|
* you will be able to read only the last of several bytes that have been pushed
|
|
* back). Has no effect with unbuffered reads (such as IO#sysread).
|
|
*
|
|
* f = File.new("testfile") #=> #<File:testfile>
|
|
* b = f.getbyte #=> 0x38
|
|
* f.ungetbyte(b) #=> nil
|
|
* f.getbyte #=> 0x38
|
|
*/
|
|
|
|
VALUE
|
|
rb_io_ungetbyte(VALUE io, VALUE b)
|
|
{
|
|
rb_io_t *fptr;
|
|
|
|
GetOpenFile(io, fptr);
|
|
rb_io_check_byte_readable(fptr);
|
|
switch (TYPE(b)) {
|
|
case T_NIL:
|
|
return Qnil;
|
|
case T_FIXNUM:
|
|
case T_BIGNUM: ;
|
|
VALUE v = rb_int_modulo(b, INT2FIX(256));
|
|
unsigned char c = NUM2INT(v) & 0xFF;
|
|
b = rb_str_new((const char *)&c, 1);
|
|
break;
|
|
default:
|
|
SafeStringValue(b);
|
|
}
|
|
io_ungetbyte(b, fptr);
|
|
return Qnil;
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* ios.ungetc(string) -> nil
|
|
*
|
|
* Pushes back one character (passed as a parameter) onto <em>ios</em>,
|
|
* such that a subsequent buffered character read will return it. Only one character
|
|
* may be pushed back before a subsequent read operation (that is,
|
|
* you will be able to read only the last of several characters that have been pushed
|
|
* back). Has no effect with unbuffered reads (such as IO#sysread).
|
|
*
|
|
* f = File.new("testfile") #=> #<File:testfile>
|
|
* c = f.getc #=> "8"
|
|
* f.ungetc(c) #=> nil
|
|
* f.getc #=> "8"
|
|
*/
|
|
|
|
VALUE
|
|
rb_io_ungetc(VALUE io, VALUE c)
|
|
{
|
|
rb_io_t *fptr;
|
|
long len;
|
|
|
|
GetOpenFile(io, fptr);
|
|
rb_io_check_char_readable(fptr);
|
|
if (NIL_P(c)) return Qnil;
|
|
if (FIXNUM_P(c)) {
|
|
c = rb_enc_uint_chr(FIX2UINT(c), io_read_encoding(fptr));
|
|
}
|
|
else if (RB_TYPE_P(c, T_BIGNUM)) {
|
|
c = rb_enc_uint_chr(NUM2UINT(c), io_read_encoding(fptr));
|
|
}
|
|
else {
|
|
SafeStringValue(c);
|
|
}
|
|
if (NEED_READCONV(fptr)) {
|
|
SET_BINARY_MODE(fptr);
|
|
len = RSTRING_LEN(c);
|
|
#if SIZEOF_LONG > SIZEOF_INT
|
|
if (len > INT_MAX)
|
|
rb_raise(rb_eIOError, "ungetc failed");
|
|
#endif
|
|
make_readconv(fptr, (int)len);
|
|
if (fptr->cbuf.capa - fptr->cbuf.len < len)
|
|
rb_raise(rb_eIOError, "ungetc failed");
|
|
if (fptr->cbuf.off < len) {
|
|
MEMMOVE(fptr->cbuf.ptr+fptr->cbuf.capa-fptr->cbuf.len,
|
|
fptr->cbuf.ptr+fptr->cbuf.off,
|
|
char, fptr->cbuf.len);
|
|
fptr->cbuf.off = fptr->cbuf.capa-fptr->cbuf.len;
|
|
}
|
|
fptr->cbuf.off -= (int)len;
|
|
fptr->cbuf.len += (int)len;
|
|
MEMMOVE(fptr->cbuf.ptr+fptr->cbuf.off, RSTRING_PTR(c), char, len);
|
|
}
|
|
else {
|
|
NEED_NEWLINE_DECORATOR_ON_READ_CHECK(fptr);
|
|
io_ungetbyte(c, fptr);
|
|
}
|
|
return Qnil;
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* ios.isatty -> true or false
|
|
* ios.tty? -> true or false
|
|
*
|
|
* Returns <code>true</code> if <em>ios</em> is associated with a
|
|
* terminal device (tty), <code>false</code> otherwise.
|
|
*
|
|
* File.new("testfile").isatty #=> false
|
|
* File.new("/dev/tty").isatty #=> true
|
|
*/
|
|
|
|
static VALUE
|
|
rb_io_isatty(VALUE io)
|
|
{
|
|
rb_io_t *fptr;
|
|
|
|
GetOpenFile(io, fptr);
|
|
if (isatty(fptr->fd) == 0)
|
|
return Qfalse;
|
|
return Qtrue;
|
|
}
|
|
|
|
#if defined(HAVE_FCNTL) && defined(F_GETFD) && defined(F_SETFD) && defined(FD_CLOEXEC)
|
|
/*
|
|
* call-seq:
|
|
* ios.close_on_exec? -> true or false
|
|
*
|
|
* Returns <code>true</code> if <em>ios</em> will be closed on exec.
|
|
*
|
|
* f = open("/dev/null")
|
|
* f.close_on_exec? #=> false
|
|
* f.close_on_exec = true
|
|
* f.close_on_exec? #=> true
|
|
* f.close_on_exec = false
|
|
* f.close_on_exec? #=> false
|
|
*/
|
|
|
|
static VALUE
|
|
rb_io_close_on_exec_p(VALUE io)
|
|
{
|
|
rb_io_t *fptr;
|
|
VALUE write_io;
|
|
int fd, ret;
|
|
|
|
write_io = GetWriteIO(io);
|
|
if (io != write_io) {
|
|
GetOpenFile(write_io, fptr);
|
|
if (fptr && 0 <= (fd = fptr->fd)) {
|
|
if ((ret = fcntl(fd, F_GETFD)) == -1) rb_sys_fail_path(fptr->pathv);
|
|
if (!(ret & FD_CLOEXEC)) return Qfalse;
|
|
}
|
|
}
|
|
|
|
GetOpenFile(io, fptr);
|
|
if (fptr && 0 <= (fd = fptr->fd)) {
|
|
if ((ret = fcntl(fd, F_GETFD)) == -1) rb_sys_fail_path(fptr->pathv);
|
|
if (!(ret & FD_CLOEXEC)) return Qfalse;
|
|
}
|
|
return Qtrue;
|
|
}
|
|
#else
|
|
#define rb_io_close_on_exec_p rb_f_notimplement
|
|
#endif
|
|
|
|
#if defined(HAVE_FCNTL) && defined(F_GETFD) && defined(F_SETFD) && defined(FD_CLOEXEC)
|
|
/*
|
|
* call-seq:
|
|
* ios.close_on_exec = bool -> true or false
|
|
*
|
|
* Sets a close-on-exec flag.
|
|
*
|
|
* f = open("/dev/null")
|
|
* f.close_on_exec = true
|
|
* system("cat", "/proc/self/fd/#{f.fileno}") # cat: /proc/self/fd/3: No such file or directory
|
|
* f.closed? #=> false
|
|
*
|
|
* Ruby sets close-on-exec flags of all file descriptors by default
|
|
* since Ruby 2.0.0.
|
|
* So you don't need to set by yourself.
|
|
* Also, unsetting a close-on-exec flag can cause file descriptor leak
|
|
* if another thread use fork() and exec() (via system() method for example).
|
|
* If you really needs file descriptor inheritance to child process,
|
|
* use spawn()'s argument such as fd=>fd.
|
|
*/
|
|
|
|
static VALUE
|
|
rb_io_set_close_on_exec(VALUE io, VALUE arg)
|
|
{
|
|
int flag = RTEST(arg) ? FD_CLOEXEC : 0;
|
|
rb_io_t *fptr;
|
|
VALUE write_io;
|
|
int fd, ret;
|
|
|
|
write_io = GetWriteIO(io);
|
|
if (io != write_io) {
|
|
GetOpenFile(write_io, fptr);
|
|
if (fptr && 0 <= (fd = fptr->fd)) {
|
|
if ((ret = fcntl(fptr->fd, F_GETFD)) == -1) rb_sys_fail_path(fptr->pathv);
|
|
if ((ret & FD_CLOEXEC) != flag) {
|
|
ret = (ret & ~FD_CLOEXEC) | flag;
|
|
ret = fcntl(fd, F_SETFD, ret);
|
|
if (ret != 0) rb_sys_fail_path(fptr->pathv);
|
|
}
|
|
}
|
|
|
|
}
|
|
|
|
GetOpenFile(io, fptr);
|
|
if (fptr && 0 <= (fd = fptr->fd)) {
|
|
if ((ret = fcntl(fd, F_GETFD)) == -1) rb_sys_fail_path(fptr->pathv);
|
|
if ((ret & FD_CLOEXEC) != flag) {
|
|
ret = (ret & ~FD_CLOEXEC) | flag;
|
|
ret = fcntl(fd, F_SETFD, ret);
|
|
if (ret != 0) rb_sys_fail_path(fptr->pathv);
|
|
}
|
|
}
|
|
return Qnil;
|
|
}
|
|
#else
|
|
#define rb_io_set_close_on_exec rb_f_notimplement
|
|
#endif
|
|
|
|
#define FMODE_PREP (1<<16)
|
|
#define IS_PREP_STDIO(f) ((f)->mode & FMODE_PREP)
|
|
#define PREP_STDIO_NAME(f) (RSTRING_PTR((f)->pathv))
|
|
|
|
static VALUE
|
|
finish_writeconv(rb_io_t *fptr, int noalloc)
|
|
{
|
|
unsigned char *ds, *dp, *de;
|
|
rb_econv_result_t res;
|
|
|
|
if (!fptr->wbuf.ptr) {
|
|
unsigned char buf[1024];
|
|
long r;
|
|
|
|
res = econv_destination_buffer_full;
|
|
while (res == econv_destination_buffer_full) {
|
|
ds = dp = buf;
|
|
de = buf + sizeof(buf);
|
|
res = rb_econv_convert(fptr->writeconv, NULL, NULL, &dp, de, 0);
|
|
while (dp-ds) {
|
|
retry:
|
|
if (fptr->write_lock && rb_mutex_owned_p(fptr->write_lock))
|
|
r = rb_write_internal2(fptr->fd, ds, dp-ds);
|
|
else
|
|
r = rb_write_internal(fptr->fd, ds, dp-ds);
|
|
if (r == dp-ds)
|
|
break;
|
|
if (0 <= r) {
|
|
ds += r;
|
|
}
|
|
if (rb_io_wait_writable(fptr->fd)) {
|
|
if (fptr->fd < 0)
|
|
return noalloc ? Qtrue : rb_exc_new3(rb_eIOError, rb_str_new_cstr(closed_stream));
|
|
goto retry;
|
|
}
|
|
return noalloc ? Qtrue : INT2NUM(errno);
|
|
}
|
|
if (res == econv_invalid_byte_sequence ||
|
|
res == econv_incomplete_input ||
|
|
res == econv_undefined_conversion) {
|
|
return noalloc ? Qtrue : rb_econv_make_exception(fptr->writeconv);
|
|
}
|
|
}
|
|
|
|
return Qnil;
|
|
}
|
|
|
|
res = econv_destination_buffer_full;
|
|
while (res == econv_destination_buffer_full) {
|
|
if (fptr->wbuf.len == fptr->wbuf.capa) {
|
|
if (io_fflush(fptr) < 0)
|
|
return noalloc ? Qtrue : INT2NUM(errno);
|
|
}
|
|
|
|
ds = dp = (unsigned char *)fptr->wbuf.ptr + fptr->wbuf.off + fptr->wbuf.len;
|
|
de = (unsigned char *)fptr->wbuf.ptr + fptr->wbuf.capa;
|
|
res = rb_econv_convert(fptr->writeconv, NULL, NULL, &dp, de, 0);
|
|
fptr->wbuf.len += (int)(dp - ds);
|
|
if (res == econv_invalid_byte_sequence ||
|
|
res == econv_incomplete_input ||
|
|
res == econv_undefined_conversion) {
|
|
return noalloc ? Qtrue : rb_econv_make_exception(fptr->writeconv);
|
|
}
|
|
}
|
|
return Qnil;
|
|
}
|
|
|
|
struct finish_writeconv_arg {
|
|
rb_io_t *fptr;
|
|
int noalloc;
|
|
};
|
|
|
|
static VALUE
|
|
finish_writeconv_sync(VALUE arg)
|
|
{
|
|
struct finish_writeconv_arg *p = (struct finish_writeconv_arg *)arg;
|
|
return finish_writeconv(p->fptr, p->noalloc);
|
|
}
|
|
|
|
static void*
|
|
nogvl_close(void *ptr)
|
|
{
|
|
int *fd = ptr;
|
|
|
|
return (void*)(intptr_t)close(*fd);
|
|
}
|
|
|
|
static int
|
|
maygvl_close(int fd, int keepgvl)
|
|
{
|
|
if (keepgvl)
|
|
return close(fd);
|
|
|
|
/*
|
|
* close() may block for certain file types (NFS, SO_LINGER sockets,
|
|
* inotify), so let other threads run.
|
|
*/
|
|
return (int)(intptr_t)rb_thread_call_without_gvl(nogvl_close, &fd, RUBY_UBF_IO, 0);
|
|
}
|
|
|
|
static void*
|
|
nogvl_fclose(void *ptr)
|
|
{
|
|
FILE *file = ptr;
|
|
|
|
return (void*)(intptr_t)fclose(file);
|
|
}
|
|
|
|
static int
|
|
maygvl_fclose(FILE *file, int keepgvl)
|
|
{
|
|
if (keepgvl)
|
|
return fclose(file);
|
|
|
|
return (int)(intptr_t)rb_thread_call_without_gvl(nogvl_fclose, file, RUBY_UBF_IO, 0);
|
|
}
|
|
|
|
static void free_io_buffer(rb_io_buffer_t *buf);
|
|
static void clear_codeconv(rb_io_t *fptr);
|
|
|
|
static void
|
|
fptr_finalize_flush(rb_io_t *fptr, int noraise, int keepgvl,
|
|
struct list_head *busy)
|
|
{
|
|
VALUE err = Qnil;
|
|
int fd = fptr->fd;
|
|
FILE *stdio_file = fptr->stdio_file;
|
|
int mode = fptr->mode;
|
|
|
|
if (fptr->writeconv) {
|
|
if (fptr->write_lock && !noraise) {
|
|
struct finish_writeconv_arg arg;
|
|
arg.fptr = fptr;
|
|
arg.noalloc = noraise;
|
|
err = rb_mutex_synchronize(fptr->write_lock, finish_writeconv_sync, (VALUE)&arg);
|
|
}
|
|
else {
|
|
err = finish_writeconv(fptr, noraise);
|
|
}
|
|
}
|
|
if (fptr->wbuf.len) {
|
|
if (noraise) {
|
|
io_flush_buffer_sync(fptr);
|
|
}
|
|
else {
|
|
if (io_fflush(fptr) < 0 && NIL_P(err))
|
|
err = INT2NUM(errno);
|
|
}
|
|
}
|
|
|
|
fptr->fd = -1;
|
|
fptr->stdio_file = 0;
|
|
fptr->mode &= ~(FMODE_READABLE|FMODE_WRITABLE);
|
|
|
|
/*
|
|
* ensure waiting_fd users do not hit EBADF, wait for them
|
|
* to exit before we call close().
|
|
*/
|
|
if (busy) {
|
|
do rb_thread_schedule(); while (!list_empty(busy));
|
|
}
|
|
|
|
if (IS_PREP_STDIO(fptr) || fd <= 2) {
|
|
/* need to keep FILE objects of stdin, stdout and stderr */
|
|
}
|
|
else if (stdio_file) {
|
|
/* stdio_file is deallocated anyway
|
|
* even if fclose failed. */
|
|
if ((maygvl_fclose(stdio_file, noraise) < 0) && NIL_P(err))
|
|
if (!noraise) err = INT2NUM(errno);
|
|
}
|
|
else if (0 <= fd) {
|
|
/* fptr->fd may be closed even if close fails.
|
|
* POSIX doesn't specify it.
|
|
* We assumes it is closed. */
|
|
|
|
/**/
|
|
keepgvl |= !(mode & FMODE_WRITABLE);
|
|
keepgvl |= noraise;
|
|
if ((maygvl_close(fd, keepgvl) < 0) && NIL_P(err))
|
|
if (!noraise) err = INT2NUM(errno);
|
|
}
|
|
|
|
if (!NIL_P(err) && !noraise) {
|
|
if (RB_INTEGER_TYPE_P(err))
|
|
rb_syserr_fail_path(NUM2INT(err), fptr->pathv);
|
|
else
|
|
rb_exc_raise(err);
|
|
}
|
|
}
|
|
|
|
static void
|
|
fptr_finalize(rb_io_t *fptr, int noraise)
|
|
{
|
|
fptr_finalize_flush(fptr, noraise, FALSE, 0);
|
|
free_io_buffer(&fptr->rbuf);
|
|
free_io_buffer(&fptr->wbuf);
|
|
clear_codeconv(fptr);
|
|
}
|
|
|
|
static void
|
|
rb_io_fptr_cleanup(rb_io_t *fptr, int noraise)
|
|
{
|
|
if (fptr->finalize) {
|
|
(*fptr->finalize)(fptr, noraise);
|
|
}
|
|
else {
|
|
fptr_finalize(fptr, noraise);
|
|
}
|
|
}
|
|
|
|
static void
|
|
free_io_buffer(rb_io_buffer_t *buf)
|
|
{
|
|
if (buf->ptr) {
|
|
ruby_sized_xfree(buf->ptr, (size_t)buf->capa);
|
|
buf->ptr = NULL;
|
|
}
|
|
}
|
|
|
|
static void
|
|
clear_readconv(rb_io_t *fptr)
|
|
{
|
|
if (fptr->readconv) {
|
|
rb_econv_close(fptr->readconv);
|
|
fptr->readconv = NULL;
|
|
}
|
|
free_io_buffer(&fptr->cbuf);
|
|
}
|
|
|
|
static void
|
|
clear_writeconv(rb_io_t *fptr)
|
|
{
|
|
if (fptr->writeconv) {
|
|
rb_econv_close(fptr->writeconv);
|
|
fptr->writeconv = NULL;
|
|
}
|
|
fptr->writeconv_initialized = 0;
|
|
}
|
|
|
|
static void
|
|
clear_codeconv(rb_io_t *fptr)
|
|
{
|
|
clear_readconv(fptr);
|
|
clear_writeconv(fptr);
|
|
}
|
|
|
|
void
|
|
rb_io_fptr_finalize_internal(void *ptr)
|
|
{
|
|
rb_io_t *fptr = ptr;
|
|
|
|
if (!ptr) return;
|
|
fptr->pathv = Qnil;
|
|
if (0 <= fptr->fd)
|
|
rb_io_fptr_cleanup(fptr, TRUE);
|
|
fptr->write_lock = 0;
|
|
free_io_buffer(&fptr->rbuf);
|
|
free_io_buffer(&fptr->wbuf);
|
|
clear_codeconv(fptr);
|
|
free(fptr);
|
|
}
|
|
|
|
#undef rb_io_fptr_finalize
|
|
int
|
|
rb_io_fptr_finalize(rb_io_t *fptr)
|
|
{
|
|
if (!fptr) {
|
|
return 0;
|
|
}
|
|
else {
|
|
rb_io_fptr_finalize_internal(fptr);
|
|
return 1;
|
|
}
|
|
}
|
|
#define rb_io_fptr_finalize(fptr) rb_io_fptr_finalize_internal(fptr)
|
|
|
|
RUBY_FUNC_EXPORTED size_t
|
|
rb_io_memsize(const rb_io_t *fptr)
|
|
{
|
|
size_t size = sizeof(rb_io_t);
|
|
size += fptr->rbuf.capa;
|
|
size += fptr->wbuf.capa;
|
|
size += fptr->cbuf.capa;
|
|
if (fptr->readconv) size += rb_econv_memsize(fptr->readconv);
|
|
if (fptr->writeconv) size += rb_econv_memsize(fptr->writeconv);
|
|
return size;
|
|
}
|
|
|
|
#ifdef _WIN32
|
|
/* keep GVL while closing to prevent crash on Windows */
|
|
# define KEEPGVL TRUE
|
|
#else
|
|
# define KEEPGVL FALSE
|
|
#endif
|
|
|
|
int rb_notify_fd_close(int fd, struct list_head *);
|
|
static rb_io_t *
|
|
io_close_fptr(VALUE io)
|
|
{
|
|
rb_io_t *fptr;
|
|
VALUE write_io;
|
|
rb_io_t *write_fptr;
|
|
struct list_head busy;
|
|
|
|
list_head_init(&busy);
|
|
write_io = GetWriteIO(io);
|
|
if (io != write_io) {
|
|
write_fptr = RFILE(write_io)->fptr;
|
|
if (write_fptr && 0 <= write_fptr->fd) {
|
|
rb_io_fptr_cleanup(write_fptr, TRUE);
|
|
}
|
|
}
|
|
|
|
fptr = RFILE(io)->fptr;
|
|
if (!fptr) return 0;
|
|
if (fptr->fd < 0) return 0;
|
|
|
|
if (rb_notify_fd_close(fptr->fd, &busy)) {
|
|
/* calls close(fptr->fd): */
|
|
fptr_finalize_flush(fptr, FALSE, KEEPGVL, &busy);
|
|
}
|
|
rb_io_fptr_cleanup(fptr, FALSE);
|
|
return fptr;
|
|
}
|
|
|
|
static void
|
|
fptr_waitpid(rb_io_t *fptr, int nohang)
|
|
{
|
|
int status;
|
|
if (fptr->pid) {
|
|
rb_last_status_clear();
|
|
rb_waitpid(fptr->pid, &status, nohang ? WNOHANG : 0);
|
|
fptr->pid = 0;
|
|
}
|
|
}
|
|
|
|
VALUE
|
|
rb_io_close(VALUE io)
|
|
{
|
|
rb_io_t *fptr = io_close_fptr(io);
|
|
if (fptr) fptr_waitpid(fptr, 0);
|
|
return Qnil;
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* ios.close -> nil
|
|
*
|
|
* Closes <em>ios</em> and flushes any pending writes to the operating
|
|
* system. The stream is unavailable for any further data operations;
|
|
* an IOError is raised if such an attempt is made. I/O streams are
|
|
* automatically closed when they are claimed by the garbage collector.
|
|
*
|
|
* If <em>ios</em> is opened by IO.popen, #close sets
|
|
* <code>$?</code>.
|
|
*
|
|
* Calling this method on closed IO object is just ignored since Ruby 2.3.
|
|
*/
|
|
|
|
static VALUE
|
|
rb_io_close_m(VALUE io)
|
|
{
|
|
rb_io_t *fptr = rb_io_get_fptr(io);
|
|
if (fptr->fd < 0) {
|
|
return Qnil;
|
|
}
|
|
rb_io_close(io);
|
|
return Qnil;
|
|
}
|
|
|
|
static VALUE
|
|
io_call_close(VALUE io)
|
|
{
|
|
rb_check_funcall(io, rb_intern("close"), 0, 0);
|
|
return io;
|
|
}
|
|
|
|
static VALUE
|
|
ignore_closed_stream(VALUE io, VALUE exc)
|
|
{
|
|
enum {mesg_len = sizeof(closed_stream)-1};
|
|
VALUE mesg = rb_attr_get(exc, idMesg);
|
|
if (!RB_TYPE_P(mesg, T_STRING) ||
|
|
RSTRING_LEN(mesg) != mesg_len ||
|
|
memcmp(RSTRING_PTR(mesg), closed_stream, mesg_len)) {
|
|
rb_exc_raise(exc);
|
|
}
|
|
return io;
|
|
}
|
|
|
|
static VALUE
|
|
io_close(VALUE io)
|
|
{
|
|
VALUE closed = rb_check_funcall(io, rb_intern("closed?"), 0, 0);
|
|
if (closed != Qundef && RTEST(closed)) return io;
|
|
rb_rescue2(io_call_close, io, ignore_closed_stream, io,
|
|
rb_eIOError, (VALUE)0);
|
|
return io;
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* ios.closed? -> true or false
|
|
*
|
|
* Returns <code>true</code> if <em>ios</em> is completely closed (for
|
|
* duplex streams, both reader and writer), <code>false</code>
|
|
* otherwise.
|
|
*
|
|
* f = File.new("testfile")
|
|
* f.close #=> nil
|
|
* f.closed? #=> true
|
|
* f = IO.popen("/bin/sh","r+")
|
|
* f.close_write #=> nil
|
|
* f.closed? #=> false
|
|
* f.close_read #=> nil
|
|
* f.closed? #=> true
|
|
*/
|
|
|
|
|
|
static VALUE
|
|
rb_io_closed(VALUE io)
|
|
{
|
|
rb_io_t *fptr;
|
|
VALUE write_io;
|
|
rb_io_t *write_fptr;
|
|
|
|
write_io = GetWriteIO(io);
|
|
if (io != write_io) {
|
|
write_fptr = RFILE(write_io)->fptr;
|
|
if (write_fptr && 0 <= write_fptr->fd) {
|
|
return Qfalse;
|
|
}
|
|
}
|
|
|
|
fptr = rb_io_get_fptr(io);
|
|
return 0 <= fptr->fd ? Qfalse : Qtrue;
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* ios.close_read -> nil
|
|
*
|
|
* Closes the read end of a duplex I/O stream (i.e., one that contains
|
|
* both a read and a write stream, such as a pipe). Will raise an
|
|
* IOError if the stream is not duplexed.
|
|
*
|
|
* f = IO.popen("/bin/sh","r+")
|
|
* f.close_read
|
|
* f.readlines
|
|
*
|
|
* <em>produces:</em>
|
|
*
|
|
* prog.rb:3:in `readlines': not opened for reading (IOError)
|
|
* from prog.rb:3
|
|
*
|
|
* Calling this method on closed IO object is just ignored since Ruby 2.3.
|
|
*/
|
|
|
|
static VALUE
|
|
rb_io_close_read(VALUE io)
|
|
{
|
|
rb_io_t *fptr;
|
|
VALUE write_io;
|
|
|
|
fptr = rb_io_get_fptr(rb_io_taint_check(io));
|
|
if (fptr->fd < 0) return Qnil;
|
|
if (is_socket(fptr->fd, fptr->pathv)) {
|
|
#ifndef SHUT_RD
|
|
# define SHUT_RD 0
|
|
#endif
|
|
if (shutdown(fptr->fd, SHUT_RD) < 0)
|
|
rb_sys_fail_path(fptr->pathv);
|
|
fptr->mode &= ~FMODE_READABLE;
|
|
if (!(fptr->mode & FMODE_WRITABLE))
|
|
return rb_io_close(io);
|
|
return Qnil;
|
|
}
|
|
|
|
write_io = GetWriteIO(io);
|
|
if (io != write_io) {
|
|
rb_io_t *wfptr;
|
|
wfptr = rb_io_get_fptr(rb_io_taint_check(write_io));
|
|
wfptr->pid = fptr->pid;
|
|
fptr->pid = 0;
|
|
RFILE(io)->fptr = wfptr;
|
|
/* bind to write_io temporarily to get rid of memory/fd leak */
|
|
fptr->tied_io_for_writing = 0;
|
|
RFILE(write_io)->fptr = fptr;
|
|
rb_io_fptr_cleanup(fptr, FALSE);
|
|
/* should not finalize fptr because another thread may be reading it */
|
|
return Qnil;
|
|
}
|
|
|
|
if ((fptr->mode & (FMODE_DUPLEX|FMODE_WRITABLE)) == FMODE_WRITABLE) {
|
|
rb_raise(rb_eIOError, "closing non-duplex IO for reading");
|
|
}
|
|
return rb_io_close(io);
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* ios.close_write -> nil
|
|
*
|
|
* Closes the write end of a duplex I/O stream (i.e., one that contains
|
|
* both a read and a write stream, such as a pipe). Will raise an
|
|
* IOError if the stream is not duplexed.
|
|
*
|
|
* f = IO.popen("/bin/sh","r+")
|
|
* f.close_write
|
|
* f.print "nowhere"
|
|
*
|
|
* <em>produces:</em>
|
|
*
|
|
* prog.rb:3:in `write': not opened for writing (IOError)
|
|
* from prog.rb:3:in `print'
|
|
* from prog.rb:3
|
|
*
|
|
* Calling this method on closed IO object is just ignored since Ruby 2.3.
|
|
*/
|
|
|
|
static VALUE
|
|
rb_io_close_write(VALUE io)
|
|
{
|
|
rb_io_t *fptr;
|
|
VALUE write_io;
|
|
|
|
write_io = GetWriteIO(io);
|
|
fptr = rb_io_get_fptr(rb_io_taint_check(write_io));
|
|
if (fptr->fd < 0) return Qnil;
|
|
if (is_socket(fptr->fd, fptr->pathv)) {
|
|
#ifndef SHUT_WR
|
|
# define SHUT_WR 1
|
|
#endif
|
|
if (shutdown(fptr->fd, SHUT_WR) < 0)
|
|
rb_sys_fail_path(fptr->pathv);
|
|
fptr->mode &= ~FMODE_WRITABLE;
|
|
if (!(fptr->mode & FMODE_READABLE))
|
|
return rb_io_close(write_io);
|
|
return Qnil;
|
|
}
|
|
|
|
if ((fptr->mode & (FMODE_DUPLEX|FMODE_READABLE)) == FMODE_READABLE) {
|
|
rb_raise(rb_eIOError, "closing non-duplex IO for writing");
|
|
}
|
|
|
|
if (io != write_io) {
|
|
fptr = rb_io_get_fptr(rb_io_taint_check(io));
|
|
fptr->tied_io_for_writing = 0;
|
|
}
|
|
rb_io_close(write_io);
|
|
return Qnil;
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* ios.sysseek(offset, whence=IO::SEEK_SET) -> integer
|
|
*
|
|
* Seeks to a given <i>offset</i> in the stream according to the value
|
|
* of <i>whence</i> (see IO#seek for values of <i>whence</i>). Returns
|
|
* the new offset into the file.
|
|
*
|
|
* f = File.new("testfile")
|
|
* f.sysseek(-13, IO::SEEK_END) #=> 53
|
|
* f.sysread(10) #=> "And so on."
|
|
*/
|
|
|
|
static VALUE
|
|
rb_io_sysseek(int argc, VALUE *argv, VALUE io)
|
|
{
|
|
VALUE offset, ptrname;
|
|
int whence = SEEK_SET;
|
|
rb_io_t *fptr;
|
|
off_t pos;
|
|
|
|
if (rb_scan_args(argc, argv, "11", &offset, &ptrname) == 2) {
|
|
whence = interpret_seek_whence(ptrname);
|
|
}
|
|
pos = NUM2OFFT(offset);
|
|
GetOpenFile(io, fptr);
|
|
if ((fptr->mode & FMODE_READABLE) &&
|
|
(READ_DATA_BUFFERED(fptr) || READ_CHAR_PENDING(fptr))) {
|
|
rb_raise(rb_eIOError, "sysseek for buffered IO");
|
|
}
|
|
if ((fptr->mode & FMODE_WRITABLE) && fptr->wbuf.len) {
|
|
rb_warn("sysseek for buffered IO");
|
|
}
|
|
errno = 0;
|
|
pos = lseek(fptr->fd, pos, whence);
|
|
if (pos < 0 && errno) rb_sys_fail_path(fptr->pathv);
|
|
|
|
return OFFT2NUM(pos);
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* ios.syswrite(string) -> integer
|
|
*
|
|
* Writes the given string to <em>ios</em> using a low-level write.
|
|
* Returns the number of bytes written. Do not mix with other methods
|
|
* that write to <em>ios</em> or you may get unpredictable results.
|
|
* Raises SystemCallError on error.
|
|
*
|
|
* f = File.new("out", "w")
|
|
* f.syswrite("ABCDEF") #=> 6
|
|
*/
|
|
|
|
static VALUE
|
|
rb_io_syswrite(VALUE io, VALUE str)
|
|
{
|
|
VALUE tmp;
|
|
rb_io_t *fptr;
|
|
long n, len;
|
|
const char *ptr;
|
|
|
|
if (!RB_TYPE_P(str, T_STRING))
|
|
str = rb_obj_as_string(str);
|
|
|
|
io = GetWriteIO(io);
|
|
GetOpenFile(io, fptr);
|
|
rb_io_check_writable(fptr);
|
|
|
|
if (fptr->wbuf.len) {
|
|
rb_warn("syswrite for buffered IO");
|
|
}
|
|
|
|
tmp = rb_str_tmp_frozen_acquire(str);
|
|
RSTRING_GETMEM(tmp, ptr, len);
|
|
n = rb_write_internal(fptr->fd, ptr, len);
|
|
if (n < 0) rb_sys_fail_path(fptr->pathv);
|
|
rb_str_tmp_frozen_release(str, tmp);
|
|
|
|
return LONG2FIX(n);
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* ios.sysread(maxlen[, outbuf]) -> string
|
|
*
|
|
* Reads <i>maxlen</i> bytes from <em>ios</em> using a low-level
|
|
* read and returns them as a string. Do not mix with other methods
|
|
* that read from <em>ios</em> or you may get unpredictable results.
|
|
*
|
|
* If the optional _outbuf_ argument is present,
|
|
* it must reference a String, which will receive the data.
|
|
* The _outbuf_ will contain only the received data after the method call
|
|
* even if it is not empty at the beginning.
|
|
*
|
|
* Raises SystemCallError on error and EOFError at end of file.
|
|
*
|
|
* f = File.new("testfile")
|
|
* f.sysread(16) #=> "This is line one"
|
|
*/
|
|
|
|
static VALUE
|
|
rb_io_sysread(int argc, VALUE *argv, VALUE io)
|
|
{
|
|
VALUE len, str;
|
|
rb_io_t *fptr;
|
|
long n, ilen;
|
|
struct io_internal_read_struct iis;
|
|
int shrinkable;
|
|
|
|
rb_scan_args(argc, argv, "11", &len, &str);
|
|
ilen = NUM2LONG(len);
|
|
|
|
shrinkable = io_setstrbuf(&str, ilen);
|
|
if (ilen == 0) return str;
|
|
|
|
GetOpenFile(io, fptr);
|
|
rb_io_check_byte_readable(fptr);
|
|
|
|
if (READ_DATA_BUFFERED(fptr)) {
|
|
rb_raise(rb_eIOError, "sysread for buffered IO");
|
|
}
|
|
|
|
/*
|
|
* FIXME: removing rb_thread_wait_fd() here changes sysread semantics
|
|
* on non-blocking IOs. However, it's still currently possible
|
|
* for sysread to raise Errno::EAGAIN if another thread read()s
|
|
* the IO after we return from rb_thread_wait_fd() but before
|
|
* we call read()
|
|
*/
|
|
rb_thread_wait_fd(fptr->fd);
|
|
|
|
rb_io_check_closed(fptr);
|
|
|
|
io_setstrbuf(&str, ilen);
|
|
iis.fd = fptr->fd;
|
|
iis.nonblock = 1; /* for historical reasons, maybe (see above) */
|
|
iis.buf = RSTRING_PTR(str);
|
|
iis.capa = ilen;
|
|
n = read_internal_locktmp(str, &iis);
|
|
|
|
if (n < 0) {
|
|
rb_sys_fail_path(fptr->pathv);
|
|
}
|
|
io_set_read_length(str, n, shrinkable);
|
|
if (n == 0 && ilen > 0) {
|
|
rb_eof_error();
|
|
}
|
|
OBJ_TAINT(str);
|
|
|
|
return str;
|
|
}
|
|
|
|
#if defined(HAVE_PREAD) || defined(HAVE_PWRITE)
|
|
struct prdwr_internal_arg {
|
|
int fd;
|
|
void *buf;
|
|
size_t count;
|
|
off_t offset;
|
|
};
|
|
#endif /* HAVE_PREAD || HAVE_PWRITE */
|
|
|
|
#if defined(HAVE_PREAD)
|
|
static VALUE
|
|
internal_pread_func(void *arg)
|
|
{
|
|
struct prdwr_internal_arg *p = arg;
|
|
return (VALUE)pread(p->fd, p->buf, p->count, p->offset);
|
|
}
|
|
|
|
static VALUE
|
|
pread_internal_call(VALUE arg)
|
|
{
|
|
struct prdwr_internal_arg *p = (struct prdwr_internal_arg *)arg;
|
|
return rb_thread_io_blocking_region(internal_pread_func, p, p->fd);
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* ios.pread(maxlen, offset[, outbuf]) -> string
|
|
*
|
|
* Reads <i>maxlen</i> bytes from <em>ios</em> using the pread system call
|
|
* and returns them as a string without modifying the underlying
|
|
* descriptor offset. This is advantageous compared to combining IO#seek
|
|
* and IO#read in that it is atomic, allowing multiple threads/process to
|
|
* share the same IO object for reading the file at various locations.
|
|
* This bypasses any userspace buffering of the IO layer.
|
|
* If the optional <i>outbuf</i> argument is present, it must
|
|
* reference a String, which will receive the data.
|
|
* Raises SystemCallError on error, EOFError at end of file and
|
|
* NotImplementedError if platform does not implement the system call.
|
|
*
|
|
* File.write("testfile", "This is line one\nThis is line two\n")
|
|
* File.open("testfile") do |f|
|
|
* p f.read # => "This is line one\nThis is line two\n"
|
|
* p f.pread(12, 0) # => "This is line"
|
|
* p f.pread(9, 8) # => "line one\n"
|
|
* end
|
|
*/
|
|
static VALUE
|
|
rb_io_pread(int argc, VALUE *argv, VALUE io)
|
|
{
|
|
VALUE len, offset, str;
|
|
rb_io_t *fptr;
|
|
ssize_t n;
|
|
struct prdwr_internal_arg arg;
|
|
int shrinkable;
|
|
|
|
rb_scan_args(argc, argv, "21", &len, &offset, &str);
|
|
arg.count = NUM2SIZET(len);
|
|
arg.offset = NUM2OFFT(offset);
|
|
|
|
shrinkable = io_setstrbuf(&str, (long)arg.count);
|
|
if (arg.count == 0) return str;
|
|
arg.buf = RSTRING_PTR(str);
|
|
|
|
GetOpenFile(io, fptr);
|
|
rb_io_check_byte_readable(fptr);
|
|
|
|
arg.fd = fptr->fd;
|
|
rb_io_check_closed(fptr);
|
|
|
|
rb_str_locktmp(str);
|
|
n = (ssize_t)rb_ensure(pread_internal_call, (VALUE)&arg, rb_str_unlocktmp, str);
|
|
|
|
if (n < 0) {
|
|
rb_sys_fail_path(fptr->pathv);
|
|
}
|
|
io_set_read_length(str, n, shrinkable);
|
|
if (n == 0 && arg.count > 0) {
|
|
rb_eof_error();
|
|
}
|
|
OBJ_TAINT(str);
|
|
|
|
return str;
|
|
}
|
|
#else
|
|
# define rb_io_pread rb_f_notimplement
|
|
#endif /* HAVE_PREAD */
|
|
|
|
#if defined(HAVE_PWRITE)
|
|
static VALUE
|
|
internal_pwrite_func(void *ptr)
|
|
{
|
|
struct prdwr_internal_arg *arg = ptr;
|
|
|
|
return (VALUE)pwrite(arg->fd, arg->buf, arg->count, arg->offset);
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* ios.pwrite(string, offset) -> integer
|
|
*
|
|
* Writes the given string to <em>ios</em> at <i>offset</i> using pwrite()
|
|
* system call. This is advantageous to combining IO#seek and IO#write
|
|
* in that it is atomic, allowing multiple threads/process to share the
|
|
* same IO object for reading the file at various locations.
|
|
* This bypasses any userspace buffering of the IO layer.
|
|
* Returns the number of bytes written.
|
|
* Raises SystemCallError on error and NotImplementedError
|
|
* if platform does not implement the system call.
|
|
*
|
|
* File.open("out", "w") do |f|
|
|
* f.pwrite("ABCDEF", 3) #=> 6
|
|
* end
|
|
*
|
|
* File.read("out") #=> "\u0000\u0000\u0000ABCDEF"
|
|
*/
|
|
static VALUE
|
|
rb_io_pwrite(VALUE io, VALUE str, VALUE offset)
|
|
{
|
|
rb_io_t *fptr;
|
|
ssize_t n;
|
|
struct prdwr_internal_arg arg;
|
|
VALUE tmp;
|
|
|
|
if (!RB_TYPE_P(str, T_STRING))
|
|
str = rb_obj_as_string(str);
|
|
|
|
arg.offset = NUM2OFFT(offset);
|
|
|
|
io = GetWriteIO(io);
|
|
GetOpenFile(io, fptr);
|
|
rb_io_check_writable(fptr);
|
|
arg.fd = fptr->fd;
|
|
|
|
tmp = rb_str_tmp_frozen_acquire(str);
|
|
arg.buf = RSTRING_PTR(tmp);
|
|
arg.count = (size_t)RSTRING_LEN(tmp);
|
|
|
|
n = (ssize_t)rb_thread_io_blocking_region(internal_pwrite_func, &arg, fptr->fd);
|
|
if (n < 0) rb_sys_fail_path(fptr->pathv);
|
|
rb_str_tmp_frozen_release(str, tmp);
|
|
|
|
return SSIZET2NUM(n);
|
|
}
|
|
#else
|
|
# define rb_io_pwrite rb_f_notimplement
|
|
#endif /* HAVE_PWRITE */
|
|
|
|
VALUE
|
|
rb_io_binmode(VALUE io)
|
|
{
|
|
rb_io_t *fptr;
|
|
|
|
GetOpenFile(io, fptr);
|
|
if (fptr->readconv)
|
|
rb_econv_binmode(fptr->readconv);
|
|
if (fptr->writeconv)
|
|
rb_econv_binmode(fptr->writeconv);
|
|
fptr->mode |= FMODE_BINMODE;
|
|
fptr->mode &= ~FMODE_TEXTMODE;
|
|
fptr->writeconv_pre_ecflags &= ~ECONV_NEWLINE_DECORATOR_MASK;
|
|
#ifdef O_BINARY
|
|
if (!fptr->readconv) {
|
|
SET_BINARY_MODE_WITH_SEEK_CUR(fptr);
|
|
}
|
|
else {
|
|
setmode(fptr->fd, O_BINARY);
|
|
}
|
|
#endif
|
|
return io;
|
|
}
|
|
|
|
static void
|
|
io_ascii8bit_binmode(rb_io_t *fptr)
|
|
{
|
|
if (fptr->readconv) {
|
|
rb_econv_close(fptr->readconv);
|
|
fptr->readconv = NULL;
|
|
}
|
|
if (fptr->writeconv) {
|
|
rb_econv_close(fptr->writeconv);
|
|
fptr->writeconv = NULL;
|
|
}
|
|
fptr->mode |= FMODE_BINMODE;
|
|
fptr->mode &= ~FMODE_TEXTMODE;
|
|
SET_BINARY_MODE_WITH_SEEK_CUR(fptr);
|
|
|
|
fptr->encs.enc = rb_ascii8bit_encoding();
|
|
fptr->encs.enc2 = NULL;
|
|
fptr->encs.ecflags = 0;
|
|
fptr->encs.ecopts = Qnil;
|
|
clear_codeconv(fptr);
|
|
}
|
|
|
|
VALUE
|
|
rb_io_ascii8bit_binmode(VALUE io)
|
|
{
|
|
rb_io_t *fptr;
|
|
|
|
GetOpenFile(io, fptr);
|
|
io_ascii8bit_binmode(fptr);
|
|
|
|
return io;
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* ios.binmode -> ios
|
|
*
|
|
* Puts <em>ios</em> into binary mode.
|
|
* Once a stream is in binary mode, it cannot be reset to nonbinary mode.
|
|
*
|
|
* - newline conversion disabled
|
|
* - encoding conversion disabled
|
|
* - content is treated as ASCII-8BIT
|
|
*/
|
|
|
|
static VALUE
|
|
rb_io_binmode_m(VALUE io)
|
|
{
|
|
VALUE write_io;
|
|
|
|
rb_io_ascii8bit_binmode(io);
|
|
|
|
write_io = GetWriteIO(io);
|
|
if (write_io != io)
|
|
rb_io_ascii8bit_binmode(write_io);
|
|
return io;
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* ios.binmode? -> true or false
|
|
*
|
|
* Returns <code>true</code> if <em>ios</em> is binmode.
|
|
*/
|
|
static VALUE
|
|
rb_io_binmode_p(VALUE io)
|
|
{
|
|
rb_io_t *fptr;
|
|
GetOpenFile(io, fptr);
|
|
return fptr->mode & FMODE_BINMODE ? Qtrue : Qfalse;
|
|
}
|
|
|
|
static const char*
|
|
rb_io_fmode_modestr(int fmode)
|
|
{
|
|
if (fmode & FMODE_APPEND) {
|
|
if ((fmode & FMODE_READWRITE) == FMODE_READWRITE) {
|
|
return MODE_BTMODE("a+", "ab+", "at+");
|
|
}
|
|
return MODE_BTMODE("a", "ab", "at");
|
|
}
|
|
switch (fmode & FMODE_READWRITE) {
|
|
default:
|
|
rb_raise(rb_eArgError, "invalid access fmode 0x%x", fmode);
|
|
case FMODE_READABLE:
|
|
return MODE_BTMODE("r", "rb", "rt");
|
|
case FMODE_WRITABLE:
|
|
return MODE_BTXMODE("w", "wb", "wt", "wx", "wbx", "wtx");
|
|
case FMODE_READWRITE:
|
|
if (fmode & FMODE_CREATE) {
|
|
return MODE_BTXMODE("w+", "wb+", "wt+", "w+x", "wb+x", "wt+x");
|
|
}
|
|
return MODE_BTMODE("r+", "rb+", "rt+");
|
|
}
|
|
}
|
|
|
|
static const char bom_prefix[] = "bom|";
|
|
static const char utf_prefix[] = "utf-";
|
|
enum {bom_prefix_len = (int)sizeof(bom_prefix) - 1};
|
|
enum {utf_prefix_len = (int)sizeof(utf_prefix) - 1};
|
|
|
|
static int
|
|
io_encname_bom_p(const char *name, long len)
|
|
{
|
|
return len > bom_prefix_len && STRNCASECMP(name, bom_prefix, bom_prefix_len) == 0;
|
|
}
|
|
|
|
int
|
|
rb_io_modestr_fmode(const char *modestr)
|
|
{
|
|
int fmode = 0;
|
|
const char *m = modestr, *p = NULL;
|
|
|
|
switch (*m++) {
|
|
case 'r':
|
|
fmode |= FMODE_READABLE;
|
|
break;
|
|
case 'w':
|
|
fmode |= FMODE_WRITABLE | FMODE_TRUNC | FMODE_CREATE;
|
|
break;
|
|
case 'a':
|
|
fmode |= FMODE_WRITABLE | FMODE_APPEND | FMODE_CREATE;
|
|
break;
|
|
default:
|
|
error:
|
|
rb_raise(rb_eArgError, "invalid access mode %s", modestr);
|
|
}
|
|
|
|
while (*m) {
|
|
switch (*m++) {
|
|
case 'b':
|
|
fmode |= FMODE_BINMODE;
|
|
break;
|
|
case 't':
|
|
fmode |= FMODE_TEXTMODE;
|
|
break;
|
|
case '+':
|
|
fmode |= FMODE_READWRITE;
|
|
break;
|
|
case 'x':
|
|
if (modestr[0] != 'w')
|
|
goto error;
|
|
fmode |= FMODE_EXCL;
|
|
break;
|
|
default:
|
|
goto error;
|
|
case ':':
|
|
p = strchr(m, ':');
|
|
if (io_encname_bom_p(m, p ? (long)(p - m) : (long)strlen(m)))
|
|
fmode |= FMODE_SETENC_BY_BOM;
|
|
goto finished;
|
|
}
|
|
}
|
|
|
|
finished:
|
|
if ((fmode & FMODE_BINMODE) && (fmode & FMODE_TEXTMODE))
|
|
goto error;
|
|
|
|
return fmode;
|
|
}
|
|
|
|
int
|
|
rb_io_oflags_fmode(int oflags)
|
|
{
|
|
int fmode = 0;
|
|
|
|
switch (oflags & O_ACCMODE) {
|
|
case O_RDONLY:
|
|
fmode = FMODE_READABLE;
|
|
break;
|
|
case O_WRONLY:
|
|
fmode = FMODE_WRITABLE;
|
|
break;
|
|
case O_RDWR:
|
|
fmode = FMODE_READWRITE;
|
|
break;
|
|
}
|
|
|
|
if (oflags & O_APPEND) {
|
|
fmode |= FMODE_APPEND;
|
|
}
|
|
if (oflags & O_TRUNC) {
|
|
fmode |= FMODE_TRUNC;
|
|
}
|
|
if (oflags & O_CREAT) {
|
|
fmode |= FMODE_CREATE;
|
|
}
|
|
if (oflags & O_EXCL) {
|
|
fmode |= FMODE_EXCL;
|
|
}
|
|
#ifdef O_BINARY
|
|
if (oflags & O_BINARY) {
|
|
fmode |= FMODE_BINMODE;
|
|
}
|
|
#endif
|
|
|
|
return fmode;
|
|
}
|
|
|
|
static int
|
|
rb_io_fmode_oflags(int fmode)
|
|
{
|
|
int oflags = 0;
|
|
|
|
switch (fmode & FMODE_READWRITE) {
|
|
case FMODE_READABLE:
|
|
oflags |= O_RDONLY;
|
|
break;
|
|
case FMODE_WRITABLE:
|
|
oflags |= O_WRONLY;
|
|
break;
|
|
case FMODE_READWRITE:
|
|
oflags |= O_RDWR;
|
|
break;
|
|
}
|
|
|
|
if (fmode & FMODE_APPEND) {
|
|
oflags |= O_APPEND;
|
|
}
|
|
if (fmode & FMODE_TRUNC) {
|
|
oflags |= O_TRUNC;
|
|
}
|
|
if (fmode & FMODE_CREATE) {
|
|
oflags |= O_CREAT;
|
|
}
|
|
if (fmode & FMODE_EXCL) {
|
|
oflags |= O_EXCL;
|
|
}
|
|
#ifdef O_BINARY
|
|
if (fmode & FMODE_BINMODE) {
|
|
oflags |= O_BINARY;
|
|
}
|
|
#endif
|
|
|
|
return oflags;
|
|
}
|
|
|
|
int
|
|
rb_io_modestr_oflags(const char *modestr)
|
|
{
|
|
return rb_io_fmode_oflags(rb_io_modestr_fmode(modestr));
|
|
}
|
|
|
|
static const char*
|
|
rb_io_oflags_modestr(int oflags)
|
|
{
|
|
#ifdef O_BINARY
|
|
# define MODE_BINARY(a,b) ((oflags & O_BINARY) ? (b) : (a))
|
|
#else
|
|
# define MODE_BINARY(a,b) (a)
|
|
#endif
|
|
int accmode;
|
|
if (oflags & O_EXCL) {
|
|
rb_raise(rb_eArgError, "exclusive access mode is not supported");
|
|
}
|
|
accmode = oflags & (O_RDONLY|O_WRONLY|O_RDWR);
|
|
if (oflags & O_APPEND) {
|
|
if (accmode == O_WRONLY) {
|
|
return MODE_BINARY("a", "ab");
|
|
}
|
|
if (accmode == O_RDWR) {
|
|
return MODE_BINARY("a+", "ab+");
|
|
}
|
|
}
|
|
switch (accmode) {
|
|
default:
|
|
rb_raise(rb_eArgError, "invalid access oflags 0x%x", oflags);
|
|
case O_RDONLY:
|
|
return MODE_BINARY("r", "rb");
|
|
case O_WRONLY:
|
|
return MODE_BINARY("w", "wb");
|
|
case O_RDWR:
|
|
if (oflags & O_TRUNC) {
|
|
return MODE_BINARY("w+", "wb+");
|
|
}
|
|
return MODE_BINARY("r+", "rb+");
|
|
}
|
|
}
|
|
|
|
/*
|
|
* Convert external/internal encodings to enc/enc2
|
|
* NULL => use default encoding
|
|
* Qnil => no encoding specified (internal only)
|
|
*/
|
|
static void
|
|
rb_io_ext_int_to_encs(rb_encoding *ext, rb_encoding *intern, rb_encoding **enc, rb_encoding **enc2, int fmode)
|
|
{
|
|
int default_ext = 0;
|
|
|
|
if (ext == NULL) {
|
|
ext = rb_default_external_encoding();
|
|
default_ext = 1;
|
|
}
|
|
if (ext == rb_ascii8bit_encoding()) {
|
|
/* If external is ASCII-8BIT, no transcoding */
|
|
intern = NULL;
|
|
}
|
|
else if (intern == NULL) {
|
|
intern = rb_default_internal_encoding();
|
|
}
|
|
if (intern == NULL || intern == (rb_encoding *)Qnil ||
|
|
(!(fmode & FMODE_SETENC_BY_BOM) && (intern == ext))) {
|
|
/* No internal encoding => use external + no transcoding */
|
|
*enc = (default_ext && intern != ext) ? NULL : ext;
|
|
*enc2 = NULL;
|
|
}
|
|
else {
|
|
*enc = intern;
|
|
*enc2 = ext;
|
|
}
|
|
}
|
|
|
|
static void
|
|
unsupported_encoding(const char *name, rb_encoding *enc)
|
|
{
|
|
rb_enc_warn(enc, "Unsupported encoding %s ignored", name);
|
|
}
|
|
|
|
static void
|
|
parse_mode_enc(const char *estr, rb_encoding *estr_enc,
|
|
rb_encoding **enc_p, rb_encoding **enc2_p, int *fmode_p)
|
|
{
|
|
const char *p;
|
|
char encname[ENCODING_MAXNAMELEN+1];
|
|
int idx, idx2;
|
|
int fmode = fmode_p ? *fmode_p : 0;
|
|
rb_encoding *ext_enc, *int_enc;
|
|
long len;
|
|
|
|
/* parse estr as "enc" or "enc2:enc" or "enc:-" */
|
|
|
|
p = strrchr(estr, ':');
|
|
len = p ? (p++ - estr) : (long)strlen(estr);
|
|
if ((fmode & FMODE_SETENC_BY_BOM) || io_encname_bom_p(estr, len)) {
|
|
estr += bom_prefix_len;
|
|
len -= bom_prefix_len;
|
|
if (!STRNCASECMP(estr, utf_prefix, utf_prefix_len)) {
|
|
fmode |= FMODE_SETENC_BY_BOM;
|
|
}
|
|
else {
|
|
rb_enc_warn(estr_enc, "BOM with non-UTF encoding %s is nonsense", estr);
|
|
fmode &= ~FMODE_SETENC_BY_BOM;
|
|
}
|
|
}
|
|
if (len == 0 || len > ENCODING_MAXNAMELEN) {
|
|
idx = -1;
|
|
}
|
|
else {
|
|
if (p) {
|
|
memcpy(encname, estr, len);
|
|
encname[len] = '\0';
|
|
estr = encname;
|
|
}
|
|
idx = rb_enc_find_index(estr);
|
|
}
|
|
if (fmode_p) *fmode_p = fmode;
|
|
|
|
if (idx >= 0)
|
|
ext_enc = rb_enc_from_index(idx);
|
|
else {
|
|
if (idx != -2)
|
|
unsupported_encoding(estr, estr_enc);
|
|
ext_enc = NULL;
|
|
}
|
|
|
|
int_enc = NULL;
|
|
if (p) {
|
|
if (*p == '-' && *(p+1) == '\0') {
|
|
/* Special case - "-" => no transcoding */
|
|
int_enc = (rb_encoding *)Qnil;
|
|
}
|
|
else {
|
|
idx2 = rb_enc_find_index(p);
|
|
if (idx2 < 0)
|
|
unsupported_encoding(p, estr_enc);
|
|
else if (!(fmode & FMODE_SETENC_BY_BOM) && (idx2 == idx)) {
|
|
int_enc = (rb_encoding *)Qnil;
|
|
}
|
|
else
|
|
int_enc = rb_enc_from_index(idx2);
|
|
}
|
|
}
|
|
|
|
rb_io_ext_int_to_encs(ext_enc, int_enc, enc_p, enc2_p, fmode);
|
|
}
|
|
|
|
int
|
|
rb_io_extract_encoding_option(VALUE opt, rb_encoding **enc_p, rb_encoding **enc2_p, int *fmode_p)
|
|
{
|
|
VALUE encoding=Qnil, extenc=Qundef, intenc=Qundef, tmp;
|
|
int extracted = 0;
|
|
rb_encoding *extencoding = NULL;
|
|
rb_encoding *intencoding = NULL;
|
|
|
|
if (!NIL_P(opt)) {
|
|
VALUE v;
|
|
v = rb_hash_lookup2(opt, sym_encoding, Qnil);
|
|
if (v != Qnil) encoding = v;
|
|
v = rb_hash_lookup2(opt, sym_extenc, Qundef);
|
|
if (v != Qnil) extenc = v;
|
|
v = rb_hash_lookup2(opt, sym_intenc, Qundef);
|
|
if (v != Qundef) intenc = v;
|
|
}
|
|
if ((extenc != Qundef || intenc != Qundef) && !NIL_P(encoding)) {
|
|
if (!NIL_P(ruby_verbose)) {
|
|
int idx = rb_to_encoding_index(encoding);
|
|
if (idx >= 0) encoding = rb_enc_from_encoding(rb_enc_from_index(idx));
|
|
rb_warn("Ignoring encoding parameter '%"PRIsVALUE"': %s_encoding is used",
|
|
encoding, extenc == Qundef ? "internal" : "external");
|
|
}
|
|
encoding = Qnil;
|
|
}
|
|
if (extenc != Qundef && !NIL_P(extenc)) {
|
|
extencoding = rb_to_encoding(extenc);
|
|
}
|
|
if (intenc != Qundef) {
|
|
if (NIL_P(intenc)) {
|
|
/* internal_encoding: nil => no transcoding */
|
|
intencoding = (rb_encoding *)Qnil;
|
|
}
|
|
else if (!NIL_P(tmp = rb_check_string_type(intenc))) {
|
|
char *p = StringValueCStr(tmp);
|
|
|
|
if (*p == '-' && *(p+1) == '\0') {
|
|
/* Special case - "-" => no transcoding */
|
|
intencoding = (rb_encoding *)Qnil;
|
|
}
|
|
else {
|
|
intencoding = rb_to_encoding(intenc);
|
|
}
|
|
}
|
|
else {
|
|
intencoding = rb_to_encoding(intenc);
|
|
}
|
|
if (extencoding == intencoding) {
|
|
intencoding = (rb_encoding *)Qnil;
|
|
}
|
|
}
|
|
if (!NIL_P(encoding)) {
|
|
extracted = 1;
|
|
if (!NIL_P(tmp = rb_check_string_type(encoding))) {
|
|
parse_mode_enc(StringValueCStr(tmp), rb_enc_get(tmp),
|
|
enc_p, enc2_p, fmode_p);
|
|
}
|
|
else {
|
|
rb_io_ext_int_to_encs(rb_to_encoding(encoding), NULL, enc_p, enc2_p, 0);
|
|
}
|
|
}
|
|
else if (extenc != Qundef || intenc != Qundef) {
|
|
extracted = 1;
|
|
rb_io_ext_int_to_encs(extencoding, intencoding, enc_p, enc2_p, 0);
|
|
}
|
|
return extracted;
|
|
}
|
|
|
|
typedef struct rb_io_enc_t convconfig_t;
|
|
|
|
static void
|
|
validate_enc_binmode(int *fmode_p, int ecflags, rb_encoding *enc, rb_encoding *enc2)
|
|
{
|
|
int fmode = *fmode_p;
|
|
|
|
if ((fmode & FMODE_READABLE) &&
|
|
!enc2 &&
|
|
!(fmode & FMODE_BINMODE) &&
|
|
!rb_enc_asciicompat(enc ? enc : rb_default_external_encoding()))
|
|
rb_raise(rb_eArgError, "ASCII incompatible encoding needs binmode");
|
|
|
|
if ((fmode & FMODE_BINMODE) && (ecflags & ECONV_NEWLINE_DECORATOR_MASK)) {
|
|
rb_raise(rb_eArgError, "newline decorator with binary mode");
|
|
}
|
|
if (!(fmode & FMODE_BINMODE) &&
|
|
(DEFAULT_TEXTMODE || (ecflags & ECONV_NEWLINE_DECORATOR_MASK))) {
|
|
fmode |= FMODE_TEXTMODE;
|
|
*fmode_p = fmode;
|
|
}
|
|
#if !DEFAULT_TEXTMODE
|
|
else if (!(ecflags & ECONV_NEWLINE_DECORATOR_MASK)) {
|
|
fmode &= ~FMODE_TEXTMODE;
|
|
*fmode_p = fmode;
|
|
}
|
|
#endif
|
|
}
|
|
|
|
static void
|
|
extract_binmode(VALUE opthash, int *fmode)
|
|
{
|
|
if (!NIL_P(opthash)) {
|
|
VALUE v;
|
|
v = rb_hash_aref(opthash, sym_textmode);
|
|
if (!NIL_P(v)) {
|
|
if (*fmode & FMODE_TEXTMODE)
|
|
rb_raise(rb_eArgError, "textmode specified twice");
|
|
if (*fmode & FMODE_BINMODE)
|
|
rb_raise(rb_eArgError, "both textmode and binmode specified");
|
|
if (RTEST(v))
|
|
*fmode |= FMODE_TEXTMODE;
|
|
}
|
|
v = rb_hash_aref(opthash, sym_binmode);
|
|
if (!NIL_P(v)) {
|
|
if (*fmode & FMODE_BINMODE)
|
|
rb_raise(rb_eArgError, "binmode specified twice");
|
|
if (*fmode & FMODE_TEXTMODE)
|
|
rb_raise(rb_eArgError, "both textmode and binmode specified");
|
|
if (RTEST(v))
|
|
*fmode |= FMODE_BINMODE;
|
|
}
|
|
|
|
if ((*fmode & FMODE_BINMODE) && (*fmode & FMODE_TEXTMODE))
|
|
rb_raise(rb_eArgError, "both textmode and binmode specified");
|
|
}
|
|
}
|
|
|
|
void
|
|
rb_io_extract_modeenc(VALUE *vmode_p, VALUE *vperm_p, VALUE opthash,
|
|
int *oflags_p, int *fmode_p, convconfig_t *convconfig_p)
|
|
{
|
|
VALUE vmode;
|
|
int oflags, fmode;
|
|
rb_encoding *enc, *enc2;
|
|
int ecflags;
|
|
VALUE ecopts;
|
|
int has_enc = 0, has_vmode = 0;
|
|
VALUE intmode;
|
|
|
|
vmode = *vmode_p;
|
|
|
|
/* Set to defaults */
|
|
rb_io_ext_int_to_encs(NULL, NULL, &enc, &enc2, 0);
|
|
|
|
vmode_handle:
|
|
if (NIL_P(vmode)) {
|
|
fmode = FMODE_READABLE;
|
|
oflags = O_RDONLY;
|
|
}
|
|
else if (!NIL_P(intmode = rb_check_to_integer(vmode, "to_int"))) {
|
|
vmode = intmode;
|
|
oflags = NUM2INT(intmode);
|
|
fmode = rb_io_oflags_fmode(oflags);
|
|
}
|
|
else {
|
|
const char *p;
|
|
|
|
SafeStringValue(vmode);
|
|
p = StringValueCStr(vmode);
|
|
fmode = rb_io_modestr_fmode(p);
|
|
oflags = rb_io_fmode_oflags(fmode);
|
|
p = strchr(p, ':');
|
|
if (p) {
|
|
has_enc = 1;
|
|
parse_mode_enc(p+1, rb_enc_get(vmode), &enc, &enc2, &fmode);
|
|
}
|
|
else {
|
|
rb_encoding *e;
|
|
|
|
e = (fmode & FMODE_BINMODE) ? rb_ascii8bit_encoding() : NULL;
|
|
rb_io_ext_int_to_encs(e, NULL, &enc, &enc2, fmode);
|
|
}
|
|
}
|
|
|
|
if (NIL_P(opthash)) {
|
|
ecflags = (fmode & FMODE_READABLE) ?
|
|
MODE_BTMODE(ECONV_DEFAULT_NEWLINE_DECORATOR,
|
|
0, ECONV_UNIVERSAL_NEWLINE_DECORATOR) : 0;
|
|
#ifdef TEXTMODE_NEWLINE_DECORATOR_ON_WRITE
|
|
ecflags |= (fmode & FMODE_WRITABLE) ?
|
|
MODE_BTMODE(TEXTMODE_NEWLINE_DECORATOR_ON_WRITE,
|
|
0, TEXTMODE_NEWLINE_DECORATOR_ON_WRITE) : 0;
|
|
#endif
|
|
SET_UNIVERSAL_NEWLINE_DECORATOR_IF_ENC2(enc2, ecflags);
|
|
ecopts = Qnil;
|
|
}
|
|
else {
|
|
VALUE v;
|
|
if (!has_vmode) {
|
|
v = rb_hash_aref(opthash, sym_mode);
|
|
if (!NIL_P(v)) {
|
|
if (!NIL_P(vmode)) {
|
|
rb_raise(rb_eArgError, "mode specified twice");
|
|
}
|
|
has_vmode = 1;
|
|
vmode = v;
|
|
goto vmode_handle;
|
|
}
|
|
}
|
|
v = rb_hash_aref(opthash, sym_flags);
|
|
if (!NIL_P(v)) {
|
|
v = rb_to_int(v);
|
|
oflags |= NUM2INT(v);
|
|
vmode = INT2NUM(oflags);
|
|
fmode = rb_io_oflags_fmode(oflags);
|
|
}
|
|
extract_binmode(opthash, &fmode);
|
|
if (fmode & FMODE_BINMODE) {
|
|
#ifdef O_BINARY
|
|
oflags |= O_BINARY;
|
|
#endif
|
|
if (!has_enc)
|
|
rb_io_ext_int_to_encs(rb_ascii8bit_encoding(), NULL, &enc, &enc2, fmode);
|
|
}
|
|
#if DEFAULT_TEXTMODE
|
|
else if (NIL_P(vmode)) {
|
|
fmode |= DEFAULT_TEXTMODE;
|
|
}
|
|
#endif
|
|
v = rb_hash_aref(opthash, sym_perm);
|
|
if (!NIL_P(v)) {
|
|
if (vperm_p) {
|
|
if (!NIL_P(*vperm_p)) {
|
|
rb_raise(rb_eArgError, "perm specified twice");
|
|
}
|
|
*vperm_p = v;
|
|
}
|
|
else {
|
|
/* perm no use, just ignore */
|
|
}
|
|
}
|
|
ecflags = (fmode & FMODE_READABLE) ?
|
|
MODE_BTMODE(ECONV_DEFAULT_NEWLINE_DECORATOR,
|
|
0, ECONV_UNIVERSAL_NEWLINE_DECORATOR) : 0;
|
|
#ifdef TEXTMODE_NEWLINE_DECORATOR_ON_WRITE
|
|
ecflags |= (fmode & FMODE_WRITABLE) ?
|
|
MODE_BTMODE(TEXTMODE_NEWLINE_DECORATOR_ON_WRITE,
|
|
0, TEXTMODE_NEWLINE_DECORATOR_ON_WRITE) : 0;
|
|
#endif
|
|
|
|
if (rb_io_extract_encoding_option(opthash, &enc, &enc2, &fmode)) {
|
|
if (has_enc) {
|
|
rb_raise(rb_eArgError, "encoding specified twice");
|
|
}
|
|
}
|
|
SET_UNIVERSAL_NEWLINE_DECORATOR_IF_ENC2(enc2, ecflags);
|
|
ecflags = rb_econv_prepare_options(opthash, &ecopts, ecflags);
|
|
}
|
|
|
|
validate_enc_binmode(&fmode, ecflags, enc, enc2);
|
|
|
|
*vmode_p = vmode;
|
|
|
|
*oflags_p = oflags;
|
|
*fmode_p = fmode;
|
|
convconfig_p->enc = enc;
|
|
convconfig_p->enc2 = enc2;
|
|
convconfig_p->ecflags = ecflags;
|
|
convconfig_p->ecopts = ecopts;
|
|
}
|
|
|
|
struct sysopen_struct {
|
|
VALUE fname;
|
|
int oflags;
|
|
mode_t perm;
|
|
};
|
|
|
|
static void *
|
|
sysopen_func(void *ptr)
|
|
{
|
|
const struct sysopen_struct *data = ptr;
|
|
const char *fname = RSTRING_PTR(data->fname);
|
|
return (void *)(VALUE)rb_cloexec_open(fname, data->oflags, data->perm);
|
|
}
|
|
|
|
static inline int
|
|
rb_sysopen_internal(struct sysopen_struct *data)
|
|
{
|
|
int fd;
|
|
fd = (int)(VALUE)rb_thread_call_without_gvl(sysopen_func, data, RUBY_UBF_IO, 0);
|
|
if (0 <= fd)
|
|
rb_update_max_fd(fd);
|
|
return fd;
|
|
}
|
|
|
|
static int
|
|
rb_sysopen(VALUE fname, int oflags, mode_t perm)
|
|
{
|
|
int fd;
|
|
struct sysopen_struct data;
|
|
|
|
data.fname = rb_str_encode_ospath(fname);
|
|
StringValueCStr(data.fname);
|
|
data.oflags = oflags;
|
|
data.perm = perm;
|
|
|
|
fd = rb_sysopen_internal(&data);
|
|
if (fd < 0) {
|
|
int e = errno;
|
|
if (rb_gc_for_fd(e)) {
|
|
fd = rb_sysopen_internal(&data);
|
|
}
|
|
if (fd < 0) {
|
|
rb_syserr_fail_path(e, fname);
|
|
}
|
|
}
|
|
return fd;
|
|
}
|
|
|
|
FILE *
|
|
rb_fdopen(int fd, const char *modestr)
|
|
{
|
|
FILE *file;
|
|
|
|
#if defined(__sun)
|
|
errno = 0;
|
|
#endif
|
|
file = fdopen(fd, modestr);
|
|
if (!file) {
|
|
int e = errno;
|
|
#if defined(__sun)
|
|
if (e == 0) {
|
|
rb_gc();
|
|
errno = 0;
|
|
file = fdopen(fd, modestr);
|
|
}
|
|
else
|
|
#endif
|
|
if (rb_gc_for_fd(e)) {
|
|
file = fdopen(fd, modestr);
|
|
}
|
|
if (!file) {
|
|
#ifdef _WIN32
|
|
if (e == 0) e = EINVAL;
|
|
#elif defined(__sun)
|
|
if (e == 0) e = EMFILE;
|
|
#endif
|
|
rb_syserr_fail(e, 0);
|
|
}
|
|
}
|
|
|
|
/* xxx: should be _IONBF? A buffer in FILE may have trouble. */
|
|
#ifdef USE_SETVBUF
|
|
if (setvbuf(file, NULL, _IOFBF, 0) != 0)
|
|
rb_warn("setvbuf() can't be honoured (fd=%d)", fd);
|
|
#endif
|
|
return file;
|
|
}
|
|
|
|
static int
|
|
io_check_tty(rb_io_t *fptr)
|
|
{
|
|
int t = isatty(fptr->fd);
|
|
if (t)
|
|
fptr->mode |= FMODE_TTY|FMODE_DUPLEX;
|
|
return t;
|
|
}
|
|
|
|
static VALUE rb_io_internal_encoding(VALUE);
|
|
static void io_encoding_set(rb_io_t *, VALUE, VALUE, VALUE);
|
|
|
|
static int
|
|
io_strip_bom(VALUE io)
|
|
{
|
|
VALUE b1, b2, b3, b4;
|
|
rb_io_t *fptr;
|
|
|
|
GetOpenFile(io, fptr);
|
|
if (!(fptr->mode & FMODE_READABLE)) return 0;
|
|
if (NIL_P(b1 = rb_io_getbyte(io))) return 0;
|
|
switch (b1) {
|
|
case INT2FIX(0xEF):
|
|
if (NIL_P(b2 = rb_io_getbyte(io))) break;
|
|
if (b2 == INT2FIX(0xBB) && !NIL_P(b3 = rb_io_getbyte(io))) {
|
|
if (b3 == INT2FIX(0xBF)) {
|
|
return rb_utf8_encindex();
|
|
}
|
|
rb_io_ungetbyte(io, b3);
|
|
}
|
|
rb_io_ungetbyte(io, b2);
|
|
break;
|
|
|
|
case INT2FIX(0xFE):
|
|
if (NIL_P(b2 = rb_io_getbyte(io))) break;
|
|
if (b2 == INT2FIX(0xFF)) {
|
|
return ENCINDEX_UTF_16BE;
|
|
}
|
|
rb_io_ungetbyte(io, b2);
|
|
break;
|
|
|
|
case INT2FIX(0xFF):
|
|
if (NIL_P(b2 = rb_io_getbyte(io))) break;
|
|
if (b2 == INT2FIX(0xFE)) {
|
|
b3 = rb_io_getbyte(io);
|
|
if (b3 == INT2FIX(0) && !NIL_P(b4 = rb_io_getbyte(io))) {
|
|
if (b4 == INT2FIX(0)) {
|
|
return ENCINDEX_UTF_32LE;
|
|
}
|
|
rb_io_ungetbyte(io, b4);
|
|
}
|
|
rb_io_ungetbyte(io, b3);
|
|
return ENCINDEX_UTF_16LE;
|
|
}
|
|
rb_io_ungetbyte(io, b2);
|
|
break;
|
|
|
|
case INT2FIX(0):
|
|
if (NIL_P(b2 = rb_io_getbyte(io))) break;
|
|
if (b2 == INT2FIX(0) && !NIL_P(b3 = rb_io_getbyte(io))) {
|
|
if (b3 == INT2FIX(0xFE) && !NIL_P(b4 = rb_io_getbyte(io))) {
|
|
if (b4 == INT2FIX(0xFF)) {
|
|
return ENCINDEX_UTF_32BE;
|
|
}
|
|
rb_io_ungetbyte(io, b4);
|
|
}
|
|
rb_io_ungetbyte(io, b3);
|
|
}
|
|
rb_io_ungetbyte(io, b2);
|
|
break;
|
|
}
|
|
rb_io_ungetbyte(io, b1);
|
|
return 0;
|
|
}
|
|
|
|
static rb_encoding *
|
|
io_set_encoding_by_bom(VALUE io)
|
|
{
|
|
int idx = io_strip_bom(io);
|
|
rb_io_t *fptr;
|
|
rb_encoding *extenc = NULL;
|
|
|
|
GetOpenFile(io, fptr);
|
|
if (idx) {
|
|
extenc = rb_enc_from_index(idx);
|
|
io_encoding_set(fptr, rb_enc_from_encoding(extenc),
|
|
rb_io_internal_encoding(io), Qnil);
|
|
}
|
|
else {
|
|
fptr->encs.enc2 = NULL;
|
|
}
|
|
return extenc;
|
|
}
|
|
|
|
static VALUE
|
|
rb_file_open_generic(VALUE io, VALUE filename, int oflags, int fmode,
|
|
const convconfig_t *convconfig, mode_t perm)
|
|
{
|
|
VALUE pathv;
|
|
rb_io_t *fptr;
|
|
convconfig_t cc;
|
|
if (!convconfig) {
|
|
/* Set to default encodings */
|
|
rb_io_ext_int_to_encs(NULL, NULL, &cc.enc, &cc.enc2, fmode);
|
|
cc.ecflags = 0;
|
|
cc.ecopts = Qnil;
|
|
convconfig = &cc;
|
|
}
|
|
validate_enc_binmode(&fmode, convconfig->ecflags,
|
|
convconfig->enc, convconfig->enc2);
|
|
|
|
MakeOpenFile(io, fptr);
|
|
fptr->mode = fmode;
|
|
fptr->encs = *convconfig;
|
|
pathv = rb_str_new_frozen(filename);
|
|
#ifdef O_TMPFILE
|
|
if (!(oflags & O_TMPFILE)) {
|
|
fptr->pathv = pathv;
|
|
}
|
|
#else
|
|
fptr->pathv = pathv;
|
|
#endif
|
|
fptr->fd = rb_sysopen(pathv, oflags, perm);
|
|
io_check_tty(fptr);
|
|
if (fmode & FMODE_SETENC_BY_BOM) io_set_encoding_by_bom(io);
|
|
|
|
return io;
|
|
}
|
|
|
|
static VALUE
|
|
rb_file_open_internal(VALUE io, VALUE filename, const char *modestr)
|
|
{
|
|
int fmode = rb_io_modestr_fmode(modestr);
|
|
const char *p = strchr(modestr, ':');
|
|
convconfig_t convconfig;
|
|
|
|
if (p) {
|
|
parse_mode_enc(p+1, rb_usascii_encoding(),
|
|
&convconfig.enc, &convconfig.enc2, &fmode);
|
|
convconfig.ecflags = 0;
|
|
convconfig.ecopts = Qnil;
|
|
}
|
|
else {
|
|
rb_encoding *e;
|
|
/* Set to default encodings */
|
|
|
|
e = (fmode & FMODE_BINMODE) ? rb_ascii8bit_encoding() : NULL;
|
|
rb_io_ext_int_to_encs(e, NULL, &convconfig.enc, &convconfig.enc2, fmode);
|
|
convconfig.ecflags = 0;
|
|
convconfig.ecopts = Qnil;
|
|
}
|
|
|
|
return rb_file_open_generic(io, filename,
|
|
rb_io_fmode_oflags(fmode),
|
|
fmode,
|
|
&convconfig,
|
|
0666);
|
|
}
|
|
|
|
VALUE
|
|
rb_file_open_str(VALUE fname, const char *modestr)
|
|
{
|
|
FilePathValue(fname);
|
|
return rb_file_open_internal(io_alloc(rb_cFile), fname, modestr);
|
|
}
|
|
|
|
VALUE
|
|
rb_file_open(const char *fname, const char *modestr)
|
|
{
|
|
return rb_file_open_internal(io_alloc(rb_cFile), rb_str_new_cstr(fname), modestr);
|
|
}
|
|
|
|
#if defined(__CYGWIN__) || !defined(HAVE_WORKING_FORK)
|
|
static struct pipe_list {
|
|
rb_io_t *fptr;
|
|
struct pipe_list *next;
|
|
} *pipe_list;
|
|
|
|
static void
|
|
pipe_add_fptr(rb_io_t *fptr)
|
|
{
|
|
struct pipe_list *list;
|
|
|
|
list = ALLOC(struct pipe_list);
|
|
list->fptr = fptr;
|
|
list->next = pipe_list;
|
|
pipe_list = list;
|
|
}
|
|
|
|
static void
|
|
pipe_del_fptr(rb_io_t *fptr)
|
|
{
|
|
struct pipe_list **prev = &pipe_list;
|
|
struct pipe_list *tmp;
|
|
|
|
while ((tmp = *prev) != 0) {
|
|
if (tmp->fptr == fptr) {
|
|
*prev = tmp->next;
|
|
free(tmp);
|
|
return;
|
|
}
|
|
prev = &tmp->next;
|
|
}
|
|
}
|
|
|
|
#if defined (_WIN32) || defined(__CYGWIN__)
|
|
static void
|
|
pipe_atexit(void)
|
|
{
|
|
struct pipe_list *list = pipe_list;
|
|
struct pipe_list *tmp;
|
|
|
|
while (list) {
|
|
tmp = list->next;
|
|
rb_io_fptr_finalize(list->fptr);
|
|
list = tmp;
|
|
}
|
|
}
|
|
#endif
|
|
|
|
static void
|
|
pipe_finalize(rb_io_t *fptr, int noraise)
|
|
{
|
|
#if !defined(HAVE_WORKING_FORK) && !defined(_WIN32)
|
|
int status = 0;
|
|
if (fptr->stdio_file) {
|
|
status = pclose(fptr->stdio_file);
|
|
}
|
|
fptr->fd = -1;
|
|
fptr->stdio_file = 0;
|
|
rb_last_status_set(status, fptr->pid);
|
|
#else
|
|
fptr_finalize(fptr, noraise);
|
|
#endif
|
|
pipe_del_fptr(fptr);
|
|
}
|
|
#endif
|
|
|
|
static void
|
|
fptr_copy_finalizer(rb_io_t *fptr, const rb_io_t *orig)
|
|
{
|
|
#if defined(__CYGWIN__) || !defined(HAVE_WORKING_FORK)
|
|
void (*const old_finalize)(struct rb_io_t*,int) = fptr->finalize;
|
|
|
|
if (old_finalize == orig->finalize) return;
|
|
#endif
|
|
|
|
fptr->finalize = orig->finalize;
|
|
|
|
#if defined(__CYGWIN__) || !defined(HAVE_WORKING_FORK)
|
|
if (old_finalize != pipe_finalize) {
|
|
struct pipe_list *list;
|
|
for (list = pipe_list; list; list = list->next) {
|
|
if (list->fptr == fptr) break;
|
|
}
|
|
if (!list) pipe_add_fptr(fptr);
|
|
}
|
|
else {
|
|
pipe_del_fptr(fptr);
|
|
}
|
|
#endif
|
|
}
|
|
|
|
void
|
|
rb_io_synchronized(rb_io_t *fptr)
|
|
{
|
|
rb_io_check_initialized(fptr);
|
|
fptr->mode |= FMODE_SYNC;
|
|
}
|
|
|
|
void
|
|
rb_io_unbuffered(rb_io_t *fptr)
|
|
{
|
|
rb_io_synchronized(fptr);
|
|
}
|
|
|
|
int
|
|
rb_pipe(int *pipes)
|
|
{
|
|
int ret;
|
|
ret = rb_cloexec_pipe(pipes);
|
|
if (ret < 0) {
|
|
if (rb_gc_for_fd(errno)) {
|
|
ret = rb_cloexec_pipe(pipes);
|
|
}
|
|
}
|
|
if (ret == 0) {
|
|
rb_update_max_fd(pipes[0]);
|
|
rb_update_max_fd(pipes[1]);
|
|
}
|
|
return ret;
|
|
}
|
|
|
|
#ifdef _WIN32
|
|
#define HAVE_SPAWNV 1
|
|
#define spawnv(mode, cmd, args) rb_w32_uaspawn((mode), (cmd), (args))
|
|
#define spawn(mode, cmd) rb_w32_uspawn((mode), (cmd), 0)
|
|
#endif
|
|
|
|
#if defined(HAVE_WORKING_FORK) || defined(HAVE_SPAWNV)
|
|
struct popen_arg {
|
|
VALUE execarg_obj;
|
|
struct rb_execarg *eargp;
|
|
int modef;
|
|
int pair[2];
|
|
int write_pair[2];
|
|
};
|
|
#endif
|
|
|
|
#ifdef HAVE_WORKING_FORK
|
|
static void
|
|
popen_redirect(struct popen_arg *p)
|
|
{
|
|
if ((p->modef & FMODE_READABLE) && (p->modef & FMODE_WRITABLE)) {
|
|
close(p->write_pair[1]);
|
|
if (p->write_pair[0] != 0) {
|
|
dup2(p->write_pair[0], 0);
|
|
close(p->write_pair[0]);
|
|
}
|
|
close(p->pair[0]);
|
|
if (p->pair[1] != 1) {
|
|
dup2(p->pair[1], 1);
|
|
close(p->pair[1]);
|
|
}
|
|
}
|
|
else if (p->modef & FMODE_READABLE) {
|
|
close(p->pair[0]);
|
|
if (p->pair[1] != 1) {
|
|
dup2(p->pair[1], 1);
|
|
close(p->pair[1]);
|
|
}
|
|
}
|
|
else {
|
|
close(p->pair[1]);
|
|
if (p->pair[0] != 0) {
|
|
dup2(p->pair[0], 0);
|
|
close(p->pair[0]);
|
|
}
|
|
}
|
|
}
|
|
|
|
#if defined(__linux__)
|
|
/* Linux /proc/self/status contains a line: "FDSize:\t<nnn>\n"
|
|
* Since /proc may not be available, linux_get_maxfd is just a hint.
|
|
* This function, linux_get_maxfd, must be async-signal-safe.
|
|
* I.e. opendir() is not usable.
|
|
*
|
|
* Note that memchr() and memcmp is *not* async-signal-safe in POSIX.
|
|
* However they are easy to re-implement in async-signal-safe manner.
|
|
* (Also note that there is missing/memcmp.c.)
|
|
*/
|
|
static int
|
|
linux_get_maxfd(void)
|
|
{
|
|
int fd;
|
|
char buf[4096], *p, *np, *e;
|
|
ssize_t ss;
|
|
fd = rb_cloexec_open("/proc/self/status", O_RDONLY|O_NOCTTY, 0);
|
|
if (fd < 0) return fd;
|
|
ss = read(fd, buf, sizeof(buf));
|
|
if (ss < 0) goto err;
|
|
p = buf;
|
|
e = buf + ss;
|
|
while ((int)sizeof("FDSize:\t0\n")-1 <= e-p &&
|
|
(np = memchr(p, '\n', e-p)) != NULL) {
|
|
if (memcmp(p, "FDSize:", sizeof("FDSize:")-1) == 0) {
|
|
int fdsize;
|
|
p += sizeof("FDSize:")-1;
|
|
*np = '\0';
|
|
fdsize = (int)ruby_strtoul(p, (char **)NULL, 10);
|
|
close(fd);
|
|
return fdsize;
|
|
}
|
|
p = np+1;
|
|
}
|
|
/* fall through */
|
|
|
|
err:
|
|
close(fd);
|
|
return (int)ss;
|
|
}
|
|
#endif
|
|
|
|
/* This function should be async-signal-safe. */
|
|
void
|
|
rb_close_before_exec(int lowfd, int maxhint, VALUE noclose_fds)
|
|
{
|
|
#if defined(HAVE_FCNTL) && defined(F_GETFD) && defined(F_SETFD) && defined(FD_CLOEXEC)
|
|
int fd, ret;
|
|
int max = (int)max_file_descriptor;
|
|
# ifdef F_MAXFD
|
|
/* F_MAXFD is available since NetBSD 2.0. */
|
|
ret = fcntl(0, F_MAXFD); /* async-signal-safe */
|
|
if (ret != -1)
|
|
maxhint = max = ret;
|
|
# elif defined(__linux__)
|
|
ret = linux_get_maxfd();
|
|
if (maxhint < ret)
|
|
maxhint = ret;
|
|
/* maxhint = max = ret; if (ret == -1) abort(); // test */
|
|
# endif
|
|
if (max < maxhint)
|
|
max = maxhint;
|
|
for (fd = lowfd; fd <= max; fd++) {
|
|
if (!NIL_P(noclose_fds) &&
|
|
RTEST(rb_hash_lookup(noclose_fds, INT2FIX(fd)))) /* async-signal-safe */
|
|
continue;
|
|
ret = fcntl(fd, F_GETFD); /* async-signal-safe */
|
|
if (ret != -1 && !(ret & FD_CLOEXEC)) {
|
|
fcntl(fd, F_SETFD, ret|FD_CLOEXEC); /* async-signal-safe */
|
|
}
|
|
# define CONTIGUOUS_CLOSED_FDS 20
|
|
if (ret != -1) {
|
|
if (max < fd + CONTIGUOUS_CLOSED_FDS)
|
|
max = fd + CONTIGUOUS_CLOSED_FDS;
|
|
}
|
|
}
|
|
#endif
|
|
}
|
|
|
|
static int
|
|
popen_exec(void *pp, char *errmsg, size_t errmsg_len)
|
|
{
|
|
struct popen_arg *p = (struct popen_arg*)pp;
|
|
|
|
return rb_exec_async_signal_safe(p->eargp, errmsg, errmsg_len);
|
|
}
|
|
#endif
|
|
|
|
#if defined(HAVE_WORKING_FORK) || defined(HAVE_SPAWNV)
|
|
static VALUE
|
|
rb_execarg_fixup_v(VALUE execarg_obj)
|
|
{
|
|
rb_execarg_parent_start(execarg_obj);
|
|
return Qnil;
|
|
}
|
|
#else
|
|
char *rb_execarg_commandline(const struct rb_execarg *eargp, VALUE *prog);
|
|
#endif
|
|
|
|
static VALUE
|
|
pipe_open(VALUE execarg_obj, const char *modestr, int fmode,
|
|
const convconfig_t *convconfig)
|
|
{
|
|
struct rb_execarg *eargp = NIL_P(execarg_obj) ? NULL : rb_execarg_get(execarg_obj);
|
|
VALUE prog = eargp ? (eargp->use_shell ? eargp->invoke.sh.shell_script : eargp->invoke.cmd.command_name) : Qfalse ;
|
|
rb_pid_t pid = 0;
|
|
rb_io_t *fptr;
|
|
VALUE port;
|
|
rb_io_t *write_fptr;
|
|
VALUE write_port;
|
|
#if defined(HAVE_WORKING_FORK)
|
|
int status;
|
|
char errmsg[80] = { '\0' };
|
|
#endif
|
|
#if defined(HAVE_WORKING_FORK) || defined(HAVE_SPAWNV)
|
|
int state;
|
|
struct popen_arg arg;
|
|
#endif
|
|
int e = 0;
|
|
#if defined(HAVE_SPAWNV)
|
|
# if defined(HAVE_SPAWNVE)
|
|
# define DO_SPAWN(cmd, args, envp) ((args) ? \
|
|
spawnve(P_NOWAIT, (cmd), (args), (envp)) : \
|
|
spawne(P_NOWAIT, (cmd), (envp)))
|
|
# else
|
|
# define DO_SPAWN(cmd, args, envp) ((args) ? \
|
|
spawnv(P_NOWAIT, (cmd), (args)) : \
|
|
spawn(P_NOWAIT, (cmd)))
|
|
# endif
|
|
# if !defined(HAVE_WORKING_FORK)
|
|
char **args = NULL;
|
|
# if defined(HAVE_SPAWNVE)
|
|
char **envp = NULL;
|
|
# endif
|
|
# endif
|
|
#endif
|
|
#if !defined(HAVE_WORKING_FORK)
|
|
struct rb_execarg sarg, *sargp = &sarg;
|
|
#endif
|
|
FILE *fp = 0;
|
|
int fd = -1;
|
|
int write_fd = -1;
|
|
#if !defined(HAVE_WORKING_FORK)
|
|
const char *cmd = 0;
|
|
|
|
if (prog)
|
|
cmd = StringValueCStr(prog);
|
|
#endif
|
|
|
|
#if defined(HAVE_WORKING_FORK) || defined(HAVE_SPAWNV)
|
|
arg.execarg_obj = execarg_obj;
|
|
arg.eargp = eargp;
|
|
arg.modef = fmode;
|
|
arg.pair[0] = arg.pair[1] = -1;
|
|
arg.write_pair[0] = arg.write_pair[1] = -1;
|
|
# if !defined(HAVE_WORKING_FORK)
|
|
if (eargp && !eargp->use_shell) {
|
|
args = ARGVSTR2ARGV(eargp->invoke.cmd.argv_str);
|
|
}
|
|
# endif
|
|
switch (fmode & (FMODE_READABLE|FMODE_WRITABLE)) {
|
|
case FMODE_READABLE|FMODE_WRITABLE:
|
|
if (rb_pipe(arg.write_pair) < 0)
|
|
rb_sys_fail_str(prog);
|
|
if (rb_pipe(arg.pair) < 0) {
|
|
e = errno;
|
|
close(arg.write_pair[0]);
|
|
close(arg.write_pair[1]);
|
|
rb_syserr_fail_str(e, prog);
|
|
}
|
|
if (eargp) {
|
|
rb_execarg_addopt(execarg_obj, INT2FIX(0), INT2FIX(arg.write_pair[0]));
|
|
rb_execarg_addopt(execarg_obj, INT2FIX(1), INT2FIX(arg.pair[1]));
|
|
}
|
|
break;
|
|
case FMODE_READABLE:
|
|
if (rb_pipe(arg.pair) < 0)
|
|
rb_sys_fail_str(prog);
|
|
if (eargp)
|
|
rb_execarg_addopt(execarg_obj, INT2FIX(1), INT2FIX(arg.pair[1]));
|
|
break;
|
|
case FMODE_WRITABLE:
|
|
if (rb_pipe(arg.pair) < 0)
|
|
rb_sys_fail_str(prog);
|
|
if (eargp)
|
|
rb_execarg_addopt(execarg_obj, INT2FIX(0), INT2FIX(arg.pair[0]));
|
|
break;
|
|
default:
|
|
rb_sys_fail_str(prog);
|
|
}
|
|
if (!NIL_P(execarg_obj)) {
|
|
rb_protect(rb_execarg_fixup_v, execarg_obj, &state);
|
|
if (state) {
|
|
if (0 <= arg.write_pair[0]) close(arg.write_pair[0]);
|
|
if (0 <= arg.write_pair[1]) close(arg.write_pair[1]);
|
|
if (0 <= arg.pair[0]) close(arg.pair[0]);
|
|
if (0 <= arg.pair[1]) close(arg.pair[1]);
|
|
rb_execarg_parent_end(execarg_obj);
|
|
rb_jump_tag(state);
|
|
}
|
|
|
|
# if defined(HAVE_WORKING_FORK)
|
|
pid = rb_fork_async_signal_safe(&status, popen_exec, &arg, arg.eargp->redirect_fds, errmsg, sizeof(errmsg));
|
|
# else
|
|
rb_execarg_run_options(eargp, sargp, NULL, 0);
|
|
# if defined(HAVE_SPAWNVE)
|
|
if (eargp->envp_str) envp = (char **)RSTRING_PTR(eargp->envp_str);
|
|
# endif
|
|
while ((pid = DO_SPAWN(cmd, args, envp)) < 0) {
|
|
/* exec failed */
|
|
switch (e = errno) {
|
|
case EAGAIN:
|
|
# if defined(EWOULDBLOCK) && EWOULDBLOCK != EAGAIN
|
|
case EWOULDBLOCK:
|
|
# endif
|
|
rb_thread_sleep(1);
|
|
continue;
|
|
}
|
|
break;
|
|
}
|
|
if (eargp)
|
|
rb_execarg_run_options(sargp, NULL, NULL, 0);
|
|
# endif
|
|
rb_execarg_parent_end(execarg_obj);
|
|
}
|
|
else {
|
|
# if defined(HAVE_WORKING_FORK)
|
|
pid = rb_fork_ruby(&status);
|
|
if (pid == 0) { /* child */
|
|
rb_thread_atfork();
|
|
popen_redirect(&arg);
|
|
rb_io_synchronized(RFILE(orig_stdout)->fptr);
|
|
rb_io_synchronized(RFILE(orig_stderr)->fptr);
|
|
return Qnil;
|
|
}
|
|
# else
|
|
rb_notimplement();
|
|
# endif
|
|
}
|
|
|
|
/* parent */
|
|
if (pid < 0) {
|
|
# if defined(HAVE_WORKING_FORK)
|
|
e = errno;
|
|
# endif
|
|
close(arg.pair[0]);
|
|
close(arg.pair[1]);
|
|
if ((fmode & (FMODE_READABLE|FMODE_WRITABLE)) == (FMODE_READABLE|FMODE_WRITABLE)) {
|
|
close(arg.write_pair[0]);
|
|
close(arg.write_pair[1]);
|
|
}
|
|
# if defined(HAVE_WORKING_FORK)
|
|
if (errmsg[0])
|
|
rb_syserr_fail(e, errmsg);
|
|
# endif
|
|
rb_syserr_fail_str(e, prog);
|
|
}
|
|
if ((fmode & FMODE_READABLE) && (fmode & FMODE_WRITABLE)) {
|
|
close(arg.pair[1]);
|
|
fd = arg.pair[0];
|
|
close(arg.write_pair[0]);
|
|
write_fd = arg.write_pair[1];
|
|
}
|
|
else if (fmode & FMODE_READABLE) {
|
|
close(arg.pair[1]);
|
|
fd = arg.pair[0];
|
|
}
|
|
else {
|
|
close(arg.pair[0]);
|
|
fd = arg.pair[1];
|
|
}
|
|
#else
|
|
cmd = rb_execarg_commandline(eargp, &prog);
|
|
if (!NIL_P(execarg_obj)) {
|
|
rb_execarg_parent_start(execarg_obj);
|
|
rb_execarg_run_options(eargp, sargp, NULL, 0);
|
|
}
|
|
fp = popen(cmd, modestr);
|
|
e = errno;
|
|
if (eargp) {
|
|
rb_execarg_parent_end(execarg_obj);
|
|
rb_execarg_run_options(sargp, NULL, NULL, 0);
|
|
}
|
|
if (!fp) rb_syserr_fail_path(e, prog);
|
|
fd = fileno(fp);
|
|
#endif
|
|
|
|
port = io_alloc(rb_cIO);
|
|
MakeOpenFile(port, fptr);
|
|
fptr->fd = fd;
|
|
fptr->stdio_file = fp;
|
|
fptr->mode = fmode | FMODE_SYNC|FMODE_DUPLEX;
|
|
if (convconfig) {
|
|
fptr->encs = *convconfig;
|
|
#if defined(RUBY_TEST_CRLF_ENVIRONMENT) || defined(_WIN32)
|
|
if (fptr->encs.ecflags & ECONV_DEFAULT_NEWLINE_DECORATOR) {
|
|
fptr->encs.ecflags |= ECONV_UNIVERSAL_NEWLINE_DECORATOR;
|
|
}
|
|
#endif
|
|
}
|
|
else {
|
|
if (NEED_NEWLINE_DECORATOR_ON_READ(fptr)) {
|
|
fptr->encs.ecflags |= ECONV_UNIVERSAL_NEWLINE_DECORATOR;
|
|
}
|
|
#ifdef TEXTMODE_NEWLINE_DECORATOR_ON_WRITE
|
|
if (NEED_NEWLINE_DECORATOR_ON_WRITE(fptr)) {
|
|
fptr->encs.ecflags |= TEXTMODE_NEWLINE_DECORATOR_ON_WRITE;
|
|
}
|
|
#endif
|
|
}
|
|
fptr->pid = pid;
|
|
|
|
if (0 <= write_fd) {
|
|
write_port = io_alloc(rb_cIO);
|
|
MakeOpenFile(write_port, write_fptr);
|
|
write_fptr->fd = write_fd;
|
|
write_fptr->mode = (fmode & ~FMODE_READABLE)| FMODE_SYNC|FMODE_DUPLEX;
|
|
fptr->mode &= ~FMODE_WRITABLE;
|
|
fptr->tied_io_for_writing = write_port;
|
|
rb_ivar_set(port, rb_intern("@tied_io_for_writing"), write_port);
|
|
}
|
|
|
|
#if defined (__CYGWIN__) || !defined(HAVE_WORKING_FORK)
|
|
fptr->finalize = pipe_finalize;
|
|
pipe_add_fptr(fptr);
|
|
#endif
|
|
return port;
|
|
}
|
|
|
|
static int
|
|
is_popen_fork(VALUE prog)
|
|
{
|
|
if (RSTRING_LEN(prog) == 1 && RSTRING_PTR(prog)[0] == '-') {
|
|
#if !defined(HAVE_WORKING_FORK)
|
|
rb_raise(rb_eNotImpError,
|
|
"fork() function is unimplemented on this machine");
|
|
#else
|
|
return TRUE;
|
|
#endif
|
|
}
|
|
return FALSE;
|
|
}
|
|
|
|
static VALUE
|
|
pipe_open_s(VALUE prog, const char *modestr, int fmode,
|
|
const convconfig_t *convconfig)
|
|
{
|
|
int argc = 1;
|
|
VALUE *argv = &prog;
|
|
VALUE execarg_obj = Qnil;
|
|
|
|
if (!is_popen_fork(prog))
|
|
execarg_obj = rb_execarg_new(argc, argv, TRUE, FALSE);
|
|
return pipe_open(execarg_obj, modestr, fmode, convconfig);
|
|
}
|
|
|
|
static VALUE
|
|
pipe_close(VALUE io)
|
|
{
|
|
rb_io_t *fptr = io_close_fptr(io);
|
|
if (fptr) {
|
|
fptr_waitpid(fptr, rb_thread_to_be_killed(rb_thread_current()));
|
|
}
|
|
return Qnil;
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* IO.popen([env,] cmd, mode="r" [, opt]) -> io
|
|
* IO.popen([env,] cmd, mode="r" [, opt]) {|io| block } -> obj
|
|
*
|
|
* Runs the specified command as a subprocess; the subprocess's
|
|
* standard input and output will be connected to the returned
|
|
* IO object.
|
|
*
|
|
* The PID of the started process can be obtained by IO#pid method.
|
|
*
|
|
* _cmd_ is a string or an array as follows.
|
|
*
|
|
* cmd:
|
|
* "-" : fork
|
|
* commandline : command line string which is passed to a shell
|
|
* [env, cmdname, arg1, ..., opts] : command name and zero or more arguments (no shell)
|
|
* [env, [cmdname, argv0], arg1, ..., opts] : command name, argv[0] and zero or more arguments (no shell)
|
|
* (env and opts are optional.)
|
|
*
|
|
* If _cmd_ is a +String+ ``<code>-</code>'',
|
|
* then a new instance of Ruby is started as the subprocess.
|
|
*
|
|
* If <i>cmd</i> is an +Array+ of +String+,
|
|
* then it will be used as the subprocess's +argv+ bypassing a shell.
|
|
* The array can contain a hash at first for environments and
|
|
* a hash at last for options similar to #spawn.
|
|
*
|
|
* The default mode for the new file object is ``r'',
|
|
* but <i>mode</i> may be set to any of the modes listed in the description for class IO.
|
|
* The last argument <i>opt</i> qualifies <i>mode</i>.
|
|
*
|
|
* # set IO encoding
|
|
* IO.popen("nkf -e filename", :external_encoding=>"EUC-JP") {|nkf_io|
|
|
* euc_jp_string = nkf_io.read
|
|
* }
|
|
*
|
|
* # merge standard output and standard error using
|
|
* # spawn option. See the document of Kernel.spawn.
|
|
* IO.popen(["ls", "/", :err=>[:child, :out]]) {|ls_io|
|
|
* ls_result_with_error = ls_io.read
|
|
* }
|
|
*
|
|
* # spawn options can be mixed with IO options
|
|
* IO.popen(["ls", "/"], :err=>[:child, :out]) {|ls_io|
|
|
* ls_result_with_error = ls_io.read
|
|
* }
|
|
*
|
|
* Raises exceptions which IO.pipe and Kernel.spawn raise.
|
|
*
|
|
* If a block is given, Ruby will run the command as a child connected
|
|
* to Ruby with a pipe. Ruby's end of the pipe will be passed as a
|
|
* parameter to the block.
|
|
* At the end of block, Ruby closes the pipe and sets <code>$?</code>.
|
|
* In this case IO.popen returns the value of the block.
|
|
*
|
|
* If a block is given with a _cmd_ of ``<code>-</code>'',
|
|
* the block will be run in two separate processes: once in the parent,
|
|
* and once in a child. The parent process will be passed the pipe
|
|
* object as a parameter to the block, the child version of the block
|
|
* will be passed +nil+, and the child's standard in and
|
|
* standard out will be connected to the parent through the pipe. Not
|
|
* available on all platforms.
|
|
*
|
|
* f = IO.popen("uname")
|
|
* p f.readlines
|
|
* f.close
|
|
* puts "Parent is #{Process.pid}"
|
|
* IO.popen("date") {|f| puts f.gets }
|
|
* IO.popen("-") {|f| $stderr.puts "#{Process.pid} is here, f is #{f.inspect}"}
|
|
* p $?
|
|
* IO.popen(%w"sed -e s|^|<foo>| -e s&$&;zot;&", "r+") {|f|
|
|
* f.puts "bar"; f.close_write; puts f.gets
|
|
* }
|
|
*
|
|
* <em>produces:</em>
|
|
*
|
|
* ["Linux\n"]
|
|
* Parent is 21346
|
|
* Thu Jan 15 22:41:19 JST 2009
|
|
* 21346 is here, f is #<IO:fd 3>
|
|
* 21352 is here, f is nil
|
|
* #<Process::Status: pid 21352 exit 0>
|
|
* <foo>bar;zot;
|
|
*/
|
|
|
|
static VALUE
|
|
rb_io_s_popen(int argc, VALUE *argv, VALUE klass)
|
|
{
|
|
const char *modestr;
|
|
VALUE pname, pmode = Qnil, port, tmp, opt = Qnil, env = Qnil, execarg_obj = Qnil;
|
|
int oflags, fmode;
|
|
convconfig_t convconfig;
|
|
|
|
if (argc > 1 && !NIL_P(opt = rb_check_hash_type(argv[argc-1]))) --argc;
|
|
if (argc > 1 && !NIL_P(env = rb_check_hash_type(argv[0]))) --argc, ++argv;
|
|
switch (argc) {
|
|
case 2:
|
|
pmode = argv[1];
|
|
case 1:
|
|
pname = argv[0];
|
|
break;
|
|
default:
|
|
{
|
|
int ex = !NIL_P(opt);
|
|
rb_error_arity(argc + ex, 1 + ex, 2 + ex);
|
|
}
|
|
}
|
|
|
|
tmp = rb_check_array_type(pname);
|
|
if (!NIL_P(tmp)) {
|
|
long len = RARRAY_LEN(tmp);
|
|
#if SIZEOF_LONG > SIZEOF_INT
|
|
if (len > INT_MAX) {
|
|
rb_raise(rb_eArgError, "too many arguments");
|
|
}
|
|
#endif
|
|
execarg_obj = rb_execarg_new((int)len, RARRAY_CONST_PTR(tmp), FALSE, FALSE);
|
|
RB_GC_GUARD(tmp);
|
|
}
|
|
else {
|
|
SafeStringValue(pname);
|
|
execarg_obj = Qnil;
|
|
if (!is_popen_fork(pname))
|
|
execarg_obj = rb_execarg_new(1, &pname, TRUE, FALSE);
|
|
}
|
|
if (!NIL_P(execarg_obj)) {
|
|
if (!NIL_P(opt))
|
|
opt = rb_execarg_extract_options(execarg_obj, opt);
|
|
if (!NIL_P(env))
|
|
rb_execarg_setenv(execarg_obj, env);
|
|
}
|
|
rb_io_extract_modeenc(&pmode, 0, opt, &oflags, &fmode, &convconfig);
|
|
modestr = rb_io_oflags_modestr(oflags);
|
|
|
|
port = pipe_open(execarg_obj, modestr, fmode, &convconfig);
|
|
if (NIL_P(port)) {
|
|
/* child */
|
|
if (rb_block_given_p()) {
|
|
rb_yield(Qnil);
|
|
rb_io_flush(rb_stdout);
|
|
rb_io_flush(rb_stderr);
|
|
_exit(0);
|
|
}
|
|
return Qnil;
|
|
}
|
|
RBASIC_SET_CLASS(port, klass);
|
|
if (rb_block_given_p()) {
|
|
return rb_ensure(rb_yield, port, pipe_close, port);
|
|
}
|
|
return port;
|
|
}
|
|
|
|
static void
|
|
rb_scan_open_args(int argc, const VALUE *argv,
|
|
VALUE *fname_p, int *oflags_p, int *fmode_p,
|
|
convconfig_t *convconfig_p, mode_t *perm_p)
|
|
{
|
|
VALUE opt, fname, vmode, vperm;
|
|
int oflags, fmode;
|
|
mode_t perm;
|
|
|
|
argc = rb_scan_args(argc, argv, "12:", &fname, &vmode, &vperm, &opt);
|
|
FilePathValue(fname);
|
|
|
|
rb_io_extract_modeenc(&vmode, &vperm, opt, &oflags, &fmode, convconfig_p);
|
|
|
|
perm = NIL_P(vperm) ? 0666 : NUM2MODET(vperm);
|
|
|
|
*fname_p = fname;
|
|
*oflags_p = oflags;
|
|
*fmode_p = fmode;
|
|
*perm_p = perm;
|
|
}
|
|
|
|
static VALUE
|
|
rb_open_file(int argc, const VALUE *argv, VALUE io)
|
|
{
|
|
VALUE fname;
|
|
int oflags, fmode;
|
|
convconfig_t convconfig;
|
|
mode_t perm;
|
|
|
|
rb_scan_open_args(argc, argv, &fname, &oflags, &fmode, &convconfig, &perm);
|
|
rb_file_open_generic(io, fname, oflags, fmode, &convconfig, perm);
|
|
|
|
return io;
|
|
}
|
|
|
|
|
|
/*
|
|
* Document-method: File::open
|
|
*
|
|
* call-seq:
|
|
* File.open(filename, mode="r" [, opt]) -> file
|
|
* File.open(filename [, mode [, perm]] [, opt]) -> file
|
|
* File.open(filename, mode="r" [, opt]) {|file| block } -> obj
|
|
* File.open(filename [, mode [, perm]] [, opt]) {|file| block } -> obj
|
|
*
|
|
* With no associated block, File.open is a synonym for
|
|
* File.new. If the optional code block is given, it will
|
|
* be passed the opened +file+ as an argument and the File object will
|
|
* automatically be closed when the block terminates. The value of the block
|
|
* will be returned from File.open.
|
|
*
|
|
* If a file is being created, its initial permissions may be set using the
|
|
* +perm+ parameter. See File.new for further discussion.
|
|
*
|
|
* See IO.new for a description of the +mode+ and +opt+ parameters.
|
|
*/
|
|
|
|
/*
|
|
* Document-method: IO::open
|
|
*
|
|
* call-seq:
|
|
* IO.open(fd, mode="r" [, opt]) -> io
|
|
* IO.open(fd, mode="r" [, opt]) {|io| block } -> obj
|
|
*
|
|
* With no associated block, IO.open is a synonym for IO.new. If
|
|
* the optional code block is given, it will be passed +io+ as an argument,
|
|
* and the IO object will automatically be closed when the block terminates.
|
|
* In this instance, IO.open returns the value of the block.
|
|
*
|
|
* See IO.new for a description of the +fd+, +mode+ and +opt+ parameters.
|
|
*/
|
|
|
|
static VALUE
|
|
rb_io_s_open(int argc, VALUE *argv, VALUE klass)
|
|
{
|
|
VALUE io = rb_class_new_instance_kw(argc, argv, klass, RB_PASS_CALLED_KEYWORDS);
|
|
|
|
if (rb_block_given_p()) {
|
|
return rb_ensure(rb_yield, io, io_close, io);
|
|
}
|
|
|
|
return io;
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* IO.sysopen(path, [mode, [perm]]) -> integer
|
|
*
|
|
* Opens the given path, returning the underlying file descriptor as a
|
|
* Integer.
|
|
*
|
|
* IO.sysopen("testfile") #=> 3
|
|
*/
|
|
|
|
static VALUE
|
|
rb_io_s_sysopen(int argc, VALUE *argv, VALUE _)
|
|
{
|
|
VALUE fname, vmode, vperm;
|
|
VALUE intmode;
|
|
int oflags, fd;
|
|
mode_t perm;
|
|
|
|
rb_scan_args(argc, argv, "12", &fname, &vmode, &vperm);
|
|
FilePathValue(fname);
|
|
|
|
if (NIL_P(vmode))
|
|
oflags = O_RDONLY;
|
|
else if (!NIL_P(intmode = rb_check_to_integer(vmode, "to_int")))
|
|
oflags = NUM2INT(intmode);
|
|
else {
|
|
SafeStringValue(vmode);
|
|
oflags = rb_io_modestr_oflags(StringValueCStr(vmode));
|
|
}
|
|
if (NIL_P(vperm)) perm = 0666;
|
|
else perm = NUM2MODET(vperm);
|
|
|
|
RB_GC_GUARD(fname) = rb_str_new4(fname);
|
|
fd = rb_sysopen(fname, oflags, perm);
|
|
return INT2NUM(fd);
|
|
}
|
|
|
|
static VALUE
|
|
check_pipe_command(VALUE filename_or_command)
|
|
{
|
|
char *s = RSTRING_PTR(filename_or_command);
|
|
long l = RSTRING_LEN(filename_or_command);
|
|
char *e = s + l;
|
|
int chlen;
|
|
|
|
if (rb_enc_ascget(s, e, &chlen, rb_enc_get(filename_or_command)) == '|') {
|
|
VALUE cmd = rb_str_new(s+chlen, l-chlen);
|
|
OBJ_INFECT(cmd, filename_or_command);
|
|
return cmd;
|
|
}
|
|
return Qnil;
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* open(path [, mode [, perm]] [, opt]) -> io or nil
|
|
* open(path [, mode [, perm]] [, opt]) {|io| block } -> obj
|
|
*
|
|
* Creates an IO object connected to the given stream, file, or subprocess.
|
|
*
|
|
* If +path+ does not start with a pipe character (<code>|</code>), treat it
|
|
* as the name of a file to open using the specified mode (defaulting to
|
|
* "r").
|
|
*
|
|
* The +mode+ is either a string or an integer. If it is an integer, it
|
|
* must be bitwise-or of open(2) flags, such as File::RDWR or File::EXCL. If
|
|
* it is a string, it is either "fmode", "fmode:ext_enc", or
|
|
* "fmode:ext_enc:int_enc".
|
|
*
|
|
* See the documentation of IO.new for full documentation of the +mode+ string
|
|
* directives.
|
|
*
|
|
* If a file is being created, its initial permissions may be set using the
|
|
* +perm+ parameter. See File.new and the open(2) and chmod(2) man pages for
|
|
* a description of permissions.
|
|
*
|
|
* If a block is specified, it will be invoked with the IO object as a
|
|
* parameter, and the IO will be automatically closed when the block
|
|
* terminates. The call returns the value of the block.
|
|
*
|
|
* If +path+ starts with a pipe character (<code>"|"</code>), a subprocess is
|
|
* created, connected to the caller by a pair of pipes. The returned IO
|
|
* object may be used to write to the standard input and read from the
|
|
* standard output of this subprocess.
|
|
*
|
|
* If the command following the pipe is a single minus sign
|
|
* (<code>"|-"</code>), Ruby forks, and this subprocess is connected to the
|
|
* parent. If the command is not <code>"-"</code>, the subprocess runs the
|
|
* command.
|
|
*
|
|
* When the subprocess is Ruby (opened via <code>"|-"</code>), the +open+
|
|
* call returns +nil+. If a block is associated with the open call, that
|
|
* block will run twice --- once in the parent and once in the child.
|
|
*
|
|
* The block parameter will be an IO object in the parent and +nil+ in the
|
|
* child. The parent's +IO+ object will be connected to the child's $stdin
|
|
* and $stdout. The subprocess will be terminated at the end of the block.
|
|
*
|
|
* === Examples
|
|
*
|
|
* Reading from "testfile":
|
|
*
|
|
* open("testfile") do |f|
|
|
* print f.gets
|
|
* end
|
|
*
|
|
* Produces:
|
|
*
|
|
* This is line one
|
|
*
|
|
* Open a subprocess and read its output:
|
|
*
|
|
* cmd = open("|date")
|
|
* print cmd.gets
|
|
* cmd.close
|
|
*
|
|
* Produces:
|
|
*
|
|
* Wed Apr 9 08:56:31 CDT 2003
|
|
*
|
|
* Open a subprocess running the same Ruby program:
|
|
*
|
|
* f = open("|-", "w+")
|
|
* if f.nil?
|
|
* puts "in Child"
|
|
* exit
|
|
* else
|
|
* puts "Got: #{f.gets}"
|
|
* end
|
|
*
|
|
* Produces:
|
|
*
|
|
* Got: in Child
|
|
*
|
|
* Open a subprocess using a block to receive the IO object:
|
|
*
|
|
* open "|-" do |f|
|
|
* if f then
|
|
* # parent process
|
|
* puts "Got: #{f.gets}"
|
|
* else
|
|
* # child process
|
|
* puts "in Child"
|
|
* end
|
|
* end
|
|
*
|
|
* Produces:
|
|
*
|
|
* Got: in Child
|
|
*/
|
|
|
|
static VALUE
|
|
rb_f_open(int argc, VALUE *argv, VALUE _)
|
|
{
|
|
ID to_open = 0;
|
|
int redirect = FALSE;
|
|
|
|
if (argc >= 1) {
|
|
CONST_ID(to_open, "to_open");
|
|
if (rb_respond_to(argv[0], to_open)) {
|
|
redirect = TRUE;
|
|
}
|
|
else {
|
|
VALUE tmp = argv[0];
|
|
FilePathValue(tmp);
|
|
if (NIL_P(tmp)) {
|
|
redirect = TRUE;
|
|
}
|
|
else {
|
|
VALUE cmd = check_pipe_command(tmp);
|
|
if (!NIL_P(cmd)) {
|
|
argv[0] = cmd;
|
|
return rb_io_s_popen(argc, argv, rb_cIO);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
if (redirect) {
|
|
VALUE io = rb_funcallv_kw(argv[0], to_open, argc-1, argv+1, RB_PASS_CALLED_KEYWORDS);
|
|
|
|
if (rb_block_given_p()) {
|
|
return rb_ensure(rb_yield, io, io_close, io);
|
|
}
|
|
return io;
|
|
}
|
|
return rb_io_s_open(argc, argv, rb_cFile);
|
|
}
|
|
|
|
static VALUE rb_io_open_generic(VALUE, VALUE, int, int, const convconfig_t *, mode_t);
|
|
|
|
static VALUE
|
|
rb_io_open(VALUE io, VALUE filename, VALUE vmode, VALUE vperm, VALUE opt)
|
|
{
|
|
int oflags, fmode;
|
|
convconfig_t convconfig;
|
|
mode_t perm;
|
|
|
|
rb_io_extract_modeenc(&vmode, &vperm, opt, &oflags, &fmode, &convconfig);
|
|
perm = NIL_P(vperm) ? 0666 : NUM2MODET(vperm);
|
|
return rb_io_open_generic(io, filename, oflags, fmode, &convconfig, perm);
|
|
}
|
|
|
|
static VALUE
|
|
rb_io_open_generic(VALUE klass, VALUE filename, int oflags, int fmode,
|
|
const convconfig_t *convconfig, mode_t perm)
|
|
{
|
|
VALUE cmd;
|
|
if (klass == rb_cIO && !NIL_P(cmd = check_pipe_command(filename))) {
|
|
return pipe_open_s(cmd, rb_io_oflags_modestr(oflags), fmode, convconfig);
|
|
}
|
|
else {
|
|
return rb_file_open_generic(io_alloc(klass), filename,
|
|
oflags, fmode, convconfig, perm);
|
|
}
|
|
}
|
|
|
|
static VALUE
|
|
io_reopen(VALUE io, VALUE nfile)
|
|
{
|
|
rb_io_t *fptr, *orig;
|
|
int fd, fd2;
|
|
off_t pos = 0;
|
|
|
|
nfile = rb_io_get_io(nfile);
|
|
GetOpenFile(io, fptr);
|
|
GetOpenFile(nfile, orig);
|
|
|
|
if (fptr == orig) return io;
|
|
if (IS_PREP_STDIO(fptr)) {
|
|
if ((fptr->stdio_file == stdin && !(orig->mode & FMODE_READABLE)) ||
|
|
(fptr->stdio_file == stdout && !(orig->mode & FMODE_WRITABLE)) ||
|
|
(fptr->stdio_file == stderr && !(orig->mode & FMODE_WRITABLE))) {
|
|
rb_raise(rb_eArgError,
|
|
"%s can't change access mode from \"%s\" to \"%s\"",
|
|
PREP_STDIO_NAME(fptr), rb_io_fmode_modestr(fptr->mode),
|
|
rb_io_fmode_modestr(orig->mode));
|
|
}
|
|
}
|
|
if (fptr->mode & FMODE_WRITABLE) {
|
|
if (io_fflush(fptr) < 0)
|
|
rb_sys_fail(0);
|
|
}
|
|
else {
|
|
flush_before_seek(fptr);
|
|
}
|
|
if (orig->mode & FMODE_READABLE) {
|
|
pos = io_tell(orig);
|
|
}
|
|
if (orig->mode & FMODE_WRITABLE) {
|
|
if (io_fflush(orig) < 0)
|
|
rb_sys_fail(0);
|
|
}
|
|
|
|
/* copy rb_io_t structure */
|
|
fptr->mode = orig->mode | (fptr->mode & FMODE_PREP);
|
|
fptr->pid = orig->pid;
|
|
fptr->lineno = orig->lineno;
|
|
if (RTEST(orig->pathv)) fptr->pathv = orig->pathv;
|
|
else if (!IS_PREP_STDIO(fptr)) fptr->pathv = Qnil;
|
|
fptr_copy_finalizer(fptr, orig);
|
|
|
|
fd = fptr->fd;
|
|
fd2 = orig->fd;
|
|
if (fd != fd2) {
|
|
if (IS_PREP_STDIO(fptr) || fd <= 2 || !fptr->stdio_file) {
|
|
/* need to keep FILE objects of stdin, stdout and stderr */
|
|
if (rb_cloexec_dup2(fd2, fd) < 0)
|
|
rb_sys_fail_path(orig->pathv);
|
|
rb_update_max_fd(fd);
|
|
}
|
|
else {
|
|
fclose(fptr->stdio_file);
|
|
fptr->stdio_file = 0;
|
|
fptr->fd = -1;
|
|
if (rb_cloexec_dup2(fd2, fd) < 0)
|
|
rb_sys_fail_path(orig->pathv);
|
|
rb_update_max_fd(fd);
|
|
fptr->fd = fd;
|
|
}
|
|
rb_thread_fd_close(fd);
|
|
if ((orig->mode & FMODE_READABLE) && pos >= 0) {
|
|
if (io_seek(fptr, pos, SEEK_SET) < 0 && errno) {
|
|
rb_sys_fail_path(fptr->pathv);
|
|
}
|
|
if (io_seek(orig, pos, SEEK_SET) < 0 && errno) {
|
|
rb_sys_fail_path(orig->pathv);
|
|
}
|
|
}
|
|
}
|
|
|
|
if (fptr->mode & FMODE_BINMODE) {
|
|
rb_io_binmode(io);
|
|
}
|
|
|
|
RBASIC_SET_CLASS(io, rb_obj_class(nfile));
|
|
return io;
|
|
}
|
|
|
|
#ifdef _WIN32
|
|
int rb_freopen(VALUE fname, const char *mode, FILE *fp);
|
|
#else
|
|
static int
|
|
rb_freopen(VALUE fname, const char *mode, FILE *fp)
|
|
{
|
|
if (!freopen(RSTRING_PTR(fname), mode, fp)) {
|
|
RB_GC_GUARD(fname);
|
|
return errno;
|
|
}
|
|
return 0;
|
|
}
|
|
#endif
|
|
|
|
/*
|
|
* call-seq:
|
|
* ios.reopen(other_IO) -> ios
|
|
* ios.reopen(path, mode [, opt]) -> ios
|
|
*
|
|
* Reassociates <em>ios</em> with the I/O stream given in
|
|
* <i>other_IO</i> or to a new stream opened on <i>path</i>. This may
|
|
* dynamically change the actual class of this stream.
|
|
* The +mode+ and +opt+ parameters accept the same values as IO.open.
|
|
*
|
|
* f1 = File.new("testfile")
|
|
* f2 = File.new("testfile")
|
|
* f2.readlines[0] #=> "This is line one\n"
|
|
* f2.reopen(f1) #=> #<File:testfile>
|
|
* f2.readlines[0] #=> "This is line one\n"
|
|
*/
|
|
|
|
static VALUE
|
|
rb_io_reopen(int argc, VALUE *argv, VALUE file)
|
|
{
|
|
VALUE fname, nmode, opt;
|
|
int oflags;
|
|
rb_io_t *fptr;
|
|
|
|
if (rb_scan_args(argc, argv, "11:", &fname, &nmode, &opt) == 1) {
|
|
VALUE tmp = rb_io_check_io(fname);
|
|
if (!NIL_P(tmp)) {
|
|
return io_reopen(file, tmp);
|
|
}
|
|
}
|
|
|
|
FilePathValue(fname);
|
|
rb_io_taint_check(file);
|
|
fptr = RFILE(file)->fptr;
|
|
if (!fptr) {
|
|
fptr = RFILE(file)->fptr = ZALLOC(rb_io_t);
|
|
}
|
|
|
|
if (!NIL_P(nmode) || !NIL_P(opt)) {
|
|
int fmode;
|
|
convconfig_t convconfig;
|
|
|
|
rb_io_extract_modeenc(&nmode, 0, opt, &oflags, &fmode, &convconfig);
|
|
if (IS_PREP_STDIO(fptr) &&
|
|
((fptr->mode & FMODE_READWRITE) & (fmode & FMODE_READWRITE)) !=
|
|
(fptr->mode & FMODE_READWRITE)) {
|
|
rb_raise(rb_eArgError,
|
|
"%s can't change access mode from \"%s\" to \"%s\"",
|
|
PREP_STDIO_NAME(fptr), rb_io_fmode_modestr(fptr->mode),
|
|
rb_io_fmode_modestr(fmode));
|
|
}
|
|
fptr->mode = fmode;
|
|
fptr->encs = convconfig;
|
|
}
|
|
else {
|
|
oflags = rb_io_fmode_oflags(fptr->mode);
|
|
}
|
|
|
|
fptr->pathv = fname;
|
|
if (fptr->fd < 0) {
|
|
fptr->fd = rb_sysopen(fptr->pathv, oflags, 0666);
|
|
fptr->stdio_file = 0;
|
|
return file;
|
|
}
|
|
|
|
if (fptr->mode & FMODE_WRITABLE) {
|
|
if (io_fflush(fptr) < 0)
|
|
rb_sys_fail(0);
|
|
}
|
|
fptr->rbuf.off = fptr->rbuf.len = 0;
|
|
|
|
if (fptr->stdio_file) {
|
|
int e = rb_freopen(rb_str_encode_ospath(fptr->pathv),
|
|
rb_io_oflags_modestr(oflags),
|
|
fptr->stdio_file);
|
|
if (e) rb_syserr_fail_path(e, fptr->pathv);
|
|
fptr->fd = fileno(fptr->stdio_file);
|
|
rb_fd_fix_cloexec(fptr->fd);
|
|
#ifdef USE_SETVBUF
|
|
if (setvbuf(fptr->stdio_file, NULL, _IOFBF, 0) != 0)
|
|
rb_warn("setvbuf() can't be honoured for %"PRIsVALUE, fptr->pathv);
|
|
#endif
|
|
if (fptr->stdio_file == stderr) {
|
|
if (setvbuf(fptr->stdio_file, NULL, _IONBF, BUFSIZ) != 0)
|
|
rb_warn("setvbuf() can't be honoured for %"PRIsVALUE, fptr->pathv);
|
|
}
|
|
else if (fptr->stdio_file == stdout && isatty(fptr->fd)) {
|
|
if (setvbuf(fptr->stdio_file, NULL, _IOLBF, BUFSIZ) != 0)
|
|
rb_warn("setvbuf() can't be honoured for %"PRIsVALUE, fptr->pathv);
|
|
}
|
|
}
|
|
else {
|
|
int tmpfd = rb_sysopen(fptr->pathv, oflags, 0666);
|
|
int err = 0;
|
|
if (rb_cloexec_dup2(tmpfd, fptr->fd) < 0)
|
|
err = errno;
|
|
(void)close(tmpfd);
|
|
if (err) {
|
|
rb_syserr_fail_path(err, fptr->pathv);
|
|
}
|
|
}
|
|
|
|
return file;
|
|
}
|
|
|
|
/* :nodoc: */
|
|
static VALUE
|
|
rb_io_init_copy(VALUE dest, VALUE io)
|
|
{
|
|
rb_io_t *fptr, *orig;
|
|
int fd;
|
|
VALUE write_io;
|
|
off_t pos;
|
|
|
|
io = rb_io_get_io(io);
|
|
if (!OBJ_INIT_COPY(dest, io)) return dest;
|
|
GetOpenFile(io, orig);
|
|
MakeOpenFile(dest, fptr);
|
|
|
|
rb_io_flush(io);
|
|
|
|
/* copy rb_io_t structure */
|
|
fptr->mode = orig->mode & ~FMODE_PREP;
|
|
fptr->encs = orig->encs;
|
|
fptr->pid = orig->pid;
|
|
fptr->lineno = orig->lineno;
|
|
if (!NIL_P(orig->pathv)) fptr->pathv = orig->pathv;
|
|
fptr_copy_finalizer(fptr, orig);
|
|
|
|
fd = ruby_dup(orig->fd);
|
|
fptr->fd = fd;
|
|
pos = io_tell(orig);
|
|
if (0 <= pos)
|
|
io_seek(fptr, pos, SEEK_SET);
|
|
if (fptr->mode & FMODE_BINMODE) {
|
|
rb_io_binmode(dest);
|
|
}
|
|
|
|
write_io = GetWriteIO(io);
|
|
if (io != write_io) {
|
|
write_io = rb_obj_dup(write_io);
|
|
fptr->tied_io_for_writing = write_io;
|
|
rb_ivar_set(dest, rb_intern("@tied_io_for_writing"), write_io);
|
|
}
|
|
|
|
return dest;
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* ios.printf(format_string [, obj, ...]) -> nil
|
|
*
|
|
* Formats and writes to <em>ios</em>, converting parameters under
|
|
* control of the format string. See Kernel#sprintf for details.
|
|
*/
|
|
|
|
VALUE
|
|
rb_io_printf(int argc, const VALUE *argv, VALUE out)
|
|
{
|
|
rb_io_write(out, rb_f_sprintf(argc, argv));
|
|
return Qnil;
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* printf(io, string [, obj ... ]) -> nil
|
|
* printf(string [, obj ... ]) -> nil
|
|
*
|
|
* Equivalent to:
|
|
* io.write(sprintf(string, obj, ...))
|
|
* or
|
|
* $stdout.write(sprintf(string, obj, ...))
|
|
*/
|
|
|
|
static VALUE
|
|
rb_f_printf(int argc, VALUE *argv, VALUE _)
|
|
{
|
|
VALUE out;
|
|
|
|
if (argc == 0) return Qnil;
|
|
if (RB_TYPE_P(argv[0], T_STRING)) {
|
|
out = rb_stdout;
|
|
}
|
|
else {
|
|
out = argv[0];
|
|
argv++;
|
|
argc--;
|
|
}
|
|
rb_io_write(out, rb_f_sprintf(argc, argv));
|
|
|
|
return Qnil;
|
|
}
|
|
|
|
static void
|
|
rb_output_fs_setter(VALUE val, ID id, VALUE *var)
|
|
{
|
|
rb_str_setter(val, id, &val);
|
|
if (!NIL_P(val)) {
|
|
rb_warn("non-nil $, will be deprecated");
|
|
}
|
|
*var = val;
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* ios.print -> nil
|
|
* ios.print(obj, ...) -> nil
|
|
*
|
|
* Writes the given object(s) to <em>ios</em>. Returns +nil+.
|
|
*
|
|
* The stream must be opened for writing.
|
|
* Each given object that isn't a string will be converted by calling
|
|
* its <code>to_s</code> method.
|
|
* When called without arguments, prints the contents of <code>$_</code>.
|
|
*
|
|
* If the output field separator (<code>$,</code>) is not +nil+,
|
|
* it is inserted between objects.
|
|
* If the output record separator (<code>$\\</code>) is not +nil+,
|
|
* it is appended to the output.
|
|
*
|
|
* $stdout.print("This is ", 100, " percent.\n")
|
|
*
|
|
* <em>produces:</em>
|
|
*
|
|
* This is 100 percent.
|
|
*/
|
|
|
|
VALUE
|
|
rb_io_print(int argc, const VALUE *argv, VALUE out)
|
|
{
|
|
int i;
|
|
VALUE line;
|
|
|
|
/* if no argument given, print `$_' */
|
|
if (argc == 0) {
|
|
argc = 1;
|
|
line = rb_lastline_get();
|
|
argv = &line;
|
|
}
|
|
for (i=0; i<argc; i++) {
|
|
if (!NIL_P(rb_output_fs) && i>0) {
|
|
rb_io_write(out, rb_output_fs);
|
|
}
|
|
rb_io_write(out, argv[i]);
|
|
}
|
|
if (argc > 0 && !NIL_P(rb_output_rs)) {
|
|
rb_io_write(out, rb_output_rs);
|
|
}
|
|
|
|
return Qnil;
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* print(obj, ...) -> nil
|
|
*
|
|
* Prints each object in turn to <code>$stdout</code>. If the output
|
|
* field separator (<code>$,</code>) is not +nil+, its
|
|
* contents will appear between each field. If the output record
|
|
* separator (<code>$\\</code>) is not +nil+, it will be
|
|
* appended to the output. If no arguments are given, prints
|
|
* <code>$_</code>. Objects that aren't strings will be converted by
|
|
* calling their <code>to_s</code> method.
|
|
*
|
|
* print "cat", [1,2,3], 99, "\n"
|
|
* $, = ", "
|
|
* $\ = "\n"
|
|
* print "cat", [1,2,3], 99
|
|
*
|
|
* <em>produces:</em>
|
|
*
|
|
* cat12399
|
|
* cat, 1, 2, 3, 99
|
|
*/
|
|
|
|
static VALUE
|
|
rb_f_print(int argc, const VALUE *argv, VALUE _)
|
|
{
|
|
rb_io_print(argc, argv, rb_stdout);
|
|
return Qnil;
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* ios.putc(obj) -> obj
|
|
*
|
|
* If <i>obj</i> is Numeric, write the character whose code is the
|
|
* least-significant byte of <i>obj</i>. If <i>obj</i> is String,
|
|
* write the first character of <i>obj</i> to <em>ios</em>. Otherwise,
|
|
* raise TypeError.
|
|
*
|
|
* $stdout.putc "A"
|
|
* $stdout.putc 65
|
|
*
|
|
* <em>produces:</em>
|
|
*
|
|
* AA
|
|
*/
|
|
|
|
static VALUE
|
|
rb_io_putc(VALUE io, VALUE ch)
|
|
{
|
|
VALUE str;
|
|
if (RB_TYPE_P(ch, T_STRING)) {
|
|
str = rb_str_substr(ch, 0, 1);
|
|
}
|
|
else {
|
|
char c = NUM2CHR(ch);
|
|
str = rb_str_new(&c, 1);
|
|
}
|
|
rb_io_write(io, str);
|
|
return ch;
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* putc(int) -> int
|
|
*
|
|
* Equivalent to:
|
|
*
|
|
* $stdout.putc(int)
|
|
*
|
|
* Refer to the documentation for IO#putc for important information regarding
|
|
* multi-byte characters.
|
|
*/
|
|
|
|
static VALUE
|
|
rb_f_putc(VALUE recv, VALUE ch)
|
|
{
|
|
if (recv == rb_stdout) {
|
|
return rb_io_putc(recv, ch);
|
|
}
|
|
return rb_funcallv(rb_stdout, rb_intern("putc"), 1, &ch);
|
|
}
|
|
|
|
|
|
int
|
|
rb_str_end_with_asciichar(VALUE str, int c)
|
|
{
|
|
long len = RSTRING_LEN(str);
|
|
const char *ptr = RSTRING_PTR(str);
|
|
rb_encoding *enc = rb_enc_from_index(ENCODING_GET(str));
|
|
int n;
|
|
|
|
if (len == 0) return 0;
|
|
if ((n = rb_enc_mbminlen(enc)) == 1) {
|
|
return ptr[len - 1] == c;
|
|
}
|
|
return rb_enc_ascget(ptr + ((len - 1) / n) * n, ptr + len, &n, enc) == c;
|
|
}
|
|
|
|
static VALUE
|
|
io_puts_ary(VALUE ary, VALUE out, int recur)
|
|
{
|
|
VALUE tmp;
|
|
long i;
|
|
|
|
if (recur) {
|
|
tmp = rb_str_new2("[...]");
|
|
rb_io_puts(1, &tmp, out);
|
|
return Qtrue;
|
|
}
|
|
ary = rb_check_array_type(ary);
|
|
if (NIL_P(ary)) return Qfalse;
|
|
for (i=0; i<RARRAY_LEN(ary); i++) {
|
|
tmp = RARRAY_AREF(ary, i);
|
|
rb_io_puts(1, &tmp, out);
|
|
}
|
|
return Qtrue;
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* ios.puts(obj, ...) -> nil
|
|
*
|
|
* Writes the given object(s) to <em>ios</em>.
|
|
* Writes a newline after any that do not already end
|
|
* with a newline sequence. Returns +nil+.
|
|
*
|
|
* The stream must be opened for writing.
|
|
* If called with an array argument, writes each element on a new line.
|
|
* Each given object that isn't a string or array will be converted
|
|
* by calling its +to_s+ method.
|
|
* If called without arguments, outputs a single newline.
|
|
*
|
|
* $stdout.puts("this", "is", ["a", "test"])
|
|
*
|
|
* <em>produces:</em>
|
|
*
|
|
* this
|
|
* is
|
|
* a
|
|
* test
|
|
*
|
|
* Note that +puts+ always uses newlines and is not affected
|
|
* by the output record separator (<code>$\\</code>).
|
|
*/
|
|
|
|
VALUE
|
|
rb_io_puts(int argc, const VALUE *argv, VALUE out)
|
|
{
|
|
int i, n;
|
|
VALUE line, args[2];
|
|
|
|
/* if no argument given, print newline. */
|
|
if (argc == 0) {
|
|
rb_io_write(out, rb_default_rs);
|
|
return Qnil;
|
|
}
|
|
for (i=0; i<argc; i++) {
|
|
if (RB_TYPE_P(argv[i], T_STRING)) {
|
|
line = argv[i];
|
|
goto string;
|
|
}
|
|
if (rb_exec_recursive(io_puts_ary, argv[i], out)) {
|
|
continue;
|
|
}
|
|
line = rb_obj_as_string(argv[i]);
|
|
string:
|
|
n = 0;
|
|
args[n++] = line;
|
|
if (RSTRING_LEN(line) == 0 ||
|
|
!rb_str_end_with_asciichar(line, '\n')) {
|
|
args[n++] = rb_default_rs;
|
|
}
|
|
rb_io_writev(out, n, args);
|
|
}
|
|
|
|
return Qnil;
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* puts(obj, ...) -> nil
|
|
*
|
|
* Equivalent to
|
|
*
|
|
* $stdout.puts(obj, ...)
|
|
*/
|
|
|
|
static VALUE
|
|
rb_f_puts(int argc, VALUE *argv, VALUE recv)
|
|
{
|
|
if (recv == rb_stdout) {
|
|
return rb_io_puts(argc, argv, recv);
|
|
}
|
|
return rb_funcallv(rb_stdout, rb_intern("puts"), argc, argv);
|
|
}
|
|
|
|
void
|
|
rb_p(VALUE obj) /* for debug print within C code */
|
|
{
|
|
VALUE args[2];
|
|
args[0] = rb_obj_as_string(rb_inspect(obj));
|
|
args[1] = rb_default_rs;
|
|
if (RB_TYPE_P(rb_stdout, T_FILE) &&
|
|
rb_method_basic_definition_p(CLASS_OF(rb_stdout), id_write)) {
|
|
io_writev(2, args, rb_stdout);
|
|
}
|
|
else {
|
|
rb_io_writev(rb_stdout, 2, args);
|
|
}
|
|
}
|
|
|
|
struct rb_f_p_arg {
|
|
int argc;
|
|
VALUE *argv;
|
|
};
|
|
|
|
static VALUE
|
|
rb_f_p_internal(VALUE arg)
|
|
{
|
|
struct rb_f_p_arg *arg1 = (struct rb_f_p_arg*)arg;
|
|
int argc = arg1->argc;
|
|
VALUE *argv = arg1->argv;
|
|
int i;
|
|
VALUE ret = Qnil;
|
|
|
|
for (i=0; i<argc; i++) {
|
|
rb_p(argv[i]);
|
|
}
|
|
if (argc == 1) {
|
|
ret = argv[0];
|
|
}
|
|
else if (argc > 1) {
|
|
ret = rb_ary_new4(argc, argv);
|
|
}
|
|
if (RB_TYPE_P(rb_stdout, T_FILE)) {
|
|
rb_io_flush(rb_stdout);
|
|
}
|
|
return ret;
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* p(obj) -> obj
|
|
* p(obj1, obj2, ...) -> [obj, ...]
|
|
* p() -> nil
|
|
*
|
|
* For each object, directly writes _obj_.+inspect+ followed by a
|
|
* newline to the program's standard output.
|
|
*
|
|
* S = Struct.new(:name, :state)
|
|
* s = S['dave', 'TX']
|
|
* p s
|
|
*
|
|
* <em>produces:</em>
|
|
*
|
|
* #<S name="dave", state="TX">
|
|
*/
|
|
|
|
static VALUE
|
|
rb_f_p(int argc, VALUE *argv, VALUE self)
|
|
{
|
|
struct rb_f_p_arg arg;
|
|
arg.argc = argc;
|
|
arg.argv = argv;
|
|
|
|
return rb_uninterruptible(rb_f_p_internal, (VALUE)&arg);
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* obj.display(port=$>) -> nil
|
|
*
|
|
* Prints <i>obj</i> on the given port (default <code>$></code>).
|
|
* Equivalent to:
|
|
*
|
|
* def display(port=$>)
|
|
* port.write self
|
|
* nil
|
|
* end
|
|
*
|
|
* For example:
|
|
*
|
|
* 1.display
|
|
* "cat".display
|
|
* [ 4, 5, 6 ].display
|
|
* puts
|
|
*
|
|
* <em>produces:</em>
|
|
*
|
|
* 1cat[4, 5, 6]
|
|
*/
|
|
|
|
static VALUE
|
|
rb_obj_display(int argc, VALUE *argv, VALUE self)
|
|
{
|
|
VALUE out;
|
|
|
|
out = (!rb_check_arity(argc, 0, 1) ? rb_stdout : argv[0]);
|
|
rb_io_write(out, self);
|
|
|
|
return Qnil;
|
|
}
|
|
|
|
static int
|
|
rb_stderr_to_original_p(void)
|
|
{
|
|
return (rb_stderr == orig_stderr || RFILE(orig_stderr)->fptr->fd < 0);
|
|
}
|
|
|
|
void
|
|
rb_write_error2(const char *mesg, long len)
|
|
{
|
|
if (rb_stderr_to_original_p()) {
|
|
#ifdef _WIN32
|
|
if (isatty(fileno(stderr))) {
|
|
if (rb_w32_write_console(rb_str_new(mesg, len), fileno(stderr)) > 0) return;
|
|
}
|
|
#endif
|
|
if (fwrite(mesg, sizeof(char), (size_t)len, stderr) < (size_t)len) {
|
|
/* failed to write to stderr, what can we do? */
|
|
return;
|
|
}
|
|
}
|
|
else {
|
|
rb_io_write(rb_stderr, rb_str_new(mesg, len));
|
|
}
|
|
}
|
|
|
|
void
|
|
rb_write_error(const char *mesg)
|
|
{
|
|
rb_write_error2(mesg, strlen(mesg));
|
|
}
|
|
|
|
void
|
|
rb_write_error_str(VALUE mesg)
|
|
{
|
|
/* a stopgap measure for the time being */
|
|
if (rb_stderr_to_original_p()) {
|
|
size_t len = (size_t)RSTRING_LEN(mesg);
|
|
#ifdef _WIN32
|
|
if (isatty(fileno(stderr))) {
|
|
if (rb_w32_write_console(mesg, fileno(stderr)) > 0) return;
|
|
}
|
|
#endif
|
|
if (fwrite(RSTRING_PTR(mesg), sizeof(char), len, stderr) < len) {
|
|
RB_GC_GUARD(mesg);
|
|
return;
|
|
}
|
|
}
|
|
else {
|
|
/* may unlock GVL, and */
|
|
rb_io_write(rb_stderr, mesg);
|
|
}
|
|
}
|
|
|
|
int
|
|
rb_stderr_tty_p(void)
|
|
{
|
|
if (rb_stderr_to_original_p())
|
|
return isatty(fileno(stderr));
|
|
return 0;
|
|
}
|
|
|
|
static void
|
|
must_respond_to(ID mid, VALUE val, ID id)
|
|
{
|
|
if (!rb_respond_to(val, mid)) {
|
|
rb_raise(rb_eTypeError, "%"PRIsVALUE" must have %"PRIsVALUE" method, %"PRIsVALUE" given",
|
|
rb_id2str(id), rb_id2str(mid),
|
|
rb_obj_class(val));
|
|
}
|
|
}
|
|
|
|
static void
|
|
stdout_setter(VALUE val, ID id, VALUE *variable)
|
|
{
|
|
must_respond_to(id_write, val, id);
|
|
*variable = val;
|
|
}
|
|
|
|
static VALUE
|
|
prep_io(int fd, int fmode, VALUE klass, const char *path)
|
|
{
|
|
rb_io_t *fp;
|
|
VALUE io = io_alloc(klass);
|
|
|
|
MakeOpenFile(io, fp);
|
|
fp->fd = fd;
|
|
fp->mode = fmode;
|
|
if (!io_check_tty(fp)) {
|
|
#ifdef __CYGWIN__
|
|
fp->mode |= FMODE_BINMODE;
|
|
setmode(fd, O_BINARY);
|
|
#endif
|
|
}
|
|
if (path) fp->pathv = rb_obj_freeze(rb_str_new_cstr(path));
|
|
rb_update_max_fd(fd);
|
|
|
|
return io;
|
|
}
|
|
|
|
VALUE
|
|
rb_io_fdopen(int fd, int oflags, const char *path)
|
|
{
|
|
VALUE klass = rb_cIO;
|
|
|
|
if (path && strcmp(path, "-")) klass = rb_cFile;
|
|
return prep_io(fd, rb_io_oflags_fmode(oflags), klass, path);
|
|
}
|
|
|
|
static VALUE
|
|
prep_stdio(FILE *f, int fmode, VALUE klass, const char *path)
|
|
{
|
|
rb_io_t *fptr;
|
|
VALUE io = prep_io(fileno(f), fmode|FMODE_PREP|DEFAULT_TEXTMODE, klass, path);
|
|
|
|
GetOpenFile(io, fptr);
|
|
fptr->encs.ecflags |= ECONV_DEFAULT_NEWLINE_DECORATOR;
|
|
#ifdef TEXTMODE_NEWLINE_DECORATOR_ON_WRITE
|
|
fptr->encs.ecflags |= TEXTMODE_NEWLINE_DECORATOR_ON_WRITE;
|
|
if (fmode & FMODE_READABLE) {
|
|
fptr->encs.ecflags |= ECONV_UNIVERSAL_NEWLINE_DECORATOR;
|
|
}
|
|
#endif
|
|
fptr->stdio_file = f;
|
|
|
|
return io;
|
|
}
|
|
|
|
FILE *
|
|
rb_io_stdio_file(rb_io_t *fptr)
|
|
{
|
|
if (!fptr->stdio_file) {
|
|
int oflags = rb_io_fmode_oflags(fptr->mode) & ~O_EXCL;
|
|
fptr->stdio_file = rb_fdopen(fptr->fd, rb_io_oflags_modestr(oflags));
|
|
}
|
|
return fptr->stdio_file;
|
|
}
|
|
|
|
static inline void
|
|
rb_io_buffer_init(rb_io_buffer_t *buf)
|
|
{
|
|
buf->ptr = NULL;
|
|
buf->off = 0;
|
|
buf->len = 0;
|
|
buf->capa = 0;
|
|
}
|
|
|
|
static inline rb_io_t *
|
|
rb_io_fptr_new(void)
|
|
{
|
|
rb_io_t *fp = ALLOC(rb_io_t);
|
|
fp->fd = -1;
|
|
fp->stdio_file = NULL;
|
|
fp->mode = 0;
|
|
fp->pid = 0;
|
|
fp->lineno = 0;
|
|
fp->pathv = Qnil;
|
|
fp->finalize = 0;
|
|
rb_io_buffer_init(&fp->wbuf);
|
|
rb_io_buffer_init(&fp->rbuf);
|
|
rb_io_buffer_init(&fp->cbuf);
|
|
fp->readconv = NULL;
|
|
fp->writeconv = NULL;
|
|
fp->writeconv_asciicompat = Qnil;
|
|
fp->writeconv_pre_ecflags = 0;
|
|
fp->writeconv_pre_ecopts = Qnil;
|
|
fp->writeconv_initialized = 0;
|
|
fp->tied_io_for_writing = 0;
|
|
fp->encs.enc = NULL;
|
|
fp->encs.enc2 = NULL;
|
|
fp->encs.ecflags = 0;
|
|
fp->encs.ecopts = Qnil;
|
|
fp->write_lock = 0;
|
|
return fp;
|
|
}
|
|
|
|
rb_io_t *
|
|
rb_io_make_open_file(VALUE obj)
|
|
{
|
|
rb_io_t *fp = 0;
|
|
|
|
Check_Type(obj, T_FILE);
|
|
if (RFILE(obj)->fptr) {
|
|
rb_io_close(obj);
|
|
rb_io_fptr_finalize(RFILE(obj)->fptr);
|
|
RFILE(obj)->fptr = 0;
|
|
}
|
|
fp = rb_io_fptr_new();
|
|
RFILE(obj)->fptr = fp;
|
|
return fp;
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* IO.new(fd [, mode] [, opt]) -> io
|
|
*
|
|
* Returns a new IO object (a stream) for the given integer file descriptor
|
|
* +fd+ and +mode+ string. +opt+ may be used to specify parts of +mode+ in a
|
|
* more readable fashion. See also IO.sysopen and IO.for_fd.
|
|
*
|
|
* IO.new is called by various File and IO opening methods such as IO::open,
|
|
* Kernel#open, and File::open.
|
|
*
|
|
* === Open Mode
|
|
*
|
|
* When +mode+ is an integer it must be combination of the modes defined in
|
|
* File::Constants (+File::RDONLY+, <code>File::WRONLY|File::CREAT</code>).
|
|
* See the open(2) man page for more information.
|
|
*
|
|
* When +mode+ is a string it must be in one of the following forms:
|
|
*
|
|
* fmode
|
|
* fmode ":" ext_enc
|
|
* fmode ":" ext_enc ":" int_enc
|
|
* fmode ":" "BOM|UTF-*"
|
|
*
|
|
* +fmode+ is an IO open mode string, +ext_enc+ is the external encoding for
|
|
* the IO and +int_enc+ is the internal encoding.
|
|
*
|
|
* ==== IO Open Mode
|
|
*
|
|
* Ruby allows the following open modes:
|
|
*
|
|
* "r" Read-only, starts at beginning of file (default mode).
|
|
*
|
|
* "r+" Read-write, starts at beginning of file.
|
|
*
|
|
* "w" Write-only, truncates existing file
|
|
* to zero length or creates a new file for writing.
|
|
*
|
|
* "w+" Read-write, truncates existing file to zero length
|
|
* or creates a new file for reading and writing.
|
|
*
|
|
* "a" Write-only, each write call appends data at end of file.
|
|
* Creates a new file for writing if file does not exist.
|
|
*
|
|
* "a+" Read-write, each write call appends data at end of file.
|
|
* Creates a new file for reading and writing if file does
|
|
* not exist.
|
|
*
|
|
* The following modes must be used separately, and along with one or more of
|
|
* the modes seen above.
|
|
*
|
|
* "b" Binary file mode
|
|
* Suppresses EOL <-> CRLF conversion on Windows. And
|
|
* sets external encoding to ASCII-8BIT unless explicitly
|
|
* specified.
|
|
*
|
|
* "t" Text file mode
|
|
*
|
|
* The exclusive access mode ("x") can be used together with "w" to ensure
|
|
* the file is created. Errno::EEXIST is raised when it already exists.
|
|
* It may not be supported with all kinds of streams (e.g. pipes).
|
|
*
|
|
* When the open mode of original IO is read only, the mode cannot be
|
|
* changed to be writable. Similarly, the open mode cannot be changed from
|
|
* write only to readable.
|
|
*
|
|
* When such a change is attempted the error is raised in different locations
|
|
* according to the platform.
|
|
*
|
|
* === IO Encoding
|
|
*
|
|
* When +ext_enc+ is specified, strings read will be tagged by the encoding
|
|
* when reading, and strings output will be converted to the specified
|
|
* encoding when writing.
|
|
*
|
|
* When +ext_enc+ and +int_enc+ are specified read strings will be converted
|
|
* from +ext_enc+ to +int_enc+ upon input, and written strings will be
|
|
* converted from +int_enc+ to +ext_enc+ upon output. See Encoding for
|
|
* further details of transcoding on input and output.
|
|
*
|
|
* If "BOM|UTF-8", "BOM|UTF-16LE" or "BOM|UTF16-BE" are used, Ruby checks for
|
|
* a Unicode BOM in the input document to help determine the encoding. For
|
|
* UTF-16 encodings the file open mode must be binary. When present, the BOM
|
|
* is stripped and the external encoding from the BOM is used. When the BOM
|
|
* is missing the given Unicode encoding is used as +ext_enc+. (The BOM-set
|
|
* encoding option is case insensitive, so "bom|utf-8" is also valid.)
|
|
*
|
|
* === Options
|
|
*
|
|
* +opt+ can be used instead of +mode+ for improved readability. The
|
|
* following keys are supported:
|
|
*
|
|
* :mode ::
|
|
* Same as +mode+ parameter
|
|
*
|
|
* :flags ::
|
|
* Specifies file open flags as integer.
|
|
* If +mode+ parameter is given, this parameter will be bitwise-ORed.
|
|
*
|
|
* :\external_encoding ::
|
|
* External encoding for the IO.
|
|
*
|
|
* :\internal_encoding ::
|
|
* Internal encoding for the IO. "-" is a synonym for the default internal
|
|
* encoding.
|
|
*
|
|
* If the value is +nil+ no conversion occurs.
|
|
*
|
|
* :encoding ::
|
|
* Specifies external and internal encodings as "extern:intern".
|
|
*
|
|
* :textmode ::
|
|
* If the value is truth value, same as "t" in argument +mode+.
|
|
*
|
|
* :binmode ::
|
|
* If the value is truth value, same as "b" in argument +mode+.
|
|
*
|
|
* :autoclose ::
|
|
* If the value is +false+, the +fd+ will be kept open after this IO
|
|
* instance gets finalized.
|
|
*
|
|
* Also, +opt+ can have same keys in String#encode for controlling conversion
|
|
* between the external encoding and the internal encoding.
|
|
*
|
|
* === Example 1
|
|
*
|
|
* fd = IO.sysopen("/dev/tty", "w")
|
|
* a = IO.new(fd,"w")
|
|
* $stderr.puts "Hello"
|
|
* a.puts "World"
|
|
*
|
|
* Produces:
|
|
*
|
|
* Hello
|
|
* World
|
|
*
|
|
* === Example 2
|
|
*
|
|
* require 'fcntl'
|
|
*
|
|
* fd = STDERR.fcntl(Fcntl::F_DUPFD)
|
|
* io = IO.new(fd, mode: 'w:UTF-16LE', cr_newline: true)
|
|
* io.puts "Hello, World!"
|
|
*
|
|
* fd = STDERR.fcntl(Fcntl::F_DUPFD)
|
|
* io = IO.new(fd, mode: 'w', cr_newline: true,
|
|
* external_encoding: Encoding::UTF_16LE)
|
|
* io.puts "Hello, World!"
|
|
*
|
|
* Both of above print "Hello, World!" in UTF-16LE to standard error output
|
|
* with converting EOL generated by #puts to CR.
|
|
*/
|
|
|
|
static VALUE
|
|
rb_io_initialize(int argc, VALUE *argv, VALUE io)
|
|
{
|
|
VALUE fnum, vmode;
|
|
rb_io_t *fp;
|
|
int fd, fmode, oflags = O_RDONLY;
|
|
convconfig_t convconfig;
|
|
VALUE opt;
|
|
#if defined(HAVE_FCNTL) && defined(F_GETFL)
|
|
int ofmode;
|
|
#else
|
|
struct stat st;
|
|
#endif
|
|
|
|
|
|
argc = rb_scan_args(argc, argv, "11:", &fnum, &vmode, &opt);
|
|
rb_io_extract_modeenc(&vmode, 0, opt, &oflags, &fmode, &convconfig);
|
|
|
|
fd = NUM2INT(fnum);
|
|
if (rb_reserved_fd_p(fd)) {
|
|
rb_raise(rb_eArgError, "The given fd is not accessible because RubyVM reserves it");
|
|
}
|
|
#if defined(HAVE_FCNTL) && defined(F_GETFL)
|
|
oflags = fcntl(fd, F_GETFL);
|
|
if (oflags == -1) rb_sys_fail(0);
|
|
#else
|
|
if (fstat(fd, &st) < 0) rb_sys_fail(0);
|
|
#endif
|
|
rb_update_max_fd(fd);
|
|
#if defined(HAVE_FCNTL) && defined(F_GETFL)
|
|
ofmode = rb_io_oflags_fmode(oflags);
|
|
if (NIL_P(vmode)) {
|
|
fmode = ofmode;
|
|
}
|
|
else if ((~ofmode & fmode) & FMODE_READWRITE) {
|
|
VALUE error = INT2FIX(EINVAL);
|
|
rb_exc_raise(rb_class_new_instance(1, &error, rb_eSystemCallError));
|
|
}
|
|
#endif
|
|
if (!NIL_P(opt) && rb_hash_aref(opt, sym_autoclose) == Qfalse) {
|
|
fmode |= FMODE_PREP;
|
|
}
|
|
MakeOpenFile(io, fp);
|
|
fp->fd = fd;
|
|
fp->mode = fmode;
|
|
fp->encs = convconfig;
|
|
clear_codeconv(fp);
|
|
io_check_tty(fp);
|
|
if (fileno(stdin) == fd)
|
|
fp->stdio_file = stdin;
|
|
else if (fileno(stdout) == fd)
|
|
fp->stdio_file = stdout;
|
|
else if (fileno(stderr) == fd)
|
|
fp->stdio_file = stderr;
|
|
|
|
if (fmode & FMODE_SETENC_BY_BOM) io_set_encoding_by_bom(io);
|
|
return io;
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* ios.set_encoding_by_bom -> encoding or nil
|
|
*
|
|
* Checks if +ios+ starts with a BOM, and then consumes it and sets
|
|
* the external encoding. Returns the result encoding if found, or
|
|
* nil. If +ios+ is not binmode or its encoding has been set
|
|
* already, an exception will be raised.
|
|
*
|
|
* File.write("bom.txt", "\u{FEFF}abc")
|
|
* ios = File.open("bom.txt", "rb")
|
|
* ios.set_encoding_by_bom #=> #<Encoding:UTF-8>
|
|
*
|
|
* File.write("nobom.txt", "abc")
|
|
* ios = File.open("nobom.txt", "rb")
|
|
* ios.set_encoding_by_bom #=> nil
|
|
*/
|
|
|
|
static VALUE
|
|
rb_io_set_encoding_by_bom(VALUE io)
|
|
{
|
|
rb_io_t *fptr;
|
|
|
|
GetOpenFile(io, fptr);
|
|
if (!(fptr->mode & FMODE_BINMODE)) {
|
|
rb_raise(rb_eArgError, "ASCII incompatible encoding needs binmode");
|
|
}
|
|
if (fptr->encs.enc2) {
|
|
rb_raise(rb_eArgError, "encoding conversion is set");
|
|
}
|
|
if (!io_set_encoding_by_bom(io)) return Qnil;
|
|
return rb_enc_from_encoding(fptr->encs.enc);
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* File.new(filename, mode="r" [, opt]) -> file
|
|
* File.new(filename [, mode [, perm]] [, opt]) -> file
|
|
*
|
|
* Opens the file named by +filename+ according to the given +mode+ and
|
|
* returns a new File object.
|
|
*
|
|
* See IO.new for a description of +mode+ and +opt+.
|
|
*
|
|
* If a file is being created, permission bits may be given in +perm+. These
|
|
* mode and permission bits are platform dependent; on Unix systems, see
|
|
* open(2) and chmod(2) man pages for details.
|
|
*
|
|
* The new File object is buffered mode (or non-sync mode), unless
|
|
* +filename+ is a tty.
|
|
* See IO#flush, IO#fsync, IO#fdatasync, and IO#sync= about sync mode.
|
|
*
|
|
* === Examples
|
|
*
|
|
* f = File.new("testfile", "r")
|
|
* f = File.new("newfile", "w+")
|
|
* f = File.new("newfile", File::CREAT|File::TRUNC|File::RDWR, 0644)
|
|
*/
|
|
|
|
static VALUE
|
|
rb_file_initialize(int argc, VALUE *argv, VALUE io)
|
|
{
|
|
if (RFILE(io)->fptr) {
|
|
rb_raise(rb_eRuntimeError, "reinitializing File");
|
|
}
|
|
if (0 < argc && argc < 3) {
|
|
VALUE fd = rb_check_to_int(argv[0]);
|
|
|
|
if (!NIL_P(fd)) {
|
|
argv[0] = fd;
|
|
return rb_io_initialize(argc, argv, io);
|
|
}
|
|
}
|
|
rb_open_file(argc, argv, io);
|
|
|
|
return io;
|
|
}
|
|
|
|
/* :nodoc: */
|
|
static VALUE
|
|
rb_io_s_new(int argc, VALUE *argv, VALUE klass)
|
|
{
|
|
if (rb_block_given_p()) {
|
|
VALUE cname = rb_obj_as_string(klass);
|
|
|
|
rb_warn("%"PRIsVALUE"::new() does not take block; use %"PRIsVALUE"::open() instead",
|
|
cname, cname);
|
|
}
|
|
return rb_class_new_instance_kw(argc, argv, klass, RB_PASS_CALLED_KEYWORDS);
|
|
}
|
|
|
|
|
|
/*
|
|
* call-seq:
|
|
* IO.for_fd(fd, mode [, opt]) -> io
|
|
*
|
|
* Synonym for IO.new.
|
|
*
|
|
*/
|
|
|
|
static VALUE
|
|
rb_io_s_for_fd(int argc, VALUE *argv, VALUE klass)
|
|
{
|
|
VALUE io = rb_obj_alloc(klass);
|
|
rb_io_initialize(argc, argv, io);
|
|
return io;
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* ios.autoclose? -> true or false
|
|
*
|
|
* Returns +true+ if the underlying file descriptor of _ios_ will be
|
|
* closed automatically at its finalization, otherwise +false+.
|
|
*/
|
|
|
|
static VALUE
|
|
rb_io_autoclose_p(VALUE io)
|
|
{
|
|
rb_io_t *fptr = RFILE(io)->fptr;
|
|
rb_io_check_closed(fptr);
|
|
return (fptr->mode & FMODE_PREP) ? Qfalse : Qtrue;
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* io.autoclose = bool -> true or false
|
|
*
|
|
* Sets auto-close flag.
|
|
*
|
|
* f = open("/dev/null")
|
|
* IO.for_fd(f.fileno)
|
|
* # ...
|
|
* f.gets # may cause Errno::EBADF
|
|
*
|
|
* f = open("/dev/null")
|
|
* IO.for_fd(f.fileno).autoclose = false
|
|
* # ...
|
|
* f.gets # won't cause Errno::EBADF
|
|
*/
|
|
|
|
static VALUE
|
|
rb_io_set_autoclose(VALUE io, VALUE autoclose)
|
|
{
|
|
rb_io_t *fptr;
|
|
GetOpenFile(io, fptr);
|
|
if (!RTEST(autoclose))
|
|
fptr->mode |= FMODE_PREP;
|
|
else
|
|
fptr->mode &= ~FMODE_PREP;
|
|
return autoclose;
|
|
}
|
|
|
|
static void
|
|
argf_mark(void *ptr)
|
|
{
|
|
struct argf *p = ptr;
|
|
rb_gc_mark(p->filename);
|
|
rb_gc_mark(p->current_file);
|
|
rb_gc_mark(p->argv);
|
|
rb_gc_mark(p->inplace);
|
|
rb_gc_mark(p->encs.ecopts);
|
|
}
|
|
|
|
static size_t
|
|
argf_memsize(const void *ptr)
|
|
{
|
|
const struct argf *p = ptr;
|
|
size_t size = sizeof(*p);
|
|
return size;
|
|
}
|
|
|
|
static const rb_data_type_t argf_type = {
|
|
"ARGF",
|
|
{argf_mark, RUBY_TYPED_DEFAULT_FREE, argf_memsize},
|
|
0, 0, RUBY_TYPED_FREE_IMMEDIATELY
|
|
};
|
|
|
|
static inline void
|
|
argf_init(struct argf *p, VALUE v)
|
|
{
|
|
p->filename = Qnil;
|
|
p->current_file = Qnil;
|
|
p->lineno = 0;
|
|
p->argv = v;
|
|
}
|
|
|
|
static VALUE
|
|
argf_alloc(VALUE klass)
|
|
{
|
|
struct argf *p;
|
|
VALUE argf = TypedData_Make_Struct(klass, struct argf, &argf_type, p);
|
|
|
|
argf_init(p, Qnil);
|
|
return argf;
|
|
}
|
|
|
|
#undef rb_argv
|
|
|
|
/* :nodoc: */
|
|
static VALUE
|
|
argf_initialize(VALUE argf, VALUE argv)
|
|
{
|
|
memset(&ARGF, 0, sizeof(ARGF));
|
|
argf_init(&ARGF, argv);
|
|
|
|
return argf;
|
|
}
|
|
|
|
/* :nodoc: */
|
|
static VALUE
|
|
argf_initialize_copy(VALUE argf, VALUE orig)
|
|
{
|
|
if (!OBJ_INIT_COPY(argf, orig)) return argf;
|
|
ARGF = argf_of(orig);
|
|
ARGF.argv = rb_obj_dup(ARGF.argv);
|
|
return argf;
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* ARGF.lineno = integer -> integer
|
|
*
|
|
* Sets the line number of +ARGF+ as a whole to the given +Integer+.
|
|
*
|
|
* +ARGF+ sets the line number automatically as you read data, so normally
|
|
* you will not need to set it explicitly. To access the current line number
|
|
* use +ARGF.lineno+.
|
|
*
|
|
* For example:
|
|
*
|
|
* ARGF.lineno #=> 0
|
|
* ARGF.readline #=> "This is line 1\n"
|
|
* ARGF.lineno #=> 1
|
|
* ARGF.lineno = 0 #=> 0
|
|
* ARGF.lineno #=> 0
|
|
*/
|
|
static VALUE
|
|
argf_set_lineno(VALUE argf, VALUE val)
|
|
{
|
|
ARGF.lineno = NUM2INT(val);
|
|
ARGF.last_lineno = ARGF.lineno;
|
|
return Qnil;
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* ARGF.lineno -> integer
|
|
*
|
|
* Returns the current line number of ARGF as a whole. This value
|
|
* can be set manually with +ARGF.lineno=+.
|
|
*
|
|
* For example:
|
|
*
|
|
* ARGF.lineno #=> 0
|
|
* ARGF.readline #=> "This is line 1\n"
|
|
* ARGF.lineno #=> 1
|
|
*/
|
|
static VALUE
|
|
argf_lineno(VALUE argf)
|
|
{
|
|
return INT2FIX(ARGF.lineno);
|
|
}
|
|
|
|
static VALUE
|
|
argf_forward(int argc, VALUE *argv, VALUE argf)
|
|
{
|
|
return rb_funcall3(ARGF.current_file, rb_frame_this_func(), argc, argv);
|
|
}
|
|
|
|
#define next_argv() argf_next_argv(argf)
|
|
#define ARGF_GENERIC_INPUT_P() \
|
|
(ARGF.current_file == rb_stdin && !RB_TYPE_P(ARGF.current_file, T_FILE))
|
|
#define ARGF_FORWARD(argc, argv) do {\
|
|
if (ARGF_GENERIC_INPUT_P())\
|
|
return argf_forward((argc), (argv), argf);\
|
|
} while (0)
|
|
#define NEXT_ARGF_FORWARD(argc, argv) do {\
|
|
if (!next_argv()) return Qnil;\
|
|
ARGF_FORWARD((argc), (argv));\
|
|
} while (0)
|
|
|
|
static void
|
|
argf_close(VALUE argf)
|
|
{
|
|
VALUE file = ARGF.current_file;
|
|
if (file == rb_stdin) return;
|
|
if (RB_TYPE_P(file, T_FILE)) {
|
|
rb_io_set_write_io(file, Qnil);
|
|
}
|
|
io_close(file);
|
|
ARGF.init_p = -1;
|
|
}
|
|
|
|
static int
|
|
argf_next_argv(VALUE argf)
|
|
{
|
|
char *fn;
|
|
rb_io_t *fptr;
|
|
int stdout_binmode = 0;
|
|
int fmode;
|
|
|
|
if (RB_TYPE_P(rb_stdout, T_FILE)) {
|
|
GetOpenFile(rb_stdout, fptr);
|
|
if (fptr->mode & FMODE_BINMODE)
|
|
stdout_binmode = 1;
|
|
}
|
|
|
|
if (ARGF.init_p == 0) {
|
|
if (!NIL_P(ARGF.argv) && RARRAY_LEN(ARGF.argv) > 0) {
|
|
ARGF.next_p = 1;
|
|
}
|
|
else {
|
|
ARGF.next_p = -1;
|
|
}
|
|
ARGF.init_p = 1;
|
|
}
|
|
else {
|
|
if (NIL_P(ARGF.argv)) {
|
|
ARGF.next_p = -1;
|
|
}
|
|
else if (ARGF.next_p == -1 && RARRAY_LEN(ARGF.argv) > 0) {
|
|
ARGF.next_p = 1;
|
|
}
|
|
}
|
|
|
|
if (ARGF.next_p == 1) {
|
|
if (ARGF.init_p == 1) argf_close(argf);
|
|
retry:
|
|
if (RARRAY_LEN(ARGF.argv) > 0) {
|
|
VALUE filename = rb_ary_shift(ARGF.argv);
|
|
FilePathValue(filename);
|
|
ARGF.filename = filename;
|
|
filename = rb_str_encode_ospath(filename);
|
|
fn = StringValueCStr(filename);
|
|
if (RSTRING_LEN(filename) == 1 && fn[0] == '-') {
|
|
ARGF.current_file = rb_stdin;
|
|
if (ARGF.inplace) {
|
|
rb_warn("Can't do inplace edit for stdio; skipping");
|
|
goto retry;
|
|
}
|
|
}
|
|
else {
|
|
VALUE write_io = Qnil;
|
|
int fr = rb_sysopen(filename, O_RDONLY, 0);
|
|
|
|
if (ARGF.inplace) {
|
|
struct stat st;
|
|
#ifndef NO_SAFE_RENAME
|
|
struct stat st2;
|
|
#endif
|
|
VALUE str;
|
|
int fw;
|
|
|
|
if (RB_TYPE_P(rb_stdout, T_FILE) && rb_stdout != orig_stdout) {
|
|
rb_io_close(rb_stdout);
|
|
}
|
|
fstat(fr, &st);
|
|
str = filename;
|
|
if (!NIL_P(ARGF.inplace)) {
|
|
VALUE suffix = ARGF.inplace;
|
|
str = rb_str_dup(str);
|
|
if (NIL_P(rb_str_cat_conv_enc_opts(str, RSTRING_LEN(str),
|
|
RSTRING_PTR(suffix), RSTRING_LEN(suffix),
|
|
rb_enc_get(suffix), 0, Qnil))) {
|
|
rb_str_append(str, suffix);
|
|
}
|
|
#ifdef NO_SAFE_RENAME
|
|
(void)close(fr);
|
|
(void)unlink(RSTRING_PTR(str));
|
|
if (rename(fn, RSTRING_PTR(str)) < 0) {
|
|
rb_warn("Can't rename %"PRIsVALUE" to %"PRIsVALUE": %s, skipping file",
|
|
filename, str, strerror(errno));
|
|
goto retry;
|
|
}
|
|
fr = rb_sysopen(str, O_RDONLY, 0);
|
|
#else
|
|
if (rename(fn, RSTRING_PTR(str)) < 0) {
|
|
rb_warn("Can't rename %"PRIsVALUE" to %"PRIsVALUE": %s, skipping file",
|
|
filename, str, strerror(errno));
|
|
close(fr);
|
|
goto retry;
|
|
}
|
|
#endif
|
|
}
|
|
else {
|
|
#ifdef NO_SAFE_RENAME
|
|
rb_fatal("Can't do inplace edit without backup");
|
|
#else
|
|
if (unlink(fn) < 0) {
|
|
rb_warn("Can't remove %"PRIsVALUE": %s, skipping file",
|
|
filename, strerror(errno));
|
|
close(fr);
|
|
goto retry;
|
|
}
|
|
#endif
|
|
}
|
|
fw = rb_sysopen(filename, O_WRONLY|O_CREAT|O_TRUNC, 0666);
|
|
#ifndef NO_SAFE_RENAME
|
|
fstat(fw, &st2);
|
|
#ifdef HAVE_FCHMOD
|
|
fchmod(fw, st.st_mode);
|
|
#else
|
|
chmod(fn, st.st_mode);
|
|
#endif
|
|
if (st.st_uid!=st2.st_uid || st.st_gid!=st2.st_gid) {
|
|
int err;
|
|
#ifdef HAVE_FCHOWN
|
|
err = fchown(fw, st.st_uid, st.st_gid);
|
|
#else
|
|
err = chown(fn, st.st_uid, st.st_gid);
|
|
#endif
|
|
if (err && getuid() == 0 && st2.st_uid == 0) {
|
|
const char *wkfn = RSTRING_PTR(filename);
|
|
rb_warn("Can't set owner/group of %"PRIsVALUE" to same as %"PRIsVALUE": %s, skipping file",
|
|
filename, str, strerror(errno));
|
|
(void)close(fr);
|
|
(void)close(fw);
|
|
(void)unlink(wkfn);
|
|
goto retry;
|
|
}
|
|
}
|
|
#endif
|
|
write_io = prep_io(fw, FMODE_WRITABLE, rb_cFile, fn);
|
|
rb_stdout = write_io;
|
|
if (stdout_binmode) rb_io_binmode(rb_stdout);
|
|
}
|
|
fmode = FMODE_READABLE;
|
|
if (!ARGF.binmode) {
|
|
fmode |= DEFAULT_TEXTMODE;
|
|
}
|
|
ARGF.current_file = prep_io(fr, fmode, rb_cFile, fn);
|
|
if (!NIL_P(write_io)) {
|
|
rb_io_set_write_io(ARGF.current_file, write_io);
|
|
}
|
|
RB_GC_GUARD(filename);
|
|
}
|
|
if (ARGF.binmode) rb_io_ascii8bit_binmode(ARGF.current_file);
|
|
GetOpenFile(ARGF.current_file, fptr);
|
|
if (ARGF.encs.enc) {
|
|
fptr->encs = ARGF.encs;
|
|
clear_codeconv(fptr);
|
|
}
|
|
else {
|
|
fptr->encs.ecflags &= ~ECONV_NEWLINE_DECORATOR_MASK;
|
|
if (!ARGF.binmode) {
|
|
fptr->encs.ecflags |= ECONV_DEFAULT_NEWLINE_DECORATOR;
|
|
#ifdef TEXTMODE_NEWLINE_DECORATOR_ON_WRITE
|
|
fptr->encs.ecflags |= TEXTMODE_NEWLINE_DECORATOR_ON_WRITE;
|
|
#endif
|
|
}
|
|
}
|
|
ARGF.next_p = 0;
|
|
}
|
|
else {
|
|
ARGF.next_p = 1;
|
|
return FALSE;
|
|
}
|
|
}
|
|
else if (ARGF.next_p == -1) {
|
|
ARGF.current_file = rb_stdin;
|
|
ARGF.filename = rb_str_new2("-");
|
|
if (ARGF.inplace) {
|
|
rb_warn("Can't do inplace edit for stdio");
|
|
rb_stdout = orig_stdout;
|
|
}
|
|
}
|
|
if (ARGF.init_p == -1) ARGF.init_p = 1;
|
|
return TRUE;
|
|
}
|
|
|
|
static VALUE
|
|
argf_getline(int argc, VALUE *argv, VALUE argf)
|
|
{
|
|
VALUE line;
|
|
long lineno = ARGF.lineno;
|
|
|
|
retry:
|
|
if (!next_argv()) return Qnil;
|
|
if (ARGF_GENERIC_INPUT_P()) {
|
|
line = rb_funcall3(ARGF.current_file, idGets, argc, argv);
|
|
}
|
|
else {
|
|
if (argc == 0 && rb_rs == rb_default_rs) {
|
|
line = rb_io_gets(ARGF.current_file);
|
|
}
|
|
else {
|
|
line = rb_io_getline(argc, argv, ARGF.current_file);
|
|
}
|
|
if (NIL_P(line) && ARGF.next_p != -1) {
|
|
argf_close(argf);
|
|
ARGF.next_p = 1;
|
|
goto retry;
|
|
}
|
|
}
|
|
if (!NIL_P(line)) {
|
|
ARGF.lineno = ++lineno;
|
|
ARGF.last_lineno = ARGF.lineno;
|
|
}
|
|
return line;
|
|
}
|
|
|
|
static VALUE
|
|
argf_lineno_getter(ID id, VALUE *var)
|
|
{
|
|
VALUE argf = *var;
|
|
return INT2FIX(ARGF.last_lineno);
|
|
}
|
|
|
|
static void
|
|
argf_lineno_setter(VALUE val, ID id, VALUE *var)
|
|
{
|
|
VALUE argf = *var;
|
|
int n = NUM2INT(val);
|
|
ARGF.last_lineno = ARGF.lineno = n;
|
|
}
|
|
|
|
static VALUE argf_gets(int, VALUE *, VALUE);
|
|
|
|
/*
|
|
* call-seq:
|
|
* gets(sep=$/ [, getline_args]) -> string or nil
|
|
* gets(limit [, getline_args]) -> string or nil
|
|
* gets(sep, limit [, getline_args]) -> string or nil
|
|
*
|
|
* Returns (and assigns to <code>$_</code>) the next line from the list
|
|
* of files in +ARGV+ (or <code>$*</code>), or from standard input if
|
|
* no files are present on the command line. Returns +nil+ at end of
|
|
* file. The optional argument specifies the record separator. The
|
|
* separator is included with the contents of each record. A separator
|
|
* of +nil+ reads the entire contents, and a zero-length separator
|
|
* reads the input one paragraph at a time, where paragraphs are
|
|
* divided by two consecutive newlines. If the first argument is an
|
|
* integer, or optional second argument is given, the returning string
|
|
* would not be longer than the given value in bytes. If multiple
|
|
* filenames are present in +ARGV+, <code>gets(nil)</code> will read
|
|
* the contents one file at a time.
|
|
*
|
|
* ARGV << "testfile"
|
|
* print while gets
|
|
*
|
|
* <em>produces:</em>
|
|
*
|
|
* This is line one
|
|
* This is line two
|
|
* This is line three
|
|
* And so on...
|
|
*
|
|
* The style of programming using <code>$_</code> as an implicit
|
|
* parameter is gradually losing favor in the Ruby community.
|
|
*/
|
|
|
|
static VALUE
|
|
rb_f_gets(int argc, VALUE *argv, VALUE recv)
|
|
{
|
|
if (recv == argf) {
|
|
return argf_gets(argc, argv, argf);
|
|
}
|
|
return rb_funcallv(argf, idGets, argc, argv);
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* ARGF.gets(sep=$/ [, getline_args]) -> string or nil
|
|
* ARGF.gets(limit [, getline_args]) -> string or nil
|
|
* ARGF.gets(sep, limit [, getline_args]) -> string or nil
|
|
*
|
|
* Returns the next line from the current file in +ARGF+.
|
|
*
|
|
* By default lines are assumed to be separated by <code>$/</code>;
|
|
* to use a different character as a separator, supply it as a +String+
|
|
* for the _sep_ argument.
|
|
*
|
|
* The optional _limit_ argument specifies how many characters of each line
|
|
* to return. By default all characters are returned.
|
|
*
|
|
* See IO.readlines for details about getline_args.
|
|
*
|
|
*/
|
|
static VALUE
|
|
argf_gets(int argc, VALUE *argv, VALUE argf)
|
|
{
|
|
VALUE line;
|
|
|
|
line = argf_getline(argc, argv, argf);
|
|
rb_lastline_set(line);
|
|
|
|
return line;
|
|
}
|
|
|
|
VALUE
|
|
rb_gets(void)
|
|
{
|
|
VALUE line;
|
|
|
|
if (rb_rs != rb_default_rs) {
|
|
return rb_f_gets(0, 0, argf);
|
|
}
|
|
|
|
retry:
|
|
if (!next_argv()) return Qnil;
|
|
line = rb_io_gets(ARGF.current_file);
|
|
if (NIL_P(line) && ARGF.next_p != -1) {
|
|
rb_io_close(ARGF.current_file);
|
|
ARGF.next_p = 1;
|
|
goto retry;
|
|
}
|
|
rb_lastline_set(line);
|
|
if (!NIL_P(line)) {
|
|
ARGF.lineno++;
|
|
ARGF.last_lineno = ARGF.lineno;
|
|
}
|
|
|
|
return line;
|
|
}
|
|
|
|
static VALUE argf_readline(int, VALUE *, VALUE);
|
|
|
|
/*
|
|
* call-seq:
|
|
* readline(sep=$/) -> string
|
|
* readline(limit) -> string
|
|
* readline(sep, limit) -> string
|
|
*
|
|
* Equivalent to Kernel::gets, except
|
|
* +readline+ raises +EOFError+ at end of file.
|
|
*/
|
|
|
|
static VALUE
|
|
rb_f_readline(int argc, VALUE *argv, VALUE recv)
|
|
{
|
|
if (recv == argf) {
|
|
return argf_readline(argc, argv, argf);
|
|
}
|
|
return rb_funcallv(argf, rb_intern("readline"), argc, argv);
|
|
}
|
|
|
|
|
|
/*
|
|
* call-seq:
|
|
* ARGF.readline(sep=$/) -> string
|
|
* ARGF.readline(limit) -> string
|
|
* ARGF.readline(sep, limit) -> string
|
|
*
|
|
* Returns the next line from the current file in +ARGF+.
|
|
*
|
|
* By default lines are assumed to be separated by <code>$/</code>;
|
|
* to use a different character as a separator, supply it as a +String+
|
|
* for the _sep_ argument.
|
|
*
|
|
* The optional _limit_ argument specifies how many characters of each line
|
|
* to return. By default all characters are returned.
|
|
*
|
|
* An +EOFError+ is raised at the end of the file.
|
|
*/
|
|
static VALUE
|
|
argf_readline(int argc, VALUE *argv, VALUE argf)
|
|
{
|
|
VALUE line;
|
|
|
|
if (!next_argv()) rb_eof_error();
|
|
ARGF_FORWARD(argc, argv);
|
|
line = argf_gets(argc, argv, argf);
|
|
if (NIL_P(line)) {
|
|
rb_eof_error();
|
|
}
|
|
|
|
return line;
|
|
}
|
|
|
|
static VALUE argf_readlines(int, VALUE *, VALUE);
|
|
|
|
/*
|
|
* call-seq:
|
|
* readlines(sep=$/) -> array
|
|
* readlines(limit) -> array
|
|
* readlines(sep, limit) -> array
|
|
*
|
|
* Returns an array containing the lines returned by calling
|
|
* <code>Kernel.gets(<i>sep</i>)</code> until the end of file.
|
|
*/
|
|
|
|
static VALUE
|
|
rb_f_readlines(int argc, VALUE *argv, VALUE recv)
|
|
{
|
|
if (recv == argf) {
|
|
return argf_readlines(argc, argv, argf);
|
|
}
|
|
return rb_funcallv(argf, rb_intern("readlines"), argc, argv);
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* ARGF.readlines(sep=$/) -> array
|
|
* ARGF.readlines(limit) -> array
|
|
* ARGF.readlines(sep, limit) -> array
|
|
*
|
|
* ARGF.to_a(sep=$/) -> array
|
|
* ARGF.to_a(limit) -> array
|
|
* ARGF.to_a(sep, limit) -> array
|
|
*
|
|
* Reads +ARGF+'s current file in its entirety, returning an +Array+ of its
|
|
* lines, one line per element. Lines are assumed to be separated by _sep_.
|
|
*
|
|
* lines = ARGF.readlines
|
|
* lines[0] #=> "This is line one\n"
|
|
*/
|
|
static VALUE
|
|
argf_readlines(int argc, VALUE *argv, VALUE argf)
|
|
{
|
|
long lineno = ARGF.lineno;
|
|
VALUE lines, ary;
|
|
|
|
ary = rb_ary_new();
|
|
while (next_argv()) {
|
|
if (ARGF_GENERIC_INPUT_P()) {
|
|
lines = rb_funcall3(ARGF.current_file, rb_intern("readlines"), argc, argv);
|
|
}
|
|
else {
|
|
lines = rb_io_readlines(argc, argv, ARGF.current_file);
|
|
argf_close(argf);
|
|
}
|
|
ARGF.next_p = 1;
|
|
rb_ary_concat(ary, lines);
|
|
ARGF.lineno = lineno + RARRAY_LEN(ary);
|
|
ARGF.last_lineno = ARGF.lineno;
|
|
}
|
|
ARGF.init_p = 0;
|
|
return ary;
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* `cmd` -> string
|
|
*
|
|
* Returns the standard output of running _cmd_ in a subshell.
|
|
* The built-in syntax <code>%x{...}</code> uses
|
|
* this method. Sets <code>$?</code> to the process status.
|
|
*
|
|
* `date` #=> "Wed Apr 9 08:56:30 CDT 2003\n"
|
|
* `ls testdir`.split[1] #=> "main.rb"
|
|
* `echo oops && exit 99` #=> "oops\n"
|
|
* $?.exitstatus #=> 99
|
|
*/
|
|
|
|
static VALUE
|
|
rb_f_backquote(VALUE obj, VALUE str)
|
|
{
|
|
VALUE port;
|
|
VALUE result;
|
|
rb_io_t *fptr;
|
|
|
|
SafeStringValue(str);
|
|
rb_last_status_clear();
|
|
port = pipe_open_s(str, "r", FMODE_READABLE|DEFAULT_TEXTMODE, NULL);
|
|
if (NIL_P(port)) return rb_str_new(0,0);
|
|
|
|
GetOpenFile(port, fptr);
|
|
result = read_all(fptr, remain_size(fptr), Qnil);
|
|
rb_io_close(port);
|
|
RFILE(port)->fptr = NULL;
|
|
rb_io_fptr_finalize(fptr);
|
|
rb_gc_force_recycle(port); /* also guards from premature GC */
|
|
|
|
return result;
|
|
}
|
|
|
|
#ifdef HAVE_SYS_SELECT_H
|
|
#include <sys/select.h>
|
|
#endif
|
|
|
|
static VALUE
|
|
select_internal(VALUE read, VALUE write, VALUE except, struct timeval *tp, rb_fdset_t *fds)
|
|
{
|
|
VALUE res, list;
|
|
rb_fdset_t *rp, *wp, *ep;
|
|
rb_io_t *fptr;
|
|
long i;
|
|
int max = 0, n;
|
|
int pending = 0;
|
|
struct timeval timerec;
|
|
|
|
if (!NIL_P(read)) {
|
|
Check_Type(read, T_ARRAY);
|
|
for (i=0; i<RARRAY_LEN(read); i++) {
|
|
GetOpenFile(rb_io_get_io(RARRAY_AREF(read, i)), fptr);
|
|
rb_fd_set(fptr->fd, &fds[0]);
|
|
if (READ_DATA_PENDING(fptr) || READ_CHAR_PENDING(fptr)) { /* check for buffered data */
|
|
pending++;
|
|
rb_fd_set(fptr->fd, &fds[3]);
|
|
}
|
|
if (max < fptr->fd) max = fptr->fd;
|
|
}
|
|
if (pending) { /* no blocking if there's buffered data */
|
|
timerec.tv_sec = timerec.tv_usec = 0;
|
|
tp = &timerec;
|
|
}
|
|
rp = &fds[0];
|
|
}
|
|
else
|
|
rp = 0;
|
|
|
|
if (!NIL_P(write)) {
|
|
Check_Type(write, T_ARRAY);
|
|
for (i=0; i<RARRAY_LEN(write); i++) {
|
|
VALUE write_io = GetWriteIO(rb_io_get_io(RARRAY_AREF(write, i)));
|
|
GetOpenFile(write_io, fptr);
|
|
rb_fd_set(fptr->fd, &fds[1]);
|
|
if (max < fptr->fd) max = fptr->fd;
|
|
}
|
|
wp = &fds[1];
|
|
}
|
|
else
|
|
wp = 0;
|
|
|
|
if (!NIL_P(except)) {
|
|
Check_Type(except, T_ARRAY);
|
|
for (i=0; i<RARRAY_LEN(except); i++) {
|
|
VALUE io = rb_io_get_io(RARRAY_AREF(except, i));
|
|
VALUE write_io = GetWriteIO(io);
|
|
GetOpenFile(io, fptr);
|
|
rb_fd_set(fptr->fd, &fds[2]);
|
|
if (max < fptr->fd) max = fptr->fd;
|
|
if (io != write_io) {
|
|
GetOpenFile(write_io, fptr);
|
|
rb_fd_set(fptr->fd, &fds[2]);
|
|
if (max < fptr->fd) max = fptr->fd;
|
|
}
|
|
}
|
|
ep = &fds[2];
|
|
}
|
|
else {
|
|
ep = 0;
|
|
}
|
|
|
|
max++;
|
|
|
|
n = rb_thread_fd_select(max, rp, wp, ep, tp);
|
|
if (n < 0) {
|
|
rb_sys_fail(0);
|
|
}
|
|
if (!pending && n == 0) return Qnil; /* returns nil on timeout */
|
|
|
|
res = rb_ary_new2(3);
|
|
rb_ary_push(res, rp?rb_ary_new():rb_ary_new2(0));
|
|
rb_ary_push(res, wp?rb_ary_new():rb_ary_new2(0));
|
|
rb_ary_push(res, ep?rb_ary_new():rb_ary_new2(0));
|
|
|
|
if (rp) {
|
|
list = RARRAY_AREF(res, 0);
|
|
for (i=0; i< RARRAY_LEN(read); i++) {
|
|
VALUE obj = rb_ary_entry(read, i);
|
|
VALUE io = rb_io_get_io(obj);
|
|
GetOpenFile(io, fptr);
|
|
if (rb_fd_isset(fptr->fd, &fds[0]) ||
|
|
rb_fd_isset(fptr->fd, &fds[3])) {
|
|
rb_ary_push(list, obj);
|
|
}
|
|
}
|
|
}
|
|
|
|
if (wp) {
|
|
list = RARRAY_AREF(res, 1);
|
|
for (i=0; i< RARRAY_LEN(write); i++) {
|
|
VALUE obj = rb_ary_entry(write, i);
|
|
VALUE io = rb_io_get_io(obj);
|
|
VALUE write_io = GetWriteIO(io);
|
|
GetOpenFile(write_io, fptr);
|
|
if (rb_fd_isset(fptr->fd, &fds[1])) {
|
|
rb_ary_push(list, obj);
|
|
}
|
|
}
|
|
}
|
|
|
|
if (ep) {
|
|
list = RARRAY_AREF(res, 2);
|
|
for (i=0; i< RARRAY_LEN(except); i++) {
|
|
VALUE obj = rb_ary_entry(except, i);
|
|
VALUE io = rb_io_get_io(obj);
|
|
VALUE write_io = GetWriteIO(io);
|
|
GetOpenFile(io, fptr);
|
|
if (rb_fd_isset(fptr->fd, &fds[2])) {
|
|
rb_ary_push(list, obj);
|
|
}
|
|
else if (io != write_io) {
|
|
GetOpenFile(write_io, fptr);
|
|
if (rb_fd_isset(fptr->fd, &fds[2])) {
|
|
rb_ary_push(list, obj);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
return res; /* returns an empty array on interrupt */
|
|
}
|
|
|
|
struct select_args {
|
|
VALUE read, write, except;
|
|
struct timeval *timeout;
|
|
rb_fdset_t fdsets[4];
|
|
};
|
|
|
|
static VALUE
|
|
select_call(VALUE arg)
|
|
{
|
|
struct select_args *p = (struct select_args *)arg;
|
|
|
|
return select_internal(p->read, p->write, p->except, p->timeout, p->fdsets);
|
|
}
|
|
|
|
static VALUE
|
|
select_end(VALUE arg)
|
|
{
|
|
struct select_args *p = (struct select_args *)arg;
|
|
int i;
|
|
|
|
for (i = 0; i < numberof(p->fdsets); ++i)
|
|
rb_fd_term(&p->fdsets[i]);
|
|
return Qnil;
|
|
}
|
|
|
|
static VALUE sym_normal, sym_sequential, sym_random,
|
|
sym_willneed, sym_dontneed, sym_noreuse;
|
|
|
|
#ifdef HAVE_POSIX_FADVISE
|
|
struct io_advise_struct {
|
|
int fd;
|
|
int advice;
|
|
off_t offset;
|
|
off_t len;
|
|
};
|
|
|
|
static VALUE
|
|
io_advise_internal(void *arg)
|
|
{
|
|
struct io_advise_struct *ptr = arg;
|
|
return posix_fadvise(ptr->fd, ptr->offset, ptr->len, ptr->advice);
|
|
}
|
|
|
|
static VALUE
|
|
io_advise_sym_to_const(VALUE sym)
|
|
{
|
|
#ifdef POSIX_FADV_NORMAL
|
|
if (sym == sym_normal)
|
|
return INT2NUM(POSIX_FADV_NORMAL);
|
|
#endif
|
|
|
|
#ifdef POSIX_FADV_RANDOM
|
|
if (sym == sym_random)
|
|
return INT2NUM(POSIX_FADV_RANDOM);
|
|
#endif
|
|
|
|
#ifdef POSIX_FADV_SEQUENTIAL
|
|
if (sym == sym_sequential)
|
|
return INT2NUM(POSIX_FADV_SEQUENTIAL);
|
|
#endif
|
|
|
|
#ifdef POSIX_FADV_WILLNEED
|
|
if (sym == sym_willneed)
|
|
return INT2NUM(POSIX_FADV_WILLNEED);
|
|
#endif
|
|
|
|
#ifdef POSIX_FADV_DONTNEED
|
|
if (sym == sym_dontneed)
|
|
return INT2NUM(POSIX_FADV_DONTNEED);
|
|
#endif
|
|
|
|
#ifdef POSIX_FADV_NOREUSE
|
|
if (sym == sym_noreuse)
|
|
return INT2NUM(POSIX_FADV_NOREUSE);
|
|
#endif
|
|
|
|
return Qnil;
|
|
}
|
|
|
|
static VALUE
|
|
do_io_advise(rb_io_t *fptr, VALUE advice, off_t offset, off_t len)
|
|
{
|
|
int rv;
|
|
struct io_advise_struct ias;
|
|
VALUE num_adv;
|
|
|
|
num_adv = io_advise_sym_to_const(advice);
|
|
|
|
/*
|
|
* The platform doesn't support this hint. We don't raise exception, instead
|
|
* silently ignore it. Because IO::advise is only hint.
|
|
*/
|
|
if (NIL_P(num_adv))
|
|
return Qnil;
|
|
|
|
ias.fd = fptr->fd;
|
|
ias.advice = NUM2INT(num_adv);
|
|
ias.offset = offset;
|
|
ias.len = len;
|
|
|
|
rv = (int)rb_thread_io_blocking_region(io_advise_internal, &ias, fptr->fd);
|
|
if (rv && rv != ENOSYS) {
|
|
/* posix_fadvise(2) doesn't set errno. On success it returns 0; otherwise
|
|
it returns the error code. */
|
|
VALUE message = rb_sprintf("%"PRIsVALUE" "
|
|
"(%"PRI_OFFT_PREFIX"d, "
|
|
"%"PRI_OFFT_PREFIX"d, "
|
|
"%"PRIsVALUE")",
|
|
fptr->pathv, offset, len, advice);
|
|
rb_syserr_fail_str(rv, message);
|
|
}
|
|
|
|
return Qnil;
|
|
}
|
|
|
|
#endif /* HAVE_POSIX_FADVISE */
|
|
|
|
static void
|
|
advice_arg_check(VALUE advice)
|
|
{
|
|
if (!SYMBOL_P(advice))
|
|
rb_raise(rb_eTypeError, "advice must be a Symbol");
|
|
|
|
if (advice != sym_normal &&
|
|
advice != sym_sequential &&
|
|
advice != sym_random &&
|
|
advice != sym_willneed &&
|
|
advice != sym_dontneed &&
|
|
advice != sym_noreuse) {
|
|
rb_raise(rb_eNotImpError, "Unsupported advice: %+"PRIsVALUE, advice);
|
|
}
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* ios.advise(advice, offset=0, len=0) -> nil
|
|
*
|
|
* Announce an intention to access data from the current file in a
|
|
* specific pattern. On platforms that do not support the
|
|
* <em>posix_fadvise(2)</em> system call, this method is a no-op.
|
|
*
|
|
* _advice_ is one of the following symbols:
|
|
*
|
|
* :normal:: No advice to give; the default assumption for an open file.
|
|
* :sequential:: The data will be accessed sequentially
|
|
* with lower offsets read before higher ones.
|
|
* :random:: The data will be accessed in random order.
|
|
* :willneed:: The data will be accessed in the near future.
|
|
* :dontneed:: The data will not be accessed in the near future.
|
|
* :noreuse:: The data will only be accessed once.
|
|
*
|
|
* The semantics of a piece of advice are platform-dependent. See
|
|
* <em>man 2 posix_fadvise</em> for details.
|
|
*
|
|
* "data" means the region of the current file that begins at
|
|
* _offset_ and extends for _len_ bytes. If _len_ is 0, the region
|
|
* ends at the last byte of the file. By default, both _offset_ and
|
|
* _len_ are 0, meaning that the advice applies to the entire file.
|
|
*
|
|
* If an error occurs, one of the following exceptions will be raised:
|
|
*
|
|
* IOError:: The IO stream is closed.
|
|
* Errno::EBADF::
|
|
* The file descriptor of the current file is invalid.
|
|
* Errno::EINVAL:: An invalid value for _advice_ was given.
|
|
* Errno::ESPIPE::
|
|
* The file descriptor of the current file refers to a FIFO or
|
|
* pipe. (Linux raises Errno::EINVAL in this case).
|
|
* TypeError::
|
|
* Either _advice_ was not a Symbol, or one of the
|
|
* other arguments was not an Integer.
|
|
* RangeError:: One of the arguments given was too big/small.
|
|
*
|
|
* This list is not exhaustive; other Errno:: exceptions are also possible.
|
|
*/
|
|
static VALUE
|
|
rb_io_advise(int argc, VALUE *argv, VALUE io)
|
|
{
|
|
VALUE advice, offset, len;
|
|
off_t off, l;
|
|
rb_io_t *fptr;
|
|
|
|
rb_scan_args(argc, argv, "12", &advice, &offset, &len);
|
|
advice_arg_check(advice);
|
|
|
|
io = GetWriteIO(io);
|
|
GetOpenFile(io, fptr);
|
|
|
|
off = NIL_P(offset) ? 0 : NUM2OFFT(offset);
|
|
l = NIL_P(len) ? 0 : NUM2OFFT(len);
|
|
|
|
#ifdef HAVE_POSIX_FADVISE
|
|
return do_io_advise(fptr, advice, off, l);
|
|
#else
|
|
((void)off, (void)l); /* Ignore all hint */
|
|
return Qnil;
|
|
#endif
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* IO.select(read_array [, write_array [, error_array [, timeout]]]) -> array or nil
|
|
*
|
|
* Calls select(2) system call.
|
|
* It monitors given arrays of IO objects, waits until one or more of
|
|
* IO objects are ready for reading, are ready for writing, and have
|
|
* pending exceptions respectively, and returns an array that contains
|
|
* arrays of those IO objects. It will return +nil+ if optional
|
|
* <i>timeout</i> value is given and no IO object is ready in
|
|
* <i>timeout</i> seconds.
|
|
*
|
|
* IO.select peeks the buffer of IO objects for testing readability.
|
|
* If the IO buffer is not empty, IO.select immediately notifies
|
|
* readability. This "peek" only happens for IO objects. It does not
|
|
* happen for IO-like objects such as OpenSSL::SSL::SSLSocket.
|
|
*
|
|
* The best way to use IO.select is invoking it after nonblocking
|
|
* methods such as #read_nonblock, #write_nonblock, etc. The methods
|
|
* raise an exception which is extended by IO::WaitReadable or
|
|
* IO::WaitWritable. The modules notify how the caller should wait
|
|
* with IO.select. If IO::WaitReadable is raised, the caller should
|
|
* wait for reading. If IO::WaitWritable is raised, the caller should
|
|
* wait for writing.
|
|
*
|
|
* So, blocking read (#readpartial) can be emulated using
|
|
* #read_nonblock and IO.select as follows:
|
|
*
|
|
* begin
|
|
* result = io_like.read_nonblock(maxlen)
|
|
* rescue IO::WaitReadable
|
|
* IO.select([io_like])
|
|
* retry
|
|
* rescue IO::WaitWritable
|
|
* IO.select(nil, [io_like])
|
|
* retry
|
|
* end
|
|
*
|
|
* Especially, the combination of nonblocking methods and IO.select is
|
|
* preferred for IO like objects such as OpenSSL::SSL::SSLSocket. It
|
|
* has #to_io method to return underlying IO object. IO.select calls
|
|
* #to_io to obtain the file descriptor to wait.
|
|
*
|
|
* This means that readability notified by IO.select doesn't mean
|
|
* readability from OpenSSL::SSL::SSLSocket object.
|
|
*
|
|
* The most likely situation is that OpenSSL::SSL::SSLSocket buffers
|
|
* some data. IO.select doesn't see the buffer. So IO.select can
|
|
* block when OpenSSL::SSL::SSLSocket#readpartial doesn't block.
|
|
*
|
|
* However, several more complicated situations exist.
|
|
*
|
|
* SSL is a protocol which is sequence of records.
|
|
* The record consists of multiple bytes.
|
|
* So, the remote side of SSL sends a partial record, IO.select
|
|
* notifies readability but OpenSSL::SSL::SSLSocket cannot decrypt a
|
|
* byte and OpenSSL::SSL::SSLSocket#readpartial will block.
|
|
*
|
|
* Also, the remote side can request SSL renegotiation which forces
|
|
* the local SSL engine to write some data.
|
|
* This means OpenSSL::SSL::SSLSocket#readpartial may invoke #write
|
|
* system call and it can block.
|
|
* In such a situation, OpenSSL::SSL::SSLSocket#read_nonblock raises
|
|
* IO::WaitWritable instead of blocking.
|
|
* So, the caller should wait for ready for writability as above
|
|
* example.
|
|
*
|
|
* The combination of nonblocking methods and IO.select is also useful
|
|
* for streams such as tty, pipe socket socket when multiple processes
|
|
* read from a stream.
|
|
*
|
|
* Finally, Linux kernel developers don't guarantee that
|
|
* readability of select(2) means readability of following read(2) even
|
|
* for a single process.
|
|
* See select(2) manual on GNU/Linux system.
|
|
*
|
|
* Invoking IO.select before IO#readpartial works well as usual.
|
|
* However it is not the best way to use IO.select.
|
|
*
|
|
* The writability notified by select(2) doesn't show
|
|
* how many bytes are writable.
|
|
* IO#write method blocks until given whole string is written.
|
|
* So, <code>IO#write(two or more bytes)</code> can block after
|
|
* writability is notified by IO.select. IO#write_nonblock is required
|
|
* to avoid the blocking.
|
|
*
|
|
* Blocking write (#write) can be emulated using #write_nonblock and
|
|
* IO.select as follows: IO::WaitReadable should also be rescued for
|
|
* SSL renegotiation in OpenSSL::SSL::SSLSocket.
|
|
*
|
|
* while 0 < string.bytesize
|
|
* begin
|
|
* written = io_like.write_nonblock(string)
|
|
* rescue IO::WaitReadable
|
|
* IO.select([io_like])
|
|
* retry
|
|
* rescue IO::WaitWritable
|
|
* IO.select(nil, [io_like])
|
|
* retry
|
|
* end
|
|
* string = string.byteslice(written..-1)
|
|
* end
|
|
*
|
|
* === Parameters
|
|
* read_array:: an array of IO objects that wait until ready for read
|
|
* write_array:: an array of IO objects that wait until ready for write
|
|
* error_array:: an array of IO objects that wait for exceptions
|
|
* timeout:: a numeric value in second
|
|
*
|
|
* === Example
|
|
*
|
|
* rp, wp = IO.pipe
|
|
* mesg = "ping "
|
|
* 100.times {
|
|
* # IO.select follows IO#read. Not the best way to use IO.select.
|
|
* rs, ws, = IO.select([rp], [wp])
|
|
* if r = rs[0]
|
|
* ret = r.read(5)
|
|
* print ret
|
|
* case ret
|
|
* when /ping/
|
|
* mesg = "pong\n"
|
|
* when /pong/
|
|
* mesg = "ping "
|
|
* end
|
|
* end
|
|
* if w = ws[0]
|
|
* w.write(mesg)
|
|
* end
|
|
* }
|
|
*
|
|
* <em>produces:</em>
|
|
*
|
|
* ping pong
|
|
* ping pong
|
|
* ping pong
|
|
* (snipped)
|
|
* ping
|
|
*/
|
|
|
|
static VALUE
|
|
rb_f_select(int argc, VALUE *argv, VALUE obj)
|
|
{
|
|
VALUE timeout;
|
|
struct select_args args;
|
|
struct timeval timerec;
|
|
int i;
|
|
|
|
rb_scan_args(argc, argv, "13", &args.read, &args.write, &args.except, &timeout);
|
|
if (NIL_P(timeout)) {
|
|
args.timeout = 0;
|
|
}
|
|
else {
|
|
timerec = rb_time_interval(timeout);
|
|
args.timeout = &timerec;
|
|
}
|
|
|
|
for (i = 0; i < numberof(args.fdsets); ++i)
|
|
rb_fd_init(&args.fdsets[i]);
|
|
|
|
return rb_ensure(select_call, (VALUE)&args, select_end, (VALUE)&args);
|
|
}
|
|
|
|
#if (defined(__linux__) && !defined(__ANDROID__)) || defined(__FreeBSD__) || defined(__NetBSD__) || defined(__OpenBSD__) || defined(__APPLE__)
|
|
typedef unsigned long ioctl_req_t;
|
|
# define NUM2IOCTLREQ(num) NUM2ULONG(num)
|
|
#else
|
|
typedef int ioctl_req_t;
|
|
# define NUM2IOCTLREQ(num) NUM2INT(num)
|
|
#endif
|
|
|
|
#ifdef HAVE_IOCTL
|
|
struct ioctl_arg {
|
|
int fd;
|
|
ioctl_req_t cmd;
|
|
long narg;
|
|
};
|
|
|
|
static VALUE
|
|
nogvl_ioctl(void *ptr)
|
|
{
|
|
struct ioctl_arg *arg = ptr;
|
|
|
|
return (VALUE)ioctl(arg->fd, arg->cmd, arg->narg);
|
|
}
|
|
|
|
static int
|
|
do_ioctl(int fd, ioctl_req_t cmd, long narg)
|
|
{
|
|
int retval;
|
|
struct ioctl_arg arg;
|
|
|
|
arg.fd = fd;
|
|
arg.cmd = cmd;
|
|
arg.narg = narg;
|
|
|
|
retval = (int)rb_thread_io_blocking_region(nogvl_ioctl, &arg, fd);
|
|
|
|
return retval;
|
|
}
|
|
#endif
|
|
|
|
#define DEFULT_IOCTL_NARG_LEN (256)
|
|
|
|
#if defined(__linux__) && defined(_IOC_SIZE)
|
|
static long
|
|
linux_iocparm_len(ioctl_req_t cmd)
|
|
{
|
|
long len;
|
|
|
|
if ((cmd & 0xFFFF0000) == 0) {
|
|
/* legacy and unstructured ioctl number. */
|
|
return DEFULT_IOCTL_NARG_LEN;
|
|
}
|
|
|
|
len = _IOC_SIZE(cmd);
|
|
|
|
/* paranoia check for silly drivers which don't keep ioctl convention */
|
|
if (len < DEFULT_IOCTL_NARG_LEN)
|
|
len = DEFULT_IOCTL_NARG_LEN;
|
|
|
|
return len;
|
|
}
|
|
#endif
|
|
|
|
static long
|
|
ioctl_narg_len(ioctl_req_t cmd)
|
|
{
|
|
long len;
|
|
|
|
#ifdef IOCPARM_MASK
|
|
#ifndef IOCPARM_LEN
|
|
#define IOCPARM_LEN(x) (((x) >> 16) & IOCPARM_MASK)
|
|
#endif
|
|
#endif
|
|
#ifdef IOCPARM_LEN
|
|
len = IOCPARM_LEN(cmd); /* on BSDish systems we're safe */
|
|
#elif defined(__linux__) && defined(_IOC_SIZE)
|
|
len = linux_iocparm_len(cmd);
|
|
#else
|
|
/* otherwise guess at what's safe */
|
|
len = DEFULT_IOCTL_NARG_LEN;
|
|
#endif
|
|
|
|
return len;
|
|
}
|
|
|
|
#ifdef HAVE_FCNTL
|
|
#ifdef __linux__
|
|
typedef long fcntl_arg_t;
|
|
#else
|
|
/* posix */
|
|
typedef int fcntl_arg_t;
|
|
#endif
|
|
|
|
static long
|
|
fcntl_narg_len(int cmd)
|
|
{
|
|
long len;
|
|
|
|
switch (cmd) {
|
|
#ifdef F_DUPFD
|
|
case F_DUPFD:
|
|
len = sizeof(fcntl_arg_t);
|
|
break;
|
|
#endif
|
|
#ifdef F_DUP2FD /* bsd specific */
|
|
case F_DUP2FD:
|
|
len = sizeof(int);
|
|
break;
|
|
#endif
|
|
#ifdef F_DUPFD_CLOEXEC /* linux specific */
|
|
case F_DUPFD_CLOEXEC:
|
|
len = sizeof(fcntl_arg_t);
|
|
break;
|
|
#endif
|
|
#ifdef F_GETFD
|
|
case F_GETFD:
|
|
len = 1;
|
|
break;
|
|
#endif
|
|
#ifdef F_SETFD
|
|
case F_SETFD:
|
|
len = sizeof(fcntl_arg_t);
|
|
break;
|
|
#endif
|
|
#ifdef F_GETFL
|
|
case F_GETFL:
|
|
len = 1;
|
|
break;
|
|
#endif
|
|
#ifdef F_SETFL
|
|
case F_SETFL:
|
|
len = sizeof(fcntl_arg_t);
|
|
break;
|
|
#endif
|
|
#ifdef F_GETOWN
|
|
case F_GETOWN:
|
|
len = 1;
|
|
break;
|
|
#endif
|
|
#ifdef F_SETOWN
|
|
case F_SETOWN:
|
|
len = sizeof(fcntl_arg_t);
|
|
break;
|
|
#endif
|
|
#ifdef F_GETOWN_EX /* linux specific */
|
|
case F_GETOWN_EX:
|
|
len = sizeof(struct f_owner_ex);
|
|
break;
|
|
#endif
|
|
#ifdef F_SETOWN_EX /* linux specific */
|
|
case F_SETOWN_EX:
|
|
len = sizeof(struct f_owner_ex);
|
|
break;
|
|
#endif
|
|
#ifdef F_GETLK
|
|
case F_GETLK:
|
|
len = sizeof(struct flock);
|
|
break;
|
|
#endif
|
|
#ifdef F_SETLK
|
|
case F_SETLK:
|
|
len = sizeof(struct flock);
|
|
break;
|
|
#endif
|
|
#ifdef F_SETLKW
|
|
case F_SETLKW:
|
|
len = sizeof(struct flock);
|
|
break;
|
|
#endif
|
|
#ifdef F_READAHEAD /* bsd specific */
|
|
case F_READAHEAD:
|
|
len = sizeof(int);
|
|
break;
|
|
#endif
|
|
#ifdef F_RDAHEAD /* Darwin specific */
|
|
case F_RDAHEAD:
|
|
len = sizeof(int);
|
|
break;
|
|
#endif
|
|
#ifdef F_GETSIG /* linux specific */
|
|
case F_GETSIG:
|
|
len = 1;
|
|
break;
|
|
#endif
|
|
#ifdef F_SETSIG /* linux specific */
|
|
case F_SETSIG:
|
|
len = sizeof(fcntl_arg_t);
|
|
break;
|
|
#endif
|
|
#ifdef F_GETLEASE /* linux specific */
|
|
case F_GETLEASE:
|
|
len = 1;
|
|
break;
|
|
#endif
|
|
#ifdef F_SETLEASE /* linux specific */
|
|
case F_SETLEASE:
|
|
len = sizeof(fcntl_arg_t);
|
|
break;
|
|
#endif
|
|
#ifdef F_NOTIFY /* linux specific */
|
|
case F_NOTIFY:
|
|
len = sizeof(fcntl_arg_t);
|
|
break;
|
|
#endif
|
|
|
|
default:
|
|
len = 256;
|
|
break;
|
|
}
|
|
|
|
return len;
|
|
}
|
|
#else /* HAVE_FCNTL */
|
|
static long
|
|
fcntl_narg_len(int cmd)
|
|
{
|
|
return 0;
|
|
}
|
|
#endif /* HAVE_FCNTL */
|
|
|
|
static long
|
|
setup_narg(ioctl_req_t cmd, VALUE *argp, int io_p)
|
|
{
|
|
long narg = 0;
|
|
VALUE arg = *argp;
|
|
|
|
if (NIL_P(arg) || arg == Qfalse) {
|
|
narg = 0;
|
|
}
|
|
else if (FIXNUM_P(arg)) {
|
|
narg = FIX2LONG(arg);
|
|
}
|
|
else if (arg == Qtrue) {
|
|
narg = 1;
|
|
}
|
|
else {
|
|
VALUE tmp = rb_check_string_type(arg);
|
|
|
|
if (NIL_P(tmp)) {
|
|
narg = NUM2LONG(arg);
|
|
}
|
|
else {
|
|
char *ptr;
|
|
long len, slen;
|
|
|
|
*argp = arg = tmp;
|
|
if (io_p)
|
|
len = ioctl_narg_len(cmd);
|
|
else
|
|
len = fcntl_narg_len((int)cmd);
|
|
rb_str_modify(arg);
|
|
|
|
slen = RSTRING_LEN(arg);
|
|
/* expand for data + sentinel. */
|
|
if (slen < len+1) {
|
|
rb_str_resize(arg, len+1);
|
|
MEMZERO(RSTRING_PTR(arg)+slen, char, len-slen);
|
|
slen = len+1;
|
|
}
|
|
/* a little sanity check here */
|
|
ptr = RSTRING_PTR(arg);
|
|
ptr[slen - 1] = 17;
|
|
narg = (long)(SIGNED_VALUE)ptr;
|
|
}
|
|
}
|
|
|
|
return narg;
|
|
}
|
|
|
|
#ifdef HAVE_IOCTL
|
|
static VALUE
|
|
rb_ioctl(VALUE io, VALUE req, VALUE arg)
|
|
{
|
|
ioctl_req_t cmd = NUM2IOCTLREQ(req);
|
|
rb_io_t *fptr;
|
|
long narg;
|
|
int retval;
|
|
|
|
narg = setup_narg(cmd, &arg, 1);
|
|
GetOpenFile(io, fptr);
|
|
retval = do_ioctl(fptr->fd, cmd, narg);
|
|
if (retval < 0) rb_sys_fail_path(fptr->pathv);
|
|
if (RB_TYPE_P(arg, T_STRING)) {
|
|
char *ptr;
|
|
long slen;
|
|
RSTRING_GETMEM(arg, ptr, slen);
|
|
if (ptr[slen-1] != 17)
|
|
rb_raise(rb_eArgError, "return value overflowed string");
|
|
ptr[slen-1] = '\0';
|
|
}
|
|
|
|
return INT2NUM(retval);
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* ios.ioctl(integer_cmd, arg) -> integer
|
|
*
|
|
* Provides a mechanism for issuing low-level commands to control or
|
|
* query I/O devices. Arguments and results are platform dependent. If
|
|
* <i>arg</i> is a number, its value is passed directly. If it is a
|
|
* string, it is interpreted as a binary sequence of bytes. On Unix
|
|
* platforms, see <code>ioctl(2)</code> for details. Not implemented on
|
|
* all platforms.
|
|
*/
|
|
|
|
static VALUE
|
|
rb_io_ioctl(int argc, VALUE *argv, VALUE io)
|
|
{
|
|
VALUE req, arg;
|
|
|
|
rb_scan_args(argc, argv, "11", &req, &arg);
|
|
return rb_ioctl(io, req, arg);
|
|
}
|
|
#else
|
|
#define rb_io_ioctl rb_f_notimplement
|
|
#endif
|
|
|
|
#ifdef HAVE_FCNTL
|
|
struct fcntl_arg {
|
|
int fd;
|
|
int cmd;
|
|
long narg;
|
|
};
|
|
|
|
static VALUE
|
|
nogvl_fcntl(void *ptr)
|
|
{
|
|
struct fcntl_arg *arg = ptr;
|
|
|
|
#if defined(F_DUPFD)
|
|
if (arg->cmd == F_DUPFD)
|
|
return (VALUE)rb_cloexec_fcntl_dupfd(arg->fd, (int)arg->narg);
|
|
#endif
|
|
return (VALUE)fcntl(arg->fd, arg->cmd, arg->narg);
|
|
}
|
|
|
|
static int
|
|
do_fcntl(int fd, int cmd, long narg)
|
|
{
|
|
int retval;
|
|
struct fcntl_arg arg;
|
|
|
|
arg.fd = fd;
|
|
arg.cmd = cmd;
|
|
arg.narg = narg;
|
|
|
|
retval = (int)rb_thread_io_blocking_region(nogvl_fcntl, &arg, fd);
|
|
if (retval != -1) {
|
|
switch (cmd) {
|
|
#if defined(F_DUPFD)
|
|
case F_DUPFD:
|
|
#endif
|
|
#if defined(F_DUPFD_CLOEXEC)
|
|
case F_DUPFD_CLOEXEC:
|
|
#endif
|
|
rb_update_max_fd(retval);
|
|
}
|
|
}
|
|
|
|
return retval;
|
|
}
|
|
|
|
static VALUE
|
|
rb_fcntl(VALUE io, VALUE req, VALUE arg)
|
|
{
|
|
int cmd = NUM2INT(req);
|
|
rb_io_t *fptr;
|
|
long narg;
|
|
int retval;
|
|
|
|
narg = setup_narg(cmd, &arg, 0);
|
|
GetOpenFile(io, fptr);
|
|
retval = do_fcntl(fptr->fd, cmd, narg);
|
|
if (retval < 0) rb_sys_fail_path(fptr->pathv);
|
|
if (RB_TYPE_P(arg, T_STRING)) {
|
|
char *ptr;
|
|
long slen;
|
|
RSTRING_GETMEM(arg, ptr, slen);
|
|
if (ptr[slen-1] != 17)
|
|
rb_raise(rb_eArgError, "return value overflowed string");
|
|
ptr[slen-1] = '\0';
|
|
}
|
|
|
|
return INT2NUM(retval);
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* ios.fcntl(integer_cmd, arg) -> integer
|
|
*
|
|
* Provides a mechanism for issuing low-level commands to control or
|
|
* query file-oriented I/O streams. Arguments and results are platform
|
|
* dependent. If <i>arg</i> is a number, its value is passed
|
|
* directly. If it is a string, it is interpreted as a binary sequence
|
|
* of bytes (Array#pack might be a useful way to build this string). On
|
|
* Unix platforms, see <code>fcntl(2)</code> for details. Not
|
|
* implemented on all platforms.
|
|
*/
|
|
|
|
static VALUE
|
|
rb_io_fcntl(int argc, VALUE *argv, VALUE io)
|
|
{
|
|
VALUE req, arg;
|
|
|
|
rb_scan_args(argc, argv, "11", &req, &arg);
|
|
return rb_fcntl(io, req, arg);
|
|
}
|
|
#else
|
|
#define rb_io_fcntl rb_f_notimplement
|
|
#endif
|
|
|
|
#if defined(HAVE_SYSCALL) || defined(HAVE___SYSCALL)
|
|
/*
|
|
* call-seq:
|
|
* syscall(num [, args...]) -> integer
|
|
*
|
|
* Calls the operating system function identified by _num_ and
|
|
* returns the result of the function or raises SystemCallError if
|
|
* it failed.
|
|
*
|
|
* Arguments for the function can follow _num_. They must be either
|
|
* +String+ objects or +Integer+ objects. A +String+ object is passed
|
|
* as a pointer to the byte sequence. An +Integer+ object is passed
|
|
* as an integer whose bit size is same as a pointer.
|
|
* Up to nine parameters may be passed.
|
|
*
|
|
* The function identified by _num_ is system
|
|
* dependent. On some Unix systems, the numbers may be obtained from a
|
|
* header file called <code>syscall.h</code>.
|
|
*
|
|
* syscall 4, 1, "hello\n", 6 # '4' is write(2) on our box
|
|
*
|
|
* <em>produces:</em>
|
|
*
|
|
* hello
|
|
*
|
|
* Calling +syscall+ on a platform which does not have any way to
|
|
* an arbitrary system function just fails with NotImplementedError.
|
|
*
|
|
* *Note:*
|
|
* +syscall+ is essentially unsafe and unportable.
|
|
* Feel free to shoot your foot.
|
|
* The DL (Fiddle) library is preferred for safer and a bit
|
|
* more portable programming.
|
|
*/
|
|
|
|
static VALUE
|
|
rb_f_syscall(int argc, VALUE *argv, VALUE _)
|
|
{
|
|
VALUE arg[8];
|
|
#if SIZEOF_VOIDP == 8 && defined(HAVE___SYSCALL) && SIZEOF_INT != 8 /* mainly *BSD */
|
|
# define SYSCALL __syscall
|
|
# define NUM2SYSCALLID(x) NUM2LONG(x)
|
|
# define RETVAL2NUM(x) LONG2NUM(x)
|
|
# if SIZEOF_LONG == 8
|
|
long num, retval = -1;
|
|
# elif SIZEOF_LONG_LONG == 8
|
|
long long num, retval = -1;
|
|
# else
|
|
# error ---->> it is asserted that __syscall takes the first argument and returns retval in 64bit signed integer. <<----
|
|
# endif
|
|
#elif defined(__linux__)
|
|
# define SYSCALL syscall
|
|
# define NUM2SYSCALLID(x) NUM2LONG(x)
|
|
# define RETVAL2NUM(x) LONG2NUM(x)
|
|
/*
|
|
* Linux man page says, syscall(2) function prototype is below.
|
|
*
|
|
* int syscall(int number, ...);
|
|
*
|
|
* But, it's incorrect. Actual one takes and returned long. (see unistd.h)
|
|
*/
|
|
long num, retval = -1;
|
|
#else
|
|
# define SYSCALL syscall
|
|
# define NUM2SYSCALLID(x) NUM2INT(x)
|
|
# define RETVAL2NUM(x) INT2NUM(x)
|
|
int num, retval = -1;
|
|
#endif
|
|
int i;
|
|
|
|
if (RTEST(ruby_verbose)) {
|
|
rb_warning("We plan to remove a syscall function at future release. DL(Fiddle) provides safer alternative.");
|
|
}
|
|
|
|
if (argc == 0)
|
|
rb_raise(rb_eArgError, "too few arguments for syscall");
|
|
if (argc > numberof(arg))
|
|
rb_raise(rb_eArgError, "too many arguments for syscall");
|
|
num = NUM2SYSCALLID(argv[0]); ++argv;
|
|
for (i = argc - 1; i--; ) {
|
|
VALUE v = rb_check_string_type(argv[i]);
|
|
|
|
if (!NIL_P(v)) {
|
|
SafeStringValue(v);
|
|
rb_str_modify(v);
|
|
arg[i] = (VALUE)StringValueCStr(v);
|
|
}
|
|
else {
|
|
arg[i] = (VALUE)NUM2LONG(argv[i]);
|
|
}
|
|
}
|
|
|
|
switch (argc) {
|
|
case 1:
|
|
retval = SYSCALL(num);
|
|
break;
|
|
case 2:
|
|
retval = SYSCALL(num, arg[0]);
|
|
break;
|
|
case 3:
|
|
retval = SYSCALL(num, arg[0],arg[1]);
|
|
break;
|
|
case 4:
|
|
retval = SYSCALL(num, arg[0],arg[1],arg[2]);
|
|
break;
|
|
case 5:
|
|
retval = SYSCALL(num, arg[0],arg[1],arg[2],arg[3]);
|
|
break;
|
|
case 6:
|
|
retval = SYSCALL(num, arg[0],arg[1],arg[2],arg[3],arg[4]);
|
|
break;
|
|
case 7:
|
|
retval = SYSCALL(num, arg[0],arg[1],arg[2],arg[3],arg[4],arg[5]);
|
|
break;
|
|
case 8:
|
|
retval = SYSCALL(num, arg[0],arg[1],arg[2],arg[3],arg[4],arg[5],arg[6]);
|
|
break;
|
|
}
|
|
|
|
if (retval == -1)
|
|
rb_sys_fail(0);
|
|
return RETVAL2NUM(retval);
|
|
#undef SYSCALL
|
|
#undef NUM2SYSCALLID
|
|
#undef RETVAL2NUM
|
|
}
|
|
#else
|
|
#define rb_f_syscall rb_f_notimplement
|
|
#endif
|
|
|
|
static VALUE
|
|
io_new_instance(VALUE args)
|
|
{
|
|
return rb_class_new_instance(2, (VALUE*)args+1, *(VALUE*)args);
|
|
}
|
|
|
|
static rb_encoding *
|
|
find_encoding(VALUE v)
|
|
{
|
|
rb_encoding *enc = rb_find_encoding(v);
|
|
if (!enc) rb_warn("Unsupported encoding %"PRIsVALUE" ignored", v);
|
|
return enc;
|
|
}
|
|
|
|
static void
|
|
io_encoding_set(rb_io_t *fptr, VALUE v1, VALUE v2, VALUE opt)
|
|
{
|
|
rb_encoding *enc, *enc2;
|
|
int ecflags = fptr->encs.ecflags;
|
|
VALUE ecopts, tmp;
|
|
|
|
if (!NIL_P(v2)) {
|
|
enc2 = find_encoding(v1);
|
|
tmp = rb_check_string_type(v2);
|
|
if (!NIL_P(tmp)) {
|
|
if (RSTRING_LEN(tmp) == 1 && RSTRING_PTR(tmp)[0] == '-') {
|
|
/* Special case - "-" => no transcoding */
|
|
enc = enc2;
|
|
enc2 = NULL;
|
|
}
|
|
else
|
|
enc = find_encoding(v2);
|
|
if (enc == enc2) {
|
|
/* Special case - "-" => no transcoding */
|
|
enc2 = NULL;
|
|
}
|
|
}
|
|
else {
|
|
enc = find_encoding(v2);
|
|
if (enc == enc2) {
|
|
/* Special case - "-" => no transcoding */
|
|
enc2 = NULL;
|
|
}
|
|
}
|
|
SET_UNIVERSAL_NEWLINE_DECORATOR_IF_ENC2(enc2, ecflags);
|
|
ecflags = rb_econv_prepare_options(opt, &ecopts, ecflags);
|
|
}
|
|
else {
|
|
if (NIL_P(v1)) {
|
|
/* Set to default encodings */
|
|
rb_io_ext_int_to_encs(NULL, NULL, &enc, &enc2, 0);
|
|
SET_UNIVERSAL_NEWLINE_DECORATOR_IF_ENC2(enc2, ecflags);
|
|
ecopts = Qnil;
|
|
}
|
|
else {
|
|
tmp = rb_check_string_type(v1);
|
|
if (!NIL_P(tmp) && rb_enc_asciicompat(enc = rb_enc_get(tmp))) {
|
|
parse_mode_enc(RSTRING_PTR(tmp), enc, &enc, &enc2, NULL);
|
|
SET_UNIVERSAL_NEWLINE_DECORATOR_IF_ENC2(enc2, ecflags);
|
|
ecflags = rb_econv_prepare_options(opt, &ecopts, ecflags);
|
|
}
|
|
else {
|
|
rb_io_ext_int_to_encs(find_encoding(v1), NULL, &enc, &enc2, 0);
|
|
SET_UNIVERSAL_NEWLINE_DECORATOR_IF_ENC2(enc2, ecflags);
|
|
ecopts = Qnil;
|
|
}
|
|
}
|
|
}
|
|
validate_enc_binmode(&fptr->mode, ecflags, enc, enc2);
|
|
fptr->encs.enc = enc;
|
|
fptr->encs.enc2 = enc2;
|
|
fptr->encs.ecflags = ecflags;
|
|
fptr->encs.ecopts = ecopts;
|
|
clear_codeconv(fptr);
|
|
|
|
}
|
|
|
|
struct io_encoding_set_args {
|
|
rb_io_t *fptr;
|
|
VALUE v1;
|
|
VALUE v2;
|
|
VALUE opt;
|
|
};
|
|
|
|
static VALUE
|
|
io_encoding_set_v(VALUE v)
|
|
{
|
|
struct io_encoding_set_args *arg = (struct io_encoding_set_args *)v;
|
|
io_encoding_set(arg->fptr, arg->v1, arg->v2, arg->opt);
|
|
return Qnil;
|
|
}
|
|
|
|
static VALUE
|
|
pipe_pair_close(VALUE rw)
|
|
{
|
|
VALUE *rwp = (VALUE *)rw;
|
|
return rb_ensure(io_close, rwp[0], io_close, rwp[1]);
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* IO.pipe -> [read_io, write_io]
|
|
* IO.pipe(ext_enc) -> [read_io, write_io]
|
|
* IO.pipe("ext_enc:int_enc" [, opt]) -> [read_io, write_io]
|
|
* IO.pipe(ext_enc, int_enc [, opt]) -> [read_io, write_io]
|
|
*
|
|
* IO.pipe(...) {|read_io, write_io| ... }
|
|
*
|
|
* Creates a pair of pipe endpoints (connected to each other) and
|
|
* returns them as a two-element array of IO objects:
|
|
* <code>[</code> <i>read_io</i>, <i>write_io</i> <code>]</code>.
|
|
*
|
|
* If a block is given, the block is called and
|
|
* returns the value of the block.
|
|
* <i>read_io</i> and <i>write_io</i> are sent to the block as arguments.
|
|
* If read_io and write_io are not closed when the block exits, they are closed.
|
|
* i.e. closing read_io and/or write_io doesn't cause an error.
|
|
*
|
|
* Not available on all platforms.
|
|
*
|
|
* If an encoding (encoding name or encoding object) is specified as an optional argument,
|
|
* read string from pipe is tagged with the encoding specified.
|
|
* If the argument is a colon separated two encoding names "A:B",
|
|
* the read string is converted from encoding A (external encoding)
|
|
* to encoding B (internal encoding), then tagged with B.
|
|
* If two optional arguments are specified, those must be
|
|
* encoding objects or encoding names,
|
|
* and the first one is the external encoding,
|
|
* and the second one is the internal encoding.
|
|
* If the external encoding and the internal encoding is specified,
|
|
* optional hash argument specify the conversion option.
|
|
*
|
|
* In the example below, the two processes close the ends of the pipe
|
|
* that they are not using. This is not just a cosmetic nicety. The
|
|
* read end of a pipe will not generate an end of file condition if
|
|
* there are any writers with the pipe still open. In the case of the
|
|
* parent process, the <code>rd.read</code> will never return if it
|
|
* does not first issue a <code>wr.close</code>.
|
|
*
|
|
* rd, wr = IO.pipe
|
|
*
|
|
* if fork
|
|
* wr.close
|
|
* puts "Parent got: <#{rd.read}>"
|
|
* rd.close
|
|
* Process.wait
|
|
* else
|
|
* rd.close
|
|
* puts "Sending message to parent"
|
|
* wr.write "Hi Dad"
|
|
* wr.close
|
|
* end
|
|
*
|
|
* <em>produces:</em>
|
|
*
|
|
* Sending message to parent
|
|
* Parent got: <Hi Dad>
|
|
*/
|
|
|
|
static VALUE
|
|
rb_io_s_pipe(int argc, VALUE *argv, VALUE klass)
|
|
{
|
|
int pipes[2], state;
|
|
VALUE r, w, args[3], v1, v2;
|
|
VALUE opt;
|
|
rb_io_t *fptr, *fptr2;
|
|
struct io_encoding_set_args ies_args;
|
|
int fmode = 0;
|
|
VALUE ret;
|
|
|
|
argc = rb_scan_args(argc, argv, "02:", &v1, &v2, &opt);
|
|
if (rb_pipe(pipes) < 0)
|
|
rb_sys_fail(0);
|
|
|
|
args[0] = klass;
|
|
args[1] = INT2NUM(pipes[0]);
|
|
args[2] = INT2FIX(O_RDONLY);
|
|
r = rb_protect(io_new_instance, (VALUE)args, &state);
|
|
if (state) {
|
|
close(pipes[0]);
|
|
close(pipes[1]);
|
|
rb_jump_tag(state);
|
|
}
|
|
GetOpenFile(r, fptr);
|
|
|
|
ies_args.fptr = fptr;
|
|
ies_args.v1 = v1;
|
|
ies_args.v2 = v2;
|
|
ies_args.opt = opt;
|
|
rb_protect(io_encoding_set_v, (VALUE)&ies_args, &state);
|
|
if (state) {
|
|
close(pipes[1]);
|
|
io_close(r);
|
|
rb_jump_tag(state);
|
|
}
|
|
|
|
args[1] = INT2NUM(pipes[1]);
|
|
args[2] = INT2FIX(O_WRONLY);
|
|
w = rb_protect(io_new_instance, (VALUE)args, &state);
|
|
if (state) {
|
|
close(pipes[1]);
|
|
if (!NIL_P(r)) rb_io_close(r);
|
|
rb_jump_tag(state);
|
|
}
|
|
GetOpenFile(w, fptr2);
|
|
rb_io_synchronized(fptr2);
|
|
|
|
extract_binmode(opt, &fmode);
|
|
|
|
if ((fmode & FMODE_BINMODE) && v1 == Qnil) {
|
|
rb_io_ascii8bit_binmode(r);
|
|
rb_io_ascii8bit_binmode(w);
|
|
}
|
|
|
|
#if DEFAULT_TEXTMODE
|
|
if ((fptr->mode & FMODE_TEXTMODE) && (fmode & FMODE_BINMODE)) {
|
|
fptr->mode &= ~FMODE_TEXTMODE;
|
|
setmode(fptr->fd, O_BINARY);
|
|
}
|
|
#if defined(RUBY_TEST_CRLF_ENVIRONMENT) || defined(_WIN32)
|
|
if (fptr->encs.ecflags & ECONV_DEFAULT_NEWLINE_DECORATOR) {
|
|
fptr->encs.ecflags |= ECONV_UNIVERSAL_NEWLINE_DECORATOR;
|
|
}
|
|
#endif
|
|
#endif
|
|
fptr->mode |= fmode;
|
|
#if DEFAULT_TEXTMODE
|
|
if ((fptr2->mode & FMODE_TEXTMODE) && (fmode & FMODE_BINMODE)) {
|
|
fptr2->mode &= ~FMODE_TEXTMODE;
|
|
setmode(fptr2->fd, O_BINARY);
|
|
}
|
|
#endif
|
|
fptr2->mode |= fmode;
|
|
|
|
ret = rb_assoc_new(r, w);
|
|
if (rb_block_given_p()) {
|
|
VALUE rw[2];
|
|
rw[0] = r;
|
|
rw[1] = w;
|
|
return rb_ensure(rb_yield, ret, pipe_pair_close, (VALUE)rw);
|
|
}
|
|
return ret;
|
|
}
|
|
|
|
struct foreach_arg {
|
|
int argc;
|
|
VALUE *argv;
|
|
VALUE io;
|
|
};
|
|
|
|
static void
|
|
open_key_args(VALUE klass, int argc, VALUE *argv, VALUE opt, struct foreach_arg *arg)
|
|
{
|
|
VALUE path, v;
|
|
VALUE vmode = Qnil, vperm = Qnil;
|
|
|
|
path = *argv++;
|
|
argc--;
|
|
FilePathValue(path);
|
|
arg->io = 0;
|
|
arg->argc = argc;
|
|
arg->argv = argv;
|
|
if (NIL_P(opt)) {
|
|
vmode = INT2NUM(O_RDONLY);
|
|
vperm = INT2FIX(0666);
|
|
}
|
|
else if (!NIL_P(v = rb_hash_aref(opt, sym_open_args))) {
|
|
int n;
|
|
|
|
v = rb_to_array_type(v);
|
|
n = RARRAY_LENINT(v);
|
|
rb_check_arity(n, 0, 3); /* rb_io_open */
|
|
rb_scan_args_kw(RB_SCAN_ARGS_LAST_HASH_KEYWORDS, n, RARRAY_CONST_PTR(v), "02:", &vmode, &vperm, &opt);
|
|
}
|
|
arg->io = rb_io_open(klass, path, vmode, vperm, opt);
|
|
}
|
|
|
|
static VALUE
|
|
io_s_foreach(VALUE v)
|
|
{
|
|
struct getline_arg *arg = (void *)v;
|
|
VALUE str;
|
|
|
|
while (!NIL_P(str = rb_io_getline_1(arg->rs, arg->limit, arg->chomp, arg->io))) {
|
|
rb_lastline_set(str);
|
|
rb_yield(str);
|
|
}
|
|
rb_lastline_set(Qnil);
|
|
return Qnil;
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* IO.foreach(name, sep=$/ [, getline_args, open_args]) {|line| block } -> nil
|
|
* IO.foreach(name, limit [, getline_args, open_args]) {|line| block } -> nil
|
|
* IO.foreach(name, sep, limit [, getline_args, open_args]) {|line| block } -> nil
|
|
* IO.foreach(...) -> an_enumerator
|
|
*
|
|
* Executes the block for every line in the named I/O port, where lines
|
|
* are separated by <em>sep</em>.
|
|
*
|
|
* If no block is given, an enumerator is returned instead.
|
|
*
|
|
* IO.foreach("testfile") {|x| print "GOT ", x }
|
|
*
|
|
* <em>produces:</em>
|
|
*
|
|
* GOT This is line one
|
|
* GOT This is line two
|
|
* GOT This is line three
|
|
* GOT And so on...
|
|
*
|
|
* If the last argument is a hash, it's the keyword argument to open.
|
|
* See IO.readlines for details about getline_args.
|
|
* And see also IO.read for details about open_args.
|
|
*
|
|
*/
|
|
|
|
static VALUE
|
|
rb_io_s_foreach(int argc, VALUE *argv, VALUE self)
|
|
{
|
|
VALUE opt;
|
|
int orig_argc = argc;
|
|
struct foreach_arg arg;
|
|
struct getline_arg garg;
|
|
|
|
argc = rb_scan_args(argc, argv, "13:", NULL, NULL, NULL, NULL, &opt);
|
|
RETURN_ENUMERATOR(self, orig_argc, argv);
|
|
extract_getline_args(argc-1, argv+1, &garg);
|
|
open_key_args(self, argc, argv, opt, &arg);
|
|
if (NIL_P(arg.io)) return Qnil;
|
|
extract_getline_opts(opt, &garg);
|
|
check_getline_args(&garg.rs, &garg.limit, garg.io = arg.io);
|
|
return rb_ensure(io_s_foreach, (VALUE)&garg, rb_io_close, arg.io);
|
|
}
|
|
|
|
static VALUE
|
|
io_s_readlines(VALUE v)
|
|
{
|
|
struct getline_arg *arg = (void *)v;
|
|
return io_readlines(arg, arg->io);
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* IO.readlines(name, sep=$/ [, getline_args, open_args]) -> array
|
|
* IO.readlines(name, limit [, getline_args, open_args]) -> array
|
|
* IO.readlines(name, sep, limit [, getline_args, open_args]) -> array
|
|
*
|
|
* Reads the entire file specified by <i>name</i> as individual
|
|
* lines, and returns those lines in an array. Lines are separated by
|
|
* <i>sep</i>.
|
|
*
|
|
* a = IO.readlines("testfile")
|
|
* a[0] #=> "This is line one\n"
|
|
*
|
|
* b = IO.readlines("testfile", chomp: true)
|
|
* b[0] #=> "This is line one"
|
|
*
|
|
* If the last argument is a hash, it's the keyword argument to open.
|
|
*
|
|
* === Options for getline
|
|
*
|
|
* The options hash accepts the following keys:
|
|
*
|
|
* :chomp::
|
|
* When the optional +chomp+ keyword argument has a true value,
|
|
* <code>\n</code>, <code>\r</code>, and <code>\r\n</code>
|
|
* will be removed from the end of each line.
|
|
*
|
|
* See also IO.read for details about open_args.
|
|
*/
|
|
|
|
static VALUE
|
|
rb_io_s_readlines(int argc, VALUE *argv, VALUE io)
|
|
{
|
|
VALUE opt;
|
|
struct foreach_arg arg;
|
|
struct getline_arg garg;
|
|
|
|
argc = rb_scan_args(argc, argv, "13:", NULL, NULL, NULL, NULL, &opt);
|
|
extract_getline_args(argc-1, argv+1, &garg);
|
|
open_key_args(io, argc, argv, opt, &arg);
|
|
if (NIL_P(arg.io)) return Qnil;
|
|
extract_getline_opts(opt, &garg);
|
|
check_getline_args(&garg.rs, &garg.limit, garg.io = arg.io);
|
|
return rb_ensure(io_s_readlines, (VALUE)&garg, rb_io_close, arg.io);
|
|
}
|
|
|
|
static VALUE
|
|
io_s_read(VALUE v)
|
|
{
|
|
struct foreach_arg *arg = (void *)v;
|
|
return io_read(arg->argc, arg->argv, arg->io);
|
|
}
|
|
|
|
struct seek_arg {
|
|
VALUE io;
|
|
VALUE offset;
|
|
int mode;
|
|
};
|
|
|
|
static VALUE
|
|
seek_before_access(VALUE argp)
|
|
{
|
|
struct seek_arg *arg = (struct seek_arg *)argp;
|
|
rb_io_binmode(arg->io);
|
|
return rb_io_seek(arg->io, arg->offset, arg->mode);
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* IO.read(name, [length [, offset]] [, opt] ) -> string
|
|
*
|
|
* Opens the file, optionally seeks to the given +offset+, then returns
|
|
* +length+ bytes (defaulting to the rest of the file). #read ensures
|
|
* the file is closed before returning.
|
|
*
|
|
* If +name+ starts with a pipe character (<code>"|"</code>), a subprocess is
|
|
* created in the same way as Kernel#open, and its output is returned.
|
|
*
|
|
* === Options
|
|
*
|
|
* The options hash accepts the following keys:
|
|
*
|
|
* :encoding::
|
|
* string or encoding
|
|
*
|
|
* Specifies the encoding of the read string. +:encoding+ will be ignored
|
|
* if +length+ is specified. See Encoding.aliases for possible encodings.
|
|
*
|
|
* :mode::
|
|
* string or integer
|
|
*
|
|
* Specifies the <i>mode</i> argument for open(). It must start
|
|
* with an "r", otherwise it will cause an error.
|
|
* See IO.new for the list of possible modes.
|
|
*
|
|
* :open_args::
|
|
* array
|
|
*
|
|
* Specifies arguments for open() as an array. This key can not be used
|
|
* in combination with either +:encoding+ or +:mode+.
|
|
*
|
|
* Examples:
|
|
*
|
|
* IO.read("testfile") #=> "This is line one\nThis is line two\nThis is line three\nAnd so on...\n"
|
|
* IO.read("testfile", 20) #=> "This is line one\nThi"
|
|
* IO.read("testfile", 20, 10) #=> "ne one\nThis is line "
|
|
* IO.read("binfile", mode: "rb") #=> "\xF7\x00\x00\x0E\x12"
|
|
*/
|
|
|
|
static VALUE
|
|
rb_io_s_read(int argc, VALUE *argv, VALUE io)
|
|
{
|
|
VALUE opt, offset;
|
|
struct foreach_arg arg;
|
|
|
|
argc = rb_scan_args(argc, argv, "13:", NULL, NULL, &offset, NULL, &opt);
|
|
open_key_args(io, argc, argv, opt, &arg);
|
|
if (NIL_P(arg.io)) return Qnil;
|
|
if (!NIL_P(offset)) {
|
|
struct seek_arg sarg;
|
|
int state = 0;
|
|
sarg.io = arg.io;
|
|
sarg.offset = offset;
|
|
sarg.mode = SEEK_SET;
|
|
rb_protect(seek_before_access, (VALUE)&sarg, &state);
|
|
if (state) {
|
|
rb_io_close(arg.io);
|
|
rb_jump_tag(state);
|
|
}
|
|
if (arg.argc == 2) arg.argc = 1;
|
|
}
|
|
return rb_ensure(io_s_read, (VALUE)&arg, rb_io_close, arg.io);
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* IO.binread(name, [length [, offset]] ) -> string
|
|
*
|
|
* Opens the file, optionally seeks to the given <i>offset</i>, then
|
|
* returns <i>length</i> bytes (defaulting to the rest of the file).
|
|
* #binread ensures the file is closed before returning. The open mode
|
|
* would be <code>"rb:ASCII-8BIT"</code>.
|
|
*
|
|
* IO.binread("testfile") #=> "This is line one\nThis is line two\nThis is line three\nAnd so on...\n"
|
|
* IO.binread("testfile", 20) #=> "This is line one\nThi"
|
|
* IO.binread("testfile", 20, 10) #=> "ne one\nThis is line "
|
|
*/
|
|
|
|
static VALUE
|
|
rb_io_s_binread(int argc, VALUE *argv, VALUE io)
|
|
{
|
|
VALUE offset;
|
|
struct foreach_arg arg;
|
|
enum {
|
|
fmode = FMODE_READABLE|FMODE_BINMODE,
|
|
oflags = O_RDONLY
|
|
#ifdef O_BINARY
|
|
|O_BINARY
|
|
#endif
|
|
};
|
|
convconfig_t convconfig = {NULL, NULL, 0, Qnil};
|
|
|
|
rb_scan_args(argc, argv, "12", NULL, NULL, &offset);
|
|
FilePathValue(argv[0]);
|
|
convconfig.enc = rb_ascii8bit_encoding();
|
|
arg.io = rb_io_open_generic(io, argv[0], oflags, fmode, &convconfig, 0);
|
|
if (NIL_P(arg.io)) return Qnil;
|
|
arg.argv = argv+1;
|
|
arg.argc = (argc > 1) ? 1 : 0;
|
|
if (!NIL_P(offset)) {
|
|
struct seek_arg sarg;
|
|
int state = 0;
|
|
sarg.io = arg.io;
|
|
sarg.offset = offset;
|
|
sarg.mode = SEEK_SET;
|
|
rb_protect(seek_before_access, (VALUE)&sarg, &state);
|
|
if (state) {
|
|
rb_io_close(arg.io);
|
|
rb_jump_tag(state);
|
|
}
|
|
}
|
|
return rb_ensure(io_s_read, (VALUE)&arg, rb_io_close, arg.io);
|
|
}
|
|
|
|
static VALUE
|
|
io_s_write0(VALUE v)
|
|
{
|
|
struct write_arg *arg = (void * )v;
|
|
return io_write(arg->io,arg->str,arg->nosync);
|
|
}
|
|
|
|
static VALUE
|
|
io_s_write(int argc, VALUE *argv, VALUE klass, int binary)
|
|
{
|
|
VALUE string, offset, opt;
|
|
struct foreach_arg arg;
|
|
struct write_arg warg;
|
|
|
|
rb_scan_args(argc, argv, "21:", NULL, &string, &offset, &opt);
|
|
|
|
if (NIL_P(opt)) opt = rb_hash_new();
|
|
else opt = rb_hash_dup(opt);
|
|
|
|
|
|
if (NIL_P(rb_hash_aref(opt,sym_mode))) {
|
|
int mode = O_WRONLY|O_CREAT;
|
|
#ifdef O_BINARY
|
|
if (binary) mode |= O_BINARY;
|
|
#endif
|
|
if (NIL_P(offset)) mode |= O_TRUNC;
|
|
rb_hash_aset(opt,sym_mode,INT2NUM(mode));
|
|
}
|
|
open_key_args(klass, argc, argv, opt, &arg);
|
|
|
|
#ifndef O_BINARY
|
|
if (binary) rb_io_binmode_m(arg.io);
|
|
#endif
|
|
|
|
if (NIL_P(arg.io)) return Qnil;
|
|
if (!NIL_P(offset)) {
|
|
struct seek_arg sarg;
|
|
int state = 0;
|
|
sarg.io = arg.io;
|
|
sarg.offset = offset;
|
|
sarg.mode = SEEK_SET;
|
|
rb_protect(seek_before_access, (VALUE)&sarg, &state);
|
|
if (state) {
|
|
rb_io_close(arg.io);
|
|
rb_jump_tag(state);
|
|
}
|
|
}
|
|
|
|
warg.io = arg.io;
|
|
warg.str = string;
|
|
warg.nosync = 0;
|
|
|
|
return rb_ensure(io_s_write0, (VALUE)&warg, rb_io_close, arg.io);
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* IO.write(name, string [, offset]) -> integer
|
|
* IO.write(name, string [, offset] [, opt]) -> integer
|
|
*
|
|
* Opens the file, optionally seeks to the given <i>offset</i>, writes
|
|
* <i>string</i>, then returns the length written. #write ensures the
|
|
* file is closed before returning. If <i>offset</i> is not given in
|
|
* write mode, the file is truncated. Otherwise, it is not truncated.
|
|
*
|
|
* IO.write("testfile", "0123456789", 20) #=> 10
|
|
* # File could contain: "This is line one\nThi0123456789two\nThis is line three\nAnd so on...\n"
|
|
* IO.write("testfile", "0123456789") #=> 10
|
|
* # File would now read: "0123456789"
|
|
*
|
|
* If the last argument is a hash, it specifies options for the internal
|
|
* open(). It accepts the following keys:
|
|
*
|
|
* :encoding::
|
|
* string or encoding
|
|
*
|
|
* Specifies the encoding of the read string.
|
|
* See Encoding.aliases for possible encodings.
|
|
*
|
|
* :mode::
|
|
* string or integer
|
|
*
|
|
* Specifies the <i>mode</i> argument for open(). It must start
|
|
* with "w", "a", or "r+", otherwise it will cause an error.
|
|
* See IO.new for the list of possible modes.
|
|
*
|
|
* :perm::
|
|
* integer
|
|
*
|
|
* Specifies the <i>perm</i> argument for open().
|
|
*
|
|
* :open_args::
|
|
* array
|
|
*
|
|
* Specifies arguments for open() as an array.
|
|
* This key can not be used in combination with other keys.
|
|
*/
|
|
|
|
static VALUE
|
|
rb_io_s_write(int argc, VALUE *argv, VALUE io)
|
|
{
|
|
return io_s_write(argc, argv, io, 0);
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* IO.binwrite(name, string, [offset] ) -> integer
|
|
* IO.binwrite(name, string, [offset], open_args ) -> integer
|
|
*
|
|
* Same as IO.write except opening the file in binary mode and
|
|
* ASCII-8BIT encoding (<code>"wb:ASCII-8BIT"</code>).
|
|
*/
|
|
|
|
static VALUE
|
|
rb_io_s_binwrite(int argc, VALUE *argv, VALUE io)
|
|
{
|
|
return io_s_write(argc, argv, io, 1);
|
|
}
|
|
|
|
struct copy_stream_struct {
|
|
VALUE src;
|
|
VALUE dst;
|
|
off_t copy_length; /* (off_t)-1 if not specified */
|
|
off_t src_offset; /* (off_t)-1 if not specified */
|
|
|
|
int src_fd;
|
|
int dst_fd;
|
|
unsigned close_src : 1;
|
|
unsigned close_dst : 1;
|
|
int error_no;
|
|
off_t total;
|
|
const char *syserr;
|
|
const char *notimp;
|
|
VALUE th;
|
|
struct stat src_stat;
|
|
struct stat dst_stat;
|
|
#ifdef HAVE_FCOPYFILE
|
|
copyfile_state_t copyfile_state;
|
|
#endif
|
|
};
|
|
|
|
static void *
|
|
exec_interrupts(void *arg)
|
|
{
|
|
VALUE th = (VALUE)arg;
|
|
rb_thread_execute_interrupts(th);
|
|
return NULL;
|
|
}
|
|
|
|
/*
|
|
* returns TRUE if the preceding system call was interrupted
|
|
* so we can continue. If the thread was interrupted, we
|
|
* reacquire the GVL to execute interrupts before continuing.
|
|
*/
|
|
static int
|
|
maygvl_copy_stream_continue_p(int has_gvl, struct copy_stream_struct *stp)
|
|
{
|
|
switch (errno) {
|
|
case EINTR:
|
|
#if defined(ERESTART)
|
|
case ERESTART:
|
|
#endif
|
|
if (rb_thread_interrupted(stp->th)) {
|
|
if (has_gvl)
|
|
rb_thread_execute_interrupts(stp->th);
|
|
else
|
|
rb_thread_call_with_gvl(exec_interrupts, (void *)stp->th);
|
|
}
|
|
return TRUE;
|
|
}
|
|
return FALSE;
|
|
}
|
|
|
|
/* non-Linux poll may not work on all FDs */
|
|
#if defined(HAVE_POLL)
|
|
# if defined(__linux__)
|
|
# define USE_POLL 1
|
|
# endif
|
|
# if defined(__FreeBSD_version) && __FreeBSD_version >= 1100000
|
|
# define USE_POLL 1
|
|
# endif
|
|
#endif
|
|
|
|
#ifndef USE_POLL
|
|
# define USE_POLL 0
|
|
#endif
|
|
|
|
#if USE_POLL
|
|
# define IOWAIT_SYSCALL "poll"
|
|
STATIC_ASSERT(pollin_expected, POLLIN == RB_WAITFD_IN);
|
|
STATIC_ASSERT(pollout_expected, POLLOUT == RB_WAITFD_OUT);
|
|
static int
|
|
nogvl_wait_for_single_fd(int fd, short events)
|
|
{
|
|
struct pollfd fds;
|
|
|
|
fds.fd = fd;
|
|
fds.events = events;
|
|
|
|
return poll(&fds, 1, -1);
|
|
}
|
|
#else /* !USE_POLL */
|
|
# include "vm_core.h"
|
|
# define IOWAIT_SYSCALL "select"
|
|
static int
|
|
nogvl_wait_for_single_fd(int fd, short events)
|
|
{
|
|
rb_fdset_t fds;
|
|
int ret;
|
|
|
|
rb_fd_init(&fds);
|
|
rb_fd_set(fd, &fds);
|
|
|
|
switch (events) {
|
|
case RB_WAITFD_IN:
|
|
ret = rb_fd_select(fd + 1, &fds, 0, 0, 0);
|
|
break;
|
|
case RB_WAITFD_OUT:
|
|
ret = rb_fd_select(fd + 1, 0, &fds, 0, 0);
|
|
break;
|
|
default:
|
|
VM_UNREACHABLE(nogvl_wait_for_single_fd);
|
|
}
|
|
|
|
rb_fd_term(&fds);
|
|
return ret;
|
|
}
|
|
#endif /* !USE_POLL */
|
|
|
|
static int
|
|
maygvl_copy_stream_wait_read(int has_gvl, struct copy_stream_struct *stp)
|
|
{
|
|
int ret;
|
|
|
|
do {
|
|
if (has_gvl) {
|
|
ret = rb_wait_for_single_fd(stp->src_fd, RB_WAITFD_IN, NULL);
|
|
}
|
|
else {
|
|
ret = nogvl_wait_for_single_fd(stp->src_fd, RB_WAITFD_IN);
|
|
}
|
|
} while (ret < 0 && maygvl_copy_stream_continue_p(has_gvl, stp));
|
|
|
|
if (ret < 0) {
|
|
stp->syserr = IOWAIT_SYSCALL;
|
|
stp->error_no = errno;
|
|
return ret;
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
static int
|
|
nogvl_copy_stream_wait_write(struct copy_stream_struct *stp)
|
|
{
|
|
int ret;
|
|
|
|
do {
|
|
ret = nogvl_wait_for_single_fd(stp->dst_fd, RB_WAITFD_OUT);
|
|
} while (ret < 0 && maygvl_copy_stream_continue_p(0, stp));
|
|
|
|
if (ret < 0) {
|
|
stp->syserr = IOWAIT_SYSCALL;
|
|
stp->error_no = errno;
|
|
return ret;
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
#if defined HAVE_COPY_FILE_RANGE || (defined __linux__ && defined __NR_copy_file_range)
|
|
# define USE_COPY_FILE_RANGE
|
|
#endif
|
|
|
|
#ifdef USE_COPY_FILE_RANGE
|
|
|
|
static ssize_t
|
|
simple_copy_file_range(int in_fd, off_t *in_offset, int out_fd, off_t *out_offset, size_t count, unsigned int flags)
|
|
{
|
|
#ifdef HAVE_COPY_FILE_RANGE
|
|
return copy_file_range(in_fd, in_offset, out_fd, out_offset, count, flags);
|
|
#else
|
|
return syscall(__NR_copy_file_range, in_fd, in_offset, out_fd, out_offset, count, flags);
|
|
#endif
|
|
}
|
|
|
|
static int
|
|
nogvl_copy_file_range(struct copy_stream_struct *stp)
|
|
{
|
|
ssize_t ss;
|
|
off_t src_size;
|
|
off_t copy_length, src_offset, *src_offset_ptr;
|
|
|
|
if (!S_ISREG(stp->src_stat.st_mode))
|
|
return 0;
|
|
|
|
src_size = stp->src_stat.st_size;
|
|
src_offset = stp->src_offset;
|
|
if (src_offset >= (off_t)0) {
|
|
src_offset_ptr = &src_offset;
|
|
}
|
|
else {
|
|
src_offset_ptr = NULL; /* if src_offset_ptr is NULL, then bytes are read from in_fd starting from the file offset */
|
|
}
|
|
|
|
copy_length = stp->copy_length;
|
|
if (copy_length < (off_t)0) {
|
|
if (src_offset < (off_t)0) {
|
|
off_t current_offset;
|
|
errno = 0;
|
|
current_offset = lseek(stp->src_fd, 0, SEEK_CUR);
|
|
if (current_offset < (off_t)0 && errno) {
|
|
stp->syserr = "lseek";
|
|
stp->error_no = errno;
|
|
return (int)current_offset;
|
|
}
|
|
copy_length = src_size - current_offset;
|
|
}
|
|
else {
|
|
copy_length = src_size - src_offset;
|
|
}
|
|
}
|
|
|
|
retry_copy_file_range:
|
|
# if SIZEOF_OFF_T > SIZEOF_SIZE_T
|
|
/* we are limited by the 32-bit ssize_t return value on 32-bit */
|
|
ss = (copy_length > (off_t)SSIZE_MAX) ? SSIZE_MAX : (ssize_t)copy_length;
|
|
# else
|
|
ss = (ssize_t)copy_length;
|
|
# endif
|
|
ss = simple_copy_file_range(stp->src_fd, src_offset_ptr, stp->dst_fd, NULL, ss, 0);
|
|
if (0 < ss) {
|
|
stp->total += ss;
|
|
copy_length -= ss;
|
|
if (0 < copy_length) {
|
|
goto retry_copy_file_range;
|
|
}
|
|
}
|
|
if (ss < 0) {
|
|
if (maygvl_copy_stream_continue_p(0, stp)) {
|
|
goto retry_copy_file_range;
|
|
}
|
|
switch (errno) {
|
|
case EINVAL:
|
|
case EPERM: /* copy_file_range(2) doesn't exist (may happen in
|
|
docker container) */
|
|
#ifdef ENOSYS
|
|
case ENOSYS:
|
|
#endif
|
|
#ifdef EXDEV
|
|
case EXDEV: /* in_fd and out_fd are not on the same filesystem */
|
|
#endif
|
|
return 0;
|
|
case EAGAIN:
|
|
#if defined(EWOULDBLOCK) && EWOULDBLOCK != EAGAIN
|
|
case EWOULDBLOCK:
|
|
#endif
|
|
{
|
|
int ret = nogvl_copy_stream_wait_write(stp);
|
|
if (ret < 0) return ret;
|
|
}
|
|
goto retry_copy_file_range;
|
|
case EBADF:
|
|
{
|
|
int e = errno;
|
|
int flags = fcntl(stp->dst_fd, F_GETFL);
|
|
|
|
if (flags != -1 && flags & O_APPEND) {
|
|
return 0;
|
|
}
|
|
errno = e;
|
|
}
|
|
}
|
|
stp->syserr = "copy_file_range";
|
|
stp->error_no = errno;
|
|
return (int)ss;
|
|
}
|
|
return 1;
|
|
}
|
|
#endif
|
|
|
|
#ifdef HAVE_FCOPYFILE
|
|
static int
|
|
nogvl_fcopyfile(struct copy_stream_struct *stp)
|
|
{
|
|
off_t cur, ss = 0;
|
|
const off_t src_offset = stp->src_offset;
|
|
int ret;
|
|
|
|
if (stp->copy_length >= (off_t)0) {
|
|
/* copy_length can't be specified in fcopyfile(3) */
|
|
return 0;
|
|
}
|
|
|
|
if (!S_ISREG(stp->src_stat.st_mode))
|
|
return 0;
|
|
|
|
if (!S_ISREG(stp->dst_stat.st_mode))
|
|
return 0;
|
|
if (lseek(stp->dst_fd, 0, SEEK_CUR) > (off_t)0) /* if dst IO was already written */
|
|
return 0;
|
|
|
|
if (src_offset > (off_t)0) {
|
|
off_t r;
|
|
|
|
/* get current offset */
|
|
errno = 0;
|
|
cur = lseek(stp->src_fd, 0, SEEK_CUR);
|
|
if (cur < (off_t)0 && errno) {
|
|
stp->error_no = errno;
|
|
return 1;
|
|
}
|
|
|
|
errno = 0;
|
|
r = lseek(stp->src_fd, src_offset, SEEK_SET);
|
|
if (r < (off_t)0 && errno) {
|
|
stp->error_no = errno;
|
|
return 1;
|
|
}
|
|
}
|
|
|
|
stp->copyfile_state = copyfile_state_alloc(); /* this will be freed by copy_stream_finalize() */
|
|
ret = fcopyfile(stp->src_fd, stp->dst_fd, stp->copyfile_state, COPYFILE_DATA);
|
|
copyfile_state_get(stp->copyfile_state, COPYFILE_STATE_COPIED, &ss); /* get copied bytes */
|
|
|
|
if (ret == 0) { /* success */
|
|
stp->total = ss;
|
|
if (src_offset > (off_t)0) {
|
|
off_t r;
|
|
errno = 0;
|
|
/* reset offset */
|
|
r = lseek(stp->src_fd, cur, SEEK_SET);
|
|
if (r < (off_t)0 && errno) {
|
|
stp->error_no = errno;
|
|
return 1;
|
|
}
|
|
}
|
|
}
|
|
else {
|
|
switch (errno) {
|
|
case ENOTSUP:
|
|
case EPERM:
|
|
case EINVAL:
|
|
return 0;
|
|
}
|
|
stp->syserr = "fcopyfile";
|
|
stp->error_no = errno;
|
|
return (int)ret;
|
|
}
|
|
return 1;
|
|
}
|
|
#endif
|
|
|
|
#ifdef HAVE_SENDFILE
|
|
|
|
# ifdef __linux__
|
|
# define USE_SENDFILE
|
|
|
|
# ifdef HAVE_SYS_SENDFILE_H
|
|
# include <sys/sendfile.h>
|
|
# endif
|
|
|
|
static ssize_t
|
|
simple_sendfile(int out_fd, int in_fd, off_t *offset, off_t count)
|
|
{
|
|
return sendfile(out_fd, in_fd, offset, (size_t)count);
|
|
}
|
|
|
|
# elif 0 /* defined(__FreeBSD__) || defined(__DragonFly__) */ || defined(__APPLE__)
|
|
/* This runs on FreeBSD8.1 r30210, but sendfiles blocks its execution
|
|
* without cpuset -l 0.
|
|
*/
|
|
# define USE_SENDFILE
|
|
|
|
static ssize_t
|
|
simple_sendfile(int out_fd, int in_fd, off_t *offset, off_t count)
|
|
{
|
|
int r;
|
|
off_t pos = offset ? *offset : lseek(in_fd, 0, SEEK_CUR);
|
|
off_t sbytes;
|
|
# ifdef __APPLE__
|
|
r = sendfile(in_fd, out_fd, pos, &count, NULL, 0);
|
|
sbytes = count;
|
|
# else
|
|
r = sendfile(in_fd, out_fd, pos, (size_t)count, NULL, &sbytes, 0);
|
|
# endif
|
|
if (r != 0 && sbytes == 0) return r;
|
|
if (offset) {
|
|
*offset += sbytes;
|
|
}
|
|
else {
|
|
lseek(in_fd, sbytes, SEEK_CUR);
|
|
}
|
|
return (ssize_t)sbytes;
|
|
}
|
|
|
|
# endif
|
|
|
|
#endif
|
|
|
|
#ifdef USE_SENDFILE
|
|
static int
|
|
nogvl_copy_stream_sendfile(struct copy_stream_struct *stp)
|
|
{
|
|
ssize_t ss;
|
|
off_t src_size;
|
|
off_t copy_length;
|
|
off_t src_offset;
|
|
int use_pread;
|
|
|
|
if (!S_ISREG(stp->src_stat.st_mode))
|
|
return 0;
|
|
|
|
src_size = stp->src_stat.st_size;
|
|
#ifndef __linux__
|
|
if ((stp->dst_stat.st_mode & S_IFMT) != S_IFSOCK)
|
|
return 0;
|
|
#endif
|
|
|
|
src_offset = stp->src_offset;
|
|
use_pread = src_offset >= (off_t)0;
|
|
|
|
copy_length = stp->copy_length;
|
|
if (copy_length < (off_t)0) {
|
|
if (use_pread)
|
|
copy_length = src_size - src_offset;
|
|
else {
|
|
off_t cur;
|
|
errno = 0;
|
|
cur = lseek(stp->src_fd, 0, SEEK_CUR);
|
|
if (cur < (off_t)0 && errno) {
|
|
stp->syserr = "lseek";
|
|
stp->error_no = errno;
|
|
return (int)cur;
|
|
}
|
|
copy_length = src_size - cur;
|
|
}
|
|
}
|
|
|
|
retry_sendfile:
|
|
# if SIZEOF_OFF_T > SIZEOF_SIZE_T
|
|
/* we are limited by the 32-bit ssize_t return value on 32-bit */
|
|
ss = (copy_length > (off_t)SSIZE_MAX) ? SSIZE_MAX : (ssize_t)copy_length;
|
|
# else
|
|
ss = (ssize_t)copy_length;
|
|
# endif
|
|
if (use_pread) {
|
|
ss = simple_sendfile(stp->dst_fd, stp->src_fd, &src_offset, ss);
|
|
}
|
|
else {
|
|
ss = simple_sendfile(stp->dst_fd, stp->src_fd, NULL, ss);
|
|
}
|
|
if (0 < ss) {
|
|
stp->total += ss;
|
|
copy_length -= ss;
|
|
if (0 < copy_length) {
|
|
goto retry_sendfile;
|
|
}
|
|
}
|
|
if (ss < 0) {
|
|
if (maygvl_copy_stream_continue_p(0, stp))
|
|
goto retry_sendfile;
|
|
switch (errno) {
|
|
case EINVAL:
|
|
#ifdef ENOSYS
|
|
case ENOSYS:
|
|
#endif
|
|
return 0;
|
|
case EAGAIN:
|
|
#if defined(EWOULDBLOCK) && EWOULDBLOCK != EAGAIN
|
|
case EWOULDBLOCK:
|
|
#endif
|
|
{
|
|
int ret;
|
|
#ifndef __linux__
|
|
/*
|
|
* Linux requires stp->src_fd to be a mmap-able (regular) file,
|
|
* select() reports regular files to always be "ready", so
|
|
* there is no need to select() on it.
|
|
* Other OSes may have the same limitation for sendfile() which
|
|
* allow us to bypass maygvl_copy_stream_wait_read()...
|
|
*/
|
|
ret = maygvl_copy_stream_wait_read(0, stp);
|
|
if (ret < 0) return ret;
|
|
#endif
|
|
ret = nogvl_copy_stream_wait_write(stp);
|
|
if (ret < 0) return ret;
|
|
}
|
|
goto retry_sendfile;
|
|
}
|
|
stp->syserr = "sendfile";
|
|
stp->error_no = errno;
|
|
return (int)ss;
|
|
}
|
|
return 1;
|
|
}
|
|
#endif
|
|
|
|
static ssize_t
|
|
maygvl_read(int has_gvl, int fd, void *buf, size_t count)
|
|
{
|
|
if (has_gvl)
|
|
return rb_read_internal(fd, buf, count);
|
|
else
|
|
return read(fd, buf, count);
|
|
}
|
|
|
|
static ssize_t
|
|
maygvl_copy_stream_read(int has_gvl, struct copy_stream_struct *stp, char *buf, size_t len, off_t offset)
|
|
{
|
|
ssize_t ss;
|
|
retry_read:
|
|
if (offset < (off_t)0) {
|
|
ss = maygvl_read(has_gvl, stp->src_fd, buf, len);
|
|
}
|
|
else {
|
|
#ifdef HAVE_PREAD
|
|
ss = pread(stp->src_fd, buf, len, offset);
|
|
#else
|
|
stp->notimp = "pread";
|
|
return -1;
|
|
#endif
|
|
}
|
|
if (ss == 0) {
|
|
return 0;
|
|
}
|
|
if (ss < 0) {
|
|
if (maygvl_copy_stream_continue_p(has_gvl, stp))
|
|
goto retry_read;
|
|
switch (errno) {
|
|
case EAGAIN:
|
|
#if defined(EWOULDBLOCK) && EWOULDBLOCK != EAGAIN
|
|
case EWOULDBLOCK:
|
|
#endif
|
|
{
|
|
int ret = maygvl_copy_stream_wait_read(has_gvl, stp);
|
|
if (ret < 0) return ret;
|
|
}
|
|
goto retry_read;
|
|
#ifdef ENOSYS
|
|
case ENOSYS:
|
|
stp->notimp = "pread";
|
|
return ss;
|
|
#endif
|
|
}
|
|
stp->syserr = offset < (off_t)0 ? "read" : "pread";
|
|
stp->error_no = errno;
|
|
}
|
|
return ss;
|
|
}
|
|
|
|
static int
|
|
nogvl_copy_stream_write(struct copy_stream_struct *stp, char *buf, size_t len)
|
|
{
|
|
ssize_t ss;
|
|
int off = 0;
|
|
while (len) {
|
|
ss = write(stp->dst_fd, buf+off, len);
|
|
if (ss < 0) {
|
|
if (maygvl_copy_stream_continue_p(0, stp))
|
|
continue;
|
|
if (errno == EAGAIN || errno == EWOULDBLOCK) {
|
|
int ret = nogvl_copy_stream_wait_write(stp);
|
|
if (ret < 0) return ret;
|
|
continue;
|
|
}
|
|
stp->syserr = "write";
|
|
stp->error_no = errno;
|
|
return (int)ss;
|
|
}
|
|
off += (int)ss;
|
|
len -= (int)ss;
|
|
stp->total += ss;
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
static void
|
|
nogvl_copy_stream_read_write(struct copy_stream_struct *stp)
|
|
{
|
|
char buf[1024*16];
|
|
size_t len;
|
|
ssize_t ss;
|
|
int ret;
|
|
off_t copy_length;
|
|
int use_eof;
|
|
off_t src_offset;
|
|
int use_pread;
|
|
|
|
copy_length = stp->copy_length;
|
|
use_eof = copy_length < (off_t)0;
|
|
src_offset = stp->src_offset;
|
|
use_pread = src_offset >= (off_t)0;
|
|
|
|
if (use_pread && stp->close_src) {
|
|
off_t r;
|
|
errno = 0;
|
|
r = lseek(stp->src_fd, src_offset, SEEK_SET);
|
|
if (r < (off_t)0 && errno) {
|
|
stp->syserr = "lseek";
|
|
stp->error_no = errno;
|
|
return;
|
|
}
|
|
src_offset = (off_t)-1;
|
|
use_pread = 0;
|
|
}
|
|
|
|
while (use_eof || 0 < copy_length) {
|
|
if (!use_eof && copy_length < (off_t)sizeof(buf)) {
|
|
len = (size_t)copy_length;
|
|
}
|
|
else {
|
|
len = sizeof(buf);
|
|
}
|
|
if (use_pread) {
|
|
ss = maygvl_copy_stream_read(0, stp, buf, len, src_offset);
|
|
if (0 < ss)
|
|
src_offset += ss;
|
|
}
|
|
else {
|
|
ss = maygvl_copy_stream_read(0, stp, buf, len, (off_t)-1);
|
|
}
|
|
if (ss <= 0) /* EOF or error */
|
|
return;
|
|
|
|
ret = nogvl_copy_stream_write(stp, buf, ss);
|
|
if (ret < 0)
|
|
return;
|
|
|
|
if (!use_eof)
|
|
copy_length -= ss;
|
|
}
|
|
}
|
|
|
|
static void *
|
|
nogvl_copy_stream_func(void *arg)
|
|
{
|
|
struct copy_stream_struct *stp = (struct copy_stream_struct *)arg;
|
|
#if defined(USE_SENDFILE) || defined(USE_COPY_FILE_RANGE) || defined(HAVE_FCOPYFILE)
|
|
int ret;
|
|
#endif
|
|
|
|
#ifdef USE_COPY_FILE_RANGE
|
|
ret = nogvl_copy_file_range(stp);
|
|
if (ret != 0)
|
|
goto finish; /* error or success */
|
|
#endif
|
|
|
|
#ifdef HAVE_FCOPYFILE
|
|
ret = nogvl_fcopyfile(stp);
|
|
if (ret != 0)
|
|
goto finish; /* error or success */
|
|
#endif
|
|
|
|
#ifdef USE_SENDFILE
|
|
ret = nogvl_copy_stream_sendfile(stp);
|
|
if (ret != 0)
|
|
goto finish; /* error or success */
|
|
#endif
|
|
|
|
nogvl_copy_stream_read_write(stp);
|
|
|
|
#if defined(USE_SENDFILE) || defined(USE_COPY_FILE_RANGE) || defined(HAVE_FCOPYFILE)
|
|
finish:
|
|
#endif
|
|
return 0;
|
|
}
|
|
|
|
static VALUE
|
|
copy_stream_fallback_body(VALUE arg)
|
|
{
|
|
struct copy_stream_struct *stp = (struct copy_stream_struct *)arg;
|
|
const int buflen = 16*1024;
|
|
VALUE n;
|
|
VALUE buf = rb_str_buf_new(buflen);
|
|
off_t rest = stp->copy_length;
|
|
off_t off = stp->src_offset;
|
|
ID read_method = id_readpartial;
|
|
|
|
if (stp->src_fd < 0) {
|
|
if (!rb_respond_to(stp->src, read_method)) {
|
|
read_method = id_read;
|
|
}
|
|
}
|
|
|
|
while (1) {
|
|
long numwrote;
|
|
long l;
|
|
if (stp->copy_length < (off_t)0) {
|
|
l = buflen;
|
|
}
|
|
else {
|
|
if (rest == 0) {
|
|
rb_str_resize(buf, 0);
|
|
break;
|
|
}
|
|
l = buflen < rest ? buflen : (long)rest;
|
|
}
|
|
if (stp->src_fd < 0) {
|
|
VALUE rc = rb_funcall(stp->src, read_method, 2, INT2FIX(l), buf);
|
|
|
|
if (read_method == id_read && NIL_P(rc))
|
|
break;
|
|
}
|
|
else {
|
|
ssize_t ss;
|
|
rb_str_resize(buf, buflen);
|
|
ss = maygvl_copy_stream_read(1, stp, RSTRING_PTR(buf), l, off);
|
|
rb_str_resize(buf, ss > 0 ? ss : 0);
|
|
if (ss < 0)
|
|
return Qnil;
|
|
if (ss == 0)
|
|
rb_eof_error();
|
|
if (off >= (off_t)0)
|
|
off += ss;
|
|
}
|
|
n = rb_io_write(stp->dst, buf);
|
|
numwrote = NUM2LONG(n);
|
|
stp->total += numwrote;
|
|
rest -= numwrote;
|
|
if (read_method == id_read && RSTRING_LEN(buf) == 0) {
|
|
break;
|
|
}
|
|
}
|
|
|
|
return Qnil;
|
|
}
|
|
|
|
static VALUE
|
|
copy_stream_fallback(struct copy_stream_struct *stp)
|
|
{
|
|
if (stp->src_fd < 0 && stp->src_offset >= (off_t)0) {
|
|
rb_raise(rb_eArgError, "cannot specify src_offset for non-IO");
|
|
}
|
|
rb_rescue2(copy_stream_fallback_body, (VALUE)stp,
|
|
(VALUE (*) (VALUE, VALUE))0, (VALUE)0,
|
|
rb_eEOFError, (VALUE)0);
|
|
return Qnil;
|
|
}
|
|
|
|
static VALUE
|
|
copy_stream_body(VALUE arg)
|
|
{
|
|
struct copy_stream_struct *stp = (struct copy_stream_struct *)arg;
|
|
VALUE src_io = stp->src, dst_io = stp->dst;
|
|
rb_io_t *src_fptr = 0, *dst_fptr = 0;
|
|
int src_fd, dst_fd;
|
|
const int common_oflags = 0
|
|
#ifdef O_NOCTTY
|
|
| O_NOCTTY
|
|
#endif
|
|
;
|
|
|
|
stp->th = rb_thread_current();
|
|
|
|
stp->total = 0;
|
|
|
|
if (src_io == argf ||
|
|
!(RB_TYPE_P(src_io, T_FILE) ||
|
|
RB_TYPE_P(src_io, T_STRING) ||
|
|
rb_respond_to(src_io, rb_intern("to_path")))) {
|
|
src_fd = -1;
|
|
}
|
|
else {
|
|
int stat_ret;
|
|
VALUE tmp_io = rb_io_check_io(src_io);
|
|
if (!NIL_P(tmp_io)) {
|
|
src_io = tmp_io;
|
|
}
|
|
else if (!RB_TYPE_P(src_io, T_FILE)) {
|
|
VALUE args[2];
|
|
FilePathValue(src_io);
|
|
args[0] = src_io;
|
|
args[1] = INT2NUM(O_RDONLY|common_oflags);
|
|
src_io = rb_class_new_instance(2, args, rb_cFile);
|
|
stp->src = src_io;
|
|
stp->close_src = 1;
|
|
}
|
|
GetOpenFile(src_io, src_fptr);
|
|
rb_io_check_byte_readable(src_fptr);
|
|
src_fd = src_fptr->fd;
|
|
|
|
stat_ret = fstat(src_fd, &stp->src_stat);
|
|
if (stat_ret < 0) {
|
|
stp->syserr = "fstat";
|
|
stp->error_no = errno;
|
|
return Qnil;
|
|
}
|
|
}
|
|
stp->src_fd = src_fd;
|
|
|
|
if (dst_io == argf ||
|
|
!(RB_TYPE_P(dst_io, T_FILE) ||
|
|
RB_TYPE_P(dst_io, T_STRING) ||
|
|
rb_respond_to(dst_io, rb_intern("to_path")))) {
|
|
dst_fd = -1;
|
|
}
|
|
else {
|
|
int stat_ret;
|
|
VALUE tmp_io = rb_io_check_io(dst_io);
|
|
if (!NIL_P(tmp_io)) {
|
|
dst_io = GetWriteIO(tmp_io);
|
|
}
|
|
else if (!RB_TYPE_P(dst_io, T_FILE)) {
|
|
VALUE args[3];
|
|
FilePathValue(dst_io);
|
|
args[0] = dst_io;
|
|
args[1] = INT2NUM(O_WRONLY|O_CREAT|O_TRUNC|common_oflags);
|
|
args[2] = INT2FIX(0666);
|
|
dst_io = rb_class_new_instance(3, args, rb_cFile);
|
|
stp->dst = dst_io;
|
|
stp->close_dst = 1;
|
|
}
|
|
else {
|
|
dst_io = GetWriteIO(dst_io);
|
|
stp->dst = dst_io;
|
|
}
|
|
GetOpenFile(dst_io, dst_fptr);
|
|
rb_io_check_writable(dst_fptr);
|
|
dst_fd = dst_fptr->fd;
|
|
|
|
stat_ret = fstat(dst_fd, &stp->dst_stat);
|
|
if (stat_ret < 0) {
|
|
stp->syserr = "fstat";
|
|
stp->error_no = errno;
|
|
return Qnil;
|
|
}
|
|
}
|
|
stp->dst_fd = dst_fd;
|
|
|
|
#ifdef O_BINARY
|
|
if (src_fptr)
|
|
SET_BINARY_MODE_WITH_SEEK_CUR(src_fptr);
|
|
#endif
|
|
if (dst_fptr)
|
|
io_ascii8bit_binmode(dst_fptr);
|
|
|
|
if (stp->src_offset < (off_t)0 && src_fptr && src_fptr->rbuf.len) {
|
|
size_t len = src_fptr->rbuf.len;
|
|
VALUE str;
|
|
if (stp->copy_length >= (off_t)0 && stp->copy_length < (off_t)len) {
|
|
len = (size_t)stp->copy_length;
|
|
}
|
|
str = rb_str_buf_new(len);
|
|
rb_str_resize(str,len);
|
|
read_buffered_data(RSTRING_PTR(str), len, src_fptr);
|
|
if (dst_fptr) { /* IO or filename */
|
|
if (io_binwrite(str, RSTRING_PTR(str), RSTRING_LEN(str), dst_fptr, 0) < 0)
|
|
rb_sys_fail(0);
|
|
}
|
|
else /* others such as StringIO */
|
|
rb_io_write(dst_io, str);
|
|
rb_str_resize(str, 0);
|
|
stp->total += len;
|
|
if (stp->copy_length >= (off_t)0)
|
|
stp->copy_length -= len;
|
|
}
|
|
|
|
if (dst_fptr && io_fflush(dst_fptr) < 0) {
|
|
rb_raise(rb_eIOError, "flush failed");
|
|
}
|
|
|
|
if (stp->copy_length == 0)
|
|
return Qnil;
|
|
|
|
if (src_fd < 0 || dst_fd < 0) {
|
|
return copy_stream_fallback(stp);
|
|
}
|
|
|
|
rb_thread_call_without_gvl(nogvl_copy_stream_func, (void*)stp, RUBY_UBF_IO, 0);
|
|
return Qnil;
|
|
}
|
|
|
|
static VALUE
|
|
copy_stream_finalize(VALUE arg)
|
|
{
|
|
struct copy_stream_struct *stp = (struct copy_stream_struct *)arg;
|
|
|
|
#ifdef HAVE_FCOPYFILE
|
|
if (stp->copyfile_state) {
|
|
copyfile_state_free(stp->copyfile_state);
|
|
}
|
|
#endif
|
|
|
|
if (stp->close_src) {
|
|
rb_io_close_m(stp->src);
|
|
}
|
|
if (stp->close_dst) {
|
|
rb_io_close_m(stp->dst);
|
|
}
|
|
if (stp->syserr) {
|
|
rb_syserr_fail(stp->error_no, stp->syserr);
|
|
}
|
|
if (stp->notimp) {
|
|
rb_raise(rb_eNotImpError, "%s() not implemented", stp->notimp);
|
|
}
|
|
return Qnil;
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* IO.copy_stream(src, dst)
|
|
* IO.copy_stream(src, dst, copy_length)
|
|
* IO.copy_stream(src, dst, copy_length, src_offset)
|
|
*
|
|
* IO.copy_stream copies <i>src</i> to <i>dst</i>.
|
|
* <i>src</i> and <i>dst</i> is either a filename or an IO-like object.
|
|
* IO-like object for <i>src</i> should have #readpartial or #read
|
|
* method. IO-like object for <i>dst</i> should have #write method.
|
|
* (Specialized mechanisms, such as sendfile system call, may be used
|
|
* on appropriate situation.)
|
|
*
|
|
* This method returns the number of bytes copied.
|
|
*
|
|
* If optional arguments are not given,
|
|
* the start position of the copy is
|
|
* the beginning of the filename or
|
|
* the current file offset of the IO.
|
|
* The end position of the copy is the end of file.
|
|
*
|
|
* If <i>copy_length</i> is given,
|
|
* No more than <i>copy_length</i> bytes are copied.
|
|
*
|
|
* If <i>src_offset</i> is given,
|
|
* it specifies the start position of the copy.
|
|
*
|
|
* When <i>src_offset</i> is specified and
|
|
* <i>src</i> is an IO,
|
|
* IO.copy_stream doesn't move the current file offset.
|
|
*
|
|
*/
|
|
static VALUE
|
|
rb_io_s_copy_stream(int argc, VALUE *argv, VALUE io)
|
|
{
|
|
VALUE src, dst, length, src_offset;
|
|
struct copy_stream_struct st;
|
|
|
|
MEMZERO(&st, struct copy_stream_struct, 1);
|
|
|
|
rb_scan_args(argc, argv, "22", &src, &dst, &length, &src_offset);
|
|
|
|
st.src = src;
|
|
st.dst = dst;
|
|
|
|
if (NIL_P(length))
|
|
st.copy_length = (off_t)-1;
|
|
else
|
|
st.copy_length = NUM2OFFT(length);
|
|
|
|
if (NIL_P(src_offset))
|
|
st.src_offset = (off_t)-1;
|
|
else
|
|
st.src_offset = NUM2OFFT(src_offset);
|
|
|
|
rb_ensure(copy_stream_body, (VALUE)&st, copy_stream_finalize, (VALUE)&st);
|
|
|
|
return OFFT2NUM(st.total);
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* io.external_encoding -> encoding
|
|
*
|
|
* Returns the Encoding object that represents the encoding of the file.
|
|
* If _io_ is in write mode and no encoding is specified, returns +nil+.
|
|
*/
|
|
|
|
static VALUE
|
|
rb_io_external_encoding(VALUE io)
|
|
{
|
|
rb_io_t *fptr;
|
|
|
|
GetOpenFile(io, fptr);
|
|
if (fptr->encs.enc2) {
|
|
return rb_enc_from_encoding(fptr->encs.enc2);
|
|
}
|
|
if (fptr->mode & FMODE_WRITABLE) {
|
|
if (fptr->encs.enc)
|
|
return rb_enc_from_encoding(fptr->encs.enc);
|
|
return Qnil;
|
|
}
|
|
return rb_enc_from_encoding(io_read_encoding(fptr));
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* io.internal_encoding -> encoding
|
|
*
|
|
* Returns the Encoding of the internal string if conversion is
|
|
* specified. Otherwise returns +nil+.
|
|
*/
|
|
|
|
static VALUE
|
|
rb_io_internal_encoding(VALUE io)
|
|
{
|
|
rb_io_t *fptr;
|
|
|
|
GetOpenFile(io, fptr);
|
|
if (!fptr->encs.enc2) return Qnil;
|
|
return rb_enc_from_encoding(io_read_encoding(fptr));
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* io.set_encoding(ext_enc) -> io
|
|
* io.set_encoding("ext_enc:int_enc") -> io
|
|
* io.set_encoding(ext_enc, int_enc) -> io
|
|
* io.set_encoding("ext_enc:int_enc", opt) -> io
|
|
* io.set_encoding(ext_enc, int_enc, opt) -> io
|
|
*
|
|
* If single argument is specified, read string from io is tagged
|
|
* with the encoding specified. If encoding is a colon separated two
|
|
* encoding names "A:B", the read string is converted from encoding A
|
|
* (external encoding) to encoding B (internal encoding), then tagged
|
|
* with B. If two arguments are specified, those must be encoding
|
|
* objects or encoding names, and the first one is the external encoding, and the
|
|
* second one is the internal encoding.
|
|
* If the external encoding and the internal encoding is specified,
|
|
* optional hash argument specify the conversion option.
|
|
*/
|
|
|
|
static VALUE
|
|
rb_io_set_encoding(int argc, VALUE *argv, VALUE io)
|
|
{
|
|
rb_io_t *fptr;
|
|
VALUE v1, v2, opt;
|
|
|
|
if (!RB_TYPE_P(io, T_FILE)) {
|
|
return rb_funcallv(io, id_set_encoding, argc, argv);
|
|
}
|
|
|
|
argc = rb_scan_args(argc, argv, "11:", &v1, &v2, &opt);
|
|
GetOpenFile(io, fptr);
|
|
io_encoding_set(fptr, v1, v2, opt);
|
|
return io;
|
|
}
|
|
|
|
void
|
|
rb_stdio_set_default_encoding(void)
|
|
{
|
|
VALUE val = Qnil;
|
|
|
|
rb_io_set_encoding(1, &val, rb_stdin);
|
|
rb_io_set_encoding(1, &val, rb_stdout);
|
|
rb_io_set_encoding(1, &val, rb_stderr);
|
|
}
|
|
|
|
static inline int
|
|
global_argf_p(VALUE arg)
|
|
{
|
|
return arg == argf;
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* ARGF.external_encoding -> encoding
|
|
*
|
|
* Returns the external encoding for files read from +ARGF+ as an +Encoding+
|
|
* object. The external encoding is the encoding of the text as stored in a
|
|
* file. Contrast with +ARGF.internal_encoding+, which is the encoding used
|
|
* to represent this text within Ruby.
|
|
*
|
|
* To set the external encoding use +ARGF.set_encoding+.
|
|
*
|
|
* For example:
|
|
*
|
|
* ARGF.external_encoding #=> #<Encoding:UTF-8>
|
|
*
|
|
*/
|
|
static VALUE
|
|
argf_external_encoding(VALUE argf)
|
|
{
|
|
if (!RTEST(ARGF.current_file)) {
|
|
return rb_enc_from_encoding(rb_default_external_encoding());
|
|
}
|
|
return rb_io_external_encoding(rb_io_check_io(ARGF.current_file));
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* ARGF.internal_encoding -> encoding
|
|
*
|
|
* Returns the internal encoding for strings read from +ARGF+ as an
|
|
* +Encoding+ object.
|
|
*
|
|
* If +ARGF.set_encoding+ has been called with two encoding names, the second
|
|
* is returned. Otherwise, if +Encoding.default_external+ has been set, that
|
|
* value is returned. Failing that, if a default external encoding was
|
|
* specified on the command-line, that value is used. If the encoding is
|
|
* unknown, +nil+ is returned.
|
|
*/
|
|
static VALUE
|
|
argf_internal_encoding(VALUE argf)
|
|
{
|
|
if (!RTEST(ARGF.current_file)) {
|
|
return rb_enc_from_encoding(rb_default_external_encoding());
|
|
}
|
|
return rb_io_internal_encoding(rb_io_check_io(ARGF.current_file));
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* ARGF.set_encoding(ext_enc) -> ARGF
|
|
* ARGF.set_encoding("ext_enc:int_enc") -> ARGF
|
|
* ARGF.set_encoding(ext_enc, int_enc) -> ARGF
|
|
* ARGF.set_encoding("ext_enc:int_enc", opt) -> ARGF
|
|
* ARGF.set_encoding(ext_enc, int_enc, opt) -> ARGF
|
|
*
|
|
* If single argument is specified, strings read from ARGF are tagged with
|
|
* the encoding specified.
|
|
*
|
|
* If two encoding names separated by a colon are given, e.g. "ascii:utf-8",
|
|
* the read string is converted from the first encoding (external encoding)
|
|
* to the second encoding (internal encoding), then tagged with the second
|
|
* encoding.
|
|
*
|
|
* If two arguments are specified, they must be encoding objects or encoding
|
|
* names. Again, the first specifies the external encoding; the second
|
|
* specifies the internal encoding.
|
|
*
|
|
* If the external encoding and the internal encoding are specified, the
|
|
* optional +Hash+ argument can be used to adjust the conversion process. The
|
|
* structure of this hash is explained in the String#encode documentation.
|
|
*
|
|
* For example:
|
|
*
|
|
* ARGF.set_encoding('ascii') # Tag the input as US-ASCII text
|
|
* ARGF.set_encoding(Encoding::UTF_8) # Tag the input as UTF-8 text
|
|
* ARGF.set_encoding('utf-8','ascii') # Transcode the input from US-ASCII
|
|
* # to UTF-8.
|
|
*/
|
|
static VALUE
|
|
argf_set_encoding(int argc, VALUE *argv, VALUE argf)
|
|
{
|
|
rb_io_t *fptr;
|
|
|
|
if (!next_argv()) {
|
|
rb_raise(rb_eArgError, "no stream to set encoding");
|
|
}
|
|
rb_io_set_encoding(argc, argv, ARGF.current_file);
|
|
GetOpenFile(ARGF.current_file, fptr);
|
|
ARGF.encs = fptr->encs;
|
|
return argf;
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* ARGF.tell -> Integer
|
|
* ARGF.pos -> Integer
|
|
*
|
|
* Returns the current offset (in bytes) of the current file in +ARGF+.
|
|
*
|
|
* ARGF.pos #=> 0
|
|
* ARGF.gets #=> "This is line one\n"
|
|
* ARGF.pos #=> 17
|
|
*
|
|
*/
|
|
static VALUE
|
|
argf_tell(VALUE argf)
|
|
{
|
|
if (!next_argv()) {
|
|
rb_raise(rb_eArgError, "no stream to tell");
|
|
}
|
|
ARGF_FORWARD(0, 0);
|
|
return rb_io_tell(ARGF.current_file);
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* ARGF.seek(amount, whence=IO::SEEK_SET) -> 0
|
|
*
|
|
* Seeks to offset _amount_ (an +Integer+) in the +ARGF+ stream according to
|
|
* the value of _whence_. See IO#seek for further details.
|
|
*/
|
|
static VALUE
|
|
argf_seek_m(int argc, VALUE *argv, VALUE argf)
|
|
{
|
|
if (!next_argv()) {
|
|
rb_raise(rb_eArgError, "no stream to seek");
|
|
}
|
|
ARGF_FORWARD(argc, argv);
|
|
return rb_io_seek_m(argc, argv, ARGF.current_file);
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* ARGF.pos = position -> Integer
|
|
*
|
|
* Seeks to the position given by _position_ (in bytes) in +ARGF+.
|
|
*
|
|
* For example:
|
|
*
|
|
* ARGF.pos = 17
|
|
* ARGF.gets #=> "This is line two\n"
|
|
*/
|
|
static VALUE
|
|
argf_set_pos(VALUE argf, VALUE offset)
|
|
{
|
|
if (!next_argv()) {
|
|
rb_raise(rb_eArgError, "no stream to set position");
|
|
}
|
|
ARGF_FORWARD(1, &offset);
|
|
return rb_io_set_pos(ARGF.current_file, offset);
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* ARGF.rewind -> 0
|
|
*
|
|
* Positions the current file to the beginning of input, resetting
|
|
* +ARGF.lineno+ to zero.
|
|
*
|
|
* ARGF.readline #=> "This is line one\n"
|
|
* ARGF.rewind #=> 0
|
|
* ARGF.lineno #=> 0
|
|
* ARGF.readline #=> "This is line one\n"
|
|
*/
|
|
static VALUE
|
|
argf_rewind(VALUE argf)
|
|
{
|
|
VALUE ret;
|
|
int old_lineno;
|
|
|
|
if (!next_argv()) {
|
|
rb_raise(rb_eArgError, "no stream to rewind");
|
|
}
|
|
ARGF_FORWARD(0, 0);
|
|
old_lineno = RFILE(ARGF.current_file)->fptr->lineno;
|
|
ret = rb_io_rewind(ARGF.current_file);
|
|
if (!global_argf_p(argf)) {
|
|
ARGF.last_lineno = ARGF.lineno -= old_lineno;
|
|
}
|
|
return ret;
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* ARGF.fileno -> integer
|
|
* ARGF.to_i -> integer
|
|
*
|
|
* Returns an integer representing the numeric file descriptor for
|
|
* the current file. Raises an +ArgumentError+ if there isn't a current file.
|
|
*
|
|
* ARGF.fileno #=> 3
|
|
*/
|
|
static VALUE
|
|
argf_fileno(VALUE argf)
|
|
{
|
|
if (!next_argv()) {
|
|
rb_raise(rb_eArgError, "no stream");
|
|
}
|
|
ARGF_FORWARD(0, 0);
|
|
return rb_io_fileno(ARGF.current_file);
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* ARGF.to_io -> IO
|
|
*
|
|
* Returns an +IO+ object representing the current file. This will be a
|
|
* +File+ object unless the current file is a stream such as STDIN.
|
|
*
|
|
* For example:
|
|
*
|
|
* ARGF.to_io #=> #<File:glark.txt>
|
|
* ARGF.to_io #=> #<IO:<STDIN>>
|
|
*/
|
|
static VALUE
|
|
argf_to_io(VALUE argf)
|
|
{
|
|
next_argv();
|
|
ARGF_FORWARD(0, 0);
|
|
return ARGF.current_file;
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* ARGF.eof? -> true or false
|
|
* ARGF.eof -> true or false
|
|
*
|
|
* Returns true if the current file in +ARGF+ is at end of file, i.e. it has
|
|
* no data to read. The stream must be opened for reading or an +IOError+
|
|
* will be raised.
|
|
*
|
|
* $ echo "eof" | ruby argf.rb
|
|
*
|
|
* ARGF.eof? #=> false
|
|
* 3.times { ARGF.readchar }
|
|
* ARGF.eof? #=> false
|
|
* ARGF.readchar #=> "\n"
|
|
* ARGF.eof? #=> true
|
|
*/
|
|
|
|
static VALUE
|
|
argf_eof(VALUE argf)
|
|
{
|
|
next_argv();
|
|
if (RTEST(ARGF.current_file)) {
|
|
if (ARGF.init_p == 0) return Qtrue;
|
|
next_argv();
|
|
ARGF_FORWARD(0, 0);
|
|
if (rb_io_eof(ARGF.current_file)) {
|
|
return Qtrue;
|
|
}
|
|
}
|
|
return Qfalse;
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* ARGF.read([length [, outbuf]]) -> string, outbuf, or nil
|
|
*
|
|
* Reads _length_ bytes from ARGF. The files named on the command line
|
|
* are concatenated and treated as a single file by this method, so when
|
|
* called without arguments the contents of this pseudo file are returned in
|
|
* their entirety.
|
|
*
|
|
* _length_ must be a non-negative integer or +nil+.
|
|
*
|
|
* If _length_ is a positive integer, +read+ tries to read
|
|
* _length_ bytes without any conversion (binary mode).
|
|
* It returns +nil+ if an EOF is encountered before anything can be read.
|
|
* Fewer than _length_ bytes are returned if an EOF is encountered during
|
|
* the read.
|
|
* In the case of an integer _length_, the resulting string is always
|
|
* in ASCII-8BIT encoding.
|
|
*
|
|
* If _length_ is omitted or is +nil+, it reads until EOF
|
|
* and the encoding conversion is applied, if applicable.
|
|
* A string is returned even if EOF is encountered before any data is read.
|
|
*
|
|
* If _length_ is zero, it returns an empty string (<code>""</code>).
|
|
*
|
|
* If the optional _outbuf_ argument is present,
|
|
* it must reference a String, which will receive the data.
|
|
* The _outbuf_ will contain only the received data after the method call
|
|
* even if it is not empty at the beginning.
|
|
*
|
|
* For example:
|
|
*
|
|
* $ echo "small" > small.txt
|
|
* $ echo "large" > large.txt
|
|
* $ ./glark.rb small.txt large.txt
|
|
*
|
|
* ARGF.read #=> "small\nlarge"
|
|
* ARGF.read(200) #=> "small\nlarge"
|
|
* ARGF.read(2) #=> "sm"
|
|
* ARGF.read(0) #=> ""
|
|
*
|
|
* Note that this method behaves like the fread() function in C.
|
|
* This means it retries to invoke read(2) system calls to read data
|
|
* with the specified length.
|
|
* If you need the behavior like a single read(2) system call,
|
|
* consider ARGF#readpartial or ARGF#read_nonblock.
|
|
*/
|
|
|
|
static VALUE
|
|
argf_read(int argc, VALUE *argv, VALUE argf)
|
|
{
|
|
VALUE tmp, str, length;
|
|
long len = 0;
|
|
|
|
rb_scan_args(argc, argv, "02", &length, &str);
|
|
if (!NIL_P(length)) {
|
|
len = NUM2LONG(argv[0]);
|
|
}
|
|
if (!NIL_P(str)) {
|
|
StringValue(str);
|
|
rb_str_resize(str,0);
|
|
argv[1] = Qnil;
|
|
}
|
|
|
|
retry:
|
|
if (!next_argv()) {
|
|
return str;
|
|
}
|
|
if (ARGF_GENERIC_INPUT_P()) {
|
|
tmp = argf_forward(argc, argv, argf);
|
|
}
|
|
else {
|
|
tmp = io_read(argc, argv, ARGF.current_file);
|
|
}
|
|
if (NIL_P(str)) str = tmp;
|
|
else if (!NIL_P(tmp)) rb_str_append(str, tmp);
|
|
if (NIL_P(tmp) || NIL_P(length)) {
|
|
if (ARGF.next_p != -1) {
|
|
argf_close(argf);
|
|
ARGF.next_p = 1;
|
|
goto retry;
|
|
}
|
|
}
|
|
else if (argc >= 1) {
|
|
long slen = RSTRING_LEN(str);
|
|
if (slen < len) {
|
|
len -= slen;
|
|
argv[0] = INT2NUM(len);
|
|
goto retry;
|
|
}
|
|
}
|
|
return str;
|
|
}
|
|
|
|
struct argf_call_arg {
|
|
int argc;
|
|
VALUE *argv;
|
|
VALUE argf;
|
|
};
|
|
|
|
static VALUE
|
|
argf_forward_call(VALUE arg)
|
|
{
|
|
struct argf_call_arg *p = (struct argf_call_arg *)arg;
|
|
argf_forward(p->argc, p->argv, p->argf);
|
|
return Qnil;
|
|
}
|
|
|
|
static VALUE argf_getpartial(int argc, VALUE *argv, VALUE argf, VALUE opts,
|
|
int nonblock);
|
|
|
|
/*
|
|
* call-seq:
|
|
* ARGF.readpartial(maxlen) -> string
|
|
* ARGF.readpartial(maxlen, outbuf) -> outbuf
|
|
*
|
|
* Reads at most _maxlen_ bytes from the ARGF stream.
|
|
*
|
|
* If the optional _outbuf_ argument is present,
|
|
* it must reference a String, which will receive the data.
|
|
* The _outbuf_ will contain only the received data after the method call
|
|
* even if it is not empty at the beginning.
|
|
*
|
|
* It raises EOFError on end of ARGF stream.
|
|
* Since ARGF stream is a concatenation of multiple files,
|
|
* internally EOF is occur for each file.
|
|
* ARGF.readpartial returns empty strings for EOFs except the last one and
|
|
* raises EOFError for the last one.
|
|
*
|
|
*/
|
|
|
|
static VALUE
|
|
argf_readpartial(int argc, VALUE *argv, VALUE argf)
|
|
{
|
|
return argf_getpartial(argc, argv, argf, Qnil, 0);
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* ARGF.read_nonblock(maxlen[, options]) -> string
|
|
* ARGF.read_nonblock(maxlen, outbuf[, options]) -> outbuf
|
|
*
|
|
* Reads at most _maxlen_ bytes from the ARGF stream in non-blocking mode.
|
|
*/
|
|
|
|
static VALUE
|
|
argf_read_nonblock(int argc, VALUE *argv, VALUE argf)
|
|
{
|
|
VALUE opts;
|
|
|
|
rb_scan_args(argc, argv, "11:", NULL, NULL, &opts);
|
|
|
|
if (!NIL_P(opts))
|
|
argc--;
|
|
|
|
return argf_getpartial(argc, argv, argf, opts, 1);
|
|
}
|
|
|
|
static VALUE
|
|
argf_getpartial(int argc, VALUE *argv, VALUE argf, VALUE opts, int nonblock)
|
|
{
|
|
VALUE tmp, str, length;
|
|
int no_exception;
|
|
|
|
rb_scan_args(argc, argv, "11", &length, &str);
|
|
if (!NIL_P(str)) {
|
|
StringValue(str);
|
|
argv[1] = str;
|
|
}
|
|
no_exception = no_exception_p(opts);
|
|
|
|
if (!next_argv()) {
|
|
if (!NIL_P(str)) {
|
|
rb_str_resize(str, 0);
|
|
}
|
|
rb_eof_error();
|
|
}
|
|
if (ARGF_GENERIC_INPUT_P()) {
|
|
VALUE (*const rescue_does_nothing)(VALUE, VALUE) = 0;
|
|
struct argf_call_arg arg;
|
|
arg.argc = argc;
|
|
arg.argv = argv;
|
|
arg.argf = argf;
|
|
tmp = rb_rescue2(argf_forward_call, (VALUE)&arg,
|
|
rescue_does_nothing, Qnil, rb_eEOFError, (VALUE)0);
|
|
}
|
|
else {
|
|
tmp = io_getpartial(argc, argv, ARGF.current_file, no_exception, nonblock);
|
|
}
|
|
if (NIL_P(tmp)) {
|
|
if (ARGF.next_p == -1) {
|
|
return io_nonblock_eof(no_exception);
|
|
}
|
|
argf_close(argf);
|
|
ARGF.next_p = 1;
|
|
if (RARRAY_LEN(ARGF.argv) == 0) {
|
|
return io_nonblock_eof(no_exception);
|
|
}
|
|
if (NIL_P(str))
|
|
str = rb_str_new(NULL, 0);
|
|
return str;
|
|
}
|
|
return tmp;
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* ARGF.getc -> String or nil
|
|
*
|
|
* Reads the next character from +ARGF+ and returns it as a +String+. Returns
|
|
* +nil+ at the end of the stream.
|
|
*
|
|
* +ARGF+ treats the files named on the command line as a single file created
|
|
* by concatenating their contents. After returning the last character of the
|
|
* first file, it returns the first character of the second file, and so on.
|
|
*
|
|
* For example:
|
|
*
|
|
* $ echo "foo" > file
|
|
* $ ruby argf.rb file
|
|
*
|
|
* ARGF.getc #=> "f"
|
|
* ARGF.getc #=> "o"
|
|
* ARGF.getc #=> "o"
|
|
* ARGF.getc #=> "\n"
|
|
* ARGF.getc #=> nil
|
|
* ARGF.getc #=> nil
|
|
*/
|
|
static VALUE
|
|
argf_getc(VALUE argf)
|
|
{
|
|
VALUE ch;
|
|
|
|
retry:
|
|
if (!next_argv()) return Qnil;
|
|
if (ARGF_GENERIC_INPUT_P()) {
|
|
ch = rb_funcall3(ARGF.current_file, rb_intern("getc"), 0, 0);
|
|
}
|
|
else {
|
|
ch = rb_io_getc(ARGF.current_file);
|
|
}
|
|
if (NIL_P(ch) && ARGF.next_p != -1) {
|
|
argf_close(argf);
|
|
ARGF.next_p = 1;
|
|
goto retry;
|
|
}
|
|
|
|
return ch;
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* ARGF.getbyte -> Integer or nil
|
|
*
|
|
* Gets the next 8-bit byte (0..255) from +ARGF+. Returns +nil+ if called at
|
|
* the end of the stream.
|
|
*
|
|
* For example:
|
|
*
|
|
* $ echo "foo" > file
|
|
* $ ruby argf.rb file
|
|
*
|
|
* ARGF.getbyte #=> 102
|
|
* ARGF.getbyte #=> 111
|
|
* ARGF.getbyte #=> 111
|
|
* ARGF.getbyte #=> 10
|
|
* ARGF.getbyte #=> nil
|
|
*/
|
|
static VALUE
|
|
argf_getbyte(VALUE argf)
|
|
{
|
|
VALUE ch;
|
|
|
|
retry:
|
|
if (!next_argv()) return Qnil;
|
|
if (!RB_TYPE_P(ARGF.current_file, T_FILE)) {
|
|
ch = rb_funcall3(ARGF.current_file, rb_intern("getbyte"), 0, 0);
|
|
}
|
|
else {
|
|
ch = rb_io_getbyte(ARGF.current_file);
|
|
}
|
|
if (NIL_P(ch) && ARGF.next_p != -1) {
|
|
argf_close(argf);
|
|
ARGF.next_p = 1;
|
|
goto retry;
|
|
}
|
|
|
|
return ch;
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* ARGF.readchar -> String or nil
|
|
*
|
|
* Reads the next character from +ARGF+ and returns it as a +String+. Raises
|
|
* an +EOFError+ after the last character of the last file has been read.
|
|
*
|
|
* For example:
|
|
*
|
|
* $ echo "foo" > file
|
|
* $ ruby argf.rb file
|
|
*
|
|
* ARGF.readchar #=> "f"
|
|
* ARGF.readchar #=> "o"
|
|
* ARGF.readchar #=> "o"
|
|
* ARGF.readchar #=> "\n"
|
|
* ARGF.readchar #=> end of file reached (EOFError)
|
|
*/
|
|
static VALUE
|
|
argf_readchar(VALUE argf)
|
|
{
|
|
VALUE ch;
|
|
|
|
retry:
|
|
if (!next_argv()) rb_eof_error();
|
|
if (!RB_TYPE_P(ARGF.current_file, T_FILE)) {
|
|
ch = rb_funcall3(ARGF.current_file, rb_intern("getc"), 0, 0);
|
|
}
|
|
else {
|
|
ch = rb_io_getc(ARGF.current_file);
|
|
}
|
|
if (NIL_P(ch) && ARGF.next_p != -1) {
|
|
argf_close(argf);
|
|
ARGF.next_p = 1;
|
|
goto retry;
|
|
}
|
|
|
|
return ch;
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* ARGF.readbyte -> Integer
|
|
*
|
|
* Reads the next 8-bit byte from ARGF and returns it as an +Integer+. Raises
|
|
* an +EOFError+ after the last byte of the last file has been read.
|
|
*
|
|
* For example:
|
|
*
|
|
* $ echo "foo" > file
|
|
* $ ruby argf.rb file
|
|
*
|
|
* ARGF.readbyte #=> 102
|
|
* ARGF.readbyte #=> 111
|
|
* ARGF.readbyte #=> 111
|
|
* ARGF.readbyte #=> 10
|
|
* ARGF.readbyte #=> end of file reached (EOFError)
|
|
*/
|
|
static VALUE
|
|
argf_readbyte(VALUE argf)
|
|
{
|
|
VALUE c;
|
|
|
|
NEXT_ARGF_FORWARD(0, 0);
|
|
c = argf_getbyte(argf);
|
|
if (NIL_P(c)) {
|
|
rb_eof_error();
|
|
}
|
|
return c;
|
|
}
|
|
|
|
#define FOREACH_ARGF() while (next_argv())
|
|
|
|
static VALUE
|
|
argf_block_call_i(RB_BLOCK_CALL_FUNC_ARGLIST(i, argf))
|
|
{
|
|
const VALUE current = ARGF.current_file;
|
|
rb_yield_values2(argc, argv);
|
|
if (ARGF.init_p == -1 || current != ARGF.current_file) {
|
|
rb_iter_break_value(Qundef);
|
|
}
|
|
return Qnil;
|
|
}
|
|
|
|
static void
|
|
argf_block_call(ID mid, int argc, VALUE *argv, VALUE argf)
|
|
{
|
|
VALUE ret = rb_block_call(ARGF.current_file, mid, argc, argv, argf_block_call_i, argf);
|
|
if (ret != Qundef) ARGF.next_p = 1;
|
|
}
|
|
|
|
static VALUE
|
|
argf_block_call_line_i(RB_BLOCK_CALL_FUNC_ARGLIST(i, argf))
|
|
{
|
|
if (!global_argf_p(argf)) {
|
|
ARGF.last_lineno = ++ARGF.lineno;
|
|
}
|
|
return argf_block_call_i(i, argf, argc, argv, blockarg);
|
|
}
|
|
|
|
static void
|
|
argf_block_call_line(ID mid, int argc, VALUE *argv, VALUE argf)
|
|
{
|
|
VALUE ret = rb_block_call(ARGF.current_file, mid, argc, argv, argf_block_call_line_i, argf);
|
|
if (ret != Qundef) ARGF.next_p = 1;
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* ARGF.each(sep=$/) {|line| block } -> ARGF
|
|
* ARGF.each(sep=$/, limit) {|line| block } -> ARGF
|
|
* ARGF.each(...) -> an_enumerator
|
|
*
|
|
* ARGF.each_line(sep=$/) {|line| block } -> ARGF
|
|
* ARGF.each_line(sep=$/, limit) {|line| block } -> ARGF
|
|
* ARGF.each_line(...) -> an_enumerator
|
|
*
|
|
* Returns an enumerator which iterates over each line (separated by _sep_,
|
|
* which defaults to your platform's newline character) of each file in
|
|
* +ARGV+. If a block is supplied, each line in turn will be yielded to the
|
|
* block, otherwise an enumerator is returned.
|
|
* The optional _limit_ argument is an +Integer+ specifying the maximum
|
|
* length of each line; longer lines will be split according to this limit.
|
|
*
|
|
* This method allows you to treat the files supplied on the command line as
|
|
* a single file consisting of the concatenation of each named file. After
|
|
* the last line of the first file has been returned, the first line of the
|
|
* second file is returned. The +ARGF.filename+ and +ARGF.lineno+ methods can
|
|
* be used to determine the filename of the current line and line number of
|
|
* the whole input, respectively.
|
|
*
|
|
* For example, the following code prints out each line of each named file
|
|
* prefixed with its line number, displaying the filename once per file:
|
|
*
|
|
* ARGF.each_line do |line|
|
|
* puts ARGF.filename if ARGF.file.lineno == 1
|
|
* puts "#{ARGF.file.lineno}: #{line}"
|
|
* end
|
|
*
|
|
* While the following code prints only the first file's name at first, and
|
|
* the contents with line number counted through all named files.
|
|
*
|
|
* ARGF.each_line do |line|
|
|
* puts ARGF.filename if ARGF.lineno == 1
|
|
* puts "#{ARGF.lineno}: #{line}"
|
|
* end
|
|
*/
|
|
static VALUE
|
|
argf_each_line(int argc, VALUE *argv, VALUE argf)
|
|
{
|
|
RETURN_ENUMERATOR(argf, argc, argv);
|
|
FOREACH_ARGF() {
|
|
argf_block_call_line(rb_intern("each_line"), argc, argv, argf);
|
|
}
|
|
return argf;
|
|
}
|
|
|
|
/*
|
|
* This is a deprecated alias for #each_line.
|
|
*/
|
|
|
|
static VALUE
|
|
argf_lines(int argc, VALUE *argv, VALUE argf)
|
|
{
|
|
rb_warn("ARGF#lines is deprecated; use #each_line instead");
|
|
if (!rb_block_given_p())
|
|
return rb_enumeratorize(argf, ID2SYM(rb_intern("each_line")), argc, argv);
|
|
return argf_each_line(argc, argv, argf);
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* ARGF.bytes {|byte| block } -> ARGF
|
|
* ARGF.bytes -> an_enumerator
|
|
*
|
|
* ARGF.each_byte {|byte| block } -> ARGF
|
|
* ARGF.each_byte -> an_enumerator
|
|
*
|
|
* Iterates over each byte of each file in +ARGV+.
|
|
* A byte is returned as an +Integer+ in the range 0..255.
|
|
*
|
|
* This method allows you to treat the files supplied on the command line as
|
|
* a single file consisting of the concatenation of each named file. After
|
|
* the last byte of the first file has been returned, the first byte of the
|
|
* second file is returned. The +ARGF.filename+ method can be used to
|
|
* determine the filename of the current byte.
|
|
*
|
|
* If no block is given, an enumerator is returned instead.
|
|
*
|
|
* For example:
|
|
*
|
|
* ARGF.bytes.to_a #=> [35, 32, ... 95, 10]
|
|
*
|
|
*/
|
|
static VALUE
|
|
argf_each_byte(VALUE argf)
|
|
{
|
|
RETURN_ENUMERATOR(argf, 0, 0);
|
|
FOREACH_ARGF() {
|
|
argf_block_call(rb_intern("each_byte"), 0, 0, argf);
|
|
}
|
|
return argf;
|
|
}
|
|
|
|
/*
|
|
* This is a deprecated alias for #each_byte.
|
|
*/
|
|
|
|
static VALUE
|
|
argf_bytes(VALUE argf)
|
|
{
|
|
rb_warn("ARGF#bytes is deprecated; use #each_byte instead");
|
|
if (!rb_block_given_p())
|
|
return rb_enumeratorize(argf, ID2SYM(rb_intern("each_byte")), 0, 0);
|
|
return argf_each_byte(argf);
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* ARGF.each_char {|char| block } -> ARGF
|
|
* ARGF.each_char -> an_enumerator
|
|
*
|
|
* Iterates over each character of each file in +ARGF+.
|
|
*
|
|
* This method allows you to treat the files supplied on the command line as
|
|
* a single file consisting of the concatenation of each named file. After
|
|
* the last character of the first file has been returned, the first
|
|
* character of the second file is returned. The +ARGF.filename+ method can
|
|
* be used to determine the name of the file in which the current character
|
|
* appears.
|
|
*
|
|
* If no block is given, an enumerator is returned instead.
|
|
*/
|
|
static VALUE
|
|
argf_each_char(VALUE argf)
|
|
{
|
|
RETURN_ENUMERATOR(argf, 0, 0);
|
|
FOREACH_ARGF() {
|
|
argf_block_call(rb_intern("each_char"), 0, 0, argf);
|
|
}
|
|
return argf;
|
|
}
|
|
|
|
/*
|
|
* This is a deprecated alias for #each_char.
|
|
*/
|
|
|
|
static VALUE
|
|
argf_chars(VALUE argf)
|
|
{
|
|
rb_warn("ARGF#chars is deprecated; use #each_char instead");
|
|
if (!rb_block_given_p())
|
|
return rb_enumeratorize(argf, ID2SYM(rb_intern("each_char")), 0, 0);
|
|
return argf_each_char(argf);
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* ARGF.each_codepoint {|codepoint| block } -> ARGF
|
|
* ARGF.each_codepoint -> an_enumerator
|
|
*
|
|
* Iterates over each codepoint of each file in +ARGF+.
|
|
*
|
|
* This method allows you to treat the files supplied on the command line as
|
|
* a single file consisting of the concatenation of each named file. After
|
|
* the last codepoint of the first file has been returned, the first
|
|
* codepoint of the second file is returned. The +ARGF.filename+ method can
|
|
* be used to determine the name of the file in which the current codepoint
|
|
* appears.
|
|
*
|
|
* If no block is given, an enumerator is returned instead.
|
|
*/
|
|
static VALUE
|
|
argf_each_codepoint(VALUE argf)
|
|
{
|
|
RETURN_ENUMERATOR(argf, 0, 0);
|
|
FOREACH_ARGF() {
|
|
argf_block_call(rb_intern("each_codepoint"), 0, 0, argf);
|
|
}
|
|
return argf;
|
|
}
|
|
|
|
/*
|
|
* This is a deprecated alias for #each_codepoint.
|
|
*/
|
|
|
|
static VALUE
|
|
argf_codepoints(VALUE argf)
|
|
{
|
|
rb_warn("ARGF#codepoints is deprecated; use #each_codepoint instead");
|
|
if (!rb_block_given_p())
|
|
return rb_enumeratorize(argf, ID2SYM(rb_intern("each_codepoint")), 0, 0);
|
|
return argf_each_codepoint(argf);
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* ARGF.filename -> String
|
|
* ARGF.path -> String
|
|
*
|
|
* Returns the current filename. "-" is returned when the current file is
|
|
* STDIN.
|
|
*
|
|
* For example:
|
|
*
|
|
* $ echo "foo" > foo
|
|
* $ echo "bar" > bar
|
|
* $ echo "glark" > glark
|
|
*
|
|
* $ ruby argf.rb foo bar glark
|
|
*
|
|
* ARGF.filename #=> "foo"
|
|
* ARGF.read(5) #=> "foo\nb"
|
|
* ARGF.filename #=> "bar"
|
|
* ARGF.skip
|
|
* ARGF.filename #=> "glark"
|
|
*/
|
|
static VALUE
|
|
argf_filename(VALUE argf)
|
|
{
|
|
next_argv();
|
|
return ARGF.filename;
|
|
}
|
|
|
|
static VALUE
|
|
argf_filename_getter(ID id, VALUE *var)
|
|
{
|
|
return argf_filename(*var);
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* ARGF.file -> IO or File object
|
|
*
|
|
* Returns the current file as an +IO+ or +File+ object.
|
|
* <code>$stdin</code> is returned when the current file is STDIN.
|
|
*
|
|
* For example:
|
|
*
|
|
* $ echo "foo" > foo
|
|
* $ echo "bar" > bar
|
|
*
|
|
* $ ruby argf.rb foo bar
|
|
*
|
|
* ARGF.file #=> #<File:foo>
|
|
* ARGF.read(5) #=> "foo\nb"
|
|
* ARGF.file #=> #<File:bar>
|
|
*/
|
|
static VALUE
|
|
argf_file(VALUE argf)
|
|
{
|
|
next_argv();
|
|
return ARGF.current_file;
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* ARGF.binmode -> ARGF
|
|
*
|
|
* Puts +ARGF+ into binary mode. Once a stream is in binary mode, it cannot
|
|
* be reset to non-binary mode. This option has the following effects:
|
|
*
|
|
* * Newline conversion is disabled.
|
|
* * Encoding conversion is disabled.
|
|
* * Content is treated as ASCII-8BIT.
|
|
*/
|
|
static VALUE
|
|
argf_binmode_m(VALUE argf)
|
|
{
|
|
ARGF.binmode = 1;
|
|
next_argv();
|
|
ARGF_FORWARD(0, 0);
|
|
rb_io_ascii8bit_binmode(ARGF.current_file);
|
|
return argf;
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* ARGF.binmode? -> true or false
|
|
*
|
|
* Returns true if +ARGF+ is being read in binary mode; false otherwise.
|
|
* To enable binary mode use +ARGF.binmode+.
|
|
*
|
|
* For example:
|
|
*
|
|
* ARGF.binmode? #=> false
|
|
* ARGF.binmode
|
|
* ARGF.binmode? #=> true
|
|
*/
|
|
static VALUE
|
|
argf_binmode_p(VALUE argf)
|
|
{
|
|
return ARGF.binmode ? Qtrue : Qfalse;
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* ARGF.skip -> ARGF
|
|
*
|
|
* Sets the current file to the next file in ARGV. If there aren't any more
|
|
* files it has no effect.
|
|
*
|
|
* For example:
|
|
*
|
|
* $ ruby argf.rb foo bar
|
|
* ARGF.filename #=> "foo"
|
|
* ARGF.skip
|
|
* ARGF.filename #=> "bar"
|
|
*/
|
|
static VALUE
|
|
argf_skip(VALUE argf)
|
|
{
|
|
if (ARGF.init_p && ARGF.next_p == 0) {
|
|
argf_close(argf);
|
|
ARGF.next_p = 1;
|
|
}
|
|
return argf;
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* ARGF.close -> ARGF
|
|
*
|
|
* Closes the current file and skips to the next file in ARGV. If there are
|
|
* no more files to open, just closes the current file. +STDIN+ will not be
|
|
* closed.
|
|
*
|
|
* For example:
|
|
*
|
|
* $ ruby argf.rb foo bar
|
|
*
|
|
* ARGF.filename #=> "foo"
|
|
* ARGF.close
|
|
* ARGF.filename #=> "bar"
|
|
* ARGF.close
|
|
*/
|
|
static VALUE
|
|
argf_close_m(VALUE argf)
|
|
{
|
|
next_argv();
|
|
argf_close(argf);
|
|
if (ARGF.next_p != -1) {
|
|
ARGF.next_p = 1;
|
|
}
|
|
ARGF.lineno = 0;
|
|
return argf;
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* ARGF.closed? -> true or false
|
|
*
|
|
* Returns _true_ if the current file has been closed; _false_ otherwise. Use
|
|
* +ARGF.close+ to actually close the current file.
|
|
*/
|
|
static VALUE
|
|
argf_closed(VALUE argf)
|
|
{
|
|
next_argv();
|
|
ARGF_FORWARD(0, 0);
|
|
return rb_io_closed(ARGF.current_file);
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* ARGF.to_s -> String
|
|
*
|
|
* Returns "ARGF".
|
|
*/
|
|
static VALUE
|
|
argf_to_s(VALUE argf)
|
|
{
|
|
return rb_str_new2("ARGF");
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* ARGF.inplace_mode -> String
|
|
*
|
|
* Returns the file extension appended to the names of modified files under
|
|
* in-place edit mode. This value can be set using +ARGF.inplace_mode=+ or
|
|
* passing the +-i+ switch to the Ruby binary.
|
|
*/
|
|
static VALUE
|
|
argf_inplace_mode_get(VALUE argf)
|
|
{
|
|
if (!ARGF.inplace) return Qnil;
|
|
if (NIL_P(ARGF.inplace)) return rb_str_new(0, 0);
|
|
return rb_str_dup(ARGF.inplace);
|
|
}
|
|
|
|
static VALUE
|
|
opt_i_get(ID id, VALUE *var)
|
|
{
|
|
return argf_inplace_mode_get(*var);
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* ARGF.inplace_mode = ext -> ARGF
|
|
*
|
|
* Sets the filename extension for in-place editing mode to the given String.
|
|
* Each file being edited has this value appended to its filename. The
|
|
* modified file is saved under this new name.
|
|
*
|
|
* For example:
|
|
*
|
|
* $ ruby argf.rb file.txt
|
|
*
|
|
* ARGF.inplace_mode = '.bak'
|
|
* ARGF.each_line do |line|
|
|
* print line.sub("foo","bar")
|
|
* end
|
|
*
|
|
* Each line of _file.txt_ has the first occurrence of "foo" replaced with
|
|
* "bar", then the new line is written out to _file.txt.bak_.
|
|
*/
|
|
static VALUE
|
|
argf_inplace_mode_set(VALUE argf, VALUE val)
|
|
{
|
|
if (rb_safe_level() >= 1 && OBJ_TAINTED(val))
|
|
rb_insecure_operation();
|
|
|
|
if (!RTEST(val)) {
|
|
ARGF.inplace = Qfalse;
|
|
}
|
|
else if (StringValueCStr(val), !RSTRING_LEN(val)) {
|
|
ARGF.inplace = Qnil;
|
|
}
|
|
else {
|
|
ARGF.inplace = rb_str_new_frozen(val);
|
|
}
|
|
return argf;
|
|
}
|
|
|
|
static void
|
|
opt_i_set(VALUE val, ID id, VALUE *var)
|
|
{
|
|
argf_inplace_mode_set(*var, val);
|
|
}
|
|
|
|
const char *
|
|
ruby_get_inplace_mode(void)
|
|
{
|
|
return RSTRING_PTR(ARGF.inplace);
|
|
}
|
|
|
|
void
|
|
ruby_set_inplace_mode(const char *suffix)
|
|
{
|
|
ARGF.inplace = !suffix ? Qfalse : !*suffix ? Qnil : rb_fstring_cstr(suffix);
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* ARGF.argv -> ARGV
|
|
*
|
|
* Returns the +ARGV+ array, which contains the arguments passed to your
|
|
* script, one per element.
|
|
*
|
|
* For example:
|
|
*
|
|
* $ ruby argf.rb -v glark.txt
|
|
*
|
|
* ARGF.argv #=> ["-v", "glark.txt"]
|
|
*
|
|
*/
|
|
static VALUE
|
|
argf_argv(VALUE argf)
|
|
{
|
|
return ARGF.argv;
|
|
}
|
|
|
|
static VALUE
|
|
argf_argv_getter(ID id, VALUE *var)
|
|
{
|
|
return argf_argv(*var);
|
|
}
|
|
|
|
VALUE
|
|
rb_get_argv(void)
|
|
{
|
|
return ARGF.argv;
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* ARGF.to_write_io -> io
|
|
*
|
|
* Returns IO instance tied to _ARGF_ for writing if inplace mode is
|
|
* enabled.
|
|
*/
|
|
static VALUE
|
|
argf_write_io(VALUE argf)
|
|
{
|
|
if (!RTEST(ARGF.current_file)) {
|
|
rb_raise(rb_eIOError, "not opened for writing");
|
|
}
|
|
return GetWriteIO(ARGF.current_file);
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* ARGF.write(string) -> integer
|
|
*
|
|
* Writes _string_ if inplace mode.
|
|
*/
|
|
static VALUE
|
|
argf_write(VALUE argf, VALUE str)
|
|
{
|
|
return rb_io_write(argf_write_io(argf), str);
|
|
}
|
|
|
|
void
|
|
rb_readwrite_sys_fail(enum rb_io_wait_readwrite writable, const char *mesg)
|
|
{
|
|
rb_readwrite_syserr_fail(writable, errno, mesg);
|
|
}
|
|
|
|
void
|
|
rb_readwrite_syserr_fail(enum rb_io_wait_readwrite writable, int n, const char *mesg)
|
|
{
|
|
VALUE arg;
|
|
arg = mesg ? rb_str_new2(mesg) : Qnil;
|
|
if (writable == RB_IO_WAIT_WRITABLE) {
|
|
switch (n) {
|
|
case EAGAIN:
|
|
rb_exc_raise(rb_class_new_instance(1, &arg, rb_eEAGAINWaitWritable));
|
|
break;
|
|
#if EAGAIN != EWOULDBLOCK
|
|
case EWOULDBLOCK:
|
|
rb_exc_raise(rb_class_new_instance(1, &arg, rb_eEWOULDBLOCKWaitWritable));
|
|
break;
|
|
#endif
|
|
case EINPROGRESS:
|
|
rb_exc_raise(rb_class_new_instance(1, &arg, rb_eEINPROGRESSWaitWritable));
|
|
break;
|
|
default:
|
|
rb_mod_sys_fail_str(rb_mWaitWritable, arg);
|
|
}
|
|
}
|
|
else if (writable == RB_IO_WAIT_READABLE) {
|
|
switch (n) {
|
|
case EAGAIN:
|
|
rb_exc_raise(rb_class_new_instance(1, &arg, rb_eEAGAINWaitReadable));
|
|
break;
|
|
#if EAGAIN != EWOULDBLOCK
|
|
case EWOULDBLOCK:
|
|
rb_exc_raise(rb_class_new_instance(1, &arg, rb_eEWOULDBLOCKWaitReadable));
|
|
break;
|
|
#endif
|
|
case EINPROGRESS:
|
|
rb_exc_raise(rb_class_new_instance(1, &arg, rb_eEINPROGRESSWaitReadable));
|
|
break;
|
|
default:
|
|
rb_mod_sys_fail_str(rb_mWaitReadable, arg);
|
|
}
|
|
}
|
|
else {
|
|
rb_bug("invalid read/write type passed to rb_readwrite_sys_fail: %d", writable);
|
|
}
|
|
}
|
|
|
|
static VALUE
|
|
get_LAST_READ_LINE(ID _x, VALUE *_y)
|
|
{
|
|
return rb_lastline_get();
|
|
}
|
|
|
|
static void
|
|
set_LAST_READ_LINE(VALUE val, ID _x, VALUE *_y)
|
|
{
|
|
rb_lastline_set(val);
|
|
}
|
|
|
|
/*
|
|
* Document-class: IOError
|
|
*
|
|
* Raised when an IO operation fails.
|
|
*
|
|
* File.open("/etc/hosts") {|f| f << "example"}
|
|
* #=> IOError: not opened for writing
|
|
*
|
|
* File.open("/etc/hosts") {|f| f.close; f.read }
|
|
* #=> IOError: closed stream
|
|
*
|
|
* Note that some IO failures raise <code>SystemCallError</code>s
|
|
* and these are not subclasses of IOError:
|
|
*
|
|
* File.open("does/not/exist")
|
|
* #=> Errno::ENOENT: No such file or directory - does/not/exist
|
|
*/
|
|
|
|
/*
|
|
* Document-class: EOFError
|
|
*
|
|
* Raised by some IO operations when reaching the end of file. Many IO
|
|
* methods exist in two forms,
|
|
*
|
|
* one that returns +nil+ when the end of file is reached, the other
|
|
* raises +EOFError+.
|
|
*
|
|
* +EOFError+ is a subclass of +IOError+.
|
|
*
|
|
* file = File.open("/etc/hosts")
|
|
* file.read
|
|
* file.gets #=> nil
|
|
* file.readline #=> EOFError: end of file reached
|
|
*/
|
|
|
|
/*
|
|
* Document-class: ARGF
|
|
*
|
|
* +ARGF+ is a stream designed for use in scripts that process files given as
|
|
* command-line arguments or passed in via STDIN.
|
|
*
|
|
* The arguments passed to your script are stored in the +ARGV+ Array, one
|
|
* argument per element. +ARGF+ assumes that any arguments that aren't
|
|
* filenames have been removed from +ARGV+. For example:
|
|
*
|
|
* $ ruby argf.rb --verbose file1 file2
|
|
*
|
|
* ARGV #=> ["--verbose", "file1", "file2"]
|
|
* option = ARGV.shift #=> "--verbose"
|
|
* ARGV #=> ["file1", "file2"]
|
|
*
|
|
* You can now use +ARGF+ to work with a concatenation of each of these named
|
|
* files. For instance, +ARGF.read+ will return the contents of _file1_
|
|
* followed by the contents of _file2_.
|
|
*
|
|
* After a file in +ARGV+ has been read +ARGF+ removes it from the Array.
|
|
* Thus, after all files have been read +ARGV+ will be empty.
|
|
*
|
|
* You can manipulate +ARGV+ yourself to control what +ARGF+ operates on. If
|
|
* you remove a file from +ARGV+, it is ignored by +ARGF+; if you add files to
|
|
* +ARGV+, they are treated as if they were named on the command line. For
|
|
* example:
|
|
*
|
|
* ARGV.replace ["file1"]
|
|
* ARGF.readlines # Returns the contents of file1 as an Array
|
|
* ARGV #=> []
|
|
* ARGV.replace ["file2", "file3"]
|
|
* ARGF.read # Returns the contents of file2 and file3
|
|
*
|
|
* If +ARGV+ is empty, +ARGF+ acts as if it contained STDIN, i.e. the data
|
|
* piped to your script. For example:
|
|
*
|
|
* $ echo "glark" | ruby -e 'p ARGF.read'
|
|
* "glark\n"
|
|
*/
|
|
|
|
/*
|
|
* The IO class is the basis for all input and output in Ruby.
|
|
* An I/O stream may be <em>duplexed</em> (that is, bidirectional), and
|
|
* so may use more than one native operating system stream.
|
|
*
|
|
* Many of the examples in this section use the File class, the only standard
|
|
* subclass of IO. The two classes are closely associated. Like the File
|
|
* class, the Socket library subclasses from IO (such as TCPSocket or
|
|
* UDPSocket).
|
|
*
|
|
* The Kernel#open method can create an IO (or File) object for these types
|
|
* of arguments:
|
|
*
|
|
* * A plain string represents a filename suitable for the underlying
|
|
* operating system.
|
|
*
|
|
* * A string starting with <code>"|"</code> indicates a subprocess.
|
|
* The remainder of the string following the <code>"|"</code> is
|
|
* invoked as a process with appropriate input/output channels
|
|
* connected to it.
|
|
*
|
|
* * A string equal to <code>"|-"</code> will create another Ruby
|
|
* instance as a subprocess.
|
|
*
|
|
* The IO may be opened with different file modes (read-only, write-only) and
|
|
* encodings for proper conversion. See IO.new for these options. See
|
|
* Kernel#open for details of the various command formats described above.
|
|
*
|
|
* IO.popen, the Open3 library, or Process#spawn may also be used to
|
|
* communicate with subprocesses through an IO.
|
|
*
|
|
* Ruby will convert pathnames between different operating system
|
|
* conventions if possible. For instance, on a Windows system the
|
|
* filename <code>"/gumby/ruby/test.rb"</code> will be opened as
|
|
* <code>"\gumby\ruby\test.rb"</code>. When specifying a Windows-style
|
|
* filename in a Ruby string, remember to escape the backslashes:
|
|
*
|
|
* "C:\\gumby\\ruby\\test.rb"
|
|
*
|
|
* Our examples here will use the Unix-style forward slashes;
|
|
* File::ALT_SEPARATOR can be used to get the platform-specific separator
|
|
* character.
|
|
*
|
|
* The global constant ARGF (also accessible as <code>$<</code>) provides an
|
|
* IO-like stream which allows access to all files mentioned on the
|
|
* command line (or STDIN if no files are mentioned). ARGF#path and its alias
|
|
* ARGF#filename are provided to access the name of the file currently being
|
|
* read.
|
|
*
|
|
* == io/console
|
|
*
|
|
* The io/console extension provides methods for interacting with the
|
|
* console. The console can be accessed from IO.console or the standard
|
|
* input/output/error IO objects.
|
|
*
|
|
* Requiring io/console adds the following methods:
|
|
*
|
|
* * IO::console
|
|
* * IO#raw
|
|
* * IO#raw!
|
|
* * IO#cooked
|
|
* * IO#cooked!
|
|
* * IO#getch
|
|
* * IO#echo=
|
|
* * IO#echo?
|
|
* * IO#noecho
|
|
* * IO#winsize
|
|
* * IO#winsize=
|
|
* * IO#iflush
|
|
* * IO#ioflush
|
|
* * IO#oflush
|
|
*
|
|
* Example:
|
|
*
|
|
* require 'io/console'
|
|
* rows, columns = $stdout.winsize
|
|
* puts "Your screen is #{columns} wide and #{rows} tall"
|
|
*/
|
|
|
|
void
|
|
Init_IO(void)
|
|
{
|
|
#undef rb_intern
|
|
#define rb_intern(str) rb_intern_const(str)
|
|
|
|
VALUE rb_cARGF;
|
|
#ifdef __CYGWIN__
|
|
#include <sys/cygwin.h>
|
|
static struct __cygwin_perfile pf[] =
|
|
{
|
|
{"", O_RDONLY | O_BINARY},
|
|
{"", O_WRONLY | O_BINARY},
|
|
{"", O_RDWR | O_BINARY},
|
|
{"", O_APPEND | O_BINARY},
|
|
{NULL, 0}
|
|
};
|
|
cygwin_internal(CW_PERFILE, pf);
|
|
#endif
|
|
|
|
rb_eIOError = rb_define_class("IOError", rb_eStandardError);
|
|
rb_eEOFError = rb_define_class("EOFError", rb_eIOError);
|
|
|
|
id_write = rb_intern("write");
|
|
id_read = rb_intern("read");
|
|
id_getc = rb_intern("getc");
|
|
id_flush = rb_intern("flush");
|
|
id_readpartial = rb_intern("readpartial");
|
|
id_set_encoding = rb_intern("set_encoding");
|
|
|
|
rb_define_global_function("syscall", rb_f_syscall, -1);
|
|
|
|
rb_define_global_function("open", rb_f_open, -1);
|
|
rb_define_global_function("printf", rb_f_printf, -1);
|
|
rb_define_global_function("print", rb_f_print, -1);
|
|
rb_define_global_function("putc", rb_f_putc, 1);
|
|
rb_define_global_function("puts", rb_f_puts, -1);
|
|
rb_define_global_function("gets", rb_f_gets, -1);
|
|
rb_define_global_function("readline", rb_f_readline, -1);
|
|
rb_define_global_function("select", rb_f_select, -1);
|
|
|
|
rb_define_global_function("readlines", rb_f_readlines, -1);
|
|
|
|
rb_define_global_function("`", rb_f_backquote, 1);
|
|
|
|
rb_define_global_function("p", rb_f_p, -1);
|
|
rb_define_method(rb_mKernel, "display", rb_obj_display, -1);
|
|
|
|
rb_cIO = rb_define_class("IO", rb_cObject);
|
|
rb_include_module(rb_cIO, rb_mEnumerable);
|
|
|
|
/* exception to wait for reading. see IO.select. */
|
|
rb_mWaitReadable = rb_define_module_under(rb_cIO, "WaitReadable");
|
|
/* exception to wait for writing. see IO.select. */
|
|
rb_mWaitWritable = rb_define_module_under(rb_cIO, "WaitWritable");
|
|
/* exception to wait for reading by EAGAIN. see IO.select. */
|
|
rb_eEAGAINWaitReadable = rb_define_class_under(rb_cIO, "EAGAINWaitReadable", rb_eEAGAIN);
|
|
rb_include_module(rb_eEAGAINWaitReadable, rb_mWaitReadable);
|
|
/* exception to wait for writing by EAGAIN. see IO.select. */
|
|
rb_eEAGAINWaitWritable = rb_define_class_under(rb_cIO, "EAGAINWaitWritable", rb_eEAGAIN);
|
|
rb_include_module(rb_eEAGAINWaitWritable, rb_mWaitWritable);
|
|
#if EAGAIN == EWOULDBLOCK
|
|
rb_eEWOULDBLOCKWaitReadable = rb_eEAGAINWaitReadable;
|
|
/* same as IO::EAGAINWaitReadable */
|
|
rb_define_const(rb_cIO, "EWOULDBLOCKWaitReadable", rb_eEAGAINWaitReadable);
|
|
rb_eEWOULDBLOCKWaitWritable = rb_eEAGAINWaitWritable;
|
|
/* same as IO::EAGAINWaitWritable */
|
|
rb_define_const(rb_cIO, "EWOULDBLOCKWaitWritable", rb_eEAGAINWaitWritable);
|
|
#else
|
|
/* exception to wait for reading by EWOULDBLOCK. see IO.select. */
|
|
rb_eEWOULDBLOCKWaitReadable = rb_define_class_under(rb_cIO, "EWOULDBLOCKWaitReadable", rb_eEWOULDBLOCK);
|
|
rb_include_module(rb_eEWOULDBLOCKWaitReadable, rb_mWaitReadable);
|
|
/* exception to wait for writing by EWOULDBLOCK. see IO.select. */
|
|
rb_eEWOULDBLOCKWaitWritable = rb_define_class_under(rb_cIO, "EWOULDBLOCKWaitWritable", rb_eEWOULDBLOCK);
|
|
rb_include_module(rb_eEWOULDBLOCKWaitWritable, rb_mWaitWritable);
|
|
#endif
|
|
/* exception to wait for reading by EINPROGRESS. see IO.select. */
|
|
rb_eEINPROGRESSWaitReadable = rb_define_class_under(rb_cIO, "EINPROGRESSWaitReadable", rb_eEINPROGRESS);
|
|
rb_include_module(rb_eEINPROGRESSWaitReadable, rb_mWaitReadable);
|
|
/* exception to wait for writing by EINPROGRESS. see IO.select. */
|
|
rb_eEINPROGRESSWaitWritable = rb_define_class_under(rb_cIO, "EINPROGRESSWaitWritable", rb_eEINPROGRESS);
|
|
rb_include_module(rb_eEINPROGRESSWaitWritable, rb_mWaitWritable);
|
|
|
|
#if 0
|
|
/* This is necessary only for forcing rdoc handle File::open */
|
|
rb_define_singleton_method(rb_cFile, "open", rb_io_s_open, -1);
|
|
#endif
|
|
|
|
rb_define_alloc_func(rb_cIO, io_alloc);
|
|
rb_define_singleton_method(rb_cIO, "new", rb_io_s_new, -1);
|
|
rb_define_singleton_method(rb_cIO, "open", rb_io_s_open, -1);
|
|
rb_define_singleton_method(rb_cIO, "sysopen", rb_io_s_sysopen, -1);
|
|
rb_define_singleton_method(rb_cIO, "for_fd", rb_io_s_for_fd, -1);
|
|
rb_define_singleton_method(rb_cIO, "popen", rb_io_s_popen, -1);
|
|
rb_define_singleton_method(rb_cIO, "foreach", rb_io_s_foreach, -1);
|
|
rb_define_singleton_method(rb_cIO, "readlines", rb_io_s_readlines, -1);
|
|
rb_define_singleton_method(rb_cIO, "read", rb_io_s_read, -1);
|
|
rb_define_singleton_method(rb_cIO, "binread", rb_io_s_binread, -1);
|
|
rb_define_singleton_method(rb_cIO, "write", rb_io_s_write, -1);
|
|
rb_define_singleton_method(rb_cIO, "binwrite", rb_io_s_binwrite, -1);
|
|
rb_define_singleton_method(rb_cIO, "select", rb_f_select, -1);
|
|
rb_define_singleton_method(rb_cIO, "pipe", rb_io_s_pipe, -1);
|
|
rb_define_singleton_method(rb_cIO, "try_convert", rb_io_s_try_convert, 1);
|
|
rb_define_singleton_method(rb_cIO, "copy_stream", rb_io_s_copy_stream, -1);
|
|
|
|
rb_define_method(rb_cIO, "initialize", rb_io_initialize, -1);
|
|
|
|
rb_output_fs = Qnil;
|
|
rb_define_hooked_variable("$,", &rb_output_fs, 0, rb_output_fs_setter);
|
|
|
|
rb_default_rs = rb_fstring_lit("\n"); /* avoid modifying RS_default */
|
|
rb_gc_register_mark_object(rb_default_rs);
|
|
rb_rs = rb_default_rs;
|
|
rb_output_rs = Qnil;
|
|
rb_define_hooked_variable("$/", &rb_rs, 0, rb_str_setter);
|
|
rb_define_hooked_variable("$-0", &rb_rs, 0, rb_str_setter);
|
|
rb_define_hooked_variable("$\\", &rb_output_rs, 0, rb_str_setter);
|
|
|
|
rb_define_virtual_variable("$_", get_LAST_READ_LINE, set_LAST_READ_LINE);
|
|
|
|
rb_define_method(rb_cIO, "initialize_copy", rb_io_init_copy, 1);
|
|
rb_define_method(rb_cIO, "reopen", rb_io_reopen, -1);
|
|
|
|
rb_define_method(rb_cIO, "print", rb_io_print, -1);
|
|
rb_define_method(rb_cIO, "putc", rb_io_putc, 1);
|
|
rb_define_method(rb_cIO, "puts", rb_io_puts, -1);
|
|
rb_define_method(rb_cIO, "printf", rb_io_printf, -1);
|
|
|
|
rb_define_method(rb_cIO, "each", rb_io_each_line, -1);
|
|
rb_define_method(rb_cIO, "each_line", rb_io_each_line, -1);
|
|
rb_define_method(rb_cIO, "each_byte", rb_io_each_byte, 0);
|
|
rb_define_method(rb_cIO, "each_char", rb_io_each_char, 0);
|
|
rb_define_method(rb_cIO, "each_codepoint", rb_io_each_codepoint, 0);
|
|
rb_define_method(rb_cIO, "lines", rb_io_lines, -1);
|
|
rb_define_method(rb_cIO, "bytes", rb_io_bytes, 0);
|
|
rb_define_method(rb_cIO, "chars", rb_io_chars, 0);
|
|
rb_define_method(rb_cIO, "codepoints", rb_io_codepoints, 0);
|
|
|
|
rb_define_method(rb_cIO, "syswrite", rb_io_syswrite, 1);
|
|
rb_define_method(rb_cIO, "sysread", rb_io_sysread, -1);
|
|
|
|
rb_define_method(rb_cIO, "pread", rb_io_pread, -1);
|
|
rb_define_method(rb_cIO, "pwrite", rb_io_pwrite, 2);
|
|
|
|
rb_define_method(rb_cIO, "fileno", rb_io_fileno, 0);
|
|
rb_define_alias(rb_cIO, "to_i", "fileno");
|
|
rb_define_method(rb_cIO, "to_io", rb_io_to_io, 0);
|
|
|
|
rb_define_method(rb_cIO, "fsync", rb_io_fsync, 0);
|
|
rb_define_method(rb_cIO, "fdatasync", rb_io_fdatasync, 0);
|
|
rb_define_method(rb_cIO, "sync", rb_io_sync, 0);
|
|
rb_define_method(rb_cIO, "sync=", rb_io_set_sync, 1);
|
|
|
|
rb_define_method(rb_cIO, "lineno", rb_io_lineno, 0);
|
|
rb_define_method(rb_cIO, "lineno=", rb_io_set_lineno, 1);
|
|
|
|
rb_define_method(rb_cIO, "readlines", rb_io_readlines, -1);
|
|
|
|
/* for prelude.rb use only: */
|
|
rb_define_private_method(rb_cIO, "__read_nonblock", io_read_nonblock, 3);
|
|
rb_define_private_method(rb_cIO, "__write_nonblock", io_write_nonblock, 2);
|
|
|
|
rb_define_method(rb_cIO, "readpartial", io_readpartial, -1);
|
|
rb_define_method(rb_cIO, "read", io_read, -1);
|
|
rb_define_method(rb_cIO, "write", io_write_m, -1);
|
|
rb_define_method(rb_cIO, "gets", rb_io_gets_m, -1);
|
|
rb_define_method(rb_cIO, "readline", rb_io_readline, -1);
|
|
rb_define_method(rb_cIO, "getc", rb_io_getc, 0);
|
|
rb_define_method(rb_cIO, "getbyte", rb_io_getbyte, 0);
|
|
rb_define_method(rb_cIO, "readchar", rb_io_readchar, 0);
|
|
rb_define_method(rb_cIO, "readbyte", rb_io_readbyte, 0);
|
|
rb_define_method(rb_cIO, "ungetbyte",rb_io_ungetbyte, 1);
|
|
rb_define_method(rb_cIO, "ungetc",rb_io_ungetc, 1);
|
|
rb_define_method(rb_cIO, "<<", rb_io_addstr, 1);
|
|
rb_define_method(rb_cIO, "flush", rb_io_flush, 0);
|
|
rb_define_method(rb_cIO, "tell", rb_io_tell, 0);
|
|
rb_define_method(rb_cIO, "seek", rb_io_seek_m, -1);
|
|
/* Set I/O position from the beginning */
|
|
rb_define_const(rb_cIO, "SEEK_SET", INT2FIX(SEEK_SET));
|
|
/* Set I/O position from the current position */
|
|
rb_define_const(rb_cIO, "SEEK_CUR", INT2FIX(SEEK_CUR));
|
|
/* Set I/O position from the end */
|
|
rb_define_const(rb_cIO, "SEEK_END", INT2FIX(SEEK_END));
|
|
#ifdef SEEK_DATA
|
|
/* Set I/O position to the next location containing data */
|
|
rb_define_const(rb_cIO, "SEEK_DATA", INT2FIX(SEEK_DATA));
|
|
#endif
|
|
#ifdef SEEK_HOLE
|
|
/* Set I/O position to the next hole */
|
|
rb_define_const(rb_cIO, "SEEK_HOLE", INT2FIX(SEEK_HOLE));
|
|
#endif
|
|
rb_define_method(rb_cIO, "rewind", rb_io_rewind, 0);
|
|
rb_define_method(rb_cIO, "pos", rb_io_tell, 0);
|
|
rb_define_method(rb_cIO, "pos=", rb_io_set_pos, 1);
|
|
rb_define_method(rb_cIO, "eof", rb_io_eof, 0);
|
|
rb_define_method(rb_cIO, "eof?", rb_io_eof, 0);
|
|
|
|
rb_define_method(rb_cIO, "close_on_exec?", rb_io_close_on_exec_p, 0);
|
|
rb_define_method(rb_cIO, "close_on_exec=", rb_io_set_close_on_exec, 1);
|
|
|
|
rb_define_method(rb_cIO, "close", rb_io_close_m, 0);
|
|
rb_define_method(rb_cIO, "closed?", rb_io_closed, 0);
|
|
rb_define_method(rb_cIO, "close_read", rb_io_close_read, 0);
|
|
rb_define_method(rb_cIO, "close_write", rb_io_close_write, 0);
|
|
|
|
rb_define_method(rb_cIO, "isatty", rb_io_isatty, 0);
|
|
rb_define_method(rb_cIO, "tty?", rb_io_isatty, 0);
|
|
rb_define_method(rb_cIO, "binmode", rb_io_binmode_m, 0);
|
|
rb_define_method(rb_cIO, "binmode?", rb_io_binmode_p, 0);
|
|
rb_define_method(rb_cIO, "sysseek", rb_io_sysseek, -1);
|
|
rb_define_method(rb_cIO, "advise", rb_io_advise, -1);
|
|
|
|
rb_define_method(rb_cIO, "ioctl", rb_io_ioctl, -1);
|
|
rb_define_method(rb_cIO, "fcntl", rb_io_fcntl, -1);
|
|
rb_define_method(rb_cIO, "pid", rb_io_pid, 0);
|
|
rb_define_method(rb_cIO, "inspect", rb_io_inspect, 0);
|
|
|
|
rb_define_method(rb_cIO, "external_encoding", rb_io_external_encoding, 0);
|
|
rb_define_method(rb_cIO, "internal_encoding", rb_io_internal_encoding, 0);
|
|
rb_define_method(rb_cIO, "set_encoding", rb_io_set_encoding, -1);
|
|
rb_define_method(rb_cIO, "set_encoding_by_bom", rb_io_set_encoding_by_bom, 0);
|
|
|
|
rb_define_method(rb_cIO, "autoclose?", rb_io_autoclose_p, 0);
|
|
rb_define_method(rb_cIO, "autoclose=", rb_io_set_autoclose, 1);
|
|
|
|
rb_define_variable("$stdin", &rb_stdin);
|
|
rb_stdin = prep_stdio(stdin, FMODE_READABLE, rb_cIO, "<STDIN>");
|
|
rb_define_hooked_variable("$stdout", &rb_stdout, 0, stdout_setter);
|
|
rb_stdout = prep_stdio(stdout, FMODE_WRITABLE, rb_cIO, "<STDOUT>");
|
|
rb_define_hooked_variable("$stderr", &rb_stderr, 0, stdout_setter);
|
|
rb_stderr = prep_stdio(stderr, FMODE_WRITABLE|FMODE_SYNC, rb_cIO, "<STDERR>");
|
|
rb_define_hooked_variable("$>", &rb_stdout, 0, stdout_setter);
|
|
orig_stdout = rb_stdout;
|
|
orig_stderr = rb_stderr;
|
|
|
|
/* Holds the original stdin */
|
|
rb_define_global_const("STDIN", rb_stdin);
|
|
/* Holds the original stdout */
|
|
rb_define_global_const("STDOUT", rb_stdout);
|
|
/* Holds the original stderr */
|
|
rb_define_global_const("STDERR", rb_stderr);
|
|
|
|
#if 0
|
|
/* Hack to get rdoc to regard ARGF as a class: */
|
|
rb_cARGF = rb_define_class("ARGF", rb_cObject);
|
|
#endif
|
|
|
|
rb_cARGF = rb_class_new(rb_cObject);
|
|
rb_set_class_path(rb_cARGF, rb_cObject, "ARGF.class");
|
|
rb_define_alloc_func(rb_cARGF, argf_alloc);
|
|
|
|
rb_include_module(rb_cARGF, rb_mEnumerable);
|
|
|
|
rb_define_method(rb_cARGF, "initialize", argf_initialize, -2);
|
|
rb_define_method(rb_cARGF, "initialize_copy", argf_initialize_copy, 1);
|
|
rb_define_method(rb_cARGF, "to_s", argf_to_s, 0);
|
|
rb_define_alias(rb_cARGF, "inspect", "to_s");
|
|
rb_define_method(rb_cARGF, "argv", argf_argv, 0);
|
|
|
|
rb_define_method(rb_cARGF, "fileno", argf_fileno, 0);
|
|
rb_define_method(rb_cARGF, "to_i", argf_fileno, 0);
|
|
rb_define_method(rb_cARGF, "to_io", argf_to_io, 0);
|
|
rb_define_method(rb_cARGF, "to_write_io", argf_write_io, 0);
|
|
rb_define_method(rb_cARGF, "each", argf_each_line, -1);
|
|
rb_define_method(rb_cARGF, "each_line", argf_each_line, -1);
|
|
rb_define_method(rb_cARGF, "each_byte", argf_each_byte, 0);
|
|
rb_define_method(rb_cARGF, "each_char", argf_each_char, 0);
|
|
rb_define_method(rb_cARGF, "each_codepoint", argf_each_codepoint, 0);
|
|
rb_define_method(rb_cARGF, "lines", argf_lines, -1);
|
|
rb_define_method(rb_cARGF, "bytes", argf_bytes, 0);
|
|
rb_define_method(rb_cARGF, "chars", argf_chars, 0);
|
|
rb_define_method(rb_cARGF, "codepoints", argf_codepoints, 0);
|
|
|
|
rb_define_method(rb_cARGF, "read", argf_read, -1);
|
|
rb_define_method(rb_cARGF, "readpartial", argf_readpartial, -1);
|
|
rb_define_method(rb_cARGF, "read_nonblock", argf_read_nonblock, -1);
|
|
rb_define_method(rb_cARGF, "readlines", argf_readlines, -1);
|
|
rb_define_method(rb_cARGF, "to_a", argf_readlines, -1);
|
|
rb_define_method(rb_cARGF, "gets", argf_gets, -1);
|
|
rb_define_method(rb_cARGF, "readline", argf_readline, -1);
|
|
rb_define_method(rb_cARGF, "getc", argf_getc, 0);
|
|
rb_define_method(rb_cARGF, "getbyte", argf_getbyte, 0);
|
|
rb_define_method(rb_cARGF, "readchar", argf_readchar, 0);
|
|
rb_define_method(rb_cARGF, "readbyte", argf_readbyte, 0);
|
|
rb_define_method(rb_cARGF, "tell", argf_tell, 0);
|
|
rb_define_method(rb_cARGF, "seek", argf_seek_m, -1);
|
|
rb_define_method(rb_cARGF, "rewind", argf_rewind, 0);
|
|
rb_define_method(rb_cARGF, "pos", argf_tell, 0);
|
|
rb_define_method(rb_cARGF, "pos=", argf_set_pos, 1);
|
|
rb_define_method(rb_cARGF, "eof", argf_eof, 0);
|
|
rb_define_method(rb_cARGF, "eof?", argf_eof, 0);
|
|
rb_define_method(rb_cARGF, "binmode", argf_binmode_m, 0);
|
|
rb_define_method(rb_cARGF, "binmode?", argf_binmode_p, 0);
|
|
|
|
rb_define_method(rb_cARGF, "write", argf_write, 1);
|
|
rb_define_method(rb_cARGF, "print", rb_io_print, -1);
|
|
rb_define_method(rb_cARGF, "putc", rb_io_putc, 1);
|
|
rb_define_method(rb_cARGF, "puts", rb_io_puts, -1);
|
|
rb_define_method(rb_cARGF, "printf", rb_io_printf, -1);
|
|
|
|
rb_define_method(rb_cARGF, "filename", argf_filename, 0);
|
|
rb_define_method(rb_cARGF, "path", argf_filename, 0);
|
|
rb_define_method(rb_cARGF, "file", argf_file, 0);
|
|
rb_define_method(rb_cARGF, "skip", argf_skip, 0);
|
|
rb_define_method(rb_cARGF, "close", argf_close_m, 0);
|
|
rb_define_method(rb_cARGF, "closed?", argf_closed, 0);
|
|
|
|
rb_define_method(rb_cARGF, "lineno", argf_lineno, 0);
|
|
rb_define_method(rb_cARGF, "lineno=", argf_set_lineno, 1);
|
|
|
|
rb_define_method(rb_cARGF, "inplace_mode", argf_inplace_mode_get, 0);
|
|
rb_define_method(rb_cARGF, "inplace_mode=", argf_inplace_mode_set, 1);
|
|
|
|
rb_define_method(rb_cARGF, "external_encoding", argf_external_encoding, 0);
|
|
rb_define_method(rb_cARGF, "internal_encoding", argf_internal_encoding, 0);
|
|
rb_define_method(rb_cARGF, "set_encoding", argf_set_encoding, -1);
|
|
|
|
argf = rb_class_new_instance(0, 0, rb_cARGF);
|
|
|
|
rb_define_readonly_variable("$<", &argf);
|
|
/*
|
|
* ARGF is a stream designed for use in scripts that process files given
|
|
* as command-line arguments or passed in via STDIN.
|
|
*
|
|
* See ARGF (the class) for more details.
|
|
*/
|
|
rb_define_global_const("ARGF", argf);
|
|
|
|
rb_define_hooked_variable("$.", &argf, argf_lineno_getter, argf_lineno_setter);
|
|
rb_define_hooked_variable("$FILENAME", &argf, argf_filename_getter, rb_gvar_readonly_setter);
|
|
ARGF.filename = rb_str_new2("-");
|
|
|
|
rb_define_hooked_variable("$-i", &argf, opt_i_get, opt_i_set);
|
|
rb_define_hooked_variable("$*", &argf, argf_argv_getter, rb_gvar_readonly_setter);
|
|
|
|
#if defined (_WIN32) || defined(__CYGWIN__)
|
|
atexit(pipe_atexit);
|
|
#endif
|
|
|
|
Init_File();
|
|
|
|
rb_define_method(rb_cFile, "initialize", rb_file_initialize, -1);
|
|
|
|
sym_mode = ID2SYM(rb_intern("mode"));
|
|
sym_perm = ID2SYM(rb_intern("perm"));
|
|
sym_flags = ID2SYM(rb_intern("flags"));
|
|
sym_extenc = ID2SYM(rb_intern("external_encoding"));
|
|
sym_intenc = ID2SYM(rb_intern("internal_encoding"));
|
|
sym_encoding = ID2SYM(rb_id_encoding());
|
|
sym_open_args = ID2SYM(rb_intern("open_args"));
|
|
sym_textmode = ID2SYM(rb_intern("textmode"));
|
|
sym_binmode = ID2SYM(rb_intern("binmode"));
|
|
sym_autoclose = ID2SYM(rb_intern("autoclose"));
|
|
sym_normal = ID2SYM(rb_intern("normal"));
|
|
sym_sequential = ID2SYM(rb_intern("sequential"));
|
|
sym_random = ID2SYM(rb_intern("random"));
|
|
sym_willneed = ID2SYM(rb_intern("willneed"));
|
|
sym_dontneed = ID2SYM(rb_intern("dontneed"));
|
|
sym_noreuse = ID2SYM(rb_intern("noreuse"));
|
|
sym_SET = ID2SYM(rb_intern("SET"));
|
|
sym_CUR = ID2SYM(rb_intern("CUR"));
|
|
sym_END = ID2SYM(rb_intern("END"));
|
|
#ifdef SEEK_DATA
|
|
sym_DATA = ID2SYM(rb_intern("DATA"));
|
|
#endif
|
|
#ifdef SEEK_HOLE
|
|
sym_HOLE = ID2SYM(rb_intern("HOLE"));
|
|
#endif
|
|
sym_wait_readable = ID2SYM(rb_intern("wait_readable"));
|
|
sym_wait_writable = ID2SYM(rb_intern("wait_writable"));
|
|
}
|