Ke 2 @sddlZddlZddlZddlZddlZddlmZddlmZm Z m Z ddl m Z m Z mZmZmZddlmZmZmZmZmZmZddlmZddlmZdZd Zd Zd Ze j Z!ej"ej#e$Z%ej&Gd d d ej'Z(GdddeZ)GdddeZ*GdddeZ+GdddeZ,e e+dddZ-e e.e e(dddZ/Gdddej0Z1e e(dddZ2e.e.e.e.e e3e e3fd d!d"Z4e.e.e.e.e eje e(d#d$d%Z5ed&de(dd'd(Z6de.e ee7dd)d*d+Z8ddde e.e e.e ee7dd,d-d.Z9e.e e.d/d0d1Z:e3dd2d3Z;dS)4N) lru_cache)ListOptionalTuple) event_logger exceptionsmessagessystemutil) BoolDataValue DataObjectFieldIncorrectTypeErrorStringDataValue data_list) state_files) serviceclientz http-proxyz https-proxyz/snap/bin/canonical-livepatchz/v1/api/kernels/supportedc@s=eZdZeZeZeZeZeZdS)LivepatchSupportN) __name__ __module__ __qualname__object SUPPORTEDKERNEL_UPGRADE_REQUIRED KERNEL_EOL UNSUPPORTEDUNKNOWNrr4/usr/lib/python3/dist-packages/uaclient/livepatch.pyr s     rc@sfeZdZededdddededdddgZeeee dd d Z d S) LivepatchPatchFixStatusnamerequiredFdict_keyNamepatchedZPatched)r r$cCs||_||_dS)N)r r$)selfr r$rrr__init__/s z LivepatchPatchFixStatus.__init__N) rrrr rr fieldsrstrboolr&rrrrr)s rc @seZdZededdddedeeddddededddd gZee ee eee d d d Z d S)LivepatchPatchStatusstater!Fr"ZStatefixesZFixesversionZVersion)r+r,r-cCs||_||_||_dS)N)r+r,r-)r%r+r,r-rrrr&Ds  zLivepatchPatchStatus.__init__N) rrrr rrrr'rr(rr&rrrrr*8s   r*c @seZdZededdddededdddededddd gZeeeeeed d d Z d S)LivepatchStatusStatuskernelr!Fr"ZKernel livepatchZ Livepatch supported Supported)r/r0r1cCs||_||_||_dS)N)r/r0r1)r%r/r0r1rrrr&`s  zLivepatchStatusStatus.__init__N) rrrr rr*r'rr(r&rrrrr.Os  r.c@sQeZdZedeeddddgZeeedddZ d S) LivepatchStatusstatusr!Fr"ZStatus)r4cCs ||_dS)N)r4)r%r4rrrr&uszLivepatchStatus.__init__N) rrrr rr.r'rrr&rrrrr3ks   r3)returnc!Cs?tstjddSy(tjtddddg\}}Wn&tjk rjtjddSYnXyt j |}WnBt j k r}ztjd|d|dSWYdd}~XnXyt j |}Wn&tk rtjd |dSYnX|jdks#t|jd kr4tjd dS|jd S) Nz$canonical-livepatch is not installedr4z --verbosez--formatjsonz7canonical-livepatch returned error when checking statuszPJSONDecodeError while parsing livepatch status, returning None. output was: "%s"exc_infozcodeZbodyZ json_dictget isinstancer)rrrrF) r%r-rHrIrJrKrLrMZresponserCZapi_supported_valrrris_kernel_supporteds6       z%UALivepatchClient.is_kernel_supportedN) rrrZcfg_url_base_attrr(rdatetimerrTrrrrrGs  rGcCs&t}|dkrdSt|jS)N)r4rFr1)Z lp_statusrrr_on_supported_kernel_clis  rV)r-rHrIrJr5cCsytjj}Wntk r-d}YnX|dk rtjjtjjtjdd}t |j |k|j |k|j |k|j |k|j|kgr|jdkrtjdd|jfSdS)zOCheck local cache of kernel support :return: (is_cache_valid, result) NZdaysz-livepatch kernel support cache has None valueTF)FN)rlivepatch_support_cachereadrPrUnowtimezoneutcZ timedeltaall cached_atr-rHrIrJr1r:r>)r-rHrIrJZ cache_dataZ one_week_agorrr_on_supported_kernel_caches$           r_)r-rHrIrJrKr5cCstjd|d|d|d|d|}d}|tjkrKd}n|tjkr`d}tjjtjd|d|d|d|d|d t j j t j j |dkrt jd |S) Nr-rHrIrJrKTFr1r^z3livepatch kernel support API response was ambiguous)rGrTrrrrrXwriteZLivepatchSupportCacheDatarUrZr[r\r:r>)r-rHrIrJrKr1Zcache_supportedrrr_on_supported_kernel_api s,       ramaxsizecCsRt}|dk r&tjd|Stj}|jdks_|jdks_|jdkrstjdt j St j |j }tjj}djd|jd|j}t||j||\}}|rtjd|dkrt j S|rt jS|st jStjdt||j|||j}|dkrNt j S|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 Nzusing livepatch cli for supportzHunable to determine enough kernel information to check livepatch supportz{major}.{minor}majorminorzusing livepatch support cachezusing livepatch support api)rVr:r;r Zget_kernel_inforHrcrdr>rrr Zstandardize_arch_nameZuname_machine_archZget_release_infoZseriesformatr_rrrarK)Zcli_saysZ kernel_inforIrJZlp_api_kernel_verZis_cache_validZ cache_saysZapi_saysrrron_supported_kernel.sF          rf) protocol_type retry_sleepsr5cCs6ts dStjtddj|gd|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. Nconfigz {}-proxy=rh)r9r r<r=re)rgrhrrrunconfigure_livepatch_proxyjs rj) http_proxy https_proxyrhr5cCsddlm}|s|r;tjtjjd|j|rftj t ddj|gd||rtj t ddj|gd|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)LivepatchEntitlementserviceriz http-proxy={}rhzhttps-proxy={}N) Zuaclient.entitlementsrmeventinforZSETTING_SERVICE_PROXYretitler r<r=)rkrlrhrmrrrconfigure_livepatch_proxy~s   rr)keyr5cCstjtdg\}}tjdj||tj}|rQ|jdnd}|rrtjdd|}|r|j SdS)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 riz ^{}: (.*)$r8Nz\"(.*)\"z\g<1>) r r<r=researchre MULTILINEgroupsubstrip)rsrArBmatchvaluerrrget_config_option_values !r|cCstjtdk S)N)r Zwhichr=rrrrr9sr9)sZ     (. )/ #  <!