polyphony 0.39 → 0.43.1

data/ext/gyro/socket.c

@@ -1,210 +0,0 @@
-#include "gyro.h"
-#include <sys/socket.h>
-
-static VALUE cTCPSocket;
-
-///////////////////////////////////////////////////////////////////////////
-
-struct rsock_send_arg {
-    int fd, flags;
-    VALUE mesg;
-    struct sockaddr *to;
-    socklen_t tolen;
-};
-
-#define StringValue(v) rb_string_value(&(v))
-#define IS_ADDRINFO(obj) rb_typeddata_is_kind_of((obj), &addrinfo_type)
-
-VALUE
-rsock_sockaddr_string_value(volatile VALUE *v)
-{
-    // VALUE val = *v;
-    // if (IS_ADDRINFO(val)) {
-    //     *v = addrinfo_to_sockaddr(val);
-    // }
-    StringValue(*v);
-    return *v;
-}
-
-#define SockAddrStringValue(v) rsock_sockaddr_string_value(&(v))
-#define RSTRING_LENINT(str) rb_long2int(RSTRING_LEN(str))
-#ifndef RSTRING_SOCKLEN
-#  define RSTRING_SOCKLEN (socklen_t)RSTRING_LENINT
-#endif
-
-#if defined __APPLE__
-# define do_write_retry(code) do {ret = code;} while (ret == -1 && errno == EPROTOTYPE)
-#else
-# define do_write_retry(code) ret = code
-#endif
-
-VALUE
-rsock_sendto_blocking(void *data)
-{
-    struct rsock_send_arg *arg = data;
-    VALUE mesg = arg->mesg;
-    ssize_t ret;
-    do_write_retry(sendto(arg->fd, RSTRING_PTR(mesg), RSTRING_LEN(mesg),
-			  arg->flags, arg->to, arg->tolen));
-    return (VALUE)ret;
-}
-
-VALUE
-rsock_send_blocking(void *data)
-{
-    struct rsock_send_arg *arg = data;
-    VALUE mesg = arg->mesg;
-    ssize_t ret;
-    do_write_retry(send(arg->fd, RSTRING_PTR(mesg), RSTRING_LEN(mesg),
-			arg->flags));
-    return (VALUE)ret;
-}
-
-///////////////////////////////////////////////////////////////////////////
-
-static VALUE BasicSocket_send(int argc, VALUE *argv, VALUE sock) {
-  VALUE underlying_socket = rb_iv_get(sock, "@io");
-  if (!NIL_P(underlying_socket)) sock = underlying_socket;
-  struct rsock_send_arg arg;
-  VALUE flags, to;
-  rb_io_t *fptr;
-  ssize_t n;
-  rb_blocking_function_t *func;
-  const char *funcname;
-  VALUE write_watcher = Qnil;
-  
-  rb_scan_args(argc, argv, "21", &arg.mesg, &flags, &to);
-
-  StringValue(arg.mesg);
-
-  if (!NIL_P(to)) {
-    SockAddrStringValue(to);
-    to = rb_str_new4(to);
-    arg.to = (struct sockaddr *)RSTRING_PTR(to);
-    arg.tolen = RSTRING_SOCKLEN(to);
-    func = rsock_sendto_blocking;
-    funcname = "sendto(2)";
-  }
-  else {
-    func = rsock_send_blocking;
-    funcname = "send(2)";
-  }
-  GetOpenFile(sock, fptr);
-  rb_io_set_nonblock(fptr);
-  arg.fd = fptr->fd;
-  arg.flags = NUM2INT(flags);
-  while ((n = (ssize_t)func(&arg)) < 0) {
-    if (NIL_P(write_watcher))
-      write_watcher = Gyro_IO_auto_io(fptr->fd, EV_WRITE);
-    Gyro_IO_await(write_watcher);
-  }
-  return SSIZET2NUM(n);
-}
-
-static VALUE BasicSocket_recv(int argc, VALUE *argv, VALUE sock) {
-  VALUE underlying_socket = rb_iv_get(sock, "@io");
-  if (!NIL_P(underlying_socket)) sock = underlying_socket;
-  long len = argc >= 1 ? NUM2LONG(argv[0]) : 8192;
-  if (len < 0) {
-    rb_raise(rb_eArgError, "negative length %ld given", len);
-  }
-
-  rb_io_t *fptr;
-  long n;
-  int shrinkable;
-  VALUE read_watcher = Qnil;
-
-
-  VALUE str = argc >= 3 ? argv[2] : Qnil;
-
-  shrinkable = io_setstrbuf(&str, len);
-  OBJ_TAINT(str);
-  GetOpenFile(sock, fptr);
-  // rb_io_set_nonblock(fptr);
-  rb_io_check_byte_readable(fptr);
-
-  if (len == 0)
-  	return str;
-
-  while (1) {
-    n = recv(fptr->fd, RSTRING_PTR(str), len, MSG_DONTWAIT);
-    if (n < 0) {
-      int e = errno;
-      if (e == EWOULDBLOCK || e == EAGAIN) {
-        if (NIL_P(read_watcher))
-          read_watcher = Gyro_IO_auto_io(fptr->fd, EV_READ);
-        Gyro_IO_await(read_watcher);
-      }
-      else
-        rb_syserr_fail(e, strerror(e));
-        // rb_syserr_fail_path(e, fptr->pathv);
-    }
-    else
-      break;
-  }
-
-  io_set_read_length(str, n, shrinkable);
-  io_enc_str(str, fptr);
-
-  if (n == 0)
-    return Qnil;
-
-  return str;
-}
-
-static VALUE Socket_accept(VALUE sock) {
-  rb_io_t *fptr;
-  int fd;
-  struct sockaddr addr;
-  socklen_t len = (socklen_t)sizeof addr;
-  VALUE read_watcher = Qnil;
-
-  GetOpenFile(sock, fptr);
-  rb_io_set_nonblock(fptr);
-
-  while (1) {
-    fd = accept(fptr->fd, &addr, &len);
-
-    if (fd < 0) {
-      int e = errno;
-      if (e == EWOULDBLOCK || e == EAGAIN) {
-        if (NIL_P(read_watcher))
-          read_watcher = Gyro_IO_auto_io(fptr->fd, EV_READ);
-        Gyro_IO_await(read_watcher);
-      }
-      else
-        rb_syserr_fail(e, strerror(e));
-        // rb_syserr_fail_path(e, fptr->pathv);
-    }
-    else {
-      VALUE connection = rb_obj_alloc(cTCPSocket);
-      rb_io_t *fp;
-      MakeOpenFile(connection, fp);
-      rb_update_max_fd(fd);
-      fp->fd = fd;
-      fp->mode = FMODE_READWRITE | FMODE_DUPLEX;
-      rb_io_ascii8bit_binmode(connection);
-      rb_io_set_nonblock(fp);
-      rb_io_synchronized(fp);
-      // if (rsock_do_not_reverse_lookup) {
-	    //   fp->mode |= FMODE_NOREVLOOKUP;
-      // }
-
-      return connection;
-    }
-  }
-}
-
-void Init_Socket() {
-  rb_require("socket");
-  VALUE cBasicSocket = rb_const_get(rb_cObject, rb_intern("BasicSocket"));
-
-  rb_define_method(cBasicSocket, "send", BasicSocket_send, -1);
-  rb_define_method(cBasicSocket, "recv", BasicSocket_recv, -1);
-
-  VALUE cSocket = rb_const_get(rb_cObject, rb_intern("Socket"));
-  
-  rb_define_method(cSocket, "accept", Socket_accept, 0);
-
-  cTCPSocket = rb_const_get(rb_cObject, rb_intern("TCPSocket"));
-}

