File: //lib/python3/dist-packages/cloudinit/__pycache__/user_data.cpython-310.pyc
o
�Ad�9 � @ 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
�e�Ze jZe jZe jZd
Zd d gZd
dgZdgZd Z dZ dZ g d�Z dZ dgZ!dd� Z"dd� Z#d dd�Z$G dd� d�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-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_filenameA s r c C s t jrt| �|�t�| � d S r )r �ERROR_ON_USER_DATA_FAILURE� Exception�LOG�warning)�
error_message�source_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 || _ t�|�| _d S r )�pathsr �fetch_ssl_details� ssl_details)�selfr r
r
r �__init__N 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 �processR 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' [ 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 )Nr3 T��decode� launch-index) �getr6 �EXAMINE_FOR_LAUNCH_INDEXr � load_yaml� get_payloadr r 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 )rA 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_filer r" �ok�
write_file�contentsr; �coder r
r �IOErrorr( r' )r# �contentrI �include_once_on�line�lc_line� include_url�include_once_fn�respr �urle�message�ioe�new_msgr
r
r rD � 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� � �typer1 r2 �text)�_subtyper rU r3 zcontent-disposition)r rX r& �setr% r �dictrV �ARCHIVE_UNDEF_TYPE�bytes�ARCHIVE_UNDEF_BINARY_TYPEr r- r? rT r r r@ r r �keysrv �ATTACHMENT_FIELDrB rG ) r# �archiverI �entries�entr� �mtyper� rO rP r �headerr
r
r rF 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 rV r[ r\ )r# � outer_msg� new_count�
fetched_countr
r
r �_multi_part_countG 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.
r2 N)r� rn �attach)r# r� rJ �
part_countr
r
r rG Y s
z UserDataProcessor._attach_partr )
�__name__�
__module__� __qualname__r$ r, r' r_ rk rn rD rF r� rG r
r
r
r r M s PE
4 r c C s | � � pd}|�� dkrdS dS )Nr� � multipartTF)�get_content_maintyperv )rJ �
part_maintyper
r
r r5 e s r5 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 )Nr1 r2 )r? r r@ )rb � content_typerO rP r r
r
r �
create_binmsgt s
z%convert_string.<locals>.create_binmsgzutf-8FrS s
mime-version:r i )r% r �encoder r9 rv �message_from_stringrT )�raw_datar� r� �bdatar r
r
r r( n s
�r( r )(re �email.mime.baser �email.mime.multipartr �email.mime.nonmultipartr �email.mime.textr � cloudinitr r r �loggingr �cloudinit.url_helperr
r � getLoggerr� r �NOT_MULTIPART_TYPErl �
OCTET_TYPErB r> rC rE r= r� r� r8 r� rW r r r r r5 r( r
r
r
r �<module> s>