CasperSecurity
o
쑛h$ � @ 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�dededededef
dd�Zdededefdd�Z d de dee defdd�Z defd d �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)�Metadatazx-ms-client-request-id)�str�uuid�uuid4)�_url� r �>/usr/lib/python3/dist-packages/cloudinit/sources/azure/imds.py�
headers_cb s
�r c @ sL e Zd ZdZddddd�dedee dee d dfd
d �Zd efd 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_deadliner 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__c C s@ | j d7 _ t|t�std| tjd� dS d}| jd ur't� | jkr'd}nd}| jd urDt|j t
j �rD| jd8 _| jdkrCd}n
|j d urQ|j | j
vrQd}| j | jkr`| j| j9 _nd}|j d urt| j|j krsd}|j | _n| jd u s�t| jt�r�t|j | j�s�d}t|j �| _|s�|s�td| j |f tjd� |S )N� z1Polling IMDS failed with unexpected exception: %r�� 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" � exception�log�retryr r r �exception_callback: s^
��
�
�
�
� � ��z&ReadUrlRetryHandler.exception_callback)
�__name__�
__module__� __qualname__�__doc__�floatr �intr# �boolr2 r r r r r s ��� �
�r T� )� log_response�timeout�url�
retry_handlerr; r< r c
C sL zt | |jtd||d�}W |jS ty% } z
td| tjd� � d}~ww )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�� exception_cbr �infinite� log_req_respr< z&Failed to fetch metadata from IMDS: %sr% N)r
r2 r r r r( r) �contents)r= r> r; r< �response�errorr r r �
_fetch_url{ s$ �����rF c
C sL t | |d�}z t�|�d��W S ty% } z
td| tjd� � d}~ww )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)rF r � load_json�decode�
ValueErrorr r( r) )r= r> �metadatarE r r r �_fetch_metadata� s
���rL r r c
C s� t || d�}z td }t||d�W S ty@ } z#|jdkr;tdtjd� t || d�}td }t||d�W Y d}~S � d}~ww ) 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=truerG i� z,Falling back to IMDS api-version: 2019-06-01r% z /instance?api-version=2019-06-01N)r �IMDS_URLrL r r- r r( r) )r r r> r= rE r r r � fetch_metadata_with_api_fallback� s,
�
����rN 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 TFr: r? z Polled IMDS z time(s)r% )
rM r r
r2 r r r r( �debugrC )r= �handlerrD r r r �fetch_reprovision_data� s( �
� �rQ )N) �loggingr �timer �typingr r r r r+ � cloudinitr � cloudinit.sources.helpers.azurer �cloudinit.url_helperr r
� getLoggerr3 r( rM r r r
r9 r8 �bytesrF rL r7 rN rQ r r r r �<module> sR
c�����
�#��
����
