o
    a8y                     @   s  d Z ddlZddlZddlZddlZddlm  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  ddlm!Z! ddlm"Z" g Z#erej$ej%ej&ej'ej(ej)ej*ej+ej,ej-ej.ej/ej0ej1iZ2n=erej$ej%ej(ej)ej*ej+ej3ej-ej,ej-ej&ej4ej5ej'iZ2nerej$ej%ej(ej)ej*ej+ej,ej-ej&ej4ej5ej'iZ2ej6ej7ej8ej9ej:ej;ej<ej=ej>ej?ej@ejAejBejCejDejEejFejGejHejIejJejKejLejMiZNeO ZPejQZQeRedZSeRedZTeRedZUeRedZVeWdi dddddd d!d"d#d$d%d&d'd(d)d*d+d,d-d.d/d0d1d2d3d4d5d6d7d8d9d:d;d<d=d>d?d@dAdBdCdDdEdFdGdHdIdJdKdLZXedMg dNZYedOg dPZZedQg dRZ[e[Z\edSg dTZ]edUdVZ^edWdXZ_eredYg dZZ`nedYg d[Z`d\d] Zad^d_ Zbd`da ZceSrdbd Zdndcd Zddded_ededf Zfeserdgdh Zgndidh Zgdjdk ZhddldmZiejjZjejkZkejlZlejmZmdndo Zndpdq Zoer*drds Zpdtdu Zqdvdw Zrdxdy Zsdzd{ Zted|d} Zud~d ZvesBerGdd ZwnejwZwdd Zxdd Zyejzdd Z{G dd de|Z}dS )z5FreeBSD, OpenBSD and NetBSD platforms implementation.    N)defaultdict)
namedtuple   )_common)_psposix)_psutil_bsd)_psutil_posix)FREEBSD)NETBSD)OPENBSD)AccessDenied)NoSuchProcess)ZombieProcess)	conn_tmap)conn_to_ntuple)memoize)memoize_when_activated)usage_percent)FileNotFoundError)PermissionError)ProcessLookupError)whichper_cpu_timesproc_num_threadsproc_open_filesproc_num_fds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)r>   r@   textdatastack	pcputimes)rY   r[   children_userchildren_systempmmap_grouped*path rss, private, ref_count, shadow_count	pmmap_ext6addr, perms path rss, private, ref_count, shadow_countsdiskio)
read_countwrite_count
read_byteswrite_bytesZ	read_timeZ
write_timeZ	busy_time)rj   rk   rl   rm   c                  C   s   t  } | \}}}}}}}}trItdd,}	|	D ]!}
|
dr*t|
 d d }q|
dr9t|
 d d }qW d   n1 sDw   Y  || | }|| | }t|| |dd}t|||||||||||S )	z&System virtual memory as a namedtuple.z/proc/meminforbs   Buffers:r   i   s
   MemShared:NZround_)	cextZvirtual_memr
   open
startswithintsplitr   rL   )ZmemrM   rP   rQ   rR   rV   rT   rS   rU   flineZavailrO   rN    rw   //usr/lib/python3/dist-packages/psutil/_psbsd.pyvirtual_memory   s$   

ry   c                  C   s4   t  \} }}}}t|| dd}t| |||||S )z@System swap memory as (total, used, free, sin, sout) namedtuple.r   ro   )rp   Zswap_memr   r   Zsswap)rM   rO   rP   ZsinZsoutrN   rw   rw   rx   swap_memory   s   rz   c                  C   s"   t  \} }}}}t| ||||S )z+Return system per-CPU times as a namedtuple)rp   	cpu_timesrW   rX   rw   rw   rx   r{      s   r{   c                  C   s>   g } t  D ]}|\}}}}}t|||||}| | q| S )'Return system CPU times as a namedtuple)rp   r   rW   append)retZcpu_trY   rZ   r[   r\   r]   itemrw   rw   rx   r      s   c                   C   s.   t  dkr	t gS tjrtddt_t gS )r|   r   &supported only starting from FreeBSD 8T)cpu_count_logicalr{   r   
__called__NotImplementedErrorrw   rw   rw   rx   r      s   
Fc                   C      t  S )z0Return the number of logical CPUs in the system.)rp   r   rw   rw   rw   rx   r         r   c                   C   s   t  dkrdS d S )Nr   )r   rw   rw   rw   rx   cpu_count_cores      r   c                  C   s   d} t  }|dur5|d}|dkr5|d|d  }t|}zt|dp)d} W |  n|  w | s>t dkr>dS | S )z-Return the number of CPU cores in the system.Nz	</groups>r-   zgroup/children/group/cpur   )	rp   Zcpu_topologyrfindETZ
fromstringlenfindallclearr   )r~   sindexrootrw   rw   rx   r      s   


