Server : Apache/2.4.52 (Ubuntu) System : Linux webserver 6.8.0-49-generic #49~22.04.1-Ubuntu SMP PREEMPT_DYNAMIC Wed Nov 6 17:42:15 UTC 2 x86_64 User : www-data ( 33) PHP Version : 8.1.2-1ubuntu2.21 Disable Function : NONE Directory : /lib/python3/dist-packages/uaclient/clouds/__pycache__/ |
o
��pfF � @ s� d dl Z d dlZd dlZd dlZd dlmZmZmZmZ d dl m
Z
m Z m Z m
Z
mZ d dlmZ e�e�e��ZdZdZdZdZd Zd
Zd d d
ddd�ZG dd� de�Z dS )� N)�Any�Dict�List�Optional)�
exceptions�http�secret_manager�system�util)�AutoAttachCloudInstancez�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�599959289349842382�2592866803419978320�2176054482269786025)�xenial�bionic�focal�jammy�noblec @ s� e Zd Zdd� Zeejejg d�d�de e
e f fdd���Z ede
fdd ��Z
edefd
d ��Zdee
fd d
�Zdefdd�Zdedefdd�ZdS )�UAAutoAttachGCPInstancec C s
d | _ d S )N��etag��self� r �5/usr/lib/python3/dist-packages/uaclient/clouds/gcp.py�__init__% s
z UAAutoAttachGCPInstance.__init__)g �?� r )� retry_sleeps�returnc C st t jtddidd�}|jdkrtj�|j� d|jiS |j� d�}|r1d|�
� v r1t j |j|d ��t j
|j|jd
��) N�Metadata-Flavor�Googler )�headers�timeout�� �
identityToken�error_descriptionzservice account)� status_code� error_msg)�code�body)r �readurl� TOKEN_URLr* r �secrets�
add_secretr+ � json_dict�get�lowerr �GCPServiceAccountError�CloudMetadataError)r �response�
error_descr r r � identity_doc, s
�
��z$UAAutoAttachGCPInstance.identity_docc C s dS )N�gcpr r r r r �
cloud_type? s z"UAAutoAttachGCPInstance.cloud_typec C s* t j�t�rt�t�}t|�� krdS dS )z$This machine is a viable GCPInstanceTF)�os�path�exists�DMI_PRODUCT_NAMEr � load_file�GCP_PRODUCT_NAME�strip)r � product_namer r r � is_viableC s
z!UAAutoAttachGCPInstance.is_viablec C sN | j d }t�|�d�d d �}t�|�d��}|�di ��di ��dg �S ) 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.
r& �.r z===zutf-8�google�compute_engine�
license_id)r7 �base64�urlsafe_b64decode�split�json�loads�decoder1 )r �token�identity�
identity_dictr r r �get_licenses_from_identityM s
�z2UAAutoAttachGCPInstance.get_licenses_from_identityc C s$ t �� j}|tvrt�d� dS dS )Nz0This series isn't supported for GCP auto-attach.FT)r �get_release_info�series� GCP_LICENSES�LOG�info)r rR r r r �should_poll_for_pro_license^ s
z3UAAutoAttachGCPInstance.should_poll_for_pro_license�wait_for_changec C s� t }|r|t7 }| jr|tj| jd�7 }tj|ddid�}|jdkr=dd� |jD �}|j �
d�| _t � � j
}t�
|�|v S t�|j� |jd krLt�� �t�� �)
Nr r! r" )r# r% c S s g | ]}|d �qS )�idr )�.0�licenser r r �
<listcomp>o s zBUAAutoAttachGCPInstance.is_pro_license_present.<locals>.<listcomp>r i� )� LICENSES_URL�WAIT_FOR_CHANGEr � LAST_ETAG�formatr r, r* � json_listr# r1 r rQ rR rS rT �errorr+ r �CancelProLicensePolling�DelayProLicensePolling)r rW �urlr5 � license_idsrR r r r �is_pro_license_presente s
z.UAAutoAttachGCPInstance.is_pro_license_presentN)�__name__�
__module__� __qualname__r �propertyr
�retryr r4 r �strr r7 r9 �boolrB r rP rV rf r r r r r $ s r ) rG rJ �loggingr: �typingr r r r �uaclientr r r r r
�uaclient.cloudsr � getLogger� replace_top_level_logger_namerg rT r- r\ r] r^ r= r? rS r r r r r �<module> s. ���