U
    ÷;gˆ ã                M   @   sª  d Z ddlm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	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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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 zddlZW n ek
r   dZY nX 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/m0Z0 ddl/m1Z1 ddl/m2Z2 ddl/m3Z3 ddl/m4Z4 ddl/m5Z5 ddl/m6Z6 zddlm7Z7 W n< ek
r`   e 8¡  e 9d¡ ddl7Z7W 5 Q R X Y nX e0rrddl:Z:nddl;ZdZ:e&rddl<m=Z= ddd d!d"d#d$d%d&d'd(d)d*d+d,d-d.d/d0d1d2d3d4d5d6d7d8d9d:d;d<d=d>d?d@dAdBdCdDdEdFdGdHdIdJdKdLdMdNdOdPdQdRdSdTdUdVdWdXdYdZd[d\d]d^d_d`dadbdcdddedfdgdhdidjgMZ>dkej?kZ@de	jAkZBdle	jAkpVdme	jAkZCeBp`eCZDdne	jAkZEdoe	jAkZFe"r¤eCr¤eGdpƒZHdqeH I¡ kZJW 5 Q R X ndrZJejKdskZLe
 M¡ dtkZNe,dudv„ ƒZOe#ræeOƒ dwkZPeOƒ dxkZQndrZPdrZQdyZRdzZSd{ZTd|ZUeDr$eRd}9 ZReUd}9 ZUeSd~9 ZSeTd}9 ZTe	jVdkr>d€e	 W¡  ZXnde	 W¡  ZXd‚ZYe0rbdƒ Zd„d…¡Z[nd†Z[e \¡  ]¡ d‡kZ^e	j_ `e	j_ ae	j_ bec¡dˆdˆ¡¡Zde	jA ed‰e	j_ aeddŠ¡¡Zfe	j_ `e	j_ bec¡¡Zgehe jid‹ƒZjehe dŒƒZkehe jidƒZlehe dŽƒZmehe jidƒZnehe jidƒZoe&oe' Zpehe d‘ƒZqehe jid’ƒZrehe jid“ƒZsehe jid”ƒZtehe d•ƒZuzeuo\eve  w¡ ƒZxW n eyk
rz   drZxY nX ehe d–ƒZzehe d—ƒZ{ehe jid˜ƒZ|e#s¨e!o²e	 }¡ dkZ~d™dš„ Zeƒ \Z€ZeGe	j‚d›ƒZƒe „eƒj…¡ dœd„ e†e ƒD ƒZ‡eˆedže‰ƒ ƒZŠe‹ƒ ZŒe‹ƒ ZG dŸdH„ dHejŽƒZd d¡„ Zedâd¢dE„ƒZ‘ed£dG„ ƒZ’d¤dF„ Z“ed¥dB„ ƒZ”ed¦d§„ ƒZ•ej–eUfd¨dC„Z—dãd©dD„Z˜dªdZ„ Z™d«dY„ ZšG d¬d­„ d­ƒZ›e›e jœdeUd®d¯d°d\„ ƒZe›e2ežfdeUd®d¯däd²d]„ƒZŸe›eždeUd®d¯d³d[„ ƒZ d´dU„ Z¡dµd¶„ Z¢ej£d·dT„ ƒZ¤d¸dV„ Z¥dåd¹dW„Z¦dæd»dX„Z§G d¼dS„ dSƒZ¨edkr<e¨ZG d½d¾„ d¾ej©ƒZ©e©e_©G d¿dN„ dNe©ƒZªej«j¬e@dÀdÁG dÂdM„ dMeªƒƒZ­dÃdQ„ Z®dÄdR„ Z¯dÅdÆ„ Z°G dÇdO„ dOƒZ±G dÈdP„ dPƒZ²eRfdÉdL„Z³dçdÊdJ„Z´dèdËdK„ZµdédÍd`„Z¶eedfdÎda„Z·ejfdÏdb„Z¸dêdÑdc„Z¹dÒdd„ Zºej£dÓde„ ƒZ»dÔd^„ Z¼dÕdÖ„ Z½d×d_„ Z¾dØdf„ Z¿dÙdg„ ZÀdÚdh„ ZÁdÛdj„ ZÂe&rpej£dëdÜdi„ƒZÃnej£dìdÝdi„ƒZÃej„dÞdß„ ƒZÄe&r¦e ej–dàdá„ ¡ dS )ízTest utilities.é    )Úprint_functionN)ÚAF_INET)ÚAF_INET6)ÚSOCK_STREAM)ÚAIX)ÚLINUX)ÚMACOS)ÚNETBSD)ÚOPENBSD)ÚPOSIX)ÚSUNOS)ÚWINDOWS)Úbytes2human)Údebug)Úmemoize)Úprint_color)Úsupports_ipv6)ÚPY3)ÚFileExistsError)ÚFileNotFoundError)Úrange)Úsuper)Úunicode)Úwhich)ÚmockÚignore)Úwait_pidÚAPPVEYORÚDEVNULLÚGLOBAL_TIMEOUTÚTOLERANCE_SYS_MEMÚ
NO_RETRIESÚPYPYÚ
PYTHON_EXEÚPYTHON_EXE_ENVÚROOT_DIRÚSCRIPTS_DIRÚTESTFN_PREFIXÚUNICODE_SUFFIXÚINVALID_UNICODE_SUFFIXÚ
CI_TESTINGÚVALID_PROC_STATUSESÚTOLERANCE_DISK_USAGEÚIS_64BITÚHAS_CPU_AFFINITYÚHAS_CPU_FREQÚHAS_ENVIRONÚHAS_PROC_IO_COUNTERSÚ
HAS_IONICEÚHAS_MEMORY_MAPSÚHAS_PROC_CPU_NUMÚ
HAS_RLIMITÚHAS_SENSORS_BATTERYÚHAS_BATTERYÚHAS_SENSORS_FANSÚHAS_SENSORS_TEMPERATURESÚHAS_NET_CONNECTIONS_UNIXÚMACOS_11PLUSÚMACOS_12PLUSÚCOVERAGEÚAARCH64Ú	QEMU_USERÚPYTEST_PARALLELÚpyrunÚ	terminateÚreap_childrenÚspawn_testprocÚspawn_zombieÚspawn_children_pairÚ
ThreadTaskÚunittestÚskip_on_access_deniedÚskip_on_not_implementedÚretry_on_failureÚTestMemoryLeakÚPsutilTestCaseÚprocess_namespaceÚsystem_namespaceÚprint_sysinfoÚis_win_secure_system_procÚfake_pytestÚchdirÚsafe_rmpathÚcreate_py_exeÚcreate_c_exeÚ
get_testfnÚ
get_winverÚkernel_versionÚ
call_untilÚwait_for_pidÚwait_for_fileÚcheck_net_addressÚfilter_proc_net_connectionsÚget_free_portÚbind_socketÚbind_unix_socketÚtcp_socketpairÚunix_socketpairÚcreate_socketsÚreload_moduleÚimport_module_by_pathÚwarnÚcopyload_shared_libÚis_namedtupleZ__pypy__ÚGITHUB_ACTIONSZCIBUILDWHEELZCOVERAGE_RUNZPYTEST_XDIST_WORKERz/proc/1/cmdlinez
/bin/qemu-Fl        Úaarch64c                  C   sn   t  ¡ d } ttt|  d¡d d… ƒƒ}|dkrjtjtj	dddgdd	id
d} ttt|  d¡d d… ƒƒ}|S )Nr   Ú.é   )é
   é   z-sSú-cz-import platform; print(platform.mac_ver()[0])ZSYSTEM_VERSION_COMPATÚ0T)ÚenvÚuniversal_newlines)
ÚplatformÚmac_verÚtupleÚmapÚintÚsplitÚ
subprocessÚcheck_outputÚsysÚ
executable)Úversion_strÚversion© r€   ú9/tmp/pip-unpacked-wheel-u0epr1ex/psutil/tests/__init__.pyÚmacos_version•   s    üø
r‚   )rn   é   )é   r   rn   i  P i    é   é   é   Újavaz$psutil-%s-z@psutil-%s-u   -Æ’Å‘Å‘s   fÀ€Úutf8Úsurrogateescapeu   fÃ€Â€)Úasciizus-asciiz..ZPSUTIL_SCRIPTS_DIRÚscriptsÚcpu_affinityÚcpu_freqÚenvironÚ
getloadavgÚioniceÚmemory_mapsÚnet_io_countersÚcpu_numÚio_countersÚrlimitÚsensors_batteryÚsensors_fansÚsensors_temperaturesÚthreadsc                  C   sà   dd„ } t j ¡ }ttdd ƒ}trFtjdkrF|d k	rFtj|d< ||fS trTtj|fS t	r²| tjƒpœ| t j
 tj¡ƒpœ| tdtjd d…  ƒƒpœ| t ¡  ¡ ƒ}|sªtdƒ‚||fS t j
 tj¡}t j
 |¡sÔt|ƒ‚||fS d S )	Nc                 S   s>   zt j| dgt jt jd W n t jk
