3 +sY@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|jj|jd}t|dksNdSd|_|j}|j}|j|}||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. krNr )NN)r lenNEWLINErsplitpopparse)rdatalinesZ remainingheaderinforrrfeed/s  z V1Parser.feedc 2Cs|}d}d}d}d}d}d}ttt|jdd\}}WdQRX||jkrRtttt|jdd\}}WdQRX||jkrt||jkrtj |ddSttt |jdd\}}WdQRXttt |jdd\}}WdQRXttt |jdd\}}WdQRXttt |jdd}WdQRX||j krltj |t j d|t|t j d|t|Stj |t jd|t|t jd|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 sourcePortZdestAddrZdestPortrrrrKsD          zV1Parser.parseN)__name__ __module__ __qualname____doc__rr!r"Z TCP6_PROTOr rrr classmethodrrrrrr sr N)r)Zzope.interfacerZtwisted.internetr _exceptionsrrrrr r Z IProxyParserobjectr rrrrs