HEX

File: //usr/lib/python3/dist-packages/twisted/internet/__pycache__/posixbase.cpython-38.pyc
U

�b�k�@s�dZddlZddlZddlZddlZddlZddlmZddlm	Z	m
Z
mZmZddl
mZmZmZddlmZmZddlmZmZmZmZmZmZmZmZmZmZddlm Z m!Z!dd	l"m#Z#m$Z$m%Z%dd
l&m'Z'm(Z(e�)d�Z*e�)d�Z+zdd
l,m-Z.Wne/k
�rdZ-YnXe.Z-zddl
m0Z1Wne/k
�rFdZ0YnXe1Z0e(dkZ2dZ3e2�rzddl
m4Z4m5Z5m6Z6m7Z7dZ3e'�8��r�zddl9Z9dZ3Wne/k
�r�dZ9YnXGdd�de
�Z:ee:�Gdd�de$j;��Z<Gdd�de$j;�Z=ee:�Gdd�de=��Z>e(dk�re>Z?ne<Z?Gdd�de=�Z@Gdd�d�ZAeeee�Gdd �d eeAe��ZBGd!d"�d"�ZCee�Gd#d$�d$eCeA��ZDe-dk	�s�e0dk	�r�eeBe�e2�r�eeBee�e3�r�eeBe�eEed%d�dk	�r�eeBe�d gZFdS)&z
Posix reactor base class
�N)�Sequence)�	Attribute�	Interface�classImplements�implementer)�error�tcp�udp)�ReactorBase�_SignalReactorMixin)
�IHalfCloseableDescriptor�
IReactorFDSet�IReactorMulticast�IReactorProcess�IReactorSocket�IReactorSSL�IReactorTCP�IReactorUDP�IReactorUNIX�IReactorUNIXDatagram)�CONNECTION_DONE�CONNECTION_LOST)�failure�log�util)�platform�platformTypezHandler has no fileno methodzFile descriptor lost)�tls)�ssl�posixF)�_signals�fdesc�process�unixTc@s8eZdZdZed�Zdd�Zdd�Zej	d�dd	�Z
d
S)�_IWakeraQ
    Interface to wake up the event loop based on the self-pipe trick.

    The U{I{self-pipe trick}<http://cr.yp.to/docs/selfpipe.html>}, used to wake
    up the main loop from another thread or a signal handler.
    This is why we have wakeUp together with doRead

    This is used by threads or signals to wake up the event loop.
    �cCsdS)z:
        Called when the event should be wake up.
        N�r&r&r&�</usr/lib/python3/dist-packages/twisted/internet/posixbase.py�wakeUpWsz_IWaker.wakeUpcCsdS)zC
        Read some data from my connection and discard it.
        Nr&r&r&r&r'�doRead\sz_IWaker.doRead��reasoncCsdS)zB
        Called when connection was closed and the pipes.
        Nr&r*r&r&r'�connectionLostasz_IWaker.connectionLostN)�__name__�
__module__�__qualname__�__doc__r�disconnectedr(r)r�Failurer,r&r&r&r'r$Js

r$c@s4eZdZdZdZdd�Zdd�Zdd�Zd	d
�ZdS)�_SocketWakerz�
    The I{self-pipe trick<http://cr.yp.to/docs/selfpipe.html>}, implemented
    using a pair of sockets rather than pipes (due to the lack of support in
    select() on Windows for pipes), used to wake up the main loop from
    another thread.
    rc	Cs�||_t�tjtj�}|�tjtjd�t�t�tjtj���4}|�	d�|�
d�|�|���|�
�\}}W5QRX|�d�|�d�||_||_|jj|_dS)�Initialize.�)z	127.0.0.1rrN)�reactor�socket�AF_INETZSOCK_STREAMZ
setsockoptZIPPROTO_TCPZTCP_NODELAY�
contextlib�closing�bindZlisten�connectZgetsocknameZacceptZsetblocking�r�w�fileno)�selfr6ZclientZserver�readerZ
clientaddrr&r&r'�__init__rs �



