eventmachine 0.12.6-java

data/java/src/com/rubyeventmachine/EmReactor.java

@@ -0,0 +1,408 @@
+/**
+ * $Id$
+ * 
+ * Author:: Francis Cianfrocca (gmail: blackhedd)
+ * Homepage::  http://rubyeventmachine.com
+ * Date:: 15 Jul 2007
+ * 
+ * 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.
+ *
+ *---------------------------------------------------------------------------
+ *
+ * 
+ */
+
+package com.rubyeventmachine;
+
+import java.io.*;
+import java.nio.channels.*;
+import java.util.*;
+import java.nio.*;
+import java.net.*;
+import java.util.concurrent.atomic.*;
+import java.security.*;
+
+public class EmReactor {
+
+	
+	public final int EM_TIMER_FIRED = 100;
+	public final int EM_CONNECTION_READ = 101;
+	public final int EM_CONNECTION_UNBOUND = 102;
+	public final int EM_CONNECTION_ACCEPTED = 103;
+	public final int EM_CONNECTION_COMPLETED = 104;
+	public final int EM_LOOPBREAK_SIGNAL = 105;
+	
+	private Selector mySelector;
+	private TreeMap<Long, String> Timers;
+	private TreeMap<String, EventableChannel> Connections;
+	private TreeMap<String, ServerSocketChannel> Acceptors;
+
+	private boolean bRunReactor;
+	private long BindingIndex;
+	private ByteBuffer EmptyByteBuffer;
+	private AtomicBoolean loopBreaker;
+	private ByteBuffer myReadBuffer;
+	private int timerQuantum;
+	
+	public EmReactor() {
+		Timers = new TreeMap<Long, String>();
+		Connections = new TreeMap<String, EventableChannel>();
+		Acceptors = new TreeMap<String, ServerSocketChannel>();
+		
+		BindingIndex = 100000;
+		EmptyByteBuffer = ByteBuffer.allocate(0);
+		loopBreaker = new AtomicBoolean();
+		loopBreaker.set(false);
+		myReadBuffer = ByteBuffer.allocate(32*1024); // don't use a direct buffer. Ruby doesn't seem to like them.
+		timerQuantum = 98;
+	}
+	
+	/**
+	 * Intended to be overridden in languages (like Ruby) that can't handle ByteBuffer. This is a stub.
+	 * Obsolete now that I figured out how to make Ruby deal with ByteBuffers.
+	 * @param sig
+	 * @param eventType
+	 * @param data
+	 */
+	/*
+ 	public void stringEventCallback (String sig, int eventType, String data) {
+		System.out.println ("Default event callback: " + sig + " " + eventType + " " + data);
+	}
+	*/
+ 	
+ 	/**
+ 	 * This is a no-op stub, intended to be overridden in user code.
+ 	 * @param sig
+ 	 * @param eventType
+ 	 * @param data
+ 	 */
+	public void eventCallback (String sig, int eventType, ByteBuffer data) {
+		System.out.println ("Default callback: "+sig+" "+eventType+" "+data);
+		//stringEventCallback (sig, eventType, new String (data.array(), data.position(), data.limit()));
+		
+	}
+	
+	public void run() throws IOException {
+ 		mySelector = Selector.open();
+ 		bRunReactor = true;
+ 		
+ 		//int n = 0;
+ 		for (;;) {
+ 			//System.out.println ("loop "+ (n++));
+ 			if (!bRunReactor) break;
+ 			runLoopbreaks();
+ 			if (!bRunReactor) break;
+ 			runTimers();
+ 			if (!bRunReactor) break;
+ 			mySelector.select(timerQuantum);
+ 			 	
+ 			Iterator<SelectionKey> it = mySelector.selectedKeys().iterator();
+ 			while (it.hasNext()) {
+ 				SelectionKey k = it.next();
+ 				it.remove();
+
+ 				try {
+ 					if (k.isAcceptable()) {
+ 						ServerSocketChannel ss = (ServerSocketChannel) k.channel();
+ 						SocketChannel sn;
+ 						while ((sn = ss.accept()) != null) {
+ 							sn.configureBlocking(false);
+ 							String b = createBinding();
+ 							EventableSocketChannel ec = new EventableSocketChannel (sn, b, mySelector);
+ 							Connections.put(b, ec);
+ 							eventCallback ((String)k.attachment(), EM_CONNECTION_ACCEPTED, ByteBuffer.wrap(b.getBytes()));
+ 						}
+ 					}
+
+ 					if (k.isReadable()) {
+ 						EventableChannel ec = (EventableChannel)k.attachment();
+ 						myReadBuffer.clear();
+ 						ec.readInboundData (myReadBuffer);
+ 						myReadBuffer.flip();
+ 						String b = ec.getBinding();
+ 						if (myReadBuffer.limit() > 0) {
+ 							eventCallback (b, EM_CONNECTION_READ, myReadBuffer);
+ 						}
+ 						else {
+ 							eventCallback (b, EM_CONNECTION_UNBOUND, EmptyByteBuffer);
+ 							Connections.remove(b);
+ 							k.channel().close();							
+ 						}
+ 						/*
+						System.out.println ("READABLE");
+ 						SocketChannel sn = (SocketChannel) k.channel();
+ 						//ByteBuffer bb = ByteBuffer.allocate(16 * 1024);
+ 						// Obviously not thread-safe, since we're using the same buffer for every connection.
+ 						// This should minimize the production of garbage, though.
+ 						// TODO, we need somehow to make a call to the EventableChannel, so we can pass the
+ 						// inbound data through an SSLEngine. Hope that won't break the strategy of using one
+ 						// global read-buffer.
+ 						myReadBuffer.clear();
+ 						int r = sn.read(myReadBuffer);
+ 						if (r > 0) {
+ 							myReadBuffer.flip();
+ 							//bb = ((EventableChannel)k.attachment()).dispatchInboundData (bb);
+ 							eventCallback (((EventableChannel)k.attachment()).getBinding(), EM_CONNECTION_READ, myReadBuffer);
+ 						}
+ 						else {
+ 							// TODO. Figure out if a socket that selects readable can ever return 0 bytes
+ 							// without it being indicative of an error condition. If Java is like C, the answer is no.
+ 							String b = ((EventableChannel)k.attachment()).getBinding();
+ 							eventCallback (b, EM_CONNECTION_UNBOUND, EmptyByteBuffer);
+ 							Connections.remove(b);
+ 							sn.close();
+ 						}
+ 						*/
+ 					}
+ 				
+
+ 					if (k.isWritable()) {
+ 						EventableChannel ec = (EventableChannel)k.attachment();
+ 						if (!ec.writeOutboundData()) {
+ 							eventCallback (ec.getBinding(), EM_CONNECTION_UNBOUND, EmptyByteBuffer);
+ 							Connections.remove (ec.getBinding());
+ 							k.channel().close();
+ 						}
+ 					}
+ 					
+ 					if (k.isConnectable()) {
+ 						EventableSocketChannel ec = (EventableSocketChannel)k.attachment();
+ 						if (ec.finishConnecting()) {
+ 							eventCallback (ec.getBinding(), EM_CONNECTION_COMPLETED, EmptyByteBuffer);
+ 						}
+ 						else {
+ 							Connections.remove (ec.getBinding());
+ 							k.channel().close();
+ 							eventCallback (ec.getBinding(), EM_CONNECTION_UNBOUND, EmptyByteBuffer);
+ 						}
+ 					}
+ 				}
+ 				catch (CancelledKeyException e) {
+ 					// No-op. We can come here if a read-handler closes a socket before we fall through
+ 					// to call isWritable.
+ 				}
+ 				
+  			}
+   		}
+ 		
+ 		close();
+	}
+	
+	void close() throws IOException {
+		mySelector.close();
+		mySelector = null;
+
+		// run down open connections and sockets.
+		Iterator<ServerSocketChannel> i = Acceptors.values().iterator();
+		while (i.hasNext()) {
+			i.next().close();
+		}
+		
+		Iterator<EventableChannel> i2 = Connections.values().iterator();
+		while (i2.hasNext())
+			i2.next().close();
+	}
+	
+	void runLoopbreaks() {
+		if (loopBreaker.getAndSet(false)) {
+			eventCallback ("", EM_LOOPBREAK_SIGNAL, EmptyByteBuffer);
+		}
+	}
+	
+	public void stop() {
+		bRunReactor = false;
+		signalLoopbreak();
+	}
+	
+	void runTimers() {
+		long now = new Date().getTime();
+		while (!Timers.isEmpty()) {
+			long k = Timers.firstKey();
+			//System.out.println (k - now);
+			if (k > now)
+				break;
+			String s = Timers.remove(k);
+			eventCallback ("", EM_TIMER_FIRED, ByteBuffer.wrap(s.getBytes()));
+		}
+	}
+	
+	public String installOneshotTimer (int milliseconds) {
+		BindingIndex++;
+		String s = createBinding();
+		Timers.put(new Date().getTime() + milliseconds, s);
+		return s;
+	}
+	
+	public String startTcpServer (SocketAddress sa) throws EmReactorException {
+		try {
+			ServerSocketChannel server = ServerSocketChannel.open();
+			server.configureBlocking(false);
+			server.socket().bind (sa);
+			String s = createBinding();
+			Acceptors.put(s, server);
+			server.register(mySelector, SelectionKey.OP_ACCEPT, s);
+			return s;
+		} catch (IOException e) {
+			// TODO, should parameterize this exception better.
+			throw new EmReactorException ("unable to open socket acceptor");
+		}
+	}
+	
+	public String startTcpServer (String address, int port) throws EmReactorException {
+		return startTcpServer (new InetSocketAddress (address, port));
+		/*
+		ServerSocketChannel server = ServerSocketChannel.open();
+		server.configureBlocking(false);
+		server.socket().bind(new java.net.InetSocketAddress(address, port));
+		String s = createBinding();
+		Acceptors.put(s, server);
+		server.register(mySelector, SelectionKey.OP_ACCEPT, s);
+		return s;
+		*/
+	}
+
+	public void stopTcpServer (String signature) throws IOException {
+		ServerSocketChannel server = Acceptors.remove(signature);
+		if (server != null)
+			server.close();
+		else
+			throw new RuntimeException ("failed to close unknown acceptor");
+	}
+	
+
+	public String openUdpSocket (String address, int port) throws IOException {
+		return openUdpSocket (new InetSocketAddress (address, port));
+	}
+	/**
+	 * 
+	 * @param address
+	 * @param port
+	 * @return
+	 * @throws IOException
+	 */
+	public String openUdpSocket (InetSocketAddress address) throws IOException {
+		// TODO, don't throw an exception out of here.
+		DatagramChannel dg = DatagramChannel.open();
+		dg.configureBlocking(false);
+		dg.socket().bind(address);
+		String b = createBinding();
+		EventableChannel ec = new EventableDatagramChannel (dg, b, mySelector);
+		dg.register(mySelector, SelectionKey.OP_READ, ec);
+		Connections.put(b, ec);
+		return b;
+	}
+	
+	public void sendData (String sig, ByteBuffer bb) throws IOException {
+		(Connections.get(sig)).scheduleOutboundData( bb );
+	}
+	public void sendData (String sig, byte[] data) throws IOException {
+		sendData (sig, ByteBuffer.wrap(data));
+		//(Connections.get(sig)).scheduleOutboundData( ByteBuffer.wrap(data.getBytes()));
+	}
+	public void setCommInactivityTimeout (String sig, long mills) {
+		(Connections.get(sig)).setCommInactivityTimeout (mills);
+	}
+	
+	/**
+	 * 
+	 * @param sig
+	 * @param data
+	 * @param length
+	 * @param recipAddress
+	 * @param recipPort
+	 */
+	public void sendDatagram (String sig, String data, int length, String recipAddress, int recipPort) {
+		sendDatagram (sig, ByteBuffer.wrap(data.getBytes()), recipAddress, recipPort);
+	}
+	
+	/**
+	 * 
+	 * @param sig
+	 * @param bb
+	 * @param recipAddress
+	 * @param recipPort
+	 */
+	public void sendDatagram (String sig, ByteBuffer bb, String recipAddress, int recipPort) {
+		(Connections.get(sig)).scheduleOutboundDatagram( bb, recipAddress, recipPort);
+	}
+
+	
+	/**
+	 * 
+	 * @param address
+	 * @param port
+	 * @return
+	 * @throws ClosedChannelException
+	 */
+	public String connectTcpServer (String address, int port) throws ClosedChannelException {
+		String b = createBinding();
+				
+		try {
+			SocketChannel sc = SocketChannel.open();
+			sc.configureBlocking(false);
+			EventableSocketChannel ec = new EventableSocketChannel (sc, b, mySelector);
+
+			if (sc.connect (new InetSocketAddress (address, port))) {
+				// Connection returned immediately. Can happen with localhost connections.
+				// WARNING, this code is untested due to lack of available test conditions.
+				// Ought to be be able to come here from a localhost connection, but that
+				// doesn't happen on Linux. (Maybe on FreeBSD?)
+				// The reason for not handling this until we can test it is that we
+				// really need to return from this function WITHOUT triggering any EM events.
+				// That's because until the user code has seen the signature we generated here,
+				// it won't be able to properly dispatch them. The C++ EM deals with this
+				// by setting pending mode as a flag in ALL eventable descriptors and making
+				// the descriptor select for writable. Then, it can send UNBOUND and
+				// CONNECTION_COMPLETED on the next pass through the loop, because writable will
+				// fire.
+				throw new RuntimeException ("immediate-connect unimplemented");
+ 			}
+			else {
+				Connections.put (b, ec);
+				ec.setConnectPending();
+			}
+		} catch (IOException e) {
+			// Can theoretically come here if a connect failure can be determined immediately.
+			// I don't know how to make that happen for testing purposes.
+			throw new RuntimeException ("immediate-connect unimplemented"); 
+		}
+		return b;
+	}
+
+	public void closeConnection (String sig, boolean afterWriting) throws ClosedChannelException {
+		Connections.get(sig).scheduleClose (afterWriting);
+	}
+	
+	String createBinding() {
+		return new String ("BND_" + (++BindingIndex));
+	}
+	
+	public void signalLoopbreak() {
+		loopBreaker.set(true);
+		mySelector.wakeup();
+	}
+	
+	public void startTls (String sig) throws NoSuchAlgorithmException, KeyManagementException {
+		Connections.get(sig).startTls();
+	}
+	
+	public void setTimerQuantum (int mills) {
+		if (mills < 5 || mills > 2500)
+			throw new RuntimeException ("attempt to set invalid timer-quantum value: "+mills);
+		timerQuantum = mills;
+	}
+}

