3

\�men�@s�dd
lZddl
m
Z

ddlmZmZmZmZ
dd
lZddlm	Z	
ddl
mZ
ddlm
Z
mZ
eje�
ZdZGdd	�d	�Zd
dd�eeeeed
�dd�Zeeed�dd�Zdeeeed�dd�Zed�
dd�Zd
S)�N)
�time)�Dict�Optional�Type�Union)
�util)
�report_diagnostic_event)�UrlError�readurlzhttp://169.254.169.254/metadatac@sFeZ
dZd
Zddddd�eeeeedd	�d
d�Zed�
d
d�Z	dS)�ReadUrlRetryHandleraF
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 time()-based deadline to retry until.
    g�?N�
�
�
��
)�logging_backoff�max_connection_errors�retry_codes�retry_deadline)rrr�returnc
Cs.|
|_||_
||_||_d
|_d|_d|_dS)Ng�?r
)rrrr�_logging_threshold�_request_count�_last_error)�selfrrrr�r�/usr/lib/python3.6/imds.py�__init__s






zReadUrlRetryHandler.__init__)
rcCsP
|jd
7_t
|t�s.td|tjd�
dSd}|jdk	rNt�|jkrNd}nd}|jdk	r�t
|j	t
j�r�|jd
8_|jdk
r�d}n|jdk	r�|j|j
kr�d}|j|jkr�|j|j9_nd}|jdk	r�|j|jkr�d}|j|_n>|jdk�
st
|jt��
st
|j	|j��
r&d}t|j	�
|_|�
s4|�
rLtd|j|ftjd�
|S)N�
z1Polling IMDS failed with unexpected exception: %r)
�logger_funcFTr
z1Polling IMDS failed attempt %d with exception: %r)r�
isinstancer	r�LOG�warningrrr�cause�requests�ConnectionError�coderrrr�type�info)rZreq_argsZ	exception�logZretryrrr�exception_callback2sJ



























z&ReadUrlRetryHandler.exception_callback)rr
rr)
�__name__�
__module__�__qualname__�__doc__�floatr�intr�boolr(rrrrrs




	rT�)�log_response�timeout)�url�
retry_handlerr1r2rc
Cs\yt||
j
d
di
d||d�}Wn6tk
rT
}
ztd|tjd�
�WYdd}~XnX|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.
    �Metadata�trueT)�exception_cb�headers�infinite�log_req_respr2z&Failed to fetch metadata from IMDS: %s)
rN)r
r(r	rrr �contents)r3r4r1r2�response�errorrrr�
_fetch_urlps










r>)r3r4rc

CsRt||
d
�}y
t
j|�
Stk
rL
}
ztd|tjd�
�WYdd}~XnXdS)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.
    )
r4z&Failed to parse metadata from IMDS: %s)
rN)r>rZ	load_json�
ValueErrorrrr )r3r4Zmetadatar=rrr�_fetch_metadata�s







r@)rrrcCs�t|
|d
�}yt
d}t||d�Stk
r|
}
z@|jdkrjtdtjd�
t|
|d
�}t
d}t||d�S�WYdd}~XnXdS)	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.
    )rrz./instance?api-version=2021-08-01&extended=true)
r4i�
z,Falling back to IMDS api-version: 2019-06-01)
rz /instance?api-version=2019-06-01N)r�IMDS_URLr@r	r$rrr )rrr4r3r=rrr� fetch_metadata_with_api_fallback�s"















rB)
rcCsTtd
}t
ddddd�}
t||
jd	d
i
dd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�
�
�
N)rrrrr5r6TFr0)r7r8r9r:r2zPolled IMDS z time(s))
r)rCrDrE)	rArr
r(rrr�debugr;)r3Zhandlerr<rrr�fetch_reprovision_data�s&












rG)
N)ZloggingrZtypingrrrrr"Z	cloudinitrZcloudinit.sources.helpers.azurerZcloudinit.url_helperr	r
Z	getLoggerr)rrAr�strr/r.�bytesr>r@r-rBrGrrrr�<module>s$




`