z_SocketWaker.__init__c
CsNzt�|jjd�Wn4tk
rH}z|jdtjkr8�W5d}~XYnXdS)zSend a byte to my connection.�xrN)r�untilConcludesr>�send�OSError�args�errnoZWSAEWOULDBLOCK�r@�er&r&r'r(�s
z_SocketWaker.wakeUpcCs*z|j�d�Wntk
r$YnXdS)z4
        Read some data from my connection.
        � N)r=ZrecvrF�r@r&r&r'r)�sz_SocketWaker.doReadcCs|j��|j��dS�N)r=�closer>)r@r+r&r&r'r,�s
z_SocketWaker.connectionLostN)	r-r.r/r0r1rBr(r)r,r&r&r&r'r3gs	r3c@s4eZdZdZdZdZdZdd�Zdd�Zdd	�Z	dS)
�_FDWakera�
    The I{self-pipe trick<http://cr.yp.to/docs/selfpipe.html>}, used to wake
    up the main loop from another thread or a signal handler.

    L{_FDWaker} is a base class for waker implementations based on
    writing to a pipe being monitored by the reactor.

    @ivar o: The file descriptor for the end of the pipe which can be
        written to wake up a reactor monitoring this waker.

    @ivar i: The file descriptor which should be monitored in order to
        be awoken by this waker.
    rNcsX|�_t��\�_�_t��j�t��j�t��j�t��j��fdd��_dS)r4cs�jSrM)�ir&rLr&r'�<lambda>��z#_FDWaker.__init__.<locals>.<lambda>N)	r6�os�piperP�or!ZsetNonBlockingZ_setCloseOnExecr?�r@r6r&rLr'rB�sz_FDWaker.__init__cCst�|��dd��dS)zA
        Read some bytes from the pipe and discard them.
        cSsdSrMr&)�datar&r&r'rQ�rRz!_FDWaker.doRead.<locals>.<lambda>N)r!Z
readFromFDr?rLr&r&r'r)�sz_FDWaker.doReadc	CsPt|d�sdS|j|jfD](}zt�|�Wqtk
r@YqXq|`|`dS)zClose both ends of my pipe.rUN)�hasattrrPrUrSrNrF)r@r+�fdr&r&r'r,�s
z_FDWaker.connectionLost)
r-r.r/r0r1rPrUrBr)r,r&r&r&r'rO�s
rOc@seZdZdZdd�ZdS)�
_UnixWakerz�
    This class provides a simple interface to wake up the event loop.

    This is used by threads or signals to wake up the event loop.
    c
CsV|jdk	rRzt�tj|jd�Wn0tk
rP}z|jtjkr@�W5d}~XYnXdS)z)Write one byte to the pipe, and flush it.NrC)rUrrDrS�writerFrHZEAGAINrIr&r&r'r(�s
z_UnixWaker.wakeUpN)r-r.r/r0r(r&r&r&r'rZ�srZc@s0eZdZdZdd�Zdd�Zdd�Zdd	�Zd
S)�
_SIGCHLDWakerz}
    L{_SIGCHLDWaker} can wake up a reactor whenever C{SIGCHLD} is
    received.

    @see: L{twisted.internet._signals}
    cCst�||�dSrM)rOrBrVr&r&r'rB�sz_SIGCHLDWaker.__init__cCst�|j�dS)zJ
        Install the handler necessary to make this waker active.
        N)r �installHandlerrUrLr&r&r'�install�sz_SIGCHLDWaker.installcCst�d�dS)zC
        Remove the handler which makes this waker active.
        ���N)r r]rLr&r&r'�	uninstall�sz_SIGCHLDWaker.uninstallcCst�|�t��dS)a
        Having woken up the reactor in response to receipt of
        C{SIGCHLD}, reap the process which exited.

        This is called whenever the reactor notices the waker pipe is
        writeable, which happens soon after any call to the C{wakeUp}
        method.
        N)rOr)r"�reapAllProcessesrLr&r&r'r)�s	
