Skip to content

Commit

Permalink
Add build_permission_change_command to ILinuxAgentCommandBuilder
Browse files Browse the repository at this point in the history
  • Loading branch information
VakarisZ committed Jun 11, 2024
1 parent 6c1489c commit a02c7f1
Showing 1 changed file with 15 additions and 1 deletion.
16 changes: 15 additions & 1 deletion agentpluginapi/i_linux_agent_command_builder.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@
from typing import Optional

from monkeytypes import InfectionMonkeyBaseModel
from pydantic import model_validator
from pydantic import model_validator, Field

from .dropper_execution_mode import DropperExecutionMode

Expand All @@ -20,6 +20,11 @@ class LinuxDownloadOptions(InfectionMonkeyBaseModel):
download_url: str


class LinuxPermissionChangeOptions(InfectionMonkeyBaseModel):
file_path: PurePosixPath
permissions: int = Field(ge=0, le=777, default=700)


class LinuxRunOptions(InfectionMonkeyBaseModel):
agent_destination_path: PurePosixPath
dropper_execution_mode: DropperExecutionMode
Expand Down Expand Up @@ -47,6 +52,15 @@ def build_download_command(self, download_options: LinuxDownloadOptions):
:param download_options: Options needed for the command to be built
"""

@abc.abstractmethod
def build_permission_change_command(self,
permission_change_options: LinuxPermissionChangeOptions):
"""
Build Agent's binary permission change command
:param permission_change_options: Options needed for the command to be built
"""

@abc.abstractmethod
def build_run_command(self, run_options: LinuxRunOptions):
"""
Expand Down

0 comments on commit a02c7f1

Please sign in to comment.