Small changes to ConnectionHandler and MessageHandler, while adding interfaces

for Packet, MessageIn and MessageOut.

1 files changed, 9 insertions, 0 deletions

diff --git a/src/connectionhandler.h b/src/connectionhandler.h
index 23bc3504..2b4bd850 100644
--- a/src/connectionhandler.h
+++ b/src/connectionhandler.h
@@ -25,6 +25,7 @@
 
 #include "netcomputer.h"
 #include "packet.h"
+#include <map>
 
 /**
  * This class represents the connection handler interface. The connection
@@ -56,6 +57,14 @@ class ConnectionHandler
          * Called when a computer sends a packet to the network session.
          */
         void receivePacket(NetComputer *computer, Packet *packet);
+
+        /**
+         * Registers a message handler to handle a certain message type.
+         */
+        void registerHandler(unsigned int msgId, MessageHandler *handler);
+
+    private:
+        std::map<unsigned int, MessageHandler*> handlers;
 };
 
 #endif