polyphony 0.36 → 0.42

data/ext/gyro/gyro_ext.c

@@ -1,33 +0,0 @@
-#include "gyro.h"
-
-void Init_Fiber();
-void Init_Gyro();
-void Init_Gyro_Async();
-void Init_Gyro_Child();
-void Init_Gyro_IO();
-void Init_Gyro_Queue();
-void Init_Gyro_Selector();
-void Init_Gyro_Signal();
-void Init_Gyro_Timer();
-void Init_Socket();
-void Init_Thread();
-void Init_Tracing();
-
-void Init_gyro_ext() {
-  ev_set_allocator(xrealloc);
-
-  Init_Gyro();
-  Init_Gyro_Async();
-  Init_Gyro_Child();
-  Init_Gyro_IO();
-  Init_Gyro_Queue();
-  Init_Gyro_Selector();
-  Init_Gyro_Signal();
-  Init_Gyro_Timer();
-
-  Init_Fiber();
-  Init_Socket();
-  Init_Thread();
-
-  Init_Tracing();
-}

data/ext/gyro/io.c

@@ -1,474 +0,0 @@
-#include "gyro.h"
-
-#ifdef GetReadFile
-# define FPTR_TO_FD(fptr) (fileno(GetReadFile(fptr)))
-#else
-# define FPTR_TO_FD(fptr) fptr->fd
-#endif /* GetReadFile */
-
-struct Gyro_IO {
-  struct  ev_io ev_io;
-  struct  ev_loop *ev_loop;
-  int     active;
-  VALUE   self;
-  VALUE   fiber;
-  VALUE   selector;
-};
-
-VALUE cGyro_IO = Qnil;
-
-ID ID_ivar_read_watcher;
-ID ID_ivar_write_watcher;
-VALUE SYM_r;
-VALUE SYM_w;
-
-static void Gyro_IO_mark(void *ptr) {
-  struct Gyro_IO *io = ptr;
-  if (io->fiber != Qnil) {
-    rb_gc_mark(io->fiber);
-  }
-  if (io->selector != Qnil) {
-    rb_gc_mark(io->selector);
-  }
-}
-
-static void Gyro_IO_free(void *ptr) {
-  struct Gyro_IO *io = ptr;
-  if (io->active) {
-    ev_clear_pending(io->ev_loop, &io->ev_io);
-    ev_io_stop(io->ev_loop, &io->ev_io);
-  }
-  xfree(io);
-}
-
-static size_t Gyro_IO_size(const void *ptr) {
-  return sizeof(struct Gyro_IO);
-}
-
-static const rb_data_type_t Gyro_IO_type = {
-    "Gyro_IO",
-    {Gyro_IO_mark, Gyro_IO_free, Gyro_IO_size,},
-    0, 0, 0
-};
-
-static VALUE Gyro_IO_allocate(VALUE klass) {
-  struct Gyro_IO *io = ALLOC(struct Gyro_IO);
-
-  return TypedData_Wrap_Struct(klass, &Gyro_IO_type, io);
-}
-
-inline void Gyro_IO_activate(struct Gyro_IO *io) {
-  if (io->active) return;
-
-  io->active = 1;
-  io->fiber = rb_fiber_current();
-  io->selector = Thread_current_event_selector();
-  io->ev_loop = Gyro_Selector_ev_loop(io->selector);
-  Gyro_Selector_add_active_watcher(io->selector, io->self);
-  ev_io_start(io->ev_loop, &io->ev_io);
-}
-
-inline void Gyro_IO_deactivate(struct Gyro_IO *io) {
-  if (!io->active) return;
-
-  ev_io_stop(io->ev_loop, &io->ev_io);
-  Gyro_Selector_remove_active_watcher(io->selector, io->self);
-  io->active = 0;
-  io->ev_loop = 0;
-  io->selector = Qnil;
-  io->fiber = Qnil;
-}
-
-void Gyro_IO_callback(struct ev_loop *ev_loop, struct ev_io *ev_io, int revents) {
-  struct Gyro_IO *io = (struct Gyro_IO*)ev_io;
-
-  Fiber_make_runnable(io->fiber, Qnil);
-  Gyro_IO_deactivate(io);
-}
-
-static int Gyro_IO_symbol2event_mask(VALUE sym) {
-  ID sym_id;
-
-  if (NIL_P(sym)) {
-    return 0;
-  }
-
-  sym_id = SYM2ID(sym);
-
-  if(sym_id == ID_R) {
-    return EV_READ;
-  } else if(sym_id == ID_W) {
-    return EV_WRITE;
-  } else if(sym_id == ID_RW) {
-    return EV_READ | EV_WRITE;
-  } else {
-    rb_raise(rb_eArgError, "invalid interest type %s (must be :r, :w, or :rw)",
-      RSTRING_PTR(rb_funcall(sym, ID_inspect, 0)));
-  }
-}
-
-#define GetGyro_IO(obj, io) TypedData_Get_Struct((obj), struct Gyro_IO, &Gyro_IO_type, (io))
-
-static const char * S_IO = "IO";
-static const char * S_to_io = "to_io";
-
-static VALUE Gyro_IO_initialize(VALUE self, VALUE io_obj, VALUE event_mask) {
-  struct Gyro_IO *io;
-  rb_io_t *fptr;
-
-  GetGyro_IO(self, io);
-
-  io->self = self;
-  io->fiber = Qnil;
-  io->selector = Qnil;
-  io->active = 0;
-  io->ev_loop = 0;
-
-  int fd;
-  if (NIL_P(io_obj)) {
-    fd = 0;
-  }
-  else {
-    GetOpenFile(rb_convert_type(io_obj, T_FILE, S_IO, S_to_io), fptr);
-    fd = FPTR_TO_FD(fptr);
-  }
-  int events = Gyro_IO_symbol2event_mask(event_mask);
-  
-  ev_io_init(&io->ev_io, Gyro_IO_callback, fd, events);
-
-  return Qnil;
-}
-
-VALUE Gyro_IO_await(VALUE self) {
-  struct Gyro_IO *io;
-  GetGyro_IO(self, io);
-
-  Gyro_IO_activate(io);
-  VALUE ret = Gyro_switchpoint();
-  Gyro_IO_deactivate(io);
-
-  TEST_RESUME_EXCEPTION(ret);
-  RB_GC_GUARD(ret);
-  return ret;
-}
-
-VALUE Gyro_IO_auto_io(int fd, int events) {
-  VALUE watcher = Fiber_auto_io(rb_fiber_current());
-  struct Gyro_IO *io;
-  GetGyro_IO(watcher, io);
-
-  ev_io_set(&io->ev_io, fd, events);
-
-  RB_GC_GUARD(watcher);
-  return watcher;
-}
-
-
-//////////////////////////////////////////////////////////////////////
-//////////////////////////////////////////////////////////////////////
-// the following is copied verbatim from the Ruby source code (io.c)
-struct io_internal_read_struct {
-    int fd;
-    int nonblock;
-    void *buf;
-    size_t capa;
-};
-
-int io_setstrbuf(VALUE *str, long len) {
-  #ifdef _WIN32
-    len = (len + 1) & ~1L;	/* round up for wide char */
-  #endif
-  if (NIL_P(*str)) {
-    *str = rb_str_new(0, len);
-    return 1;
-  }
-  else {
-    VALUE s = StringValue(*str);
-    long clen = RSTRING_LEN(s);
-    if (clen >= len) {
-      rb_str_modify(s);
-      return 0;
-    }
-    len -= clen;
-  }
-  rb_str_modify_expand(*str, len);
-  return 0;
-}
-
-#define MAX_REALLOC_GAP 4096
-static void io_shrink_read_string(VALUE str, long n) {
-  if (rb_str_capacity(str) - n > MAX_REALLOC_GAP) {
-    rb_str_resize(str, n);
-  }
-}
-
-void io_set_read_length(VALUE str, long n, int shrinkable) {
-  if (RSTRING_LEN(str) != n) {
-    rb_str_modify(str);
-    rb_str_set_len(str, n);
-    if (shrinkable) io_shrink_read_string(str, n);
-  }
-}
-
-static rb_encoding* io_read_encoding(rb_io_t *fptr) {
-    if (fptr->encs.enc) {
-	return fptr->encs.enc;
-    }
-    return rb_default_external_encoding();
-}
-
-VALUE io_enc_str(VALUE str, rb_io_t *fptr) {
-    OBJ_TAINT(str);
-    rb_enc_associate(str, io_read_encoding(fptr));
-    return str;
-}
-
-//////////////////////////////////////////////////////////////////////
-//////////////////////////////////////////////////////////////////////
-
-static VALUE IO_read(int argc, VALUE *argv, VALUE io) {
-  VALUE underlying_io = rb_iv_get(io, "@io");
-  if (!NIL_P(underlying_io)) io = underlying_io;
-
-  long len = argc == 1 ? NUM2LONG(argv[0]) : (1 << 30);
-
-  rb_io_t *fptr;
-  long n;
-  int shrinkable;
-  VALUE read_watcher = Qnil;
-
-  if (len < 0) {
-    rb_raise(rb_eArgError, "negative length %ld given", len);
-  }
-
-  VALUE str = argc >= 2 ? argv[1] : Qnil;
-
-  shrinkable = io_setstrbuf(&str, len);
-  OBJ_TAINT(str);
-  GetOpenFile(io, fptr);
-  rb_io_check_byte_readable(fptr);
-  rb_io_set_nonblock(fptr);
-
-  if (len == 0)
-  	return str;
-  
-  char *buf = RSTRING_PTR(str);
-  long total = 0;
-
-  while (1) {
-    n = read(fptr->fd, buf, len);
-    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 if (n == 0)
-      break;
-    else {
-      total = total + n;
-      buf += n;
-      len -= n;
-      if (len == 0)
-        break;
-    }
-  }
-
-  if (total == 0)
-    return Qnil;
-
-  io_set_read_length(str, total, shrinkable);
-  io_enc_str(str, fptr);
-
-  return str;
-}
-
-#define READ_DATA_PENDING_COUNT(fptr) ((fptr)->rbuf.len)
-#define MEMMOVE(p1,p2,type,n) memmove((p1), (p2), sizeof(type)*(size_t)(n))
-
-static long
-read_buffered_data(char *ptr, long len, rb_io_t *fptr)
-{
-    int n;
-
-    n = READ_DATA_PENDING_COUNT(fptr);
-    if (n <= 0) return 0;
-    if (n > len) n = (int)len;
-    MEMMOVE(ptr, fptr->rbuf.ptr+fptr->rbuf.off, char, n);
-    fptr->rbuf.off += n;
-    fptr->rbuf.len -= n;
-    return n;
-}
-
-static VALUE IO_readpartial(int argc, VALUE *argv, VALUE io) {
-  VALUE underlying_io = rb_iv_get(io, "@io");
-  if (!NIL_P(underlying_io)) io = underlying_io;
-
-  long len = argc >= 1 ? NUM2LONG(argv[0]) : 8192;
-
-  rb_io_t *fptr;
-  long n;
-  int shrinkable;
-  VALUE read_watcher = Qnil;
-
-  if (len < 0) {
-    rb_raise(rb_eArgError, "negative length %ld given", len);
-  }
-
-  VALUE str = argc >= 2 ? argv[1] : Qnil;
-
-  shrinkable = io_setstrbuf(&str, len);
-  OBJ_TAINT(str);
-  GetOpenFile(io, fptr);
-  rb_io_set_nonblock(fptr);
-  rb_io_check_byte_readable(fptr);
-
-  if (len == 0)
-  	return str;
-
-  n = read_buffered_data(RSTRING_PTR(str), len, fptr);
-  if (n <= 0) {
-    while (1) {
-      n = read(fptr->fd, RSTRING_PTR(str), len);
-      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);
-
-  // ensure yielding to reactor if haven't yielded while reading
-  // if (read_watcher == Qnil) {
-  //   Gyro_snooze(Qnil);
-  // }
-
-  if (n == 0)
-    return Qnil;
-
-  return str;
-}
-
-static VALUE IO_write(int argc, VALUE *argv, VALUE io) {
-  VALUE underlying_io = rb_iv_get(io, "@io");
-  if (!NIL_P(underlying_io)) io = underlying_io;
-
-  long i;
-  long n;
-  long total = 0;
-  rb_io_t *fptr;
-
-  io = rb_io_get_write_io(io);
-  VALUE write_watcher = Qnil;
-
-  GetOpenFile(io, fptr);
-  rb_io_check_writable(fptr);
-  rb_io_set_nonblock(fptr);
-
-  for (i = 0; i < argc; i++) {
-    VALUE str = argv[i];
-    if (!RB_TYPE_P(str, T_STRING))
-	    str = rb_obj_as_string(str);
-    char *buf = RSTRING_PTR(str);
-    long len = RSTRING_LEN(str);
-    RB_GC_GUARD(str);
-    while (1) {
-      n = write(fptr->fd, buf, len);
-
-      if (n < 0) {
-        int e = errno;
-        if (e == EWOULDBLOCK || e == EAGAIN) {
-          if (NIL_P(write_watcher))
-            write_watcher = Gyro_IO_auto_io(fptr->fd, EV_WRITE);
-          Gyro_IO_await(write_watcher);
-        }
-        else {
-          rb_syserr_fail(e, strerror(e));
-          // rb_syserr_fail_path(e, fptr->pathv);
-        }
-      }
-      else {
-        total += n;
-        if (n < len) {
-          buf += n;
-          len -= n;
-        }
-        else break;
-      }
-    }
-  }
-
-  // ensure yielding to reactor if haven't yielded while writing
-  // if (write_watcher == Qnil) {
-  //   Gyro_snooze(Qnil);
-  // }
-
-  return LONG2FIX(total);
-}
-
-static VALUE IO_write_chevron(VALUE io, VALUE str) {
-  IO_write(1, &str, io);
-  return io;
-}
-
-VALUE IO_read_watcher(VALUE self) {
-  VALUE watcher = rb_ivar_get(self, ID_ivar_read_watcher);
-  if (watcher == Qnil) {
-    VALUE args[] = {self, SYM_r};
-    watcher = rb_class_new_instance(2, args, cGyro_IO);
-    rb_ivar_set(self, ID_ivar_read_watcher, watcher);
-  }
-  return watcher;
-}
-
-VALUE IO_write_watcher(VALUE self) {
-  VALUE watcher = rb_ivar_get(self, ID_ivar_write_watcher);
-  if (watcher == Qnil) {
-    VALUE args[] = {self, SYM_w};
-    watcher = rb_class_new_instance(2, args, cGyro_IO);
-    rb_ivar_set(self, ID_ivar_write_watcher, watcher);
-  }
-  return watcher;
-}
-
-void Init_Gyro_IO() {
-  cGyro_IO = rb_define_class_under(mGyro, "IO", rb_cData);
-  rb_define_alloc_func(cGyro_IO, Gyro_IO_allocate);
-
-  rb_define_method(cGyro_IO, "initialize", Gyro_IO_initialize, 2);
-  rb_define_method(cGyro_IO, "await", Gyro_IO_await, 0);
-
-  VALUE cIO = rb_const_get(rb_cObject, rb_intern("IO"));
-  // rb_define_method(cIO, "gets", IO_gets, -1);
-  rb_define_method(cIO, "read", IO_read, -1);
-  rb_define_method(cIO, "readpartial", IO_readpartial, -1);
-  rb_define_method(cIO, "write", IO_write, -1);
-  rb_define_method(cIO, "write_nonblock", IO_write, -1);
-  rb_define_method(cIO, "<<", IO_write_chevron, 1);
-  rb_define_method(cIO, "read_watcher", IO_read_watcher, 0);
-  rb_define_method(cIO, "write_watcher", IO_write_watcher, 0);
-
-  ID_ivar_read_watcher = rb_intern("@read_watcher");
-  ID_ivar_write_watcher = rb_intern("@write_watcher");
-  SYM_r = ID2SYM(rb_intern("r"));
-  SYM_w = ID2SYM(rb_intern("w"));
-
-  rb_global_variable(&SYM_r);
-  rb_global_variable(&SYM_w);
-}