U n5{i. @sdZddlmZddlmZddlmZddlmZdZ e ddd d d fe e e e e e e d d dZ e ddddd fe e e e e e e d ddZ e ddddfe e e e e e dddZ GdddejZdS)z7 Test cases for L{twisted.protocols.haproxy.V2Parser}. )address)unittest) _v2parser)InvalidProxyHeaders QUIT !s$s s")sigverComfamProto addrLengthaddrsportsreturncCs||||||S)aV Construct a version 2 IPv6 header with custom bytes. @param sig: The protocol signature; defaults to valid L{V2_SIGNATURE}. @type sig: L{bytes} @param verCom: Protocol version and command. Defaults to V2 PROXY. @type verCom: L{bytes} @param famProto: Address family and protocol. Defaults to AF_INET6/STREAM. @type famProto: L{bytes} @param addrLength: Network-endian byte length of payload. Defaults to description of default addrs/ports. @type addrLength: L{bytes} @param addrs: Address payload. Defaults to C{::1} for source and destination. @type addrs: L{bytes} @param ports: Source and destination ports. Defaults to 8080 for source 8888 for destination. @type ports: L{bytes} @return: A packet with header, addresses, and ports. @rtype: L{bytes} rr r r r r rrP/tmp/pip-unpacked-wheel-mnhg7r0q/twisted/protocols/haproxy/test/test_v2parser.py_makeHeaderIPv6s#rs scCs||||||S)aX Construct a version 2 IPv4 header with custom bytes. @param sig: The protocol signature; defaults to valid L{V2_SIGNATURE}. @type sig: L{bytes} @param verCom: Protocol version and command. Defaults to V2 PROXY. @type verCom: L{bytes} @param famProto: Address family and protocol. Defaults to AF_INET/STREAM. @type famProto: L{bytes} @param addrLength: Network-endian byte length of payload. Defaults to description of default addrs/ports. @type addrLength: L{bytes} @param addrs: Address payload. Defaults to 127.0.0.1 for source and destination. @type addrs: L{bytes} @param ports: Source and destination ports. Defaults to 8080 for source 8888 for destination. @type ports: L{bytes} @return: A packet with header, addresses, and ports. @rtype: L{bytes} rrrrr_makeHeaderIPv46s#r1ss/home/tests/mysockets/sock/home/tests/mysockets/sock)rr r r r rcCs|||||S)a Construct a version 2 IPv4 header with custom bytes. @param sig: The protocol signature; defaults to valid L{V2_SIGNATURE}. @type sig: L{bytes} @param verCom: Protocol version and command. Defaults to V2 PROXY. @type verCom: L{bytes} @param famProto: Address family and protocol. Defaults to AF_UNIX/STREAM. @type famProto: L{bytes} @param addrLength: Network-endian byte length of payload. Defaults to 108 bytes for 2 null terminated paths. @type addrLength: L{bytes} @param addrs: Address payload. Defaults to C{/home/tests/mysockets/sock} for source and destination paths. @type addrs: L{bytes} @return: A packet with header, addresses, and8 ports. @rtype: L{bytes} r)rr r r r rrr_makeHeaderUnix\s#rc@s`eZdZdZddddZddddZdddd Zddd d Zddd d ZddddZ ddddZ ddddZ ddddZ ddddZ ddddZddddZddddZddddZddd d!Zddd"d#Zddd$d%Zddd&d'Zddd(d)Zddd*d+Zddd,d-Zddd.d/Zddd0d1Zddd2d3ZdS)4 V2ParserTestsz? Test L{twisted.protocols.haproxy.V2Parser} behaviour. N)rcCst}|tj|dS)zL Test if a well formed IPv4 header is parsed without error. N)r assertTruerV2Parserparseselfheaderrrrtest_happyPathIPv4sz V2ParserTests.test_happyPathIPv4cCst}|tj|dS)zL Test if a well formed IPv6 header is parsed without error. N)rrrrrrrrrtest_happyPathIPv6sz V2ParserTests.test_happyPathIPv6cCst}|tj|dS)zL Test if a well formed UNIX header is parsed without error. N)rrrrrrrrrtest_happyPathUnixsz V2ParserTests.test_happyPathUnixcCs tdd}|ttjj|dS)zN Test if an invalid signature block raises InvalidProxyError. s )rNr assertRaisesrrrrrrrrtest_invalidSignatures  z#V2ParserTests.test_invalidSignaturecCs tdd}|ttjj|dS)zF Test if an invalid version raises InvalidProxyError. rr Nr!rrrrtest_invalidVersions  z!V2ParserTests.test_invalidVersioncCs tdd}|ttjj|dS)zF Test if an invalid command raises InvalidProxyError. #r$Nr!rrrrtest_invalidCommands  z!V2ParserTests.test_invalidCommandcCs tdd}|ttjj|dS)zE Test if an invalid family raises InvalidProxyError. @r Nr!rrrrtest_invalidFamilys  z V2ParserTests.test_invalidFamilycCs tdd}|ttjj|dS)zG Test if an invalid protocol raises InvalidProxyError. $r)Nr!rrrrtest_invalidProtos  zV2ParserTests.test_invalidProtocCs2tdd}tj|}||j||jdS)zU Test that local does not return endpoint data for IPv4 connections.  r$Nrrrr assertFalsesource destinationrrinforrrtest_localCommandIpv4s   z#V2ParserTests.test_localCommandIpv4cCs2tdd}tj|}||j||jdS)zU Test that local does not return endpoint data for IPv6 connections. r-r$Nrrrrr/r0r1r2rrrtest_localCommandIpv6s   z#V2ParserTests.test_localCommandIpv6cCs2tdd}tj|}||j||jdS)zU Test that local does not return endpoint data for UNIX connections. r-r$Nrrrrr/r0r1r2rrrtest_localCommandUnixs   z#V2ParserTests.test_localCommandUnixcCsRtdd}tj|}||j||jtj||j ||j tjdS)zM Test that proxy returns endpoint data for IPv4 connections. rr$N) rrrrrr0assertIsInstancer IPv4Addressr1r2rrrtest_proxyCommandIpv4s     z#V2ParserTests.test_proxyCommandIpv4cCsRtdd}tj|}||j||jtj||j ||j tjdS)zM Test that proxy returns endpoint data for IPv6 connections. rr$N) rrrrrr0r9r IPv6Addressr1r2rrrtest_proxyCommandIpv6s     z#V2ParserTests.test_proxyCommandIpv6cCsRtdd}tj|}||j||jtj||j ||j tjdS)zM Test that proxy returns endpoint data for UNIX connections. rr$N) rrrrrr0r9rZ UNIXAddressr1r2rrrtest_proxyCommandUnixs     z#V2ParserTests.test_proxyCommandUnixcCs2tdd}tj|}||j||jdS)V Test that UNSPEC does not return endpoint data for IPv4 connections. r)Nr.r2rrrtest_unspecFamilyIpv4s   z#V2ParserTests.test_unspecFamilyIpv4cCs2tdd}tj|}||j||jdS)V Test that UNSPEC does not return endpoint data for IPv6 connections. r@r)Nr5r2rrrtest_unspecFamilyIpv6s   z#V2ParserTests.test_unspecFamilyIpv6cCs2tdd}tj|}||j||jdS)V Test that UNSPEC does not return endpoint data for UNIX connections. r@r)Nr7r2rrrtest_unspecFamilyUnix!s   z#V2ParserTests.test_unspecFamilyUnixcCs2tdd}tj|}||j||jdS)r?r)Nr.r2rrrtest_unspecProtoIpv4*s   z"V2ParserTests.test_unspecProtoIpv4cCs2tdd}tj|}||j||jdS)rBr-r)Nr5r2rrrtest_unspecProtoIpv63s   z"V2ParserTests.test_unspecProtoIpv6cCs2tdd}tj|}||j||jdS)rD0r)Nr7r2rrrtest_unspecProtoUnix<s   z"V2ParserTests.test_unspecProtoUnixcCs>d}t|}t}||\}}|||||dS)zU Test that overflow bits are preserved during feed parsing for IPv4. TEST DATA TEST DATAN)rrrfeedr assertEqualr testValuerparserr3Zoverflowrrrtest_overflowIpv4Es   zV2ParserTests.test_overflowIpv4cCs>d}t|}t}||\}}|||||dS)zU Test that overflow bits are preserved during feed parsing for IPv6. rKN)rrrrLrrMrNrrrtest_overflowIpv6Ps   zV2ParserTests.test_overflowIpv6cCs>d}t|}t}||\}}|||||dS)zU Test that overflow bits are preserved during feed parsing for Unix. rKN)rrrrLrrMrNrrrtest_overflowUnix[s   zV2ParserTests.test_overflowUnixcCs d}t}|t|j|dS)zK Test that an initial payload of less than 16 bytes fails. s NEEDMOREDATAN)rrr"rrL)rrOrPrrrtest_segmentTooSmallfsz"V2ParserTests.test_segmentTooSmall)__name__ __module__ __qualname____doc__rrr r#r%r'r*r,r4r6r8r;r=r>rArCrErGrHrJrQrRrSrTrrrrrs2                    rN)rXZtwisted.internetrZ twisted.trialrr _exceptionsrZ V2_SIGNATUREbytesrrrZTestCaserrrrrsb     ' ' &