The message encoders and decoders now catch exceptions, and throw InvalidMessageException.

Removed the big switch statement from BinaryMessageDecoder - it now uses the decoder factory instead.
issue #35 #36
#story[1095]

racevisionGame/src/main/java/network/BinaryMessageDecoder.java

@@ -17,42 +17,71 @@ import java.util.zip.CRC32;
  */
 public class BinaryMessageDecoder {
 
-    ///Length of the header.
+    /**
+     * Length of the header.
+     */
     private static final int headerLength = 15;
-    ///Length of the CRC.
-    private static final int CRCLength = 4;//TODO these should probably be static defined somewhere else to be shared.
+    /**
+     * Length of the CRC.
+     */
+    private static final int CRCLength = 4;
 
-    ///The value the first sync byte should have.
+    /**
+     * The value the first sync byte should have.
+     */
     private static final byte syncByte1 = (byte) 0x47;
-    //The value the second sync byte should have.
+    /**
+     * The value the second sync byte should have.
+     */
     private static final byte syncByte2 = (byte) 0x83;
 
-    ///The full message.
+    /**
+     * The full message.
+     */
     private byte[] fullMessage;
-    ///The messageHeader.
+    /**
+     * The messageHeader.
+     */
     private byte[] messageHeader;
-    ///The messageBody.
+    /**
+     * The messageBody.
+     */
     private byte[] messageBody;
 
-    ///The sync bytes from the header..
+    /**
+     * The sync bytes from the header.
+     */
     private byte headerSync1;
     private byte headerSync2;
 
-    ///The message type from the header.
+    /**
+     * The message type from the header.
+     */
     private byte headerMessageType;
 
-    ///The timestamp from the header.
+    /**
+     * The timestamp from the header.
+     */
     private long headerTimeStamp;
 
-    ///The source ID from the header.
+    /**
+     * The source ID from the header.
+     */
     private int headerSourceID;
 
-    ///The message body length from the header.
+    /**
+     * The message body length from the header.
+     */
     private int messageBodyLength;
 
-    ///CRC value read from message header.
+    /**
+     * CRC value read from message header.
+     */
     private long messageCRCValue;
-    ///Calculated CRC value from message.
+
+    /**
+     * Calculated CRC value from message.
+     */
     private long calculatedCRCValue;
 
 
@@ -129,90 +158,20 @@ public class BinaryMessageDecoder {
         }
 
         //Now we create the message object based on what is actually in the message body.
-        MessageType mType = MessageType.fromByte(headerMessageType);
+        MessageType messageType = MessageType.fromByte(headerMessageType);
 
-        /*MessageDecoder decoder = null;
+        MessageDecoder decoder = null;
         try {
-            decoder = DecoderFactory.create(mType);
+            decoder = DecoderFactory.create(messageType);
+
         } catch (InvalidMessageTypeException e) {
-            throw new InvalidMessageException("Could not create decoder for MessageType: " + mType, e);
+            throw new InvalidMessageException("Could not create decoder for MessageType: " + messageType, e);
+
         }
 
-        return decoder.decode(messageBody);*/
-
-
-        switch(mType) {
-            case HEARTBEAT:
-                //System.out.println("Decoding HeartBeat Message!");
-                HeartBeatDecoder heartBeatDecoder = new HeartBeatDecoder();
-                return heartBeatDecoder.decode(messageBody);
-
-            case RACESTATUS:
-                //System.out.println("Race Status Message");
-                RaceStatusDecoder rsdecoder = new RaceStatusDecoder();
-                return rsdecoder.decode(messageBody);
-
-            case DISPLAYTEXTMESSAGE:
-                //System.out.println("Display Text Message");
-                //No decoder for this.
-                //throw new InvalidMessageException("Cannot decode DISPLAYTEXTMESSAGE - no decoder.");
-
-            case XMLMESSAGE:
-                //System.out.println("XML Message!");
-                XMLMessageDecoder xmdecoder = new XMLMessageDecoder();
-                return xmdecoder.decode(messageBody);
-
-            case RACESTARTSTATUS:
-                //System.out.println("Race Start Status Message");
-                RaceStartStatusDecoder rssDecoder = new RaceStartStatusDecoder();
-                return rssDecoder.decode(messageBody);
-
-            case YACHTEVENTCODE:
-                //System.out.println("Yacht Action Code!");
-                //No decoder for this.
-                //throw new InvalidMessageException("Cannot decode YACHTEVENTCODE - no decoder.");
-
-            case YACHTACTIONCODE:
-                //System.out.println("Yacht Action Code!");
-                //No decoder for this.
-                //throw new InvalidMessageException("Cannot decode YACHTACTIONCODE - no decoder.");
-
-            case CHATTERTEXT:
-                //System.out.println("Chatter Text Message!");
-                //No decoder for this.
-                //throw new InvalidMessageException("Cannot decode CHATTERTEXT - no decoder.");
-
-            case BOATLOCATION:
-                //System.out.println("Boat Location Message!");
-                BoatLocationDecoder blDecoder = new BoatLocationDecoder();
-                return blDecoder.decode(messageBody);
-
-            case MARKROUNDING:
-                //System.out.println("Mark Rounding Message!");
-                MarkRoundingDecoder mrDecoder = new MarkRoundingDecoder();
-                return mrDecoder.decode(messageBody);
-
-            case COURSEWIND:
-                //System.out.println("Course Wind Message!");
-                CourseWindsDecoder cwDecoder = new CourseWindsDecoder();
-                return cwDecoder.decode(messageBody);
-
-            case AVGWIND:
-                //System.out.println("Average Wind Message!");
-                AverageWindDecoder awDecoder = new AverageWindDecoder();
-                return awDecoder.decode(messageBody);
-
-            case BOATACTION:
-                BoatActionDecoder baDecoder = new BoatActionDecoder();
-                return baDecoder.decode(messageBody);
-
-            default:
-                //System.out.println("Broken Message!");
-                //throw new InvalidMessageException("Broken message! Did not recognise message type: " + headerMessageType + ".");
-                return null;
+        return decoder.decode(messageBody);
 
     }
-    }
 
 
     /**

racevisionGame/src/main/java/network/MessageDecoders/AverageWindDecoder.java

@@ -1,6 +1,7 @@
 package network.MessageDecoders;
 
 
+import network.Exceptions.InvalidMessageException;
 import network.Messages.AC35Data;
 import network.Messages.AverageWind;
 import network.Utils.ByteConverter;
@@ -32,9 +33,10 @@ public class AverageWindDecoder implements MessageDecoder {
 
 
     @Override
-    public AC35Data decode(byte[] encodedMessage) {
+    public AC35Data decode(byte[] encodedMessage) throws InvalidMessageException {
         this.encodedMessage = encodedMessage;
 
+        try {
 
             byte messageVersionNumber = encodedMessage[0];
 
@@ -75,8 +77,6 @@ public class AverageWindDecoder implements MessageDecoder {
             double speed4Knots = unpackMMperSecToKnots(intSpeed4);
 
 
-
-
             message = new AverageWind(
                     messageVersionNumber,
                     time,
@@ -91,6 +91,10 @@ public class AverageWindDecoder implements MessageDecoder {
 
             return message;
 
+        } catch (Exception e) {
+            throw new InvalidMessageException("Could not decode AverageWind message.", e);
+        }
+
     }
 
     /**

racevisionGame/src/main/java/network/MessageDecoders/BoatActionDecoder.java

@@ -1,5 +1,6 @@
 package network.MessageDecoders;
 
+import network.Exceptions.InvalidMessageException;
 import network.Messages.AC35Data;
 import network.Messages.BoatAction;
 import network.Messages.Enums.BoatActionEnum;
@@ -29,14 +30,19 @@ public class BoatActionDecoder implements MessageDecoder {
     }
 
     @Override
-    public AC35Data decode(byte[] encodedMessage) {
+    public AC35Data decode(byte[] encodedMessage) throws InvalidMessageException {
         this.encodedMessage = encodedMessage;
 
+        try {
             BoatActionEnum boatActionEnum = BoatActionEnum.fromByte(encodedMessage[0]);
 
             message = new BoatAction(boatActionEnum);
 
             return message;
+
+        } catch (Exception e) {
+            throw new InvalidMessageException("Could not decode BoatAction message.", e);
+        }
     }
 
 

racevisionGame/src/main/java/network/MessageDecoders/BoatLocationDecoder.java

@@ -1,6 +1,7 @@
 package network.MessageDecoders;
 
 
+import network.Exceptions.InvalidMessageException;
 import network.Messages.AC35Data;
 import network.Messages.BoatLocation;
 import network.Messages.Enums.BoatLocationDeviceEnum;
@@ -39,9 +40,11 @@ public class BoatLocationDecoder implements MessageDecoder {
 
 
     @Override
-    public AC35Data decode(byte[] encodedMessage) {
+    public AC35Data decode(byte[] encodedMessage) throws InvalidMessageException {
         this.encodedMessage = encodedMessage;
 
+        try {
+
             byte[] messageVersionNumberBytes = Arrays.copyOfRange(encodedMessage, 0, 1);
             byte messageVersionNumber = messageVersionNumberBytes[0];
 
@@ -149,6 +152,10 @@ public class BoatLocationDecoder implements MessageDecoder {
 
             return message;
 
+        } catch (Exception e) {
+            throw new InvalidMessageException("Could not decode BoatLocation message.", e);
+        }
+
     }
 
 

racevisionGame/src/main/java/network/MessageDecoders/BoatStatusDecoder.java

@@ -1,6 +1,7 @@
 package network.MessageDecoders;
 
 
+import network.Exceptions.InvalidMessageException;
 import network.Messages.BoatStatus;
 import network.Messages.Enums.BoatStatusEnum;
 
@@ -39,10 +40,12 @@ public class BoatStatusDecoder {
      * Decodes the contained message.
      * @param encodedMessage The message to decode.
      * @return The decoded message.
+     * @throws InvalidMessageException Thrown if the encoded message is invalid in some way, or cannot be decoded.
      */
