MattHulse-eventmachine 0.0.1

data/lib/evma.rb

@@ -0,0 +1,32 @@
+#--
+#
+# Author:: Francis Cianfrocca (gmail: blackhedd)
+# Homepage::  http://rubyeventmachine.com
+# Date:: 8 Apr 2006
+# 
+# See EventMachine and EventMachine::Connection for documentation and
+# usage examples.
+#
+#----------------------------------------------------------------------------
+#
+# Copyright (C) 2006-07 by Francis Cianfrocca. All Rights Reserved.
+# Gmail: blackhedd
+# 
+# This program is free software; you can redistribute it and/or modify
+# it under the terms of either: 1) the GNU General Public License
+# as published by the Free Software Foundation; either version 2 of the
+# License, or (at your option) any later version; or 2) Ruby's License.
+# 
+# See the file COPYING for complete licensing information.
+#
+#-------------------------------------------------------------------
+#
+
+
+require 'rubyeventmachine'
+require 'evma/reactor'
+require 'evma/callback'
+require 'evma/protocol'
+require 'evma/factory'
+require 'evma/container'
+

data/lib/evma/callback.rb

@@ -0,0 +1,32 @@
+# $Id$
+#
+# Author:: Francis Cianfrocca (gmail: blackhedd)
+# Homepage::  http://rubyeventmachine.com
+# Date:: 19 May 2006
+# 
+# See EventMachine and EventMachine::Connection for documentation and
+# usage examples.
+#
+#----------------------------------------------------------------------------
+#
+# Copyright (C) 2006-07 by Francis Cianfrocca. All Rights Reserved.
+# Gmail: blackhedd
+# 
+# This program is free software; you can redistribute it and/or modify
+# it under the terms of either: 1) the GNU General Public License
+# as published by the Free Software Foundation; either version 2 of the
+# License, or (at your option) any later version; or 2) Ruby's License.
+# 
+# See the file COPYING for complete licensing information.
+#
+#---------------------------------------------------------------------------
+#
+
+
+module EventMachine
+
+def self.event_callback target, opcode, data
+  Evma::Container.callback target, opcode, data
+end
+
+end # module EventMachine

data/lib/evma/container.rb

@@ -0,0 +1,75 @@
+# $Id$
+#
+# Author:: Francis Cianfrocca (gmail: blackhedd)
+# Homepage::  http://rubyeventmachine.com
+# Date:: 19 May 2006
+# 
+# See EventMachine and EventMachine::Connection for documentation and
+# usage examples.
+#
+#----------------------------------------------------------------------------
+#
+# Copyright (C) 2006-07 by Francis Cianfrocca. All Rights Reserved.
+# Gmail: blackhedd
+# 
+# This program is free software; you can redistribute it and/or modify
+# it under the terms of either: 1) the GNU General Public License
+# as published by the Free Software Foundation; either version 2 of the
+# License, or (at your option) any later version; or 2) Ruby's License.
+# 
+# See the file COPYING for complete licensing information.
+#
+#---------------------------------------------------------------------------
+#
+
+
+
+require 'singleton'
+
+module Evma
+
+class ContainerHasObject < Exception; end
+class UnsupportedCallback < Exception; end
+class UnknownTarget < Exception; end
+
+class Container
+  include Singleton
+
+  def initialize
+    @objects = {}
+  end
+
+  def self.store obj
+    instance.store obj
+  end
+
+  def self.callback target, opcode, data
+    instance.callback target, opcode, data
+  end
+
+  def store obj
+    sig = obj.signature
+    raise ContainerHasObject.new(sig) if @objects.has_key?(sig)
+    @objects[sig] = obj
+  end
+
+  def callback target, opcode, data
+    case opcode
+    when 101 # received data
+      obj = @objects[target] or raise UnknownTarget.new( target )
+      obj.receive_data data
+    when 102 # unbind
+      obj = @objects[target] or raise UnknownTarget.new( target )
+      obj.unbind
+      @objects.delete obj.signature
+    when 103 # accept
+      obj = @objects[target] or raise UnknownTarget.new( target )
+      obj.accept data
+    else
+      raise UnsupportedCallback.new( opcode.to_s )
+    end
+  end
+
+end # class Container
+end # module Evma
+

data/lib/evma/factory.rb