data/java/src/com/rubyeventmachine/EmReactorException.java

@@ -0,0 +1,40 @@
+/**
+ * $Id$
+ * 
+ * Author:: Francis Cianfrocca (gmail: blackhedd)
+ * Homepage::  http://rubyeventmachine.com
+ * Date:: 15 Jul 2007
+ * 
+ * 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.
+ *
+ *---------------------------------------------------------------------------
+ *
+ * 
+ */
+
+package com.rubyeventmachine;
+
+/**
+ * @author francis
+ *
+ */
+public class EmReactorException extends Exception {
+	static final long serialVersionUID = 0;
+	public EmReactorException (String msg) {
+		super (msg);
+	}
+}

data/java/src/com/rubyeventmachine/EventableChannel.java

@@ -0,0 +1,57 @@
+/**
+ * $Id$
+ * 
+ * Author:: Francis Cianfrocca (gmail: blackhedd)
+ * Homepage::  http://rubyeventmachine.com
+ * Date:: 15 Jul 2007
+ * 
+ * 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.
+ *
+ *---------------------------------------------------------------------------
+ *
+ * 
+ */
+
+
+package com.rubyeventmachine;
+
+import java.nio.ByteBuffer;
+
+public interface EventableChannel {
+	
+	public void scheduleOutboundData (ByteBuffer bb);
+	
+	public void scheduleOutboundDatagram (ByteBuffer bb, String recipAddress, int recipPort);
+	
+	public void scheduleClose (boolean afterWriting);
+	
+	public void startTls();
+	
+	public String getBinding();
+	
+	public void readInboundData (ByteBuffer dst);
+	
+	/**
+	 * This is called by the reactor after it finishes running.
+	 * The idea is to free network resources.
+	 */
+	public void close();
+	
+	public boolean writeOutboundData();
+
+	public void setCommInactivityTimeout (long seconds);
+}