uvrb 0.1.0

data/lib/uv/net.rb

@@ -0,0 +1,31 @@
+require 'socket'
+
+module UV
+  module Net
+    private
+
+    def get_sockaddr_and_len
+      sockaddr = FFI::MemoryPointer.new(UV::Sockaddr)
+      len = FFI::MemoryPointer.new(:int)
+      len.put_int(0, UV::Sockaddr.size)
+      [sockaddr, len]
+    end
+
+    def get_ip_and_port(sockaddr, len=nil)
+      if sockaddr[:sa_family] == Socket::Constants::AF_INET6
+        len ||= Socket::Constants::INET6_ADDRSTRLEN
+        sockaddr_in6 = UV::SockaddrIn6.new(sockaddr.pointer)
+        ip_ptr = FFI::MemoryPointer.new(:char, len)
+        UV.ip6_name(sockaddr_in6, ip_ptr, len)
+        port = UV.ntohs(sockaddr_in6[:sin6_port])
+      else
+        len ||= Socket::Constants::INET_ADDRSTRLEN
+        sockaddr_in = UV::SockaddrIn.new(sockaddr.pointer)
+        ip_ptr = FFI::MemoryPointer.new(:char, len)
+        UV.ip4_name(sockaddr_in, ip_ptr, len)
+        port = UV.ntohs(sockaddr_in[:sin_port])
+      end
+      [ip_ptr.read_string, port]
+    end
+  end
+end

data/lib/uv/pipe.rb

@@ -0,0 +1,47 @@
+module UV
+  class Pipe
+    include Stream
+
+    def open(fileno)
+      assert_type(Integer, fileno, "io#fileno must return an integer file descriptor")
+
+      check_result! UV.pipe_open(handle, fileno)
+
+      self
+    end
+
+    def bind(name)
+      assert_type(String, name, "name must be a String")
+
+      check_result! UV.pipe_bind(handle, name)
+
+      self
+    end
+
+    def connect(name, &block)
+      assert_block(block)
+      assert_arity(1, block)
+      assert_type(String, name, "name must be a String")
+
+      @connect_block = block
+
+      UV.pipe_connect(UV.create_request(:uv_connect), handle, name, callback(:on_connect))
+
+      self
+    end
+
+    def pending_instances=(count)
+      assert_type(Integer, count, "count must be an Integer")
+
+      UV.pipe_pending_instances(handle, count)
+
+      self
+    end
+
+    private
+    def on_connect(req, status)
+      UV.free(req)
+      @connect_block.call(check_result(status))
+    end
+  end
+end

data/lib/uv/prepare.rb

@@ -0,0 +1,27 @@
+module UV
+  class Prepare
+    include Handle
+
+    def start(&block)
+      assert_block(block)
+      assert_arity(1, block)
+
+      @prepare_block = block
+
+      check_result! UV.prepare_start(handle, callback(:on_prepare))
+
+      self
+    end
+
+    def stop
+      check_result! UV.prepare_stop(handle)
+
+      self
+    end
+
+    private
+    def on_prepare(handle, status)
+      @prepare_block.call(check_result(status))
+    end
+  end
+end

data/lib/uv/resource.rb

@@ -0,0 +1,20 @@
+module UV
+  module Resource
+    def check_result(rc)
+      @loop.last_error if rc == -1
+    end
+
+    def check_result!(rc)
+      e = check_result(rc)
+      raise e if e
+    end
+
+    def to_ptr
+      @pointer
+    end
+
+    protected
+
+    attr_reader :loop
+  end
+end

data/lib/uv/stream.rb

