Your IP : 216.73.217.13
U
�Ad� � @ s� d dl mZ d dlZd dlmZ d dlmZ d dlmZ d dl m
Z
m Z e� e
�ZdZG dd � d �Zd
d d d
�eeeeed�dd�Zdeeed�dd�Zdeed�dd�Zed�dd�ZdS ) � )�DictN)�log)�util)�report_diagnostic_event)�UrlError�readurlz http://169.254.169.254/metadatac @ s6 e Zd Zdddd�eedd�dd�Zed �d
d �ZdS ) �ReadUrlRetryHandler)� � i� i� �
� �?)� retry_codes�max_connection_errors�logging_backoffN)r r �returnc C s" || _ || _|| _d| _d| _d S )Nr r )r r r
�_logging_threshold�_request_count)�selfr
r r � r �>/usr/lib/python3/dist-packages/cloudinit/sources/azure/imds.py�__init__ s
z ReadUrlRetryHandler.__init__�r c C s� | j d7 _ t|t�s.td| tjd� dS d}d}t|jtjtj f�rh| j
d8 _
| j
dk rxd}n|j | j krxd}| j | j
kr�| j
| j9 _
nd}|s�|s�td| 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�warning�cause�requests�ConnectionErrorZTimeoutr �coder
r r �info)r Zreq_argsZ exceptionr Zretryr r r �exception_callback& s>
��
�
��z&ReadUrlRetryHandler.exception_callback)�__name__�
__module__� __qualname__�int�floatr �boolr" r r r r r s � � r Tr � )� log_response�retries�timeout)�urlr* r+ r, r c
C sd t � }z t| |jddid|||d�}W n6 tk
r\ } ztd| tjd� � W 5 d}~X Y nX |jS )zLFetch URL from IMDS.
:raises UrlError: on error fetching metadata.
�Metadata�trueF)� exception_cb�headers�infinite� log_req_respr+ r, z&Failed to fetch metadata from IMDS: %sr N)r r r" r r r r �contents)r- r* r+ r, �handler�response�errorr r r �
_fetch_urlM s$ �
�r8 )r- r+ r c
C sT t | |d�}z t�|�W S tk
rN } ztd| tjd� � W 5 d}~X Y nX dS )zFetch IMDS metadata.
:raises UrlError: on error fetching metadata.
:raises ValueError: on error parsing metadata.
�r+ z&Failed to parse metadata from IMDS: %sr N)r8 r Z load_json�
ValueErrorr r r )r- r+ Zmetadatar7 r r r �_fetch_metadataj s �r; )r+ r c
C st zt d }t|| d�W S tk
rn } z:|jdkr\tdtjd� t d }t|| d� W Y �S � W 5 d}~X Y nX dS )z�Fetch extended metadata, falling back to non-extended as required.
:raises UrlError: on error fetching metadata.
:raises ValueError: on error parsing metadata.
z./instance?api-version=2021-08-01&extended=truer9 i� z,Falling back to IMDS api-version: 2019-06-01r z /instance?api-version=2019-06-01N)�IMDS_URLr; r r r r r )r+ r- r7 r r r � fetch_metadata_with_api_fallback s
�r= r c C sR t d } tdddd�}t| |jddidd d
d �}td |jd
� d�tjd� |jS )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 r r
r. r/ TFr) )r0 r1 r2 r3 r, z Polled IMDS r z time(s)r ) r< r r r" r r r �debugr4 )r- r5 r6 r r r �fetch_reprovision_data� s& �� �r? )r )r )�typingr r Z cloudinitr Zloggingr Z cloudinit.sources.helpers.azurer Zcloudinit.url_helperr r Z getLoggerr# r r< r �strr( r& �bytesr8 r; r= r? r r r r �<module> s4
; � � ��