File: //lib/python3/dist-packages/cloudinit/sources/azure/__pycache__/imds.cpython-310.pyc
o
�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
�dededededef
dd�Z ddededefdd�Zddedefdd�Zdefdd�ZdS ) � )�DictN)�log)�util)�report_diagnostic_event)�UrlError�readurlz http://169.254.169.254/metadatac @ s: e Zd Zdddd�dededdfd d
�Zdefd d �ZdS )
�ReadUrlRetryHandler)� � i� i� �
� �?)� retry_codes�max_connection_errors�logging_backoffr r �returnNc 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__c C s� | j d7 _ t|t�std| tjd� dS d}d}t|jtjtj f�r4| j
d8 _
| j
dk r3d}n|j | j vr<d}| j | j
krK| j
| j9 _
nd}|sQ|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�warning�cause�requests�ConnectionError�Timeoutr �coder
r r �info)r �req_args� exceptionr �retryr 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/ r0 r c
C sX t � }zt| |jddid|||d�}W |jS ty+ } z
td| tjd� � d}~ww )zLFetch URL from IMDS.
:raises UrlError: on error fetching metadata.
�Metadata�trueF)� exception_cb�headers�infinite� log_req_respr/ r0 z&Failed to fetch metadata from IMDS: %sr N)r r r& r r r r �contents)r1 r. r/ r0 �handler�response�errorr r r �
_fetch_urlM s( �����r< c
C sF t | |d�}zt�|�W S ty" } z
td| tjd� � d}~ww )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)r< r � load_json�
ValueErrorr r r )r1 r/ �metadatar; r r r �_fetch_metadataj s ���rA c
C sj z t d }t|| d�W S ty4 } z |jdkr/tdtjd� t d }t|| d�W Y d}~S � d}~ww )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=truer= i� z,Falling back to IMDS api-version: 2019-06-01r z /instance?api-version=2019-06-01N)�IMDS_URLrA r r! r r r )r/ r1 r; r r r � fetch_metadata_with_api_fallback s
���rC 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
r2 r3 TFr- )r4 r5 r6 r7 r0 z Polled IMDS r z time(s)r ) rB r r r& r r r �debugr8 )r1 r9 r: r r r �fetch_reprovision_data� s& �� �rE )r )�typingr r � cloudinitr �loggingr � cloudinit.sources.helpers.azurer �cloudinit.url_helperr r � getLoggerr' r rB r �strr, r* �bytesr< rA rC rE r r r r �<module> s>
;�����
� ���
