This commit is contained in:
@@ -0,0 +1,316 @@
|
||||
#--
|
||||
#
|
||||
# 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 'rubyeventmachine'
|
||||
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)
|
||||
if field.getType.toString == 'int'
|
||||
field.setInt(self, value)
|
||||
else
|
||||
field.set(self, value)
|
||||
end
|
||||
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.
|
||||
# @private
|
||||
TimerFired = 100
|
||||
# @private
|
||||
ConnectionData = 101
|
||||
# @private
|
||||
ConnectionUnbound = 102
|
||||
# @private
|
||||
ConnectionAccepted = 103
|
||||
# @private
|
||||
ConnectionCompleted = 104
|
||||
# @private
|
||||
LoopbreakSignalled = 105
|
||||
# @private
|
||||
ConnectionNotifyReadable = 106
|
||||
# @private
|
||||
ConnectionNotifyWritable = 107
|
||||
# @private
|
||||
SslHandshakeCompleted = 108
|
||||
# @private
|
||||
SslVerify = 109
|
||||
|
||||
# @private
|
||||
EM_PROTO_SSLv2 = 2
|
||||
# @private
|
||||
EM_PROTO_SSLv3 = 4
|
||||
# @private
|
||||
EM_PROTO_TLSv1 = 8
|
||||
# @private
|
||||
EM_PROTO_TLSv1_1 = 16
|
||||
# @private
|
||||
EM_PROTO_TLSv1_2 = 32
|
||||
|
||||
# Exceptions that are defined in rubymain.cpp
|
||||
class ConnectionError < RuntimeError; end
|
||||
class ConnectionNotBound < RuntimeError; end
|
||||
class UnknownTimerFired < RuntimeError; end
|
||||
class Unsupported < RuntimeError; end
|
||||
|
||||
# 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
|
||||
nil
|
||||
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
|
||||
rescue java.lang.NullPointerException
|
||||
0
|
||||
end
|
||||
def self.send_datagram sig, data, length, address, port
|
||||
@em.sendDatagram sig, data.to_java_bytes, 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.set_pending_connect_timeout sig, val
|
||||
end
|
||||
def self.set_heartbeat_interval val
|
||||
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.get_sockname sig
|
||||
if sockName = @em.getSockName(sig)
|
||||
Socket.pack_sockaddr_in(*sockName)
|
||||
end
|
||||
end
|
||||
# @private
|
||||
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? Integer
|
||||
end
|
||||
|
||||
if fileno == 0
|
||||
raise "can't open STDIN as selectable in Java =("
|
||||
elsif fileno.is_a? Integer
|
||||
# 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
|
||||
|
||||
def self.pause_connection(sig)
|
||||
@em.pauseConnection(sig)
|
||||
end
|
||||
def self.resume_connection(sig)
|
||||
@em.resumeConnection(sig)
|
||||
end
|
||||
def self.connection_paused?(sig)
|
||||
@em.isConnectionPaused(sig)
|
||||
end
|
||||
def self._get_outbound_data_size(sig)
|
||||
@em.getOutboundDataSize(sig)
|
||||
end
|
||||
|
||||
|
||||
def self.set_tls_parms(sig, params)
|
||||
end
|
||||
def self.start_tls(sig)
|
||||
end
|
||||
def self.send_file_data(sig, filename)
|
||||
end
|
||||
|
||||
class Connection
|
||||
def associate_callback_target sig
|
||||
# No-op for the time being
|
||||
end
|
||||
def get_outbound_data_size
|
||||
EM._get_outbound_data_size @signature
|
||||
end
|
||||
end
|
||||
end
|
||||
|
||||
Reference in New Issue
Block a user