Cs[�v � 3 @ s� d Z d d l Z d d l Z d d l Z d d l Z d d l j j 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 d l m" Z" d d l m# Z# g Z$ e r�e j% e j& e j' e j( e j) e j* e j+ e j, e j- e j. e j/ e j0 e j1 e j2 i Z3 n� e s�e rZe j% e j& e j) e j* e j+ e j, e j4 e j. e j- e j. e j' e j5 e j6 e j( i Z3 n` e r�e j% e j& e j7 e j( e j8 e j. e j+ e j, e j- e j. e j4 e j9 e j: e j; i Z3 e j< e j= e j> e j? e j@ e jA e jB e jC e jD e jE e jF e jG e jH e jI e jJ e jK e jL e jM e jN e jO e jP e jQ e jR e jS i ZT e rhe jU d � ZV n e jU d � ZV e jW ZW eX d d d d d d d d d d! d" d# d$ d% d& d' d( d) d* d+ d, d- d. d/ d0 d1 d2 d3 d4 d5 d6 d7 d8 d9 d: d; d< d= d> d? d@ dA dB dC dD dE dF dG dH dI � ZY e
dJ dK dL dM dN dO dP dQ dR dS dT dU g � ZZ e
dV dW dX dY dZ d[ g � Z[ e
d\ d< d> d] d^ d_ g � Z\ e\ Z] e
d` dW dY da db g � Z^ e
dc dd � Z_ e
de df � Z` e r�e
dg dh di dj dk dl dm dn g � Za n e
dg dh di dj dk g � Za do dp � Zb dq dr � Zc ds dt � Zd ee e du � rTdv du � Zf n dw du � Zf dx ef _g dy dz � Zh e s�e r�d{ d| � Zi n d} d| � Zi d~ d � Zj dx d� d� � Zk e jl Zl e jm Zm e jn Zn e jo Zo d� d� � Zp d� d� � Zq e rd� d� � Zr d� d� � Zs d� d� � Zt e d� d� � � Zu d� d� � Zv e sGe rVd� d� � Zw n e jw Zw d� d� � Zx e jy d� d� � � Zz Gd� d� � d� e{ � Z| d S)�z5FreeBSD, OpenBSD and NetBSD platforms implementation.� N)�
namedtuple)�AF_INET� )�_common)�_psposix)� _psutil_bsd)�
_psutil_posix)�AF_INET6)� conn_tmap)�FREEBSD)�memoize)�memoize_when_activated)�NETBSD)�OPENBSD)�sockfam_to_enum)�socktype_to_enum)�
usage_percent)�which)� AccessDenied)�
NoSuchProcess)�
ZombieProcess� SC_PAGESIZE� SC_PAGE_SIZE�ppid�status�real_uid� �
effective_uid� � saved_uid� �real_gid� �
effective_gid� � saved_gid� �ttynr� � create_time� �ctx_switches_vol�
�ctx_switches_unvol� �
read_io_count� �write_io_count�
� user_time� �sys_time� � ch_user_time� � ch_sys_time� �rss� �vms� �memtext� �memdata� �memstack� �cpunum� �name� �svmem�totalZ available�percent�used�free�active�inactive�buffers�cached�shared�wired� scputimes�user�nice�system�idle�irq�pmem�text�data�stack� pcputimes�
children_user�children_system�
pmmap_groupedz*path rss, private, ref_count, shadow_count� pmmap_extz6addr, perms path rss, private, ref_count, shadow_count�sdiskioZ
read_countZ write_countZ
read_bytesZ write_bytesZ read_timeZ
write_timeZ busy_timec C s t j � } | \ } } } } } } } } t r� t d d � �m } xc | D][ }
|
j d � r{ t |
j � d � d } qI |
j d � rI t |
j � d � d } qI WWd QRX| | | } | | | } t | | | d d �}
t | | |
| | | | | | | | � S) z&System virtual memory as a namedtuple.z
/proc/meminfo�rbs Buffers:r i s
MemShared:N�round_) �cextZ virtual_memr �open�
startswith�int�splitr rI )ZmemrJ rM rN rO rS rQ rP rR �f�lineZavailrL rK � rm �
/_psbsd.py�virtual_memory� s
%ro c C sL t j � \ } } } } } t | | d d �} t j | | | | | | � S)z@System swap memory as (total, used, free, sin, sout) namedtuple.re r )rf Zswap_memr r Zsswap)rJ rL rM �sinZsoutrK rm rm rn � swap_memory� s rq c C s1 t j � \ } } } } } t | | | | | � S)z+Return system per-CPU times as a namedtuple)rf � cpu_timesrT )rU rV rW rX rY rm rm rn rr � s rr �
per_cpu_timesc C s[ g } xN t j � D]@ } | \ } } } } } t | | | | | � } | j | � q W| S)z'Return system CPU times as a namedtuple)rf rs rT �append)�retZcpu_trU rV rW rX rY �itemrm rm rn rs � s c C sA t � d k r t � g St j r. t d � � d t _ t � g S)z'Return system CPU times as a namedtupler z&supported only starting from FreeBSD 8T)�cpu_count_logicalrr rs �
__called__�NotImplementedErrorrm rm rm rn rs � s
Fc C s
t j � S)z0Return the number of logical CPUs in the system.)rf rw rm rm rm rn rw � s rw c C s t � d k r d Sd S)Nr )rw rm rm rm rn �cpu_count_physical� s rz c
C s� d } t j � } | d k r� | j d � } | d k r� | d | d � } t j | � } z t | j d � � pw d } Wd | j � X| s� t � d k r� d S| S)z1Return the number of physical CPUs in the system.Nz </groups>r r* zgroup/children/group/cpu���) rf Zcpu_count_phys�rfind�ET�
fromstring�len�findall�clearrw )ru �s�index�rootrm rm rn rz � s c C s� t r$ t j � \ } } } } } n� t r� t j � \ } } } } } } } t d d � �= } x3 | D]+ } | j d � rd t | j � d � } qd WWd QRXn' t r� t j � \ } } } } } } } t j
| | | | � S)z*Return various CPU stats as a named tuple.z
/proc/statrd s intrr N) r rf � cpu_statsr rg rh ri rj r r Z scpustats) ZctxswZintrsZ
soft_intrsZsyscallsZtrapsZfaultsZforksrk rl rm rm rn r� s !
$!r� c C s^ g } t j � } xE | D]= } | \ } } } } t j | | | | � } | j | � q W| S)z�Return mounted disk partitions as a list of namedtuples.
'all' argument is ignored, see:
https://github.com/giampaolo/psutil/issues/906
)rf �disk_partitionsr Z sdiskpartrt ) �all�retlistZ
partitions� partitionZdeviceZ
mountpointZfstypeZopts�ntuplerm rm rn r� < s
r� c C s� t � j � } i } x� | D]� } y7 t j | � } t j | � } t j | � \ } } Wn: t k
r� } z | j t j k r� � WYd d } ~ Xq Xt t
d � r� t
j | � } t
j | | | | � | | <q W| S)z)Get NIC stats (isup, duplex, speed, mtu).N� NicDuplex)
�net_io_counters�keys�
cext_posixZ
net_if_mtuZ net_if_flagsZnet_if_duplex_speed�OSError�errnoZENODEV�hasattrr r� Z snicstats)�namesru rG ZmtuZisup�duplexZspeed�errrm rm rn � net_if_statsW s
r� c C s t r� g } x� t � D]} } y t | � j | � } Wn t t f k
rR w Yq Xx= | D]5 } t | � } | j | � | j t j | � � qZ Wq W| S| t j
k r� t d | d j d d � t
D� � f � � t
| \ } } t
� } t rt j d � } n t j � } x� | D]� } | \ } }
} } }
} } |
| k r| | k ry t | } Wn t k
r�t t j } Yn X|
t t f k r�| r�t j | � } |
r�t j |
� }
t |
� }
t | � } t j | |
| | |
| | � } | j | � qWt | � S)z System-wide network connections.z+invalid %r kind argument; choose between %sz, c S s g | ] } t | � � q Srm )�repr)�.0�xrm rm rn �
<listcomp>} s z#net_connections.<locals>.<listcomp>r r{ )r �pids�Process� connectionsr r �listrt r Zsconnr
�
ValueError�join�setr rf �net_connections� TCP_STATUSES�KeyError�PSUTIL_CONN_NONEr r �addrr r �add)�kindru �pidZcons�conn�families�types�rawlistrv �fd�fam�type�laddr�raddrr �ntrm rm rn r� k sJ
&
!r� c C s� y t j � \ } } } Wn t k
r1 d SYn X| d k } | rP t j } n" | d k rh t j } n
| d } t j | | | � S)zReturn battery info.Nr �< r{ )rf �sensors_batteryry r ZPOWER_TIME_UNLIMITEDZPOWER_TIME_UNKNOWNZsbattery)rK ZminsleftZ
power_pluggedZsecsleftrm rm rn r� � s
r� c C s
t j � S)z:The system boot time expressed in seconds since the epoch.)rf � boot_timerm rm rm rn r� � s r� c C s� g } t j � } x~ | D]v } | \ } } } } } | d k rR t sL t � d } | d k ra q t j | | ps d | | | � } | j | � q W| S)z:Return currently connected users as a list of namedtuples.r N�~r{ )rf �usersr �AssertionErrorr Zsuserrt ) r� r� rv rU Ztty�hostnameZtstampr� r� rm rm rn r� � s
!r� c C sJ y t d � j � Wn+ t k
r, d SYn t k
rA d SYn Xd Sd S)Nr FT)r� rG r r rm rm rm rn �
_pid_0_exists� s
r� c C s; t j � } t r7 d | k r7 t � r7 | j d d � | S)z7Returns a list of PIDs currently running on the system.r )rf r� r r� �insert)ru rm rm rn r� � s r� c C s* t j | � } | s" | t � k Sd Sd S)zReturn True if pid exists.TN)r �
pid_existsr� )r� �existsrm rm rn r� � s
r� c s% t j � � � f d d � � } | S)z`Decorator which translates bare OSError exceptions into
NoSuchProcess and AccessDenied.
c s y � | | | � SWn� t k
r� } z� | j d k rb d t � k r_ t | j | j � � n � | j t j 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)Nr )
r� r� r� r �_namer� �ESRCHr� r r �_ppid�EPERM�EACCES)�self�args�kwargsr� )�funrm rn �wrapper� s z wrap_exceptions.<locals>.wrapper)� functools�wraps)r� r� rm )r� rn �wrap_exceptions� s !r� c c s� y d VWn� t k
r� } z� | 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)z8Same as above, for routines relying on reading /proc fs.N)
�EnvironmentErrorr� �ENOENTr� r� r� r r� r r� r� r� r )�instr� rm rm rn �wrap_exceptions_procfs s r� c @ s� e Z d Z d Z d d d g Z d d � Z e 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 � � Z e Z e
d! d" � � Z e
d# d$ � � Z e
d% d& � � Z e
d' d( � � Z e
d) d* d+ � � Z e
d, d- d. � � Z e
d/ d0 � � Z e
d1 d2 � � Z e
d3 d4 � � Z e
d5 d6 � � Z e
d7 d8 � � Z! e" d9 d: � Z# e" d9 d; � Z$ d< d= � Z% e& e' d> � r1e
d? d@ � � Z( n e% Z( e& e' dA � r[e
dB dC � � Z) n e% Z) e r�e
dD dE � � Z* e
dF dG � � Z+ e
dH dI � � Z, d, S)Jr� z1Wrapper class around underlying C implementation.r� r� r� c C s | | _ d | _ d | _ d S)N)r� r� r� )r� r� rm rm rn �__init__( s zProcess.__init__c C s4 t j | j � } t | � t t � k s0 t � | S)z;Retrieves multiple process info in one shot as a raw tuple.)rf Zproc_oneshot_infor� r �kinfo_proc_mapr� )r� ru rm rm rn �oneshot- s zProcess.oneshotc C s | j j � d S)N)r� Zcache_activate)r� rm rm rn �
oneshot_enter4 s zProcess.oneshot_enterc C s | j j � d S)N)r� Zcache_deactivate)r� rm rm rn � oneshot_exit7 s zProcess.oneshot_exitc C s4 | j � t d } | d k r$ | St j | j � S)NrG )r� r� rf Z proc_namer� )r� rG rm rm rn rG : s z Process.namec
C s� t r t j | j � St rZ | j d k r/ d St | � � t j d | j � SWd QRXn$ | j � } | rz t | d � Sd Sd S)Nr � z /proc/%s/exe)
r rf Zproc_exer� r r� �os�readlink�cmdliner )r� r� rm rm rn �exe? s
z Process.exec C s� t r | j d k r g St r� y t j | j � SWq� t k
r� } z_ | j t j k r� t | j � s� t | j | j
� � q� t | j | j
| j � � n � WYd d } ~ Xq� Xn t j | j � Sd S)Nr )
r r� r rf Z proc_cmdliner� r� �EINVALr� r r� r r� )r� r� rm rm rn r� U s zProcess.cmdlinec C sI | j � t d } t j � } y | | SWn t k
rD d SYn Xd S)Nr' )r� r� r Zget_terminal_mapr� )r� Ztty_nrZtmaprm rm rn �terminalk s
zProcess.terminalc C s | j � t d | _ | j S)Nr )r� r� r� )r� rm rm rn r t s z Process.ppidc C s7 | j � } t j | t d | t d | t d � S)Nr r r )r� r Zpuidsr� )r� �rawtuplerm rm rn �uidsy s
z Process.uidsc C s7 | j � } t j | t d | t d | t d � S)Nr! r# r% )r� r Zpgidsr� )r� r� rm rm rn �gids� s
z Process.gidsc C sB | j � } t j | t d | t d | t d | t d � S)Nr3 r5 r7 r9 )r� r r^ r� )r� r� rm rm rn rr � s zProcess.cpu_timesc C s | j � t d S)NrE )r� r� )r� rm rm rn �cpu_num� s zProcess.cpu_numc C sJ | j � } t | t d | t d | t d | t d | t d � S)Nr; r= r? rA rC )r� rZ r� )r� r� rm rm rn � memory_info� s zProcess.memory_infoc C s | j � t d S)Nr) )r� r� )r� rm rm rn r) � s zProcess.create_timec C s3 t t d � r t j | j � St | j � � Sd S)N�proc_num_threads)r� rf r� r� r �threads)r� rm rm rn � num_threads� s zProcess.num_threadsc C s, | j � } t j | t d | t d � S)Nr+ r- )r� r Zpctxswr� )r� r� rm rm rn �num_ctx_switches� s zProcess.num_ctx_switchesc C sh t j | j � } g } x9 | D]1 \ } } } t j | | | � } | j | � q Wt rd | j � | S)N)rf Z proc_threadsr� r Zpthreadrt r rG )r� r� r� Z thread_id�utimeZstimer� rm rm rn r� � s
zProcess.threadsZinetc C sy | t k r8 t d | d j d d � t D� � f � � t r�t | \ } } t � } t j | j � } x| D]� } | \ } } } }
} } }
|
| j k s� t � | | k rp | | k rp y t | } Wn t
k
r� t t j } Yn X| t t
f k r*|
rt j |
� }
| r*t j | � } t | � } t | � } t j | | | |
| | � } | j | � qp W| j � t | � St | \ } } t j | j | | � } g } x� | D]� } | \ } } } }
} } | t t
f k r|
r�t j |
� }
| rt j | � } t | � } t | � } t | } t j | | | |
| | � } | j | � q�Wt ru| j � | S)Nz+invalid %r kind argument; choose between %sz, c S s g | ] } t | � � q Srm )r� )r� r� rm rm rn r� � s z'Process.connections.<locals>.<listcomp>)r
r� r� r r� rf r� r� r� r� r� r� r r r r� r r Zpconnr� rG r� Zproc_connectionsrt r )r� r� r� r� ru r� rv r� r� r� r� r� r r� r� rm rm rn r� � sX &
zProcess.connectionsNc C s t j | j | | j � S)N)r Zwait_pidr� r� )r� �timeoutrm rm rn �wait� s z Process.waitc C s t j | j � S)N)r� � getpriorityr� )r� rm rm rn �nice_get s zProcess.nice_getc C s t j | j | � S)N)r� � setpriorityr� )r� �valuerm rm rn �nice_set s zProcess.nice_setc C s$ | j � t d } t j | d � S)Nr �?)r� r� �
PROC_STATUSES�get)r� �coderm rm rn r s zProcess.statusc C s2 | j � } t j | t d | t d d d � S)Nr/ r1 r r{ r{ )r� r Zpior� )r� r� rm rm rn � io_counters s zProcess.io_countersc
C s� t r | j d k r d St rJ t | � � t j d | j � SWd QRXn= t t d � ro t j | j � pn d St t
r~ d n d � � d S)z)Return process current working directory.r Nz /proc/%s/cwd�proc_open_filesz&supported only starting from FreeBSD 8r� ) r r� r r� r� r� r� rf Zproc_cwdry r )r� rm rm rn �cwd s
z Process.cwdZmmapz*path rss, private, ref_count, shadow_countz6addr, perms path rss, private, ref_count, shadow_countc C s
t � d S)N)ry )r� rm rm rn �_not_implemented/ s zProcess._not_implementedr� c C s# t j | j � } d d � | D� S)z8Return files opened by process as a list of namedtuples.c S s( g | ] \ } } t j | | � � q Srm )r Z popenfile)r� �pathr� rm rm rn r� 9 s z&Process.open_files.<locals>.<listcomp>)rf r� r� )r� r� rm rm rn �
open_files5 s zProcess.open_files� proc_num_fdsc C s& t j | j � } t r" | j � | S)z=Return the number of file descriptors opened by this process.)rf r� r� r rG )r� ru rm rm rn �num_fds@ s
zProcess.num_fdsc C s t j | j � S)N)rf Zproc_cpu_affinity_getr� )r� rm rm rn �cpu_affinity_getP s zProcess.cpu_affinity_getc C s� t t t t � � � � } x0 | D]( } | | k r" t d | | f � � q" Wy t j | j | � Wnv t k
r� } zV | j t j
t j f k r� x0 | D]( } | | k r� t d | | f � � q� W� WYd d } ~ Xn Xd S)Nz#invalid CPU #%i (choose between %s)) �tuple�ranger rs r� rf Zproc_cpu_affinity_setr� r� r� r� ZEDEADLK)r� ZcpusZallcpusZcpur� rm rm rn �cpu_affinity_setT s
zProcess.cpu_affinity_setc C s t j | j � S)N)rf Zproc_memory_mapsr� )r� rm rm rn � memory_mapsm s zProcess.memory_maps)-�__name__�
__module__� __qualname__�__doc__� __slots__r� r
r� r� r� r� rG r� r� r� r r� r� rr r r� r� Zmemory_full_infor) r� r� r� r� r� r� r� r r� r� r Znt_mmap_groupedZ nt_mmap_extr� r� rf r� r� r� r� r� rm rm rm rn r� # s\
6
r� )}r �
contextlibr� r� r� Zxml.etree.ElementTreeZetreeZ ElementTreer} � collectionsr �socketr r� r r r rf r r� r r
r r r
r r r r r Z_compatr � _exceptionsr r r Z__extra__all__ZSIDLZ STATUS_IDLEZSRUNZSTATUS_RUNNINGZSSLEEPZSTATUS_SLEEPINGZSSTOPZSTATUS_STOPPEDZSZOMBZ
STATUS_ZOMBIEZSWAITZSTATUS_WAITINGZSLOCKZ
STATUS_LOCKEDr� ZSDEADZ
STATUS_WAKINGZSONPROCZSACTIVEZSDYINGZ STATUS_DEADZ
SSUSPENDEDZSTATUS_SUSPENDEDZTCPS_ESTABLISHEDZCONN_ESTABLISHEDZ
TCPS_SYN_SENTZ
CONN_SYN_SENTZTCPS_SYN_RECEIVEDZ
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_CLOSINGr� Z CONN_NONEr� �sysconfZPAGESIZEZAF_LINK�dictr� rI rT rZ Zpfullmemr^ ra rb rc ro rq rr r� rs rx rw rz r� r� �
disk_usageZdisk_io_countersr� Z net_if_addrsr� r� r� r� r� r� r� r� r� �contextmanagerr� �objectr� rm rm rm rn �<module> s
# 6