← Back
Editing: posixbase.cpython-38.pyc
U ��W[�f � @ s� d Z ddlmZmZ ddlZddlZddlZddlZddlm Z m Z ddl m Z m Z mZ ddlmZmZ ddlmZmZ ddlmZmZmZmZmZmZm Z m Z m 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+ W nF e,k �r6 dZ+zddl m-Z- W n e,k �r0 dZ-Y nX Y nX e%dkZ.dZ/e.�rnddl m0Z0m1Z1 ddl m2Z2m3Z3 dZ/e&�4� �r�zddl5Z5dZ/W n e,k �r� dZ5Y nX G dd� de!j6�Z7G dd� de!j6e8�Z9G dd� de9�Z:e%dk�r�e:Z;ne7Z;G dd� de9�Z<G d d � d e8�Z=e eee �G d d � d ee=e��Z>G d d!� d!e8�Z?e e �G d"d#� d#e?e=��Z@e+dk �sle-dk �rve e>e� e.�r�e e>ee� e/�r�e e>e � eAed$d�dk �r�e e>e� d gZBdS )%z Posix reactor base class � )�division�absolute_importN)� implementer�classImplements)�error�udp�tcp)� ReactorBase�_SignalReactorMixin)�CONNECTION_DONE�CONNECTION_LOST) � IReactorUNIX�IReactorUNIXDatagram� IReactorTCP� IReactorUDP� IReactorSSL�IReactorSocket�IHalfCloseableDescriptor�IReactorProcess�IReactorMulticast� IReactorFDSet)�log�failure�util)� platformType�platformz Handler has no fileno methodzFile descriptor lost)�tls)�ssl�posixF)�fdesc�unix)�process�_signalsTc @ s4 e Zd ZdZdZdd� Zdd� Zdd� Zd d � Zd S ) � _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. r c C s� || _ t�tjtj�}t�tjtj�}|�tjtjd� |�d� |�d� |� |� � � |� � \}}|� d� |� d� || _ || _| j j| _dS )�Initialize. � )z 127.0.0.1r r N)�reactor�socket�AF_INETZ SOCK_STREAMZ setsockoptZ IPPROTO_TCPZ TCP_NODELAYZbindZlisten�connectZ getsocknameZacceptZ setblocking�r�w�fileno)�selfr& ZserverZclient�readerZ clientaddr� r/ �</usr/lib/python3/dist-packages/twisted/internet/posixbase.py�__init__E s z_SocketWaker.__init__c C sP zt �| jjd� W n6 tjk rJ } z|jd tjkr:� W 5 d}~X Y nX dS )z&Send a byte to my connection. � xr N) r �untilConcludesr+ �sendr' r �args�errnoZWSAEWOULDBLOCK�r- �er/ r/ r0 �wakeUpW s z_SocketWaker.wakeUpc C s, z| j �d� W n tjk r& Y nX dS )z+Read some data from my connection. � N)r* Zrecvr' r �r- r/ r/ r0 �doRead` s z_SocketWaker.doReadc C s | j �� | j�� d S �N)r* �closer+ )r- �reasonr/ r/ r0 �connectionLosth s z_SocketWaker.connectionLostN) �__name__� __module__� __qualname__�__doc__� disconnectedr1 r9 r< r@ r/ r/ r/ r0 r# < s r# c @ s4 e Zd ZdZdZdZdZdd� 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. r Nc sX |� _ t�� \� _� _t�� j� t�� j� t�� j� t�� j� � fdd�� _dS )r$ c s � j S r= )�ir/ r; r/ r0 �<lambda>� � z#_FDWaker.__init__.<locals>.<lambda>N) r& �os�piperG �or ZsetNonBlockingZ_setCloseOnExecr, �r- r& r/ r; r0 r1 � s z_FDWaker.__init__c C s t �| �� dd� � dS )zA Read some bytes from the pipe and discard them. c S s d S r= r/ )�datar/ r/ r0 rH � rI z!_FDWaker.doRead.<locals>.<lambda>N)r Z readFromFDr, r; r/ r/ r0 r< � s z_FDWaker.doReadc C sP t | d�sdS | j| jfD ](}zt�|� W q tk r@ Y qX q| `| `dS )z$Close both ends of my pipe. rL N)�hasattrrG rL rJ r> �IOError)r- r? �fdr/ r/ r0 r@ � s z_FDWaker.connectionLost) rA rB rC rD rE rG rL r1 r< r@ r/ r/ r/ r0 rF n s rF c @ s e Zd ZdZdd� 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 C sV | j dk rRzt�tj| j d� W n0 tk rP } z|jtjkr@� W 5 d}~X Y nX dS )z2Write one byte to the pipe, and flush it. Nr2 )rL r r3 rJ �write�OSErrorr6 ZEAGAINr7 r/ r/ r0 r9 � s z_UnixWaker.wakeUpN)rA rB rC rD r9 r/ r/ r/ r0 rR � s rR c @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd S ) � _SIGCHLDWakerz} L{_SIGCHLDWaker} can wake up a reactor whenever C{SIGCHLD} is received. @see: L{twisted.internet._signals} c C s t �| |� d S r= )rF r1 rM r/ r/ r0 r1 � s z_SIGCHLDWaker.__init__c C s t �| j� dS )zJ Install the handler necessary to make this waker active. N)r" �installHandlerrL r; r/ r/ r0 �install� s z_SIGCHLDWaker.installc C s t �d� dS )zC Remove the handler which makes this waker active. ���N)r" rV r; r/ r/ r0 � uninstall� s z_SIGCHLDWaker.uninstallc C s t �| � 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)rF r< r! �reapAllProcessesr; r/ r/ r0 r< � s z_SIGCHLDWaker.doReadN)rA rB rC rD r1 rW rY r< r/ r/ r/ r0 rU � s rU c @ s< e Zd ZdZeje�e�� �eje�e�� �ifdd�Z dS )�_DisconnectSelectableMixinz> Mixin providing the C{_disconnectSelectable} method. c C st | � |� |�|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 Z providedByZreadConnectionLost� removeWriterr@ r �Failure)r- � selectable�whyZisReadZfaildict�fr/ r/ r0 �_disconnectSelectable� s � z0_DisconnectSelectableMixin._disconnectSelectableN) rA rB rC rD r r_ r ra ZConnectionLostre r/ r/ r/ r0 r[ � s �r[ c @ s� e Zd ZdZeZdd� ZdZdd� Zdd� Z d i dddd dfd d �Z d/dd�Z d0dd�Z d1dd�Z d2dd�Zd3dd �Zd4d d �Zer�ejejejfZn ejejfZd d � Zd!d"� Zd5d#d$�Zd6d%d&�Zd7d'd(�Zd8d)d*�Z d9d+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. c C s0 | j s,| �| �| _ | 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� addReaderr; r/ r/ r0 � installWaker s z PosixReactorBase.installWakerNc C sV t �| � 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. r N) r �_handleSignalsr �processEnabled� _childWakerrU rh ri rj rW r! rZ r; r/ r/ r0 rl " s z PosixReactorBase._handleSignalsc C s | j r| 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)rn rY r; r/ r/ r0 �_uninstallHandler5 s z"PosixReactorBase._uninstallHandlerr/ r c C s� | � ||�\}}tdkrb|rF| d k r,td��t�| ||||||||� S t�| |||||||| � S nttdkr�|d k rztd��|d k r�td��|r�td��| r�td��tr�dd lm} | | |||||�S td ��ntd ��d S ) Nr z1Using childFDs is not supported with usePTY=True.Zwin32z,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.) Z_checkProcessArgsr � ValueErrorr! Z PTYProcessrp � win32processZ twisted.internet._dumbwin32proc�NotImplementedError) r- ZprocessProtocol� executabler5 �env�pathZuid�gidZusePTYZchildFDsrp r/ r/ r0 � spawnProcessH sF � � ��z PosixReactorBase.spawnProcess� r: c C s t �||||| �}|�� |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/ r0 � listenUDPk s zPosixReactorBase.listenUDPFc C s 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/ r0 �listenMulticastv s z PosixReactorBase.listenMulticast� c C s* t s td��t�|||| |�}|�� |S �N�UNIX support is not present)� unixEnabled�AssertionErrorr � Connectorr) )r- �address�factory�timeoutZcheckPID�cr/ r/ r0 � connectUNIX� s z PosixReactorBase.connectUNIX�2 � c C s, t s td��t�||||| |�}|�� |S r� )r� r� r rz r{ )r- r� r� �backlog�modeZwantPIDr� r/ r/ r0 � listenUNIX� s zPosixReactorBase.listenUNIXc C s* t s td��t�||||| �}|�� |S )z� Connects a given L{DatagramProtocol} to the given path. EXPERIMENTAL. @returns: object conforming to L{IListeningPort}. r� )r� r� r Z DatagramPortr{ )r- r� r} r r� r� r/ r/ r0 �listenUNIXDatagram� s z#PosixReactorBase.listenUNIXDatagramc C s, t s td��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/ r0 �connectUNIXDatagram� s z$PosixReactorBase.connectUNIXDatagramc C sR || j krt�|��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} ) �_supportedAddressFamiliesr �UnsupportedAddressFamilyr� r' �AF_UNIXr rz �_fromListeningDescriptorr r{ )r- �fileDescriptor� addressFamilyr� r� r/ r/ r0 �adoptStreamPort� s � �z PosixReactorBase.adoptStreamPortc C sH || j krt�|��tr2|tjkr2tj�||| �S t j�|||| �S dS )zg @see: L{twisted.internet.interfaces.IReactorSocket.adoptStreamConnection} N) r� r r� r� r' r� r ZServerZ_fromConnectedSocketr )r- r� r� r� r/ r/ r0 �adoptStreamConnection� s � �z&PosixReactorBase.adoptStreamConnectionc C s< |t jt jfkrt�|��tjj| ||||d�}|�� |S )N)r ) r' r( �AF_INET6r r� r rz r� r{ )r- r� r� r} r r� r/ r/ r0 �adoptDatagramPort� s �z"PosixReactorBase.adoptDatagramPortc C s t �||||| �}|�� |S r= )r rz r{ )r- r| r� r� r~ r� r/ r/ r0 � listenTCP� s zPosixReactorBase.listenTCPc C s t �|||||| �}|�� |S r= )r r� r) )r- �hostr| r� r� r� r� r/ r/ r0 � connectTCP� s zPosixReactorBase.connectTCPc C sb t d k r(t �|d|�}| �|||||�S td k rRt�||||||| �}|�� |S ds^td��d S )NTF�SSL support is not present)r �TLSMemoryBIOFactoryr� r r� r) r� ) r- r� r| r� �contextFactoryr� r� � tlsFactoryr� r/ r/ r0 � connectSSL� s �zPosixReactorBase.connectSSLc C sh t d k r0t �|d|�}| �||||�}d|_|S td k rXt�|||||| �}|�� |S dsdtd��d S )NFZTLSr� )r r� r� Z_typer rz r{ r� )r- r| r� r� r� r~ r� r� r/ r/ r0 � listenSSL s" �zPosixReactorBase.listenSSLc C sJ t |�| 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. )�setrh r\ r` �list)r- ZreadersZwritersZremovedReadersr. ZremovedWriters�writerr/ r/ r0 � _removeAll s zPosixReactorBase._removeAll)ry r: )ry r: F)r� r )r� r� r )r: r� )r: r� N)r: )r� ry )r� N)r� N)r� ry ) rA rB rC rD �_Wakerrg rk rn rl ro rx r� r� r� r� r� r� r� r' r( r� r� r� r� r� r� r� r� r� r� r� r/ r/ r/ r0 rf sT � # � � � � � rf c @ s e Zd ZdZdd� 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). c C s� d}d}|| j @ r6|| j@ s6|| jkr0d}t}q�t}nhzF|�� dkrJt}n0|| j@ r`|�� }d}|sz|| j@ rz|� � }d}W n t � � d }t � � Y nX |r�| �|||� dS )zg fd is available for read or write, do the work and raise errors if necessary. NFTrX r% )�_POLL_DISCONNECTED�_POLL_INZ_readsr r r, � _NO_FILEDESCr< � _POLL_OUTZdoWrite�sys�exc_infor �errre )r- rb rQ Zeventrc ZinReadr/ r/ r0 �_doReadOrWrite< s* z _PollLikeMixin._doReadOrWriteN)rA rB rC rD r� r/ r/ r/ r0 r� - s r� c @ s| e Zd ZdZdZdZdZdd� Zdd� Zd d � Z d d � Z d d� Z dd� Z dd� Z dd� Zdd� Zdd� Zdd� Zdd � Zd S ) �_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. r% � � c C s || _ d | _t� | _t� | _d S r= )�_reactor�_loopr� �_readers�_writersrM r/ r/ r0 r1 � s z_ContinuousPolling.__init__c C sh | j s | 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 )� LoopingCall�_EPSILONF)Znow) r� r� r� Ztwisted.internet.taskr� r� �iterater� Zclock�start�stop)r- r� r� r/ r/ r0 � _checkLoop� s z _ContinuousPolling._checkLoopc C sD t | 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- r. r� r/ r/ r0 r� � s z_ContinuousPolling.iteratec C s | j �|� | �� dS )zU Add a C{FileDescriptor} for notification of data available to read. N)r� ri r� �r- r. r/ r/ r0 rj � s z _ContinuousPolling.addReaderc C s | j �|� | �� dS )zV Add a C{FileDescriptor} for notification of data available to write. N)r� ri r� �r- r� r/ r/ r0 � addWriter� s z _ContinuousPolling.addWriterc C s4 z| j �|� W n tk r& Y dS X | �� dS )zY Remove a C{FileDescriptor} from notification of data available to read. N)r� �remove�KeyErrorr� r� r/ r/ r0 r\ � s z _ContinuousPolling.removeReaderc C s4 z| j �|� W n tk r& Y dS X | �� dS )zb Remove a C{FileDescriptor} from notification of data available to write. N)r� r� r� r� r� r/ r/ r0 r` � s z _ContinuousPolling.removeWriterc C s( t | j| jB �}| j�� | j�� |S )z1 Remove all readers and writers. )r� r� r� �clear)r- �resultr/ r/ r0 � removeAll� s z _ContinuousPolling.removeAllc C s t | j�S )z/ Return a list of the readers. )r� r� r; r/ r/ r0 � getReaders� s z _ContinuousPolling.getReadersc C s t | j�S )z/ Return a list of the writers. )r� r� r; r/ r/ r0 � getWriters� s z _ContinuousPolling.getWritersc C s || j kS )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- rQ r/ r/ r0 � isReading� s z _ContinuousPolling.isReadingc C s || j kS )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/ r0 � isWriting s z _ContinuousPolling.isWritingN)rA rB rC rD r� r� r� r1 r� r� rj r� r\ r` r� r� r� r� r� r/ r/ r/ r0 r� w s r� Zfromfd)CrD Z __future__r r r' r6 rJ r� Zzope.interfacer r Ztwisted.internetr r r Ztwisted.internet.baser r Ztwisted.internet.mainr r Ztwisted.internet.interfacesr r r r r r r r r r Ztwisted.pythonr r r Ztwisted.python.runtimer r ZConnectionFdescWentAwayZ _NO_FILENOr� Ztwisted.protocolsr � ImportErrorr r� rm r r r! r" Z isWindowsrr ZLoggerr# �objectrF rR r� rU r[ rf r� r� �getattr�__all__r/ r/ r/ r0 �<module> sz 0 24 ( � (J
Save File
Cancel