- HTTP: configurable server_header, response_code, fake_app presets (apache/nginx/wordpress/phpmyadmin/iis), extra_headers, custom_body, static files directory mount - SSH/Cowrie: configurable kernel_version, hardware_platform, ssh_banner, and users/passwords via COWRIE_USERDB_ENTRIES; switched to build mode so cowrie.cfg.j2 persona fields and userdb.txt generation work - SMTP: configurable banner and MTA hostname - MySQL: configurable version string in protocol greeting - Redis: configurable redis_version and os string in INFO response - BYOS: [custom-*] INI sections define bring-your-own Docker services - Stealth: rename all *_honeypot.py → server.py; replace HONEYPOT_NAME env var with NODE_NAME across all 22+ service templates and plugins; strip "honeypot" from all in-container file content - Config: DeckyConfig.service_config dict; INI [decky-N.svc] subsections; composer passes service_cfg to compose_fragment - 350 tests passing (100%) Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
159 lines
4.4 KiB
Python
159 lines
4.4 KiB
Python
"""
|
|
Tests for the INI loader — subsection parsing, custom service definitions,
|
|
and per-service config propagation.
|
|
"""
|
|
|
|
import pytest
|
|
import textwrap
|
|
from pathlib import Path
|
|
from decnet.ini_loader import load_ini, IniConfig
|
|
|
|
|
|
def _write_ini(tmp_path: Path, content: str) -> Path:
|
|
f = tmp_path / "decnet.ini"
|
|
f.write_text(textwrap.dedent(content))
|
|
return f
|
|
|
|
|
|
# ---------------------------------------------------------------------------
|
|
# Basic decky parsing (regression)
|
|
# ---------------------------------------------------------------------------
|
|
|
|
def test_basic_decky_parsed(tmp_path):
|
|
ini_file = _write_ini(tmp_path, """
|
|
[general]
|
|
net = 192.168.1.0/24
|
|
gw = 192.168.1.1
|
|
|
|
[decky-01]
|
|
ip = 192.168.1.101
|
|
services = ssh, http
|
|
""")
|
|
cfg = load_ini(ini_file)
|
|
assert len(cfg.deckies) == 1
|
|
assert cfg.deckies[0].name == "decky-01"
|
|
assert cfg.deckies[0].services == ["ssh", "http"]
|
|
assert cfg.deckies[0].service_config == {}
|
|
|
|
|
|
# ---------------------------------------------------------------------------
|
|
# Per-service subsection parsing
|
|
# ---------------------------------------------------------------------------
|
|
|
|
def test_subsection_parsed_into_service_config(tmp_path):
|
|
ini_file = _write_ini(tmp_path, """
|
|
[decky-01]
|
|
ip = 192.168.1.101
|
|
services = ssh
|
|
|
|
[decky-01.ssh]
|
|
kernel_version = 5.15.0-76-generic
|
|
hardware_platform = x86_64
|
|
""")
|
|
cfg = load_ini(ini_file)
|
|
svc_cfg = cfg.deckies[0].service_config
|
|
assert "ssh" in svc_cfg
|
|
assert svc_cfg["ssh"]["kernel_version"] == "5.15.0-76-generic"
|
|
assert svc_cfg["ssh"]["hardware_platform"] == "x86_64"
|
|
|
|
|
|
def test_multiple_subsections_for_same_decky(tmp_path):
|
|
ini_file = _write_ini(tmp_path, """
|
|
[decky-01]
|
|
services = ssh, http
|
|
|
|
[decky-01.ssh]
|
|
users = root:toor
|
|
|
|
[decky-01.http]
|
|
server_header = nginx/1.18.0
|
|
fake_app = wordpress
|
|
""")
|
|
cfg = load_ini(ini_file)
|
|
svc_cfg = cfg.deckies[0].service_config
|
|
assert svc_cfg["ssh"]["users"] == "root:toor"
|
|
assert svc_cfg["http"]["server_header"] == "nginx/1.18.0"
|
|
assert svc_cfg["http"]["fake_app"] == "wordpress"
|
|
|
|
|
|
def test_subsection_for_unknown_decky_is_ignored(tmp_path):
|
|
ini_file = _write_ini(tmp_path, """
|
|
[decky-01]
|
|
services = ssh
|
|
|
|
[ghost.ssh]
|
|
kernel_version = 5.15.0
|
|
""")
|
|
cfg = load_ini(ini_file)
|
|
# ghost.ssh must not create a new decky or error out
|
|
assert len(cfg.deckies) == 1
|
|
assert cfg.deckies[0].name == "decky-01"
|
|
assert cfg.deckies[0].service_config == {}
|
|
|
|
|
|
def test_plain_decky_without_subsections_has_empty_service_config(tmp_path):
|
|
ini_file = _write_ini(tmp_path, """
|
|
[decky-01]
|
|
services = http
|
|
""")
|
|
cfg = load_ini(ini_file)
|
|
assert cfg.deckies[0].service_config == {}
|
|
|
|
|
|
# ---------------------------------------------------------------------------
|
|
# Bring-your-own service (BYOS) parsing
|
|
# ---------------------------------------------------------------------------
|
|
|
|
def test_custom_service_parsed(tmp_path):
|
|
ini_file = _write_ini(tmp_path, """
|
|
[general]
|
|
net = 10.0.0.0/24
|
|
gw = 10.0.0.1
|
|
|
|
[custom-myservice]
|
|
binary = my-image:latest
|
|
exec = /usr/bin/myapp -p 8080
|
|
ports = 8080
|
|
""")
|
|
cfg = load_ini(ini_file)
|
|
assert len(cfg.custom_services) == 1
|
|
cs = cfg.custom_services[0]
|
|
assert cs.name == "myservice"
|
|
assert cs.image == "my-image:latest"
|
|
assert cs.exec_cmd == "/usr/bin/myapp -p 8080"
|
|
assert cs.ports == [8080]
|
|
|
|
|
|
def test_custom_service_without_ports(tmp_path):
|
|
ini_file = _write_ini(tmp_path, """
|
|
[custom-scanner]
|
|
binary = scanner:1.0
|
|
exec = /usr/bin/scanner
|
|
""")
|
|
cfg = load_ini(ini_file)
|
|
assert cfg.custom_services[0].ports == []
|
|
|
|
|
|
def test_custom_service_not_added_to_deckies(tmp_path):
|
|
ini_file = _write_ini(tmp_path, """
|
|
[decky-01]
|
|
services = ssh
|
|
|
|
[custom-myservice]
|
|
binary = foo:bar
|
|
exec = /bin/foo
|
|
""")
|
|
cfg = load_ini(ini_file)
|
|
assert len(cfg.deckies) == 1
|
|
assert cfg.deckies[0].name == "decky-01"
|
|
assert len(cfg.custom_services) == 1
|
|
|
|
|
|
def test_no_custom_services_gives_empty_list(tmp_path):
|
|
ini_file = _write_ini(tmp_path, """
|
|
[decky-01]
|
|
services = http
|
|
""")
|
|
cfg = load_ini(ini_file)
|
|
assert cfg.custom_services == []
|