-    public BoatStatus decode(byte[] encodedMessage) {
+    public BoatStatus decode(byte[] encodedMessage) throws InvalidMessageException {
         this.encodedMessage = encodedMessage;
 
+        try {
 
         byte[] sourceIDBytes = Arrays.copyOfRange(encodedMessage, 0, 4);
         int sourceID = bytesToInt(sourceIDBytes);
@@ -76,6 +79,10 @@ public class BoatStatusDecoder {
 
         return message;
 
+        } catch (Exception e) {
+            throw new InvalidMessageException("Could not decode BoatStatus message.", e);
+        }
+
 
     }
 

racevisionGame/src/main/java/network/MessageDecoders/CourseWindDecoder.java

@@ -1,6 +1,7 @@
 package network.MessageDecoders;
 
 
+import network.Exceptions.InvalidMessageException;
 import network.Messages.CourseWind;
 import shared.model.Bearing;
 
@@ -38,10 +39,13 @@ public class CourseWindDecoder {
      * Decodes the contained message.
      * @param encodedMessage The message to decode.
      * @return The decoded message.
+     * @throws InvalidMessageException Thrown if the encoded message is invalid in some way, or cannot be decoded.
      */
-    public CourseWind decode(byte[] encodedMessage) {
+    public CourseWind decode(byte[] encodedMessage) throws InvalidMessageException {
         this.encodedMessage = encodedMessage;
 
+        try {
+
             byte[] windId = Arrays.copyOfRange(encodedMessage, 0, 1);
 
             byte[] timeBytes = Arrays.copyOfRange(encodedMessage, 1, 7);
@@ -69,7 +73,6 @@ public class CourseWindDecoder {
             byte[] flags = Arrays.copyOfRange(encodedMessage, 19, 20);
 
 
-
             message = new CourseWind(
                     windId[0],
                     time,
@@ -81,6 +84,10 @@ public class CourseWindDecoder {
                     flags[0]);
 
             return message;
+
+        } catch (Exception e) {
+            throw new InvalidMessageException("Could not decode CourseWind message.", e);
+        }
     }
 
 

racevisionGame/src/main/java/network/MessageDecoders/CourseWindsDecoder.java

@@ -1,6 +1,7 @@
 package network.MessageDecoders;
 
 
+import network.Exceptions.InvalidMessageException;
 import network.Messages.AC35Data;
 import network.Messages.CourseWind;
 import network.Messages.CourseWinds;
@@ -38,9 +39,11 @@ public class CourseWindsDecoder implements MessageDecoder {
 
 
     @Override
-    public AC35Data decode(byte[] encodedMessage) {
+    public AC35Data decode(byte[] encodedMessage) throws InvalidMessageException {
         this.encodedMessage = encodedMessage;
 
+        try {
+
             //The header is three bytes.
             byte messageVersionNumber = encodedMessage[0];
             byte byteWindID = encodedMessage[1];
@@ -66,13 +69,16 @@ public class CourseWindsDecoder implements MessageDecoder {
             }
 
 
-
             message = new CourseWinds(
                     messageVersionNumber,
                     byteWindID,
                     loopMessages);
 
             return message;
+
+        } catch (Exception e) {
+            throw new InvalidMessageException("Could not decode CourseWinds message.", e);
+        }
     }
 
 

racevisionGame/src/main/java/network/MessageDecoders/HeartBeatDecoder.java

@@ -1,5 +1,6 @@
 package network.MessageDecoders;
 
+import network.Exceptions.InvalidMessageException;
 import network.Messages.AC35Data;
 import network.Messages.Enums.BoatActionEnum;
 import network.Messages.HeartBeat;
@@ -30,12 +31,18 @@ public class HeartBeatDecoder implements MessageDecoder {
     }
 
     @Override
-    public AC35Data decode(byte[] encodedMessage) {
+    public AC35Data decode(byte[] encodedMessage) throws InvalidMessageException {
         this.encodedMessage = encodedMessage;
 
+        try {
+
             message = new HeartBeat(bytesToLong(encodedMessage));
 
             return message;
+
+        } catch (Exception e) {
+            throw new InvalidMessageException("Could not decode HeartBeat message.", e);
+        }
     }
 
 

racevisionGame/src/main/java/network/MessageDecoders/JoinAcceptanceDecoder.java

@@ -1,6 +1,7 @@
 package network.MessageDecoders;
 
 
+import network.Exceptions.InvalidMessageException;
 import network.Messages.AC35Data;
 import network.Messages.Enums.JoinAcceptanceEnum;
 import network.Messages.JoinAcceptance;
@@ -32,9 +33,11 @@ public class JoinAcceptanceDecoder implements MessageDecoder {
 
 
     @Override
-    public AC35Data decode(byte[] encodedMessage) {
+    public AC35Data decode(byte[] encodedMessage) throws InvalidMessageException {
         this.encodedMessage = encodedMessage;
 
+        try {
+
             //SourceID is first four bytes.
             byte[] sourceIdBytes = Arrays.copyOfRange(encodedMessage, 0, 4);
 
@@ -42,7 +45,6 @@ public class JoinAcceptanceDecoder implements MessageDecoder {
             byte[] acceptanceBytes = Arrays.copyOfRange(encodedMessage, 4, 5);
 
 
-
             //SourceID is an int.
             int sourceID = ByteConverter.bytesToInt(sourceIdBytes);
 
@@ -53,6 +55,10 @@ public class JoinAcceptanceDecoder implements MessageDecoder {
             message = new JoinAcceptance(acceptanceType, sourceID);
 
             return message;
+
+        } catch (Exception e) {
+            throw new InvalidMessageException("Could not decode JoinAcceptance message.", e);
+        }
     }
 
 

racevisionGame/src/main/java/network/MessageDecoders/MarkRoundingDecoder.java

@@ -1,6 +1,7 @@
 package network.MessageDecoders;
 
 
+import network.Exceptions.InvalidMessageException;
 import network.Messages.AC35Data;
 import network.Messages.Enums.MarkRoundingBoatStatusEnum;
 import network.Messages.Enums.MarkRoundingSideEnum;
@@ -33,9 +34,11 @@ public class MarkRoundingDecoder implements MessageDecoder {
     }
 
     @Override
-    public AC35Data decode(byte[] encodedMessage) {
+    public AC35Data decode(byte[] encodedMessage) throws InvalidMessageException {
         this.encodedMessage = encodedMessage;
 
+        try {
+
             byte messageVersionNumber = encodedMessage[0];
 
             byte[] byteTime = Arrays.copyOfRange(encodedMessage, 1, 7);
@@ -75,6 +78,10 @@ public class MarkRoundingDecoder implements MessageDecoder {
 
 
             return message;
+
+        } catch (Exception e) {
+            throw new InvalidMessageException("Could not decode AverageWind message.", e);
+        }
     }
 
     /**

racevisionGame/src/main/java/network/MessageDecoders/MessageDecoder.java

@@ -1,6 +1,7 @@
 package network.MessageDecoders;
 
 
+import network.Exceptions.InvalidMessageException;
 import network.Messages.AC35Data;
 
 
@@ -15,7 +16,8 @@ public interface MessageDecoder {
      * Decodes a given message.
      * @param encodedMessage The message to decode.
      * @return The decoded message.
+     * @throws InvalidMessageException Thrown if the encoded message is invalid in some way, or cannot be decoded.
      */
-    public AC35Data decode(byte[] encodedMessage);
+    public AC35Data decode(byte[] encodedMessage) throws InvalidMessageException;
 
 }

racevisionGame/src/main/java/network/MessageDecoders/RaceStartStatusDecoder.java

@@ -1,6 +1,7 @@
 package network.MessageDecoders;
 
 
+import network.Exceptions.InvalidMessageException;
 import network.Messages.AC35Data;
 import network.Messages.Enums.RaceStartTypeEnum;
 import network.Messages.RaceStartStatus;
@@ -35,9 +36,10 @@ public class RaceStartStatusDecoder implements MessageDecoder {
 
 
     @Override
-    public AC35Data decode(byte[] encodedMessage) {
+    public AC35Data decode(byte[] encodedMessage) throws InvalidMessageException {
         this.encodedMessage = encodedMessage;
 
+        try {
 
             byte messageVersion = encodedMessage[0];
 
@@ -56,7 +58,6 @@ public class RaceStartStatusDecoder implements MessageDecoder {
             byte notificationType = encodedMessage[19];
 
 
-
             message = new RaceStartStatus(
                     messageVersion,
                     time,
@@ -68,6 +69,10 @@ public class RaceStartStatusDecoder implements MessageDecoder {
 
             return message;
 
+        } catch (Exception e) {
+            throw new InvalidMessageException("Could not decode RaceStartStatus message.", e);
+        }
+
     }
 
 

racevisionGame/src/main/java/network/MessageDecoders/RaceStatusDecoder.java

@@ -1,6 +1,7 @@
 package network.MessageDecoders;
 
 
+import network.Exceptions.InvalidMessageException;
 import network.Messages.AC35Data;
 import network.Messages.BoatStatus;
 import network.Messages.Enums.RaceStatusEnum;
@@ -43,9 +44,10 @@ public class RaceStatusDecoder implements MessageDecoder {
 
 
     @Override
-    public AC35Data decode(byte[] encodedMessage) {
+    public AC35Data decode(byte[] encodedMessage) throws InvalidMessageException {
         this.encodedMessage = encodedMessage;
 
+        try {
 
             byte[] versionNumBytes = Arrays.copyOfRange(encodedMessage, 0, 1);
             byte versionNum = versionNumBytes[0];
@@ -107,6 +109,10 @@ public class RaceStatusDecoder implements MessageDecoder {
                     boatStatuses);
 
             return message;
+
+        } catch (Exception e) {
+            throw new InvalidMessageException("Could not decode RaceStatus message.", e);
+        }
     }
 
 

racevisionGame/src/main/java/network/MessageDecoders/RequestToJoinDecoder.java

@@ -1,6 +1,7 @@
 package network.MessageDecoders;
 
 
+import network.Exceptions.InvalidMessageException;
 import network.Messages.AC35Data;
 import network.Messages.Enums.RequestToJoinEnum;
 import network.Messages.RequestToJoin;
@@ -31,9 +32,11 @@ public class RequestToJoinDecoder implements MessageDecoder{
 
 
     @Override
-    public AC35Data decode(byte[] encodedRequest) {
+    public AC35Data decode(byte[] encodedRequest) throws InvalidMessageException {
         this.encodedRequest = encodedRequest;
 
+        try {
+
             //Request type is first four bytes.
             byte[] requestTypeBytes = Arrays.copyOfRange(encodedRequest, 0, 4);
 
@@ -45,6 +48,10 @@ public class RequestToJoinDecoder implements MessageDecoder{
             message = new RequestToJoin(requestType);
 
             return message;
+
+        } catch (Exception e) {
+            throw new InvalidMessageException("Could not decode RequestToJoin message.", e);
+        }
     }
 
 

racevisionGame/src/main/java/network/MessageDecoders/XMLMessageDecoder.java

@@ -1,5 +1,6 @@
 package network.MessageDecoders;
 
+import network.Exceptions.InvalidMessageException;
 import network.Messages.AC35Data;
 import network.Messages.Enums.XMLMessageType;
 import network.Messages.XMLMessage;
@@ -35,9 +36,11 @@ public class XMLMessageDecoder implements MessageDecoder {
 
 
     @Override
-    public AC35Data decode(byte[] encodedMessage) {
+    public AC35Data decode(byte[] encodedMessage) throws InvalidMessageException {
         this.encodedMessage = encodedMessage;
 
+        try {
+
             byte[] messageVersionNumberBytes = Arrays.copyOfRange(encodedMessage, 0, 1);
             byte[] ackNumberBytes = Arrays.copyOfRange(encodedMessage, 1, 3);
             byte[] timeStampBytes = Arrays.copyOfRange(encodedMessage, 3, 9);
@@ -65,6 +68,10 @@ public class XMLMessageDecoder implements MessageDecoder {
                     xmlMessage);
 
             return message;
+
+        } catch (Exception e) {
+            throw new InvalidMessageException("Could not decode XMLMessage message.", e);
+        }
     }
 
 

racevisionGame/src/main/java/network/MessageEncoders/AverageWindEncoder.java

@@ -1,6 +1,7 @@
 package network.MessageEncoders;
 
 
+import network.Exceptions.InvalidMessageException;
 import network.Messages.AC35Data;
 import network.Messages.AverageWind;
 
@@ -24,7 +25,9 @@ public class AverageWindEncoder implements MessageEncoder {
 
 
     @Override
-    public byte[] encode(AC35Data message) {
+    public byte[] encode(AC35Data message) throws InvalidMessageException {
+
+        try {
 
             //Downcast.
             AverageWind averageWind = (AverageWind) message;
@@ -68,7 +71,6 @@ public class AverageWindEncoder implements MessageEncoder {
             byte[] byteSpeed4 = intToBytes(speed4Int, 2);
 
 
-
             ByteBuffer result = ByteBuffer.allocate(23);
             result.put(messageVersionNumber);
             result.put(byteTime);
@@ -82,5 +84,9 @@ public class AverageWindEncoder implements MessageEncoder {
             result.put(byteSpeed4);
             return result.array();
 
+        } catch (Exception e) {
+            throw new InvalidMessageException("Could not encode AverageWind message.", e);
+        }
+
     }
 }

racevisionGame/src/main/java/network/MessageEncoders/BoatActionEncoder.java

@@ -1,6 +1,7 @@
 package network.MessageEncoders;
 
 
+import network.Exceptions.InvalidMessageException;
 import network.Messages.AC35Data;
 import network.Messages.BoatAction;
 
@@ -22,7 +23,9 @@ public class BoatActionEncoder implements MessageEncoder {
 
 
     @Override
-    public byte[] encode(AC35Data message) {
+    public byte[] encode(AC35Data message) throws InvalidMessageException {
+
+        try {
 
             //Downcast.
             BoatAction boatAction = (BoatAction) message;
@@ -36,5 +39,9 @@ public class BoatActionEncoder implements MessageEncoder {
 
             return result;
 
+        } catch (Exception e) {
+            throw new InvalidMessageException("Could not encode BoatAction message.", e);
+        }
+
     }
 }

racevisionGame/src/main/java/network/MessageEncoders/BoatLocationEncoder.java

@@ -1,6 +1,7 @@
 package network.MessageEncoders;
 
 
+import network.Exceptions.InvalidMessageException;
 import network.Messages.AC35Data;
 import network.Messages.BoatLocation;
 
@@ -24,7 +25,9 @@ public class BoatLocationEncoder implements MessageEncoder {
 
 
     @Override
-    public byte[] encode(AC35Data message) {
+    public byte[] encode(AC35Data message) throws InvalidMessageException {
+
+        try {
 
             //Downcast.
             BoatLocation boatLocation = (BoatLocation) message;
@@ -80,5 +83,9 @@ public class BoatLocationEncoder implements MessageEncoder {
 
             return result.array();
 
+        } catch (Exception e) {
+            throw new InvalidMessageException("Could not encode BoatLocation message.", e);
+        }
+
     }
 }

racevisionGame/src/main/java/network/MessageEncoders/BoatStatusEncoder.java

@@ -1,6 +1,7 @@
 package network.MessageEncoders;
 
 
+import network.Exceptions.InvalidMessageException;
 import network.Messages.BoatStatus;
 
 import java.nio.ByteBuffer;
@@ -25,8 +26,12 @@ public class BoatStatusEncoder {
      * Encodes a given BoatStatus message.
      * @param message The message to encode.
      * @return The encoded message.
+     * @throws InvalidMessageException Thrown if the message is invalid in some way, or cannot be encoded.
      */
-    public byte[] encode(BoatStatus message) {
+    public byte[] encode(BoatStatus message) throws InvalidMessageException {
+
+        try {
+
 
             //Downcast.
             BoatStatus boatStatus = (BoatStatus) message;
@@ -52,5 +57,9 @@ public class BoatStatusEncoder {
 
             return boatStatusBuffer.array();
 
+        } catch (Exception e) {
+            throw new InvalidMessageException("Could not encode BoatStatus message.", e);
+        }
+
     }
 }

racevisionGame/src/main/java/network/MessageEncoders/CourseWindEncoder.java

@@ -1,6 +1,7 @@
 package network.MessageEncoders;
 
 
+import network.Exceptions.InvalidMessageException;
 import network.Messages.CourseWind;
 import shared.model.Bearing;
 
@@ -28,8 +29,11 @@ public class CourseWindEncoder {
      * Encodes a given CourseWind message.
      * @param message The message to encode.
      * @return The encoded message.
+     * @throws InvalidMessageException Thrown if the message is invalid in some way, or cannot be encoded.
      */
-    public byte[] encode(CourseWind message) {
+    public byte[] encode(CourseWind message) throws InvalidMessageException {
+
+        try {
 
             CourseWind courseWind = message;
 
@@ -69,5 +73,9 @@ public class CourseWindEncoder {
 
             return courseWindBuffer.array();
 
+        } catch (Exception e) {
+            throw new InvalidMessageException("Could not encode CourseWind message.", e);
+        }
+
     }
 }

racevisionGame/src/main/java/network/MessageEncoders/CourseWindsEncoder.java

@@ -1,6 +1,7 @@
 package network.MessageEncoders;
 
 
+import network.Exceptions.InvalidMessageException;
 import network.Messages.AC35Data;
 import network.Messages.BoatAction;
 import network.Messages.CourseWind;
@@ -24,7 +25,9 @@ public class CourseWindsEncoder implements MessageEncoder {
 
 
     @Override
-    public byte[] encode(AC35Data message) {
+    public byte[] encode(AC35Data message) throws InvalidMessageException {
+
+        try {
 
             //Downcast.
             CourseWinds courseWinds = (CourseWinds) message;
@@ -52,6 +55,10 @@ public class CourseWindsEncoder implements MessageEncoder {
             }
             return result.array();
 
+        } catch (Exception e) {
+            throw new InvalidMessageException("Could not encode CourseWinds message.", e);
+        }
+
 
     }
 }

racevisionGame/src/main/java/network/MessageEncoders/HeartBeatEncoder.java

@@ -1,6 +1,7 @@
 package network.MessageEncoders;
 
 
+import network.Exceptions.InvalidMessageException;
 import network.Messages.AC35Data;
 import network.Messages.HeartBeat;
 
@@ -22,7 +23,9 @@ public class HeartBeatEncoder implements MessageEncoder {
 
 
     @Override
-    public byte[] encode(AC35Data message) {
+    public byte[] encode(AC35Data message) throws InvalidMessageException {
+
+        try {
 
             //Downcast.
             HeartBeat heartbeat = (HeartBeat) message;
@@ -35,5 +38,9 @@ public class HeartBeatEncoder implements MessageEncoder {
 
             return result;
 
+        } catch (Exception e) {
+            throw new InvalidMessageException("Could not encode HeartBeat message.", e);
+        }
+
     }
 }

racevisionGame/src/main/java/network/MessageEncoders/JoinAcceptanceEncoder.java

@@ -1,6 +1,7 @@
 package network.MessageEncoders;
 
 
+import network.Exceptions.InvalidMessageException;
 import network.Messages.AC35Data;
 import network.Messages.JoinAcceptance;
 import network.Utils.ByteConverter;
@@ -23,7 +24,9 @@ public class JoinAcceptanceEncoder implements MessageEncoder {
 
 
     @Override
-    public byte[] encode(AC35Data message) {
+    public byte[] encode(AC35Data message) throws InvalidMessageException {
+
+        try {
 
             //Downcast.
             JoinAcceptance joinAcceptance = (JoinAcceptance) message;
@@ -40,5 +43,9 @@ public class JoinAcceptanceEncoder implements MessageEncoder {
 
             return result;
 
+        } catch (Exception e) {
+            throw new InvalidMessageException("Could not encode JoinAcceptance message.", e);
+        }
+
     }
 }

racevisionGame/src/main/java/network/MessageEncoders/MarkRoundingEncoder.java

@@ -1,6 +1,7 @@
 package network.MessageEncoders;
 
 
+import network.Exceptions.InvalidMessageException;
 import network.Messages.AC35Data;
 import network.Messages.MarkRounding;
 
@@ -23,7 +24,9 @@ public class MarkRoundingEncoder implements MessageEncoder {
 
 
     @Override
-    public byte[] encode(AC35Data message) {
+    public byte[] encode(AC35Data message) throws InvalidMessageException {
+
+        try {
 
             //Downcast.
             MarkRounding markRounding = (MarkRounding) message;
@@ -53,5 +56,9 @@ public class MarkRoundingEncoder implements MessageEncoder {
 
             return result.array();
 
+        } catch (Exception e) {
+            throw new InvalidMessageException("Could not encode MarkRounding message.", e);
+        }
+
     }
 }

racevisionGame/src/main/java/network/MessageEncoders/MessageEncoder.java

@@ -1,6 +1,7 @@
 package network.MessageEncoders;
 
 
+import network.Exceptions.InvalidMessageException;
 import network.Messages.AC35Data;
 
 
@@ -15,7 +16,8 @@ public interface MessageEncoder {
      * Encodes a given message.
      * @param message The message to encode.
      * @return Message in byte encoded form.
+     * @throws InvalidMessageException Thrown if the message is invalid in some way, or cannot be encoded.
      */
-    public byte[] encode(AC35Data message);
+    public byte[] encode(AC35Data message) throws InvalidMessageException;
 
 }

racevisionGame/src/main/java/network/MessageEncoders/RaceStartStatusEncoder.java

@@ -1,6 +1,7 @@
 package network.MessageEncoders;
 
 
+import network.Exceptions.InvalidMessageException;
 import network.Messages.AC35Data;
 import network.Messages.RaceStartStatus;
 import network.Utils.ByteConverter;
@@ -24,7 +25,9 @@ public class RaceStartStatusEncoder implements MessageEncoder {
 
 
     @Override
-    public byte[] encode(AC35Data message) {
+    public byte[] encode(AC35Data message) throws InvalidMessageException {
+
+        try {
 
             //Downcast.
             RaceStartStatus raceStartStatus = (RaceStartStatus) message;
@@ -47,5 +50,9 @@ public class RaceStartStatusEncoder implements MessageEncoder {
 
             return result.array();
 
+        } catch (Exception e) {
+            throw new InvalidMessageException("Could not encode RaceStartStatus message.", e);
+        }
+
     }
 }

racevisionGame/src/main/java/network/MessageEncoders/RaceStatusEncoder.java

@@ -1,6 +1,7 @@
 package network.MessageEncoders;
 
 
+import network.Exceptions.InvalidMessageException;
 import network.Messages.AC35Data;
 import network.Messages.BoatStatus;
 import network.Messages.RaceStatus;
@@ -28,7 +29,9 @@ public class RaceStatusEncoder implements MessageEncoder {
 
 
     @Override
-    public byte[] encode(AC35Data message) {
+    public byte[] encode(AC35Data message) throws InvalidMessageException {
+
+        try {
 
             //Downcast.
             RaceStatus raceStatus = (RaceStatus) message;
@@ -91,5 +94,9 @@ public class RaceStatusEncoder implements MessageEncoder {
 
             return raceStatusMessage.array();
 
+        } catch (Exception e) {
+            throw new InvalidMessageException("Could not encode RaceStatus message.", e);
+        }
+
     }
 }

racevisionGame/src/main/java/network/MessageEncoders/RaceVisionByteEncoder.java

@@ -113,8 +113,10 @@ public class RaceVisionByteEncoder {
         MessageEncoder encoder = null;
         try {
             encoder = EncoderFactory.create(message.getType());
+
         } catch (InvalidMessageTypeException e) {
             throw new InvalidMessageException("Could not create encoder for MessageType: " + message.getType(), e);
+
         }
 
         byte[] encodedMessage = encoder.encode(message);

racevisionGame/src/main/java/network/MessageEncoders/RequestToJoinEncoder.java

@@ -1,6 +1,7 @@
 package network.MessageEncoders;
 
 
+import network.Exceptions.InvalidMessageException;
 import network.Messages.AC35Data;
 import network.Messages.RequestToJoin;
 import network.Utils.ByteConverter;
@@ -21,7 +22,9 @@ public class RequestToJoinEncoder implements MessageEncoder {
 
 
     @Override
-    public byte[] encode(AC35Data message) {
+    public byte[] encode(AC35Data message) throws InvalidMessageException {
+
+        try {
 
             //Downcast.
             RequestToJoin requestToJoin = (RequestToJoin) message;
@@ -35,5 +38,9 @@ public class RequestToJoinEncoder implements MessageEncoder {
 
             return result;
 
+        } catch (Exception e) {
+            throw new InvalidMessageException("Could not encode RequestToJoin message.", e);
+        }
+
     }
 }

racevisionGame/src/main/java/network/MessageEncoders/XMLMessageEncoder.java

@@ -1,6 +1,7 @@
 package network.MessageEncoders;
 
 
+import network.Exceptions.InvalidMessageException;
 import network.Messages.AC35Data;
 import network.Messages.XMLMessage;
 
@@ -24,7 +25,9 @@ public class XMLMessageEncoder implements MessageEncoder {
 
 
     @Override
-    public byte[] encode(AC35Data message) {
+    public byte[] encode(AC35Data message) throws InvalidMessageException {
+
+        try {
 
             //Downcast.
             XMLMessage xmlMessage = (XMLMessage) message;
@@ -58,5 +61,9 @@ public class XMLMessageEncoder implements MessageEncoder {
 
             return tempOutputByteBuffer.array();
 
+        } catch (Exception e) {
+            throw new InvalidMessageException("Could not encode XMLMessage message.", e);
+        }
+
     }
 }

racevisionGame/src/main/java/visualiser/gameController/ControllerClient.java

@@ -56,18 +56,21 @@ public class ControllerClient  {
             BoatAction boatAction = new BoatAction(protocolCode);
 
             //Encode BoatAction.
-            byte[] encodedBoatAction = new byte[0];
             try {
-                encodedBoatAction = RaceVisionByteEncoder.encode(boatAction);
-            } catch (InvalidMessageException e) {
-                Logger.getGlobal().log(Level.WARNING, "Could not encode BoatAction: " + boatAction, e);
-            }
+                byte[] encodedBoatAction = RaceVisionByteEncoder.encode(boatAction);
 
                 BinaryMessageEncoder binaryMessage = new BinaryMessageEncoder(MessageType.BOATACTION, System.currentTimeMillis(), 0,
                         (short) encodedBoatAction.length, encodedBoatAction);
 
                 System.out.println("Sending out key: " + protocolCode);
                 outputStream.write(binaryMessage.getFullMessage());
+
+            } catch (InvalidMessageException e) {
+                Logger.getGlobal().log(Level.WARNING, "Could not encode BoatAction: " + boatAction, e);
+
+            }
+
+
         }
     }
 }

racevisionGame/src/main/java/visualiser/gameController/ControllerServer.java

@@ -1,6 +1,7 @@
 package visualiser.gameController;
 
 import network.BinaryMessageDecoder;
+import network.Exceptions.InvalidMessageException;
 import network.MessageDecoders.BoatActionDecoder;
 import network.Messages.BoatAction;
 import network.Messages.Enums.BoatActionEnum;
@@ -10,6 +11,8 @@ import visualiser.gameController.Keys.KeyFactory;
 import java.io.DataInputStream;
 import java.io.IOException;
 import java.net.Socket;
+import java.util.logging.Level;
+import java.util.logging.Logger;
 
 /**
  * Service for dispatching key press data to race from client
@@ -46,12 +49,22 @@ public class ControllerServer implements Runnable {
             byte[] message = new byte[20];
             try {
                 if (inputStream.available() > 0) {
+
                     inputStream.read(message);
+
                     BinaryMessageDecoder encodedMessage = new BinaryMessageDecoder(message);
                     BoatActionDecoder boatActionDecoder = new BoatActionDecoder();
+
+                    try {
                         boatActionDecoder.decode(encodedMessage.getMessageBody());
                         BoatAction boatAction = boatActionDecoder.getMessage();
                         System.out.println("Received key: " + boatAction.getBoatAction());
+
+                    } catch (InvalidMessageException e) {
+                        Logger.getGlobal().log(Level.WARNING, "Could not decode BoatAction message.", e);
+                    }
+
+
                 }
             } catch (IOException e) {
                 e.printStackTrace();

racevisionGame/src/test/java/network/MessageDecoders/BoatStatusDecoderTest.java

@@ -1,5 +1,6 @@
 package network.MessageDecoders;
 
+import network.Exceptions.InvalidMessageException;
 import network.MessageEncoders.BoatStatusEncoder;
 import network.MessageEncoders.RaceVisionByteEncoder;
 import network.Messages.BoatStatus;
@@ -55,8 +56,9 @@ public class BoatStatusDecoderTest {
      * Encodes and decodes a BoatStatus, and returns it.
      * @param boatStatus The BoatStatus to encode and decode.
      * @return The decoded BoatStatus.
+     * @throws InvalidMessageException Thrown if message cannot be encoded or decoded.
      */
-    private static BoatStatus encodeDecodeBoatStatus(BoatStatus boatStatus) {
+    private static BoatStatus encodeDecodeBoatStatus(BoatStatus boatStatus) throws InvalidMessageException {
 
         BoatStatusEncoder boatStatusEncoder = new BoatStatusEncoder();
         byte[] boatStatusEncoded = boatStatusEncoder.encode(boatStatus);

racevisionGame/src/test/java/network/MessageDecoders/CourseWindDecoderTest.java

@@ -1,5 +1,6 @@
 package network.MessageDecoders;
 
+import network.Exceptions.InvalidMessageException;
 import network.MessageEncoders.CourseWindEncoder;
 import network.Messages.BoatStatus;
 import network.Messages.CourseWind;
@@ -43,8 +44,9 @@ public class CourseWindDecoderTest {
      * Encodes and decodes a CourseWind, and returns it.
      * @param courseWind The CourseWind to encode and decode.
      * @return The decoded CourseWind.
+     * @throws InvalidMessageException Thrown if message cannot be encoded or decoded.
      */
-    private static CourseWind encodeDecodeCourseWind(CourseWind courseWind) {
+    private static CourseWind encodeDecodeCourseWind(CourseWind courseWind) throws InvalidMessageException {
 
         CourseWindEncoder courseWindEncoder = new CourseWindEncoder();
         byte[] courseWindEncoded = courseWindEncoder.encode(courseWind);