OwlCyberSecurity - MANAGER
Edit File: resource_sharer.cpython-312.pyc
� !��g � � � d dl Z d dlZd dlZd dlZd dlZddlmZ ddlmZ ddlm Z dgZ ej dk( re dgz Z G d � de � Z ne d gz Z G d � d e � Z G d � d e � Z e� Zej" Zy)� N� )�process)� reduction)�util�stop�win32� DupSocketc � � e Zd ZdZd� Zd� Zy)r z Picklable wrapper for a socket.c �z �� |j � ��fd�}t j |�j � | _ y )Nc �J �� �j |� }| j |� y �N)�share� send_bytes)�conn�pidr �new_socks ��F/opt/alt/python312/lib64/python3.12/multiprocessing/resource_sharer.py�sendz DupSocket.__init__.<locals>.send s �� � ���s�+������&� )�dup�_resource_sharer�register�close�_id)�self�sockr r s @r �__init__zDupSocket.__init__ s- �� ��x�x�z�H� '� (�0�0��x�~�~�F�D�Hr c � � t j | j � 5 }|j � }t j |� cddd� S # 1 sw Y yxY w)z1Get the socket. This should only be called once.N)r �get_connectionr � recv_bytes�socket� fromshare)r r r s r �detachzDupSocket.detach$ s< � �!�0�0����:�d� ���)�� �'�'��.� ;�:�:�s �%A�AN��__name__� __module__� __qualname__�__doc__r r# � r r r r s � �-� G� /r �DupFdc � � e Zd ZdZd� Zd� Zy)r* z-Wrapper for fd which can be used at any time.c �z �� t j |� ��fd�}�fd�}t j ||� | _ y )Nc �4 �� t j | �|� y r )r � send_handle)r r �new_fds �r r z DupFd.__init__.<locals>.send1 s �� ��%�%�d�F�C�8r c �0 �� t j � � y r )�osr )r/ s �r r z DupFd.__init__.<locals>.close3 s �� ����� r )r1 r r r r )r �fdr r r/ s @r r zDupFd.__init__/ s. �� ��V�V�B�Z�F� 9� !�'�0�0��u�=�D�Hr c � � t j | j � 5 }t j |� cddd� S # 1 sw Y yxY w)z-Get the fd. This should only be called once.N)r r r r � recv_handle)r r s r r# z DupFd.detach7 s0 � �!�0�0����:�d� �,�,�T�2� ;�:�:�s �?�ANr$ r) r r r* r* - s � �;� >� 3r c �F � e Zd ZdZd� Zd� Zed� � Zd d�Zd� Z d� Z d � Z y) �_ResourceSharerz.Manager for resources using background thread.c �� � d| _ i | _ t j � | _ d | _ d | _ d | _ t j | t j � y )Nr ) �_key�_cache� threading�Lock�_lock� _listener�_address�_threadr �register_after_forkr6 � _afterfork)r s r r z_ResourceSharer.__init__? sI � ��� ��� � �^�^�%�� � ��� �� ��� � � � ���'A�'A�Br c � � | j 5 | j �| j � | xj dz c_ ||f| j | j < | j | j fcddd� S # 1 sw Y yxY w)z+Register resource, returning an identifier.Nr )r< r> �_startr8 r9 )r r r s r r z_ResourceSharer.registerH sY � � �Z�Z��}�}�$�� � � � �I�I��N�I�&*�E�]�D�K�K�� � � "��M�M�4�9�9�-� �Z�Z�s �A$A;�;Bc � � ddl m} | \ }} ||t j � j �� }|j |t j � f� |S )z<Return connection from which to receive identified resource.r ��Client��authkey)� connectionrF r �current_processrH r r1 �getpid)�identrF �address�key�cs r r z _ResourceSharer.get_connectionQ sG � � '� � ��� �7�G�$;�$;�$=�$E�$E� F�� ����R�Y�Y�[�!�"��r Nc � � ddl m} | j 5 | j �� || j t j � j �� }|j d� |j � | j j |� | j j � rt j d� | j j � d| _ d| _ d| _ | j j! � D ] \ }\ }} |� � | j j# � ddd� y# 1 sw Y yxY w)z:Stop the background thread and clear registered resources.r rE NrG z._ResourceSharer thread did not stop when asked)rI rF r< r> r rJ rH r r r? �join�is_aliver � sub_warningr= r9 �items�clear)r �timeoutrF rO rN r r s r r z_ResourceSharer.stopZ s� � �&� �Z�Z��}�}�(��4�=�=�#*�#:�#:�#<�#D�#D�F�����t� ���� �� � �!�!�'�*��<�<�(�(�*��$�$� &;� <����$�$�&� #�� � $�� �!%���*.�+�+�*;�*;�*=�&�C� �$���G� +>�� � �!�!�#�! �Z�Z�s �D D:�:Ec �: � | j j � D ] \ }\ }} |� � | j j � | j j � | j �| j j � d | _ d | _ d | _ y r ) r9 rT rU r< �_at_fork_reinitr= r r> r? )r rN r r s r rA z_ResourceSharer._afterforko sv � �"&�+�+�"3�"3�"5� �C� �$�� �G� #6� � � ���� � � �"�"�$� �>�>� %� �N�N� � � "� ��� �� ��� r c �h � ddl m} | j �J d� �t j d� |t j � j d�� | _ | j j | _ t j | j �� }d|_ |j � || _ y ) Nr )�ListenerzAlready have Listenerz0starting listener and thread for sending handles� )rH �backlog)�targetT)rI rZ r= r �debugr rJ rH rM r> r: �Thread�_serve�daemon�startr? )r rZ �ts r rC z_ResourceSharer._startz s� � �(��~�~�%�>�'>�>�%� � � �E�F�!�'�*A�*A�*C�*K�*K�UX�Y��� ���.�.�� � � � �D�K�K� 0����� ��� ��� r c �* � t t d� r6t j t j t j � � | j j � 5 }|j � }|� d d d � y |\ }}| j j |� \ }} |||� |� d d d � �v# |� w xY w# 1 sw Y �xY w# t j � s$t j t j � � Y �UxY w)N�pthread_sigmask)�hasattr�signalre � SIG_BLOCK� valid_signalsr= �accept�recvr9 �popr � is_exiting�sys� excepthook�exc_info)r r �msgrN �destination_pidr r s r r` z_ResourceSharer._serve� s� � � �6�,� -� � "� "�6�#3�#3�V�5I�5I�5K� L�� 4��^�^�*�*�,�� �)�)�+�C��{� � -�,� ,/�(�C� �"&�+�+�/�/�#�"6�K�D�%� � �T�?�3� �� -� �� ��� -�,�� 4����(��N�N�C�L�L�N�3�sH � C �#C �7C � #C �$ B>�-C �5C �> C�C � C�C �:Dr ) r% r& r' r( r r � staticmethodr r rA rC r` r) r r r6 r6 = s8 � �8�C�.� �� ��$�* � �4r r6 )r1 rg r! rn r: � r �contextr r �__all__�platform�objectr r* r6 r r r) r r �<module>ry s� �� � � � � � � � � �(�� �<�<�7�� � �}� �G� /�F� /� ��y��G� 3�� 3� Y4�f� Y4�x #�$� �� � �r