[PATCH] Handle case where an AI sends an invalid message and then crashes

[progcomp2012.git] / judge / manager / network_controller.h

diff --git a/judge/manager/network_controller.h b/judge/manager/network_controller.h
index 31b0c92..130bede 100644 (file)
--- a/judge/manager/network_controller.h
+++ b/judge/manager/network_controller.h
@@ -27,10 +27,10 @@ class NetworkSender : public NetworkController
 
                virtual bool Valid() const {return NetworkController::Valid() && controller->Valid();}
 
 
                virtual bool Valid() const {return NetworkController::Valid() && controller->Valid();}
 

-               virtual void Message(const char * message) 
+               virtual bool Message(const char * message) 

                {
                        //fprintf(stderr,"NetworkSender sending message %s to underlying controller\n", message);
                {
                        //fprintf(stderr,"NetworkSender sending message %s to underlying controller\n", message);

-                       controller->Message(message);
+                       return (controller->Message(message));

                }
 
                virtual MovementResult QuerySetup(const char * opponentName, std::string setup[]);
                }
 
                virtual MovementResult QuerySetup(const char * opponentName, std::string setup[]);


@@ -47,7 +47,7 @@ class NetworkReceiver : public NetworkController
                NetworkReceiver(const Piece::Colour & colour, Network * newNetwork) : NetworkController(colour, newNetwork) {}
                virtual ~NetworkReceiver() {}
 
                NetworkReceiver(const Piece::Colour & colour, Network * newNetwork) : NetworkController(colour, newNetwork) {}
                virtual ~NetworkReceiver() {}
 

-               virtual void Message(const char * message) {} //Do nothing (Counter intuitive much)
+               virtual bool Message(const char * message) {return true;} //Do nothing (Counter intuitive much)

                virtual MovementResult QuerySetup(const char * opponentName, std::string setup[]);
                virtual MovementResult QueryMove(std::string & buffer);
                
                virtual MovementResult QuerySetup(const char * opponentName, std::string setup[]);
                virtual MovementResult QueryMove(std::string & buffer);