o pj@sdZddlZddlZddlmZmZmZmZmZddl m Z ddl m Z m Z ddlmZddlmZdd lmZmZdd lmZmZmZmZGd d d e ZGd dde ZdZdZdZdZe ej GdddZ!dS)zD IProxyParser implementation for version two of the PROXY protocol. N)CallableLiteralTupleTypeUnion) implementer) ValueConstantValues)address)compat)_info _interfaces)InvalidNetworkProtocolInvalidProxyHeaderMissingAddressData convertErrorc@s0eZdZdZedZedZedZedZdS) NetFamilyz( Values for the 'family' field. r 0N) __name__ __module__ __qualname____doc__rUNSPECINETINET6UNIXrrE/usr/lib/python3/dist-packages/twisted/protocols/haproxy/_v2parser.pyrs  rc@s(eZdZdZedZedZedZdS) NetProtocolz& Values for 'protocol' field. rr N)rrrrrrSTREAMDGRAMrrrr r!(s  r!LOCALPROXYc@seZdZdZdZdgZeedZdddddddZ dd d Z d e d e e eje fe dffddZede d e fddZede d e fddZede d ejfddZd S)V2Parserzn PROXY protocol version two header parser. Version two of the PROXY protocol is a binary format. s QUIT r)rr z!4s4s2Hz !16s16s2Hz !108s108s)!"12returnNcCs d|_dS)N)buffer)selfrrr __init__Os zV2Parser.__init__dataNNcCs|j|7_t|jdkrttd|jdddd}t|j|kr)dS|jd||j|d}}d|_||}||fS)a Consume a chunk of data and attempt to parse it. @param data: A bytestring. @type data: bytes @return: A two-tuple containing, in order, a L{_interfaces.IProxyInfo} and any bytes fed to the parser that followed the end of the header. Both of these values are None until a complete header is parsed. @raises InvalidProxyHeader: If the bytes fed to the parser create an invalid PROXY header. rz!Hrr6Nr1)r2lenrstructunpackparse)r3r5sizeheader remaininginforrr feedRs z V2Parser.feed bytestringcCsdddt|DS)a7 Convert packed 32-bit IPv4 address bytes into a dotted-quad ASCII bytes representation of that address. @param bytestring: 4 octets representing an IPv4 address. @type bytestring: L{bytes} @return: a dotted-quad notation IPv4 address. @rtype: L{bytes} .css$|] }dt|fdVqdS)z%iasciiN)ordencode.0brrr |s z(V2Parser._bytesToIPv4..)joinr iterbytesrArrr _bytesToIPv4ps zV2Parser._bytesToIPv4cs*t|dfddtdddDS)a= Convert packed 128-bit IPv6 address bytes into a colon-separated ASCII bytes representation of that address. @param bytestring: 16 octets representing an IPv6 address. @type bytestring: L{bytes} @return: a dotted-quad notation IPv6 address. @rtype: L{bytes} :c3s0|]}t||ddddVqdS)rxrCN)intrErF hexStringrr rIs  z(V2Parser._bytesToIPv6..rrrO)binasciib2a_hexrJrangerLrrRr _bytesToIPv6s  zV2Parser._bytesToIPv6linec CsX|dd}d}tttt|dd}t|dd}Wdn1s(wY||jkr5t|t@|t@}}||jvsH||jvrKt|j|t krYt |ddS|t@|t@}} tt t t|}t| } Wdn1s|wY|tjus| tjurt |ddS|j|} |ddt| }|tjurttjtt| |\} } Wdn1swYt |t| dt| dSd} | tjurd} tj}|j}|tjurtj }|j!}ttjtt| |}|\} } }}Wdn 1swYt ||| || "||| || "|S) a Parse a bytestring as a full PROXY protocol header. @param line: A bytestring that represents a valid HAProxy PROXY protocol version 2 header. @type line: bytes @return: A L{_interfaces.IProxyInfo} containing the parsed data. @raises InvalidProxyHeader: If the bytestring does not represent a valid PROXY header. N r7rTCPUDP)#r IndexErrorrrDPREFIX_HIGH_LOWVERSIONSCOMMANDS _LOCALCOMMANDr ProxyInfo ValueErrorrr lookupByValuer!rADDRESSFORMATSr9calcsizererrorrr:r UNIXAddressrstripr$ IPv4AddressrMr IPv6AddressrWdecode)clsrXprefixaddrInfoversionCommand familyProtoversioncommandfamilynetproto addressFormatsourcedestaddrTypeaddrCls addrParserr?sPortdPortrrr r;sb           zV2Parser.parse)r0N)rrrrr_rbrd _PROXYCOMMANDrcrhr4bytesrrr rer@ staticmethodrMrW classmethodr;rrrr r)8s0   r))"rrTr9typingrrrrrzope.interfacer constantlyrr twisted.internetr twisted.pythonr r r _exceptionsrrrrrr!r`rardr IProxyParserr)rrrr s$