diff options
author | nobu <nobu@b2dd03c8-39d4-4d8f-98ff-823fe69b080e> | 2003-07-23 09:26:53 +0000 |
---|---|---|
committer | nobu <nobu@b2dd03c8-39d4-4d8f-98ff-823fe69b080e> | 2003-07-23 09:26:53 +0000 |
commit | eb5a52e0706c7321eaf04f1c3240618532db8ca3 (patch) | |
tree | c7535f1c700e1b070efa8bd26e7a2b1146bd6722 /ext/io/wait/wait.c | |
parent | 2436525ed8cd7b95a3a98173afd23804fedf6676 (diff) | |
download | ruby-eb5a52e0706c7321eaf04f1c3240618532db8ca3.tar.gz |
* ext/io/wait: imported.
git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@4124 b2dd03c8-39d4-4d8f-98ff-823fe69b080e
Diffstat (limited to 'ext/io/wait/wait.c')
-rw-r--r-- | ext/io/wait/wait.c | 106 |
1 files changed, 106 insertions, 0 deletions
diff --git a/ext/io/wait/wait.c b/ext/io/wait/wait.c new file mode 100644 index 0000000000..53d5bd7d18 --- /dev/null +++ b/ext/io/wait/wait.c @@ -0,0 +1,106 @@ +/********************************************************************** + + io/wait.c - + + $Author$ + $Date$ + created at: Tue Aug 28 09:08:06 JST 2001 + + All the files in this distribution are covered under the Ruby's + license (see the file COPYING). + +**********************************************************************/ + +#include "ruby.h" +#include "rubyio.h" + +#include <sys/types.h> +#include FIONREAD_HEADER + +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)); + +/* +=begin += IO wait methods. +=end + */ + +/* +=begin +--- IO#ready? + returns non-nil if input available without blocking, or nil. +=end +*/ +static VALUE +io_ready_p(io) + VALUE io; +{ + OpenFile *fptr; + FILE *fp; + int n; + + GetOpenFile(io, fptr); + rb_io_check_readable(fptr); + fp = fptr->f; + if (feof(fp)) return Qfalse; + if (rb_read_pending(fp)) return Qtrue; + if (ioctl(fileno(fp), FIONREAD, &n)) rb_sys_fail(0); + if (n > 0) return INT2NUM(n); + return Qnil; +} + +/* +=begin +--- IO#wait([timeout]) + waits until input available or timed out and returns self, or nil + when EOF reached. +=end +*/ +static VALUE +io_wait(argc, argv, io) + int argc; + VALUE *argv; + VALUE io; +{ + OpenFile *fptr; + fd_set rd; + FILE *fp; + int fd, n; + VALUE timeout; + struct timeval *tp, timerec; + + GetOpenFile(io, fptr); + rb_io_check_readable(fptr); + rb_scan_args(argc, argv, "01", &timeout); + if (NIL_P(timeout)) { + tp = 0; + } + else { + timerec = rb_time_interval(timeout); + tp = &timerec; + } + + fp = fptr->f; + if (feof(fp)) return Qfalse; + if (rb_read_pending(fp)) return Qtrue; + fd = fileno(fp); + FD_ZERO(&rd); + FD_SET(fd, &rd); + if (rb_thread_select(fd + 1, &rd, NULL, NULL, tp) < 0) + rb_sys_fail(0); + rb_io_check_closed(fptr); + if (ioctl(fileno(fp), FIONREAD, &n)) rb_sys_fail(0); + if (n > 0) return io; + return Qnil; +} + +void +Init_wait() +{ + rb_define_method(rb_cIO, "ready?", io_ready_p, 0); + rb_define_method(rb_cIO, "wait", io_wait, -1); +} |