File: //usr/lib/python3/dist-packages/cloudinit/__pycache__/user_data.cpython-39.pyc
a
��`09 � @ s d dl Z d dlmZ d dlmZ d dlmZ d dlmZ d dl m
Z
d dl m Z d dl m
Z
d d lmZmZ d d
l mZ e �e�Ze
jZe
jZe
jZd Zd d
gZddgZdgZd Z dZ dZ g d�Z dZ dgZ!dd� Z"dd� Z#d!dd�Z$G dd � d e%�Z&d d � Z'efd d �Z(dS )"� N)�MIMEBase)�
MIMEMultipart)�MIMENonMultipart)�MIMEText)�handlers)�log)�features)�read_file_or_url�UrlError)�utilz Content-Typez
text/plainztext/x-not-multipartztext/x-include-urlztext/x-include-once-urlztext/cloud-config-archiveztext/cloud-configzapplication/octet-stream)zapplication/gzipzapplication/gzip-compressedzapplication/gzippedzapplication/x-compresszapplication/x-compressedzapplication/x-gunzipzapplication/x-gzipz application/x-gzip-compressedzNumber-Attachmentsc C s | |= || |< d S �N� )�msg�key�valuer
r
�5/usr/lib/python3/dist-packages/cloudinit/user_data.py�_replace_header= s r c C s | d= | j ddt|�d� d S )NzContent-Disposition�
attachment)�filename)�
add_header�str)r r r
r
r �
_set_filenameB s �r c C s t jrt| �|�n
t�| � d S r )r ZERROR_ON_USER_DATA_FAILURE� Exception�LOG�warning)�
error_messageZsource_exceptionr
r
r �
_handle_errorH s r c @ s^ e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
ddd�Z dd� Z dS )�UserDataProcessorc C s || _ t�|�| _d S r )�pathsr �fetch_ssl_details� ssl_details)�selfr r
r
r �__init__P s zUserDataProcessor.__init__c C s@ t � }t|t�r,|D ]}| �t|�|� qn| �t|�|� |S r )r �
isinstance�list� _process_msg�convert_string)r! �blobZaccumulating_msg�br
r
r �processT s
zUserDataProcessor.processc C s� dd� }|� � D �]�}t|�r qd }|�� }t�|�}d}|tv r�ztj|dd�}d }d}W nN tjy� } z4d�|t |�| �}
t
|
| � W Y d } ~ qW Y d } ~ n
d } ~ 0 0 |s�t }|t dg v r�||�}|d u r�|}|�rF|�
dd �\} } t| | �}
|
�|� |�� �rt|
|�� � d
D ]$}||v �r t|
|t|| �� �q |
}||k�r\t|t|� |tv �rt| �||� q|tv �r�| �||� qt|t|� | �||� qd S ) Nc S s
t �| �S r )r �type_from_starts_with)�payloadr
r
r �
find_ctype_ s z2UserDataProcessor._process_msg.<locals>.find_ctypeF)�quietTz<Failed decompressing payload from {} of length {} due to: {}ztext/x-shellscript�/� )� Launch-Index)�walk� is_skippable�get_content_typer �fully_decoded_payload� DECOMP_TYPES� decomp_gzip�DecompressionError�format�lenr �
UNDEF_TYPE� TYPE_NEEDED�splitr � set_payload� get_filenamer r r � CONTENT_TYPE�
INCLUDE_TYPES� _do_include�
ARCHIVE_TYPES�_explode_archive� _attach_part)r! Zbase_msg�
append_msgr, �part�ctypeZ
ctype_origr+ Zwas_compressed�er �maintype�subtypeZn_part�hr
r
r r% ] s\
��
"
z UserDataProcessor._process_msgc C s� |� dd �}d }|�� tv rTz$t�|jdd��}|r>|� d�}W n tyR Y n0 |d ur`|}|d u rl|}|d ur�z|�dtt |��� W n t
t fy� Y n0 d S )Nr0 T��decode� launch-index) �getr3 �EXAMINE_FOR_LAUNCH_INDEXr � load_yaml� get_payloadr r r �int�
ValueError� TypeError)r! r Z
header_idxZ payload_idxr+ r
r
r �_attach_launch_index� s$ z&UserDataProcessor._attach_launch_indexc C s& t �|dd�}tj�| j�d�d|�S )N�md5�@ �dataZurlcache)r � hash_blob�os�path�joinr �
get_ipath_cur)r! �entryZentry_fnr
r
r �_get_include_once_filename� s �z,UserDataProcessor._get_include_once_filenamec C s$ |� � st|t| � | �|� d S r )r> r � PART_FN_TPLrV )r! r Z attached_idr
r
r �_process_before_attach� s z(UserDataProcessor._process_before_attachc C s� d}|� � D �]�}|�� }|�d�r>|td�d � �� }d}n"|�d�r`|td�d � �� }d}|�d�rlq |�� }|szq d }d }|r�| �|�}|r�tj� |�r�t
� |�}n�zZt |dd| j
d�}|r�|�� r�t
j||jd d
� |�� r�|j}nd �||j�} t| � W n� t�yV }
z6t|
�} || v�r8| d �|�7 } t| |
� W Y d }
~
nDd }
~
0 t�y� } z"d
�|| �} t| | � W Y d } ~ n
d } ~ 0 0 |d ur t|�}
| �|
|� q d S )NFz
#include-onceTz#include�#� �
)�timeout�retriesr i� )�modez6Fetching from {} resulted in a invalid http code of {}z
for url: {0}z Fetching from {} resulted in {})�
splitlines�lower�
startswithr9 �lstrip�stripr` r[ r\ �isfiler � load_filer r �ok�
write_file�contentsr8 �coder r
r �IOErrorr& r% )r! �contentrE Zinclude_once_on�lineZlc_lineZ include_urlZinclude_once_fn�respr Zurle�messageZioeZnew_msgr
r
r rA � s`
�
���
� z UserDataProcessor._do_includec C sL t j|g ttfd�}|D �],}t|t�r0d|i}t|t�s<q|�dd�}|�d�}|stt}t|t �rht
}t � ||�}|�
dd�\}} |dkr�t|t �r�|�� }t|| d�}
nt|| �}
|
�|� d |v r�t|
|d � d
|v r�|
�d t|d
�� t|�� �D ]:} | �� dd dd
d t�� t�� fv �r(q�|
�| || � q�| �||
� qd S )
N)�default�allowedru � �typer. r/ �text)�_subtyper rN r0 zcontent-disposition)r rQ r$ �setr# r �dictrO �ARCHIVE_UNDEF_TYPE�bytes�ARCHIVE_UNDEF_BINARY_TYPEr r* r< rM r r r= r r �keysrj �ATTACHMENT_FIELDr? rD ) r! �archiverE �entries�entru �mtypery rI rJ r �headerr
r
r rC sB
�z"UserDataProcessor._explode_archiveNc C sj t |vrd|t < |dur(t|t t|�� d}zt|�t ��}W n& ttfyd t|t t|�� Y n0 |S )z
Return the number of attachments to this MIMEMultipart by looking
at its 'Number-Attachments' header.
�0Nr )r� r r rS rO rT rU )r! � outer_msgZ new_countZ
fetched_countr
r
r �_multi_part_count; s z#UserDataProcessor._multi_part_countc C s8 | � |�}| �||d � |�|� | � ||d � dS )z�
Attach a message to an outer message. outermsg must be a MIMEMultipart.
Modifies a header in the outer message to keep track of number of
attachments.
r/ N)r� rb �attach)r! r� rF �
part_countr
r
r rD M s
z UserDataProcessor._attach_part)N)
�__name__�
__module__� __qualname__r" r) r% rV r` rb rA rC r� rD r
r
r
r r O s P=/
r c C s | � � p
d}|�� dkr dS dS )Nr{ � multipartTF)�get_content_maintyperj )rF Z
part_maintyper
r
r r2 Y s r2 c C sl | sd} dd� }t | t�r&| �d�}n| }tj|dd�}d|dd � �� v r^t�|�d��}n
|||�}|S )
zJconvert a string (more likely bytes) or a message into
a mime message.� c S s( |� dd�\}}t||�}|�| � |S )Nr. r/ )r< r r= )rY � content_typerI rJ r r
r
r �
create_binmsgh s
z%convert_string.<locals>.create_binmsgzutf-8FrL s
mime-version:r i )r# r �encoder r6 rj �message_from_stringrM )�raw_datar� r� Zbdatar r
r
r r& b s
r& )N))r[ Zemail.mime.baser Zemail.mime.multipartr Zemail.mime.nonmultipartr Zemail.mime.textr � cloudinitr r �loggingr Zcloudinit.url_helperr r
r � getLoggerr� r ZNOT_MULTIPART_TYPEra Z
OCTET_TYPEr? r; r@ rB r: r� r� r5 r� rP r r r �objectr r2 r& r
r
r
r �<module> s>