ok
Mini Shell
from subprocess import DEVNULL, CalledProcessError, check_output
from typing import Any
from defence360agent.utils import stub_unexpected_error
def _run_cmd(cmd):
try:
out = check_output(cmd, stderr=DEVNULL)
except (FileNotFoundError, CalledProcessError):
return None
return out.decode("utf-8", errors="ignore").strip()
@stub_unexpected_error
def _get_virtualization_type():
systemd_virt = _run_cmd(["systemd-detect-virt"])
if systemd_virt:
return systemd_virt
virt_what = _run_cmd(["virt-what"])
if virt_what:
return virt_what
demicode = _run_cmd(["dmidecode", "-s", "system-manufacturer"])
if demicode:
return demicode
return "fail to detect"
@stub_unexpected_error
def _get_total_ram():
import psutil
return psutil.virtual_memory().total // 2**20
_TAGS = None
def _tags():
global _TAGS
if _TAGS is None:
from defence360agent.utils import OsReleaseInfo
_TAGS = {
"av_version": None,
"version": None,
"os_details": stub_unexpected_error(OsReleaseInfo.pretty_name)(),
"ip": None,
"hosting_panel": None,
"total_ram": _get_total_ram(),
"firewall": None,
"strategy": None,
"virtualization": _get_virtualization_type(),
"server_id": None,
"name": None,
}
return _TAGS
def set_firewall_type(firewall: str) -> None:
_tags()["firewall"] = firewall
def set_hosting_panel(panel: str) -> None:
_tags()["hosting_panel"] = panel
def set_strategy(strategy: str) -> None:
_tags()["strategy"] = strategy
def set_ip(ip: str) -> None:
_tags()["ip"] = ip
def set_product_name(product: str) -> None:
_tags()["name"] = product
def set_server_id(id: str) -> None:
_tags()["server_id"] = id
def set_version(version: str) -> None:
_tags()["version"] = version
def set_av_version(version: str) -> None:
_tags()["av_version"] = version
def tags() -> dict:
return _tags().copy()
def tag(name: str) -> Any:
return _tags()[name]
Zerion Mini Shell 1.0