o
    adfc                     @   s6  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! g d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)g d*ZUed+g d,ZVed-g d.ZWed/d0d1gZXeXZYed2g d3ZZed4d5d6[eZj\ Z]d7d8 Z^d9d: Z_d;d< Z`d=d> Zad?d@ ZbdAdB ZcdCdD ZdejeZeejfZfdYdFdGZgejhZhejiZidZdIdJZjdKdL ZkdMdN ZldOdP ZmdQdR ZndSdT ZodUdV ZpG dWdX dXZqdS )[z'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systemidleZiowait	pcputimes)r2   r3   children_userchildren_systemsvmem)total	availablepercentusedfreepmemr&   r'   pmmap_grouped)pathr&   Z	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   Zround_)ossysconf	PAGE_SIZEr   r8   )r9   r=   Zavailr<   r;    rJ   X/var/www/html/software/conda/envs/catlas/lib/python3.10/site-packages/psutil/_pssunos.pyvirtual_memory   s
   rL   c               	   C   s  t  \} }tjddtjd  ddgtjd}| \}}tr&|	t
jj}|jdkr2td|j | d	d
d }|sEd}t|d }}|D ]"}	|	 }	|	dd \}
}|tt|
d 7 }|tt|d 7 }qK|| }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   rF   )cextZswap_mem
subprocessPopenrG   environPIPEcommunicater   decodesysrN   encoding
returncodeRuntimeErrorstripsplitintr   r   ZsswaprI   )sinZsoutprN   _linesmsgr9   r=   linetfr<   r;   rJ   rJ   rK   swap_memory   s:   	
rf   c                  C   s   t  } tdd t|  D  S )z.Return system-wide CPU times as a named tuple.c                 S      g | ]}t |qS rJ   )sum.0xrJ   rJ   rK   
<listcomp>       zcpu_times.<locals>.<listcomp>)rP   per_cpu_timesr1   zipretrJ   rJ   rK   	cpu_times   s   rr   c                  C   s   t  } dd | D S )z6Return system per-CPU times as a list of named tuples.c                 S   s   g | ]}t | qS rJ   )r1   ri   rJ   rJ   rK   rl      rm   z!per_cpu_times.<locals>.<listcomp>)rP   rn   rp   rJ   rJ   rK   rn      s   rn   c                   C   s"   zt dW S  ty   Y dS w )z0Return the number of logical CPUs in the system.SC_NPROCESSORS_ONLNN)rG   rH   
ValueErrorrJ   rJ   rJ   rK   cpu_count_logical   s
   ru   c                   C      t  S )z-Return the number of CPU cores in the system.)rP   cpu_count_coresrJ   rJ   rJ   rK   rw         rw   c                  C   s$   t  \} }}}d}t| |||S )z*Return various CPU stats as a named tuple.r   )rP   	cpu_statsr   Z	scpustats)Zctx_switchesZ
interruptsZsyscallsZtrapsZsoft_interruptsrJ   rJ   rK   ry      s
   ry   Fc                 C   s   g }t  }|D ]G}|\}}}}|dkrd}| s<z	t|js W qW n ty; } ztd||f  W Y d}~qd}~ww d }	}
t|||||	|
}|| q|S )zReturn system disk partitions.none zskipping %r: %sN)	rP   disk_partitions
disk_usager9   OSErrorr   r   Z	sdiskpartappend)allretlistZ
partitions	partitionZdeviceZ
mountpointZfstypeoptserrZmaxfileZmaxpathZntuplerJ   rJ   rK   r|      s,   
r|   c              
   C   s  t j }|dkr|dd | |vr#td| ddd |D f t j|  \}}t|}t }|D ]T}|\}}	}
}}}}|	|vrDq4|
|vrIq4|	t	t
fv r]|rVt j| }|r]t j| }t| }t|	}	t|
}
|dkryt ||	|
||||}n
t ||	|
|||}|| q4t|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).
    r   unixr   z+invalid %r kind argument; choose between %sz, c                 S   rg   rJ   )reprri   rJ   rJ   rK   rl     rm   z#net_connections.<locals>.<listcomp>)r   Z	conn_tmapcopypoprt   joinrP   net_connectionssetr   r	   addrTCP_STATUSESr   r   Zsconnpconnaddlist)kind_pidZcmapZfamiliestypesrawlistrq   itemfdfamtype_laddrraddrr+   pidntrJ   rJ   rK   r   
  s>   



r   c                  C   sT   t  } |  D ]\}}|\}}}}ttdrt|}t||||d| |< q| S )z)Get NIC stats (isup, duplex, speed, mtu).	NicDuplexr{   )rP   net_if_statsitemshasattrr   r   Z	snicstats)rq   namer   ZisupZduplexspeedZmturJ   rJ   rK   r   1  s   

r   c                   C   rv   )z:The system boot time expressed in seconds since the epoch.)rP   	boot_timerJ   rJ   rJ   rK   r   A  rx   r   c                  C   s\   g } t  }d}|D ]!}|\}}}}}}	|sq
||v rd}t|||||	}
| |
 q
| S )z:Return currently connected users as a list of namedtuples.)z:0.0z:0	localhost)rP   usersr   Zsuserr   )r   r   r   r   r2   ttyhostnameZtstampZuser_processr   r   rJ   rJ   rK   r   F  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 rJ   )isdigitr]   ri   rJ   rJ   rK   rl   `  s    zpids.<locals>.<listcomp>)rG   listdirr   r   rJ   rJ   rJ   rK   pids^     r   c                 C   s
   t | S )z&Check for the existence of a unix pid.)r   
