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.base; 3459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 3559b2e6871c65f58fdad78cd7229c292f6a177578Scott Bartaimport com.jme3.network.Message; 3659b2e6871c65f58fdad78cd7229c292f6a177578Scott Bartaimport com.jme3.network.serializing.Serializer; 3759b2e6871c65f58fdad78cd7229c292f6a177578Scott Bartaimport java.io.IOException; 3859b2e6871c65f58fdad78cd7229c292f6a177578Scott Bartaimport java.nio.ByteBuffer; 3959b2e6871c65f58fdad78cd7229c292f6a177578Scott Bartaimport java.util.LinkedList; 4059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 4159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta/** 4259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * Consolidates the conversion of messages to/from byte buffers 4359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * and provides a rolling message buffer. ByteBuffers can be 4459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * pushed in and messages will be extracted, accumulated, and 4559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * available for retrieval. This is not thread safe and is meant 4659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * to be used within a single message processing thread. 4759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * 4859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * <p>The protocol is based on a simple length + data format 4959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * where two bytes represent the (short) length of the data 5059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * and the rest is the raw data for the Serializers class.</p> 5159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * 5259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * @version $Revision: 8843 $ 5359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * @author Paul Speed 5459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta */ 5559b2e6871c65f58fdad78cd7229c292f6a177578Scott Bartapublic class MessageProtocol 5659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta{ 5759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta private LinkedList<Message> messages = new LinkedList<Message>(); 5859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta private ByteBuffer current; 5959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta private int size; 6059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta private Byte carry; 6159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 6259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta /** 6359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * Converts a message to a ByteBuffer using the Serializer 6459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * and the (short length) + data protocol. If target is null 6559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * then a 32k byte buffer will be created and filled. 6659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta */ 6759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta public static ByteBuffer messageToBuffer( Message message, ByteBuffer target ) 6859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta { 6959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta // Could let the caller pass their own in 7059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta ByteBuffer buffer = target == null ? ByteBuffer.allocate( 32767 + 2 ) : target; 7159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 7259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta try { 7359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta buffer.position( 2 ); 7459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta Serializer.writeClassAndObject( buffer, message ); 7559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta buffer.flip(); 7659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta short dataLength = (short)(buffer.remaining() - 2); 7759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta buffer.putShort( dataLength ); 7859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta buffer.position( 0 ); 7959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 8059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta return buffer; 8159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } catch( IOException e ) { 8259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta throw new RuntimeException( "Error serializing message", e ); 8359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 8459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 8559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 8659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta /** 8759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * Retrieves and removes an extracted message from the accumulated buffer 8859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * or returns null if there are no more messages. 8959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta */ 9059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta public Message getMessage() 9159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta { 9259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta if( messages.isEmpty() ) { 9359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta return null; 9459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 9559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 9659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta return messages.removeFirst(); 9759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 9859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 9959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta /** 10059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * Adds the specified buffer, extracting the contained messages 10159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * and making them available to getMessage(). The left over 10259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * data is buffered to be combined with future data. 10359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta & 10459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * @return The total number of queued messages after this call. 10559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta */ 10659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta public int addBuffer( ByteBuffer buffer ) 10759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta { 10859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta // push the data from the buffer into as 10959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta // many messages as we can 11059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta while( buffer.remaining() > 0 ) { 11159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 11259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta if( current == null ) { 11359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 11459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta // If we have a left over carry then we need to 11559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta // do manual processing to get the short value 11659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta if( carry != null ) { 11759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta byte high = carry; 11859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta byte low = buffer.get(); 11959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 12059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta size = (high & 0xff) << 8 | (low & 0xff); 12159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta carry = null; 12259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 12359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta else if( buffer.remaining() < 2 ) { 12459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta // It's possible that the supplied buffer only has one 12559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta // byte in it... and in that case we will get an underflow 12659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta // when attempting to read the short below. 12759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 12859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta // It has to be 1 or we'd never get here... but one 12959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta // isn't enough so we stash it away. 13059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta carry = buffer.get(); 13159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta break; 13259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } else { 13359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta // We are not currently reading an object so 13459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta // grab the size. 13559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta // Note: this is somewhat limiting... int would 13659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta // be better. 13759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta size = buffer.getShort(); 13859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 13959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 14059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta // Allocate the buffer into which we'll feed the 14159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta // data as we get it 14259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta current = ByteBuffer.allocate(size); 14359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 14459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 14559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta if( current.remaining() <= buffer.remaining() ) { 14659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta // We have at least one complete object so 14759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta // copy what we can into current, create a message, 14859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta // and then continue pulling from buffer. 14959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 15059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta // Artificially set the limit so we don't overflow 15159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta int extra = buffer.remaining() - current.remaining(); 15259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta buffer.limit( buffer.position() + current.remaining() ); 15359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 15459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta // Now copy the data 15559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta current.put( buffer ); 15659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta current.flip(); 15759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 15859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta // Now set the limit back to a good value 15959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta buffer.limit( buffer.position() + extra ); 16059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 16159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta createMessage( current ); 16259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 16359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta current = null; 16459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } else { 16559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 16659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta // Not yet a complete object so just copy what we have 16759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta current.put( buffer ); 16859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 16959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 17059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 17159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta return messages.size(); 17259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 17359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 17459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta /** 17559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * Creates a message from the properly sized byte buffer 17659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * and adds it to the messages queue. 17759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta */ 17859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta protected void createMessage( ByteBuffer buffer ) 17959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta { 18059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta try { 18159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta Object obj = Serializer.readClassAndObject( buffer ); 18259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta Message m = (Message)obj; 18359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta messages.add(m); 18459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } catch( IOException e ) { 18559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta throw new RuntimeException( "Error deserializing object", e ); 18659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 18759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 18859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta} 18959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 19059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 19159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 192