CasperSecurity
o
쑛h�9 � @ s� d dl Z d dlZd dlmZ d dlmZ d dlmZ d dlm Z d dl
m Z m Z m
Z
d dlmZmZ e �e�Ze jZe jZe jZdZd d
gZd d gZd
gZd ZdZ dZ g d�Z dZ dgZ dd� Z!dd� Z"d dd�Z#G dd� d�Z$dd� Z%efd d �Z&dS ) � N)�MIMEBase)�
MIMEMultipart)�MIMENonMultipart)�MIMEText)�features�handlers�util)�UrlError�read_file_or_urlz 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_filename@ s r c C s t jrt| �|�t�| � d S r )r �ERROR_ON_USER_DATA_FAILURE� RuntimeError�LOG�warning)�
error_message�source_exceptionr r r �
_handle_errorE 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__M s zUserDataProcessor.__init__c C sB t � }t|t�r|D ]
}| �t|�|� q
|S | �t|�|� |S r )r �
isinstance�list� _process_msg�convert_string)r" �blob�accumulating_msg�br r r �processQ s
�zUserDataProcessor.processc C s� dd� }|� � D ]�}t|�rqd }|�� }t�|�}d}|tv rOz
tj|dd�}d }d}W n! tjyN } zd�|t |�| �}
t
|
| � W Y d } ~ qd } ~ ww |sSt }|t dg v r^||�}|d u rd|}|r�|�
dd �\} } t| | �}
|
�|� |�� r�t|
|�� � d
D ]}||v r�t|
|t|| �� q�|
}||kr�t|t|� |tv r�| �||� 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" �base_msg�
append_msgr. �part�ctype�
ctype_origr- �was_compressed�er �maintype�subtype�n_part�hr r r r& Z sf
��
��
� �z UserDataProcessor._process_msgc C s� |� dd �}d }|�� tv r*zt�|jdd��}|r |� d�}W n ty) Y nw |d ur0|}|d u r6|}|d urTz
|�dtt |��� W d S t
t fyS Y d S w d S )Nr2 T��decode� launch-index) �getr5 �EXAMINE_FOR_LAUNCH_INDEXr � load_yaml� get_payload� Exceptionr r �int�
ValueError� TypeError)r" r
�
header_idx� 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�@ �data�urlcache)r � hash_blob�os�path�joinr �
get_ipath_cur)r" �entry�entry_fnr r r �_get_include_once_filename� s �z,UserDataProcessor._get_include_once_filenamec C s$ |� � s t|t| � | �|� d S r )r@ r � PART_FN_TPLr_ )r" r
� 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�r5q|�� }|s<qd }d }|rG| �|�}|rUtj� |�rUt
� |�}npz-t |dd| j
d�}|rn|�� rnt
j||jd d
� |�� rv|j}n d �||j�} t| � W nB ty� }
zt|
�} || vr�| d �|�7 } t| |
� W Y d }
~
n!d }
~
w ty� } zd
�|| �} t| | � W Y d } ~ nd } ~ ww |d ur�t|�}
| �|
|� qd 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�
startswithr; �lstrip�striprk re rf �isfiler �load_text_filer
r! �ok�
write_file�contentsr: �coder r r �IOErrorr' r& )r" �contentrH �include_once_on�line�lc_line� include_url�include_once_fn�respr �urle�message�ioe�new_msgr r r rC � st
� �������� ��z UserDataProcessor._do_includec C sH t j|g ttfd�}|D ]�}t|t�rd|i}t|t�s q |�dd�}|�d�}|s9t}t|t �r3t
}t � ||�}|�
dd�\}} |dkrUt|t �rN|�� }t|| d�}
n
t|| �}
|
�|� d |v rjt|
|d � d
|v rx|
�d t|d
�� t|�� �D ] } | �� dd dd
d t�� t�� fv r�q~|
�| || � q~| �||
� q d S )
N)�default�allowedr� � �typer0 r1 �text)�_subtyper rT r2 zcontent-disposition)r rW r% �setr$ r �dictrU �ARCHIVE_UNDEF_TYPE�bytes�ARCHIVE_UNDEF_BINARY_TYPEr r, r> rS r r r? r r �keysrv �ATTACHMENT_FIELDrA rF ) r" �archiverH �entries�entr� �mtyper� rN rO r
�headerr r r rE sL
� �z"UserDataProcessor._explode_archiveNc C sj t |vrd|t < |durt|t t|�� d}z
t|�t ��}W |S ttfy4 t|t t|�� Y |S w )z
Return the number of attachments to this MIMEMultipart by looking
at its 'Number-Attachments' header.
�0Nr )r� r r rZ rU r[ r\ )r" � outer_msg� new_count�
fetched_countr r r �_multi_part_countF 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.
r1 N)r� rn �attach)r" r� rI �
part_countr r r rF X s
z UserDataProcessor._attach_partr )
�__name__�
__module__� __qualname__r# r+ r& r_ rk rn rC rE r� rF r r r r r L s PE
4 r c C s | � � pd}|�� dkrdS dS )Nr� � multipartTF)�get_content_maintyperv )rI �
part_maintyper r r r4 d s r4 c C sn | sd} dd� }t | t�r| �d�}n| }tj|dd�}d|dd � �� v r0t�|�d��}|S |||�}|S )
zJconvert a string (more likely bytes) or a message into
a mime message.� c S s( |� dd�\}}t||�}|�| � |S )Nr0 r1 )r> r r? )rb � content_typerN rO r
r r r �
create_binmsgs s
z%convert_string.<locals>.create_binmsgzutf-8FrR s
mime-version:r i )r$ r �encoder r8 rv �message_from_stringrS )�raw_datar� r� �bdatar
r r r r' m s
�r' r )'�loggingre �email.mime.baser �email.mime.multipartr �email.mime.nonmultipartr �email.mime.textr � cloudinitr r r �cloudinit.url_helperr r
� getLoggerr� r �NOT_MULTIPART_TYPErl �
OCTET_TYPErA r= rB rD r< r� r� r7 r� rV r r r r r4 r' r r r r �<module> s<