File manager - Edit - /opt/saltstack/salt/lib/python3.10/site-packages/salt/transport/__pycache__/tcp.cpython-310.pyc
Back
o ;jG� � @ s> d Z ddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl Z ddl Z ddl Z ddlZ ddlZ ddlZ ddlZ ddlZ ddlZ ddlZ ddlZ ddlZ ddlZ ddlZ ddlZ ddlZ ddlZ ddl m Z m Z ddl m Z ddl!m"Z" e j#j$�%� r�dZ&ndZ&e&r�ddl'Z e�(e)�Z*G dd � d e+�Z,d d � Z-d d � Z.dd� Z/G dd� de"�Z0G dd� d�Z1G dd� de j2j3j4�Z5G dd� de j2j3j6�Z7G dd� de j8j9j:j;�Z<G dd� de<�Z=G d d � d e j8j9j>j?�Z@G d d � d �ZAG d d!� d!�ZBG d"d#� d#e j8j9j:j;�ZCG d$d%� d%e j2j3jD�ZEG d&d'� d'e j2j3jF�ZGdS )(zh TCP transport classes Wire protocol: "len(payload) msgpack({'head': SOMEHEADER, 'body': SOMEBODY})" � N)�SaltClientError�SaltReqTimeoutError)� ip_bracket)�SignalHandlingProcessTFc @ s e Zd ZdZdS )� ClosingError� N)�__name__� __module__� __qualname__�__doc__� r r �F/opt/saltstack/salt/lib/python3.10/site-packages/salt/transport/tcp.pyr 3 s r c C s& t j}| �dd�r t j}t � |t j�S )NZipv6F)�socket�AF_INET�get�AF_INET6� SOCK_STREAM)�opts�familyr r r � _get_socket7 s r c C s t | d dd�t| | �fS )NZ interfaceT��strip)r �int)r Z port_typer r r �_get_bind_addr> s �r c C sb t td�r�|�dd�r�| �tjtjd� t td�rgt td�r3|�dd�}|d kr3| �tjtjt|�� t td �rM|�d d�}|d krM| �tjtj t|�� t td �rg|�d d�}|d krg| �tjtj t|�� t td�r�|�dd�}|�d d�}|d ks�|d kr�|d kr�d}|d kr�d}| � tj dt|d �t|d �f� dS dS dS | �tjtjd � dS dS )z< Ensure that TCP keepalives are set for the socket. � SO_KEEPALIVEZ tcp_keepaliveF� �SOL_TCP� TCP_KEEPIDLE�tcp_keepalive_idle���r � TCP_KEEPCNT�tcp_keepalive_cnt� TCP_KEEPINTVL�tcp_keepalive_intvl�SIO_KEEPALIVE_VALSi i� N) �hasattrr r � setsockopt� SOL_SOCKETr r r r r r" �ioctlr$ )�sockr r r! r# r r r �_set_tcp_keepaliveE sV � � � ����r* c s< e Zd ZdZdZ� fdd�Zdd� Zdd� Zd d � Z� Z S ) �LoadBalancerServera Raw TCP server which runs in its own process and will listen for incoming connections. Each incoming connection will be sent via multiprocessing queue to the workers. Since the queue is shared amongst workers, only one worker will handle a given connection. � c s( t � jdi |�� || _|| _d | _d S )Nr )�super�__init__r � socket_queue�_socket)�selfr r/ �kwargs�� __class__r r r. � s zLoadBalancerServer.__init__c C s0 | j d ur| j �tj� | j �� d | _ d S d S �N)r0 �shutdownr � SHUT_RDWR�close�r1 r r r r8 � s �zLoadBalancerServer.closec C � | � � d S r5 �r8 r9 r r r �__del__� � zLoadBalancerServer.__del__c C sn t | j�| _| j�tjtjd� t| j| j� | j�d� | j� t | jd�� | j� | j � | jr�| jj s�z| j�� \}}| j�||fdd� W n_ ty� } zS| j}|du r^W Y d}~dS |�� }t|t�rk|d }tjjj�|�tjkr|W Y d}~q.|jtj kr�t � d|� W Y d}~dS |jtj kr�t � d|� W Y d}~dS � d}~ww | jr�| jj r5dS dS dS dS )z) Start the load balancer r �ret_portTNr zSocket shutdown: %szSocket closed: %s) r r r0 r& r r' � SO_REUSEADDRr* � setblocking�bindr �listen�backlog�_closed�acceptr/ �put�OSError� getsockname� isinstance�tuple�salt�ext�tornado�utilZerrno_from_exception�errnoZ ECONNABORTED�EINVAL�log�trace�EBADF)r1 � connection�address�er) �namer r r �run� s: �� �zLoadBalancerServer.run) r r r r rC r. r8 r< rX � __classcell__r r r3 r r+ � s r+ c @ s& e Zd ZdZeddd��Zdd� ZdS ) �ResolverF� c C s t jjjjjd|d� d| _d S )Nz)salt.ext.tornado.netutil.ThreadedResolver)� num_threadsT)rK rL rM ZnetutilrZ Z configure�_resolver_configured)�clsr\ r r r �_config_resolver� s � zResolver._config_resolverc O s | j s | �� d S d S r5 )r] r_ )r1 �argsr2 r r r r. � s �zResolver.__init__N)r[ )r r r r] � classmethodr_ r. r r r r rZ � s rZ c s� e Zd ZdZdZ� fdd�Zdd� Zejj j j ddd ��Z ejj j j d d � �Z ejj j j d d � �Zdd� Zdd� Zdd� Z� ZS )� TCPPubClientz& Tornado based TCP Pub Client �tcpc s@ t � j||fi |�� || _|| _d | _d| _d| _t� | _d S �NF) r- r. r �io_loop�message_client� connected�_closingrZ �resolver)r1 r re r2 r3 r r r. � s zTCPPubClient.__init__c C s2 | j rd S d| _ | jd ur| j�� d | _d S d S �NT�rh rf r8 r9 r r r r8 � s �zTCPPubClient.closeNc c s^ � d| _ || _t| j| jd t| j�| j||| j�d�| j�d�d�| _| j�� V d| _ d S )NTZ master_ip� source_ipZsource_publish_port)re �connect_callback�disconnect_callbackrl � source_port) �_connect_called� publish_port� MessageClientr r re r rf �connectrg )r1 rq rm rn r r r rs � s � � zTCPPubClient.connectc C s< t |t�stjj�|�}tjj�|�}n|}tj j j � |��r5 ) rI �dictrK �utils�msgpack�loads� transport�frame�decode_embedded_strsrL rM �gen�Return)r1 �messages�bodyr r r �_decode_messages s z TCPPubClient._decode_messagesc c s � | j j�|�V d S r5 )rf �_stream�write)r1 �msgr r r �send s �zTCPPubClient.sendc C s | j �|�S )z. Register an on_recv callback )rf �on_recv)r1 �callbackr r r r� s zTCPPubClient.on_recvc C � | S r5 r r9 r r r � __enter__ � zTCPPubClient.__enter__c C r: r5 r; )r1 �exc_typeZexc_valZexc_tbr r r �__exit__ r= zTCPPubClient.__exit__�NN)r r r r �ttyper. r8 rK rL rM r{ � coroutiners r r� r� r� r� rY r r r3 r rb � s rb c @ sn e Zd ZdZdZdd� Zedd� �Zdd� Zd d � Z d d � Z d d� Z dd� Z e jjjjddd��Zdd� ZdS )� TCPReqServerzc Tornado based TCP Request/Reply Server :param dict opts: Salt master config options. � c C s || _ d | _d | _d S r5 )r r0 � req_server�r1 r r r r r. - s zTCPReqServer.__init__c C s | j S r5 )r0 r9 r r r r 2 s zTCPReqServer.socketc C s� | j d ur6z | j �tj� W n ty( } z|jtjkr n� W Y d }~nd }~ww | jd u r3| j �� d | _ | jd urhz| j�� W n tyb } z|jdkrP� t � dt |�� W Y d }~nd }~ww d | _d S d S )N� z4TCPReqServerChannel close generated an exception: %s) r0 r6 r r7 rG rO �ENOTCONNr� r8 rQ � exception�str)r1 �excr r r r8 6 s4 ��� ��� �zTCPReqServer.closec C r� r5 r r9 r r r r� T r� zTCPReqServer.__enter__c G r: r5 r; )r1 r` r r r r� W r= zTCPReqServer.__exit__c C s� t rt�� | _|jt| j| jfdd� dS tjj � � sDt | j�| _ | j � tjtjd� t| j | j� | j �d� | j �t| jd�� dS dS )zB Pre-fork we need to create the zmq router device r+ )r` rW r r r> N)�USE_LOAD_BALANCER�multiprocessing�Queuer/ � add_processr+ r rK ru �platform� is_windowsr r0 r& r r' r? r* r@ rA r �r1 Zprocess_managerr r r �pre_forkZ s � �zTCPReqServer.pre_forkc C s� || _ tjj�|��j tr t| j| j| j � d�d�| _ nPtjj � � rIt| j �| _| j�tjtjd� t| j| j � | j�d� | j�t| j d�� t| j| j � d�|d�| _ | j �| j� | j�| j� W d � dS W d � dS 1 sww Y dS )z� After forking we need to create all of the local sockets to listen to the router message_handler: function to call with your payloads �ssl�� ssl_optionsr r r> )r� re N) �message_handlerrK ru � asynchronous�current_ioloopr� �LoadBalancerWorkerr/ �handle_messager r r� r� r� r r0 r& r r' r? r* r@ rA r �SaltMessageServer� add_socketrB rC )r1 r� re r r r � post_forkl s0 � ��"�zTCPReqServer.post_forkNc c s4 � | � |�}| �|�V }|�tjjj||d�� d S )N��header)�decode_payloadr� r� rK rx ry � frame_msg)r1 �stream�payloadr� Zreplyr r r r� � s � zTCPReqServer.handle_messagec C � |S r5 r )r1 r� r r r r� � r� zTCPReqServer.decode_payloadr5 )r r r r rC r. �propertyr r8 r� r� r� r� rK rL rM r{ r� r� r� r r r r r� # s r� c sP e Zd ZdZ� fdd�Zejjjj ejjj j fdd��Z dd� Z dd � Z� ZS ) r� z{ Raw TCP server which will receive all of the TCP streams and re-assemble messages that are sent through to us c sJ |� dd �p tjjjj�� }d| _t� j |i |�� || _ g | _ || _ d S )Nre F) �poprK rL rM �ioloop�IOLoop�currentrh r- r. re �clientsr� )r1 r� r` r2 re r3 r r r. � s � zSaltMessageServer.__init__c c s � t �d|� | j�||f� tjj�� }z+ |jddd�V }|� |� |D ]}tj j � |�}|d }| j �| j||d |� q&q |y] t �d|� tjj�� }| �||f� Y d S ty� } z t jd|dd � tjj�� }| �||f� |�� W Y d }~d S d }~ww ) zP Handle incoming streams and add messages to the incoming queue zReq client %s connectedT� ��partial�headr~ zreq client disconnected %sz other master-side exception: %s��exc_infoN)rQ rR r� �appendrK ru rv �Unpacker� read_bytes�feedrx ry rz re �spawn_callbackr� � remove_client� Exceptionr8 ) r1 r� rU �_StreamClosedError�unpacker� wire_bytes� framed_msgr� rV r r r � handle_stream� s4 � �� ��z SaltMessageServer.handle_streamc C s2 z | j �|� W d S ty t�d� Y d S w )Nz/Message server client was not in list to remove)r� �remove� ValueErrorrQ rR �r1 �clientr r r r� � s �z SaltMessageServer.remove_clientc C s| | j rdS d| _ t| j�D ]}|\}}|�� | �|� q z| �� W dS ty= } z |jdkr2� W Y d}~dS d}~ww )z" Close the server NTr� )rh �listr� r8 r� �stoprG rO )r1 �itemr� rU r� r r r r8 � s ���zSaltMessageServer.close)r r r r r. rK rL rM r{ r� �iostream�StreamClosedErrorr� r� r8 rY r r r3 r r� � s � r� c s4 e Zd ZdZ� fdd�Z� fdd�Zdd� Z� ZS )r� z� This will receive TCP connections from 'LoadBalancerServer' via a multiprocessing queue. Since the queue is shared amongst workers, only one worker will handle a given connection. c sH t � j|g|�R i |�� || _t�� | _tj| jd�| _| j� � d S )N)�target) r- r. r/ � threading�Event�_stop�Thread�socket_queue_thread�thread�start)r1 r/ r� r` r2 r3 r r r. � s zLoadBalancerWorker.__init__c s" | j �� | j�� t� �� d S r5 )r� �setr� �joinr- r8 r9 r3 r r r8 � s zLoadBalancerWorker.closec C sn z* z | j �dd�\}}W n tjy | j�� r Y W d S Y qw | j�| j||� q t t fy6 Y d S w �NTr ) r/ r �queue�Emptyr� �is_setre r� Z_handle_connection�KeyboardInterrupt� SystemExit)r1 Z client_socketrU r r r r� � s ��� �z&LoadBalancerWorker.socket_queue_thread)r r r r r. r8 r� rY r r r3 r r� � s r� c s* e Zd ZdZd� fdd� Zdd� Z� ZS )�TCPClientKeepAlivezN Override _create_stream() in TCPClient to enable keep alive support. Nc s || _ t� j|d� d S )N�ri )r r- r. )r1 r ri r3 r r r. s zTCPClientKeepAlive.__init__c K s4 t | j�}t|| j� tjjjj||d�}|�|�S )z� Override _create_stream() in TCPClient. Tornado 4.5 added the kwargs 'source_ip' and 'source_port'. Due to this, use **kwargs to swallow these and any future kwargs to maintain compatibility. )�max_buffer_size) r r r* rK rL rM r� ZIOStreamrs )r1 r� �af�addrr2 r) r� r r r �_create_stream s � z!TCPClientKeepAlive._create_streamr5 )r r r r r. r� rY r r r3 r r� s r� c @ s� e Zd ZdZ d dd�Zdd� Zejjj j dd� �Z d d � Z ejjj j d d � �Z ejjj j d d� �Zejjj j dd� �Zdd� Zdd� Zdd� Zdd� Zejjj j d dd��ZdS ) rr z* Low-level message sending client Nc C s� || _ || _|| _|| _| | _|| _|| _|p tjj j j � � | _ tjj�| j �� t||d�| _W d � n1 s:w Y i | _d | _d | _d| _d| _tjj j�� | _d| _d | _ |� dd�| _ d S )Nr� FZtcp_reconnect_backoffr ) r �host�portrl ro rm rn rK rL rM r� r� r� re ru r� r� r� � _tcp_client�send_future_map�_read_until_future�_on_recvrh rD � concurrent�FutureZ_connecting_future�_stream_return_runningr� r �backoff) r1 r r� r� re ri rm rn rl ro r r r r. * s( �zMessageClient.__init__c C s* | j s| jrd S d| _ | j�d| j� d S r� )rh rD re � add_timeout� check_closer9 r r r r8 N s zMessageClient.closec C sJ | j s| j�� | jr| j�� d | _d| _d| _d S | j�d| j� d S )NTFr ) r� r� r8 r� rD rh re r� r� r9 r r r r� T s zMessageClient.check_closec C r: r5 r; r9 r r r r< a r= zMessageClient.__del__c k s� � | j s| jr| j | jd�}d }|d u rg| jsg| jsgz| jjt| jdd�| jfd| j � d�i|��V }W n( t y\ } z t � d| j| j|| j� tjjj�| j�V W Y d }~nd }~ww |d u rg| jsg| jrtjjj�|��)N)rl ro Tr r� r� zgTCP Message Client encountered an exception while connecting to %s:%s: %r, will reconnect in %d seconds)rl ro rD rh r� rs r r� r� r r r� rQ �warningr� rK rL rM r{ �sleepr| )r1 r2 r� r� r r r � getstreamf s: � � � � �� ���zMessageClient.getstreamc c s^ � | j d u r)| �� V | _ | j r+d| _d| _| js | j�| j� | jr-| �d� d S d S d S d S )NFT) r� r� rh rD r� re r� �_stream_returnrm r9 r r r rs � s � �zMessageClient.connectc c sb � d| _ tjj�� }| j�s,| j�s,zM| jjddd�V }|� |� |D ]:}tj j � |�}|d }|d }|� d�}|| jv rH| j�|��|� q#| jd urW| j�| j||� q#t�d|� q#W n� tjjjjy� } zJt�d| j| j � | j� � D ]}|� |� qyi | _| js�| jr�W Y d }~d S | j r�| � � | j} d | _| r�| � � tjj�� }| �!� V W Y d }~npd }~w t"y� d | j#v r�t�$d � nt%�Y nW t&�y$ } zJtjd dd � | j� � D ]}|� |� q�i | _| js�| jr�W Y d }~d S | j �r| � � | j} d | _| �r| � � tjj�� }| �!� V W Y d }~nd }~ww | j�s,| jrd | _ d S )NTr� r� r� r~ �midz7Got response for message_id %s that we are not trackingz*tcp stream to %s:%s closed, unable to recv� detect_modez[There was an error trying to use TCP transport; attempting to fallback to another transportzException parsing responser� F)'r� rK ru rv r� rD rh r� r� r� rx ry rz r r� r� Z set_resultr� re r� rQ �errorrL rM r� r� �debugr� r� �values� set_exceptionrn r8 rs � TypeErrorr �infor r� ) r1 r� r� r� r� r~ � message_idrV �futurer� r r r r� � s� � ��� � �� ��� Az MessageClient._stream_returnc C s t t�� �S r5 )r� �uuidZuuid4r9 r r r � _message_id� r= zMessageClient._message_idc s( � du r � | _ dS � fdd�}|| _ dS )zU Register a callback for received messages (that we didn't initiate) Nc s � |� d S r5 r )r� r~ �r� r r � wrap_recv� r= z(MessageClient.on_recv.<locals>.wrap_recv)r� )r1 r� r r r r r� � s zMessageClient.on_recvc C s* || j vrd S | j �|�}| j�|� d S r5 )Zsend_timeout_mapr� re Zremove_timeout)r1 r �timeoutr r r �remove_message_timeout� s z$MessageClient.remove_message_timeoutc C s8 || j vrd S | j �|�}|d ur|�td�� d S d S )NzMessage timed out)r� r� r� r )r1 r r� r r r r �timeout_message� s �z MessageClient.timeout_messageFc # s� � �j rt� ���� }d|i}tjjj�� }� d ur&� �fdd�}|�|� |�j |< �j � d�du r5d}|d urC�j � |�j||� tjjj||d��tjjjj��fdd ��} �j �| � |V } tjjj�| ��) Nr� c s | � � }�j�� |� d S r5 )�resultre � add_callback)r �response)r� r1 r r � handle_future� s z)MessageClient.send.<locals>.handle_futurer� Tr r� c 3 s( � �� � V �jr�j�� �V d S d S r5 )rs r� r� r )r� r1 r r �_do_send s � �z$MessageClient.send.<locals>._do_send)rh r r rK rL rM r� r� Zadd_done_callbackr� r r re Z call_laterr rx ry r� r{ r� r r| ) r1 r� r r� �rawr r� r r r �recvr )r� r� r1 r r� � s( � zMessageClient.send)NNNNNN)NNF)r r r r r. r8 rK rL rM r{ r� r� r< r� rs r� r r� r r r� r r r r rr % s2 �$ F rr c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) � Subscriberz= Client object for use with the TCP publisher server c C s" || _ || _d| _d | _d | _d S rd )r� rU rh r� �id_)r1 r� rU r r r r. ! � zSubscriber.__init__c C sR | j rd S d| _ | j�� s#| j�� | jd ur%| j�� r'| j�� d S d S d S d S rj )rh r� �closedr8 r� Zdoner� r9 r r r r8 ( s �zSubscriber.closec C r: r5 r; r9 r r r r< 8 r= zSubscriber.__del__N)r r r r r. r8 r< r r r r r s r c sr e Zd ZdZ d� fdd� Zdd� Zdd� Zejj j j ejj j j fd d ��Zd d � Zejj j j dd d��Z� ZS )� PubServerz TCP publisher Nc sf t � j|�d�d� || _|| _d| _t� | _d| _|r || _ ndd� | _ |r,|| _ d S dd� | _ d S )Nr� r� Fc S r� r5 r )� subscriberr� r r r �<lambda>O � z$PubServer.__init__.<locals>.<lambda>c S r� r5 r )r r r r r S r ) r- r. r re r rh r� r� Zpresence_events�presence_callback�remove_presence_callback)r1 r re r r r3 r r r. C s zPubServer.__init__c C s6 | j rd S d| _ t| j�D ]}|�� q | j�� d S rj )rh r� r� r8 �clearr� r r r r8 U s zPubServer.closec C r: r5 r; r9 r r r r< ^ r= zPubServer.__del__c c s � t jj�� }| js�z-|jjddd�|_|jV }|�|� |D ]}t j j � |�}|d }| j r5| � ||� q W nC |y` } z t �d|j� |�� | �|� | j�|� W Y d }~d S d }~w tyz } zt jd|jdd� W Y d }~qd }~ww | jr d S d S )Nr� Tr� r~ z'tcp stream to %s closed, unable to recvz"Exception parsing response from %sr� )rK ru rv r� rh r� r� r� r� rx ry rz r rQ r� rU r8 r r� �discardr� r� )r1 r� r� r� r� r� r~ rV r r r � _stream_readc s: � �� �� ���zPubServer._stream_readc C s6 t �d|� t||�}| j�|� | j�| j|� d S )NzSubscriber at %s connected)rQ r� r r� �addre r� r )r1 r� rU r� r r r r� ~ s zPubServer.handle_streamc c s" � t �d||� tjj�|�}g }|rN|D ]7}d}| jD ]%}||jkrAz |j� |�V d}W q tj j j j y@ |�|� Y q w q |sLt �d|| j� qn!| jD ] }z |j� |�V W qQ tj j j j yn |�|� Y qQw |D ]}t �d|j� |�� | �|� | j�|� qqt �d� d S )Nz'TCP PubServer sending payload: %s %rFTz"Publish target %s not connected %rz0Subscriber at %s has disconnected from publisherz)TCP PubServer finished publishing payload)rQ rR rK rx ry r� r� r r� r� rL rM r� r� r� r� rU r8 r r )r1 �packageZ topic_listr� � to_removeZtopic�sentr� r r r �publish_payload� sD � ���� �� zPubServer.publish_payload)NNNr5 )r r r r r. r8 r< rK rL rM r{ r� r� r� r r� r! rY r r r3 r r > s � � r c @ sz e Zd ZdZdZdd� Zedd� �Zdd� Zd d � Z dd d �Z dd� Z e j jjjdd� �Zdd� Zdd� Zdd� Zd S )�TCPPublishServerz) Tornado based TCP PublishServer r, c C s" || _ d | _d | _d | _d| _d S rd )r �pub_sock� pub_serverre rh r� r r r r. � r zTCPPublishServer.__init__c C s | j �dd� S )NZ order_mastersF)r r r9 r r r � topic_support� s z TCPPublishServer.topic_supportc C s | � |d � d S �Nr )r. )r1 �stater r r � __setstate__� s z TCPPublishServer.__setstate__c C s d| j iS r& )r r9 r r r � __getstate__� s z TCPPublishServer.__getstate__Nc C sp t jjj�� }|�� || _t| j|||d� | _ }t | j�}|� t j t jd� t|| j� |�d� |�t| jd�� |�| j� |�|� | j�dd�dkrZt| j�dd ��}n tj�| jd d �}|| _ t jj j |||d �}t � d |� t j j!�"d�� |�#� W d � n1 s�w Y z z|�#� W n t$t%fy� Y nw W |�&� dS W |�&� dS |�&� w )zK Bind to the interface specified in the configuration file )re r r r r rq �ipc_mode� rc �tcp_master_publish_pull� �sock_dir�publish_pull.ipc)re Zpayload_handlerz Starting the Salt Puller on %s� N)'rK rL rM r� r� Z make_currentre r r r$ r r& r r' r? r* r@ rA r rB rC r� r r �os�pathr� rx �ipcZIPCMessageServerrQ r� ru �filesZ set_umaskr� r� r� r8 ) r1 r! r r re r$ r) �pull_uriZ pull_sockr r r �publish_daemon� sL � � � �� �z TCPPublishServer.publish_daemonc C s |j | j| jjd� dS )z� Do anything necessary pre-fork. Since this is on the master side this will primarily be used to create IPC channels and create our daemon process to do the actual publishing )rW N)r� r6 r4 r r� r r r r� � s zTCPPublishServer.pre_forkc g s( � | j j|g|�R � V }tjjj�|��r5 )r$ r! rK rL rM r{ r| )r1 r� r` �retr r r r! s �z TCPPublishServer.publish_payloadc K sx | j �dd�dkrt| j �dd��}n tj�| j d d�}| js4tjj j tj j j |fdd �| _| j�� | j�|� d S ) z+ Publish "load" to minions r* r+ rc r, r- r. r/ re )Z loop_kwargN)r r r r1 r2 r� r# rK ru r� Z SyncWrapperrx r3 ZIPCMessageClientrs r� )r1 r� r2 r5 r r r �publish s � zTCPPublishServer.publishc C s^ d| _ | jr| j�� d | _| jr| j�� d | _| jr-| j�� | jjdd� d | _d S d S )NT)Zall_fds)rh r# r8 r$ re r� r9 r r r r8 s �zTCPPublishServer.closec C s$ | j stjd| ��t| d� d S d S )Nzunclosed publish server )�source)rh �warnings�warn�ResourceWarningr9 r r r r< % s ��zTCPPublishServer.__del__r� )r r r r rC r. r� r% r( r) r6 r� rK rL rM r{ r� r! r8 r8 r< r r r r r" � s" �6 r" c sV e Zd ZdZdZ� fdd�Zejjj j dd� �Z ejjj j d dd ��Z d d � Z � ZS ) � TCPReqClientz) Tornado based TCP RequestClient rc c s� t � j||fi |�� || _|| _tj�| jd �}|j�dd�\}}|t |�f}|� d�}t j j j||t |�|||� d�|� d�d�| _d| _d S ) NZ master_uri�:r ri rl Zsource_ret_port)re ri rl ro F)r- r. r re �urllib�parse�urlparse�netloc�rsplitr r rK rx rc rr rf rh ) r1 r re r2 r@ Z master_hostZ master_portZ master_addrri r3 r r r. 5 s" � zTCPReqClient.__init__c c s � d| _ | j�� V d S rj )rp rf rs r9 r r r rs I s �zTCPReqClient.connect�< c c s$ � | j j||d�V }tjjj�|��)N)r )rf r� rK rL rM r{ r| )r1 �loadr r7 r r r r� N s �zTCPReqClient.sendc C s | j rd S d| _ | j�� d S rj rk r9 r r r r8 S s zTCPReqClient.close)rD )r r r r r� r. rK rL rM r{ r� rs r� r8 rY r r r3 r r= . s r= )Hr rO �loggingr� r1 r� r r� r? r r: Zsalt.ext.tornadorK Zsalt.ext.tornado.concurrentZsalt.ext.tornado.genZsalt.ext.tornado.iostreamZsalt.ext.tornado.netutilZsalt.ext.tornado.tcpclientZsalt.ext.tornado.tcpserverZ salt.masterZ salt.payloadZsalt.transport.frameZsalt.transport.ipcZsalt.utils.asynchronousZsalt.utils.filesZsalt.utils.msgpackZsalt.utils.platformZsalt.utils.versionsZsalt.exceptionsr r Zsalt.utils.networkr Zsalt.utils.processr ru r� r� r� Zsalt.ext.tornado.util� getLoggerr rQ r� r r r r* r+ rZ rx �baseZ PublishClientrb ZDaemonizedRequestServerr� rL rM Z tcpserverZ TCPServerr� r� Z tcpclientZ TCPClientr� rr r r ZDaemonizedPublishServerr" Z RequestClientr= r r r r �<module> sl >JErI' x"k
| ver. 1.4 |
Github
|
.
| PHP 8.2.30 | Generation time: 0 |
proxy
|
phpinfo
|
Settings