data/ext/gyro/thread.c

@@ -1,308 +0,0 @@
-#include "gyro.h"
-
-static VALUE cQueue;
-
-static ID ID_create_event_selector;
-static ID ID_deactivate_all_watchers_post_fork;
-static ID ID_empty;
-static ID ID_fiber_ref_count;
-static ID ID_ivar_event_selector_proc;
-static ID ID_ivar_event_selector;
-static ID ID_ivar_join_wait_queue;
-static ID ID_ivar_main_fiber;
-static ID ID_ivar_result;
-static ID ID_ivar_terminated;
-static ID ID_pop;
-static ID ID_push;
-static ID ID_run_queue;
-static ID ID_runnable_next;
-static ID ID_stop;
-
-VALUE event_selector_factory_proc(RB_BLOCK_CALL_FUNC_ARGLIST(args, klass)) {
-  return rb_funcall(klass, ID_new, 1, rb_ary_entry(args, 0));
-}
-
-static VALUE Thread_event_selector_set_proc(VALUE self, VALUE proc) {
-  if (!rb_obj_is_proc(proc)) {
-    proc = rb_proc_new(event_selector_factory_proc, proc);
-  }
-  rb_ivar_set(self, ID_ivar_event_selector_proc, proc);
-  return self;
-}
-
-static VALUE Thread_create_event_selector(VALUE self, VALUE thread) {
-  VALUE selector_proc = rb_ivar_get(self, ID_ivar_event_selector_proc);
-  if (selector_proc == Qnil) {
-    rb_raise(rb_eRuntimeError, "No event_selector_proc defined");
-  }
-
-  return rb_funcall(selector_proc, ID_call, 1, thread);
-}
-
-static VALUE Thread_setup_fiber_scheduling(VALUE self) {
-  rb_ivar_set(self, ID_ivar_main_fiber, rb_fiber_current());
-  rb_ivar_set(self, ID_fiber_ref_count, INT2NUM(0));
-  VALUE queue = rb_ary_new();
-  rb_ivar_set(self, ID_run_queue, queue);
-  VALUE selector = rb_funcall(rb_cThread, ID_create_event_selector, 1, self);
-  rb_ivar_set(self, ID_ivar_event_selector, selector);
-
-  return self;
-}
-
-static VALUE Thread_stop_event_selector(VALUE self) {
-  VALUE selector = rb_ivar_get(self, ID_ivar_event_selector);
-  if (selector != Qnil) {
-    rb_funcall(selector, ID_stop, 0);
-  }
-  // Nullify the selector in order to prevent running the
-  // selector after the thread is done running.
-  rb_ivar_set(self, ID_ivar_event_selector, Qnil);
-
-  return self;
-}
-
-static VALUE Thread_deactivate_all_watchers_post_fork(VALUE self) {
-  VALUE selector = rb_ivar_get(self, ID_ivar_event_selector);
-  if (selector != Qnil) {
-    rb_funcall(selector, ID_deactivate_all_watchers_post_fork, 0);
-  }
-
-  return self;
-}
-
-VALUE Thread_ref(VALUE self) {
-  VALUE count = rb_ivar_get(self, ID_fiber_ref_count);
-  int new_count = NUM2INT(count) + 1;
-  rb_ivar_set(self, ID_fiber_ref_count, INT2NUM(new_count));
-  return self;
-}
-
-VALUE Thread_unref(VALUE self) {
-  VALUE count = rb_ivar_get(self, ID_fiber_ref_count);
-  int new_count = NUM2INT(count) - 1;
-  rb_ivar_set(self, ID_fiber_ref_count, INT2NUM(new_count));
-  return self;
-}
-
-int Thread_fiber_ref_count(VALUE self) {
-  VALUE count = rb_ivar_get(self, ID_fiber_ref_count);
-  return NUM2INT(count);
-}
-
-void Thread_fiber_reset_ref_count(VALUE self) {
-  rb_ivar_set(self, ID_fiber_ref_count, INT2NUM(0));
-}
-
-static VALUE SYM_scheduled_fibers;
-static VALUE SYM_pending_watchers;
-
-static VALUE Thread_fiber_scheduling_stats(VALUE self) {
-  VALUE stats = rb_hash_new();
-  VALUE queue = rb_ivar_get(self, ID_run_queue);
-  VALUE selector = rb_ivar_get(self, ID_ivar_event_selector);
-  
-  long scheduled_count = RARRAY_LEN(queue);
-  rb_hash_aset(stats, SYM_scheduled_fibers, INT2NUM(scheduled_count));
-
-  long pending_count = Gyro_Selector_pending_count(selector);
-  rb_hash_aset(stats, SYM_pending_watchers, INT2NUM(pending_count));
-
-  return stats;
-}
-
-VALUE Thread_schedule_fiber(VALUE self, VALUE fiber, VALUE value) {
-  if (rb_fiber_alive_p(fiber) != Qtrue) {
-    return self;
-  }
-
-  FIBER_TRACE(3, SYM_fiber_schedule, fiber, value);
-  // if fiber is already scheduled, just set the scheduled value, then return
-  rb_ivar_set(fiber, ID_runnable_value, value);
-  if (rb_ivar_get(fiber, ID_runnable) != Qnil) {
-    return self;
-  }
-
-  VALUE queue = rb_ivar_get(self, ID_run_queue);
-  rb_ary_push(queue, fiber);
-  rb_ivar_set(fiber, ID_runnable, Qtrue);
-
-  if (rb_thread_current() != self) {
-    // if the fiber scheduling is done across threads, we need to make sure the
-    // target thread is woken up in case it is in the middle of running its
-    // event selector. Otherwise it's gonna be stuck waiting for an event to
-    // happen, not knowing that it there's already a fiber ready to run in its
-    // run queue.
-    VALUE selector = rb_ivar_get(self, ID_ivar_event_selector);
-    if (selector != Qnil) {
-      Gyro_Selector_break_out_of_ev_loop(selector);
-    }
-  }
-  return self;
-}
-
-VALUE Thread_schedule_fiber_with_priority(VALUE self, VALUE fiber, VALUE value) {
-  if (rb_fiber_alive_p(fiber) != Qtrue) {
-    return self;
-  }
-  FIBER_TRACE(3, SYM_fiber_schedule, fiber, value);
-  rb_ivar_set(fiber, ID_runnable_value, value);
-
-  VALUE queue = rb_ivar_get(self, ID_run_queue);
-
-  // if fiber is already scheduled, remove it from the run queue
-  if (rb_ivar_get(fiber, ID_runnable) != Qnil) {
-    rb_ary_delete(queue, fiber);
-  } else {
-    rb_ivar_set(fiber, ID_runnable, Qtrue);
-  }
-
-  // the fiber is given priority by putting it at the front of the run queue
-  rb_ary_unshift(queue, fiber);
-
-  if (rb_thread_current() != self) {
-    // if the fiber scheduling is done across threads, we need to make sure the
-    // target thread is woken up in case it is in the middle of running its
-    // event selector. Otherwise it's gonna be stuck waiting for an event to
-    // happen, not knowing that it there's already a fiber ready to run in its
-    // run queue.
-    VALUE selector = rb_ivar_get(self, ID_ivar_event_selector);
-    if (selector != Qnil) {
-      Gyro_Selector_break_out_of_ev_loop(selector);
-    }
-  }
-  return self;
-}
-
-VALUE Thread_switch_fiber(VALUE self) {
-  VALUE current_fiber = rb_fiber_current();
-  if (__tracing_enabled__) {
-    if (rb_ivar_get(current_fiber, ID_ivar_running) != Qfalse) {
-      rb_funcall(rb_cObject, ID_fiber_trace, 2, SYM_fiber_switchpoint, current_fiber);
-    }
-  }
-  VALUE queue = rb_ivar_get(self, ID_run_queue);
-  VALUE selector = rb_ivar_get(self, ID_ivar_event_selector);
-
-  VALUE next_fiber;
-
-  while (1) {
-    next_fiber = rb_ary_shift(queue);
-    // if (break_flag != 0) {
-    //   return Qnil;
-    // }
-    int ref_count = Thread_fiber_ref_count(self);
-    if (next_fiber != Qnil) {
-      if (ref_count > 0) {
-        // this mechanism prevents event starvation in case the run queue never
-        // empties
-        Gyro_Selector_run_no_wait(selector, current_fiber, RARRAY_LEN(queue));
-      }
-      break;
-    }
-    if (ref_count == 0) {
-      break;
-    }
-
-    Gyro_Selector_run(selector, current_fiber);
-  }
-
-  if (next_fiber == Qnil) {
-    return Qnil;
-  }
-
-  // run next fiber
-  VALUE value = rb_ivar_get(next_fiber, ID_runnable_value);
-  FIBER_TRACE(3, SYM_fiber_run, next_fiber, value);
-
-  rb_ivar_set(next_fiber, ID_runnable, Qnil);
-  RB_GC_GUARD(next_fiber);
-  RB_GC_GUARD(value);
-  return rb_funcall(next_fiber, ID_transfer, 1, value);
-}
-
-VALUE Thread_reset_fiber_scheduling(VALUE self) {
-  VALUE queue = rb_ivar_get(self, ID_run_queue);
-  rb_ary_clear(queue);
-  Thread_fiber_reset_ref_count(self);
-  return self;
-}
-
-VALUE Thread_post_fork(VALUE self) {
-  VALUE selector = rb_ivar_get(self, ID_ivar_event_selector);
-  Gyro_Selector_post_fork(selector);
-
-  return self;
-}
-
-VALUE Gyro_switchpoint() {
-  VALUE thread = rb_thread_current();
-  Thread_ref(thread);
-  VALUE ret = Thread_switch_fiber(thread);
-  Thread_unref(thread);
-  RB_GC_GUARD(ret);
-  return ret;
-}
-
-VALUE Thread_current_event_selector() {
-  return rb_ivar_get(rb_thread_current(), ID_ivar_event_selector);
-}
-
-VALUE Thread_fiber_break_out_of_ev_loop(VALUE self, VALUE fiber, VALUE resume_obj) {
-  VALUE selector = rb_ivar_get(self, ID_ivar_event_selector);
-  if (fiber != Qnil) {
-    Thread_schedule_fiber_with_priority(self, fiber, resume_obj);
-  }
-
-  if (Gyro_Selector_break_out_of_ev_loop(selector) == Qnil) {
-    // we're not inside the ev_loop, so we just do a switchpoint
-    Thread_switch_fiber(self);
-  }
-
-  return self;
-}
-
-void Init_Thread() {
-  cQueue = rb_const_get(rb_cObject, rb_intern("Queue"));
-
-  rb_define_singleton_method(rb_cThread, "event_selector=", Thread_event_selector_set_proc, 1);
-  rb_define_singleton_method(rb_cThread, "create_event_selector", Thread_create_event_selector, 1);
-
-  rb_define_method(rb_cThread, "fiber_ref", Thread_ref, 0);
-  rb_define_method(rb_cThread, "fiber_unref", Thread_unref, 0);
-
-  rb_define_method(rb_cThread, "post_fork", Thread_post_fork, 0);
-
-  rb_define_method(rb_cThread, "setup_fiber_scheduling", Thread_setup_fiber_scheduling, 0);
-  rb_define_method(rb_cThread, "stop_event_selector", Thread_stop_event_selector, 0);
-  rb_define_method(rb_cThread, "deactivate_all_watchers_post_fork", Thread_deactivate_all_watchers_post_fork, 0);
-  rb_define_method(rb_cThread, "reset_fiber_scheduling", Thread_reset_fiber_scheduling, 0);
-  rb_define_method(rb_cThread, "fiber_scheduling_stats", Thread_fiber_scheduling_stats, 0);
-  rb_define_method(rb_cThread, "break_out_of_ev_loop", Thread_fiber_break_out_of_ev_loop, 2);
-
-  rb_define_method(rb_cThread, "schedule_fiber", Thread_schedule_fiber, 2);
-  rb_define_method(rb_cThread, "schedule_fiber_with_priority",
-    Thread_schedule_fiber_with_priority, 2);
-  rb_define_method(rb_cThread, "switch_fiber", Thread_switch_fiber, 0);
-
-  ID_create_event_selector    = rb_intern("create_event_selector");
-  ID_deactivate_all_watchers_post_fork = rb_intern("deactivate_all_watchers_post_fork");
-  ID_empty                    = rb_intern("empty?");
-  ID_fiber_ref_count          = rb_intern("fiber_ref_count");
-  ID_ivar_event_selector      = rb_intern("@event_selector");
-  ID_ivar_event_selector_proc = rb_intern("@event_selector_proc");
-  ID_ivar_join_wait_queue     = rb_intern("@join_wait_queue");
-  ID_ivar_main_fiber          = rb_intern("@main_fiber");
-  ID_ivar_result              = rb_intern("@result");
-  ID_ivar_terminated          = rb_intern("@terminated");
-  ID_pop                      = rb_intern("pop");
-  ID_push                     = rb_intern("push");
-  ID_run_queue                = rb_intern("run_queue");
-  ID_runnable_next            = rb_intern("runnable_next");
-  ID_stop                     = rb_intern("stop");
-
-  SYM_scheduled_fibers = ID2SYM(rb_intern("scheduled_fibers"));
-  SYM_pending_watchers = ID2SYM(rb_intern("pending_watchers"));
-  rb_global_variable(&SYM_scheduled_fibers);
-  rb_global_variable(&SYM_pending_watchers);
-}