U  W[@sjdZddlmZddlmZddlmZmZmZm Z ddl m Z ddl m Z ee j Gdd d eZd S) zD IProxyParser implementation for version one of the PROXY protocol. ) implementer)address) convertErrorInvalidProxyHeaderInvalidNetworkProtocolMissingAddressData)_info) _interfacesc@sJeZdZdZdZdZdZdZeeefZdZ ddZ d d Z e d d Z d S)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 cCs d|_dS)N)buffer)selfrE/usr/lib/python3/dist-packages/twisted/protocols/haproxy/_v1parser.py__init__+szV1Parser.__init__cCsv|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. kr)NNr )r lenNEWLINErsplitpopparse)rdatalinesZ remainingheaderinforrrfeed/s  z V1Parser.feedc 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 krlt |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 ValueErrorrrPROXYSTRrALLOWED_NET_PROTOS UNKNOWN_PROTOr Z ProxyInfor TCP4_PROTOrZ IPv4AddressintZ IPv6Address) clslineZ originalLineZproxyStrZnetworkProtocolZ sourceAddrZ sourcePortZdestAddrZdestPortrrrrKsH          zV1Parser.parseN)__name__ __module__ __qualname____doc__rr!r"Z TCP6_PROTOr rrr classmethodrrrrrr sr N)r)Zzope.interfacerZtwisted.internetr _exceptionsrrrrr r Z IProxyParserobjectr rrrrs