­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ 3 ;j[#@sdddlmZddlZddlZddlZddlZddlZddlZddlZddl Z ddl m Z ddl m Z ej dZdZedZed ZedZdddZd?ddZddZd@ddZdedfddZddZdd Zd!d"Zd#d$Zd%d&Zd'd(Z ed)d*Z!d+d,Z"dAd-d.Z#dBd/d0Z$d1d2Z%dCd4d5Z&d6d7Z'd8d9Z(de)fdfd:d;Z*dS)D)print_functionN)datetime) constantsz^(\d+[.]\d+[-]\d+)Fwc Cstjj|}|r(tjj| r(tj|ytj|j}Wntk rP|}YnXtj ||tjj |ddd<}tj |j ||j ||jtj|j |j}WdQRXtj|tj} ztj| Wdtj| Xtj||dS)N.F)modedirprefixdelete)ospathdirnameexistsmakedirsstatst_mode ExceptiontempfileNamedTemporaryFilebasenamefchmodfilenowriteflushfsyncnameopenO_RDONLYcloserename) fnamecontent ensure_dirr Z create_modeZdnamerfZ tmp_fnameZ folder_fdr)+/usr/libexec/kcare/python/kcarectl/utils.py atomic_write"s$      r+utf-8cCs4t|tkr|St|tkr&|j|S|j|SdS)N)typentypebtypedecodeencode)dataencodingr)r)r*nstr>s    r4latin1cCst|tkr|j|}|S)N)r-utyper1)r2r3r)r)r*bstrHs  r7cCst|tkr|j|}|S)N)r-r/r0)r2r3r)r)r*ustrOs  r8cs4itjfdd}|_j|_|_|S)Nc sH|tt|jf}y|Stk r0YnX||}|<|S)N)tuplesorteditemsKeyError)argskwargsZ cache_keyresult)cachefnr)r*innerZszcached..inner) functoolswrapsr@clearorig)rArBr))r@rAr*cachedVs  rGcsFdkrtjdkrtjdkr*tjifdd}|S)Ncsfdd}|S)Ncs}xltD]`}y ||Stk rJ}z|s:WYdd}~XnXtj|t|tjdtj}qWy ||Stk r}zt |dWYdd}~XnXdS)Nrattempts) rangertimesleepminrandomuniformrRETRY_MAX_DELAYsetattr)r=r>Zldelay_exZfinal_ex)backoff check_retrycountdelayrAstater)r*rBvs     z'retry..decorator..innerr))rArB)rSrTrUrVrW)rAr* decoratoruszretry..decorator)r RETRY_DELAY RETRY_COUNT RETRY_BACKOFF)rTrUrVrSrXr))rSrTrUrVrWr*retryjsr\c Cstjj|sdSg}tj|}|dk r2tj||}xD|D]<}tjj||}||krh|jtj|j |f|j ddq8WxF||dD]6\}} tjj | stjj | rtj | qtj| qWdS)NT)reverse)rrrlistdirfnmatchfilterjoinappendrst_mtimesortisfileislinkremoveshutilrmtree) Z directory exclude_pathkeep_npatternr2r;item full_pathrQentryr)r)r*clean_directorys     rpcCsNttjjtjdddttjjtjdddtjjtjrJtjtjdS)Nmodulesr)rkpatches) rprrrar PATCH_CACHErCACHE_KEY_DUMP_PATHunlinkr)r)r)r*clear_all_cachesrvc Cs^tjj|}tjj|s"tj|t|d(}tj|||jtj |j WdQRXdS)Nwb) rrrrrr!rh copyfileobjrrr)responsedstZ parent_dirr(r)r)r* save_to_files     r{cCstj|}|r|jdp|S)Nr) VERSION_REmatchgroup)versionr}r)r)r*strip_version_timestamps rcCs0|jd\}}}|s$|jd\}}}tj|dS)NT z%Y-%m-%d) partitionrstrptime)Zstr_rawZstr_dateseprQr)r)r*parse_response_datesrcGs0ddlm}djddtd|jf|DS)Nr) ipv6_support/css|]}|jdVqdS)rN)strip).0itr)r)r* sz'get_patch_server_url..)rrrar`get_patch_server)partsrr)r)r*get_patch_server_urls rc Cs0tjj|sdSt|}|jjSQRXdS)N)rrrr!readr)filenamer(r)r)r* try_to_reads  rcCs ttjS)N)rrrtr)r)r)r* get_cache_keysrc Cs.tjj|s|St|| }|jSQRXdS)N)rrrr!r)r%r defaultr(r)r)r* _read_files  rcCst|d|}|S)Nr)r)r%rr?r)r)r* read_files rcCst|d|}|S)Nrb)r)r%rr?r)r)r* read_file_bins rcCsBi}|j}x0|D](}|r|jd\}}}|r|j||<qW|S)N:) splitlinesrr)r2r?Z data_lineslinekey delimitervaluer)r)r* data_as_dicts rcvecCs0t}x$|D]}|j|}|r |j|q W|S)a Extract unique CVEs from a list of patches. Args: patches: List of patch dictionaries cve_field: Field name to extract CVE from (default 'cve' for userspace, 'kpatch-cve' for kernel) Returns: Set of unique CVE strings )setgetadd)rr cve_fieldZ unique_cvespatchrr)r)r*extract_unique_cvess   rcCstttjS)N)strintrJr)r)r)r* timestamp_strsrcGs t|dS)za workaround to fix T201N)print)valuesr)r)r* print_wrappersrcsfdd}|S)Ncs tjfdd}|S)Ncsny ||Sk rh}zBrZdd|Ddd|jD}djjdj||Sd}~XnXdS)NcSsg|] }t|qSr))r)rar)r)r* &szBcatch_errors..decorator..inner..cSsg|]\}}dj||qS)z{0}={1})format)rkvr)r)r*r&sz{0}({1}) failed: {2}z, )r;r__name__ra)r=r>earg_list)default_returnerrorsrAloggerr)r*rBs  z.catch_errors..decorator..inner)rCrD)rArB)rrr)rAr*rXs zcatch_errors..decoratorr))rrrrXr))rrrr* catch_errorssr)Fr r )r,)r5)r5)NNN)N)N)r)+ __future__rr_rCrrMrerhrrJrrrcompiler|Z CACHE_ENTRIESr-r.r/r6r+r4r7r8rGr\rprvr{rrrrrrrrrrrrrrr)r)r)r*sJ        $