CasperSecurity
o
`�wi;) � @ s> d dl Z d dlZd dlZd dlZd dlmZ G dd� d�Ze� Zd%dd�Ze� de� d d
� Z
d � � D ]Z e� e e
� q3d d
� Z
d� � D ]
Ze� de e
� qDdd� Ze� de� dd� Zdd� Zdd� Zdd� Z d&d d �Ze�ee� d'd d!�Ze�ejje� " d(d#d$�Zee e fD ]Z e�e e� q�dS ))� N)�
quoprimimec @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
S )�ContentManagerc C s i | _ i | _d S �N)� get_handlers� set_handlers)�self� r �+/usr/lib/python3.10/email/contentmanager.py�__init__ s
zContentManager.__init__c C � || j |< d S r )r )r �key�handlerr r r �add_get_handler
� z ContentManager.add_get_handlerc O s� |� � }|| jv r| j| |g|�R i |��S |�� }|| jv r.| j| |g|�R i |��S d| jv rA| jd |g|�R i |��S t|��)N� )�get_content_typer �get_content_maintype�KeyError)r �msg�args�kw� content_type�maintyper r r � get_content s
zContentManager.get_contentc C r r )r )r �typekeyr
r r r �add_set_handler r z ContentManager.add_set_handlerc O sD |� � dkr
td��| �||�}|�� |||g|�R i |�� d S )N� multipartz"set_content not valid on multipart)r � TypeError�_find_set_handler�
clear_content)r r �objr r r
r r r � set_content s
zContentManager.set_contentc C s� d }t |�jD ]O}|| jv r| j| S |j}t|dd�}|r'd�||f�n|}|d u r/|}|| jv r;| j| S || jv rG| j| S |j}|| jv rV| j| S qd | jv ra| jd S t|��)N�
__module__r �.)�type�__mro__r � __qualname__�getattr�join�__name__r ) r r r �full_path_for_error�typ�qname�modname� full_path�namer r r r ' s(
�
z ContentManager._find_set_handlerN) r) r" r& r
r r r r! r r r r r r s r �replacec C s& | j dd�}| �dd�}|j||d�S )NT��decode�charset�ASCII)�errors)� get_payload� get_paramr2 )r r5 �contentr3 r r r �get_text_content@ s r9 �textc C s | j dd�S )NTr1 �r6 �r r r r �get_non_text_contentG s r= z audio image video applicationc C s
| � d�S �Nr r; r< r r r �get_message_contentM s
r? zrfc822 external-bodyzmessage/c C s t | �d��S r> )�bytesr6 r< r r r �%get_and_fixup_unknown_message_contentS s rA �messagec
s� d� ||f�| d< |rOt|d d�s | j� � fdd�|D �}z|D ]}|jr+|jd �|| |j< q!W d S tjjyN } z
td� |j
| jd���|�d }~ww d S ) N�/z Content-Typer r/ c s g | ] }� j � �|g�� �qS r )�header_factory�header_source_parse)�.0�header��mpr r �
<listcomp>d s �z _prepare_set.<locals>.<listcomp>zInvalid header: {})�policy) r( �hasattrrK �defectsr/ �emailr5 � HeaderDefect�
ValueError�format�fold)r r �subtype�headersrG �excr rH r � _prepare_set_ s,
�
� �����rV c C s| |d u r
|d ur
d}|d ur|| d< |d ur | j d|ddd� |d ur'|| d< |d ur:|�� D ] \}}| � ||� q/d S d S )N�
attachmentzContent-Disposition�filenameT)rG r0 z
Content-ID)� set_param�items)r � dispositionrX �cid�paramsr �valuer r r �
_finalize_setp s" ��r_ c C sV g }|d d }t dt| �|�D ]}| ||| � }|�t�|��d�� qd�|�S )N� � r �asciir )�range�len�append�binascii�
b2a_base64r2 r( )�data�max_line_length�
encoded_lines�unencoded_bytes_per_line�i�thisliner r r �_encode_base64� s
rn c s� | � |��� }|j� d�� � fdd�}dd� }|d u rztdd� |D �dd �|jkrLz
d
||��d�fW S ty< Y nw |jd krLd ||��dd �fS ||d d
� �}t� |�d�|j�}t
� |�} t |�t | �krnd}n d}t |�d
krz||fS |d
kr�||��d�}
||
fS |d kr�||��dd �}
||
fS |dkr�t� ||��d�|j�}
||
fS |dkr�t
||�|j�}
||
fS td�|���)Nrb c s � � | �� S r �r( ��lines��linesepr r �
embedded_body� � z#_encode_text.<locals>.embedded_bodyc S s d� | �d S )N�
ro rp r r r � normal_body� ru z!_encode_text.<locals>.normal_bodyc s s � | ]}t |�V qd S r )rd )rF �xr r r � <genexpr>� s � z _encode_text.<locals>.<genexpr>r )�default�7bit�8bit�surrogateescape�
zlatin-1�base64�quoted-printablez$Unknown content transfer encoding {})�encode�
splitlinesrs �maxri r2 �UnicodeDecodeError�cte_typer � body_encoderf rg rd rn rP rQ ) �stringr3 �cterK rq rt rw �sniff�sniff_qp� sniff_base64rh r rr r � _encode_text� sL �
�
�����r� �plain�utf-8c
C sd t | d|| � t|||| j�\}}
| �|
� | jdtjj�||�dd� || d< t | ||||� d S )Nr: r3 T�r0 �Content-Transfer-Encoding)
rV r� rK � set_payloadrY rN r3 �ALIASES�getr_ ) r r� rS r3 r� r[ rX r\ r] rT �payloadr r r �set_text_content� s
�r� �rfc822c C s� |dkrt d��|dkr |dvrt d�|���|d u r dn|}n|dkr2|dvr/t d �|���d
}n|d u r8d
}t| d ||� | �|g� || d < t| ||||� d S )
N�partialz4message/partial is not supported for Message objectsr� )Nr{ r| �binaryz*message/rfc822 parts do not support cte={}r| z
external-body)Nr{ z1message/external-body parts do not support cte={}r{ rB r� )rP rQ rV r� r_ ) r rB rS r� r[ rX r\ r] rT r r r �set_message_content� s( �� r� r c
C s� t | ||| � |dkrt|| jjd�}n'|dkr'tj|dddd�}|�d�}n|dkr1|�d�}n
|d v r;|�dd
�}| �|� || d < t| ||||� d S ) Nr )ri r� FT)�istextrG � quotetabsrb r{ )r| r� r} r� ) rV rn rK ri rf �b2a_qpr2 r� r_ )
r rh r rS r� r[ rX r\ r] rT r r r �set_bytes_content� s
r� r� )r� r� NNNNNN)r� NNNNNN)r NNNNN) rf �
email.charsetrN �
email.message� email.errorsr r �raw_data_managerr9 r r= �splitr r? rS rA rV r_ rn r� r� r �strr� rB �Messager� r@ � bytearray�
memoryviewr+ r r r r �<module> sP 6
� '
�
�
��