�
��>fY � �� � d dl Z d dlZd dlmZmZ d dlmZmZmZm Z m
Z
d dl m Z e j e
j e� � ZdZdZedz ez edz ez d �Zd
Zd Zd Zd
Z G d� de � Zy)� N)�Any�Dict)�
exceptions�http�secret_manager�system�util)�AutoAttachCloudInstancez http://169.254.169.254/metadata/z
2020-09-01z attested/document?api-version=z instance/compute?api-version=)�pkcs7�computez#/sys/class/dmi/id/chassis_asset_tagz%/var/lib/cloud/seed/azure/ovf-env.xmlz 7783-7084-3265-9085-8269-3286-77�
UBUNTU_PROc � � e Zd Ze ej
ej g d��� dee e
f fd�� � Z ede fd�� Z ede
fd�� Zde
fd�Zde
de
fd �Zy
) �UAAutoAttachAzureInstance)� r r )� retry_sleeps�returnc � � i }t t j � � D ]� \ }}t j |ddid�� }|j
dk7 r+t
j |j
|j �� �|dk( r4|j d }|||< t j j |� ��|j ||< �� |S ) N�Metadata�truer )�headers�timeout�� )�code�bodyr � signature)
�sorted� IMDS_URLS�itemsr �readurlr r �CloudMetadataErrorr � json_dictr �secrets�
add_secret)�self� responses�key�url�responser s �7/usr/lib/python3/dist-packages/uaclient/clouds/azure.py� identity_docz&UAAutoAttachAzureInstance.identity_doc s� � � � � �y��� 0�1�
4�H�C���|�|��j�&� 1�1��H� �}�}��#� �3�3�!� � �X�]�]�� � �g�~� $� .� .�{� ;� �!*� �#�� �&�&�1�1�)�<�!)�!3�!3� �#��
4� �� c � � y)N�azure� �r$ s r) �
cloud_typez$UAAutoAttachAzureInstance.cloud_type/ s � �r+ c �� � t j j t � r1t j
t � }t |j � k( ryt j j t � S )z&This machine is a viable AzureInstanceT) �os�path�exists�DMI_CHASSIS_ASSET_TAGr � load_file�AZURE_CHASSIS_ASSET_TAG�strip�AZURE_OVF_ENV_FILE)r$ �chassis_asset_tags r) � is_viablez#UAAutoAttachAzureInstance.is_viable3 sP � � �7�7�>�>�/� 0� &� 0� 0�1F� G� �&�*;�*A�*A�*C�C���w�w�~�~� 0�1�1r+ c � � y)NTr. r/ s r) �should_poll_for_pro_licensez5UAAutoAttachAzureInstance.should_poll_for_pro_license< s � �r+ �wait_for_changec �� � |rt j � �t j dd� } t j
|ddi�� }|j dk7 r3t j |j � t j � �|j j d� t k( S # t $ r.}t j |� t j � �d }~ww xY w)Nr � r r )r r � licenseType)
r �CancelProLicensePollingr �getr r �OSError�LOG�errorr r r! �AZURE_PRO_LICENSE_TYPE)r$ r> r' r( �es r) �is_pro_license_presentz0UAAutoAttachAzureInstance.is_pro_license_present@ s� � � � �4�4�6� 6��m�m�I�r�*�� 7��|�|�C�*�f�1E�F�H�
�=�=�C� � �I�I�h�m�m� $� �4�4�6� 6��!�!�%�%�m�4�8N�N�N�� � 7� �I�I�a�L� �4�4�6� 6�� 7�s �B+ �+ C"�4)C � C"N)�__name__�
__module__� __qualname__�propertyr �retryr r r �strr r* r0 �boolr; r= rI r. r+ r) r r s� � � ��T�Z�Z�
�-�-�I�F��d�3��8�n� � G� ��$ ��C� � �� �2�4� 2� �2��T� �O�� O�$� Or+ r )�loggingr2 �typingr r �uaclientr r r r r �uaclient.cloudsr
� getLogger� replace_top_level_logger_namerJ rE �
IMDS_BASE_URL� API_VERSIONr r5 r9 r7 rG r r. r+ r) �<module>rY s� �� � � � C� C� 3��g���:��:�:�8�D�E��2�
�� �
� =�
=� �
K� � >�>��L�
� �
>� �<� �<� �%� �7O� 7� 7Or+ |