B
    "Occ                 @   sL  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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# Z$ej%Z%ej&dkZ'dZ(dZ)ej*ej+ej,ej-ej.ej/ej0ej1ej2ej3ej4ej-ej5ej6iZ7e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jMejNejOejPe(ejQe)iZReSddd d!d"d#d$d%d&d'd(d)d*ZTed+d,d-d.d/gZUed0d,d-d1d2gZVed3d4d5d6d7d8gZWed9d:d;gZXeXZYed<d=d:d>d?gZZed@dAdB[eZj\ Z]dCdD Z^dEdF Z_dGdH Z`dIdJ ZadKdL ZbdMdN ZcdOdP ZdejeZeejfZfdedRdSZgejhZhejiZidfdUdVZjdWdX ZkdYdZ Zld[d\ Zmd]d^ Znd_d` Zodadb ZpG dcdd ddeqZrdS )gz'Sun OS Solaris platform implementation.    N)
namedtuple)AF_INET   )_common)_psposix)_psutil_posix)_psutil_sunos)AF_INET6)AccessDenied)NoSuchProcess)ZombieProcess)debug)get_procfs_path)isfile_strict)memoize_when_activated)sockfam_to_enum)socktype_to_enum)usage_percent)PY3)FileNotFoundError)PermissionError)ProcessLookupError)b	CONN_IDLE
CONN_BOUNDZPROCFS_PATHl        ZIDLEZBOUND                     	   
      )ppidrssvmscreate_timenicenum_threadsstatusttynruideuidgidegid	scputimesusersystemZidleZiowait	pcputimeschildren_userchildren_systemsvmemtotalZ	availablepercentusedfreepmemr&   r'   pmmap_groupedpathZ	anonymouslocked	pmmap_extzaddr perms  c              C   sF   t dt } t dt  }}| | }t|| dd}t| ||||S )zReport virtual memory metrics.SC_PHYS_PAGESSC_AVPHYS_PAGESr   )round_)ossysconf	PAGE_SIZEr   r7   )r8   r;   Zavailr:   r9    rH   .lib/python3.7/site-packages/psutil/_pssunos.pyvirtual_memory{   s
    rJ   c           	   C   s  t  \} }tjddtjd  ddgtjd}| \}}trL|	t
jj}|jdkrdtd|j | d	d
d }|stdd }}xL|D ]D}| }|dd \}	}
|tt|	d 7 }|tt|
d 7 }qW || }t||d
d}t||||| t |t S )zReport swap memory metrics.z/usr/bin/envzPATH=/usr/sbin:/sbin:%sPATHZswapz-l)stdoutr   z'swap -l' failed (retcode=%s)
r   Nzno swap device(s) configuredr   r   i   )rD   )cextZswap_mem
subprocessPopenrE   environPIPEcommunicater   decodesysrL   encoding
returncodeRuntimeErrorstripsplitintr   r   ZsswaprG   )ZsinZsoutprL   stderrlinesr8   r;   linetfr:   r9   rH   rH   rI   swap_memory   s,    


rb   c              C   s   t  } tdd t|  D  S )z-Return system-wide CPU times as a named tuplec             S   s   g | ]}t |qS rH   )sum).0xrH   rH   rI   
<listcomp>   s    zcpu_times.<locals>.<listcomp>)rN   per_cpu_timesr1   zip)retrH   rH   rI   	cpu_times   s    rj   c              C   s   t  } dd | D S )z5Return system per-CPU times as a list of named tuplesc             S   s   g | ]}t | qS rH   )r1   )rd   re   rH   rH   rI   rf      s    z!per_cpu_times.<locals>.<listcomp>)rN   rg   )ri   rH   rH   rI   rg      s    rg   c               C   s$   y
t dS  tk
r   dS X dS )z0Return the number of logical CPUs in the system.SC_NPROCESSORS_ONLNN)rE   rF   
ValueErrorrH   rH   rH   rI   cpu_count_logical   s    
rm   c               C   s   t  S )z-Return the number of CPU cores in the system.)rN   cpu_count_coresrH   rH   rH   rI   rn      s    rn   c              C   s$   t  \} }}}d}t| |||S )z*Return various CPU stats as a named tuple.r   )rN   	cpu_statsr   Z	scpustats)Zctx_switchesZ
interruptsZsyscallsZtrapsZsoft_interruptsrH   rH   rI   ro      s    
ro   Fc             C   s   g }t  }x|D ]}|\}}}}|dkr.d}| sxyt|js@wW n4 tk
rv } ztd||f  wW dd}~X Y nX d }	}
t|||||	|
}|| qW |S )zReturn system disk partitions.Znone zskipping %r: %sN)	rN   disk_partitions
disk_usager8   OSErrorr   r   Z	sdiskpartappend)allretlistZ
partitions	partitionZdeviceZ
mountpointZfstypeZoptserrZmaxfileZmaxpathZntuplerH   rH   rI   rq      s$    