c            	      C   s   t rt \} }}}}nBtrAt \} }}}}}}tdd}|D ]}|dr0t| d }q!W d   n1 s;w   Y  ntrNt \} }}}}}}t	
| |||S )z*Return various CPU stats as a named tuple.z
/proc/statrn   s   intrr   N)r	   rp   	cpu_statsr
   rq   rr   rs   rt   r   r   Z	scpustats)	ZctxswZintrsZ
soft_intrsZsyscallsZtrapsZfaultsZforksru   rv   rw   rw   rx   r     s"   
r   c              	   C   sL   g }t  }|D ]}|\}}}}d }}	t||||||	}
||
 q|S )zReturn mounted disk partitions as a list of namedtuples.
    'all' argument is ignored, see:
    https://github.com/giampaolo/psutil/issues/906
    N)rp   disk_partitionsr   Z	sdiskpartr}   )allretlistZ
partitions	partitionZdeviceZ
mountpointZfstypeZoptsZmaxfileZmaxpathntuplerw   rw   rx   r   >  s   r   c                  C   s   t   } i }| D ]C}zt|}t|}t|\}}W n ty7 } z|jtjkr- W Y d}~q	d}~ww t	t
drBt
|}t
||||||< q	|S )z)Get NIC stats (isup, duplex, speed, mtu).N	NicDuplex)net_io_counterskeys
cext_posixZ
net_if_mtuZnet_if_is_runningZnet_if_duplex_speedOSErrorerrnoZENODEVhasattrr   r   Z	snicstats)namesr~   rJ   ZmtuZisupZduplexZspeederrrw   rw   rx   net_if_stats[  s"   




r   c              
   C   s  t r7g }t D ]-}z	t|| }W n ttfy   Y qw |D ]}t|}|| |tj	|  q q|S | tj
vrLtd| ddd t
D f t
|  \}}t }tr]td}nt }|D ]#}|\}	}
}}}}}|
|v r||v rt|	|
||||t|}|| qct|S )z System-wide network connections.+invalid %r kind argument; choose between %s, c                 S      g | ]}t |qS rw   repr.0xrw   rw   rx   
<listcomp>      z#net_connections.<locals>.<listcomp>r   )r   pidsProcessconnectionsr   r   listr}   r   Zsconnr   
ValueErrorjoinsetr
   rp   net_connectionsr   TCP_STATUSESadd)kindr~   pidZconsZconnfamiliestypesrawlistr   fdfamtypeladdrraddrr   ntrw   rw   rx   r   o  s@   



r   c                  C   sb   z	t  \} }}W n
 ty   Y dS w |dk}|rtj}n|dkr&tj}n|d }t| ||S )zReturn battery info.Nr   r   <   )rp   sensors_batteryr   r   ZPOWER_TIME_UNLIMITEDZPOWER_TIME_UNKNOWNZsbattery)rN   ZminsleftZpower_pluggedZsecsleftrw   rw   rx   r     s   r   c               	   C   sr   t t} t }t|D ]+}z t|\}}|dkrd}d| }| d t|||| W q t	y6   Y qw | S )z?Return CPU cores temperatures if available, else an empty dict.r   NzCore %sZcoretemp)
r   r   r   rangerp   Zsensors_cpu_temperaturer}   r   Zshwtempr   )r~   num_cpuscpucurrentZhighrJ   rw   rw   rx   sensors_temperatures  s   r   c               
   C   s   g } t  }t|D ]]}z	t|\}}W n	 ty   Y q	w |r\zt|dd dd }W n ttfy=   d}Y nw zt|dd dd }W n ttfy[   d}Y nw | 	t
||| q	| S )zReturn frequency metrics for CPUs. As of Dec 2018 only
        CPU 0 appears to be supported by FreeBSD and all other cores
        match the frequency of CPU 0.
         r   /r   N)r   r   rp   Zcpu_frequencyr   rs   rt   
