0x1949 Team - FAZEMRX - MANAGER
Edit File: user_data.cpython-310.pyc
o ���f�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<