r4   Y d S X | S d S )Nz-V)ÚstdoutÚstderr)rz   Ú
check_callÚPIPEÚCalledProcessError)Úexer€   r€   r   Úattempt÷   s      ÿ
z_get_py_exe.<locals>.attemptÚ_base_executable)r†   é   Ú__PYVENV_LAUNCHER__zpython%s.%srm   z"can't find python exe real abspath)Úosr   ÚcopyÚgetattrr|   r   Úversion_infor}   rj   r   ÚpathÚrealpathr   ÚpsutilÚProcessr    Ú
ValueErrorÚexistsÚAssertionError)r¡   rr   Úbaser    r€   r€   r   Ú_get_py_exeö   s,    




ÿþür±   zr+c                 C   s    g | ]}|  d ¡rtt|ƒ‘qS )ZSTATUS_©Ú
startswithr§   r«   ©Ú.0Úxr€   r€   r   Ú
<listcomp>$  s    
 r·   ÚAF_UNIXc                       sP   e Zd ZdZ‡ fdd„Zdd„ Zdd„ Zdd	„ Zd
d„ Zdd„ Z	dd„ Z
‡  ZS )rG   z6A thread task which does nothing expect staying alive.c                    s$   t ƒ  ¡  d| _d| _t ¡ | _d S )NFçü©ñÒMbP?)r   Ú__init__Ú_runningÚ	_intervalÚ	threadingÚEventÚ_flag©Úself©Ú	__class__r€   r   rº   5  s    
zThreadTask.__init__c                 C   s   | j j}d|| jt| ƒf S )Nz<%s running=%s at %#x>)rÃ   Ú__name__r»   Úid©rÁ   Únamer€   r€   r   Ú__repr__;  s    zThreadTask.__repr__c                 C   s   |   ¡  | S ©N)ÚstartrÀ   r€   r€   r   Ú	__enter__?  s    zThreadTask.__enter__c                 O   s   |   ¡  d S rÉ   )Ústop)rÁ   ÚargsÚkwargsr€   r€   r   Ú__exit__C  s    zThreadTask.__exit__c                 C   s(   | j rtdƒ‚tj | ¡ | j ¡  dS )zStart thread and keep it running until an explicit
        stop() request. Polls for shutdown every 'timeout' seconds.
        zalready startedN)r»   r­   r½   ÚThreadrÊ   r¿   ÚwaitrÀ   r€   r€   r   rÊ   F  s    zThreadTask.startc                 C   s(   d| _ | j ¡  | j r$t | j¡ qd S )NT)r»   r¿   ÚsetÚtimeÚsleepr¼   rÀ   r€   r€   r   ÚrunO  s    
zThreadTask.runc                 C   s    | j stdƒ‚d| _ |  ¡  dS )z8Stop thread execution and and waits until it is stopped.zalready stoppedFN)r»   r­   ÚjoinrÀ   r€   r€   r   rÌ   U  s    zThreadTask.stop)rÄ   Ú
__module__Ú__qualname__Ú__doc__rº   rÈ   rË   rÏ   rÊ   rÕ   rÌ   Ú__classcell__r€   r€   rÂ   r   rG   2  s   	c                    s   t  ˆ ¡‡ fdd„ƒ}|S )Nc                     s.   zˆ | |ŽW S  t k
r(   tƒ  ‚ Y nX d S rÉ   )Ú	ExceptionrC   ©rÍ   rÎ   ©Úfunr€   r   Úwrapperc  s
    z&_reap_children_on_err.<locals>.wrapper©Ú	functoolsÚwraps©rÞ   rß   r€   rÝ   r   Ú_reap_children_on_errb  s    rä   c                 K   sÞ   |  dt¡ |  dt¡ |  dt ¡ ¡ |  dt¡ trHd}|  d|¡ | dkr¸tt ¡ d}zLt|ƒ d	d
|  d }td|g} t	j
| f|Ž}t |¡ t|ddd W 5 t|ƒ X n"t	j
| f|Ž}t |¡ t|jƒ |S )a  Create a python subprocess which does nothing for some secs and
    return it as a subprocess.Popen instance.
    If "cmd" is specified that is used instead of python.
    By default stdin and stdout are redirected to /dev/null.
    It also attempts to make sure the process is in a reasonably
    initialized state.
    The process is registered for cleanup on reap_children().
    Ústdinr›   Úcwdrr   é   ÚcreationflagsN©Údirzimport time;zopen(r'%s', 'w').close();z&[time.sleep(0.1) for x in range(100)];rp   T©ÚdeleteÚempty)Ú
setdefaultr   r¥   Úgetcwdr$   r   rW   rT   r#   rz   ÚPopenÚ_subprocesses_startedÚaddr\   r[   Úpid)ÚcmdÚkwdsZCREATE_NO_WINDOWÚtestfnZpylineÚsprocr€   r€   r   rD   n  s4    
ÿþÿ



c                  C   s¬   d} t t ¡ d}zzt dtj |¡tf ¡}t	rDt
|dd\}} nt
|ƒ\}} t |j¡}tt|dddƒ}t |¡ t |¡}||fW ¢S t|ƒ | dk	r¦t| ƒ X dS )	a  Create a subprocess which creates another one as in:
    A (us) -> B (child) -> C (grandchild).
    Return a (child, grandchild) tuple.
    The 2 processes are fully initialized and will live for 60 secs
    and are registered for cleanup on reap_children().
    Nré   aV              import subprocess, os, sys, time
            s = "import os, time;"
            s += "f = open('%s', 'w');"
            s += "f.write(str(os.getpid()));"
            s += "f.close();"
            s += "[time.sleep(0.1) for x in range(100 * 6)];"
            p = subprocess.Popen([r'%s', '-c', s])
            p.wait()
            r   )rè   TFrë   )rW   r¥   rï   rT   ÚtextwrapÚdedentr©   Úbasenamer#   r   rA   r«   r¬   ró   rx   r\   Ú_pids_startedrò   )Útfilerö   ÚsÚsubpÚchildZgrandchild_pidZ
grandchildr€   r€   r   rF   ˜  s"    	÷

c                     sÜ   t js
t‚tƒ } t d|  ¡}d}t| ƒ}zŠ| 	t
¡ t|ƒ\}}| ¡ \}}zXt | ¡ gg g t
¡ t| d¡ƒ}t |¡ t  |¡‰ t‡ fdd„ƒ |ˆ fW ¢W ¢S | ¡  X W 5 | ¡  t| ƒ |dk	rÖt|ƒ X dS )z±Create a zombie process and return a (parent, zombie) process tuple.
    In order to kill the zombie parent must be terminate()d first, then
    zombie must be wait()ed on.
    a          import os, sys, time, socket, contextlib
        child_pid = os.fork()
        if child_pid > 0:
            time.sleep(3000)
        else:
            # this is the zombie process
            s = socket.socket(socket.AF_UNIX)
            with contextlib.closing(s):
                s.connect('%s')
                if sys.version_info < (3, ):
                    pid = str(os.getpid())
                else:
                    pid = bytes(str(os.getpid()), 'ascii')
                s.sendall(pid)
        Né   c                      s   ˆ   ¡ tjkS rÉ   )Ústatusr«   ÚSTATUS_ZOMBIEr€   ©Úzombier€   r   Ú<lambda>á  ó    zspawn_zombie.<locals>.<lambda>)r«   r   r¯   rW   rø   rù   ra   ÚcloserT   Ú
settimeoutr   rA   ÚacceptÚselectÚfilenorx   Úrecvrû   rò   r¬   rZ   )Z	unix_fileÚsrcrü   ÚsockÚparentÚconnÚ_Zzpidr€   r  r   rE   ¿  s.    
ñ


c              	   K   sˆ   |  dd¡ |  dd¡ tƒ }zFt|dƒ}| | ¡ W 5 Q R X tt|jgf|Ž}t|jƒ ||fW S  t	k
r‚   t
|ƒ ‚ Y nX dS )zRun python 'src' code string in a separate interpreter.
    Returns a subprocess.Popen instance and the test file where the source
    code was written.
    r›   Nrœ   Úw)rî   rW   ÚopenÚwriterD   r#   rÇ   r[   ró   rÛ   rT   )r  rõ   ÚsrcfileÚfrþ   r€   r€   r   rA   ì  s    

c                 K   sÊ   t rdnd}| dtj¡ | dtj¡ | dd¡ | d|¡ t| tƒrTt | ¡} tj| f|Ž}t	 
|¡ tr‚|jtd\}}n| ¡ \}}|jdkr¤t|| ƒ‚|r°t|ƒ | d	¡rÆ|d
d… }|S )zURun cmd in a subprocess and return its output.
    raises RuntimeError on error.
    rç   r   r›   rœ   rs   Trè   ©ÚtimeoutÚ
Néÿÿÿÿ)r   rî   rz   rž   Ú
isinstanceÚstrÚshlexry   rð   rñ   rò   r   Úcommunicater   Ú
returncodeÚRuntimeErrorrg   Úendswith)rô   rõ   ÚflagsÚpr›   rœ   r€   r€   r   Úsh   s&    




r$  c              
      sò   dd„ ‰dd„ ‰ ‡ ‡‡fdd„}‡ ‡‡fdd„‰‡fd	d