rq   c          
   C   s  t j }|dkr|dd | |krFtd| ddd |D f t j|  \}}t|}t }x|D ]}|\}}	}
}}}}|	|krqj|
|krqj|	t	t
fkr|rt j| }|rt j| }t| }t|	}	t|
}
|dkrt ||	|
||||}nt ||	|
|||}|| qjW t|S )zReturn 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).
    ry   unixr   z+invalid %r kind argument; choose between %sz, c             S   s   g | ]}t |qS rH   )repr)rd   re   rH   rH   rI   rf     s    z#net_connections.<locals>.<listcomp>)r   Z	conn_tmapcopypoprl   joinrN   net_connectionssetr   r	   addrTCP_STATUSESr   r   Zsconnpconnaddlist)kind_pidZcmapZfamiliestypesrawlistri   itemfdZfamZtype_ZladdrZraddrr+   pidntrH   rH   rI   r      s8    




r   c              C   sX   t  } xJ|  D ]>\}}|\}}}}ttdr:t|}t||||d| |< qW | S )z)Get NIC stats (isup, duplex, speed, mtu).	NicDuplexrp   )rN   net_if_statsitemshasattrr   r   Z	snicstats)ri   namer   ZisupZduplexZspeedZmturH   rH   rI   r   !  s    

r   c               C   s   t  S )z:The system boot time expressed in seconds since the epoch.)rN   	boot_timerH   rH   rH   rI   r   1  s    r   c              C   s`   g } t  }d}xJ|D ]B}|\}}}}}}	|s0q||kr<d}t|||||	}
| |
 qW | S )z:Return currently connected users as a list of namedtuples.)z:0.0z:0	localhost)rN   usersr   Zsuserrt   )rv   r   r   r   r2   ttyZhostnameZtstampZuser_processr   r   rH   rH   rI   r   6  s    
r   c               C   s   dd t tt D S )z7Returns a list of PIDs currently running on the system.c             S   s   g | ]}|  rt|qS rH   )isdigitr[   )rd   re   rH   rH   rI   rf   P  s    zpids.<locals>.<listcomp>)rE   listdirr   r   rH   rH   rH   rI   pidsN  s    r   c             C   s
   t | S )z&Check for the existence of a unix pid.)r   
pid_exists)r   rH   rH   rI   r   S  s    r   c                s   t   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 | f||S  t tfk
rT   t| js>t| j| jnt| j| j| jY n\ tk
rt   t	| j| jY n< t
k
r   | jdkrdt krt	| j| jn  Y nX d S )Nr   )r   r   r   r   r   _namer   _ppidr   r
   rs   r   )selfargskwargs)funrH   rI   wrapper\  s    


z wrap_exceptions.<locals>.wrapper)	functoolswraps)r   r   rH   )r   rI   wrap_exceptionsX  s    r   c               @   s  e Zd ZdZdddddgZdd Zd	d
 Zdd Zdd Ze	e
dd Ze	e
dd Ze	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 Ze	d1d2 Ze	d3d4 ZeZe	d5d6 Ze	d7d8 Z e	d9d: Z!d;d< Z"e	dLd>d?Z#e$d@dAZ%e$d@dBZ&e	dCdD Z'e	dEdF Z(e	dGdH Z)e	dMdJdKZ*dIS )NProcessz1Wrapper class around underlying C implementation.r   r   r   _procfs_path_cachec             C   s   || _ d | _d | _t | _d S )N)r   r   r   r   r   )r   r   rH   rH   rI   __init__y  s    zProcess.__init__c             C   s   t d| j| jf  dS )z+Raise NSP if the process disappeared on us.z%s/%sN)rE   statr   r   )r   rH   rH   rI   _assert_alive  s    zProcess._assert_alivec             C   s(   | j |  | j|  | j|  d S )N)_proc_name_and_argsZcache_activate_proc_basic_info
_proc_cred)r   rH   rH   rI   oneshot_enter  s    zProcess.oneshot_enterc             C   s(   | j |  | j|  | j|  d S )N)r   Zcache_deactivater   r   )r   rH   rH   rI   oneshot_exit  s    zProcess.oneshot_exitc             C   s   t | j| jS )N)rN   Zproc_name_and_argsr   r   )r   rH   rH   rI   r     s    zProcess._proc_name_and_argsc             C   sT   | j dkr,tjd| j| j f s,t| j t| j | j}t|tt	ksPt
|S )Nr   z%s/%s/psinfo)r   rE   r>   existsr   r
   rN   Zproc_basic_infolenproc_info_mapAssertionError)r   ri   rH   rH   rI   r     s    

zProcess._proc_basic_infoc             C   s   t | j| jS )N)rN   Z	proc_credr   r   )r   rH   rH   rI   r     s    zProcess._proc_credc             C   s   |   d S )Nr   )r   )r   rH   rH   rI   r     s    zProcess.namec             C   s8   yt d| j| jf S  tk
r*   Y nX |   dS )Nz%s/%s/path/a.outrp   )rE   readlinkr   r   rs   cmdline)r   rH   rH   rI   exe  s    zProcess.exec             C   s   |   d dS )Nr   rA   )r   rZ   )r   rH   rH   rI   r     s    zProcess.cmdlinec             C   s   t | j| jS )N)rN   Zproc_environr   r   )r   rH   rH   rI   rQ     s    zProcess.environc             C   s   |   td  S )Nr(   )r   r   )r   rH   rH   rI   r(     s    zProcess.create_timec             C   s   |   td  S )Nr*   )r   r   )r   rH   rH   rI   r*     s    zProcess.num_threadsc             C   s   |   td  S )Nr)   )r   r   )r   rH   rH   rI   nice_get  s    zProcess.nice_getc             C   s&   | j dkrt| j | jt| j |S )N)r   r   )r   r
   r   
cext_posixsetpriority)r   valuerH   rH   rI   nice_set  s    
zProcess.nice_setc             C   s   |   td  | _| jS )Nr%   )r   r   r   )r   rH   rH   rI   r%     s    zProcess.ppidc             C   s`   y|   \}}}}}}W n8 tk
rP   |  td  }|  td  }d }Y nX t|||S )Nr-   r.   )r   r
   r   r   r   puids)r   real	effectivesaved_rH   rH   rI   uids  s    