@@ -0,0 +1,105 @@
+module UV
+  module Stream
+    include Handle
+
+    def listen(backlog, &block)
+      assert_block(block)
+      assert_arity(1, block)
+      assert_type(Integer, backlog, "backlog must be an Integer")
+
+      @listen_block = block
+
+      check_result! UV.listen(handle, Integer(backlog), callback(:on_listen))
+
+      self
+    end
+
+    def accept
+      client = loop.send(handle_name)
+
+      check_result! UV.accept(handle, client.handle)
+
+      client
+    end
+
+    def start_read(&block)
+      assert_block(block)
+      assert_arity(2, block)
+
+      @read_block = block
+
+      check_result! UV.read_start(handle, callback(:on_allocate), callback(:on_read))
+
+      self
+    end
+
+    def stop_read
+      check_result! UV.read_stop(handle)
+
+      self
+    end
+
+    def write(data, &block)
+      assert_block(block)
+      assert_arity(1, block)
+      assert_type(String, data, "data must be a String")
+
+      @write_block = block
+      size         = data.respond_to?(:bytesize) ? data.bytesize : data.size
+      buffer       = UV.buf_init(FFI::MemoryPointer.from_string(data), size)
+
+      check_result! UV.write(UV.create_request(:uv_write), handle, buffer, 1, callback(:on_write))
+
+      self
+    end
+
+    def shutdown(&block)
+      assert_block(block)
+      assert_arity(1, block)
+
+      @shutdown_block = block
+
+      check_result! UV.shutdown(UV.create_request(:uv_shutdown), handle, callback(:on_shutdown))
+
+      self
+    end
+
+    def readable?
+      UV.is_readable(handle) > 0
+    end
+
+    def writable?
+      UV.is_writable(handle) > 0
+    end
+
+    private
+
+    def on_listen(server, status)
+      @listen_block.call(check_result(status))
+    end
+
+    def on_allocate(client, suggested_size)
+      UV.buf_init(UV.malloc(suggested_size), suggested_size)
+    end
+
+    def on_read(handle, nread, buf)
+      e = check_result(nread)
+      base = buf[:base]
+      unless e
+        data = base.read_string(nread)
+      end
+      UV.free(base)
+      @read_block.call(e, data)
+    end
+
+    def on_write(req, status)
+      UV.free(req)
+      @write_block.call(check_result(status))
+    end
+
+    def on_shutdown(req, status)
+      UV.free(req)
+      @shutdown_block.call(check_result(status))
+    end
+  end
+end

data/lib/uv/tasks.rb

@@ -0,0 +1,27 @@
+module FFI::Platform
+  def self.ia32?
+    ARCH == "i386"
+  end
+
+  def self.x64?
+    ARCH == "x86_64"
+  end
+end
+
+file 'ext/libuv/build' do
+  system "git", "submodule", "update", "--init"
+end
+
+file 'ext/libuv/build/gyp' => 'ext/libuv/build' do
+  system "svn", "export", "-r1214", "http://gyp.googlecode.com/svn/trunk", "ext/libuv/build/gyp"
+end
+
+CLEAN.include('ext/libuv/build/gyp')
+
+if FFI::Platform.windows?
+  require 'uv/tasks/win'
+elsif FFI::Platform.mac?
+  require 'uv/tasks/mac'
+else # UNIX
+  require 'uv/tasks/unix'
+end

data/lib/uv/tasks/mac.rb

@@ -0,0 +1,23 @@
+file 'ext/libuv/uv.xcodeproj' => 'ext/libuv/build/gyp' do
+  target_arch = 'ia32'if FFI::Platform.ia32?
+  target_arch = 'x64' if FFI::Platform.x64?
+
+  abort "Don't know how to build on #{FFI::Platform::ARCH} (yet)" unless target_arch
+
+  Dir.chdir("ext/libuv") do |path|
+    system "./gyp_uv -f xcode -Dtarget_arch=#{target_arch}"
+  end
+end
+
+file "ext/libuv/build/Release/libuv.#{FFI::Platform::LIBSUFFIX}" => 'ext/libuv/uv.xcodeproj' do
+  Dir.chdir("ext/libuv") do |path|
+    system 'xcodebuild -project uv.xcodeproj -configuration Release -target All'
+  end
+end
+
+file "ext/libuv.#{FFI::Platform::LIBSUFFIX}" => "ext/libuv/build/Release/libuv.#{FFI::Platform::LIBSUFFIX}" do
+  File.symlink("libuv/build/Release/libuv.#{FFI::Platform::LIBSUFFIX}", "ext/libuv.#{FFI::Platform::LIBSUFFIX}")
+end
+
+CLEAN.include('ext/libuv/uv.xcodeproj')
+CLOBBER.include("ext/libuv/build/Release/libuv.#{FFI::Platform::LIBSUFFIX}")

data/lib/uv/tasks/unix.rb

@@ -0,0 +1,23 @@
+file 'ext/libuv/out' => 'ext/libuv/build/gyp' do
+  target_arch = 'ia32'if FFI::Platform.ia32?
+  target_arch = 'x64' if FFI::Platform.x64?
+
+  abort "Don't know how to build on #{FFI::Platform::ARCH} (yet)" unless target_arch
+
+  Dir.chdir("ext/libuv") do |path|
+    system "./gyp_uv -f make -Dtarget_arch=#{target_arch}"
+  end
+end
+
+file "ext/libuv/out/Release/lib.target/libuv.#{FFI::Platform::LIBSUFFIX}" => 'ext/libuv/out' do
+  Dir.chdir("ext/libuv") do |path|
+    system 'make -C out BUILDTYPE=Release'
+  end
+end
+
+file "ext/libuv.#{FFI::Platform::LIBSUFFIX}" => "ext/libuv/out/Release/lib.target/libuv.#{FFI::Platform::LIBSUFFIX}" do
+  File.symlink("libuv/out/Release/lib.target/libuv.#{FFI::Platform::LIBSUFFIX}", "ext/libuv.#{FFI::Platform::LIBSUFFIX}")
+end
+
+CLEAN.include('ext/libuv/out')
+CLOBBER.include("ext/libuv/out/Release/lib.target/libuv.#{FFI::Platform::LIBSUFFIX}")