pid_exists)r   rJ   rJ   rK   r   c  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   z | g|R i |W S  t tfy*   t| js!t| j| jt| j| j| j ty7   t	| j| j t
yP   | jdkrOdt v rNt	| j| j  w Nr   )r   r   r   r   r   _namer   _ppidr   r
   r~   r   )selfargskwargsfunrJ   rK   wrapperm  s   


z wrap_exceptions.<locals>.wrapper)	functoolswraps)r   r   rJ   r   rK   wrap_exceptionsh  s   r   c                   @   s  e Zd ZdZg d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+d, Ze	d-d. Ze	d/d0 ZeZe	d1d2 Ze	d3d4 Z e	d5d6 Z!d7d8 Z"e	dHd:d;Z#e$d<d=Z%e$d<d>Z&e	d?d@ Z'e	dAdB Z(e	dCdD Z)e	dIdFdGZ*dES )J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   rJ   rJ   rK   __init__  s   zProcess.__init__c                 C   s   t d| j| jf  dS )z+Raise NSP if the process disappeared on us.%s/%sN)rG   statr   r   r   rJ   rJ   rK   _assert_alive  s   zProcess._assert_alivec                 C   (   | j |  | j|  | j|  d S r   )_proc_name_and_argsZcache_activate_proc_basic_info
_proc_credr   rJ   rJ   rK   oneshot_enter     zProcess.oneshot_enterc                 C   r   r   )r   Zcache_deactivater   r   r   rJ   rJ   rK   oneshot_exit  r   zProcess.oneshot_exitc                 C      t | j| jS r   )rP   Zproc_name_and_argsr   r   r   rJ   rJ   rK   r        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s(J |S )Nr   z%s/%s/psinfo)
r   rG   r@   existsr   r
   rP   Zproc_basic_infolenproc_info_map)r   rq   rJ   rJ   rK   r     s   
zProcess._proc_basic_infoc                 C   r   r   )rP   Z	proc_credr   r   r   rJ   rJ   rK   r     r   zProcess._proc_credc                 C   s   |   d S r   )r   r   rJ   rJ   rK   r     s   zProcess.namec                 C   s8   zt d| j| jf W S  ty   Y nw |   dS )Nz%s/%s/path/a.outr{   )rG   readlinkr   r   r~   cmdliner   rJ   rJ   rK   exe  s   zProcess.exec                 C   s   |   d dS )Nr   rC   )r   r\   r   rJ   rJ   rK   r        zProcess.cmdlinec                 C   r   r   )rP   Zproc_environr   r   r   rJ   rJ   rK   rS        zProcess.environc                 C      |   td  S )Nr(   r   r   r   rJ   rJ   rK   r(     r   zProcess.create_timec                 C   r   )Nr*   r   r   rJ   rJ   rK   r*     r   zProcess.num_threadsc                 C   r   )Nr)   r   r   rJ   rJ   rK   nice_get  s   zProcess.nice_getc                 C   s&   | j dv rt| j | jt| j |S )N)r   r   )r   r
   r   
