o b@spdZddlmZmZddlmZddlmZddlm Z m Z ddl m Z m Z mZmZee jGdd d Zd S) zD IProxyParser implementation for version one of the PROXY protocol. )TupleUnion) implementer)address)_info _interfaces)InvalidNetworkProtocolInvalidProxyHeaderMissingAddressData convertErrorc@sxeZdZdZdZdZdZdZeeefZdZ dd d Z d e de e eje fe d ffd dZede dejfddZdS)V1Parserz PROXY protocol version one header parser. Version one of the PROXY protocol is a human readable format represented by a single, newline delimited binary string that contains all of the relevant source and destination data. sPROXYsUNKNOWNsTCP4sTCP6s returnNcCs d|_dS)N)buffer)selfrE/usr/lib/python3/dist-packages/twisted/protocols/haproxy/_v1parser.py__init__,s zV1Parser.__init__dataNNcCsv|j|7_t|jdkr|j|jvrt|j|jd}t|dks'dSd|_|}|}||}||fS)a Consume a chunk of data and attempt to parse it. @param data: A bytestring. @type data: L{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. krrr)rlenNEWLINEr splitpopparse)rrlines remainingheaderinforrrfeed/s  z V1Parser.feedlinec Cs|}d}d}d}d}d}d}ttt|dd\}}Wdn1s&wY||jkr3tttt|dd\}}Wdn1sKwY||jvrXt||jkrdt |ddSttt |dd\}}Wdn1s|wYttt |dd\}}Wdn1swYttt |dd\}}Wdn1swYttt |dd}Wdn1swY||j krt |t d|t|t d|t|St |t d|t|t d|t|S)a Parse a bytestring as a full PROXY protocol header line. @param line: A bytestring that represents a valid HAProxy PROXY protocol header line. @type line: bytes @return: A L{_interfaces.IProxyInfo} containing the parsed data. @raises InvalidProxyHeader: If the bytestring does not represent a valid PROXY header. @raises InvalidNetworkProtocol: When no protocol can be parsed or is not one of the allowed values. @raises MissingAddressData: When the protocol is TCP* but the header does not contain a complete set of addresses and ports. N rrTCP)r ValueErrorr rPROXYSTRr ALLOWED_NET_PROTOS UNKNOWN_PROTOr ProxyInfor TCP4_PROTOr IPv4Addressdecodeint IPv6Address) clsr" originalLineproxyStrnetworkProtocol sourceAddr sourcePortdestAddrdestPortrrrrLsT          zV1Parser.parse)rN)__name__ __module__ __qualname____doc__r&r(r* TCP6_PROTOr'rrbytesrrrr)r! classmethodrrrrrr s&  r N)r:typingrrzope.interfacertwisted.internetrrr _exceptionsr r r r IProxyParserr rrrrs