1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
|
/**********************************************************************
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;
int n;
GetOpenFile(io, fptr);
rb_io_check_readable(fptr);
if (rb_io_read_pending(fptr)) return Qtrue;
if (ioctl(fptr->fd, FIONREAD, &n)) rb_sys_fail(0);
if (n > 0) return INT2NUM(n);
return Qnil;
}
struct wait_readable_arg {
rb_fdset_t fds;
struct timeval *timeout;
};
#ifdef HAVE_RB_FD_INIT
static VALUE
wait_readable(p)
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
/*
=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;
struct wait_readable_arg arg;
int fd, 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;
fd = fptr->fd;
rb_fd_init(&arg.fds);
rb_fd_set(fd, &arg.fds);
#ifdef HAVE_RB_FD_INIT
n = (int)rb_ensure(wait_readable, (VALUE)&arg,
(VALUE (*)_((VALUE)))rb_fd_term, (VALUE)&arg.fds);
#else
n = rb_thread_select(fd + 1, rb_fd_ptr(&rd), NULL, NULL, tp)
#endif
if (n < 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;
}
void
Init_wait()
{
rb_define_method(rb_cIO, "ready?", io_ready_p, 0);
rb_define_method(rb_cIO, "wait", io_wait, -1);
}
|
