File: //usr/lib/python3.9/site-packages/cloudinit/sources/azure/__pycache__/imds.cpython-39.pyc
a
'�Dg. � @ 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.9/site-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 urNt� | jkrNd}nd}| jd ur�t|j t
j �r�| jd8 _| jdkr�d}n|j d ur�|j | j
vr�d}| j | jkr�| j| j9 _nd}|j d ur�| j|j kr�d}|j | _n:| jd u �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 Zreq_args� exception�logZretryr r r �exception_callback: sd
���
� �
�
�
�
� � ��z&ReadUrlRetryHandler.exception_callback)
�__name__�
__module__� __qualname__�__doc__�floatr �intr �boolr. r r r r r s �� r T� )� log_response�timeout)�url�
retry_handlerr7 r8 r c
C sZ zt | |jtd||d�}W n8 tyR } z td| tjd� � W Y d}~n
d}~0 0 |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_respr8 z&Failed to fetch metadata from IMDS: %sr# N)r
r. r r r r% r& �contents)r9 r: r7 r8 �response�errorr r r �
_fetch_url{ s �
�r? )r9 r: r c
C s\ t | |d�}zt�|�d��W S tyV } z td| tjd� � W Y d}~n
d}~0 0 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.
�r: zutf-8z&Failed to parse metadata from IMDS: %sr# N)r? r Z load_json�decode�
ValueErrorr r% r& )r9 r: �metadatar> r r r �_fetch_metadata� s
�rD )r r r c
C s� t || d�}ztd }t||d�W S ty� } zR|jdkrvtdtjd� t || d�}td }t||d�W Y d}~S � W Y d}~n
d}~0 0 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_URLrD r r* r r% r& )r r r: r9 r> r r r � fetch_metadata_with_api_fallback� s(
�
��rF 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 TFr6 r; z Polled IMDS z time(s)r# )
rE r r
r. r r r r% �debugr< )r9 Zhandlerr= r r r �fetch_reprovision_data� s( �
� �rH )N) Zloggingr
�timer �typingr r r r r( Z cloudinitr Z cloudinit.sources.helpers.azurer Zcloudinit.url_helperr r
Z getLoggerr/ r% rE r r r r5 r4 �bytesr? rD r3 rF rH r r r r �<module> s<
c�� $� ��