@@ -0,0 +1,77 @@
+# $Id$
+#
+# Author:: Francis Cianfrocca (gmail: blackhedd)
+# Homepage::  http://rubyeventmachine.com
+# Date:: 19 May 2006
+# 
+# See EventMachine and EventMachine::Connection for documentation and
+# usage examples.
+#
+#----------------------------------------------------------------------------
+#
+# Copyright (C) 2006-07 by Francis Cianfrocca. All Rights Reserved.
+# Gmail: blackhedd
+# 
+# This program is free software; you can redistribute it and/or modify
+# it under the terms of either: 1) the GNU General Public License
+# as published by the Free Software Foundation; either version 2 of the
+# License, or (at your option) any later version; or 2) Ruby's License.
+# 
+# See the file COPYING for complete licensing information.
+#
+#---------------------------------------------------------------------------
+#
+
+
+module Evma
+class ProtocolFactory < Protocol
+
+  #--
+  # default implementation raises an exception.
+  # we expect subclasses to override this.
+  # we can't do anything reasonable here because
+  def accept new_object
+    # don't bother calling Evma::Reactor.instance, since only Reactor can call accept
+    Evma::Container.store Evma::Protocol.new( new_object )
+    EventMachine.close_connection new_object, false
+  end
+
+
+end # class ProtocolFactory
+end # module Evma
+
+######################################
+
+module Evma
+class TcpSocket
+
+  def self.connect server, port, protocol_handler = Evma::Protocol
+    Evma::Reactor.instance # ensure initialization
+    sig = EventMachine.connect_server server, port
+    Evma::Container.store protocol_handler.new( sig )
+  end
+
+end
+end # module Evma
+
+######################################
+
+module Evma
+class TcpServerFactory < Evma::ProtocolFactory
+
+  def initialize server, port, protocol_handler = Evma::Protocol
+    Evma::Reactor.instance # ensure initialization
+    sig = EventMachine.start_tcp_server server, port
+    super sig
+    @protocol_handler = protocol_handler
+    Evma::Container.store self
+  end
+
+  def accept new_obj
+    # don't bother calling Evma::Reactor.instance, since only Reactor can call accept
+    Evma::Container.store @protocol_handler.new( new_obj )
+  end
+
+end
+end # module Evma
+

data/lib/evma/protocol.rb

@@ -0,0 +1,87 @@
+# $Id$
+#
+# Author:: Francis Cianfrocca (gmail: blackhedd)
+# Homepage::  http://rubyeventmachine.com
+# Date:: 19 May 2006
+# 
+# See EventMachine and EventMachine::Connection for documentation and
+# usage examples.
+#
+#----------------------------------------------------------------------------
+#
+# Copyright (C) 2006-07 by Francis Cianfrocca. All Rights Reserved.
+# Gmail: blackhedd
+# 
+# This program is free software; you can redistribute it and/or modify
+# it under the terms of either: 1) the GNU General Public License
+# as published by the Free Software Foundation; either version 2 of the
+# License, or (at your option) any later version; or 2) Ruby's License.
+# 
+# See the file COPYING for complete licensing information.
+#
+#---------------------------------------------------------------------------
+#
+
+
+module Evma
+class Protocol
+
+  attr_reader :signature
+
+  def initialize sig
+    @signature = sig
+  end
+
+  def unbind
+  end
+
+  def close
+    Evma::Reactor.instance # ensure initialized
+    EventMachine.close_connection signature, false
+  end
+
+  def close_after_writing
+    Evma::Reactor.instance # ensure initialized
+    EventMachine.close_connection signature, true
+  end
+
+end # class Protocol
+end # module Evma
+
+
+###########################################
+
+module Evma
+class StreamProtocol < Protocol
+
+  def initialize sig
+    super
+  end
+
+  def send_data data
+    Evma::Reactor.instance # ensure initialized
+    EventMachine.send_data signature, data, data.length
+  end
+
+end # class Protocol
+end # module Evma
+
+
+###########################################
+
+module Evma
+class DatagramProtocol < Protocol
+
+  def initialize sig
+    super
+  end
+
+  def send_message data
+    Evma::Reactor.instance # ensure initialized
+    raise "unimplemented"
+  end
+
+end # class Protocol
+end # module Evma
+
+

data/lib/evma/reactor.rb

