CasperSecurity
o
`�wi�# � @ s� d dl Z d dlZd dlZd dlmZ e�d�ZdZdZdd� Z dd � Z
d
d � Z G d d
� d
ej �Z G dd� dej
�ZG dd� dej�ZG dd� de ej�ZG dd� de ej�Zdd� ZdS )� N)� ucd_3_2_0u [.。．｡]s xn--zxn--c C s� g }| D ]}t �|�r q|�t �|�� qd�|�} t�d| �} | D ]5}t �|�sQt �|�sQt � |�sQt �
|�sQt � |�sQt � |�sQt �
|�sQt �|�sQt �|�rWtd| ��q"dd� | D �}t|�r|tdd� | D ��rptd��|d rx|d
s|td ��| S ) N� �NFKCzInvalid character %rc S s g | ]}t �|��qS � )�
stringprep� in_table_d1��.0�xr r �%/usr/lib/python3.10/encodings/idna.py�
<listcomp>) s z nameprep.<locals>.<listcomp>c s s � | ]}t �|�V qd S )N)r � in_table_d2r r r r � <genexpr>1 s � znameprep.<locals>.<genexpr>z Violation of BIDI requirement 2r ���z Violation of BIDI requirement 3)r � in_table_b1�append� map_table_b2�join� unicodedata� normalize� in_table_c12� in_table_c22� in_table_c3� in_table_c4� in_table_c5� in_table_c6� in_table_c7� in_table_c8� in_table_c9� UnicodeError�any)�label�newlabel�c�RandALr r r �nameprep sD
�������� � r% c C s� z| � d�} W n ty Y nw dt| � k r dk r#| S td��td��t| �} z| � d�} W n ty; Y nw dt| � k rIdk rN| S td��td��| �t�r[td��| � d�} t| } dt| � k rqdk rv| S td��td��)N�asciir �@ �label empty or too longz Label starts with ACE prefix�punycode)�encoder �lenr% �
startswith� sace_prefix�
ace_prefix)r! r r r �ToASCII< s6 �� ��
�r/ c C s� t | t�rd}nz | �d�} d}W n ty d}Y nw |s6t| �} z| �d�} W n ty5 td��w | �t�s@t| d�S | tt�d � }|� d�}t
|�}t| d�� � t|d�krctd| |��|S )NTr& Fz Invalid character in IDN labelr) zIDNA does not round-trip) �
isinstance�bytesr* r r% r, r. �strr+ �decoder/ �lower)r! �
pure_ascii�label1�result�label2r r r � ToUnicodeg s.
� �
r9 c @ s e Zd Zddd�Zddd�ZdS )�Codec�strictc C s |dkr
t d| ��|sdS z|�d�}W n ty Y n5w |�d�}|d d� D ]}dt|� k r;dk s@t d �� t d ��q*t|d �dkrMt d
��|t|�fS t� }t�|�}|rg|d sgd}|d= nd }|D ]}|rt|�d� |�t|�� qkt || �t|�fS ) Nr; �unsupported error handling �� r r& � .r r r' r( zlabel too longr> )
r r* �UnicodeEncodeError�splitr+ � bytearray�dots�extendr/ r1 )�self�input�errorsr7 �labelsr! � trailing_dotr r r r* � s< �
��
z Codec.encodec C s� |dkr
t d| ��|sdS t|t�st|�}t|vr/z
|�d�t|�fW S ty. Y nw |�d�}|rDt|d �dkrDd}|d= nd }g }|D ] }|�t |�� qJd�
|�| t|�fS )
Nr; �Unsupported error handling �r r r& r? r r �.r ) r r0 r1 r. r3 r+ �UnicodeDecodeErrorrA r r9 r )rE rF rG rH rI r7 r! r r r r3 � s*
�
z Codec.decodeN)r; )�__name__�
__module__� __qualname__r* r3 r r r r r: � s
%r: c @ � e Zd Zdd� ZdS )�IncrementalEncoderc C s� |dkr
t d| ��|sdS t�|�}d}|r*|d s!d}|d= n |s*|d= |r*d}t� }d}|D ]}|r>|�d� |d7 }|�t|�� |t|�7 }q1||7 }|t|�7 }t|�|fS ) Nr; r< r= r> r r? r � )r rC rA rB rD r/ r+ r1 � rE rF rG �finalrH rI r7 �sizer! r r r �_buffer_encode� s2
z!IncrementalEncoder._buffer_encodeN)rN rO rP rW r r r r rR � � rR c @ rQ )�IncrementalDecoderc C s� |dkr
t d| ��|sdS t|t�rt�|�}n
t|d�}|�d�}d}|r:|d s1d}|d= n |s:|d= |r:d}g }d}|D ]}|�t|�� |rO|d 7 }|t|�7 }q@d�|�| }|t|�7 }||fS )
Nr; rJ rK r& rL r r r rS ) r r0 r2 rC rA r r9 r+ r rT r r r �_buffer_decode� s6
z!IncrementalDecoder._buffer_decodeN)rN rO rP rZ r r r r rY � rX rY c @ � e Zd ZdS )� StreamWriterN�rN rO rP r r r r r\ " � r\ c @ r[ )� StreamReaderNr] r r r r r_ % r^ r_ c C s t jdt� jt� jttttd�S )N�idna)�namer* r3 �incrementalencoder�incrementaldecoder� streamwriter� streamreader) �codecs� CodecInfor: r* r3 rR rY r\ r_ r r r r � getregentry* s �rh )r �rerf r r �compilerC r. r- r% r/ r9 r: �BufferedIncrementalEncoderrR �BufferedIncrementalDecoderrY r\ r_ rh r r r r �<module> s
.+)H#'