Your IP : 216.73.216.213
B
;��]�a � @ sD d Z ddlmZmZ ddlZddlmZ ddlmZ ddl m
Z
m Z m Z ddl
mZmZmZmZmZmZ ddlmZ G d d
� d
�ZG d d � d eej�ZG d
d� deej�ZG dd� de�ZG dd� de �Z G dd� dej�Z G dd� dej �Z G dd� dej �Z!G dd� dej �Z"e�#ed��s,de _$de!_$e�%ed��s@d e"_$dS ) zG
Tests for implementations of L{IReactorUDP} and L{IReactorMulticast}.
� )�division�absolute_importN)�unittest)�
intToBytes)�Deferred�
gatherResults�
maybeDeferred)�protocol�reactor�error�defer�
interfaces�udp)�runtimec @ s0 e Zd ZdZdZdZdd� Zdd� Zdd� ZdS ) �Mixinr Nc C s
g | _ d S )N)�packets)�self� r �H/opt/alt/python37/lib64/python3.7/site-packages/twisted/test/test_udp.py�__init__ s zMixin.__init__c C s, d| _ | jd k r(| jd }| _|�d � d S )N� )�started�startedDeferred�callback)r �dr r r �
startProtocol s
zMixin.startProtocolc C s
d| _ d S )Nr )�stopped)r r r r � stopProtocol% s zMixin.stopProtocol) �__name__�
__module__� __qualname__r r r r r r r r r r r s r c @ s e Zd ZdZdZdd� ZdS )�ServerNr c C s6 | j �||f� | jd k r2| jd }| _|�d � d S )N)r �append�packetReceivedr )r �data�addrr r r r �datagramReceived. s
zServer.datagramReceived)r r r r# �refusedr& r r r r r! ) s r! c @ s, e Zd ZdZdZdd� Zdd� Zdd� ZdS ) �ClientNr c C s2 | j �|� | jd k r.| jd }| _|�d � d S )N)r r" r# r )r r$ r r r r r& ; s
zClient.datagramReceivedc C s, | j d k r"| j d }| _ |�|� || _d S )N)r �errback�failure)r r* r r r r �connectionFailedA s
zClient.connectionFailedc C s2 | j d k r(| j d }| _ |�t�d�� d| _d S )N�yupr )r r) r �ConnectionRefusedErrorr' )r r r r r �connectionRefusedG s
zClient.connectionRefused)r r r r# r' r& r+ r. r r r r r( 6 s
r( c @ s e Zd Zdd� ZdS )�
GoodClientc C s2 | j d k r(| j d }| _ |�t�d�� d| _d S )Nr, r )r r) r r- r' )r r r r r r. P s
z GoodClient.connectionRefusedN)r r r r. r r r r r/ N s r/ c @ s e Zd ZdZdS )�BadClientErrorzf
Raised by BadClient at the end of every datagramReceived call to try and
screw stuff up.
N)r r r �__doc__r r r r r0 X s r0 c @ s$ e Zd ZdZdZdd� Zdd� ZdS )� BadClientz�
A DatagramProtocol which always raises an exception from datagramReceived.
Used to test error handling behavior in the reactor for that method.
Nc C s
|| _ dS )ze
Set the Deferred which will be called back when datagramReceived is
called.
N)r )r r r r r � setDeferredg s zBadClient.setDeferredc C s. | j d k r"| j d }| _ |�|� td��d S )Nz Application code is very buggy!)r r r0 )r �bytesr% r r r r r& o s
zBadClient.datagramReceived)r r r r1 r r3 r& r r r r r2 ` s r2 c @ sj e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Ze j
� d
�dkrNde_ dd� Z
dd� Zdd� ZdS )�UDPTestsc s< t � }t�� }|_tjd|dd�� � �fdd�}|�|�S )zu
The C{type} of the host address of a listening L{DatagramProtocol}'s
transport is C{"UDP"}.
r z 127.0.0.1)� interfacec s � � � }��|jd� � �� S )NZUDP)�getHost� assertEqual�type�
stopListening)�ignoredr% )�pr r r � cbStarted� s z+UDPTests.test_oldAddress.<locals>.cbStarted)r! r r r r
� listenUDP� addCallback)r �serverr r= r )r< r r �test_oldAddressy s
zUDPTests.test_oldAddressc sR t � �t�� }�_tjd�dd�� � ��fdd�}��fdd�}|�|��|�S )z�
The L{DatagramProtocol}'s C{startProtocol} and C{stopProtocol}
methods are called when its transports starts and stops listening,
respectively.
r z 127.0.0.1)r6 c s$ �� �jd� �� �jd� � �� S )Nr r )r8 r r r: )r; )�port1r r@ r r r= � s z*UDPTests.test_startStop.<locals>.cbStartedc s � � �jd� d S )Nr )r8 r )r; )r r@ r r � cbStopped� s z*UDPTests.test_startStop.<locals>.cbStopped)r! r r r r
r> r? )r r r= rC r )rB r r@ r �test_startStop� s zUDPTests.test_startStopc sF t � �t�� }�_tjd�dd�}dd� � � �fdd�}|�� |�S )zr
Re-listening with the same L{DatagramProtocol} re-invokes the
C{startProtocol} callback.
r z 127.0.0.1)r6 c S s |� � S )N)r: )r; �portr r r r= � s z'UDPTests.test_rebind.<locals>.cbStartedc s* t �� }�_tjd�dd�}|�� |�S )Nr z 127.0.0.1)r6 )r r r r
r> r? )r; r r< )r= r@ r r rC � s z'UDPTests.test_rebind.<locals>.cbStopped)r! r r r r
r> r? )r r r< rC r )r= r@ r � test_rebind� s zUDPTests.test_rebindc sX t � �t�� }�_tjd�dd�� � ��fdd�}|�|� � fdd�}|�|� |S )z�
A L{CannotListenError} exception is raised when attempting to bind a
second protocol instance to an already bound port
r z 127.0.0.1)r6 c s> �� � �� �j�� � t� }�jtjtj� �� j |dd� d S )Nz 127.0.0.1)r6 )
r8 r7 � transportr! � assertRaisesr ZCannotListenErrorr
r> rE )r; Zserver2)rE r r@ r r r= � s z*UDPTests.test_bindError.<locals>.cbStartedc s � � � S )N)r: )r; )rE r r �
cbFinished� s z+UDPTests.test_bindError.<locals>.cbFinished)r! r r r r
r> r? )r r r= rI r )rE r r@ r �test_bindError� s
zUDPTests.test_bindErrorc s� t � �t�� }�_tjd�dd��t� � t�� �� _� ��fdd�}|�|�}� �fdd�}|�|� � ��fdd �}|�|� ��fd
d �}|�|� |S ) z�
Datagrams can be sent with the transport's C{write} method and
received via the C{datagramReceived} callback method.
r z 127.0.0.1)r6 c s t jd� dd��_�S )Nr z 127.0.0.1)r6 )r
r> �port2)r; )�client�
clientStartedr r r �cbServerStarted� s z2UDPTests.test_sendPackets.<locals>.cbServerStartedc s� �j �d�j �� j� �j �� }�j �� }t�� }�_�j �d|j|jf� ddd|j|jffg�� ���fdd�� tj � d �|gdd �S )
Nz 127.0.0.1s hello)� a)� bN� cc s6 �r2t �� }�_|�� � �jj��d�� |S d S )Nr )r r r# r? rG �write�pop)r; ZnextClientWrite)� cbClientSendrL � clientWritesr@ r r rT � s
zHUDPTests.test_sendPackets.<locals>.cbClientStarted.<locals>.cbClientSendT)�fireOnOneErrback)
rG �connectr7 rE r r r# rR �host� DeferredList)r; �cAddr�sAddrZ
serverSend)rL r@ )rT rU r �cbClientStarted� s
z2UDPTests.test_sendPackets.<locals>.cbClientStartedc s` � j �� }�j �� }��� jd|j|jffg� |j|jf}���jd|fd|fd|fg� d S )Ns hellorO rP rQ )rG r7 r8 r rX rE )r; rZ r[ Z
clientAddr)rL r r@ r r �cbSendsFinished� s
z2UDPTests.test_sendPackets.<locals>.cbSendsFinishedc s$ t jt �� j�t ��jj�gdd�S )NT)rV )r rY r r: rK )r; )rB r r r rI s
z-UDPTests.test_sendPackets.<locals>.cbFinished)r! r r r r
r> r/ r? )r �
serverStartedrN r r\ r] rI r )rL rM rB r r@ r �test_sendPackets� s
zUDPTests.test_sendPacketsc s� t � � t�� }� _tjd� dd��t� �t�� }�_tjd�dd��tj||gdd�}� ��fdd�}|�|� ��fdd �}|�|� |S )
z�
A L{ConnectionRefusedError} exception is raised when a connection
attempt is actively refused by the other end.
Note: This test assumes no one is listening on port 80 UDP.
r z 127.0.0.1)r6 T)rV c s^ t �� }� _� j�dd� x2td�D ]&}� j�t|�� �j�t|�d� q&W ��|t j
�S )Nz 127.0.0.1�P �
)z 127.0.0.1r` ) r r r rG rW �rangerR r �
assertFailurer r- )r; r. �i)rL r r@ r r r= $ s z2UDPTests.test_connectionRefused.<locals>.cbStartedc s" t jt �� j�t ��j�gdd�S )NT)rV )r rY r r: )r; )rE rK r r rI 2 s
z3UDPTests.test_connectionRefused.<locals>.cbFinished) r/ r r r r
r> r! rY r? )r rM r^ r r= rI r )rL rE rK r r@ r �test_connectionRefused s
z UDPTests.test_connectionRefused�INFRASTRUCTURE�AZUREPIPELINESz"Hangs on Pipelines due to firewallc C sV t � }tjd|dd�}| �tj|jjdd� |j�dd� | �t|jjdd� |� � S )a
A call to the transport's connect method fails with an
L{InvalidAddressError} when a non-IP address is passed as the host
value.
A call to a transport's connect method fails with a L{RuntimeError}
when the transport is already connected.
r z 127.0.0.1)r6 Z localhostr` )
r/ r
r> rH r ZInvalidAddressErrorrG rW � RuntimeErrorr: )r rL rE r r r �test_badConnect? s zUDPTests.test_badConnectc s� t �� ��fdd�}��|� t� �tjd�dd���fdd�}��|� ��� � tt d���g �� ������fd d
���� �S ) zr
When datagramReceived raises an exception it is logged but the port
is not disconnected.
c s* � � t�}� �t|�ddt|�f � dS )z�
Flush the exceptions which the reactor should have logged and make
sure they're actually there.
� z'Incorrectly found %d errors, expected 2N)�flushLoggedErrorsr0 r8 �len)�ignZerrs)r r r � cbCompletedZ s
z8UDPTests.test_datagramReceivedError.<locals>.cbCompletedr z 127.0.0.1)r6 c s t ��j��� fdd��S )z�
Disconnect the port we started and pass on whatever was given to us
in case it was a Failure.
c s � S )Nr )rm )�resultr r �<lambda>k � zHUDPTests.test_datagramReceivedError.<locals>.cbCleanup.<locals>.<lambda>)r r r: �addBoth)ro )rE )ro r � cbCleanupf s z6UDPTests.test_datagramReceivedError.<locals>.cbCleanup�< c s� �s(y�� d� W n ��� Y nX ���jdd� t��d��} t�� }��|� �j� | �j
�j f� ���� fdd�}�fdd�}|� ||� |�
�j� t�d |jt�d
��� dS ) a6
Send one packet to the listening BadClient. Set up a 0.1 second
timeout to do re-transmits in case the packet is dropped. When two
packets have been received by the BadClient, stop sending and let
the finalDeferred's callbacks do some assertions.
zNot enough packets receivedNz UDP Protocol lost its transportr c s: �� � ��| � t��dkr0t�d� jd� n�� dS )z�
A packet arrived. Cancel the timeout for it, record it, and
maybe finish the test.
rj r N)�cancelr" rl r
� callLaterr )�packet)�
finalDeferred� makeAttempt�succeededAttempts� timeoutCallr r �cbPacketReceived� s
zRUDPTests.test_datagramReceivedError.<locals>.makeAttempt.<locals>.cbPacketReceivedc s
� � dS )a�
The packet wasn't received quickly enough. Try sending another
one. It doesn't matter if the packet for which this was the
timeout eventually arrives: makeAttempt throws away the
Deferred on which this function is the errback, so when
datagramReceived callbacks, so it won't be on this Deferred, so
it won't raise an AlreadyCalledError.
Nr )�err)ry r r �ebPacketTimeout� s zQUDPTests.test_datagramReceivedError.<locals>.makeAttempt.<locals>.ebPacketTimeoutg�������?z&Timed out in testDatagramReceivedError)Zfailr) ZfailIfIdenticalrG r rS r r r3 rR rX rE Z addCallbacksZ
addErrbackr
rv r � TimeoutError)rw ZpacketDeferredr| r~ )r% �attemptsrL rx ry r rz )r{ r ry { s$
z8UDPTests.test_datagramReceivedError.<locals>.makeAttempt)
r r r? r2 r
r> rr r7 �listrb )r rn rs r )r% r� rL rx ry rE r rz r �test_datagramReceivedErrorS s
7z#UDPTests.test_datagramReceivedErrorc C sR G dd� d�}t �dt� �}|� |_|�dd� | �| jg�}| �t|�d� dS )ai
C{'<broadcast>'} is an alternative way to say C{'255.255.255.255'}
({socket.gethostbyname("<broadcast>")} returns C{'255.255.255.255'}),
so because it becomes a valid IP address, no deprecation warning about
passing hostnames to L{twisted.internet.udp.Port.write} needs to be
emitted by C{write()} in this case.
c @ s e Zd Zdd� ZdS )z6UDPTests.test_NoWarningOnBroadcast.<locals>.fakeSocketc S s d S )Nr )r ZfooZbarr r r �sendto� s z=UDPTests.test_NoWarningOnBroadcast.<locals>.fakeSocket.sendtoN)r r r r� r r r r �
fakeSocket� s r� r s test)z <broadcast>i� N) r ZPortr! �socketrR Z
flushWarnings�test_NoWarningOnBroadcastr8 rl )r r� r< �warningsr r r r� � s z"UDPTests.test_NoWarningOnBroadcastN)r r r rA rD rF rJ r_ re �os�environ�get�skipri r� r� r r r r r5 w s K*cr5 c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) � ReactorShutdownInteractionTestsz!Test reactor shutdown interactionc C s t � | _tjd| jdd�| _dS )zStart a UDP portr z 127.0.0.1)r6 N)r! r@ r
r> rE )r r r r �setUp� s z%ReactorShutdownInteractionTests.setUpc C s
| j �� S )zStop the UDP port)rE r: )r r r r �tearDown� s z(ReactorShutdownInteractionTests.tearDownc sh t �� � t �� }�j_� �fdd�}|�|� �fdd�}� �|� �jj�dd�jj�� jf� � S )z0Test reactor shutdown while in a recvfrom() loopc s �j j�� t�d� jd � d S )Nr )r@ rG ZconnectionLostr
rv r )r; )�finishedr r r �pktRece� s zQReactorShutdownInteractionTests.testShutdownFromDatagramReceived.<locals>.pktRecec s � � � d S )N)rk )r; )r r r � flushErrors� s zUReactorShutdownInteractionTests.testShutdownFromDatagramReceived.<locals>.flushErrorss@ z 127.0.0.1) r r r@ r# r? rG rR r7 rE )r Zprr� r� r )r� r r � testShutdownFromDatagramReceived� s
z@ReactorShutdownInteractionTests.testShutdownFromDatagramReceivedN)r r r r1 r� r� r� r r r r r� � s r� c @ s� e Zd Zej�d�dkr&ej�� r&dZ dd� Z
dd� Z dd � Z d
d � Z
d d
� Zdd� Zdd� Zej�� rxej�� sxde_ dd� Zdd� Zej�� r�de_ dS )�MulticastTestsrf rg z Does not work on Azure Pipelinesc C sN t � | _t� | _t�d| j�| _t�d| j�| _| jj� d| jj�
� j � d S )Nr z 127.0.0.1) r! r@ r( rL r
�listenMulticastrB rK rG rW r7 rE )r r r r r� s zMulticastTests.setUpc C s t t| jj�t| jj�g�S )N)r r rB r: rK )r r r r r� s
zMulticastTests.tearDownc C sJ xD| j | jfD ]4}| �|j�� d� |j�d� | �|j�� d� qW d S )Nr rj )rL r@ r8 rG ZgetTTLZsetTTL)r �or r r �testTTL s zMulticastTests.testTTLc sx �� �jj�� d� �jj�� � �jj�d�}� �fdd�}|�|� � �fdd�}|�|� �fdd�}|�|� |S ) zv
Test that after loopback mode has been set, multicast packets are
delivered to their sender.
r z 225.0.0.250c s( t � }�j_�jj�dd� jf� |S )Ns helloz 225.0.0.250)r r@ r# rG rR rE )r; r )r% r r r �cbJoined$ s z.MulticastTests.test_loopback.<locals>.cbJoinedc sf �� t�jj�d� �jj�d� �� �jj�� d� �jj�dd� jf� t � }t
� d|j d � |S )Nr r s helloz 225.0.0.250)
r8 rl r@ r rG ZsetLoopbackMode�getLoopbackModerR rE r r
rv r )r; r )r% r r r �cbPacket* s z.MulticastTests.test_loopback.<locals>.cbPacketc s � � t� jj�d� d S )Nr )r8 rl r@ r )r; )r r r �
cbNoPacket6 s z0MulticastTests.test_loopback.<locals>.cbNoPacket)r8 r@ rG r� r7 � joinGroupr? )r �joinedr� r� r� r )r% r r �
test_loopback s
z MulticastTests.test_loopbackc sj � � � jj�� d� � � � jj�� d� � jj�d�}� jj�d�}t||g�}� fdd�}|�|� |S )zK
Test C{getOutgoingInterface} and C{setOutgoingInterface}.
z0.0.0.0z 127.0.0.1c s, � � � jj�� d� � � � jj�� d� d S )Nz 127.0.0.1)r8 rL rG �getOutgoingInterfacer@ )r; )r r r � cbInterfacesJ s z3MulticastTests.test_interface.<locals>.cbInterfaces)r8 rL rG r� r@ ZsetOutgoingInterfacer r? )r �d1�d2ro r� r )r r �test_interface= s
z MulticastTests.test_interfacec sT � j j�d�}� fdd�}|�|� � fdd�}|�|� � fdd�}|�|� |S )zE
Test that multicast a group can be joined and left.
z 225.0.0.250c s � j j�d�S )Nz 225.0.0.250)rL rG �
leaveGroup)r; )r r r � clientJoinedY s z3MulticastTests.test_joinLeave.<locals>.clientJoinedc s � j j�d�S )Nz 225.0.0.250)r@ rG r� )r; )r r r �
clientLeft] s z1MulticastTests.test_joinLeave.<locals>.clientLeftc s � j j�d�S )Nz 225.0.0.250)r@ rG r� )r; )r r r � serverJoineda s z3MulticastTests.test_joinLeave.<locals>.serverJoined)rL rG r� r? )r r r� r� r� r )r r �test_joinLeaveS s
z MulticastTests.test_joinLeavec C s | � | jj�d�tj�S )z�
Test that an attempt to join an address which is not a multicast
address fails with L{error.MulticastJoinError}.
z 127.0.0.1)rc rL rG r� r ZMulticastJoinError)r r r r �test_joinFailureh s z MulticastTests.test_joinFailurez2Windows' UDP multicast is not yet fully supported.c sv t � �t�d����jj�� � �jj�d�}� ��fdd�}|�|� �fdd�}|�|� �fdd�}|�|� |S ) zn
Test that a multicast group can be joined and messages sent to and
received from it.
r z 225.0.0.250c s& t � }�j_�j�dd� jf� |S )Ns hello worldz 225.0.0.250)r r@ r# rG rR rE )r; r )r% �cr r r r� � s z/MulticastTests.test_multicast.<locals>.cbJoinedc s � � � jjd d d� d S )Nr s hello world)r8 r@ r )r; )r r r r� � s z/MulticastTests.test_multicast.<locals>.cbPacketc s t �j�}|�� fdd�� |S )Nc s � S )Nr )rm )� passthroughr r rp � rq z@MulticastTests.test_multicast.<locals>.cleanup.<locals>.<lambda>)r r: r? )r� ro )r< )r� r �cleanup� s
z.MulticastTests.test_multicast.<locals>.cleanup)r! r
r� r@ rG r7 r� r? )r r� r� r� r� r )r% r� r< r r �test_multicastw s
z MulticastTests.test_multicastc s� t � � tjd� dd����� j�t � �tj��dd��d�t�jj�����������g�}� ���fdd�}|� |� � ��fdd�}|� |� ��fd d
�}|�
|� |S ) z�
Test that multiple sockets can listen on the same multicast port and
that they both receive multicast messages directed to that address.
r T)ZlistenMultiplez 225.0.0.250c s6 t � }� _t � }�_� j�d��f� t||g�S )Ns hello world)r r# rG rR r )r; r� r� )� firstClient�portno� secondClient�theGroupr r r� � s z5MulticastTests.test_multiListen.<locals>.serverJoinedc s0 �� � jd d d� �� �jd d d� d S )Nr s hello world)r8 r )r; )r� r� r r r �
gotPackets� s z3MulticastTests.test_multiListen.<locals>.gotPacketsc s. t t�j�t�j�g�}|�� fdd�� |S )Nc s � S )Nr )rm )r� r r rp � rq zBMulticastTests.test_multiListen.<locals>.cleanup.<locals>.<lambda>)r r r: r? )r� ro )� firstPort�
secondPort)r� r r� � s
z0MulticastTests.test_multiListen.<locals>.cleanup) r! r
r� r7 rE r r@ rG r� r? rr )r r� r� r� r� r )r� r� r� r� r� r r� r �test_multiListen� s$
z MulticastTests.test_multiListenzjon non-linux platforms it appears multiple processes can listen, but not multiple sockets in same process?N)r r r r� r� r� r �platformZisMacOSXr� r� r� r� r� r� r� r� Z isWindowsZisVistar� r� r r r r r� � s
" )
r� z!This reactor does not support UDPz'This reactor does not support multicast)&r1 �
__future__r r r� Z
twisted.trialr Ztwisted.python.compatr Ztwisted.internet.deferr r r Ztwisted.internetr r
r r r
r Ztwisted.pythonr r ZDatagramProtocolr! ZConnectedDatagramProtocolr( r/ � Exceptionr0 r2 ZTestCaser5 r� r� Z IReactorUDPr� ZIReactorMulticastr r r r �<module> s0
V1 I