File: //lib/python3/dist-packages/cloudinit/sources/azure/__pycache__/imds.cpython-38.pyc
U
Ӈg$ � @ s� d dl Z d dlZd dlmZ d dlmZmZmZmZ d dl Z d dl
m Z d dl m
Z
d dlmZmZ e �e�ZdZdd � ZG d
d � d �Zd d
d�eeeeed�dd�Zeeed�dd�Z de ee ed�dd�Z ed�dd�Z dS ) � N)� monotonic)�Dict�Optional�Type�Union)�util)�report_diagnostic_event)�UrlError�readurlz http://169.254.169.254/metadatac C s dt t�� �d�S )N�true)ZMetadatazx-ms-client-request-id)�str�uuidZuuid4)Z_url� r �>/usr/lib/python3/dist-packages/cloudinit/sources/azure/imds.py�
headers_cb s
�r c @ sF e Zd ZdZddddd�eee ee dd�dd�Zed �d
d �Z dS ) �ReadUrlRetryHandleraK Manager for readurl retry behavior using exception_callback().
:param logging_backoff: Backoff to limit logging.
:param max_connection_errors: Number of connection errors to retry on.
:param retry_codes: Set of http codes to retry on.
:param retry_deadline: Optional monotonic()-based deadline to retry until.
� �?N)� � � i� ��logging_backoff�max_connection_errors� retry_codes�retry_deadline)r r r �returnc C s. || _ || _|| _|| _d| _d| _d | _d S )Nr r )r r r r �_logging_threshold�_request_count� _last_error)�selfr r r r r r r �__init__% s
z ReadUrlRetryHandler.__init__�r c C sJ | j d7 _ t|t�s.td| tjd� dS d}| jd k rNt� | jkrNd}nd}| jd k r�t|j t
j �r�| jd8 _| jdkr�d}n|j d k r�|j | j
kr�d}| j | jkr�| j| j9 _nd}|j d k r�| j|j kr�d}|j | _n:| jd k�st| jt��rt|j | j��s"d}t|j �| _|�s.|�sFtd| j |f tjd� |S )N� z1Polling IMDS failed with unexpected exception: %r�Z logger_funcFTr z1Polling IMDS failed attempt %d with exception: %r)r �
isinstancer r �LOG�warningr r r �cause�requests�ConnectionError�coder r r r �type)r Z exception�logZretryr r r �exception_callback: sf
���
� �
�
�
�
� � ��z&ReadUrlRetryHandler.exception_callback)
�__name__�
__module__� __qualname__�__doc__�floatr �intr �boolr- r r r r r s �� r T� )� log_response�timeout)�url�
retry_handlerr6 r7 r c
C sX zt | |jtd||d�}W n6 tk
rP } ztd| tjd� � W 5 d}~X Y nX |jS )a' Fetch URL from IMDS.
:param url: url to fetch.
:param log_response: log responses in readurl().
:param retry_deadline: time()-based deadline to retry until.
:param timeout: Read/connection timeout in seconds for readurl().
:raises UrlError: on error fetching metadata.
T�Z exception_cbr ZinfiniteZ log_req_respr7 z&Failed to fetch metadata from IMDS: %sr# N)r
r- r r r r% r& �contents)r8 r9 r6 r7 �response�errorr r r �
_fetch_url{ s �
�r> )r8 r9 r c
C sZ t | |d�}zt�|�d��W S tk
rT } ztd| tjd� � W 5 d}~X Y nX dS )z�Fetch IMDS metadata.
:param url: url to fetch.
:param retry_deadline: time()-based deadline to retry until.
:raises UrlError: on error fetching metadata.
:raises ValueError: on error parsing metadata.
�r9 zutf-8z&Failed to parse metadata from IMDS: %sr# N)r> r Z load_json�decode�
ValueErrorr r% r& )r8 r9 Zmetadatar= r r r �_fetch_metadata� s
�rB )r r r c
C s� t || d�}ztd }t||d�W S tk
r� } zF|jdkrttdtjd� t || d�}td }t||d� W Y �S � W 5 d}~X Y nX dS ) z�Fetch extended metadata, falling back to non-extended as required.
:param retry_deadline: time()-based deadline to retry until.
:raises UrlError: on error fetching metadata.
:raises ValueError: on error parsing metadata.
)r r z./instance?api-version=2021-08-01&extended=truer? i� z,Falling back to IMDS api-version: 2019-06-01r# z /instance?api-version=2019-06-01N)r �IMDS_URLrB r r* r r% r& )r r r9 r8 r= r r r � fetch_metadata_with_api_fallback� s(
�
��rD r! c C sP t d } tddddd�}t| |jtddd d
�}td |jd � d �tjd
� |j S )zhFetch extended metadata, falling back to non-extended as required.
:raises UrlError: on error.
z'/reprovisiondata?api-version=2019-06-01g @r" )r r r Nr TFr5 r: z Polled IMDS z time(s)r# )
rC r r
r- r r r r% �debugr; )r8 Zhandlerr< r r r �fetch_reprovision_data� s( �
� �rF )N) Zloggingr
�timer �typingr r r r r( Z cloudinitr Z cloudinit.sources.helpers.azurer Zcloudinit.url_helperr r
Z getLoggerr. r% rC r r r r4 r3 �bytesr> rB r2 rD rF r r r r �<module> s>
c�� $� � �