File: //lib/python3/dist-packages/uaclient/__pycache__/livepatch.cpython-310.pyc
o
8�-d , �
@ s� d dl Z d dlZd dlZd dlZd dlmZ d dlmZmZm Z d dl
m Z m Z m
Z
mZmZmZ d dlmZmZmZmZmZmZ d dlmZ dZdZd Z d
Z e � � Z G d d � d e�Z G d
d� de�Z!G dd� de�Z"G dd� de�Z#dee" fdd�Z$G dd� dej%�Z&dee' fdd�Z(de)de)d e)d e)de e'ee' f f
d d �Z*de)de)d e)d e)dee' f
d d!�Z+edd"�dee' fd#d$��Z, d2d%e)d&eee- ddfd'd(�Z. d3d)ee) d*ee) d&eee- ddfd+d,�Z/d-e)dee) fd.d/�Z0de'fd0d1�Z1dS )4� N)� lru_cache)�List�Optional�Tuple)� event_logger�
exceptions�messages�
serviceclient�system�util)�
BoolDataValue�
DataObject�Field�IncorrectTypeError�StringDataValue� data_list)� state_filesz
http-proxyz https-proxyz /snap/bin/canonical-livepatchz/v1/api/kernels/supportedc @ sF e Zd Zededdd�ededdd�gZdee dee fdd�Z
d S )
�LivepatchPatchFixStatus�nameF�Name��required�dict_key�patched�Patchedc C � || _ || _d S �N)r r )�selfr r � r �4/usr/lib/python3/dist-packages/uaclient/livepatch.py�__init__* �
z LivepatchPatchFixStatus.__init__N) �__name__�
__module__� __qualname__r r r �fieldsr �str�boolr r r r r r $ s ���r c @ sN e Zd Zededdd�edee�ddd�gZdee dee
e fdd�Z d S )
�LivepatchPatchStatus�stateF�Stater �fixes�Fixesc C r r )r) r+ )r r) r+ r r r r > r! z LivepatchPatchStatus.__init__N) r" r# r$ r r r r r% r r& r r r r r r r( 3 s ��
�
�r( c @ s\ e Zd Zededdd�ededdd�ededdd�gZdee dee dee fd d
�Z d S ) �LivepatchStatusStatus�kernelF�Kernelr � livepatch� Livepatch� supported� Supportedc C s || _ || _|| _d S r )r. r0 r2 )r r. r0 r2 r r r r X s
z LivepatchStatusStatus.__init__N)
r" r# r$ r r r( r% r r& r r r r r r- G s, ������r- c @ s8 e Zd Zedee�ddd�gZdeee fdd�Z dS )�LivepatchStatus�statusF�Statusr c C s
|| _ d S r )r5 )r r5 r r r r m s
zLivepatchStatus.__init__N)
r" r# r$ r r r- r% r r r r r r r r4 c s ��
�r4 �returnc C sn t � s
t�d� d S z
t�tdddg�\} }W n& tjy= t� � � t�
d� W d � Y d S 1 s5w Y Y d S w zt � | �}W n) t j
yn t� � � t�
d�| �� W d � Y d S 1 sfw Y Y d S w zt�|�}W n( ty� t� � � t�
d�| �� W d � Y d S 1 s�w Y Y d S w |jd u s�t|j�dk r�t�d � d S |jd
S ) Nz$canonical-livepatch is not installedr5 z--format�jsonz7canonical-livepatch returned error when checking statusz4canonical-livepatch status returned invalid json: {}z<canonical-livepatch status returned unexpected structure: {}� z!canonical-livepatch has no statusr )�is_livepatch_installed�logging�debugr
�subp�
LIVEPATCH_CMDr �ProcessExecutionErrorr �disable_log_to_console�warningr8 �loads�JSONDecodeError�formatr4 � from_dictr r5 �len)�out�_� status_json� status_rootr r r r5 t s\
�
���
��
���
�
���
r5 c @ s8 e Zd ZdZejZdededededee f
dd�Z
d S )
�UALivepatchClient�
livepatch_url�version�flavor�arch�codenamer7 c
C s� ||||d�}| � � }z | jt||d�\}}W n1 tyH } z%t�� � t�d� t�| � W d � n1 s8w Y W Y d} ~ dS d} ~ ww t|t �sUt�d� dS t
|� dd��S )z�
:returns: True if supported
False if unsupported
None if API returns error or ambiguous response
)zkernel-version�flavour� architecturerP )� query_params�headersz4error while checking livepatch supported kernels APINz2livepatch api returned something that isn't a dictr3 F) rT � request_url�"LIVEPATCH_API_V1_KERNELS_SUPPORTED� Exceptionr r@ r; rA �
isinstance�dictr' �get)
r rM rN rO rP rS rT �result�_headers�er r r �is_kernel_supported� s8 ��
� ���
�z%UALivepatchClient.is_kernel_supportedN) r" r# r$ �cfg_url_base_attrr �UrlError�
api_error_clsr& r r' r^ r r r r rK � s �����rK c C s2 t � } | d u r d S | jdkrdS | jdkrdS d S )Nr2 T� unsupportedF)r5 r2 )� lp_statusr r r �_on_supported_kernel_cli� s
rd rM rN rO rP c C s� zt j�� }W n ty d}Y nw |dur`tj�tjj�tjdd� }t |j
|k|j | k|j |k|j
|k|j|kg�r`|jdu r[t�� �
t�d� W d � n1 sVw Y d|jfS dS )zOCheck local cache of kernel support
:return: (is_cache_valid, result)
N� )�daysz-livepatch kernel support cache has None valueT)FN)r �livepatch_support_cache�readrW �datetime�now�timezone�utc� timedelta�all� cached_atrM rN rO rP r2 r r@ r; rA )rM rN rO rP �
cache_data� one_week_agor r r �_on_supported_kernel_cache� s6 ��
���
� �
rr c C s~ t � j| |||d�}tj�tj| ||||tj�tjj �d�� |d u r=t
� � � t �
d� W d � |S 1 s8w Y |S )N)rM rN rO rP )rM rN rO rP r2 ro z3livepatch kernel support API response was ambiguous)rK r^ r rg �write�LivepatchSupportCacheDatari rj rk rl r r@ r; rA )rM rN rO rP r2 r r r �_on_supported_kernel_api� s2 ���
�
��ru )�maxsizec C s� t � } | durt�d� | S t�� }|jdu s!|jdu s!|jdu r(t�d� dS t �
t� � �}t� � d }dj
|j|jd�}t||j||�\}}|rQt�d� |S t�d� t||j||�S ) z�
Checks CLI, local cache, and API in that order for kernel support
If all checks fail to return an authoritative answer, we return None
Nz using livepatch cli for supportzHunable to determine enough kernel information to check livepatch support�seriesz{major}.{minor})�major�minorz using livepatch support cachezusing livepatch support api)rd r; r< r
�get_kernel_inforN rx ry rA r �standardize_arch_name�
get_dpkg_arch�get_platform_inforD rr ru )�cli_says� kernel_inforO rP �lp_api_kernel_ver�is_cache_valid�
cache_saysr r r �on_supported_kernel s<
���� �
�
�r� �
protocol_type� retry_sleepsc C s( t � sdS tjtdd�| �g|d� dS )a�
Unset livepatch configuration settings for http and https proxies.
:param protocol_type: String either http or https
:param retry_sleeps: Optional list of sleep lengths to apply between
retries. Specifying a list of [0.5, 1] tells subp to retry twice
on failure; sleeping half a second before the first retry and 1 second
before the second retry.
N�configz {}-proxy=�r� )r: r
r= r> rD )r� r� r r r �unconfigure_livepatch_proxyD s
�r� �
http_proxy� https_proxyc C sn ddl m} | s
|rt�tjj|jd�� | r$tj t
dd�| �g|d� |r5tj t
dd�|�g|d� dS dS ) a�
Configure livepatch to use http and https proxies.
:param http_proxy: http proxy to be used by livepatch. If None, it will
not be configured
:param https_proxy: https proxy to be used by livepatch. If None, it will
not be configured
:@param retry_sleeps: Optional list of sleep lengths to apply between
snap calls
r )�LivepatchEntitlement)�servicer� z
http-proxy={}r� zhttps-proxy={}N) �uaclient.entitlementsr� �event�infor �SETTING_SERVICE_PROXYrD �titler
r= r> )r� r� r� r� r r r �configure_livepatch_proxyX s$ ���
��r� �keyc C s\ t �tdg�\}}t�d�| �|tj�}|r|�d�nd}|r&t�dd|�}|r,|� � S dS )z�
Gets the config value from livepatch.
:param key: can be any valid livepatch config option
:return: the value of the livepatch config option, or None if not set
r� z
^{}: (.*)$r9 Nz\"(.*)\"z\g<1>)
r
r= r> �re�searchrD � MULTILINE�group�sub�strip)r� rG rH �match�valuer r r �get_config_option_value} s r� c C s t �t�d uS r )r
�whichr> r r r r r: � s r: r )NNN)2ri r8 r; r� � functoolsr �typingr r r �uaclientr r r r r
r �uaclient.data_typesr r
r r r r �uaclient.filesr �HTTP_PROXY_OPTION�HTTPS_PROXY_OPTIONr> rV �get_event_loggerr� r r( r- r4 r5 �UAServiceClientrK r' rd r& rr ru r� �floatr� r� r� r: r r r r �<module> s� *+ ����
�"����
� 0��
�
����
�
�%