IndexErrorr   r}   r   Zscpufreq)r~   r   r   r   Zavailable_freqZmin_freqZmax_freqrw   rw   rx   cpu_freq  s*     r   c                   C   r   )z:The system boot time expressed in seconds since the epoch.)rp   	boot_timerw   rw   rw   rx   r     r   r   c            	      C   sf   g } t  }|D ](}|\}}}}}|dkrtsJ d}|dkr qt||p&d|||}| | q| S )z:Return currently connected users as a list of namedtuples.r   N~)rp   usersr   r   Zsuserr}   )	r   r   r   rY   ZttyZhostnameZtstampr   r   rw   rw   rx   r     s   r   c                   C   s:   z	t d  W dS  ty   Y dS  ty   Y dS w )Nr   FT)r   rJ   r   r   rw   rw   rw   rx   _pid_0_exists  s   r   c                  C   s*   t  } trd| vrt r| dd | S )z7Returns a list of PIDs currently running on the system.r   )rp   r   r   r   insert)r~   rw   rw   rx   r     s   r   c                 C   s   t | }|s| t v S dS )zReturn True if pid exists.T)r   
pid_existsr   )r   existsrw   rw   rx   r   	  s   

r   c                 C   s4   zt | td  }|t jkW S  ty   Y dS w )Nr   F)rp   proc_oneshot_infokinfo_proc_mapSZOMB	Exception)r   strw   rw   rx   	is_zombie  s   r   c                    s   t   fdd}|S )z`Decorator which translates bare OSError exceptions into
    NoSuchProcess and AccessDenied.
    c                    s   z | g|R i |W S  t y(   t| jr!t| j| j| jt| j| j ty5   t| j| j t	yN   | jdkrMdt
 v rLt| j| j  w Nr   )r   r   r   r   _name_ppidr   r   r   r   r   )selfargskwargsfunrw   rx   wrapper"  s   


z wrap_exceptions.<locals>.wrapper)	functoolswraps)r   r   rw   r   rx   wrap_exceptions  s   r   c              	   c   sf    zdV  W dS  t tfy%   t| jrt| j| j| jt| j| j ty2   t	| j| jw )z8Same as above, for routines relying on reading /proc fs.N)
r   r   r   r   r   r   r   r   r   r   )instrw   rw   rx   wrap_exceptions_procfs7  s   
r   c                   @   s  e Zd ZdZg dZdd Zdd Ze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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dLd,d-ZedMd/d0Zed1d2 Zed3d4 Z ed5d6 Z!ed7d8 Z"ed9d: Z#e$d;d<Z%e$d;d=Z&d>d? Z'e(red@dA Z)ne'Z)e*redBdC Z+ne'Z+eredDdE Z,edFdG Z-edHdI Z.edMdJdKZ/d.S d.S )Nr   z1Wrapper class around underlying C implementation.)r   r   r   _cachec                 C   s   || _ d | _d | _d S N)r   r   r   )r   r   rw   rw   rx   __init__M  s   
zProcess.__init__c                 C   s   t | j dS )z+Raise NSP if the process disappeared on us.N)rp   	proc_namer   r   rw   rw   rx   _assert_aliveR  s   zProcess._assert_alivec                 C   s$   t | j}t|ttksJ |S )z;Retrieves multiple process info in one shot as a raw tuple.)rp   r   r   r   r   r   r~   rw   rw   rx   oneshotX  s   zProcess.oneshotc                 C      | j |  d S r   )r   Zcache_activater   rw   rw   rx   oneshot_enter`     zProcess.oneshot_enterc                 C   r   r   )r   Zcache_deactivater   rw   rw   rx   oneshot_exitc  r   zProcess.oneshot_exitc                 C   s(   |   td  }|d ur|S t| jS )NrJ   )r   r   rp   r   r   )r   rJ   rw   rw   rx   rJ   f  s   zProcess.namec                 C   s   t r| jdkr	dS t| jS tr6| jdkrdS t|  td| j W  d    S 1 s/w   Y  d S |  }|rDt	|d pCdS dS )Nr    z/proc/%s/exe)
r	   r   rp   Zproc_exer
   r   osreadlinkcmdliner   )r   r   rw   rw   rx   exek  s   


