polyphony 0.43 → 0.43.5

data/ext/polyphony/socket.c

@@ -1,213 +0,0 @@
-#include "polyphony.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 agent = 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(agent))
-      agent = rb_ivar_get(rb_thread_current(), ID_ivar_agent);
-    LibevAgent_wait_io(agent, sock, Qtrue);
-  }
-  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 agent = 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(agent))
-          agent = rb_ivar_get(rb_thread_current(), ID_ivar_agent);
-        LibevAgent_wait_io(agent, sock, Qnil);
-      }
-      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 agent = 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(agent))
-          agent = rb_ivar_get(rb_thread_current(), ID_ivar_agent);
-        LibevAgent_wait_io(agent, sock, Qnil);
-      }
-      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() {
-  VALUE cBasicSocket;
-  VALUE cSocket;
-  
-  rb_require("socket");
-  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);
-
-  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"));
-}