o b.@sdZddlmZddlmZddlmZddlmZdZ e ddd d d fd e d e de de de de de fddZ e ddddd fd e d e de de de de de fddZ e ddddfd e d e de de de de f ddZ Gdd d ejZd!S)"z7 Test cases for L{twisted.protocols.haproxy.V2Parser}. )address)unittest) _v2parser)InvalidProxyHeaders QUIT !s$s s"sigverComfamProto addrLengthaddrsportsreturncC||||||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 rrN/usr/lib/python3/dist-packages/twisted/protocols/haproxy/test/test_v2parser.py_makeHeaderIPv6#rs scCr)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_makeHeaderIPv46rr1ss/home/tests/mysockets/sock/home/tests/mysockets/sockcCs|||||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@seZdZdZd4ddZd4ddZd4dd Zd4d d Zd4d d Zd4ddZ d4ddZ d4ddZ d4ddZ d4ddZ d4ddZd4ddZd4ddZd4ddZd4d d!Zd4d"d#Zd4d$d%Zd4d&d'Zd4d(d)Zd4d*d+Zd4d,d-Zd4d.d/Zd4d0d1Zd4d2d3ZdS)5 V2ParserTestsz? Test L{twisted.protocols.haproxy.V2Parser} behaviour. rNcCt}|tj|dS)zL Test if a well formed IPv4 header is parsed without error. N)r assertTruerV2Parserparseselfheaderrrrtest_happyPathIPv4z V2ParserTests.test_happyPathIPv4cCr)zL Test if a well formed IPv6 header is parsed without error. N)rrrrrrrrrtest_happyPathIPv6r"z V2ParserTests.test_happyPathIPv6cCr)zL Test if a well formed UNIX header is parsed without error. N)rrrrrrrrrtest_happyPathUnixr"z V2ParserTests.test_happyPathUnixcC tdd}|ttjj|dS)zN Test if an invalid signature block raises InvalidProxyError. s )rNr assertRaisesrrrrrrrrtest_invalidSignature z#V2ParserTests.test_invalidSignaturecCr%)zF Test if an invalid version raises InvalidProxyError. rr Nr&rrrrtest_invalidVersionr)z!V2ParserTests.test_invalidVersioncCr%)zF Test if an invalid command raises InvalidProxyError. #r*Nr&rrrrtest_invalidCommandr)z!V2ParserTests.test_invalidCommandcCr%)zE Test if an invalid family raises InvalidProxyError. @r Nr&rrrrtest_invalidFamilyr)z V2ParserTests.test_invalidFamilycCr%)zG Test if an invalid protocol raises InvalidProxyError. $r/Nr&rrrrtest_invalidProtor)zV2ParserTests.test_invalidProtocC2tdd}tj|}||j||jdS)zU Test that local does not return endpoint data for IPv4 connections.  r*Nrrrr assertFalsesource destinationrr inforrrtest_localCommandIpv4   z#V2ParserTests.test_localCommandIpv4cCr3)zU Test that local does not return endpoint data for IPv6 connections. r4r*Nrrrrr6r7r8r9rrrtest_localCommandIpv6r<z#V2ParserTests.test_localCommandIpv6cCr3)zU Test that local does not return endpoint data for UNIX connections. r4r*Nrrrrr6r7r8r9rrrtest_localCommandUnixr<z#V2ParserTests.test_localCommandUnixcCRtdd}tj|}||j||jtj||j ||j tjdS)zM Test that proxy returns endpoint data for IPv4 connections. rr*N) rrrrrr7assertIsInstancer IPv4Addressr8r9rrrtest_proxyCommandIpv4    z#V2ParserTests.test_proxyCommandIpv4cCrA)zM Test that proxy returns endpoint data for IPv6 connections. rr*N) rrrrrr7rBr IPv6Addressr8r9rrrtest_proxyCommandIpv6rEz#V2ParserTests.test_proxyCommandIpv6cCrA)zM Test that proxy returns endpoint data for UNIX connections. rr*N) rrrrrr7rBr UNIXAddressr8r9rrrtest_proxyCommandUnixrEz#V2ParserTests.test_proxyCommandUnixcCr3)V Test that UNSPEC does not return endpoint data for IPv4 connections. r/Nr5r9rrrtest_unspecFamilyIpv4r<z#V2ParserTests.test_unspecFamilyIpv4cCr3)V Test that UNSPEC does not return endpoint data for IPv6 connections. rKr/Nr=r9rrrtest_unspecFamilyIpv6r<z#V2ParserTests.test_unspecFamilyIpv6cCr3)V Test that UNSPEC does not return endpoint data for UNIX connections. rKr/Nr?r9rrrtest_unspecFamilyUnix!r<z#V2ParserTests.test_unspecFamilyUnixcCr3)rJr/Nr5r9rrrtest_unspecProtoIpv4*r<z"V2ParserTests.test_unspecProtoIpv4cCr3)rMr4r/Nr=r9rrrtest_unspecProtoIpv63r<z"V2ParserTests.test_unspecProtoIpv6cCr3)rO0r/Nr?r9rrrtest_unspecProtoUnix<r<z"V2ParserTests.test_unspecProtoUnixcC>d}t|}t}||\}}|||||dS)zU Test that overflow bits are preserved during feed parsing for IPv4. TEST DATA TEST DATAN)rrrfeedr assertEqualr testValuer parserr:overflowrrrtest_overflowIpv4E   zV2ParserTests.test_overflowIpv4cCrV)zU Test that overflow bits are preserved during feed parsing for IPv6. rWN)rrrrXrrYrZrrrtest_overflowIpv6Pr_zV2ParserTests.test_overflowIpv6cCrV)zU Test that overflow bits are preserved during feed parsing for Unix. rWN)rrrrXrrYrZrrrtest_overflowUnix[r_zV2ParserTests.test_overflowUnixcCs d}t}|t|j|dS)zK Test that an initial payload of less than 16 bytes fails. s NEEDMOREDATAN)rrr'rrX)rr[r\rrrtest_segmentTooSmallfsz"V2ParserTests.test_segmentTooSmall)rN)__name__ __module__ __qualname____doc__r!r#r$r(r+r-r0r2r;r>r@rDrGrIrLrNrPrRrSrUr^r`rarbrrrrrs4      rN)rftwisted.internetr twisted.trialrr _exceptionsr V2_SIGNATUREbytesrrrTestCaserrrrrs     ' ' &