XhndZddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl Z ddl Z ddl Z ddl Z ddlZddlZddlZddlZddlZddlZddlZddlZddlZddlZddlZddlZddlZddlZddlZddlmZddlmZddlmZ ddl Z ddl"Z"ddl"m#Z#ddl"m$Z$ddl"m%Z%dd l"m&Z&dd l"m'Z'dd l"m(Z(dd l"m)Z)dd l"m*Z*ddl+m,Z,ddl+m-Z-ddl+m.Z.ddl+m/Z/ddl+m0Z0e(rddl1m2Z2gdZ3dejhvZ5de jlvxsde jlvZ7e7Z8de jlvZ9de jlvZ:ejvdkDZ<e jzj}dvZ?e jzdk(Z@e.dZAe%reAdkDZBeAdk\ZCnd ZBd ZCd!ZDd"ZEd#ZFd$ZGe8reDd%zZDeGd%zZGeEd&zZEeFd%zZFe jd'k(rd(e jd)ZJnd*e jd)ZJd+ZKd,jd-d.ZMejj}d/vZOe jje jje jjeTd0d0ZUe jljd1e jjeUd2ZWe jje jjeTZXeYe"jd3Z[eYe"jd4Z\eYe"d5Z]eYe"jd6Z^eYe"jd7Z_e(xre) Z`eYe"d8ZaeYe"jd9ZbeYe"jd:ZceYe"jd;ZdeYe"d<ZeeYe"d=ZfeYe"d>ZgeYe"jd?Zhe%xse#xre jdk7Zj eexreke"jZm eYe"d@xreke"jZtdAZueu\ZvZwexe jdBZzejezje|e"Dcgc]}|jdCse~e"|c}Ze~edDeZeZeZGdEdFZeej_ne #eZ eejdG<ejGdHdIejZdJZeddKZedLZdMZedNZedOZej$eGfdPZddQZdRZdSZGdTdUZee"j0deGdVWdXZeeefdeGdVWddYZeee jjfdeGdVWdZZd[Zd\Zej>d]Zd^Zdd_Zdd`ZGdadbejHZe jLjOe5dcdGdedfeZdgZdhZdiZGdjdkZGdldmZeDfdnZddoZddpZddqZeedfdrZej>fdsZddtZduZej>dvZdwZdxZdyZdzZd{Zd|Zd}Ze(rej>dd~Znej>ddZejdZe(r ej"ej$dyy#e!$rdZ YFwxYw#en$r8ejejeqejd ZmY-wxYw#en$r8ejejeqejd ZtYJwxYwcc}w)zTest utilities.N)AF_INET)AF_INET6) SOCK_STREAM)AIX)LINUX)MACOS)NETBSD)OPENBSD)POSIX)SUNOS)WINDOWS) bytes2human)debug)memoize) print_color) supports_ipv6)wait_pid)KDEVNULLGLOBAL_TIMEOUTTOLERANCE_SYS_MEM NO_RETRIESPYPY PYTHON_EXEPYTHON_EXE_ENVROOT_DIR SCRIPTS_DIR TESTFN_PREFIXUNICODE_SUFFIXINVALID_UNICODE_SUFFIX CI_TESTINGVALID_PROC_STATUSESTOLERANCE_DISK_USAGEIS_64BITHAS_CPU_AFFINITY HAS_CPU_FREQ HAS_ENVIRONHAS_PROC_IO_COUNTERS HAS_IONICEHAS_MEMORY_MAPSHAS_PROC_CPU_NUM HAS_RLIMITHAS_SENSORS_BATTERY HAS_BATTERYHAS_SENSORS_FANSHAS_SENSORS_TEMPERATURESHAS_NET_CONNECTIONS_UNIX MACOS_11PLUS MACOS_12PLUSCOVERAGEAARCH64PYTEST_PARALLELpyrun terminate reap_children spawn_subproc spawn_zombiespawn_children_pair ThreadTaskunittestskip_on_access_deniedskip_on_not_implementedretry_on_failureTestMemoryLeakPsutilTestCaseprocess_namespacesystem_namespace print_sysinfois_win_secure_system_proc fake_pytestchdir safe_rmpath create_py_exe create_c_exe get_testfn get_winverkernel_version call_until wait_for_pid wait_for_filecheck_net_addressfilter_proc_net_connections get_free_port bind_socketbind_unix_sockettcp_socketpairunix_socketpaircreate_sockets reload_moduleimport_module_by_pathwarncopyload_shared_lib is_namedtuple__pypy__GITHUB_ACTIONS CIBUILDWHEEL COVERAGE_RUNPYTEST_XDIST_WORKERl>arm64aarch64riscv64cBtjd}ttt|j ddd}|dk(rWt jtjdddgdd id }ttt|j ddd}|S) Nr.) z-sS-cz-import platform; print(platform.mac_ver()[0])SYSTEM_VERSION_COMPAT0T)envuniversal_newlines) platformmac_vertuplemapintsplit subprocess check_outputsys executable) version_strversions I/var/www/html/myenv/lib/python3.12/site-packages/psutil/tests/__init__.py macos_versionr~zs""$Q'KC[..s3BQ789G(!--?   )#.#  C!2!23!7!;<= N)rj) rFrjiPijavaz$psutil--z@psutil-u-ƒőősfutf8surrogateescape>us-asciiasciiz..PSUTIL_SCRIPTS_DIRscripts cpu_affinityenviron getloadavgionice memory_mapsnet_io_counterscpu_num io_countersrlimitsensors_battery sensors_fanssensors_temperaturesthreadscpu_freqc 6d}tjj}ttdd}t r,tj dk\r|tj|d<||fStrtj|fStr|tjxs|tjjtjxsc|tjdjtj ddxs(|tj j#}|s t%d||fStjjtj}tjj'|sJ|||fS)Nc tj|dgtjtj|S#tj$rYywxYw)Nz-V)stdoutstderr)rw check_callPIPECalledProcessError)exes r}attemptz_get_py_exe..attemptsI   ! !d JOOJOO  J,,  s6:AA_base_executable)r__PYVENV_LAUNCHER__z python{}.{}riz"can't find python exe real abspath)osrcopygetattrryr version_inforzr`rpathrealpathshutilwhichformatpsutilProcessr ValueErrorexists)rrobasers r} _get_py_exersI **// C 3*D 1D3##v-$2B&)^^ !"Sy ~~s""  CNN # /rww''78 / 1]1133C3CBQ3GHI / v~~'++-. AB BCxggs~~.ww~~c"'C'"Cxrzr+STATUS_AF_UNIXceZdZdZedZedZed dZed dZed dZ ed dZ Gd d Z y) rGzA 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 c4d}tj|y)Nc<tjdtdy)Nz._warn_on_exit's MMN r)atexitregister)rs r}rzfake_pytest._warn_on_exit%s   &rctjjt}tjdj ||S)zMimics pytest.main(). It has the same effect as running `python3 -m unittest -v` from the project root directory. ri) verbosity)r= TestLoaderdiscoverHERETextTestRunnerrun)argskwsuites r}mainzfake_pytest.main0s; ##%..t4!,007 rNc^Gddtjdfd }|||S)zMimics `pytest.raises`.c eZdZdZedZy))fake_pytest.raises..ExceptionInfoNc|jSN)_excselfs r}valuez/fake_pytest.raises..ExceptionInfo.value@s yy r)__name__ __module__ __qualname__rpropertyrrrr} ExceptionInfor=sD  ! !rrc3K} |t|d#|$rF}|r3tj|t|sd|d|d}t|||_Yd}~yd}~wwxYww)Nz not raised"z" does not match ")AssertionErrorresearchstrr)excmatcheinfoerrmsgrs r}contextz#fake_pytest.raises..contextDsz!OE < %wk%:;;  !5#c(!;eW$6se1=C(--   !s*A,A,A))rrridrs r}__repr__zThreadTask.__repr__s6~~&&4& $--RXbMCCrc&|j|Sr)startrs r} __enter__zThreadTask.__enter__s  rc$|jyr)stop)rrkwargss r}__exit__zThreadTask.__exit__s  rc|jr tdtjj ||j j y)zStart thread and keep it running until an explicit stop() request. Polls for shutdown every 'timeout' seconds. zalready startedN)rrrThreadr rwaitrs r}r zThreadTask.starts9 ==./ /t$ rcd|_|jj|jr-tj|j |jr,yy)NT)rrsettimesleeprrs r}rzThreadTask.runs7  mm JJt~~ &mmrc`|js tdd|_|jy)z8Stop thread execution and and waits until it is stopped.zalready stoppedFN)rrjoinrs r}r#zThreadTask.stops%}}./ /  r) rrrr rrr!r%r rr# __classcell__)rs@r}r<r<s*@' D' rr<cBtjfd}|S)NcF |i|S#t$r twxYwr) Exceptionr8)rr$funs r}wrapperz&_reap_children_on_err..wrappers/ '' '  O  s   functoolswraps)r3r4s` r}_reap_children_on_errr8s%__S Nrc |jdt|jdt|jdtj|jdtt rd}|jd||t tj} t|d|d }td |g}tj|fi|}tj|t|d d t||Stj|fi|}tj|t|j|S#t|wxYw) aCreate 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(). stdinrcwdro creationflagsdirzimport time;open(r'z6', 'w').close();[time.sleep(0.1) for x in range(100)];rlTdeleteempty) setdefaultrrgetcwdrr rLrIrrwPopen_subprocesses_startedaddrQrPpid)cmdkwdsCREATE_NO_WINDOWtestfnpylinesprocs r}r9r9s% OOGW%OOHg&OOE299;'OOE>*% )9: { ,   "99  tV,C$$S1D1E ! % %e , &T :   L  --!!%(UYY L  s "AE Ec>d}ttj} tjdtj j |dtd}trt|d\}}nt|\}}tj|j}tt|dd }tj!|tj|}||ft#|| t#|SS#t#|| t#|wwxYw) aCreate 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>zl import subprocess, os, sys, time s = "import os, time;" s += "f = open('z', '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'z.', '-c', s]) p.wait() r)r=TFr@)rLrrDtextwrapdedentrbasenamerr r6rrrHrurQ _pids_startedrGrI)tfilerLssubpchildgrandchild_pid grandchilds r}r;r;s E BIIK (F OO! WW--f567%&0L1    3KD%(KD%txx(]6$eLM.)^^N3 z"F      F     s CDDctjsJt}tjd|d}d}t |} |j tt|\}}|j\}} tj|jgggtt|jd}tj|tj |t#fd|f|j%|j%t'|| t'|SS#|j%wxYw#|j%t'|| t'|wwxYw)zCreate 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 with socket.socket(socket.AF_UNIX) as s: s.connect('za') pid = bytes(str(os.getpid()), 'ascii') s.sendall(pid) NcFjtjk(Sr)statusr STATUS_ZOMBIE)zombiesr}zspawn_zombie..0sv}}&2F2FFr)rr rLrPrQrV settimeoutrr6acceptselectfilenorurecvrSrGrrOcloserI) unix_filesrcrTsockparentconn_zpidr_s @r}r:r:s7 <<< I //&;'   C E I &D 'c ++-a  MM4;;=/2r> Btyy'D   d #^^D)F F GF# JJL I      JJL I     s%6E xYw#t$r t|wxYw)zRun python 'src' code string in a separate interpreter. Returns a subprocess.Popen instance and the test file where the source code was written. rNrw) rCrLopenwriter9rrrPrHr2rI)rhrJsrcfilefrVs r}r6r6;s  OOHd#OOHd#lG '3  1 GGCL j!&&1:T:TXXg    G s" BB ]:NFF||q6F?++  V  t Mrc d d fd} fd fd}d}|} t|tru|||t|tjtjfr||t|tr|n |j }t j |rJ|St|tjtjfru ||t|tjtjfr||t|tr|n |j }t j |rJ|St|tjru|||t|tjtjfr||t|tr|n |j }t j |rJ|Std|#t|tjtjfr||t|tr|n |j }t j |rJ|wxYw)aTerminate 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|j|trJt|tjr/ t j |jj|Syy#t j$rYywxYwr) r(r ryrwrErrrH NoSuchProcess)procrvs r}r(zterminate..waitssd ' z$ (8(89 ~~dhh/44W==:7''  s-A##A98A9ctrtrtj}tr2|tjk7r|j tj |j |yr)rr`signalSIGKILLr send_signalSIGCONT)rsigs r}sendsigzterminate..sendsig|s@ ^..C SFNN*   V^^ , rc |||S#t$rYt$r}tr|jdk(r d}~wwxYw)N)ProcessLookupErrorOSErrorr winerror)rrvrrrr(s r}term_subprocess_procz'terminate..term_subprocess_procsR  D# D'"" "   3<<1,  s  AA?Ac\ |||S#tj$rYwxYwr)rr)rrvrrr(s r}term_psutil_procz#terminate..term_psutil_procs;  D# D'""##   s ++c tj|}||S#tj$rtrt ||cYSYywxYwr)rrrr r)rHrvrrs r}term_pidzterminate..term_pidsQ 3>>#&D $D'2 2 ## .W-- .s!%A  A c|jr|jj|jr|jj|jr|jjyyr)rrfrr:)rs r} flush_popenzterminate..flush_popensL ;; KK    ;; KK    :: JJ    rz wrong type ) ryrurwrErrH pid_existsr TypeError) proc_or_pidr wait_timeoutrrrrrHrrr(s ` @@@r}r7r7hs ##3 A / a A|, a***FLL9 : Na%a155$$S).3.))FNNFLL9 :#A|4 a***FLL9 : Na%a155$$S).3.)):++ ,'<8 a***FLL9 : Na%a155$$S).3.)) k!/0 0 a***FLL9 : Na%a155$$S).3.))sG%)2G%"G%G%%A.Ictjj|}tr&tj }t |tr&t r&t j }t |t r&|rd|D]}t |dtj|t\}}|D],}td|dt |tj.yy)aTerminate 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)rruzcouldn't terminate process z; attempting kill())r) rrchildrenrFpopr7rS wait_procsrr\rr)rrrVrHrrlalives r}r8r8s~~((9(=H $((*$ !#   ,A ad + ,$$X~F5 -A .qe3FG H aV^^ , - rc|ts tdd}tjd}|D]}|j s|dk(r||z }n|st d|d}d}|j d}t|d}t|dk\rt|d}t|dk\rt|d}|||fS) z"Return a tuple such as (2, 6, 36).z not POSIXr rirhz can't parse rrr) r NotImplementedErrorrunameisdigitrrvrulen)rUrcminormicronumsmajors r}rNrNs !+.. A HHJqME  99;!s( FA   <y122 E E 773.wrapper*spC /// I~~C{{. C(JJL sA/AA)r6r7 decorator)rr3r4s`` r}rzretry.__call__)s-      !r) rrrr r2rrr,rrrr}rrs*  &rrr)rrrvrc|tjvrtj|tj|y)zWait 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)rpidsrr)rHs r}rPrP?s/ &++-""3'' NN3rct|d5}|j}ddd|ssJ|r t|S#1swYxYw)z8Wait for a file to be written on disk with some content.rbN)rpreadrI)fnamerArBrsdatas r}rQrQNsJ eT avvx  t E K s ;Ac|}|sJ|S)z1Keep calling function until it evaluates to True.r)r3rets r}rOrO_s %C J3 JrcPd} tj|}tj|jr%t j t j|}n$t j tj|}tr|y||y#t$rYywxYw)z?Convenience function for removing temporary test files or dirs.c6tjtz}tj|kr |S#t$rYn%t$r}|}t d|Yd}~nd}~wwxYwtj dtj|krg`)Nz ignoring g{Gz?)r+rFileNotFoundErrorrr\r,)r3rrlrs r} retry_funzsafe_rmpath..retry_funts ))+.iikG# (u  %  (y&'' ( JJt iikG#s= A*A*A%%A*N) rstatS_ISDIRst_moder6partialrrmtreeremover r)rrstr3s r}rIrIqsv"  WWT] << ###FMM48C##BIIt4C  E cN    sB BB B%$B%cN tj|y#t$rYywxYw)z.Convenience function for creating a directory.N)rmkdirFileExistsErrorr>s r} safe_mkdirrs$      s  $$c#Ktj} tj|dtj|y#tj|wxYww)z@Context manager which temporarily changes the current directory.N)rrDrH)dirnamecurdirs r}rHrHs?YY[F   sA!AA!AA!cRtjj|rJ|tjt |t jt|trFtj|}tj||jtjz|S)z6Create a Python executable file in the given location.)rrrrrrIrcopyfilerr rchmodrS_IEXEC)rrs r}rJrJsiww~~d#)T) # OOK& OOJ% WWT] rzzDLL01 KrcVtjj|rJ|tjdst j d|tjd}nt|tsJ|tjt|ttdd5}|j!|ddd t#j$dj&d|gt|j&|S#1swYDxYw#tj&wxYw) z5Create a compiled C executable in the given location.gcczgcc is not installedNz #include int main() { pause(); return 1; } z.csuffixroz-o)rrrrrrrrPrQryrrrrIrprLrqrwrr)rc_coderss r}rKrKsww~~d#)T) # << kk011 ~"&#&..& OOK& j%s +q uaffdD9:AFF K  AFFs0D #DDD(c tjt||}tjj |s;tjj |}tjt||Sw)zReturn 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. )prefixrr?) tempfilemktemprrrrrrrrI)rr?rrs r}rLrLsU mFLww~~d#77##D)D OOK .K rcTeZdZdZdZddZdZdZdZdZ d Z d Z d Z d Z d Zy)rBzTest class providing auto-cleanup wrappers on top of process test utilities. All test classes should derive from this one, even if we use pytest. c|jj}|jdsd|z}dj||jj|j S)Nzpsutil.z psutil.tests.z{}.{}.{})rr startswithrr_testMethodName)rfqmods r}__str__zPsutilTestCase.__str__sU)) *#e+E   NN # #    rNcLt||}|jt||S)N)rr?)rL addCleanuprI)rrr?rs r}rLzPsutilTestCase.get_testfns!&c2  U+ rcHt|i|}|jt||Sr)r9rr7)rrrJrNs r}r9zPsutilTestCase.spawn_subprocs$t,t,  5) rc|j|i|} tj|jS#tj$r|j |jwxYwr)r9rrrHrassert_pid_gone)rrr$rNs r} spawn_psproczPsutilTestCase.spawn_psprocs["""D3F3 >>%)), ,##    +  s 30A#c|t\}}|jt||jt|||fSr)r;rr7)rchild1child2s r}r;z"PsutilTestCase.spawn_children_pairs5,.  6*  6*rc|t\}}|jt||jt|||fSr)r:rr7)rrjr_s r}r:zPsutilTestCase.spawn_zombie s4%  6*  6*rczt|i|\}}|jt||jt||Sr)r6rrIr7)rrrJrNrrs r}r6zPsutilTestCase.pyruns6--w  W-  5) rct|tjsJ|j|jk(sJ|j|j k(sJ|jr|jsJt|tj r8|j|jk(sJ|j|jdk\sJt|t|yNr) ryrErrorrHr_name ZombieProcessppid_ppidrrepr)rrrs r}_check_proc_exczPsutilTestCase._check_proc_excs#v||,,,ww$(("""xx4::%%% 8888O8 c6// 088tzz) ))xx#xx1}$} C S rc2tjtj5} tj| dddjj|k(sJ|jjJtj|rJ||tjvsJ|tjDcgc]}|jc}vsJy#tj $rtj dwxYw#1swYxYwcc}w)Nz&wasn't supposed to raise ZombieProcess) rrrrrrrrrHrrr process_iter)rrHcmxs r}rzPsutilTestCase.assert_pid_gone"s ]]6// 0 LB Ls# L xx||s"""xx}}$$$$$S).3.)&++-'''&*=*=*?@Q155@@@@ '' Lkk"JKK L L LAs"DCD)DDDc|j|jt|}|j|jdD]<\}}|j t ||5 |}d|d|}t||jdy#tj$rtj$r}|j||Yd}~nd}~wwxYw ddd#1swYxYw)NT clear_cacherrzProcess.z!() didn't raise NSP and returned rru) rrHrCiterallsubTestrrrrrrr()rrnsr3rrrrs r}assert_proc_gonezPsutilTestCase.assert_proc_gone.s TXX& t $T: .IC3t948 . .%C#4&)7$)-- . ! ++++4((s3343 . . .s6&C%(B/C%&C<CC%CC%%C. ctj|j}||k(sJts%tst st |t |k(sJ|jtjk(sJ|jsJtj|jsJ|j|jtjvsJ|jtjDcgc]}|jc}vsJit_|jtjDcgc]}|jc}vsJt|}|j!|j"dD]2\}}|j%t'||5 |ddd4t.rt1j2tj(5}|j5ddd|j-|j6t1j2tj(5}|j9ddd|j-||j6t1j2tj(5}|j;ddd|j-||j6|j=|j?|jA|jC|jsJtj|jsJ|jtjvsJ|jtjDcgc]}|jc}vsJit_|jtjDcgc]}|jc}vsJycc}wcc}w#tj(tj*f$r}|j-||Yd}~sd}~wwxYw#1swYxYw#1swYExYw#1swYxYw#1swYxYwcc}wcc}w)NTr r )"rrrHr r r hashr]r^ is_runningras_dictrr_pmaprCr r rrr AccessDeniedrrrrcmdlinerrrsuspendresumer7kill) rrclonerrr3rrrs r}assert_proc_zombiez!PsutilTestCase.assert_proc_zombieAsEtxx( u}}6U:e, ,,{{} 4 4444     *** xx6;;=(((xx6+>+>+@AaAEEAAAA xx6+>+>+@AaAEEAAAA t $T: 4IC3t948 44E 4 4 4 v334     rxx 0v334     rxx 0v334 #  " #  rxx 0         ***xx6;;=(((xx6+>+>+@AaAEEAAAA xx6+>+>+@AaAEEAAAAABA,,f.A.AB4((s334 4 4     # #BAsl;OO#P'%O$ P48QQ;QQ $#P$PP'P$$P''P1 4P>Q Qr N)rrrr rrLr9rr;r:r6rrrrrrr}rBrBs@       A&3BrrBzunreliable on PYPYrceZdZdZdZdZdZerdndZdZ e jZ e ejdZedZed Zd Zd Zd Zd ZdZdZdZ ddZdZy)rAaTest 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) rjrrT PSUTIL_DEBUGc.tjdy)NF)r _set_debugclss r} setUpClasszTestMemoryLeak.setUpClasss% rcBtj|jyr)rr"_psutil_debug_origr#s r} tearDownClasszTestMemoryLeak.tearDownClasss#001rcd|jj}t|d|jS)Nuss) _thisprocmemory_full_inforrss)rmems r}_get_memzTestMemoryLeak._get_mems)nn--/sE377++rcvtr|jjS|jjSr)r r+num_fds num_handlesrs r} _get_num_fdszTestMemoryLeak._get_num_fdss+ >>))+ +>>--/ /rcV|jrt|dtjyy)Nyellow)colorfile)verboserryr)rrs r}_logzTestMemoryLeak._logs << 8#** = rc |j}|j||j}||z }|dkrd|d}tj||dkDr3trdnd}|dkDr|dz }|d|d |}tj|y ) zMakes sure num_fds() (POSIX) or num_handles() (Windows) does not increase after calling a function. Used to discover forgotten close(2) and CloseHandle syscalls. rznegative diff z8 (gc probably collected a resource from a previous test)fdhandlerrUz unclosed z after calling N)r3callrrr )rr3beforeafterdiffrtype_s r} _check_fdszTestMemoryLeak._check_fdss ""$ #!!#v~ !8 )22 ++c" " !8!DxEax F*UG?3'BC++c" " rctjd|j}t|D]}|j |}~~tjd|j}tj gk(sJ||z }|S)zGet 2 distinct memory samples, before and after having called fun repeatedly, and return the memory difference. r) generation)gccollectr/rr=garbage)rr3timesmem1rrmem2r@s r} _call_ntimeszTestMemoryLeak._call_ntimessy a }}u A))C.C3  a }}zzRd{ rc g}d}|}td|dzD]}|j||} dj|t| t| |z |} |j | | |kxs| |k} | r|dkDr|j | y|dk(r t |j | ||z }| }tjdj|)Nrrz,Run #{}: extra-mem={}, per-call={}, calls={}z. ) rrKrrappendr9printrrr.) rr3rHr tolerancemessagesprev_memincreaseidxr.rsuccesss r} _check_memzTestMemoryLeak._check_memsGaK( C##C/C@GGC C%K( C OOC Y&9#/G7IIcN!8G #!' (kk$))H-..rc|Srr)rr3s r}r=zTestMemoryLeak.calls u rNc||n |j}||n |j}||n |j}||n |j} |dk\sJd|dk\sJd|dk\sJd|dk\sJd |j|||j||j||||y#t$r}t t |d}~wwxYw) zTest a callable.Nrztimes must be >= 1rzwarmup_times must be >= 0zretries must be >= 0ztolerance must be >= 0)rHrrO) rH warmup_timesrrOrrrrKrBrU)rr3rHrXrrOrs r}executezTestMemoryLeak.execute s* (4L$:K:K %0'dll!*!6IDNN  'A: 33 3:1$ A&A A$a< 7!7 7<> ;#; ;> #|,  5'YO  'SX& & 's0B,, C5C  Cc :fd}|j|fi|y)znConvenience method to test a callable while making sure it raises an exception on every call. c^ tjd#$rYywxYw)Nz did not raise )rr)rr3sr}r=z*TestMemoryLeak.execute_w_exc..call%s> @kkSE">??  s $,,N)rY)rrr3r$r=s `` r} execute_w_exczTestMemoryLeak.execute_w_exc s  @  T$V$r)NNNN)rrrr rHrXrOr rr8rrr+boolrgetenvr' classmethodr%r(r/r3r9rBrKrUr=rYr\rrr}rArAs> ELIbAGG Iibii78!!22, 0 >#, /8KOP, %rrAc4 ddl}ddl}ddl}ddl}ddl} ddl} ddl}|j}tjr$tjdrtd|d<ntjrdtj d|d<ntj"rjddj%t't(tj*z|d<t-tdrQ|dxxd tj.zz cc<n.tj0dtj2|d<d j%t5tj6tj8gz|d <tj:rtj<d |d <d j%tj>tj@tjBg|d <tE|dd|d<||dxxd|jFdz cc<tj:rftjdr.tddg}t)|jIdd|d<nd|d<tjJd} | r| |d<tMjN|d<|jQ} | dd | d|d<|jjStjTjWd|d<|jjYjWd|d<|j[|d<t\j^jad|d<t]jb|d <td|d!<tjf|d"<t]jh|d#<tjj|d$<d%jltod&tjpD|d'<tjr} d(jmtu| jvty| jzty| j||d)<tj~} d(jmtu| jvty| jzty| j||d*<ttjD cgc]\} }| js|d+us| c}} }d,j%||d-<ttj|d.<tjj}|jd/dtt\jD cic]} | t\j| c} |d0<|j||d1<td2tLj3|jD]3\} }td4jm| d5z|tLj35td2tLj3tLjjy#t $rd}Y_wxYw#t $rd}YlwxYwcc}} wcc} w)6Nr lsb_releasezlsb_release -d -sOSzDarwin zWindows  win32_editionz, archrikernelpython __version__z not installedpipz (wheel=)rz --versionrwrglibcz fs-encodinglangz%Y-%m-%d %H:%M:%Sz boot-timer+user~homer;pyexehostnamePIDcpusz{:.1f}%, {:.1f}%, {:.1f}%c3PK|]}|tjz dz yw)dN)r cpu_count).0rs r} z print_sysinfo..zs#Iq6##%%+Is$&loadavgz{}%%, used={}, total={}memoryswapTz constantsrrrrzF======================================================================r7z {:<17} {}:)O collectionsdatetimegetpasslocalepprintri ImportErrorwheel OrderedDictrrrrrOSXrqrrr r.rtr win32_verhasattrrdsystemr|list architecturemachiner rpython_implementationpython_versionpython_compilerrrhrvlibc_verrygetfilesystemencoding getlocale fromtimestamp boot_timestrftimenowgetuserrr expanduserrDrnodegetpidrvrrsrvirtual_memoryrupercentrusedtotal swap_memorysortedglobalsitemsisupperrrrrrrpformatrNrrflush)rrrrrririnfooutrUrlr.r{kvr|pinfos r}rErE0sM  " " $D|| ]3+,T x//1!456T #((3sH4F4F4H+I"JJT 8_ - J$!7!7!99 9J )*!H,<,<,>+?@T 99 X " " $%)9)9);(<<DL||!)!,XYY&&(!  " DN #}o>DK  U %"3"3!4A66 || << e[)*Cc(...q1DK)DK     " DM335D    D1gYba *DL ))77h"# $$((*334GHDL??$DL77%%c*DL))+DKDM}}D))+DK##%DL8188 IV5F5F5HI IDO    !C.55 CKKCHHCIIDN    D,33 DLLDIIDJJDLy(Ftq!AIIKAIFI/33I>Dv{{}%DL NN  $ $ &E IImT"282DEQ2::a=(EE)>>%(DL ($ ?1 k  S!,3::>? ($JJE  \ GFs:Y+Y=.Z Z Z Z+ Y:9Y:= Z  Z cTtd} ||dk(S#t$rYywxYw)Nci}td}|jddD]H}|jdDcgc]}|jdd}}|dt |d}}|||<J|Scc}w)Nztasklist.exe /NH /FO csvr,rr r)r splitlinesrvreplaceru)rrlinerbitsrrHs r} get_procsz,is_win_secure_system_proc..get_procss+,NN$QR( D04 3@1AIIc2&@D@QT!W#DCH  AsA.z Secure SystemF)rKeyError)rHrs r}rFrFs=  {3?22 s  ''ctj}t|dr|jSt|dr#t j |j Sy)Nrrr)rrrrrandomchoicer)rs r}_get_eligible_cpursGAq)yy{ N #}}Q^^-.. rceZdZdZddifddifgZddifddddifd difd difd difd difd difddifddifg Zddifddifddifddifddifddifddifddifddddifddifddifddifddifd difd!difd"difd#difgZer$ed$difgz Zed%difgz Zed&difgz Zed'difgz Zer ed(difgz Ze r ed)difgz Ze red*e jfifgz Ze r ed+difgz Zer ed,difgz Zer ed-difgz Zer ed.difgz Zer ed/dd0d1ifgz ZgZer eddifgz Znede j&fifgz Ze red*e jd2fifgz Ze r,ered)e j*d3fifgz Zned)e j,fifgz Ze red+egfifgz Zd4ej2fifd5difd6difd7difd8difgZer(ed4ej6fifgz Zed4ej8fifgz ZeezezezZd9Zd?d:Zd;Z e!d<Z"e!d=Z#y>)@rCaA 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_percentrmemory_percentrrrT connectionsroneshotrjparentsrHr()rr cpu_times create_timer;rr, memory_infornet_connectionskindr nicenum_ctx_switches num_threads open_filesrr]rusernameuidsgidsterminalr1rrrrrrr2rgroupedF)r[irrrrr7rc||_yr)_proc)rrs r}rzprocess_namespace.__init__s  rc#Kt|}tj||D]M\}}}|r|jt |j |}t j|g|i|}||fOywz_Given a list of tuples yields a set of (fun, fun_name) tuples in random order. N)rrshuffler rrr6r)rlsr fun_namerrJr3s r}r zprocess_namespace.itersv"Xr$& " HdD  "$**h/C##C7$7$7C/ !  "sA4A6cf|jj|jjdy)z&Clear the cache of a Process instance.T) _ignore_nspN)r_initrHrs r}r zprocess_namespace.clear_cache&s! T:rc|D]>\}}}d|z}t||r|jjd|d}t|y)z}Given a TestCase instance and a list of tuples checks that the class defines the required test method names. test_z class should define a z methodN)rrrAttributeError)r$ test_classrrrl meth_namers r}test_class_coveragez%process_namespace.test_class_coverage*sb ! *NHa(*I:y1!++4478!}G-%S)) *rc |jDchc]}|d }}|jDchc]}|d }}ttjDchc] }|ddk7s |}}||z|z }|rt d|ycc}wcc}wcc}w)Nrrlzuncovered Process class names: )r ignoredr?rrr)r$rthisrklassleftouts r}testzprocess_namespace.test8s!gg&!&&!$-A1Q4--/?q1Q43;??'>U* >wkJK K  '-?s B B B "B N)T)$rrrr utilsrgettersr r'r(r+r RLIMIT_NOFILEr$r*r&r r)settersNORMAL_PRIORITY_CLASSrIOPRIO_CLASS_NONE IOPRIO_NORMALrrSIGTERMkillers CTRL_C_EVENTCTRL_BREAK_EVENTr rr r r_rrrrr}rCrCsR $'7R&@ AE B R+t,- B r2 B 2r B B r G B b" B B B R$ B R B0 R R$ B r2 R 2r B R#G& VR$%%VR$%%ZR())YB'((]B+,,Xr2&''X 4 46;<<^R,--YB'((YB'((]B+,,]BE(:;<<G VT2&''Vf::CDD '/G #g -C "; * *LLrrCceZdZdZddifddddifddddifddifd dd difd dd difd dd difd dddifdej fifddddifddifddifddddifdej fifddifddifddifddifgZere re rn eddd difgz Ze r eddifgz Ze r eddifgz Ze r eddifgz Zer eddifgz Zered difgz Zed!d"ifgz Zd#difd$ej"gfifd%difd&difgZeZed'Zej.Zy())rDzA 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() rrrvlogicalFT cpu_statsrpercpudisk_io_countersperdiskdisk_partitionsr  disk_usagerr net_if_addrs net_if_statsrpernicrrrusersrrrrrrwin_service_iterwin_service_get)algrrrcpu_times_percentc#Kt|}tj||D]5\}}}tt|}t j |g|i|}||f7ywr)rrrrrr6r)rrrrJr3s r}r zsystem_namespace.iterysc "Xr$& " HdD&(+C##C7$7$7C/ ! "sAAN)rrrr rrDrrr%rr4HAS_GETLOADAVGr/r.r,r rrrr r r rCrrrr}rDrDBs b" b9e,- b9d+, b" b8U+, b8T*+ R)T!23 B .   ~r* B0 R R B4 01   ~r* R B "b 2r"%G( W  R(D)9:; ;G\2r*+++R455^R,--&B/00'R011&"566 R )*,b1 B b"% G C " ",??rrDcbd}tttjjfd||S)zZDecorator which runs a test function and retries N times before actually failing. c@t|dtjy)Nz , retryingr})rNryr)rs r}rz retry_on_failure..logfuns z"4rN)rrvrr)rrrrr2)rrs r}r@r@s1 5 !6;;#8#89  rcfd}|S)z,Decorator to Ignore AccessDenied exceptions.cFtjfd}|S)Ncx |i|S#tj$rstjdwxYw)Nzraises AccessDenied)rrrr)rr$r3only_ifs r}r4z9skip_on_access_denied..decorator..wrappersI 9D+F++&& 9&"kk"788  9s .9r5r3r4rs` r}rz(skip_on_access_denied..decorators%   9  9rrrrs` r}r>r>s  rcfd}|S)z3Decorator to Ignore NotImplementedError exceptions.cFtjfd}|S)Nc |i|S#t$r*sjd}tj|wxYw)Nz2 was skipped because it raised NotImplementedError)rrrr)rr$rr3rs r}r4z;skip_on_not_implemented..decorator..wrappers] 'D+F++& '&"||&'++kk#&& 's 3>r5rs` r}rz*skip_on_not_implemented..decorators%   '  'rrrs` r}r?r?s" rctj5}|j|df|jdcdddS#1swYyxYw)z6Return an unused TCP port. Subject to race conditions.rrN)socketbind getsockname)hostris r}rTrTsC %D 4)!!$%%%s &AAc||tthvrd}tj||} tjdvr/|j tj tjd|j||tjk(r|jd|S#t$r|jwxYw)zBinds a generic socket.r r>ntcygwinrr) rrr rr setsockopt SOL_SOCKET SO_REUSEADDRr rlistenr2rf)familytypeaddrris r}rUrUs |7H"55 == &D  77* * OOF--v/B/BA F $ 6%% % KKN   s A7B""B=cftjsJtjj |rJ|t j t j |} |j||t jk(r|jd|S#t$r|jwxYw)zBind a UNIX socket.r) rr rrrr rr rrr2rf)rrris r}rVrVs <<<ww~~d#)T) # == .D $ 6%% % KKN K   s 5BB0ctj|t5}|j||jd|j }tj|t} |j ||j } |j \}}||k(r ||fcdddS|j6#t$r|jwxYw#1swYyxYw)z^Build a pair of TCP sockets connected to each other. Return a (server, client) tuple. rN) r rr rrconnectrbrfr)rrllrcaddras r}rWrWs v{ +r   ! ~~ MM&+ .  IIdOMMOE))+45=q6    GGI  s$A C)=C0CCCC(ctjsJdx}} t|tj}|j dtjtj tj}|j d|j|||fS#t$r&||j||jwxYw)zBuild a pair of UNIX sockets connected to each other through the same UNIX file name. Return a (server, client) tuple. Nrr) rr rVr r setblockingrrr2rf)rserverclients r}rXrXs <<<FV !$V-?-?@1v~~v/A/AB1t F     LLN   LLN s BB/C c#<Kg}dx}} |jttjtjttjtj ft r_|jttjtjttjtj ftr^trXt}t}t|\}}t|tj }|||fD]}|j|||D]}|j||fD]}|t|y#|D]}|j||fD]}|t|wxYww)z1Open as many socket families / types as possible.Nr")extendrUr rr SOCK_DGRAMrrr r0rLrXrVrMrfrI)socksfname1fname2s1s2s3rUrs r}rYrYsg EFV# (:(: ; (9(9 :   ? LLFOOV-?-?@FOOV->->?  -\F\F$V,FB!&v/@/@AB"b\  Q   A GGI f% #E E" # A GGI f% #E E" #s)FD0E): FF)!F FFcXt|tjsJ||tjk(rs|j dDcgc] }t |}}t|dk(sJ||D]}d|cxkr dkrJ|J|tj|y|tjk(r-t|tsJ|tj|y|tjk(rtj d|J|yt#d|cc}w)z[Check a net address validity. Supported families are IPv4, IPv6 and MAC addresses. rhrrz([a-fA-F0-9]{2}[:|\-]?){6}Nzunknown family )ryenumIntEnumr rrvrur ipaddress IPv4Addressrr IPv6AddressrAF_LINKrrr)rrroctsnums r}rRrR<s fdll +3V3 +  $ 301A004yA~#t#~ )C?s? (D (? (D (? )d# 6?? "$$*d*$d# 6>> !xx5t<HN$NH?6*5661sD'crd}d}d}d}d}||||||||||y)z*Check validity of a connection namedtuple.cJt|dk(}t|dvsJt||d|jk(sJ|j|d|jk(sJ|j|d|jk(sJ|j|d|jk(sJ|j|d|j k(sJ|j |d|j k(sJ|j |r$|d |jk(sJ|jyy) Nr>rrrrrirrrr)rr;rrladdrraddrr]rH)rkhas_pids r} check_ntuplez-check_connection_ntuple..check_ntupleSs d)q.4yF"-CI-"Aw$''!*477*!Aw$++%2t{{2%Aw$))#.TYY.#Aw$**$0djj0$Aw$**$0djj0$Aw$++%2t{{2% 7dhh& 0 0& rc|jttthvsJ|jt |jt j sJ||jtk(rUtj|j|j5} |j|jddfdddy|jtk(r/|jtj k(sJ|jyy#t$r(}|jtjk7rYd}~xd}~wwxYw#1swYyxYwr)rrrrryr1r2r rr r;rerrno EADDRNOTAVAILr]r CONN_NONE)rkrUrs r} check_familyz-check_connection_ntuple..check_family_s{{w'::GDKKG:$++t||4:d:4 ;;' ! t{{DII6 !FFDJJqM1-.   [[G #;;&"2"22 ?DKK ?2$yyE$7$778  s0D9 D D6D1,D91D66D99Ecttdt}|jtjtj |hvsJ|jt |jtjsJ||jtj k(r/|jtjk(sJ|jyy)NSOCK_SEQPACKET) rr objectrrr(ryr1r2r]rrB)rkrEs r} check_typez+check_connection_ntuple..check_typeps )968Dyy           99   $))T\\28D82 99)) );;&"2"22 ?DKK ?2 *rc|j|jfD]}|jtthvrt |t sJt||s>t |jtsJt|jd|jcxkrdksnJ|jt|j|j|jtk(st |trJt|y)Nri)r;r<rrrryrsrportrurRiprr)rkrs r} check_addrsz,check_connection_ntuple..check_addrs|sZZ, 9D{{w11!$.:T :.!$))S1B4 ?B1DII..9 9.!$''4;;7'!$,8d4j8, 9rc*t|jtsJ|jttDcgc]$}|j dst t|&}}|j|vsJ|j|jtthvrB|jtk(r/|jtjk7sJ|jy|jtjk(sJ|jycc}w)NCONN_) ryr]rr?rrrrrrrrrB)rkrvalidss r} check_statusz-check_connection_ntuple..check_statuss$++s+8T[[8+(+F  #$q||G7LGFA   {{f$1dkk1$ ;;7H- -$)){2J;;&"2"22 ?DKK ?2;;&"2"22 ?DKK ?2 s DDNr)rkr>rCrGrKrOs r}check_connection_ntuplerPPsH 1@" @ 9 @trcg}|D]Y}tr@|jtjk(r#trd|j vrt d|I|j|[|S)ztOur process may start with some open UNIX sockets which are not initialized by us, invalidating unit tests. z/syslogz skipping )r rr rrr<rrM)consnewrks r}rSrSs] C T[[FNN2djj0 $() 4  Jrc,tj|Sr) importlibreload)modules r}rZrZs   F ##rc8tjjtjj|d}tj j ||}tj j|}|jj||Sr) rrsplitextrRrUutilspec_from_file_locationmodule_from_specloader exec_module)rrspecmods r}r[r[sk 77  BGG,,T2 3A 6D >> 1 1$ =D .. ) )$ /CKKC Jrc<tj|tdy)zRaise a warning msg.rirNr)rs r}r\r\s MM#{q1rct|}|j}t|dk7s |dturyt |dd}t |tsyt d|DS)z-Check if object is an instance of namedtuple.rrF_fieldsNc3<K|]}t|tywr)ryr)rwns r}rxz is_namedtuple..s-az!S!-s)r __bases__rrsrryr )rtbrss r}r^r^sY QA A 1v{ad%'9d#A a  -1- --rc#Ktrdnd}d}t||z}tjj Dcgc]Y}t j j|j d|k(r(||j jvr |j [}}tj|}tj|| tj||t|ycc}w#t|wxYww)zCtx 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. pypyrgz.sorrN)rrLrrrrrrYlowerrrrrctypesCDLLrI)rrextdstrlibsrhs r}r]r]s f( -^^%113 ww'*c1cQVV\\^6K FF  mmD!S!  KK I     s+ADAC3 -DC8'D8 DDc#Kddlm}ddlm}d}t||z}t j j Dcgc]}|jjj|radtjj|jjvr(d|jjvr |j}}trt|srt j j Dcgc]G}dtjj|jjvr |jI}}tj|}tj ||d } tj"|}||Ttj$j&j(} |j*g| _| |j.} | dk(r|t1|y cc}wcc}w#|Ttj$j&j(} |j*g| _| |j.} | dk(r|t1|wxYww) zCtx 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.dllrrgwow64rjN)rlrrrsrLrrrrrkr~rrRrrrrrWinDLLwindllkernel32 FreeLibraryHMODULEargtypes_handlerI) rrrrsrnrorrprhcfilerxrs r}r]r]s $# -^^%113 vv||~&&s+BGG,,QVV4::<<qvv||~- FF    )557RWW--aff5;;==D mmD!S! MM#&EI $mm44@@ (0(8(8'9 $!%--0!8"*$  ? " $mm44@@ (0(8(8'9 $!%--0!8"*$  s:AI*BG:.I*A G?/I*?HA,I*A#I''I*ctdy)NTr)r8rrr}cleanup_test_procsr~s D!rc,tj|Sr)ryexit)rrls r}r`r`(s#rr)F)TFr)z 127.0.0.1)rr )r rrrlr1r@r6rErUr3rrqrrrcrzrrr rrwryrrPrr+ tracebackr=rrrrrrrrrrr r r r r psutil._commonrrrrrpsutil._psposixr__all__builtin_module_namesrrr`r r3r5maxsizer#rrkr4RISCV64r~r1r2rrr"rrrrrdecoderrASCII_FSrrr.r__file__rgetrrrrr$r&rr(r)r0HAS_NET_IO_COUNTERSr*r'r+r,r.r/ HAS_THREADSgetuid SKIP_SYSCONSr]rr-r2rrrN format_excrr%rrrrpdevnullrrfr?rrr!rFrr*rFrSrGrrmodulesrr'r<r8r9r;r:r6rrr7r8rNrMrrrPrrQrOrIrrrHrJrKrLrrBrrrArErFrrCrDr@r>r?rTrUrVrWrXrYrRrPrSrZr[r\r^r]r~)rs0r}rs        & "&((# ZS---!RZZ/O>RZZ3O  RZZ ''2::5 ;;  (    " " $(< < (    )  (  ?X-L ?g-LLL  #' !OJaNA 77fyryy{m1-Myryy{m1-M%,,V5FG $3 $ $ & , , .2G G 77  GGLL*D$7 jjnn"'',,x;  ww126>>>:fnni0 . V^^X . &..-8 .Yf&786>>95v~~}= V^^X . f&786>2"6+ABfnni0  2)"))+"2 %H$/Ev/E/E/G*HK6:.J48I3JL)X)] N rzz4  !$F q||I/FGFA &)VX . U#U#r,  > F'CKK(!!(` &&R##L&R&0 &~~NQ/h-J!066r""        .1       v{{445        D  4 &VBX&&VBND!56g%^g%7g%Tk\$ FLFLRC@C@L(  &:%[t"!' 2 2 .. ##<7(HV &$2 . .**f"" FMM&..">? }? FRFOO%I%%e-AY-A-A-CDEK FOO%I%%e-AY-A-A-CDELpsB[[$"\ ]# ][[9\\9]]