„}dd„ }| }zbt |tƒrb|||ƒW ¢LS t |tjtjfƒr‚ˆ||ƒW ¢,S t |tjƒrœ|||ƒW ¢S t	d| ƒ‚W 5 t |tjtjfƒrÆ||ƒ t |tƒrÔ|n|j}t |¡rìt|ƒ‚X dS )a˜  Terminate a process and wait() for it.
    Process can be a PID or an instance of psutil.Process(),
    subprocess.Popen() or psutil.Popen().
    If it's a subprocess.Popen() or psutil.Popen() instance also closes
    its stdin / stdout / stderr fds.
    PID is wait()ed even if the process is already gone (kills zombies).
    Does nothing if the process does not exist.
    Return process exit status.
    c                 S   sd   t | tjƒrts|  ¡  n
|  |¡ tr`t | tjƒr`zt | j¡ |¡W S  tj	k
r^   Y nX d S rÉ   )
r  rz   rð   r   rÑ   r   r«   r¬   ró   ÚNoSuchProcess©Úprocr  r€   r€   r   rÑ   '  s    

zterminate.<locals>.waitc                 S   s6   t rtrtj}tr(|tjkr(|  tj¡ |  |¡ d S rÉ   )r   rj   ÚsignalÚSIGKILLr   Úsend_signalÚSIGCONT)r'  Úsigr€   r€   r   Úsendsig3  s
    zterminate.<locals>.sendsigc              
      sZ   zˆ | ˆƒ W n@ t k
rN } z"tr0|jdkr0n|jtjkr>‚ W 5 d }~X Y nX ˆ| |ƒS )Né   )ÚOSErrorr   ÚwinerrorÚerrnoZESRCH)r'  r  Úerr©r-  r,  rÑ   r€   r   Úterm_subprocess_proc=  s    z'terminate.<locals>.term_subprocess_procc                    s0   zˆ | ˆƒ W n t jk
r$   Y nX ˆ| |ƒS rÉ   )r«   r%  r&  r3  r€   r   Úterm_psutil_procG  s
    z#terminate.<locals>.term_psutil_procc                    sF   zt  | ¡}W n( t jk
r6   tr2t| |ƒ Y S Y nX ˆ ||ƒS d S rÉ   )r«   r¬   r%  r   r   )ró   r  r'  )r5  r€   r   Úterm_pidN  s    zterminate.<locals>.term_pidc                 S   s4   | j r| j  ¡  | jr | j ¡  | jr0| j ¡  d S rÉ   )r›   r  rœ   rå   )r'  r€   r€   r   Úflush_popenX  s    

zterminate.<locals>.flush_popenzwrong type %rN)
r  rz   rð   r«   rx   ró   Ú
pid_existsr¯   r¬   Ú	TypeError)Zproc_or_pidr,  Úwait_timeoutr4  r6  r7  r#  ró   r€   )r-  r,  r5  rÑ   r   rB     s&    


	
c                 C   s   t  ¡ j| d}tr&t ¡ }t|ƒ qtr<t ¡ }t|ƒ q&|rŒ|D ]}t|dd qDt j|td\}}|D ]}t	d| ƒ t|t
jd qldS )a  Terminate and wait() any subprocess started by this test suite
    and any children currently running, ensuring that no processes stick
    around to hog resources.
    If recursive is True it also tries to terminate and wait()
    all grandchildren started by this process.
    ©Ú	recursiveN)r:  r  z0couldn't terminate process %r; attempting kill())r,  )r«   r¬   Úchildrenrñ   ÚpoprB   rû   Ú
wait_procsr   rg   r(  r)  )r<  r=  rþ   ró   r#  r  Úaliver€   r€   r   rC   r  s    


c                  C   s¬   t stdƒ‚d} t ¡ d }|D ]"}| ¡ s4|dkr>| |7 } q  qDq | sTtd| ƒ‚d}d}|  d¡}t|d ƒ}t|ƒdkrŠt|d ƒ}t|ƒdkr¢t|d ƒ}|||fS )	z"Return a tuple such as (2, 6, 36).z	not POSIXÚ rm   rl   zcan't parse %rr   é   r†   )	r   ÚNotImplementedErrorr¥   ÚunameÚisdigitr­   ry   rx   Úlen)rý   rD  ÚcÚminorÚmicroÚnumsÚmajorr€   r€   r   rY   —  s&    

c                  C   sb   t stdƒ‚t ¡ } t| dƒr*| jp&d}n&t d| d ¡}|rLt| 	d¡ƒnd}| d | d |fS )Nznot WINDOWSÚservice_pack_majorr   z\s\d$r‡   rB  )
r   rC  r|   ÚgetwindowsversionÚhasattrrL  ÚreÚsearchrx   Úgroup)ZwvÚspÚrr€   r€   r   rX   ¯  s    
c                   @   s<   e Zd ZdZeddddfdd„Zdd„ Zdd	„ Zd
d„ ZdS )ÚretryzA retry decorator.Nr¹   c                 C   s2   |r|rt dƒ‚|| _|| _|| _|| _|| _d S )Nz/timeout and retries args are mutually exclusive)r­   Ú	exceptionr  ÚretriesÚintervalÚlogfun)rÁ   rU  r  rV  rW  rX  r€   r€   r   rº   Ã  s    zretry.__init__c                 c   sT   | j r*t ¡ | j  }t ¡ |k rPd V  qn&| jrHt| jƒD ]
}d V  q:nd V  qHd S rÉ   )r  rÓ   rV  r   )rÁ   Ústop_atr  r€   r€   r   Ú__iter__Ó  s    

zretry.__iter__c                 C   s   | j d k	rt | j ¡ d S rÉ   )rW  rÓ   rÔ   rÀ   r€   r€   r   rÔ   ß  s    
zretry.sleepc                    s"   t  ˆ ¡‡ ‡fdd„ƒ}ˆ|_|S )Nc                     s|   d }ˆD ]b}zˆ | |ŽW   S  ˆj k
rh } z,|}ˆjd k	rHˆ |¡ ˆ ¡  W Y ¢qW 5 d }~X Y qX qtrv|‚n‚ d S rÉ   )rU  rX  rÔ   r   )rÍ   rÎ   Úexcr  ©rÞ   rÁ   r€   r   rß   ä  s    

zretry.__call__.<locals>.wrapper)rá   râ   Ú	decorator)rÁ   rÞ   rß   r€   r\  r   Ú__call__ã  s    zretry.__call__)	rÄ   r×   rØ   rÙ   rÛ   rº   rZ  rÔ   r^  r€   r€   r€   r   rT  À  s   ú
rT  r¹   )rU  rX  r  rW  c                 C   s$   | t  ¡ krt  | ¡‚t  | ¡ dS )z„Wait for pid to show up in the process list then return.
    Used in the test suite to give time the sub process to initialize.
    N)r«   Úpidsr%  r¬   ©ró   r€   r€   r   r[   û  s    

Tc              	   C   s:   t | dƒ}| ¡ }W 5 Q R X |s*|s*t‚|r6t| ƒ |S )z8Wait for a file to be written on disk with some content.Úrb)r  Úreadr¯   rT   )Úfnamerì   rí   r  Údatar€   r€   r   r\   
  s    c                 C   s   | ƒ }|st ‚|S )z1Keep calling function until it evaluates to True.)r¯   )rÞ   Úretr€   r€   r   rZ     s    c                 C   sn   dd„ }zLt  | ¡}t |j¡r0t tj| ¡}nt t j| ¡}t	rJ|ƒ  n||ƒ W n t
k
rh   Y nX dS )z?Convenience function for removing temporary test files or dirs.c              
   S   s~   t   ¡ t }t   ¡ |k rvz| ƒ W S  tk
r4   Y n6 tk
rh } z|}tdt|ƒ ƒ W 5 d }~X Y nX t  d¡ q|‚d S )Nzignoring %sg{®Gáz„?)rÓ   r   r   ZWindowsErrorrg   r  rÔ   )rÞ   rY  r  r2  r€   r€   r   Ú	retry_fun0  s    "zsafe_rmpath.<locals>.retry_funN)r¥   ÚstatÚS_ISDIRÚst_moderá   ÚpartialÚshutilÚrmtreeÚremover   r   )r©   rf  ÚstrÞ   r€   r€   r   rT   -  s    
c                 C   s(   zt  | ¡ W n tk
r"   Y nX dS )z.Convenience function for creating a directory.N)r¥   Úmkdirr   ré   r€   r€   r   Ú
safe_mkdirO  s    rp  c              	   c   s.   t  ¡ }zt  | ¡ dV  W 5 t  |¡ X dS )z@Context manager which temporarily changes the current directory.N)r¥   rï   rS   )ÚdirnameÚcurdirr€   r€   r   rS   W  s
    

