o a @s4dZddlmZddlZddlZddlZddlZddlZddlZddl Z ddl Z ddl Z ddl Z ddl Z ddlZddlZddlZddlZddlZddlZddlZddlZddlZddlZddlZddlmZddlmZddlmZddlZddlmZddlmZdd lmZdd lm Z dd lm!Z!dd lm"Z"dd lm#Z#ddl$m%Z%ddl$m&Z&ddl$m'Z'ddl(m)Z)ddl(m*Z*ddl(m+Z+ddl(m,Z,ddl(m-Z-ddl(m.Z.ddl(m/Z/ddl(m0Z0e)rddl1Z1nddl2Z1zddl1m3Z3Wn(e4ye5e6dddl3Z3Wdn 1swYYnwej7dkr)ddl8Z8ndZ8e!r4ddl9m:Z:gdZ;dejvZ?d e j>vpLd!e j>vZ@e?pQe@ZAejBd"kZCd#ZDd$ZEd%ZFd&ZGeArreDd'9ZDeGd'9ZGeEd'9ZEeFd'9ZFe jHd(krd)e IZJnd*e IZJe.d+ZKe)rd,Ld-d.ZMnd/ZMeNOd0vZPe jQRe jQSe jQTeUd1d1ZVe jQSeVd2ZWe jQRe jQTeUZXe!oe" ZYeZej[d3Z\eZed4Z]eZed5Z^eZej[d6Z_eZej[d7Z`eZej[d8ZaeZed9ZbeZej[d:ZceZej[d;ZdeZej[d<ZeeZed=Zfz efo egehZiWn ejyd>ZiYnweZed?ZkeZed@ZleZej[dAZme s0eo5e ndkZodBdCZpepZqere jsdDZteuetjvdEdFeweDZxeyedGezZ{e|Z}e|Z~GdHdIdIejZdJdKZeddLdMZedNdOZdPdQZedRdSZedTdUZejeGfdVdWZddXdYZdZd[Zd\d]ZGd^d_d_ezZeejdeGd`dadbdcZee+efdeGd`daddedfZeedeGd`dadgdhZdidjZdkdlZejdmdnZddodpZddrdsZGdtdudue1jZee1_GdvdwdweZe1e=dxGdydzdzeZd{d|Zd}d~ZGdddZGdddZddZeDfddZdddZdddZdddZeedfddZejfddZdddZddZejddZddZddZddZddZddZddZe!r{ejdddZnejdddZejuddZe!reejdddSdS)z Test utilities. )print_functionN)AF_INET)AF_INET6) SOCK_STREAM)AIX)FREEBSD)LINUX)MACOS)POSIX)SUNOS)WINDOWS) bytes2human) print_color) supports_ipv6)PY3)FileExistsError)FileNotFoundError)range)super)u)unicode)which)mockignore)wait_pid)EAPPVEYORDEVNULLGLOBAL_TIMEOUTTOLERANCE_SYS_MEM NO_RETRIESPYPY PYTHON_EXEROOT_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_TEMPERATURESZHAS_MEMORY_FULL_INFOpyrun terminate reap_childrenspawn_testproc spawn_zombiespawn_children_pairZThreadTaskunittestskip_on_access_deniedskip_on_not_implementedretry_on_failureTestMemoryLeakPsutilTestCaseprocess_namespacesystem_namespace print_sysinfoZ install_pipZinstall_test_depschdir safe_rmpath create_exeZ decode_pathZ encode_path get_testfn get_winverkernel_version call_until wait_for_pid wait_for_filecheck_net_address get_free_port bind_socketbind_unix_sockettcp_socketpairunix_socketpaircreate_sockets reload_moduleimport_module_by_pathwarncopyload_shared_lib is_namedtupleZ__pypy__rGITHUB_ACTIONSZ CIBUILDWHEELl iPirjavaz $psutil-%s-z @psutil-%s-u-ƒőősfutf8surrogateescapeufÀ€)asciizus-asciiz..Zscripts cpu_affinitycpu_freq getloadavgenvironionice memory_mapsnet_io_counterscpu_num io_countersrlimitsensors_batteryF sensors_fanssensors_temperaturesthreadscCsdd}trtrtrtdStdStrtjtj StdSt rL|tj pC|tjtj pC|tdtj ddpC|t }|sJtd|Stjtj }tj|s]J||S) NcSs4ztj|dgtjtjdW|StyYdSw)Nz-V)stdoutstderr) subprocess check_callPIPE Exception)exery7/usr/lib/python3/dist-packages/psutil/tests/__init__.pyattempts z_get_py_exe..attemptZpypy3pypypythonz python%s.%sz"can't find python exe real abspath)r]r#rrrospathrealpathsys executabler version_infopsutilProcessrx ValueErrorexists)r{rxryryrz _get_py_exes*  rzr+cC g|] }|drtt|qS)ZSTATUS_ startswithgetattrr.0xryryrz rAF_UNIXcsPeZdZdZfddZddZddZdd Zd d Zd d Z ddZ Z S) ThreadTaskz6A thread task which does nothing expect staying alive.cs$td|_d|_t|_dS)NFMbP?)r__init___running _interval threadingZEvent_flagself __class__ryrzrs zThreadTask.__init__cCs|jj}d||jt|fS)Nz<%s running=%s at %#x>)r__name__rid)rnameryryrz__repr__szThreadTask.__repr__cCs ||SN)startrryryrz __enter__szThreadTask.__enter__cOs |dSr)stop)rargskwargsryryrz__exit__ s zThreadTask.__exit__cCs(|jrtdtj||jdS)zStart thread and keep it running until an explicit stop() request. Polls for shutdown every 'timeout' seconds. zalready startedN)rrrThreadrrwaitrryryrzr s zThreadTask.startcCs0d|_|j|jrt|j|js dSdS)NT)rrsettimesleeprrryryrzruns   zThreadTask.runcCs |jstdd|_|dS)z8Stop thread execution and and waits until it is stopped.zalready stoppedFN)rrjoinrryryrzrs zThreadTask.stop) r __module__ __qualname____doc__rrrrrrr __classcell__ryryrrzrs  rcstfdd}|S)Ncs(z|i|WStytwr)rwr<rrfunryrzwrapper*s  z&_reap_children_on_err..wrapper functoolswrapsrrryrrz_reap_children_on_err)srcKs|dt|dt|dt|dtjtr%d}|d||dur[t}z)t|d|}td |g}t j |fi|}t |t |d d d Wt||St|wt j |fi|}t |t|j|S) a}Creates a python subprocess which does nothing for 60 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 attemps to make sure the process is in a reasonably initialized state. The process is registered for cleanup on reap_children(). stdinrrcwdenv creationflagsNz:from time import sleep;open(r'%s', 'w').close();sleep(60);z-cTdeleteempty) setdefaultrrgetcwdrgr rKrIr$rtPopen_subprocesses_startedaddrPrOpid)cmdkwdsZCREATE_NO_WINDOWtestfnZpylinesprocryryrzr=4s2       r=cCsd}ttd}zItdtj|tf}tr"t |dd\}}nt |\}}t |j }t t|ddd}t|t |}||fWt||durRt|SSt||dur`t|ww) 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(). Ndira; 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(60);" p = subprocess.Popen([r'%s', '-c', s]) p.wait() r)rTFr)rKrrtextwrapdedentrbasenamer$r r:rrrintrP _pids_startedrrI)tfilerssubpchildZgrandchild_pidZ grandchildryryrzr?\s,        r?cstjsJt}td|}d}t|}zW|tt|\}}| \}}z@t | gggtt | d}t|t|tfddd|fW|W|t||durgt|SS|w|t||dur~t|ww)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) NcsSr)statusryzombieryrzszspawn_zombie..zret == psutil.STATUS_ZOMBIE)rr rKrrrTZ settimeoutr r:acceptselectfilenorZrecvrrrrNcloserI)Z unix_filesrcrsockparentconn_Zzpidryrrzr>s<          r>cKs|dd|ddt}z/t|d }||Wdn1s%wYtt|jgfi|}t|j||fWSt yJt |w)zRun python 'src' code string in a separate interpreter. Returns a subprocess.Popen instance and the test file where the source code was written. rrNrswt) rrKopenwriter=r$rrOrrwrI)rrsrcfilefrryryrzr:s       r:cKst|ttfr dnd}tr|rdnd}|d||dtj|dtj|dd|d |tj|fi|}t |t rL|j t d \}}n| \}}|j dkr[t||rat||d rl|d d }|S)zUrun cmd in a subprocess and return its output. raises RuntimeError on error. TFrrshellrrrsZuniversal_newlinesrtimeout N) isinstancestrrr rrtrvrrrrZ communicater returncode RuntimeErrorrZendswith)rrrflagsprrrsryryrzshs&        rc sddddfdd}fddfd d }d d }|}zt|trP|||Wt|tjtjfr<||t|trC|n|j}t|rOJ|St|tjtjfr||Wt|tjtjfrk||t|trr|n|j}t|r~J|St|tjr|||Wt|tjtjfr||t|tr|n|j}t|rJ|Std |t|tjtjfr||t|tr|n|j}t|rJ|w)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. cSsht|tjr ts |n||tr0t|tjr2z t|j|WStj y/YdSwdSdSr) rrtrrrr rrr NoSuchProcessprocrryryrzrs  zterminate..waitcSs6trtrtj}tr|tjkr|tj||dSr)r r]signalSIGKILLr send_signalSIGCONT)rsigryryrzsendsigs  zterminate..sendsigc s\z|Wn!ty(}ztr|jdkrn|jtjkrWYd}~nd}~ww||S)N)OSErrorr winerrorerrnoZESRCH)rrerrrrrryrz term_subprocs  zterminate..term_subproccs.z|Wn tjyYnw||Sr)rrrr ryrz term_psproc s  zterminate..term_psproccsBzt|}Wntjytrt||YSYdSw||Sr)rrrr r)rrr)r ryrzterm_pids zterminate..term_pidcSs8|jr|j|jr|j|jr|jdSdSr)rrrrsr)rryryrz flush_popens  zterminate..flush_popenz wrong type %r) rrrtrrr pid_existsr TypeError)Z proc_or_pidr wait_timeoutr r r rrry)rrr rrzr;s>           r;cCstj|d}trt}t|ts tr t}t|ts|rH|D]}t|ddq$tj|td\}}|D]}t d|t|t j dq8dSdS)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 resursive is True it also tries to terminate and wait() all grandchildren started by this process.  recursiveN)rrz0couldn't terminate process %r; attempting kill())r) rrchildrenrpopr;r wait_procsr rZrr)rrrrrZgoneZaliveryryrzr<5s$  r<cCststdd}td}|D]}|s|dkr||7}q|s(td|d}d}|d}t|d}t|dkrCt|d}t|dkrOt|d}|||fS) z"Return a tuple such as (2, 6, 36).z not POSIXr~.zcan't parse %rrr) r NotImplementedErrorrunameisdigitrsplitrlen)rrcminormicroZnumsmajorryryrzrMZs&          rMcCsdtstdt}t|dr|jpd}ntd|d}|r't| d}nd}|d|d|fS)Nz not WINDOWSservice_pack_majorrz\s\d$rr) r rrZgetwindowsversionhasattrr"researchrgroup)ZwvsprryryrzrLrs  rLc@s<eZdZdZeddddfddZddZdd Zd d ZdS) retryzA retry decorator.NrcCs2|r|rtd||_||_||_||_||_dS)Nz/timeout and retries args are mutually exclusive)r exceptionrretriesintervallogfun)rr*rr+r,r-ryryrzrs zretry.__init__ccsf|jrt|j}t|krdVt|ksdSdS|jr.t|jD]}dVq&dS dVq/r)rrr+r)rstop_atrryryrz__iter__s zretry.__iter__cCs|jdur t|jdSdSr)r,rrrryryrzrs z retry.sleepcs"tfdd}|_|S)Nc srd}D]/}z |i|WSjy3}z|}jdur%|WYd}~qd}~wwtr8|r)r*r-rr)rrexcrrrryrzrs   zretry.__call__..wrapper)rr decorator)rrrryr1rz__call__szretry.__call__) rrrrrwrr/rr3ryryryrzr)s  r)r)r*r-rr,cCs t|trtddSdS)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. {Gz?N)rrr rr)rryryrzrOs rOTcCsNt|d }|}Wdn1swY|s|sJ|r%t||S)z8Wait for a file to be written on disk with some content.rbN)rreadrI)fnamerrrdataryryrzrPs  rPcCs|}t|s J|S)zVKeep calling function for timeout secs and exit if eval() expression is True. )eval)rexprretryryrzrNs rNcCspdd}z)t|}t|jrttj|}nttj|}t r'|WdS||WdSt y7YdSw)z>Convenience function for removing temporary test files or dirsc Sstt}t|kr@z|WStyYnty4}z|}tdt|WYd}~nd}~wwtdt|ks |)Nz ignoring %sr4)rr rZ WindowsErrorrZrr)rr.rrryryrz retry_funs      zsafe_rmpath..retry_funN) rstatS_ISDIRst_moderpartialshutilrmtreeremover r)rr<strryryrzrIs    rIcCs&zt|WdStyYdSw)z-Convenience function for creating a directoryN)rmkdirrrryryrz safe_mkdir s  rFc cs8t}zt|dVWt|dSt|w)z@Context manager which temporarily changes the current directory.N)rrrH)dirnamecurdirryryrzrHs  rHcCstj|r J||r]tdstdt|trtd}t|t s'J|t t ddd }| |Wdn1s?wYzt d|jd|gWt|jdSt|jwtt|trvt|}t||jtjBdSdS) z1Creates an executable file in the given location.gcczgcc is not installedz #include int main() { pause(); return 1; } z.csuffixrNz-o)rrrrrrboolrrrrrKrrtrurrIrAcopyfiler$r r=chmodr?S_IEXEC)ZoutpathZc_coderrDryryrzrJs(    rJrcCs, tjt||d}tj|stj|Sq)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. T)prefixrKr)tempfileZmktempr'rrrr)rKrrryryrzrK;s   rKc@s8eZdZddZeejdsejjZe sddZ dSdS)TestCasecCs.|jj}|ds d|}d||jj|jfS)Nzpsutil.z psutil.tests.z%s.%s.%s)rrrrZ_testMethodName)rZfqmodryryrz__str__Ps   zTestCase.__str__assertRaisesRegexcCsdSrryrryryrzrunTest^szTestCase.runTestN) rrrrSr#unittestrRZassertRaisesRegexprTrrUryryryrzrRLs  rRc@sBeZdZdZdddZddZdd Zd d Zd d ZddZ dS)rDzUTest class providing auto-cleanup wrappers on top of process test utilities. rNcCst||d}|t||S)N)rKr)rK addCleanuprI)rrKrr7ryryrzrKks  zPsutilTestCase.get_testfncOst|i|}|t||Sr)r=rWr;)rrrrryryrzr=ps zPsutilTestCase.spawn_testproccC*t\}}|t||t|||fSr)r?rWr;)rZchild1Zchild2ryryrzr?u   z"PsutilTestCase.spawn_children_paircCrXr)r>rWr;)rrrryryrzr>{rYzPsutilTestCase.spawn_zombiecOs.t|i|\}}|t||t||Sr)r:rWrIr;)rrrrrryryrzr:s  zPsutilTestCase.pyruncCs|tjtj|jt|tjtjfr6|rJz|}Wn tjy)Ynwt d||j ddt |jrAJ|j| |jt dS)Nz3Process.status() didn't raise exception (status=%s)rr) assertRaisesrrrrrr is_runningrAssertionErrorrrZ assertNotInpids)rrrryryrzassertProcessGones   z PsutilTestCase.assertProcessGonerN) rrrrrKr=r?r>r:r^ryryryrzrDfs  rDzunreliable on PYPYc@seZdZdZdZdZdZerdndZdZ e Z e eddZedd Zed d Zd d ZddZddZddZddZddZddZ  dddZddZdS) rCaTest 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) r^rr_TZ PSUTIL_DEBUGcCstddS)NF)r _set_debugclsryryrz setUpClasszTestMemoryLeak.setUpClasscCst|jdSr)rra_psutil_debug_origrbryryrz tearDownClasszTestMemoryLeak.tearDownClasscCs|j}t|d|jS)NZuss) _thisprocmemory_full_inforZrss)rmemryryrz_get_mems zTestMemoryLeak._get_memcCstr|jS|jSr)r rinum_fds num_handlesrryryrz _get_num_fdss  zTestMemoryLeak._get_num_fdscCs|jr t|dtjddSdS)NZyellow)Zcolorfile)verboserrrs)rmsgryryrz_logszTestMemoryLeak._logcCsx|}|||}||}|dkr|d||dkr:tr$dnd}|dkr.|d7}d|||f}||dS) 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)fdZhandlerrz%s unclosed %s after calling %rN)rocallfailr )rrZbeforeZafterdiffZtype_rrryryrz _check_fdss   zTestMemoryLeak._check_fdscCs^tjdd|}t|D] }||}~~qtjdd|}|tjg||}|S)zGet 2 distinct memory samples, before and after having called fun repeadetly, and return the memory difference. r)Z generation)gcZcollectrlrruZ assertEqualZgarbage)rrtimesZmem1rr;Zmem2rwryryrz _call_ntimess    zTestMemoryLeak._call_ntimesc Csg}d}|}td|dD]C} |||} d| t| t| ||f} || | |kp/| |k} | r>| dkr;|| dS| dkrEt|| ||7}| }q |d|)Nrrz,Run #%s: extra-mem=%s, per-call=%s, calls=%sz. )rr{r appendrsprintrvr) rrrz warmup_timesr+ tolerancemessagesZprev_memZincreaseidxrkrrZsuccessryryrz _check_mems(    zTestMemoryLeak._check_memcCs|Srry)rrryryrzruszTestMemoryLeak.callNc Cs|dur|n|j}|dur|n|j}|dur|n|j}|dur!|n|j}z"|dks-Jd|dks5Jd|dks=Jd|dksEJdWntyX}ztt|d}~ww||||||j |||||ddS) zTest a callable.Nrztimes must be >= 1rzwarmup_times must be >= 0zretries must be >= 0ztolerance must be >= 0)rzr~r+r) rzr~r+rr\rrr{rxr)rrrzr~r+rrryryrzexecutes(      zTestMemoryLeak.executec s&fdd}j|fi|dS)znConvenience method to test a callable while making sure it raises an exception on every call. csdSr)rZryr0rrryrzru-sz*TestMemoryLeak.execute_w_exc..callN)r)rr0rrruryrrz execute_w_exc)szTestMemoryLeak.execute_w_exc)NNNN)rrrrrzr~rr*r+rqrrrirLrgetenvrf classmethodrdrgrlrorsrxr{rrurrryryryrzrCs0    rCcCsddl}ddl}ddl}ddl}ddl}ddl}zddl}Wn ty)d}Ynwzddl}Wn ty;d}Ynw| }t j rNt drNt d|d<n=t jr\d|d|d<n/t jrddtt||d<t|dr~|dd |7<n d ||f|d<d t||g|d <t jr|d |d <d ||| g|d<t!|dd|d<|dur|dd|j"7<t jrt drt ddg} t| #dd|d<nd|d<|$d} | r| |d<t%&|d<|'} d| d| df|d<|j(t )*d|d<|j+*d|d<|,|d<t-j./d|d <t-0|d!<t1|d"<|2|d#<t-3|d$<t 4|d%<d&t5d'd(t 6D|d)<t 7} d*t8| j9t:| j;t:| j<f|d+<t =} d*t8| j9t:| j;t:| j<f|d,<t>t ?|d-<t @A}|Bd.d|C||d/<tDd0t%jEd1|FD]\}}tDd2|d3|ft%jEd1qtDd0t%jEd1t%jGHdS)4NrZ lsb_releasezlsb_release -d -sZOSz Darwin %szWindows  win32_editionz, z%s %sZarchr~Zkernelr} __version__z not installedpipz (wheel=%s)rIz --versionrrZglibcz fs-encodingz%s, %slangz%Y-%m-%d %H:%M:%Sz boot-timeruser~homerZpyexeZhostnameZPIDZcpusz%.1f%%, %.1f%%, %.1f%%cSsg|] }|tdqS)d)r cpu_countrryryrzr{z!print_sysinfo..Zloadavgz%s%%, used=%s, total=%sZmemoryswapr]rirzF======================================================================rpz%-17s %s:)I collectionsdatetimegetpasslocaleplatformpprintr ImportErrorwheel OrderedDictrrrrZOSXZmac_verr rmaprZ win32_verr#rsystemversionlistZ architecturemachiner rZpython_implementationZpython_versionZpython_compilerrrrZlibc_verrgetfilesystemencoding getlocaleZ fromtimestamp boot_timestrftimeZnowZgetuserrr expanduserrr$Znodegetpidrtuplerfvirtual_memoryrpercentr usedtotal swap_memoryrr]ras_dictrZpformatr}rsitemsrrflush)rrrrrrrrinfooutrrrkrZpinfokvryryrzrG3s                   rGcCs6t}t|dr |St|drt|SdS)Nrkrdr)rrr#rkrandomchoicerd)rryryrz_get_eligible_cpus   rc@s0eZdZdZddifddifgZddifddddifd difd difd difd difd difddifddifg Zddifddddifddifddifddifddifddifddifddifddifddifddifddifd difd!difd"difd#difgZered$difg7Zed%difg7Zed&difg7Zed'difg7Zered(difg7Ze red)difg7Ze red*e j fifg7Ze red+difg7Zered,difg7Zered-difg7Zered.difg7Zered/dd0d1ifg7ZgZereddifg7Zn ede jfifg7Ze red*e j d2fifg7Ze r8er.ed)e jd3fifg7Zn ed)e jfifg7Ze rFed+egfifg7Zd4ejfifd5difd6difd7difd8difgZerued4ejfifg7Zed4ejfifg7ZeeeeZd9d:ZdDd;d<Zd=d>Z e!d?d@Z"e!dAdBZ#dCS)ErEaA 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_percentryZmemory_percentrrrTr[Zmemory_info_exZoneshotrparentsrrrZcmdlineZ connectionskindall cpu_timesZ create_timerrxrjZ memory_inforniceZnum_ctx_switchesZ num_threadsZ open_filesZppidrrqZusernameZuidsZgidsZterminalrmrlrhrmrdrkrgrnriZgroupedF)rirrZsuspendZresumer;killcCs ||_dSr)_proc)rrryryrzrs zprocess_namespace.__init__ccsbt|}t||D]"\}}}|r|t|j|}tj|g|Ri|}||fVq dSz_Given a list of tuples yields a set of (fun, fun_name) tuples in random order. N)rrshuffle clear_cacherrrr@)rlsrfun_namerrrryryrziters   zprocess_namespace.itercCs|jj|jjdddS)z&Clear the cache of a Process instance.T)Z _ignore_nspN)rZ_initrrryryrzrszprocess_namespace.clear_cachecCs>|D]\}}}d|}t||sd|jj|f}t|qdS)z}Given a TestCase instance and a list of tuples checks that the class defines the required test method names. Ztest_z$%r class should define a '%s' methodN)r#rrAttributeError)rcZ test_classrrrZ meth_namerrryryrztest_class_coverage s z%process_namespace.test_class_coveragecCs`tdd|jD}tdd|jD}tddttjD}||B|A}|r.td|dS)NcSg|]}|dqSrryrryryrzrz*process_namespace.test..cSrrryrryryrzrrcSsg|] }|ddkr|qS)rrryrryryrzrrz!uncovered Process class names: %r)rrignoredrrrr)rcthisrklassZleftoutryryrztests  zprocess_namespace.testN)T)$rrrrZutilsrgettersr r1r2r5rZ RLIMIT_NOFILEr.r4r0r r3ZsettersZNORMAL_PRIORITY_CLASSrZIOPRIO_CLASS_NONEZ IOPRIO_NORMALrrSIGTERMZkillersZ CTRL_C_EVENTZCTRL_BREAK_EVENTrrrrrrrryryryrzrEs        rEc@seZdZdZddifddddifddddifddifd dd difd dd difd dd difd dddifdefifddddifddifddifddddifdefifddifddifddifddifgZerreddd difg7Ze r|eddifg7Ze reddifg7Ze reddifg7Ze reddifg7Ze red difg7Zed!d"ifg7Zd#difd$egfifd%difd&difgZeZed'd(ZejZd)S)*rFzA 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() rryrZlogicalFTZ cpu_statsrZpercpuZdisk_io_countersZperdiskZdisk_partitionsr disk_usageZnet_connectionsrZ net_if_addrsZ net_if_statsrjZpernicrr]rZusersrrerfrprornZwin_service_iterZwin_service_get)ZalgZ process_iterrrZcpu_times_percentccsTt|}t||D]\}}}tt|}tj|g|Ri|}||fVq dSr)rrrrrrr@)rrrrrryryrzrSs   zsystem_namespace.iterN)rrrrrrrrr/HAS_GETLOADAVGr9r8r6r rrrr staticmethodrrErryryryrzrF sV          rFcCst|s J|d|_|S)zA decorator to mark a TestCase class. When running parallel tests, class' unit tests will be run serially (1 process). T)inspectZisclassZ _serialrun)rryryrz serialrunbsrcCsdd}ttd||dS)zZDecorator which runs a test function and retries N times before actually failing. cSstd|tjddS)Nz %r, retryingr)r}rrs)r0ryryrzr-psz retry_on_failure..logfunN)r*rr+r-)r)r\)r+r-ryryrzrBlsrBcfdd}|S)z,Decorator to Ignore AccessDenied exceptions.ctfdd}|S)Ncs:z|i|WStjydurstdw)Nzraises AccessDenied)rZ AccessDeniedrVSkipTestrronly_ifryrzrzs z9skip_on_access_denied..decorator..wrapperrrrrrzr2ysz(skip_on_access_denied..decoratorryrr2ryrrzr@ws  r@cr)z3Decorator to Ignore NotImplementedError exceptions.cr)NcsBz|i|WSty dursdj}t|w)Nz4%r was skipped because it raised NotImplementedError)rrrVr)rrrrrryrzrs  z;skip_on_not_implemented..decorator..wrapperrrrrrzr2s z*skip_on_not_implemented..decoratorryrryrrzrAs  rA 127.0.0.1cCsLtt}||df|dWdS1swYdS)z6Return an unused TCP port. Subject to race conditions.rrN) contextlibclosingsocketbind getsockname)ZhostrryryrzrRs $rRcCs~|dur |ttfvr d}t||}z tjdvr!|tjtjd|||tj kr0| d|WSt y>| w)zBinds a generic socket.Nrr)ntcygwinrr_) rrrrrZ setsockoptZ SOL_SOCKETZ SO_REUSEADDRrrlistenrwr)familytypeaddrrryryrzrSs      rScCsptjsJtj|rJ|ttj|}z|||tjkr)| dW|SW|St y7| w)zBind a UNIX socket.r_) rr rrrrrrrrrwr)rrrryryrzrTs     rTrc Cstt|tG}|||d|}t|t}z%|||} |\}}||kr?||fWWdS| q)t yO| w1sSwYdS)z^Build a pair of TCP sockets connected to each other. Return a (server, client) tuple. r_TN) rrrrrrrconnectrrr)rrZllrZcaddraryryrzrUs(       rUcCstjsJd}}z#t|tjd}|dttjtj}|d||W||fStyD|dur;| |durC| w)zBuild a pair of UNIX sockets connected to each other through the same UNIX file name. Return a (server, client) tuple. Nrr) rr rTrrZ setblockingrrrwr)rZserverZclientryryrzrVs     rVc csg}d}}zm|ttjtj|ttjtjtr3|ttjtj|ttjtjtrWt rWt }t }t |\}}t |tjd}|||fD]}||qO|VW|D]}| q]||fD] }|durrt|qhdS|D]}| qw||fD] }|durt|qw)z1Open as many socket families / types as possible.Nr)r|rSrrr SOCK_DGRAMrrr HAS_CONNECTIONS_UNIXrKrVrTrrI)ZsocksZfname1Zfname2s1s2Zs3rr7ryryrzrWs>      rWcCsddl}trtrtst|tjsJ||tjkrQdd|dD}t |dks-J||D]}d|kr>dksCJ|J|q/tsJt |}| |dS|tj krlt|t s_J|tset |}||dS|tjkrtd|dus}J|dStd |) z[Check a net address validity. Supported families are IPv4, IPv6 and MAC addresses. rNcSsg|]}t|qSry)rrryryrzrrz%check_net_address..rrz([a-fA-F0-9]{2}[:|\-]?){6}zunknown family %r) ipaddressenumrr#rIntEnumrrrrrZ IPv4AddressrrZ IPv6AddressrZAF_LINKr$matchr)rrrZoctsZnumryryrzrQs&  &   rQcCsTdd}dd}dd}dd}d d }||||||||||d S) z*Check validity of a connection namedtuple.cSst|dk}t|dvsJt||d|jksJ|j|d|jks*J|j|d|jks6J|j|d|jksBJ|j|d|jksNJ|j|d|jksZJ|j|rh|d |jksjJ|jdSdS) N)rrrrr~rrr_r)rrtrrladdrraddrrr)rZhas_pidryryrz check_ntuple+s z-check_connection_ntuple..check_ntuplec Ss|jtttfvs J|jtdurt|jtjsJ|n t|jts'J||jtkr{t|j|j }t |8z | |j ddfWntjy`}z |jtjkrVWYd}~n d}~wwWddSWddS1stwYdS|jtkr|jtjksJ|jdSdS)Nr)rrrrrrrrrrrrrrerrorrZ EADDRNOTAVAILrr CONN_NONE)rrrryryrz check_family7s.    " z-check_connection_ntuple..check_familycSsttdt}|jtjtj|fvsJ|jtdur&t|jtjs%J|n t|jt s0J||jtjkrA|j t j ksCJ|j dSdS)NSOCK_SEQPACKET) rrobjectrrrrrrrrrr)rrryryrz check_typeLs  z+check_connection_ntuple..check_typecSs|j|jfD]M}|jttfvrCt|tsJt||sqt|jt s*Jt|jd|jkr5dks;nJ|jt |j |jq|jt krSt|t sSJt|qdS)Nri)rrrrrrrrZportrrQZiprr)rrryryrz check_addrsXs" z,check_connection_ntuple..check_addrscSst|jts J|jddttD}|j|vsJ|j|jttfvr7|jt kr7|jtj ks5J|jdS|jtj ksBJ|jdS)NcSr)ZCONN_rrryryrzrgrzAcheck_connection_ntuple..check_status..) rrrrrrrrrrr)rZvalidsryryrz check_statuses z-check_connection_ntuple..check_statusNry)rrrrrrryryrzcheck_connection_ntuple)s     rcCsJz ddl}t|ds tWntyddl}||YSw||S)z,Backport of importlib.reload of Python 3.3+.rNreload) importlibr#rimpr)modulerrryryrzrX{s   rXcCstjtj|d}tjddkrddl}|||Stjdddkr3ddlm }||| Sddl }|j ||}|j |}|j||S)Nrr~r)SourceFileLoader)rrsplitextrrrrZ load_sourceimportlib.machineryr  load_moduleimportlib.utilutilspec_from_file_locationmodule_from_specloader exec_module)rrrr rspecmodryryrzrYs    rYcCst|tdS)zRaise a warning msg.N)warningsrZ UserWarning)rrryryrzrZrhrZcCsVt|}|j}t|dks|dtkrdSt|dd}t|ts"dStdd|DS)z-Check if object is an instance of namedtuple.rrF_fieldsNcss|] }t|tkVqdSr)rr)rnryryrz sz is_namedtuple..)r __bases__rrrrr)rtbrryryrzr\s  r\c#s|trdnddt|d}fddtD}t|}t||zt ||VWt |dSt |w)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. r|r}z.sorJcs6g|]}tj|jdkr|jvr|jqS)r)rrr lowerrrxextryrzrs   'copyload_shared_lib..N) r#rKrrrirrrArMctypesZCDLLrI)rKdstlibsrryrrzr[s    r[c #sddlm}ddlm}dt|d}fddtD}tr2|s2ddtD}t |}t ||d }z)t |}|VW|d urctj jj}|jg|_||j}|dkrc|t|d S|d urtj jj}|jg|_||j}|dkr|t|w) 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.dllrJcsFg|]}|jrdtj|jvrd|jvr|jqS)r}Zwow64)rrrrrrrryrzrs  r cSs(g|]}dtj|jvr|jqS)r|)rrrrrryryrzrs N)r!r$r%rKrrrir#rrrArMZWinDLLZwindllZkernel32 FreeLibraryZHMODULEargtypesZ_handlerI) rKr$r%r"r#rcfiler'r;ryr&rzr[s8             cCstdddS)NTr)r<ryryryrzcleanup_test_procsrer*cCs t|Sr)rexit)rframeryryrzrs rr)F)TFr_)r)r)r)rZ __future__ratexitrr!rrryrrrr$rrArrr=rtrrQrrrrrrrrrrrr r r r Zpsutil._commonr rrZpsutil._compatrrrrrrrrrVZ unittest2rrcatch_warnings simplefilterrrZpsutil._psposixr__all__builtin_module_namesr#rgrr]r*maxsizer-r"r!r,r rrr'r(decoder)rrZASCII_FSrrrrG__file__r%r&ZHERErr#rr.r/rr0r2r3ZHAS_NET_IO_COUNTERSr4r1r5r6rLrnr7rwr8r9Z HAS_THREADSgetuidZ SKIP_SYSCONSrr$rdevnullrregisterrrr+rrrrrrrrrr=r?r>r:rrr;r<rMrLr)rrOr\rPrNrIrFcontextmanagerrHrJrKrRrDZskipIfrCrGrrErFrrBr@rArRrSrTrUrVrWrQrrXrYrZr\r[r*ryryryrzs                              -                      0 ' &-   U% :     !   1[  B    R   ,