Edit File: resource_limits.cpython-311.pyc
� a�nh+ � � � 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 d�Zd efd�Zd efd�ZdS )� N)�Enum)�fsdecode)�Path)�List)� OsReleaseInfoz /usr/libexec/run-with-intensityz/usr/sbin/lvectlz/proc/lve/listc � � e Zd ZdZdZdZdS )� LimitsMethod�nice�lve�cgroupsN)�__name__� __module__� __qualname__�NICE�LVE�CGROUPS� � �Z/opt/imunify360/venv/lib/python3.11/site-packages/defence360agent/utils/resource_limits.pyr r s � � � � � � �D� �C��G�G�Gr 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)�stdout�stderrNr r r z>Parsing of used limitation method failed stdout: {} stderr: {})�asyncio�create_subprocess_exec�RUN_WITH_INTENSITY� subprocess�PIPE� communicater �stripr r r r � LookupError�format)�procr r s r �get_current_methodr% s � � � ��/����!�&��!�&� � � � � � � � � �D� �+�+�-�-�-�-�-�-�-�-�N�F�F� �f� � � #� #� %� %�F� ����� � � ����� � � �����#�#� �J� �����(�(�.�.�0�0� 1� 1� � � r �cmd�key� 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 �str�extendr r )r& r' r( r) �subprocess_kwargs� limits_cmds r �create_subprocessr0 2 s� � � � �$ � �� �M���� �L��� �J� ���w��n�%�%�%��/� �s� �� 0�� � � � � � � � r c �Z � t � � � ot j � � S )z1Checks that LVE-utils is active resource limiter.)�PROC_LVE_LIST_PATH�existsr � is_cloudlinuxr r r � is_lve_activer5 R s$ � � � $� $� &� &� H�=�+F�+H�+H�Hr c �4 � t � � � S )z#Checks that LVE-utils is installed.)�LVECTL_BIN_PATHr3 r r r � has_lvectlr8 Y s � � � !� !� #� #�#r ) r �logging�enumr �osr �pathlibr �typingr �defence360agent.utilsr � getLoggerr �loggerr r7 r2 r r% r, �intr �Processr0 �boolr5 r8 r r r �<module>rD s� �� ���� ���� � � � � � � � � � � � � � � � � � � � � � � � � /� /� /� /� /� /� �� �8� $� $�� 7� ��$�)�*�*���T�*�+�+� �� � � � �4� � � � �,� � � � �0 � �c�� � � � � � � � �� � � � � �@I�t� I� I� I� I�$�D� $� $� $� $� $� $r