Пример #1
0
    def __init__(self, conf, init=False):
        self.bot_server = None

        self.servicebot_dispatcher = None
        self.submitbot_dispatcher = None
        self.scoreboardbot_dispatcher = None

        self.conf = conf
        self.servicebot_conf = conf.getSection("SERVICE_BOT")
        self.logger = conf.buildLogger("StandardCTF")

        self.scoring = BasicScoring(conf)
        self.service_status = None
Пример #2
0
	def __init__(self,conf,init=False):
		self.bot_server = None

		self.servicebot_dispatcher = None
		self.submitbot_dispatcher = None
		self.scoreboardbot_dispatcher = None

		self.conf = conf
		self.servicebot_conf = conf.getSection("SERVICE_BOT")
		self.logger = conf.buildLogger("StandardCTF")

		self.scoring = BasicScoring(conf)
		self.service_status = None
Пример #3
0
class StandardCTF(GameStateLogic):
    def __init__(self, conf, init=False):
        self.bot_server = None

        self.servicebot_dispatcher = None
        self.submitbot_dispatcher = None
        self.scoreboardbot_dispatcher = None

        self.conf = conf
        self.servicebot_conf = conf.getSection("SERVICE_BOT")
        self.logger = conf.buildLogger("StandardCTF")

        self.scoring = BasicScoring(conf)
        self.service_status = None

    def setup(self, bot_server):
        self.bot_server = bot_server

    def handleBotMessage(self, msg, dispatcher):

        if (msg.type == "CLIENT_HELLO"):
            self.__handleNewClient(msg.data, dispatcher)

        elif (msg.type == "SERVICE_RESULTS"):
            self.__handleServiceResults(msg.data)

        elif (msg.type == "COLLECT_FLAGS_RESULT"):
            self.__handleSubmitResults(msg.data)

        else:
            self.logger.error("Unknown message: %r %r" % (msg.type, msg.data))

    def __handleNewClient(self, name, dispatcher):
        self.logger.info("New client: %s", name)

        if (name == "SERVICE_BOT"):
            if (self.servicebot_dispatcher == None):
                self.servicebot_dispatcher = dispatcher
            else:
                self.logger.error("Service Bot tried to connect again!")
                return

        if (name == "SUBMIT_BOT"):
            if (self.submitbot_dispatcher == None):
                self.submitbot_dispatcher = dispatcher
            else:
                self.logger.error("Submit Bot tried to connect again!")
                return

        if (name == "SCOREBOARD_BOT"):
            if (self.scoreboardbot_dispatcher == None):
                self.scoreboardbot_dispatcher = dispatcher
            else:
                self.logger.error("Scoreboard Bot tried to connect again!")
                return

        if (self.__checkStart() == True):
            self.__startGame()

    def __checkStart(self):

        if (self.servicebot_dispatcher == None):
            return False

        if (self.submitbot_dispatcher == None):
            return False

        if (self.scoreboardbot_dispatcher == None):
            return False

        return True

    def __startGame(self):
        init_game = True

        if (init_game):
            self.round = 1

        execute_msg = BotMessage("EXECUTE_ROUND", self.round)
        self.servicebot_dispatcher.sendMsg(execute_msg)

    def __handleServiceResults(self, service_results):
        round, data = service_results
        assert (self.round == round)
        self.service_status = data
        self.scoring.updateDefensiveInfo(data)
        self.submitbot_dispatcher.sendMsg(BotMessage("COLLECT_FLAGS", None))

    def __handleSubmitResults(self, submit_results):
        self.scoring.updateOffensiveInfo(submit_results)
        self.__endRound()

    def __endRound(self):
        self.logger.debug("Ending round %d" % self.round)
        team_off_scores, team_def_scores = self.scoring.updateRoundScores(
            self.round)
        self.scoreboardbot_dispatcher.sendMsg(
            BotMessage(
                "UPDATE_SCORES",
                (team_off_scores, team_def_scores, self.service_status)))

        self.round += 1
        execute_msg = BotMessage("EXECUTE_ROUND", self.round)
        self.servicebot_dispatcher.sendMsg(execute_msg)