z_SIGCHLDWaker.doReadN)r-r.r/r0rBr^r`r)r&r&r&r'r\�s
r\c@s<eZdZdZeje�e���eje�e���ifdd�Z	dS)�_DisconnectSelectableMixinz>
    Mixin providing the C{_disconnectSelectable} method.
    cCst|�|�|�|j�}|rV|r@|jtjkr@t�|�r@|�|�qp|�|�|�	|�n|�|�|�	t
�|��dS)z�
        Utility function for disconnecting a selectable.

        Supports half-close notification, isRead should be boolean indicating
        whether error resulted from doRead().
        N)�removeReader�get�	__class__r�ConnectionDoner�
providedByZreadConnectionLost�removeWriterr,rr2)r@�
selectable�whyZisReadZfaildict�fr&r&r'�_disconnectSelectables
�
��

z0_DisconnectSelectableMixin._disconnectSelectableN)
r-r.r/r0rrfrr2ZConnectionLostrlr&r&r&r'rbs
��rbc@seZdZUdZeZdd�ZdZdd�Zdd�Z	d	idddd
dfdd�Z
d0dd�Zd1dd�Zd2dd�Z
d3dd�Zd4dd�Zd5dd�Zer�ejejejfZeejed<nejejfZd d!�Zd"d#�Zd6d$d%�Zd7d&d'�Zd8d(d)�Zd9d*d+�Zd:d,d-�Z d.d/�Z!dS);�PosixReactorBasez�
    A basis for reactors that use file descriptors.

    @ivar _childWaker: L{None} or a reference to the L{_SIGCHLDWaker}
        which is used to properly notice child process termination.
    cCs0|js,|�|�|_|j�|j�|�|j�dS)z�
        Install a `waker' to allow threads and signals to wake up the IO thread.

        We use the self-pipe trick (http://cr.yp.to/docs/selfpipe.html) to wake
        the reactor. On Windows we use a pair of sockets.
        N)Zwaker�
_wakerFactory�_internalReaders�add�	addReaderrLr&r&r'�installWaker?szPosixReactorBase.installWakerNcCsVt�|�tdkrRtrR|js@t|�|_|j�|j�|�|j�|j�	�t
��dS)z�
        Extend the basic signal handling logic to also support
        handling SIGCHLD to know when to try to reap child processes.
        rN)r�_handleSignalsr�processEnabled�_childWakerr\rorprqr^r"rarLr&r&r'rsMs


zPosixReactorBase._handleSignalscCs|jr|j��dS)a�
        If a child waker was created and installed, uninstall it now.

        Since this disables reactor functionality and is only called
        when the reactor is stopping, it doesn't provide any directly
        useful functionality, but the cleanup of reactor-related
        process-global state that it does helps in unit tests
        involving multiple reactors and is generally just a nice
        thing.
        N)rur`rLr&r&r'�_uninstallHandler`sz"PosixReactorBase._uninstallHandlerr&rc
Cs�tdkrR|r6|	dk	rtd��t�|||||||||�	St�|||||||||	�	Snttdkr�|dk	rjtd��|dk	rztd��|r�td��|	r�td��tr�dd	lm}
|
||||||�Std
��ntd��dS)Nrz1Using childFDs is not supported with usePTY=True.�win32z,Setting UID is unsupported on this platform.z,Setting GID is unsupported on this platform.z1The usePTY parameter is not supported on Windows.z1Customizing childFDs is not supported on Windows.r)�Processz:spawnProcess not available since pywin32 is not installed.z0spawnProcess only available on Windows or POSIX.)r�
ValueErrorr"Z
PTYProcessrx�win32processZtwisted.internet._dumbwin32proc�NotImplementedError)r@ZprocessProtocol�
executablerG�env�pathZuid�gidZusePTYZchildFDsrxr&r&r'�spawnProcessss\�����zPosixReactorBase.spawnProcessr%rKcCst�|||||�}|��|S)z�Connects a given L{DatagramProtocol} to the given numeric UDP port.

        @returns: object conforming to L{IListeningPort}.
        )r	�Port�startListening)r@�port�protocol�	interface�
maxPacketSize�pr&r&r'�	listenUDP�szPosixReactorBase.listenUDPFcCs t�||||||�}|��|S)z�Connects a given DatagramProtocol to the given numeric UDP port.

        EXPERIMENTAL.

        @returns: object conforming to IListeningPort.
        )r	Z
MulticastPortr�)r@r�r�r�r�ZlistenMultipler�r&r&r'�listenMulticast�s	�z PosixReactorBase.listenMulticast�cCs*tstd��t�|||||�}|��|S�N�UNIX support is not present)�unixEnabled�AssertionErrorr#�	Connectorr<)r@�address�factory�timeoutZcheckPID�cr&r&r'�connectUNIX�szPosixReactorBase.connectUNIX�2�cCs,tstd��t�||||||�}|��|Sr�)r�r�r#r�r�)r@r�r��backlog�modeZwantPIDr�r&r&r'�
listenUNIX�szPosixReactorBase.listenUNIXcCs*tstd��t�|||||�}|��|S)z�
        Connects a given L{DatagramProtocol} to the given path.

        EXPERIMENTAL.

        @returns: object conforming to L{IListeningPort}.
        r�)r�r�r#ZDatagramPortr�)r@r�r�r�r�r�r&r&r'�listenUNIXDatagram�sz#PosixReactorBase.listenUNIXDatagramcCs,tstd��t�||||||�}|��|S)zd
        Connects a L{ConnectedDatagramProtocol} instance to a path.

        EXPERIMENTAL.
        r�)r�r�r#ZConnectedDatagramPortr�)r@r�r�r�r��bindAddressr�r&r&r'�connectUNIXDatagram�s�z$PosixReactorBase.connectUNIXDatagram�_supportedAddressFamiliescCsR||jkrt�|��tr4|tjkr4tj�|||�}nt	j�||||�}|�
�|S)a0
        Create a new L{IListeningPort} from an already-initialized socket.

        This just dispatches to a suitable port implementation (eg from
        L{IReactorTCP}, etc) based on the specified C{addressFamily}.

        @see: L{twisted.internet.interfaces.IReactorSocket.adoptStreamPort}
        )r�r�UnsupportedAddressFamilyr�r7�AF_UNIXr#r��_fromListeningDescriptorrr�)r@�fileDescriptor�
addressFamilyr�r�r&r&r'�adoptStreamPorts	

�z PosixReactorBase.adoptStreamPortcCsH||jkrt�|��tr2|tjkr2tj�|||�St	j�||||�SdS)zg
        @see:
            L{twisted.internet.interfaces.IReactorSocket.adoptStreamConnection}
        N)
r�rr�r�r7r�r#ZServerZ_fromConnectedSocketr)r@r�r�r�r&r&r'�adoptStreamConnections

