o
    iw                     @   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 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j(ej)iZ*n=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j-ejiZ*n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iZ*ej.ej/ej0e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jEiZFeG ZHejIZIeJedZKeLdi dddddd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ZMedGg dHZNedIg dJZOedKg dLZPePZQedMg dNZRedOdPZSedQdRZTeredSg dTZUnedSg dUZUdVdW ZVdXdY ZWdZd[ ZXd\d] ZYd^d_ ZZeserd`da Z[ndbda Z[dcdd Z\erdedf Z]nerdgdf Z]ddidjZ^ej_Z_ej`Z`ejaZaejbZbdkdl Zcdmdn Zder dodp Zedqdr Zfdsdt Zger-zeg ZhW n eiy( Zj zeduej dZhW Y dZj[jndZj[jww dvdw Zkdxdy Zledzd{ Zmd|d} ZnerCd~d ZonerKdd ZonejoZodd Zpejqdd ZrG dd dZsdS )z5FreeBSD, OpenBSD and NetBSD platforms implementation.    N)defaultdict)
namedtuple)ElementTree   )_common)_psposix)_psutil_bsd)FREEBSD)NETBSD)OPENBSD)AccessDenied)NoSuchProcess)ZombieProcess)	conn_tmap)conn_to_ntuple)debug)memoize)memoize_when_activated)usage_percentproc_num_threads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)total	availablepercentusedfreeactiveinactivebufferscachedsharedwired	scputimesusernicesystemidleirqpmem)r8   r:   textdatastack	pcputimes)rT   rV   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_bytes	read_time
write_time	busy_time)re   rf   rg   rh   c                  C   s   t  } trP| \}}}}}}tdd,}|D ]!}|dr(t| d d }	q|dr7t| d d }
qW d    n1 sBw   Y  || }|| }n| \}}}}}}}	}
|| | }|| | }t|| |dd}t||||||||	||
|S )Nz/proc/meminforbs   Buffers:r   i   s
   MemShared:round_)	cextvirtual_memr
   open
startswithintsplitr   rF   )memrG   rK   rL   rM   rQ   rO   flinerN   rP   rJ   availrI    ry   P/home/ubuntu/.local/lib/python3.10/site-packages/ddtrace/vendor/psutil/_psbsd.pyvirtual_memory   s>   


r{   c                  C   s4   t  \} }}}}t|| dd}t| |||||S )z@System swap memory as (total, used, free, sin, sout) namedtuple.r   rm   )ro   swap_memr   r   sswap)rG   rJ   rK   sinsoutrI   ry   ry   rz   swap_memory   s   r   c                  C   s"   t  \} }}}}t| ||||S )z,Return system per-CPU times as a namedtuple.)ro   	cpu_timesrR   rS   ry   ry   rz   r      s   r   c                  C   s>   g } t  D ]}|\}}}}}t|||||}| | q| S )z(Return system CPU times as a namedtuple.)ro   per_cpu_timesrR   append)retcpu_trT   rU   rV   rW   rX   itemry   ry   rz   r      s   r   c                   C      t  S )z0Return the number of logical CPUs in the system.)ro   cpu_count_logicalry   ry   ry   rz   r         r   c                   C   s   t  dkrdS d S )Nr   )r   ry   ry   ry   rz   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   )	ro   cpu_topologyrfindr   
fromstringlenfindallclearr   )r   sindexrootry   ry   rz   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/statrl   s   intrr   N)r	   ro   	cpu_statsr
   rq   rr   rs   rt   r   r   	scpustats)	ctxswintrs
soft_intrssyscalls_traps_faults_forksrv   rw   ry   ry   rz   r      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   rangero   cpu_freqNotImplementedErrorrs   rt   
IndexError
ValueErrorr   r   scpufreq)r   num_cpuscpucurrentavailable_freqmin_freqmax_freqry   ry   rz   r   B  s*     r   c                  C   s   t t } t| ddgS )Ng        )floatro   r   r   r   )currry   ry   rz   r   \  s   Fc           	      C   s@   g }t  }|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.
    )ro   disk_partitionsr   	sdiskpartr   )	allretlist
partitions	partitiondevice
mountpointfstypeoptsntuplery   ry   rz   r   f  s   r   c            
      C   s   t   } i }| D ]M}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
|}d|}d|v }	t
|	||||||< q	|S )z)Get NIC stats (isup, duplex, speed, mtu).N	NicDuplex,running)net_io_counterskeysro   
net_if_mtunet_if_flagsnet_if_duplex_speedOSErrorerrnoENODEVhasattrr   r   join	snicstats)
namesr   rD   mtuflagsduplexspeederroutput_flagsisupry   ry   rz   net_if_stats  s*   







r   c              
   C   s   t |  \}}t }trtd||}ntrtd| }nt||}|D ]}|\}}}}	}
}}t||||	|
|t|}|| q$t	|S )z System-wide network connections.r   )
r   setr   ro   net_connectionsr
   r   TCP_STATUSESaddlist)kindfamiliestypesr   rawlistr   fdfamtypeladdrraddrr   pidntry   ry   rz   r     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   <   )ro   sensors_batteryr   r   POWER_TIME_UNLIMITEDPOWER_TIME_UNKNOWNsbattery)rI   minsleftpower_pluggedsecsleftry   ry   rz   r     s   r   c               	   C   st   t t} t }t|D ],}z!t|\}}|dkrd}d| }| d t|||| W q t	y7   Y qw | S )z?Return CPU cores temperatures if available, else an empty dict.r   NzCore coretemp)
r   r   r   r   ro   sensors_cpu_temperaturer   r   shwtempr   )r   r   r   r   highrD   ry   ry   rz   sensors_temperatures  s   
r   c                   C   r   )z:The system boot time expressed in seconds since the epoch.)ro   	boot_timery   ry   ry   rz   r     r   r   zignoring exception on import: c                 C   sN   t dkr| S t t  }|dkst|dk r| S td |dk r#| | S | | S )z!Account for system clock updates.r   r   z9system clock was updated; adjusting process create_time())INIT_BOOT_TIMEr   absr   )ctimediffry   ry   rz   adjust_proc_create_time  s   
r   c            	      C   sR   g } t  }|D ]}|\}}}}}|dkrqt||pd|||}| | q| S )z:Return currently connected users as a list of namedtuples.~N)ro   usersr   suserr   )	r   r   r   rT   ttyhostnametstampr   r   ry   ry   rz   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)ProcessrD   r   r   ry   ry   ry   rz   _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   )ro   pidsr   r   insert)r   ry   ry   rz   r     s   r   c                 C   s   t | }|s| t v S dS )NTr   
pid_existsr   r   existsry   ry   rz   r   #  s   

r   c                 C   s   t | }|s	dS | t v S )NFr   r   ry   ry   rz   r   .  s   

c                    s   t   fdd}|S )z`Decorator which translates bare OSError exceptions into
    NoSuchProcess and AccessDenied.
    c              
      s   | j | j| j}}}z | g|R i |W S  ty5 } zt|r+t||||t|||d }~w tyF } zt	|||d }~w tj
