Cs[6D � @ s� d 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 g 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) e
j* e
j+ e
j, e
j- e
j. e
j/ e
j0 e
j1 e
j2 e
j3 e
j4 e
j5 e
j6 e
j7 i Z8 e
j9 e
j: e
j; e
j< e
j= e
j> e
j? e
j@ e
jA e
jB i ZC eD d d d d d d d d d d d d d d! d" d# d$ d% d& d' d( d) � ZE eD d* d d+ d d, d d- d d. d d/ d d0 d! d1 d# � ZF e d2 d3 d4 d5 d6 g � ZG e d7 d8 d9 d: d; d< d= d> d? g � ZH e d@ d, d- d. d/ g � ZI e dA eI jJ dn � ZK e dC dD � ZL e dE dF dG jM eL jJ � � ZN dH dI � ZO dJ dK � ZP dL dM � ZQ dN dO � ZR dP dQ � ZS dR dS � ZT dT dU � ZU dV dW � ZV e jW ZW e
jX ZX dX dY dZ � ZY d[ d\ � ZZ e
j[ Z[ e j\ Z\ d] d^ d_ � Z] d` da � Z^ db dc � Z_ dd de � Z` df dg � Za e jb Zb dh di � Zc e jd dj dk � � Ze Gdl dm � dm ef � Zg d S)oz macOS platform implementation.� N)�AF_INET)�
namedtuple� )�_common)�_psposix)� _psutil_osx)�
_psutil_posix)�AF_INET6)� conn_tmap)�
isfile_strict)�memoize_when_activated)�parse_environ_block)�sockfam_to_enum)�socktype_to_enum)�
usage_percent)� AccessDenied)�
NoSuchProcess)�
ZombieProcess� SC_PAGE_SIZE�ppid�ruid�euid� �suid� �rgid� �egid� �sgid� �ttynr� �ctime� �status� �name�
�cpuutime�cpustime�rss�vms�pfaults�pageins�
numthreads�volctxsw� scputimes�user�nice�system�idle�svmem�totalZ available�percent�used�free�active�inactive�wired�pmem�pfullmem�uss�
pmmap_groupedz7path rss private swapped dirtied ref_count shadow_depth� pmmap_extz addr perms � c C sk t j � \ } } } } } | | } | | | } t | | | d d �} t | | | | | | | | � S)z&System virtual memory as a namedtuple.�round_r )�cextZ virtual_memr r6 )r7 r; r< r= r: Zavailr9 r8 � rF �
/_psosx.py�virtual_memoryx s
rH c C sL t j � \ } } } } } t | | d d �} t j | | | | | | � S)z=Swap system memory as a (total, used, free, sin, sout) tuple.rD r )rE Zswap_memr r Zsswap)r7 r9 r: �sinZsoutr8 rF rF rG � swap_memory� s rJ c C s+ t j � \ } } } } t | | | | � S)z(Return system CPU times as a namedtuple.)rE � cpu_timesr1 )r2 r3 r4 r5 rF rF rG rK � s rK c C sU g } xH t j � D]: } | \ } } } } t | | | | � } | j | � q W| S)z(Return system CPU times as a named tuple)rE �
per_cpu_timesr1 �append)�retZcpu_tr2 r3 r4 r5 �itemrF rF rG rL � s rL c C s
t j � S)z0Return the number of logical CPUs in the system.)rE �cpu_count_logicalrF rF rF rG rP � s rP c C s
t j � S)z1Return the number of physical CPUs in the system.)rE Zcpu_count_physrF rF rF rG �cpu_count_physical� s rQ c C s1 t j � \ } } } } } t j | | | | � S)N)rE � cpu_statsr Z scpustats)Z ctx_switchesZ
interruptsZsoft_interruptsZsyscallsZtrapsrF rF rG rR � s rR c C s+ t j � \ } } } t j | | | � g S)z�Return CPU frequency.
On macOS per-cpu frequency is not supported.
Also, the returned frequency never changes, see:
https://arstechnica.com/civis/viewtopic.php?f=19&t=465002
)rE �cpu_freqr Zscpufreq)ZcurrZmin_Zmax_rF rF rG rS � s rS Fc C s� g } t j � } x� | D]~ } | \ } } } } | d k rC d } | sr t j j | � s t j j | � rr q t j | | | | � } | j | � q W| S)z8Return mounted disk partitions as a list of namedtuples.Znone� ) rE �disk_partitions�os�path�isabs�existsr Z sdiskpartrM ) �all�retlistZ
partitions� partitionZdeviceZ
mountpointZfstypeZopts�ntuplerF rF rG rU � s
&rU 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 information.Nr �< ���)rE �sensors_battery�NotImplementedErrorr ZPOWER_TIME_UNLIMITEDZPOWER_TIME_UNKNOWNZsbattery)r8 ZminsleftZ
power_pluggedZsecsleftrF rF rG r` � s
r` �inetc
C s� g } x� t � D]w } y t | � j | � } Wn t k
rF w Yq X| r x7 | D]/ } t | � | g } | j t j | � � qT Wq W| S)z System-wide network connections.)�pids�Process� connectionsr �listrM r Zsconn)�kindrN �pidZcons�crF rF rG �net_connections� s
rj 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 rk Z snicstats)�namesrN r' ZmtuZisup�duplexZspeed�errrF rF rG � net_if_stats s
ru c C s
t j � S)z:The system boot time expressed in seconds since the epoch.)rE � boot_timerF rF rF rG rv s rv c C s� g } t j � } xo | D]g } | \ } } } } } | d k rC q | sL q t j | | p^ d | pg d | | � } | j | � q W| S)z:Return currently connected users as a list of namedtuples.�~N)rE �usersr ZsuserrM ) r[ �rawlistrO r2 Ztty�hostnameZtstamprh �ntrF rF rG rx ! s
'rx c C sv t j � } d | k rr y$ t d � j � | j d d � Wn3 t k
rP Yn" t k
rq | j d d � Yn X| S)Nr )rE rc rd � create_time�insertr r )ZlsrF rF rG rc 5 s
rc 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 t j k rP t | j | j � � | j t j t j f k r� t | j | j � � � WYd d } ~ Xn0 t j
k
r� t | j | j | j � � Yn Xd S)N)
ro rp �ESRCHr rh �_name�EPERMZEACCESr rE ZZombieProcessErrorr �_ppid)�self�args�kwargsrt )�funrF rG �wrapperL s z wrap_exceptions.<locals>.wrapper)� functools�wraps)r� r� rF )r� rG �wrap_exceptionsH s ! r� c c s� y d VWn� t t f k
r� } z� t | t � sE | j t j k r� y | j � } Wn t k
ro | � Yq� X| t j k r� t | j
| j | j � � q� t
| j
| j � � n � WYd d } ~ Xn Xd S)z�There are some poor C APIs which incorrectly raise ESRCH when
the process is still alive or it's a zombie, or even RuntimeError
(those who don't set errno). This is here in order to solve:
https://github.com/giampaolo/psutil/issues/1044
N)ro � RuntimeError�
isinstancerp r~ r% r r �
STATUS_ZOMBIEr rh r r� r )�procrt r% rF rF rG � catch_zombie[ s !
r� c @ s# e Z d Z d Z d d d g Z d d � Z e 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 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 d1 � � Z e d2 d3 � � Z e d4 d5 d6 � � Z e d7 d8 � � Z e d9 d: � � Z e d; d< � � Z! e d= d> � � Z" e d? d@ � � Z# d4 S)Ard z1Wrapper class around underlying C implementation.rh r r� c C s | | _ d | _ d | _ d S)N)rh r r� )r� rh rF rF rG �__init__z s zProcess.__init__c C s t j | j � } | S)N)rE Zproc_kinfo_oneshotrh )r� rN rF rF rG �_get_kinfo_proc s zProcess._get_kinfo_procc C s* t | � � t j | j � } Wd QRX| S)N)r� rE Zproc_pidtaskinfo_oneshotrh )r� rN rF rF rG �_get_pidtaskinfo� s
zProcess._get_pidtaskinfoc C s | j j � | j j � d S)N)r� Zcache_activater� )r� rF rF rG �
oneshot_enter� s
zProcess.oneshot_enterc C s | j j � | j j � d S)N)r� Zcache_deactivater� )r� rF rF rG � oneshot_exit� s
zProcess.oneshot_exitc C s4 | j � t d } | d k r$ | St j | j � S)Nr' )r� �kinfo_proc_maprE Z proc_namerh )r� r' rF rF rG r' � s z Process.namec C s( t | � � t j | j � SWd QRXd S)N)r� rE Zproc_exerh )r� rF rF rG �exe� s
z Process.exec C s( t | � � t j | j � SWd QRXd S)N)r� rE Z proc_cmdlinerh )r� rF rF rG �cmdline� s
zProcess.cmdlinec
C s. t | � � t t j | j � � SWd QRXd S)N)r� r
rE Z proc_environrh )r� rF rF rG �environ� s
zProcess.environc C s | j � t d | _ | j S)Nr )r� r� r� )r� rF rF rG r � s z Process.ppidc C s( t | � � t j | j � SWd QRXd S)N)r� rE Zproc_cwdrh )r� rF rF rG �cwd� s
z Process.cwdc C s7 | j � } t j | t d | t d | t d � S)Nr r r )r� r �puidsr� )r� �rawtuplerF rF rG �uids� s
z Process.uidsc C s7 | j � } t j | t d | t d | t d � S)Nr r r )r� r r� r� )r� r� rF rF rG �gids� s
z Process.gidsc C sI | j � t d } t j � } y | | SWn t k
rD d SYn Xd S)Nr! )r� r� r Zget_terminal_map�KeyError)r� Ztty_nrZtmaprF rF rG �terminal� s
zProcess.terminalc C s? | j � } t | t d | t d | t d | t d � S)Nr+ r, r- r. )r� r> �pidtaskinfo_map)r� r� rF rF rG � memory_info� s zProcess.memory_infoc C s/ | j � } t j | j � } t | | f � S)N)r� rE Zproc_memory_ussrh r? )r� Z basic_memr@ rF rF rG �memory_full_info� s zProcess.memory_full_infoc C s2 | j � } t j | t d | t d d d � S)Nr) r* g )r� r Z pcputimesr� )r� r� rF rF rG rK � s
zProcess.cpu_timesc C s | j � t d S)Nr# )r� r� )r� rF rF rG r| � s zProcess.create_timec C s$ | j � t d } t j | d � S)Nr0 r )r� r� r Zpctxsw)r� ZvolrF rF rG �num_ctx_switches� s zProcess.num_ctx_switchesc C s | j � t d S)Nr/ )r� r� )r� rF rF rG � num_threads� s zProcess.num_threadsc C s� | j d k r g Sg } t | � � t j | j � } Wd QRXx? | D]7 \ } } t | � rF t j | | � } | j | � qF W| S)Nr )rh r� rE Zproc_open_filesr r Z popenfilerM )r� �filesry rW �fdr] rF rF rG �
open_files� s
zProcess.open_filesrb c C s0 | t k r8 t d | d j d d � t D� � f � � t | \ } } t | � � t j | j | | � } Wd QRXg } x� | D]� } | \ } } } }
} } t | } t | � } t | � } | t
t f k r� |
r� t j
|
� }
| r� t j
| � } t j | | | |
| | � }
| j |
� q� W| S)Nz+invalid %r kind argument; choose between %sz, c S s g | ] } t | � � q SrF )�repr)�.0�xrF rF rG �
<listcomp> s z'Process.connections.<locals>.<listcomp>)r
�
ValueError�joinr� rE Zproc_connectionsrh � TCP_STATUSESr r r r r �addrZpconnrM )r� rg �families�typesry rN rO r� Zfam�type�laddr�raddrr% r{ rF rF rG re s( &
zProcess.connectionsc C s; | j d k r d St | � � t j | j � SWd QRXd S)Nr )rh r� rE Z proc_num_fds)r� rF rF rG �num_fds s
zProcess.num_fdsNc C s t j | j | | j � S)N)r Zwait_pidrh r )r� �timeoutrF rF rG �wait! s z Process.waitc C s( t | � � t j | j � SWd QRXd S)N)r� rn � getpriorityrh )r� rF rF rG �nice_get% s
zProcess.nice_getc
C s+ t | � � t j | j | � SWd QRXd S)N)r� rn � setpriorityrh )r� �valuerF rF rG �nice_set* s
zProcess.nice_setc C s$ | j � t d } t j | d � S)Nr% �?)r� r� �
PROC_STATUSES�get)r� �coderF rF rG r% / s zProcess.statusc C sX t j | j � } g } x9 | D]1 \ } } } t j | | | � } | j | � q W| S)N)rE Z proc_threadsrh r ZpthreadrM )r� ry r[ Z thread_id�utimeZstimer] rF rF rG �threads5 s zProcess.threadsc C s t j | j � S)N)rE Zproc_memory_mapsrh )r� rF rF rG � memory_maps> s zProcess.memory_maps)$�__name__�
__module__� __qualname__�__doc__� __slots__r� r r� r� r� r� r� r' r� r� r� r r� r� r� r� r� r� rK r| r� r� r� re r� r� r� r� r% r� r� rF rF rF rG rd u sB
rd )r@ )hr� �
contextlibrp r� rV �socketr � collectionsr rT r r r rE r rn r r
r r r
r r r � _exceptionsr r r Z__extra__all__�sysconfZPAGESIZEZAF_LINKZTCPS_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_CLOSINGZPSUTIL_CONN_NONEZ CONN_NONEr� ZSIDLZ STATUS_IDLEZSRUNZSTATUS_RUNNINGZSSLEEPZSTATUS_SLEEPINGZSSTOPZSTATUS_STOPPEDZSZOMBr� r� �dictr� r� r1 r6 r> �_fieldsr? rA r� rB rH rJ rK rL rP rQ rR rS �
disk_usageZdisk_io_countersrU r` rl Z net_if_addrsrj ru rv rx rc Z
pid_existsr� �contextmanagerr� �objectrd rF rF rF rG �<module> s