File: //lib/python3/dist-packages/uaclient/__pycache__/system.cpython-310.pyc
o
S��cQB � @ 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�de$fdd��Z%e
dd�de"fd d ��Z&e
dd�de"fd d ��Z'e
dd�de"fd d!��Z(e
dd�dee"e"f fd"d#��Z)e
dd�d$e"de*fd%d&��Z+e
dd�de*fd'd(��Z,e
dd�d$e"de*fd)d*��Z-e
dd�d$e"de*fd+d,��Z.e
dd�de*fd-d.��Z/e
dd�d_d0e"de*fd1d2��Z0e
dd�d`d3ee" dee"e"f fd4d5��Z1e
dd�d$e"de!fd6d7��Z2d8e"dee" fd9d:�Z3 dad;eee" d<eee" de*fd=d>�Z4d?e"de*fd@dA�Z5dbdCe"dDe*de"fdEdF�Z6dcdCe"dHe#ddfdIdJ�Z7dcdCe"dKe"dHe#ddfdLdM�Z8dNe"ddfdOdP�Z9 Q dddRee" dSeee# dTe*dUee: dVeee"e"f dee"e"f f dWdX�Z; Q dedRee" dSeee# dTe*dUee: dYeee: dVeee"e"f dee"e"f fdZd[�Z<d\e"ddfd]d^�Z=dS )f� 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.csvz0(?P<release>\d+\.\d+) (LTS )?\((?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 } z t d�}|�� d } W n ty t�d� Y nw t�� j�� }t �
t |�}|d u rCt�t j
j|d�� t|| d d d d d d�S t|| t|�d��t|�d��t|�d��|�d �|�d
�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 �KERNEL_PARSE_ERROR�formatr �int�group)r �proc_version_signature_fullr � uname_match� r/ �1/usr/lib/python3/dist-packages/uaclient/system.py�get_kernel_info9 s> � ��
�r1 c C sd t dg�\} }| �� D ]}|�� �d�r%|�d�d �� }|r#| S nq tj}tj|j |j
d��)zused for livepatch�lscpuz
Architecture:�:r )�msg�msg_code) �subp�
splitlinesr% �
startswithr r �LSCPU_ARCH_PARSE_ERRORr �UserFacingErrorr4 �name)�out�_err�line�arch� error_msgr/ r/ r0 �get_lscpu_arch] s ��rA c C s t ddg�\} }| �� S )N�dpkgz--print-architecture)r6 r% )r< r= r/ r/ r0 �
get_dpkg_archn � rC c C s~ | j r| j �di ��d�}|r|S | �d�}tt|fD ]}tj�|�r0t|�� d�}|r0| S qt
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.
�machineTokenInfo� machineIdz
machine-id�
)�
machine_token�get� data_path�ETC_MACHINE_ID�DBUS_MACHINE_IDr" �path�existsr �rstrip�str�uuid�uuid4� write_cache)�cfg�cfg_machine_id�fallback_machine_id_filerM �content�
machine_idr/ r/ r0 �get_machine_idt s �
� rY c C s� t � } | �dd�dd�}| d }t�dd|�}||d< t�t|�}|s-td �| d |���|�� }|� |d
|d �
� d �� t � j |d
< 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.
�NAME�UNKNOWN�Linux)� distribution�type�VERSIONz\.\d LTSz LTS�versionz<Could not parse /etc/os-release VERSION: {} (modified to {})r$ �series)r$ ra r r? )�parse_os_releaserI r& �subr' �REGEX_OS_RELEASE_VERSION� RuntimeErrorr* � groupdict�update�lowerr1 r rC )�
os_release�
platform_infor` r' �
match_dictr/ r/ r0 �get_platform_info� s,
� �
��
rl ra c C � t ddg�\}}| |v S )N�/usr/bin/ubuntu-distro-infoz--supported-esm�r6 �ra r< r= r/ r/ r0 �is_lts� rD rq c C � t t� d �S �Nra )rq rl r/ r/ r/ r0 �is_current_series_lts� � rt c C rm )Nrn z --supportedro rp r/ r/ r0 � is_supported� rD rv c C s, t | �sdS tdd| dg�\}}t|�dkS )zCReturn True when Ubuntu series supports ESM and is actively in ESM.Frn z--seriesz-yeolr )rq r6 r+ rp r/ r/ r0 �
is_active_esm� s
� rw c C rr rs )rw rl r/ r/ r/ r0 � is_current_series_active_esm� ru rx �/run�run_pathc C s~ zt dg� W dS tjy Y nw z t g d�� W dS ttfy' Y nw dD ]}tj�| |�}tj�|�r< dS q*dS )z>Checks to see if this code running in a container of some sort�ischrootF)zsystemd-detect-virtz--quietz --containerT)�container_typezsystemd/container) r6 r �ProcessExecutionError�IOError�OSErrorr" rM �joinrN )rz �filenamerM r/ r/ r0 � is_container� s$
� � �r� � release_filec C sH | sd} i }t | ��� D ]}|�dd�\}}|r!|�� �d�||< q |S )Nz/etc/os-release�=r �")r r7 r r% )r� �datar> �key�valuer/ r/ r0 rb � s �rb c C s� zt t��� }W n
ty t�tj��w |D ]:}|�d�}|d | krR| dkr,d}nd|d v r6|d n|d }t t
j
� |d d �� � t
j
� |d �� � d
� S qt�tj
�| ���) N�,� �xenialz
2026-04-23�LTSr � � z%Y-%m-%d)r r )r �DISTRO_INFO_CSVr7 �FileNotFoundErrorr r: r �MISSING_DISTRO_INFO_FILEr r �datetime�strptime�date�"MISSING_SERIES_IN_DISTRO_INFO_FILEr* )ra �linesr> �valuesr r/ r/ r0 �get_distro_info� s& �
��
�r� �programc C sr t jj| v r t| �r | S dd� t j�dd��t j�D �}dd� |D �}|D ]}t j�|| �}t|�r6| S q%dS )z;Find whether the provided program is executable in our PATHc S s g | ]}|� d ��qS )r� )r% ��.0�pr/ r/ r0 �
<listcomp> s
�zwhich.<locals>.<listcomp>�PATH� c S s g | ]}t j�|��qS r/ )r" rM �abspathr� r/ r/ r0 r� s N) r" rM �sep�is_exe�environrI r �pathsepr� )r� �paths�normalized_pathsrM � program_pathr/ r/ r0 �which s ��r� �installed_pkgs�installed_pkgs_regexc C s� t j�t�sdS | du r|du rdS z ttt��d��}W n
ty' Y dS w | dur7t | �
|��dkr7dS |durO|D ]}|D ] }t � ||�rM dS qAq=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.
FNTrG r )
r" rM rN �REBOOT_FILE_CHECK_PATH�setr �REBOOT_PKGS_FILE_PATHr r� �len� intersectionr& �search)r� r� �reboot_required_pkgs�pkg_name� pkg_regexr/ r/ r0 �
should_reboot& s, � � ��r� rM c C s t j�| �o t �| t j�S �N)r" rM �isfile�access�X_OK)rM r/ r/ r0 r� \ s r� Tr� �decodec C sH t �d| � t| d�� }|�� }W d � n1 sw Y |�d�S )z!Read filename and decode content.zReading file: %s�rbN�utf-8)r �debug�open�readr� )r� r� �streamrW r/ r/ r0 r a s
�
r � �modec C s@ t �d| � tjtj�| �dd� t�| ��� t� | |� d S )NzCreating file: %sT��exist_ok)
r r� r" �makedirsrM �dirname�pathlib�Path�touch�chmod)r� r� r/ r/ r0 � create_filei s r� rW c
C s� d}z@t jt j�| �dd� tjddt j�| �d�}t�d| |j� |� |�
d�� |� � |� � t �
|j|� t �|j| � W dS tyZ } z |durTt �|j� |�d}~ww ) 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� rM r� �tempfile�NamedTemporaryFiler r� r; �write�encode�flush�closer� �renamer �unlink)r� rW r� �tmpf�er/ r/ r0 �
write_filep s( �� ��r� � file_pathc C s* t j�| �rt�d| � t �| � dS dS )z<Remove a file if it exists, logging a message about removal.zRemoving file: %sN)r" rM rN r r� r� )r� r/ r/ r0 �ensure_file_absent� s �r� F�args�rcs�capture�timeout�envc
C s dd� | D �}|r|� tj� |du rdg}t�d�| ��}ztj|tjtj|d�}|j |d�\}} W n% t
yX zt j ||j
|�d�| �d�d �� tyW t j |d
��w w |j
|vrnt j ||j
|�d�| �d�d ��|ryt�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� )r� �xr/ r/ r0 r� � s �z_subp.<locals>.<listcomp>Nr � )�stdout�stderrr� )r� r� )�cmd� exit_coder� r� )r� z Ran cmd: %s, rc: %s stderr: %s)rg r" r� r
�redact_sensitive_logsr� �
subprocess�Popen�PIPE� communicater r r} �
returncoder� �UnboundLocalErrorr r� )
r� r� r� r� r� �
bytes_args� redacted_cmd�procr< �errr/ r/ r0 �_subp� sP � � � ��
��r� � retry_sleepsc
C s� |dur|� � nd} zt| ||||d�\}}W ||fS tjyg } z?|r@t�t�t|��� d� |j
|j �} t� t�| �� |sC� dt
|� }
t�t�t|�|
�� t�|�d�� W Y d}~nd}~ww q )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.
NT)r� zStderr: {}
Stdout: {}z Retrying %d more times.r )�copyr� r r} r r� r
r� rP r* r� r� r! r� �time�sleep�pop) r� r� r� r� r� r� r< r� r� r4 � retry_msgr/ r/ r0 r6 � s& � ���r6 � folder_pathc C s( t j�| �rt�d| � t| � d S d S )NzRemoving folder: %s)r" rM rN r r� r )r� r/ r/ r0 �ensure_folder_absent� s �r� )ry r� )NN)T)r� )NFNN)NFNNN)>r� r r" r� r& r� r� r� rQ � functoolsr �shutilr �typingr r r r r r r
�uaclientr r r
r� r� rK rL r� rd r( r� r rP r+ r r1 rA rC rY rl �boolrq rt rv rw rx r� rb r� r� r� r� r r� r� r� �floatr� r6 r� r/ r/ r/ r0 �<module> s� $��
��#&
" �
�
�
�6 ��
����
�@��
���
��
�.