$zProcess.exec              
   C   s   t r	| jdkr	g S trJzt| jW S  tyI } z+|jtjkrDt| jr.t	| j| j
| jt| js<t| j| j
| jg W  Y d }~S  d }~ww t| jS r   )r   r   r
   rp   Zproc_cmdliner   r   EINVALr   r   r   r   r   r   )r   r   rw   rw   rx   r     s    

zProcess.cmdlinec                 C      t | jS r   )rp   Zproc_environr   r   rw   rw   rx   environ     zProcess.environc                 C   s8   |   td  }t }z|| W S  ty   Y d S w )Nr*   )r   r   r   Zget_terminal_mapKeyError)r   Ztty_nrZtmaprw   rw   rx   terminal  s   
zProcess.terminalc                 C   s   |   td  | _| jS )Nr   )r   r   r   r   rw   rw   rx   r     s   zProcess.ppidc                 C   .   |   }t|td  |td  |td  S )Nr   r    r"   )r   r   Zpuidsr   r   Zrawtuplerw   rw   rx   uids     


zProcess.uidsc                 C   r   )Nr$   r&   r(   )r   r   Zpgidsr   r   rw   rw   rx   gids  r   zProcess.gidsc                 C   s8   |   }t|td  |td  |td  |td  S )Nr6   r8   r:   r<   )r   r   rb   r   r   rw   rw   rx   r{     s   



zProcess.cpu_timesc                 C      |   td  S )NrH   r   r   r   rw   rw   rx   cpu_num     zProcess.cpu_numc                 C   s@   |   }t|td  |td  |td  |td  |td  S )Nr>   r@   rB   rD   rF   )r   r^   r   r   rw   rw   rx   memory_info  s   




zProcess.memory_infoc                 C   r  )Nr,   r  r   rw   rw   rx   r,     r  zProcess.create_timec                 C   s   t rt| jS t|  S r   )HAS_PROC_NUM_THREADSrp   r   r   r   threadsr   rw   rw   rx   num_threads  s   zProcess.num_threadsc                 C   s$   |   }t|td  |td  S )Nr.   r0   )r   r   Zpctxswr   r   rw   rw   rx   num_ctx_switches  s
   

zProcess.num_ctx_switchesc                 C   sH   t | j}g }|D ]\}}}t|||}|| q
tr"|   |S r   )rp   Zproc_threadsr   r   Zpthreadr}   r   r   )r   r   r   Z	thread_idutimeZstimer   rw   rw   rx   r    s   zProcess.threadsinetc              	   C   s  |t vrtd|ddd t D f trXt | \}}g }t| j}|D ])}|\}}}	}
}}}|| jks8J ||v rO|	|v rOt|||	|
||t}|	| q&| 
  t|S t | \}}t| j||}g }|D ]}|\}}}	}
}}t|||	|
||t}|	| qjtr| 
  |S )Nr   r   c                 S   r   rw   r   r   rw   rw   rx   r     r   z'Process.connections.<locals>.<listcomp>)r   r   r   r
   rp   r   r   r   r   r}   r   r   Zproc_connectionsr   )r   r   r   r   r~   r   r   r   r   r   r   r   r   r   r   rw   rw   rx   r     s>   
zProcess.connectionsNc                 C   s   t | j|| jS r   )r   Zwait_pidr   r   )r   Ztimeoutrw   rw   rx   wait  r   zProcess.waitc                 C   r   r   )r   getpriorityr   r   rw   rw   rx   nice_get  r   zProcess.nice_getc                 C   s   t | j|S r   )r   setpriorityr   )r   valuerw   rw   rx   nice_set"  s   zProcess.nice_setc                 C   s   |   td  }t|dS )Nr   ?)r   r   PROC_STATUSESget)r   coderw   rw   rx   r   &  s   zProcess.statusc                 C   s(   |   }t|td  |td  ddS )Nr2   r4   r   )r   r   Zpior   r   rw   rw   rx   io_counters,  s   

zProcess.io_countersc                 C   s<   t r	| jdkr	dS tstrt| jpdS ttrdd)z)Return process current working directory.r   Nr   r   )r   r   r
   HAS_PROC_OPEN_FILESrp   Zproc_cwdr   r	   r   rw   rw   rx   cwd5  s   zProcess.cwdZmmaprf   rh   c                 C   s   t r   )r   r   rw   rw   rx   _not_implementedJ  s   zProcess._not_implementedc                 C   s   t | j}dd |D S )z8Return files opened by process as a list of namedtuples.c                 S   s   g | ]
