def test_send_malware_to_agent(self):
result = None
try:
# run agent stub
agent, agent_id = agent_stub.agent()
agent.setDaemon(True)
agent.start()
# get Online Agent
result = requests.get("http://{0}:{1}/agents".format(
self.cnc_ip, self.cnc_port))
if (len(result.json()["agents"]) <= 0): self.fail("no agents")
# select agent
agent = result.json()["agents"][0]
values = {'malware_id': 42, 'agent_id': agent["id"]}
result = requests.get("http://{0}:{1}/send_malware".format(
self.cnc_ip, self.cnc_port),
params=values)
# clean delete agent
agent_stub.delete_agent_from_db(agent_id)
if (result.json()["status"] == False):
self.fail(result.json()["msg"])
print(result.text)
except NameError:
self.fail(NameError)
def test_send_malware_to_agent_malware_id_not_found(self):
result = None
try:
# run agent stub
agent, agent_id = agent_stub.agent()
agent.setDaemon(True)
agent.start()
# get Online Agent
result = requests.get("http://{0}:{1}/agents".format(
self.cnc_ip, self.cnc_port))
if (len(result.json()["agents"]) <= 0): self.fail("no agents")
# select agent
agent = result.json()["agents"][0]
values = {'malware_id': "32323242", 'agent_id': agent["id"]}
result = requests.get("http://{0}:{1}/send_malware".format(
self.cnc_ip, self.cnc_port),
params=values)
# clean delete agent
agent_stub.delete_agent_from_db(agent_id)
if (result.json()["status"] == True):
self.fail("staus should not be True ")
except Exception as e:
self.fail(e.__str__())
def test_malware_run_revert(self):
result = None
try:
# run agent stub
agent, agent_id = agent_stub.agent()
agent.setDaemon(True)
agent.start()
# get Online Agent
result = requests.get("http://{0}:{1}/agents".format(
self.cnc_ip, self.cnc_port))
if (len(result.json()["agents"]) <= 0): self.fail("no agents")
# select agent
agent = result.json()["agents"][0]
values = {'malware_id': "42", 'agent_id': agent["id"]}
result = requests.get("http://{0}:{1}/send_malware".format(
self.cnc_ip, self.cnc_port),
params=values)
if (result.json()["status"] == False):
self.fail(result.json()["msg"])
# run malware
result = requests.get("http://{0}:{1}/run_malware".format(
self.cnc_ip, self.cnc_port),
params=values)
if (result.json()["status"] == False):
self.fail(result.json()["msg"])
# revert malware
result = requests.get("http://{0}:{1}/revert_malware".format(
self.cnc_ip, self.cnc_port),
params=values)
if (result.json()["status"] == False):
self.fail(result.json()["msg"])
print(result.json())
# clean delete agent
agent_stub.delete_agent_from_db(agent_id)
except Exception as e:
self.fail(e.__str__())