data/lib/uv/tasks/win.rb

@@ -0,0 +1,2 @@
+abort "can't build on windows using this rake task (yet)"
+configure_command = 'vcbuild.bat'

data/lib/uv/tcp.rb

@@ -0,0 +1,162 @@
+require 'ipaddr'
+
+module UV
+  class TCP
+    include Stream, Net
+
+    def bind(ip, port)
+      assert_type(String, ip, "ip must be a String")
+      assert_type(Integer, port, "port must be an Integer")
+
+      @socket = create_socket(IPAddr.new(ip), port)
+
+      @socket.bind
+
+      self
+    end
+
+    def connect(ip, port, &block)
+      assert_block(block)
+      assert_arity(1, block)
+      assert_type(String, ip, "ip must be a String")
+      assert_type(Integer, port, "port must be an Integer")
+
+      @connect_block = block
+      @socket        = create_socket(IPAddr.new(ip), port)
+
+      @socket.connect(callback(:on_connect))
+
+      self
+    end
+
+    def sockname
+      sockaddr, len = get_sockaddr_and_len
+      check_result! UV.tcp_getsockname(handle, sockaddr, len)
+      get_ip_and_port(UV::Sockaddr.new(sockaddr), len.get_int(0))
+    end
+
+    def peername
+      sockaddr, len = get_sockaddr_and_len
+      check_result! UV.tcp_getpeername(handle, sockaddr, len)
+      get_ip_and_port(UV::Sockaddr.new(sockaddr), len.get_int(0))
+    end
+
+    def enable_nodelay
+      check_result! UV.tcp_nodelay(handle, 1)
+
+      self
+    end
+
+    def disable_nodelay
+      check_result! UV.tcp_nodelay(handle, 0)
+
+      self
+    end
+
+    def enable_keepalive(delay)
+      assert_type(Integer, delay, "delay must be an Integer")
+
+      check_result! UV.tcp_keepalive(handle, 1, delay)
+
+      self
+    end
+
+    def disable_keepalive
+      check_result! UV.tcp_keepalive(handle, 0, 0)
+
+      self
+    end
+
+    def enable_simultaneous_accepts
+      check_result! UV.tcp_simultaneous_accepts(handle, 1)
+
+      self
+    end
+
+    def disable_simultaneous_accepts
+      check_result! UV.tcp_simultaneous_accepts(handle, 0)
+
+      self
+    end
+
+    private
+    def create_socket(ip, port)
+      if ip.ipv4?
+        Socket4.new(@loop, handle, ip.to_s, port)
+      else
+        Socket6.new(@loop, handle, ip.to_s, port)
+      end
+    end
+
+    def on_connect(req, status)
+      UV.free(req)
+      @connect_block.call(check_result(status))
+    end
+
+    module SocketMethods
+      include Resource
+
+      def initialize(loop, tcp, ip, port)
+        @loop, @tcp, @sockaddr = loop, tcp, ip_addr(ip, port)
+      end
+
+      def bind
+        check_result! tcp_bind
+      end
+
+      def connect(callback)
+        check_result! tcp_connect(callback)
+      end
+
+      private
+
+      def connect_req
+        UV.create_request(:uv_connect)
+      end
+    end
+
+    class Socket4
+      include SocketMethods
+
+      private
+      def ip_addr(ip, port)
+        UV.ip4_addr(ip, port)
+      end
+
+      def tcp_bind
+        UV.tcp_bind(@tcp, @sockaddr)
+      end
+
+      def tcp_connect(callback)
+        UV.tcp_connect(
+          connect_req,
+          @tcp,
+          @sockaddr,
+          callback
+        )
+      end
+    end
+
+    class Socket6
+      include SocketMethods
+
+      private
+      def ip_addr(ip, port)
+        UV.ip6_addr(ip, port)
+      end
+
+      def tcp_bind
+        UV.tcp_bind6(@tcp, @sockaddr)
+      end
+
+      def tcp_connect(callback)
+        UV.tcp_connect6(
+          connect_req,
+          @tcp,
+          @sockaddr,
+          callback
+        )
+      end
+    end
+  end
+end