CasperSecurity
o
���c| � @ s� d dl Z d dlmZmZ d dlmZ d dlmZmZ d dl m
Z
d dl m Z d dl
mZmZ d dlmZ G d d
� d
e�Zdd d
�ZG dd� de�Zdi dddd dddf dd�ZdS )� N)� DeferredList�Deferred)�ProcessProtocol)�Process�
ProcessReader)�reactor)�Failure)�
itervalues�
networkString)�
encode_valuesc @ s e Zd ZdZdS )� SignalErrorz3An error if the process was terminated by a signal.N)�__name__�
__module__� __qualname__�__doc__� r r �</usr/lib/python3/dist-packages/landscape/lib/twisted_util.pyr
s r Fc C s. t | d|d�}|�dd� � |�dd� � |S )N� )�fireOnOneErrback�
consumeErrorsc S s dd� | D �S )Nc S s g | ]}|d �qS )r r )�.0�xr r r �
<listcomp> s z4gather_results.<locals>.<lambda>.<locals>.<listcomp>r )�rr r r �<lambda> s z gather_results.<locals>.<lambda>c S s | j jS �N)�value�
subFailure)�fr r r r s )r � addCallback�
addErrback)� deferreds�consume_errors�dr r r �gather_results s �r$ c @ s2 e Zd ZdZd dd�Zdd� Zdd� Zd d
� ZdS ) �AllOutputProcessProtocolz<A process protocol for getting stdout, stderr and exit code.Nc C s: || _ t�� | _t�� | _| jj| _|| _|| _d| _ d S �N� )
�deferred�io�BytesIO�outBuf�errBuf�write� errReceived�stdin�
line_received�
_partial_line)�selfr( r/ r0 r r r �__init__ s
z!AllOutputProcessProtocol.__init__c C s. | j d ur| j�t| j �� | j�� d S d S r )r/ � transportr- r
�
closeStdin)r2 r r r �connectionMade% s
�z'AllOutputProcessProtocol.connectionMadec C sX | j �|� | jd u r
d S |�d�}| j|d |d< |�� | _|D ]}| �|� q"d S )N�
r )r+ r- r0 �splitr1 �pop)r2 �data�lines�liner r r � outReceived* s
�z$AllOutputProcessProtocol.outReceivedc C sv | j r | �| j � d| _ | j�� }| j�� }|j}|j}|jr0tt |||j��}| j
� |� d S | j
� |||f� d S r& )
r1 r0 r+ �getvaluer, r �exitCode�signalr r r( �errback�callback)r2 �reason�out�err�e�code�failurer r r � processEnded: s
z%AllOutputProcessProtocol.processEnded)NN)r
r r r r3 r6 r= rI r r r r r% s
r% r Tc
s` | g}
|
� |� t� } t| | |d�} t|�}tj| | |
|||||d�� |s.� fdd�}
|
� _| S )a�
Spawn a process using Twisted reactor.
Return a deferred which will be called with process stdout, stderr and exit
code.
@param wait_pipes: if set to False, don't wait for stdin/stdout pipes to
close when process ends.
@param line_received: an optional callback called with every line of
output from the process as parameter.
@note: compared to reactor.spawnProcess, this version does NOT require the
executable name as first element of args.
)r/ r0 )�args�env�path�uid�gid�usePTYc sJ � j r � js t� j �D ]} t| t�r| �� | �� q i � _ t�� � dS )a� A less strict version of Process.maybeCallProcessEnded.
This behaves exactly like the original method, but in case the
process has ended already and sent us a SIGCHLD, it doesn't wait
for the stdin/stdout pipes to close, because the child process
itself might have passed them to its own child processes.
@note: Twisted 8.2 now has a processExited hook that could
be used in place of this workaround.
N) �pipes�pidr �
isinstancer �doRead� stopReadingr �maybeCallProcessEnded)�pipe��processr r rU h s
z,spawn_process.<locals>.maybeCallProcessEnded)�extendr r% r r � spawnProcessrU )�
executablerJ rK rL rM rN rO �
wait_pipesr0 r/ � list_args�result�protocolrU r rW r �
spawn_processI s
�
� r` )F)r) �twisted.internet.deferr r �twisted.internet.protocolr �twisted.internet.processr r �twisted.internetr �twisted.python.failurer �twisted.python.compatr r
�landscape.lib.encodingr � Exceptionr r$ r% r` r r r r �<module> s
0