ÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂ 3 ôn»j&ã@slddlmZddlZddlZddlZddlZddlZddlZddlZddl Z ddl m Z ddlmZej dƒZdZedƒZed ƒZedƒZd?d d„Zd@dd„Zdd„ZdAdd„ZdBdd„Zdd„ZdCdd„Zdedfdd„Zd d!„Zd"d#„Zd$d%„Zd&d'„Zd(d)„Z d*d+„Z!ed,d-„ƒZ"d.d/„Z#dDd0d1„Z$dEd2d3„Z%d4d5„Z&dFd7d8„Z'd9d:„Z(d;d<„Z)de*fdfd=d>„Z+dS)Gé)Úprint_functionN)Údatetimeé)Ú constantsz^(\d+[.]\d+[-]\d+)éÚóFÚwé¤c Csìtjj|ƒ}|r(tjj|ƒr(tj|ƒytj|ƒj}Wntk rP|}YnXtj ||tjj |ƒddd<}tj|jƒ|ƒ|j |ƒ|jƒtj|jƒƒ|j}WdQRXtj|tjƒ} ztj| ƒWdtj| ƒXtj||ƒdS)NÚ.F)ÚmodeÚdirÚprefixÚdelete)ÚosÚpathÚdirnameÚexistsÚmakedirsÚstatÚst_modeÚ ExceptionÚtempfileÚNamedTemporaryFileÚbasenameÚfchmodÚfilenoÚwriteÚflushÚfsyncÚnameÚopenÚO_RDONLYÚcloseÚrename) ÚfnameÚcontentÚ ensure_dirrZcreate_modeZdnamerÚfZ tmp_fnameZ folder_fd©r)ú+/usr/libexec/kcare/python/kcarectl/utils.pyÚatomic_write"s$ r+úutf-8ÚstrictcCs8t|ƒtkr|St|ƒtkr(|j||ƒS|j||ƒSdS)N)ÚtypeÚntypeÚbtypeÚdecodeÚencode)ÚdataÚencodingÚerrorsr)r)r*Únstr>s r6cCs&t|dƒ}t|jƒddSQRXdS)aRead ``path`` as UTF-8 text, independent of the locale codec. Under a C/POSIX locale Python 3 opens text files with an ASCII codec, so a non-ASCII byte in the file (e.g. an em-dash in a patch-info skip reason) makes a plain text read raise UnicodeDecodeError. Reading the bytes and decoding them as UTF-8 matches how the same data is decoded when fetched over HTTP. A byte that is not valid UTF-8 is replaced rather than raising, so a malformed file degrades output instead of aborting the command. ÚrbÚreplace)r5N)r!r6Úread)rr(r)r)r*Úread_text_fileHsr:Úlatin1cCst|ƒtkr|j|ƒ}|S)N)r.Úutyper2)r3r4r)r)r*ÚbstrXs r=cCst|ƒtkr|j|ƒ}|S)N)r.r0r1)r3r4r)r)r*Úustr_s r>cs4i‰tjˆƒ‡‡fdd„ƒ}ˆ|_ˆj|_ˆ|_|S)NcsH|tt|jƒƒƒf}yˆ|Stk r0YnXˆ||Ž}ˆ|<|S)N)ÚtupleÚsortedÚitemsÚKeyError)ÚargsÚkwargsZ cache_keyÚresult)ÚcacheÚfnr)r*Úinnerjszcached..inner)Ú functoolsÚwrapsrFÚclearÚorig)rGrHr))rFrGr*Úcachedfs rMcsFˆdkrtj‰ˆdkrtj‰ˆdkr*tj‰i‰‡‡‡‡‡fdd„}|S)Ncs‡‡‡‡‡‡fdd„}|S)Ncs²ˆ}xltˆƒD]`}y ˆ||ŽStk rJ}zˆ|ˆƒs:‚WYdd}~XnXtj|ƒt|tjdˆƒtjƒ}qWy ˆ||ŽStk r¬}zt |dˆƒ‚WYdd}~XnXdS)NrÚattempts) ÚrangerÚtimeÚsleepÚminÚrandomÚuniformrÚRETRY_MAX_DELAYÚsetattr)rCrDZldelayÚ_ÚexZfinal_ex)ÚbackoffÚcheck_retryÚcountÚdelayrGÚstater)r*rH†s z'retry..decorator..innerr))rGrH)rYrZr[r\r])rGr*Ú decorator…szretry..decorator)rÚRETRY_DELAYÚRETRY_COUNTÚ RETRY_BACKOFF)rZr[r\rYr^r))rYrZr[r\r]r*Úretryzsrbc CsÄtjj|ƒsdSg}tj|ƒ}|dk r2tj||ƒ}xD|D]<}tjj||ƒ}||krh|jtj|ƒj |fƒ|j ddq8WxF||d…D]6\}} tjj| ƒs¦tjj| ƒr²tj | ƒq†tj| ƒq†WdS)NT)Úreverse)rrrÚlistdirÚfnmatchÚfilterÚjoinÚappendrÚst_mtimeÚsortÚisfileÚislinkÚremoveÚshutilÚrmtree) Z directoryÚexclude_pathÚkeep_nÚpatternr3rAÚitemÚ full_pathrWÚentryr)r)r*Úclean_directoryžs rvcCsNttjjtjdƒddttjjtjdƒddtjjtjƒrJtjtjƒdS)NÚmodulesr)rqÚpatches) rvrrrgrÚPATCH_CACHErÚCACHE_KEY_DUMP_PATHÚunlinkr)r)r)r*Úclear_all_cache²sr|c Cs^tjj|ƒ}tjj|ƒs"tj|ƒt|dƒ(}tj||ƒ|jƒtj |j ƒƒWdQRXdS)NÚwb)rrrrrr!rnÚcopyfileobjrrr)ÚresponseÚdstZ parent_dirr(r)r)r*Úsave_to_fileºs rcCstj|ƒ}|r|jdƒp|S)Nr)Ú VERSION_REÚmatchÚgroup)Úversionrƒr)r)r*Ústrip_version_timestampÅs r†cCs0|jdƒ\}}}|s$|jdƒ\}}}tj|dƒS)NÚTú z%Y-%m-%d)Ú partitionrÚstrptime)Zstr_rawZstr_dateÚseprWr)r)r*Úparse_response_dateËsrŒcGs0ddlm}djdd„td|jƒf|ƒDƒƒS)Nr)Úipv6_supportú/css|]}|jdƒVqdS)rŽN)Ústrip)Ú.0Úitr)r)r*ú Ùsz'get_patch_server_url..)rrrgrfÚget_patch_server)Úpartsrr)r)r*Úget_patch_server_urlÕsr•c Cs0tjj|ƒsdSt|ƒ}|jƒjƒSQRXdS)N)rrrr!r9r)Úfilenamer(r)r)r*Útry_to_readÜs r—cCs ttjƒS)N)r—rrzr)r)r)r*Ú get_cache_keyåsr˜c Cs.tjj|ƒs|St||ƒ }|jƒSQRXdS)N)rrrr!r9)r%rÚdefaultr(r)r)r*Ú _read_fileësršcCst|d|ƒ}|S)NÚr)rš)r%r™rEr)r)r*Ú read_fileósrœcCst|d|ƒ}|S)Nr7)rš)r%r™rEr)r)r*Ú read_file_binùsrcCsBi}|jƒ}x0|D](}|r|jdƒ\}}}|r|jƒ||<qW|S)Nú:)Ú splitlinesr‰r)r3rEZ data_linesÚlineÚkeyÚ delimiterÚvaluer)r)r*Údata_as_dictÿs r¤ÚcvecCs0tƒ}x$|D]}|j|ƒ}|r|j|ƒqW|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 )ÚsetÚgetÚadd)rxÚ cve_fieldZunique_cvesÚpatchr¥r)r)r*Úextract_unique_cvess r«cCstttjƒƒƒS)N)ÚstrÚintrPr)r)r)r*Ú timestamp_strsr®cGst|ŽdS)za workaround to fix T201N)Úprint)Úvaluesr)r)r*Ú print_wrapper$sr±cs‡‡‡fdd„}|S)Ncs tjˆƒ‡‡‡‡fdd„ƒ}|S)Ncsny ˆ||ŽSˆk rh}zBˆrZdd„|Dƒdd„|jƒDƒ}ˆdjˆjdj|ƒ|ƒƒˆSd}~XnXdS)NcSsg|]}t|ƒ‘qSr))r¬)rÚar)r)r*ú 6szBcatch_errors..decorator..inner..cSsg|]\}}dj||ƒ‘qS)z{0}={1})Úformat)rÚkÚvr)r)r*r³6sz{0}({1}) failed: {2}z, )rAr´Ú__name__rg)rCrDÚeÚarg_list)Údefault_returnr5rGÚloggerr)r*rH/s z.catch_errors..decorator..inner)rIrJ)rGrH)rºr5r»)rGr*r^-szcatch_errors..decoratorr))r»r5rºr^r))rºr5r»r*Úcatch_errors*sr¼)Fr r )r,r-)r;)r;)NNN)N)N)r¥),Ú __future__rrerIrrSÚrernrrPrrrÚcompiler‚Z CACHE_ENTRIESr.r/r0r<r+r6r:r=r>rMrbrvr|rr†rŒr•r—r˜ršrœrr¤r«r®r±rr¼r)r)r)r*ÚsL $