PATH:
opt
/
cloudlinux
/
venv
/
lib
/
python3.11
/
site-packages
/
ssa
/
__pycache__
� �d�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
[+]
..
[-] website_isolation.cpython-311.pyc
[open]
[-] clos_ssa_ini.cpython-311.pyc
[open]
[-] __init__.cpython-311.pyc
[open]
[-] db.cpython-311.pyc
[open]
[-] agent.cpython-311.pyc
[open]
[-] manager.cpython-311.pyc
[open]