c                 C   sR   t j | ¡rt| ƒ‚t t| ¡ t t	| ¡ t
rNt  | ¡}t  | |jtjB ¡ | S )z6Create a Python executable file in the given location.)r¥   r©   r®   r¯   ÚatexitÚregisterrT   rk  Úcopyfiler#   r   rg  Úchmodri  ÚS_IEXEC)r©   rn  r€   r€   r   rU   b  s    
c              	   C   s¨   t j | ¡rt| ƒ‚tdƒs&t d¡‚|dkr:t d¡}nt	|t
ƒsLt|ƒ‚t t| ¡ ttdddƒ}| |¡ W 5 Q R X zt d|jd| g¡ W 5 t|jƒ X | S )	z5Create a compiled C executable in the given location.Úgcczgcc is not installedNz‡
            #include <unistd.h>
            int main() {
                pause();
                return 1;
            }
            z.c©Úsuffixr  z-o)r¥   r©   r®   r¯   r   ÚpytestÚskiprø   rù   r  r  rs  rt  rT   r  rW   r  rÇ   rz   r   )r©   Zc_coder  r€   r€   r   rV   m  s    
rA  c                 C   s>   t jt| |d}tj |¡s tj |¡}t t	|¡ |S q dS )zîReturn an absolute pathname of a file or dir that did not
    exist at the time this call is made. Also schedule it for safe
    deletion at interpreter exit. It's technically racy but probably
    not really due to the time variant.
    )Úprefixrz  rê   N)
ÚtempfileÚmktempr'   r¥   r©   r®   rª   rs  rt  rT   )rz  rê   rÇ   r©   r€   r€   r   rW   ‡  s
    c                   @   sT   e Zd ZdZedd„ ƒZeddd„ƒZeddd„ƒZedd
d„ƒZG dd„ dƒZ	dS )rR   zéA class that mimics some basic pytest APIs. This is meant for
    when unit tests are run in production, where pytest may not be
    installed. Still, the user can test psutil installation via:

        $ python3 -m psutil.tests
    c                  O   s4   t  ¡  t¡}t jdd |¡ tjdtdd |S )z‚Mimics pytest.main(). It has the same effect as running
        `python3 -m unittest -v` from the project root directory.
        rm   )Ú	verbosityz<Fake pytest module was used. Test results may be inaccurate.rB  ©Ú
stacklevel)	rH   Z
TestLoaderZdiscoverÚHEREZTextTestRunnerrÕ   Úwarningsrg   ÚUserWarning)rÍ   ÚkwZsuiter€   r€   r   Úmain¢  s    ýzfake_pytest.mainNc                    s.   G dd„ dƒ‰ t jd‡ fdd„	ƒ}|| |dS )zMimics `pytest.raises`.c                   @   s   e Zd ZdZedd„ ƒZdS )z)fake_pytest.raises.<locals>.ExceptionInfoNc                 S   s   | j S rÉ   )Ú_excrÀ   r€   r€   r   Úvalue·  s    z/fake_pytest.raises.<locals>.ExceptionInfo.value)rÄ   r×   rØ   rˆ  Úpropertyr‰  r€   r€   r€   r   ÚExceptionInfo´  s   r‹  Nc              
   3   sv   ˆ ƒ }z
|V  W nT | k
rd } z6|rNt  |t|ƒ¡sNd |t|ƒ¡}t|ƒ‚||_W 5 d }~X Y nX td|  ƒ‚d S )Nz"{}" does not match "{}"z%r not raised)rO  rP  r  Úformatr¯   rˆ  )r[  ÚmatchZeinfor2  Úmsg©r‹  r€   r   Úcontext»  s    
z#fake_pytest.raises.<locals>.context)r  )N)Ú
contextlibÚcontextmanager)r[  r  r  r€   r  r   Úraises°  s    zfake_pytest.raisesc                 C   s"   |rt  ¡  | |¡S t  ¡  | ¡S )zMimics `pytest.warns`.)rH   ÚTestCaseZassertWarnsRegexZassertWarns)Úwarningr  r€   r€   r   ÚwarnsÊ  s    zfake_pytest.warnsrA  c                 C   s   t  | ¡‚dS )zMimics `unittest.SkipTest`.N)rH   ZSkipTest©Úreasonr€   r€   r   r|  Ñ  s    zfake_pytest.skipc                   @   s(   e Zd Zeddd„ƒZG dd„ dƒZdS )zfake_pytest.markrA  c                 C   s   t  | |¡S )z'Mimics `@pytest.mark.skipif` decorator.)rH   ZskipIf)Ú	conditionr˜  r€   r€   r   ÚskipifØ  s    zfake_pytest.mark.skipifc                   @   s"   e Zd ZdZddd„Zdd„ ZdS )zfake_pytest.mark.xdist_groupz4Mimics `@pytest.mark.xdist_group` decorator (no-op).Nc                 C   s   d S rÉ   r€   rÆ   r€   r€   r   rº   à  s    z%fake_pytest.mark.xdist_group.__init__c                 C   s   |S rÉ   r€   )rÁ   Zcls_or_methr€   r€   r   r^  ã  s    z%fake_pytest.mark.xdist_group.__call__)N)rÄ   r×   rØ   rÙ   rº   r^  r€   r€   r€   r   Úxdist_groupÝ  s   
r›  N)rA  )rÄ   r×   rØ   Ústaticmethodrš  r›  r€   r€   r€   r   ÚmarkÖ  s   r  )N)N)rA  )
rÄ   r×   rØ   rÙ   rœ  r‡  r“  r–  r|  r  r€   r€   r€   r   rR   š  s   
c                   @   s&   e Zd Zes"dd„ Zejdd„ ƒZdS )r”  c                 C   s   d S rÉ   r€   rÀ   r€   r€   r   ÚrunTestï  s    zTestCase.runTestc                 o   s
   d V  d S rÉ   r€   )rÁ   rÍ   r†  r€   r€   r   ÚsubTestò  s    zTestCase.subTestN)rÄ   r×   rØ   r   rž  r‘  r’  rŸ  r€   r€   r€   r   r”  ë  s   r”  c                   @   sZ   e Zd ZdZddd„Zdd„ Zdd	„ Zd
d„ Zdd„ Zdd„ Z	dd„ Z
dd„ Zdd„ ZdS )rM   zžTest class providing auto-cleanup wrappers on top of process
    test utilities. All test classes should derive from this one, even
    if we use pytest.
    rA  Nc                 C   s   t ||d}|  t|¡ |S )N)rz  rê   )rW   Ú
addCleanuprT   )rÁ   rz  rê   rc  r€   r€   r   rW     s    zPsutilTestCase.get_testfnc                 O   s   t ||Ž}|  t|¡ |S rÉ   )rD   r   rB   )rÁ   rÍ   rõ   r÷   r€   r€   r   rD     s    
zPsutilTestCase.spawn_testprocc                 C   s*   t ƒ \}}|  t|¡ |  t|¡ ||fS rÉ   )rF   r   rB   )rÁ   Zchild1Zchild2r€   r€   r   rF     s    
z"PsutilTestCase.spawn_children_pairc                 C   s*   t ƒ \}}|  t|¡ |  t|¡ ||fS rÉ   )rE   r   rB   )rÁ   r  r  r€   r€   r   rE     s    
zPsutilTestCase.spawn_zombiec                 O   s*   t ||Ž\}}|  t|¡ |  t|¡ |S rÉ   )rA   r   rT   rB   )rÁ   rÍ   rõ   r÷   r  r€   r€   r   rA     s    zPsutilTestCase.pyrunc                 C   sˆ   t |tjƒst‚|j|jks t‚|j|jks0t‚|jr@|js@t‚t |tjƒrt|j|j	ks\t‚|jd k	rt|jdkstt‚t
|ƒ t|ƒ d S ©Nr   )r  r«   ÚErrorr¯   ró   rÇ   Ú_nameÚZombieProcessÚppidZ_ppidr  Úrepr)rÁ   r'  r[  r€   r€   r   Ú_check_proc_exc  s    

