B
    *dH                 @   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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gZe edZ!e edZ"e edZ#e$ Z%ej&Z&ej'e
j(ej)e
j*ej+e
j,ej-e
j,ej.e
j/iZ0ej1e
j2ej3e
j4ej5e
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
jHiZIeJdddddddd d!ZKed"d#d$gZLeLZMed%d&d'd(d)gZNed*d+d,d-d.d/gZOd0d1 ZPd2d3 ZQd4d5 ZRd6d7 ZSd8d9 ZTd:d; ZUd<d= ZVejWZWejXZXdRd?d@ZYejZZZe"rej[Z[dSdBdCZ\dDdE Z]dFdG Z^dHdI Z_dJdK Z`dLdM ZadNdO ZbG dPdQ dQecZddS )TzAIX platform implementation.    N)
namedtuple   )_common)_psposix)_psutil_aix)_psutil_posix)NIC_DUPLEX_FULL)NIC_DUPLEX_HALF)NIC_DUPLEX_UNKNOWN)AccessDenied)NoSuchProcess)ZombieProcess)conn_to_ntuple)get_procfs_path)memoize_when_activated)usage_percent)PY3)FileNotFoundError)PermissionError)ProcessLookupErrorZPROCFS_PATHproc_threadsnet_io_countersproc_io_counters                  )ppidrssvmscreate_timenicenum_threadsstatusttynrpmemr    r!   	scputimesusersystemidleZiowaitsvmemtotal	availablepercentusedfreec              C   s4   t  \} }}}}t| | | dd}t| ||||S )Nr   )round_)cextZvirtual_memr   r,   )r-   Zavailr1   pinnedZinuser/    r5   Z/work/yifan.wang/ringdown/master-ringdown-env/lib/python3.7/site-packages/psutil/_psaix.pyvirtual_memoryj   s    r7   c              C   s:   t  \} }}}| | }t|| dd}t| |||||S )z=Swap system memory as a (total, used, free, sin, sout) tuple.r   )r2   )r3   Zswap_memr   r   Zsswap)r-   r1   sinZsoutr0   r/   r5   r5   r6   swap_memoryp   s    r9   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 r5   )sum).0xr5   r5   r6   
<listcomp>   s    zcpu_times.<locals>.<listcomp>)r3   per_cpu_timesr(   zip)retr5   r5   r6   	cpu_times}   s    rA   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 r5   )r(   )r;   r<   r5   r5   r6   r=      s    z!per_cpu_times.<locals>.<listcomp>)r3   r>   )r@   r5   r5   r6   r>      s    r>   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)ossysconf
ValueErrorr5   r5   r5   r6   cpu_count_logical   s    
rF   c              C   sr   d} t j| dt jt jd}| \}}tr@dd ||fD \}}|jdkrZtd| |f |  }t	|ppd S )Nzlsdev -Cc processorT)shellstdoutstderrc             S   s   g | ]}| tjjqS r5   )decodesysrH   encoding)r;   r<   r5   r5   r6   r=      s   z#cpu_count_cores.<locals>.<listcomp>r   z%r command error
%s)

subprocessPopenPIPEcommunicater   
returncodeRuntimeErrorstrip
splitlineslen)cmdprH   rI   Z
processorsr5   r5   r6   cpu_count_cores   s    

rX   c              C   s    t  \} }}}t| |||S )z*Return various CPU stats as a named tuple.)r3   	cpu_statsr   Z	scpustats)Zctx_switchesZ
interruptsZsoft_interruptsZsyscallsr5   r5   r6   rY      s    rY   Fc          	   C   sl   g }t  }xZ|D ]R}|\}}}}|dkr.d}| s>t|js>qd }}	t||||||	}
||
 qW |S )zReturn system disk partitions.none N)r3   disk_partitions
disk_usager-   r   Z	sdiskpartappend)allretlistZ
partitions	partitionZdeviceZ
mountpointZfstypeoptsZmaxfileZmaxpathntupler5   r5   r6   r\      s    

r\   c             C   s   t j}| |kr.td| ddd |D f t j|  \}}t|}g }x`|D ]X}|\}}	}
}}}}|	|krpqP|
|krzqPt||	|
|||t|dkr|ndd}|| qPW |S )zReturn socket connections.  If pid == -1 return system-wide
    connections (as opposed to connections opened by one process only).
    z+invalid %r kind argument; choose between %sz, c             S   s   g | ]}t |qS r5   )repr)r;   r<   r5   r5   r6   r=      s    z#net_connections.<locals>.<listcomp>rd   N)pid)	r   Z	conn_tmaprE   joinr3   net_connectionsr   TCP_STATUSESr^   )kind_pidZcmapZfamiliestypesrawlistr@   itemfdfamtype_laddrraddrr%   rf   ntr5   r5   r6   rh      s"    

rh   c              C   s   t td} tdd t D }i }x|D ]}t|}t|}d}d}tjdd|gtj	tj	d}|
 \}	}
