Current File : //lib/python3/dist-packages/cloudinit/sources/azure/__pycache__/imds.cpython-312.pyc
�
���f. � � � d dl Z d dlZd dlmZ d dlmZmZmZmZ d dl Z d dl
m Z d dl m
Z
d dlmZmZ e j" e� ZdZd� Z G d � d
� Zd d d
�dededededef
d�Zdededefd�Z dde dee defd�Z defd�Z y)� N)� monotonic)�Dict�Optional�Type�Union)�util)�report_diagnostic_event)�UrlError�readurlz http://169.254.169.254/metadatac �B � dt t j � � d�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 �L � e Zd ZdZddddd�dedee dee d dfd
�Zd efd �Z y) �ReadUrlRetryHandleraK 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 �f � || _ || _ || _ || _ d| _ d| _ d | _ y )Nr r )r r r r �_logging_threshold�_request_count� _last_error)�selfr r r r s r �__init__z ReadUrlRetryHandler.__init__% s< � � /�� �%:��"�&��� ,���"%�� � ���37��r c �� � | xj dz
c_ t |t � s t d|z t j
�� yd}| j �t � | j k\ rd}nd}| j �Kt |j t j � r'| xj dz c_ | j dk r)d}n&|j �|j | j vrd}| j | j k\ r | xj | j z c_ nd}|j �-| j |j k7 rvd}|j | _ nb| j �:t | j t" � r t |j | j � s d}t# |j � | _ |s|s*t d| j |fz t j
�� |S )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& �req_args� exception�log�retrys r �exception_callbackz&ReadUrlRetryHandler.exception_callback: s� � � ���q� ���)�X�.� #�C� � � �K�K�
�
��� � � � +�� �t� 2� 2�2��E��E�
� %� %� 1�j� �O�O�X� 5� 5�7
�
� &� &�!� +� &��)�)�Q�.� �� �N�N� &����d�&6�&6�6��E� � � �$�"9�"9� 9� � #� #�t�';�';� ;� #��C� �>�>� %�� � �9�>�>�1���#,�>�>�� �
� � � $� �d� .� .��5� �i�o�o�t�/?�/?�@��C� #�I�O�O� 4�D� � �e� #�C��&�&� �2�3� �K�K�
�
� r )
�__name__�
__module__� __qualname__�__doc__�floatr �intr' �boolr8 � r r r r sZ � �� "%�/3�
� +/�8� �8� (��}� 8� !���8�
�8�*>�� >r r T� )� log_response�timeout�url�
retry_handlerrB rC r! c � � t | |j t d||�� }|j S # t $ r$}t d|z t
j �� � d}~ww xY w)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_resprC z&Failed to fetch metadata from IMDS: %sr* N)r r8 r r
r r- r. �contents)rD rE rB rC �response�errors r �
_fetch_urlrN { se � � �� �&�9�9�!��%��
�� � � � �� � � � 4�u� <�� � �
� �� �s � - � A� A�Ac �� � t | |�� } t j |j d� � S # t $ r$}t d|z t j �� � d}~ww xY w)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.
�rE zutf-8z&Failed to parse metadata from IMDS: %sr* N)rN r � load_json�decode�
ValueErrorr r- r. )rD rE �metadatarM s r �_fetch_metadatarU � sY � � �#�]�;�H���~�~�h�o�o�g� 6�7�7�� � � � 4�u� <�� � �
� �� �s �#3 � A � A�A r r c � � t || �� } t dz }t ||�� S # t $ rX}|j dk( rCt dt j �� t || �� }t dz }t ||�� cY d}~S � d}~ww xY w) 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=truerP i� z,Falling back to IMDS api-version: 2019-06-01r* z /instance?api-version=2019-06-01N)r �IMDS_URLrU r
r2 r r- r. )r r rE rD rM s r � fetch_metadata_with_api_fallbackrX � s� � � (� 3�%��M���I�I�� �s�-�@�@�� � � �:�:�� � #�>� �K�K�
� 0�&;� -� �M� � ?�?�C�"�3�m�D� D�
�� �s! �% � B�A B�:B� B�Bc �� � t dz } t dddd�� }t | |j t ddd �
� }t d |j dz � d �t j �
� |j S )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 Nr TFrA rG z Polled IMDS z time(s)r* )
rW r r r8 r r r$ r- �debugrK )rD �handlerrL s r �fetch_reprovision_datar\ � s� � �
�>�
>�C�!�� �
�
� �G� � � �/�/� ����
�H� �
�w�-�-�a�/�0��9��I�I�� � � � r )N) �loggingr �timer �typingr r r r r0 � cloudinitr � cloudinit.sources.helpers.azurer �cloudinit.url_helperr
r � getLoggerr9 r- rW r r r r? r> �bytesrN rU r= rX r\ r@ r r �<module>re s� ��
� � � .� .� � � C� 2��g����!�� ,���\� \�F �� � � � '� � � �
� � �
�F� �� '��
� �4 CG� �� �2:�3�-� � � �@ �� r
Mini Shell