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

Island: Init Configuration resource

This commit is contained in:
Ilija Lazoroski 2022-06-21 12:50:52 +02:00 committed by Shreya Malviya
parent 138480f021
commit 5f253e79b3
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
monkey/monkey_island/cc/resources/configuration.py
View File

@ -8,7 +8,7 @@ from flask import jsonify, request
from common.configuration.agent_configuration import AgentConfigurationSchema from common.configuration.agent_configuration import AgentConfigurationSchema
from common.utils.exceptions import InvalidConfigurationError from common.utils.exceptions import InvalidConfigurationError
from monkey_island.cc.repository import FileAgentConfigurationRepository from monkey_island.cc.repository import IAgentConfigurationRepository
from monkey_island.cc.resources.AbstractResource import AbstractResource from monkey_island.cc.resources.AbstractResource import AbstractResource
from monkey_island.cc.resources.request_authentication import jwt_required from monkey_island.cc.resources.request_authentication import jwt_required
@ -40,6 +40,9 @@ class ResponseContents:
class Configuration(AbstractResource): class Configuration(AbstractResource):
urls = ["/api/configuration/<string:configuration_type>"] urls = ["/api/configuration/<string:configuration_type>"]
def __init__(self, file_agent_configuration_repository: IAgentConfigurationRepository):
self._file_agent_configuration_repository = file_agent_configuration_repository
@jwt_required @jwt_required
def get(self, configuration_type: str): def get(self, configuration_type: str):
# Q: we probably still need this because of credential fields, HTTP ports, etc in the # Q: we probably still need this because of credential fields, HTTP ports, etc in the
@ -47,7 +50,7 @@ class Configuration(AbstractResource):
if configuration_type == ConfigurationTypeEnum.ISLAND: if configuration_type == ConfigurationTypeEnum.ISLAND:
pass pass
elif configuration_type == ConfigurationTypeEnum.AGENT: elif configuration_type == ConfigurationTypeEnum.AGENT:
configuration = FileAgentConfigurationRepository.get_configuration() configuration = self._file_agent_configuration_repository.get_configuration()
return jsonify(configuration=configuration) return jsonify(configuration=configuration)
@jwt_required @jwt_required
@ -63,7 +66,7 @@ class Configuration(AbstractResource):
# Q: in what format/schema are we getting the config from the Island? # Q: in what format/schema are we getting the config from the Island?
# Q: when does flattening the config go away? # Q: when does flattening the config go away?
configuration_object = schema.loads(configuration_json) configuration_object = schema.loads(configuration_json)
FileAgentConfigurationRepository.store_configuration( self._file_agent_configuration_repository.store_configuration(
configuration_object configuration_object
) # check error handling ) # check error handling
return ResponseContents().form_response() return ResponseContents().form_response()