관리-도구
편집 파일: forkserver.cpython-35.opt-2.pyc
���] � @ sd d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l m Z d d l m Z d d l m Z d d l m Z d d l m Z d d l m Z d d d d g Z d Z e j d � Z Gd d � d e � Z d d d d � Z d d � Z d d � Z d d � Z e � Z e j Z e j Z e j Z e j Z d S)� N� )� connection)�process)� reduction)�semaphore_tracker)�spawn)�util�ensure_running�get_inherited_fds�connect_to_new_process�set_forkserver_preload� �Qc @ sL e Z d Z d d � Z d d � Z d d � Z d d � Z d d � Z d S) � ForkServerc C s: d | _ d | _ d | _ t j � | _ d g | _ d S)N�__main__)�_forkserver_address�_forkserver_alive_fd�_inherited_fds� threading�Lock�_lock�_preload_modules)�self� r �//opt/alt/python35/lib64/python3.5/forkserver.py�__init__! s zForkServer.__init__c C s5 t d d � | j D� � s( t d � � | | _ d S)Nc s s! | ] } t | � t k Vq d S)N)�type�str)�.0�modr r r � <genexpr>* s z4ForkServer.set_forkserver_preload.<locals>.<genexpr>z&module_names must be a list of strings)�allr � TypeError)r Z modules_namesr r r r ( s z!ForkServer.set_forkserver_preloadc C s | j S)N)r )r r r r r . s z ForkServer.get_inherited_fdsc C s | j � t | � d t k r, t d � � t j t j � �� } | j | j � t j � \ } } t j � \ } } | | | j t j � g } | | 7} zJ y t j | | � | | f SWn% t j | � t j | � � Yn XWd t j | � t j | � XWd QRXd S)N� z too many fds)r �len�MAXFDS_TO_SEND� ValueError�socket�AF_UNIX�connectr �os�piper r Zgetfdr �sendfds�close)r �fds�clientZparent_r�child_w�child_rZparent_wZallfdsr r r r 6 s& z!ForkServer.connect_to_new_processc s� | j ��t j � | j d k r' d Sd d } | j r} d d h � t j d � } t � f d d � | j � D� � } n i } t j t j � � } t j d � } | j | � t j | d � | j � t j � \ } } z� yy | j � | g } | | j � | | j | f ;} t j � } | g t j � } | d | g 7} t j | | | � } Wn t j | � � Yn XWd t j | � X| | _ | | _ Wd QRXWd QRXd S) Nz-from multiprocessing.forkserver import main; zmain(%d, %d, %r, **%r)� main_path�sys_path�ignorec 3 s- | ]# \ } } | � k r | | f Vq d S)Nr )r �x�y)� desired_keysr r r e s z,ForkServer.ensure_running.<locals>.<genexpr>r( i� z-c)r r r r r r Zget_preparation_data�dict�itemsr' r( r �arbitrary_address�bindr* �chmod�listenr+ �filenoZget_executabler � _args_from_interpreter_flags�spawnv_passfdsr- r ) r �cmd�data�listener�address�alive_rZalive_wZ fds_to_passZexe�args�pidr )r7 r r S s> ( zForkServer.ensure_runningN)�__name__� __module__� __qualname__r r r r r r r r r r s r c 4 C sH | r� d | k rN | d k rN d t j � _ z t j | � Wd t j � ` Xx1 | D]) } y t | � WqU t k r} YqU XqU Wt j � t j t j t j t j i } d d � | j � D� } t j t j d | ��f} t j � �P} | j � t _ | j | t j � | j | t j � xy� x$ d d � | j � D� } | r+Pq+W| | k rat � | j � d �� } d } t j � d k r�zP y t | | | | � Wn2 t k r�t j t j! � � t j" j# � Yn XWd t j$ | � XWd QRXWq%t% k r1} z | j& t& j' k r � WYd d } ~ Xq%Xq%WWd QRXWd QRXd S) Nr Tc S s+ i | ]! \ } } t j | | � | � q Sr )�signal)r �sig�valr r r � <dictcomp>� s zmain.<locals>.<dictcomp>r> c S s g | ] \ } } | j � q Sr )�fileobj)r �key�eventsr r r � <listcomp>� s zmain.<locals>.<listcomp>r r )(r �current_processZ _inheritingr Zimport_main_path� __import__� ImportErrorr � _close_stdinrK �SIGCHLD�SIG_IGN�SIGINT�SIG_DFLr9 r' r( � selectors�DefaultSelector� getsockname� _forkserverr �register� EVENT_READ�select� SystemExit�acceptr* �fork� _serve_one� Exception�sys� excepthook�exc_info�stderr�flush�_exit�OSError�errnoZ ECONNABORTED)Z listener_fdrE Zpreloadr2 r3 �modname�handlersZ old_handlersrC �selectorZrfds�s�code�er r r �main� sR ru c C s� | j � t j | � x* | j � D] \ } } t j | | � q$ Wt j | t d � } | j � | ^ } } t _ } t _ | t j _ t | t j � � d t j k r� d d l } | j � t j | � } t | | � d S)Nr �randomr )r- r* r9 rK r �recvfdsr% r^ r r r Z_semaphore_trackerZ_fd�write_unsigned�getpidrg �modulesrv �seedr �_main) rr rC rE rp rL rM r. r1 r0 Zstfdrv rs r r r re � s re c C sq d } t j } xN t | � | k r_ t j | | t | � � } | sR t d � � | | 7} q Wt j | � d S)N� zunexpected EOFr )�UNSIGNED_STRUCT�sizer$ r* �read�EOFError�unpack)�fdrB �lengthrr r r r � read_unsigned� s r� c C sZ t j | � } xD | rU t j | | � } | d k rB t d � � | | d � } q Wd S)Nr zshould not get here)r~ �packr* �write� RuntimeError)r� �n�msg�nbytesr r r rx � s rx ) rn r* r[ rK r' �structrg r � r r r r r r �__all__r% �Structr~ �objectr ru re r� rx r^ r r r r r r r r �<module> s6 h= !