libc-eventmachine 0.12.5.42

data/java/src/com/rubyeventmachine/EventableSocketChannel.java

@@ -0,0 +1,244 @@
+/**
+ * $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
+ *
+ */
+
+import java.nio.channels.*;
+import java.nio.*;
+import java.util.*;
+import java.io.*;
+import javax.net.ssl.*;
+import javax.net.ssl.SSLEngineResult.*;
+
+import java.security.*;
+
+public class EventableSocketChannel implements EventableChannel {
+	
+	// TODO, must refactor this to permit channels that aren't sockets.
+	SocketChannel channel;
+	String binding;
+	Selector selector;
+	LinkedList<ByteBuffer> outboundQ;
+	boolean bCloseScheduled;
+	boolean bConnectPending;
+	
+	SSLEngine sslEngine;
+	
+	
+	SSLContext sslContext;
+
+
+	public EventableSocketChannel (SocketChannel sc, String _binding, Selector sel) throws ClosedChannelException {
+		channel = sc;
+		binding = _binding;
+		selector = sel;
+		bCloseScheduled = false;
+		bConnectPending = false;
+		outboundQ = new LinkedList<ByteBuffer>();
+		
+		sc.register(selector, SelectionKey.OP_READ, this);
+	}
+	
+	public String getBinding() {
+		return binding;
+	}
+	
+	/**
+	 * Terminate with extreme prejudice. Don't assume there will be another pass through
+	 * the reactor core.
+	 */
+	public void close() {
+		try {
+			channel.close();
+		} catch (IOException e) {
+		}
+	}
+	
+	public void scheduleOutboundData (ByteBuffer bb) {
+		try {
+			if ((!bCloseScheduled) && (bb.remaining() > 0)) {
+				if (sslEngine != null) {
+					ByteBuffer b = ByteBuffer.allocate(32*1024); // TODO, preallocate this buffer.
+					sslEngine.wrap(bb, b);
+					b.flip();
+					outboundQ.addLast(b);
+				}
+				else {
+					outboundQ.addLast(bb);
+				}
+				channel.register(selector, SelectionKey.OP_WRITE | SelectionKey.OP_READ | (bConnectPending ? SelectionKey.OP_CONNECT : 0), this);
+			}
+		} catch (ClosedChannelException e) {
+			throw new RuntimeException ("no outbound data");			
+		} catch (SSLException e) {
+			throw new RuntimeException ("no outbound data");
+		}
+	}
+	
+	public void scheduleOutboundDatagram (ByteBuffer bb, String recipAddress, int recipPort) {
+		throw new RuntimeException ("datagram sends not supported on this channel");
+	}
+	
+	/**
+	 * Called by the reactor when we have selected readable.
+	 */
+	public void readInboundData (ByteBuffer bb) {
+		try {
+			channel.read(bb);
+		} catch (IOException e) {
+			throw new RuntimeException ("i/o error");
+		}
+	}
+	/**
+	 * Called by the reactor when we have selected writable.
+	 * Return false to indicate an error that should cause the connection to close.
+	 * We can get here with an empty outbound buffer if bCloseScheduled is true.
+	 * TODO, VERY IMPORTANT: we're here because we selected writable, but it's always
+	 * possible to become unwritable between the poll and when we get here. The way
+	 * this code is written, we're depending on a nonblocking write NOT TO CONSUME
+	 * the whole outbound buffer in this case, rather than firing an exception.
+	 * We should somehow verify that this is indeed Java's defined behavior.
+	 * Also TODO, see if we can use gather I/O rather than one write at a time.
+	 * Ought to be a big performance enhancer.
+	 * @return
+	 */
+	public boolean writeOutboundData(){
+		while (!outboundQ.isEmpty()) {
+			ByteBuffer b = outboundQ.getFirst();
+			try {
+				if (b.remaining() > 0)
+					channel.write(b);
+			}
+			catch (IOException e) {
+				return false;
+			}
+
+			// Did we consume the whole outbound buffer? If yes,
+			// pop it off and keep looping. If no, the outbound network
+			// buffers are full, so break out of here.
+			if (b.remaining() == 0)
+				outboundQ.removeFirst();
+			else
+				break;
+		}
+
+		if (outboundQ.isEmpty()) {
+			try {
+				channel.register(selector, SelectionKey.OP_READ, this);
+			} catch (ClosedChannelException e) {
+			}
+		}
+		
+		// ALWAYS drain the outbound queue before triggering a connection close.
+		// If anyone wants to close immediately, they're responsible for clearing
+		// the outbound queue.
+		return (bCloseScheduled && outboundQ.isEmpty()) ? false : true;
+ 	}
+	
+	public void setConnectPending() throws ClosedChannelException {
+		channel.register(selector, SelectionKey.OP_CONNECT, this);
+		bConnectPending = true;
+	}
+	
+	/**
+	 * Called by the reactor when we have selected connectable.
+	 * Return false to indicate an error that should cause the connection to close.
+	 * @throws ClosedChannelException
+	 */
+	public boolean finishConnecting() throws ClosedChannelException {
+		try {
+			channel.finishConnect();
+		}
+		catch (IOException e) {
+			return false;
+		}
+		bConnectPending = false;
+		channel.register(selector, SelectionKey.OP_READ | (outboundQ.isEmpty() ? 0 : SelectionKey.OP_WRITE), this);
+		return true;
+	}
+	
+	public void scheduleClose (boolean afterWriting) {
+		// TODO: What the hell happens here if bConnectPending is set?
+		if (!afterWriting)
+			outboundQ.clear();
+		try {
+			channel.register(selector, SelectionKey.OP_READ|SelectionKey.OP_WRITE, this);
+		} catch (ClosedChannelException e) {
+			throw new RuntimeException ("unable to schedule close"); // TODO, get rid of this.
+		}
+		bCloseScheduled = true;
+	}
+	public void startTls() {
+		if (sslEngine == null) {
+			try {
+				sslContext = SSLContext.getInstance("TLS");
+				sslContext.init(null, null, null); // TODO, fill in the parameters.
+				sslEngine = sslContext.createSSLEngine(); // TODO, should use the parameterized version, to get Kerb stuff and session re-use.
+				sslEngine.setUseClientMode(false);
+			} catch (NoSuchAlgorithmException e) {
+				throw new RuntimeException ("unable to start TLS"); // TODO, get rid of this.				
+			} catch (KeyManagementException e) {
+				throw new RuntimeException ("unable to start TLS"); // TODO, get rid of this.				
+			}
+		}
+		System.out.println ("Starting TLS");
+	}
+	
+	public ByteBuffer dispatchInboundData (ByteBuffer bb) throws SSLException {
+		if (sslEngine != null) {
+			if (true) throw new RuntimeException ("TLS currently unimplemented");
+			System.setProperty("javax.net.debug", "all");
+			ByteBuffer w = ByteBuffer.allocate(32*1024); // TODO, WRONG, preallocate this buffer.
+			SSLEngineResult res = sslEngine.unwrap(bb, w);
+			if (res.getHandshakeStatus() == HandshakeStatus.NEED_TASK) {
+				Runnable r;
+				while ((r = sslEngine.getDelegatedTask()) != null) {
+					r.run();
+				}
+			}
+			System.out.println (bb);
+			w.flip();
+			return w;
+		}
+		else
+			return bb;
+	}
+
+	public void setCommInactivityTimeout (long seconds) {
+		// TODO
+		System.out.println ("SOCKET: SET COMM INACTIVITY UNIMPLEMENTED " + seconds);
+	}
+}

