SecureMessage

import java.io.BufferedReader;
import java.io.DataOutputStream;
import java.io.DataOutputStream;
import java.io.InputStreamReader;
import java.io.ObjectInputStream;
import java.io.ObjectOutputStream;
import java.io.Serializable;
import java.net.InetAddress;
import java.net.ServerSocket;
import java.net.Socket;
import java.util.ArrayList;
import java.io.IOException;

  enum MessageType {
    MESSAGE, DISCONNECT, GETCLIENTS
};

public class SecureMessage implements Serializable {

 private String message;
 private InetAddress receiver;
 private InetAddress sender;
 private MessageType type; // can be MESSAGE, DISCONNECT, GETCLIENTS
 private static final long serialVersionUID = 42L;

 /**
    * Wrapper for any message that is sent between clients. The object gets serialized and sent through tcp.
    * @param   The message as a string object
    * @param   The InetAddress of the receiver
    * @param   The InetAddress of the sender
    */
 public SecureMessage(String message,MessageType messageType, InetAddress receiver, InetAddress sender) {
     this.message = message;
     this.type = messageType;
     this.receiver = receiver;
     this.sender = sender;
 }

 /**
    * Returns the message
    * @return A message
    */
 public String getMessage() {
     return this.message;
 }

 /**
    * Returns the receiver of the message
    * @return A InetAddress representing the receiver
    */
 public InetAddress getReceiver() {
     return this.receiver;
 }

 /**
    * Returns the sender of the message
    * @return A InetAddress representing the sender
    */
 public InetAddress getSender() {
     return this.sender;
 }

 public MessageType getType() {
     return this.type;
 }
}