Your IP : 216.73.216.52
U
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#ee" d�dd�Z$G dd� dej%�Z&ee' d�dd�Z(e)e)e)e)e e'ee' f d�dd �Z*e)e)e)e)ee' d�d d �Z+edd �ee' d�d d!��Z,d-e)eee- dd"�d#d$�Z.d.ee) ee) eee- dd%�d&d'�Z/e)ee) d(�d)d*�Z0e'd�d+d,�Z1dS )/� 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 @ sD e Zd Zededdd�ededdd�gZee ee d�dd �Z
d
S ) �LivepatchPatchFixStatus�nameF�Name�ZrequiredZdict_key�patchedZPatched�r r c C s || _ || _d S �Nr )�selfr r � r �4/usr/lib/python3/dist-packages/uaclient/livepatch.py�__init__* s z LivepatchPatchFixStatus.__init__N) �__name__�
__module__� __qualname__r r r �fieldsr �str�boolr r r r r r $ s ��r c @ sL e Zd Zededdd�edee�ddd�gZee ee
e d�dd �Z d
S ) �LivepatchPatchStatus�stateF�Stater �fixesZFixes�r% r' c C s || _ || _d S r r( )r r% r' r r r r > s 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 @ sX e Zd Zededdd�ededdd�ededdd�gZee ee ee d �d
d �Z d S )
�LivepatchStatusStatus�kernelFZKernelr � livepatchZ Livepatch� supported� Supported�r* r+ r, c C s || _ || _|| _d S r r. )r r* r+ r, 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�gZeee d�dd�Z dS ) �LivepatchStatus�statusFZStatusr �r0 c C s
|| _ d S r r1 )r r0 r r r r m s zLivepatchStatus.__init__N)
r r r r r r) r! r r r r r r r r/ c s ��
�r/ )�returnc C s2 t � st�d� d S zt�tdddg�\} }W n6 tjk
rd t� � � t�
d� W 5 Q R X Y d S X zt � | �}W n< t j
k
r� t� � � t�
d�| �� W 5 Q R X Y d S X zt�|�}W n< tk
� r� t� � � t�
d�| �� W 5 Q R X Y d S X |jd k�st|j�dk �r(t�d � d S |jd
S ) Nz$canonical-livepatch is not installedr0 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 ZProcessExecutionErrorr �disable_log_to_console�warningr3 �loadsZJSONDecodeError�formatr/ Z from_dictr r0 �len)�out�_Z status_jsonZ status_rootr r r r0 t sF
�
��
��
r0 c @ s0 e Zd ZdZejZeeeeee d�dd�Z
dS )�UALivepatchClientZ
livepatch_url��version�flavor�arch�codenamer2 c
C s� ||||d�}| � � }z| jt||d�\}}W nP tk
r~ } z2t�� � t�d� t�| � W 5 Q R X W Y �dS d} ~ X Y nX t|t �s�t�d� dS t
|� dd��S )z�
:returns: True if supported
False if unsupported
None if API returns error or ambiguous response
)zkernel-versionZflavourZ architecturerF )� query_params�headersz4error while checking livepatch supported kernels APINz2livepatch api returned something that isn't a dictr- F) rH Z request_url�"LIVEPATCH_API_V1_KERNELS_SUPPORTED� Exceptionr r: r6 r; �
isinstance�dictr# �get)
r rC rD rE rF rG rH �resultZ_headers�er r r �is_kernel_supported� s2 ��
�
�z%UALivepatchClient.is_kernel_supportedN) r r r Zcfg_url_base_attrr ZUrlErrorZ
api_error_clsr" r r# rP r r r r rA � s �rA c C s2 t � } | d krd S | jdkr dS | jdkr.dS d S )Nr, TZ unsupportedF)r0 r, )Z lp_statusr r r �_on_supported_kernel_cli� s
rQ rB c C s� zt j�� }W n tk
r& d}Y nX |dk r�tj�tjj�tjdd� }t |j
|k|j | k|j |k|j
|k|j|kg�r�|jdkr�t�� � t�d� W 5 Q R X d|jfS dS )zOCheck local cache of kernel support
:return: (is_cache_valid, result)
N� )Zdaysz-livepatch kernel support cache has None valueT)FN)r �livepatch_support_cache�readrJ �datetime�now�timezone�utcZ timedelta�all� cached_atrC rD rE rF r, r r: r6 r; )rC rD rE rF Z
cache_dataZ one_week_agor r r �_on_supported_kernel_cache� s2
�
���
�
r[ c C sh t � j| |||d�}tj�tj| ||||tj�tjj �d�� |d krdt
� � � t �
d� W 5 Q R X |S )N)rC rD rE rF )rC rD rE rF r, rZ z3livepatch kernel support API response was ambiguous)rA rP r rS �writeZLivepatchSupportCacheDatarU rV rW rX r r: r6 r; )rC rD rE rF r, r r r �_on_supported_kernel_api� s, ���
�r] )�maxsizec C s� t � } | dk r t�d� | S t�� }|jdksB|jdksB|jdkrPt�d� dS t �
t� � �}t� � d }dj
|j|jd�}t||j||�\}}|r�t�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 supportZseriesz{major}.{minor})�major�minorz using livepatch support cachezusing livepatch support api)rQ r6 r7 r
Zget_kernel_inforD r_ r` r; r Zstandardize_arch_nameZ
get_dpkg_archZget_platform_infor= r[ r] )Zcli_saysZ kernel_inforE rF Zlp_api_kernel_verZis_cache_validZ
cache_saysr r r �on_supported_kernel sJ
���� � �
�ra )�
protocol_type� retry_sleepsr2 c C s( t � s
dS 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=�rc )r5 r
r8 r9 r= )rb rc r r r �unconfigure_livepatch_proxyD s �rf )�
http_proxy� https_proxyrc r2 c C sj ddl m} | s|r*t�tjj|jd�� | rHtj t
dd�| �g|d� |rftj t
dd�|�g|d� 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)�servicerd z
http-proxy={}re zhttps-proxy={}N) Zuaclient.entitlementsri �event�infor ZSETTING_SERVICE_PROXYr= �titler
r8 r9 )rg rh rc ri r r r �configure_livepatch_proxyX s" ����rn )�keyr2 c C s\ t �tdg�\}}t�d�| �|tj�}|r6|�d�nd}|rLt�dd|�}|rX|� � 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
rd z
^{}: (.*)$r4 Nz\"(.*)\"z\g<1>)
r
r8 r9 �re�searchr= � MULTILINE�group�sub�strip)ro r? r@ �match�valuer r r �get_config_option_value} s rx c C s t �t�d k S r )r
Zwhichr9 r r r r r5 � s r5 )N)NNN)2rU r3 r6 rp � functoolsr �typingr r r Zuaclientr r r r r
r Zuaclient.data_typesr r
r r r r Zuaclient.filesr ZHTTP_PROXY_OPTIONZHTTPS_PROXY_OPTIONr9 rI Zget_event_loggerrk r r$ r) r/ r0 ZUAServiceClientrA r# rQ r" r[ r] ra �floatrf rn rx r5 r r r r �<module> sf *+ � # � 0 �
� �
� %