UdpConnector.java revision 59b2e6871c65f58fdad78cd7229c292f6a177578
159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta/* 259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * Copyright (c) 2011 jMonkeyEngine 359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * All rights reserved. 459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * 559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * Redistribution and use in source and binary forms, with or without 659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * modification, are permitted provided that the following conditions are 759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * met: 859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * 959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * * Redistributions of source code must retain the above copyright 1059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * notice, this list of conditions and the following disclaimer. 1159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * 1259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * * Redistributions in binary form must reproduce the above copyright 1359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * notice, this list of conditions and the following disclaimer in the 1459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * documentation and/or other materials provided with the distribution. 1559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * 1659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * * Neither the name of 'jMonkeyEngine' nor the names of its contributors 1759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * may be used to endorse or promote products derived from this software 1859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * without specific prior written permission. 1959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * 2059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS 2159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED 2259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR 2359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR 2459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, 2559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, 2659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR 2759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF 2859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING 2959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS 3059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 3159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta */ 3259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 3359b2e6871c65f58fdad78cd7229c292f6a177578Scott Bartapackage com.jme3.network.kernel.udp; 3459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 3559b2e6871c65f58fdad78cd7229c292f6a177578Scott Bartaimport com.jme3.network.kernel.Connector; 3659b2e6871c65f58fdad78cd7229c292f6a177578Scott Bartaimport com.jme3.network.kernel.ConnectorException; 3759b2e6871c65f58fdad78cd7229c292f6a177578Scott Bartaimport java.io.IOException; 3859b2e6871c65f58fdad78cd7229c292f6a177578Scott Bartaimport java.net.*; 3959b2e6871c65f58fdad78cd7229c292f6a177578Scott Bartaimport java.nio.ByteBuffer; 4059b2e6871c65f58fdad78cd7229c292f6a177578Scott Bartaimport java.util.concurrent.atomic.AtomicBoolean; 4159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 4259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 4359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta/** 4459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * A straight forward datagram socket-based UDP connector 4559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * implementation. 4659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * 4759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * @version $Revision: 8843 $ 4859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * @author Paul Speed 4959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta */ 5059b2e6871c65f58fdad78cd7229c292f6a177578Scott Bartapublic class UdpConnector implements Connector 5159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta{ 5259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta private DatagramSocket sock = new DatagramSocket(); 5359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta private SocketAddress remoteAddress; 5459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta private byte[] buffer = new byte[65535]; 5559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta private AtomicBoolean connected = new AtomicBoolean(false); 5659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 5759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta /** 5859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * In order to provide proper available() checking, we 5959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * potentially queue one datagram. 6059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta */ 6159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta private DatagramPacket pending; 6259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 6359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta /** 6459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * Creates a new UDP connection that send datagrams to the 6559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * specified address and port. 6659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta */ 6759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta public UdpConnector( InetAddress remote, int remotePort ) throws IOException 6859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta { 6959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta InetSocketAddress localSocketAddress = new InetSocketAddress(0); 7059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta this.sock = new DatagramSocket( localSocketAddress ); 7159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta remoteAddress = new InetSocketAddress( remote, remotePort ); 7259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 7359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta // Setup to receive only from the remote address 7459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta sock.connect( remoteAddress ); 7559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 7659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta connected.set(true); 7759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 7859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 7959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta protected void checkClosed() 8059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta { 8159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta if( sock == null ) 8259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta throw new ConnectorException( "Connection is closed:" + remoteAddress ); 8359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 8459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 8559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta public boolean isConnected() 8659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta { 8759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta if( sock == null ) 8859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta return false; 8959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta return sock.isConnected(); 9059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 9159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 9259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta public void close() 9359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta { 9459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta checkClosed(); 9559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta DatagramSocket temp = sock; 9659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta sock = null; 9759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta connected.set(false); 9859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta temp.close(); 9959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 10059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 10159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta /** 10259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * This always returns false since the simple DatagramSocket usage 10359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * cannot be run in a non-blocking way. 10459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta */ 10559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta public boolean available() 10659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta { 10759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta // It would take a separate thread or an NIO Selector based implementation to get this 10859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta // to work. If a polling strategy is never employed by callers then it doesn't 10959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta // seem worth it to implement all of that just for this method. 11059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta checkClosed(); 11159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta return false; 11259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 11359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 11459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta public ByteBuffer read() 11559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta { 11659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta checkClosed(); 11759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 11859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta try { 11959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta DatagramPacket packet = new DatagramPacket( buffer, buffer.length ); 12059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta sock.receive(packet); 12159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 12259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta // Wrap it in a ByteBuffer for the caller 12359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta return ByteBuffer.wrap( buffer, 0, packet.getLength() ); 12459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } catch( IOException e ) { 12559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta if( !connected.get() ) { 12659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta // Nothing to see here... just move along 12759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta return null; 12859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 12959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta throw new ConnectorException( "Error reading from connection to:" + remoteAddress, e ); 13059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 13159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 13259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 13359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta public void write( ByteBuffer data ) 13459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta { 13559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta checkClosed(); 13659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 13759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta try { 13859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta DatagramPacket p = new DatagramPacket( data.array(), data.position(), data.remaining(), 13959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta remoteAddress ); 14059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta sock.send(p); 14159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } catch( IOException e ) { 14259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta throw new ConnectorException( "Error writing to connection:" + remoteAddress, e ); 14359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 14459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 14559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta} 14659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 147