def testCookie(self):
conf = Config()
servicebot_conf = ServiceBotConfig()
servicebot_conf.min_duration_seconds = 2
servicebot_conf.max_duration_seconds = 2
servicebot_conf.addServiceInfo(
"Service",
modulePath()+"/testservices/StoreService.py",
1,
1,1)
conf.addSection("SERVICE_BOT",servicebot_conf)
conf.addTeamInfo("Team1","127.0.0.1","127.0.0.0/24")
conf.addTeamInfo("Team1","127.0.0.1","127.0.0.0/24")
gamestate = GameStateServerInfo(
"localhost",
4242,
"0123456789012345",
"ABCDEFGH")
conf.setGameStateServerInfo(gamestate)
scheduler = ServiceTaskScheduler(conf,True)
scheduler.execute(0)
results = scheduler.execute(1)
for team,service,task in results:
self.assertEquals(task.cookie(),"Worked")
scheduler.quit()
def testStore(self):
conf = Config()
servicebot_conf = ServiceBotConfig()
servicebot_conf.min_duration_seconds = 2
servicebot_conf.max_duration_seconds = 2
servicebot_conf.addServiceInfo(
"Service",
modulePath() + "/testservices/StoreService.py", 1, 1, 1)
conf.addSection("SERVICE_BOT", servicebot_conf)
conf.addTeamInfo("Team1", "127.0.0.1", "127.0.0.0/24")
conf.addTeamInfo("Team1", "127.0.0.1", "127.0.0.0/24")
gamestate = GameStateServerInfo("localhost", 4242, "0123456789012345",
"ABCDEFGH")
conf.setGameStateServerInfo(gamestate)
scheduler = ServiceTaskScheduler(conf, True)
scheduler.execute(0)
results = scheduler.execute(1)
scheduler.terminateAtRoundEnd()
for i in range(0, conf.numTeams()):
for j in range(0, servicebot_conf.numServices()):
self.assert_(results[i][j].store() == "Worked")
class TestAttackManager(unittest.TestCase):
def setUp(self):
self.conf = Config()
self.conf.addTeamInfo("Team1","127.0.0.1","127.0.0.0/24")
self.conf.addTeamInfo("Team2","127.0.1.1","127.0.1.0/24")
atkcfg = AttackConfig(True)
atkcfg.exploit_dir = os.path.join(modulePath(),"testexploits")
atkcfg.exploit_timeout = 5
atkcfg.round_interval = 5
atkcfg.gather_interval = 1
self.conf.addSection("ATTACK_BOT",atkcfg)
def testUpdateExploits(self):
manager = AttackManager(self.conf,None,True,False)
manager.start()
manager.cmd(AttackManager.UPDATE_EXPLOITS)
exploit_set = manager.test(AttackManager.TEST_GET_EXPLOITS)
self.assertTrue("Succesful.py" in exploit_set)
self.assertTrue("Forever.py" in exploit_set)
self.assertTrue("Timeout.py" in exploit_set)
def testLaunchExploit(self):
manager = AttackManager(self.conf,None,True,False)
manager.start()
manager.cmd(AttackManager.UPDATE_EXPLOITS)
manager.cmd(AttackManager.LAUNCH_EXPLOIT,"Succesful.py")
time.sleep(1)
manager.cmd(AttackManager.PROCESS_OUTPUT)
time.sleep(1)
manager.cmd(AttackManager.GATHER_FLAGS)
time.sleep(1)
results = manager.getFlags()
self.assertEquals(len(results),2)
r1 = results[0]
r2 = results[1]
self.assertEquals(r1[0],0)
self.assertEquals(r2[0],1)
self.assertEquals(r1[2],['foobarbaz'])
self.assertEquals(r2[2],['foobarbaz'])
def testStore(self):
conf = Config()
servicebot_conf = ServiceBotConfig()
servicebot_conf.min_duration_seconds = 2
servicebot_conf.max_duration_seconds = 2
servicebot_conf.addServiceInfo("Service", modulePath() + "/testservices/StoreService.py", 1, 1, 1)
conf.addSection("SERVICE_BOT", servicebot_conf)
conf.addTeamInfo("Team1", "127.0.0.1", "127.0.0.0/24")
conf.addTeamInfo("Team1", "127.0.0.1", "127.0.0.0/24")
gamestate = GameStateServerInfo("localhost", 4242, "0123456789012345", "ABCDEFGH")
conf.setGameStateServerInfo(gamestate)
scheduler = ServiceTaskScheduler(conf, True)
scheduler.execute(0)
results = scheduler.execute(1)
scheduler.terminateAtRoundEnd()
for i in range(0, conf.numTeams()):
for j in range(0, servicebot_conf.numServices()):
self.assert_(results[i][j].store() == "Worked")
class TestServiceTaskScheduler(unittest.TestCase):
def setUp(self):
#Building a fake config
timeout = 1
self.conf = Config()
self.conf.addTeamInfo("Team1","127.0.0.1","127.0.0.0/24")
self.conf.addTeamInfo("Team2","127.0.1.1","127.0.1.0/24")
self.servicebot_conf = ServiceBotConfig()
self.servicebot_conf.min_duration_seconds = 5
self.servicebot_conf.max_duration_seconds = 5
self.servicebot_conf.addServiceInfo(
"Service1",
modulePath()+"/testservices/ErrorService.py",
timeout,
1,1)
self.servicebot_conf.addServiceInfo(
"Service2",
modulePath()+"/testservices/ErrorService.py",
timeout,
1,1)
self.servicebot_conf.addServiceInfo(
"Service3",
modulePath()+"/testservices/ErrorService.py",
timeout,
1,1)
gamestate = GameStateServerInfo(
"localhost",
4242,
"0123456789012345",
"ABCDEFGH")
self.conf.setGameStateServerInfo(gamestate)
self.conf.addSection("SERVICE_BOT",self.servicebot_conf)
def testSimpleSchedule(self):
round = 0
scheduler = ServiceTaskScheduler(self.conf,True)
start = time.time()
results = scheduler.execute(round)
end = time.time()
run_time = end - start
self.assert_(4.0 < run_time and run_time < 6.0)
for team,service,task in results:
self.assertNotEqual(None,task.error())
self.assertEquals(None,task.prevFlag())
scheduler.quit()
def testCookie(self):
conf = Config()
servicebot_conf = ServiceBotConfig()
servicebot_conf.min_duration_seconds = 2
servicebot_conf.max_duration_seconds = 2
servicebot_conf.addServiceInfo(
"Service",
modulePath()+"/testservices/StoreService.py",
1,
1,1)
conf.addSection("SERVICE_BOT",servicebot_conf)
conf.addTeamInfo("Team1","127.0.0.1","127.0.0.0/24")
conf.addTeamInfo("Team1","127.0.0.1","127.0.0.0/24")
gamestate = GameStateServerInfo(
"localhost",
4242,
"0123456789012345",
"ABCDEFGH")
conf.setGameStateServerInfo(gamestate)
scheduler = ServiceTaskScheduler(conf,True)
scheduler.execute(0)
results = scheduler.execute(1)
for team,service,task in results:
self.assertEquals(task.cookie(),"Worked")
scheduler.quit()
class TestServiceTaskScheduler(unittest.TestCase):
def setUp(self):
# Building a fake config
timeout = 1
self.conf = Config()
self.conf.addTeamInfo("Team1", "127.0.0.1", "127.0.0.0/24")
self.conf.addTeamInfo("Team2", "127.0.1.1", "127.0.1.0/24")
self.servicebot_conf = ServiceBotConfig()
self.servicebot_conf.min_duration_seconds = 5
self.servicebot_conf.max_duration_seconds = 5
self.servicebot_conf.addServiceInfo("Service1", modulePath() + "/testservices/ErrorService.py", timeout, 1, 1)
self.servicebot_conf.addServiceInfo("Service2", modulePath() + "/testservices/ErrorService.py", timeout, 1, 1)
self.servicebot_conf.addServiceInfo("Service3", modulePath() + "/testservices/ErrorService.py", timeout, 1, 1)
gamestate = GameStateServerInfo("localhost", 4242, "0123456789012345", "ABCDEFGH")
self.conf.setGameStateServerInfo(gamestate)
self.conf.addSection("SERVICE_BOT", self.servicebot_conf)
def testSimpleSchedule(self):
round = 0
scheduler = ServiceTaskScheduler(self.conf, True)
start = time.time()
results = scheduler.execute(round)
scheduler.terminateAtRoundEnd()
end = time.time()
run_time = end - start
self.assert_(4.0 < run_time and run_time < 6.0)
for i in range(0, self.conf.numTeams()):
for j in range(0, self.servicebot_conf.numServices()):
self.assert_(results[i][j].status() == ServiceTask.ERROR)
def testStore(self):
conf = Config()
servicebot_conf = ServiceBotConfig()
servicebot_conf.min_duration_seconds = 2
servicebot_conf.max_duration_seconds = 2
servicebot_conf.addServiceInfo("Service", modulePath() + "/testservices/StoreService.py", 1, 1, 1)
conf.addSection("SERVICE_BOT", servicebot_conf)
conf.addTeamInfo("Team1", "127.0.0.1", "127.0.0.0/24")
conf.addTeamInfo("Team1", "127.0.0.1", "127.0.0.0/24")
gamestate = GameStateServerInfo("localhost", 4242, "0123456789012345", "ABCDEFGH")
conf.setGameStateServerInfo(gamestate)
scheduler = ServiceTaskScheduler(conf, True)
scheduler.execute(0)
results = scheduler.execute(1)
scheduler.terminateAtRoundEnd()
for i in range(0, conf.numTeams()):
for j in range(0, servicebot_conf.numServices()):
self.assert_(results[i][j].store() == "Worked")
class TestServiceTaskFactory(unittest.TestCase):
def setUp(self):
#Building a fake config
timeout = 1
self.conf = Config()
self.conf.addTeamInfo("Team1","127.0.0.1","127.0.0.0/24")
self.conf.addTeamInfo("Team2","127.0.1.1","127.0.1.0/24")
self.servicebot_conf = ServiceBotConfig()
self.servicebot_conf.addServiceInfo(
"Service1",
modulePath()+"/testservices/GoodService.py",
timeout,
1,1)
self.servicebot_conf.addServiceInfo(
"Service2",
modulePath()+"/testservices/StoreService.py",
timeout,
1,1)
gamestate = GameStateServerInfo(
"localhost",
4242,
"0123456789012345",
"ABCDEFGH")
self.conf.setGameStateServerInfo(gamestate)
self.conf.addSection("SERVICE_BOT",self.servicebot_conf)
def testBuildSimple(self):
factory = ServiceTaskFactory(self.conf,True)
task = factory.build(0,0,0)
task.start()
self.assert_(task.status() == ServiceTask.OK)
def testStorePersistance(self):
factory = ServiceTaskFactory(self.conf,True)
tasks = [
factory.build(0,1,0),
factory.build(1,1,0),
]
for task in tasks:
task.start()
factory.update(tasks[0],0,1,1)
factory.update(tasks[1],1,1,1)
task = factory.build(1,1,0)
task.start()
self.assert_(task.store() == "Worked")
def testSave(self):
factory = ServiceTaskFactory(self.conf,True)
tasks = [
factory.build(0,1,1),
factory.build(1,1,1),
]
for task in tasks:
task.start()
factory.update(tasks[0],0,1,0)
factory.update(tasks[1],1,1,0)
#Test a "crash recovery" by building a new factory
factory.save()
new_factory = ServiceTaskFactory(self.conf,False)
task = new_factory.build(1,1,0)
task.start()
self.assert_(task.store() == "Worked")
def testFlagUpdate(self):
factory = ServiceTaskFactory(self.conf,True)
#Simulate the service being down - prev flag should be invalid
task = factory.build(0,0,6)
task.start()
self.assert_(task.status() == ServiceTask.OK)
factory.update(task,0,0,6)
self.assert_(task.status() == ServiceTask.INVALID_FLAG)
#Simulate a round passing - prev flag should be valid
task = factory.build(0,0,7)
task.start()
factory.update(task,0,0,7)
self.assert_(task.status() == ServiceTask.OK)
def testUpdateRoundOne(self):
factory = ServiceTaskFactory(self.conf,True)
#Clear prev flag
task = factory.build(0,0,42)
task.start()
#Round 1 should be ok, even if the prev_flag is invalid
task = factory.build(0,0,1)
task.start()
self.assertEquals(task.status(),ServiceTask.OK)
factory.update(task,0,0,1)
self.assertEquals(task.status(),ServiceTask.OK)
def testStress(self):
self.conf = Config()
for i in range(0,1000):
self.conf.addTeamInfo("Team","127.0.0.1","127.0.0.0/24")
self.conf.addServiceInfo(
"Service",
modulePath()+"/testservices/GoodService.py",
1,
1,1)
factory1 = ServiceTaskFactory(self.conf,True)
factory2 = ServiceTaskFactory(self.conf,False)
class TestServiceTaskFactory(unittest.TestCase):
def setUp(self):
#Building a fake config
timeout = 1
self.conf = Config()
self.conf.addTeamInfo("Team1", "127.0.0.1", "127.0.0.0/24")
self.conf.addTeamInfo("Team2", "127.0.1.1", "127.0.1.0/24")
self.servicebot_conf = ServiceBotConfig()
self.servicebot_conf.addServiceInfo(
"Service1",
modulePath() + "/testservices/GoodService.py", timeout, 1, 1)
self.servicebot_conf.addServiceInfo(
"Service2",
modulePath() + "/testservices/StoreService.py", timeout, 1, 1)
gamestate = GameStateServerInfo("localhost", 4242, "0123456789012345",
"ABCDEFGH")
self.conf.setGameStateServerInfo(gamestate)
self.conf.addSection("SERVICE_BOT", self.servicebot_conf)
def testBuildSimple(self):
factory = ServiceTaskFactory(self.conf, True)
task = factory.build(0, 0, 0)
task.start()
self.assert_(task.status() == ServiceTask.OK)
def testStorePersistance(self):
factory = ServiceTaskFactory(self.conf, True)
tasks = [
factory.build(0, 1, 0),
factory.build(1, 1, 0),
]
for task in tasks:
task.start()
factory.update(tasks[0], 0, 1, 1)
factory.update(tasks[1], 1, 1, 1)
task = factory.build(1, 1, 0)
task.start()
self.assert_(task.store() == "Worked")
def testSave(self):
factory = ServiceTaskFactory(self.conf, True)
tasks = [
factory.build(0, 1, 1),
factory.build(1, 1, 1),
]
for task in tasks:
task.start()
factory.update(tasks[0], 0, 1, 0)
factory.update(tasks[1], 1, 1, 0)
#Test a "crash recovery" by building a new factory
factory.save()
new_factory = ServiceTaskFactory(self.conf, False)
task = new_factory.build(1, 1, 0)
task.start()
self.assert_(task.store() == "Worked")
def testFlagUpdate(self):
factory = ServiceTaskFactory(self.conf, True)
#Simulate the service being down - prev flag should be invalid
task = factory.build(0, 0, 6)
task.start()
self.assert_(task.status() == ServiceTask.OK)
factory.update(task, 0, 0, 6)
self.assert_(task.status() == ServiceTask.INVALID_FLAG)
#Simulate a round passing - prev flag should be valid
task = factory.build(0, 0, 7)
task.start()
factory.update(task, 0, 0, 7)
self.assert_(task.status() == ServiceTask.OK)
def testUpdateRoundOne(self):
factory = ServiceTaskFactory(self.conf, True)
#Clear prev flag
task = factory.build(0, 0, 42)
task.start()
#Round 1 should be ok, even if the prev_flag is invalid
task = factory.build(0, 0, 1)
task.start()
self.assertEquals(task.status(), ServiceTask.OK)
factory.update(task, 0, 0, 1)
self.assertEquals(task.status(), ServiceTask.OK)
def testStress(self):
self.conf = Config()
for i in range(0, 1000):
self.conf.addTeamInfo("Team", "127.0.0.1", "127.0.0.0/24")
self.conf.addServiceInfo(
"Service",
modulePath() + "/testservices/GoodService.py", 1, 1, 1)
factory1 = ServiceTaskFactory(self.conf, True)
factory2 = ServiceTaskFactory(self.conf, False)
class TestServiceTaskScheduler(unittest.TestCase):
def setUp(self):
#Building a fake config
timeout = 1
self.conf = Config()
self.conf.addTeamInfo("Team1", "127.0.0.1", "127.0.0.0/24")
self.conf.addTeamInfo("Team2", "127.0.1.1", "127.0.1.0/24")
self.servicebot_conf = ServiceBotConfig()
self.servicebot_conf.min_duration_seconds = 5
self.servicebot_conf.max_duration_seconds = 5
self.servicebot_conf.addServiceInfo(
"Service1",
modulePath() + "/testservices/ErrorService.py", timeout, 1, 1)
self.servicebot_conf.addServiceInfo(
"Service2",
modulePath() + "/testservices/ErrorService.py", timeout, 1, 1)
self.servicebot_conf.addServiceInfo(
"Service3",
modulePath() + "/testservices/ErrorService.py", timeout, 1, 1)
gamestate = GameStateServerInfo("localhost", 4242, "0123456789012345",
"ABCDEFGH")
self.conf.setGameStateServerInfo(gamestate)
self.conf.addSection("SERVICE_BOT", self.servicebot_conf)
def testSimpleSchedule(self):
round = 0
scheduler = ServiceTaskScheduler(self.conf, True)
start = time.time()
results = scheduler.execute(round)
scheduler.terminateAtRoundEnd()
end = time.time()
run_time = end - start
self.assert_(4.0 < run_time and run_time < 6.0)
for i in range(0, self.conf.numTeams()):
for j in range(0, self.servicebot_conf.numServices()):
self.assert_(results[i][j].status() == ServiceTask.ERROR)
def testStore(self):
conf = Config()
servicebot_conf = ServiceBotConfig()
servicebot_conf.min_duration_seconds = 2
servicebot_conf.max_duration_seconds = 2
servicebot_conf.addServiceInfo(
"Service",
modulePath() + "/testservices/StoreService.py", 1, 1, 1)
conf.addSection("SERVICE_BOT", servicebot_conf)
conf.addTeamInfo("Team1", "127.0.0.1", "127.0.0.0/24")
conf.addTeamInfo("Team1", "127.0.0.1", "127.0.0.0/24")
gamestate = GameStateServerInfo("localhost", 4242, "0123456789012345",
"ABCDEFGH")
conf.setGameStateServerInfo(gamestate)
scheduler = ServiceTaskScheduler(conf, True)
scheduler.execute(0)
results = scheduler.execute(1)
scheduler.terminateAtRoundEnd()
for i in range(0, conf.numTeams()):
for j in range(0, servicebot_conf.numServices()):
self.assert_(results[i][j].store() == "Worked")