zPsutilTestCase._check_proc_excc              	   C   s¦   t  tj¡4}zt |¡ W n tjk
r:   tdƒ‚Y nX W 5 Q R X |jj|ksVt‚|jj	d ksft‚t 
|¡rxt|ƒ‚|t ¡ ksˆt‚|dd„ t ¡ D ƒks¢t‚d S )Nz&wasn't supposed to raise ZombieProcessc                 S   s   g | ]
}|j ‘qS r€   r`  r´   r€   r€   r   r·   5  s     z0PsutilTestCase.assertPidGone.<locals>.<listcomp>)r{  r“  r«   r%  r¬   r¤  r¯   r‰  ró   rÇ   r8  r_  Úprocess_iter)rÁ   ró   Úcmr€   r€   r   ÚassertPidGone+  s    zPsutilTestCase.assertPidGonec                 C   s¾   |   |j¡ t|ƒ}|j|jddD ]ˆ\}}| j||dl z
|ƒ }W nF tjk
r^   ‚ Y nD tjk
rŒ } z|  	||¡ W 5 d }~X Y nX d||f }t
|ƒ‚W 5 Q R X q$|jdd d S )NT©Úclear_cache©r'  rÇ   z-Process.%s() didn't raise NSP and returned %rr   r  )rª  ró   rN   ÚiterÚallrŸ  r«   r¤  r%  r§  r¯   rÑ   )rÁ   r'  ÚnsrÞ   rÇ   re  r[  rŽ  r€   r€   r   ÚassertProcessGone7  s     
þz PsutilTestCase.assertProcessGonec                 C   sd  t  |j¡}||kst‚ts4ts4t|ƒt|ƒks4t‚| ¡ t jksFt‚| 	¡ sRt‚t  
|j¡sbt‚| ¡  |jt  ¡ ks|t‚|jdd„ t  ¡ D ƒks˜t‚i t _|jdd„ t  ¡ D ƒksºt‚t|ƒ}|j|jddD ]f\}}| j||dJ z
|ƒ  W n8 t jt jfk
r, } z|  ||¡ W 5 d }~X Y nX W 5 Q R X qÒtrÊt t j¡}| ¡  W 5 Q R X |  ||j¡ t t j¡}| ¡  W 5 Q R X |  ||j¡ t t j¡}| ¡  W 5 Q R X |  ||j¡ | ¡  | ¡  | ¡  |  ¡  | 	¡ søt‚t  
|j¡s
t‚|jt  ¡ kst‚|jdd„ t  ¡ D ƒks<t‚i t _|jdd„ t  ¡ D ƒks`t‚d S )	Nc                 S   s   g | ]
}|j ‘qS r€   r`  r´   r€   r€   r   r·   [  s     z6PsutilTestCase.assertProcessZombie.<locals>.<listcomp>c                 S   s   g | ]
}|j ‘qS r€   r`  r´   r€   r€   r   r·   ]  s     Tr«  r­  c                 S   s   g | ]
}|j ‘qS r€   r`  r´   r€   r€   r   r·   y  s     c                 S   s   g | ]
}|j ‘qS r€   r`  r´   r€   r€   r   r·   {  s     )!r«   r¬   ró   r¯   r
   r	   Úhashr  r  Ú
is_runningr8  Úas_dictr_  r¨  Z_pmaprN   r®  r¯  rŸ  r¤  ÚAccessDeniedr§  r   r{  r“  Úcmdliner‰  r    r’   ÚsuspendÚresumerB   Úkill)rÁ   r'  Úcloner°  rÞ   rÇ   r[  r©  r€   r€   r   ÚassertProcessZombieJ  sN    
*z"PsutilTestCase.assertProcessZombie)rA  N)rÄ   r×   rØ   rÙ   rW   rD   rF   rE   rA   r§  rª  r±  r»  r€   r€   r€   r   rM   ü  s   
zunreliable on PYPYr—  c                   @   s¤   e Zd ZdZdZdZdZer dndZdZ	e
 ¡ Zee d¡ƒZedd	„ ƒZed
d„ ƒZdd„ Zdd„ Zdd„ Zdd„ Zdd„ Zdd„ Zdd„ Zddd„Zdd„ ZdS ) rL   aç  Test framework class for detecting function memory leaks,
    typically functions implemented in C which forgot to free() memory
    from the heap. It does so by checking whether the process memory
    usage increased before and after calling the function many times.

    Note that this is hard (probably impossible) to do reliably, due
    to how the OS handles memory, the GC and so on (memory can even
    decrease!). In order to avoid false positives, in case of failure
    (mem > 0) we retry the test for up to 5 times, increasing call
    repetitions each time. If the memory keeps increasing then it's a
    failure.

    If available (Linux, OSX, Windows), USS memory is used for comparison,
    since it's supposed to be more precise, see:
    https://gmpy.dev/blog/2016/real-process-memory-and-environ-in-python
    If not, RSS memory is used. mallinfo() on Linux and _heapwalk() on
    Windows may give even more precision, but at the moment are not
    implemented.

    PyPy appears to be completely unstable for this framework, probably
    because of its JIT, so tests on PYPY are skipped.

    Usage:

        class TestLeaks(psutil.tests.TestMemoryLeak):

            def test_fun(self):
                self.execute(some_function)
    éÈ   rn   r   r…   TZPSUTIL_DEBUGc                 C   s   t  d¡ d S )NF)r«   Ú
_set_debug©Úclsr€   r€   r   Ú
setUpClassµ  s    zTestMemoryLeak.setUpClassc                 C   s   t  | j¡ d S rÉ   )r«   r½  Ú_psutil_debug_origr¾  r€   r€   r   ÚtearDownClass¹  s    zTestMemoryLeak.tearDownClassc                 C   s   | j  ¡ }t|d|jƒS )NZuss)Ú	_thisprocÚmemory_full_infor§   Zrss)rÁ   Úmemr€   r€   r   Ú_get_mem½  s    
zTestMemoryLeak._get_memc                 C   s   t r| j ¡ S | j ¡ S d S rÉ   )r   rÃ  Únum_fdsÚnum_handlesrÀ   r€   r€   r   Ú_get_num_fdsÃ  s    
zTestMemoryLeak._get_num_fdsc                 C   s   | j rt|dtjd d S )NZyellow)ÚcolorÚfile)Úverboser   r|   rœ   )rÁ   rŽ  r€   r€   r   Ú_logÉ  s    zTestMemoryLeak._logc                 C   sx   |   ¡ }|  |¡ |   ¡ }|| }|dk r8|  d| ¡‚|dkrttrHdnd}|dkr\|d7 }d|||f }|  |¡‚dS )	z»Makes sure num_fds() (POSIX) or num_handles() (Windows) does
        not increase after calling a function.  Used to discover forgotten
        close(2) and CloseHandle syscalls.
        r   zHnegative diff %r (gc probably collected a resource from a previous test)ÚfdÚhandlerB  rý   z%s unclosed %s after calling %rN)rÉ  ÚcallÚfailr   )rÁ   rÞ   ÚbeforeÚafterÚdiffÚtype_rŽ  r€   r€   r   Ú
_check_fdsÍ  s     
ÿÿzTestMemoryLeak._check_fdsc                 C   s^   t jdd |  ¡ }t|ƒD ]}|  |¡}~~qt jdd |  ¡ }t jg ksRt‚|| }|S )z€Get 2 distinct memory samples, before and after having
        called fun repeatedly, and return the memory difference.
        rB  )Z
generation)ÚgcZcollectrÆ  r   rÐ  Úgarbager¯   )rÁ   rÞ   ÚtimesZmem1r¶   re  Zmem2rÔ  r€   r€   r   Ú_call_ntimesâ  s    
zTestMemoryLeak._call_ntimesc                 C   s¶   g }d}|}t d|d ƒD ]†}|  ||¡}	d|t|	ƒt|	| ƒ|f }
| |
¡ |	|kp^|	|k}|r||dkrv|  |
¡  d S |dkrŠtƒ  |  |
¡ ||7 }|	}q|  d |¡¡‚d S )Nr   rB  z,Run #%s: extra-mem=%s, per-call=%s, calls=%sz. )r   rÚ  r   ÚappendrÍ  ÚprintrÑ  rÖ   )rÁ   rÞ   rÙ  rV  Ú	toleranceÚmessagesZprev_memZincreaseÚidxrÅ  rŽ  Úsuccessr€   r€   r   Ú
_check_memñ  s.    
ü


zTestMemoryLeak._check_memc                 C   s   |ƒ S rÉ   r€   )rÁ   rÞ   r€   r€   r   rÐ    s    zTestMemoryLeak.callNc              
   C   sè   |dk	r|n| j }|dk	r|n| j}|dk	r0|n| j}|dk	rB|n| j}zD|dksZtdƒ‚|dksjtdƒ‚|dksztdƒ‚|dksŠtdƒ‚W n. tk
rº } ztt|ƒƒ‚W 5 d}~X Y nX |  ||¡ |  |¡ | j	||||d dS )	zTest a callable.NrB  ztimes must be >= 1r   zwarmup_times must be >= 0zretries must be >= 0ztolerance must be >= 0)rÙ  rV  rÝ  )
rÙ  Úwarmup_timesrV  rÝ  r¯   r­   r  rÚ  rÖ  rá  )rÁ   rÞ   rÙ  râ  rV  rÝ  r2  r€   r€   r   Úexecute  s    ÿ
zTestMemoryLeak.executec                    s"   ‡ ‡‡fdd„}ˆj |f|Ž dS )znConvenience method to test a callable while making sure it
        raises an exception on every call.
        c                      s   ˆ  ˆ ˆ¡ d S rÉ   )ÚassertRaisesr€   ©r[  rÞ   rÁ   r€   r   rÐ  +  s    z*TestMemoryLeak.execute_w_exc.<locals>.callN)rã  )rÁ   r[  rÞ   rÎ   rÐ  r€   rå  r   Úexecute_w_exc&  s    zTestMemoryLeak.execute_w_exc)NNNN)rÄ   r×   rØ   rÙ   rÙ  râ  rÝ  r*   rV  rÌ  r«   r¬   rÃ  Úboolr¥   ÚgetenvrÁ  ÚclassmethodrÀ  rÂ  rÆ  rÉ  rÍ  rÖ  rÚ  rá  rÐ  rã  ræ  r€   r€   r€   r   rL   Œ  s2   

       ÿ
