�
��>f� � �� � d dl Z d dlmZmZ d dlmZmZmZmZm Z d dl
m Z dZ dZ
e e
fZdZdZd Zd
Zd Zd Zd
Zedz Z e j. e j0 e� � Z G d� de � Zy)� N)�Any�Dict)�
exceptions�http�secret_manager�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 � � e Zd ZdZdZd� Ze ej e j g d��� de e e
f fd�� � Zd� Z ej e j g d��� d � � Zede fd
�� Zedefd �� Zdefd �Zd
edefd�Zy)�UAAutoAttachAWSInstanceNc � � | j � }t j t j | j
� |d�� }|j dk( r |j S t j |j |j �� �)N� )�headers�timeout�� ��code�body)
� _request_imds_v2_token_headersr �readurl�IMDS_URL�format� _ip_addressr r r �CloudMetadataError)�selfr �responses �5/usr/lib/python3/dist-packages/uaclient/clouds/aws.py�_get_imds_url_responsez.UAAutoAttachAWSInstance._get_imds_url_response sl � ��5�5�7���<�<� �O�O�D� ,� ,� -�w��
�� �=�=�C� ��=�=� � �/�/� �]�]�� � ��
� )g �?r r )� retry_sleeps�returnc �h � | j � }t j j |� d|iS )N�pkcs7)r r �secrets�
add_secret)r �imds_url_responses r � identity_docz$UAAutoAttachAWSInstance.identity_doc+ s4 � � !� 7� 7� 9��� � �)�)�*;�<��*�+�+r c �, � t D ] } | j |�� }|| _ n | j �)t j dj t � �� �S # t $ r,}t j d|t
|dd� � Y d }~��d }~ww xY w)N)�
ip_addressz*Could not reach AWS IMDS at http://%s: %s
�reason� z, )� addresses)
�IMDS_IP_ADDRESS�_get_imds_v2_token_headersr � Exception�LOG�warning�getattrr �AWSNoValidIMDS�join)r �addressr �es r r z6UAAutoAttachAWSInstance._request_imds_v2_token_headers2 s� � �&� �G�
� �9�9�W�9�M�� $+�� �� � � � � #� �+�+� �)�)�O�4��
� ��� �
�� � �A���A�x��,�� ��
�s �A � B�'"B�B)r � � c � � | j dk( ry | j rt | j iS t j t j |� dt t id�� }|j dk( rL|j | _ t j j | j � t | j iS |j dk( rd| _ y t j |j |j �� �)N�IMDSv1�PUTr )�methodr r r i� r )�
_api_token�AWS_TOKEN_PUT_HEADERr r �IMDS_V2_TOKEN_URLr �AWS_TOKEN_REQ_HEADER�AWS_TOKEN_TTL_SECONDSr r r r% r& r r )r r* r s r r/ z2UAAutoAttachAWSInstance._get_imds_v2_token_headersE s� � � �?�?�h� &��
�_�_�(�$�/�/�:� :��<�<� � $� $�Z� 0��)�+@�A��
�� �=�=�C� � &�m�m�D�O� � "� "� -� -�d�o�o� >�(�$�/�/�:� :� �=�=�C� � &�D�O���+�+�� � �X�]�]�
�
r c � � y)N�aws� �r s r �
cloud_typez"UAAutoAttachAWSInstance.cloud_type^ � � �r c �. � t j t � }d|dd k( ry t j t � j � }t j t � j � }d|dd cxk( r|dd k( ry yy# t $ r Y �nw xY w)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_serials r � is_viablez!UAAutoAttachAWSInstance.is_viableb s� � � � $� .� .�/J� K�O����!�,�,�� -� �#�#�$4�5�;�;�=���%�%�&8�9�?�?�A�
� �H�Q�q�M� 4�Z��!�_� 4�� 5��� !� �
� �s �!B � B�Bc � � y)� UnsupportedFrE rF s r �should_poll_for_pro_licensez3UAAutoAttachAWSInstance.should_poll_for_pro_licenset rH r �wait_for_changec �* � t j � �)N)r � InPlaceUpgradeNotSupportedError)r rY s r �is_pro_license_presentz.UAAutoAttachAWSInstance.is_pro_license_presentx s � ��8�8�:�:r )�__name__�
__module__� __qualname__r> r r �propertyr �retryr r r �strr r( r r/ rG �boolrU rX r\ rE r r r
r
s� � ��J��K�
� ��T�Z�Z�
�-�-�K�H�,�d�3��8�n� ,� I� �,�
�& �T�Z�Z�
�-�-�I�F�
� G�
�0 ��C� � �� ��4� � ��"�T� �;�� ;�$� ;r r
) �logging�typingr r �uaclientr r r r r �uaclient.cloudsr
�IMDS_IPV4_ADDRESS�IMDS_IPV6_ADDRESSr. r@ r rM rQ rO rB r? rA � getLogger� replace_top_level_logger_namer] r1 r
rE r r �<module>rl s� �� � � C� C� 3�%� �%� �$�&7�8��0� � =�� 4� �7� �3� � � �1� �+�n�<� ��g���:��:�:�8�D�E��`;� 5� `;r |