forkserver.pyc
1 o 2 5�Hc�0 � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlZddl m 3 Z 4 ddl m Z ddl m Z ddl mZ ddl mZ ddl mZ g d �Zd 5 Ze�d �ZG d d � d e�Zddd�Zdd� Zdd� Zdd� Ze� Zej Z ej Z ej Z ej Z dS )� N� )� 6 connection)�process)� reduction)�resource_tracker)�spawn)�util)�ensure_running�get_inherited_fds�connect_to_new_process�set_forkserver_preload� �qc @ sD e Zd Zdd� Zdd� Zdd� Zdd� Zd d 7 � Zd d � Zd d� Z dS )� 8 ForkServerc C s. d | _ d | _d | _d | _t�� | _dg| _d S )N�__main__)�_forkserver_address�_forkserver_alive_fd�_forkserver_pid�_inherited_fds� threading�Lock�_lock�_preload_modules��self� r ��C:\Program Files\WindowsApps\PythonSoftwareFoundation.Python.3.10_3.10.2800.0_x64__qbz5n2kfra8p0\lib\multiprocessing\forkserver.py�__init__" s 9 zForkServer.__init__c C s4 | j � | �� W d � d S 1 sw Y d S �N)r �_stop_unlockedr r r r �_stop* s 10 "�zForkServer._stopc C sV | j d u rd S t�| j� d | _t�| j d� d | _ t�| j�s&t�| j� d | _d S )Nr ) r �os�closer �waitpidr � is_abstract_socket_namespacer �unlinkr r r r r / s 11 12 zForkServer._stop_unlockedc C s& t dd� | jD ��std��|| _dS )z>Set list of module names to try to load in forkserver process.c s s � | ] }t |�tu V qd S r )�type�str)�.0�modr r r � <genexpr>@ s � z4ForkServer.set_forkserver_preload.<locals>.<genexpr>z&module_names must be a list of stringsN)�allr � TypeError)r Z modules_namesr r r r > s 13 z!ForkServer.set_forkserver_preloadc C s | j S )z�Return list of fds inherited from parent process. 14 15 This returns None if the current process was not started by fork 16 server. 17 )r r r r r r 18 D s z ForkServer.get_inherited_fdsc 19 C s� | � � t|�d tkrtd��t�tj��X}|�| j� t� � \}}t� � \}}||| j 20 t � � g}||7 }z,z t �||� ||fW W t�|� t�|� W d � S t�|� t�|� � t�|� t�|� w 1 sqw Y dS )a; Request forkserver to create a child process. 21 22 Returns a pair of fds (status_r, data_w). The calling process can read 23 the child process's pid and (eventually) its returncode from status_r. 24 The calling process should write to data_w the pickled preparation and 25 process data. 26 � z too many fdsN)r �len�MAXFDS_TO_SEND� 27 ValueError�socket�AF_UNIX�connectr r! �piper r �getfdr �sendfdsr" )r �fds�client�parent_r�child_w�child_r�parent_wZallfdsr r r r L s0 � 28 29 30 � 31 32 33 34 �z!ForkServer.connect_to_new_processc s� | j �� t�� | jdur1t�| jtj�\}}|s" W d � dS t�| j� d| _ d| _d| _d}| j 35 rKddh� t � d�}� fdd�|� � D �}ni }t�tj��s}t�d�}|�|� t�|�sit�|d � |�� t�� \}}z<z+|�� |g} ||�� || j 36 |f; }t �� } 37 | 38 gt�� } | d 39 |g7 } t�| 40 | | �}W n t�|� � W t�|� nt�|� w || _ || _|| _W d � n1 s�w Y W d � dS W d � dS 1 s�w Y dS ) z�Make sure that a fork server is running. 41 42 This can be called from any process. Note that usually a child 43 process will just reuse the forkserver started by its parent, so 44 ensure_running() will do nothing. 45 NzCfrom multiprocessing.forkserver import main; main(%d, %d, %r, **%r)� main_path�sys_path�ignorec s i | ] \}}|� v r||�qS r r )r( �x�y�Z desired_keysr r � 46 <dictcomp>� s z-ForkServer.ensure_running.<locals>.<dictcomp>r2 i� z-c) r r r r r! r# �WNOHANGr" r r r r �get_preparation_data�itemsr1 r2 r �arbitrary_address�bindr r$ �chmod�listenr4 �fileno�get_executable� _args_from_interpreter_flags�spawnv_passfds) r �pid�status�cmd�data�listener�address�alive_rZalive_w� fds_to_pass�exe�argsr rB r r i sZ 47 � 48 49 50 51 � 52 � ��"�zForkServer.ensure_runningN) 53 �__name__� 54 __module__� __qualname__r r r r r 55 r r r r r r r s r c C s� |r4d|v r |dur dt �� _z t�|� W t �� `nt �� `w |D ]}zt|� W q" ty3 Y q"w t�� t � 56 � \}}t � |d� t � |d� dd� }t j |t jt ji} dd� | �� D �} 57 t �|� i } tjtj| d ���y} t�� ��b} | �� t_| �| tj� | �|tj� | �|tj� �z( d 58 d � | �� D �}|r�nq�||v r�t � |d �d ks�J d��t �||v �r t � |d� z t � dt j �\}}W n t y� Y n5w |dkr�n/| �!|d�}|du�rt �"|�}zt#||� W n t$y� Y nw t �%|� nt&�'d| � q�| |v �r�| �(� d ��}t)�*|t+d �}t,|�t+k�r-t-d�.t,|����|^}}}|�%� t �/� }|dk�r�d }z<z | �%� | �%� ||||g}|�0| �1� � t2|||| 59 �}W n t3�yv t4j5t4�6� � t4j7�8� Y nw W t �9|� n,t �9|� w zt#||� W n 60 t$�y� Y nw || |< t �%|� |D ]}t �%|� �q�W d � n 1 �s�w Y W n t:�y� } z|j;t;j<k�r̂ W Y d}~nd}~ww q�1 �s�w Y W d � dS 1 �s�w Y dS )zRun forkserver.r NTFc W s d S r r )Z_unusedr r r �sigchld_handler� s z main.<locals>.sigchld_handlerc S s i | ] \}}|t � ||��qS r )�signal)r( �sig�valr r r rC � s �zmain.<locals>.<dictcomp>)rK c S s g | ]\}}|j �qS r )�fileobj)r( �key�eventsr r r � 61 <listcomp>� s zmain.<locals>.<listcomp>r � z Not at EOF?i �����r z.forkserver: waitpid returned unexpected pid %dz Too many ({0:n}) fds to send)=r �current_process� _inheritingr �import_main_path� 62 __import__� ImportErrorr � _close_stdinr! r4 � set_blockingr] �SIGCHLD�SIGINT�SIG_IGNrF � set_wakeup_fdr1 r2 � selectors�DefaultSelector� getsockname� _forkserverr �register� 63 EVENT_READ�select�read� 64 SystemExitr# rD �ChildProcessError�pop�waitstatus_to_exitcode� write_signed�BrokenPipeErrorr" �warnings�warn�acceptr �recvfdsr/ r. � RuntimeError�format�fork�extend�values� 65 _serve_one� Exception�sys� 66 excepthook�exc_info�stderr�flush�_exit�OSError�errno� ECONNABORTED)Z listener_fdrU �preloadr= r> �modnameZsig_rZsig_wr\ �handlers� old_handlersZ pid_to_fdrS �selectorZrfdsrO �stsr: � 67 returncode�sr7 r; �code� 68 unused_fds�fd�er r r �main� s� 69 � �� 70 � 71 � 72 � 73 74 � �� 75 �� 76 77 ���� 78 � ��& ����8�r� c C sh t �d� |�� D ] 79 \}}t � ||� q |D ]}t�|� q|^t_tj_ t_ 80 t� | �}t � | |�}|S )Nre )r] rp rF r! r" rt r r �_resource_tracker�_fdr �dupr �_main) r; r7 r� r� r^ r_ r� �parent_sentinelr� r r r r� + s 81 � 82 r� c C sX d}t j}t|�|k r%t�| |t|� �}|std��||7 }t|�|k s t �|�d S )Nrd zunexpected EOFr )� SIGNED_STRUCT�sizer. r! rx �EOFError�unpack)r� rR �lengthr� r r r � read_signedB s �r� c C sB t �|�}|r t�| |�}|dkrtd��||d � }|sd S d S )Nr zshould not get here)r� �packr! �writer� )r� �n�msg�nbytesr r r r} L s 83 �r} )NN) r� r! rq r] r1 �structr� r r � r r �contextr r r r �__all__r/ �Structr� �objectr r� r� r� r} rt r r 84 r r r r r r �<module> s< 85 86 87 88