| Server IP : 104.21.4.30 / Your IP : 162.159.115.11 Web Server : Apache System : Linux keen-cori.18-142-40-148.plesk.page 5.15.0-1084-aws #91~20.04.1-Ubuntu SMP Fri May 2 06:59:36 UTC 2025 x86_64 User : simottodesign.com_2tntp341vs7 ( 10011) PHP Version : 8.3.31 Disable Function : opcache_get_status MySQL : OFF | cURL : ON | WGET : ON | Perl : ON | Python : ON | Sudo : ON | Pkexec : ON Directory : /usr/lib/python3/dist-packages/uaclient/__pycache__/ |
Upload File : |
U
2�d�( �
@ s� d dl Z d dlZd dlZd dlZd dlmZmZ d dlmZm Z m
Z
m Z m Z m
Z
d dlmZ d dlmZ d dlmZ d dlmZmZmZ d dlmZ d d lmZ d d
lmZmZm Z d d l m Z m Z m Z e�!e�"e#��Z$d Z%d
Z&e j'e(e)dd�dd�Z*d+e j'ejdd�dd�Z+dddddd�e j'e(e)e)e)e(eee( d�dd�Z,ddd�e j'ee( e)d �d d �Zd,e(eee- dd �d d!�Z.e j'd"�d#d$�Z/e j'e(d%�d&d'�Z0e j'ed(�d)d*�Z1dS )-� N)�List�Optional)�clouds�config�contract� entitlements�
exceptions� livepatch)�log)�messages)�status)�system�timer�util)�AutoAttachCloudInstance)�identity)�CLOUD_BUILD_INFO�DEFAULT_CONFIG_FILE�DEFAULT_LOG_PREFIX)�AttachmentData�attachment_data_file�timer_jobs_state_file)zua-timer.servicezua-timer.timerzua-auto-attach.pathzua-auto-attach.servicezua-reboot-cmds.servicezubuntu-advantage.service�
)�cfg�token� allow_enable�returnc C sz ddl m} t�| �}tjjtjjd�}z|j||d�}W nN t j
k
r� } z.t � � � t
�t|�� W 5 Q R X t �� �W 5 d}~X Y nX | j�|� tj�� |�di ��dt�| ��}| �d|� zt�| i | jj|� W n\ t j
t jfk
�r: } z4t�t |d �� t j | d
� || � |� � | �W 5 d} ~ X Y nX t �!� }
|
�rV| �d |
� t�t |d �� || � t"�#� dS ) a:
Common functionality to take a token and attach via contract backend
:raise UrlError: On unexpected connectivity issues to contract
server or inability to access identity doc from metadata service.
:raise ContractAPIError: On unexpected errors when talking to the contract
server.
r )�update_motd_messages)Ztz)Zcontract_tokenZ
attachment_dtNZmachineTokenInfoZ machineIdz
machine-id)� attached_at�r z instance-id)$Z uaclient.timer.update_messagingr r �UAContractClient�datetimeZnow�timezoneZutcZadd_contract_machiner ZUrlErrorr Zdisable_log_to_console�LOGZ exception�strZConnectivityErrorZmachine_token_file�writer
Zget_machine_id� cache_clear�getZ write_cacheZprocess_entitlements_deltar �UserFacingErrorr r � ua_statusr Zupdate_activity_tokenr Zget_instance_idr �start) r r r r �contract_clientr Znew_machine_token�eZ
machine_id�excZ current_iid� r. �2/usr/lib/python3/dist-packages/uaclient/actions.py�attach_with_token/ sL
�
� � r0 T)r �cloudr c C s0 t �| �}|j|d�}|d }t| ||d� dS )aS
:raise UrlError: On unexpected connectivity issues to contract
server or inability to access identity doc from metadata service.
:raise ContractAPIError: On unexpected errors when talking to the contract
server.
:raise NonAutoAttachImageError: If this cloud type does not have
auto-attach support.
)�instanceZ
contractToken)r r N)r r Z%get_contract_token_for_cloud_instancer0 )r r1 r r+ Z
tokenResponser r. r. r/ � auto_attachj s
�r3 F� )�
assume_yes�
allow_beta� access_only�variant�
extra_args)r �namer5 r6 r7 r8 r9 c C s, t j| ||d�}|| |||||d�}|�� S )z�
Constructs an entitlement based on the name provided. Passes kwargs onto
the entitlement constructor.
:raise EntitlementNotFoundError: If no entitlement with the given name is
found, then raises this error.
)r r: r8 )r5 r6 Z called_namer7 r9 )r Zentitlement_factory�enable) r r: r5 r6 r7 r8 r9 Zent_cls� entitlementr. r. r/ �enable_entitlement_by_name� s ��r= )�simulate_with_token�show_all)r r> r? c C s4 |rt j| ||d�\}}nt j| |d�}d}||fS )z6
Construct the current Pro status dictionary.
)r r r? )r r? r )r) Zsimulate_statusr )r r> r? r Zretr. r. r/ r � s � r )�filename� return_codesr c
C sf zt j| �� |d�\}}W n: tjk
rT } zt �d�|�t|�� W 5 d}~X Y nX t �||� dS )zCHelper which runs a command and writes output or error to filename.)Zrcsz{}-errorN)r
Zsubp�splitr ZProcessExecutionError�
write_file�formatr$ )�cmdr@ rA �out�_r, r. r. r/ � _write_command_output_to_file� s
(rH r c C s0 | j pt| j| j| jtjjtfdd� t j
D ��S )Nc s s, | ]$}t |tjj�r|jj|jd �V qdS ))r: N)�
issubclassr ZrepoZRepoEntitlementZrepo_list_file_tmplrD r: )�.0r< r. r. r/ � <genexpr>� s �z#_get_state_files.<locals>.<genexpr>) Zcfg_pathr �log_fileZtimer_log_fileZdaemon_log_filer Zua_file�pathr r ZENTITLEMENT_CLASSESr r. r. r/ �_get_state_files� s ���rN )r �
output_dirc
C s� t dd�|�� t dd�|�� t d�tj�d�|�� t dd�|�� t d �d
�d d � tD ���d
�|�� tD ]$}t d�|�d�||�ddgd� qrt| �}t�� r�t �
� dt � nt � � g}t
|�D ]p\}}z0t�t�|��}t�tj�|d�|��|� W q� tk
�r8 } zt�d|t|�� W 5 d}~X Y q�X q�|t�td � D ]�} tj�| ��rNzt�| �}W n@ tk
�r� } z t�d| t|�� W Y ��qNW 5 d}~X Y nX t�|�}t�� �r�t�| |� t�tj�|tj�| ��|� �qNdS )zG
Write all relevant Ubuntu Pro logs to the specified directory
zcloud-idz{}/cloud-id.txtzpro status --format jsonz{}/ua-status.jsonz {} statusz{}/livepatch-status.txtzsystemctl list-timers --allz{}/systemd-timers.txtzxjournalctl --boot=0 -o short-precise {} -u cloud-init-local.service -u cloud-init-config.service -u cloud-config.service� c S s g | ]}d |krd� |��qS )z.servicez-u {})rD )rJ �sr. r. r/ �
<listcomp>� s z collect_logs.<locals>.<listcomp>z{}/journalctl.txtzsystemctl status {}z {}/{}.txtr � )rA Nz
user{}.logz&Failed to collect user log file: %s
%s�*zFailed to load file: %s
%s) rH rD r Z
LIVEPATCH_CMD�join� UA_SERVICESrN r Zwe_are_currently_root�pro_logZget_all_user_log_files�USER_LOG_COLLECTED_LIMITZget_user_log_file� enumerateZredact_sensitive_logsr
Z load_filerC �osrM � Exceptionr# Zwarningr$ �globr �isfile�basename)
r rO �serviceZ state_filesZuser_log_filesZ log_file_idxrL Zcontentr, �fr. r. r/ � collect_logs� s~ � �
��� ��
�
���� �
�ra )r r c
C s� d }z t �� }W n� tjk
r� } zjt|tj�r@tjtjdd��t|tj �rXt�tj
��t|tj �r~tj tj
j|jd�dd��t�tj��W 5 d }~X Y nX |s�t�tj��|S )Nz auto-attach-cloud-type-error)Zmsg_code)�
cloud_typez(auto-attach-unsupported-cloud-type-error)r Zcloud_instance_factoryr ZCloudFactoryError�
isinstanceZCloudFactoryNoCloudErrorr( r Z UNABLE_TO_DETERMINE_CLOUD_TYPEZ CloudFactoryNonViableCloudErrorZUNSUPPORTED_AUTO_ATTACHZ!CloudFactoryUnsupportedCloudErrorZNonAutoAttachImageErrorZ"UNSUPPORTED_AUTO_ATTACH_CLOUD_TYPErD rb )r r2 r, r. r. r/ �get_cloud_instance! s4 � ����rd )T)N)2r! r\ ZloggingrZ �typingr r Zuaclientr r r r r r r
rW r r r) r
r r Zuaclient.cloudsr r Zuaclient.defaultsr r r Zuaclient.files.state_filesr r r Z getLoggerZ replace_top_level_logger_name�__name__r# rV rX ZUAConfigr$ �boolr0 r3 r= �intrH rN ra rd r. r. r. r/ �<module> sv � > �� �
� !�� �
� P