yY } zt||||d }~w tyv } z|dkrodt v rot	||||d d }~ww )Nr   )r   _ppid_nameProcessLookupErrorro   proc_is_zombier   r   PermissionErrorr   ZombieProcessErrorr   r   )selfargskwargsr   r   rD   r   funry   rz   wrapperA  s(   
z wrap_exceptions.<locals>.wrapper)	functoolswraps)r  r  ry   r  rz   wrap_exceptions<  s   r  c              
   c   s    | j | j| j}}}zdV  W dS  ttfy3 } zt| j r)t||||t|||d}~w t	yD } zt
|||d}~ww )z8Same as above, for routines relying on reading /proc fs.N)r   r  r  r  FileNotFoundErrorro   r  r   r   r  r   )instr   rD   r   r   ry   ry   rz   wrap_exceptions_procfsV  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dKd$d%Zed&d' Zed(d) Zed*d+ ZedLd-d.ZedMd0d1Zed2d3 Zed4d5 Z ed6d7 Z!ed8d9 Z"ed:d; Z#e$d<d=Z%e$d<d>Z&ed?d@ Z'edAdB Z(eredCdD Z)edEdF Z*edGdH Z+edMdIdJZ,d/S d/S )Nr   z1Wrapper class around underlying C implementation.)_cacher  r  r   c                 C   s   || _ d | _d | _d S N)r   r  r  )r  r   ry   ry   rz   __init__m  s   
zProcess.__init__c                 C   s   t | j dS )z+Raise NSP if the process disappeared on us.N)ro   	proc_namer   r  ry   ry   rz   _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.)ro   proc_oneshot_infor   r   kinfo_proc_mapr  r   ry   ry   rz   oneshotx  s   zProcess.oneshotc                 C      | j |  d S r  )r  cache_activater  ry   ry   rz   oneshot_enter     zProcess.oneshot_enterc                 C   r  r  )r  cache_deactivater  ry   ry   rz   oneshot_exit  r!  zProcess.oneshot_exitc                 C   s(   |   td  }|d ur|S t| jS )NrD   )r  r  ro   r  r   )r  rD   ry   ry   rz   rD     s   zProcess.namec                 C   s   t r| jdkr	dS t| jS tr8| jdkrdS t|  td| j dW  d    S 1 s1w   Y  d S dd l}| 	 }|rK|
|d pJdS dS )Nr    z/proc/z/exe)r	   r   ro   proc_exer
   r  osreadlinkshutilcmdlinewhich)r  r(  r)  ry   ry   rz   exe  s   


