← Back
Editing: _v2parser.cpython-38.pyc
U W[� � @ s� d Z ddlZddlZddlmZmZ ddlmZ ddlm Z ddl m Z ddl m Z mZmZmZ dd lmZ dd lmZ G d d � d e�ZG d d� de�ZdZdZdZdZeej�G dd� de��Z dS )zD IProxyParser implementation for version two of the PROXY protocol. � N)�Values� ValueConstant)� implementer)�address)�compat� )� convertError�InvalidProxyHeader�InvalidNetworkProtocol�MissingAddressData)�_info)� _interfacesc @ s0 e Zd ZdZed�Zed�Zed�Zed�ZdS )� NetFamilyz( Values for the 'family' field. r � � �0 N) �__name__� __module__� __qualname__�__doc__r �UNSPECZINET�INET6�UNIX� r r �E/usr/lib/python3/dist-packages/twisted/protocols/haproxy/_v2parser.pyr s r c @ s( e Zd ZdZed�Zed�Zed�ZdS )� NetProtocolz& Values for 'protocol' field. r r � N)r r r r r r ZSTREAM�DGRAMr r r r r % s r �� � ZLOCALZPROXYc @ sj e Zd ZdZdZdgZeed�Zddddddd�Z d d � Z d d � Z e d d� �Z e dd� �Zedd� �ZdS )�V2Parserzn PROXY protocol version two header parser. Version two of the PROXY protocol is a binary format. s QUIT r )r r z!4s4s2Hz !16s16s2Hz !108s108s)� � �! �" �1 �2 c C s d| _ d S )N� )�buffer)�selfr r r �__init__J s zV2Parser.__init__c C s� | j |7 _ t| j �dk r"t� �t�d| j dd� �d d }t| j �|k rRdS | j d|� | 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. r z!H� r )NNNr' )r( �lenr �struct�unpack�parse)r) �data�size�headerZ remaining�infor r r �feedN s z V2Parser.feedc C s d� dd� t�| �D ��S )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} � .c s s" | ]}d t |�f �d�V qdS )z%i�asciiN)�ord�encode��.0�br r r � <genexpr>w s �z(V2Parser._bytesToIPv4.<locals>.<genexpr>)�joinr Z iterbytes�Z bytestringr r r � _bytesToIPv4k s �zV2Parser._bytesToIPv4c s* t �| �� d�� fdd�tddd�D ��S )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} � :c 3 s0 | ](}d t � ||d � d�f �d�V qdS )z%x� r r6 N)�intr8 r9 �Z hexStringr r r< � s �z(V2Parser._bytesToIPv6.<locals>.<genexpr>r r rA )�binasciiZb2a_hexr= �ranger> r rC r � _bytesToIPv6} s �zV2Parser._bytesToIPv6c C s |dd� }d}t tt��& t|dd� �}t|dd� �}W 5 Q R X || jkrVt� �|t@ |t@ }}|| jks||| jkr�t� �| j| t kr�t � |dd�S |t@ |t@ }} t t t �� t�|�}t�| �} W 5 Q R X |tjks�| tjk� r�t � |dd�S | j| } |ddt�| � � }|tjk�rxt tjt�� t�| |�\} } W 5 Q R X t � |t�| �d��t�| �d���S d} | tj k�r�d} tj }| j }|tj k�r�tj }| j!}t tjt�� t�| |�}|\} } }}W 5 Q R X t � ||| || �|�|| || �|��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+ r � ZTCPZUDP)"r � IndexErrorr r7 �PREFIX�_HIGH�_LOW�VERSIONS�COMMANDS� _LOCALCOMMANDr Z ProxyInfo� ValueErrorr r Z lookupByValuer r �ADDRESSFORMATSr- Zcalcsizer �errorr r. r Z UNIXAddress�rstripr Z IPv4Addressr? r Z IPv6AddressrF )�cls�line�prefixZaddrInfoZversionCommandZ familyProto�versionZcommandZfamilyZnetprotoZ addressFormat�source�destZaddrTypeZaddrClsZ addrParserr3 ZsPortZdPortr r r r/ � s^ �� � �zV2Parser.parseN)r r r r rK rN rP � _PROXYCOMMANDrO rR r* r4 � staticmethodr? rF � classmethodr/ r r r r r 3 s& � r ) r rD r- Z constantlyr r Zzope.interfacer Ztwisted.internetr Ztwisted.pythonr � _exceptionsr r r r � r r r r rL rM rP r[ Z IProxyParser�objectr r r r r �<module> s"
Save File
Cancel