\}}t ||qS rw   )r   Z	popenfile)r   pathr   rw   rw   rx   r   T  s    z&Process.open_files.<locals>.<listcomp>)rp   r   r   )r   r   rw   rw   rx   
open_filesP  s   zProcess.open_filesc                 C   s   t | j}tr|   |S )z=Return the number of file descriptors opened by this process.)rp   r   r   r
   r   r   rw   rw   rx   num_fds[  s   zProcess.num_fdsc                 C   r   r   )rp   Zproc_cpu_affinity_getr   r   rw   rw   rx   cpu_affinity_geti  r   zProcess.cpu_affinity_getc              
   C   s   t ttt }|D ]}||vrtd||f qz
t| j| W d S  tyK } z|j	t	j
t	jfv rF|D ]}||vrEtd||f q7 d }~ww )Nz#invalid CPU #%i (choose between %s))tupler   r   r   r   rp   Zproc_cpu_affinity_setr   r   r   r   ZEDEADLK)r   ZcpusZallcpusr   r   rw   rw   rx   cpu_affinity_setm  s,   zProcess.cpu_affinity_setc                 C   r   r   )rp   Zproc_memory_mapsr   r   rw   rw   rx   memory_maps  r   zProcess.memory_mapsc                 C   sL   |d u rt | j|S t|dkrtdt| |\}}t | j|||S )Nr   z4second argument must be a (soft, hard) tuple, got %s)rp   Zproc_getrlimitr   r   r   r   Zproc_setrlimit)r   ZresourceZlimitsZsoftZhardrw   rw   rx   rlimit  s   zProcess.rlimit)r  r   )0__name__
__module____qualname____doc__	__slots__r   r   r   r   r   r   r   rJ   r   r   r   r   r   r   r  r{   r	   r  r  Zmemory_full_infor,   r	  r
  r  r   r  r  r  r   r  r  r   Znt_mmap_groupedZnt_mmap_extr  r  r  HAS_PROC_NUM_FDSr  r  r   r!  r"  rw   rw   rw   rx   r   H  s    










	



!







r   rw   )F)~r&  
contextlibr   r   r   Zxml.etree.ElementTreeZetreeZElementTreer   collectionsr   r   r   r   r   r   rp   r   r   r	   r
   r   r   r   r   r   r   r   r   r   Z_compatr   r   r   r   Z__extra__all__ZSIDLZSTATUS_IDLEZSRUNZSTATUS_RUNNINGZSSLEEPZSTATUS_SLEEPINGZSSTOPZSTATUS_STOPPEDr   ZSTATUS_ZOMBIEZSWAITZSTATUS_WAITINGZSLOCKZSTATUS_LOCKEDr  ZSDEADZSTATUS_WAKINGZSONPROCZTCPS_ESTABLISHEDZCONN_ESTABLISHEDZTCPS_SYN_SENTZCONN_SYN_SENTZTCPS_SYN_RECEIVEDZCONN_SYN_RECVZTCPS_FIN_WAIT_1ZCONN_FIN_WAIT1ZTCPS_FIN_WAIT_2ZCONN_FIN_WAIT2ZTCPS_TIME_WAITZCONN_TIME_WAITZTCPS_CLOSEDZ
CONN_CLOSEZTCPS_CLOSE_WAITZCONN_CLOSE_WAITZTCPS_LAST_ACKZCONN_LAST_ACKZTCPS_LISTENZCONN_LISTENZTCPS_CLOSINGZCONN_CLOSINGZPSUTIL_CONN_NONEZ	CONN_NONEr   ZgetpagesizeZPAGESIZEZAF_LINKr   ZHAS_PER_CPU_TIMESr  r  r(  dictr   rL   rW   r^   Zpfullmemrb   re   rg   ri   ry   rz   r{   r   r   r   r   r   r   
disk_usageZdisk_io_countersr   Znet_if_addrsr   r   r   r   r   r   r   r   r   r   r   r   contextmanagerr   objectr   rw   rw   rw   rx   <module>   sh  	




	
#	
	

#(




