U lHJe 2@s.ddlZddlZddlZddlZddlZddlmZddlmZm Z m Z ddl m Z m Z mZmZmZddlmZmZmZmZmZmZddlmZddlmZdZd Zd Zd Ze Z!e"e#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%Z5edd&e(dd'd(Z6d4e.e ee7dd)d*d+Z8d5e e.e e.e ee7dd,d-d.Z9e.e e.d/d0d1Z:e3dd2d3Z;dS)6N) 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@sDeZdZededddededddgZeeee ddd Z d S) LivepatchPatchFixStatusnameFNameZrequiredZdict_keypatchedZPatchedr r#cCs||_||_dSNr$)selfr r#rrr__init__/sz LivepatchPatchFixStatus.__init__N) rrrr rr fieldsrstrboolr'rrrrr)s rc@s`eZdZededddedeedddededddgZee ee eee d d d Z d S) LivepatchPatchStatusstateFStater"fixesZFixesversionZVersionr,r.r/cCs||_||_||_dSr%r0)r&r,r.r/rrrr'DszLivepatchPatchStatus.__init__N) rrrr rrrr(rr)rr'rrrrr+8s  r+c@sXeZdZededddededddededddgZeeeeeed d d Z d S) LivepatchStatusStatuskernelFZKernelr" livepatchZ Livepatch supported Supportedr2r3r4cCs||_||_||_dSr%r6)r&r2r3r4rrrr'`szLivepatchStatusStatus.__init__N) rrrr rr+r(rr)r'rrrrr1Os$r1c@s8eZdZedeedddgZeeedddZ dS) LivepatchStatusstatusFZStatusr"r8cCs ||_dSr%r9)r&r8rrrr'uszLivepatchStatus.__init__N) rrrr rr1r(rrr'rrrrr7ks r7)returnc CststddSzttddddg\}}Wn"tjk rRtdYdSXzt |}Wn:t j k r}ztjd||dWYdSd}~XYnXzt |}Wn"tk rtd |YdSX|jdkst|jd krtd dS|jd S) Nz$canonical-livepatch is not installedr8z --verbosez--formatjsonz7canonical-livepatch returned error when checking statuszPJSONDecodeError while parsing livepatch status, returning None. output was: "%s")exc_infoz    r8) status_strr:cCsJ|dkrtjS|dkrtjS|dkr*tjS|dkr8tjS|dkrFtjSdS)Nr4zkernel-upgrade-requiredzkernel-end-of-lifeZ unsupportedunknown)rrrrrr)rIrrr(_convert_str_to_livepatch_support_statussrKc@s2eZdZdZeeeeeejeedddZdS)UALivepatchClientZ livepatch_urlr/flavorarchcodename build_dater:c Cs|||||dk r|ndd}|}z|jt||d}Wn<tk rx} ztdt| WYdSd} ~ XYnX|jdkrtdt|jdS|j d} | dkst | t r| rt jSt jSt| S)NrJ)zkernel-versionZflavourZ architecturerPz build-date) query_paramsheadersz4error while checking livepatch supported kernels APIz0livepatch supported kernels API was unsuccessfulr5)Z isoformatrSZ request_url"LIVEPATCH_API_V1_KERNELS_SUPPORTED Exceptionr?rCcodeZbodyZ json_dictget isinstancer*rrrrK) r&r/rNrOrPrQrRrSZresponserHZapi_supported_valrrris_kernel_supporteds:          z%UALivepatchClient.is_kernel_supportedN) rrrZcfg_url_base_attrr)rdatetimerrZrrrrrLsrLcCst}|dkrdSt|jSr%)r8rKr4)Z lp_statusrrr_on_supported_kernel_clisr\)r/rNrOrPr:cCsztj}Wntk r&d}YnX|dk rtjtjjtjdd}t |j |k|j |k|j |k|j |k|j|kgr|jdkrtdd|jfSdS)zOCheck local cache of kernel support :return: (is_cache_valid, result) N)Zdaysz-livepatch kernel support cache has None valueT)FN)rlivepatch_support_cachereadrVr[nowtimezoneutcZ timedeltaall cached_atr/rNrOrPr4r?rC)r/rNrOrPZ cache_dataZ one_week_agorrr_on_supported_kernel_caches,     rerMc Csxtj|||||d}d}|tjkr*d}n|tjkr8d}tjtj|||||t j t j j d|dkrtt d|S)N)r/rNrOrPrQTF)r/rNrOrPr4rdz3livepatch kernel support API response was ambiguous)rLrZrrrrr^writeZLivepatchSupportCacheDatar[r`rarbr?rC)r/rNrOrPrQr4Zcache_supportedrrr_on_supported_kernel_api s2     rg)maxsizecCst}|dk rtd|St}|jdksB|jdksB|jdkrRtdt j St |j }tj}dj|j|jd}t||j||\}}|rtd|dkrt j S|rt jS|st jStdt||j|||j}|dkrt 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)r\r?r@r Zget_kernel_inforNrirjrCrrr Zstandardize_arch_nameZuname_machine_archZget_release_infoZseriesformatrerrrgrQ)Zcli_saysZ kernel_inforOrPZlp_api_kernel_verZis_cache_validZ cache_saysZapi_saysrrron_supported_kernel.s\     rl) protocol_type retry_sleepsr:cCs(ts dStjtdd|g|ddS)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=rn)r>r rArBrk)rmrnrrrunconfigure_livepatch_proxyjs rq) http_proxy https_proxyrnr:cCsjddlm}|s|r*ttjj|jd|rHtj t dd|g|d|rftj t dd|g|ddS) 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)serviceroz http-proxy={}rpzhttps-proxy={}N) Zuaclient.entitlementsrteventinforZSETTING_SERVICE_PROXYrktitler rArB)rrrsrnrtrrrconfigure_livepatch_proxy~s" ry)keyr:cCs\ttdg\}}td||tj}|r6|dnd}|rLtdd|}|rX| 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 roz ^{}: (.*)$r=Nz\"(.*)\"z\g<1>) r rArBresearchrk MULTILINEgroupsubstrip)rzrFrGmatchvaluerrrget_config_option_values rcCsttdk Sr%)r ZwhichrBrrrrr>sr>)N)NNN)rrrrsx    ) /  %<    %