File: //lib/python3/dist-packages/twisted/internet/__pycache__/_dumbwin32proc.cpython-38.pyc
U
W[33 � @ sJ d Z ddlmZmZmZ ddlZddlZddlZddlZddl Z ddl
Z
ddl Z ddl Z ddl
Z
ddlZe
�� Zde_ddlmZ ddlmZmZmZ ddlmZmZ ddlm Z dd l m Z dd
l m Z dd l m!Z! dd l"m#Z# e ee$ed
d��dd� �Z%dd� Z&G dd� de!j'�Z(dd� Z)dd� Z*eeee�G dd� de!j+e#��Z,dS )z-
http://isometri.cc/strips/gates_in_the_head
� )�absolute_import�division�print_functionN� )� implementer)�IProcessTransport� IConsumer� IProducer)�items�_PY3)�quoteArguments)�
_replaceIf)�error)� _pollingfile)� BaseProcess�fsdecodec C s t | t�r| �t�� �S | S dS )z�
Decode a string to a L{unicode} representation, passing
through existing L{unicode} unchanged.
@param x: The string to be conditionally decoded.
@type x: L{bytes} or L{unicode}
@return: L{unicode}
N)�
isinstance�bytes�decode�sys�getfilesystemencoding)�x� r �A/usr/lib/python3/dist-packages/twisted/internet/_dumbwin32proc.py� _fsdecode* s
r c C s t | � tj�� d S �N)�printr �stdout�flush)�msgr r r �debug< s r c @ s e Zd Zdd� Zdd� ZdS )�_Reaperc C s
|| _ d S r )�proc)�selfr" r r r �__init__D s z_Reaper.__init__c C s@ t �| jjd�t jkrdS t�| jj�}| �� | j�|� dS �Nr ) �
win32eventZWaitForSingleObjectr" �hProcessZ
WAIT_OBJECT_0� win32processZGetExitCodeProcessZ
deactivateZ processEnded)r# ZexitCoder r r � checkWorkH s z_Reaper.checkWorkN)�__name__�
__module__� __qualname__r$ r) r r r r r! B s r! c
C sH t | d��4}|�d�dkr:|�d��d�}|W 5 Q R � S W 5 Q R X dS )a
Look for a #! line, and return the value following the #! if one exists, or
None if this file is not a script.
I don't know if there are any conventions for quoting in Windows shebang
lines, so this doesn't support any; therefore, you may not pass any
arguments to scripts invoked as filters. That's probably wrong, so if
somebody knows more about the cultural expectations on Windows, please feel
free to fix.
This shebang line support was added in support of the CGI tests;
appropriately enough, I determined that shebang lines are culturally
accepted in the Windows world through this page::
http://www.cgi101.com/learn/connect/winxp.html
@param filename: str representing a filename
@return: a str representing another filename.
ZrU� z#!i �
N)�open�read�readline�strip)�filename�f�exer r r � _findShebangR s r6 c C s | j d dkS )z�
Determine if a pywintypes.error is telling us that the given process is
'not a valid win32 application', i.e. not a PE format executable.
@param pywinerr: a pywintypes.error instance raised by CreateProcess
@return: a boolean
r � )�args)Zpywinerrr r r �_invalidWin32Appn s r9 c @ s� e Zd ZdZdZdd� Zdd� Zdd� Zd d
� Zd d � Z d
d� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+d,� Zd-d.� Zd/S )0�Processa�
A process that integrates with the Twisted event loop.
If your subprocess is a python program, you need to:
- Run python.exe with the '-u' command line option - this turns on
unbuffered I/O. Buffering stdout/err/in can cause problems, see e.g.
http://support.microsoft.com/default.aspx?scid=kb;EN-US;q1903
- If you don't want Windows messing with data passed over
stdin/out/err, set the pipes to be in binary mode::
import os, sys, mscvrt
msvcrt.setmode(sys.stdin.fileno(), os.O_BINARY)
msvcrt.setmode(sys.stdout.fileno(), os.O_BINARY)
msvcrt.setmode(sys.stderr.fileno(), os.O_BINARY)
r c
s> t j��|� t��|� t�� }d|_t�|d�\�_ }t�|d�\�_
} t�|d�\}
�_ t� �j tj
dd� t�� � |� _| � _|
� _tj� _t�� } t�| �j | ddtj�} t��j � | �_ t�| �j
| ddtj�} t��j
� | �_
t�| �j | ddtj�} t��j � | �_ t j � � ��� |�p0i � i }
t ��D ]"\}}t!|�}t!|�}||
|< �q@|
�dd� |D �}t"|����r�t!��n����r�t!��n��� �����fdd�}z
|� W n� t#j$k
�r� } z�t%|��s�t&|��n�t'��}|dk�r t&d� ��nxt(|�}|�)d�� t"|���}|�z
|� W nH t#j$k
�r� } z&t%|��rjt&d ||f ��t&|��W 5 d}~X Y nX W 5 d}~X Y nX t�| � t�|� t�|
� t �*�j �fd
d ��j+��_,t �*�j
�fd d ��j-��_.t �/�j �j0��_1�j,�j.�j1fD ]}��2|� �q�j3�4�� ��2t5��� dS )
z-
Create a new child process.
r r Nc S s g | ] }t |��qS r )r )�.0r r r r �
<listcomp>� s z$Process.__init__.<locals>.<listcomp>c s2 t j} t���d d d| ��� � \�_�_�_}d S �Nr )�win32conZCREATE_NO_WINDOWr( Z
CreateProcessr' �hThread�pid)�flagsZdwTid�Z StartupInfoZcmdline�command�env�pathr# r r �doCreate� s �z"Process.__init__.<locals>.doCreatezD%r is neither a Windows executable, nor a script with a shebang linez<%r has an invalid shebang line: %r is not a valid executablec s � j �d| �S r= ��protoZchildDataReceived��data�r# r r �<lambda> � z"Process.__init__.<locals>.<lambda>c s � j �d| �S �Nr- rG rI rK r r rL rM )6r �
_PollingTimerr$ r �
win32security�SECURITY_ATTRIBUTES�bInheritHandle� win32pipeZ
CreatePipeZhStdoutRZhStderrRZhStdinWZSetNamedPipeHandleStateZ PIPE_NOWAITr( Z STARTUPINFOZ
hStdOutputZ hStdErrorZ hStdInputZSTARTF_USESTDHANDLESZdwFlags�win32apiZGetCurrentProcessZDuplicateHandler> ZDUPLICATE_SAME_ACCESS� win32file� CloseHandle�os�environ�copy�updater
r r �
pywintypesr r9 �OSErrorr6 �list�insertZ_PollableReadPipe�outConnectionLostr �errConnectionLost�stderrZ_PollableWritePipe�inConnectionLost�stdinZ_addPollableResourcerH ZmakeConnectionr! )r# ZreactorZprotocolrC r8 Z environmentrE ZsAttrsZhStdoutWZhStderrWZhStdinRZ
currentPidZtmpZnewenv�key�valuerF ZpwteZshebZorigcmdZpwte2Z pipewatcherr rB r r$ � s� �� � �
��
���,
�
� � zProcess.__init__c C s, | j d krt�� �|dkr(t�| jd� d S )N)ZINTZTERMZKILLr )r@ r ZProcessExitedAlreadyr( ZTerminateProcessr' )r# ZsignalIDr r r �
signalProcess s
zProcess.signalProcessc C s |dkrt �|�S t �|�S r% )r Z ProcessDoneZProcessTerminated)r# Zstatusr r r �
_getReason& s
zProcess._getReasonc C s | j �|� dS )zQ
Write data to the process' stdin.
@type data: C{bytes}
N)rc �write)r# rJ r r r rh , s z
Process.writec C s | j �|� dS )z\
Write data to the process' stdin.
@type data: C{list} of C{bytes}
N)rc �
writeSequence)r# �seqr r r ri 5 s zProcess.writeSequencec C s" |dkr| j �|� nt|��dS )a;
Similar to L{ITransport.write} but also allows the file descriptor in
the child process which will receive the bytes to be specified.
This implementation is limited to writing to the child's standard input.
@param fd: The file descriptor to which to write. Only stdin (C{0}) is
supported.
@type fd: C{int}
@param data: The bytes to write.
@type data: C{bytes}
@return: L{None}
@raise KeyError: If C{fd} is anything other than the stdin file
descriptor (C{0}).
r N)rc rh �KeyError)r# �fdrJ r r r � writeToChild> s zProcess.writeToChildc C sB |dkr| � � n,|dkr$| �� n|dkr6| �� ntd��d S )Nr r r- z4Only standard-IO file descriptors available on win32)�
closeStdin� closeStdout� closeStderr�NotImplementedError)r# rl r r r � closeChildFDW s
zProcess.closeChildFDc C s | j �� dS )z"Close the process' stdin.
N)rc �closerK r r r rn b s zProcess.closeStdinc C s | j �� d S r )ra rs rK r r r rp h s zProcess.closeStderrc C s | j �� d S r )r rs rK r r r ro l s zProcess.closeStdoutc C s | � � | �� | �� dS )z8
Close the process' stdout, in and err.
N)rn ro rp rK r r r �loseConnectionp s zProcess.loseConnectionc C s | j �d� | �� d S r= �rH ZchildConnectionLost�connectionLostNotifyrK r r r r_ y s zProcess.outConnectionLostc C s | j �d� | �� d S rN ru rK r r r r` ~ s zProcess.errConnectionLostc C s | j �d� | �� d S r% ru rK r r r rb � s zProcess.inConnectionLostc C s | j d7 _ | �� dS )zS
Will be called 3 times, by stdout/err threads and process handle.
r N)�closedNotifies�maybeCallProcessEndedrK r r r rv � s z Process.connectionLostNotifyc C sB | j dkr>| jr>t�| j� t�| j� d | _d | _t�| � d S )N� )rw Z lostProcessrU rV r' r? r rx rK r r r rx � s z Process.maybeCallProcessEndedc C s | j �||� d S r )rc �registerProducer)r# ZproducerZ streamingr r r rz � s zProcess.registerProducerc C s | j �� d S r )rc �unregisterProducerrK r r r r{ � s zProcess.unregisterProducerc C s | � � d S r )Z_pauserK r r r �pauseProducing� s zProcess.pauseProducingc C s | � � d S r )Z_unpauserK r r r �resumeProducing� s zProcess.resumeProducingc C s | � � d S r )rt rK r r r �
stopProducing� s zProcess.stopProducingc C s d| j j| jf S )z@
Return a string representation of the process.
z <%s pid=%s>)� __class__r* r@ rK r r r �__repr__� s zProcess.__repr__N)r* r+ r, �__doc__rw r$ rf rg rh ri rm rr rn rp ro rt r_ r` rb rv rx rz r{ r| r} r~ r� r r r r r: � s2
r: )-r� Z
__future__r r r rW r rT r> r& rU rS r( rP r[ rQ ZPIPE_ATTRS_INHERITABLErR Zzope.interfacer Ztwisted.internet.interfacesr r r Ztwisted.python.compatr
r Ztwisted.python.win32r Ztwisted.python.utilr
Ztwisted.internetr r Z twisted.internet._baseprocessr �getattrr r Z_PollableResourcer! r6 r9 rO r: r r r r �<module> s: