�
��g( � � � S SK r S SKrS SKrS SKrS SKJrJrJrJr S SK J
r
J r J r J
r
Jr S SKJr \R" " \R$ " \5 5 rSrSrSrSrS rS
rS /S S
/SS/SS/SS/S.r " S S\5 r g)� N)�Any�Dict�List�Optional)�
exceptions�http�secret_manager�system�util)� PublicCloudAutoAttachInstancez�http://metadata/computeMetadata/v1/instance/service-accounts/default/identity?audience=contracts.canonical.com&format=full&licenses=TRUEzThttp://metadata.google.internal/computeMetadata/v1/instance/licenses/?recursive=truez&wait_for_change=truez&last_etag={etag}z /sys/class/dmi/id/product_namezGoogle Compute Engine�8045211386737108299�6022427724719891830�7427070211152946628�599959289349842382�7818678586103571931�2592866803419978320�160084055690847662�2176054482269786025�4168323874790319525)�xenial�bionic�focal�jammy�noblec �� � \ rS rSrS r\\R " \R / SQS9S\
\ \ 4 4S j5 5 r
\S\ 4S j5 r\S\4S j5 rS\\ 4S jrS\4S
jrS \S\4S jrS
rg)�GCPAutoAttachInstance�$ c � � S U l g )N��etag��selfs �5/usr/lib/python3/dist-packages/uaclient/clouds/gcp.py�__init__� GCPAutoAttachInstance.__init__% s � � �� � )g �?� r' )� retry_sleeps�returnc �� � [ R " [ SS0SS9nUR S:X a7 [ R
R
UR 5 SUR 0$ UR R S5 nU( a3 SUR 5 ; a [ R " UR US 9e[ R " UR UR S
9e) N�Metadata-Flavor�Googler' )�headers�timeout�� �
identityToken�error_descriptionzservice account)� status_code� error_msg)�code�body)r �readurl� TOKEN_URLr4 r �secrets�
add_secretr5 � json_dict�get�lowerr �GCPServiceAccountError�CloudMetadataError)r" �response�
error_descs r# � identity_doc�"GCPAutoAttachInstance.identity_doc, s� � � �<�<� � 1�8� <�a�
�� �=�=�C� � � "� "� -� -�h�m�m� <�#�X�]�]�3� 3� �'�'�+�+�,?�@�
� �+�z�/?�/?�/A�A� �3�3� $�M�M�Z��
� �+�+�� � �X�]�]�
�
r& c � � g)N�gcp� r! s r# �
cloud_type� GCPAutoAttachInstance.cloud_type? s � �r& c � � [ R R [ 5 ( a3 [ R
" [ 5 n[ UR 5 :X a gg)z$This machine is a viable GCPInstanceTF)�os�path�exists�DMI_PRODUCT_NAMEr
� load_file�GCP_PRODUCT_NAME�strip)r" � product_names r# � is_viable� GCPAutoAttachInstance.is_viableC s@ � � �7�7�>�>�*� +� +�!�+�+�,<�=�L� �<�#5�#5�#7�7��r& c �$ � U R S n[ R " UR S5 S S- 5 n[ R
" UR
S5 5 nUR S0 5 R S0 5 R S/ 5 $ ) z�Get a list of licenses from the GCP metadata.
Instance identity token (jwt) carries a list of licenses
associated with the instance itself.
Returns an empty list if licenses are not present in the metadata.
r0 �.r' z===zutf-8�google�compute_engine�
license_id)rA �base64�urlsafe_b64decode�split�json�loads�decoder; )r" �token�identity�
identity_dicts r# �get_licenses_from_identity�0GCPAutoAttachInstance.get_licenses_from_identityM s| � � �!�!�/�2���+�+�E�K�K��,<�Q�,?�%�,G�H�� �
�
�8�?�?�7�#;�<�
� � � �h�� +�
�S�!�2�
&�
�S� �r�
"�
r& c � � [ R " 5 R nU[ ; a [ R S5 gg)Nz0This series isn't supported for GCP auto-attach.FT)r
�get_release_info�series� GCP_LICENSES�LOG�info)r" re s r# �should_poll_for_pro_license�1GCPAutoAttachInstance.should_poll_for_pro_license^ s1 � ��(�(�*�1�1�� � � %� �H�H�G� H��r& �wait_for_changec � � [ nU( a: U[ -
nU R ( a U[ R U R S9-
n[
R " USS0S9nUR S:X a� UR Vs/ s H oDS PM nnUR R S5 U l [ R " 5 R n[ R U/ 5 H
nXu; d M
g g [ R! UR" 5 UR S
:X a [$ R& " 5 e[$ R( " 5 es snf ) Nr r+ r, )r- r/ �idr TFi� )� LICENSES_URL�WAIT_FOR_CHANGEr � LAST_ETAG�formatr r6 r4 � json_listr- r; r
rd re rf rg �errorr5 r �CancelProLicensePolling�DelayProLicensePolling)r" rk �urlr? �license� license_idsre �lics r# �is_pro_license_present�,GCPAutoAttachInstance.is_pro_license_presente s � ��� � �?� "�C��y�y��y�'�'�T�Y�Y�'�7�7���<�<��.?��-J�K�� �=�=�C� �8@�8J�8J�K�8J�W�4�=�8J�K�K� �(�(�,�,�V�4�D�I��,�,�.�5�5�F�#�'�'���3���%� � 4� � � � �(�-�-� � �=�=�C� � �4�4�6� 6� �3�3�5� 5�� Ls �=E
r N)�__name__�
__module__� __qualname__�__firstlineno__r$ �propertyr �retryr r> r �strr rA rF �boolrQ r ra ri rz �__static_attributes__rE r& r# r r $ s� � �� � �Z�Z�
�-�-�K�H�
�d�3��8�n�
� I� �
�" ��C� � �� ��4� � ��
�D��I�
�"�T� �6�� 6�$� 6r& r ) rX r[ �loggingrI �typingr r r r �uaclientr r r r
r �uaclient.cloudsr � getLogger� replace_top_level_logger_namer| rg r7 rn ro rp rL rN rf r rE r& r# �<module>r� s� ��
� � � � ,� ,� C� C� 9�
����:�:�8�D�E�� �
� �
� *�� � �3� �*� � %�%�$�&;�<�"�$9�
:�#�%9�
:�#�%:�
;� � �W6� 9� W6r&