c                  C   s®  dd l } dd l}dd l}dd l}dd l}zdd l}W n tk
rL   d }Y nX zdd l}W n tk
rr   d }Y nX |  ¡ }t	j
r˜tdƒr˜tdƒ|d< nzt	jr´dt ¡ d  |d< n^t	jrúdd ttt ¡ ƒ¡ |d< ttdƒrø|d  d	t ¡  7  < nd
t ¡ t ¡ f |d< d	 tt ¡ ƒt ¡ g ¡|d< t	jrJt ¡ d |d< d	 t ¡ t ¡ t  ¡ g¡|d< t!|ddƒ|d< |d k	rš|d  d|j" 7  < t	jròtdƒrÐtddgƒ}t|ƒ #d¡d |d< nd|d< t $¡ d }	|	rò|	|d< t% &¡ |d< | '¡ }
d|
d |
d f |d< |j (t	 )¡ ¡ *d¡|d< |j +¡  *d¡|d< | ,¡ |d< t-j. /d¡|d < t- 0¡ |d!< t1|d"< t 2¡ |d#< t- 3¡ |d$< t	 4¡ |d%< d&t5d'd(„ t	 6¡ D ƒƒ |d)< t	 7¡ }d*t8|j9ƒt:|j;ƒt:|j<ƒf |d+< t	 =¡ }d*t8|j9ƒt:|j;ƒt:|j<ƒf |d,< t>t	 ?¡ ƒ|d-< t	 @¡  A¡ }| Bd.d ¡ | C|¡|d/< tDd0t%jEd1 | F¡ D ]$\}}tDd2|d3 |f t%jEd1 qbtDd0t%jEd1 t%jG H¡  t%jG H¡  d S )4Nr   Úlsb_releasezlsb_release -d -sZOSz	Darwin %szWindows ú Úwin32_editionz, z%s %sÚarchrm   ÚkernelÚpythonÚ__version__znot installedÚpipz (wheel=%s)rx  z	--versionr  rB  Úglibczfs-encodingz%s, %sÚlangz%Y-%m-%d %H:%M:%Sz	boot-timerÓ   Úuserú~Úhomeræ   ZpyexeÚhostnameZPIDZcpusz%.1f%%, %.1f%%, %.1f%%c                 S   s   g | ]}|t  ¡  d  ‘qS )éd   )r«   Ú	cpu_countr´   r€   r€   r   r·   {  s     z!print_sysinfo.<locals>.<listcomp>Zloadavgz%s%%, used=%s, total=%sZmemoryÚswapr_  r’   r'  zF======================================================================©rË  z%-17s %sú:)IÚcollectionsÚdatetimeÚgetpassÚlocaleÚpprintrñ  ÚImportErrorÚwheelÚOrderedDictr«   r   r   r$  ZOSXrt   ru   r   rÖ   rw   r  Ú	win32_verrN  rì  Úsystemr   ÚlistÚarchitectureÚmachiner   rD  Úpython_implementationÚpython_versionÚpython_compilerr§   rð  ry   Úlibc_verr|   ÚgetfilesystemencodingÚ	getlocaleÚfromtimestampÚ	boot_timeÚstrftimeÚnowÚgetuserr¥   r©   Ú
expanduserrï   r#   ÚnodeÚgetpidrù  rv   r   Úvirtual_memoryrx   Úpercentr   ÚusedÚtotalÚswap_memoryrF  r_  r¬   r´  r>  ÚpformatrÜ  rœ   Úitemsr›   Úflush)rý  rþ  rÿ  r   r  rñ  r  ÚinfoÚoutrý   ró  rÅ  rú  ZpinfoÚkÚvr€   r€   r   rP   1  sª    


ÿý


ÿþÿý
ý

c                 C   s8   t dd„ ƒ}z|ƒ |  dkW S  tk
r2   Y dS X d S )Nc                  S   sX   i } t dƒ}| ¡ dd … D ]6}dd„ | d¡D ƒ}|d t|d ƒ }}|| |< q| S )Nztasklist.exe /NH /FO csvrB  c                 S   s   g | ]}|  d d¡‘qS )ú"rA  )Úreplacer´   r€   r€   r   r·   ¤  s     z@is_win_secure_system_proc.<locals>.get_procs.<locals>.<listcomp>ú,r   )r$  Ú
splitlinesry   rx   )re  r!  ÚlineÚbitsrÇ   ró   r€   r€   r   Ú	get_procsŸ  s    
z,is_win_secure_system_proc.<locals>.get_procszSecure SystemF)r   ÚKeyError)ró   r*  r€   r€   r   rQ     s    
	c                  C   s6   t  ¡ } t| dƒr|  ¡ S t| dƒr2t |  ¡ ¡S dS )Nr”   r   r   )r«   r¬   rN  r”   ÚrandomÚchoicer   )r#  r€   r€   r   Ú_get_eligible_cpu¯  s    

r.  c                   @   sJ  e Zd ZdZddi fddi fgZddi fddddifd	di fd
di fddi fddi fddi fddi fddi fddi fg
Zddi fddi fddi fddi fddi fddi fddi fddi fddddifddi fddi fddi fd di fd!di fd"di fd#di fd$di fgZerNed%di fg7 Zed&di fg7 Zed'di fg7 Zed(di fg7 Zerded)di fg7 Ze	rzed*di fg7 Ze
r”ed+ejfi fg7 Zerªed,di fg7 ZerÀed-di fg7 ZerÖed.di fg7 Zerìed/di fg7 Zered0dd1d2ifg7 Zg Zer"eddi fg7 Znedejfi fg7 Ze
rRed+ejd3fi fg7 Ze	rŠerved*ejd4fi fg7 Zned*ejfi fg7 Zer¦ed,eƒ gfi fg7 Zd5ejfi fd6di fd7di fd8di fd9di fgZered5ejfi fg7 Zed5ejfi fg7 Zee e e Zd:d;„ ZdEd<d=„Zd>d?„ Z e!d@dA„ ƒZ"e!dBdC„ ƒZ#dDS )FrN   a  A container that lists all Process class method names + some
    reasonable parameters to be called with. Utility methods (parent(),
    children(), ...) are excluded.

    >>> ns = process_namespace(psutil.Process())
    >>> for fun, name in ns.iter(ns.getters):
    ...    fun()
    Úcpu_percentr€   Zmemory_percentr´  r=  r<  TÚconnectionsr³  Zmemory_info_exZoneshotr  Úparentsró   rÑ   ©r   r¶  Ú	cpu_timesZcreate_timeræ   r    rÄ  Zmemory_inforÇ   Únet_connectionsÚkindr¯  ÚniceZnum_ctx_switchesZnum_threadsZ
open_filesr¥  r  rš   ÚusernameZuidsZgidsZterminalrÇ  r•   r‘   r–   r   r”   r   rÈ  r’   ZgroupedF)r   i   r   r*  r·  r¸  rB   r¹  c                 C   s
   || _ d S rÉ   )Ú_proc)rÁ   r'  r€   r€   r   rº     s    zprocess_namespace.__init__c                 c   sZ   t |ƒ}t |¡ |D ]>\}}}|r,|  ¡  t| j|ƒ}tj|f|ž|Ž}||fV  qdS ©z_Given a list of tuples yields a set of (fun, fun_name) tuples
        in random order.
        N)r  r,  Úshuffler¬  r§   r8  rá   rj  )rÁ   Úlsr¬  Úfun_namerÍ   rõ   rÞ   r€   r€   r   r®    s    
zprocess_namespace.iterc                 C   s   | j j| j jdd dS )z&Clear the cache of a Process instance.T)Z_ignore_nspN)r8  Ú_initró   rÀ   r€   r€   r   r¬  &  s    zprocess_namespace.clear_cachec                 C   s>   |D ]4\}}}d| }t ||ƒsd|jj|f }t|ƒ‚qdS )z}Given a TestCase instance and a list of tuples checks that
        the class defines the required test method names.
        Ztest_z$%r class should define a '%s' methodN)rN  rÃ   rÄ   ÚAttributeError)r¿  Z
