3 ³bfZ<ã @sžddlmZddlZddlZddlZddlZddlZddlZddlZddl Z ddl mZejj dƒZyddlmZeƒZWndZYnXdZesœerÎedƒed ƒes¸ed ƒesÄedƒejdƒdd lmZmZddlmZmZmZmZddlmZmZGdd„de ƒZ!Gdd„de!ƒZ"dZ#dZ$dZ%dZ&dZ'Gdd„de ƒZ(dd„Z)dd„Z*dd„Z+Gdd„de!ƒZ,Gd d!„d!e!ƒZ-Gd"d#„d#e!ƒZ.d$d%„Z/dS)&é)Úprint_functionN)Úcmdlinez /proc/self/io)ÚVmStatFTz= 2.6.20 withz~ - I/O accounting support (CONFIG_TASKSTATS, CONFIG_TASK_DELAY_ACCT, CONFIG_TASK_IO_ACCOUNTING, kernel.task_delayacct sysctl)z0 - VM event counters (CONFIG_VM_EVENT_COUNTERS)é)ÚioprioÚvmstat)Ú ConnectionÚNETLINK_GENERICÚU32AttrÚ NLM_F_REQUEST)Ú ControllerÚGeNlMessagec@seZdZdZdd„ZdS)ÚDumpableObjectzFBase class for all objects that allows easy introspection when printedcCs$dtt|ƒƒdd…tj|jƒfS)Nz%s: %s>réÿÿÿÿ)ÚstrÚtypeÚpprintZpformatÚ__dict__)Úself©rú/usr/lib/python3.6/data.pyÚ__repr__?szDumpableObject.__repr__N)Ú__name__Ú __module__Ú__qualname__Ú__doc__rrrrrr=src@sLeZdZdddddgZdZdd „Zddd„Zdd„Zdd„Ze dd„ƒZ dS)ÚStatsÚblkio_delay_totalé(Úswapin_delay_totalé8Ú read_byteséøÚwrite_byteséÚcancelled_write_byteséNcCsP|j}x6tjD],\}}|||d…}tjd|ƒd||<qWtjsLtƒt_dS)NéÚQr)rrÚmembers_offsetsÚstructÚunpackÚhas_blkio_delay_totalÚsysctl_task_delayacct)rZtask_stats_bufferÚsdÚnameÚoffsetÚdatarrrÚ__init__RszStats.__init__rc CsB|j}|j}|j}x*tjD] \}}|||||||<qWdS)z3Update destination from operator(self, other_stats)N)rrr)) rÚother_statsÚdestinationÚcoeffZddr.ZodÚmemberr0rrrÚ accumulate]s zStats.accumulatecCs|j||ddS)z*Update destination with self - other_statsr)r5r)r7)rr3r4rrrÚdeltaeszStats.deltacCs.|j}x"tjD]\}}||dkrdSqWdS)NrFT)rrr))rr.r/r0rrrÚis_all_zerois zStats.is_all_zerocCs0tjtƒ}|j}xtjD]\}}d||<qW|S)Nr)rÚ__new__rr))ÚstatsZstdr/r0rrrÚbuild_all_zerops zStats.build_all_zero)rr)rr )r!r")r#r$)r%r&)r)rrrr)r,r2r7r8r9Ústaticmethodr<rrrrrGs rééc@s$eZdZdd„Zdd„Zdd„ZdS)ÚTaskStatsNetlinkcCs*||_ttƒ|_t|jƒ}|jdƒ|_dS)NZ TASKSTATS)Úoptionsrr Ú connectionrZ get_family_idÚ family_id)rrAZ controllerrrrr2…s zTaskStatsNetlink.__init__cCst|jttt|ƒgtdS)N)ÚcmdÚattrsÚflags)r rCÚTASKSTATS_CMD_GETr ÚTASKSTATS_CMD_ATTR_PIDr)rÚtidrrrÚ build_request‹s zTaskStatsNetlink.build_requestcCsÈ|jj|jƒytj|jƒ}Wn4tk rR}z|jtjkr@dS‚WYdd}~XnXx,|jj ƒD]\}}|t kr`|jƒ}Pq`WdS|tj }t|ƒdkrœdStjd|dd…ƒd}|dksÀt‚t|ƒS)NiÚHérr>)Útask_stats_requestÚsendrBr ZrecvÚOSErrorÚerrnoZESRCHrEÚitemsÚTASKSTATS_TYPE_AGGR_PIDÚnestedÚTASKSTATS_TYPE_STATSr1Úlenr*r+ÚAssertionErrorr)rÚthreadZreplyÚeZ attr_typeZ attr_valueZtaskstats_dataZtaskstats_versionrrrÚget_single_task_statss$ z&TaskStatsNetlink.get_single_task_statsN)rrrr2rJrYrrrrr@‚sr@cCs g|_d}x‚|jpgD]t}yt|ƒ}WnRtk rxytj|ƒ}Wn(tk rltd|tj dd}YnX|j }YnX|s|jj|ƒqW|rœtjdƒdS)zDBuild options.uids from options.users by resolving usernames to UIDsFz Unknown user:)ÚfileTrN) ÚuidsZusersÚintÚ ValueErrorÚpwdÚgetpwnamÚKeyErrorÚprintÚsysÚstderrZpw_uidÚappendÚexit)rAÚerrorÚuÚuidZpasswdrrrÚ find_uids«s ricCsTi}y6x0td|ƒD] }|jddƒ\}}|jƒ||<qWWntk rNYnX|S)Nz/proc/%d/statusz: r)ÚopenÚsplitÚstripÚIOError)ÚpidZresult_dictÚlineÚkeyÚvaluerrrÚparse_proc_pid_statusÀsrrcCs<y |jdƒStk r$|jdƒStk r6|SXdS)Nzutf-8Z string_escape)ÚdecodeÚUnicodeDecodeErrorÚencodeÚAttributeError)ÚsrrrÚsafe_utf8_decodeËs rxc@s0eZdZdZdd„Zdd„Zdd„Zdd „Zd S)Ú ThreadInfozStats for a single threadcCs.||_d|_d|_tjtƒ|_|j|ƒ|_dS)NT)rIÚmarkÚstats_totalrr:Ústats_deltarJrM)rrIÚtaskstats_connectionrrrr2Õs zThreadInfo.__init__cCstj|jƒS)N)rÚgetrI)rrrrÚ get_ioprioÜszThreadInfo.get_iopriocCstjtj|j||ƒS)N)rÚ set_ioprioZIOPRIO_WHO_PROCESSrI)rÚioprio_classÚioprio_datarrrr€ßszThreadInfo.set_iopriocCs&|js||_|j|j|jƒ||_dS)N)r{r8r|)rr;rrrÚupdate_statsãszThreadInfo.update_statsN)rrrrr2rr€rƒrrrrryÓs ryc@sheZdZdZdd„Zdd„Zdd„Zdd „Zd d„Zdd „Z dd„Z dd„Zdd„Zdd„Z dd„ZdS)ÚProcessInfoz˜Stats for a single process (a single line in the output): if options.processes is set, it is a collection of threads, otherwise a single thread.cCs:||_d|_d|_i|_tjƒ|_tjƒ|_tjƒ|_ dS)N) rnrhÚuserÚthreadsrr<r|Ústats_accumÚtimeZstats_accum_timestamp)rrnrrrr2îs zProcessInfo.__init__cCs:|jr|jr|j|jkrdS|jr6|jƒ|jkr6dSdS)NFT)ÚpidsÚ processesrnr[Úget_uid)rrArrrÚis_monitored÷szProcessInfo.is_monitoredcCsZ|jr|jSytjd|jƒtj}Wntk r>d}YnX||jkrVd|_||_|S)Nz/proc/%d)rhÚosÚstatrnÚST_UIDrOr…)rrhrrrr‹s zProcessInfo.get_uidcCs\|jƒ}|dk rR|jrRyttj|ƒjƒ|_Wn"ttfk rPt|ƒ|_YnX|jpZdS)Nz{none}) r‹r…rxr^ÚgetpwuidZpw_namer`rvr)rrhrrrÚget_userszProcessInfo.get_usercCs ytd|jƒ}|jdƒ}Wntk r0dSXt|jƒ}|sb|jddƒ}|rZd|}nd}|Sd}t|jd|jƒƒ}||jkr²t|ƒjddƒ}|jddƒ}||kr²|d |7}|jd ƒ} | djdƒrð| dj dƒd } | d| d…| d<dj | ƒjƒ}t||ƒS)Nz/proc/%d/cmdlineiz{no such process}ÚNameÚz[%s]z {no name}ZTgidz [%s]úrú/rú ) rjrnÚreadrmrrr~r\rkÚ startswithÚrfindÚjoinrlrx)rZproc_cmdlinerZproc_statusZproc_status_nameÚsuffixÚtgidZ tgid_nameZthread_nameÚpartsZfirst_command_charrrrÚget_cmdlines2 zProcessInfo.get_cmdlinecCs6|r|jjƒSx |jjƒD]}|jjƒsdSqWdS)NTF)r‡r9r†Úvaluesr|)rZaccumulatedÚtrrrÚdid_some_io<s zProcessInfo.did_some_iocCs0tdd„|jjƒDƒƒ}t|ƒdkr,|jƒSdS)Ncss|]}|jƒVqdS)N)r)Ú.0r rrrú Esz)ProcessInfo.get_ioprio..rz?dif)Úsetr†rŸrUÚpop)rZ prioritiesrrrrDszProcessInfo.get_iopriocCs$x|jjƒD]}|j||ƒqWdS)N)r†rŸr€)rrr‚rWrrrr€JszProcessInfo.set_iopriocCstj|jƒƒS)N)rZsort_keyr)rrrrÚioprio_sort_keyNszProcessInfo.ioprio_sort_keycCs*|jj|dƒ}|s&t||ƒ}||j|<|S)N)r†r~ry)rrIr}rWrrrÚ get_threadQs zProcessInfo.get_threadcCs˜tjƒ}x*|jjƒD]\}}|js|j|j|ƒqWtdd„|jjƒDƒƒ|_t|jƒ}|s`dS|j |_ |j |_ ||_|jj|j|jƒdS)NcSsg|]\}}|js||f‘qSr)rz)r¢rIrWrrrú ]sz,ProcessInfo.update_stats..FT)rr<r†rQrzr7r|ÚdictrUrrr‡)rr|rIrWZ nr_threadsrrrrƒXs zProcessInfo.update_statsN)rrrrr2rŒr‹r‘ržr¡rr€r¦r§rƒrrrrr„ês r„c@sDeZdZdd„Zdd„Zdd„Zdd„Zd d „Zdd„Zd d„Z dS)ÚProcessListcCs2i|_||_||_tjƒ|_tjƒ|_|jƒdS)N)rŠr}rArˆÚ timestamprrÚupdate_process_counts)rr}rArrrr2ms zProcessList.__init__cCs8|jj|dƒ}|s$t|ƒ}||j|<|j|jƒr4|SdS)zyEither get the specified PID from self.processes or build a new ProcessInfo if we see this PID for the first timeN)rŠr~r„rŒrA)rrnÚprocessrrrÚget_processxs zProcessList.get_processc Csš|jjr|jjStjdƒ}|jjr0dd„|DƒSg}x`|D]X}d|dkoTdknr:y"|jtttjd|dƒƒƒWq:tk rYq:Xq:W|S) Nz/proccSs0g|](}d|dkodknrt|ƒ‘qS)Ú0rÚ9)r\)r¢rœrrrr¨‰sz*ProcessList.list_tgids..r¯rr°z/proc/z/task) rAr‰rÚlistdirrŠÚextendÚmapr\rO)rZtgidsÚtidsrœrrrÚ list_tgidsƒs " zProcessList.list_tgidscCsf|jjs|gSyttttjd|ƒƒƒ}Wntk r>gSX|jjrbtt |jjƒj t |ƒƒƒ}|S)Nz /proc/%d/task)rArŠÚlistr³r\rr±rOr‰r¤Úintersection)rrœr´rrrÚ list_tids•szProcessList.list_tidsc Cs²tjƒ}||j|_||_d}}x||jƒD]p}|j|ƒ}|s@q,xZ|j|ƒD]L}|j||jƒ}|jj|ƒ}|rL|j |ƒ|j } || j7}|| j7}d|_ qLWq,W||f|jjƒfS)NrF)rˆr«Zdurationrµr®r¸r§r}rYrƒr|r!r#rzrr8) rZ new_timestampZ total_readZtotal_writerœrrIrWr;r8rrrr¬£s$ z!ProcessList.update_process_countscCsTx,|jjƒD]}x|jjƒD] }d|_qWqW|jƒ}tdd„|jjƒDƒƒ|_|S)NTcSs g|]\}}|jƒr||f‘qSr)rƒ)r¢rnrrrrr¨Àsz1ProcessList.refresh_processes..)rŠrŸr†rzr¬r©rQ)rrrWZtotal_read_and_writerrrÚrefresh_processes¹szProcessList.refresh_processescCs i|_dS)N)rŠ)rrrrÚclearÆszProcessList.clearN) rrrr2r®rµr¸r¬r¹rºrrrrrªls rªcCsdtƒjƒkS)a| WAS: try: with open('/proc/sys/kernel/task_delayacct') as f: return bool(int(f.read().strip())) except FileNotFoundError: return None Because /proc/sys/kernel/task_delayacct doesn't exist on RHEL8, it always returns None, which is equivalent to False in the end. On RHEL8, delayacct_on kernel variable is enabled by default Znodelayacct)rÚkeysrrrrr-Êsr-)0Z __future__rrPrrr^rŽr*rbrˆZprocfsrÚpathÚexistsZioaccountingZiotop.vmstatrZvmstat_fZvm_event_countersrareZiotoprrZ iotop.netlinkrr r rZiotop.genetlinkrr ÚobjectrrrGrHrRZTASKSTATS_TYPE_PIDrTr@rirrrxryr„rªr-rrrrÚsV 5)^