U  W[@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"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!Hr)NNNr')r(lenr structunpackparse)r)datasizeheaderZ remaininginforrrfeedNs z V2Parser.feedcCsdddt|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 wsz(V2Parser._bytesToIPv4..)joinrZ iterbytesZ bytestringrrr _bytesToIPv4ks 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|](}dt||ddfdVqdS)z%xrr6N)intr8r9Z hexStringrrr<sz(V2Parser._bytesToIPv6..rrrA)binasciiZb2a_hexr=ranger>rrCr _bytesToIPv6}s  zV2Parser._bytesToIPv6c 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 r+rZTCPZUDP)"r IndexErrorr r7PREFIX_HIGH_LOWVERSIONSCOMMANDS _LOCALCOMMANDr Z ProxyInfo ValueErrorr rZ lookupByValuerrADDRESSFORMATSr-Zcalcsizererrorr r.rZ UNIXAddressrstriprZ IPv4Addressr?rZ IPv6AddressrF)clslineprefixZaddrInfoZversionCommandZ familyProtoversionZcommandZfamilyZnetprotoZ addressFormatsourcedestZaddrTypeZaddrClsZ addrParserr3ZsPortZdPortrrrr/s^          zV2Parser.parseN)rrrrrKrNrP _PROXYCOMMANDrOrRr*r4 staticmethodr?rF classmethodr/rrrrr 3s&    r )rrDr-Z constantlyrrZzope.interfacerZtwisted.internetrZtwisted.pythonr _exceptionsrr r r r r rrrLrMrPr[Z IProxyParserobjectr rrrrs"