From fcbf63e62c627deae76c1b8cb8c0876c536ed811 Mon Sep 17 00:00:00 2001 From: Jari Vetoniemi Date: Mon, 16 Mar 2020 18:49:26 +0900 Subject: Fresh start --- jni/ruby/ext/io/nonblock/nonblock.c | 138 ++++++++++++++++++++++++++++++++++++ 1 file changed, 138 insertions(+) create mode 100644 jni/ruby/ext/io/nonblock/nonblock.c (limited to 'jni/ruby/ext/io/nonblock/nonblock.c') diff --git a/jni/ruby/ext/io/nonblock/nonblock.c b/jni/ruby/ext/io/nonblock/nonblock.c new file mode 100644 index 0000000..792b6f0 --- /dev/null +++ b/jni/ruby/ext/io/nonblock/nonblock.c @@ -0,0 +1,138 @@ +/********************************************************************** + + io/wait.c - + + $Author: usa $ + created at: Tue Jul 14 21:53:18 2009 + + All the files in this distribution are covered under the Ruby's + license (see the file COPYING). + +**********************************************************************/ + +#include "ruby.h" +#include "ruby/io.h" +#ifdef HAVE_UNISTD_H +#include +#endif +#include + +#ifdef F_GETFL +static int +io_nonblock_mode(int fd) +{ + int f = fcntl(fd, F_GETFL); + if (f == -1) rb_sys_fail(0); + return f; +} +#else +#define io_nonblock_mode(fd) ((void)(fd), 0) +#endif + +#ifdef F_GETFL +/* + * call-seq: + * io.nonblock? -> boolean + * + * Returns +true+ if an IO object is in non-blocking mode. + */ +static VALUE +rb_io_nonblock_p(VALUE io) +{ + rb_io_t *fptr; + GetOpenFile(io, fptr); + if (io_nonblock_mode(fptr->fd) & O_NONBLOCK) + return Qtrue; + return Qfalse; +} +#else +#define rb_io_nonblock_p rb_f_notimplement +#endif + +#ifdef F_SETFL +static void +io_nonblock_set(int fd, int f, int nb) +{ + if (nb) { + if ((f & O_NONBLOCK) != 0) + return; + f |= O_NONBLOCK; + } + else { + if ((f & O_NONBLOCK) == 0) + return; + f &= ~O_NONBLOCK; + } + if (fcntl(fd, F_SETFL, f) == -1) + rb_sys_fail(0); +} + +/* + * call-seq: + * io.nonblock = boolean -> boolean + * + * Enables non-blocking mode on a stream when set to + * +true+, and blocking mode when set to +false+. + */ +static VALUE +rb_io_nonblock_set(VALUE io, VALUE nb) +{ + rb_io_t *fptr; + GetOpenFile(io, fptr); + if (RTEST(nb)) + rb_io_set_nonblock(fptr); + else + io_nonblock_set(fptr->fd, io_nonblock_mode(fptr->fd), RTEST(nb)); + return io; +} + +static VALUE +io_nonblock_restore(VALUE arg) +{ + int *restore = (int *)arg; + if (fcntl(restore[0], F_SETFL, restore[1]) == -1) + rb_sys_fail(0); + return Qnil; +} + +/* + * call-seq: + * io.nonblock {|io| } -> io + * io.nonblock(boolean) {|io| } -> io + * + * Yields +self+ in non-blocking mode. + * + * When +false+ is given as an argument, +self+ is yielded in blocking mode. + * The original mode is restored after the block is executed. + */ +static VALUE +rb_io_nonblock_block(int argc, VALUE *argv, VALUE io) +{ + int nb = 1; + rb_io_t *fptr; + int f, restore[2]; + + GetOpenFile(io, fptr); + if (argc > 0) { + VALUE v; + rb_scan_args(argc, argv, "01", &v); + nb = RTEST(v); + } + f = io_nonblock_mode(fptr->fd); + restore[0] = fptr->fd; + restore[1] = f; + io_nonblock_set(fptr->fd, f, nb); + return rb_ensure(rb_yield, io, io_nonblock_restore, (VALUE)restore); +} +#else +#define rb_io_nonblock_set rb_f_notimplement +#define rb_io_nonblock_block rb_f_notimplement +#endif + +void +Init_nonblock(void) +{ + rb_define_method(rb_cIO, "nonblock?", rb_io_nonblock_p, 0); + rb_define_method(rb_cIO, "nonblock=", rb_io_nonblock_set, 1); + rb_define_method(rb_cIO, "nonblock", rb_io_nonblock_block, -1); +} -- cgit v1.2.3