W
d dl Z d dlZd dlmZ d dlmZ d dlmZ d dlm Z d dl
m Z ej e
ZdZ ed Z ed Z G d
d e Zd efd
Zde e dededed e j j f
dZd efdZd efdZdS ) N)Enum)fsdecode)Path)List)
OsReleaseInfoz /usr/libexec/run-with-intensityz/usr/sbin/lvectlz/proc/lve/listc e Zd ZdZdZdZdS ) LimitsMethodnicelvecgroupsN)__name__
__module__ __qualname__NICELVECGROUPS Z/opt/imunify360/venv/lib/python3.11/site-packages/defence360agent/utils/resource_limits.pyr r s D
CGGGr r returnc K t j t dt j j t j j d{V } | d{V \ }}t
| }|dk r t j S |dk r t j
S |dk r t j S t d
|t
| )z6Returns limit method, used in run-with-intensity tool.show)stdoutstderrNr
r r z>Parsing of used limitation method failed
stdout: {}
stderr: {})asynciocreate_subprocess_execRUN_WITH_INTENSITY
subprocessPIPE communicater stripr r r r LookupErrorformat)procr r s r get_current_methodr% s /!&!& D ++--------NFF
f
#
#
%
%F
##
J ((..00 1 1 r cmdkey
intensity_cpu intensity_ioc K t ddt | dt | g}| d|g t j || z i | d{V S )aS
Creates asyncio.Process with limited resources (cpu & io),
using run-with-intensity tool.
:param cmd: command to execute
:param intensity_cpu: cpu intensity limit
:param intensity_io: io intensity limit
:param subprocess_kwargs: keyword arguments for create_subprocess_exec func
:return: executed Process
runz--intensity-cpuz--intensity-ioz--keyN)r strextendr r )r& r' r( r) subprocess_kwargs
limits_cmds r create_subprocessr0 2 s $
M L
J wn%%%/
s
0 r c Z t ot j S )z1Checks that LVE-utils is active resource limiter.)PROC_LVE_LIST_PATHexistsr
is_cloudlinuxr r r
is_lve_activer5 R s$ $ $ & & H=+F+H+HHr c 4 t S )z#Checks that LVE-utils is installed.)LVECTL_BIN_PATHr3 r r r
has_lvectlr8 Y s ! ! # ##r ) r loggingenumr osr pathlibr typingr defence360agent.utilsr getLoggerr
loggerr r7 r2 r r% r, intr Processr0 boolr5 r8 r r r rD s / / / / / / 8 $ $ 7 $)**T*++ 4 , 0
c
@It I I I I$D $ $ $ $ $ $r