File: //usr/lib/python3/dist-packages/twisted/protocols/haproxy/__pycache__/_v2parser.cpython-310.pyc
o
^h�b� � @ s� d Z ddlZddlZddlmZmZmZmZmZ ddl m
Z
ddl m Z m
Z
ddlmZ ddlmZ dd lmZmZ dd
lmZmZmZmZ G d d � d e
�ZG d
d� de
�ZdZ dZ dZ dZ e
ej �G dd� d��Z!dS )zD
IProxyParser implementation for version two of the PROXY protocol.
� N)�Callable�Literal�Tuple�Type�Union)� implementer)�
ValueConstant�Values)�address)�compat� )�_info� _interfaces)�InvalidNetworkProtocol�InvalidProxyHeader�MissingAddressData� convertErrorc @ s0 e Zd ZdZed�Zed�Zed�Zed�ZdS )� NetFamilyz(
Values for the 'family' field.
r � � �0 N) �__name__�
__module__� __qualname__�__doc__r �UNSPEC�INET�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 �STREAM�DGRAMr r r r r! ( s
r! �� � �LOCAL�PROXYc @ s� e Zd ZdZdZdgZeed�Zddddddd�Z dd d �Z
d
e d e e
eje f e
d f fdd�Zede d e fdd��Zede d e fdd��Zede d ejfdd��Zd
S )�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 �returnNc C s
d| _ d S )N� )�buffer)�selfr r r �__init__O s
zV2Parser.__init__�data�NNc C s� | j |7 _ t| j �dk rt� �t�d| j dd� �d d }t| j �|k r)dS | 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 r6 Nr1 )r2 �lenr �struct�unpack�parse)r3 r5 �size�header� remaining�infor r r �feedR s
z
V2Parser.feed�
bytestringc 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>| s �
�z(V2Parser._bytesToIPv4.<locals>.<genexpr>)�joinr � iterbytes�rA r r r � _bytesToIPv4p 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 � | ]}t � ||d � d�d��d�V qdS )� r �xrC N)�intrE rF �� hexStringr r rI � s
� �
�z(V2Parser._bytesToIPv6.<locals>.<genexpr>r r rO )�binascii�b2a_hexrJ �rangerL r rR r � _bytesToIPv6� s
�zV2Parser._bytesToIPv6�linec C sX |dd� }d}t tt�� t|dd� �}t|dd� �}W d � n1 s(w Y || jkr5t� �|t@ |t@ }}|| jvsH|| jvrKt� �| j| t krYt
� |dd�S |t@ |t@ }} t t t
�� t�|�}t�| �} W d � n1 s|w Y |tju s�| tju r�t
� |dd�S | j| }
|ddt�|
� � }|tju r�t tjt�� t�|
|�\} } W d � n1 s�w Y t
� |t�| �d��t�| �d���S d}
| tj u r�d}
tj }| j }|tj u r�tj }| j!}t tjt�� t�|
|�}|\} } }}W d � n 1 �sw Y 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� �
r7 r � �TCP�UDP)#r �
IndexErrorr rD �PREFIX�_HIGH�_LOW�VERSIONS�COMMANDS�
_LOCALCOMMANDr
� ProxyInfo�
ValueErrorr r �
lookupByValuer! r �ADDRESSFORMATSr9 �calcsizer �errorr r: r
� UNIXAddress�rstripr$ � IPv4AddressrM r � IPv6AddressrW �decode)�clsrX �prefix�addrInfo�versionCommand� familyProto�version�command�family�netproto�
addressFormat�source�dest�addrType�addrCls�
addrParserr? �sPort�dPortr r r r; � sb �
�
��
�
��zV2Parser.parse)r0 N)r r r r r_ rb rd �
_PROXYCOMMANDrc rh r4 �bytesr r r
re r@ � staticmethodrM rW � classmethodr; r r r r r) 8 s0
�
�
� r) )"r rT r9 �typingr r r r r �zope.interfacer �
constantlyr r �twisted.internetr
�twisted.pythonr � r
r � _exceptionsr r r r r r! r` ra rd r� � IProxyParserr) r r r r �<module> s$