o i}w @s*dZddlmZddlZddlmZmZmZmZm Z m Z m Z m Z m Z mZmZddlmZddlmZmZmZmZmZmZddlmZmZddlmZmZmZdd lm Z dd l!m"Z"m#Z#m$Z$dd l%m&Z&dd l'm(Z(dd l)m*Z*ddl+m,Z,m-Z-GdddZ.Gddde.ej/Z0Gddde.ej1Z2Gddde0Z3Gddde4Z5Gdddej/Z6ee7ed dGddde-Z8ee7ed dGddde-Z9d7d*d+Z:ee#ed d,ee:eeed-d. d/Gd0d1d1e-Z;ee#ed d,e*ee:ee e d2d3 d4Gd5d6d6e;Zr2r(r(r)r@Ms  zClient.datagramReceivedcCs,|jdur|jd}|_||||_dSr$)r/errbackfailure)r'rEr2r(r(r)connectionFailedSs   zClient.connectionFailedcC2|jdur|jd}|_|tdd|_dSNZyupr-r/rDrConnectionRefusedErrorrBr1r(r(r)connectionRefusedY  zClient.connectionRefused)r6r7r8r:rBr@rFrKr(r(r(r)rCIs  rCc@eZdZddZdS) GoodClientcCrGrHrIr1r(r(r)rKarLzGoodClient.connectionRefusedN)r6r7r8rKr(r(r(r)rN` rNc@seZdZdZdS)BadClientErrorzf Raised by BadClient at the end of every datagramReceived call to try and screw stuff up. N)r6r7r8__doc__r(r(r(r)rPhsrPc@s$eZdZdZdZddZddZdS) BadClientz A DatagramProtocol which always raises an exception from datagramReceived. Used to test error handling behavior in the reactor for that method. NcCs ||_dS)ze Set the Deferred which will be called back when datagramReceived is called. N)r2r1r(r(r) setDeferredws zBadClient.setDeferredcCs*|jdur|jd}|_||td)NzApplication code is very buggy!)r2r0rP)r'bytesr?r2r(r(r)r@~s  zBadClient.datagramReceived)r6r7r8rQr2rSr@r(r(r(r)rRos  rR!This reactor does not support UDPc@sreZdZddZddZddZddZd d Zee j d d kd ddZ ddZ ddZddZddZdS)UDPTestscs<t}t}|_tjd|ddfdd}||S)zu The C{type} of the host address of a listening L{DatagramProtocol}'s transport is C{"UDP"}. r 127.0.0.1 interfacecs}|jdS)NZUDP)getHost assertEqualtype stopListening)ignoredr?pr'r(r) cbStartedsz+UDPTests.test_oldAddress..cbStartedNr9rrr/r listenUDP addCallback)r'serverr2rar(r_r)test_oldAddresss  zUDPTests.test_oldAddresscsRtt}_tjdddfdd}fdd}|||S) z The L{DatagramProtocol}'s C{startProtocol} and C{stopProtocol} methods are called when its transports starts and stops listening, respectively. rrWrXcs$jdjdS)Nr-r)r[r.r4r]r^port1r'rer(r)rasz*UDPTests.test_startStop..cbStartedcsjddSr,)r[r4rg)r'rer(r) cbStoppedsz*UDPTests.test_startStop..cbStoppedNrb)r'r2rarjr(rhr)test_startStops zUDPTests.test_startStopcsFtt}_tjddd}ddfdd}||S) zr Re-listening with the same L{DatagramProtocol} re-invokes the C{startProtocol} callback. rrWrXcSs|Sr$r])r^portr(r(r)raz'UDPTests.test_rebind..cbStartedcs*t}_tjddd}||SNrrWrX)rrr/rrcrd)r^r2r`rarer(r)rjs z'UDPTests.test_rebind..cbStoppedNrb)r'r2r`rjr(rpr) test_rebinds  zUDPTests.test_rebindcsXtt}_tjdddfdd}||fdd}|||S) z A L{CannotListenError} exception is raised when attempting to bind a second protocol instance to an already bound port rrWrXcs>jt}jtjtjj |dddS)NrWrX) r[rZr;r9 assertRaisesrZCannotListenErrorrrcrm)r^Zserver2rmr'rer(r)ras z*UDPTests.test_bindError..cbStartedcsSr$rlrgrmr(r) cbFinishedrnz+UDPTests.test_bindError..cbFinishedNrb)r'r2rarur(rsr)test_bindErrors  zUDPTests.test_bindErrorcstt}_tjdddtt_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. rrWrXcstjddd_Sro)rrcport2rg)client clientStartedr'r(r)cbServerStartedsz2UDPTests.test_sendPackets..cbServerStartedcsjdjjj}j}t}_jd|j|jfddd|j|jffgfddtj d|gdd S) NrWhello)a)bNccs6rt}_|jjd|SdSNr)rrr:rdr;writepop)r^ZnextClientWrite) cbClientSendrx clientWritesrer(r)rs  zHUDPTests.test_sendPackets..cbClientStarted..cbClientSendTZfireOnOneErrback) r;connectrZrmrrr:rhost DeferredList)r^cAddrsAddrZ serverSend)rxre)rrr)cbClientStarteds   z2UDPTests.test_sendPackets..cbClientStartedcs`j}j}jd|j|jffg|j|jf}jd|fd|fd|fgdS)Nr{r|r}r~)r;rZr[r%rrm)r^rrZ clientAddrrxr'rer(r)cbSendsFinisheds   z2UDPTests.test_sendPackets..cbSendsFinishedcs$tjtjtjjgddSNTr)rrrr]rwrg)rir'r(r)rus   z-UDPTests.test_sendPackets..cbFinishedN)r9rrr/rrcrNrd)r' serverStartedrzr2rrrur()rxryrir'rer)test_sendPacketss    zUDPTests.test_sendPacketsZINFRASTRUCTUREZAZUREPIPELINESz"Hangs on Pipelines due to firewallcstt}_tjdddtt}_tjdddtj||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. rrWrXTrcs^t}_jddtdD]}jd|fjd|fdq|tj S)NrWP %drWr) rrr/r;rranger assertFailurerrJ)r^rKirr(r)ra3s  z2UDPTests.test_connectionRefused..cbStartedc"tjtjtjgddSrrrrr]rg)rmrwr(r)ru?   z3UDPTests.test_connectionRefused..cbFinishedN) rNrrr/rrcr9rrd)r'ryrr2rarur()rxrmrwr'rer)test_connectionRefuseds   zUDPTests.test_connectionRefusedc stt}_tjddddtt}_t}_tjdddtj||gdd}fdd}| |tj||gdd}fd d }| |fd d }| ||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. rrWrXsSending test packet to serverTrcs&jddjdjfdS)Ns$write to port no one is listening torrW)r;rZ_realPortNumberrg)rxre serverPorttest_data_to_sendr(r)cbClientAndServerStartedbs  zAUDPTests.test_serverReadFailure..cbClientAndServerStartedcsjdddSr)r[r%rg)r'rerr(r)verify_server_got_datausz?UDPTests.test_serverReadFailure..verify_server_got_datacrrrrg) clientPortrr(r)cleanupzrz0UDPTests.test_serverReadFailure..cleanupN) rNrrr/rrcr9r:rrd) r'ryrZ serverGotDataZserver_client_started_drZ all_data_sentrrr()rxrr'rerrr)test_serverReadFailureKs*    zUDPTests.test_serverReadFailurecCsVt}tjd|dd}|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. rrWrX localhostrN) rNrrcrrrZInvalidAddressErrorr;r RuntimeErrorr])r'rxrmr(r(r)test_badConnects zUDPTests.test_badConnectcstfdd}|ttjdddfdd}|tt dgfd d S) zr When datagramReceived raises an exception it is logged but the port is not disconnected. cs*t}t|ddt|fdS)z Flush the exceptions which the reactor should have logged and make sure they're actually there. z'Incorrectly found %d errors, expected 2N)flushLoggedErrorsrPr[len)ignZerrsr&r(r) cbCompleteds z8UDPTests.test_datagramReceivedError..cbCompletedrrWrXcstjfddS)z Disconnect the port we started and pass on whatever was given to us in case it was a Failure. cSr$r(rresultr(r)zHUDPTests.test_datagramReceivedError..cbCleanup..N)rrr]addBothrrtrr) cbCleanupsz6UDPTests.test_datagramReceivedError..cbCleanup<csszdWn tyYnwjddddf}t}|j |j j ffdd}fdd }| ||| jtd |jtd 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 transportrrcs<|tdkrtdjddSdS)z A packet arrived. Cancel the timeout for it, record it, and maybe finish the test. rrN)cancelr=rr callLaterr0)packet) finalDeferred makeAttemptsucceededAttempts timeoutCallr(r)cbPacketReceiveds    zRUDPTests.test_datagramReceivedError..makeAttempt..cbPacketReceivedcs 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)rr(r)ebPacketTimeouts zQUDPTests.test_datagramReceivedError..makeAttempt..ebPacketTimeoutg?z&Timed out in testDatagramReceivedError)Zfail ExceptionrDZfailIfIdenticalr;rrrrSrrrmZ addCallbacksZ addErrbackrrr TimeoutError)rZpacketDeferredrr)r?attemptsrxrrr'r)rr)rs,     z8UDPTests.test_datagramReceivedError..makeAttemptN) rrrdrRrrcrrZlistr)r'rrr()r?rrxrrrmr'rr)test_datagramReceivedErrors     :z#UDPTests.test_datagramReceivedErrorcCsRGddd}tdt}||_|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. c@rM)z6UDPTests.test_NoWarningOnBroadcast..fakeSocketcSsdSr$r()r'Zfoobarr(r(r)sendto sz=UDPTests.test_NoWarningOnBroadcast..fakeSocket.sendtoN)r6r7r8rr(r(r(r) fakeSocket rOrrstest)z iN) rZPortr9r rZ flushWarningstest_NoWarningOnBroadcastr[r)r'rr`warningsr(r(r)rs   z"UDPTests.test_NoWarningOnBroadcastN)r6r7r8rfrkrqrvrrosenvirongetrrrrrr(r(r(r)rVsD )< irVc@s8eZdZdZeedsdZddZddZ dd Z dS) ReactorShutdownInteractionTestsz!Test reactor shutdown interactionNrUcCs t|_tjd|jdd|_dS)zStart a UDP portrrWrXN)r9rerrcrmr&r(r(r)setUpsz%ReactorShutdownInteractionTests.setUpcCs |jS)zStop the UDP portN)rmr]r&r(r(r)tearDown#s z(ReactorShutdownInteractionTests.tearDowncshtt}j_fdd}||fdd}|jjddjjjfS)z0Test reactor shutdown while in a recvfrom() loopcs jjtdjddSr)rer;ZconnectionLostrrr0rgfinishedr'r(r)pktRece6s zQReactorShutdownInteractionTests.testShutdownFromDatagramReceived..pktRececs dSr$)rrgr&r(r) flushErrors@s zUReactorShutdownInteractionTests.testShutdownFromDatagramReceived..flushErrorss@rWN) rrrer:rdr;rrZrm)r'prrrr(rr) testShutdownFromDatagramReceived's    z@ReactorShutdownInteractionTests.testShutdownFromDatagramReceived) r6r7r8rQr IReactorUDPrskiprrrr(r(r(r)rs  rafr ipprotointjoingroupstrbindtoreturnboolcCs|t|t}||dft||}t||}z"z |||||Wnty1YW|dSwW|dS|w)a  Bind a socket to the given network interface and attempt to join a multicast group with the given address family. This is used to determine whether the local networking stack configuration allows for multicast within the given address family. rFTN)r r bindr setsockoptOSErrorclose)rrrrrsZ packgroupZpackifr(r(r)checkMulticastAvailabilityNs     rz'This reactor does not support multicast 225.0.0.2500.0.0.0z:The local networking stack does not enable IPv4 multicast.c@seZdZUdZdZded<dZded<dZded<d Zded <dZ ded <d Z ded <dZ ded<dZ ded<ddZ ddZd*ddZd*ddZd*ddZd*ddZd*dd Zeejocej d!d"d#Zd$d%Zeejd&d*d'd(Zd)S)+MulticastTestszs Tests for sending and receiving multicast datagrams. (This suite supports IPv4, see below for IPv6.) rrrY str | intexpectedInterfacerW clientAddressrmulticastGroupalternateInterfacerinterfaceSynonym invalidGroup::1wrongAddressFamilycCsXt|_t|_td|j|j|_td|j|j|_|jj |j |jj j dSr)r9rerCrxrlistenMulticastrYrirwr;rrrZrmr&r(r(r)rs zMulticastTests.setUpcCstt|jjt|jjgSr$)rrrir]rwr&r(r(r)rs   zMulticastTests.tearDownrNonecs<jjjtkr tddfdd }|j|jdS) Nz+only IPv4 multicast has TTLs, IPv6 has hopsoServer | ClientrrcsDt|jsJ|jd|jd|jddS)Nr-r)rZ providedByr;r[ZgetTTLZsetTTL)rr&r(r)checkttls z(MulticastTests.testTTL..checkttl)rrrr)rer;Z addressFamilyrr!rx)r'rr(r&r)testTTLs  zMulticastTests.testTTLcs|j}||jjd|jj}|jj|IdHt|j_|jj d||j f|jjIdH|t |jj d|jj d||jjd|jj d||j ftttdIdH|t |jj ddS)zv Test that after loopback mode has been set, multicast packets are delivered to their sender. r-Nr{r)rr[rer;ZgetLoopbackModerZ joinGrouprr:rrmrr%ZsetLoopbackModerrr)r'rr?r(r(r) test_loopbacks  zMulticastTests.test_loopbackcsN|t|jjj|j|jdIdHWddS1s wYdS)z A sensible error will be reported when using the wrong address family literal for the 'interface' argument to 'joinGroup'. rXN)rrrrer;rrrr&r(r(r)test_wrongFamilyInterfaceJoins  "z,MulticastTests.test_wrongFamilyInterfaceJoincs||jj|j||jj|j|jj|jIdH|jj|jIdH||jj|j||jj|jdS)zK Test C{getOutgoingInterface} and C{setOutgoingInterface}. N) r[rxr;ZgetOutgoingInterfacerreZsetOutgoingInterfacerrr&r(r(r)test_interfaces"  zMulticastTests.test_interfacecs^|jj|jIdH|jj|jIdH|jj|jIdH|jj|jIdHdS)zE Test that multicast a group can be joined and left. N)rxr;rrZ leaveGrouprer&r(r(r)test_joinLeaves zMulticastTests.test_joinLeavez2Windows' UDP multicast is not yet fully supported.cCs||jj|jtjS)z Test that an attempt to join an address which is not a multicast address fails with L{error.MulticastJoinError}. N)rrxr;rrrrr&r(r(r)test_joinFailures zMulticastTests.test_joinFailurecs|ttdjjjjjj}fdd}| |fdd}| |fdd}| ||S) zn Test that a multicast group can be joined and messages sent to and received from it. rcs(t}j_jdjjf|S)N hello world)rrer:r;rrrm)r^r2)r?cr'r(r)cbJoinedsz/MulticastTests.test_multicast..cbJoinedcsjjddddS)Nrr)r[rer%rgr&r(r)cbPacketsz/MulticastTests.test_multicast..cbPacketcs tj}|fdd|S)Ncrr$r(r passthroughr(r)r rz@MulticastTests.test_multicast..cleanup..)rr]rd)rr)r`rr)rs z.MulticastTests.test_multicast..cleanupN) r9rrrYrer;rZrrrd)r'Zjoinedrrrr()r?rr`r'r)test_multicasts      zMulticastTests.test_multicastzjon non-linux platforms it appears multiple processes can listen, but not multiple sockets in same process?c st}tt}|jd|d|jd}|}t|ttfsJ|j }t}|j||d|jd}t |j j |j| |j| |jgIdHt}|_t} |_|j d|j|ft || gIdH||jddd||jdddt t|jt|jgIdHdS)z Test that multiple sockets can listen on the same multicast port and that they both receive multicast messages directed to that address. rT)ZlistenMultiplerYNr)r9rrrrYrZ isinstancerrrmrrer;rrrr:rr[r%rr]) r'Z firstClientZmreactorZ firstPortZfpAddrZportnoZ secondClientZ secondPortZd1Zd2r(r(r)test_multiListensB     zMulticastTests.test_multiListenN)rr)r6r7r8rQrYrArrrrrrrrrrrrrrrrplatformZ isWindowsZisVistarrrr(r(r(r)res8                rff03::1::z/The local stack does not enable IPv6 multicast.c@seZdZUdZdZded<dZded<dZded<ed d e DZ d ed <e Z d ed <dZ ded<dZ d ed<dZded<dS)MulticastTestsIPv6zC Tests for sending and receiving IPv6 multicast datagrams. rrrYrrrrccs&|]}|ddr|dVqdS)r-lorN) startswith).0Zidxnmr(r(r) ^s$zMulticastTestsIPv6.rrrrrrrWrN)r6r7r8rQrYrArrnextr rrrrrr(r(r(r)rHs       r) rr rrrrrrrrrr)=rQ __future__rrr rrrrrrr r r r ZunittestrZtwisted.internetrrrrrrZtwisted.internet.addressrrZtwisted.internet.deferrrrZtwisted.internet.errorrZtwisted.internet.interfacesrrrZtwisted.internet.taskrZtwisted.pythonrZtwisted.test.testutilsr Ztwisted.trial.unittestr!r"r#DatagramProtocolr9ZConnectedDatagramProtocolrCrNrrPrRrrVrrrrr(r(r(r)sh 4       6  Z