zProcess.uidsc             C   s`   y|   \}}}}}}W n8 tk
rP   |  td  }|  td  }d }Y nX t|||S )Nr/   r0   )r   r
   r   r   r   r   )r   r   r   r   r   rH   rH   rI   gids  s    
zProcess.gidsc          
   C   sZ   yt | j| j}W n: tk
rN } z|jtjkr<ts<d}n W d d }~X Y nX tj	| S )N)g        g        g        g        )
rN   Zproc_cpu_timesr   r   rs   errno	EOVERFLOW	IS_64_BITr   r4   )r   timesrx   rH   rH   rI   rj     s    zProcess.cpu_timesc             C   s   t | j| jS )N)rN   Zproc_cpu_numr   r   )r   rH   rH   rI   cpu_num  s    zProcess.cpu_numc          	   C   sx   | j }d}t|  td  }|tjkrhx>dD ]6}ytd|| j|f S  t	k
rb   d}w.Y q.X q.W |rt| 
  d S )NFr,   )r   r   r      z%s/%d/path/%dT)r   r   r   r   rN   ZPRNODEVrE   r   r   r   r   )r   procfs_path
hit_enoentr   re   rH   rH   rI   terminal  s    

zProcess.terminalc             C   sH   | j }ytd|| jf S  tk
rB   td|| jf  d S X d S )Nz%s/%s/path/cwdz%s/%s)r   rE   r   r   r   r   )r   r   rH   rH   rI   cwd  s    zProcess.cwdc             C   s2   |   }|td  d }|td  d }t||S )Nr&   i   r'   )r   r   r<   )r   ri   r&   r'   rH   rH   rI   memory_info#  s    zProcess.memory_infoc             C   s   |   td  }t|dS )Nr+   ?)r   r   PROC_STATUSESget)r   coderH   rH   rI   r+   ,  s    zProcess.statusc       
      C   s   | j }g }td|| jf }d}x|D ]}t|}yt| j||\}}W nH tk
r } z*|jtj	krpt
spw(|jtjkrd}w( W d d }~X Y q(X t|||}	||	 q(W |r|   |S )Nz	%s/%d/lwpFT)r   rE   r   r   r[   rN   Zquery_process_threadEnvironmentErrorr   r   r   ENOENTr   Zpthreadrt   r   )
r   r   ri   Ztidsr   tidutimeZstimerx   r   rH   rH   rI   threads2  s*    
zProcess.threadsc          	   C   s   g }d}| j }d|| jf }x~td|| jf D ]f}tj||}tj|r2yt|}W n tk
rx   d}w2Y q2X t	|r2|
t|t| q2W |r|   |S )NFz
%s/%d/pathz%s/%d/fdT)r   r   rE   r   r>   r~   islinkr   r   r   rt   r   Z	popenfiler[   r   )r   rv   r   r   Zpathdirr   r>   filerH   rH   rI   
open_filesS  s"    zProcess.open_filesc             c   s,  d| }t j|dt jt jd}| \}}trDdd ||fD \}}|jdkrd| krht| j| j	d| krt
| j| j	td	||f |d
dd }xt|D ]v\}}| }|dr|ddd }	||d   }
|
dkrtj}
n|
dkr
tj}
nd}
dtj|
|	dtjfV  qW dS )z<Get UNIX sockets used by process by parsing 'pfiles' output.z	pfiles %sT)shellrL   r]   c             S   s   g | ]}| tjjqS rH   )rT   rU   rL   rV   )rd   re   rH   rH   rI   rf   q  s   z-Process._get_unix_sockets.<locals>.<listcomp>r   zpermission deniedzno such processz%r command error
%srM   r   Nzsockname: AF_UNIXrA   SOCK_STREAM
SOCK_DGRAMry   rp   )rO   rP   rR   rS   r   rW   lowerr
   r   r   r   rX   rZ   	enumeratelstrip
startswithrY   socketr   r   ZAF_UNIXr   	CONN_NONE)r   r   cmdr\   rL   r]   r^   ir_   r>   typerH   rH   rI   _get_unix_socketsh  s2    



zProcess._get_unix_socketsinetc             C   sP   t || jd}|s(td| j| jf  |dkrL|dd | | jD  |S )N)r   z%s/%s)ru   rz   c             S   s   g | ]}t j| qS rH   )r   r   )rd   ZconnrH   rH   rI   rf     s    z'Process.connections.<locals>.<listcomp>)r   r   rE   r   r   extendr   )r   r   ri   rH   rH   rI   connections  s    
zProcess.connectionsZmmapzpath rss anon lockedzaddr perms path rss anon lockedc             C   s   dd }| j }g }yt| j|}W n8 tk
r\ } z|jtjkrJtsJg S  W d d }~X Y nX d}x|D ]}|\}}	}
}}}}|||	}|dsyt	
d|| j|f }W nF tk
r } z(|jtjkrd|| j|f }d}n W d d }~X Y nX |||
||||f qhW |r|   |S )Nc             S   s0   dt | dd  dt |dd  df S )Nz%s-%sr   L)hexrY   )startendrH   rH   rI   toaddr  s    z#Process.memory_maps.<locals>.toaddrF[z%s/%s/path/%sT)r   rN   Zproc_memory_mapsr   rs   r   r   r   r   rE   r   r   rt   r   )r   r   r   rv   r   rx   r   r   r   ZaddrsizeZpermr   r&   Zanonr?   rH   rH   rI   memory_maps  s4    


zProcess.memory_mapsc             C   s   t td| j| jf S )Nz%s/%s/fd)r   rE   r   r   r   )r   rH   rH   rI   num_fds  s    zProcess.num_fdsc             C   s   t jt| j| j S )N)r   ZpctxswrN   Zproc_num_ctx_switchesr   r   )r   rH   rH   rI   num_ctx_switches  s    zProcess.num_ctx_switchesNc             C   s   t | j|| jS )N)r   Zwait_pidr   r   )r   ZtimeoutrH   rH   rI   wait  s    zProcess.wait)r   )N)+__name__
__module____qualname____doc__	__slots__r   r   r   r   r   r   r   r   r   r   r   r   rQ   r(   r*   r   r   r%   r   r   rj   r   r   r   r   Zmemory_full_infor+   r   r   r   r   r   Znt_mmap_groupedZnt_mmap_extr   r   r   r   rH   rH   rH   rI   r   t  sT   	


! 

/r   )F)ry   )sr   r   r   rE   r   rO   rU   collectionsr   r   rp   r   r   r   r   r   rN   r	   r
   r   r   r   r   r   r   r   r   r   Z_compatr   r   r   r   r   Z__extra__all__ZgetpagesizerG   ZAF_LINKmaxsizer   r   r   ZSSLEEPZSTATUS_SLEEPINGZSRUNZSTATUS_RUNNINGZSZOMBZSTATUS_ZOMBIEZSSTOPZSTATUS_STOPPEDZSIDLZSTATUS_IDLEZSONPROCZSWAITZSTATUS_WAITINGr   ZTCPS_ESTABLISHEDZCONN_ESTABLISHEDZTCPS_SYN_SENTZCONN_SYN_SENTZTCPS_SYN_RCVDZ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_NONEr   Z	TCPS_IDLEZ
TCPS_BOUNDr   dictr   r1   r4   r7   r<   Zpfullmemr=   r~   _fieldsr@   rJ   rb   rj   rg   rm   rn   ro   Zdisk_io_countersrr   rq   Znet_io_countersZnet_if_addrsr   r   r   r   r   r   r   objectr   rH   rH   rH   rI   <module>   s   


	&	
!
%