Пример #4
0
class StandardCTF(GameStateLogic):

	def __init__(self,conf,init=False):
		self.bot_server = None

		self.servicebot_dispatcher = None
		self.submitbot_dispatcher = None
		self.scoreboardbot_dispatcher = None
		self.staticflagbot_dispatcher = None

		self.conf = conf
		self.servicebot_conf = conf.getSection("SERVICE_BOT")
		self.logger = conf.buildLogger("StandardCTF")

		self.scoring = BasicScoring(conf)
		self.service_status = None

	def setup(self,bot_server):
		self.bot_server = bot_server

	def handleBotMessage(self,msg,dispatcher):
		self.logger.info("Bot Message:%s"%(msg.type))
		if(msg.type == "CLIENT_HELLO"):
			self.__handleNewClient(msg.data,dispatcher)

		elif(msg.type == "SERVICE_RESULTS"):
			self.__handleServiceResults(msg.data)

		elif(msg.type == "COLLECT_FLAGS_RESULT"):
			self.__handleSubmitResults(msg.data)
			
		elif(msg.type == "COLLECT_STATIC_FLAGS_RESULT"):
			self.__handleStaticFlagResults(msg.data)

		else:
			self.logger.error("Unknown message: %r %r" % (msg.type,msg.data))

	def __handleNewClient(self, name, dispatcher):
		self.logger.info("New client: %s",name)

		if(name == "SERVICE_BOT"):
			if(self.servicebot_dispatcher == None):
				self.servicebot_dispatcher = dispatcher
			else:
				self.logger.error("Service Bot tried to connect again!")
				return

		if(name == "SUBMIT_BOT"):
			if(self.submitbot_dispatcher == None):
				self.submitbot_dispatcher = dispatcher
			else:
				self.logger.error("Submit Bot tried to connect again!")
				return
			
		if(name == "STATICFLAG_BOT"):
			if(self.staticflagbot_dispatcher == None):
				self.staticflagbot_dispatcher = dispatcher
			else:
				self.logger.error("Static Flag Bot tried to connect again!")
				return
		
		if(name == "SCOREBOARD_BOT"):
			if(self.scoreboardbot_dispatcher == None):
				self.scoreboardbot_dispatcher = dispatcher
			else:
				self.logger.error("Scoreboard Bot tried to connect again!")
				return

		if(self.__checkStart() == True):
			self.__startGame()

	def __checkStart(self):

		if(self.servicebot_dispatcher == None):
			return False
		
		if(self.submitbot_dispatcher == None):
			return False
		
		if(self.staticflagbot_dispatcher == None):
			return False

		if(self.scoreboardbot_dispatcher == None):
			return False

		return True

	def __startGame(self):
		init_game = True

		if(init_game):
			self.round = 1

		execute_msg = BotMessage("EXECUTE_ROUND",self.round)
		self.servicebot_dispatcher.sendMsg(execute_msg)

	def __handleServiceResults(self,service_results):
		round, data = service_results
		assert(self.round == round)
		self.service_status = data
		self.scoring.updateDefensiveInfo(data)
		#round ending so send msg to the submitbot to pass back flags
		self.submitbot_dispatcher.sendMsg(BotMessage("COLLECT_FLAGS",None))
		self.staticflagbot_dispatcher.sendMsg(BotMessage("COLLECT_STATIC_FLAGS",None))

	def __handleSubmitResults(self,submit_results):
		#deals wit the "COLLECT_FLAGS_RESULT"
		self.scoring.updateOffensiveInfo(submit_results)
		#self.__endRound()
		
	def __handleStaticFlagResults(self,submit_results):
		#deals wit the "COLLECT_STATIC_FLAGS_RESULT"
		self.scoring.updateStaticFlagInfo(submit_results)
		self.__endRound()

	def __endRound(self):
		self.logger.debug("Ending round %d" % self.round)
		team_off_scores,team_def_scores, team_egg_scores = self.scoring.updateRoundScores(self.round)
		self.scoreboardbot_dispatcher.sendMsg(BotMessage(
			"UPDATE_SCORES",(team_off_scores,team_def_scores,team_egg_scores, self.round, self.service_status)))

		self.round += 1
		execute_msg = BotMessage("EXECUTE_ROUND",self.round)
		self.servicebot_dispatcher.sendMsg(execute_msg)