3 +sY@sdZddlZddlZddlmZmZddlmZddlm Z ddl m Z ddl m Z mZmZmZdd lmZdd lmZGd d d eZGd ddeZdZdZdZdZeejGdddeZdS)zD IProxyParser implementation for version two of the PROXY protocol. N)Values ValueConstant) implementer)address)compat) convertErrorInvalidProxyHeaderInvalidNetworkProtocolMissingAddressData)_info) _interfacesc@s0eZdZdZedZedZedZedZdS) NetFamilyz( Values for the 'family' field. r 0N) __name__ __module__ __qualname____doc__rUNSPECZINETINET6UNIXrrE/usr/lib/python3/dist-packages/twisted/protocols/haproxy/_v2parser.pyrs rc@s(eZdZdZedZedZedZdS) NetProtocolz& Values for 'protocol' field. rrN)rrrrrrZSTREAMDGRAMrrrrr%srZLOCALZPROXYc@sjeZdZdZdZdgZeedZdddddddZ d d Z d d Z e d dZ e ddZeddZdS)V2Parserzn PROXY protocol version two header parser. Version two of the PROXY protocol is a binary format. s QUIT r)rrz!4s4s2Hz !16s16s2Hz !108s108s)!"12cCs d|_dS)N)buffer)selfrrr__init__JszV2Parser.__init__cCs|j|7_t|jdkr"ttjd|jdddd}t|j|krRdS|jd||j|d}}d|_|j|}||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!HrNr')NN)r(lenr structunpackparse)r)datasizeheaderZ remaininginforrrfeedNs z V2Parser.feedcCsdjddtj|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|fjdVqdS)z%iasciiN)ordencode).0brrr xsz(V2Parser._bytesToIPv4..)joinrZ iterbytes) bytestringrrr _bytesToIPv4ks zV2Parser._bytesToIPv4cs*tj|djfddtdddDS)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|](}dt||ddfjdVqdS)z%xrr6N)intr8)r9r:) hexStringrrr;sz(V2Parser._bytesToIPv6..rrr@)binasciiZb2a_hexr<range)r=r)rBr _bytesToIPv6}s  zV2Parser._bytesToIPv6c#Cs|dd}d}ttt&t|dd}t|dd}WdQRX||jkrVt|t@|t@}}||jks|||jkrt|j|t krt j |ddS|t@|t@}} tt t tj|}tj| } WdQRX|tjks| tjkrt j |ddS|j|} |ddtj| }|tjkrxttjttj| |\} } WdQRXt j |tj| jdtj| jdSd} | tjkrd} tj}|j}|tjkrtj }|j!}ttjttj| |}|\} } }}WdQRXt j ||| || ||| || |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 r+rZTCPZUDP)"r IndexErrorr r7PREFIX_HIGH_LOWVERSIONSCOMMANDS _LOCALCOMMANDr Z ProxyInfo ValueErrorr rZ lookupByValuerrADDRESSFORMATSr-Zcalcsizererrorr r.rZ UNIXAddressrstriprZ IPv4Addressr>rZ IPv6AddressrE)clslineprefixZaddrInfoZversionCommandZ familyProtoversionZcommandZfamilyZnetprotoZ addressFormatsourcedestZaddrTypeZaddrClsZ addrParserr3ZsPortZdPortrrrr/sV            zV2Parser.parseN)rrrrrJrMrO _PROXYCOMMANDrNrQr*r4 staticmethodr>rE classmethodr/rrrrr 3s   r )rrCr-Z constantlyrrZzope.interfacerZtwisted.internetrZtwisted.pythonr _exceptionsrr r r r r rrrKrLrOrZZ IProxyParserobjectr rrrrs"