p59342861
/
monkey
forked from p15670423/monkey
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Agent: Take OperatingSystem in IIslandAPIClient.get_agent_binary()

This commit is contained in:
Mike Salvatore 2022-09-20 11:07:09 -04:00
parent 741d192eab
commit 930256f581
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
monkey/infection_monkey/island_api_client/i_island_api_client.py
View File

@ -1,6 +1,7 @@
from abc import ABC, abstractmethod
from typing import Optional, Sequence
from common import OperatingSystem
from common.agent_events import AbstractAgentEvent
@ -59,11 +60,11 @@ class IIslandAPIClient(ABC):
"""
@abstractmethod
def get_agent_binary(self, os_name: str) -> Optional[bytes]:
def get_agent_binary(self, operating_system: OperatingSystem) -> Optional[bytes]:
"""
Get an agent binary for the given OS from the island
:param os_name: The OS on which the agent binary will run
:param operating_system: The OS on which the agent binary will run
:return: The agent binary file
:raises IslandAPIConnectionError: If the client cannot successfully connect to the island
:raises IslandAPIRequestError: If an error occurs while attempting to connect to the