�z&PosixReactorBase.adoptStreamConnectioncCs<|tjtjfkrt�|��tjj|||||d�}|��|S)N)r�)	r7r8�AF_INET6rr�r	r�r�r�)r@r�r�r�r�r�r&r&r'�adoptDatagramPort%s
�z"PosixReactorBase.adoptDatagramPortcCst�|||||�}|��|SrM)rr�r�)r@r�r�r�r�r�r&r&r'�	listenTCP3szPosixReactorBase.listenTCPcCs t�||||||�}|��|SrM)rr�r<)r@�hostr�r�r�r�r�r&r&r'�
connectTCP8szPosixReactorBase.connectTCPc		Csbtdk	r(t�|d|�}|�|||||�Stdk	rRt�|||||||�}|��|Sds^td��dS)NTF�SSL support is not present)r�TLSMemoryBIOFactoryr�rr�r<r�)	r@r�r�r��contextFactoryr�r��
tlsFactoryr�r&r&r'�
connectSSL?s �zPosixReactorBase.connectSSLcCshtdk	r0t�|d|�}|�||||�}d|_|Stdk	rXt�||||||�}|��|Sdsdtd��dS)NFZTLSr�)rr�r�Z_typerr�r�r�)r@r�r�r�r�r�r�r�r&r&r'�	listenSSLNszPosixReactorBase.listenSSLcCsJt|�|j}|D]}|�|�qt|�}|D]}|�|�q.t||B�S)ag
        Remove all readers and writers, and list of removed L{IReadDescriptor}s
        and L{IWriteDescriptor}s.

        Meant for calling from subclasses, to implement removeAll, like::

          def removeAll(self):
              return self._removeAll(self._reads, self._writes)

        where C{self._reads} and C{self._writes} are iterables.
        )�setrorcrh�list)r@ZreadersZwritersZremovedReadersrAZremovedWriters�writerr&r&r'�
_removeAll[szPosixReactorBase._removeAll)r%rK)r%rKF)r�r)r�r�r)rKr�)rKr�N)rK)r�r%)r�N)r�N)r�r%)"r-r.r/r0�_Wakerrnrrrursrvr�r�r�r�r�r�r�r�r7r8r�r�r�rZ
AddressFamily�__annotations__r�r�r�r�r�r�r�r�r&r&r&r'rm2sZ
	�
:
�



�
���


�


