p59342861
/
monkey
forked from p15670423/monkey
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Island: Add endpoint to delete agent data 

This endpoint allows to delete the data gathered by agents without resetting config related databases
This commit is contained in:
vakarisz 2022-04-21 17:38:10 +03:00
parent fb2f85ba24
commit 551439dcc2
2 changed files with 16 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
monkey/monkey_island/cc/resources/root.py
View File

@ -19,6 +19,8 @@ class Root(flask_restful.Resource):
if not action: if not action:
return self.get_server_info() return self.get_server_info()
elif action == "delete-agent-data":
return jwt_required(Database.reset_db)(reset_config=False)
elif action == "reset": elif action == "reset":
return jwt_required(Database.reset_db)() return jwt_required(Database.reset_db)()
elif action == "is-up": elif action == "is-up":

16
monkey/monkey_island/cc/services/database.py
View File

@ -3,8 +3,10 @@ import logging
from flask import jsonify from flask import jsonify
from monkey_island.cc.database import mongo from monkey_island.cc.database import mongo
from monkey_island.cc.models import Config
from monkey_island.cc.models.agent_controls import AgentControls from monkey_island.cc.models.agent_controls import AgentControls
from monkey_island.cc.models.attack.attack_mitigations import AttackMitigations from monkey_island.cc.models.attack.attack_mitigations import AttackMitigations
from monkey_island.cc.models.island_mode_model import IslandMode
from monkey_island.cc.services.config import ConfigService from monkey_island.cc.services.config import ConfigService
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
@ -15,19 +17,29 @@ class Database(object):
pass pass
@staticmethod @staticmethod
def reset_db(): def reset_db(reset_config=True):
logger.info("Resetting database") logger.info("Resetting database")
# We can't drop system collections. # We can't drop system collections.
[ [
Database.drop_collection(x) Database.drop_collection(x)
for x in mongo.db.collection_names() for x in mongo.db.collection_names()
if not x.startswith("system.") and not x == AttackMitigations.COLLECTION_NAME if Database._should_drop(x, reset_config)
] ]
ConfigService.init_config() ConfigService.init_config()
Database.init_agent_controls() Database.init_agent_controls()
logger.info("DB was reset") logger.info("DB was reset")
return jsonify(status="OK") return jsonify(status="OK")
@staticmethod
def _should_drop(collection: str, drop_config: bool) -> bool:
if not drop_config:
if collection == IslandMode.COLLECTION_NAME or collection == Config.COLLECTION_NAME:
return False
return (
not collection.startswith("system.")
and not collection == AttackMitigations.COLLECTION_NAME
)
@staticmethod @staticmethod
def drop_collection(collection_name: str): def drop_collection(collection_name: str):
mongo.db[collection_name].drop() mongo.db[collection_name].drop()