p20319658
/
monkey
forked from p34709852/monkey
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Show tunnel info and remove config from queried node

This commit is contained in:
Itay Mizeretz 2017-09-13 18:01:41 +03:00
parent 9a9906326a
commit 5b3e526d49
1 changed files with 7 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
monkey_island/cc/services/node.py
View File

@ -13,7 +13,7 @@ class NodeService:
@staticmethod
def get_displayed_node_by_id(node_id):
if ObjectId(node_id) == ObjectId("000000000000000000000000"):
if ObjectId(node_id) == NodeService.get_monkey_island_pseudo_id():
return NodeService.get_monkey_island_node()
edges = EdgeService.get_displayed_edges_by_to(node_id)
@ -31,8 +31,7 @@ class NodeService:
# node is infected
new_node = NodeService.monkey_to_net_node(monkey)
for key in monkey:
# TODO: do something with tunnel
if key not in ["_id", "modifytime", "parent", "tunnel", "dead"]:
if key not in ["_id", "modifytime", "parent", "dead", "config"]:
new_node[key] = monkey[key]
else:
@ -53,8 +52,6 @@ class NodeService:
if len(edges) > 0:
new_node["services"] = edges[-1]["services"]
# TODO: add exploited by
return new_node
@staticmethod
@ -216,11 +213,15 @@ class NodeService:
return monkey
return None
@staticmethod
def get_monkey_island_pseudo_id():
return ObjectId("000000000000000000000000")
@staticmethod
def get_monkey_island_pseudo_net_node():
return\
{
"id": ObjectId("000000000000000000000000"),
"id": NodeService.get_monkey_island_pseudo_id(),
"label": "MonkeyIsland",
"group": "islandClean",
}