Your IP : 216.73.217.13
U
�Ad�" �
@ s� d dl Z d dlZd dlmZ d dlmZmZ e�e�Z e
ej g�Z G dd� d�Z
G dd� d�Zdd � Zdd
d�Zddd�Zddd�Zddd�ZdS )� N)�log)�
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
�?/usr/lib/python3/dist-packages/cloudinit/sources/helpers/ec2.py�_maybe_json_object s z&MetadataLeafDecoder._maybe_json_objectc
C s� |sdS zt �|�}W n tk
r. | Y S X | �|�r|z t�|�W S ttfk
rz } zt� d||� W 5 d }~X Y nX |�
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 kr*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||kr�|�|� q6|�dd�}|} t|�dkr�|\}
} t �|
�}
|
d k r�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 )Nr0 )r
�rstripr1 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�childrenr3 r5 r Z
field_name�contents�resourceZidentZ sub_contentsr
r
r �_parseA s6
zMetadataMaterializer._parsec C s( | j d k r| 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 ]&}
|
|kr�t
� d|� q�| |
||
< q�|S )Nr0 z&Duplicate key found in results from %s) r? r � combine_urlr
r* r@ �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 r@ r s(
z!MetadataMaterializer._materialize)N)r! r" r# r/ r? rA r@ r
r
r
r r% 6 s
+r% c C s
|j | kS )z/Returns False if cause.code is in status_codes.)�code)Z status_codesZ
_request_args�causer
r
r �skip_retry_on_codes� s rI �latest�http://169.254.169.254� c
C s� t �|| �}t �|d�}d} z2|s.t�tt�}t j|||||||d�}
|
j} W n\ t jk
r� } z | j tkrzt
� t d|� W 5 d } ~ X Y n" t
k
r� t
� t d|� Y nX | S )Nz user-datar )� ssl_details�timeout�retries� exception_cb�
headers_cb�headers_redactz$Failed fetching userdata from url %s)r rB � functools�partialrI �SKIP_USERDATA_CODES�read_file_or_urlr= �UrlErrorrG r �logexcr � Exception) � api_version�metadata_addressrM rN rO rQ rR rP Zud_urlZ user_data�responser r
r
r �get_instance_userdata� s2
��
r] c s� t �||| �} tjt j|||||| d�� � �fdd�} z6� | �}
t|
j| | |d�}|�� }t|t �shi }|W S t
k
r� t � t
d| | � i Y S X d S )N)rM rN rO rQ rR rP c
s\ z � | �j W S tjk
rV } z*�r,�|�s0� nt�d| � W Y �dS W 5 d }~X Y nX d S )Nz&Skipped retrieval of the content of %sz (skipped))r= r rW r r )Zurlr �r- � retrieval_exception_ignore_cbr
r �mcaller� s �� z'_get_instance_metadata.<locals>.mcaller)r. z Failed fetching %s from url %s)r rB rS rT rV r% r= rA �
isinstance�dictrY r rX r )�treerZ r[ rM rN rO r. rQ rR rP r_ Zmd_urlr` r\ Z materializerZmdr
r^ r �_get_instance_metadata� s4
�
�
rd c
C s t d| ||||||||| d� S )Nz
meta-data/) rc rZ r[ rM rN rO r. rR rQ rP r_ �rd )
rZ r[ rM rN rO r. rQ rR rP r_ r
r
r �get_instance_metadata� s �rf c C s t d| ||||||||d�
S )Nzdynamic/instance-identity)
rc rZ r[ rM rN rO r. rR rQ rP re ) rZ r[ rM rN rO r. rQ rR rP r
r
r �get_instance_identity s �rg )rJ rK NrL rL NNN)
rJ rK NrL rL NNNNN)
rJ rK NrL rL NNNNN) rJ rK NrL rL NNNN)rS r Z cloudinitr Zloggingr r Z getLoggerr! r � frozensetZ NOT_FOUNDrU r r% rI r] rd rf rg r
r
r
r �<module> sb
#T �
) �
4 �
