U ;ghK@sdZddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl Z ddl Z ddl Z ddl Z ddlZddlZddlZddlZddlZddlZddlZddlZddlZddlZddlZddlZddlZddlZddlmZddlmZddlmZz ddlZWne k r&dZYnXddl!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*m/Z/e'rddl0m1Z1ddddddddddddd d!d"d#d$d%d&d'd(d)d*d+d,d-d.d/d0d1d2d3d4d5d6d7d8d9d:d;dd?d@dAdBdCdDdEdFdGdHdIdJdKdLdMdNdOdPdQdRdSdTdUdVdWdXdYdZd[d\d]d^gKZ2d_ej3kZ4d`e j5kpdae j5kZ6e6Z7dbe j5kZ8dce j5kZ9ej:ddkZ;e <dekZ=e-dfdgZ>e$re>dhkZ?e>dikZ@ndjZ?djZ@dkZAdlZBdmZCdnZDe7r,eAdo9ZAeDdo9ZDeBdp9ZBeCdo9ZCe jEdqkrJdre FdsZGndte FdsZGduZHdvIdwdxZJeKLdykZMe jNOe jNPe jNQeRdzdzZSe j5Td{e jNPeSd|ZUe jNOe jNQeRZVeWe!jXd}ZYeWe!d~ZZeWe!jXdZ[eWe!dZ\eWe!jXdZ]eWe!jXdZ^e'oe( Z_eWe!dZ`eWe!jXdZaeWe!jXdZbeWe!jXdZceWe!dZdzedo`eee!fZgWnehk r~djZgYnXeWe!dZieWe!dZjeWe!jXdZke$se"oe ldkZmddZnen\ZoZpeqe jrdZsetesjuddeve!DZwexedeyZze{Z|e{Z}Gdd<dZddd?ZdddTZeedfddUZejfddVZdddWZddXZejddYZddRZddƄZddSZddZZdd[Zdd\Zdd^Ze'r^ejddd]Znejddd]ZejtddτZe'reejddфdS)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_pidDEVNULLGLOBAL_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_childrenspawn_testproc 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_namedtupleZ__pypy__GITHUB_ACTIONSZ CIBUILDWHEELZ COVERAGE_RUNZPYTEST_XDIST_WORKERlaarch64cCsntd}ttt|ddd}|dkrjtjtj dddgdd id d }ttt|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_strversionruU/opt/bot/vods_filmes/script/venv/lib/python3.8/site-packages/psutil/tests/__init__.py macos_versionzs  rw)rb) rFrbiPijavaz$psutil--z@psutil-u-ƒőősfutf8surrogateescape>us-asciiasciiz..ZPSUTIL_SCRIPTS_DIRscripts cpu_affinitycpu_freqenviron getloadavgionice memory_mapsnet_io_counterscpu_num io_countersrlimitsensors_battery sensors_fanssensors_temperaturesthreadscCsdd}tj}ttdd}trFtjdkrF|dk rFtj|d<||fStrTtj|fSt r|tjp|tj tjp|t djtjddp|t}|std||fStj tj}tj |st|||fSdS) NcSs>ztj|dgtjtjdWntjk r4YdSX|SdS)Nz-V)stdoutstderr)ro check_callPIPECalledProcessError)exerururvattempts z_get_py_exe..attempt_base_executable)r{__PYVENV_LAUNCHER__z python{}.{}raz"can't find python exe real abspath)osrcopygetattrrqr version_inforrr^rpathrealpathshutilwhichformatpsutilProcessr ValueErrorexistsAssertionError)rrgbaserrururv _get_py_exes0     rzr+cCs g|]}|drtt|qS)ZSTATUS_ startswithrr.0xrururv s rAF_UNIXcsPeZdZdZfddZddZddZdd Zd d Zd d Z ddZ Z S)r;z6A thread task which does nothing expect staying alive.cs$td|_d|_t|_dS)NFMbP?)super__init___running _interval threadingEvent_flagself __class__rurvrs zThreadTask.__init__cCs(|jj}d|d|jdt|ddS)N)r__name__ridrnamerururv__repr__szThreadTask.__repr__cCs ||SN)startrrururv __enter__#szThreadTask.__enter__cOs |dSr)stop)rargskwargsrururv__exit__'szThreadTask.__exit__cCs(|jrtdtj||jdS)zStart thread and keep it running until an explicit stop() request. Polls for shutdown every 'timeout' seconds. zalready startedN)rrrThreadrrwaitrrururvr*s zThreadTask.startcCs(d|_|j|jr$t|jqdS)NT)rrsettimesleeprrrururvrun3s zThreadTask.runcCs |jstdd|_|dS)z8Stop thread execution and and waits until it is stopped.zalready stoppedFN)rrjoinrrururvr9szThreadTask.stop) r __module__ __qualname____doc__rrrrrrr __classcell__rururrvr;s  cstfdd}|S)Ncs.z ||WStk r(tYnXdSr) Exceptionr7rrfunrurvwrapperGs  z&_reap_children_on_err..wrapper functoolswrapsrrrurrv_reap_children_on_errFsrcKs|dt|dt|dt|dttrHd}|d||dkrttd}zHt|d |d }td |g}t j |f|}t |t |d d d W5t|Xn"t j |f|}t |t|j|S)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(). stdinrcwdrg creationflagsNdirzimport time;open(r'z6', 'w').close();[time.sleep(0.1) for x in range(100)];rdTdeleteempty) setdefaultrrgetcwdrr rKrHrroPopen_subprocesses_startedaddrPrOpid)cmdkwdsZCREATE_NO_WINDOWtestfnZpylinesprocrururvr8Rs,          cCsd}ttd}ztdtj|dtd}t rJt |dd\}}n t |\}}t |j }tt|dd d }t|t |}||fWSt||dk rt|XdS) 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(0.1) for x in range(100 * 6)];" p = subprocess.Popen([r'z.', '-c', s]) p.wait() r)rTFr)rKrrrHtextwrapdedentrbasenamerr r5rrrrmrP _pids_startedr)tfilerssubpchildZgrandchild_pidZ grandchildrururvr:|s&       cstjs tt}td|d}d}t|}z| t t |\}}| \}}zXt |gggt t|d}t|t|tfdd|fWWS|XW5|t||dk rt|XdS)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) NcstjkSr)statusr STATUS_ZOMBIEruzombierurvzspawn_zombie..)rr rrKrrrUcloserH settimeoutrr5acceptselectfilenormrecvrrrrN)Z unix_filesrcrsockparentconn_Zzpidrurrvr9s.       c Ks|dd|ddt}zFt|d}||W5QRXtt|jgf|}t|j||fWSt k rt |YnXdS)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) rrKopenwriter8rrrOrrrH)r rsrcfilefrrururvr5s     cKstrdnd}|dtj|dtj|dd|d|t|trTt|}tj|f|}t ||j t d\}}|j dkrt|||rt||d r|d d }|S) zURun cmd in a subprocess and return its output. raises RuntimeError on error. rrrrrhTrtimeout N)r rror isinstancestrshlexrnrrr communicater returncode RuntimeErrorr[endswith)rrflagsprrrururvshs"          r!c sddddfdd}fddfd d }d d }|}zdt|trb|||WNSt|tjtjfr||W.St|tjr|||WSt d |W5t|tjtjfr||t|tr|n|j}t|rt|XdS)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. cSsJ||trFt|tjrFzt|j|WStjk rDYnXdSr) rr rrorrrr NoSuchProcessprocrrururvrs  zterminate..waitcSs6trtrtj}tr(|tjkr(|tj||dSr)rr^signalSIGKILLr send_signalSIGCONT)r$sigrururvsendsig s  zterminate..sendsigc s^z|WnDtk r"Yn2tk rR}ztr@|jdkr@W5d}~XYnX||S)N)ProcessLookupErrorOSErrorr winerror)r$rerrr*r)rrurvterm_subprocess_procsz'terminate..term_subprocess_proccs0z|Wntjk r$YnX||Sr)rr"r#r0rurvterm_psutil_proc"s z#terminate..term_psutil_proccsFzt|}Wn(tjk r6tr2t||YSYn X||SdSr)rrr"r r)rrr$)r2rurvterm_pid)s zterminate..term_pidcSs4|jr|j|jr |j|jr0|jdSr)rrrr)r$rururv flush_popen3s   zterminate..flush_popenz wrong type N) rrorrrmr pid_existsrr TypeError)Z proc_or_pidr) wait_timeoutr1r3r4r rru)r*r)r2rrvr6s&        cCstj|d}tr&t}t|qtr||7}q qDq |sVtd|d}d}|d}t|d}t|dkrt|d}t|dkrt|d}|||fS) z"Return a tuple such as (2, 6, 36).z not POSIXrar`z can't parse rr{) r NotImplementedErrorrunameisdigitrrnrmlen)rrAcminormicroZnumsmajorrururvrMrs&        cCs0ts tdt}|jpd}|d|d|fS)Nz not WINDOWSrr?)r r@rqgetwindowsversionservice_pack_major)ZwvsprururvrLs  c@s<eZdZdZeddddfddZddZdd Zd d ZdS) retryzA retry decorator.NrcCs2|r|rtd||_||_||_||_||_dS)Nz/timeout and retries args are mutually exclusive)r exceptionrretriesintervallogfun)rrLrrMrNrOrururvrszretry.__init__ccsT|jr*t|j}t|krPdVqn&|jrHt|jD] }dVq:ndVqHdSr)rrrMrange)rstop_atrrururv__iter__s   zretry.__iter__cCs|jdk rt|jdSr)rNrrrrururvrs z retry.sleepcs"tfdd}|_|S)Nc std}D]b}z||WSjk rh}z,|}jdk rH|WYqW5d}~XYqXq|dSr)rLrOr)rrexcrrrrurvrs  zretry.__call__..wrapper)rr decorator)rrrrurTrv__call__szretry.__call__) rrrrrrrRrrVrurururvrKs  rKr)rLrOrrNcCs$|tkrt|t|dS)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)rpidsr"rrrururvrOs  Tc Cs:t|d}|}W5QRX|s*|s*t|r6t||S)z8Wait for a file to be written on disk with some content.rb)rreadrrH)fnamerrrdatarururvrPs cCs|}|st|S)z1Keep calling function until it evaluates to True.)r)rretrururvrNscCsndd}zLt|}t|jr0ttj|}nttj|}t rJ|n||Wnt k rhYnXdS)z?Convenience function for removing temporary test files or dirs.c Ss|tt}t|krtz|WStk r4Yn4tk rf}z|}td|W5d}~XYnXtdq |dS)Nz ignoring g{Gz?)rrFileNotFoundErrorr-r[r)rrQrr/rururv retry_funs    zsafe_rmpath..retry_funN) rstatS_ISDIRst_moderpartialrrmtreeremover r^)rr_strrururvrHs   cCs(zt|Wntk r"YnXdS)z.Convenience function for creating a directory.N)rmkdirFileExistsErrorrrururv safe_mkdir$sric cs.t}zt|dVW5t|XdS)z@Context manager which temporarily changes the current directory.N)rrrG)dirnamecurdirrururvrG,s   cCsRtj|rt|tt|tt |t rNt |}t ||j t jB|S)z6Create a Python executable file in the given location.)rrrratexitregisterrHrcopyfilerr r`chmodrbS_IEXEC)rrfrururvrI7s   c Cstj|rt|tds(td|dkr int main() { pause(); return 1; } z.csuffixrz-o)rrrrrrpytestskiprrrrrlrmrHrrKrrror)rZc_coderrururvrJBs     r>cCs>tjt||d}tj|stj|}tt ||SqdS)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. )prefixrsrN) tempfilemktemprrrrrrlrmrH)rsrrrrururvrK\s    c@sTeZdZdZeddZedddZedddZedd d ZGd d d Z dS)rFzA 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 cOs4tt}tjdd|tjdtdd|S)zMimics pytest.main(). It has the same effect as running `python3 -m unittest -v` from the project root directory. ra) verbosityz.ExceptionInfoNcSs|jSr)_excrrururvvaluesz/fake_pytest.raises..ExceptionInfo.value)rrrrpropertyrrurururv ExceptionInfosrNc 3sz}z |VWnV|k rf}z8|rPt|t|sPd|d|d}t|||_W5d}~XYnXt|ddS)N"z" does not match "z not raised)researchrrr)rSmatchZeinfor/msgrrurvcontexts z#fake_pytest.raises..context)r)N) contextlibcontextmanager)rSrrrurrvraisess zfake_pytest.raisescCs"|rt||St|S)zMimics `pytest.warns`.)r<TestCaseZassertWarnsRegexZ assertWarns)warningrrururvwarnsszfake_pytest.warnsr>cCst|dS)zMimics `unittest.SkipTest`.N)r<ZSkipTestreasonrururvruszfake_pytest.skipc@s(eZdZedddZGdddZdS)zfake_pytest.markr>cCs t||S)z'Mimics `@pytest.mark.skipif` decorator.)r<ZskipIf) conditionrrururvskipifszfake_pytest.mark.skipifc@s"eZdZdZdddZddZdS)zfake_pytest.mark.xdist_groupz4Mimics `@pytest.mark.xdist_group` decorator (no-op).NcCsdSrrurrururvrsz%fake_pytest.mark.xdist_group.__init__cCs|Srru)rZ cls_or_methrururvrVsz%fake_pytest.mark.xdist_group.__call__)N)rrrrrrVrurururv xdist_groups rN)r>)rrr staticmethodrrrurururvmarks r)N)N)r>) rrrrrrrrrurrurururvrFos    c@sZeZdZdZdddZddZdd Zd d Zd d ZddZ ddZ ddZ ddZ dS)rAzTest class providing auto-cleanup wrappers on top of process test utilities. All test classes should derive from this one, even if we use pytest. r>NcCst||d}|t||S)N)rsr)rK addCleanuprH)rrsrr[rururvrKs  zPsutilTestCase.get_testfncOst||}|t||Sr)r8rr6)rrrrrururvr8s  zPsutilTestCase.spawn_testproccCs*t\}}|t||t|||fSr)r:rr6)rZchild1Zchild2rururvr:s   z"PsutilTestCase.spawn_children_paircCs*t\}}|t||t|||fSr)r9rr6)rr rrururvr9s   zPsutilTestCase.spawn_zombiecOs*t||\}}|t||t||Sr)r5rrHr6)rrrrrrururvr5s  zPsutilTestCase.pyruncCst|tjst|j|jks t|j|jks0t|jr@|js@tt|tjrt|j|j ks\t|jdk rt|jdksttt |t |dSNr) rrErrorrrr_name ZombieProcessppidZ_ppidrrepr)rr$rSrururv_check_proc_excs   zPsutilTestCase._check_proc_excc Csttj4}zt|Wntjk r:tdYnXW5QRX|jj|ksVt|jj dksftt |rxt||t kst|ddt DkstdS)Nz&wasn't supposed to raise ZombieProcesscSsg|] }|jqSrurXrrururvrsz0PsutilTestCase.assertPidGone..) rtrrr"rrrrrrr5rW process_iter)rrcmrururv assertPidGoneszPsutilTestCase.assertPidGonec Cs||jt|}|j|jddD]\}}|j||dpz |}WnFtjk r^YnHtjk r}z| ||W5d}~XYnXd|d|}t |W5QRXq$|j dddS)NT clear_cacher$rzProcess.z!() didn't raise NSP and returned rr) rrrBiterallsubTestrrr"rrr)rr$nsrrr]rSrrururvassertProcessGones  z PsutilTestCase.assertProcessGonec Csdt|j}||kstts4ts4t|t|ks4t|tjksFt| sRtt |jsbt| |jt ks|t|jddt Dkstit_|jddt Dkstt|}|j|jddD]f\}}|j||dJz |Wn8tjtjfk r,}z|||W5d}~XYnXW5QRXqtrttj}|W5QRX|||jttj}|W5QRX|||jttj}|W5QRX|||j|||| | stt |js t|jt kst|jddt Dks.cSsg|] }|jqSrurXrrururvr!sTrrcSsg|] }|jqSrurXrrururvr=scSsg|] }|jqSrurXrrururvr?s)!rrrrr rhashrr is_runningr5as_dictrWrZ_pmaprBrrrr AccessDeniedrrrtrcmdlinerrrsuspendresumer6kill)rr$clonerrrrSrrururvassertProcessZombiesN    *z"PsutilTestCase.assertProcessZombie)r>N) rrrrrKr8r:r9r5rrrrrurururvrAs   zunreliable on PYPYrc@seZdZdZdZdZdZer dndZdZ e Z e edZedd Zed d Zd d ZddZddZddZddZddZddZdddZddZdS) r@aTest 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) rbrrzTZ PSUTIL_DEBUGcCstddS)NF)r _set_debugclsrururv setUpClassyszTestMemoryLeak.setUpClasscCst|jdSr)rr_psutil_debug_origrrururv tearDownClass}szTestMemoryLeak.tearDownClasscCs|j}t|d|jS)NZuss) _thisprocmemory_full_inforZrss)rmemrururv_get_mems zTestMemoryLeak._get_memcCstr|jS|jSdSr)r rnum_fds num_handlesrrururv _get_num_fdss zTestMemoryLeak._get_num_fdscCs|jrt|dtjddS)Nyellow)colorfile)verboserrqr)rrrururv_logszTestMemoryLeak._logcCs|}|||}||}|dkr@d|d}|||dkrtrPdnd}|dkrd|d7}|d|d |}||d S) 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)fdhandler?rz unclosed z after calling N)rcallfailr )rrbeforeafterdiffrtype_rururv _check_fdss    zTestMemoryLeak._check_fdscCs^tjdd|}t|D]}||}~~qtjdd|}tjgksRt||}|S)zGet 2 distinct memory samples, before and after having called fun repeatedly, and return the memory difference. r?)Z generation)gcZcollectrrPrgarbager)rrtimesZmem1rr]Zmem2rrururv _call_ntimess    zTestMemoryLeak._call_ntimesc Csg}d}|}td|dD]}|||} d|t| t| ||} || | |kp^| |k} | r||dkrv|| dS|dkrt|| ||7}| }q|d|dS)Nrr?z,Run #{}: extra-mem={}, per-call={}, calls={}z. ) rPrrr appendrprintrr) rrrrM tolerancemessagesZprev_memZincreaseidxrrsuccessrururv _check_mems.     zTestMemoryLeak._check_memcCs|Srru)rrrururvrszTestMemoryLeak.callNc Cs|dk r |n|j}|dk r|n|j}|dk r0|n|j}|dk rB|n|j}zD|dksZtd|dksjtd|dksztd|dkstdWn.tk r}ztt|W5d}~XYnX||||||j ||||ddS) zTest a callable.Nr?ztimes must be >= 1rzwarmup_times must be >= 0zretries must be >= 0ztolerance must be >= 0)rrMr) r warmup_timesrMrrrrrrr)rrrrrMrr/rururvexecutes  zTestMemoryLeak.executec s"fdd}j|f|dS)znConvenience method to test a callable while making sure it raises an exception on every call. csdSr)Z assertRaisesrurSrrrurvrsz*TestMemoryLeak.execute_w_exc..callN)r)rrSrrrrurrv execute_w_excszTestMemoryLeak.execute_w_exc)NNNN)rrrrrrrrrMrrrrboolrgetenvr classmethodrrrrrrrrrrrrurururvr@Ps2    cCsddl}ddl}ddl}ddl}ddl}z ddl}Wntk rLd}YnXz ddl}Wntk rrd}YnX|}t j rt drt d|d<n~t jrdtd|d<n`t jrddttt|d<ttdr|dd t7<ntdt|d<d tttg|d <t jrPtd |d <d tt t!g|d <t"|dd|d<|dk r|dd|j#d7<t jrt drt ddg}t|$dd|d<nd|d<t%d} | r| |d<t&'|d<|(} | dd | d|d<|j)t *+d|d<|j,+d|d<|-|d<t.j/0d|d<t.1|d <t2|d!<t3|d"<t.4|d#<t 5|d$<d%j6t7d&d't 8D|d(<t 9} d)6t:| j;t<| j=t<| j>|d*<t ?} d)6t:| j;t<| j=t<| j>|d+<t@t A|d,<t BC} | Dd-d|E| |d.<tFd/t&jGd0|HD]$\}}tFd16|d2|t&jGd0qrtFd/t&jGd0t&jIJt&jIJdS)3Nr lsb_releasezlsb_release -d -sZOSzDarwin zWindows  win32_editionz, archraZkernelpython __version__z not installedpipz (wheel=)rqz --versionrr?glibcz fs-encodinglangz%Y-%m-%d %H:%M:%Sz boot-timeruser~homerZpyexehostnameZPIDZcpusz{:.1f}%, {:.1f}%, {:.1f}%css|]}|tdVqdS)dN)r cpu_countrrururv @sz print_sysinfo..Zloadavgz{}%%, used={}, total={}ZmemoryswaprWrr$zF======================================================================rz {:<17} {}:)K collectionsdatetimegetpasslocalepprintr ImportErrorwheel OrderedDictrrrrr!ZOSXrirjr rrlr win32_verhasattrrsystemrtlist architecturemachiner rApython_implementationpython_versionpython_compilerrrrnlibc_verrqgetfilesystemencoding getlocale fromtimestamp boot_timestrftimenowgetuserrr expanduserrrnodegetpidrrrkrvirtual_memoryrmpercentr usedtotal swap_memoryrCrWrrr;pformatrritemsrflush)rrrrrrrinfooutrrrrZpinfokvrururvrDs                   cCs8tdd}z||dkWStk r2YdSXdS)NcSsXi}td}|ddD]6}dd|dD}|dt|d}}|||<q|S)Nztasklist.exe /NH /FO csvr?cSsg|]}|ddqS)rr>)replacerrururvrisz@is_win_secure_system_proc..get_procs..,r)r! splitlinesrnrm)r]rlinebitsrrrururv get_procsds z,is_win_secure_system_proc..get_procsz Secure SystemF)rKeyError)rr&rururvrEbs  cCs6t}t|dr|St|dr2t|SdS)Nrrr)rrrrrandomchoicer)r rururv_get_eligible_cputs   r*c@sBeZdZdZddifddifgZddifddddifd difd difd difd difd difddifddifg Zddifddifddifddifddifddifddifddifddddifddifddifddifddifd difd!difd"difd#difgZerFed$difg7Zed%difg7Zed&difg7Zed'difg7Zer\ed(difg7Ze rred)difg7Ze red*e j fifg7Ze red+difg7Zered,difg7Zered-difg7Zered.difg7Zered/dd0d1ifg7ZgZereddifg7Znede jfifg7Ze rJed*e j d2fifg7Ze rerned)e jd3fifg7Zned)e jfifg7Ze red+egfifg7Zd4ejfifd5difd6difd7difd8difgZered4ejfifg7Zed4ejfifg7ZeeeeZd9d:ZdDd;d<Zd=d>Z e!d?d@Z"e!dAdBZ#dCS)ErBaA 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_percentruZmemory_percentrr:r9T connectionsrZoneshotr parentsrrrr cpu_timesZ create_timerrrZ memory_infornet_connectionskindrniceZnum_ctx_switchesZ num_threadsZ open_filesrrrusernameZuidsZgidsZterminalrrrrrrrrrgroupedF)rirr'rrr6rcCs ||_dSr)_proc)rr$rururvrszprocess_namespace.__init__ccsZt|}t||D]>\}}}|r,|t|j|}tj|f||}||fVqdSz_Given a list of tuples yields a set of (fun, fun_name) tuples in random order. N)rr(shufflerrr5rrc)rlsrfun_namerrrrururvrs  zprocess_namespace.itercCs|jj|jjdddS)z&Clear the cache of a Process instance.T)Z _ignore_nspN)r5_initrrrururvrszprocess_namespace.clear_cachecCsB|D]8\}}}d|}t||s|jjd|d}t|qdS)z}Given a TestCase instance and a list of tuples checks that the class defines the required test method names. Ztest_z class should define a z methodN)rrrAttributeError)rZ test_classr8r9r meth_namerrururvtest_class_coverages  z%process_namespace.test_class_coveragecCsVdd|jD}dd|jD}ddttjD}||B|A}|rRtd|dS)NcSsh|] }|dqSr.rurrururv sz)process_namespace.test..cSsh|] }|dqSr.rurrururvr>scSsh|]}|ddkr|qS)rrrurrururvr>s zuncovered Process class names: )rignoredrrrr)rthisr?klassZleftoutrururvtests  zprocess_namespace.testN)T)$rrrrutilsr?gettersr r&r'r*rZ RLIMIT_NOFILEr#r)r%r r(ZsettersZNORMAL_PRIORITY_CLASSrZIOPRIO_CLASS_NONEZ IOPRIO_NORMALr*r%SIGTERMZkillersZ CTRL_C_EVENTZCTRL_BREAK_EVENTrrrrrr=rBrurururvrB}s       c@seZdZdZddifddddifddddifddifd dd difd dd difd dd difd dddifdefifddddifddifddifddddifdefifddifddifddifddifgZere re dkrneddd difg7Ze r eddifg7Ze r"eddifg7Zer8eddifg7ZerNed difg7Zerted!difg7Zed"d#ifg7Zd$difd%egfifd&difd'difgZeZed(d)ZejZd*S)+rCzA 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() rrurZlogicalFTZ cpu_statsr/ZpercpuZdisk_io_countersZperdiskZdisk_partitionsr disk_usager0r1Z net_if_addrsZ net_if_statsrZpernicr5rWrZusersrarm64rrrrrZwin_service_iterZwin_service_get)algrr<r+Zcpu_times_percentccsLt|}t||D]0\}}}tt|}tj|f||}||fVqdSr6)rr(r7rrrrc)r8r9rrrrururvr=s   zsystem_namespace.iterN)rrrrrrrrDr$rrirHAS_GETLOADAVGr.r-r+r rrr?rrrrBr=rurururvrCsX          cCsdd}ttd||dS)zZDecorator which runs a test function and retries N times before actually failing. cSst|dtjddS)Nz , retryingr)rrqr)rSrururvrOQsz retry_on_failure..logfunN)rLrrMrO)rKr)rMrOrururvr?Lscsfdd}|S)z,Decorator to Ignore AccessDenied exceptions.cstfdd}|S)Ncs@z ||WStjk r:dk r,s,tdYnXdS)Nzraises AccessDenied)rrrtrurronly_ifrurvr]s z9skip_on_access_denied..decorator..wrapperrrrKrrvrU\s z(skip_on_access_denied..decoratorrurKrUrurLrvr=Ys csfdd}|S)z3Decorator to Ignore NotImplementedError exceptions.cstfdd}|S)NcsJz ||WStk rDdk r*s*jd}t|YnXdS)Nz2 was skipped because it raised NotImplementedError)r@rrtru)rrrrJrurvrps  z;skip_on_not_implemented..decorator..wrapperrrrLrrvrUos z*skip_on_not_implemented..decoratorrurMrurLrvr>ls  127.0.0.1c Cs:t(}||df|dW5QRSQRXdS)z6Return an unused TCP port. Subject to race conditions.rr?N)socketbind getsockname)hostr rururvrSs cCs|dkr|tthkrd}t||}z@tjdkrB|tjtjd|||tj kr`| d|WSt k r| YnXdS)zBinds a generic socket.Nr>r>cygwinntr?rz) rrrOrr setsockopt SOL_SOCKET SO_REUSEADDRrPrlistenrr)familytypeaddrr rururvrTs     cCsrtjs ttj|rt|ttj|}z"|||tj krL| dWnt k rl| YnX|S)zBind a UNIX socket.rz) rr rrrrrOrrPrrYrr)rr[r rururvrUs   rSc Cst|t}|||d|}t|t}zJ|||}|\}}||krt||fWW5QRS|qJWntk r|YnXW5QRXdS)z^Build a pair of TCP sockets connected to each other. Return a (server, client) tuple. rzN) rOrrPrYrQconnectrrr-)rZr\ZllrDZcaddrarururvrVs     cCstjs td}}z@t|tjd}|dttjtj}|d||Wn6t k r|dk rr| |dk r| YnX||fS)zBuild a pair of UNIX sockets connected to each other through the same UNIX file name. Return a (server, client) tuple. Nr[r) rr rrUrOr setblockingrr]rr)rserverclientrururvrWs   c csg}d}}z|ttjtjttjtjftrX|ttj tjttj tjft rt rt }t }t |\}}t|tjd}|||fD]}||q|VW5|D] }|q||fD]}|dk rt|qXdS)z1Open as many socket families / types as possible.Nr_)rrHextendrTrOrr SOCK_DGRAMrrr r/rKrWrUr)socksZfname1Zfname2rr[s1s2Zs3rururvrXs2         cCst|tjst||tjkrzdd|dD}t|dksFt||D]"}d|krbdksJnt|qJt |nZ|tj krt|t st|t |n2|t jkrtd|dk st|ntd |dS) z[Check a net address validity. Supported families are IPv4, IPv6 and MAC addresses. cSsg|] }t|qSru)rmrrururvrsz%check_net_address..r`r|rz([a-fA-F0-9]{2}[:|\-]?){6}Nzunknown family )renumIntEnumrrOrrnrC ipaddress IPv4Addressrr IPv6AddressrZAF_LINKrrr)r\rZZoctsnumrururvrQs      cCsTdd}dd}dd}dd}d d }||||||||||d S) z*Check validity of a connection namedtuple.cSst|dk}t|dks$tt||d|jksr+rrr?rar{r|rzr+) rCrrrZr[laddrraddrrr)r Zhas_pidrururv check_ntuples z-check_connection_ntuple..check_ntuplec Ss|jttthkst|jt|jtjs0t||jtkrt|j|j P}z| |j ddfWn0t k r}z|j t jkrW5d}~XYnXW5QRXn |jtkr|jtjkst|jdSr)rZrrrrrrirjrOr[rPror-errnoZ EADDRNOTAVAILrr CONN_NONE)r rr/rururv check_family s    z-check_connection_ntuple..check_familycSshttdt}|jtjtj|hks,t|jt|jtj sBt||jtjkrd|j t j ksdt|j dS)NSOCK_SEQPACKET) rrOobjectr[rrdrrrirjrrrs)r rurururv check_type1s z+check_connection_ntuple..check_typecSs|j|jfD]}|jtthkrt|ts4tt||s:q t|j t sTtt|j d|j krjdksvnt|j t |j |jq |jt kr t|ts tt|q dS)Nri)rorprZrrrrkrr[portrmrQiprr)r r\rururv check_addrs=s" z,check_connection_ntuple..check_addrscSst|jtst|jddttD}|j|ks.check_status..) rrrrrrrZrrr[rrs)r Zvalidsrururv check_statusJsz-check_connection_ntuple..check_statusNru)r rqrtrwrzr{rururvcheck_connection_ntuples    r|cCsJg}|D]<}tr:|jtjkr:tr:d|jkr:td|q||q|S)ztOur process may start with some open UNIX sockets which are not initialized by us, invalidating unit tests. z/syslogz skipping )r rZrOrrrprr)Zconsnewr rururvrR\s cCs t|Sr) importlibreload)modulerururvrYoscCsBtjtj|d}tj||}tj|}|j ||Sr) rrsplitextrr~utilspec_from_file_locationmodule_from_specloader exec_module)rrspecmodrururvrZss   cCstj|tdddS)zRaise a warning msg.rarzN)r~r[r)rrururvr[scCsVt|}|j}t|dks&|dtk r*dSt|dd}t|tsDdStdd|DS)z-Check if object is an instance of namedtuple.r?rF_fieldsNcss|]}t|tVqdSr)rr)rnrururvrsz is_namedtuple..)r[ __bases__rCrkrrr)rtbrrururvr]s  c#sttrdnddt|d}fddtD}t|}t||zt ||VW5t |XdS)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. pypyrz.sorrcs6g|].}tj|jdkr|jkr|jqS)r?)rrrlowerrrextrurvrs'copyload_shared_lib..N) rrKrrrr(r)rrnrHctypesCDLL)rsdstlibsr rurrvr\s       c #sddlm}ddlm}dt|d}fddtD}trb|sbddtD}t |}t ||d }zt|}|VW5|d k rtj j j}|jg|_||j}|dkr|t|Xd S) 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.dllrrcsFg|]>}|jrdtj|jkrd|jkr|jqS)rZwow64)rrrrrrrrurvrs rcSs(g|] }dtj|jkr|jqS)r)rrrrrrururvrsN)rrrrKrrrrr(r)rrnwindllkernel32 FreeLibraryZHMODULEZargtypesZ_handlerHZWinDLL) rsrrrrr cfilerr]rurrvr\s0            cCstdddS)NTr8)r7rurururvcleanup_test_procssrcCs t|Sr)rqexit)r)rrururvrrr)N)F)TF)N)r>N)N)N)rN)rS)r>)r>)rrlrrrirrrrr~rkrrir(rrrrr%rOr`rorqrwrrrr<r~rrrrtrrrrrrr r r r Zpsutil._commonr rrrrZpsutil._psposixr__all__builtin_module_namesrrr^rr2r4maxsizer"rr3rwr0r1rrr!rrrrrdecoderr rZASCII_FSrrrrj__file__rgetrr}rrr#r$r%rIr'r(r/ZHAS_NET_IO_COUNTERSr)r&r*r+rrr,rr-r.Z HAS_THREADSgetuidZ SKIP_SYSCONSrrrrdevnullrrmrrr rrvrrrrrr;rr8r:r9r5r!rEr6r7rMrLrKr"rOr^rrPrNrHrirrGrIrJrKrFrrArrr@rDrEr*rBrCr?r=r>rSrTrUrVrWrXrQr|rRrYrZr[r]r\rrurururvs*                   -                           ,   0  ) &)  T % 9   "   M &l  F     K   3