cext_posixsetpriority)r   valuerJ   rJ   rK   nice_set  s   
zProcess.nice_setc                 C   s   |   td  | _| jS )Nr%   )r   r   r   r   rJ   rJ   rK   r%     s   zProcess.ppidc                 C   s^   z|   \}}}}}}W n ty'   |  td  }|  td  }d }Y nw t|||S )Nr-   r.   r   r
   r   r   r   Zpuids)r   real	effectivesavedr`   rJ   rJ   rK   uids     zProcess.uidsc                 C   s^   z|   \}}}}}}W n ty'   |  td  }|  td  }d }Y nw t|||S )Nr/   r0   r   )r   r`   r   r   r   rJ   rJ   rK   gids  r   zProcess.gidsc              
   C   s\   z
t | j| j}W n ty( } z|jtjkrtsd}n W Y d }~nd }~ww tj	| S )N)        r   r   r   )
rP   Zproc_cpu_timesr   r   r~   errno	EOVERFLOW	IS_64_BITr   r5   )r   timesr   rJ   rJ   rK   rr     s   
zProcess.cpu_timesc                 C   r   r   )rP   Zproc_cpu_numr   r   r   rJ   rJ   rK   cpu_num  r   zProcess.cpu_numc              	   C   sx   | j }d}t|  td  }|tjkr2dD ]}ztd|| j|f W   S  t	y1   d}Y qw |r:| 
  d S d S )NFr,   )r   r   r      z%s/%d/path/%dT)r   r   r   r   rP   ZPRNODEVrG   r   r   r   r   )r   procfs_path
hit_enoentr   rk   rJ   rJ   rK   terminal  s    

zProcess.terminalc                 C   sF   | j }ztd|| jf W S  ty"   td|| jf  Y dS w )Nz%s/%s/path/cwdr   r{   )r   rG   r   r   r   r   )r   r   rJ   rJ   rK   cwd*  s   zProcess.cwdc                 C   s2   |   }|td  d }|td  d }t||S )Nr&   i   r'   )r   r   r>   )r   rq   r&   r'   rJ   rJ   rK   memory_info7  s   
zProcess.memory_infoc                 C   s   |   td  }t|dS )Nr+   ?)r   r   PROC_STATUSESget)r   coderJ   rJ   rK   r+   @  s   zProcess.statusc           
      C   s   | j }g }td|| jf }d}|D ]H}t|}zt| j||\}}W n) tyN } z|jtj	kr;t
s;W Y d }~q|jtjkrId}W Y d }~q d }~ww t|||}	||	 q|rb|   |S )Nz	%s/%d/lwpFT)r   rG   r   r   r]   rP   Zquery_process_threadEnvironmentErrorr   r   r   ENOENTr   Zpthreadr   r   )
r   r   rq   Ztidsr   tidutimeZstimer   r   rJ   rJ   rK   threadsF  s0   zProcess.threadsc              	   C   s   g }d}| j }d|| jf }td|| jf D ]1}tj||}tj|rIzt|}W n ty9   d}Y qw t	|rI|
t|t| q|rP|   |S )NFz
%s/%d/pathz%s/%d/fdT)r   r   rG   r   r@   r   islinkr   r   r   r   r   Z	popenfiler]   r   )r   r   r   r   Zpathdirr   r@   filerJ   rJ   rK   
open_filesh  s&   zProcess.open_filesc                 c   s*   dt |g}tj|tjtjd}| \}}tr$dd ||fD \}}|jdkrKd| v r6t| j	| j
d| v rCt| j	| j
td||f |d	d
d }t|D ]:\}}| }|dr|dd
d
 }	||d
   }