$zProcess.exec              
   C   s   t r	| jdkr	g S trZzt| jW S  tyY } z;|jtjkrT| j| j| j	}}}t
| jr8t||||t| jsDt||||td|d g W  Y d }~S  d }~ww t| jS )Nr   z	ignoring z and returning an empty list)r   r   r
   ro   proc_cmdliner   r   EINVALr  r  r  r   r   r   r   )r  r   r   rD   r   ry   ry   rz   r)    s$   
zProcess.cmdlinec                 C      t | jS r  )ro   proc_environr   r  ry   ry   rz   environ     zProcess.environc                 C   s8   |   td  }t }z|| W S  ty   Y d S w )Nr$   )r  r  r   get_terminal_mapKeyError)r  tty_nrtmapry   ry   rz   terminal  s   
zProcess.terminalc                 C   s   |   td  | _| jS )Nr   )r  r  r  r  ry   ry   rz   r     s   zProcess.ppidc                 C   .   |   }t|td  |td  |td  S )Nr   r   r   )r  r   puidsr  r  rawtuplery   ry   rz   uids     


zProcess.uidsc                 C   r7  )Nr   r    r"   )r  r   pgidsr  r9  ry   ry   rz   gids  r<  zProcess.gidsc                 C   s8   |   }t|td  |td  |td  |td  S )Nr0   r2   r4   r6   )r  r   r]   r  r9  ry   ry   rz   r     s   



zProcess.cpu_timesc                 C   s   |   td  S )NrB   )r  r  r  ry   ry   rz   cpu_num  s   zProcess.cpu_numc                 C   s@   |   }t|td  |td  |td  |td  |td  S )Nr8   r:   r<   r>   r@   )r  rY   r  r9  ry   ry   rz   memory_info  s   




zProcess.memory_infoFc                 C   s$   |   td  }tr|st|}|S )Nr&   )r  r  r
   r   )r  	monotonicr   ry   ry   rz   r&     s   zProcess.create_timec                 C   s   t rt| jS t|  S r  )HAS_PROC_NUM_THREADSro   r   r   r   threadsr  ry   ry   rz   num_threads	  s   zProcess.num_threadsc                 C   s$   |   }t|td  |td  S )Nr(   r*   )r  r   pctxswr  r9  ry   ry   rz   num_ctx_switches  s
   

zProcess.num_ctx_switchesc                 C   sH   t | j}g }|D ]\}}}t|||}|| q
tr"|   |S r  )ro   proc_threadsr   r   pthreadr   r   r  )r  r   r   	thread_idutimestimer   ry   ry   rz   rC    s   zProcess.threadsinetc              	   C   s   t | \}}g }trt| j|}ntrt| j||}nt| j||}|D ](}|d d \}}}	}
}}tr@||vs?|	|vr@q't|||	|
||t	}|
| q'|   |S )Nr!   )r   r
   ro   r   r   r   proc_net_connectionsr	   r   r   r   r  )r  r   r   r   r   r   r   r   r   r   r   r   r   r   ry   ry   rz   r   %  s$   zProcess.net_connectionsNc                 C   s   t | j|| jS r  )r   wait_pidr   r  )r  timeoutry   ry   rz   wait>  r   zProcess.waitc                 C   r.  r  )ro   proc_priority_getr   r  ry   ry   rz   nice_getB  r1  zProcess.nice_getc                 C   s   t | j|S r  )ro   proc_priority_setr   )r  valuery   ry   rz   nice_setF  s   zProcess.nice_setc                 C   s   |   td  }t|dS )Nr   ?)r  r  PROC_STATUSESget)r  codery   ry   rz   r   J  s   zProcess.statusc                 C   s(   |   }t|td  |td  ddS )Nr,   r.   r   )r  r   pior  r9  ry   ry   rz   io_countersP  s   

