U j@sdZddlZddlZddlmZmZmZmZddlm Z ddl m Z m Z ddl mZddlmZddlmZd d lmZmZd d lmZmZmZmZGd d d e ZGddde ZdZdZdZdZ e ej!GdddZ"dS)zD IProxyParser implementation for version two of the PROXY protocol. N)CallableTupleTypeUnion) implementer) ValueConstantValues)Literal)address)compat)_info _interfaces)InvalidNetworkProtocolInvalidProxyHeaderMissingAddressData convertErrorc@s0eZdZdZedZedZedZedZdS) NetFamilyz( Values for the 'family' field. r 0N) __name__ __module__ __qualname____doc__rUNSPECZINETINET6UNIXrrG/tmp/pip-unpacked-wheel-j5h_c8_q/twisted/protocols/haproxy/_v2parser.pyrs rc@s(eZdZdZedZedZedZdS) NetProtocolz& Values for 'protocol' field. rr N)rrrrrrZSTREAMDGRAMrrrrr )sr ZLOCALZPROXYc@seZdZdZdZdgZeedZdddddddZ d d d d Z e e e eje fe d fdddZee e dddZee e dddZee ejdddZd 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)!"12N)returncCs d|_dS)N)buffer)selfrrr__init__PszV2Parser.__init__NN)datar,cCs|j|7_t|jdkr"ttd|jdddd}t|j|krRdS|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!Hrr1Nr-)r.lenrstructunpackparse)r/r2sizeheader remaininginforrrfeedSs z V2Parser.feed) bytestringr,cCsdddt|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 }sz(V2Parser._bytesToIPv4..)joinr iterbytesr=rrr _bytesToIPv4qs 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} :c3s.|]&}t||ddddVqdS)rxr?N)intrArBZ hexStringrrrEsz(V2Parser._bytesToIPv6..rrrK)binasciib2a_hexrFrangerHrrNr _bytesToIPv6s  zV2Parser._bytesToIPv6)liner,c Cs|dd}d}ttt&t|dd}t|dd}W5QRX||jkrVt|t@|t@}}||jks|||jkrt|j|t krt |ddS|t@|t@}} tt t t|}t| } W5QRX|tjks| tjkrt |ddS|j|} |ddt| }|tjkrxttjtt| |\} } W5QRXt |t| dt| dSd} | tjkrd} tj}|j}|tjkrtj }|j!}ttjtt| |}|\} } }}W5QRXt ||| || "||| || "|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 r3rZTCPZUDP)#r IndexErrorrr@PREFIX_HIGH_LOWVERSIONSCOMMANDS _LOCALCOMMANDr ProxyInfo ValueErrorrrZ lookupByValuer rADDRESSFORMATSr5calcsizererrorrr6r Z UNIXAddressrstripr" IPv4AddressrIr IPv6AddressrRdecode)clsrSprefixZaddrInfoZversionCommandZ familyProtoversioncommandfamilyZnetprotoZ addressFormatsourcedestZaddrTypeZaddrClsZ addrParserr;ZsPortZdPortrrrr7sZ          zV2Parser.parse)rrrrrXr[r] _PROXYCOMMANDr\r`r0bytesrrr r^r< staticmethodrIrR classmethodr7rrrrr%9s*   r%)#rrOr5typingrrrrZzope.interfacerZ constantlyrrZtyping_extensionsr Ztwisted.internetr Ztwisted.pythonr r r _exceptionsrrrrrr rYrZr]rnZ IProxyParserr%rrrrs$