CasperSecurity
o
�K�ic � @ s� d dl Z d dlZd dlmZmZ d dlmZmZmZm Z m
Z
d dl m Z e �
e
�e��ZdZdZed e ed e d �Zd
Zd Zd Zd
ZG dd� de �ZdS )� N)�Any�Dict)�
exceptions�http�secret_manager�system�util)� PublicCloudAutoAttachInstancez 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 @ s~ e Zd Zeejejg d�d�dee e
f fdd���Z ede fdd��Z ede
fdd ��Zde
fd
d �Zd e
de
fd
d�ZdS )�AzureAutoAttachInstance)� r r )� retry_sleeps�returnc C s� i }t t�� �D ]5\}}tj|ddidd�}|jdkr$tj|j|jd��|dkr8|j d }|||< t
j � |� q|j ||< q|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 � r& �7/usr/lib/python3/dist-packages/uaclient/clouds/azure.py� identity_doc s
�
�
z$AzureAutoAttachInstance.identity_docc C � dS )N�azurer& �r! r& r& r' �
cloud_type/ � z"AzureAutoAttachInstance.cloud_typec C s2 t j�t�rt�t�}t|�� krdS t 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_tagr& r& r' � is_viable3 s
z!AzureAutoAttachInstance.is_viablec C r) )NTr& r+ r& r& r' �should_poll_for_pro_license< r- z3AzureAutoAttachInstance.should_poll_for_pro_license�wait_for_changec
C s� |rt �� �t�dd�}z tj|ddid�}W n ty, } z t�|� t �� �d }~ww |j dkr<t�|j
� t �� �|j �d�t kS )Nr � r r )r r � licenseType)
r �CancelProLicensePollingr �getr r �OSError�LOG�errorr r r �AZURE_PRO_LICENSE_TYPE)r! r9 r$ r% �er& r& r' �is_pro_license_present@ s
��
z.AzureAutoAttachInstance.is_pro_license_presentN)�__name__�
__module__� __qualname__�propertyr �retryr r r �strr r( r, �boolr7 r8 rC r& r& r& r' r
s r
)�loggingr. �typingr r �uaclientr r r r r �uaclient.cloudsr � getLogger� replace_top_level_logger_namerD r? �
IMDS_BASE_URL� API_VERSIONr r1 r5 r3 rA r
r& r& r& r' �<module> s
