File: //usr/lib/python3/dist-packages/twisted/internet/iocpreactor/__pycache__/tcp.cpython-38.pyc
U
��W[�N � @ s� d Z ddlZddlZddlZddlZddlmZmZ ddlm Z m
Z
m Z m Z m
Z
ddlmZ ddlmZmZ ddlmZmZmZmZmZmZmZ ddl m Z m Z m Z dd l m!Z!m"Z" dd
l#m$Z%m&Z& dd l'm(Z( dd l)m*Z* dd
l)m+Z+ ddl)m,Z, ddl)m-Z- ddl)m.Z. zddl/m0Z1 W n e2k
�r6 dZ1Y nX e-ej3e.ej4iZ5ee(e j6e j7�G dd� de&j8ee��Z9e1dk �r�ee9e j:� G dd� deee9�Z;G dd� de9�Z<G dd� de�Zee j=�G dd� dee��Z>dS ) z
TCP support for IOCP reactor
� N)� implementer�classImplements)�
interfaces�error�address�main�defer)�Protocol)� _LogOwner�
isIPv6Address)�
_SocketCloser� Connector�_AbortingMixin�_BaseBaseClient�_BaseTCPClient� _resolveIPv6� _getsockname)�log�failure�reflect)�_PY3� nativeString)� iocpsupport�abstract)�IReadWriteHandle)�ERROR_IO_PENDING)�SO_UPDATE_CONNECT_CONTEXT)�SO_UPDATE_ACCEPT_CONTEXT)�ERROR_CONNECTION_REFUSED)�ERROR_NETWORK_UNREACHABLE)�startTLSc @ s� e Zd ZdZdZd+dd�Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd� Zd d � Zedk r�d,d d �Zd!d"� Zd#d$� Zd-d%d&�Zd'd(� Zd)d*� ZdS ).�
Connectionz�
@ivar TLS: C{False} to indicate the connection is in normal TCP mode,
C{True} to indicate that TLS has been started and that operations must
be routed through the L{TLSMemoryBIOProtocol} instance.
FNc C s& t j�| |� || _|j| _|| _d S �N)r �
FileHandle�__init__�socket�fileno�
getFileHandle�protocol)�self�sock�proto�reactor� r- �B/usr/lib/python3/dist-packages/twisted/internet/iocpreactor/tcp.pyr$ 3 s zConnection.__init__c C s | j S r" )r% �r) r- r- r. � getHandle: s zConnection.getHandlec C s@ t |t�r n$t |t�r t|�}ntdt|� ��| j�|� dS )z`
@param rbuffer: Data received.
@type rbuffer: L{bytes} or L{bytearray}
z%data must be bytes or bytearray, not N)�
isinstance�bytes� bytearray� TypeError�typer( � dataReceived)r) Zrbufferr- r- r. r6 > s
�zConnection.dataReceivedc C s t �| �� ||�S r" )�_iocpZrecvr' )r) Zbufflist�evtr- r- r. �readFromHandleO s zConnection.readFromHandlec C s( t |�}t�| �� |d| j� �� |�S )z�
Send C{buff} to current file handle using C{_iocp.send}. The buffer
sent is limited to a size of C{self.SEND_LIMIT}.
r )�
memoryviewr7 �sendr' Z
SEND_LIMIT�tobytes)r) �buffr8 Z writeViewr- r- r. �
writeToHandleS s
�zConnection.writeToHandlec C sr z| j �d� W n t jk
r& Y nX t�| jd �}|rnz |�� W n& t�� }t �
� | � |� Y nX d S )N� ) r% Zshutdownr r �IHalfCloseableProtocolr( ZwriteConnectionLostr �Failurer �err�connectionLost)r) �p�fr- r- r. �_closeWriteConnection] s z Connection._closeWriteConnectionc C sR t �| jd �}|rDz |�� W qN t�� | �t�� � Y qNX n
| �|� d S r" ) r r@ r( �readConnectionLostr rB rC r rA )r) �reasonrD r- r- r. rG l s z Connection.readConnectionLostc C sX | j r
d S tj�| |� |d kp,|�tj� }| �|� | j}| `| ` | `
|�|� d S r" ) � disconnectedr r# rC Zcheckr ZConnectionAborted� _closeSocketr( r% r' )r) rH ZisCleanr( r- r- r. rC x s �
zConnection.connectionLostc C s | j S )zN
Return the prefix to log with when I own the logging thread.
)�logstrr/ r- r- r. � logPrefix� s zConnection.logPrefixc C s t �| j�tjtj��S r" )�operator�truthr% �
getsockopt� IPPROTO_TCP� TCP_NODELAYr/ r- r- r. �
getTcpNoDelay� s �zConnection.getTcpNoDelayc C s | j �t jt j|� d S r" )r% �
setsockoptrP rQ �r) �enabledr- r- r. �
setTcpNoDelay� s zConnection.setTcpNoDelayc C s t �| j�tjtj��S r" )rM rN r% rO �
SOL_SOCKET� SO_KEEPALIVEr/ r- r- r. �getTcpKeepAlive� s �zConnection.getTcpKeepAlivec C s | j �t jt j|� d S r" )r% rS rW rX rT r- r- r. �setTcpKeepAlive� s zConnection.setTcpKeepAliveTc C s t | ||tj� dS )z9
@see: L{ITLSTransport.startTLS}
N)� _startTLSr r# )r) ZcontextFactoryZnormalr- r- r. r � s zConnection.startTLSc C s0 | j r
dS | jr | j�|� ntj�| |� dS )z�
Write some data, either directly to the underlying handle or, if TLS
has been started, to the L{TLSMemoryBIOProtocol} for it to encrypt and
send.
@see: L{twisted.internet.interfaces.ITransport.write}
N)rI �TLSr( �writer r# )r) �datar- r- r. r] � s
zConnection.writec C s0 | j r
dS | jr | j�|� ntj�| |� dS )z�
Write some data, either directly to the underlying handle or, if TLS
has been started, to the L{TLSMemoryBIOProtocol} for it to encrypt and
send.
@see: L{twisted.internet.interfaces.ITransport.writeSequence}
N)rI r\ r( �
writeSequencer r# )r) Ziovecr- r- r. r_ � s
zConnection.writeSequencec C s0 | j r | jr,| js,| j�� ntj�| |� dS )z�
Close the underlying handle or, if TLS has been started, first shut it
down.
@see: L{twisted.internet.interfaces.ITransport.loseConnection}
N)r\ � connected�
disconnectingr( �loseConnectionr r# )r) rH r- r- r. rb � s zConnection.loseConnectionc C s* | j r| j�||� ntj�| ||� dS )zc
Register a producer.
If TLS is enabled, the TLS connection handles this.
N)r\ r( �registerProducerr r# )r) ZproducerZ streamingr- r- r. rc � s zConnection.registerProducerc C s" | j r| j�� n tj�| � dS )ze
Unregister a producer.
If TLS is enabled, the TLS connection handles this.
N)r\ r( �unregisterProducerr r# r/ r- r- r. rd � s z Connection.unregisterProducer)N)T)N)�__name__�
__module__� __qualname__�__doc__r\ r$ r0 r6 r9 r> rF rG rC rL rR rV rY rZ r[ r r] r_ rb rc rd r- r- r- r. r! ( s,
r! c @ sT e Zd ZdZejZejZdZ e
Z dd� Z dd� Z
dd� Zd d
� Zd d � Zd
d� ZdS )�Clientz�
@ivar _tlsClientDefault: Always C{True}, indicating that this is a client
connection, and by default when TLS is negotiated this class will act as
a TLS client.
Tc C s* |d kr d}|| _ t�| |||||� d S )N)� r )r, r r$ )r) �host�port� bindAddress� connectorr, r- r- r. r$ s �zClient.__init__c C s | j �| j| j�S )zd
Create a socket registered with the IOCP reactor.
@see: L{_BaseTCPClient}
)r, � createSocket�
addressFamily�
socketTyper/ r- r- r. �createInternetSocket
s zClient.createInternetSocketc C s | ` | `dS )z�
Clean up potentially circular references to the socket and to its
C{getFileHandle} method.
@see: L{_BaseBaseClient}
N)r% r' r/ r- r- r. �_collectSocketDetails s z Client._collectSocketDetailsc C s | j �| � dS )z^
Remove the active handle from the reactor.
@see: L{_BaseBaseClient}
N)r, �removeActiveHandler/ r- r- r. �_stopReadingAndWriting s z Client._stopReadingAndWritingc C s� |r0t �||�}| �t�|tj�|d�f�� n~| j�tj t
t � d| j�
� �� | j�| �� �| _d| _| �| j�}|d | _| jd kr�t� | _| �� n| j�| � | �� d S )Nz
Unknown error�PTz,client)�connectExErrors�getZfailIfNotConnectedr ZgetConnectError�errno� errorcoder% rS rW r �struct�packr& rn �
buildProtocol�getPeerr( r` �
_getLogPrefixrK r rb �makeConnection� startReading)r) �rcr^ r8 rL r- r- r. � cbConnect& s&
�
�
zClient.cbConnectc C sf t | d�sd S tjst�| j�| � t�| j| �}t�| j �
� | j |�}|rb|t krb| �|d|� d S )Nrn r )
�hasattrr7 Zhave_connectex�AssertionErrorr, �addActiveHandle�Eventr� Zconnectr% r& Z realAddressr )r) r8 r� r- r- r. � doConnectB s
zClient.doConnectN)re rf rg rh r% �AF_INETrp � SOCK_STREAMrq �_tlsClientDefaultr! Z_commonConnectionr$ rr rs ru r� r� r- r- r- r. ri � s
ri c @ s4 e Zd ZdZdZdd� Zdd� Zdd� Zd d
� Zd S ) �ServeraV
Serverside socket-stream connection class.
I am a serverside network connection transport; a socket which came from an
accept() on a server.
@ivar _tlsClientDefault: Always C{False}, indicating that this is a server
connection, and by default when TLS is negotiated this class will act as
a TLS server.
Fc C sp t �| |||� || _|| _|| _| �| j�}d||| jjf | _d| jj j
| j| jj f | _ d| _
| �� dS )a
Server(sock, protocol, client, server, sessionno)
Initialize me with a socket, a protocol, a descriptor for my peer (a
tuple of host, port describing the other end of the connection), an
instance of Port, and a session number.
z%s,%s,%sz<%s #%s on %s>TN)r! r$ �
serverAddr�
clientAddr� sessionnor r( rk rK � __class__re rl �repstrr` r� )r) r* r( r� r� r� r, rL r- r- r. r$ ` s
�zServer.__init__c C s | j S )z=
A string representation of this connection.
)r� r/ r- r- r. �__repr__t s zServer.__repr__c C s | j S )zW
Returns an IPv4Address.
This indicates the server's address.
)r� r/ r- r- r. �getHost{ s zServer.getHostc C s | j S )zW
Returns an IPv4Address.
This indicates the client's address.
)r� r/ r- r- r. r~ � s zServer.getPeerN) re rf rg rh r� r$ r� r� r~ r- r- r- r. r� Q s r� c @ s e Zd Zdd� ZdS )r
c C s t | j| j| j| | j�S r" )ri rk rl rm r, r/ r- r- r. �_makeTransport� s �zConnector._makeTransportN)re rf rg r� r- r- r- r. r
� s r
c @ s� e Zd ZdZdZdZejZej Z
e j Z
dZdZdZd dd�Zd d
� Zd d � Ze�ej�fd
d�ZeZdd� Zdd� Zdd� Z dd� Z dd� Z dd� Z dd � Z dS ) �PortFr N�TCP�2 rj c C s: || _ || _|| _|| _|| _t|�r6tj| _t j
| _ d S r" ) rl �factory�backlog� interfacer, r r% �AF_INET6rp r Z IPv6Address� _addressType)r) rl r� r� r� r, r- r- r. r$ � s z
Port.__init__c C s6 | j d k r d| j| jj| j f S d| j| jjf S d S )Nz<%s of %s on %s>z<%s of %s (not listening)>)�_realPortNumberr� r� r/ r- r- r. r� � s
��z
Port.__repr__c
C s zH| j �| j| j�}| jtjkr0t| j| j�}n | j| jf}|� |� W n6 tj
k
r~ } zt
� | j| j|��W 5 d }~X Y nX t �
|�� �| _|�� d | _t�d| �| j�| jf � | j�� |�| j� d| _d| _| j �| � || _| jj| _ | � � d S )Nr? z%s starting on %sTF) r, ro rp rq r% r� r r� rl Zbindr ZCannotListenErrorr7 Z
maxAddrLenr& �addrLenZ getsocknamer� r �msgr r� ZdoStartZlistenr� r` rI r� r' �doAccept)r) ZsktZaddr�ler- r- r. �startListening� s.
� $�
zPort.startListeningc C s2 d| _ | jr.t�� | _| j�d| j|� | jS dS )z�
Stop accepting connections on this port.
This will shut down my socket and call self.connectionLost().
It returns a deferred which will fire successfully when the
port is actually closed.
Tr N)ra r` r ZDeferred�deferredr, Z callLaterrC )r) ZconnDoner- r- r. rb � s
zPort.loseConnectionc C s t �d| j| jf � dS )z.
Log message for closing port
z(%s Port %s Closed)N)r r� �_typer� r/ r- r- r. �_logConnectionLostMsg� s zPort._logConnectionLostMsgc C s� | � � d| _d}t| d�r&| j}| `d| _| j�| � d| _| �d� | ` | `
z| j � � W n, d| _
|dk r�|�t�� � n� Y nX d| _
|dk r�|�d� dS )z'
Cleans up the socket.
Nr� TF)r� r� r� r� rI r, rt r` rJ r% r' r� ZdoStopra Zerrbackr rA �callback)r) rH �dr- r- r. rC � s,
zPort.connectionLostc C s t �| jj�S )zK
Returns the name of my class, to prefix log entries with.
)r Zqualr� r� r/ r- r- r. rL s zPort.logPrefixc C s | j dt| j��� S )zf
Returns an IPv4Address or IPv6Address.
This indicates the server's address.
r� )r� )r� r r% r/ r- r- r. r� s z Port.getHostc C s$ | � ||� | js | js | �� d S r" )� handleAcceptra rI r� )r) r� r^ r8 r- r- r. �cbAccept' s z
Port.cbAcceptc
C s� | j s | jrdS |r4t�dtj�|d�|f � dS |j�t j
t t �
d| j �� �� t�|j�� |j�\}}}ts�t|d �|d f}t|d �|d f}|| jks�t�d|d kr�t|d �d�d �}|d |d d|f}d|d k�rt|d �d�d �}|d |d d|f}| j�| jd
|�� �}|d k�rB|j�� nD| j }|d | _ t |j|| jd |�� | jd |�� || j �} |� | � d S d S )
NFz*Could not accept new connection -- %s (%s)z
unknown errorrv r r? �%r� T)r� )r� )r� ) ra rI r r� ry rz rx �newsktrS r% rW r r{ r| r& r7 Zget_accept_addrsr= r r rp r� �int�splitr� r} r� �closer� r� r, r� )
r) r� r8 ZfamilyZlAddrZrAddrZscoper( �sZ transportr- r- r. r� - sP � � �
�
�
zPort.handleAcceptc C st t �| j| �}td| jd � |_}| j�| j| j � |_
}t � | j �
� |�
� ||�}|rp|tkrp| �||� d S )N� � )r7 r� r� r3 r� r= r, ro rp rq r� Zacceptr% r& r r� )r) r8 r= r� r� r- r- r. r� ^ s
�
z
Port.doAccept)r� rj N)!re rf rg r` rI ra r% r� rp r� rq r Z IPv4Addressr� r� r� r� r$ r� r� r rA r ZCONNECTION_DONErb Z
stopListeningr� rC rL r� r� r� r� r- r- r- r. r� � s*
1r� )?rh r% rM ry r{ Zzope.interfacer r Ztwisted.internetr r r r r Ztwisted.internet.protocolr Ztwisted.internet.abstractr
r Ztwisted.internet.tcpr r
Z TCPConnectorr r r r r Ztwisted.pythonr r r Ztwisted.python.compatr r Z twisted.internet.iocpreactorr r7 r Z'twisted.internet.iocpreactor.interfacesr Z"twisted.internet.iocpreactor.constr r r r r Ztwisted.internet._newtlsr r[ � ImportErrorZWSAECONNREFUSEDZWSAENETUNREACHrw Z
ITCPTransportZ
ISystemHandler# r! Z
ITLSTransportri r� ZIListeningPortr� r- r- r- r. �<module> sH $
�� G
\=