File: //lib/python3.6/site-packages/cloudinit/__pycache__/user_data.cpython-36.pyc
3
�� b: � @ s d dl Z d dlmZ d dlmZ d dlmZ d dlmZ d dl m
Z
m Z d dl m Z
d dl mZ d d lmZmZ e
je�Ze jZe jZe jZd
Zd d gZd
dgZdgZd Z dZ dZ ddddddddgZ dZ dgZ!dd � Z"d d � Z#d'd d �Z$G d!d"� d"e%�Z&d#d$� Z'efd%d&�Z(dS )(� N)�MIMEBase)�
MIMEMultipart)�MIMENonMultipart)�MIMEText)�features�handlers)�log)�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-streamzapplication/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 � /usr/lib/python3.6/user_data.py�_replace_header< s r c C s | d= | j ddt|�d� d S )NzContent-DispositionZ
attachment)�filename)�
add_header�str)r
r r r r �
_set_filenameA s r c C s t jrt| �|�n
tj| � d S )N)r ZERROR_ON_USER_DATA_FAILURE� Exception�LOGZwarning)�
error_messageZsource_exceptionr r r �
_handle_errorF 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 || _ tj|�| _d S )N)�pathsr Zfetch_ssl_details� ssl_details)�selfr r r r �__init__N s zUserDataProcessor.__init__c C sD t � }t|t�r0x.|D ]}| jt|�|� qW n| jt|�|� |S )N)r �
isinstance�list� _process_msg�convert_string)r ZblobZaccumulating_msg�br r r �processR s
zUserDataProcessor.processc C s� dd� }�x�|j � D �]�}t|�r$qd }|j� }tj|�}d}|tkr�ytj|dd�}d }d}W nB tjk
r� } z$dj|t |�| �}
t
|
| � wW Y d d } ~ X nX |s�t }|t dg kr�||�}|d kr�|}|�rB|j
dd �\} } t| | �}
|
j|� |j� �rt|
|j� � x,d D ]$}||k�rt|
|t|| �� �qW |
}||k�rXt|t|� |tk�rp| j||� q|tk�r�| j||� qt|t|� | j||� qW d S ) Nc S s
t j| �S )N)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)r+ )�walk� is_skippable�get_content_typer Zfully_decoded_payload� DECOMP_TYPES� decomp_gzipZDecompressionError�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' �partZctypeZ
ctype_origr& Zwas_compressed�er �maintype�subtypeZn_part�hr r r r! [ sX
z UserDataProcessor._process_msgc C s� |j dd �}d }|j� tkrVy$tj|jdd��}|r>|j d�}W n tk
rT Y nX |d k rb|}|d krn|}|d k r�y|jdtt |��� W n t
t fk
r� Y nX d S )Nz Launch-IndexT)�decodez launch-index) �getr. �EXAMINE_FOR_LAUNCH_INDEXr � load_yamlZ 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 j|dd�}tjj| jjd�d|�S )NZmd5�@ �dataZurlcache)r Z hash_blob�os�path�joinr Z
get_ipath_cur)r �entryZentry_fnr r r �_get_include_once_filename� s z,UserDataProcessor._get_include_once_filenamec C s$ |j � st|t| � | j|� d S )N)r7 r � PART_FN_TPLrK )r r
Z attached_idr r r �_process_before_attach� s z(UserDataProcessor._process_before_attachc C s� d}�x�|j � D �]�}|j� }|jd�rB|td�d � j� }d}n"|jd�rd|td�d � j� }d}|jd�rpq|j� }|s~qd }d }|r�| j|�}|r�tjj |�r�t
j |�}n�yZt |dd| j
d�}|r�|j� r�t
j||jd d
� |j� r�|j}nd j||j�} t| � W n� tk
�rX }
z0t|
�} || k�r>| d j|�7 } t| |
� W Y d d }
~
X n: tk
�r� } z d
j|| �} t| | � W Y d d } ~ X nX |d k rt|�}
| j|
|� qW d S )NFz
#include-onceTz#include�#� �
)ZtimeoutZretriesr i� )�modez6Fetching from {} resulted in a invalid http code of {}z
for url: {0}z Fetching from {} resulted in {})�
splitlines�lower�
startswithr2 �lstrip�striprR rN rO �isfiler Z load_filer r �okZ
write_file�contentsr1 �coder r
r �IOErrorr" r! )r �contentr> Zinclude_once_on�lineZlc_lineZ include_urlZinclude_once_fnZrespr Zurle�messageZioeZnew_msgr r r r: � s\
z UserDataProcessor._do_includec
C sZ t j|g ttfd�}�x>|D �]4}t|t�r4d|i}t|t�s@q |jdd�}|jd�}|sxt}t|t �rlt
}t j ||�}|j
dd�\}} |dkr�t|t �r�|j� }t|| d�}
nt|| �}
|
j|� d |kr�t|
|d � d
|kr�|
jd t|d
�� xNt|j� �D ]>} | j� dd dd
d tj� tj� fk�r0�q|
j| || � �qW | j||
� q W d S )
N)�defaultZallowedrc � �typer) r* �text)Z_subtyper z launch-indexz Launch-Indexzcontent-disposition)r rG r �setr r �dictrE �ARCHIVE_UNDEF_TYPE�bytes�ARCHIVE_UNDEF_BINARY_TYPEr r% r5 rD r r r6 r r �keysrZ �ATTACHMENT_FIELDr8 r= ) r �archiver> ZentriesZentrc Zmtyperf rA rB r
�headerr r r r< sH
z"UserDataProcessor._explode_archiveNc C sl t |krd|t < |dk r(t|t t|�� d}yt|jt ��}W n( ttfk
rf t|t t|�� Y nX |S )z
Return the number of attachments to this MIMEMultipart by looking
at its 'Number-Attachments' header.
�0Nr )rp r r rH rE rI rJ )r � outer_msgZ new_countZ
fetched_countr r r �_multi_part_countH s z#UserDataProcessor._multi_part_countc C s8 | j |�}| j||d � |j|� | j ||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)ru rT Zattach)r rt r? Z
part_countr r r r= Z s
z UserDataProcessor._attach_part)N)
�__name__�
__module__� __qualname__r r$ r! rK rR rT r: r< ru r= r r r r r M s QE4
r c C s | j � p
d}|j� dkr dS dS )Nrg Z multipartTF)Zget_content_maintyperZ )r? Z
part_maintyper r r r- f s r- c C sl | sd} dd� }t | t�r&| jd�}n| }tj|dd�}d|dd � j� kr^tj|jd��}n
|||�}|S )
zJconvert a string (more likely bytes) or a message into
a mime message.� c S s( |j dd�\}}t||�}|j| � |S )Nr) r* )r5 r r6 )rM � content_typerA rB r
r r r �
create_binmsgu s
z%convert_string.<locals>.create_binmsgzutf-8F)rD s
mime-version:r i )r r �encoder r0 rZ Zmessage_from_stringrD )Zraw_datarz r{ Zbdatar
r r r r" o s
r" )N))rN Zemail.mime.baser Zemail.mime.multipartr Zemail.mime.nonmultipartr Zemail.mime.textr Z cloudinitr r r Zloggingr Zcloudinit.url_helperr
r Z getLoggerrv r ZNOT_MULTIPART_TYPErS Z
OCTET_TYPEr8 r4 r9 r; r3 rl rn r/ rp rF r r r �objectr r- r" r r r r �<module> sJ