| Current File : //usr/lib/python3/dist-packages/uaclient/clouds/__pycache__/aws.cpython-312.pyc |
�
��g� � �� � d dl Z d dlZd dlmZmZ d dlmZmZmZm Z m
Z
d dl m Z dZ
dZe
efZdZdZd Zd
Zd Zd Zd
ZdZedz Z ej2 e
j4 e� � Z G d� de � Z y)� N)�Any�Dict)�
exceptions�http�secret_manager�system�util)� PublicCloudAutoAttachInstancez169.254.169.254z[fd00:ec2::254]zhttp://{}/latest/api/tokenz0http://{}/latest/dynamic/instance-identity/pkcs7z3http://{}/latest/dynamic/instance-identity/documentz/sys/hypervisor/uuidz /sys/class/dmi/id/product_serialz /sys/class/dmi/id/product_uuid�21600zX-aws-ec2-metadata-tokenz -ttl-secondsc � � e Zd ZdZdZdefd�Ze ej e
j g d��� de ee
f fd�� � Z ej e
j g d��� de fd�� Zedefd �� Zd
� Z ej e
j g d ��� d � � Zedefd
�� Zedefd�� Zdefd�Zdedefd�Zy)�AWSAutoAttachInstanceN�urlc � � t j ||d�� }|j dk( r |j S t j
|j |j �� �)N� )�headers�timeout�� ��code�body)r �readurlr r r �CloudMetadataError)�selfr r �responses �5/usr/lib/python3/dist-packages/uaclient/clouds/aws.py�_get_imds_url_responsez,AWSAutoAttachInstance._get_imds_url_response sJ � ��<�<��W�a�@�� �=�=�C� ��=�=� � �/�/� �]�]�� � ��
� )g �?r r )� retry_sleeps�returnc �� � | j � }t j | j � }| j ||�� }t
j j |� d|iS )N�r �pkcs7)� _request_imds_v2_token_headers�IMDS_URL�format� _ip_addressr r �secrets�
add_secret)r r r �imds_url_responses r � identity_docz"AWSAutoAttachInstance.identity_doc* s] � � �5�5�7���o�o�d� .� .�/�� � 7� 7��W� 7� M��� � �)�)�*;�<��*�+�+r c �" � | j � }t j | j � } t j
| j
||�� � }|S # t j $ r"}t j d|� i cY d}~S d}~ww xY w)a
Get the instance identity doc associated with the current instance.
See
https://docs.aws.amazon.com/AWSEC2/latest/UserGuide/retrieve-iid.html
for more context.
@return: Dict containing the instance identity document.
r! z-Error decoding instance identity document: %sN)
r# �
_IMDS_IID_URLr% r& �json�loadsr �JSONDecodeError�LOG�debug)r r r �ii_doc�es r � _get_ii_docz!AWSAutoAttachInstance._get_ii_doc3 s� � � �5�5�7���"�"�4�#3�#3�4�� ��Z�Z��+�+�C��+�A��F� �
�� �#�#� � �I�I�E�q� I��I�� �s �&A �B�,B �B� Bc � � | j � }|j dd� }|j dd� }t |� xs t |� S )a
Determines if the instance is likely Ubuntu Pro.
Criteria: if any billing-product or marketplace-product-code is
present, then is likely a Pro instance.
@return: Boolean indicating if the instance is likely pro or not.
�billingProductsN�marketplaceProductCodes)r4 �get�bool)r r2 �billing_products�marketplace_product_codess r �
is_likely_proz#AWSAutoAttachInstance.is_likely_proI sM � � �!�!�#��!�:�:�&7��>��$*�J�J�/H�$�$O�!��$�%�H��.G�)H�Hr 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& � Exceptionr0 �warning�getattrr �AWSNoValidIMDS�join)r �addressr r3 s r r# z4AWSAutoAttachInstance._request_imds_v2_token_headersX 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 rC z0AWSAutoAttachInstance._get_imds_v2_token_headersk 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 AWSAutoAttachInstance.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 AWSAutoAttachInstance.is_viable� s� � � � $� .� .�/J� K�O����!�,�,�� -� �#�#�$4�5�;�;�=���%�%�&8�9�?�?�A�
� �H�Q�q�M� 4�Z��!�_� 4�� 5��� !� �
� �s �!B � B�Bc � � y)� UnsupportedFrW rX s r �should_poll_for_pro_licensez1AWSAutoAttachInstance.should_poll_for_pro_license� rZ r �wait_for_changec �* � t j � �)N)r � InPlaceUpgradeNotSupportedError)r rk s r �is_pro_license_presentz,AWSAutoAttachInstance.is_pro_license_present� s � ��8�8�:�:r )�__name__�
__module__� __qualname__rP r& �strr �propertyr �retryr r r r r* r4 r9 r< r# rC rY rg rj rn rW r r r
r
s& � ��J��K��#� � ��T�Z�Z�
�-�-�K�H�,�d�3��8�n� ,� I� �,� �T�Z�Z�
�-�-�K�H��T� � I��* � I�t� I� � I� �& �T�Z�Z�
�-�-�I�F�
� G�
�0 ��C� � �� ��4� � ��"�T� �;�� ;�$� ;r r
) r- �logging�typingr r �uaclientr r r r r �uaclient.cloudsr
�IMDS_IPV4_ADDRESS�IMDS_IPV6_ADDRESSrB rR r$ r, r_ rc ra rT rQ rS � getLogger� replace_top_level_logger_namero r0 r
rW r r �<module>r} s� �� � � � C� C� 9�%� �%� �$�&7�8��0� � =��E�
� 4� �7� �3� � � �1� �+�n�<� ��g���:��:�:�8�D�E��D;� 9� D;r