socket_helper 0.1.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 822a9459f5cb9c1ecc2d1095b4fec9a99a6c4c20
+  data.tar.gz: db2fe8c8ac41a291312f41b0699111f52c2d134e
+SHA512:
+  metadata.gz: 5aca37c5fec6a0e5d0a261b5152f256fb058b17f0fa91bd2b4591a1f7ec1fd6e94b48e78ccf16e947a8af73462008a3477f1bd1d160e7d31ce36f3424ebb9af2
+  data.tar.gz: 69bcb96529be2ec4f6d40c05a7c5a04896b7182d36bb3af9f5d7933a695635b722b31342e4b7397e4293ae85185559fa518778a457d1737eee1d938481a26255

data/Rakefile

@@ -0,0 +1,7 @@
+# rake-compiler
+require 'rake' unless defined? Rake
+
+gem 'rake-compiler', '>= 0.7.5'
+require "rake/extensiontask"
+
+Rake::ExtensionTask.new('socket_helper')

data/ext/socket_helper/extconf.rb

@@ -0,0 +1,7 @@
+require 'mkmf'
+
+$CFLAGS << " -Wall "
+
+extension_name = 'socket_helper'
+dir_config(extension_name)
+create_makefile(extension_name)

data/ext/socket_helper/socket_helper.c

@@ -0,0 +1,88 @@
+#include <ruby.h>
+#include <ruby/io.h>
+#include <sys/socket.h>
+
+extern VALUE rb_thread_io_blocking_region(rb_blocking_function_t *func, void *data1, int fd);
+#define BLOCKING_REGION_FD(func, arg) (long)rb_thread_io_blocking_region((func), (arg), (arg)->fd)
+
+struct recv_args {
+	VALUE buffer;
+	int fd, flags;
+	long pos;
+};
+
+static VALUE
+recv_loop(void *data)
+{
+	struct recv_args *arg = data;
+	ssize_t ret;
+
+	ret = recv(arg->fd,
+		RSTRING_PTR(arg->buffer) + arg->pos,
+		RSTRING_LEN(arg->buffer) - arg->pos,
+		arg->flags
+	);
+
+	if (ret < 0)
+		return (VALUE)ret;
+
+	arg->pos += ret;
+	return (VALUE)ret;
+}
+
+static VALUE
+rb_socket_read_in_full(int argc, VALUE *argv, VALUE self)
+{
+	rb_io_t *fptr;
+	VALUE rb_sock, rb_buffer, rb_klass;
+	struct recv_args arg;
+	long buflen;
+
+	VALUE len, flg;
+	rb_scan_args(argc, argv, "21", &rb_sock, &len, &flg);
+
+	if (flg == Qnil)
+		arg.flags = 0;
+	else
+		arg.flags = NUM2INT(flg);
+
+	buflen = NUM2INT(len);
+
+	GetOpenFile(rb_sock, fptr);
+	if (rb_io_read_pending(fptr)) {
+		rb_raise(rb_eIOError, "recv for buffered IO");
+	}
+
+	arg.fd = fptr->fd;
+	arg.pos = 0;
+	arg.buffer = rb_buffer = rb_tainted_str_new(0, buflen);
+
+	rb_klass = RBASIC(rb_buffer)->klass;
+	rb_obj_hide(rb_buffer);
+
+	while (arg.pos < buflen) {
+		long slen;
+
+		rb_io_check_closed(fptr);
+		slen = BLOCKING_REGION_FD(recv_loop, &arg);
+
+		if (!slen)
+			break;
+
+		if (slen < 0 && !rb_io_wait_readable(fptr->fd))
+			rb_sys_fail("recv(2)");
+	}
+
+	rb_obj_reveal(rb_buffer, rb_klass);
+	if (arg.pos < buflen)
+		rb_raise(rb_eRuntimeError, "connection closed");
+
+	rb_obj_taint(rb_buffer);
+	return rb_buffer;
+}
+
+void Init_socket_helper(void)
+{
+	VALUE rb_mSocketHelper = rb_define_module("SocketHelper");
+	rb_define_module_function(rb_mSocketHelper, "read_in_full", rb_socket_read_in_full, -1);
+}

data/socket_helper.gemspec

@@ -0,0 +1,17 @@
+Gem::Specification.new do |s|
+  s.name = 'socket_helper'
+  s.version = '0.1.0'
+  s.authors = ["Vicent Marti"]
+  s.date = Time.now.utc.strftime("%Y-%m-%d")
+  s.email = %q{tanoku@gmail.com}
+  s.extensions = ["ext/socket_helper/extconf.rb"]
+  s.files = `git ls-files`.split("\n")
+  s.homepage = %q{http://github.com/vmg/socket_helper}
+  s.rdoc_options = ["--charset=UTF-8"]
+  s.require_paths = ["lib"]
+  s.rubygems_version = %q{1.4.2}
+  s.summary = %q{A thing for sockets}
+
+  # tests
+  s.add_development_dependency 'rake-compiler', ">= 0.7.5"
+end

metadata

@@ -0,0 +1,62 @@
+--- !ruby/object:Gem::Specification
+name: socket_helper
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- Vicent Marti
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2014-09-18 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: rake-compiler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 0.7.5
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 0.7.5
+description: 
+email: tanoku@gmail.com
+executables: []
+extensions:
+- ext/socket_helper/extconf.rb
+extra_rdoc_files: []
+files:
+- Rakefile
+- ext/socket_helper/extconf.rb
+- ext/socket_helper/socket_helper.c
+- socket_helper.gemspec
+homepage: http://github.com/vmg/socket_helper
+licenses: []
+metadata: {}
+post_install_message: 
+rdoc_options:
+- "--charset=UTF-8"
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.2.2
+signing_key: 
+specification_version: 4
+summary: A thing for sockets
+test_files: []