File: //opt/cloudlinux/venv/lib/python3.11/site-packages/clwpos/__pycache__/socket_utils.cpython-311.pyc
�
�h � � � d dl mZ d dlZd dlZd dlZd dlZd dlmZ dZdZ dej de
fd�Z d e de
fd
�Zdej d e
d e
dee
fd
�Zdej dee fd�Zdej dee fd�Zdej de fd�ZdS )� )�absolute_importN)�Optionalz>I�
� sock_object�returnc � � d}| � t j t j t j |� � � � }t j ||� � \ }}}|S )z�
Retrieve credentials from SO_PEERCRED option
:param sock_object: Socket object
:return: uid of user, which connects to this socket.
�3I)�
getsockopt�socket�
SOL_SOCKET� SO_PEERCRED�struct�calcsize�unpack)r �_format_string�creds�_�_uids �J/opt/cloudlinux/venv/lib64/python3.11/site-packages/clwpos/socket_utils.py�get_uid_from_socketr sM � � �N� � "� "�6�#4�f�6H�&�/�Zh�Ji�Ji� j� j�E�� �~�u�5�5�J�A�t�Q� �K� � data_dictc � � t j | � � � d� � }t j t
t
|� � � � |z S )z�
Prefix message with a 4-byte length
:param data_dict: Data dict for send
:return: byte array for send to socket
�utf-8)�json�dumps�encoder �pack�_format�len)r � msg_bytess r �pack_data_for_socketr" ' s@ � � �
�9�%�%�,�,�W�5�5�I� �;�w��I��� /� /�)� ;�;r � num_bytes� timeout_secc �� � t � � }t |dz � � D ]E}|| � |� � z
}t |� � |k r|c S t j d� � �FdS )z�
Read amount data from socket
:param sock_object: Socket object to read data from
:param num_bytes: Bytes number to read
:param timeout_sec: Read timeout, None - timeout expired, data not received
r g�������?N)�bytes�range�recvr �time�sleep)r r# r$ �msg�is r �$_read_bytes_from_socket_with_timeoutr- 4 so � � �'�'�C�
�;��#�
$�
$� � �� �{� � � �*�*�*�� �s�8�8�y� � ��J�J�J� �
�3����� �4r c �V � | � d� � | � t � � t | dt � � }|�dS t j t |� � d }t | |t � � }|�dS t j |� d� � � � S )��
Read length-prefixed amount of data from socket
:param sock_object: Socket object to read data
:return: Data received from socket dictionary. None - socket data format error
F� Nr r )
� setblocking�
settimeout� _WPOS_SOCKET_READ_TIMEOUT_SECr- r r r r �loads�decode�r �
raw_msglen�msglenr+ s r �'read_unpack_response_from_socket_daemonr9 E s� � � ���E�"�"�"����8�9�9�9�5�k�1�Fc�d�d�J����t�
�]�7�J�
/�
/��
2�F�
.�{�F�Da�
b�
b�C�
�{��t� �:�c�j�j��)�)� *� *�*r c � � | � d� � }t j t |� � d }t � � }t |� � |k r+|| � d� � z
}t |� � |k �+n# t j $ r Y dS w xY wt j |�
d� � � � S )r/ r0 r i Nr ) r( r r r r& r r �timeoutr r4 r5 r6 s r �'read_unpack_response_from_socket_clientr< \ s� � �� �%�%�a�(�(�
�� �w�
�3�3�A�6���g�g���#�h�h�&� � � �;�#�#�D�)�)� )�C� �#�h�h�&� � ��� �>� � � ��t�t����� �:�c�j�j��)�)� *� *�*s �BB �B�B�
connection� data_to_sendc �v � t |� � }| � |� � | � � � dS )z�
Sends dictionary to socket connection and close it
:param connection: Socket connection to send data
:param data_to_send: Data dict to send
N)r" �sendall�close)r= r> �
bytes_to_sends r �(send_dict_to_socket_connection_and_closerC p s= � � )� �6�6�M����}�%�%�%��������r )�
__future__r r r r r) �typingr r r3 �intr �dictr& r" r- r9 r< rC � r r �<module>rI s^ �� '� &� &� &� &� &� � � � �
�
�
�
�
�
�
�
� � � � � � � � � � � �� !#� �
�V�]�
�s�
�
�
�
�
<�D�
<�U�
<�
<�
<�
<��f�m� �PS� �69��>F�u�o�� � � �"+�� � +�8�TX�>� +� +� +� +�.+�� � +�8�TX�>� +� +� +� +�(�� � �VZ� � � � � � r