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

Island: Make AWSCommandStatus enum json serializable

This commit is contained in:
vakarisz 2022-05-11 11:46:13 +03:00
parent 30fb57c37f
commit 0b5a507f38
1 changed files with 12 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
monkey/monkey_island/cc/resources/remote_run.py
View File

@ -54,8 +54,20 @@ class RemoteRun(flask_restful.Resource):
resp = {} resp = {}
if body.get("type") == "aws": if body.get("type") == "aws":
result = self.run_aws_monkeys(body) result = self.run_aws_monkeys(body)
result = self._get_encodable_results(result)
resp["result"] = result resp["result"] = result
return jsonify(resp) return jsonify(resp)
# default action # default action
return make_response({"error": "Invalid action"}, 500) return make_response({"error": "Invalid action"}, 500)
@staticmethod
def _get_encodable_results(results: Sequence[AWSCommandResults]) -> str:
results_copy = []
for result in results:
results_copy.append(
AWSCommandResults(
result.response_code, result.stdout, result.stderr, result.status.name.lower()
)
)
return results_copy