def __init__(self, db_col, cfg_file="scorebotcfg.json", obj_id=None):
threading.Thread.__init__(self)
self.col = db_col
self.flag_queue = Queue.Queue()
self.flag_answer_queue = Queue.Queue()
self.msg_queue = Queue.Queue()
self.logger_obj = Logger("scorebot")
self.pp = pprint.PrettyPrinter(indent=2)
if obj_id:
bson_id = ObjectId(obj_id)
self.obj_id = bson_id
json_obj = self.col.find_one(self.obj_id)
if globalvars.debug:
self.pp.pprint(json_obj)
game = jaraco.modb.decode(decode_dict(json_obj))
if "config" in game:
self.flag_store = FlagStore(self.logger_obj, self.flag_queue, self.flag_answer_queue)
self.json_cfg_obj = JsonConfig(cfg_file, self.flag_store)
(self.blue_teams, self.injects) = self.json_cfg_obj.process(game["config"])
else:
raise Exception("No config section in restore object!")
self.message_store = MessageStore(self.logger_obj, self.msg_queue)
self.flag_server = FlagServer(self.logger_obj, self.flag_queue, self.flag_answer_queue, self.msg_queue)
#self.myscoreboard = Scoreboard(self.blue_teams, self.flag_store, self.message_store)
for team in self.blue_teams.keys():
self.blue_teams[team].add_queue(self.flag_queue)
if "scores" in game:
scores = game["scores"]
if "blueteams" in scores:
for team in self.blue_teams:
if team in scores["blueteams"]:
blue_score = scores["blueteams"][team]
self.blue_teams[team].set_scores(blue_score)
else:
raise Exception("Missing scores for team %s in restore object!" % team)
else:
raise Exception("Missing blueteams score block in restore object!" % team)
else:
raise Exception("Missing scores block in restore object!" % team)
if "flags" in game:
all_flags = game["flags"]
self.flag_store.restore(all_flags)
else:
self.obj_id = obj_id
self.flag_store = FlagStore(self.logger_obj, self.flag_queue, self.flag_answer_queue)
self.json_cfg_obj = JsonConfig(cfg_file, self.flag_store)
(self.blue_teams, self.injects) = self.json_cfg_obj.process()
self.message_store = MessageStore(self.logger_obj, self.msg_queue)
self.flag_server = FlagServer(self.logger_obj, self.flag_queue, self.flag_answer_queue, self.msg_queue)
#self.myscoreboard = Scoreboard(self.blue_teams, self.flag_store, self.message_store)
for team in self.blue_teams.keys():
self.blue_teams[team].add_queue(self.flag_queue)
self.movies = Movies(self.logger_obj)
#self.movies.set_movie()
self.bottle_server = BottleServer(self.blue_teams, self.flag_store, self.message_store, self.movies, '0.0.0.0', 8090)
def __init__(self, db_col, cfg_file="scorebotcfg.json", obj_id=None):
threading.Thread.__init__(self)
self.col = db_col
self.flag_queue = Queue.Queue()
self.flag_answer_queue = Queue.Queue()
self.msg_queue = Queue.Queue()
self.logger_obj = Logger("scorebot")
self.oqueue = Queue.Queue()
self.equeue = Queue.Queue()
self.logger = ThreadedLogger("ctfgame.log", self.oqueue, self.equeue)
self.logger.start()
self.pp = pprint.PrettyPrinter(indent=2)
self.blues_queues = {}
self.teams_rounds = {}
# todo - fix why this isn't actually the first round!
self.this_round = 1
self.interval = 120
self.inround = False
self.go_time = time.time()
self.save_time = time.time()
self.save_interval = 10
if obj_id:
bson_id = ObjectId(obj_id)
self.obj_id = bson_id
json_obj = self.col.find_one(self.obj_id)
if globalvars.debug:
self.pp.pprint(json_obj)
game = jaraco.modb.decode(decode_dict(json_obj))
if "config" in game:
self.flag_store = FlagStore(self.logger_obj, self.flag_queue, self.flag_answer_queue)
self.json_cfg_obj = JsonConfig(cfg_file, self.flag_store)
(self.blue_teams, self.injects) = self.json_cfg_obj.process(game["config"])
else:
raise Exception("No config section in restore object!")
self.message_store = MessageStore(self.logger_obj, self.msg_queue)
self.flag_server = FlagServer(self.logger_obj, self.flag_queue, self.flag_answer_queue, self.msg_queue)
for team in self.blue_teams.keys():
self.blue_teams[team].add_queue(self.flag_queue)
if "scores" in game:
scores = game["scores"]
if "blueteams" in scores:
for team in self.blue_teams:
if team in scores["blueteams"]:
blue_score = scores["blueteams"][team]
self.blue_teams[team].set_scores(blue_score)
else:
raise Exception("Missing scores for team %s in restore object!" % team)
else:
raise Exception("Missing blueteams score block in restore object!" % team)
else:
raise Exception("Missing scores block in restore object!" % team)
if "flags" in game:
all_flags = game["flags"]
self.flag_store.restore(all_flags)
else:
self.obj_id = obj_id
self.flag_store = FlagStore(self.logger_obj, self.flag_queue, self.flag_answer_queue)
self.json_cfg_obj = JsonConfig(cfg_file, self.flag_store)
(self.blue_teams, self.injects) = self.json_cfg_obj.process()
self.message_store = MessageStore(self.logger_obj, self.msg_queue)
self.flag_server = FlagServer(self.logger_obj, self.flag_queue, self.flag_answer_queue, self.msg_queue)
for team in self.blue_teams.keys():
self.blue_teams[team].add_queue(self.flag_queue)
blue_queue = Queue.Queue()
self.blues_queues[team] = blue_queue
self.blue_teams[team].set_queue(blue_queue)
self.teams_rounds[team] = True
self.movies = Movies(self.logger_obj)
#self.movies.set_movie()
self.bottle_server = BottleServer(self.blue_teams, self.flag_store, self.message_store, self.movies, '0.0.0.0', 8090)
class CTFgame(threading.Thread):
def __init__(self, db_col, cfg_file="scorebotcfg.json", obj_id=None):
threading.Thread.__init__(self)
self.col = db_col
self.flag_queue = Queue.Queue()
self.flag_answer_queue = Queue.Queue()
self.msg_queue = Queue.Queue()
self.logger_obj = Logger("scorebot")
self.oqueue = Queue.Queue()
self.equeue = Queue.Queue()
self.logger = ThreadedLogger("ctfgame.log", self.oqueue, self.equeue)
self.logger.start()
self.pp = pprint.PrettyPrinter(indent=2)
self.blues_queues = {}
self.teams_rounds = {}
# todo - fix why this isn't actually the first round!
self.this_round = 1
self.interval = 120
self.inround = False
self.go_time = time.time()
self.save_time = time.time()
self.save_interval = 10
if obj_id:
bson_id = ObjectId(obj_id)
self.obj_id = bson_id
json_obj = self.col.find_one(self.obj_id)
if globalvars.debug:
self.pp.pprint(json_obj)
game = jaraco.modb.decode(decode_dict(json_obj))
if "config" in game:
self.flag_store = FlagStore(self.logger_obj, self.flag_queue, self.flag_answer_queue)
self.json_cfg_obj = JsonConfig(cfg_file, self.flag_store)
(self.blue_teams, self.injects) = self.json_cfg_obj.process(game["config"])
else:
raise Exception("No config section in restore object!")
self.message_store = MessageStore(self.logger_obj, self.msg_queue)
self.flag_server = FlagServer(self.logger_obj, self.flag_queue, self.flag_answer_queue, self.msg_queue)
for team in self.blue_teams.keys():
self.blue_teams[team].add_queue(self.flag_queue)
if "scores" in game:
scores = game["scores"]
if "blueteams" in scores:
for team in self.blue_teams:
if team in scores["blueteams"]:
blue_score = scores["blueteams"][team]
self.blue_teams[team].set_scores(blue_score)
else:
raise Exception("Missing scores for team %s in restore object!" % team)
else:
raise Exception("Missing blueteams score block in restore object!" % team)
else:
raise Exception("Missing scores block in restore object!" % team)
if "flags" in game:
all_flags = game["flags"]
self.flag_store.restore(all_flags)
else:
self.obj_id = obj_id
self.flag_store = FlagStore(self.logger_obj, self.flag_queue, self.flag_answer_queue)
self.json_cfg_obj = JsonConfig(cfg_file, self.flag_store)
(self.blue_teams, self.injects) = self.json_cfg_obj.process()
self.message_store = MessageStore(self.logger_obj, self.msg_queue)
self.flag_server = FlagServer(self.logger_obj, self.flag_queue, self.flag_answer_queue, self.msg_queue)
for team in self.blue_teams.keys():
self.blue_teams[team].add_queue(self.flag_queue)
blue_queue = Queue.Queue()
self.blues_queues[team] = blue_queue
self.blue_teams[team].set_queue(blue_queue)
self.teams_rounds[team] = True
self.movies = Movies(self.logger_obj)
#self.movies.set_movie()
self.bottle_server = BottleServer(self.blue_teams, self.flag_store, self.message_store, self.movies, '0.0.0.0', 8090)
def run(self):
pp = pprint.PrettyPrinter(indent=2)
self.inround = False
while True:
now = time.time()
if self.save_time <= now:
json_obj = self.build_json()
if self.obj_id:
json_obj["_id"] = self.obj_id
if globalvars.debug:
pp.pprint(json_obj)
self.obj_id = self.col.save(json_obj)
self.oqueue.put("Saved Game ID %s" % self.obj_id)
self.save_time += self.save_interval
for team in self.blues_queues:
try:
item = self.blues_queues[team].get(False)
if item == "Done":
#process the host message
if team in self.teams_rounds:
#print "Found message from team %s" % team
self.teams_rounds[team] = True
else:
raise Exception("Unknown team %s\n" % team)
else:
self.blues_queues[team].put(item)
except Queue.Empty:
pass
except:
traceback.print_exc(file=self.logger_obj)
score_round = True
# Check to see if all hosts have finished the last round
donemsg = ""
finished = []
not_finished = []
for team in self.teams_rounds:
if self.teams_rounds[team]:
donemsg += "%s, " % team
finished.append(team)
else:
score_round = False
not_finished.append(team)
if donemsg:
failmsg = "Failed: "
for team in self.teams_rounds:
if not self.teams_rounds[team]:
failmsg += team
#sys.stdout.write("Done: %s\n" % donemsg)
#sys.stdout.write(failmsg)
if score_round:
self.inround = False
for team in self.teams_rounds:
self.teams_rounds[team] = False
now = time.time()
statfile = open("scorebot.status", "w")
statfile.write("Round %s, teams %s not finished\n" % (self.this_round, ",".join(not_finished)))
statfile.write("Round %s, teams %s finished\n" % (self.this_round, ", ".join(finished)))
statfile.write("Go time: %s\nNow time: %s\n" % (self.go_time, now))
statfile.close()
if self.go_time <= now and not self.inround:
try:
# Report times so that we know whether or not the last round ran too long
for team in self.blues_queues:
self.oqueue.put("Starting Service check for Blueteam %s. Go time was %s, now is %s." % (team, self.go_time, now))
self.blues_queues[team].put(["Go", self.this_round], 1)
self.this_round += 1
self.go_time += self.interval
self.oqueue.put("New go time is %s" % self.go_time)
self.inround = True
except:
traceback.print_exc(file=self.logger_obj)
else:
time.sleep(1)
def build_json(self):
# master hash for the whole thing
game = {}
# Get the config json
json_config = self.json_cfg_obj.get_json()
game["config"] = json_config
# build the hash for the service scores
scores = {}
blueteam_scores = {}
for blueteam in self.blue_teams:
blue_score = self.blue_teams[blueteam].get_scores()
if blue_score:
blueteam_scores[blueteam] = blue_score
scores["blueteams"] = blueteam_scores
# Add the scores json to the game level json
game["scores"] = scores
# Get the flags json
flags = self.flag_store.get_all_flags()
#Add the flags to the game hash
game["flags"] = flags
#Convert the master hash to json
json_obj = jaraco.modb.encode(game)
return json_obj
def start_game(self):
t = threading.Thread(target=self.flag_server.serve_forever)
t.start()
# Get this party started!
#self.myscoreboard.start()
for team in self.blue_teams.keys():
self.blue_teams[team].start()
self.flag_store.start()
self.message_store.start()
self.injects.start()
self.bottle_server.start()
elif o in ("-b", "--binjitsu"):
globalvars.binjitsu = True
elif o in ("-v", "--verbose"):
globalvars.verbose = True
elif o in ("-q", "--quick"):
globalvars.quick = True
elif o in ("-q", "--quick"):
globalvars.quick = True
elif o in ("-n", "--nomovie"):
globalvars.nomovie = True
else:
assert False, "unhandled option"
flag_queue_obj = Queue.Queue()
message_queue_obj = Queue.Queue()
logger_obj = Logger("scorebot")
flag_store = FlagStore(logger_obj, flag_queue_obj)
message_store = MessageStore(logger_obj, message_queue_obj)
flag_server = FlagServer(logger_obj, flag_queue_obj, message_queue_obj)
t = threading.Thread(target=flag_server.serve_forever)
t.start()
blue_teams = read_config(cfg_file, flag_store)
myscoreboard = Scoreboard(blue_teams, flag_store, message_store)
myscoreboard.start()
for team in blue_teams.keys():
blue_teams[team].add_queue(flag_queue_obj)
blue_teams[team].start()
flag_store.start()
message_store.start()
injects.start()
if __name__ == "__main__":
def proc_flags(self, blue_obj, flag_name, flag):
if "value" in flag:
value = flag["value"]
else:
raise Exception("No value given for flag %s!" % flag_name)
if "answer" in flag:
this_answer = flag["answer"]
else:
raise Exception("No answer given for flag %s!" % flag_name)
if "score" in flag:
this_score = flag["score"]
blue_obj.add_flag(flag_name,
value,
score=this_score,
answer=this_answer)
else:
blue_obj.add_flag(flag_name, value, answer=this_answer)
if __name__ == "__main__":
from FlagStore import FlagStore
from Logger import Logger
from Queue import Queue
queue_obj = Queue()
logger_obj = Logger("JsonConfigTesT")
flag_store = FlagStore(logger_obj, queue_obj, "test")
json_obj = JsonConfig("scorebotcfg.json", flag_store)
(blueteams, injects) = json_obj.process()
for blueteam in blueteams:
print blueteam.get_teamname()
def __init__(self, db_col, cfg_file="scorebotcfg.json", obj_id=None):
threading.Thread.__init__(self)
self.col = db_col
self.flag_queue = Queue.Queue()
self.flag_answer_queue = Queue.Queue()
self.msg_queue = Queue.Queue()
self.logger_obj = Logger("scorebot")
self.oqueue = Queue.Queue()
self.equeue = Queue.Queue()
self.logger = ThreadedLogger("ctfgame.log", self.oqueue, self.equeue)
self.logger.start()
self.pp = pprint.PrettyPrinter(indent=2)
self.blues_queues = {}
self.teams_rounds = {}
# todo - fix why this isn't actually the first round!
self.this_round = 1
self.interval = 120
self.inround = False
self.go_time = time.time()
self.save_time = time.time()
self.save_interval = 10
if obj_id:
bson_id = ObjectId(obj_id)
self.obj_id = bson_id
json_obj = self.col.find_one(self.obj_id)
if globalvars.debug:
self.pp.pprint(json_obj)
game = jaraco.modb.decode(decode_dict(json_obj))
if "config" in game:
self.flag_store = FlagStore(self.logger_obj, self.flag_queue,
self.flag_answer_queue)
self.json_cfg_obj = JsonConfig(cfg_file, self.flag_store)
(self.blue_teams,
self.injects) = self.json_cfg_obj.process(game["config"])
else:
raise Exception("No config section in restore object!")
self.message_store = MessageStore(self.logger_obj, self.msg_queue)
self.flag_server = FlagServer(self.logger_obj, self.flag_queue,
self.flag_answer_queue,
self.msg_queue)
for team in self.blue_teams.keys():
self.blue_teams[team].add_queue(self.flag_queue)
if "scores" in game:
scores = game["scores"]
if "blueteams" in scores:
for team in self.blue_teams:
if team in scores["blueteams"]:
blue_score = scores["blueteams"][team]
self.blue_teams[team].set_scores(blue_score)
else:
raise Exception(
"Missing scores for team %s in restore object!"
% team)
else:
raise Exception(
"Missing blueteams score block in restore object!" %
team)
else:
raise Exception("Missing scores block in restore object!" %
team)
if "flags" in game:
all_flags = game["flags"]
self.flag_store.restore(all_flags)
else:
self.obj_id = obj_id
self.flag_store = FlagStore(self.logger_obj, self.flag_queue,
self.flag_answer_queue)
self.json_cfg_obj = JsonConfig(cfg_file, self.flag_store)
(self.blue_teams, self.injects) = self.json_cfg_obj.process()
self.message_store = MessageStore(self.logger_obj, self.msg_queue)
self.flag_server = FlagServer(self.logger_obj, self.flag_queue,
self.flag_answer_queue,
self.msg_queue)
for team in self.blue_teams.keys():
self.blue_teams[team].add_queue(self.flag_queue)
blue_queue = Queue.Queue()
self.blues_queues[team] = blue_queue
self.blue_teams[team].set_queue(blue_queue)
self.teams_rounds[team] = True
self.movies = Movies(self.logger_obj)
#self.movies.set_movie()
self.bottle_server = BottleServer(self.blue_teams, self.flag_store,
self.message_store, self.movies,
'0.0.0.0', 8090)
class CTFgame(threading.Thread):
def __init__(self, db_col, cfg_file="scorebotcfg.json", obj_id=None):
threading.Thread.__init__(self)
self.col = db_col
self.flag_queue = Queue.Queue()
self.flag_answer_queue = Queue.Queue()
self.msg_queue = Queue.Queue()
self.logger_obj = Logger("scorebot")
self.oqueue = Queue.Queue()
self.equeue = Queue.Queue()
self.logger = ThreadedLogger("ctfgame.log", self.oqueue, self.equeue)
self.logger.start()
self.pp = pprint.PrettyPrinter(indent=2)
self.blues_queues = {}
self.teams_rounds = {}
# todo - fix why this isn't actually the first round!
self.this_round = 1
self.interval = 120
self.inround = False
self.go_time = time.time()
self.save_time = time.time()
self.save_interval = 10
if obj_id:
bson_id = ObjectId(obj_id)
self.obj_id = bson_id
json_obj = self.col.find_one(self.obj_id)
if globalvars.debug:
self.pp.pprint(json_obj)
game = jaraco.modb.decode(decode_dict(json_obj))
if "config" in game:
self.flag_store = FlagStore(self.logger_obj, self.flag_queue,
self.flag_answer_queue)
self.json_cfg_obj = JsonConfig(cfg_file, self.flag_store)
(self.blue_teams,
self.injects) = self.json_cfg_obj.process(game["config"])
else:
raise Exception("No config section in restore object!")
self.message_store = MessageStore(self.logger_obj, self.msg_queue)
self.flag_server = FlagServer(self.logger_obj, self.flag_queue,
self.flag_answer_queue,
self.msg_queue)
for team in self.blue_teams.keys():
self.blue_teams[team].add_queue(self.flag_queue)
if "scores" in game:
scores = game["scores"]
if "blueteams" in scores:
for team in self.blue_teams:
if team in scores["blueteams"]:
blue_score = scores["blueteams"][team]
self.blue_teams[team].set_scores(blue_score)
else:
raise Exception(
"Missing scores for team %s in restore object!"
% team)
else:
raise Exception(
"Missing blueteams score block in restore object!" %
team)
else:
raise Exception("Missing scores block in restore object!" %
team)
if "flags" in game:
all_flags = game["flags"]
self.flag_store.restore(all_flags)
else:
self.obj_id = obj_id
self.flag_store = FlagStore(self.logger_obj, self.flag_queue,
self.flag_answer_queue)
self.json_cfg_obj = JsonConfig(cfg_file, self.flag_store)
(self.blue_teams, self.injects) = self.json_cfg_obj.process()
self.message_store = MessageStore(self.logger_obj, self.msg_queue)
self.flag_server = FlagServer(self.logger_obj, self.flag_queue,
self.flag_answer_queue,
self.msg_queue)
for team in self.blue_teams.keys():
self.blue_teams[team].add_queue(self.flag_queue)
blue_queue = Queue.Queue()
self.blues_queues[team] = blue_queue
self.blue_teams[team].set_queue(blue_queue)
self.teams_rounds[team] = True
self.movies = Movies(self.logger_obj)
#self.movies.set_movie()
self.bottle_server = BottleServer(self.blue_teams, self.flag_store,
self.message_store, self.movies,
'0.0.0.0', 8090)
def run(self):
pp = pprint.PrettyPrinter(indent=2)
self.inround = False
while True:
now = time.time()
if self.save_time <= now:
json_obj = self.build_json()
if self.obj_id:
json_obj["_id"] = self.obj_id
if globalvars.debug:
pp.pprint(json_obj)
self.obj_id = self.col.save(json_obj)
self.oqueue.put("Saved Game ID %s" % self.obj_id)
self.save_time += self.save_interval
for team in self.blues_queues:
try:
item = self.blues_queues[team].get(False)
if item == "Done":
#process the host message
if team in self.teams_rounds:
#print "Found message from team %s" % team
self.teams_rounds[team] = True
else:
raise Exception("Unknown team %s\n" % team)
else:
self.blues_queues[team].put(item)
except Queue.Empty:
pass
except:
traceback.print_exc(file=self.logger_obj)
score_round = True
# Check to see if all hosts have finished the last round
donemsg = ""
finished = []
not_finished = []
for team in self.teams_rounds:
if self.teams_rounds[team]:
donemsg += "%s, " % team
finished.append(team)
else:
score_round = False
not_finished.append(team)
if donemsg:
failmsg = "Failed: "
for team in self.teams_rounds:
if not self.teams_rounds[team]:
failmsg += team
#sys.stdout.write("Done: %s\n" % donemsg)
#sys.stdout.write(failmsg)
if score_round:
self.inround = False
for team in self.teams_rounds:
self.teams_rounds[team] = False
now = time.time()
statfile = open("scorebot.status", "w")
statfile.write("Round %s, teams %s not finished\n" %
(self.this_round, ",".join(not_finished)))
statfile.write("Round %s, teams %s finished\n" %
(self.this_round, ", ".join(finished)))
statfile.write("Go time: %s\nNow time: %s\n" %
(self.go_time, now))
statfile.close()
if self.go_time <= now and not self.inround:
try:
# Report times so that we know whether or not the last round ran too long
for team in self.blues_queues:
self.oqueue.put(
"Starting Service check for Blueteam %s. Go time was %s, now is %s."
% (team, self.go_time, now))
self.blues_queues[team].put(["Go", self.this_round], 1)
self.this_round += 1
self.go_time += self.interval
self.oqueue.put("New go time is %s" % self.go_time)
self.inround = True
except:
traceback.print_exc(file=self.logger_obj)
else:
time.sleep(1)
def build_json(self):
# master hash for the whole thing
game = {}
# Get the config json
json_config = self.json_cfg_obj.get_json()
game["config"] = json_config
# build the hash for the service scores
scores = {}
blueteam_scores = {}
for blueteam in self.blue_teams:
blue_score = self.blue_teams[blueteam].get_scores()
if blue_score:
blueteam_scores[blueteam] = blue_score
scores["blueteams"] = blueteam_scores
# Add the scores json to the game level json
game["scores"] = scores
# Get the flags json
flags = self.flag_store.get_all_flags()
#Add the flags to the game hash
game["flags"] = flags
#Convert the master hash to json
json_obj = jaraco.modb.encode(game)
return json_obj
def start_game(self):
t = threading.Thread(target=self.flag_server.serve_forever)
t.start()
# Get this party started!
#self.myscoreboard.start()
for team in self.blue_teams.keys():
self.blue_teams[team].start()
self.flag_store.start()
self.message_store.start()
self.injects.start()
self.bottle_server.start()
elif o in ("-b", "--binjitsu"):
globalvars.binjitsu = True
elif o in ("-v", "--verbose"):
globalvars.verbose = True
elif o in ("-q", "--quick"):
globalvars.quick = True
elif o in ("-q", "--quick"):
globalvars.quick = True
elif o in ("-n", "--nomovie"):
globalvars.nomovie = True
else:
assert False, "unhandled option"
flag_queue_obj = Queue.Queue()
message_queue_obj = Queue.Queue()
logger_obj = Logger("scorebot")
flag_store = FlagStore(logger_obj, flag_queue_obj)
message_store = MessageStore(logger_obj, message_queue_obj)
flag_server = FlagServer(logger_obj, flag_queue_obj, message_queue_obj)
t = threading.Thread(target=flag_server.serve_forever)
t.start()
blue_teams = read_config(cfg_file, flag_store)
myscoreboard = Scoreboard(blue_teams, flag_store, message_store)
myscoreboard.start()
for team in blue_teams.keys():
blue_teams[team].add_queue(flag_queue_obj)
blue_teams[team].start()
flag_store.start()
message_store.start()
injects.start()
class CTFgame(threading.Thread):
def __init__(self, db_col, cfg_file="scorebotcfg.json", obj_id=None):
threading.Thread.__init__(self)
self.col = db_col
self.flag_queue = Queue.Queue()
self.flag_answer_queue = Queue.Queue()
self.flag_game_queue = Queue.Queue()
self.flag_servers = {}
self.default_port = 50007
self.msg_queue = Queue.Queue()
self.logger_obj = Logger("scorebot")
self.logger = ThreadedLogger("ctfgame.log")
self.oqueue = self.logger.get_oqueue()
self.equeue = self.logger.get_equeue()
self.logger.start()
self.pp = pprint.PrettyPrinter(indent=2)
self.blues_queues = {}
self.teams_rounds = {}
# todo - fix why this isn't actually the first round!
self.this_round = 1
self.interval = 120
self.inround = False
self.go_time = time.time()
self.save_time = time.time()
self.save_interval = 10
if obj_id:
bson_id = ObjectId(obj_id)
self.obj_id = bson_id
json_obj = self.col.find_one(self.obj_id)
if globalvars.debug:
self.pp.pprint(json_obj)
game = jaraco.modb.decode(decode_dict(json_obj))
if "config" in game:
self.flag_store = FlagStore(self.logger_obj, self.flag_queue, self.flag_answer_queue)
self.json_cfg_obj = JsonConfig(cfg_file, self.flag_store)
(self.blue_teams, self.injects) = self.json_cfg_obj.process(game["config"])
else:
raise Exception("No config section in restore object!")
self.message_store = MessageStore(self.logger_obj, self.msg_queue)
self.flag_servers[self.default_port] = FlagServer(self.logger_obj, self.flag_queue, self.flag_answer_queue, self.msg_queue, self.flag_game_queue, self.default_port)
for team in self.blue_teams.keys():
self.blue_teams[team].add_queue(self.flag_queue)
if "scores" in game:
scores = game["scores"]
if "blueteams" in scores:
for team in self.blue_teams:
if team in scores["blueteams"]:
blue_score = scores["blueteams"][team]
self.blue_teams[team].set_scores(blue_score)
else:
raise Exception("Missing scores for team %s in restore object!" % team)
else:
raise Exception("Missing blueteams score block in restore object!" % team)
else:
raise Exception("Missing scores block in restore object!" % team)
if "flags" in game:
all_flags = game["flags"]
self.flag_store.restore(all_flags)
else:
self.obj_id = obj_id
self.flag_store = FlagStore(self.logger_obj, self.flag_queue, self.flag_answer_queue)
self.json_cfg_obj = JsonConfig(cfg_file, self.flag_store)
(self.blue_teams, self.injects) = self.json_cfg_obj.process()
self.message_store = MessageStore(self.logger_obj, self.msg_queue)
self.flag_servers[self.default_port] = FlagServer(self.logger_obj, self.flag_queue, self.flag_answer_queue, self.msg_queue, self.flag_game_queue, self.default_port)
for team in self.blue_teams.keys():
self.blue_teams[team].add_queue(self.flag_queue)
blue_queue = Queue.Queue()
self.blues_queues[team] = blue_queue
self.blue_teams[team].set_queue(blue_queue)
self.teams_rounds[team] = True
self.movies = Movies(self.logger_obj)
#self.movies.set_movie()
self.bottle_server = BottleServer(self.blue_teams, self.flag_store, self.message_store, self.movies, '0.0.0.0', 8090)
def run(self):
pp = pprint.PrettyPrinter(indent=2)
self.inround = False
while True:
now = time.time()
newservice = None
if self.save_time <= now:
json_obj = self.build_json()
if self.obj_id:
json_obj["_id"] = self.obj_id
if globalvars.debug:
pp.pprint(json_obj)
self.obj_id = self.col.save(json_obj)
self.oqueue.put("Saved Game ID %s\n" % self.obj_id)
self.save_time += self.save_interval
try:
queue_item = self.flag_game_queue.get(False)
if len(queue_item)> 5:
self.flag_game_queue.put(queue_item)
else:
newservice = int(queue_item)
self.oqueue.put("Received new service request for %s, processing...\n" % newservice)
except Queue.Empty, err:
pass
if newservice:
if newservice in self.flag_servers:
self.flag_game_queue.put("Failure, service already open")
elif len(self.flag_servers) > 10:
self.flag_game_queue.put("Failure, too many custom services open (10)")
else:
self.flag_servers[newservice] = FlagServer(self.logger_obj, self.flag_queue, self.flag_answer_queue, self.msg_queue, self.flag_game_queue, newservice)
t = threading.Thread(target=self.flag_servers[newservice].serve_forever)
t.start()
self.flag_game_queue.put("New service started on %s" % newservice)
for team in self.blues_queues:
try:
item = self.blues_queues[team].get(False)
if item == "Done":
#process the host message
if team in self.teams_rounds:
#print "Found message from team %s" % team
self.teams_rounds[team] = True
else:
raise Exception("Unknown team %s\n" % team)
else:
self.blues_queues[team].put(item)
except Queue.Empty:
pass
except:
traceback.print_exc(file=self.logger_obj)
score_round = True
# Check to see if all hosts have finished the last round
donemsg = ""
finished = []
not_finished = []
for team in self.teams_rounds:
if self.teams_rounds[team]:
donemsg += "%s, " % team
finished.append(team)
else:
score_round = False
not_finished.append(team)
if donemsg:
failmsg = "Failed: "
for team in self.teams_rounds:
if not self.teams_rounds[team]:
failmsg += team
#sys.stdout.write("Done: %s\n" % donemsg)
#sys.stdout.write(failmsg)
if score_round:
self.inround = False
for team in self.teams_rounds:
self.teams_rounds[team] = False
now = time.time()
statfile = open("status/scorebot.status", "w")
statfile.write("Round %s, teams %s not finished\n" % (self.this_round, ",".join(not_finished)))
statfile.write("Round %s, teams %s finished\n" % (self.this_round, ", ".join(finished)))
statfile.write("Go time: %s\nNow time: %s\n" % (self.go_time, now))
statfile.close()
if self.go_time <= now and not self.inround:
try:
# Report times so that we know whether or not the last round ran too long
self.oqueue.put("Starting Service check for Blueteam %s. Go time was %s, now is %s.\n" % (team, self.go_time, now))
for team in self.blues_queues:
self.blues_queues[team].put(["Go", self.this_round], 1)
self.this_round += 1
self.go_time += self.interval
self.oqueue.put("New go time is %s\n" % self.go_time)
self.inround = True
except:
class CTFgame(threading.Thread):
def __init__(self, db_col, cfg_file="scorebotcfg.json", obj_id=None):
threading.Thread.__init__(self)
self.col = db_col
self.flag_queue = Queue.Queue()
self.flag_answer_queue = Queue.Queue()
self.msg_queue = Queue.Queue()
self.logger_obj = Logger("scorebot")
self.pp = pprint.PrettyPrinter(indent=2)
if obj_id:
bson_id = ObjectId(obj_id)
self.obj_id = bson_id
json_obj = self.col.find_one(self.obj_id)
if globalvars.debug:
self.pp.pprint(json_obj)
game = jaraco.modb.decode(decode_dict(json_obj))
if "config" in game:
self.flag_store = FlagStore(self.logger_obj, self.flag_queue, self.flag_answer_queue)
self.json_cfg_obj = JsonConfig(cfg_file, self.flag_store)
(self.blue_teams, self.injects) = self.json_cfg_obj.process(game["config"])
else:
raise Exception("No config section in restore object!")
self.message_store = MessageStore(self.logger_obj, self.msg_queue)
self.flag_server = FlagServer(self.logger_obj, self.flag_queue, self.flag_answer_queue, self.msg_queue)
#self.myscoreboard = Scoreboard(self.blue_teams, self.flag_store, self.message_store)
for team in self.blue_teams.keys():
self.blue_teams[team].add_queue(self.flag_queue)
if "scores" in game:
scores = game["scores"]
if "blueteams" in scores:
for team in self.blue_teams:
if team in scores["blueteams"]:
blue_score = scores["blueteams"][team]
self.blue_teams[team].set_scores(blue_score)
else:
raise Exception("Missing scores for team %s in restore object!" % team)
else:
raise Exception("Missing blueteams score block in restore object!" % team)
else:
raise Exception("Missing scores block in restore object!" % team)
if "flags" in game:
all_flags = game["flags"]
self.flag_store.restore(all_flags)
else:
self.obj_id = obj_id
self.flag_store = FlagStore(self.logger_obj, self.flag_queue, self.flag_answer_queue)
self.json_cfg_obj = JsonConfig(cfg_file, self.flag_store)
(self.blue_teams, self.injects) = self.json_cfg_obj.process()
self.message_store = MessageStore(self.logger_obj, self.msg_queue)
self.flag_server = FlagServer(self.logger_obj, self.flag_queue, self.flag_answer_queue, self.msg_queue)
#self.myscoreboard = Scoreboard(self.blue_teams, self.flag_store, self.message_store)
for team in self.blue_teams.keys():
self.blue_teams[team].add_queue(self.flag_queue)
self.movies = Movies(self.logger_obj)
#self.movies.set_movie()
self.bottle_server = BottleServer(self.blue_teams, self.flag_store, self.message_store, self.movies, '0.0.0.0', 8090)
def run(self):
pp = pprint.PrettyPrinter(indent=2)
runtime = 0
while True:
json_obj = self.build_json()
if self.obj_id:
json_obj["_id"] = self.obj_id
if globalvars.debug:
pp.pprint(json_obj)
self.obj_id = self.col.save(json_obj)
print "Saved Game ID %s" % self.obj_id
time.sleep(10)
def build_json(self):
# master hash for the whole thing
game = {}
# Get the config json
json_config = self.json_cfg_obj.get_json()
game["config"] = json_config
# build the hash for the service scores
scores = {}
blueteam_scores = {}
for blueteam in self.blue_teams:
blue_score = self.blue_teams[blueteam].get_scores()
if blue_score:
blueteam_scores[blueteam] = blue_score
scores["blueteams"] = blueteam_scores
# Add the scores json to the game level json
game["scores"] = scores
# Get the flags json
flags = self.flag_store.get_all_flags()
#Add the flags to the game hash
game["flags"] = flags
#Convert the master hash to json
json_obj = jaraco.modb.encode(game)
return json_obj
def start_game(self):
t = threading.Thread(target=self.flag_server.serve_forever)
t.start()
# Get this party started!
#self.myscoreboard.start()
for team in self.blue_teams.keys():
self.blue_teams[team].start()
self.flag_store.start()
self.message_store.start()
self.injects.start()
self.bottle_server.start()
