ok
Mini Shell
import logging
from functools import lru_cache, partial
from typing import Set
from pathlib import Path
from defence360agent.subsys.panels.directadmin import DirectAdmin as Base
from defence360agent.subsys.panels.directadmin.config import ConfigOptions
from imav.malwarelib.subsys.pure_ftpd import PureFTPBaseConfig
from im360.subsys.panels.base import PanelInterface
from .mod_security import DirectAdminModSecurity, get_custombuild_webserver
from .remoteip import RemoteIP
PURE_FTP_CONFIG = (
"/usr/local/directadmin/custombuild/configure/pureftpd/pure-ftpd.conf"
)
SETUP_TXT_OLD = Path("/usr/local/directadmin/scripts/setup.txt")
SETUP_TXT = Path("/usr/local/directadmin/conf/setup.txt")
VERSIONS = "/usr/local/directadmin/custombuild/versions.txt"
logger = logging.getLogger(__name__)
class DirectAdmin(Base, PanelInterface, DirectAdminModSecurity, RemoteIP):
pure_ftp_conf_cls = partial(PureFTPBaseConfig, filename=PURE_FTP_CONFIG)
@lru_cache(maxsize=1)
def _get_config_options(self):
config = {"is_SSL_on": False, "SSL_port": False, "port": "2222"}
try:
config["is_SSL_on"] = (ConfigOptions("SSL").get() or "0") == "1"
config["SSL_port"] = ConfigOptions("ssl_port").get() or False
config["port"] = ConfigOptions("port").get() or "2222"
except (OSError, KeyError):
logger.exception("Failed to obtain options from DA config")
return config
@lru_cache(maxsize=1)
def _get_setup_file(self):
if SETUP_TXT.exists():
return SETUP_TXT
elif SETUP_TXT_OLD.exists():
return SETUP_TXT_OLD
else:
raise FileNotFoundError("setup.txt file not found")
async def _get_all_admin_emails(self) -> list:
emails = ConfigOptions("email", filename=self._get_setup_file()).get()
if emails:
return [email.strip() for email in emails.split(",") if email]
return []
def http_ports(self) -> Set[int]:
config = self._get_config_options()
if config["is_SSL_on"]:
return set()
try:
return {int(config["port"])}
except ValueError:
return set()
def https_ports(self) -> Set[int]:
config = self._get_config_options()
ports = set()
try:
if config["SSL_port"]:
ports.add(int(config["SSL_port"]))
except ValueError:
logger.warning(
"Bad value for `SSL_port` in DirectAdmin configuration: %s",
config["SSL_port"],
)
try:
if config["is_SSL_on"]:
ports.add(int(config["port"]))
except ValueError:
logger.warning(
"Bad value for `port` in DirectAdmin configuration: %s",
config["port"],
)
return ports
def remoteip_supported(self) -> bool:
return True
async def get_web_server(self):
return await get_custombuild_webserver()
Zerion Mini Shell 1.0