Your IP : 216.73.216.31
U
8�-d`D � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl m
Z
d dl m Z d dl
mZmZmZmZmZmZmZ d dlmZmZmZ dZdZdZd Z d
Z d Z d Z ed
de j fde j fg�Z!edde"fdee" fdee# fdee# fdee# fdee" fdee" fg�Z$e
dd�e$d�dd��Z%e
dd�e"d�d d ��Z&e
dd�e"d�d d ��Z'e
dd�e"d�d d!��Z(e
dd�ee"e"f d�d"d#��Z)e
dd�e"e*d$�d%d&��Z+e
dd�e*d�d'd(��Z,e
dd�e"e*d$�d)d*��Z-e
dd�e"e*d$�d+d,��Z.e
dd�e*d�d-d.��Z/e
dd�d\e"e*d0�d1d2��Z0e
dd�d]ee" ee"e"f d3�d4d5��Z1e
dd�e"e!d$�d6d7��Z2e"ee" d8�d9d:�Z3d^eee" eee" e*d;�d<d=�Z4e"e*d>�d?d@�Z5d_e"e*e"dB�dCdD�Z6d`e"e#ddF�dGdH�Z7dae"e"e#ddI�dJdK�Z8e"ddL�dMdN�Z9dbee" eee# e*ee: eee"e"f ee"e"f dP�dQdR�Z;dcee" eee# e*ee: eee: eee"e"f ee"e"f dS�dTdU�Z<e"ddV�dWdX�Z=e"e*dY�dZd[�Z>dS )d� N)� lru_cache)�rmtree)�Dict�List�
NamedTuple�Optional�Sequence�Set�Tuple)�
exceptions�messages�utilz/var/run/reboot-requiredz /var/run/reboot-required.pkgsz/etc/machine-idz/var/lib/dbus/machine-idz!/usr/share/distro-info/ubuntu.csvz5(?P<release>\d+\.\d+) (LTS\s*)?(\((?P<series>\w+))?.*zd^(?P<major>[\d]+)[.-](?P<minor>[\d]+)[.-](?P<patch>[\d]+)-(?P<abi>[\d]+)-(?P<flavor>[A-Za-z0-9_-]+)$�
DistroInfo�eol�eol_esm�
KernelInfo�
uname_release� proc_version_signature_version�major�minor�patch�abi�flavor)�maxsize)�returnc
C s� d } zt d�}|�� d } W n tk
r: t�d� Y nX t�� j�� }t �
t |�}|d kr�t�t j
j|d�� t|| d d d d d d�S t|| t|�d��t|�d��t|�d��|�d �|�d
�d�S d S ) Nz/proc/version_signature� z*failed to process /proc/version_signature.)�kernel)r r r r r r r r r r r r )� load_file�split� Exception�logging�warning�os�uname�release�strip�re�match�RE_KERNEL_UNAMEr ZKERNEL_PARSE_ERROR�formatr �int�group)r Zproc_version_signature_fullr Z uname_match� r, �1/usr/lib/python3/dist-packages/uaclient/system.py�get_kernel_info; s< ��
�r. c C s t ddg�\} }| �� S )NZdpkgz--print-architecture)�subpr% )�out�_errr, r, r- �
get_dpkg_arch_ s r2 c C s6 zt dg�\} }| �� W S tjk
r0 Y dS X d S )N�systemd-detect-virt� )r/ r% r �ProcessExecutionError)r0 �_r, r, r- �
get_virt_typee s
r7 c C s~ | j r"| j �di ��d�}|r"|S | �d�}tt|fD ]*}tj�|�r6t|�� d�}|r6| S q6t
t � � �}| �
d|� |S )z�
Get system's unique machine-id or create our own in data_dir.
We first check for the machine-id in machine-token.json before
looking at the system file.
ZmachineTokenInfoZ machineIdz
machine-id�
)Z
machine_token�getZ data_path�ETC_MACHINE_ID�DBUS_MACHINE_IDr" �path�existsr �rstrip�str�uuidZuuid4Z write_cache)ZcfgZcfg_machine_idZfallback_machine_id_filer< �contentZ
machine_idr, r, r- �get_machine_idn s �
rB c C s� t � } | �dd�dd�}t�dd| �dd��}||d < | �d
d�}| �d d�}|rV|s�t�t|�}|s|tj| �dd�|d ��|�� }|p�|�d
d�}|s�tj |d��|p�|�dd�}|�
||� � t � j
t� t� d�� |S )z�
Returns a dict of platform information.
N.B. This dict is sent to the contract server, which requires the
distribution, type and release keys.
�NAMEZUNKNOWNZLinux)Z distribution�typez\.\d LTSz LTSZVERSIONr4 �versionZVERSION_CODENAMEZ
VERSION_ID)Zorig_verZmod_ver�series)rE r$ )r$ rF r ZarchZvirt)�parse_os_releaser9 r& �subr' �REGEX_OS_RELEASE_VERSIONr ZParsingErrorOnOSReleaseFile� groupdictZ MissingSeriesOnOSReleaseFile�update�lowerr. r r2 r7 )Z
os_releaseZ
platform_inforE rF r$ r' Z
match_dictr, r, r- �get_platform_info� s:
�
� ��
rM )rF r c C s t ddg�\}}| |kS )N�/usr/bin/ubuntu-distro-infoz--supported-esm�r/ �rF r0 r1 r, r, r- �is_lts� s rQ c C s t t� d �S �NrF )rQ rM r, r, r, r- �is_current_series_lts� s rS c C s t ddg�\}}| |kS )NrN z --supportedrO rP r, r, r- � is_supported� s rT c C s, t | �s dS tdd| dg�\}}t|�dkS )zCReturn True when Ubuntu series supports ESM and is actively in ESM.FrN z--seriesz-yeolr )rQ r/ r* rP r, r, r- �
is_active_esm� s
�rU c C s t t� d �S rR )rU rM r, r, r, r- � is_current_series_active_esm� s rV �/run)�run_pathr c C s� zt dg� W dS tjk
r& Y nX zt dddg� W dS ttfk
rT Y nX dD ]$}tj�| |�}tj�|�rZ dS qZdS )z>Checks to see if this code running in a container of some sortZischrootFr3 z--quietz --containerT)Zcontainer_typezsystemd/container) r/ r r5 �IOError�OSErrorr" r< �joinr= )rX �filenamer< r, r, r- � is_container� s
r] )� release_filer c C sH | sd} i }t | ��� D ]*}|�dd�\}}|r|�� �d�||< q|S )Nz/etc/os-release�=r �")r �
splitlinesr r% )r^ �data�line�key�valuer, r, r- rG � s rG c C s� zt t��� }W n tk
r0 t�tj��Y nX |D ]t}|�d�}|d | kr6| dkr^d}n d|d krr|d n|d }t t
j
� |d d �� � t
j
� |d �� � d
� S q6t�tj
�| ���d S ) N�,� Zxenialz
2026-04-23ZLTSr � � z%Y-%m-%d)r r )r �DISTRO_INFO_CSVra �FileNotFoundErrorr ZUserFacingErrorr ZMISSING_DISTRO_INFO_FILEr r �datetime�strptime�dateZ"MISSING_SERIES_IN_DISTRO_INFO_FILEr) )rF �linesrc �valuesr r, r, r- �get_distro_info� s"
�
�rq )�programr c C sr t jj| krt| �r| S dd� t j�dd��t j�D �}dd� |D �}|D ]"}t j�|| �}t|�rJ| S qJdS )z;Find whether the provided program is executable in our PATHc S s g | ]}|� d ��qS )r` )r% ��.0�pr, r, r- �
<listcomp> s zwhich.<locals>.<listcomp>�PATHr4 c S s g | ]}t j�|��qS r, )r" r< �abspathrs r, r, r- rv s N) r" r< �sep�is_exe�environr9 r �pathsepr[ )rr �pathsZnormalized_pathsr< Z program_pathr, r, r- �which s �
r~ )�installed_pkgs�installed_pkgs_regexr c C s� t j�t�sdS | dkr$|dkr$dS zttt��d��}W n tk
rP Y dS X | dk rpt | �
|��dkrpdS |dk r�|D ]"}|D ]}t � ||�r� dS q�q|dS )a� Check if the system needs to be rebooted.
:param installed_pkgs: If provided, verify if the any packages in
the list are present on /var/run/reboot-required.pkgs. If that
param is provided, we will only return true if we have the
reboot-required marker file and any package in reboot-required.pkgs
file. When both installed_pkgs and installed_pkgs_regex are
provided, they act as an OR, so only one of the two lists must have
a match to return True.
:param installed_pkgs_regex: If provided, verify if the any regex in
the list matches any line in /var/run/reboot-required.pkgs. If that
param is provided, we will only return true if we have the
reboot-required marker file and any match in reboot-required.pkgs
file. When both installed_pkgs and installed_pkgs_regex are
provided, they act as an OR, so only one of the two lists must have
a match to return True.
FNTr8 r )
r" r< r= �REBOOT_FILE_CHECK_PATH�setr �REBOOT_PKGS_FILE_PATHr rk �len� intersectionr& �search)r r� Zreboot_required_pkgs�pkg_nameZ pkg_regexr, r, r- �
should_reboot' s& � r� )r< r c C s t j�| �ot �| t j�S )N)r" r< �isfile�access�X_OK)r< r, r, r- rz ] s rz T)r\ �decoder c C s4 t �d| � t| d��}|�� }W 5 Q R X |�d�S )z!Read filename and decode content.zReading file: %s�rb�utf-8)r �debug�open�readr� )r\ r� �streamrA r, r, r- r b s r � )r\ �moder c C s@ t �d| � tjtj�| �dd� t�| ��� t� | |� d S )NzCreating file: %sT��exist_ok)
r r� r" �makedirsr< �dirname�pathlib�PathZtouch�chmod)r\ r� r, r, r- � create_filej s r� )r\ rA r� r c
C s� d}z~t jt j�| �dd� tjddt j�| �d�}t�d| |j� |� |�
d�� |� � |� � t �
|j|� t �|j| � W n: tk
r� } z |dk r�t �|j� |�W 5 d}~X Y nX dS ) z�Write content to the provided filename encoding it if necessary.
@param filename: The full path of the file to write.
@param content: The content to write to the file.
@param mode: The filesystem mode to set on the file.
NTr� �wbF)r� �delete�dirz*Writing file %s atomically via tempfile %sr� )r" r� r< r� �tempfileZNamedTemporaryFiler r� �name�write�encode�flush�closer� �renamer �unlink)r\ rA r� Ztmpf�er, r, r- �
write_fileq s,
� � r� )� file_pathr c C s& t j�| �r"t�d| � t �| � dS )z<Remove a file if it exists, logging a message about removal.zRemoving file: %sN)r" r< r= r r� r� )r� r, r, r- �ensure_file_absent� s r� F)�args�rcs�capture�timeout�envr c
C s dd� | D �}|r |� tj� |dkr,dg}t�d�| ��}z*tj|tjtj|d�}|j |d�\}} W nZ t
k
r� z$t j ||j
|�d�| �d�d ��W n tk
r� t j |d
��Y nX Y nX |j
|kr�t j ||j
|�d�| �d�d ��|�rt�d ||j
| � |�d�| �d�fS ) a� Run a command and return a tuple of decoded stdout, stderr.
@param args: A list of arguments to feed to subprocess.Popen
@param rcs: A list of allowed return_codes. If returncode not in rcs
raise a ProcessExecutionError.
@param capture: Boolean set True to log the command and response.
@param timeout: Optional float indicating number of seconds to wait for
subp to return.
@param env: Optional dictionary of environment variable to pass to Popen.
@return: Tuple of utf-8 decoded stdout, stderr
@raises ProcessExecutionError on invalid command or returncode not in rcs.
@raises subprocess.TimeoutError when timeout specified and the command
exceeds that number of seconds.
c S s$ g | ] }t |t�r|n|�d ��qS )r� )�
isinstance�bytesr� )rt �xr, r, r- rv � s z_subp.<locals>.<listcomp>Nr � )�stdout�stderrr� )r� r� )�cmdZ exit_coder� r� )r� z Ran cmd: %s, rc: %s stderr: %s)rK r" r{ r
Zredact_sensitive_logsr[ �
subprocess�Popen�PIPEZ communicaterZ r r5 �
returncoder� �UnboundLocalErrorr r� )
r� r� r� r� r� Z
bytes_argsZ redacted_cmd�procr0 �errr, r, r- �_subp� sR � ��
��r� )r� r� r� r� � retry_sleepsr� r c
C s� |dk r|� � nd}z t| ||||d�\}}W q�W q tjk
r� } zd|rrt�t|�� d�|j|j �} t�
| � |sx� dt |� }
t�t|�|
� t �
|�d�� W 5 d}~X Y qX q||fS )a� Run a command and return a tuple of decoded stdout, stderr.
@param subp: A list of arguments to feed to subprocess.Popen
@param rcs: A list of allowed return_codes. If returncode not in rcs
raise a ProcessExecutionError.
@param capture: Boolean set True to log the command and response.
@param timeout: Optional float indicating number of seconds to wait for a
subp call to return.
@param retry_sleeps: Optional list of sleep lengths to apply between
retries. Specifying a list of [0.5, 1] instructs subp to retry twice
on failure; sleeping half a second before the first retry and 1 second
before the next retry.
@param env: Optional dictionary of environment variables to provide to
subp.
@return: Tuple of utf-8 decoded stdout, stderr
@raises ProcessExecutionError on invalid command or returncode not in rcs.
@raises subprocess.TimeoutError when timeout specified and the command
exceeds that number of seconds.
N)r� zStderr: {}
Stdout: {}z Retrying %d more times.r )�copyr� r r5 r r� r? r) r� r� r! r� �time�sleep�pop) r� r� r� r� r� r� r0 r� r� �msgZ retry_msgr, r, r- r/ � s
$r/ )� folder_pathr c C s$ t j�| �r t�d| � t| � d S )NzRemoving folder: %s)r" r< r= r r� r )r� r, r, r- �ensure_folder_absent� s r� )�job_namer c
C sV zt dd| g�\}}W n* tjk
r@ } z
|j}W 5 d}~X Y nX |sJdS |�� dkS )aZ
Get if the systemd job is active in the system. Note that any status
different from "active" will make this function return False.
Additionally, if the system doesn't exist we will also return False
here.
@param job_name: Name of the systemd job to look at
@return: A Boolean specifying if the job is active or not
Z systemctlz is-activeNFZactive)r/ r r5 r� r% )r� r0 r6 r� r, r, r- �get_systemd_job_state s r� )rW )N)NN)T)r� )r� )NFNN)NFNNN)?rl r r" r� r& r� r� r� r@ � functoolsr Zshutilr �typingr r r r r r r
Zuaclientr r r
r� r� r: r; rj rI r( rn r r? r* r r. r2 r7 rB rM �boolrQ rS rT rU rV r] rG rq r~ r� rz r r� r� r� �floatr� r/ r� r� r, r, r, r- �<module> s� $�� �
��#-
�
� 6 �
� @ �
� .