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/goruby.c | 60 +++++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 60 insertions(+) create mode 100644 jni/ruby/goruby.c (limited to 'jni/ruby/goruby.c') diff --git a/jni/ruby/goruby.c b/jni/ruby/goruby.c new file mode 100644 index 0000000..99042f8 --- /dev/null +++ b/jni/ruby/goruby.c @@ -0,0 +1,60 @@ +void Init_golf(void); +#define ruby_options goruby_options +#define ruby_run_node goruby_run_node +#include "main.c" +#undef ruby_options +#undef ruby_run_node + +#if defined _WIN32 +#include +#include +#define pipe(p) _pipe(p, 32L, _O_NOINHERIT) +#elif defined HAVE_UNISTD_H +#include +#endif + +RUBY_EXTERN void *ruby_options(int argc, char **argv); +RUBY_EXTERN int ruby_run_node(void*); +RUBY_EXTERN void ruby_init_ext(const char *name, void (*init)(void)); + +static VALUE +init_golf(VALUE arg) +{ + ruby_init_ext("golf.so", Init_golf); + return arg; +} + +void * +goruby_options(int argc, char **argv) +{ + static const char cmd[] = "END{require 'irb';IRB.start}"; + int rw[2], infd; + void *ret; + + if ((isatty(0) && isatty(1) && isatty(2)) && (pipe(rw) == 0)) { + ssize_t n; + infd = dup(0); + if (infd < 0) return NULL; + dup2(rw[0], 0); + close(rw[0]); + n = write(rw[1], cmd, sizeof(cmd) - 1); + close(rw[1]); + ret = n > 0 ? ruby_options(argc, argv) : NULL; + dup2(infd, 0); + close(infd); + return ret; + } + else { + return ruby_options(argc, argv); + } +} + +int +goruby_run_node(void *arg) +{ + int state; + if (NIL_P(rb_protect(init_golf, Qtrue, &state))) { + return state == EXIT_SUCCESS ? EXIT_FAILURE : state; + } + return ruby_run_node(arg); +} -- cgit v1.2.3