diff options
Diffstat (limited to 'ext/io/wait/wait.c')
| -rw-r--r-- | ext/io/wait/wait.c | 172 |
1 files changed, 4 insertions, 168 deletions
diff --git a/ext/io/wait/wait.c b/ext/io/wait/wait.c index d8bb55fc47..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,178 +11,13 @@ **********************************************************************/ -#include "ruby.h" -#include "ruby/io.h" - -#include <sys/types.h> -#if defined(HAVE_UNISTD_H) && (defined(__sun)) -#include <unistd.h> -#endif -#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_readable _((int argc, VALUE *argv, VALUE io)); -static VALUE io_wait_writable _((int argc, VALUE *argv, VALUE io)); -void Init_wait _((void)); - -/* - * call-seq: - * io.nread -> int - * - * Returns number of bytes that can be read without blocking. - * Returns zero if no information available. - */ - -static VALUE -io_nread(VALUE io) -{ - rb_io_t *fptr; - int len; - ioctl_arg n; - - GetOpenFile(io, fptr); - rb_io_check_readable(fptr); - len = rb_io_read_pending(fptr); - if (len > 0) return len; - if (!FIONREAD_POSSIBLE_P(fptr->fd)) return INT2FIX(0); - if (ioctl(fptr->fd, FIONREAD, &n)) return INT2FIX(0); - if (n > 0) return ioctl_arg2num(n); - return INT2FIX(0); -} - -/* - * call-seq: - * io.ready? -> true, false or nil - * - * Returns true if input available without blocking, or false. - * Returns nil if no information available. - */ - -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 Qnil; - if (ioctl(fptr->fd, FIONREAD, &n)) return Qnil; - if (n > 0) return Qtrue; - return Qfalse; -} - -/* - * call-seq: - * io.wait -> IO, true, false or nil - * io.wait(timeout) -> IO, true, false or nil - * io.wait_readable -> IO, true, false or nil - * io.wait_readable(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_readable(int argc, VALUE *argv, VALUE io) -{ - rb_io_t *fptr; - int i; - ioctl_arg n; - VALUE timeout; - struct timeval timerec; - struct timeval *tv; - - GetOpenFile(io, fptr); - rb_io_check_readable(fptr); - rb_scan_args(argc, argv, "01", &timeout); - if (NIL_P(timeout)) { - tv = NULL; - } - else { - timerec = rb_time_interval(timeout); - tv = &timerec; - } - - if (rb_io_read_pending(fptr)) return Qtrue; - if (!FIONREAD_POSSIBLE_P(fptr->fd)) return Qfalse; - i = rb_wait_for_single_fd(fptr->fd, RB_WAITFD_IN, tv); - 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; -} - -/* - * call-seq: - * io.wait_writable -> IO - * io.wait_writable(timeout) -> IO or nil - * - * Waits until IO writable is available or times out and returns self or - * nil when EOF is reached. - */ -static VALUE -io_wait_writable(int argc, VALUE *argv, VALUE io) -{ - rb_io_t *fptr; - int i; - VALUE timeout; - struct timeval timerec; - struct timeval *tv; - - GetOpenFile(io, fptr); - rb_io_check_writable(fptr); - rb_scan_args(argc, argv, "01", &timeout); - if (NIL_P(timeout)) { - tv = NULL; - } - else { - timerec = rb_time_interval(timeout); - tv = &timerec; - } - - i = rb_wait_for_single_fd(fptr->fd, RB_WAITFD_OUT, tv); - if (i < 0) - rb_sys_fail(0); - rb_io_check_closed(fptr); - if (i & RB_WAITFD_OUT) - 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, "nread", io_nread, 0); - rb_define_method(rb_cIO, "ready?", io_ready_p, 0); - rb_define_method(rb_cIO, "wait", io_wait_readable, -1); - rb_define_method(rb_cIO, "wait_readable", io_wait_readable, -1); - rb_define_method(rb_cIO, "wait_writable", io_wait_writable, -1); } |
