f dZddlmZddlZddlZddlZddlZddlZddlZddl Z ddl Z ddl Z ddl Z ddl Z ddlZddlZddlZddlZddlZddlZddlZddlZddlZddlZddlZddlZddlZddlZddlmZddlmZddlmZddlZddlm Z ddlm!Z!dd lm"Z"dd lm#Z#dd lm$Z$dd lm%Z%dd lm&Z&ddlm'Z'ddl(m)Z)ddl(m*Z*ddl(m+Z+ddl(m,Z,ddl(m-Z-ddl.m/Z/ddl.m0Z0ddl.m1Z1ddl.m2Z2ddl.m3Z3ddl.m4Z4ddl.m5Z5ddl.m6Z6 ddlm7Z7e/rddl;Z;ndZ;e%rddld ej~vZ@d!e jvZBd"e jvxsd#e jvZCeBxseCZDd$e jvZEejd%kDZGe+d&ZHe"reHd'kDZIeHd(k\ZJnd)ZId)ZJd*ZKd+ZLd,ZMd-ZNeDreKd.zZKeNd.zZNeLd/zZLeMd.zZMe jd0k(rd1e jzZQnd2e jzZQe4d3ZRe/rd4jd5d6ZTnd7ZTejjd8vZWe jje jje jje\d9d9Z]e jjd:e jje]d;Z_e jje jje\Z`e%xre& Zaebejd<Zdebed=Zeebed>Zfebejd?Zgebejd@ZhebejdAZiebedBZjebejdCZkebejdDZlebejdEZmebedFZn enxreoejZqebedGZsebedHZtebejdIZue"xse xre jdk7ZwdJZxex\ZyZze{e jdKZ}eje}jeeDcgc]}|jdLsee| c}ZeedMeZeZeZGdNdOejZdPZeddQZedRZdSZedTZedUZej"eNfdVZddWZdXZdYZGdZd[Zeej.deNd\]d^Zee1efdeNd\]dd_ZeedeNd\]d`ZdaZdbZej<dcZddZddeZddfZGdgdhejFZee_GdidjeZejJe@dkGdldmeZdnZdoZdpZGdqdrZGdsdtZduZeKfdvZddwZddxZddyZeedfdzZej<fd{Zdd|Zd}Zej<d~ZdZdZdZdZdZdZdZe%rej<ddZnej<ddZejdZe%r ej"ej"dyy#e8$rAejr5ejtdddl7Z7dddn #1swYnxYwYwxYw#er$rd)ZqYwxYwcc}w)zTest 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)u)unicode)which)mockignore)wait_pid)HAPPVEYORDEVNULLGLOBAL_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_TEMPERATURES MACOS_11PLUS MACOS_12PLUSCOVERAGEpyrun terminate reap_childrenspawn_testproc spawn_zombiespawn_children_pair ThreadTaskunittestskip_on_access_deniedskip_on_not_implementedretry_on_failureTestMemoryLeakPsutilTestCaseprocess_namespacesystem_namespace print_sysinfois_win_secure_system_procchdir safe_rmpath create_py_exe create_c_exe get_testfn get_winverkernel_version call_until wait_for_pid wait_for_filecheck_net_addressfilter_proc_connections get_free_port bind_socketbind_unix_sockettcp_socketpairunix_socketpaircreate_sockets reload_moduleimport_module_by_pathwarncopyload_shared_lib is_namedtuple__pypy__rGITHUB_ACTIONS CIBUILDWHEEL COVERAGE_RUNlcBtjd}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 7/usr/lib/python3/dist-packages/psutil/tests/__init__.py macos_versionrs""$Q'KC[..s3BQ789G(!--?   )#.#  C!2!23!7!;<= N)rn) rFrniPijavaz $psutil-%s-z @psutil-%s-u-ƒőősfutf8surrogateescapeufÀ€)asciizus-asciiz..PSUTIL_SCRIPTS_DIRscripts cpu_affinitycpu_freq getloadavgenvironionice memory_mapsnet_io_counterscpu_num io_countersrlimitsensors_battery sensors_fanssensors_temperaturesthreadscd}tjj}ttdd}t r,tj dk\r|tj|d<||fStrtj|fStr|tjxs|tjjtjxsO|tdtj ddzxs(|tjj}|s t!d||fStjjtj}tjj#|sJ|||fS)Nc tj|dgtjtj|S#tj$rYywxYw)Nz-V)stdoutstderr)r{ check_callPIPECalledProcessError)exes rattemptz_get_py_exe..attemptsI   ! !d JOOJOO  J,,  s6:AA_base_executable)r__PYVENV_LAUNCHER__z python%s.%srmz"can't find python exe real abspath)osrcopygetattrr}r version_infor~rhr pathrealpathrpsutilProcessr ValueErrorexists)rrsbasers r _get_py_exers9 **// C 3*D 1D3##v-$2B&)^^ !"Sy ~~s""  CNN # /rww''78 /u]S-=-=bq-AABC /v~~'++-. AB BCxggs~~.ww~~c"'C'"Cxrzr+STATUS_AF_UNIXcFeZdZdZfdZdZdZdZdZdZ dZ xZ S) rEz6A thread task which does nothing expect staying alive.ctjd|_d|_t j |_y)NFMbP?)r__init___running _interval threadingEvent_flag)self __class__s rrzThreadTask.__init__(s-  __& rcd|jj}d||jt|fzS)Nz<%s running=%s at %#x>)r__name__rid)rnames r__repr__zThreadTask.__repr__.s+~~&&'44*IIIrc&|j|SN)startrs r __enter__zThreadTask.__enter__2s  rc$|jyr)stop)rargskwargss r__exit__zThreadTask.__exit__6s  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)rrrThreadrrwaitrs rrzThreadTask.start9s9 ==./ /t$ rcd|_|jj|jr-tj|j |jr,yy)NT)rrsettimesleeprrs rrunzThreadTask.runBs7  mm JJt~~ &mmrc`|js tdd|_|jy)z8Stop thread execution and and waits until it is stopped.zalready stoppedFN)rrjoinrs rrzThreadTask.stopHs%}}./ /  r) r __module__ __qualname____doc__rrrrrrr __classcell__)rs@rrErE%s*@' J' rrEcBtjfd}|S)NcF |i|S#t$r twxYwr) ExceptionrA)rrfuns rwrapperz&_reap_children_on_err..wrapperVs/ '' '  O  s   functoolswraps)rrs` r_reap_children_on_errrUs%__S Nrc |jdt|jdt|jdtj|jdtt rd}|jd||t tj} t|dd |zzd z}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(). stdinrcwdrs creationflagsdirz import time;zopen(r'%s', 'w').close();z&[time.sleep(0.1) for x in range(100)];rpTdeleteempty) setdefaultr rgetcwdr&rrTrQr%r{Popen_subprocesses_startedaddrYrXpid)cmdkwdsCREATE_NO_WINDOWtestfnpylinesprocs rrBrBas) OOGW%OOHg&OOE299;'OOE>*% )9: { ,  -67:;  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(). Nrzl 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(60);" p = subprocess.Popen([r'z.', '-c', s]) p.wait() r)rTFr)rTrrtextwrapdedentrbasenamer%rr?rrrryrY _pids_startedrrQ)tfilerssubpchildgrandchild_pid grandchilds rrDrDs E BIIK (F OO77##F+Z 9 : 3KD%(KD%txx(]6$eLM.)^^N3 z"F      F     s CDDctjsJt}tjd|z}d}t |} |j tt|\}}|j\}} tj|jgggtt|jd}tj|tj |}t#|j$d||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 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) Nzret == psutil.STATUS_ZOMBIE)rr rTrrr^ settimeoutr!r?acceptselectfilenoryrecvrrrrWstatuscloserQ) unix_filesrcrsockparentconn_zpidzombies rrCrCs5 <<< I //  C E I &D 'c ++-a  MM4;;=/2r> Btyy'D   d #^^D)F v}}&C DF# JJL I      JJL I     s%6E:BD<E<EE+E<c `|jdd|jddt} t|d5}|j|dddt t j gfi|}t|j||fS#1swY>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) rrTopenwriterBr%rrXrrrQ)rrsrcfilefrs rr?r?s  OOHd#OOHd#lG '3  1 GGCL z1662;d;TXXg    G s" BB ||q6""  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|z#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. cZt|tjrts|j n|j |t rJt|tjr/ t j|jj |Syy#t j$rYywxYwr) r%r{rrrrrrr NoSuchProcess)procr"s rrzterminate..waits dJ,, -c IIK IIg  z$ (8(89 ~~dhh/44W==:7''  s$-BB*)B*ctrtrtj}tr2|tjk7r|j tj |j |yr)r rhsignalSIGKILLr send_signalSIGCONT)r2sigs rsendsigzterminate..sendsig&s@ ^..C SFNN*   V^^ , rc |||S#t$r>}tr|jdk(rn|jtjk7rYd}~Kd}~wwxYw)N)OSErrorrwinerrorerrnoESRCH)r2r"errr9r8rs rterm_subprocess_procz'terminate..term_subprocess_proc0sX  D#  D'""  3<<1,ekk)  s  A4AAc\ |||S#tj$rYwxYwr)rr1)r2r"r9r8rs rterm_psutil_procz#terminate..term_psutil_proc:s;  D# D'""##   s ++c tj|}||S#tj$rtrt ||cYSYywxYwr)rrr1r r)rr"r2rCs rterm_pidzterminate..term_pidAsQ 3>>#&D $D'2 2 ## .W-- .s!%A  A c|jr|jj|jr|jj|jr|jjyyr)rrrr)r2s r flush_popenzterminate..flush_popenKsL ;; KK    ;; KK    :: JJ    rz wrong type %r) r%ryr{rrr pid_existsr TypeError) proc_or_pidr8 wait_timeoutrArErGr-rr9rCrs ` @@@rr@r@s ##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.)) Oa/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&|rc|D]}t |dtj|t\}}|D]+}td|zt |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)rKr!z0couldn't terminate process %r; attempting kill())r8) rrchildrenrpopr@r wait_procsr!rdr4r5)rNrOrrr-ralives rrArAes~~((9(=H $((*$ !#   ,A ad + ,$$X~F5 -A CaG H aV^^ , - rc|ts tdd}tjd}|D]}|j s|dk(r||z }n|st d|zd}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 POSIXrmrlzcan't parse %rrr) r NotImplementedErrorrunameisdigitrrzrylen)rrWcminormicronumsmajors rrVrVs !+.. A HHJqME  99;!s( FA   )E122 E E 773.wrappersvC ///  ~~C{{. C(JJL sA&/A!!A&)rr decorator)rrrs`` r__call__zretry.__call__s-     $!r) rrrrrrrrrrxrrrjrjs*  &rrjr)rlror"rnc|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)rpidsr1r)rs rrXrXs/ &++-""3'' NN3rct|d5}|j}ddd|ssJ|r t|S#1swYxYw)z8Wait for a file to be written on disk with some content.rbN)rreadrQ)fnamerrrdatas rrYrYsJ eT avvx  t E K s ;Ac.|}t|sJ|S)zVKeep calling function for timeout secs and exit if eval() expression is True. )eval)rexprrets rrWrWs %C :: 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.cHtjtz}tj|kr |S#t$rYn.t$r#}|}t dt |zYd}~nd}~wwxYwtj dtj|krpi)Nz ignoring %sg{Gz?)rr!r WindowsErrorrdr&r)rrqrr@s r retry_funzsafe_rmpath..retry_fun%s ))+.iikG# 1u  %  1]c#h/00 1 JJt iikG#s= A3A3A..A3N) rstatS_ISDIRst_moderpartialshutilrmtreeremover r)rrstrs rrQrQ"sv"  WWT] << ###FMM48C##BIIt4C  E cN    sB BB B%$B%cN tj|y#t$rYywxYw)z.Convenience function for creating a directory.N)rmkdirrrs r safe_mkdirrDs$      s  $$c#Ktj} tj|dtj|y#tj|wxYww)z@Context manager which temporarily changes the current directory.N)rrrP)dirnamecurdirs rrPrPLs?YY[F   sA!AA!AA!cRtjj|rJ|tjt |t jt|trFtj|}tj||jtjz|S)z6Create a Python executable file in the given location.)rrratexitregisterrQrcopyfiler%r rchmodrS_IEXEC)rrs rrRrRWsiww~~d#)T) # OOK& OOJ% WWT] rzzDLL01 KrcBtjj|rJ|tdst j d|t jd}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.csuffixrz-o)rrrrrFSkipTestrrr%r&rrrQrrTrr{rr)rc_coders rrSrSbsww~~d#)T) # < 677 ~"&#&..& OOK& j%s +q uaffdD9:AFF K  AFFs&C;#D;DDc 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) tempfilemktempr)rrrrrrrQ)rrrrs rrTrT|sU mFLww~~d#77##D)D OOK .K rceZdZdZeej dsej jZe sdZ e jdZ yy)TestCasec|jj}|jdsd|z}|d|jjd|jS)Nzpsutil.z psutil.tests.rl)rr startswithr_testMethodName)rfqmods r__str__zTestCase.__str__sL)) *#e+E  NN # #    rassertRaisesRegexcyrryrs rrunTestzTestCase.runTests rc/Kdywrry)rrkws rsubTestzTestCase.subTests  sN)rrrrrbrFrassertRaisesRegexprrr contextlibcontextmanagerrryrrrrsU  8$$&9 :$--@@    " "  #  rrcHeZdZdZd dZdZdZdZdZdZ d Z d Z d Z y) rKzUTest class providing auto-cleanup wrappers on top of process test utilities. NcLt||}|jt||S)N)rr)rT addCleanuprQ)rrrrs rrTzPsutilTestCase.get_testfns!&c2  U+ rcHt|i|}|jt||Sr)rBrr@)rrrrs rrBzPsutilTestCase.spawn_testprocs$--  5) rc|t\}}|jt||jt|||fSr)rDrr@)rchild1child2s rrDz"PsutilTestCase.spawn_children_pairs5,.  6*  6*rc|t\}}|jt||jt|||fSr)rCrr@)rrrs rrCzPsutilTestCase.spawn_zombies4%  6*  6*rczt|i|\}}|jt||jt||Sr)r?rrQr@)rrrrrs rr?zPsutilTestCase.pyruns6--w  W-  5) rc(|j|tj|j|j|j|j|j |j |j r|j|j dt|tjrN|j|j|j|j|j|jdt|t|y)NrTr)assertIsInstancerError assertEqualrr_nameassertNotEqualr% ZombieProcessppid_ppidassertGreaterEqualr&repr)rr2rvs r_check_proc_exczPsutilTestCase._check_proc_excs c6<<0 $((+ 4::. 88   " - c6// 0   SXXtzz 2xx#''!4 C S rcx|jtj5} tj| ddd|j jj||j |jjdtj|rJ||j|tj|j|tjDcgc]}|jc}y#tj$r t dwxYw#1swYxYwcc}w)Nz&wasn't supposed to raise ZombieProcess) assertRaisesrr1rrAssertionErrorrrlrrrH assertNotInr{ process_iter)rrcmxs r assertPidGonezPsutilTestCase.assertPidGones   v33 4 O Os# O ))3/ **D1$$S).3.) fkkm, f.A.A.CDquuDE '' O$%MNN O O OEs"D+D .D7 D((D++D4c|j|jt|}|j|jdD]3\}}|j ||5 |}d|d|}t ||jdy#tj$rtj$r}|j||Yd}~nd}~wwxYw ddd#1swYxYw)NT clear_cacher2rzProcess.z!() didn't raise NSP and returned rr!) rrrLiterallrrrrr1rr)rr2nsrrrmsgrvs rassertProcessGonez PsutilTestCase.assertProcessGones 488$ t $T: .IC4d3 . .%CC)-- . ! ++++4((s3343 . . .s6CB &C &C3C C CCC% cd tj|j}|j||ts*t s$|jt |t ||j|jtj|jsJtj|jsJ|j|j|jtj|j|jtjDcgc]}|jc}it_|j|jtjDcgc]}|jc}t!|}|j#|j$dD])\}}|j'||5 |ddd+t.r|j1tj(5}|j3ddd|j-|j4|j1tj(5}|j7ddd|j-||j4|j1tj(5}|j9ddd|j-||j4|j;|j=|j?|jA|jsJtj|jsJ|j|jtj|j|jtjDcgc]}|jc}it_|j|jtjDcgc]}|jc}ycc}wcc}w#tj(tj*f$r}|j-||Yd}~d}~wwxYw#1swYxYw#1swYaxYw#1swYxYw#1swYxYwcc}wcc}w)NTrr)!rrrrr r hashr STATUS_ZOMBIE is_runningrHas_dictassertInr{r_pmaprLrrrr AccessDeniedrrrcmdlinerlrrsuspendresumer@kill) rr2clonerrrrrvrs rassertProcessZombiez"PsutilTestCase.assertProcessZombiesMtxx( u%6   T$Ze 5 (<(<=     ***  dhh . dhh0C0C0E F1 FG  dhh0C0C0E F1 FG t $T: 4IC4d3 44E 4 4 4 ""6#7#78 B    r|| 4""6#7#78 B    r|| 4""6#7#78 #B  " #  r|| 4         *** dhh . dhh0C0C0E F1 FG  dhh0C0C0E F1 FGA!G F,,f.A.AB4((s334 4 4     # #!G Fsl-P' >P, Q4P1R%R9R;R( R- 1#Q1Q,&Q4,Q11Q44Q> R RR%rTN) rrrrrTrBrDrCr?rrrrryrrrKrKs6      F&1HrrKzunreliable on PYPYceZdZdZdZdZdZerdndZdZ e jZ e ejdZedZed Zd Zd Zd Zd ZdZdZdZ ddZdZy)rJaTest 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) rnrrT PSUTIL_DEBUGc.tjdy)NF)r _set_debugclss r setUpClasszTestMemoryLeak.setUpClassjs% rcBtj|jyr)rr_psutil_debug_origrs r tearDownClasszTestMemoryLeak.tearDownClassns#001rcd|jj}t|d|jS)Nuss) _thisprocmemory_full_inforrss)rmems r_get_memzTestMemoryLeak._get_memrs)nn--/sE377++rcvtr|jjS|jjSr)r rnum_fds num_handlesrs r _get_num_fdszTestMemoryLeak._get_num_fdsxs+ >>))+ +>>--/ /rcV|jrt|dtjyy)Nyellow)colorfile)verboserr}r)rrs r_logzTestMemoryLeak._log~s << 8#** = rc |j}|j||j}||z }|dkr|jd|z|dkDr/trdnd}|dkDr|dz }|d|d|}|j|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. rzHnegative diff %r (gc probably collected a resource from a previous test)fdhandlerUrz unclosed z after calling N)rcallfailr )rrbeforeafterdifftype_rs r _check_fdszTestMemoryLeak._check_fdss ""$ #!!#v~ !8))1378  !8!DxEax 7;UCHC))C. rc.tjd|j}t|D]}|j |}~~tjd|j}|j tj g||z }|S)zGet 2 distinct memory samples, before and after having called fun repeatedly, and return the memory difference. rU) generation)gccollectrrr rgarbage)rrtimesmem1rrmem2rs r _call_ntimeszTestMemoryLeak._call_ntimess{ a }}u A))C.C3  a }} R(d{ rc g}d}|}td|dzD]}|j||} d|dt| dt| |z d|} |j| | |kxs| |k} | r|dkDr|j | y|dk(r t |j | ||z }| }|j dj|)NrrUzRun #z : extra-mem=z , per-call=z, calls=z. )rrrappendrprintr r) rrrrm tolerancemessagesprev_memincreaseidxrrsuccesss r _check_memzTestMemoryLeak._check_memsGaK( C##C/CC C%K( C OOC Y&9#/G7IIcN!8G #!' (ii (+,,rc|Srry)rrs rr 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.NrUztimes must be >= 1rzwarmup_times must be >= 0zretries must be >= 0ztolerance must be >= 0)rrmr) r warmup_timesrmrrrr&rrr$)rrrr'rmrr@s rexecutezTestMemoryLeak.executes* (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*jyr)r)rvrrsrr z*TestMemoryLeak.execute_w_exc..calls   c3 'rN)r()rrvrrr s``` r execute_w_exczTestMemoryLeak.execute_w_excs  (  T$V$r)NNNN)rrrrrr'rr,rmrrrrboolrgetenvr classmethodrrrrrrrr$r r(r+ryrrrJrJAs> ELIbAGG Iibii78!!22, 0 >!* -8KOP,%rrJc  ddl}ddl}ddl}ddl}ddl} ddl} ddl}|j}tjrtdrtd|d<ntjrdtjdz|d<ntj rjddj#t%t&tj(z|d<t+tdrQ|dxxd tj,zz cc<n.tj.dtj0|d<d j#t3tj4tj6gz|d <tj8rtj:d |d <d j#tj<tj>tj@g|d <tC|dd|d<||dxxd|jDzz cc<tj8r\tdr.tddg}t'|jGdd|d<nd|d<tjHd} | r| |d<tKjL|d<|jO} | dd | d|d<|jjQtjRjUd|d<|jjWjUd|d<|jY|d<tZj\j_d|d<t[j`|d<tb|d <tjd|d!<t[jf|d"<tjh|d#<d$tktjlD cgc]} | tjhz d%zc} z|d&<tjn} tq| jrd'tu| jvd(tu| jx|d)<tjz} tq| jrd'tu| jvd(tu| jx|d*<t}tj~|d+<tjj}|jd,d|j||d-<td.tJj/|jD])\}}t|d0zd1d|tJj/+td.tJj/tJjjt rt[j.d2n td3rt[j.d4td.tJj/tJjjy#t $rd}YwxYw#t $rd}Y+wxYwcc} w)5Nr lsb_releasezlsb_release -d -sOSz Darwin %szWindows  win32_editionz, archrmkernelpython __version__z not installedpipz (wheel=%s)rz --versionr#rUglibcz fs-encodinglangz%Y-%m-%d %H:%M:%Sz boot-timeruser~homerpyexehostnamePIDcpusz%.1f%%, %.1f%%, %.1f%%dloadavgz%, used=z, total=memoryswapr{rr2zF======================================================================r:17tasklistpszps aux)I collectionsdatetimegetpasslocalepprintr8 ImportErrorwheel OrderedDictrrrr.OSXrurvrrrxr& win32_verrbr3systemrlist architecturemachiner rWpython_implementationpython_versionpython_compilerrr7rzlibc_verr}getfilesystemencoding getlocale fromtimestamp boot_timestrftimenowgetuserrr expanduserrr%nodegetpid cpu_countrwrvirtual_memoryrypercentrusedtotal swap_memoryrYr{rrrPpformatrritemsrflush)rKrLrMrNrOr8rQinfooutrr:rrrEpinfokvs rrNrNs  " " $D||m,+,T  8#3#3#5a#88T #((3sH4F4F4H+I"JJT 8_ - J$!7!7!99 9J ( 183C3C3EFT 99 X " " $%)9)9);(<<DL||!)!,XYY&&(!  " DN #}o>DK  U }u'8'888 || <e[)*Cc(...q1DK)DK     " DM335D    D#AwQ0DL ))77h"# $$((*334GHDL??$DL77%%c*DL))+DKDM}}D))+DK##%DL. V5F5F5HIq6##%%+IJDO    !C CKKCHHCIIDN    D DLLDIIDJJDL v{{}%DL NN  $ $ &E IImT">>%(DL ($ :1 AGQ'cjj9: ($JJ * t ( ($JJ  zJs(X*X<!Y * X98X9< Y  Y cTtd} ||dk(S#t$rYywxYw)Nci}td}|jddD]H}|jdDcgc]}|jdd}}|dt |d}}|||<J|Scc}w)Nztasklist.exe /NH /FO csvrU,"rTr)r. splitlinesrzreplacery)rrqlinerbitsrrs r get_procsz,is_win_secure_system_proc..get_procsSs+,NN$QR( D04 3@1AIIc2&@D@QT!W#DCH  AsA.z Secure SystemF)rKeyError)rr}s rrOrOQs=  {3?22 s  ''ctj}t|dr|jSt|dr#t j |j Sy)Nrrr)rrrbrrandomchoicer)r-s r_get_eligible_cpurcsGAq)yy{ N #}}Q^^-.. rceZdZdZddifddifgZddifddddifd difd difd difd difd difddifddifg Zddifddddifddifddifddifddifddifddifddifddifddifddifddifd 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>)@rLaA 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_percentrymemory_percentrrOrNTrmemory_info_exoneshotrparentsrr)rr connectionskindr cpu_times create_timerrr memory_infornicenum_ctx_switches num_threads open_filesrrrusernameuidsgidsterminalrrrrrrrrrgroupedF)r irr6rrr@rc||_yr)_proc)rr2s rrzprocess_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)rVrshufflerrrrr)rlsrfun_namerrrs rrzprocess_namespace.itersv"Xr$& " HdD  "$**h/C##C7$7$7C/ !  "sA4A6cf|jj|jjdy)z&Clear the cache of a Process instance.T) _ignore_nspN)r_initrrs rrzprocess_namespace.clear_caches! 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)rbrrAttributeError)r test_classrrr meth_namers rtest_class_coveragez%process_namespace.test_class_coveragesS ! *NHa(*I:y1((11%S)) *rc\t|jDcgc]}|d c}}t|jDcgc]}|d c}}tttj Dcgc] }|ddk7s |c}}||z|z }|rt d|zycc}wcc}wcc}w)Nrrz!uncovered Process class names: %r)rrignoredrrrr)rrthisrklassleftouts rtestzprocess_namespace.tests#''*QAaD*+S[[1qt12FNN 3C1qts{QCD'>U* @7JK K  +1Cs B B$- B);B)N)T)$rrrrutilsrgettersr r3r4r7r RLIMIT_NOFILEr0r6r2rr5settersNORMAL_PRIORITY_CLASSrIOPRIO_CLASS_NONE IOPRIO_NORMALrr4SIGTERMkillers CTRL_C_EVENTCTRL_BREAK_EVENTrrrrr.rrryrrrLrLlsR $'7R&@ AE B R+t,- r2 2r" B 2r B B r G B VUO, b" B B B R$ B R 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 "; * *LLrrLceZdZdZddifddddifddddifddifd dd difd dd difd dd difd dddifdej fifddddifddifddifddddifdej fifddifddifddifddifgZer eddd difgz Ze r eddifgz Ze r eddifgz Ze r eddifgz Ze r eddifgz Ze red difgz Zed!d"ifgz Zd#difd$ejgfifd%difd&difgZeZed'Zej*Zy())rMzA 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`ryrglogicalFT cpu_statsrpercpudisk_io_countersperdiskdisk_partitionsr disk_usagenet_connectionsr net_if_addrs net_if_statsrpernicrHr{rlusersrhrrrrrwin_service_iterwin_service_get)algrrQrcpu_times_percentc#Kt|}tj||D]5\}}}tt|}t j |g|i|}||f7ywr)rVrrrrrr)rrrrrs rrzsystem_namespace.iter)sc "Xr$& " HdD&(+C##C7$7$7C/ ! "sAAN)rrrrrrrfrr1HAS_GETLOADAVGr;r:r8rrrrr staticmethodrrLrryrrrMrMs b" b9e,- b9d+, b" b8U+, b8T*+ R)T!23 B .   ~r* B0 R R B4 01   ~r* R B "b 2r"%G(Zh%5677\2r*+++R455^R,--&B/00'R011&"566 R )*,b1 B b"% G C " ",??rrMcLtj|sJ|d|_|S)zA decorator to mark a TestCase class. When running parallel tests, class' unit tests will be run serially (1 process). T)inspectisclass _serialrun)rs r serialrunr8s( ??5 !(5( !E Lrc.d}ttd||S)zZDecorator which runs a test function and retries N times before actually failing. c@td|ztjy)Nz %r, retryingrF)rr}r)rvs rroz retry_on_failure..logfunGs ns"4rN)rlr"rmro)rjr)rmros rrIrIBs 5  $ rcfd}|S)z,Decorator to Ignore AccessDenied exceptions.cFtjfd}|S)Ncx |i|S#tj$rstjdwxYw)Nzraises AccessDenied)rrrFr)rrronly_ifs rrz9skip_on_access_denied..decorator..wrapperSsK ?D+F++&& ?&"''(=>>  ?s .9rrrrs` rrwz(skip_on_access_denied..decoratorRs%   ?  ?rryrrws` rrGrGOs  rcfd}|S)z3Decorator to Ignore NotImplementedError exceptions.cFtjfd}|S)Nc |i|S#t$r*sdjz}tj|wxYw)Nz4%r was skipped because it raised NotImplementedError)rVrrFr)rrrrrs rrz;skip_on_not_implemented..decorator..wrapperfsZ -D+F++& -&"Jll#'',, -s 3>rrs` rrwz*skip_on_not_implemented..decoratores%   -  -rryrs` rrHrHbs" rctjtj5}|j|df|j dcdddS#1swYyxYw)z6Return an unused TCP port. Subject to race conditions.rrUN)rclosingsocketbind getsockname)hostrs rr\r\sN   FMMO ,% 4)!!$%%%s &AA!c||ttfvrd}tj||} tjdvr/|j tj tjd|j||tjk(r|jd|S#t$r|jwxYw)zBinds a generic socket.rTr)ntcygwinrUr) rrrrr setsockopt SOL_SOCKET SO_REUSEADDRrrlistenrr)familytypeaddrrs rr]r]s |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 rrrrrrrrrr)rrrs rr^r^s <<<ww~~d#)T) # == .D $ 6%% % KKN K   s 5BB0ctjtj|t5}|j ||j d|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) rrrrrrrconnectr rr<)rrllrZcaddras rr_r_s   FMM&+> ?2   ! ~~ MM&+ .  IIdOMMOE))+45=q6    GGI  s$A C2<=CCC//C22C;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 r^rr setblockingrrrr)rserverclients rr`r`s <<<FV !$V-?-?@1v~~v/A/AB1t F     LLN   LLN s BB/C c#xKg}dx}} |jttjtj|jttjtj t rn|jttjtj|jttjtj tr^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)rr]rrr SOCK_DGRAMrrr HAS_CONNECTIONS_UNIXrTr`r^rrQ)socksfname1fname2s1s2s3rrs rrarase EFV# [1C1CDE [1B1BCD ? LLV__f6H6HI J LLV__f6G6GH I )\F\F$V,FB!&v/@/@AB"b\  Q   A GGI f% #E E" # A GGI f% #E E" #s)F:EF F:9F:!F7)F77F:cddl}tr-tr'ts!t |tj sJ||t jk(r|jdDcgc] }t|}}t|dk(sJ||D]}d|cxkr dkrJ|J|ts t|}|j|y|t jk(r:t |tsJ|ts t|}|j|y|t j"k(rt%j&d|J|yt)d|zcc}w)z[Check a net address validity. Supported families are IPv4, IPv6 and MAC addresses. rNrlrz([a-fA-F0-9]{2}[:|\-]?){6}zunknown family %r) ipaddressenumrr$r%IntEnumrrrzryrYr IPv4Addressrr& IPv6AddressrAF_LINKrcmatchr)rrrroctsnums rrZrZs) D&$,,/77/  $ 301A004yA~#t#~ )C?s? (D (? (D (? )4=Dd# 6?? "$$*d*$4=Dd# 6>> !xx5t<HN$NH,v5661sEcrd}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)r;rrrUrmrrrr;)rYr rrladdrraddrrr)rhas_pids r check_ntuplez-check_connection_ntuple..check_ntuple s d)q.4yF"-CI-"Aw$''!*477*!Aw$++%2t{{2%Aw$))#.TYY.#Aw$**$0djj0$Aw$**$0djj0$Aw$++%2t{{2% 7dhh& 0 0& rc|jtttfvsJ|jt+t |jtj s(J|t |jtsJ||jtk(rjtj|j|j}tj|5 |j|jddfdddy|jtk(r/|j"t$j&k(sJ|j"yy#tj$r(}|jtj k7rYd}~d}~wwxYw#1swYyxYw)Nr)rrrrrr%rryrrrrrrerrorr> EADDRNOTAVAILrr CONN_NONE)rrr@s r check_familyz-check_connection_ntuple..check_familys0{{w'::GDKKG:  dkk4<<8 >$ >8dkk3/ 5 5/ ;;' !  dkk4995A##A& FFDJJqM1-.   [[G #;;&"2"22 ?DKK ?2$||yyE$7$778  s0E? EE<E72E?7E<<E??Fcttdt}|jtjtj |fvsJ|jt +t|jt js(J|t|jtsJ||jtj k(r/|jtjk(sJ|jyy)NSOCK_SEQPACKET) rrobjectrrrrr%rryrrr)rrs r check_typez+check_connection_ntuple..check_type.s )968Dyy           99    dii6 < <6dii- 3t 3- 99)) );;&"2"22 ?DKK ?2 *rc|j|jfD]}|jttfvrt |t sJt||s>t |jtsJt|jd|jcxkrdksnJ|jt|j|j|jtk(st |trJt|y)Nri)rrrrrr%rwrportryrZiprr&)rrs 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|jttfvrB|jtk(r/|jtjk7sJ|jy|jtjk(sJ|jycc}w)NCONN_) r%rr&rrrrrrrrrr)rrvalidss r check_statusz-check_connection_ntuple..check_statusJs$++s+8T[[8+(+F  #$q||G7LGFA   {{f$1dkk1$ ;;7H- -$)){2J;;&"2"22 ?DKK ?2;;&"2"22 ?DKK ?2 s DDNry)rrrrrr"s rcheck_connection_ntupler# sH 1@* @ 9 @trcg}|D]b}trI|jtjk(r,tr&d|j vrt dt|zR|j|d|S)ztOur process may start with some open UNIX sockets which are not initialized by us, invalidating unit tests. z/syslogz skipping %s) r rrrr rrr&r)consnewrs rr[r[\s_ C T[[FNN2djj0mc$i/0 4  Jrc ddl}t|dst |j|S#t$rddl}|j|cYSwxYw)z,Backport of importlib.reload of Python 3.3+.rNreload) importlibrbrPr(imp)moduler)r*s rrbrbosU (y(+ ,'' "zz&!!"s*A  A ctjjtjj|d}tj ddkrddl}|j||Sddl}|jj||}|jj|}|jj||S)Nrr)rrsplitextrr}rr* load_sourceimportlib.utilutilspec_from_file_locationmodule_from_specloader exec_module)rrr*r)specmods rrcrc~s 77  BGG,,T2 3A 6D QtT**~~55dDAnn--d3 $ rc<tj|tdy)zRaise a warning msg.rm) stacklevelN)warningsrd UserWarning)rs rrdrds MM#{q1rct|}|j}t|dk7s |dtk7ryt |dd}t |tsyt d|DS)z-Check if object is an instance of namedtuple.rUrF_fieldsNc3<K|]}t|tywr)r%r&).0ns r z is_namedtuple..s-az!S!-s)r __bases__rYrwrr%r)rtbrs rrfrfsX QA A 1v{adem9d#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. pypyr6z.sorrUN)r$rTrrrrrr-lowerrrrrctypesCDLLrQ)rrextdstrlibsrs rreres 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.dllrr6wow64rEN)rGrMrNrTrrrrrFr+rrr$rrrrWinDLLwindllkernel32 FreeLibraryHMODULEargtypes_handlerQ) rrMrNrIrJrrKrcfilerSrs rreres $# -^^%113 vv||~&&s+BGG,,QVV4::<<qvv||~- FF    )557RWW--aff5;;==D mmD!S! MM#&EI $mm44@@ (0(8(8'9 $!%--0!8J  ? " $mm44@@ (0(8(8'9 $!%--0!8J  s:AI*BG:.I*A G?/I*?HA,I*A#I''I*ctdy)NTrM)rAryrrcleanup_test_procsrYs D!rc,tj|Sr)r}exit)r8rs rr\s#rr)F)TFr)z 127.0.0.1)r)rT)r __future__rrrrGr>rrrrrurrcr r'rr4rrr{r}rrrrrFr9rrrrrrr r r r r rpsutil._commonrrrrrpsutil._compatrrrrrrrrrrPcatch_warnings simplefilterrpsutil._psposixr__all__builtin_module_namesr$rrrhr,r>maxsizer/rr<r=r#r"r.r!rrfr)r*decoder+r]rFASCII_FSrrrr__file__r'getr(HERErrbrr0r1rr2r4r5HAS_NET_IO_COUNTERSr6r3r7r8r,rr9rr:r; HAS_THREADSgetuid SKIP_SYSCONSrr%r&rdevnullr rrrrrr-rrrrrrrErrBrDrCr?r.rr@rArVrUrjr1rXrrYrWrQrrrPrRrSrTrrKskipIfrJrNrOrrLrMrrIrGrHr\r]r^r_r`rarZr#r[rbrcrdrfrerY)rs0rrqso%        & "&(*, "   D(# ZS---  #!RZZ/O>RZZ3O  ' RZZ ' ;;   (  ?X-L ?g-LLL  #' !OJaNA 77f!IBIIK/M!IBIIK/M9)009JK( $3 $ $ & , , .2G G 77  GGLL*D$7 jjnn"'',,x;  ww12*U6>>>:vz* .fnni0 V^^X . &..-8f&786>>95v~~}= V^^X . f&78%H$/Ev/E/E/G*HK6>2"6+ABfnni0  2)"))+"2 'T)] N rzz4  !$F q||I/FGFA &)VX . (!!(` &&R##L*Z&6 &~~NS/l-J!0 "88v""        .1               D  4 &x  @~HX~H^+,a%^a%-a%HhV$ FLFLR@@@@F( &:%[t"!' 2 2 .. ##476Od & ( (2 . .**f"" FMM&..">? q=  "h'zKpsHY,Z5?[ [,Z2Z" Z2"Z+ 'Z21Z25[?[