From bfeea19c88155700db46a747f2cb6bf6a4921b8b Mon Sep 17 00:00:00 2001 From: Mike Salvatore Date: Tue, 10 May 2022 08:47:40 -0400 Subject: [PATCH] Island: Fix broken AWS queries in aws_command_runner --- monkey/monkey_island/cc/services/aws/aws_command_runner.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/monkey/monkey_island/cc/services/aws/aws_command_runner.py b/monkey/monkey_island/cc/services/aws/aws_command_runner.py index 6795868d7..88eb7e3b4 100644 --- a/monkey/monkey_island/cc/services/aws/aws_command_runner.py +++ b/monkey/monkey_island/cc/services/aws/aws_command_runner.py @@ -69,13 +69,13 @@ def _run_command_async( doc_name = "AWS-RunShellScript" if target_os == "linux" else "AWS-RunPowerShellScript" logger.debug(f'Running command on {target_instance_id} -- {doc_name}: "{command}"') - command_response = aws_client.ssm.send_command( + command_response = aws_client.send_command( DocumentName=doc_name, Parameters={"commands": [command]}, InstanceIds=[target_instance_id], ) - command_id = command_response["CommandId"] + command_id = command_response["Command"]["CommandId"] logger.debug( f"Started command on AWS instance {target_instance_id} with command ID {command_id}" ) @@ -93,7 +93,7 @@ def _wait_for_command_to_complete( sleep_time = min((timer.time_remaining - STATUS_CHECK_SLEEP_TIME), STATUS_CHECK_SLEEP_TIME) time.sleep(sleep_time) - command_status = aws_client.ssm.get_command_invocation( + command_status = aws_client.get_command_invocation( CommandId=command_id, InstanceId=target_instance_id )["Status"] logger.debug(f"Command {command_id} status: {command_status}")