rmc@seZdZdZdd�ZdS)�_PollLikeMixina�
    Mixin for poll-like reactors.

    Subclasses must define the following attributes::

      - _POLL_DISCONNECTED - Bitmask for events indicating a connection was
        lost.
      - _POLL_IN - Bitmask for events indicating there is input to read.
      - _POLL_OUT - Bitmask for events indicating output can be written.

    Must be mixed in to a subclass of PosixReactorBase (for
    _disconnectSelectable).
    cCs�d}d}||j@r6||j@s6||jkr0d}t}q�t}npzF|��dkrJt}n0||j@r`|��}d}|sz||j@rz|�	�}d}Wn(t
k
r�t��d}t
��YnX|r�|�|||�dS)zg
        fd is available for read or write, do the work and raise errors if
        necessary.
        NFTr_r5)�_POLL_DISCONNECTED�_POLL_IN�_readsrrr?�_NO_FILEDESCr)�	_POLL_OUTZdoWrite�
BaseException�sys�exc_infor�errrl)r@rirY�eventrjZinReadr&r&r'�_doReadOrWrite�s*

z_PollLikeMixin._doReadOrWriteN)r-r.r/r0r�r&r&r&r'r�rsr�c@s|eZdZdZdZdZdZdd�Zdd�Zd	d
�Z	dd�Z
d
d�Zdd�Zdd�Z
dd�Zdd�Zdd�Zdd�Zdd�ZdS)�_ContinuousPollinga�
    Schedule reads and writes based on the passage of time, rather than
    notification.

    This is useful for supporting polling filesystem files, which C{epoll(7)}
    does not support.

    The implementation uses L{_PollLikeMixin}, which is a bit hacky, but
    re-implementing and testing the relevant code yet again is unappealing.

    @ivar _reactor: The L{EPollReactor} that is using this instance.

    @ivar _loop: A C{LoopingCall} that drives the polling, or L{None}.

    @ivar _readers: A C{set} of C{FileDescriptor} objects that should be read
        from.

    @ivar _writers: A C{set} of C{FileDescriptor} objects that should be
        written to.
    r5��cCs ||_d|_t�|_t�|_dSrM)�_reactor�_loopr��_readers�_writersrVr&r&r'rB�sz_ContinuousPolling.__init__cCsh|js|jrN|jdkrdddlm}m}||j�|_|j|j_|jj	|dd�n|jrd|j�
�d|_dS)zh
        Start or stop a C{LoopingCall} based on whether there are readers and
        writers.
        Nr)�_EPSILON�LoopingCallF)Znow)r�r�r�Ztwisted.internet.taskr�r��iterater�Zclock�start�stop)r@r�r�r&r&r'�
_checkLoop�s


z_ContinuousPolling._checkLoopcCsDt|j�D]}|�|||j�q
t|j�D]}|�|||j�q*dS)zX
        Call C{doRead} and C{doWrite} on all readers and writers respectively.
        N)r�r�r�r�r�r�)r@rAr�r&r&r'r��sz_ContinuousPolling.iteratecCs|j�|�|��dS)zU
        Add a C{FileDescriptor} for notification of data available to read.
        N)r�rpr��r@rAr&r&r'rq�sz_ContinuousPolling.addReadercCs|j�|�|��dS)zV
        Add a C{FileDescriptor} for notification of data available to write.
        N)r�rpr��r@r�r&r&r'�	addWriter�sz_ContinuousPolling.addWritercCs4z|j�|�Wntk
r&YdSX|��dS)zY
        Remove a C{FileDescriptor} from notification of data available to read.
        N)r��remove�KeyErrorr�r�r&r&r'rcs
z_ContinuousPolling.removeReadercCs4z|j�|�Wntk
r&YdSX|��dS)zb
        Remove a C{FileDescriptor} from notification of data available to
        write.
        N)r�r�r�r�r�r&r&r'rhs
z_ContinuousPolling.removeWritercCs(t|j|jB�}|j��|j��|S)z1
        Remove all readers and writers.
        )r�r�r��clear)r@�resultr&r&r'�	removeAlls

z_ContinuousPolling.removeAllcCs
t|j�S)z/
        Return a list of the readers.
        )r�r�rLr&r&r'�
getReaders&sz_ContinuousPolling.getReaderscCs
t|j�S)z/
        Return a list of the writers.
        )r�r�rLr&r&r'�
getWriters,sz_ContinuousPolling.getWriterscCs
||jkS)aj
        Checks if the file descriptor is currently being observed for read
        readiness.

        @param fd: The file descriptor being checked.
        @type fd: L{twisted.internet.abstract.FileDescriptor}
        @return: C{True} if the file descriptor is being observed for read
            readiness, C{False} otherwise.
        @rtype: C{bool}
        )r��r@rYr&r&r'�	isReading2sz_ContinuousPolling.isReadingcCs
||jkS)al
        Checks if the file descriptor is currently being observed for write
        readiness.

        @param fd: The file descriptor being checked.
        @type fd: L{twisted.internet.abstract.FileDescriptor}
        @return: C{True} if the file descriptor is being observed for write
            readiness, C{False} otherwise.
        @rtype: C{bool}
        )r�r�r&r&r'�	isWriting?sz_ContinuousPolling.isWritingN)r-r.r/r0r�r�r�rBr�r�rqr�rcrhr�r�r�r�r�r&r&r&r'r��s 	

r��fromfd)Gr0r9rHrSr7r��typingr�zope.interfacerrrr�twisted.internetrrr	Ztwisted.internet.baser
r�twisted.internet.interfacesrr
rrrrrrrr�twisted.internet.mainrr�twisted.pythonrrr�twisted.python.runtimerrZConnectionFdescWentAwayZ
_NO_FILENOr�Ztwisted.protocolsrZ_tls�ImportErrorrZ_sslr�rtr r!r"r#�	isWindowsrzr$ZLoggerr3rOrZr�r\rbrmr�r��getattr�__all__r&r&r&r'�<module>s�0





30
$&
AI