@@ -0,0 +1,48 @@
+# $Id$
+#
+# Author:: Francis Cianfrocca (gmail: blackhedd)
+# Homepage::  http://rubyeventmachine.com
+# Date:: 18 May 2006
+# 
+# See EventMachine and EventMachine::Connection for documentation and
+# usage examples.
+#
+#----------------------------------------------------------------------------
+#
+# Copyright (C) 2006-07 by Francis Cianfrocca. All Rights Reserved.
+# Gmail: blackhedd
+# 
+# This program is free software; you can redistribute it and/or modify
+# it under the terms of either: 1) the GNU General Public License
+# as published by the Free Software Foundation; either version 2 of the
+# License, or (at your option) any later version; or 2) Ruby's License.
+# 
+# See the file COPYING for complete licensing information.
+#
+#---------------------------------------------------------------------------
+#
+
+
+
+require 'singleton'
+
+module Evma
+class Reactor
+  include Singleton
+
+  #--
+  def initialize
+    EventMachine.initialize_event_machine
+  end
+
+  #--
+  #
+  def run
+    EventMachine.run_machine
+  end
+
+end # class Reactor
+end # module Evma
+
+
+

data/lib/jeventmachine.rb

@@ -0,0 +1,246 @@
+#--
+#
+# Author:: Francis Cianfrocca (gmail: blackhedd)
+# Homepage::  http://rubyeventmachine.com
+# Date:: 8 Apr 2006
+# 
+# See EventMachine and EventMachine::Connection for documentation and
+# usage examples.
+#
+#----------------------------------------------------------------------------
+#
+# Copyright (C) 2006-07 by Francis Cianfrocca. All Rights Reserved.
+# Gmail: blackhedd
+# 
+# This program is free software; you can redistribute it and/or modify
+# it under the terms of either: 1) the GNU General Public License
+# as published by the Free Software Foundation; either version 2 of the
+# License, or (at your option) any later version; or 2) Ruby's License.
+# 
+# See the file COPYING for complete licensing information.
+#
+#---------------------------------------------------------------------------
+#
+# 
+
+# This module provides "glue" for the Java version of the EventMachine reactor core.
+# For C++ EventMachines, the analogous functionality is found in ext/rubymain.cpp,
+# which is a garden-variety Ruby-extension glue module.
+
+require 'java'
+require 'em_reactor'
+require 'socket'
+
+java_import java.io.FileDescriptor
+java_import java.nio.channels.SocketChannel
+java_import java.lang.reflect.Field
+
+module JavaFields
+  def set_field(key, value)
+    field = getClass.getDeclaredField(key)
+    field.setAccessible(true)
+    field.set(self, value)
+  end
+
+  def get_field(key)
+    field = getClass.getDeclaredField(key)
+    field.setAccessible(true)
+    field.get(self)
+  end
+end
+
+FileDescriptor.send :include, JavaFields
+SocketChannel.send :include, JavaFields
+
+module EventMachine
+  # TODO: These event numbers are defined in way too many places.
+  # DRY them up.
+  TimerFired = 100
+  ConnectionData = 101
+  ConnectionUnbound = 102
+  ConnectionAccepted = 103
+  ConnectionCompleted = 104
+  LoopbreakSignalled = 105
+  ConnectionNotifyReadable = 106
+  ConnectionNotifyWritable = 107
+
+  # This thunk class used to be called EM, but that caused conflicts with
+  # the alias "EM" for module EventMachine. (FC, 20Jun08)
+  class JEM < com.rubyeventmachine.EmReactor
+    def eventCallback a1, a2, a3, a4
+      s = String.from_java_bytes(a3.array[a3.position...a3.limit]) if a3
+      EventMachine::event_callback a1, a2, s || a4
+    end
+  end
+  # class Connection < com.rubyeventmachine.Connection
+  #   def associate_callback_target sig
+  #     # No-op for the time being.
+  #   end
+  # end
+  def self.initialize_event_machine
+    @em = JEM.new
+  end
+  def self.release_machine
+    @em = nil
+  end
+  def self.add_oneshot_timer interval
+    @em.installOneshotTimer interval
+  end
+  def self.run_machine
+    @em.run
+  end
+  def self.stop
+    @em.stop
+  end
+  def self.start_tcp_server server, port
+    @em.startTcpServer server, port
+  end
+  def self.stop_tcp_server sig
+    @em.stopTcpServer sig
+  end
+  def self.start_unix_server filename
+    # TEMPORARILY unsupported until someone figures out how to do it.
+    raise "unsupported on this platform"
+  end
+  def self.send_data sig, data, length
+    @em.sendData sig, data.to_java_bytes
+  end
+  def self.send_datagram sig, data, length, address, port
+    @em.sendDatagram sig, data, length, address, port
+  end
+  def self.connect_server server, port
+    bind_connect_server nil, nil, server, port
+  end
+  def self.bind_connect_server bind_addr, bind_port, server, port
+    @em.connectTcpServer bind_addr, bind_port.to_i, server, port
+  end
+  def self.close_connection sig, after_writing
+    @em.closeConnection sig, after_writing
+  end
+  def self.set_comm_inactivity_timeout sig, interval
+    @em.setCommInactivityTimeout sig, interval
+  end
+  def self.start_tls sig
+    @em.startTls sig
+  end
+  def self.ssl?
+    false
+  end
+  def self.signal_loopbreak
+    @em.signalLoopbreak
+  end
+  def self.set_timer_quantum q
+    @em.setTimerQuantum q
+  end
+  def self.epoll
+    # Epoll is a no-op for Java.
+    # The latest Java versions run epoll when possible in NIO.
+  end
+  def self.epoll= val
+  end
+  def self.kqueue
+  end
+  def self.kqueue= val
+  end
+  def self.epoll?
+    false
+  end
+  def self.kqueue?
+    false
+  end
+  def self.set_rlimit_nofile n_descriptors
+    # Currently a no-op for Java.
+  end
+  def self.open_udp_socket server, port
+    @em.openUdpSocket server, port
+  end
+  def self.invoke_popen cmd
+    # TEMPORARILY unsupported until someone figures out how to do it.
+    raise "unsupported on this platform"
+  end
+  def self.read_keyboard
+    # TEMPORARILY unsupported until someone figures out how to do it.
+    raise "temporarily unsupported on this platform"
+  end
+  def self.set_max_timer_count num
+    # harmless no-op in Java. There's no built-in timer limit.
+    @max_timer_count = num
+  end
+  def self.get_max_timer_count
+    # harmless no-op in Java. There's no built-in timer limit.
+    @max_timer_count || 100_000
+  end
+  def self.library_type
+    :java
+  end
+  def self.get_peername sig
+    if peer = @em.getPeerName(sig)
+      Socket.pack_sockaddr_in *peer
+    end
+  end
+  def self.attach_fd fileno, watch_mode
+    # 3Aug09: We could pass in the actual SocketChannel, but then it would be modified (set as non-blocking), and
+    # we would need some logic to make sure detach_fd below didn't clobber it. For now, we just always make a new
+    # SocketChannel for the underlying file descriptor
+    # if fileno.java_kind_of? SocketChannel
+    #   ch = fileno
+    #   ch.configureBlocking(false)
+    #   fileno = nil
+    # elsif fileno.java_kind_of? java.nio.channels.Channel
+
+    if fileno.java_kind_of? java.nio.channels.Channel
+      field = fileno.getClass.getDeclaredField('fdVal')
+      field.setAccessible(true)
+      fileno = field.get(fileno)
+    else
+      raise ArgumentError, 'attach_fd requires Java Channel or POSIX fileno' unless fileno.is_a? Fixnum
+    end
+
+    if fileno == 0
+      raise "can't open STDIN as selectable in Java =("
+    elsif fileno.is_a? Fixnum
+      # 8Aug09: The following code is specific to the sun jvm's SocketChannelImpl. Is there a cross-platform
+      # way of implementing this? If so, also remember to update EventableSocketChannel#close and #cleanup
+      fd = FileDescriptor.new
+      fd.set_field 'fd', fileno
+
+      ch = SocketChannel.open
+      ch.configureBlocking(false)
+      ch.kill
+      ch.set_field 'fd', fd
+      ch.set_field 'fdVal', fileno
+      ch.set_field 'state', ch.get_field('ST_CONNECTED')
+    end
+
+    @em.attachChannel(ch,watch_mode)
+  end
+  def self.detach_fd sig
+    if ch = @em.detachChannel(sig)
+      ch.get_field 'fdVal'
+    end
+  end
+
+  def self.set_notify_readable sig, mode
+    @em.setNotifyReadable(sig, mode)
+  end
+  def self.set_notify_writable sig, mode
+    @em.setNotifyWritable(sig, mode)
+  end
+
+  def self.is_notify_readable sig
+    @em.isNotifyReadable(sig)
+  end
+  def self.is_notify_writable sig
+    @em.isNotifyWritable(sig)
+  end
+  def self.get_connection_count
+    @em.getConnectionCount
+  end
+
+  class Connection
+    def associate_callback_target sig
+      # No-op for the time being
+    end
+  end
+end
+