| Current File : //usr/lib/python3.12/multiprocessing/__pycache__/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 ��6/usr/lib/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$ sE � �!�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 s9 � �!�0�0����:�
3�d� �,�,�T�2�
3�
3�
3�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 sd � �
�Z�Z� .��}�}�$�� � �
� �I�I��N�I�&*�E�]�D�K�K�� � � "��M�M�4�9�9�-� .� .� .�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� �� � �!�!�#�! $� $� $�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 s{ � �"&�+�+�"3�"3�"5� � �C� �$�� �G� � � � ���� �
�
�"�"�$� �>�>� %� �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