�
x[h$ � � � S SK r S SKrS SKJr S SKJrJrJrJr S SK r S SK
J r S SK J
r
S SKJrJr \ R" " \5 rSrS r " S S
5 rS S S
.S\S\S\S\S\4
S jjrS\S\S\4S jr SS\ S\\ S\4S jjr S\4S jr g)� N)� monotonic)�Dict�Optional�Type�Union)�util)�report_diagnostic_event)�UrlError�readurlz http://169.254.169.254/metadatac �D � S[ [ R " 5 5 S.$ )N�true)�Metadatazx-ms-client-request-id)�str�uuid�uuid4)�_urls �>/usr/lib/python3/dist-packages/cloudinit/sources/azure/imds.py�
headers_cbr s � ��"%�d�j�j�l�"3� � � c �Z � \ rS rSrSrSSSSS.S\S\\ S \\ S
S4S jjrS
\ 4S jr
S
r g)�ReadUrlRetryHandler� a7 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 �^ � Xl X l X0l X@l SU l SU l S U l g )Nr r )r r r r! �_logging_threshold�_request_count� _last_error)�selfr r r r! s r �__init__� ReadUrlRetryHandler.__init__% s4 � � /� �%:�"�&�� ,��"%�� � ���37��r c �* � U =R S-
sl [ U[ 5 ( d [ SU- [ R
S9 gSnU R b [ 5 U R :� a SnOSnU R bQ [ UR [ R 5 ( a( U =R S- sl U R S:: a SnO)UR b UR U R ; a SnU R U R :� a U =R U R - sl OSnUR b. U R UR :w a SnUR U l OmU R bD [ U R [" 5 ( a% [ UR U R 5 ( d Sn[# UR 5 U l U( d U( d' [ SU R U4- [ R
S9 U$ )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�retrys r �exception_callback�&ReadUrlRetryHandler.exception_callback: s� � � ���q� ���)�X�.�.� #�C� � � �K�K�
�
��� � � � +�� �t� 2� 2�2��E��E�
� %� %� 1�j� �O�O�X� 5� 5�7
� 7
�
� &� &�!� +� &��)�)�Q�.� ��� �N�N� &����d�&6�&6�6��E� � � �$�"9�"9� 9� � #� #�t�';�';� ;� #��C� �>�>� %�� � �9�>�>�1���#,�>�>�� ��
� � � $� �d� .� .��5�5� �i�o�o�t�/?�/?�@�@��C� #�I�O�O� 4�D� � �e� #�C��&�&� �2�3� �K�K�
�
� r )r& r$ r% r r r r! ) �__name__�
__module__� __qualname__�__firstlineno__�__doc__�floatr �intr( �boolr9 �__static_attributes__� r r r r sZ � �� "%�/3�
� +/�8� �8� (��}� 8� !���8�
�8�*>�t� >r r T� )� log_response�timeout�url�
retry_handlerrF rG r" c � � [ U UR [ SUUS9nUR $ ! [ a! n[ SU- [
R S9 e SnAff = f)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_resprG z&Failed to fetch metadata from IMDS: %sr, N)r r9 r r
r r/ r0 �contents)rH rI rF rG �response�errors r �
_fetch_urlrR { se � � �� �&�9�9�!��%��
�� � � � �� � � � 4�u� <�� � �
� �� �s � * �
A� A�Ac � � [ XS9n [ R " UR S5 5 $ ! [ a! n[ SU- [ R S9 e SnAff = f)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.
�rI zutf-8z&Failed to parse metadata from IMDS: %sr, N)rR r � load_json�decode�
ValueErrorr r/ r0 )rH rI �metadatarQ s r �_fetch_metadatarY � sW � � �#�;�H���~�~�h�o�o�g� 6�7�7�� � � � 4�u� <�� � �
� �� �s �$0 �
A� A�Ar! r c �� � [ UU S9n [ S- n[ X2S9$ ! [ aO nUR S:X a9 [ S[ R S9 [ UU S9n[ S- n[ X2S9s SnA$ e SnAff = f) 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=truerT i� z,Falling back to IMDS api-version: 2019-06-01r, z /instance?api-version=2019-06-01N)r �IMDS_URLrY r
r4 r r/ r0 )r! r rI rH rQ s r � fetch_metadata_with_api_fallbackr\ � s� � � (� 3�%��M���I�I�� �s�@�@�� � � �:�:�� � #�>� �K�K�
� 0�&;� -� �M� � ?�?�C�"�3�D� D�
�� �s! � �
A7�AA2�+A7�1A2�2A7c �� � [ S- n [ SSSSS9n[ U UR [ SSS S
9n[ S UR S- S 3[ R S
9 UR $ )z`Fetch 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 TFrE rK z Polled IMDS z time(s)r, )
r[ r r r9 r r r% r/ �debugrO )rH �handlerrP s r �fetch_reprovision_datar` � s� � �
�>�
>�C�!�� �
�
� �G� � � �/�/� ����
�H� �
�w�-�-�a�/�0��9��I�I�� � � � r )N) �loggingr �timer �typingr r r r r2 � cloudinitr � cloudinit.sources.helpers.azurer �cloudinit.url_helperr
r � getLoggerr; r/ r[ r r r rB rA �bytesrR rY r@ r\ r` rD r r �<module>ri s� ��
� � � .� .� � � C� 2�
����!�� ,���\� \�F �� � � � '� � � �
� � �
�F� �� '��
� �4 CG� �� �2:�3�-� � � �@ �� r