test_classr;  r<  r  Ú	meth_namerŽ  r€   r€   r   Útest_class_coverage*  s    
þz%process_namespace.test_class_coveragec                 C   s`   t dd„ | jD ƒƒ}t dd„ | jD ƒƒ}t dd„ ttjƒD ƒƒ}||B |A }|r\td| ƒ‚d S )Nc                 S   s   g | ]}|d  ‘qS r2  r€   r´   r€   r€   r   r·   :  s     z*process_namespace.test.<locals>.<listcomp>c                 S   s   g | ]}|d  ‘qS r2  r€   r´   r€   r€   r   r·   ;  s     c                 S   s   g | ]}|d  dkr|‘qS )r   r  r€   r´   r€   r€   r   r·   <  s      z!uncovered Process class names: %r)rÒ   r¯  Úignoredrê   r«   r¬   r­   )r¿  ÚthisrA  ÚklassZleftoutr€   r€   r   Útest8  s    zprocess_namespace.testN)T)$rÄ   r×   rØ   rÙ   ÚutilsrA  Úgettersr   r1   r2   r5   r«   ZRLIMIT_NOFILEr.   r4   r0   r   r3   ZsettersZNORMAL_PRIORITY_CLASSr   ZIOPRIO_CLASS_NONEZIOPRIO_NORMALr.  r(  ÚSIGTERMZkillersZCTRL_C_EVENTZCTRL_BREAK_EVENTr¯  rº   r®  r¬  ré  r@  rD  r€   r€   r€   r   rN   ¸  s¢   	öïû

c                   @   sº  e Zd ZdZddi fddddifddddifddi fd	dd
difd	dd
difddddifddddifde ¡ fi fddddifddi fddi fddddifde ¡ fi fddi fddi fddi fddi fgZeröe	râe
 ¡ dkrâneddd
difg7 Zereddi fg7 Zer"eddi fg7 Zer8eddi fg7 ZerNed di fg7 Zerted!di fg7 Zed"d#i fg7 Zd$di fd%e ¡ gfi fd&di fd'di fgZeZed(d)„ ƒZejZd*S )+rO   zÝA container that lists all the module-level, system-related APIs.
    Utilities such as cpu_percent() are excluded. Usage:

    >>> ns = system_namespace
    >>> for fun, name in ns.iter(ns.getters):
    ...    fun()
    r  r€   rù  ÚlogicalFTZ	cpu_statsr3  ZpercpuZdisk_io_countersZperdiskZdisk_partitionsr¯  Ú
disk_usager4  r5  Znet_if_addrsZnet_if_statsr“   Zpernicr8  r_  r  Zusersr  Zarm64rŽ   r   r™   r˜   r—   Zwin_service_iterZwin_service_get)Úalgr¨  r?  r/  Zcpu_times_percentc                 c   sL   t | ƒ} t | ¡ | D ]0\}}}tt|ƒ}tj|f|ž|Ž}||fV  qdS r9  )r  r,  r:  r§   r«   rá   rj  )r;  r<  rÍ   rõ   rÞ   r€   r€   r   r®  y  s    

zsystem_namespace.iterN)rÄ   r×   rØ   rÙ   r¥   rï   r  rF  r/   r   rt   r	  ÚHAS_GETLOADAVGr9   r8   r6   r   r«   r¬   rA  r¯  rœ  r®  rN   r@  r€   r€   r€   r   rO   B  sX   	îü
c                 C   s   dd„ }t td| |dS )zZDecorator which runs a test function and retries N times before
    actually failing.
    c                 S   s   t d|  tjd d S )Nz%r, retryingrû  )rÜ  r|   rœ   )r[  r€   r€   r   rX    s    z retry_on_failure.<locals>.logfunN)rU  r  rV  rX  )rT  r¯   )rV  rX  r€   r€   r   rK   ˆ  s       ÿc                    s   ‡ fdd„}|S )z,Decorator to Ignore AccessDenied exceptions.c                    s   t  ˆ ¡‡ ‡fdd„ƒ}|S )Nc                     s@   zˆ | |ŽW S  t jk
r:   ˆd k	r,ˆs,‚ t d¡‚Y nX d S )Nzraises AccessDenied)r«   rµ  r{  r|  rÜ   ©rÞ   Úonly_ifr€   r   rß   ™  s    z9skip_on_access_denied.<locals>.decorator.<locals>.wrapperrà   rã   ©rM  rÝ   r   r]  ˜  s    	z(skip_on_access_denied.<locals>.decoratorr€   ©rM  r]  r€   rN  r   rI   •  s    c                    s   ‡ fdd„}|S )z3Decorator to Ignore NotImplementedError exceptions.c                    s   t  ˆ ¡‡ ‡fdd„ƒ}|S )Nc                     sH   zˆ | |ŽW S  t k
rB   ˆd k	r*ˆs*‚ dˆ j }t |¡‚Y nX d S )Nz4%r was skipped because it raised NotImplementedError)rC  rÄ   r{  r|  )rÍ   rÎ   rŽ  rL  r€   r   rß   ¬  s    ÿÿz;skip_on_not_implemented.<locals>.decorator.<locals>.wrapperrà   rã   rN  rÝ   r   r]  «  s    z*skip_on_not_implemented.<locals>.decoratorr€   rO  r€   rN  r   rJ   ¨  s    ú	127.0.0.1c              
   C   s@   t  t ¡ ¡(}| | df¡ | ¡ d W  5 Q R £ S Q R X dS )z6Return an unused TCP port. Subject to race conditions.r   rB  N)r‘  ÚclosingÚsocketÚbindÚgetsockname)Úhostr  r€   r€   r   r_   Å  s    c                 C   sˆ   |dkr| t tfkrd}t | |¡}z@tjdkrB| tjtjd¡ | |¡ |tj	kr`| 
d¡ |W S  tk
r‚   | ¡  ‚ Y nX dS )zBinds a generic socket.N©rA  r   )ÚntÚcygwinrB  r…   )r   r   rR  r¥   rÇ   Ú
setsockoptÚ
SOL_SOCKETÚSO_REUSEADDRrS  r   ÚlistenrÛ   r  )ÚfamilyÚtypeÚaddrr  r€   r€   r   r`   Ì  s    



c                 C   sr   t js
t‚tj | ¡rt| ƒ‚t tj|¡}z"| | ¡ |tj	krL| 
d¡ W n tk
rl   | ¡  ‚ Y nX |S )zBind a UNIX socket.r…   )r«   r   r¯   r¥   r©   r®   rR  r¸   rS  r   r\  rÛ   r  )rÇ   r^  r  r€   r€   r   ra   Ý  s    


rV  c              
   C   s´   t  t | t¡¡˜}| |¡ | d¡ | ¡ }t | t¡}zJ| |¡ | ¡ }| ¡ \}}||krz||fW W  5 Q R £ S | 	¡  qPW n t
k
r¤   | 	¡  ‚ Y nX W 5 Q R X dS )z^Build a pair of TCP sockets connected to each other.
    Return a (server, client) tuple.
    r…   N)r‘  rQ  rR  r   rS  r\  rT  Úconnectr	  r  r/  )r]  r_  ZllrG  ZcaddrÚar€   r€   r   rb   ì  s    


c                 C   s’   t js
t‚d }}z@t| tjd}| d¡ t tjtj¡}| d¡ | | ¡ W n6 t	k
rˆ   |dk	rr| 
¡  |dk	r‚| 
¡  ‚ Y nX ||fS )zƒBuild a pair of UNIX sockets connected to each other through
    the same UNIX file name.
    Return a (server, client) tuple.
    N©r^  r   )r«   r   r¯   ra   rR  r   Úsetblockingr¸   r`  rÛ   r  )rÇ   ÚserverÚclientr€   r€   r   rc     s    


