ruạṛ
a '�DgU" � @ s� d dl Z d dlZd dlZd dlmZmZ e�e�Ze ej g�Z G dd� d�Z G dd� d�Z dd� Zdd d �Zddd�Zddd�Zddd�ZdS )� N)� url_helper�utilc @ s e Zd ZdZdd� Zdd� ZdS )�MetadataLeafDecoderz.Decodes a leaf blob into something meaningful.c C s, |sdS |� � }|�d�r(|�d�r(dS dS )NF�{�}T)�strip� startswith�endswith)�self�text� r �A/usr/lib/python3.9/site-packages/cloudinit/sources/helpers/ec2.py�_maybe_json_object s z&MetadataLeafDecoder._maybe_json_objectc C s� |sdS zt �|�}W n ty, | Y S 0 | �|�r|z t�|�W S ttfyz } zt� d||� W Y d }~n d }~0 0 |� d�dkr�|� � S |S )N� z6Field %s looked like a json object, but it was not: %s� ���) r � decode_binary�UnicodeDecodeErrorr �json�loads� ValueError� TypeError�LOG�warning�find� splitlines)r �field�blob�er r r �__call__ s$ �z MetadataLeafDecoder.__call__N)�__name__� __module__� __qualname__�__doc__r r r r r r r s r c @ s. e Zd Zd dd�Zdd� Zdd� Zdd � ZdS ) �MetadataMaterializerNc C s4 || _ d | _|| _|| _|d u r*t� | _n|| _d S �N)�_blob�_md� _base_url�_callerr � _leaf_decoder)r r �base_url�caller� leaf_decoderr r r �__init__7 s z MetadataMaterializer.__init__c C s� i }g }t �|�}|s ||fS dd� }dd� }|�� D ]�}|�� }||�}|r6|sTq6|dkr^q6||�rz||vr�|�|� q6|�dd�}|} t|�dkr�|\} } t �| �} | d ur�d| } | }| ||< q6||fS ) Nc S s | � d�rdS dS d S )N�/TF)r ��itemr r r � has_childrenI s z1MetadataMaterializer._parse.<locals>.has_childrenc S s | � d�r| �d�S | S )Nr/ )r �rstripr0 r r r �get_nameO s z-MetadataMaterializer._parse.<locals>.get_namezsecurity-credentials�=� z%s/openssh-key)r r r r �append�split�lenZsafe_int) r r �leaves�childrenr2 r4 r � field_name�contents�resource�identZ sub_contentsr r r �_parseA s6 zMetadataMaterializer._parsec C s( | j d ur| j S | �| j| j�| _ | j S r% )r' � _materializer& r( )r r r r � materializel s z MetadataMaterializer.materializec C s� | � |�\}}i }|D ]<}t�||�}|�d�s8|d7 }| �|�}| �||�||< qi } |�� D ].\} } t�|| �} | �| �} | �| | �| | < q`i }|�|� | � � D ]&} | |v r�t � d|� q�| | || < q�|S )Nr/ z&Duplicate key found in results from %s) r@ r � combine_urlr r) rA �itemsr* �update�keysr r )r r r+ r: r; Zchild_contents�cZ child_urlZ child_blobZ leaf_contentsr r> Zleaf_urlZ leaf_blobZjoinedr r r rA r s( z!MetadataMaterializer._materialize)N)r r! r" r. r@ rB rA r r r r r$ 6 s +r$ c C s |j | vS )z/Returns False if cause.code is in status_codes.)�code)Z status_codesZ _request_args�causer r r �skip_retry_on_codes� s rJ �latest�http://169.254.169.254� c C s� t �|| �}t �|d�}d} z2|s.t�tt�}t j|||||||d�} | j} W n\ t jy� } z$| j tvrxt � t d|� W Y d } ~ n(d } ~ 0 t y� t � t d|� Y n0 | S )Nz user-data� )� ssl_details�timeout�retries� exception_cb� headers_cb�headers_redactz$Failed fetching userdata from url %s)r rC � functools�partialrJ �SKIP_USERDATA_CODES�read_file_or_urlr= �UrlErrorrH r �logexcr � Exception) � api_version�metadata_addressrO rP rQ rS rT rR Zud_urlZ user_data�responser r r r �get_instance_userdata� s0 �� " r_ c s� t �||| �} tjt j|||||| d�� � �fdd�} z6� | �} t| j| | |d�}|�� }t|t �shi }|W S t y� t � t d| | � i Y S 0 d S )N)rO rP rQ rS rT rR c sb z � | �j W S tjy\ } z6�r*�|�s.� nt�d| � W Y d }~dS W Y d }~n d }~0 0 d S )Nz&Skipped retrieval of the content of %sz (skipped))r= r rY r r )�urlr �r, � retrieval_exception_ignore_cbr r �mcaller� s �� z'_get_instance_metadata.<locals>.mcaller)r- z Failed fetching %s from url %s)r rC rU rV rX r$ r= rB � isinstance�dictr[ r rZ r )�treer\ r] rO rP rQ r- rS rT rR rb Zmd_urlrc r^ Z materializerZmdr ra r �_get_instance_metadata� s. � � rg c C s t d| ||||||||| d� S )Nz meta-data/) rf r\ r] rO rP rQ r- rT rS rR rb �rg ) r\ r] rO rP rQ r- rS rT rR rb r r r �get_instance_metadata� s �ri c C s t d| ||||||||d� S )Nzdynamic/instance-identity) rf r\ r] rO rP rQ r- rT rS rR rh ) r\ r] rO rP rQ r- rS rT rR r r r �get_instance_identity s �rj )rK rL NrM rM NNN) rK rL NrM rM NNNNN) rK rL NrM rM NNNNN) rK rL NrM rM NNNN)rU r ZloggingZ cloudinitr r Z getLoggerr r � frozensetZ NOT_FOUNDrW r r$ rJ r_ rg ri rj r r r r �<module> sb #T � ) � 4 � �
cải xoăn