|
dkr}tj}
n
|
dkrtj}
nd}
dtj|
|	dtjfV  qXdS )z<Get UNIX sockets used by process by parsing 'pfiles' output.Zpfiles)rN   stderrc                 s   s    | ]
}| tjjV  qd S r   )rV   rW   rN   rX   ri   rJ   rJ   rK   	<genexpr>  s    
z,Process._get_unix_sockets.<locals>.<genexpr>r   zpermission deniedzno such processz%r command error
%srO   r   Nzsockname: AF_UNIXrC   SOCK_STREAM
SOCK_DGRAMr   r{   )strrQ   rR   rT   rU   r   rY   lowerr
   r   r   r   rZ   r\   	enumeratelstrip
startswithr[   socketr   r   AF_UNIXr   	CONN_NONE)r   r   cmdr_   rN   r   ra   irc   r@   typerJ   rJ   rK   _get_unix_sockets}  s<   



zProcess._get_unix_socketsinetc                 C   sP   t || jd}|std| j| jf  |dv r&|dd | | jD  |S )N)r   r   )r   r   c                 S   s   g | ]}t j| qS rJ   )r   r   )rj   connrJ   rJ   rK   rl     s    z'Process.connections.<locals>.<listcomp>)r   r   rG   r   r   extendr	  )r   r   rq   rJ   rJ   rK   connections  s   

zProcess.connectionsmmapzpath rss anon lockedzaddr perms path rss anon lockedc                 C   s  dd }| j }g }z	t| j|}W n ty/ } z|jtjkr*ts*g W  Y d }~S  d }~ww d}|D ]R}|\}}	}
}}}}|||	}|ds{zt	
d|| j|f }W n$ tyz } z|jtjkrod|| j|f }d}n W Y d }~nd }~ww |||
||||f q4|r|   |S )Nc                 S   s0   dt | dd  dt |dd  df S )Nz%s-%sr   L)hexr[   )startendrJ   rJ   rK   toaddr  s   z#Process.memory_maps.<locals>.toaddrF[z%s/%s/path/%sT)r   rP   Zproc_memory_mapsr   r~   r   r   r   r  rG   r   r   r   r   )r   r  r   r   r   r   r   r   r   Zaddrsizepermr   r&   ZanonrA   rJ   rJ   rK   memory_maps  s@   

zProcess.memory_mapsc                 C   s   t td| j| jf S )Nz%s/%s/fd)r   rG   r   r   r   r   rJ   rJ   rK   num_fds  r   zProcess.num_fdsc                 C   s   t jt| j| j S r   )r   ZpctxswrP   Zproc_num_ctx_switchesr   r   r   rJ   rJ   rK   num_ctx_switches  s   zProcess.num_ctx_switchesNc                 C   s   t | j|| jS r   )r   Zwait_pidr   r   )r   timeoutrJ   rJ   rK   wait  r   zProcess.wait)r
  r   )+__name__
__module____qualname____doc__	__slots__r   r   r   r   r   r   r   r   r   r   r   r   rS   r(   r*   r   r   r%   r   r   rr   r   r   r   r   Zmemory_full_infor+   r   r   r	  r  r   Znt_mmap_groupedZnt_mmap_extr  r  r  r  rJ   rJ   rJ   rK   r     s    	







	

	
	






!
"


1

r   )F)r   )rr  r   r   rG   r  rQ   rW   collectionsr   r   r{   r   r   r   r   r   rP   r	   r
   r   r   r   r   r   r   r   r   r   _compatr   r   r   r   r   Z__extra__all__ZgetpagesizerI   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   r5   r8   r>   Zpfullmemr?   r   _fieldsrB   rL   rf   rr   rn   ru   rw   ry   Zdisk_io_countersr}   r|   Znet_io_countersZnet_if_addrsr   r   r   r   r   r   r   r   rJ   rJ   rJ   rK   <module>   s   

.	
"
'