File: //usr/lib/python3/dist-packages/twisted/internet/iocpreactor/__pycache__/tcp.cpython-310.pyc
o
�b�M � @ s� d Z ddlZddlZddlZddlmZ ddlmZmZ ddl m
Z
m Z m Z m
Z
mZ ddlmZmZ ddlmZmZ ddlmZmZmZmZmZ dd l m Z dd
l m Z dd l m!Z"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z( dd l)m*Z*m+Z+m,Z, zdd
l-m.Z/ W n e0y� dZ1Y nw e/Z1eej2eej3iZ4ee e
j5e
j6�G dd� dej7e(e#��Z8e1dur�ee8e
j9� G dd� de$e%e8�Z:G dd� de8�Z;G dd� de"�Z!ee
j<�G dd� de(e��Z=dS )z
TCP support for IOCP reactor
� N)�Optional)�classImplements� implementer)�address�defer�error�
interfaces�main)� _LogOwner�
isIPv6Address)�abstract� iocpsupport)�ERROR_CONNECTION_REFUSED�ERROR_IO_PENDING�ERROR_NETWORK_UNREACHABLE�SO_UPDATE_ACCEPT_CONTEXT�SO_UPDATE_CONNECT_CONTEXT)�IReadWriteHandle)�Protocol)� Connector�_AbortingMixin�_BaseBaseClient�_BaseTCPClient� _getsockname� _resolveIPv6�
_SocketCloser)�failure�log�reflect)�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 � ZedurFd0d d �Zd!d"� Zd#d$� Zd/d%d&�Zd'd(� Zd)d*� Zd+d,� Zd-d.� ZdS )1�
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# ? s
zConnection.__init__c C � | j S r! )r$ �r( r, r, r- � getHandleE s zConnection.getHandlec C s@ t |t�rnt |t�rt|�}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( �rbufferr, r, r- r6 H s
zConnection.dataReceivedc C s t �| �� ||�S r! )�_iocp�recvr& )r( �bufflist�evtr, r, r- �readFromHandleW 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 )�
memoryviewr8 �sendr& �
SEND_LIMIT�tobytes)r( �buffr; � writeViewr, r, r- �
writeToHandleZ s �zConnection.writeToHandlec C sx z| j �d� W n ty Y nw t�| jd �}|r:z|�� W d S ty9 t� � }t
� � | � |� Y d S w d S )N� )
r$ �shutdown�OSErrorr �IHalfCloseableProtocolr' �writeConnectionLost�
BaseExceptionr �Failurer �err�connectionLost)r( �p�fr, r, r- �_closeWriteConnectiond s � ��z Connection._closeWriteConnectionc C sZ t �| jd �}|r&z|�� W d S ty% t�� | �t� � � Y d S w | �|� d S r! )
r rG r' �readConnectionLostrI r rK rL r rJ )r( �reasonrM r, r, r- rP r s �z Connection.readConnectionLostc C sX | j rd S tj�| |� |d u p|�tj� }| �|� | j}| `| ` | `
|�|� d S r! ) � disconnectedr r" rL �checkr �ConnectionAborted� _closeSocketr' r$ r& )r( rQ �isCleanr' r, r, r- rL } s
zConnection.connectionLostc C r. )zN
Return the prefix to log with when I own the logging thread.
)�logstrr/ r, r, r- � logPrefix� � zConnection.logPrefixc C � t | j�tjtj��S r! )�boolr$ �
getsockopt� IPPROTO_TCP� TCP_NODELAYr/ r, r, r- �
getTcpNoDelay� � zConnection.getTcpNoDelayc C � | j �t jt j|� d S r! )r$ �
setsockoptr] r^ �r( �enabledr, r, r- �
setTcpNoDelay� � zConnection.setTcpNoDelayc C rZ r! )r[ r$ r\ �
SOL_SOCKET� SO_KEEPALIVEr/ r, r, r- �getTcpKeepAlive� r` zConnection.getTcpKeepAlivec C ra r! )r$ rb rg rh rc r, r, r- �setTcpKeepAlive� rf zConnection.setTcpKeepAliveTc C s t | ||tj� dS )z9
@see: L{ITLSTransport.startTLS}
N)� _startTLSr r" )r( �contextFactory�normalr, r, r- r � s zConnection.startTLSc C �2 | j rdS | jr| j�|� dS tj�| |� 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)rR �TLSr' �writer r" )r( �datar, r, r- rp � �
zConnection.writec C rn )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)rR ro r' �
writeSequencer r" )r( �iovecr, r, r- rs � rr zConnection.writeSequencec C s: | j r| jr| js| j�� dS dS dS tj�| |� dS )z�
Close the underlying handle or, if TLS has been started, first shut it
down.
@see: L{twisted.internet.interfaces.ITransport.loseConnection}
N)ro � connected�
disconnectingr' �loseConnectionr r" )r( rQ r, r, r- rw � s
�zConnection.loseConnectionc C s, | j r | j�||� dS tj�| ||� dS )zc
Register a producer.
If TLS is enabled, the TLS connection handles this.
N)ro r' �registerProducerr r" )r( �producer� streamingr, r, r- rx � s zConnection.registerProducerc C s$ | j r
| j�� dS tj�| � dS )ze
Unregister a producer.
If TLS is enabled, the TLS connection handles this.
N)ro r' �unregisterProducerr r" r/ r, r, r- r{ � s z Connection.unregisterProducerc C � d S r! r, r/ r, r, r- �getHost� � zConnection.getHostc C r| r! r, r/ r, r, r- �getPeer� r~ zConnection.getPeerr! )T)�__name__�
__module__� __qualname__�__doc__ro r# r0 r6 r< rC rO rP rL rX r_ re ri rj rk r rp rs rw rx r{ r} r r, r, r, r- r 5 s2
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 u rd}|| _ 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� |rt �||�}| �t�|tj�|d�f�� d S | j�tj t
t � d| j�
� �� | j�| �� �| _d| _| �| j�}|d | _| jd u rOt� | _| �� d S | j�| � | �� d S )Nz
Unknown error�PTz,client)�connectExErrors�get�failIfNotConnectedr �getConnectError�errno� errorcoder$ rb rg r �struct�packr% r� �
buildProtocolr r' ru �
_getLogPrefixrW r rw �makeConnection� startReading)r( �rcrq r; rX r, r, r- � cbConnect" s& ��
zClient.cbConnectc C sn t | d�sd S tjs J �| j�| � t�| j| �}t�| j� � | j
|�}|r3|t kr5| �|d|� d S d S d S )Nr� r ) �hasattrr8 �have_connectexr+ �addActiveHandle�Eventr� �connectr$ r% � realAddressr )r( r; r� r, r, r- � doConnect@ s
�zClient.doConnectN)r� r� r� r� r$ �AF_INETr� � SOCK_STREAMr� �_tlsClientDefaultr �_commonConnectionr# r� r� r� r� r� r, r, r, r- r� � s r� c @ s: e Zd ZdZdZdd� Zdefdd�Zdd � Zd
d � Z d 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 sv t �| |||� || _|| _|| _| �| j�}|� d|� d| jj� �| _d� | jj
j | j| jj �| _
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<{} #{} on {}>TN)r r# �
serverAddr�
clientAddr� sessionnor� r' r� rW �format� __class__r� r� �repstrru r� )r( r) r' r� r� r� r+ rX r, r, r- r# \ s � zServer.__init__�returnc C r. )z=
A string representation of this connection.
)r� r/ r, r, r- �__repr__r rY zServer.__repr__c C r. )zW
Returns an IPv4Address.
This indicates the server's address.
)r� r/ r, r, r- r} x � zServer.getHostc C r. )zW
Returns an IPv4Address.
This indicates the client's address.
)r� r/ r, r, r- r � r� zServer.getPeerN)
r� r� r� r� r� r# �strr� r} r r, r, r, r- r� N s r� c @ s e Zd Zdd� ZdS )r c C s t | j| j| j| | j�S r! )r� r� r� r� r+ r/ r, r, r- �_makeTransport� rf zConnector._makeTransportN)r� r� r� r� r, r, r, r- r � s r c @ s� e Zd ZU dZdZdZejZej Z
e j Z
dZdZee ed< dZd dd �Zd
efd d �Zd
d� Ze�ej�fdd�Z e Z dd� Z dd� Z dd� Z dd� Z!dd� Z"dd � Z#d d � Z$dS ) �PortFr N�_realPortNumber�TCP�2 r� c C s> || _ || _|| _|| _|| _t|�r tj| _t j
| _ d S d S r! ) r� �factory�backlog� interfacer+ r r$ �AF_INET6r� r � IPv6Address� _addressType)r( r� r� r� r� r+ r, r, r- r# � s �z
Port.__init__r� c C s2 | j d urd�| j| jj| j �S d�| j| jj�S )Nz<{} of {} on {}>z<{} of {} (not listening)>)r� r� r� r� r/ r, r, r- r� � s
��z
Port.__repr__c
C s� z$| j �| j| j�}| jtjkrt| j| j�}n| j| jf}|� |� W n t
y9 } z t � | j| j|��d }~ww t
�|�� �| _|�� d | _t�d| �| j�| jf � | j�� |�| j� d| _d| _| j � | � || _| jj| _ | � � d S )NrD z%s starting on %sTF) r+ r� r� r� r$ r� r r� r� �bindrF r �CannotListenErrorr8 �
maxAddrLenr% �addrLen� getsocknamer� r �msgr� r� �doStart�listenr� ru rR r� r& �doAccept)r( �skt�addr�ler, r, r- �startListening� s2 ����
zPort.startListeningc C s2 d| _ | jrt�� | _| 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)rv ru r �Deferred�deferredr+ � callLaterrL )r( �connDoner, r, r- rw � s
�zPort.loseConnectionc C s t �d| j� d| j� d�� dS )z.
Log message for closing port
�(z Port z 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 t
yH d| _|durG|�t�� � Y dS � w d| _|durW|�d� dS dS )z'
Cleans up the socket.
Nr� TF)r� r� r� r� rR r+ r� ru rU r$ r& r� �doStoprI rv �errbackr rJ �callback)r( rQ �dr, r, r- rL � s0
��zPort.connectionLostc C s t �| jj�S )zK
Returns the name of my class, to prefix log entries with.
)r �qualr� r� r/ r, r, r- rX s zPort.logPrefixc C s | j dgt| j��R � S )zf
Returns an IPv4Address or IPv6Address.
This indicates the server's address.
r� )r� r r$ r/ r, r, r- r} s z Port.getHostc C s, | � ||� | js| js| �� d S d S d S r! )� handleAcceptrv rR r� )r( r� rq r; r, r, r- �cbAccept s �z
Port.cbAcceptc
C sh | j s| jrdS |rt�dtj�|d�|f � dS |j�t j
t t �
d| j �� �� t�|j�� |j�\}}}|| jks>J �d|d v rYt|d �d�d �}|d |d d|f}d|d v rtt|d �d�d �}|d |d d|f}| j�| jdg|�R � �}|d u r�|j�� d S | j}|d | _t|j|| jdg|�R � | jdg|�R � || j�} |� | � d S )
NFz*Could not accept new connection -- %s (%s)z
unknown errorr� �%r rD r� T) rv rR r r� r� r� r� �newsktrb r$ rg r r� r� r% r8 �get_accept_addrsrA r� �int�splitr� r� r� �closer� r� r+ r� )
r( r� r; �family�lAddr�rAddr�scoper' �s� transportr, r, r- r� " sL ���
�
�
zPort.handleAcceptc C s| t �| j| �}td| jd � |_}| j�| j| j � |_
}t � | j �
� |�
� ||�}|r:|tkr<| �||� d S d S d S )N� � )r8 r� r� r3 r� rA r+ r� r� r� r� �acceptr$ r% r r� )r( r; rA r� r� r, r, r- r� U s
� �z
Port.doAccept)r� r� N)%r� r� r� ru rR rv r$ r� r� r� r� r � IPv4Addressr� r� r� r r� �__annotations__r� r# r� r� r� r rJ r �CONNECTION_DONErw �
stopListeningr� rL rX r} r� r� r� r, r, r, r- r� � s,
3r� )>r� r� r$ r� �typingr �zope.interfacer r �twisted.internetr r r r r �twisted.internet.abstractr
r � twisted.internet.iocpreactorr r
r8 �"twisted.internet.iocpreactor.constr r r r r �'twisted.internet.iocpreactor.interfacesr �twisted.internet.protocolr �twisted.internet.tcpr � TCPConnectorr r r r r r �twisted.pythonr r r �twisted.internet._newtlsr �
__startTLS� ImportErrorrk �WSAECONNREFUSED�WSAENETUNREACHr� �
ITCPTransport�
ISystemHandler" r �
ITLSTransportr� r� �IListeningPortr� r, r, r, r- �<module> sB $ �� < Y;