o ´iŒd+<ã @sÌddlmZddlZddlZddlZddlZddlZddlZddlZddl Z ze e dƒs5ddlmZee _Wnee fyEe j e _Ynwej d¡ZzddlmZeƒZWneybdZYnwdZeries‚ed ƒed ƒeswedƒes}edƒe d ¡ddlmZmZddlmZmZmZmZddl m!Z!m"Z"Gdd„de#ƒZ$Gdd„de$ƒZ%d Z&d Z'dZ(d Z)dZ*Gdd„de#ƒZ+dd„Z,dd„Z-dd„Z.Gdd „d e$ƒZ/Gd!d"„d"e$ƒZ0Gd#d$„d$e$ƒZ1dS)%é)Úprint_functionNÚ monotonic)rz /proc/self/io)ÚVmStatFTz= 2.6.20 withz` - I/O accounting support (CONFIG_TASKSTATS, CONFIG_TASK_DELAY_ACCT, CONFIG_TASK_IO_ACCOUNTING)z0 - VM event counters (CONFIG_VM_EVENT_COUNTERS)é)ÚioprioÚvmstat)Ú ConnectionÚNETLINK_GENERICÚU32AttrÚ NLM_F_REQUEST)Ú ControllerÚGeNlMessagec@seZdZdZdd„ZdS)ÚDumpableObjectzBBase class for objects that allows easy introspection when printedcCs$dtt|ƒƒdd…t |j¡fS)Nz%s: %s>éÿÿÿÿ)ÚstrÚtypeÚpprintÚpformatÚ__dict__©Úself©rú,/usr/lib/python3/dist-packages/iotop/data.pyÚ__repr__Ks ÿzDumpableObject.__repr__N)Ú__name__Ú __module__Ú__qualname__Ú__doc__rrrrrrIsrc@sFeZdZgd¢ZdZdd„Zddd„Zdd „Zd d„Ze dd „ƒZ dS)ÚStats))Úblkio_delay_totalé()Úswapin_delay_totalé8)Ú read_byteséø)Úwrite_bytesé)Úcancelled_write_bytesiFcCsT|j}tjD]\}}|||d…}t d|¡d||<qtjs(|jdkt_dSdS)NéÚQr)rrÚmembers_offsetsÚstructÚunpackÚhas_blkio_delay_totalr)rÚtask_stats_bufferÚsdÚnameÚoffsetÚdatarrrÚ__init___sÿzStats.__init__rc Cs>|j}|j}|j}tjD]\}}|||||||<qdS)z3Update destination from operator(self, other_stats)N©rrr*) rÚother_statsÚdestinationÚcoeffÚddr/ÚodÚmemberr1rrrÚ accumulatejsÿzStats.accumulatecCs|j||ddS)z*Update destination with self - other_statsr)r7)r;)rr5r6rrrÚdeltarszStats.deltacCs,|j}tjD] \}}||dkrdSqdS)NrFTr4)rr/r0r1rrrÚis_all_zerovsÿzStats.is_all_zerocCs,t t¡}|j}tjD]\}}d||<q|S)Nr)rÚ__new__rr*)ÚstatsÚstdr0r1rrrÚbuild_all_zero}s zStats.build_all_zeroN)r)rrrr*r-r3r;r<r=ÚstaticmethodrArrrrrTs rééc@s$eZdZdd„Zdd„Zdd„ZdS)ÚTaskStatsNetlinkcCs*||_ttƒ|_t|jƒ}| d¡|_dS)NÚ TASKSTATS)Úoptionsrr Ú connectionrÚ get_family_idÚ family_id)rrGÚ controllerrrrr3“s zTaskStatsNetlink.__init__cCst|jttt|ƒgtdS)N)ÚcmdÚattrsÚflags)r rJÚTASKSTATS_CMD_GETr ÚTASKSTATS_CMD_ATTR_PIDr)rÚtidrrrÚ build_request™s þzTaskStatsNetlink.build_requestc CsÆ|j |j¡zt |j¡}Wnty)}z|jtjkr$WYd}~dS‚d}~ww|j ¡D]\}}|t kr=| ¡}nq/dS|tj }t|ƒdkrMdSt d|dd…¡d}|dks_J‚t|ƒS)NiÚHérrC)Útask_stats_requestÚsendrHr ÚrecvÚOSErrorÚerrnoÚESRCHrMÚitemsÚTASKSTATS_TYPE_AGGR_PIDÚnestedÚTASKSTATS_TYPE_STATSr2Úlenr+r,r)rÚthreadÚreplyÚeÚ attr_typeÚ attr_valueÚtaskstats_dataÚtaskstats_versionrrrÚget_single_task_statsžs*€üþ z&TaskStatsNetlink.get_single_task_statsN)rrrr3rRrgrrrrrEsrEcCsœg|_d}|jp gD]8}zt|ƒ}Wn'ty9zt |¡}Wnty3td|tj dd}Ynw|j }Ynw|sB|j |¡q |rLt d¡dSdS)zDBuild options.uids from options.users by resolving usernames to UIDsFz Unknown user:)ÚfileTrN) ÚuidsÚusersÚintÚ ValueErrorÚpwdÚgetpwnamÚKeyErrorÚprintÚsysÚstderrÚpw_uidÚappendÚexit)rGÚerrorÚuÚuidÚpasswdrrrÚ find_uidsºs*þ€ù€ÿrzcCsRi}ztd|ddD]}| dd¡\}}| ¡||<qW|Sty(Y|Sw)Nz/proc/%d/statusÚreplace)Úerrorsú:r)ÚopenÚsplitÚstripÚIOError)ÚpidÚresult_dictÚlineÚkeyÚvaluerrrÚparse_proc_pid_statusÏsþþþr‡cCs>z| d¡WSty| d¡YSty|YSw)Nzutf-8Ú string_escape)ÚdecodeÚUnicodeDecodeErrorÚencodeÚAttributeError)ÚsrrrÚsafe_utf8_decodeÚsÿrŽc@s0eZdZdZdd„Zdd„Zdd„Zdd „Zd S)Ú ThreadInfozStats for a single threadcCs.||_d|_d|_t t¡|_| |¡|_dS)NT)rQÚmarkÚstats_totalrr>Ústats_deltarRrU)rrQÚtaskstats_connectionrrrr3ås zThreadInfo.__init__cCst |j¡S©N)rÚgetrQrrrrÚ get_ioprioìszThreadInfo.get_iopriocCst tj|j||¡Sr”)rÚ set_ioprioÚIOPRIO_WHO_PROCESSrQ)rÚioprio_classÚioprio_datarrrr—ïsÿzThreadInfo.set_iopriocCs&|js||_| |j|j¡||_dSr”)r‘r<r’)rr?rrrÚupdate_statsós zThreadInfo.update_statsN)rrrrr3r–r—r›rrrrrãsrc@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|_t ¡|_t ¡|_t ¡|_ dSr”)r‚rxÚuserÚthreadsrrAr’Ústats_accumÚtimerÚstats_accum_timestamp)rr‚rrrr3þs zProcessInfo.__init__cCs8|jr|js|j|jvrdS|jr| ¡|jvrdSdS)NFT)ÚpidsÚ processesr‚riÚget_uid)rrGrrrÚis_monitoredszProcessInfo.is_monitoredcCsX|jr|jSz t d|j¡tj}Wntyd}Ynw||jkr*d|_||_|S)Nz/proc/%d)rxÚosÚstatr‚ÚST_UIDrXr©rrxrrrr¤sþ zProcessInfo.get_uidc CsX| ¡}|dur'|js'ztt |¡jƒ|_Wnttfy&t|ƒ|_Ynw|jp+dS)Nz{none}) r¤rrŽrmÚgetpwuidÚpw_namerorŒrr©rrrÚget_user$sÿ zProcessInfo.get_usercCsztd|jƒ}| d¡}Wn tyYdSwt|jƒ}|s2| dd¡}|r.d|}|Sd}|Sd}t| d|j¡ƒ}||jkrZt|ƒ dd¡}| dd¡}||krZ|d |7}| d ¡} | d d¡ry| d d¡d } | d| d…| d<d | ¡ ¡}t||ƒS)Nz/proc/%d/cmdlineiz{no such process}ÚNameÚz[%s]z {no name}ÚTgidz [%s]úrú/rú ) r~r‚Úreadrr‡r•rkrÚ startswithÚrfindÚjoinr€rŽ)rÚproc_cmdlineÚcmdlineÚproc_statusÚproc_status_nameÚsuffixÚtgidÚ tgid_nameÚthread_nameÚpartsÚfirst_command_charrrrÚget_cmdline-s6ÿ ÿ zProcessInfo.get_cmdlinecCs4|r|j ¡S|j ¡D] }|j ¡sdSq dS)NTF)rŸr=ržÚvaluesr’)rÚaccumulatedÚtrrrÚdid_some_ioLs ÿzProcessInfo.did_some_iocCs0tdd„|j ¡Dƒƒ}t|ƒdkr| ¡SdS)Ncss|]}| ¡VqdSr”)r–)Ú.0rÄrrrÚ Us€z)ProcessInfo.get_ioprio..rz?dif)ÚsetržrÂr_Úpop)rÚ prioritiesrrrr–TszProcessInfo.get_iopriocCs |j ¡D]}| ||¡qdSr”)ržrÂr—)rr™ršr`rrrr—ZsÿzProcessInfo.set_iopriocCst | ¡¡Sr”)rÚsort_keyr–rrrrÚioprio_sort_key^szProcessInfo.ioprio_sort_keycCs*|j |d¡}|st||ƒ}||j|<|Sr”)ržr•r)rrQr“r`rrrÚ get_threadas zProcessInfo.get_threadcCs”t ¡}|j ¡D]\}}|js| |j|¡q tdd„|j ¡Dƒƒ|_t|jƒ}|s.dS|j |_ |j |_ ||_|j |j|j¡dS)NcSsg|]\}}|js||f‘qSr)r)rÆrQr`rrrÚ msÿz,ProcessInfo.update_stats..FT)rrAržr[rr;r’Údictr_rr!rŸ)rr’rQr`Ú nr_threadsrrrr›hs € ÿ zProcessInfo.update_statsN)rrrrr3r¥r¤r¬rÁrÅr–r—rÌrÍr›rrrrrœús rœc@sDeZdZdd„Zdd„Zdd„Zdd„Zd d „Zdd„Zd d„Z dS)ÚProcessListcCs2i|_||_||_t ¡|_t ¡|_| ¡dSr”) r£r“rGr rÚ timestamprrÚupdate_process_counts)rr“rGrrrr3~s zProcessList.__init__cCs8|j |d¡}|st|ƒ}||j|<| |j¡r|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¥rG)rr‚ÚprocessrrrÚget_process‰s ÿzProcessList.get_processc Cs”|jjr|jjSt d¡}|jjrdd„|DƒSg}|D]+}d|dkr*dkrGnqz| ttt d|d¡ƒ¡WqtyFYqwq|S) Nz/proccSs0g|]}d|dkrdkrnnt|ƒ‘qS)Ú0rÚ9)rk)rÆr¼rrrrÎšs0z*ProcessList.list_tgids..rÖrr×z/proc/z/task) rGr¢r¦Úlistdirr£ÚextendÚmaprkrX)rÚtgidsÚtidsr¼rrrÚ list_tgids”s" ÿþüzProcessList.list_tgidscCsh|jjs|gSztttt d|¡ƒƒ}Wnty gYSw|jjr2tt |jjƒ t |ƒ¡ƒ}|S)Nz /proc/%d/task)rGr£ÚlistrÚrkr¦rØrXr¢rÈÚintersection)rr¼rÜrrrÚ list_tids§sÿzProcessList.list_tidsc Csªt ¡}||j|_||_d}}| ¡D]6}| |¡}|sq| |¡D]&}| ||j¡}|j |¡}|rJ| |¡|j} || j7}|| j 7}d|_q$q||f|j ¡fS)NrF)r rrÒÚdurationrÝrÕràrÍr“rgr›r’r#r%rrr<) rÚ new_timestampÚ total_readÚtotal_writer¼rÔrQr`r?r<rrrrÓµs( €ø z!ProcessList.update_process_countscCsL|j ¡D] }|j ¡D]}d|_qq| ¡}tdd„|j ¡Dƒƒ|_|S)NTcSs g|]\}}| ¡r||f‘qSr)r›)rÆr‚rÔrrrrÎÒsþz1ProcessList.refresh_processes..)r£rÂržrrÓrÏr[)rrÔr`Útotal_read_and_writerrrÚrefresh_processesËsÿ ÿzProcessList.refresh_processescCs i|_dSr”)r£rrrrÚclearØs zProcessList.clearN) rrrr3rÕrÝràrÓrærçrrrrrÑ}s rÑ)2Ú __future__rrYr¦rrmr§r+rqr ÚhasattrrÚImportErrorÚRuntimeErrorÚpathÚexistsÚioaccountingÚiotop.vmstatrÚvmstat_frÚvm_event_countersrpruÚiotoprrÚ iotop.netlinkrr r rÚiotop.genetlinkrr ÚobjectrrrOrPr\ÚTASKSTATS_TYPE_PIDr^rErzr‡rŽrrœrÑrrrrÚsf €ÿ ÿ 5*