Add support for get_peername to jruby reactor

zadams · Jun 16, 2009 · d12f6e5 · d12f6e5
1 parent 17ed668
commit d12f6e5
Show file tree

Hide file tree

Showing 5 changed files with 27 additions and 0 deletions.
diff --git a/java/src/com/rubyeventmachine/EmReactor.java b/java/src/com/rubyeventmachine/EmReactor.java
@@ -436,4 +436,8 @@ public void setTimerQuantum (int mills) {
 			throw new RuntimeException ("attempt to set invalid timer-quantum value: "+mills);
 		timerQuantum = mills;
 	}
+
+	public Object[] getPeerName (String sig) {
+		return Connections.get(sig).getPeerName();
+	}
 }
diff --git a/java/src/com/rubyeventmachine/EventableChannel.java b/java/src/com/rubyeventmachine/EventableChannel.java
@@ -54,4 +54,6 @@ public interface EventableChannel {
 	public boolean writeOutboundData();
 
 	public void setCommInactivityTimeout (long seconds);
+
+	public Object[] getPeerName();
 }
diff --git a/java/src/com/rubyeventmachine/EventableDatagramChannel.java b/java/src/com/rubyeventmachine/EventableDatagramChannel.java
@@ -168,4 +168,13 @@ public void setCommInactivityTimeout (long seconds) {
 		// TODO
 		System.out.println ("DATAGRAM: SET COMM INACTIVITY UNIMPLEMENTED " + seconds);
 	}
+
+	public Object[] getPeerName () {
+		if (returnAddress != null) {
+			InetSocketAddress inetAddr = (InetSocketAddress) returnAddress;
+			return new Object[]{ inetAddr.getPort(), inetAddr.getHostName() };
+		} else {
+			return null;
+		}
+	}
 }
diff --git a/java/src/com/rubyeventmachine/EventableSocketChannel.java b/java/src/com/rubyeventmachine/EventableSocketChannel.java
@@ -40,6 +40,7 @@
 import java.nio.*;
 import java.util.*;
 import java.io.*;
+import java.net.Socket;
 import javax.net.ssl.*;
 import javax.net.ssl.SSLEngineResult.*;
 
@@ -241,4 +242,9 @@ public void setCommInactivityTimeout (long seconds) {
 		// TODO
 		System.out.println ("SOCKET: SET COMM INACTIVITY UNIMPLEMENTED " + seconds);
 	}
+
+	public Object[] getPeerName () {
+		Socket sock = channel.socket();
+		return new Object[]{ sock.getPort(), sock.getInetAddress().getHostAddress() };
+	}
 }
diff --git a/lib/jeventmachine.rb b/lib/jeventmachine.rb
@@ -29,6 +29,7 @@
 
 require 'java'
 require 'em_reactor'
+require 'socket'
 
 module EventMachine
   # TODO: These event numbers are defined in way too many places.
@@ -132,6 +133,11 @@ def self.set_max_timer_count num
   def self.library_type
     :java
   end
+  def self.get_peername sig
+    if peer = @em.getPeerName(sig)
+      Socket.pack_sockaddr_in *peer
+    end
+  end
 
   class Connection
     def associate_callback_target sig