�
��g� � �� � S SK r S SKrS SKJrJr S SKJrJrJrJ r J
r
S SK J r Sr
Sr\
\4rSrSrS rS
rS rS rS
rSr\S- r\R2 " \
R4 " \5 5 r " S S\ 5 r g)� 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 � � \ rS rSrSrSrS\4S jr\\ R " \ R / SQS9S\
\\4 4S j5 5 r\ R " \ R / SQS9S\
4S j5 r\S\4S
j5 rS r\ R " \ R / S QS9S
5 r\S\4S j5 r\S\4S j5 rS\4S jrS\S\4S jrSrg)�AWSAutoAttachInstance� N�urlc � � [ R " XSS9nUR S:X a UR $ [ R
" UR UR S9e)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_response�,AWSAutoAttachInstance._get_imds_url_response sH � ��<�<��a�@�� �=�=�C� ��=�=� � �/�/� �]�]�� � ��
� )g �?r r )� retry_sleeps�returnc �� � U R 5 n[ R U R 5 nU R X!S9n[
R R U5 SU0$ )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_doc�"AWSAutoAttachInstance.identity_doc* s[ � � �5�5�7���o�o�d� .� .�/�� � 7� 7�� 7� M��� � �)�)�*;�<��*�+�+r c � � U R 5 n[ R U R 5 n [ R
" U R
X!S95 nU$ ! [ R a" n[ R SU5 0 s SnA$ SnAff = f)z�
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_doc�!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 �B c � � U R 5 nUR SS5 nUR SS5 n[ U5 =( d [ U5 $ )z�
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)r7 �get�bool)r r5 �billing_products�marketplace_product_codess r �
is_likely_pro�#AWSAutoAttachInstance.is_likely_proI sM � � �!�!�#��!�:�:�&7��>��$*�J�J�/H�$�$O�!��$�%�H��.G�)H�Hr c �( � [ H n U R US9nXl O U R c' [ R " SR [ 5 S9eW$ ! [ a- n[ R SU[
USS5 5 S nAM� S nAff = f)N)�
ip_addressz*Could not reach AWS IMDS at http://%s: %s
�reason� z, )� addresses)
�IMDS_IP_ADDRESS�_get_imds_v2_token_headersr( � Exceptionr3 �warning�getattrr �AWSNoValidIMDS�join)r �addressr r6 s r r% �4AWSAutoAttachInstance._request_imds_v2_token_headersX s� � �&�G�
� �9�9�W�9�M�� $+� �� '� � � � #� �+�+� �)�)�O�4��
� ��� �
�� � �A���A�x��,�� ��
�s �A�
B�$"B � B)r � � c � � U R S:X a g U R ( a [ U R 0$ [ R " [ R U5 S[ [ 0SS9nUR S:X aL UR U l [ R R U R 5 [ U R 0$ UR S:X a SU l g [ R " UR UR S9e)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 rC r s r rH �0AWSAutoAttachInstance._get_imds_v2_token_headersk s� � � �?�?�h� &��
�_�_�(�$�/�/�:� :��<�<� � $� $�Z� 0��)�+@�A��
�� �=�=�C� � &�m�m�D�O� � "� "� -� -�d�o�o� >�(�$�/�/�:� :� �=�=�C� � &�D�O���+�+�� � �X�]�]�
�
r c � � g)N�aws� �r s r �
cloud_type� AWSAutoAttachInstance.cloud_type� � � �r c �> � [ R " [ 5 nSUSS :X a g [ R " [ 5 R 5 n[ R " [ 5 R 5 nSUSS s=:X a
USS :X a g gg! [ a Ntf = f)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_viable� AWSAutoAttachInstance.is_viable� s� � � � $� .� .�/J� K�O����!�,�,�� -� �#�#�$4�5�;�;�=���%�%�&8�9�?�?�A�
� �H�Q�q�M� 4�Z��!�_� 4�� 5��� !� �
� �s �#B �
B �B c � � g)� UnsupportedFr^ r_ s r �should_poll_for_pro_license�1AWSAutoAttachInstance.should_poll_for_pro_license� rb r �wait_for_changec �, � [ R " 5 e)N)r � InPlaceUpgradeNotSupportedError)r ru s r �is_pro_license_present�,AWSAutoAttachInstance.is_pro_license_present� s � ��8�8�:�:r )rV r( )�__name__�
__module__� __qualname__�__firstlineno__rV r( �strr �propertyr �retryr r r r r, r7 r= r@ r% rH r` ro rs rx �__static_attributes__r^ r r r
r
s � ��J��K��#� � � �Z�Z�
�-�-�K�H�,�d�3��8�n� ,� I� �,�
�Z�Z�
�-�-�K�H��T� � I��* � I�t� I� � I� �&
�Z�Z�
�-�-�I�F�
� G�
�0 ��C� � �� ��4� � ��"�T� �;�� ;�$� ;r r
) r0 �logging�typingr r �uaclientr r r r r �uaclient.cloudsr
�IMDS_IPV4_ADDRESS�IMDS_IPV6_ADDRESSrG rX r& r/ rg rk ri rZ rW rY � getLogger� replace_top_level_logger_namerz r3 r
r^ r r �<module>r� s� �� � � � C� C� 9�%� �%� �$�&7�8��0� � =��E�
� 4� �7� �3� � � �1� �+�n�<� �
����:�:�8�D�E��D;� 9� D;r