U lj@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 ddd d Z e e e eje fe d fd d dZee ejdddZdS)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 N)returncCs d|_dS)N)buffer)selfrG/tmp/pip-unpacked-wheel-aghyhbke/twisted/protocols/haproxy/_v1parser.py__init__,szV1Parser.__init__NN)datarcCsv|j|7_t|jdkr.|j|jkr.t|j|jd}t|dksNdSd|_|}|}||}||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.feed)linerc Cs|}d}d}d}d}d}d}ttt|dd\}}W5QRX||jkrRtttt|dd\}}W5QRX||jkrt||jkrt |ddSttt |dd\}}W5QRXttt |dd\}}W5QRXttt |dd\}}W5QRXttt |dd}W5QRX||j krtt |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 rrZTCP)r ValueErrorr rPROXYSTRr ALLOWED_NET_PROTOS UNKNOWN_PROTOr ProxyInfor TCP4_PROTOr IPv4Addressdecodeint IPv6Address) clsr"Z originalLineZproxyStrZnetworkProtocolZ sourceAddrZ sourcePortZdestAddrZdestPortrrrrLsH          zV1Parser.parse)__name__ __module__ __qualname____doc__r%r'r)Z TCP6_PROTOr&rrbytesrrrr(r! classmethodrrrrrr s  r N)r2typingrrZzope.interfacerZtwisted.internetrrr _exceptionsr r r r Z IProxyParserr rrrrs