zProcess.io_countersc                 C   s   t r	| jdkr	dS t| jS )z)Return process current working directory.r   r$  )r   r   ro   proc_cwdr  ry   ry   rz   cwdZ  s   zProcess.cwdmmapra   rc   c                 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 ry   )r   	popenfile).0pathr   ry   ry   rz   
<listcomp>n  s    z&Process.open_files.<locals>.<listcomp>)ro   proc_open_filesr   )r  r   ry   ry   rz   
open_filesj  s   zProcess.open_filesc                 C   s   t | j}tr|   |S )z=Return the number of file descriptors opened by this process.)ro   proc_num_fdsr   r
   r  r  ry   ry   rz   num_fdsp  s   zProcess.num_fdsc                 C   r.  r  )ro   proc_cpu_affinity_getr   r  ry   ry   rz   cpu_affinity_get|  r1  zProcess.cpu_affinity_getc              
   C   s   t ttt }|D ]}||vrd|d| d}t|qz
t| j| W d S  tyV } z!|j	t	j
t	jhv rQ|D ]}||vrPd|d| d}t||q< d }~ww )Nzinvalid CPU z (choose between ))r   r   r   r   r   ro   proc_cpu_affinity_setr   r   r   r-  EDEADLK)r  cpusallcpusr   msgr   ry   ry   rz   cpu_affinity_set  s*   
zProcess.cpu_affinity_setc                 C   r.  r  )ro   proc_memory_mapsr   r  ry   ry   rz   memory_maps  r1  zProcess.memory_mapsc                 C   sN   |d u rt | j|S t|dkrd|}t||\}}t | j|||S )Nr   z2second argument must be a (soft, hard) tuple, got )ro   proc_getrlimitr   r   r   proc_setrlimit)r  resourcelimitsrn  softhardry   ry   rz   rlimit  s   zProcess.rlimitF)rL  r  )-__name__
__module____qualname____doc__	__slots__r  r  r  r   r  r   r#  rD   r+  r)  r0  r6  r   r;  r>  r   r	   r?  r@  memory_full_infor&   rD  rF  rC  r   rP  rR  rU  r   r[  r]  r   nt_mmap_groupednt_mmap_extrd  rf  rh  ro  rq  rx  ry   ry   ry   rz   r   h  s    








	









	


	


r   ry   ry  )tr}  
contextlibr   r  r&  collectionsr   r   	xml.etreer   r$  r   r   r   ro   r	   r
   r   r   r   r   r   r   r   r   r   r   __extra__all__SIDLSTATUS_IDLESRUNSTATUS_RUNNINGSSLEEPSTATUS_SLEEPINGSSTOPSTATUS_STOPPEDSZOMBSTATUS_ZOMBIESWAITSTATUS_WAITINGSLOCKSTATUS_LOCKEDrW  SDEADSTATUS_WAKINGSONPROCTCPS_ESTABLISHEDCONN_ESTABLISHEDTCPS_SYN_SENTCONN_SYN_SENTTCPS_SYN_RECEIVEDCONN_SYN_RECVTCPS_FIN_WAIT_1CONN_FIN_WAIT1TCPS_FIN_WAIT_2CONN_FIN_WAIT2TCPS_TIME_WAITCONN_TIME_WAITTCPS_CLOSED
CONN_CLOSETCPS_CLOSE_WAITCONN_CLOSE_WAITTCPS_LAST_ACKCONN_LAST_ACKTCPS_LISTENCONN_LISTENTCPS_CLOSINGCONN_CLOSINGPSUTIL_CONN_NONE	CONN_NONEr   getpagesizePAGESIZEAF_LINKr   rB  dictr  rF   rR   rY   pfullmemr]   r`   rb   rd   r{   r   r   r   r   r   r   r   r   
disk_usagedisk_io_countersr   net_if_addrsr   r   r   r   r   r   	Exceptionr   r   r   r   r   r   r  contextmanagerr  r   ry   ry   ry   rz   <module>   sn  	

	
$
/

 






	

