File: //usr/lib/python3/dist-packages/uaclient/clouds/__pycache__/aws.cpython-310.pyc
o
��wc� � @ s� d dl Z d dlmZmZ d dlmZ d dlmZmZm Z d dl
m Z dZ dZ
e e
fZdZd Zd
Zd Zd Zd
ZdZed ZG dd� de �ZdS )� N)�Any�Dict)� HTTPError)�
exceptions�system�util)�AutoAttachCloudInstancez169.254.169.254z[fd00:ec2::254]zhttp://{}/latest/api/tokenz0http://{}/latest/dynamic/instance-identity/pkcs7z/sys/hypervisor/uuidz /sys/class/dmi/id/product_serialz /sys/class/dmi/id/product_uuid�21600zX-aws-ec2-metadata-tokenz -ttl-secondsc @ s� e Zd ZdZdZdd� Zeeje g d�d�de
e e f fdd���Z
d d
� Zeje g d �d�d d
� �Zede fdd��Zedefdd��Zdefdd�Zdedefdd�ZdS )�UAAutoAttachAWSInstanceNc C s | � � }tjt�| j�|dd�S )N� )�headers�timeout)� _request_imds_v2_token_headersr �readurl�IMDS_URL�format� _ip_address)�selfr � r �5/usr/lib/python3/dist-packages/uaclient/clouds/aws.py�_get_imds_url_response s �z.UAAutoAttachAWSInstance._get_imds_url_response)g �?r r )� retry_sleeps�returnc C s | � � \}}d|iS )N�pkcs7)r )r �response�_headersr r r � identity_doc% s z$UAAutoAttachAWSInstance.identity_docc C s� t D ]=}z| j|d�}W n/ ty } z|�d }~w ty; } zdj|t|dd�d�}t�|� W Y d }~qd }~ww || _ | jd u rOt �
dd� t � ��|S )N)�
ip_addressz8Could not reach AWS IMDS at http://{endpoint}: {reason}
�reason� )�endpointr z6No valid AWS IMDS endpoint discovered at addresses: %sz, ) �IMDS_IP_ADDRESS�_get_imds_v2_token_headersr � Exceptionr �getattr�logging�debugr r �UserFacingError�join)r �addressr �e�msgr r r r + s0 � ����
��z6UAAutoAttachAWSInstance._request_imds_v2_token_headers)r � � c
C s� | j dkrd S | j rt| j iS ztjt�|�dttidd�\}}W n ty= } z|j dkr8d| _ W Y d }~d S � d }~ww || _ t| j iS )N�IMDSv1�PUTr )�methodr r
i� )
�
_api_token�AWS_TOKEN_PUT_HEADERr r �IMDS_V2_TOKEN_URLr �AWS_TOKEN_REQ_HEADER�AWS_TOKEN_TTL_SECONDSr �code)r r r r r* r r r r" C s(
�
��
z2UAAutoAttachAWSInstance._get_imds_v2_token_headersc C � dS )N�awsr �r r r r �
cloud_typeZ � z"UAAutoAttachAWSInstance.cloud_typec C s� zt �t�}d|dd� krW dS W n ty Y nw t �t��� }t �t��� }d|dd� kr=|dd� kr@dS dS dS )z$This machine is a viable AWSInstance�ec2r � TF)r � load_file�SYS_HYPERVISOR_PRODUCT_UUID�FileNotFoundError�DMI_PRODUCT_UUID�lower�DMI_PRODUCT_SERIAL)r �hypervisor_uuid�dmi_uuid�
dmi_serialr r r � is_viable^ s
� �"�z!UAAutoAttachAWSInstance.is_viablec C r7 )� UnsupportedFr r9 r r r �should_poll_for_pro_licensep r; z3UAAutoAttachAWSInstance.should_poll_for_pro_license�wait_for_changec C s t �� �)N)r � InPlaceUpgradeNotSupportedError)r rJ r r r �is_pro_license_presentt s z.UAAutoAttachAWSInstance.is_pro_license_present)�__name__�
__module__� __qualname__r1 r r �propertyr �retryr r �strr r r r" r: �boolrG rI rL r r r r r
s
r
)r% �typingr r � urllib.errorr �uaclientr r r �uaclient.cloudsr �IMDS_IPV4_ADDRESS�IMDS_IPV6_ADDRESSr! r3 r r? rC rA r5 r2 r4 r
r r r r �<module> s"