Cs[K @ sA d Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l m Z d d l m
Z
d d l m Z d d l m
Z
d d l m Z d d l m Z d d
l m Z d d l m Z d d l m Z d d
l m Z d d l m Z d d l m Z d d l m Z d d l m Z d d l m Z d d l m Z d d l m Z d d l m Z d g Z e! e d Z" e j# d Z$ e j% Z% e j& e j' e j( e j) e j* e j+ e j, e j+ e j- e j. i Z/ e j0 e j1 e j2 e j3 e j4 e j5 e j6 e j7 e j8 e j9 e j: e j; e j< e j= e j> e j? e j@ e jA e jB e jC e jD e jE e jF e jG i ZH eI d d d d d d d d d d d! d" d# d$ d% d& ZJ e d' d d g ZK eK ZL e d( d) d* d+ d, g ZM e d- d. d/ d0 d1 d2 g ZN e d3 d4 d d5 d6 g ZO e d7 d8 d9 jP eO jQ ZR d: d; ZS d< d= ZT d> d? ZU d@ dA ZV dB dC ZW dD dE ZX dF dG ZY dH dI ZZ e j[ Z[ e
j\ Z\ dJ dK dL Z] e j^ Z^ e j_ Z_ d] dM dN Z` dO dP Za dQ dR Zb dS dT Zc dU dV Zd dW dX Ze dY dZ Zf Gd[ d\ d\ eg Zh d S)^z AIX platform implementation. N)
namedtuple)AF_INET )_common)_psposix) _psutil_aix)
_psutil_posix)AF_INET6)memoize_when_activated)NIC_DUPLEX_FULL)NIC_DUPLEX_HALF)NIC_DUPLEX_UNKNOWN)sockfam_to_enum)socktype_to_enum)
usage_percent)PY3) AccessDenied)
NoSuchProcess)
ZombieProcess PROCFS_PATH proc_threads SC_PAGE_SIZEppidrssvms create_time nice num_threads status ttynr pmem scputimesusersystemZidleZiowaitsvmemtotalZ availablepercentusedfree
pmmap_groupedpathZanonlocked pmmap_extz addr perms c C s t j d j S)z+Return updated psutil.PROCFS_PATH constant.Zpsutil)sysmodulesr r6 r6
/_psaix.pyget_procfs_pathl s r8 c C sJ t j \ } } } } } t | | | d d } t | | | | | S)Nround_r )cextZ virtual_memr r* )r+ Zavailr. ZpinnedZinuser, r6 r6 r7 virtual_memoryv s r; c C sS t j \ } } } } | | } t | | d d } t j | | | | | | S)z=Swap system memory as a (total, used, free, sin, sout) tuple.r9 r )r: Zswap_memr r Zsswap)r+ r. sinZsoutr- r, r6 r6 r7 swap_memory| s
r= c C s) t j } t d d t | D S)z-Return system-wide CPU times as a named tuplec S s g | ] } t | q Sr6 )sum).0xr6 r6 r7
<listcomp> s z cpu_times.<locals>.<listcomp>)r:
per_cpu_timesr' zip)retr6 r6 r7 cpu_times s rE c C s t j } d d | D S)z5Return system per-CPU times as a list of named tuplesc S s g | ] } t | q Sr6 )r' )r? r@ r6 r6 r7 rA s z!per_cpu_times.<locals>.<listcomp>)r: rB )rD r6 r6 r7 rB s rB c C s. y t j d SWn t k
r) d SYn Xd S)z0Return the number of logical CPUs in the system.SC_NPROCESSORS_ONLNN)ossysconf
ValueErrorr6 r6 r6 r7 cpu_count_logical s
rJ c C s d } t j | d d d t j d t j } | j \ } } t rd d d | | f D \ } } | j d k r t d | | f | j j } t | p d S)
Nzlsdev -Cc processorshellTstdoutstderrc S s% g | ] } | j t j j q Sr6 )decoder4 rL encoding)r? r@ r6 r6 r7 rA s z&cpu_count_physical.<locals>.<listcomp>r z%r command error
%s)
subprocessPopenPIPE communicater
returncode RuntimeErrorstrip
splitlineslen)cmdprL rM Z
processorsr6 r6 r7 cpu_count_physical s r[ c C s. t j \ } } } } t j | | | | S)z*Return various CPU stats as a named tuple.)r: cpu_statsr Z scpustats)Z ctx_switchesZ
interruptsZsoft_interruptsZsyscallsr6 r6 r7 r\ s r\ Fc C s g } t j } xo | D]g } | \ } } } } | d k rC d } | s[ t | j s[ q t j | | | | } | j | q W| S)z Return system disk partitions.Znone )r: disk_partitions
disk_usager+ r Z sdiskpartappend) allretlistZ
partitions partitionZdeviceZ
mountpointZfstypeZoptsntupler6 r6 r7 r^ s
r^ c C sy t j } | | k rA t d | d j d d | D f t j | \ } } t j | } t } x | D] } | \ } } }
} } }
} | | k r qs |
| k r qs t |
}
| t t f k r | r t j
| } | r t j
| } t | } t |
}
| d k r@t j
| | |
| | |
| } n t j | | |
| | |
} | j | qs Wt | S)zReturn socket connections. If pid == -1 return system-wide
connections (as opposed to connections opened by one process only).
z+invalid %r kind argument; choose between %sz, c S s g | ] } t | q Sr6 )repr)r? r@ r6 r6 r7 rA s z#net_connections.<locals>.<listcomp>r )r Z conn_tmaprI joinr: net_connectionsset TCP_STATUSESr r addrr r ZsconnZpconnaddlist)kind_pidZcmapfamiliestypesrawlistrD itemfdZfamZtype_laddrraddrr" pidntr6 r6 r7 rh s4 &
$ rh c C sJ d t d t i } t d d t D } i } x| D]} t j | \ } } d } d } t j d d | g d t j d
t j } | j \ } }
t
r d d | |
f D \ } }
| j d k rt j
d | } | d
k rt | j d } | j d } | j | t } t j | | | | | | <q; W| S)z)Get NIC stats (isup, duplex, speed, mtu).FullZHalfc S s g | ] } | d q S)r r6 )r? r@ r6 r6 r7 rA s z net_if_stats.<locals>.<listcomp>r] r z/usr/bin/entstatz-drL rM c S s% g | ] } | j t j j q Sr6 )rN r4 rL rO )r? r@ r6 r6 r7 rA
s z"Running: (\d+) Mbps.*?(\w+) DuplexNr r )r r ri net_if_addrsr: net_if_statsrP rQ rR rS r rT researchintgroupgetr
r Z snicstats) Z
duplex_mapnamesrD nameZisupZmtuduplexZspeedrZ rL rM Z re_resultr6 r6 r7 r{ s.
r{ c C s
t j S)z:The system boot time expressed in seconds since the epoch.)r: boot_timer6 r6 r6 r7 r s r c C s g } t j } d } xi | D]a } | \ } } } } } } | sF q | | k rX d } t j | | | | | }
| j |
q W| S)z:Return currently connected users as a list of namedtuples.:0.0:0 localhost)r r )r: usersr Zsuserr` ) rb rr r rs r( ZttyhostnameZtstampZ user_processrw rx r6 r6 r7 r % s
r c C s d d t j t D S)z7Returns a list of PIDs currently running on the system.c S s( g | ] } | j r t | q Sr6 )isdigitr~ )r? r@ r6 r6 r7 rA ? s zpids.<locals>.<listcomp>)rG listdirr8 r6 r6 r6 r7 pids= s r c C s+ t j j t j j t t | d S)z&Check for the existence of a unix pid.Zpsinfo)rG r0 existsrg r8 str)rw r6 r6 r7
pid_existsB s r c s f d d } | S)zCall callable into a try/except clause and translate ENOENT,
EACCES and EPERM in NoSuchProcess or AccessDenied exceptions.
c s y | | | SWn t k
r } z t d k sM t d k sM t d k rP | j t j t j f k r t | j s t | j | j n t | j | j | j
| j t j t j f k r t | j | j WYd d } ~ Xn Xd S)N)
EnvironmentErrorr r r errnoENOENTZESRCHr rw _name_ppidEPERMZEACCES)selfargskwargserr)funr6 r7 wrapperL s z wrap_exceptions.<locals>.wrapperr6 )r r r6 )r r7 wrap_exceptionsG s r c @ s e Z d Z d Z d d d d g Z d d Z d d Z d
d Z e d d
Z e d d Z
e d d Z e d d Z
e d d Z e d d Z e d d Z e d d Z e r e d d Z e d d d Z e d! d" Z e d# d$ Z e d% d& Z e d' d( Z e d) d* Z e d+ d, Z e d- d. Z e d/ d0 Z e d1 d2 Z e Z e d3 d4 Z d5 d6 Z e d7 d8 Z! e d9 d: Z" e d; d< d= Z# e d> d? Z$ d; S)@Processz1Wrapper class around underlying C implementation.rw r r _procfs_pathc C s+ | | _ d | _ d | _ t | _ d S)N)rw r r r8 r )r rw r6 r6 r7 __init__g s zProcess.__init__c C s+ | j j | j j | j j d S)N)_proc_name_and_argsZcache_activate_proc_basic_info
_proc_cred)r r6 r6 r7
oneshot_enterm s
zProcess.oneshot_enterc C s+ | j j | j j | j j d S)N)r Zcache_deactivater r )r r6 r6 r7 oneshot_exitr s
zProcess.oneshot_exitc C s t j | j | j S)N)r: Zproc_name_and_argsrw r )r r6 r6 r7 r w s zProcess._proc_name_and_argsc C s t j | j | j S)N)r: Zproc_basic_inforw r )r r6 r6 r7 r { s zProcess._proc_basic_infoc C s t j | j | j S)N)r: Z proc_credrw r )r r6 r6 r7 r s zProcess._proc_credc C s* | j d k r d S| j d j d S)Nr Zswapper )rw r rstrip)r r6 r6 r7 r s z Process.namec C s | j d } t j j | k r t j j | s[ t j j t j j | j | } t j j | r t j j | r t j | t j
r | St j j | } xj t j d j
d D]R } t j j t j j | | } t j j | r t j | t j
r | Sq Wd S)Nr PATH:r] )cmdlinerG r0 sepisabsabspathrg cwdisfileaccessX_OKbasenameenvironsplit)r exer0 Z possible_exer6 r6 r7 r s ' !z Process.exec C s | j d j d S)Nr r3 )r r )r r6 r6 r7 r s zProcess.cmdlinec C s | j t d S)Nr )r
proc_info_map)r r6 r6 r7 r s zProcess.create_timec C s | j t d S)Nr )r r )r r6 r6 r7 r s zProcess.num_threadsc C s{ t j | j } g } x9 | D]1 \ } } } t j | | | } | j | q W| sw t j d | j | j f | S)Nz%s/%s) r: r rw r Zpthreadr` rG statr )r rr rb Z thread_idutimeZstimerd r6 r6 r7 threads s zProcess.threadsZinetc C s< t | d | j } | s8 t j d | j | j f | S)Nro z%s/%s)rh rw rG r r )r rn rD r6 r6 r7 connections s zProcess.connectionsc C s t j | j S)N)
cext_posix getpriorityrw )r r6 r6 r7 nice_get s zProcess.nice_getc C s t j | j | S)N)r setpriorityrw )r valuer6 r6 r7 nice_set s zProcess.nice_setc C s | j t d | _ | j S)Nr )r r r )r r6 r6 r7 r s z Process.ppidc C s1 | j \ } } } } } } t j | | | S)N)r r puids)r real effectivesaved_r6 r6 r7 uids s z Process.uidsc C s1 | j \ } } } } } } t j | | | S)N)r r r )r r r r r r6 r6 r7 gids s z Process.gidsc C s% t j | j | j } t j | S)N)r: Zproc_cpu_timesrw r r Z pcputimes)r rE r6 r6 r7 rE s zProcess.cpu_timesc C sd | j t d } | d @d ?| d @B} x3 t j d D]" } t j | j | k r: | Sq: Wd S)Nr$ l i z /dev/**/*)r r globrG r st_rdev)r ZttydevZdevr6 r6 r7 terminal s zProcess.terminalc C s | j } y- t j d | | j f } | j d SWnX t k
r } z8 | j t j k r{ t j d | | j f d S WYd d } ~ Xn Xd S)Nz %s/%s/cwd/z%s/%s) r rG readlinkrw r OSErrorr r r )r Z procfs_pathresultr r6 r6 r7 r s z Process.cwdc C s= | j } | t d d } | t d d } t | | S)Nr i r )r r r& )r rD r r r6 r6 r7 memory_info s zProcess.memory_infoc C s$ | j t d } t j | d S)Nr" ?)r r
PROC_STATUSESr )r coder6 r6 r7 r" s zProcess.statusc C s# t j d d t | j g d t j d t j } | j \ } } t rj d d | | f D \ } } d | j k r t | j | j t
j d | } g } xs | D]k \ } } | j } | j
d r | d
d } | j d k r q | j t j | t | q W| S) Nz/usr/bin/procfilesz-nrL rM c S s% g | ] } | j t j j q Sr6 )rN r4 rL rO )r? r@ r6 r6 r7 rA s z&Process.open_files.<locals>.<listcomp>zno such processz (\d+): S_IFREG.*\s*.*name:(.*)\nz//r zcannot be retrieved)rP rQ r rw rR rS r lowerr r r| findallrV
startswithr` r Z popenfiler~ )r rZ rL rM Z procfilesrb rt r0 r6 r6 r7
open_files s$ #zProcess.open_filesc C s6 | j d k r d St t j d | j | j f S)Nr z%s/%s/fd)rw rX rG r r )r r6 r6 r7 num_fds% s zProcess.num_fdsc C s t j t j | j S)N)r Zpctxswr: Zproc_num_ctx_switchesrw )r r6 r6 r7 num_ctx_switches+ s zProcess.num_ctx_switchesNc C s t j | j | | j S)N)r Zwait_pidrw r )r timeoutr6 r6 r7 wait0 s z Process.waitc
C st y" t j | j \ } } } } Wn9 t k
r] t | j sV t | j | j Yn Xt j | | | | S)N) r: Zproc_io_countersrw r r r r r Zpio)r rcZwcrbwbr6 r6 r7 io_counters4 s "
zProcess.io_counters)%__name__
__module__ __qualname____doc__ __slots__r r r r
r r r r r r r r r HAS_THREADSr r r r r r r rE r r r Zmemory_full_infor" r r r r r r6 r6 r6 r7 r b sD r rf )ir r r rG r| rP r4 collectionsr socketr r] r r r r: r r r r
r r r
r r r Z_compatr _exceptionsr r r Z__extra__all__hasattrr rH Z PAGE_SIZEZAF_LINKZSIDLZ STATUS_IDLEZSZOMBZ
STATUS_ZOMBIEZSACTIVEZSTATUS_RUNNINGZSSWAPZSSTOPZSTATUS_STOPPEDr ZTCPS_ESTABLISHEDZCONN_ESTABLISHEDZ
TCPS_SYN_SENTZ
CONN_SYN_SENTZ
TCPS_SYN_RCVDZ
CONN_SYN_RECVZTCPS_FIN_WAIT_1ZCONN_FIN_WAIT1ZTCPS_FIN_WAIT_2ZCONN_FIN_WAIT2ZTCPS_TIME_WAITZCONN_TIME_WAITZ TCPS_CLOSEDZ
CONN_CLOSEZTCPS_CLOSE_WAITZCONN_CLOSE_WAITZ
TCPS_LAST_ACKZ
CONN_LAST_ACKZ TCPS_LISTENZ CONN_LISTENZ TCPS_CLOSINGZ CONN_CLOSINGZPSUTIL_CONN_NONEZ CONN_NONErj dictr r& Zpfullmemr' r* r/ rg _fieldsr2 r8 r; r= rE rB rJ r[ r\ Zdisk_io_countersr_ r^ rz Znet_io_countersrh r{ r r r r r objectr r6 r6 r6 r7 <module> s
! %