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

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

diff --git a/judge/manager/ai_controller.h b/judge/manager/ai_controller.h
index f4682eb..a5c791d 100644 (file)
--- a/judge/manager/ai_controller.h
+++ b/judge/manager/ai_controller.h
@@ -19,7 +19,11 @@ class AI_Controller : public Controller, private Program
                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);
 

-               virtual void Message(const char * message) {Program::SendMessage(message);}
+               virtual bool Message(const char * message) 
+               {
+                       //fprintf(stderr, "Sending message \"%s\" to AI program...\n", message);
+                       return Program::SendMessage(message);
+               }

                virtual void Pause() {Program::Pause();} //Hack wrapper
                virtual void Continue() {Program::Continue();} //Hack wrapper
 
                virtual void Pause() {Program::Pause();} //Hack wrapper
                virtual void Continue() {Program::Continue();} //Hack wrapper