\d1kRdZddlZddlmZddlmZmZmZmZm Z m Z ddl m Z m Z mZddlmZddlmZGdd ZGd d eejZGd d eejZGddeZGddeZGddejZeeje d dGddeZeeje d dGddeZeeje d dGddeZ dS)zG Tests for implementations of L{IReactorUDP} and L{IReactorMulticast}. N)skipIf)defererror interfacesprotocolreactorudp)Deferred gatherResults maybeDeferred)runtime)TestCasec,eZdZdZdZdZdZdZdZdS)MixinrNcg|_dSN)packetsselfs 5lib/python3.11/site-packages/twisted/test/test_udp.py__init__zMixin.__init__s  cnd|_|j&|jdc}|_|ddSdSN)startedstartedDeferredcallbackrds r startProtocolzMixin.startProtocolsD   +&*&:D #At# JJt      , +rcd|_dSr)stoppedrs r stopProtocolzMixin.stopProtocol#s  r) __name__ __module__ __qualname__rr#rrr!r$rrrrsPGGO rrceZdZdZdZdZdS)ServerNrc|j||f|j&|jdc}|_|ddSdSrrappendpacketReceivedr)rdataaddrr s rdatagramReceivedzServer.datagramReceived+sW T4L)))   *%)%8$ "At" JJt      + *r)r%r&r'r.refusedr1r(rrr*r*'s-NGrr*c(eZdZdZdZdZdZdZdS)ClientNrc|j||j&|jdc}|_|ddSdSrr,)rr/r s rr1zClient.datagramReceived7sS D!!!   *%)%8$ "At" JJt      + *rcj|j$|jdc}|_||||_dSr)rerrbackfailure)rr8r s rconnectionFailedzClient.connectionFailed=s:   +&*&:D #At# IIg    rc|j6|jdc}|_|tjdd|_dSNyuprrr7rConnectionRefusedErrorr2rs rconnectionRefusedzClient.connectionRefusedCE   +&*&:D #At# IIe2599 : : : r)r%r&r'r.r2r1r9r?r(rrr4r42sKNG  rr4ceZdZdZdS) GoodClientc|j6|jdc}|_|tjdd|_dSr;r=rs rr?zGoodClient.connectionRefusedKr@rN)r%r&r'r?r(rrrBrBJs#rrBceZdZdZdS)BadClientErrorzf Raised by BadClient at the end of every datagramReceived call to try and screw stuff up. N)r%r&r'__doc__r(rrrErERsrrEc"eZdZdZdZdZdZdS) BadClientz A DatagramProtocol which always raises an exception from datagramReceived. Used to test error handling behavior in the reactor for that method. Nc||_dS)ze Set the Deferred which will be called back when datagramReceived is called. N)r rs r setDeferredzBadClient.setDeferredas rcv|j$|jdc}|_||td)NzApplication code is very buggy!)r rrE)rbytesr0r s rr1zBadClient.datagramReceivedhs: 6 IAtv JJu   >???r)r%r&r'rFr rJr1r(rrrHrHYsH A@@@@@rrH!This reactor does not support UDPceZdZdZdZdZdZdZee j ddkdd Z d Z d Zd Zd ZdS)UDPTestsct}tjx}|_t jd|dfd}||S)zu The C{type} of the host address of a listening L{DatagramProtocol}'s transport is C{"UDP"}. r 127.0.0.1 interfacec}|jdS)NUDP)getHost assertEqualtype stopListening)ignoredr0prs r cbStartedz+UDPTests.test_oldAddress..cbStartedzs899;;D   TY . . .??$$ $rr*rr rr listenUDP addCallback)rserverr r\r[s` @rtest_oldAddresszUDPTests.test_oldAddressqsm %*^%5%55F "  a; ? ? ? % % % % % % }}Y'''rcttjx}_t jddfd}fd}|||S)z The L{DatagramProtocol}'s C{startProtocol} and C{stopProtocol} methods are called when its transports starts and stops listening, respectively. rrQrRcjdjdS)Nrr)rWrr#rY)rZport1rr`s rr\z*UDPTests.test_startStop..cbStartedsE   V^Q / / /   V^Q / / /&&(( (rc>jddSr)rWr#)rZrr`s r cbStoppedz*UDPTests.test_startStop..cbStoppeds!   V^Q / / / / /rr])rr r\rfrdr`s` @@rtest_startStopzUDPTests.test_startStops %*^%5%55F "!!V{CCC ) ) ) ) ) ) )  0 0 0 0 0 0}}Y''33I>>>rcttjx}_t jdd}dfd}||S)zr Re-listening with the same L{DatagramProtocol} re-invokes the C{startProtocol} callback. rrQrRc*|SrrYrZports rr\z'UDPTests.test_rebind..cbStarteds%%'' 'rctjx}_tjdd}||SNrrQrR)rr rrr^r_)rZr r[r\r`s rrfz'UDPTests.test_rebind..cbStoppedsC).)9)9 9A&!!V{CCCA==A.. .rr])rr r[rfr\r`s @@r test_rebindzUDPTests.test_rebinds~ %*^%5%55F "  a; ? ? ? ( ( ( / / / / / / }}Y***rcttjx}_t jddfd}||fd}|||S)z A L{CannotListenError} exception is raised when attempting to bind a second protocol instance to an already bound port rrQrRc,jt}t jtjj |ddS)NrQrR) rWrV transportr* assertRaisesrCannotListenErrorrr^rl)rZserver2rlrr`s rr\z*UDPTests.test_bindError..cbStarteds   T\\^^V-=-E-E-G-G H H HhhG   '! #%      rc,Srrjrks r cbFinishedz+UDPTests.test_bindError..cbFinisheds%%'' 'rr])rr r\rwrlr`s` @@rtest_bindErrorzUDPTests.test_bindErrors %*^%5%55F " FkBBB        i    ( ( ( ( ( j!!!rc t tjx} _t jd d t tjx_fd}||} fd}|| fd}|| fd}|||S)z Datagrams can be sent with the transport's C{write} method and received via the C{datagramReceived} callback method. rrQrRc@tjdd_Srn)rr^port2)rZclient clientStartedrs rcbServerStartedz2UDPTests.test_sendPackets..cbServerStarteds" *1f LLLDJ rcjdjjj}j}t jx}_jd|j|jfddd|j|jffgfdt j d|gdS) NrQhello)a)bNccrStjx}_|jjd|SdSNr)rr r.r_rrwritepop)rZnextClientWrite cbClientSendr| clientWritesr`s rrzHUDPTests.test_sendPackets..cbClientStarted..cbClientSendse+>Cn>N>NNOf&;#// ===*F$*L,<,.cbClientStarteds   $ $[&2B2J2J2L2L2Q R R R$,,..E$,,..E161A1A AJ.   " "8ej%*-E F F F#\D5:uz:R3STL + + + + + + + +%d##Z04 rc$j}j}jd|j|jffg|j|jf}jd|fd|fd|fgdS)Nrrrr)rrrVrWrrrl)rZrr clientAddrr|rr`s rcbSendsFinishedz2UDPTests.test_sendPackets..cbSendsFinisheds$,,..E$,,..E   V^x%*ej9Q.R-S T T T*ej1J    #dJ%7$ 9KL     rctjtjjtjjjgdSNTr)rrr rYr{)rZrdrs rrwz-UDPTests.test_sendPackets..cbFinishedsK%'(;<<' (@AA"&  r)r*rr rrr^rBr_) r serverStartedr~r rrrwr|r}rdr`s ` @@@@rtest_sendPacketszUDPTests.test_sendPacketss4 161A1AA .!!V{CCC161A1AA . ! ! ! ! ! ! !  % %o 6 6      0 o&&&        o&&&       j!!!rINFRASTRUCTUREAZUREPIPELINESz"Hangs on Pipelines due to firewallc ttjx}_t jddt tjx} _t jd dtj||gd} fd}||fd}|||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. rrQrRTrcLtjx}_jddt dD]?}jd|fzjd|fzd@|tj S)NrQP %drQr) rr rrrrranger assertFailurerr>)rZr?ir|rr`s rr\z2UDPTests.test_connectionRefused..cbStarteds9>9I9I I  6   $ $[" 5 5 52YY H H &&ut|444 &&ut|5FGGGG%%&79UVV VrctjtjjtjjgdSrrrr rY)rZrlr{s rrwz3UDPTests.test_connectionRefused..cbFinished)sI%'(:;;'(;<<"&  r) rBrr rrr^r*rr_) rr}rr r\rwr|rlr{r`s ` @@@@rtest_connectionRefusedzUDPTests.test_connectionRefuseds161A1AA . FkBBB161A1AA .!!V{CCC   }=PT U U U W W W W W W W i          j!!!rcN t tjx} _t jd d d t tjx} _tjx} _t jd d tj||gd} fd}| |tj||gd} fd}| | fd }| ||S) as When a server fails to successfully read a packet the server should still be able to process future packets. The IOCP reactor had a historical problem where a failure to read caused the reactor to ignore any future reads. This test should prevent a regression. Note: This test assumes no one is listening on port 80 UDP. rrQrRsSending test packet to serverTrcjddjdjfdS)Ns$write to port no one is listening torrQ)rrr_realPortNumber)rZr|r` serverPorttest_data_to_sends rcbClientAndServerStartedzAUDPTests.test_serverReadFailure..cbClientAndServerStartedLsZ   " "79J      " "!K1K#L     rcVjdddSrrWr)rZrr`rs rverify_server_got_dataz?UDPTests.test_serverReadFailure..verify_server_got_data_s,   V^A.q13D E E E E ErctjtjjtjjgdSrr)rZ clientPortrs rcleanupz0UDPTests.test_serverReadFailure..cleanupdsI%' (@AA' (@AA"&  r) rBrr rrr^r*r.rr_)rr}r serverGotDataserver_client_started_dr all_data_sentrrr|rr`rrs` @@@@@rtest_serverReadFailurezUDPTests.test_serverReadFailure5s161A1AA .&q&KHHH <161A1AA .050@0@@ -&q&KHHH "'"4 M *T# # #          ++,DEEE* $m 4t    F F F F F F F !!"8999       !!'***rcPt}tjd|d}|tj|jjdd|jdd|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. rrQrR localhostr) rBrr^rsrInvalidAddressErrorrrr RuntimeErrorrY)rr|rls rtest_badConnectzUDPTests.test_badConnectqs FkBBB   %v'7'?b      b111 ,(8(@+rRRR!!###rc| tjfd}|tt jddfd}|ttdg  fdS)zr When datagramReceived raises an exception it is logged but the port is not disconnected. ct}t|ddt|fzdS)z Flush the exceptions which the reactor should have logged and make sure they're actually there. z'Incorrectly found %d errors, expected 2N)flushLoggedErrorsrErWlen)ignerrsrs r cbCompletedz8UDPTests.test_datagramReceivedError..cbCompletedsU )).99D   D 1G3t99,V     rrrQrRcdtjjfdS)z Disconnect the port we started and pass on whatever was given to us in case it was a Failure. cSrr()rresults rzHUDPTests.test_datagramReceivedError..cbCleanup..svr)rr rYaddBoth)rrls`r cbCleanupz6UDPTests.test_datagramReceivedError..cbCleanups2 &t'9::BBCUCUCUCUVV Vr<c`s; dn$#t$rYnwxYw jddddfz}t j}|j |j j f fd} fd}| ||| jtjd|jt!jd 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 receivedNzUDP Protocol lost its transportrrc|tdkrtjdjddSdS)z A packet arrived. Cancel the timeout for it, record it, and maybe finish the test. rrN)cancelr-rr callLaterr)packet finalDeferred makeAttemptsucceededAttempts timeoutCalls rcbPacketReceivedzRUDPTests.test_datagramReceivedError..makeAttempt..cbPacketReceivedsn ""$$$!((000())Q.. %a)?FFFFFKMMMMMrcdS)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()errrs rebPacketTimeoutzQUDPTests.test_datagramReceivedError..makeAttempt..ebPacketTimeouts rg?z&Timed out in testDatagramReceivedError)fail Exceptionr7failIfIdenticalrrrrr rJrrrl addCallbacks addErrbackrrr TimeoutError) rpacketDeferredrrrr0attemptsr|rrrrs @rrz8UDPTests.test_datagramReceivedError..makeAttempts| ,,II;<<<< ,,,!))+++++,   $(I   hll1oo//F"^--N   ~ . . .   " "6DIty+A B B B " " " " " " " "       ' '(8/ J J J  % %m&; < < <!+&"#KLLKKKs ==) rr r_rHrr^rrVlistr) rrrr0rr|rrrlrs ` @@@@@@@rtest_datagramReceivedErrorz#UDPTests.test_datagramReceivedErrors  ((       !!+... FkBBB W W W W W i(((||~~b ??8 8 8 8 8 8 8 8 8 8 8 t  rc(Gdd}tjdt}||_|dd||jg}|t|ddS)ai C{''} is an alternative way to say C{'255.255.255.255'} ({socket.gethostbyname("")} 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. ceZdZdZdS)6UDPTests.test_NoWarningOnBroadcast..fakeSocketcdSrr()rfoobars rsendtoz=UDPTests.test_NoWarningOnBroadcast..fakeSocket.sendtosrN)r%r&r'rr(rr fakeSocketrs#     rrrstest)z iN) r Portr*socketr flushWarningstest_NoWarningOnBroadcastrWr)rrr[warningss rrz"UDPTests.test_NoWarningOnBroadcasts         HQ ! !:<< .///%%t'E&FGG X*****rN)r%r&r'rargrorxrrosenvirongetrrrrrr(rrrOrOos((( ???(+++&8BBBH V '((,<<,'' 'R:::x$$$$gggR+++++rrOcJeZdZdZejedsdZdZdZ dZ dS)ReactorShutdownInteractionTestsz!Test reactor shutdown interactionNrMcnt|_tjd|jd|_dS)zStart a UDP portrrQrRN)r*r`rr^rlrs rsetUpz%ReactorShutdownInteractionTests.setUps,hh %a LLL rc4|jS)zStop the UDP port)rlrYrs rtearDownz(ReactorShutdownInteractionTests.tearDown sy&&(((rc^tjtjx}j_fd}||fd}|jjddjjjfS)z0Test reactor shutdown while in a recvfrom() loopczjjtjdjddSr)r`rrconnectionLostrrr)rZfinishedrs rpktRecezQReactorShutdownInteractionTests.testShutdownFromDatagramReceived..pktRece s: K ! 0 0 2 2 2  a!2D 9 9 9 9 9rc0dSr)rrZrs r flushErrorszUReactorShutdownInteractionTests.testShutdownFromDatagramReceived..flushErrors*s  " " $ $ $ $ $rs@rQ) rr r`r.r_rrrrVrl)rprrrrs` @r testShutdownFromDatagramReceivedz@ReactorShutdownInteractionTests.testShutdownFromDatagramReceiveds>##*/.*:*::T[ ' : : : : : : w % % % % % [))) ## dk&;&C&C&E&E&JK   r) r%r&r'rFr IReactorUDPrskiprrrr(rrrrsf++ !: !'4 0 032MMM )))$$$$$rr'This reactor does not support multicastceZdZejddkrejrdZ e j e dsdZ dZ dZdZd Zd Zd Zeejoej d d ZdZeejddZdS)MulticastTestsrrz Does not work on Azure PipelinesNrcPt|_t|_t jd|j|_t jd|j|_|jj d|jj j dS)NrrQ) r*r`r4r|rlistenMulticastrdr{rrrrVrlrs rrzMulticastTests.setUpGsyhh hh ,Q << ,Q <<  %%k4;3H3P3P3R3R3WXXXXXrctt|jjt|jjgSr)r r rdrYr{rs rrzMulticastTests.tearDownOs9dj677dj677    rc|j|jfD]v}||jd|jd||jdwdS)Nrr)r|r`rWrrgetTTLsetTTL)ros rtestTTLzMulticastTests.testTTLWsdk) 6 6A   Q[//111 5 5 5 K  q ! ! !   Q[//111 5 5 5 5 6 6rcjjdjjjjd}fd}||fd}||fd}|||S)zv Test that after loopback mode has been set, multicast packets are delivered to their sender. r 225.0.0.250ctx}j_jjddjf|S)Nrrr r`r.rrrrlrZr r0rs rcbJoinedz.MulticastTests.test_loopback..cbJoinedfs=-5ZZ 7A * K ! ' '=$)2L M M MHrctjjdjjdjjdjjddjft}tj d|j d|S)Nrrrr) rWrr`rrrsetLoopbackModegetLoopbackModerrlr rrrrs rcbPacketz.MulticastTests.test_loopback..cbPacketms   S!455q 9 9 9 K ! 1 1! 4 4 4   T[2BBDDa H H H K ! ' '=$)2L M M M A  aT 2 2 2HrcbtjjddSr)rWrr`rrs r cbNoPacketz0MulticastTests.test_loopback..cbNoPacketzs,   S!455q 9 9 9 9 9r)rWr`rrrrV joinGroupr_)rjoinedrrrr0s` @r test_loopbackzMulticastTests.test_loopback]s .>>@@!DDD{$,,..&00??       8$$$       8$$$ : : : : : :&&& rcjjdjjdjjd}jjd}t ||g}fd}|||S)zK Test C{getOutgoingInterface} and C{setOutgoingInterface}. z0.0.0.0rQcjjdjjddS)NrQ)rWr|rrgetOutgoingInterfacer`rs r cbInterfacesz3MulticastTests.test_interface..cbInterfacessW   T[2GGII; W W W   T[2GGII; W W W W Wr)rWr|rrr"r`setOutgoingInterfacer r_)rd1d2rr#s` rtest_interfacezMulticastTests.test_interfaces .CCEEyQQQ .CCEEyQQQ [ " 7 7 D D [ " 7 7 D DBx(( X X X X X <((( rcjjd}fd}||fd}||fd}|||S)zE Test that multicast a group can be joined and left. rcBjjdSNr)r|rr leaveGrouprs r clientJoinedz3MulticastTests.test_joinLeave..clientJoined;(33MBB BrcBjjdSr*)r`rrrrs r clientLeftz1MulticastTests.test_joinLeave..clientLefts;(22=AA ArcBjjdSr*)r`rrr+rs r serverJoinedz3MulticastTests.test_joinLeave..serverJoinedr-r)r|rrrr_)rr r,r/r1s` rtest_joinLeavezMulticastTests.test_joinLeaves K ! + +M : : C C C C C l### B B B B B j!!! C C C C C l###rz2Windows' UDP multicast is not yet fully supported.c|||jjdtjS)z Test that an attempt to join an address which is not a multicast address fails with L{error.MulticastJoinError}. rQ)rr|rrrrMulticastJoinErrorrs rtest_joinFailurezMulticastTests.test_joinFailures7!! K ! + +K 8 8%:R   rcnttjdjjjjd}fd}||fd}||fd}|||S)zn Test that a multicast group can be joined and messages sent to and received from it. rrctx}j_jddjf|S)N hello worldrr)rZr r0crs rrz/MulticastTests.test_multicast..cbJoineds:-5ZZ 7A * K  n}di.H I I IHrc`jjddddSNrr8)rWr`rrs rrz/MulticastTests.test_multicast..cbPackets.   T[03A6 G G G G Grcbtj}|fd|S)NcSrr(r passthroughs rrz@MulticastTests.test_multicast..cleanup..;r)r rYr_)r?rr[s` rrz.MulticastTests.test_multicast..cleanups6"1?33F   6666 7 7 7Mr)r*rr r`rrrVrr_)rrrrrr0r9r[s` @@@rtest_multicastzMulticastTests.test_multicasts HH  #Aq ) ){$,,..&00??        8$$$ H H H H H 8$$$      7### rzjon non-linux platforms it appears multiple processes can listen, but not multiple sockets in same process?c0 ttjddjttjd d t jj    g} fd}| |fd}| | fd}| ||S)z Test that multiple sockets can listen on the same multicast port and that they both receive multicast messages directed to that address. rT)listenMultiplerctx}_tx}_jdft ||gS)Nr8)r r.rrrr )rZr%r& firstClientportno secondClienttheGroups rr1z5MulticastTests.test_multiListen..serverJoinedsU.6jj 8B+/7zz 9B,  ! ' '68J K K K "b** *rcjdddjddddSr;r)rZrErGrs r gotPacketsz3MulticastTests.test_multiListen..gotPacketssO   [03A6 G G G   \1!4Q7 H H H H Hrcttjtjg}|fd|S)NcSrr(r>s rrzBMulticastTests.test_multiListen..cleanup.. r@r)r r rYr_)r?r firstPort secondPorts` rrz0MulticastTests.test_multiListen..cleanupsY"!)"9::!*":;;F   6666 7 7 7Mr) r*rr rVrlr r`rrrr_r) rrr1rJrrErMrFrGrNrHs ` @@@@@@rtest_multiListenzMulticastTests.test_multiListensfhh +A{4PPP ""$$)xx ,V\RVWWW   %//99##H--$$X..    + + + + + + + + <((( I I I I I I I :&&&       w r)r%r&r'rrrr platformisMacOSXrrIReactorMulticastrrrrrr'r2r isWindowsisVistar5rArOr(rrrr8sp '((,<<<   % % ' ' =2 ': ' 6 698YYY   666 """H$0 V""$$GW-=-E-E-G-G)G<   @ V""$$  //  ///rr)!rFrunittestrtwisted.internetrrrrrr twisted.internet.deferr r r twisted.pythonr twisted.trial.unittestrrDatagramProtocolr*ConnectedDatagramProtocolr4rBrrErHrrOrrRrr(rrr\s   MMMMMMMMMMMMMMMMIIIIIIIIII""""""++++++(UH-UH60Y@@@@@)@@@, "J "7D 1 113VWWN+N+N+N+N+xN+N+XWN+b  "J "7D 1 113VWW33333h33XW3l$ $Wd333-RRRRRXRR RRRr