%PDF-
%PDF-
Mini Shell
Mini Shell
�
x[h$ � � � 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& � 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' �boolr7 � r r r r sZ � �� "%�/3�
� +/�8� �8� (��}� 8� !���8�
�8�*>�t� >r r T� )� log_response�timeout�url�
retry_handlerrA rB 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_resprB z&Failed to fetch metadata from IMDS: %sr* N)r r7 r r
r r- r. �contents)rC rD rA rB �response�errors r �
_fetch_urlrM { 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.
�rD zutf-8z&Failed to parse metadata from IMDS: %sr* N)rM r � load_json�decode�
ValueErrorr r- r. )rC rD �metadatarL s r �_fetch_metadatarT � 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=truerO i� z,Falling back to IMDS api-version: 2019-06-01r* z /instance?api-version=2019-06-01N)r �IMDS_URLrT r
r2 r r- r. )r r rD rC rL s r � fetch_metadata_with_api_fallbackrW � 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 TFr@ rF z Polled IMDS z time(s)r* )
rV r r r7 r r r$ r- �debugrJ )rC �handlerrK 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 � getLoggerr8 r- rV r r r r> r= �bytesrM rT r< rW r[ r? r r �<module>rd s� ��
� � � .� .� � � C� 2��g����!�� ,���\� \�F �� � � � '� � � �
� � �
�F� �� '��
� �4 CG� �� �2:�3�-� � � �@ �� r
Zerion Mini Shell 1.0