File: //opt/cloudlinux/venv/lib64/python3.11/site-packages/ssa/__pycache__/agent.cpython-311.pyc
�
�e�i_ � � � d Z ddlZddlZddlZddlZddlZddlmZ ddl m
Z
ddl m Z ddl
mZ ddlmZ dd lmZ d
Z G d � d � � ZdS )
zH
This module contains contains classes implementing SSA Agent behaviour
� N)�ThreadPoolExecutor)�current_thread� )�
agent_sock)�SSAError)�
create_socket)�RequestProcessor�2 c � � e Zd ZdZd� Zd� Zdd�Z eh d�� � Ze dd de
fd
�� � Z e d e
defd �� � Zdd
�Zddde
fd�ZdS )� SimpleAgentz
SSA Simple Agent class
c �� � t j d� � | _ t � � | _ t t �� � | _ t j | j
� � | � � � d S )N�agent)� max_workers) �logging� getLogger�loggerr �request_processorr � MAX_WORKERS�executor�atexit�register� _shutdown�listen��selfs �@/opt/cloudlinux/venv/lib64/python3.11/site-packages/ssa/agent.py�__init__zSimpleAgent.__init__! sW � � �'��0�0�� �!1�!3�!3�� �*�{�C�C�C��
�����'�'�'� � � �
�
�
�
�
� c �< � | j � d�� � dS )z-Gracefully shutdown the thread pool executor.F)�waitN)r �shutdownr s r r zSimpleAgent._shutdown) s! � � �
� � �E� �*�*�*�*�*r �returnNc �� � t t � � } |� � � \ }}| j � | j |� � | j � d� � �R)z(
Start listening socket
Tz[ThreadPool] Submitted task)r r �acceptr �submit�handler �debug)r �_socket�
connection�addresss r r zSimpleAgent.listen- s` � � �
�+�+�� =�")�.�.�"2�"2� �J�� �M� � ���j� 9� 9� 9� �K� � � ;� <� <� <� =r > �url�duration� timestamp� wordpress�hitting_limits�throttled_time�io_throttled_timer) �
socket objectc � � | � t j t j t j d� � � � }t j d|� � \ }}}|S )z�
Get the UID of the peer process using SO_PEERCRED.
:param connection: socket object
:return: UID of the connecting process
�3i)�
getsockopt�
socket_module�
SOL_SOCKET� SO_PEERCRED�struct�calcsize�unpack)r) �cred�_pid�uid�_gids r �
_get_peer_uidzSimpleAgent._get_peer_uid= sP � � �$�$� � $� � %� �O�D� !� !�
�
��
!�-��d�3�3���c�4��
r �datac � � t |t � � r|sdS t |� � � � � | j k S )z�
Validate that input data contains exactly the expected metric fields.
The PHP extension always sends all 7 fields (see dump.c), so we
require an exact match to reject both unknown and partial payloads.
F)�
isinstance�dict�set�keys�_REQUIRED_FIELDS)�clsrA s r �_validate_inputzSimpleAgent._validate_inputL s@ � � �$��%�%� �T� ��5��4�9�9�;�;� � �3�#7�7�7r c � � | � |� � }nr# t t j f$ rY}| j � dt � � j t |� � � � |� � � Y d}~dS d}~ww xY w|� d�� � } | �
|� � }| � |� � s�| j � dt � � j |t |t � � r!t |� � � � � nt# |� � j � � |� � � |� � � dS | j � |� � n�# t* t, j t0 f$ rD}| j � dt � � j t |� � � � Y d}~nTd}~wt2 $ rD}| j � dt � � j t |� � � � Y d}~nd}~ww xY w|� � � |� � � dS # |� � � |� � � w xY w)z�
Handle incoming connection
:param connection: socket object usable to
send and receive data on the connection
z'[%s] Failed to get peer credentials: %sN�ignore)�errorsz2[%s] Rejected invalid payload from UID=%d: keys=%sz Handled exception in [%s]: %sz Unexpected exception in [%s]: %s)r@ �OSErrorr9 �errorr r �name�str�close�makefile�
read_inputrI �warningrC rD �sortedrF �type�__name__r r& r �json�JSONDecodeError�
ValueError� Exception� exception)r r) �peer_uid�e�fileobj�
input_datas r r& zSimpleAgent.handleW s� � � ��)�)�*�5�5�H�H���� �&� � � � �K� � �9� � � �%�s�1�v�v�
/�
/�
/�
� � � � � � �F�F�F�F�F����� ���� �%�%�X�%�6�6�� � � � ��1�1�J��'�'�
�3�3�
�� �#�#�H�"�$�$�)�8�1;�J��1M�1M�l�F�:�?�?�,�,�-�-�-�SW�Xb�Sc�Sc�Sl�n� n� n� �
�M�M�O�O�O� � � � � � � � �
� "� )� )�*� 5� 5� 5� 5���$�.�
�;� =� =� =� �K� � � =� ,� .� .� 3�S��V�V�
=�
=�
=�
=�
=�
=�
=�
=������ A� A� A� �K� !� !�"D�"0�"2�"2�"7��Q���
A�
A�
A�
A�
A�
A�
A�
A����� A����
�M�M�O�O�O� � � � � � � � ��
�M�M�O�O�O� � � � � � � ���s^ � �B�AB�B�!B F �,F �I � H3�#:G"� I �"
H3�/:H.�)I �.H3�3I � *J
�fileioz file objectc � � |� � � }| j � dt � � j t |� � � � � |� � |r(t j |� � � d�� � S i S )zy
Read input data and return decoded json
:param fileio: a file-like object providing read method
z [%s] I received %i bytes: %sF)�strict)
�readr �infor rO �len�encoderX �loads�strip)r ra rA s r rS zSimpleAgent.read_inputz s{ � �
�{�{�}�}�� � ���7�'�)�)�.��D�K�K�M�M�0B�0B� � � � � � ��:�d�j�j�l�l�5�9�9�9� 9��Ir )r" N)r) r2 r" N)rW �
__module__� __qualname__�__doc__r r r � frozensetrG � staticmethod�intr@ � classmethodrD �boolrI r& rS � r r r r s � � � � � �� �� � �+� +� +�=� =� =� =� !�y� "� "� "� � ��
� �/� �c� � � � �\� � �8�4� 8�D� 8� 8� 8� �[�8�! � ! � ! � ! �F � � �4� � � � � � r r )rl r rX r �socketr6 r9 �concurrent.futuresr � threadingr �internal.constantsr �internal.exceptionsr �internal.utilsr �modules.processorr r r rr r r �<module>rz s� ��� � �
�
�
� � � � � ���� � � � �
�
�
�
� 1� 1� 1� 1� 1� 1� $� $� $� $� $� $� *� *� *� *� *� *� )� )� )� )� )� )� )� )� )� )� )� )� /� /� /� /� /� /� � �j� j� j� j� j� j� j� j� j� jr