[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 0a4baeb..a5c791d 100644 (file)
--- a/judge/manager/ai_controller.h
+++ b/judge/manager/ai_controller.h
@@ -19,10 +19,10 @@ class AI_Controller : public Controller, private Program
                virtual MovementResult QuerySetup(const char * opponentName,std::string setup[]);
                virtual MovementResult QueryMove(std::string & buffer);
 
-               virtual void Message(const char * message) 
+               virtual bool Message(const char * message) 
                {
                        //fprintf(stderr, "Sending message \"%s\" to AI program...\n", message);
-                       Program::SendMessage(message);
+                       return Program::SendMessage(message);
                }
                virtual void Pause() {Program::Pause();} //Hack wrapper
                virtual void Continue() {Program::Continue();} //Hack wrapper