c               	   c   sì   g } d }}z¨|  ttjtjƒ¡ |  ttjtjƒ¡ tƒ rd|  ttj	tjƒ¡ |  ttj	tjƒ¡ t
r¬tr¬tƒ }tƒ }t|ƒ\}}t|tjd}|||fD ]}|  |¡ qœ| V  W 5 | D ]}|  ¡  qº||fD ]}|dk	rÐt|ƒ qÐX dS )z1Open as many socket families / types as possible.Nrb  )r  rT   rÛ  r`   rR  r   r   Ú
SOCK_DGRAMr   r   r   r:   rW   rc   ra   )ÚsocksZfname1Zfname2rý   rc  Ús1Ús2Zs3r€   r€   r   rd     s*    

c                 C   s  ddl }tr(tr(ts(t|tjƒs(t|ƒ‚|tjkršdd„ |  	d¡D ƒ}t
|ƒdksZt| ƒ‚|D ]"}d|  krvdks^n t| ƒ‚q^tsŽt| ƒ} | | ¡ nd|tjkrÎt| tƒs¶t| ƒ‚tsÂt| ƒ} | | ¡ n0|tjkròt d| ¡dk	sþt| ƒ‚ntd	| ƒ‚dS )
z[Check a net address validity. Supported families are IPv4,
    IPv6 and MAC addresses.
    r   Nc                 S   s   g | ]}t |ƒ‘qS r€   )rx   r´   r€   r€   r   r·   >  s     z%check_net_address.<locals>.<listcomp>rl   r‡   éÿ   z([a-fA-F0-9]{2}[:|\-]?){6}zunknown family %r)Ú	ipaddressÚenumr   r"   r  ÚIntEnumr¯   rR  r   ry   rF  r   ÚIPv4Addressr   r  ÚIPv6Addressr«   ZAF_LINKrO  r  r­   )r_  r]  rk  ZoctsÚnumr€   r€   r   r]   5  s&    
 

c                 C   sT   dd„ }dd„ }dd„ }dd„ }d	d
„ }|| ƒ || ƒ || ƒ || ƒ || ƒ dS )z*Check validity of a connection namedtuple.c                 S   sÔ   t | ƒdk}t | ƒdks$tt | ƒƒ‚| d | jks<t| jƒ‚| d | jksTt| jƒ‚| d | jkslt| jƒ‚| d | jks„t| jƒ‚| d | jksœt| jƒ‚| d | jks´t| jƒ‚|rÐ| d	 | jksÐt| jƒ‚d S )
Nr£   )r.  r£   r   rB  rm   r†   r‡   r…   r.  )	rF  r¯   rÎ  r]  r^  ÚladdrÚraddrr  ró   )r  Zhas_pidr€   r€   r   Úcheck_ntupleS  s    z-check_connection_ntuple.<locals>.check_ntuplec                 S   sð   | j tttfkst| j ƒ‚td k	r:t| j tjƒsNt| ƒ‚nt| j tƒsNt| ƒ‚| j tkrÌt	 	| j | j
¡}t |¡R z| | jd df¡ W n2 t	jk
r¾ } z|jtjkr®‚ W 5 d }~X Y nX W 5 Q R X n | j tkrì| jtjksìt| jƒ‚d S r¡  )r]  r   r   r¸   r¯   rl  r  rm  rx   rR  r^  r‘  rQ  rS  rq  Úerrorr1  ZEADDRNOTAVAILr  r«   Ú	CONN_NONE)r  rý   r2  r€   r€   r   Úcheck_family_  s    
 
z-check_connection_ntuple.<locals>.check_familyc                 S   s†   t tdtƒ ƒ}| jtjtj|fks,t| jƒ‚td k	rLt| jtj	ƒs`t| ƒ‚nt| jt
ƒs`t| ƒ‚| jtjkr‚| jtjks‚t| jƒ‚d S )NÚSOCK_SEQPACKET)r§   rR  Úobjectr^  r   rf  r¯   rl  r  rm  rx   r  r«   ru  )r  rw  r€   r€   r   Ú
check_typet  s    ýüz+check_connection_ntuple.<locals>.check_typec                 S   s¬   | j | jfD ]š}| jttfkr†t|tƒs4tt|ƒƒ‚|s:qt|j	t
ƒsTtt|j	ƒƒ‚d|j	  krjdksvn t|j	ƒ‚t|j| jƒ q| jtkrt|tƒstt|ƒƒ‚qd S )Nr   iÿÿ  )rq  rr  r]  r   r   r  rv   r¯   r^  Úportrx   r]   Úipr¸   r  )r  r_  r€   r€   r   Úcheck_addrsƒ  s    "
z,check_connection_ntuple.<locals>.check_addrsc                 S   s†   t | jtƒst| jƒ‚dd„ ttƒD ƒ}| j|ks<t| jƒ‚| jttfkrl| j	t
krl| jtjks‚t| jƒ‚n| jtjks‚t| jƒ‚d S )Nc                 S   s    g | ]}|  d ¡rtt|ƒ‘qS )ZCONN_r²   r´   r€   r€   r   r·   ’  s    
 zAcheck_connection_ntuple.<locals>.check_status.<locals>.<listcomp>)r  r  r  r¯   rê   r«   r]  r   r   r^  r   ru  )r  Zvalidsr€   r€   r   Úcheck_status  s    ÿz-check_connection_ntuple.<locals>.check_statusNr€   )r  rs  rv  ry  r|  r}  r€   r€   r   Úcheck_connection_ntupleP  s    r~  c                 C   sL   g }| D ]>}t r<|jtjkr<tr<d|jkr<tdt|ƒ ƒ q| |¡ q|S )ztOur process may start with some open UNIX sockets which are not
    initialized by us, invalidating unit tests.
    z/syslogzskipping %s)	r   r]  rR  r¸   r   rr  r   r  rÛ  )ZconsÚnewr  r€   r€   r   r^   ¢  s    c                 C   sP   zddl }t|dƒst‚W n& tk
r@   ddl}| | ¡ Y S X | | ¡S dS )z,Backport of importlib.reload of Python 3.3+.r   NÚreload)Ú	importlibrN  r  Úimpr€  )Úmoduler  r‚  r€   r€   r   re   µ  s    
c                 C   sp   t j t j | ¡¡d }tjd dk r:dd l}| || ¡S dd l}|j	 
|| ¡}|j	 |¡}|j |¡ |S d S )Nr   r†   )r¥   r©   Úsplitextrú   r|   r¨   r‚  Zload_sourceÚimportlib.utilÚutilÚspec_from_file_locationÚmodule_from_specÚloaderÚexec_module)r©   rÇ   r‚  r  ÚspecÚmodr€   r€   r   rf   Ä  s    c                 C   s   t j| tdd dS )zRaise a warning msg.rm   r  N)r„  rg   r…  )rŽ  r€   r€   r   rg   Ø  s    c                 C   sV   t | ƒ}|j}t|ƒdks&|d tk	r*dS t|ddƒ}t|tƒsDdS tdd„ |D ƒƒS )z-Check if object is an instance of namedtuple.rB  r   FÚ_fieldsNc                 s   s   | ]}t |tƒV  qd S rÉ   )r  r  )rµ   Únr€   r€   r   Ú	<genexpr>æ  s     z is_namedtuple.<locals>.<genexpr>)r^  Ú	__bases__rF  rv   r§   r  r¯  )r¶   ÚtÚbr  r€   r€   r   ri   Ý  s    
c                 #   st   t rdnd‰ d‰t| ˆ d}‡ ‡fdd„t ¡  ¡ D ƒ}t |¡}t ||¡ zt
 |¡ |V  W 5 t	|ƒ X dS )zÂCtx manager which picks up a random shared CO lib used
        by this process, copies it in another location and loads it
        in memory via ctypes. Return the new absolutized path.
        Úpypyrï  z.sory  c                    s6   g | ].}t j |j¡d  ˆkrˆ |j ¡ kr|j‘qS )rB  )r¥   r©   r„  Úlowerr´   ©r    Úextr€   r   r·   ô  s    þú'copyload_shared_lib.<locals>.<listcomp>N)r"   rW   r«   r¬   r’   r,  r-  rk  ru  rT   ÚctypesÚCDLL)rz  ÚdstÚlibsr  r€   r•  r   rh   ë  s    
þ


c           	      #   sÔ   ddl m} ddl m} d‰ t| ˆ  d}‡ fdd„t ¡  ¡ D ƒ}trb|sbdd„ t ¡  ¡ D ƒ}t 	|¡}t
 ||¡ d	}zt  |¡}|V  W 5 |d	k	rÆt jjj}|jg|_||jƒ}|dkrÆ|ƒ  t|ƒ X d	S )
zÖCtx manager which picks up a random shared DLL lib used
        by this process, copies it in another location and loads it
        in memory via ctypes.
        Return the new absolutized, normcased path.
        r   )ÚWinError)Úwintypesz.dllry  c                    sF   g | ]>}|j  ¡  ˆ ¡rd tj  |j ¡ ¡ krd|j  ¡ kr|j ‘qS )rï  Zwow64)r©   r”  r!  r¥   rú   r´   ©r–  r€   r   r·     s
   ür—  c                 S   s(   g | ] }d t j |j¡ ¡ kr|j‘qS )r“  )r¥   r©   rú   r”  r´   r€   r€   r   r·     s   þN)r˜  rœ  r  rW   r«   r¬   r’   r"   r,  r-  rk  ru  ÚwindllÚkernel32ÚFreeLibraryZHMODULEÚargtypesÚ_handlerT   ZWinDLL)	rz  rœ  r  rš  r›  r  Úcfiler¡  re  r€   rž  r   rh     s0    

þ
þ





c                   C   s   t dd d S )NTr;  )rC   r€   r€   r€   r   Úcleanup_test_procs7  s    r¥  c                 C   s
   t  | ¡S rÉ   )r|   Úexit)r,  r  r€   r€   r   r  A  r  r  )N)F)TF)N)rA  N)N)N)rP  )rV  )rA  )rA  )ÅrÙ   Ú
__future__r   rs  r‘  r˜  r1  rá   r×  r¥   rt   r,  rO  r
  r  rk  r(  rR  rg  rz   r|   r~  rø   r½   rÓ   rH   r„  r   r   r   r{  r  r«   r   r   r   r	   r
   r   r   r   Zpsutil._commonr   r   r   r   r   Zpsutil._compatr   r   r   r   r   r   r   r   Úcatch_warningsÚsimplefilterrl  Z	unittest2Zpsutil._psposixr   Ú__all__Úbuiltin_module_namesr"   r   r   rj   r*   r=   r@   r  r  rb  r?   Úmaxsizer-   r	  r>   r‚   r;   r<   r!   r    r,   r   rÇ   r  r'   r(   Údecoder)   r  r”  ZASCII_FSr©   rª   rÖ   rq  Ú__file__r%   Úgetr&   rƒ  rN  r¬   r.   r/   r0   rK  r2   r3   r:   ZHAS_NET_IO_COUNTERSr4   r1   r5   r6   rç  r—   r7   rÛ   r8   r9   ZHAS_THREADSÚgetuidZSKIP_SYSCONSr±   r#   r$   Údevnullr   rt  r  rê   r+   r§   rx  r¸   rÒ   rñ   rû   rÐ   rG   rä   rD   rF   rE   rA   r$  rG  rB   rC   rY   rX   rT  r%  r[   r¯   r\   rZ   rT   rp  r’  rS   rU   rV   rW   rR   r”  rM   r  rš  rL   rP   rQ   r.  rN   rO   rK   rI   rJ   r_   r`   ra   rb   rc   rd   r]   r~  r^   re   rf   rg   ri   rh   r¥  r€   r€   r€   r   Ú<module>   s^  



                                                     Þ-








ÿ ÿ






*
ÿ0)
&-

V
%;ü
	üü
"



M
  %l	 F




R3
