Cs[jc � @ s� 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 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 d g Z e j d � Z e
j Z e j! dk k Z" d Z# d Z$ e j% 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 i Z2 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 e
jH e jI e
jJ e jK e# e jL e$ i ZM eN d d d d d d d d! d" d# d$ d% d& d' d( d) d* d+ d, d- d. d/ d0 d1 � ZO e d2 d3 d4 d5 d6 g � ZP e d7 d3 d4 d8 d9 g � ZQ e d: d; d< d= d> d? g � ZR e d@ d d g � ZS eS ZT e dA dB d dC dD g � ZU e dE dF dG jV eU jW � � ZX dH dI � ZY dJ dK � ZZ dL dM � Z[ dN dO � Z\ dP dQ � Z] dR dS � Z^ dT dU � Z_ dV dW � Z` e ja Za e jb Zb dX dY dZ � Zc e jd Zd e
je Ze dl d[ d\ � Zf d] d^ � Zg d_ d` � Zh da db � Zi dc dd � Zj de df � Zk dg dh � Zl Gdi dj � dj em � Zn d S)mz'Sun OS Solaris platform implementation.� N)�
namedtuple)�AF_INET� )�_common)�_psposix)�
_psutil_posix)�
_psutil_sunos)�AF_INET6)�
isfile_strict)�memoize_when_activated)�sockfam_to_enum)�socktype_to_enum)�
usage_percent)�b)�PY3)� AccessDenied)�
NoSuchProcess)�
ZombieProcess� CONN_IDLE�
CONN_BOUND� PROCFS_PATH� SC_PAGE_SIZE� � ZIDLEZBOUND�ppid�rss�vms� create_time� �nice� � num_threads� �status� �ttynr� �uid� �euid� �gid�
�egid� � scputimes�user�systemZidleZiowait� pcputimes�
children_user�children_system�svmem�totalZ available�percent�used�free�pmem�
pmmap_grouped�pathZ anonymous�locked� pmmap_extz addr perms � c C s t j d j S)z+Return updated psutil.PROCFS_PATH constant.Zpsutil)�sys�modulesr � rB rB � /_pssunos.py�get_procfs_pathu s rD c C s_ t j d � t } t j d � t } } | | } t | | d d �} t | | | | | � S)z Report virtual memory metrics.�
SC_PHYS_PAGES�SC_AVPHYS_PAGES�round_r )�os�sysconf� PAGE_SIZEr r5 )r6 r9 Zavailr8 r7 rB rB rC �virtual_memory s
rK c
C su t j � \ } } t j d d t j d d d g d t j �} | j � \ } } t rn | j t
j j � } | j
d k r� t d | j
� � | j � j d � d
d � } | s� t d � � d } } xd | D]\ } | j � } | d d � \ } }
| t t | � d � 7} | t t |
� d � 7} q� W| | } t | | d d
�} t j | | | | | t | t � S)zReport swap memory metrics.z /usr/bin/envzPATH=/usr/sbin:/sbin:%s�PATHZswapz-l�stdoutr z 'swap -l' failed (retcode=%s)�
r Nz no swap device(s) configuredr i rG ���)�cextZswap_mem�
subprocess�PopenrH �environ�PIPE� communicater �decoder@ rM �encoding�
returncode� RuntimeError�strip�split�intr r ZsswaprJ )
�sinZsout�prM �stderr�linesr6 r9 �line�t�fr8 r7 rB rB rC � swap_memory� s,
rd 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 SrB )�sum)�.0�xrB rB rC �
<listcomp>� s z cpu_times.<locals>.<listcomp>)rP �
per_cpu_timesr/ �zip)�retrB rB rC � cpu_times� s rl 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 SrB )r/ )rf rg rB rB rC rh � s z!per_cpu_times.<locals>.<listcomp>)rP ri )rk rB rB rC ri � s ri 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)rH rI �
ValueErrorrB rB rB rC �cpu_count_logical� s
ro c C s
t j � S)z1Return the number of physical CPUs in the system.)rP Zcpu_count_physrB rB rB rC �cpu_count_physical� s rp c C s4 t j � \ } } } } d } t j | | | | � S)z*Return various CPU stats as a named tuple.r )rP � cpu_statsr Z scpustats)Z ctx_switchesZ
interruptsZsyscallsZtrapsZsoft_interruptsrB rB rC rq � s rq 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� )rP �disk_partitions�
disk_usager6 r Z sdiskpart�append) �all�retlistZ
partitions� partitionZdeviceZ
mountpointZfstypeZoptsZntuplerB rB rC rs � s
rs c C s� t j j � } | d k r+ | j d d � | | k rc t d | d j d d � | D� � f � � t j | \ } } t j | � } t � } x | D]� } | \ } } }
} } }
} | | k r� q� |
| k r� q� | t t
f k r| r� t j | � } | rt j | � } t |
}
t
| � } t |
� }
| d k rbt j | | |
| | |
| � } n t j | | |
| | |
� } | j | � q� Wt | � S)
z�Return socket connections. If pid == -1 return system-wide
connections (as opposed to connections opened by one process only).
Only INET sockets are returned (UNIX are not).
r �unixr z+invalid %r kind argument; choose between %sz, c S s g | ] } t | � � q SrB )�repr)rf rg rB rB rC rh s z#net_connections.<locals>.<listcomp>���r{ )r Z conn_tmap�copy�poprn �joinrP �net_connections�setr r �addr� TCP_STATUSESr r
Zsconn�pconn�add�list)�kind�_pidZcmap�families�types�rawlistrk �item�fdZfamZtype_�laddr�raddrr# �pid�ntrB rB rC r � s8 &
$ r c C sy t j � } xf | j � D]X \ } } | \ } } } } t t d � rU t j | � } t j | | | | � | | <q W| S)z)Get NIC stats (isup, duplex, speed, mtu).� NicDuplex)rP � net_if_stats�items�hasattrr r� Z snicstats)rk �namer� Zisup�duplexZspeedZmturB rB rC r� s r� c C s
t j � S)z:The system boot time expressed in seconds since the epoch.)rP � boot_timerB rB rB rC 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� )rP �usersr Zsuserru ) rw r� r� r� r0 �tty�hostnameZtstampZ user_processr� r� rB rB rC r� 2 s
r� c C s# d d � t j t t � � � D� S)z7Returns a list of PIDs currently running on the system.c S s( g | ] } | j � r t | � � q SrB )�isdigitr\ )rf rg rB rB rC rh L s zpids.<locals>.<listcomp>)rH �listdirr rD rB rB rB rC �pidsJ s r� c C s
t j | � S)z&Check for the existence of a unix pid.)r �
pid_exists)r� rB rB rC r� O s r� c s � f d d � } | S)z�Call 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� | j d k rb d t � k r_ t | j | j � � n � | 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)Nr )�EnvironmentErrorr� r� r �_name�errno�ENOENTZESRCHr� r r �_ppid�EPERMZEACCES)�self�args�kwargs�err)�funrB rC �wrapperY s z wrap_exceptions.<locals>.wrapperrB )r� r� rB )r� rC �wrap_exceptionsT s r� c @ sh 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 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 d. d/ � � Z e d0 d1 � � Z e Z e d2 d3 � � Z e d4 d5 � � Z e d6 d7 � � Z d8 d9 � Z! e d: d; d< � � Z" e# d= d> � Z$ e# d= d? � Z% e d@ dA � � Z& e dB dC � � Z' e dD dE � � Z( e dF dG dH � � Z) dF S)I�Processz1Wrapper class around underlying C implementation.r� r� r� � _procfs_pathc C s+ | | _ d | _ d | _ t � | _ d S)N)r� r� r� rD r� )r� r� rB rB rC �__init__u 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� rB rB rC �
oneshot_enter{ s
zProcess.oneshot_enterc C s+ | j j � | j j � | j j � d S)N)r� Zcache_deactivater� r� )r� rB rB rC � oneshot_exit� s
zProcess.oneshot_exitc C s t j | j | j � S)N)rP Zproc_name_and_argsr� r� )r� rB rB rC r� � s zProcess._proc_name_and_argsc C s: t j | j | j � } t | � t t � k s6 t � | S)N)rP Zproc_basic_infor� r� �len�
proc_info_map�AssertionError)r� rk rB rB rC r� � s zProcess._proc_basic_infoc C s t d d � � } | | � S)Nc S s t j | j | j � S)N)rP � proc_credr� r� )r� rB rB rC r� � s z%Process._proc_cred.<locals>.proc_cred)r� )r� r� rB rB rC r� � s zProcess._proc_credc C s | j � d S)Nr )r� )r� rB rB rC r� � s z Process.namec C sD y! t j d | j | j f � SWn t k
r5 Yn X| j � d S)Nz%s/%s/path/a.outrr )rH �readlinkr� r� �OSError�cmdline)r� rB rB rC �exe� s
z Process.exec C s | j � d j d � S)Nr r? )r� r[ )r� rB rB rC r� � s zProcess.cmdlinec C s t j | j | j � S)N)rP Z proc_environr� r� )r� rB rB rC rS � s zProcess.environc C s | j � t d S)Nr )r� r� )r� rB rB rC r � s zProcess.create_timec C s | j � t d S)Nr! )r� r� )r� rB rB rC r! � s zProcess.num_threadsc C s | j � t d S)Nr )r� r� )r� rB rB rC �nice_get� s zProcess.nice_getc C s7 | j d k r$ t | j | j � � t j | j | � S)Nr r )r r )r� r r� �
cext_posix� setpriority)r� �valuerB rB rC �nice_set� s zProcess.nice_setc C s | j � t d | _ | j S)Nr )r� r� r� )r� rB rB rC r � s z Process.ppidc C sx y" | j � \ } } } } } } Wn@ t k
rd | j � t d } | j � t d } d } Yn Xt j | | | � S)Nr' r) )r� r r� r� r �puids)r� �real� effective�saved�_rB rB rC �uids� s "
z Process.uidsc C sx y" | j � \ } } } } } } Wn@ t k
rd | j � t d } | j � t d } d } Yn Xt j | | | � S)Nr+ r- )r� r r� r� r r� )r� r� r� r� r� rB rB rC �gids� s "
z Process.gidsc C sv y t j | j | j � } WnJ t k
rh } z* | j t j k rS t rS d } n � WYd d } ~ Xn Xt j | � S)N� )r� r� r� r� )
rP Zproc_cpu_timesr� r� r� r� � EOVERFLOW� IS_64_BITr r2 )r� �timesr� rB rB rC rl � s zProcess.cpu_timesc C s t j | j | j � S)N)rP Z proc_cpu_numr� r� )r� rB rB rC �cpu_num� s zProcess.cpu_numc C s� | j } d } t | j � t d � } | t j k r� xu d
D]m } y! t j d | | j | f � SWq? t k
r� } z# | j
t
j k r� d } w? � WYd d } ~ Xq? Xq? W| r� t j d | | j f � d S) NFr% r r r � z
%s/%d/path/%dTz%s/%s)r r r r� )
r� r� r� r� rP ZPRNODEVrH r� r� r� r� r� �stat)r� � procfs_path�
hit_enoentr� rg r� rB rB rC �terminal� s
zProcess.terminalc C s� | j } y t j d | | j f � SWnX t k
r� } z8 | j t j k rl t j d | | j f � d S� WYd d } ~ Xn Xd S)Nz%s/%s/path/cwdz%s/%s)r� rH r� r� r� r� r� r� )r� r� r� rB rB rC �cwd s z Process.cwdc C s= | j � } | t d d } | t d d } t | | � S)Nr i r )r� r� r: )r� rk r r rB rB rC � memory_info s zProcess.memory_infoc C s$ | j � t d } t j | d � S)Nr# �?)r� r� �
PROC_STATUSES�get)r� �coderB rB rC r# $ s zProcess.statusc
C s | j } g } t j d | | j f � } d } x� | D]� } t | � } y" t j | j | | � \ } } Wn_ t k
r� } z? | j t j k r� t
r� w8 | j t j k r� d } w8 � WYd d } ~ Xq8 Xt j
| | | � } | j | � q8 W| rt j d | | j f � | S)Nz %s/%d/lwpFTz%s/%s)r� rH r� r� r\ rP Zquery_process_threadr� r� r� r� r� r Zpthreadru r� )
r� r� rk Ztidsr� �tid�utimeZstimer� r� rB rB rC �threads* s*
zProcess.threadsc C s g } d } | j } d | | j f } x� t j d | | j f � D]� } t j j | | � } t j j | � rE y t j | � } WnC t k
r� } z# | j t j
k r� d } wE � WYd d } ~ XqE Xt | � rE | j t
j | t | � � � qE W| rt j d | | j f � | S)NFz
%s/%d/pathz%s/%d/fdTz%s/%s)r� r� rH r� r< r~ �islinkr� r� r� r� r
ru r Z popenfiler\ r� ) r� rw r� r� Zpathdirr� r< �filer� rB rB rC �
open_filesL s& # #zProcess.open_filesc c s� d | } t j | d d d t j d t j �} | j � \ } } t rh d d � | | f D� \ } } | j d k r� d | j � k r� t | j | j � � d
| j � k r� t
| j | j � � t d | | f � � | j d � d
d � } x� t
| � D]� \ } } | j � } | j d � r| j d d
� d
} | | d
j � }
|
d k rjt j }
n |
d k r�t j }
n d }
d t j |
| d t j f VqWd S)z<Get UNIX sockets used by process by parsing 'pfiles' output.z pfiles %s�shellTrM r_ c S s% g | ] } | j t j j � � q SrB )rV r@ rM rW )rf rg rB rB rC rh n s z-Process._get_unix_sockets.<locals>.<listcomp>r zpermission deniedzno such processz%r command error
%srN r Nzsockname: AF_UNIXr? � SOCK_STREAM�
SOCK_DGRAMr rr r{ r{ )rQ rR rT rU r rX �lowerr r� r� r rY r[ � enumerate�lstrip�
startswithrZ �socketr� r� �AF_UNIXr � CONN_NONE) r� r� �cmdr^ rM r_ r` �ira r< �typerB rB rC �_get_unix_socketse s2
zProcess._get_unix_socketsZinetc C sn t | d | j �} | s8 t j d | j | j f � | d k rj | j d d � | j | j � D� � | S)Nr� z%s/%srv ry c S s g | ] } t j | � � q SrB )r r� )rf �connrB rB rC rh � s z'Process.connections.<locals>.<listcomp>)rv ry )r r� rH r� r� �extendr� )r� r� rk rB rB rC � connections� s zProcess.connectionsZmmapzpath rss anon lockedz addr perms path rss anon lockedc $ C s� d d � } | j } g } y t j | j | � } WnE t k
r{ } z% | j t j k rf t rf g S� WYd d } ~ Xn Xd } x� | D]� } | \ } } }
} } }
} | | | � } | j d � sGy# t j
d | | j | f � } WnY t k
rF} z9 | j t j k r1d | | j | f } d } n � WYd d } ~ Xn X| j | |
| | |
| f � q� W| r�t j
d | | j f � | S)Nc S s@ d t | � d d � j d � t | � d d � j d � f S)Nz%s-%sr �L)�hexrZ )�start�endrB rB rC �toaddr� s z#Process.memory_maps.<locals>.toaddrF�[z
%s/%s/path/%sTz%s/%s)r� rP Zproc_memory_mapsr� r� r� r� r� r� rH r� r� ru r� )r� r� r� rw r� r� r� r� r� ZaddrsizeZpermr� r Zanonr= rB rB rC � memory_maps� s4
#zProcess.memory_mapsc C s# t t j d | j | j f � � S)Nz%s/%s/fd)r� rH r� r� r� )r� rB rB rC �num_fds� s zProcess.num_fdsc C s t j t j | j | j � � S)N)r ZpctxswrP Zproc_num_ctx_switchesr� r� )r� rB rB rC �num_ctx_switches� s zProcess.num_ctx_switchesNc C s t j | j | | j � S)N)r Zwait_pidr� r� )r� �timeoutrB rB rC �wait� s z Process.wait)*�__name__�
__module__� __qualname__�__doc__� __slots__r� r� r� r r� r� r� r� r� r� r� rS r r! r� r� r r� r� rl r� r� r� r� Zmemory_full_infor# r� r� r� r� r Znt_mmap_groupedZ nt_mmap_extr� r� r� r� rB rB rB rC r� p sL
" 0r� l r{ )or r� rH r� rQ r@ � collectionsr r rr r r r r� r rP r r
r r r
r Z_compatr r � _exceptionsr r r Z__extra__all__rI rJ ZAF_LINK�maxsizer� r r ZSSLEEPZSTATUS_SLEEPINGZSRUNZSTATUS_RUNNINGZSZOMBZ
STATUS_ZOMBIEZSSTOPZSTATUS_STOPPEDZSIDLZ STATUS_IDLEZSONPROCZSWAITZSTATUS_WAITINGr� 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_NONEr� Z TCPS_IDLEZ
TCPS_BOUNDr� �dictr� r/ r2 r5 r: Zpfullmemr; r~ �_fieldsr> rD rK rd rl ri ro rp rq Zdisk_io_countersrt rs Znet_io_countersZ net_if_addrsr r� r� r� r� r� r� �objectr� rB rB rB rC �<module> s�
&
$