From 71c4e4d8dcd629b4590d94a92aee8cc3fb118400 Mon Sep 17 00:00:00 2001 From: Mike Salvatore Date: Thu, 2 Sep 2021 14:04:40 -0400 Subject: [PATCH] Agent: Fix incorrect host arch identification in PowerShellClient --- .../exploit/powershell_utils/powershell_client.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/monkey/infection_monkey/exploit/powershell_utils/powershell_client.py b/monkey/infection_monkey/exploit/powershell_utils/powershell_client.py index e739901f1..ad5854d4a 100644 --- a/monkey/infection_monkey/exploit/powershell_utils/powershell_client.py +++ b/monkey/infection_monkey/exploit/powershell_utils/powershell_client.py @@ -85,8 +85,8 @@ class PowerShellClient(IPowerShellClient): return output def get_host_architecture(self) -> Union[WIN_ARCH_32, WIN_ARCH_64]: - output = self._client.execute_cmd(GET_ARCH_WINDOWS) - if "64-bit" in output: + stdout, _, _ = self._client.execute_cmd(GET_ARCH_WINDOWS) + if "64-bit" in stdout: return WIN_ARCH_64 return WIN_ARCH_32