NGECore2 Code

src/protocol/Message.java

@@ -0,0 +1,204 @@
+/*******************************************************************************
+ * Copyright (c) 2013 <Project SWG>
+ * 
+ * This File is part of NGECore2.
+ * 
+ * This program is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU Lesser General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ * 
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU Lesser General Public License for more details.
+ * 
+ * You should have received a copy of the GNU Lesser General Public License
+ * along with this program.  If not, see <http://www.gnu.org/licenses/>.
+ * 
+ * Using NGEngine to work with NGECore2 is making a combined work based on NGEngine. 
+ * Therefore all terms and conditions of the GNU Lesser General Public License cover the combination.
+ ******************************************************************************/
+package protocol;
+
+import java.io.UnsupportedEncodingException;
+import java.nio.ByteBuffer;
+import java.nio.ByteOrder;
+
+import org.apache.mina.core.buffer.IoBuffer;
+
+public abstract class Message {
+	
+	protected byte[] data;
+	protected int opcode;
+	protected short sequence;
+	protected IoBuffer result;
+	
+	public Message() {
+		
+	}
+	
+	public Message(byte[] data) { 
+		this.data = data; 
+	}
+	
+	public int getOpcode() {
+		return opcode;
+	}
+	
+	public int getSize() {
+		return (data == null) ? 0 : data.length;
+	}
+	
+	public byte[] getData() {
+		return data == null ? new byte[] { } : data.clone();
+	}
+	
+	public abstract void    deserialize(IoBuffer data);
+	public abstract IoBuffer	serialize();
+	
+	protected String getAsciiString(IoBuffer buffer)
+	{
+		return getString(buffer, "US-ASCII");
+	}
+	
+	protected String getUnicodeString(IoBuffer buffer)
+	{
+		return getString(buffer, "UTF-16LE");
+	}
+	
+	protected byte[] getAsciiString(String string)
+	{
+		return getString(string, "US-ASCII");
+	}
+	
+	protected byte[] getUnicodeString(String string)
+	{
+		return getString(string, "UTF-16LE");
+	}
+	
+	protected String getString(IoBuffer buffer, String charFormat)
+	{
+		
+		String result;
+		
+		int length;
+		
+		if (charFormat == "UTF-16LE") length = buffer.order(ByteOrder.LITTLE_ENDIAN).getInt();
+		else length = buffer.order(ByteOrder.LITTLE_ENDIAN).getShort();
+		
+		int bufferPosition = buffer.position();
+		
+		try
+		{
+			
+			result = new String(buffer.array(), bufferPosition, length, charFormat);
+			
+		}
+		catch (UnsupportedEncodingException e)
+		{
+			
+			e.printStackTrace();
+			return "";
+			
+		}
+		
+		buffer.position(bufferPosition + length);
+		
+		return result;
+		
+	}
+	private byte[] getString(String string, String charFormat)
+	{
+		
+		ByteBuffer result;
+		int length = 2 + string.length();
+		
+		if (charFormat == "UTF-16LE")
+		{
+			
+			result = ByteBuffer.allocate(length * 2).order(ByteOrder.LITTLE_ENDIAN);
+			result.putInt(string.length());
+			
+		}
+		else
+		{
+			
+			result = ByteBuffer.allocate(length).order(ByteOrder.LITTLE_ENDIAN);
+			result.putShort((short) string.length());
+			
+		}
+		
+		try
+		{
+			
+			result.put(string.getBytes(charFormat));
+			
+		}
+		catch (UnsupportedEncodingException e)
+		{
+			
+			e.printStackTrace();
+			return new byte[] {};
+			
+		}
+		
+		return result.array();
+		
+	}
+	
+	
+	/**
+	 * Reads the next string in. It gets the length of the
+	 * string from the short/int at the beginning of it
+	 * @param bb The ByteBuffer to read from
+	 * @param charset Charset of the string
+	 * @return The read string. Returns "" if exception is thrown
+	 */
+	protected String getNextString(IoBuffer bb, String charset) {
+		try {
+			int length = 0;
+			if (charset.equals("US-ASCII") || charset.equals("UTF-8")) {
+				length = bb.getShort();
+			} else {
+				length = bb.getInt();
+			}
+			if (length > bb.remaining())
+				return "";
+			byte [] data = new byte[length];
+			bb.get(data);
+			return new String(data, charset);
+		} catch (Exception e) {
+			e.printStackTrace();
+		}
+		return "";
+	}
+	
+	/**
+	 * Reads the next array with specified size and returns it
+	 * @param bb The ByteBuffer to read from
+	 * @param size Size of the array
+	 * @return The array that has been read, or an empty array if an exception
+	 */
+	protected byte [] getNextArray(IoBuffer bb, int size) {
+		try {
+			if (bb.remaining() < size)
+				return new byte[0];
+			byte [] tmp = new byte[size];
+			bb.get(tmp);
+			return tmp;
+		} catch (Exception e) {
+			e.printStackTrace();
+		}
+		return new byte[0];
+	}
+	
+	
+	protected String getNextAsciiString(IoBuffer buffer) {
+		return getNextString(buffer, "US-ASCII");
+	}
+	protected String getNextUnicodeString(IoBuffer buffer) {
+		return getNextString(buffer, "UTF-16LE");
+	}
+
+}