Name	Size	Mode	Actions
errors.cpython-36.opt-1.pyc	6494	0644	editdlrm
errors.cpython-36.pyc	6494	0644	editdlrm
identity.cpython-36.opt-1.pyc	2959	0644	editdlrm
identity.cpython-36.pyc	2959	0644	editdlrm
imds.cpython-36.opt-1.pyc	5148	0644	editdlrm
imds.cpython-36.pyc	5148	0644	editdlrm
kvp.cpython-36.opt-1.pyc	1708	0644	editdlrm
kvp.cpython-36.pyc	1708	0644	editdlrm
__init__.cpython-36.opt-1.pyc	113	0644	editdlrm
__init__.cpython-36.pyc	113	0644	editdlrm

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�s$



 `