trd	d |	|
fD \}	}
|jdkrtd
|	}|dk	rt|d}|d}d|}d|k}| |t}t|||||||< q(W |S )z)Get NIC stats (isup, duplex, speed, mtu).)FullZHalfc             S   s   g | ]}|d  qS )r   r5   )r;   r<   r5   r5   r6   r=      s    z net_if_stats.<locals>.<listcomp>r[   r   z/usr/bin/entstatz-d)rH   rI   c             S   s   g | ]}| tjjqS r5   )rJ   rK   rH   rL   )r;   r<   r5   r5   r6   r=      s   z"Running: (\d+) Mbps.*?(\w+) DuplexNr   r   ,running)r   r	   setnet_if_addrs
cext_posixZ
net_if_mtuZnet_if_flagsrM   rN   rO   rP   r   rQ   researchintgrouprg   getr
   r   Z	snicstats)Z
duplex_mapnamesr@   nameZmtuflagsZduplexspeedrW   rH   rI   Z	re_resultZoutput_flagsZisupr5   r5   r6   net_if_stats   s4    





r   c               C   s   t  S )z:The system boot time expressed in seconds since the epoch.)r3   	boot_timer5   r5   r5   r6   r     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)r3   usersr   Zsuserr^   )r`   rm   r   rn   r)   ttyhostnameZtstampZuser_processrf   rt   r5   r5   r6   r     s    
r   c               C   s   dd t t D S )z7Returns a list of PIDs currently running on the system.c             S   s   g | ]}|  rt|qS r5   )isdigitr}   )r;   r<   r5   r5   r6   r=   0  s    zpids.<locals>.<listcomp>)rC   listdirr   r5   r5   r5   r6   pids.  s    r   c             C   s   t jt jt t| dS )z&Check for the existence of a unix pid.Zpsinfo)rC   pathexistsrg   r   str)rf   r5   r5   r6   
pid_exists3  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          	      sz   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X d S )N)
r   r   r   rf   r   _namer   _ppidr   r   )selfargskwargs)funr5   r6   wrapper<  s    
z wrap_exceptions.<locals>.wrapper)	functoolswraps)r   r   r5   )r   r6   wrap_exceptions8  s    r   c               @   sx  e Zd ZdZdddddgZdd Zd	d
 Z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redd ZedAd 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d0d1 Zed2d3 ZeZed4d5 Zd6d7 Z ed8d9 Z!ed:d; Z"edBd=d>Z#e$rted?d@ Z%d<S )CProcessz1Wrapper class around underlying C implementation.rf   r   r   _procfs_path_cachec             C   s   || _ d | _d | _t | _d S )N)rf   r   r   r   r   )r   rf   r5   r5   r6   __init__R  s    zProcess.__init__c             C   s   | j |  | j|  d S )N)_proc_basic_infoZcache_activate
_proc_cred)r   r5   r5   r6   oneshot_enterX  s    zProcess.oneshot_enterc             C   s   | j |  | j|  d S )N)r   Zcache_deactivater   )r   r5   r5   r6   oneshot_exit\  s    zProcess.oneshot_exitc             C   s   t | j| jS )N)r3   Zproc_basic_inforf   r   )r   r5   r5   r6   r   `  s    zProcess._proc_basic_infoc             C   s   t | j| jS )N)r3   Z	proc_credrf   r   )r   r5   r5   r6   r   e  s    zProcess._proc_credc             C   s$   | j dkrdS t| j | jdS )Nr   Zswapper )rf   r3   Z	proc_namer   rstrip)r   r5   r5   r6   r   j  s    
zProcess.namec             C   s   |   }|sdS |d }tjj|krtj|sJtjtj|  |}tj|rttj|rtt	|tj
rt|S tj|}xLtjd dD ]8}tjtj||}tj|rt	|tj
r|S qW dS )Nr[   r   PATH:)cmdlinerC   r   sepisabsabspathrg   cwdisfileaccessX_OKbasenameenvironsplit)r   r   exer   Zpossible_exer5   r5   r6   r   q  s$    zProcess.exec             C   s   t | jS )N)r3   Z	proc_argsrf   )r   r5   r5   r6   r     s    zProcess.cmdlinec             C   s   t | jS )N)r3   Zproc_environrf   )r   r5   r5   r6   r     s    zProcess.environc             C   s   |   td  S )Nr"   )r   proc_info_map)r   r5   r5   r6   r"     s    zProcess.create_timec             C   s   |   td  S )Nr$   )r   r   )r   r5   r5   r6   r$     s    zProcess.num_threadsc             C   sZ   t | j}g }x*|D ]"\}}}t|||}|| qW |sVtd| j| jf  |S )Nz%s/%s)	r3   r   rf   r   Zpthreadr^   rC   statr   )r   rm   r`   	thread_idutimeZstimerc   r5   r5   r6   threads  s    zProcess.threadsinetc             C   s,   t || jd}|s(td| j| jf  |S )N)rk   z%s/%s)rh   rf   rC   r   r   )r   rj   r@   r5   r5   r6   connections  s    zProcess.connectionsc             C   s   t | jS )N)rz   getpriorityrf   )r   r5   r5   r6   nice_get  s    zProcess.nice_getc             C   s   t | j|S )N)rz   setpriorityrf   )r   valuer5   r5   r6   nice_set  s    zProcess.nice_setc             C   s   |   td  | _| jS )Nr   )r   r   r   )r   r5   r5   r6   r     s    zProcess.ppidc             C   s"   |   \}}}}}}t|||S )N)r   r   puids)r   real	effectivesaved_r5   r5   r6   uids  s    zProcess.uidsc             C   s"   |   \}}}}}}t|||S )N)r   r   r   )r   r   r   r   r   r5   r5   r6   gids  s    zProcess.gidsc             C   s   t | j| j}tj| S )N)r3   Zproc_cpu_timesrf   r   r   Z	pcputimes)r   tr5   r5   r6   rA     s    zProcess.cpu_timesc             C   sP   |   td  }|d@ d? |d@ B }x&tdD ]}t|j|kr0|S q0W d S )Nr&   l           i  z	/dev/**/*)r   r   globrC   r   st_rdev)r   Zttydevdevr5   r5   r6   terminal  s    zProcess.terminalc             C   sR   | j }ytd|| jf }|dS  tk
rL   td|| jf  dS X d S )Nz	%s/%s/cwd/z%s/%sr[   )r   rC   readlinkrf   r   r   r   )r   Zprocfs_pathresultr5   r5   r6   r     s    
zProcess.cwdc             C   s2   |   }|td  d }|td  d }t||S )Nr    i   r!   )r   r   r'   )r   r@   r    r!   r5   r5   r6   memory_info  s    zProcess.memory_infoc             C   s   |   td  }t|dS )Nr%   ?)r   r   PROC_STATUSESr   )r   coder5   r5   r6   r%     s    zProcess.statusc             C   s   t jddt| jgt jt jd}| \}}trFdd ||fD \}}d| kr`t| j| j	t
d|}g }xR|D ]J\}}| }|dr|d	d  }| d
krqv|t|t| qvW |S )Nz/usr/bin/procfilesz-n)rH   rI   c             S   s   g | ]}| tjjqS r5   )rJ   rK   rH   rL   )r;   r<   r5   r5   r6   r=     s   z&Process.open_files.<locals>.<listcomp>zno such processz (\d+): S_IFREG.*\s*.*name:(.*)\nz//r   zcannot be retrieved)rM   rN   r   rf   rO   rP   r   lowerr   r   r{   findallrS   
startswithr^   r   Z	popenfiler}   )r   rW   rH   rI   Z	procfilesr`   ro   r   r5   r5   r6   
open_files  s$    
zProcess.open_filesc             C   s(   | j dkrdS ttd| j| j f S )Nr   z%s/%s/fd)rf   rU   rC   r   r   )r   r5   r5   r6   num_fds  s    
zProcess.num_fdsc             C   s   t jt| j S )N)r   Zpctxswr3   Zproc_num_ctx_switchesrf   )r   r5   r5   r6   num_ctx_switches  s    zProcess.num_ctx_switchesNc             C   s   t | j|| jS )N)r   Zwait_pidrf   r   )r   timeoutr5   r5   r6   wait  s    zProcess.waitc             C   sX   yt | j\}}}}W n. tk
rF   t| js@t| j| j Y nX t||||S )N)	r3   r   rf   OSErrorr   r   r   r   Zpio)r   rcwcrbwbr5   r5   r6   io_counters!  s    
zProcess.io_counters)r   )N)&__name__
__module____qualname____doc__	__slots__r   r   r   r   r   r   r   r   r   r   r   r"   r$   HAS_THREADSr   r   r   r   r   r   r   rA   r   r   r   Zmemory_full_infor%   r   r   r   r   HAS_PROC_IO_COUNTERSr   r5   r5   r5   r6   r   M  sJ   
r   )F)rd   )er   r   r   rC   r{   rM   rK   collectionsr   r[   r   r   r   r3   r   rz   r   r	   r
   r   r   r   r   r   r   r   Z_compatr   r   r   r   Z__extra__all__hasattrr   ZHAS_NET_IO_COUNTERSr   ZgetpagesizeZ	PAGE_SIZEZAF_LINKZSIDLZSTATUS_IDLEZSZOMBZSTATUS_ZOMBIEZSACTIVEZSTATUS_RUNNINGZSSWAPZSSTOPZSTATUS_STOPPEDr   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_NONEZ	CONN_NONEri   dictr   r'   Zpfullmemr(   r,   r7   r9   rA   r>   rF   rX   rY   Zdisk_io_countersr]   r\   ry   r   rh   r   r   r   r   r   r   objectr   r5   r5   r5   r6   <module>   s   


		

(