diff options
Diffstat (limited to 'ext/io/wait/wait.c')
| -rw-r--r-- | ext/io/wait/wait.c | 126 |
1 files changed, 4 insertions, 122 deletions
diff --git a/ext/io/wait/wait.c b/ext/io/wait/wait.c index 42f5e8fc48..f7575191fe 100644 --- a/ext/io/wait/wait.c +++ b/ext/io/wait/wait.c @@ -1,3 +1,4 @@ +/* -*- c-file-style: "ruby"; indent-tabs-mode: t -*- */ /********************************************************************** io/wait.c - @@ -10,132 +11,13 @@ **********************************************************************/ -#include "ruby.h" -#include "ruby/io.h" - -#include <sys/types.h> -#if defined(HAVE_SYS_IOCTL_H) -#include <sys/ioctl.h> -#endif -#if defined(FIONREAD_HEADER) -#include FIONREAD_HEADER -#endif - -#ifdef HAVE_RB_W32_IOCTLSOCKET -#define ioctl ioctlsocket -#define ioctl_arg u_long -#define ioctl_arg2num(i) ULONG2NUM(i) -#else -#define ioctl_arg int -#define ioctl_arg2num(i) INT2NUM(i) -#endif - -#ifdef HAVE_RB_W32_IS_SOCKET -#define FIONREAD_POSSIBLE_P(fd) rb_w32_is_socket(fd) -#else -#define FIONREAD_POSSIBLE_P(fd) ((void)(fd),Qtrue) -#endif - -static VALUE io_ready_p _((VALUE io)); -static VALUE io_wait _((int argc, VALUE *argv, VALUE io)); -void Init_wait _((void)); - -EXTERN struct timeval rb_time_interval _((VALUE time)); - -/* - * call-seq: - * io.ready? -> true, false or nil - * - * Returns non-nil if input available without blocking, or nil. - */ - -static VALUE -io_ready_p(VALUE io) -{ - rb_io_t *fptr; - ioctl_arg n; - - GetOpenFile(io, fptr); - rb_io_check_readable(fptr); - if (rb_io_read_pending(fptr)) return Qtrue; - if (!FIONREAD_POSSIBLE_P(fptr->fd)) return Qfalse; - if (ioctl(fptr->fd, FIONREAD, &n)) rb_sys_fail(0); - if (n > 0) return ioctl_arg2num(n); - return Qnil; -} - -struct wait_readable_arg { - rb_fdset_t fds; - struct timeval *timeout; -}; - -#ifdef HAVE_RB_FD_INIT -static VALUE -wait_readable(VALUE p) -{ - struct wait_readable_arg *arg = (struct wait_readable_arg *)p; - rb_fdset_t *fds = &arg->fds; - - return (VALUE)rb_thread_select(rb_fd_max(fds), rb_fd_ptr(fds), NULL, NULL, arg->timeout); -} -#endif - -/* - * call-seq: - * io.wait -> IO, true, false or nil - * io.wait(timeout) -> IO, true, false or nil - * - * Waits until input is available or times out and returns self or nil when - * EOF is reached. - */ - -static VALUE -io_wait(int argc, VALUE *argv, VALUE io) -{ - rb_io_t *fptr; - struct wait_readable_arg arg; - int fd, i; - ioctl_arg n; - VALUE timeout; - struct timeval timerec; - - GetOpenFile(io, fptr); - rb_io_check_readable(fptr); - rb_scan_args(argc, argv, "01", &timeout); - if (NIL_P(timeout)) { - arg.timeout = 0; - } - else { - timerec = rb_time_interval(timeout); - arg.timeout = &timerec; - } - - if (rb_io_read_pending(fptr)) return Qtrue; - if (!FIONREAD_POSSIBLE_P(fptr->fd)) return Qfalse; - fd = fptr->fd; - rb_fd_init(&arg.fds); - rb_fd_set(fd, &arg.fds); -#ifdef HAVE_RB_FD_INIT - i = (int)rb_ensure(wait_readable, (VALUE)&arg, - (VALUE (*)_((VALUE)))rb_fd_term, (VALUE)&arg.fds); -#else - i = rb_thread_select(fd + 1, rb_fd_ptr(&arg.fds), NULL, NULL, arg.timeout); -#endif - if (i < 0) - rb_sys_fail(0); - rb_io_check_closed(fptr); - if (ioctl(fptr->fd, FIONREAD, &n)) rb_sys_fail(0); - if (n > 0) return io; - return Qnil; -} +#include "ruby.h" /* abi_version */ /* - * IO wait methods + * IO wait methods are built in ruby now, just for backward compatibility. */ void -Init_wait() +Init_wait(void) { - rb_define_method(rb_cIO, "ready?", io_ready_p, 0); - rb_define_method(rb_cIO, "wait", io_wait, -1); } |
