Add chmod options

guardicode · Jun 11, 2024 · abe3f7e · abe3f7e
1 parent 0a763c0
commit abe3f7e
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/agentpluginapi/i_linux_agent_command_builder.py b/agentpluginapi/i_linux_agent_command_builder.py
@@ -20,6 +20,11 @@ class LinuxDownloadOptions(InfectionMonkeyBaseModel):
     download_url: str
 
 
+class LinuxChmodOptions(InfectionMonkeyBaseModel):
+    file_path: PurePosixPath
+    permission_string = "u+x"
+
+
 class LinuxRunOptions(InfectionMonkeyBaseModel):
     agent_destination_path: PurePosixPath
     dropper_execution_mode: DropperExecutionMode
@@ -48,9 +53,11 @@ def build_download_command(self, download_options: LinuxDownloadOptions):
         """
 
     @abc.abstractmethod
-    def build_chmod_command(self):
+    def build_chmod_command(self, chmod_options: LinuxChmodOptions):
         """
         Build Agent's binary permission change command
+
+        :param chmod_options: Options needed for the command to be built
         """
 
     @abc.abstractmethod