ok
Mini Shell
import json
import logging
import subprocess
from ipaddress import summarize_address_range, IPv4Address
from defence360agent.subsys.panels.cpanel import cPanel
from defence360agent.utils import run
WHMAPI = "/usr/sbin/whmapi1"
logger = logging.getLogger(__name__)
async def is_running():
try:
rc, *_ = await run(
["/usr/local/cpanel/scripts/restartsrv_cphulkd", "--status"]
)
return rc == 0
except FileNotFoundError:
return False
def cphulk_parse_ip(range_str, comment):
parsed_ips = []
ips = range_str.split("-")
if len(ips) == 1:
return [(ips[0], comment)]
try:
parsed_ips = [
(str(cidr), comment)
for cidr in summarize_address_range(
IPv4Address(ips[0]), IPv4Address(ips[1])
)
]
except ValueError as e:
logger.warning("Got invalid IP error: %s", str(e))
finally:
return parsed_ips
def ips_from_list(list_name):
"""
Obtain white/black lists from cphulk using cpanel api
# whmapi1 read_cphulk_records list_name=black --output=json
:param list_name: name of the list to obtain from
:return: list of ips and networks
"""
list_name = list_name.lower()
ips = []
if not cPanel.is_installed():
return ips
logger.info("Loading from chpulk %s list", list_name)
try:
try:
cp = subprocess.run(
[
WHMAPI,
"read_cphulk_records",
"--output=json",
"list_name=" + list_name,
],
stdout=subprocess.PIPE,
check=True,
timeout=20,
)
except OSError:
logger.info("whmapi not found, skipping import from cphulk")
return []
try:
output = json.loads(cp.stdout.decode())
if not output["metadata"]["result"]:
logger.warning(
"Got error from whmapi1: %s", output["metadata"]["reason"]
)
else:
for rng, comment in output["data"]["ips_in_list"].items():
ips.extend(cphulk_parse_ip(rng, comment))
except (json.JSONDecodeError, KeyError):
logger.error("Broken output from whmapi1: %s", cp.stdout)
except Exception:
logger.exception("Failed to obtain ip list from cphulk")
return ips
Zerion Mini Shell 1.0