data/java/src/com/rubyeventmachine/PeriodicTimer.java

@@ -0,0 +1,38 @@
+/**
+ * $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;
+
+public class PeriodicTimer extends Timer {
+
+	public void _fire() {
+		fire();
+		application.addTimer(interval, this);
+	}
+}

data/java/src/com/rubyeventmachine/Timer.java

@@ -0,0 +1,54 @@
+/**
+ * $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;
+
+public class Timer {
+	/**
+	 * User code is expected to call a method on a controlling Application,
+	 * which will fill in this field so subsequent user code can access it.
+	 */
+	public Application application;
+	public double interval;
+	
+	/**
+	 * The reactor calls here, and it may be overridden in subclasses.
+	 * User code should never call this method.
+	 */
+	public void _fire() {
+		fire();
+	}
+
+	/**
+	 * User code is expected to override this method.
+	 */
+	public void fire() {
+	}
+
+}

data/java/src/com/rubyeventmachine/tests/ApplicationTest.java

@@ -0,0 +1,108 @@
+/**
+ * $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.tests;
+
+
+import org.junit.After;
+import org.junit.AfterClass;
+import org.junit.Before;
+import org.junit.BeforeClass;
+import org.junit.Test;
+import org.junit.Assert;
+import java.net.*;
+import java.io.*;
+import java.nio.*;
+
+import com.rubyeventmachine.*;
+
+public class ApplicationTest {
+
+	@BeforeClass
+	public static void setUpBeforeClass() throws Exception {
+	}
+
+	@AfterClass
+	public static void tearDownAfterClass() throws Exception {
+	}
+
+	@Before
+	public void setUp() throws Exception {
+	}
+
+	@After
+	public void tearDown() throws Exception {
+	}
+
+	@Test
+	public void testRunnableArgument() {
+		final Application a = new Application();
+		a.run (new Runnable() {
+			public void run() {
+				a.stop();
+			}
+		});
+	}
+	
+
+	
+	class F implements ConnectionFactory {
+		public Connection connection() {
+			return new Connection() {
+				public void receiveData (ByteBuffer bb) {
+					application.stop();
+				}
+			};
+		}
+		
+	};
+	
+	@Test
+	public void testTcpServer() throws EmReactorException {
+		final Application a = new Application();
+		final SocketAddress saddr = new InetSocketAddress ("127.0.0.1", 9008);
+		a.run (new Runnable() {
+			public void run() {
+				try {
+					a.startServer (saddr, new F());
+				} catch (EmReactorException e) { Assert.fail(); }
+				new Thread() {
+					public void run() {
+						try {
+							Socket s = new Socket ("127.0.0.1", 9008);
+							s.getOutputStream().write(new String ("boo").getBytes());
+						} catch (UnknownHostException e) {
+						} catch (IOException e) {}
+					}
+				}.start();
+			}
+		});
+	}
+}

data/java/src/com/rubyeventmachine/tests/ConnectTest.java

@@ -0,0 +1,124 @@
+/**
+ * $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.tests;
+
+
+import org.junit.After;
+import org.junit.AfterClass;
+import org.junit.Before;
+import org.junit.BeforeClass;
+import org.junit.Test;
+import java.io.*;
+import java.nio.*;
+import java.nio.channels.*;
+
+import com.rubyeventmachine.*;
+
+public class ConnectTest {
+
+	@BeforeClass
+	public static void setUpBeforeClass() throws Exception {
+	}
+
+	@AfterClass
+	public static void tearDownAfterClass() throws Exception {
+	}
+
+	@Before
+	public void setUp() throws Exception {
+	}
+
+	@After
+	public void tearDown() throws Exception {
+	}
+
+	@Test
+	public final void test1() throws IOException, ClosedChannelException {
+		Application a = new Application();
+		a.addTimer(0, new Timer() {
+			public void fire() {
+				application.connect("www.bayshorenetworks.com", 80, new Connection() {
+					public void connectionCompleted() {
+						close();
+					}
+					public void unbind() {
+						application.stop();
+					}
+				});
+			}
+		});
+		a.run();
+	}
+	
+	@Test
+	public final void test2() throws IOException {
+		class Bays extends Connection {
+			public void connectionCompleted() {
+				sendData (ByteBuffer.wrap( new String ("GET / HTTP/1.1\r\nHost: _\r\n\r\n").getBytes()));
+			}
+			public void receiveData (ByteBuffer b) {
+				System.out.println (new String(b.array()));
+				application.stop();
+			}
+		};
+		
+		Application a = new Application();
+		a.addTimer(0, new Timer() {
+			public void fire() {
+				application.connect("www.bayshorenetworks.com", 80, new Bays());
+			}
+		});
+		a.run();
+	}
+	
+	
+	
+	class C1 extends Connection {
+		Application application;
+		public C1 (Application a) {
+			application = a;
+		}
+		public void postInit() {
+			application.stop();
+		}
+	}
+	@Test
+	public final void test3() {
+		final Application a = new Application();
+		a.run (new Runnable() {
+			public void run() {
+				a.connect("www.bayshorenetworks.com", 80, new C1(a));
+			}
+		});
+	}
+	
+	
+
+}

data/java/src/com/rubyeventmachine/tests/EMTest.java

@@ -0,0 +1,80 @@
+/**
+ * $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.tests;
+
+//import static org.junit.Assert.*;
+
+import org.junit.After;
+import org.junit.AfterClass;
+import org.junit.Before;
+import org.junit.BeforeClass;
+import org.junit.Test;
+
+import com.rubyeventmachine.EmReactor;
+
+import java.io.*;
+import java.nio.*;
+
+
+public class EMTest {
+
+	class ShortTimer extends EmReactor {
+		public void eventCallback (String sig, int eventCode, ByteBuffer data) {
+			System.out.println ("Short Callback "+sig+" "+eventCode+" "+data);
+			this.stop();
+		}
+	}
+	
+	@BeforeClass
+	public static void setUpBeforeClass() throws Exception {
+	}
+
+	@AfterClass
+	public static void tearDownAfterClass() throws Exception {
+	}
+
+	@Before
+	public void setUp() throws Exception {
+	}
+
+	@After
+	public void tearDown() throws Exception {
+	}
+ 
+
+	@Test
+	public final void testOneShort() throws IOException {
+		EmReactor em = new ShortTimer();
+		em.installOneshotTimer(1050);
+		em.run();
+	}
+	
+
+}

data/java/src/com/rubyeventmachine/tests/TestDatagrams.java

@@ -0,0 +1,53 @@
+package com.rubyeventmachine.tests;
+
+import com.rubyeventmachine.*;
+import java.net.*;
+import java.nio.*;
+
+import org.junit.After;
+import org.junit.AfterClass;
+import org.junit.Before;
+import org.junit.BeforeClass;
+import org.junit.Test;
+
+
+public class TestDatagrams {
+
+	@BeforeClass
+	public static void setUpBeforeClass() throws Exception {
+	}
+
+	@AfterClass
+	public static void tearDownAfterClass() throws Exception {
+	}
+
+	@Before
+	public void setUp() throws Exception {
+	}
+
+	@After
+	public void tearDown() throws Exception {
+	}
+
+	class A extends Connection {
+		public void receiveData (ByteBuffer bb) {
+			application.stop();
+		}
+	}
+	class B extends Connection {
+		public void postInit() {
+			this.sendDatagram(ByteBuffer.wrap(new String("ABC").getBytes()), new InetSocketAddress ("127.0.0.1", 9550));
+		}
+		
+	}
+	@Test
+	public final void testA() {
+		final Application a = new Application();
+		a.run (new Runnable() {
+			public void run() {
+				a.openDatagramSocket( new InetSocketAddress ("0.0.0.0", 9550), new A() );
+				a.openDatagramSocket( new B() );
+			}
+		});
+	}
+}