File manager - Edit - /home/newsbmcs.com/public_html/static/img/logo/psutil.tar
Back
tests/test_misc.py 0000644 00000071750 15030211765 0010264 0 ustar 00 #!/usr/bin/env python3 # Copyright (c) 2009, Giampaolo Rodola'. All rights reserved. # Use of this source code is governed by a BSD-style license that can be # found in the LICENSE file. """Miscellaneous tests.""" import collections import contextlib import io import json import os import pickle import socket import sys from unittest import mock import psutil import psutil.tests from psutil import WINDOWS from psutil._common import bcat from psutil._common import cat from psutil._common import debug from psutil._common import isfile_strict from psutil._common import memoize from psutil._common import memoize_when_activated from psutil._common import parse_environ_block from psutil._common import supports_ipv6 from psutil._common import wrap_numbers from psutil.tests import HAS_NET_IO_COUNTERS from psutil.tests import PsutilTestCase from psutil.tests import process_namespace from psutil.tests import pytest from psutil.tests import reload_module from psutil.tests import system_namespace # =================================================================== # --- Test classes' repr(), str(), ... # =================================================================== class TestSpecialMethods(PsutilTestCase): def test_check_pid_range(self): with pytest.raises(OverflowError): psutil._psplatform.cext.check_pid_range(2**128) with pytest.raises(psutil.NoSuchProcess): psutil.Process(2**128) def test_process__repr__(self, func=repr): p = psutil.Process(self.spawn_testproc().pid) r = func(p) assert "psutil.Process" in r assert f"pid={p.pid}" in r assert f"name='{p.name()}'" in r.replace("name=u'", "name='") assert "status=" in r assert "exitcode=" not in r p.terminate() p.wait() r = func(p) assert "status='terminated'" in r assert "exitcode=" in r with mock.patch.object( psutil.Process, "name", side_effect=psutil.ZombieProcess(os.getpid()), ): p = psutil.Process() r = func(p) assert f"pid={p.pid}" in r assert "status='zombie'" in r assert "name=" not in r with mock.patch.object( psutil.Process, "name", side_effect=psutil.NoSuchProcess(os.getpid()), ): p = psutil.Process() r = func(p) assert f"pid={p.pid}" in r assert "terminated" in r assert "name=" not in r with mock.patch.object( psutil.Process, "name", side_effect=psutil.AccessDenied(os.getpid()), ): p = psutil.Process() r = func(p) assert f"pid={p.pid}" in r assert "name=" not in r def test_process__str__(self): self.test_process__repr__(func=str) def test_error__repr__(self): assert repr(psutil.Error()) == "psutil.Error()" def test_error__str__(self): assert str(psutil.Error()) == "" def test_no_such_process__repr__(self): assert ( repr(psutil.NoSuchProcess(321)) == "psutil.NoSuchProcess(pid=321, msg='process no longer exists')" ) assert ( repr(psutil.NoSuchProcess(321, name="name", msg="msg")) == "psutil.NoSuchProcess(pid=321, name='name', msg='msg')" ) def test_no_such_process__str__(self): assert ( str(psutil.NoSuchProcess(321)) == "process no longer exists (pid=321)" ) assert ( str(psutil.NoSuchProcess(321, name="name", msg="msg")) == "msg (pid=321, name='name')" ) def test_zombie_process__repr__(self): assert ( repr(psutil.ZombieProcess(321)) == 'psutil.ZombieProcess(pid=321, msg="PID still ' 'exists but it\'s a zombie")' ) assert ( repr(psutil.ZombieProcess(321, name="name", ppid=320, msg="foo")) == "psutil.ZombieProcess(pid=321, ppid=320, name='name'," " msg='foo')" ) def test_zombie_process__str__(self): assert ( str(psutil.ZombieProcess(321)) == "PID still exists but it's a zombie (pid=321)" ) assert ( str(psutil.ZombieProcess(321, name="name", ppid=320, msg="foo")) == "foo (pid=321, ppid=320, name='name')" ) def test_access_denied__repr__(self): assert repr(psutil.AccessDenied(321)) == "psutil.AccessDenied(pid=321)" assert ( repr(psutil.AccessDenied(321, name="name", msg="msg")) == "psutil.AccessDenied(pid=321, name='name', msg='msg')" ) def test_access_denied__str__(self): assert str(psutil.AccessDenied(321)) == "(pid=321)" assert ( str(psutil.AccessDenied(321, name="name", msg="msg")) == "msg (pid=321, name='name')" ) def test_timeout_expired__repr__(self): assert ( repr(psutil.TimeoutExpired(5)) == "psutil.TimeoutExpired(seconds=5, msg='timeout after 5" " seconds')" ) assert ( repr(psutil.TimeoutExpired(5, pid=321, name="name")) == "psutil.TimeoutExpired(pid=321, name='name', seconds=5, " "msg='timeout after 5 seconds')" ) def test_timeout_expired__str__(self): assert str(psutil.TimeoutExpired(5)) == "timeout after 5 seconds" assert ( str(psutil.TimeoutExpired(5, pid=321, name="name")) == "timeout after 5 seconds (pid=321, name='name')" ) def test_process__eq__(self): p1 = psutil.Process() p2 = psutil.Process() assert p1 == p2 p2._ident = (0, 0) assert p1 != p2 assert p1 != 'foo' def test_process__hash__(self): s = {psutil.Process(), psutil.Process()} assert len(s) == 1 # =================================================================== # --- Misc, generic, corner cases # =================================================================== class TestMisc(PsutilTestCase): def test__all__(self): dir_psutil = dir(psutil) for name in dir_psutil: if name in { 'debug', 'tests', 'test', 'PermissionError', 'ProcessLookupError', }: continue if not name.startswith('_'): try: __import__(name) except ImportError: if name not in psutil.__all__: fun = getattr(psutil, name) if fun is None: continue if ( fun.__doc__ is not None and 'deprecated' not in fun.__doc__.lower() ): raise self.fail(f"{name!r} not in psutil.__all__") # Import 'star' will break if __all__ is inconsistent, see: # https://github.com/giampaolo/psutil/issues/656 # Can't do `from psutil import *` as it won't work # so we simply iterate over __all__. for name in psutil.__all__: assert name in dir_psutil def test_version(self): assert ( '.'.join([str(x) for x in psutil.version_info]) == psutil.__version__ ) def test_process_as_dict_no_new_names(self): # See https://github.com/giampaolo/psutil/issues/813 p = psutil.Process() p.foo = '1' assert 'foo' not in p.as_dict() def test_serialization(self): def check(ret): json.loads(json.dumps(ret)) a = pickle.dumps(ret) b = pickle.loads(a) assert ret == b # --- process APIs proc = psutil.Process() check(psutil.Process().as_dict()) ns = process_namespace(proc) for fun, name in ns.iter(ns.getters, clear_cache=True): with self.subTest(proc=proc, name=name): try: ret = fun() except psutil.Error: pass else: check(ret) # --- system APIs ns = system_namespace() for fun, name in ns.iter(ns.getters): if name in {"win_service_iter", "win_service_get"}: continue with self.subTest(name=name): try: ret = fun() except psutil.AccessDenied: pass else: check(ret) # --- exception classes b = pickle.loads( pickle.dumps( psutil.NoSuchProcess(pid=4567, name='name', msg='msg') ) ) assert isinstance(b, psutil.NoSuchProcess) assert b.pid == 4567 assert b.name == 'name' assert b.msg == 'msg' b = pickle.loads( pickle.dumps( psutil.ZombieProcess(pid=4567, name='name', ppid=42, msg='msg') ) ) assert isinstance(b, psutil.ZombieProcess) assert b.pid == 4567 assert b.ppid == 42 assert b.name == 'name' assert b.msg == 'msg' b = pickle.loads( pickle.dumps(psutil.AccessDenied(pid=123, name='name', msg='msg')) ) assert isinstance(b, psutil.AccessDenied) assert b.pid == 123 assert b.name == 'name' assert b.msg == 'msg' b = pickle.loads( pickle.dumps( psutil.TimeoutExpired(seconds=33, pid=4567, name='name') ) ) assert isinstance(b, psutil.TimeoutExpired) assert b.seconds == 33 assert b.pid == 4567 assert b.name == 'name' def test_ad_on_process_creation(self): # We are supposed to be able to instantiate Process also in case # of zombie processes or access denied. with mock.patch.object( psutil.Process, '_get_ident', side_effect=psutil.AccessDenied ) as meth: psutil.Process() assert meth.called with mock.patch.object( psutil.Process, '_get_ident', side_effect=psutil.ZombieProcess(1) ) as meth: psutil.Process() assert meth.called with mock.patch.object( psutil.Process, '_get_ident', side_effect=ValueError ) as meth: with pytest.raises(ValueError): psutil.Process() assert meth.called with mock.patch.object( psutil.Process, '_get_ident', side_effect=psutil.NoSuchProcess(1) ) as meth: with self.assertRaises(psutil.NoSuchProcess): psutil.Process() assert meth.called def test_sanity_version_check(self): # see: https://github.com/giampaolo/psutil/issues/564 with mock.patch( "psutil._psplatform.cext.version", return_value="0.0.0" ): with pytest.raises(ImportError) as cm: reload_module(psutil) assert "version conflict" in str(cm.value).lower() # =================================================================== # --- psutil/_common.py utils # =================================================================== class TestMemoizeDecorator(PsutilTestCase): def setUp(self): self.calls = [] tearDown = setUp def run_against(self, obj, expected_retval=None): # no args for _ in range(2): ret = obj() assert self.calls == [((), {})] if expected_retval is not None: assert ret == expected_retval # with args for _ in range(2): ret = obj(1) assert self.calls == [((), {}), ((1,), {})] if expected_retval is not None: assert ret == expected_retval # with args + kwargs for _ in range(2): ret = obj(1, bar=2) assert self.calls == [((), {}), ((1,), {}), ((1,), {'bar': 2})] if expected_retval is not None: assert ret == expected_retval # clear cache assert len(self.calls) == 3 obj.cache_clear() ret = obj() if expected_retval is not None: assert ret == expected_retval assert len(self.calls) == 4 # docstring assert obj.__doc__ == "My docstring." def test_function(self): @memoize def foo(*args, **kwargs): """My docstring.""" baseclass.calls.append((args, kwargs)) return 22 baseclass = self self.run_against(foo, expected_retval=22) def test_class(self): @memoize class Foo: """My docstring.""" def __init__(self, *args, **kwargs): baseclass.calls.append((args, kwargs)) def bar(self): return 22 baseclass = self self.run_against(Foo, expected_retval=None) assert Foo().bar() == 22 def test_class_singleton(self): # @memoize can be used against classes to create singletons @memoize class Bar: def __init__(self, *args, **kwargs): pass assert Bar() is Bar() assert id(Bar()) == id(Bar()) assert id(Bar(1)) == id(Bar(1)) assert id(Bar(1, foo=3)) == id(Bar(1, foo=3)) assert id(Bar(1)) != id(Bar(2)) def test_staticmethod(self): class Foo: @staticmethod @memoize def bar(*args, **kwargs): """My docstring.""" baseclass.calls.append((args, kwargs)) return 22 baseclass = self self.run_against(Foo().bar, expected_retval=22) def test_classmethod(self): class Foo: @classmethod @memoize def bar(cls, *args, **kwargs): """My docstring.""" baseclass.calls.append((args, kwargs)) return 22 baseclass = self self.run_against(Foo().bar, expected_retval=22) def test_original(self): # This was the original test before I made it dynamic to test it # against different types. Keeping it anyway. @memoize def foo(*args, **kwargs): """Foo docstring.""" calls.append(None) return (args, kwargs) calls = [] # no args for _ in range(2): ret = foo() expected = ((), {}) assert ret == expected assert len(calls) == 1 # with args for _ in range(2): ret = foo(1) expected = ((1,), {}) assert ret == expected assert len(calls) == 2 # with args + kwargs for _ in range(2): ret = foo(1, bar=2) expected = ((1,), {'bar': 2}) assert ret == expected assert len(calls) == 3 # clear cache foo.cache_clear() ret = foo() expected = ((), {}) assert ret == expected assert len(calls) == 4 # docstring assert foo.__doc__ == "Foo docstring." class TestCommonModule(PsutilTestCase): def test_memoize_when_activated(self): class Foo: @memoize_when_activated def foo(self): calls.append(None) f = Foo() calls = [] f.foo() f.foo() assert len(calls) == 2 # activate calls = [] f.foo.cache_activate(f) f.foo() f.foo() assert len(calls) == 1 # deactivate calls = [] f.foo.cache_deactivate(f) f.foo() f.foo() assert len(calls) == 2 def test_parse_environ_block(self): def k(s): return s.upper() if WINDOWS else s assert parse_environ_block("a=1\0") == {k("a"): "1"} assert parse_environ_block("a=1\0b=2\0\0") == { k("a"): "1", k("b"): "2", } assert parse_environ_block("a=1\0b=\0\0") == {k("a"): "1", k("b"): ""} # ignore everything after \0\0 assert parse_environ_block("a=1\0b=2\0\0c=3\0") == { k("a"): "1", k("b"): "2", } # ignore everything that is not an assignment assert parse_environ_block("xxx\0a=1\0") == {k("a"): "1"} assert parse_environ_block("a=1\0=b=2\0") == {k("a"): "1"} # do not fail if the block is incomplete assert parse_environ_block("a=1\0b=2") == {k("a"): "1"} def test_supports_ipv6(self): self.addCleanup(supports_ipv6.cache_clear) if supports_ipv6(): with mock.patch('psutil._common.socket') as s: s.has_ipv6 = False supports_ipv6.cache_clear() assert not supports_ipv6() supports_ipv6.cache_clear() with mock.patch( 'psutil._common.socket.socket', side_effect=OSError ) as s: assert not supports_ipv6() assert s.called supports_ipv6.cache_clear() with mock.patch( 'psutil._common.socket.socket', side_effect=socket.gaierror ) as s: assert not supports_ipv6() supports_ipv6.cache_clear() assert s.called supports_ipv6.cache_clear() with mock.patch( 'psutil._common.socket.socket.bind', side_effect=socket.gaierror, ) as s: assert not supports_ipv6() supports_ipv6.cache_clear() assert s.called else: with pytest.raises(OSError): sock = socket.socket(socket.AF_INET6, socket.SOCK_STREAM) try: sock.bind(("::1", 0)) finally: sock.close() def test_isfile_strict(self): this_file = os.path.abspath(__file__) assert isfile_strict(this_file) assert not isfile_strict(os.path.dirname(this_file)) with mock.patch('psutil._common.os.stat', side_effect=PermissionError): with pytest.raises(OSError): isfile_strict(this_file) with mock.patch( 'psutil._common.os.stat', side_effect=FileNotFoundError ): assert not isfile_strict(this_file) with mock.patch('psutil._common.stat.S_ISREG', return_value=False): assert not isfile_strict(this_file) def test_debug(self): with mock.patch.object(psutil._common, "PSUTIL_DEBUG", True): with contextlib.redirect_stderr(io.StringIO()) as f: debug("hello") sys.stderr.flush() msg = f.getvalue() assert msg.startswith("psutil-debug"), msg assert "hello" in msg assert __file__.replace('.pyc', '.py') in msg # supposed to use repr(exc) with mock.patch.object(psutil._common, "PSUTIL_DEBUG", True): with contextlib.redirect_stderr(io.StringIO()) as f: debug(ValueError("this is an error")) msg = f.getvalue() assert "ignoring ValueError" in msg assert "'this is an error'" in msg # supposed to use str(exc), because of extra info about file name with mock.patch.object(psutil._common, "PSUTIL_DEBUG", True): with contextlib.redirect_stderr(io.StringIO()) as f: exc = OSError(2, "no such file") exc.filename = "/foo" debug(exc) msg = f.getvalue() assert "no such file" in msg assert "/foo" in msg def test_cat_bcat(self): testfn = self.get_testfn() with open(testfn, "w") as f: f.write("foo") assert cat(testfn) == "foo" assert bcat(testfn) == b"foo" with pytest.raises(FileNotFoundError): cat(testfn + '-invalid') with pytest.raises(FileNotFoundError): bcat(testfn + '-invalid') assert cat(testfn + '-invalid', fallback="bar") == "bar" assert bcat(testfn + '-invalid', fallback="bar") == "bar" # =================================================================== # --- Tests for wrap_numbers() function. # =================================================================== nt = collections.namedtuple('foo', 'a b c') class TestWrapNumbers(PsutilTestCase): def setUp(self): wrap_numbers.cache_clear() tearDown = setUp def test_first_call(self): input = {'disk1': nt(5, 5, 5)} assert wrap_numbers(input, 'disk_io') == input def test_input_hasnt_changed(self): input = {'disk1': nt(5, 5, 5)} assert wrap_numbers(input, 'disk_io') == input assert wrap_numbers(input, 'disk_io') == input def test_increase_but_no_wrap(self): input = {'disk1': nt(5, 5, 5)} assert wrap_numbers(input, 'disk_io') == input input = {'disk1': nt(10, 15, 20)} assert wrap_numbers(input, 'disk_io') == input input = {'disk1': nt(20, 25, 30)} assert wrap_numbers(input, 'disk_io') == input input = {'disk1': nt(20, 25, 30)} assert wrap_numbers(input, 'disk_io') == input def test_wrap(self): # let's say 100 is the threshold input = {'disk1': nt(100, 100, 100)} assert wrap_numbers(input, 'disk_io') == input # first wrap restarts from 10 input = {'disk1': nt(100, 100, 10)} assert wrap_numbers(input, 'disk_io') == {'disk1': nt(100, 100, 110)} # then it remains the same input = {'disk1': nt(100, 100, 10)} assert wrap_numbers(input, 'disk_io') == {'disk1': nt(100, 100, 110)} # then it goes up input = {'disk1': nt(100, 100, 90)} assert wrap_numbers(input, 'disk_io') == {'disk1': nt(100, 100, 190)} # then it wraps again input = {'disk1': nt(100, 100, 20)} assert wrap_numbers(input, 'disk_io') == {'disk1': nt(100, 100, 210)} # and remains the same input = {'disk1': nt(100, 100, 20)} assert wrap_numbers(input, 'disk_io') == {'disk1': nt(100, 100, 210)} # now wrap another num input = {'disk1': nt(50, 100, 20)} assert wrap_numbers(input, 'disk_io') == {'disk1': nt(150, 100, 210)} # and again input = {'disk1': nt(40, 100, 20)} assert wrap_numbers(input, 'disk_io') == {'disk1': nt(190, 100, 210)} # keep it the same input = {'disk1': nt(40, 100, 20)} assert wrap_numbers(input, 'disk_io') == {'disk1': nt(190, 100, 210)} def test_changing_keys(self): # Emulate a case where the second call to disk_io() # (or whatever) provides a new disk, then the new disk # disappears on the third call. input = {'disk1': nt(5, 5, 5)} assert wrap_numbers(input, 'disk_io') == input input = {'disk1': nt(5, 5, 5), 'disk2': nt(7, 7, 7)} assert wrap_numbers(input, 'disk_io') == input input = {'disk1': nt(8, 8, 8)} assert wrap_numbers(input, 'disk_io') == input def test_changing_keys_w_wrap(self): input = {'disk1': nt(50, 50, 50), 'disk2': nt(100, 100, 100)} assert wrap_numbers(input, 'disk_io') == input # disk 2 wraps input = {'disk1': nt(50, 50, 50), 'disk2': nt(100, 100, 10)} assert wrap_numbers(input, 'disk_io') == { 'disk1': nt(50, 50, 50), 'disk2': nt(100, 100, 110), } # disk 2 disappears input = {'disk1': nt(50, 50, 50)} assert wrap_numbers(input, 'disk_io') == input # then it appears again; the old wrap is supposed to be # gone. input = {'disk1': nt(50, 50, 50), 'disk2': nt(100, 100, 100)} assert wrap_numbers(input, 'disk_io') == input # remains the same input = {'disk1': nt(50, 50, 50), 'disk2': nt(100, 100, 100)} assert wrap_numbers(input, 'disk_io') == input # and then wraps again input = {'disk1': nt(50, 50, 50), 'disk2': nt(100, 100, 10)} assert wrap_numbers(input, 'disk_io') == { 'disk1': nt(50, 50, 50), 'disk2': nt(100, 100, 110), } def test_real_data(self): d = { 'nvme0n1': (300, 508, 640, 1571, 5970, 1987, 2049, 451751, 47048), 'nvme0n1p1': (1171, 2, 5600256, 1024, 516, 0, 0, 0, 8), 'nvme0n1p2': (54, 54, 2396160, 5165056, 4, 24, 30, 1207, 28), 'nvme0n1p3': (2389, 4539, 5154, 150, 4828, 1844, 2019, 398, 348), } assert wrap_numbers(d, 'disk_io') == d assert wrap_numbers(d, 'disk_io') == d # decrease this ↓ d = { 'nvme0n1': (100, 508, 640, 1571, 5970, 1987, 2049, 451751, 47048), 'nvme0n1p1': (1171, 2, 5600256, 1024, 516, 0, 0, 0, 8), 'nvme0n1p2': (54, 54, 2396160, 5165056, 4, 24, 30, 1207, 28), 'nvme0n1p3': (2389, 4539, 5154, 150, 4828, 1844, 2019, 398, 348), } out = wrap_numbers(d, 'disk_io') assert out['nvme0n1'][0] == 400 # --- cache tests def test_cache_first_call(self): input = {'disk1': nt(5, 5, 5)} wrap_numbers(input, 'disk_io') cache = wrap_numbers.cache_info() assert cache[0] == {'disk_io': input} assert cache[1] == {'disk_io': {}} assert cache[2] == {'disk_io': {}} def test_cache_call_twice(self): input = {'disk1': nt(5, 5, 5)} wrap_numbers(input, 'disk_io') input = {'disk1': nt(10, 10, 10)} wrap_numbers(input, 'disk_io') cache = wrap_numbers.cache_info() assert cache[0] == {'disk_io': input} assert cache[1] == { 'disk_io': {('disk1', 0): 0, ('disk1', 1): 0, ('disk1', 2): 0} } assert cache[2] == {'disk_io': {}} def test_cache_wrap(self): # let's say 100 is the threshold input = {'disk1': nt(100, 100, 100)} wrap_numbers(input, 'disk_io') # first wrap restarts from 10 input = {'disk1': nt(100, 100, 10)} wrap_numbers(input, 'disk_io') cache = wrap_numbers.cache_info() assert cache[0] == {'disk_io': input} assert cache[1] == { 'disk_io': {('disk1', 0): 0, ('disk1', 1): 0, ('disk1', 2): 100} } assert cache[2] == {'disk_io': {'disk1': {('disk1', 2)}}} def check_cache_info(): cache = wrap_numbers.cache_info() assert cache[1] == { 'disk_io': { ('disk1', 0): 0, ('disk1', 1): 0, ('disk1', 2): 100, } } assert cache[2] == {'disk_io': {'disk1': {('disk1', 2)}}} # then it remains the same input = {'disk1': nt(100, 100, 10)} wrap_numbers(input, 'disk_io') cache = wrap_numbers.cache_info() assert cache[0] == {'disk_io': input} check_cache_info() # then it goes up input = {'disk1': nt(100, 100, 90)} wrap_numbers(input, 'disk_io') cache = wrap_numbers.cache_info() assert cache[0] == {'disk_io': input} check_cache_info() # then it wraps again input = {'disk1': nt(100, 100, 20)} wrap_numbers(input, 'disk_io') cache = wrap_numbers.cache_info() assert cache[0] == {'disk_io': input} assert cache[1] == { 'disk_io': {('disk1', 0): 0, ('disk1', 1): 0, ('disk1', 2): 190} } assert cache[2] == {'disk_io': {'disk1': {('disk1', 2)}}} def test_cache_changing_keys(self): input = {'disk1': nt(5, 5, 5)} wrap_numbers(input, 'disk_io') input = {'disk1': nt(5, 5, 5), 'disk2': nt(7, 7, 7)} wrap_numbers(input, 'disk_io') cache = wrap_numbers.cache_info() assert cache[0] == {'disk_io': input} assert cache[1] == { 'disk_io': {('disk1', 0): 0, ('disk1', 1): 0, ('disk1', 2): 0} } assert cache[2] == {'disk_io': {}} def test_cache_clear(self): input = {'disk1': nt(5, 5, 5)} wrap_numbers(input, 'disk_io') wrap_numbers(input, 'disk_io') wrap_numbers.cache_clear('disk_io') assert wrap_numbers.cache_info() == ({}, {}, {}) wrap_numbers.cache_clear('disk_io') wrap_numbers.cache_clear('?!?') @pytest.mark.skipif(not HAS_NET_IO_COUNTERS, reason="not supported") def test_cache_clear_public_apis(self): if not psutil.disk_io_counters() or not psutil.net_io_counters(): raise pytest.skip("no disks or NICs available") psutil.disk_io_counters() psutil.net_io_counters() caches = wrap_numbers.cache_info() for cache in caches: assert 'psutil.disk_io_counters' in cache assert 'psutil.net_io_counters' in cache psutil.disk_io_counters.cache_clear() caches = wrap_numbers.cache_info() for cache in caches: assert 'psutil.net_io_counters' in cache assert 'psutil.disk_io_counters' not in cache psutil.net_io_counters.cache_clear() caches = wrap_numbers.cache_info() assert caches == ({}, {}, {}) tests/test_scripts.py 0000644 00000017055 15030211765 0011016 0 ustar 00 #!/usr/bin/env python3 # Copyright (c) 2009, Giampaolo Rodola'. All rights reserved. # Use of this source code is governed by a BSD-style license that can be # found in the LICENSE file. """Test various scripts.""" import ast import os import shutil import stat import subprocess import pytest from psutil import POSIX from psutil import WINDOWS from psutil.tests import CI_TESTING from psutil.tests import HAS_BATTERY from psutil.tests import HAS_MEMORY_MAPS from psutil.tests import HAS_SENSORS_BATTERY from psutil.tests import HAS_SENSORS_FANS from psutil.tests import HAS_SENSORS_TEMPERATURES from psutil.tests import PYTHON_EXE from psutil.tests import PYTHON_EXE_ENV from psutil.tests import ROOT_DIR from psutil.tests import SCRIPTS_DIR from psutil.tests import PsutilTestCase from psutil.tests import import_module_by_path from psutil.tests import psutil from psutil.tests import sh INTERNAL_SCRIPTS_DIR = os.path.join(SCRIPTS_DIR, "internal") SETUP_PY = os.path.join(ROOT_DIR, 'setup.py') # =================================================================== # --- Tests scripts in scripts/ directory # =================================================================== @pytest.mark.skipif( CI_TESTING and not os.path.exists(SCRIPTS_DIR), reason="can't find scripts/ directory", ) class TestExampleScripts(PsutilTestCase): @staticmethod def assert_stdout(exe, *args, **kwargs): kwargs.setdefault("env", PYTHON_EXE_ENV) exe = os.path.join(SCRIPTS_DIR, exe) cmd = [PYTHON_EXE, exe] for arg in args: cmd.append(arg) try: out = sh(cmd, **kwargs).strip() except RuntimeError as err: if 'AccessDenied' in str(err): return str(err) else: raise assert out, out return out @staticmethod def assert_syntax(exe): exe = os.path.join(SCRIPTS_DIR, exe) with open(exe, encoding="utf8") as f: src = f.read() ast.parse(src) def test_coverage(self): # make sure all example scripts have a test method defined meths = dir(self) for name in os.listdir(SCRIPTS_DIR): if name.endswith('.py'): if 'test_' + os.path.splitext(name)[0] not in meths: # self.assert_stdout(name) raise self.fail( "no test defined for" f" {os.path.join(SCRIPTS_DIR, name)!r} script" ) @pytest.mark.skipif(not POSIX, reason="POSIX only") def test_executable(self): for root, dirs, files in os.walk(SCRIPTS_DIR): for file in files: if file.endswith('.py'): path = os.path.join(root, file) if not stat.S_IXUSR & os.stat(path)[stat.ST_MODE]: raise self.fail(f"{path!r} is not executable") def test_disk_usage(self): self.assert_stdout('disk_usage.py') def test_free(self): self.assert_stdout('free.py') def test_meminfo(self): self.assert_stdout('meminfo.py') def test_procinfo(self): self.assert_stdout('procinfo.py', str(os.getpid())) @pytest.mark.skipif(CI_TESTING and not psutil.users(), reason="no users") def test_who(self): self.assert_stdout('who.py') def test_ps(self): self.assert_stdout('ps.py') def test_pstree(self): self.assert_stdout('pstree.py') def test_netstat(self): self.assert_stdout('netstat.py') def test_ifconfig(self): self.assert_stdout('ifconfig.py') @pytest.mark.skipif(not HAS_MEMORY_MAPS, reason="not supported") def test_pmap(self): self.assert_stdout('pmap.py', str(os.getpid())) def test_procsmem(self): if 'uss' not in psutil.Process().memory_full_info()._fields: raise pytest.skip("not supported") self.assert_stdout('procsmem.py') def test_killall(self): self.assert_syntax('killall.py') def test_nettop(self): self.assert_syntax('nettop.py') def test_top(self): self.assert_syntax('top.py') def test_iotop(self): self.assert_syntax('iotop.py') def test_pidof(self): output = self.assert_stdout('pidof.py', psutil.Process().name()) assert str(os.getpid()) in output @pytest.mark.skipif(not WINDOWS, reason="WINDOWS only") def test_winservices(self): self.assert_stdout('winservices.py') def test_cpu_distribution(self): self.assert_syntax('cpu_distribution.py') @pytest.mark.skipif(not HAS_SENSORS_TEMPERATURES, reason="not supported") def test_temperatures(self): if not psutil.sensors_temperatures(): raise pytest.skip("no temperatures") self.assert_stdout('temperatures.py') @pytest.mark.skipif(not HAS_SENSORS_FANS, reason="not supported") def test_fans(self): if not psutil.sensors_fans(): raise pytest.skip("no fans") self.assert_stdout('fans.py') @pytest.mark.skipif(not HAS_SENSORS_BATTERY, reason="not supported") @pytest.mark.skipif(not HAS_BATTERY, reason="no battery") def test_battery(self): self.assert_stdout('battery.py') @pytest.mark.skipif(not HAS_SENSORS_BATTERY, reason="not supported") @pytest.mark.skipif(not HAS_BATTERY, reason="no battery") def test_sensors(self): self.assert_stdout('sensors.py') # =================================================================== # --- Tests scripts in scripts/internal/ directory # =================================================================== @pytest.mark.skipif( CI_TESTING and not os.path.exists(INTERNAL_SCRIPTS_DIR), reason="can't find scripts/internal/ directory", ) class TestInternalScripts(PsutilTestCase): @staticmethod def ls(): for name in os.listdir(INTERNAL_SCRIPTS_DIR): if name.endswith(".py"): yield os.path.join(INTERNAL_SCRIPTS_DIR, name) def test_syntax_all(self): for path in self.ls(): with open(path, encoding="utf8") as f: data = f.read() ast.parse(data) @pytest.mark.skipif(CI_TESTING, reason="not on CI") def test_import_all(self): for path in self.ls(): try: import_module_by_path(path) except SystemExit: pass # =================================================================== # --- Tests for setup.py script # =================================================================== @pytest.mark.skipif( CI_TESTING and not os.path.exists(SETUP_PY), reason="can't find setup.py" ) class TestSetupScript(PsutilTestCase): def test_invocation(self): module = import_module_by_path(SETUP_PY) with pytest.raises(SystemExit): module.setup() assert module.get_version() == psutil.__version__ @pytest.mark.skipif( not shutil.which("python2.7"), reason="python2.7 not installed" ) def test_python2(self): # There's a duplicate of this test in scripts/internal # directory, which is only executed by CI. We replicate it here # to run it when developing locally. p = subprocess.Popen( [shutil.which("python2.7"), SETUP_PY], stdout=subprocess.PIPE, stderr=subprocess.PIPE, universal_newlines=True, ) stdout, stderr = p.communicate() assert p.wait() == 1 assert not stdout assert "psutil no longer supports Python 2.7" in stderr assert "Latest version supporting Python 2.7 is" in stderr tests/test_memleaks.py 0000644 00000035421 15030211765 0011122 0 ustar 00 #!/usr/bin/env python3 # Copyright (c) 2009, Giampaolo Rodola'. All rights reserved. # Use of this source code is governed by a BSD-style license that can be # found in the LICENSE file. """Tests for detecting function memory leaks (typically the ones implemented in C). It does so by calling a function many times and checking whether process memory usage keeps increasing between calls or over time. Note that this may produce false positives (especially on Windows for some reason). PyPy appears to be completely unstable for this framework, probably because of how its JIT handles memory, so tests are skipped. """ import functools import os import platform import psutil import psutil._common from psutil import LINUX from psutil import MACOS from psutil import OPENBSD from psutil import POSIX from psutil import SUNOS from psutil import WINDOWS from psutil.tests import HAS_CPU_AFFINITY from psutil.tests import HAS_CPU_FREQ from psutil.tests import HAS_ENVIRON from psutil.tests import HAS_IONICE from psutil.tests import HAS_MEMORY_MAPS from psutil.tests import HAS_NET_IO_COUNTERS from psutil.tests import HAS_PROC_CPU_NUM from psutil.tests import HAS_PROC_IO_COUNTERS from psutil.tests import HAS_RLIMIT from psutil.tests import HAS_SENSORS_BATTERY from psutil.tests import HAS_SENSORS_FANS from psutil.tests import HAS_SENSORS_TEMPERATURES from psutil.tests import TestMemoryLeak from psutil.tests import create_sockets from psutil.tests import get_testfn from psutil.tests import process_namespace from psutil.tests import pytest from psutil.tests import skip_on_access_denied from psutil.tests import spawn_testproc from psutil.tests import system_namespace from psutil.tests import terminate cext = psutil._psplatform.cext thisproc = psutil.Process() FEW_TIMES = 5 def fewtimes_if_linux(): """Decorator for those Linux functions which are implemented in pure Python, and which we want to run faster. """ def decorator(fun): @functools.wraps(fun) def wrapper(self, *args, **kwargs): if LINUX: before = self.__class__.times try: self.__class__.times = FEW_TIMES return fun(self, *args, **kwargs) finally: self.__class__.times = before else: return fun(self, *args, **kwargs) return wrapper return decorator # =================================================================== # Process class # =================================================================== class TestProcessObjectLeaks(TestMemoryLeak): """Test leaks of Process class methods.""" proc = thisproc def test_coverage(self): ns = process_namespace(None) ns.test_class_coverage(self, ns.getters + ns.setters) @fewtimes_if_linux() def test_name(self): self.execute(self.proc.name) @fewtimes_if_linux() def test_cmdline(self): self.execute(self.proc.cmdline) @fewtimes_if_linux() def test_exe(self): self.execute(self.proc.exe) @fewtimes_if_linux() def test_ppid(self): self.execute(self.proc.ppid) @pytest.mark.skipif(not POSIX, reason="POSIX only") @fewtimes_if_linux() def test_uids(self): self.execute(self.proc.uids) @pytest.mark.skipif(not POSIX, reason="POSIX only") @fewtimes_if_linux() def test_gids(self): self.execute(self.proc.gids) @fewtimes_if_linux() def test_status(self): self.execute(self.proc.status) def test_nice(self): self.execute(self.proc.nice) def test_nice_set(self): niceness = thisproc.nice() self.execute(lambda: self.proc.nice(niceness)) @pytest.mark.skipif(not HAS_IONICE, reason="not supported") def test_ionice(self): self.execute(self.proc.ionice) @pytest.mark.skipif(not HAS_IONICE, reason="not supported") def test_ionice_set(self): if WINDOWS: value = thisproc.ionice() self.execute(lambda: self.proc.ionice(value)) else: self.execute(lambda: self.proc.ionice(psutil.IOPRIO_CLASS_NONE)) fun = functools.partial(cext.proc_ioprio_set, os.getpid(), -1, 0) self.execute_w_exc(OSError, fun) @pytest.mark.skipif(not HAS_PROC_IO_COUNTERS, reason="not supported") @fewtimes_if_linux() def test_io_counters(self): self.execute(self.proc.io_counters) @pytest.mark.skipif(POSIX, reason="worthless on POSIX") def test_username(self): # always open 1 handle on Windows (only once) psutil.Process().username() self.execute(self.proc.username) @fewtimes_if_linux() def test_create_time(self): self.execute(self.proc.create_time) @fewtimes_if_linux() @skip_on_access_denied(only_if=OPENBSD) def test_num_threads(self): self.execute(self.proc.num_threads) @pytest.mark.skipif(not WINDOWS, reason="WINDOWS only") def test_num_handles(self): self.execute(self.proc.num_handles) @pytest.mark.skipif(not POSIX, reason="POSIX only") @fewtimes_if_linux() def test_num_fds(self): self.execute(self.proc.num_fds) @fewtimes_if_linux() def test_num_ctx_switches(self): self.execute(self.proc.num_ctx_switches) @fewtimes_if_linux() @skip_on_access_denied(only_if=OPENBSD) def test_threads(self): self.execute(self.proc.threads) @fewtimes_if_linux() def test_cpu_times(self): self.execute(self.proc.cpu_times) @fewtimes_if_linux() @pytest.mark.skipif(not HAS_PROC_CPU_NUM, reason="not supported") def test_cpu_num(self): self.execute(self.proc.cpu_num) @fewtimes_if_linux() def test_memory_info(self): self.execute(self.proc.memory_info) @fewtimes_if_linux() def test_memory_full_info(self): self.execute(self.proc.memory_full_info) @pytest.mark.skipif(not POSIX, reason="POSIX only") @fewtimes_if_linux() def test_terminal(self): self.execute(self.proc.terminal) def test_resume(self): times = FEW_TIMES if POSIX else self.times self.execute(self.proc.resume, times=times) @fewtimes_if_linux() def test_cwd(self): self.execute(self.proc.cwd) @pytest.mark.skipif(not HAS_CPU_AFFINITY, reason="not supported") def test_cpu_affinity(self): self.execute(self.proc.cpu_affinity) @pytest.mark.skipif(not HAS_CPU_AFFINITY, reason="not supported") def test_cpu_affinity_set(self): affinity = thisproc.cpu_affinity() self.execute(lambda: self.proc.cpu_affinity(affinity)) self.execute_w_exc(ValueError, lambda: self.proc.cpu_affinity([-1])) @fewtimes_if_linux() def test_open_files(self): with open(get_testfn(), 'w'): self.execute(self.proc.open_files) @pytest.mark.skipif(not HAS_MEMORY_MAPS, reason="not supported") @fewtimes_if_linux() def test_memory_maps(self): self.execute(self.proc.memory_maps) @pytest.mark.skipif(not LINUX, reason="LINUX only") @pytest.mark.skipif(not HAS_RLIMIT, reason="not supported") def test_rlimit(self): self.execute(lambda: self.proc.rlimit(psutil.RLIMIT_NOFILE)) @pytest.mark.skipif(not LINUX, reason="LINUX only") @pytest.mark.skipif(not HAS_RLIMIT, reason="not supported") def test_rlimit_set(self): limit = thisproc.rlimit(psutil.RLIMIT_NOFILE) self.execute(lambda: self.proc.rlimit(psutil.RLIMIT_NOFILE, limit)) self.execute_w_exc((OSError, ValueError), lambda: self.proc.rlimit(-1)) @fewtimes_if_linux() # Windows implementation is based on a single system-wide # function (tested later). @pytest.mark.skipif(WINDOWS, reason="worthless on WINDOWS") def test_net_connections(self): # TODO: UNIX sockets are temporarily implemented by parsing # 'pfiles' cmd output; we don't want that part of the code to # be executed. with create_sockets(): kind = 'inet' if SUNOS else 'all' self.execute(lambda: self.proc.net_connections(kind)) @pytest.mark.skipif(not HAS_ENVIRON, reason="not supported") def test_environ(self): self.execute(self.proc.environ) @pytest.mark.skipif(not WINDOWS, reason="WINDOWS only") def test_proc_info(self): self.execute(lambda: cext.proc_info(os.getpid())) class TestTerminatedProcessLeaks(TestProcessObjectLeaks): """Repeat the tests above looking for leaks occurring when dealing with terminated processes raising NoSuchProcess exception. The C functions are still invoked but will follow different code paths. We'll check those code paths. """ @classmethod def setUpClass(cls): super().setUpClass() cls.subp = spawn_testproc() cls.proc = psutil.Process(cls.subp.pid) cls.proc.kill() cls.proc.wait() @classmethod def tearDownClass(cls): super().tearDownClass() terminate(cls.subp) def call(self, fun): try: fun() except psutil.NoSuchProcess: pass if WINDOWS: def test_kill(self): self.execute(self.proc.kill) def test_terminate(self): self.execute(self.proc.terminate) def test_suspend(self): self.execute(self.proc.suspend) def test_resume(self): self.execute(self.proc.resume) def test_wait(self): self.execute(self.proc.wait) def test_proc_info(self): # test dual implementation def call(): try: return cext.proc_info(self.proc.pid) except ProcessLookupError: pass self.execute(call) @pytest.mark.skipif(not WINDOWS, reason="WINDOWS only") class TestProcessDualImplementation(TestMemoryLeak): def test_cmdline_peb_true(self): self.execute(lambda: cext.proc_cmdline(os.getpid(), use_peb=True)) def test_cmdline_peb_false(self): self.execute(lambda: cext.proc_cmdline(os.getpid(), use_peb=False)) # =================================================================== # system APIs # =================================================================== class TestModuleFunctionsLeaks(TestMemoryLeak): """Test leaks of psutil module functions.""" def test_coverage(self): ns = system_namespace() ns.test_class_coverage(self, ns.all) # --- cpu @fewtimes_if_linux() def test_cpu_count(self): # logical self.execute(lambda: psutil.cpu_count(logical=True)) @fewtimes_if_linux() def test_cpu_count_cores(self): self.execute(lambda: psutil.cpu_count(logical=False)) @fewtimes_if_linux() def test_cpu_times(self): self.execute(psutil.cpu_times) @fewtimes_if_linux() def test_per_cpu_times(self): self.execute(lambda: psutil.cpu_times(percpu=True)) @fewtimes_if_linux() def test_cpu_stats(self): self.execute(psutil.cpu_stats) @fewtimes_if_linux() # TODO: remove this once 1892 is fixed @pytest.mark.skipif( MACOS and platform.machine() == 'arm64', reason="skipped due to #1892" ) @pytest.mark.skipif(not HAS_CPU_FREQ, reason="not supported") def test_cpu_freq(self): self.execute(psutil.cpu_freq) @pytest.mark.skipif(not WINDOWS, reason="WINDOWS only") def test_getloadavg(self): psutil.getloadavg() self.execute(psutil.getloadavg) # --- mem def test_virtual_memory(self): self.execute(psutil.virtual_memory) # TODO: remove this skip when this gets fixed @pytest.mark.skipif(SUNOS, reason="worthless on SUNOS (uses a subprocess)") def test_swap_memory(self): self.execute(psutil.swap_memory) def test_pid_exists(self): times = FEW_TIMES if POSIX else self.times self.execute(lambda: psutil.pid_exists(os.getpid()), times=times) # --- disk def test_disk_usage(self): times = FEW_TIMES if POSIX else self.times self.execute(lambda: psutil.disk_usage('.'), times=times) def test_disk_partitions(self): self.execute(psutil.disk_partitions) @pytest.mark.skipif( LINUX and not os.path.exists('/proc/diskstats'), reason="/proc/diskstats not available on this Linux version", ) @fewtimes_if_linux() def test_disk_io_counters(self): self.execute(lambda: psutil.disk_io_counters(nowrap=False)) # --- proc @fewtimes_if_linux() def test_pids(self): self.execute(psutil.pids) # --- net @fewtimes_if_linux() @pytest.mark.skipif(not HAS_NET_IO_COUNTERS, reason="not supported") def test_net_io_counters(self): self.execute(lambda: psutil.net_io_counters(nowrap=False)) @fewtimes_if_linux() @pytest.mark.skipif(MACOS and os.getuid() != 0, reason="need root access") def test_net_connections(self): # always opens and handle on Windows() (once) psutil.net_connections(kind='all') with create_sockets(): self.execute(lambda: psutil.net_connections(kind='all')) def test_net_if_addrs(self): # Note: verified that on Windows this was a false positive. tolerance = 80 * 1024 if WINDOWS else self.tolerance self.execute(psutil.net_if_addrs, tolerance=tolerance) def test_net_if_stats(self): self.execute(psutil.net_if_stats) # --- sensors @fewtimes_if_linux() @pytest.mark.skipif(not HAS_SENSORS_BATTERY, reason="not supported") def test_sensors_battery(self): self.execute(psutil.sensors_battery) @fewtimes_if_linux() @pytest.mark.skipif(not HAS_SENSORS_TEMPERATURES, reason="not supported") def test_sensors_temperatures(self): self.execute(psutil.sensors_temperatures) @fewtimes_if_linux() @pytest.mark.skipif(not HAS_SENSORS_FANS, reason="not supported") def test_sensors_fans(self): self.execute(psutil.sensors_fans) # --- others @fewtimes_if_linux() def test_boot_time(self): self.execute(psutil.boot_time) def test_users(self): self.execute(psutil.users) def test_set_debug(self): self.execute(lambda: psutil._set_debug(False)) if WINDOWS: # --- win services def test_win_service_iter(self): self.execute(cext.winservice_enumerate) def test_win_service_get(self): pass def test_win_service_get_config(self): name = next(psutil.win_service_iter()).name() self.execute(lambda: cext.winservice_query_config(name)) def test_win_service_get_status(self): name = next(psutil.win_service_iter()).name() self.execute(lambda: cext.winservice_query_status(name)) def test_win_service_get_description(self): name = next(psutil.win_service_iter()).name() self.execute(lambda: cext.winservice_query_descr(name)) tests/test_posix.py 0000644 00000041443 15030211765 0010467 0 ustar 00 #!/usr/bin/env python3 # Copyright (c) 2009, Giampaolo Rodola'. All rights reserved. # Use of this source code is governed by a BSD-style license that can be # found in the LICENSE file. """POSIX specific tests.""" import datetime import errno import os import re import shutil import subprocess import time from unittest import mock import psutil from psutil import AIX from psutil import BSD from psutil import LINUX from psutil import MACOS from psutil import OPENBSD from psutil import POSIX from psutil import SUNOS from psutil.tests import AARCH64 from psutil.tests import HAS_NET_IO_COUNTERS from psutil.tests import PYTHON_EXE from psutil.tests import PsutilTestCase from psutil.tests import pytest from psutil.tests import retry_on_failure from psutil.tests import sh from psutil.tests import skip_on_access_denied from psutil.tests import spawn_testproc from psutil.tests import terminate if POSIX: import mmap import resource from psutil._psutil_posix import getpagesize def ps(fmt, pid=None): """Wrapper for calling the ps command with a little bit of cross-platform support for a narrow range of features. """ cmd = ['ps'] if LINUX: cmd.append('--no-headers') if pid is not None: cmd.extend(['-p', str(pid)]) elif SUNOS or AIX: cmd.append('-A') else: cmd.append('ax') if SUNOS: fmt = fmt.replace("start", "stime") cmd.extend(['-o', fmt]) output = sh(cmd) output = output.splitlines() if LINUX else output.splitlines()[1:] all_output = [] for line in output: line = line.strip() try: line = int(line) except ValueError: pass all_output.append(line) if pid is None: return all_output else: return all_output[0] # ps "-o" field names differ wildly between platforms. # "comm" means "only executable name" but is not available on BSD platforms. # "args" means "command with all its arguments", and is also not available # on BSD platforms. # "command" is like "args" on most platforms, but like "comm" on AIX, # and not available on SUNOS. # so for the executable name we can use "comm" on Solaris and split "command" # on other platforms. # to get the cmdline (with args) we have to use "args" on AIX and # Solaris, and can use "command" on all others. def ps_name(pid): field = "command" if SUNOS: field = "comm" command = ps(field, pid).split() return command[0] def ps_args(pid): field = "command" if AIX or SUNOS: field = "args" out = ps(field, pid) # observed on BSD + Github CI: '/usr/local/bin/python3 -E -O (python3.9)' out = re.sub(r"\(python.*?\)$", "", out) return out.strip() def ps_rss(pid): field = "rss" if AIX: field = "rssize" return ps(field, pid) def ps_vsz(pid): field = "vsz" if AIX: field = "vsize" return ps(field, pid) def df(device): try: out = sh(f"df -k {device}").strip() except RuntimeError as err: if "device busy" in str(err).lower(): raise pytest.skip("df returned EBUSY") raise line = out.split('\n')[1] fields = line.split() sys_total = int(fields[1]) * 1024 sys_used = int(fields[2]) * 1024 sys_free = int(fields[3]) * 1024 sys_percent = float(fields[4].replace('%', '')) return (sys_total, sys_used, sys_free, sys_percent) @pytest.mark.skipif(not POSIX, reason="POSIX only") class TestProcess(PsutilTestCase): """Compare psutil results against 'ps' command line utility (mainly).""" @classmethod def setUpClass(cls): cls.pid = spawn_testproc( [PYTHON_EXE, "-E", "-O"], stdin=subprocess.PIPE ).pid @classmethod def tearDownClass(cls): terminate(cls.pid) def test_ppid(self): ppid_ps = ps('ppid', self.pid) ppid_psutil = psutil.Process(self.pid).ppid() assert ppid_ps == ppid_psutil def test_uid(self): uid_ps = ps('uid', self.pid) uid_psutil = psutil.Process(self.pid).uids().real assert uid_ps == uid_psutil def test_gid(self): gid_ps = ps('rgid', self.pid) gid_psutil = psutil.Process(self.pid).gids().real assert gid_ps == gid_psutil def test_username(self): username_ps = ps('user', self.pid) username_psutil = psutil.Process(self.pid).username() assert username_ps == username_psutil def test_username_no_resolution(self): # Emulate a case where the system can't resolve the uid to # a username in which case psutil is supposed to return # the stringified uid. p = psutil.Process() with mock.patch("psutil.pwd.getpwuid", side_effect=KeyError) as fun: assert p.username() == str(p.uids().real) assert fun.called @skip_on_access_denied() @retry_on_failure() def test_rss_memory(self): # give python interpreter some time to properly initialize # so that the results are the same time.sleep(0.1) rss_ps = ps_rss(self.pid) rss_psutil = psutil.Process(self.pid).memory_info()[0] / 1024 assert rss_ps == rss_psutil @skip_on_access_denied() @retry_on_failure() def test_vsz_memory(self): # give python interpreter some time to properly initialize # so that the results are the same time.sleep(0.1) vsz_ps = ps_vsz(self.pid) vsz_psutil = psutil.Process(self.pid).memory_info()[1] / 1024 assert vsz_ps == vsz_psutil def test_name(self): name_ps = ps_name(self.pid) # remove path if there is any, from the command name_ps = os.path.basename(name_ps).lower() name_psutil = psutil.Process(self.pid).name().lower() # ...because of how we calculate PYTHON_EXE; on MACOS this may # be "pythonX.Y". name_ps = re.sub(r"\d.\d", "", name_ps) name_psutil = re.sub(r"\d.\d", "", name_psutil) # ...may also be "python.X" name_ps = re.sub(r"\d", "", name_ps) name_psutil = re.sub(r"\d", "", name_psutil) assert name_ps == name_psutil def test_name_long(self): # On UNIX the kernel truncates the name to the first 15 # characters. In such a case psutil tries to determine the # full name from the cmdline. name = "long-program-name" cmdline = ["long-program-name-extended", "foo", "bar"] with mock.patch("psutil._psplatform.Process.name", return_value=name): with mock.patch( "psutil._psplatform.Process.cmdline", return_value=cmdline ): p = psutil.Process() assert p.name() == "long-program-name-extended" def test_name_long_cmdline_ad_exc(self): # Same as above but emulates a case where cmdline() raises # AccessDenied in which case psutil is supposed to return # the truncated name instead of crashing. name = "long-program-name" with mock.patch("psutil._psplatform.Process.name", return_value=name): with mock.patch( "psutil._psplatform.Process.cmdline", side_effect=psutil.AccessDenied(0, ""), ): p = psutil.Process() assert p.name() == "long-program-name" def test_name_long_cmdline_nsp_exc(self): # Same as above but emulates a case where cmdline() raises NSP # which is supposed to propagate. name = "long-program-name" with mock.patch("psutil._psplatform.Process.name", return_value=name): with mock.patch( "psutil._psplatform.Process.cmdline", side_effect=psutil.NoSuchProcess(0, ""), ): p = psutil.Process() with pytest.raises(psutil.NoSuchProcess): p.name() @pytest.mark.skipif(MACOS or BSD, reason="ps -o start not available") def test_create_time(self): time_ps = ps('start', self.pid) time_psutil = psutil.Process(self.pid).create_time() time_psutil_tstamp = datetime.datetime.fromtimestamp( time_psutil ).strftime("%H:%M:%S") # sometimes ps shows the time rounded up instead of down, so we check # for both possible values round_time_psutil = round(time_psutil) round_time_psutil_tstamp = datetime.datetime.fromtimestamp( round_time_psutil ).strftime("%H:%M:%S") assert time_ps in {time_psutil_tstamp, round_time_psutil_tstamp} def test_exe(self): ps_pathname = ps_name(self.pid) psutil_pathname = psutil.Process(self.pid).exe() try: assert ps_pathname == psutil_pathname except AssertionError: # certain platforms such as BSD are more accurate returning: # "/usr/local/bin/python3.7" # ...instead of: # "/usr/local/bin/python" # We do not want to consider this difference in accuracy # an error. adjusted_ps_pathname = ps_pathname[: len(ps_pathname)] assert ps_pathname == adjusted_ps_pathname # On macOS the official python installer exposes a python wrapper that # executes a python executable hidden inside an application bundle inside # the Python framework. # There's a race condition between the ps call & the psutil call below # depending on the completion of the execve call so let's retry on failure @retry_on_failure() def test_cmdline(self): ps_cmdline = ps_args(self.pid) psutil_cmdline = " ".join(psutil.Process(self.pid).cmdline()) if AARCH64 and len(ps_cmdline) < len(psutil_cmdline): assert psutil_cmdline.startswith(ps_cmdline) else: assert ps_cmdline == psutil_cmdline # On SUNOS "ps" reads niceness /proc/pid/psinfo which returns an # incorrect value (20); the real deal is getpriority(2) which # returns 0; psutil relies on it, see: # https://github.com/giampaolo/psutil/issues/1082 # AIX has the same issue @pytest.mark.skipif(SUNOS, reason="not reliable on SUNOS") @pytest.mark.skipif(AIX, reason="not reliable on AIX") def test_nice(self): ps_nice = ps('nice', self.pid) psutil_nice = psutil.Process().nice() assert ps_nice == psutil_nice @pytest.mark.skipif(not POSIX, reason="POSIX only") class TestSystemAPIs(PsutilTestCase): """Test some system APIs.""" @retry_on_failure() def test_pids(self): # Note: this test might fail if the OS is starting/killing # other processes in the meantime pids_ps = sorted(ps("pid")) pids_psutil = psutil.pids() # on MACOS and OPENBSD ps doesn't show pid 0 if MACOS or (OPENBSD and 0 not in pids_ps): pids_ps.insert(0, 0) # There will often be one more process in pids_ps for ps itself if len(pids_ps) - len(pids_psutil) > 1: difference = [x for x in pids_psutil if x not in pids_ps] + [ x for x in pids_ps if x not in pids_psutil ] raise self.fail("difference: " + str(difference)) # for some reason ifconfig -a does not report all interfaces # returned by psutil @pytest.mark.skipif(SUNOS, reason="unreliable on SUNOS") @pytest.mark.skipif(not shutil.which("ifconfig"), reason="no ifconfig cmd") @pytest.mark.skipif(not HAS_NET_IO_COUNTERS, reason="not supported") def test_nic_names(self): output = sh("ifconfig -a") for nic in psutil.net_io_counters(pernic=True): for line in output.split(): if line.startswith(nic): break else: raise self.fail( f"couldn't find {nic} nic in 'ifconfig -a'" f" output\n{output}" ) # @pytest.mark.skipif(CI_TESTING and not psutil.users(), # reason="unreliable on CI") @retry_on_failure() def test_users(self): out = sh("who -u") if not out.strip(): raise pytest.skip("no users on this system") lines = out.split('\n') users = [x.split()[0] for x in lines] terminals = [x.split()[1] for x in lines] assert len(users) == len(psutil.users()) with self.subTest(psutil=psutil.users(), who=out): for idx, u in enumerate(psutil.users()): assert u.name == users[idx] assert u.terminal == terminals[idx] if u.pid is not None: # None on OpenBSD psutil.Process(u.pid) @retry_on_failure() def test_users_started(self): out = sh("who -u") if not out.strip(): raise pytest.skip("no users on this system") tstamp = None # '2023-04-11 09:31' (Linux) started = re.findall(r"\d\d\d\d-\d\d-\d\d \d\d:\d\d", out) if started: tstamp = "%Y-%m-%d %H:%M" else: # 'Apr 10 22:27' (macOS) started = re.findall(r"[A-Z][a-z][a-z] \d\d \d\d:\d\d", out) if started: tstamp = "%b %d %H:%M" else: # 'Apr 10' started = re.findall(r"[A-Z][a-z][a-z] \d\d", out) if started: tstamp = "%b %d" else: # 'apr 10' (sunOS) started = re.findall(r"[a-z][a-z][a-z] \d\d", out) if started: tstamp = "%b %d" started = [x.capitalize() for x in started] if not tstamp: raise pytest.skip(f"cannot interpret tstamp in who output\n{out}") with self.subTest(psutil=psutil.users(), who=out): for idx, u in enumerate(psutil.users()): psutil_value = datetime.datetime.fromtimestamp( u.started ).strftime(tstamp) assert psutil_value == started[idx] def test_pid_exists_let_raise(self): # According to "man 2 kill" possible error values for kill # are (EINVAL, EPERM, ESRCH). Test that any other errno # results in an exception. with mock.patch( "psutil._psposix.os.kill", side_effect=OSError(errno.EBADF, "") ) as m: with pytest.raises(OSError): psutil._psposix.pid_exists(os.getpid()) assert m.called def test_os_waitpid_let_raise(self): # os.waitpid() is supposed to catch EINTR and ECHILD only. # Test that any other errno results in an exception. with mock.patch( "psutil._psposix.os.waitpid", side_effect=OSError(errno.EBADF, "") ) as m: with pytest.raises(OSError): psutil._psposix.wait_pid(os.getpid()) assert m.called def test_os_waitpid_eintr(self): # os.waitpid() is supposed to "retry" on EINTR. with mock.patch( "psutil._psposix.os.waitpid", side_effect=OSError(errno.EINTR, "") ) as m: with pytest.raises(psutil._psposix.TimeoutExpired): psutil._psposix.wait_pid(os.getpid(), timeout=0.01) assert m.called def test_os_waitpid_bad_ret_status(self): # Simulate os.waitpid() returning a bad status. with mock.patch( "psutil._psposix.os.waitpid", return_value=(1, -1) ) as m: with pytest.raises(ValueError): psutil._psposix.wait_pid(os.getpid()) assert m.called # AIX can return '-' in df output instead of numbers, e.g. for /proc @pytest.mark.skipif(AIX, reason="unreliable on AIX") @retry_on_failure() def test_disk_usage(self): tolerance = 4 * 1024 * 1024 # 4MB for part in psutil.disk_partitions(all=False): usage = psutil.disk_usage(part.mountpoint) try: sys_total, sys_used, sys_free, sys_percent = df(part.device) except RuntimeError as err: # see: # https://travis-ci.org/giampaolo/psutil/jobs/138338464 # https://travis-ci.org/giampaolo/psutil/jobs/138343361 err = str(err).lower() if ( "no such file or directory" in err or "raw devices not supported" in err or "permission denied" in err ): continue raise else: assert abs(usage.total - sys_total) < tolerance assert abs(usage.used - sys_used) < tolerance assert abs(usage.free - sys_free) < tolerance assert abs(usage.percent - sys_percent) <= 1 @pytest.mark.skipif(not POSIX, reason="POSIX only") class TestMisc(PsutilTestCase): def test_getpagesize(self): pagesize = getpagesize() assert pagesize > 0 assert pagesize == resource.getpagesize() assert pagesize == mmap.PAGESIZE tests/test_unicode.py 0000644 00000024230 15030211765 0010746 0 ustar 00 #!/usr/bin/env python3 # Copyright (c) 2009, Giampaolo Rodola'. All rights reserved. # Use of this source code is governed by a BSD-style license that can be # found in the LICENSE file. """Notes about unicode handling in psutil ======================================. Starting from version 5.3.0 psutil adds unicode support, see: https://github.com/giampaolo/psutil/issues/1040 The notes below apply to *any* API returning a string such as process exe(), cwd() or username(): * all strings are encoded by using the OS filesystem encoding (sys.getfilesystemencoding()) which varies depending on the platform (e.g. "UTF-8" on macOS, "mbcs" on Win) * no API call is supposed to crash with UnicodeDecodeError * instead, in case of badly encoded data returned by the OS, the following error handlers are used to replace the corrupted characters in the string: * sys.getfilesystemencodeerrors() or "surrogatescape" on POSIX and "replace" on Windows. For a detailed explanation of how psutil handles unicode see #1040. Tests ===== List of APIs returning or dealing with a string: ('not tested' means they are not tested to deal with non-ASCII strings): * Process.cmdline() * Process.cwd() * Process.environ() * Process.exe() * Process.memory_maps() * Process.name() * Process.net_connections('unix') * Process.open_files() * Process.username() (not tested) * disk_io_counters() (not tested) * disk_partitions() (not tested) * disk_usage(str) * net_connections('unix') * net_if_addrs() (not tested) * net_if_stats() (not tested) * net_io_counters() (not tested) * sensors_fans() (not tested) * sensors_temperatures() (not tested) * users() (not tested) * WindowsService.binpath() (not tested) * WindowsService.description() (not tested) * WindowsService.display_name() (not tested) * WindowsService.name() (not tested) * WindowsService.status() (not tested) * WindowsService.username() (not tested) In here we create a unicode path with a funky non-ASCII name and (where possible) make psutil return it back (e.g. on name(), exe(), open_files(), etc.) and make sure that: * psutil never crashes with UnicodeDecodeError * the returned path matches """ import os import shutil import warnings from contextlib import closing import psutil from psutil import BSD from psutil import POSIX from psutil import WINDOWS from psutil.tests import ASCII_FS from psutil.tests import CI_TESTING from psutil.tests import HAS_ENVIRON from psutil.tests import HAS_MEMORY_MAPS from psutil.tests import HAS_NET_CONNECTIONS_UNIX from psutil.tests import INVALID_UNICODE_SUFFIX from psutil.tests import PYPY from psutil.tests import TESTFN_PREFIX from psutil.tests import UNICODE_SUFFIX from psutil.tests import PsutilTestCase from psutil.tests import bind_unix_socket from psutil.tests import chdir from psutil.tests import copyload_shared_lib from psutil.tests import create_py_exe from psutil.tests import get_testfn from psutil.tests import pytest from psutil.tests import safe_mkdir from psutil.tests import safe_rmpath from psutil.tests import skip_on_access_denied from psutil.tests import spawn_testproc from psutil.tests import terminate def try_unicode(suffix): """Return True if both the fs and the subprocess module can deal with a unicode file name. """ sproc = None testfn = get_testfn(suffix=suffix) try: safe_rmpath(testfn) create_py_exe(testfn) sproc = spawn_testproc(cmd=[testfn]) shutil.copyfile(testfn, testfn + '-2') safe_rmpath(testfn + '-2') except (UnicodeEncodeError, OSError): return False else: return True finally: if sproc is not None: terminate(sproc) safe_rmpath(testfn) # =================================================================== # FS APIs # =================================================================== class BaseUnicodeTest(PsutilTestCase): funky_suffix = None @classmethod def setUpClass(cls): super().setUpClass() cls.skip_tests = False cls.funky_name = None if cls.funky_suffix is not None: if not try_unicode(cls.funky_suffix): cls.skip_tests = True else: cls.funky_name = get_testfn(suffix=cls.funky_suffix) create_py_exe(cls.funky_name) def setUp(self): super().setUp() if self.skip_tests: raise pytest.skip("can't handle unicode str") @pytest.mark.xdist_group(name="serial") @pytest.mark.skipif(ASCII_FS, reason="ASCII fs") class TestFSAPIs(BaseUnicodeTest): """Test FS APIs with a funky, valid, UTF8 path name.""" funky_suffix = UNICODE_SUFFIX def expect_exact_path_match(self): with warnings.catch_warnings(): warnings.simplefilter("ignore") return self.funky_name in os.listdir(".") # --- def test_proc_exe(self): cmd = [ self.funky_name, "-c", "import time; [time.sleep(0.1) for x in range(100)]", ] subp = self.spawn_testproc(cmd) p = psutil.Process(subp.pid) exe = p.exe() assert isinstance(exe, str) if self.expect_exact_path_match(): assert os.path.normcase(exe) == os.path.normcase(self.funky_name) def test_proc_name(self): cmd = [ self.funky_name, "-c", "import time; [time.sleep(0.1) for x in range(100)]", ] subp = self.spawn_testproc(cmd) name = psutil.Process(subp.pid).name() assert isinstance(name, str) if self.expect_exact_path_match(): assert name == os.path.basename(self.funky_name) def test_proc_cmdline(self): cmd = [ self.funky_name, "-c", "import time; [time.sleep(0.1) for x in range(100)]", ] subp = self.spawn_testproc(cmd) p = psutil.Process(subp.pid) cmdline = p.cmdline() for part in cmdline: assert isinstance(part, str) if self.expect_exact_path_match(): assert cmdline == cmd def test_proc_cwd(self): dname = self.funky_name + "2" self.addCleanup(safe_rmpath, dname) safe_mkdir(dname) with chdir(dname): p = psutil.Process() cwd = p.cwd() assert isinstance(p.cwd(), str) if self.expect_exact_path_match(): assert cwd == dname @pytest.mark.skipif(PYPY and WINDOWS, reason="fails on PYPY + WINDOWS") def test_proc_open_files(self): p = psutil.Process() start = set(p.open_files()) with open(self.funky_name, 'rb'): new = set(p.open_files()) path = (new - start).pop().path assert isinstance(path, str) if BSD and not path: # XXX - see https://github.com/giampaolo/psutil/issues/595 raise pytest.skip("open_files on BSD is broken") if self.expect_exact_path_match(): assert os.path.normcase(path) == os.path.normcase(self.funky_name) @pytest.mark.skipif(not POSIX, reason="POSIX only") def test_proc_net_connections(self): name = self.get_testfn(suffix=self.funky_suffix) sock = bind_unix_socket(name) with closing(sock): conn = psutil.Process().net_connections('unix')[0] assert isinstance(conn.laddr, str) assert conn.laddr == name @pytest.mark.skipif(not POSIX, reason="POSIX only") @pytest.mark.skipif( not HAS_NET_CONNECTIONS_UNIX, reason="can't list UNIX sockets" ) @skip_on_access_denied() def test_net_connections(self): def find_sock(cons): for conn in cons: if os.path.basename(conn.laddr).startswith(TESTFN_PREFIX): return conn raise ValueError("connection not found") name = self.get_testfn(suffix=self.funky_suffix) sock = bind_unix_socket(name) with closing(sock): cons = psutil.net_connections(kind='unix') conn = find_sock(cons) assert isinstance(conn.laddr, str) assert conn.laddr == name def test_disk_usage(self): dname = self.funky_name + "2" self.addCleanup(safe_rmpath, dname) safe_mkdir(dname) psutil.disk_usage(dname) @pytest.mark.skipif(not HAS_MEMORY_MAPS, reason="not supported") @pytest.mark.skipif(PYPY, reason="unstable on PYPY") def test_memory_maps(self): with copyload_shared_lib(suffix=self.funky_suffix) as funky_path: def normpath(p): return os.path.realpath(os.path.normcase(p)) libpaths = [ normpath(x.path) for x in psutil.Process().memory_maps() ] # ...just to have a clearer msg in case of failure libpaths = [x for x in libpaths if TESTFN_PREFIX in x] assert normpath(funky_path) in libpaths for path in libpaths: assert isinstance(path, str) @pytest.mark.skipif(CI_TESTING, reason="unreliable on CI") class TestFSAPIsWithInvalidPath(TestFSAPIs): """Test FS APIs with a funky, invalid path name.""" funky_suffix = INVALID_UNICODE_SUFFIX def expect_exact_path_match(self): return True # =================================================================== # Non fs APIs # =================================================================== class TestNonFSAPIS(BaseUnicodeTest): """Unicode tests for non fs-related APIs.""" funky_suffix = UNICODE_SUFFIX @pytest.mark.skipif(not HAS_ENVIRON, reason="not supported") @pytest.mark.skipif(PYPY and WINDOWS, reason="segfaults on PYPY + WINDOWS") def test_proc_environ(self): # Note: differently from others, this test does not deal # with fs paths. env = os.environ.copy() env['FUNNY_ARG'] = self.funky_suffix sproc = self.spawn_testproc(env=env) p = psutil.Process(sproc.pid) env = p.environ() for k, v in env.items(): assert isinstance(k, str) assert isinstance(v, str) assert env['FUNNY_ARG'] == self.funky_suffix tests/test_osx.py 0000644 00000014253 15030211765 0010135 0 ustar 00 #!/usr/bin/env python3 # Copyright (c) 2009, Giampaolo Rodola'. All rights reserved. # Use of this source code is governed by a BSD-style license that can be # found in the LICENSE file. """macOS specific tests.""" import platform import re import time import psutil from psutil import MACOS from psutil import POSIX from psutil.tests import CI_TESTING from psutil.tests import HAS_BATTERY from psutil.tests import TOLERANCE_DISK_USAGE from psutil.tests import TOLERANCE_SYS_MEM from psutil.tests import PsutilTestCase from psutil.tests import pytest from psutil.tests import retry_on_failure from psutil.tests import sh from psutil.tests import spawn_testproc from psutil.tests import terminate if POSIX: from psutil._psutil_posix import getpagesize def sysctl(cmdline): """Expects a sysctl command with an argument and parse the result returning only the value of interest. """ out = sh(cmdline) result = out.split()[1] try: return int(result) except ValueError: return result def vm_stat(field): """Wrapper around 'vm_stat' cmdline utility.""" out = sh('vm_stat') for line in out.split('\n'): if field in line: break else: raise ValueError("line not found") return int(re.search(r'\d+', line).group(0)) * getpagesize() @pytest.mark.skipif(not MACOS, reason="MACOS only") class TestProcess(PsutilTestCase): @classmethod def setUpClass(cls): cls.pid = spawn_testproc().pid @classmethod def tearDownClass(cls): terminate(cls.pid) def test_process_create_time(self): output = sh(f"ps -o lstart -p {self.pid}") start_ps = output.replace('STARTED', '').strip() hhmmss = start_ps.split(' ')[-2] year = start_ps.split(' ')[-1] start_psutil = psutil.Process(self.pid).create_time() assert hhmmss == time.strftime( "%H:%M:%S", time.localtime(start_psutil) ) assert year == time.strftime("%Y", time.localtime(start_psutil)) @pytest.mark.skipif(not MACOS, reason="MACOS only") class TestSystemAPIs(PsutilTestCase): # --- disk @retry_on_failure() def test_disks(self): # test psutil.disk_usage() and psutil.disk_partitions() # against "df -a" def df(path): out = sh(f'df -k "{path}"').strip() lines = out.split('\n') lines.pop(0) line = lines.pop(0) dev, total, used, free = line.split()[:4] if dev == 'none': dev = '' total = int(total) * 1024 used = int(used) * 1024 free = int(free) * 1024 return dev, total, used, free for part in psutil.disk_partitions(all=False): usage = psutil.disk_usage(part.mountpoint) dev, total, used, free = df(part.mountpoint) assert part.device == dev assert usage.total == total assert abs(usage.free - free) < TOLERANCE_DISK_USAGE assert abs(usage.used - used) < TOLERANCE_DISK_USAGE # --- cpu def test_cpu_count_logical(self): num = sysctl("sysctl hw.logicalcpu") assert num == psutil.cpu_count(logical=True) def test_cpu_count_cores(self): num = sysctl("sysctl hw.physicalcpu") assert num == psutil.cpu_count(logical=False) # TODO: remove this once 1892 is fixed @pytest.mark.skipif( MACOS and platform.machine() == 'arm64', reason="skipped due to #1892" ) def test_cpu_freq(self): freq = psutil.cpu_freq() assert freq.current * 1000 * 1000 == sysctl("sysctl hw.cpufrequency") assert freq.min * 1000 * 1000 == sysctl("sysctl hw.cpufrequency_min") assert freq.max * 1000 * 1000 == sysctl("sysctl hw.cpufrequency_max") # --- virtual mem def test_vmem_total(self): sysctl_hwphymem = sysctl('sysctl hw.memsize') assert sysctl_hwphymem == psutil.virtual_memory().total @pytest.mark.skipif( CI_TESTING and MACOS and platform.machine() == 'arm64', reason="skipped on MACOS + ARM64 + CI_TESTING", ) @retry_on_failure() def test_vmem_free(self): vmstat_val = vm_stat("free") psutil_val = psutil.virtual_memory().free assert abs(psutil_val - vmstat_val) < TOLERANCE_SYS_MEM @retry_on_failure() def test_vmem_active(self): vmstat_val = vm_stat("active") psutil_val = psutil.virtual_memory().active assert abs(psutil_val - vmstat_val) < TOLERANCE_SYS_MEM @retry_on_failure() def test_vmem_inactive(self): vmstat_val = vm_stat("inactive") psutil_val = psutil.virtual_memory().inactive assert abs(psutil_val - vmstat_val) < TOLERANCE_SYS_MEM @retry_on_failure() def test_vmem_wired(self): vmstat_val = vm_stat("wired") psutil_val = psutil.virtual_memory().wired assert abs(psutil_val - vmstat_val) < TOLERANCE_SYS_MEM # --- swap mem @retry_on_failure() def test_swapmem_sin(self): vmstat_val = vm_stat("Pageins") psutil_val = psutil.swap_memory().sin assert abs(psutil_val - vmstat_val) < TOLERANCE_SYS_MEM @retry_on_failure() def test_swapmem_sout(self): vmstat_val = vm_stat("Pageout") psutil_val = psutil.swap_memory().sout assert abs(psutil_val - vmstat_val) < TOLERANCE_SYS_MEM # --- network def test_net_if_stats(self): for name, stats in psutil.net_if_stats().items(): try: out = sh(f"ifconfig {name}") except RuntimeError: pass else: assert stats.isup == ('RUNNING' in out), out assert stats.mtu == int(re.findall(r'mtu (\d+)', out)[0]) # --- sensors_battery @pytest.mark.skipif(not HAS_BATTERY, reason="no battery") def test_sensors_battery(self): out = sh("pmset -g batt") percent = re.search(r"(\d+)%", out).group(1) drawing_from = re.search(r"Now drawing from '([^']+)'", out).group(1) power_plugged = drawing_from == "AC Power" psutil_result = psutil.sensors_battery() assert psutil_result.power_plugged == power_plugged assert psutil_result.percent == int(percent) tests/__pycache__/test_aix.cpython-310.pyc 0000644 00000005771 15030211765 0014451 0 ustar 00 o �h8 � @ sj d Z ddlZddlZddlmZ ddlmZ ddlmZ ddlmZ ejj e dd�G d d � d e��Z dS ) zAIX specific tests.� N)�AIX)�PsutilTestCase)�pytest)�shzAIX only)�reasonc @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd S ) �AIXSpecificTestCasec C s� t d�}d}dD ] }|d|� d�7 }qt�||�}|d us J �d}t|�d��| }t|�d��| }t|�d ��| }t|�d ��| } t�� } d | | } | j|ksVJ �t| j | �| k saJ �t| j | �| k slJ �t| j | �| k swJ �d S ) Nz/usr/bin/svmon -O unit=KBz memory\s*)�size�inuse�free�pin�virtual� available�mmode�(?P<�>\S+)\s+i r r r r � ) r �re�search�int�group�psutil�virtual_memory�total�abs�usedr r ) �self�out� re_pattern�field�matchobj�KBr r r r � psutil_result�TOLERANCE_SYS_MEM� r# �H/usr/local/CyberCP/lib/python3.10/site-packages/psutil/tests/test_aix.py�test_virtual_memory s" z'AIXSpecificTestCase.test_virtual_memoryc C sT t d�}t�d|�}|d usJ �t|�d��}d}t�� }t|j| �|ks(J �d S )Nz/usr/sbin/lsps -az=(?P<space>\S+)\s+(?P<vol>\S+)\s+(?P<vg>\S+)\s+(?P<size>\d+)MBr i )r r r r r r � swap_memoryr )r r r �total_mb�MBr! r# r# r$ �test_swap_memory8 s � z$AIXSpecificTestCase.test_swap_memoryc C s� t d�}d}dD ] }|d|� d�7 }qt�||�}|d us J �d}t�� }t|jt|�d�� �|k s5J �t|j t|�d�� �|k sEJ �t|j t|�d �� �|k sUJ �t|j t|�d �� �|k seJ �d S ) N�/usr/bin/mpstat -azALL\s*)�min�maj�mpcs�mpcr�dev�soft�dec�ph�cs�ics�bound�rq�push�S3pull�S3grd�S0rd�S1rd�S2rd�S3rd�S4rd�S5rd�syscr r i� r3 r@ r/ r0 ) r r r r � cpu_statsr � ctx_switchesr r �syscalls� interrupts�soft_interrupts)r r r r r �CPU_STATS_TOLERANCEr! r# r# r$ �test_cpu_statsO s( ��� �z"AIXSpecificTestCase.test_cpu_statsc C s: t d�}tt�d|��d��}tjdd�}||ksJ �d S )Nr* z lcpu=(\d+)� T)�logical)r r r r r r � cpu_count)r r � mpstat_lcpu� psutil_lcpur# r# r$ �test_cpu_count_logical� s z*AIXSpecificTestCase.test_cpu_count_logicalc C s4 t d�}t|�� �}tt�� �� �}||ksJ �d S )Nz/etc/ifconfig -l)r �set�splitr � net_if_addrs�keys)r r �ifconfig_names� psutil_namesr# r# r$ �test_net_if_addrs_names� s z+AIXSpecificTestCase.test_net_if_addrs_namesN)�__name__� __module__� __qualname__r% r) rG rM rT r# r# r# r$ r s "5 r ) �__doc__r r r � psutil.testsr r r �mark�skipifr r# r# r# r$ �<module> s tests/__pycache__/__init__.cpython-310.pyc 0000644 00000151563 15030211765 0014371 0 ustar 00 o �hh� � @ s d Z ddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl Z ddl Z ddl Z ddl Z ddlZddlZddlZddlZddlZddlZddlZddlZddlZddlZddlZddlZddlZddlZddlm Z ddlm Z ddlm Z zddl Z W n e y� dZ Y nw 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*m/Z/ e'r�ddl0m1Z1 g d�Z2dej3v Z4de j5v p�de j5v Z6e6Z7de j5v Z8de j5v Z9ej:dkZ;e �<� dkZ=e-d d � �Z>e$�r*e>� d kZ?e>� d kZ@nd Z?d Z@d!ZAd"ZBd#ZCd$ZDe7�rIeAd%9 ZAeDd%9 ZDeBd&9 ZBeCd%9 ZCe jEd'k�rXd(e �F� � d)�ZGnd*e �F� � d)�ZGd+ZHd,�Id-d.�ZJe�K� �L� d/v ZMe jN�Oe jN�Pe jN�QeR�d0d0��ZSe j5�Td1e jN�PeSd2��ZUe jN�Oe jN�QeR��ZVeWe!jXd3�ZYeWe!d4�ZZeWe!jXd5�Z[eWe!d6�Z\eWe!jXd7�Z]eWe!jXd8�Z^e'�o�e( Z_eWe!d9�Z`eWe!jXd:�ZaeWe!jXd;�ZbeWe!jXd<�ZceWe!d=�Zdz ed�o�eee!�f� �ZgW n eh�y� d ZgY nw eWe!d>�ZieWe!d?�ZjeWe!jXd@�Zke$�se"�o e �l� dkZmdAdB� Znen� \ZoZpeqe jrdC�Zse�tesju� dDdE� eve!�D �ZwexedFey� �Zze{� Z|e{� Z}G dGdH� dHej~�ZdIdJ� Z�e�d�dKdL��Z�e�dMdN� �Z�dOdP� Z�e�dQdR� �Z�e�dSdT� �Z�ej�eDfdUdV�Z�d�dWdX�Z�dYdZ� Z�d[d\� Z�G d]d^� d^�Z�e�e!j�deDd_d`�dadb� �Z�e�e�e�fdeDd_d`�d�ddde��Z�e�e�deDd_d`�dfdg� �Z�dhdi� Z�djdk� Z�ej�dldm� �Z�dndo� Z�d�dpdq�Z�d�dsdt�Z�G dudv� dv�Z�e du �r�e�Z G dwdx� dxej��Z�e j�j�e4dydz�G d{d|� d|e���Z�d}d~� Z�dd�� Z�d�d�� Z�G d�d�� d��Z�G d�d�� d��Z�eAfd�d��Z�d�d�d��Z�d�d�d��Z�d�d�d��Z�e e dfd�d��Z�ej fd�d��Z�d�d�d��Z�d�d�� Z�ej�d�d�� �Z�d�d�� Z�d�d�� Z�d�d�� Z�d�d�� Z�d�d�� Z�d�d�� Z�d�d�� Z�e'�raej�d�d�d���Z�nej�d�d�d���Z�ejtd�d�� �Z�e'�r~e�ej�d�d�� � dS 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_pid)K�DEVNULL�GLOBAL_TIMEOUT�TOLERANCE_SYS_MEM� NO_RETRIES�PYPY� PYTHON_EXE�PYTHON_EXE_ENV�ROOT_DIR� SCRIPTS_DIR� TESTFN_PREFIX�UNICODE_SUFFIX�INVALID_UNICODE_SUFFIX� CI_TESTING�VALID_PROC_STATUSES�TOLERANCE_DISK_USAGE�IS_64BIT�HAS_CPU_AFFINITY� HAS_CPU_FREQ� HAS_ENVIRON�HAS_PROC_IO_COUNTERS� HAS_IONICE�HAS_MEMORY_MAPS�HAS_PROC_CPU_NUM� HAS_RLIMIT�HAS_SENSORS_BATTERY� HAS_BATTERY�HAS_SENSORS_FANS�HAS_SENSORS_TEMPERATURES�HAS_NET_CONNECTIONS_UNIX� MACOS_11PLUS� MACOS_12PLUS�COVERAGE�AARCH64�PYTEST_PARALLEL�pyrun� terminate� reap_children�spawn_testproc� spawn_zombie�spawn_children_pair� ThreadTask�unittest�skip_on_access_denied�skip_on_not_implemented�retry_on_failure�TestMemoryLeak�PsutilTestCase�process_namespace�system_namespace� print_sysinfo�is_win_secure_system_proc� fake_pytest�chdir� safe_rmpath� create_py_exe� create_c_exe� get_testfn� get_winver�kernel_version� call_until� wait_for_pid� wait_for_file�check_net_address�filter_proc_net_connections� get_free_port� bind_socket�bind_unix_socket�tcp_socketpair�unix_socketpair�create_sockets� reload_module�import_module_by_path�warn�copyload_shared_lib� is_namedtuple�__pypy__�GITHUB_ACTIONS� CIBUILDWHEEL� COVERAGE_RUN�PYTEST_XDIST_WORKERl �aarch64c C sn t �� d } ttt| �d�d d� ��}|dkr5tjtj dddgdd id d �} ttt| �d�d d� ��}|S ) Nr �.� )� � z-sS�-cz-import platform; print(platform.mac_ver()[0])�SYSTEM_VERSION_COMPAT�0T)�env�universal_newlines) �platform�mac_ver�tuple�map�int�split� subprocess� check_output�sys� executable)� version_str�version� ry �H/usr/local/CyberCP/lib/python3.10/site-packages/psutil/tests/__init__.py� macos_versionz s �� r{ )rf � )� r Frf i P i � � � � �javaz$psutil-�-z@psutil-u -ƒőős f���utf8�surrogateescape> �us-ascii�asciiz..�PSUTIL_SCRIPTS_DIR�scripts� cpu_affinity�cpu_freq�environ� getloadavg�ionice� memory_maps�net_io_counters�cpu_num� io_counters�rlimit�sensors_battery� sensors_fans�sensors_temperatures�threadsc C s� dd� } t j�� }ttdd �}tr#tjdkr#|d ur#tj|d< ||fS tr*tj|fS t r[| tj�pP| t j � tj��pP| t � djtjd d� � ��pP| t�� �� �}|sWtd��||fS t j � tj�}t j �|�slJ |��||fS ) Nc S s6 zt j| dgt jt jd� W | S t jy Y d S w )Nz-V)�stdout�stderr)rs � check_call�PIPE�CalledProcessError)�exery ry rz �attempt� s ���z _get_py_exe.<locals>.attempt�_base_executable)r � �__PYVENV_LAUNCHER__z python{}.{}re z"can't find python exe real abspath)�osr� �copy�getattrru r � version_inforv r_ r �path�realpath�shutil�which�format�psutil�Processr� � ValueError�exists)r� rk �baser� ry ry rz � _get_py_exe� s0 ����r� zr+c C � g | ] }|� d �rtt|��qS )�STATUS_�� startswithr� r� ��.0�xry ry rz � <listcomp> � ��r� �AF_UNIXc sP e Zd ZdZ� fdd�Zdd� Zdd� Zdd � Zd d � Zd d � Z dd� Z � Z S )r; z6A thread task which does nothing expect staying alive.c s$ t � �� d| _d| _t�� | _d S )NF���MbP?)�super�__init__�_running� _interval� threading�Event�_flag��self�� __class__ry rz r� s zThreadTask.__init__c C s( | j j}d|� d| j� dt| �d�d�S )N�<z running=z at z#x�>)r� �__name__r� �id�r� �namery ry rz �__repr__ s zThreadTask.__repr__c C s | � � | S �N)�startr� ry ry rz � __enter__# s zThreadTask.__enter__c O s | � � d S r� )�stop)r� �args�kwargsry ry rz �__exit__' s zThreadTask.__exit__c C s( | j rtd��tj�| � | j�� dS )zStart thread and keep it running until an explicit stop() request. Polls for shutdown every 'timeout' seconds. zalready startedN)r� r� r� �Threadr� r� �waitr� ry ry rz r� * s zThreadTask.startc C s0 d| _ | j�� | j rt�| j� | j s d S d S )NT)r� r� �set�time�sleepr� r� ry ry rz �run3 s �zThreadTask.runc C s | j std��d| _ | �� dS )z8Stop thread execution and and waits until it is stopped.zalready stoppedFN)r� r� �joinr� ry ry rz r� 9 s zThreadTask.stop) r� � __module__� __qualname__�__doc__r� r� r� r� r� r� r� � __classcell__ry ry r� rz r; s r; c s t �� �� fdd��}|S )Nc s( z� | i |��W S t y t� � w r� )� Exceptionr7 �r� r� ��funry rz �wrapperG s �z&_reap_children_on_err.<locals>.wrapper�� functools�wraps�r� r� ry r� rz �_reap_children_on_errF s r� c K s� |� dt� |� dt� |� dt�� � |� dt� tr$d}|� d|� | du r`tt�� d�}z+t|� d |� d �}td |g} t j | fi |��}t � |� t |d d d � W t|� |S t|� w t j | fi |��}t � |� t|j� |S )a Create a python subprocess which does nothing for some secs and return it as a subprocess.Popen instance. If "cmd" is specified that is used instead of python. By default stdin and stdout are redirected to /dev/null. It also attempts to make sure the process is in a reasonably initialized state. The process is registered for cleanup on reap_children(). �stdinr� �cwdrk � � creationflagsN��dirzimport time;open(r'z6', 'w').close();[time.sleep(0.1) for x in range(100)];rh T��delete�empty)� setdefaultr r� �getcwdr r rK rH r rs �Popen�_subprocesses_started�addrP rO �pid)�cmd�kwds�CREATE_NO_WINDOW�testfn�pyline�sprocry ry rz r8 R s4 �� � r8 c C s� d} t t�� d�}zLt�dtj�|�� dt� d��}tr%t |dd�\}} nt |�\}} t � |j �}t t|dd d ��}t�|� t � |�}||fW t|� | durUt| � S S t|� | durct| � w w ) a Create a subprocess which creates another one as in: A (us) -> B (child) -> C (grandchild). Return a (child, grandchild) tuple. The 2 processes are fully initialized and will live for 60 secs and are registered for cleanup on reap_children(). Nr� 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� )rK r� r� �textwrap�dedentr� �basenamer r r5 r� r� r� rq rP � _pids_startedr� rH )�tfiler� �s�subp�child�grandchild_pid� grandchildry ry rz r: | s0 � � �� �r: c s t jsJ �t� } t�d| � d��}d}t| �}zV|�t� t|�\}}|� � \}}z?t � |� � gg g t� t |� d��}t�|� t �|�� t� fdd�� |� fW |�� W |�� t| � |durht|� S S |�� w |�� t| � |durt|� w w )z�Create a zombie process and return a (parent, zombie) process tuple. In order to kill the zombie parent must be terminate()d first, then zombie must be wait()ed on. a import os, sys, time, socket, contextlib child_pid = os.fork() if child_pid > 0: time.sleep(3000) else: # this is the zombie process with socket.socket(socket.AF_UNIX) as s: s.connect('za') pid = bytes(str(os.getpid()), 'ascii') s.sendall(pid) N� c s � � � tjkS r� )�statusr� � STATUS_ZOMBIEry ��zombiery rz �<lambda>� s z spawn_zombie.<locals>.<lambda>)r� r rK r� r� rU � settimeoutr r5 �accept�select�filenorq �recvr r� r� rN �closerH )� unix_file�srcr �sock�parent�conn�_�zpidry r rz r9 � s< � � � �r9 c K s� |� dd� |� dd� t� }z/t|d�� }|�| � W d � n1 s%w Y tt|jgfi |��}t|j� ||fW S t yJ t |� � w )z�Run python 'src' code string in a separate interpreter. Returns a subprocess.Popen instance and the test file where the source code was written. r� Nr� �w) r� rK �open�writer8 r r� rO r� r� rH )r r� �srcfile�fr ry ry rz r5 � s � �r5 c K s� t rdnd}|�dtj� |�dtj� |�dd� |�d|� t| t�r*t�| �} tj| fi |��}t � |� |j t d�\}}|j dkrKt|| ��|rQt|� |�d �r\|d d � }|S ) zURun cmd in a subprocess and return its output. raises RuntimeError on error. r� r r� r� rl Tr� ��timeout� N���)r r� rs r� � isinstance�str�shlexrr r� r� r� � communicater � returncode� RuntimeErrorr[ �endswith)r� r� �flags�pr� r� ry ry rz �sh� s" r- c s� dd� �dd� � � ��fdd�}� ��fdd���fd d �}d d � }| }z�t |t�rP|||�W t |tjtjf�r<||� t |t�rC|n|j}t�|�rOJ |��S t |tjtjf�r�||�W t |tjtjf�rk||� t |t�rr|n|j}t�|�r~J |��S t |tj�r�|||�W t |tjtjf�r�||� t |t�r�|n|j}t�|�r�J |��S td |����t |tjtjf�r�||� t |t�r�|n|j}t�|�r�J |��w )a� Terminate a process and wait() for it. Process can be a PID or an instance of psutil.Process(), subprocess.Popen() or psutil.Popen(). If it's a subprocess.Popen() or psutil.Popen() instance also closes its stdin / stdout / stderr fds. PID is wait()ed even if the process is already gone (kills zombies). Does nothing if the process does not exist. Return process exit status. c S sN | � |� tr#t| tj�r%z t�| j�� |�W S tjy" Y d S w d S d S r� ) r� r r$ rs r� r� r� r� � NoSuchProcess��procr! ry ry rz r� s ��zterminate.<locals>.waitc S s6 t rtrtj}tr|tjkr| �tj� | �|� d S r� )r r_ �signal�SIGKILLr � send_signal�SIGCONT)r0 �sigry ry rz �sendsig s zterminate.<locals>.sendsigc sT z� | �� W n t y Y n ty$ } z tr |jdkr � d }~ww �| |�S )N� )�ProcessLookupError�OSErrorr �winerror)r0 r! �err�r6 r5 r� ry rz �term_subprocess_proc s �� z'terminate.<locals>.term_subprocess_procc s. z� | �� W n t jy Y nw �| |�S r� )r� r. r/ r< ry rz �term_psutil_proc" s � z#terminate.<locals>.term_psutil_procc sB zt �| �}W n t jy trt| |� Y S Y d S w � ||�S r� )r� r� r. r r )r� r! r0 )r> ry rz �term_pid) s �� zterminate.<locals>.term_pidc S s8 | j r| j �� | jr| j�� | jr| j�� d S d S r� )r� r r� r� )r0 ry ry rz � flush_popen3 s �z terminate.<locals>.flush_popenz wrong type ) r$ rq rs r� r� r� � pid_existsr� � TypeError)� proc_or_pidr5 � wait_timeoutr= r? r@ r, r� ry )r6 r5 r> r� rz r6 � s> � � �r6 c C s� t �� j| d�}trt�� }t|� ts tr t�� }t|� ts|rJ|D ]}t|dd� q$t j|td�\}}|D ]}t d|�d�� t|t j d� q8dS dS )a Terminate and wait() any subprocess started by this test suite and any children currently running, ensuring that no processes stick around to hog resources. If recursive is True it also tries to terminate and wait() all grandchildren started by this process. �� recursiveN)rD r zcouldn't terminate process z; attempting kill())r5 ) r� r� �childrenr� �popr6 r � wait_procsr r[ r1 r2 )rF rG r r� r, r �alivery ry rz r7 M s$ ���r7 c C s� t std��d} t�� d }|D ]}|�� s|dkr | |7 } q | s)td|����d}d}| �d�}t|d �}t|�dkrDt|d �}t|�dkrPt|d �}|||fS ) z"Return a tuple such as (2, 6, 36).z not POSIX� re rd z can't parse r � r ) r �NotImplementedErrorr� �uname�isdigitr� rr rq �len)r rN �c�minor�micro�nums�majorry ry rz rM r s& rM c C s0 t std��t�� } | jpd}| d | d |fS )Nz not WINDOWSr rL )r rM ru �getwindowsversion�service_pack_major)�wv�spry ry rz rL � s rL c @ s< e Zd ZdZeddddfdd�Zdd� Zdd � Zd d � ZdS ) �retryzA retry decorator.Nr� c C s2 |r|rt d��|| _|| _|| _|| _|| _d S )Nz/timeout and retries args are mutually exclusive)r� � exceptionr! �retries�interval�logfun)r� r[ r! r\ r] r^ ry ry rz r� � s zretry.__init__c c sf � | j r t�� | j }t�� |k r d V t�� |k sd S d S | jr.t| j�D ]}d V q&d S d V q/r� )r! r� r\ �range)r� �stop_atr ry ry rz �__iter__� s � ���zretry.__iter__c C s | j d ur t�| j � d S d S r� )r] r� r� r� ry ry rz r� � s �z retry.sleepc s" t �� �� �fdd��}�|_|S )Nc sl d }�D ]/}z � | i |��W S �j y3 } z|}�jd ur%��|� ��� W Y d }~qd }~ww |�r� )r[ r^ r� )r� r� �excr �r� r� ry rz r� � s ��z retry.__call__.<locals>.wrapper)r� r� � decorator)r� r� r� ry rc rz �__call__� s zretry.__call__) r� r� r� r� r� r� ra r� re ry ry ry rz rZ � s � rZ r� )r[ r^ r! r] c C s$ | t �� vr t �| ��t �| � dS )z�Wait for pid to show up in the process list then return. Used in the test suite to give time the sub process to initialize. N)r� �pidsr. r� �r� ry ry rz rO � s rO Tc C sN t | d�� }|�� }W d � n1 sw Y |s |s J �|r%t| � |S )z8Wait for a file to be written on disk with some content.�rbN)r �readrH )�fnamer� r� r �datary ry rz rP � s �rP c C s | � }|sJ �|S )z1Keep calling function until it evaluates to True.ry )r� �retry ry rz rN � s rN c C sp dd� }z)t �| �}t�|j�rt�tj| �}nt�t j| �}t r'|� W dS ||� W dS t y7 Y dS w )z?Convenience function for removing temporary test files or dirs.c S s� t � � t }t � � |k r?z| � W S ty Y n ty3 } z|}td|� �� W Y d }~nd }~ww t �d� t � � |k s |�)Nz ignoring g{�G�z�?)r� r �FileNotFoundErrorr9 r[ r� )r� r` r r; ry ry rz � retry_fun s �� � z safe_rmpath.<locals>.retry_funN) r� �stat�S_ISDIR�st_moder� �partialr� �rmtree�remover rm )r� rn �str� ry ry rz rH s �rH c C s& zt �| � W dS ty Y dS w )z.Convenience function for creating a directory.N)r� �mkdir�FileExistsErrorr� ry ry rz � safe_mkdir$ s �rx c c s8 � t �� }zt �| � dV W t �|� dS t �|� w )z@Context manager which temporarily changes the current directory.N)r� r� rG )�dirname�curdirry ry rz rG , s � rG c C sR t j�| �r J | ��t�t| � t�t| � t r't � | �}t � | |j t j B � | S )z6Create a Python executable file in the given location.)r� r� r� �atexit�registerrH r� �copyfiler r ro �chmodrq �S_IEXEC)r� ru ry ry rz rI 7 s rI c C s� t j�| �r J | ��t�d�st�d��|du r t�d�}n t |t �s'J |��t � t | � ttdd�d�� }|�|� W d � n1 sEw Y zt�d|jd| g� W t |j� | S t |j� w ) z5Create a compiled C executable in the given location.�gcczgcc is not installedNz� #include <unistd.h> int main() { pause(); return 1; } z.c��suffixr z-o)r� r� r� r� r� �pytest�skipr� r� r$ r% r{ r| rH r rK r rs r� r� )r� �c_coder ry ry rz rJ B s � �rJ rK c C s< t jt| |d�}tj�|�s tj�|�}t�t |� |S q)z�Return an absolute pathname of a file or dir that did not exist at the time this call is made. Also schedule it for safe deletion at interpreter exit. It's technically racy but probably not really due to the time variant. T)�prefixr� r� ) �tempfile�mktempr r� r� r� r� r{ r| rH )r� r� r� r� ry ry rz rK \ s �rK c @ sT e Zd ZdZedd� �Zeddd��Zeddd��Zedd d ��ZG d d � d �Z dS )rF z�A class that mimics some basic pytest APIs. This is meant for when unit tests are run in production, where pytest may not be installed. Still, the user can test psutil installation via: $ python3 -m psutil.tests c O s4 t �� �t�}t jdd��|� tjdtdd� |S )z�Mimics pytest.main(). It has the same effect as running `python3 -m unittest -v` from the project root directory. re )� verbosityz<Fake pytest module was used. Test results may be inaccurate.rL �� stacklevel) r< � TestLoader�discover�HERE�TextTestRunnerr� �warningsr[ � UserWarning)r� �kw�suitery ry rz �mainw s �zfake_pytest.mainNc s. G dd� d�� t jd� fdd� �}|| |d�S )zMimics `pytest.raises`.c @ s e Zd ZdZedd� �ZdS )z)fake_pytest.raises.<locals>.ExceptionInfoNc S s | j S r� )�_excr� ry ry rz �value� s z/fake_pytest.raises.<locals>.ExceptionInfo.value)r� r� r� r� �propertyr� ry ry ry rz � ExceptionInfo� s r� Nc 3 s| � � � }z|V W n- | y6 } z!|r(t �|t|��s(d|� d|� d�}t|��||_W Y d }~d S d }~ww t| �d���)N�"z" does not match "z not raised)�re�searchr% �AssertionErrorr� )rb �match�einfor; �msg�r� ry rz �context� s � ��z#fake_pytest.raises.<locals>.context)r� r� )� contextlib�contextmanager)rb r� r� ry r� rz �raises� s zfake_pytest.raisesc C s" |r t �� �| |�S t �� �| �S )zMimics `pytest.warns`.)r< �TestCase�assertWarnsRegex� assertWarns)�warningr� ry ry rz �warns� s zfake_pytest.warnsrK c C s t �| ��)zMimics `unittest.SkipTest`.)r< �SkipTest��reasonry ry rz r� � s zfake_pytest.skipc @ s( e Zd Zeddd��ZG dd� d�ZdS )zfake_pytest.markrK c C s t �| |�S )z'Mimics `@pytest.mark.skipif` decorator.)r< �skipIf)� conditionr� ry ry rz �skipif� s zfake_pytest.mark.skipifc @ s" e Zd ZdZddd�Zdd� ZdS )z fake_pytest.mark.xdist_groupz4Mimics `@pytest.mark.xdist_group` decorator (no-op).Nc C s d S r� ry r� ry ry rz r� � � z%fake_pytest.mark.xdist_group.__init__c C s |S r� ry )r� � cls_or_methry ry rz re � r� z%fake_pytest.mark.xdist_group.__call__r� )r� r� r� r� r� re ry ry ry rz � xdist_group� s r� N�rK )r� r� r� � staticmethodr� r� ry ry ry rz �mark� s r� r� r� ) r� r� r� r� r� r� r� r� r� r� ry ry ry rz rF o s rF c @ sZ e Zd ZdZddd�Zdd� Zdd � Zd d � Zd d � Zdd� Z dd� Z dd� Z dd� Z dS )rA z�Test class providing auto-cleanup wrappers on top of process test utilities. All test classes should derive from this one, even if we use pytest. rK Nc C s t ||d�}| �t|� |S )N)r� r� )rK � addCleanuprH )r� r� r� rj ry ry rz rK � s zPsutilTestCase.get_testfnc O s t |i |��}| �t|� |S r� )r8 r� r6 )r� r� r� r� ry ry rz r8 � s z PsutilTestCase.spawn_testprocc C �* t � \}}| �t|� | �t|� ||fS r� )r: r� r6 )r� �child1�child2ry ry rz r: � � z"PsutilTestCase.spawn_children_pairc C r� r� )r9 r� r6 )r� r r ry ry rz r9 � r� zPsutilTestCase.spawn_zombiec O s. t |i |��\}}| �t|� | �t|� |S r� )r5 r� rH r6 )r� r� r� r� r ry ry rz r5 � s zPsutilTestCase.pyrunc C s� t |tj�sJ �|j|jksJ �|j|jksJ �|jr |js J �t |tj�r:|j|jks.J �|jd ur:|jdks:J �t |� t |� d S �Nr ) r$ r� �Errorr� r� �_name� ZombieProcess�ppid�_ppidr% �repr)r� r0 rb ry ry rz �_check_proc_exc� s z PsutilTestCase._check_proc_excc C s� t �tj�� }zt�|� W n tjy td��w W d � n1 s%w Y |jj|ks2J �|jj d u s:J �t� |�rCJ |��|t� � vsKJ �|dd� t� � D �vsXJ �d S )Nz&wasn't supposed to raise ZombieProcessc S � g | ]}|j �qS ry rg r� ry ry rz r� � � z0PsutilTestCase.assertPidGone.<locals>.<listcomp>) r� r� r� r. r� r� r� r� r� r� rA rf � process_iter)r� r� �cmry ry rz � assertPidGone� s �� � z PsutilTestCase.assertPidGonec C s� | � |j� t|�}|j|jdd�D ]N\}}| j||d��; z|� }W n! tjy, � tjyD } z | � ||� W Y d }~nd }~ww d|� d|��}t |��W d � n1 s[w Y q|j dd� d S )NT�� clear_cache�r0 r� zProcess.z!() didn't raise NSP and returned r r ) r� r� rB �iter�all�subTestr� r� r. r� r� r� )r� r0 �nsr� r� rl rb r� ry ry rz �assertProcessGone� s* ����� �� z PsutilTestCase.assertProcessGonec C s� t �|j�}||ks J �tstst|�t|�ksJ �|�� t jks#J �|�� s)J �t � |j�s1J �|� � |jt � � v s>J �|jdd� t � � D �v sLJ �i t _ |jdd� t � � D �v s]J �t|�}|j|jdd�D ]=\}}| j||d��* z|� W n t jt jfy� } z | �||� W Y d }~nd }~ww W d � n1 s�w Y qit�rt�t j�� }|�� W d � n1 s�w Y | �||j� t�t j�� }|�� W d � n1 s�w Y | �||j� t�t j�� }|�� W d � n 1 �sw Y | �||j� |� � |� � |� � |� � |�� �s%J �t � |j��s.J �|jt � � v �s8J �|jdd� t � � D �v �sGJ �i t _ |jdd� t � � D �v �sYJ �d S ) Nc S r� ry rg r� ry ry rz r� r� z6PsutilTestCase.assertProcessZombie.<locals>.<listcomp>c S r� ry rg r� ry ry rz r� ! r� Tr� r� c S r� ry rg r� ry ry rz r� = r� c S r� ry rg r� ry ry rz r� ? r� ) r� r� r� r r �hashr r � is_runningrA �as_dictrf r� �_pmaprB r� r� r� r� � AccessDeniedr� r r� r� �cmdliner� r� r� �suspend�resumer6 �kill)r� r0 �cloner� r� r� rb r� ry ry rz �assertProcessZombie s^ ��� �� � � � "z"PsutilTestCase.assertProcessZombie�rK N) r� r� r� r� rK r8 r: r9 r5 r� r� r� r� ry ry ry rz rA � s rA zunreliable on PYPYr� c @ s� e Zd ZdZdZdZdZerdndZdZ e � � Z e e�d��Zedd � �Zed d � �Zd d � Zdd� Zdd� Zdd� Zdd� Zdd� Zdd� Z d dd �Zd d � Z dS ) r@ a� Test framework class for detecting function memory leaks, typically functions implemented in C which forgot to free() memory from the heap. It does so by checking whether the process memory usage increased before and after calling the function many times. Note that this is hard (probably impossible) to do reliably, due to how the OS handles memory, the GC and so on (memory can even decrease!). In order to avoid false positives, in case of failure (mem > 0) we retry the test for up to 5 times, increasing call repetitions each time. If the memory keeps increasing then it's a failure. If available (Linux, OSX, Windows), USS memory is used for comparison, since it's supposed to be more precise, see: https://gmpy.dev/blog/2016/real-process-memory-and-environ-in-python If not, RSS memory is used. mallinfo() on Linux and _heapwalk() on Windows may give even more precision, but at the moment are not implemented. PyPy appears to be completely unstable for this framework, probably because of its JIT, so tests on PYPY are skipped. Usage: class TestLeaks(psutil.tests.TestMemoryLeak): def test_fun(self): self.execute(some_function) �� rf r r~ T� PSUTIL_DEBUGc C s t �d� d S )NF)r� � _set_debug��clsry ry rz � setUpClassy � zTestMemoryLeak.setUpClassc C s t �| j� d S r� )r� r� �_psutil_debug_origr� ry ry rz � tearDownClass} s z TestMemoryLeak.tearDownClassc C s | j �� }t|d|j�S )N�uss)� _thisproc�memory_full_infor� �rss)r� �memry ry rz �_get_mem� s zTestMemoryLeak._get_memc C s t r| j�� S | j�� S r� )r r� �num_fds� num_handlesr� ry ry rz � _get_num_fds� s zTestMemoryLeak._get_num_fdsc C s | j r t|dtjd� d S d S )N�yellow)�color�file)�verboser ru r� )r� r� ry ry rz �_log� s �zTestMemoryLeak._logc C s� | � � }| �|� | � � }|| }|dk r d|�d�}| �|��|dkrAtr(dnd}|dkr2|d7 }|� d|� d |��}| �|��d S ) z�Makes sure num_fds() (POSIX) or num_handles() (Windows) does not increase after calling a function. Used to discover forgotten close(2) and CloseHandle syscalls. r znegative diff z8 (gc probably collected a resource from a previous test)�fd�handlerL r z unclosed z after calling N)r� �call�failr )r� r� �before�after�diffr� �type_ry ry rz � _check_fds� s � �zTestMemoryLeak._check_fdsc C s^ t jdd� | �� }t|�D ] }| �|�}~~qt jdd� | �� }t jg ks)J �|| }|S )z�Get 2 distinct memory samples, before and after having called fun repeatedly, and return the memory difference. rL )� generation)�gc�collectr� r_ r� �garbage)r� r� �times�mem1r� rl �mem2r� ry ry rz � _call_ntimes� s zTestMemoryLeak._call_ntimesc C s� g }d}|}t d|d �D ]C}| �||�} d�|t| �t| | �|�} |�| � | |kp/| |k} | r>|dkr;| �| � d S |dkrEt� | �| � ||7 }| }q | �d�|���)Nr rL z,Run #{}: extra-mem={}, per-call={}, calls={}z. ) r_ r r� r �appendr� �printr� r� ) r� r� r r\ � tolerance�messages�prev_mem�increase�idxr� r� �successry ry rz � _check_mem� s. � zTestMemoryLeak._check_memc C s |� S r� ry )r� r� ry ry rz r� � s zTestMemoryLeak.callNc C s� |dur|n| j }|dur|n| j}|dur|n| j}|dur!|n| j}z"|dks-J d��|dks5J d��|dks=J d��|dksEJ d��W n tyX } ztt|���d}~ww | �||� | �|� | j ||||d� dS ) zTest a callable.NrL ztimes must be >= 1r zwarmup_times must be >= 0zretries must be >= 0ztolerance must be >= 0)r r\ r ) r � warmup_timesr\ r r� r� r% r r� r )r� r� r r r\ r r; ry ry rz �execute� s" � �� zTestMemoryLeak.executec s& � ��fdd�}�j |fi |�� dS )znConvenience method to test a callable while making sure it raises an exception on every call. c s �� � �� d S r� )� assertRaisesry �rb r� r� ry rz r� � s z*TestMemoryLeak.execute_w_exc.<locals>.callN)r )r� rb r� r� r� ry r rz � execute_w_exc� s z TestMemoryLeak.execute_w_exc)NNNN) r� r� r� r� r r r r r\ r� r� r� r� �boolr� �getenvr� � classmethodr� r� r� r� r� r� r r r� r r ry ry ry rz r@ P s. � r@ c C s� dd l } dd l}dd l}dd l}dd l}zdd l}W n ty% d }Y nw zdd l}W n ty7 d }Y nw | �� }t j rKt � d�rKt d�|d< n?t jrZdt�� d � �|d< n0t jr}dd�ttt�� �� |d< ttd�r||d d t�� 7 < n t�� � dt�� � �|d< d �tt�� �t� � g �|d < t j r�t� � d |d < d �t� � t� � t�!� g�|d < t"|dd�|d< |d ur�|d d|j#� d�7 < t j r�t � d�r�t ddg�}t|��$d�d |d< nd|d< t�%� d } | r�| |d< t&�'� |d< |�(� } | d � d | d � �|d< |j�)t �*� ��+d�|d< |j�,� �+d�|d < |�-� |d < t.j/�0d �|d < t.�1� |d < t2|d!< t�3� |d"< t.�4� |d#< t �5� |d$< d%j6t7d&d'� t �8� D ��� |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 �B� �C� } | �Dd-d � |�E| �|d.< tFd/t&jGd0� |�H� D ]\}}tFd1�6|d2 |�t&jGd0� �q�tFd/t&jGd0� t&jI�J� t&jI�J� d S )3Nr � lsb_releasezlsb_release -d -s�OSzDarwin zWindows � � win32_editionz, �archre �kernel�python� __version__z not installed�pipz (wheel=�)r� z --versionr" rL �glibcz fs-encoding�langz%Y-%m-%d %H:%M:%Sz boot-timer� �user�~�homer� �pyexe�hostname�PID�cpusz{:.1f}%, {:.1f}%, {:.1f}%c s s � | ] }|t �� d V qdS )�d N)r� � cpu_countr� ry ry rz � <genexpr>@ s � z print_sysinfo.<locals>.<genexpr>�loadavgz{}%%, used={}, total={}�memory�swaprf r� r0 zF======================================================================�r� z {:<17} {}�:)K� collections�datetime�getpass�locale�pprintr � ImportError�wheel� OrderedDictr� r r� r� r- �OSXrm rn r r� rp r% � win32_ver�hasattrr �systemrx �list� architecture�machiner rN �python_implementation�python_version�python_compilerr� r rr �libc_verru �getfilesystemencoding� getlocale� fromtimestamp� boot_time�strftime�now�getuserr� r� � expanduserr� r �node�getpidr) r� ro r� �virtual_memoryrq �percentr �used�total� swap_memoryrP rf r� r� rH �pformatr r� �itemsr� �flush)r0 r1 r2 r3 r4 r r6 �info�outr r r� r- �pinfo�k�vry ry rz rD � s� � � �� � �� ��� rD c C s2 t dd� �}z|� | dkW S ty Y dS w )Nc S sX i } t d�}|�� dd � D ]}dd� |�d�D �}|d t|d �}}|| |< q| S )Nztasklist.exe /NH /FO csvrL c S s g | ]}|� d d��qS )r� rK )�replacer� ry ry rz r� i s z@is_win_secure_system_proc.<locals>.get_procs.<locals>.<listcomp>�,r )r- � splitlinesrr rq )rl rV �line�bitsr� r� ry ry rz � get_procsd s z,is_win_secure_system_proc.<locals>.get_procsz Secure SystemF)r �KeyError)r� r_ ry ry rz rE b s �rE c C s6 t �� } t| d�r | �� S t| d�rt�| �� �S dS )Nr� r� r )r� r� r: r� �random�choicer� )r, ry ry rz �_get_eligible_cput s rc c @ s0 e Zd ZdZddi fddi fgZddi fddddifd di fd di fd di fd di fd di fddi fddi fg Zddi fddi fddi fddi fddi fddi fddi fddi fddddifd di fd di fd di fd di fd di fd!di fd"di fd#di fgZer�ed$di fg7 Zed%di fg7 Zed&di fg7 Zed'di fg7 Zer�ed(di fg7 Ze r�ed)di fg7 Ze r�ed*e j fi fg7 Ze r�ed+di fg7 Zer�ed,di fg7 Zer�ed-di fg7 Zer�ed.di fg7 Zer�ed/dd0d1ifg7 Zg Ze�red di fg7 Zn ed e jfi fg7 Ze �r ed*e j d2fi fg7 Ze �r8e�r.ed)e jd3fi fg7 Zn ed)e jfi fg7 Ze �rFed+e� gfi fg7 Zd4ejfi fd5di fd6di fd7di fd8di fgZe�rued4ejfi fg7 Zed4ej fi fg7 Zee e e Z d9d:� Z dDd;d<�Z d=d>� Z e!d?d@� �Z"e!dAdB� �Z#dCS )ErB a A container that lists all Process class method names + some reasonable parameters to be called with. Utility methods (parent(), children(), ...) are excluded. >>> ns = process_namespace(psutil.Process()) >>> for fun, name in ns.iter(ns.getters): ... fun() � cpu_percentry �memory_percentr� rG rF T� connectionsr� �oneshotr �parentsr� r� �r r� � cpu_times� create_timer� r� r� � memory_infor� �net_connections�kindr� �nice�num_ctx_switches� num_threads� open_filesr� r r� �username�uids�gids�terminalr� r� r� r� r� r� r� r� r� �groupedF)r i r r3 r� r� r6 r� c C s || _ d S r� )�_proc)r� r0 ry ry rz r� � � zprocess_namespace.__init__c c sb � t |�}t�|� |D ]"\}}}|r| �� t| j|�}tj|g|�R i |��}||fV q dS �z_Given a list of tuples yields a set of (fun, fun_name) tuples in random order. N)r< ra �shuffler� r� rx r� rr )r� �lsr� �fun_namer� r� r� ry ry rz r� � s � �zprocess_namespace.iterc C s | j j| j jdd� dS )z&Clear the cache of a Process instance.T)� _ignore_nspN)rx �_initr� r� ry ry rz r� � s z process_namespace.clear_cachec C sB |D ] \}}}d| }t ||�s |jj�d|�d�}t|��qdS )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)r: r� r� �AttributeError)r� � test_classr| r} r � meth_namer� ry ry rz �test_class_coverage� s ����z%process_namespace.test_class_coveragec C sV dd� | j D �}dd� | jD �}dd� ttj�D �}||B |A }|r)td|����d S )Nc S � h | ]}|d �qS ri ry r� ry ry rz � <setcomp>� � z)process_namespace.test.<locals>.<setcomp>c S r� ri ry r� ry ry rz r� � r� c S s h | ] }|d dkr|�qS )r r ry r� ry ry rz r� s z uncovered Process class names: )r� �ignoredr� r� r� r� )r� �thisr� �klass�leftoutry ry rz �test� s �zprocess_namespace.testN)T)$r� r� r� r� �utilsr� �gettersr r& r' r* r� � RLIMIT_NOFILEr# r) r% r r( �setters�NORMAL_PRIORITY_CLASSr �IOPRIO_CLASS_NONE� IOPRIO_NORMALrc r1 �SIGTERM�killers� CTRL_C_EVENT�CTRL_BREAK_EVENTr� r� r� r� r r� r� ry ry ry rz rB } s� � � � rB c @ s� e Zd ZdZddi fddddifddddifddi fd dd difd dd difd dd difd dddifde�� fi fddddifddi fddi fddddifde�� fi fddi fddi fddi fddi fgZer{e rqe � � dkrqn ed dd difg7 Ze r�ed di fg7 Ze r�ed di fg7 Zer�ed di fg7 Zer�ed di fg7 Zer�ed!di fg7 Zed"d#i fg7 Zd$di fd%e�� gfi fd&di fd'di fgZeZed(d)� �ZejZd*S )+rC z�A container that lists all the module-level, system-related APIs. Utilities such as cpu_percent() are excluded. Usage: >>> ns = system_namespace >>> for fun, name in ns.iter(ns.getters): ... fun() rF ry r) �logicalFT� cpu_statsrj �percpu�disk_io_counters�perdisk�disk_partitionsr� � disk_usagerm rn � net_if_addrs� net_if_statsr� �pernicrA rf rQ �usersrM �arm64r� r� r� r� r� �win_service_iter�win_service_get)�algr� rI rd �cpu_times_percentc c sT � t | �} t�| � | D ]\}}}tt|�}tj|g|�R i |��}||fV q dS rz )r< ra r{ r� r� r� rr )r| r} r� r� r� ry ry rz r� = s � �zsystem_namespace.iterN)r� r� r� r� r� r� rL r� r$ r rm r> �HAS_GETLOADAVGr. r- r+ r r� r� r� r� r� r� rB r� ry ry ry rz rC sZ �� rC c C s dd� }t td| |d�S )zZDecorator which runs a test function and retries N times before actually failing. c S s t | �d�tjd� d S )Nz , retryingr. )r ru r� )rb ry ry rz r^ Q s z retry_on_failure.<locals>.logfunN)r[ r! r\ r^ )rZ r� )r\ r^ ry ry rz r? L s �r? c � � fdd�}|S )z,Decorator to Ignore AccessDenied exceptions.c � t �� �� �fdd��}|S )Nc s: z� | i |��W S t jy �d ur�s� t�d��w )Nzraises AccessDenied)r� r� r� r� r� �r� �only_ifry rz r� ] s �z9skip_on_access_denied.<locals>.decorator.<locals>.wrapperr� r� �r� r� rz rd \ s z(skip_on_access_denied.<locals>.decoratorry �r� rd ry r� rz r= Y s r= c r� )z3Decorator to Ignore NotImplementedError exceptions.c r� )Nc sD z� | i |��W S t y! �d ur�s� � j�d�}t�|��w )Nz2 was skipped because it raised NotImplementedError)rM r� r� r� )r� r� r� r� ry rz r� p s � �z;skip_on_not_implemented.<locals>.decorator.<locals>.wrapperr� r� r� r� rz rd o s z*skip_on_not_implemented.<locals>.decoratorry r� ry r� rz r> l s r> � 127.0.0.1c C sF t � � �}|�| df� |�� d W d � S 1 s w Y dS )z6Return an unused TCP port. Subject to race conditions.r rL N)�socket�bind� getsockname)�hostr ry ry rz rS � s $�rS c C s~ |du r | t thv r d}t�| |�}z tjdvr!|�tjtjd� |�|� |tj kr0|� d� |W S t y> |� � � w )zBinds a generic socket.N�rK r > �nt�cygwinrL r~ ) r r r� r� r� � setsockopt� SOL_SOCKET� SO_REUSEADDRr� r �listenr� r )�family�type�addrr ry ry rz rT � s �rT c C sp t jsJ �tj�| �rJ | ��t�tj|�}z|�| � |tjkr)|� d� W |S W |S t y7 |� � � w )zBind a UNIX socket.r~ ) r� r r� r� r� r� r� r� r r� r� r )r� r� r ry ry rz rU � s � ��rU r� c C s� t � | t��G}|�|� |�d� |�� }t � | t�}z%|�|� |�� } |�� \}}||kr<||fW W d � S |�� q& tyL |�� � w 1 sPw Y dS )z^Build a pair of TCP sockets connected to each other. Return a (server, client) tuple. r~ TN) r� r r� r� r� �connectr r r9 )r� r� �llrQ �caddr�ary ry rz rV � s( � � ��rV c C s� t jsJ �d }}z#t| tjd�}|�d� t�tjtj�}|�d� |�| � W ||fS tyD |dur;|� � |durC|� � � w )z�Build a pair of UNIX sockets connected to each other through the same UNIX file name. Return a (server, client) tuple. N�r� r ) r� r rU r� r � setblockingr� r� r� r )r� �server�clientry ry rz rW � s ��rW c c s � g } d }}zg| � ttjtj�ttjtj�f� t� r-| � ttjtj�ttjtj�f� trQt rQt � }t � }t |�\}}t |tjd�}|||fD ]}| � |� qI| V W | D ]}|�� qW||fD ] }|durlt|� qbdS | D ]}|�� qq||fD ] }|dur�t|� q|w )z1Open as many socket families / types as possible.Nr� )�extendrT r� r r � SOCK_DGRAMr r r r/ rK rW rU r r rH )�socks�fname1�fname2�s1�s2�s3r rj ry ry rz rX � sF � � � ��� ��rX c C s� t |tj�s J |��|tjkrAdd� | �d�D �}t|�dks#J | ��|D ]}d| kr4dks9J | �� J | ��q%t�| � dS |tj krVt | t �sOJ | ��t� | � dS |t j krit�d| �dusgJ | ��dS td |����) z[Check a net address validity. Supported families are IPv4, IPv6 and MAC addresses. c S s g | ]}t |��qS ry )rq r� ry ry rz r� r� z%check_net_address.<locals>.<listcomp>rd r� r � z([a-fA-F0-9]{2}[:|\-]?){6}Nzunknown family )r$ �enum�IntEnumr� r rr rP � ipaddress� IPv4Addressr r% � IPv6Addressr� �AF_LINKr� r� r� )r� r� �octs�numry ry rz rQ � s & rQ c C sT dd� }dd� }dd� }dd� }d d � }|| � || � || � || � || � d S ) z*Check validity of a connection namedtuple.c S s� t | �dk}t | �dv sJ t | ���| d | jks J | j��| d | jks*J | j��| d | jks6J | j��| d | jksBJ | j��| d | jksNJ | j��| d | jksZJ | j��|rh| d | jksjJ | j��d S d S ) Nr� > r7 r� r rL re r r� r~ r7 )rP r� r� r� �laddr�raddrr r� )r �has_pidry ry rz � check_ntuple s �z-check_connection_ntuple.<locals>.check_ntuplec S s� | j ttthv s J | j ��t| j tj�sJ | ��| j tkrft�| j | j��7}z |� | j d df� W n t yK } z |j t j krA� W Y d }~n d }~ww W d � d S W d � d S 1 s_w Y d S | j tkrv| jtjksxJ | j��d S d S r� )r� r r r� r$ r� r� r� r� r� r� r9 �errno� EADDRNOTAVAILr r� � CONN_NONE)r r r; ry ry rz � check_family s( �����"� �z-check_connection_ntuple.<locals>.check_familyc S sl t tdt� �}| jtjtj|hv sJ | j��t| jtj�s!J | ��| jtjkr2| j t j ks4J | j ��d S d S )N�SOCK_SEQPACKET) r� r� �objectr� r r� r$ r� r� r r� r� )r r� ry ry rz � check_type1 s �� �z+check_connection_ntuple.<locals>.check_typec S s� | j | jfD ]M}| jtthv rCt|t�sJ t|���|s qt|jt �s*J t|j���d|j kr5dks;n J |j��t |j | j� q| jt krSt|t �sSJ t|���qd S )Nr i�� )r� r� r� r r r$ ro r� �portrq rQ �ipr� r% )r r� ry ry rz � check_addrs= s " ��z,check_connection_ntuple.<locals>.check_addrsc S s� t | jt�s J | j��dd� tt�D �}| j|v s J | j��| jtthv r7| jt kr7| jtj ks5J | j��d S | jtj ksBJ | j��d S )Nc S r� )�CONN_r� r� ry ry rz r� L r� zAcheck_connection_ntuple.<locals>.check_status.<locals>.<listcomp>) r$ r r% r� r� r� r r r� r r� )r �validsry ry rz � check_statusJ s �z-check_connection_ntuple.<locals>.check_statusNry )r r� r� r� r� r� ry ry rz �check_connection_ntuple s r� c C sJ g }| D ] }t r |jtjkr tr d|jv r 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 r� r� r� r r� r r )�cons�newr ry ry rz rR \ s rR c C � t �| �S r� )� importlib�reload)�modulery ry rz rY o ry rY c C sB t j�t j�| ��d }tj�|| �}tj�|�}|j� |� |S r� ) r� r� �splitextr r� �util�spec_from_file_location�module_from_spec�loader� exec_module)r� r� �spec�modry ry rz rZ s s rZ c C s t j| tdd� dS )zRaise a warning msg.re r� N)r� r[ r� )r� ry ry rz r[ � s r[ c C sV t | �}|j}t|�dks|d turdS t|dd�}t|t�s"dS tdd� |D ��S )z-Check if object is an instance of namedtuple.rL r F�_fieldsNc s s � | ]}t |t�V qd S r� )r$ r% )r� �nry ry rz r* � s � z is_namedtuple.<locals>.<genexpr>)r� � __bases__rP ro r� r$ r� )r� �t�br ry ry rz r] � s r] c # s| � t rdnd� d�t| � d�}� �fdd�t�� �� D �}t�|�}t�||� zt � |� |V W t |� dS t |� w )z�Ctx manager which picks up a random shared CO lib used by this process, copies it in another location and loads it in memory via ctypes. Return the new absolutized path. �pypyr z.sor� c s6 g | ]}t j�|j�d �kr� |j�� v r|j�qS )rL )r� r� r� �lowerr� �r� �extry rz r� � s $��'copyload_shared_lib.<locals>.<listcomp>N) r rK r� r� r� ra rb r� r} �ctypes�CDLLrH )r� �dst�libsr ry r� rz r\ � s � � r\ c # s � ddl m} ddl m} d� t| � d�}� fdd�t�� �� D �}tr2|s2dd� t�� �� D �}t� |�}t � ||� d }z)t � |�}|V W |d urct j jj}|jg|_||j�}|dkrc|� �t|� d S |d ur�t j jj}|jg|_||j�}|dkr�|� �t|� w ) z�Ctx manager which picks up a random shared DLL lib used by this process, copies it in another location and loads it in memory via ctypes. Return the new absolutized, normcased path. r )�WinError)�wintypesz.dllr� c sF g | ] }|j �� �� �r!d tj �|j ��� v r!d|j �� vr|j �qS )r �wow64)r� r� r* r� r r� �r� ry rz r� � s ��r c S s( g | ]}d t j�|j��� v r|j�qS )r� )r� r� r r� r� ry ry rz r� � s �N)r r r rK r� r� r� r ra rb r� r} �WinDLL�windll�kernel32� FreeLibrary�HMODULE�argtypes�_handlerH ) r� r r r r r �cfiler rl ry r rz r\ � s@ � � � � c C s t dd� d S )NTrE )r7 ry ry ry rz �cleanup_test_procs� r� r c C r� r� )ru �exit)r5 r ry ry rz r � s r r� )F)TFr� )r� )r� r� )�r� r{ r� r r� r� r� r� r� r� r� rm ra r� r r& r� r1 r� ro rs ru r� r� r� r� r< r� r r r r� r5 r� r r r r r r r r �psutil._commonr r r r r �psutil._psposixr �__all__�builtin_module_namesr r� r_ r r2 r4 �maxsizer" r> r3 r{ r0 r1 r r r! r r� rL r r �decoder rC r� �ASCII_FSr� r� r� ry �__file__r �getr r� r: r� r# r$ r% r� r' r( r/ �HAS_NET_IO_COUNTERSr) r& r* r+ r r� r, r� r- r. � HAS_THREADS�getuid� SKIP_SYSCONSr� r r r �devnullr r| r r� r r� r� r� r� r� r r� r; r� r8 r: r9 r5 r- r� r6 r7 rM rL rZ r. rO rm r� rP rN rH rx r� rG rI rJ rK rF r� rA r� r� r@ rD rE rc rB rC r? r= r> rS rT rU rV rW rX rQ r� rR rY rZ r[ r] r\ r ry ry ry rz �<module> s� � - �� � , �0 ) &) T% 9� � � " M &l F K 3 � tests/__pycache__/test_osx.cpython-310.pyc 0000644 00000014613 15030211765 0014474 0 ustar 00 o �h� � @ s d Z ddlZddlZddlZddlZddlmZ ddlmZ ddlmZ ddlm Z ddlm Z ddlm Z dd lm Z dd lm Z dd lmZ dd lmZ dd lmZ ddlmZ erbddlmZ dd� Zdd� Ze jje dd�G dd� de ��Ze jje dd�G dd� de ��ZdS )zmacOS specific tests.� N)�MACOS)�POSIX)� CI_TESTING)� HAS_BATTERY)�TOLERANCE_DISK_USAGE)�TOLERANCE_SYS_MEM)�PsutilTestCase)�pytest)�retry_on_failure)�sh)�spawn_testproc)� terminate)� getpagesizec C s6 t | �}|�� d }zt|�W S ty | Y S w )zmExpects a sysctl command with an argument and parse the result returning only the value of interest. � )r �split�int� ValueError)�cmdline�out�result� r �H/usr/local/CyberCP/lib/python3.10/site-packages/psutil/tests/test_osx.py�sysctl s �r c C sH t d�}|�d�D ]}| |v r nq td��tt�d|��d��t� S )z)Wrapper around 'vm_stat' cmdline utility.�vm_stat� zline not foundz\d+r )r r r r �re�search�groupr )�fieldr �liner r r r , s � r z MACOS only��reasonc @ s, e Zd Zedd� �Zedd� �Zdd� ZdS )� TestProcessc C s t � j| _d S �N)r �pid��clsr r r � setUpClass9 � zTestProcess.setUpClassc C s t | j� d S r# )r r$ r% r r r � tearDownClass= r( zTestProcess.tearDownClassc C s� t d| j� ��}|�dd��� }|�d�d }|�d�d }t�| j��� }|t� dt� |��ks3J �|t� dt� |��ks@J �d S ) Nzps -o lstart -p �STARTED� � ������z%H:%M:%Sz%Y) r r$ �replace�stripr �psutil�Process� create_time�time�strftime� localtime)�self�output�start_ps�hhmmss�year� start_psutilr r r �test_process_create_timeA s � z$TestProcess.test_process_create_timeN)�__name__� __module__� __qualname__� classmethodr' r) r= r r r r r" 7 s r" c @ s� e Zd Ze� dd� �Zdd� Zdd� Zejj e o e � � dkdd �d d � �Z d d � Zejj eo5e o5e � � dkdd �e� dd� ��Ze� dd� �Ze� dd� �Ze� dd� �Ze� dd� �Ze� dd� �Zdd � Zejj e d d �d d � �Zd S )!�TestSystemAPIsc C s� dd� }t jdd�D ]5}t �|j�}||j�\}}}}|j|ks"J �|j|ks)J �t|j| �tk s4J �t|j | �tk s?J �q d S )Nc S s� t d| � d���� }|�d�}|�d� |�d�}|�� d d� \}}}}|dkr+d}t|�d }t|�d }t|�d }||||fS ) Nzdf -k "�"r r � �noner+ i )r r0 r �popr )�pathr �linesr �dev�total�used�freer r r �dfV s z%TestSystemAPIs.test_disks.<locals>.dfF)�all) r1 �disk_partitions� disk_usage� mountpoint�devicerJ �absrL r rK )r7 rM �part�usagerI rJ rK rL r r r � test_disksR s �zTestSystemAPIs.test_disksc C � t d�}|tjdd�ksJ �d S )Nzsysctl hw.logicalcpuT��logical�r r1 � cpu_count�r7 �numr r r �test_cpu_count_logicalm � z%TestSystemAPIs.test_cpu_count_logicalc C rW )Nzsysctl hw.physicalcpuFrX rZ r\ r r r �test_cpu_count_coresq r_ z#TestSystemAPIs.test_cpu_count_cores�arm64zskipped due to #1892r c C sZ t �� }|jd d td�ksJ �|jd d td�ks J �|jd d td�ks+J �d S )Ni� zsysctl hw.cpufrequencyzsysctl hw.cpufrequency_minzsysctl hw.cpufrequency_max)r1 �cpu_freq�currentr �min�max)r7 �freqr r r � test_cpu_freqv s z TestSystemAPIs.test_cpu_freqc C s t d�}|t�� jks J �d S )Nzsysctl hw.memsize)r r1 �virtual_memoryrJ )r7 �sysctl_hwphymemr r r �test_vmem_total� s z TestSystemAPIs.test_vmem_totalz%skipped on MACOS + ARM64 + CI_TESTINGc C �* t d�}t�� j}t|| �tk sJ �d S )NrL )r r1 rh rL rS r �r7 � vmstat_val� psutil_valr r r �test_vmem_free� s z TestSystemAPIs.test_vmem_freec C rk )N�active)r r1 rh rp rS r rl r r r �test_vmem_active� � z TestSystemAPIs.test_vmem_activec C rk )N�inactive)r r1 rh rs rS r rl r r r �test_vmem_inactive� rr z!TestSystemAPIs.test_vmem_inactivec C rk )N�wired)r r1 rh ru rS r rl r r r �test_vmem_wired� rr z TestSystemAPIs.test_vmem_wiredc C rk )N�Pageins)r r1 � swap_memory�sinrS r rl r r r �test_swapmem_sin� rr z TestSystemAPIs.test_swapmem_sinc C rk )N�Pageout)r r1 rx �soutrS r rl r r r �test_swapmem_sout� rr z TestSystemAPIs.test_swapmem_soutc C st t �� �� D ]1\}}z td|� ��}W n ty Y qw |jd|v ks(J |��|jtt� d|�d �ks7J �qd S )Nz ifconfig �RUNNINGz mtu (\d+)r ) r1 � net_if_stats�itemsr � RuntimeError�isup�mtur r �findall)r7 �name�statsr r r r �test_net_if_stats� s � �z TestSystemAPIs.test_net_if_statsz no batteryc C s` t d�}t�d|��d�}t�d|��d�}|dk}t�� }|j|ks%J �|jt|�ks.J �d S )Nz pmset -g battz(\d+)%r zNow drawing from '([^']+)'zAC Power) r r r r r1 �sensors_battery� power_plugged�percentr )r7 r r� � drawing_fromr� � psutil_resultr r r �test_sensors_battery� s z#TestSystemAPIs.test_sensors_batteryN)r> r? r@ r rV r^ r` r �mark�skipifr �platform�machinerg rj r ro rq rt rv rz r} r� r r� r r r r rB M s: � � rB )�__doc__r� r r4 r1 r r � psutil.testsr r r r r r r r r r �psutil._psutil_posixr r r r� r� r" rB r r r r �<module> s2 tests/__pycache__/test_linux.cpython-310.pyc 0000644 00000220536 15030211765 0015025 0 ustar 00 o �h?[ � @ s� d Z ddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl Z ddl Z ddl Z ddl Z ddlmZ ddlZddlmZ ddlmZ ddlmZ ddlmZ ddlmZ dd lmZ dd lmZ dd lmZ dd lmZ dd lmZ ddlm Z ddlm Z ddlm Z ddlm Z ddlm Z ddlm!Z! ddlm"Z" ddlm#Z# ddlm$Z$ ddlm%Z% ddlm&Z& er�ddl'm(Z( ddl'm)Z) ddl'm*Z* dd l'm+Z+ ej,�-ej,�.e/��Z0d Z1d Z2d Z3d Z4er�d!Z5d"d#� Z6d$d%� Z7d&d'� Z8d(d)� Z9d*d+� Z:d,d-� Z;d.d/� Z<d0d1� Z=d2d3� Z>ej?d4d5� �Z@ej?d6d7� �ZAe!jBjCe d8d9�G d:d;� d;e ��ZDe!jBjCe d8d9�G d<d=� d=e ��ZEe!jBjCe d8d9�G d>d?� d?e ��ZFe!jBjCe d8d9�G d@dA� dAe ��ZGe!jBjCe d8d9�G dBdC� dCe ��ZHe!jBjCe d8d9�G dDdE� dEe ��ZIe!jBjCe d8d9�G dFdG� dGe ��ZJe!jBjCe d8d9�G dHdI� dIe ��ZKe!jBjCe d8d9�G dJdK� dKe ��ZLe!jBjCe d8d9�G dLdM� dMe ��ZMe!jBjCe d8d9�G dNdO� dOe ��ZNe!jBjCe d8d9�G dPdQ� dQe ��ZOe!jBjCe d8d9�G dRdS� dSe ��ZPe!jBjCe d8d9�G dTdU� dUe ��ZQe!jBjCe d8d9�G dVdW� dWe ��ZRe!jBjCe d8d9�G dXdY� dYe ��ZSe!jBjCe d8d9�G dZd[� d[e ��ZTe!jBjCe d8d9�G d\d]� d]e ��ZUe!jBjCe d8d9�e!jBjCe d^d9�G d_d`� d`e ���ZVe!jBjCe d8d9�G dadb� dbe ��ZWe!jBjCe d8d9�G dcdd� dde ��ZXe!jBjCe d8d9�G dedf� dfe ��ZYe!jBjCe d8d9�G dgdh� dhe ��ZZe!jBjCe d8d9�G didj� dje ��Z[e!jBjCe d8d9�G dkdl� dle ��Z\dS )mzLinux specific tests.� N)�mock)�LINUX)�AARCH64)�GITHUB_ACTIONS)�GLOBAL_TIMEOUT)� HAS_BATTERY)� HAS_CPU_FREQ)�HAS_GETLOADAVG)� HAS_RLIMIT)�PYPY)�PYTEST_PARALLEL)�TOLERANCE_DISK_USAGE)�TOLERANCE_SYS_MEM)�PsutilTestCase)� ThreadTask)� call_until)�pytest)� reload_module)�retry_on_failure)� safe_rmpath)�sh)�skip_on_not_implemented)� CLOCK_TICKS)�RootFsDeviceFinder)�calculate_avail_vmem)� open_binaryi� i'� i� i� i c C �v dd l }t| d d� d�} t�tjtj�� }t�|�|�� tt � d| ��dd� �W d � S 1 s4w Y d S �Nr � �ascii�256s� � ) �fcntl�bytes�socket�AF_INET� SOCK_DGRAM� inet_ntoa�ioctl�fileno� SIOCGIFADDR�struct�pack��ifnamer# �s� r1 �J/usr/local/CyberCP/lib/python3.10/site-packages/psutil/tests/test_linux.py�get_ipv4_addressD s ��$�r3 c C r r ) r# r$ r% r& r' r( r) r* �SIOCGIFNETMASKr, r- r. r1 r1 r2 �get_ipv4_netmaskP � ���$�r5 c C r r ) r# r$ r% r& r' r( r) r* �SIOCGIFBRDADDRr, r- r. r1 r1 r2 �get_ipv4_broadcast\ r6 r8 c s� t d��+}g }|D ]}|�� }|d | kr|�|� q t|�dkr(td| ����W d � n1 s2w Y tt|��D ],}|| d � � fdd�tdt� �d�D �}d�|�}t�tj |�}t� tj |�||< q=|S ) Nz/proc/net/if_inet6���r zcould not find interface c s g | ] }� ||d � �qS )� r1 )�.0�j�� unformattedr1 r2 � <listcomp>u s �z&get_ipv6_addresses.<locals>.<listcomp>r: �:) �open�split�append�len� ValueError�range�joinr% � inet_pton�AF_INET6� inet_ntop) r/ �f� all_fields�line�fields�i�groups� formatted�packedr1 r= r2 �get_ipv6_addressesh s( � � � � rS c C s� dd l }t| d d� d�} t�tjtj��'}|�|�� tt� d| ��}d� dd� |dd � D ��d d � W d � S 1 s?w Y d S ) Nr r r r � c S s g | ]}|d �d��qS )�02xr@ r1 )r; �charr1 r1 r2 r? � s z#get_mac_address.<locals>.<listcomp>� r" r9 ) r# r$ r% r& r' r) r* � SIOCGIFHWADDRr, r- rG )r/ r# r0 �infor1 r1 r2 �get_mac_address~ s �"$�rZ c C sx t ddgddid�} | �d�}|D ]#}|�d�r4|�� \}}}}t�dd�}|t|�t|�t|�� S qtd | � ���) zQParse 'free' cmd and return swap memory's s total, used and free values. �free�-b�LANG�C.UTF-8��env� �Swapztotal used freez$can't find 'Swap' in 'free' output: )r rB � startswith� collections� namedtuple�intrE )�out�linesrM �_�total�usedr[ �ntr1 r1 r2 � free_swap� s �rm c C s� t ddgddid�} | �d�}|D ](}|�d�r9dd � |�� d d � D �\}}}}t�dd �}|||||| � S qtd | � ���)zSParse 'free' cmd and return physical memory's total, used and free values. r[ r\ r] r^ r_ ra �Memc s s � | ]}t |�V qd S �N)rf �r; �xr1 r1 r2 � <genexpr>� s � z free_physmem.<locals>.<genexpr>� � z total used free shared outputz#can't find 'Mem' in 'free' output: )r rB rc rd re rE )rg rh rM rj rk r[ �sharedrl r1 r1 r2 � free_physmem� s "��rv c C sZ t ddgddid�}|�d�D ]}|�� }| |v r$t|�d�d � S qtd | �d ���) N�vmstatz-sr] r^ r_ ra � r � can't find z in 'vmstat' output)r rB �striprf rE )�statrg rM r1 r1 r2 rw � s �rw c C s@ t ddg��� } d| v rt�d��tttt�d| � � d ���S )Nr[ z-V�UNKNOWNz can't determine free versionz\d+r9 ) r rz r �skip�tuple�maprf �re�findallrB )rg r1 r1 r2 �get_free_version_info� s r� c # sP � � �fdd�}t � tjdd|d�� }|V W d � dS 1 s!w Y dS )z�Mock open() builtin and forces it to return a certain content for a given path. `pairs` is a {"path": "content", ...} dict. c sD | �v r�| }t |t�rt�|�S t�|�S � | g|�R i |��S ro )� isinstance�str�io�StringIO�BytesIO)�name�args�kwargs�content�� orig_open�pairsr1 r2 � open_mock� s z$mock_open_content.<locals>.open_mock� builtins.openT��create� side_effectN�rA r �patch)r� r� �mr1 r� r2 �mock_open_content� s � "�r� c # sR � � ��fdd�}t �tjdd|d�� }|V W d � dS 1 s"w Y dS )zZMock open() builtin and raises `exc` if the path being opened matches `for_path`. c s" | �kr� ��| g|�R i |��S ro r1 �r� r� r� ��exc�for_pathr� r1 r2 r� � s z&mock_open_exception.<locals>.open_mockr� Tr� Nr� )r� r� r� r� r1 r� r2 �mock_open_exception� s �"�r� z LINUX only��reasonc @ sL e Zd Zdd� Ze� dd� �Ze� dd� �Ze� dd� �Ze� d d � �Zd S ) �"TestSystemVirtualMemoryAgainstFreec C s" t � j}t�� j}||ksJ �d S ro )rv rj �psutil�virtual_memory��self� cli_value� psutil_valuer1 r1 r2 � test_total� s z-TestSystemVirtualMemoryAgainstFree.test_totalc C sR t � dk r t�d��t � dkrt�d��t� j}t�� j}t|| �tk s'J �d S )N�� r� � �free version too old�r: r r �free version too recent) r� r r} rv rk r� r� �absr r� r1 r1 r2 � test_used� s z,TestSystemVirtualMemoryAgainstFree.test_usedc C �* t � j}t�� j}t|| �tk sJ �d S ro )rv r[ r� r� r� r r� r1 r1 r2 � test_free� � z,TestSystemVirtualMemoryAgainstFree.test_freec C sV t � }|j}|dkrt�d��t�� j}t|| �tk s)J |� d|� d|j� ���d S )Nr z%free does not support 'shared' columnrx z ) rv ru r r} r� r� r� r �output)r� r[ � free_valuer� r1 r1 r2 � test_shared s �z.TestSystemVirtualMemoryAgainstFree.test_sharedc C sb t ddg�}|�d�}d|d vrt�d��t|d �� d �}t�� j}t|| �t k s/J �d S ) Nr[ r\ ra � availabler z(free does not support 'available' columnrs r9 ) r rB r r} rf r� r� r� r� r )r� rg rh r� r� r1 r1 r2 �test_available s z1TestSystemVirtualMemoryAgainstFree.test_availableN) �__name__� __module__� __qualname__r� r r� r� r� r� r1 r1 r1 r2 r� � s r� c @ sZ e Zd Zdd� Ze� dd� �Ze� dd� �Ze� dd� �Ze� d d � �Ze� d d � �Z d S )�$TestSystemVirtualMemoryAgainstVmstatc C �. t d�d }t�� j}t|| �tk sJ �d S )Nz total memory� )rw r� r� rj r� r �r� � vmstat_valuer� r1 r1 r2 r� s z/TestSystemVirtualMemoryAgainstVmstat.test_totalc C sV t � dk r t�d��t � dkrt�d��td�d }t�� j}t|| �tk s)J �d S )Nr� r� r� r� z used memoryr� ) r� r r} rw r� r� rk r� r r� r1 r1 r2 r� # s z.TestSystemVirtualMemoryAgainstVmstat.test_usedc C r� )Nz free memoryr� )rw r� r� r[ r� r r� r1 r1 r2 r� 5 � z.TestSystemVirtualMemoryAgainstVmstat.test_freec C r� )Nz buffer memoryr� )rw r� r� �buffersr� r r� r1 r1 r2 � test_buffers; r� z1TestSystemVirtualMemoryAgainstVmstat.test_buffersc C r� )Nz active memoryr� )rw r� r� �activer� r r� r1 r1 r2 � test_activeA r� z0TestSystemVirtualMemoryAgainstVmstat.test_activec C r� )Nzinactive memoryr� )rw r� r� �inactiver� r r� r1 r1 r2 � test_inactiveG r� z2TestSystemVirtualMemoryAgainstVmstat.test_inactiveN) r� r� r� r� r r� r� r� r� r� r1 r1 r1 r2 r� s r� c @ sB e Zd Zdd� Ze� dd� �Zdd� Zdd� Zd d � Zd d � Z d S )� TestSystemVirtualMemoryMocksc C s� t �d��� }td|i���}tjdd���}t�d� t�� }|j s#J �t |�dks+J �|d }dt |j �v s8J �d t |j �v sAJ �d t |j �v sJJ �d t |j �v sSJ �d t |j �v s\J �d t |j �v seJ �dt |j �v snJ �|j dksuJ �|jdks|J �|jdks�J �|jdks�J �|jdks�J �|jdks�J �|jdks�J �W d � n1 s�w Y W d � d S W d � d S 1 s�w Y d S )NaL Active(anon): 6145416 kB Active(file): 2950064 kB Inactive(anon): 574764 kB Inactive(file): 1567648 kB MemAvailable: -1 kB MemFree: 2057400 kB MemTotal: 16325648 kB SReclaimable: 346648 kB � /proc/meminfoT��record�alwaysrs r z#memory stats couldn't be determined�cachedru r� r� r� r� )�textwrap�dedent�encoder� �warnings�catch_warnings� simplefilterr� r� �calledrD r� �messager� r� r� ru r� r� �slab�r� r� r� �ws�ret�wr1 r1 r2 �test_warnings_on_missesP s8 � ��"�z4TestSystemVirtualMemoryMocks.test_warnings_on_missesc C s� i }t d�� }|D ]}|�� }t|d �d ||d < q W d � n1 s&w Y t|�}d|v rG|d }t|| �| d }|dk sIJ �d S d S )Nr� rs r� r s MemAvailable:�d r )r rB rf r r� )r� �memsrK rM rN �a�b� diff_percentr1 r1 r2 �test_avail_old_percentt s � � �z3TestSystemVirtualMemoryMocks.test_avail_old_percentc C � t �d��� }td|i��<}tjdd�� }t�� }W d � n1 s#w Y |js-J �|j dks4J �|d }dt |j �v sAJ �W d � d S 1 sLw Y d S )Na Active: 9444728 kB Active(anon): 6145416 kB Active(file): 2950064 kB Buffers: 287952 kB Cached: 4818144 kB Inactive(file): 1578132 kB Inactive(anon): 574764 kB Inactive(file): 1567648 kB MemAvailable: 6574984 kB MemFree: 2057400 kB MemTotal: 16325648 kB Shmem: 577588 kB SReclaimable: 346648 kB r� Tr� l �" r �,inactive memory stats couldn't be determined� r� r� r� r� r� r� r� r� r� r� r� r� r� r1 r1 r2 � test_avail_old_comes_from_kernel� s � � �"�z=TestSystemVirtualMemoryMocks.test_avail_old_comes_from_kernelc C r� )Nat Active: 9444728 kB Active(anon): 6145416 kB Buffers: 287952 kB Cached: 4818144 kB Inactive(file): 1578132 kB Inactive(anon): 574764 kB MemFree: 2057400 kB MemTotal: 16325648 kB Shmem: 577588 kB r� Tr� � `LG r r� r� r� r1 r1 r2 � test_avail_old_missing_fields� s � � �"�z:TestSystemVirtualMemoryMocks.test_avail_old_missing_fieldsc C s� t �d��� }td|i��T tdt��6 tjdd�� }t� � }|j dks&J �|d }dt |j �v s3J �W d � n1 s=w Y W d � n1 sLw Y W d � d S W d � d S 1 sdw Y d S ) Na� Active: 9444728 kB Active(anon): 6145416 kB Active(file): 2950064 kB Buffers: 287952 kB Cached: 4818144 kB Inactive(file): 1578132 kB Inactive(anon): 574764 kB Inactive(file): 1567648 kB MemFree: 2057400 kB MemTotal: 16325648 kB Shmem: 577588 kB SReclaimable: 346648 kB r� z/proc/zoneinfoTr� r� r r� ) r� r� r� r� r� �FileNotFoundErrorr� r� r� r� r� r� r� )r� r� r� r� r� r1 r1 r2 � test_avail_old_missing_zoneinfo� s$ � � �� ��"�z<TestSystemVirtualMemoryMocks.test_avail_old_missing_zoneinfoc C s� t �d��� }td|i��Q}t�� }|jsJ �|jdks J �|jdks%J �|j dks,J �|j dks3J �|j dks:J �|j dksAJ �|j d ksHJ �|jd ksOJ �|jd ksVJ �W d � d S 1 saw Y d S ) Na� MemTotal: 100 kB MemFree: 2 kB MemAvailable: 3 kB Buffers: 4 kB Cached: 5 kB SwapCached: 6 kB Active: 7 kB Inactive: 8 kB Active(anon): 9 kB Inactive(anon): 10 kB Active(file): 11 kB Inactive(file): 12 kB Unevictable: 13 kB Mlocked: 14 kB SwapTotal: 15 kB SwapFree: 16 kB Dirty: 17 kB Writeback: 18 kB AnonPages: 19 kB Mapped: 20 kB Shmem: 21 kB Slab: 22 kB SReclaimable: 23 kB SUnreclaim: 24 kB KernelStack: 25 kB PageTables: 26 kB NFS_Unstable: 27 kB Bounce: 28 kB WritebackTmp: 29 kB CommitLimit: 30 kB Committed_AS: 31 kB VmallocTotal: 32 kB VmallocUsed: 33 kB VmallocChunk: 34 kB HardwareCorrupted: 35 kB AnonHugePages: 36 kB ShmemHugePages: 37 kB ShmemPmdMapped: 38 kB CmaTotal: 39 kB CmaFree: 40 kB HugePages_Total: 41 kB HugePages_Free: 42 kB HugePages_Rsvd: 43 kB HugePages_Surp: 44 kB Hugepagesize: 45 kB DirectMap46k: 46 kB DirectMap47M: 47 kB DirectMap48G: 48 kB r� i � i � i p i T i i i X � )r� r� r� r� r� r� r� rj r[ r� r� ru r� r� r� r� )r� r� r� �memr1 r1 r2 �test_virtual_memory_mocked� s 1�2 "�z7TestSystemVirtualMemoryMocks.test_virtual_memory_mockedN) r� r� r� r� r r� r� r� r� r� r1 r1 r1 r2 r� N s $ r� c @ s\ e Zd Zedd� �Zdd� Ze� dd� �Ze� dd� �Zd d � Z d d � Z d d� Z dd� Z dS )�TestSystemSwapMemoryc C s@ t d�� } | �� }W d � n1 sw Y d|v o d|v S )z3Return True if /proc/meminfo provides swap metrics.r� Nz SwapTotal:z SwapFree:)rA �read)rK �datar1 r1 r2 �meminfo_has_swap_info s �z*TestSystemSwapMemory.meminfo_has_swap_infoc C r� ro )rm rj r� � swap_memoryr� r �r� r� r� r1 r1 r2 r� & � z TestSystemSwapMemory.test_totalc C r� ro )rm rk r� r� r� r r� r1 r1 r2 r� + r� z TestSystemSwapMemory.test_usedc C r� ro )rm r[ r� r� r� r r� r1 r1 r2 r� 1 r� z TestSystemSwapMemory.test_freec C s� t jddd��X}tjdd��9}t�d� t�� }|js J �t|�dks%J �|d }dt |j �v s2J �|j dks9J �|j dks@J �W d � n1 sJw Y W d � d S W d � d S 1 sbw Y d S ) N�psutil._common.openT�r� r� r� rs r z9'sin' and 'sout' swap memory stats couldn't be determined) r r� r� r� r� r� r� r� rD r� r� �sin�sout�r� r� r� r� r� r1 r1 r2 �test_missing_sin_sout7 s � ��"�z*TestSystemSwapMemory.test_missing_sin_soutc C s� t dt��X}tjdd��9}t�d� t�� }|jsJ �t|�dks#J �|d }dt |j �v s0J �|j dks7J �|j dks>J �W d � n1 sHw Y W d � d S W d � d S 1 s`w Y d S )Nz /proc/vmstatTr� r� rs r zK'sin' and 'sout' swap memory stats couldn't be determined and were set to 0) r� r� r� r� r� r� r� r� rD r� r� r� r� r� r1 r1 r2 �test_no_vmstat_mockedF s � ��"�z*TestSystemSwapMemory.test_no_vmstat_mockedc C s� | � � s t�d��t�d�� }t�� }W d � n1 s w Y |jr'J �dd lm } |� � \}}}}}}}||9 }||9 }|j |ksGJ �t |j | �tk sRJ �d S )Nz!/proc/meminfo has no swap metricsz"psutil._pslinux.cext.linux_sysinfor )r� r r} r r� r� r� r� �psutil._psutil_linux� _psutil_linux� linux_sysinforj r� r[ r )r� r� �swap�cextri rj r[ �unit_multiplierr1 r1 r2 � test_meminfo_against_sysinfoW s � z1TestSystemSwapMemory.test_meminfo_against_sysinfoc C sD t ddi��}t�� |jsJ �W d � d S 1 sw Y d S )Nr� � )r� r� r� r� �r� r� r1 r1 r2 �#test_emulate_meminfo_has_no_metricsh s "�z8TestSystemSwapMemory.test_emulate_meminfo_has_no_metricsN) r� r� r� � staticmethodr� r� r r� r� r� r� r r r1 r1 r1 r2 r� s r� c @ � e Zd Zdd� ZdS )�TestSystemCPUTimesc C s� t �� j}t�dt�� d �d }ttt |� d���}|dkr&d|v s%J �nd|vs,J �|dkr7d|v s6J �nd|vs=J �|d krId |v sGJ �d S d |vsOJ �d S ) Nz \d+\.\d+\.\d+� r �.)r � � �steal)r r r" �guest)r� r r � guest_nice) r� � cpu_times�_fieldsr� r� �os�unamer~ r rf rB )r� rN � kernel_ver�kernel_ver_infor1 r1 r2 � test_fieldsx s z TestSystemCPUTimes.test_fieldsN)r� r� r� r r1 r1 r1 r2 r v � r c @ s� e Zd Zejjej�d� dd�dd� �Z ejjej�d� dd�dd � �Z ejje � d � d d�d d � �Z ejje � d� dd�dd� �Zdd� ZdS )�TestSystemCPUCountLogical� /sys/devices/system/cpu/onlinez-/sys/devices/system/cpu/online does not existr� c C sn t d��}|�� �� }W d � n1 sw Y dt|�v r3t|�d�d �d }t�� |ks5J �d S d S )Nr �-rs )rA r� rz r� rf rB r� � cpu_count)r� rK �valuer1 r1 r2 � test_against_sysdev_cpu_online� s � �z8TestSystemCPUCountLogical.test_against_sysdev_cpu_online�/sys/devices/system/cpuz&/sys/devices/system/cpu does not existc C s0 t �d�}tdd� |D ��}t�� |ksJ �d S )Nr c S s g | ] }t �d |�dur|�qS )zcpu\d+$N)r� �searchrp r1 r1 r2 r? � s zITestSystemCPUCountLogical.test_against_sysdev_cpu_num.<locals>.<listcomp>)r �listdirrD r� r )r� �ls�countr1 r1 r2 �test_against_sysdev_cpu_num� s z5TestSystemCPUCountLogical.test_against_sysdev_cpu_num�nprocznproc utility not availablec C s$ t td��}tjdd�|ksJ �d S )Nz nproc --allT��logical)rf r r� r )r� �numr1 r1 r2 �test_against_nproc� s z,TestSystemCPUCountLogical.test_against_nproc�lscpu�lscpu utility not availablec C s8 t d�}tdd� |�d�D ��}tjdd�|ksJ �d S )N�lscpu -pc S s g | ] }|� d �s|�qS )�#�rc rp r1 r1 r2 r? � s z@TestSystemCPUCountLogical.test_against_lscpu.<locals>.<listcomp>ra Tr# )r rD rB r� r )r� rg r% r1 r1 r2 �test_against_lscpu� s z,TestSystemCPUCountLogical.test_against_lscpuc C s� dd l }|j�� }tjdtd���}|j�� |ksJ �|js J �tjddd��#}|j�� d u s0J �|jdks7J �|jd d dksBJ �W d � n1 sLw Y t d d �� }|� � }W d � n1 sew Y t � |�}tjd|dd ��}|j�� |ks�J �W d � n1 s�w Y t d d i��}|j�� |ks�J �|js�J �W d � n1 s�w Y W d � d S W d � d S 1 s�w Y d S ) Nr zpsutil._pslinux.os.sysconf�r� r� Tr� r � /proc/stat� /proc/cpuinfo�rb�� return_valuer� r )�psutil._pslinux�_pslinux�cpu_count_logicalr r� rE r� � call_count� call_argsrA r� r� r� r� )r� r� �originalr� rK � cpuinfo_data� fake_filer1 r1 r2 �test_emulate_fallbacks� s< � � � � � ��"�z0TestSystemCPUCountLogical.test_emulate_fallbacksN)r� r� r� r �mark�skipifr �path�existsr r! �shutil�whichr& r, r; r1 r1 r1 r2 r � s( � � � � r c @ sT e Zd Zejje�d� dd�dd� �Zejje � � dvdd�dd � �Z d d � Z d S ) �TestSystemCPUCountCoresr' r( r� c C s\ t d�}t� }|�d�D ]}|�d�s |�d�}|�|d � q tjdd�t|�ks,J �d S )Nr) ra r* �,rs Fr# )r �setrB rc �addr� r rD )r� rg �core_idsrM rN r1 r1 r2 r, � s � z*TestSystemCPUCountCores.test_against_lscpu> �i686�x86_64zx86_64/i686 onlyc C sh t j�� }tjdg d��}t j�� }|jsJ �W d � n1 s!w Y |d ur0||ks2J �d S d S )N� glob.glob�r2 )r� r4 �cpu_count_coresr r� r� )r� �meth_1r� �meth_2r1 r1 r2 � test_method_2� s � �z%TestSystemCPUCountCores.test_method_2c C s� t jdg d��(}t jddd��}tj�� d u sJ �W d � n1 s#w Y W d � n1 s2w Y |js<J �|jsAJ �d S )NrI rJ r� Tr� )r r� r� r4 rK r� )r� �m1�m2r1 r1 r2 �test_emulate_none� s �� � z)TestSystemCPUCountCores.test_emulate_noneN) r� r� r� r r< r= r@ rA r, �platform�machinerN rQ r1 r1 r1 r2 rB � s � � rB c @ s� e Zd Zejje dd�ejjedd�dd� ��Zejje dd�ejjedd�dd� ��Z ejje dd�d d � �Z ejje dd�d d � �Z ejje dd�d d� �Z dS )�TestSystemCPUFrequency� not supportedr� z(aarch64 does not always expose frequencyc sV � fdd�}t jj� tjd|dd�� t�� sJ �W d � d S 1 s$w Y d S )Nc � | � d�rdS � | �S )N�&/sys/devices/system/cpu/cpufreq/policyFr+ �r> �� orig_existsr1 r2 �path_exists_mock� � zMTestSystemCPUFrequency.test_emulate_use_second_file.<locals>.path_exists_mock�os.path.existsT)r� r� )r r> r? r r� r� �cpu_freq)r� r[ r1 rY r2 � test_emulate_use_second_file� s �"�z3TestSystemCPUFrequency.test_emulate_use_second_filez,aarch64 does not report mhz in /proc/cpuinfoc s� � fdd�}t jj� zctjd|d��< ttj� t�� }|s"J |��|j dks)J �|j dks0J �tjdd�D ]}|j dks?J �|j dksFJ �q6W d � n1 sQw Y W ttj� tt� d S W ttj� tt� d S ttj� tt� w )Nc rV )Nz/sys/devices/system/cpu/Fr+ rX ��os_path_existsr1 r2 r[ r\ zITestSystemCPUFrequency.test_emulate_use_cpuinfo.<locals>.path_exists_mockr] r- � T��percpu) r r> r? r r� r r� r4 r^ �max�min)r� r[ r� �freqr1 r` r2 �test_emulate_use_cpuinfo s* � � � � z/TestSystemCPUFrequency.test_emulate_use_cpuinfoc s� � fdd�}t � tjd|d��K tjddd��+ t�� }|jdks#J �|jd kr/|jd ks/J �|jd kr;|jd ks;J �W d � n1 sEw Y W d � d S W d � d S 1 s]w Y d S ) Nc s� | � d�r| �d�rt�d�S | � d�r | �d�r t�d�S | � d�r-| �d�r-t�d�S | dkr6t�d �S � | g|�R i |��S ) N�/scaling_cur_freqrW � 500000�/scaling_min_freq� 600000�/scaling_max_freqs 700000r/ s cpu MHz : 500��endswithrc r� r� r� �r� r1 r2 r� ' s � � � z;TestSystemCPUFrequency.test_emulate_data.<locals>.open_mockr� r- r] TrJ � @@rb � ��@g �@�rA r r� r� r^ �currentrf re �r� r� rg r1 rp r2 �test_emulate_data% s � ��"�z(TestSystemCPUFrequency.test_emulate_datac s` � fdd�}t � tjd|d��� tjddd��w tjdd d��` tjdd �}|d jd ks/J �|d jd kr?|d jdks?J �|d jd krO|d jdksOJ �|d jdksXJ �|d jd krh|d jdkshJ �|d jd krx|d jdksxJ �W d � n1 s�w Y W d � n1 s�w Y W d � d S W d � d S 1 s�w Y d S )Nc s� | }|� d�r|�d�rt�d�S |� d�r |�d�r t�d�S |� d�r/|�d�r/t�d�S |� d�r>|�d�r>t�d �S |� d�rM|�d�rMt�d �S |� d�r\|�d�r\t�d �S | d kret�d �S � | g|�R i |��S )Nri z'/sys/devices/system/cpu/cpufreq/policy0s 100000rk � 200000rm s 300000z'/sys/devices/system/cpu/cpufreq/policy1s 400000rj rl r/ s# cpu MHz : 100 cpu MHz : 400rn )r� r� r� �nrp r1 r2 r� G s8 � � � � � � z@TestSystemCPUFrequency.test_emulate_multi_cpu.<locals>.open_mockr� r- r] TrJ �!psutil._pslinux.cpu_count_logicalr rc r � Y@rb g i@g �r@rs g y@rq rr rs ru r1 rp r2 �test_emulate_multi_cpuE s2 � � �� ��"�z-TestSystemCPUFrequency.test_emulate_multi_cpuc s� � fdd�}t � tjd|d��J tjddd��* tjdd d�� t�� }|jd ks+J �W d � n1 s5w Y W d � n1 sDw Y W d � d S W d � d S 1 s\w Y d S ) Nc sJ | � d�rt�| � d�rt�d�S | dkrt�d�S � | g|�R i |��S )Nri z/cpuinfo_cur_freqrw r/ s cpu MHz : 200)ro r� r� r� r� rp r1 r2 r� { s zOTestSystemCPUFrequency.test_emulate_no_scaling_cur_freq_file.<locals>.open_mockr� r- r] TrJ ry rs �� )rA r r� r� r^ rt ru r1 rp r2 �%test_emulate_no_scaling_cur_freq_filex s � �� ��"�z<TestSystemCPUFrequency.test_emulate_no_scaling_cur_freq_fileN) r� r� r� r r< r= r r r_ rh rv r{ r} r1 r1 r1 r2 rT � s" � � 2rT c @ r )�TestSystemCPUStatsc C s* t d�}t�� j}t|| �dk sJ �d S )N� interruptsi� )rw r� � cpu_statsr r� r� r1 r1 r2 �test_interrupts� r� z"TestSystemCPUStats.test_interruptsN)r� r� r� r� r1 r1 r1 r2 r~ � s r~ c @ s& e Zd Zejje dd�dd� �ZdS )� TestLoadAvgrU r� c C s� t �� }td��}|�� �� }W d � n1 sw Y tt|d �|d �dk s.J �tt|d �|d �dk s>J �tt|d �|d �dk sNJ �d S )Nz /proc/loadavgr rs r )r� � getloadavgrA r� rB r� �float)r� r� rK � proc_valuer1 r1 r2 �test_getloadavg� s � $zTestLoadAvg.test_getloadavgN)r� r� r� r r< r= r r� r1 r1 r1 r2 r� � s r� c @ r )�TestSystemNetIfAddrsc C s� t �� �� D ]_\}}|D ]X}|jt jkr |jt|�ks J �q |jtjkrN|jt |�ks-J �|j t |�ks6J �|j d urE|j t |�ksDJ �q t |�dksMJ �q |jtjkrd|j�d�d }|t|�v sdJ �q qd S )Nz0.0.0.0�%r )r� � net_if_addrs�items�family�AF_LINK�addressrZ r% r& r3 �netmaskr5 � broadcastr8 rI rB rS )r� r� �addrs�addrr� r1 r1 r2 �test_ips� s ���z TestSystemNetIfAddrs.test_ipsN)r� r� r� r� r1 r1 r1 r2 r� � r r� c @ sT e Zd Zejje�d� dd�dd� �Zdd� Z ejje�d� dd�dd � �Z d S ) �TestSystemNetIfStats�ifconfig� ifconfig utility not availabler� c C st t �� �� D ]1\}}z td|� ��}W n ty Y qw |jd|v ks(J |��|jtt� d|�d �ks7J �qd S )N� ifconfig �RUNNINGz(?i)MTU[: ](\d+)r ) r� � net_if_statsr� r � RuntimeError�isup�mturf r� r� )r� r� �statsrg r1 r1 r2 �test_against_ifconfig� s � ��z*TestSystemNetIfStats.test_against_ifconfigc C sd t �� �� D ])\}}td|� d���}|jt|�� �� �ks J �W d � n1 s*w Y qd S )Nz/sys/class/net/z/mtu)r� r� r� rA r� rf r� rz )r� r� r� rK r1 r1 r2 �test_mtu� s ���z TestSystemNetIfStats.test_mtuc C s d}t �� �� D ]s\}}z td|� ��}W n ty Y qw t�d|�}|rNt|�� �dkrN|d7 }t |� d�� � � d��}t |j � d��}||ksMJ �qt�d|�}|r{t|�� �dkr{|d7 }t |� d�� � � � �}t |j � d��}||ks{J �q|s�| �d ��d S ) Nr r� zflags=(\d+)?<(.*?)>r rs rC z (.*) MTU:(\d+) Metric:(\d+)r� zno matches were found)r� r� r� r r� r� r rD rP rD �group�lowerrB �flags�fail)r� � matches_foundr� r� rg �match�ifconfig_flags� psutil_flagsr1 r1 r2 � test_flags� s. � � �z TestSystemNetIfStats.test_flagsN) r� r� r� r r< r= r@ rA r� r� r� r1 r1 r1 r2 r� � s � �r� c @ s2 e Zd Zejje�d� dd�e� dd� ��Z dS )�TestSystemNetIOCountersr� r� r� c C s dd� }t jddd�}|�� D ]|\}}z||�}W n ty" Y qw t|j|d �dk s0J �t|j|d �dk s=J �t|j|d �d k sJJ �t|j|d �d k sWJ �t|j |d �d k sdJ �t|j |d �d k sqJ �t|j |d �d k s~J �t|j |d �d k s�J �qd S )Nc S s� i }t d| � ��}tt�d|�d �|d< tt�d|�d �|d< tt�d|�d �|d< tt�d|�d �|d < tt�d |�d �|d < tt�d |�d �|d < tt�d|�d �|d< tt�d|�d �|d< |S )Nr� zRX packets[: ](\d+)r � packets_recvzTX packets[: ](\d+)� packets_sentzerrors[: ](\d+)�errinrs �erroutzdropped[: ](\d+)�dropin�dropoutz#RX (?:packets \d+ +)?bytes[: ](\d+)� bytes_recvz#TX (?:packets \d+ +)?bytes[: ](\d+)� bytes_sent)r rf r� r� )�nicr� rg r1 r1 r2 r� s& ����z?TestSystemNetIOCounters.test_against_ifconfig.<locals>.ifconfigTF)�pernic�nowrapr� i ( r� r� r� r� r� � r� r� r� ) r� �net_io_countersr� r� r� r� r� r� r� r� r� r� r� )r� r� �nior� r� � ifconfig_retr1 r1 r2 r� s" � �z-TestSystemNetIOCounters.test_against_ifconfigN) r� r� r� r r< r= r@ rA r r� r1 r1 r1 r2 r� s �r� c @ s8 e Zd Zejded�ejddd�dd� ��Zdd � Zd S ) �TestSystemNetConnectionsz psutil._pslinux.socket.inet_ntopr- z psutil._pslinux.supports_ipv6FrJ c C sN zt � t jt j�}| �|j� |�d� W n ty Y nw tjdd� d S )N)z::1r �inet6��kind) r% rI � SOCK_STREAM� addCleanup�close�bind�OSErrorr� �net_connections)r� � supports_ipv6rJ r0 r1 r1 r2 � test_emulate_ipv6_unsupportedF s �z6TestSystemNetConnections.test_emulate_ipv6_unsupportedc C sR t �d�}td|i��}tjdd� |jsJ �W d � d S 1 s"w Y d S )Na 0: 00000003 000 000 0001 03 462170 @/tmp/dbus-Qw2hMPIU3n 0: 00000003 000 000 0001 03 35010 @/tmp/dbus-tB2X8h69BQ 0: 00000003 000 000 0001 03 34424 @/tmp/dbus-cHy80Y8O 000000000000000000000000000000000000000000000000000000 z/proc/net/unix�unixr� )r� r� r� r� r� r� )r� r� r� r1 r1 r2 �test_emulate_unixR s "�z*TestSystemNetConnections.test_emulate_unixN)r� r� r� r r� rE r� r� r1 r1 r1 r2 r� D s r� c @ sB e Zd Zejjeed� dd�e� dd� ��Z dd� Z dd � Z d S ) �TestSystemDiskPartitions�statvfszos.statvfs() not availabler� c C sv dd� }t jdd�D ].}t �|j�}||j�\}}}}|j|ks"J �t|j| �tk s-J �t|j| �tk s8J �q d S )Nc S s~ t d| � d���� }|�d�}|�d� |�d�}|�� d d� \}}}}|dkr+d}t|�t|�t|�}}}||||fS )Nz df -P -B 1 "�"ra r r: �nonerT )r rz rB �poprf )r> rg rh rM �devrj rk r[ r1 r1 r2 �dfl s z4TestSystemDiskPartitions.test_against_df.<locals>.dfF)�all) r� �disk_partitions� disk_usage� mountpointrj r� r[ r rk )r� r� �part�usageri rj rk r[ r1 r1 r2 �test_against_dfe s �z(TestSystemDiskPartitions.test_against_dfc C s t d�� }|�� }W d � n1 sw Y d|v r+t�� D ] }|jdkr* d S q t�d�}tjd|dd��D}tjddgd ��#}t�� }|j sKJ �|j sPJ �|sTJ �|d jdks]J �W d � n1 sgw Y W d � d S W d � d S 1 sw Y d S ) Nz/proc/filesystems�zfsz nodev zfs r� Tr1 �$psutil._pslinux.cext.disk_partitions)z /dev/sdb3�/r� �rwrJ r ) rA r� r� r� �fstyper� r� r r� r� )r� rK r� r� r: rO rP r� r1 r1 r2 � test_zfs_fs~ s6 � � �� ��"�z$TestSystemDiskPartitions.test_zfs_fsc C s� zAt jddd��&}t�t�� t�� W d � n1 s w Y |js'J �W d � n1 s1w Y W dt_d S W dt_d S dt_w )Nzos.path.realpathz /non/existentrJ �/proc) r r� r �raisesr� r� r� r� � PROCFS_PATHr r1 r1 r2 �test_emulate_realpath_fail� s � � � �z3TestSystemDiskPartitions.test_emulate_realpath_failN) r� r� r� r r< r= �hasattrr r r� r� r� r1 r1 r1 r2 r� c s � r� c @ sD e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dS )�TestSystemDiskIoCountersc C � d}t d|i��q tjddd��Q tjdd�}|jdks J �|jd ks%J �|jd t ks.J �|j d ks5J �|j d ks<J �|j d ksCJ �|j dt ksLJ �|j dksSJ �|jdksZJ �W d � n1 sdw Y W d � d S W d � d S 1 s|w Y d S )Nz+ 3 0 1 hda 2 3 4 5 6 7 8 9 10 11 12�/proc/diskstats�!psutil._pslinux.is_storage_deviceTrJ F�r� rs r r� r: rt r � � r� �r� r r� r� �disk_io_counters� read_count�read_merged_count� read_bytes� SECTOR_SIZE� read_time� write_count�write_merged_count� write_bytes� write_time� busy_time�r� r� r� r1 r1 r2 �test_emulate_kernel_2_4� s& � ��"�z0TestSystemDiskIoCounters.test_emulate_kernel_2_4c C r� )Nz' 3 0 hda 1 2 3 4 5 6 7 8 9 10 11r� r� TrJ Fr� rs r r� r: rt r r� r� r� r� r� r1 r1 r2 � test_emulate_kernel_2_6_full� s& � ��"�z5TestSystemDiskIoCounters.test_emulate_kernel_2_6_fullc C s t ddi��q tjddd��Q tjdd�}|jdks J �|jd t ks%J �|jd ks,J �|j d t ks5J �|j d ks<J �|j d ksCJ �|j d ksJJ �|j d ksQJ �|jd ksXJ �W d � n1 sbw Y W d � d S W d � d S 1 szw Y d S ) Nr� z 3 1 hda 1 2 3 4r� TrJ Fr� rs r r� r: r )r� r r� r� r� r� r� r� r� r� r� r� r� r� r� )r� r� r1 r1 r2 � test_emulate_kernel_2_6_limited� s$ � ��"�z8TestSystemDiskIoCounters.test_emulate_kernel_2_6_limitedc C s� t �d�}td|i��[ tjddd��; tjddd�}t|�dks#J �|d jd ks,J �|d jd ks5J �|d j d ks>J �|d j d ksGJ �W d � n1 sQw Y W d � d S W d � d S 1 siw Y d S ) N�x 3 0 nvme0n1 1 2 3 4 5 6 7 8 9 10 11 3 0 nvme0n1p1 1 2 3 4 5 6 7 8 9 10 11 r� r� FrJ T��perdiskr� r �nvme0n1rs � nvme0n1p1rt ) r� r� r� r r� r� r� rD r� r� r� r1 r1 r2 � test_emulate_include_partitions� s � ��"�z8TestSystemDiskIoCounters.test_emulate_include_partitionsc C s, t �d�}td|i��, tjddd�� tjddd�}|d u s!J �W d � n1 s+w Y W d � n1 s:w Y dd� }t �d�}td|i��> tjdd |d �� tjddd�}|jd ksfJ �|jd ksmJ �W d � n1 sww Y W d � d S W d � d S 1 s�w Y d S ) Nr� r� r� FrJ r� c S � | dkS )Nr� r1 )r� r1 r1 r2 �is_storage_device � zSTestSystemDiskIoCounters.test_emulate_exclude_partitions.<locals>.is_storage_deviceTr� rs rt ) r� r� r� r r� r� r� r� r� )r� r� r� r� r1 r1 r2 � test_emulate_exclude_partitions� s2 � �� � � ��"�z8TestSystemDiskIoCounters.test_emulate_exclude_partitionsc C sh dd� }t jdd�}tjdd|d�� t jdd�}W d � n1 s#w Y t|�t|�ks2J �d S )Nc S r� )Nr� r1 rX r1 r1 r2 r? r� z?TestSystemDiskIoCounters.test_emulate_use_sysfs.<locals>.existsT)r� � psutil._pslinux.os.path.existsr� )r� r� r r� rD )r� r? �wprocfs�wsysfsr1 r1 r2 �test_emulate_use_sysfs s � �z/TestSystemDiskIoCounters.test_emulate_use_sysfsc C s� dd� }t jdd|d��* t�t�� t�� W d � n1 s!w Y W d � d S W d � d S 1 s9w Y d S )Nc S s dS )NFr1 rX r1 r1 r2 r? ! s z>TestSystemDiskIoCounters.test_emulate_not_impl.<locals>.existsr Tr� )r r� r r� �NotImplementedErrorr� r� )r� r? r1 r1 r2 �test_emulate_not_impl s � ��"�z.TestSystemDiskIoCounters.test_emulate_not_implN) r� r� r� r� r� r� r� r� r r r1 r1 r1 r2 r� � s r� c @ sl e Zd Zdd� Zdd� Zejjedd�dd� �Z ejje � d � d d�ejjedd�d d � ��Z d d� Z dS )�TestRootFsDeviceFinderc C s( t �d�j}t �|�| _t �|�| _d S )Nr� )r r{ �st_dev�major�minor)r� r� r1 r1 r2 �setUp- s z TestRootFsDeviceFinder.setUpc C s� t � }tj�d�r|�� nt�t�� |�� W d � n1 s"w Y tj�d| j� d| j � d��r;|� � nt�t�� |� � W d � n1 sOw Y |� � d S �Nz/proc/partitionsz/sys/dev/block/r@ z/uevent) r r r> r? �ask_proc_partitionsr r� r� r r �ask_sys_dev_block�ask_sys_class_block)r� �finderr1 r1 r2 �test_call_methods2 s � � z(TestRootFsDeviceFinder.test_call_methodsz unsupported on GITHUB_ACTIONSr� c C s� t � }|�� d us J �d } }}tj�d�r|�� }tj�d| j� d| j� d��r.|�� }|� � }|p7|p7|}|rB|rB||ksBJ �|rL|rL||ksLJ �|rV|rX||ksZJ �d S d S d S r ) r �findr r> r? r r r r r )r� r r� r� �c�baser1 r1 r2 �test_comparisons@ s" �z'TestRootFsDeviceFinder.test_comparisons�findmntz findmnt utility not availablec C s" t � �� }td�}||ksJ �d S )Nzfindmnt -o SOURCE -rn /)r r r )r� r� � findmnt_valuer1 r1 r2 �test_against_findmntT s z+TestRootFsDeviceFinder.test_against_findmntc C s� t jddgd��7}t�� d }|jsJ �ts(|jdks J �|jt� �� ks'J �n|jdks/J �W d � d S W d � d S 1 sBw Y d S )Nr� )� /dev/rootr� �ext4r� rJ r r ) r r� r� r� r� r �devicer r )r� r� r� r1 r1 r2 �test_disk_partitions_mocked] s � �"�z2TestRootFsDeviceFinder.test_disk_partitions_mockedN)r� r� r� r r r r< r= r r r@ rA r r r1 r1 r1 r2 r + s � r c @ sb e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Ze � e j j e d d�dd� ��Zdd� ZdS )�TestMiscc C s( t d�}t�� }t|�t|�ksJ �d S )Nz boot time)rw r� � boot_timerf r� r1 r1 r2 �test_boot_timer s zTestMisc.test_boot_timec sJ | � � }t�|� ttj�|d�d��}|�d� |�d� |�d� W d � n1 s-w Y �z]t� � fdd�}tjd|d ���9 t t � t � t �� t �� W d � n1 s]w Y t � t �� t jd d � W d � n1 sxw Y t � t �� t �� W d � n1 s�w Y t � t �� t jd d � W d � n1 s�w Y t � t �� t �� W d � n1 s�w Y t � t �� t jd d � W d � n1 s�w Y |t _t �� d ks�J �tt �� �d ks�J �t jd d �}t|�d k�s J �t jd d �}ttt|��d k�sJ �ttj�|d�d��}|�d � |�d� |�d� W d � n 1 �s@w Y t �� d k�sNJ �tt jd d ��d k�s[J �tt �� �d k�sfJ �tttt jd d ���d k�svJ �W d � n 1 �s�w Y W t�|� t t � n t�|� t t � w t jdk�s�J �d S )Nr{ r� zcpu 0 0 0 0 0 0 0 0 0 0 zcpu0 0 0 0 0 0 0 0 0 0 0 zcpu1 0 0 0 0 0 0 0 0 0 0 c s$ | � d�rt�� | g|�R i |��S )Nr� )rc r� r� rp r1 r2 r� � s z4TestMisc.test_no_procfs_on_import.<locals>.open_mockr� r- Trc r zcpu 1 0 0 0 0 0 0 0 0 0 zcpu0 1 0 0 0 0 0 0 0 0 0 zcpu1 1 0 0 0 0 0 0 0 0 0 r� )� get_testfnr �mkdirrA r> rG �writer r� r r� r r� r� r � cpu_percent�cpu_times_percentr� �sumr r@ �rmtree)r� � my_procfsrK r� �per_cpu_percent�per_cpu_times_percentr1 rp r2 �test_no_procfs_on_importw sj � � � � � � � �" �� + � z!TestMisc.test_no_procfs_on_importc C sb t �d��� }td|i��!}t�� |jsJ �tjdd� t�� tjdd� W d � n1 s1w Y t �d��� }td|i��d t�� }|jsMJ �tjdd�}t�� }tjdd�}|dkscJ �t|�dkskJ �t|�dkssJ �t|�dks{J �tt t|��dks�J �tt t|��dks�J �|j dks�J �|j dks�J �W d � d S 1 s�w Y d S )Nz~ cpu 0 0 0 0 0 0 0 1 0 0 cpu0 0 0 0 0 0 0 0 1 0 0 cpu1 0 0 0 0 0 0 0 1 0 0 r. Trc z~ cpu 1 0 0 0 0 0 0 0 0 0 cpu0 1 0 0 0 0 0 0 0 0 0 cpu1 1 0 0 0 0 0 0 0 0 0 r rz ) r� r� r� r� r� r" r� r# r$ r r �user)r� r� r� r" �cpu_percent_percpur# �cpu_times_percent_percpur1 r1 r2 �test_cpu_steal_decrease� s8 � � � "�z TestMisc.test_cpu_steal_decreasec C sr t jddd��(}t�t�� tj�� W d � n1 s w Y |js'J �W d � d S 1 s2w Y d S )Nr� Tr� ) r r� r r� r� r� r4 r r� r r1 r1 r2 �test_boot_time_mocked� s � "�z TestMisc.test_boot_time_mockedc C s t �� D ] }|jdvs J �qd S )N> �:0�:0.0)r� �users�host)r� r* r1 r1 r2 � test_users� s �zTestMisc.test_usersc C s | � � }t�|� z�|t_t�t�� t�� W d � n1 s!w Y t�t�� t� � W d � n1 s:w Y t�t�� tj dd� W d � n1 sUw Y t�t�� t� � W d � n1 snw Y t�t�� t� � W d � n1 s�w Y t�t�� t� � W d � n1 s�w Y t�t�� t� � W d � n1 s�w Y t�t�� t�� W d � n1 s�w Y t�tj�� t�� W d � n1 s�w Y W dt_d S W dt_d S dt_w )NTrc r� )r r r r� r� r r� r� r� r r r� r� r� r� � NoSuchProcess�Process)r� �tdirr1 r1 r2 �test_procfs_path� sD � � � � � � � � � �zTestMisc.test_procfs_pathzskip if pytest-parallelr� c C s� t � �< t�� }|�� }t|�dksJ �t|dd� d�d j}|j|ks&J �t�|�}|�� |t� � vs7J �W d � d S 1 sBw Y d S )Nr c S s | j S ro )�id)rq r1 r1 r2 �<lambda> s z)TestMisc.test_issue_687.<locals>.<lambda>)�keyrs ) r r� r5 �threadsrD �sortedr8 �pid�as_dict�pids)r� �pr; �tid�ptr1 r1 r2 �test_issue_687 s "�zTestMisc.test_issue_687c C sN t ddi��}t�t�� �sJ �|jsJ �W d � d S 1 s w Y d S )Nz/proc/%s/statusrT )r� r� � pid_existsr �getpidr� r r1 r1 r2 � test_pid_exists_no_proc_status s "�z'TestMisc.test_pid_exists_no_proc_statusN)r� r� r� r r) r- r. r3 r7 r r r< r= r rC rF r1 r1 r1 r2 r p s A' r z no batteryc @ sl e Zd Zejje�d� dd�dd� �Zdd� Z dd � Z d d � Z d d � Z dd� Z dd� Zdd� Zdd� ZdS )�TestSensorsBattery�acpizacpi utility not availabler� c C sH t d�}t|�d�d �� �dd��}t�� j}t|| �dk s"J �d S )Nzacpi -brC rs r� rT ) r rf rB rz �replacer� �sensors_battery�percentr� )r� rg � acpi_valuer� r1 r1 r2 � test_percent* s z TestSensorsBattery.test_percentc st � fdd�}t � tjd|d��!}t�� jdu sJ �t�� jtjks#J �|js(J �W d � d S 1 s3w Y d S )Nc �* | � d�r t�d�S � | g|�R i |��S )N�z AC0/onlinez AC/online� 1�ro r� r� r� rp r1 r2 r� 5 � z@TestSensorsBattery.test_emulate_power_plugged.<locals>.open_mockr� r- T) rA r r� r� rJ � power_plugged�secsleft�POWER_TIME_UNLIMITEDr� �r� r� r� r1 rp r2 �test_emulate_power_plugged3 s � "�z-TestSensorsBattery.test_emulate_power_pluggedc �` � fdd�}t � tjd|d��}t�� jdu sJ �|js J �W d � d S 1 s)w Y d S )Nc �8 | � d�rt�| � d�rt�d�S � | g|�R i |��S )NrO �/status�charging�ro r� r� r� r� rp r1 r2 r� G � zBTestSensorsBattery.test_emulate_power_plugged_2.<locals>.open_mockr� r- T�rA r r� r� rJ rS r� rV r1 rp r2 � test_emulate_power_plugged_2D � "�z/TestSensorsBattery.test_emulate_power_plugged_2c rX )Nc rN )NrO � 0rQ r� rp r1 r2 r� V rR zDTestSensorsBattery.test_emulate_power_not_plugged.<locals>.open_mockr� r- Fr^ rV r1 rp r2 � test_emulate_power_not_pluggedT s "�z1TestSensorsBattery.test_emulate_power_not_pluggedc rX )Nc rY )NrO rZ � dischargingr\ r� rp r1 r2 r� d r] zFTestSensorsBattery.test_emulate_power_not_plugged_2.<locals>.open_mockr� r- Fr^ rV r1 rp r2 � test_emulate_power_not_plugged_2a r` z3TestSensorsBattery.test_emulate_power_not_plugged_2c s` � fdd�}t � tjd|d��}t�� jd u sJ �|js J �W d � d S 1 s)w Y d S )Nc rY )N)�"/sys/class/power_supply/AC0/online�!/sys/class/power_supply/AC/online�#/sys/class/power_supply/BAT0/statuss ???)rc r� r� r� r� rp r1 r2 r� t s zETestSensorsBattery.test_emulate_power_undetermined.<locals>.open_mockr� r- r^ rV r1 rp r2 � test_emulate_power_undeterminedq s "�z2TestSensorsBattery.test_emulate_power_undeterminedc C sN t ddi��}t�� jdksJ �|jsJ �W d � d S 1 s w Y d S )N�(/sys/class/power_supply/BAT0/energy_fullra r )r� r� rJ rK r� r r1 r1 r2 �test_emulate_energy_full_0� s � "�z-TestSensorsBattery.test_emulate_energy_full_0c C s� t dt��E t dt��' tddi�� t�� jdks J �W d � n1 s&w Y W d � n1 s5w Y W d � d S W d � d S 1 sMw Y d S )Nri z(/sys/class/power_supply/BAT0/charge_fullz%/sys/class/power_supply/BAT0/capacitys 88�X )r� r� r� r� rJ rK �r� r1 r1 r2 �"test_emulate_energy_full_not_avail� s$ ��� �� ��"�z5TestSensorsBattery.test_emulate_energy_full_not_availc C s� t dt��D t dt��& t dt�� t�� jd u sJ �W d � n1 s%w Y W d � n1 s4w Y W d � d S W d � d S 1 sLw Y d S )Nrf re rg )r� r� r� rJ rS rl r1 r1 r2 �test_emulate_no_power� s" ��� �� ��"�z(TestSensorsBattery.test_emulate_no_powerN)r� r� r� r r< r= r@ rA rM rW r_ rb rd rh rj rm rn r1 r1 r1 r2 rG ' s � rG c @ r )�TestSensorsBatteryEmulatedc s� � fdd�}t � tjddgd��'}tjd|d��}t�� d us!J �W d � n1 s+w Y W d � n1 s:w Y |jsDJ �|jsIJ �d S )Nc �R | � d�r t�d�S | � d�rt�d�S | � d�r t�d�S � | g|�R i |��S )Nz /energy_now�60000000z /power_now�0z /energy_full�60000001�ro r� r� r� rp r1 r2 r� � � z5TestSensorsBatteryEmulated.test_it.<locals>.open_mockz os.listdir�BAT0rJ r� r- )rA r r� r� rJ r� )r� r� �mlistdir�mopenr1 rp r2 �test_it� s �� � z"TestSensorsBatteryEmulated.test_itN)r� r� r� ry r1 r1 r1 r2 ro � r ro c @ s e Zd Zdd� Zdd� ZdS )�TestSensorsTemperaturesc s� � fdd�}t � tjd|d��M tjddgd��, t�� d d }|jd ks(J �|jd ks/J �|jd ks6J �|jd ks=J �W d � n1 sGw Y W d � d S W d � d S 1 s_w Y d S )Nc sz | � d�r t�d�S | � d�rt�d�S | � d�r t�d�S | � d�r(t�d�S | � d �r2t�d �S � | g|�R i |��S ) N�/namer� z /temp1_label�labelz /temp1_input� 30000z /temp1_maxs 40000z /temp1_crit� 50000)ro r� r� r� r� rp r1 r2 r� � s zCTestSensorsTemperatures.test_emulate_class_hwmon.<locals>.open_mockr� r- rI z /sys/class/hwmon/hwmon0/temp1rJ r� r r| � >@g D@� I@� rA r r� r� �sensors_temperaturesr| rt �high�critical)r� r� �tempr1 rp r2 �test_emulate_class_hwmon� s � ��"�z0TestSensorsTemperatures.test_emulate_class_hwmonc s� � fdd�}dd� }t � tjd|d��M tjdd|d ��, t�� d d }|jd ks,J �|jd ks3J �|jdks:J �|jdksAJ �W d � n1 sKw Y W d � d S W d � d S 1 scw Y d S )Nc sf | � d�r t�d�S | � d�rt�d�S | � d�r t�d�S | � d�r(t�d�S � | g|�R i |��S ) N�0_tempr~ r� r} �0_typer� �typer� )ro r� r� r� r� rp r1 r2 r� � s zETestSensorsTemperatures.test_emulate_class_thermal.<locals>.open_mockc S s. | dv rg S | dkr dgS | dkrddgS g S )N> � /sys/class/hwmon/hwmon*/temp*_*�&/sys/class/hwmon/hwmon*/device/temp*_*z /sys/class/thermal/thermal_zone*z /sys/class/thermal/thermal_zone0z,/sys/class/thermal/thermal_zone0/trip_point*z2/sys/class/thermal/thermal_zone1/trip_point_0_typez2/sys/class/thermal/thermal_zone1/trip_point_0_tempr1 rX r1 r1 r2 � glob_mock� s �zETestSensorsTemperatures.test_emulate_class_thermal.<locals>.glob_mockr� r- rI Tr� r� r rT r r� r� )r� r� r� r� r1 rp r2 �test_emulate_class_thermal� s ��"�z2TestSensorsTemperatures.test_emulate_class_thermalN)r� r� r� r� r� r1 r1 r1 r2 rz � s rz c @ r )�TestSensorsFansc s� � fdd�}t � tjd|d��? tjddgd�� t�� d d }|jd ks(J �|jd ks/J �W d � n1 s9w Y W d � d S W d � d S 1 sQw Y d S ) Nc rp )Nr{ r� z /fan1_labelr| z /fan1_input�2000rt r� rp r1 r2 r� ru z4TestSensorsFans.test_emulate_data.<locals>.open_mockr� r- rI z /sys/class/hwmon/hwmon2/fan1rJ r� r r| i� )rA r r� r� � sensors_fansr| rt )r� r� �fanr1 rp r2 rv s � ��"�z!TestSensorsFans.test_emulate_dataN)r� r� r� rv r1 r1 r1 r2 r� r r� c @ s� e Zd Ze� dd� �Zdd� Zejje dd�dd� �Z d d � Z d d � Z d d� Z dd� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd � Zd d � Zd d � Zejje d!d�d"d#� �Zd$d%� Zd&d'� Zd(d)� Zd*S )+� TestProcessc C s� | � � }tj�|j��� \}}}t�|j�jdd�}t|tdd� |D �� �dk s+J �t|tdd� |D �� �dk s<J �t|tdd� |D �� �dk sMJ �d S )NF)�groupedc s s � | ] }|j |j V qd S ro )� private_dirty� private_cleanrp r1 r1 r2 rr ' s � z>TestProcess.test_parse_smaps_vs_memory_maps.<locals>.<genexpr>r� c s � � | ]}|j V qd S ro )�pssrp r1 r1 r2 rr * � � c s r� ro )r� rp r1 r1 r2 rr + r� ) �spawn_testprocr� r4 r5 r= � _parse_smaps� memory_mapsr� r$ )r� �sproc�ussr� r� �mapsr1 r1 r2 � test_parse_smaps_vs_memory_maps! s �"&z+TestProcess.test_parse_smaps_vs_memory_mapsc C s� t �d��� }tdt�� � d�|i��/}tj�t�� �}|� � \}}}|j s(J �|dks.J �|dks4J �|dks:J �W d � d S 1 sEw Y d S )Nan fffff0 r-xp 00000000 00:00 0 [vsyscall] Size: 1 kB Rss: 2 kB Pss: 3 kB Shared_Clean: 4 kB Shared_Dirty: 5 kB Private_Clean: 6 kB Private_Dirty: 7 kB Referenced: 8 kB Anonymous: 9 kB LazyFree: 10 kB AnonHugePages: 11 kB ShmemPmdMapped: 12 kB Shared_Hugetlb: 13 kB Private_Hugetlb: 14 kB Swap: 15 kB SwapPss: 16 kB KernelPageSize: 17 kB MMUPageSize: 18 kB Locked: 19 kB VmFlags: rd ex �/proc/�/smapsi l r� i < ) r� r� r� r� r rE r� r4 r5 r� r� )r� r� r� r@ r� r� r� r1 r1 r2 �test_parse_smaps_mocked- s � "�z#TestProcess.test_parse_smaps_mockedzunreliable on PYPYr� c C s dd� }| � � }t|d�� ||�jdksJ �W d � n1 s!w Y t|�� ||�jdks4J �W d � n1 s>w Y t|d�� ||�jdksRJ �W d � n1 s\w Y t|d�� ||�jdkspJ �W d � n1 szw Y t|d�� ||�jdks�J �W d � n1 s�w Y t|d�� ||�jdks�J �W d � n1 s�w Y t|� t|d �� ||�jdks�J �W d � n1 s�w Y t|� t|d �� ||�jdks�J �W d � d S 1 s�w Y d S ) Nc S sP t �� }t�� t } |�� D ]}|jtj�| �kr | S t�� |kr& nqq ro ) r� r5 �timer � open_filesr> r �abspathr� )�fnamer@ � giveup_at�filer1 r1 r2 � get_test_fileQ s ��z7TestProcess.test_open_files_mode.<locals>.get_test_filer� �rr� zr+zw+za+rq zx+)r rA �moder )r� r� �testfnr1 r1 r2 �test_open_files_modeO s8 � � � � � � � "�z TestProcess.test_open_files_modec s� t �� ���� � t| �� d��f t� �fdd�� tjdtd��}��� g ks)J �|j s.J �W d � n1 s8w Y tjdt t j d�d��}��� g ksQJ �|j sVJ �W d � n1 s`w Y W d � d S W d � d S 1 sxw Y d S )Nr� c � t ��� �t � �kS ro �rD r� r1 ��filesr@ r1 r2 r9 x � z7TestProcess.test_open_files_file_gone.<locals>.<lambda>�psutil._pslinux.os.readlinkr- rT ) r� r5 r� rA r r r r� r� r� r� �errno�EINVALr r1 r� r2 �test_open_files_file_goneq s. � � � �� "�z%TestProcess.test_open_files_file_gonec s� t �� ���� � t| �� d��> t� �fdd�� tjdtd��}��� g ks)J �|j s.J �W d � n1 s8w Y W d � d S W d � d S 1 sPw Y d S )Nr� c r� ro r� r1 r� r1 r2 r9 � r� z5TestProcess.test_open_files_fd_gone.<locals>.<lambda>r� r- ) r� r5 r� rA r r r r� r� r� r r1 r� r2 �test_open_files_fd_gone� s � ��"�z#TestProcess.test_open_files_fd_gonec s� t �� ���� � t| �� d��Y t� �fdd�� d}tj|tt j d�d��*}t�d�� ��� g ks5J �|j s:J �W d � n1 sDw Y W d � n1 sSw Y W d � d S W d � d S 1 skw Y d S )Nr� c r� ro r� r1 r� r1 r2 r9 � r� z:TestProcess.test_open_files_enametoolong.<locals>.<lambda>r� rT r- �psutil._pslinux.debug) r� r5 r� rA r r r r� r� r� � ENAMETOOLONGr� )r� � patch_pointr� r1 r� r2 � test_open_files_enametoolong� s$ � �� ��"�z(TestProcess.test_open_files_enametoolongc C sZ t jdi d�� }tj�t�� ��� d u sJ �|jsJ �W d � d S 1 s&w Y d S )Nz)psutil._pslinux._psposix.get_terminal_maprJ ) r r� r� r4 r5 r rE �terminalr� r r1 r1 r2 �test_terminal_mocked� s � "�z TestProcess.test_terminal_mockedc C � t �� }t�d�}tjd|dd��}|�� ddgks J �|js!J �W d � n1 s+w Y t�d�}tjd|dd��}|�� g d�ksHJ �|jsMJ �W d � d S 1 sXw Y d S ) Nzfoo bar r� Tr1 �foo�barz foo bar �r� r� rT �r� r5 r� r� r r� �cmdliner� �r� r@ r: r� r1 r1 r2 �test_cmdline_mocked� �" � � � "�z TestProcess.test_cmdline_mockedc C r� ) Nzfoo bar r� Tr1 r� r� z foo bar r� r� r� r1 r1 r2 �test_cmdline_spaces_mocked� r� z&TestProcess.test_cmdline_spaces_mockedc C sf t �� }t�d�}tjd|dd��}|�� ddgks J �|js!J �W d � d S 1 s,w Y d S )Nzfoo bar r� Tr1 r� r� r� r� r1 r1 r2 � test_cmdline_mixed_separators� s � "�z)TestProcess.test_cmdline_mixed_separatorsc C s\ t jddd�� t�� �� dksJ �t�� �� dks J �W d � d S 1 s'w Y d S )Nr� z/home/foo (deleted)rJ z /home/foo)r r� r� r5 �exe�cwdrl r1 r1 r2 �!test_readlink_path_deleted_mocked� s �"�z-TestProcess.test_readlink_path_deleted_mockedc s� � fdd�}t � tjd|d��}t�� �� }|jsJ �|g ks!J �W d � n1 s+w Y � fdd�}tjd|d��- t�tj �� t�� �� W d � n1 sUw Y W d � d S W d � d S 1 smw Y d S )Nc �0 | � dt�� � d��r t�� | g|�R i |��S �Nr� z/task)rc r rE r� r� rp r1 r2 � open_mock_1� � z4TestProcess.test_threads_mocked.<locals>.open_mock_1r� r- c r� r� )rc r rE �PermissionErrorr� rp r1 r2 � open_mock_2� r� z4TestProcess.test_threads_mocked.<locals>.open_mock_2) rA r r� r� r5 r; r� r r� � AccessDenied)r� r� r� r� r� r1 rp r2 �test_threads_mocked� s � ��"�z TestProcess.test_threads_mockedc C s� t jdtd��9}t jdg d�� t�� �� }|jsJ �|dks!J �W d � n1 s+w Y W d � d S W d � d S 1 sCw Y d S )Nzpsutil._pslinux.readlinkr- z psutil._pslinux.Process.cmdlinerJ rT )r r� r� r� r5 r� r� )r� r� r� r1 r1 r2 �test_exe_mocked s �� ��"�zTestProcess.test_exe_mockedc C s� t dt�� � d�t��+}t�� }t�t�� |�� W d � n1 s$w Y |j s.J �W d � d S 1 s9w Y d S )Nr� r� ) r� r rE r� r� r5 r r� r� r� �r� r� r@ r1 r1 r2 �test_issue_1014 s � � "�zTestProcess.test_issue_1014c C s� t �� }tdt�� � d�t��B tjddd��" t� t j �� |� � W d � n1 s-w Y W d � n1 s<w Y W d � d S W d � d S 1 sTw Y d S )Nr� z/statmr] FrJ ) r� r5 r� r rE r� r r� r r� r4 � memory_info)r� r@ r1 r1 r2 �test_issue_2418 s � �� ��"�zTestProcess.test_issue_2418rU c C s� t jdttjd�d��C}t jddd��,}t�� }|�� t� tj ��}|� tj � W d � n1 s3w Y W d � n1 sBw Y W d � n1 sQw Y |j s[J �|j s`J �|jj|jksiJ �|jj|�� kssJ �d S )Nzresource.prlimitrT r- z"psutil._pslinux.Process._is_zombieTrJ )r r� r� r� �ENOSYSr� r5 r� r r� � ZombieProcess�rlimit� RLIMIT_NOFILEr� r r= )r� rO rP r@ �cmr1 r1 r2 �test_rlimit_zombie& s* �� �� �� � z TestProcess.test_rlimit_zombiec C s g d�}d� |��� }tdt�� � d�|i��m t�� }|�� dks$J �|�� tj ks-J �|� � dks5J �|� � dt t� � ksCJ �|�� }|jdt ksPJ �|jd t ksYJ �|jd t ksbJ �|jd t kskJ �|jd t kstJ �|�� dks|J �W d � d S 1 s�w Y d S ) N)*rr z(cat)�Z�1rr rr rr rr rr rr rr rr rr �2�3�4�5rr rr rr rr �6rr rr rr rr rr rr rr rr rr rr rr rr rr rr rr rr r� rr rr �7rx r� z/stat�catrs r r r� r: rt r� )rG r� r� r rE r� r5 r� �status� STATUS_ZOMBIE�ppid� create_timer r r r* �system� children_user�children_system�iowait�cpu_num)r� r� r� r@ �cpur1 r1 r2 �test_stat_file_parsing: s , "�z"TestProcess.test_stat_file_parsingc C s t �d��� }tdt�� � d�|i��f t�� }|�� j dks!J �|�� j dks*J �|� � dks2J �|� � }|j dks=J �|jdksDJ �|jd ksKJ �|�� }|j d ksVJ �|jd ks]J �|jd ksdJ �|j�� ttd ��ksqJ �W d � d S 1 s|w Y d S )Nz� Uid: 1000 1001 1002 1003 Gid: 1004 1005 1006 1007 Threads: 66 Cpus_allowed: f Cpus_allowed_list: 0-7 voluntary_ctxt_switches: 12 nonvoluntary_ctxt_switches: 13r� rZ r� � �B i� i� i� i� i� i� r� )r� r� r� r� r rE r� r5 �num_ctx_switches� voluntary� involuntary� num_threads�uids�real� effective�saved�gids�_proc�_get_eligible_cpus�listrF )r� r� r@ r� r� r1 r1 r2 �test_status_file_parsingv s$ � "�z$TestProcess.test_status_file_parsingc C s� t jdttjd�d��7}t�� }t �d�� |�� g ks J �|js#J �W d � n1 s-w Y W d � d S W d � d S 1 sEw Y d S )Nr� rT r- r� ) r r� r� r� r� r� r5 r� r� r� r1 r1 r2 �!test_net_connections_enametoolong� s � ��"�z-TestProcess.test_net_connections_enametoolongN) r� r� r� r r� r� r r< r= r r� r� r� r� r� r� r� r� r� r� r� r� r� r r� r� r r r1 r1 r1 r2 r� s. " ! < r� c @ sr e Zd ZdZedd� �Zdd� Zdd� Zdd � Zd d � Z d d � Z dd� Z dd� Z e � dd� �Zdd� Zdd� ZdS )�TestProcessAgainstStatusa /proc/pid/stat and /proc/pid/status have many values in common. Whenever possible, psutil uses /proc/pid/stat (it's faster). For all those cases we check that the value found in /proc/pid/stat (by psutil) matches the one found in /proc/pid/status. c C s t �� | _d S ro )r� r5 �proc)�clsr1 r1 r2 � setUpClass� s z#TestProcessAgainstStatus.setUpClassc C s� t j�d| jj� d���@}|D ]5}|�� }|�|�rD|�d�d }zt|�W W d � S t yC | Y W d � S w qt d|����1 sOw Y d S )Nr� rZ � r ry ) r� � _psplatform� open_textr r= rz rc � partitionrf rE )r� � linestartrK rM r r1 r1 r2 �read_status_file� s$ � � � ��z)TestProcessAgainstStatus.read_status_filec C � | � d�}| j�� |ksJ �d S )NzName:)r r r� �r� r r1 r1 r2 � test_name� � z"TestProcessAgainstStatus.test_namec C sH | � d�}||�d�d |�d�� }|�dd�}| j�� |ks"J �d S )NzState:�(rs �)rx r )r r �rfindrI r r� r r1 r1 r2 � test_status� s z$TestProcessAgainstStatus.test_statusc C r )NzPPid:)r r r� r r1 r1 r2 � test_ppid� r z"TestProcessAgainstStatus.test_ppidc C r )NzThreads:)r r r� r r1 r1 r2 �test_num_threads� r z)TestProcessAgainstStatus.test_num_threadsc C �: | � d�}ttt|�� dd� ��}| j�� |ksJ �d S )NzUid:rs r: )r r~ r rf rB r r� r r1 r1 r2 � test_uids� � z"TestProcessAgainstStatus.test_uidsc C r )NzGid:rs r: )r r~ r rf rB r r� r r1 r1 r2 � test_gids� r z"TestProcessAgainstStatus.test_gidsc C s@ | � d�}| j�� j|ksJ �| � d�}| j�� j|ks J �d S )Nzvoluntary_ctxt_switches:znonvoluntary_ctxt_switches:)r r r� r� r� r r1 r1 r2 �test_num_ctx_switches� s z.TestProcessAgainstStatus.test_num_ctx_switchesc C sR | � d�}dt|�v r%tt|�d��\}}| j�� tt||d ��ks'J �d S d S )N�Cpus_allowed_list:r rs ) r r� r rf rB r � cpu_affinityr rF )r� r �min_�max_r1 r1 r2 �test_cpu_affinity� s �z*TestProcessAgainstStatus.test_cpu_affinityc C sh | � d�}t�d��}| jj�� W d � n1 sw Y dt|�v r-|jr+J �d S |js2J �d S )Nr z psutil._pslinux.per_cpu_timesr )r r r� r r� r� r� r� )r� r r� r1 r1 r2 � test_cpu_affinity_eligible_cpus� s � z8TestProcessAgainstStatus.test_cpu_affinity_eligible_cpusN)r� r� r� �__doc__� classmethodr r r r r r r r r r r r! r1 r1 r1 r2 r � s r c @ r )� TestUtilsc C sR t jddd��}tj�d�dksJ �|jsJ �W d � d S 1 s"w Y d S )Nz os.readlinkz foo (deleted)rJ r� r� )r r� r� r �readlinkr� r r1 r1 r2 � test_readlink� s "�zTestUtils.test_readlinkN)r� r� r� r& r1 r1 r1 r2 r$ � r r$ )]r" rd � contextlibr� r� r rR r� r@ r% r, r� r� r� �unittestr r� r � psutil.testsr r r r r r r r r r r r r r r r r r r r r3 r r r r r> r� �dirname�__file__�HEREr+ rX r4 r7 r� r3 r5 r8 rS rZ rm rv rw r� �contextmanagerr� r� r< r= r� r� r� r� r r rB rT r~ r� r� r� r� r� r� r� r r rG ro rz r� r� r r$ r1 r1 r1 r2 �<module> s� 61 OXI (63 ? D 7 B ~R tests/__pycache__/test_bsd.cpython-310.pyc 0000644 00000045535 15030211765 0014442 0 ustar 00 o �h�N � @ s� d Z ddlZddlZddlZddlZddlZddlZddlmZ ddlmZ ddlm Z ddlm Z ddl m Z ddl m Z dd l mZ dd l mZ dd l mZ dd l mZ dd l mZ ddl mZ eryddlmZ e� Ze�� dkowe�d�ZndZdZdd� Zdd� Zej j e dd�G dd� de��Z ej j e dd�G dd� de��Z ej j e dd�G d d � d e��Z ej j e d d�G d d � d e��Z!ej j e d!d�G d"d#� d#e��Z"dS )$z$Tests specific to all BSD platforms.� N)�BSD)�FREEBSD)�NETBSD)�OPENBSD)� HAS_BATTERY)�TOLERANCE_SYS_MEM)�PsutilTestCase)�pytest)�retry_on_failure)�sh)�spawn_testproc)� terminate)� getpagesize�museFc C sh t d| �}tr||�d�d d� }ntstr#||�d�d d� }zt|�W S ty3 | Y S w )zmExpects a sysctl command with an argument and parse the result returning only the value of interest. zsysctl z: � N�=� )r r �findr r �int� ValueError)�cmdline�result� r �H/usr/local/CyberCP/lib/python3.10/site-packages/psutil/tests/test_bsd.py�sysctl, s �r c C s> t d�}|�d�D ] }|�| �r nq td��t|�� d �S )z+Thin wrapper around 'muse' cmdline utility.r � zline not foundr )r �split� startswithr r )�field�out�liner r r r ; s �zBSD only��reasonc @ s� e Zd ZdZedd� �Zedd� �Zejj e dd�dd � �Z d d � Z ejj e �d � d d�dd� �Zejj e �d � d d�ejj e dd�dd� ��Zejj e �d� dd�dd� �ZdS )� BSDTestCasez)Generic tests common to all BSD variants.c C � t � j| _d S �N�r �pid��clsr r r � setUpClassO � zBSDTestCase.setUpClassc C � t | j� d S r% �r r' r( r r r � tearDownClassS r+ zBSDTestCase.tearDownClassz -o lstart doesn't work on NETBSDr! c C sR t d| j� ��}|�dd��� }t�| j��� }t�dt� |��}||ks'J �d S )Nzps -o lstart -p �STARTED� z%a %b %e %H:%M:%S %Y) r r' �replace�strip�psutil�Process� create_time�time�strftime� localtime)�self�output�start_ps� start_psutilr r r �test_process_create_timeW s �z$BSDTestCase.test_process_create_timec C s� dd� }t jdd�D ]I}t �|j�}||j�\}}}}|j|ks"J �|j|ks)J �t|j| �dkr>| �d|j� d|� ���t|j | �dkrS| �d|j � d|� ���q d S )Nc S s� t d| � d���� }|�d�}|�d� |�d�}|�� d d� \}}}}|dkr+d}t|�d }t|�d }t|�d }||||fS ) Nzdf -k "�"r r � �noner0 � )r r2 r �popr )�pathr �linesr �dev�total�used�freer r r �dfd s z"BSDTestCase.test_disks.<locals>.dfF)�alli � zpsutil=z, df=) r3 �disk_partitions� disk_usage� mountpoint�devicerF �absrH �failrG )r9 rI �part�usagerE rF rG rH r r r � test_disksa s ��zBSDTestCase.test_disksr zsysctl cmd not availablec C s t d�}tjdd�|ksJ �d S )Nzhw.ncpuT)�logical)r r3 � cpu_count�r9 �systr r r �test_cpu_count_logical| s z"BSDTestCase.test_cpu_count_logicalzskipped on NETBSDc C s t d�}|t�� jks J �d S )Nz hw.physmem)r r3 �virtual_memoryrF �r9 �numr r r �test_virtual_memory_total� s z%BSDTestCase.test_virtual_memory_total�ifconfigzifconfig cmd not availablec C sx t �� �� D ]3\}}z td|� ��}W n ty Y qw |jd|v ks&J �d|v r9|jtt� d|�d �ks9J �qd S )Nz ifconfig �RUNNING�mtuz mtu (\d+)r ) r3 � net_if_stats�itemsr � RuntimeError�isupr_ r �re�findall)r9 �name�statsr r r r �test_net_if_stats� s � ��z BSDTestCase.test_net_if_statsN)�__name__� __module__� __qualname__�__doc__� classmethodr* r. r �mark�skipifr r= rS �shutil�whichrX r\ rh r r r r r# K s0 � �� �r# z FREEBSD onlyc @ sf e Zd Zedd� �Zedd� �Ze� dd� �Zdd� Zd d � Z d d � Z e� d d� �Z e� dd� �Z dS )�FreeBSDPsutilTestCasec C r$ r% r& r( r r r r* � r+ z FreeBSDPsutilTestCase.setUpClassc C r, r% r- r( r r r r. � r+ z#FreeBSDPsutilTestCase.tearDownClassc C s� t d| j� ��}t�| j�jdd�}|�d�dd � }|r\|�� }|�� }|d d� \}}}} } |�� } |� d|� �| jks@J �t| �| j ksIJ �| j � d�sX|d | j ksXJ �|s d S d S ) Nz procstat -v F)�groupedr r � �-�[� ) r r' r3 r4 � memory_mapsr rB �addrr �rssrC r ) r9 r �mapsrD r �fields�_�start�stop�_perms�res�mapr r r �test_memory_maps� s �z&FreeBSDPsutilTestCase.test_memory_mapsc C s> t d| j� ��}t�| j��� |�d�d �� d ks J �d S )Nz procstat -b r r ���)r r' r3 r4 �exer �r9 r r r r �test_exe� s .z FreeBSDPsutilTestCase.test_exec C sN t d| j� ��}d�t�| j��� �d�|�d�d �� dd � �ks%J �d S )Nz procstat -c � r r r )r r' �joinr3 r4 r r r� r r r � test_cmdline� s �z"FreeBSDPsutilTestCase.test_cmdlinec C s� t d| j� ��}|�d�d �� dd� \}}}}}}t�| j�}|�� } |�� } | jt|�ks2J �| j t|�ks;J �| j t|�ksDJ �| jt|�ksMJ �| j t|�ksVJ �| j t|�ks_J �d S )Nz procstat -s r r r � ) r r' r r3 r4 �uids�gids�realr � effective�saved) r9 r �euid�ruid�suid�egid�rgid�sgid�pr� r� r r r �test_uids_gids� s & z$FreeBSDPsutilTestCase.test_uids_gidsc C s� g }t d| j� ��}t�| j�}|�d�D ]A}|�� �� }d|v r:t|�� d �}|�� j }||ks4J �|� d � qd|v rVt|�� d �}|�� j }||ksQJ �|� d � qt |�dkrat d��d S )N� procstat -r r z voluntary contextr� z involuntary contextr �)couldn't find lines match in procstat out)r r' r3 r4 r �lowerr2 r �num_ctx_switches� voluntary�append� involuntary�lenrb �r9 �testedr r� r � pstat_value� psutil_valuer r r �test_ctx_switches� s& � �z'FreeBSDPsutilTestCase.test_ctx_switchesc C s� g }t d| j� ��}t�| j�}|�d�D ]O}|�� �� }d|v rAtd|�� d �d�d �}|�� j }||ks;J �|� d � qd|v rdtd|�� d �d�d �}|�� j }||ks_J �|� d � qt |�dkrot d ��d S ) Nr� r z user timez0.r� �.z system timer r� )r r' r3 r4 r r� r2 �float� cpu_times�userr� �systemr� rb r� r r r �test_cpu_times� s& � �z$FreeBSDPsutilTestCase.test_cpu_timesN) ri rj rk rm r* r. r r� r� r� r� r� r� r r r r rr � s rr c @ s� e Zd Zedd� �Zdd� Ze� dd� �Ze� dd� �Ze� d d � �Z e� d d � �Z e� d d� �Z e� dd� �Z e jje dd�dd� �Ze jje dd�e� dd� ��Ze jje dd�e� dd� ��Ze jje dd�e� dd� ��Ze jje dd�e� dd � ��Ze jje dd�e� d d � ��Ze jje dd�e� d d � ��Zd!d"� Zd#d$� Zd%d&� Ze� d'd(� �Zd)d*� Z d+d,� Z d-d.� Z d/d0� Z e jje d1d�d2d3� �Z!e jje d1d�d4d5� �Z"e jje d6d�d7d8� �Z#d9d:� Z$d;S )<�FreeBSDSystemTestCasec C sT t d��� d } t�d| �}|std| � ���dd� |dd� D �\}}}|||fS ) Nz swapinfo -kr� z\s+zCan't parse swapinfo: c s s � | ] }t |�d V qdS )rA N)r )�.0r� r r r � <genexpr> s � z7FreeBSDSystemTestCase.parse_swapinfo.<locals>.<genexpr>r r? )r � splitlinesrd r r )r: �partsrF rG rH r r r �parse_swapinfo� s z$FreeBSDSystemTestCase.parse_swapinfoc C s� d}zt t|��}W n ty t�d��w t�� j|ks J �d}t|�}t |�� d �d�d �}t |�� d �d�d �}t�� j |ksIJ �t�� j |ksRJ �d S )Nzdev.cpu.0.freqz#frequencies not supported by kernelzdev.cpu.0.freq_levelsr �/r� ) r r rb r �skipr3 �cpu_freq�currentr �max�min)r9 �sensor� sysctl_result�max_freq�min_freqr r r �!test_cpu_frequency_against_sysctl s �z7FreeBSDSystemTestCase.test_cpu_frequency_against_sysctlc C �* t d�t }tt�� j| �tk sJ �d S )Nzvm.stats.vm.v_active_count)r �PAGESIZErO r3 rY �activer rV r r r �test_vmem_active � z&FreeBSDSystemTestCase.test_vmem_activec C r� )Nz vm.stats.vm.v_inactive_count)r r� rO r3 rY �inactiver rV r r r �test_vmem_inactive$ r� z(FreeBSDSystemTestCase.test_vmem_inactivec C r� )Nzvm.stats.vm.v_wire_count)r r� rO r3 rY �wiredr rV r r r �test_vmem_wired) r� z%FreeBSDSystemTestCase.test_vmem_wiredc C r� )Nzvm.stats.vm.v_cache_count)r r� rO r3 rY �cachedr rV r r r �test_vmem_cached. r� z&FreeBSDSystemTestCase.test_vmem_cachedc C r� )Nzvm.stats.vm.v_free_count)r r� rO r3 rY rH r rV r r r �test_vmem_free3 r� z$FreeBSDSystemTestCase.test_vmem_freec C �& t d�}tt�� j| �tk sJ �d S )Nz vfs.bufspace)r rO r3 rY �buffersr rV r r r �test_vmem_buffers8 s z'FreeBSDSystemTestCase.test_vmem_bufferszmuse not installedr! c C s t d�}t�� j|ks J �d S )N�Total)r r3 rY rF rZ r r r �test_muse_vmem_total? s z*FreeBSDSystemTestCase.test_muse_vmem_totalc C r� )N�Active)r rO r3 rY r� r rZ r r r �test_muse_vmem_activeD � z+FreeBSDSystemTestCase.test_muse_vmem_activec C r� )N�Inactive)r rO r3 rY r� r rZ r r r �test_muse_vmem_inactiveJ r� z-FreeBSDSystemTestCase.test_muse_vmem_inactivec C r� )N�Wired)r rO r3 rY r� r rZ r r r �test_muse_vmem_wiredP r� z*FreeBSDSystemTestCase.test_muse_vmem_wiredc C r� )N�Cache)r rO r3 rY r� r rZ r r r �test_muse_vmem_cachedV r� z+FreeBSDSystemTestCase.test_muse_vmem_cachedc C r� )N�Free)r rO r3 rY rH r rZ r r r �test_muse_vmem_free\ r� z)FreeBSDSystemTestCase.test_muse_vmem_freec C r� )N�Buffer)r rO r3 rY r� r rZ r r r �test_muse_vmem_buffersb r� z,FreeBSDSystemTestCase.test_muse_vmem_buffersc C �" t t�� jtd� �dk sJ �d S )Nzvm.stats.sys.v_swtch�� )rO r3 � cpu_stats� ctx_switchesr �r9 r r r �test_cpu_stats_ctx_switchesh � �� �z1FreeBSDSystemTestCase.test_cpu_stats_ctx_switchesc C r� )Nzvm.stats.sys.v_intrr� )rO r3 r� � interruptsr r� r r r �test_cpu_stats_interruptsq s �z/FreeBSDSystemTestCase.test_cpu_stats_interruptsc C r� )Nzvm.stats.sys.v_softr� )rO r3 r� �soft_interruptsr r� r r r � test_cpu_stats_soft_interruptsw r� z4FreeBSDSystemTestCase.test_cpu_stats_soft_interruptsc C r� )Nzvm.stats.sys.v_syscalli@ )rO r3 r� �syscallsr r� r r r �test_cpu_stats_syscalls� s �z-FreeBSDSystemTestCase.test_cpu_stats_syscallsc C s, | � � \}}}tt�� j| �tk sJ �d S r% )r� rO r3 � swap_memoryrH r )r9 �_total�_usedrH r r r �test_swapmem_free� � z'FreeBSDSystemTestCase.test_swapmem_freec C s, | � � \}}}tt�� j| �tk sJ �d S r% )r� rO r3 r� rG r )r9 r� rG �_freer r r �test_swapmem_used� r� z'FreeBSDSystemTestCase.test_swapmem_usedc C s, | � � \}}}tt�� j| �tk sJ �d S r% )r� rO r3 r� rF r )r9 rF r� r� r r r �test_swapmem_total� r� z(FreeBSDSystemTestCase.test_swapmem_totalc C sL t d�}||�d�d d � }|d |�d�� }t|�}|t�� ks$J �d S )Nzsysctl kern.boottimez sec = � �,)r r r r3 � boot_time)r9 �s�btimer r r �test_boot_time� s z$FreeBSDSystemTestCase.test_boot_timez no batteryc C s� dd� }t d�}dd� |�d�D �}t�� }t|d �dd ��}|d }|j|ks+J �|d kr9|jtjks7J �d S ||j�|ksBJ �d S ) Nc S s4 t | d�\}}t |d�\}}t|�� dt|�d��S )N�< �:�02)�divmodr )�secs�m�_s�hr r r � secs2hours� s z>FreeBSDSystemTestCase.test_sensors_battery.<locals>.secs2hoursz acpiconf -i 0c S s& i | ]}|� d �d |� d �d �qS )� r r� )r )r� �xr r r � <dictcomp>� s & z>FreeBSDSystemTestCase.test_sensors_battery.<locals>.<dictcomp>r zRemaining capacity:�%r0 zRemaining time:�unknown) r r r3 �sensors_batteryr r1 �percent�secsleft�POWER_TIME_UNLIMITED)r9 r� r r| �metricsr �remaining_timer r r �test_sensors_battery� s z*FreeBSDSystemTestCase.test_sensors_batteryc C sn t �� jtd�ks J �t �� jtd�dkksJ �t �� j}|dk r+td�dks)J �d S |td�d ks5J �d S )N�hw.acpi.battery.life�hw.acpi.acliner r �hw.acpi.battery.timer� r� )r3 r r r � power_pluggedr )r9 r r r r �#test_sensors_battery_against_sysctl� s � � z9FreeBSDSystemTestCase.test_sensors_battery_against_sysctlz has batteryc C sV t �t�� td� td� td� W d � n1 s w Y t�� d u s)J �d S )Nr r r )r �raisesrb r r3 r r� r r r � test_sensors_battery_no_battery� s �z5FreeBSDSystemTestCase.test_sensors_battery_no_batteryc C s� t �d�}t|�D ]S}d|� d�}zttt|�d d� ��}W n ty+ t�d��w t t � � d | j | �dk s=J �d|� d�}ttt|�d d� ��}t � � d | j |ks\J �q d S ) NTzdev.cpu.z .temperaturer� z$temperatures not supported by kernel�coretemprw z.coretemp.tjmax) r3 rU �ranger r� r rb r r� rO �sensors_temperaturesr� �high)r9 �num_cpus�cpur� r� r r r �(test_sensors_temperatures_against_sysctl� s* ���� ��z>FreeBSDSystemTestCase.test_sensors_temperatures_against_sysctlN)%ri rj rk � staticmethodr� r� r r� r� r� r� r� r� r rn ro �MUSE_AVAILABLEr� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r r r r r r r� � sh r� z OPENBSD onlyc @ s e Zd Zdd� ZdS )�OpenBSDTestCasec C s6 t d�}tj�|d�}tj�t�� �}||ksJ �d S )Nz kern.boottimez%a %b %d %H:%M:%S %Y)r �datetime�strptime� fromtimestampr3 r� )r9 r� �sys_bt� psutil_btr r r r� � s z OpenBSDTestCase.test_boot_timeN)ri rj rk r� r r r r r � s r z NETBSD onlyc @ sh e Zd Zedd� �Zdd� Zdd� Zdd� Zd d � Zd d � Z d d� Z dd� Z dd� Z dd� Z dd� ZdS )�NetBSDTestCasec C sp t d��%}|D ]}|�| �r!t|�� d �d W d � S qW d � n1 s,w Y td| � ���)Nz /proc/meminfor rA z can't find )�openr r r r )�look_for�fr r r r � parse_meminfo s �� �z NetBSDTestCase.parse_meminfoc C s t �� j| �d�ks J �d S )Nz MemTotal:)r3 rY rF r% r� r r r �test_vmem_total s z NetBSDTestCase.test_vmem_totalc C �$ t t�� j| �d� �tk sJ �d S )NzMemFree:)rO r3 rY rH r% r r� r r r r� � �z NetBSDTestCase.test_vmem_freec C r' )NzBuffers:)rO r3 rY r� r% r r� r r r r� � �� �z NetBSDTestCase.test_vmem_buffersc C r' )Nz MemShared:)rO r3 rY �sharedr% r r� r r r �test_vmem_shared r) z NetBSDTestCase.test_vmem_sharedc C r' )NzCached:)rO r3 rY r� r% r r� r r r r� % r( z NetBSDTestCase.test_vmem_cachedc C r' )Nz SwapTotal:)rO r3 r� rF r% r r� r r r r� - r( z!NetBSDTestCase.test_swapmem_totalc C r' )Nz SwapFree:)rO r3 r� rH r% r r� r r r r� 3 r( z NetBSDTestCase.test_swapmem_freec C s" t �� }|j|j|j ksJ �d S r% )r3 r� rG rF rH )r9 �smemr r r r� 9 s z NetBSDTestCase.test_swapmem_usedc C �x t dd�� }|D ]}|�d�rt|�� d �} nqtd��W d � n1 s(w Y tt�� j| �dk s:J �d S )N� /proc/stat�rbs intrr �couldn't find liner� ) r" r r r r rO r3 r� r� )r9 r$ r r� r r r r� ? � �� � z(NetBSDTestCase.test_cpu_stats_interruptsc C r- )Nr. r/ s ctxtr r0 r� ) r" r r r r rO r3 r� r� )r9 r$ r r� r r r r� I r1 z*NetBSDTestCase.test_cpu_stats_ctx_switchesN)ri rj rk r r% r& r� r� r+ r� r� r� r� r� r� r r r r r! � s r! )#rl r �osrd rp r6 r3 r r r r � psutil.testsr r r r r r r r �psutil._psutil_posixr r� �getuidrq r r r rn ro r# rr r� r r! r r r r �<module> sL UX v tests/__pycache__/test_contracts.cpython-310.pyc 0000644 00000031246 15030211765 0015664 0 ustar 00 o �h�. � @ sl d Z ddlZddlZddlZddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlm Z dd lm Z dd lm Z dd lm Z dd l mZ dd l mZ ddl mZ ddl mZ ddl mZ ddl mZ ddl mZ ddl mZ ddl mZ ddl mZ ddl mZ ddl mZ G dd� de�ZG dd� de�ZG d d � d e�Z G d d � d e�Z G d d!� d!e�Z dS )"z�Contracts tests. These tests mainly check API sanity in terms of returned types and APIs availability. Some of these are duplicates of tests test_system.py and test_process.py. � N)�AIX)�FREEBSD)�LINUX)�MACOS)�NETBSD)�OPENBSD)�POSIX)�SUNOS)�WINDOWS)�GITHUB_ACTIONS)� HAS_CPU_FREQ)�HAS_NET_IO_COUNTERS)�HAS_SENSORS_FANS)�HAS_SENSORS_TEMPERATURES)� SKIP_SYSCONS)�PsutilTestCase)�create_sockets)�enum)� is_namedtuple)�kernel_version)�pytestc @ sH e Zd Zdd� Zdd� Zdd� Zdd� Zejj e oe d d �d d � �Z d S )�TestAvailConstantsAPIsc C s t td�tp tp tks J �d S )N� PROCFS_PATH)�hasattr�psutilr r r ��self� r �N/usr/local/CyberCP/lib/python3.10/site-packages/psutil/tests/test_contracts.py�test_PROCFS_PATH0 s z'TestAvailConstantsAPIs.test_PROCFS_PATHc C sj | j }|ttd�t� |ttd�t� |ttd�t� |ttd�t� |ttd�t� |ttd�t� d S )N�ABOVE_NORMAL_PRIORITY_CLASS�BELOW_NORMAL_PRIORITY_CLASS�HIGH_PRIORITY_CLASS�IDLE_PRIORITY_CLASS�NORMAL_PRIORITY_CLASS�REALTIME_PRIORITY_CLASS�� assertEqualr r r �r �aer r r �test_win_priority3 s z(TestAvailConstantsAPIs.test_win_priorityc C �J | j }|ttd�t� |ttd�t� |ttd�t� |ttd�t� d S )N�IOPRIO_CLASS_NONE�IOPRIO_CLASS_RT�IOPRIO_CLASS_BE�IOPRIO_CLASS_IDLE)r' r r r r( r r r �test_linux_ioprio_linux< � z.TestAvailConstantsAPIs.test_linux_ioprio_linuxc C r+ )N� IOPRIO_HIGH� IOPRIO_NORMAL� IOPRIO_LOW�IOPRIO_VERYLOWr& r( r r r �test_linux_ioprio_windowsC r1 z0TestAvailConstantsAPIs.test_linux_ioprio_windows�%unsupported on GITHUB_ACTIONS + LINUX��reasonc C s� | j }|ttd�tp t� |ttd�tpt� |ttd�tp t� |ttd�tp)t� |ttd�tp3t� |ttd�tp=t� |ttd�tpGt� |ttd�tpQt� |ttd �tp[t� |ttd �tpet� |ttd �tpot� |ttd �t� tr�t� d kr�|ttd�t� t� dkr�|ttd�t� t� dkr�|ttd�t� t� dkr�|ttd�t� t� d kr�|ttd�t� |ttd�t� |ttd�t� |ttd�t� d S )N� RLIM_INFINITY� RLIMIT_AS� RLIMIT_CORE� RLIMIT_CPU� RLIMIT_DATA� RLIMIT_FSIZE�RLIMIT_MEMLOCK� RLIMIT_NOFILE� RLIMIT_NPROC� RLIMIT_RSS� RLIMIT_STACK� RLIMIT_LOCKS)� � � �RLIMIT_MSGQUEUE)rF rG � � RLIMIT_NICE� RLIMIT_RTPRIO)rF rG � � RLIMIT_RTTIME�RLIMIT_SIGPENDING� RLIMIT_SWAP� RLIMIT_SBSIZE� RLIMIT_NPTS)r' r r r r r r r( r r r � test_rlimitJ s6 z"TestAvailConstantsAPIs.test_rlimitN) �__name__� __module__� __qualname__r r* r0 r6 r �mark�skipifr r rS r r r r r / s �r c @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd S )�TestAvailSystemAPIsc C � t td�tks J �d S )N�win_service_iter�r r r r r r r �test_win_service_itero � z)TestAvailSystemAPIs.test_win_service_iterc C rZ )N�win_service_getr\ r r r r �test_win_service_getr r^ z(TestAvailSystemAPIs.test_win_service_getc C s& t td�tp tp tp tp tksJ �d S )N�cpu_freq)r r r r r r r r r r r � test_cpu_frequ s �z!TestAvailSystemAPIs.test_cpu_freqc C s t td�tptks J �d S )N�sensors_temperatures)r r r r r r r r �test_sensors_temperaturesz s z-TestAvailSystemAPIs.test_sensors_temperaturesc C rZ )N� sensors_fans)r r r r r r r �test_sensors_fans} r^ z%TestAvailSystemAPIs.test_sensors_fansc C s" t td�tp tp tp tksJ �d S )N�sensors_battery)r r r r r r r r r r � test_battery� s �z TestAvailSystemAPIs.test_batteryN) rT rU rV r] r` rb rd rf rh r r r r rY n s rY c @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zej j e o e d d �d d� �Z dd� Zdd� Zdd� Zdd� Zdd� Zdd� ZdS ) �TestAvailProcessAPIsc C s4 t tjd�tptptptptptpt pt ksJ �d S )N�environ) r r �Processr r r r r r r r r r r r � test_environ� s ������ �z!TestAvailProcessAPIs.test_environc C � t tjd�tks J �d S �N�uids�r r rk r r r r r � test_uids� � z TestAvailProcessAPIs.test_uidsc C rm rn rp r r r r � test_gids� rr z TestAvailProcessAPIs.test_gidsc C rm )N�terminalrp r r r r � test_terminal� rr z"TestAvailProcessAPIs.test_terminalc C � t tjd�tptks J �d S )N�ionice)r r rk r r r r r r � test_ionice� s z TestAvailProcessAPIs.test_ionicer7 r8 c C rv )N�rlimit)r r rk r r r r r r rS � s z TestAvailProcessAPIs.test_rlimitc C s" t tjd�}|tp t ksJ �d S )N� io_counters)r r rk r r �r �hasitr r r �test_io_counters� s z%TestAvailProcessAPIs.test_io_countersc C rm )N�num_fdsrp r r r r � test_num_fds� rr z!TestAvailProcessAPIs.test_num_fdsc C rm )N� num_handles)r r rk r r r r r �test_num_handles� rr z%TestAvailProcessAPIs.test_num_handlesc C � t tjd�tp tp tksJ �d S )N� cpu_affinity)r r rk r r r r r r r �test_cpu_affinity� � �z&TestAvailProcessAPIs.test_cpu_affinityc C r� )N�cpu_num)r r rk r r r r r r r � test_cpu_num� r� z!TestAvailProcessAPIs.test_cpu_numc C s* t tjd�}|tptptpt ksJ �d S )N� memory_maps)r r rk r r r r r{ r r r �test_memory_maps� s z%TestAvailProcessAPIs.test_memory_mapsN)rT rU rV rl rq rs ru rx r rW rX r r rS r} r r� r� r� r� r r r r ri � s" � ri c @ s e Zd ZdZedd� �Zedfdd�Zdd� Zd d � Z d d � Z d d� Z e j jeo-e�� dkdd�e j je dd�dd� ��Zdd� Zdd� Ze j jedd�dd� �Zd d � Zd d � Ze j je dd�d d!� �Ze j je dd�d"d#� �Z e j je dd�d$d%� �Z d&d'� Z d(d)� Z!d*S )+�TestSystemAPITypeszhCheck the return types of system related APIs. https://github.com/giampaolo/psutil/issues/1039. c C s t �� | _d S �N)r rk �proc)�clsr r r � setUpClass� s z TestSystemAPITypes.setUpClassTc C s8 t |�sJ �|D ]}t||�sJ �|r|dksJ �qd S )Nr )r � isinstance)r �nt�type_�gezero�nr r r �assert_ntuple_of_nums� s ��z(TestSystemAPITypes.assert_ntuple_of_numsc C s. | � t�� � tjdd�D ]}| � |� q d S )NT)�percpu)r� r � cpu_times)r r� r r r �test_cpu_times� s �z!TestSystemAPITypes.test_cpu_timesc C s0 t tjd d�t�s J �t tjdd�t�sJ �d S )N��intervalg�h㈵��>)r� r � cpu_percent�floatr r r r �test_cpu_percent� s z#TestSystemAPITypes.test_cpu_percentc C s( | � tjd d�� | � tjdd�� d S )Nr� g-C ��6?)r� r �cpu_times_percentr r r r �test_cpu_times_percent� s z)TestSystemAPITypes.test_cpu_times_percentc C � t t�� t�s J �d S r� )r� r � cpu_count�intr r r r �test_cpu_count� r^ z!TestSystemAPITypes.test_cpu_count�arm64zskipped due to #1892r8 z not supportedc C s0 t �� d u r t�d��| jt �� ttfd� d S )Nzcpu_freq() returns None�r� )r ra r �skipr� r� r� r r r r rb � s z TestSystemAPITypes.test_cpu_freqc C s: t jdd��� D ]\}}t|t�sJ �| j|td� qd S )NT)�perdiskr� )r �disk_io_counters�itemsr� �strr� r� )r �k�vr r r �test_disk_io_counters� s �z(TestSystemAPITypes.test_disk_io_countersc C sR t �� D ]"}t|jt�sJ �t|jt�sJ �t|jt�s J �t|jt�s&J �qd S r� )r �disk_partitionsr� �devicer� � mountpoint�fstype�opts)r �diskr r r �test_disk_partitions� s �z'TestSystemAPITypes.test_disk_partitionsz requires rootc C sd t � �% t�d�}t|�tt|��ksJ �|D ]}t|�s J �qW d � d S 1 s+w Y d S )N�all)r r �net_connections�len�setr )r �ret�connr r r �test_net_connections� s �"�z'TestSystemAPITypes.test_net_connectionsc C s� t �� �� D ]9\}}t|t�sJ �|D ]+}t|jtj�s J �t|jt�s&J �t|j tt d �f�s2J �t|j tt d �f�s>J �qqd S r� ) r � net_if_addrsr� r� r� �familyr �IntEnum�address�netmask�type� broadcast)r �ifname�addrs�addrr r r �test_net_if_addrs s ��z$TestSystemAPITypes.test_net_if_addrsc C sj t �� �� D ],\}}t|t�sJ �t|jt�sJ �t|jtj �s"J �t|j t �s*J �t|j t �s2J �qd S r� ) r � net_if_statsr� r� r� �isup�bool�duplexr r� �speedr� �mtu)r r� �infor r r �test_net_if_stats s �z$TestSystemAPITypes.test_net_if_statsc C s$ t jdd�D ] }t|t�sJ �qd S )NT)�pernic)r �net_io_countersr� r� )r r� r r r �test_net_io_counters s �z'TestSystemAPITypes.test_net_io_countersc C s\ t �� �� D ]%\}}t|t�sJ �|D ]}t|jt�s J �t|jttt d �f�s*J �qqd S r� ) r re r� r� r� �label�currentr� r� r� �r �name�units�unitr r r rf s ��z$TestSystemAPITypes.test_sensors_fansc C s� t �� �� D ]?\}}t|t�sJ �|D ]1}t|jt�s J �t|jttt d �f�s*J �t|j ttt d �f�s7J �t|j ttt d �f�sDJ �qqd S r� ) r rc r� r� r� r� r� r� r� r� �high�criticalr� r r r rd & s ��z,TestSystemAPITypes.test_sensors_temperaturesc C r� r� )r� r � boot_timer� r r r r �test_boot_time1 s z!TestSystemAPITypes.test_boot_timec C sj t �� D ].}t|jt�sJ �t|jttd �f�sJ �t|jttd �f�s&J �t|jt td �f�s2J �qd S r� ) r �usersr� r� r� rt r� �host�pidr� )r �userr r r � test_users5 s �z TestSystemAPITypes.test_usersN)"rT rU rV �__doc__� classmethodr� r� r� r� r� r� r� r rW rX r �platform�machiner rb r� r� r r� r� r� r r� r rf r rd r� r� r r r r r� � s8 � r� c @ s& e Zd Zejje dd�dd� �ZdS )�TestProcessWaitTypez not POSIXr8 c C sD t �| �� j�}|�� |�� }|tj ksJ �t|t j �s J �d S r� ) r rk �spawn_testprocr� � terminate�wait�signal�SIGTERMr� r r� )r �p�coder r r �test_negative_signal? s z(TestProcessWaitType.test_negative_signalN)rT rU rV r rW rX r r� r r r r r� > s r� ) r� r� r� r r r r r r r r r r � psutil.testsr r r r r r r r r r r r r rY ri r� r� r r r r �<module> s< ?>z tests/__pycache__/test_system.cpython-310.pyc 0000644 00000070274 15030211765 0015214 0 ustar 00 o �h � � @ sh d Z ddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl Z ddl Z ddl m Z ddlZddlmZ ddlmZ ddlmZ ddlmZ ddlmZ dd lmZ dd lmZ dd lmZ dd lmZ dd lmZ ddlmZ ddlm Z ddlm Z ddlm Z ddlm Z ddlm Z ddlm!Z! ddlm"Z" ddlm#Z# ddlm$Z$ ddlm%Z% ddlm&Z& ddlm'Z' ddlm(Z( dd lm)Z) dd lm*Z* dd lm+Z+ dd lm,Z, dd lm-Z- dd!lm.Z. dd"lm/Z/ G d#d$� d$e,�Z0G d%d&� d&e,�Z1G d'd(� d(e,�Z2G d)d*� d*e,�Z3G d+d,� d,e,�Z4G d-d.� d.e,�Z5G d/d0� d0e,�Z6G d1d2� d2e,�Z7dS )3zTests for system APIS.� N)�mock)�AIX)�BSD)�FREEBSD)�LINUX)�MACOS)�NETBSD)�OPENBSD)�POSIX)�SUNOS)�WINDOWS)�broadcast_addr)�AARCH64)�ASCII_FS)� CI_TESTING)�GITHUB_ACTIONS)�GLOBAL_TIMEOUT)� HAS_BATTERY)� HAS_CPU_FREQ)�HAS_GETLOADAVG)�HAS_NET_IO_COUNTERS)�HAS_SENSORS_BATTERY)�HAS_SENSORS_FANS)�HAS_SENSORS_TEMPERATURES)�IS_64BIT)� MACOS_12PLUS)�PYPY)�UNICODE_SUFFIX)�PsutilTestCase)�check_net_address)�pytest)�retry_on_failurec @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd S )�TestProcessIterc C s~ t �� dd� t�� D �v sJ �| �� }|jdd� t�� D �v s!J �t�|j�}|�� |�� |jdd� t�� D �vs=J �d S )Nc S � g | ]}|j �qS � ��pid��.0�xr$ r$ �K/usr/local/CyberCP/lib/python3.10/site-packages/psutil/tests/test_system.py� <listcomp>? � z5TestProcessIter.test_pid_presence.<locals>.<listcomp>c S r# r$ r% r' r$ r$ r* r+ A r, c S r# r$ r% r' r$ r$ r* r+ E r, ) �os�getpid�psutil� process_iter�spawn_testprocr& �Process�kill�wait��self�sproc�pr$ r$ r* �test_pid_presence> s z!TestProcessIter.test_pid_presencec C s8 t t�� �}t|dd� d�tt|�dd� d�ksJ �d S )Nc S � | j S �Nr% �r) r$ r$ r* �<lambda>I � z4TestProcessIter.test_no_duplicates.<locals>.<lambda>)�keyc S r: r; r% r< r$ r$ r* r= J r> )�listr/ r0 �sorted�set)r6 �lsr$ r$ r* �test_no_duplicatesG s �z"TestProcessIter.test_no_duplicatesc C sv t t�� � td�D ].}tjdt�t�� �d�� t tjdgd��r$J �W d � n1 s.w Y tj� � q d S �N� zpsutil.Process.as_dict�� side_effect� cpu_times��attrs) r@ r/ r0 �ranger �patch� NoSuchProcessr- r. � cache_clear�r6 r) r$ r$ r* �test_emulate_nspM s � � �z TestProcessIter.test_emulate_nspc C s� t t�� � td�D ]B}tjdt�t�� �d��' t � tj�� t tjdgd�� W d � n1 s3w Y W d � n1 sBw Y tj� � q d S rE ) r@ r/ r0 rL r rM � AccessDeniedr- r. r �raisesrO rP r$ r$ r* �test_emulate_access_deniedW s � �� � �z*TestProcessIter.test_emulate_access_deniedc C s� t jdgd�D ]}t|j�� �dgksJ �qt jdgd�D ]}t|j�� �dgks+J �q t�t�� tt jdgd�� W d � n1 sEw Y tj dt � dd�d��*}t jddgd�D ]}|jd d u siJ �|jd dksrJ �q^|j sxJ �W d � n1 s�w Y tj dt � dd�d��/}t � }t jddg|d �D ]}|jd |u s�J �|jd dks�J �q�|j s�J �W d � d S 1 s�w Y d S ) Nr& rJ �fooz$psutil._psplatform.Process.cpu_timesr � rG rI )rK �ad_value) r/ r0 r@ �info�keysr rS � ValueErrorr rM rR �called�object)r6 r8 �m�flagr$ r$ r* � test_attrsb s< � � � � � "�zTestProcessIter.test_attrsc C s. t t�� � tjs J �tj�� tjrJ �d S r; )r@ r/ r0 �_pmaprO �r6 r$ r$ r* �test_cache_clear~ s z TestProcessIter.test_cache_clearN) �__name__� __module__� __qualname__r9 rD rQ rT r_ rb r$ r$ r$ r* r" = s r" c @ sT e Zd Zejjeo edd�dd� �Zejjeoedd�dd� �Z dd� Z d d � Z d S ) �TestProcessAPIsz-spawn_testproc() unreliable on PYPY + WINDOWS��reasonc s � fdd�}g � | � � }| � � }| � � }dd� |||fD �}t�t�� tj|dd� W d � n1 s5w Y t�t�� tj|dd� W d � n1 sQw Y t�� }tj|d |d �\}}t�� | d k snJ �|rrJ �t|�d kszJ �� r~J �|D ] } t | d �r�J �q�t d�dd� �} |� � | ||�\}}|j dd� |D �v s�J �t r�|�� jtj ks�J �n |�� jdks�J �� |j gks�J �|D ] } t | d �r�J �q�t d�dd� �} |� � |� � | ||�\}}t� �|j |j |j hks�J �|D ] } t | d ��sJ �q�d S )Nc s � � | j� d S r; )�appendr& )r8 ��pidsr$ r* �callback� s z1TestProcessAPIs.test_wait_procs.<locals>.callbackc S � g | ]}t �|j��qS r$ �r/ r2 r& r' r$ r$ r* r+ � � z3TestProcessAPIs.test_wait_procs.<locals>.<listcomp>���)�timeout� )rl g{�G�z�?�rq rl g �?� � returncode� c S �< t j| d|d�\}}t|�dksJ �t|�dksJ �||fS )N� ��Q��?rs rr rF �r/ � wait_procs�len��procsrl �gone�aliver$ r$ r* �test_1� � �z/TestProcessAPIs.test_wait_procs.<locals>.test_1c S r# r$ r% r' r$ r$ r* r+ � r, c S rw )Nrx rs rt r ry r| r$ r$ r* �test_2� r� z/TestProcessAPIs.test_wait_procs.<locals>.test_2)r1 r rS rZ r/ rz � TypeError�timer{ �hasattrr! � terminater& r �popru �signal�SIGTERMrB ) r6 rl �sproc1�sproc2�sproc3r} �tr~ r r8 r� r� r$ rj r* �test_wait_procs� sP � � �z TestProcessAPIs.test_wait_procsc C sL | � � }| � � }| � � }dd� |||fD �}|D ]}|�� qt�|� d S )Nc S rm r$ rn r' r$ r$ r* r+ � ro z>TestProcessAPIs.test_wait_procs_no_timeout.<locals>.<listcomp>)r1 r� r/ rz )r6 r� r� r� r} r8 r$ r$ r* �test_wait_procs_no_timeout� s z*TestProcessAPIs.test_wait_procs_no_timeoutc C sp | � � }t�|j�s J �t�|j�}|�� |�� t�|j�r"J �t�d�r)J �t�d�dt�� v ks6J �d S )Nrp r )r1 r/ � pid_existsr& r2 r3 r4 rk r5 r$ r$ r* �test_pid_exists� s z TestProcessAPIs.test_pid_existsc C s� t �� }|D ]!}z t �|�sJ �W q ty' t�d� |t �� vs%J �Y qw tt|�d t|�d �}|D ] }t �|�r@J �q7d S )Ng�������?i�: i�> )r/ rk r� �AssertionErrorr� �sleeprL �max)r6 rk r& r$ r$ r* �test_pid_exists_2� s ��z!TestProcessAPIs.test_pid_exists_2N) rc rd re r �mark�skipifr r r� r� r� r� r$ r$ r$ r* rf � s � ;� rf c @ s> e Zd Zdd� Zejjeoe� � dd�dd� �Z dd� Z d S ) � TestMiscAPIsc C s6 t �� }t|t�s J �|dksJ �|t�� k sJ �d S �Nr )r/ � boot_time� isinstance�floatr� )r6 �btr$ r$ r* �test_boot_time� s zTestMiscAPIs.test_boot_time�unreliable on CIrg c C s� t �� }|sJ �|D ]h}| j|d��X |jsJ �t|jt�s J �t|jttd �f�s,J �|jd ur=t|jttd �f�s=J �|j |j |j dksJJ �t j � |j � t sUt r]|jd u s\J �nt �|j� W d � n1 smw Y q d S )N)�user� )r/ �users�subTest�namer� �str�terminal�type�host�started�datetime� fromtimestampr r r& r2 )r6 r� r� r$ r$ r* � test_users� s( � ���zTestMiscAPIs.test_usersc C sj g d�}|D ]}t tt|�t�sJ |��qtjdkr�tjs J �tjr$J �|�d� dt j � � v r;tj s5J �|�d� njdt j � � v rktj sGJ �tjtjtjg�d�dksVJ �|�d � |�d � |�d � |�d � n:d t j � � v sydt j � � v r�tjs~J �|�d� n!dt j � � v r�tjs�J �|�d� ntjs�J �tjr�J �|�d� |D ] }tt|�r�J |��q�d S )N) r r r r r r r r r �posixr �linuxr �bsdTrr r r r r �sunos�solarisr �darwinr r )r� �getattrr/ �boolr- r� r r �remove�sys�platform�lowerr r r r r �countr r )r6 �namesr� r$ r$ r* �test_os_constants sF �� � �z TestMiscAPIs.test_os_constantsN) rc rd re r� r r� r� r r/ r� r� r� r$ r$ r$ r* r� � s � r� c @ s e Zd Zdd� Zdd� ZdS )�TestMemoryAPIsc C s� t �� }|jdks J |��|jdksJ |��d|j kr$dks)J |�� J |��|jdks2J |��|jdks;J |��|jD ]<}t||�}|dkrPt |t �sPJ �|dkrz|dksc| � |�d|� d���||jkrz| � |�d|j� d|� d |� d���q>d S ) Nr �d �percent�totalz < 0 (�)z > total (total=z, �=) r/ �virtual_memoryr� � availabler� �used�free�_fieldsr� r� �int�fail)r6 �memr� �valuer$ r$ r* �test_virtual_memory; s& & ���z"TestMemoryAPIs.test_virtual_memoryc C s� t �� }|jdks J �|jdksJ |��|jdks J |��|jdkr,|jdks+J |��n |jdks5J |��d|j krCdksHJ |�� J |��|jdksQJ |��|jdksZJ |��d S )N)r� r� r� r� �sin�soutr r� ) r/ � swap_memoryr� r� r� r� r� r� r� )r6 r� r$ r$ r* �test_swap_memoryN s &z TestMemoryAPIs.test_swap_memoryN)rc rd re r� r� r$ r$ r$ r* r� : s r� c @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z e j j e o&edd�ed�dd� ��Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d!� Ze j j eoXe�� d"kd#d�e j j e d$d�d%d&� ��Z e j j e d$d�d'd(� �Z d)S )*� TestCpuAPIsc C s� t �� }|d us J �|tt jdd��ksJ �|dks J �tj�d�rCtd�� }|�� }W d � n1 s5w Y d|vrEt � d��d S d S )NT��percpurr z /proc/cpuinfoz physical idz#cpuinfo doesn't include physical id) r/ � cpu_countr{ rI r- �path�exists�open�readr �skip)r6 �logical�fd� cpuinfo_datar$ r$ r* �test_cpu_count_logicalf s � �z"TestCpuAPIs.test_cpu_count_logicalc C sj t �� }t jdd�}|d u rt�d��tr't�� d d� dkr'|d u s%J �d S |dks-J �||ks3J �d S )NF�r� zcpu_count_cores() is NonerF )� rr rr )r/ r� r r� r r� �getwindowsversion)r6 r� �coresr$ r$ r* �test_cpu_count_coresr s z TestCpuAPIs.test_cpu_count_coresc C s� dD ]L}t jd|d��}t�� d u sJ �|jsJ �W d � n1 s#w Y t jd|d��}tjdd�d u s:J �|js?J �W d � n1 sIw Y qd S )N)rp r Nz$psutil._psplatform.cpu_count_logical�� return_valuez"psutil._psplatform.cpu_count_coresFr� )r rM r/ r� r[ )r6 �valr] r$ r$ r* �test_cpu_count_none} s" � �� ���z TestCpuAPIs.test_cpu_count_nonec C sj d}t �� }t|� |D ]}t|t�sJ �|dksJ �||7 }q tt|t|� �d�dks/J �t|� d S )Nr r� r� )r/ rI �sumr� r� �round�absr� )r6 r� �times�cp_timer$ r$ r* �test_cpu_times� s zTestCpuAPIs.test_cpu_timesc C sR t t�� �}t�� t }t�� |k r$t t�� �}||kr d S t�� |k s| �d��)Nztime remained the same)r� r/ rI r� r r� )r6 �t1�stop_at�t2r$ r$ r* � test_cpu_times_time_increases� s � z)TestCpuAPIs.test_cpu_times_time_increasesc C s� t jdd�D ]1}d}t|� |D ]}t|t�sJ �|dks J �||7 }qtt|t|� �d�dks3J �t|� qtt jdd�d �tt jdd��ksLJ �d S )NTr� r r� r� F) r/ rI r� r� r� r� r� r� r{ )r6 r� r� r� r$ r$ r* �test_per_cpu_times� s �z TestCpuAPIs.test_per_cpu_timesc C s~ t jdd�}t�� t } t�� |kr| �d�S t jdd�}t||�D ]\}}t �|�t �|�}}|| }|dkr= d S q#q )NTr� rq g�������?)r/ rI r� r r� �zip�_cpu_busy_time)r6 �tot1� giveup_at�tot2r� r� � differencer$ r$ r* �test_per_cpu_times_2� s ��z TestCpuAPIs.test_per_cpu_times_2zunreliable on OPENBSD + CIrg rv c C s� t �� }t jdd�}|�dd� t|� D ��}|jD ]*}| j|||d�� tt||�t||� �dk s4J �W d � n1 s>w Y qd S )NTr� c S s g | ]}t |��qS r$ )r� )r( �numr$ r$ r* r+ � s z9TestCpuAPIs.test_cpu_times_comparison.<locals>.<listcomp>)�field�base�per_cpurF )r/ rI �_maker� r� r� r� r� )r6 r� r� � summed_valuesr� r$ r$ r* �test_cpu_times_comparison� s � ���z%TestCpuAPIs.test_cpu_times_comparisonc C sl zt |t�sJ �|dksJ �|dt�� ksJ �W d S ty5 } ztd�|t�|�t�|����d }~ww )Nr� g Y@z {} last={} new={})r� r� r/ r� r� �format�pprint�pformat)r6 r� �last_ret�new_ret�errr$ r$ r* �_test_cpu_percent� s ����z TestCpuAPIs._test_cpu_percentc C st t jdd�}td�D ]}t jd d�}| �|||� |}q t�t�� t jdd� W d � d S 1 s3w Y d S �N���MbP?)�intervalr� rp )r/ � cpu_percentrL r r rS rZ )r6 �last�_�newr$ r$ r* �test_cpu_percent s "�z TestCpuAPIs.test_cpu_percentc C s� t jddd�}t|�t �� ksJ �td�D ]}t jd dd�}|D ] }| �|||� q |}qt�t�� t jddd� W d � d S 1 sEw Y d S )Nr T�r r� r� rp ) r/ r r{ r� rL r r rS rZ �r6 r r r r� r$ r$ r* �test_per_cpu_percent s "�z TestCpuAPIs.test_per_cpu_percentc C s� t jdd�}td�D ] }t jd d�}|D ] }| �|||� q| �t|�||� |}q t�t�� t jdd� W d � d S 1 sAw Y d S r )r/ �cpu_times_percentrL r r� r rS rZ r r$ r$ r* �test_cpu_times_percent s "�z"TestCpuAPIs.test_cpu_times_percentc C sz t jddd�}t|�t �� ksJ �td�D ]%}t jd dd�}|D ]}|D ] }| �|||� q$| �t|�||� q |}qd S )Nr Tr r� )r/ r r{ r� rL r r� )r6 r r r �cpur� r$ r$ r* �test_per_cpu_times_percent! s �z&TestCpuAPIs.test_per_cpu_times_percentc C s� t jdd� dd� t jdd�D �}tjd|d�� t jdd�D ]}|D ] }| �|d d � q#q W d � d S 1 s9w Y d S )NTr� c S s* g | ]}|� d d� tt|j��D ���qS )c S s g | ]}d �qS )r r$ r' r$ r$ r* r+ 0 s zNTestCpuAPIs.test_per_cpu_times_percent_negative.<locals>.<listcomp>.<listcomp>)r� rL r{ r� r' r$ r$ r* r+ / s ��zCTestCpuAPIs.test_per_cpu_times_percent_negative.<locals>.<listcomp>zpsutil.cpu_timesr� )r/ r rI r rM r )r6 � zero_timesr r� r$ r$ r* �#test_per_cpu_times_percent_negative, s ���"�z/TestCpuAPIs.test_per_cpu_times_percent_negativec C sT t �� }|jdks J �|jD ]}t||�}|dksJ �ts'|dv r'|dks'J �qd S )N)� ctx_switches� interrupts�soft_interrupts�syscallsr > r r )r/ � cpu_statsr� r� r )r6 �infosr� r� r$ r$ r* �test_cpu_stats8 s ��zTestCpuAPIs.test_cpu_stats�arm64zskipped due to #1892� not supportedc C sh dd� }t jdd�}tstr|st�d��|sJ |��|t jdd�g� tr0t|�t �� ks2J �d S d S )Nc S sj | D ]0}|j dks J �|jdkr|j|jksJ �|j D ]}t||�}t|ttf�s+J �|dks1J �qqd S )N)�current�minr� r� r )r� r� r r� r� r� r� )rC �ntr� r� r$ r$ r* �check_lsN s ��z+TestCpuAPIs.test_cpu_freq.<locals>.check_lsTr� z/returns empty list on FreeBSD and Linux aarch64F) r/ �cpu_freqr r r r� r r{ r� )r6 r# rC r$ r$ r* � test_cpu_freqI s � �zTestCpuAPIs.test_cpu_freqc C s@ t �� }t|�dks J �|D ]}t|t�sJ �|dks J �qd S )Nrt r� )r/ � getloadavgr{ r� r� )r6 �loadavg�loadr$ r$ r* �test_getloadavgd s �zTestCpuAPIs.test_getloadavgN) rc rd re r� r� r� r� r� r� r� r r� r� r r r! r� r r r r r r r r r� �machiner r% r r) r$ r$ r$ r* r� e s6 "� � r� c @ s� e Zd Zejjeo e dd�dd� �Zejje dd�dd� �Z dd � Z d d � Z ejje o3ej�d � d d�ejjeo?e�� dd�dd� ��Zdd� ZdS )� TestDiskAPIsz unreliable on PYPY32 + 32BITrg c C s> t �t�� �}|jdksJ �|jdksJ |��|jdks J |��|jdks)J |��|j|jks3J |��|j|jks=J |��d|j krHdksNn J |j��t t d�r~t �t�� �}d}|j|jksdJ �t |j|j �|k spJ �t s~t |j|j �|k s~J �| � � }t�t�� t �|� W d � d S 1 s�w Y d S )N)r� r� r� r� r r� � disk_usagei P )r/ r, r- �getcwdr� r� r� r� r� r� �shutilr� r � get_testfnr rS �FileNotFoundError)r6 �usage� shutil_usage� tolerance�fnamer$ r$ r* �test_disk_usagen s& " "�z TestDiskAPIs.test_disk_usageznot an ASCII fsc C s: t �t�� t�t� W d � d S 1 sw Y d S r; )r rS �UnicodeEncodeErrorr/ r, r ra r$ r$ r* �test_disk_usage_unicode� s "�z$TestDiskAPIs.test_disk_usage_unicodec C s t �d� d S )N� .)r/ r, ra r$ r$ r* �test_disk_usage_bytes� s z"TestDiskAPIs.test_disk_usage_bytesc C sr dd� }t jdd�}|sJ �|D ]1}||� tr d|jv r qts,tj�|j�s+J |��n|j tj�|j �s:J |��|j sAJ |��qt jdd�}|sLJ �t jdd�D ]K}||� ts�|j r�zt� |j � W n, t y� } z t r}tr}|jtjkr}W Y d }~qR|jtjtjhvr�� W Y d }~qRd }~ww tj�|j �s�J |��qRdd� }|t�}d d � t jdd�D �}||v s�J �d S ) Nc S sD t | jt�sJ �t | jt�sJ �t | jt�sJ �t | jt�s J �d S r; )r� �devicer� � mountpoint�fstype�opts�r" r$ r$ r* � check_ntuple� s z7TestDiskAPIs.test_disk_partitions.<locals>.check_ntupleF)�all�cdromTc S s8 t j�| �} t j�| �st j�| �} t j�| �r | �� S r; )r- r� �abspath�ismount�dirnamer� )r� r$ r$ r* �find_mount_point� s �z;TestDiskAPIs.test_disk_partitions.<locals>.find_mount_pointc S s g | ] }|j r|j �� �qS r$ )r; r� r' r$ r$ r* r+ � s ��z5TestDiskAPIs.test_disk_partitions.<locals>.<listcomp>)r/ �disk_partitionsr r= r r- r� r� r: r; r< �stat�OSErrorr r �errno�EIO�EPERM�EACCES�__file__)r6 r? rC �diskr rE �mount�mountsr$ r$ r* �test_disk_partitions� sF ���� �z!TestDiskAPIs.test_disk_partitionsz/proc/diskstatsz3/proc/diskstats not available on this linux versionr� c C sv dd� }t jdd�}|d usJ d��||� t jdd�}t|�tt|��ks(J �|D ]}|s2J |��||| � q*d S )Nc S s� | d | j ks J �| d | jksJ �| d | jksJ �| d | jks$J �tsctsc| d | jks1J �| d | jks:J �trX| d | j ksEJ �| d | j ksNJ �| d | j ksWJ �n t rc| d | j kscJ �| j D ] }t| |�dkssJ | ��qfd S ) Nr rr rF rt � � r� � � )� read_count� write_count� read_bytes� write_bytesr r � read_time� write_timer �read_merged_count�write_merged_count� busy_timer r� r� )r" r� r$ r$ r* r? � s �z8TestDiskAPIs.test_disk_io_counters.<locals>.check_ntupleF��perdiskzno disks on this system?T)r/ �disk_io_countersr{ rB �r6 r? �retr? r$ r$ r* �test_disk_io_counters� s �z"TestDiskAPIs.test_disk_io_countersc C �f t jdi d��"}tjdd�d u sJ �tjdd�i ks J �|js!J �W d � d S 1 s,w Y d S )Nz#psutil._psplatform.disk_io_countersr� Fr_ T)r rM r/ ra r[ �r6 r] r$ r$ r* � test_disk_io_counters_no_disks� s � "�z+TestDiskAPIs.test_disk_io_counters_no_disksN)rc rd re r r� r� r r r5 r r7 r9 rQ r r- r� r� r r/ ra rd rg r$ r$ r$ r* r+ m s$ � :�� r+ c @ sz e Zd Zejje dd�dd� �Zejje dd�dd� �Zdd� Z d d � Z d d � Z ejje p2e p2e d d�dd� �ZdS )� TestNetAPIsr rg c C sd dd� }t jdd�}||� t jdd�}|g ksJ �|D ]}|s"J �t|t�s)J �||| � q d S )Nc S s$ | d | j ks J �| d | jksJ �| d | jksJ �| d | jks$J �| d | jks-J �| d | jks6J �| d | jks?J �| d | jksHJ �| j dksQJ | ��| jdksZJ | ��| jdkscJ | ��| jdkslJ | ��| jdksuJ | ��| jdks~J | ��| jdks�J | ��| jdks�J | ��d S ) Nr rr rF rt rR rS r� rT )� bytes_sent� bytes_recv� packets_sent� packets_recv�errin�errout�dropin�dropoutr> r$ r$ r* r? � s z6TestNetAPIs.test_net_io_counters.<locals>.check_ntupleF��pernicT)r/ �net_io_countersr� r� rb r$ r$ r* �test_net_io_counters� s �z TestNetAPIs.test_net_io_countersc C re )Nz"psutil._psplatform.net_io_countersr� Frq T)r rM r/ rs r[ rf r$ r$ r* � test_net_io_counters_no_nics s � "�z(TestNetAPIs.test_net_io_counters_no_nicsc C s� t �� }|s J |��t �� }tjtjt jh}|�� D �]\}}t|t �s&J �t t |��t |�ks2J �|D ]�}t|j t �s>J �t|jt �sFJ �t|jt td �f�sRJ �t|jt td �f�s^J �|j |v seJ �t|j tj�snJ �|| jr�|j tjkr�t�|j ��}|�|jdf� W d � n1 s�w Y n9|j tjkr�t�|jdtjtjdtj�d }|\} } } } } t�| | | �� }|�| � W d � n1 s�w Y |j|j|j|jfD ]}|d ur�|j tjkr�t||j � q�|jr�|jd u s�J �n |j�r|jd u �sJ �|j�r!|j�r!|j tjtjhv �r!|jt|�k�s!J �q4qt �s,t �s,t �r?t td��r;t jtjk�s=J �d S d S t �rMt jtj!k�sKJ �d S t"�rXt jdk�sZJ �d S d S )Nr �AF_LINKrp )#r/ � net_if_addrs� net_if_stats�socket�AF_INET�AF_INET6rv �itemsr� r� r{ rB �familyr� �address�netmaskr� � broadcast�enum�IntEnum�isup�bind� getaddrinfo� SOCK_STREAM� AI_PASSIVE�ptpr r r r r r� r � AF_PACKETr )r6 �nics� nic_stats�families�nic�addrs�addr�srX �af�socktype�proto� _canonname�sa�ipr$ r$ r* �test_net_if_addrs% s� �� �� �� �����3 ��z TestNetAPIs.test_net_if_addrsc C s� t r dtjdd d d fg}ndg}tjd|d��/}t�� d d }|js%J �t r/|jdks.J �n|jdks6J �W d � d S W d � d S 1 sIw Y d S ) N�em1z06:3d:29)r� rp z06-3d-29NNNz psutil._psplatform.net_if_addrsr� r z06:3d:29:00:00:00z06-3d-29-00-00-00)r r/ rv r rM rw r[ r~ )r6 rc r] r� r$ r$ r* � test_net_if_addrs_mac_null_byteso s � �"�z,TestNetAPIs.test_net_if_addrs_mac_null_bytesc C s� t �� }|s J |��t jt jt jf}|�� D ]8\}}t|t�s!J �|\}}}}} t|t�s/J �||v s5J �||v s;J �|dksAJ �|dksGJ �t| t�sNJ �qd S r� ) r/ rx �NIC_DUPLEX_FULL�NIC_DUPLEX_HALF�NIC_DUPLEX_UNKNOWNr| r� r� r� ) r6 r� � all_duplexesr� �statsr� �duplex�speed�mtu�flagsr$ r$ r* �test_net_if_stats� s � �z TestNetAPIs.test_net_if_statsz LINUX or BSD or MACOS specificc C sZ t jdttjd�d��}t�� }|i ksJ �|jsJ �W d � d S 1 s&w Y d S )Nz psutil._psutil_posix.net_if_mturV rG )r rM rH rI �ENODEVr/ rx r[ )r6 r] rc r$ r$ r* �test_net_if_stats_enodev� s � "�z$TestNetAPIs.test_net_if_stats_enodevN)rc rd re r r� r� r rt ru r� r� r� r r r r� r$ r$ r$ r* rh � s J�rh c @ s� e Zd Zejje dd�dd� �Zejje dd�dd� �Zejje dd�ejje dd�dd � ��Z ejje dd�d d � �Z d S ) �TestSensorsAPIsr rg c C s� t �� }|�� D ]<\}}t|t�sJ �|D ].}t|jt�s J �|jd ur+|jdks+J �|jd ur7|jdks7J �|jd urC|jdksCJ �qqd S r� ) r/ �sensors_temperaturesr| r� r� �labelr �high�critical)r6 �tempsr� �entries�entryr$ r$ r* �test_sensors_temperatures� s ���z)TestSensorsAPIs.test_sensors_temperaturesc C s� ddgi}t jd|d��-}tjdd�d d }|js J �|jdks#J �|jd ks*J �|jd ks1J �W d � d S 1 s<w Y d S ) N�coretemp)r� g I@g N@g �Q@z'psutil._psplatform.sensors_temperaturesr� T)� fahrenheitr g �^@g �a@g �c@)r rM r/ r� r[ r r� r� )r6 �dr] r� r$ r$ r* �#test_sensors_temperatures_fahreneit� s � "�z3TestSensorsAPIs.test_sensors_temperatures_fahreneitz no batteryc C sp t �� }|jdks J �|jdksJ �|jt jt jhvr#|jdks"J �n |jt jkr.|js.J �t|jt�s6J �d S )Nr r� ) r/ �sensors_batteryr� �secsleft�POWER_TIME_UNKNOWN�POWER_TIME_UNLIMITED� power_pluggedr� r� )r6 rc r$ r$ r* �test_sensors_battery� s � z$TestSensorsAPIs.test_sensors_batteryc C sd t �� }|�� D ]'\}}t|t�sJ �|D ]}t|jt�s J �t|jt�s'J �|jdks.J �qqd S r� )r/ � sensors_fansr| r� r� r� r r� )r6 �fansr� r� r� r$ r$ r* �test_sensors_fans� s ��z!TestSensorsAPIs.test_sensors_fansN)rc rd re r r� r� r r� r� r r r� r r� r$ r$ r$ r* r� � s r� )8�__doc__r� r� rI r- r� r r. r� ry r� r� �unittestr r/ r r r r r r r r r r �psutil._commonr � psutil.testsr r r r r r r r r r r r r r r r r r r r! r" rf r� r� r� r+ rh r� r$ r$ r$ r* �<module> sr HfO+ & tests/__pycache__/test_misc.cpython-310.pyc 0000644 00000060114 15030211765 0014613 0 ustar 00 o �h�s � @ s� d Z ddlZddlZddlZddlZddlZddlZddlZddlZddl m Z ddl Z ddl Z ddl m Z ddlmZ ddlmZ ddlmZ ddlmZ dd lmZ dd lmZ dd lmZ dd lmZ dd lmZ ddl mZ ddl mZ ddl mZ ddl mZ ddl m Z ddl m Z G dd� de�Z G dd� de�Z G dd� de�Z G dd� de�Z!e�"d d �Z#G d d � d e�Z$dS ) zMiscellaneous tests.� N)�mock)�WINDOWS)�bcat)�cat)�debug)� isfile_strict)�memoize)�memoize_when_activated��parse_environ_block)� supports_ipv6�� wrap_numbers)�HAS_NET_IO_COUNTERS)�PsutilTestCase)�process_namespace)�pytest)� reload_module)�system_namespacec @ s� e Zd Zdd� Zefdd�Zdd� Zdd� Zd d � Zd d � Z d d� Z dd� Z dd� Z dd� Z dd� Zdd� Zdd� Zdd � Zd d � Zd S ) �TestSpecialMethodsc C s| t �t�� tjj�dd � W d � n1 sw Y t �tj�� t�dd � W d � d S 1 s7w Y d S )N� � ) r �raises� OverflowError�psutil� _psplatform�cext�check_pid_range� NoSuchProcess�Process��self� r" �I/usr/local/CyberCP/lib/python3.10/site-packages/psutil/tests/test_misc.py�test_check_pid_range- s �"�z'TestSpecialMethods.test_check_pid_rangec C s t �| �� j�}||�}d|v sJ �d|j� �|v s J �d|�� � d�|�dd�v s,J �d|v s2J �d|vs8J �|�� |�� ||�}d|v sJJ �d|v sPJ �tj j t jd t � t � � �d ��& t �� }||�}d|j� �|v srJ �d |v sxJ �d |vs~J �W d � n1 s�w Y tj j t jd t �t � � �d ��& t �� }||�}d|j� �|v s�J �d |v s�J �d |vs�J �W d � n1 s�w Y tj j t jd t �t � � �d ��! t �� }||�}d|j� �|v s�J �d |vs�J �W d � d S 1 s�w Y d S )Nzpsutil.Processzpid=zname='�'zname=u'zstatus=z exitcode=zstatus='terminated'�name�� side_effectzstatus='zombie'zname=� terminated)r r �spawn_testproc�pidr& �replace� terminate�waitr �patch�object� ZombieProcess�os�getpidr � AccessDenied)r! �func�p�rr" r" r# �test_process__repr__3 sX � � � � �"�z'TestSpecialMethods.test_process__repr__c C s | j td� d S )N)r5 )r8 �strr r" r" r# �test_process__str___ � z&TestSpecialMethods.test_process__str__c C � t t�� �dks J �d S )Nzpsutil.Error())�reprr �Errorr r" r" r# �test_error__repr__b � z%TestSpecialMethods.test_error__repr__c C r< )N� )r9 r r> r r" r" r# �test_error__str__e r@ z$TestSpecialMethods.test_error__str__c C �6 t t�d��dks J �t tjdddd��dksJ �d S )N�A z=psutil.NoSuchProcess(pid=321, msg='process no longer exists')r& �msg�r& rE z5psutil.NoSuchProcess(pid=321, name='name', msg='msg'))r= r r r r" r" r# � test_no_such_process__repr__h � � �z/TestSpecialMethods.test_no_such_process__repr__c C rC )NrD z"process no longer exists (pid=321)r& rE rF �msg (pid=321, name='name'))r9 r r r r" r" r# �test_no_such_process__str__r rH z.TestSpecialMethods.test_no_such_process__str__c C �8 t t�d��dks J �t tjddddd��dksJ �d S )NrD zGpsutil.ZombieProcess(pid=321, msg="PID still exists but it's a zombie")r& �@ �foo�r& �ppidrE z?psutil.ZombieProcess(pid=321, ppid=320, name='name', msg='foo'))r= r r1 r r" r" r# �test_zombie_process__repr__| s � �z.TestSpecialMethods.test_zombie_process__repr__c C rK )NrD z,PID still exists but it's a zombie (pid=321)r& rL rM rN z$foo (pid=321, ppid=320, name='name'))r9 r r1 r r" r" r# �test_zombie_process__str__� s � �z-TestSpecialMethods.test_zombie_process__str__c C rC )NrD z psutil.AccessDenied(pid=321)r& rE rF z4psutil.AccessDenied(pid=321, name='name', msg='msg'))r= r r4 r r" r" r# �test_access_denied__repr__� � �z-TestSpecialMethods.test_access_denied__repr__c C rC )NrD z (pid=321)r& rE rF rI )r9 r r4 r r" r" r# �test_access_denied__str__� rS z,TestSpecialMethods.test_access_denied__str__c C rC )N� z?psutil.TimeoutExpired(seconds=5, msg='timeout after 5 seconds')rD r& �r+ r& zUpsutil.TimeoutExpired(pid=321, name='name', seconds=5, msg='timeout after 5 seconds'))r= r �TimeoutExpiredr r" r" r# � test_timeout_expired__repr__� s � �z/TestSpecialMethods.test_timeout_expired__repr__c C rC )NrU ztimeout after 5 secondsrD r& rV z.timeout after 5 seconds (pid=321, name='name'))r9 r rW r r" r" r# �test_timeout_expired__str__� rS z.TestSpecialMethods.test_timeout_expired__str__c C s> t �� }t �� }||ksJ �d|_||ksJ �|dks J �d S )N)r r rM )r r �_ident)r! �p1�p2r" r" r# �test_process__eq__� s z%TestSpecialMethods.test_process__eq__c C s$ t �� t �� h}t|�dksJ �d S )N� )r r �len)r! �sr" r" r# �test_process__hash__� s z'TestSpecialMethods.test_process__hash__N)�__name__� __module__� __qualname__r$ r= r8 r: r? rB rG rJ rP rQ rR rT rX rY r] ra r" r" r" r# r , s , r c @ �< e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd S )�TestMiscc C s� t t�}|D ]@}|dv r q|�d�sFzt|� W q tyE |tjvrCtt|�}|d u r/Y q|jd urCd|j�� vrC| � |�d���Y qw qtjD ]}||v sRJ �qJd S )N> �testr �tests�PermissionError�ProcessLookupError�_� deprecatedz not in psutil.__all__) �dirr � startswith� __import__� ImportError�__all__�getattr�__doc__�lower�fail)r! � dir_psutilr& �funr" r" r# � test__all__� s* ��� �zTestMisc.test__all__c C s$ d� dd� tjD ��tjksJ �d S )N�.c S s g | ]}t |��qS r" )r9 )�.0�xr" r" r# � <listcomp>� s z)TestMisc.test_version.<locals>.<listcomp>)�joinr � version_info� __version__r r" r" r# � test_version� s �zTestMisc.test_versionc C s" t �� }d|_d|�� vsJ �d S )N�1rM )r r rM �as_dict)r! r6 r" r" r# �!test_process_as_dict_no_new_names� s z*TestMisc.test_process_as_dict_no_new_namesc C s� dd� }t �� }|t �� �� � t|�}|j|jdd�D ]/\}}| j||d�� z|� }W n t jy7 Y nw ||� W d � n1 sFw Y q t� }|�|j�D ]3\}}|dv r^qU| j|d�� z|� }W n t j yt Y nw ||� W d � n1 s�w Y qUt � t � t j dd d d ���}t|t j �s�J �|jdks�J �|jd ks�J �|jd ks�J �t � t � t jdd d d d ���}t|t j�s�J �|jdks�J �|jd ks�J �|jd ks�J �|jd ks�J �t � t � t j dd d d ���}t|t j �s�J �|jdk�sJ �|jd k�s J �|jd k�sJ �t � t � t jddd d���}t|t j��s,J �|jdk�s4J �|jdk�s<J �|jd k�sDJ �d S )Nc S s4 t �t �| �� t�| �}t�|�}| |ksJ �d S �N)�json�loads�dumps�pickle)�ret�a�br" r" r# �check� s z*TestMisc.test_serialization.<locals>.checkT)� clear_cache)�procr& > �win_service_get�win_service_iter)r& i� r& rE )r+ r& rE �* )r+ r& rO rE �{ �! )�secondsr+ r& )r r r� r �iter�getters�subTestr> r r4 r� r� r� r � isinstancer+ r& rE r1 rO rW r� )r! r� r� �nsrw r& r� r� r" r" r# �test_serialization� s� �� �� �� �� �������zTestMisc.test_serializationc C s� t jjtjdtjd��}t�� |jsJ �W d � n1 s w Y t jjtjdt�d�d��}t�� |js;J �W d � n1 sEw Y t jjtjdtd��&}t � t�� t�� W d � n1 siw Y |jssJ �W d � n1 s}w Y t jjtjdt� d�d��(}| � tj �� t�� W d � n1 s�w Y |js�J �W d � d S 1 s�w Y d S )N� _get_identr' r^ ) r r/ r0 r r r4 �calledr1 � ValueErrorr r r � assertRaises)r! �methr" r" r# �test_ad_on_process_creation@ s@ � �� �� � �� � "�z$TestMisc.test_ad_on_process_creationc C s| t jddd��- t�t�� }tt� W d � n1 s w Y dt|j�� � v s,J �W d � d S 1 s7w Y d S )Nz psutil._psplatform.cext.versionz0.0.0�� return_valuezversion conflict) r r/ r r rp r r r9 �valuert )r! �cmr" r" r# �test_sanity_version_check] s � �"�z"TestMisc.test_sanity_version_checkN) rb rc rd rx r� r� r� r� r� r" r" r" r# rf � s N rf c @ sR e Zd Zdd� ZeZddd�Zdd� Zdd � Zd d � Zd d � Z dd� Z dd� Z dS )�TestMemoizeDecoratorc C s g | _ d S r� ��callsr r" r" r# �setUpm s zTestMemoizeDecorator.setUpNc C s* t d�D ]}|� }| jdi fgksJ �|d ur ||ks J �qt d�D ] }|d�}| jdi fdi fgks5J �|d ur?||ks?J �q"t d�D ]$}|ddd�}| jdi fdi fdddifgks^J �|d urh||kshJ �qDt| j�dksrJ �|�� |� }|d ur�||ks�J �t| j�dks�J �|jd ks�J �d S ) Nr r" r^ �r^ ��barr� � � � My docstring.)�ranger� r_ � cache_clearrs )r! �obj�expected_retvalrk r� r" r" r# � run_againstr s2 � � $ � z TestMemoizeDecorator.run_againstc s& t � fdd��}| � | j|dd� d S )Nc � � j �| |f� dS �r� � �r� �append��args�kwargs�� baseclassr" r# rM � s z/TestMemoizeDecorator.test_function.<locals>.foor� �r� )r r� )r! rM r" r� r# � test_function� s z"TestMemoizeDecorator.test_functionc s> t G � fdd�d��}| � | j|d d� |� �� dks J �d S )Nc s$ e Zd ZdZ� fdd�Zdd� ZdS )z,TestMemoizeDecorator.test_class.<locals>.Foor� c s � j �||f� d S r� r� �r! r� r� r� r" r# �__init__� s z5TestMemoizeDecorator.test_class.<locals>.Foo.__init__c S s dS )Nr� r" r r" r" r# r� � � z0TestMemoizeDecorator.test_class.<locals>.Foo.barN)rb rc rd rs r� r� r" r� r" r# �Foo� s r� r� r� )r r� r� �r! r� r" r� r# � test_class� s z TestMemoizeDecorator.test_classc C s� t G dd� d��}|� |� u sJ �t|� �t|� �ks J �t|d��t|d��ks+J �t|ddd��t|ddd��ks=J �t|d��t|d��ksKJ �d S )Nc @ s e Zd Zdd� ZdS )z6TestMemoizeDecorator.test_class_singleton.<locals>.Barc _ s d S r� r" r� r" r" r# r� � r� z?TestMemoizeDecorator.test_class_singleton.<locals>.Bar.__init__N)rb rc rd r� r" r" r" r# �Bar� s r� r^ r� )rM r )r �id)r! r� r" r" r# �test_class_singleton� s $ z)TestMemoizeDecorator.test_class_singletonc �, G � fdd�d�}| � | j |� jdd� d S )Nc � e Zd Zee� fdd���ZdS )z3TestMemoizeDecorator.test_staticmethod.<locals>.Fooc r� r� r� r� r� r" r# r� � � z7TestMemoizeDecorator.test_staticmethod.<locals>.Foo.barN)rb rc rd � staticmethodr r� r" r� r" r# r� � � r� r� r� �r� r� r� r" r� r# �test_staticmethod� � z&TestMemoizeDecorator.test_staticmethodc r� )Nc r� )z2TestMemoizeDecorator.test_classmethod.<locals>.Fooc s � j �||f� dS r� r� )�clsr� r� r� r" r# r� � r� z6TestMemoizeDecorator.test_classmethod.<locals>.Foo.barN)rb rc rd � classmethodr r� r" r� r" r# r� � r� r� r� r� r� r� r" r� r# �test_classmethod� r� z%TestMemoizeDecorator.test_classmethodc s t � fdd��}g � td�D ]}|� }di f}||ks J �t� �dks%J �qtd�D ]}|d�}di f}||ks:J �t� �dksBJ �q*td�D ] }|ddd�}dddif}||ks[J �t� �d kscJ �qG|�� |� }di f}||ksuJ �t� �d ks}J �|jd ks�J �d S ) Nc s � � d� | |fS )�Foo docstring.N�r� r� r� r" r# rM � s z/TestMemoizeDecorator.test_original.<locals>.foor r" r^ r� r� r� r� r� r� )r r� r_ r� rs )r! rM rk r� �expectedr" r� r# � test_original� s0 z"TestMemoizeDecorator.test_originalr� ) rb rc rd r� �tearDownr� r� r� r� r� r� r� r" r" r" r# r� l s r� c @ re )�TestCommonModulec s� G � fdd�d�}|� }g � |� � |� � t� �dks J �g � |j �|� |� � |� � t� �dks6J �g � |j �|� |� � |� � t� �dksNJ �d S )Nc s e Zd Ze� fdd��ZdS )z9TestCommonModule.test_memoize_when_activated.<locals>.Fooc s � � d � d S r� r� r r� r" r# rM � s z=TestCommonModule.test_memoize_when_activated.<locals>.Foo.fooN)rb rc rd r rM r" r� r" r# r� � s r� r r^ )rM r_ �cache_activate�cache_deactivate)r! r� �fr" r� r# �test_memoize_when_activated� s z,TestCommonModule.test_memoize_when_activatedc C s� dd� }t d�|d�diksJ �t d�|d�d|d�diks J �t d �|d�d|d�d iks0J �t d �|d�d|d�diks@J �t d �|d�diksLJ �t d �|d�diksXJ �t d�|d�diksdJ �d S )Nc S s t r| �� S | S r� )r �upper)r` r" r" r# �k r; z4TestCommonModule.test_parse_environ_block.<locals>.kza=1 r� r� z a=1 b=2 r� �2za=1 b= rA z a=1 b=2 c=3 zxxx a=1 z a=1 =b=2 za=1 b=2r )r! r� r" r" r# �test_parse_environ_block s � � z)TestCommonModule.test_parse_environ_blockc C s� | � tj� t� r�t�d��}d|_t�� t� rJ �W d � n1 s%w Y t�� tjdtd��}t� r;J �|js@J �W d � n1 sJw Y t�� tjdtj d��}t� raJ �t�� |jsjJ �W d � n1 stw Y t�� tjdtj d��}t� r�J �t�� |js�J �W d � d S 1 s�w Y d S t � t��" t�tj tj �}z |�d� W |�� n|�� w W d � d S 1 s�w Y d S )Nzpsutil._common.socketFz psutil._common.socket.socketr' z!psutil._common.socket.socket.bind)z::1r )� addCleanupr r� r r/ �has_ipv6�OSErrorr� �socket�gaierrorr r �AF_INET6� SOCK_STREAM�bind�close)r! r` �sockr" r" r# �test_supports_ipv6# sP �� �� �� "� "�z#TestCommonModule.test_supports_ipv6c C s t j�t�}t|�s J �tt j�|��rJ �tjdtd��! t � t �� t|� W d � n1 s2w Y W d � n1 sAw Y tjdt d�� t|�rTJ �W d � n1 s^w Y tjddd�� t|�rqJ �W d � d S 1 s|w Y d S )Nzpsutil._common.os.statr' zpsutil._common.stat.S_ISREGFr� ) r2 �path�abspath�__file__r �dirnamer r/ ri r r r� �FileNotFoundError)r! � this_filer" r" r# �test_isfile_strictJ s" �� �� �"�z#TestCommonModule.test_isfile_strictc C s� t j�tjdd��( t�t�� ��}t d� t j � � W d � n1 s%w Y W d � n1 s4w Y |� � }|�d�sFJ |��d|v sLJ �t�dd�|v sVJ �t j�tjdd��% t�t�� ��}t td�� W d � n1 sxw Y W d � n1 s�w Y |� � }d|v s�J �d |v s�J �t j�tjdd��+ t�t�� ��}td d �}d |_t |� W d � n1 s�w Y W d � n1 s�w Y |� � }d |v s�J �d |v s�J �d S ) N� PSUTIL_DEBUGT�helloz psutil-debugz.pycz.pyzthis is an errorzignoring ValueErrorz'this is an error'r z no such filez/foo)r r/ r0 r �_common� contextlib�redirect_stderr�io�StringIOr �sys�stderr�flush�getvaluern r� r, r� r� �filename)r! r� rE �excr" r" r# � test_debugX s> �� � �� � �� � zTestCommonModule.test_debugc C s� | � � }t|d�� }|�d� W d � n1 sw Y t|�dks&J �t|�dks.J �t�t�� t|d � W d � n1 sDw Y t�t�� t|d � W d � n1 s_w Y t|d dd�dkspJ �t|d dd�dks|J �d S )N�wrM s fooz-invalidr� )�fallback)� get_testfn�open�writer r r r r� )r! �testfnr� r" r" r# � test_cat_bcatt s � � � z TestCommonModule.test_cat_bcatN) rb rc rd r� r� r� r� r r r" r" r" r# r� � s ' r� rM za b cc @ s� e Zd Zdd� ZeZdd� Zdd� Zdd� Zd d � Zd d � Z d d� Z dd� Z dd� Z dd� Z dd� Zdd� Zdd� Zejje dd �d d � �Zd S ) �TestWrapNumbersc C s t �� d S r� )r r� r r" r" r# r� � s zTestWrapNumbers.setUpc C s& dt ddd�i}t|d�|ksJ �d S �N�disk1rU �disk_io��ntr �r! �inputr" r" r# �test_first_call� s z TestWrapNumbers.test_first_callc C s8 dt ddd�i}t|d�|ksJ �t|d�|ksJ �d S r r r r" r" r# �test_input_hasnt_changed� s z(TestWrapNumbers.test_input_hasnt_changedc C s� dt ddd�i}t|d�|ksJ �dt ddd�i}t|d�|ks"J �dt ddd�i}t|d�|ks3J �dt ddd�i}t|d�|ksDJ �d S ) Nr rU r � � � � � r r r" r" r# �test_increase_but_no_wrap� s z)TestWrapNumbers.test_increase_but_no_wrapc C s� dt ddd�i}t|d�|ksJ �dt ddd�i}t|d�dt ddd�iks(J �dt ddd�i}t|d�dt ddd�iks?J �dt ddd�i}t|d�dt ddd�iksVJ �dt ddd�i}t|d�dt ddd �iksmJ �dt ddd�i}t|d�dt ddd �iks�J �dt d dd�i}t|d�dt d dd �iks�J �dt d dd�i}t|d�dt ddd �iks�J �dt d dd�i}t|d�dt ddd �iks�J �d S ) Nr �d r r �n �Z � r �� �2 � �( r r r" r" r# � test_wrap� s$ "zTestWrapNumbers.test_wrapc C st dt ddd�i}t|d�|ksJ �t ddd�t ddd�d�}t|d�|ks'J �dt ddd�i}t|d�|ks8J �d S )Nr rU r � �r �disk2� r r r" r" r# �test_changing_keys� s z"TestWrapNumbers.test_changing_keysc C s. t ddd�t ddd�d�}t|d�|ksJ �t ddd�t ddd�d�}t|d�t ddd�t ddd�d�ks7J �dt ddd�i}t|d�|ksHJ �t ddd�t ddd�d�}t|d�|ks^J �t ddd�t ddd�d�}t|d�|kstJ �t ddd�t ddd�d�}t|d�t ddd�t ddd�d�ks�J �d S )Nr r r$ r r r r r r r" r" r# �test_changing_keys_w_wrap� s$ � �z)TestWrapNumbers.test_changing_keys_w_wrapc C sb ddddd�}t |d�|ksJ �t |d�|ksJ �ddddd�}t |d�}|d d d ks/J �d S ) N) i, � � �# �R �� � �� �ȷ ) i� r i tU i i r r r r& ) �6 r1 i �$ i �N r� � r i� � ) iU i� i" r i� i4 i� i� i\ )�nvme0n1� nvme0n1p1� nvme0n1p2� nvme0n1p3r ) r r) r* r+ r, r- r. r/ r0 r4 r i� r )r! �d�outr" r" r# �test_real_data� s �� z TestWrapNumbers.test_real_datac C sb dt ddd�i}t|d� t�� }|d d|iksJ �|d di iks%J �|d di iks/J �d S )Nr rU r r r^ r �r r � cache_info�r! r �cacher" r" r# �test_cache_first_call� s z%TestWrapNumbers.test_cache_first_callc C s� dt ddd�i}t|d� dt ddd�i}t|d� t�� }|d d|iks(J �|d ddddd�iks6J �|d di iks@J �d S ) Nr rU r r r r^ �)r r )r r^ �r r r r; r= r" r" r# �test_cache_call_twice s �z%TestWrapNumbers.test_cache_call_twicec C sv dt ddd�i}t|d� dt ddd�i}t|d� t�� }|d d|iks(J �|d ddddd�iks6J �|d ddd hiiksCJ �d d � }dt ddd�i}t|d� t�� }|d d|iksbJ �|� dt ddd �i}t|d� t�� }|d d|iks�J �|� dt ddd �i}t|d� t�� }|d d|iks�J �|d ddddd�iks�J �|d ddd hiiks�J �d S )Nr r r r r r^ r@ r rA c S sB t �� } | d ddddd�iksJ �| d dddhiiks J �d S ) Nr^ r r r r@ r r rA )r r< )r> r" r" r# �check_cache_info s � � z9TestWrapNumbers.test_cache_wrap.<locals>.check_cache_infor r r r; )r! r r> rC r" r" r# �test_cache_wrap s: � � z TestWrapNumbers.test_cache_wrapc C s� dt ddd�i}t|d� t ddd�t ddd�d�}t|d� t�� }|d d|iks-J �|d ddddd�iks;J �|d di iksEJ �d S ) Nr rU r r# r$ r r^ r@ r r; r= r" r" r# �test_cache_changing_keysA s �z(TestWrapNumbers.test_cache_changing_keysc C s\ dt ddd�i}t|d� t|d� t�d� t�� i i i fks"J �t�d� t�d� d S )Nr rU r z?!?)r r r� r< r r" r" r# �test_cache_clearM s z TestWrapNumbers.test_cache_clearz not supported)�reasonc C s� t �� rt �� s t�d��t �� t �� t�� }|D ]}d|v s#J �d|v s)J �qt j�� t�� }|D ]}d|v s=J �d|vsCJ �q5t j�� t�� }|i i i fksVJ �d S )Nzno disks or NICs availablezpsutil.disk_io_counterszpsutil.net_io_counters)r �disk_io_counters�net_io_countersr �skipr r< r� )r! �cachesr> r" r" r# � test_cache_clear_public_apisV s z,TestWrapNumbers.test_cache_clear_public_apisN)rb rc rd r� r� r r r r" r' r( r: r? rB rD rE rF r �mark�skipifr rL r" r" r" r# r � s" 2 r )%rs � collectionsr� r� r� r2 r� r� r� �unittestr r � psutil.testsr �psutil._commonr r r r r r r r r r r r r r r r rf r� r� � namedtupler r r" r" r" r# �<module> sL ( tests/__pycache__/test_scripts.cpython-310.pyc 0000644 00000020635 15030211765 0015353 0 ustar 00 o �h- � @ s� d Z ddlZddlZddlZddlZddlZddlZddlmZ ddlm Z ddl m Z ddl m Z ddl m Z ddl mZ dd l mZ dd l mZ dd l mZ dd l mZ dd l mZ ddl mZ ddl mZ ddl mZ ddl mZ ddl mZ ej�ed�Zej�ed�Zej j e o�ej� e� dd�G dd� de��Z ej j e o�ej� e� dd�G dd� de��Z ej j e o�ej� e� d d�G d d � d e��Z!dS ) zTest various scripts.� N)�POSIX)�WINDOWS)� CI_TESTING)� HAS_BATTERY)�HAS_MEMORY_MAPS)�HAS_SENSORS_BATTERY)�HAS_SENSORS_FANS)�HAS_SENSORS_TEMPERATURES)� PYTHON_EXE)�PYTHON_EXE_ENV)�ROOT_DIR)� SCRIPTS_DIR)�PsutilTestCase)�import_module_by_path)�psutil)�sh�internalzsetup.pyz can't find scripts/ directory��reasonc @ s� e Zd Zedd� �Zedd� �Zdd� Zejj e dd�d d � �Z d d � Z d d� Z dd� Zdd� Zejj eo:e�� dd�dd� �Zdd� Zdd� Zdd� Zd d � Zejj e d d�d d � �Zd!d"� Zd#d$� Zd%d&� Z d'd(� Z d)d*� Z d+d,� Z ejj e d-d�d.d/� �Z!d0d1� Z"ejj e# d d�d2d3� �Z$ejj e% d d�d4d5� �Z&ejj e' d d�ejj e( d6d�d7d8� ��Z)ejj e' d d�ejj e( d6d�d9d:� ��Z*d;S )<�TestExampleScriptsc O s� |� dt� tj�t| �} t| g}|D ]}|�|� qz t|fi |��� � }W n t yD } zdt |�v r?t |�W Y d }~S � d }~ww |sKJ |��|S )N�env� AccessDenied) � setdefaultr �os�path�joinr r �appendr �strip� RuntimeError�str)�exe�args�kwargs�cmd�arg�out�err� r' �L/usr/local/CyberCP/lib/python3.10/site-packages/psutil/tests/test_scripts.py� assert_stdout1 s �� z TestExampleScripts.assert_stdoutc C sP t j�t| �} t| dd�� }|�� }W d � n1 s w Y t�|� d S �N�utf8)�encoding)r r r r �open�read�ast�parse)r �f�srcr' r' r( � assert_syntaxB s �z TestExampleScripts.assert_syntaxc C sZ t | �}t�t�D ]!}|�d�r*dtj�|�d |vr*| �dtj�t|��d���q d S )N�.py�test_r zno test defined for z script) �dirr �listdirr �endswithr �splitext�failr )�self�meths�namer' r' r( � test_coverageI s ����z TestExampleScripts.test_coveragez POSIX onlyr c C sb t �t�D ])\}}}|D ]!}|�d�r-t j�||�}tjt �|�tj @ s-| � |�d���q qd S )Nr4 z is not executable) r �walkr r8 r r �stat�S_IXUSR�ST_MODEr: )r; �root�dirs�files�filer r' r' r( �test_executableU s ���z"TestExampleScripts.test_executablec C � | � d� d S )Nz disk_usage.py�r) �r; r' r' r( �test_disk_usage^ � z"TestExampleScripts.test_disk_usagec C rH )Nzfree.pyrI rJ r' r' r( � test_freea rL z TestExampleScripts.test_freec C rH )Nz meminfo.pyrI rJ r' r' r( � test_meminfod rL z TestExampleScripts.test_meminfoc C � | � dtt�� �� d S )Nz procinfo.py�r) r r �getpidrJ r' r' r( � test_procinfog s z TestExampleScripts.test_procinfozno usersc C rH )Nzwho.pyrI rJ r' r' r( �test_whoj � zTestExampleScripts.test_whoc C rH )Nzps.pyrI rJ r' r' r( �test_psn rL zTestExampleScripts.test_psc C rH )Nz pstree.pyrI rJ r' r' r( � test_pstreeq rL z TestExampleScripts.test_pstreec C rH )Nz netstat.pyrI rJ r' r' r( � test_netstatt rL z TestExampleScripts.test_netstatc C rH )Nz ifconfig.pyrI rJ r' r' r( � test_ifconfigw rL z TestExampleScripts.test_ifconfig� not supportedc C rO )Nzpmap.pyrP rJ r' r' r( � test_pmapz s z TestExampleScripts.test_pmapc C s* dt �� �� jvrt�d��| �d� d S )N�ussrY z procsmem.py)r �Process�memory_full_info�_fields�pytest�skipr) rJ r' r' r( � test_procsmem~ s z TestExampleScripts.test_procsmemc C rH )Nz killall.py�r3 rJ r' r' r( � test_killall� rL z TestExampleScripts.test_killallc C rH )Nz nettop.pyrb rJ r' r' r( � test_nettop� rL z TestExampleScripts.test_nettopc C rH )Nztop.pyrb rJ r' r' r( �test_top� rL zTestExampleScripts.test_topc C rH )Nziotop.pyrb rJ r' r' r( � test_iotop� rL z TestExampleScripts.test_iotopc C s, | � dt�� �� �}tt�� �|v sJ �d S )Nzpidof.py)r) r r\ r= r r rQ )r; �outputr' r' r( � test_pidof� s z TestExampleScripts.test_pidofz WINDOWS onlyc C rH )Nzwinservices.pyrI rJ r' r' r( �test_winservices� rT z#TestExampleScripts.test_winservicesc C rH )Nzcpu_distribution.pyrb rJ r' r' r( �test_cpu_distribution� rL z(TestExampleScripts.test_cpu_distributionc C � t �� s t�d��| �d� d S )Nzno temperaturesztemperatures.py)r �sensors_temperaturesr_ r` r) rJ r' r' r( �test_temperatures� � z$TestExampleScripts.test_temperaturesc C rk )Nzno fanszfans.py)r � sensors_fansr_ r` r) rJ r' r' r( � test_fans� rn z TestExampleScripts.test_fansz no batteryc C rH )Nz battery.pyrI rJ r' r' r( � test_battery� � z TestExampleScripts.test_batteryc C rH )Nz sensors.pyrI rJ r' r' r( � test_sensors� rr z TestExampleScripts.test_sensorsN)+�__name__� __module__� __qualname__� staticmethodr) r3 r> r_ �mark�skipifr rG rK rM rN rR r r �usersrS rU rV rW rX r rZ ra rc rd re rf rh r ri rj r rm r rp r r rq rs r' r' r' r( r , sN r z&can't find scripts/internal/ directoryc @ s8 e Zd Zedd� �Zdd� Zejje dd�dd� �Z d S ) �TestInternalScriptsc c s0 � t �t�D ]} | �d�rt j�t| �V qd S )Nr4 )r r7 �INTERNAL_SCRIPTS_DIRr8 r r )r= r' r' r( �ls� s � ��zTestInternalScripts.lsc C sP | � � D ]!}t|dd�� }|�� }W d � n1 sw Y t�|� qd S r* )r} r- r. r/ r0 )r; r r1 �datar' r' r( �test_syntax_all� s � �z#TestInternalScripts.test_syntax_allz not on CIr c C s0 | � � D ]}zt|� W q ty Y qw d S �N)r} r � SystemExit)r; r r' r' r( �test_import_all� s ��z#TestInternalScripts.test_import_allN) rt ru rv rw r} r r_ rx ry r r� r' r' r' r( r{ � s r{ zcan't find setup.pyc @ s4 e Zd Zdd� Zejje�d� dd�dd� �Z dS ) �TestSetupScriptc C sP t t�}t�t�� |�� W d � n1 sw Y |�� tjks&J �d S r� ) r �SETUP_PYr_ �raisesr� �setup� get_versionr � __version__)r; �moduler' r' r( �test_invocation� s �z TestSetupScript.test_invocation� python2.7zpython2.7 not installedr c C s` t jt�d�tgt jt jdd�}|�� \}}|�� dks J �|r"J �d|v s(J �d|v s.J �d S )Nr� T)�stdout�stderr�universal_newlines� z$psutil no longer supports Python 2.7z'Latest version supporting Python 2.7 is)� subprocess�Popen�shutil�whichr� �PIPE� communicate�wait)r; �pr� r� r' r' r( � test_python2� s � z TestSetupScript.test_python2N) rt ru rv r� r_ rx ry r� r� r� r' r' r' r( r� � s �r� )"�__doc__r/ r r� r@ r� r_ r r r � psutil.testsr r r r r r r r r r r r r r r r| r� rx ry �existsr r{ r� r' r' r' r( �<module> sP � �� tests/__pycache__/test_testutils.cpython-310.pyc 0000644 00000053541 15030211765 0015726 0 ustar 00 o �h�G � @ s� d Z ddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddlm Z ddl Z ddl Z ddl m Z ddl mZ ddl mZ ddlmZ ddlmZ dd lmZ dd l mZ dd l mZ dd l mZ dd l mZ ddl mZ ddl mZ ddl mZ ddl mZ 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% dd l m&Z& dd l m'Z' dd l m(Z( dd l m)Z) dd l m*Z* dd!l m+Z+ dd"l m,Z, dd#l m-Z- dd$l m.Z. dd%l m/Z/ dd&l m0Z0 dd'l m1Z1 G d(d)� d)e�Z2G d*d+� d+e�Z3G d,d-� d-e�Z4G d.d/� d/e�Z5G d0d1� d1e�Z6e&j7j8d2d3�G d4d5� d5e��Z9G d6d7� d7e�Z:G d8d9� d9e�Z;G d:d;� d;e�Z<dS )<z1Tests for testing utils (psutil.tests namespace).� N)�mock)�FREEBSD)�NETBSD)�POSIX)� open_binary)� open_text)� supports_ipv6)� CI_TESTING)�COVERAGE)�HAS_NET_CONNECTIONS_UNIX)�HERE)� PYTHON_EXE)�PYTHON_EXE_ENV)�PsutilTestCase)�TestMemoryLeak)� bind_socket)�bind_unix_socket�� call_until)�chdir)�create_sockets)� fake_pytest)�filter_proc_net_connections)� get_free_port)� is_namedtuple)�process_namespace)�pytest)� reap_children)�retry)�retry_on_failure)� safe_mkdir)� safe_rmpath)�system_namespace)�tcp_socketpair)� terminate)�unix_socketpair)� wait_for_file)� wait_for_pidc @ sx e Zd Ze�d�dd� �Ze�d�dd� �Ze�d�dd� �Ze�d�dd � �Ze�d�d d � �Z e�d�d d � �Z dS )�TestRetryDecoratorz time.sleepc sF t ddd d�� fdd��}ttd��� |� dksJ �|jdks!J �d S )N� � ��retries�interval�logfunc � � r � � � dd � sdS �Nr* r ��pop� ��queuer3 �N/usr/local/CyberCP/lib/python3.10/site-packages/psutil/tests/test_testutils.py�fooF � �z2TestRetryDecorator.test_retry_success.<locals>.foo� )r �list�range� call_count��self�sleepr7 r3 r4 r6 �test_retry_successB s z%TestRetryDecorator.test_retry_successc sh t ddd d�� fdd��}ttd��� t�t�� |� W d � n1 s&w Y |jdks2J �d S )Nr) r* r+ c r/ r0 r1 r3 r4 r3 r6 r7 T r8 z2TestRetryDecorator.test_retry_failure.<locals>.foo� )r r: r; r �raises�ZeroDivisionErrorr< r= r3 r4 r6 �test_retry_failureQ s �z%TestRetryDecorator.test_retry_failurec C sV t tdd�dd� �}t�t�� |� W d � n1 s w Y |jdks)J �d S )Nr* )� exceptionr- c S s t ��N)� TypeErrorr3 r3 r3 r6 r7 b � z2TestRetryDecorator.test_exception_arg.<locals>.foor )r � ValueErrorr rB rG r< r= r3 r3 r6 �test_exception_arg` s �z%TestRetryDecorator.test_exception_argc C sX t dd d d�dd� �}t�t�� |� W d � n1 s w Y |jdks*J �d S )Nr) r+ c S � dd d S r0 r3 r3 r3 r3 r6 r7 n � z4TestRetryDecorator.test_no_interval_arg.<locals>.foor �r r rB rC r< r= r3 r3 r6 �test_no_interval_argj s �z'TestRetryDecorator.test_no_interval_argc C sX t ddd d�dd� �}t�t�� |� W d � n1 s w Y |jdks*J �d S )Nr) r* r+ c S rK r0 r3 r3 r3 r3 r6 r7 x rL z0TestRetryDecorator.test_retries_arg.<locals>.foorM r= r3 r3 r6 �test_retries_argv s �z#TestRetryDecorator.test_retries_argc C s< t �t�� tddd� W d � d S 1 sw Y d S )Nr) r* )r, �timeout)r rB rI r )r> r? r3 r3 r6 � test_retries_and_timeout_args� s "�z0TestRetryDecorator.test_retries_and_timeout_argsN) �__name__� __module__� __qualname__r �patchr@ rD rJ rN rO rQ r3 r3 r3 r6 r( A s r( c @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd S )�TestSyncTestUtilsc C s� t t�� � tt�� �d }tjdtdg�d��+ t � tj �� t |� W d � n1 s.w Y W d � d S W d � d S 1 sFw Y d S )Ni�� �psutil.tests.retry.__iter__r �� return_value) r' �os�getpid�max�psutil�pidsr rU �iterr rB � NoSuchProcess)r> �nopidr3 r3 r6 �test_wait_for_pid� s ��"�z#TestSyncTestUtils.test_wait_for_pidc C sX | � � }t|d�� }|�d� W d � n1 sw Y t|� tj�|�r*J �d S )N�wr7 �� get_testfn�open�writer& rZ �path�exists�r> �testfn�fr3 r3 r6 �test_wait_for_file� s �z$TestSyncTestUtils.test_wait_for_filec C sR | � � }t|d�� W d � n1 sw Y t|dd� tj�|�r'J �d S )Nrc T)�empty)re rf r& rZ rh ri �r> rk r3 r3 r6 �test_wait_for_file_empty� s � z*TestSyncTestUtils.test_wait_for_file_emptyc C s� | � � }tjdtdg�d��* t�t�� t|� W d � n1 s#w Y W d � d S W d � d S 1 s;w Y d S )NrW r rX )re r rU r_ r rB �OSErrorr& ro r3 r3 r6 �test_wait_for_file_no_file� s ��"�z,TestSyncTestUtils.test_wait_for_file_no_filec C s\ | � � }t|d�� }|�d� W d � n1 sw Y t|dd� tj�|�s,J �d S )Nrc r7 F)�deleterd rj r3 r3 r6 � test_wait_for_file_no_delete� s � z.TestSyncTestUtils.test_wait_for_file_no_deletec C s t dd� � d S )Nc S � dS �Nr* r3 r3 r3 r3 r6 �<lambda>� � z3TestSyncTestUtils.test_call_until.<locals>.<lambda>r �r> r3 r3 r6 �test_call_until� s z!TestSyncTestUtils.test_call_untilN) rR rS rT rb rm rp rr rt rz r3 r3 r3 r6 rV � s rV c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd S ) �TestFSTestUtilsc C �< t t��}|jdks J �W d � d S 1 sw Y d S )N�r)r �__file__�mode�r> rl r3 r3 r6 �test_open_text� � "�z TestFSTestUtils.test_open_textc C r| )N�rb)r r~ r r� r3 r3 r6 �test_open_binary� r� z TestFSTestUtils.test_open_binaryc C s< | � � }t|� tj�|�sJ �t|� tj�|�s J �d S rF )re r rZ rh �isdirro r3 r3 r6 �test_safe_mkdir� s z TestFSTestUtils.test_safe_mkdirc C s� | � � }t|d��� t|� tj�|�rJ �t|� t�|� t|� tj�|�r,J �tj dt t j d�d��'}t �t �� t|� W d � n1 sLw Y |jsVJ �W d � d S 1 saw Y d S )Nrc zpsutil.tests.os.stat� )� side_effect)re rf �closer! rZ rh ri �mkdirr rU rq �errno�EINVALr rB �called)r> rk �mr3 r3 r6 �test_safe_rmpath� s" � � "�z TestFSTestUtils.test_safe_rmpathc C sp | � � }t�� }t�|� t|�� t�� tj�||�ks J �W d � n1 s)w Y t�� |ks6J �d S rF )re rZ �getcwdr� r rh �join)r> rk �baser3 r3 r6 � test_chdir� s �zTestFSTestUtils.test_chdirN)rR rS rT r� r� r� r� r� r3 r3 r3 r6 r{ � s r{ c @ s> e Zd Zdd� Zdd� Zejje dd�dd� �Z d d � Z d S ) �TestProcessUtilsc C sN | � � }t�|j�}|�� sJ �t� |�� rJ �tjjr J �tjjr%J �d S rF ) �spawn_testprocr] �Process�pid� is_runningr �tests� _pids_started�_subprocesses_started)r> �subp�pr3 r3 r6 �test_reap_children� s z#TestProcessUtils.test_reap_childrenc C s� | � � \}}|j|jksJ �|�� sJ �|�� sJ �t�� �� }||gks'J �t�� jdd�}t|�dks7J �||v s=J �||v sCJ �|�� t� � ksMJ �|�� |jksVJ �t |� |�� r`J �|�� sfJ �t |� |�� rpJ �d S )NT)� recursive� ) �spawn_children_pairr� r� r] r� �children�len�ppidrZ r[ r$ )r> �child� grandchildr� r3 r3 r6 �test_spawn_children_pair� s" z)TestProcessUtils.test_spawn_children_pair� POSIX only��reasonc C s" | � � \}}|�� tjksJ �d S rF )� spawn_zombie�statusr] � STATUS_ZOMBIE)r> �_parent�zombier3 r3 r6 �test_spawn_zombie� s z"TestProcessUtils.test_spawn_zombiec C s� | � � }t|� | �|j� t|� t�| � � j�}t|� | �|j� t|� tddg}tj|tj tj t d�}t|� | �|j� t|� | � � j}t|� | �|j� t|� t rw| � � \}}t|� t|� | �|j� | �|j� d S d S )Nz-cz3import time; [time.sleep(0.1) for x in range(100)];)�stdout�stderr�env) r� r$ � assertPidGoner� r] r� r �Popen� subprocess�PIPEr r r� )r> r� �cmdr� �parentr� r3 r3 r6 �test_terminate� s@ �� �z TestProcessUtils.test_terminateN) rR rS rT r� r� r �mark�skipifr r� r� r3 r3 r3 r6 r� � s r� c @ sl e Zd Zdd� Zejje dd�dd� �Zdd� Z ejje dd�ejje p'e d d�d d � ��Z d d � Z dS )� TestNetUtilsc C sN t � }td|fd��}|�� d |ksJ �W d � d S 1 s w Y d S )Nr� ��addrr* )r r � getsockname)r> �port�sr3 r3 r6 r ( s "�zTestNetUtils.bind_socketr� r� c C s� | � � }t|��3}|jtjksJ �|jtjksJ �|�� |ks!J �tj � |�s)J �t � t� |�j �s4J �W d � n1 s>w Y | � � }t|tjd��}|jtjksWJ �W d � d S 1 sbw Y d S )N)�type)re r �family�socket�AF_UNIXr� � SOCK_STREAMr� rZ rh ri �stat�S_ISSOCK�st_mode� SOCK_DGRAM)r> �name�sockr3 r3 r6 �test_bind_unix_socket- s �"�z"TestNetUtils.test_bind_unix_socketc C s� dt � f}ttj|d�\}}|�; |� |�� |ks J �|�� |ks$J �|�� |ks,J �W d � n1 s6w Y W d � d S W d � d S 1 sNw Y d S )Nz 127.0.0.1r� )r r# r� �AF_INETr� � getpeername)r> r� �server�clientr3 r3 r6 �test_tcp_socketpair; s P�z TestNetUtils.test_tcp_socketpairz*/var/run/log UNIX socket opened by defaultc C s� t �� }|�� }t|jdd��rJ �| �� }t|�\}}zFtj� |�s%J �t � t� |�j �s0J �|�� | dks:J �t t|jdd���dksHJ �|�� |ksPJ �|�� |ksXJ �W |�� |�� d S |�� |�� w )N�unix)�kindr� )r] r� �num_fdsr �net_connectionsre r% rZ rh ri r� r� r� r� r� r� r� )r> r� r� r� r� r� r3 r3 r6 �test_unix_socketpairD s( �� � z!TestNetUtils.test_unix_socketpairc C s� t � �c}t�t�}t�t�}|D ]}||j d7 < ||�tjtj� d7 < q|tj dks3J �t � r?|tj dks?J �t rLt rL|tj dksLJ �|tj dksUJ �|tj dks^J �W d � d S 1 siw Y d S )Nr* r� )r � collections� defaultdict�intr� � getsockoptr� � SOL_SOCKET�SO_TYPEr� r �AF_INET6r r r� r� r� )r> �socks�fams�typesr� r3 r3 r6 �test_create_sockets^ s "�z TestNetUtils.test_create_socketsN)rR rS rT r r r� r� r r� r� r r r� r� r3 r3 r3 r6 r� ' s � r� �serial�r� c @ sh e Zd Ze� dd� �Zdd� Ze� ejje dd�ejje dd�dd � ���Z d d � Z d d � Z dd� ZdS )�TestMemLeakClassc s8 � fdd�}ddi� | j |ddd� � d dksJ �d S ) Nc s � d d7 < d S )N�cntr* r3 r3 �r� r3 r6 �funs s z(TestMemLeakClass.test_times.<locals>.funr� r � � )�times� warmup_times� )�execute)r> r� r3 r� r6 � test_timesq s zTestMemLeakClass.test_timesc C s2 t �t�� | jdd� dd� W d � n1 sw Y t �t�� | jdd� dd� W d � n1 s7w Y t �t�� | jdd� dd� W d � n1 sUw Y t �t�� | jd d� dd � W d � n1 ssw Y t �t�� | jd d� dd � W d � d S 1 s�w Y d S ) Nc S ru �Nr r3 r3 r3 r3 r6 rw | rx z1TestMemLeakClass.test_param_err.<locals>.<lambda>r �r� c S ru r� r3 r3 r3 r3 r6 rw ~ rx ���c S ru r� r3 r3 r3 r3 r6 rw � rx )r� c S ru r� r3 r3 r3 r3 r6 rw � rx )� tolerancec S ru r� r3 r3 r3 r3 r6 rw � rx )r, )r rB rI r� ry r3 r3 r6 �test_param_errz s � � � � "�z TestMemLeakClass.test_param_errz skipped on CIr� z skipped during test coveragec C s` g }|fdd�}z%t jtdd�� | j|dd� W d � W ~d S 1 s%w Y W ~d S ~w )Nc S s | � dd � d S )N�xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx� ��append��lsr3 r3 r6 r� � � z+TestMemLeakClass.test_leak_mem.<locals>.funz extra-mem��match�d r� )r rB �AssertionErrorr� )r> r� r� r3 r3 r6 � test_leak_mem� s �� z TestMemLeakClass.test_leak_memc s` � �fdd�}g � t r dnd}tjtd| d�� ��|� W d � d S 1 s)w Y d S )Nc s" t t�} ��| j� � �| � d S rF )rf r~ � addCleanupr� r� )rl ��boxr> r3 r6 r� � s z1TestMemLeakClass.test_unclosed_files.<locals>.fun�fd�handlez unclosed r� )r r rB r r� )r> r� r� r3 r r6 �test_unclosed_files� s "�z$TestMemLeakClass.test_unclosed_filesc s> � fdd�}g � d}| j ||ddd� t� �|d ks J �d S )Nc s � � dd � d S )N�xxxxxxxxxxxxxxxxxxxxxxxxr� r� r3 r� r3 r6 r� � r� z,TestMemLeakClass.test_tolerance.<locals>.funr� r i � )r� r� r� r* )r� r� )r> r� r� r3 r� r6 �test_tolerance� s �z TestMemLeakClass.test_tolerancec C s� dd� }| � t|� t�t�� | � t|� W d � n1 s w Y dd� }t�t�� | � t|� W d � d S 1 s@w Y d S )Nc S rK r0 r3 r3 r3 r3 r6 �fun_1� s z2TestMemLeakClass.test_execute_w_exc.<locals>.fun_1c S s d S rF r3 r3 r3 r3 r6 �fun_2� � z2TestMemLeakClass.test_execute_w_exc.<locals>.fun_2)� execute_w_excrC r rB rq r )r> r r r3 r3 r6 �test_execute_w_exc� s � "�z#TestMemLeakClass.test_execute_w_excN)rR rS rT r r� r� r r� r� r r r r r r r3 r3 r3 r6 r� o s r� c @ sD e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dS )�TestFakePytestc C s( t �� }|�|� t �� }|�|�}|S rF )�unittest� TestSuite�addTest�TextTestRunner�run)r> �klass�suite�runner�resultr3 r3 r6 �run_test_class� s z TestFakePytest.run_test_classc C s� t �t�� }dd W d � n1 sw Y t|jt�s!J �t jtdd��}td��1 s0w Y zt jtdd��}td��1 sEw Y W n tyf } zt|�dks[J �W Y d }~d S d }~ww | �d��)Nr* r r7 r� �barz"foo" does not match "bar"�exception not raised) r rB rC � isinstance�valuerI r �str�fail)r> �cm�errr3 r3 r6 � test_raises� s ��� �� zTestFakePytest.test_raisesc C sV t jjdd�dd� �}|� dksJ �t jjdd�G dd� d��}|� �� dks)J �d S )Nr� r� c S ru rv r3 r3 r3 r3 r6 r7 � rH z%TestFakePytest.test_mark.<locals>.foor* c @ � e Zd Zdd� ZdS )z%TestFakePytest.test_mark.<locals>.Fooc S ru rv r3 ry r3 r3 r6 r � r z)TestFakePytest.test_mark.<locals>.Foo.barN)rR rS rT r r3 r3 r3 r6 �Foo� s r$ )r r� � xdist_groupr )r> r7 r$ r3 r3 r6 � test_mark� s zTestFakePytest.test_markc C s� G dd� dt j�}| �|d��}|�� sJ �t|j�dks J �|jd d dks*J �G dd� dt j�}| �|d��}|�� s@J �t|j�dksIJ �d S )Nc @ �$ e Zd Zejjddd�dd� �ZdS )�,TestFakePytest.test_skipif.<locals>.TestCaseTr� r� c S � ddksJ �d S rv r3 ry r3 r3 r6 r7 � � �0TestFakePytest.test_skipif.<locals>.TestCase.fooN�rR rS rT r r� r� r7 r3 r3 r3 r6 �TestCase� � r- r7 r* r r� c @ r' )r( Fr� r� c S r) rv r3 ry r3 r3 r6 r7 � r* r+ Nr, r3 r3 r3 r6 r- � r. �r r- r � wasSuccessfulr� �skipped�r> r- r r3 r3 r6 � test_skipif� s zTestFakePytest.test_skipifc C sX G dd� dt j�}| �|d��}|�� sJ �t|j�dks J �|jd d dks*J �d S )Nc @ r# )z*TestFakePytest.test_skip.<locals>.TestCasec S s t �d� ddks J �d S )Nr� r* r )r �skipry r3 r3 r6 r7 � s z.TestFakePytest.test_skip.<locals>.TestCase.fooN)rR rS rT r7 r3 r3 r3 r6 r- � � r- r7 r* r r� r/ r2 r3 r3 r6 � test_skip� s zTestFakePytest.test_skipc C s | j td�}t�|� ttj�|d�d�� W d � n1 s w Y ttj�|d�d��}|�t� d�� � � W d � n1 sDw Y t j � tjd|��3 | �td�� t�� }|�� dksfJ �W d � n1 spw Y W d � d S W d � d S 1 s�w Y d S ) N)�dirz __init__.pyrc z test_file.pyz� import unittest class TestCase(unittest.TestCase): def test_passed(self): pass r zFake pytest module was usedr* )re r rZ r� rf rh r� rg �textwrap�dedent�lstripr rU �objectr] r� �assertWarnsRegex� UserWarningr �main�countTestCases)r> �tmpdirrl r r3 r3 r6 � test_main s( � � �� ��"�zTestFakePytest.test_mainc C s6 t �t�� tjdtdd� W d � n1 sw Y z t �t�� tjdtdd� W d � n1 s6w Y W n tyE Y nw | �d��t jtdd�� tjdtdd� W d � n1 sew Y z!t jtdd�� tjdtdd� W d � n1 s�w Y W n ty� Y d S w | �d��)Nr7 r* )� stacklevelr r� r )r �warnsr= �warnings�warn�DeprecationWarningr r ry r3 r3 r6 � test_warns s0 � �� � � �� � zTestFakePytest.test_warnsN) rR rS rT r r" r& r3 r6 rA rG r3 r3 r3 r6 r � s r c @ s e Zd Zdd� Zdd� ZdS )�TestTestingUtilsc C sL t �� }t|�}|�� tdd� |�|j�D ��d }|� |�� ks$J �d S )Nc s � � | ] }|d dkr|V qdS )r* r� Nr3 ��.0�xr3 r3 r6 � <genexpr>5 � � z:TestTestingUtils.test_process_namespace.<locals>.<genexpr>r )r] r� r �test�nextr_ �gettersr� )r> r� �nsr� r3 r3 r6 �test_process_namespace1 s z'TestTestingUtils.test_process_namespacec C s: t � }tdd� |�|j�D ��d }|� t�� ksJ �d S )Nc s rI )r* � net_if_addrsNr3 rJ r3 r3 r6 rM : rN z9TestTestingUtils.test_system_namespace.<locals>.<genexpr>r )r" rP r_ rQ r] rT )r> rR r� r3 r3 r6 �test_system_namespace8 s z&TestTestingUtils.test_system_namespaceN)rR rS rT rS rU r3 r3 r3 r6 rH 0 s rH c @ r# )�TestOtherUtilsc C s. t t�dd�ddd��sJ �t t� �rJ �d S )Nr7 za b cr* r� r9 )r r� � namedtuple�tuplery r3 r3 r6 �test_is_namedtuple? s z!TestOtherUtils.test_is_namedtupleN)rR rS rT rY r3 r3 r3 r6 rV > r5 rV )=�__doc__r� r� rZ r� r� r� r8 r rD r r] � psutil.testsr r r �psutil._commonr r r r r r r r r r r r r r r r r r r r r r r r r r r! r" r# r$ r% r& r' r( rV r{ r� r� r� r% r� r rH rV r3 r3 r3 r6 �<module> sv E(-L HKu tests/__pycache__/__main__.cpython-310.pyc 0000644 00000000513 15030211765 0014336 0 ustar 00 o �h5 � @ s" d Z ddlmZ e�g d�� dS )z>Run unit tests. This is invoked by: $ python -m psutil.tests. � )�pytest)z-vz-sz --tb=shortN)�__doc__� psutil.testsr �main� r r �H/usr/local/CyberCP/lib/python3.10/site-packages/psutil/tests/__main__.py�<module> s tests/__pycache__/test_windows.cpython-310.pyc 0000644 00000073006 15030211765 0015356 0 ustar 00 o �h�� � @ s~ d Z ddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl Z ddl Z ddl m Z ddlZddlmZ ddlmZ ddlmZ ddlmZ ddlmZ dd lmZ dd lmZ dd lmZ dd lmZ dd lmZ ddlmZ ddlmZ ddlm Z er�es�e � � � e � d� ddl Z ddl Z ddl!Z!ddl"Z"W d � n1 s�w Y er�ddl#m$Z$ ej%j&Z&ej'j(e dd�ej'j(edd�G dd� de���Z)dd� Z*e+fdd�Z,G d d � d e)�Z-G d d � d e)�Z.G d d!� d!e)�Z/G d"d#� d#e)�Z0G d$d%� d%e)�Z1ej'j(e dd�G d&d'� d'e��Z2ej'j(e dd�G d(d)� d)e��Z3ej'j(e dd�G d*d+� d+e��Z4dS ),zWindows specific tests.� N)�mock)�WINDOWS)�GITHUB_ACTIONS)� HAS_BATTERY)�IS_64BIT)�PYPY)�TOLERANCE_DISK_USAGE)�TOLERANCE_SYS_MEM)�PsutilTestCase)�pytest)�retry_on_failure)�sh)�spawn_testproc)� terminate�ignore)�convert_oserrorz WINDOWS only��reasonz pywin32 not available on PYPYc @ s e Zd ZdS )�WindowsTestCaseN)�__name__� __module__� __qualname__� r r �L/usr/local/CyberCP/lib/python3.10/site-packages/psutil/tests/test_windows.pyr 5 s r c C s( t �d�s t�d��d| � d�}t|�S )z�Currently not used, but available just in case. Usage: >>> powershell( "Get-CIMInstance Win32_PageFileUsage | Select AllocatedBaseSize") zpowershell.exez powershell.exe not availablezhpowershell.exe -ExecutionPolicy Bypass -NoLogo -NonInteractive -NoProfile -WindowStyle Hidden -Command "�")�shutil�whichr �skipr )�cmd�cmdliner r r � powershell; s ��r c sh t d| � d|� ���� }d�|�� dd� ��� }� dur2d|v r.t� fdd�|�� D ��S � |�S |S ) z�Currently not used, but available just in case. Usage: >>> wmic("Win32_OperatingSystem", "FreePhysicalMemory") 2134124534 z wmic path z get � � N�,c 3 s � | ]}� |�V qd S �Nr ��.0�x�� converterr r � <genexpr>T s � zwmic.<locals>.<genexpr>)r �strip�join� splitlines�tuple�split)�path�whatr) �out�datar r( r �wmicJ s r4 c @ sR e Zd Zejjdejvdd�dd� �Zdd� Z dd � Z d d � Z d d � Z dd� Z dS )� TestCpuAPIs�NUMBER_OF_PROCESSORSz-NUMBER_OF_PROCESSORS env var is not availabler c C s" t tjd �}|t�� ksJ �d S )Nr6 )�int�os�environ�psutil� cpu_count)�self�num_cpusr r r �&test_cpu_count_vs_NUMBER_OF_PROCESSORSa s z2TestCpuAPIs.test_cpu_count_vs_NUMBER_OF_PROCESSORSc C s$ t �� d }t�� }||ksJ �d S )N� )�win32api� GetSystemInfor: r; �r< � sys_value� psutil_valuer r r � test_cpu_count_vs_GetSystemInfok s z+TestCpuAPIs.test_cpu_count_vs_GetSystemInfoc C s2 t �� }tdd� |�� D ��}t�� |ksJ �d S )Nc s � � | ]}|j V qd S r$ )�NumberOfLogicalProcessors�r& �procr r r r* t s � �z<TestCpuAPIs.test_cpu_count_logical_vs_wmi.<locals>.<genexpr>��wmi�WMI�sum�Win32_Processorr: r; )r< �w�procsr r r � test_cpu_count_logical_vs_wmir s �z)TestCpuAPIs.test_cpu_count_logical_vs_wmic C s6 t �� }tdd� |�� D ��}tjdd�|ksJ �d S )Nc s rF r$ )� NumberOfCoresrH r r r r* { s � z:TestCpuAPIs.test_cpu_count_cores_vs_wmi.<locals>.<genexpr>F)�logicalrJ )r< rO �coresr r r �test_cpu_count_cores_vs_wmiy s z'TestCpuAPIs.test_cpu_count_cores_vs_wmic C s t �� tt jdd��ksJ �d S )NT)�percpu)r: r; �len� cpu_times�r< r r r �test_cpu_count_vs_cpu_times~ s z'TestCpuAPIs.test_cpu_count_vs_cpu_timesc C s@ t �� }|�� d }|jt�� jksJ �|jt�� jks J �d S �Nr ) rK rL rN �CurrentClockSpeedr: �cpu_freq�current� MaxClockSpeed�max)r< rO rI r r r � test_cpu_freq� s zTestCpuAPIs.test_cpu_freqN)r r r r �mark�skipifr8 r9 r> rE rQ rU rZ ra r r r r r5 ` s � r5 c @ s~ e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Ze� d d � �Z e� d d� �Z e� dd� �Z dd� Z dd� Z dd� Zdd� ZdS )�TestSystemAPIsc C sT t d�}tjdd��� }|D ]}d|�dd��� v rq||vr'| �|�d���qd S )Nz ipconfig /allT)�perniczpseudo-interface� �-z+ nic wasn't found in 'ipconfig /all' output)r r: �net_io_counters�keys�replace�lower�fail)r< r2 �nics�nicr r r �test_nic_names� s ���z TestSystemAPIs.test_nic_namesc C s, t �� �� d }t|j�t�� jksJ �d S r[ )rK rL �Win32_ComputerSystemr7 �TotalPhysicalMemoryr: �virtual_memory�total�r< rO r r r �test_total_phymem� s z TestSystemAPIs.test_total_phymemc C s4 t �� �� d }tt|j�t�� j �t k sJ �d S r[ ) rK rL � Win32_PerfRawData_PerfOS_Memory�absr7 �AvailableBytesr: rr �freer rt r r r �test_free_phymem� s �z TestSystemAPIs.test_free_phymemc C sl t �� �� d }t|j�t�� j t�� jksJ �t�� jdkr2t�� j dks)J �t�� j dks4J �d S d S r[ ) rK rL rv r7 � CommitLimitr: rr rs � swap_memoryry �usedrt r r r �test_total_swapmem� s ��z!TestSystemAPIs.test_total_swapmemc C s� t �� jdkr<t�� jdd�d }t|j�d t|j� }t �� j dks&J �t t �� j | �dk s3J �t �� j dks>J �d S d S )Nr �_Total)�Name�d r? ) r: r| rs rK rL �#Win32_PerfRawData_PerfOS_PagingFiler7 � PercentUsage�PercentUsage_Base�percentrw )r< rO � percentSwapr r r �test_percent_swapmem� s �z#TestSystemAPIs.test_percent_swapmemc C s6 t �� �� }dd� |D �}tt�� �}||ksJ �d S )Nc S s h | ]}|j �qS r �� ProcessIdr% r r r � <setcomp>� s z+TestSystemAPIs.test_pids.<locals>.<setcomp>)rK rL � Win32_Process�setr: �pids)r< rO �wmi_pids� psutil_pidsr r r � test_pids� s zTestSystemAPIs.test_pidsc C s� t jdd�}t�� �� }|D ]m}|D ]`}|j�dd�|jkrr|js# nXd|j v r* nQ|j� d�r2 nIzt � |j�}W n t yD Y n7w |j t|j�ksOJ �t|j�}|j|ks[J �t|j| �dkrp| �d|j� d |� ��� n q| �d |����qd S ) NT��all�\r! �cdrom�A:i � zpsutil=z, wmi=zcan't find partition )r: �disk_partitionsrK rL �Win32_LogicalDisk�devicerj �DeviceID� mountpoint�opts� startswith� disk_usage�FileNotFoundErrorrs r7 �Size� FreeSpacery rw rl )r< �ps_parts� wmi_parts�ps_part�wmi_part�usage�wmi_freer r r � test_disks� s4 � ���zTestSystemAPIs.test_disksc C s� t �� D ]9}d|jv r qt�|j�}t �|j�}t|d |j �t k s%J �t|d |j �t k s2J �|j |j |j ks=J �qd S )Nr� r r" ) r: r� r� r@ �GetDiskFreeSpaceExr� r� rw ry r rs r} )r< �diskrC rD r r r �test_disk_usage� s �z TestSystemAPIs.test_disk_usagec C s> dd� t �� �d�D �}dd� tjdd�D �}||ks J �d S )Nc S s" g | ] }|r|� d �s|d �qS )r� r� )r� r% r r r � <listcomp>� s ���z7TestSystemAPIs.test_disk_partitions.<locals>.<listcomp>z\ c S s g | ] }|j �d �s|j �qS )r� )r� r� r% r r r r� � s ��Tr� )r@ �GetLogicalDriveStringsr/ r: r� rB r r r �test_disk_partitions� s � �z#TestSystemAPIs.test_disk_partitionsc C s` t t�� �}t�� �� }t � }|D ]}|�|j� |�|j� q||@ s.J d|� d|� ���d S )Nzno common entries in z, ) r� �cext� net_if_statsrK rL �Win32_NetworkAdapter�addr� �NetConnectionID)r< �ps_names� wmi_adapters� wmi_names� wmi_adapterr r r �test_net_if_stats s ��z TestSystemAPIs.test_net_if_statsc C s^ t �� �� }|d j�d�d }tj�|d�}tj�t� � �}t || � � �}|dks-J �d S )Nr �.� %Y%m%d%H%M%Sr? ) rK rL �Win32_OperatingSystem�LastBootUpTimer/ �datetime�strptime� fromtimestampr: � boot_timerw � total_seconds)r< �wmi_os� wmi_btime_str� wmi_btime_dt� psutil_dt�diffr r r �test_boot_time s �z TestSystemAPIs.test_boot_timec C s� t jddd�� t�� dksJ �W d � n1 sw Y t jddd�� t�� dks/J �W d � n1 s9w Y t jddd�� t�� dksNJ �W d � n1 sXw Y t jddd�� t�� dksmJ �W d � d S 1 sxw Y d S )Nz psutil._pswindows.cext.boot_timer? �� return_value� � iM )r �patchr: r� rY r r r �test_boot_time_fluctuation s � � �"�z)TestSystemAPIs.test_boot_time_fluctuationN)r r r ro ru rz r~ r� r r� r� r� r� r� r� r� r r r r rd � s rd c @ sh e Zd Zdd� Zejje dd�dd� �Zejje dd�dd� �Z d d � Z d d � Z d d� Z dd� Z dS )�TestSensorsBatteryc C s4 t �� d rt�� d usJ �d S t�� d u sJ �d S )N�SystemBatteriesPresent)r@ �GetPwrCapabilitiesr: �sensors_batteryrY r r r �test_has_battery' s z#TestSensorsBattery.test_has_batteryz no batteryr c C s: t �� }|�d�d }t�� }t|j|j �dk sJ �d S )N�select * from Win32_Batteryr r" )rK rL �queryr: r� rw r� �EstimatedChargeRemaining�r< rO � battery_wmi�battery_psutilr r r � test_percent- s �z TestSensorsBattery.test_percentc C s6 t �� }|�d�d }t�� }|j|jdkksJ �d S )Nr� r � )rK rL r� r: r� � power_plugged� BatteryStatusr� r r r �test_power_plugged7 s z%TestSensorsBattery.test_power_pluggedc C sN t jddd��}t�� d u sJ �|jsJ �W d � d S 1 s w Y d S )N�&psutil._pswindows.cext.sensors_battery)r � r r r� )r r� r: r� �called�r< �mr r r �test_emulate_no_battery@ s � "�z*TestSensorsBattery.test_emulate_no_batteryc C �R t jddd��}t�� jtjksJ �|jsJ �W d � d S 1 s"w Y d S )Nr� )r" r r r r� �r r� r: r� �secsleft�POWER_TIME_UNLIMITEDr� r� r r r � test_emulate_power_connectedH � �� "�z/TestSensorsBattery.test_emulate_power_connectedc C r� )Nr� )r � r r r� r� r� r r r �test_emulate_power_chargingR r� z.TestSensorsBattery.test_emulate_power_chargingc C r� )Nr� )r r r ���r� )r r� r: r� r� �POWER_TIME_UNKNOWNr� r� r r r � test_emulate_secs_left_unknown\ s � "�z1TestSensorsBattery.test_emulate_secs_left_unknownN)r r r r� r rb rc r r� r� r� r� r� r� r r r r r� & s r� c @ s� e Zd Zedd� �Zedd� �Zdd� Zdd� Zd d � Zd d � Z d d� Z dd� Z dd� Z dd� Z dd� Zdd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#S )$� TestProcessc C � t � j| _d S r$ �r �pid��clsr r r � setUpClassm � zTestProcess.setUpClassc C � t | j� d S r$ �r r� r� r r r � tearDownClassq r� zTestProcess.tearDownClassc C sD t �d�}t�t j�� |�� W d � d S 1 sw Y d S r[ )r: �Processr �raises� AccessDenied�kill�r< �pr r r � test_issue_24u s "�zTestProcess.test_issue_24c C s� t �d�}|�� dks J �t|� |�� |�� dks J �z |�� d d� \}}W n t jy= t� � d dvr:� Y d S w |dksDJ �d S )Nr� �Systemg r� r" > �win-7�win7�vistar ) r: r� �name�str�username� create_time� memory_infor� �platform�uname)r< r� �rss�_vmsr r r �test_special_pidz s ��z TestProcess.test_special_pidc C sH t �| j�}t�t�� |�tj� W d � d S 1 s w Y d S r$ ) r: r� r� r r� � ValueError� send_signal�signal�SIGINTr� r r r �test_send_signal� s "�z TestProcess.test_send_signalc C sb t �t�� �}|�� }t�tjtj t�� �}|�� }||d ks"J �t� |� |�� |ks/J �d S )Nr" ) r: r� r8 �getpid� num_handlesr@ � OpenProcess�win32con�PROCESS_QUERY_INFORMATION�FALSE� CloseHandle)r< r� �before�handle�afterr r r �test_num_handles_increment� s � z&TestProcess.test_num_handles_incrementc C s� t �| �� j�}|�tj� |�tj� |�� |� � t � t j �� |�tj� W d � n1 s3w Y t � t j �� |�tj� W d � d S 1 sPw Y d S r$ ) r: r� r r� r r � CTRL_C_EVENT�CTRL_BREAK_EVENTr� �waitr r� � NoSuchProcessr� r r r �test_ctrl_signals� s �"�z TestProcess.test_ctrl_signalsc C s8 t �tj�}|�d�rt�d��t�� � � |ksJ �d S )N�$zrunning as service account) r@ � GetUserNameExr �NameSamCompatible�endswithr r r: r� r )r< r r r r � test_username� s zTestProcess.test_usernamec C sf t �ddt�� ��� }d�t�� �� �}|d d kr"|d kr+n n|� ddd�}||ks1J �d S )Nz[ ]+rf r r r! r� ) �re�subr@ �GetCommandLiner+ r, r: r� r rj rB r r r � test_cmdline� s zTestProcess.test_cmdlinec C sJ t �tjtjt�� �}| �t j|� t � |�}t � � � � }||ks#J �d S r$ )r@ r r r r r8 r � addCleanupr � win32process�GetPriorityClassr: r� �nice�r< r rC rD r r r � test_nice� s � zTestProcess.test_nicec C s� t �tjtj| j�}| �t j|� t� |�}t � | j�� � }|d |j ks'J �|d |jks0J �|d |jks9J �|d |jksBJ �|d |jksKJ �|d |jksTJ �|d |jks]J �|d |jksfJ �|j|jksnJ �|j|jksvJ �d S ) N�PeakWorkingSetSize�WorkingSetSize�QuotaPeakPagedPoolUsage�QuotaPagedPoolUsage�QuotaPeakNonPagedPoolUsage�QuotaNonPagedPoolUsage� PagefileUsage�PeakPagefileUsage)r@ r r r r r� r, r r- �GetProcessMemoryInfor: r� r � peak_wset�wset�peak_paged_pool� paged_pool�peak_nonpaged_pool� nonpaged_pool�pagefile� peak_pagefiler �vmsr0 r r r �test_memory_info� s( � ��z TestProcess.test_memory_infoc C sX t �tjtj| j�}| �t j|� t� | j�}|� � |� � }t � |�}||ks*J �d S r$ )r@ r r r r r� r, r r: r� r r r- �GetExitCodeProcess)r< r r� rD rC r r r � test_wait� s � zTestProcess.test_waitc C s\ dd� }t �tjtj| j�}| �t j|� |t� |�d �}t � | j�� � }||ks,J �d S )Nc s � fdd�t d�D �S )Nc s g | ] }d |> � @ r|�qS )r" r )r& �i�r' r r r� � s zGTestProcess.test_cpu_affinity.<locals>.from_bitmask.<locals>.<listcomp>�@ )�rangerH r rH r � from_bitmask� s z3TestProcess.test_cpu_affinity.<locals>.from_bitmaskr ) r@ r r r r r� r, r r- �GetProcessAffinityMaskr: r� � cpu_affinity)r< rK r rC rD r r r �test_cpu_affinity� s � �z TestProcess.test_cpu_affinityc C s� t �tjtjt�� �}| �t j|� t � |�}t � � � � }|j|d ks&J �|j|d ks/J �|j|d ks8J �|j|d ksAJ �|j|d ksJJ �|j|d ksSJ �d S )N�ReadOperationCount�WriteOperationCount�ReadTransferCount�WriteTransferCount�OtherOperationCount�OtherTransferCount)r@ r r r r r8 r r, r r- �GetProcessIoCountersr: r� � io_counters� read_count� write_count� read_bytes� write_bytes� other_count� other_bytesr0 r r r �test_io_counters s � z TestProcess.test_io_countersc C s� dd l }dd l}d}|jj�|d| j�}| �|jjj|� |j� � }|jj� ||� |�� |j }t �| j��� }||ks>J �d S )Nr � )�ctypes�ctypes.wintypes�windll�kernel32r r� r, r �wintypes�DWORD�GetProcessHandleCount�byref�valuer: r� r )r< r_ r r �hndcntrC rD r r r �test_num_handles s � �z TestProcess.test_num_handlesc C s� t � }d|_tjd|d��; t�d��&}t�� }t�tj�� |� � W d � n1 s-w Y W d � n1 s<w Y W d � n1 sKw Y |j dksWJ �d S )Ni+ z psutil._psplatform.cext.proc_cwd�� side_effectz time.sleepr? ) �OSError�winerrorr r� r: r� r r� r� �cwd� call_count)r< �excr� r� r r r �test_error_partial_copy& s �� �� �z#TestProcess.test_error_partial_copyc C sV t �� d d }t j�|�}t�t j�� |�� W d � d S 1 s$w Y d S )Nr� i�� )r: r� � _psplatformr� r r� r! �exe)r< r� rI r r r �test_exe1 s "�zTestProcess.test_exeN)r r r � classmethodr� r� r� r r r r" r' r+ r1 rD rF rN r] ri rq rt r r r r r� l s( r� c @ s| e Zd ZdZedd� �Zedd� �Zdd� Zej j e dd �d d � �Z d d � Z dd� Ze� dd� �Ze� dd� �Zdd� ZdS )�TestProcessWMIz%Compare Process API results with WMI.c C r� r$ r� r� r r r r� = r� zTestProcessWMI.setUpClassc C r� r$ r� r� r r r r� A r� z TestProcessWMI.tearDownClassc C s8 t �� j| jd�d }t�| j�}|�� |jksJ �d S �Nr� r )rK rL r� r� r: r� r �Caption�r< rO r� r r r � test_nameE s zTestProcessWMI.test_namez!unreliable path on GITHUB_ACTIONSr c C s@ t �� j| jd�d }t�| j�}|�� �� |j�� ks J �d S rw ) rK rL r� r� r: r� rs rk �ExecutablePathry r r r rt K s zTestProcessWMI.test_exec C sF t �� j| jd�d }t�| j�}d�|�� �|j� dd�ks!J �d S )Nr� r rf r r! ) rK rL r� r� r: r� r, r � CommandLinerj ry r r r r+ U s $zTestProcessWMI.test_cmdlinec C sR t �� j| jd�d }t�| j�}|�� \}}}|� d|� �}|�� |ks'J �d S )Nr� r r� )rK rL r� r� r: r� �GetOwnerr )r< rO r� �domain�_r r r r r' Z s z TestProcessWMI.test_usernamec C sB t �� j| jd�d }t�| j�}|�� j}|t|j �ks J �d S rw ) rK rL r� r� r: r� r r r7 r3 )r< rO r� r r r r �test_memory_rssa s z TestProcessWMI.test_memory_rssc C s` t �� j| jd�d }t�| j�}|�� j}t|j �}|||d hvr.| � d|� d|� ���d S )Nr� r r^ zwmi=z , psutil=) rK rL r� r� r: r� r rC r7 � PageFileUsagerl )r< rO r� rC � wmi_usager r r �test_memory_vmsh s �z TestProcessWMI.test_memory_vmsc C s\ t �� j| jd�d }t�| j�}t|j�d�d �}t � dt � |� � ��}||ks,J �d S )Nr� r r� r� ) rK rL r� r� r: r� r � CreationDater/ �time�strftime� localtimer )r< rO r� � wmic_create� psutil_creater r r �test_create_timeu s �z TestProcessWMI.test_create_timeN)r r r �__doc__ru r� r� rz r rb rc r rt r+ r' r r� r� r� r r r r rv : s$ � rv c @ sX e Zd ZdZedd� �Zedd� �Zdd� Zdd � Zd d � Z d d � Z dd� Z dd� Z dS )� TestDualProcessImplementationaw Certain APIs on Windows have 2 internal implementations, one based on documented Windows APIs, another one based NtQuerySystemInformation() which gets called as fallback in case the first fails because of limited permission error. Here we test that the two methods return the exact same value, see: https://github.com/giampaolo/psutil/issues/304. c C r� r$ r� r� r r r r� � r� z(TestDualProcessImplementation.setUpClassc C r� r$ r� r� r r r r� � r� z+TestDualProcessImplementation.tearDownClassc C s� t �| j��� }tjdtd��G}t �| j��� }t|�t|�ks"J �tt|��D ] }|| dks2J �|| dks:J �t || || �dk sHJ �q(|j sNJ �W d � d S 1 sYw Y d S )Nz(psutil._psplatform.cext.proc_memory_inforj r i ) r: r� r� r r r� �PermissionErrorrW rJ rw r� )r< �mem_1�fun�mem_2rG r r r rD � s � "�z.TestDualProcessImplementation.test_memory_infoc C �f t �| j��� }tjdtd��}t �| j��� |ks J �|js!J �W d � d S 1 s,w Y d S )N�"psutil._psplatform.cext.proc_timesrj )r: r� r� r r r� r� r� )r< �ctimer� r r r r� � � � "�z.TestDualProcessImplementation.test_create_timec C s� t �| j��� }tjdtd��.}t �| j��� }|js J �t|j |j �dk s)J �t|j |j �dk s5J �W d � d S 1 s@w Y d S )Nr� rj g{�G�z�?) r: r� r� rX r r� r� r� rw �user�system)r< � cpu_times_1r� � cpu_times_2r r r �test_cpu_times� s � "�z,TestDualProcessImplementation.test_cpu_timesc C s� t �| j��� }tjdtd��-}t �| j��� }tt|��D ]}t || || �dk s.J �q |j s4J �W d � d S 1 s?w Y d S )Nz(psutil._psplatform.cext.proc_io_countersrj r? ) r: r� r� rV r r� r� rJ rW rw r� )r< � io_counters_1r� � io_counters_2rG r r r r] � s � "�z.TestDualProcessImplementation.test_io_countersc C r� )Nz(psutil._psplatform.cext.proc_num_handlesrj )r: r� r� r r r� r� r� )r< r r� r r r ri � r� z.TestDualProcessImplementation.test_num_handlesc C s� t �� D ]9}ztj|dd�}tj|dd�}W n ty6 } zt|�}t|t jt jf�s,� W Y d }~qd }~ww ||ks=J �qd S )NT)�use_pebF) r: r� r� � proc_cmdlinerl r � isinstancer� r! )r< r� �a�b�errr r r r+ � s � ����z*TestDualProcessImplementation.test_cmdlineN) r r r r� ru r� r� rD r� r� r] ri r+ r r r r r� � s r� c sp e Zd ZdZedd� �ZddgZ� fdd�Z� fdd �Zd d � Z d d � Z dd� Z dd� Z dd� Z dd� Z� ZS )�RemoteProcessTestCasez�Certain functions require calling ReadProcessMemory. This trivially works when called on the current process. Check that this works on other processes, especially when they have a different bitness. c C s\ d} t � d�D ]$}tj|d| gtjtjd�}|�� \}}|�� |tt �kr+| S qd S )Nz6import sys; sys.stdout.write(str(sys.maxsize > 2**32))zC:\Python*\python.exe�-c)�args�stdout�stderr) �glob� subprocess�Popen�PIPE�STDOUT� communicater r r )�code�filenamerI �outputr r r r �find_other_interpreter� s � ��z,RemoteProcessTestCase.find_other_interpreterr� z import sys; sys.stdin.read()c s� t � �� | �� }|d u rt�d��tr tj| _|| _ n|| _tj| _ t j � � }t t �� �|d< | j| j g| j |tjd�| _| j| jg| j |tjd�| _d S )Nz0could not find interpreter with opposite bitness�THINK_OF_A_NUMBER)�env�stdin)�super�setUpr� r r r �sys� executable�python64�python32r8 r9 �copyr r r � test_argsr� r� �proc32�proc64)r< � other_pythonr� �� __class__r r r� � s& � � �zRemoteProcessTestCase.setUpc s" t � �� | j�� | j�� d S r$ )r� �tearDownr� r� r� rY r� r r r� s z RemoteProcessTestCase.tearDownc C �@ t �| jj�}t|�� �dksJ �|�� dd � | jks J �d S �N� r" )r: r� r� r� rW r r� r� r r r �test_cmdline_32 � z%RemoteProcessTestCase.test_cmdline_32c C r� r� )r: r� r� r� rW r r� r� r r r �test_cmdline_64 r� z%RemoteProcessTestCase.test_cmdline_64c C �& t �| jj�}|�� t�� ksJ �d S r$ )r: r� r� r� rn r8 �getcwdr� r r r � test_cwd_32 � z!RemoteProcessTestCase.test_cwd_32c C r� r$ )r: r� r� r� rn r8 r� r� r r r � test_cwd_64! r� z!RemoteProcessTestCase.test_cwd_64c C s> t �| jj�}|�� }d|v sJ �|d tt�� �ks J �d S )Nr� )r: r� r� r� r9 r r8 r )r< r� �er r r �test_environ_32% s z%RemoteProcessTestCase.test_environ_32c C s4 t �| jj�}z|�� W d S t jy Y d S w r$ )r: r� r� r� r9 r� r� r r r �test_environ_64+ s �z%RemoteProcessTestCase.test_environ_64)r r r r� � staticmethodr� r� r� r� r� r� r� r� r� r� � __classcell__r r r� r r� � s r� c @ s e Zd Zdd� Zdd� ZdS )� TestServicesc C sJ h d�}h d�}h d�}t �� D ]�}|�� }t|d t�s J �|d �� s'J �t|d t�s0J �t|d t�s9J �|d |v sAJ �|d d urNt �|d � t|d t�sWJ �t|d t�s`J �t|d t�siJ �|d |v sqJ �|d |v syJ �t|d t�s�J �|�� }|d ur�t �|�}|�� s�J �t � |� � �}||ks�J �qd S ) N> �stop�pause�start�paused�running�stopped�continue> �manual�disabled� automatic> r� r� r� � stop_pending� pause_pending� start_pending�continue_pendingr � display_namer �statusr� �binpath� start_type� description) r: �win_service_iter�as_dictr� r r+ r� r� � is_running�win_service_getr )r< �valid_statuses�valid_start_types�servr3 r� r� �sr r r �test_win_service_iter: s2 �z"TestServices.test_win_service_iterc C s� t jjj}t jjj}tt �� ��� }t� t j ��}t � |d � W d � n1 s*w Y |j j|d ks9J �t � |�}t ddd�}||_tjd|d��" t� t j �� |�� W d � n1 sdw Y W d � n1 ssw Y tjd|d��" t� t j �� |�� W d � n1 s�w Y W d � n1 s�w Y t ddd�}||_tjd|d��" t� t j�� |�� W d � n1 s�w Y W d � n1 s�w Y tjd|d��# t� t j�� |�� W d � n 1 �sw Y W d � n 1 �sw Y |�� t|�v �s!J �|�� t|�v �s,J �|�� t|�v �s7J �|�� t|�v �sBJ �d S )Nz???r �msgz/psutil._psplatform.cext.winservice_query_statusrj z/psutil._psplatform.cext.winservice_query_config)r: rr r� � ERROR_SERVICE_DOES_NOT_EXIST�ERROR_ACCESS_DENIED�nextr� r r r� r! r� rg rl rm r r� r� r r� r r� �repr)r< r� r� r �cm�servicerp r r r �test_win_service_getf sb � � � �� �� �� � � �� �� �� �z!TestServices.test_win_service_getN)r r r r� r� r r r r r� 8 s ,r� )5r� r� r� r8 r r( r r r� r� r� �warnings�unittestr r: r � psutil.testsr r r r r r r r r r r r �catch_warnings� simplefilterr@ r r- rK �psutil._pswindowsr rr r� rb rc r r r7 r4 r5 rd r� r� rv r� r� r� r r r r �<module> sp � ( F OHW] tests/__pycache__/test_process.cpython-310.pyc 0000644 00000141306 15030211765 0015341 0 ustar 00 o �h�� � @ s� d Z ddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl Z ddl Z ddl Z ddl Z ddlZddlZddlmZ ddlZddlmZ ddlmZ ddlmZ ddlmZ ddlmZ dd lmZ dd lmZ dd lmZ dd lmZ dd l 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 m(Z( ddl m)Z) ddl m*Z* ddl m+Z+ ddl m,Z, dd l m-Z- dd l m.Z. dd l m/Z/ dd l m0Z0 dd l m1Z1 dd!l m2Z2 dd"l m3Z3 dd#l m4Z4 dd$l m5Z5 dd%l m6Z6 dd&l m7Z7 dd'l m8Z8 dd(l m9Z9 dd)l m:Z: dd*l m;Z; G d+d,� d,e.�Z<G d-d.� d.e.�Z=dS )/� Tests for psutil.Process class.� N)�mock)�AIX)�BSD)�LINUX)�MACOS)�NETBSD)�OPENBSD)�OSX)�POSIX)�WINDOWS)� open_text)� CI_TESTING)�GITHUB_ACTIONS)�GLOBAL_TIMEOUT)�HAS_CPU_AFFINITY)� HAS_ENVIRON)� HAS_IONICE)�HAS_MEMORY_MAPS)�HAS_PROC_CPU_NUM)�HAS_PROC_IO_COUNTERS)� HAS_RLIMIT)� HAS_THREADS)� MACOS_11PLUS)�PYPY)� PYTHON_EXE)�PYTHON_EXE_ENV)�PsutilTestCase)� ThreadTask)� call_until)�copyload_shared_lib)� create_c_exe)� create_py_exe)�process_namespace)�pytest�� reap_children)�retry_on_failure)�sh)�skip_on_access_denied)�skip_on_not_implemented)� wait_for_pidc @ sJ e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Ze j j e d d �dd� �Z dd� Ze j j edd �dd� �Zdd� Zdd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Ze j j e d#d �d$d%� �Zd&d'� Ze j j e d(d �d)d*� �Ze j j e d#d �e e d+�d,d-� ��Z e j j e d#d �e j j e d.d �d/d0� ��Z!e j j e d#d �e j j e" d1d �d2d3� ��Z#e j j e$ d#d �d4d5� �Z%e j j e$ d#d �d6d7� �Z&e j j e$ d#d �d8d9� �Z'e j j e$ d#d �d:d;� �Z(e j j e$ d#d �d<d=� �Z)d>d?� Z*e j j e" d@d �dAdB� �Z+e j j e, d#d �dCdD� �Z-e.� e/e0d+�e j j e, d#d �dEdF� ���Z1e.� dGdH� �Z2dIdJ� Z3e j j e4 d#d �dKdL� �Z5e j j e4 d#d �dMdN� �Z6dOdP� Z7dQdR� Z8dSdT� Z9dUdV� Z:e j j e;dWd �dXdY� �Z<dZd[� Z=e j j e;d\d �d]d^� �Z>e j j e d(d �d_d`� �Z?e j j e d(d �dadb� �Z@dcdd� ZAdedf� ZBdgdh� ZCdidj� ZDdkdl� ZEe j j eF d#d �dmdn� �ZGe j j eF d#d �dodp� �ZHe j j eF d#d �dqdr� �ZIe j j eJdsd �dtdu� �ZKe j j eJdsd �dvdw� �ZLe j j e d(d �dxdy� �ZMe e d+�e j j eN�p�edzd �d{d|� ��ZOd}d~� ZPdd�� ZQd�d�� ZRe.� d�d�� �ZSd�d�� ZTd�d�� ZUd�d�� ZVd�d�� ZWd�d�� ZXd�d�� ZYd�d�� ZZd�d�� Z[d�d�� Z\d�d�� Z]d�d�� Z^e j j e d(d �d�d�� �Z_e j j e d(d �d�d�� �Z`e j j e d(d �d�d�� �Zad�d�� Zbd�d�� Zce j j ed d#d �d�d�� �Zee j j ed d#d �e j j e d(d �e j j efd�d �e j j ed�d �d�d�� ����Zgd�S )�� TestProcessr c O s> | j |i |��}zt�|j�W S tjy | �|j� � w �N)�spawn_testproc�psutil�Process�pid� NoSuchProcess� assertPidGone)�self�args�kwargs�sproc� r8 �L/usr/local/CyberCP/lib/python3.10/site-packages/psutil/tests/test_process.py� spawn_psprocL s �zTestProcess.spawn_psprocc C sP t �� }|jt�� ks J �t�t�� d|_W d � d S 1 s!w Y d S )N�! )r/ r0 r1 �os�getpidr$ �raises�AttributeError�r4 �pr8 r8 r9 �test_pidV s "�zTestProcess.test_pidc C sJ | � � }|�� |�� }tr|tjksJ �n|tj ks J �| �|� d S r- )r: �kill�waitr �signal�SIGTERM�SIGKILL�assertProcessGone�r4 rA �coder8 r8 r9 � test_kill\ � zTestProcess.test_killc C sJ | � � }|�� |�� }tr|tjksJ �n|tj ks J �| �|� d S r- )r: � terminaterD r rE rF rH rI r8 r8 r9 �test_terminatef rL zTestProcess.test_terminatec C sX t rtjntj}| �� }|�|� |�� }tr ||ks J �n|| ks%J �| �|� d S r- ) r rE rG rF r: � send_signalrD r rH )r4 �sigrA rJ r8 r8 r9 �test_send_signalp s z TestProcess.test_send_signalz not POSIX)�reasonc C s� t j}| �� }tjdtd��# t�tj �� |� |� W d � n1 s%w Y W d � n1 s4w Y | �� }tjdt d��, t�tj �� |� |� W d � n1 s[w Y W d � d S W d � d S 1 ssw Y d S )Nzpsutil.os.kill�� side_effect) rE rF r: r �patch�ProcessLookupErrorr$ r> r/ r2 rO �PermissionError� AccessDenied)r4 rP rA r8 r8 r9 �test_send_signal_mocked{ s �� � ��"�z#TestProcess.test_send_signal_mockedc C s� t ddg}| �|�}|�� }|dksJ �| �|� t ddg}| j|tjd�}|�� }|dks0J �| �|� t ddg}| �|�}|�� }|dksIJ �| �|� t dd g}| �|�}|�� }|dksbJ �| �|� d S ) N�-c�passr z1 / 0)�stderr� zimport sys; sys.exit(5);� zimport os; os._exit(5);)r r: rD rH � subprocess�PIPE)r4 �cmdrA rJ r8 r8 r9 �test_wait_exited� s( z TestProcess.test_wait_exitedzfails on NETBSDc C s| | � � }trf|�tj� t�tj�� |j dd� W d � n1 s#w Y |�tj � t�tj�� |j dd� W d � n1 sEw Y |�tj � |� � tj ksZJ �|� � tj ksdJ �d S |� � t�tj�� |j dd� W d � n1 s�w Y |� � t�tj�� |j dd� W d � n1 s�w Y |�� |� � tj ks�J �|� � tj ks�J �d S )N���MbP?)�timeout)r: r rO rE �SIGSTOPr$ r> r/ �TimeoutExpiredrD �SIGCONTrF �suspend�resumerM r@ r8 r8 r9 �test_wait_stopped� s0 � � � �z TestProcess.test_wait_stoppedc C s� | � � \}}t�tj�� |�d� W d � n1 s w Y t�tj�� |�d� W d � n1 s7w Y |�� |�� |�� }|�� }tr^|tj ksVJ �|d u s\J �d S |tj kseJ �|tj kslJ �d S )N�{�G�z�?) �spawn_children_pairr$ r> r/ rf rD rM r rE rF )r4 �child� grandchild� child_ret�grandchild_retr8 r8 r9 �test_wait_non_children� s � �z"TestProcess.test_wait_non_childrenc C s� | � � }|�� t�tj�� |�d� W d � n1 s w Y t�tj�� |�d� W d � n1 s9w Y t�t�� |�d� W d � d S 1 sTw Y d S )Nrk r ���)r: �namer$ r> r/ rf rD � ValueErrorr@ r8 r8 r9 �test_wait_timeout� s � � "�z TestProcess.test_wait_timeoutc C s� | � � }t�tj�� |�d� W d � n1 sw Y |�� t�� t }t�� |k rGz|�d�}W n tjy@ Y nw t�� |k s/| � d��t rW|t j ksVJ �n|t j ks^J �| �|� d S )Nr rd )r: r$ r> r/ rf rD rC �timer �failr rE rG rF rH )r4 rA �stop_atrJ r8 r8 r9 � test_wait_timeout_nonblocking� s& � � � z)TestProcess.test_wait_timeout_nonblockingc C s� t �� }|jdd� |jdd� td�D ]}|jd d�}t|t�s#J �|dks)J �qt�t�� |jdd� W d � d S 1 sAw Y d S )Nrc )�interval�d � rr ) r/ r0 � cpu_percent�range� isinstance�floatr$ r> rt )r4 rA �_�percentr8 r8 r9 �test_cpu_percent� s "�z TestProcess.test_cpu_percentc C sJ t jdd d��}t�� �� |jsJ �W d � d S 1 s w Y d S )Nzpsutil.cpu_count)� return_value)r rU r/ r0 r} �called)r4 �mr8 r8 r9 � test_cpu_percent_numcpus_none� s "�z)TestProcess.test_cpu_percent_numcpus_nonec C s� t �� �� }|jdksJ |��|jdksJ |��|jdks!J |��|jdks*J |��tr5|jdks5J |��|j D ]}t � dt � t ||��� q8d S )Nr| z%H:%M:%S)r/ r0 � cpu_times�user�system� children_user�children_systemr �iowait�_fieldsrv �strftime� localtime�getattr)r4 �timesrs r8 r8 r9 �test_cpu_times s �zTestProcess.test_cpu_timesc C sN dd� }|� t �� �� }t�� }| j|j|jdd� | j|j|jdd� d S )Nc S sH t �� jd } t �� j| k r"td�D ]}|dC }qt �� j| k sd S d S )Ng�������?i�� � )r< r� r� r~ )rx �xr8 r8 r9 � waste_cpu s �z/TestProcess.test_cpu_times_2.<locals>.waste_cpu皙�����?)�delta)r/ r0 r� r< r� �assertAlmostEqualr� r� )r4 r� �a�br8 r8 r9 �test_cpu_times_2 s z TestProcess.test_cpu_times_2z not supportedc C sP t �� }|�� }|dksJ �t �� dkr|dksJ �|�� tt �� �v s&J �d S �Nr r] )r/ r0 �cpu_num� cpu_countr~ )r4 rA �numr8 r8 r9 � test_cpu_num s zTestProcess.test_cpu_numc C sb | � � }t�� }|�� }t|| �}|dkr$| �d|� d|� d|� ���t�dt�|�� �� d S )Nr� z expected: z , found: z, difference: z%Y %m %d %H:%M:%S)r: rv � create_time�absrw r� r� )r4 rA �nowr� � differencer8 r8 r9 �test_create_time% s ��z TestProcess.test_create_timez POSIX onlyc C sV t �� �� }|d ur'z tj�td��}W n ty t� d��w ||ks)J �d S d S )N�ttyzcan't rely on `tty` CLI) r/ r0 �terminalr< �path�realpathr( � RuntimeErrorr$ �skip)r4 r� r� r8 r8 r9 � test_terminal7 s � �zTestProcess.test_terminal)�only_ifc C s� t �� }|�� }ttd�� }|�� W d � n1 s w Y |�� }tsLtsL|j|jks1J �|j |j ks9J �t rK|j |j ksCJ �|j |j ksKJ �n|j |j ksTJ �|j|jks\J �|�� }t| �� d��}|�tdd d�� W d � n1 s|w Y |�� }|j |j ks�J �|j|jks�J �|j|jks�J �|j |j ks�J �t r�|j |j ks�J �|j |j ks�J �tt|��D ]}tr�|dkr�q�|| dks�J �|| dks�J �q�d S )N�rb�wbr� i@B �asciir� r )r/ r0 � io_counters�openr �readr r � read_count� write_countr � read_chars� write_chars� read_bytes� write_bytes� get_testfn�write�bytesr~ �len)r4 rA �io1�f�io2�ir8 r8 r9 �test_io_countersC sB �� � �z TestProcess.test_io_countersz linux onlyc sH � fdd�}t �� � ts� �� d t jksJ �t jdks J �t jdks%J �t jdks,J �t jdks3J �� �� }| �||� � �t j� t � �� �t jdfksPJ �t � t �� � jt jdd� W d � n1 shw Y � �t j� t � �� �t jdfks�J �� jt jdd� t � �� �t jdfks�J �t � t �� � jt jd d� W d � n1 s�w Y z � jt jdd� W n t j y� Y nw t j t d d �� � �t jd� W d � n1 s�w Y t j t d d �� � �t jd� W d � n1 s�w Y t j t d d �� � jdd� W d � d S 1 �s w Y d S ) Nc s& | \}}|t jkr d}� �||� d S �Nr )r/ �IOPRIO_CLASS_NONE�ionice)�init�ioclass�value�rA r8 r9 �cleanupn s z.TestProcess.test_ionice_linux.<locals>.cleanupr r] r� � � �r� � zioclass accepts no value��matchz$'ioclass' argument must be specified)r/ r0 r r� r� �IOPRIO_CLASS_RT�IOPRIO_CLASS_BE�IOPRIO_CLASS_IDLE� addCleanup�tupler$ r> rt rX )r4 r� r� r8 r� r9 �test_ionice_linuxk sL � �� � ��$�z TestProcess.test_ionice_linuxz!not supported on this win versionc C s* t �� }ts|�� t jksJ �|�� }| �|j|� |�t j� |�� t jks)J �|�t j� |�� t jks8J �z|�t j� W n t j yJ Y n w |�� t jksTJ �t j t dd�� |jt jdd� W d � n1 snw Y t j t dd�� |�t jd � W d � d S 1 s�w Y d S )Nz&value argument not accepted on Windowsr� r] r� zis not a valid priority)r/ r0 r r� � IOPRIO_NORMALr� �IOPRIO_VERYLOW� IOPRIO_LOW� IOPRIO_HIGHrX r$ r> � TypeErrorrt )r4 rA r� r8 r8 r9 �test_ionice_win� s. �� �"�zTestProcess.test_ionice_winc C s� dd l }t�t�� �}dd� tt�D �}|sJ |��|D ]I}tt|�}|dks)J �|t|�v rH|t||�ks8J �tr;q |�|�|� |�ksGJ �q |�|�}t |�dksUJ �|d dks]J �|d dkseJ �q d S )Nr c S s g | ] }|� d �r|�qS )�RLIMIT)� startswith��.0r� r8 r8 r9 � <listcomp>� � z/TestProcess.test_rlimit_get.<locals>.<listcomp>r� rr r] ) �resourcer/ r0 r< r= �dirr� r �rlimit� getrlimitr� )r4 r� rA �namesrs r� �retr8 r8 r9 �test_rlimit_get� s" �zTestProcess.test_rlimit_getc C s� | � � }|�tjd� |�tj�dksJ �tr7tjtdd�� tj� d��d� W d � n1 s2w Y t�t�� |�tjd� W d � d S 1 sOw Y d S )N)r^ r^ zcan't use prlimitr� r )r^ r^ r^ ) r: r� r/ � RLIMIT_NOFILEr r$ r> rt � _psplatformr0 r@ r8 r8 r9 �test_rlimit_set� s � "�zTestProcess.test_rlimit_setc C s4 t �� }| �� }|�t j�\}}zs|�t jd|f� t|d�� }|�d� W d � n1 s/w Y t�t ��"}t|d�� }|�d� W d � n1 sOw Y W d � n1 s^w Y |j j t j kslJ �W |�t j||f� |�t j�||fks�J �d S |�t j||f� |�t j�||fks�J �w )N� r� s XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXs XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX) r/ r0 r� r� � RLIMIT_FSIZEr� r� r$ r> �OSErrorr� �errno�EFBIG)r4 rA �testfn�soft�hardr� �excr8 r8 r9 � test_rlimit� s& � �� � �zTestProcess.test_rlimitc C s� t �� }|�t j�\}}zG|�t jd|f� |�t jt j|f� t| �� d�� }|�d� W d � n1 s7w Y W |�t j||f� |�t j�||fksRJ �d S |�t j||f� |�t j�||fksiJ �w )Nr� r� s XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX)r/ r0 r� r� � RLIM_INFINITYr� r� r� )r4 rA r� r� r� r8 r8 r9 �test_rlimit_infinity� s �� �z TestProcess.test_rlimit_infinityc C s< t �� }|�t j�\}}|t jksJ �|�t j||f� d S r- )r/ r0 r� r� r� )r4 rA r� r� r8 r8 r9 �test_rlimit_infinity_value� s z&TestProcess.test_rlimit_infinity_valuec C s� t �� }trz|�� }W n t jy t�d��w |�� }t� � |�� }||d ks.J �W d � d S 1 s9w Y d S )N�$on OpenBSD this requires root accessr] )r/ r0 r � num_threadsrX r$ r� r )r4 rA �step1�step2r8 r8 r9 �test_num_threads s �"�z TestProcess.test_num_threadsz WINDOWS onlyc C s t �� }|�� dks J �d S r� )r/ r0 � num_handlesr@ r8 r8 r9 �test_num_handles s z TestProcess.test_num_handlesc C s� t �� }trz|�� }W n t jy t�d��w |�� }t� �8 |�� }t|�t|�d ks2J �|d }|j |d ks?J �|j |d ksHJ �|j |d ksQJ �W d � d S 1 s\w Y d S )Nr� r] r r� ) r/ r0 r �threadsrX r$ r� r r� �id� user_time� system_time)r4 rA r� r� �athreadr8 r8 r9 � test_threads s �"�zTestProcess.test_threadsc C s� | � � }trz|�� W n tjy t�d��w t|�� j t dd� |�� D �� �dk s0J �t|�� j t dd� |�� D �� �dk sFJ �d S )Nr� c s � � | ]}|j V qd S r- )r r� r8 r8 r9 � <genexpr>? � � z-TestProcess.test_threads_2.<locals>.<genexpr>r� c s r r- )r r� r8 r8 r9 r C r ) r: r r r/ rX r$ r� r� r� r� �sumr� r@ r8 r8 r9 �test_threads_24 s �"�" �zTestProcess.test_threads_2c C s� t �� }|�� d d� \}}|�� }|dksJ �|dks J �d gd }|�� d d� \}}|�� }||ks7J �||ks=J �||ksCJ �~trZ|�� } | j| jksRJ �| j| jksZJ �|�� } | j D ] } t | | �dkslJ �qad S )Nr� r i`� ) r/ r0 � memory_info�memory_percentr �rss�wset�vms�pagefiler� r� ) r4 rA �rss1�vms1�percent1�memarr�rss2�vms2�percent2�memrs r8 r8 r9 �test_memory_infoG s( �z TestProcess.test_memory_infoc C s� t �� }t �� j}|�� }|jD ] }t||�}|dks J �|dkr#ts%tr&q||ks,J �qts3t s3t r:|j dks:J �trJ|j dksCJ �|j dksLJ �d S d S )Nr r )r/ r0 �virtual_memory�total�memory_full_infor� r� r r r r �uss�pss�swap)r4 rA r r rs r� r8 r8 r9 �test_memory_full_infof s �z!TestProcess.test_memory_full_infoc C s� t �� }|�� }t|�tt|��ksJ �|jdd�}|D ]�}|j�d�r%q tr-|jdkr-q tj� |j�s9J |j��t r~ztj� |j�sOtj� |j�sOJ |j��W q t y} tsZ� td�� }|�� }W d � n1 smw Y |j� d�|vr{� Y q w dtj�|j�vr�zt�|j�}W n ty� Y q w t�|j�s�J |j��q |D ],}|jD ]&}t||�} |dkr�q�|d v r�| s�J | ��q�t| t�s�J �| d ks�J | ��q�q�d S ) NF)�grouped�[�pvclockz/proc/self/smapsz (deleted)�64r� > �addr�permsr )r/ r0 � memory_mapsr� �setr� r� r r< �isabsr �exists�islink�AssertionErrorr r r� �basename�stat�FileNotFoundError�S_ISREG�st_moder� r� r �int) r4 rA �maps�ext_maps�ntr� �data�st�fnamer� r8 r8 r9 �test_memory_mapsv s^ ��� ��� �� ��z TestProcess.test_memory_mapsc sb t �� }t� � }dd� � � fdd�|�� D �}� |�|v s J �W d � d S 1 s*w Y d S )Nc S s t j�t j�| ��S r- )r< r� r� �normcaser� r8 r8 r9 �normpath� s z8TestProcess.test_memory_maps_lists_lib.<locals>.normpathc s g | ]}� |j ��qS r8 )r� r� �r= r8 r9 r� � s z:TestProcess.test_memory_maps_lists_lib.<locals>.<listcomp>)r/ r0 r r) )r4 rA r� �libpathsr8 r> r9 �test_memory_maps_lists_lib� s "�z&TestProcess.test_memory_maps_lists_libc C sf t �� }|�� t�t�� |jdd� W d � n1 s w Y ts)ts)tr1|jdd� d S d S )Nz?!?)�memtyper ) r/ r0 r r$ r> rt r r r r@ r8 r8 r9 �test_memory_percent� s � �z TestProcess.test_memory_percentc C sL | � � }|�� s J �|�� sJ �|�� |�� |�� r J �|�� r$J �d S r- )r: � is_runningrC rD r@ r8 r8 r9 �test_is_running� s zTestProcess.test_is_runningc C s� | � � }|�� }z|tksJ �W nI tyY tr0t|�tt�kr0tjj}||�|t�ks/J �n't j d � dt j d � �}z|� |d�t� |d�ksLJ �W n tyV Y nw Y nw t |ddg�}|dksgJ �d S )Nr �.r] � rZ zimport os; print('hey')�hey) r: �exer r. r r� r< r� r< �sys� version_info�replacer( )r4 rA rH r< �ver�outr8 r8 r9 �test_exe� s$ ���zTestProcess.test_exec C s� t ddg}| �|�}tr|�� g krt�d��ts ts tr)|�� d t ks'J �d S trOt rO|�� d }|t krOd� |�� dd � �d� |dd � �ksMJ �d S d� |�� �d� |�ks]J �d S )NrZ �2import time; [time.sleep(0.1) for x in range(100)]�OPENBSD: returned EBUSYr � r] ) r r: r �cmdliner$ r� r r r r �join)r4 rR rA �pyexer8 r8 r9 � test_cmdline� s � , zTestProcess.test_cmdlinezbroken on PYPYc C s� t g}|�dgd � |�ddg� | �|�}tr2z |�� |ks"J �W d S tjy1 t�d��w |�� }t rA|g krAt�d��||ksGJ �d S )Nz-v�2 rZ rO z#OPENBSD: process turned into zombierP ) r �extendr: r rR r/ � ZombieProcessr$ r� r )r4 rR rA r� r8 r8 r9 �test_long_cmdline� s � � z TestProcess.test_long_cmdlinec C sH | � � }|�� �� }tj�tj�tj���� }|� |�s"J ||f��d S r- ) r: rs �lowerr< r� r/ r� rI � executabler� )r4 rA rs rT r8 r8 r9 � test_name s zTestProcess.test_namezunreliable on PYPYc C s� t | jtjd d��}|ddg}| �|�}trGz|�� tj� |�ks$J �W d S t yF |� � t j krBtj� |��|�� �sAJ �n� Y d S w |�� tj� |�ksSJ �d S )Nr� )�suffixrZ rO )r" r� �string�digitsr: r rs r< r� r/ r. �statusr/ � STATUS_ZOMBIEr� )r4 rT rR rA r8 r8 r9 �test_long_name s � �� zTestProcess.test_long_namec C �\ t �� }|�� \}}}|t�� ksJ �|t�� ksJ �ttd�r*t�� |�� ks,J �d S d S �N� getresuid)r/ r0 �uidsr< �getuid�geteuid�hasattrre �r4 rA �real� effective�_savedr8 r8 r9 � test_uidsD � �zTestProcess.test_uidsc C rc rd )r/ r0 �gidsr< �getgid�getegidri � getresgidrj r8 r8 r9 � test_gidsQ ro zTestProcess.test_gidsc s� � fdd�}t �� � t�t�� � �d� W d � n1 s w Y � �� }| �||� tr�d }t jt j t j t j t j t j fD ]O}| j|d��? z� �|� W n t jyZ Y n&w � �� }|t j t j t j hv rz||ksq|d u ry|}||ksyJ �n||ks�J �W d � n1 s�w Y q@d S zJttd�r�t�tjt�� �� �� ks�J �� �d� � �� dks�J �ttd�r�t�tjt�� �� �� ks�J �ts �d� � �� dks�J �W d S W d S t jy� Y d S w )Nc s( z� � | � W d S tjy Y d S w r- )�nicer/ rX )r� r� r8 r9 r� _ s �z&TestProcess.test_nice.<locals>.cleanup�str)�prio� getpriorityr] r )r/ r0 r$ r> r� ru r� r �IDLE_PRIORITY_CLASS�BELOW_NORMAL_PRIORITY_CLASS�NORMAL_PRIORITY_CLASS�ABOVE_NORMAL_PRIORITY_CLASS�HIGH_PRIORITY_CLASS�REALTIME_PRIORITY_CLASS�subTestrX ri r< rx � PRIO_PROCESSr= r )r4 r� r� � highest_priorw �new_prior8 r� r9 � test_nice^ sp � ��� � � ��� � � ��zTestProcess.test_nicec C s t �� }|�� t jks J �d S r- )r/ r0 r` �STATUS_RUNNINGr@ r8 r8 r9 � test_status� s zTestProcess.test_statusc C s� | � � }|�� }tr7|�d�\}}t�� }|�d�r t�d��||ks%J �dt j v r3|t j d ks5J �d S d S |t�� ks?J �d S )N�\�$zrunning as service account� USERDOMAIN) r: �usernamer �split�getpass�getuser�endswithr$ r� r< �environ)r4 rA r� �domain� getpass_userr8 r8 r9 � test_username� s �zTestProcess.test_usernamec C s | � � }|�� t�� ksJ �d S r- )r: �cwdr< �getcwdr@ r8 r8 r9 �test_cwd� s zTestProcess.test_cwdc s( t ddg}| �|�� t� fdd�� d S )NrZ zFimport os, time; os.chdir('..'); [time.sleep(0.1) for x in range(100)]c s � � � tj�t�� �kS r- )r� r< r� �dirnamer� r8 r� r8 r9 �<lambda>� s z(TestProcess.test_cwd_2.<locals>.<lambda>)r r: r )r4 ra r8 r� r9 � test_cwd_2� s � zTestProcess.test_cwd_2c C s� t �� }|�� }|sJ |��| �|j|� ttd�r&|tt�|j��ks&J �t |�t t |��ks2J �tt t t j dd����}|D ]5}|�|g� |�� |gksQJ �ttd�rd|�� tt�|j��ksdJ �t|d�ru|�� d |� � ksuJ �q@|�g � tr�|�� |j�� ks�J �n|�� |ks�J �ttd�r�|�� tt�|j��ks�J �t�t�� |�d� W d � n1 s�w Y |�|� |�t |�� |�t|�� d S )N�sched_getaffinityT��percpu�num_cpur r] )r/ r0 � cpu_affinityr� ri r< �listr� r1 r� r* r~ r} r� r �_proc�_get_eligible_cpusr$ r> r� r� )r4 rA �initial�all_cpus�nr8 r8 r9 �test_cpu_affinity� s8 � � z TestProcess.test_cpu_affinityc C s | � � }ttjdd��d g}t�t�� |�|� W d � n1 s$w Y t�t�� |�tdd�� W d � n1 sAw Y t�t tf�� |�ddg� W d � n1 s_w Y t�t�� |�ddg� W d � d S 1 s|w Y d S ) NTr� � i' i�* r �1rr ) r: r� r/ r� r$ r> rt r� r~ r� )r4 rA � invalid_cpur8 r8 r9 �test_cpu_affinity_errs� s � � � "�z"TestProcess.test_cpu_affinity_errsc C s� t �� }|�� }|sJ |��| �|j|� t|�dkr!|d d� }g }tt|�d �D ]}|�dd� t�||�D �� q+|D ]}|�|� t |�� �t |�ksQJ �q>d S )N� r] c s s � | ] }|rt |�V qd S r- )r� )r� �subsetr8 r8 r9 r s � � �zATestProcess.test_cpu_affinity_all_combinations.<locals>.<genexpr>) r/ r0 r� r� r� r~ rW � itertools� combinations�sorted)r4 rA r� �combosr� �combor8 r8 r9 �"test_cpu_affinity_all_combinations� s � �z.TestProcess.test_cpu_affinity_all_combinationsz broken on BSDc sj t �� �| �� }��� � |� vsJ �t|d��B}|�d� |�� t� �fdd�� ��� � dd� � D �}tj � |�|v s?J �t rR� D ]}|j |krQ|j dksQJ �qCW d � n1 s\w Y � D ] }tj � |j �spJ |��qcd|� d �}| �td |g��td �D ]}d d� ��� D �}||v r� nt�d � q�tj � |�|v s�J �|D ] }tj � |�s�J |��q�d S )Nr� s xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxc s t ��� �t � �kS r- )r� � open_filesr8 ��filesrA r8 r9 r� s z-TestProcess.test_open_files.<locals>.<lambda>c S � g | ] }t j�|j��qS r8 �r< r� r< r� r8 r8 r9 r� r� z/TestProcess.test_open_files.<locals>.<listcomp>r� zimport time; f = open(r'z/', 'r'); [time.sleep(0.1) for x in range(100)];rZ r{ c S r� r8 r� r� r8 r8 r9 r� - r� rk )r/ r0 r� r� r� r� �flushr r< r� r< r �position�isfiler: r r~ rv �sleep)r4 r� r� � filenames�filerR r� r8 r� r9 �test_open_files s> � � � �zTestProcess.test_open_filesc C s t �� }tjj}| �� }t|d��m}|�� D ]}||j�||j�ks)|j |� � kr+ n q| � d|�� ����||j�||j�ksBJ �t rL|j dksKJ �n |j |� � ksUJ �|�� d }|d |jksdJ �|d |j ksmJ �|j|�� vsvJ �W d � d S 1 s�w Y d S )N�wzno file found; files=rr r r] ) r/ r0 r< r� r< r� r� r� rs �fd�filenorw r )r4 rA r< r� �fileobjr� �ntupler8 r8 r9 �test_open_files_27 s& � "�z TestProcess.test_open_files_2c C s� t �� }| �� }|�� }t|d�}| �|j� |�� |d ks!J �t�� }| �|j� |�� |d ks5J �|�� |�� |�� |ksEJ �d S )Nr� r] r� )r/ r0 r� �num_fdsr� r� �close�socket)r4 rA r� �startr� �sockr8 r8 r9 � test_num_fdsR s zTestProcess.test_num_fdsz not reliable on OPENBSD & NETBSDc C sP t �� }t|�� �}td�D ]}t�d� t|�� �}||kr" d S q| �d��)Nr� g�������?z2num ctx switches still the same after 2 iterations)r/ r0 r �num_ctx_switchesr~ rv r� rw )r4 rA �beforer� �afterr8 r8 r9 �test_num_ctx_switchesa s � z!TestProcess.test_num_ctx_switchesc C sF t �� }ttd�r|�� t�� ksJ �| �� }|�� t�� ks!J �d S )N�getppid)r/ r0 ri r< �ppidr� r: r= r@ r8 r8 r9 � test_ppido s zTestProcess.test_ppidc C sD | � � }|�� jt�� ksJ �t�� d }t�|��� d u s J �d S r� )r: �parentr1 r< r= r/ �pidsr0 )r4 rA � lowest_pidr8 r8 r9 � test_parentv s zTestProcess.test_parentc C s8 t �� }| �� \}}|�� |ksJ �|�� |ksJ �d S r- )r/ r0 rl r� �r4 r� rm rn r8 r8 r9 �test_parent_multi} s z TestProcess.test_parent_multic C s` t �� }|�� s J �| �� \}}|�� d |ksJ �|�� d |ks$J �|�� d |ks.J �d S r� )r/ r0 �parentsrl r� r8 r8 r9 � test_parents� s zTestProcess.test_parentsc C s� t �� }|�� r J �|jdd�rJ �| jdd�}|�� }|jdd�}||fD ] }t|�dks0J �|d j|jks:J �|d �� |jksEJ �q&d S )NT�� recursiver )� creationflagsr] )r/ r0 �childrenr: r� r1 r� )r4 r� rm � children1� children2r� r8 r8 r9 � test_children� s �zTestProcess.test_childrenc C sb t �� }| �� \}}|�� |gksJ �|jdd�||gks J �|�� |�� |jdd�r/J �d S )NTr� )r/ r0 rl r� rM rD r� r8 r8 r9 �test_children_recursive� s z#TestProcess.test_children_recursivec C s� t �t�}t�� D ]}z ||�� d7 < W q tjy! Y q w t|�� dd� d�d }t r9|dkr9t � d��t� |�}z|j dd�}W n tjyQ Y d S w t|�tt|��ks^J �d S ) Nr] c S s | d S )Nr] r8 )r� r8 r8 r9 r� � s z6TestProcess.test_children_duplicates.<locals>.<lambda>)�keyr zPID 0Tr� )� collections� defaultdictr4 r/ � process_iterr� �Error�max�itemsr r$ r� r0 r� rX r� r* )r4 �tablerA r1 �cr8 r8 r9 �test_children_duplicates� s" � � z$TestProcess.test_children_duplicatesc C s| t �� }| �� \}}|jdd�}t|�dksJ �|d |ks J �|d |ks(J �|�� }|d |ks4J �|d |ks<J �d S )NTr� r� r r] )r/ r0 rl r� r� r� )r4 r� rm rn r� r� r8 r8 r9 �test_parents_and_children� s z%TestProcess.test_parents_and_childrenc C sX | � � }|�� td�D ]}|�� tjkr nt�d� q |�� |�� tjks*J �d S )Nr{ rk ) r: rh r~ r` r/ �STATUS_STOPPEDrv r� ri )r4 rA r� r8 r8 r9 �test_suspend_resume� s z TestProcess.test_suspend_resumec C sn t �t�� t�d� W d � n1 sw Y t �t�� t�d� W d � d S 1 s0w Y d S )Nr� rr )r$ r> r� r/ r0 rt �r4 r8 r8 r9 �test_invalid_pid� s � "�z TestProcess.test_invalid_pidc C s� t �� }|jddgd�}t|�� �ddgksJ �t �tt �� ��}|jdgdd�}t|d t�s8|d dks8J �t j ddt j d �� |jd gd d�d d iksPJ �W d � n1 sZw Y t j ddt � |j d�d ��% t�t j �� |jd gd� W d � n1 s�w Y W d � n1 s�w Y t j ddt �|j d�d �� |jd gdd�d diks�J �W d � n1 s�w Y t j ddtd ��2 |�� }d t|�� �vs�J �t�t�� |jd gd� W d � n1 s�w Y W d � n 1 �sw Y t�t�� |�d� W d � n 1 �s w Y t�t�� |�dg� W d � n 1 �s9w Y t�t�� |�dd g� W d � d S 1 �sWw Y d S ) NrH rs )�attrs�net_connections�foo)r� �ad_valuezpsutil.Process.niceT)�createrT ru r] �bar)r/ r0 �as_dictr� �keys�minr� r r� r rU rX r2 r1 r$ r> rX �NotImplementedErrorr� rt )r4 rA �dr8 r8 r9 � test_as_dict� s` � � � �� � � � � �� � � � $�zTestProcess.test_as_dictc C s� t �� }t�d��+}|�� � |�� |�� W d � n1 s!w Y |jdks-J �W d � n1 s7w Y t�d��}|�� |�� W d � n1 sTw Y |jdks`J �d S )N�$psutil._psplatform.Process.cpu_timesr] r� �r/ r0 r rU �oneshotr� � call_count�r4 rA r� r8 r8 r9 � test_oneshot s � � �zTestProcess.test_oneshotc C s4 t �� }t�d��c}t�d��N}|�� �, |�� |�� |�� � |�� |�� W d � n1 s4w Y W d � n1 sCw Y |jdksOJ �|jdksVJ �W d � n1 s`w Y W d � n1 sow Y t�d��}|�� |�� W d � n1 s�w Y |jdks�J �d S )Nr� z(psutil._psplatform.Process.oneshot_enterr] r� r� )r4 rA �m1�m2r� r8 r8 r9 �test_oneshot_twice s, �� � �� � �z TestProcess.test_oneshot_twicec C s� | � � \}}|�� }|�� }||ksJ �|�� � |�� |ks!J �|�� |ks)J �W d � n1 s3w Y |�� � |�� |ksEJ �|�� |ksMJ �W d � d S 1 sXw Y d S r- )rl r� r� )r4 �p1�p2�p1_ppid�p2_ppidr8 r8 r9 �test_oneshot_cache. s � "�z TestProcess.test_oneshot_cachec sn �fdd�}�� � � � �� � �� tr t� fdd�� ��� � t� �}|�|j�D ] \}}|||� q+d S )Nc sz z| � }W n$ t jy � t jy Y d S t jy) tr(|dv r(Y d S � w tr2|dv r2d S � �| �d|�d���)N> r r� > rH rs z didn't raise NSP and returned z instead)r/ rX r2 rX r r rw )�fun�fun_namer� r� r8 r9 �assert_raises_nspE s � �zFTestProcess.test_halfway_terminated_process.<locals>.assert_raises_nspc s � j t�� vS r- )r1 r/ r� r8 r� r8 r9 r� \ s z=TestProcess.test_halfway_terminated_process.<locals>.<lambda>) r: rM rD r r rH r# �iter�all)r4 r �nsr rs r8 )rA r4 r9 � test_halfway_terminated_process= s �z+TestProcess.test_halfway_terminated_processc C s | � � \}}| �|� d S r- )� spawn_zombie�assertProcessZombie)r4 �_parent�zombier8 r8 r9 �test_zombie_processc s z TestProcess.test_zombie_processc C sX t �� }tjdt �d�d��}|�� sJ �|jsJ �W d � d S 1 s%w Y d S )Nzpsutil.Processr rS )r/ r0 r rU rX rC r� r� r8 r8 r9 �$test_zombie_process_is_running_w_exch s � "�z0TestProcess.test_zombie_process_is_running_w_excc C s^ t �� }tjdt �d�d��}|�� t jksJ �|js J �W d � d S 1 s(w Y d S )Nz!psutil._psplatform.Process.statusr rS )r/ r0 r rU rX r` ra r� r� r8 r8 r9 � test_zombie_process_status_w_excs s � "�z,TestProcess.test_zombie_process_status_w_excc C s� | � � }t�|j�}|j|�� d f|_tt�� � |jtjv s"J �|� � r(J �t j � tj dd��% t�t�� ��}tt�� � W d � n1 sJw Y W d � n1 sYw Y d|j� �|�� v sjJ �|jtjvsrJ �|t�|j�ks|J �d}t|�}|j|j|j dd�D ]5\}}| j|d��# tjtj|d �� |� W d � n1 s�w Y W d � n1 s�w Y q�d t|�v s�J �d t |�v s�J �tjtj|d �� |� � W d � n1 s�w Y tjtj|d �� |� � W d � n 1 �sw Y tjtj|d �� |� � W d � n 1 �s$w Y tjtj|d �� |� � W d � d S 1 �sBw Y d S ) Nr{ � PSUTIL_DEBUGTz+refreshing Process instance for reused PID z4process no longer exists and its PID has been reusedF)� clear_cache)rs r� zterminated + PID reused)!r. r/ r0 r1 r� �_identr� r� �_pmaprC r rU �object�_common� contextlib�redirect_stderr�io�StringIO�getvaluer# r �setters�killersr r$ r> r2 rv �reprr� r� r� r� )r4 �subprA r� �msgr r rs r8 r8 r9 �test_reused_pid sR �� � � �� �� � � � $�zTestProcess.test_reused_pidc C s� dt �� vr5t�t j�� t �d� W d � n1 s w Y t �d�r(J �t �d��� dks3J �d S t �d�}tr?t j nt }t�|�� |� � W d � n1 sUw Y t�|�� |� � W d � n1 snw Y t�|�� |� � W d � n1 s�w Y t�|�� |�� W d � n1 s�w Y t�|�� |�� W d � n1 s�w Y t�|�� |�tj� W d � n1 s�w Y t|�}|�|j|j �D ]D\}}z|� }W n t j y� Y q�w |dv �r|jdk�sJ �q�|dk�r t�rdnd}|�� |k�s J �q�|dk�r*|�s*J |��q�t�s?dt �� v �s7J �t �d��sAJ �d S d S )Nr r] > rp rf r� zNT AUTHORITY\SYSTEM�rootrs )r/ r� r$ r> r2 r0 � pid_existsr� r rX rt rD rM rh ri rC rO rE rF r# r �gettersr! rk r� r )r4 rA r� r r rs r� r� r8 r8 r9 � test_pid_0� s^ � � � � � � � � ��zTestProcess.test_pid_0c C sP dd� }d | _ t�� }||�� �}|tj�� �}ts"tr$||ks&J �d S d S d S )Nc S sB g d�}t r |�g d�� |D ]}| �|d � qdd� | �� D �S )N)�PLAT�HOME�PYTEST_CURRENT_TEST�PYTEST_VERSION)�__CF_USER_TEXT_ENCODING� VERSIONER_PYTHON_PREFER_32_BIT�VERSIONER_PYTHON_VERSIONr1 c S s6 i | ]\}}|� d d�� dd�|� d d�� dd��qS )� rF � )rK )r� �k�vr8 r8 r9 � <dictcomp>� s ���z@TestProcess.test_environ.<locals>.clean_dict.<locals>.<dictcomp>)r rW �popr� )r� �excluders r8 r8 r9 � clean_dict� s �z,TestProcess.test_environ.<locals>.clean_dict)�maxDiffr/ r0 r� r< �copyr r )r4 r9 rA �d1�d2r8 r8 r9 � test_environ� s �zTestProcess.test_environz<macOS 11+ can't get another process environment, issue #2084z(sometimes fails on `assert is_running()`c C s� t �d�}t| �� |d�}| j|gtjtjd�}t�|j �}t |j � |� � s)J �|j � � dks2J �trHtrHz|�� }W n tjyG Y d S w |�� }|ddd�ksUJ �|�� |jdks`J �d S ) Na� #include <unistd.h> #include <fcntl.h> char * const argv[] = {"cat", 0}; char * const envp[] = {"A=1", "X", "C=3", 0}; int main(void) { // Close stderr on exec so parent can wait for the // execve to finish. if (fcntl(2, F_SETFD, FD_CLOEXEC) != 0) return 0; return execve("/bin/cat", argv, envp); } )�c_code)�stdinr\ � r� �3)�A�Cr )�textwrap�dedentr! r� r. r_ r` r/ r0 r1 r+ rC r\ r� r r r� rX � communicate� returncode)r4 rJ �cexer7 rA �envr8 r8 r9 �test_weird_environ� s& � �z TestProcess.test_weird_environN)h�__name__� __module__� __qualname__�__doc__r: rB rK rN rQ r$ �mark�skipifr rY rb r rj rq ru ry r� r� r� r� r r� r� r� r r* r r� r r� r r� r r� r� r� r� r� r� r r r r' r) r r r r" r r; r@ rB rD rN rU r rY r\ rb rn rt r� r� r� r� r� r r� r� r� r r� r� r� r r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r r r r& r* r r> r rK r8 r8 r8 r9 r, I s & +� 0 - B & & � 4 & ). ��r, c @ s< e Zd ZdZedd� �Zdd� Zdd� Zdd � Zd d � Z d S ) � TestPopenz Tests for psutil.Popen class.c C s t � d S r- r% )�clsr8 r8 r9 � tearDownClass* s zTestPopen.tearDownClassc C s� t ddg}tj|tjtjtd��5}|�� |�� |j t |�s"J �t � t �� |j W d � n1 s5w Y |�� W d � n1 sHw Y tr\|�d�tj ksZJ �d S |�d�tjksfJ �d S )NrZ �3import time; [time.sleep(0.1) for x in range(100)];��stdoutr\ rJ r^ )r r/ �Popenr_ r` r rs r� r@ r� r$ r> r? r� rM r rD rE rF �r4 ra �procr8 r8 r9 � test_misc. s. �� � � zTestPopen.test_miscc C s| t jtdgtjtjtjtd�� }|�� W d � n1 s w Y |jjs)J �|j js/J �|j js5J �|j dks<J �d S )Nz-V)rW r\ r@ rJ r ) r/ rX r r_ r` r rG rW �closedr\ r@ rH )r4 rZ r8 r8 r9 �test_ctx_managerI s � � zTestPopen.test_ctx_managerc C s� t ddg}tj|tjtjtd���}|�� |�� t� tj �� |�� W d � n1 s.w Y t� tj �� |� � W d � n1 sHw Y t� tj �� |� t j� W d � n1 sdw Y tr�t� tj �� |� t j� W d � n1 s�w Y t� tj �� |� t j� W d � n1 s�w Y W d � d S W d � d S W d � d S 1 s�w Y d S �NrZ rU rV )r r/ rX r_ r` r rM rD r$ r> r2 rC rO rE rF r � CTRL_C_EVENT�CTRL_BREAK_EVENTrY r8 r8 r9 �test_kill_terminateW sD �� � � � � ���"�z TestPopen.test_kill_terminatec C s� t ddg}tj|tjtjtd��1}|�� |�� t� t �� |j W d � n1 s,w Y W d � d S W d � d S 1 sDw Y d S r^ ) r r/ rX r_ r` r rM rD r$ r> r? r� rY r8 r8 r9 �test__getattribute__t s& �� ��"�z TestPopen.test__getattribute__N) rL rM rN rO � classmethodrT r[ r] ra rb r8 r8 r8 r9 rR ' s rR )>rO r� r r� r� r r� r< rE r� r0 r^ r_ rI rE rv �unittestr r/ r r r r r r r r r �psutil._commonr � psutil.testsr r r r r r r r r r r r r r r r r r r r! r" r# r$ r& r' r( r) r* r+ r, rR r8 r8 r8 r9 �<module> s� i tests/__pycache__/test_posix.cpython-310.pyc 0000644 00000033256 15030211765 0015031 0 ustar 00 o �h#C � @ s� d Z ddlZddlZddlZddlZddlZddlZddlZddlm Z ddl Z ddl m Z ddl m Z ddl m Z ddl mZ ddl mZ dd l mZ dd l mZ dd lmZ dd lmZ dd lmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlm Z er�ddl Z ddl Z ddl m Z d*dd�Z!dd� Z"dd� Z#d d � Z$d d � Z%d d!� Z&ej'j(e d"d#�G d$d%� d%e��Z)ej'j(e d"d#�G d&d'� d'e��Z*ej'j(e d"d#�G d(d)� d)e��Z+dS )+zPOSIX specific tests.� N)�mock)�AIX)�BSD)�LINUX)�MACOS)�OPENBSD)�POSIX)�SUNOS)�AARCH64)�HAS_NET_IO_COUNTERS)� PYTHON_EXE)�PsutilTestCase)�pytest)�retry_on_failure)�sh)�skip_on_access_denied)�spawn_testproc)� terminate)� getpagesizec C s� dg}t r |�d� |dur|�dt|�g� nts tr"|�d� n|�d� tr/| �dd�} |�d | g� t|�}t r@|�� n|�� d d� }g }|D ]}|� � }zt |�}W n t ya Y nw |�|� qL|du rn|S |d S ) zwWrapper for calling the ps command with a little bit of cross-platform support for a narrow range of features. �psz --no-headersNz-pz-A�ax�start�stimez-o� r ) r �append�extend�strr r �replacer � splitlines�strip�int� ValueError)�fmt�pid�cmd�output� all_output�line� r( �J/usr/local/CyberCP/lib/python3.10/site-packages/psutil/tests/test_posix.pyr - s2 � r c C s" d}t rd}t|| ��� }|d S )N�command�commr )r r �split)r# �fieldr* r( r( r) �ps_named s r. c C s0 d}t strd}t|| �}t�dd|�}|�� S )Nr* �argsz\(python.*?\)$� )r r r �re�subr )r# r- �outr( r( r) �ps_argsl s r4 c C � d}t rd}t|| �S )N�rss�rssize�r r �r# r- r( r( r) �ps_rssv � r: c C r5 )N�vsz�vsizer8 r9 r( r( r) �ps_vsz} r; r> c C s� z t d| � ���� }W n ty% } zdt|��� v r t�d��� d }~ww |�d�d }|�� }t|d �d }t|d �d }t|d �d }t |d � d d ��}||||fS ) Nzdf -k z device busyzdf returned EBUSY� r � � � � �%r0 ) r r � RuntimeErrorr �lowerr �skipr, r �floatr ) �devicer3 �errr' �fields� sys_total�sys_used�sys_free� sys_percentr( r( r) �df� s �� rP z POSIX only��reasonc @ s� e Zd ZdZedd� �Zedd� �Zdd� Zdd � Zd d � Z d d � Z dd� Z e � e � dd� ��Ze � e � dd� ��Zdd� Zdd� Zdd� Zdd� ZejjepPed d �d d � �Zd d!� Ze � d"d#� �Zejje d$d �ejje d%d �d&d'� ��Z d(S ))� TestProcesszBCompare psutil results against 'ps' command line utility (mainly).c C s t tddgtjd�j| _d S )Nz-Ez-O)�stdin)r r � subprocess�PIPEr# ��clsr( r( r) � setUpClass� s ��zTestProcess.setUpClassc C s t | j� d S �N)r r# rW r( r( r) � tearDownClass� s zTestProcess.tearDownClassc C �, t d| j�}t�| j��� }||ksJ �d S )N�ppid)r r# �psutil�Processr] )�self�ppid_ps� ppid_psutilr( r( r) � test_ppid� � zTestProcess.test_ppidc C �. t d| j�}t�| j��� j}||ksJ �d S )N�uid)r r# r^ r_ �uids�real)r` �uid_ps� uid_psutilr( r( r) �test_uid� � zTestProcess.test_uidc C re )N�rgid)r r# r^ r_ �gidsrh )r` �gid_ps� gid_psutilr( r( r) �test_gid� rl zTestProcess.test_gidc C r\ )N�user)r r# r^ r_ �username)r` � username_ps�username_psutilr( r( r) � test_username� rd zTestProcess.test_usernamec C s` t �� }tjdtd��}|�� t|�� j�ksJ �|j s J �W d � d S 1 s)w Y d S )Nzpsutil.pwd.getpwuid�� side_effect) r^ r_ r �patch�KeyErrorrs r rg rh �called)r` �p�funr( r( r) �test_username_no_resolution� s "�z'TestProcess.test_username_no_resolutionc C �< t �d� t| j�}t�| j��� d d }||ks J �d S )N皙�����?r r@ )�time�sleepr: r# r^ r_ � memory_info)r` �rss_ps� rss_psutilr( r( r) �test_rss_memory� � zTestProcess.test_rss_memoryc C r )Nr� r r@ )r� r� r> r# r^ r_ r� )r` �vsz_ps� vsz_psutilr( r( r) �test_vsz_memory� r� zTestProcess.test_vsz_memoryc C sv t | j�}tj�|��� }t�| j��� �� }t � dd|�}t � dd|�}t � dd|�}t � dd|�}||ks9J �d S )Nz\d.\dr0 z\d) r. r# �os�path�basenamerF r^ r_ �namer1 r2 )r` �name_ps� name_psutilr( r( r) � test_name� s zTestProcess.test_namec C s� d}g d�}t jd|d��4 t jd|d�� t�� }|�� dks"J �W d � n1 s,w Y W d � d S W d � d S 1 sDw Y d S )N�long-program-name)�long-program-name-extended�foo�bar� psutil._psplatform.Process.name�� return_value�"psutil._psplatform.Process.cmdliner� )r ry r^ r_ r� )r` r� �cmdliner| r( r( r) �test_name_long� s � ��"�zTestProcess.test_name_longc C s� d}t jd|d��8 t jdt�dd�d�� t�� }|�� dks"J �W d � n1 s,w Y W d � d S W d � d S 1 sDw Y d S �Nr� r� r� r� r r0 rw )r ry r^ � AccessDeniedr_ r� �r` r� r| r( r( r) � test_name_long_cmdline_ad_exc� s � ��"�z)TestProcess.test_name_long_cmdline_ad_excc C s� d}t jd|d��J t jdt�dd�d��& t�� }t�tj�� |�� W d � n1 s/w Y W d � n1 s>w Y W d � d S W d � d S 1 sVw Y d S r� )r ry r^ � NoSuchProcessr_ r �raisesr� r� r( r( r) � test_name_long_cmdline_nsp_exc� s � �� ��"�z*TestProcess.test_name_long_cmdline_nsp_exczps -o start not availablerQ c C s\ t d| j�}t�| j��� }tj�|��d�}t|�}tj�|��d�}|||hv s,J �d S )Nr z%H:%M:%S) r r# r^ r_ � create_time�datetime� fromtimestamp�strftime�round)r` �time_ps� time_psutil�time_psutil_tstamp�round_time_psutil�round_time_psutil_tstampr( r( r) �test_create_time s ����z TestProcess.test_create_timec C s^ t | j�}t�| j��� }z ||ksJ �W d S ty. |d t|�� }||ks+J �Y d S w rZ )r. r# r^ r_ �exe�AssertionError�len)r` � ps_pathname�psutil_pathname�adjusted_ps_pathnamer( r( r) �test_exe s �zTestProcess.test_exec C sV t | j�}d�t�| j��� �}tr#t|�t|�k r#|�|�s!J �d S ||ks)J �d S )N� ) r4 r# �joinr^ r_ r� r r� � startswith)r` � ps_cmdline�psutil_cmdliner( r( r) � test_cmdline+ s zTestProcess.test_cmdlineznot reliable on SUNOSznot reliable on AIXc C s( t d| j�}t�� �� }||ksJ �d S )N�nice)r r# r^ r_ r� )r` �ps_nice� psutil_nicer( r( r) � test_nice9 s zTestProcess.test_niceN) �__name__� __module__� __qualname__�__doc__� classmethodrY r[ rc rk rq rv r~ r r r� r� r� r� r� r� r �mark�skipifr r r� r� r� r r r� r( r( r( r) rS � s: rS c @ s� e Zd ZdZe� dd� �Zejje dd�ejje � d� dd�ejje dd�d d � ���Z e� d d � �Ze� d d� �Zdd� Zdd� Zdd� Zdd� Zejjedd�e� dd� ��ZdS )�TestSystemAPIszTest some system APIs.c s~ t td��� t�� �tstrd� vr� �dd� t� �t�� dkr=� fdd��D ��fdd�� D � }| �dt |� ��d S )Nr# r r c � g | ]}|� vr|�qS r( r( ��.0�x)�pids_psr( r) � <listcomp>R � z,TestSystemAPIs.test_pids.<locals>.<listcomp>c r� r( r( r� )� pids_psutilr( r) r� R s z difference: ) �sortedr r^ �pidsr r �insertr� �failr )r` � differencer( )r� r� r) � test_pidsE s ��zTestSystemAPIs.test_pidszunreliable on SUNOSrQ �ifconfigzno ifconfig cmdz not supportedc C sP t d�}tjdd�D ]}|�� D ] }|�|�r n q| �d|� d|� ���q d S )Nz ifconfig -aT)�perniczcouldn't find z nic in 'ifconfig -a' output )r r^ �net_io_countersr, r� r� )r` r% �nicr' r( r( r) �test_nic_namesY s �����z TestSystemAPIs.test_nic_namesc C s� t d�}|�� s t�d��|�d�}dd� |D �}dd� |D �}t|�tt�� �ks,J �| jt�� |d��1 t t�� �D ]!\}}|j || ksIJ �|j || ksRJ �|j d ur]t� |j � q<W d � d S 1 siw Y d S )N�who -u�no users on this systemr? c S � g | ]}|� � d �qS )r �r, r� r( r( r) r� p r� z-TestSystemAPIs.test_users.<locals>.<listcomp>c S r� )r r� r� r( r( r) r� q r� �r^ �who)r r r rG r, r� r^ �users�subTest� enumerater� �terminalr# r_ )r` r3 �linesr� � terminals�idx�ur( r( r) � test_usersj s ��"�zTestSystemAPIs.test_usersc C s t d�}|�� s t�d��d }t�d|�}|rd}n't�d|�}|r%d}n t�d|�}|r0d}nt�d |�}|rAd}d d � |D �}|sKt�d |� ���| jt�� |d ��& t t�� �D ]\}}t j � |j �� |�}||| ksqJ �q[W d � d S 1 s}w Y d S )Nr� r� z \d\d\d\d-\d\d-\d\d \d\d:\d\dz%Y-%m-%d %H:%Mz [A-Z][a-z][a-z] \d\d \d\d:\d\dz %b %d %H:%Mz[A-Z][a-z][a-z] \d\dz%b %dz[a-z][a-z][a-z] \d\dc S s g | ]}|� � �qS r( )� capitalizer� r( r( r) r� � s z5TestSystemAPIs.test_users_started.<locals>.<listcomp>z&cannot interpret tstamp in who output r� )r r r rG r1 �findallr� r^ r� r� r� r� �startedr� )r` r3 �tstampr� r� r� � psutil_valuer( r( r) �test_users_startedz s: ���"�z!TestSystemAPIs.test_users_startedc C � t jdttjd�d��+}t�t�� tj� t � � � W d � n1 s$w Y |j s.J �W d � d S 1 s9w Y d S )Nzpsutil._psposix.os.killr0 rw ) r ry �OSError�errno�EBADFr r� r^ �_psposix� pid_existsr� �getpidr{ �r` �mr( r( r) �test_pid_exists_let_raise� s � � "�z(TestSystemAPIs.test_pid_exists_let_raisec C r� )N�psutil._psposix.os.waitpidr0 rw ) r ry r� r� r� r r� r^ r� �wait_pidr� r� r{ r� r( r( r) �test_os_waitpid_let_raise� s � � "�z(TestSystemAPIs.test_os_waitpid_let_raisec C s� t jdttjd�d��/}t�tjj �� tjj t � � dd� W d � n1 s(w Y |j s2J �W d � d S 1 s=w Y d S )Nr� r0 rw g{�G�z�?)�timeout)r ry r� r� �EINTRr r� r^ r� �TimeoutExpiredr� r� r� r{ r� r( r( r) �test_os_waitpid_eintr� s � � "�z$TestSystemAPIs.test_os_waitpid_eintrc C sx t jddd��+}t�t�� tj�t� � � W d � n1 s w Y |j s*J �W d � d S 1 s5w Y d S )Nr� )r ���r� ) r ry r r� r! r^ r� r� r� r� r{ r� r( r( r) � test_os_waitpid_bad_ret_status� s � � "�z-TestSystemAPIs.test_os_waitpid_bad_ret_statuszunreliable on AIXc C s� d}t jdd�D ]e}t �|j�}z t|j�\}}}}W n% ty@ } zt|��� }d|v s5d|v s5d|v r;W Y d }~q� d }~ww t |j | �|k sLJ �t |j | �|k sWJ �t |j | �|k sbJ �t |j | �dksmJ �qd S )Ni @ F)�allzno such file or directoryzraw devices not supportedzpermission deniedr )r^ �disk_partitions� disk_usage� mountpointrP rI rE r rF �abs�total�used�free�percent) r` � tolerance�part�usagerL rM rN rO rJ r( r( r) �test_disk_usage� s& �� �z TestSystemAPIs.test_disk_usageN)r� r� r� r� r r� r r� r� r �shutil�whichr r� r� r� r� r� r� r r r r( r( r( r) r� A s& $ r� c @ s e Zd Zdd� ZdS )�TestMiscc C s4 t � }|dks J �|t� � ksJ �|tjksJ �d S )Nr )r �resource�mmap�PAGESIZE)r` �pagesizer( r( r) �test_getpagesize� s zTestMisc.test_getpagesizeN)r� r� r� r r( r( r( r) r � s r rZ ),r� r� r� r� r1 r rU r� �unittestr r^ r r r r r r r � psutil.testsr r r r r r r r r r r r �psutil._psutil_posixr r r. r4 r: r> rP r� r� rS r� r r( r( r( r) �<module> sZ 7 - ! tests/__pycache__/test_sunos.cpython-310.pyc 0000644 00000002614 15030211765 0015030 0 ustar 00 o �h� � @ sj d Z ddlZddlZddlmZ ddlmZ ddlmZ ddlmZ ejj e dd�G d d � d e��Z dS ) zSun OS specific tests.� N)�SUNOS)�PsutilTestCase)�pytest)�shz SUNOS only)�reasonc @ s e Zd Zdd� Zdd� ZdS )�SunOSSpecificTestCasec C s� t dtjd � d��}|�� �d�dd � }|s td��d }}|D ]}|�� }t|d �d }t|d �d }q"|| }t�� }|j |ksHJ �|j |ksOJ �|j |ksVJ �d S ) Nzenv PATH=/usr/sbin:/sbin:�PATHz swap -l� � z no swap device(s) configuredr � i � ) r �os�environ�strip�split� ValueError�int�psutil� swap_memory�total�used�free) �self�out�linesr r �line�fieldsr � psutil_swap� r �J/usr/local/CyberCP/lib/python3.10/site-packages/psutil/tests/test_sunos.py�test_swap_memory s z&SunOSSpecificTestCase.test_swap_memoryc C s& t d�}t�� t|�d��ksJ �d S )Nz/usr/sbin/psrinfor )r r � cpu_count�lenr )r r r r r �test_cpu_count% s z$SunOSSpecificTestCase.test_cpu_countN)�__name__� __module__� __qualname__r r# r r r r r s r ) �__doc__r r r � psutil.testsr r r �mark�skipifr r r r r �<module> s tests/__pycache__/test_process_all.cpython-310.pyc 0000644 00000037551 15030211765 0016177 0 ustar 00 o �h�G � @ sp d Z ddlZddlZddlZddlZddlZddlZddlZddlZddlm Z ddlm Z ddlm Z ddlm Z ddlm Z ddlmZ dd lmZ dd lmZ dd lmZ dd lmZ dd lmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlm Z ddlm Z e o�e o�e Z dd� Z G dd� de�Z G dd � d e�Z!dS ) zeIterate over all process PIDs and for each one of them invoke and test all psutil.Process() methods. � N)�AIX)�BSD)�FREEBSD)�LINUX)�MACOS)�NETBSD)�OPENBSD)�OSX)�POSIX)�WINDOWS)� CI_TESTING)�PYTEST_PARALLEL)�VALID_PROC_STATUSES)�PsutilTestCase)�check_connection_ntuple)�create_sockets)� is_namedtuple)�is_win_secure_system_proc)�process_namespace)�pytestc s t � ���fdd�� � ����fdd�}zt����W n tjy, ���� i Y S w z ��ddg�}W n tjyF ���� Y d S w |d |d ��d�ji}t��}|j |j dd �D ]%\}}z|� ||< W qa tj y� } z � |���� W Y d }~qad }~ww |� |S ) Nc s� �� | j� � | jd ur�� | j|� t| tj�r2��|� | jd ur1��| jd� �� | j|� n t| tj �r=�� |� t | � t | � d S �Nr ) � assertEqual�pid�name� isinstance�psutil� ZombieProcess�assertProcessZombie�ppid�assertGreaterEqual� NoSuchProcess�assertProcessGone�str�repr)�exc�procr r )r �tcase� �P/usr/local/CyberCP/lib/python3.10/site-packages/psutil/tests/test_process_all.py�check_exception2 s � z"proc_info.<locals>.check_exceptionc sT �dkr(z�� d� W d S tjy' } z� | ���� W Y d } ~ d S d } ~ ww d S r )�waitr �Error)r$ )r) r r r r% r' r( �do_wait@ s ���zproc_info.<locals>.do_waitr r r F)� clear_cache) r r �Processr � assertPidGone�as_dictr! r r �iter�gettersr+ )r r, �d�info�ns�fun�fun_namer$ r' )r) r r r r% r& r( � proc_info/ s6 �� ��r8 c @ s8 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd d � Z dd� Z dd� Z dd� Z dd� Z dd� Zdd� Zdd� Zd d � Zd d � Zd d!� Zd"d#� Zd$d%� Zd&d'� Zd(d)� Zd*d+� Zd,d-� Zd.d/� Zd0d1� Zd2d3� Z d4d5� Z d6d7� Z d8d9� Z d:d;� Z d<d=� Z!d>d?� Z"d@dA� Z#dBdC� Z$dDdE� Z%dFdG� Z&dHdI� Z'dJdK� Z(dLS )M�TestFetchAllProcessesz�Test which iterates over all running processes and performs some sanity checks against Process API's returned values. Uses a process pool to get info about all processes. c C s t �d� trt�� | _d S d S �NF)r � _set_debug� USE_PROC_POOL�multiprocessing�Pool�pool��selfr' r' r( �setUpf s �zTestFetchAllProcesses.setUpc C s* t �d� tr| j�� | j�� d S d S �NT)r r; r< r? � terminate�joinr@ r' r' r( �tearDownm s �z TestFetchAllProcesses.tearDownc s< ddl m� tr| j�� t�� �S � fdd�t�� D �}|S )Nr �r8 c s g | ]}� |��qS r' r' )�.0r rG r' r( � <listcomp>{ s z8TestFetchAllProcesses.iter_proc_info.<locals>.<listcomp>)� psutil.tests.test_process_allr8 r< r? �imap_unorderedr �pids)rA �lsr' rG r( �iter_proc_infos s z$TestFetchAllProcesses.iter_proc_infoc C s� g }| � � D ]a}|�� D ]Z\}}t| |�}z|||� W n9 tyU d}|d�||d t|�|�7 }|d7 }|dt�� � �7 }d�dd� |� � D ��d }|� |� Y q w |dd g d d i fvrf|sfJ |��q q|rr| � d �|���d S ) NzH ====================================================================== z+FAIL: name=test_{}, pid={}, ret={} info={} r zF----------------------------------------------------------------------� c s s � | ]}d | V qdS )z Nr' )rH �ir' r' r( � <genexpr>� s � z1TestFetchAllProcesses.test_all.<locals>.<genexpr>r � � ) rN �items�getattr� Exception�formatr# � traceback� format_excrE � splitlines�append�fail)rA �failuresr4 r �value�meth�sr' r' r( �test_all~ s8 �� � ���z TestFetchAllProcesses.test_allc C s* t |t�sJ �|D ] }t |t�sJ �q d S �N)r �listr" )rA �retr4 �partr' r' r( �cmdline� s �z TestFetchAllProcesses.cmdlinec C s� t |t�sJ �|�� |ksJ �|rYtr|�d�sd S tj�|�s$J |��tr[tj� |�r]t td�r_t td�raz t� |tj �s@J �W d S t yX tj�|�rRtsU� Y d S Y d S w d S d S d S d S d S )Nz.exe�access�X_OK)r r" �stripr �endswith�os�path�isabsr �isfile�hasattrrg rh �AssertionError�existsr �rA rd r4 r' r' r( �exe� s$ ���zTestFetchAllProcesses.exec C � t |t�sJ �|dks J �d S r �r �intrr r' r' r( r � � zTestFetchAllProcesses.pidc C s& t |t�sJ �|dks J �t|� d S r )r rv r8 rr r' r' r( r � s zTestFetchAllProcesses.ppidc C sB t |t�sJ �tr|st|d �rd S ts |s J t|���d S d S )Nr )r r" r r r r# rr r' r' r( r � s �zTestFetchAllProcesses.namec C s^ t |t�sJ �z|dksJ �W n ty# tr |d tjkr n� Y nw t�dt�|�� d S )Nr �statusz%Y %m %d %H:%M:%S) r �floatrp r r � STATUS_ZOMBIE�time�strftime� localtimerr r' r' r( � create_time� s �� z!TestFetchAllProcesses.create_timec C �4 t |�sJ �|D ]}t|t�sJ �|dksJ �qd S r �r r rv )rA rd r4 �uidr' r' r( �uids� � �zTestFetchAllProcesses.uidsc C s< t |�sJ �|D ]}t|t�sJ �tsts|dksJ �qd S r )r r rv r r )rA rd r4 �gidr' r' r( �gids� s ��zTestFetchAllProcesses.gidsc C s. t |t�sJ �|�� |ksJ �|�� sJ �d S rb )r r" ri rr r' r' r( �username� s z TestFetchAllProcesses.usernamec C s6 t |t�sJ �|s J |��|dksJ �|tv sJ �d S )N�?)r r" r rr r' r' r( rx � s z TestFetchAllProcesses.statusc C s< t |�sJ �|D ]}t|t�sJ �|dkr|dksJ �qd S )N���r r� )rA rd r4 �fieldr' r' r( � io_counters� s ��z!TestFetchAllProcesses.io_countersc C s� t r"t|jt�s J �t|jt�sJ �|jdksJ �|jdks J �d S tjtjtjtj g}t|t�s3J �|dks9J �||v s?J �d S r ) r r �ioclassrv r^ r �IOPRIO_VERYLOW� IOPRIO_LOW� IOPRIO_NORMAL� IOPRIO_HIGH)rA rd r4 �choicesr' r' r( �ionice� s � z TestFetchAllProcesses.ionicec C s: t |t�sJ �tr|dkrt|d �rd S |dksJ �d S )Nr r � )r rv r r rr r' r' r( � num_threads s z!TestFetchAllProcesses.num_threadsc C sn t |t�sJ �|D ]+}t|�sJ �|jdksJ �|jdks J �|jdks&J �|D ] }t |ttf�s3J �q(q d S r )r rc r �id� user_time� system_timerv ry )rA rd r4 �tr� r' r' r( �threads s ��z TestFetchAllProcesses.threadsc C r r )r r ry )rA rd r4 �nr' r' r( � cpu_times r� z TestFetchAllProcesses.cpu_timesc C �6 t |t�sJ �d| krdksJ |�� J |��d S )NrR g Y@�r ry rr r' r' r( � cpu_percent � (z!TestFetchAllProcesses.cpu_percentc C s^ t |t�sJ �tr|dkrd S |dksJ �t�� dkr!|dks!J �|ttt�� ��v s-J �d S )Nr� r r� )r rv r r � cpu_countrc �rangerr r' r' r( �cpu_num! s z TestFetchAllProcesses.cpu_numc C s| t |�sJ �|D ]}t|t�sJ �|dksJ �qtr:|j|jks"J �|j|jks*J �|j|j ks2J �|j |j ks<J �d S d S r ) r r rv r � peak_wset�wset�peak_paged_pool� paged_pool�peak_nonpaged_pool� nonpaged_pool� peak_pagefile�pagefile�rA rd r4 r^ r' r' r( � memory_info* s �z!TestFetchAllProcesses.memory_infoc C s� t |�sJ �t�� j}|jD ]%}t||�}t|t�s J �|dks"J �ts*t r+|dv r+q||ks3J |��qtr>|j |j ks@J �d S d S )Nr > �vms�data) r r �virtual_memory�total�_fieldsrU r rv r r �pss�uss)rA rd r4 r� r r^ r' r' r( �memory_full_info5 s �z&TestFetchAllProcesses.memory_full_infoc C s t |t�sJ �|D ]�}t |jt�sJ �t |jt�sJ �|j�� |jks%J �tr/|jdks.J �n6tr_t |j t�s9J �t |j t�sAJ �t |j t�sIJ �|j dksPJ �|j dv sWJ �|j dks^J �nt re|jseq t j�|j�spJ |��zt �|j�}W n ty� Y q w t�|j�s�J |��q d S )Nr� r > �a+�r+�a�r�w)r rc �fdrv rl r" ri r r �position�mode�flagsr rk rm �stat�FileNotFoundError�S_ISREG�st_mode)rA rd r4 �f�str' r' r( � open_filesE s0 ��z TestFetchAllProcesses.open_filesc C rt r ru rr r' r' r( �num_fds_ rw z TestFetchAllProcesses.num_fdsc C sb t � �$ t|�tt|��ksJ �|D ] }t|�sJ �t|� qW d � d S 1 s*w Y d S rb )r �len�setr r )rA rd r4 �connr' r' r( �net_connectionsc s �"�z%TestFetchAllProcesses.net_connectionsc C s� t |t�sJ �|�� |ksJ �|rUtj�|�sJ |��zt�|�}W n* tyL } z tr3t j � |�r3n|j t j kr:� W Y d }~d S W Y d }~d S d }~ww t�|j�sWJ �d S d S rb )r r" ri rk rl rm r� �OSErrorr r � _psplatform�is_permission_err�errno�ENOENT�S_ISDIRr� )rA rd r4 r� �errr' r' r( �cwdj s$ ����� zTestFetchAllProcesses.cwdc C r� )Nr �d r� rr r' r' r( �memory_percentz r� z$TestFetchAllProcesses.memory_percentc C s t |t�sJ �d S rb )r �boolrr r' r' r( � is_running~ s z TestFetchAllProcesses.is_runningc C sR t |t�sJ �|g ks J �ttt�� ��}|D ]}t |t�s J �||v s&J �qd S rb )r rc r� r r� rv )rA rd r4 �cpusr� r' r' r( � cpu_affinity� s �z"TestFetchAllProcesses.cpu_affinityc C sN t |ttd �f�s J �|d ur#tj�|�sJ |��tj�|�s%J |��d S d S rb )r r" �typerk rl rm rq rr r' r' r( �terminal� s �z TestFetchAllProcesses.terminalc C s� |D ]l}t |jt�s J �t |jt�sJ �t |jt�s J �|jD ]N}t||�}|dkrD|�d�r0q tr7|dkr7q t j� |j�sCJ |j��q |dkrQ|sPJ t |���q |dkr`t s_|s_J t |���q t |t �sgJ �|dksmJ �q qd S )Nrl )�[z anon_inode:�pvclock�addr�permsr )r r� r" r� rl r� rU � startswithr rk rm r# r rv )rA rd r4 �nt�fnamer^ r' r' r( � memory_maps� s, ���z!TestFetchAllProcesses.memory_mapsc C rt r ru rr r' r' r( � num_handles� rw z!TestFetchAllProcesses.num_handlesc C sl t |t�sJ �tr d| krdksJ |�� J |��d S dd� tt�D �}||v s,J �t |tj�s4J �d S )Ni��� c S s g | ] }|� d �rtt|��qS )�_PRIORITY_CLASS)rj rU r )rH �xr' r' r( rI � s ��z.TestFetchAllProcesses.nice.<locals>.<listcomp>)r rv r �dirr �enum�IntEnum)rA rd r4 � prioritiesr' r' r( �nice� s (� zTestFetchAllProcesses.nicec C r r r� r� r' r' r( �num_ctx_switches� r� z&TestFetchAllProcesses.num_ctx_switchesc C sB t |t�sJ �t|�dksJ �|d dksJ �|d dks J �d S )N� r r� r� )r �tupler� rr r' r' r( �rlimit� s z TestFetchAllProcesses.rlimitc C s@ t |t�sJ �|�� D ]\}}t |t�sJ �t |t�s J �q d S rb )r �dictrT r" )rA rd r4 �k�vr' r' r( �environ� s �z TestFetchAllProcesses.environN))�__name__� __module__� __qualname__�__doc__rB rF rN ra rf rs r r r r~ r� r� r� rx r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r' r' r' r( r9 ` sN r9 c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) � TestPidsRangea@ Given pid_exists() return value for a range of PIDs which may or may not exist, make sure that psutil.Process() and psutil.pids() agree with pid_exists(). This guarantees that the 3 APIs are all consistent with each other. See: https://github.com/giampaolo/psutil/issues/2359 XXX - Note about Windows: it turns out there are some "hidden" PIDs which are not returned by psutil.pids() and are also not revealed by taskmgr.exe and ProcessHacker, still they can be instantiated by psutil.Process() and queried. One of such PIDs is "conhost.exe". Running as_dict() for it reveals that some Process() APIs erroneously raise NoSuchProcess, so we know we have problem there. Let's ignore this for now, since it's quite a corner case (who even imagined hidden PIDs existed on Windows?). c C � t �d� d S r: �r r; r@ r' r' r( rB � � zTestPidsRange.setUpc C r� rC r� r@ r' r' r( rF � r� zTestPidsRange.tearDownc C sf dd� }dd� }t dd�D ]#}tr||�rq | j|d�� ||� W d � n1 s+w Y q d S )Nc S s� z t d| � d�d�}W n ty Y dS w |�$ |D ] }|�d�r7t|�� d �}|| k W d � S qtd��1 s?w Y d S )Nz/proc/z/status�rbFs Tgid:r� z'Tgid' line not found)�openr� r� rv �split� ValueError)r r� �line�tgidr' r' r( � is_linux_tid� s � ��z+TestPidsRange.test_it.<locals>.is_linux_tidc S s� d} t �| �}zD|r t �| � ts| t �� v s J �W d S W d S ts=t�t j�� t �| � W d � n1 s8w Y tsG| t �� vsJJ �W d S W d S t j t fya |d8 }|dkr_� Y nw q)N� Tr� r ) r � pid_existsr. r rL r r �raisesr r+ rp )r r� rq r' r' r( �check� s: ��� �������z$TestPidsRange.test_it.<locals>.checkr� i� )r )r� r �subTest)rA r r r r' r' r( �test_it� s ���zTestPidsRange.test_itN)r� r� r� r� rB rF r r' r' r' r( r� � s r� )"r� r� r� r= rk r� r{ rX r r r r r r r r r r r � psutil.testsr r r r r r r r r r r< r8 r9 r� r' r' r' r( �<module> sF 1 n tests/__pycache__/test_connections.cpython-310.pyc 0000644 00000036331 15030211765 0016206 0 ustar 00 o �h�R � @ sP d Z ddlZddlZddlZddlmZ ddlmZ ddlmZ ddlmZ ddlm Z ddl Z ddl m Z dd l m Z dd l m Z dd l mZ dd l mZ dd l mZ ddl mZ ddl mZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlm Z ddlm Z ddlm Z ddlm Z ddlm Z dd lm!Z! dd lm"Z" dd lm#Z# dd lm$Z$ dd lm%Z% e&ed!e'� �Z(d"d#� Z)e j*j+d$d%�G d&d'� d'e��Z,G d(d)� d)e,�Z-e j*j+d$d%�G d*d+� d+e,��Z.e j*j+d$d%�G d,d-� d-e,��Z/G d.d/� d/e,�Z0e j*j1ed0d1�G d2d3� d3e,��Z2G d4d5� d5e�Z3dS )6zFTests for psutil.net_connections() and Process.net_connections() APIs.� N)�closing)�AF_INET)�AF_INET6)� SOCK_DGRAM)� SOCK_STREAM)�FREEBSD)�LINUX)�MACOS)�NETBSD)�OPENBSD)�POSIX)�SUNOS)�WINDOWS)� supports_ipv6)�AF_UNIX)�HAS_NET_CONNECTIONS_UNIX)� SKIP_SYSCONS)�PsutilTestCase)� bind_socket)�bind_unix_socket)�check_connection_ntuple)�create_sockets)�filter_proc_net_connections)�pytest)� reap_children)�retry_on_failure)�skip_on_access_denied)�tcp_socketpair)�unix_socketpair)� wait_for_file�SOCK_SEQPACKETc C s$ t �� j| d�}| dv rt|�S |S )N��kind> �all�unix)�psutil�Process�net_connectionsr )r" �cons� r) �P/usr/local/CyberCP/lib/python3.10/site-packages/psutil/tests/test_connections.py�this_proc_net_connections1 s r+ �serial)�namec @ s& e Zd Zdd� Zdd� Zd dd�ZdS ) �ConnectionTestCasec C � t dd�g ks J �d S �Nr# r! �r+ ��selfr) r) r* �setUp: s zConnectionTestCase.setUpc C r/ r0 r1 r2 r) r) r* �tearDown= s zConnectionTestCase.tearDownr# c s` zt j|d�}W n t jy trY dS � w � fdd�|D �}|�� |�� ||ks.J �dS )z�Given a process PID and its list of connections compare those against system-wide connections retrieved via psutil.net_connections. r! Nc s" g | ] }|j � kr|d d� �qS )N�����pid��.0�cr7 r) r* � <listcomp>P s " zBConnectionTestCase.compare_procsys_connections.<locals>.<listcomp>)r% r' � AccessDeniedr �sort)r3 r8 � proc_consr" �sys_consr) r7 r* �compare_procsys_connectionsA s �z.ConnectionTestCase.compare_procsys_connectionsN)r# )�__name__� __module__� __qualname__r4 r5 rA r) r) r) r* r. 8 s r. c @ s4 e Zd Zejjedd�dd� �Zdd� Zdd� Z d S ) �TestBasicOperations� requires root��reasonc C sF t � � tjdd�D ]}t|� q W d � d S 1 s w Y d S r0 )r r% r' r �r3 �connr) r) r* � test_systemW s �"�z TestBasicOperations.test_systemc C sD t � � tdd�D ]}t|� q W d � d S 1 sw Y d S r0 )r r+ r rI r) r) r* � test_process] s �"�z TestBasicOperations.test_processc C sp t �t�� tdd� W d � n1 sw Y t �t�� tjdd� W d � d S 1 s1w Y d S )Nz???r! )r �raises� ValueErrorr+ r% r' r2 r) r) r* �test_invalid_kindb s � "�z%TestBasicOperations.test_invalid_kindN) rB rC rD r �mark�skipifr rK rL rO r) r) r) r* rE V s rE c @ s� e Zd ZdZdd� Zdd� Zdd� Zejj e � dd �d d � �Z d d � Z ejj e � dd �dd� �Z ejj e dd �dd� �Zejj e dd �dd� �ZdS )�TestUnconnectedSocketsz;Tests sockets which are open but not connected to anything.c C sl t dd�}dd� |D �}tstr||�� S t|�dks J �|d jdkr2||�� j|�� ks2J �|d S )Nr# r! c S s i | ]}|j |�qS r) )�fdr9 r) r) r* � <dictcomp>o s z=TestUnconnectedSockets.get_conn_from_sock.<locals>.<dictcomp>� r r6 )r+ r r �fileno�lenrS )r3 �sockr( �smapr) r) r* �get_conn_from_sockm s z)TestUnconnectedSockets.get_conn_from_sockc C s� | � |�}t|� |jdkr|j|�� ksJ �|j|jks J �|j|�tjtj �ks,J �|� � }|s;t |t �r;|� � }|jtkrF|dd� }|j|ksMJ �|jtkrctrctdd�}| jt�� |dd� |S )z�Given a socket, makes sure it matches the one obtained via psutil. It assumes this process created one connection only (the one supposed to be checked). r6 N� r# r! )rZ r rS rV �family�type� getsockopt�socket� SOL_SOCKET�SO_TYPE� getsockname� isinstance�bytes�decoder �laddrr r r+ rA �os�getpid)r3 rX rJ rf r( r) r) r* � check_socketz s z#TestUnconnectedSockets.check_socketc C �d d}t ttt|d��� }| �|�}|jdksJ �|jtjks J �W d � d S 1 s+w Y d S �N�� 127.0.0.1r ��addrr) ) r r r r ri �raddr�statusr% � CONN_LISTEN�r3 ro rX rJ r) r) r* � test_tcp_v4� � "�z"TestUnconnectedSockets.test_tcp_v4zIPv6 not supportedrG c C rj �N)�::1r rn r) ) r r r r ri rp rq r% rr rs r) r) r* � test_tcp_v6� � "�z"TestUnconnectedSockets.test_tcp_v6c C rj rk ) r r r r ri rp rq r% � CONN_NONErs r) r) r* � test_udp_v4� ru z"TestUnconnectedSockets.test_udp_v4c C rj rv ) r r r r ri rp rq r% rz rs r) r) r* � test_udp_v6� ry z"TestUnconnectedSockets.test_udp_v6� POSIX onlyc C �f | � � }tt|td��� }| �|�}|jdksJ �|jtjks!J �W d � d S 1 s,w Y d S �N)r] � � � get_testfnr r r ri rp rq r% rz �r3 �testfnrX rJ r) r) r* � test_unix_tcp� � "�z$TestUnconnectedSockets.test_unix_tcpc C r~ r r� r� r) r) r* � test_unix_udp� r� z$TestUnconnectedSockets.test_unix_udpN)rB rC rD �__doc__rZ ri rt r rP rQ r rx r{ r| r r� r� r) r) r) r* rR i s rR c @ sB e Zd ZdZejjedd�dd� �Zejje dd�dd� �Z d S ) �TestConnectedSocketzFTest socket pairs which are actually connected to each other. zunreliable on SUONSrG c C s� d}t dd�g ks J �tt|d�\}}z,t dd�}t|�dks!J �|d jtjks+J �|d jtjks5J �W |�� |�� d S |�� |�� w )Nrl �tcp4r! rn r[ r rU )r+ r r rW rq r% �CONN_ESTABLISHED�close)r3 ro �server�clientr( r) r) r* �test_tcp� s � z TestConnectedSocket.test_tcpr} c C s* | � � }t|�\}}z�tdd�}|d jr |d jr J |��|d jr,|d jr,J |��ts0tr7dd� |D �}t|�dks?J �tsGtsGt sGt rh|d jdksPJ �|d jdksYJ �||d jpc|d jksgJ �n|d jpq|d j|ks�J �W |� � |� � d S W |� � |� � d S |� � |� � w ) Nr$ r! r rU c S s g | ] }|j d kr|�qS )z /var/run/log)rp r9 r) r) r* r< � � z1TestConnectedSocket.test_unix.<locals>.<listcomp>r[ r� ) r� r r+ rf rp r r rW r r r r� )r3 r� r� r� r( r) r) r* � test_unix� s, � � � z TestConnectedSocket.test_unixN) rB rC rD r� r rP rQ r r� r r� r) r) r) r* r� � s r� c @ s. e Zd Zdd� Zeed�dd� �Zdd� ZdS ) � TestFiltersc C s� dd� }t � �n |dtttgtttg� |dttgttg� |dtgttg� |dttgtg� |dtgtg� |dtgtg� |d ttgtg� |d tgtg� |d tgtg� trm|d tgtttg� W d � d S W d � d S 1 sxw Y d S ) Nc S sf t | d�D ]}|j|v sJ �|j|v sJ �qts/tj| d�D ]}|j|v s'J �|j|v s.J �q d S d S )Nr! )r+ r\ r] r r% r' )r" �families�typesrJ r) r) r* �check s �z'TestFilters.test_filters.<locals>.checkr# �inet�inet4�tcpr� �tcp6�udp�udp4�udp6r$ )r r r r r r r r )r3 r� r) r) r* � test_filters s0 ���"�zTestFilters.test_filters)�only_ifc s� t � � fdd�}t�d�}t�d�}tj�� jt�� d��}|jt t �d|d�}|jt t �d|d�}|jt t �d|d�}|jt t �d|d�}� � |�} t t|d d ��} � � |�} t t|d d ��} t� r� � |�} t t|d d ��}� � |�}t t|d d ��}nd } d }d }d }t�� �� D ]d}|�� }t|�d ks�J �|D ]S}|j| jkr�|||t t| d tjd � q�|j| jkr�|||t t| d tjd� q�|jt| dd �kr�|||t t|d tjd� q�|jt|dd �kr�|||t t|d tjd� q�q�d S )Nc s� d}t |� |j|ks J �|j|ksJ �|j|ksJ �|j|ks"J �|j|ks)J �|D ]} | j| d�} | |v r>| g ks=J �q+| g ksDJ �q+trQ� �| j |g� d S d S )N) r# r� r� �inet6r� r� r� r� r� r� r! ) r r\ r] rf rp rq r' r rA r8 ) �procrJ r\ r] rf rp rq �kinds� all_kindsr" r( r2 r) r* � check_conn# s �z+TestFilters.test_combos.<locals>.check_conna4 import socket, time s = socket.socket({family}, socket.SOCK_STREAM) s.bind(('{addr}', 0)) s.listen(5) with open('{testfn}', 'w') as f: f.write(str(s.getsockname()[:2])) [time.sleep(0.1) for x in range(100)] a import socket, time s = socket.socket({family}, socket.SOCK_DGRAM) s.bind(('{addr}', 0)) with open('{testfn}', 'w') as f: f.write(str(s.getsockname()[:2])) [time.sleep(0.1) for x in range(100)] )�dirrm )r\ ro r� rw T)�deleterU r) )r# r� r� r� r� )r# r� r� r� r� r8 )r# r� r� r� r� )r# r� r� r� r� )r �textwrap�dedentrg �path�basenamer� �getcwd�format�intr r �pyrun�evalr r r% r& �childrenr' rW r8 r rr r rz �getattr)r3 r� � tcp_template� udp_template�testfile� tcp4_template� udp4_template� tcp6_template� udp6_template� tcp4_proc� tcp4_addr� udp4_proc� udp4_addr� tcp6_proc� tcp6_addr� udp6_proc� udp6_addr�pr( rJ r) r2 r* � test_combos s� � � � � � � � ����zTestFilters.test_combosc C s t � ��s tdd�}t|�t� rdndksJ �|D ]}|jtthv s$J �|jtks+J �qtdd�}t|�dks9J �|d jtksBJ �|d jtksKJ �t� rmtdd�}t|�dks[J �|d jtksdJ �|d jtksmJ �tdd�}t|�t� rzdndksJ �|D ]}|jtthv s�J �|jt ks�J �q�td d�}t|�dks�J �|d jtks�J �|d jt ks�J �t� r�td d�}t|�dks�J �|d jtks�J �|d jt ks�J �td d�}t|�t� r�d ndks�J �|D ]}|jtthv s�J �|jtt hv s�J �q�t� �r(td d�}t|�dk�sJ �|D ]}|jtk�s J �|jtt hv �s&J ��qt �rWt �s_t �sgtdd�}t|�dk�s?J �|D ]-}|jt k�sKJ �|jtt hv �sUJ ��qAW d � d S W d � d S W d � d S W d � d S 1 �s{w Y d S )Nr� r! r[ rU r� r r� r� r� r� r� � r� r$ � )r r+ rW r r\ r r r] r r r r r r )r3 r( rJ r) r) r* � test_count� sl ��2�2�5$�zTestFilters.test_countN)rB rC rD r� r r r� r� r) r) r) r* r� � s r� rF rG c @ s& e Zd ZdZdd� Ze� dd� �ZdS )�TestSystemWideConnectionsz Tests for net_connections().c C s� dd� }t � �: ddlm} |�� D ]&\}}|dkr ts q|\}}t�|�}t|�tt|��ks2J �||||� qW d � d S 1 sDw Y d S )Nc S s< | D ]}|j |v s J �|j tkr|j|v sJ �t|� qd S )N)r\ r r] r )r( r� �types_rJ r) r) r* r� � s �z0TestSystemWideConnections.test_it.<locals>.checkr )� conn_tmapr$ ) r �psutil._commonr� �itemsr r% r' rW �set)r3 r� r� r" �groupsr� r� r( r) r) r* �test_it� s �"�z!TestSystemWideConnections.test_itc s� t � � }t|�}W d � n1 sw Y g �d}g }t|�D ] }| �� }|�|� t�d|� d��}| �|�}��|j� q!|D ]}t |� qC�fdd�t j dd�D �} �D ]!� t� fdd�| D ��|ksjJ �t � � �} t| � d��|kszJ �qYd S ) N� z� import time, os from psutil.tests import create_sockets with create_sockets(): with open(r'z�', 'w') as f: f.write("hello") [time.sleep(0.1) for x in range(100)] c s g | ] }|j � v r|�qS r) r7 �r: �x)�pidsr) r* r< s zFTestSystemWideConnections.test_multi_sockets_procs.<locals>.<listcomp>r# r! c s g | ] }|j � kr|�qS r) r7 r� r7 r) r* r< r� ) r rW �ranger� �appendr� r� r� r8 r r% r' r& ) r3 �socks�expected�times�fnames�_�fname�src�sproc�sysconsr� r) )r8 r� r* �test_multi_sockets_procs� s0 � � � �z2TestSystemWideConnections.test_multi_sockets_procsN)rB rC rD r� r� r r� r) r) r) r* r� � s r� c @ s e Zd Zdd� ZdS )�TestMiscc C s� g }g }t t�D ].}|�d�r6tt|�}t|�}|�� s J |��t|vs&J �||vs,J �|�|� |�|� qtr?tj tj t rFtj d S d S )N�CONN_) r� r% � startswithr� �str�isupperr� r � CONN_IDLE� CONN_BOUNDr �CONN_DELETE_TCB)r3 �ints�strsr- �num�str_r) r) r* � test_net_connection_constants& s$ � �z&TestMisc.test_net_connection_constantsN)rB rC rD r� r) r) r) r* r� % s r� )4r� rg r_ r� � contextlibr r r r r r% r r r r r r r r r� r � psutil.testsr r r r r r r r r r r r r r r r r� �objectr r+ rP � xdist_groupr. rE rR r� r� rQ r� r� r) r) r) r* �<module> sb ]7 g? tests/__pycache__/test_memleaks.cpython-310.pyc 0000644 00000053443 15030211765 0015465 0 ustar 00 o �h; � @ s� d Z ddlZddlZddlZddlZddlZddlmZ ddlmZ ddlmZ ddlm Z ddlm Z ddlm Z dd l m Z dd l mZ dd l mZ dd l mZ dd l mZ ddl mZ ddl mZ ddl mZ ddl mZ ddl mZ ddl mZ ddl mZ ddl mZ ddl mZ ddl mZ ddl m Z ddl m Z ddl m Z ddl m Z dd l m Z dd l m!Z! ej"j#Z#e�$� Z%d Z&d d � Z'G d!d"� d"e�Z(G d#d$� d$e(�Z)e j*j+e d%d&�G d'd(� d(e��Z,G d)d*� d*e�Z-dS )+a� Tests for detecting function memory leaks (typically the ones implemented in C). It does so by calling a function many times and checking whether process memory usage keeps increasing between calls or over time. Note that this may produce false positives (especially on Windows for some reason). PyPy appears to be completely unstable for this framework, probably because of how its JIT handles memory, so tests are skipped. � N)�LINUX)�MACOS)�OPENBSD)�POSIX)�SUNOS)�WINDOWS)�HAS_CPU_AFFINITY)� HAS_CPU_FREQ)� HAS_ENVIRON)� HAS_IONICE)�HAS_MEMORY_MAPS)�HAS_NET_IO_COUNTERS)�HAS_PROC_CPU_NUM)�HAS_PROC_IO_COUNTERS)� HAS_RLIMIT)�HAS_SENSORS_BATTERY)�HAS_SENSORS_FANS)�HAS_SENSORS_TEMPERATURES)�TestMemoryLeak)�create_sockets)� get_testfn)�process_namespace)�pytest)�skip_on_access_denied)�spawn_testproc)�system_namespace)� terminate� c C s dd� } | S )zsDecorator for those Linux functions which are implemented in pure Python, and which we want to run faster. c s t �� �� fdd��}|S )Nc sV t r | jj}zt| j_� | g|�R i |��W || j_S || j_w � | g|�R i |��S �N)r � __class__�times� FEW_TIMES)�self�args�kwargs�before��fun� �M/usr/local/CyberCP/lib/python3.10/site-packages/psutil/tests/test_memleaks.py�wrapper@ s z5fewtimes_if_linux.<locals>.decorator.<locals>.wrapper)� functools�wraps)r' r* r( r& r) � decorator? s z$fewtimes_if_linux.<locals>.decoratorr( )r- r( r( r) �fewtimes_if_linux: s r. c @ s0 e Zd ZdZeZdd� Ze� dd� �Ze� dd� �Z e� dd � �Z e� d d � �Z e j je d d �e� dd� ��Ze j je d d �e� dd� ��Ze� dd� �Zdd� Zdd� Ze j je dd �dd� �Ze j je dd �dd � �Ze j je dd �e� d d � ��Ze j jed d �d d!� �Ze� d"d#� �Ze� e e d$�d%d&� ��Z e j je d'd �d(d)� �Z e j je d d �e� d*d+� ��Z!e� d,d-� �Z"e� e e d$�d.d/� ��Z#e� d0d1� �Z$e� e j je% dd �d2d3� ��Z&e� d4d5� �Z'e� d6d7� �Z(e j je d d �e� d8d9� ��Z)d:d;� Z*e� d<d=� �Z+e j je, dd �d>d?� �Z-e j je, dd �d@dA� �Z.e� dBdC� �Z/e j je0 dd �e� dDdE� ��Z1e j je2 dFd �e j je3 dd �dGdH� ��Z4e j je2 dFd �e j je3 dd �dIdJ� ��Z5e� e j je dKd �dLdM� ��Z6e j je7 dd �dNdO� �Z8e j je d'd �dPdQ� �Z9dRS )S�TestProcessObjectLeaksz$Test leaks of Process class methods.c C s t d �}|�| |j|j � d S r )r �test_class_coverage�getters�setters�r" �nsr( r( r) � test_coverage[ � z$TestProcessObjectLeaks.test_coveragec C � | � | jj� d S r )�execute�proc�name�r" r( r( r) � test_name_ � z TestProcessObjectLeaks.test_namec C r7 r )r8 r9 �cmdliner; r( r( r) � test_cmdlinec r= z#TestProcessObjectLeaks.test_cmdlinec C r7 r )r8 r9 �exer; r( r( r) �test_exeg r= z TestProcessObjectLeaks.test_exec C r7 r )r8 r9 �ppidr; r( r( r) � test_ppidk r= z TestProcessObjectLeaks.test_ppidz POSIX only��reasonc C r7 r )r8 r9 �uidsr; r( r( r) � test_uidso � z TestProcessObjectLeaks.test_uidsc C r7 r )r8 r9 �gidsr; r( r( r) � test_gidst rH z TestProcessObjectLeaks.test_gidsc C r7 r )r8 r9 �statusr; r( r( r) � test_statusy r= z"TestProcessObjectLeaks.test_statusc C r7 r )r8 r9 �nicer; r( r( r) � test_nice} � z TestProcessObjectLeaks.test_nicec s t �� � ��� �fdd�� d S )Nc � �j �� �S r )r9 rM r( ��nicenessr" r( r) �<lambda>� � z6TestProcessObjectLeaks.test_nice_set.<locals>.<lambda>)�thisprocrM r8 r; r( rQ r) � test_nice_set� r6 z$TestProcessObjectLeaks.test_nice_set� not supportedc C r7 r )r8 r9 �ionicer; r( r( r) � test_ionice� r= z"TestProcessObjectLeaks.test_ionicec s\ t rt�� �� �� �fdd�� d S � �� fdd�� t�tjt� � dd�}� � t |� d S )Nc s � j ���S r )r9 rX r( �r" �valuer( r) rS � rT z8TestProcessObjectLeaks.test_ionice_set.<locals>.<lambda>c � � j �tj�S r )r9 rX �psutil�IOPRIO_CLASS_NONEr( r; r( r) rS � � ���r ) r rU rX r8 r+ �partial�cext�proc_ioprio_set�os�getpid� execute_w_exc�OSError�r" r' r( rZ r) �test_ionice_set� s z&TestProcessObjectLeaks.test_ionice_setc C r7 r )r8 r9 � io_countersr; r( r( r) �test_io_counters� rH z'TestProcessObjectLeaks.test_io_counterszworthless on POSIXc C s t �� �� | �| jj� d S r )r] �Process�usernamer8 r9 r; r( r( r) � test_username� s z$TestProcessObjectLeaks.test_usernamec C r7 r )r8 r9 � create_timer; r( r( r) �test_create_time� r= z'TestProcessObjectLeaks.test_create_time)�only_ifc C r7 r )r8 r9 � num_threadsr; r( r( r) �test_num_threads� rH z'TestProcessObjectLeaks.test_num_threads� WINDOWS onlyc C r7 r )r8 r9 � num_handlesr; r( r( r) �test_num_handles� r= z'TestProcessObjectLeaks.test_num_handlesc C r7 r )r8 r9 �num_fdsr; r( r( r) � test_num_fds� rH z#TestProcessObjectLeaks.test_num_fdsc C r7 r )r8 r9 �num_ctx_switchesr; r( r( r) �test_num_ctx_switches� r= z,TestProcessObjectLeaks.test_num_ctx_switchesc C r7 r )r8 r9 �threadsr; r( r( r) � test_threads� rH z#TestProcessObjectLeaks.test_threadsc C r7 r )r8 r9 � cpu_timesr; r( r( r) �test_cpu_times� r= z%TestProcessObjectLeaks.test_cpu_timesc C r7 r )r8 r9 �cpu_numr; r( r( r) � test_cpu_num� rH z#TestProcessObjectLeaks.test_cpu_numc C r7 r )r8 r9 � memory_infor; r( r( r) �test_memory_info� r= z'TestProcessObjectLeaks.test_memory_infoc C r7 r )r8 r9 �memory_full_infor; r( r( r) �test_memory_full_info� r= z,TestProcessObjectLeaks.test_memory_full_infoc C r7 r )r8 r9 �terminalr; r( r( r) � test_terminal� rH z$TestProcessObjectLeaks.test_terminalc C s$ t rtn| j}| j| jj|d� d S )N�r )r r! r r8 r9 �resume�r" r r( r( r) � test_resume� � z"TestProcessObjectLeaks.test_resumec C r7 r )r8 r9 �cwdr; r( r( r) �test_cwd� r= z TestProcessObjectLeaks.test_cwdc C r7 r )r8 r9 � cpu_affinityr; r( r( r) �test_cpu_affinity� r= z(TestProcessObjectLeaks.test_cpu_affinityc s4 t �� � ��� �fdd�� ��t�fdd�� d S )Nc rP r �r9 r� r( ��affinityr" r( r) rS � rT z>TestProcessObjectLeaks.test_cpu_affinity_set.<locals>.<lambda>c s � j �dg�S �Nr` r� r( r; r( r) rS � r_ )rU r� r8 rf � ValueErrorr; r( r� r) �test_cpu_affinity_set� s z,TestProcessObjectLeaks.test_cpu_affinity_setc C s@ t t� d�� | �| jj� W d � d S 1 sw Y d S )N�w)�openr r8 r9 � open_filesr; r( r( r) �test_open_files� s "�z&TestProcessObjectLeaks.test_open_filesc C r7 r )r8 r9 � memory_mapsr; r( r( r) �test_memory_maps� rH z'TestProcessObjectLeaks.test_memory_mapsz LINUX onlyc s � � � fdd�� d S )Nc r\ r �r9 �rlimitr] � RLIMIT_NOFILEr( r; r( r) rS � r_ z4TestProcessObjectLeaks.test_rlimit.<locals>.<lambda>�r8 r; r( r; r) � test_rlimit� s z"TestProcessObjectLeaks.test_rlimitc s< t �tj�� ��� �fdd�� ��ttf�fdd�� d S )Nc s �j �tj� �S r r� r( ��limitr" r( r) rS � s z8TestProcessObjectLeaks.test_rlimit_set.<locals>.<lambda>c s � j �d�S r� )r9 r� r( r; r( r) rS � rT )rU r� r] r� r8 rf rg r� r; r( r� r) �test_rlimit_set� s z&TestProcessObjectLeaks.test_rlimit_setzworthless on WINDOWSc sL t � � trdnd� ��� �fdd�� W d � d S 1 s w Y d S )N�inet�allc rP r )r9 �net_connectionsr( ��kindr" r( r) rS rT z=TestProcessObjectLeaks.test_net_connections.<locals>.<lambda>)r r r8 r; r( r� r) �test_net_connections� s "�z+TestProcessObjectLeaks.test_net_connectionsc C r7 r )r8 r9 �environr; r( r( r) � test_environ r= z#TestProcessObjectLeaks.test_environc C � | � dd� � d S )Nc S � t �t�� �S r )rb � proc_inford re r( r( r( r) rS r_ z7TestProcessObjectLeaks.test_proc_info.<locals>.<lambda>r� r; r( r( r) �test_proc_info r= z%TestProcessObjectLeaks.test_proc_infoN):�__name__� __module__� __qualname__�__doc__rU r9 r5 r. r< r? rA rC r �mark�skipifr rG rJ rL rN rV r rY ri r rk rn rp r r rs r rv rx rz r| r~ r r� r� r� r� r� r� r r� r� r� r r� r r r� r� r� r r� r� r( r( r( r) r/ V s� r/ c sx e Zd ZdZe� fdd��Ze� fdd��Zdd� Zer8dd � Z d d � Z d d � Z dd� Z dd� Z dd� Z� ZS � ZS )�TestTerminatedProcessLeaksz�Repeat the tests above looking for leaks occurring when dealing with terminated processes raising NoSuchProcess exception. The C functions are still invoked but will follow different code paths. We'll check those code paths. c s: t � �� t� | _t�| jj�| _| j�� | j� � d S r ) �super� setUpClassr �subpr] rl �pidr9 �kill�wait��cls�r r( r) r� s z%TestTerminatedProcessLeaks.setUpClassc s t � �� t| j� d S r )r� � tearDownClassr r� r� r� r( r) r� s z(TestTerminatedProcessLeaks.tearDownClassc C s$ z|� W d S t jy Y d S w r )r] � NoSuchProcessrh r( r( r) �call s �z TestTerminatedProcessLeaks.callc C r7 r )r8 r9 r� r; r( r( r) � test_kill' rO z$TestTerminatedProcessLeaks.test_killc C r7 r )r8 r9 r r; r( r( r) �test_terminate* rO z)TestTerminatedProcessLeaks.test_terminatec C r7 r )r8 r9 �suspendr; r( r( r) � test_suspend- rO z'TestTerminatedProcessLeaks.test_suspendc C r7 r )r8 r9 r� r; r( r( r) r� 0 rO z&TestTerminatedProcessLeaks.test_resumec C r7 r )r8 r9 r� r; r( r( r) � test_wait3 rO z$TestTerminatedProcessLeaks.test_waitc s � fdd�}� � |� d S )Nc s& zt �� jj�W S ty Y d S w r )rb r� r9 r� �ProcessLookupErrorr( r; r( r) r� 8 s �z7TestTerminatedProcessLeaks.test_proc_info.<locals>.callr� )r" r� r( r; r) r� 6 s z)TestTerminatedProcessLeaks.test_proc_info)r� r� r� r� � classmethodr� r� r� r r� r� r� r� r� r� � __classcell__r( r( r� r) r� s �r� rt rD c @ s e Zd Zdd� Zdd� ZdS )� TestProcessDualImplementationc C r� )Nc S � t jt�� dd�S )NT��use_peb�rb � proc_cmdlinerd re r( r( r( r) rS D � zETestProcessDualImplementation.test_cmdline_peb_true.<locals>.<lambda>r� r; r( r( r) �test_cmdline_peb_trueC rO z3TestProcessDualImplementation.test_cmdline_peb_truec C r� )Nc S r� )NFr� r� r( r( r( r) rS G r� zFTestProcessDualImplementation.test_cmdline_peb_false.<locals>.<lambda>r� r; r( r( r) �test_cmdline_peb_falseF rO z4TestProcessDualImplementation.test_cmdline_peb_falseN)r� r� r� r� r� r( r( r( r) r� A s r� c @ s0 e Zd ZdZdd� Ze� dd� �Ze� dd� �Ze� dd � �Ze� d d � �Z e� d d � �Z e� e j j eo9e�� dkdd�e j j e dd�dd� ���Ze j j e dd�dd� �Zdd� Ze j j edd�dd� �Zd d � Zd d � Zd d!� Ze j j eoe j � d"� d#d�e� d$d%� ��Z e� d&d'� �Z e� e j j e! dd�d(d)� ��Z"e� e j j eo�e �#� d*kd+d�d,d-� ��Z$d.d/� Z%d0d1� Z&e� e j j e' dd�d2d3� ��Z(e� e j j e) dd�d4d5� ��Z*e� e j j e+ dd�d6d7� ��Z,e� d8d9� �Z-d:d;� Z.d<d=� Z/e�rd>d?� Z0d@dA� Z1dBdC� Z2dDdE� Z3dFdG� Z4dHS dHS )I�TestModuleFunctionsLeaksz&Test leaks of psutil module functions.c C s t � }|�| |j� d S r )r r0 r� r3 r( r( r) r5 R s z&TestModuleFunctionsLeaks.test_coveragec C r� )Nc S � t jdd�S )NT��logical�r] � cpu_countr( r( r( r) rS Z rT z9TestModuleFunctionsLeaks.test_cpu_count.<locals>.<lambda>r� r; r( r( r) �test_cpu_countX r= z'TestModuleFunctionsLeaks.test_cpu_countc C r� )Nc S r� )NFr� r� r( r( r( r) rS ^ rT z?TestModuleFunctionsLeaks.test_cpu_count_cores.<locals>.<lambda>r� r; r( r( r) �test_cpu_count_cores\ r= z-TestModuleFunctionsLeaks.test_cpu_count_coresc C � | � tj� d S r )r8 r] r} r; r( r( r) r~ ` � z'TestModuleFunctionsLeaks.test_cpu_timesc C r� )Nc S r� )NT)�percpu)r] r} r( r( r( r) rS f rT z=TestModuleFunctionsLeaks.test_per_cpu_times.<locals>.<lambda>r� r; r( r( r) �test_per_cpu_timesd r= z+TestModuleFunctionsLeaks.test_per_cpu_timesc C r� r )r8 r] � cpu_statsr; r( r( r) �test_cpu_statsh r� z'TestModuleFunctionsLeaks.test_cpu_stats�arm64zskipped due to #1892rD rW c C r� r )r8 r] �cpu_freqr; r( r( r) � test_cpu_freql s z&TestModuleFunctionsLeaks.test_cpu_freqrt c C s t �� | �t j� d S r )r] � getloadavgr8 r; r( r( r) �test_getloadavgu s z(TestModuleFunctionsLeaks.test_getloadavgc C r� r )r8 r] �virtual_memoryr; r( r( r) �test_virtual_memory| � z,TestModuleFunctionsLeaks.test_virtual_memoryz&worthless on SUNOS (uses a subprocess)c C r� r )r8 r] � swap_memoryr; r( r( r) �test_swap_memory� r� z)TestModuleFunctionsLeaks.test_swap_memoryc C �$ t rtn| j}| jdd� |d� d S )Nc S r� r )r] � pid_existsrd re r( r( r( r) rS � r_ z:TestModuleFunctionsLeaks.test_pid_exists.<locals>.<lambda>r� �r r! r r8 r� r( r( r) �test_pid_exists� r� z(TestModuleFunctionsLeaks.test_pid_existsc C r� )Nc S � t �d�S )N�.)r] � disk_usager( r( r( r) rS � � z:TestModuleFunctionsLeaks.test_disk_usage.<locals>.<lambda>r� r� r� r( r( r) �test_disk_usage� r� z(TestModuleFunctionsLeaks.test_disk_usagec C r� r )r8 r] �disk_partitionsr; r( r( r) �test_disk_partitions� r� z-TestModuleFunctionsLeaks.test_disk_partitionsz/proc/diskstatsz3/proc/diskstats not available on this Linux versionc C r� )Nc S r� �NF)�nowrap)r] �disk_io_countersr( r( r( r) rS � rT z@TestModuleFunctionsLeaks.test_disk_io_counters.<locals>.<lambda>r� r; r( r( r) �test_disk_io_counters� s z.TestModuleFunctionsLeaks.test_disk_io_countersc C r� r )r8 r] �pidsr; r( r( r) � test_pids� r� z"TestModuleFunctionsLeaks.test_pidsc C r� )Nc S r� r� )r] �net_io_countersr( r( r( r) rS � rT z?TestModuleFunctionsLeaks.test_net_io_counters.<locals>.<lambda>r� r; r( r( r) �test_net_io_counters� rH z-TestModuleFunctionsLeaks.test_net_io_countersr zneed root accessc C sF t jdd� t� � | �dd� � W d � d S 1 s w Y d S )Nr� �r� c S r� )Nr� r� )r] r� r( r( r( r) rS � rT z?TestModuleFunctionsLeaks.test_net_connections.<locals>.<lambda>)r] r� r r8 r; r( r( r) r� � s "�z-TestModuleFunctionsLeaks.test_net_connectionsc C s" t rdn| j}| jtj|d� d S )Ni @ )� tolerance)r r r8 r] � net_if_addrs)r" r r( r( r) �test_net_if_addrs� s z*TestModuleFunctionsLeaks.test_net_if_addrsc C r� r )r8 r] � net_if_statsr; r( r( r) �test_net_if_stats� r� z*TestModuleFunctionsLeaks.test_net_if_statsc C r� r )r8 r] �sensors_batteryr; r( r( r) �test_sensors_battery� � z-TestModuleFunctionsLeaks.test_sensors_batteryc C r� r )r8 r] �sensors_temperaturesr; r( r( r) �test_sensors_temperatures� r z2TestModuleFunctionsLeaks.test_sensors_temperaturesc C r� r )r8 r] � sensors_fansr; r( r( r) �test_sensors_fans� r z*TestModuleFunctionsLeaks.test_sensors_fansc C r� r )r8 r] � boot_timer; r( r( r) �test_boot_time� r� z'TestModuleFunctionsLeaks.test_boot_timec C r� r )r8 r] �usersr; r( r( r) � test_users� r� z#TestModuleFunctionsLeaks.test_usersc C r� )Nc S r� )NF)r] � _set_debugr( r( r( r) rS � r� z9TestModuleFunctionsLeaks.test_set_debug.<locals>.<lambda>r� r; r( r( r) �test_set_debug� rO z'TestModuleFunctionsLeaks.test_set_debugc C r� r )r8 rb �winservice_enumerater; r( r( r) �test_win_service_iter� r� z.TestModuleFunctionsLeaks.test_win_service_iterc C s d S r r( r; r( r( r) �test_win_service_get� s z-TestModuleFunctionsLeaks.test_win_service_getc �& t t�� ��� � | �� fdd�� d S )Nc � t �� �S r )rb �winservice_query_configr( �r: r( r) rS � r� zFTestModuleFunctionsLeaks.test_win_service_get_config.<locals>.<lambda>��nextr] �win_service_iterr: r8 r; r( r r) �test_win_service_get_config� � z4TestModuleFunctionsLeaks.test_win_service_get_configc r )Nc r r )rb �winservice_query_statusr( r r( r) rS � r� zFTestModuleFunctionsLeaks.test_win_service_get_status.<locals>.<lambda>r r; r( r r) �test_win_service_get_status� r z4TestModuleFunctionsLeaks.test_win_service_get_statusc r )Nc r r )rb �winservice_query_descrr( r r( r) rS � r� zKTestModuleFunctionsLeaks.test_win_service_get_description.<locals>.<lambda>r r; r( r r) � test_win_service_get_description� r z9TestModuleFunctionsLeaks.test_win_service_get_descriptionN)5r� r� r� r� r5 r. r� r� r~ r� r� r r� r� r �platform�machiner r� r r� r� r r� r� r� r� r rd �path�existsr� r� r r� �getuidr� r r r r r r r r r r r r r r r r! r( r( r( r) r� O s~ � � �r� ).r� r+ rd r" r] �psutil._commonr r r r r r � psutil.testsr r r r r r r r r r r r r r r r r r r r r � _psplatformrb rl rU r! r. r/ r� r� r� r� r� r( r( r( r) �<module> sV 66 tests/__pycache__/test_unicode.cpython-310.pyc 0000644 00000024264 15030211765 0015314 0 ustar 00 o �h�( � @ s� d Z ddlZddlZddlZddlmZ ddlZddlmZ ddlmZ ddlm Z ddl m Z ddl m Z dd l m Z dd l mZ dd l mZ dd l mZ dd l mZ ddl mZ ddl mZ ddl mZ ddl mZ ddl mZ ddl mZ ddl mZ ddl mZ ddl mZ ddl mZ ddl m Z ddl m Z ddl m Z ddl m Z d d � Z G d d � d e�Z!ej"j#d d!�ej"j$e d"d#�G d$d%� d%e!���Z%ej"j$e d&d#�G d'd(� d(e%��Z&G d)d*� d*e!�Z'dS )+aY Notes about unicode handling in psutil ======================================. Starting from version 5.3.0 psutil adds unicode support, see: https://github.com/giampaolo/psutil/issues/1040 The notes below apply to *any* API returning a string such as process exe(), cwd() or username(): * all strings are encoded by using the OS filesystem encoding (sys.getfilesystemencoding()) which varies depending on the platform (e.g. "UTF-8" on macOS, "mbcs" on Win) * no API call is supposed to crash with UnicodeDecodeError * instead, in case of badly encoded data returned by the OS, the following error handlers are used to replace the corrupted characters in the string: * sys.getfilesystemencodeerrors() or "surrogatescape" on POSIX and "replace" on Windows. For a detailed explanation of how psutil handles unicode see #1040. Tests ===== List of APIs returning or dealing with a string: ('not tested' means they are not tested to deal with non-ASCII strings): * Process.cmdline() * Process.cwd() * Process.environ() * Process.exe() * Process.memory_maps() * Process.name() * Process.net_connections('unix') * Process.open_files() * Process.username() (not tested) * disk_io_counters() (not tested) * disk_partitions() (not tested) * disk_usage(str) * net_connections('unix') * net_if_addrs() (not tested) * net_if_stats() (not tested) * net_io_counters() (not tested) * sensors_fans() (not tested) * sensors_temperatures() (not tested) * users() (not tested) * WindowsService.binpath() (not tested) * WindowsService.description() (not tested) * WindowsService.display_name() (not tested) * WindowsService.name() (not tested) * WindowsService.status() (not tested) * WindowsService.username() (not tested) In here we create a unicode path with a funky non-ASCII name and (where possible) make psutil return it back (e.g. on name(), exe(), open_files(), etc.) and make sure that: * psutil never crashes with UnicodeDecodeError * the returned path matches � N)�closing)�BSD)�POSIX)�WINDOWS)�ASCII_FS)� CI_TESTING)� HAS_ENVIRON)�HAS_MEMORY_MAPS)�HAS_NET_CONNECTIONS_UNIX)�INVALID_UNICODE_SUFFIX)�PYPY�� TESTFN_PREFIX)�UNICODE_SUFFIX)�PsutilTestCase)�bind_unix_socket)�chdir)�copyload_shared_lib)� create_py_exe)� get_testfn)�pytest)� safe_mkdir)� safe_rmpath)�skip_on_access_denied)�spawn_testproc)� terminatec C s� d}t | d�}zGz t|� t|� t|gd�}t�||d � t|d � W n ttfy? Y W |dur9t|� t|� dS w W |durIt|� t|� dS |durWt|� t|� w )z`Return True if both the fs and the subprocess module can deal with a unicode file name. N��suffix)�cmdz-2FT) r r r r �shutil�copyfile�UnicodeEncodeError�OSErrorr )r �sproc�testfn� r% �L/usr/local/CyberCP/lib/python3.10/site-packages/psutil/tests/test_unicode.py� try_unicodee s* � � r' c s0 e Zd ZdZe� fdd��Z� fdd�Z� ZS )�BaseUnicodeTestNc sT t � �� d| _d | _| jd ur(t| j�sd| _d S t| jd�| _t| j� d S d S )NFTr )�super� setUpClass� skip_tests� funky_name� funky_suffixr' r r )�cls�� __class__r% r& r* � s �zBaseUnicodeTest.setUpClassc s t � �� | jr t�d��d S )Nzcan't handle unicode str)r) �setUpr+ r �skip��selfr/ r% r& r1 � s �zBaseUnicodeTest.setUp)�__name__� __module__� __qualname__r- � classmethodr* r1 � __classcell__r% r% r/ r& r( � s r( �serial)�namezASCII fs��reasonc @ s� e Zd ZdZeZdd� Zdd� Zdd� Zdd � Z d d � Z e j j eo"ed d �dd� �Ze j j e dd �dd� �Ze j j e dd �e j j e dd �e� dd� ���Zdd� Ze j j e dd �e j j edd �dd� ��Zd S ) � TestFSAPIsz1Test FS APIs with a funky, valid, UTF8 path name.c C sF t �� � t �d� | jt�d�v W d � S 1 s w Y d S )N�ignore�.)�warnings�catch_warnings� simplefilterr, �os�listdirr3 r% r% r& �expect_exact_path_match� s $�z"TestFSAPIs.expect_exact_path_matchc C sf | j ddg}| �|�}t�|j�}|�� }t|t�s J �| �� r/t j � |�t j � | j �ks1J �d S d S �Nz-cz2import time; [time.sleep(0.1) for x in range(100)]) r, r �psutil�Process�pid�exe� isinstance�strrF rD �path�normcase)r4 r �subp�prK r% r% r& � test_proc_exe� s � �zTestFSAPIs.test_proc_exec C sZ | j ddg}| �|�}t�|j��� }t|t�sJ �| �� r)|t j � | j �ks+J �d S d S rG ) r, r rH rI rJ r; rL rM rF rD rN �basename)r4 r rP r; r% r% r& �test_proc_name� s � �zTestFSAPIs.test_proc_namec C s^ | j ddg}| �|�}t�|j�}|�� }|D ] }t|t�s J �q| �� r+||ks-J �d S d S rG ) r, r rH rI rJ �cmdlinerL rM rF )r4 r rP rQ rU �partr% r% r& �test_proc_cmdline� s � �z TestFSAPIs.test_proc_cmdlinec C s� | j d }| �t|� t|� t|�� t�� }|�� }W d � n1 s&w Y t|�� t �s4J �| � � r>||ks@J �d S d S �N�2) r, � addCleanupr r r rH rI �cwdrL rM rF )r4 �dnamerQ r[ r% r% r& � test_proc_cwd� s � �zTestFSAPIs.test_proc_cwdzfails on PYPY + WINDOWSr< c C s� t �� }t|�� �}t| jd�� t|�� �}W d � n1 s!w Y || �� j}t|t �s4J �t r=|s=t � d��| � � rPtj�|�tj�| j�ksRJ �d S d S )N�rbzopen_files on BSD is broken)rH rI �set� open_files�openr, �poprN rL rM r r r2 rF rD rO )r4 rQ �start�newrN r% r% r& �test_proc_open_files� s � �z TestFSAPIs.test_proc_open_filesz POSIX onlyc C st | j | jd�}t|�}t|��! t�� �d�d }t|jt �s!J �|j|ks(J �W d � d S 1 s3w Y d S )Nr �unixr ) r r- r r rH rI �net_connectionsrL �laddrrM )r4 r; �sock�connr% r% r& �test_proc_net_connections� s "�z$TestFSAPIs.test_proc_net_connectionszcan't list UNIX socketsc C s~ dd� }| j | jd�}t|�}t|��" tjdd�}||�}t|jt�s&J �|j|ks-J �W d � d S 1 s8w Y d S )Nc S s. | D ]}t j�|j��t�r| S qtd��)Nzconnection not found)rD rN rS rh � startswithr � ValueError)�consrj r% r% r& � find_sock� s �z2TestFSAPIs.test_net_connections.<locals>.find_sockr rf )�kind) r r- r r rH rg rL rh rM )r4 ro r; ri rn rj r% r% r& �test_net_connections� s "�z TestFSAPIs.test_net_connectionsc C s, | j d }| �t|� t|� t�|� d S rX )r, rZ r r rH � disk_usage)r4 r\ r% r% r& �test_disk_usage s zTestFSAPIs.test_disk_usage� not supportedzunstable on PYPYc s� t | jd��5}dd� � � fdd�t�� �� D �}dd� |D �}� |�|v s'J �|D ] }t|t�s2J �q)W d � d S 1 s>w Y d S )Nr c S s t j�t j�| ��S )N)rD rN �realpathrO )rQ r% r% r& �normpath s z-TestFSAPIs.test_memory_maps.<locals>.normpathc s g | ]}� |j ��qS r% )rN ��.0�x�rv r% r& � <listcomp> s �z/TestFSAPIs.test_memory_maps.<locals>.<listcomp>c S s g | ]}t |v r|�qS r% r rw r% r% r& r{ s )r r- rH rI � memory_mapsrL rM )r4 � funky_path�libpathsrN r% rz r& �test_memory_maps s ��"�zTestFSAPIs.test_memory_mapsN)r5 r6 r7 �__doc__r r- rF rR rT rW r] r �mark�skipifr r re r rk r r rq rs r r r% r% r% r& r> � s, �r> zunreliable on CIc @ s e Zd ZdZeZdd� ZdS )�TestFSAPIsWithInvalidPathz-Test FS APIs with a funky, invalid path name.c C s dS )NTr% r3 r% r% r& rF s z1TestFSAPIsWithInvalidPath.expect_exact_path_matchN)r5 r6 r7 r� r r- rF r% r% r% r& r� s r� c @ sB e Zd ZdZeZejje dd�ejje oe dd�dd� ��Z dS )� TestNonFSAPISz&Unicode tests for non fs-related APIs.rt r< zsegfaults on PYPY + WINDOWSc C sx t j�� }| j|d< | j|d�}t�|j�}|�� }|�� D ]\}}t |t �s)J �t |t �s0J �q |d | jks:J �d S )N� FUNNY_ARG)�env) rD �environ�copyr- r rH rI rJ �itemsrL rM )r4 r� r# rQ �k�vr% r% r& �test_proc_environ, s z TestNonFSAPIS.test_proc_environN) r5 r6 r7 r� r r- r r� r� r r r r� r% r% r% r& r� ' s r� )(r� rD r rA � contextlibr rH r r r � psutil.testsr r r r r r r r r r r r r r r r r r r r r r' r( r� � xdist_groupr� r> r� r� r% r% r% r& �<module> sN > tests/test_testutils.py 0000644 00000043642 15030211765 0011370 0 ustar 00 #!/usr/bin/env python3 # Copyright (c) 2009, Giampaolo Rodola'. All rights reserved. # Use of this source code is governed by a BSD-style license that can be # found in the LICENSE file. """Tests for testing utils (psutil.tests namespace).""" import collections import errno import os import socket import stat import subprocess import textwrap import unittest import warnings from unittest import mock import psutil import psutil.tests from psutil import FREEBSD from psutil import NETBSD from psutil import POSIX from psutil._common import open_binary from psutil._common import open_text from psutil._common import supports_ipv6 from psutil.tests import CI_TESTING from psutil.tests import COVERAGE from psutil.tests import HAS_NET_CONNECTIONS_UNIX from psutil.tests import HERE from psutil.tests import PYTHON_EXE from psutil.tests import PYTHON_EXE_ENV from psutil.tests import PsutilTestCase from psutil.tests import TestMemoryLeak from psutil.tests import bind_socket from psutil.tests import bind_unix_socket from psutil.tests import call_until from psutil.tests import chdir from psutil.tests import create_sockets from psutil.tests import fake_pytest from psutil.tests import filter_proc_net_connections from psutil.tests import get_free_port from psutil.tests import is_namedtuple from psutil.tests import process_namespace from psutil.tests import pytest from psutil.tests import reap_children from psutil.tests import retry from psutil.tests import retry_on_failure from psutil.tests import safe_mkdir from psutil.tests import safe_rmpath from psutil.tests import system_namespace from psutil.tests import tcp_socketpair from psutil.tests import terminate from psutil.tests import unix_socketpair from psutil.tests import wait_for_file from psutil.tests import wait_for_pid # =================================================================== # --- Unit tests for test utilities. # =================================================================== class TestRetryDecorator(PsutilTestCase): @mock.patch('time.sleep') def test_retry_success(self, sleep): # Fail 3 times out of 5; make sure the decorated fun returns. @retry(retries=5, interval=1, logfun=None) def foo(): while queue: queue.pop() 1 / 0 # noqa: B018 return 1 queue = list(range(3)) assert foo() == 1 assert sleep.call_count == 3 @mock.patch('time.sleep') def test_retry_failure(self, sleep): # Fail 6 times out of 5; th function is supposed to raise exc. @retry(retries=5, interval=1, logfun=None) def foo(): while queue: queue.pop() 1 / 0 # noqa: B018 return 1 queue = list(range(6)) with pytest.raises(ZeroDivisionError): foo() assert sleep.call_count == 5 @mock.patch('time.sleep') def test_exception_arg(self, sleep): @retry(exception=ValueError, interval=1) def foo(): raise TypeError with pytest.raises(TypeError): foo() assert sleep.call_count == 0 @mock.patch('time.sleep') def test_no_interval_arg(self, sleep): # if interval is not specified sleep is not supposed to be called @retry(retries=5, interval=None, logfun=None) def foo(): 1 / 0 # noqa: B018 with pytest.raises(ZeroDivisionError): foo() assert sleep.call_count == 0 @mock.patch('time.sleep') def test_retries_arg(self, sleep): @retry(retries=5, interval=1, logfun=None) def foo(): 1 / 0 # noqa: B018 with pytest.raises(ZeroDivisionError): foo() assert sleep.call_count == 5 @mock.patch('time.sleep') def test_retries_and_timeout_args(self, sleep): with pytest.raises(ValueError): retry(retries=5, timeout=1) class TestSyncTestUtils(PsutilTestCase): def test_wait_for_pid(self): wait_for_pid(os.getpid()) nopid = max(psutil.pids()) + 99999 with mock.patch('psutil.tests.retry.__iter__', return_value=iter([0])): with pytest.raises(psutil.NoSuchProcess): wait_for_pid(nopid) def test_wait_for_file(self): testfn = self.get_testfn() with open(testfn, 'w') as f: f.write('foo') wait_for_file(testfn) assert not os.path.exists(testfn) def test_wait_for_file_empty(self): testfn = self.get_testfn() with open(testfn, 'w'): pass wait_for_file(testfn, empty=True) assert not os.path.exists(testfn) def test_wait_for_file_no_file(self): testfn = self.get_testfn() with mock.patch('psutil.tests.retry.__iter__', return_value=iter([0])): with pytest.raises(OSError): wait_for_file(testfn) def test_wait_for_file_no_delete(self): testfn = self.get_testfn() with open(testfn, 'w') as f: f.write('foo') wait_for_file(testfn, delete=False) assert os.path.exists(testfn) def test_call_until(self): call_until(lambda: 1) # TODO: test for timeout class TestFSTestUtils(PsutilTestCase): def test_open_text(self): with open_text(__file__) as f: assert f.mode == 'r' def test_open_binary(self): with open_binary(__file__) as f: assert f.mode == 'rb' def test_safe_mkdir(self): testfn = self.get_testfn() safe_mkdir(testfn) assert os.path.isdir(testfn) safe_mkdir(testfn) assert os.path.isdir(testfn) def test_safe_rmpath(self): # test file is removed testfn = self.get_testfn() open(testfn, 'w').close() safe_rmpath(testfn) assert not os.path.exists(testfn) # test no exception if path does not exist safe_rmpath(testfn) # test dir is removed os.mkdir(testfn) safe_rmpath(testfn) assert not os.path.exists(testfn) # test other exceptions are raised with mock.patch( 'psutil.tests.os.stat', side_effect=OSError(errno.EINVAL, "") ) as m: with pytest.raises(OSError): safe_rmpath(testfn) assert m.called def test_chdir(self): testfn = self.get_testfn() base = os.getcwd() os.mkdir(testfn) with chdir(testfn): assert os.getcwd() == os.path.join(base, testfn) assert os.getcwd() == base class TestProcessUtils(PsutilTestCase): def test_reap_children(self): subp = self.spawn_testproc() p = psutil.Process(subp.pid) assert p.is_running() reap_children() assert not p.is_running() assert not psutil.tests._pids_started assert not psutil.tests._subprocesses_started def test_spawn_children_pair(self): child, grandchild = self.spawn_children_pair() assert child.pid != grandchild.pid assert child.is_running() assert grandchild.is_running() children = psutil.Process().children() assert children == [child] children = psutil.Process().children(recursive=True) assert len(children) == 2 assert child in children assert grandchild in children assert child.ppid() == os.getpid() assert grandchild.ppid() == child.pid terminate(child) assert not child.is_running() assert grandchild.is_running() terminate(grandchild) assert not grandchild.is_running() @pytest.mark.skipif(not POSIX, reason="POSIX only") def test_spawn_zombie(self): _parent, zombie = self.spawn_zombie() assert zombie.status() == psutil.STATUS_ZOMBIE def test_terminate(self): # by subprocess.Popen p = self.spawn_testproc() terminate(p) self.assertPidGone(p.pid) terminate(p) # by psutil.Process p = psutil.Process(self.spawn_testproc().pid) terminate(p) self.assertPidGone(p.pid) terminate(p) # by psutil.Popen cmd = [ PYTHON_EXE, "-c", "import time; [time.sleep(0.1) for x in range(100)];", ] p = psutil.Popen( cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE, env=PYTHON_EXE_ENV, ) terminate(p) self.assertPidGone(p.pid) terminate(p) # by PID pid = self.spawn_testproc().pid terminate(pid) self.assertPidGone(p.pid) terminate(pid) # zombie if POSIX: parent, zombie = self.spawn_zombie() terminate(parent) terminate(zombie) self.assertPidGone(parent.pid) self.assertPidGone(zombie.pid) class TestNetUtils(PsutilTestCase): def bind_socket(self): port = get_free_port() with bind_socket(addr=('', port)) as s: assert s.getsockname()[1] == port @pytest.mark.skipif(not POSIX, reason="POSIX only") def test_bind_unix_socket(self): name = self.get_testfn() with bind_unix_socket(name) as sock: assert sock.family == socket.AF_UNIX assert sock.type == socket.SOCK_STREAM assert sock.getsockname() == name assert os.path.exists(name) assert stat.S_ISSOCK(os.stat(name).st_mode) # UDP name = self.get_testfn() with bind_unix_socket(name, type=socket.SOCK_DGRAM) as sock: assert sock.type == socket.SOCK_DGRAM def test_tcp_socketpair(self): addr = ("127.0.0.1", get_free_port()) server, client = tcp_socketpair(socket.AF_INET, addr=addr) with server, client: # Ensure they are connected and the positions are correct. assert server.getsockname() == addr assert client.getpeername() == addr assert client.getsockname() != addr @pytest.mark.skipif(not POSIX, reason="POSIX only") @pytest.mark.skipif( NETBSD or FREEBSD, reason="/var/run/log UNIX socket opened by default" ) def test_unix_socketpair(self): p = psutil.Process() num_fds = p.num_fds() assert not filter_proc_net_connections(p.net_connections(kind='unix')) name = self.get_testfn() server, client = unix_socketpair(name) try: assert os.path.exists(name) assert stat.S_ISSOCK(os.stat(name).st_mode) assert p.num_fds() - num_fds == 2 assert ( len( filter_proc_net_connections(p.net_connections(kind='unix')) ) == 2 ) assert server.getsockname() == name assert client.getpeername() == name finally: client.close() server.close() def test_create_sockets(self): with create_sockets() as socks: fams = collections.defaultdict(int) types = collections.defaultdict(int) for s in socks: fams[s.family] += 1 # work around http://bugs.python.org/issue30204 types[s.getsockopt(socket.SOL_SOCKET, socket.SO_TYPE)] += 1 assert fams[socket.AF_INET] >= 2 if supports_ipv6(): assert fams[socket.AF_INET6] >= 2 if POSIX and HAS_NET_CONNECTIONS_UNIX: assert fams[socket.AF_UNIX] >= 2 assert types[socket.SOCK_STREAM] >= 2 assert types[socket.SOCK_DGRAM] >= 2 @pytest.mark.xdist_group(name="serial") class TestMemLeakClass(TestMemoryLeak): @retry_on_failure() def test_times(self): def fun(): cnt['cnt'] += 1 cnt = {'cnt': 0} self.execute(fun, times=10, warmup_times=15) assert cnt['cnt'] == 26 def test_param_err(self): with pytest.raises(ValueError): self.execute(lambda: 0, times=0) with pytest.raises(ValueError): self.execute(lambda: 0, times=-1) with pytest.raises(ValueError): self.execute(lambda: 0, warmup_times=-1) with pytest.raises(ValueError): self.execute(lambda: 0, tolerance=-1) with pytest.raises(ValueError): self.execute(lambda: 0, retries=-1) @retry_on_failure() @pytest.mark.skipif(CI_TESTING, reason="skipped on CI") @pytest.mark.skipif(COVERAGE, reason="skipped during test coverage") def test_leak_mem(self): ls = [] def fun(ls=ls): ls.append("x" * 248 * 1024) try: # will consume around 60M in total with pytest.raises(AssertionError, match="extra-mem"): self.execute(fun, times=100) finally: del ls def test_unclosed_files(self): def fun(): f = open(__file__) # noqa: SIM115 self.addCleanup(f.close) box.append(f) box = [] kind = "fd" if POSIX else "handle" with pytest.raises(AssertionError, match="unclosed " + kind): self.execute(fun) def test_tolerance(self): def fun(): ls.append("x" * 24 * 1024) ls = [] times = 100 self.execute( fun, times=times, warmup_times=0, tolerance=200 * 1024 * 1024 ) assert len(ls) == times + 1 def test_execute_w_exc(self): def fun_1(): 1 / 0 # noqa: B018 self.execute_w_exc(ZeroDivisionError, fun_1) with pytest.raises(ZeroDivisionError): self.execute_w_exc(OSError, fun_1) def fun_2(): pass with pytest.raises(AssertionError): self.execute_w_exc(ZeroDivisionError, fun_2) class TestFakePytest(PsutilTestCase): def run_test_class(self, klass): suite = unittest.TestSuite() suite.addTest(klass) runner = unittest.TextTestRunner() result = runner.run(suite) return result def test_raises(self): with fake_pytest.raises(ZeroDivisionError) as cm: 1 / 0 # noqa: B018 assert isinstance(cm.value, ZeroDivisionError) with fake_pytest.raises(ValueError, match="foo") as cm: raise ValueError("foo") try: with fake_pytest.raises(ValueError, match="foo") as cm: raise ValueError("bar") except AssertionError as err: assert str(err) == '"foo" does not match "bar"' else: raise self.fail("exception not raised") def test_mark(self): @fake_pytest.mark.xdist_group(name="serial") def foo(): return 1 assert foo() == 1 @fake_pytest.mark.xdist_group(name="serial") class Foo: def bar(self): return 1 assert Foo().bar() == 1 def test_skipif(self): class TestCase(unittest.TestCase): @fake_pytest.mark.skipif(True, reason="reason") def foo(self): assert 1 == 1 # noqa: PLR0133 result = self.run_test_class(TestCase("foo")) assert result.wasSuccessful() assert len(result.skipped) == 1 assert result.skipped[0][1] == "reason" class TestCase(unittest.TestCase): @fake_pytest.mark.skipif(False, reason="reason") def foo(self): assert 1 == 1 # noqa: PLR0133 result = self.run_test_class(TestCase("foo")) assert result.wasSuccessful() assert len(result.skipped) == 0 def test_skip(self): class TestCase(unittest.TestCase): def foo(self): fake_pytest.skip("reason") assert 1 == 0 # noqa: PLR0133 result = self.run_test_class(TestCase("foo")) assert result.wasSuccessful() assert len(result.skipped) == 1 assert result.skipped[0][1] == "reason" def test_main(self): tmpdir = self.get_testfn(dir=HERE) os.mkdir(tmpdir) with open(os.path.join(tmpdir, "__init__.py"), "w"): pass with open(os.path.join(tmpdir, "test_file.py"), "w") as f: f.write(textwrap.dedent("""\ import unittest class TestCase(unittest.TestCase): def test_passed(self): pass """).lstrip()) with mock.patch.object(psutil.tests, "HERE", tmpdir): with self.assertWarnsRegex( UserWarning, "Fake pytest module was used" ): suite = fake_pytest.main() assert suite.countTestCases() == 1 def test_warns(self): # success with fake_pytest.warns(UserWarning): warnings.warn("foo", UserWarning, stacklevel=1) # failure try: with fake_pytest.warns(UserWarning): warnings.warn("foo", DeprecationWarning, stacklevel=1) except AssertionError: pass else: raise self.fail("exception not raised") # match success with fake_pytest.warns(UserWarning, match="foo"): warnings.warn("foo", UserWarning, stacklevel=1) # match failure try: with fake_pytest.warns(UserWarning, match="foo"): warnings.warn("bar", UserWarning, stacklevel=1) except AssertionError: pass else: raise self.fail("exception not raised") class TestTestingUtils(PsutilTestCase): def test_process_namespace(self): p = psutil.Process() ns = process_namespace(p) ns.test() fun = next(x for x in ns.iter(ns.getters) if x[1] == 'ppid')[0] assert fun() == p.ppid() def test_system_namespace(self): ns = system_namespace() fun = next(x for x in ns.iter(ns.getters) if x[1] == 'net_if_addrs')[0] assert fun() == psutil.net_if_addrs() class TestOtherUtils(PsutilTestCase): def test_is_namedtuple(self): assert is_namedtuple(collections.namedtuple('foo', 'a b c')(1, 2, 3)) assert not is_namedtuple(tuple()) tests/test_bsd.py 0000644 00000047337 15030211765 0010105 0 ustar 00 #!/usr/bin/env python3 # Copyright (c) 2009, Giampaolo Rodola'. All rights reserved. # Use of this source code is governed by a BSD-style license that can be # found in the LICENSE file. # TODO: (FreeBSD) add test for comparing connections with 'sockstat' cmd. """Tests specific to all BSD platforms.""" import datetime import os import re import shutil import time import psutil from psutil import BSD from psutil import FREEBSD from psutil import NETBSD from psutil import OPENBSD from psutil.tests import HAS_BATTERY from psutil.tests import TOLERANCE_SYS_MEM from psutil.tests import PsutilTestCase from psutil.tests import pytest from psutil.tests import retry_on_failure from psutil.tests import sh from psutil.tests import spawn_testproc from psutil.tests import terminate if BSD: from psutil._psutil_posix import getpagesize PAGESIZE = getpagesize() # muse requires root privileges MUSE_AVAILABLE = os.getuid() == 0 and shutil.which("muse") else: PAGESIZE = None MUSE_AVAILABLE = False def sysctl(cmdline): """Expects a sysctl command with an argument and parse the result returning only the value of interest. """ result = sh("sysctl " + cmdline) if FREEBSD: result = result[result.find(": ") + 2 :] elif OPENBSD or NETBSD: result = result[result.find("=") + 1 :] try: return int(result) except ValueError: return result def muse(field): """Thin wrapper around 'muse' cmdline utility.""" out = sh('muse') for line in out.split('\n'): if line.startswith(field): break else: raise ValueError("line not found") return int(line.split()[1]) # ===================================================================== # --- All BSD* # ===================================================================== @pytest.mark.skipif(not BSD, reason="BSD only") class BSDTestCase(PsutilTestCase): """Generic tests common to all BSD variants.""" @classmethod def setUpClass(cls): cls.pid = spawn_testproc().pid @classmethod def tearDownClass(cls): terminate(cls.pid) @pytest.mark.skipif(NETBSD, reason="-o lstart doesn't work on NETBSD") def test_process_create_time(self): output = sh(f"ps -o lstart -p {self.pid}") start_ps = output.replace('STARTED', '').strip() start_psutil = psutil.Process(self.pid).create_time() start_psutil = time.strftime( "%a %b %e %H:%M:%S %Y", time.localtime(start_psutil) ) assert start_ps == start_psutil def test_disks(self): # test psutil.disk_usage() and psutil.disk_partitions() # against "df -a" def df(path): out = sh(f'df -k "{path}"').strip() lines = out.split('\n') lines.pop(0) line = lines.pop(0) dev, total, used, free = line.split()[:4] if dev == 'none': dev = '' total = int(total) * 1024 used = int(used) * 1024 free = int(free) * 1024 return dev, total, used, free for part in psutil.disk_partitions(all=False): usage = psutil.disk_usage(part.mountpoint) dev, total, used, free = df(part.mountpoint) assert part.device == dev assert usage.total == total # 10 MB tolerance if abs(usage.free - free) > 10 * 1024 * 1024: raise self.fail(f"psutil={usage.free}, df={free}") if abs(usage.used - used) > 10 * 1024 * 1024: raise self.fail(f"psutil={usage.used}, df={used}") @pytest.mark.skipif( not shutil.which("sysctl"), reason="sysctl cmd not available" ) def test_cpu_count_logical(self): syst = sysctl("hw.ncpu") assert psutil.cpu_count(logical=True) == syst @pytest.mark.skipif( not shutil.which("sysctl"), reason="sysctl cmd not available" ) @pytest.mark.skipif( NETBSD, reason="skipped on NETBSD" # we check /proc/meminfo ) def test_virtual_memory_total(self): num = sysctl('hw.physmem') assert num == psutil.virtual_memory().total @pytest.mark.skipif( not shutil.which("ifconfig"), reason="ifconfig cmd not available" ) def test_net_if_stats(self): for name, stats in psutil.net_if_stats().items(): try: out = sh(f"ifconfig {name}") except RuntimeError: pass else: assert stats.isup == ('RUNNING' in out) if "mtu" in out: assert stats.mtu == int(re.findall(r'mtu (\d+)', out)[0]) # ===================================================================== # --- FreeBSD # ===================================================================== @pytest.mark.skipif(not FREEBSD, reason="FREEBSD only") class FreeBSDPsutilTestCase(PsutilTestCase): @classmethod def setUpClass(cls): cls.pid = spawn_testproc().pid @classmethod def tearDownClass(cls): terminate(cls.pid) @retry_on_failure() def test_memory_maps(self): out = sh(f"procstat -v {self.pid}") maps = psutil.Process(self.pid).memory_maps(grouped=False) lines = out.split('\n')[1:] while lines: line = lines.pop() fields = line.split() _, start, stop, _perms, res = fields[:5] map = maps.pop() assert f"{start}-{stop}" == map.addr assert int(res) == map.rss if not map.path.startswith('['): assert fields[10] == map.path def test_exe(self): out = sh(f"procstat -b {self.pid}") assert psutil.Process(self.pid).exe() == out.split('\n')[1].split()[-1] def test_cmdline(self): out = sh(f"procstat -c {self.pid}") assert ' '.join(psutil.Process(self.pid).cmdline()) == ' '.join( out.split('\n')[1].split()[2:] ) def test_uids_gids(self): out = sh(f"procstat -s {self.pid}") euid, ruid, suid, egid, rgid, sgid = out.split('\n')[1].split()[2:8] p = psutil.Process(self.pid) uids = p.uids() gids = p.gids() assert uids.real == int(ruid) assert uids.effective == int(euid) assert uids.saved == int(suid) assert gids.real == int(rgid) assert gids.effective == int(egid) assert gids.saved == int(sgid) @retry_on_failure() def test_ctx_switches(self): tested = [] out = sh(f"procstat -r {self.pid}") p = psutil.Process(self.pid) for line in out.split('\n'): line = line.lower().strip() if ' voluntary context' in line: pstat_value = int(line.split()[-1]) psutil_value = p.num_ctx_switches().voluntary assert pstat_value == psutil_value tested.append(None) elif ' involuntary context' in line: pstat_value = int(line.split()[-1]) psutil_value = p.num_ctx_switches().involuntary assert pstat_value == psutil_value tested.append(None) if len(tested) != 2: raise RuntimeError("couldn't find lines match in procstat out") @retry_on_failure() def test_cpu_times(self): tested = [] out = sh(f"procstat -r {self.pid}") p = psutil.Process(self.pid) for line in out.split('\n'): line = line.lower().strip() if 'user time' in line: pstat_value = float('0.' + line.split()[-1].split('.')[-1]) psutil_value = p.cpu_times().user assert pstat_value == psutil_value tested.append(None) elif 'system time' in line: pstat_value = float('0.' + line.split()[-1].split('.')[-1]) psutil_value = p.cpu_times().system assert pstat_value == psutil_value tested.append(None) if len(tested) != 2: raise RuntimeError("couldn't find lines match in procstat out") @pytest.mark.skipif(not FREEBSD, reason="FREEBSD only") class FreeBSDSystemTestCase(PsutilTestCase): @staticmethod def parse_swapinfo(): # the last line is always the total output = sh("swapinfo -k").splitlines()[-1] parts = re.split(r'\s+', output) if not parts: raise ValueError(f"Can't parse swapinfo: {output}") # the size is in 1k units, so multiply by 1024 total, used, free = (int(p) * 1024 for p in parts[1:4]) return total, used, free def test_cpu_frequency_against_sysctl(self): # Currently only cpu 0 is frequency is supported in FreeBSD # All other cores use the same frequency. sensor = "dev.cpu.0.freq" try: sysctl_result = int(sysctl(sensor)) except RuntimeError: raise pytest.skip("frequencies not supported by kernel") assert psutil.cpu_freq().current == sysctl_result sensor = "dev.cpu.0.freq_levels" sysctl_result = sysctl(sensor) # sysctl returns a string of the format: # <freq_level_1>/<voltage_level_1> <freq_level_2>/<voltage_level_2>... # Ordered highest available to lowest available. max_freq = int(sysctl_result.split()[0].split("/")[0]) min_freq = int(sysctl_result.split()[-1].split("/")[0]) assert psutil.cpu_freq().max == max_freq assert psutil.cpu_freq().min == min_freq # --- virtual_memory(); tests against sysctl @retry_on_failure() def test_vmem_active(self): syst = sysctl("vm.stats.vm.v_active_count") * PAGESIZE assert abs(psutil.virtual_memory().active - syst) < TOLERANCE_SYS_MEM @retry_on_failure() def test_vmem_inactive(self): syst = sysctl("vm.stats.vm.v_inactive_count") * PAGESIZE assert abs(psutil.virtual_memory().inactive - syst) < TOLERANCE_SYS_MEM @retry_on_failure() def test_vmem_wired(self): syst = sysctl("vm.stats.vm.v_wire_count") * PAGESIZE assert abs(psutil.virtual_memory().wired - syst) < TOLERANCE_SYS_MEM @retry_on_failure() def test_vmem_cached(self): syst = sysctl("vm.stats.vm.v_cache_count") * PAGESIZE assert abs(psutil.virtual_memory().cached - syst) < TOLERANCE_SYS_MEM @retry_on_failure() def test_vmem_free(self): syst = sysctl("vm.stats.vm.v_free_count") * PAGESIZE assert abs(psutil.virtual_memory().free - syst) < TOLERANCE_SYS_MEM @retry_on_failure() def test_vmem_buffers(self): syst = sysctl("vfs.bufspace") assert abs(psutil.virtual_memory().buffers - syst) < TOLERANCE_SYS_MEM # --- virtual_memory(); tests against muse @pytest.mark.skipif(not MUSE_AVAILABLE, reason="muse not installed") def test_muse_vmem_total(self): num = muse('Total') assert psutil.virtual_memory().total == num @pytest.mark.skipif(not MUSE_AVAILABLE, reason="muse not installed") @retry_on_failure() def test_muse_vmem_active(self): num = muse('Active') assert abs(psutil.virtual_memory().active - num) < TOLERANCE_SYS_MEM @pytest.mark.skipif(not MUSE_AVAILABLE, reason="muse not installed") @retry_on_failure() def test_muse_vmem_inactive(self): num = muse('Inactive') assert abs(psutil.virtual_memory().inactive - num) < TOLERANCE_SYS_MEM @pytest.mark.skipif(not MUSE_AVAILABLE, reason="muse not installed") @retry_on_failure() def test_muse_vmem_wired(self): num = muse('Wired') assert abs(psutil.virtual_memory().wired - num) < TOLERANCE_SYS_MEM @pytest.mark.skipif(not MUSE_AVAILABLE, reason="muse not installed") @retry_on_failure() def test_muse_vmem_cached(self): num = muse('Cache') assert abs(psutil.virtual_memory().cached - num) < TOLERANCE_SYS_MEM @pytest.mark.skipif(not MUSE_AVAILABLE, reason="muse not installed") @retry_on_failure() def test_muse_vmem_free(self): num = muse('Free') assert abs(psutil.virtual_memory().free - num) < TOLERANCE_SYS_MEM @pytest.mark.skipif(not MUSE_AVAILABLE, reason="muse not installed") @retry_on_failure() def test_muse_vmem_buffers(self): num = muse('Buffer') assert abs(psutil.virtual_memory().buffers - num) < TOLERANCE_SYS_MEM def test_cpu_stats_ctx_switches(self): assert ( abs( psutil.cpu_stats().ctx_switches - sysctl('vm.stats.sys.v_swtch') ) < 1000 ) def test_cpu_stats_interrupts(self): assert ( abs(psutil.cpu_stats().interrupts - sysctl('vm.stats.sys.v_intr')) < 1000 ) def test_cpu_stats_soft_interrupts(self): assert ( abs( psutil.cpu_stats().soft_interrupts - sysctl('vm.stats.sys.v_soft') ) < 1000 ) @retry_on_failure() def test_cpu_stats_syscalls(self): # pretty high tolerance but it looks like it's OK. assert ( abs(psutil.cpu_stats().syscalls - sysctl('vm.stats.sys.v_syscall')) < 200000 ) # def test_cpu_stats_traps(self): # self.assertAlmostEqual(psutil.cpu_stats().traps, # sysctl('vm.stats.sys.v_trap'), delta=1000) # --- swap memory def test_swapmem_free(self): _total, _used, free = self.parse_swapinfo() assert abs(psutil.swap_memory().free - free) < TOLERANCE_SYS_MEM def test_swapmem_used(self): _total, used, _free = self.parse_swapinfo() assert abs(psutil.swap_memory().used - used) < TOLERANCE_SYS_MEM def test_swapmem_total(self): total, _used, _free = self.parse_swapinfo() assert abs(psutil.swap_memory().total - total) < TOLERANCE_SYS_MEM # --- others def test_boot_time(self): s = sysctl('sysctl kern.boottime') s = s[s.find(" sec = ") + 7 :] s = s[: s.find(',')] btime = int(s) assert btime == psutil.boot_time() # --- sensors_battery @pytest.mark.skipif(not HAS_BATTERY, reason="no battery") def test_sensors_battery(self): def secs2hours(secs): m, _s = divmod(secs, 60) h, m = divmod(m, 60) return f"{int(h)}:{int(m):02}" out = sh("acpiconf -i 0") fields = {x.split('\t')[0]: x.split('\t')[-1] for x in out.split("\n")} metrics = psutil.sensors_battery() percent = int(fields['Remaining capacity:'].replace('%', '')) remaining_time = fields['Remaining time:'] assert metrics.percent == percent if remaining_time == 'unknown': assert metrics.secsleft == psutil.POWER_TIME_UNLIMITED else: assert secs2hours(metrics.secsleft) == remaining_time @pytest.mark.skipif(not HAS_BATTERY, reason="no battery") def test_sensors_battery_against_sysctl(self): assert psutil.sensors_battery().percent == sysctl( "hw.acpi.battery.life" ) assert psutil.sensors_battery().power_plugged == ( sysctl("hw.acpi.acline") == 1 ) secsleft = psutil.sensors_battery().secsleft if secsleft < 0: assert sysctl("hw.acpi.battery.time") == -1 else: assert secsleft == sysctl("hw.acpi.battery.time") * 60 @pytest.mark.skipif(HAS_BATTERY, reason="has battery") def test_sensors_battery_no_battery(self): # If no battery is present one of these calls is supposed # to fail, see: # https://github.com/giampaolo/psutil/issues/1074 with pytest.raises(RuntimeError): sysctl("hw.acpi.battery.life") sysctl("hw.acpi.battery.time") sysctl("hw.acpi.acline") assert psutil.sensors_battery() is None # --- sensors_temperatures def test_sensors_temperatures_against_sysctl(self): num_cpus = psutil.cpu_count(True) for cpu in range(num_cpus): sensor = f"dev.cpu.{cpu}.temperature" # sysctl returns a string in the format 46.0C try: sysctl_result = int(float(sysctl(sensor)[:-1])) except RuntimeError: raise pytest.skip("temperatures not supported by kernel") assert ( abs( psutil.sensors_temperatures()["coretemp"][cpu].current - sysctl_result ) < 10 ) sensor = f"dev.cpu.{cpu}.coretemp.tjmax" sysctl_result = int(float(sysctl(sensor)[:-1])) assert ( psutil.sensors_temperatures()["coretemp"][cpu].high == sysctl_result ) # ===================================================================== # --- OpenBSD # ===================================================================== @pytest.mark.skipif(not OPENBSD, reason="OPENBSD only") class OpenBSDTestCase(PsutilTestCase): def test_boot_time(self): s = sysctl('kern.boottime') sys_bt = datetime.datetime.strptime(s, "%a %b %d %H:%M:%S %Y") psutil_bt = datetime.datetime.fromtimestamp(psutil.boot_time()) assert sys_bt == psutil_bt # ===================================================================== # --- NetBSD # ===================================================================== @pytest.mark.skipif(not NETBSD, reason="NETBSD only") class NetBSDTestCase(PsutilTestCase): @staticmethod def parse_meminfo(look_for): with open('/proc/meminfo') as f: for line in f: if line.startswith(look_for): return int(line.split()[1]) * 1024 raise ValueError(f"can't find {look_for}") # --- virtual mem def test_vmem_total(self): assert psutil.virtual_memory().total == self.parse_meminfo("MemTotal:") def test_vmem_free(self): assert ( abs(psutil.virtual_memory().free - self.parse_meminfo("MemFree:")) < TOLERANCE_SYS_MEM ) def test_vmem_buffers(self): assert ( abs( psutil.virtual_memory().buffers - self.parse_meminfo("Buffers:") ) < TOLERANCE_SYS_MEM ) def test_vmem_shared(self): assert ( abs( psutil.virtual_memory().shared - self.parse_meminfo("MemShared:") ) < TOLERANCE_SYS_MEM ) def test_vmem_cached(self): assert ( abs(psutil.virtual_memory().cached - self.parse_meminfo("Cached:")) < TOLERANCE_SYS_MEM ) # --- swap mem def test_swapmem_total(self): assert ( abs(psutil.swap_memory().total - self.parse_meminfo("SwapTotal:")) < TOLERANCE_SYS_MEM ) def test_swapmem_free(self): assert ( abs(psutil.swap_memory().free - self.parse_meminfo("SwapFree:")) < TOLERANCE_SYS_MEM ) def test_swapmem_used(self): smem = psutil.swap_memory() assert smem.used == smem.total - smem.free # --- others def test_cpu_stats_interrupts(self): with open('/proc/stat', 'rb') as f: for line in f: if line.startswith(b'intr'): interrupts = int(line.split()[1]) break else: raise ValueError("couldn't find line") assert abs(psutil.cpu_stats().interrupts - interrupts) < 1000 def test_cpu_stats_ctx_switches(self): with open('/proc/stat', 'rb') as f: for line in f: if line.startswith(b'ctxt'): ctx_switches = int(line.split()[1]) break else: raise ValueError("couldn't find line") assert abs(psutil.cpu_stats().ctx_switches - ctx_switches) < 1000 tests/test_connections.py 0000644 00000051245 15030211765 0011650 0 ustar 00 #!/usr/bin/env python3 # Copyright (c) 2009, Giampaolo Rodola'. All rights reserved. # Use of this source code is governed by a BSD-style license that can be # found in the LICENSE file. """Tests for psutil.net_connections() and Process.net_connections() APIs.""" import os import socket import textwrap from contextlib import closing from socket import AF_INET from socket import AF_INET6 from socket import SOCK_DGRAM from socket import SOCK_STREAM import psutil from psutil import FREEBSD from psutil import LINUX from psutil import MACOS from psutil import NETBSD from psutil import OPENBSD from psutil import POSIX from psutil import SUNOS from psutil import WINDOWS from psutil._common import supports_ipv6 from psutil.tests import AF_UNIX from psutil.tests import HAS_NET_CONNECTIONS_UNIX from psutil.tests import SKIP_SYSCONS from psutil.tests import PsutilTestCase from psutil.tests import bind_socket from psutil.tests import bind_unix_socket from psutil.tests import check_connection_ntuple from psutil.tests import create_sockets from psutil.tests import filter_proc_net_connections from psutil.tests import pytest from psutil.tests import reap_children from psutil.tests import retry_on_failure from psutil.tests import skip_on_access_denied from psutil.tests import tcp_socketpair from psutil.tests import unix_socketpair from psutil.tests import wait_for_file SOCK_SEQPACKET = getattr(socket, "SOCK_SEQPACKET", object()) def this_proc_net_connections(kind): cons = psutil.Process().net_connections(kind=kind) if kind in {"all", "unix"}: return filter_proc_net_connections(cons) return cons @pytest.mark.xdist_group(name="serial") class ConnectionTestCase(PsutilTestCase): def setUp(self): assert this_proc_net_connections(kind='all') == [] def tearDown(self): # Make sure we closed all resources. assert this_proc_net_connections(kind='all') == [] def compare_procsys_connections(self, pid, proc_cons, kind='all'): """Given a process PID and its list of connections compare those against system-wide connections retrieved via psutil.net_connections. """ try: sys_cons = psutil.net_connections(kind=kind) except psutil.AccessDenied: # On MACOS, system-wide connections are retrieved by iterating # over all processes if MACOS: return else: raise # Filter for this proc PID and exlucde PIDs from the tuple. sys_cons = [c[:-1] for c in sys_cons if c.pid == pid] sys_cons.sort() proc_cons.sort() assert proc_cons == sys_cons class TestBasicOperations(ConnectionTestCase): @pytest.mark.skipif(SKIP_SYSCONS, reason="requires root") def test_system(self): with create_sockets(): for conn in psutil.net_connections(kind='all'): check_connection_ntuple(conn) def test_process(self): with create_sockets(): for conn in this_proc_net_connections(kind='all'): check_connection_ntuple(conn) def test_invalid_kind(self): with pytest.raises(ValueError): this_proc_net_connections(kind='???') with pytest.raises(ValueError): psutil.net_connections(kind='???') @pytest.mark.xdist_group(name="serial") class TestUnconnectedSockets(ConnectionTestCase): """Tests sockets which are open but not connected to anything.""" def get_conn_from_sock(self, sock): cons = this_proc_net_connections(kind='all') smap = {c.fd: c for c in cons} if NETBSD or FREEBSD: # NetBSD opens a UNIX socket to /var/log/run # so there may be more connections. return smap[sock.fileno()] else: assert len(cons) == 1 if cons[0].fd != -1: assert smap[sock.fileno()].fd == sock.fileno() return cons[0] def check_socket(self, sock): """Given a socket, makes sure it matches the one obtained via psutil. It assumes this process created one connection only (the one supposed to be checked). """ conn = self.get_conn_from_sock(sock) check_connection_ntuple(conn) # fd, family, type if conn.fd != -1: assert conn.fd == sock.fileno() assert conn.family == sock.family # see: http://bugs.python.org/issue30204 assert conn.type == sock.getsockopt(socket.SOL_SOCKET, socket.SO_TYPE) # local address laddr = sock.getsockname() if not laddr and isinstance(laddr, bytes): # See: http://bugs.python.org/issue30205 laddr = laddr.decode() if sock.family == AF_INET6: laddr = laddr[:2] assert conn.laddr == laddr # XXX Solaris can't retrieve system-wide UNIX sockets if sock.family == AF_UNIX and HAS_NET_CONNECTIONS_UNIX: cons = this_proc_net_connections(kind='all') self.compare_procsys_connections(os.getpid(), cons, kind='all') return conn def test_tcp_v4(self): addr = ("127.0.0.1", 0) with closing(bind_socket(AF_INET, SOCK_STREAM, addr=addr)) as sock: conn = self.check_socket(sock) assert conn.raddr == () assert conn.status == psutil.CONN_LISTEN @pytest.mark.skipif(not supports_ipv6(), reason="IPv6 not supported") def test_tcp_v6(self): addr = ("::1", 0) with closing(bind_socket(AF_INET6, SOCK_STREAM, addr=addr)) as sock: conn = self.check_socket(sock) assert conn.raddr == () assert conn.status == psutil.CONN_LISTEN def test_udp_v4(self): addr = ("127.0.0.1", 0) with closing(bind_socket(AF_INET, SOCK_DGRAM, addr=addr)) as sock: conn = self.check_socket(sock) assert conn.raddr == () assert conn.status == psutil.CONN_NONE @pytest.mark.skipif(not supports_ipv6(), reason="IPv6 not supported") def test_udp_v6(self): addr = ("::1", 0) with closing(bind_socket(AF_INET6, SOCK_DGRAM, addr=addr)) as sock: conn = self.check_socket(sock) assert conn.raddr == () assert conn.status == psutil.CONN_NONE @pytest.mark.skipif(not POSIX, reason="POSIX only") def test_unix_tcp(self): testfn = self.get_testfn() with closing(bind_unix_socket(testfn, type=SOCK_STREAM)) as sock: conn = self.check_socket(sock) assert conn.raddr == "" assert conn.status == psutil.CONN_NONE @pytest.mark.skipif(not POSIX, reason="POSIX only") def test_unix_udp(self): testfn = self.get_testfn() with closing(bind_unix_socket(testfn, type=SOCK_STREAM)) as sock: conn = self.check_socket(sock) assert conn.raddr == "" assert conn.status == psutil.CONN_NONE @pytest.mark.xdist_group(name="serial") class TestConnectedSocket(ConnectionTestCase): """Test socket pairs which are actually connected to each other. """ # On SunOS, even after we close() it, the server socket stays around # in TIME_WAIT state. @pytest.mark.skipif(SUNOS, reason="unreliable on SUONS") def test_tcp(self): addr = ("127.0.0.1", 0) assert this_proc_net_connections(kind='tcp4') == [] server, client = tcp_socketpair(AF_INET, addr=addr) try: cons = this_proc_net_connections(kind='tcp4') assert len(cons) == 2 assert cons[0].status == psutil.CONN_ESTABLISHED assert cons[1].status == psutil.CONN_ESTABLISHED # May not be fast enough to change state so it stays # commenteed. # client.close() # cons = this_proc_net_connections(kind='all') # self.assertEqual(len(cons), 1) # self.assertEqual(cons[0].status, psutil.CONN_CLOSE_WAIT) finally: server.close() client.close() @pytest.mark.skipif(not POSIX, reason="POSIX only") def test_unix(self): testfn = self.get_testfn() server, client = unix_socketpair(testfn) try: cons = this_proc_net_connections(kind='unix') assert not (cons[0].laddr and cons[0].raddr), cons assert not (cons[1].laddr and cons[1].raddr), cons if NETBSD or FREEBSD: # On NetBSD creating a UNIX socket will cause # a UNIX connection to /var/run/log. cons = [c for c in cons if c.raddr != '/var/run/log'] assert len(cons) == 2 if LINUX or FREEBSD or SUNOS or OPENBSD: # remote path is never set assert cons[0].raddr == "" assert cons[1].raddr == "" # one local address should though assert testfn == (cons[0].laddr or cons[1].laddr) else: # On other systems either the laddr or raddr # of both peers are set. assert (cons[0].laddr or cons[1].laddr) == testfn finally: server.close() client.close() class TestFilters(ConnectionTestCase): def test_filters(self): def check(kind, families, types): for conn in this_proc_net_connections(kind=kind): assert conn.family in families assert conn.type in types if not SKIP_SYSCONS: for conn in psutil.net_connections(kind=kind): assert conn.family in families assert conn.type in types with create_sockets(): check( 'all', [AF_INET, AF_INET6, AF_UNIX], [SOCK_STREAM, SOCK_DGRAM, SOCK_SEQPACKET], ) check('inet', [AF_INET, AF_INET6], [SOCK_STREAM, SOCK_DGRAM]) check('inet4', [AF_INET], [SOCK_STREAM, SOCK_DGRAM]) check('tcp', [AF_INET, AF_INET6], [SOCK_STREAM]) check('tcp4', [AF_INET], [SOCK_STREAM]) check('tcp6', [AF_INET6], [SOCK_STREAM]) check('udp', [AF_INET, AF_INET6], [SOCK_DGRAM]) check('udp4', [AF_INET], [SOCK_DGRAM]) check('udp6', [AF_INET6], [SOCK_DGRAM]) if HAS_NET_CONNECTIONS_UNIX: check( 'unix', [AF_UNIX], [SOCK_STREAM, SOCK_DGRAM, SOCK_SEQPACKET], ) @skip_on_access_denied(only_if=MACOS) def test_combos(self): reap_children() def check_conn(proc, conn, family, type, laddr, raddr, status, kinds): all_kinds = ( "all", "inet", "inet4", "inet6", "tcp", "tcp4", "tcp6", "udp", "udp4", "udp6", ) check_connection_ntuple(conn) assert conn.family == family assert conn.type == type assert conn.laddr == laddr assert conn.raddr == raddr assert conn.status == status for kind in all_kinds: cons = proc.net_connections(kind=kind) if kind in kinds: assert cons != [] else: assert cons == [] # compare against system-wide connections # XXX Solaris can't retrieve system-wide UNIX # sockets. if HAS_NET_CONNECTIONS_UNIX: self.compare_procsys_connections(proc.pid, [conn]) tcp_template = textwrap.dedent(""" import socket, time s = socket.socket({family}, socket.SOCK_STREAM) s.bind(('{addr}', 0)) s.listen(5) with open('{testfn}', 'w') as f: f.write(str(s.getsockname()[:2])) [time.sleep(0.1) for x in range(100)] """) udp_template = textwrap.dedent(""" import socket, time s = socket.socket({family}, socket.SOCK_DGRAM) s.bind(('{addr}', 0)) with open('{testfn}', 'w') as f: f.write(str(s.getsockname()[:2])) [time.sleep(0.1) for x in range(100)] """) # must be relative on Windows testfile = os.path.basename(self.get_testfn(dir=os.getcwd())) tcp4_template = tcp_template.format( family=int(AF_INET), addr="127.0.0.1", testfn=testfile ) udp4_template = udp_template.format( family=int(AF_INET), addr="127.0.0.1", testfn=testfile ) tcp6_template = tcp_template.format( family=int(AF_INET6), addr="::1", testfn=testfile ) udp6_template = udp_template.format( family=int(AF_INET6), addr="::1", testfn=testfile ) # launch various subprocess instantiating a socket of various # families and types to enrich psutil results tcp4_proc = self.pyrun(tcp4_template) tcp4_addr = eval(wait_for_file(testfile, delete=True)) udp4_proc = self.pyrun(udp4_template) udp4_addr = eval(wait_for_file(testfile, delete=True)) if supports_ipv6(): tcp6_proc = self.pyrun(tcp6_template) tcp6_addr = eval(wait_for_file(testfile, delete=True)) udp6_proc = self.pyrun(udp6_template) udp6_addr = eval(wait_for_file(testfile, delete=True)) else: tcp6_proc = None udp6_proc = None tcp6_addr = None udp6_addr = None for p in psutil.Process().children(): cons = p.net_connections() assert len(cons) == 1 for conn in cons: # TCP v4 if p.pid == tcp4_proc.pid: check_conn( p, conn, AF_INET, SOCK_STREAM, tcp4_addr, (), psutil.CONN_LISTEN, ("all", "inet", "inet4", "tcp", "tcp4"), ) # UDP v4 elif p.pid == udp4_proc.pid: check_conn( p, conn, AF_INET, SOCK_DGRAM, udp4_addr, (), psutil.CONN_NONE, ("all", "inet", "inet4", "udp", "udp4"), ) # TCP v6 elif p.pid == getattr(tcp6_proc, "pid", None): check_conn( p, conn, AF_INET6, SOCK_STREAM, tcp6_addr, (), psutil.CONN_LISTEN, ("all", "inet", "inet6", "tcp", "tcp6"), ) # UDP v6 elif p.pid == getattr(udp6_proc, "pid", None): check_conn( p, conn, AF_INET6, SOCK_DGRAM, udp6_addr, (), psutil.CONN_NONE, ("all", "inet", "inet6", "udp", "udp6"), ) def test_count(self): with create_sockets(): # tcp cons = this_proc_net_connections(kind='tcp') assert len(cons) == (2 if supports_ipv6() else 1) for conn in cons: assert conn.family in {AF_INET, AF_INET6} assert conn.type == SOCK_STREAM # tcp4 cons = this_proc_net_connections(kind='tcp4') assert len(cons) == 1 assert cons[0].family == AF_INET assert cons[0].type == SOCK_STREAM # tcp6 if supports_ipv6(): cons = this_proc_net_connections(kind='tcp6') assert len(cons) == 1 assert cons[0].family == AF_INET6 assert cons[0].type == SOCK_STREAM # udp cons = this_proc_net_connections(kind='udp') assert len(cons) == (2 if supports_ipv6() else 1) for conn in cons: assert conn.family in {AF_INET, AF_INET6} assert conn.type == SOCK_DGRAM # udp4 cons = this_proc_net_connections(kind='udp4') assert len(cons) == 1 assert cons[0].family == AF_INET assert cons[0].type == SOCK_DGRAM # udp6 if supports_ipv6(): cons = this_proc_net_connections(kind='udp6') assert len(cons) == 1 assert cons[0].family == AF_INET6 assert cons[0].type == SOCK_DGRAM # inet cons = this_proc_net_connections(kind='inet') assert len(cons) == (4 if supports_ipv6() else 2) for conn in cons: assert conn.family in {AF_INET, AF_INET6} assert conn.type in {SOCK_STREAM, SOCK_DGRAM} # inet6 if supports_ipv6(): cons = this_proc_net_connections(kind='inet6') assert len(cons) == 2 for conn in cons: assert conn.family == AF_INET6 assert conn.type in {SOCK_STREAM, SOCK_DGRAM} # Skipped on BSD becayse by default the Python process # creates a UNIX socket to '/var/run/log'. if HAS_NET_CONNECTIONS_UNIX and not (FREEBSD or NETBSD): cons = this_proc_net_connections(kind='unix') assert len(cons) == 3 for conn in cons: assert conn.family == AF_UNIX assert conn.type in {SOCK_STREAM, SOCK_DGRAM} @pytest.mark.skipif(SKIP_SYSCONS, reason="requires root") class TestSystemWideConnections(ConnectionTestCase): """Tests for net_connections().""" def test_it(self): def check(cons, families, types_): for conn in cons: assert conn.family in families if conn.family != AF_UNIX: assert conn.type in types_ check_connection_ntuple(conn) with create_sockets(): from psutil._common import conn_tmap for kind, groups in conn_tmap.items(): # XXX: SunOS does not retrieve UNIX sockets. if kind == 'unix' and not HAS_NET_CONNECTIONS_UNIX: continue families, types_ = groups cons = psutil.net_connections(kind) assert len(cons) == len(set(cons)) check(cons, families, types_) @retry_on_failure() def test_multi_sockets_procs(self): # Creates multiple sub processes, each creating different # sockets. For each process check that proc.net_connections() # and psutil.net_connections() return the same results. # This is done mainly to check whether net_connections()'s # pid is properly set, see: # https://github.com/giampaolo/psutil/issues/1013 with create_sockets() as socks: expected = len(socks) pids = [] times = 10 fnames = [] for _ in range(times): fname = self.get_testfn() fnames.append(fname) src = textwrap.dedent(f"""\ import time, os from psutil.tests import create_sockets with create_sockets(): with open(r'{fname}', 'w') as f: f.write("hello") [time.sleep(0.1) for x in range(100)] """) sproc = self.pyrun(src) pids.append(sproc.pid) # sync for fname in fnames: wait_for_file(fname) syscons = [ x for x in psutil.net_connections(kind='all') if x.pid in pids ] for pid in pids: assert len([x for x in syscons if x.pid == pid]) == expected p = psutil.Process(pid) assert len(p.net_connections('all')) == expected class TestMisc(PsutilTestCase): def test_net_connection_constants(self): ints = [] strs = [] for name in dir(psutil): if name.startswith('CONN_'): num = getattr(psutil, name) str_ = str(num) assert str_.isupper(), str_ assert str not in strs assert num not in ints ints.append(num) strs.append(str_) if SUNOS: psutil.CONN_IDLE # noqa: B018 psutil.CONN_BOUND # noqa: B018 if WINDOWS: psutil.CONN_DELETE_TCB # noqa: B018 tests/test_linux.py 0000644 00000255477 15030211765 0010502 0 ustar 00 #!/usr/bin/env python3 # Copyright (c) 2009, Giampaolo Rodola'. All rights reserved. # Use of this source code is governed by a BSD-style license that can be # found in the LICENSE file. """Linux specific tests.""" import collections import contextlib import errno import io import os import platform import re import shutil import socket import struct import textwrap import time import warnings from unittest import mock import psutil from psutil import LINUX from psutil.tests import AARCH64 from psutil.tests import GITHUB_ACTIONS from psutil.tests import GLOBAL_TIMEOUT from psutil.tests import HAS_BATTERY from psutil.tests import HAS_CPU_FREQ from psutil.tests import HAS_GETLOADAVG from psutil.tests import HAS_RLIMIT from psutil.tests import PYPY from psutil.tests import PYTEST_PARALLEL from psutil.tests import TOLERANCE_DISK_USAGE from psutil.tests import TOLERANCE_SYS_MEM from psutil.tests import PsutilTestCase from psutil.tests import ThreadTask from psutil.tests import call_until from psutil.tests import pytest from psutil.tests import reload_module from psutil.tests import retry_on_failure from psutil.tests import safe_rmpath from psutil.tests import sh from psutil.tests import skip_on_not_implemented if LINUX: from psutil._pslinux import CLOCK_TICKS from psutil._pslinux import RootFsDeviceFinder from psutil._pslinux import calculate_avail_vmem from psutil._pslinux import open_binary HERE = os.path.abspath(os.path.dirname(__file__)) SIOCGIFADDR = 0x8915 SIOCGIFHWADDR = 0x8927 SIOCGIFNETMASK = 0x891B SIOCGIFBRDADDR = 0x8919 if LINUX: SECTOR_SIZE = 512 # ===================================================================== # --- utils # ===================================================================== def get_ipv4_address(ifname): import fcntl ifname = bytes(ifname[:15], "ascii") with socket.socket(socket.AF_INET, socket.SOCK_DGRAM) as s: return socket.inet_ntoa( fcntl.ioctl(s.fileno(), SIOCGIFADDR, struct.pack('256s', ifname))[ 20:24 ] ) def get_ipv4_netmask(ifname): import fcntl ifname = bytes(ifname[:15], "ascii") with socket.socket(socket.AF_INET, socket.SOCK_DGRAM) as s: return socket.inet_ntoa( fcntl.ioctl( s.fileno(), SIOCGIFNETMASK, struct.pack('256s', ifname) )[20:24] ) def get_ipv4_broadcast(ifname): import fcntl ifname = bytes(ifname[:15], "ascii") with socket.socket(socket.AF_INET, socket.SOCK_DGRAM) as s: return socket.inet_ntoa( fcntl.ioctl( s.fileno(), SIOCGIFBRDADDR, struct.pack('256s', ifname) )[20:24] ) def get_ipv6_addresses(ifname): with open("/proc/net/if_inet6") as f: all_fields = [] for line in f: fields = line.split() if fields[-1] == ifname: all_fields.append(fields) if len(all_fields) == 0: raise ValueError(f"could not find interface {ifname!r}") for i in range(len(all_fields)): unformatted = all_fields[i][0] groups = [ unformatted[j : j + 4] for j in range(0, len(unformatted), 4) ] formatted = ":".join(groups) packed = socket.inet_pton(socket.AF_INET6, formatted) all_fields[i] = socket.inet_ntop(socket.AF_INET6, packed) return all_fields def get_mac_address(ifname): import fcntl ifname = bytes(ifname[:15], "ascii") with socket.socket(socket.AF_INET, socket.SOCK_DGRAM) as s: info = fcntl.ioctl( s.fileno(), SIOCGIFHWADDR, struct.pack('256s', ifname) ) return "".join([f"{char:02x}:" for char in info[18:24]])[:-1] def free_swap(): """Parse 'free' cmd and return swap memory's s total, used and free values. """ out = sh(["free", "-b"], env={"LANG": "C.UTF-8"}) lines = out.split('\n') for line in lines: if line.startswith('Swap'): _, total, used, free = line.split() nt = collections.namedtuple('free', 'total used free') return nt(int(total), int(used), int(free)) raise ValueError(f"can't find 'Swap' in 'free' output:\n{out}") def free_physmem(): """Parse 'free' cmd and return physical memory's total, used and free values. """ # Note: free can have 2 different formats, invalidating 'shared' # and 'cached' memory which may have different positions so we # do not return them. # https://github.com/giampaolo/psutil/issues/538#issuecomment-57059946 out = sh(["free", "-b"], env={"LANG": "C.UTF-8"}) lines = out.split('\n') for line in lines: if line.startswith('Mem'): total, used, free, shared = (int(x) for x in line.split()[1:5]) nt = collections.namedtuple( 'free', 'total used free shared output' ) return nt(total, used, free, shared, out) raise ValueError(f"can't find 'Mem' in 'free' output:\n{out}") def vmstat(stat): out = sh(["vmstat", "-s"], env={"LANG": "C.UTF-8"}) for line in out.split("\n"): line = line.strip() if stat in line: return int(line.split(' ')[0]) raise ValueError(f"can't find {stat!r} in 'vmstat' output") def get_free_version_info(): out = sh(["free", "-V"]).strip() if 'UNKNOWN' in out: raise pytest.skip("can't determine free version") return tuple(map(int, re.findall(r'\d+', out.split()[-1]))) @contextlib.contextmanager def mock_open_content(pairs): """Mock open() builtin and forces it to return a certain content for a given path. `pairs` is a {"path": "content", ...} dict. """ def open_mock(name, *args, **kwargs): if name in pairs: content = pairs[name] if isinstance(content, str): return io.StringIO(content) else: return io.BytesIO(content) else: return orig_open(name, *args, **kwargs) orig_open = open with mock.patch("builtins.open", create=True, side_effect=open_mock) as m: yield m @contextlib.contextmanager def mock_open_exception(for_path, exc): """Mock open() builtin and raises `exc` if the path being opened matches `for_path`. """ def open_mock(name, *args, **kwargs): if name == for_path: raise exc return orig_open(name, *args, **kwargs) orig_open = open with mock.patch("builtins.open", create=True, side_effect=open_mock) as m: yield m # ===================================================================== # --- system virtual memory # ===================================================================== @pytest.mark.skipif(not LINUX, reason="LINUX only") class TestSystemVirtualMemoryAgainstFree(PsutilTestCase): def test_total(self): cli_value = free_physmem().total psutil_value = psutil.virtual_memory().total assert cli_value == psutil_value @retry_on_failure() def test_used(self): # Older versions of procps used slab memory to calculate used memory. # This got changed in: # https://gitlab.com/procps-ng/procps/commit/ # 05d751c4f076a2f0118b914c5e51cfbb4762ad8e # Newer versions of procps are using yet another way to compute used # memory. # https://gitlab.com/procps-ng/procps/commit/ # 2184e90d2e7cdb582f9a5b706b47015e56707e4d if get_free_version_info() < (3, 3, 12): raise pytest.skip("free version too old") if get_free_version_info() >= (4, 0, 0): raise pytest.skip("free version too recent") cli_value = free_physmem().used psutil_value = psutil.virtual_memory().used assert abs(cli_value - psutil_value) < TOLERANCE_SYS_MEM @retry_on_failure() def test_free(self): cli_value = free_physmem().free psutil_value = psutil.virtual_memory().free assert abs(cli_value - psutil_value) < TOLERANCE_SYS_MEM @retry_on_failure() def test_shared(self): free = free_physmem() free_value = free.shared if free_value == 0: raise pytest.skip("free does not support 'shared' column") psutil_value = psutil.virtual_memory().shared assert ( abs(free_value - psutil_value) < TOLERANCE_SYS_MEM ), f"{free_value} {psutil_value} \n{free.output}" @retry_on_failure() def test_available(self): # "free" output format has changed at some point: # https://github.com/giampaolo/psutil/issues/538#issuecomment-147192098 out = sh(["free", "-b"]) lines = out.split('\n') if 'available' not in lines[0]: raise pytest.skip("free does not support 'available' column") free_value = int(lines[1].split()[-1]) psutil_value = psutil.virtual_memory().available assert abs(free_value - psutil_value) < TOLERANCE_SYS_MEM @pytest.mark.skipif(not LINUX, reason="LINUX only") class TestSystemVirtualMemoryAgainstVmstat(PsutilTestCase): def test_total(self): vmstat_value = vmstat('total memory') * 1024 psutil_value = psutil.virtual_memory().total assert abs(vmstat_value - psutil_value) < TOLERANCE_SYS_MEM @retry_on_failure() def test_used(self): # Older versions of procps used slab memory to calculate used memory. # This got changed in: # https://gitlab.com/procps-ng/procps/commit/ # 05d751c4f076a2f0118b914c5e51cfbb4762ad8e # Newer versions of procps are using yet another way to compute used # memory. # https://gitlab.com/procps-ng/procps/commit/ # 2184e90d2e7cdb582f9a5b706b47015e56707e4d if get_free_version_info() < (3, 3, 12): raise pytest.skip("free version too old") if get_free_version_info() >= (4, 0, 0): raise pytest.skip("free version too recent") vmstat_value = vmstat('used memory') * 1024 psutil_value = psutil.virtual_memory().used assert abs(vmstat_value - psutil_value) < TOLERANCE_SYS_MEM @retry_on_failure() def test_free(self): vmstat_value = vmstat('free memory') * 1024 psutil_value = psutil.virtual_memory().free assert abs(vmstat_value - psutil_value) < TOLERANCE_SYS_MEM @retry_on_failure() def test_buffers(self): vmstat_value = vmstat('buffer memory') * 1024 psutil_value = psutil.virtual_memory().buffers assert abs(vmstat_value - psutil_value) < TOLERANCE_SYS_MEM @retry_on_failure() def test_active(self): vmstat_value = vmstat('active memory') * 1024 psutil_value = psutil.virtual_memory().active assert abs(vmstat_value - psutil_value) < TOLERANCE_SYS_MEM @retry_on_failure() def test_inactive(self): vmstat_value = vmstat('inactive memory') * 1024 psutil_value = psutil.virtual_memory().inactive assert abs(vmstat_value - psutil_value) < TOLERANCE_SYS_MEM @pytest.mark.skipif(not LINUX, reason="LINUX only") class TestSystemVirtualMemoryMocks(PsutilTestCase): def test_warnings_on_misses(self): # Emulate a case where /proc/meminfo provides few info. # psutil is supposed to set the missing fields to 0 and # raise a warning. content = textwrap.dedent("""\ Active(anon): 6145416 kB Active(file): 2950064 kB Inactive(anon): 574764 kB Inactive(file): 1567648 kB MemAvailable: -1 kB MemFree: 2057400 kB MemTotal: 16325648 kB SReclaimable: 346648 kB """).encode() with mock_open_content({'/proc/meminfo': content}) as m: with warnings.catch_warnings(record=True) as ws: warnings.simplefilter("always") ret = psutil.virtual_memory() assert m.called assert len(ws) == 1 w = ws[0] assert "memory stats couldn't be determined" in str(w.message) assert "cached" in str(w.message) assert "shared" in str(w.message) assert "active" in str(w.message) assert "inactive" in str(w.message) assert "buffers" in str(w.message) assert "available" in str(w.message) assert ret.cached == 0 assert ret.active == 0 assert ret.inactive == 0 assert ret.shared == 0 assert ret.buffers == 0 assert ret.available == 0 assert ret.slab == 0 @retry_on_failure() def test_avail_old_percent(self): # Make sure that our calculation of avail mem for old kernels # is off by max 15%. mems = {} with open_binary('/proc/meminfo') as f: for line in f: fields = line.split() mems[fields[0]] = int(fields[1]) * 1024 a = calculate_avail_vmem(mems) if b'MemAvailable:' in mems: b = mems[b'MemAvailable:'] diff_percent = abs(a - b) / a * 100 assert diff_percent < 15 def test_avail_old_comes_from_kernel(self): # Make sure "MemAvailable:" coluimn is used instead of relying # on our internal algorithm to calculate avail mem. content = textwrap.dedent("""\ Active: 9444728 kB Active(anon): 6145416 kB Active(file): 2950064 kB Buffers: 287952 kB Cached: 4818144 kB Inactive(file): 1578132 kB Inactive(anon): 574764 kB Inactive(file): 1567648 kB MemAvailable: 6574984 kB MemFree: 2057400 kB MemTotal: 16325648 kB Shmem: 577588 kB SReclaimable: 346648 kB """).encode() with mock_open_content({'/proc/meminfo': content}) as m: with warnings.catch_warnings(record=True) as ws: ret = psutil.virtual_memory() assert m.called assert ret.available == 6574984 * 1024 w = ws[0] assert "inactive memory stats couldn't be determined" in str( w.message ) def test_avail_old_missing_fields(self): # Remove Active(file), Inactive(file) and SReclaimable # from /proc/meminfo and make sure the fallback is used # (free + cached), content = textwrap.dedent("""\ Active: 9444728 kB Active(anon): 6145416 kB Buffers: 287952 kB Cached: 4818144 kB Inactive(file): 1578132 kB Inactive(anon): 574764 kB MemFree: 2057400 kB MemTotal: 16325648 kB Shmem: 577588 kB """).encode() with mock_open_content({"/proc/meminfo": content}) as m: with warnings.catch_warnings(record=True) as ws: ret = psutil.virtual_memory() assert m.called assert ret.available == 2057400 * 1024 + 4818144 * 1024 w = ws[0] assert "inactive memory stats couldn't be determined" in str( w.message ) def test_avail_old_missing_zoneinfo(self): # Remove /proc/zoneinfo file. Make sure fallback is used # (free + cached). content = textwrap.dedent("""\ Active: 9444728 kB Active(anon): 6145416 kB Active(file): 2950064 kB Buffers: 287952 kB Cached: 4818144 kB Inactive(file): 1578132 kB Inactive(anon): 574764 kB Inactive(file): 1567648 kB MemFree: 2057400 kB MemTotal: 16325648 kB Shmem: 577588 kB SReclaimable: 346648 kB """).encode() with mock_open_content({"/proc/meminfo": content}): with mock_open_exception("/proc/zoneinfo", FileNotFoundError): with warnings.catch_warnings(record=True) as ws: ret = psutil.virtual_memory() assert ret.available == 2057400 * 1024 + 4818144 * 1024 w = ws[0] assert ( "inactive memory stats couldn't be determined" in str(w.message) ) def test_virtual_memory_mocked(self): # Emulate /proc/meminfo because neither vmstat nor free return slab. content = textwrap.dedent("""\ MemTotal: 100 kB MemFree: 2 kB MemAvailable: 3 kB Buffers: 4 kB Cached: 5 kB SwapCached: 6 kB Active: 7 kB Inactive: 8 kB Active(anon): 9 kB Inactive(anon): 10 kB Active(file): 11 kB Inactive(file): 12 kB Unevictable: 13 kB Mlocked: 14 kB SwapTotal: 15 kB SwapFree: 16 kB Dirty: 17 kB Writeback: 18 kB AnonPages: 19 kB Mapped: 20 kB Shmem: 21 kB Slab: 22 kB SReclaimable: 23 kB SUnreclaim: 24 kB KernelStack: 25 kB PageTables: 26 kB NFS_Unstable: 27 kB Bounce: 28 kB WritebackTmp: 29 kB CommitLimit: 30 kB Committed_AS: 31 kB VmallocTotal: 32 kB VmallocUsed: 33 kB VmallocChunk: 34 kB HardwareCorrupted: 35 kB AnonHugePages: 36 kB ShmemHugePages: 37 kB ShmemPmdMapped: 38 kB CmaTotal: 39 kB CmaFree: 40 kB HugePages_Total: 41 kB HugePages_Free: 42 kB HugePages_Rsvd: 43 kB HugePages_Surp: 44 kB Hugepagesize: 45 kB DirectMap46k: 46 kB DirectMap47M: 47 kB DirectMap48G: 48 kB """).encode() with mock_open_content({"/proc/meminfo": content}) as m: mem = psutil.virtual_memory() assert m.called assert mem.total == 100 * 1024 assert mem.free == 2 * 1024 assert mem.buffers == 4 * 1024 # cached mem also includes reclaimable memory assert mem.cached == (5 + 23) * 1024 assert mem.shared == 21 * 1024 assert mem.active == 7 * 1024 assert mem.inactive == 8 * 1024 assert mem.slab == 22 * 1024 assert mem.available == 3 * 1024 # ===================================================================== # --- system swap memory # ===================================================================== @pytest.mark.skipif(not LINUX, reason="LINUX only") class TestSystemSwapMemory(PsutilTestCase): @staticmethod def meminfo_has_swap_info(): """Return True if /proc/meminfo provides swap metrics.""" with open("/proc/meminfo") as f: data = f.read() return 'SwapTotal:' in data and 'SwapFree:' in data def test_total(self): free_value = free_swap().total psutil_value = psutil.swap_memory().total assert abs(free_value - psutil_value) < TOLERANCE_SYS_MEM @retry_on_failure() def test_used(self): free_value = free_swap().used psutil_value = psutil.swap_memory().used assert abs(free_value - psutil_value) < TOLERANCE_SYS_MEM @retry_on_failure() def test_free(self): free_value = free_swap().free psutil_value = psutil.swap_memory().free assert abs(free_value - psutil_value) < TOLERANCE_SYS_MEM def test_missing_sin_sout(self): with mock.patch('psutil._common.open', create=True) as m: with warnings.catch_warnings(record=True) as ws: warnings.simplefilter("always") ret = psutil.swap_memory() assert m.called assert len(ws) == 1 w = ws[0] assert ( "'sin' and 'sout' swap memory stats couldn't be determined" in str(w.message) ) assert ret.sin == 0 assert ret.sout == 0 def test_no_vmstat_mocked(self): # see https://github.com/giampaolo/psutil/issues/722 with mock_open_exception("/proc/vmstat", FileNotFoundError) as m: with warnings.catch_warnings(record=True) as ws: warnings.simplefilter("always") ret = psutil.swap_memory() assert m.called assert len(ws) == 1 w = ws[0] assert ( "'sin' and 'sout' swap memory stats couldn't " "be determined and were set to 0" in str(w.message) ) assert ret.sin == 0 assert ret.sout == 0 def test_meminfo_against_sysinfo(self): # Make sure the content of /proc/meminfo about swap memory # matches sysinfo() syscall, see: # https://github.com/giampaolo/psutil/issues/1015 if not self.meminfo_has_swap_info(): raise pytest.skip("/proc/meminfo has no swap metrics") with mock.patch('psutil._pslinux.cext.linux_sysinfo') as m: swap = psutil.swap_memory() assert not m.called import psutil._psutil_linux as cext _, _, _, _, total, free, unit_multiplier = cext.linux_sysinfo() total *= unit_multiplier free *= unit_multiplier assert swap.total == total assert abs(swap.free - free) < TOLERANCE_SYS_MEM def test_emulate_meminfo_has_no_metrics(self): # Emulate a case where /proc/meminfo provides no swap metrics # in which case sysinfo() syscall is supposed to be used # as a fallback. with mock_open_content({"/proc/meminfo": b""}) as m: psutil.swap_memory() assert m.called # ===================================================================== # --- system CPU # ===================================================================== @pytest.mark.skipif(not LINUX, reason="LINUX only") class TestSystemCPUTimes(PsutilTestCase): def test_fields(self): fields = psutil.cpu_times()._fields kernel_ver = re.findall(r'\d+\.\d+\.\d+', os.uname()[2])[0] kernel_ver_info = tuple(map(int, kernel_ver.split('.'))) if kernel_ver_info >= (2, 6, 11): assert 'steal' in fields else: assert 'steal' not in fields if kernel_ver_info >= (2, 6, 24): assert 'guest' in fields else: assert 'guest' not in fields if kernel_ver_info >= (3, 2, 0): assert 'guest_nice' in fields else: assert 'guest_nice' not in fields @pytest.mark.skipif(not LINUX, reason="LINUX only") class TestSystemCPUCountLogical(PsutilTestCase): @pytest.mark.skipif( not os.path.exists("/sys/devices/system/cpu/online"), reason="/sys/devices/system/cpu/online does not exist", ) def test_against_sysdev_cpu_online(self): with open("/sys/devices/system/cpu/online") as f: value = f.read().strip() if "-" in str(value): value = int(value.split('-')[1]) + 1 assert psutil.cpu_count() == value @pytest.mark.skipif( not os.path.exists("/sys/devices/system/cpu"), reason="/sys/devices/system/cpu does not exist", ) def test_against_sysdev_cpu_num(self): ls = os.listdir("/sys/devices/system/cpu") count = len([x for x in ls if re.search(r"cpu\d+$", x) is not None]) assert psutil.cpu_count() == count @pytest.mark.skipif( not shutil.which("nproc"), reason="nproc utility not available" ) def test_against_nproc(self): num = int(sh("nproc --all")) assert psutil.cpu_count(logical=True) == num @pytest.mark.skipif( not shutil.which("lscpu"), reason="lscpu utility not available" ) def test_against_lscpu(self): out = sh("lscpu -p") num = len([x for x in out.split('\n') if not x.startswith('#')]) assert psutil.cpu_count(logical=True) == num def test_emulate_fallbacks(self): import psutil._pslinux original = psutil._pslinux.cpu_count_logical() # Here we want to mock os.sysconf("SC_NPROCESSORS_ONLN") in # order to cause the parsing of /proc/cpuinfo and /proc/stat. with mock.patch( 'psutil._pslinux.os.sysconf', side_effect=ValueError ) as m: assert psutil._pslinux.cpu_count_logical() == original assert m.called # Let's have open() return empty data and make sure None is # returned ('cause we mimic os.cpu_count()). with mock.patch('psutil._common.open', create=True) as m: assert psutil._pslinux.cpu_count_logical() is None assert m.call_count == 2 # /proc/stat should be the last one assert m.call_args[0][0] == '/proc/stat' # Let's push this a bit further and make sure /proc/cpuinfo # parsing works as expected. with open('/proc/cpuinfo', 'rb') as f: cpuinfo_data = f.read() fake_file = io.BytesIO(cpuinfo_data) with mock.patch( 'psutil._common.open', return_value=fake_file, create=True ) as m: assert psutil._pslinux.cpu_count_logical() == original # Finally, let's make /proc/cpuinfo return meaningless data; # this way we'll fall back on relying on /proc/stat with mock_open_content({"/proc/cpuinfo": b""}) as m: assert psutil._pslinux.cpu_count_logical() == original assert m.called @pytest.mark.skipif(not LINUX, reason="LINUX only") class TestSystemCPUCountCores(PsutilTestCase): @pytest.mark.skipif( not shutil.which("lscpu"), reason="lscpu utility not available" ) def test_against_lscpu(self): out = sh("lscpu -p") core_ids = set() for line in out.split('\n'): if not line.startswith('#'): fields = line.split(',') core_ids.add(fields[1]) assert psutil.cpu_count(logical=False) == len(core_ids) @pytest.mark.skipif( platform.machine() not in {"x86_64", "i686"}, reason="x86_64/i686 only" ) def test_method_2(self): meth_1 = psutil._pslinux.cpu_count_cores() with mock.patch('glob.glob', return_value=[]) as m: meth_2 = psutil._pslinux.cpu_count_cores() assert m.called if meth_1 is not None: assert meth_1 == meth_2 def test_emulate_none(self): with mock.patch('glob.glob', return_value=[]) as m1: with mock.patch('psutil._common.open', create=True) as m2: assert psutil._pslinux.cpu_count_cores() is None assert m1.called assert m2.called @pytest.mark.skipif(not LINUX, reason="LINUX only") class TestSystemCPUFrequency(PsutilTestCase): @pytest.mark.skipif(not HAS_CPU_FREQ, reason="not supported") @pytest.mark.skipif( AARCH64, reason="aarch64 does not always expose frequency" ) def test_emulate_use_second_file(self): # https://github.com/giampaolo/psutil/issues/981 def path_exists_mock(path): if path.startswith("/sys/devices/system/cpu/cpufreq/policy"): return False else: return orig_exists(path) orig_exists = os.path.exists with mock.patch( "os.path.exists", side_effect=path_exists_mock, create=True ): assert psutil.cpu_freq() @pytest.mark.skipif(not HAS_CPU_FREQ, reason="not supported") @pytest.mark.skipif( AARCH64, reason="aarch64 does not report mhz in /proc/cpuinfo" ) def test_emulate_use_cpuinfo(self): # Emulate a case where /sys/devices/system/cpu/cpufreq* does not # exist and /proc/cpuinfo is used instead. def path_exists_mock(path): if path.startswith('/sys/devices/system/cpu/'): return False else: return os_path_exists(path) os_path_exists = os.path.exists try: with mock.patch("os.path.exists", side_effect=path_exists_mock): reload_module(psutil._pslinux) ret = psutil.cpu_freq() assert ret, ret assert ret.max == 0.0 assert ret.min == 0.0 for freq in psutil.cpu_freq(percpu=True): assert freq.max == 0.0 assert freq.min == 0.0 finally: reload_module(psutil._pslinux) reload_module(psutil) @pytest.mark.skipif(not HAS_CPU_FREQ, reason="not supported") def test_emulate_data(self): def open_mock(name, *args, **kwargs): if name.endswith('/scaling_cur_freq') and name.startswith( "/sys/devices/system/cpu/cpufreq/policy" ): return io.BytesIO(b"500000") elif name.endswith('/scaling_min_freq') and name.startswith( "/sys/devices/system/cpu/cpufreq/policy" ): return io.BytesIO(b"600000") elif name.endswith('/scaling_max_freq') and name.startswith( "/sys/devices/system/cpu/cpufreq/policy" ): return io.BytesIO(b"700000") elif name == '/proc/cpuinfo': return io.BytesIO(b"cpu MHz : 500") else: return orig_open(name, *args, **kwargs) orig_open = open with mock.patch("builtins.open", side_effect=open_mock): with mock.patch('os.path.exists', return_value=True): freq = psutil.cpu_freq() assert freq.current == 500.0 # when /proc/cpuinfo is used min and max frequencies are not # available and are set to 0. if freq.min != 0.0: assert freq.min == 600.0 if freq.max != 0.0: assert freq.max == 700.0 @pytest.mark.skipif(not HAS_CPU_FREQ, reason="not supported") def test_emulate_multi_cpu(self): def open_mock(name, *args, **kwargs): n = name if n.endswith('/scaling_cur_freq') and n.startswith( "/sys/devices/system/cpu/cpufreq/policy0" ): return io.BytesIO(b"100000") elif n.endswith('/scaling_min_freq') and n.startswith( "/sys/devices/system/cpu/cpufreq/policy0" ): return io.BytesIO(b"200000") elif n.endswith('/scaling_max_freq') and n.startswith( "/sys/devices/system/cpu/cpufreq/policy0" ): return io.BytesIO(b"300000") elif n.endswith('/scaling_cur_freq') and n.startswith( "/sys/devices/system/cpu/cpufreq/policy1" ): return io.BytesIO(b"400000") elif n.endswith('/scaling_min_freq') and n.startswith( "/sys/devices/system/cpu/cpufreq/policy1" ): return io.BytesIO(b"500000") elif n.endswith('/scaling_max_freq') and n.startswith( "/sys/devices/system/cpu/cpufreq/policy1" ): return io.BytesIO(b"600000") elif name == '/proc/cpuinfo': return io.BytesIO(b"cpu MHz : 100\ncpu MHz : 400") else: return orig_open(name, *args, **kwargs) orig_open = open with mock.patch("builtins.open", side_effect=open_mock): with mock.patch('os.path.exists', return_value=True): with mock.patch( 'psutil._pslinux.cpu_count_logical', return_value=2 ): freq = psutil.cpu_freq(percpu=True) assert freq[0].current == 100.0 if freq[0].min != 0.0: assert freq[0].min == 200.0 if freq[0].max != 0.0: assert freq[0].max == 300.0 assert freq[1].current == 400.0 if freq[1].min != 0.0: assert freq[1].min == 500.0 if freq[1].max != 0.0: assert freq[1].max == 600.0 @pytest.mark.skipif(not HAS_CPU_FREQ, reason="not supported") def test_emulate_no_scaling_cur_freq_file(self): # See: https://github.com/giampaolo/psutil/issues/1071 def open_mock(name, *args, **kwargs): if name.endswith('/scaling_cur_freq'): raise FileNotFoundError if name.endswith('/cpuinfo_cur_freq'): return io.BytesIO(b"200000") elif name == '/proc/cpuinfo': return io.BytesIO(b"cpu MHz : 200") else: return orig_open(name, *args, **kwargs) orig_open = open with mock.patch("builtins.open", side_effect=open_mock): with mock.patch('os.path.exists', return_value=True): with mock.patch( 'psutil._pslinux.cpu_count_logical', return_value=1 ): freq = psutil.cpu_freq() assert freq.current == 200 @pytest.mark.skipif(not LINUX, reason="LINUX only") class TestSystemCPUStats(PsutilTestCase): # XXX: fails too often. # def test_ctx_switches(self): # vmstat_value = vmstat("context switches") # psutil_value = psutil.cpu_stats().ctx_switches # self.assertAlmostEqual(vmstat_value, psutil_value, delta=500) def test_interrupts(self): vmstat_value = vmstat("interrupts") psutil_value = psutil.cpu_stats().interrupts assert abs(vmstat_value - psutil_value) < 500 @pytest.mark.skipif(not LINUX, reason="LINUX only") class TestLoadAvg(PsutilTestCase): @pytest.mark.skipif(not HAS_GETLOADAVG, reason="not supported") def test_getloadavg(self): psutil_value = psutil.getloadavg() with open("/proc/loadavg") as f: proc_value = f.read().split() assert abs(float(proc_value[0]) - psutil_value[0]) < 1 assert abs(float(proc_value[1]) - psutil_value[1]) < 1 assert abs(float(proc_value[2]) - psutil_value[2]) < 1 # ===================================================================== # --- system network # ===================================================================== @pytest.mark.skipif(not LINUX, reason="LINUX only") class TestSystemNetIfAddrs(PsutilTestCase): def test_ips(self): for name, addrs in psutil.net_if_addrs().items(): for addr in addrs: if addr.family == psutil.AF_LINK: assert addr.address == get_mac_address(name) elif addr.family == socket.AF_INET: assert addr.address == get_ipv4_address(name) assert addr.netmask == get_ipv4_netmask(name) if addr.broadcast is not None: assert addr.broadcast == get_ipv4_broadcast(name) else: assert get_ipv4_broadcast(name) == '0.0.0.0' elif addr.family == socket.AF_INET6: # IPv6 addresses can have a percent symbol at the end. # E.g. these 2 are equivalent: # "fe80::1ff:fe23:4567:890a" # "fe80::1ff:fe23:4567:890a%eth0" # That is the "zone id" portion, which usually is the name # of the network interface. address = addr.address.split('%')[0] assert address in get_ipv6_addresses(name) # XXX - not reliable when having virtual NICs installed by Docker. # @pytest.mark.skipif(not shutil.which("ip"), # reason="'ip' utility not available") # def test_net_if_names(self): # out = sh("ip addr").strip() # nics = [x for x in psutil.net_if_addrs().keys() if ':' not in x] # found = 0 # for line in out.split('\n'): # line = line.strip() # if re.search(r"^\d+:", line): # found += 1 # name = line.split(':')[1].strip() # self.assertIn(name, nics) # self.assertEqual(len(nics), found, msg="{}\n---\n{}".format( # pprint.pformat(nics), out)) @pytest.mark.skipif(not LINUX, reason="LINUX only") class TestSystemNetIfStats(PsutilTestCase): @pytest.mark.skipif( not shutil.which("ifconfig"), reason="ifconfig utility not available" ) def test_against_ifconfig(self): for name, stats in psutil.net_if_stats().items(): try: out = sh(f"ifconfig {name}") except RuntimeError: pass else: assert stats.isup == ('RUNNING' in out), out assert stats.mtu == int( re.findall(r'(?i)MTU[: ](\d+)', out)[0] ) def test_mtu(self): for name, stats in psutil.net_if_stats().items(): with open(f"/sys/class/net/{name}/mtu") as f: assert stats.mtu == int(f.read().strip()) @pytest.mark.skipif( not shutil.which("ifconfig"), reason="ifconfig utility not available" ) def test_flags(self): # first line looks like this: # "eth0: flags=4163<UP,BROADCAST,RUNNING,MULTICAST> mtu 1500" matches_found = 0 for name, stats in psutil.net_if_stats().items(): try: out = sh(f"ifconfig {name}") except RuntimeError: pass else: match = re.search(r"flags=(\d+)?<(.*?)>", out) if match and len(match.groups()) >= 2: matches_found += 1 ifconfig_flags = set(match.group(2).lower().split(",")) psutil_flags = set(stats.flags.split(",")) assert ifconfig_flags == psutil_flags else: # ifconfig has a different output on CentOS 6 # let's try that match = re.search(r"(.*) MTU:(\d+) Metric:(\d+)", out) if match and len(match.groups()) >= 3: matches_found += 1 ifconfig_flags = set(match.group(1).lower().split()) psutil_flags = set(stats.flags.split(",")) assert ifconfig_flags == psutil_flags if not matches_found: raise self.fail("no matches were found") @pytest.mark.skipif(not LINUX, reason="LINUX only") class TestSystemNetIOCounters(PsutilTestCase): @pytest.mark.skipif( not shutil.which("ifconfig"), reason="ifconfig utility not available" ) @retry_on_failure() def test_against_ifconfig(self): def ifconfig(nic): ret = {} out = sh(f"ifconfig {nic}") ret['packets_recv'] = int( re.findall(r'RX packets[: ](\d+)', out)[0] ) ret['packets_sent'] = int( re.findall(r'TX packets[: ](\d+)', out)[0] ) ret['errin'] = int(re.findall(r'errors[: ](\d+)', out)[0]) ret['errout'] = int(re.findall(r'errors[: ](\d+)', out)[1]) ret['dropin'] = int(re.findall(r'dropped[: ](\d+)', out)[0]) ret['dropout'] = int(re.findall(r'dropped[: ](\d+)', out)[1]) ret['bytes_recv'] = int( re.findall(r'RX (?:packets \d+ +)?bytes[: ](\d+)', out)[0] ) ret['bytes_sent'] = int( re.findall(r'TX (?:packets \d+ +)?bytes[: ](\d+)', out)[0] ) return ret nio = psutil.net_io_counters(pernic=True, nowrap=False) for name, stats in nio.items(): try: ifconfig_ret = ifconfig(name) except RuntimeError: continue assert ( abs(stats.bytes_recv - ifconfig_ret['bytes_recv']) < 1024 * 10 ) assert ( abs(stats.bytes_sent - ifconfig_ret['bytes_sent']) < 1024 * 10 ) assert ( abs(stats.packets_recv - ifconfig_ret['packets_recv']) < 1024 ) assert ( abs(stats.packets_sent - ifconfig_ret['packets_sent']) < 1024 ) assert abs(stats.errin - ifconfig_ret['errin']) < 10 assert abs(stats.errout - ifconfig_ret['errout']) < 10 assert abs(stats.dropin - ifconfig_ret['dropin']) < 10 assert abs(stats.dropout - ifconfig_ret['dropout']) < 10 @pytest.mark.skipif(not LINUX, reason="LINUX only") class TestSystemNetConnections(PsutilTestCase): @mock.patch('psutil._pslinux.socket.inet_ntop', side_effect=ValueError) @mock.patch('psutil._pslinux.supports_ipv6', return_value=False) def test_emulate_ipv6_unsupported(self, supports_ipv6, inet_ntop): # see: https://github.com/giampaolo/psutil/issues/623 try: s = socket.socket(socket.AF_INET6, socket.SOCK_STREAM) self.addCleanup(s.close) s.bind(("::1", 0)) except OSError: pass psutil.net_connections(kind='inet6') def test_emulate_unix(self): content = textwrap.dedent("""\ 0: 00000003 000 000 0001 03 462170 @/tmp/dbus-Qw2hMPIU3n 0: 00000003 000 000 0001 03 35010 @/tmp/dbus-tB2X8h69BQ 0: 00000003 000 000 0001 03 34424 @/tmp/dbus-cHy80Y8O 000000000000000000000000000000000000000000000000000000 """) with mock_open_content({"/proc/net/unix": content}) as m: psutil.net_connections(kind='unix') assert m.called # ===================================================================== # --- system disks # ===================================================================== @pytest.mark.skipif(not LINUX, reason="LINUX only") class TestSystemDiskPartitions(PsutilTestCase): @pytest.mark.skipif( not hasattr(os, 'statvfs'), reason="os.statvfs() not available" ) @skip_on_not_implemented() def test_against_df(self): # test psutil.disk_usage() and psutil.disk_partitions() # against "df -a" def df(path): out = sh(f'df -P -B 1 "{path}"').strip() lines = out.split('\n') lines.pop(0) line = lines.pop(0) dev, total, used, free = line.split()[:4] if dev == 'none': dev = '' total, used, free = int(total), int(used), int(free) return dev, total, used, free for part in psutil.disk_partitions(all=False): usage = psutil.disk_usage(part.mountpoint) _, total, used, free = df(part.mountpoint) assert usage.total == total assert abs(usage.free - free) < TOLERANCE_DISK_USAGE assert abs(usage.used - used) < TOLERANCE_DISK_USAGE def test_zfs_fs(self): # Test that ZFS partitions are returned. with open("/proc/filesystems") as f: data = f.read() if 'zfs' in data: for part in psutil.disk_partitions(): if part.fstype == 'zfs': return # No ZFS partitions on this system. Let's fake one. fake_file = io.StringIO("nodev\tzfs\n") with mock.patch( 'psutil._common.open', return_value=fake_file, create=True ) as m1: with mock.patch( 'psutil._pslinux.cext.disk_partitions', return_value=[('/dev/sdb3', '/', 'zfs', 'rw')], ) as m2: ret = psutil.disk_partitions() assert m1.called assert m2.called assert ret assert ret[0].fstype == 'zfs' def test_emulate_realpath_fail(self): # See: https://github.com/giampaolo/psutil/issues/1307 try: with mock.patch( 'os.path.realpath', return_value='/non/existent' ) as m: with pytest.raises(FileNotFoundError): psutil.disk_partitions() assert m.called finally: psutil.PROCFS_PATH = "/proc" @pytest.mark.skipif(not LINUX, reason="LINUX only") class TestSystemDiskIoCounters(PsutilTestCase): def test_emulate_kernel_2_4(self): # Tests /proc/diskstats parsing format for 2.4 kernels, see: # https://github.com/giampaolo/psutil/issues/767 content = " 3 0 1 hda 2 3 4 5 6 7 8 9 10 11 12" with mock_open_content({'/proc/diskstats': content}): with mock.patch( 'psutil._pslinux.is_storage_device', return_value=True ): ret = psutil.disk_io_counters(nowrap=False) assert ret.read_count == 1 assert ret.read_merged_count == 2 assert ret.read_bytes == 3 * SECTOR_SIZE assert ret.read_time == 4 assert ret.write_count == 5 assert ret.write_merged_count == 6 assert ret.write_bytes == 7 * SECTOR_SIZE assert ret.write_time == 8 assert ret.busy_time == 10 def test_emulate_kernel_2_6_full(self): # Tests /proc/diskstats parsing format for 2.6 kernels, # lines reporting all metrics: # https://github.com/giampaolo/psutil/issues/767 content = " 3 0 hda 1 2 3 4 5 6 7 8 9 10 11" with mock_open_content({"/proc/diskstats": content}): with mock.patch( 'psutil._pslinux.is_storage_device', return_value=True ): ret = psutil.disk_io_counters(nowrap=False) assert ret.read_count == 1 assert ret.read_merged_count == 2 assert ret.read_bytes == 3 * SECTOR_SIZE assert ret.read_time == 4 assert ret.write_count == 5 assert ret.write_merged_count == 6 assert ret.write_bytes == 7 * SECTOR_SIZE assert ret.write_time == 8 assert ret.busy_time == 10 def test_emulate_kernel_2_6_limited(self): # Tests /proc/diskstats parsing format for 2.6 kernels, # where one line of /proc/partitions return a limited # amount of metrics when it bumps into a partition # (instead of a disk). See: # https://github.com/giampaolo/psutil/issues/767 with mock_open_content({"/proc/diskstats": " 3 1 hda 1 2 3 4"}): with mock.patch( 'psutil._pslinux.is_storage_device', return_value=True ): ret = psutil.disk_io_counters(nowrap=False) assert ret.read_count == 1 assert ret.read_bytes == 2 * SECTOR_SIZE assert ret.write_count == 3 assert ret.write_bytes == 4 * SECTOR_SIZE assert ret.read_merged_count == 0 assert ret.read_time == 0 assert ret.write_merged_count == 0 assert ret.write_time == 0 assert ret.busy_time == 0 def test_emulate_include_partitions(self): # Make sure that when perdisk=True disk partitions are returned, # see: # https://github.com/giampaolo/psutil/pull/1313#issuecomment-408626842 content = textwrap.dedent("""\ 3 0 nvme0n1 1 2 3 4 5 6 7 8 9 10 11 3 0 nvme0n1p1 1 2 3 4 5 6 7 8 9 10 11 """) with mock_open_content({"/proc/diskstats": content}): with mock.patch( 'psutil._pslinux.is_storage_device', return_value=False ): ret = psutil.disk_io_counters(perdisk=True, nowrap=False) assert len(ret) == 2 assert ret['nvme0n1'].read_count == 1 assert ret['nvme0n1p1'].read_count == 1 assert ret['nvme0n1'].write_count == 5 assert ret['nvme0n1p1'].write_count == 5 def test_emulate_exclude_partitions(self): # Make sure that when perdisk=False partitions (e.g. 'sda1', # 'nvme0n1p1') are skipped and not included in the total count. # https://github.com/giampaolo/psutil/pull/1313#issuecomment-408626842 content = textwrap.dedent("""\ 3 0 nvme0n1 1 2 3 4 5 6 7 8 9 10 11 3 0 nvme0n1p1 1 2 3 4 5 6 7 8 9 10 11 """) with mock_open_content({"/proc/diskstats": content}): with mock.patch( 'psutil._pslinux.is_storage_device', return_value=False ): ret = psutil.disk_io_counters(perdisk=False, nowrap=False) assert ret is None def is_storage_device(name): return name == 'nvme0n1' content = textwrap.dedent("""\ 3 0 nvme0n1 1 2 3 4 5 6 7 8 9 10 11 3 0 nvme0n1p1 1 2 3 4 5 6 7 8 9 10 11 """) with mock_open_content({"/proc/diskstats": content}): with mock.patch( 'psutil._pslinux.is_storage_device', create=True, side_effect=is_storage_device, ): ret = psutil.disk_io_counters(perdisk=False, nowrap=False) assert ret.read_count == 1 assert ret.write_count == 5 def test_emulate_use_sysfs(self): def exists(path): return path == '/proc/diskstats' wprocfs = psutil.disk_io_counters(perdisk=True) with mock.patch( 'psutil._pslinux.os.path.exists', create=True, side_effect=exists ): wsysfs = psutil.disk_io_counters(perdisk=True) assert len(wprocfs) == len(wsysfs) def test_emulate_not_impl(self): def exists(path): return False with mock.patch( 'psutil._pslinux.os.path.exists', create=True, side_effect=exists ): with pytest.raises(NotImplementedError): psutil.disk_io_counters() @pytest.mark.skipif(not LINUX, reason="LINUX only") class TestRootFsDeviceFinder(PsutilTestCase): def setUp(self): dev = os.stat("/").st_dev self.major = os.major(dev) self.minor = os.minor(dev) def test_call_methods(self): finder = RootFsDeviceFinder() if os.path.exists("/proc/partitions"): finder.ask_proc_partitions() else: with pytest.raises(FileNotFoundError): finder.ask_proc_partitions() if os.path.exists(f"/sys/dev/block/{self.major}:{self.minor}/uevent"): finder.ask_sys_dev_block() else: with pytest.raises(FileNotFoundError): finder.ask_sys_dev_block() finder.ask_sys_class_block() @pytest.mark.skipif(GITHUB_ACTIONS, reason="unsupported on GITHUB_ACTIONS") def test_comparisons(self): finder = RootFsDeviceFinder() assert finder.find() is not None a = b = c = None if os.path.exists("/proc/partitions"): a = finder.ask_proc_partitions() if os.path.exists(f"/sys/dev/block/{self.major}:{self.minor}/uevent"): b = finder.ask_sys_class_block() c = finder.ask_sys_dev_block() base = a or b or c if base and a: assert base == a if base and b: assert base == b if base and c: assert base == c @pytest.mark.skipif( not shutil.which("findmnt"), reason="findmnt utility not available" ) @pytest.mark.skipif(GITHUB_ACTIONS, reason="unsupported on GITHUB_ACTIONS") def test_against_findmnt(self): psutil_value = RootFsDeviceFinder().find() findmnt_value = sh("findmnt -o SOURCE -rn /") assert psutil_value == findmnt_value def test_disk_partitions_mocked(self): with mock.patch( 'psutil._pslinux.cext.disk_partitions', return_value=[('/dev/root', '/', 'ext4', 'rw')], ) as m: part = psutil.disk_partitions()[0] assert m.called if not GITHUB_ACTIONS: assert part.device != "/dev/root" assert part.device == RootFsDeviceFinder().find() else: assert part.device == "/dev/root" # ===================================================================== # --- misc # ===================================================================== @pytest.mark.skipif(not LINUX, reason="LINUX only") class TestMisc(PsutilTestCase): def test_boot_time(self): vmstat_value = vmstat('boot time') psutil_value = psutil.boot_time() assert int(vmstat_value) == int(psutil_value) def test_no_procfs_on_import(self): my_procfs = self.get_testfn() os.mkdir(my_procfs) with open(os.path.join(my_procfs, 'stat'), 'w') as f: f.write('cpu 0 0 0 0 0 0 0 0 0 0\n') f.write('cpu0 0 0 0 0 0 0 0 0 0 0\n') f.write('cpu1 0 0 0 0 0 0 0 0 0 0\n') try: orig_open = open def open_mock(name, *args, **kwargs): if name.startswith('/proc'): raise FileNotFoundError return orig_open(name, *args, **kwargs) with mock.patch("builtins.open", side_effect=open_mock): reload_module(psutil) with pytest.raises(OSError): psutil.cpu_times() with pytest.raises(OSError): psutil.cpu_times(percpu=True) with pytest.raises(OSError): psutil.cpu_percent() with pytest.raises(OSError): psutil.cpu_percent(percpu=True) with pytest.raises(OSError): psutil.cpu_times_percent() with pytest.raises(OSError): psutil.cpu_times_percent(percpu=True) psutil.PROCFS_PATH = my_procfs assert psutil.cpu_percent() == 0 assert sum(psutil.cpu_times_percent()) == 0 # since we don't know the number of CPUs at import time, # we awkwardly say there are none until the second call per_cpu_percent = psutil.cpu_percent(percpu=True) assert sum(per_cpu_percent) == 0 # ditto awkward length per_cpu_times_percent = psutil.cpu_times_percent(percpu=True) assert sum(map(sum, per_cpu_times_percent)) == 0 # much user, very busy with open(os.path.join(my_procfs, 'stat'), 'w') as f: f.write('cpu 1 0 0 0 0 0 0 0 0 0\n') f.write('cpu0 1 0 0 0 0 0 0 0 0 0\n') f.write('cpu1 1 0 0 0 0 0 0 0 0 0\n') assert psutil.cpu_percent() != 0 assert sum(psutil.cpu_percent(percpu=True)) != 0 assert sum(psutil.cpu_times_percent()) != 0 assert ( sum(map(sum, psutil.cpu_times_percent(percpu=True))) != 0 ) finally: shutil.rmtree(my_procfs) reload_module(psutil) assert psutil.PROCFS_PATH == '/proc' def test_cpu_steal_decrease(self): # Test cumulative cpu stats decrease. We should ignore this. # See issue #1210. content = textwrap.dedent("""\ cpu 0 0 0 0 0 0 0 1 0 0 cpu0 0 0 0 0 0 0 0 1 0 0 cpu1 0 0 0 0 0 0 0 1 0 0 """).encode() with mock_open_content({"/proc/stat": content}) as m: # first call to "percent" functions should read the new stat file # and compare to the "real" file read at import time - so the # values are meaningless psutil.cpu_percent() assert m.called psutil.cpu_percent(percpu=True) psutil.cpu_times_percent() psutil.cpu_times_percent(percpu=True) content = textwrap.dedent("""\ cpu 1 0 0 0 0 0 0 0 0 0 cpu0 1 0 0 0 0 0 0 0 0 0 cpu1 1 0 0 0 0 0 0 0 0 0 """).encode() with mock_open_content({"/proc/stat": content}): # Increase "user" while steal goes "backwards" to zero. cpu_percent = psutil.cpu_percent() assert m.called cpu_percent_percpu = psutil.cpu_percent(percpu=True) cpu_times_percent = psutil.cpu_times_percent() cpu_times_percent_percpu = psutil.cpu_times_percent(percpu=True) assert cpu_percent != 0 assert sum(cpu_percent_percpu) != 0 assert sum(cpu_times_percent) != 0 assert sum(cpu_times_percent) != 100.0 assert sum(map(sum, cpu_times_percent_percpu)) != 0 assert sum(map(sum, cpu_times_percent_percpu)) != 100.0 assert cpu_times_percent.steal == 0 assert cpu_times_percent.user != 0 def test_boot_time_mocked(self): with mock.patch('psutil._common.open', create=True) as m: with pytest.raises(RuntimeError): psutil._pslinux.boot_time() assert m.called def test_users(self): # Make sure the C extension converts ':0' and ':0.0' to # 'localhost'. for user in psutil.users(): assert user.host not in {":0", ":0.0"} def test_procfs_path(self): tdir = self.get_testfn() os.mkdir(tdir) try: psutil.PROCFS_PATH = tdir with pytest.raises(OSError): psutil.virtual_memory() with pytest.raises(OSError): psutil.cpu_times() with pytest.raises(OSError): psutil.cpu_times(percpu=True) with pytest.raises(OSError): psutil.boot_time() # self.assertRaises(OSError, psutil.pids) with pytest.raises(OSError): psutil.net_connections() with pytest.raises(OSError): psutil.net_io_counters() with pytest.raises(OSError): psutil.net_if_stats() # self.assertRaises(OSError, psutil.disk_io_counters) with pytest.raises(OSError): psutil.disk_partitions() with pytest.raises(psutil.NoSuchProcess): psutil.Process() finally: psutil.PROCFS_PATH = "/proc" @retry_on_failure() @pytest.mark.skipif(PYTEST_PARALLEL, reason="skip if pytest-parallel") def test_issue_687(self): # In case of thread ID: # - pid_exists() is supposed to return False # - Process(tid) is supposed to work # - pids() should not return the TID # See: https://github.com/giampaolo/psutil/issues/687 with ThreadTask(): p = psutil.Process() threads = p.threads() assert len(threads) == 2 tid = sorted(threads, key=lambda x: x.id)[1].id assert p.pid != tid pt = psutil.Process(tid) pt.as_dict() assert tid not in psutil.pids() def test_pid_exists_no_proc_status(self): # Internally pid_exists relies on /proc/{pid}/status. # Emulate a case where this file is empty in which case # psutil is supposed to fall back on using pids(). with mock_open_content({"/proc/%s/status": ""}) as m: assert psutil.pid_exists(os.getpid()) assert m.called # ===================================================================== # --- sensors # ===================================================================== @pytest.mark.skipif(not LINUX, reason="LINUX only") @pytest.mark.skipif(not HAS_BATTERY, reason="no battery") class TestSensorsBattery(PsutilTestCase): @pytest.mark.skipif( not shutil.which("acpi"), reason="acpi utility not available" ) def test_percent(self): out = sh("acpi -b") acpi_value = int(out.split(",")[1].strip().replace('%', '')) psutil_value = psutil.sensors_battery().percent assert abs(acpi_value - psutil_value) < 1 def test_emulate_power_plugged(self): # Pretend the AC power cable is connected. def open_mock(name, *args, **kwargs): if name.endswith(('AC0/online', 'AC/online')): return io.BytesIO(b"1") else: return orig_open(name, *args, **kwargs) orig_open = open with mock.patch("builtins.open", side_effect=open_mock) as m: assert psutil.sensors_battery().power_plugged is True assert ( psutil.sensors_battery().secsleft == psutil.POWER_TIME_UNLIMITED ) assert m.called def test_emulate_power_plugged_2(self): # Same as above but pretend /AC0/online does not exist in which # case code relies on /status file. def open_mock(name, *args, **kwargs): if name.endswith(('AC0/online', 'AC/online')): raise FileNotFoundError if name.endswith("/status"): return io.StringIO("charging") else: return orig_open(name, *args, **kwargs) orig_open = open with mock.patch("builtins.open", side_effect=open_mock) as m: assert psutil.sensors_battery().power_plugged is True assert m.called def test_emulate_power_not_plugged(self): # Pretend the AC power cable is not connected. def open_mock(name, *args, **kwargs): if name.endswith(('AC0/online', 'AC/online')): return io.BytesIO(b"0") else: return orig_open(name, *args, **kwargs) orig_open = open with mock.patch("builtins.open", side_effect=open_mock) as m: assert psutil.sensors_battery().power_plugged is False assert m.called def test_emulate_power_not_plugged_2(self): # Same as above but pretend /AC0/online does not exist in which # case code relies on /status file. def open_mock(name, *args, **kwargs): if name.endswith(('AC0/online', 'AC/online')): raise FileNotFoundError if name.endswith("/status"): return io.StringIO("discharging") else: return orig_open(name, *args, **kwargs) orig_open = open with mock.patch("builtins.open", side_effect=open_mock) as m: assert psutil.sensors_battery().power_plugged is False assert m.called def test_emulate_power_undetermined(self): # Pretend we can't know whether the AC power cable not # connected (assert fallback to False). def open_mock(name, *args, **kwargs): if name.startswith(( '/sys/class/power_supply/AC0/online', '/sys/class/power_supply/AC/online', )): raise FileNotFoundError if name.startswith("/sys/class/power_supply/BAT0/status"): return io.BytesIO(b"???") else: return orig_open(name, *args, **kwargs) orig_open = open with mock.patch("builtins.open", side_effect=open_mock) as m: assert psutil.sensors_battery().power_plugged is None assert m.called def test_emulate_energy_full_0(self): # Emulate a case where energy_full files returns 0. with mock_open_content( {"/sys/class/power_supply/BAT0/energy_full": b"0"} ) as m: assert psutil.sensors_battery().percent == 0 assert m.called def test_emulate_energy_full_not_avail(self): # Emulate a case where energy_full file does not exist. # Expected fallback on /capacity. with mock_open_exception( "/sys/class/power_supply/BAT0/energy_full", FileNotFoundError, ): with mock_open_exception( "/sys/class/power_supply/BAT0/charge_full", FileNotFoundError, ): with mock_open_content( {"/sys/class/power_supply/BAT0/capacity": b"88"} ): assert psutil.sensors_battery().percent == 88 def test_emulate_no_power(self): # Emulate a case where /AC0/online file nor /BAT0/status exist. with mock_open_exception( "/sys/class/power_supply/AC/online", FileNotFoundError ): with mock_open_exception( "/sys/class/power_supply/AC0/online", FileNotFoundError ): with mock_open_exception( "/sys/class/power_supply/BAT0/status", FileNotFoundError, ): assert psutil.sensors_battery().power_plugged is None @pytest.mark.skipif(not LINUX, reason="LINUX only") class TestSensorsBatteryEmulated(PsutilTestCase): def test_it(self): def open_mock(name, *args, **kwargs): if name.endswith("/energy_now"): return io.StringIO("60000000") elif name.endswith("/power_now"): return io.StringIO("0") elif name.endswith("/energy_full"): return io.StringIO("60000001") else: return orig_open(name, *args, **kwargs) orig_open = open with mock.patch('os.listdir', return_value=["BAT0"]) as mlistdir: with mock.patch("builtins.open", side_effect=open_mock) as mopen: assert psutil.sensors_battery() is not None assert mlistdir.called assert mopen.called @pytest.mark.skipif(not LINUX, reason="LINUX only") class TestSensorsTemperatures(PsutilTestCase): def test_emulate_class_hwmon(self): def open_mock(name, *args, **kwargs): if name.endswith('/name'): return io.StringIO("name") elif name.endswith('/temp1_label'): return io.StringIO("label") elif name.endswith('/temp1_input'): return io.BytesIO(b"30000") elif name.endswith('/temp1_max'): return io.BytesIO(b"40000") elif name.endswith('/temp1_crit'): return io.BytesIO(b"50000") else: return orig_open(name, *args, **kwargs) orig_open = open with mock.patch("builtins.open", side_effect=open_mock): # Test case with /sys/class/hwmon with mock.patch( 'glob.glob', return_value=['/sys/class/hwmon/hwmon0/temp1'] ): temp = psutil.sensors_temperatures()['name'][0] assert temp.label == 'label' assert temp.current == 30.0 assert temp.high == 40.0 assert temp.critical == 50.0 def test_emulate_class_thermal(self): def open_mock(name, *args, **kwargs): if name.endswith('0_temp'): return io.BytesIO(b"50000") elif name.endswith('temp'): return io.BytesIO(b"30000") elif name.endswith('0_type'): return io.StringIO("critical") elif name.endswith('type'): return io.StringIO("name") else: return orig_open(name, *args, **kwargs) def glob_mock(path): if path in { '/sys/class/hwmon/hwmon*/temp*_*', '/sys/class/hwmon/hwmon*/device/temp*_*', }: return [] elif path == '/sys/class/thermal/thermal_zone*': return ['/sys/class/thermal/thermal_zone0'] elif path == '/sys/class/thermal/thermal_zone0/trip_point*': return [ '/sys/class/thermal/thermal_zone1/trip_point_0_type', '/sys/class/thermal/thermal_zone1/trip_point_0_temp', ] return [] orig_open = open with mock.patch("builtins.open", side_effect=open_mock): with mock.patch('glob.glob', create=True, side_effect=glob_mock): temp = psutil.sensors_temperatures()['name'][0] assert temp.label == '' assert temp.current == 30.0 assert temp.high == 50.0 assert temp.critical == 50.0 @pytest.mark.skipif(not LINUX, reason="LINUX only") class TestSensorsFans(PsutilTestCase): def test_emulate_data(self): def open_mock(name, *args, **kwargs): if name.endswith('/name'): return io.StringIO("name") elif name.endswith('/fan1_label'): return io.StringIO("label") elif name.endswith('/fan1_input'): return io.StringIO("2000") else: return orig_open(name, *args, **kwargs) orig_open = open with mock.patch("builtins.open", side_effect=open_mock): with mock.patch( 'glob.glob', return_value=['/sys/class/hwmon/hwmon2/fan1'] ): fan = psutil.sensors_fans()['name'][0] assert fan.label == 'label' assert fan.current == 2000 # ===================================================================== # --- test process # ===================================================================== @pytest.mark.skipif(not LINUX, reason="LINUX only") class TestProcess(PsutilTestCase): @retry_on_failure() def test_parse_smaps_vs_memory_maps(self): sproc = self.spawn_testproc() uss, pss, swap = psutil._pslinux.Process(sproc.pid)._parse_smaps() maps = psutil.Process(sproc.pid).memory_maps(grouped=False) assert ( abs(uss - sum(x.private_dirty + x.private_clean for x in maps)) < 4096 ) assert abs(pss - sum(x.pss for x in maps)) < 4096 assert abs(swap - sum(x.swap for x in maps)) < 4096 def test_parse_smaps_mocked(self): # See: https://github.com/giampaolo/psutil/issues/1222 content = textwrap.dedent("""\ fffff0 r-xp 00000000 00:00 0 [vsyscall] Size: 1 kB Rss: 2 kB Pss: 3 kB Shared_Clean: 4 kB Shared_Dirty: 5 kB Private_Clean: 6 kB Private_Dirty: 7 kB Referenced: 8 kB Anonymous: 9 kB LazyFree: 10 kB AnonHugePages: 11 kB ShmemPmdMapped: 12 kB Shared_Hugetlb: 13 kB Private_Hugetlb: 14 kB Swap: 15 kB SwapPss: 16 kB KernelPageSize: 17 kB MMUPageSize: 18 kB Locked: 19 kB VmFlags: rd ex """).encode() with mock_open_content({f"/proc/{os.getpid()}/smaps": content}) as m: p = psutil._pslinux.Process(os.getpid()) uss, pss, swap = p._parse_smaps() assert m.called assert uss == (6 + 7 + 14) * 1024 assert pss == 3 * 1024 assert swap == 15 * 1024 # On PYPY file descriptors are not closed fast enough. @pytest.mark.skipif(PYPY, reason="unreliable on PYPY") def test_open_files_mode(self): def get_test_file(fname): p = psutil.Process() giveup_at = time.time() + GLOBAL_TIMEOUT while True: for file in p.open_files(): if file.path == os.path.abspath(fname): return file elif time.time() > giveup_at: break raise RuntimeError("timeout looking for test file") testfn = self.get_testfn() with open(testfn, "w"): assert get_test_file(testfn).mode == "w" with open(testfn): assert get_test_file(testfn).mode == "r" with open(testfn, "a"): assert get_test_file(testfn).mode == "a" with open(testfn, "r+"): assert get_test_file(testfn).mode == "r+" with open(testfn, "w+"): assert get_test_file(testfn).mode == "r+" with open(testfn, "a+"): assert get_test_file(testfn).mode == "a+" safe_rmpath(testfn) with open(testfn, "x"): assert get_test_file(testfn).mode == "w" safe_rmpath(testfn) with open(testfn, "x+"): assert get_test_file(testfn).mode == "r+" def test_open_files_file_gone(self): # simulates a file which gets deleted during open_files() # execution p = psutil.Process() files = p.open_files() with open(self.get_testfn(), 'w'): # give the kernel some time to see the new file call_until(lambda: len(p.open_files()) != len(files)) with mock.patch( 'psutil._pslinux.os.readlink', side_effect=FileNotFoundError, ) as m: assert p.open_files() == [] assert m.called # also simulate the case where os.readlink() returns EINVAL # in which case psutil is supposed to 'continue' with mock.patch( 'psutil._pslinux.os.readlink', side_effect=OSError(errno.EINVAL, ""), ) as m: assert p.open_files() == [] assert m.called def test_open_files_fd_gone(self): # Simulate a case where /proc/{pid}/fdinfo/{fd} disappears # while iterating through fds. # https://travis-ci.org/giampaolo/psutil/jobs/225694530 p = psutil.Process() files = p.open_files() with open(self.get_testfn(), 'w'): # give the kernel some time to see the new file call_until(lambda: len(p.open_files()) != len(files)) with mock.patch( "builtins.open", side_effect=FileNotFoundError ) as m: assert p.open_files() == [] assert m.called def test_open_files_enametoolong(self): # Simulate a case where /proc/{pid}/fd/{fd} symlink # points to a file with full path longer than PATH_MAX, see: # https://github.com/giampaolo/psutil/issues/1940 p = psutil.Process() files = p.open_files() with open(self.get_testfn(), 'w'): # give the kernel some time to see the new file call_until(lambda: len(p.open_files()) != len(files)) patch_point = 'psutil._pslinux.os.readlink' with mock.patch( patch_point, side_effect=OSError(errno.ENAMETOOLONG, "") ) as m: with mock.patch("psutil._pslinux.debug"): assert p.open_files() == [] assert m.called # --- mocked tests def test_terminal_mocked(self): with mock.patch( 'psutil._pslinux._psposix.get_terminal_map', return_value={} ) as m: assert psutil._pslinux.Process(os.getpid()).terminal() is None assert m.called # TODO: re-enable this test. # def test_num_ctx_switches_mocked(self): # with mock.patch('psutil._common.open', create=True) as m: # self.assertRaises( # NotImplementedError, # psutil._pslinux.Process(os.getpid()).num_ctx_switches) # assert m.called def test_cmdline_mocked(self): # see: https://github.com/giampaolo/psutil/issues/639 p = psutil.Process() fake_file = io.StringIO('foo\x00bar\x00') with mock.patch( 'psutil._common.open', return_value=fake_file, create=True ) as m: assert p.cmdline() == ['foo', 'bar'] assert m.called fake_file = io.StringIO('foo\x00bar\x00\x00') with mock.patch( 'psutil._common.open', return_value=fake_file, create=True ) as m: assert p.cmdline() == ['foo', 'bar', ''] assert m.called def test_cmdline_spaces_mocked(self): # see: https://github.com/giampaolo/psutil/issues/1179 p = psutil.Process() fake_file = io.StringIO('foo bar ') with mock.patch( 'psutil._common.open', return_value=fake_file, create=True ) as m: assert p.cmdline() == ['foo', 'bar'] assert m.called fake_file = io.StringIO('foo bar ') with mock.patch( 'psutil._common.open', return_value=fake_file, create=True ) as m: assert p.cmdline() == ['foo', 'bar', ''] assert m.called def test_cmdline_mixed_separators(self): # https://github.com/giampaolo/psutil/issues/ # 1179#issuecomment-552984549 p = psutil.Process() fake_file = io.StringIO('foo\x20bar\x00') with mock.patch( 'psutil._common.open', return_value=fake_file, create=True ) as m: assert p.cmdline() == ['foo', 'bar'] assert m.called def test_readlink_path_deleted_mocked(self): with mock.patch( 'psutil._pslinux.os.readlink', return_value='/home/foo (deleted)' ): assert psutil.Process().exe() == "/home/foo" assert psutil.Process().cwd() == "/home/foo" def test_threads_mocked(self): # Test the case where os.listdir() returns a file (thread) # which no longer exists by the time we open() it (race # condition). threads() is supposed to ignore that instead # of raising NSP. def open_mock_1(name, *args, **kwargs): if name.startswith(f"/proc/{os.getpid()}/task"): raise FileNotFoundError return orig_open(name, *args, **kwargs) orig_open = open with mock.patch("builtins.open", side_effect=open_mock_1) as m: ret = psutil.Process().threads() assert m.called assert ret == [] # ...but if it bumps into something != ENOENT we want an # exception. def open_mock_2(name, *args, **kwargs): if name.startswith(f"/proc/{os.getpid()}/task"): raise PermissionError return orig_open(name, *args, **kwargs) with mock.patch("builtins.open", side_effect=open_mock_2): with pytest.raises(psutil.AccessDenied): psutil.Process().threads() def test_exe_mocked(self): with mock.patch( 'psutil._pslinux.readlink', side_effect=FileNotFoundError ) as m: # de-activate guessing from cmdline() with mock.patch( 'psutil._pslinux.Process.cmdline', return_value=[] ): ret = psutil.Process().exe() assert m.called assert ret == "" def test_issue_1014(self): # Emulates a case where smaps file does not exist. In this case # wrap_exception decorator should not raise NoSuchProcess. with mock_open_exception( f"/proc/{os.getpid()}/smaps", FileNotFoundError ) as m: p = psutil.Process() with pytest.raises(FileNotFoundError): p.memory_maps() assert m.called def test_issue_2418(self): p = psutil.Process() with mock_open_exception( f"/proc/{os.getpid()}/statm", FileNotFoundError ): with mock.patch("os.path.exists", return_value=False): with pytest.raises(psutil.NoSuchProcess): p.memory_info() @pytest.mark.skipif(not HAS_RLIMIT, reason="not supported") def test_rlimit_zombie(self): # Emulate a case where rlimit() raises ENOSYS, which may # happen in case of zombie process: # https://travis-ci.org/giampaolo/psutil/jobs/51368273 with mock.patch( "resource.prlimit", side_effect=OSError(errno.ENOSYS, "") ) as m1: with mock.patch( "psutil._pslinux.Process._is_zombie", return_value=True ) as m2: p = psutil.Process() p.name() with pytest.raises(psutil.ZombieProcess) as cm: p.rlimit(psutil.RLIMIT_NOFILE) assert m1.called assert m2.called assert cm.value.pid == p.pid assert cm.value.name == p.name() def test_stat_file_parsing(self): args = [ "0", # pid "(cat)", # name "Z", # status "1", # ppid "0", # pgrp "0", # session "0", # tty "0", # tpgid "0", # flags "0", # minflt "0", # cminflt "0", # majflt "0", # cmajflt "2", # utime "3", # stime "4", # cutime "5", # cstime "0", # priority "0", # nice "0", # num_threads "0", # itrealvalue "6", # starttime "0", # vsize "0", # rss "0", # rsslim "0", # startcode "0", # endcode "0", # startstack "0", # kstkesp "0", # kstkeip "0", # signal "0", # blocked "0", # sigignore "0", # sigcatch "0", # wchan "0", # nswap "0", # cnswap "0", # exit_signal "6", # processor "0", # rt priority "0", # policy "7", # delayacct_blkio_ticks ] content = " ".join(args).encode() with mock_open_content({f"/proc/{os.getpid()}/stat": content}): p = psutil.Process() assert p.name() == 'cat' assert p.status() == psutil.STATUS_ZOMBIE assert p.ppid() == 1 assert p.create_time() == 6 / CLOCK_TICKS + psutil.boot_time() cpu = p.cpu_times() assert cpu.user == 2 / CLOCK_TICKS assert cpu.system == 3 / CLOCK_TICKS assert cpu.children_user == 4 / CLOCK_TICKS assert cpu.children_system == 5 / CLOCK_TICKS assert cpu.iowait == 7 / CLOCK_TICKS assert p.cpu_num() == 6 def test_status_file_parsing(self): content = textwrap.dedent("""\ Uid:\t1000\t1001\t1002\t1003 Gid:\t1004\t1005\t1006\t1007 Threads:\t66 Cpus_allowed:\tf Cpus_allowed_list:\t0-7 voluntary_ctxt_switches:\t12 nonvoluntary_ctxt_switches:\t13""").encode() with mock_open_content({f"/proc/{os.getpid()}/status": content}): p = psutil.Process() assert p.num_ctx_switches().voluntary == 12 assert p.num_ctx_switches().involuntary == 13 assert p.num_threads() == 66 uids = p.uids() assert uids.real == 1000 assert uids.effective == 1001 assert uids.saved == 1002 gids = p.gids() assert gids.real == 1004 assert gids.effective == 1005 assert gids.saved == 1006 assert p._proc._get_eligible_cpus() == list(range(8)) def test_net_connections_enametoolong(self): # Simulate a case where /proc/{pid}/fd/{fd} symlink points to # a file with full path longer than PATH_MAX, see: # https://github.com/giampaolo/psutil/issues/1940 with mock.patch( 'psutil._pslinux.os.readlink', side_effect=OSError(errno.ENAMETOOLONG, ""), ) as m: p = psutil.Process() with mock.patch("psutil._pslinux.debug"): assert p.net_connections() == [] assert m.called @pytest.mark.skipif(not LINUX, reason="LINUX only") class TestProcessAgainstStatus(PsutilTestCase): """/proc/pid/stat and /proc/pid/status have many values in common. Whenever possible, psutil uses /proc/pid/stat (it's faster). For all those cases we check that the value found in /proc/pid/stat (by psutil) matches the one found in /proc/pid/status. """ @classmethod def setUpClass(cls): cls.proc = psutil.Process() def read_status_file(self, linestart): with psutil._psplatform.open_text( f"/proc/{self.proc.pid}/status" ) as f: for line in f: line = line.strip() if line.startswith(linestart): value = line.partition('\t')[2] try: return int(value) except ValueError: return value raise ValueError(f"can't find {linestart!r}") def test_name(self): value = self.read_status_file("Name:") assert self.proc.name() == value def test_status(self): value = self.read_status_file("State:") value = value[value.find('(') + 1 : value.rfind(')')] value = value.replace(' ', '-') assert self.proc.status() == value def test_ppid(self): value = self.read_status_file("PPid:") assert self.proc.ppid() == value def test_num_threads(self): value = self.read_status_file("Threads:") assert self.proc.num_threads() == value def test_uids(self): value = self.read_status_file("Uid:") value = tuple(map(int, value.split()[1:4])) assert self.proc.uids() == value def test_gids(self): value = self.read_status_file("Gid:") value = tuple(map(int, value.split()[1:4])) assert self.proc.gids() == value @retry_on_failure() def test_num_ctx_switches(self): value = self.read_status_file("voluntary_ctxt_switches:") assert self.proc.num_ctx_switches().voluntary == value value = self.read_status_file("nonvoluntary_ctxt_switches:") assert self.proc.num_ctx_switches().involuntary == value def test_cpu_affinity(self): value = self.read_status_file("Cpus_allowed_list:") if '-' in str(value): min_, max_ = map(int, value.split('-')) assert self.proc.cpu_affinity() == list(range(min_, max_ + 1)) def test_cpu_affinity_eligible_cpus(self): value = self.read_status_file("Cpus_allowed_list:") with mock.patch("psutil._pslinux.per_cpu_times") as m: self.proc._proc._get_eligible_cpus() if '-' in str(value): assert not m.called else: assert m.called # ===================================================================== # --- test utils # ===================================================================== @pytest.mark.skipif(not LINUX, reason="LINUX only") class TestUtils(PsutilTestCase): def test_readlink(self): with mock.patch("os.readlink", return_value="foo (deleted)") as m: assert psutil._psplatform.readlink("bar") == "foo" assert m.called tests/test_process.py 0000644 00000164751 15030211765 0011013 0 ustar 00 #!/usr/bin/env python3 # Copyright (c) 2009, Giampaolo Rodola'. All rights reserved. # Use of this source code is governed by a BSD-style license that can be # found in the LICENSE file. """Tests for psutil.Process class.""" import collections import contextlib import errno import getpass import io import itertools import os import signal import socket import stat import string import subprocess import sys import textwrap import time from unittest import mock import psutil from psutil import AIX from psutil import BSD from psutil import LINUX from psutil import MACOS from psutil import NETBSD from psutil import OPENBSD from psutil import OSX from psutil import POSIX from psutil import WINDOWS from psutil._common import open_text from psutil.tests import CI_TESTING from psutil.tests import GITHUB_ACTIONS from psutil.tests import GLOBAL_TIMEOUT from psutil.tests import HAS_CPU_AFFINITY from psutil.tests import HAS_ENVIRON from psutil.tests import HAS_IONICE from psutil.tests import HAS_MEMORY_MAPS from psutil.tests import HAS_PROC_CPU_NUM from psutil.tests import HAS_PROC_IO_COUNTERS from psutil.tests import HAS_RLIMIT from psutil.tests import HAS_THREADS from psutil.tests import MACOS_11PLUS from psutil.tests import PYPY from psutil.tests import PYTHON_EXE from psutil.tests import PYTHON_EXE_ENV from psutil.tests import PsutilTestCase from psutil.tests import ThreadTask from psutil.tests import call_until from psutil.tests import copyload_shared_lib from psutil.tests import create_c_exe from psutil.tests import create_py_exe from psutil.tests import process_namespace from psutil.tests import pytest from psutil.tests import reap_children from psutil.tests import retry_on_failure from psutil.tests import sh from psutil.tests import skip_on_access_denied from psutil.tests import skip_on_not_implemented from psutil.tests import wait_for_pid # =================================================================== # --- psutil.Process class tests # =================================================================== class TestProcess(PsutilTestCase): """Tests for psutil.Process class.""" def spawn_psproc(self, *args, **kwargs): sproc = self.spawn_testproc(*args, **kwargs) try: return psutil.Process(sproc.pid) except psutil.NoSuchProcess: self.assertPidGone(sproc.pid) raise # --- def test_pid(self): p = psutil.Process() assert p.pid == os.getpid() with pytest.raises(AttributeError): p.pid = 33 def test_kill(self): p = self.spawn_psproc() p.kill() code = p.wait() if WINDOWS: assert code == signal.SIGTERM else: assert code == -signal.SIGKILL self.assertProcessGone(p) def test_terminate(self): p = self.spawn_psproc() p.terminate() code = p.wait() if WINDOWS: assert code == signal.SIGTERM else: assert code == -signal.SIGTERM self.assertProcessGone(p) def test_send_signal(self): sig = signal.SIGKILL if POSIX else signal.SIGTERM p = self.spawn_psproc() p.send_signal(sig) code = p.wait() if WINDOWS: assert code == sig else: assert code == -sig self.assertProcessGone(p) @pytest.mark.skipif(not POSIX, reason="not POSIX") def test_send_signal_mocked(self): sig = signal.SIGTERM p = self.spawn_psproc() with mock.patch('psutil.os.kill', side_effect=ProcessLookupError): with pytest.raises(psutil.NoSuchProcess): p.send_signal(sig) p = self.spawn_psproc() with mock.patch('psutil.os.kill', side_effect=PermissionError): with pytest.raises(psutil.AccessDenied): p.send_signal(sig) def test_wait_exited(self): # Test waitpid() + WIFEXITED -> WEXITSTATUS. # normal return, same as exit(0) cmd = [PYTHON_EXE, "-c", "pass"] p = self.spawn_psproc(cmd) code = p.wait() assert code == 0 self.assertProcessGone(p) # exit(1), implicit in case of error cmd = [PYTHON_EXE, "-c", "1 / 0"] p = self.spawn_psproc(cmd, stderr=subprocess.PIPE) code = p.wait() assert code == 1 self.assertProcessGone(p) # via sys.exit() cmd = [PYTHON_EXE, "-c", "import sys; sys.exit(5);"] p = self.spawn_psproc(cmd) code = p.wait() assert code == 5 self.assertProcessGone(p) # via os._exit() cmd = [PYTHON_EXE, "-c", "import os; os._exit(5);"] p = self.spawn_psproc(cmd) code = p.wait() assert code == 5 self.assertProcessGone(p) @pytest.mark.skipif(NETBSD, reason="fails on NETBSD") def test_wait_stopped(self): p = self.spawn_psproc() if POSIX: # Test waitpid() + WIFSTOPPED and WIFCONTINUED. # Note: if a process is stopped it ignores SIGTERM. p.send_signal(signal.SIGSTOP) with pytest.raises(psutil.TimeoutExpired): p.wait(timeout=0.001) p.send_signal(signal.SIGCONT) with pytest.raises(psutil.TimeoutExpired): p.wait(timeout=0.001) p.send_signal(signal.SIGTERM) assert p.wait() == -signal.SIGTERM assert p.wait() == -signal.SIGTERM else: p.suspend() with pytest.raises(psutil.TimeoutExpired): p.wait(timeout=0.001) p.resume() with pytest.raises(psutil.TimeoutExpired): p.wait(timeout=0.001) p.terminate() assert p.wait() == signal.SIGTERM assert p.wait() == signal.SIGTERM def test_wait_non_children(self): # Test wait() against a process which is not our direct # child. child, grandchild = self.spawn_children_pair() with pytest.raises(psutil.TimeoutExpired): child.wait(0.01) with pytest.raises(psutil.TimeoutExpired): grandchild.wait(0.01) # We also terminate the direct child otherwise the # grandchild will hang until the parent is gone. child.terminate() grandchild.terminate() child_ret = child.wait() grandchild_ret = grandchild.wait() if POSIX: assert child_ret == -signal.SIGTERM # For processes which are not our children we're supposed # to get None. assert grandchild_ret is None else: assert child_ret == signal.SIGTERM assert child_ret == signal.SIGTERM def test_wait_timeout(self): p = self.spawn_psproc() p.name() with pytest.raises(psutil.TimeoutExpired): p.wait(0.01) with pytest.raises(psutil.TimeoutExpired): p.wait(0) with pytest.raises(ValueError): p.wait(-1) def test_wait_timeout_nonblocking(self): p = self.spawn_psproc() with pytest.raises(psutil.TimeoutExpired): p.wait(0) p.kill() stop_at = time.time() + GLOBAL_TIMEOUT while time.time() < stop_at: try: code = p.wait(0) break except psutil.TimeoutExpired: pass else: raise self.fail('timeout') if POSIX: assert code == -signal.SIGKILL else: assert code == signal.SIGTERM self.assertProcessGone(p) def test_cpu_percent(self): p = psutil.Process() p.cpu_percent(interval=0.001) p.cpu_percent(interval=0.001) for _ in range(100): percent = p.cpu_percent(interval=None) assert isinstance(percent, float) assert percent >= 0.0 with pytest.raises(ValueError): p.cpu_percent(interval=-1) def test_cpu_percent_numcpus_none(self): # See: https://github.com/giampaolo/psutil/issues/1087 with mock.patch('psutil.cpu_count', return_value=None) as m: psutil.Process().cpu_percent() assert m.called def test_cpu_times(self): times = psutil.Process().cpu_times() assert times.user >= 0.0, times assert times.system >= 0.0, times assert times.children_user >= 0.0, times assert times.children_system >= 0.0, times if LINUX: assert times.iowait >= 0.0, times # make sure returned values can be pretty printed with strftime for name in times._fields: time.strftime("%H:%M:%S", time.localtime(getattr(times, name))) def test_cpu_times_2(self): def waste_cpu(): stop_at = os.times().user + 0.2 while os.times().user < stop_at: for x in range(100000): x **= 2 waste_cpu() a = psutil.Process().cpu_times() b = os.times() self.assertAlmostEqual(a.user, b.user, delta=0.1) self.assertAlmostEqual(a.system, b.system, delta=0.1) @pytest.mark.skipif(not HAS_PROC_CPU_NUM, reason="not supported") def test_cpu_num(self): p = psutil.Process() num = p.cpu_num() assert num >= 0 if psutil.cpu_count() == 1: assert num == 0 assert p.cpu_num() in range(psutil.cpu_count()) def test_create_time(self): p = self.spawn_psproc() now = time.time() create_time = p.create_time() # Use time.time() as base value to compare our result using a # tolerance of +/- 1 second. # It will fail if the difference between the values is > 2s. difference = abs(create_time - now) if difference > 2: raise self.fail( f"expected: {now}, found: {create_time}, difference:" f" {difference}" ) # make sure returned value can be pretty printed with strftime time.strftime("%Y %m %d %H:%M:%S", time.localtime(p.create_time())) @pytest.mark.skipif(not POSIX, reason="POSIX only") def test_terminal(self): terminal = psutil.Process().terminal() if terminal is not None: try: tty = os.path.realpath(sh('tty')) except RuntimeError: # Note: happens if pytest is run without the `-s` opt. raise pytest.skip("can't rely on `tty` CLI") else: assert terminal == tty @pytest.mark.skipif(not HAS_PROC_IO_COUNTERS, reason="not supported") @skip_on_not_implemented(only_if=LINUX) def test_io_counters(self): p = psutil.Process() # test reads io1 = p.io_counters() with open(PYTHON_EXE, 'rb') as f: f.read() io2 = p.io_counters() if not BSD and not AIX: assert io2.read_count > io1.read_count assert io2.write_count == io1.write_count if LINUX: assert io2.read_chars > io1.read_chars assert io2.write_chars == io1.write_chars else: assert io2.read_bytes >= io1.read_bytes assert io2.write_bytes >= io1.write_bytes # test writes io1 = p.io_counters() with open(self.get_testfn(), 'wb') as f: f.write(bytes("x" * 1000000, 'ascii')) io2 = p.io_counters() assert io2.write_count >= io1.write_count assert io2.write_bytes >= io1.write_bytes assert io2.read_count >= io1.read_count assert io2.read_bytes >= io1.read_bytes if LINUX: assert io2.write_chars > io1.write_chars assert io2.read_chars >= io1.read_chars # sanity check for i in range(len(io2)): if BSD and i >= 2: # On BSD read_bytes and write_bytes are always set to -1. continue assert io2[i] >= 0 assert io2[i] >= 0 @pytest.mark.skipif(not HAS_IONICE, reason="not supported") @pytest.mark.skipif(not LINUX, reason="linux only") def test_ionice_linux(self): def cleanup(init): ioclass, value = init if ioclass == psutil.IOPRIO_CLASS_NONE: value = 0 p.ionice(ioclass, value) p = psutil.Process() if not CI_TESTING: assert p.ionice()[0] == psutil.IOPRIO_CLASS_NONE assert psutil.IOPRIO_CLASS_NONE == 0 assert psutil.IOPRIO_CLASS_RT == 1 # high assert psutil.IOPRIO_CLASS_BE == 2 # normal assert psutil.IOPRIO_CLASS_IDLE == 3 # low init = p.ionice() self.addCleanup(cleanup, init) # low p.ionice(psutil.IOPRIO_CLASS_IDLE) assert tuple(p.ionice()) == (psutil.IOPRIO_CLASS_IDLE, 0) with pytest.raises(ValueError): # accepts no value p.ionice(psutil.IOPRIO_CLASS_IDLE, value=7) # normal p.ionice(psutil.IOPRIO_CLASS_BE) assert tuple(p.ionice()) == (psutil.IOPRIO_CLASS_BE, 0) p.ionice(psutil.IOPRIO_CLASS_BE, value=7) assert tuple(p.ionice()) == (psutil.IOPRIO_CLASS_BE, 7) with pytest.raises(ValueError): p.ionice(psutil.IOPRIO_CLASS_BE, value=8) try: p.ionice(psutil.IOPRIO_CLASS_RT, value=7) except psutil.AccessDenied: pass # errs with pytest.raises(ValueError, match="ioclass accepts no value"): p.ionice(psutil.IOPRIO_CLASS_NONE, 1) with pytest.raises(ValueError, match="ioclass accepts no value"): p.ionice(psutil.IOPRIO_CLASS_IDLE, 1) with pytest.raises( ValueError, match="'ioclass' argument must be specified" ): p.ionice(value=1) @pytest.mark.skipif(not HAS_IONICE, reason="not supported") @pytest.mark.skipif( not WINDOWS, reason="not supported on this win version" ) def test_ionice_win(self): p = psutil.Process() if not CI_TESTING: assert p.ionice() == psutil.IOPRIO_NORMAL init = p.ionice() self.addCleanup(p.ionice, init) # base p.ionice(psutil.IOPRIO_VERYLOW) assert p.ionice() == psutil.IOPRIO_VERYLOW p.ionice(psutil.IOPRIO_LOW) assert p.ionice() == psutil.IOPRIO_LOW try: p.ionice(psutil.IOPRIO_HIGH) except psutil.AccessDenied: pass else: assert p.ionice() == psutil.IOPRIO_HIGH # errs with pytest.raises( TypeError, match="value argument not accepted on Windows" ): p.ionice(psutil.IOPRIO_NORMAL, value=1) with pytest.raises(ValueError, match="is not a valid priority"): p.ionice(psutil.IOPRIO_HIGH + 1) @pytest.mark.skipif(not HAS_RLIMIT, reason="not supported") def test_rlimit_get(self): import resource p = psutil.Process(os.getpid()) names = [x for x in dir(psutil) if x.startswith('RLIMIT')] assert names, names for name in names: value = getattr(psutil, name) assert value >= 0 if name in dir(resource): assert value == getattr(resource, name) # XXX - On PyPy RLIMIT_INFINITY returned by # resource.getrlimit() is reported as a very big long # number instead of -1. It looks like a bug with PyPy. if PYPY: continue assert p.rlimit(value) == resource.getrlimit(value) else: ret = p.rlimit(value) assert len(ret) == 2 assert ret[0] >= -1 assert ret[1] >= -1 @pytest.mark.skipif(not HAS_RLIMIT, reason="not supported") def test_rlimit_set(self): p = self.spawn_psproc() p.rlimit(psutil.RLIMIT_NOFILE, (5, 5)) assert p.rlimit(psutil.RLIMIT_NOFILE) == (5, 5) # If pid is 0 prlimit() applies to the calling process and # we don't want that. if LINUX: with pytest.raises(ValueError, match="can't use prlimit"): psutil._psplatform.Process(0).rlimit(0) with pytest.raises(ValueError): p.rlimit(psutil.RLIMIT_NOFILE, (5, 5, 5)) @pytest.mark.skipif(not HAS_RLIMIT, reason="not supported") def test_rlimit(self): p = psutil.Process() testfn = self.get_testfn() soft, hard = p.rlimit(psutil.RLIMIT_FSIZE) try: p.rlimit(psutil.RLIMIT_FSIZE, (1024, hard)) with open(testfn, "wb") as f: f.write(b"X" * 1024) # write() or flush() doesn't always cause the exception # but close() will. with pytest.raises(OSError) as exc: with open(testfn, "wb") as f: f.write(b"X" * 1025) assert exc.value.errno == errno.EFBIG finally: p.rlimit(psutil.RLIMIT_FSIZE, (soft, hard)) assert p.rlimit(psutil.RLIMIT_FSIZE) == (soft, hard) @pytest.mark.skipif(not HAS_RLIMIT, reason="not supported") def test_rlimit_infinity(self): # First set a limit, then re-set it by specifying INFINITY # and assume we overridden the previous limit. p = psutil.Process() soft, hard = p.rlimit(psutil.RLIMIT_FSIZE) try: p.rlimit(psutil.RLIMIT_FSIZE, (1024, hard)) p.rlimit(psutil.RLIMIT_FSIZE, (psutil.RLIM_INFINITY, hard)) with open(self.get_testfn(), "wb") as f: f.write(b"X" * 2048) finally: p.rlimit(psutil.RLIMIT_FSIZE, (soft, hard)) assert p.rlimit(psutil.RLIMIT_FSIZE) == (soft, hard) @pytest.mark.skipif(not HAS_RLIMIT, reason="not supported") def test_rlimit_infinity_value(self): # RLIMIT_FSIZE should be RLIM_INFINITY, which will be a really # big number on a platform with large file support. On these # platforms we need to test that the get/setrlimit functions # properly convert the number to a C long long and that the # conversion doesn't raise an error. p = psutil.Process() soft, hard = p.rlimit(psutil.RLIMIT_FSIZE) assert hard == psutil.RLIM_INFINITY p.rlimit(psutil.RLIMIT_FSIZE, (soft, hard)) def test_num_threads(self): # on certain platforms such as Linux we might test for exact # thread number, since we always have with 1 thread per process, # but this does not apply across all platforms (MACOS, Windows) p = psutil.Process() if OPENBSD: try: step1 = p.num_threads() except psutil.AccessDenied: raise pytest.skip("on OpenBSD this requires root access") else: step1 = p.num_threads() with ThreadTask(): step2 = p.num_threads() assert step2 == step1 + 1 @pytest.mark.skipif(not WINDOWS, reason="WINDOWS only") def test_num_handles(self): # a better test is done later into test/_windows.py p = psutil.Process() assert p.num_handles() > 0 @pytest.mark.skipif(not HAS_THREADS, reason="not supported") def test_threads(self): p = psutil.Process() if OPENBSD: try: step1 = p.threads() except psutil.AccessDenied: raise pytest.skip("on OpenBSD this requires root access") else: step1 = p.threads() with ThreadTask(): step2 = p.threads() assert len(step2) == len(step1) + 1 athread = step2[0] # test named tuple assert athread.id == athread[0] assert athread.user_time == athread[1] assert athread.system_time == athread[2] @retry_on_failure() @skip_on_access_denied(only_if=MACOS) @pytest.mark.skipif(not HAS_THREADS, reason="not supported") def test_threads_2(self): p = self.spawn_psproc() if OPENBSD: try: p.threads() except psutil.AccessDenied: raise pytest.skip("on OpenBSD this requires root access") assert ( abs(p.cpu_times().user - sum(x.user_time for x in p.threads())) < 0.1 ) assert ( abs(p.cpu_times().system - sum(x.system_time for x in p.threads())) < 0.1 ) @retry_on_failure() def test_memory_info(self): p = psutil.Process() # step 1 - get a base value to compare our results rss1, vms1 = p.memory_info()[:2] percent1 = p.memory_percent() assert rss1 > 0 assert vms1 > 0 # step 2 - allocate some memory memarr = [None] * 1500000 rss2, vms2 = p.memory_info()[:2] percent2 = p.memory_percent() # step 3 - make sure that the memory usage bumped up assert rss2 > rss1 assert vms2 >= vms1 # vms might be equal assert percent2 > percent1 del memarr if WINDOWS: mem = p.memory_info() assert mem.rss == mem.wset assert mem.vms == mem.pagefile mem = p.memory_info() for name in mem._fields: assert getattr(mem, name) >= 0 def test_memory_full_info(self): p = psutil.Process() total = psutil.virtual_memory().total mem = p.memory_full_info() for name in mem._fields: value = getattr(mem, name) assert value >= 0 if (name == "vms" and OSX) or LINUX: continue assert value <= total if LINUX or WINDOWS or MACOS: assert mem.uss >= 0 if LINUX: assert mem.pss >= 0 assert mem.swap >= 0 @pytest.mark.skipif(not HAS_MEMORY_MAPS, reason="not supported") def test_memory_maps(self): p = psutil.Process() maps = p.memory_maps() assert len(maps) == len(set(maps)) ext_maps = p.memory_maps(grouped=False) for nt in maps: if nt.path.startswith('['): continue if BSD and nt.path == "pvclock": continue assert os.path.isabs(nt.path), nt.path if POSIX: try: assert os.path.exists(nt.path) or os.path.islink( nt.path ), nt.path except AssertionError: if not LINUX: raise # https://github.com/giampaolo/psutil/issues/759 with open_text('/proc/self/smaps') as f: data = f.read() if f"{nt.path} (deleted)" not in data: raise elif '64' not in os.path.basename(nt.path): # XXX - On Windows we have this strange behavior with # 64 bit dlls: they are visible via explorer but cannot # be accessed via os.stat() (wtf?). try: st = os.stat(nt.path) except FileNotFoundError: pass else: assert stat.S_ISREG(st.st_mode), nt.path for nt in ext_maps: for fname in nt._fields: value = getattr(nt, fname) if fname == 'path': continue if fname in {'addr', 'perms'}: assert value, value else: assert isinstance(value, int) assert value >= 0, value @pytest.mark.skipif(not HAS_MEMORY_MAPS, reason="not supported") def test_memory_maps_lists_lib(self): # Make sure a newly loaded shared lib is listed. p = psutil.Process() with copyload_shared_lib() as path: def normpath(p): return os.path.realpath(os.path.normcase(p)) libpaths = [normpath(x.path) for x in p.memory_maps()] assert normpath(path) in libpaths def test_memory_percent(self): p = psutil.Process() p.memory_percent() with pytest.raises(ValueError): p.memory_percent(memtype="?!?") if LINUX or MACOS or WINDOWS: p.memory_percent(memtype='uss') def test_is_running(self): p = self.spawn_psproc() assert p.is_running() assert p.is_running() p.kill() p.wait() assert not p.is_running() assert not p.is_running() def test_exe(self): p = self.spawn_psproc() exe = p.exe() try: assert exe == PYTHON_EXE except AssertionError: if WINDOWS and len(exe) == len(PYTHON_EXE): # on Windows we don't care about case sensitivity normcase = os.path.normcase assert normcase(exe) == normcase(PYTHON_EXE) else: # certain platforms such as BSD are more accurate returning: # "/usr/local/bin/python3.7" # ...instead of: # "/usr/local/bin/python" # We do not want to consider this difference in accuracy # an error. ver = f"{sys.version_info[0]}.{sys.version_info[1]}" try: assert exe.replace(ver, '') == PYTHON_EXE.replace(ver, '') except AssertionError: # Typically MACOS. Really not sure what to do here. pass out = sh([exe, "-c", "import os; print('hey')"]) assert out == 'hey' def test_cmdline(self): cmdline = [ PYTHON_EXE, "-c", "import time; [time.sleep(0.1) for x in range(100)]", ] p = self.spawn_psproc(cmdline) if NETBSD and p.cmdline() == []: # https://github.com/giampaolo/psutil/issues/2250 raise pytest.skip("OPENBSD: returned EBUSY") # XXX - most of the times the underlying sysctl() call on Net # and Open BSD returns a truncated string. # Also /proc/pid/cmdline behaves the same so it looks # like this is a kernel bug. # XXX - AIX truncates long arguments in /proc/pid/cmdline if NETBSD or OPENBSD or AIX: assert p.cmdline()[0] == PYTHON_EXE else: if MACOS and CI_TESTING: pyexe = p.cmdline()[0] if pyexe != PYTHON_EXE: assert ' '.join(p.cmdline()[1:]) == ' '.join(cmdline[1:]) return assert ' '.join(p.cmdline()) == ' '.join(cmdline) @pytest.mark.skipif(PYPY, reason="broken on PYPY") def test_long_cmdline(self): cmdline = [PYTHON_EXE] cmdline.extend(["-v"] * 50) cmdline.extend( ["-c", "import time; [time.sleep(0.1) for x in range(100)]"] ) p = self.spawn_psproc(cmdline) if OPENBSD: # XXX: for some reason the test process may turn into a # zombie (don't know why). try: assert p.cmdline() == cmdline except psutil.ZombieProcess: raise pytest.skip("OPENBSD: process turned into zombie") else: ret = p.cmdline() if NETBSD and ret == []: # https://github.com/giampaolo/psutil/issues/2250 raise pytest.skip("OPENBSD: returned EBUSY") assert ret == cmdline def test_name(self): p = self.spawn_psproc() name = p.name().lower() pyexe = os.path.basename(os.path.realpath(sys.executable)).lower() assert pyexe.startswith(name), (pyexe, name) @pytest.mark.skipif(PYPY, reason="unreliable on PYPY") def test_long_name(self): pyexe = create_py_exe(self.get_testfn(suffix=string.digits * 2)) cmdline = [ pyexe, "-c", "import time; [time.sleep(0.1) for x in range(100)]", ] p = self.spawn_psproc(cmdline) if OPENBSD: # XXX: for some reason the test process may turn into a # zombie (don't know why). Because the name() is long, all # UNIX kernels truncate it to 15 chars, so internally psutil # tries to guess the full name() from the cmdline(). But the # cmdline() of a zombie on OpenBSD fails (internally), so we # just compare the first 15 chars. Full explanation: # https://github.com/giampaolo/psutil/issues/2239 try: assert p.name() == os.path.basename(pyexe) except AssertionError: if p.status() == psutil.STATUS_ZOMBIE: assert os.path.basename(pyexe).startswith(p.name()) else: raise else: assert p.name() == os.path.basename(pyexe) # XXX: fails too often # @pytest.mark.skipif(SUNOS, reason="broken on SUNOS") # @pytest.mark.skipif(AIX, reason="broken on AIX") # @pytest.mark.skipif(PYPY, reason="broken on PYPY") # def test_prog_w_funky_name(self): # # Test that name(), exe() and cmdline() correctly handle programs # # with funky chars such as spaces and ")", see: # # https://github.com/giampaolo/psutil/issues/628 # pyexe = create_py_exe(self.get_testfn(suffix='foo bar )')) # cmdline = [ # pyexe, # "-c", # "import time; [time.sleep(0.1) for x in range(100)]", # ] # p = self.spawn_psproc(cmdline) # assert p.cmdline() == cmdline # assert p.name() == os.path.basename(pyexe) # assert os.path.normcase(p.exe()) == os.path.normcase(pyexe) @pytest.mark.skipif(not POSIX, reason="POSIX only") def test_uids(self): p = psutil.Process() real, effective, _saved = p.uids() # os.getuid() refers to "real" uid assert real == os.getuid() # os.geteuid() refers to "effective" uid assert effective == os.geteuid() # No such thing as os.getsuid() ("saved" uid), but we have # os.getresuid() which returns all of them. if hasattr(os, "getresuid"): assert os.getresuid() == p.uids() @pytest.mark.skipif(not POSIX, reason="POSIX only") def test_gids(self): p = psutil.Process() real, effective, _saved = p.gids() # os.getuid() refers to "real" uid assert real == os.getgid() # os.geteuid() refers to "effective" uid assert effective == os.getegid() # No such thing as os.getsgid() ("saved" gid), but we have # os.getresgid() which returns all of them. if hasattr(os, "getresuid"): assert os.getresgid() == p.gids() def test_nice(self): def cleanup(init): try: p.nice(init) except psutil.AccessDenied: pass p = psutil.Process() with pytest.raises(TypeError): p.nice("str") init = p.nice() self.addCleanup(cleanup, init) if WINDOWS: highest_prio = None for prio in [ psutil.IDLE_PRIORITY_CLASS, psutil.BELOW_NORMAL_PRIORITY_CLASS, psutil.NORMAL_PRIORITY_CLASS, psutil.ABOVE_NORMAL_PRIORITY_CLASS, psutil.HIGH_PRIORITY_CLASS, psutil.REALTIME_PRIORITY_CLASS, ]: with self.subTest(prio=prio): try: p.nice(prio) except psutil.AccessDenied: pass else: new_prio = p.nice() # The OS may limit our maximum priority, # even if the function succeeds. For higher # priorities, we match either the expected # value or the highest so far. if prio in { psutil.ABOVE_NORMAL_PRIORITY_CLASS, psutil.HIGH_PRIORITY_CLASS, psutil.REALTIME_PRIORITY_CLASS, }: if new_prio == prio or highest_prio is None: highest_prio = prio assert new_prio == highest_prio else: assert new_prio == prio else: try: if hasattr(os, "getpriority"): assert ( os.getpriority(os.PRIO_PROCESS, os.getpid()) == p.nice() ) p.nice(1) assert p.nice() == 1 if hasattr(os, "getpriority"): assert ( os.getpriority(os.PRIO_PROCESS, os.getpid()) == p.nice() ) # XXX - going back to previous nice value raises # AccessDenied on MACOS if not MACOS: p.nice(0) assert p.nice() == 0 except psutil.AccessDenied: pass def test_status(self): p = psutil.Process() assert p.status() == psutil.STATUS_RUNNING def test_username(self): p = self.spawn_psproc() username = p.username() if WINDOWS: domain, username = username.split('\\') getpass_user = getpass.getuser() if getpass_user.endswith('$'): # When running as a service account (most likely to be # NetworkService), these user name calculations don't produce # the same result, causing the test to fail. raise pytest.skip('running as service account') assert username == getpass_user if 'USERDOMAIN' in os.environ: assert domain == os.environ['USERDOMAIN'] else: assert username == getpass.getuser() def test_cwd(self): p = self.spawn_psproc() assert p.cwd() == os.getcwd() def test_cwd_2(self): cmd = [ PYTHON_EXE, "-c", ( "import os, time; os.chdir('..'); [time.sleep(0.1) for x in" " range(100)]" ), ] p = self.spawn_psproc(cmd) call_until(lambda: p.cwd() == os.path.dirname(os.getcwd())) @pytest.mark.skipif(not HAS_CPU_AFFINITY, reason="not supported") def test_cpu_affinity(self): p = psutil.Process() initial = p.cpu_affinity() assert initial, initial self.addCleanup(p.cpu_affinity, initial) if hasattr(os, "sched_getaffinity"): assert initial == list(os.sched_getaffinity(p.pid)) assert len(initial) == len(set(initial)) all_cpus = list(range(len(psutil.cpu_percent(percpu=True)))) for n in all_cpus: p.cpu_affinity([n]) assert p.cpu_affinity() == [n] if hasattr(os, "sched_getaffinity"): assert p.cpu_affinity() == list(os.sched_getaffinity(p.pid)) # also test num_cpu() if hasattr(p, "num_cpu"): assert p.cpu_affinity()[0] == p.num_cpu() # [] is an alias for "all eligible CPUs"; on Linux this may # not be equal to all available CPUs, see: # https://github.com/giampaolo/psutil/issues/956 p.cpu_affinity([]) if LINUX: assert p.cpu_affinity() == p._proc._get_eligible_cpus() else: assert p.cpu_affinity() == all_cpus if hasattr(os, "sched_getaffinity"): assert p.cpu_affinity() == list(os.sched_getaffinity(p.pid)) with pytest.raises(TypeError): p.cpu_affinity(1) p.cpu_affinity(initial) # it should work with all iterables, not only lists p.cpu_affinity(set(all_cpus)) p.cpu_affinity(tuple(all_cpus)) @pytest.mark.skipif(not HAS_CPU_AFFINITY, reason="not supported") def test_cpu_affinity_errs(self): p = self.spawn_psproc() invalid_cpu = [len(psutil.cpu_times(percpu=True)) + 10] with pytest.raises(ValueError): p.cpu_affinity(invalid_cpu) with pytest.raises(ValueError): p.cpu_affinity(range(10000, 11000)) with pytest.raises((TypeError, ValueError)): p.cpu_affinity([0, "1"]) with pytest.raises(ValueError): p.cpu_affinity([0, -1]) @pytest.mark.skipif(not HAS_CPU_AFFINITY, reason="not supported") def test_cpu_affinity_all_combinations(self): p = psutil.Process() initial = p.cpu_affinity() assert initial, initial self.addCleanup(p.cpu_affinity, initial) # All possible CPU set combinations. if len(initial) > 12: initial = initial[:12] # ...otherwise it will take forever combos = [] for i in range(len(initial) + 1): combos.extend( list(subset) for subset in itertools.combinations(initial, i) if subset ) for combo in combos: p.cpu_affinity(combo) assert sorted(p.cpu_affinity()) == sorted(combo) # TODO: #595 @pytest.mark.skipif(BSD, reason="broken on BSD") def test_open_files(self): p = psutil.Process() testfn = self.get_testfn() files = p.open_files() assert testfn not in files with open(testfn, 'wb') as f: f.write(b'x' * 1024) f.flush() # give the kernel some time to see the new file call_until(lambda: len(p.open_files()) != len(files)) files = p.open_files() filenames = [os.path.normcase(x.path) for x in files] assert os.path.normcase(testfn) in filenames if LINUX: for file in files: if file.path == testfn: assert file.position == 1024 for file in files: assert os.path.isfile(file.path), file # another process cmdline = ( f"import time; f = open(r'{testfn}', 'r'); [time.sleep(0.1) for x" " in range(100)];" ) p = self.spawn_psproc([PYTHON_EXE, "-c", cmdline]) for x in range(100): filenames = [os.path.normcase(x.path) for x in p.open_files()] if testfn in filenames: break time.sleep(0.01) else: assert os.path.normcase(testfn) in filenames for file in filenames: assert os.path.isfile(file), file # TODO: #595 @pytest.mark.skipif(BSD, reason="broken on BSD") def test_open_files_2(self): # test fd and path fields p = psutil.Process() normcase = os.path.normcase testfn = self.get_testfn() with open(testfn, 'w') as fileobj: for file in p.open_files(): if ( normcase(file.path) == normcase(fileobj.name) or file.fd == fileobj.fileno() ): break else: raise self.fail(f"no file found; files={p.open_files()!r}") assert normcase(file.path) == normcase(fileobj.name) if WINDOWS: assert file.fd == -1 else: assert file.fd == fileobj.fileno() # test positions ntuple = p.open_files()[0] assert ntuple[0] == ntuple.path assert ntuple[1] == ntuple.fd # test file is gone assert fileobj.name not in p.open_files() @pytest.mark.skipif(not POSIX, reason="POSIX only") def test_num_fds(self): p = psutil.Process() testfn = self.get_testfn() start = p.num_fds() file = open(testfn, 'w') # noqa: SIM115 self.addCleanup(file.close) assert p.num_fds() == start + 1 sock = socket.socket() self.addCleanup(sock.close) assert p.num_fds() == start + 2 file.close() sock.close() assert p.num_fds() == start @skip_on_not_implemented(only_if=LINUX) @pytest.mark.skipif( OPENBSD or NETBSD, reason="not reliable on OPENBSD & NETBSD" ) def test_num_ctx_switches(self): p = psutil.Process() before = sum(p.num_ctx_switches()) for _ in range(2): time.sleep(0.05) # this shall ensure a context switch happens after = sum(p.num_ctx_switches()) if after > before: return raise self.fail("num ctx switches still the same after 2 iterations") def test_ppid(self): p = psutil.Process() if hasattr(os, 'getppid'): assert p.ppid() == os.getppid() p = self.spawn_psproc() assert p.ppid() == os.getpid() def test_parent(self): p = self.spawn_psproc() assert p.parent().pid == os.getpid() lowest_pid = psutil.pids()[0] assert psutil.Process(lowest_pid).parent() is None def test_parent_multi(self): parent = psutil.Process() child, grandchild = self.spawn_children_pair() assert grandchild.parent() == child assert child.parent() == parent @retry_on_failure() def test_parents(self): parent = psutil.Process() assert parent.parents() child, grandchild = self.spawn_children_pair() assert child.parents()[0] == parent assert grandchild.parents()[0] == child assert grandchild.parents()[1] == parent def test_children(self): parent = psutil.Process() assert not parent.children() assert not parent.children(recursive=True) # On Windows we set the flag to 0 in order to cancel out the # CREATE_NO_WINDOW flag (enabled by default) which creates # an extra "conhost.exe" child. child = self.spawn_psproc(creationflags=0) children1 = parent.children() children2 = parent.children(recursive=True) for children in (children1, children2): assert len(children) == 1 assert children[0].pid == child.pid assert children[0].ppid() == parent.pid def test_children_recursive(self): # Test children() against two sub processes, p1 and p2, where # p1 (our child) spawned p2 (our grandchild). parent = psutil.Process() child, grandchild = self.spawn_children_pair() assert parent.children() == [child] assert parent.children(recursive=True) == [child, grandchild] # If the intermediate process is gone there's no way for # children() to recursively find it. child.terminate() child.wait() assert not parent.children(recursive=True) def test_children_duplicates(self): # find the process which has the highest number of children table = collections.defaultdict(int) for p in psutil.process_iter(): try: table[p.ppid()] += 1 except psutil.Error: pass # this is the one, now let's make sure there are no duplicates pid = max(table.items(), key=lambda x: x[1])[0] if LINUX and pid == 0: raise pytest.skip("PID 0") p = psutil.Process(pid) try: c = p.children(recursive=True) except psutil.AccessDenied: # windows pass else: assert len(c) == len(set(c)) def test_parents_and_children(self): parent = psutil.Process() child, grandchild = self.spawn_children_pair() # forward children = parent.children(recursive=True) assert len(children) == 2 assert children[0] == child assert children[1] == grandchild # backward parents = grandchild.parents() assert parents[0] == child assert parents[1] == parent def test_suspend_resume(self): p = self.spawn_psproc() p.suspend() for _ in range(100): if p.status() == psutil.STATUS_STOPPED: break time.sleep(0.01) p.resume() assert p.status() != psutil.STATUS_STOPPED def test_invalid_pid(self): with pytest.raises(TypeError): psutil.Process("1") with pytest.raises(ValueError): psutil.Process(-1) def test_as_dict(self): p = psutil.Process() d = p.as_dict(attrs=['exe', 'name']) assert sorted(d.keys()) == ['exe', 'name'] p = psutil.Process(min(psutil.pids())) d = p.as_dict(attrs=['net_connections'], ad_value='foo') if not isinstance(d['net_connections'], list): assert d['net_connections'] == 'foo' # Test ad_value is set on AccessDenied. with mock.patch( 'psutil.Process.nice', create=True, side_effect=psutil.AccessDenied ): assert p.as_dict(attrs=["nice"], ad_value=1) == {"nice": 1} # Test that NoSuchProcess bubbles up. with mock.patch( 'psutil.Process.nice', create=True, side_effect=psutil.NoSuchProcess(p.pid, "name"), ): with pytest.raises(psutil.NoSuchProcess): p.as_dict(attrs=["nice"]) # Test that ZombieProcess is swallowed. with mock.patch( 'psutil.Process.nice', create=True, side_effect=psutil.ZombieProcess(p.pid, "name"), ): assert p.as_dict(attrs=["nice"], ad_value="foo") == {"nice": "foo"} # By default APIs raising NotImplementedError are # supposed to be skipped. with mock.patch( 'psutil.Process.nice', create=True, side_effect=NotImplementedError ): d = p.as_dict() assert 'nice' not in list(d.keys()) # ...unless the user explicitly asked for some attr. with pytest.raises(NotImplementedError): p.as_dict(attrs=["nice"]) # errors with pytest.raises(TypeError): p.as_dict('name') with pytest.raises(ValueError): p.as_dict(['foo']) with pytest.raises(ValueError): p.as_dict(['foo', 'bar']) def test_oneshot(self): p = psutil.Process() with mock.patch("psutil._psplatform.Process.cpu_times") as m: with p.oneshot(): p.cpu_times() p.cpu_times() assert m.call_count == 1 with mock.patch("psutil._psplatform.Process.cpu_times") as m: p.cpu_times() p.cpu_times() assert m.call_count == 2 def test_oneshot_twice(self): # Test the case where the ctx manager is __enter__ed twice. # The second __enter__ is supposed to resut in a NOOP. p = psutil.Process() with mock.patch("psutil._psplatform.Process.cpu_times") as m1: with mock.patch("psutil._psplatform.Process.oneshot_enter") as m2: with p.oneshot(): p.cpu_times() p.cpu_times() with p.oneshot(): p.cpu_times() p.cpu_times() assert m1.call_count == 1 assert m2.call_count == 1 with mock.patch("psutil._psplatform.Process.cpu_times") as m: p.cpu_times() p.cpu_times() assert m.call_count == 2 def test_oneshot_cache(self): # Make sure oneshot() cache is nonglobal. Instead it's # supposed to be bound to the Process instance, see: # https://github.com/giampaolo/psutil/issues/1373 p1, p2 = self.spawn_children_pair() p1_ppid = p1.ppid() p2_ppid = p2.ppid() assert p1_ppid != p2_ppid with p1.oneshot(): assert p1.ppid() == p1_ppid assert p2.ppid() == p2_ppid with p2.oneshot(): assert p1.ppid() == p1_ppid assert p2.ppid() == p2_ppid def test_halfway_terminated_process(self): # Test that NoSuchProcess exception gets raised in case the # process dies after we create the Process object. # Example: # >>> proc = Process(1234) # >>> time.sleep(2) # time-consuming task, process dies in meantime # >>> proc.name() # Refers to Issue #15 def assert_raises_nsp(fun, fun_name): try: ret = fun() except psutil.ZombieProcess: # differentiate from NSP raise except psutil.NoSuchProcess: pass except psutil.AccessDenied: if OPENBSD and fun_name in {'threads', 'num_threads'}: return raise else: # NtQuerySystemInformation succeeds even if process is gone. if WINDOWS and fun_name in {'exe', 'name'}: return raise self.fail( f"{fun!r} didn't raise NSP and returned {ret!r} instead" ) p = self.spawn_psproc() p.terminate() p.wait() if WINDOWS: # XXX call_until(lambda: p.pid not in psutil.pids()) self.assertProcessGone(p) ns = process_namespace(p) for fun, name in ns.iter(ns.all): assert_raises_nsp(fun, name) @pytest.mark.skipif(not POSIX, reason="POSIX only") def test_zombie_process(self): _parent, zombie = self.spawn_zombie() self.assertProcessZombie(zombie) @pytest.mark.skipif(not POSIX, reason="POSIX only") def test_zombie_process_is_running_w_exc(self): # Emulate a case where internally is_running() raises # ZombieProcess. p = psutil.Process() with mock.patch( "psutil.Process", side_effect=psutil.ZombieProcess(0) ) as m: assert p.is_running() assert m.called @pytest.mark.skipif(not POSIX, reason="POSIX only") def test_zombie_process_status_w_exc(self): # Emulate a case where internally status() raises # ZombieProcess. p = psutil.Process() with mock.patch( "psutil._psplatform.Process.status", side_effect=psutil.ZombieProcess(0), ) as m: assert p.status() == psutil.STATUS_ZOMBIE assert m.called def test_reused_pid(self): # Emulate a case where PID has been reused by another process. subp = self.spawn_testproc() p = psutil.Process(subp.pid) p._ident = (p.pid, p.create_time() + 100) list(psutil.process_iter()) assert p.pid in psutil._pmap assert not p.is_running() # make sure is_running() removed PID from process_iter() # internal cache with mock.patch.object(psutil._common, "PSUTIL_DEBUG", True): with contextlib.redirect_stderr(io.StringIO()) as f: list(psutil.process_iter()) assert ( f"refreshing Process instance for reused PID {p.pid}" in f.getvalue() ) assert p.pid not in psutil._pmap assert p != psutil.Process(subp.pid) msg = "process no longer exists and its PID has been reused" ns = process_namespace(p) for fun, name in ns.iter(ns.setters + ns.killers, clear_cache=False): with self.subTest(name=name): with pytest.raises(psutil.NoSuchProcess, match=msg): fun() assert "terminated + PID reused" in str(p) assert "terminated + PID reused" in repr(p) with pytest.raises(psutil.NoSuchProcess, match=msg): p.ppid() with pytest.raises(psutil.NoSuchProcess, match=msg): p.parent() with pytest.raises(psutil.NoSuchProcess, match=msg): p.parents() with pytest.raises(psutil.NoSuchProcess, match=msg): p.children() def test_pid_0(self): # Process(0) is supposed to work on all platforms except Linux if 0 not in psutil.pids(): with pytest.raises(psutil.NoSuchProcess): psutil.Process(0) # These 2 are a contradiction, but "ps" says PID 1's parent # is PID 0. assert not psutil.pid_exists(0) assert psutil.Process(1).ppid() == 0 return p = psutil.Process(0) exc = psutil.AccessDenied if WINDOWS else ValueError with pytest.raises(exc): p.wait() with pytest.raises(exc): p.terminate() with pytest.raises(exc): p.suspend() with pytest.raises(exc): p.resume() with pytest.raises(exc): p.kill() with pytest.raises(exc): p.send_signal(signal.SIGTERM) # test all methods ns = process_namespace(p) for fun, name in ns.iter(ns.getters + ns.setters): try: ret = fun() except psutil.AccessDenied: pass else: if name in {"uids", "gids"}: assert ret.real == 0 elif name == "username": user = 'NT AUTHORITY\\SYSTEM' if WINDOWS else 'root' assert p.username() == user elif name == "name": assert name, name if not OPENBSD: assert 0 in psutil.pids() assert psutil.pid_exists(0) @pytest.mark.skipif(not HAS_ENVIRON, reason="not supported") def test_environ(self): def clean_dict(d): exclude = ["PLAT", "HOME", "PYTEST_CURRENT_TEST", "PYTEST_VERSION"] if MACOS: exclude.extend([ "__CF_USER_TEXT_ENCODING", "VERSIONER_PYTHON_PREFER_32_BIT", "VERSIONER_PYTHON_VERSION", "VERSIONER_PYTHON_VERSION", ]) for name in exclude: d.pop(name, None) return { k.replace("\r", "").replace("\n", ""): v.replace( "\r", "" ).replace("\n", "") for k, v in d.items() } self.maxDiff = None p = psutil.Process() d1 = clean_dict(p.environ()) d2 = clean_dict(os.environ.copy()) if not OSX and GITHUB_ACTIONS: assert d1 == d2 @pytest.mark.skipif(not HAS_ENVIRON, reason="not supported") @pytest.mark.skipif(not POSIX, reason="POSIX only") @pytest.mark.skipif( MACOS_11PLUS, reason="macOS 11+ can't get another process environment, issue #2084", ) @pytest.mark.skipif( NETBSD, reason="sometimes fails on `assert is_running()`" ) def test_weird_environ(self): # environment variables can contain values without an equals sign code = textwrap.dedent(""" #include <unistd.h> #include <fcntl.h> char * const argv[] = {"cat", 0}; char * const envp[] = {"A=1", "X", "C=3", 0}; int main(void) { // Close stderr on exec so parent can wait for the // execve to finish. if (fcntl(2, F_SETFD, FD_CLOEXEC) != 0) return 0; return execve("/bin/cat", argv, envp); } """) cexe = create_c_exe(self.get_testfn(), c_code=code) sproc = self.spawn_testproc( [cexe], stdin=subprocess.PIPE, stderr=subprocess.PIPE ) p = psutil.Process(sproc.pid) wait_for_pid(p.pid) assert p.is_running() # Wait for process to exec or exit. assert sproc.stderr.read() == b"" if MACOS and CI_TESTING: try: env = p.environ() except psutil.AccessDenied: # XXX: fails sometimes with: # PermissionError from 'sysctl(KERN_PROCARGS2) -> EIO' return else: env = p.environ() assert env == {"A": "1", "C": "3"} sproc.communicate() assert sproc.returncode == 0 # =================================================================== # --- psutil.Popen tests # =================================================================== class TestPopen(PsutilTestCase): """Tests for psutil.Popen class.""" @classmethod def tearDownClass(cls): reap_children() def test_misc(self): # XXX this test causes a ResourceWarning because # psutil.__subproc instance doesn't get properly freed. # Not sure what to do though. cmd = [ PYTHON_EXE, "-c", "import time; [time.sleep(0.1) for x in range(100)];", ] with psutil.Popen( cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE, env=PYTHON_EXE_ENV, ) as proc: proc.name() proc.cpu_times() proc.stdin # noqa: B018 assert dir(proc) with pytest.raises(AttributeError): proc.foo # noqa: B018 proc.terminate() if POSIX: assert proc.wait(5) == -signal.SIGTERM else: assert proc.wait(5) == signal.SIGTERM def test_ctx_manager(self): with psutil.Popen( [PYTHON_EXE, "-V"], stdout=subprocess.PIPE, stderr=subprocess.PIPE, stdin=subprocess.PIPE, env=PYTHON_EXE_ENV, ) as proc: proc.communicate() assert proc.stdout.closed assert proc.stderr.closed assert proc.stdin.closed assert proc.returncode == 0 def test_kill_terminate(self): # subprocess.Popen()'s terminate(), kill() and send_signal() do # not raise exception after the process is gone. psutil.Popen # diverges from that. cmd = [ PYTHON_EXE, "-c", "import time; [time.sleep(0.1) for x in range(100)];", ] with psutil.Popen( cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE, env=PYTHON_EXE_ENV, ) as proc: proc.terminate() proc.wait() with pytest.raises(psutil.NoSuchProcess): proc.terminate() with pytest.raises(psutil.NoSuchProcess): proc.kill() with pytest.raises(psutil.NoSuchProcess): proc.send_signal(signal.SIGTERM) if WINDOWS: with pytest.raises(psutil.NoSuchProcess): proc.send_signal(signal.CTRL_C_EVENT) with pytest.raises(psutil.NoSuchProcess): proc.send_signal(signal.CTRL_BREAK_EVENT) def test__getattribute__(self): cmd = [ PYTHON_EXE, "-c", "import time; [time.sleep(0.1) for x in range(100)];", ] with psutil.Popen( cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE, env=PYTHON_EXE_ENV, ) as proc: proc.terminate() proc.wait() with pytest.raises(AttributeError): proc.foo # noqa: B018 tests/__main__.py 0000644 00000000465 15030211765 0010005 0 ustar 00 # Copyright (c) 2009, Giampaolo Rodola'. All rights reserved. # Use of this source code is governed by a BSD-style license that can be # found in the LICENSE file. """Run unit tests. This is invoked by: $ python -m psutil.tests. """ from psutil.tests import pytest pytest.main(["-v", "-s", "--tb=short"]) tests/test_sunos.py 0000644 00000002246 15030211765 0010472 0 ustar 00 #!/usr/bin/env python3 # Copyright (c) 2009, Giampaolo Rodola'. All rights reserved. # Use of this source code is governed by a BSD-style license that can be # found in the LICENSE file. """Sun OS specific tests.""" import os import psutil from psutil import SUNOS from psutil.tests import PsutilTestCase from psutil.tests import pytest from psutil.tests import sh @pytest.mark.skipif(not SUNOS, reason="SUNOS only") class SunOSSpecificTestCase(PsutilTestCase): def test_swap_memory(self): out = sh(f"env PATH=/usr/sbin:/sbin:{os.environ['PATH']} swap -l") lines = out.strip().split('\n')[1:] if not lines: raise ValueError('no swap device(s) configured') total = free = 0 for line in lines: fields = line.split() total = int(fields[3]) * 512 free = int(fields[4]) * 512 used = total - free psutil_swap = psutil.swap_memory() assert psutil_swap.total == total assert psutil_swap.used == used assert psutil_swap.free == free def test_cpu_count(self): out = sh("/usr/sbin/psrinfo") assert psutil.cpu_count() == len(out.split('\n')) tests/test_system.py 0000644 00000106413 15030211765 0010650 0 ustar 00 #!/usr/bin/env python3 # Copyright (c) 2009, Giampaolo Rodola'. All rights reserved. # Use of this source code is governed by a BSD-style license that can be # found in the LICENSE file. """Tests for system APIS.""" import datetime import enum import errno import os import platform import pprint import shutil import signal import socket import sys import time from unittest import mock import psutil from psutil import AIX from psutil import BSD from psutil import FREEBSD from psutil import LINUX from psutil import MACOS from psutil import NETBSD from psutil import OPENBSD from psutil import POSIX from psutil import SUNOS from psutil import WINDOWS from psutil._common import broadcast_addr from psutil.tests import AARCH64 from psutil.tests import ASCII_FS from psutil.tests import CI_TESTING from psutil.tests import GITHUB_ACTIONS from psutil.tests import GLOBAL_TIMEOUT from psutil.tests import HAS_BATTERY from psutil.tests import HAS_CPU_FREQ from psutil.tests import HAS_GETLOADAVG from psutil.tests import HAS_NET_IO_COUNTERS from psutil.tests import HAS_SENSORS_BATTERY from psutil.tests import HAS_SENSORS_FANS from psutil.tests import HAS_SENSORS_TEMPERATURES from psutil.tests import IS_64BIT from psutil.tests import MACOS_12PLUS from psutil.tests import PYPY from psutil.tests import UNICODE_SUFFIX from psutil.tests import PsutilTestCase from psutil.tests import check_net_address from psutil.tests import pytest from psutil.tests import retry_on_failure # =================================================================== # --- System-related API tests # =================================================================== class TestProcessIter(PsutilTestCase): def test_pid_presence(self): assert os.getpid() in [x.pid for x in psutil.process_iter()] sproc = self.spawn_testproc() assert sproc.pid in [x.pid for x in psutil.process_iter()] p = psutil.Process(sproc.pid) p.kill() p.wait() assert sproc.pid not in [x.pid for x in psutil.process_iter()] def test_no_duplicates(self): ls = list(psutil.process_iter()) assert sorted(ls, key=lambda x: x.pid) == sorted( set(ls), key=lambda x: x.pid ) def test_emulate_nsp(self): list(psutil.process_iter()) # populate cache for x in range(2): with mock.patch( 'psutil.Process.as_dict', side_effect=psutil.NoSuchProcess(os.getpid()), ): assert not list(psutil.process_iter(attrs=["cpu_times"])) psutil.process_iter.cache_clear() # repeat test without cache def test_emulate_access_denied(self): list(psutil.process_iter()) # populate cache for x in range(2): with mock.patch( 'psutil.Process.as_dict', side_effect=psutil.AccessDenied(os.getpid()), ): with pytest.raises(psutil.AccessDenied): list(psutil.process_iter(attrs=["cpu_times"])) psutil.process_iter.cache_clear() # repeat test without cache def test_attrs(self): for p in psutil.process_iter(attrs=['pid']): assert list(p.info.keys()) == ['pid'] # yield again for p in psutil.process_iter(attrs=['pid']): assert list(p.info.keys()) == ['pid'] with pytest.raises(ValueError): list(psutil.process_iter(attrs=['foo'])) with mock.patch( "psutil._psplatform.Process.cpu_times", side_effect=psutil.AccessDenied(0, ""), ) as m: for p in psutil.process_iter(attrs=["pid", "cpu_times"]): assert p.info['cpu_times'] is None assert p.info['pid'] >= 0 assert m.called with mock.patch( "psutil._psplatform.Process.cpu_times", side_effect=psutil.AccessDenied(0, ""), ) as m: flag = object() for p in psutil.process_iter( attrs=["pid", "cpu_times"], ad_value=flag ): assert p.info['cpu_times'] is flag assert p.info['pid'] >= 0 assert m.called def test_cache_clear(self): list(psutil.process_iter()) # populate cache assert psutil._pmap psutil.process_iter.cache_clear() assert not psutil._pmap class TestProcessAPIs(PsutilTestCase): @pytest.mark.skipif( PYPY and WINDOWS, reason="spawn_testproc() unreliable on PYPY + WINDOWS", ) def test_wait_procs(self): def callback(p): pids.append(p.pid) pids = [] sproc1 = self.spawn_testproc() sproc2 = self.spawn_testproc() sproc3 = self.spawn_testproc() procs = [psutil.Process(x.pid) for x in (sproc1, sproc2, sproc3)] with pytest.raises(ValueError): psutil.wait_procs(procs, timeout=-1) with pytest.raises(TypeError): psutil.wait_procs(procs, callback=1) t = time.time() gone, alive = psutil.wait_procs(procs, timeout=0.01, callback=callback) assert time.time() - t < 0.5 assert not gone assert len(alive) == 3 assert not pids for p in alive: assert not hasattr(p, 'returncode') @retry_on_failure(30) def test_1(procs, callback): gone, alive = psutil.wait_procs( procs, timeout=0.03, callback=callback ) assert len(gone) == 1 assert len(alive) == 2 return gone, alive sproc3.terminate() gone, alive = test_1(procs, callback) assert sproc3.pid in [x.pid for x in gone] if POSIX: assert gone.pop().returncode == -signal.SIGTERM else: assert gone.pop().returncode == 1 assert pids == [sproc3.pid] for p in alive: assert not hasattr(p, 'returncode') @retry_on_failure(30) def test_2(procs, callback): gone, alive = psutil.wait_procs( procs, timeout=0.03, callback=callback ) assert len(gone) == 3 assert len(alive) == 0 return gone, alive sproc1.terminate() sproc2.terminate() gone, alive = test_2(procs, callback) assert set(pids) == {sproc1.pid, sproc2.pid, sproc3.pid} for p in gone: assert hasattr(p, 'returncode') @pytest.mark.skipif( PYPY and WINDOWS, reason="spawn_testproc() unreliable on PYPY + WINDOWS", ) def test_wait_procs_no_timeout(self): sproc1 = self.spawn_testproc() sproc2 = self.spawn_testproc() sproc3 = self.spawn_testproc() procs = [psutil.Process(x.pid) for x in (sproc1, sproc2, sproc3)] for p in procs: p.terminate() psutil.wait_procs(procs) def test_pid_exists(self): sproc = self.spawn_testproc() assert psutil.pid_exists(sproc.pid) p = psutil.Process(sproc.pid) p.kill() p.wait() assert not psutil.pid_exists(sproc.pid) assert not psutil.pid_exists(-1) assert psutil.pid_exists(0) == (0 in psutil.pids()) def test_pid_exists_2(self): pids = psutil.pids() for pid in pids: try: assert psutil.pid_exists(pid) except AssertionError: # in case the process disappeared in meantime fail only # if it is no longer in psutil.pids() time.sleep(0.1) assert pid not in psutil.pids() pids = range(max(pids) + 15000, max(pids) + 16000) for pid in pids: assert not psutil.pid_exists(pid) class TestMiscAPIs(PsutilTestCase): def test_boot_time(self): bt = psutil.boot_time() assert isinstance(bt, float) assert bt > 0 assert bt < time.time() @pytest.mark.skipif( CI_TESTING and not psutil.users(), reason="unreliable on CI" ) def test_users(self): users = psutil.users() assert users for user in users: with self.subTest(user=user): assert user.name assert isinstance(user.name, str) assert isinstance(user.terminal, (str, type(None))) if user.host is not None: assert isinstance(user.host, (str, type(None))) user.terminal # noqa: B018 user.host # noqa: B018 assert user.started > 0.0 datetime.datetime.fromtimestamp(user.started) if WINDOWS or OPENBSD: assert user.pid is None else: psutil.Process(user.pid) def test_os_constants(self): names = [ "POSIX", "WINDOWS", "LINUX", "MACOS", "FREEBSD", "OPENBSD", "NETBSD", "BSD", "SUNOS", ] for name in names: assert isinstance(getattr(psutil, name), bool), name if os.name == 'posix': assert psutil.POSIX assert not psutil.WINDOWS names.remove("POSIX") if "linux" in sys.platform.lower(): assert psutil.LINUX names.remove("LINUX") elif "bsd" in sys.platform.lower(): assert psutil.BSD assert [psutil.FREEBSD, psutil.OPENBSD, psutil.NETBSD].count( True ) == 1 names.remove("BSD") names.remove("FREEBSD") names.remove("OPENBSD") names.remove("NETBSD") elif ( "sunos" in sys.platform.lower() or "solaris" in sys.platform.lower() ): assert psutil.SUNOS names.remove("SUNOS") elif "darwin" in sys.platform.lower(): assert psutil.MACOS names.remove("MACOS") else: assert psutil.WINDOWS assert not psutil.POSIX names.remove("WINDOWS") # assert all other constants are set to False for name in names: assert not getattr(psutil, name), name class TestMemoryAPIs(PsutilTestCase): def test_virtual_memory(self): mem = psutil.virtual_memory() assert mem.total > 0, mem assert mem.available > 0, mem assert 0 <= mem.percent <= 100, mem assert mem.used > 0, mem assert mem.free >= 0, mem for name in mem._fields: value = getattr(mem, name) if name != 'percent': assert isinstance(value, int) if name != 'total': if not value >= 0: raise self.fail(f"{name!r} < 0 ({value})") if value > mem.total: raise self.fail( f"{name!r} > total (total={mem.total}, {name}={value})" ) def test_swap_memory(self): mem = psutil.swap_memory() assert mem._fields == ( 'total', 'used', 'free', 'percent', 'sin', 'sout', ) assert mem.total >= 0, mem assert mem.used >= 0, mem if mem.total > 0: # likely a system with no swap partition assert mem.free > 0, mem else: assert mem.free == 0, mem assert 0 <= mem.percent <= 100, mem assert mem.sin >= 0, mem assert mem.sout >= 0, mem class TestCpuAPIs(PsutilTestCase): def test_cpu_count_logical(self): logical = psutil.cpu_count() assert logical is not None assert logical == len(psutil.cpu_times(percpu=True)) assert logical >= 1 if os.path.exists("/proc/cpuinfo"): with open("/proc/cpuinfo") as fd: cpuinfo_data = fd.read() if "physical id" not in cpuinfo_data: raise pytest.skip("cpuinfo doesn't include physical id") def test_cpu_count_cores(self): logical = psutil.cpu_count() cores = psutil.cpu_count(logical=False) if cores is None: raise pytest.skip("cpu_count_cores() is None") if WINDOWS and sys.getwindowsversion()[:2] <= (6, 1): # <= Vista assert cores is None else: assert cores >= 1 assert logical >= cores def test_cpu_count_none(self): # https://github.com/giampaolo/psutil/issues/1085 for val in (-1, 0, None): with mock.patch( 'psutil._psplatform.cpu_count_logical', return_value=val ) as m: assert psutil.cpu_count() is None assert m.called with mock.patch( 'psutil._psplatform.cpu_count_cores', return_value=val ) as m: assert psutil.cpu_count(logical=False) is None assert m.called def test_cpu_times(self): # Check type, value >= 0, str(). total = 0 times = psutil.cpu_times() sum(times) for cp_time in times: assert isinstance(cp_time, float) assert cp_time >= 0.0 total += cp_time assert round(abs(total - sum(times)), 6) == 0 str(times) # CPU times are always supposed to increase over time # or at least remain the same and that's because time # cannot go backwards. # Surprisingly sometimes this might not be the case (at # least on Windows and Linux), see: # https://github.com/giampaolo/psutil/issues/392 # https://github.com/giampaolo/psutil/issues/645 # if not WINDOWS: # last = psutil.cpu_times() # for x in range(100): # new = psutil.cpu_times() # for field in new._fields: # new_t = getattr(new, field) # last_t = getattr(last, field) # self.assertGreaterEqual( # new_t, last_t, # msg="{} {}".format(new_t, last_t)) # last = new def test_cpu_times_time_increases(self): # Make sure time increases between calls. t1 = sum(psutil.cpu_times()) stop_at = time.time() + GLOBAL_TIMEOUT while time.time() < stop_at: t2 = sum(psutil.cpu_times()) if t2 > t1: return raise self.fail("time remained the same") def test_per_cpu_times(self): # Check type, value >= 0, str(). for times in psutil.cpu_times(percpu=True): total = 0 sum(times) for cp_time in times: assert isinstance(cp_time, float) assert cp_time >= 0.0 total += cp_time assert round(abs(total - sum(times)), 6) == 0 str(times) assert len(psutil.cpu_times(percpu=True)[0]) == len( psutil.cpu_times(percpu=False) ) # Note: in theory CPU times are always supposed to increase over # time or remain the same but never go backwards. In practice # sometimes this is not the case. # This issue seemd to be afflict Windows: # https://github.com/giampaolo/psutil/issues/392 # ...but it turns out also Linux (rarely) behaves the same. # last = psutil.cpu_times(percpu=True) # for x in range(100): # new = psutil.cpu_times(percpu=True) # for index in range(len(new)): # newcpu = new[index] # lastcpu = last[index] # for field in newcpu._fields: # new_t = getattr(newcpu, field) # last_t = getattr(lastcpu, field) # self.assertGreaterEqual( # new_t, last_t, msg="{} {}".format(lastcpu, newcpu)) # last = new def test_per_cpu_times_2(self): # Simulate some work load then make sure time have increased # between calls. tot1 = psutil.cpu_times(percpu=True) giveup_at = time.time() + GLOBAL_TIMEOUT while True: if time.time() >= giveup_at: return self.fail("timeout") tot2 = psutil.cpu_times(percpu=True) for t1, t2 in zip(tot1, tot2): t1, t2 = psutil._cpu_busy_time(t1), psutil._cpu_busy_time(t2) difference = t2 - t1 if difference >= 0.05: return None @pytest.mark.skipif( CI_TESTING and OPENBSD, reason="unreliable on OPENBSD + CI" ) @retry_on_failure(30) def test_cpu_times_comparison(self): # Make sure the sum of all per cpu times is almost equal to # base "one cpu" times. On OpenBSD the sum of per-CPUs is # higher for some reason. base = psutil.cpu_times() per_cpu = psutil.cpu_times(percpu=True) summed_values = base._make([sum(num) for num in zip(*per_cpu)]) for field in base._fields: with self.subTest(field=field, base=base, per_cpu=per_cpu): assert ( abs(getattr(base, field) - getattr(summed_values, field)) < 2 ) def _test_cpu_percent(self, percent, last_ret, new_ret): try: assert isinstance(percent, float) assert percent >= 0.0 assert percent <= 100.0 * psutil.cpu_count() except AssertionError as err: raise AssertionError( "\n{}\nlast={}\nnew={}".format( err, pprint.pformat(last_ret), pprint.pformat(new_ret) ) ) def test_cpu_percent(self): last = psutil.cpu_percent(interval=0.001) for _ in range(100): new = psutil.cpu_percent(interval=None) self._test_cpu_percent(new, last, new) last = new with pytest.raises(ValueError): psutil.cpu_percent(interval=-1) def test_per_cpu_percent(self): last = psutil.cpu_percent(interval=0.001, percpu=True) assert len(last) == psutil.cpu_count() for _ in range(100): new = psutil.cpu_percent(interval=None, percpu=True) for percent in new: self._test_cpu_percent(percent, last, new) last = new with pytest.raises(ValueError): psutil.cpu_percent(interval=-1, percpu=True) def test_cpu_times_percent(self): last = psutil.cpu_times_percent(interval=0.001) for _ in range(100): new = psutil.cpu_times_percent(interval=None) for percent in new: self._test_cpu_percent(percent, last, new) self._test_cpu_percent(sum(new), last, new) last = new with pytest.raises(ValueError): psutil.cpu_times_percent(interval=-1) def test_per_cpu_times_percent(self): last = psutil.cpu_times_percent(interval=0.001, percpu=True) assert len(last) == psutil.cpu_count() for _ in range(100): new = psutil.cpu_times_percent(interval=None, percpu=True) for cpu in new: for percent in cpu: self._test_cpu_percent(percent, last, new) self._test_cpu_percent(sum(cpu), last, new) last = new def test_per_cpu_times_percent_negative(self): # see: https://github.com/giampaolo/psutil/issues/645 psutil.cpu_times_percent(percpu=True) zero_times = [ x._make([0 for x in range(len(x._fields))]) for x in psutil.cpu_times(percpu=True) ] with mock.patch('psutil.cpu_times', return_value=zero_times): for cpu in psutil.cpu_times_percent(percpu=True): for percent in cpu: self._test_cpu_percent(percent, None, None) def test_cpu_stats(self): # Tested more extensively in per-platform test modules. infos = psutil.cpu_stats() assert infos._fields == ( 'ctx_switches', 'interrupts', 'soft_interrupts', 'syscalls', ) for name in infos._fields: value = getattr(infos, name) assert value >= 0 # on AIX, ctx_switches is always 0 if not AIX and name in {'ctx_switches', 'interrupts'}: assert value > 0 # TODO: remove this once 1892 is fixed @pytest.mark.skipif( MACOS and platform.machine() == 'arm64', reason="skipped due to #1892" ) @pytest.mark.skipif(not HAS_CPU_FREQ, reason="not supported") def test_cpu_freq(self): def check_ls(ls): for nt in ls: assert nt._fields == ('current', 'min', 'max') if nt.max != 0.0: assert nt.current <= nt.max for name in nt._fields: value = getattr(nt, name) assert isinstance(value, (int, float)) assert value >= 0 ls = psutil.cpu_freq(percpu=True) if (FREEBSD or AARCH64) and not ls: raise pytest.skip( "returns empty list on FreeBSD and Linux aarch64" ) assert ls, ls check_ls([psutil.cpu_freq(percpu=False)]) if LINUX: assert len(ls) == psutil.cpu_count() @pytest.mark.skipif(not HAS_GETLOADAVG, reason="not supported") def test_getloadavg(self): loadavg = psutil.getloadavg() assert len(loadavg) == 3 for load in loadavg: assert isinstance(load, float) assert load >= 0.0 class TestDiskAPIs(PsutilTestCase): @pytest.mark.skipif( PYPY and not IS_64BIT, reason="unreliable on PYPY32 + 32BIT" ) def test_disk_usage(self): usage = psutil.disk_usage(os.getcwd()) assert usage._fields == ('total', 'used', 'free', 'percent') assert usage.total > 0, usage assert usage.used > 0, usage assert usage.free > 0, usage assert usage.total > usage.used, usage assert usage.total > usage.free, usage assert 0 <= usage.percent <= 100, usage.percent if hasattr(shutil, 'disk_usage'): # py >= 3.3, see: http://bugs.python.org/issue12442 shutil_usage = shutil.disk_usage(os.getcwd()) tolerance = 5 * 1024 * 1024 # 5MB assert usage.total == shutil_usage.total assert abs(usage.free - shutil_usage.free) < tolerance if not MACOS_12PLUS: # see https://github.com/giampaolo/psutil/issues/2147 assert abs(usage.used - shutil_usage.used) < tolerance # if path does not exist OSError ENOENT is expected across # all platforms fname = self.get_testfn() with pytest.raises(FileNotFoundError): psutil.disk_usage(fname) @pytest.mark.skipif(not ASCII_FS, reason="not an ASCII fs") def test_disk_usage_unicode(self): # See: https://github.com/giampaolo/psutil/issues/416 with pytest.raises(UnicodeEncodeError): psutil.disk_usage(UNICODE_SUFFIX) def test_disk_usage_bytes(self): psutil.disk_usage(b'.') def test_disk_partitions(self): def check_ntuple(nt): assert isinstance(nt.device, str) assert isinstance(nt.mountpoint, str) assert isinstance(nt.fstype, str) assert isinstance(nt.opts, str) # all = False ls = psutil.disk_partitions(all=False) assert ls for disk in ls: check_ntuple(disk) if WINDOWS and 'cdrom' in disk.opts: continue if not POSIX: assert os.path.exists(disk.device), disk else: # we cannot make any assumption about this, see: # http://goo.gl/p9c43 disk.device # noqa: B018 # on modern systems mount points can also be files assert os.path.exists(disk.mountpoint), disk assert disk.fstype, disk # all = True ls = psutil.disk_partitions(all=True) assert ls for disk in psutil.disk_partitions(all=True): check_ntuple(disk) if not WINDOWS and disk.mountpoint: try: os.stat(disk.mountpoint) except OSError as err: if GITHUB_ACTIONS and MACOS and err.errno == errno.EIO: continue # http://mail.python.org/pipermail/python-dev/ # 2012-June/120787.html if err.errno not in {errno.EPERM, errno.EACCES}: raise else: assert os.path.exists(disk.mountpoint), disk # --- def find_mount_point(path): path = os.path.abspath(path) while not os.path.ismount(path): path = os.path.dirname(path) return path.lower() mount = find_mount_point(__file__) mounts = [ x.mountpoint.lower() for x in psutil.disk_partitions(all=True) if x.mountpoint ] assert mount in mounts @pytest.mark.skipif( LINUX and not os.path.exists('/proc/diskstats'), reason="/proc/diskstats not available on this linux version", ) @pytest.mark.skipif( CI_TESTING and not psutil.disk_io_counters(), reason="unreliable on CI" ) # no visible disks def test_disk_io_counters(self): def check_ntuple(nt): assert nt[0] == nt.read_count assert nt[1] == nt.write_count assert nt[2] == nt.read_bytes assert nt[3] == nt.write_bytes if not (OPENBSD or NETBSD): assert nt[4] == nt.read_time assert nt[5] == nt.write_time if LINUX: assert nt[6] == nt.read_merged_count assert nt[7] == nt.write_merged_count assert nt[8] == nt.busy_time elif FREEBSD: assert nt[6] == nt.busy_time for name in nt._fields: assert getattr(nt, name) >= 0, nt ret = psutil.disk_io_counters(perdisk=False) assert ret is not None, "no disks on this system?" check_ntuple(ret) ret = psutil.disk_io_counters(perdisk=True) # make sure there are no duplicates assert len(ret) == len(set(ret)) for key in ret: assert key, key check_ntuple(ret[key]) def test_disk_io_counters_no_disks(self): # Emulate a case where no disks are installed, see: # https://github.com/giampaolo/psutil/issues/1062 with mock.patch( 'psutil._psplatform.disk_io_counters', return_value={} ) as m: assert psutil.disk_io_counters(perdisk=False) is None assert psutil.disk_io_counters(perdisk=True) == {} assert m.called class TestNetAPIs(PsutilTestCase): @pytest.mark.skipif(not HAS_NET_IO_COUNTERS, reason="not supported") def test_net_io_counters(self): def check_ntuple(nt): assert nt[0] == nt.bytes_sent assert nt[1] == nt.bytes_recv assert nt[2] == nt.packets_sent assert nt[3] == nt.packets_recv assert nt[4] == nt.errin assert nt[5] == nt.errout assert nt[6] == nt.dropin assert nt[7] == nt.dropout assert nt.bytes_sent >= 0, nt assert nt.bytes_recv >= 0, nt assert nt.packets_sent >= 0, nt assert nt.packets_recv >= 0, nt assert nt.errin >= 0, nt assert nt.errout >= 0, nt assert nt.dropin >= 0, nt assert nt.dropout >= 0, nt ret = psutil.net_io_counters(pernic=False) check_ntuple(ret) ret = psutil.net_io_counters(pernic=True) assert ret != [] for key in ret: assert key assert isinstance(key, str) check_ntuple(ret[key]) @pytest.mark.skipif(not HAS_NET_IO_COUNTERS, reason="not supported") def test_net_io_counters_no_nics(self): # Emulate a case where no NICs are installed, see: # https://github.com/giampaolo/psutil/issues/1062 with mock.patch( 'psutil._psplatform.net_io_counters', return_value={} ) as m: assert psutil.net_io_counters(pernic=False) is None assert psutil.net_io_counters(pernic=True) == {} assert m.called def test_net_if_addrs(self): nics = psutil.net_if_addrs() assert nics, nics nic_stats = psutil.net_if_stats() # Not reliable on all platforms (net_if_addrs() reports more # interfaces). # self.assertEqual(sorted(nics.keys()), # sorted(psutil.net_io_counters(pernic=True).keys())) families = {socket.AF_INET, socket.AF_INET6, psutil.AF_LINK} for nic, addrs in nics.items(): assert isinstance(nic, str) assert len(set(addrs)) == len(addrs) for addr in addrs: assert isinstance(addr.family, int) assert isinstance(addr.address, str) assert isinstance(addr.netmask, (str, type(None))) assert isinstance(addr.broadcast, (str, type(None))) assert addr.family in families assert isinstance(addr.family, enum.IntEnum) if nic_stats[nic].isup: # Do not test binding to addresses of interfaces # that are down if addr.family == socket.AF_INET: with socket.socket(addr.family) as s: s.bind((addr.address, 0)) elif addr.family == socket.AF_INET6: info = socket.getaddrinfo( addr.address, 0, socket.AF_INET6, socket.SOCK_STREAM, 0, socket.AI_PASSIVE, )[0] af, socktype, proto, _canonname, sa = info with socket.socket(af, socktype, proto) as s: s.bind(sa) for ip in ( addr.address, addr.netmask, addr.broadcast, addr.ptp, ): if ip is not None: # TODO: skip AF_INET6 for now because I get: # AddressValueError: Only hex digits permitted in # u'c6f3%lxcbr0' in u'fe80::c8e0:fff:fe54:c6f3%lxcbr0' if addr.family != socket.AF_INET6: check_net_address(ip, addr.family) # broadcast and ptp addresses are mutually exclusive if addr.broadcast: assert addr.ptp is None elif addr.ptp: assert addr.broadcast is None # check broadcast address if ( addr.broadcast and addr.netmask and addr.family in {socket.AF_INET, socket.AF_INET6} ): assert addr.broadcast == broadcast_addr(addr) if BSD or MACOS or SUNOS: if hasattr(socket, "AF_LINK"): assert psutil.AF_LINK == socket.AF_LINK elif LINUX: assert psutil.AF_LINK == socket.AF_PACKET elif WINDOWS: assert psutil.AF_LINK == -1 def test_net_if_addrs_mac_null_bytes(self): # Simulate that the underlying C function returns an incomplete # MAC address. psutil is supposed to fill it with null bytes. # https://github.com/giampaolo/psutil/issues/786 if POSIX: ret = [('em1', psutil.AF_LINK, '06:3d:29', None, None, None)] else: ret = [('em1', -1, '06-3d-29', None, None, None)] with mock.patch( 'psutil._psplatform.net_if_addrs', return_value=ret ) as m: addr = psutil.net_if_addrs()['em1'][0] assert m.called if POSIX: assert addr.address == '06:3d:29:00:00:00' else: assert addr.address == '06-3d-29-00-00-00' def test_net_if_stats(self): nics = psutil.net_if_stats() assert nics, nics all_duplexes = ( psutil.NIC_DUPLEX_FULL, psutil.NIC_DUPLEX_HALF, psutil.NIC_DUPLEX_UNKNOWN, ) for name, stats in nics.items(): assert isinstance(name, str) isup, duplex, speed, mtu, flags = stats assert isinstance(isup, bool) assert duplex in all_duplexes assert duplex in all_duplexes assert speed >= 0 assert mtu >= 0 assert isinstance(flags, str) @pytest.mark.skipif( not (LINUX or BSD or MACOS), reason="LINUX or BSD or MACOS specific" ) def test_net_if_stats_enodev(self): # See: https://github.com/giampaolo/psutil/issues/1279 with mock.patch( 'psutil._psutil_posix.net_if_mtu', side_effect=OSError(errno.ENODEV, ""), ) as m: ret = psutil.net_if_stats() assert ret == {} assert m.called class TestSensorsAPIs(PsutilTestCase): @pytest.mark.skipif(not HAS_SENSORS_TEMPERATURES, reason="not supported") def test_sensors_temperatures(self): temps = psutil.sensors_temperatures() for name, entries in temps.items(): assert isinstance(name, str) for entry in entries: assert isinstance(entry.label, str) if entry.current is not None: assert entry.current >= 0 if entry.high is not None: assert entry.high >= 0 if entry.critical is not None: assert entry.critical >= 0 @pytest.mark.skipif(not HAS_SENSORS_TEMPERATURES, reason="not supported") def test_sensors_temperatures_fahreneit(self): d = {'coretemp': [('label', 50.0, 60.0, 70.0)]} with mock.patch( "psutil._psplatform.sensors_temperatures", return_value=d ) as m: temps = psutil.sensors_temperatures(fahrenheit=True)['coretemp'][0] assert m.called assert temps.current == 122.0 assert temps.high == 140.0 assert temps.critical == 158.0 @pytest.mark.skipif(not HAS_SENSORS_BATTERY, reason="not supported") @pytest.mark.skipif(not HAS_BATTERY, reason="no battery") def test_sensors_battery(self): ret = psutil.sensors_battery() assert ret.percent >= 0 assert ret.percent <= 100 if ret.secsleft not in { psutil.POWER_TIME_UNKNOWN, psutil.POWER_TIME_UNLIMITED, }: assert ret.secsleft >= 0 elif ret.secsleft == psutil.POWER_TIME_UNLIMITED: assert ret.power_plugged assert isinstance(ret.power_plugged, bool) @pytest.mark.skipif(not HAS_SENSORS_FANS, reason="not supported") def test_sensors_fans(self): fans = psutil.sensors_fans() for name, entries in fans.items(): assert isinstance(name, str) for entry in entries: assert isinstance(entry.label, str) assert isinstance(entry.current, int) assert entry.current >= 0 tests/test_aix.py 0000644 00000010470 15030211765 0010102 0 ustar 00 #!/usr/bin/env python3 # Copyright (c) 2009, Giampaolo Rodola' # Copyright (c) 2017, Arnon Yaari # All rights reserved. # Use of this source code is governed by a BSD-style license that can be # found in the LICENSE file. """AIX specific tests.""" import re import psutil from psutil import AIX from psutil.tests import PsutilTestCase from psutil.tests import pytest from psutil.tests import sh @pytest.mark.skipif(not AIX, reason="AIX only") class AIXSpecificTestCase(PsutilTestCase): def test_virtual_memory(self): out = sh('/usr/bin/svmon -O unit=KB') re_pattern = r"memory\s*" for field in [ "size", "inuse", "free", "pin", "virtual", "available", "mmode", ]: re_pattern += rf"(?P<{field}>\S+)\s+" matchobj = re.search(re_pattern, out) assert matchobj is not None KB = 1024 total = int(matchobj.group("size")) * KB available = int(matchobj.group("available")) * KB used = int(matchobj.group("inuse")) * KB free = int(matchobj.group("free")) * KB psutil_result = psutil.virtual_memory() # TOLERANCE_SYS_MEM from psutil.tests is not enough. For some reason # we're seeing differences of ~1.2 MB. 2 MB is still a good tolerance # when compared to GBs. TOLERANCE_SYS_MEM = 2 * KB * KB # 2 MB assert psutil_result.total == total assert abs(psutil_result.used - used) < TOLERANCE_SYS_MEM assert abs(psutil_result.available - available) < TOLERANCE_SYS_MEM assert abs(psutil_result.free - free) < TOLERANCE_SYS_MEM def test_swap_memory(self): out = sh('/usr/sbin/lsps -a') # From the man page, "The size is given in megabytes" so we assume # we'll always have 'MB' in the result # TODO maybe try to use "swap -l" to check "used" too, but its units # are not guaranteed to be "MB" so parsing may not be consistent matchobj = re.search( r"(?P<space>\S+)\s+" r"(?P<vol>\S+)\s+" r"(?P<vg>\S+)\s+" r"(?P<size>\d+)MB", out, ) assert matchobj is not None total_mb = int(matchobj.group("size")) MB = 1024**2 psutil_result = psutil.swap_memory() # we divide our result by MB instead of multiplying the lsps value by # MB because lsps may round down, so we round down too assert int(psutil_result.total / MB) == total_mb def test_cpu_stats(self): out = sh('/usr/bin/mpstat -a') re_pattern = r"ALL\s*" for field in [ "min", "maj", "mpcs", "mpcr", "dev", "soft", "dec", "ph", "cs", "ics", "bound", "rq", "push", "S3pull", "S3grd", "S0rd", "S1rd", "S2rd", "S3rd", "S4rd", "S5rd", "sysc", ]: re_pattern += rf"(?P<{field}>\S+)\s+" matchobj = re.search(re_pattern, out) assert matchobj is not None # numbers are usually in the millions so 1000 is ok for tolerance CPU_STATS_TOLERANCE = 1000 psutil_result = psutil.cpu_stats() assert ( abs(psutil_result.ctx_switches - int(matchobj.group("cs"))) < CPU_STATS_TOLERANCE ) assert ( abs(psutil_result.syscalls - int(matchobj.group("sysc"))) < CPU_STATS_TOLERANCE ) assert ( abs(psutil_result.interrupts - int(matchobj.group("dev"))) < CPU_STATS_TOLERANCE ) assert ( abs(psutil_result.soft_interrupts - int(matchobj.group("soft"))) < CPU_STATS_TOLERANCE ) def test_cpu_count_logical(self): out = sh('/usr/bin/mpstat -a') mpstat_lcpu = int(re.search(r"lcpu=(\d+)", out).group(1)) psutil_lcpu = psutil.cpu_count(logical=True) assert mpstat_lcpu == psutil_lcpu def test_net_if_addrs_names(self): out = sh('/etc/ifconfig -l') ifconfig_names = set(out.split()) psutil_names = set(psutil.net_if_addrs().keys()) assert ifconfig_names == psutil_names tests/test_contracts.py 0000644 00000027341 15030211765 0011326 0 ustar 00 #!/usr/bin/env python3 # Copyright (c) 2009, Giampaolo Rodola'. All rights reserved. # Use of this source code is governed by a BSD-style license that can be # found in the LICENSE file. """Contracts tests. These tests mainly check API sanity in terms of returned types and APIs availability. Some of these are duplicates of tests test_system.py and test_process.py. """ import platform import signal import psutil from psutil import AIX from psutil import FREEBSD from psutil import LINUX from psutil import MACOS from psutil import NETBSD from psutil import OPENBSD from psutil import POSIX from psutil import SUNOS from psutil import WINDOWS from psutil.tests import GITHUB_ACTIONS from psutil.tests import HAS_CPU_FREQ from psutil.tests import HAS_NET_IO_COUNTERS from psutil.tests import HAS_SENSORS_FANS from psutil.tests import HAS_SENSORS_TEMPERATURES from psutil.tests import SKIP_SYSCONS from psutil.tests import PsutilTestCase from psutil.tests import create_sockets from psutil.tests import enum from psutil.tests import is_namedtuple from psutil.tests import kernel_version from psutil.tests import pytest # =================================================================== # --- APIs availability # =================================================================== # Make sure code reflects what doc promises in terms of APIs # availability. class TestAvailConstantsAPIs(PsutilTestCase): def test_PROCFS_PATH(self): assert hasattr(psutil, "PROCFS_PATH") == (LINUX or SUNOS or AIX) def test_win_priority(self): ae = self.assertEqual ae(hasattr(psutil, "ABOVE_NORMAL_PRIORITY_CLASS"), WINDOWS) ae(hasattr(psutil, "BELOW_NORMAL_PRIORITY_CLASS"), WINDOWS) ae(hasattr(psutil, "HIGH_PRIORITY_CLASS"), WINDOWS) ae(hasattr(psutil, "IDLE_PRIORITY_CLASS"), WINDOWS) ae(hasattr(psutil, "NORMAL_PRIORITY_CLASS"), WINDOWS) ae(hasattr(psutil, "REALTIME_PRIORITY_CLASS"), WINDOWS) def test_linux_ioprio_linux(self): ae = self.assertEqual ae(hasattr(psutil, "IOPRIO_CLASS_NONE"), LINUX) ae(hasattr(psutil, "IOPRIO_CLASS_RT"), LINUX) ae(hasattr(psutil, "IOPRIO_CLASS_BE"), LINUX) ae(hasattr(psutil, "IOPRIO_CLASS_IDLE"), LINUX) def test_linux_ioprio_windows(self): ae = self.assertEqual ae(hasattr(psutil, "IOPRIO_HIGH"), WINDOWS) ae(hasattr(psutil, "IOPRIO_NORMAL"), WINDOWS) ae(hasattr(psutil, "IOPRIO_LOW"), WINDOWS) ae(hasattr(psutil, "IOPRIO_VERYLOW"), WINDOWS) @pytest.mark.skipif( GITHUB_ACTIONS and LINUX, reason="unsupported on GITHUB_ACTIONS + LINUX", ) def test_rlimit(self): ae = self.assertEqual ae(hasattr(psutil, "RLIM_INFINITY"), LINUX or FREEBSD) ae(hasattr(psutil, "RLIMIT_AS"), LINUX or FREEBSD) ae(hasattr(psutil, "RLIMIT_CORE"), LINUX or FREEBSD) ae(hasattr(psutil, "RLIMIT_CPU"), LINUX or FREEBSD) ae(hasattr(psutil, "RLIMIT_DATA"), LINUX or FREEBSD) ae(hasattr(psutil, "RLIMIT_FSIZE"), LINUX or FREEBSD) ae(hasattr(psutil, "RLIMIT_MEMLOCK"), LINUX or FREEBSD) ae(hasattr(psutil, "RLIMIT_NOFILE"), LINUX or FREEBSD) ae(hasattr(psutil, "RLIMIT_NPROC"), LINUX or FREEBSD) ae(hasattr(psutil, "RLIMIT_RSS"), LINUX or FREEBSD) ae(hasattr(psutil, "RLIMIT_STACK"), LINUX or FREEBSD) ae(hasattr(psutil, "RLIMIT_LOCKS"), LINUX) if POSIX: if kernel_version() >= (2, 6, 8): ae(hasattr(psutil, "RLIMIT_MSGQUEUE"), LINUX) if kernel_version() >= (2, 6, 12): ae(hasattr(psutil, "RLIMIT_NICE"), LINUX) if kernel_version() >= (2, 6, 12): ae(hasattr(psutil, "RLIMIT_RTPRIO"), LINUX) if kernel_version() >= (2, 6, 25): ae(hasattr(psutil, "RLIMIT_RTTIME"), LINUX) if kernel_version() >= (2, 6, 8): ae(hasattr(psutil, "RLIMIT_SIGPENDING"), LINUX) ae(hasattr(psutil, "RLIMIT_SWAP"), FREEBSD) ae(hasattr(psutil, "RLIMIT_SBSIZE"), FREEBSD) ae(hasattr(psutil, "RLIMIT_NPTS"), FREEBSD) class TestAvailSystemAPIs(PsutilTestCase): def test_win_service_iter(self): assert hasattr(psutil, "win_service_iter") == WINDOWS def test_win_service_get(self): assert hasattr(psutil, "win_service_get") == WINDOWS def test_cpu_freq(self): assert hasattr(psutil, "cpu_freq") == ( LINUX or MACOS or WINDOWS or FREEBSD or OPENBSD ) def test_sensors_temperatures(self): assert hasattr(psutil, "sensors_temperatures") == (LINUX or FREEBSD) def test_sensors_fans(self): assert hasattr(psutil, "sensors_fans") == LINUX def test_battery(self): assert hasattr(psutil, "sensors_battery") == ( LINUX or WINDOWS or FREEBSD or MACOS ) class TestAvailProcessAPIs(PsutilTestCase): def test_environ(self): assert hasattr(psutil.Process, "environ") == ( LINUX or MACOS or WINDOWS or AIX or SUNOS or FREEBSD or OPENBSD or NETBSD ) def test_uids(self): assert hasattr(psutil.Process, "uids") == POSIX def test_gids(self): assert hasattr(psutil.Process, "uids") == POSIX def test_terminal(self): assert hasattr(psutil.Process, "terminal") == POSIX def test_ionice(self): assert hasattr(psutil.Process, "ionice") == (LINUX or WINDOWS) @pytest.mark.skipif( GITHUB_ACTIONS and LINUX, reason="unsupported on GITHUB_ACTIONS + LINUX", ) def test_rlimit(self): assert hasattr(psutil.Process, "rlimit") == (LINUX or FREEBSD) def test_io_counters(self): hasit = hasattr(psutil.Process, "io_counters") assert hasit == (not (MACOS or SUNOS)) def test_num_fds(self): assert hasattr(psutil.Process, "num_fds") == POSIX def test_num_handles(self): assert hasattr(psutil.Process, "num_handles") == WINDOWS def test_cpu_affinity(self): assert hasattr(psutil.Process, "cpu_affinity") == ( LINUX or WINDOWS or FREEBSD ) def test_cpu_num(self): assert hasattr(psutil.Process, "cpu_num") == ( LINUX or FREEBSD or SUNOS ) def test_memory_maps(self): hasit = hasattr(psutil.Process, "memory_maps") assert hasit == (not (OPENBSD or NETBSD or AIX or MACOS)) # =================================================================== # --- API types # =================================================================== class TestSystemAPITypes(PsutilTestCase): """Check the return types of system related APIs. https://github.com/giampaolo/psutil/issues/1039. """ @classmethod def setUpClass(cls): cls.proc = psutil.Process() def assert_ntuple_of_nums(self, nt, type_=float, gezero=True): assert is_namedtuple(nt) for n in nt: assert isinstance(n, type_) if gezero: assert n >= 0 def test_cpu_times(self): self.assert_ntuple_of_nums(psutil.cpu_times()) for nt in psutil.cpu_times(percpu=True): self.assert_ntuple_of_nums(nt) def test_cpu_percent(self): assert isinstance(psutil.cpu_percent(interval=None), float) assert isinstance(psutil.cpu_percent(interval=0.00001), float) def test_cpu_times_percent(self): self.assert_ntuple_of_nums(psutil.cpu_times_percent(interval=None)) self.assert_ntuple_of_nums(psutil.cpu_times_percent(interval=0.0001)) def test_cpu_count(self): assert isinstance(psutil.cpu_count(), int) # TODO: remove this once 1892 is fixed @pytest.mark.skipif( MACOS and platform.machine() == 'arm64', reason="skipped due to #1892" ) @pytest.mark.skipif(not HAS_CPU_FREQ, reason="not supported") def test_cpu_freq(self): if psutil.cpu_freq() is None: raise pytest.skip("cpu_freq() returns None") self.assert_ntuple_of_nums(psutil.cpu_freq(), type_=(float, int)) def test_disk_io_counters(self): # Duplicate of test_system.py. Keep it anyway. for k, v in psutil.disk_io_counters(perdisk=True).items(): assert isinstance(k, str) self.assert_ntuple_of_nums(v, type_=int) def test_disk_partitions(self): # Duplicate of test_system.py. Keep it anyway. for disk in psutil.disk_partitions(): assert isinstance(disk.device, str) assert isinstance(disk.mountpoint, str) assert isinstance(disk.fstype, str) assert isinstance(disk.opts, str) @pytest.mark.skipif(SKIP_SYSCONS, reason="requires root") def test_net_connections(self): with create_sockets(): ret = psutil.net_connections('all') assert len(ret) == len(set(ret)) for conn in ret: assert is_namedtuple(conn) def test_net_if_addrs(self): # Duplicate of test_system.py. Keep it anyway. for ifname, addrs in psutil.net_if_addrs().items(): assert isinstance(ifname, str) for addr in addrs: assert isinstance(addr.family, enum.IntEnum) assert isinstance(addr.address, str) assert isinstance(addr.netmask, (str, type(None))) assert isinstance(addr.broadcast, (str, type(None))) def test_net_if_stats(self): # Duplicate of test_system.py. Keep it anyway. for ifname, info in psutil.net_if_stats().items(): assert isinstance(ifname, str) assert isinstance(info.isup, bool) assert isinstance(info.duplex, enum.IntEnum) assert isinstance(info.speed, int) assert isinstance(info.mtu, int) @pytest.mark.skipif(not HAS_NET_IO_COUNTERS, reason="not supported") def test_net_io_counters(self): # Duplicate of test_system.py. Keep it anyway. for ifname in psutil.net_io_counters(pernic=True): assert isinstance(ifname, str) @pytest.mark.skipif(not HAS_SENSORS_FANS, reason="not supported") def test_sensors_fans(self): # Duplicate of test_system.py. Keep it anyway. for name, units in psutil.sensors_fans().items(): assert isinstance(name, str) for unit in units: assert isinstance(unit.label, str) assert isinstance(unit.current, (float, int, type(None))) @pytest.mark.skipif(not HAS_SENSORS_TEMPERATURES, reason="not supported") def test_sensors_temperatures(self): # Duplicate of test_system.py. Keep it anyway. for name, units in psutil.sensors_temperatures().items(): assert isinstance(name, str) for unit in units: assert isinstance(unit.label, str) assert isinstance(unit.current, (float, int, type(None))) assert isinstance(unit.high, (float, int, type(None))) assert isinstance(unit.critical, (float, int, type(None))) def test_boot_time(self): # Duplicate of test_system.py. Keep it anyway. assert isinstance(psutil.boot_time(), float) def test_users(self): # Duplicate of test_system.py. Keep it anyway. for user in psutil.users(): assert isinstance(user.name, str) assert isinstance(user.terminal, (str, type(None))) assert isinstance(user.host, (str, type(None))) assert isinstance(user.pid, (int, type(None))) class TestProcessWaitType(PsutilTestCase): @pytest.mark.skipif(not POSIX, reason="not POSIX") def test_negative_signal(self): p = psutil.Process(self.spawn_testproc().pid) p.terminate() code = p.wait() assert code == -signal.SIGTERM assert isinstance(code, enum.IntEnum) tests/test_windows.py 0000644 00000100676 15030211765 0011023 0 ustar 00 #!/usr/bin/env python3 # Copyright (c) 2009, Giampaolo Rodola'. All rights reserved. # Use of this source code is governed by a BSD-style license that can be # found in the LICENSE file. """Windows specific tests.""" import datetime import glob import os import platform import re import shutil import signal import subprocess import sys import time import warnings from unittest import mock import psutil from psutil import WINDOWS from psutil.tests import GITHUB_ACTIONS from psutil.tests import HAS_BATTERY from psutil.tests import IS_64BIT from psutil.tests import PYPY from psutil.tests import TOLERANCE_DISK_USAGE from psutil.tests import TOLERANCE_SYS_MEM from psutil.tests import PsutilTestCase from psutil.tests import pytest from psutil.tests import retry_on_failure from psutil.tests import sh from psutil.tests import spawn_testproc from psutil.tests import terminate if WINDOWS and not PYPY: with warnings.catch_warnings(): warnings.simplefilter("ignore") import win32api # requires "pip install pywin32" import win32con import win32process import wmi # requires "pip install wmi" / "make install-pydeps-test" if WINDOWS: from psutil._pswindows import convert_oserror cext = psutil._psplatform.cext @pytest.mark.skipif(not WINDOWS, reason="WINDOWS only") @pytest.mark.skipif(PYPY, reason="pywin32 not available on PYPY") class WindowsTestCase(PsutilTestCase): pass def powershell(cmd): """Currently not used, but available just in case. Usage: >>> powershell( "Get-CIMInstance Win32_PageFileUsage | Select AllocatedBaseSize") """ if not shutil.which("powershell.exe"): raise pytest.skip("powershell.exe not available") cmdline = ( "powershell.exe -ExecutionPolicy Bypass -NoLogo -NonInteractive " f"-NoProfile -WindowStyle Hidden -Command \"{cmd}\"" # noqa: Q003 ) return sh(cmdline) def wmic(path, what, converter=int): """Currently not used, but available just in case. Usage: >>> wmic("Win32_OperatingSystem", "FreePhysicalMemory") 2134124534 """ out = sh(f"wmic path {path} get {what}").strip() data = "".join(out.splitlines()[1:]).strip() # get rid of the header if converter is not None: if "," in what: return tuple(converter(x) for x in data.split()) else: return converter(data) else: return data # =================================================================== # System APIs # =================================================================== class TestCpuAPIs(WindowsTestCase): @pytest.mark.skipif( 'NUMBER_OF_PROCESSORS' not in os.environ, reason="NUMBER_OF_PROCESSORS env var is not available", ) def test_cpu_count_vs_NUMBER_OF_PROCESSORS(self): # Will likely fail on many-cores systems: # https://stackoverflow.com/questions/31209256 num_cpus = int(os.environ['NUMBER_OF_PROCESSORS']) assert num_cpus == psutil.cpu_count() def test_cpu_count_vs_GetSystemInfo(self): # Will likely fail on many-cores systems: # https://stackoverflow.com/questions/31209256 sys_value = win32api.GetSystemInfo()[5] psutil_value = psutil.cpu_count() assert sys_value == psutil_value def test_cpu_count_logical_vs_wmi(self): w = wmi.WMI() procs = sum( proc.NumberOfLogicalProcessors for proc in w.Win32_Processor() ) assert psutil.cpu_count() == procs def test_cpu_count_cores_vs_wmi(self): w = wmi.WMI() cores = sum(proc.NumberOfCores for proc in w.Win32_Processor()) assert psutil.cpu_count(logical=False) == cores def test_cpu_count_vs_cpu_times(self): assert psutil.cpu_count() == len(psutil.cpu_times(percpu=True)) def test_cpu_freq(self): w = wmi.WMI() proc = w.Win32_Processor()[0] assert proc.CurrentClockSpeed == psutil.cpu_freq().current assert proc.MaxClockSpeed == psutil.cpu_freq().max class TestSystemAPIs(WindowsTestCase): def test_nic_names(self): out = sh('ipconfig /all') nics = psutil.net_io_counters(pernic=True).keys() for nic in nics: if "pseudo-interface" in nic.replace(' ', '-').lower(): continue if nic not in out: raise self.fail( f"{nic!r} nic wasn't found in 'ipconfig /all' output" ) def test_total_phymem(self): w = wmi.WMI().Win32_ComputerSystem()[0] assert int(w.TotalPhysicalMemory) == psutil.virtual_memory().total def test_free_phymem(self): w = wmi.WMI().Win32_PerfRawData_PerfOS_Memory()[0] assert ( abs(int(w.AvailableBytes) - psutil.virtual_memory().free) < TOLERANCE_SYS_MEM ) def test_total_swapmem(self): w = wmi.WMI().Win32_PerfRawData_PerfOS_Memory()[0] assert ( int(w.CommitLimit) - psutil.virtual_memory().total == psutil.swap_memory().total ) if psutil.swap_memory().total == 0: assert psutil.swap_memory().free == 0 assert psutil.swap_memory().used == 0 def test_percent_swapmem(self): if psutil.swap_memory().total > 0: w = wmi.WMI().Win32_PerfRawData_PerfOS_PagingFile(Name="_Total")[0] # calculate swap usage to percent percentSwap = int(w.PercentUsage) * 100 / int(w.PercentUsage_Base) # exact percent may change but should be reasonable # assert within +/- 5% and between 0 and 100% assert psutil.swap_memory().percent >= 0 assert abs(psutil.swap_memory().percent - percentSwap) < 5 assert psutil.swap_memory().percent <= 100 # @pytest.mark.skipif(wmi is None, reason="wmi module is not installed") # def test__UPTIME(self): # # _UPTIME constant is not public but it is used internally # # as value to return for pid 0 creation time. # # WMI behaves the same. # w = wmi.WMI().Win32_Process(ProcessId=self.pid)[0] # p = psutil.Process(0) # wmic_create = str(w.CreationDate.split('.')[0]) # psutil_create = time.strftime("%Y%m%d%H%M%S", # time.localtime(p.create_time())) # Note: this test is not very reliable @retry_on_failure() def test_pids(self): # Note: this test might fail if the OS is starting/killing # other processes in the meantime w = wmi.WMI().Win32_Process() wmi_pids = {x.ProcessId for x in w} psutil_pids = set(psutil.pids()) assert wmi_pids == psutil_pids @retry_on_failure() def test_disks(self): ps_parts = psutil.disk_partitions(all=True) wmi_parts = wmi.WMI().Win32_LogicalDisk() for ps_part in ps_parts: for wmi_part in wmi_parts: if ps_part.device.replace('\\', '') == wmi_part.DeviceID: if not ps_part.mountpoint: # this is usually a CD-ROM with no disk inserted break if 'cdrom' in ps_part.opts: break if ps_part.mountpoint.startswith('A:'): break # floppy try: usage = psutil.disk_usage(ps_part.mountpoint) except FileNotFoundError: # usually this is the floppy break assert usage.total == int(wmi_part.Size) wmi_free = int(wmi_part.FreeSpace) assert usage.free == wmi_free # 10 MB tolerance if abs(usage.free - wmi_free) > 10 * 1024 * 1024: raise self.fail(f"psutil={usage.free}, wmi={wmi_free}") break else: raise self.fail(f"can't find partition {ps_part!r}") @retry_on_failure() def test_disk_usage(self): for disk in psutil.disk_partitions(): if 'cdrom' in disk.opts: continue sys_value = win32api.GetDiskFreeSpaceEx(disk.mountpoint) psutil_value = psutil.disk_usage(disk.mountpoint) assert abs(sys_value[0] - psutil_value.free) < TOLERANCE_DISK_USAGE assert ( abs(sys_value[1] - psutil_value.total) < TOLERANCE_DISK_USAGE ) assert psutil_value.used == psutil_value.total - psutil_value.free def test_disk_partitions(self): sys_value = [ x + '\\' for x in win32api.GetLogicalDriveStrings().split("\\\x00") if x and not x.startswith('A:') ] psutil_value = [ x.mountpoint for x in psutil.disk_partitions(all=True) if not x.mountpoint.startswith('A:') ] assert sys_value == psutil_value def test_net_if_stats(self): ps_names = set(cext.net_if_stats()) wmi_adapters = wmi.WMI().Win32_NetworkAdapter() wmi_names = set() for wmi_adapter in wmi_adapters: wmi_names.add(wmi_adapter.Name) wmi_names.add(wmi_adapter.NetConnectionID) assert ( ps_names & wmi_names ), f"no common entries in {ps_names}, {wmi_names}" def test_boot_time(self): wmi_os = wmi.WMI().Win32_OperatingSystem() wmi_btime_str = wmi_os[0].LastBootUpTime.split('.')[0] wmi_btime_dt = datetime.datetime.strptime( wmi_btime_str, "%Y%m%d%H%M%S" ) psutil_dt = datetime.datetime.fromtimestamp(psutil.boot_time()) diff = abs((wmi_btime_dt - psutil_dt).total_seconds()) assert diff <= 5 def test_boot_time_fluctuation(self): # https://github.com/giampaolo/psutil/issues/1007 with mock.patch('psutil._pswindows.cext.boot_time', return_value=5): assert psutil.boot_time() == 5 with mock.patch('psutil._pswindows.cext.boot_time', return_value=4): assert psutil.boot_time() == 5 with mock.patch('psutil._pswindows.cext.boot_time', return_value=6): assert psutil.boot_time() == 5 with mock.patch('psutil._pswindows.cext.boot_time', return_value=333): assert psutil.boot_time() == 333 # =================================================================== # sensors_battery() # =================================================================== class TestSensorsBattery(WindowsTestCase): def test_has_battery(self): if win32api.GetPwrCapabilities()['SystemBatteriesPresent']: assert psutil.sensors_battery() is not None else: assert psutil.sensors_battery() is None @pytest.mark.skipif(not HAS_BATTERY, reason="no battery") def test_percent(self): w = wmi.WMI() battery_wmi = w.query('select * from Win32_Battery')[0] battery_psutil = psutil.sensors_battery() assert ( abs(battery_psutil.percent - battery_wmi.EstimatedChargeRemaining) < 1 ) @pytest.mark.skipif(not HAS_BATTERY, reason="no battery") def test_power_plugged(self): w = wmi.WMI() battery_wmi = w.query('select * from Win32_Battery')[0] battery_psutil = psutil.sensors_battery() # Status codes: # https://msdn.microsoft.com/en-us/library/aa394074(v=vs.85).aspx assert battery_psutil.power_plugged == (battery_wmi.BatteryStatus == 2) def test_emulate_no_battery(self): with mock.patch( "psutil._pswindows.cext.sensors_battery", return_value=(0, 128, 0, 0), ) as m: assert psutil.sensors_battery() is None assert m.called def test_emulate_power_connected(self): with mock.patch( "psutil._pswindows.cext.sensors_battery", return_value=(1, 0, 0, 0) ) as m: assert ( psutil.sensors_battery().secsleft == psutil.POWER_TIME_UNLIMITED ) assert m.called def test_emulate_power_charging(self): with mock.patch( "psutil._pswindows.cext.sensors_battery", return_value=(0, 8, 0, 0) ) as m: assert ( psutil.sensors_battery().secsleft == psutil.POWER_TIME_UNLIMITED ) assert m.called def test_emulate_secs_left_unknown(self): with mock.patch( "psutil._pswindows.cext.sensors_battery", return_value=(0, 0, 0, -1), ) as m: assert ( psutil.sensors_battery().secsleft == psutil.POWER_TIME_UNKNOWN ) assert m.called # =================================================================== # Process APIs # =================================================================== class TestProcess(WindowsTestCase): @classmethod def setUpClass(cls): cls.pid = spawn_testproc().pid @classmethod def tearDownClass(cls): terminate(cls.pid) def test_issue_24(self): p = psutil.Process(0) with pytest.raises(psutil.AccessDenied): p.kill() def test_special_pid(self): p = psutil.Process(4) assert p.name() == 'System' # use __str__ to access all common Process properties to check # that nothing strange happens str(p) p.username() assert p.create_time() >= 0.0 try: rss, _vms = p.memory_info()[:2] except psutil.AccessDenied: # expected on Windows Vista and Windows 7 if platform.uname()[1] not in {'vista', 'win-7', 'win7'}: raise else: assert rss > 0 def test_send_signal(self): p = psutil.Process(self.pid) with pytest.raises(ValueError): p.send_signal(signal.SIGINT) def test_num_handles_increment(self): p = psutil.Process(os.getpid()) before = p.num_handles() handle = win32api.OpenProcess( win32con.PROCESS_QUERY_INFORMATION, win32con.FALSE, os.getpid() ) after = p.num_handles() assert after == before + 1 win32api.CloseHandle(handle) assert p.num_handles() == before def test_ctrl_signals(self): p = psutil.Process(self.spawn_testproc().pid) p.send_signal(signal.CTRL_C_EVENT) p.send_signal(signal.CTRL_BREAK_EVENT) p.kill() p.wait() with pytest.raises(psutil.NoSuchProcess): p.send_signal(signal.CTRL_C_EVENT) with pytest.raises(psutil.NoSuchProcess): p.send_signal(signal.CTRL_BREAK_EVENT) def test_username(self): name = win32api.GetUserNameEx(win32con.NameSamCompatible) if name.endswith('$'): # When running as a service account (most likely to be # NetworkService), these user name calculations don't produce the # same result, causing the test to fail. raise pytest.skip('running as service account') assert psutil.Process().username() == name def test_cmdline(self): sys_value = re.sub(r"[ ]+", " ", win32api.GetCommandLine()).strip() psutil_value = ' '.join(psutil.Process().cmdline()) if sys_value[0] == '"' != psutil_value[0]: # The PyWin32 command line may retain quotes around argv[0] if they # were used unnecessarily, while psutil will omit them. So remove # the first 2 quotes from sys_value if not in psutil_value. # A path to an executable will not contain quotes, so this is safe. sys_value = sys_value.replace('"', '', 2) assert sys_value == psutil_value # XXX - occasional failures # def test_cpu_times(self): # handle = win32api.OpenProcess(win32con.PROCESS_QUERY_INFORMATION, # win32con.FALSE, os.getpid()) # self.addCleanup(win32api.CloseHandle, handle) # sys_value = win32process.GetProcessTimes(handle) # psutil_value = psutil.Process().cpu_times() # self.assertAlmostEqual( # psutil_value.user, sys_value['UserTime'] / 10000000.0, # delta=0.2) # self.assertAlmostEqual( # psutil_value.user, sys_value['KernelTime'] / 10000000.0, # delta=0.2) def test_nice(self): handle = win32api.OpenProcess( win32con.PROCESS_QUERY_INFORMATION, win32con.FALSE, os.getpid() ) self.addCleanup(win32api.CloseHandle, handle) sys_value = win32process.GetPriorityClass(handle) psutil_value = psutil.Process().nice() assert psutil_value == sys_value def test_memory_info(self): handle = win32api.OpenProcess( win32con.PROCESS_QUERY_INFORMATION, win32con.FALSE, self.pid ) self.addCleanup(win32api.CloseHandle, handle) sys_value = win32process.GetProcessMemoryInfo(handle) psutil_value = psutil.Process(self.pid).memory_info() assert sys_value['PeakWorkingSetSize'] == psutil_value.peak_wset assert sys_value['WorkingSetSize'] == psutil_value.wset assert ( sys_value['QuotaPeakPagedPoolUsage'] == psutil_value.peak_paged_pool ) assert sys_value['QuotaPagedPoolUsage'] == psutil_value.paged_pool assert ( sys_value['QuotaPeakNonPagedPoolUsage'] == psutil_value.peak_nonpaged_pool ) assert ( sys_value['QuotaNonPagedPoolUsage'] == psutil_value.nonpaged_pool ) assert sys_value['PagefileUsage'] == psutil_value.pagefile assert sys_value['PeakPagefileUsage'] == psutil_value.peak_pagefile assert psutil_value.rss == psutil_value.wset assert psutil_value.vms == psutil_value.pagefile def test_wait(self): handle = win32api.OpenProcess( win32con.PROCESS_QUERY_INFORMATION, win32con.FALSE, self.pid ) self.addCleanup(win32api.CloseHandle, handle) p = psutil.Process(self.pid) p.terminate() psutil_value = p.wait() sys_value = win32process.GetExitCodeProcess(handle) assert psutil_value == sys_value def test_cpu_affinity(self): def from_bitmask(x): return [i for i in range(64) if (1 << i) & x] handle = win32api.OpenProcess( win32con.PROCESS_QUERY_INFORMATION, win32con.FALSE, self.pid ) self.addCleanup(win32api.CloseHandle, handle) sys_value = from_bitmask( win32process.GetProcessAffinityMask(handle)[0] ) psutil_value = psutil.Process(self.pid).cpu_affinity() assert psutil_value == sys_value def test_io_counters(self): handle = win32api.OpenProcess( win32con.PROCESS_QUERY_INFORMATION, win32con.FALSE, os.getpid() ) self.addCleanup(win32api.CloseHandle, handle) sys_value = win32process.GetProcessIoCounters(handle) psutil_value = psutil.Process().io_counters() assert psutil_value.read_count == sys_value['ReadOperationCount'] assert psutil_value.write_count == sys_value['WriteOperationCount'] assert psutil_value.read_bytes == sys_value['ReadTransferCount'] assert psutil_value.write_bytes == sys_value['WriteTransferCount'] assert psutil_value.other_count == sys_value['OtherOperationCount'] assert psutil_value.other_bytes == sys_value['OtherTransferCount'] def test_num_handles(self): import ctypes import ctypes.wintypes PROCESS_QUERY_INFORMATION = 0x400 handle = ctypes.windll.kernel32.OpenProcess( PROCESS_QUERY_INFORMATION, 0, self.pid ) self.addCleanup(ctypes.windll.kernel32.CloseHandle, handle) hndcnt = ctypes.wintypes.DWORD() ctypes.windll.kernel32.GetProcessHandleCount( handle, ctypes.byref(hndcnt) ) sys_value = hndcnt.value psutil_value = psutil.Process(self.pid).num_handles() assert psutil_value == sys_value def test_error_partial_copy(self): # https://github.com/giampaolo/psutil/issues/875 exc = OSError() exc.winerror = 299 with mock.patch("psutil._psplatform.cext.proc_cwd", side_effect=exc): with mock.patch("time.sleep") as m: p = psutil.Process() with pytest.raises(psutil.AccessDenied): p.cwd() assert m.call_count >= 5 def test_exe(self): # NtQuerySystemInformation succeeds if process is gone. Make sure # it raises NSP for a non existent pid. pid = psutil.pids()[-1] + 99999 proc = psutil._psplatform.Process(pid) with pytest.raises(psutil.NoSuchProcess): proc.exe() class TestProcessWMI(WindowsTestCase): """Compare Process API results with WMI.""" @classmethod def setUpClass(cls): cls.pid = spawn_testproc().pid @classmethod def tearDownClass(cls): terminate(cls.pid) def test_name(self): w = wmi.WMI().Win32_Process(ProcessId=self.pid)[0] p = psutil.Process(self.pid) assert p.name() == w.Caption # This fail on github because using virtualenv for test environment @pytest.mark.skipif( GITHUB_ACTIONS, reason="unreliable path on GITHUB_ACTIONS" ) def test_exe(self): w = wmi.WMI().Win32_Process(ProcessId=self.pid)[0] p = psutil.Process(self.pid) # Note: wmi reports the exe as a lower case string. # Being Windows paths case-insensitive we ignore that. assert p.exe().lower() == w.ExecutablePath.lower() def test_cmdline(self): w = wmi.WMI().Win32_Process(ProcessId=self.pid)[0] p = psutil.Process(self.pid) assert ' '.join(p.cmdline()) == w.CommandLine.replace('"', '') def test_username(self): w = wmi.WMI().Win32_Process(ProcessId=self.pid)[0] p = psutil.Process(self.pid) domain, _, username = w.GetOwner() username = f"{domain}\\{username}" assert p.username() == username @retry_on_failure() def test_memory_rss(self): w = wmi.WMI().Win32_Process(ProcessId=self.pid)[0] p = psutil.Process(self.pid) rss = p.memory_info().rss assert rss == int(w.WorkingSetSize) @retry_on_failure() def test_memory_vms(self): w = wmi.WMI().Win32_Process(ProcessId=self.pid)[0] p = psutil.Process(self.pid) vms = p.memory_info().vms # http://msdn.microsoft.com/en-us/library/aa394372(VS.85).aspx # ...claims that PageFileUsage is represented in Kilo # bytes but funnily enough on certain platforms bytes are # returned instead. wmi_usage = int(w.PageFileUsage) if vms not in {wmi_usage, wmi_usage * 1024}: raise self.fail(f"wmi={wmi_usage}, psutil={vms}") def test_create_time(self): w = wmi.WMI().Win32_Process(ProcessId=self.pid)[0] p = psutil.Process(self.pid) wmic_create = str(w.CreationDate.split('.')[0]) psutil_create = time.strftime( "%Y%m%d%H%M%S", time.localtime(p.create_time()) ) assert wmic_create == psutil_create # --- @pytest.mark.skipif(not WINDOWS, reason="WINDOWS only") class TestDualProcessImplementation(PsutilTestCase): """Certain APIs on Windows have 2 internal implementations, one based on documented Windows APIs, another one based NtQuerySystemInformation() which gets called as fallback in case the first fails because of limited permission error. Here we test that the two methods return the exact same value, see: https://github.com/giampaolo/psutil/issues/304. """ @classmethod def setUpClass(cls): cls.pid = spawn_testproc().pid @classmethod def tearDownClass(cls): terminate(cls.pid) def test_memory_info(self): mem_1 = psutil.Process(self.pid).memory_info() with mock.patch( "psutil._psplatform.cext.proc_memory_info", side_effect=PermissionError, ) as fun: mem_2 = psutil.Process(self.pid).memory_info() assert len(mem_1) == len(mem_2) for i in range(len(mem_1)): assert mem_1[i] >= 0 assert mem_2[i] >= 0 assert abs(mem_1[i] - mem_2[i]) < 512 assert fun.called def test_create_time(self): ctime = psutil.Process(self.pid).create_time() with mock.patch( "psutil._psplatform.cext.proc_times", side_effect=PermissionError, ) as fun: assert psutil.Process(self.pid).create_time() == ctime assert fun.called def test_cpu_times(self): cpu_times_1 = psutil.Process(self.pid).cpu_times() with mock.patch( "psutil._psplatform.cext.proc_times", side_effect=PermissionError, ) as fun: cpu_times_2 = psutil.Process(self.pid).cpu_times() assert fun.called assert abs(cpu_times_1.user - cpu_times_2.user) < 0.01 assert abs(cpu_times_1.system - cpu_times_2.system) < 0.01 def test_io_counters(self): io_counters_1 = psutil.Process(self.pid).io_counters() with mock.patch( "psutil._psplatform.cext.proc_io_counters", side_effect=PermissionError, ) as fun: io_counters_2 = psutil.Process(self.pid).io_counters() for i in range(len(io_counters_1)): assert abs(io_counters_1[i] - io_counters_2[i]) < 5 assert fun.called def test_num_handles(self): num_handles = psutil.Process(self.pid).num_handles() with mock.patch( "psutil._psplatform.cext.proc_num_handles", side_effect=PermissionError, ) as fun: assert psutil.Process(self.pid).num_handles() == num_handles assert fun.called def test_cmdline(self): for pid in psutil.pids(): try: a = cext.proc_cmdline(pid, use_peb=True) b = cext.proc_cmdline(pid, use_peb=False) except OSError as err: err = convert_oserror(err) if not isinstance( err, (psutil.AccessDenied, psutil.NoSuchProcess) ): raise else: assert a == b @pytest.mark.skipif(not WINDOWS, reason="WINDOWS only") class RemoteProcessTestCase(PsutilTestCase): """Certain functions require calling ReadProcessMemory. This trivially works when called on the current process. Check that this works on other processes, especially when they have a different bitness. """ @staticmethod def find_other_interpreter(): # find a python interpreter that is of the opposite bitness from us code = "import sys; sys.stdout.write(str(sys.maxsize > 2**32))" # XXX: a different and probably more stable approach might be to access # the registry but accessing 64 bit paths from a 32 bit process for filename in glob.glob(r"C:\Python*\python.exe"): proc = subprocess.Popen( args=[filename, "-c", code], stdout=subprocess.PIPE, stderr=subprocess.STDOUT, ) output, _ = proc.communicate() proc.wait() if output == str(not IS_64BIT): return filename test_args = ["-c", "import sys; sys.stdin.read()"] def setUp(self): super().setUp() other_python = self.find_other_interpreter() if other_python is None: raise pytest.skip( "could not find interpreter with opposite bitness" ) if IS_64BIT: self.python64 = sys.executable self.python32 = other_python else: self.python64 = other_python self.python32 = sys.executable env = os.environ.copy() env["THINK_OF_A_NUMBER"] = str(os.getpid()) self.proc32 = self.spawn_testproc( [self.python32] + self.test_args, env=env, stdin=subprocess.PIPE ) self.proc64 = self.spawn_testproc( [self.python64] + self.test_args, env=env, stdin=subprocess.PIPE ) def tearDown(self): super().tearDown() self.proc32.communicate() self.proc64.communicate() def test_cmdline_32(self): p = psutil.Process(self.proc32.pid) assert len(p.cmdline()) == 3 assert p.cmdline()[1:] == self.test_args def test_cmdline_64(self): p = psutil.Process(self.proc64.pid) assert len(p.cmdline()) == 3 assert p.cmdline()[1:] == self.test_args def test_cwd_32(self): p = psutil.Process(self.proc32.pid) assert p.cwd() == os.getcwd() def test_cwd_64(self): p = psutil.Process(self.proc64.pid) assert p.cwd() == os.getcwd() def test_environ_32(self): p = psutil.Process(self.proc32.pid) e = p.environ() assert "THINK_OF_A_NUMBER" in e assert e["THINK_OF_A_NUMBER"] == str(os.getpid()) def test_environ_64(self): p = psutil.Process(self.proc64.pid) try: p.environ() except psutil.AccessDenied: pass # =================================================================== # Windows services # =================================================================== @pytest.mark.skipif(not WINDOWS, reason="WINDOWS only") class TestServices(PsutilTestCase): def test_win_service_iter(self): valid_statuses = { "running", "paused", "start", "pause", "continue", "stop", "stopped", } valid_start_types = {"automatic", "manual", "disabled"} valid_statuses = { "running", "paused", "start_pending", "pause_pending", "continue_pending", "stop_pending", "stopped", } for serv in psutil.win_service_iter(): data = serv.as_dict() assert isinstance(data['name'], str) assert data['name'].strip() assert isinstance(data['display_name'], str) assert isinstance(data['username'], str) assert data['status'] in valid_statuses if data['pid'] is not None: psutil.Process(data['pid']) assert isinstance(data['binpath'], str) assert isinstance(data['username'], str) assert isinstance(data['start_type'], str) assert data['start_type'] in valid_start_types assert data['status'] in valid_statuses assert isinstance(data['description'], str) pid = serv.pid() if pid is not None: p = psutil.Process(pid) assert p.is_running() # win_service_get s = psutil.win_service_get(serv.name()) # test __eq__ assert serv == s def test_win_service_get(self): ERROR_SERVICE_DOES_NOT_EXIST = ( psutil._psplatform.cext.ERROR_SERVICE_DOES_NOT_EXIST ) ERROR_ACCESS_DENIED = psutil._psplatform.cext.ERROR_ACCESS_DENIED name = next(psutil.win_service_iter()).name() with pytest.raises(psutil.NoSuchProcess) as cm: psutil.win_service_get(name + '???') assert cm.value.name == name + '???' # test NoSuchProcess service = psutil.win_service_get(name) exc = OSError(0, "msg", 0) exc.winerror = ERROR_SERVICE_DOES_NOT_EXIST with mock.patch( "psutil._psplatform.cext.winservice_query_status", side_effect=exc ): with pytest.raises(psutil.NoSuchProcess): service.status() with mock.patch( "psutil._psplatform.cext.winservice_query_config", side_effect=exc ): with pytest.raises(psutil.NoSuchProcess): service.username() # test AccessDenied exc = OSError(0, "msg", 0) exc.winerror = ERROR_ACCESS_DENIED with mock.patch( "psutil._psplatform.cext.winservice_query_status", side_effect=exc ): with pytest.raises(psutil.AccessDenied): service.status() with mock.patch( "psutil._psplatform.cext.winservice_query_config", side_effect=exc ): with pytest.raises(psutil.AccessDenied): service.username() # test __str__ and __repr__ assert service.name() in str(service) assert service.display_name() in str(service) assert service.name() in repr(service) assert service.display_name() in repr(service) tests/test_process_all.py 0000644 00000043653 15030211765 0011640 0 ustar 00 #!/usr/bin/env python3 # Copyright (c) 2009, Giampaolo Rodola'. All rights reserved. # Use of this source code is governed by a BSD-style license that can be # found in the LICENSE file. """Iterate over all process PIDs and for each one of them invoke and test all psutil.Process() methods. """ import enum import errno import multiprocessing import os import stat import time import traceback import psutil from psutil import AIX from psutil import BSD from psutil import FREEBSD from psutil import LINUX from psutil import MACOS from psutil import NETBSD from psutil import OPENBSD from psutil import OSX from psutil import POSIX from psutil import WINDOWS from psutil.tests import CI_TESTING from psutil.tests import PYTEST_PARALLEL from psutil.tests import VALID_PROC_STATUSES from psutil.tests import PsutilTestCase from psutil.tests import check_connection_ntuple from psutil.tests import create_sockets from psutil.tests import is_namedtuple from psutil.tests import is_win_secure_system_proc from psutil.tests import process_namespace from psutil.tests import pytest # Cuts the time in half, but (e.g.) on macOS the process pool stays # alive after join() (multiprocessing bug?), messing up other tests. USE_PROC_POOL = LINUX and not CI_TESTING and not PYTEST_PARALLEL def proc_info(pid): tcase = PsutilTestCase() def check_exception(exc, proc, name, ppid): tcase.assertEqual(exc.pid, pid) if exc.name is not None: tcase.assertEqual(exc.name, name) if isinstance(exc, psutil.ZombieProcess): tcase.assertProcessZombie(proc) if exc.ppid is not None: tcase.assertGreaterEqual(exc.ppid, 0) tcase.assertEqual(exc.ppid, ppid) elif isinstance(exc, psutil.NoSuchProcess): tcase.assertProcessGone(proc) str(exc) repr(exc) def do_wait(): if pid != 0: try: proc.wait(0) except psutil.Error as exc: check_exception(exc, proc, name, ppid) try: proc = psutil.Process(pid) except psutil.NoSuchProcess: tcase.assertPidGone(pid) return {} try: d = proc.as_dict(['ppid', 'name']) except psutil.NoSuchProcess: tcase.assertProcessGone(proc) else: name, ppid = d['name'], d['ppid'] info = {'pid': proc.pid} ns = process_namespace(proc) # We don't use oneshot() because in order not to fool # check_exception() in case of NSP. for fun, fun_name in ns.iter(ns.getters, clear_cache=False): try: info[fun_name] = fun() except psutil.Error as exc: check_exception(exc, proc, name, ppid) continue do_wait() return info class TestFetchAllProcesses(PsutilTestCase): """Test which iterates over all running processes and performs some sanity checks against Process API's returned values. Uses a process pool to get info about all processes. """ def setUp(self): psutil._set_debug(False) # Using a pool in a CI env may result in deadlock, see: # https://github.com/giampaolo/psutil/issues/2104 if USE_PROC_POOL: self.pool = multiprocessing.Pool() def tearDown(self): psutil._set_debug(True) if USE_PROC_POOL: self.pool.terminate() self.pool.join() def iter_proc_info(self): # Fixes "can't pickle <function proc_info>: it's not the # same object as test_process_all.proc_info". from psutil.tests.test_process_all import proc_info if USE_PROC_POOL: return self.pool.imap_unordered(proc_info, psutil.pids()) else: ls = [proc_info(pid) for pid in psutil.pids()] return ls def test_all(self): failures = [] for info in self.iter_proc_info(): for name, value in info.items(): meth = getattr(self, name) try: meth(value, info) except Exception: # noqa: BLE001 s = '\n' + '=' * 70 + '\n' s += ( "FAIL: name=test_{}, pid={}, ret={}\ninfo={}\n".format( name, info['pid'], repr(value), info, ) ) s += '-' * 70 s += f"\n{traceback.format_exc()}" s = "\n".join((" " * 4) + i for i in s.splitlines()) + "\n" failures.append(s) else: if value not in (0, 0.0, [], None, '', {}): assert value, value if failures: raise self.fail(''.join(failures)) def cmdline(self, ret, info): assert isinstance(ret, list) for part in ret: assert isinstance(part, str) def exe(self, ret, info): assert isinstance(ret, str) assert ret.strip() == ret if ret: if WINDOWS and not ret.endswith('.exe'): return # May be "Registry", "MemCompression", ... assert os.path.isabs(ret), ret # Note: os.stat() may return False even if the file is there # hence we skip the test, see: # http://stackoverflow.com/questions/3112546/os-path-exists-lies if POSIX and os.path.isfile(ret): if hasattr(os, 'access') and hasattr(os, "X_OK"): # XXX: may fail on MACOS try: assert os.access(ret, os.X_OK) except AssertionError: if os.path.exists(ret) and not CI_TESTING: raise def pid(self, ret, info): assert isinstance(ret, int) assert ret >= 0 def ppid(self, ret, info): assert isinstance(ret, int) assert ret >= 0 proc_info(ret) def name(self, ret, info): assert isinstance(ret, str) if WINDOWS and not ret and is_win_secure_system_proc(info['pid']): # https://github.com/giampaolo/psutil/issues/2338 return # on AIX, "<exiting>" processes don't have names if not AIX: assert ret, repr(ret) def create_time(self, ret, info): assert isinstance(ret, float) try: assert ret >= 0 except AssertionError: # XXX if OPENBSD and info['status'] == psutil.STATUS_ZOMBIE: pass else: raise # this can't be taken for granted on all platforms # self.assertGreaterEqual(ret, psutil.boot_time()) # make sure returned value can be pretty printed # with strftime time.strftime("%Y %m %d %H:%M:%S", time.localtime(ret)) def uids(self, ret, info): assert is_namedtuple(ret) for uid in ret: assert isinstance(uid, int) assert uid >= 0 def gids(self, ret, info): assert is_namedtuple(ret) # note: testing all gids as above seems not to be reliable for # gid == 30 (nodoby); not sure why. for gid in ret: assert isinstance(gid, int) if not MACOS and not NETBSD: assert gid >= 0 def username(self, ret, info): assert isinstance(ret, str) assert ret.strip() == ret assert ret.strip() def status(self, ret, info): assert isinstance(ret, str) assert ret, ret assert ret != '?' # XXX assert ret in VALID_PROC_STATUSES def io_counters(self, ret, info): assert is_namedtuple(ret) for field in ret: assert isinstance(field, int) if field != -1: assert field >= 0 def ionice(self, ret, info): if LINUX: assert isinstance(ret.ioclass, int) assert isinstance(ret.value, int) assert ret.ioclass >= 0 assert ret.value >= 0 else: # Windows, Cygwin choices = [ psutil.IOPRIO_VERYLOW, psutil.IOPRIO_LOW, psutil.IOPRIO_NORMAL, psutil.IOPRIO_HIGH, ] assert isinstance(ret, int) assert ret >= 0 assert ret in choices def num_threads(self, ret, info): assert isinstance(ret, int) if WINDOWS and ret == 0 and is_win_secure_system_proc(info['pid']): # https://github.com/giampaolo/psutil/issues/2338 return assert ret >= 1 def threads(self, ret, info): assert isinstance(ret, list) for t in ret: assert is_namedtuple(t) assert t.id >= 0 assert t.user_time >= 0 assert t.system_time >= 0 for field in t: assert isinstance(field, (int, float)) def cpu_times(self, ret, info): assert is_namedtuple(ret) for n in ret: assert isinstance(n, float) assert n >= 0 # TODO: check ntuple fields def cpu_percent(self, ret, info): assert isinstance(ret, float) assert 0.0 <= ret <= 100.0, ret def cpu_num(self, ret, info): assert isinstance(ret, int) if FREEBSD and ret == -1: return assert ret >= 0 if psutil.cpu_count() == 1: assert ret == 0 assert ret in list(range(psutil.cpu_count())) def memory_info(self, ret, info): assert is_namedtuple(ret) for value in ret: assert isinstance(value, int) assert value >= 0 if WINDOWS: assert ret.peak_wset >= ret.wset assert ret.peak_paged_pool >= ret.paged_pool assert ret.peak_nonpaged_pool >= ret.nonpaged_pool assert ret.peak_pagefile >= ret.pagefile def memory_full_info(self, ret, info): assert is_namedtuple(ret) total = psutil.virtual_memory().total for name in ret._fields: value = getattr(ret, name) assert isinstance(value, int) assert value >= 0 if LINUX or (OSX and name in {'vms', 'data'}): # On Linux there are processes (e.g. 'goa-daemon') whose # VMS is incredibly high for some reason. continue assert value <= total, name if LINUX: assert ret.pss >= ret.uss def open_files(self, ret, info): assert isinstance(ret, list) for f in ret: assert isinstance(f.fd, int) assert isinstance(f.path, str) assert f.path.strip() == f.path if WINDOWS: assert f.fd == -1 elif LINUX: assert isinstance(f.position, int) assert isinstance(f.mode, str) assert isinstance(f.flags, int) assert f.position >= 0 assert f.mode in {'r', 'w', 'a', 'r+', 'a+'} assert f.flags > 0 elif BSD and not f.path: # XXX see: https://github.com/giampaolo/psutil/issues/595 continue assert os.path.isabs(f.path), f try: st = os.stat(f.path) except FileNotFoundError: pass else: assert stat.S_ISREG(st.st_mode), f def num_fds(self, ret, info): assert isinstance(ret, int) assert ret >= 0 def net_connections(self, ret, info): with create_sockets(): assert len(ret) == len(set(ret)) for conn in ret: assert is_namedtuple(conn) check_connection_ntuple(conn) def cwd(self, ret, info): assert isinstance(ret, str) assert ret.strip() == ret if ret: assert os.path.isabs(ret), ret try: st = os.stat(ret) except OSError as err: if WINDOWS and psutil._psplatform.is_permission_err(err): pass # directory has been removed in mean time elif err.errno != errno.ENOENT: raise else: assert stat.S_ISDIR(st.st_mode) def memory_percent(self, ret, info): assert isinstance(ret, float) assert 0 <= ret <= 100, ret def is_running(self, ret, info): assert isinstance(ret, bool) def cpu_affinity(self, ret, info): assert isinstance(ret, list) assert ret != [] cpus = list(range(psutil.cpu_count())) for n in ret: assert isinstance(n, int) assert n in cpus def terminal(self, ret, info): assert isinstance(ret, (str, type(None))) if ret is not None: assert os.path.isabs(ret), ret assert os.path.exists(ret), ret def memory_maps(self, ret, info): for nt in ret: assert isinstance(nt.addr, str) assert isinstance(nt.perms, str) assert isinstance(nt.path, str) for fname in nt._fields: value = getattr(nt, fname) if fname == 'path': if value.startswith(("[", "anon_inode:")): # linux continue if BSD and value == "pvclock": # seen on FreeBSD continue assert os.path.isabs(nt.path), nt.path # commented as on Linux we might get # '/foo/bar (deleted)' # assert os.path.exists(nt.path), nt.path elif fname == 'addr': assert value, repr(value) elif fname == 'perms': if not WINDOWS: assert value, repr(value) else: assert isinstance(value, int) assert value >= 0 def num_handles(self, ret, info): assert isinstance(ret, int) assert ret >= 0 def nice(self, ret, info): assert isinstance(ret, int) if POSIX: assert -20 <= ret <= 20, ret else: priorities = [ getattr(psutil, x) for x in dir(psutil) if x.endswith('_PRIORITY_CLASS') ] assert ret in priorities assert isinstance(ret, enum.IntEnum) def num_ctx_switches(self, ret, info): assert is_namedtuple(ret) for value in ret: assert isinstance(value, int) assert value >= 0 def rlimit(self, ret, info): assert isinstance(ret, tuple) assert len(ret) == 2 assert ret[0] >= -1 assert ret[1] >= -1 def environ(self, ret, info): assert isinstance(ret, dict) for k, v in ret.items(): assert isinstance(k, str) assert isinstance(v, str) class TestPidsRange(PsutilTestCase): """Given pid_exists() return value for a range of PIDs which may or may not exist, make sure that psutil.Process() and psutil.pids() agree with pid_exists(). This guarantees that the 3 APIs are all consistent with each other. See: https://github.com/giampaolo/psutil/issues/2359 XXX - Note about Windows: it turns out there are some "hidden" PIDs which are not returned by psutil.pids() and are also not revealed by taskmgr.exe and ProcessHacker, still they can be instantiated by psutil.Process() and queried. One of such PIDs is "conhost.exe". Running as_dict() for it reveals that some Process() APIs erroneously raise NoSuchProcess, so we know we have problem there. Let's ignore this for now, since it's quite a corner case (who even imagined hidden PIDs existed on Windows?). """ def setUp(self): psutil._set_debug(False) def tearDown(self): psutil._set_debug(True) def test_it(self): def is_linux_tid(pid): try: f = open(f"/proc/{pid}/status", "rb") # noqa: SIM115 except FileNotFoundError: return False else: with f: for line in f: if line.startswith(b"Tgid:"): tgid = int(line.split()[1]) # If tgid and pid are different then we're # dealing with a process TID. return tgid != pid raise ValueError("'Tgid' line not found") def check(pid): # In case of failure retry up to 3 times in order to avoid # race conditions, especially when running in a CI # environment where PIDs may appear and disappear at any # time. x = 3 while True: exists = psutil.pid_exists(pid) try: if exists: psutil.Process(pid) if not WINDOWS: # see docstring assert pid in psutil.pids() else: # On OpenBSD thread IDs can be instantiated, # and oneshot() succeeds, but other APIs fail # with EINVAL. if not OPENBSD: with pytest.raises(psutil.NoSuchProcess): psutil.Process(pid) if not WINDOWS: # see docstring assert pid not in psutil.pids() except (psutil.Error, AssertionError): x -= 1 if x == 0: raise else: return for pid in range(1, 3000): if LINUX and is_linux_tid(pid): # On Linux a TID (thread ID) can be passed to the # Process class and is querable like a PID (process # ID). Skip it. continue with self.subTest(pid=pid): check(pid) tests/__init__.py 0000644 00000175150 15030211765 0010030 0 ustar 00 # Copyright (c) 2009, Giampaolo Rodola'. All rights reserved. # Use of this source code is governed by a BSD-style license that can be # found in the LICENSE file. """Test utilities.""" import atexit import contextlib import ctypes import enum import errno import functools import gc import importlib import ipaddress import os import platform import random import re import select import shlex import shutil import signal import socket import stat import subprocess import sys import tempfile import textwrap import threading import time import unittest import warnings from socket import AF_INET from socket import AF_INET6 from socket import SOCK_STREAM try: import pytest except ImportError: pytest = None import psutil from psutil import AIX from psutil import LINUX from psutil import MACOS from psutil import NETBSD from psutil import OPENBSD from psutil import POSIX from psutil import SUNOS from psutil import WINDOWS from psutil._common import bytes2human from psutil._common import debug from psutil._common import memoize from psutil._common import print_color from psutil._common import supports_ipv6 if POSIX: from psutil._psposix import wait_pid # fmt: off __all__ = [ # constants 'DEVNULL', 'GLOBAL_TIMEOUT', 'TOLERANCE_SYS_MEM', 'NO_RETRIES', 'PYPY', 'PYTHON_EXE', 'PYTHON_EXE_ENV', 'ROOT_DIR', 'SCRIPTS_DIR', 'TESTFN_PREFIX', 'UNICODE_SUFFIX', 'INVALID_UNICODE_SUFFIX', 'CI_TESTING', 'VALID_PROC_STATUSES', 'TOLERANCE_DISK_USAGE', 'IS_64BIT', "HAS_CPU_AFFINITY", "HAS_CPU_FREQ", "HAS_ENVIRON", "HAS_PROC_IO_COUNTERS", "HAS_IONICE", "HAS_MEMORY_MAPS", "HAS_PROC_CPU_NUM", "HAS_RLIMIT", "HAS_SENSORS_BATTERY", "HAS_BATTERY", "HAS_SENSORS_FANS", "HAS_SENSORS_TEMPERATURES", "HAS_NET_CONNECTIONS_UNIX", "MACOS_11PLUS", "MACOS_12PLUS", "COVERAGE", 'AARCH64', "PYTEST_PARALLEL", # subprocesses 'pyrun', 'terminate', 'reap_children', 'spawn_testproc', 'spawn_zombie', 'spawn_children_pair', # threads 'ThreadTask', # test utils 'unittest', 'skip_on_access_denied', 'skip_on_not_implemented', 'retry_on_failure', 'TestMemoryLeak', 'PsutilTestCase', 'process_namespace', 'system_namespace', 'print_sysinfo', 'is_win_secure_system_proc', 'fake_pytest', # fs utils 'chdir', 'safe_rmpath', 'create_py_exe', 'create_c_exe', 'get_testfn', # os 'get_winver', 'kernel_version', # sync primitives 'call_until', 'wait_for_pid', 'wait_for_file', # network 'check_net_address', 'filter_proc_net_connections', 'get_free_port', 'bind_socket', 'bind_unix_socket', 'tcp_socketpair', 'unix_socketpair', 'create_sockets', # compat 'reload_module', 'import_module_by_path', # others 'warn', 'copyload_shared_lib', 'is_namedtuple', ] # fmt: on # =================================================================== # --- constants # =================================================================== # --- platforms PYPY = '__pypy__' in sys.builtin_module_names # whether we're running this test suite on a Continuous Integration service GITHUB_ACTIONS = 'GITHUB_ACTIONS' in os.environ or 'CIBUILDWHEEL' in os.environ CI_TESTING = GITHUB_ACTIONS COVERAGE = 'COVERAGE_RUN' in os.environ PYTEST_PARALLEL = "PYTEST_XDIST_WORKER" in os.environ # `make test-parallel` # are we a 64 bit process? IS_64BIT = sys.maxsize > 2**32 AARCH64 = platform.machine() == "aarch64" @memoize def macos_version(): version_str = platform.mac_ver()[0] version = tuple(map(int, version_str.split(".")[:2])) if version == (10, 16): # When built against an older macOS SDK, Python will report # macOS 10.16 instead of the real version. version_str = subprocess.check_output( [ sys.executable, "-sS", "-c", "import platform; print(platform.mac_ver()[0])", ], env={"SYSTEM_VERSION_COMPAT": "0"}, universal_newlines=True, ) version = tuple(map(int, version_str.split(".")[:2])) return version if MACOS: MACOS_11PLUS = macos_version() > (10, 15) MACOS_12PLUS = macos_version() >= (12, 0) else: MACOS_11PLUS = False MACOS_12PLUS = False # --- configurable defaults # how many times retry_on_failure() decorator will retry NO_RETRIES = 10 # bytes tolerance for system-wide related tests TOLERANCE_SYS_MEM = 5 * 1024 * 1024 # 5MB TOLERANCE_DISK_USAGE = 10 * 1024 * 1024 # 10MB # the timeout used in functions which have to wait GLOBAL_TIMEOUT = 5 # be more tolerant if we're on CI in order to avoid false positives if CI_TESTING: NO_RETRIES *= 3 GLOBAL_TIMEOUT *= 3 TOLERANCE_SYS_MEM *= 4 TOLERANCE_DISK_USAGE *= 3 # --- file names # Disambiguate TESTFN for parallel testing. if os.name == 'java': # Jython disallows @ in module names TESTFN_PREFIX = f"$psutil-{os.getpid()}-" else: TESTFN_PREFIX = f"@psutil-{os.getpid()}-" UNICODE_SUFFIX = "-ƒőő" # An invalid unicode string. INVALID_UNICODE_SUFFIX = b"f\xc0\x80".decode('utf8', 'surrogateescape') ASCII_FS = sys.getfilesystemencoding().lower() in {"ascii", "us-ascii"} # --- paths ROOT_DIR = os.path.realpath( os.path.join(os.path.dirname(__file__), '..', '..') ) SCRIPTS_DIR = os.environ.get( "PSUTIL_SCRIPTS_DIR", os.path.join(ROOT_DIR, 'scripts') ) HERE = os.path.realpath(os.path.dirname(__file__)) # --- support HAS_CPU_AFFINITY = hasattr(psutil.Process, "cpu_affinity") HAS_CPU_FREQ = hasattr(psutil, "cpu_freq") HAS_ENVIRON = hasattr(psutil.Process, "environ") HAS_GETLOADAVG = hasattr(psutil, "getloadavg") HAS_IONICE = hasattr(psutil.Process, "ionice") HAS_MEMORY_MAPS = hasattr(psutil.Process, "memory_maps") HAS_NET_CONNECTIONS_UNIX = POSIX and not SUNOS HAS_NET_IO_COUNTERS = hasattr(psutil, "net_io_counters") HAS_PROC_CPU_NUM = hasattr(psutil.Process, "cpu_num") HAS_PROC_IO_COUNTERS = hasattr(psutil.Process, "io_counters") HAS_RLIMIT = hasattr(psutil.Process, "rlimit") HAS_SENSORS_BATTERY = hasattr(psutil, "sensors_battery") try: HAS_BATTERY = HAS_SENSORS_BATTERY and bool(psutil.sensors_battery()) except Exception: # noqa: BLE001 HAS_BATTERY = False HAS_SENSORS_FANS = hasattr(psutil, "sensors_fans") HAS_SENSORS_TEMPERATURES = hasattr(psutil, "sensors_temperatures") HAS_THREADS = hasattr(psutil.Process, "threads") SKIP_SYSCONS = (MACOS or AIX) and os.getuid() != 0 # --- misc def _get_py_exe(): def attempt(exe): try: subprocess.check_call( [exe, "-V"], stdout=subprocess.PIPE, stderr=subprocess.PIPE ) except subprocess.CalledProcessError: return None else: return exe env = os.environ.copy() # On Windows, starting with python 3.7, virtual environments use a # venv launcher startup process. This does not play well when # counting spawned processes, or when relying on the PID of the # spawned process to do some checks, e.g. connections check per PID. # Let's use the base python in this case. base = getattr(sys, "_base_executable", None) if WINDOWS and sys.version_info >= (3, 7) and base is not None: # We need to set __PYVENV_LAUNCHER__ to sys.executable for the # base python executable to know about the environment. env["__PYVENV_LAUNCHER__"] = sys.executable return base, env elif GITHUB_ACTIONS: return sys.executable, env elif MACOS: exe = ( attempt(sys.executable) or attempt(os.path.realpath(sys.executable)) or attempt( shutil.which("python{}.{}".format(*sys.version_info[:2])) ) or attempt(psutil.Process().exe()) ) if not exe: raise ValueError("can't find python exe real abspath") return exe, env else: exe = os.path.realpath(sys.executable) assert os.path.exists(exe), exe return exe, env PYTHON_EXE, PYTHON_EXE_ENV = _get_py_exe() DEVNULL = open(os.devnull, 'r+') # noqa: SIM115 atexit.register(DEVNULL.close) VALID_PROC_STATUSES = [ getattr(psutil, x) for x in dir(psutil) if x.startswith('STATUS_') ] AF_UNIX = getattr(socket, "AF_UNIX", object()) _subprocesses_started = set() _pids_started = set() # =================================================================== # --- threads # =================================================================== class ThreadTask(threading.Thread): """A thread task which does nothing expect staying alive.""" def __init__(self): super().__init__() self._running = False self._interval = 0.001 self._flag = threading.Event() def __repr__(self): name = self.__class__.__name__ return f"<{name} running={self._running} at {id(self):#x}>" def __enter__(self): self.start() return self def __exit__(self, *args, **kwargs): self.stop() def start(self): """Start thread and keep it running until an explicit stop() request. Polls for shutdown every 'timeout' seconds. """ if self._running: raise ValueError("already started") threading.Thread.start(self) self._flag.wait() def run(self): self._running = True self._flag.set() while self._running: time.sleep(self._interval) def stop(self): """Stop thread execution and and waits until it is stopped.""" if not self._running: raise ValueError("already stopped") self._running = False self.join() # =================================================================== # --- subprocesses # =================================================================== def _reap_children_on_err(fun): @functools.wraps(fun) def wrapper(*args, **kwargs): try: return fun(*args, **kwargs) except Exception: reap_children() raise return wrapper @_reap_children_on_err def spawn_testproc(cmd=None, **kwds): """Create a python subprocess which does nothing for some secs and return it as a subprocess.Popen instance. If "cmd" is specified that is used instead of python. By default stdin and stdout are redirected to /dev/null. It also attempts to make sure the process is in a reasonably initialized state. The process is registered for cleanup on reap_children(). """ kwds.setdefault("stdin", DEVNULL) kwds.setdefault("stdout", DEVNULL) kwds.setdefault("cwd", os.getcwd()) kwds.setdefault("env", PYTHON_EXE_ENV) if WINDOWS: # Prevents the subprocess to open error dialogs. This will also # cause stderr to be suppressed, which is suboptimal in order # to debug broken tests. CREATE_NO_WINDOW = 0x8000000 kwds.setdefault("creationflags", CREATE_NO_WINDOW) if cmd is None: testfn = get_testfn(dir=os.getcwd()) try: safe_rmpath(testfn) pyline = ( "import time;" f"open(r'{testfn}', 'w').close();" "[time.sleep(0.1) for x in range(100)];" # 10 secs ) cmd = [PYTHON_EXE, "-c", pyline] sproc = subprocess.Popen(cmd, **kwds) _subprocesses_started.add(sproc) wait_for_file(testfn, delete=True, empty=True) finally: safe_rmpath(testfn) else: sproc = subprocess.Popen(cmd, **kwds) _subprocesses_started.add(sproc) wait_for_pid(sproc.pid) return sproc @_reap_children_on_err def spawn_children_pair(): """Create a subprocess which creates another one as in: A (us) -> B (child) -> C (grandchild). Return a (child, grandchild) tuple. The 2 processes are fully initialized and will live for 60 secs and are registered for cleanup on reap_children(). """ tfile = None testfn = get_testfn(dir=os.getcwd()) try: s = textwrap.dedent(f"""\ import subprocess, os, sys, time s = "import os, time;" s += "f = open('{os.path.basename(testfn)}', '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'{PYTHON_EXE}', '-c', s]) p.wait() """) # On Windows if we create a subprocess with CREATE_NO_WINDOW flag # set (which is the default) a "conhost.exe" extra process will be # spawned as a child. We don't want that. if WINDOWS: subp, tfile = pyrun(s, creationflags=0) else: subp, tfile = pyrun(s) child = psutil.Process(subp.pid) grandchild_pid = int(wait_for_file(testfn, delete=True, empty=False)) _pids_started.add(grandchild_pid) grandchild = psutil.Process(grandchild_pid) return (child, grandchild) finally: safe_rmpath(testfn) if tfile is not None: safe_rmpath(tfile) def spawn_zombie(): """Create a zombie process and return a (parent, zombie) process tuple. In order to kill the zombie parent must be terminate()d first, then zombie must be wait()ed on. """ assert psutil.POSIX unix_file = get_testfn() src = textwrap.dedent(f"""\ 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('{unix_file}') pid = bytes(str(os.getpid()), 'ascii') s.sendall(pid) """) tfile = None sock = bind_unix_socket(unix_file) try: sock.settimeout(GLOBAL_TIMEOUT) parent, tfile = pyrun(src) conn, _ = sock.accept() try: select.select([conn.fileno()], [], [], GLOBAL_TIMEOUT) zpid = int(conn.recv(1024)) _pids_started.add(zpid) zombie = psutil.Process(zpid) call_until(lambda: zombie.status() == psutil.STATUS_ZOMBIE) return (parent, zombie) finally: conn.close() finally: sock.close() safe_rmpath(unix_file) if tfile is not None: safe_rmpath(tfile) @_reap_children_on_err def pyrun(src, **kwds): """Run python 'src' code string in a separate interpreter. Returns a subprocess.Popen instance and the test file where the source code was written. """ kwds.setdefault("stdout", None) kwds.setdefault("stderr", None) srcfile = get_testfn() try: with open(srcfile, "w") as f: f.write(src) subp = spawn_testproc([PYTHON_EXE, f.name], **kwds) wait_for_pid(subp.pid) return (subp, srcfile) except Exception: safe_rmpath(srcfile) raise @_reap_children_on_err def sh(cmd, **kwds): """Run cmd in a subprocess and return its output. raises RuntimeError on error. """ # Prevents subprocess to open error dialogs in case of error. flags = 0x8000000 if WINDOWS else 0 kwds.setdefault("stdout", subprocess.PIPE) kwds.setdefault("stderr", subprocess.PIPE) kwds.setdefault("universal_newlines", True) kwds.setdefault("creationflags", flags) if isinstance(cmd, str): cmd = shlex.split(cmd) p = subprocess.Popen(cmd, **kwds) _subprocesses_started.add(p) stdout, stderr = p.communicate(timeout=GLOBAL_TIMEOUT) if p.returncode != 0: raise RuntimeError(stdout + stderr) if stderr: warn(stderr) if stdout.endswith('\n'): stdout = stdout[:-1] return stdout def terminate(proc_or_pid, sig=signal.SIGTERM, wait_timeout=GLOBAL_TIMEOUT): """Terminate a process and wait() for it. Process can be a PID or an instance of psutil.Process(), subprocess.Popen() or psutil.Popen(). If it's a subprocess.Popen() or psutil.Popen() instance also closes its stdin / stdout / stderr fds. PID is wait()ed even if the process is already gone (kills zombies). Does nothing if the process does not exist. Return process exit status. """ def wait(proc, timeout): proc.wait(timeout) if WINDOWS and isinstance(proc, subprocess.Popen): # Otherwise PID may still hang around. try: return psutil.Process(proc.pid).wait(timeout) except psutil.NoSuchProcess: pass def sendsig(proc, sig): # XXX: otherwise the build hangs for some reason. if MACOS and GITHUB_ACTIONS: sig = signal.SIGKILL # If the process received SIGSTOP, SIGCONT is necessary first, # otherwise SIGTERM won't work. if POSIX and sig != signal.SIGKILL: proc.send_signal(signal.SIGCONT) proc.send_signal(sig) def term_subprocess_proc(proc, timeout): try: sendsig(proc, sig) except ProcessLookupError: pass except OSError as err: if WINDOWS and err.winerror == 6: # "invalid handle" pass raise return wait(proc, timeout) def term_psutil_proc(proc, timeout): try: sendsig(proc, sig) except psutil.NoSuchProcess: pass return wait(proc, timeout) def term_pid(pid, timeout): try: proc = psutil.Process(pid) except psutil.NoSuchProcess: # Needed to kill zombies. if POSIX: return wait_pid(pid, timeout) else: return term_psutil_proc(proc, timeout) def flush_popen(proc): if proc.stdout: proc.stdout.close() if proc.stderr: proc.stderr.close() # Flushing a BufferedWriter may raise an error. if proc.stdin: proc.stdin.close() p = proc_or_pid try: if isinstance(p, int): return term_pid(p, wait_timeout) elif isinstance(p, (psutil.Process, psutil.Popen)): return term_psutil_proc(p, wait_timeout) elif isinstance(p, subprocess.Popen): return term_subprocess_proc(p, wait_timeout) else: raise TypeError(f"wrong type {p!r}") finally: if isinstance(p, (subprocess.Popen, psutil.Popen)): flush_popen(p) pid = p if isinstance(p, int) else p.pid assert not psutil.pid_exists(pid), pid def reap_children(recursive=False): """Terminate and wait() any subprocess started by this test suite and any children currently running, ensuring that no processes stick around to hog resources. If recursive is True it also tries to terminate and wait() all grandchildren started by this process. """ # Get the children here before terminating them, as in case of # recursive=True we don't want to lose the intermediate reference # pointing to the grandchildren. children = psutil.Process().children(recursive=recursive) # Terminate subprocess.Popen. while _subprocesses_started: subp = _subprocesses_started.pop() terminate(subp) # Collect started pids. while _pids_started: pid = _pids_started.pop() terminate(pid) # Terminate children. if children: for p in children: terminate(p, wait_timeout=None) _, alive = psutil.wait_procs(children, timeout=GLOBAL_TIMEOUT) for p in alive: warn(f"couldn't terminate process {p!r}; attempting kill()") terminate(p, sig=signal.SIGKILL) # =================================================================== # --- OS # =================================================================== def kernel_version(): """Return a tuple such as (2, 6, 36).""" if not POSIX: raise NotImplementedError("not POSIX") s = "" uname = os.uname()[2] for c in uname: if c.isdigit() or c == '.': s += c else: break if not s: raise ValueError(f"can't parse {uname!r}") minor = 0 micro = 0 nums = s.split('.') major = int(nums[0]) if len(nums) >= 2: minor = int(nums[1]) if len(nums) >= 3: micro = int(nums[2]) return (major, minor, micro) def get_winver(): if not WINDOWS: raise NotImplementedError("not WINDOWS") wv = sys.getwindowsversion() sp = wv.service_pack_major or 0 return (wv[0], wv[1], sp) # =================================================================== # --- sync primitives # =================================================================== class retry: """A retry decorator.""" def __init__( self, exception=Exception, timeout=None, retries=None, interval=0.001, logfun=None, ): if timeout and retries: raise ValueError("timeout and retries args are mutually exclusive") self.exception = exception self.timeout = timeout self.retries = retries self.interval = interval self.logfun = logfun def __iter__(self): if self.timeout: stop_at = time.time() + self.timeout while time.time() < stop_at: yield elif self.retries: for _ in range(self.retries): yield else: while True: yield def sleep(self): if self.interval is not None: time.sleep(self.interval) def __call__(self, fun): @functools.wraps(fun) def wrapper(*args, **kwargs): exc = None for _ in self: try: return fun(*args, **kwargs) except self.exception as _: exc = _ if self.logfun is not None: self.logfun(exc) self.sleep() continue raise exc # This way the user of the decorated function can change config # parameters. wrapper.decorator = self return wrapper @retry( exception=psutil.NoSuchProcess, logfun=None, timeout=GLOBAL_TIMEOUT, interval=0.001, ) def wait_for_pid(pid): """Wait for pid to show up in the process list then return. Used in the test suite to give time the sub process to initialize. """ if pid not in psutil.pids(): raise psutil.NoSuchProcess(pid) psutil.Process(pid) @retry( exception=(FileNotFoundError, AssertionError), logfun=None, timeout=GLOBAL_TIMEOUT, interval=0.001, ) def wait_for_file(fname, delete=True, empty=False): """Wait for a file to be written on disk with some content.""" with open(fname, "rb") as f: data = f.read() if not empty: assert data if delete: safe_rmpath(fname) return data @retry( exception=AssertionError, logfun=None, timeout=GLOBAL_TIMEOUT, interval=0.001, ) def call_until(fun): """Keep calling function until it evaluates to True.""" ret = fun() assert ret return ret # =================================================================== # --- fs # =================================================================== def safe_rmpath(path): """Convenience function for removing temporary test files or dirs.""" def retry_fun(fun): # On Windows it could happen that the file or directory has # open handles or references preventing the delete operation # to succeed immediately, so we retry for a while. See: # https://bugs.python.org/issue33240 stop_at = time.time() + GLOBAL_TIMEOUT while time.time() < stop_at: try: return fun() except FileNotFoundError: pass except OSError as _: err = _ warn(f"ignoring {err}") time.sleep(0.01) raise err try: st = os.stat(path) if stat.S_ISDIR(st.st_mode): fun = functools.partial(shutil.rmtree, path) else: fun = functools.partial(os.remove, path) if POSIX: fun() else: retry_fun(fun) except FileNotFoundError: pass def safe_mkdir(dir): """Convenience function for creating a directory.""" try: os.mkdir(dir) except FileExistsError: pass @contextlib.contextmanager def chdir(dirname): """Context manager which temporarily changes the current directory.""" curdir = os.getcwd() try: os.chdir(dirname) yield finally: os.chdir(curdir) def create_py_exe(path): """Create a Python executable file in the given location.""" assert not os.path.exists(path), path atexit.register(safe_rmpath, path) shutil.copyfile(PYTHON_EXE, path) if POSIX: st = os.stat(path) os.chmod(path, st.st_mode | stat.S_IEXEC) return path def create_c_exe(path, c_code=None): """Create a compiled C executable in the given location.""" assert not os.path.exists(path), path if not shutil.which("gcc"): raise pytest.skip("gcc is not installed") if c_code is None: c_code = textwrap.dedent(""" #include <unistd.h> int main() { pause(); return 1; } """) else: assert isinstance(c_code, str), c_code atexit.register(safe_rmpath, path) with open(get_testfn(suffix='.c'), "w") as f: f.write(c_code) try: subprocess.check_call(["gcc", f.name, "-o", path]) finally: safe_rmpath(f.name) return path def get_testfn(suffix="", dir=None): """Return an absolute pathname of a file or dir that did not exist at the time this call is made. Also schedule it for safe deletion at interpreter exit. It's technically racy but probably not really due to the time variant. """ while True: name = tempfile.mktemp(prefix=TESTFN_PREFIX, suffix=suffix, dir=dir) if not os.path.exists(name): # also include dirs path = os.path.realpath(name) # needed for OSX atexit.register(safe_rmpath, path) return path # =================================================================== # --- testing # =================================================================== class fake_pytest: """A class that mimics some basic pytest APIs. This is meant for when unit tests are run in production, where pytest may not be installed. Still, the user can test psutil installation via: $ python3 -m psutil.tests """ @staticmethod def main(*args, **kw): # noqa: ARG004 """Mimics pytest.main(). It has the same effect as running `python3 -m unittest -v` from the project root directory. """ suite = unittest.TestLoader().discover(HERE) unittest.TextTestRunner(verbosity=2).run(suite) warnings.warn( "Fake pytest module was used. Test results may be inaccurate.", UserWarning, stacklevel=1, ) return suite @staticmethod def raises(exc, match=None): """Mimics `pytest.raises`.""" class ExceptionInfo: _exc = None @property def value(self): return self._exc @contextlib.contextmanager def context(exc, match=None): einfo = ExceptionInfo() try: yield einfo except exc as err: if match and not re.search(match, str(err)): msg = f'"{match}" does not match "{err}"' raise AssertionError(msg) einfo._exc = err else: raise AssertionError(f"{exc!r} not raised") return context(exc, match=match) @staticmethod def warns(warning, match=None): """Mimics `pytest.warns`.""" if match: return unittest.TestCase().assertWarnsRegex(warning, match) return unittest.TestCase().assertWarns(warning) @staticmethod def skip(reason=""): """Mimics `unittest.SkipTest`.""" raise unittest.SkipTest(reason) class mark: @staticmethod def skipif(condition, reason=""): """Mimics `@pytest.mark.skipif` decorator.""" return unittest.skipIf(condition, reason) class xdist_group: """Mimics `@pytest.mark.xdist_group` decorator (no-op).""" def __init__(self, name=None): pass def __call__(self, cls_or_meth): return cls_or_meth if pytest is None: pytest = fake_pytest class PsutilTestCase(unittest.TestCase): """Test class providing auto-cleanup wrappers on top of process test utilities. All test classes should derive from this one, even if we use pytest. """ def get_testfn(self, suffix="", dir=None): fname = get_testfn(suffix=suffix, dir=dir) self.addCleanup(safe_rmpath, fname) return fname def spawn_testproc(self, *args, **kwds): sproc = spawn_testproc(*args, **kwds) self.addCleanup(terminate, sproc) return sproc def spawn_children_pair(self): child1, child2 = spawn_children_pair() self.addCleanup(terminate, child2) self.addCleanup(terminate, child1) # executed first return (child1, child2) def spawn_zombie(self): parent, zombie = spawn_zombie() self.addCleanup(terminate, zombie) self.addCleanup(terminate, parent) # executed first return (parent, zombie) def pyrun(self, *args, **kwds): sproc, srcfile = pyrun(*args, **kwds) self.addCleanup(safe_rmpath, srcfile) self.addCleanup(terminate, sproc) # executed first return sproc def _check_proc_exc(self, proc, exc): assert isinstance(exc, psutil.Error) assert exc.pid == proc.pid assert exc.name == proc._name if exc.name: assert exc.name if isinstance(exc, psutil.ZombieProcess): assert exc.ppid == proc._ppid if exc.ppid is not None: assert exc.ppid >= 0 str(exc) repr(exc) def assertPidGone(self, pid): with pytest.raises(psutil.NoSuchProcess) as cm: try: psutil.Process(pid) except psutil.ZombieProcess: raise AssertionError("wasn't supposed to raise ZombieProcess") assert cm.value.pid == pid assert cm.value.name is None assert not psutil.pid_exists(pid), pid assert pid not in psutil.pids() assert pid not in [x.pid for x in psutil.process_iter()] def assertProcessGone(self, proc): self.assertPidGone(proc.pid) ns = process_namespace(proc) for fun, name in ns.iter(ns.all, clear_cache=True): with self.subTest(proc=proc, name=name): try: ret = fun() except psutil.ZombieProcess: raise except psutil.NoSuchProcess as exc: self._check_proc_exc(proc, exc) else: msg = ( f"Process.{name}() didn't raise NSP and returned" f" {ret!r}" ) raise AssertionError(msg) proc.wait(timeout=0) # assert not raise TimeoutExpired def assertProcessZombie(self, proc): # A zombie process should always be instantiable. clone = psutil.Process(proc.pid) # Cloned zombie on Open/NetBSD has null creation time, see: # https://github.com/giampaolo/psutil/issues/2287 assert proc == clone if not (OPENBSD or NETBSD): assert hash(proc) == hash(clone) # Its status always be querable. assert proc.status() == psutil.STATUS_ZOMBIE # It should be considered 'running'. assert proc.is_running() assert psutil.pid_exists(proc.pid) # as_dict() shouldn't crash. proc.as_dict() # It should show up in pids() and process_iter(). assert proc.pid in psutil.pids() assert proc.pid in [x.pid for x in psutil.process_iter()] psutil._pmap = {} assert proc.pid in [x.pid for x in psutil.process_iter()] # Call all methods. ns = process_namespace(proc) for fun, name in ns.iter(ns.all, clear_cache=True): with self.subTest(proc=proc, name=name): try: fun() except (psutil.ZombieProcess, psutil.AccessDenied) as exc: self._check_proc_exc(proc, exc) if LINUX: # https://github.com/giampaolo/psutil/pull/2288 with pytest.raises(psutil.ZombieProcess) as cm: proc.cmdline() self._check_proc_exc(proc, cm.value) with pytest.raises(psutil.ZombieProcess) as cm: proc.exe() self._check_proc_exc(proc, cm.value) with pytest.raises(psutil.ZombieProcess) as cm: proc.memory_maps() self._check_proc_exc(proc, cm.value) # Zombie cannot be signaled or terminated. proc.suspend() proc.resume() proc.terminate() proc.kill() assert proc.is_running() assert psutil.pid_exists(proc.pid) assert proc.pid in psutil.pids() assert proc.pid in [x.pid for x in psutil.process_iter()] psutil._pmap = {} assert proc.pid in [x.pid for x in psutil.process_iter()] # Its parent should 'see' it (edit: not true on BSD and MACOS). # descendants = [x.pid for x in psutil.Process().children( # recursive=True)] # self.assertIn(proc.pid, descendants) # __eq__ can't be relied upon because creation time may not be # querable. # self.assertEqual(proc, psutil.Process(proc.pid)) # XXX should we also assume ppid() to be usable? Note: this # would be an important use case as the only way to get # rid of a zombie is to kill its parent. # self.assertEqual(proc.ppid(), os.getpid()) @pytest.mark.skipif(PYPY, reason="unreliable on PYPY") class TestMemoryLeak(PsutilTestCase): """Test framework class for detecting function memory leaks, typically functions implemented in C which forgot to free() memory from the heap. It does so by checking whether the process memory usage increased before and after calling the function many times. Note that this is hard (probably impossible) to do reliably, due to how the OS handles memory, the GC and so on (memory can even decrease!). In order to avoid false positives, in case of failure (mem > 0) we retry the test for up to 5 times, increasing call repetitions each time. If the memory keeps increasing then it's a failure. If available (Linux, OSX, Windows), USS memory is used for comparison, since it's supposed to be more precise, see: https://gmpy.dev/blog/2016/real-process-memory-and-environ-in-python If not, RSS memory is used. mallinfo() on Linux and _heapwalk() on Windows may give even more precision, but at the moment are not implemented. PyPy appears to be completely unstable for this framework, probably because of its JIT, so tests on PYPY are skipped. Usage: class TestLeaks(psutil.tests.TestMemoryLeak): def test_fun(self): self.execute(some_function) """ # Configurable class attrs. times = 200 warmup_times = 10 tolerance = 0 # memory retries = 10 if CI_TESTING else 5 verbose = True _thisproc = psutil.Process() _psutil_debug_orig = bool(os.getenv('PSUTIL_DEBUG')) @classmethod def setUpClass(cls): psutil._set_debug(False) # avoid spamming to stderr @classmethod def tearDownClass(cls): psutil._set_debug(cls._psutil_debug_orig) def _get_mem(self): # USS is the closest thing we have to "real" memory usage and it # should be less likely to produce false positives. mem = self._thisproc.memory_full_info() return getattr(mem, "uss", mem.rss) def _get_num_fds(self): if POSIX: return self._thisproc.num_fds() else: return self._thisproc.num_handles() def _log(self, msg): if self.verbose: print_color(msg, color="yellow", file=sys.stderr) def _check_fds(self, fun): """Makes sure num_fds() (POSIX) or num_handles() (Windows) does not increase after calling a function. Used to discover forgotten close(2) and CloseHandle syscalls. """ before = self._get_num_fds() self.call(fun) after = self._get_num_fds() diff = after - before if diff < 0: msg = ( f"negative diff {diff!r} (gc probably collected a" " resource from a previous test)" ) raise self.fail(msg) if diff > 0: type_ = "fd" if POSIX else "handle" if diff > 1: type_ += "s" msg = f"{diff} unclosed {type_} after calling {fun!r}" raise self.fail(msg) def _call_ntimes(self, fun, times): """Get 2 distinct memory samples, before and after having called fun repeatedly, and return the memory difference. """ gc.collect(generation=1) mem1 = self._get_mem() for x in range(times): ret = self.call(fun) del x, ret gc.collect(generation=1) mem2 = self._get_mem() assert gc.garbage == [] diff = mem2 - mem1 # can also be negative return diff def _check_mem(self, fun, times, retries, tolerance): messages = [] prev_mem = 0 increase = times for idx in range(1, retries + 1): mem = self._call_ntimes(fun, times) msg = "Run #{}: extra-mem={}, per-call={}, calls={}".format( idx, bytes2human(mem), bytes2human(mem / times), times, ) messages.append(msg) success = mem <= tolerance or mem <= prev_mem if success: if idx > 1: self._log(msg) return else: if idx == 1: print() # noqa: T201 self._log(msg) times += increase prev_mem = mem raise self.fail(". ".join(messages)) # --- def call(self, fun): return fun() def execute( self, fun, times=None, warmup_times=None, retries=None, tolerance=None ): """Test a callable.""" times = times if times is not None else self.times warmup_times = ( warmup_times if warmup_times is not None else self.warmup_times ) retries = retries if retries is not None else self.retries tolerance = tolerance if tolerance is not None else self.tolerance try: assert times >= 1, "times must be >= 1" assert warmup_times >= 0, "warmup_times must be >= 0" assert retries >= 0, "retries must be >= 0" assert tolerance >= 0, "tolerance must be >= 0" except AssertionError as err: raise ValueError(str(err)) self._call_ntimes(fun, warmup_times) # warm up self._check_fds(fun) self._check_mem(fun, times=times, retries=retries, tolerance=tolerance) def execute_w_exc(self, exc, fun, **kwargs): """Convenience method to test a callable while making sure it raises an exception on every call. """ def call(): self.assertRaises(exc, fun) self.execute(call, **kwargs) def print_sysinfo(): import collections import datetime import getpass import locale import pprint try: import pip except ImportError: pip = None try: import wheel except ImportError: wheel = None info = collections.OrderedDict() # OS if psutil.LINUX and shutil.which("lsb_release"): info['OS'] = sh('lsb_release -d -s') elif psutil.OSX: info['OS'] = f"Darwin {platform.mac_ver()[0]}" elif psutil.WINDOWS: info['OS'] = "Windows " + ' '.join(map(str, platform.win32_ver())) if hasattr(platform, 'win32_edition'): info['OS'] += ", " + platform.win32_edition() else: info['OS'] = f"{platform.system()} {platform.version()}" info['arch'] = ', '.join( list(platform.architecture()) + [platform.machine()] ) if psutil.POSIX: info['kernel'] = platform.uname()[2] # python info['python'] = ', '.join([ platform.python_implementation(), platform.python_version(), platform.python_compiler(), ]) info['pip'] = getattr(pip, '__version__', 'not installed') if wheel is not None: info['pip'] += f" (wheel={wheel.__version__})" # UNIX if psutil.POSIX: if shutil.which("gcc"): out = sh(['gcc', '--version']) info['gcc'] = str(out).split('\n')[0] else: info['gcc'] = 'not installed' s = platform.libc_ver()[1] if s: info['glibc'] = s # system info['fs-encoding'] = sys.getfilesystemencoding() lang = locale.getlocale() info['lang'] = f"{lang[0]}, {lang[1]}" info['boot-time'] = datetime.datetime.fromtimestamp( psutil.boot_time() ).strftime("%Y-%m-%d %H:%M:%S") info['time'] = datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S") info['user'] = getpass.getuser() info['home'] = os.path.expanduser("~") info['cwd'] = os.getcwd() info['pyexe'] = PYTHON_EXE info['hostname'] = platform.node() info['PID'] = os.getpid() # metrics info['cpus'] = psutil.cpu_count() info['loadavg'] = "{:.1f}%, {:.1f}%, {:.1f}%".format( *tuple(x / psutil.cpu_count() * 100 for x in psutil.getloadavg()) ) mem = psutil.virtual_memory() info['memory'] = "{}%%, used={}, total={}".format( int(mem.percent), bytes2human(mem.used), bytes2human(mem.total), ) swap = psutil.swap_memory() info['swap'] = "{}%%, used={}, total={}".format( int(swap.percent), bytes2human(swap.used), bytes2human(swap.total), ) info['pids'] = len(psutil.pids()) pinfo = psutil.Process().as_dict() pinfo.pop('memory_maps', None) info['proc'] = pprint.pformat(pinfo) print("=" * 70, file=sys.stderr) # noqa: T201 for k, v in info.items(): print("{:<17} {}".format(k + ":", v), file=sys.stderr) # noqa: T201 print("=" * 70, file=sys.stderr) # noqa: T201 sys.stdout.flush() # if WINDOWS: # os.system("tasklist") # elif shutil.which("ps"): # os.system("ps aux") # print("=" * 70, file=sys.stderr) sys.stdout.flush() def is_win_secure_system_proc(pid): # see: https://github.com/giampaolo/psutil/issues/2338 @memoize def get_procs(): ret = {} out = sh("tasklist.exe /NH /FO csv") for line in out.splitlines()[1:]: bits = [x.replace('"', "") for x in line.split(",")] name, pid = bits[0], int(bits[1]) ret[pid] = name return ret try: return get_procs()[pid] == "Secure System" except KeyError: return False def _get_eligible_cpu(): p = psutil.Process() if hasattr(p, "cpu_num"): return p.cpu_num() elif hasattr(p, "cpu_affinity"): return random.choice(p.cpu_affinity()) return 0 class process_namespace: """A container that lists all Process class method names + some reasonable parameters to be called with. Utility methods (parent(), children(), ...) are excluded. >>> ns = process_namespace(psutil.Process()) >>> for fun, name in ns.iter(ns.getters): ... fun() """ utils = [('cpu_percent', (), {}), ('memory_percent', (), {})] ignored = [ ('as_dict', (), {}), ('children', (), {'recursive': True}), ('connections', (), {}), # deprecated ('is_running', (), {}), ('oneshot', (), {}), ('parent', (), {}), ('parents', (), {}), ('pid', (), {}), ('wait', (0,), {}), ] getters = [ ('cmdline', (), {}), ('cpu_times', (), {}), ('create_time', (), {}), ('cwd', (), {}), ('exe', (), {}), ('memory_full_info', (), {}), ('memory_info', (), {}), ('name', (), {}), ('net_connections', (), {'kind': 'all'}), ('nice', (), {}), ('num_ctx_switches', (), {}), ('num_threads', (), {}), ('open_files', (), {}), ('ppid', (), {}), ('status', (), {}), ('threads', (), {}), ('username', (), {}), ] if POSIX: getters += [('uids', (), {})] getters += [('gids', (), {})] getters += [('terminal', (), {})] getters += [('num_fds', (), {})] if HAS_PROC_IO_COUNTERS: getters += [('io_counters', (), {})] if HAS_IONICE: getters += [('ionice', (), {})] if HAS_RLIMIT: getters += [('rlimit', (psutil.RLIMIT_NOFILE,), {})] if HAS_CPU_AFFINITY: getters += [('cpu_affinity', (), {})] if HAS_PROC_CPU_NUM: getters += [('cpu_num', (), {})] if HAS_ENVIRON: getters += [('environ', (), {})] if WINDOWS: getters += [('num_handles', (), {})] if HAS_MEMORY_MAPS: getters += [('memory_maps', (), {'grouped': False})] setters = [] if POSIX: setters += [('nice', (0,), {})] else: setters += [('nice', (psutil.NORMAL_PRIORITY_CLASS,), {})] if HAS_RLIMIT: setters += [('rlimit', (psutil.RLIMIT_NOFILE, (1024, 4096)), {})] if HAS_IONICE: if LINUX: setters += [('ionice', (psutil.IOPRIO_CLASS_NONE, 0), {})] else: setters += [('ionice', (psutil.IOPRIO_NORMAL,), {})] if HAS_CPU_AFFINITY: setters += [('cpu_affinity', ([_get_eligible_cpu()],), {})] killers = [ ('send_signal', (signal.SIGTERM,), {}), ('suspend', (), {}), ('resume', (), {}), ('terminate', (), {}), ('kill', (), {}), ] if WINDOWS: killers += [('send_signal', (signal.CTRL_C_EVENT,), {})] killers += [('send_signal', (signal.CTRL_BREAK_EVENT,), {})] all = utils + getters + setters + killers def __init__(self, proc): self._proc = proc def iter(self, ls, clear_cache=True): """Given a list of tuples yields a set of (fun, fun_name) tuples in random order. """ ls = list(ls) random.shuffle(ls) for fun_name, args, kwds in ls: if clear_cache: self.clear_cache() fun = getattr(self._proc, fun_name) fun = functools.partial(fun, *args, **kwds) yield (fun, fun_name) def clear_cache(self): """Clear the cache of a Process instance.""" self._proc._init(self._proc.pid, _ignore_nsp=True) @classmethod def test_class_coverage(cls, test_class, ls): """Given a TestCase instance and a list of tuples checks that the class defines the required test method names. """ for fun_name, _, _ in ls: meth_name = 'test_' + fun_name if not hasattr(test_class, meth_name): msg = ( f"{test_class.__class__.__name__!r} class should define a" f" {meth_name!r} method" ) raise AttributeError(msg) @classmethod def test(cls): this = {x[0] for x in cls.all} ignored = {x[0] for x in cls.ignored} klass = {x for x in dir(psutil.Process) if x[0] != '_'} leftout = (this | ignored) ^ klass if leftout: raise ValueError(f"uncovered Process class names: {leftout!r}") class system_namespace: """A container that lists all the module-level, system-related APIs. Utilities such as cpu_percent() are excluded. Usage: >>> ns = system_namespace >>> for fun, name in ns.iter(ns.getters): ... fun() """ getters = [ ('boot_time', (), {}), ('cpu_count', (), {'logical': False}), ('cpu_count', (), {'logical': True}), ('cpu_stats', (), {}), ('cpu_times', (), {'percpu': False}), ('cpu_times', (), {'percpu': True}), ('disk_io_counters', (), {'perdisk': True}), ('disk_partitions', (), {'all': True}), ('disk_usage', (os.getcwd(),), {}), ('net_connections', (), {'kind': 'all'}), ('net_if_addrs', (), {}), ('net_if_stats', (), {}), ('net_io_counters', (), {'pernic': True}), ('pid_exists', (os.getpid(),), {}), ('pids', (), {}), ('swap_memory', (), {}), ('users', (), {}), ('virtual_memory', (), {}), ] if HAS_CPU_FREQ: if MACOS and platform.machine() == 'arm64': # skipped due to #1892 pass else: getters += [('cpu_freq', (), {'percpu': True})] if HAS_GETLOADAVG: getters += [('getloadavg', (), {})] if HAS_SENSORS_TEMPERATURES: getters += [('sensors_temperatures', (), {})] if HAS_SENSORS_FANS: getters += [('sensors_fans', (), {})] if HAS_SENSORS_BATTERY: getters += [('sensors_battery', (), {})] if WINDOWS: getters += [('win_service_iter', (), {})] getters += [('win_service_get', ('alg',), {})] ignored = [ ('process_iter', (), {}), ('wait_procs', ([psutil.Process()],), {}), ('cpu_percent', (), {}), ('cpu_times_percent', (), {}), ] all = getters @staticmethod def iter(ls): """Given a list of tuples yields a set of (fun, fun_name) tuples in random order. """ ls = list(ls) random.shuffle(ls) for fun_name, args, kwds in ls: fun = getattr(psutil, fun_name) fun = functools.partial(fun, *args, **kwds) yield (fun, fun_name) test_class_coverage = process_namespace.test_class_coverage def retry_on_failure(retries=NO_RETRIES): """Decorator which runs a test function and retries N times before actually failing. """ def logfun(exc): print(f"{exc!r}, retrying", file=sys.stderr) # noqa: T201 return retry( exception=AssertionError, timeout=None, retries=retries, logfun=logfun ) def skip_on_access_denied(only_if=None): """Decorator to Ignore AccessDenied exceptions.""" def decorator(fun): @functools.wraps(fun) def wrapper(*args, **kwargs): try: return fun(*args, **kwargs) except psutil.AccessDenied: if only_if is not None: if not only_if: raise raise pytest.skip("raises AccessDenied") return wrapper return decorator def skip_on_not_implemented(only_if=None): """Decorator to Ignore NotImplementedError exceptions.""" def decorator(fun): @functools.wraps(fun) def wrapper(*args, **kwargs): try: return fun(*args, **kwargs) except NotImplementedError: if only_if is not None: if not only_if: raise msg = ( f"{fun.__name__!r} was skipped because it raised" " NotImplementedError" ) raise pytest.skip(msg) return wrapper return decorator # =================================================================== # --- network # =================================================================== # XXX: no longer used def get_free_port(host='127.0.0.1'): """Return an unused TCP port. Subject to race conditions.""" with socket.socket() as sock: sock.bind((host, 0)) return sock.getsockname()[1] def bind_socket(family=AF_INET, type=SOCK_STREAM, addr=None): """Binds a generic socket.""" if addr is None and family in {AF_INET, AF_INET6}: addr = ("", 0) sock = socket.socket(family, type) try: if os.name not in {'nt', 'cygwin'}: sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1) sock.bind(addr) if type == socket.SOCK_STREAM: sock.listen(5) return sock except Exception: sock.close() raise def bind_unix_socket(name, type=socket.SOCK_STREAM): """Bind a UNIX socket.""" assert psutil.POSIX assert not os.path.exists(name), name sock = socket.socket(socket.AF_UNIX, type) try: sock.bind(name) if type == socket.SOCK_STREAM: sock.listen(5) except Exception: sock.close() raise return sock def tcp_socketpair(family, addr=("", 0)): """Build a pair of TCP sockets connected to each other. Return a (server, client) tuple. """ with socket.socket(family, SOCK_STREAM) as ll: ll.bind(addr) ll.listen(5) addr = ll.getsockname() c = socket.socket(family, SOCK_STREAM) try: c.connect(addr) caddr = c.getsockname() while True: a, addr = ll.accept() # check that we've got the correct client if addr == caddr: return (a, c) a.close() except OSError: c.close() raise def unix_socketpair(name): """Build a pair of UNIX sockets connected to each other through the same UNIX file name. Return a (server, client) tuple. """ assert psutil.POSIX server = client = None try: server = bind_unix_socket(name, type=socket.SOCK_STREAM) server.setblocking(0) client = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM) client.setblocking(0) client.connect(name) # new = server.accept() except Exception: if server is not None: server.close() if client is not None: client.close() raise return (server, client) @contextlib.contextmanager def create_sockets(): """Open as many socket families / types as possible.""" socks = [] fname1 = fname2 = None try: socks.extend(( bind_socket(socket.AF_INET, socket.SOCK_STREAM), bind_socket(socket.AF_INET, socket.SOCK_DGRAM), )) if supports_ipv6(): socks.extend(( bind_socket(socket.AF_INET6, socket.SOCK_STREAM), bind_socket(socket.AF_INET6, socket.SOCK_DGRAM), )) if POSIX and HAS_NET_CONNECTIONS_UNIX: fname1 = get_testfn() fname2 = get_testfn() s1, s2 = unix_socketpair(fname1) s3 = bind_unix_socket(fname2, type=socket.SOCK_DGRAM) for s in (s1, s2, s3): socks.append(s) yield socks finally: for s in socks: s.close() for fname in (fname1, fname2): if fname is not None: safe_rmpath(fname) def check_net_address(addr, family): """Check a net address validity. Supported families are IPv4, IPv6 and MAC addresses. """ assert isinstance(family, enum.IntEnum), family if family == socket.AF_INET: octs = [int(x) for x in addr.split('.')] assert len(octs) == 4, addr for num in octs: assert 0 <= num <= 255, addr ipaddress.IPv4Address(addr) elif family == socket.AF_INET6: assert isinstance(addr, str), addr ipaddress.IPv6Address(addr) elif family == psutil.AF_LINK: assert re.match(r'([a-fA-F0-9]{2}[:|\-]?){6}', addr) is not None, addr else: raise ValueError(f"unknown family {family!r}") def check_connection_ntuple(conn): """Check validity of a connection namedtuple.""" def check_ntuple(conn): has_pid = len(conn) == 7 assert len(conn) in {6, 7}, len(conn) assert conn[0] == conn.fd, conn.fd assert conn[1] == conn.family, conn.family assert conn[2] == conn.type, conn.type assert conn[3] == conn.laddr, conn.laddr assert conn[4] == conn.raddr, conn.raddr assert conn[5] == conn.status, conn.status if has_pid: assert conn[6] == conn.pid, conn.pid def check_family(conn): assert conn.family in {AF_INET, AF_INET6, AF_UNIX}, conn.family assert isinstance(conn.family, enum.IntEnum), conn if conn.family == AF_INET: # actually try to bind the local socket; ignore IPv6 # sockets as their address might be represented as # an IPv4-mapped-address (e.g. "::127.0.0.1") # and that's rejected by bind() with socket.socket(conn.family, conn.type) as s: try: s.bind((conn.laddr[0], 0)) except OSError as err: if err.errno != errno.EADDRNOTAVAIL: raise elif conn.family == AF_UNIX: assert conn.status == psutil.CONN_NONE, conn.status def check_type(conn): # SOCK_SEQPACKET may happen in case of AF_UNIX socks SOCK_SEQPACKET = getattr(socket, "SOCK_SEQPACKET", object()) assert conn.type in { socket.SOCK_STREAM, socket.SOCK_DGRAM, SOCK_SEQPACKET, }, conn.type assert isinstance(conn.type, enum.IntEnum), conn if conn.type == socket.SOCK_DGRAM: assert conn.status == psutil.CONN_NONE, conn.status def check_addrs(conn): # check IP address and port sanity for addr in (conn.laddr, conn.raddr): if conn.family in {AF_INET, AF_INET6}: assert isinstance(addr, tuple), type(addr) if not addr: continue assert isinstance(addr.port, int), type(addr.port) assert 0 <= addr.port <= 65535, addr.port check_net_address(addr.ip, conn.family) elif conn.family == AF_UNIX: assert isinstance(addr, str), type(addr) def check_status(conn): assert isinstance(conn.status, str), conn.status valids = [ getattr(psutil, x) for x in dir(psutil) if x.startswith('CONN_') ] assert conn.status in valids, conn.status if conn.family in {AF_INET, AF_INET6} and conn.type == SOCK_STREAM: assert conn.status != psutil.CONN_NONE, conn.status else: assert conn.status == psutil.CONN_NONE, conn.status check_ntuple(conn) check_family(conn) check_type(conn) check_addrs(conn) check_status(conn) def filter_proc_net_connections(cons): """Our process may start with some open UNIX sockets which are not initialized by us, invalidating unit tests. """ new = [] for conn in cons: if POSIX and conn.family == socket.AF_UNIX: if MACOS and "/syslog" in conn.raddr: debug(f"skipping {conn}") continue new.append(conn) return new # =================================================================== # --- import utils # =================================================================== def reload_module(module): return importlib.reload(module) def import_module_by_path(path): name = os.path.splitext(os.path.basename(path))[0] spec = importlib.util.spec_from_file_location(name, path) mod = importlib.util.module_from_spec(spec) spec.loader.exec_module(mod) return mod # =================================================================== # --- others # =================================================================== def warn(msg): """Raise a warning msg.""" warnings.warn(msg, UserWarning, stacklevel=2) def is_namedtuple(x): """Check if object is an instance of namedtuple.""" t = type(x) b = t.__bases__ if len(b) != 1 or b[0] is not tuple: return False f = getattr(t, '_fields', None) if not isinstance(f, tuple): return False return all(isinstance(n, str) for n in f) if POSIX: @contextlib.contextmanager def copyload_shared_lib(suffix=""): """Ctx manager which picks up a random shared CO lib used by this process, copies it in another location and loads it in memory via ctypes. Return the new absolutized path. """ exe = 'pypy' if PYPY else 'python' ext = ".so" dst = get_testfn(suffix=suffix + ext) libs = [ x.path for x in psutil.Process().memory_maps() if os.path.splitext(x.path)[1] == ext and exe in x.path.lower() ] src = random.choice(libs) shutil.copyfile(src, dst) try: ctypes.CDLL(dst) yield dst finally: safe_rmpath(dst) else: @contextlib.contextmanager def copyload_shared_lib(suffix=""): """Ctx manager which picks up a random shared DLL lib used by this process, copies it in another location and loads it in memory via ctypes. Return the new absolutized, normcased path. """ from ctypes import WinError from ctypes import wintypes ext = ".dll" dst = get_testfn(suffix=suffix + ext) libs = [ x.path for x in psutil.Process().memory_maps() if x.path.lower().endswith(ext) and 'python' in os.path.basename(x.path).lower() and 'wow64' not in x.path.lower() ] if PYPY and not libs: libs = [ x.path for x in psutil.Process().memory_maps() if 'pypy' in os.path.basename(x.path).lower() ] src = random.choice(libs) shutil.copyfile(src, dst) cfile = None try: cfile = ctypes.WinDLL(dst) yield dst finally: # Work around OverflowError: # - https://ci.appveyor.com/project/giampaolo/psutil/build/1207/ # job/o53330pbnri9bcw7 # - http://bugs.python.org/issue30286 # - http://stackoverflow.com/questions/23522055 if cfile is not None: FreeLibrary = ctypes.windll.kernel32.FreeLibrary FreeLibrary.argtypes = [wintypes.HMODULE] ret = FreeLibrary(cfile._handle) if ret == 0: raise WinError() safe_rmpath(dst) # =================================================================== # --- Exit funs (first is executed last) # =================================================================== # this is executed first @atexit.register def cleanup_test_procs(): reap_children(recursive=True) # atexit module does not execute exit functions in case of SIGTERM, which # gets sent to test subprocesses, which is a problem if they import this # module. With this it will. See: # https://gmpy.dev/blog/2016/how-to-always-execute-exit-functions-in-python if POSIX: signal.signal(signal.SIGTERM, lambda sig, _: sys.exit(sig)) _pslinux.py 0000644 00000250014 15030211765 0006761 0 ustar 00 # Copyright (c) 2009, Giampaolo Rodola'. All rights reserved. # Use of this source code is governed by a BSD-style license that can be # found in the LICENSE file. """Linux platform implementation.""" import base64 import collections import enum import errno import functools import glob import os import re import resource import socket import struct import sys import warnings from collections import defaultdict from collections import namedtuple from . import _common from . import _psposix from . import _psutil_linux as cext from . import _psutil_posix as cext_posix from ._common import ENCODING from ._common import NIC_DUPLEX_FULL from ._common import NIC_DUPLEX_HALF from ._common import NIC_DUPLEX_UNKNOWN from ._common import AccessDenied from ._common import NoSuchProcess from ._common import ZombieProcess from ._common import bcat from ._common import cat from ._common import debug from ._common import decode from ._common import get_procfs_path from ._common import isfile_strict from ._common import memoize from ._common import memoize_when_activated from ._common import open_binary from ._common import open_text from ._common import parse_environ_block from ._common import path_exists_strict from ._common import supports_ipv6 from ._common import usage_percent # fmt: off __extra__all__ = [ 'PROCFS_PATH', # io prio constants "IOPRIO_CLASS_NONE", "IOPRIO_CLASS_RT", "IOPRIO_CLASS_BE", "IOPRIO_CLASS_IDLE", # connection status constants "CONN_ESTABLISHED", "CONN_SYN_SENT", "CONN_SYN_RECV", "CONN_FIN_WAIT1", "CONN_FIN_WAIT2", "CONN_TIME_WAIT", "CONN_CLOSE", "CONN_CLOSE_WAIT", "CONN_LAST_ACK", "CONN_LISTEN", "CONN_CLOSING", ] if hasattr(resource, "prlimit"): __extra__all__.extend( [x for x in dir(cext) if x.startswith('RLIM') and x.isupper()] ) # fmt: on # ===================================================================== # --- globals # ===================================================================== POWER_SUPPLY_PATH = "/sys/class/power_supply" HAS_PROC_SMAPS = os.path.exists(f"/proc/{os.getpid()}/smaps") HAS_PROC_SMAPS_ROLLUP = os.path.exists(f"/proc/{os.getpid()}/smaps_rollup") HAS_PROC_IO_PRIORITY = hasattr(cext, "proc_ioprio_get") HAS_CPU_AFFINITY = hasattr(cext, "proc_cpu_affinity_get") # Number of clock ticks per second CLOCK_TICKS = os.sysconf("SC_CLK_TCK") PAGESIZE = cext_posix.getpagesize() BOOT_TIME = None # set later LITTLE_ENDIAN = sys.byteorder == 'little' # "man iostat" states that sectors are equivalent with blocks and have # a size of 512 bytes. Despite this value can be queried at runtime # via /sys/block/{DISK}/queue/hw_sector_size and results may vary # between 1k, 2k, or 4k... 512 appears to be a magic constant used # throughout Linux source code: # * https://stackoverflow.com/a/38136179/376587 # * https://lists.gt.net/linux/kernel/2241060 # * https://github.com/giampaolo/psutil/issues/1305 # * https://github.com/torvalds/linux/blob/ # 4f671fe2f9523a1ea206f63fe60a7c7b3a56d5c7/include/linux/bio.h#L99 # * https://lkml.org/lkml/2015/8/17/234 DISK_SECTOR_SIZE = 512 AddressFamily = enum.IntEnum( 'AddressFamily', {'AF_LINK': int(socket.AF_PACKET)} ) AF_LINK = AddressFamily.AF_LINK # ioprio_* constants http://linux.die.net/man/2/ioprio_get class IOPriority(enum.IntEnum): IOPRIO_CLASS_NONE = 0 IOPRIO_CLASS_RT = 1 IOPRIO_CLASS_BE = 2 IOPRIO_CLASS_IDLE = 3 globals().update(IOPriority.__members__) # See: # https://github.com/torvalds/linux/blame/master/fs/proc/array.c # ...and (TASK_* constants): # https://github.com/torvalds/linux/blob/master/include/linux/sched.h PROC_STATUSES = { "R": _common.STATUS_RUNNING, "S": _common.STATUS_SLEEPING, "D": _common.STATUS_DISK_SLEEP, "T": _common.STATUS_STOPPED, "t": _common.STATUS_TRACING_STOP, "Z": _common.STATUS_ZOMBIE, "X": _common.STATUS_DEAD, "x": _common.STATUS_DEAD, "K": _common.STATUS_WAKE_KILL, "W": _common.STATUS_WAKING, "I": _common.STATUS_IDLE, "P": _common.STATUS_PARKED, } # https://github.com/torvalds/linux/blob/master/include/net/tcp_states.h TCP_STATUSES = { "01": _common.CONN_ESTABLISHED, "02": _common.CONN_SYN_SENT, "03": _common.CONN_SYN_RECV, "04": _common.CONN_FIN_WAIT1, "05": _common.CONN_FIN_WAIT2, "06": _common.CONN_TIME_WAIT, "07": _common.CONN_CLOSE, "08": _common.CONN_CLOSE_WAIT, "09": _common.CONN_LAST_ACK, "0A": _common.CONN_LISTEN, "0B": _common.CONN_CLOSING, } # ===================================================================== # --- named tuples # ===================================================================== # fmt: off # psutil.virtual_memory() svmem = namedtuple( 'svmem', ['total', 'available', 'percent', 'used', 'free', 'active', 'inactive', 'buffers', 'cached', 'shared', 'slab']) # psutil.disk_io_counters() sdiskio = namedtuple( 'sdiskio', ['read_count', 'write_count', 'read_bytes', 'write_bytes', 'read_time', 'write_time', 'read_merged_count', 'write_merged_count', 'busy_time']) # psutil.Process().open_files() popenfile = namedtuple( 'popenfile', ['path', 'fd', 'position', 'mode', 'flags']) # psutil.Process().memory_info() pmem = namedtuple('pmem', 'rss vms shared text lib data dirty') # psutil.Process().memory_full_info() pfullmem = namedtuple('pfullmem', pmem._fields + ('uss', 'pss', 'swap')) # psutil.Process().memory_maps(grouped=True) pmmap_grouped = namedtuple( 'pmmap_grouped', ['path', 'rss', 'size', 'pss', 'shared_clean', 'shared_dirty', 'private_clean', 'private_dirty', 'referenced', 'anonymous', 'swap']) # psutil.Process().memory_maps(grouped=False) pmmap_ext = namedtuple( 'pmmap_ext', 'addr perms ' + ' '.join(pmmap_grouped._fields)) # psutil.Process.io_counters() pio = namedtuple('pio', ['read_count', 'write_count', 'read_bytes', 'write_bytes', 'read_chars', 'write_chars']) # psutil.Process.cpu_times() pcputimes = namedtuple('pcputimes', ['user', 'system', 'children_user', 'children_system', 'iowait']) # fmt: on # ===================================================================== # --- utils # ===================================================================== def readlink(path): """Wrapper around os.readlink().""" assert isinstance(path, str), path path = os.readlink(path) # readlink() might return paths containing null bytes ('\x00') # resulting in "TypeError: must be encoded string without NULL # bytes, not str" errors when the string is passed to other # fs-related functions (os.*, open(), ...). # Apparently everything after '\x00' is garbage (we can have # ' (deleted)', 'new' and possibly others), see: # https://github.com/giampaolo/psutil/issues/717 path = path.split('\x00')[0] # Certain paths have ' (deleted)' appended. Usually this is # bogus as the file actually exists. Even if it doesn't we # don't care. if path.endswith(' (deleted)') and not path_exists_strict(path): path = path[:-10] return path def file_flags_to_mode(flags): """Convert file's open() flags into a readable string. Used by Process.open_files(). """ modes_map = {os.O_RDONLY: 'r', os.O_WRONLY: 'w', os.O_RDWR: 'w+'} mode = modes_map[flags & (os.O_RDONLY | os.O_WRONLY | os.O_RDWR)] if flags & os.O_APPEND: mode = mode.replace('w', 'a', 1) mode = mode.replace('w+', 'r+') # possible values: r, w, a, r+, a+ return mode def is_storage_device(name): """Return True if the given name refers to a root device (e.g. "sda", "nvme0n1") as opposed to a logical partition (e.g. "sda1", "nvme0n1p1"). If name is a virtual device (e.g. "loop1", "ram") return True. """ # Re-adapted from iostat source code, see: # https://github.com/sysstat/sysstat/blob/ # 97912938cd476645b267280069e83b1c8dc0e1c7/common.c#L208 # Some devices may have a slash in their name (e.g. cciss/c0d0...). name = name.replace('/', '!') including_virtual = True if including_virtual: path = f"/sys/block/{name}" else: path = f"/sys/block/{name}/device" return os.access(path, os.F_OK) @memoize def set_scputimes_ntuple(procfs_path): """Set a namedtuple of variable fields depending on the CPU times available on this Linux kernel version which may be: (user, nice, system, idle, iowait, irq, softirq, [steal, [guest, [guest_nice]]]) Used by cpu_times() function. """ global scputimes with open_binary(f"{procfs_path}/stat") as f: values = f.readline().split()[1:] fields = ['user', 'nice', 'system', 'idle', 'iowait', 'irq', 'softirq'] vlen = len(values) if vlen >= 8: # Linux >= 2.6.11 fields.append('steal') if vlen >= 9: # Linux >= 2.6.24 fields.append('guest') if vlen >= 10: # Linux >= 3.2.0 fields.append('guest_nice') scputimes = namedtuple('scputimes', fields) try: set_scputimes_ntuple("/proc") except Exception as err: # noqa: BLE001 # Don't want to crash at import time. debug(f"ignoring exception on import: {err!r}") scputimes = namedtuple('scputimes', 'user system idle')(0.0, 0.0, 0.0) # ===================================================================== # --- system memory # ===================================================================== def calculate_avail_vmem(mems): """Fallback for kernels < 3.14 where /proc/meminfo does not provide "MemAvailable", see: https://blog.famzah.net/2014/09/24/. This code reimplements the algorithm outlined here: https://git.kernel.org/cgit/linux/kernel/git/torvalds/linux.git/ commit/?id=34e431b0ae398fc54ea69ff85ec700722c9da773 We use this function also when "MemAvailable" returns 0 (possibly a kernel bug, see: https://github.com/giampaolo/psutil/issues/1915). In that case this routine matches "free" CLI tool result ("available" column). XXX: on recent kernels this calculation may differ by ~1.5% compared to "MemAvailable:", as it's calculated slightly differently. It is still way more realistic than doing (free + cached) though. See: * https://gitlab.com/procps-ng/procps/issues/42 * https://github.com/famzah/linux-memavailable-procfs/issues/2 """ # Note about "fallback" value. According to: # https://git.kernel.org/cgit/linux/kernel/git/torvalds/linux.git/ # commit/?id=34e431b0ae398fc54ea69ff85ec700722c9da773 # ...long ago "available" memory was calculated as (free + cached), # We use fallback when one of these is missing from /proc/meminfo: # "Active(file)": introduced in 2.6.28 / Dec 2008 # "Inactive(file)": introduced in 2.6.28 / Dec 2008 # "SReclaimable": introduced in 2.6.19 / Nov 2006 # /proc/zoneinfo: introduced in 2.6.13 / Aug 2005 free = mems[b'MemFree:'] fallback = free + mems.get(b"Cached:", 0) try: lru_active_file = mems[b'Active(file):'] lru_inactive_file = mems[b'Inactive(file):'] slab_reclaimable = mems[b'SReclaimable:'] except KeyError as err: debug( f"{err.args[0]} is missing from /proc/meminfo; using an" " approximation for calculating available memory" ) return fallback try: f = open_binary(f"{get_procfs_path()}/zoneinfo") except OSError: return fallback # kernel 2.6.13 watermark_low = 0 with f: for line in f: line = line.strip() if line.startswith(b'low'): watermark_low += int(line.split()[1]) watermark_low *= PAGESIZE avail = free - watermark_low pagecache = lru_active_file + lru_inactive_file pagecache -= min(pagecache / 2, watermark_low) avail += pagecache avail += slab_reclaimable - min(slab_reclaimable / 2.0, watermark_low) return int(avail) def virtual_memory(): """Report virtual memory stats. This implementation mimics procps-ng-3.3.12, aka "free" CLI tool: https://gitlab.com/procps-ng/procps/blob/ 24fd2605c51fccc375ab0287cec33aa767f06718/proc/sysinfo.c#L778-791 The returned values are supposed to match both "free" and "vmstat -s" CLI tools. """ missing_fields = [] mems = {} with open_binary(f"{get_procfs_path()}/meminfo") as f: for line in f: fields = line.split() mems[fields[0]] = int(fields[1]) * 1024 # /proc doc states that the available fields in /proc/meminfo vary # by architecture and compile options, but these 3 values are also # returned by sysinfo(2); as such we assume they are always there. total = mems[b'MemTotal:'] free = mems[b'MemFree:'] try: buffers = mems[b'Buffers:'] except KeyError: # https://github.com/giampaolo/psutil/issues/1010 buffers = 0 missing_fields.append('buffers') try: cached = mems[b"Cached:"] except KeyError: cached = 0 missing_fields.append('cached') else: # "free" cmdline utility sums reclaimable to cached. # Older versions of procps used to add slab memory instead. # This got changed in: # https://gitlab.com/procps-ng/procps/commit/ # 05d751c4f076a2f0118b914c5e51cfbb4762ad8e cached += mems.get(b"SReclaimable:", 0) # since kernel 2.6.19 try: shared = mems[b'Shmem:'] # since kernel 2.6.32 except KeyError: try: shared = mems[b'MemShared:'] # kernels 2.4 except KeyError: shared = 0 missing_fields.append('shared') try: active = mems[b"Active:"] except KeyError: active = 0 missing_fields.append('active') try: inactive = mems[b"Inactive:"] except KeyError: try: inactive = ( mems[b"Inact_dirty:"] + mems[b"Inact_clean:"] + mems[b"Inact_laundry:"] ) except KeyError: inactive = 0 missing_fields.append('inactive') try: slab = mems[b"Slab:"] except KeyError: slab = 0 used = total - free - cached - buffers if used < 0: # May be symptomatic of running within a LCX container where such # values will be dramatically distorted over those of the host. used = total - free # - starting from 4.4.0 we match free's "available" column. # Before 4.4.0 we calculated it as (free + buffers + cached) # which matched htop. # - free and htop available memory differs as per: # http://askubuntu.com/a/369589 # http://unix.stackexchange.com/a/65852/168884 # - MemAvailable has been introduced in kernel 3.14 try: avail = mems[b'MemAvailable:'] except KeyError: avail = calculate_avail_vmem(mems) else: if avail == 0: # Yes, it can happen (probably a kernel bug): # https://github.com/giampaolo/psutil/issues/1915 # In this case "free" CLI tool makes an estimate. We do the same, # and it matches "free" CLI tool. avail = calculate_avail_vmem(mems) if avail < 0: avail = 0 missing_fields.append('available') elif avail > total: # If avail is greater than total or our calculation overflows, # that's symptomatic of running within a LCX container where such # values will be dramatically distorted over those of the host. # https://gitlab.com/procps-ng/procps/blob/ # 24fd2605c51fccc375ab0287cec33aa767f06718/proc/sysinfo.c#L764 avail = free percent = usage_percent((total - avail), total, round_=1) # Warn about missing metrics which are set to 0. if missing_fields: msg = "{} memory stats couldn't be determined and {} set to 0".format( ", ".join(missing_fields), "was" if len(missing_fields) == 1 else "were", ) warnings.warn(msg, RuntimeWarning, stacklevel=2) return svmem( total, avail, percent, used, free, active, inactive, buffers, cached, shared, slab, ) def swap_memory(): """Return swap memory metrics.""" mems = {} with open_binary(f"{get_procfs_path()}/meminfo") as f: for line in f: fields = line.split() mems[fields[0]] = int(fields[1]) * 1024 # We prefer /proc/meminfo over sysinfo() syscall so that # psutil.PROCFS_PATH can be used in order to allow retrieval # for linux containers, see: # https://github.com/giampaolo/psutil/issues/1015 try: total = mems[b'SwapTotal:'] free = mems[b'SwapFree:'] except KeyError: _, _, _, _, total, free, unit_multiplier = cext.linux_sysinfo() total *= unit_multiplier free *= unit_multiplier used = total - free percent = usage_percent(used, total, round_=1) # get pgin/pgouts try: f = open_binary(f"{get_procfs_path()}/vmstat") except OSError as err: # see https://github.com/giampaolo/psutil/issues/722 msg = ( "'sin' and 'sout' swap memory stats couldn't " f"be determined and were set to 0 ({err})" ) warnings.warn(msg, RuntimeWarning, stacklevel=2) sin = sout = 0 else: with f: sin = sout = None for line in f: # values are expressed in 4 kilo bytes, we want # bytes instead if line.startswith(b'pswpin'): sin = int(line.split(b' ')[1]) * 4 * 1024 elif line.startswith(b'pswpout'): sout = int(line.split(b' ')[1]) * 4 * 1024 if sin is not None and sout is not None: break else: # we might get here when dealing with exotic Linux # flavors, see: # https://github.com/giampaolo/psutil/issues/313 msg = "'sin' and 'sout' swap memory stats couldn't " msg += "be determined and were set to 0" warnings.warn(msg, RuntimeWarning, stacklevel=2) sin = sout = 0 return _common.sswap(total, used, free, percent, sin, sout) # ===================================================================== # --- CPU # ===================================================================== def cpu_times(): """Return a named tuple representing the following system-wide CPU times: (user, nice, system, idle, iowait, irq, softirq [steal, [guest, [guest_nice]]]) Last 3 fields may not be available on all Linux kernel versions. """ procfs_path = get_procfs_path() set_scputimes_ntuple(procfs_path) with open_binary(f"{procfs_path}/stat") as f: values = f.readline().split() fields = values[1 : len(scputimes._fields) + 1] fields = [float(x) / CLOCK_TICKS for x in fields] return scputimes(*fields) def per_cpu_times(): """Return a list of namedtuple representing the CPU times for every CPU available on the system. """ procfs_path = get_procfs_path() set_scputimes_ntuple(procfs_path) cpus = [] with open_binary(f"{procfs_path}/stat") as f: # get rid of the first line which refers to system wide CPU stats f.readline() for line in f: if line.startswith(b'cpu'): values = line.split() fields = values[1 : len(scputimes._fields) + 1] fields = [float(x) / CLOCK_TICKS for x in fields] entry = scputimes(*fields) cpus.append(entry) return cpus def cpu_count_logical(): """Return the number of logical CPUs in the system.""" try: return os.sysconf("SC_NPROCESSORS_ONLN") except ValueError: # as a second fallback we try to parse /proc/cpuinfo num = 0 with open_binary(f"{get_procfs_path()}/cpuinfo") as f: for line in f: if line.lower().startswith(b'processor'): num += 1 # unknown format (e.g. amrel/sparc architectures), see: # https://github.com/giampaolo/psutil/issues/200 # try to parse /proc/stat as a last resort if num == 0: search = re.compile(r'cpu\d') with open_text(f"{get_procfs_path()}/stat") as f: for line in f: line = line.split(' ')[0] if search.match(line): num += 1 if num == 0: # mimic os.cpu_count() return None return num def cpu_count_cores(): """Return the number of CPU cores in the system.""" # Method #1 ls = set() # These 2 files are the same but */core_cpus_list is newer while # */thread_siblings_list is deprecated and may disappear in the future. # https://www.kernel.org/doc/Documentation/admin-guide/cputopology.rst # https://github.com/giampaolo/psutil/pull/1727#issuecomment-707624964 # https://lkml.org/lkml/2019/2/26/41 p1 = "/sys/devices/system/cpu/cpu[0-9]*/topology/core_cpus_list" p2 = "/sys/devices/system/cpu/cpu[0-9]*/topology/thread_siblings_list" for path in glob.glob(p1) or glob.glob(p2): with open_binary(path) as f: ls.add(f.read().strip()) result = len(ls) if result != 0: return result # Method #2 mapping = {} current_info = {} with open_binary(f"{get_procfs_path()}/cpuinfo") as f: for line in f: line = line.strip().lower() if not line: # new section try: mapping[current_info[b'physical id']] = current_info[ b'cpu cores' ] except KeyError: pass current_info = {} elif line.startswith((b'physical id', b'cpu cores')): # ongoing section key, value = line.split(b'\t:', 1) current_info[key] = int(value) result = sum(mapping.values()) return result or None # mimic os.cpu_count() def cpu_stats(): """Return various CPU stats as a named tuple.""" with open_binary(f"{get_procfs_path()}/stat") as f: ctx_switches = None interrupts = None soft_interrupts = None for line in f: if line.startswith(b'ctxt'): ctx_switches = int(line.split()[1]) elif line.startswith(b'intr'): interrupts = int(line.split()[1]) elif line.startswith(b'softirq'): soft_interrupts = int(line.split()[1]) if ( ctx_switches is not None and soft_interrupts is not None and interrupts is not None ): break syscalls = 0 return _common.scpustats( ctx_switches, interrupts, soft_interrupts, syscalls ) def _cpu_get_cpuinfo_freq(): """Return current CPU frequency from cpuinfo if available.""" with open_binary(f"{get_procfs_path()}/cpuinfo") as f: return [ float(line.split(b':', 1)[1]) for line in f if line.lower().startswith(b'cpu mhz') ] if os.path.exists("/sys/devices/system/cpu/cpufreq/policy0") or os.path.exists( "/sys/devices/system/cpu/cpu0/cpufreq" ): def cpu_freq(): """Return frequency metrics for all CPUs. Contrarily to other OSes, Linux updates these values in real-time. """ cpuinfo_freqs = _cpu_get_cpuinfo_freq() paths = glob.glob( "/sys/devices/system/cpu/cpufreq/policy[0-9]*" ) or glob.glob("/sys/devices/system/cpu/cpu[0-9]*/cpufreq") paths.sort(key=lambda x: int(re.search(r"[0-9]+", x).group())) ret = [] pjoin = os.path.join for i, path in enumerate(paths): if len(paths) == len(cpuinfo_freqs): # take cached value from cpuinfo if available, see: # https://github.com/giampaolo/psutil/issues/1851 curr = cpuinfo_freqs[i] * 1000 else: curr = bcat(pjoin(path, "scaling_cur_freq"), fallback=None) if curr is None: # Likely an old RedHat, see: # https://github.com/giampaolo/psutil/issues/1071 curr = bcat(pjoin(path, "cpuinfo_cur_freq"), fallback=None) if curr is None: online_path = f"/sys/devices/system/cpu/cpu{i}/online" # if cpu core is offline, set to all zeroes if cat(online_path, fallback=None) == "0\n": ret.append(_common.scpufreq(0.0, 0.0, 0.0)) continue msg = "can't find current frequency file" raise NotImplementedError(msg) curr = int(curr) / 1000 max_ = int(bcat(pjoin(path, "scaling_max_freq"))) / 1000 min_ = int(bcat(pjoin(path, "scaling_min_freq"))) / 1000 ret.append(_common.scpufreq(curr, min_, max_)) return ret else: def cpu_freq(): """Alternate implementation using /proc/cpuinfo. min and max frequencies are not available and are set to None. """ return [_common.scpufreq(x, 0.0, 0.0) for x in _cpu_get_cpuinfo_freq()] # ===================================================================== # --- network # ===================================================================== net_if_addrs = cext_posix.net_if_addrs class _Ipv6UnsupportedError(Exception): pass class NetConnections: """A wrapper on top of /proc/net/* files, retrieving per-process and system-wide open connections (TCP, UDP, UNIX) similarly to "netstat -an". Note: in case of UNIX sockets we're only able to determine the local endpoint/path, not the one it's connected to. According to [1] it would be possible but not easily. [1] http://serverfault.com/a/417946 """ def __init__(self): # The string represents the basename of the corresponding # /proc/net/{proto_name} file. tcp4 = ("tcp", socket.AF_INET, socket.SOCK_STREAM) tcp6 = ("tcp6", socket.AF_INET6, socket.SOCK_STREAM) udp4 = ("udp", socket.AF_INET, socket.SOCK_DGRAM) udp6 = ("udp6", socket.AF_INET6, socket.SOCK_DGRAM) unix = ("unix", socket.AF_UNIX, None) self.tmap = { "all": (tcp4, tcp6, udp4, udp6, unix), "tcp": (tcp4, tcp6), "tcp4": (tcp4,), "tcp6": (tcp6,), "udp": (udp4, udp6), "udp4": (udp4,), "udp6": (udp6,), "unix": (unix,), "inet": (tcp4, tcp6, udp4, udp6), "inet4": (tcp4, udp4), "inet6": (tcp6, udp6), } self._procfs_path = None def get_proc_inodes(self, pid): inodes = defaultdict(list) for fd in os.listdir(f"{self._procfs_path}/{pid}/fd"): try: inode = readlink(f"{self._procfs_path}/{pid}/fd/{fd}") except (FileNotFoundError, ProcessLookupError): # ENOENT == file which is gone in the meantime; # os.stat(f"/proc/{self.pid}") will be done later # to force NSP (if it's the case) continue except OSError as err: if err.errno == errno.EINVAL: # not a link continue if err.errno == errno.ENAMETOOLONG: # file name too long debug(err) continue raise else: if inode.startswith('socket:['): # the process is using a socket inode = inode[8:][:-1] inodes[inode].append((pid, int(fd))) return inodes def get_all_inodes(self): inodes = {} for pid in pids(): try: inodes.update(self.get_proc_inodes(pid)) except (FileNotFoundError, ProcessLookupError, PermissionError): # os.listdir() is gonna raise a lot of access denied # exceptions in case of unprivileged user; that's fine # as we'll just end up returning a connection with PID # and fd set to None anyway. # Both netstat -an and lsof does the same so it's # unlikely we can do any better. # ENOENT just means a PID disappeared on us. continue return inodes @staticmethod def decode_address(addr, family): """Accept an "ip:port" address as displayed in /proc/net/* and convert it into a human readable form, like: "0500000A:0016" -> ("10.0.0.5", 22) "0000000000000000FFFF00000100007F:9E49" -> ("::ffff:127.0.0.1", 40521) The IP address portion is a little or big endian four-byte hexadecimal number; that is, the least significant byte is listed first, so we need to reverse the order of the bytes to convert it to an IP address. The port is represented as a two-byte hexadecimal number. Reference: http://linuxdevcenter.com/pub/a/linux/2000/11/16/LinuxAdmin.html """ ip, port = addr.split(':') port = int(port, 16) # this usually refers to a local socket in listen mode with # no end-points connected if not port: return () ip = ip.encode('ascii') if family == socket.AF_INET: # see: https://github.com/giampaolo/psutil/issues/201 if LITTLE_ENDIAN: ip = socket.inet_ntop(family, base64.b16decode(ip)[::-1]) else: ip = socket.inet_ntop(family, base64.b16decode(ip)) else: # IPv6 ip = base64.b16decode(ip) try: # see: https://github.com/giampaolo/psutil/issues/201 if LITTLE_ENDIAN: ip = socket.inet_ntop( socket.AF_INET6, struct.pack('>4I', *struct.unpack('<4I', ip)), ) else: ip = socket.inet_ntop( socket.AF_INET6, struct.pack('<4I', *struct.unpack('<4I', ip)), ) except ValueError: # see: https://github.com/giampaolo/psutil/issues/623 if not supports_ipv6(): raise _Ipv6UnsupportedError from None raise return _common.addr(ip, port) @staticmethod def process_inet(file, family, type_, inodes, filter_pid=None): """Parse /proc/net/tcp* and /proc/net/udp* files.""" if file.endswith('6') and not os.path.exists(file): # IPv6 not supported return with open_text(file) as f: f.readline() # skip the first line for lineno, line in enumerate(f, 1): try: _, laddr, raddr, status, _, _, _, _, _, inode = ( line.split()[:10] ) except ValueError: msg = ( f"error while parsing {file}; malformed line" f" {lineno} {line!r}" ) raise RuntimeError(msg) from None if inode in inodes: # # We assume inet sockets are unique, so we error # # out if there are multiple references to the # # same inode. We won't do this for UNIX sockets. # if len(inodes[inode]) > 1 and family != socket.AF_UNIX: # raise ValueError("ambiguous inode with multiple " # "PIDs references") pid, fd = inodes[inode][0] else: pid, fd = None, -1 if filter_pid is not None and filter_pid != pid: continue else: if type_ == socket.SOCK_STREAM: status = TCP_STATUSES[status] else: status = _common.CONN_NONE try: laddr = NetConnections.decode_address(laddr, family) raddr = NetConnections.decode_address(raddr, family) except _Ipv6UnsupportedError: continue yield (fd, family, type_, laddr, raddr, status, pid) @staticmethod def process_unix(file, family, inodes, filter_pid=None): """Parse /proc/net/unix files.""" with open_text(file) as f: f.readline() # skip the first line for line in f: tokens = line.split() try: _, _, _, _, type_, _, inode = tokens[0:7] except ValueError: if ' ' not in line: # see: https://github.com/giampaolo/psutil/issues/766 continue msg = ( f"error while parsing {file}; malformed line {line!r}" ) raise RuntimeError(msg) # noqa: B904 if inode in inodes: # noqa: SIM108 # With UNIX sockets we can have a single inode # referencing many file descriptors. pairs = inodes[inode] else: pairs = [(None, -1)] for pid, fd in pairs: if filter_pid is not None and filter_pid != pid: continue else: path = tokens[-1] if len(tokens) == 8 else '' type_ = _common.socktype_to_enum(int(type_)) # XXX: determining the remote endpoint of a # UNIX socket on Linux is not possible, see: # https://serverfault.com/questions/252723/ raddr = "" status = _common.CONN_NONE yield (fd, family, type_, path, raddr, status, pid) def retrieve(self, kind, pid=None): self._procfs_path = get_procfs_path() if pid is not None: inodes = self.get_proc_inodes(pid) if not inodes: # no connections for this process return [] else: inodes = self.get_all_inodes() ret = set() for proto_name, family, type_ in self.tmap[kind]: path = f"{self._procfs_path}/net/{proto_name}" if family in {socket.AF_INET, socket.AF_INET6}: ls = self.process_inet( path, family, type_, inodes, filter_pid=pid ) else: ls = self.process_unix(path, family, inodes, filter_pid=pid) for fd, family, type_, laddr, raddr, status, bound_pid in ls: if pid: conn = _common.pconn( fd, family, type_, laddr, raddr, status ) else: conn = _common.sconn( fd, family, type_, laddr, raddr, status, bound_pid ) ret.add(conn) return list(ret) _net_connections = NetConnections() def net_connections(kind='inet'): """Return system-wide open connections.""" return _net_connections.retrieve(kind) def net_io_counters(): """Return network I/O statistics for every network interface installed on the system as a dict of raw tuples. """ with open_text(f"{get_procfs_path()}/net/dev") as f: lines = f.readlines() retdict = {} for line in lines[2:]: colon = line.rfind(':') assert colon > 0, repr(line) name = line[:colon].strip() fields = line[colon + 1 :].strip().split() ( # in bytes_recv, packets_recv, errin, dropin, _fifoin, # unused _framein, # unused _compressedin, # unused _multicastin, # unused # out bytes_sent, packets_sent, errout, dropout, _fifoout, # unused _collisionsout, # unused _carrierout, # unused _compressedout, # unused ) = map(int, fields) retdict[name] = ( bytes_sent, bytes_recv, packets_sent, packets_recv, errin, errout, dropin, dropout, ) return retdict def net_if_stats(): """Get NIC stats (isup, duplex, speed, mtu).""" duplex_map = { cext.DUPLEX_FULL: NIC_DUPLEX_FULL, cext.DUPLEX_HALF: NIC_DUPLEX_HALF, cext.DUPLEX_UNKNOWN: NIC_DUPLEX_UNKNOWN, } names = net_io_counters().keys() ret = {} for name in names: try: mtu = cext_posix.net_if_mtu(name) flags = cext_posix.net_if_flags(name) duplex, speed = cext.net_if_duplex_speed(name) except OSError as err: # https://github.com/giampaolo/psutil/issues/1279 if err.errno != errno.ENODEV: raise debug(err) else: output_flags = ','.join(flags) isup = 'running' in flags ret[name] = _common.snicstats( isup, duplex_map[duplex], speed, mtu, output_flags ) return ret # ===================================================================== # --- disks # ===================================================================== disk_usage = _psposix.disk_usage def disk_io_counters(perdisk=False): """Return disk I/O statistics for every disk installed on the system as a dict of raw tuples. """ def read_procfs(): # OK, this is a bit confusing. The format of /proc/diskstats can # have 3 variations. # On Linux 2.4 each line has always 15 fields, e.g.: # "3 0 8 hda 8 8 8 8 8 8 8 8 8 8 8" # On Linux 2.6+ each line *usually* has 14 fields, and the disk # name is in another position, like this: # "3 0 hda 8 8 8 8 8 8 8 8 8 8 8" # ...unless (Linux 2.6) the line refers to a partition instead # of a disk, in which case the line has less fields (7): # "3 1 hda1 8 8 8 8" # 4.18+ has 4 fields added: # "3 0 hda 8 8 8 8 8 8 8 8 8 8 8 0 0 0 0" # 5.5 has 2 more fields. # See: # https://www.kernel.org/doc/Documentation/iostats.txt # https://www.kernel.org/doc/Documentation/ABI/testing/procfs-diskstats with open_text(f"{get_procfs_path()}/diskstats") as f: lines = f.readlines() for line in lines: fields = line.split() flen = len(fields) # fmt: off if flen == 15: # Linux 2.4 name = fields[3] reads = int(fields[2]) (reads_merged, rbytes, rtime, writes, writes_merged, wbytes, wtime, _, busy_time, _) = map(int, fields[4:14]) elif flen == 14 or flen >= 18: # Linux 2.6+, line referring to a disk name = fields[2] (reads, reads_merged, rbytes, rtime, writes, writes_merged, wbytes, wtime, _, busy_time, _) = map(int, fields[3:14]) elif flen == 7: # Linux 2.6+, line referring to a partition name = fields[2] reads, rbytes, writes, wbytes = map(int, fields[3:]) rtime = wtime = reads_merged = writes_merged = busy_time = 0 else: msg = f"not sure how to interpret line {line!r}" raise ValueError(msg) yield (name, reads, writes, rbytes, wbytes, rtime, wtime, reads_merged, writes_merged, busy_time) # fmt: on def read_sysfs(): for block in os.listdir('/sys/block'): for root, _, files in os.walk(os.path.join('/sys/block', block)): if 'stat' not in files: continue with open_text(os.path.join(root, 'stat')) as f: fields = f.read().strip().split() name = os.path.basename(root) # fmt: off (reads, reads_merged, rbytes, rtime, writes, writes_merged, wbytes, wtime, _, busy_time) = map(int, fields[:10]) yield (name, reads, writes, rbytes, wbytes, rtime, wtime, reads_merged, writes_merged, busy_time) # fmt: on if os.path.exists(f"{get_procfs_path()}/diskstats"): gen = read_procfs() elif os.path.exists('/sys/block'): gen = read_sysfs() else: msg = ( f"{get_procfs_path()}/diskstats nor /sys/block are available on" " this system" ) raise NotImplementedError(msg) retdict = {} for entry in gen: # fmt: off (name, reads, writes, rbytes, wbytes, rtime, wtime, reads_merged, writes_merged, busy_time) = entry if not perdisk and not is_storage_device(name): # perdisk=False means we want to calculate totals so we skip # partitions (e.g. 'sda1', 'nvme0n1p1') and only include # base disk devices (e.g. 'sda', 'nvme0n1'). Base disks # include a total of all their partitions + some extra size # of their own: # $ cat /proc/diskstats # 259 0 sda 10485760 ... # 259 1 sda1 5186039 ... # 259 1 sda2 5082039 ... # See: # https://github.com/giampaolo/psutil/pull/1313 continue rbytes *= DISK_SECTOR_SIZE wbytes *= DISK_SECTOR_SIZE retdict[name] = (reads, writes, rbytes, wbytes, rtime, wtime, reads_merged, writes_merged, busy_time) # fmt: on return retdict class RootFsDeviceFinder: """disk_partitions() may return partitions with device == "/dev/root" or "rootfs". This container class uses different strategies to try to obtain the real device path. Resources: https://bootlin.com/blog/find-root-device/ https://www.systutorials.com/how-to-find-the-disk-where-root-is-on-in-bash-on-linux/. """ __slots__ = ['major', 'minor'] def __init__(self): dev = os.stat("/").st_dev self.major = os.major(dev) self.minor = os.minor(dev) def ask_proc_partitions(self): with open_text(f"{get_procfs_path()}/partitions") as f: for line in f.readlines()[2:]: fields = line.split() if len(fields) < 4: # just for extra safety continue major = int(fields[0]) if fields[0].isdigit() else None minor = int(fields[1]) if fields[1].isdigit() else None name = fields[3] if major == self.major and minor == self.minor: if name: # just for extra safety return f"/dev/{name}" def ask_sys_dev_block(self): path = f"/sys/dev/block/{self.major}:{self.minor}/uevent" with open_text(path) as f: for line in f: if line.startswith("DEVNAME="): name = line.strip().rpartition("DEVNAME=")[2] if name: # just for extra safety return f"/dev/{name}" def ask_sys_class_block(self): needle = f"{self.major}:{self.minor}" files = glob.iglob("/sys/class/block/*/dev") for file in files: try: f = open_text(file) except FileNotFoundError: # race condition continue else: with f: data = f.read().strip() if data == needle: name = os.path.basename(os.path.dirname(file)) return f"/dev/{name}" def find(self): path = None if path is None: try: path = self.ask_proc_partitions() except OSError as err: debug(err) if path is None: try: path = self.ask_sys_dev_block() except OSError as err: debug(err) if path is None: try: path = self.ask_sys_class_block() except OSError as err: debug(err) # We use exists() because the "/dev/*" part of the path is hard # coded, so we want to be sure. if path is not None and os.path.exists(path): return path def disk_partitions(all=False): """Return mounted disk partitions as a list of namedtuples.""" fstypes = set() procfs_path = get_procfs_path() if not all: with open_text(f"{procfs_path}/filesystems") as f: for line in f: line = line.strip() if not line.startswith("nodev"): fstypes.add(line.strip()) else: # ignore all lines starting with "nodev" except "nodev zfs" fstype = line.split("\t")[1] if fstype == "zfs": fstypes.add("zfs") # See: https://github.com/giampaolo/psutil/issues/1307 if procfs_path == "/proc" and os.path.isfile('/etc/mtab'): mounts_path = os.path.realpath("/etc/mtab") else: mounts_path = os.path.realpath(f"{procfs_path}/self/mounts") retlist = [] partitions = cext.disk_partitions(mounts_path) for partition in partitions: device, mountpoint, fstype, opts = partition if device == 'none': device = '' if device in {"/dev/root", "rootfs"}: device = RootFsDeviceFinder().find() or device if not all: if not device or fstype not in fstypes: continue ntuple = _common.sdiskpart(device, mountpoint, fstype, opts) retlist.append(ntuple) return retlist # ===================================================================== # --- sensors # ===================================================================== def sensors_temperatures(): """Return hardware (CPU and others) temperatures as a dict including hardware name, label, current, max and critical temperatures. Implementation notes: - /sys/class/hwmon looks like the most recent interface to retrieve this info, and this implementation relies on it only (old distros will probably use something else) - lm-sensors on Ubuntu 16.04 relies on /sys/class/hwmon - /sys/class/thermal/thermal_zone* is another one but it's more difficult to parse """ ret = collections.defaultdict(list) basenames = glob.glob('/sys/class/hwmon/hwmon*/temp*_*') # CentOS has an intermediate /device directory: # https://github.com/giampaolo/psutil/issues/971 # https://github.com/nicolargo/glances/issues/1060 basenames.extend(glob.glob('/sys/class/hwmon/hwmon*/device/temp*_*')) basenames = sorted({x.split('_')[0] for x in basenames}) # Only add the coretemp hwmon entries if they're not already in # /sys/class/hwmon/ # https://github.com/giampaolo/psutil/issues/1708 # https://github.com/giampaolo/psutil/pull/1648 basenames2 = glob.glob( '/sys/devices/platform/coretemp.*/hwmon/hwmon*/temp*_*' ) repl = re.compile(r"/sys/devices/platform/coretemp.*/hwmon/") for name in basenames2: altname = repl.sub('/sys/class/hwmon/', name) if altname not in basenames: basenames.append(name) for base in basenames: try: path = base + '_input' current = float(bcat(path)) / 1000.0 path = os.path.join(os.path.dirname(base), 'name') unit_name = cat(path).strip() except (OSError, ValueError): # A lot of things can go wrong here, so let's just skip the # whole entry. Sure thing is Linux's /sys/class/hwmon really # is a stinky broken mess. # https://github.com/giampaolo/psutil/issues/1009 # https://github.com/giampaolo/psutil/issues/1101 # https://github.com/giampaolo/psutil/issues/1129 # https://github.com/giampaolo/psutil/issues/1245 # https://github.com/giampaolo/psutil/issues/1323 continue high = bcat(base + '_max', fallback=None) critical = bcat(base + '_crit', fallback=None) label = cat(base + '_label', fallback='').strip() if high is not None: try: high = float(high) / 1000.0 except ValueError: high = None if critical is not None: try: critical = float(critical) / 1000.0 except ValueError: critical = None ret[unit_name].append((label, current, high, critical)) # Indication that no sensors were detected in /sys/class/hwmon/ if not basenames: basenames = glob.glob('/sys/class/thermal/thermal_zone*') basenames = sorted(set(basenames)) for base in basenames: try: path = os.path.join(base, 'temp') current = float(bcat(path)) / 1000.0 path = os.path.join(base, 'type') unit_name = cat(path).strip() except (OSError, ValueError) as err: debug(err) continue trip_paths = glob.glob(base + '/trip_point*') trip_points = { '_'.join(os.path.basename(p).split('_')[0:3]) for p in trip_paths } critical = None high = None for trip_point in trip_points: path = os.path.join(base, trip_point + "_type") trip_type = cat(path, fallback='').strip() if trip_type == 'critical': critical = bcat( os.path.join(base, trip_point + "_temp"), fallback=None ) elif trip_type == 'high': high = bcat( os.path.join(base, trip_point + "_temp"), fallback=None ) if high is not None: try: high = float(high) / 1000.0 except ValueError: high = None if critical is not None: try: critical = float(critical) / 1000.0 except ValueError: critical = None ret[unit_name].append(('', current, high, critical)) return dict(ret) def sensors_fans(): """Return hardware fans info (for CPU and other peripherals) as a dict including hardware label and current speed. Implementation notes: - /sys/class/hwmon looks like the most recent interface to retrieve this info, and this implementation relies on it only (old distros will probably use something else) - lm-sensors on Ubuntu 16.04 relies on /sys/class/hwmon """ ret = collections.defaultdict(list) basenames = glob.glob('/sys/class/hwmon/hwmon*/fan*_*') if not basenames: # CentOS has an intermediate /device directory: # https://github.com/giampaolo/psutil/issues/971 basenames = glob.glob('/sys/class/hwmon/hwmon*/device/fan*_*') basenames = sorted({x.split("_")[0] for x in basenames}) for base in basenames: try: current = int(bcat(base + '_input')) except OSError as err: debug(err) continue unit_name = cat(os.path.join(os.path.dirname(base), 'name')).strip() label = cat(base + '_label', fallback='').strip() ret[unit_name].append(_common.sfan(label, current)) return dict(ret) def sensors_battery(): """Return battery information. Implementation note: it appears /sys/class/power_supply/BAT0/ directory structure may vary and provide files with the same meaning but under different names, see: https://github.com/giampaolo/psutil/issues/966. """ null = object() def multi_bcat(*paths): """Attempt to read the content of multiple files which may not exist. If none of them exist return None. """ for path in paths: ret = bcat(path, fallback=null) if ret != null: try: return int(ret) except ValueError: return ret.strip() return None bats = [ x for x in os.listdir(POWER_SUPPLY_PATH) if x.startswith('BAT') or 'battery' in x.lower() ] if not bats: return None # Get the first available battery. Usually this is "BAT0", except # some rare exceptions: # https://github.com/giampaolo/psutil/issues/1238 root = os.path.join(POWER_SUPPLY_PATH, min(bats)) # Base metrics. energy_now = multi_bcat(root + "/energy_now", root + "/charge_now") power_now = multi_bcat(root + "/power_now", root + "/current_now") energy_full = multi_bcat(root + "/energy_full", root + "/charge_full") time_to_empty = multi_bcat(root + "/time_to_empty_now") # Percent. If we have energy_full the percentage will be more # accurate compared to reading /capacity file (float vs. int). if energy_full is not None and energy_now is not None: try: percent = 100.0 * energy_now / energy_full except ZeroDivisionError: percent = 0.0 else: percent = int(cat(root + "/capacity", fallback=-1)) if percent == -1: return None # Is AC power cable plugged in? # Note: AC0 is not always available and sometimes (e.g. CentOS7) # it's called "AC". power_plugged = None online = multi_bcat( os.path.join(POWER_SUPPLY_PATH, "AC0/online"), os.path.join(POWER_SUPPLY_PATH, "AC/online"), ) if online is not None: power_plugged = online == 1 else: status = cat(root + "/status", fallback="").strip().lower() if status == "discharging": power_plugged = False elif status in {"charging", "full"}: power_plugged = True # Seconds left. # Note to self: we may also calculate the charging ETA as per: # https://github.com/thialfihar/dotfiles/blob/ # 013937745fd9050c30146290e8f963d65c0179e6/bin/battery.py#L55 if power_plugged: secsleft = _common.POWER_TIME_UNLIMITED elif energy_now is not None and power_now is not None: try: secsleft = int(energy_now / power_now * 3600) except ZeroDivisionError: secsleft = _common.POWER_TIME_UNKNOWN elif time_to_empty is not None: secsleft = int(time_to_empty * 60) if secsleft < 0: secsleft = _common.POWER_TIME_UNKNOWN else: secsleft = _common.POWER_TIME_UNKNOWN return _common.sbattery(percent, secsleft, power_plugged) # ===================================================================== # --- other system functions # ===================================================================== def users(): """Return currently connected users as a list of namedtuples.""" retlist = [] rawlist = cext.users() for item in rawlist: user, tty, hostname, tstamp, pid = item nt = _common.suser(user, tty or None, hostname, tstamp, pid) retlist.append(nt) return retlist def boot_time(): """Return the system boot time expressed in seconds since the epoch.""" global BOOT_TIME path = f"{get_procfs_path()}/stat" with open_binary(path) as f: for line in f: if line.startswith(b'btime'): ret = float(line.strip().split()[1]) BOOT_TIME = ret return ret msg = f"line 'btime' not found in {path}" raise RuntimeError(msg) # ===================================================================== # --- processes # ===================================================================== def pids(): """Returns a list of PIDs currently running on the system.""" path = get_procfs_path().encode(ENCODING) return [int(x) for x in os.listdir(path) if x.isdigit()] def pid_exists(pid): """Check for the existence of a unix PID. Linux TIDs are not supported (always return False). """ if not _psposix.pid_exists(pid): return False else: # Linux's apparently does not distinguish between PIDs and TIDs # (thread IDs). # listdir("/proc") won't show any TID (only PIDs) but # os.stat("/proc/{tid}") will succeed if {tid} exists. # os.kill() can also be passed a TID. This is quite confusing. # In here we want to enforce this distinction and support PIDs # only, see: # https://github.com/giampaolo/psutil/issues/687 try: # Note: already checked that this is faster than using a # regular expr. Also (a lot) faster than doing # 'return pid in pids()' path = f"{get_procfs_path()}/{pid}/status" with open_binary(path) as f: for line in f: if line.startswith(b"Tgid:"): tgid = int(line.split()[1]) # If tgid and pid are the same then we're # dealing with a process PID. return tgid == pid msg = f"'Tgid' line not found in {path}" raise ValueError(msg) except (OSError, ValueError): return pid in pids() def ppid_map(): """Obtain a {pid: ppid, ...} dict for all running processes in one shot. Used to speed up Process.children(). """ ret = {} procfs_path = get_procfs_path() for pid in pids(): try: with open_binary(f"{procfs_path}/{pid}/stat") as f: data = f.read() except (FileNotFoundError, ProcessLookupError): # Note: we should be able to access /stat for all processes # aka it's unlikely we'll bump into EPERM, which is good. pass else: rpar = data.rfind(b')') dset = data[rpar + 2 :].split() ppid = int(dset[1]) ret[pid] = ppid return ret def wrap_exceptions(fun): """Decorator which translates bare OSError and OSError exceptions into NoSuchProcess and AccessDenied. """ @functools.wraps(fun) def wrapper(self, *args, **kwargs): pid, name = self.pid, self._name try: return fun(self, *args, **kwargs) except PermissionError as err: raise AccessDenied(pid, name) from err except ProcessLookupError as err: self._raise_if_zombie() raise NoSuchProcess(pid, name) from err except FileNotFoundError as err: self._raise_if_zombie() # /proc/PID directory may still exist, but the files within # it may not, indicating the process is gone, see: # https://github.com/giampaolo/psutil/issues/2418 if not os.path.exists(f"{self._procfs_path}/{pid}/stat"): raise NoSuchProcess(pid, name) from err raise return wrapper class Process: """Linux process implementation.""" __slots__ = ["_cache", "_name", "_ppid", "_procfs_path", "pid"] def __init__(self, pid): self.pid = pid self._name = None self._ppid = None self._procfs_path = get_procfs_path() def _is_zombie(self): # Note: most of the times Linux is able to return info about the # process even if it's a zombie, and /proc/{pid} will exist. # There are some exceptions though, like exe(), cmdline() and # memory_maps(). In these cases /proc/{pid}/{file} exists but # it's empty. Instead of returning a "null" value we'll raise an # exception. try: data = bcat(f"{self._procfs_path}/{self.pid}/stat") except OSError: return False else: rpar = data.rfind(b')') status = data[rpar + 2 : rpar + 3] return status == b"Z" def _raise_if_zombie(self): if self._is_zombie(): raise ZombieProcess(self.pid, self._name, self._ppid) def _raise_if_not_alive(self): """Raise NSP if the process disappeared on us.""" # For those C function who do not raise NSP, possibly returning # incorrect or incomplete result. os.stat(f"{self._procfs_path}/{self.pid}") @wrap_exceptions @memoize_when_activated def _parse_stat_file(self): """Parse /proc/{pid}/stat file and return a dict with various process info. Using "man proc" as a reference: where "man proc" refers to position N always subtract 3 (e.g ppid position 4 in 'man proc' == position 1 in here). The return value is cached in case oneshot() ctx manager is in use. """ data = bcat(f"{self._procfs_path}/{self.pid}/stat") # Process name is between parentheses. It can contain spaces and # other parentheses. This is taken into account by looking for # the first occurrence of "(" and the last occurrence of ")". rpar = data.rfind(b')') name = data[data.find(b'(') + 1 : rpar] fields = data[rpar + 2 :].split() ret = {} ret['name'] = name ret['status'] = fields[0] ret['ppid'] = fields[1] ret['ttynr'] = fields[4] ret['utime'] = fields[11] ret['stime'] = fields[12] ret['children_utime'] = fields[13] ret['children_stime'] = fields[14] ret['create_time'] = fields[19] ret['cpu_num'] = fields[36] try: ret['blkio_ticks'] = fields[39] # aka 'delayacct_blkio_ticks' except IndexError: # https://github.com/giampaolo/psutil/issues/2455 debug("can't get blkio_ticks, set iowait to 0") ret['blkio_ticks'] = 0 return ret @wrap_exceptions @memoize_when_activated def _read_status_file(self): """Read /proc/{pid}/stat file and return its content. The return value is cached in case oneshot() ctx manager is in use. """ with open_binary(f"{self._procfs_path}/{self.pid}/status") as f: return f.read() @wrap_exceptions @memoize_when_activated def _read_smaps_file(self): with open_binary(f"{self._procfs_path}/{self.pid}/smaps") as f: return f.read().strip() def oneshot_enter(self): self._parse_stat_file.cache_activate(self) self._read_status_file.cache_activate(self) self._read_smaps_file.cache_activate(self) def oneshot_exit(self): self._parse_stat_file.cache_deactivate(self) self._read_status_file.cache_deactivate(self) self._read_smaps_file.cache_deactivate(self) @wrap_exceptions def name(self): # XXX - gets changed later and probably needs refactoring return decode(self._parse_stat_file()['name']) @wrap_exceptions def exe(self): try: return readlink(f"{self._procfs_path}/{self.pid}/exe") except (FileNotFoundError, ProcessLookupError): self._raise_if_zombie() # no such file error; might be raised also if the # path actually exists for system processes with # low pids (about 0-20) if os.path.lexists(f"{self._procfs_path}/{self.pid}"): return "" raise @wrap_exceptions def cmdline(self): with open_text(f"{self._procfs_path}/{self.pid}/cmdline") as f: data = f.read() if not data: # may happen in case of zombie process self._raise_if_zombie() return [] # 'man proc' states that args are separated by null bytes '\0' # and last char is supposed to be a null byte. Nevertheless # some processes may change their cmdline after being started # (via setproctitle() or similar), they are usually not # compliant with this rule and use spaces instead. Google # Chrome process is an example. See: # https://github.com/giampaolo/psutil/issues/1179 sep = '\x00' if data.endswith('\x00') else ' ' if data.endswith(sep): data = data[:-1] cmdline = data.split(sep) # Sometimes last char is a null byte '\0' but the args are # separated by spaces, see: https://github.com/giampaolo/psutil/ # issues/1179#issuecomment-552984549 if sep == '\x00' and len(cmdline) == 1 and ' ' in data: cmdline = data.split(' ') return cmdline @wrap_exceptions def environ(self): with open_text(f"{self._procfs_path}/{self.pid}/environ") as f: data = f.read() return parse_environ_block(data) @wrap_exceptions def terminal(self): tty_nr = int(self._parse_stat_file()['ttynr']) tmap = _psposix.get_terminal_map() try: return tmap[tty_nr] except KeyError: return None # May not be available on old kernels. if os.path.exists(f"/proc/{os.getpid()}/io"): @wrap_exceptions def io_counters(self): fname = f"{self._procfs_path}/{self.pid}/io" fields = {} with open_binary(fname) as f: for line in f: # https://github.com/giampaolo/psutil/issues/1004 line = line.strip() if line: try: name, value = line.split(b': ') except ValueError: # https://github.com/giampaolo/psutil/issues/1004 continue else: fields[name] = int(value) if not fields: msg = f"{fname} file was empty" raise RuntimeError(msg) try: return pio( fields[b'syscr'], # read syscalls fields[b'syscw'], # write syscalls fields[b'read_bytes'], # read bytes fields[b'write_bytes'], # write bytes fields[b'rchar'], # read chars fields[b'wchar'], # write chars ) except KeyError as err: msg = ( f"{err.args[0]!r} field was not found in {fname}; found" f" fields are {fields!r}" ) raise ValueError(msg) from None @wrap_exceptions def cpu_times(self): values = self._parse_stat_file() utime = float(values['utime']) / CLOCK_TICKS stime = float(values['stime']) / CLOCK_TICKS children_utime = float(values['children_utime']) / CLOCK_TICKS children_stime = float(values['children_stime']) / CLOCK_TICKS iowait = float(values['blkio_ticks']) / CLOCK_TICKS return pcputimes(utime, stime, children_utime, children_stime, iowait) @wrap_exceptions def cpu_num(self): """What CPU the process is on.""" return int(self._parse_stat_file()['cpu_num']) @wrap_exceptions def wait(self, timeout=None): return _psposix.wait_pid(self.pid, timeout, self._name) @wrap_exceptions def create_time(self): ctime = float(self._parse_stat_file()['create_time']) # According to documentation, starttime is in field 21 and the # unit is jiffies (clock ticks). # We first divide it for clock ticks and then add uptime returning # seconds since the epoch. # Also use cached value if available. bt = BOOT_TIME or boot_time() return (ctime / CLOCK_TICKS) + bt @wrap_exceptions def memory_info(self): # ============================================================ # | FIELD | DESCRIPTION | AKA | TOP | # ============================================================ # | rss | resident set size | | RES | # | vms | total program size | size | VIRT | # | shared | shared pages (from shared mappings) | | SHR | # | text | text ('code') | trs | CODE | # | lib | library (unused in Linux 2.6) | lrs | | # | data | data + stack | drs | DATA | # | dirty | dirty pages (unused in Linux 2.6) | dt | | # ============================================================ with open_binary(f"{self._procfs_path}/{self.pid}/statm") as f: vms, rss, shared, text, lib, data, dirty = ( int(x) * PAGESIZE for x in f.readline().split()[:7] ) return pmem(rss, vms, shared, text, lib, data, dirty) if HAS_PROC_SMAPS_ROLLUP or HAS_PROC_SMAPS: def _parse_smaps_rollup(self): # /proc/pid/smaps_rollup was added to Linux in 2017. Faster # than /proc/pid/smaps. It reports higher PSS than */smaps # (from 1k up to 200k higher; tested against all processes). # IMPORTANT: /proc/pid/smaps_rollup is weird, because it # raises ESRCH / ENOENT for many PIDs, even if they're alive # (also as root). In that case we'll use /proc/pid/smaps as # fallback, which is slower but has a +50% success rate # compared to /proc/pid/smaps_rollup. uss = pss = swap = 0 with open_binary( f"{self._procfs_path}/{self.pid}/smaps_rollup" ) as f: for line in f: if line.startswith(b"Private_"): # Private_Clean, Private_Dirty, Private_Hugetlb uss += int(line.split()[1]) * 1024 elif line.startswith(b"Pss:"): pss = int(line.split()[1]) * 1024 elif line.startswith(b"Swap:"): swap = int(line.split()[1]) * 1024 return (uss, pss, swap) @wrap_exceptions def _parse_smaps( self, # Gets Private_Clean, Private_Dirty, Private_Hugetlb. _private_re=re.compile(br"\nPrivate.*:\s+(\d+)"), _pss_re=re.compile(br"\nPss\:\s+(\d+)"), _swap_re=re.compile(br"\nSwap\:\s+(\d+)"), ): # /proc/pid/smaps does not exist on kernels < 2.6.14 or if # CONFIG_MMU kernel configuration option is not enabled. # Note: using 3 regexes is faster than reading the file # line by line. # # You might be tempted to calculate USS by subtracting # the "shared" value from the "resident" value in # /proc/<pid>/statm. But at least on Linux, statm's "shared" # value actually counts pages backed by files, which has # little to do with whether the pages are actually shared. # /proc/self/smaps on the other hand appears to give us the # correct information. smaps_data = self._read_smaps_file() # Note: smaps file can be empty for certain processes. # The code below will not crash though and will result to 0. uss = sum(map(int, _private_re.findall(smaps_data))) * 1024 pss = sum(map(int, _pss_re.findall(smaps_data))) * 1024 swap = sum(map(int, _swap_re.findall(smaps_data))) * 1024 return (uss, pss, swap) @wrap_exceptions def memory_full_info(self): if HAS_PROC_SMAPS_ROLLUP: # faster try: uss, pss, swap = self._parse_smaps_rollup() except (ProcessLookupError, FileNotFoundError): uss, pss, swap = self._parse_smaps() else: uss, pss, swap = self._parse_smaps() basic_mem = self.memory_info() return pfullmem(*basic_mem + (uss, pss, swap)) else: memory_full_info = memory_info if HAS_PROC_SMAPS: @wrap_exceptions def memory_maps(self): """Return process's mapped memory regions as a list of named tuples. Fields are explained in 'man proc'; here is an updated (Apr 2012) version: https://git.kernel.org/pub/scm/linux/kernel/git/torvalds/linux.git/tree/Documentation/filesystems/proc.txt?id=b76437579d1344b612cf1851ae610c636cec7db0. /proc/{PID}/smaps does not exist on kernels < 2.6.14 or if CONFIG_MMU kernel configuration option is not enabled. """ def get_blocks(lines, current_block): data = {} for line in lines: fields = line.split(None, 5) if not fields[0].endswith(b':'): # new block section yield (current_block.pop(), data) current_block.append(line) else: try: data[fields[0]] = int(fields[1]) * 1024 except ValueError: if fields[0].startswith(b'VmFlags:'): # see issue #369 continue msg = f"don't know how to interpret line {line!r}" raise ValueError(msg) from None yield (current_block.pop(), data) data = self._read_smaps_file() # Note: smaps file can be empty for certain processes or for # zombies. if not data: self._raise_if_zombie() return [] lines = data.split(b'\n') ls = [] first_line = lines.pop(0) current_block = [first_line] for header, data in get_blocks(lines, current_block): hfields = header.split(None, 5) try: addr, perms, _offset, _dev, _inode, path = hfields except ValueError: addr, perms, _offset, _dev, _inode, path = hfields + [''] if not path: path = '[anon]' else: path = decode(path) path = path.strip() if path.endswith(' (deleted)') and not path_exists_strict( path ): path = path[:-10] item = ( decode(addr), decode(perms), path, data.get(b'Rss:', 0), data.get(b'Size:', 0), data.get(b'Pss:', 0), data.get(b'Shared_Clean:', 0), data.get(b'Shared_Dirty:', 0), data.get(b'Private_Clean:', 0), data.get(b'Private_Dirty:', 0), data.get(b'Referenced:', 0), data.get(b'Anonymous:', 0), data.get(b'Swap:', 0), ) ls.append(item) return ls @wrap_exceptions def cwd(self): return readlink(f"{self._procfs_path}/{self.pid}/cwd") @wrap_exceptions def num_ctx_switches( self, _ctxsw_re=re.compile(br'ctxt_switches:\t(\d+)') ): data = self._read_status_file() ctxsw = _ctxsw_re.findall(data) if not ctxsw: msg = ( "'voluntary_ctxt_switches' and" " 'nonvoluntary_ctxt_switches'lines were not found in" f" {self._procfs_path}/{self.pid}/status; the kernel is" " probably older than 2.6.23" ) raise NotImplementedError(msg) return _common.pctxsw(int(ctxsw[0]), int(ctxsw[1])) @wrap_exceptions def num_threads(self, _num_threads_re=re.compile(br'Threads:\t(\d+)')): # Using a re is faster than iterating over file line by line. data = self._read_status_file() return int(_num_threads_re.findall(data)[0]) @wrap_exceptions def threads(self): thread_ids = os.listdir(f"{self._procfs_path}/{self.pid}/task") thread_ids.sort() retlist = [] hit_enoent = False for thread_id in thread_ids: fname = f"{self._procfs_path}/{self.pid}/task/{thread_id}/stat" try: with open_binary(fname) as f: st = f.read().strip() except (FileNotFoundError, ProcessLookupError): # no such file or directory or no such process; # it means thread disappeared on us hit_enoent = True continue # ignore the first two values ("pid (exe)") st = st[st.find(b')') + 2 :] values = st.split(b' ') utime = float(values[11]) / CLOCK_TICKS stime = float(values[12]) / CLOCK_TICKS ntuple = _common.pthread(int(thread_id), utime, stime) retlist.append(ntuple) if hit_enoent: self._raise_if_not_alive() return retlist @wrap_exceptions def nice_get(self): # with open_text(f"{self._procfs_path}/{self.pid}/stat") as f: # data = f.read() # return int(data.split()[18]) # Use C implementation return cext_posix.getpriority(self.pid) @wrap_exceptions def nice_set(self, value): return cext_posix.setpriority(self.pid, value) # starting from CentOS 6. if HAS_CPU_AFFINITY: @wrap_exceptions def cpu_affinity_get(self): return cext.proc_cpu_affinity_get(self.pid) def _get_eligible_cpus( self, _re=re.compile(br"Cpus_allowed_list:\t(\d+)-(\d+)") ): # See: https://github.com/giampaolo/psutil/issues/956 data = self._read_status_file() match = _re.findall(data) if match: return list(range(int(match[0][0]), int(match[0][1]) + 1)) else: return list(range(len(per_cpu_times()))) @wrap_exceptions def cpu_affinity_set(self, cpus): try: cext.proc_cpu_affinity_set(self.pid, cpus) except (OSError, ValueError) as err: if isinstance(err, ValueError) or err.errno == errno.EINVAL: eligible_cpus = self._get_eligible_cpus() all_cpus = tuple(range(len(per_cpu_times()))) for cpu in cpus: if cpu not in all_cpus: msg = ( f"invalid CPU {cpu!r}; choose between" f" {eligible_cpus!r}" ) raise ValueError(msg) from None if cpu not in eligible_cpus: msg = ( f"CPU number {cpu} is not eligible; choose" f" between {eligible_cpus}" ) raise ValueError(msg) from err raise # only starting from kernel 2.6.13 if HAS_PROC_IO_PRIORITY: @wrap_exceptions def ionice_get(self): ioclass, value = cext.proc_ioprio_get(self.pid) ioclass = IOPriority(ioclass) return _common.pionice(ioclass, value) @wrap_exceptions def ionice_set(self, ioclass, value): if value is None: value = 0 if value and ioclass in { IOPriority.IOPRIO_CLASS_IDLE, IOPriority.IOPRIO_CLASS_NONE, }: msg = f"{ioclass!r} ioclass accepts no value" raise ValueError(msg) if value < 0 or value > 7: msg = "value not in 0-7 range" raise ValueError(msg) return cext.proc_ioprio_set(self.pid, ioclass, value) if hasattr(resource, "prlimit"): @wrap_exceptions def rlimit(self, resource_, limits=None): # If pid is 0 prlimit() applies to the calling process and # we don't want that. We should never get here though as # PID 0 is not supported on Linux. if self.pid == 0: msg = "can't use prlimit() against PID 0 process" raise ValueError(msg) try: if limits is None: # get return resource.prlimit(self.pid, resource_) else: # set if len(limits) != 2: msg = ( "second argument must be a (soft, hard) " f"tuple, got {limits!r}" ) raise ValueError(msg) resource.prlimit(self.pid, resource_, limits) except OSError as err: if err.errno == errno.ENOSYS: # I saw this happening on Travis: # https://travis-ci.org/giampaolo/psutil/jobs/51368273 self._raise_if_zombie() raise @wrap_exceptions def status(self): letter = self._parse_stat_file()['status'] letter = letter.decode() # XXX is '?' legit? (we're not supposed to return it anyway) return PROC_STATUSES.get(letter, '?') @wrap_exceptions def open_files(self): retlist = [] files = os.listdir(f"{self._procfs_path}/{self.pid}/fd") hit_enoent = False for fd in files: file = f"{self._procfs_path}/{self.pid}/fd/{fd}" try: path = readlink(file) except (FileNotFoundError, ProcessLookupError): # ENOENT == file which is gone in the meantime hit_enoent = True continue except OSError as err: if err.errno == errno.EINVAL: # not a link continue if err.errno == errno.ENAMETOOLONG: # file name too long debug(err) continue raise else: # If path is not an absolute there's no way to tell # whether it's a regular file or not, so we skip it. # A regular file is always supposed to be have an # absolute path though. if path.startswith('/') and isfile_strict(path): # Get file position and flags. file = f"{self._procfs_path}/{self.pid}/fdinfo/{fd}" try: with open_binary(file) as f: pos = int(f.readline().split()[1]) flags = int(f.readline().split()[1], 8) except (FileNotFoundError, ProcessLookupError): # fd gone in the meantime; process may # still be alive hit_enoent = True else: mode = file_flags_to_mode(flags) ntuple = popenfile( path, int(fd), int(pos), mode, flags ) retlist.append(ntuple) if hit_enoent: self._raise_if_not_alive() return retlist @wrap_exceptions def net_connections(self, kind='inet'): ret = _net_connections.retrieve(kind, self.pid) self._raise_if_not_alive() return ret @wrap_exceptions def num_fds(self): return len(os.listdir(f"{self._procfs_path}/{self.pid}/fd")) @wrap_exceptions def ppid(self): return int(self._parse_stat_file()['ppid']) @wrap_exceptions def uids(self, _uids_re=re.compile(br'Uid:\t(\d+)\t(\d+)\t(\d+)')): data = self._read_status_file() real, effective, saved = _uids_re.findall(data)[0] return _common.puids(int(real), int(effective), int(saved)) @wrap_exceptions def gids(self, _gids_re=re.compile(br'Gid:\t(\d+)\t(\d+)\t(\d+)')): data = self._read_status_file() real, effective, saved = _gids_re.findall(data)[0] return _common.pgids(int(real), int(effective), int(saved)) __pycache__/_psposix.cpython-310.pyc 0000644 00000007730 15030211765 0013330 0 ustar 00 o �h� � @ s� d Z ddlZddlZddlZddlZddlZddlmZ ddlmZ ddlm Z ddlm Z ddlm Z er<dd l m Z g d �Zd d � Ze�d dd� ejD ��Zdd� Zddejeedej�eejefdd�Zdd� Ze dd� �ZdS )z%Routines common to all posix systems.� N� )�MACOS��TimeoutExpired)�memoize)� sdiskusage)� usage_percent)� _psutil_osx)� pid_exists�wait_pid� disk_usage�get_terminal_mapc C sF | dkrdS z t �| d� W dS ty Y dS ty" Y dS w )z6Check whether pid exists in the current process table.r TF)�os�kill�ProcessLookupError�PermissionError)�pid� r �B/usr/local/CyberCP/lib/python3.10/site-packages/psutil/_psposix.pyr s � �r � Negsignalc C s i | ]}|j |j �qS r )�name�value)�.0�xr r r � <dictcomp>2 s r c C s" zt | �W S ty | Y S w )z+Convert a negative signal value to an enum.)r � ValueError)�numr r r �negsig_to_enum6 s �r � monotonicc s �dkr d}t |��d} d} �dur | tjO } �� � �� ������fdd�} z t��| �\} } W n" ty@ | | �} Y n? tyU |��rR| | �} |��sJY dS w | dkr_| | �} q(t�| �rit�| �S t�| �rvt t� | � �S d| ��}t |��q)) a� Wait for a process PID to terminate. If the process terminated normally by calling exit(3) or _exit(2), or by returning from main(), the return value is the positive integer passed to *exit(). If it was terminated by a signal it returns the negated value of the signal which caused the termination (e.g. -SIGTERM). If PID is not a children of os.getpid() (current process) just wait until the process disappears and return None. If PID does not exist at all return None immediately. If *timeout* != None and process is still alive raise TimeoutExpired. timeout=0 is also possible (either return immediately or raise). r zcan't wait for PID 0g-C ��6?Nc s6 �d ur�� �krt ���d���| � � | d d�S )N)r r � g{�G�z�?r )�interval��_min�_sleep�_timerr � proc_name�stop_at�timeoutr r �sleepc s zwait_pid.<locals>.sleepTz unknown process exit status ) r r �WNOHANG�waitpid�InterruptedError�ChildProcessError� WIFEXITED� WEXITSTATUS� WIFSIGNALEDr �WTERMSIG)r r' r% �_waitpidr$ r" r# � _pid_exists�msgr �flagsr( �retpid�statusr r! r r > s< �� �r c C sl t �| �}|j|j }|j|j }|j|j }|| }tr#t�| |�}|| }t ||dd�}t ||||d�S )a. Return disk usage associated with path. Note: UNIX usually reserves 5% disk space which is not accessible by user. In this function "total" and "used" values reflect the total and used disk space whereas "free" and "percent" represent the "free" and "used percent" user disk space. r )�round_)�total�used�free�percent) r �statvfs�f_blocks�f_frsize�f_bfree�f_bavailr r �disk_usage_usedr r )�path�str8 � avail_to_root� avail_to_userr9 � total_user�usage_percent_userr r r r � s �r c C s\ i } t � d�t � d� }|D ] }|| vsJ |��z || t�|�j< W q ty+ Y qw | S )zNGet a map of device-id -> path as a dict. Used by Process.terminal(). z /dev/tty*z /dev/pts/*)�globr �stat�st_rdev�FileNotFoundError)�ret�lsr r r r r � s �r )�__doc__�enumrH r �signal�time�_commonr r r r r � r �__all__r �IntEnum�Signalsr r r* �getattr�minr( r r r r r r r �<module> s< � �`$ __pycache__/_pssunos.cpython-310.pyc 0000644 00000042657 15030211765 0013344 0 ustar 00 o �hXa � @ s d Z ddlZddlZddlZddlZddlZddlZddlmZ ddlm Z ddl m Z ddl m Z ddl m Z dd l mZ dd l mZ dd l mZ dd l mZ dd l mZ ddl mZ ddl mZ ddl mZ ddl mZ ddl mZ ddl mZ ddl mZ ddl m Z g d�Z e� � Z ej Z ej!dkZ"dZ#dZ$ej%e j&ej'e j(ej)e j*ej+e j,ej-e j.ej/e j(ej0e j1iZ2ej3e j4ej5e j6ej7e j8ej9e j:ej;e j<ej=e j>ej?e j@ejAe jBejCe jDejEe jFejGe jHejIe jJejKe#ejLe$iZMeNddddd d d d d d!d"d#d$� ZOed%g d&��ZPed'g d(��ZQed)g d*��ZRed+d,d-g�ZSeSZTed.g d/��ZUed0d1d2�VeUjW� �ZXd3d4� ZYd5d6� ZZd7d8� Z[d9d:� Z\d;d<� Z]d=d>� Z^d?d@� Z_ej`Z`e jaZadUdBdC�ZbejcZcejdZddVdEdF�ZedGdH� ZfdIdJ� ZgdKdL� ZhdMdN� ZidOdP� ZjdQdR� ZkG dSdT� dT�ZldS )Wz'Sun OS Solaris platform implementation.� N)� namedtuple)�AF_INET� )�_common)�_psposix)� _psutil_posix)� _psutil_sunos)�AF_INET6)�ENCODING)� AccessDenied)� NoSuchProcess)� ZombieProcess)�debug)�get_procfs_path)� isfile_strict)�memoize_when_activated)�sockfam_to_enum)�socktype_to_enum)� usage_percent)� CONN_IDLE� CONN_BOUND� PROCFS_PATHl �IDLE�BOUND� � � � � � � � � � ) �ppid�rss�vms� create_time�nice� num_threads�status�ttynr�uid�euid�gid�egid� scputimes)�user�system�idle�iowait� pcputimes)r1 r2 � children_user�children_system�svmem)�total� available�percent�used�free�pmemr% r& � pmmap_grouped)�pathr% � anonymous�locked� pmmap_extz addr perms � c C sF t �d�t } t �d�t }}| | }t|| dd�}t| ||||�S )z Report virtual memory metrics.� SC_PHYS_PAGES�SC_AVPHYS_PAGESr ��round_)�os�sysconf� PAGE_SIZEr r8 )r9 r= �availr< r; � rM �B/usr/local/CyberCP/lib/python3.10/site-packages/psutil/_pssunos.py�virtual_memory{ s rO c C s t �� \} }tjddtjd � �ddgtjd�}|�� \}}|�t j j �}|j dkr5d|j � d �}t |��|�� �d �d d � }|sHd }t |��d }}|D ]"} | �� } | dd� \} } |tt| �d �7 }|tt| �d �7 }qN|| } t| |d d�} t�|| || | t |t �S )zReport swap memory metrics.z /usr/bin/envzPATH=/usr/sbin:/sbin:�PATH�swapz-l)�stdoutr z'swap -l' failed (retcode=�)� r Nz no swap device(s) configuredr r i rG )�cext�swap_mem� subprocess�PopenrI �environ�PIPE� communicate�decode�sysrR �encoding� returncode� RuntimeError�strip�split�intr r �sswaprK )�sin�sout�prR �_�msg�linesr9 r= �line�t�fr< r; rM rM rN � swap_memory� s: �� �rn c C s t �� } tdd� t| � D �� S )z.Return system-wide CPU times as a named tuple.c S s g | ]}t |��qS rM )�sum��.0�xrM rM rN � <listcomp>� � z cpu_times.<locals>.<listcomp>)rU � per_cpu_timesr0 �zip��retrM rM rN � cpu_times� s ry c C s t �� } dd� | D �S )z6Return system per-CPU times as a list of named tuples.c S s g | ]}t |� �qS rM )r0 rp rM rM rN rs � rt z!per_cpu_times.<locals>.<listcomp>)rU ru rw rM rM rN ru � s ru c C s" zt �d�W S ty Y dS w )z0Return the number of logical CPUs in the system.�SC_NPROCESSORS_ONLNN)rI rJ � ValueErrorrM rM rM rN �cpu_count_logical� s �r| c C � t �� S )z-Return the number of CPU cores in the system.)rU �cpu_count_coresrM rM rM rN r~ � � r~ c C s$ t �� \} }}}d}t�| |||�S )z*Return various CPU stats as a named tuple.r )rU � cpu_statsr � scpustats)� ctx_switches� interrupts�syscalls�_traps�soft_interruptsrM rM rN r� � s �r� Fc C s� g }t �� }|D ]C}|\}}}}|dkrd}| s>z t|�js W qW n ty= } ztd|�d|� �� W Y d}~qd}~ww t�||||�} |�| � q|S )z Return system disk partitions.�none� z skipping z: N) rU �disk_partitions� disk_usager9 �OSErrorr r � sdiskpart�append) �all�retlist� partitions� partition�device� mountpoint�fstype�opts�err�ntuplerM rM rN r� � s& � �� r� ���c C s� t j| \}}t�|�}t� }|D ]T}|\}}} } } } } ||vr!q| |vr&q|tthv r:| r3t j| � } | r:t j| � } t| } t |�}t | �} |dkrVt � ||| | | | | �}n t � ||| | | | �}|� |� qt|�S )z�Return socket connections. If pid == -1 return system-wide connections (as opposed to connections opened by one process only). Only INET sockets are returned (UNIX are not). r� )r � conn_tmaprU �net_connections�setr r �addr� TCP_STATUSESr r �sconn�pconn�add�list)�kind�_pid�families�types�rawlistrx �item�fd�fam�type_�laddr�raddrr* �pid�ntrM rM rN r� s, r� c C sT t �� } | �� D ] \}}|\}}}}ttd�r t�|�}t�||||d�| |< q| S )z)Get NIC stats (isup, duplex, speed, mtu).� NicDuplexr� )rU � net_if_stats�items�hasattrr r� � snicstats)rx �namer� �isup�duplex�speed�mturM rM rN r� " s r� c C r} )z:The system boot time expressed in seconds since the epoch.)rU � boot_timerM rM rM rN r� 2 r r� c C s\ g } t �� }d}|D ]!}|\}}}}}} |sq ||v r d}t�||||| �} | �| � q | S )z:Return currently connected users as a list of namedtuples.)z:0.0z:0� localhost)rU �usersr �suserr� ) r� r� r� r� r1 �tty�hostname�tstamp� user_processr� r� rM rM rN r� 7 s r� c C s t � �t�} dd� t�| �D �S )z7Returns a list of PIDs currently running on the system.c S s g | ] }|� � rt|��qS rM )�isdigitrc rp rM rM rN rs R s zpids.<locals>.<listcomp>)r �encoder rI �listdir)r@ rM rM rN �pidsO s r� c C s t �| �S )z&Check for the existence of a unix pid.)r � pid_exists)r� rM rM rN r� U s r� c s t �� �� fdd��}|S )z�Call callable into a try/except clause and translate ENOENT, EACCES and EPERM in NoSuchProcess or AccessDenied exceptions. c s� | j | j| j}}}z � | g|�R i |��W S ttfy6 } zt|�s+t||�|�t|||�|�d }~w tyG } zt ||�|�d }~w t yc } z|dkr^dt � v r]t ||�|�� � d }~ww �Nr ) r� �_ppid�_name�FileNotFoundError�ProcessLookupErrorr� r r �PermissionErrorr r� r� )�self�args�kwargsr� r$ r� r� ��funrM rN �wrapper_ s& � � ��z wrap_exceptions.<locals>.wrapper)� functools�wraps)r� r� rM r� rN �wrap_exceptionsZ s r� c @ s� e Zd ZdZg d�Zdd� Zdd� Zdd� Zd d � Ze e d d � ��Z e e d d� ��Z e e dd� ��Z e dd� �Ze dd� �Ze dd� �Ze dd� �Ze dd� �Ze dd � �Ze d d � �Ze d d � �Ze d!d"� �Ze d#d$� �Ze d%d&� �Ze d'd(� �Ze d)d*� �Ze d+d,� �Ze d-d.� �Z e d/d0� �Z e Z e d1d2� �Z e d3d4� �Z e d5d6� �Z!d7d8� Z"e dHd:d;��Z#e$d<d=�Z%e$d<d>�Z&e d?d@� �Z'e dAdB� �Z(e dCdD� �Z)e dIdFdG��Z*dES )J�Processz1Wrapper class around underlying C implementation.)�_cacher� r� � _procfs_pathr� c C s || _ d | _d | _t� | _d S �N)r� r� r� r r� )r� r� rM rM rN �__init__| s zProcess.__init__c C s t �| j� d| j� �� dS )z+Raise NSP if the process disappeared on us.�/N)rI �statr� r� �r� rM rM rN � _assert_alive� s zProcess._assert_alivec C �( | j �| � | j�| � | j�| � d S r� )�_proc_name_and_args�cache_activate�_proc_basic_info� _proc_credr� rM rM rN � oneshot_enter� � zProcess.oneshot_enterc C r� r� )r� �cache_deactivater� r� r� rM rM rN � oneshot_exit� r� zProcess.oneshot_exitc C � t �| j| j�S r� )rU �proc_name_and_argsr� r� r� rM rM rN r� � � zProcess._proc_name_and_argsc C sX | j dkrtj�| j� d| j � d��st| j ��t�| j | j�}t|�tt �ks*J �|S )Nr r� z/psinfo) r� rI r@ �existsr� r rU �proc_basic_info�len� proc_info_map)r� rx rM rM rN r� � s � zProcess._proc_basic_infoc C r� r� )rU � proc_credr� r� r� rM rM rN r� � r� zProcess._proc_credc C s | � � d S r� )r� r� rM rM rN r� � s z Process.namec C s< zt �| j� d| j� d��W S ty Y nw | �� dS )Nr� z /path/a.outr� )rI �readlinkr� r� r� �cmdliner� rM rM rN �exe� s �z Process.exec C s | � � d �d�S )Nr rD )r� rb r� rM rM rN r� � � zProcess.cmdlinec C r� r� )rU � proc_environr� r� r� rM rM rN rY � � zProcess.environc C � | � � td S )Nr' �r� r� r� rM rM rN r' � r� zProcess.create_timec C r� )Nr) r� r� rM rM rN r) � r� zProcess.num_threadsc C r� )Nr( r� r� rM rM rN �nice_get� s zProcess.nice_getc C s& | j dv r t| j | j��t�| j |�S )N> r r )r� r r� � cext_posix� setpriority)r� �valuerM rM rN �nice_set� s zProcess.nice_setc C s | � � td | _| jS )Nr$ )r� r� r� r� rM rM rN r$ � s z Process.ppidc C s^ z | � � \}}}}}}W n ty' | �� td }| �� td }d }Y nw t�|||�S )Nr, r- �r� r r� r� r �puids)r� �real� effective�savedrh rM rM rN �uids� � �z Process.uidsc C s^ z | � � \}}}}}}W n ty' | �� td }| �� td }d }Y nw t�|||�S )Nr. r/ r� )r� rh r� r� r rM rM rN �gids� r z Process.gidsc C s\ z t �| j| j�}W n ty( } z|jtjkr ts d}n� W Y d }~nd }~ww tj |� S )N)� r r r ) rU �proc_cpu_timesr� r� r� �errno� EOVERFLOW� IS_64_BITr r5 )r� �timesr� rM rM rN ry � s ��� zProcess.cpu_timesc C r� r� )rU � proc_cpu_numr� r� r� rM rM rN �cpu_num r� zProcess.cpu_numc C s~ | j }d}t| �� td �}|tjkr5dD ] }zt�|� d| j� d|� ��W S t y4 d}Y qw |r=| � � d S d S )NFr+ )r r r � r� �/path/T) r� r� r� r� rU �PRNODEVrI r� r� r� r� )r� � procfs_path� hit_enoentr� rr rM rM rN �terminal s " � �zProcess.terminalc C sL | j }z t�|� d| j� d��W S ty% t�|� d| j� �� Y dS w )Nr� z /path/cwdr� )r� rI r� r� r� r� )r� r rM rM rN �cwd s �z Process.cwdc C s2 | � � }|td d }|td d }t||�S )Nr% i r& )r� r� r> )r� rx r% r& rM rM rN � memory_info$ s zProcess.memory_infoc C s | � � td }t�|d�S )Nr* �?)r� r� � PROC_STATUSES�get)r� �coderM rM rN r* - s zProcess.statusc C s� | j }g }t�|� d| j� d��}d}|D ]H}t|�}z t�| j||�\}}W n) tyP } z |jtj kr=t s=W Y d }~q|jtj krKd}W Y d }~q� d }~ww t � |||�} |�| � q|rd| �� |S )Nr� z/lwpFT)r� rI r� r� rc rU �query_process_threadr� r r r �ENOENTr �pthreadr� r� ) r� r rx �tidsr �tid�utime�stimer� r� rM rM rN �threads3 s0 � �� zProcess.threadsc C s� g }d}| j }|� d| j� d�}t�|� d| j� d��D ]1}tj�||�}tj�|�rMzt�|�}W n ty= d}Y q w t |�rM|� t � |t |��� q |rT| �� |S )NFr� z/path�/fdT)r� r� rI r� r@ �join�islinkr� r� r r� r � popenfilerc r� )r� r� r r �pathdirr� r@ �filerM rM rN � open_filesU s& ��zProcess.open_filesc c s, � dt |�g}tj|tjtjd�}|�� \}}dd� ||fD �\}}|jdkrLd|�� v r4t| j| j ��d|�� v rAt | j| j ��|�d|� �}t |��|� d �d d � }t |�D ]:\}} | �� } | �d �r�| � d d �d } ||d �� } | dkr~tj} n | dkr�tj} nd} dtj| | dtjfV qYd S )z<Get UNIX sockets used by process by parsing 'pfiles' output.�pfiles)rR �stderrc s s � | ] }|� tjj�V qd S r� )r\ r] rR r^ rp rM rM rN � <genexpr>s s � �z,Process._get_unix_sockets.<locals>.<genexpr>r zpermission deniedzno such processz command error rT r Nzsockname: AF_UNIXrD � SOCK_STREAM� SOCK_DGRAMr� r� )�strrW rX rZ r[ r_ �lowerr r� r� r r` rb � enumerate�lstrip� startswithra �socketr* r+ �AF_UNIXr � CONN_NONE) r� r� �cmdrg rR r( ri rj �irk r@ �typerM rM rN �_get_unix_socketsj s< � � � ��zProcess._get_unix_sockets�inetc C sR t || jd�}|st�| j� d| j� �� |dv r'|�dd� | �| j�D �� |S )N)r� r� > r� �unixc S s g | ]}t j|� �qS rM )r r� )rq �connrM rM rN rs � s ��z+Process.net_connections.<locals>.<listcomp>)r� r� rI r� r� �extendr7 )r� r� rx rM rM rN r� � s �zProcess.net_connections�mmapzpath rss anon lockedz addr perms path rss anon lockedc C s* dd� }| j }g }z t�| j|�}W n ty/ } z|jtjkr*ts*g W Y d }~S � d }~ww d}|D ]X}|\}} } } } } }||| �}| �d�s�zt � |� d| j� d| � ��} W n' ty� } z|jtj kru|� d| j� d| � �} d}n� W Y d }~nd }~ww |� || | | | |f� q4|r�| � � |S )Nc S s0 d� t| �dd � �d�t|�dd � �d��S )Nz{}-{}r �L)�format�hexra )�start�endrM rM rN �toaddr� s (�z#Process.memory_maps.<locals>.toaddrF�[r� r T)r� rU �proc_memory_mapsr� r� r r r r0 rI r� r r� r� )r� rB r r� r� r� r r� r� �addrsize�permr� r% �anonrB rM rM rN � memory_maps� s< �� ��� zProcess.memory_mapsc C s t t�| j� d| j� d���S )Nr� r )r� rI r� r� r� r� rM rM rN �num_fds� s zProcess.num_fdsc C s t jt�| j| j�� S r� )r �pctxswrU �proc_num_ctx_switchesr� r� r� rM rM rN �num_ctx_switches� s �zProcess.num_ctx_switchesNc C s t �| j|| j�S r� )r �wait_pidr� r� )r� �timeoutrM rM rN �wait� r� z Process.wait)r8 r� )+�__name__� __module__� __qualname__�__doc__� __slots__r� r� r� r� r� r r� r� r� r� r� r� rY r' r) r� r� r$ r r ry r r r r �memory_full_infor* r r& r7 r� r �nt_mmap_grouped� nt_mmap_extrH rI rL rO rM rM rM rN r� w s� ! " . r� )F)r� )mrS r r� rI r1 rW r] � collectionsr r r� r r r r� r rU r r r r r r r r r r r r �__extra__all__� getpagesizerK �AF_LINK�maxsizer r r �SSLEEP�STATUS_SLEEPING�SRUN�STATUS_RUNNING�SZOMB� STATUS_ZOMBIE�SSTOP�STATUS_STOPPED�SIDL� STATUS_IDLE�SONPROC�SWAIT�STATUS_WAITINGr �TCPS_ESTABLISHED�CONN_ESTABLISHED� TCPS_SYN_SENT� CONN_SYN_SENT� TCPS_SYN_RCVD� CONN_SYN_RECV�TCPS_FIN_WAIT_1�CONN_FIN_WAIT1�TCPS_FIN_WAIT_2�CONN_FIN_WAIT2�TCPS_TIME_WAIT�CONN_TIME_WAIT� TCPS_CLOSED� CONN_CLOSE�TCPS_CLOSE_WAIT�CONN_CLOSE_WAIT� TCPS_LAST_ACK� CONN_LAST_ACK� TCPS_LISTEN� CONN_LISTEN� TCPS_CLOSING� CONN_CLOSING�PSUTIL_CONN_NONEr3 � TCPS_IDLE� TCPS_BOUNDr� �dictr� r0 r5 r8 r> �pfullmemr? r! �_fieldsrC rO rn ry ru r| r~ r� �disk_io_countersr� r� �net_io_counters� net_if_addrsr� r� r� r� r� r� r� r� rM rM rM rN �<module> s� � ����� . __pycache__/__init__.cpython-310.pyc 0000644 00000174053 15030211765 0013226 0 ustar 00 o �h�R � @ s^ d Z ddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl Z ddl Z zddl Z W n e y? dZ Y nw ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ dd lmZ dd lmZ dd lmZ dd lmZ dd lmZ ddlmZ ddlmZ ddlmZ ddlm Z ddlm Z ddlm Z ddlm Z ddlm Z ddlm!Z! ddlm"Z" ddlm#Z# ddlm$Z$ ddlm%Z% ddlm&Z& dd lm'Z' dd lm(Z( dd lm)Z) dd lm*Z* dd lm+Z+ dd!lm,Z, dd"lm-Z- dd#lm.Z. dd$lm/Z/ dd%lm0Z0 dd&lm1Z1 dd'lm2Z2 dd(lm3Z3 dd)lm4Z4 dd*lm5Z5 dd+lm6Z6 dd,lm7Z7 dd-lm8Z8 dd.lm9Z9 dd/lm:Z: dd0lm;Z; dd1lm<Z< dd2lm=Z= dd3lm>Z> dd4lm?Z@ e �r�d5ZAdd6lmBZC dd7lBmDZD dd8lBmEZE dd9lBmFZF dd:lBmGZG n�e7�r�dd;lmHZC dd<lImJZJ dd=lImKZK dd>lImLZL dd?lImMZM dd@lImNZN ddAlImOZO ddBlHmPZP ddClHmQZQ ddDlHmRZR ddElHmSZS ddFlHmTZT nCe �r�ddGlmUZC n9e�r�ddHlmVZC n/e6�rddIlmWZC ddJlWmXZX ddKlWmYZY d5ZAne�rddLlmZZC d5ZAn dMe j[� dN�Z\e]e\��g dO�Z^e^�_eCj`� eaeCjbdP��rVddQlmcZc ed� ZedZfegec�D ]Zfef�hdR��rRef�i� �rRejecef�eeef< e^�kef� �q9[e[feCjlZldSZmdTZneodUdV� en�pdW�D ��Zqeje dXe j �Zrdasdateu� Zvewen�xdWdY��ejeCjydZd�k�r�d[eCjyjz�d\�Z\e\d]7 Z\eaeCjydZ��r�dW�{e|e}eCjyj~���Ze\d^e� d_en� d`�7 Z\ne\daen� d`�7 Z\ejeCjydbdc�Z�e\dde�� �7 Z\e\de7 Z\e e\��eaeCdf��r�eCj�Z�ndgdh� Z�didj� Z�dkdl� Z�G dmdn� dn�Zbdodp� egeb�D �Z�G dqdr� dreb�Z�dsdt� Z�dudv� Z�i a�e�� Z�d�dwdx�Z�dydz� e�_�d{e�j�_ d�d|d}�Z�d�dd��Z�d�d�d��Z�z e ��� j�e�� iZ�W n e��y= i Z�Y nw z e ��� j�e�d~d��iZ�W n e��yV i Z�Y nw d�d�� Z�d�d�� Z�d�d�� Z�d�d�d��Z�e���� Z�e���� Z�d�d�d��Z�d�d�� Z�eaeCd���r�d�d�d��Z�e^�kd�� eaed���s�eaeCd���r�eaed���r�ej�Z�neCj�Z�e^�kd�� d�d�� Z�d�d�� Z�d�d�� Z�d�d�d��Z�d�d�d��Z�e��e@j�d��e�_�d�e�j�_ d�d�d��Z�e��e@j�d��e�_�d�e�j�_ d�d�d��Z�d�d�� Z�d�d�� Z�eaeCd���r�d�d�d��Z�e^�kd�� eaeCd���rd�d�� Z�e^�kd�� eaeCd���rd�d�� Z�e^�kd�� d�d�� Z�d�d�� Z�e7�r(d�d�� Z�d�d�� Z�d�d�� Z�[>dS )�a; psutil is a cross-platform library for retrieving information on running processes and system utilization (CPU, memory, disks, network, sensors) in Python. Supported platforms: - Linux - Windows - macOS - FreeBSD - OpenBSD - NetBSD - Sun Solaris - AIX Supported Python versions are cPython 3.6+ and PyPy. � N� )�_common)�AIX)�BSD)� CONN_CLOSE)�CONN_CLOSE_WAIT)� CONN_CLOSING)�CONN_ESTABLISHED)�CONN_FIN_WAIT1)�CONN_FIN_WAIT2)� CONN_LAST_ACK)� CONN_LISTEN)� CONN_NONE)� CONN_SYN_RECV)� CONN_SYN_SENT)�CONN_TIME_WAIT)�FREEBSD)�LINUX)�MACOS)�NETBSD)�NIC_DUPLEX_FULL)�NIC_DUPLEX_HALF)�NIC_DUPLEX_UNKNOWN)�OPENBSD)�OSX)�POSIX)�POWER_TIME_UNKNOWN)�POWER_TIME_UNLIMITED)� STATUS_DEAD)�STATUS_DISK_SLEEP)� STATUS_IDLE)� STATUS_LOCKED)� STATUS_PARKED)�STATUS_RUNNING)�STATUS_SLEEPING)�STATUS_STOPPED)�STATUS_TRACING_STOP)�STATUS_WAITING)� STATUS_WAKING)� STATUS_ZOMBIE)�SUNOS)�WINDOWS)� AccessDenied)�Error)� NoSuchProcess)�TimeoutExpired)� ZombieProcess)�debug)�memoize_when_activated)� wrap_numbersz/proc)�_pslinux)�IOPRIO_CLASS_BE)�IOPRIO_CLASS_IDLE)�IOPRIO_CLASS_NONE)�IOPRIO_CLASS_RT)� _pswindows)�ABOVE_NORMAL_PRIORITY_CLASS)�BELOW_NORMAL_PRIORITY_CLASS)�HIGH_PRIORITY_CLASS)�IDLE_PRIORITY_CLASS)�NORMAL_PRIORITY_CLASS)�REALTIME_PRIORITY_CLASS)�CONN_DELETE_TCB)� IOPRIO_HIGH)� IOPRIO_LOW)� IOPRIO_NORMAL)�IOPRIO_VERYLOW)�_psosx)�_psbsd)�_pssunos)� CONN_BOUND)� CONN_IDLE)�_psaixz platform z is not supported)Gr- r. r0 r, r/ � version_info� __version__r# r r$ r r% r& r) r r( r! r' r! r" r r r r r r r r r r r r �AF_LINKr r r r r r r r r r r r r r* r+ r �Process�Popen� pid_exists�pids� process_iter� wait_procs�virtual_memory� swap_memory� cpu_times� cpu_percent�cpu_times_percent� cpu_count� cpu_stats�net_io_counters�net_connections� net_if_addrs� net_if_stats�disk_io_counters�disk_partitions� disk_usage�users� boot_time�rlimit)� _psutil_posix�RLIMzGiampaolo Rodola'z7.0.0c c � � | ]}t |�V qd S �N)�int)�.0�num� rl �B/usr/local/CyberCP/lib/python3.10/site-packages/psutil/__init__.py� <genexpr>� � � rn �.� monotonic� �versionzversion conflict: z C extension z.module was built for another version of psutilz (z instead of �)z (different than �__file__z%the existing psutil install directoryz9; you may try to 'pip uninstall psutil', manually remove z1 or clean the virtual env somehow, then reinstall�ppid_mapc C s@ i } t � D ]}z t�|��� | |< W q ttfy Y qw | S )z{Return a {pid: ppid, ...} dict for all running processes in one shot. Used to speed up Process.children(). )rQ � _psplatformrN �ppidr. r0 )�ret�pidrl rl rm � _ppid_map� s �r{ c C s6 t � � }t|| �}|dk rdnd}tj�| ��|�S )z(Format seconds in a human readable form.i�Q z%H:%M:%Sz%Y-%m-%d %H:%M:%S)�timeri �datetime� fromtimestamp�strftime)�secs�now�secs_ago�fmtrl rl rm � _pprint_secs s r� c C s. t tj�}| |vrd| �d|� �}t|��dS )z+Check net_connections()'s `kind` parameter.zinvalid kind argument z; valid ones are: N)�tupler � conn_tmap� ValueError)�kind�kinds�msgrl rl rm �_check_conn_kind s �r� c @ sv e Zd ZdZd}dd�Zd~dd�Zdd � Zd d � ZeZd d � Z dd� Z dd� Z dd� Z e dd� �Zejdd� �Zddd�Zdd� Zd d � Zd d � Zed d!� �Zd"d#� Zd$d%� Zd&d'� Zd(d)� Zd*d+� Z d,d-� Z d.d/� Z d}d0d1�Z e r�ed2d3� �Z!d4d5� Z"d6d7� Z#d8d9� Z$e%e&j'd:�r�d;d<� Z(e%e&j'd=�r�dd>d?�Z)e%e&j'd@�r�d}dAdB�Z*e%e&j'dC�r�d}dDdE�Z+e%e&j'dF�r�dGdH� Z,e%e&j'dI�r�dJdK� Z-e.r�dLdM� Z/dNdO� Z0dPdQ� Z1e%e&j'dR�r�dSdT� Z2d~dUdV�Z3d}dWdX�Z4edYdZ� �Z5ed[d\� �Z6d]d^� Z7d�d`da�Z8e%e&j'db��rd�ddde�Z9dfdg� Z:d�didj�Z;e<j=dkdl�d�dmdn��Z>e �r dodp� Z?dqdr� Z@dsdt� ZAdudv� ZBdwdx� ZCdydz� ZDd}d{d|�ZEdS )�rN a Represents an OS process with the given PID. If PID is omitted current process PID (os.getpid()) is used. Raise NoSuchProcess if PID does not exist. Note that most of the methods of this class do not make sure that the PID of the process being queried has been reused. That means that you may end up retrieving information for another process. The only exceptions for which process identity is pre-emptively checked and guaranteed are: - parent() - children() - nice() (set) - ionice() (set) - rlimit() (set) - cpu_affinity (set) - suspend() - resume() - send_signal() - terminate() - kill() To prevent this problem for all other methods you can use is_running() before querying the process. Nc C s | � |� d S rh )�_init)�selfrz rl rl rm �__init__< s zProcess.__init__Fc C s4 |d u r t �� }n,|dk rd|� d�}t|��ztj�|� W n ty4 } z d}t||d�|�d }~ww || _d | _ d | _ d | _ d| _ d| _ d | _t�� | _d | _t�|�| _d | _d | _t| _| jd f| _z| �� | _W d S t yy Y d S t y� Y d S ty� |s�d}t||d�d �d| _ Y d S w ) Nr z$pid must be a positive integer (got rt zprocess PID out of range�r� Fzprocess PID not foundT) �os�getpidr� rw �cext�check_pid_range� OverflowErrorr. �_pid�_name�_exe� _create_time�_gone� _pid_reused�_hash� threading�RLock�_lock�_ppidrN �_proc�_last_sys_cpu_times�_last_proc_cpu_times� _SENTINEL� _exitcoderz �_ident� _get_identr, r0 )r� rz � _ignore_nspr� �errrl rl rm r� ? sL �� �z Process._initc C s. t r| jjdd�| _| j| jfS | j| �� fS )a Return a (pid, uid) tuple which is supposed to identify a Process instance univocally over time. The PID alone is not enough, as it can be assigned to a new process after this one terminates, so we add process creation time to the mix. We need this in order to prevent killing the wrong process later on. This is also known as PID reuse or PID recycling problem. The reliability of this strategy mostly depends on create_time() precision, which is 0.01 secs on Linux. The assumption is that, after a process terminates, the kernel won't reuse the same PID after such a short period of time (0.01 secs). Technically this is inherently racy, but practically it should be good enough. T)� fast_only)r+ r� � create_timer� rz �r� rl rl rm r� n s zProcess._get_identc C s t �� }| j|d< | jr| j|d< | �� �m | jr d|d< n0z| �� |d< | �� |d< W n! ty9 d|d< Y n t yE d|d< Y n t yM Y nw | j t d hvrZ| j |d< | j d urft| j �|d< d �| jj| jjd �d d � |�� D ���W d � S 1 s�w Y d S ) Nrz �namezterminated + PID reused�status�zombie� terminated�exitcode�startedz {}.{}({})�, c S s g | ] \}}|� d |���qS )�=rl )rj �k�vrl rl rm � <listcomp>� s z#Process.__str__.<locals>.<listcomp>)� collections� OrderedDictrz r� �oneshotr� r� r� r0 r. r, r� r� r� r� �format� __class__� __module__�__name__�join�items)r� �inforl rl rm �__str__� s6 � �$�zProcess.__str__c C sh t |t�stS ts tr.| j\}}|j\}}||kr.|r.|s.z| �� tkW S ty- Y nw | j|jkS rh ) � isinstancerN �NotImplementedr r r� r� r) r- )r� �other�pid1�ident1�pid2�ident2rl rl rm �__eq__� s � zProcess.__eq__c C s | |k S rh rl )r� r� rl rl rm �__ne__� � zProcess.__ne__c C s | j d u r t| j�| _ | j S rh )r� �hashr� r� rl rl rm �__hash__� s zProcess.__hash__c C s2 | j s | �� s| j rd}t| j| j|d��dS dS )z9Raises NoSuchProcess in case process PID has been reused.z4process no longer exists and its PID has been reusedr� N)r� � is_runningr. rz r� )r� r� rl rl rm �_raise_if_pid_reused� s �z Process._raise_if_pid_reusedc C s | j S )zThe process PID.)r� r� rl rl rm rz � s z Process.pidc c s � | j �~ t| d�rdV ndzC| j�| � | j�| � | j�| � tr)| j�| � | j� � dV W | j� | � | j� | � | j� | � trL| j� | � | j� � n(| j� | � | j� | � | j� | � trl| j� | � | j� � w W d � dS W d � dS 1 s�w Y dS )a# Utility context manager which considerably speeds up the retrieval of multiple process information at the same time. Internally different process info (e.g. name, ppid, uids, gids, ...) may be fetched by using the same routine, but only one information is returned and the others are discarded. When using this context manager the internal routine is executed once (in the example below on name()) and the other info are cached. The cache is cleared when exiting the context manager block. The advice is to use this every time you retrieve more than one information about the process. If you're lucky, you'll get a hell of a speedup. >>> import psutil >>> p = psutil.Process() >>> with p.oneshot(): ... p.name() # collect multiple info ... p.cpu_times() # return cached value ... p.cpu_percent() # return cached value ... p.create_time() # return cached value ... >>> �_cacheN) r� �hasattrrV �cache_activate� memory_inforx r �uidsr� � oneshot_enter�cache_deactivate� oneshot_exitr� rl rl rm r� � s8 � � ��&"�zProcess.oneshotc C s t }|dur<t|ttttf�sdt|�� �}t|��t|�}|| }|r<d�t |�dkr.dndd� t t |���}t |��i }|pA|}| �� �= |D ]1}z|dkrT| j} nt| |�} | � } W n ttfyj |} Y n tyu |rs� Y qIw | ||< qIW d � |S 1 s�w Y |S ) a� Utility method returning process information as a hashable dictionary. If *attrs* is specified it must be a list of strings reflecting available Process class' attribute names (e.g. ['cpu_times', 'name']) else all public (read only) attributes are assumed. *ad_value* is the value which gets assigned in case AccessDenied or ZombieProcess exception is raised when retrieving that particular process information. Nzinvalid attrs type zinvalid attr name{} {}r �srr r� rz )�_as_dict_attrnamesr� �listr� �set� frozenset�type� TypeErrorr� �lenr� �map�reprr� r� rz �getattrr, r0 �NotImplementedError) r� �attrs�ad_value� valid_namesr� � invalid_names�retdict�lsr� ry �methrl rl rm �as_dict sH � � � � ��zProcess.as_dictc C sv t durt nt� d }| j|krdS | �� }|dur9| �� }zt|�}|�� |kr,|W S W dS ty8 Y dS w dS )z�Return the parent process as a Process object pre-emptively checking whether PID has been reused. If no parent is known return None. Nr )� _LOWEST_PIDrQ rz rx r� rN r. )r� � lowest_pidrx �ctime�parentrl rl rm r� J s � ��zProcess.parentc C s2 g }| � � }|dur|�|� |� � }|dus |S )z�Return the parents of this process as a list of Process instances. If no parents are known return an empty list. N)r� �append)r� �parents�procrl rl rm r� ] s �zProcess.parentsc C sp | j s| jrdS z| t| j�k| _| jr t�| j� t| j��W dS ty+ Y dS ty7 d| _ Y dS w )z�Return whether this process is running. It also checks if PID has been reused by another process, in which case it will remove the process from `process_iter()` internal cache and return False. FT)r� r� rN rz � _pids_reused�addr. r0 r� rl rl rm r� h s �zProcess.is_runningc C s. | � � tr | j�� S | jp| j�� | _| jS )z`The process parent PID. On Windows the return value is cached after first call. )r� r r� rx r� r� rl rl rm rx � s z Process.ppidc C s� t r | jdur | jS | j�� }tr:t|�dkr:z| �� }W n ttfy( Y nw |r:t j � |d �}|� |�r:|}|| _|| j_|S )z>The process name. The return value is cached after first call.N� r ) r+ r� r� r� r r� �cmdliner, r0 r� �path�basename� startswith)r� r� r� � extended_namerl rl rm r� � s � z Process.namec s� � fdd�}� j du r@z� j�� }W n ty) } z ||d�W Y d}~S d}~ww |s=z||d�}W n ty< Y nw |� _ � j S )z�The process executable as an absolute path. May also be an empty string. The return value is cached after first call. c sd � � � }|r)ttd�r)ttd�r)|d }tj�|�r)tj�|�r)t�|tj�r)|S t| t �r0| �| S )N�access�X_OKr ) r� r� r� r� �isabs�isfiler� r� r� r, )�fallbackr� �exer� rl rm �guess_it� s � � � z Process.exe.<locals>.guess_itN)r� )r� r� r r, )r� r r r� rl r� rm r � s �� �z Process.exec C � | j �� S )z3The command line this process has been called with.)r� r� r� rl rl rm r� � � zProcess.cmdlinec C s$ z| j �� W S ty t Y S w )z2The process current status as a STATUS_* constant.)r� r� r0 r) r� rl rl rm r� � s �zProcess.statusc C sV t r&tdu r d}t|��| �� j}zt�|�jW S ty% t|� Y S w | j � � S )ztThe name of the user that owns the process. On UNIX this is calculated by using *real* process uid. Nz0requires pwd module shipped with standard python) r �pwd� ImportErrorr� �real�getpwuid�pw_name�KeyError�strr� �username)r� r� �real_uidrl rl rm r � s � zProcess.usernamec C s | j du r | j�� | _ | j S )z�The process creation time as a floating point number expressed in seconds since the epoch. The return value is cached after first call. N)r� r� r� r� rl rl rm r� � s zProcess.create_timec C r )z6Process current working directory as an absolute path.)r� �cwdr� rl rl rm r r z Process.cwdc C s* |du r | j �� S | �� | j �|� dS )z'Get or set process niceness (priority).N)r� �nice_getr� �nice_set)r� �valuerl rl rm �nice s z Process.nicec C r )zVReturn process UIDs as a (real, effective, saved) namedtuple. )r� r� r� rl rl rm r� � z Process.uidsc C r )zVReturn process GIDs as a (real, effective, saved) namedtuple. )r� �gidsr� rl rl rm r � z Process.gidsc C r )zVThe terminal associated with this process, if any, else None. )r� �terminalr� rl rl rm r " r zProcess.terminalc C r )zcReturn the number of file descriptors opened by this process (POSIX only). )r� �num_fdsr� rl rl rm r ( r zProcess.num_fds� io_countersc C r )a Return process I/O statistics as a (read_count, write_count, read_bytes, write_bytes) namedtuple. Those are the number of read/write calls performed and the amount of bytes read and written by the process. )r� r r� rl rl rm r 1 s zProcess.io_counters� ionice_getc C s< |du r|durd}t |��| j�� S | �� | j�||�S )a� Get or set process I/O niceness (priority). On Linux *ioclass* is one of the IOPRIO_CLASS_* constants. *value* is a number which goes from 0 to 7. The higher the value, the lower the I/O priority of the process. On Windows only *ioclass* is used and it can be set to 2 (normal), 1 (low) or 0 (very low). Available on Linux and Windows > Vista only. Nz$'ioclass' argument must be specified)r� r� r r� � ionice_set)r� �ioclassr r� rl rl rm �ionice= s zProcess.ionicerd c C s |dur| � � | j�||�S )a- Get or set process resource limits as a (soft, hard) tuple. *resource* is one of the RLIMIT_* constants. *limits* is supposed to be a (soft, hard) tuple. See "man prlimit" for further info. Available on Linux and FreeBSD only. N)r� r� rd )r� �resource�limitsrl rl rm rd U s zProcess.rlimit�cpu_affinity_getc C sl |du r t t| j�� ��S | �� |s*t| jd�r | j�� }n ttt t dd����}| j� t t|��� dS )a- Get or set process CPU affinity. If specified, *cpus* must be a list of CPUs for which you want to set the affinity (e.g. [0, 1]). If an empty list is passed, all egible CPUs are assumed (and set). (Windows, Linux and BSD only). N�_get_eligible_cpusT��percpu) �sortedr� r� r r� r� r r� �ranger� rV �cpu_affinity_setr� )r� �cpusrl rl rm � cpu_affinityf s zProcess.cpu_affinity�cpu_numc C r )aZ Return what CPU this process is currently running on. The returned number should be <= psutil.cpu_count() and <= len(psutil.cpu_percent(percpu=True)). It may be used in conjunction with psutil.cpu_percent(percpu=True) to observe the system workload distributed across CPUs. )r� r' r� rl rl rm r' | s zProcess.cpu_num�environc C r )z�The environment variables of the process as a dict. Note: this might not reflect changes made after the process started. )r� r( r� rl rl rm r( � r zProcess.environc C r )z\Return the number of handles opened by this process (Windows only). )r� � num_handlesr� rl rl rm r) � r zProcess.num_handlesc C r )zkReturn the number of voluntary and involuntary context switches performed by this process. )r� �num_ctx_switchesr� rl rl rm r* � r zProcess.num_ctx_switchesc C r )z2Return the number of threads used by this process.)r� � num_threadsr� rl rl rm r+ � r zProcess.num_threads�threadsc C r )z�Return threads opened by process as a list of (id, user_time, system_time) namedtuples representing thread id and thread CPU times (user/system). On OpenBSD this method requires root access. )r� r, r� rl rl rm r, � s zProcess.threadsc C s, | � � t� }g }|s:|�� D ](\}}|| jkr7zt|�}| �� |�� kr*|�|� W q ttfy6 Y qw q|S t � t �}|�� D ] \}}|| �|� qCt � }| jg} | r�| � � }||v raqV|�|� || D ]'} zt| �}| �� |�� k} | r�|�|� | �| � W qj ttfy� Y qjw | sX|S )u( Return the children of this process as a list of Process instances, pre-emptively checking whether PID has been reused. If *recursive* is True return all the parent descendants. Example (A == this process): A ─┐ │ ├─ B (child) ─┐ │ └─ X (grandchild) ─┐ │ └─ Y (great grandchild) ├─ C (child) └─ D (child) >>> import psutil >>> p = psutil.Process() >>> p.children() B, C, D >>> p.children(recursive=True) B, X, Y, C, D Note that in the example above if process X disappears process Y won't be listed as the reference to process A is lost. )r� r{ r� rz rN r� r� r. r0 r� � defaultdictr� r� �popr� ) r� � recursiverv ry rz rx �child�reverse_ppid_map�seen�stack� child_pid�intimerl rl rm �children� sN ���% � ���zProcess.childrenc s |duo|dk}|dur|dk rd|�d�}t |��t� p d� � fdd�}|r=|� }| j�� }t�|� |� }| j�� }n | j}| j}|� }| j�� }|du sS|du r[|| _|| _dS |j|j |j |j } || } || _|| _z| | d } W n t y� Y dS w | � } t | d�S ) a Return a float representing the current process CPU utilization as a percentage. When *interval* is 0.0 or None (default) compares process times to system CPU times elapsed since last call, returning immediately (non-blocking). That means that the first time this is called it will return a meaningful 0.0 value. When *interval* is > 0.0 compares process times to system CPU times elapsed before and after the interval (blocking). In this case is recommended for accuracy that this function be called with at least 0.1 seconds between calls. A value > 100.0 can be returned in case of processes running multiple threads on different CPU cores. The returned value is explicitly NOT split evenly between all available logical CPUs. This means that a busy loop process running on a system with 2 logical CPUs will be reported as having 100% CPU utilization instead of 50%. Examples: >>> import psutil >>> p = psutil.Process(os.getpid()) >>> # blocking >>> p.cpu_percent(interval=1) 2.0 >>> # non-blocking (percentage since last call) >>> p.cpu_percent(interval=None) 2.9 >>> N� r � interval is not positive (got rt r c s t � � S rh )�_timerrl ��num_cpusrl rm �timer r� z"Process.cpu_percent.<locals>.timer�d ) r� rY r� rV r| �sleepr� r� �user�system�ZeroDivisionError�round) r� �interval�blockingr� r< �st1�pt1�st2�pt2� delta_proc� delta_time�overall_cpus_percent�single_cpu_percentrl r: rm rW � s> # � zProcess.cpu_percentc C r )a% Return a (user, system, children_user, children_system) namedtuple representing the accumulated process time, in seconds. This is similar to os.times() but per-process. On macOS and Windows children_user and children_system are always set to 0. )r� rV r� rl rl rm rV N � zProcess.cpu_timesc C r )a Return a namedtuple with variable fields depending on the platform, representing memory information about the process. The "portable" fields available on all platforms are `rss` and `vms`. All numbers are expressed in bytes. )r� r� r� rl rl rm r� Y rM zProcess.memory_infoc C r )a] This method returns the same information as memory_info(), plus, on some platform (Linux, macOS, Windows), also provides additional metrics (USS, PSS and swap). The additional metrics provide a better representation of actual process memory usage. Namely USS is the memory which is unique to a process and which would be freed if the process was terminated right now. It does so by passing through the whole process address. As such it usually requires higher user privileges than memory_info() and is considerably slower. )r� �memory_full_infor� rl rl rm rN d s zProcess.memory_full_info�rssc C s� t tjj�}||vrd|�dt|���}t|��|tjjv r!| jn| j}|� }t ||�}t p1t � j }|dks@d|�d�}t|��|t |� d S )a� Compare process memory to total physical system memory and calculate process memory utilization as a percentage. *memtype* argument is a string that dictates what type of process memory you want to compare against (defaults to "rss"). The list of available strings can be obtained like this: >>> psutil.Process().memory_info()._fields ('rss', 'vms', 'shared', 'text', 'lib', 'data', 'dirty', 'uss', 'pss') zinvalid memtype z; valid types are r z]can't calculate process memory percent because total physical system memory is not positive (rt r= )r� rw �pfullmem�_fieldsr� r� �pmemr� rN r� � _TOTAL_PHYMEMrT �total�float)r� �memtype� valid_typesr� �fun�metricsr � total_phymemrl rl rm �memory_percentt s* �� �� ��zProcess.memory_percent� memory_mapsTc s� | j �� }|rBi � |D ])}|d }|dd� }zttdd� � | |��� |< W q ty4 |� |< Y q w tj�� �fdd�� D �S tj��fdd�|D �S ) a� Return process' mapped memory regions as a list of namedtuples whose fields are variable depending on the platform. If *grouped* is True the mapped regions with the same 'path' are grouped together and the different memory fields are summed. If *grouped* is False every mapped region is shown as a single entity and the namedtuple will also include the mapped region's address space ('addr') and permission set ('perms'). � � Nc S s | | S rh rl )�x�yrl rl rm �<lambda>� � z%Process.memory_maps.<locals>.<lambda>c s g | ] }�|g� | �R � �qS rl rl )rj r� ��d�ntrl rm r� � s z'Process.memory_maps.<locals>.<listcomp>c s g | ]}� |� �qS rl rl �rj r_ )re rl rm r� � � )r� r\ r� r� r rw � pmmap_grouped� pmmap_ext)r� �grouped�it�tuplr� �numsrl rc rm r\ � s �zProcess.memory_mapsc C r )z�Return files opened by process as a list of (path, fd) namedtuples including the absolute file name and file descriptor number. )r� � open_filesr� rl rl rm rn � r zProcess.open_files�inetc C s t |� | j�|�S )aT Return socket connections opened by process as a list of (fd, family, type, laddr, raddr, status) namedtuples. The *kind* parameter filters for connections that match the following criteria: +------------+----------------------------------------------------+ | Kind Value | Connections using | +------------+----------------------------------------------------+ | inet | IPv4 and IPv6 | | inet4 | IPv4 | | inet6 | IPv6 | | tcp | TCP | | tcp4 | TCP over IPv4 | | tcp6 | TCP over IPv6 | | udp | UDP | | udp4 | UDP over IPv4 | | udp6 | UDP over IPv6 | | unix | UNIX socket (both UDP and TCP protocols) | | all | the sum of all the possible families and protocols | +------------+----------------------------------------------------+ )r� r� r\ �r� r� rl rl rm r\ � s zProcess.net_connectionsr\ )� replacementc C s | j |d�S )N�r� )r\ rp rl rl rm � connections� s zProcess.connectionsc C s� | j dk r J | j ��| �� | j | j| j}}}|dkr#d}t|��z t�||� W d S tyN } ztrAt |�rAt |||�|�d| _ t ||�|�d }~w t y_ } zt||�|�d }~ww )Nr z�preventing sending signal to process with PID 0 as it would affect every process in the process group of the calling process (os.getpid()) instead of PID 0T)rz r� r� r� r� r� �kill�ProcessLookupErrorr rP r0 r� r. �PermissionErrorr, )r� �sigrz rx r� r� r� rl rl rm � _send_signal� s& � � ��zProcess._send_signalc C sR t r | �|� dS | �� |tjkr!| �� s!d}t| j| j|d��| j � |� dS )z�Send a signal *sig* to process pre-emptively checking whether PID has been reused (see signal module constants) . On Windows only SIGTERM is valid and is treated as an alias for kill(). zprocess no longer existsr� N) r rx r� �signal�SIGTERMr� r. rz r� r� � send_signal)r� rw r� rl rl rm r{ � s zProcess.send_signalc C �* t r | �tj� dS | �� | j�� dS )z�Suspend process execution with SIGSTOP pre-emptively checking whether PID has been reused. On Windows this has the effect of suspending all process threads. N)r rx ry �SIGSTOPr� r� �suspendr� rl rl rm r~ � zProcess.suspendc C r| )z�Resume process execution with SIGCONT pre-emptively checking whether PID has been reused. On Windows this has the effect of resuming all process threads. N)r rx ry �SIGCONTr� r� �resumer� rl rl rm r� r zProcess.resumec C r| )z�Terminate the process with SIGTERM pre-emptively checking whether PID has been reused. On Windows this is an alias for kill(). N)r rx ry rz r� r� rt r� rl rl rm � terminate r zProcess.terminatec C r| )zjKill the current process with SIGKILL pre-emptively checking whether PID has been reused. N)r rx ry �SIGKILLr� r� rt r� rl rl rm rt & s z Process.killc C s@ |dur|dksd}t |��| jtur| jS | j�|�| _| jS )a� Wait for process to terminate and, if process is a children of os.getpid(), also return its exit code, else None. On Windows there's no such limitation (exit code is always returned). If the process is already terminated immediately return None instead of raising NoSuchProcess. If *timeout* (in seconds) is specified and process is still alive raise TimeoutExpired. To wait for multiple Process(es) use psutil.wait_procs(). Nr z"timeout must be a positive integer)r� r� r� r� �wait)r� �timeoutr� rl rl rm r� 0 s z Process.waitrh �F�NN)rO �T�ro )Fr� r� � __qualname__�__doc__r� r� r� r� �__repr__r� r� r� r� �propertyrz � contextlib�contextmanagerr� r� r� r� r� r2 rx r� r r� r� r r� r r r r� r r r r� rw rN r r rd r& r' r( r+ r) r* r+ r, r6 rW rV r� rN r[ r\ rn r\ r �deprecated_methodrs rx r{ r~ r� r� rt r� rl rl rl rm rN s� / B/ * F] $ rN c C s" h | ] }|� d �s|dvr|�qS )�_> rt r� r� r� rd r� r� r� r~ r6 r� r� rs r{ )r� rf rl rl rm � <setcomp>I s � �r� c sJ e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd� fd d� Z � Z S )rO a Same as subprocess.Popen, but in addition it provides all psutil.Process methods in a single class. For the following methods which are common to both classes, psutil implementation takes precedence: * send_signal() * terminate() * kill() This is done in order to avoid killing another process in case its PID has been reused, fixing BPO-6973. >>> import psutil >>> from subprocess import PIPE >>> p = psutil.Popen(["python", "-c", "print 'hi'"], stdout=PIPE) >>> p.name() 'python' >>> p.uids() user(real=1000, effective=1000, saved=1000) >>> p.username() 'giampaolo' >>> p.communicate() ('hi', None) >>> p.terminate() >>> p.wait(timeout=2) 0 >>> c O s( t j|i |��| _| j| jjdd� d S )NT)r� )� subprocessrO �_Popen__subprocr� rz �r� �args�kwargsrl rl rm r� u s zPopen.__init__c C s t ttt�ttj� ��S rh )r" r� �dirrO r� r� rl rl rm �__dir__| s z Popen.__dir__c C s t | jd�r | j�� | S )N� __enter__)r� r� r� r� rl rl rm r� s zPopen.__enter__c O sv t | jd�r| jj|i |��S | jr| j�� | jr | j�� z| jr/| j�� W | �� d S W | �� d S | �� w )N�__exit__)r� r� r� �stdout�close�stderr�stdinr� r� rl rl rm r� � s �zPopen.__exit__c C s\ zt �| |�W S ty- z t �| j|�W Y S ty, | j�d|��}t|�d �w w )Nz has no attribute )�object�__getattribute__�AttributeErrorr� r� )r� r� r� rl rl rm r� � s ��zPopen.__getattribute__Nc s, | j jd ur | j jS t� �|�}|| j _|S rh )r� � returncode�superr� )r� r� ry �r� rl rm r� � s z Popen.waitrh ) r� r� r� r� r� r� r� r� r� r� � __classcell__rl rl r� rm rO W s rO c C s t t�� �} | d a| S )z&Return a list of current running PIDs.r )r" rw rQ r� �ry rl rl rm rQ � s rQ c C s, | dk rdS | dkrt r| t� v S t�| �S )z�Return True if given PID exists in the current process list. This is faster than doing "pid in psutil.pids()" and should be preferred. r F)r rQ rw rP �rz rl rl rm rP � s rP c # s � � fdd�}� fdd�}t �� � tt� �}t� �� �}|| }|| }|D ]}||� q&tr@t�� }td|� �� ||� ts/zCtt � � � �t t � |�� � � �} | D ]*\}} z| du ra||�} | durm| j | |d�| _| V W qT ty~ ||� Y qTw W � a dS � a w )a� Return a generator yielding a Process instance for all running processes. Every new Process instance is only created once and then cached into an internal table which is updated every time this is used. Cache can optionally be cleared via `process_iter.clear_cache()`. The sorting order in which processes are yielded is based on their PIDs. *attrs* and *ad_value* have the same meaning as in Process.as_dict(). If *attrs* is specified as_dict() is called and the resulting dict is stored as a 'info' attribute attached to returned Process instance. If *attrs* is an empty list it will retrieve all process info (slow). c s t | �}|� |j< |S rh )rN rz )rz r� ��pmaprl rm r� � s zprocess_iter.<locals>.addc s � � | d � d S rh )r. r� r� rl rm �remove� s z process_iter.<locals>.removez+refreshing Process instance for reused PID N)r� r� )�_pmap�copyr� rQ �keysr� r. r1 r" r� r� �dict�fromkeysr� r� r. ) r� r� r� r� �a�b�new_pids� gone_pidsrz r� r� rl r� rm rR � s: � �" �� rR c C � t �� S rh )r� �clearrl rl rl rm ra � rb ra z$Clear process_iter() internal cache.c s � �fdd�}|dur|dksd|� �}t |��t� �t| �}� dur1t� �s1d� �d�}t|��|dur:t� | }|rs|durE|dkrEn.|D ]%}dt|� }|durgt|t� |�}|dkra n |||� qG|||� qG|� }|s<|r�|D ]}||d� qw|� }t��t|�fS ) a, Convenience function which waits for a list of processes to terminate. Return a (gone, alive) tuple indicating which processes are gone and which ones are still alive. The gone ones will have a new *returncode* attribute indicating process exit status (may be None). *callback* is a function which gets called every time a process terminates (a Process instance is passed as callback argument). Function will return as soon as all processes terminate or when *timeout* occurs. Differently from Process.wait() it will not raise TimeoutExpired if *timeout* occurs. Typical use case is: - send SIGTERM to a list of processes - give them some time to terminate - send SIGKILL to those ones which are still alive Example: >>> def on_terminate(proc): ... print("process {} terminated".format(proc)) ... >>> for p in procs: ... p.terminate() ... >>> gone, alive = wait_procs(procs, timeout=3, callback=on_terminate) >>> for p in alive: ... p.kill() c sh z| j |d�}W n ttjfy Y d S w |d us | �� s2|| _��| � � d ur0� | � d S d S d S )N)r� )r� r/ r� r� r� r� )r� r� r� ��callback�gonerl rm � check_gone( s � ��z wait_procs.<locals>.check_goneNr z(timeout must be a positive integer, got z callback z is not a callableg �?)r� r� �callabler� r9 r� �minr� ) �procsr� r� r� r� �alive�deadliner� � max_timeoutrl r� rm rS s: % � rS Tc C s. | rt �� }nt �� }|dur|dk rd}|S )al Return the number of logical CPUs in the system (same as os.cpu_count()). If *logical* is False return the number of physical cores only (e.g. hyper thread CPUs are excluded). Return None if undetermined. The return value is cached after first call. If desired cache can be cleared like this: >>> psutil.cpu_count.cache_clear() Nr )rw �cpu_count_logical�cpu_count_cores)�logicalry rl rl rm rY c s rY Fc C s | st �� S t �� S )a� Return system-wide CPU times as a namedtuple. Every CPU time represents the seconds the CPU has spent in the given mode. The namedtuple's fields availability varies depending on the platform: - user - system - idle - nice (UNIX) - iowait (Linux) - irq (Linux, FreeBSD) - softirq (Linux) - steal (Linux >= 2.6.11) - guest (Linux >= 2.6.24) - guest_nice (Linux >= 3.2.0) When *percpu* is True return a list of namedtuples for each CPU. First element of the list refers to first CPU, second element to second CPU and so on. The order of the list is consistent across calls. )rw rV � per_cpu_timesr rl rl rm rV z s rV r c C s0 t | �}tr|t| dd�8 }|t| dd�8 }|S )zWGiven a cpu_time() ntuple calculates the total CPU time (including idle time). �guestr � guest_nice)�sumr r� )�times�totrl rl rm � _cpu_tot_time� s r� c C s& t | �}|| j8 }|t| dd�8 }|S )zlGiven a cpu_time() ntuple calculates the busy CPU time. We do so by subtracting all idle CPU times. �iowaitr )r� �idler� )r� �busyrl rl rm �_cpu_busy_time� s r� c C s\ | j |j ks J | |f��g }tjj D ]}t||�t| |� }td|�}|�|� qtj|� S )Nr )rQ rw � scputimesr� �maxr� )�t1�t2� field_deltas�field� field_deltarl rl rm �_cpu_times_deltas� s r� c C s� t �� j}| duo | dk}| dur | dk r d| � d�}t|��dd� }|sD|r0t� }t�| � nt�|�p7t� }t� t|< ||t| �S g }|rStdd �}t�| � n t �|�p\tdd �}tdd �t |< t |t | �D ] \}} |� ||| �� qk|S ) a� Return a float representing the current system-wide CPU utilization as a percentage. When *interval* is > 0.0 compares system CPU times elapsed before and after the interval (blocking). When *interval* is 0.0 or None compares system CPU times elapsed since last call or module import, returning immediately (non blocking). That means the first time this is called it will return a meaningless 0.0 value which you should ignore. In this case is recommended for accuracy that this function be called with at least 0.1 seconds between calls. When *percpu* is True returns a list of floats representing the utilization as a percentage for each CPU. First element of the list refers to first CPU, second element to second CPU and so on. The order of the list is consistent across calls. Examples: >>> # blocking, system-wide >>> psutil.cpu_percent(interval=1) 2.0 >>> >>> # blocking, per-cpu >>> psutil.cpu_percent(interval=1, percpu=True) [2.0, 1.0] >>> >>> # non-blocking (percentage since last call) >>> psutil.cpu_percent(interval=None) 2.9 >>> Nr7 r r8 rt c S sJ t | |�}t|�}t|�}z|| d }W n ty Y dS w t|d�S )Nr= r7 r )r� r� r� rA rB )r� r� � times_delta� all_delta� busy_delta� busy_percrl rl rm � calculate s � z cpu_percent.<locals>.calculateTr ) r� �current_thread�identr� rV r| r> �_last_cpu_times�get�_last_per_cpu_times�zipr� � rC r! �tidrD r� r� r� ry �tot1r� rl rl rm rW � s, # rW c C s� t �� j}| duo | dk}| dur | dk r d| �d�}t|��dd� }|sD|r0t� }t�| � nt�|�p7t� }t� t|< ||t| �S g }|rStdd �}t�| � n t �|�p\tdd �}tdd �t |< t |t | �D ] \}} |� ||| �� qk|S ) a� Same as cpu_percent() but provides utilization percentages for each specific CPU time as is returned by cpu_times(). For instance, on Linux we'll get: >>> cpu_times_percent() cpupercent(user=4.8, nice=0.0, system=4.8, idle=90.5, iowait=0.0, irq=0.0, softirq=0.0, steal=0.0, guest=0.0, guest_nice=0.0) >>> *interval* and *percpu* arguments have the same meaning as in cpu_percent(). Nr7 r r8 rt c S sd g }t | |�}t|�}dtd|� }|D ]}|| }t|d�}ttd|�d�}|�|� qtj|� S )Ng Y@r r7 )r� r� r� rB r� r� rw r� )r� r� rm r� r� �scaler� � field_percrl rl rm r� B s z$cpu_times_percent.<locals>.calculateTr ) r� r� r� r� rV r| r> �_last_cpu_times_2r� �_last_per_cpu_times_2r� r� r� rl rl rm rX / s, rX c C r� )zReturn CPU statistics.)rw rZ rl rl rl rm rZ j s rZ �cpu_freqc C s� t �� }| r|S tt|��}|dkrdS |dkr |d S d\}}}d}|D ]}||j7 }tr6|jdu r6d}q%||j7 }||j7 }q%|| }|rLd } } n|| } || } t� || | �S )a: Return CPU frequency as a namedtuple including current, min and max frequency expressed in Mhz. If *percpu* is True and the system supports per-cpu frequency retrieval (Linux only) a list of frequencies is returned for each CPU. If not a list with one element is returned. r Nr )r7 r7 r7 FT) rw r� rU r� �currentr r� r� r �scpufreq) r! ry r; �currs�mins�maxs�set_none�cpur� �min_�max_rl rl rm r� q s. � getloadavgc C s t �� } | ja| S )a� Return statistics about system memory usage as a namedtuple including the following fields, expressed in bytes: - total: total physical memory available. - available: the memory that can be given instantly to processes without the system going into swap. This is calculated by summing different memory values depending on the platform and it is supposed to be used to monitor actual memory usage in a cross platform fashion. - percent: the percentage usage calculated as (total - available) / total * 100 - used: memory used, calculated differently depending on the platform and designed for informational purposes only: macOS: active + wired BSD: active + wired + cached Linux: total - free - free: memory not being used at all (zeroed) that is readily available; note that this doesn't reflect the actual memory available (use 'available' instead) Platform-specific fields: - active (UNIX): memory currently in use or very recently used, and so it is in RAM. - inactive (UNIX): memory that is marked as not used. - buffers (BSD, Linux): cache for things like file system metadata. - cached (BSD, macOS): cache for various things. - wired (macOS, BSD): memory that is marked to always stay in RAM. It is never moved to disk. - shared (BSD): memory that may be simultaneously accessed by multiple processes. The sum of 'used' and 'available' does not necessarily equal total. On Windows 'available' and 'free' are the same. )rw rT rT rS r� rl rl rm rT � s 5rT c C r� )a� Return system swap memory statistics as a namedtuple including the following fields: - total: total swap memory in bytes - used: used swap memory in bytes - free: free swap memory in bytes - percent: the percentage usage - sin: no. of bytes the system has swapped in from disk (cumulative) - sout: no. of bytes the system has swapped out from disk (cumulative) 'sin' and 'sout' on Windows are meaningless and always set to 0. )rw rU rl rl rl rm rU � s rU c C � t �| �S )z�Return disk usage statistics about the given *path* as a namedtuple including total, used and free space expressed in bytes plus the percentage usage. )rw ra )r� rl rl rm ra � r ra c C r� )a3 Return mounted partitions as a list of (device, mountpoint, fstype, opts) namedtuple. 'opts' field is a raw string separated by commas indicating mount options which may vary depending on the platform. If *all* parameter is False return physical devices only and ignore all others. )rw r` )�allrl rl rm r` rM r` c C s� t rt| d�ni }tjdi |��}|s| ri S dS |r t|d�}ttdtj�}| r:|�� D ] \}}||� ||< q-|S |dd� t |� � � D �� S )a� Return system disk I/O statistics as a namedtuple including the following fields: - read_count: number of reads - write_count: number of writes - read_bytes: number of bytes read - write_bytes: number of bytes written - read_time: time spent reading from disk (in ms) - write_time: time spent writing to disk (in ms) Platform specific: - busy_time: (Linux, FreeBSD) time spent doing actual I/Os (in ms) - read_merged_count (Linux): number of merged reads - write_merged_count (Linux): number of merged writes If *perdisk* is True return the same information for every physical disk installed on the system as a dictionary with partition names as the keys and the namedtuple described above as the values. If *nowrap* is True it detects and adjust the numbers which overflow and wrap (restart from 0) and add "old value" to "new value" so that the returned numbers will always be increasing or remain the same, but never decrease. "disk_io_counters.cache_clear()" can be used to invalidate the cache. On recent Windows versions 'diskperf -y' command may need to be executed first otherwise this function won't find any disk. )�perdiskN�psutil.disk_io_counters�sdiskioc s rg rh �r� rf rl rl rm rn < ro z#disk_io_counters.<locals>.<genexpr>rl ) r r� rw r_ � _wrap_numbersr� r r� r� r� �values)r� �nowrapr� �rawdictre �disk�fieldsrl rl rm r_ s r_ r� z Clears nowrap argument cachec C sj t �� }|s | r i S dS |rt|d�}| r'|�� D ] \}}tj|� ||< q|S tjdd� t|�� � D �� S )ac Return network I/O statistics as a namedtuple including the following fields: - bytes_sent: number of bytes sent - bytes_recv: number of bytes received - packets_sent: number of packets sent - packets_recv: number of packets received - errin: total number of errors while receiving - errout: total number of errors while sending - dropin: total number of incoming packets which were dropped - dropout: total number of outgoing packets which were dropped (always 0 on macOS and BSD) If *pernic* is True return the same information for every network interface installed on the system as a dictionary with network interface names as the keys and the namedtuple described above as the values. If *nowrap* is True it detects and adjust the numbers which overflow and wrap (restart from 0) and add "old value" to "new value" so that the returned numbers will always be increasing or remain the same, but never decrease. "net_io_counters.cache_clear()" can be used to invalidate the cache. N�psutil.net_io_countersc S s g | ]}t |��qS rl r� rf rl rl rm r� n rg z#net_io_counters.<locals>.<listcomp>)rw r[ r� r� r �snetior� r� )�pernicr� r� �nicr� rl rl rm r[ J s r[ r ro c C s t | � t�| �S )a� Return system-wide socket connections as a list of (fd, family, type, laddr, raddr, status, pid) namedtuples. In case of limited privileges 'fd' and 'pid' may be set to -1 and None respectively. The *kind* parameter filters for connections that fit the following criteria: +------------+----------------------------------------------------+ | Kind Value | Connections using | +------------+----------------------------------------------------+ | inet | IPv4 and IPv6 | | inet4 | IPv4 | | inet6 | IPv6 | | tcp | TCP | | tcp4 | TCP over IPv4 | | tcp6 | TCP over IPv6 | | udp | UDP | | udp4 | UDP over IPv4 | | udp6 | UDP over IPv6 | | unix | UNIX socket (both UDP and TCP protocols) | | all | the sum of all the possible families and protocols | +------------+----------------------------------------------------+ On macOS this function requires root privileges. )r� rw r\ rr rl rl rm r\ w s r\ c C sT t �� } | jdd� d� t�t�}| D ]�\}}}}}}zt�|�}W n tyB t r3|dkr3t j }n t t d�r@|t j kr@t j }Y nw |t j krct rLdnd}|� |�dk rc||� d �7 }|� |�dk sUt�|||||�} t r�|tjtjhv r�zt�| �}W n ty� } z t| � W Y d } ~ nd } ~ ww |d ur�| j|d � || �| � qt|�S ) a* Return the addresses associated to each NIC (network interface card) installed on the system as a dictionary whose keys are the NIC names and value is a list of namedtuples for each address assigned to the NIC. Each namedtuple includes 5 fields: - family: can be either socket.AF_INET, socket.AF_INET6 or psutil.AF_LINK, which refers to a MAC address. - address: is the primary address and it is always set. - netmask: and 'broadcast' and 'ptp' may be None. - ptp: stands for "point to point" and references the destination address on a point to point interface (typically a VPN). - broadcast: and *ptp* are mutually exclusive. Note: you can have more than one address of the same family associated with each interface. c S s | d S )Nr rl )r_ rl rl rm ra � rb z net_if_addrs.<locals>.<lambda>)�key���rM �:�-� �00N)� broadcast)rw r] �sortr� r- r� �socket� AddressFamilyr� r+ rM r� r �countr �snicaddr�AF_INET�AF_INET6�broadcast_addr� Exceptionr1 �_replacer� r� ) �rawlistry r� �fam�addr�maskr �ptp� separatorre r� rl rl rm r] � s@ � �� ��� r] c C r� )a Return information about each NIC (network interface card) installed on the system as a dictionary whose keys are the NIC names and value is a namedtuple with the following fields: - isup: whether the interface is up (bool) - duplex: can be either NIC_DUPLEX_FULL, NIC_DUPLEX_HALF or NIC_DUPLEX_UNKNOWN - speed: the NIC speed expressed in mega bits (MB); if it can't be determined (e.g. 'localhost') it will be set to 0. - mtu: the maximum transmission unit expressed in bytes. )rw r^ rl rl rl rm r^ � s r^ �sensors_temperaturesc s� � fdd�}t �t�}t�� }|�� D ]7\}}|rJ|�d�\}}}} ||�}||�}|| �} |r5| s5|} n| r;|s;| }|| �t� |||| �� |sqt |�S )a< Return hardware temperatures. Each entry is a namedtuple representing a certain hardware sensor (it may be a CPU, an hard disk or something else, depending on the OS and its configuration). All temperatures are expressed in celsius unless *fahrenheit* is set to True. c s( | d ur� rt | �d d d S | S d S )N� r � )rU )�n�� fahrenheitrl rm �convert� s �z%sensors_temperatures.<locals>.convertr ) r� r- r� rw r r� r. r� r �shwtempr� ) r r! ry r� r� r� �labelr� �high�criticalrl r rm r � s& ���� sensors_fansc C r� )z�Return fans speed. Each entry is a namedtuple representing a certain hardware sensor. All speed are expressed in RPM (rounds per minute). )rw r& rl rl rl rm r& s �sensors_batteryc C r� )a� Return battery information. If no battery is installed returns None. - percent: battery power left as a percentage. - secsleft: a rough approximation of how many seconds are left before the battery runs out of power. May be POWER_TIME_UNLIMITED or POWER_TIME_UNLIMITED. - power_plugged: True if the AC power cable is connected. )rw r' rl rl rl rm r' � c C r� )zAReturn the system boot time expressed in seconds since the epoch.)rw rc rl rl rl rm rc 2 � rc c C r� )a� Return users currently connected on the system as a list of namedtuples including the following fields. - user: the name of the user - terminal: the tty or pseudo-tty associated with the user, if any. - host: the host name associated with the entry, if any. - started: the creation time as a floating point number expressed in seconds since the epoch. )rw rb rl rl rl rm rb 9 r( rb c C r� )zjReturn a generator yielding a WindowsService instance for all Windows services installed. )rw �win_service_iterrl rl rl rm r* M r) r* c C r� )zjGet a Windows service by *name*. Raise NoSuchProcess if no service with such name exists. )rw �win_service_get)r� rl rl rm r+ S r r+ c C s( ddl }t| �|j_tj�t| �� dS )zZEnable or disable PSUTIL_DEBUG option, which prints debugging messages to stderr. r N)�psutil._common�boolr � PSUTIL_DEBUGrw r� � set_debug)r �psutilrl rl rm � _set_debug] s r1 r� r� r� )NF)FTr� )�r� r� r� r} � functoolsr� ry r r� �sysr� r| r r rr r r r r r r r r r r r r r r r r r r r r r r r r r r r r r r r! r" r# r$ r% r& r' r( r) r* r+ r, r- r. r/ r0 r1 r2 r3 r� � PROCFS_PATHr4 rw r5 r6 r7 r8 r9 �_psutil_windowsr: r; r< r= r> r? r@ rA rB rC rD rE rF rG rH rI rJ �platformr� r� �__all__�extend�__extra__all__r� rN re �globals�_globalsr� r� r� �isupperr� r� rM � __author__rL r� �splitrK r9 rS r� r� r� ri �replacer� ru r� r� r rs r� �whatrv r{ r� r� r� rO rQ rP r� r� r� rR � cache_clearrS rY rV r� r� r� r r� r� r� r� rW r� r� r� rX rZ r� r� rT rU ra r` r_ �partialr[ r\ r] r^ r r& r' rc rb r* r+ r1 rl rl rl rm �<module> s� � 1 ��� 1�T 6 ` ��� N ; ( ; /� '� = " __pycache__/_common.cpython-310.pyc 0000644 00000054276 15030211765 0013122 0 ustar 00 o �h�o � @ s� d Z ddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddlm Z ddlm Z ddlm Z ddlm Z zddlmZ W n eyQ dZY nw zddlmZ W n eye dZY nw ee�d ��Ze� Zg d �Zejd kZejd kZej�d �Z ej�d�Z e Z ej�d�Z ej�d�Z ej�d�Z!e p�e p�e!Z"ej�d�Z#ej�d�Z$dZ%dZ&dZ'dZ(dZ)dZ*dZ+dZ,d Z-d Z.d Z/d Z0d Z1d!Z2d"Z3d#Z4d$Z5d%Z6d&Z7d'Z8d(Z9d)Z:d*Z;d+Z<d,Z=d-Z>G d.d/� d/ej?�Z@eA� �Be@jC� G d0d1� d1ej?�ZDeA� �BeDjC� e�E� ZFe�G� ZHe d2g d3��ZIe d4g d5��ZJe d6g d7��ZKe d8g d9��ZLe d:g d;��ZMe d<g d=��ZNe d>g d?��ZOe d@g dA��ZPe dBg dC��ZQe dDg dE��ZRe dFg dG��ZSe dHg dI��ZTe dJg dK��ZUe dLdMdNg�ZVe dOg dP��ZWe dQdRdSg�ZXe dTg dU��ZYe dVg dW��ZZe dXg dW��Z[e dYg dZ��Z\e d[d\d]g�Z]e d^d_d`g�Z^e dag db��Z_e dcdddeg�Z`e eege e gfe ege gfe ge gfe ege gfe ge gfe ege e gfe ge e gfege e gfdf�Zaedu�r�ea�Bege gfege gfdg�� edu�re#�sea�Bdhege e gfi� G didj� djeb�ZcG dkdl� dlec�ZdG dmdn� dned�ZeG dodp� dpec�ZfG dqdr� drec�Zgd�dsdt�Zhdudv� Zidwdx� Zjdydz� Zkd{d|� Zleid}d~� �Zmdd�� Znd�d�� Zod�d�� Zpd�d�d��Zqd�d�� Zrd�d�� ZsG d�d�� d��Ztd�d�� Zuet� Zvevjweu_wevjxeu_xd�Zyd�d�� Zzd�d�� Z{ee{fd�d��Z|efd�d��Z}d�d�d��Z~d�d�� Zd�d�� Z�eiej�fd�d���Z�d�d�d��Z�dd�ej�fd�d��Z�d�d�� Z�dS )�z�Common objects shared by __init__.py and _ps*.py modules. Note: this module is imported by setup.py, so it should not import psutil or third-party modules. � N)� namedtuple)�AF_INET)� SOCK_DGRAM)� SOCK_STREAM)�AF_INET6)�AF_UNIX� PSUTIL_DEBUG)R�FREEBSD�BSD�LINUX�NETBSD�OPENBSD�MACOS�OSX�POSIX�SUNOS�WINDOWS� CONN_CLOSE�CONN_CLOSE_WAIT� CONN_CLOSING�CONN_ESTABLISHED�CONN_FIN_WAIT1�CONN_FIN_WAIT2� CONN_LAST_ACK� CONN_LISTEN� CONN_NONE� CONN_SYN_RECV� CONN_SYN_SENT�CONN_TIME_WAIT�NIC_DUPLEX_FULL�NIC_DUPLEX_HALF�NIC_DUPLEX_UNKNOWN� STATUS_DEAD�STATUS_DISK_SLEEP� STATUS_IDLE� STATUS_LOCKED�STATUS_RUNNING�STATUS_SLEEPING�STATUS_STOPPED�STATUS_SUSPENDED�STATUS_TRACING_STOP�STATUS_WAITING�STATUS_WAKE_KILL� STATUS_WAKING� STATUS_ZOMBIE� STATUS_PARKED�ENCODING� ENCODING_ERRSr �pconn� pcputimes�pctxsw�pgids�pio�pionice� popenfile�pthread�puids�sconn� scpustats�sdiskio� sdiskpart� sdiskusage�snetio�snicaddr� snicstats�sswap�suser� conn_tmap�deprecated_method� isfile_strict�memoize�parse_environ_block�path_exists_strict� usage_percent� supports_ipv6�sockfam_to_enum�socktype_to_enum� wrap_numbers� open_text� open_binary�cat�bcat� bytes2human�conn_to_ntuple�debug�hilite�term_supports_colors� print_color�posix�nt�linux�darwin)�freebsd� midnightbsd�openbsd�netbsd)�sunos�solaris�aix�running�sleepingz disk-sleep�stoppedz tracing-stop�zombie�deadz wake-kill�waking�idle�locked�waiting� suspended�parked� ESTABLISHED�SYN_SENT�SYN_RECV� FIN_WAIT1� FIN_WAIT2� TIME_WAIT�CLOSE� CLOSE_WAIT�LAST_ACK�LISTEN�CLOSING�NONEc @ s e Zd ZdZdZdZdS )� NicDuplex� � r N)�__name__� __module__� __qualname__r r r! � r� r� �A/usr/local/CyberCP/lib/python3.10/site-packages/psutil/_common.pyr| � s r| c @ s e Zd ZdZdZdS )� BatteryTime������N)r r� r� �POWER_TIME_UNKNOWN�POWER_TIME_UNLIMITEDr� r� r� r� r� � s r� rC )�total�used�free�percent�sin�soutr? )r� r� r� r� r= )� read_count� write_count� read_bytes� write_bytes� read_time� write_timer> )�device� mountpoint�fstype�optsr@ )� bytes_sent� bytes_recv� packets_sent� packets_recv�errin�errout�dropin�dropoutrD )�name�terminal�host�started�pidr; )�fd�family�type�laddr�raddr�statusr� rA )r� �address�netmask� broadcast�ptprB )�isup�duplex�speed�mtu�flagsr< )� ctx_switches� interrupts�soft_interrupts�syscalls�scpufreq)�current�min�max�shwtemp)�labelr� �high�critical�sbattery)r� �secsleft� power_plugged�sfanr� r� r3 )�user�system� children_user�children_systemr8 �pathr� r9 )�id� user_time� system_timer: )�real� effective�savedr5 r6 )r� r� r� r� r7 �ioclass�valuer4 � voluntary� involuntaryr2 )r� r� r� r� r� r� �addr�ip�port)�all�tcp�tcp4�udp�udp4�inet�inet4�inet6)�tcp6�udp6�unixc @ s, e Zd ZdZdZdd� Zdd� Zdd� Zd S ) �ErrorzQBase exception class. All other psutil exceptions inherit from this one. �psutilc C s> t �� }|D ]}t| |d �}|s|dkr |dkr |||< q|S )Nr� r )� collections� OrderedDict�getattr)�self�attrs�infor� r� r� r� r� � _infodict s �zError._infodictc C sR | � d�}|rd�d�dd� |�� D ���}nd }d�dd� t| dd �|fD ��S ) N)r� �ppidr� z({})�, c S � g | ] \}}|� d |���qS ��=r� ��.0�k�vr� r� r� � <listcomp> � z!Error.__str__.<locals>.<listcomp>� c S s g | ]}|r|�qS r� r� )r� �xr� r� r� r� s �msg� )r� �format�join�itemsr� �r� r� �detailsr� r� r� �__str__ s � z Error.__str__c C s8 | � d�}d�dd� |�� D ��}d| jj� d|� d�S )N)r� r� r� �secondsr� r� c S r� r� r� r� r� r� r� r� " r� z"Error.__repr__.<locals>.<listcomp>zpsutil.�(�))r� r� r� � __class__r r� r� r� r� �__repr__ s zError.__repr__N)r r� r� �__doc__r� r� r r� r� r� r� r� s r� c @ �& e Zd ZdZdZddd�Zdd� ZdS ) � NoSuchProcesszXException raised when a process with a certain PID doesn't or no longer exists. r� Nc C �$ t �| � || _|| _|pd| _d S )Nzprocess no longer exists�r� �__init__r� r� r� �r� r� r� r� r� r� r� r - � zNoSuchProcess.__init__c C � | j | j| j| jffS �N�r r� r� r� �r� r� r� r� � __reduce__3 � zNoSuchProcess.__reduce__�NN�r r� r� r r r r� r� r� r� r & s r c @ r ) � ZombieProcessa1 Exception raised when querying a zombie process. This is raised on macOS, BSD and Solaris only, and not always: depending on the query the OS may be able to succeed anyway. On Linux all zombie processes are querable (hence this is never raised). Windows doesn't have zombie processes. r� Nc C s$ t �| |||� || _|pd| _d S )Nz"PID still exists but it's a zombie)r r r� r� )r� r� r� r� r� r� r� r� r A s zZombieProcess.__init__c C s | j | j| j| j| jffS r )r r� r� r� r� r r� r� r� r F s zZombieProcess.__reduce__�NNNr r� r� r� r� r 7 s r c @ r ) � AccessDeniedz@Exception raised when permission to perform an action is denied.r� Nc C r )Nr� r r r� r� r� r O r zAccessDenied.__init__c C r r r r r� r� r� r U r zAccessDenied.__reduce__r r r� r� r� r� r J s r c @ r ) �TimeoutExpiredzWRaised on Process.wait(timeout) if timeout expires and process is still alive. r� Nc C s. t �| � || _|| _|| _d|� d�| _d S )Nztimeout after z seconds)r� r r r� r� r� )r� r r� r� r� r� r� r ` s zTimeoutExpired.__init__c C r r )r r r� r� r r� r� r� r g r zTimeoutExpired.__reduce__r r r� r� r� r� r Y s r c C s@ z t | �| d }W n ty Y dS w |dur t||�}|S )z5Calculate percentage usage of 'used' against 'total'.�d g N)�float�ZeroDivisionError�round)r� r� �round_�retr� r� r� rK p s � rK c s2 t ���� �fdd��}� fdd�}i � ||_|S )a� A simple memoize decorator for functions supporting (hashable) positional arguments. It also provides a cache_clear() function for clearing the cache: >>> @memoize ... def foo() ... return 1 ... >>> foo() 1 >>> foo.cache_clear() >>> It supports: - functions - classes (acts as a @singleton) - staticmethods - classmethods It does NOT support: - methods c sn | t t|�� ��f}z� | W S ty6 z�| i |�� }� |< W | Y S ty5 } z|d �d }~ww w r )� frozenset�sortedr� �KeyError� Exception)�args�kwargs�keyr �err��cache�funr� r� �wrapper� s ����zmemoize.<locals>.wrapperc s � � � dS )z Clear cache.N)�clearr� )r( r� r� � cache_clear� � z memoize.<locals>.cache_clear)� functools�wrapsr, )r) r* r, r� r' r� rH | s rH c s6 t �� �� fdd��}dd� }dd� }||_||_|S )a� A memoize decorator which is disabled by default. It can be activated and deactivated on request. For efficiency reasons it can be used only against class methods accepting no arguments. >>> class Foo: ... @memoize ... def foo() ... print(1) ... >>> f = Foo() >>> # deactivated (default) >>> foo() 1 >>> foo() 1 >>> >>> # activated >>> foo.cache_activate(self) >>> foo() 1 >>> foo() >>> foo() >>> c s� z| j � }W |S ty& z� | �W Y S ty% } z|d �d }~ww tyW z� | �}W n tyA } z|d �d }~ww z || j � < W Y |S tyV Y Y |S w w r )�_cache�AttributeErrorr" r! )r� r r& �r) r� r� r* � s2 ��� �� ���z'memoize_when_activated.<locals>.wrapperc S s i | _ dS )zsActivate cache. Expects a Process instance. Cache will be stored as a "_cache" instance attribute. N)r0 ��procr� r� r� �cache_activate� s z.memoize_when_activated.<locals>.cache_activatec S s z| ` W dS ty Y dS w )zDeactivate and clear cache.N)r0 r1 r3 r� r� r� �cache_deactivate� s �z0memoize_when_activated.<locals>.cache_deactivate)r. r/ r5 r6 )r) r* r5 r6 r� r2 r� �memoize_when_activated� s r7 c C s> zt �| �}W n ty � ty Y dS w t�|j�S )z�Same as os.path.isfile() but does not swallow EACCES / EPERM exceptions, see: http://mail.python.org/pipermail/python-dev/2012-June/120787.html. F)�os�stat�PermissionError�OSError�S_ISREG�st_mode)r� �str� r� r� rG � s � rG c C s4 zt �| � W dS ty � ty Y dS w )z�Same as os.path.exists() but does not swallow EACCES / EPERM exceptions. See: http://mail.python.org/pipermail/python-dev/2012-June/120787.html. FT)r8 r9 r: r; )r� r� r� r� rJ � s � �rJ c C sj t jrtdu r dS z!t � tt j��} | �d� W d � W dS 1 s#w Y W dS ty4 Y dS w )z2Return True if IPv6 is supported on this platform.NF)z::1r T)�socket�has_ipv6r r �bindr; )�sockr� r� r� rL s �� �rL c C sx i }d}t } | �d|�}||kr |S | �d||�}||kr7| ||� }| |d |� }|r3|�� }|||< |d }q)zCParse a C environ block of environment variables into a dictionary.r T� r� r~ )r �find�upper)�datar �pos�WINDOWS_�next_pos� equal_posr% r� r� r� r� rI s" � �rI c C �$ zt �| �W S ty | Y S w )z�Convert a numeric socket family value to an IntEnum member. If it's not a known member, return the numeric value itself. )r? � AddressFamily� ValueError��numr� r� r� rM 7 � �rM c C rK )zConvert a numeric socket type value to an IntEnum member. If it's not a known member, return the numeric value itself. )r? � SocketKindrM rN r� r� r� rN A rP rN c C s� |t jthv r|r t|� }|rt|� }|t jkr%|tthv r%|�|t�}nt}t|�}t|�}|du r<t | |||||�S t | ||||||�S )z2Convert a raw connection tuple to a proper ntuple.N) r? r r r� r �getr rM rN r2 r; )r� �fam�type_r� r� r� � status_mapr� r� r� r� rU K s rU c C sx ddl }| jr | js dS | jtjkr#t|j| j� d| j� �dd�j�S | jtj kr:t|j | j� d| j� �dd�j�S dS )zbGiven the address ntuple returned by ``net_if_addrs()`` calculates the broadcast address. r N�/F)�strict) � ipaddressr� r� r� r? r �str� IPv4Network�broadcast_addressr � IPv6Network)r� rX r� r� r� �broadcast_addr^ s$ �� ���r] c s � fdd�}|S )z�A decorator which can be used to mark a method as deprecated 'replcement' is the method name which will be called instead. c s> | j � d�� d�� | jd u r� | _t�| �� �fdd��}|S )Nz*() is deprecated and will be removed; use z () insteadc s$ t j� tdd� t| ��|i |��S )Nr} )�category� stacklevel)�warnings�warn�DeprecationWarningr� )r� r# r$ )r� � replacementr� r� �inner� s z/deprecated_method.<locals>.outer.<locals>.inner)r r r. r/ )r) rd �rc )r� r� �outery s �� z deprecated_method.<locals>.outerr� )rc rf r� re r� rF t s rF c @ sB e Zd ZdZdd� Zdd� Zdd� Zdd � Zdd d �Zd d� Z d S )� _WrapNumberszNWatches numbers so that they don't overflow and wrap (reset to zero). c C s t �� | _i | _i | _i | _d S r )� threading�Lock�lockr( � reminders� reminder_keysr r� r� r� r � s z_WrapNumbers.__init__c C sX || j vsJ �|| jvsJ �|| jvsJ �|| j |< t�t�| j|< t�t�| j|< d S r )r( rk rl r� � defaultdict�int�set)r� � input_dictr� r� r� r� � _add_dict� s z_WrapNumbers._add_dictc C s\ | j | }t|�� �t|�� � }|D ]}| j| | D ]}| j| |= q | j| |= qdS )z�In case the number of keys changed between calls (e.g. a disk disappears) this removes the entry from self.reminders. N)r( ro �keysrl rk )r� rp r� �old_dict� gone_keys�gone_key�remkeyr� r� r� �_remove_dead_reminders� s �z#_WrapNumbers._remove_dead_remindersc C s || j vr | �||� |S | �||� | j | }i }|D ]\}|| }z|| }W n ty5 |||< Y q w g }tt|��D ]3} || } || } || f} | | k re| j| | | 7 < | j| | �| � |� | | j| | � q>t |�||< q || j |< |S )zlCache dict and sum numbers which overflow and wrap. Return an updated copy of `input_dict`. ) r( rq rw r! �range�lenrk rl �add�append�tuple) r� rp r� rs �new_dictr% � input_tuple� old_tuple�bits�i� input_value� old_valuerv r� r� r� �run� s4 � z_WrapNumbers.runNc C s� | j �: |du r| j�� | j�� | j�� n | j�|d� | j�|d� | j�|d� W d � dS W d � dS 1 s@w Y dS )z>Clear the internal cache, optionally only for function 'name'.N)rj r( r+ rk rl �pop)r� r� r� r� r� r, � s �"�z_WrapNumbers.cache_clearc C s: | j � | j| j| jfW d � S 1 sw Y dS )z5Return internal cache dicts as a tuple of 3 elements.N)rj r( rk rl r r� r� r� � cache_info� s $�z_WrapNumbers.cache_infor ) r r� r� r r rq rw r� r, r� r� r� r� r� rg � s ' rg c C s6 t j� t �| |�W d � S 1 sw Y dS )z�Given an `input_dict` and a function `name`, adjust the numbers which "wrap" (restart from zero) across different calls by adding "old value" to "new value" and return an updated dict. N)�_wnrj r� )rp r� r� r� r� rO � s $�rO i � c C s t | dtd�S )N�rb)� buffering)�open�FILE_READ_BUFFER_SIZE)�fnamer� r� r� rQ � � rQ c C sH t | tttd�}zt|_W |S ty Y |S ty# |�� � w )zaOpen a file in text mode by using the proper FS encoding and en/decoding error handlers. )r� �encoding�errors)r� r� r0 r1 � _CHUNK_SIZEr1 r" �close)r� �fobjr� r� r� rP s � � ��rP c C s� |t u r || �� }|�� W d � S 1 sw Y dS z || �� }|�� W d � W S 1 s3w Y W dS tyE | Y S w )z�Read entire file content and return it as a string. File is opened in text mode. If specified, `fallback` is the value returned in case of error, either if the file does not exist or it can't be read(). N)�_DEFAULT�readr; )r� �fallback�_open�fr� r� r� rR s $� (� �rR c C s t | |td�S )z,Same as above but opens file in binary mode.)r� r� )rR rQ )r� r� r� r� r� rS , s rS �%(value).1f%(symbol)sc C s� d}i }t |dd� �D ]\}}d|d d > ||< q t|dd� �D ]}t| �|| kr<t| �|| }|t� S q#|t|d | d� S )z�Used by various scripts. See: https://code.activestate.com/recipes/578019-bytes-to-human-human-to-bytes-converter/?in=user-4178764. >>> bytes2human(10000) '9.8K' >>> bytes2human(100001221) '95.4M' ) �B�K�M�G�T�P�E�Z�Yr~ N� r )�symbolr� )� enumerate�reversed�absr �locals�dict)�nr� �symbols�prefixr� �sr� r� r� r� r� rT 1 s �rT c C s t jd jS )z+Return updated psutil.PROCFS_PATH constant.r� )�sys�modules� PROCFS_PATHr� r� r� r� �get_procfs_pathD r- r� c C s | j ttd�S )N)r� r� )�decoder0 r1 )r� r� r� r� r� I r� r� c C sX t jdkrdS zdd l}| �� sJ �|�� |�d�dks J �W dS ty+ Y dS w )Nr[ Tr �colorsF)r8 r� �curses�isatty� setupterm�tigetnumr" )�filer� r� r� r� rX R s ��rX Fc C s� t � s| S g }tddddddddd d � }d |d < z|| }W n ty6 d |�dt|�� �� �}t|�d �w |�|� |rC|�d� dd�|�� d| � d�S )z*Return an highlighted version of 'string'.�34�33�30�32�37�36�91�35�93) �blue�brown�darkgrey�green�grey� lightblue�red�violet�yellow�29N�invalid color z; choose amongst �1z[�;�mz[0m)rX r� r! �listrr rM r{ r� )r� �color�bold�attrr� r� r� r� r� rW b s2 � � rW c C s t � s t| |d� dS trtt| ||�|d� dS ddl}d}|jjj}|jjj}t ddddd�}||d< z|| }W n t yS d |�d t |� � ���} t | �d�w |r^|dkr^|d 7 }|tju red nd } |j|_|| �} || |� zt| |d� W || |� dS || |� w )z$Print a colorized version of string.�r� r N� r} � � )r� r� r� r� r� z; choose between � i�i����)rX �printr rW �ctypes�windll�Kernel32� GetStdHandle�SetConsoleTextAttributer� r! r� rr rM r� �stderr�c_ulong�restype) r� r� r� r� r� � DEFAULT_COLORr� r� r� r� � handle_id�handler� r� r� rY ~ s: � �� � rY c C sx t r:ddl}|�|�� j�\}}}}}t| t�r(t| t�r#d| � �} nd| ��} td|� d|� d| � �t j d� dS dS )z@If PSUTIL_DEBUG env var is set, print a debug message to stderr.r Nz ignoring zpsutil-debug [�:z]> r� ) r �inspect� getframeinfo� currentframe�f_back� isinstancer" r; r� r� r� )r� r� r� �lineno�_�_lines�_indexr� r� r� rV � s � ��rV r )r� )NF)�r r� �enumr. r8 r? r9 r� rh r` r r r r r � ImportErrorr �bool�getenvr �objectr� �__all__r� r r �platform� startswithr r r r r r r r �AIXr& r' r# r( r* r. r" r, r- r$ r% r+ r) r/ r r r r r r r r r r r r �IntEnumr| �globals�update� __members__r� �getfilesystemencodingr0 �getfilesystemencodeerrorsr1 rC r? r= r> r@ rD r; rA rB r< r� r� r� r� r3 r8 r9 r: r5 r6 r7 r4 r2 r� rE r" r� r r r r rK rH r7 rG rJ rL rI rM rN rU r] rF rg rO r� r, r� r� rQ rP rR rS rT r� r� �stdoutrX rW rY rV r� r� r� r� �<module> sB � � & ������� � �! -G W � ( __pycache__/_psaix.cpython-310.pyc 0000644 00000034717 15030211765 0012754 0 ustar 00 o �hLG � @ s� d Z ddlZddlZddlZddlZddlZddlZddlmZ ddl m Z ddl m Z ddl m Z ddl mZ dd l mZ dd l mZ dd l mZ dd l mZ dd l mZ ddl mZ ddl mZ ddl mZ ddl mZ ddl mZ dgZee d�Z ee d�Z ee d�Z e� � Z ej!Z!e j"e j#e j$e j%e j&e j'e j(e j'e j)e j*iZ+e j,e j-e j.e j/e j0e j1e j2e j3e j4e j5e j6e j7e j8e j9e j:e j;e j<e j=e j>e j?e j@e jAe jBe jCi ZDeEdddddddd d �ZFed d d g�ZGeGZHed!g d"��ZIed#g d$��ZJd%d&� ZKd'd(� ZLd)d*� ZMd+d,� ZNd-d.� ZOd/d0� ZPd1d2� ZQe jRZRe jSZSdGd4d5�ZTejUZUe �r(e jVZVdHd7d8�ZWd9d:� ZXd;d<� ZYd=d>� ZZd?d@� Z[dAdB� Z\dCdD� Z]G dEdF� dF�Z^dS )Iz AIX platform implementation.� N)� namedtuple� )�_common)�_psposix)� _psutil_aix)� _psutil_posix)�NIC_DUPLEX_FULL)�NIC_DUPLEX_HALF)�NIC_DUPLEX_UNKNOWN)� AccessDenied)� NoSuchProcess)� ZombieProcess)�conn_to_ntuple)�get_procfs_path)�memoize_when_activated)� usage_percent� PROCFS_PATH� proc_threads�net_io_counters�proc_io_counters� � � � � � )�ppid�rss�vms� create_time�nice� num_threads�status�ttynr�pmemr r � scputimes)�user�system�idle�iowait�svmem)�total� available�percent�used�freec C s4 t �� \} }}}}t| | | dd�}t| ||||�S )Nr ��round_)�cext� virtual_memr r* )r+ �availr/ �_pinned�inuser- � r7 �@/usr/local/CyberCP/lib/python3.10/site-packages/psutil/_psaix.py�virtual_memoryg s r9 c C s: t �� \} }}}| | }t|| dd�}t�| |||||�S )z=Swap system memory as a (total, used, free, sin, sout) tuple.r r0 )r2 �swap_memr r �sswap)r+ r/ �sin�soutr. r- r7 r7 r8 � swap_memorym s r> c C s t �� } tdd� t| � D �� S )z.Return system-wide CPU times as a named tuple.c S s g | ]}t |��qS r7 )�sum��.0�xr7 r7 r8 � <listcomp>} � z cpu_times.<locals>.<listcomp>)r2 � per_cpu_timesr% �zip��retr7 r7 r8 � cpu_timesz s rI c C s t �� } dd� | D �S )z6Return system per-CPU times as a list of named tuples.c S s g | ]}t |� �qS r7 )r% r@ r7 r7 r8 rC � rD z!per_cpu_times.<locals>.<listcomp>)r2 rE rG r7 r7 r8 rE � s rE c C s" zt �d�W S ty Y dS w )z0Return the number of logical CPUs in the system.�SC_NPROCESSORS_ONLNN)�os�sysconf� ValueErrorr7 r7 r7 r8 �cpu_count_logical� s �rN c C sv g d�} t j| t jt jd�}|�� \}}dd� ||fD �\}}|jdkr/| �d|� �}t|��|�� �� }t|�p:d S )N)�lsdevz-Cc� processor��stdout�stderrc s � � | ] }|� tjj�V qd S �N��decode�sysrR �encodingr@ r7 r7 r8 � <genexpr>� s � z"cpu_count_cores.<locals>.<genexpr>r z command error ) � subprocess�Popen�PIPE� communicate� returncode� RuntimeError�strip� splitlines�len)�cmd�prR rS �msg� processorsr7 r7 r8 �cpu_count_cores� s rh c C s t �� \} }}}t�| |||�S )z*Return various CPU stats as a named tuple.)r2 � cpu_statsr � scpustats)� ctx_switches� interrupts�soft_interrupts�syscallsr7 r7 r8 ri � s �ri Fc C s\ g }t �� }|D ]#}|\}}}}|dkrd}| s t|�js qt�||||�}|�|� q|S )z Return system disk partitions.�none� )r2 �disk_partitions� disk_usager+ r � sdiskpart�append) �all�retlist� partitions� partition�device� mountpoint�fstype�opts�ntupler7 r7 r8 rq � s rq ���c C s~ t j| \}}t�|�}g }|D ],}|\}}} } } } } ||vr q| |vr%qt||| | | | t|dkr3| ndd�}|�|� q|S )z�Return socket connections. If pid == -1 return system-wide connections (as opposed to connections opened by one process only). r~ N��pid)r � conn_tmapr2 �net_connectionsr � TCP_STATUSESrt )�kind�_pid�families�types�rawlistrH �item�fd�fam�type_�laddr�raddrr" r� �ntr7 r7 r8 r� � s* � r� c C s� t td�} dd� t� D �}i }|D ]c}t�|�}t�|�}d}d}tjdd|gtjtjd�}|� � \} } d d � | | fD �\} } |j dkrZt � d | �} | d urZt | �d ��}| �d�}d�|�} d|v } | �|t�}t�| |||| �||< q|S )z)Get NIC stats (isup, duplex, speed, mtu).)�Full�Halfc S s h | ]}|d �qS )r r7 r@ r7 r7 r8 � <setcomp>� rD z net_if_stats.<locals>.<setcomp>rp r z/usr/bin/entstatz-drQ c s rT rU rV r@ r7 r7 r8 rZ � � � �z net_if_stats.<locals>.<genexpr>z"Running: (\d+) Mbps.*?(\w+) DuplexNr r �,�running)r r � net_if_addrs� cext_posix� net_if_mtu� net_if_flagsr[ r\ r] r^ r_ �re�search�int�group�join�getr r � snicstats)� duplex_map�namesrH �name�mtu�flags�duplex�speedre rR rS � re_result� output_flags�isupr7 r7 r8 � net_if_stats� s: � � � r� c C s t �� S )z:The system boot time expressed in seconds since the epoch.)r2 � boot_timer7 r7 r7 r8 r� s r� c C s\ g } t �� }d}|D ]!}|\}}}}}} |sq ||v r d}t�||||| �} | �| � q | S )z:Return currently connected users as a list of namedtuples.)z:0.0z:0� localhost)r2 �usersr �suserrt ) rv r� r� r� r& �tty�hostname�tstamp� user_processr� r� r7 r7 r8 r� s r� c C s dd� t �t� �D �S )z7Returns a list of PIDs currently running on the system.c S s g | ] }|� � rt|��qS r7 )�isdigitr� r@ r7 r7 r8 rC 1 s zpids.<locals>.<listcomp>)rK �listdirr r7 r7 r7 r8 �pids/ s r� c C s t j�t j�t� t| �d��S )z&Check for the existence of a unix pid.�psinfo)rK �path�existsr� r �strr r7 r7 r8 � pid_exists4 s r� c s t �� �� fdd��}|S )z�Call callable into a try/except clause and translate ENOENT, EACCES and EPERM in NoSuchProcess or AccessDenied exceptions. c s� | j | j| j}}}z � | g|�R i |��W S ttfy6 } zt|�s+t||�|�t|||�|�d }~w tyG } zt ||�|�d }~ww rU ) r� �_ppid�_name�FileNotFoundError�ProcessLookupErrorr� r r �PermissionErrorr )�self�args�kwargsr� r r� �err��funr7 r8 �wrapper> s � ��z wrap_exceptions.<locals>.wrapper)� functools�wraps)r� r� r7 r� r8 �wrap_exceptions9 s r� c @ st e Zd ZdZg d�Zdd� Zdd� Zdd� Zee d d � ��Z ee d d � ��Z ed d� �Z edd� �Z edd� �Zedd� �Zedd� �Zedd� �ZerRedd� �Zed=d d ��Zed d � �Zed d!� �Zed"d#� �Zed$d%� �Zed&d'� �Zed(d)� �Zed*d+� �Zed,d-� �Z ed.d/� �Z e Z ed0d1� �Z d2d3� Z ed4d5� �Z!ed6d7� �Z"ed>d9d:��Z#e$r�ed;d<� �Z%d8S d8S )?�Processz1Wrapper class around underlying C implementation.)�_cacher� r� � _procfs_pathr� c C s || _ d | _d | _t� | _d S rU )r� r� r� r r� )r� r� r7 r7 r8 �__init__U s zProcess.__init__c C � | j �| � | j�| � d S rU )�_proc_basic_info�cache_activate� _proc_cred�r� r7 r7 r8 � oneshot_enter[ � zProcess.oneshot_enterc C r� rU )r� �cache_deactivater� r� r7 r7 r8 � oneshot_exit_ r� zProcess.oneshot_exitc C � t �| j| j�S rU )r2 �proc_basic_infor� r� r� r7 r7 r8 r� c � zProcess._proc_basic_infoc C r� rU )r2 � proc_credr� r� r� r7 r7 r8 r� h r� zProcess._proc_credc C s$ | j dkrdS t�| j | j��d�S )Nr �swapper� )r� r2 � proc_namer� �rstripr� r7 r7 r8 r� m s z Process.namec C s� | � � }|sdS |d }tjj|v r@tj�|�s%tj�tj�| �� |��}tj�|�r:tj�|�r:t� |tj �r:|S tj� |�}tj d � d�D ] }tj�tj�||��}tj�|�rft� |tj �rf| S qHdS )Nrp r �PATH�:)�cmdlinerK r� �sep�isabs�abspathr� �cwd�isfile�access�X_OK�basename�environ�split)r� r� �exer� � possible_exer7 r7 r8 r� t s. � � � ��z Process.exec C � t �| j�S rU )r2 � proc_argsr� r� r7 r7 r8 r� � � zProcess.cmdlinec C r� rU )r2 � proc_environr� r� r7 r7 r8 r� � r� zProcess.environc C � | � � td S )Nr �r� � proc_info_mapr� r7 r7 r8 r � � zProcess.create_timec C r� )Nr! r� r� r7 r7 r8 r! � r� zProcess.num_threadsc C sX t �| j�}g }|D ]\}}}t�|||�}|�|� q |s*t�| j� d| j� �� |S )N�/) r2 r r� r �pthreadrt rK �statr� )r� r� rv � thread_id�utime�stimer} r7 r7 r8 �threads� s zProcess.threads�inetc C s. t || jd�}|st�| j� d| j� �� |S )N)r� r� )r� r� rK r� r� )r� r� rH r7 r7 r8 r� � s zProcess.net_connectionsc C r� rU )r� � getpriorityr� r� r7 r7 r8 �nice_get� r� zProcess.nice_getc C s t �| j|�S rU )r� � setpriorityr� )r� �valuer7 r7 r8 �nice_set� s zProcess.nice_setc C s | � � td | _| jS )Nr )r� r� r� r� r7 r7 r8 r � s z Process.ppidc C s" | � � \}}}}}}t�|||�S rU �r� r �puids)r� �real� effective�saved�_r7 r7 r8 �uids� � z Process.uidsc C s" | � � \}}}}}}t�|||�S rU r )r� r r r r r7 r7 r8 �gids� r z Process.gidsc C s t �| j| j�}tj|� S rU )r2 �proc_cpu_timesr� r� r � pcputimes)r� �tr7 r7 r8 rI � s zProcess.cpu_timesc C sP | � � td }|d@ d? |d@ B }t�d�D ]}t�|�j|kr%| S qd S )Nr# l � � i�� z /dev/**/*)r� r� �globrK r� �st_rdev)r� �ttydev�devr7 r7 r8 �terminal� s �zProcess.terminalc C sV | j }zt�|� d| j� d��}|�d�W S ty* t�|� d| j� �� Y dS w )Nr� z/cwdrp )r� rK �readlinkr� r� r� r� )r� � procfs_path�resultr7 r7 r8 r� � s �z Process.cwdc C s2 | � � }|td d }|td d }t||�S )Nr i r )r� r� r$ )r� rH r r r7 r7 r8 � memory_info� s zProcess.memory_infoc C s | � � td }t�|d�S )Nr" �?)r� r� � PROC_STATUSESr� )r� �coder7 r7 r8 r" � s zProcess.statusc C s� t jddt| j�gt jt jd�}|�� \}}dd� ||fD �\}}d|�� v r.t| j| j��t � d|�}g }|D ]%\}}|� � }|� d�rK|d d � }|�� d krRq8|� t�|t|��� q8|S ) Nz/usr/bin/procfilesz-nrQ c s rT rU rV r@ r7 r7 r8 rZ r� z%Process.open_files.<locals>.<genexpr>zno such processz(\d+): S_IFREG.*name:(.*)\nz//r zcannot be retrieved)r[ r\ r� r� r] r^ �lowerr r� r� �findallra � startswithrt r � popenfiler� )r� re rR rS � procfilesrv r� r� r7 r7 r8 � open_files s* � � zProcess.open_filesc C s, | j dkrdS tt�| j� d| j � d���S )Nr r� z/fd)r� rc rK r� r� r� r7 r7 r8 �num_fds s zProcess.num_fdsc C s t jt�| j�� S rU )r �pctxswr2 �proc_num_ctx_switchesr� r� r7 r7 r8 �num_ctx_switches! � zProcess.num_ctx_switchesNc C s t �| j|| j�S rU )r �wait_pidr� r� )r� �timeoutr7 r7 r8 �wait% r' z Process.waitc C s^ z t �| j�\}}}}W n ty& } zt| j�s!t| j| j�|�� d }~ww t�||||�S rU ) r2 r r� �OSErrorr� r r� r �pio)r� �rc�wc�rb�wbr� r7 r7 r8 � io_counters+ s ��zProcess.io_counters)r� rU )&�__name__� __module__� __qualname__�__doc__� __slots__r� r� r� r� r r� r� r� r� r� r� r r! � HAS_THREADSr� r� r r r r r rI r r� r �memory_full_infor" r" r# r& r* �HAS_PROC_IO_COUNTERSr1 r7 r7 r7 r8 r� P sz �r� )F)r~ )_r5 r� r rK r� r[ rX � collectionsr rp r r r r2 r r� r r r r r r r r r r �__extra__all__�hasattrr7 �HAS_NET_IO_COUNTERSr9 � getpagesize� PAGE_SIZE�AF_LINK�SIDL� STATUS_IDLE�SZOMB� STATUS_ZOMBIE�SACTIVE�STATUS_RUNNING�SSWAP�SSTOP�STATUS_STOPPEDr �TCPS_ESTABLISHED�CONN_ESTABLISHED� TCPS_SYN_SENT� CONN_SYN_SENT� TCPS_SYN_RCVD� CONN_SYN_RECV�TCPS_FIN_WAIT_1�CONN_FIN_WAIT1�TCPS_FIN_WAIT_2�CONN_FIN_WAIT2�TCPS_TIME_WAIT�CONN_TIME_WAIT� TCPS_CLOSED� CONN_CLOSE�TCPS_CLOSE_WAIT�CONN_CLOSE_WAIT� TCPS_LAST_ACK� CONN_LAST_ACK� TCPS_LISTEN� CONN_LISTEN� TCPS_CLOSING� CONN_CLOSING�PSUTIL_CONN_NONE� CONN_NONEr� �dictr� r$ �pfullmemr% r* r9 r> rI rE rN rh ri �disk_io_countersrr rq r� r r� r� r� r� r� r� r� r� r7 r7 r7 r8 �<module> s� � �� + __pycache__/_pswindows.cpython-310.pyc 0000644 00000066032 15030211765 0013660 0 ustar 00 o �hm� � @ s� d Z ddlZddlZddlZddlZddlZddlZddlZddlm Z ddl m Z ddl m Z ddl m Z ddl mZ dd l mZ dd l mZ dd l mZ dd l mZ dd l mZ ddl mZ ddl mZ ddl mZ ddl mZ ddlmZ ddlmZ ddlmZ ddlm Z ddlm Z ddlm Z zddl mZ W n- e y� Z! z!e"e!��#� �$d�r�e�%� d dk r�dZ&e&d 7 Z&e&d 7 Z&e'e&�e!�� dZ![!ww g d �Z(d Z)d Z*d!ej+v Z,e�-d"d#d$i�Z.e.j/Z/e j0e j1e j2e j3e j4e j5e j6e j7e j8e j9e j:e j;e j<e j=e j>e j?e j@e jAe jBe jCe jDe jEe jFe)e jGe jHi ZIG d%d&� d&ej-�ZJeK� �LeJjM� G d'd(� d(ej-�ZNeK� �LeNjM� eOd�i d)d�d*d�d+d,�d-d.�d/d0�d1d2�d3d�d4d5�d6d7�d8d9�d:d;�d<d=�d>d?�d@dA�dBdC�dDdE�dFdG�dHdI�dJdK�dLdM�dNdO�dPdQ��ZPe dRg dS��ZQe dTg dU��ZRe dVg dW��ZSe dXeSjTdY �ZUe dZd[d\g�ZVe d]d^d_�WeVjT� �ZXe d`g da��ZYejZdbdc�ddde� �Z[edfdg� �Z\dhdi� Z]djdk� Z^e j_Z_dldm� Z`dndo� Zadpdq� Zbdrds� Zcdtdu� Zddvdw� Zedxdy� Zfdzd{� Zgd|ahd}d~� Zid�dd��Zjd�d�� Zkd�d�� Zld�d�� Zmd�d�� Zndaod�d�� Zpd�d�� Zqd�d�� Zrd�d�� ZsG d�d�� d��Zte juZue jvZve jwZwd�d�� Zxd�d�d��Zyd�d�� Zzd�d�� Z{G d�d�� d��Z|dS )�z Windows platform implementation.� N)� namedtuple� )�_common)�ENCODING)� AccessDenied)� NoSuchProcess)�TimeoutExpired)� conn_tmap)�conn_to_ntuple)�debug)� isfile_strict)�memoize)�memoize_when_activated)�parse_environ_block)� usage_percent)�ABOVE_NORMAL_PRIORITY_CLASS)�BELOW_NORMAL_PRIORITY_CLASS)�HIGH_PRIORITY_CLASS)�IDLE_PRIORITY_CLASS)�NORMAL_PRIORITY_CLASS)�REALTIME_PRIORITY_CLASS)�_psutil_windowszdll load failed� z3this Windows version is too old (< Windows Vista); z:psutil 3.4.2 is the latest version which supports Windows z2000, XP and 2003 server)�win_service_iter�win_service_getr r r r r r �IOPRIO_VERYLOW� IOPRIO_LOW� IOPRIO_NORMAL� IOPRIO_HIGH�CONN_DELETE_TCB�AF_LINK� DELETE_TCBi+ �__pypy__� AddressFamilyr ���c @ s$ e Zd ZeZeZeZeZeZeZdS )�PriorityN) �__name__� __module__� __qualname__r r r r r r � r) r) �D/usr/local/CyberCP/lib/python3.10/site-packages/psutil/_pswindows.pyr% e s r% c @ s e Zd ZdZdZdZdZdS )� IOPriorityr r � � N)r& r' r( r r r r r) r) r) r* r+ q s r+ � num_handles� ctx_switches� user_timer, � kernel_timer- � create_time� � num_threads� � io_rcount� io_wcount� � io_rbytes� � io_wbytes� �io_count_others� �io_bytes_others� �num_page_faults� � peak_wset� �wset� �peak_paged_pool� � paged_pool� �peak_non_paged_pool� �non_paged_pool� �pagefile� � peak_pagefile� � mem_private� � scputimes)�user�system�idle� interrupt�dpc�svmem)�total� available�percent�used�free�pmem) �rss�vmsrA rC rE rG rI �peak_nonpaged_pool� nonpaged_poolrO rQ �private�pfullmem)�uss� pmmap_grouped�pathrb � pmmap_extz addr perms � �pio)� read_count� write_count� read_bytes� write_bytes� other_count� other_bytesi )�maxsizec C s@ d� | �d�dd� �}t�|�}| t|�d� }tj� ||�S )z�Convert paths using native DOS format like: "\Device\HarddiskVolume1\Windows\systemew\file.txt" into: "C:\Windows\systemew\file.txt". �\Nr- )�join�split�cext�QueryDosDevice�len�osrj )�s�rawdrive� driveletter� remainderr) r) r* �convert_dos_path� s r� c C � t �� S �N)rx � getpagesizer) r) r) r* r� � � r� c C sJ t �� } | \}}}}|}|}|}|| }t|| |dd�} t||| ||�S )z&System virtual memory as a namedtuple.r ��round_)rx � virtual_memr r[ ) �mem�totphys� availphys�_totsys� _availsysr\ �availr` r_ r^ r) r) r* �virtual_memory� s r� c C sp t �� } | d }| d }|| }|dkr!t �� }td| | �}nd}d}|| }t|d�}t�||||dd�S )z=Swap system memory as a (total, used, free, sin, sout) tuple.r r, g{�G�z�?� r )rx r� � swap_percent�int�roundr �sswap)r� � total_phys� total_systemr\ � percentswapr_ r` r^ r) r) r* � swap_memory� s r� c C sL t | t�r | jtdd�} t�| �\}}|| }t||dd�}t�||||�S )z'Return disk usage associated with path.�strict)�errorsr r� ) � isinstance�bytes�decoder rx � disk_usager r � sdiskusage)rj r\ r` r_ r^ r) r) r* r� � s r� c C s t �| �}dd� |D �S )zReturn disk partitions.c S s g | ]}t j|� �qS r) )r � sdiskpart)�.0�xr) r) r* � <listcomp> s z#disk_partitions.<locals>.<listcomp>)rx �disk_partitions)�all�rawlistr) r) r* r� s r� c C s< t �� \} }}tdd� tt �� � D �� }t| |||j|j�S )z)Return system CPU times as a named tuple.c S s g | ]}t |��qS r) )�sum)r� �nr) r) r* r� s z cpu_times.<locals>.<listcomp>)rx � cpu_timesrU �zip� per_cpu_timesrY rZ )rV rW rX � percpu_summedr) r) r* r� s �r� c C s: g } t �� D ]\}}}}}t|||||�}| �|� q| S )z6Return system per-CPU times as a list of named tuples.)rx r� rU �append)�retrV rW rX rY rZ �itemr) r) r* r� s r� c C r� )z0Return the number of logical CPUs in the system.)rx �cpu_count_logicalr) r) r) r* r� ) r� r� c C r� )z-Return the number of CPU cores in the system.)rx �cpu_count_coresr) r) r) r* r� . r� r� c C s$ t �� \} }}}d}t�| |||�S )zReturn CPU statistics.r )rx � cpu_statsr � scpustats)r/ � interrupts�_dpcs�syscalls�soft_interruptsr) r) r* r� 3 s �r� c C s( t �� \} }d}t�t| �|t|��gS )zMReturn CPU frequency. On Windows per-cpu frequency is not supported. r� )rx �cpu_freqr �scpufreq�float)�curr�max_�min_r) r) r* r� < s r� Fc C s* t st�� da t�� } tdd� | D ��S )z�Return the number of processes in the system run queue averaged over the last 1, 5, and 15 minutes respectively as a tuple. Tc s s � | ]}t |d �V qdS )r, N)r� )r� �loadr) r) r* � <genexpr>T s � z getloadavg.<locals>.<genexpr>)�_loadavg_inititializedrx �init_loadavg_counter� getloadavg�tuple)� raw_loadsr) r) r* r� H s r� c C sr t | \}}t�|||�}t� }|D ]"}|\}}} } } } } t||| | | | t|dkr+| ndd�}|�|� qt|�S )z�Return socket connections. If pid == -1 return system-wide connections (as opposed to connections opened by one process only). r$ N)�pid)r rx �net_connections�setr � TCP_STATUSES�add�list)�kind�_pid�families�typesr� r� r� �fd�fam�type�laddr�raddr�statusr� �ntr) r) r* r� \ s" � r� c C sX i } t �� }|�� D ] \}}|\}}}}ttd�r t�|�}t�||||d�| |< q | S )z)Get NIC stats (isup, duplex, speed, mtu).� NicDuplex� )rx � net_if_stats�items�hasattrr r� � snicstats)r� �rawdict�namer� �isup�duplex�speed�mtur) r) r* r� s s r� c C r� )zsReturn network I/O statistics for every network interface installed on the system as a dict of raw tuples. )rx �net_io_countersr) r) r) r* r� s r� c C r� )z,Return the addresses associated to each NIC.)rx � net_if_addrsr) r) r) r* r� � r� r� c C sd t �� \} }}}| dk}t|d@ �}t|d@ �}|r dS |s |r$tj}n|dkr+tj}t�|||�S )zReturn battery information.r � r: Nr$ )rx �sensors_battery�boolr �POWER_TIME_UNLIMITED�POWER_TIME_UNKNOWN�sbattery)� acline_status�flagsr^ �secsleft� power_plugged� no_battery�chargingr) r) r* r� � s r� c C s( t t�� �} t| t �dkrtS | a| S )z:The system boot time expressed in seconds since the epoch.r )r� rx � boot_time�abs� _last_btime)r� r) r) r* r� � s r� c C s@ g } t �� }|D ]}|\}}}t�|d||d�}| �|� q| S )z:Return currently connected users as a list of namedtuples.N)rx �usersr �suserr� )�retlistr� r� rV �hostname�tstampr� r) r) r* r� � s r� c c s$ � t �� D ] \} }t| |�V qdS )z*Yields a list of WindowsService instances.N)rx �winservice_enumerate�WindowsService)r� � display_namer) r) r* r � s ��r c C s t | d�}|�� d |_|S )zBOpen a Windows service and return it as a WindowsService instance.Nr� )r� � _query_config� _display_name)r� �servicer) r) r* r � s r c @ s� e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd d � Z dd� Z e j dd� �Z dd� Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d!� Zd"d#� Zd$S )%r� z(Represents an installed Windows service.c C s || _ || _d S r� )�_namer� )�selfr� r� r) r) r* �__init__� s zWindowsService.__init__c C s& d| j �d| j�d�}| jj� |� �S )Nz(name=z, display_name=�))r� r� � __class__r& )r� �detailsr) r) r* �__str__� s zWindowsService.__str__c C s d| � � � dt| �� d�S )N�<z at �>)r �id�r� r) r) r* �__repr__� s zWindowsService.__repr__c C s t |t�stS | j|jkS r� )r� r� �NotImplementedr� �r� �otherr) r) r* �__eq__� s zWindowsService.__eq__c C s | |k S r� r) r r) r) r* �__ne__� s zWindowsService.__ne__c C sL | � � � t�| j�\}}}}W d � n1 sw Y t||||d�S )N)r� �binpath�username� start_type)�_wrap_exceptionsrx �winservice_query_configr� �dict)r� r� r r r r) r) r* r� � s � ��z WindowsService._query_configc C sP | � � � t�| j�\}}W d � n1 sw Y |dkr"d }t||d�S )Nr )r� r� )r rx �winservice_query_statusr� r )r� r� r� r) r) r* � _query_status� s � z WindowsService._query_statusc c s� � zdV W dS t y@ } z-| j}t|�r$d|�d�}td||d�|�|jtjtjhv r;d|�d�}td||d�|�� d}~ww )z{Ctx manager which translates bare OSError and WindowsError exceptions into NoSuchProcess and AccessDenied. Nzservice z( is not querable (not enough privileges)�r� r� �msgz does not exist) �OSErrorr� �is_permission_errr �winerrorrx �ERROR_INVALID_NAME� ERROR_SERVICE_DOES_NOT_EXISTr )r� �errr� r r) r) r* r s$ � �� ��z WindowsService._wrap_exceptionsc C � | j S )z�The service name. This string is how a service is referenced and can be passed to win_service_get() to get a new WindowsService instance. )r� r r) r) r* r� s zWindowsService.namec C r )z_The service display name. The value is cached when this class is instantiated. )r� r r) r) r* r� # s zWindowsService.display_namec C � | � � d S )zwThe fully qualified path to the service binary/exe file as a string, including command line arguments. r �r� r r) r) r* r ) � zWindowsService.binpathc C r )z,The name of the user that owns this service.r r r r) r) r* r / � zWindowsService.usernamec C r )zRA string which can either be "automatic", "manual" or "disabled". r r r r) r) r* r 3 r! zWindowsService.start_typec C r )zzThe process PID, if any, else None. This can be passed to Process class to control the service's process. r� �r r r) r) r* r� ; r! zWindowsService.pidc C r )zService status as a string.r� r# r r) r) r* r� A r" zWindowsService.statusc C s t �| �� �S )zService long description.)rx �winservice_query_descrr� r r) r) r* � descriptionE � zWindowsService.descriptionc C s> | � � }|�| �� � | �� |d< | �� |d< | �� |d< |S )zUUtility method retrieving all the information above as a dictionary. r� r� r% )r� �updater r� r� r% )r� �dr) r) r* �as_dictK s zWindowsService.as_dictN)r& r' r( �__doc__r� r r r r r� r � contextlib�contextmanagerr r� r� r r r r� r� r% r) r) r) r) r* r� � s( r� c C s. t | t�s J | ��t | t�p| jtjtjhv S )z*Return True if this is a permission error.)r� r �PermissionErrorr rx �ERROR_ACCESS_DENIED�ERROR_PRIVILEGE_NOT_HELD)�excr) r) r* r � s �r c C s@ t | t�s J | ��t| �rt||d�S t | t�r t||d�S | �)z3Convert OSError into NoSuchProcess or AccessDenied.�r� r� )r� r r r �ProcessLookupErrorr )r0 r� r� r) r) r* �convert_oserror� s r3 c � t �� �� fdd��}|S )zDDecorator which converts OSError into NoSuchProcess or AccessDenied.c sF z � | g|�R i |��W S t y" } z t|| j| jd�|�d }~ww )Nr1 )r r3 r� r� )r� �args�kwargsr ��funr) r* �wrapper� s ��z wrap_exceptions.<locals>.wrapper�� functools�wraps�r8 r9 r) r7 r* �wrap_exceptions� s r> c r4 )z�Workaround for https://github.com/giampaolo/psutil/issues/875. See: https://stackoverflow.com/questions/4457745#4457745. c s� d}d}t |�D ]6}z� | g|�R i |��W S ty> } z|}|jtkr9t�|� t|d d�}W Y d }~q� d }~ww � � d|� d|� �}t| j| j |d��)N�-C ��6?�! r, �{�G�z�?z retried z: times, converted to AccessDenied as it's still returning r ) �ranger r �ERROR_PARTIAL_COPY�time�sleep�minr r� r� )r� r5 r6 �delay�times�_r r r7 r) r* r9 � s& �� ��z)retry_error_partial_copy.<locals>.wrapperr: r= r) r7 r* �retry_error_partial_copy� s rJ c @ s� e Zd ZdZg d�Zdd� Zdd� Zdd� Zed d � �Z d d � Z e ed d� ��Z e e dd� ��Ze e dd� ��Zdd� Zdd� Ze dd� �Ze dd� �Zdd � Ze d d � �Ze d d � �Ze dNd"d#��Ze d$d%� �Ze dOd'd(��Ze d)d*� �Ze d+d,� �Ze d-d.� �Z e d/d0� �Z e d1d2� �Z e e d3d4� ��Z e d5d6� �Z e dPd8d9��Z!e d:d;� �Z"e d<d=� �Z#e d>d?� �Z$e d@dA� �Z%e dBdC� �Z&e dDdE� �Z'e dFdG� �Z(e dHdI� �Z)e dJdK� �Z*e dLdM� �Z+d!S )Q�Processz1Wrapper class around underlying C implementation.)�_cacher� �_ppidr� c C s || _ d | _d | _d S r� )r� r� rM )r� r� r) r) r* r� � s zProcess.__init__c C � | j �| � | j�| � d S r� )� _proc_info�cache_activate�exer r) r) r* � oneshot_enter� � zProcess.oneshot_enterc C rN r� )rO �cache_deactivaterQ r r) r) r* � oneshot_exit� rS zProcess.oneshot_exitc C s$ t �| j�}t|�tt�ksJ �|S )zOReturn multiple information about this process as a raw tuple. )rx � proc_infor� rz � pinfo_map�r� r� r) r) r* rO � s zProcess._proc_infoc C s, | j dkrdS | j dkrdS tj�| �� �S )zbReturn process name, which on Windows is always the final part of the executable. r zSystem Idle Processr3 �System)r� r{ rj �basenamerQ r r) r) r* r� � s z Process.namec C sz t r,zt�| j�}W n' ty+ } z|jdkr&t|�d�� t| j| j�|�� d }~ww t�| j�}|� d�r;t |�S |S )N� z translated into AccessDeniedru ) �PYPYrx �proc_exer� r �errnor r r� � startswithr� )r� rQ r r) r) r* rQ � s �� z Process.exec C sn t jt jkr/z t j| jdd�W S ty. } zt|�r)t j| jdd�W Y d }~S � d }~ww t j| jdd�S )NT)�use_pebF)rx �WINVER� WINDOWS_8_1� proc_cmdliner� r r �r� r r) r) r* �cmdline s ��zProcess.cmdlinec C s t �| j�}t|�S r� )rx � proc_environr� r )r� r| r) r) r* �environ s zProcess.environc C s. zt � | j W S ty t| j| j�d �w r� )�ppid_mapr� �KeyErrorr r� r r) r) r* �ppid s �z Process.ppidc C s� zt �| j�W S tyZ } zGt|�rUtd� | �� }|td |td |td |td |td |td |td |td |td |td f W Y d }~S � d }~ww ) Nz*attempting memory_info() fallback (slower)rA rC rE rG rI rK rM rO rQ rS )rx �proc_memory_infor� r r r rO rW )r� r �infor) r) r* �_get_raw_meminfo s( � ��zProcess._get_raw_meminfoc C s( | � � }|d }|d }t||f| � S )Nr, r8 )rm ra )r� �trb rc r) r) r* � memory_info3 s zProcess.memory_infoc C s, | � � }t�| j�}|t� 9 }t||f � S r� )ro rx �proc_memory_ussr� r� rg )r� � basic_memrh r) r) r* �memory_full_info= s zProcess.memory_full_infoc c sr � zt �| j�}W n ty } z t|| j| j�|�d }~ww |D ]\}}}}t|�}t|�}||||fV q!d S r� )rx �proc_memory_mapsr� r r3 r� r� �hex)r� �rawr �addr�permrj rb r) r) r* � memory_mapsD s ����zProcess.memory_mapsc C s t �| j�S r� )rx � proc_killr� r r) r) r* �killQ r" z Process.killc C sH |t jkr t�| j� d S |t jt jhv r t�| j|� d S d}t |��)NzPonly SIGTERM, CTRL_C_EVENT and CTRL_BREAK_EVENT signals are supported on Windows) �signal�SIGTERMrx ry r� � CTRL_C_EVENT�CTRL_BREAK_EVENTr{ rz � ValueError)r� �sigr r) r) r* � send_signalU s �zProcess.send_signalNc C s� |d u rt j}nt|d �}ttdtj�}|d ur |� | nd }z t �| j|�}W n! t jy? } z t|| j| j�|�d }~w t j yJ d }Y nw d} t | j�sU|S |re|� |kret|| j| jd��t� |� t |d d�}qN)Ni� � monotonicr? Tr1 r, rA ) rx �INFINITEr� �getattrrD � proc_waitr� r r� �TimeoutAbandoned� pid_existsrE rF )r� �timeout� cext_timeout�timer�stop_at� exit_coder rG r) r) r* �waitb s, � � �z Process.waitc C s, | j dv rdS t�| j �\}}|� d|� �S )N� r r3 zNT AUTHORITY\SYSTEMru )r� rx � proc_username)r� �domainrV r) r) r* r � s zProcess.usernameFc C sf z t �| j�\}}}|W S ty2 } zt|�r-|r� td� | �� td W Y d }~S � d }~ww )Nz*attempting create_time() fallback (slower)r2 )rx � proc_timesr� r r r rO rW )r� � fast_only�_user�_system�createdr r) r) r* r2 � s ��zProcess.create_timec C s | � � td S )Nr4 )rO rW r r) r) r* r4 � s zProcess.num_threadsc C s< t �| j�}g }|D ]\}}}t�|||�}|�|� q |S r� )rx � proc_threadsr� r �pthreadr� )r� r� r� � thread_id�utime�stime�ntupler) r) r* �threads� s zProcess.threadsc C s~ z t �| j�\}}}W n+ ty6 } z t|�s� td� | �� }|td }|td }W Y d }~nd }~ww t� ||dd�S )Nz(attempting cpu_times() fallback (slower)r0 r1 r� ) rx r� r� r r r rO rW r � pcputimes)r� rV rW �_createdr rl r) r) r* r� � s ��zProcess.cpu_timesc C � t �| jd� d S )NT�rx �proc_suspend_or_resumer� r r) r) r* �suspend� � zProcess.suspendc C r� )NFr� r r) r) r* �resume� r� zProcess.resumec C s0 | j dv r t| j | j��t�| j �}tj�|�S )Nr� )r� r r� rx �proc_cwdr{ rj �normpath)r� rj r) r) r* �cwd� s z Process.cwdc C sX | j dv rg S t� }t�| j �}|D ]}t|�}t|�r't�|d�}|�|� qt |�S )Nr� r$ ) r� r� rx �proc_open_filesr� r r � popenfiler� r� )r� r� �raw_file_names�filer� r) r) r* � open_files� s �zProcess.open_files�inetc C s t || jd�S )N)r� )r� r� )r� r� r) r) r* r� � r& zProcess.net_connectionsc C � t �| j�}t|�}|S r� )rx �proc_priority_getr� r% �r� �valuer) r) r* �nice_get� � zProcess.nice_getc C s t �| j|�S r� )rx �proc_priority_setr� r� r) r) r* �nice_set� r& zProcess.nice_setc C r� r� )rx �proc_io_priority_getr� r+ rX r) r) r* � ionice_get� r� zProcess.ionice_getc C sL |rd}t |��|tjtjtjtjhvr |� d�}t|��t�| j |� d S )Nz&value argument not accepted on Windowsz is not a valid priority) � TypeErrorr+ r r r r r rx �proc_io_priority_setr� )r� �ioclassr� r r) r) r* � ionice_set� s � zProcess.ionice_setc C s� z t �| j�}W t|� S tyM } z6t|�s� td� | �� }|td |td |td |td |td |td f}W Y d }~t|� S d }~ww )Nz*attempting io_counters() fallback (slower)r6 r7 r9 r; r= r? ) rx �proc_io_countersr� r r r rO rW rm )r� r� r rl r) r) r* � io_counters s$ � ���zProcess.io_countersc C s t �| j�}|r tjS tjS r� )rx �proc_is_suspendedr� r �STATUS_STOPPED�STATUS_RUNNING)r� � suspendedr) r) r* r� s zProcess.statusc C s dd� }t �| j�}||�S )Nc s � fdd�t d�D �S )Nc s g | ] }d |> � @ r|�qS )r r) )r� �i�r� r) r* r� " s zBProcess.cpu_affinity_get.<locals>.from_bitmask.<locals>.<listcomp>�@ )rB r� r) r� r* � from_bitmask! s z.Process.cpu_affinity_get.<locals>.from_bitmask)rx �proc_cpu_affinity_getr� )r� r� �bitmaskr) r) r* �cpu_affinity_get s zProcess.cpu_affinity_getc C sv dd� }t ttt� ���}|D ] }||vr-t|t�s$d|�d�}t|��d|��}t|��q||�}t� | j |� d S )Nc S s4 | s d| ��}t |��d}| D ]}|d| O }q|S )Nzinvalid argument r r, )r )�lsr �out�br) r) r* � to_bitmask) s z,Process.cpu_affinity_set.<locals>.to_bitmaskz invalid CPU z; an integer is required) r� rB rz r� r� r� r� r rx �proc_cpu_affinity_setr� )r� r� r� �allcpus�cpur r� r) r) r* �cpu_affinity_set' s �zProcess.cpu_affinity_setc C sV zt �| j�W S ty* } zt|�r%td� | �� td W Y d }~S � d }~ww )Nz*attempting num_handles() fallback (slower)r. )rx �proc_num_handlesr� r r r rO rW rd r) r) r* r. A s ��zProcess.num_handlesc C s | � � td }t�|d�S )Nr/ r )rO rW r �pctxsw)r� r/ r) r) r* �num_ctx_switchesK s zProcess.num_ctx_switchesr� )F)r� ),r& r' r( r* � __slots__r� rR rU r rO r� r> rQ rJ re rg rj rm ro rr rx rz r� r� r r2 r4 r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r. r� r) r) r) r* rK � s� * rK r) )r$ )NN)}r* r+ �enumr; r{ r{ �sysrD � collectionsr r� r r r r r r r r r r r r r r r r r r r r rx � ImportErrorr �str�lowerr_ �getwindowsversionr � RuntimeError�__extra__all__r rC �builtin_module_namesr\ �IntEnumr# r �MIB_TCP_STATE_ESTAB�CONN_ESTABLISHED�MIB_TCP_STATE_SYN_SENT� CONN_SYN_SENT�MIB_TCP_STATE_SYN_RCVD� CONN_SYN_RECV�MIB_TCP_STATE_FIN_WAIT1�CONN_FIN_WAIT1�MIB_TCP_STATE_FIN_WAIT2�CONN_FIN_WAIT2�MIB_TCP_STATE_TIME_WAIT�CONN_TIME_WAIT�MIB_TCP_STATE_CLOSED� CONN_CLOSE�MIB_TCP_STATE_CLOSE_WAIT�CONN_CLOSE_WAIT�MIB_TCP_STATE_LAST_ACK� CONN_LAST_ACK�MIB_TCP_STATE_LISTEN� CONN_LISTEN�MIB_TCP_STATE_CLOSING� CONN_CLOSING�MIB_TCP_STATE_DELETE_TCB�PSUTIL_CONN_NONE� CONN_NONEr� r% �globalsr' � __members__r+ r rW rU r[ ra �_fieldsrg ri rv rk rm � lru_cacher� r� r� r� �disk_io_countersr� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r� �pidsr� rh r r3 r> rJ rK r) r) r) r* �<module> s@ � �� � �������� � � � � ����������!��� - __pycache__/_pslinux.cpython-310.pyc 0000644 00000146723 15030211765 0013333 0 ustar 00 o �h P � @ s" d Z ddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl Z ddl Z ddl Z ddl Z ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ dd lmZ dd lmZ dd lmZ dd lmZ dd lmZ ddlmZ ddlm Z ddlm Z ddlm Z ddlm Z ddlm Z ddlm!Z! ddlm"Z" ddlm#Z# ddlm$Z$ ddlm%Z% ddlm&Z& ddlm'Z' ddlm(Z( dd lm)Z) dd lm*Z* dd lm+Z+ g d �Z,e-e d �r�e,�.d!d"� e/e�D �� d#Z0ej1�2d$e�3� � d%��Z4ej1�2d$e�3� � d&��Z5e-ed'�Z6e-ed(�Z7e�8d)�Z9e�:� Z;da<e j=d*kZ>d+Z?e�@d,d-eAe jB�i�ZCeCjDZDG d.d/� d/ej@�ZEeF� �GeEjH� ejIejJejKejLejMejNejOejOejPejQejRejSd0� ZTejUejVejWejXejYejZej[ej\ej]ej^ej_d1� Z`ed2g d3��Zaed4g d5��Zbed6g d7��Zced8d9�Zded:edjed; �Zfed<g d=��Zged>d?d@�hegje� �ZiedAg dB��ZjedCg dD��ZkdEdF� ZldGdH� ZmdIdJ� Zne$dKdL� �ZozeodM� W n# ep�y� Zq ze dNeq��� edOdP�dQdQdQ�arW Y dZq[qndZq[qww dRdS� ZsdTdU� ZtdVdW� ZudXdY� ZvdZd[� Zwd\d]� Zxd^d_� Zyd`da� Zzdbdc� Z{ej1�2dd��s%ej1�2de��r*dfdg� Z|ndhdg� Z|ej}Z}G didj� djep�Z~G dkdl� dl�Ze� Z�d�dndo�Z�dpdq� Z�drds� Z�ej�Z�d�dudv�Z�G dwdx� dx�Z�d�dydz�Z�d{d|� Z�d}d~� Z�dd�� Z�d�d�� Z�d�d�� Z�d�d�� Z�d�d�� Z�d�d�� Z�d�d�� Z�G d�d�� d��Z�dS )�z Linux platform implementation.� N)� defaultdict)� namedtuple� )�_common)�_psposix)� _psutil_linux)� _psutil_posix)�ENCODING)�NIC_DUPLEX_FULL)�NIC_DUPLEX_HALF)�NIC_DUPLEX_UNKNOWN)� AccessDenied)� NoSuchProcess)� ZombieProcess)�bcat)�cat)�debug)�decode)�get_procfs_path)� isfile_strict)�memoize)�memoize_when_activated)� open_binary)� open_text)�parse_environ_block)�path_exists_strict)� supports_ipv6)� usage_percent)� PROCFS_PATH�IOPRIO_CLASS_NONE�IOPRIO_CLASS_RT�IOPRIO_CLASS_BE�IOPRIO_CLASS_IDLE�CONN_ESTABLISHED� CONN_SYN_SENT� CONN_SYN_RECV�CONN_FIN_WAIT1�CONN_FIN_WAIT2�CONN_TIME_WAIT� CONN_CLOSE�CONN_CLOSE_WAIT� CONN_LAST_ACK� CONN_LISTEN� CONN_CLOSING�prlimitc C s" g | ] }|� d �r|�� r|�qS )�RLIM)� startswith�isupper��.0�x� r5 �B/usr/local/CyberCP/lib/python3.10/site-packages/psutil/_pslinux.py� <listcomp>A s " r7 z/sys/class/power_supply�/proc/�/smaps� /smaps_rollup�proc_ioprio_get�proc_cpu_affinity_get� SC_CLK_TCK�littlei � AddressFamily�AF_LINKc @ s e Zd ZdZdZdZdZdS )� IOPriorityr r � � N)�__name__� __module__� __qualname__r r r! r" r5 r5 r5 r6 rA k s rA ) �R�S�D�T�t�Z�Xr4 �K�W�I�P) �01�02�03�04�05�06�07�08�09�0A�0B�svmem) �total� available�percent�used�free�active�inactive�buffers�cached�shared�slab�sdiskio) � read_count� write_count� read_bytes� write_bytes� read_time� write_time�read_merged_count�write_merged_count� busy_time� popenfile)�path�fd�position�mode�flags�pmemz"rss vms shared text lib data dirty�pfullmem)�uss�pss�swap� pmmap_grouped) rt �rss�sizer| � shared_clean� shared_dirty� private_clean� private_dirty� referenced� anonymousr} � pmmap_extz addr perms � �pio)rj rk rl rm � read_chars� write_chars� pcputimes)�user�system� children_user�children_system�iowaitc C sL t | t�s J | ��t�| �} | �d�d } | �d�r$t| �s$| dd� } | S )z Wrapper around os.readlink().� r � (deleted)N���)� isinstance�str�os�readlink�split�endswithr �rt r5 r5 r6 r� � s r� c C sX t jdt jdt jdi}|| t jt jB t jB @ }| t j@ r$|�ddd�}|�dd�}|S )zZConvert file's open() flags into a readable string. Used by Process.open_files(). �r�wzw+�ar zr+)r� �O_RDONLY�O_WRONLY�O_RDWR�O_APPEND�replace)rx � modes_maprw r5 r5 r6 �file_flags_to_mode� s r� c C s: | � dd�} d}|rd| � �}nd| � d�}t�|tj�S )z�Return True if the given name refers to a root device (e.g. "sda", "nvme0n1") as opposed to a logical partition (e.g. "sda1", "nvme0n1p1"). If name is a virtual device (e.g. "loop1", "ram") return True. �/�!Tz /sys/block/z/device)r� r� �access�F_OK)�name�including_virtualrt r5 r5 r6 �is_storage_device� s r� c C s� t | � d���}|�� �� dd� }W d � n1 s w Y g d�}t|�}|dkr2|�d� |dkr;|�d� |d krD|�d � td |�adS ) z�Set a namedtuple of variable fields depending on the CPU times available on this Linux kernel version which may be: (user, nice, system, idle, iowait, irq, softirq, [steal, [guest, [guest_nice]]]) Used by cpu_times() function. �/statr N)r� �nicer� �idler� �irq�softirq� �steal� �guest� � guest_nice� scputimes)r �readliner� �len�appendr r� )� procfs_path�f�values�fields�vlenr5 r5 r6 �set_scputimes_ntuple� s � r� �/procz ignoring exception on import: r� zuser system idle� c C sF | d }|| � dd� }z| d }| d }| d }W n ty8 } zt|jd � d�� |W Y d}~S d}~ww z tt� � d ��}W n tyN | Y S w d}|� |D ]} | �� } | �d �rk|t | � � d �7 }qVW d � n1 svw Y |t 9 }|| } || } | t | d |�8 } | | 7 } | |t |d |� 7 } t | �S )ay Fallback for kernels < 3.14 where /proc/meminfo does not provide "MemAvailable", see: https://blog.famzah.net/2014/09/24/. This code reimplements the algorithm outlined here: https://git.kernel.org/cgit/linux/kernel/git/torvalds/linux.git/ commit/?id=34e431b0ae398fc54ea69ff85ec700722c9da773 We use this function also when "MemAvailable" returns 0 (possibly a kernel bug, see: https://github.com/giampaolo/psutil/issues/1915). In that case this routine matches "free" CLI tool result ("available" column). XXX: on recent kernels this calculation may differ by ~1.5% compared to "MemAvailable:", as it's calculated slightly differently. It is still way more realistic than doing (free + cached) though. See: * https://gitlab.com/procps-ng/procps/issues/42 * https://github.com/famzah/linux-memavailable-procfs/issues/2 � MemFree:� Cached:r s Active(file):s Inactive(file):� SReclaimable:zW is missing from /proc/meminfo; using an approximation for calculating available memoryNz /zoneinfos lowr rB g @) �get�KeyErrorr �argsr r �OSError�stripr0 �intr� �PAGESIZE�min) �memsrb �fallback�lru_active_file�lru_inactive_file�slab_reclaimable�errr� � watermark_low�line�avail� pagecacher5 r5 r6 �calculate_avail_vmem! sD ��� � �� �r� c C s� g } i }t t� � d��� }|D ]}|�� }t|d �d ||d < qW d � n1 s,w Y |d }|d }z|d }W n tyO d}| �d � Y nw z|d }W n tyf d}| �d � Y n w ||�d d�7 }z|d } W n ty� z|d } W n ty� d} | �d� Y nw Y nw z|d } W n ty� d} | �d� Y nw z|d } W n( ty� z|d |d |d } W n ty� d} | �d� Y nw Y nw z|d } W n ty� d} Y nw || | | } | dk r�|| } z|d }W n t�y t|�}Y n w |dk�r t|�}|dk �r)d}| �d� n||k�r0|}t|| |dd�}| �rVd� d � | �t | �dk�rKd nd �}t j |td d � t|||| || | ||| | � S )!a7 Report virtual memory stats. This implementation mimics procps-ng-3.3.12, aka "free" CLI tool: https://gitlab.com/procps-ng/procps/blob/ 24fd2605c51fccc375ab0287cec33aa767f06718/proc/sysinfo.c#L778-791 The returned values are supposed to match both "free" and "vmstat -s" CLI tools. �/meminfor � r Ns MemTotal:r� s Buffers:re r� rf r� s Shmem:s MemShared:rg s Active:rc s Inactive:s Inact_dirty:s Inact_clean:s Inact_laundry:rd s Slab:s MemAvailable:r_ ��round_z6{} memory stats couldn't be determined and {} set to 0z, �was�wererB �� stacklevel)r r r� r� r� r� r� r� r �format�joinr� �warnings�warn�RuntimeWarningr] )�missing_fieldsr� r� r� r� r^ rb re rf rg rc rd rh ra r� r` �msgr5 r5 r6 �virtual_memory` s� � � � � ��� � ��� ��� � � ��r� c C s� i } t t� � d��� }|D ]}|�� }t|d �d | |d < q W d � n1 s*w Y z | d }| d }W n tyU t�� \}}}}}}}||9 }||9 }Y nw || }t||dd�} z t t� � d ��}W n$ ty� } zd | � d �} t j | t d d � d } } W Y d} ~ nad} ~ ww |�R d } } |D ]1}|� d�r�t|�d�d �d d } n|� d�r�t|�d�d �d d } | dur�| dur� nq�d} | d7 } t j | t d d � d } } W d � n1 s�w Y t �|||| | | �S )zReturn swap memory metrics.r� r r� r Ns SwapTotal:s SwapFree:r� z/vmstatzM'sin' and 'sout' swap memory stats couldn't be determined and were set to 0 (�)rB r� s pswpin� � s pswpoutz,'sin' and 'sout' swap memory stats couldn't z be determined and were set to 0)r r r� r� r� �cext� linux_sysinfor r� r� r� r� r0 r �sswap)r� r� r� r� r^ rb �_�unit_multiplierra r` r� r� �sin�soutr5 r5 r6 � swap_memory� sZ � � ����� �� �r� c C st t � } t| � t| � d���}|�� �� }W d � n1 s w Y |dttj�d � }dd� |D �}t|� S )z�Return a named tuple representing the following system-wide CPU times: (user, nice, system, idle, iowait, irq, softirq [steal, [guest, [guest_nice]]]) Last 3 fields may not be available on all Linux kernel versions. r� Nr c S � g | ]}t |�t �qS r5 ��float� CLOCK_TICKSr2 r5 r5 r6 r7 - � z cpu_times.<locals>.<listcomp>)r r� r r� r� r� r� �_fields)r� r� r� r� r5 r5 r6 � cpu_times! s �r� c C s� t � } t| � g }t| � d���7}|�� |D ]&}|�d�r=|�� }|dttj�d � }dd� |D �}t|� }|� |� q|W d � S 1 sJw Y dS )zfReturn a list of namedtuple representing the CPU times for every CPU available on the system. r� s cpur c S r� r5 r� r2 r5 r5 r6 r7 ? r� z!per_cpu_times.<locals>.<listcomp>N) r r� r r� r0 r� r� r� r� r� )r� �cpusr� r� r� r� �entryr5 r5 r6 � per_cpu_times1 s �$�r� c C s� zt �d�W S tyx d} tt� � d���}|D ] }|�� �d�r'| d7 } qW d � n1 s2w Y | dkrmt�d�}t t� � d��� }|D ]}|� d �d }|� |�r]| d7 } qKW d � n1 shw Y | dkrtY dS | Y S w ) z0Return the number of logical CPUs in the system.�SC_NPROCESSORS_ONLNr �/cpuinfos processorr Nzcpu\dr� r� ) r� �sysconf� ValueErrorr r �lowerr0 �re�compiler r� �match)�numr� r� �searchr5 r5 r6 �cpu_count_logicalE s2 �� � �� ��r c C s0 t � } d}d}t�|�pt�|�D ] }t|��}| �|�� �� � W d � n1 s+w Y qt| �}|dkr;|S i }i }tt� � d���?}|D ]4}|�� �� }|skz |d ||d < W n t yg Y nw i }qJ|� d�r~|� d d �\} } t | �|| < qJW d � n1 s�w Y t |�� �}|p�dS ) z-Return the number of CPU cores in the system.z9/sys/devices/system/cpu/cpu[0-9]*/topology/core_cpus_listz?/sys/devices/system/cpu/cpu[0-9]*/topology/thread_siblings_listNr r � cpu cores� physical id)r r s :r )�set�globr �add�readr� r� r r r� r0 r� r� �sumr� ) �ls�p1�p2rt r� �result�mapping� current_infor� �key�valuer5 r5 r6 �cpu_count_coresb sB �� � � �� � r c C s� t t� � d���J} d}d}d}| D ]9}|�d�r!t|�� d �}n|�d�r/t|�� d �}n |�d�r<t|�� d �}|durJ|durJ|durJ nqW d � n1 sUw Y d}t�||||�S )z*Return various CPU stats as a named tuple.r� Ns ctxtr s intrs softirqr )r r r0 r� r� r � scpustats)r� � ctx_switches� interrupts�soft_interruptsr� �syscallsr5 r5 r6 � cpu_stats� s* � ��r c C sB t t� � d���} dd� | D �W d � S 1 sw Y dS )z7Return current CPU frequency from cpuinfo if available.r c S s. g | ]}|� � �d �rt|�dd�d ��qS )s cpu mhz� :r )r r0 r� r� )r3 r� r5 r5 r6 r7 � s ��z)_cpu_get_cpuinfo_freq.<locals>.<listcomp>N)r r )r� r5 r5 r6 �_cpu_get_cpuinfo_freq� s �$�r! z'/sys/devices/system/cpu/cpufreq/policy0z$/sys/devices/system/cpu/cpu0/cpufreqc C s( t � } t�d�p t�d�}|jdd� d� g }tjj}t|�D ]r\}}t|�t| �kr2| | d }n t||d�dd �}|du rkt||d �dd �}|du rkd |� d �}t |dd �d kre|� t � ddd�� q d}t |��t|�d }tt||d���d } tt||d���d } |� t � || | �� q |S )z�Return frequency metrics for all CPUs. Contrarily to other OSes, Linux updates these values in real-time. z,/sys/devices/system/cpu/cpufreq/policy[0-9]*z)/sys/devices/system/cpu/cpu[0-9]*/cpufreqc S s t t�d| ��� �S )Nz[0-9]+)r� r r �group)r4 r5 r5 r6 �<lambda>� s zcpu_freq.<locals>.<lambda>)r i� �scaling_cur_freqN�r� �cpuinfo_cur_freqz/sys/devices/system/cpu/cpuz/onlinez0 r� z!can't find current frequency file�scaling_max_freq�scaling_min_freq)r! r �sortr� rt r� � enumerater� r r r� r �scpufreq�NotImplementedErrorr� ) � cpuinfo_freqs�paths�ret�pjoin�irt �curr� online_pathr� �max_�min_r5 r5 r6 �cpu_freq� s6 �� r6 c C s dd� t � D �S )z}Alternate implementation using /proc/cpuinfo. min and max frequencies are not available and are set to None. c S s g | ] }t �|d d ��qS )r� )r r+ r2 r5 r5 r6 r7 � � z cpu_freq.<locals>.<listcomp>)r! r5 r5 r5 r6 r6 � s c @ s e Zd ZdS )�_Ipv6UnsupportedErrorN)rD rE rF r5 r5 r5 r6 r8 � s r8 c @ sZ e Zd ZdZdd� Zdd� Zdd� Zedd � �Zedd d ��Z edd d��Z ddd�Z d S )�NetConnectionsaw A wrapper on top of /proc/net/* files, retrieving per-process and system-wide open connections (TCP, UDP, UNIX) similarly to "netstat -an". Note: in case of UNIX sockets we're only able to determine the local endpoint/path, not the one it's connected to. According to [1] it would be possible but not easily. [1] http://serverfault.com/a/417946 c C s� dt jt jf}dt jt jf}dt jt jf}dt jt jf}dt jd f}|||||f||f|f|f||f|f|f|f||||f||f||fd� | _d | _d S )N�tcp�tcp6�udp�udp6�unix) �allr: �tcp4r; r< �udp4r= r> �inet�inet4�inet6)�socket�AF_INET� SOCK_STREAM�AF_INET6� SOCK_DGRAM�AF_UNIX�tmap� _procfs_path)�selfr@ r; rA r= r> r5 r5 r6 �__init__� s$ � zNetConnections.__init__c C s� t t�}t�| j� d|� d��D ]_}zt| j� d|� d|� ��}W n3 ttfy, Y q tyT } z |j t j kr?W Y d }~q|j t j krOt |� W Y d }~q� d }~ww |� d�ro|dd � d d� }|| �|t|�f� q|S )Nr� �/fd�/fd/zsocket:[r� ���)r �listr� �listdirrL r� �FileNotFoundError�ProcessLookupErrorr� �errno�EINVAL� ENAMETOOLONGr r0 r� r� )rM �pid�inodesru �inoder� r5 r5 r6 �get_proc_inodes s( �� �z NetConnections.get_proc_inodesc C s@ i }t � D ]}z |�| �|�� W q tttfy Y qw |S �N)�pids�updater\ rT rU �PermissionError)rM rZ rY r5 r5 r6 �get_all_inodes* s � z NetConnections.get_all_inodesc C s� | � d�\}}t|d�}|sdS |�d�}|tjkr5tr+t�|t�|�ddd� �}nGt�|t�|��}n=t�|�}z)trPt�tj t j dgt � d|��R � �}nt�tj t j dgt � d|��R � �}W n t yq t� sptd�� w t�||�S ) a� Accept an "ip:port" address as displayed in /proc/net/* and convert it into a human readable form, like: "0500000A:0016" -> ("10.0.0.5", 22) "0000000000000000FFFF00000100007F:9E49" -> ("::ffff:127.0.0.1", 40521) The IP address portion is a little or big endian four-byte hexadecimal number; that is, the least significant byte is listed first, so we need to reverse the order of the bytes to convert it to an IP address. The port is represented as a two-byte hexadecimal number. Reference: http://linuxdevcenter.com/pub/a/linux/2000/11/16/LinuxAdmin.html �:� r5 �asciiNrQ z>4Iz<4I)r� r� �encoderE rF � LITTLE_ENDIAN� inet_ntop�base64� b16decoderH �struct�pack�unpackr r r8 r �addr)rm �family�ip�portr5 r5 r6 �decode_address: s6 ��� � z NetConnections.decode_addressNc c sP � | � d�rtj�| �sdS t| ���}|�� t|d�D ]y\}}z|�� dd� \ }} } } }}}}}} W n tyK d| � d|� d|��} t | �d�w | |v rY|| d \}}nd \}}|durf||krfq |t j krpt | } nt j} zt�| |�} t�| |�} W n ty� Y q w |||| | | |fV q W d � dS 1 s�w Y dS ) z.Parse /proc/net/tcp* and /proc/net/udp* files.�6Nr r� �error while parsing �; malformed line r� r �NrQ )r� r� rt �existsr r� r* r� r � RuntimeErrorrE rG � TCP_STATUSESr � CONN_NONEr9 rq r8 )�filern �type_rZ � filter_pidr� �linenor� r� �laddr�raddr�statusr[ r� rY ru r5 r5 r6 � process_inetm sF � � ��� � ��"�zNetConnections.process_inetc c s � t | ��}}|�� |D ]m}|�� }z|dd� \}}}}}}} W n ty: d|vr.Y q d| � d|��} t| ��w | |v rD|| } ndg} | D ]/\} } |durV|| krVqIt|�dkr`|d nd }t�t|��}d }tj }| |||||| fV qIq W d � dS 1 s�w Y dS ) zParse /proc/net/unix files.r � r� rs rt ru Nr� rQ � ) r r� r� r rw r� r �socktype_to_enumr� ry )rz rn rZ r| r� r� �tokensr� r{ r[ r� �pairsrY ru rt r r� r5 r5 r6 � process_unix� s8 � �� ��"�zNetConnections.process_unixc C s� t � | _|d ur| �|�}|sg S n| �� }t� }| j| D ]R\}}}| j� d|� �}|tjtjhv r>| j |||||d�} n | j ||||d�} | D ]&\} }}} } } }|r_t � | ||| | | �}n t � | ||| | | |�}|�|� qIq t|�S )Nz/net/)r| )r rL r\ ra r rK rE rF rH r� r� r �pconn�sconnr rR )rM �kindrY rZ r/ � proto_namern r{ rt r ru r~ r r� � bound_pid�connr5 r5 r6 �retrieve� s4 � � �� � zNetConnections.retriever] ) rD rE rF �__doc__rN r\ ra � staticmethodrq r� r� r� r5 r5 r5 r6 r9 � s 2 * #r9 rB c C s t �| �S )z$Return system-wide open connections.)�_net_connectionsr� )r� r5 r5 r6 �net_connections� s r� c C s� t t� � d��� } | �� }W d � n1 sw Y i }|dd� D ]F}|�d�}|dks5J t|���|d|� �� }||d d� �� �� }tt|�\}}} } } } } }}}}}}}}}||||| || |f||< q$|S )zsReturn network I/O statistics for every network interface installed on the system as a dict of raw tuples. z/net/devNrB rb r r ) r r � readlines�rfind�reprr� r� �mapr� )r� �lines�retdictr� �colonr� r� � bytes_recv� packets_recv�errin�dropin�_fifoin�_framein� _compressedin� _multicastin� bytes_sent� packets_sent�errout�dropout�_fifoout�_collisionsout� _carrierout�_compressedoutr5 r5 r6 �net_io_counters� sJ � � � r� c C s� t jtt jtt jti} t� �� }i }|D ]I}zt � |�}t � |�}t � |�\}}W n t yF } z|jtjkr8� t|� W Y d}~qd}~ww d�|�} d|v } t�| | | ||| �||< q|S )z)Get NIC stats (isup, duplex, speed, mtu).N�,�running)r� � DUPLEX_FULLr � DUPLEX_HALFr �DUPLEX_UNKNOWNr r� �keys� cext_posix� net_if_mtu� net_if_flags�net_if_duplex_speedr� rV �ENODEVr r� r � snicstats) � duplex_map�namesr/ r� �mturx �duplex�speedr� � output_flags�isupr5 r5 r6 � net_if_stats s. � �� �r� Fc C s� dd� }dd� }t j�t� � d��r|� }nt j�d�r |� }n t� � d�}t|��i }|D ]*}|\ }}} } } } } }}}| sCt|�sCq.| t9 } | t9 } || | | | | |||f ||< q.|S )zcReturn disk I/O statistics for every disk installed on the system as a dict of raw tuples. c s sJ � t t� � d��� } | �� }W d � n1 sw Y |D ]�}|�� }t|�}|dkrK|d }t|d �}tt|dd� �\ }}} } } } } }}}nJ|dksS|dkrl|d }tt|dd� �\ }}}} } } } } }}}n)|dkr�|d }tt|dd � �\}}} } d } } } } }n d |��}t|��||| || | | || |f V q d S ) N� /diskstats� rC rB r� � � r� r z not sure how to interpret line )r r r� r� r� r� r� r )r� r� r� r� �flenr� �reads� reads_merged�rbytes�rtime�writes� writes_merged�wbytes�wtimer� rr r� r5 r5 r6 � read_procfs; s8 � � � � ��z%disk_io_counters.<locals>.read_procfsc s s� � t �d�D ]^} t �t j�d| ��D ]Q\}}}d|vr qtt j�|d���}|�� �� �� }W d � n1 s8w Y t j� |�}t t |d d� �\ }}} } } } } }}}||| | | | ||| |f V qqd S )N� /sys/block�statr� ) r� rS �walkrt r� r r r� r� �basenamer� r� )�block�rootr� �filesr� r� r� r� r� r� r� r� r� r� r� rr r5 r5 r6 � read_sysfsi s" � � � ���z$disk_io_counters.<locals>.read_sysfsr� r� z6/diskstats nor /sys/block are available on this system)r� rt rv r r, r� �DISK_SECTOR_SIZE)�perdiskr� r� �genr� r� r� r� r� r� r� r� r� r� r� r� rr r5 r5 r6 �disk_io_counters6 s, . �� �r� c @ s@ e Zd ZdZddgZdd� Zdd� Zdd � Zd d � Zd d � Z dS )�RootFsDeviceFinderaF disk_partitions() may return partitions with device == "/dev/root" or "rootfs". This container class uses different strategies to try to obtain the real device path. Resources: https://bootlin.com/blog/find-root-device/ https://www.systutorials.com/how-to-find-the-disk-where-root-is-on-in-bash-on-linux/. �major�minorc C s( t �d�j}t �|�| _t �|�| _d S )Nr� )r� r� �st_devr� r� )rM �devr5 r5 r6 rN � � zRootFsDeviceFinder.__init__c C s� t t� � d���Y}|�� dd � D ]G}|�� }t|�dk r q|d �� r*t|d �nd }|d �� r8t|d �nd }|d }|| jkrX|| jkrX|rXd|� � W d � S qW d � d S 1 sdw Y d S )Nz /partitionsrB r� r r rC �/dev/) r r r� r� r� �isdigitr� r� r� )rM r� r� r� r� r� r� r5 r5 r6 �ask_proc_partitions� s ��"�z&RootFsDeviceFinder.ask_proc_partitionsc C s� d| j � d| j� d�}t|��,}|D ] }|�d�r2|�� �d�d }|r2d|� � W d � S qW d � d S 1 s>w Y d S )Nz/sys/dev/block/rb z/ueventzDEVNAME=rB r� )r� r� r r0 r� � rpartition)rM rt r� r� r� r5 r5 r6 �ask_sys_dev_block� s ��"�z$RootFsDeviceFinder.ask_sys_dev_blockc C s� | j � d| j� �}t�d�}|D ]F}zt|�}W n ty! Y qw |�* |�� �� }||krGtj � tj � |��}d|� �W d � S W d � n1 sQw Y qd S )Nrb z/sys/class/block/*/devr� ) r� r� r �iglobr rT r r� r� rt r� �dirname)rM �needler� rz r� �datar� r5 r5 r6 �ask_sys_class_block� s$ � � ���z&RootFsDeviceFinder.ask_sys_class_blockc C s� d }|d u r#z| � � }W n ty" } z t|� W Y d }~nd }~ww |d u rDz| �� }W n tyC } z t|� W Y d }~nd }~ww |d u rez| �� }W n tyd } z t|� W Y d }~nd }~ww |d urqtj�|�rs|S d S d S r] )r� r� r r� r� r� rt rv )rM rt r� r5 r5 r6 �find� s2 �� �� ���zRootFsDeviceFinder.findN) rD rE rF r� � __slots__rN r� r� r� r� r5 r5 r5 r6 r� � s r� c C s4 t � }t� }| sEt|� d���.}|D ]#}|�� }|�d�s%|�|�� � q|�d�d }|dkr5|�d� qW d � n1 s@w Y |dkrVtj� d�rVtj� d�}n tj� |� d ��}g }t � |�}|D ]/} | \} } }} | d krvd } | d v r�t � �� p�| } | s�| r�||vr�qht�| | || �} |�| � qh|S ) z8Return mounted disk partitions as a list of namedtuples.z /filesystems�nodev� r �zfsNr� z /etc/mtabz /self/mounts�noner� > � /dev/root�rootfs)r r r r� r0 r r� r� rt �isfile�realpathr� �disk_partitionsr� r� r � sdiskpartr� )r? �fstypesr� r� r� �fstype� mounts_path�retlist� partitions� partition�device� mountpoint�opts�ntupler5 r5 r6 r� � s> �� � r� c C s8 t �t�} t�d�}|�t�d�� tdd� |D ��}t�d�}t�d�}|D ]}|�d|�}||vr8|� |� q'|D ]�}z |d }t t |��d }t j �t j �|�d �}t|��� } W n ttfyg Y q;w t |d d d �} t |d d d �} t|d dd ��� } | d ur�zt | �d } W n ty� d } Y nw | d ur�zt | �d } W n ty� d } Y nw | | � | || | f� q;|�s�t�d�}tt|��}|D ]�}z t j �|d�}t t |��d }t j �|d�}t|��� } W n ttf�y } z t| � W Y d } ~ q�d } ~ ww t�|d �}dd� |D �}d } d } |D ]o}t j �||d �}t|dd ��� }|dk�rBt t j �||d �d d �} n|dk�rTt t j �||d �d d �} | d u�rnzt | �d } W n t�ym d } Y nw | d u�r�z t | �d } W �q t�y� d } Y �q w �q | | � d|| | f� q�t| �S )a� Return hardware (CPU and others) temperatures as a dict including hardware name, label, current, max and critical temperatures. Implementation notes: - /sys/class/hwmon looks like the most recent interface to retrieve this info, and this implementation relies on it only (old distros will probably use something else) - lm-sensors on Ubuntu 16.04 relies on /sys/class/hwmon - /sys/class/thermal/thermal_zone* is another one but it's more difficult to parse z /sys/class/hwmon/hwmon*/temp*_*z&/sys/class/hwmon/hwmon*/device/temp*_*c S � h | ] }|� d �d �qS �r� r �r� r2 r5 r5 r6 � <setcomp>) r7 z'sensors_temperatures.<locals>.<setcomp>z5/sys/devices/platform/coretemp.*/hwmon/hwmon*/temp*_*z'/sys/devices/platform/coretemp.*/hwmon/z/sys/class/hwmon/�_inputg @�@r� �_maxNr% �_crit�_labelr� z /sys/class/thermal/thermal_zone*�temp�typez /trip_point*c S s, h | ]}d � tj�|��d �dd� ��qS )r� r rC )r� r� rt r� r� )r3 �pr5 r5 r6 r j s ���_type�critical�_temp�high)� collectionsr rR r �extend�sortedr r �subr� r� r r� rt r� r� r r� r� r r r �dict)r/ � basenames� basenames2�replr� �altname�basert �current� unit_namer r �labelr� � trip_paths� trip_points� trip_point� trip_typer5 r5 r6 �sensors_temperatures s� � � � � � ��� � � � ��r c C s� t �t�} t�d�}|st�d�}tdd� |D ��}|D ]G}z tt|d ��}W n ty> } z t|� W Y d}~q d}~ww t t j � t j � |�d���� }t |d d d ��� }| | �t�||�� q t| �S ) a� Return hardware fans info (for CPU and other peripherals) as a dict including hardware label and current speed. Implementation notes: - /sys/class/hwmon looks like the most recent interface to retrieve this info, and this implementation relies on it only (old distros will probably use something else) - lm-sensors on Ubuntu 16.04 relies on /sys/class/hwmon z /sys/class/hwmon/hwmon*/fan*_*z%/sys/class/hwmon/hwmon*/device/fan*_*c S r� r r r2 r5 r5 r6 r � r7 z sensors_fans.<locals>.<setcomp>r Nr� r r� r% )r r rR r r r� r r� r r r� rt r� r� r� r� r �sfanr )r/ r r r r� r r r5 r5 r6 � sensors_fans� s" �� r! c s� t � � � fdd�} dd� t�t�D �}|sdS tj�tt|��}| |d |d �}| |d |d �}| |d |d �}| |d �}|dur]|dur]zd | | }W n ty\ d}Y nw tt |d dd��}|dkrmdS d}| tj�td�tj�td��} | dur�| dk}nt |d dd�� � � � } | dkr�d}n| dv r�d}|r�t j } n4|dur�|dur�z t|| d �} W n! ty� t j} Y nw |dur�t|d �} | d k r�t j} nt j} t �|| |�S ) a Return battery information. Implementation note: it appears /sys/class/power_supply/BAT0/ directory structure may vary and provide files with the same meaning but under different names, see: https://github.com/giampaolo/psutil/issues/966. c sP | D ]#}t |� d�}|� kr%zt|�W S ty$ |�� Y S w qdS )zvAttempt to read the content of multiple files which may not exist. If none of them exist return None. r% N)r r� r r� )r. rt r/ ��nullr5 r6 � multi_bcat� s ��z#sensors_battery.<locals>.multi_bcatc S s& g | ]}|� d �sd|�� v r|�qS )�BAT�battery)r0 r r2 r5 r5 r6 r7 � s � �z#sensors_battery.<locals>.<listcomp>Nz /energy_nowz /charge_nowz /power_nowz /current_nowz /energy_fullz /charge_fullz/time_to_empty_nowg Y@r� z /capacityrQ r% z AC0/onlinez AC/onliner �/statusr� � dischargingF> �full�chargingTi �< r )�objectr� rS �POWER_SUPPLY_PATHrt r� r� �ZeroDivisionErrorr� r r� r r �POWER_TIME_UNLIMITED�POWER_TIME_UNKNOWN�sbattery) r$ �batsr� � energy_now� power_now� energy_full� time_to_emptyr` � power_plugged�onliner� �secsleftr5 r" r6 �sensors_battery� s` � � � � �r: c C sH g } t �� }|D ]}|\}}}}}t�||pd|||�}| �|� q| S )z:Return currently connected users as a list of namedtuples.N)r� �usersr �suserr� ) r� �rawlist�itemr� �tty�hostname�tstamprY �ntr5 r5 r6 r; s r; c C s~ t � � d�} t| ��+}|D ] }|�d�r+t|�� �� d �}|a| W d � S q d| � �}t|��1 s8w Y dS )zAReturn the system boot time expressed in seconds since the epoch.r� s btimer Nzline 'btime' not found in )r r r0 r� r� r� � BOOT_TIMErw )rt r� r� r/ r� r5 r5 r6 � boot_time s � �rD c C s t � �t�} dd� t�| �D �S )z7Returns a list of PIDs currently running on the system.c S s g | ] }|� � rt|��qS r5 )r� r� r2 r5 r5 r6 r7 + s zpids.<locals>.<listcomp>)r re r r� rS r� r5 r5 r6 r^ ( s r^ c C s� t �| �sdS zBt� � d| � d�}t|��*}|D ] }|�d�r5t|�� d �}|| k W d � W S qd|� �}t|��1 sBw Y W dS ttfyY | t � v Y S w )zcCheck for the existence of a unix PID. Linux TIDs are not supported (always return False). Fr� r' s Tgid:r Nz'Tgid' line not found in ) r � pid_existsr r r0 r� r� r r� r^ )rY rt r� r� �tgidr� r5 r5 r6 rE . s" � � �rE c C s� i } t � }t� D ]G}z t|� d|� d��� }|�� }W d � n1 s$w Y W n ttfy5 Y qw |�d�}||d d� �� }t|d �}|| |< q| S )zsObtain a {pid: ppid, ...} dict for all running processes in one shot. Used to speed up Process.children(). r� r� N� )rB r ) r r^ r r rT rU r� r� r� )r/ r� rY r� r� �rpar�dset�ppidr5 r5 r6 �ppid_mapO s ��� rK c s t �� �� fdd��}|S )zlDecorator which translates bare OSError and OSError exceptions into NoSuchProcess and AccessDenied. c s� | j | j}}z � | g|�R i |��W S ty% } zt||�|�d }~w ty: } z | �� t||�|�d }~w ty] } z| �� tj � | j � d|� d��sXt||�|�� d }~ww )Nr� r� ) rY �_namer` r rU �_raise_if_zombier rT r� rt rv rL )rM r� �kwargsrY r� r� ��funr5 r6 �wrapperj s" � � ��z wrap_exceptions.<locals>.wrapper)� functools�wraps)rP rQ r5 rO r6 �wrap_exceptionse s rT c @ s� e Zd ZdZg d�Zdd� Zdd� Zdd� Zd d � Ze e d d � ��Z e e d d� ��Z e e dd� ��Z dd� Zdd� Ze dd� �Ze dd� �Ze dd� �Ze dd � �Ze d d � �Zej�d e�� � d ��rje d!d"� �Ze d#d$� �Ze d%d&� �Ze dfd(d)��Z e d*d+� �Z e d,d-� �Z e s�e r�d.d/� Z!e e"�#d0�e"�#d1�e"�#d2�fd3d4��Z$e d5d6� �Z%ne Z%e r�e d7d8� �Z&e d9d:� �Z'e e"�#d;�fd<d=��Z(e e"�#d>�fd?d@��Z)e dAdB� �Z*e dCdD� �Z+e dEdF� �Z,e-r�e dGdH� �Z.e"�#dI�fdJdK�Z/e dLdM� �Z0e1�r e dNdO� �Z2e dPdQ� �Z3e4e5dR��re dfdSdT��Z6e dUdV� �Z7e dWdX� �Z8e dgdZd[��Z9e d\d]� �Z:e d^d_� �Z;e e"�#d`�fdadb��Z<e e"�#dc�fddde��Z=d'S )h�Processz Linux process implementation.)�_cacherL �_ppidrL rY c C s || _ d | _d | _t� | _d S r] )rY rL rW r rL )rM rY r5 r5 r6 rN � s zProcess.__init__c C sX zt | j� d| j� d��}W n ty Y dS w |�d�}||d |d � }|dkS )Nr� r� FrG rB rC � Z)r rL rY r� r� )rM r� rH r� r5 r5 r6 � _is_zombie� s � zProcess._is_zombiec C s | � � r t| j| j| j��d S r] )rY r rY rL rW �rM r5 r5 r6 rM � s �zProcess._raise_if_zombiec C s t �| j� d| j� �� dS )z+Raise NSP if the process disappeared on us.r� N)r� r� rL rY rZ r5 r5 r6 �_raise_if_not_alive� s zProcess._raise_if_not_alivec C s� t | j� d| j� d��}|�d�}||�d�d |� }||d d� �� }i }||d< |d |d < |d |d < |d |d < |d |d< |d |d< |d |d< |d |d< |d |d< |d |d< z |d |d< W |S ty} td � d |d< Y |S w ) aY Parse /proc/{pid}/stat file and return a dict with various process info. Using "man proc" as a reference: where "man proc" refers to position N always subtract 3 (e.g ppid position 4 in 'man proc' == position 1 in here). The return value is cached in case oneshot() ctx manager is in use. r� r� rG � (r rB Nr� r r� rJ r� �ttynr� �utime� �stime� �children_utimer� �children_stime� � create_time�$ �cpu_num�' � blkio_ticksz&can't get blkio_ticks, set iowait to 0)r rL rY r� r� r� � IndexErrorr )rM r� rH r� r� r/ r5 r5 r6 �_parse_stat_file� s. � �zProcess._parse_stat_filec C sD t | j� d| j� d��� }|�� W d � S 1 sw Y dS )z�Read /proc/{pid}/stat file and return its content. The return value is cached in case oneshot() ctx manager is in use. r� r' N)r rL rY r �rM r� r5 r5 r6 �_read_status_file� s $�zProcess._read_status_filec C sH t | j� d| j� d���}|�� �� W d � S 1 s w Y d S )Nr� r9 )r rL rY r r� rm r5 r5 r6 �_read_smaps_file� s $�zProcess._read_smaps_filec C �( | j �| � | j�| � | j�| � d S r] )rl �cache_activatern ro rZ r5 r5 r6 � oneshot_enter� r� zProcess.oneshot_enterc C rp r] )rl �cache_deactivatern ro rZ r5 r5 r6 � oneshot_exit� r� zProcess.oneshot_exitc C � t | �� d �S )Nr� )r rl rZ r5 r5 r6 r� � � z Process.namec C sX z t | j� d| j� d��W S ttfy+ | �� tj�| j� d| j� ��r*Y dS � w )Nr� z/exer� ) r� rL rY rT rU rM r� rt �lexistsrZ r5 r5 r6 �exe� s �z Process.exec C s� t | j� d| j� d��� }|�� }W d � n1 sw Y |s(| �� g S |�d�r/dnd}|�|�r<|d d� }|�|�}|dkrTt|�dkrTd|v rT|�d�}|S )Nr� z/cmdliner� r� rQ r )r rL rY r rM r� r� r� )rM r� r� �sep�cmdliner5 r5 r6 rz � s � zProcess.cmdlinec C sN t | j� d| j� d���}|�� }W d � t|�S 1 s w Y t|�S )Nr� z/environ)r rL rY r r )rM r� r� r5 r5 r6 �environ s ��zProcess.environc C s8 t | �� d �}t�� }z|| W S ty Y d S w )Nr] )r� rl r �get_terminal_mapr� )rM �tty_nrrK r5 r5 r6 �terminal s �zProcess.terminalr8 �/ioc C s | j � d| j� d�}i }t|��,}|D ]!}|�� }|r4z |�d�\}}W n ty- Y qw t|�||< qW d � n1 s?w Y |sO|� d�}t|��zt|d |d |d |d |d |d �W S t y� } z|j d �d |� d |��}t|�d �d }~ww )Nr� r s : z file was emptys syscrs syscws read_bytess write_bytess rchars wcharr z field was not found in z; found fields are ) rL rY r r� r� r r� rw r� r� r� ) rM �fnamer� r� r� r� r r� r� r5 r5 r6 � io_counters$ sF � �� � ��� ��zProcess.io_countersc C sh | � � }t|d �t }t|d �t }t|d �t }t|d �t }t|d �t }t|||||�S )Nr_ ra rc rd rj )rl r� r� r� )rM r� r_ ra rc rd r� r5 r5 r6 r� G s zProcess.cpu_timesc C ru )zWhat CPU the process is on.rh �r� rl rZ r5 r5 r6 rh Q rv zProcess.cpu_numNc C s t �| j|| j�S r] )r �wait_pidrY rL )rM �timeoutr5 r5 r6 �waitV s z Process.waitc C s& t | �� d �}tp t� }|t | S )Nrf )r� rl rC rD r� )rM �ctime�btr5 r5 r6 rf Z s zProcess.create_timec C sx t | j� d| j� d��� }dd� |�� �� d d� D �\}}}}}}}W d � n1 s-w Y t|||||||�S )Nr� z/statmc s s � | ] }t |�t V qd S r] )r� r� r2 r5 r5 r6 � <genexpr>s s � �z&Process.memory_info.<locals>.<genexpr>r� )r rL rY r� r� ry ) rM r� �vmsr rg �text�libr� �dirtyr5 r5 r6 � memory_infoe s � �zProcess.memory_infoc C s� d } }}t | j� d| j� d���>}|D ]3}|�d�r)|t|�� d �d 7 }q|�d�r9t|�� d �d }q|�d�rHt|�� d �d }qW d � n1 sSw Y |||fS ) Nr r� r: s Private_r r� � Pss:� Swap:)r rL rY r0 r� r� )rM r{ r| r} r� r� r5 r5 r6 �_parse_smaps_rollupz s � �� � zProcess._parse_smaps_rollups \nPrivate.*:\s+(\d+)s \nPss\:\s+(\d+)s \nSwap\:\s+(\d+)c C sZ | � � }ttt|�|���d }ttt|�|���d }ttt|�|���d }|||fS )Nr� )ro r r� r� �findall)rM � _private_re�_pss_re�_swap_re� smaps_datar{ r| r} r5 r5 r6 � _parse_smaps� s zProcess._parse_smapsc C sd t r z | �� \}}}W n ttfy | �� \}}}Y nw | �� \}}}| �� }t||||f � S r] )�HAS_PROC_SMAPS_ROLLUPr� rU rT r� r� rz )rM r{ r| r} � basic_memr5 r5 r6 �memory_full_info� s �zProcess.memory_full_infoc C sX dd� }| � � }|s| �� g S |�d�}g }|�d�}|g}|||�D ]�\}}|�dd�}z |\} } } } } }W n tyL |dg \} } } } } }Y nw |sRd}nt|�}|�� }|�d �rit|�si|dd � }t| �t| �||� d d�|� d d�|� d d�|� dd�|� dd�|� dd�|� dd�|� dd�|� dd�|� dd�f }|� |� q$|S )a� Return process's mapped memory regions as a list of named tuples. Fields are explained in 'man proc'; here is an updated (Apr 2012) version: https://git.kernel.org/pub/scm/linux/kernel/git/torvalds/linux.git/tree/Documentation/filesystems/proc.txt?id=b76437579d1344b612cf1851ae610c636cec7db0. /proc/{PID}/smaps does not exist on kernels < 2.6.14 or if CONFIG_MMU kernel configuration option is not enabled. c s s� � i }| D ]D}|� d d�}|d �d�s!|�� |fV |�|� qzt|d �d ||d < W q tyI |d �d�r?Y qd|��}t|�d �w |�� |fV d S )N� r r r r� s VmFlags:z!don't know how to interpret line )r� r� �popr� r� r r0 )r� � current_blockr� r� r� r� r5 r5 r6 � get_blocks� s � �z'Process.memory_maps.<locals>.get_blocks� r Nr� r� z[anon]r� r� s Rss:s Size:r� s Shared_Clean:s Shared_Dirty:s Private_Clean:s Private_Dirty:s Referenced:s Anonymous:r� ) ro rM r� r� r r r� r� r r� r� )rM r� r� r� r � first_liner� �header�hfieldsrm �perms�_offset�_dev�_inodert r> r5 r5 r6 � memory_maps� sP � � � zProcess.memory_mapsc C s t | j� d| j� d��S )Nr� z/cwd)r� rL rY rZ r5 r5 r6 �cwd s z Process.cwds ctxt_switches:\t(\d+)c C sP | � � }|�|�}|sd| j� d| j� d�}t|��t�t|d �t|d ��S )NzR'voluntary_ctxt_switches' and 'nonvoluntary_ctxt_switches'lines were not found in r� z1/status; the kernel is probably older than 2.6.23r r )rn r� rL rY r, r �pctxswr� )rM � _ctxsw_rer� �ctxswr� r5 r5 r6 �num_ctx_switches s ��� zProcess.num_ctx_switchess Threads:\t(\d+)c C s | � � }t|�|�d �S �Nr )rn r� r� )rM �_num_threads_rer� r5 r5 r6 � num_threads s zProcess.num_threadsc C s t �| j� d| j� d��}|�� g }d}|D ]g}| j� d| j� d|� d�}z t|��}|�� �� }W d � n1 s<w Y W n tt fyO d}Y qw ||� d�d d � }|� d �}t |d �t } t |d �t } t�t|�| | �} |�| � q|r�| �� |S ) Nr� z/taskFz/task/r� TrG rB r� r^ r` )r� rS rL rY r) r r r� rT rU r� r� r� r� r �pthreadr� r� r[ ) rM � thread_idsr� � hit_enoent� thread_idr� r� �str� r_ ra r� r5 r5 r6 �threads! s0 ��� zProcess.threadsc C � t �| j�S r] )r� � getpriorityrY rZ r5 r5 r6 �nice_get< s zProcess.nice_getc C s t �| j|�S r] )r� � setpriorityrY )rM r r5 r5 r6 �nice_setE s zProcess.nice_setc C r� r] )r� r< rY rZ r5 r5 r6 �cpu_affinity_getL s zProcess.cpu_affinity_gets Cpus_allowed_list:\t(\d+)-(\d+)c C sR | � � }|�|�}|r ttt|d d �t|d d �d ��S tttt� ���S )Nr r )rn r� rR �ranger� r� r� )rM �_rer� r r5 r5 r6 �_get_eligible_cpusP s *zProcess._get_eligible_cpusc C s� z t �| j|� W d S ttfyX } z@t|t�s |jtjkrS| �� }t t t t � ���}|D ]$}||vrAd|�d|��}t|�d �||vrRd|� d|� �}t|�|�q.� d }~ww )Nz invalid CPU z; choose between z CPU number z! is not eligible; choose between ) r� �proc_cpu_affinity_setrY r� r r� rV rW r� �tupler� r� r� )rM r� r� � eligible_cpus�all_cpus�cpur� r5 r5 r6 �cpu_affinity_set[ s. �� �� ���zProcess.cpu_affinity_setc C s$ t �| j�\}}t|�}t�||�S r] )r� r; rY rA r �pionice)rM �ioclassr r5 r5 r6 � ionice_getu s zProcess.ionice_getc C s^ |d u rd}|r|t jt jhv r|�d�}t|��|dk s!|dkr'd}t|��t�| j||�S )Nr z ioclass accepts no valuer� zvalue not in 0-7 range)rA r" r r r� �proc_ioprio_setrY )rM r� r r� r5 r5 r6 � ionice_set{ s � zProcess.ionice_setr. c C s� | j dkr d}t|��z&|d u rt�| j |�W S t|�dkr'd|��}t|��t�| j ||� W d S tyH } z |jtjkrC| �� � d }~ww )Nr z)can't use prlimit() against PID 0 processrB z2second argument must be a (soft, hard) tuple, got ) rY r �resourcer. r� r� rV �ENOSYSrM )rM � resource_�limitsr� r� r5 r5 r6 �rlimit� s& �� ��zProcess.rlimitc C s | � � d }|�� }t�|d�S )Nr� �?)rl r � PROC_STATUSESr� )rM �letterr5 r5 r6 r� � s zProcess.statusc C s� g }t �| j� d| j� d��}d}|D ]�}| j� d| j� d|� �}zt|�}W n5 ttfy4 d}Y q ty\ } z |jtj krGW Y d }~q|jtj krWt |� W Y d }~q� d }~ww |� d�r�t |�r�| j� d| j� d|� �}z+t|�� }t|�� �� d �} t|�� �� d d�} W d � n1 s�w Y W n ttfy� d}Y qw t| �} t|t|�t| �| | �} |�| � q|r�| �� |S ) Nr� rO FrP Tz/fdinfo/r r� )r� rS rL rY r� rT rU r� rV rW rX r r0 r r r� r� r� r� rs r� r[ ) rM r� r� r� ru rz rt r� r� �posrx rw r� r5 r5 r6 � open_files� sN �� ���� �zProcess.open_filesrB c C s t �|| j�}| �� |S r] )r� r� rY r[ )rM r� r/ r5 r5 r6 r� � s zProcess.net_connectionsc C s t t�| j� d| j� d���S )Nr� rO )r� r� rS rL rY rZ r5 r5 r6 �num_fds� s zProcess.num_fdsc C ru )NrJ r� rZ r5 r5 r6 rJ � s z Process.ppids Uid:\t(\d+)\t(\d+)\t(\d+)c C �6 | � � }|�|�d \}}}t�t|�t|�t|��S r� )rn r� r �puidsr� )rM �_uids_rer� �real� effective�savedr5 r5 r6 �uids� � z Process.uidss Gid:\t(\d+)\t(\d+)\t(\d+)c C r� r� )rn r� r �pgidsr� )rM �_gids_rer� r� r� r� r5 r5 r6 �gids� r� z Process.gidsr] �rB )>rD rE rF r� r� rN rY rM r[ rT r rl rn ro rr rt r� rx rz r{ r~ r� rt rv �getpidr� r� rh r� rf r� r� �HAS_PROC_SMAPSr� r r r� r� r� r� r� r� r� r� r� �HAS_CPU_AFFINITYr� r� r� �HAS_PROC_IO_PRIORITYr� r� �hasattrr� r� r� r� r� r� rJ r� r� r5 r5 r5 r6 rU � s� % " � G � � / rU r� )F)�r� rh r �enumrV rR r r� r r� rE rj �sysr� r r r� r r r r� r r� r r r r r r r r r r r r r r r r r r r r r �__extra__all__r� r �dirr- rt rv r� r� r� r� r� r r� � getpagesizer� rC � byteorderrf r� �IntEnumr� � AF_PACKETr? r@ rA �globalsr_ � __members__�STATUS_RUNNING�STATUS_SLEEPING�STATUS_DISK_SLEEP�STATUS_STOPPED�STATUS_TRACING_STOP� STATUS_ZOMBIE� STATUS_DEAD�STATUS_WAKE_KILL� STATUS_WAKING� STATUS_IDLE� STATUS_PARKEDr� r# r$ r% r& r' r( r) r* r+ r, r- rx r] ri rs ry r� rz r~ r� r� r� r� r� r� r� r� � Exceptionr� r� r� r� r� r� r� r r r r! r6 � net_if_addrsr8 r9 r� r� r� r� � disk_usager� r� r� r r! r: r; rD r^ rE rK rT rU r5 r5 r5 r6 �<module> s< � ������ ��� �� ? < * � ' o /! i K,v _ ! __pycache__/_psbsd.cpython-310.pyc 0000644 00000050141 15030211765 0012730 0 ustar 00 o �h | � @ s� d Z ddlZddlZddlZddlZddlmZ ddlmZ ddlm Z ddl m Z ddl m Z dd l m Z dd l mZ dd l mZ dd l mZ dd l mZ ddl mZ ddl mZ ddl mZ ddl mZ ddl mZ ddl mZ ddl mZ ddl mZ ddl m Z g Z er�ej e j ej e j!ej"e j#ej$e j%ej&e j'ej(e j)ej*e j+iZ,n=er�ej e j ej"e j#ej$e j%ej-e j'ej&e j'ej e j.ej/e j!iZ,n er�ej e j ej"e j#ej$e j%ej&e j'ej e j.ej/e j!iZ,ej0e j1ej2e j3ej4e j5ej6e j7ej8e j9ej:e j;ej<e j=ej>e j?ej@e jAejBe jCejDe jEejFe jGi ZHe�I� ZJejKZKeLed�ZMeLed�ZNeLed�ZOeLed�ZPeQd�i dd�d d�d d �d d �d!d"�d#d$�d%d&�d'd(�d)d*�d+d,�d-d.�d/d0�d1d2�d3d4�d5d6�d7d8�d9d:�d;d<�d=d>�d?d@�dAdB�dCdD�dEdF�dGdH�dIdJ��ZRedKg dL��ZSedMg dN��ZTedOg dP��ZUeUZVedQg dR��ZWedSdT�ZXedUdV�ZYe�r�edWg dX��ZZnedWg dY��ZZdZd[� Z[d\d]� Z\d^d_� Z]eM�r�d`d� Z^ndad� Z^dbe^__dcdd� Z`e�s�e�r�dedf� Zandgdf� Zadhdi� Zbe�r�djdk� Zcne�r�dldk� Zcd�dmdn�Zde jeZeejfZfejgZgejhZhdodp� Zidqdr� Zje�r dsdt� Zkdudv� Zldwdx� Zmdydz� Zned{d|� �Zod}d~� Zpe�r:dd�� Zqn e�rBd�d�� Zqne jqZqd�d�� Zrd�d�� Zsejtd�d�� �ZuG d�d�� d��ZvdS )�z5FreeBSD, OpenBSD and NetBSD platforms implementation.� N)� defaultdict)� namedtuple)� ElementTree� )�_common)�_psposix)� _psutil_bsd)� _psutil_posix)�FREEBSD)�NETBSD)�OPENBSD)� AccessDenied)� NoSuchProcess)� ZombieProcess)� conn_tmap)�conn_to_ntuple)�debug)�memoize)�memoize_when_activated)� usage_percent� per_cpu_times�proc_num_threads�proc_open_files� proc_num_fds�ppid�status�real_uid� � effective_uid� � saved_uid� �real_gid� � effective_gid� � saved_gid� �ttynr� � create_time� �ctx_switches_vol� �ctx_switches_unvol� � read_io_count� �write_io_count� � user_time� �sys_time� � ch_user_time� � ch_sys_time� �rss� �vms� �memtext� �memdata� �memstack� �cpunum� �name� �svmem) �total� available�percent�used�free�active�inactive�buffers�cached�shared�wired� scputimes��user�nice�system�idle�irq�pmem)r< r> �text�data�stack� pcputimes)rX rZ � children_user�children_system� pmmap_grouped�*path rss, private, ref_count, shadow_count� pmmap_ext�6addr, perms path rss, private, ref_count, shadow_count�sdiskio)� read_count� write_count� read_bytes� write_bytes� read_time� write_time� busy_time)ri rj rk rl c C s� t �� } trP| \}}}}}}tdd��,}|D ]!}|�d�r(t|�� d �d } q|�d�r7t|�� d �d } qW d � n1 sBw Y || } || } n| \}}}}}}} } || | } || | } t|| |dd�} t|| | | |||| || |� S )Nz /proc/meminfo�rbs Buffers:r i s MemShared:��round_) �cext� virtual_memr �open� startswith�int�splitr rJ )�memrK rO rP rQ rU rS �f�linerR rT rN �availrM � r} �@/usr/local/CyberCP/lib/python3.10/site-packages/psutil/_psbsd.py�virtual_memory� s> �� � �r c C s4 t �� \} }}}}t|| dd�}t�| |||||�S )z@System swap memory as (total, used, free, sin, sout) namedtuple.r rq )rs �swap_memr r �sswap)rK rN rO �sin�soutrM r} r} r~ � swap_memory� s r� c C s" t �� \} }}}}t| ||||�S )z,Return system per-CPU times as a namedtuple.)rs � cpu_timesrV rW r} r} r~ r� � s r� c C s> g } t �� D ]}|\}}}}}t|||||�}| �|� q| S )�(Return system CPU times as a namedtuple.)rs r rV �append)�ret�cpu_trX rY rZ r[ r\ �itemr} r} r~ r � s c C s2 t � dkr t� gS tjrd} t| ��dt_t� gS )r� r �&supported only starting from FreeBSD 8T)�cpu_count_logicalr� r � __called__�NotImplementedError)�msgr} r} r~ r s Fc C � t �� S )z0Return the number of logical CPUs in the system.)rs r� r} r} r} r~ r� � r� c C s t � dkrdS d S )Nr )r� r} r} r} r~ �cpu_count_cores � r� c C s� d} t �� }|dur5|�d�}|dkr5|d|d � }t�|�}zt|�d��p)d} W |�� n|�� w | s>t� dkr>dS | S )z-Return the number of CPU cores in the system.Nz </groups>���r+ zgroup/children/group/cpur ) rs � cpu_topology�rfindr � fromstring�len�findall�clearr� )r� �s�index�rootr} r} r~ r� s c C s� t r t�� \} }}}}nBtrAt�� \} }}}}}}tdd��}|D ]}|�d�r0t|�� d �}q!W d � n1 s;w Y n trNt�� \} }}}}}}t � | |||�S )z*Return various CPU stats as a named tuple.z /proc/statrp s intrr N) r rs � cpu_statsr ru rv rw rx r r � scpustats) �ctxsw�intrs� soft_intrs�syscalls�_traps�_faults�_forksrz r{ r} r} r~ r� ; s" � �� ���r� c C s� g } t � }t|�D ]]}z t�|�\}}W n ty Y q w |r\zt|�d�d �d�d �}W n ttfy= d}Y nw zt|�d�d �d�d �}W n ttfy[ d}Y nw | � t � |||�� q | S )z�Return frequency metrics for CPUs. As of Dec 2018 only CPU 0 appears to be supported by FreeBSD and all other cores match the frequency of CPU 0. � r� �/r N) r� �rangers �cpu_freqr� rw rx � IndexError� ValueErrorr� r �scpufreq)r� �num_cpus�cpu�current�available_freq�min_freq�max_freqr} r} r~ r� ] s* � � �r� c C s t t�� �} t�| dd�gS )Ng )�floatrs r� r r� )�currr} r} r~ r� w s c C s@ g }t �� }|D ]}|\}}}}t�||||�}|�|� q|S )z�Return mounted disk partitions as a list of namedtuples. 'all' argument is ignored, see: https://github.com/giampaolo/psutil/issues/906. )rs �disk_partitionsr � sdiskpartr� ) �all�retlist� partitions� partition�device� mountpoint�fstype�opts�ntupler} r} r~ r� � s r� c C s� t � �� } i }| D ]M}zt�|�}t�|�}t�|�\}}W n ty7 } z |jtjkr-� W Y d}~q d}~ww t t d�rBt � |�}d� |�}d|v } t � | ||||�||< q |S )z)Get NIC stats (isup, duplex, speed, mtu).N� NicDuplex�,�running)�net_io_counters�keys� cext_posix� net_if_mtu� net_if_flags�net_if_duplex_speed�OSError�errno�ENODEV�hasattrr r� �join� snicstats) �namesr� rH �mtu�flags�duplex�speed�err� output_flags�isupr} r} r~ � net_if_stats� s* ��� �r� c C s� t | \}}t� }trt�d||�}ntr t�d| �}nt�||�}|D ]}|\}}}} } } } t|||| | | t| �} |�| � q$t |�S )z System-wide network connections.r� ) r �setr rs �net_connectionsr r � TCP_STATUSES�add�list)�kind�families�typesr� �rawlistr� �fd�fam�type�laddr�raddrr �pid�ntr} r} r~ r� � s � r� c C sb z t �� \} }}W n ty Y dS w |dk}|r tj}n |dkr&tj}n|d }t�| ||�S )zReturn battery info.Nr r� �< )rs �sensors_batteryr� r �POWER_TIME_UNLIMITED�POWER_TIME_UNKNOWN�sbattery)rM �minsleft� power_plugged�secsleftr} r} r~ r� � s �r� c C st t t�} t� }t|�D ],}z!t�|�\}}|dkrd}d|� �}| d �t�||||�� W q t y7 Y q w | S )z?Return CPU cores temperatures if available, else an empty dict.r NzCore �coretemp) r r� r� r� rs �sensors_cpu_temperaturer� r �shwtempr� )r� r� r� r� �highrH r} r} r~ �sensors_temperatures� s � �r� c C r� )z:The system boot time expressed in seconds since the epoch.)rs � boot_timer} r} r} r~ r� � r� r� c C sf g } t �� }|D ](}|\}}}}}|dkrtsJ �d}|dkr qt�||p&d|||�}| �|� q| S )z:Return currently connected users as a list of namedtuples.r� N�~)rs �usersr r �suserr� ) r� r� r� rX �tty�hostname�tstampr� r� r} r} r~ r� � s r� c C s: z t d��� W dS ty Y dS ty Y dS w )Nr FT)�ProcessrH r r r} r} r} r~ � _pid_0_exists s � �r� c C s* t �� } trd| vrt� r| �dd� | S )z7Returns a list of PIDs currently running on the system.r )rs �pidsr r� �insert)r� r} r} r~ r� s r� c C s t �| �}|s | t� v S dS )NT�r � pid_existsr� �r� �existsr} r} r~ r ) s r c C s t �| �}|s dS | t� v S )NFr r r} r} r~ r 4 s c C s: zt �| �td }t�|�tjkW S ty Y dS w )Nr F)rs �proc_oneshot_info�kinfo_proc_map� PROC_STATUSES�getr � STATUS_ZOMBIEr� )r� �str} r} r~ � is_zombieB s �r c s t �� �� fdd��}|S )z`Decorator which translates bare OSError exceptions into NoSuchProcess and AccessDenied. c s� | j | j| j}}}z � | g|�R i |��W S ty4 } zt|�r*t|||�|�t||�|�d }~w tyE } zt||�|�d }~w t y` } z|dkr[dt � v r[t||�|�� d }~ww )Nr ) r� �_ppid�_name�ProcessLookupErrorr r r �PermissionErrorr r� r� )�self�args�kwargsr� r rH r� ��funr} r~ �wrapperO s" � � ��z wrap_exceptions.<locals>.wrapper)� functools�wraps)r r r} r r~ �wrap_exceptionsJ s r c c s� � | j | j| j}}}zdV W dS ttfy2 } zt| j �r(t|||�|�t||�|�d}~w tyC } zt ||�|�d}~ww )z8Same as above, for routines relying on reading /proc fs.N) r� r r r �FileNotFoundErrorr r r r r )�instr� rH r r� r} r} r~ �wrap_exceptions_procfsb s � � ��r c @ s� e Zd ZdZg d�Zdd� Zdd� Zeedd� ��Z d d � Z d d � Z ed d� �Z edd� �Z edd� �Zedd� �Zedd� �Zedd� �Zedd� �Zedd � �Zed d � �Zer`ed d � �Zed!d"� �ZeZed#d$� �Zed%d&� �Zed'd(� �Zed)d*� �Z edLd,d-��Z edMd/d0��Z ed1d2� �Z ed3d4� �Z ed5d6� �Z!ed7d8� �Z"ed9d:� �Z#e$d;d<�Z%e$d;d=�Z&d>d?� Z'e(r�ed@dA� �Z)ne'Z)e*r�edBdC� �Z+ne'Z+er�edDdE� �Z,edFdG� �Z-edHdI� �Z.edMdJdK��Z/d.S d.S )Nr� z1Wrapper class around underlying C implementation.)�_cacher r r� c C s || _ d | _d | _d S �N)r� r r )r r� r} r} r~ �__init__y s zProcess.__init__c C s t �| j� dS )z+Raise NSP if the process disappeared on us.N)rs � proc_namer� �r r} r} r~ � _assert_alive~ s zProcess._assert_alivec C s$ t �| j�}t|�tt�ksJ �|S )z;Retrieves multiple process info in one shot as a raw tuple.)rs r r� r� r �r r� r} r} r~ �oneshot� s zProcess.oneshotc C � | j �| � d S r )r# �cache_activater r} r} r~ � oneshot_enter� � zProcess.oneshot_enterc C r$ r )r# �cache_deactivater r} r} r~ � oneshot_exit� r' zProcess.oneshot_exitc C s( | � � td }|d ur|S t�| j�S )NrH )r# r rs r r� )r rH r} r} r~ rH � s z Process.namec C s� t r| jdkr dS t�| j�S tr8| jdkrdS t| �� t�d| j� d��W d � S 1 s1w Y d S dd l}| � � }|rK|� |d �pJdS dS )Nr � z/proc/z/exe) r r� rs �proc_exer r �os�readlink�shutil�cmdline�which)r r. r/ r} r} r~ �exe� s $�z Process.exec C s� t r | jdkr g S trYzt�| j�W S tyX } z:|jtjkrS| j| j| j }}}t | j�r7t |||�|�t | j�sCt |||�|�td|�d�� g W Y d }~S � d }~ww t�| j�S )Nr z ignoring z and returning an empty list)r r� r rs � proc_cmdliner� r� �EINVALr r r r r r r )r r� r� rH r r} r} r~ r/ � s$ �� zProcess.cmdlinec C � t �| j�S r )rs � proc_environr� r r} r} r~ �environ� � zProcess.environc C s8 | � � td }t�� }z|| W S ty Y d S w )Nr( )r# r r �get_terminal_map�KeyError)r �tty_nr�tmapr} r} r~ �terminal� s �zProcess.terminalc C s | � � td | _| jS )Nr )r# r r r r} r} r~ r � s z Process.ppidc C �. | � � }t�|td |td |td �S )Nr r r )r# r �puidsr �r �rawtupler} r} r~ �uids� � �z Process.uidsc C r= )Nr" r$ r& )r# r �pgidsr r? r} r} r~ �gids� rB z Process.gidsc C s8 | � � }t�|td |td |td |td �S )Nr4 r6 r8 r: )r# r ra r r? r} r} r~ r� � s �zProcess.cpu_timesc C � | � � td S )NrF �r# r r r} r} r~ �cpu_num� � zProcess.cpu_numc C s@ | � � }t|td |td |td |td |td �S )Nr< r> r@ rB rD )r# r] r r? r} r} r~ � memory_info s �zProcess.memory_infoc C rE )Nr* rF r r} r} r~ r* rH zProcess.create_timec C s t rt�| j�S t| �� �S r )�HAS_PROC_NUM_THREADSrs r r� r� �threadsr r} r} r~ � num_threads s zProcess.num_threadsc C s$ | � � }t�|td |td �S )Nr, r. )r# r �pctxswr r? r} r} r~ �num_ctx_switches s �zProcess.num_ctx_switchesc C sH t �| j�}g }|D ]\}}}t�|||�}|�|� q tr"| �� |S r )rs � proc_threadsr� r �pthreadr� r r! )r r� r� � thread_id�utime�stimer� r} r} r~ rK ! s zProcess.threads�inetc C s� t | \}}g }trt�| j|�}ntr t�| j||�}nt�| j||�}|D ](}|d d� \}}} } } } tr@||vs?| |vr@q't||| | | | t �} |� | � q'| � � |S )Nr% ) r r rs r� r� r �proc_net_connectionsr r r� r� r! )r r� r� r� r� r� r� r� r� r� r� r� r r� r} r} r~ r� - s$ � zProcess.net_connectionsNc C s t �| j|| j�S r )r �wait_pidr� r )r �timeoutr} r} r~ �waitF r� z Process.waitc C r4 r )r� � getpriorityr� r r} r} r~ �nice_getJ r7 zProcess.nice_getc C s t �| j|�S r )r� � setpriorityr� )r �valuer} r} r~ �nice_setN s zProcess.nice_setc C s | � � td }t�|d�S )Nr �?)r# r r r )r �coder} r} r~ r R s zProcess.statusc C s( | � � }t�|td |td dd�S )Nr0 r2 r� )r# r �pior r? r} r} r~ � io_countersX s �zProcess.io_countersc C s8 t r | jdkr dS ts trt�| j�S ttrd��d��)z)Return process current working directory.r r* r� )r r� r �HAS_PROC_OPEN_FILESrs �proc_cwdr� r r r} r} r~ �cwdb s ��z Process.cwd�mmapre rg c C s t �r )r� r r} r} r~ �_not_implementedy s zProcess._not_implementedc C s t �| j�}dd� |D �S )z8Return files opened by process as a list of namedtuples.c S s g | ] \}}t �||��qS r} )r � popenfile)�.0�pathr� r} r} r~ � <listcomp>� s z&Process.open_files.<locals>.<listcomp>)rs r r� )r r� r} r} r~ � open_files� s zProcess.open_filesc C s t �| j�}tr | �� |S )z=Return the number of file descriptors opened by this process.)rs r r� r r! r" r} r} r~ �num_fds� s zProcess.num_fdsc C r4 r )rs �proc_cpu_affinity_getr� r r} r} r~ �cpu_affinity_get� r7 zProcess.cpu_affinity_getc C s� t ttt� ���}|D ]}||vr d|�d|� d�}t|��q z t�| j|� W d S tyV } z!|j t j t j hv rQ|D ]}||vrPd|�d|� d�}t|�|�q<� d }~ww )Nz invalid CPU z (choose between �)) r� r� r� r r� rs �proc_cpu_affinity_setr� r� r� r3 �EDEADLK)r �cpus�allcpusr� r� r� r} r} r~ �cpu_affinity_set� s* ��� ���zProcess.cpu_affinity_setc C r4 r )rs �proc_memory_mapsr� r r} r} r~ � memory_maps� r7 zProcess.memory_mapsc C sN |d u r t �| j|�S t|�dkrd|��}t|��|\}}t �| j|||�S )Nr z2second argument must be a (soft, hard) tuple, got )rs �proc_getrlimitr� r� r� �proc_setrlimit)r �resource�limitsr� �soft�hardr} r} r~ �rlimit� s ��zProcess.rlimit)rT r )0�__name__� __module__� __qualname__�__doc__� __slots__r r! r r r# r& r) rH r1 r/ r6 r< r rA rD r� r rG rI �memory_full_infor* rL rN rK r� rX rZ r] r ra rd r �nt_mmap_grouped� nt_mmap_extrf rb rk �HAS_PROC_NUM_FDSrl rn rt rv r} r} r} r} r~ r� t s� �� �r� r} )F)wr� � contextlibr� r r, � collectionsr r � xml.etreer r* r r r rs r r� r r r r r r r r r r r r �__extra__all__�SIDL� STATUS_IDLE�SRUN�STATUS_RUNNING�SSLEEP�STATUS_SLEEPING�SSTOP�STATUS_STOPPED�SZOMBr �SWAIT�STATUS_WAITING�SLOCK� STATUS_LOCKEDr �SDEAD� STATUS_WAKING�SONPROC�TCPS_ESTABLISHED�CONN_ESTABLISHED� TCPS_SYN_SENT� CONN_SYN_SENT�TCPS_SYN_RECEIVED� CONN_SYN_RECV�TCPS_FIN_WAIT_1�CONN_FIN_WAIT1�TCPS_FIN_WAIT_2�CONN_FIN_WAIT2�TCPS_TIME_WAIT�CONN_TIME_WAIT� TCPS_CLOSED� CONN_CLOSE�TCPS_CLOSE_WAIT�CONN_CLOSE_WAIT� TCPS_LAST_ACK� CONN_LAST_ACK� TCPS_LISTEN� CONN_LISTEN� TCPS_CLOSING� CONN_CLOSING�PSUTIL_CONN_NONE� CONN_NONEr� � getpagesize�PAGESIZE�AF_LINKr� �HAS_PER_CPU_TIMESrJ rb r� �dictr rJ rV r] �pfullmemra rd rf rh r r� r� r r� r� r� r� r� r� � disk_usage�disk_io_countersr� � net_if_addrsr� r� r� r� r� r� r� r� r r r �contextmanagerr r� r} r} r} r~ �<module> sl � �� � �������� � � � � �������������$����� / __pycache__/_psosx.cpython-310.pyc 0000644 00000031460 15030211765 0012774 0 ustar 00 o �h> � @ s� d Z ddlZddlZddlZddlmZ ddlmZ ddlmZ ddlm Z ddlm Z dd lm Z dd lmZ dd lmZ dd lmZ dd lmZ ddlmZ ddlmZ ddlmZ ddlmZ g Ze �� Ze jZe jeje j ej e j ej e j ej!e j"ej#e j$ej%e j&ej'e j(ej)e j*ej+e j,ej-e j.ej/e j0ej1i Z2e j3ej4e j5ej6e j7ej8e j9ej:e j;ej<iZ=e>dddddddddddd� Z?e>ddddddddd �Z@ed g d ��ZAed g d ��ZBed!g d"��ZCed#eCjDd$ �ZEd%d&� ZFd'd(� ZGd)d*� ZHd+d,� ZId-d.� ZJd/d0� ZKd1d2� ZLd3d4� ZMejNZNe jOZOdKd6d7�ZPd8d9� ZQe jRZRe jSZSdLd;d<�ZTd=d>� ZUd?d@� ZVdAdB� ZWdCdD� ZXejYZYdEdF� ZZdGdH� Z[G dIdJ� dJ�Z\dS )Mz macOS platform implementation.� N)� namedtuple� )�_common)�_psposix)� _psutil_osx)� _psutil_posix)� AccessDenied)� NoSuchProcess)� ZombieProcess)� conn_tmap)�conn_to_ntuple)� isfile_strict)�memoize_when_activated)�parse_environ_block)� usage_percent� � � � � � � � � ) �ppid�ruid�euid�suid�rgid�egid�sgid�ttynr�ctime�status�name)�cpuutime�cpustime�rss�vms�pfaults�pageins� numthreads�volctxsw� scputimes��user�nice�system�idle�svmem)�total� available�percent�used�free�active�inactive�wired�pmem)r' r( r) r* �pfullmem)�ussc C sT t �� \} }}}}}|| }|| }||8 }t| | | dd�}t| |||||||�S )z&System virtual memory as a namedtuple.r ��round_)�cext� virtual_memr r3 ) r4 r9 r: r; r8 � speculative�availr7 r6 � rE �@/usr/local/CyberCP/lib/python3.10/site-packages/psutil/_psosx.py�virtual_memoryo s rG c C s4 t �� \} }}}}t|| dd�}t�| |||||�S )z=Swap system memory as a (total, used, free, sin, sout) tuple.r r? )rA �swap_memr r �sswap)r4 r7 r8 �sin�soutr6 rE rE rF � swap_memory~ s rL c C s t �� \} }}}t| |||�S )z(Return system CPU times as a namedtuple.)rA � cpu_timesr- r. rE rE rF rM � s rM c C s: g } t �� D ]}|\}}}}t||||�}| �|� q| S )z)Return system CPU times as a named tuple.)rA � per_cpu_timesr- �append)�ret�cpu_tr/ r0 r1 r2 �itemrE rE rF rN � s rN c C � t �� S )z0Return the number of logical CPUs in the system.)rA �cpu_count_logicalrE rE rE rF rT � � rT c C rS )z-Return the number of CPU cores in the system.)rA �cpu_count_coresrE rE rE rF rV � rU rV c C s" t �� \} }}}}t�| |||�S �N)rA � cpu_statsr � scpustats)� ctx_switches� interrupts�soft_interrupts�syscalls�_trapsrE rE rF rX � s ��rX c C s t �� \} }}t�| ||�gS )z�Return CPU frequency. On macOS per-cpu frequency is not supported. Also, the returned frequency never changes, see: https://arstechnica.com/civis/viewtopic.php?f=19&t=465002. )rA �cpu_freqr �scpufreq)�curr�min_�max_rE rE rF r_ � s r_ Fc C sj g }t �� }|D ]*}|\}}}}|dkrd}| s%tj�|�r$tj�|�s%qt�||||�}|�|� q|S )z8Return mounted disk partitions as a list of namedtuples.�none� ) rA �disk_partitions�os�path�isabs�existsr � sdiskpartrO ) �all�retlist� partitions� partition�device� mountpoint�fstype�opts�ntuplerE rE rF rf � s rf c C sb z t �� \} }}W n ty Y dS w |dk}|r tj}n |dkr&tj}n|d }t�| ||�S )zReturn battery information.Nr ����< )rA �sensors_battery�NotImplementedErrorr �POWER_TIME_UNLIMITED�POWER_TIME_UNKNOWN�sbattery)r6 �minsleft� power_plugged�secsleftrE rE rF rw � s �rw �inetc C sf g }t � D ]+}z t|��| �}W n ty Y qw |r0|D ]}t|�|g }|�tj|� � q q|S )z System-wide network connections.)�pids�Process�net_connectionsr �listrO r �sconn)�kindrP �pid�cons�crE rE rF r� � s ��r� c C s� t � �� } i }| D ]M}zt�|�}t�|�}t�|�\}}W n ty7 } z |jtjkr-� W Y d}~q d}~ww t t d�rBt � |�}d� |�}d|v } t � | ||||�||< q |S )z)Get NIC stats (isup, duplex, speed, mtu).N� NicDuplex�,�running)�net_io_counters�keys� cext_posix� net_if_mtu� net_if_flags�net_if_duplex_speed�OSError�errno�ENODEV�hasattrr r� �join� snicstats) �namesrP r$ �mtu�flags�duplex�speed�err� output_flags�isuprE rE rF � net_if_stats s* ��� �r� c C rS )z:The system boot time expressed in seconds since the epoch.)rA � boot_timerE rE rE rF r� rU r� c C s\ g } t �� }|D ]#}|\}}}}}|dkrq|sqt�||p d|p"d||�}| �|� q| S )z:Return currently connected users as a list of namedtuples.�~N)rA �usersr �suserrO ) rm �rawlistrR r/ �tty�hostname�tstampr� �ntrE rE rF r� # s r� c C sf t �� } d| vr1ztd��� | �dd� W | S ty! Y | S ty0 | �dd� Y | S w | S �Nr )rA r� r� � create_time�insertr r )�lsrE rE rF r� 7 s � ��r� c C s4 zt �| �td }|t jkW S ty Y dS w )Nr# F)rA �proc_kinfo_oneshot�kinfo_proc_map�SZOMBr� )r� �strE rE rF � is_zombieJ s �r� c s t �� �� fdd��}|S )z`Decorator which translates bare OSError exceptions into NoSuchProcess and AccessDenied. c s� | j | j| j}}}z � | g|�R i |��W S ty4 } zt|�r*t|||�|�t||�|�d }~w tyE } zt||�|�d }~ww rW ) r� �_ppid�_name�ProcessLookupErrorr� r r �PermissionErrorr )�self�args�kwargsr� r r$ r� ��funrE rF �wrapperW s � ��z wrap_exceptions.<locals>.wrapper)� functools�wraps)r� r� rE r� rF �wrap_exceptionsR s r� c @ sh e Zd ZdZg d�Zdd� Zeedd� ��Zeedd� ��Z d d � Z d d � Z ed d� �Z edd� �Z edd� �Zedd� �Zedd� �Zedd� �Zedd� �Zedd � �Zed d � �Zed d � �Zed!d"� �Zed#d$� �Zed%d&� �Zed'd(� �Zed)d*� �Zed+d,� �Zed=d.d/��Z ed0d1� �Z ed>d3d4��Z ed5d6� �Z ed7d8� �Z ed9d:� �Z!ed;d<� �Z"d2S )?r� z1Wrapper class around underlying C implementation.)�_cacher� r� r� c C s || _ d | _d | _d S rW )r� r� r� )r� r� rE rE rF �__init__k s zProcess.__init__c C �$ t �| j�}t|�tt�ksJ �|S rW )rA r� r� �lenr� �r� rP rE rE rF �_get_kinfo_procp � zProcess._get_kinfo_procc C r� rW )rA �proc_pidtaskinfo_oneshotr� r� �pidtaskinfo_mapr� rE rE rF �_get_pidtaskinfox r� zProcess._get_pidtaskinfoc C � | j �| � | j�| � d S rW )r� �cache_activater� �r� rE rE rF � oneshot_enter� � zProcess.oneshot_enterc C r� rW )r� �cache_deactivater� r� rE rE rF � oneshot_exit� r� zProcess.oneshot_exitc C s( | � � td }|d ur|S t�| j�S )Nr$ )r� r� rA � proc_namer� )r� r$ rE rE rF r$ � s z Process.namec C � t �| j�S rW )rA �proc_exer� r� rE rE rF �exe� � z Process.exec C r� rW )rA � proc_cmdliner� r� rE rE rF �cmdline� r� zProcess.cmdlinec C s t t�| j��S rW )r rA � proc_environr� r� rE rE rF �environ� � zProcess.environc C s | � � td | _| jS )Nr )r� r� r� r� rE rE rF r � s z Process.ppidc C r� rW )rA �proc_cwdr� r� rE rE rF �cwd� r� z Process.cwdc C �. | � � }t�|td |td |td �S )Nr r r �r� r �puidsr� �r� �rawtuplerE rE rF �uids� � �z Process.uidsc C r� )Nr r r r� r� rE rE rF �gids� r� z Process.gidsc C s8 | � � td }t�� }z|| W S ty Y d S w )Nr! )r� r� r �get_terminal_map�KeyError)r� �tty_nr�tmaprE rE rF �terminal� s �zProcess.terminalc C s6 | � � }t|td |td |td |td �S )Nr' r( r) r* )r� r< r� r� rE rE rF � memory_info� s �zProcess.memory_infoc C s" | � � }t�| j�}t||f � S rW )r� rA �proc_memory_ussr� r= )r� � basic_memr> rE rE rF �memory_full_info� s zProcess.memory_full_infoc C s( | � � }t�|td |td dd�S )Nr% r& g )r� r � pcputimesr� r� rE rE rF rM � s �zProcess.cpu_timesc C � | � � td S )Nr" )r� r� r� rE rE rF r� � r� zProcess.create_timec C � | � � td }t�|d�S )Nr, r )r� r� r �pctxsw)r� �volrE rE rF �num_ctx_switches� s zProcess.num_ctx_switchesc C r� )Nr+ )r� r� r� rE rE rF � num_threads� r� zProcess.num_threadsc C sN | j dkrg S g }t�| j �}|D ]\}}t|�r$t�||�}|�|� q|S r� )r� rA �proc_open_filesr r � popenfilerO )r� �filesr� rh �fdrt rE rE rF � open_files� s �zProcess.open_filesr c C s\ t | \}}t�| j||�}g }|D ]}|\}}} } } } t||| | | | t�} |�| � q|S rW )r rA �proc_net_connectionsr� r � TCP_STATUSESrO )r� r� �families�typesr� rP rR r� �fam�type�laddr�raddrr# r� rE rE rF r� � s � zProcess.net_connectionsc C s | j dkrdS t�| j �S r� )r� rA � proc_num_fdsr� rE rE rF �num_fds s zProcess.num_fdsNc C s t �| j|| j�S rW )r �wait_pidr� r� )r� �timeoutrE rE rF �wait s z Process.waitc C r� rW )r� � getpriorityr� r� rE rE rF �nice_get r� zProcess.nice_getc C s t �| j|�S rW )r� � setpriorityr� )r� �valuerE rE rF �nice_set s zProcess.nice_setc C r� )Nr# �?)r� r� � PROC_STATUSES�get)r� �coderE rE rF r# s zProcess.statusc C s< t �| j�}g }|D ]\}}}t�|||�}|�|� q |S rW )rA � proc_threadsr� r �pthreadrO )r� r� rm � thread_id�utime�stimert rE rE rF �threads s zProcess.threads�r rW )#�__name__� __module__� __qualname__�__doc__� __slots__r� r� r r� r� r� r� r$ r� r� r� r r� r� r� r� r� r� rM r� r� r� r� r� r r r r r# r rE rE rE rF r� f st r� )Fr )]r r� r� rg � collectionsr re r r r rA r r� r r r r r r r r r �__extra__all__� getpagesize�PAGESIZE�AF_LINK�TCPS_ESTABLISHED�CONN_ESTABLISHED� TCPS_SYN_SENT� CONN_SYN_SENT�TCPS_SYN_RECEIVED� CONN_SYN_RECV�TCPS_FIN_WAIT_1�CONN_FIN_WAIT1�TCPS_FIN_WAIT_2�CONN_FIN_WAIT2�TCPS_TIME_WAIT�CONN_TIME_WAIT� TCPS_CLOSED� CONN_CLOSE�TCPS_CLOSE_WAIT�CONN_CLOSE_WAIT� TCPS_LAST_ACK� CONN_LAST_ACK� TCPS_LISTEN� CONN_LISTEN� TCPS_CLOSING� CONN_CLOSING�PSUTIL_CONN_NONE� CONN_NONEr� �SIDL� STATUS_IDLE�SRUN�STATUS_RUNNING�SSLEEP�STATUS_SLEEPING�SSTOP�STATUS_STOPPEDr� � STATUS_ZOMBIEr �dictr� r� r- r3 r< �_fieldsr= rG rL rM rN rT rV rX r_ � disk_usage�disk_io_countersrf rw r� � net_if_addrsr� r� r� r� r� � pid_existsr� r� r� rE rE rE rF �<module> s� ����� _psosx.py 0000644 00000037005 15030211765 0006436 0 ustar 00 # Copyright (c) 2009, Giampaolo Rodola'. All rights reserved. # Use of this source code is governed by a BSD-style license that can be # found in the LICENSE file. """macOS platform implementation.""" import errno import functools import os from collections import namedtuple from . import _common from . import _psposix from . import _psutil_osx as cext from . import _psutil_posix as cext_posix from ._common import AccessDenied from ._common import NoSuchProcess from ._common import ZombieProcess from ._common import conn_tmap from ._common import conn_to_ntuple from ._common import isfile_strict from ._common import memoize_when_activated from ._common import parse_environ_block from ._common import usage_percent __extra__all__ = [] # ===================================================================== # --- globals # ===================================================================== PAGESIZE = cext_posix.getpagesize() AF_LINK = cext_posix.AF_LINK TCP_STATUSES = { cext.TCPS_ESTABLISHED: _common.CONN_ESTABLISHED, cext.TCPS_SYN_SENT: _common.CONN_SYN_SENT, cext.TCPS_SYN_RECEIVED: _common.CONN_SYN_RECV, cext.TCPS_FIN_WAIT_1: _common.CONN_FIN_WAIT1, cext.TCPS_FIN_WAIT_2: _common.CONN_FIN_WAIT2, cext.TCPS_TIME_WAIT: _common.CONN_TIME_WAIT, cext.TCPS_CLOSED: _common.CONN_CLOSE, cext.TCPS_CLOSE_WAIT: _common.CONN_CLOSE_WAIT, cext.TCPS_LAST_ACK: _common.CONN_LAST_ACK, cext.TCPS_LISTEN: _common.CONN_LISTEN, cext.TCPS_CLOSING: _common.CONN_CLOSING, cext.PSUTIL_CONN_NONE: _common.CONN_NONE, } PROC_STATUSES = { cext.SIDL: _common.STATUS_IDLE, cext.SRUN: _common.STATUS_RUNNING, cext.SSLEEP: _common.STATUS_SLEEPING, cext.SSTOP: _common.STATUS_STOPPED, cext.SZOMB: _common.STATUS_ZOMBIE, } kinfo_proc_map = dict( ppid=0, ruid=1, euid=2, suid=3, rgid=4, egid=5, sgid=6, ttynr=7, ctime=8, status=9, name=10, ) pidtaskinfo_map = dict( cpuutime=0, cpustime=1, rss=2, vms=3, pfaults=4, pageins=5, numthreads=6, volctxsw=7, ) # ===================================================================== # --- named tuples # ===================================================================== # fmt: off # psutil.cpu_times() scputimes = namedtuple('scputimes', ['user', 'nice', 'system', 'idle']) # psutil.virtual_memory() svmem = namedtuple( 'svmem', ['total', 'available', 'percent', 'used', 'free', 'active', 'inactive', 'wired']) # psutil.Process.memory_info() pmem = namedtuple('pmem', ['rss', 'vms', 'pfaults', 'pageins']) # psutil.Process.memory_full_info() pfullmem = namedtuple('pfullmem', pmem._fields + ('uss', )) # fmt: on # ===================================================================== # --- memory # ===================================================================== def virtual_memory(): """System virtual memory as a namedtuple.""" total, active, inactive, wired, free, speculative = cext.virtual_mem() # This is how Zabbix calculate avail and used mem: # https://github.com/zabbix/zabbix/blob/master/src/libs/zbxsysinfo/osx/memory.c # Also see: https://github.com/giampaolo/psutil/issues/1277 avail = inactive + free used = active + wired # This is NOT how Zabbix calculates free mem but it matches "free" # cmdline utility. free -= speculative percent = usage_percent((total - avail), total, round_=1) return svmem(total, avail, percent, used, free, active, inactive, wired) def swap_memory(): """Swap system memory as a (total, used, free, sin, sout) tuple.""" total, used, free, sin, sout = cext.swap_mem() percent = usage_percent(used, total, round_=1) return _common.sswap(total, used, free, percent, sin, sout) # ===================================================================== # --- CPU # ===================================================================== def cpu_times(): """Return system CPU times as a namedtuple.""" user, nice, system, idle = cext.cpu_times() return scputimes(user, nice, system, idle) def per_cpu_times(): """Return system CPU times as a named tuple.""" ret = [] for cpu_t in cext.per_cpu_times(): user, nice, system, idle = cpu_t item = scputimes(user, nice, system, idle) ret.append(item) return ret def cpu_count_logical(): """Return the number of logical CPUs in the system.""" return cext.cpu_count_logical() def cpu_count_cores(): """Return the number of CPU cores in the system.""" return cext.cpu_count_cores() def cpu_stats(): ctx_switches, interrupts, soft_interrupts, syscalls, _traps = ( cext.cpu_stats() ) return _common.scpustats( ctx_switches, interrupts, soft_interrupts, syscalls ) def cpu_freq(): """Return CPU frequency. On macOS per-cpu frequency is not supported. Also, the returned frequency never changes, see: https://arstechnica.com/civis/viewtopic.php?f=19&t=465002. """ curr, min_, max_ = cext.cpu_freq() return [_common.scpufreq(curr, min_, max_)] # ===================================================================== # --- disks # ===================================================================== disk_usage = _psposix.disk_usage disk_io_counters = cext.disk_io_counters def disk_partitions(all=False): """Return mounted disk partitions as a list of namedtuples.""" retlist = [] partitions = cext.disk_partitions() for partition in partitions: device, mountpoint, fstype, opts = partition if device == 'none': device = '' if not all: if not os.path.isabs(device) or not os.path.exists(device): continue ntuple = _common.sdiskpart(device, mountpoint, fstype, opts) retlist.append(ntuple) return retlist # ===================================================================== # --- sensors # ===================================================================== def sensors_battery(): """Return battery information.""" try: percent, minsleft, power_plugged = cext.sensors_battery() except NotImplementedError: # no power source - return None according to interface return None power_plugged = power_plugged == 1 if power_plugged: secsleft = _common.POWER_TIME_UNLIMITED elif minsleft == -1: secsleft = _common.POWER_TIME_UNKNOWN else: secsleft = minsleft * 60 return _common.sbattery(percent, secsleft, power_plugged) # ===================================================================== # --- network # ===================================================================== net_io_counters = cext.net_io_counters net_if_addrs = cext_posix.net_if_addrs def net_connections(kind='inet'): """System-wide network connections.""" # Note: on macOS this will fail with AccessDenied unless # the process is owned by root. ret = [] for pid in pids(): try: cons = Process(pid).net_connections(kind) except NoSuchProcess: continue else: if cons: for c in cons: c = list(c) + [pid] ret.append(_common.sconn(*c)) return ret def net_if_stats(): """Get NIC stats (isup, duplex, speed, mtu).""" names = net_io_counters().keys() ret = {} for name in names: try: mtu = cext_posix.net_if_mtu(name) flags = cext_posix.net_if_flags(name) duplex, speed = cext_posix.net_if_duplex_speed(name) except OSError as err: # https://github.com/giampaolo/psutil/issues/1279 if err.errno != errno.ENODEV: raise else: if hasattr(_common, 'NicDuplex'): duplex = _common.NicDuplex(duplex) output_flags = ','.join(flags) isup = 'running' in flags ret[name] = _common.snicstats( isup, duplex, speed, mtu, output_flags ) return ret # ===================================================================== # --- other system functions # ===================================================================== def boot_time(): """The system boot time expressed in seconds since the epoch.""" return cext.boot_time() def users(): """Return currently connected users as a list of namedtuples.""" retlist = [] rawlist = cext.users() for item in rawlist: user, tty, hostname, tstamp, pid = item if tty == '~': continue # reboot or shutdown if not tstamp: continue nt = _common.suser(user, tty or None, hostname or None, tstamp, pid) retlist.append(nt) return retlist # ===================================================================== # --- processes # ===================================================================== def pids(): ls = cext.pids() if 0 not in ls: # On certain macOS versions pids() C doesn't return PID 0 but # "ps" does and the process is querable via sysctl(): # https://travis-ci.org/giampaolo/psutil/jobs/309619941 try: Process(0).create_time() ls.insert(0, 0) except NoSuchProcess: pass except AccessDenied: ls.insert(0, 0) return ls pid_exists = _psposix.pid_exists def is_zombie(pid): try: st = cext.proc_kinfo_oneshot(pid)[kinfo_proc_map['status']] return st == cext.SZOMB except OSError: return False def wrap_exceptions(fun): """Decorator which translates bare OSError exceptions into NoSuchProcess and AccessDenied. """ @functools.wraps(fun) def wrapper(self, *args, **kwargs): pid, ppid, name = self.pid, self._ppid, self._name try: return fun(self, *args, **kwargs) except ProcessLookupError as err: if is_zombie(pid): raise ZombieProcess(pid, name, ppid) from err raise NoSuchProcess(pid, name) from err except PermissionError as err: raise AccessDenied(pid, name) from err return wrapper class Process: """Wrapper class around underlying C implementation.""" __slots__ = ["_cache", "_name", "_ppid", "pid"] def __init__(self, pid): self.pid = pid self._name = None self._ppid = None @wrap_exceptions @memoize_when_activated def _get_kinfo_proc(self): # Note: should work with all PIDs without permission issues. ret = cext.proc_kinfo_oneshot(self.pid) assert len(ret) == len(kinfo_proc_map) return ret @wrap_exceptions @memoize_when_activated def _get_pidtaskinfo(self): # Note: should work for PIDs owned by user only. ret = cext.proc_pidtaskinfo_oneshot(self.pid) assert len(ret) == len(pidtaskinfo_map) return ret def oneshot_enter(self): self._get_kinfo_proc.cache_activate(self) self._get_pidtaskinfo.cache_activate(self) def oneshot_exit(self): self._get_kinfo_proc.cache_deactivate(self) self._get_pidtaskinfo.cache_deactivate(self) @wrap_exceptions def name(self): name = self._get_kinfo_proc()[kinfo_proc_map['name']] return name if name is not None else cext.proc_name(self.pid) @wrap_exceptions def exe(self): return cext.proc_exe(self.pid) @wrap_exceptions def cmdline(self): return cext.proc_cmdline(self.pid) @wrap_exceptions def environ(self): return parse_environ_block(cext.proc_environ(self.pid)) @wrap_exceptions def ppid(self): self._ppid = self._get_kinfo_proc()[kinfo_proc_map['ppid']] return self._ppid @wrap_exceptions def cwd(self): return cext.proc_cwd(self.pid) @wrap_exceptions def uids(self): rawtuple = self._get_kinfo_proc() return _common.puids( rawtuple[kinfo_proc_map['ruid']], rawtuple[kinfo_proc_map['euid']], rawtuple[kinfo_proc_map['suid']], ) @wrap_exceptions def gids(self): rawtuple = self._get_kinfo_proc() return _common.puids( rawtuple[kinfo_proc_map['rgid']], rawtuple[kinfo_proc_map['egid']], rawtuple[kinfo_proc_map['sgid']], ) @wrap_exceptions def terminal(self): tty_nr = self._get_kinfo_proc()[kinfo_proc_map['ttynr']] tmap = _psposix.get_terminal_map() try: return tmap[tty_nr] except KeyError: return None @wrap_exceptions def memory_info(self): rawtuple = self._get_pidtaskinfo() return pmem( rawtuple[pidtaskinfo_map['rss']], rawtuple[pidtaskinfo_map['vms']], rawtuple[pidtaskinfo_map['pfaults']], rawtuple[pidtaskinfo_map['pageins']], ) @wrap_exceptions def memory_full_info(self): basic_mem = self.memory_info() uss = cext.proc_memory_uss(self.pid) return pfullmem(*basic_mem + (uss,)) @wrap_exceptions def cpu_times(self): rawtuple = self._get_pidtaskinfo() return _common.pcputimes( rawtuple[pidtaskinfo_map['cpuutime']], rawtuple[pidtaskinfo_map['cpustime']], # children user / system times are not retrievable (set to 0) 0.0, 0.0, ) @wrap_exceptions def create_time(self): return self._get_kinfo_proc()[kinfo_proc_map['ctime']] @wrap_exceptions def num_ctx_switches(self): # Unvoluntary value seems not to be available; # getrusage() numbers seems to confirm this theory. # We set it to 0. vol = self._get_pidtaskinfo()[pidtaskinfo_map['volctxsw']] return _common.pctxsw(vol, 0) @wrap_exceptions def num_threads(self): return self._get_pidtaskinfo()[pidtaskinfo_map['numthreads']] @wrap_exceptions def open_files(self): if self.pid == 0: return [] files = [] rawlist = cext.proc_open_files(self.pid) for path, fd in rawlist: if isfile_strict(path): ntuple = _common.popenfile(path, fd) files.append(ntuple) return files @wrap_exceptions def net_connections(self, kind='inet'): families, types = conn_tmap[kind] rawlist = cext.proc_net_connections(self.pid, families, types) ret = [] for item in rawlist: fd, fam, type, laddr, raddr, status = item nt = conn_to_ntuple( fd, fam, type, laddr, raddr, status, TCP_STATUSES ) ret.append(nt) return ret @wrap_exceptions def num_fds(self): if self.pid == 0: return 0 return cext.proc_num_fds(self.pid) @wrap_exceptions def wait(self, timeout=None): return _psposix.wait_pid(self.pid, timeout, self._name) @wrap_exceptions def nice_get(self): return cext_posix.getpriority(self.pid) @wrap_exceptions def nice_set(self, value): return cext_posix.setpriority(self.pid, value) @wrap_exceptions def status(self): code = self._get_kinfo_proc()[kinfo_proc_map['status']] # XXX is '?' legit? (we're not supposed to return it anyway) return PROC_STATUSES.get(code, '?') @wrap_exceptions def threads(self): rawlist = cext.proc_threads(self.pid) retlist = [] for thread_id, utime, stime in rawlist: ntuple = _common.pthread(thread_id, utime, stime) retlist.append(ntuple) return retlist _psbsd.py 0000644 00000076014 15030211765 0006400 0 ustar 00 # Copyright (c) 2009, Giampaolo Rodola'. All rights reserved. # Use of this source code is governed by a BSD-style license that can be # found in the LICENSE file. """FreeBSD, OpenBSD and NetBSD platforms implementation.""" import contextlib import errno import functools import os from collections import defaultdict from collections import namedtuple from xml.etree import ElementTree # noqa: ICN001 from . import _common from . import _psposix from . import _psutil_bsd as cext from . import _psutil_posix as cext_posix from ._common import FREEBSD from ._common import NETBSD from ._common import OPENBSD from ._common import AccessDenied from ._common import NoSuchProcess from ._common import ZombieProcess from ._common import conn_tmap from ._common import conn_to_ntuple from ._common import debug from ._common import memoize from ._common import memoize_when_activated from ._common import usage_percent __extra__all__ = [] # ===================================================================== # --- globals # ===================================================================== if FREEBSD: PROC_STATUSES = { cext.SIDL: _common.STATUS_IDLE, cext.SRUN: _common.STATUS_RUNNING, cext.SSLEEP: _common.STATUS_SLEEPING, cext.SSTOP: _common.STATUS_STOPPED, cext.SZOMB: _common.STATUS_ZOMBIE, cext.SWAIT: _common.STATUS_WAITING, cext.SLOCK: _common.STATUS_LOCKED, } elif OPENBSD: PROC_STATUSES = { cext.SIDL: _common.STATUS_IDLE, cext.SSLEEP: _common.STATUS_SLEEPING, cext.SSTOP: _common.STATUS_STOPPED, # According to /usr/include/sys/proc.h SZOMB is unused. # test_zombie_process() shows that SDEAD is the right # equivalent. Also it appears there's no equivalent of # psutil.STATUS_DEAD. SDEAD really means STATUS_ZOMBIE. # cext.SZOMB: _common.STATUS_ZOMBIE, cext.SDEAD: _common.STATUS_ZOMBIE, cext.SZOMB: _common.STATUS_ZOMBIE, # From http://www.eecs.harvard.edu/~margo/cs161/videos/proc.h.txt # OpenBSD has SRUN and SONPROC: SRUN indicates that a process # is runnable but *not* yet running, i.e. is on a run queue. # SONPROC indicates that the process is actually executing on # a CPU, i.e. it is no longer on a run queue. # As such we'll map SRUN to STATUS_WAKING and SONPROC to # STATUS_RUNNING cext.SRUN: _common.STATUS_WAKING, cext.SONPROC: _common.STATUS_RUNNING, } elif NETBSD: PROC_STATUSES = { cext.SIDL: _common.STATUS_IDLE, cext.SSLEEP: _common.STATUS_SLEEPING, cext.SSTOP: _common.STATUS_STOPPED, cext.SZOMB: _common.STATUS_ZOMBIE, cext.SRUN: _common.STATUS_WAKING, cext.SONPROC: _common.STATUS_RUNNING, } TCP_STATUSES = { cext.TCPS_ESTABLISHED: _common.CONN_ESTABLISHED, cext.TCPS_SYN_SENT: _common.CONN_SYN_SENT, cext.TCPS_SYN_RECEIVED: _common.CONN_SYN_RECV, cext.TCPS_FIN_WAIT_1: _common.CONN_FIN_WAIT1, cext.TCPS_FIN_WAIT_2: _common.CONN_FIN_WAIT2, cext.TCPS_TIME_WAIT: _common.CONN_TIME_WAIT, cext.TCPS_CLOSED: _common.CONN_CLOSE, cext.TCPS_CLOSE_WAIT: _common.CONN_CLOSE_WAIT, cext.TCPS_LAST_ACK: _common.CONN_LAST_ACK, cext.TCPS_LISTEN: _common.CONN_LISTEN, cext.TCPS_CLOSING: _common.CONN_CLOSING, cext.PSUTIL_CONN_NONE: _common.CONN_NONE, } PAGESIZE = cext_posix.getpagesize() AF_LINK = cext_posix.AF_LINK HAS_PER_CPU_TIMES = hasattr(cext, "per_cpu_times") HAS_PROC_NUM_THREADS = hasattr(cext, "proc_num_threads") HAS_PROC_OPEN_FILES = hasattr(cext, 'proc_open_files') HAS_PROC_NUM_FDS = hasattr(cext, 'proc_num_fds') kinfo_proc_map = dict( ppid=0, status=1, real_uid=2, effective_uid=3, saved_uid=4, real_gid=5, effective_gid=6, saved_gid=7, ttynr=8, create_time=9, ctx_switches_vol=10, ctx_switches_unvol=11, read_io_count=12, write_io_count=13, user_time=14, sys_time=15, ch_user_time=16, ch_sys_time=17, rss=18, vms=19, memtext=20, memdata=21, memstack=22, cpunum=23, name=24, ) # ===================================================================== # --- named tuples # ===================================================================== # fmt: off # psutil.virtual_memory() svmem = namedtuple( 'svmem', ['total', 'available', 'percent', 'used', 'free', 'active', 'inactive', 'buffers', 'cached', 'shared', 'wired']) # psutil.cpu_times() scputimes = namedtuple( 'scputimes', ['user', 'nice', 'system', 'idle', 'irq']) # psutil.Process.memory_info() pmem = namedtuple('pmem', ['rss', 'vms', 'text', 'data', 'stack']) # psutil.Process.memory_full_info() pfullmem = pmem # psutil.Process.cpu_times() pcputimes = namedtuple('pcputimes', ['user', 'system', 'children_user', 'children_system']) # psutil.Process.memory_maps(grouped=True) pmmap_grouped = namedtuple( 'pmmap_grouped', 'path rss, private, ref_count, shadow_count') # psutil.Process.memory_maps(grouped=False) pmmap_ext = namedtuple( 'pmmap_ext', 'addr, perms path rss, private, ref_count, shadow_count') # psutil.disk_io_counters() if FREEBSD: sdiskio = namedtuple('sdiskio', ['read_count', 'write_count', 'read_bytes', 'write_bytes', 'read_time', 'write_time', 'busy_time']) else: sdiskio = namedtuple('sdiskio', ['read_count', 'write_count', 'read_bytes', 'write_bytes']) # fmt: on # ===================================================================== # --- memory # ===================================================================== def virtual_memory(): mem = cext.virtual_mem() if NETBSD: total, free, active, inactive, wired, cached = mem # On NetBSD buffers and shared mem is determined via /proc. # The C ext set them to 0. with open('/proc/meminfo', 'rb') as f: for line in f: if line.startswith(b'Buffers:'): buffers = int(line.split()[1]) * 1024 elif line.startswith(b'MemShared:'): shared = int(line.split()[1]) * 1024 # Before avail was calculated as (inactive + cached + free), # same as zabbix, but it turned out it could exceed total (see # #2233), so zabbix seems to be wrong. Htop calculates it # differently, and the used value seem more realistic, so let's # match htop. # https://github.com/htop-dev/htop/blob/e7f447b/netbsd/NetBSDProcessList.c#L162 # https://github.com/zabbix/zabbix/blob/af5e0f8/src/libs/zbxsysinfo/netbsd/memory.c#L135 used = active + wired avail = total - used else: total, free, active, inactive, wired, cached, buffers, shared = mem # matches freebsd-memory CLI: # * https://people.freebsd.org/~rse/dist/freebsd-memory # * https://www.cyberciti.biz/files/scripts/freebsd-memory.pl.txt # matches zabbix: # * https://github.com/zabbix/zabbix/blob/af5e0f8/src/libs/zbxsysinfo/freebsd/memory.c#L143 avail = inactive + cached + free used = active + wired + cached percent = usage_percent((total - avail), total, round_=1) return svmem( total, avail, percent, used, free, active, inactive, buffers, cached, shared, wired, ) def swap_memory(): """System swap memory as (total, used, free, sin, sout) namedtuple.""" total, used, free, sin, sout = cext.swap_mem() percent = usage_percent(used, total, round_=1) return _common.sswap(total, used, free, percent, sin, sout) # ===================================================================== # --- CPU # ===================================================================== def cpu_times(): """Return system per-CPU times as a namedtuple.""" user, nice, system, idle, irq = cext.cpu_times() return scputimes(user, nice, system, idle, irq) if HAS_PER_CPU_TIMES: def per_cpu_times(): """Return system CPU times as a namedtuple.""" ret = [] for cpu_t in cext.per_cpu_times(): user, nice, system, idle, irq = cpu_t item = scputimes(user, nice, system, idle, irq) ret.append(item) return ret else: # XXX # Ok, this is very dirty. # On FreeBSD < 8 we cannot gather per-cpu information, see: # https://github.com/giampaolo/psutil/issues/226 # If num cpus > 1, on first call we return single cpu times to avoid a # crash at psutil import time. # Next calls will fail with NotImplementedError def per_cpu_times(): """Return system CPU times as a namedtuple.""" if cpu_count_logical() == 1: return [cpu_times()] if per_cpu_times.__called__: msg = "supported only starting from FreeBSD 8" raise NotImplementedError(msg) per_cpu_times.__called__ = True return [cpu_times()] per_cpu_times.__called__ = False def cpu_count_logical(): """Return the number of logical CPUs in the system.""" return cext.cpu_count_logical() if OPENBSD or NETBSD: def cpu_count_cores(): # OpenBSD and NetBSD do not implement this. return 1 if cpu_count_logical() == 1 else None else: def cpu_count_cores(): """Return the number of CPU cores in the system.""" # From the C module we'll get an XML string similar to this: # http://manpages.ubuntu.com/manpages/precise/man4/smp.4freebsd.html # We may get None in case "sysctl kern.sched.topology_spec" # is not supported on this BSD version, in which case we'll mimic # os.cpu_count() and return None. ret = None s = cext.cpu_topology() if s is not None: # get rid of padding chars appended at the end of the string index = s.rfind("</groups>") if index != -1: s = s[: index + 9] root = ElementTree.fromstring(s) try: ret = len(root.findall('group/children/group/cpu')) or None finally: # needed otherwise it will memleak root.clear() if not ret: # If logical CPUs == 1 it's obvious we' have only 1 core. if cpu_count_logical() == 1: return 1 return ret def cpu_stats(): """Return various CPU stats as a named tuple.""" if FREEBSD: # Note: the C ext is returning some metrics we are not exposing: # traps. ctxsw, intrs, soft_intrs, syscalls, _traps = cext.cpu_stats() elif NETBSD: # XXX # Note about intrs: the C extension returns 0. intrs # can be determined via /proc/stat; it has the same value as # soft_intrs thought so the kernel is faking it (?). # # Note about syscalls: the C extension always sets it to 0 (?). # # Note: the C ext is returning some metrics we are not exposing: # traps, faults and forks. ctxsw, intrs, soft_intrs, syscalls, _traps, _faults, _forks = ( cext.cpu_stats() ) with open('/proc/stat', 'rb') as f: for line in f: if line.startswith(b'intr'): intrs = int(line.split()[1]) elif OPENBSD: # Note: the C ext is returning some metrics we are not exposing: # traps, faults and forks. ctxsw, intrs, soft_intrs, syscalls, _traps, _faults, _forks = ( cext.cpu_stats() ) return _common.scpustats(ctxsw, intrs, soft_intrs, syscalls) if FREEBSD: def cpu_freq(): """Return frequency metrics for CPUs. As of Dec 2018 only CPU 0 appears to be supported by FreeBSD and all other cores match the frequency of CPU 0. """ ret = [] num_cpus = cpu_count_logical() for cpu in range(num_cpus): try: current, available_freq = cext.cpu_freq(cpu) except NotImplementedError: continue if available_freq: try: min_freq = int(available_freq.split(" ")[-1].split("/")[0]) except (IndexError, ValueError): min_freq = None try: max_freq = int(available_freq.split(" ")[0].split("/")[0]) except (IndexError, ValueError): max_freq = None ret.append(_common.scpufreq(current, min_freq, max_freq)) return ret elif OPENBSD: def cpu_freq(): curr = float(cext.cpu_freq()) return [_common.scpufreq(curr, 0.0, 0.0)] # ===================================================================== # --- disks # ===================================================================== def disk_partitions(all=False): """Return mounted disk partitions as a list of namedtuples. 'all' argument is ignored, see: https://github.com/giampaolo/psutil/issues/906. """ retlist = [] partitions = cext.disk_partitions() for partition in partitions: device, mountpoint, fstype, opts = partition ntuple = _common.sdiskpart(device, mountpoint, fstype, opts) retlist.append(ntuple) return retlist disk_usage = _psposix.disk_usage disk_io_counters = cext.disk_io_counters # ===================================================================== # --- network # ===================================================================== net_io_counters = cext.net_io_counters net_if_addrs = cext_posix.net_if_addrs def net_if_stats(): """Get NIC stats (isup, duplex, speed, mtu).""" names = net_io_counters().keys() ret = {} for name in names: try: mtu = cext_posix.net_if_mtu(name) flags = cext_posix.net_if_flags(name) duplex, speed = cext_posix.net_if_duplex_speed(name) except OSError as err: # https://github.com/giampaolo/psutil/issues/1279 if err.errno != errno.ENODEV: raise else: if hasattr(_common, 'NicDuplex'): duplex = _common.NicDuplex(duplex) output_flags = ','.join(flags) isup = 'running' in flags ret[name] = _common.snicstats( isup, duplex, speed, mtu, output_flags ) return ret def net_connections(kind): """System-wide network connections.""" families, types = conn_tmap[kind] ret = set() if OPENBSD: rawlist = cext.net_connections(-1, families, types) elif NETBSD: rawlist = cext.net_connections(-1, kind) else: # FreeBSD rawlist = cext.net_connections(families, types) for item in rawlist: fd, fam, type, laddr, raddr, status, pid = item nt = conn_to_ntuple( fd, fam, type, laddr, raddr, status, TCP_STATUSES, pid ) ret.add(nt) return list(ret) # ===================================================================== # --- sensors # ===================================================================== if FREEBSD: def sensors_battery(): """Return battery info.""" try: percent, minsleft, power_plugged = cext.sensors_battery() except NotImplementedError: # See: https://github.com/giampaolo/psutil/issues/1074 return None power_plugged = power_plugged == 1 if power_plugged: secsleft = _common.POWER_TIME_UNLIMITED elif minsleft == -1: secsleft = _common.POWER_TIME_UNKNOWN else: secsleft = minsleft * 60 return _common.sbattery(percent, secsleft, power_plugged) def sensors_temperatures(): """Return CPU cores temperatures if available, else an empty dict.""" ret = defaultdict(list) num_cpus = cpu_count_logical() for cpu in range(num_cpus): try: current, high = cext.sensors_cpu_temperature(cpu) if high <= 0: high = None name = f"Core {cpu}" ret["coretemp"].append( _common.shwtemp(name, current, high, high) ) except NotImplementedError: pass return ret # ===================================================================== # --- other system functions # ===================================================================== def boot_time(): """The system boot time expressed in seconds since the epoch.""" return cext.boot_time() def users(): """Return currently connected users as a list of namedtuples.""" retlist = [] rawlist = cext.users() for item in rawlist: user, tty, hostname, tstamp, pid = item if pid == -1: assert OPENBSD pid = None if tty == '~': continue # reboot or shutdown nt = _common.suser(user, tty or None, hostname, tstamp, pid) retlist.append(nt) return retlist # ===================================================================== # --- processes # ===================================================================== @memoize def _pid_0_exists(): try: Process(0).name() except NoSuchProcess: return False except AccessDenied: return True else: return True def pids(): """Returns a list of PIDs currently running on the system.""" ret = cext.pids() if OPENBSD and (0 not in ret) and _pid_0_exists(): # On OpenBSD the kernel does not return PID 0 (neither does # ps) but it's actually querable (Process(0) will succeed). ret.insert(0, 0) return ret if NETBSD: def pid_exists(pid): exists = _psposix.pid_exists(pid) if not exists: # We do this because _psposix.pid_exists() lies in case of # zombie processes. return pid in pids() else: return True elif OPENBSD: def pid_exists(pid): exists = _psposix.pid_exists(pid) if not exists: return False else: # OpenBSD seems to be the only BSD platform where # _psposix.pid_exists() returns True for thread IDs (tids), # so we can't use it. return pid in pids() else: # FreeBSD pid_exists = _psposix.pid_exists def is_zombie(pid): try: st = cext.proc_oneshot_info(pid)[kinfo_proc_map['status']] return PROC_STATUSES.get(st) == _common.STATUS_ZOMBIE except OSError: return False def wrap_exceptions(fun): """Decorator which translates bare OSError exceptions into NoSuchProcess and AccessDenied. """ @functools.wraps(fun) def wrapper(self, *args, **kwargs): pid, ppid, name = self.pid, self._ppid, self._name try: return fun(self, *args, **kwargs) except ProcessLookupError as err: if is_zombie(pid): raise ZombieProcess(pid, name, ppid) from err raise NoSuchProcess(pid, name) from err except PermissionError as err: raise AccessDenied(pid, name) from err except OSError as err: if pid == 0 and 0 in pids(): raise AccessDenied(pid, name) from err raise return wrapper @contextlib.contextmanager def wrap_exceptions_procfs(inst): """Same as above, for routines relying on reading /proc fs.""" pid, name, ppid = inst.pid, inst._name, inst._ppid try: yield except (ProcessLookupError, FileNotFoundError) as err: # ENOENT (no such file or directory) gets raised on open(). # ESRCH (no such process) can get raised on read() if # process is gone in meantime. if is_zombie(inst.pid): raise ZombieProcess(pid, name, ppid) from err else: raise NoSuchProcess(pid, name) from err except PermissionError as err: raise AccessDenied(pid, name) from err class Process: """Wrapper class around underlying C implementation.""" __slots__ = ["_cache", "_name", "_ppid", "pid"] def __init__(self, pid): self.pid = pid self._name = None self._ppid = None def _assert_alive(self): """Raise NSP if the process disappeared on us.""" # For those C function who do not raise NSP, possibly returning # incorrect or incomplete result. cext.proc_name(self.pid) @wrap_exceptions @memoize_when_activated def oneshot(self): """Retrieves multiple process info in one shot as a raw tuple.""" ret = cext.proc_oneshot_info(self.pid) assert len(ret) == len(kinfo_proc_map) return ret def oneshot_enter(self): self.oneshot.cache_activate(self) def oneshot_exit(self): self.oneshot.cache_deactivate(self) @wrap_exceptions def name(self): name = self.oneshot()[kinfo_proc_map['name']] return name if name is not None else cext.proc_name(self.pid) @wrap_exceptions def exe(self): if FREEBSD: if self.pid == 0: return '' # else NSP return cext.proc_exe(self.pid) elif NETBSD: if self.pid == 0: # /proc/0 dir exists but /proc/0/exe doesn't return "" with wrap_exceptions_procfs(self): return os.readlink(f"/proc/{self.pid}/exe") else: # OpenBSD: exe cannot be determined; references: # https://chromium.googlesource.com/chromium/src/base/+/ # master/base_paths_posix.cc # We try our best guess by using which against the first # cmdline arg (may return None). import shutil cmdline = self.cmdline() if cmdline: return shutil.which(cmdline[0]) or "" else: return "" @wrap_exceptions def cmdline(self): if OPENBSD and self.pid == 0: return [] # ...else it crashes elif NETBSD: # XXX - most of the times the underlying sysctl() call on # NetBSD and OpenBSD returns a truncated string. Also # /proc/pid/cmdline behaves the same so it looks like this # is a kernel bug. try: return cext.proc_cmdline(self.pid) except OSError as err: if err.errno == errno.EINVAL: pid, name, ppid = self.pid, self._name, self._ppid if is_zombie(self.pid): raise ZombieProcess(pid, name, ppid) from err if not pid_exists(self.pid): raise NoSuchProcess(pid, name, ppid) from err # XXX: this happens with unicode tests. It means the C # routine is unable to decode invalid unicode chars. debug(f"ignoring {err!r} and returning an empty list") return [] else: raise else: return cext.proc_cmdline(self.pid) @wrap_exceptions def environ(self): return cext.proc_environ(self.pid) @wrap_exceptions def terminal(self): tty_nr = self.oneshot()[kinfo_proc_map['ttynr']] tmap = _psposix.get_terminal_map() try: return tmap[tty_nr] except KeyError: return None @wrap_exceptions def ppid(self): self._ppid = self.oneshot()[kinfo_proc_map['ppid']] return self._ppid @wrap_exceptions def uids(self): rawtuple = self.oneshot() return _common.puids( rawtuple[kinfo_proc_map['real_uid']], rawtuple[kinfo_proc_map['effective_uid']], rawtuple[kinfo_proc_map['saved_uid']], ) @wrap_exceptions def gids(self): rawtuple = self.oneshot() return _common.pgids( rawtuple[kinfo_proc_map['real_gid']], rawtuple[kinfo_proc_map['effective_gid']], rawtuple[kinfo_proc_map['saved_gid']], ) @wrap_exceptions def cpu_times(self): rawtuple = self.oneshot() return _common.pcputimes( rawtuple[kinfo_proc_map['user_time']], rawtuple[kinfo_proc_map['sys_time']], rawtuple[kinfo_proc_map['ch_user_time']], rawtuple[kinfo_proc_map['ch_sys_time']], ) if FREEBSD: @wrap_exceptions def cpu_num(self): return self.oneshot()[kinfo_proc_map['cpunum']] @wrap_exceptions def memory_info(self): rawtuple = self.oneshot() return pmem( rawtuple[kinfo_proc_map['rss']], rawtuple[kinfo_proc_map['vms']], rawtuple[kinfo_proc_map['memtext']], rawtuple[kinfo_proc_map['memdata']], rawtuple[kinfo_proc_map['memstack']], ) memory_full_info = memory_info @wrap_exceptions def create_time(self): return self.oneshot()[kinfo_proc_map['create_time']] @wrap_exceptions def num_threads(self): if HAS_PROC_NUM_THREADS: # FreeBSD return cext.proc_num_threads(self.pid) else: return len(self.threads()) @wrap_exceptions def num_ctx_switches(self): rawtuple = self.oneshot() return _common.pctxsw( rawtuple[kinfo_proc_map['ctx_switches_vol']], rawtuple[kinfo_proc_map['ctx_switches_unvol']], ) @wrap_exceptions def threads(self): # Note: on OpenSBD this (/dev/mem) requires root access. rawlist = cext.proc_threads(self.pid) retlist = [] for thread_id, utime, stime in rawlist: ntuple = _common.pthread(thread_id, utime, stime) retlist.append(ntuple) if OPENBSD: self._assert_alive() return retlist @wrap_exceptions def net_connections(self, kind='inet'): families, types = conn_tmap[kind] ret = [] if NETBSD: rawlist = cext.net_connections(self.pid, kind) elif OPENBSD: rawlist = cext.net_connections(self.pid, families, types) else: rawlist = cext.proc_net_connections(self.pid, families, types) for item in rawlist: fd, fam, type, laddr, raddr, status = item[:6] if FREEBSD: if (fam not in families) or (type not in types): continue nt = conn_to_ntuple( fd, fam, type, laddr, raddr, status, TCP_STATUSES ) ret.append(nt) self._assert_alive() return ret @wrap_exceptions def wait(self, timeout=None): return _psposix.wait_pid(self.pid, timeout, self._name) @wrap_exceptions def nice_get(self): return cext_posix.getpriority(self.pid) @wrap_exceptions def nice_set(self, value): return cext_posix.setpriority(self.pid, value) @wrap_exceptions def status(self): code = self.oneshot()[kinfo_proc_map['status']] # XXX is '?' legit? (we're not supposed to return it anyway) return PROC_STATUSES.get(code, '?') @wrap_exceptions def io_counters(self): rawtuple = self.oneshot() return _common.pio( rawtuple[kinfo_proc_map['read_io_count']], rawtuple[kinfo_proc_map['write_io_count']], -1, -1, ) @wrap_exceptions def cwd(self): """Return process current working directory.""" # sometimes we get an empty string, in which case we turn # it into None if OPENBSD and self.pid == 0: return "" # ...else it would raise EINVAL elif NETBSD or HAS_PROC_OPEN_FILES: # FreeBSD < 8 does not support functions based on # kinfo_getfile() and kinfo_getvmmap() return cext.proc_cwd(self.pid) else: raise NotImplementedError( "supported only starting from FreeBSD 8" if FREEBSD else "" ) nt_mmap_grouped = namedtuple( 'mmap', 'path rss, private, ref_count, shadow_count' ) nt_mmap_ext = namedtuple( 'mmap', 'addr, perms path rss, private, ref_count, shadow_count' ) def _not_implemented(self): raise NotImplementedError # FreeBSD < 8 does not support functions based on kinfo_getfile() # and kinfo_getvmmap() if HAS_PROC_OPEN_FILES: @wrap_exceptions def open_files(self): """Return files opened by process as a list of namedtuples.""" rawlist = cext.proc_open_files(self.pid) return [_common.popenfile(path, fd) for path, fd in rawlist] else: open_files = _not_implemented # FreeBSD < 8 does not support functions based on kinfo_getfile() # and kinfo_getvmmap() if HAS_PROC_NUM_FDS: @wrap_exceptions def num_fds(self): """Return the number of file descriptors opened by this process.""" ret = cext.proc_num_fds(self.pid) if NETBSD: self._assert_alive() return ret else: num_fds = _not_implemented # --- FreeBSD only APIs if FREEBSD: @wrap_exceptions def cpu_affinity_get(self): return cext.proc_cpu_affinity_get(self.pid) @wrap_exceptions def cpu_affinity_set(self, cpus): # Pre-emptively check if CPUs are valid because the C # function has a weird behavior in case of invalid CPUs, # see: https://github.com/giampaolo/psutil/issues/586 allcpus = set(range(len(per_cpu_times()))) for cpu in cpus: if cpu not in allcpus: msg = f"invalid CPU {cpu!r} (choose between {allcpus})" raise ValueError(msg) try: cext.proc_cpu_affinity_set(self.pid, cpus) except OSError as err: # 'man cpuset_setaffinity' about EDEADLK: # <<the call would leave a thread without a valid CPU to run # on because the set does not overlap with the thread's # anonymous mask>> if err.errno in {errno.EINVAL, errno.EDEADLK}: for cpu in cpus: if cpu not in allcpus: msg = ( f"invalid CPU {cpu!r} (choose between" f" {allcpus})" ) raise ValueError(msg) from err raise @wrap_exceptions def memory_maps(self): return cext.proc_memory_maps(self.pid) @wrap_exceptions def rlimit(self, resource, limits=None): if limits is None: return cext.proc_getrlimit(self.pid, resource) else: if len(limits) != 2: msg = ( "second argument must be a (soft, hard) tuple, got" f" {limits!r}" ) raise ValueError(msg) soft, hard = limits return cext.proc_setrlimit(self.pid, resource, soft, hard) _pssunos.py 0000644 00000060530 15030211765 0006773 0 ustar 00 # Copyright (c) 2009, Giampaolo Rodola'. All rights reserved. # Use of this source code is governed by a BSD-style license that can be # found in the LICENSE file. """Sun OS Solaris platform implementation.""" import errno import functools import os import socket import subprocess import sys from collections import namedtuple from socket import AF_INET from . import _common from . import _psposix from . import _psutil_posix as cext_posix from . import _psutil_sunos as cext from ._common import AF_INET6 from ._common import ENCODING from ._common import AccessDenied from ._common import NoSuchProcess from ._common import ZombieProcess from ._common import debug from ._common import get_procfs_path from ._common import isfile_strict from ._common import memoize_when_activated from ._common import sockfam_to_enum from ._common import socktype_to_enum from ._common import usage_percent __extra__all__ = ["CONN_IDLE", "CONN_BOUND", "PROCFS_PATH"] # ===================================================================== # --- globals # ===================================================================== PAGE_SIZE = cext_posix.getpagesize() AF_LINK = cext_posix.AF_LINK IS_64_BIT = sys.maxsize > 2**32 CONN_IDLE = "IDLE" CONN_BOUND = "BOUND" PROC_STATUSES = { cext.SSLEEP: _common.STATUS_SLEEPING, cext.SRUN: _common.STATUS_RUNNING, cext.SZOMB: _common.STATUS_ZOMBIE, cext.SSTOP: _common.STATUS_STOPPED, cext.SIDL: _common.STATUS_IDLE, cext.SONPROC: _common.STATUS_RUNNING, # same as run cext.SWAIT: _common.STATUS_WAITING, } TCP_STATUSES = { cext.TCPS_ESTABLISHED: _common.CONN_ESTABLISHED, cext.TCPS_SYN_SENT: _common.CONN_SYN_SENT, cext.TCPS_SYN_RCVD: _common.CONN_SYN_RECV, cext.TCPS_FIN_WAIT_1: _common.CONN_FIN_WAIT1, cext.TCPS_FIN_WAIT_2: _common.CONN_FIN_WAIT2, cext.TCPS_TIME_WAIT: _common.CONN_TIME_WAIT, cext.TCPS_CLOSED: _common.CONN_CLOSE, cext.TCPS_CLOSE_WAIT: _common.CONN_CLOSE_WAIT, cext.TCPS_LAST_ACK: _common.CONN_LAST_ACK, cext.TCPS_LISTEN: _common.CONN_LISTEN, cext.TCPS_CLOSING: _common.CONN_CLOSING, cext.PSUTIL_CONN_NONE: _common.CONN_NONE, cext.TCPS_IDLE: CONN_IDLE, # sunos specific cext.TCPS_BOUND: CONN_BOUND, # sunos specific } proc_info_map = dict( ppid=0, rss=1, vms=2, create_time=3, nice=4, num_threads=5, status=6, ttynr=7, uid=8, euid=9, gid=10, egid=11, ) # ===================================================================== # --- named tuples # ===================================================================== # psutil.cpu_times() scputimes = namedtuple('scputimes', ['user', 'system', 'idle', 'iowait']) # psutil.cpu_times(percpu=True) pcputimes = namedtuple( 'pcputimes', ['user', 'system', 'children_user', 'children_system'] ) # psutil.virtual_memory() svmem = namedtuple('svmem', ['total', 'available', 'percent', 'used', 'free']) # psutil.Process.memory_info() pmem = namedtuple('pmem', ['rss', 'vms']) pfullmem = pmem # psutil.Process.memory_maps(grouped=True) pmmap_grouped = namedtuple( 'pmmap_grouped', ['path', 'rss', 'anonymous', 'locked'] ) # psutil.Process.memory_maps(grouped=False) pmmap_ext = namedtuple( 'pmmap_ext', 'addr perms ' + ' '.join(pmmap_grouped._fields) ) # ===================================================================== # --- memory # ===================================================================== def virtual_memory(): """Report virtual memory metrics.""" # we could have done this with kstat, but IMHO this is good enough total = os.sysconf('SC_PHYS_PAGES') * PAGE_SIZE # note: there's no difference on Solaris free = avail = os.sysconf('SC_AVPHYS_PAGES') * PAGE_SIZE used = total - free percent = usage_percent(used, total, round_=1) return svmem(total, avail, percent, used, free) def swap_memory(): """Report swap memory metrics.""" sin, sout = cext.swap_mem() # XXX # we are supposed to get total/free by doing so: # http://cvs.opensolaris.org/source/xref/onnv/onnv-gate/ # usr/src/cmd/swap/swap.c # ...nevertheless I can't manage to obtain the same numbers as 'swap' # cmdline utility, so let's parse its output (sigh!) p = subprocess.Popen( [ '/usr/bin/env', f"PATH=/usr/sbin:/sbin:{os.environ['PATH']}", 'swap', '-l', ], stdout=subprocess.PIPE, ) stdout, _ = p.communicate() stdout = stdout.decode(sys.stdout.encoding) if p.returncode != 0: msg = f"'swap -l' failed (retcode={p.returncode})" raise RuntimeError(msg) lines = stdout.strip().split('\n')[1:] if not lines: msg = 'no swap device(s) configured' raise RuntimeError(msg) total = free = 0 for line in lines: line = line.split() t, f = line[3:5] total += int(int(t) * 512) free += int(int(f) * 512) used = total - free percent = usage_percent(used, total, round_=1) return _common.sswap( total, used, free, percent, sin * PAGE_SIZE, sout * PAGE_SIZE ) # ===================================================================== # --- CPU # ===================================================================== def cpu_times(): """Return system-wide CPU times as a named tuple.""" ret = cext.per_cpu_times() return scputimes(*[sum(x) for x in zip(*ret)]) def per_cpu_times(): """Return system per-CPU times as a list of named tuples.""" ret = cext.per_cpu_times() return [scputimes(*x) for x in ret] def cpu_count_logical(): """Return the number of logical CPUs in the system.""" try: return os.sysconf("SC_NPROCESSORS_ONLN") except ValueError: # mimic os.cpu_count() behavior return None def cpu_count_cores(): """Return the number of CPU cores in the system.""" return cext.cpu_count_cores() def cpu_stats(): """Return various CPU stats as a named tuple.""" ctx_switches, interrupts, syscalls, _traps = cext.cpu_stats() soft_interrupts = 0 return _common.scpustats( ctx_switches, interrupts, soft_interrupts, syscalls ) # ===================================================================== # --- disks # ===================================================================== disk_io_counters = cext.disk_io_counters disk_usage = _psposix.disk_usage def disk_partitions(all=False): """Return system disk partitions.""" # TODO - the filtering logic should be better checked so that # it tries to reflect 'df' as much as possible retlist = [] partitions = cext.disk_partitions() for partition in partitions: device, mountpoint, fstype, opts = partition if device == 'none': device = '' if not all: # Differently from, say, Linux, we don't have a list of # common fs types so the best we can do, AFAIK, is to # filter by filesystem having a total size > 0. try: if not disk_usage(mountpoint).total: continue except OSError as err: # https://github.com/giampaolo/psutil/issues/1674 debug(f"skipping {mountpoint!r}: {err}") continue ntuple = _common.sdiskpart(device, mountpoint, fstype, opts) retlist.append(ntuple) return retlist # ===================================================================== # --- network # ===================================================================== net_io_counters = cext.net_io_counters net_if_addrs = cext_posix.net_if_addrs def net_connections(kind, _pid=-1): """Return socket connections. If pid == -1 return system-wide connections (as opposed to connections opened by one process only). Only INET sockets are returned (UNIX are not). """ families, types = _common.conn_tmap[kind] rawlist = cext.net_connections(_pid) ret = set() for item in rawlist: fd, fam, type_, laddr, raddr, status, pid = item if fam not in families: continue if type_ not in types: continue # TODO: refactor and use _common.conn_to_ntuple. if fam in {AF_INET, AF_INET6}: if laddr: laddr = _common.addr(*laddr) if raddr: raddr = _common.addr(*raddr) status = TCP_STATUSES[status] fam = sockfam_to_enum(fam) type_ = socktype_to_enum(type_) if _pid == -1: nt = _common.sconn(fd, fam, type_, laddr, raddr, status, pid) else: nt = _common.pconn(fd, fam, type_, laddr, raddr, status) ret.add(nt) return list(ret) def net_if_stats(): """Get NIC stats (isup, duplex, speed, mtu).""" ret = cext.net_if_stats() for name, items in ret.items(): isup, duplex, speed, mtu = items if hasattr(_common, 'NicDuplex'): duplex = _common.NicDuplex(duplex) ret[name] = _common.snicstats(isup, duplex, speed, mtu, '') return ret # ===================================================================== # --- other system functions # ===================================================================== def boot_time(): """The system boot time expressed in seconds since the epoch.""" return cext.boot_time() def users(): """Return currently connected users as a list of namedtuples.""" retlist = [] rawlist = cext.users() localhost = (':0.0', ':0') for item in rawlist: user, tty, hostname, tstamp, user_process, pid = item # note: the underlying C function includes entries about # system boot, run level and others. We might want # to use them in the future. if not user_process: continue if hostname in localhost: hostname = 'localhost' nt = _common.suser(user, tty, hostname, tstamp, pid) retlist.append(nt) return retlist # ===================================================================== # --- processes # ===================================================================== def pids(): """Returns a list of PIDs currently running on the system.""" path = get_procfs_path().encode(ENCODING) return [int(x) for x in os.listdir(path) if x.isdigit()] def pid_exists(pid): """Check for the existence of a unix pid.""" return _psposix.pid_exists(pid) def wrap_exceptions(fun): """Call callable into a try/except clause and translate ENOENT, EACCES and EPERM in NoSuchProcess or AccessDenied exceptions. """ @functools.wraps(fun) def wrapper(self, *args, **kwargs): pid, ppid, name = self.pid, self._ppid, self._name try: return fun(self, *args, **kwargs) except (FileNotFoundError, ProcessLookupError) as err: # ENOENT (no such file or directory) gets raised on open(). # ESRCH (no such process) can get raised on read() if # process is gone in meantime. if not pid_exists(pid): raise NoSuchProcess(pid, name) from err raise ZombieProcess(pid, name, ppid) from err except PermissionError as err: raise AccessDenied(pid, name) from err except OSError as err: if pid == 0: if 0 in pids(): raise AccessDenied(pid, name) from err raise raise return wrapper class Process: """Wrapper class around underlying C implementation.""" __slots__ = ["_cache", "_name", "_ppid", "_procfs_path", "pid"] def __init__(self, pid): self.pid = pid self._name = None self._ppid = None self._procfs_path = get_procfs_path() def _assert_alive(self): """Raise NSP if the process disappeared on us.""" # For those C function who do not raise NSP, possibly returning # incorrect or incomplete result. os.stat(f"{self._procfs_path}/{self.pid}") def oneshot_enter(self): self._proc_name_and_args.cache_activate(self) self._proc_basic_info.cache_activate(self) self._proc_cred.cache_activate(self) def oneshot_exit(self): self._proc_name_and_args.cache_deactivate(self) self._proc_basic_info.cache_deactivate(self) self._proc_cred.cache_deactivate(self) @wrap_exceptions @memoize_when_activated def _proc_name_and_args(self): return cext.proc_name_and_args(self.pid, self._procfs_path) @wrap_exceptions @memoize_when_activated def _proc_basic_info(self): if self.pid == 0 and not os.path.exists( f"{self._procfs_path}/{self.pid}/psinfo" ): raise AccessDenied(self.pid) ret = cext.proc_basic_info(self.pid, self._procfs_path) assert len(ret) == len(proc_info_map) return ret @wrap_exceptions @memoize_when_activated def _proc_cred(self): return cext.proc_cred(self.pid, self._procfs_path) @wrap_exceptions def name(self): # note: max len == 15 return self._proc_name_and_args()[0] @wrap_exceptions def exe(self): try: return os.readlink(f"{self._procfs_path}/{self.pid}/path/a.out") except OSError: pass # continue and guess the exe name from the cmdline # Will be guessed later from cmdline but we want to explicitly # invoke cmdline here in order to get an AccessDenied # exception if the user has not enough privileges. self.cmdline() return "" @wrap_exceptions def cmdline(self): return self._proc_name_and_args()[1].split(' ') @wrap_exceptions def environ(self): return cext.proc_environ(self.pid, self._procfs_path) @wrap_exceptions def create_time(self): return self._proc_basic_info()[proc_info_map['create_time']] @wrap_exceptions def num_threads(self): return self._proc_basic_info()[proc_info_map['num_threads']] @wrap_exceptions def nice_get(self): # Note #1: getpriority(3) doesn't work for realtime processes. # Psinfo is what ps uses, see: # https://github.com/giampaolo/psutil/issues/1194 return self._proc_basic_info()[proc_info_map['nice']] @wrap_exceptions def nice_set(self, value): if self.pid in {2, 3}: # Special case PIDs: internally setpriority(3) return ESRCH # (no such process), no matter what. # The process actually exists though, as it has a name, # creation time, etc. raise AccessDenied(self.pid, self._name) return cext_posix.setpriority(self.pid, value) @wrap_exceptions def ppid(self): self._ppid = self._proc_basic_info()[proc_info_map['ppid']] return self._ppid @wrap_exceptions def uids(self): try: real, effective, saved, _, _, _ = self._proc_cred() except AccessDenied: real = self._proc_basic_info()[proc_info_map['uid']] effective = self._proc_basic_info()[proc_info_map['euid']] saved = None return _common.puids(real, effective, saved) @wrap_exceptions def gids(self): try: _, _, _, real, effective, saved = self._proc_cred() except AccessDenied: real = self._proc_basic_info()[proc_info_map['gid']] effective = self._proc_basic_info()[proc_info_map['egid']] saved = None return _common.puids(real, effective, saved) @wrap_exceptions def cpu_times(self): try: times = cext.proc_cpu_times(self.pid, self._procfs_path) except OSError as err: if err.errno == errno.EOVERFLOW and not IS_64_BIT: # We may get here if we attempt to query a 64bit process # with a 32bit python. # Error originates from read() and also tools like "cat" # fail in the same way (!). # Since there simply is no way to determine CPU times we # return 0.0 as a fallback. See: # https://github.com/giampaolo/psutil/issues/857 times = (0.0, 0.0, 0.0, 0.0) else: raise return _common.pcputimes(*times) @wrap_exceptions def cpu_num(self): return cext.proc_cpu_num(self.pid, self._procfs_path) @wrap_exceptions def terminal(self): procfs_path = self._procfs_path hit_enoent = False tty = wrap_exceptions(self._proc_basic_info()[proc_info_map['ttynr']]) if tty != cext.PRNODEV: for x in (0, 1, 2, 255): try: return os.readlink(f"{procfs_path}/{self.pid}/path/{x}") except FileNotFoundError: hit_enoent = True continue if hit_enoent: self._assert_alive() @wrap_exceptions def cwd(self): # /proc/PID/path/cwd may not be resolved by readlink() even if # it exists (ls shows it). If that's the case and the process # is still alive return None (we can return None also on BSD). # Reference: https://groups.google.com/g/comp.unix.solaris/c/tcqvhTNFCAs procfs_path = self._procfs_path try: return os.readlink(f"{procfs_path}/{self.pid}/path/cwd") except FileNotFoundError: os.stat(f"{procfs_path}/{self.pid}") # raise NSP or AD return "" @wrap_exceptions def memory_info(self): ret = self._proc_basic_info() rss = ret[proc_info_map['rss']] * 1024 vms = ret[proc_info_map['vms']] * 1024 return pmem(rss, vms) memory_full_info = memory_info @wrap_exceptions def status(self): code = self._proc_basic_info()[proc_info_map['status']] # XXX is '?' legit? (we're not supposed to return it anyway) return PROC_STATUSES.get(code, '?') @wrap_exceptions def threads(self): procfs_path = self._procfs_path ret = [] tids = os.listdir(f"{procfs_path}/{self.pid}/lwp") hit_enoent = False for tid in tids: tid = int(tid) try: utime, stime = cext.query_process_thread( self.pid, tid, procfs_path ) except OSError as err: if err.errno == errno.EOVERFLOW and not IS_64_BIT: # We may get here if we attempt to query a 64bit process # with a 32bit python. # Error originates from read() and also tools like "cat" # fail in the same way (!). # Since there simply is no way to determine CPU times we # return 0.0 as a fallback. See: # https://github.com/giampaolo/psutil/issues/857 continue # ENOENT == thread gone in meantime if err.errno == errno.ENOENT: hit_enoent = True continue raise else: nt = _common.pthread(tid, utime, stime) ret.append(nt) if hit_enoent: self._assert_alive() return ret @wrap_exceptions def open_files(self): retlist = [] hit_enoent = False procfs_path = self._procfs_path pathdir = f"{procfs_path}/{self.pid}/path" for fd in os.listdir(f"{procfs_path}/{self.pid}/fd"): path = os.path.join(pathdir, fd) if os.path.islink(path): try: file = os.readlink(path) except FileNotFoundError: hit_enoent = True continue else: if isfile_strict(file): retlist.append(_common.popenfile(file, int(fd))) if hit_enoent: self._assert_alive() return retlist def _get_unix_sockets(self, pid): """Get UNIX sockets used by process by parsing 'pfiles' output.""" # TODO: rewrite this in C (...but the damn netstat source code # does not include this part! Argh!!) cmd = ["pfiles", str(pid)] p = subprocess.Popen( cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE ) stdout, stderr = p.communicate() stdout, stderr = ( x.decode(sys.stdout.encoding) for x in (stdout, stderr) ) if p.returncode != 0: if 'permission denied' in stderr.lower(): raise AccessDenied(self.pid, self._name) if 'no such process' in stderr.lower(): raise NoSuchProcess(self.pid, self._name) msg = f"{cmd!r} command error\n{stderr}" raise RuntimeError(msg) lines = stdout.split('\n')[2:] for i, line in enumerate(lines): line = line.lstrip() if line.startswith('sockname: AF_UNIX'): path = line.split(' ', 2)[2] type = lines[i - 2].strip() if type == 'SOCK_STREAM': type = socket.SOCK_STREAM elif type == 'SOCK_DGRAM': type = socket.SOCK_DGRAM else: type = -1 yield (-1, socket.AF_UNIX, type, path, "", _common.CONN_NONE) @wrap_exceptions def net_connections(self, kind='inet'): ret = net_connections(kind, _pid=self.pid) # The underlying C implementation retrieves all OS connections # and filters them by PID. At this point we can't tell whether # an empty list means there were no connections for process or # process is no longer active so we force NSP in case the PID # is no longer there. if not ret: # will raise NSP if process is gone os.stat(f"{self._procfs_path}/{self.pid}") # UNIX sockets if kind in {'all', 'unix'}: ret.extend([ _common.pconn(*conn) for conn in self._get_unix_sockets(self.pid) ]) return ret nt_mmap_grouped = namedtuple('mmap', 'path rss anon locked') nt_mmap_ext = namedtuple('mmap', 'addr perms path rss anon locked') @wrap_exceptions def memory_maps(self): def toaddr(start, end): return "{}-{}".format( hex(start)[2:].strip('L'), hex(end)[2:].strip('L') ) procfs_path = self._procfs_path retlist = [] try: rawlist = cext.proc_memory_maps(self.pid, procfs_path) except OSError as err: if err.errno == errno.EOVERFLOW and not IS_64_BIT: # We may get here if we attempt to query a 64bit process # with a 32bit python. # Error originates from read() and also tools like "cat" # fail in the same way (!). # Since there simply is no way to determine CPU times we # return 0.0 as a fallback. See: # https://github.com/giampaolo/psutil/issues/857 return [] else: raise hit_enoent = False for item in rawlist: addr, addrsize, perm, name, rss, anon, locked = item addr = toaddr(addr, addrsize) if not name.startswith('['): try: name = os.readlink(f"{procfs_path}/{self.pid}/path/{name}") except OSError as err: if err.errno == errno.ENOENT: # sometimes the link may not be resolved by # readlink() even if it exists (ls shows it). # If that's the case we just return the # unresolved link path. # This seems an inconsistency with /proc similar # to: http://goo.gl/55XgO name = f"{procfs_path}/{self.pid}/path/{name}" hit_enoent = True else: raise retlist.append((addr, perm, name, rss, anon, locked)) if hit_enoent: self._assert_alive() return retlist @wrap_exceptions def num_fds(self): return len(os.listdir(f"{self._procfs_path}/{self.pid}/fd")) @wrap_exceptions def num_ctx_switches(self): return _common.pctxsw( *cext.proc_num_ctx_switches(self.pid, self._procfs_path) ) @wrap_exceptions def wait(self, timeout=None): return _psposix.wait_pid(self.pid, timeout, self._name) _psaix.py 0000644 00000043514 15030211765 0006410 0 ustar 00 # Copyright (c) 2009, Giampaolo Rodola' # Copyright (c) 2017, Arnon Yaari # All rights reserved. # Use of this source code is governed by a BSD-style license that can be # found in the LICENSE file. """AIX platform implementation.""" import functools import glob import os import re import subprocess import sys from collections import namedtuple from . import _common from . import _psposix from . import _psutil_aix as cext from . import _psutil_posix as cext_posix from ._common import NIC_DUPLEX_FULL from ._common import NIC_DUPLEX_HALF from ._common import NIC_DUPLEX_UNKNOWN from ._common import AccessDenied from ._common import NoSuchProcess from ._common import ZombieProcess from ._common import conn_to_ntuple from ._common import get_procfs_path from ._common import memoize_when_activated from ._common import usage_percent __extra__all__ = ["PROCFS_PATH"] # ===================================================================== # --- globals # ===================================================================== HAS_THREADS = hasattr(cext, "proc_threads") HAS_NET_IO_COUNTERS = hasattr(cext, "net_io_counters") HAS_PROC_IO_COUNTERS = hasattr(cext, "proc_io_counters") PAGE_SIZE = cext_posix.getpagesize() AF_LINK = cext_posix.AF_LINK PROC_STATUSES = { cext.SIDL: _common.STATUS_IDLE, cext.SZOMB: _common.STATUS_ZOMBIE, cext.SACTIVE: _common.STATUS_RUNNING, cext.SSWAP: _common.STATUS_RUNNING, # TODO what status is this? cext.SSTOP: _common.STATUS_STOPPED, } TCP_STATUSES = { cext.TCPS_ESTABLISHED: _common.CONN_ESTABLISHED, cext.TCPS_SYN_SENT: _common.CONN_SYN_SENT, cext.TCPS_SYN_RCVD: _common.CONN_SYN_RECV, cext.TCPS_FIN_WAIT_1: _common.CONN_FIN_WAIT1, cext.TCPS_FIN_WAIT_2: _common.CONN_FIN_WAIT2, cext.TCPS_TIME_WAIT: _common.CONN_TIME_WAIT, cext.TCPS_CLOSED: _common.CONN_CLOSE, cext.TCPS_CLOSE_WAIT: _common.CONN_CLOSE_WAIT, cext.TCPS_LAST_ACK: _common.CONN_LAST_ACK, cext.TCPS_LISTEN: _common.CONN_LISTEN, cext.TCPS_CLOSING: _common.CONN_CLOSING, cext.PSUTIL_CONN_NONE: _common.CONN_NONE, } proc_info_map = dict( ppid=0, rss=1, vms=2, create_time=3, nice=4, num_threads=5, status=6, ttynr=7, ) # ===================================================================== # --- named tuples # ===================================================================== # psutil.Process.memory_info() pmem = namedtuple('pmem', ['rss', 'vms']) # psutil.Process.memory_full_info() pfullmem = pmem # psutil.Process.cpu_times() scputimes = namedtuple('scputimes', ['user', 'system', 'idle', 'iowait']) # psutil.virtual_memory() svmem = namedtuple('svmem', ['total', 'available', 'percent', 'used', 'free']) # ===================================================================== # --- memory # ===================================================================== def virtual_memory(): total, avail, free, _pinned, inuse = cext.virtual_mem() percent = usage_percent((total - avail), total, round_=1) return svmem(total, avail, percent, inuse, free) def swap_memory(): """Swap system memory as a (total, used, free, sin, sout) tuple.""" total, free, sin, sout = cext.swap_mem() used = total - free percent = usage_percent(used, total, round_=1) return _common.sswap(total, used, free, percent, sin, sout) # ===================================================================== # --- CPU # ===================================================================== def cpu_times(): """Return system-wide CPU times as a named tuple.""" ret = cext.per_cpu_times() return scputimes(*[sum(x) for x in zip(*ret)]) def per_cpu_times(): """Return system per-CPU times as a list of named tuples.""" ret = cext.per_cpu_times() return [scputimes(*x) for x in ret] def cpu_count_logical(): """Return the number of logical CPUs in the system.""" try: return os.sysconf("SC_NPROCESSORS_ONLN") except ValueError: # mimic os.cpu_count() behavior return None def cpu_count_cores(): cmd = ["lsdev", "-Cc", "processor"] p = subprocess.Popen(cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE) stdout, stderr = p.communicate() stdout, stderr = (x.decode(sys.stdout.encoding) for x in (stdout, stderr)) if p.returncode != 0: msg = f"{cmd!r} command error\n{stderr}" raise RuntimeError(msg) processors = stdout.strip().splitlines() return len(processors) or None def cpu_stats(): """Return various CPU stats as a named tuple.""" ctx_switches, interrupts, soft_interrupts, syscalls = cext.cpu_stats() return _common.scpustats( ctx_switches, interrupts, soft_interrupts, syscalls ) # ===================================================================== # --- disks # ===================================================================== disk_io_counters = cext.disk_io_counters disk_usage = _psposix.disk_usage def disk_partitions(all=False): """Return system disk partitions.""" # TODO - the filtering logic should be better checked so that # it tries to reflect 'df' as much as possible retlist = [] partitions = cext.disk_partitions() for partition in partitions: device, mountpoint, fstype, opts = partition if device == 'none': device = '' if not all: # Differently from, say, Linux, we don't have a list of # common fs types so the best we can do, AFAIK, is to # filter by filesystem having a total size > 0. if not disk_usage(mountpoint).total: continue ntuple = _common.sdiskpart(device, mountpoint, fstype, opts) retlist.append(ntuple) return retlist # ===================================================================== # --- network # ===================================================================== net_if_addrs = cext_posix.net_if_addrs if HAS_NET_IO_COUNTERS: net_io_counters = cext.net_io_counters def net_connections(kind, _pid=-1): """Return socket connections. If pid == -1 return system-wide connections (as opposed to connections opened by one process only). """ families, types = _common.conn_tmap[kind] rawlist = cext.net_connections(_pid) ret = [] for item in rawlist: fd, fam, type_, laddr, raddr, status, pid = item if fam not in families: continue if type_ not in types: continue nt = conn_to_ntuple( fd, fam, type_, laddr, raddr, status, TCP_STATUSES, pid=pid if _pid == -1 else None, ) ret.append(nt) return ret def net_if_stats(): """Get NIC stats (isup, duplex, speed, mtu).""" duplex_map = {"Full": NIC_DUPLEX_FULL, "Half": NIC_DUPLEX_HALF} names = {x[0] for x in net_if_addrs()} ret = {} for name in names: mtu = cext_posix.net_if_mtu(name) flags = cext_posix.net_if_flags(name) # try to get speed and duplex # TODO: rewrite this in C (entstat forks, so use truss -f to follow. # looks like it is using an undocumented ioctl?) duplex = "" speed = 0 p = subprocess.Popen( ["/usr/bin/entstat", "-d", name], stdout=subprocess.PIPE, stderr=subprocess.PIPE, ) stdout, stderr = p.communicate() stdout, stderr = ( x.decode(sys.stdout.encoding) for x in (stdout, stderr) ) if p.returncode == 0: re_result = re.search( r"Running: (\d+) Mbps.*?(\w+) Duplex", stdout ) if re_result is not None: speed = int(re_result.group(1)) duplex = re_result.group(2) output_flags = ','.join(flags) isup = 'running' in flags duplex = duplex_map.get(duplex, NIC_DUPLEX_UNKNOWN) ret[name] = _common.snicstats(isup, duplex, speed, mtu, output_flags) return ret # ===================================================================== # --- other system functions # ===================================================================== def boot_time(): """The system boot time expressed in seconds since the epoch.""" return cext.boot_time() def users(): """Return currently connected users as a list of namedtuples.""" retlist = [] rawlist = cext.users() localhost = (':0.0', ':0') for item in rawlist: user, tty, hostname, tstamp, user_process, pid = item # note: the underlying C function includes entries about # system boot, run level and others. We might want # to use them in the future. if not user_process: continue if hostname in localhost: hostname = 'localhost' nt = _common.suser(user, tty, hostname, tstamp, pid) retlist.append(nt) return retlist # ===================================================================== # --- processes # ===================================================================== def pids(): """Returns a list of PIDs currently running on the system.""" return [int(x) for x in os.listdir(get_procfs_path()) if x.isdigit()] def pid_exists(pid): """Check for the existence of a unix pid.""" return os.path.exists(os.path.join(get_procfs_path(), str(pid), "psinfo")) def wrap_exceptions(fun): """Call callable into a try/except clause and translate ENOENT, EACCES and EPERM in NoSuchProcess or AccessDenied exceptions. """ @functools.wraps(fun) def wrapper(self, *args, **kwargs): pid, ppid, name = self.pid, self._ppid, self._name try: return fun(self, *args, **kwargs) except (FileNotFoundError, ProcessLookupError) as err: # ENOENT (no such file or directory) gets raised on open(). # ESRCH (no such process) can get raised on read() if # process is gone in meantime. if not pid_exists(pid): raise NoSuchProcess(pid, name) from err raise ZombieProcess(pid, name, ppid) from err except PermissionError as err: raise AccessDenied(pid, name) from err return wrapper class Process: """Wrapper class around underlying C implementation.""" __slots__ = ["_cache", "_name", "_ppid", "_procfs_path", "pid"] def __init__(self, pid): self.pid = pid self._name = None self._ppid = None self._procfs_path = get_procfs_path() def oneshot_enter(self): self._proc_basic_info.cache_activate(self) self._proc_cred.cache_activate(self) def oneshot_exit(self): self._proc_basic_info.cache_deactivate(self) self._proc_cred.cache_deactivate(self) @wrap_exceptions @memoize_when_activated def _proc_basic_info(self): return cext.proc_basic_info(self.pid, self._procfs_path) @wrap_exceptions @memoize_when_activated def _proc_cred(self): return cext.proc_cred(self.pid, self._procfs_path) @wrap_exceptions def name(self): if self.pid == 0: return "swapper" # note: max 16 characters return cext.proc_name(self.pid, self._procfs_path).rstrip("\x00") @wrap_exceptions def exe(self): # there is no way to get executable path in AIX other than to guess, # and guessing is more complex than what's in the wrapping class cmdline = self.cmdline() if not cmdline: return '' exe = cmdline[0] if os.path.sep in exe: # relative or absolute path if not os.path.isabs(exe): # if cwd has changed, we're out of luck - this may be wrong! exe = os.path.abspath(os.path.join(self.cwd(), exe)) if ( os.path.isabs(exe) and os.path.isfile(exe) and os.access(exe, os.X_OK) ): return exe # not found, move to search in PATH using basename only exe = os.path.basename(exe) # search for exe name PATH for path in os.environ["PATH"].split(":"): possible_exe = os.path.abspath(os.path.join(path, exe)) if os.path.isfile(possible_exe) and os.access( possible_exe, os.X_OK ): return possible_exe return '' @wrap_exceptions def cmdline(self): return cext.proc_args(self.pid) @wrap_exceptions def environ(self): return cext.proc_environ(self.pid) @wrap_exceptions def create_time(self): return self._proc_basic_info()[proc_info_map['create_time']] @wrap_exceptions def num_threads(self): return self._proc_basic_info()[proc_info_map['num_threads']] if HAS_THREADS: @wrap_exceptions def threads(self): rawlist = cext.proc_threads(self.pid) retlist = [] for thread_id, utime, stime in rawlist: ntuple = _common.pthread(thread_id, utime, stime) retlist.append(ntuple) # The underlying C implementation retrieves all OS threads # and filters them by PID. At this point we can't tell whether # an empty list means there were no connections for process or # process is no longer active so we force NSP in case the PID # is no longer there. if not retlist: # will raise NSP if process is gone os.stat(f"{self._procfs_path}/{self.pid}") return retlist @wrap_exceptions def net_connections(self, kind='inet'): ret = net_connections(kind, _pid=self.pid) # The underlying C implementation retrieves all OS connections # and filters them by PID. At this point we can't tell whether # an empty list means there were no connections for process or # process is no longer active so we force NSP in case the PID # is no longer there. if not ret: # will raise NSP if process is gone os.stat(f"{self._procfs_path}/{self.pid}") return ret @wrap_exceptions def nice_get(self): return cext_posix.getpriority(self.pid) @wrap_exceptions def nice_set(self, value): return cext_posix.setpriority(self.pid, value) @wrap_exceptions def ppid(self): self._ppid = self._proc_basic_info()[proc_info_map['ppid']] return self._ppid @wrap_exceptions def uids(self): real, effective, saved, _, _, _ = self._proc_cred() return _common.puids(real, effective, saved) @wrap_exceptions def gids(self): _, _, _, real, effective, saved = self._proc_cred() return _common.puids(real, effective, saved) @wrap_exceptions def cpu_times(self): t = cext.proc_cpu_times(self.pid, self._procfs_path) return _common.pcputimes(*t) @wrap_exceptions def terminal(self): ttydev = self._proc_basic_info()[proc_info_map['ttynr']] # convert from 64-bit dev_t to 32-bit dev_t and then map the device ttydev = ((ttydev & 0x0000FFFF00000000) >> 16) | (ttydev & 0xFFFF) # try to match rdev of /dev/pts/* files ttydev for dev in glob.glob("/dev/**/*"): if os.stat(dev).st_rdev == ttydev: return dev return None @wrap_exceptions def cwd(self): procfs_path = self._procfs_path try: result = os.readlink(f"{procfs_path}/{self.pid}/cwd") return result.rstrip('/') except FileNotFoundError: os.stat(f"{procfs_path}/{self.pid}") # raise NSP or AD return "" @wrap_exceptions def memory_info(self): ret = self._proc_basic_info() rss = ret[proc_info_map['rss']] * 1024 vms = ret[proc_info_map['vms']] * 1024 return pmem(rss, vms) memory_full_info = memory_info @wrap_exceptions def status(self): code = self._proc_basic_info()[proc_info_map['status']] # XXX is '?' legit? (we're not supposed to return it anyway) return PROC_STATUSES.get(code, '?') def open_files(self): # TODO rewrite without using procfiles (stat /proc/pid/fd/* and then # find matching name of the inode) p = subprocess.Popen( ["/usr/bin/procfiles", "-n", str(self.pid)], stdout=subprocess.PIPE, stderr=subprocess.PIPE, ) stdout, stderr = p.communicate() stdout, stderr = ( x.decode(sys.stdout.encoding) for x in (stdout, stderr) ) if "no such process" in stderr.lower(): raise NoSuchProcess(self.pid, self._name) procfiles = re.findall(r"(\d+): S_IFREG.*name:(.*)\n", stdout) retlist = [] for fd, path in procfiles: path = path.strip() if path.startswith("//"): path = path[1:] if path.lower() == "cannot be retrieved": continue retlist.append(_common.popenfile(path, int(fd))) return retlist @wrap_exceptions def num_fds(self): if self.pid == 0: # no /proc/0/fd return 0 return len(os.listdir(f"{self._procfs_path}/{self.pid}/fd")) @wrap_exceptions def num_ctx_switches(self): return _common.pctxsw(*cext.proc_num_ctx_switches(self.pid)) @wrap_exceptions def wait(self, timeout=None): return _psposix.wait_pid(self.pid, timeout, self._name) if HAS_PROC_IO_COUNTERS: @wrap_exceptions def io_counters(self): try: rc, wc, rb, wb = cext.proc_io_counters(self.pid) except OSError as err: # if process is terminated, proc_io_counters returns OSError # instead of NSP if not pid_exists(self.pid): raise NoSuchProcess(self.pid, self._name) from err raise return _common.pio(rc, wc, rb, wb) _psutil_posix.abi3.so 0000755 00000213730 15030211765 0010636 0 ustar 00 ELF > �" @ @ 8 @ # " P P ] ] @ @ @ �M �] �] �M �] �] � � 8 8 8 $ $ P�td �B �B �B � � Q�td R�td �M �] �] @ @ GNU ��6Б��C�@ ��;um`� * !�8 � � � * . 4 �ȷԸf���W�~� ���:hde���l�H�����������$���r: � � � � B � � � � M < � � � ( � � � � � � - n � 6 N v � � 8 � R " � ~ �4 W � �# � ] p% W � �% ) c P4 � �4 i a @# � ~ �b % j T7 � P5 � p$ � __gmon_start__ _init _fini _ITM_deregisterTMCloneTable _ITM_registerTMCloneTable __cxa_finalize psutil_PyErr_SetFromOSErrnoWithSyscall __errno_location strerror sprintf PyExc_OSError PyObject_CallFunction PyErr_SetObject _Py_Dealloc NoSuchProcess AccessDenied psutil_check_pid_range PyArg_ParseTuple _Py_NoneStruct PyExc_ValueError PyErr_SetString psutil_set_debug PyObject_IsTrue PSUTIL_DEBUG psutil_setup getenv socket strncpy ioctl close Py_BuildValue PyErr_SetFromErrno _Py_TrueStruct _Py_FalseStruct setpriority getnameinfo PyList_New getifaddrs PyList_Append freeifaddrs getpriority PyUnicode_FromString psutil_getpagesize sysconf psutil_pid_exists kill psutil_raise_for_pid PyExc_RuntimeError PyErr_Format PyInit__psutil_posix PyModule_Create2 PyModule_AddIntConstant PyLong_FromLong PyModule_AddObject libpthread.so.0 libc.so.6 GLIBC_2.2.5 GLIBC_2.3 + ui E ; ii Q ui E �] 0# �] �" �] �] ha VB �a �a �a dB �a `4 �a pB �a �, b |B b ) b �B (b P- @b �B Hb �& `b �B hb & �b �B �b �' �_ �_ �_ �_ �_ �_ �_ �_ �_ 1 �_ & �_ ( ` ` (` 0` 8` @` H` P` X` `` h` p` . x` �` �` �` �` �` �` �` �` �` �` 0 �` �` �` �` * �` + �` a ! a " a # a $ a % (a ' 0a ( 8a ) H��H��? H��t�+ H��� �5�? �%�? @ �%�? h ����%�? h �����%�? h ����%�? h ����%�? h ����%�? h ����%�? h ����%�? h �p����%�? h �`����%�? h �P����%�? h �@����%�? h �0����%�? h � ����%z? h �����%r? h � ����%j? h ���%b? h ����%Z? h �����%R? h ����%J? h ����%B? h ����%:? h ����%2? h ����%*? h �p����%"? h �`����%? h �P����%? h �@����% ? h �0����%? h � ����%�> h �����%�> h � ����%�> h ���%�> h ����%�> h! �����%�> h" ����%�> h# ����%�> h$ ���H�=9@ H�2@ H9�tH�= H��t �� � � � H�= @ H�5@ H)�H��H��?H��H�H�tH�= H��t��f D � � �=�? u/UH�=�< H��t H�=�: �M����h�����? ]� � � � �{���f. � �AUATI��UH�� � ���I��8H�����L��H�5� L��H��1������U 1�L��L�%9< H�5� I�<$���I�<$H��H��� ���H��tH�m tH�� 1�]A\A]� � H������H�� 1�]A\A]ÐATH��H�5D 1�USH�� I��L���>���1�L�� H� �; H�5� H�;�.���H�;H��H�����H��tH�m tH�� 1�[]A\�f D H���H���H�� 1�[]A\�f. � ATH��H�5� 1�USH�� I��L�����1�L�� H� ; H�5n H�;���H�;H��H����H��tH�m tH�� 1�[]A\�f D H�����H�� 1�[]A\�f. � H��H��H�5� 1�H�T$ �����t"�D$ ��x*H��: H� H���f. � 1�H���f � H�a: H�52 H�8�Z���1���f D H��H��H�5� 1�H�T$�6�����t2H�|$������x$H�E: ���҉H�&: H� H��� D 1�H���f � H��H�=B �P���H��t H��9 � 1�H��� � UH��1�H�5� H��@H�T$���A��1�E��tc1Ҿ � �:����Ń��tXL�D$H�t$� L��� �����!� �D$ H��1�������t ����t$ H�=/ 1�����H��@]�f��k���H�$9 H�8���H��@]�f D UH��1�H�5� H��@H�T$���A��1�E��tm1Ҿ � ����Ń��thL�D$H�t$� L���l������ �D$ H��1��������t/�����D$ @tCH�5�8 H�=� 1�� ���H��@]� � ���H�d8 H�8�,���H��@]�f D H�5Q8 H�=� 1�����H��@]� D H��H��H�5 1�H�L$ H�T$����A��1�E��t �T$ �t$1��������tH�8 H� H��� H��7 H�8�����ff. � @ H���� AW��AV��AUATUSH�� ��u\��� � H��E�H��E1�E1�j� H�����H���� H�v7 H� H�� []A\A]A^A_� � ��u�D�w M��t�I��H��N�,7L��L�% D �S H��L��1�H��H������I9�u�K�vL��H�=� �D� 1�����f D H��H�=� 1��g����h���f�H��6 H� � @ AW1�AVAUATUSH��(����H�D$H���� H�|$�c�������f H�l$H���� �, � ��0 H�i6 H� I��I��H��H�uD��M��AUI��L��H�=" 1�����I��XZM���� H�|$L��L�$�t���L�$���� I�*�N I�,$�V H�+�^ I�/�f I�m �n H�m H���w H�}H��t�D�7D�����H;�5 I��t�H���= H�} D�����H��H���E �E�� ���H�}(D�����I��H�l5 H� I��M�������E1�H�|$H��t L�$�^���L�$H�T$H�H�$H��H�� M��t I�*�B M��t I�,$�R H��t H�+�c M��t I�/�t M��t I�m �� H�D$ H�D$H��([]A\A]A^A_� � L���x���I�,$�����L���e���H�+�����H���S���I�/�����L���A���I�m �����L���.���H�m H��������H�l$H���c����s���f D H�}(D���\���I��H�*4 H� I��M���������� D H��L�$���L�$M����������f. � L������� L����M����������f. � L���x�H����������f. � H���X�M����������f. � L���8����� H�|$E1�E1�1�E1�H��� �������f�H�|$E1�E1�E1�H��������H� 3 E1�E1�1�E1�H�8���E1����f�UH��SH����H�T$ H�5� H��� H��1����A��1�E��t!�t$ 1��n�Ƌ��uH�=� 1��8�H��[]ÐH��2 H�8�Q�H��[]�f. � AU1�ATUH��H��@���H��t(I��H�T$1�H��H�5 �?��uI�,$�` E1�H��@L��]A\A]�1Ҿ � ��Ń���T H�t$L�D$� L����� ���D$ H��1������6 �����l$ @���� @���> @���� @���: @���� @�� �� @��@� @�ŀ�R f�� �� f�� �� f�� �! f�� �� f�� �� f�� � f�� @�E f�����H�=� �P�H��H������H��L���i��H�E �W H��H�E �����H�������f�L������� H�=� ���I��H���m���H��L��� ��I�E �� H��I�E ����L���[����f D H�=� ��I��H��� ���H��L�����I�E uuH��I�E �����L�����|���f. � H�=N �T�I��H������H��L���m��I�E u%H��I�E �C���L�����6���f. � H��I�E �����L�����z��� D H�=� �t��d��� � H�=� �\�I�,$� ��E1���H��@L��]A\A]�f. � H�=� ��I��H��� ���H��L�����I�E �a���H��I�E �����L�����|���f D H�=M �D�I��H�������H��L���]��I�E ����H��I�E �C���L�����6���f D H�=2 ���I��H���m���H��L��� ��I�E �����H��I�E �=���L���[��0���L���N����f � H�=� ��I��H��� ���H��L������I�E �a���H��I�E ���L��������f D H�=� �D�I��H�������H��L���]���I�E ����H��I�E �����L�������f D H�= ���I��H���m���H��L��� ���I�E �����H��I�E �����L���[���f D H�=I ��I��H��� ���H��L������I�E �q���H��I�E �����L��� ����f D H�=t �T�I��H������H��L���m���I�E �!���H��I�E �q���L�����d���f D H�=* ��I��H���}���H��L��� ���I�E ����H��I�E �,���L���k�� ���f D H�=� ��I��H���-���H��L�������I�E �����H��I�E ���L��������f D H�=� �d�I��H������H��L���}���I�E �1���H��I�E �����L��������f D H�=j ��I��H�������H��L���-���I�E ���H��I�E �~���L���{��q���H��H�E �M���H���`��@���ff. � � ���f D H���w�H�=� H��H��1����f��� 1��f � H��1����A��� E��t*��� ��t ��tH� + H�8����������1�H���f � UH��SH��H���o�� ��tH��H��[]�� ������uH��H��[]�� � H��* H��H�5* H�8H��1�[]��� � AT� H�=�+ � �H���� � H�5� H��I��������� � H�5� L�������� 1�H�5� L�������� � H�5� L���|����l � H�5� L���`����P � H�5� L���D����4 � H�5� L���(���� � H�5� L��� ����� � H�5� L��������� � H�5p L��������� � H�5a L�������� � H�5R L�������� � H�5F L������ut� H�5: L���h���u\� H�50 L���P���uD� H�5& L���8���u,H�������H��H��tH�5 L������ � E1�L��A\� H��H��� %s (originated from %s) (is) PSUTIL_DEBUG assume no such process (originated from %s) assume access denied (originated from %s) pid must be a positive integer ii %02x: (siOOOO) socket(SOCK_DGRAM) ioctl(SIOCGIFFLAGS) up broadcast debug loopback pointopoint notrailers noarp promisc allmulti master slave multicast portsel automedia dynamic %s syscall failed RLIMIT_AS RLIMIT_CORE RLIMIT_CPU RLIMIT_DATA RLIMIT_FSIZE RLIMIT_MEMLOCK RLIMIT_NOFILE RLIMIT_NPROC RLIMIT_RSS RLIMIT_STACK RLIMIT_LOCKS RLIMIT_MSGQUEUE RLIMIT_NICE RLIMIT_RTPRIO RLIMIT_RTTIME RLIMIT_SIGPENDING RLIM_INFINITY _psutil_posix getpagesize getpriority net_if_addrs net_if_flags net_if_is_running net_if_mtu setpriority ;� `��� ��� �( ��h @�� ��� �� @� ��, ��` @�| P�� �< ��p �� ��� ��� � ��8 zR x � $ ���` FJ w� ?;*3$" < D ���� B�B�D �G�j CBBH O CBB< � ���� B�M�A �G�L CABG O CAB< � @�� B�M�A �G�L CABG O CAB ��j D q K F J $ ��W D F F F D �) Dd ( \ 8�� A�PP{ AC XA 0 � ��� A�PP� AI X AG YA � h�a D H D ` � �� K�D�E �B(�A0�A8�G�\�H�P�Z 8A0A(B BBBI ������� X < h�� B�D�B �B(�A0�A8�D`]hLpXhA`l 8A0A(B BBBH 0 � ��v A�D�D0S AAB SAA@ � �� B�D�A �G`y DBBA � DBBK �� $ �� DP < ��W TB< T �i A�D�G O DAH O DAL XCA � P� B�� 0# �" �] + ; T7 �] �] ���o ` � � [ ` x � � ���o � ���o �o D ���o �] 6 F V f v � � � � � � � � ! ! &! 6! F! V! f! v! �! �! �! �! �! �! �! �! " " &" 6" F" V" f" v" VB ���������a dB `4 pB �, |B ) �B P- �B �& �B & �B �' GCC: (GNU) 10.2.1 20210130 (Red Hat 10.2.1-11) , @# � , p & Q l � ] @# � � � � � � X int W � x �W � �c � � � � � fv W � z bj } \ T� g G � 8 + : W ; W � 5 j; = l� u m@ ) ; � n F � �F � �R ; �R � W g ^ �} �b � 3� � � � ,W � R � R� R � * W R � L R � N3 R R : v R P R � � l W m � � � �� � W � 2 � �W �% ) �� �% � U @ � �R p% W �t } � R b �,R C = � �R �h x � W � � �% � f U�TT &B Q�h �% � � �R % j � } �"R � � b �2R � � pid � � �l % � � U�TT �@ Q�l f% � T �@ P {R p$ � �( {� G C exc |R � � msg } ( ��w � � !m �R � � "� !� �R �$ Uv �$ P � U��wT `@ Q�U �$ 3 T @ Q=R��w �$ Tv #� 9 $- � N jR �# � �L j� G C exc kR � � msg l ( ��w ` � !m qR � � "� !� qR X$ Uv $ P U��wT 0@ Q�U "$ 3 7 T @ Q3R��w 0$ Tv % SR @# � � S4� I C � T ( ��w exc [R � � � !m _R � � "0 !� _R " �# Uv T# � a# m x# P 4 U��wT @ R| �# 3 Z T @ R��w �# Tv 8 A � F & Q o � �9 � � � � � X int c � x �c E �N � �o � �� � � � � R �N � � f� c � z bv G h� } T � N I/ � ; � h � Y � R � � � � � � 4 � � q � � � � � _ � � ! L " � # 1 $ � % � & O ' � ( � ) * , + � , } - � . � / v 0 � 1 � 2 � 3 � 4 0 5 | 6 $ 7 H 8 9 L : p ; � < , < � = � > ? � @ � A � B C 3 D � E � F � G � H ] I � J � K � L / M k N � O � P Q R # S � T U F V M W � X � Y 7 Z � [ � \ � ] � ^ � _ � ` � a b � c d C e q f � g � h % i � j ^ k � l m � n � o � p > q r s � t � u v � w x 3 y y z � { $ | k } ~ ~ K � � � � E � � � � � � � � � � � q � < � U � � � 6 � & � � � � � ~ � � � ` � � � � = � K � { � q � � � � � � � � : � ] � � � � � � f � � � � � 8 � � � � p � x � � � � � 3 � $ � � � % � � � � � � � � � � L � ' � � � � � P � � 4 � � � O � � � k � � � ^ � � � f � � � U � � � � � h � Y � � � � � � � � � � � � � � � � � � � 0@ � 1 G < 3N \ T� g G � 8� + : c ; c m � 5 j� = lS u m� ) � � n� � � � � � � � c � �, 2 c F � � ^ �R X c q � � � ?~ � � � � �� � & � A 1� � � 9 6 � 7� � 8� ] 9c ;� 9 = � � ( +U � ,� � -Z ^ .S � / � � U � 0 <� � = c � > � T h J � K` � L� ( j M� 0 � NS 8 � O @ � P H � QF P 1 R � X S q ` l � �� � �� ? N � � 6 � � � � Q � � � � 8 . 8 W g f N $� z � X L ) � � N ? y � I k � 9 H � � G �s w �? � � x K � � 9 K � � � � � J � � � | � � � 9 � � � � �) � �? # �- z � h �� � � . � �� � ? � - G � � P G 9 9 � ' � � � � � � � � � � � \ � � � ' � � � s � & � 1 � < � G ) R � ] � h � s � ~ � � � � � @ � 9 N +1 d � 3 � - � @ g �� ^ ` � Q ] @� � � o� O q9 q r9 � sG irq t@ dma u@ � v@ �� � �� � � 9 �M H �K b �K � �K �K �K ! � \ � �c � �c � �1 w �� �� � � � (~u � � � & � � &� � ,� � -� 8 � ! � � "N � $� X %� 5 .u ( � 8 � 0 � E G { - � " � x; �g A � g � � w = � � / w 9 ; � 9 G � 9 � �� r � 9 � G k � # 3 F G � � � c � G � @ � @ � � g j � P 9 � f@ �a u� @a !� ^ c � � � "� l c � � � # "@ � c � ! � � c $ E � !� B c !" 9c 4 � � "& �� K � %L ^ � !I 4� t S !k (� � � # "� a c � c !j ) c � c 9 # !� !r c � c c c !t b c � � � c !* c � � # !� Kc : � � � !a "� P o !� Lc p � � o !/ �� � � c � "� � � � � # !N V � � � ! W � � � !� �� � � &� $2 � ! # c v c "U ko . c '� �� P5 �w (mod �� \ X (v �� � � )c5 p � *U @a *T3 )�5 P � *U| *T uA *Q9 )�5 P � *U| *T A *Q4 )�5 P *U| *T �A *Q0 )�5 P E *U| *T �A *Q2 )�5 P o *U| *T �A *Q1 ) 6 P � *U| *T �A *Q8 )(6 P � *U| *T �A *Q7 )D6 P � *U| *T �A *Q6 )`6 P *U| *T �A *Q5 )|6 P A *U| *T �A *Q3 )�6 P k *U| *T �A *Q: )�6 P � *U| *T �A *Q< )�6 P � *U| *T B *Q= )�6 P � *U| *T B *Q> ) 7 P *U| *T (B *Q? )7 P = *U| *T 6B *Q; )(7 : U *U � +?7 *U| *T HB , �� �& � � -} �$� � � -b �4� � � � � � ��.� � c R F (ret � c � � /ifr �M �@0� �)�& � E *U�T*T �B *Q�� )�& � f *U2*T2*Q0 )' 8 � *U�@*Q? )' � � *Uv *T � )&' � � *Uv )B' t � *U &B 1U' � 1d' � +�' t *U &B ,� �� P- � ��* -} � � -b �/� W M � � � ��.� � c � � (ret � c � � /ifr �M ��.� �� � � .� �\ G ? 0� t2` & .� u� � � )�. K *U| +R1 K *U| 3�* �. � J5 4�* 4�* W S 5� 6�* � � 7�* � � 6�* +@4 K *Uv 8�* �. � 6�* / - 9�* �. 6�* T R +�. K *Uv )�. 4 *U [A +�. *U| *Tv :�* / / P �: 4�* y w 4�* � � 6�* � � 8�* %/ � ;�* 8�* // ! 6�* 2 0 9�* // ! 6�* W U +E/ K *U} ) / 4 *U �@ +#/ *U| *T} :�* P/ P/ P �? 4�* | z 4�* � � 6�* � � 8�* u/ � ;�* 8�* {/ % 6�* 5 3 9�* {/ % 6�* Z X +�/ K *U} )\/ 4 $ *U �@ +s/ *U| *T} :�* �/ �/ P �D 4�* } 4�* � � 6�* � � 8�* �/ � ;�* 8�* �/ % 6�* 8 6 9�* �/ % 6�* ] [ +�/ K *U} )�/ 4 ) *U �@ +�/ *U| *T} 3�* �3 DS 4�* � � 4�* � � 5 6�* � � 7�* @ � 6�* ; 9 +0 K *U} 8�* 4 6�* ` ^ 9�* 4 6�* � � +%4 K *U} )�3 4 7 *U QA +4 *U| *T} :�* `0 `0 P X 4�* � � 4�* � � 6�* � 8�* �0 � ;�* 8�* �0 ! 6�* c a 9�* �0 ! 6�* � � +�0 K *U} )l0 4 = *U �@ +�0 *U| *T} :�* �0 �0 P ] 4�* � � 4�* � � 6�* � 8�* �0 � ;�* 8�* �0 ! # 6�* f d 9�* �0 ! 6�* � � +�0 K *U} )�0 4 B *U A +�0 *U| *T} :�* 1 1 J 2b! 4�* � � 4�* � � 6�* 8�* %1 � ;�* 8�* /1 (! 6�* i g 9�* /1 6�* � � +E1 K *U} ) 1 4 G! *U 9A +#1 *U| *T} :�* `1 `1 P 8g" 4�* � � 4�* � � 6�* 8�* �1 �! ;�* 8�* �1 ! -" 6�* l j 9�* �1 ! 6�* � � +�1 K *U} )l1 4 L" *U ?A +�1 *U| *T} :�* �1 �1 P >l# 4�* � � 4�* � � 6�* 8�* �1 �" ;�* 8�* �1 ! 2# 6�* o m 9�* �1 ! 6�* � � +�1 K *U} )�1 4 Q# *U IA +�1 *U| *T} :�* 2 2 P q$ 4�* � � 4�* � � 6�* 8�* %2 �# ;�* 8�* /2 ! 7$ 6�* r p 9�* /2 ! 6�* � � +E2 K *U} ) 2 4 V$ *U A +#2 *U| *T} :�* P2 P2 P v% 4�* � � 4�* � � 6�* 8�* u2 �$ ;�* 8�* 2 ! <% 6�* u s 9�* 2 ! 6�* � � +�2 K *U} )\2 4 [% *U �B +s2 *U| *T} :�* �2 �2 P {& 4�* � � 4�* � � 6�* 8�* �2 �% ;�* 8�* �2 ! A& 6�* x v 9�* �2 ! 6�* � � +�2 K *U} )�2 4 `& *U A +�2 *U| *T} :�* �2 �2 P �' 4�* � � 4�* � � 6�* 8�* 3 �& ;�* 8�* 3 ! F' 6�* { y 9�* 3 ! 6�* � � +53 K *U} )�2 4 e' *U !A +3 *U| *T} :�* @3 @3 P &�( 4�* � � 4�* � � 6�* 8�* e3 �' ;�* 8�* o3 ! K( 6�* ~ | 9�* o3 ! 6�* � � +�3 K *U} )L3 4 j( *U )A +c3 *U| *T} :�* �3 �3 P ,�) 4�* � � 4�* � � 6�* ! 8�* �3 �( ;�* 8�* �3 ! P) 6�* � 9�* �3 ! 6�* � � +�3 K *U} )�3 4 o) *U 2A +�3 *U| *T} )c- ^ �) *U0 )�- � �) *Uv *T �B *Q�� )�- � �) *U2*T2*Q0 )�- 8 * *U��*Q? )�- � +* *Uv *T � )�- � C* *Uv ) 0 � b* *U �@ )40 � �* *U �@ +I0 � *Uv <I �c + =� �� =� �0� > �� ?�* >� � � @>� �� @>� �� ,� �� & � �~, -} � � � � -b �-� � � � ��.� � c b X (ret � c � � /ifr �M �@0� �)& � �+ *U�T*T �B *Q�� )6& � �+ *U2*T2*Q0 )T& 8 , *U�@*Q? )j& � ,, *Uv *T !� )v& � D, *Uv )�& t c, *U �@ 1�& � 1�& � ,� -� ) � ��1 -} - � -b -/� N J B . ��(ifa . � � .� / c � � .� 1� j d .9 2� � � .� 3� � � . 4� � � .r 5� � .� 6� D! *! A� y* 2P �- .� o � Y" Q" 5P .� o � �" �" +(+ K *U�� 2� 3. .� p � %# !# 5� .� p � _# [# +;+ K *U| 2� �. .� q � �# �# 5� .� q � �# �# +M+ K *Us 2� �. .� r � $ $ 5� .� r � G$ C$ +_+ K *U 2 / .� s � �$ }$ 5 .� s � �$ �$ +r+ K *U} 2@ R/ .� |� �$ �$ +�+ K *U�� 2p �/ .m }� J% D% 5� .� }� �% �% 1, K 2� �/ .m ~� �% �% 5 .� ~� & & +(, K *U| 2@ 50 .m � \& V& 5p .� � �& �& +H, K *Us 2� �0 .m �� �& �& 5� .� �� 2' .' +h, K *U 2 �0 .m �� l' h' 50 .� �� �' �' +�+ K *U} )%) ^ �0 *U0 )=) � 1 *U�� )�) t 81 *U �@ *Q~ *R| *Xs *Y )�) Z1 *U��*T�� ) * �1 r1 *T~ )?* �1 �1 *T~ )b* �1 �1 *T~ 1�* � )�+ � �1 *Uv )�+ �1 �1 *T~ 1�, � B �� x2 C� �(� C� �2c Dbuf � x2 Derr � c E, � c Dn � - Dlen � - E� �� Dptr � � @>k �2 � �2 F9 � Bg �� �2 C} �$� Cb �4� Dpid � � E� � c E � c B� �� 3 C} �$� Cb �4� Dpid � � E� � c Go �� `4 ��3 H} �(� �' �' Hb �8� ( ( 1i4 |4 I~4 t *U OA J� ��4 i �R4 Kpid �o d( V( H �&� ) ) 1�4 � L5 � �3 *U�T )5 R4 4 *Us L!5 � /4 *U�T II5 � *T cA *Q�T M� `c |4 Npid `� Dret a c O� Ko P4 ��4 IZ4 *UN P�2 �' a ��5 4�2 �) �) 4�2 0* ** Q�2 �hQ�2 �l6�2 �* |* R�2 �' �' �a5 4�2 �* �* 4�2 �* �* ;�2 ;�2 ;�2 1�' � )�' � �5 *U�T*T �@ *Q�h*R�l +�' � *U0 S�1 ( �c6 T�1 UT 2 TQ2 ��w;"2 ;.2 ;:2 ;D2 ;P2 ;\2 U�1 �( �V 2 V�1 5 Q2 ��w;"2 ;.2 ;:2 QD2 ^;P2 ;\2 9h2 �( ;i2 P�2 �, v �p7 4�2 + + 4�2 M+ C+ Q3 �\63 �+ �+ R�2 0- 0- �7 4�2 , , 4�2 ;, 9, ;3 ;3 1?- � 1�, � )�, � =7 *Uv *T �@ *Q�\ )- ~ T7 *U0 +(- t *U �@ PR4 �4 W �8 4c4 b, ^, 6o4 �, �, RR4 �4 �4 `�7 4c4 �, �, ;o4 1�4 � )�4 � 8 *T0 1�4 � Wi _ % % $ > $ > & I : ; 9 I I : ; 9 : ; 9 I8 7 I < 4 : ;9 I?< 4 : ; 9 I?< 4 : ; 9 I 4 G: ; 9 .?: ;9 'I< I .?: ; 9 '< .?: ; 9 'I< .?: ;9 '< . ?: ; 9 'I< .?: ;9 'I@�B ��1 �� �B .?: ; 9 'I@�B : ; 9 I�B 4 : ; 9 I 4 : ; 9 I�B ��1 �� 1 4 : ; 9 I U !4 : ; 9 I�B " U #I $! I/ %.?: ; 9 'I@�B % : ; 9 I $ > $ > & I I 7 I : ;9 I > I: ; 9 ( : ; 9 : ; 9 I8 < 'I I ' 4 : ;9 I?< 4 : ; 9 I?< 'I > I: ; 9 > I: ;9 ( ( I ! I/ : ;9 I8 : ; 9 I8 : ; 9 : ; 9 I 4 : ; 9 I 4 : ;9 I !.?: ; 9 'I< ".?: ;9 'I< # $.?: ; 9 '< %.?: ;9 '< &. ?: ; 9 'I< '.?: ;9 'I@�B (4 : ;9 I�B )��1 *�� �B +��1 ,.: ;9 'I@�B - : ;9 I�B .4 : ;9 I�B /4 : ;9 I 0 : ;9 1�� 1 2 U 3 1R�BUX YW 4 1�B 5 U 64 1�B 7 1U 8 1 9 1 : 1R�BX YW ;4 1 <.: ;9 'I = : ;9 I >4 : ;9 I ? @ A : ;9 B.: ; 9 'I C : ; 9 I D4 : ; 9 I E4 : ; 9 I F! I/ G.: ; 9 'I@�B H : ; 9 I�B I���B1 J.?: ; 9 '@�B K : ; 9 I�B L���B1 M.?: ; 9 'I N : ; 9 I O.?: ; 9 'I@�B P.1@�B Q4 1 R 1R�BX Y W S.1@�B T 1 U 1R�BUX Y W V 1 W. ?<n: ; k . � psutil /usr/include/bits /usr/include /usr/include/sys /opt/python/cp36-cp36m/include/python3.6m _psutil_common.c types.h stdio.h types.h pyport.h time.h object.h pyerrors.h _psutil_common.h stdlib.h modsupport.h abstract.h string.h errno.h = @# � =w: �X:X<� �� I=� X t � � � �$n$\$��� � ;=� X t � � � �#�#\#��� � ;=� X t � � � �8�8yJ Ct�L��z�4 �Y2�2J @t�L � M t��t� .�� K� Y � � � psutil /opt/rh/devtoolset-10/root/usr/lib/gcc/x86_64-redhat-linux/10/include /usr/include/bits /usr/include /usr/include/sys /opt/python/cp36-cp36m/include/python3.6m /usr/include/netinet /usr/include/net /usr/include/asm-generic /usr/include/linux _psutil_posix.c stddef.h types.h stdio.h types.h unistd.h stdint.h pyport.h time.h object.h boolobject.h methodobject.h moduleobject.h pyerrors.h resource.h resource.h socket_type.h sockaddr.h socket.h in.h confname.h if.h ifaddrs.h int-ll64.h types.h if_packet.h _psutil_common.h netdb.h listobject.h unicodeobject.h modsupport.h ioctl.h socket.h longobject.h signal.h errno.h <built-in> 3 & �3v J3v� J�=-[ /_f sY � t.Z] ' Y �X:�t:z P:z� P�=-[ /[f sY � u.[Y uR� Y �X ztPX:�|f: J At=-^�ZtKp� t�: �:s .:s. .:s< � Y � 8 @ � -�� ^� ^X # Y Y�� �L IY \K� Y `� lt�95`5 25* �XY� ��XJ �� ; B ' � � � � � � � % L) J � J Z K �q ? 0 �� = � < � �tK <�<XX KXK � X � X � X � X � X � � l� � � � � � � � � � % L) J 7�X� ]� �tK < . t X J t � � � � � � � � � � �z��� A�o;;- D�v <:�~t:� Z�h >�-^�.L �fp. �f5��5 35) �Y� t��J� =�~� /�f sY ��u X��������������� ��~�� � /Ij��t�� �~D�� � /Ij��� 7��� � /I2��� =��� � /I2��� � ����� / ;Z� ����� � /Ij��� � ��� � /Ij��� � ��� � /Ij��X � � B��~�� � /Ij��� � �~�� � /Ij��� � ��� � /Ij��� � ��� � /Ij��� � ��� � /Ij��� � ��� � /Ij��� � ��� � /Ij��� � ��� � /Ij��� ��~�� � /Ij��X ���}� >� �> KXu I ��L .j� l Jt=W [ X .\dX &�h� ./�/ �X g EA � t K G? �= I/ X P5 � / � X� <� � <� <� <� <� <� <� <� <� <� <N <N <N <J � Y���� < tz_dsttime syscall psutil_setup /project _typeobject _object ob_refcnt double NoSuchProcess Py_ssize_t float _py_xdecref_tmp long long unsigned int timezone unsigned char psutil_set_debug strerror _py_decref_tmp PSUTIL_DEBUG short unsigned int PyObject_IsTrue psutil_PyErr_SetFromOSErrnoWithSyscall tz_minuteswest PyObject_CallFunction AccessDenied psutil/_psutil_common.c ob_type self PyErr_SetObject __errno_location PyObject __ssize_t fullmsg long long int GNU C17 10.2.1 20210130 (Red Hat 10.2.1-11) -mtune=generic -march=x86-64 -g -O3 -fwrapv -fPIC PyArg_ParseTuple PyExc_ValueError _Py_Dealloc short int args PSUTIL_CONN_NONE __pid_t _Py_NoneStruct psutil_check_pid_range PyErr_SetString sprintf getenv PyExc_OSError _SC_EQUIV_CLASS_MAX ifr_ifrn m_copy ifa_addr _SC_THREAD_PRIO_PROTECT sin6_flowinfo ifr_ifru PyModule_Create2 getnameinfo __priority_which_t _SC_VERSION Py_BuildValue _SC_NL_NMAX m_methods _longobject slot _SC_SYNCHRONIZED_IO _SC_THREAD_PRIORITY_SCHEDULING _SC_NPROCESSORS_ONLN sa_data m_base _SC_TIMEOUTS sockaddr _SC_BASE _SC_PII_OSI_COTS sockaddr_un _SC_THREAD_SAFE_FUNCTIONS _SC_ATEXIT_MAX sockaddr_ns _SC_STREAM_MAX ifru_slave psutil_posix_getpriority _SC_PRIORITIZED_IO py_ptp _SC_V6_ILP32_OFF32 _SC_THREAD_SPORADIC_SERVER sll_pkttype __socket_type ifu_dstaddr _SC_SHRT_MIN _SC_USHRT_MAX _SC_NL_TEXTMAX _SC_STREAMS __rlimit_resource IFF_PORTSEL _SC_THREAD_DESTRUCTOR_ITERATIONS _SC_PIPE _SC_BC_DIM_MAX psutil_getpagesize _SC_MAPPED_FILES _SC_2_C_BIND _SC_MQ_OPEN_MAX _SC_XOPEN_SHM _SC_INT_MAX _SC_2_FORT_DEV kill _SC_XOPEN_XPG2 _SC_NZERO _SC_XOPEN_XPG4 IFF_NOTRAILERS _SC_CPUTIME _SC_PII_INTERNET _SC_V7_LP64_OFF64 _SC_LEVEL3_CACHE_SIZE _SC_MB_LEN_MAX ifreq __u6_addr16 __caddr_t _py_tmp sockaddr_ipx sll_hatype getifaddrs ifru_broadaddr _SC_REALTIME_SIGNALS __RLIMIT_LOCKS m_free moduledef py_netmask _SC_DEVICE_SPECIFIC_R uint32_t in_addr_t mem_start _SC_SAVED_IDS __RLIM_NLIMITS _SC_2_C_DEV _SC_XBS5_LPBIG_OFFBIG ob_base _SC_2_C_VERSION _SC_SEM_VALUE_MAX _SC_SCHAR_MAX ifru_map _SC_SSIZE_MAX _SC_2_UPE PyExc_RuntimeError freeifaddrs RLIMIT_CORE ifa_name _SC_INT_MIN _SC_BC_BASE_MAX __u6_addr8 _SC_POLL sysconf ml_flags _SC_SYSTEM_DATABASE_R sockaddr_dl sll_halen __RLIMIT_NPROC _SC_2_LOCALEDEF IFF_MULTICAST sin_family _SC_XOPEN_XPG3 _SC_T_IOV_MAX _SC_LEVEL1_ICACHE_ASSOC in_port_t py_str _SC_SYMLOOP_MAX retval _SC_TRACE_LOG ifa_ifu __u6_addr32 append_flag _SC_THREAD_CPUTIME sin_zero _SC_CHAR_MAX _SC_XBS5_ILP32_OFFBIG _SC_PII_INTERNET_DGRAM s_addr py_address _SC_2_PBS_TRACK __u16 psutil_net_if_flags _SC_FILE_ATTRIBUTES _SC_ASYNCHRONOUS_IO _SC_FSYNC __RLIMIT_NICE sockaddr_inarp _SC_DEVICE_SPECIFIC sockaddr_ax25 ifa_netmask IFF_UP _SC_THREAD_ATTR_STACKSIZE _SC_REGEX_VERSION _SC_MEMLOCK _SC_DELAYTIMER_MAX _SC_LONG_BIT _SC_SEM_NSEMS_MAX sockaddr_iso PyModule_AddObject PyModule_AddIntConstant _SC_XOPEN_STREAMS _SC_LEVEL1_ICACHE_LINESIZE _SC_SIGNALS sll_family __RLIMIT_OFILE IFF_SLAVE ioctl _SC_2_PBS_ACCOUNTING _SC_AIO_MAX _SC_LEVEL2_CACHE_LINESIZE _SC_XOPEN_VERSION mod_methods sa_family_t _SC_CLK_TCK _SC_SHELL sll_protocol _SC_UIO_MAXIOV _SC_TZNAME_MAX ifru_data _SC_SPORADIC_SERVER _SC_MEMLOCK_RANGE _SC_AVPHYS_PAGES __RLIMIT_NLIMITS _SC_V7_ILP32_OFFBIG _SC_PII_XTI close sll_ifindex flag_name sock _SC_V7_LPBIG_OFFBIG _SC_C_LANG_SUPPORT_R SOCK_DCCP _SC_LEVEL3_CACHE_ASSOC uint8_t _SC_FILE_SYSTEM ifru_netmask _SC_PAGESIZE _SC_V6_ILP32_OFFBIG SOCK_PACKET __id_t _SC_SIGQUEUE_MAX _SC_SPAWN psutil_posix_setpriority _SC_DEVICE_IO _SC_V6_LPBIG_OFFBIG _SC_2_VERSION ifru_mtu _SC_LEVEL4_CACHE_SIZE ml_name SOCK_DGRAM m_size IFF_LOOPBACK _SC_USER_GROUPS_R PyModuleDef_Slot visitproc sin_port _SC_LINE_MAX sockaddr_eon psutil/_psutil_posix.c IFF_AUTOMEDIA lladdr py_broadcast _SC_LEVEL1_DCACHE_ASSOC _SC_HOST_NAME_MAX _SC_C_LANG_SUPPORT __RLIMIT_RSS _SC_THREAD_STACK_MIN IFF_RUNNING _SC_SEMAPHORES _SC_UINT_MAX _SC_CHILD_MAX __RLIMIT_MSGQUEUE PyUnicode_FromString _SC_NGROUPS_MAX _SC_SINGLE_PROCESS IFF_ALLMULTI __be16 __in6_u sin_addr _SC_TTY_NAME_MAX _SC_PII_INTERNET_STREAM RLIMIT_AS _SC_MEMORY_PROTECTION PyModuleDef_Base m_name _SC_XOPEN_CRYPT PyCFunction psutil_net_if_is_running ifru_newname _Py_FalseStruct RLIMIT_FSIZE _SC_CHAR_BIT _SC_LEVEL1_DCACHE_SIZE IFF_NOARP _SC_CLOCK_SELECTION m_init ifrn_name sin6_addr psutil_net_if_mtu _SC_TIMERS freefunc _SC_BC_SCALE_MAX _SC_ULONG_MAX uint16_t sin6_port _SC_MQ_PRIO_MAX _SC_TRACE _SC_SPIN_LOCKS ifru_flags addrlen _SC_LEVEL1_DCACHE_LINESIZE _SC_BC_STRING_MAX PyLong_FromLong mem_end SOCK_STREAM RLIMIT_DATA _SC_V7_ILP32_OFF32 _SC_TRACE_SYS_MAX _SC_FD_MGMT _SC_REGEXP sin6_family IFF_PROMISC RLIMIT_CPU PyInit__psutil_posix _SC_LEVEL1_ICACHE_SIZE _SC_ADVISORY_INFO __RLIMIT_RTPRIO _SC_SHRT_MAX _SC_XBS5_LP64_OFF64 __builtin_strncpy _SC_READER_WRITER_LOCKS _SC_SYSTEM_DATABASE _SC_XOPEN_REALTIME_THREADS _SC_THREAD_ROBUST_PRIO_PROTECT _SC_2_CHAR_TERM _SC_PASS_MAX _SC_FIFO m_slots _SC_ARG_MAX ifru_hwaddr ml_doc _SC_2_PBS_CHECKPOINT PyMethodDef _SC_XOPEN_REALTIME PRIO_USER ifru_dstaddr psutil_getpagesize_pywrapper _SC_2_FORT_RUN socket _SC_TRACE_EVENT_FILTER psutil_raise_for_pid _SC_RE_DUP_MAX _SC_SCHAR_MIN sockaddr_in _SC_THREAD_ROBUST_PRIO_INHERIT _SC_THREADS PyList_Append _SC_PII __RLIMIT_SIGPENDING _SC_TRACE_INHERIT _SC_WORD_BIT _SC_XBS5_ILP32_OFF32 _SC_PII_OSI_M RLIMIT_NOFILE m_traverse _SC_2_SW_DEV _SC_OPEN_MAX _SC_XOPEN_UNIX sockaddr_in6 _SC_AIO_LISTIO_MAX PyErr_SetFromErrno _SC_PII_OSI _SC_UCHAR_MAX _SC_MONOTONIC_CLOCK m_clear py_tuple ifaddr SOCK_RAW _SC_PRIORITY_SCHEDULING m_doc _SC_SELECT _SC_NETWORKING _SC_TIMER_MAX _SC_TRACE_EVENT_NAME_MAX _SC_V6_LP64_OFF64 _SC_GETGR_R_SIZE_MAX _SC_LEVEL2_CACHE_SIZE _SC_LOGIN_NAME_MAX _SC_EXPR_NEST_MAX IFF_POINTOPOINT _SC_NPROCESSORS_CONF __RLIMIT_RTTIME ifru_addr __socklen_t traverseproc SOCK_SEQPACKET PRIO_PROCESS sockaddr_at psutil_pid_exists ifa_flags _SC_SS_REPL_MAX _SC_RAW_SOCKETS _SC_BARRIERS _SC_CHAR_MIN _SC_LEVEL4_CACHE_LINESIZE psutil_convert_ipaddr _SC_THREAD_PROCESS_SHARED _SC_NL_MSGMAX _Py_TrueStruct sin6_scope_id m_index _SC_LEVEL4_CACHE_ASSOC _SC_CHARCLASS_NAME_MAX _SC_PII_OSI_CLTS psutil_net_if_addrs py_retlist _SC_TYPED_MEMORY_OBJECTS ml_meth __RLIMIT_MEMLOCK base_addr _SC_2_PBS _SC_PHYS_PAGES IFF_MASTER _SC_PII_SOCKET IFF_DEBUG _SC_MULTI_PROCESS _SC_LEVEL2_CACHE_ASSOC __priority_which sa_family ifmap _SC_THREAD_KEYS_MAX PyErr_Format IFF_BROADCAST _SC_THREAD_THREADS_MAX _SC_GETPW_R_SIZE_MAX ifa_data SOCK_NONBLOCK PRIO_PGRP _SC_2_PBS_MESSAGE _SC_RTSIG_MAX sockaddr_ll _SC_THREAD_ATTR_STACKADDR PyList_New PyModuleDef _SC_FILE_LOCKING _SC_SHARED_MEMORY_OBJECTS _SC_TRACE_NAME_MAX _SC_COLL_WEIGHTS_MAX _SC_XOPEN_ENH_I18N _SC_XOPEN_LEGACY _SC_JOB_CONTROL IFF_DYNAMIC ifa_next RLIMIT_STACK _SC_MESSAGE_PASSING _SC_NL_LANGMAX _SC_IOV_MAX _SC_USER_GROUPS SOCK_CLOEXEC _SC_IPV6 _SC_LEVEL3_CACHE_LINESIZE _SC_2_PBS_LOCATE ifru_ivalue _SC_NL_SETMAX _SC_NL_ARGMAX ifu_broadaddr nic_name _SC_THREAD_PRIO_INHERIT sll_addr _SC_TRACE_USER_EVENT_MAX inquiry sockaddr_x25 SOCK_RDM _SC_AIO_PRIO_DELTA_MAX _SC_XOPEN_XCU_VERSION 0 7 U7 � �U� 0 > T> I UI � �T� X c P � � U� * �U� � � T� � U� * �T� 0 Q UQ � �U� x P � V� � V � � V� � V � � V� � V � � U� & �U� � � P� V # V � V # V � � V # V U G \G � �U� ^ e Pe | V� � V f | V� � V k r V� � V c � P� A \ + > P � � U� � �U� � � T� � U� � �T� � � ��� P B VH T UT i Vj � V % PH T P P T UT E �U� P b Tb � V� � �T�� � V� E �T� P � ��� � P� � V� �� P Q VV J ��J W VW E �� � � P! 3 P c r Pr � \� D \V E \ � � V� VV J VW * V � � \� � \4 D \J W \ � � [A �* @ [A � � � \* @ \ � � 0�� � P� � V� � 0�* @ V * E V � � V � � V � E �@ � � E \ � 0� " P" / ]/ E 0� / J ] / J ] J � �@ � J � \ J _ 0�_ r Pr { ]{ � 0� { � ] { � ] � � �@ � � � \ � � 0�� � P� � ]� � 0� � � ] � � ] � % QA � � % \ � � 0�� P ] % 0� � ] * ] * ] V � �@ � V � \ V o 0�o � P� � ]� � 0� � � ] � � ] � � A � � � \ � � 0�� � P� � ]� � 0� � � ] � � ] � E 9A � � E \ � 0� " P" / ]/ E 0� / J ] / J ] W � ?A � W � \ W o 0�o � P� � ]� � 0� � � ] � � ] � � IA � � � \ � � 0�� � P� � ]� � 0� � � ] � � ] � E A � � E \ � 0� " P" / ]/ E 0� / J ] / J ] J � �B � J � \ J _ 0�_ r Pr ] � 0� � ] � ] � � A � � � \ � � 0�� � P� � ]� � 0� � � ] � � ] � 5 !A � � 5 \ � � 0�� P ] 5 0� : ] : ] : � )A � : � \ : O 0�O b Pb o ]o � 0� o � ] o � ] � � 2A � � � \ � � 0�� � P� � ]� � 0� � � ] � � ] U � �U� T U � �T� 8 ��8 S PS � V� � U� � V j u P� � P U � �U� $ T$ � �T� K V � V� � Vm � V Y � ^ ^ ^� � ^m � ^ * < P< �� � �� * � 0�� � P� � Z� � ��� 0� � Z� � �� � 0�m � 0�� � S� � 0�� � Z� � 0� * Y 0�Y � \� � 0�( > P> � \ ( \( 0�� m \m � P� � \� � 0� * Y 0�Y � S� B 0�B N PN � S ; S; � 0�� m Sm � 0�� � S� � P� � 0� * s 0�y � _� e 0�e l Pl | _ � _ M _M � 0�� m _m � 0�� � S� � 0�� � Z� � 0� * s 0�y � ]� p 0� � ] _ ]_ � 0�� � P� m ]m � 0�� � S� � 0�� � Z� � 0� � � Z� � �� ' Z' �� � � Z� � �� ' Z' �� � � \( \ � � \( \ � � S; S � � S; S � � _M _ � � _M _ � � ]_ ] � � ]_ ] � � ��� � Q� m �� � � Z� � Z� Z � � Z� Z � � \� � \ m \ � � \ 6 \ � � S� � S( m S � � S6 V S � � _� � _H m _ � � _V m _ � � ]� � ] � � ]� � ] ` h Uh ~ �U� ` h Th ~ �T� � � U� � S� �U� S ! �U�! C SC I �U� � � T� V U �T� V U ! �T�! D VD H QH I �T� � � U� � �U� � � T� � U� � �T� � � P� � P � � �U� � � �T� � � U� F �U� � � T� . V. / �T�/ E VE F �T� P ' T/ > T / ? �U� / ? V � � U� � �U� � � P� � X � � �U� f r � � k r � � � � � � � � � � � � � � [ f � � � � ( � � ( ; � � ; M � � M _ � � _ r � � � � � � � � � � � � ( � � ( � � ( H � � @ H � � H p � � ` p � � � � � � � � � 4 ? J R � � * E � � * E � � * � 8 ` � � D � � �" T7 @ �B hC �] �] �] �] �_ ` @a �b � �" �" ! �" 7 �b C �] j 0# v �] � � � � � & � � �& � � �' a � ( ) � $ �, v = P- � Q `4 n @a h x �a � � G � � �] � �] � �B � �b � ` � � � 2 G ` | � � � T7 � � P5 � � � P4 & 9 L e u � � � � � � � � �4 i @# � 3 B X h y �4 W � �# � � � � p% W � � � p$ � �b ' 4 I c y % j � " � � �% ) � crtstuff.c deregister_tm_clones __do_global_dtors_aux completed.0 __do_global_dtors_aux_fini_array_entry frame_dummy __frame_dummy_init_array_entry _psutil_common.c _psutil_posix.c psutil_net_if_mtu psutil_net_if_is_running psutil_posix_setpriority psutil_convert_ipaddr psutil_net_if_addrs psutil_posix_getpriority psutil_net_if_flags psutil_getpagesize_pywrapper moduledef mod_methods __FRAME_END__ __dso_handle _DYNAMIC __GNU_EH_FRAME_HDR __TMC_END__ _GLOBAL_OFFSET_TABLE_ getenv@@GLIBC_2.2.5 PyList_New PyModule_AddIntConstant __errno_location@@GLIBC_2.2.5 strncpy@@GLIBC_2.2.5 getpriority@@GLIBC_2.2.5 _ITM_deregisterTMCloneTable PyErr_SetFromErrno _Py_Dealloc PyErr_SetObject _fini PyExc_RuntimeError PyInit__psutil_posix PyErr_SetString PyExc_ValueError setpriority@@GLIBC_2.2.5 psutil_getpagesize ioctl@@GLIBC_2.2.5 close@@GLIBC_2.2.5 getnameinfo@@GLIBC_2.2.5 PyLong_FromLong PyList_Append PyExc_OSError _Py_FalseStruct __gmon_start__ PyObject_CallFunction Py_BuildValue kill@@GLIBC_2.2.5 PyModule_Create2 psutil_raise_for_pid psutil_PyErr_SetFromOSErrnoWithSyscall _Py_NoneStruct getifaddrs@@GLIBC_2.3 PyObject_IsTrue PyArg_ParseTuple psutil_pid_exists NoSuchProcess _Py_TrueStruct freeifaddrs@@GLIBC_2.3 psutil_set_debug PyUnicode_FromString PyModule_AddObject AccessDenied sysconf@@GLIBC_2.2.5 PSUTIL_DEBUG PyErr_Format sprintf@@GLIBC_2.2.5 _ITM_registerTMCloneTable strerror@@GLIBC_2.2.5 psutil_check_pid_range __cxa_finalize@@GLIBC_2.2.5 _init psutil_setup socket@@GLIBC_2.2.5 .symtab .strtab .shstrtab .note.gnu.build-id .gnu.hash .dynsym .dynstr .gnu.version .gnu.version_r .rela.dyn .rela.plt .init .text .fini .rodata .eh_frame_hdr .eh_frame .init_array .fini_array .data.rel.ro .dynamic .got .got.plt .data .bss .comment .debug_aranges .debug_info .debug_abbrev .debug_line .debug_str .debug_loc .debug_ranges 8 8 $ . ���o ` ` ` 8 � � ( @ � � [ H ���o D D n U ���o � � P d � n B � � x x s ` ~ �" �" � � T7 T7 � 2 @ @ � � �B �B � � hC hC � � �] �M � �] �M � �] �M � �] �M � � �_ �O X � ` P @ � @a @Q � � �b �R � 0 �R / � �R ` OS �? ے r # M� 5 / 0 �� M : �� �, E � p @� � ! < � � � S _pswindows.py 0000644 00000106155 15030211765 0007322 0 ustar 00 # Copyright (c) 2009, Giampaolo Rodola'. All rights reserved. # Use of this source code is governed by a BSD-style license that can be # found in the LICENSE file. """Windows platform implementation.""" import contextlib import enum import functools import os import signal import sys import time from collections import namedtuple from . import _common from ._common import ENCODING from ._common import AccessDenied from ._common import NoSuchProcess from ._common import TimeoutExpired from ._common import conn_tmap from ._common import conn_to_ntuple from ._common import debug from ._common import isfile_strict from ._common import memoize from ._common import memoize_when_activated from ._common import parse_environ_block from ._common import usage_percent from ._psutil_windows import ABOVE_NORMAL_PRIORITY_CLASS from ._psutil_windows import BELOW_NORMAL_PRIORITY_CLASS from ._psutil_windows import HIGH_PRIORITY_CLASS from ._psutil_windows import IDLE_PRIORITY_CLASS from ._psutil_windows import NORMAL_PRIORITY_CLASS from ._psutil_windows import REALTIME_PRIORITY_CLASS try: from . import _psutil_windows as cext except ImportError as err: if ( str(err).lower().startswith("dll load failed") and sys.getwindowsversion()[0] < 6 ): # We may get here if: # 1) we are on an old Windows version # 2) psutil was installed via pip + wheel # See: https://github.com/giampaolo/psutil/issues/811 msg = "this Windows version is too old (< Windows Vista); " msg += "psutil 3.4.2 is the latest version which supports Windows " msg += "2000, XP and 2003 server" raise RuntimeError(msg) from err else: raise # process priority constants, import from __init__.py: # http://msdn.microsoft.com/en-us/library/ms686219(v=vs.85).aspx # fmt: off __extra__all__ = [ "win_service_iter", "win_service_get", # Process priority "ABOVE_NORMAL_PRIORITY_CLASS", "BELOW_NORMAL_PRIORITY_CLASS", "HIGH_PRIORITY_CLASS", "IDLE_PRIORITY_CLASS", "NORMAL_PRIORITY_CLASS", "REALTIME_PRIORITY_CLASS", # IO priority "IOPRIO_VERYLOW", "IOPRIO_LOW", "IOPRIO_NORMAL", "IOPRIO_HIGH", # others "CONN_DELETE_TCB", "AF_LINK", ] # fmt: on # ===================================================================== # --- globals # ===================================================================== CONN_DELETE_TCB = "DELETE_TCB" ERROR_PARTIAL_COPY = 299 PYPY = '__pypy__' in sys.builtin_module_names AddressFamily = enum.IntEnum('AddressFamily', {'AF_LINK': -1}) AF_LINK = AddressFamily.AF_LINK TCP_STATUSES = { cext.MIB_TCP_STATE_ESTAB: _common.CONN_ESTABLISHED, cext.MIB_TCP_STATE_SYN_SENT: _common.CONN_SYN_SENT, cext.MIB_TCP_STATE_SYN_RCVD: _common.CONN_SYN_RECV, cext.MIB_TCP_STATE_FIN_WAIT1: _common.CONN_FIN_WAIT1, cext.MIB_TCP_STATE_FIN_WAIT2: _common.CONN_FIN_WAIT2, cext.MIB_TCP_STATE_TIME_WAIT: _common.CONN_TIME_WAIT, cext.MIB_TCP_STATE_CLOSED: _common.CONN_CLOSE, cext.MIB_TCP_STATE_CLOSE_WAIT: _common.CONN_CLOSE_WAIT, cext.MIB_TCP_STATE_LAST_ACK: _common.CONN_LAST_ACK, cext.MIB_TCP_STATE_LISTEN: _common.CONN_LISTEN, cext.MIB_TCP_STATE_CLOSING: _common.CONN_CLOSING, cext.MIB_TCP_STATE_DELETE_TCB: CONN_DELETE_TCB, cext.PSUTIL_CONN_NONE: _common.CONN_NONE, } class Priority(enum.IntEnum): ABOVE_NORMAL_PRIORITY_CLASS = ABOVE_NORMAL_PRIORITY_CLASS BELOW_NORMAL_PRIORITY_CLASS = BELOW_NORMAL_PRIORITY_CLASS HIGH_PRIORITY_CLASS = HIGH_PRIORITY_CLASS IDLE_PRIORITY_CLASS = IDLE_PRIORITY_CLASS NORMAL_PRIORITY_CLASS = NORMAL_PRIORITY_CLASS REALTIME_PRIORITY_CLASS = REALTIME_PRIORITY_CLASS globals().update(Priority.__members__) class IOPriority(enum.IntEnum): IOPRIO_VERYLOW = 0 IOPRIO_LOW = 1 IOPRIO_NORMAL = 2 IOPRIO_HIGH = 3 globals().update(IOPriority.__members__) pinfo_map = dict( num_handles=0, ctx_switches=1, user_time=2, kernel_time=3, create_time=4, num_threads=5, io_rcount=6, io_wcount=7, io_rbytes=8, io_wbytes=9, io_count_others=10, io_bytes_others=11, num_page_faults=12, peak_wset=13, wset=14, peak_paged_pool=15, paged_pool=16, peak_non_paged_pool=17, non_paged_pool=18, pagefile=19, peak_pagefile=20, mem_private=21, ) # ===================================================================== # --- named tuples # ===================================================================== # fmt: off # psutil.cpu_times() scputimes = namedtuple('scputimes', ['user', 'system', 'idle', 'interrupt', 'dpc']) # psutil.virtual_memory() svmem = namedtuple('svmem', ['total', 'available', 'percent', 'used', 'free']) # psutil.Process.memory_info() pmem = namedtuple( 'pmem', ['rss', 'vms', 'num_page_faults', 'peak_wset', 'wset', 'peak_paged_pool', 'paged_pool', 'peak_nonpaged_pool', 'nonpaged_pool', 'pagefile', 'peak_pagefile', 'private']) # psutil.Process.memory_full_info() pfullmem = namedtuple('pfullmem', pmem._fields + ('uss', )) # psutil.Process.memory_maps(grouped=True) pmmap_grouped = namedtuple('pmmap_grouped', ['path', 'rss']) # psutil.Process.memory_maps(grouped=False) pmmap_ext = namedtuple( 'pmmap_ext', 'addr perms ' + ' '.join(pmmap_grouped._fields)) # psutil.Process.io_counters() pio = namedtuple('pio', ['read_count', 'write_count', 'read_bytes', 'write_bytes', 'other_count', 'other_bytes']) # fmt: on # ===================================================================== # --- utils # ===================================================================== @functools.lru_cache(maxsize=512) def convert_dos_path(s): r"""Convert paths using native DOS format like: "\Device\HarddiskVolume1\Windows\systemew\file.txt" into: "C:\Windows\systemew\file.txt". """ rawdrive = '\\'.join(s.split('\\')[:3]) driveletter = cext.QueryDosDevice(rawdrive) remainder = s[len(rawdrive) :] return os.path.join(driveletter, remainder) @memoize def getpagesize(): return cext.getpagesize() # ===================================================================== # --- memory # ===================================================================== def virtual_memory(): """System virtual memory as a namedtuple.""" mem = cext.virtual_mem() totphys, availphys, _totsys, _availsys = mem total = totphys avail = availphys free = availphys used = total - avail percent = usage_percent((total - avail), total, round_=1) return svmem(total, avail, percent, used, free) def swap_memory(): """Swap system memory as a (total, used, free, sin, sout) tuple.""" mem = cext.virtual_mem() total_phys = mem[0] total_system = mem[2] # system memory (commit total/limit) is the sum of physical and swap # thus physical memory values need to be subtracted to get swap values total = total_system - total_phys # commit total is incremented immediately (decrementing free_system) # while the corresponding free physical value is not decremented until # pages are accessed, so we can't use free system memory for swap. # instead, we calculate page file usage based on performance counter if total > 0: percentswap = cext.swap_percent() used = int(0.01 * percentswap * total) else: percentswap = 0.0 used = 0 free = total - used percent = round(percentswap, 1) return _common.sswap(total, used, free, percent, 0, 0) # ===================================================================== # --- disk # ===================================================================== disk_io_counters = cext.disk_io_counters def disk_usage(path): """Return disk usage associated with path.""" if isinstance(path, bytes): # XXX: do we want to use "strict"? Probably yes, in order # to fail immediately. After all we are accepting input here... path = path.decode(ENCODING, errors="strict") total, free = cext.disk_usage(path) used = total - free percent = usage_percent(used, total, round_=1) return _common.sdiskusage(total, used, free, percent) def disk_partitions(all): """Return disk partitions.""" rawlist = cext.disk_partitions(all) return [_common.sdiskpart(*x) for x in rawlist] # ===================================================================== # --- CPU # ===================================================================== def cpu_times(): """Return system CPU times as a named tuple.""" user, system, idle = cext.cpu_times() # Internally, GetSystemTimes() is used, and it doesn't return # interrupt and dpc times. cext.per_cpu_times() does, so we # rely on it to get those only. percpu_summed = scputimes(*[sum(n) for n in zip(*cext.per_cpu_times())]) return scputimes( user, system, idle, percpu_summed.interrupt, percpu_summed.dpc ) def per_cpu_times(): """Return system per-CPU times as a list of named tuples.""" ret = [] for user, system, idle, interrupt, dpc in cext.per_cpu_times(): item = scputimes(user, system, idle, interrupt, dpc) ret.append(item) return ret def cpu_count_logical(): """Return the number of logical CPUs in the system.""" return cext.cpu_count_logical() def cpu_count_cores(): """Return the number of CPU cores in the system.""" return cext.cpu_count_cores() def cpu_stats(): """Return CPU statistics.""" ctx_switches, interrupts, _dpcs, syscalls = cext.cpu_stats() soft_interrupts = 0 return _common.scpustats( ctx_switches, interrupts, soft_interrupts, syscalls ) def cpu_freq(): """Return CPU frequency. On Windows per-cpu frequency is not supported. """ curr, max_ = cext.cpu_freq() min_ = 0.0 return [_common.scpufreq(float(curr), min_, float(max_))] _loadavg_inititialized = False def getloadavg(): """Return the number of processes in the system run queue averaged over the last 1, 5, and 15 minutes respectively as a tuple. """ global _loadavg_inititialized if not _loadavg_inititialized: cext.init_loadavg_counter() _loadavg_inititialized = True # Drop to 2 decimal points which is what Linux does raw_loads = cext.getloadavg() return tuple(round(load, 2) for load in raw_loads) # ===================================================================== # --- network # ===================================================================== def net_connections(kind, _pid=-1): """Return socket connections. If pid == -1 return system-wide connections (as opposed to connections opened by one process only). """ families, types = conn_tmap[kind] rawlist = cext.net_connections(_pid, families, types) ret = set() for item in rawlist: fd, fam, type, laddr, raddr, status, pid = item nt = conn_to_ntuple( fd, fam, type, laddr, raddr, status, TCP_STATUSES, pid=pid if _pid == -1 else None, ) ret.add(nt) return list(ret) def net_if_stats(): """Get NIC stats (isup, duplex, speed, mtu).""" ret = {} rawdict = cext.net_if_stats() for name, items in rawdict.items(): isup, duplex, speed, mtu = items if hasattr(_common, 'NicDuplex'): duplex = _common.NicDuplex(duplex) ret[name] = _common.snicstats(isup, duplex, speed, mtu, '') return ret def net_io_counters(): """Return network I/O statistics for every network interface installed on the system as a dict of raw tuples. """ return cext.net_io_counters() def net_if_addrs(): """Return the addresses associated to each NIC.""" return cext.net_if_addrs() # ===================================================================== # --- sensors # ===================================================================== def sensors_battery(): """Return battery information.""" # For constants meaning see: # https://msdn.microsoft.com/en-us/library/windows/desktop/ # aa373232(v=vs.85).aspx acline_status, flags, percent, secsleft = cext.sensors_battery() power_plugged = acline_status == 1 no_battery = bool(flags & 128) charging = bool(flags & 8) if no_battery: return None if power_plugged or charging: secsleft = _common.POWER_TIME_UNLIMITED elif secsleft == -1: secsleft = _common.POWER_TIME_UNKNOWN return _common.sbattery(percent, secsleft, power_plugged) # ===================================================================== # --- other system functions # ===================================================================== _last_btime = 0 def boot_time(): """The system boot time expressed in seconds since the epoch.""" # This dirty hack is to adjust the precision of the returned # value which may have a 1 second fluctuation, see: # https://github.com/giampaolo/psutil/issues/1007 global _last_btime ret = float(cext.boot_time()) if abs(ret - _last_btime) <= 1: return _last_btime else: _last_btime = ret return ret def users(): """Return currently connected users as a list of namedtuples.""" retlist = [] rawlist = cext.users() for item in rawlist: user, hostname, tstamp = item nt = _common.suser(user, None, hostname, tstamp, None) retlist.append(nt) return retlist # ===================================================================== # --- Windows services # ===================================================================== def win_service_iter(): """Yields a list of WindowsService instances.""" for name, display_name in cext.winservice_enumerate(): yield WindowsService(name, display_name) def win_service_get(name): """Open a Windows service and return it as a WindowsService instance.""" service = WindowsService(name, None) service._display_name = service._query_config()['display_name'] return service class WindowsService: # noqa: PLW1641 """Represents an installed Windows service.""" def __init__(self, name, display_name): self._name = name self._display_name = display_name def __str__(self): details = f"(name={self._name!r}, display_name={self._display_name!r})" return f"{self.__class__.__name__}{details}" def __repr__(self): return f"<{self.__str__()} at {id(self)}>" def __eq__(self, other): # Test for equality with another WindosService object based # on name. if not isinstance(other, WindowsService): return NotImplemented return self._name == other._name def __ne__(self, other): return not self == other def _query_config(self): with self._wrap_exceptions(): display_name, binpath, username, start_type = ( cext.winservice_query_config(self._name) ) # XXX - update _self.display_name? return dict( display_name=display_name, binpath=binpath, username=username, start_type=start_type, ) def _query_status(self): with self._wrap_exceptions(): status, pid = cext.winservice_query_status(self._name) if pid == 0: pid = None return dict(status=status, pid=pid) @contextlib.contextmanager def _wrap_exceptions(self): """Ctx manager which translates bare OSError and WindowsError exceptions into NoSuchProcess and AccessDenied. """ try: yield except OSError as err: name = self._name if is_permission_err(err): msg = ( f"service {name!r} is not querable (not enough privileges)" ) raise AccessDenied(pid=None, name=name, msg=msg) from err elif err.winerror in { cext.ERROR_INVALID_NAME, cext.ERROR_SERVICE_DOES_NOT_EXIST, }: msg = f"service {name!r} does not exist" raise NoSuchProcess(pid=None, name=name, msg=msg) from err else: raise # config query def name(self): """The service name. This string is how a service is referenced and can be passed to win_service_get() to get a new WindowsService instance. """ return self._name def display_name(self): """The service display name. The value is cached when this class is instantiated. """ return self._display_name def binpath(self): """The fully qualified path to the service binary/exe file as a string, including command line arguments. """ return self._query_config()['binpath'] def username(self): """The name of the user that owns this service.""" return self._query_config()['username'] def start_type(self): """A string which can either be "automatic", "manual" or "disabled". """ return self._query_config()['start_type'] # status query def pid(self): """The process PID, if any, else None. This can be passed to Process class to control the service's process. """ return self._query_status()['pid'] def status(self): """Service status as a string.""" return self._query_status()['status'] def description(self): """Service long description.""" return cext.winservice_query_descr(self.name()) # utils def as_dict(self): """Utility method retrieving all the information above as a dictionary. """ d = self._query_config() d.update(self._query_status()) d['name'] = self.name() d['display_name'] = self.display_name() d['description'] = self.description() return d # actions # XXX: the necessary C bindings for start() and stop() are # implemented but for now I prefer not to expose them. # I may change my mind in the future. Reasons: # - they require Administrator privileges # - can't implement a timeout for stop() (unless by using a thread, # which sucks) # - would require adding ServiceAlreadyStarted and # ServiceAlreadyStopped exceptions, adding two new APIs. # - we might also want to have modify(), which would basically mean # rewriting win32serviceutil.ChangeServiceConfig, which involves a # lot of stuff (and API constants which would pollute the API), see: # http://pyxr.sourceforge.net/PyXR/c/python24/lib/site-packages/ # win32/lib/win32serviceutil.py.html#0175 # - psutil is typically about "read only" monitoring stuff; # win_service_* APIs should only be used to retrieve a service and # check whether it's running # def start(self, timeout=None): # with self._wrap_exceptions(): # cext.winservice_start(self.name()) # if timeout: # giveup_at = time.time() + timeout # while True: # if self.status() == "running": # return # else: # if time.time() > giveup_at: # raise TimeoutExpired(timeout) # else: # time.sleep(.1) # def stop(self): # # Note: timeout is not implemented because it's just not # # possible, see: # # http://stackoverflow.com/questions/11973228/ # with self._wrap_exceptions(): # return cext.winservice_stop(self.name()) # ===================================================================== # --- processes # ===================================================================== pids = cext.pids pid_exists = cext.pid_exists ppid_map = cext.ppid_map # used internally by Process.children() def is_permission_err(exc): """Return True if this is a permission error.""" assert isinstance(exc, OSError), exc return isinstance(exc, PermissionError) or exc.winerror in { cext.ERROR_ACCESS_DENIED, cext.ERROR_PRIVILEGE_NOT_HELD, } def convert_oserror(exc, pid=None, name=None): """Convert OSError into NoSuchProcess or AccessDenied.""" assert isinstance(exc, OSError), exc if is_permission_err(exc): return AccessDenied(pid=pid, name=name) if isinstance(exc, ProcessLookupError): return NoSuchProcess(pid=pid, name=name) raise exc def wrap_exceptions(fun): """Decorator which converts OSError into NoSuchProcess or AccessDenied.""" @functools.wraps(fun) def wrapper(self, *args, **kwargs): try: return fun(self, *args, **kwargs) except OSError as err: raise convert_oserror(err, pid=self.pid, name=self._name) from err return wrapper def retry_error_partial_copy(fun): """Workaround for https://github.com/giampaolo/psutil/issues/875. See: https://stackoverflow.com/questions/4457745#4457745. """ @functools.wraps(fun) def wrapper(self, *args, **kwargs): delay = 0.0001 times = 33 for _ in range(times): # retries for roughly 1 second try: return fun(self, *args, **kwargs) except OSError as _: err = _ if err.winerror == ERROR_PARTIAL_COPY: time.sleep(delay) delay = min(delay * 2, 0.04) continue raise msg = ( f"{fun} retried {times} times, converted to AccessDenied as it's " f"still returning {err}" ) raise AccessDenied(pid=self.pid, name=self._name, msg=msg) return wrapper class Process: """Wrapper class around underlying C implementation.""" __slots__ = ["_cache", "_name", "_ppid", "pid"] def __init__(self, pid): self.pid = pid self._name = None self._ppid = None # --- oneshot() stuff def oneshot_enter(self): self._proc_info.cache_activate(self) self.exe.cache_activate(self) def oneshot_exit(self): self._proc_info.cache_deactivate(self) self.exe.cache_deactivate(self) @memoize_when_activated def _proc_info(self): """Return multiple information about this process as a raw tuple. """ ret = cext.proc_info(self.pid) assert len(ret) == len(pinfo_map) return ret def name(self): """Return process name, which on Windows is always the final part of the executable. """ # This is how PIDs 0 and 4 are always represented in taskmgr # and process-hacker. if self.pid == 0: return "System Idle Process" if self.pid == 4: return "System" return os.path.basename(self.exe()) @wrap_exceptions @memoize_when_activated def exe(self): if PYPY: try: exe = cext.proc_exe(self.pid) except OSError as err: # 24 = ERROR_TOO_MANY_OPEN_FILES. Not sure why this happens # (perhaps PyPy's JIT delaying garbage collection of files?). if err.errno == 24: debug(f"{err!r} translated into AccessDenied") raise AccessDenied(self.pid, self._name) from err raise else: exe = cext.proc_exe(self.pid) if exe.startswith('\\'): return convert_dos_path(exe) return exe # May be "Registry", "MemCompression", ... @wrap_exceptions @retry_error_partial_copy def cmdline(self): if cext.WINVER >= cext.WINDOWS_8_1: # PEB method detects cmdline changes but requires more # privileges: https://github.com/giampaolo/psutil/pull/1398 try: return cext.proc_cmdline(self.pid, use_peb=True) except OSError as err: if is_permission_err(err): return cext.proc_cmdline(self.pid, use_peb=False) else: raise else: return cext.proc_cmdline(self.pid, use_peb=True) @wrap_exceptions @retry_error_partial_copy def environ(self): s = cext.proc_environ(self.pid) return parse_environ_block(s) def ppid(self): try: return ppid_map()[self.pid] except KeyError: raise NoSuchProcess(self.pid, self._name) from None def _get_raw_meminfo(self): try: return cext.proc_memory_info(self.pid) except OSError as err: if is_permission_err(err): # TODO: the C ext can probably be refactored in order # to get this from cext.proc_info() debug("attempting memory_info() fallback (slower)") info = self._proc_info() return ( info[pinfo_map['num_page_faults']], info[pinfo_map['peak_wset']], info[pinfo_map['wset']], info[pinfo_map['peak_paged_pool']], info[pinfo_map['paged_pool']], info[pinfo_map['peak_non_paged_pool']], info[pinfo_map['non_paged_pool']], info[pinfo_map['pagefile']], info[pinfo_map['peak_pagefile']], info[pinfo_map['mem_private']], ) raise @wrap_exceptions def memory_info(self): # on Windows RSS == WorkingSetSize and VSM == PagefileUsage. # Underlying C function returns fields of PROCESS_MEMORY_COUNTERS # struct. t = self._get_raw_meminfo() rss = t[2] # wset vms = t[7] # pagefile return pmem(*(rss, vms) + t) @wrap_exceptions def memory_full_info(self): basic_mem = self.memory_info() uss = cext.proc_memory_uss(self.pid) uss *= getpagesize() return pfullmem(*basic_mem + (uss,)) def memory_maps(self): try: raw = cext.proc_memory_maps(self.pid) except OSError as err: # XXX - can't use wrap_exceptions decorator as we're # returning a generator; probably needs refactoring. raise convert_oserror(err, self.pid, self._name) from err else: for addr, perm, path, rss in raw: path = convert_dos_path(path) addr = hex(addr) yield (addr, perm, path, rss) @wrap_exceptions def kill(self): return cext.proc_kill(self.pid) @wrap_exceptions def send_signal(self, sig): if sig == signal.SIGTERM: cext.proc_kill(self.pid) elif sig in {signal.CTRL_C_EVENT, signal.CTRL_BREAK_EVENT}: os.kill(self.pid, sig) else: msg = ( "only SIGTERM, CTRL_C_EVENT and CTRL_BREAK_EVENT signals " "are supported on Windows" ) raise ValueError(msg) @wrap_exceptions def wait(self, timeout=None): if timeout is None: cext_timeout = cext.INFINITE else: # WaitForSingleObject() expects time in milliseconds. cext_timeout = int(timeout * 1000) timer = getattr(time, 'monotonic', time.time) stop_at = timer() + timeout if timeout is not None else None try: # Exit code is supposed to come from GetExitCodeProcess(). # May also be None if OpenProcess() failed with # ERROR_INVALID_PARAMETER, meaning PID is already gone. exit_code = cext.proc_wait(self.pid, cext_timeout) except cext.TimeoutExpired as err: # WaitForSingleObject() returned WAIT_TIMEOUT. Just raise. raise TimeoutExpired(timeout, self.pid, self._name) from err except cext.TimeoutAbandoned: # WaitForSingleObject() returned WAIT_ABANDONED, see: # https://github.com/giampaolo/psutil/issues/1224 # We'll just rely on the internal polling and return None # when the PID disappears. Subprocess module does the same # (return None): # https://github.com/python/cpython/blob/ # be50a7b627d0aa37e08fa8e2d5568891f19903ce/ # Lib/subprocess.py#L1193-L1194 exit_code = None # At this point WaitForSingleObject() returned WAIT_OBJECT_0, # meaning the process is gone. Stupidly there are cases where # its PID may still stick around so we do a further internal # polling. delay = 0.0001 while True: if not pid_exists(self.pid): return exit_code if stop_at and timer() >= stop_at: raise TimeoutExpired(timeout, pid=self.pid, name=self._name) time.sleep(delay) delay = min(delay * 2, 0.04) # incremental delay @wrap_exceptions def username(self): if self.pid in {0, 4}: return 'NT AUTHORITY\\SYSTEM' domain, user = cext.proc_username(self.pid) return f"{domain}\\{user}" @wrap_exceptions def create_time(self, fast_only=False): # Note: proc_times() not put under oneshot() 'cause create_time() # is already cached by the main Process class. try: _user, _system, created = cext.proc_times(self.pid) return created except OSError as err: if is_permission_err(err): if fast_only: raise debug("attempting create_time() fallback (slower)") return self._proc_info()[pinfo_map['create_time']] raise @wrap_exceptions def num_threads(self): return self._proc_info()[pinfo_map['num_threads']] @wrap_exceptions def threads(self): rawlist = cext.proc_threads(self.pid) retlist = [] for thread_id, utime, stime in rawlist: ntuple = _common.pthread(thread_id, utime, stime) retlist.append(ntuple) return retlist @wrap_exceptions def cpu_times(self): try: user, system, _created = cext.proc_times(self.pid) except OSError as err: if not is_permission_err(err): raise debug("attempting cpu_times() fallback (slower)") info = self._proc_info() user = info[pinfo_map['user_time']] system = info[pinfo_map['kernel_time']] # Children user/system times are not retrievable (set to 0). return _common.pcputimes(user, system, 0.0, 0.0) @wrap_exceptions def suspend(self): cext.proc_suspend_or_resume(self.pid, True) @wrap_exceptions def resume(self): cext.proc_suspend_or_resume(self.pid, False) @wrap_exceptions @retry_error_partial_copy def cwd(self): if self.pid in {0, 4}: raise AccessDenied(self.pid, self._name) # return a normalized pathname since the native C function appends # "\\" at the and of the path path = cext.proc_cwd(self.pid) return os.path.normpath(path) @wrap_exceptions def open_files(self): if self.pid in {0, 4}: return [] ret = set() # Filenames come in in native format like: # "\Device\HarddiskVolume1\Windows\systemew\file.txt" # Convert the first part in the corresponding drive letter # (e.g. "C:\") by using Windows's QueryDosDevice() raw_file_names = cext.proc_open_files(self.pid) for file in raw_file_names: file = convert_dos_path(file) if isfile_strict(file): ntuple = _common.popenfile(file, -1) ret.add(ntuple) return list(ret) @wrap_exceptions def net_connections(self, kind='inet'): return net_connections(kind, _pid=self.pid) @wrap_exceptions def nice_get(self): value = cext.proc_priority_get(self.pid) value = Priority(value) return value @wrap_exceptions def nice_set(self, value): return cext.proc_priority_set(self.pid, value) @wrap_exceptions def ionice_get(self): ret = cext.proc_io_priority_get(self.pid) ret = IOPriority(ret) return ret @wrap_exceptions def ionice_set(self, ioclass, value): if value: msg = "value argument not accepted on Windows" raise TypeError(msg) if ioclass not in { IOPriority.IOPRIO_VERYLOW, IOPriority.IOPRIO_LOW, IOPriority.IOPRIO_NORMAL, IOPriority.IOPRIO_HIGH, }: msg = f"{ioclass} is not a valid priority" raise ValueError(msg) cext.proc_io_priority_set(self.pid, ioclass) @wrap_exceptions def io_counters(self): try: ret = cext.proc_io_counters(self.pid) except OSError as err: if not is_permission_err(err): raise debug("attempting io_counters() fallback (slower)") info = self._proc_info() ret = ( info[pinfo_map['io_rcount']], info[pinfo_map['io_wcount']], info[pinfo_map['io_rbytes']], info[pinfo_map['io_wbytes']], info[pinfo_map['io_count_others']], info[pinfo_map['io_bytes_others']], ) return pio(*ret) @wrap_exceptions def status(self): suspended = cext.proc_is_suspended(self.pid) if suspended: return _common.STATUS_STOPPED else: return _common.STATUS_RUNNING @wrap_exceptions def cpu_affinity_get(self): def from_bitmask(x): return [i for i in range(64) if (1 << i) & x] bitmask = cext.proc_cpu_affinity_get(self.pid) return from_bitmask(bitmask) @wrap_exceptions def cpu_affinity_set(self, value): def to_bitmask(ls): if not ls: msg = f"invalid argument {ls!r}" raise ValueError(msg) out = 0 for b in ls: out |= 2**b return out # SetProcessAffinityMask() states that ERROR_INVALID_PARAMETER # is returned for an invalid CPU but this seems not to be true, # therefore we check CPUs validy beforehand. allcpus = list(range(len(per_cpu_times()))) for cpu in value: if cpu not in allcpus: if not isinstance(cpu, int): msg = f"invalid CPU {cpu!r}; an integer is required" raise TypeError(msg) msg = f"invalid CPU {cpu!r}" raise ValueError(msg) bitmask = to_bitmask(value) cext.proc_cpu_affinity_set(self.pid, bitmask) @wrap_exceptions def num_handles(self): try: return cext.proc_num_handles(self.pid) except OSError as err: if is_permission_err(err): debug("attempting num_handles() fallback (slower)") return self._proc_info()[pinfo_map['num_handles']] raise @wrap_exceptions def num_ctx_switches(self): ctx_switches = self._proc_info()[pinfo_map['ctx_switches']] # only voluntary ctx switches are supported return _common.pctxsw(ctx_switches, 0) _common.py 0000644 00000067742 15030211765 0006565 0 ustar 00 # Copyright (c) 2009, Giampaolo Rodola'. All rights reserved. # Use of this source code is governed by a BSD-style license that can be # found in the LICENSE file. """Common objects shared by __init__.py and _ps*.py modules. Note: this module is imported by setup.py, so it should not import psutil or third-party modules. """ import collections import enum import functools import os import socket import stat import sys import threading import warnings from collections import namedtuple from socket import AF_INET from socket import SOCK_DGRAM from socket import SOCK_STREAM try: from socket import AF_INET6 except ImportError: AF_INET6 = None try: from socket import AF_UNIX except ImportError: AF_UNIX = None PSUTIL_DEBUG = bool(os.getenv('PSUTIL_DEBUG')) _DEFAULT = object() # fmt: off __all__ = [ # OS constants 'FREEBSD', 'BSD', 'LINUX', 'NETBSD', 'OPENBSD', 'MACOS', 'OSX', 'POSIX', 'SUNOS', 'WINDOWS', # connection constants 'CONN_CLOSE', 'CONN_CLOSE_WAIT', 'CONN_CLOSING', 'CONN_ESTABLISHED', 'CONN_FIN_WAIT1', 'CONN_FIN_WAIT2', 'CONN_LAST_ACK', 'CONN_LISTEN', 'CONN_NONE', 'CONN_SYN_RECV', 'CONN_SYN_SENT', 'CONN_TIME_WAIT', # net constants 'NIC_DUPLEX_FULL', 'NIC_DUPLEX_HALF', 'NIC_DUPLEX_UNKNOWN', # noqa: F822 # process status constants 'STATUS_DEAD', 'STATUS_DISK_SLEEP', 'STATUS_IDLE', 'STATUS_LOCKED', 'STATUS_RUNNING', 'STATUS_SLEEPING', 'STATUS_STOPPED', 'STATUS_SUSPENDED', 'STATUS_TRACING_STOP', 'STATUS_WAITING', 'STATUS_WAKE_KILL', 'STATUS_WAKING', 'STATUS_ZOMBIE', 'STATUS_PARKED', # other constants 'ENCODING', 'ENCODING_ERRS', 'AF_INET6', # named tuples 'pconn', 'pcputimes', 'pctxsw', 'pgids', 'pio', 'pionice', 'popenfile', 'pthread', 'puids', 'sconn', 'scpustats', 'sdiskio', 'sdiskpart', 'sdiskusage', 'snetio', 'snicaddr', 'snicstats', 'sswap', 'suser', # utility functions 'conn_tmap', 'deprecated_method', 'isfile_strict', 'memoize', 'parse_environ_block', 'path_exists_strict', 'usage_percent', 'supports_ipv6', 'sockfam_to_enum', 'socktype_to_enum', "wrap_numbers", 'open_text', 'open_binary', 'cat', 'bcat', 'bytes2human', 'conn_to_ntuple', 'debug', # shell utils 'hilite', 'term_supports_colors', 'print_color', ] # fmt: on # =================================================================== # --- OS constants # =================================================================== POSIX = os.name == "posix" WINDOWS = os.name == "nt" LINUX = sys.platform.startswith("linux") MACOS = sys.platform.startswith("darwin") OSX = MACOS # deprecated alias FREEBSD = sys.platform.startswith(("freebsd", "midnightbsd")) OPENBSD = sys.platform.startswith("openbsd") NETBSD = sys.platform.startswith("netbsd") BSD = FREEBSD or OPENBSD or NETBSD SUNOS = sys.platform.startswith(("sunos", "solaris")) AIX = sys.platform.startswith("aix") # =================================================================== # --- API constants # =================================================================== # Process.status() STATUS_RUNNING = "running" STATUS_SLEEPING = "sleeping" STATUS_DISK_SLEEP = "disk-sleep" STATUS_STOPPED = "stopped" STATUS_TRACING_STOP = "tracing-stop" STATUS_ZOMBIE = "zombie" STATUS_DEAD = "dead" STATUS_WAKE_KILL = "wake-kill" STATUS_WAKING = "waking" STATUS_IDLE = "idle" # Linux, macOS, FreeBSD STATUS_LOCKED = "locked" # FreeBSD STATUS_WAITING = "waiting" # FreeBSD STATUS_SUSPENDED = "suspended" # NetBSD STATUS_PARKED = "parked" # Linux # Process.net_connections() and psutil.net_connections() CONN_ESTABLISHED = "ESTABLISHED" CONN_SYN_SENT = "SYN_SENT" CONN_SYN_RECV = "SYN_RECV" CONN_FIN_WAIT1 = "FIN_WAIT1" CONN_FIN_WAIT2 = "FIN_WAIT2" CONN_TIME_WAIT = "TIME_WAIT" CONN_CLOSE = "CLOSE" CONN_CLOSE_WAIT = "CLOSE_WAIT" CONN_LAST_ACK = "LAST_ACK" CONN_LISTEN = "LISTEN" CONN_CLOSING = "CLOSING" CONN_NONE = "NONE" # net_if_stats() class NicDuplex(enum.IntEnum): NIC_DUPLEX_FULL = 2 NIC_DUPLEX_HALF = 1 NIC_DUPLEX_UNKNOWN = 0 globals().update(NicDuplex.__members__) # sensors_battery() class BatteryTime(enum.IntEnum): POWER_TIME_UNKNOWN = -1 POWER_TIME_UNLIMITED = -2 globals().update(BatteryTime.__members__) # --- others ENCODING = sys.getfilesystemencoding() ENCODING_ERRS = sys.getfilesystemencodeerrors() # =================================================================== # --- namedtuples # =================================================================== # --- for system functions # fmt: off # psutil.swap_memory() sswap = namedtuple('sswap', ['total', 'used', 'free', 'percent', 'sin', 'sout']) # psutil.disk_usage() sdiskusage = namedtuple('sdiskusage', ['total', 'used', 'free', 'percent']) # psutil.disk_io_counters() sdiskio = namedtuple('sdiskio', ['read_count', 'write_count', 'read_bytes', 'write_bytes', 'read_time', 'write_time']) # psutil.disk_partitions() sdiskpart = namedtuple('sdiskpart', ['device', 'mountpoint', 'fstype', 'opts']) # psutil.net_io_counters() snetio = namedtuple('snetio', ['bytes_sent', 'bytes_recv', 'packets_sent', 'packets_recv', 'errin', 'errout', 'dropin', 'dropout']) # psutil.users() suser = namedtuple('suser', ['name', 'terminal', 'host', 'started', 'pid']) # psutil.net_connections() sconn = namedtuple('sconn', ['fd', 'family', 'type', 'laddr', 'raddr', 'status', 'pid']) # psutil.net_if_addrs() snicaddr = namedtuple('snicaddr', ['family', 'address', 'netmask', 'broadcast', 'ptp']) # psutil.net_if_stats() snicstats = namedtuple('snicstats', ['isup', 'duplex', 'speed', 'mtu', 'flags']) # psutil.cpu_stats() scpustats = namedtuple( 'scpustats', ['ctx_switches', 'interrupts', 'soft_interrupts', 'syscalls']) # psutil.cpu_freq() scpufreq = namedtuple('scpufreq', ['current', 'min', 'max']) # psutil.sensors_temperatures() shwtemp = namedtuple( 'shwtemp', ['label', 'current', 'high', 'critical']) # psutil.sensors_battery() sbattery = namedtuple('sbattery', ['percent', 'secsleft', 'power_plugged']) # psutil.sensors_fans() sfan = namedtuple('sfan', ['label', 'current']) # fmt: on # --- for Process methods # psutil.Process.cpu_times() pcputimes = namedtuple( 'pcputimes', ['user', 'system', 'children_user', 'children_system'] ) # psutil.Process.open_files() popenfile = namedtuple('popenfile', ['path', 'fd']) # psutil.Process.threads() pthread = namedtuple('pthread', ['id', 'user_time', 'system_time']) # psutil.Process.uids() puids = namedtuple('puids', ['real', 'effective', 'saved']) # psutil.Process.gids() pgids = namedtuple('pgids', ['real', 'effective', 'saved']) # psutil.Process.io_counters() pio = namedtuple( 'pio', ['read_count', 'write_count', 'read_bytes', 'write_bytes'] ) # psutil.Process.ionice() pionice = namedtuple('pionice', ['ioclass', 'value']) # psutil.Process.ctx_switches() pctxsw = namedtuple('pctxsw', ['voluntary', 'involuntary']) # psutil.Process.net_connections() pconn = namedtuple( 'pconn', ['fd', 'family', 'type', 'laddr', 'raddr', 'status'] ) # psutil.net_connections() and psutil.Process.net_connections() addr = namedtuple('addr', ['ip', 'port']) # =================================================================== # --- Process.net_connections() 'kind' parameter mapping # =================================================================== conn_tmap = { "all": ([AF_INET, AF_INET6, AF_UNIX], [SOCK_STREAM, SOCK_DGRAM]), "tcp": ([AF_INET, AF_INET6], [SOCK_STREAM]), "tcp4": ([AF_INET], [SOCK_STREAM]), "udp": ([AF_INET, AF_INET6], [SOCK_DGRAM]), "udp4": ([AF_INET], [SOCK_DGRAM]), "inet": ([AF_INET, AF_INET6], [SOCK_STREAM, SOCK_DGRAM]), "inet4": ([AF_INET], [SOCK_STREAM, SOCK_DGRAM]), "inet6": ([AF_INET6], [SOCK_STREAM, SOCK_DGRAM]), } if AF_INET6 is not None: conn_tmap.update({ "tcp6": ([AF_INET6], [SOCK_STREAM]), "udp6": ([AF_INET6], [SOCK_DGRAM]), }) if AF_UNIX is not None and not SUNOS: conn_tmap.update({"unix": ([AF_UNIX], [SOCK_STREAM, SOCK_DGRAM])}) # ===================================================================== # --- Exceptions # ===================================================================== class Error(Exception): """Base exception class. All other psutil exceptions inherit from this one. """ __module__ = 'psutil' def _infodict(self, attrs): info = collections.OrderedDict() for name in attrs: value = getattr(self, name, None) if value or (name == "pid" and value == 0): info[name] = value return info def __str__(self): # invoked on `raise Error` info = self._infodict(("pid", "ppid", "name")) if info: details = "({})".format( ", ".join([f"{k}={v!r}" for k, v in info.items()]) ) else: details = None return " ".join([x for x in (getattr(self, "msg", ""), details) if x]) def __repr__(self): # invoked on `repr(Error)` info = self._infodict(("pid", "ppid", "name", "seconds", "msg")) details = ", ".join([f"{k}={v!r}" for k, v in info.items()]) return f"psutil.{self.__class__.__name__}({details})" class NoSuchProcess(Error): """Exception raised when a process with a certain PID doesn't or no longer exists. """ __module__ = 'psutil' def __init__(self, pid, name=None, msg=None): Error.__init__(self) self.pid = pid self.name = name self.msg = msg or "process no longer exists" def __reduce__(self): return (self.__class__, (self.pid, self.name, self.msg)) class ZombieProcess(NoSuchProcess): """Exception raised when querying a zombie process. This is raised on macOS, BSD and Solaris only, and not always: depending on the query the OS may be able to succeed anyway. On Linux all zombie processes are querable (hence this is never raised). Windows doesn't have zombie processes. """ __module__ = 'psutil' def __init__(self, pid, name=None, ppid=None, msg=None): NoSuchProcess.__init__(self, pid, name, msg) self.ppid = ppid self.msg = msg or "PID still exists but it's a zombie" def __reduce__(self): return (self.__class__, (self.pid, self.name, self.ppid, self.msg)) class AccessDenied(Error): """Exception raised when permission to perform an action is denied.""" __module__ = 'psutil' def __init__(self, pid=None, name=None, msg=None): Error.__init__(self) self.pid = pid self.name = name self.msg = msg or "" def __reduce__(self): return (self.__class__, (self.pid, self.name, self.msg)) class TimeoutExpired(Error): """Raised on Process.wait(timeout) if timeout expires and process is still alive. """ __module__ = 'psutil' def __init__(self, seconds, pid=None, name=None): Error.__init__(self) self.seconds = seconds self.pid = pid self.name = name self.msg = f"timeout after {seconds} seconds" def __reduce__(self): return (self.__class__, (self.seconds, self.pid, self.name)) # =================================================================== # --- utils # =================================================================== def usage_percent(used, total, round_=None): """Calculate percentage usage of 'used' against 'total'.""" try: ret = (float(used) / total) * 100 except ZeroDivisionError: return 0.0 else: if round_ is not None: ret = round(ret, round_) return ret def memoize(fun): """A simple memoize decorator for functions supporting (hashable) positional arguments. It also provides a cache_clear() function for clearing the cache: >>> @memoize ... def foo() ... return 1 ... >>> foo() 1 >>> foo.cache_clear() >>> It supports: - functions - classes (acts as a @singleton) - staticmethods - classmethods It does NOT support: - methods """ @functools.wraps(fun) def wrapper(*args, **kwargs): key = (args, frozenset(sorted(kwargs.items()))) try: return cache[key] except KeyError: try: ret = cache[key] = fun(*args, **kwargs) except Exception as err: # noqa: BLE001 raise err from None return ret def cache_clear(): """Clear cache.""" cache.clear() cache = {} wrapper.cache_clear = cache_clear return wrapper def memoize_when_activated(fun): """A memoize decorator which is disabled by default. It can be activated and deactivated on request. For efficiency reasons it can be used only against class methods accepting no arguments. >>> class Foo: ... @memoize ... def foo() ... print(1) ... >>> f = Foo() >>> # deactivated (default) >>> foo() 1 >>> foo() 1 >>> >>> # activated >>> foo.cache_activate(self) >>> foo() 1 >>> foo() >>> foo() >>> """ @functools.wraps(fun) def wrapper(self): try: # case 1: we previously entered oneshot() ctx ret = self._cache[fun] except AttributeError: # case 2: we never entered oneshot() ctx try: return fun(self) except Exception as err: # noqa: BLE001 raise err from None except KeyError: # case 3: we entered oneshot() ctx but there's no cache # for this entry yet try: ret = fun(self) except Exception as err: # noqa: BLE001 raise err from None try: self._cache[fun] = ret except AttributeError: # multi-threading race condition, see: # https://github.com/giampaolo/psutil/issues/1948 pass return ret def cache_activate(proc): """Activate cache. Expects a Process instance. Cache will be stored as a "_cache" instance attribute. """ proc._cache = {} def cache_deactivate(proc): """Deactivate and clear cache.""" try: del proc._cache except AttributeError: pass wrapper.cache_activate = cache_activate wrapper.cache_deactivate = cache_deactivate return wrapper def isfile_strict(path): """Same as os.path.isfile() but does not swallow EACCES / EPERM exceptions, see: http://mail.python.org/pipermail/python-dev/2012-June/120787.html. """ try: st = os.stat(path) except PermissionError: raise except OSError: return False else: return stat.S_ISREG(st.st_mode) def path_exists_strict(path): """Same as os.path.exists() but does not swallow EACCES / EPERM exceptions. See: http://mail.python.org/pipermail/python-dev/2012-June/120787.html. """ try: os.stat(path) except PermissionError: raise except OSError: return False else: return True @memoize def supports_ipv6(): """Return True if IPv6 is supported on this platform.""" if not socket.has_ipv6 or AF_INET6 is None: return False try: with socket.socket(AF_INET6, socket.SOCK_STREAM) as sock: sock.bind(("::1", 0)) return True except OSError: return False def parse_environ_block(data): """Parse a C environ block of environment variables into a dictionary.""" # The block is usually raw data from the target process. It might contain # trailing garbage and lines that do not look like assignments. ret = {} pos = 0 # localize global variable to speed up access. WINDOWS_ = WINDOWS while True: next_pos = data.find("\0", pos) # nul byte at the beginning or double nul byte means finish if next_pos <= pos: break # there might not be an equals sign equal_pos = data.find("=", pos, next_pos) if equal_pos > pos: key = data[pos:equal_pos] value = data[equal_pos + 1 : next_pos] # Windows expects environment variables to be uppercase only if WINDOWS_: key = key.upper() ret[key] = value pos = next_pos + 1 return ret def sockfam_to_enum(num): """Convert a numeric socket family value to an IntEnum member. If it's not a known member, return the numeric value itself. """ try: return socket.AddressFamily(num) except ValueError: return num def socktype_to_enum(num): """Convert a numeric socket type value to an IntEnum member. If it's not a known member, return the numeric value itself. """ try: return socket.SocketKind(num) except ValueError: return num def conn_to_ntuple(fd, fam, type_, laddr, raddr, status, status_map, pid=None): """Convert a raw connection tuple to a proper ntuple.""" if fam in {socket.AF_INET, AF_INET6}: if laddr: laddr = addr(*laddr) if raddr: raddr = addr(*raddr) if type_ == socket.SOCK_STREAM and fam in {AF_INET, AF_INET6}: status = status_map.get(status, CONN_NONE) else: status = CONN_NONE # ignore whatever C returned to us fam = sockfam_to_enum(fam) type_ = socktype_to_enum(type_) if pid is None: return pconn(fd, fam, type_, laddr, raddr, status) else: return sconn(fd, fam, type_, laddr, raddr, status, pid) def broadcast_addr(addr): """Given the address ntuple returned by ``net_if_addrs()`` calculates the broadcast address. """ import ipaddress if not addr.address or not addr.netmask: return None if addr.family == socket.AF_INET: return str( ipaddress.IPv4Network( f"{addr.address}/{addr.netmask}", strict=False ).broadcast_address ) if addr.family == socket.AF_INET6: return str( ipaddress.IPv6Network( f"{addr.address}/{addr.netmask}", strict=False ).broadcast_address ) def deprecated_method(replacement): """A decorator which can be used to mark a method as deprecated 'replcement' is the method name which will be called instead. """ def outer(fun): msg = ( f"{fun.__name__}() is deprecated and will be removed; use" f" {replacement}() instead" ) if fun.__doc__ is None: fun.__doc__ = msg @functools.wraps(fun) def inner(self, *args, **kwargs): warnings.warn(msg, category=DeprecationWarning, stacklevel=2) return getattr(self, replacement)(*args, **kwargs) return inner return outer class _WrapNumbers: """Watches numbers so that they don't overflow and wrap (reset to zero). """ def __init__(self): self.lock = threading.Lock() self.cache = {} self.reminders = {} self.reminder_keys = {} def _add_dict(self, input_dict, name): assert name not in self.cache assert name not in self.reminders assert name not in self.reminder_keys self.cache[name] = input_dict self.reminders[name] = collections.defaultdict(int) self.reminder_keys[name] = collections.defaultdict(set) def _remove_dead_reminders(self, input_dict, name): """In case the number of keys changed between calls (e.g. a disk disappears) this removes the entry from self.reminders. """ old_dict = self.cache[name] gone_keys = set(old_dict.keys()) - set(input_dict.keys()) for gone_key in gone_keys: for remkey in self.reminder_keys[name][gone_key]: del self.reminders[name][remkey] del self.reminder_keys[name][gone_key] def run(self, input_dict, name): """Cache dict and sum numbers which overflow and wrap. Return an updated copy of `input_dict`. """ if name not in self.cache: # This was the first call. self._add_dict(input_dict, name) return input_dict self._remove_dead_reminders(input_dict, name) old_dict = self.cache[name] new_dict = {} for key in input_dict: input_tuple = input_dict[key] try: old_tuple = old_dict[key] except KeyError: # The input dict has a new key (e.g. a new disk or NIC) # which didn't exist in the previous call. new_dict[key] = input_tuple continue bits = [] for i in range(len(input_tuple)): input_value = input_tuple[i] old_value = old_tuple[i] remkey = (key, i) if input_value < old_value: # it wrapped! self.reminders[name][remkey] += old_value self.reminder_keys[name][key].add(remkey) bits.append(input_value + self.reminders[name][remkey]) new_dict[key] = tuple(bits) self.cache[name] = input_dict return new_dict def cache_clear(self, name=None): """Clear the internal cache, optionally only for function 'name'.""" with self.lock: if name is None: self.cache.clear() self.reminders.clear() self.reminder_keys.clear() else: self.cache.pop(name, None) self.reminders.pop(name, None) self.reminder_keys.pop(name, None) def cache_info(self): """Return internal cache dicts as a tuple of 3 elements.""" with self.lock: return (self.cache, self.reminders, self.reminder_keys) def wrap_numbers(input_dict, name): """Given an `input_dict` and a function `name`, adjust the numbers which "wrap" (restart from zero) across different calls by adding "old value" to "new value" and return an updated dict. """ with _wn.lock: return _wn.run(input_dict, name) _wn = _WrapNumbers() wrap_numbers.cache_clear = _wn.cache_clear wrap_numbers.cache_info = _wn.cache_info # The read buffer size for open() builtin. This (also) dictates how # much data we read(2) when iterating over file lines as in: # >>> with open(file) as f: # ... for line in f: # ... ... # Default per-line buffer size for binary files is 1K. For text files # is 8K. We use a bigger buffer (32K) in order to have more consistent # results when reading /proc pseudo files on Linux, see: # https://github.com/giampaolo/psutil/issues/2050 # https://github.com/giampaolo/psutil/issues/708 FILE_READ_BUFFER_SIZE = 32 * 1024 def open_binary(fname): return open(fname, "rb", buffering=FILE_READ_BUFFER_SIZE) def open_text(fname): """Open a file in text mode by using the proper FS encoding and en/decoding error handlers. """ # See: # https://github.com/giampaolo/psutil/issues/675 # https://github.com/giampaolo/psutil/pull/733 fobj = open( # noqa: SIM115 fname, buffering=FILE_READ_BUFFER_SIZE, encoding=ENCODING, errors=ENCODING_ERRS, ) try: # Dictates per-line read(2) buffer size. Defaults is 8k. See: # https://github.com/giampaolo/psutil/issues/2050#issuecomment-1013387546 fobj._CHUNK_SIZE = FILE_READ_BUFFER_SIZE except AttributeError: pass except Exception: fobj.close() raise return fobj def cat(fname, fallback=_DEFAULT, _open=open_text): """Read entire file content and return it as a string. File is opened in text mode. If specified, `fallback` is the value returned in case of error, either if the file does not exist or it can't be read(). """ if fallback is _DEFAULT: with _open(fname) as f: return f.read() else: try: with _open(fname) as f: return f.read() except OSError: return fallback def bcat(fname, fallback=_DEFAULT): """Same as above but opens file in binary mode.""" return cat(fname, fallback=fallback, _open=open_binary) def bytes2human(n, format="%(value).1f%(symbol)s"): """Used by various scripts. See: https://code.activestate.com/recipes/578019-bytes-to-human-human-to-bytes-converter/?in=user-4178764. >>> bytes2human(10000) '9.8K' >>> bytes2human(100001221) '95.4M' """ symbols = ('B', 'K', 'M', 'G', 'T', 'P', 'E', 'Z', 'Y') prefix = {} for i, s in enumerate(symbols[1:]): prefix[s] = 1 << (i + 1) * 10 for symbol in reversed(symbols[1:]): if abs(n) >= prefix[symbol]: value = float(n) / prefix[symbol] return format % locals() return format % dict(symbol=symbols[0], value=n) def get_procfs_path(): """Return updated psutil.PROCFS_PATH constant.""" return sys.modules['psutil'].PROCFS_PATH def decode(s): return s.decode(encoding=ENCODING, errors=ENCODING_ERRS) # ===================================================================== # --- shell utils # ===================================================================== @memoize def term_supports_colors(file=sys.stdout): # pragma: no cover if os.name == 'nt': return True try: import curses assert file.isatty() curses.setupterm() assert curses.tigetnum("colors") > 0 except Exception: # noqa: BLE001 return False else: return True def hilite(s, color=None, bold=False): # pragma: no cover """Return an highlighted version of 'string'.""" if not term_supports_colors(): return s attr = [] colors = dict( blue='34', brown='33', darkgrey='30', green='32', grey='37', lightblue='36', red='91', violet='35', yellow='93', ) colors[None] = '29' try: color = colors[color] except KeyError: msg = f"invalid color {color!r}; choose amongst {list(colors.keys())}" raise ValueError(msg) from None attr.append(color) if bold: attr.append('1') return f"\x1b[{';'.join(attr)}m{s}\x1b[0m" def print_color( s, color=None, bold=False, file=sys.stdout ): # pragma: no cover """Print a colorized version of string.""" if not term_supports_colors(): print(s, file=file) elif POSIX: print(hilite(s, color, bold), file=file) else: import ctypes DEFAULT_COLOR = 7 GetStdHandle = ctypes.windll.Kernel32.GetStdHandle SetConsoleTextAttribute = ( ctypes.windll.Kernel32.SetConsoleTextAttribute ) colors = dict(green=2, red=4, brown=6, yellow=6) colors[None] = DEFAULT_COLOR try: color = colors[color] except KeyError: msg = ( f"invalid color {color!r}; choose between" f" {list(colors.keys())!r}" ) raise ValueError(msg) from None if bold and color <= 7: color += 8 handle_id = -12 if file is sys.stderr else -11 GetStdHandle.restype = ctypes.c_ulong handle = GetStdHandle(handle_id) SetConsoleTextAttribute(handle, color) try: print(s, file=file) finally: SetConsoleTextAttribute(handle, DEFAULT_COLOR) def debug(msg): """If PSUTIL_DEBUG env var is set, print a debug message to stderr.""" if PSUTIL_DEBUG: import inspect fname, lineno, _, _lines, _index = inspect.getframeinfo( inspect.currentframe().f_back ) if isinstance(msg, Exception): if isinstance(msg, OSError): # ...because str(exc) may contain info about the file name msg = f"ignoring {msg}" else: msg = f"ignoring {msg!r}" print( # noqa: T201 f"psutil-debug [{fname}:{lineno}]> {msg}", file=sys.stderr ) _psposix.py 0000644 00000015746 15030211765 0006777 0 ustar 00 # Copyright (c) 2009, Giampaolo Rodola'. All rights reserved. # Use of this source code is governed by a BSD-style license that can be # found in the LICENSE file. """Routines common to all posix systems.""" import enum import glob import os import signal import time from ._common import MACOS from ._common import TimeoutExpired from ._common import memoize from ._common import sdiskusage from ._common import usage_percent if MACOS: from . import _psutil_osx __all__ = ['pid_exists', 'wait_pid', 'disk_usage', 'get_terminal_map'] def pid_exists(pid): """Check whether pid exists in the current process table.""" if pid == 0: # According to "man 2 kill" PID 0 has a special meaning: # it refers to <<every process in the process group of the # calling process>> so we don't want to go any further. # If we get here it means this UNIX platform *does* have # a process with id 0. return True try: os.kill(pid, 0) except ProcessLookupError: return False except PermissionError: # EPERM clearly means there's a process to deny access to return True # According to "man 2 kill" possible error values are # (EINVAL, EPERM, ESRCH) else: return True Negsignal = enum.IntEnum( 'Negsignal', {x.name: -x.value for x in signal.Signals} ) def negsig_to_enum(num): """Convert a negative signal value to an enum.""" try: return Negsignal(num) except ValueError: return num def wait_pid( pid, timeout=None, proc_name=None, _waitpid=os.waitpid, _timer=getattr(time, 'monotonic', time.time), # noqa: B008 _min=min, _sleep=time.sleep, _pid_exists=pid_exists, ): """Wait for a process PID to terminate. If the process terminated normally by calling exit(3) or _exit(2), or by returning from main(), the return value is the positive integer passed to *exit(). If it was terminated by a signal it returns the negated value of the signal which caused the termination (e.g. -SIGTERM). If PID is not a children of os.getpid() (current process) just wait until the process disappears and return None. If PID does not exist at all return None immediately. If *timeout* != None and process is still alive raise TimeoutExpired. timeout=0 is also possible (either return immediately or raise). """ if pid <= 0: # see "man waitpid" msg = "can't wait for PID 0" raise ValueError(msg) interval = 0.0001 flags = 0 if timeout is not None: flags |= os.WNOHANG stop_at = _timer() + timeout def sleep(interval): # Sleep for some time and return a new increased interval. if timeout is not None: if _timer() >= stop_at: raise TimeoutExpired(timeout, pid=pid, name=proc_name) _sleep(interval) return _min(interval * 2, 0.04) # See: https://linux.die.net/man/2/waitpid while True: try: retpid, status = os.waitpid(pid, flags) except InterruptedError: interval = sleep(interval) except ChildProcessError: # This has two meanings: # - PID is not a child of os.getpid() in which case # we keep polling until it's gone # - PID never existed in the first place # In both cases we'll eventually return None as we # can't determine its exit status code. while _pid_exists(pid): interval = sleep(interval) return None else: if retpid == 0: # WNOHANG flag was used and PID is still running. interval = sleep(interval) continue if os.WIFEXITED(status): # Process terminated normally by calling exit(3) or _exit(2), # or by returning from main(). The return value is the # positive integer passed to *exit(). return os.WEXITSTATUS(status) elif os.WIFSIGNALED(status): # Process exited due to a signal. Return the negative value # of that signal. return negsig_to_enum(-os.WTERMSIG(status)) # elif os.WIFSTOPPED(status): # # Process was stopped via SIGSTOP or is being traced, and # # waitpid() was called with WUNTRACED flag. PID is still # # alive. From now on waitpid() will keep returning (0, 0) # # until the process state doesn't change. # # It may make sense to catch/enable this since stopped PIDs # # ignore SIGTERM. # interval = sleep(interval) # continue # elif os.WIFCONTINUED(status): # # Process was resumed via SIGCONT and waitpid() was called # # with WCONTINUED flag. # interval = sleep(interval) # continue else: # Should never happen. msg = f"unknown process exit status {status!r}" raise ValueError(msg) def disk_usage(path): """Return disk usage associated with path. Note: UNIX usually reserves 5% disk space which is not accessible by user. In this function "total" and "used" values reflect the total and used disk space whereas "free" and "percent" represent the "free" and "used percent" user disk space. """ st = os.statvfs(path) # Total space which is only available to root (unless changed # at system level). total = st.f_blocks * st.f_frsize # Remaining free space usable by root. avail_to_root = st.f_bfree * st.f_frsize # Remaining free space usable by user. avail_to_user = st.f_bavail * st.f_frsize # Total space being used in general. used = total - avail_to_root if MACOS: # see: https://github.com/giampaolo/psutil/pull/2152 used = _psutil_osx.disk_usage_used(path, used) # Total space which is available to user (same as 'total' but # for the user). total_user = used + avail_to_user # User usage percent compared to the total amount of space # the user can use. This number would be higher if compared # to root's because the user has less space (usually -5%). usage_percent_user = usage_percent(used, total_user, round_=1) # NB: the percentage is -5% than what shown by df due to # reserved blocks that we are currently not considering: # https://github.com/giampaolo/psutil/issues/829#issuecomment-223750462 return sdiskusage( total=total, used=used, free=avail_to_user, percent=usage_percent_user ) @memoize def get_terminal_map(): """Get a map of device-id -> path as a dict. Used by Process.terminal(). """ ret = {} ls = glob.glob('/dev/tty*') + glob.glob('/dev/pts/*') for name in ls: assert name not in ret, name try: ret[os.stat(name).st_rdev] = name except FileNotFoundError: pass return ret _psutil_linux.abi3.so 0000755 00000341210 15030211765 0010626 0 ustar 00 ELF > 0$ @ ȹ @ 8 @ # " U% U% P P P h h �] �m �m � � �] �m �m � � 8 8 8 $ $ P�td �T �T �T � � Q�td R�td �] �m �m X X GNU ��T�K�fM ���ggs�: G c�9 �$҈� �� G H J M O P S T V W X [ \ ����� U��� �ȷ��r:�H���$����"ոf�ذ�u U;hdeG�� ������w������l�� ������W�~�_aY � � j � � � B K � � � � � � M < � � � � � ( � ( � � � 1 � l � � � � � ` A - \ n 9 � 6 � N / v � t � � � 8 J � R " � � LE � p? � ~ 06 W � & � ~ �u � 7 � �B z � �% � + 9 � n @= & � �6 i W �< y c 6 �& j � �9 � a �$ � � A Q @ @< t ] ' W � �' ) � `B d __gmon_start__ _init _fini _ITM_deregisterTMCloneTable _ITM_registerTMCloneTable __cxa_finalize psutil_PyErr_SetFromOSErrnoWithSyscall __errno_location strerror sprintf PyExc_OSError PyObject_CallFunction PyErr_SetObject _Py_Dealloc NoSuchProcess AccessDenied psutil_check_pid_range PyArg_ParseTuple _Py_NoneStruct PyExc_ValueError PyErr_SetString psutil_set_debug PyObject_IsTrue PSUTIL_DEBUG psutil_setup getenv socket strncpy ioctl close Py_BuildValue PyErr_SetFromErrno _Py_TrueStruct _Py_FalseStruct setpriority getnameinfo PyList_New getifaddrs PyList_Append freeifaddrs getpriority PyUnicode_FromString psutil_getpagesize sysconf psutil_pid_exists kill psutil_raise_for_pid PyExc_RuntimeError PyErr_Format PyInit__psutil_posix PyModule_Create2 PyModule_AddIntConstant PyLong_FromLong PyModule_AddObject PyInit__psutil_linux psutil_proc_ioprio_get psutil_proc_ioprio_set psutil_proc_cpu_affinity_get psutil_proc_cpu_affinity_set psutil_disk_partitions psutil_users psutil_net_if_duplex_speed psutil_linux_sysinfo setutent getutent PyUnicode_DecodeFSDefault endutent syscall __sched_cpualloc sched_getaffinity __sched_cpufree PyExc_OverflowError PyErr_NoMemory __sched_cpucount stderr fprintf fwrite fputc PySequence_Check PySequence_Size PyErr_Occurred PySequence_GetItem PyLong_AsLong PyExc_TypeError sched_setaffinity PyEval_SaveThread setmntent PyEval_RestoreThread getmntent endmntent PyErr_SetFromErrnoWithFilename libpthread.so.0 libc.so.6 GLIBC_2.2.5 GLIBC_2.7 GLIBC_2.3 GLIBC_2.3.4 GLIBC_2.6 � ui � � ii � ii � ti � ii � ui � �m �$ �m �$ �m �m Hr MR `r �r �r [R �r 6 �r gR �r �. �r sR �r �* s �R s / s �R (s `( @s �R Hs �' `s �R hs @) �s �R �s t t �R @t S `t S �t )S �t ?S �t OS �t US u iS u wS @u �S �o �o �o �o �o �o ! �o # �o % �o . �o 4 �o L �o A �o D �o E (t Y Ht S ht Q �t H �t N �t V �t X u \ (u U Hu Z p p (p 0p 8p @p Hp Pp Xp `p hp pp xp �p �p �p �p T �p �p �p �p �p �p �p �p �p �p �p �p " q # q $ q & q ' q ( (q ) 0q * 8q W @q + Hq , Pq - Xq / `q 0 hq 1 pq 2 xq J �q O �q 3 �q 5 �q 6 �q 7 �q 8 �q 9 �q : �q ; �q < �q = �q > �q ? �q @ �q B �q C r D r [ r F H��H��O H��t�� H��� �5�O �%�O @ �%�O h ����%�O h �����%�O h ����%�O h ����%�O h ����%�O h ����%�O h ����%�O h �p����%�O h �`����%�O h �P����%�O h �@����%�O h �0����%�O h � ����%zO h �����%rO h � ����%jO h ���%bO h ����%ZO h �����%RO h ����%JO h ����%BO h ����%:O h ����%2O h ����%*O h �p����%"O h �`����%O h �P����%O h �@����% O h �0����%O h � ����%�N h �����%�N h � ����%�N h ���%�N h ����%�N h! �����%�N h" ����%�N h# ����%�N h$ ����%�N h% ����%�N h& ����%�N h' �p����%�N h( �`����%�N h) �P����%�N h* �@����%�N h+ �0����%�N h, � ����%zN h- �����%rN h. � ����%jN h/ ���%bN h0 ����%ZN h1 �����%RN h2 ����%JN h3 ����%BN h4 ����%:N h5 ����%2N h6 ����%*N h7 �p����%"N h8 �`����%N h9 �P����%N h: �@����% N h; �0����%N h< � ����%�M h= �����%�M h> � ����%�M h? ��H�=IQ H�BQ H9�tH�FK H��t �� � � � H�=Q H�5Q H)�H��H��?H��H�H�tH�]K H��t��f D � � �=�P u/UH�=>K H��t H�=�H �=����h�����P ]� � � � �{���f. � �AUATI��UH�� �|���I��8H������L��H�5�* L��H��1�����U 1�L��L�%yJ H�5�* I�<$����I�<$H��H�����H��tH�m tH�� 1�]A\A]� � H���P���H�� 1�]A\A]ÐATH��H�5�* 1�USH�� I��L������1�L�� H� �I H�5N* H�;�>���H�;H��H������H��tH�m tH�� 1�[]A\�f D H������H�� 1�[]A\�f. � ATH��H�54* 1�USH�� I��L���~���1�L�� H� ]I H�5�) H�;���H�;H��H�����H��tH�m tH�� 1�[]A\�f D H���8���H�� 1�[]A\�f. � H��H��H�5�, 1�H�T$ ������t"�D$ ��x*H��H H� H���f. � 1�H���f � H��H H�5�) H�8���1���f D H��H��H�5�* 1�H�T$�����t2H�|$�����x$H��H ���҉H�nH H� H��� D 1�H���f � H��H�=�( ���H��t H�DH � 1�H��� � UH��1�H�5�* H��@H�T$����A��1�E��tc1Ҿ � �:����Ń��tXL�D$H�t$� L�������!� �D$ H��1��&������t �*����t$ H�=s+ 1����H��@]�f�� ���H�dG H�8�l���H��@]�f D UH��1�H�5 * H��@H�T$�e���A��1�E��tm1Ҿ � ����Ń��thL�D$H�t$� L��������� �D$ H��1��v������t/�z����D$ @tCH�5�F H�=2) 1��.���H��@]� � �K���H��F H�8���H��@]�f D H�5�F H�=�( 1����H��@]� D H��H��H�5Y* 1�H�L$ H�T$���A��1�E��t �T$ �t$1�������tH�LF H� H��� H�F H�8�!�����ff. � @ H���� AW��AV��AUATUSH�� ��u\��� � H��E�H��E1�E1�j� H���p���H���� H��E H� H�� []A\A]A^A_� � ��u�D�w M��t�I��H��N�,7L��L�%\& D �S H��L��1�H��H���O���I9�u�K�vL��H�=( �D� 1�����f D H��H�=�' 1��w����h���f�H�E H� � @ AW1�AVAUATUSH��(�k���H�D$H���� H�|$��������f H�l$H���� �, � ��0 H��D H� I��I��H��H�uD��M��AUI��L��H�=o% 1�����I��XZM���� H�|$L��L�$�d���L�$���� I�*�N I�,$�V H�+�^ I�/�f I�m �n H�m H���w H�}H��t�D�7D�����H;�C I��t�H���= H�} D�����H��H���E �E�� ���H�}(D�����I��H��C H� I��M�������E1�H�|$H��t L�$����L�$H�T$H�H�$H��H�� M��t I�*�B M��t I�,$�R H��t H�+�c M��t I�/�t M��t I�m �� H�D$ H�D$H��([]A\A]A^A_� � L����I�,$�����L�����H�+�����H�����I�/�����L����I�m �����L����H�m H��������H�l$H������s���f D H�}(D���\���I��H�rB H� I��M���������� D H��L�$�L�L�$M����������f. � L���(���� L����M����������f. � L����H����������f. � H�����M����������f. � L�������� H�|$E1�E1�1�E1�H��� �������f�H�|$E1�E1�E1�H��������H�MA E1�E1�1�E1�H�8�J�E1����f�UH��SH�����H�T$ H�5 % H��� H��1��3�A��1�E��t!�t$ 1����Ƌ��uH�=�$ 1��H�H��[]ÐH��@ H�8���H��[]�f. � AU1�ATUH��H��@�-�H��t(I��H�T$1�H��H�5_# ���uI�,$�` E1�H��@L��]A\A]�1Ҿ � ��Ń���T H�t$L�D$� L��� �� ���D$ H��1������6 �����l$ @���� @���> @���� @���: @���� @�� �� @��@� @�ŀ�R f�� �� f�� �� f�� �! f�� �� f�� �� f�� � f�� @�E f�����H�=� ���H��H������H��L���Y��H�E �W H��H�E �����H���7����f�L���(���� H�=. ��I��H���m���H��L�����I�E �� H��I�E ����L��������f D H�=� �D�I��H��� ���H��L�����I�E uuH��I�E �����L�����|���f. � H�=4" ���I��H������H��L���]��I�E u%H��I�E �C���L���?��6���f. � H��I�E �����L�����z��� D H�=� ���d��� � H�=� ��I�,$� ��E1��W�H��@L��]A\A]�f. � H�=� �4�I��H��� ���H��L������I�E �a���H��I�E �����L���{��|���f D H�=� ���I��H�������H��L���M���I�E ����H��I�E �C���L���+��6���f D H�=y ��I��H���m���H��L������I�E �����H��I�E �=���L������0���L��������f � H�= �4�I��H��� ���H��L������I�E �a���H��I�E ���L���{�����f D H�=� ���I��H�������H��L���M���I�E ����H��I�E �����L���+����f D H�=P ��I��H���m���H��L������I�E �����H��I�E �����L�������f D H�=� �D�I��H��� ���H��L������I�E �q���H��I�E �����L�������f D H�=� ���I��H������H��L���]���I�E �!���H��I�E �q���L���;��d���f D H�=q ��I��H���}���H��L��� ���I�E ����H��I�E �,���L������ ���f D H�=) �T�I��H���-���H��L������I�E �����H��I�E ���L��������f D H�=� ��I��H������H��L���m���I�E �1���H��I�E �����L���K����f D H�=� ��I��H�������H��L��� ���I�E ���H��I�E �~���L�����q���H��H�E �M���H������@���ff. � � �v�f D H����H�=& H��H��1����f��� 1��f � H��1����A��� E��t*�#�� ��t ��tH�@9 H�8�H��������1�H���f � UH��SH��H������ ��tH��H��[]�� ���1���uH��H��[]�/� � H��8 H��H�5q H�8H��1�[]�� � AT� H�=; �=�H���� � H�5D H��I���=����� � H�5/ L���!����� 1�H�5" L�������� � H�5 L��������l � H�5 L��������P � H�5� L�������4 � H�5� L������� � H�5� L���|����� � H�5� L���`����� � H�5� L���D����� � H�5� L���(����� � H�5� L��� ����� � H�5� L�������ut� H�5� L�������u\� H�5w L������uD� H�5m L������u,H��������H��H��tH�5X L����� � E1�L��A\�f. � D AT� H�=�: �-�H��tp�� H�5� H��I���1���uU1�H�5r L��� ���u@� H�5f L������u(�� H�5Z L�������u��L��A\� D E1�L��A\�f. � D AW1�AVAUATUSH���{�I��H��� �Z�f. � ��H��H���� f�;u�H�{,�@�I��H���] H�{�+�I��H���� �{L:�� �{M0�� �{N �� H�=P ��H��H���� f�D�CH��L���*�T L��� H�=$ ��I��H���� H��L���8����� I�/�� I�.�� H�m �� I�,$����L������H��H������ D ��H��L��[]A\A]A^A_ÐL�CL� H�=� L����� ��� ���L����H��H���!���E1�I�/�� I�.�� H��t H�m �� M��tI�,$ttI�m tX�+�E1��p��� L���H�I�.�"���L���6�H�m ����H���#�I�,$�(���L����� ��� L��E1����������L�����I�m u���L�����I�.�M���L������@���H�����C���I�/�E���L�����8���f. � D H��H��H�5V 1�H�T$ ��A��1�E��t3�T$ � �� �y����t$��H�= 1��� �� ��H��� @ H� 3 H�8��H���ff. � �H��H��H�5� 1�L�D$ H�L$H�T$��A��1�E��t.�L$�T$� �� �� L$ ������tH��2 H� H��� � H��2 H�8���� � AWH��1�H�5V AVA� AUA�@ ATUSH��H�T$ �p���u� � E�Ic�L�?��M��I��H��I��H���� �|$ H��L���+��Å��� H�������8� A��u�H��1 H�5 E1�H�8� �H��L��[]A\A]A^A_�H��I�uL����H����I�m �/ E1��ŐH��1 � ���� ��I��� D 1����I��H��twL��H��I����A�ą�u�_f � ��Hc�L9�s�H��H��H�D� H��s���I��H���i���H��L������I��B���H��I�t3��A��u�H��������� D H��0 H�8���I������L�������H� �0 �b H�� 1�H�5� H�;��H� � � H�=� ��H�3� �<����L���o�����f. � AWH��1�H�5o AVAUATUSH�� H�L$H�T$�D����� H�|$������ H�|$� �I��H���� L�l$ � � L���H��� 1�A� �> � I���tb�=�H��uXI��� wL��L��D��H��H��I T� H��I9�tyH�|$H���$�H��H��t&H����H�m I��u�H���r�����H��t`1�H�Ĩ []A\A]A^A_�f�H�D$H�5� H�PH�/ H�81����� D �|$L�꾀 �����u/H� / H� �H��. H�5 H�D$H�:�+�H�D$�H��. H�8�����l���ATH��1�H�5} UE1�H��xH�T$������ 1Ҿ � ����Ń���� L�D$H�t$� L���5��f�F� ��I��H�D$@D$@H�D$ L��1��D$ H�D$` �D$h �D$@ D$P������tF�T$\�D$L�t$N�� ¸ H�H�=� 1��L�I��H��t`���m��H��xL��]A\�f����� ��_t��u/1Ҿ� �f D H�=D �T�H��xI��]L��A\�f D H�=- �4���E1�� ��H��xL��]A\�f. � D H��xH���$���u@�D$hH�= P1��t$PL�L$PL�D$@H�L$HH�T$8H�t$0���ZYH��x�f � H��, H�8���H��x�f. � f�AW1�AVAUATUH��SH���X��H���w I��H�T$1�H��H�5� �������V �I�H�|$H�5 L�5� H���n��H��I���s��M���� f. � L���8��H��H���� H�;����H��H���� H�{���I��H���3 H�KL�CH��H��L��1��{��H��H���� H��L��������� H�m t%I�/t-H�+�u���H�������h��� � H������I�/u�L������H�+t�L���|��H��H���D���L���8���Af D H��+ � ���� H�h+ H�t$H�8�K��I�,$u L���l�� @ E1�H��L��[]A\A]A^A_� L������H�m �� I�/tH��t�H�+u�H���$���f�L�������f D L�����H�m u�H������y��� @ H� + �# H�� 1�H�5� H�;����H� � � H�=� ���H�3� �f�������L���(������H������I��� H��H��� %s (originated from %s) (is) PSUTIL_DEBUG assume no such process (originated from %s) assume access denied (originated from %s) pid must be a positive integer %02x: (siOOOO) socket(SOCK_DGRAM) ioctl(SIOCGIFFLAGS) up broadcast loopback pointopoint notrailers noarp promisc allmulti master slave multicast portsel automedia dynamic %s syscall failed RLIMIT_AS RLIMIT_CORE RLIMIT_CPU RLIMIT_DATA RLIMIT_FSIZE RLIMIT_MEMLOCK RLIMIT_NOFILE RLIMIT_NPROC RLIMIT_RSS RLIMIT_STACK RLIMIT_LOCKS RLIMIT_MSGQUEUE RLIMIT_NICE RLIMIT_RTPRIO RLIMIT_RTTIME RLIMIT_SIGPENDING RLIM_INFINITY _psutil_posix getpagesize getpriority net_if_addrs net_if_flags net_if_is_running net_if_mtu setpriority version DUPLEX_HALF DUPLEX_FULL DUPLEX_UNKNOWN _psutil_linux proc_ioprio_get proc_ioprio_set proc_cpu_affinity_get proc_cpu_affinity_set disk_partitions users net_if_duplex_speed linux_sysinfo check_pid_range set_debug :0.0 localhost OOOdi iii psutil/arch/linux/proc.c psutil-debug [%s:%d]> CPU_ALLOC() failed iO invalid CPU value could not allocate a large enough CPU set sequence argument expected, got %R socket() ioctl(SIOCETHTOOL) [ii] (kkkkkkI) psutil/arch/linux/disk.c setmntent() failed (OOss) ;� d�� 4��, ���l d��� ���� d�� ���, ���D ���p ���� ���� ��$ ���� D��� D�� T� t�$ ��< D�| T�� ��� �� �, ��H ��� D�� ��, �\ zR x � $ X�� FJ w� ?;*3$" < D ��� B�B�D �G�j CBBH O CBB< � `��� B�M�A �G�L CABG O CAB< � ���� B�M�A �G�L CABG O CAB ��j D q K F J $ P��W D F F F D ���) Dd ( \ ���� A�PP{ AC XA 0 � ,��� A�PP� AI X AG YA � ���a D H D ` � ,�� K�D�E �B(�A0�A8�G�\�H�P�Z 8A0A(B BBBI ������� X < ���� B�D�B �B(�A0�A8�D`]hLpXhA`l 8A0A(B BBBH 0 � <��v A�D�D0S AAB SAA@ � ���� B�D�A �G`y DBBA � DBBK D�� $ @�� DP < H��W TB< T ���i A�D�G O DAH O DAL XCA � ��� B�� � ���� B�� F HL � 0�� B�D�B �B(�A0�A8�D@1 8D0A(B BBBB $ p�t D W E S D ��y D \ H H ` 4�& B�N�H �H(�A0�A8�DP� 8D0A(B BBBA L � �� B�N�B �B(�A0�A8�G�� 8A0A(B BBBC D � X�Q B�M�G�� DBC p DEG ZDB , D p�d D�X�F�_�A�D J S H t ��z B�D�B �B(�A0�D8�DP� 8D0A(B BBBD �$ �$ �m � � LE �m �m ���o ` � � p � ( ���o x ���o �o � ���o �m 6 F V f v � � � � � � � � ! ! &! 6! F! V! f! v! �! �! �! �! �! �! �! �! " " &" 6" F" V" f" v" �" �" �" �" �" �" �" �" # # &# 6# F# V# f# v# �# �# �# �# �# �# �# �# $ $ &$ MR ���������r [R 6 gR �. sR �* �R / �R `( �R �' �R @) �R �������� t �R S S )S ?S OS US iS wS �S GCC: (GNU) 10.2.1 20210130 (Red Hat 10.2.1-11) , �$ � , p �' Q , �? 9 � , �E �9 � , ;N @< � , �] A Q , ai `B d , �l �B z l � ] �$ � � � � � � X int W � x �W � �c � � � � � fv W � z bj } \ T� g G � 8 + : W ; W � 5 j; = l� u m@ ) ; � n F � �F � �R ; �R � W g ^ �} �u � 3� � � � ,W � R � R� R � * W R � L R � N3 R R : v R P R � � l W m � � � �� � W � 2 � �W �' ) �� �' � U P � �R ' W �t } � R b �,R C = � �R �h x � W � � :' � f U�TT R Q�h H' � � �R �& j � } �"R � � b �2R � � pid � � �l �& � � U�TT �S Q�l ' � T �P P {R & � �( {� G C exc |R � � msg } ( ��w � � !m �R � � "� !� �R �& Uv B& P � U��wT `P Q�U b& 3 T P Q=R��w p& Tv #� 9 $- � N jR �% � �L j� G C exc kR � � msg l ( ��w ` � !m qR � � "� !� qR & Uv �% P U��wT 0P Q�U �% 3 7 T P Q3R��w �% Tv % SR �$ � � S4� I C � T ( ��w exc [R � � � !m _R � � "0 !� _R " �% Uv % � % m (% P 4 U��wT P R| G% 3 Z T P R��w V% Tv 8 A � F �' Q o � �9 � � � � � X int c � x �c E �N � �o � �� � � � � R �N � � f� c � z bv G h� } T � N I/ � ; � h � Y � R � � � � � � 4 � � q � � � � � _ � � ! L " � # 1 $ � % � & O ' � ( � ) * , + � , } - � . � / v 0 � 1 � 2 � 3 � 4 0 5 | 6 $ 7 H 8 9 L : p ; � < , < � = � > ? � @ � A � B C 3 D � E � F � G � H ] I � J � K � L / M k N � O � P Q R # S � T U F V M W � X � Y 7 Z � [ � \ � ] � ^ � _ � ` � a b � c d C e q f � g � h % i � j ^ k � l m � n � o � p > q r s � t � u v � w x 3 y y z � { $ | k } ~ ~ K � � � � E � � � � � � � � � � � q � < � U � � � 6 � & � � � � � ~ � � � ` � � � � = � K � { � q � � � � � � � � : � ] � � � � � � f � � � � � 8 � � � � p � x � � � � � 3 � $ � � � % � � � � � � � � � � L � ' � � � � � P � � 4 � � � O � � � k � � � ^ � � � f � � � U � � � � � h � Y � � � � � � � � � � � � � � � � � � � 0@ � 1 G < 3N \ T� g G � 8� + : c ; c m � 5 j� = lS u m� ) � � n� � � � � � � � c � �, 2 c F � � ^ �R X c q � � � ?~ � � � � �� � & � A 1� � � 9 6 � 7� � 8� ] 9c ;� 9 = � � ( +U � ,� � -Z ^ .S � / � � U � 0 <� � = c � > � T h J � K` � L� ( j M� 0 � NS 8 � O @ � P H � QF P 1 R � X S q ` l � �� � �� ? N � � 6 � � � � Q � � � � 8 . 8 W g f N $� z � X L ) � � N ? y � I k � 9 H � � G �s w �? � � x K � � 9 K � � � � � J � � � | � � � 9 � � � � �) � �? # �- z � h �� � � . � �� � ? � - G � � P G 9 9 � ' � � � � � � � � � � � \ � � � ' � � � s � & � 1 � < � G ) R � ] � h � s � ~ � � � � � @ � 9 N +1 d � 3 � - � @ g �� ^ ` � Q ] @� � � o� O q9 q r9 � sG irq t@ dma u@ � v@ �� � �� � � 9 �M H �K b �K � �K �K �K ! � \ � �c � �c � �1 w �� �� � � � (~u � � � & � � &� � ,� � -� 8 � ! � � "N � $� X %� 5 .u ( � 8 � 0 � E G { - � " � x; �g A � g � � w = � � / w 9 ; � 9 G � 9 � �� r � 9 � G k � # 3 F G � � � c � G � @ � @ � � g j � P 9 � f@ �r u� r !� ^ c � � � "� l c � � � # "@ � c � ! � � c $ E � !� B c !" 9c 4 � � "& �� K � %L ^ � !I 4� t S !k (� � � # "� a c � c !j ) c � c 9 # !� !r c � c c c !t b c � � � c !* c � � # !� Kc : � � � !a "� P o !� Lc p � � o !/ �� � � c � "� � � � � # !N V � � � ! W � � � !� �� � � &� $2 � ! # c v c "U ko . c '� �� 7 �w (mod �� \ X (v �� � � )7 p � *U r *T3 )37 P � *U| *T lQ *Q9 )O7 P � *U| *T vQ *Q4 )h7 P *U| *T �Q *Q0 )�7 P E *U| *T �Q *Q2 )�7 P o *U| *T �Q *Q1 )�7 P � *U| *T �Q *Q8 )�7 P � *U| *T �Q *Q7 )�7 P � *U| *T �Q *Q6 )8 P *U| *T �Q *Q5 ),8 P A *U| *T �Q *Q3 )H8 P k *U| *T �Q *Q: )d8 P � *U| *T �Q *Q< )�8 P � *U| *T R *Q= )�8 P � *U| *T R *Q> )�8 P *U| *T R *Q? )�8 P = *U| *T -R *Q; )�8 : U *U � +�8 *U| *T ?R , �� `( � � -} �$� � � -b �4� � � � � � ��.� � c R F (ret � c � � /ifr �M �@0� �){( � E *U�T*T �R *Q�� )�( � f *U2*T2*Q0 )�( 8 � *U�@*Q? )�( � � *Uv *T � )�( � � *Uv )�( t � *U R 1) � 1) � +5) t *U R ,� �� / � ��* -} � � -b �/� W M � � � ��.� � c � � (ret � c � � /ifr �M ��.� �� � � .� �\ G ? 0� t2` & .� u� � � )�0 K *U| +3 K *U| 3�* T0 � J5 4�* 4�* W S 5� 6�* � � 7�* � � 6�* +�5 K *Uv 8�* �0 � 6�* / - 9�* �0 6�* T R +�0 K *Uv )`0 4 *U RQ +w0 *U| *Tv :�* �0 �0 P �: 4�* y w 4�* � � 6�* � � 8�* �0 � ;�* 8�* �0 ! 6�* 2 0 9�* �0 ! 6�* W U +�0 K *U} )�0 4 *U �P +�0 *U| *T} :�* 1 1 P �? 4�* | z 4�* � � 6�* � � 8�* %1 � ;�* 8�* +1 % 6�* 5 3 9�* +1 % 6�* Z X +A1 K *U} ) 1 4 $ *U �P +#1 *U| *T} :�* P1 P1 P �D 4�* } 4�* � � 6�* � � 8�* u1 � ;�* 8�* {1 % 6�* 8 6 9�* {1 % 6�* ] [ +�1 K *U} )\1 4 ) *U �S +s1 *U| *T} 3�* �5 DS 4�* � � 4�* � � 5 6�* � � 7�* @ � 6�* ; 9 +�1 K *U} 8�* �5 6�* ` ^ 9�* �5 6�* � � +�5 K *U} )�5 4 7 *U HQ +�5 *U| *T} :�* 2 2 P X 4�* � � 4�* � � 6�* � 8�* 52 � ;�* 8�* ?2 ! 6�* c a 9�* ?2 ! 6�* � � +U2 K *U} ) 2 4 = *U �P +32 *U| *T} :�* `2 `2 P ] 4�* � � 4�* � � 6�* � 8�* �2 � ;�* 8�* �2 ! # 6�* f d 9�* �2 ! 6�* � � +�2 K *U} )l2 4 B *U �P +�2 *U| *T} :�* �2 �2 J 2b! 4�* � � 4�* � � 6�* 8�* �2 � ;�* 8�* �2 (! 6�* i g 9�* �2 6�* � � +�2 K *U} )�2 4 G! *U 0Q +�2 *U| *T} :�* 3 3 P 8g" 4�* � � 4�* � � 6�* 8�* 53 �! ;�* 8�* ?3 ! -" 6�* l j 9�* ?3 ! 6�* � � +U3 K *U} ) 3 4 L" *U 6Q +33 *U| *T} :�* `3 `3 P >l# 4�* � � 4�* � � 6�* 8�* �3 �" ;�* 8�* �3 ! 2# 6�* o m 9�* �3 ! 6�* � � +�3 K *U} )l3 4 Q# *U @Q +�3 *U| *T} :�* �3 �3 P q$ 4�* � � 4�* � � 6�* 8�* �3 �# ;�* 8�* �3 ! 7$ 6�* r p 9�* �3 ! 6�* � � +�3 K *U} )�3 4 V$ *U Q +�3 *U| *T} :�* 4 4 P v% 4�* � � 4�* � � 6�* 8�* %4 �$ ;�* 8�* /4 ! <% 6�* u s 9�* /4 ! 6�* � � +E4 K *U} ) 4 4 [% *U �R +#4 *U| *T} :�* P4 P4 P {& 4�* � � 4�* � � 6�* 8�* u4 �% ;�* 8�* 4 ! A& 6�* x v 9�* 4 ! 6�* � � +�4 K *U} )\4 4 `& *U Q +s4 *U| *T} :�* �4 �4 P �' 4�* � � 4�* � � 6�* 8�* �4 �& ;�* 8�* �4 ! F' 6�* { y 9�* �4 ! 6�* � � +�4 K *U} )�4 4 e' *U Q +�4 *U| *T} :�* �4 �4 P &�( 4�* � � 4�* � � 6�* 8�* 5 �' ;�* 8�* 5 ! K( 6�* ~ | 9�* 5 ! 6�* � � +55 K *U} )�4 4 j( *U Q +5 *U| *T} :�* @5 @5 P ,�) 4�* � � 4�* � � 6�* ! 8�* e5 �( ;�* 8�* o5 ! P) 6�* � 9�* o5 ! 6�* � � +�5 K *U} )L5 4 o) *U )Q +c5 *U| *T} )/ ^ �) *U0 )1/ � �) *Uv *T �R *Q�� )a/ � �) *U2*T2*Q0 )�/ 8 * *U��*Q? )�/ � +* *Uv *T � )�/ � C* *Uv )�1 � b* *U �P )�1 � �* *U �P +�1 � *Uv <I �c + =� �� =� �0� > �� ?�* >� � � @>� �� @>� �� ,� �� �' � �~, -} � � � � -b �-� � � � ��.� � c b X (ret � c � � /ifr �M �@0� �)�' � �+ *U�T*T �R *Q�� )�' � �+ *U2*T2*Q0 )( 8 , *U�@*Q? )( � ,, *Uv *T !� )&( � D, *Uv )8( t c, *U �S 1E( � 1T( � ,� -� �* � ��1 -} - � -b -/� N J B . ��(ifa . � � .� / c � � .� 1� j d .9 2� � � .� 3� � � . 4� � � .r 5� � .� 6� D! *! A� y/, 2P �- .� o � Y" Q" 5P .� o � �" �" +�, K *U�� 2� 3. .� p � %# !# 5� .� p � _# [# +�, K *U| 2� �. .� q � �# �# 5� .� q � �# �# +�, K *Us 2� �. .� r � $ $ 5� .� r � G$ C$ +- K *U 2 / .� s � �$ }$ 5 .� s � �$ �$ +"- K *U} 2@ R/ .� |� �$ �$ +�- K *U�� 2p �/ .m }� J% D% 5� .� }� �% �% 1�- K 2� �/ .m ~� �% �% 5 .� ~� & & +�- K *U| 2@ 50 .m � \& V& 5p .� � �& �& +�- K *Us 2� �0 .m �� �& �& 5� .� �� 2' .' +. K *U 2 �0 .m �� l' h' 50 .� �� �' �' +�- K *U} )�* ^ �0 *U0 )�* � 1 *U�� )M+ t 81 *U �P *Q~ *R| *Xs *Y )l+ Z1 *U��*T�� )�+ �1 r1 *T~ )�+ �1 �1 *T~ ), �1 �1 *T~ 1B, � )=- � �1 *Uv )T- �1 �1 *T~ 1v. � B �� x2 C� �(� C� �2c Dbuf � x2 Derr � c E, � c Dn � - Dlen � - E� �� Dptr � � @>k �2 � �2 F9 � Bg �� �2 C} �$� Cb �4� Dpid � � E� � c E � c B� �� 3 C} �$� Cb �4� Dpid � � E� � c Go �� 6 ��3 H} �(� �' �' Hb �8� ( ( 16 |4 I.6 t *U FQ J� ��6 i �R4 Kpid �o d( V( H �&� ) ) 1�6 � L�6 � �3 *U�T )�6 R4 4 *Us L�6 � /4 *U�T I�6 � *T ZQ *Q�T M� `c |4 Npid `� Dret a c O� Ko 6 ��4 I 6 *UN P�2 @) a ��5 4�2 �) �) 4�2 0* ** Q�2 �hQ�2 �l6�2 �* |* R�2 �) �) �a5 4�2 �* �* 4�2 �* �* ;�2 ;�2 ;�2 1�) � )_) � �5 *U�T*T �S *Q�h*R�l +x) � *U0 S�1 �) �c6 T�1 UT 2 TQ2 ��w;"2 ;.2 ;:2 ;D2 ;P2 ;\2 U�1 5* �V 2 V�1 5 Q2 ��w;"2 ;.2 ;:2 QD2 ^;P2 ;\2 9h2 5* ;i2 P�2 �. v �p7 4�2 + + 4�2 M+ C+ Q3 �\63 �+ �+ R�2 �. �. �7 4�2 , , 4�2 ;, 9, ;3 ;3 1�. � 1�. � )�. � =7 *Uv *T �S *Q�\ )�. ~ T7 *U0 +�. t *U �S PR4 06 W �8 4c4 b, ^, 6o4 �, �, RR4 i6 i6 `�7 4c4 �, �, ;o4 1x6 � )K6 � 8 *T0 1]6 � Wi _ % � r � 9 � 5 � � � � � X int W � � �c � x � fl � } \ T� g G � 8� + : W ; W � � 5 j = l� u m ) � n� $ < 0 P 0 0 �\ b W q 0 �} � W � 0 j ^ �� � W � 0 q j � ?� � � j � 6 9 6. � 7� � 8� ] 9W ;� 9 = � � (+| � ,$ � -� ^ .� � / 0 0 | � 0: <� � = W � > j T hJ> � K� � L� ( j M� 0 � N� 8 � O> @ � P D H � Q� P 1 R P X S � ` . � g ^ �. g - � W t 5� �s � ^ 0 � 0 0 � ] 0 � 0 0 � 0 � 0 0 � 0 � 0 0 j 0 0 0 w 0 5 0 0 � 0 P 0 0 M 0 k 0 0 � 0 � 0 0 � 0 � 0 0 _W � LW � 0 � c / �0 � � W � C0 9 � �mod D0 - �, #9 � D U �s T3 ?9 � p U| T �R Q � T9 � � U| T �R Q0 l9 � � U| T �R Q1 �9 � � U| T �R Q� �9 � � � � n �9 � � � �9 � � @ � � � X int h � x �h � �t � � � � 9 � � 9 � � f� � z b{ = �h } � 9 \ T� g G � 8I + : h ; h ! I 5 j| = l u m� ) | � nT � � � 9 � R 2� D 4a 5a I� ^ K � < L � � � :� � < a = � J >� � ?� ( � @� , � A� L+ B� L H � P� M� T� S � \A T� l � � 9 g o �� D 3 >� � L � � " 9h � � � k (� � � 4� + � � A I 4� I j � �9 � �� } � P- L- b (� �- �- ut � �- �- � � K. ?. 9 � �. �. 5 � �/ �/ � n0 Z0 � � V1 @1 � ? !: � " $ - e $ - D2 <2 !: � $ � �2 �2 � $ h 3 3 ; � " $/- e $/- �3 �3 � $/� � $/h !� � 7 � �3 �3 "� � 7 � �3 �3 #�; � $U !� j � 8 � 4 4 "� � 8 � Y4 U4 #�; � $U~ ! � � 9 � �4 �4 " � 9 � �4 �4 #�; � $Uv !p � : � 5 5 "� � : � A5 =5 %�: � $U| #�; � $U| !@ ~ m @� }5 w5 "@ � @� �5 �5 %�; � h $U #,< � $U !� � m A� 6 6 "� � A� k6 e6 #< � $U~ !� m B� �6 �6 "0 � B� 7 7 #< � $Uv !` b m C� Y7 S7 "� � C� �7 �7 #�; � $U| !� � � D� �7 �7 &�; � %�9 3 � $U0 &�9 + &�9 � % : � $Us, %: � $Us %K: $U �S %�: � > $U �S $T $Q~ $Rv %�: � \ $U} $T| &�: � &�: � %*; � $Us� &u; � &�; � G p � � � @< � � � �9 � � � � � X int c � R �o � �o x �c � �o � � � � 0� � ��e � �c Z � � � � � � � � N � � \ � ( � � 0 � � 8 � @ � � H � � P � � X � ` g � h � c p � c t / v x G � ? U � � � � � � � 3 &� � � / � � � 0 � � � 1 � � � 2 � � � 3 - � � 5c � � 7� � � �I �� � �� j �� �c m � � � 9 e � � 9 � � � f� � �� c � z b� } \ T� g G � 8Y + : c ; c 1 Y 5 j� = l u m� ) � � nd � � �� � �� � �� , �� ; �� � � N w9 � ~ � � � $ 9 �� $ � c g j �N c � < u c � � - � 0 � V� ; o � � n . � � � � � � � � � � q c � L % � " 9c @ � � a � V o � c q - � I 4� � � R� � � � 2 � � � � � $ p y c � � - � � �� T d c � � ` �� ! - 8o 9 o k (� P � � �� f � * c � � � � �� p? � �� } �(� E8 A8 b �8� �8 ~8 / �$ ��~ len � - �8 �8 !pid � ��~ i � 9 9 "7 � d9 ^9 , �� ��~#� 7 "� �� �9 �9 "� �o : : # � "m � � {: u: $ "� � � �: �: %^@ &Uv #� � "� � - ; ; '@ � (<@ � � &Ts (L@ � &Uv 'e@ � %�@ � &T �S (�? f k &U�T&T �S &Q��~&R��~ '�? � '�? � (�@ � � &T 8T (�@ c � &T�&Q} '�@ P M R� @= & �w } R(� <; 8; b R8� {; u; cpu S c �; �; "� Sc S< A< "& Sc $= = "� S c �= �= !pid T ��"$ U - �= �= "T V� g> Y> " W� ? �> )� �#p � "� x - �? �? #` � " y� �@ �@ #� + "� }� A A %> &U~ #� ] "� � � ;A 7A %? &U~ (�> @ { &Us $ & %�> % &U} &T~ # � "m �� wA qA $@ "� �� �A �A %a? &U} (p= f &U�T&T �S &Q�� (�= ' &UA (�= � E &T| &Qv (�= � ] &Uv '�= � (�= � � &T T ( > � � &Uv 'F> � (W> q � &U0 (n> V � &U| &Tv (�> � � &Uv '�> P (.? � : &T �S &Q �S &Rb (G? ] c &U �S &T1&QB %T? h &U: � <� �< y �� } <"� �A �A b <2� >B 8B !pid = �d"Y > c �B �B >c �h � >c �l" ? c �B �B *� �< Fp +& C C + PC NC + vC tC % = ! &U�&T1 (�< f � &U�T&T �S &Q�d&R�h&X�l '7= P � *� @< t �� } *"� �C �C b *2� �C �C !pid + �l"Y , c 1D )D " ,c �D �D "� ,c E �D ,3 d< d< /� +P ZE XE +D �E ~E %w< ! &U�&T1 (Z< f � &U�T&T �S &Q�l (�< 9 � &U �S '�< P -� c 3 .q c /who c .Y $c -� c ] .q c /who c 0� � 0� � � � � �! A Q p" � �9 � � � � � X int c � � �o � � � � � fx c � } < 3N \ T� g G � 8 + : c ; c � 5 j= = l� u mB ) = � n H � N � y � I k � 9 H � � G �� w �� � � � � � � 9 � � � � � J | % % / 9 : : D � O O Y � d d n ' y y � � � � � � � � � \ � � � ' � � � � � � � � * ? # T . i 9 ~ D � O � Z � e � p � { ! @ � G ! N ' � �! (N ] )N �! *G " +� - % �! .N ] /N �! 0G " 1N �! 2� 4 U P! 5G � 6G � 71 9 � ! :N $! ;N " <N " =N " >N lmi ?G dce @G �! Aa C � � DN � E� G � HN �! I � ) 9 e! J� L Y �! MN ! NN )! O5 � �� O �9 q �9 � �G irq �@ dma �@ � �@ � �! � �! �$ fr �* W" �0 1" �6 i" �< te1 �B U Y � � ) � % �! �} x �N � �N � �� �� � � �9 H �� b �� � �� �� �� ! � \ � �c � �c � �e w � � � v � �H � ( �a � �} & �� 5! ,aL cmd b� �! c� D! d� � e� I" f� � g� ! h� =" i� � j� h k� � l� � m� � n� Y! o� w! p� A! q� � rQ $ a � a 9 g j �� a c � c k (T � � � 2 � j ) c � c 9 P" >v � v c - W T � � � r c c c c * c 6 T � � :T A Q �� } :&T �E �E b :6T �E �E � ; � ��~ � < c AF 3F !ret = c �F �F I" > c G G ^" ? � 8G 6G � @ c qG oG "ifr A9 �� � Ba �� � CT �G �G #� t<B $� �A P W_ %� �G �G & A � 'U�T'T �R 'Q��~ &9A � � 'U2'T2'Q| &[A � � 'U��'Q? &�A � � 'Uv 'T F� &�A � 'U wT &�A n " 'Uv (�A � & B � N 'U [T &<B � m 'U dT )FB n 'Uv *u #� � +�! #4� L ,i _ B D � �" `B d �% � � � � � X int W � � �c � v } � fj � } \ T� g G � 8� + : W ; W � � 5 j = l� u m ) � n� " � �. � ; ! B �" c �" - � p 3 �" X t" 3 �" d �" d (�" d 0�" d 8�" d @�" d Hn" @ P pad @ R�" d Xz" d `�" L h _f C l d C - v S - g ^ �k (. w � � �. � . � W � � p � . `B d �} . &H "H b 0. cH _H r! p ��lB � U�� �B ` 7 U |T �B w � � � �# �B z �' � �9 � � � � � X int c � R �o � �o � �o � � � � 0� � ��Y � �c Z � � � � � � � � N � � \ � ( � � 0 � � 8 � @ � � H � � P � � X � ` g � h � c p � c t / v x G � ? U � � � � � � � 3 &� � � / � � � 0 � � � 1 � � � 2 � � � 3 - � � 5c � � 7� � � �I �� � �� j �� �c a � � � 9 Y � � 9 � � � f� � �� � } \ T� g G � 8; + : c ; c ; 5 jn = l� u ms ) n � nF y # R� _ts � �� � �� � � &# ( 5 Q# 7 � ?# 8 � �# 9 � |# : � �# ; c q# < c $ � c g j �## W c J � �# G` ` � � L y � " 9c � � � k (� � � � 4� � � � � � � � �# �� � � � T d c � � \# �) ) � G# B� J � � -# � ) * c r � � I 4� � � w � �B z �� } "� �H �H b 2� �H �H �# � 3I +I �# ` �I �I �# � �� # � ,J J �# � �J �J 9 � �K �K � � �L {L � C � � �# ) �L �L C J !2C / � "T .Q #=C "Uv � < � 0M .M $ � < � UM SM #�C f "Uv 0 X � = � |M xM $0 � = � �M �M #D f "U � � � > � �M �M $� � > � *N &N #�C f "Us ` m F� jN `N $� � F� �N �N !�D f � "Uv #EE f "Uv T m G� 4O *O $p � G� �O �O #�D f "U � � m H� �O �O %�D � H� P P #�D f "Us &UD � � I� CP AP #dD f "U| !�B r "U0 ! C V - "Uv "T �R "Q�� !XC J E "U} lC � �C � !�C � � "U~ "Tv "Q !�C y � "U| "Ts !D J � "U} !(D 4 � "U} UD � !�D 4 � "U} !�D 4 "U} !E � @ "T �S "Q �T "R# ! E � i "U �T "T1"QB !*E � � "U: #8E 4 "U} '� � '� � % $ > $ > & I : ; 9 I I : ; 9 : ; 9 I8 7 I < 4 : ;9 I?< 4 : ; 9 I?< 4 : ; 9 I 4 G: ; 9 .?: ;9 'I< I .?: ; 9 '< .?: ; 9 'I< .?: ;9 '< . ?: ; 9 'I< .?: ;9 'I@�B ��1 �� �B .?: ; 9 'I@�B : ; 9 I�B 4 : ; 9 I 4 : ; 9 I�B ��1 �� 1 4 : ; 9 I U !4 : ; 9 I�B " U #I $! I/ %.?: ; 9 'I@�B % : ; 9 I $ > $ > & I I 7 I : ;9 I > I: ; 9 ( : ; 9 : ; 9 I8 < 'I I ' 4 : ;9 I?< 4 : ; 9 I?< 'I > I: ; 9 > I: ;9 ( ( I ! I/ : ;9 I8 : ; 9 I8 : ; 9 : ; 9 I 4 : ; 9 I 4 : ;9 I !.?: ; 9 'I< ".?: ;9 'I< # $.?: ; 9 '< %.?: ;9 '< &. ?: ; 9 'I< '.?: ;9 'I@�B (4 : ;9 I�B )��1 *�� �B +��1 ,.: ;9 'I@�B - : ;9 I�B .4 : ;9 I�B /4 : ;9 I 0 : ;9 1�� 1 2 U 3 1R�BUX YW 4 1�B 5 U 64 1�B 7 1U 8 1 9 1 : 1R�BX YW ;4 1 <.: ;9 'I = : ;9 I >4 : ;9 I ? @ A : ;9 B.: ; 9 'I C : ; 9 I D4 : ; 9 I E4 : ; 9 I F! I/ G.: ; 9 'I@�B H : ; 9 I�B I���B1 J.?: ; 9 '@�B K : ; 9 I�B L���B1 M.?: ; 9 'I N : ; 9 I O.?: ; 9 'I@�B P.1@�B Q4 1 R 1R�BX Y W S.1@�B T 1 U 1R�BUX Y W V 1 W. ?<n: ; % $ > $ > & I : ; 9 I I : ; 9 : ; 9 I8 7 I < 'I I : ;9 I ' 'I 4 : ; 9 I I ! I/ 4 : ; 9 I .?: ; 9 'I< . ?: ; 9 'I< .?: ; 9 'I@�B 4 : ; 9 I�B ��1 �� �B �� 1 % : ; 9 I $ > & I $ > I ! I/ I : ; 9 : ; 9 I8 7 I < : ; 9 : ; 9 : ; 9 I8 4 : ; 9 I . ?: ; 9 '< . ?: ; 9 'I< .?: ;9 '< I .?: ; 9 'I< .?: ;9 'I< .?: ; 9 'I@�B : ; 9 I�B 4 : ; 9 I�B 4 : ; 9 I�B : ; 9 4 : ; 9 I ! U " U #��1 $�� �B %��1 &�� 1 % : ; 9 I $ > $ > & I I : ; 9 : ; 9 I8 : ;9 I8 : ; 9 I ! I/ 7 I 4 : ; 9 I?< < 4 : ;9 I?< : ; 9 4 : ; 9 I > I: ; 9 ( .?: ; 9 'I< I . ?: ; 9 'I< .?: ;9 'I< .?: ;9 '< .?: ; 9 '< .?: ; 9 'I@�B : ; 9 I�B 4 : ; 9 I 4 : ; 9 I�B !4 : ; 9 I "4 : ; 9 I�B # U $ U %��1 &�� �B '�� 1 (��1 ) : ; 9 * 1R�B UX Y W + 1�B , 1R�B X Y W -.: ; 9 'I . : ; 9 I / : ; 9 I 0. ?<n: ; % : ; 9 I $ > $ > & I I : ; 9 : ; 9 I8 7 I < > I: ; 9 ( ( ( I ! I/ : ; 9 : ; 9 I8 : ; 9 : ; 9 I : ; 9 I 4 : ; 9 I .?: ;9 'I< I .?: ; 9 'I< . ?: ; 9 'I< .?: ; 9 'I@�B : ; 9 I�B 4 : ; 9 I 4 : ; 9 I�B !4 : ; 9 I�B "4 : ; 9 I # : ; 9 $ 1R�B UX Y W % 1�B &��1 '�� �B (�� 1 )��1 *.: ; 9 'I + : ; 9 I ,. ?<n: ; % $ > $ > & I : ; 9 I I : ; 9 : ; 9 I8 7 I < 4 : ; 9 I?< : ; 9 I8 I ! I/ ! I7 4 : ; 9 I .?: ; 9 'I< I .?: ; 9 'I@�B : ; 9 I�B 4 : ; 9 I ��1 �� �B �� 1 % : ; 9 I $ > $ > & I I : ; 9 : ; 9 I8 : ;9 I8 : ; 9 I ! I/ 7 I 4 : ; 9 I?< < < 4 : ; 9 I .?: ; 9 'I< I .?: ;9 '< .?: ;9 'I< .?: ; 9 '< . ?: ; 9 'I< .?: ; 9 'I@�B : ; 9 I�B 4 : ; 9 I�B 4 : ; 9 I : ; 9 U �� 1 !��1 "�� �B #��1 $ U % & '. ?<n: ; k . � psutil /usr/include/bits /usr/include /usr/include/sys /opt/python/cp36-cp36m/include/python3.6m _psutil_common.c types.h stdio.h types.h pyport.h time.h object.h pyerrors.h _psutil_common.h stdlib.h modsupport.h abstract.h string.h errno.h = �$ � =w: �X:X<� �� I=� X t � � � �$n$\$��� � ;=� X t � � � �#�#\#��� � ;=� X t � � � �8�8yJ Ct�L��z�4 �Y2�2J @t�L � M t��t� .�� K� Y � � � psutil /opt/rh/devtoolset-10/root/usr/lib/gcc/x86_64-redhat-linux/10/include /usr/include/bits /usr/include /usr/include/sys /opt/python/cp36-cp36m/include/python3.6m /usr/include/netinet /usr/include/net /usr/include/asm-generic /usr/include/linux _psutil_posix.c stddef.h types.h stdio.h types.h unistd.h stdint.h pyport.h time.h object.h boolobject.h methodobject.h moduleobject.h pyerrors.h resource.h resource.h socket_type.h sockaddr.h socket.h in.h confname.h if.h ifaddrs.h int-ll64.h types.h if_packet.h _psutil_common.h netdb.h listobject.h unicodeobject.h modsupport.h ioctl.h socket.h longobject.h signal.h errno.h <built-in> 3 �' �3v J3v� J�=-[ /_f sY � t.Z] ' Y �X:�t:z P:z� P�=-[ /[f sY � u.[Y uR� Y �X ztPX:�|f: J At=-^�ZtKp� t�: �:s .:s. .:s< � Y � 8 @ � -�� ^� ^X # Y Y�� �L IY \K� Y `� lt�95`5 25* �XY� ��XJ �� ; B ' � � � � � � � % L) J � J Z K �q ? 0 �� = � < � �tK <�<XX KXK � X � X � X � X � X � � l� � � � � � � � � � % L) J 7�X� ]� �tK < . t X J t � � � � � � � � � � �z��� A�o;;- D�v <:�~t:� Z�h >�-^�.L �fp. �f5��5 35) �Y� t��J� =�~� /�f sY ��u X��������������� ��~�� � /Ij��t�� �~D�� � /Ij��� 7��� � /I2��� =��� � /I2��� � ����� / ;Z� ����� � /Ij��� � ��� � /Ij��� � ��� � /Ij��X � � B��~�� � /Ij��� � �~�� � /Ij��� � ��� � /Ij��� � ��� � /Ij��� � ��� � /Ij��� � ��� � /Ij��� � ��� � /Ij��� � ��� � /Ij��� ��~�� � /Ij��X ���}� >� �> KXu I ��L .j� l Jt=W [ X .\dX &�h� ./�/ �X g EA � t K G? �= I/ X 7 � / � X� <� � <� <� <� <� <� <� <� <� <� <N <N <N <J � Y���� < � J � psutil /usr/include/bits /usr/include /opt/python/cp36-cp36m/include/python3.6m /usr/include/sys psutil/arch/linux _psutil_linux.c types.h stdio.h pyport.h time.h object.h methodobject.h moduleobject.h _psutil_common.h mem.h net.h users.h disk.h proc.h modsupport.h 9 � / _ XK K <K <LZX pf< � � � psutil/arch/linux /opt/rh/devtoolset-10/root/usr/lib/gcc/x86_64-redhat-linux/10/include /usr/include/bits /usr/include /usr/include/sys /opt/python/cp36-cp36m/include/python3.6m psutil/arch/linux/../.. users.c stddef.h types.h stdio.h types.h pyport.h time.h object.h utmp.h listobject.h modsupport.h unicodeobject.h utmp.h _psutil_common.h . �9 . 0., �X=�Y � ` � � h3JX = �.JX = � � � � � �� � � � � � � � � � ` �"�Y / ]. t, . � � u�#<� � X � X t t Y < q� � � � � � �� > : X Y U � � � � � ��� � � � psutil/arch/linux /opt/rh/devtoolset-10/root/usr/lib/gcc/x86_64-redhat-linux/10/include /usr/include/bits /usr/include /usr/include/sys /opt/python/cp36-cp36m/include/python3.6m psutil/arch/linux/../.. proc.c stddef.h types.h stdio.h libio.h types.h pyport.h time.h object.h pyerrors.h sched.h _psutil_common.h sched.h longobject.h abstract.h listobject.h errno.h unistd.h modsupport.h <built-in> 8 @< )8J ?t�=-Zg .Z - 0� e >Y �t�88 J AtX>,\U) WJ)� <WJX+ZtK �t> >y _>y� {� � o< uWu; K � � . �� X � g� :� q � � �� Wft � �t�[ hJX='.�1'< < X!!X = ��/; j � . v1 ' < J. �� _ f � b $ � ( >m>y _>y� ��� �����f f .� z f X ] � J t J Y � [ �W = . � . XXj#. `<( Nu�.� i ? � psutil/arch/linux /opt/rh/devtoolset-10/root/usr/lib/gcc/x86_64-redhat-linux/10/include /usr/include/bits /usr/include /opt/python/cp36-cp36m/include/python3.6m /usr/include/sys /usr/include/asm-generic /usr/include/linux/hdlc /usr/include/linux psutil/arch/linux/../.. net.c stddef.h types.h stdio.h stdint.h pyport.h time.h object.h sockaddr.h socket.h int-ll64.h ioctl.h if.h ethtool.h unistd.h modsupport.h ioctl.h string.h _psutil_common.h socket.h socket_type.h errno.h <built-in> < A 9<u X<u� <t< J�� /�i Mzt<� YzX[� [Z YS)XX- [L 1 O�� �=Zu h�X .�- e�.�RJ.< q�� / ;Y� � o � psutil/arch/linux /usr/include/bits /usr/include /opt/python/cp36-cp36m/include/python3.6m /usr/include/sys /usr/include/asm-generic /usr/include/linux psutil/arch/linux/../.. mem.c types.h stdio.h pyport.h time.h object.h int-ll64.h posix_types.h sysinfo.h modsupport.h pyerrors.h sysinfo.h _psutil_common.h 6 `B 6 M�M 2 t� � � � psutil/arch/linux /opt/rh/devtoolset-10/root/usr/lib/gcc/x86_64-redhat-linux/10/include /usr/include/bits /usr/include /opt/python/cp36-cp36m/include/python3.6m /usr/include/sys psutil/arch/linux/../.. disk.c stddef.h types.h stdio.h libio.h pyport.h time.h object.h pystate.h pyerrors.h mntent.h _psutil_common.h listobject.h modsupport.h unicodeobject.h ceval.h <built-in> 8 �B 8y 58y. �Z� t�� Y�lt= Y ;=X . j � �� = �� = �.X A � � � t f � J � f � f j ��� b� = t � =yJ �� f X f � � �t�� \ $ �!. �X� tz_dsttime syscall psutil_setup /project _typeobject _object ob_refcnt double NoSuchProcess Py_ssize_t float _py_xdecref_tmp long long unsigned int timezone unsigned char psutil_set_debug strerror _py_decref_tmp PSUTIL_DEBUG short unsigned int PyObject_IsTrue psutil_PyErr_SetFromOSErrnoWithSyscall tz_minuteswest PyObject_CallFunction AccessDenied psutil/_psutil_common.c ob_type self PyErr_SetObject __errno_location PyObject __ssize_t fullmsg long long int GNU C17 10.2.1 20210130 (Red Hat 10.2.1-11) -mtune=generic -march=x86-64 -g -O3 -fwrapv -fPIC PyArg_ParseTuple PyExc_ValueError _Py_Dealloc short int args PSUTIL_CONN_NONE __pid_t _Py_NoneStruct psutil_check_pid_range PyErr_SetString sprintf getenv PyExc_OSError _SC_EQUIV_CLASS_MAX ifr_ifrn m_copy ifa_addr _SC_THREAD_PRIO_PROTECT sin6_flowinfo ifr_ifru PyModule_Create2 getnameinfo __priority_which_t _SC_VERSION Py_BuildValue _SC_NL_NMAX m_methods _longobject slot _SC_SYNCHRONIZED_IO _SC_THREAD_PRIORITY_SCHEDULING _SC_NPROCESSORS_ONLN sa_data m_base _SC_TIMEOUTS sockaddr _SC_BASE _SC_PII_OSI_COTS sockaddr_un _SC_THREAD_SAFE_FUNCTIONS _SC_ATEXIT_MAX sockaddr_ns _SC_STREAM_MAX ifru_slave psutil_posix_getpriority _SC_PRIORITIZED_IO py_ptp _SC_V6_ILP32_OFF32 _SC_THREAD_SPORADIC_SERVER sll_pkttype __socket_type ifu_dstaddr _SC_SHRT_MIN _SC_USHRT_MAX _SC_NL_TEXTMAX _SC_STREAMS __rlimit_resource IFF_PORTSEL _SC_THREAD_DESTRUCTOR_ITERATIONS _SC_PIPE _SC_BC_DIM_MAX psutil_getpagesize _SC_MAPPED_FILES _SC_2_C_BIND _SC_MQ_OPEN_MAX _SC_XOPEN_SHM _SC_INT_MAX _SC_2_FORT_DEV kill _SC_XOPEN_XPG2 _SC_NZERO _SC_XOPEN_XPG4 IFF_NOTRAILERS _SC_CPUTIME _SC_PII_INTERNET _SC_V7_LP64_OFF64 _SC_LEVEL3_CACHE_SIZE _SC_MB_LEN_MAX ifreq __u6_addr16 __caddr_t _py_tmp sockaddr_ipx sll_hatype getifaddrs ifru_broadaddr _SC_REALTIME_SIGNALS __RLIMIT_LOCKS m_free moduledef py_netmask _SC_DEVICE_SPECIFIC_R uint32_t in_addr_t mem_start _SC_SAVED_IDS __RLIM_NLIMITS _SC_2_C_DEV _SC_XBS5_LPBIG_OFFBIG ob_base _SC_2_C_VERSION _SC_SEM_VALUE_MAX _SC_SCHAR_MAX ifru_map _SC_SSIZE_MAX _SC_2_UPE PyExc_RuntimeError freeifaddrs RLIMIT_CORE ifa_name _SC_INT_MIN _SC_BC_BASE_MAX __u6_addr8 _SC_POLL sysconf ml_flags _SC_SYSTEM_DATABASE_R sockaddr_dl sll_halen __RLIMIT_NPROC _SC_2_LOCALEDEF IFF_MULTICAST sin_family _SC_XOPEN_XPG3 _SC_T_IOV_MAX _SC_LEVEL1_ICACHE_ASSOC in_port_t py_str _SC_SYMLOOP_MAX retval _SC_TRACE_LOG ifa_ifu __u6_addr32 append_flag _SC_THREAD_CPUTIME sin_zero _SC_CHAR_MAX _SC_XBS5_ILP32_OFFBIG _SC_PII_INTERNET_DGRAM s_addr py_address _SC_2_PBS_TRACK __u16 psutil_net_if_flags _SC_FILE_ATTRIBUTES _SC_ASYNCHRONOUS_IO _SC_FSYNC __RLIMIT_NICE sockaddr_inarp _SC_DEVICE_SPECIFIC sockaddr_ax25 ifa_netmask IFF_UP _SC_THREAD_ATTR_STACKSIZE _SC_REGEX_VERSION _SC_MEMLOCK _SC_DELAYTIMER_MAX _SC_LONG_BIT _SC_SEM_NSEMS_MAX sockaddr_iso PyModule_AddObject PyModule_AddIntConstant _SC_XOPEN_STREAMS _SC_LEVEL1_ICACHE_LINESIZE _SC_SIGNALS sll_family __RLIMIT_OFILE IFF_SLAVE ioctl _SC_2_PBS_ACCOUNTING _SC_AIO_MAX _SC_LEVEL2_CACHE_LINESIZE _SC_XOPEN_VERSION mod_methods sa_family_t _SC_CLK_TCK _SC_SHELL sll_protocol _SC_UIO_MAXIOV _SC_TZNAME_MAX ifru_data _SC_SPORADIC_SERVER _SC_MEMLOCK_RANGE _SC_AVPHYS_PAGES __RLIMIT_NLIMITS _SC_V7_ILP32_OFFBIG _SC_PII_XTI close sll_ifindex flag_name sock _SC_V7_LPBIG_OFFBIG _SC_C_LANG_SUPPORT_R SOCK_DCCP _SC_LEVEL3_CACHE_ASSOC uint8_t _SC_FILE_SYSTEM ifru_netmask _SC_PAGESIZE _SC_V6_ILP32_OFFBIG SOCK_PACKET __id_t _SC_SIGQUEUE_MAX _SC_SPAWN psutil_posix_setpriority _SC_DEVICE_IO _SC_V6_LPBIG_OFFBIG _SC_2_VERSION ifru_mtu _SC_LEVEL4_CACHE_SIZE ml_name SOCK_DGRAM m_size IFF_LOOPBACK _SC_USER_GROUPS_R PyModuleDef_Slot visitproc sin_port _SC_LINE_MAX sockaddr_eon psutil/_psutil_posix.c IFF_AUTOMEDIA lladdr py_broadcast _SC_LEVEL1_DCACHE_ASSOC _SC_HOST_NAME_MAX _SC_C_LANG_SUPPORT __RLIMIT_RSS _SC_THREAD_STACK_MIN IFF_RUNNING _SC_SEMAPHORES _SC_UINT_MAX _SC_CHILD_MAX __RLIMIT_MSGQUEUE PyUnicode_FromString _SC_NGROUPS_MAX _SC_SINGLE_PROCESS IFF_ALLMULTI __be16 __in6_u sin_addr _SC_TTY_NAME_MAX _SC_PII_INTERNET_STREAM RLIMIT_AS _SC_MEMORY_PROTECTION PyModuleDef_Base m_name _SC_XOPEN_CRYPT PyCFunction psutil_net_if_is_running ifru_newname _Py_FalseStruct RLIMIT_FSIZE _SC_CHAR_BIT _SC_LEVEL1_DCACHE_SIZE IFF_NOARP _SC_CLOCK_SELECTION m_init ifrn_name sin6_addr psutil_net_if_mtu _SC_TIMERS freefunc _SC_BC_SCALE_MAX _SC_ULONG_MAX uint16_t sin6_port _SC_MQ_PRIO_MAX _SC_TRACE _SC_SPIN_LOCKS ifru_flags addrlen _SC_LEVEL1_DCACHE_LINESIZE _SC_BC_STRING_MAX PyLong_FromLong mem_end SOCK_STREAM RLIMIT_DATA _SC_V7_ILP32_OFF32 _SC_TRACE_SYS_MAX _SC_FD_MGMT _SC_REGEXP sin6_family IFF_PROMISC RLIMIT_CPU PyInit__psutil_posix _SC_LEVEL1_ICACHE_SIZE _SC_ADVISORY_INFO __RLIMIT_RTPRIO _SC_SHRT_MAX _SC_XBS5_LP64_OFF64 __builtin_strncpy _SC_READER_WRITER_LOCKS _SC_SYSTEM_DATABASE _SC_XOPEN_REALTIME_THREADS _SC_THREAD_ROBUST_PRIO_PROTECT _SC_2_CHAR_TERM _SC_PASS_MAX _SC_FIFO m_slots _SC_ARG_MAX ifru_hwaddr ml_doc _SC_2_PBS_CHECKPOINT PyMethodDef _SC_XOPEN_REALTIME PRIO_USER ifru_dstaddr psutil_getpagesize_pywrapper _SC_2_FORT_RUN socket _SC_TRACE_EVENT_FILTER psutil_raise_for_pid _SC_RE_DUP_MAX _SC_SCHAR_MIN sockaddr_in _SC_THREAD_ROBUST_PRIO_INHERIT _SC_THREADS PyList_Append _SC_PII __RLIMIT_SIGPENDING _SC_TRACE_INHERIT _SC_WORD_BIT _SC_XBS5_ILP32_OFF32 _SC_PII_OSI_M RLIMIT_NOFILE m_traverse _SC_2_SW_DEV _SC_OPEN_MAX _SC_XOPEN_UNIX sockaddr_in6 _SC_AIO_LISTIO_MAX PyErr_SetFromErrno _SC_PII_OSI _SC_UCHAR_MAX _SC_MONOTONIC_CLOCK m_clear py_tuple ifaddr SOCK_RAW _SC_PRIORITY_SCHEDULING m_doc _SC_SELECT _SC_NETWORKING _SC_TIMER_MAX _SC_TRACE_EVENT_NAME_MAX _SC_V6_LP64_OFF64 _SC_GETGR_R_SIZE_MAX _SC_LEVEL2_CACHE_SIZE _SC_LOGIN_NAME_MAX _SC_EXPR_NEST_MAX IFF_POINTOPOINT _SC_NPROCESSORS_CONF __RLIMIT_RTTIME ifru_addr __socklen_t traverseproc SOCK_SEQPACKET PRIO_PROCESS sockaddr_at psutil_pid_exists ifa_flags _SC_SS_REPL_MAX _SC_RAW_SOCKETS _SC_BARRIERS _SC_CHAR_MIN _SC_LEVEL4_CACHE_LINESIZE psutil_convert_ipaddr _SC_THREAD_PROCESS_SHARED _SC_NL_MSGMAX _Py_TrueStruct sin6_scope_id m_index _SC_LEVEL4_CACHE_ASSOC _SC_CHARCLASS_NAME_MAX _SC_PII_OSI_CLTS psutil_net_if_addrs py_retlist _SC_TYPED_MEMORY_OBJECTS ml_meth __RLIMIT_MEMLOCK base_addr _SC_2_PBS _SC_PHYS_PAGES IFF_MASTER _SC_PII_SOCKET IFF_DEBUG _SC_MULTI_PROCESS _SC_LEVEL2_CACHE_ASSOC __priority_which sa_family ifmap _SC_THREAD_KEYS_MAX PyErr_Format IFF_BROADCAST _SC_THREAD_THREADS_MAX _SC_GETPW_R_SIZE_MAX ifa_data SOCK_NONBLOCK PRIO_PGRP _SC_2_PBS_MESSAGE _SC_RTSIG_MAX sockaddr_ll _SC_THREAD_ATTR_STACKADDR PyList_New PyModuleDef _SC_FILE_LOCKING _SC_SHARED_MEMORY_OBJECTS _SC_TRACE_NAME_MAX _SC_COLL_WEIGHTS_MAX _SC_XOPEN_ENH_I18N _SC_XOPEN_LEGACY _SC_JOB_CONTROL IFF_DYNAMIC ifa_next RLIMIT_STACK _SC_MESSAGE_PASSING _SC_NL_LANGMAX _SC_IOV_MAX _SC_USER_GROUPS SOCK_CLOEXEC _SC_IPV6 _SC_LEVEL3_CACHE_LINESIZE _SC_2_PBS_LOCATE ifru_ivalue _SC_NL_SETMAX _SC_NL_ARGMAX ifu_broadaddr nic_name _SC_THREAD_PRIO_INHERIT sll_addr _SC_TRACE_USER_EVENT_MAX inquiry sockaddr_x25 SOCK_RDM _SC_AIO_PRIO_DELTA_MAX _SC_XOPEN_XCU_VERSION psutil_proc_cpu_affinity_get psutil_users psutil_disk_partitions psutil_proc_ioprio_set psutil_net_if_duplex_speed psutil_linux_sysinfo psutil_proc_ioprio_get psutil_proc_cpu_affinity_set PyInit__psutil_linux psutil/_psutil_linux.c py_username __unused ut_line exit_status tv_sec __s2_len psutil/arch/linux/users.c __s1 ut_id ut_host ut_addr_v6 ut_type setutent ut_user py_hostname __result ut_tv PyUnicode_DecodeFSDefault endutent utmp py_tty e_exit ut_session ut_pid __s1_len ut_exit getutent tv_usec e_termination __off_t _IO_read_ptr _chain PySequence_GetItem _shortbuf ncpus cpucount_s _IO_buf_base __builtin_fwrite __bits _fileno __builtin_fputc _IO_lock_t PyExc_OverflowError PyErr_NoMemory _IO_buf_end _cur_column PySequence_Size _old_offset PyLong_AsLong _IO_marker fprintf _IO_write_ptr _sbuf sched_getaffinity _IO_save_base __sched_cpufree _lock PyErr_Occurred _flags2 _mode _IO_read_end _IO_write_end item iodata _IO_FILE IOPRIO_WHO_PROCESS _pos _markers py_list cpu_num setsize py_cpu_set seq_len _vtable_offset __cpu_mask ioprio __sched_cpualloc PySequence_Check __cpu __off64_t _IO_read_base _IO_save_end __pad1 __pad2 __pad3 __pad4 __pad5 psutil/arch/linux/proc.c _unused2 stderr _IO_backup_base cpu_set_t ioclass __sched_cpucount PyExc_TypeError sched_setaffinity _IO_write_base clock_type mdio_support psutil_ethtool_cmd_speed ifru_settings maxrxpkt raw_hdlc_proto speed_hi dlci ethcmd ifs_ifsu parity fr_proto_pvc reserved autoneg maxtxpkt __u8 __u32 timeout phy_address t391 t392 cisco_proto ethtool_cmd lp_advertising encoding eth_tp_mdix fr_proto_pvc_info eth_tp_mdix_ctrl interval master psutil/arch/linux/net.c raw_hdlc fr_proto cisco if_settings te1_settings clock_rate loopback ecmd supported n391 n392 n393 slot_map sync_serial_settings fr_pvc_info transceiver duplex memset fr_pvc uint_speed sync procs loads freehigh totalram uptime __kernel_long_t bufferram sharedram totalhigh freeswap __kernel_ulong_t totalswap mem_unit freeram psutil/arch/linux/mem.c py_dev PyThreadState endmntent PyEval_SaveThread mnt_dir setmntent mnt_fsname PyEval_RestoreThread mnt_passno mnt_opts py_mountp entry file PyErr_SetFromErrnoWithFilename mnt_type _save mtab_path psutil/arch/linux/disk.c getmntent mnt_freq 0 7 U7 � �U� 0 > T> I UI � �T� X c P � � U� * �U� � � T� � U� * �T� 0 Q UQ � �U� x P � V� � V � � V� � V � � V� � V � � U� & �U� � � P� V # V � V # V � � V # V U G \G � �U� ^ e Pe | V� � V f | V� � V k r V� � V c � P� A \ + > P � � U� � �U� � � T� � U� � �T� � � ��� P B VH T UT i Vj � V % PH T P P T UT E �U� P b Tb � V� � �T�� � V� E �T� P � ��� � P� � V� �� P Q VV J ��J W VW E �� � � P! 3 P c r Pr � \� D \V E \ � � V� VV J VW * V � � \� � \4 D \J W \ � � RQ �* @ RQ � � � \* @ \ � � 0�� � P� � V� � 0�* @ V * E V � � V � � V � E �P � � E \ � 0� " P" / ]/ E 0� / J ] / J ] J � �P � J � \ J _ 0�_ r Pr { ]{ � 0� { � ] { � ] � � �S � � � \ � � 0�� � P� � ]� � 0� � � ] � � ] � % HQ � � % \ � � 0�� P ] % 0� � ] * ] * ] V � �P � V � \ V o 0�o � P� � ]� � 0� � � ] � � ] � � �P � � � \ � � 0�� � P� � ]� � 0� � � ] � � ] � E 0Q � � E \ � 0� " P" / ]/ E 0� / J ] / J ] W � 6Q � W � \ W o 0�o � P� � ]� � 0� � � ] � � ] � � @Q � � � \ � � 0�� � P� � ]� � 0� � � ] � � ] � E Q � � E \ � 0� " P" / ]/ E 0� / J ] / J ] J � �R � J � \ J _ 0�_ r Pr ] � 0� � ] � ] � � Q � � � \ � � 0�� � P� � ]� � 0� � � ] � � ] � 5 Q � � 5 \ � � 0�� P ] 5 0� : ] : ] : � Q � : � \ : O 0�O b Pb o ]o � 0� o � ] o � ] � � )Q � � � \ � � 0�� � P� � ]� � 0� � � ] � � ] U � �U� T U � �T� 8 ��8 S PS � V� � U� � V j u P� � P U � �U� $ T$ � �T� K V � V� � Vm � V Y � ^ ^ ^� � ^m � ^ * < P< �� � �� * � 0�� � P� � Z� � ��� 0� � Z� � �� � 0�m � 0�� � S� � 0�� � Z� � 0� * Y 0�Y � \� � 0�( > P> � \ ( \( 0�� m \m � P� � \� � 0� * Y 0�Y � S� B 0�B N PN � S ; S; � 0�� m Sm � 0�� � S� � P� � 0� * s 0�y � _� e 0�e l Pl | _ � _ M _M � 0�� m _m � 0�� � S� � 0�� � Z� � 0� * s 0�y � ]� p 0� � ] _ ]_ � 0�� � P� m ]m � 0�� � S� � 0�� � Z� � 0� � � Z� � �� ' Z' �� � � Z� � �� ' Z' �� � � \( \ � � \( \ � � S; S � � S; S � � _M _ � � _M _ � � ]_ ] � � ]_ ] � � ��� � Q� m �� � � Z� � Z� Z � � Z� Z � � \� � \ m \ � � \ 6 \ � � S� � S( m S � � S6 V S � � _� � _H m _ � � _V m _ � � ]� � ] � � ]� � ] ` h Uh ~ �U� ` h Th ~ �T� � � U� � S� �U� S ! �U�! C SC I �U� � � T� V U �T� V U ! �T�! D VD H QH I �T� � � U� � �U� � � T� � U� � �T� � � P� � P � � �U� � � �T� � � U� F �U� � � T� . V. / �T�/ E VE F �T� P ' T/ > T / ? �U� / ? V � � U� � �U� � � P� � X � � �U� . P. � \� � P U � �U� T � �T� 8 O PO * S* < P< = SO � S % P% = ]O � ]� ] " U- � ] � 0�� � P� \ = 0�O � 0�� � \� � \� 0�> F \j | 0� S 0�S d Pd � _� = 0�O � _� 0� � _ h 0�h � P� � ^� = 0�O h Ph � ^� � ^� 0�> F ^j | 0� � 0�� � P� V = 0�O } 0�} � P� � V� � V� 0�> F Vj | 0� q 3 2�O � 2�� 2�- j 2� q * s� �O � s� �� s� �- j s� � q { s� ��: �{ � s� ��0 �� � s� ��� O � 4� � 3 _� _ � 3 _� _ � 3 ^� ^ � 3 ^� ^ 3 V� V 3 V� V 3 \� \ 3 \� \ � � _> F _j | _ � � _- j _j � _ � � ^F ] ^| | 0� � � ^- > ^F j ^ � � V- > V] j V| � 0� � � V] j V � � \- > \| � 0� � � \- > \ � � ] ] " U5 > ] 0 5 U5 � �U� 0 > T> [ U[ � �T� c � ��� � �� � � 0�� * S� � S � � P� * \c � \ � � V� P * V� � V � � _ P * _� � _ � � V * V� � V � � V * V� � V � � _ U & �U� T / U/ & �T� � � S1 7 0�7 � S� � S & S 9 � ]� � A�� ] A� ] ) A�� � ]� � A�� ] � � \1 7 P7 � \� � |�� � \� � \ & \ 1 7 P 9 C \] � \� 1 \� � \� \ 9 0�9 C V] t Pt � V� � V� � P� & V � 0�� � ]� 0� - P- � ]� � 0�� � ]� 0� & ] � � s $ &�7 C UF a Ua � s $ &�� � s $ &�� � s $ &� & s $ &� � � ^e x Px � ^� � ^ & ^ � � ^ � � ^� � ^ � � ]� � 0� & ] � � ] & ] � � U� � �U� � � T� � U� � �T� � � �h�=$�l�!�� � R � � P� � P � � �h�=$�l�!�� � R � � �d � � 1� U t �U� T U t �T� 7 I PI O QO R T\ g P < I p =&�I O q =&�O R t =&�R V T < I p � �I O q � �O V Q $ 6 �l $ 7 1� U Q �U� T U Q �T� ; 0�; Z PZ � V� V P $ V* N V � � P� � P � � T � � �L� ��@$��� ��!� � � Q � 0�� � P� � \� < 0� � � ��� U d �U� T d �T� U z �U� T ] V] z �T� h 0�h l Pl � ]� z ] � � P� � SG W PW Z S� S_ m S � 0�� � P� � V� � 0�� V m 0�m z V � 0�� � P� � Q� � _� 0� ( _( � 0�� � _� m 0�m z _ � 0�� � P� S 0� 6 S6 � 0�� � S� m 0�m z S + P+ � \� z \ ] a Pa v VZ � V _ V � P V � P V � _( P _ � _( P _ S6 G S S6 G S � � 0�� � V� Vh m 0�m z V � � V� Vm z V � � 0�� � _ 0�h m 0�u z _ � � _ � � 0�� � Sh m 0� � � S � � \ f r � � k r � � � � � � � � � � � � � � [ f � � � � ( � � ( ; � � ; M � � M _ � � _ r � � � � � � � � � � � � ( � � ( � � ( H � � @ H � � H p � � ` p � � � � � � � � � 4 ? J R � � * E � � * E � � * � 0 0 " � � � � � � � � � � � � > F j | � � F ] | | � � F ] � � ] j | | � � ] j � � - 5 | � � � - 5 � � # 5 > � � � � � � � � ] � � � � � { ~ � � � � � � � � & � � & % ) C ] � � � * � � � � � � � � � � � � � � B S Z m � � 6 < � � ( � ( 6 � � � � � h h m u � � � m u � � � � � � h h u z � � � � � � � � h m 8 ` � � x � 0$ LE P �T �U �m �m �m �m �o p r �u � 0$ `$ ! �$ 7 �u C �m j �$ v �m � � � � � �' � � `( � � @) a � �) �* � $ �. v = / � Q 6 n r h x �r � � n �s h x t ` � � � � � � � � � � � � d[ � � �m � �m � �T � �u � p # . ? V n � � � � � � 9 � . > Z @= & w p? � � LE � A Q � � 7 � � � ' 6 : M ` s � � � � � � @< t ' < L a p � � � � � � � �6 i � �< y �$ � ; Q h � � � � � 06 W � �B z �% � 3 B Y o ' W � � � � � & � � � �u �9 � % D Q f z � � � �& j � " � �' ) # `B d 8 crtstuff.c deregister_tm_clones __do_global_dtors_aux completed.0 __do_global_dtors_aux_fini_array_entry frame_dummy __frame_dummy_init_array_entry _psutil_common.c _psutil_posix.c psutil_net_if_mtu psutil_net_if_is_running psutil_posix_setpriority psutil_convert_ipaddr psutil_net_if_addrs psutil_posix_getpriority psutil_net_if_flags psutil_getpagesize_pywrapper moduledef mod_methods _psutil_linux.c users.c proc.c net.c mem.c disk.c __FRAME_END__ __dso_handle _DYNAMIC __GNU_EH_FRAME_HDR __TMC_END__ _GLOBAL_OFFSET_TABLE_ getenv@@GLIBC_2.2.5 PyList_New PySequence_Check endmntent@@GLIBC_2.2.5 PyModule_AddIntConstant __errno_location@@GLIBC_2.2.5 strncpy@@GLIBC_2.2.5 getpriority@@GLIBC_2.2.5 _ITM_deregisterTMCloneTable PyEval_RestoreThread PyErr_SetFromErrno PyInit__psutil_linux _Py_Dealloc PyErr_NoMemory PyErr_SetObject __sched_cpucount@@GLIBC_2.6 psutil_proc_cpu_affinity_get psutil_proc_cpu_affinity_set _fini psutil_net_if_duplex_speed PyExc_RuntimeError PyInit__psutil_posix PyErr_SetString PyExc_ValueError PyExc_TypeError setpriority@@GLIBC_2.2.5 psutil_getpagesize ioctl@@GLIBC_2.2.5 close@@GLIBC_2.2.5 PySequence_GetItem getnameinfo@@GLIBC_2.2.5 sched_setaffinity@@GLIBC_2.3.4 fputc@@GLIBC_2.2.5 endutent@@GLIBC_2.2.5 PyLong_FromLong PySequence_Size psutil_proc_ioprio_get PyList_Append PyExc_OSError fprintf@@GLIBC_2.2.5 _Py_FalseStruct syscall@@GLIBC_2.2.5 __gmon_start__ PyObject_CallFunction PyExc_OverflowError Py_BuildValue kill@@GLIBC_2.2.5 PyErr_Occurred PyModule_Create2 PyLong_AsLong psutil_raise_for_pid psutil_proc_ioprio_set psutil_PyErr_SetFromOSErrnoWithSyscall getutent@@GLIBC_2.2.5 getmntent@@GLIBC_2.2.5 setmntent@@GLIBC_2.2.5 _Py_NoneStruct getifaddrs@@GLIBC_2.3 __sched_cpufree@@GLIBC_2.7 PyObject_IsTrue PyArg_ParseTuple psutil_pid_exists psutil_disk_partitions NoSuchProcess __sched_cpualloc@@GLIBC_2.7 _Py_TrueStruct freeifaddrs@@GLIBC_2.3 setutent@@GLIBC_2.2.5 psutil_set_debug PyUnicode_DecodeFSDefault PyUnicode_FromString PyEval_SaveThread PyModule_AddObject AccessDenied sysconf@@GLIBC_2.2.5 PSUTIL_DEBUG psutil_users sysinfo@@GLIBC_2.2.5 PyErr_SetFromErrnoWithFilename PyErr_Format sprintf@@GLIBC_2.2.5 fwrite@@GLIBC_2.2.5 _ITM_registerTMCloneTable sched_getaffinity@@GLIBC_2.3.4 strerror@@GLIBC_2.2.5 psutil_check_pid_range __cxa_finalize@@GLIBC_2.2.5 _init psutil_setup stderr@@GLIBC_2.2.5 psutil_linux_sysinfo socket@@GLIBC_2.2.5 .symtab .strtab .shstrtab .note.gnu.build-id .gnu.hash .dynsym .dynstr .gnu.version .gnu.version_r .rela.dyn .rela.plt .init .text .fini .rodata .eh_frame_hdr .eh_frame .init_array .fini_array .data.rel.ro .dynamic .got .got.plt .data .bss .comment .debug_aranges .debug_info .debug_abbrev .debug_line .debug_str .debug_loc .debug_ranges 8 8 $ . ���o ` ` � 8 � @ � � � H ���o � � � U ���o x x � d � � ( n B x s ~ 0$ 0$ ! � LE LE � 2 P P � � �T �T � � �U �U � � �m �] � �m �] � �m �] � �m �] � � �o �_ p � p ` � r b ` � �u �e � 0 �e / � �e � /g Sw �� � # 9� �, / 0 � �# : �B fP E F� � (� ! D (� L t� S __init__.py 0000644 00000251214 15030211765 0006662 0 ustar 00 # Copyright (c) 2009, Giampaolo Rodola'. All rights reserved. # Use of this source code is governed by a BSD-style license that can be # found in the LICENSE file. """psutil is a cross-platform library for retrieving information on running processes and system utilization (CPU, memory, disks, network, sensors) in Python. Supported platforms: - Linux - Windows - macOS - FreeBSD - OpenBSD - NetBSD - Sun Solaris - AIX Supported Python versions are cPython 3.6+ and PyPy. """ import collections import contextlib import datetime import functools import os import signal import socket import subprocess import sys import threading import time try: import pwd except ImportError: pwd = None from . import _common from ._common import AIX from ._common import BSD from ._common import CONN_CLOSE from ._common import CONN_CLOSE_WAIT from ._common import CONN_CLOSING from ._common import CONN_ESTABLISHED from ._common import CONN_FIN_WAIT1 from ._common import CONN_FIN_WAIT2 from ._common import CONN_LAST_ACK from ._common import CONN_LISTEN from ._common import CONN_NONE from ._common import CONN_SYN_RECV from ._common import CONN_SYN_SENT from ._common import CONN_TIME_WAIT from ._common import FREEBSD from ._common import LINUX from ._common import MACOS from ._common import NETBSD from ._common import NIC_DUPLEX_FULL from ._common import NIC_DUPLEX_HALF from ._common import NIC_DUPLEX_UNKNOWN from ._common import OPENBSD from ._common import OSX # deprecated alias from ._common import POSIX from ._common import POWER_TIME_UNKNOWN from ._common import POWER_TIME_UNLIMITED from ._common import STATUS_DEAD from ._common import STATUS_DISK_SLEEP from ._common import STATUS_IDLE from ._common import STATUS_LOCKED from ._common import STATUS_PARKED from ._common import STATUS_RUNNING from ._common import STATUS_SLEEPING from ._common import STATUS_STOPPED from ._common import STATUS_TRACING_STOP from ._common import STATUS_WAITING from ._common import STATUS_WAKING from ._common import STATUS_ZOMBIE from ._common import SUNOS from ._common import WINDOWS from ._common import AccessDenied from ._common import Error from ._common import NoSuchProcess from ._common import TimeoutExpired from ._common import ZombieProcess from ._common import debug from ._common import memoize_when_activated from ._common import wrap_numbers as _wrap_numbers if LINUX: # This is public API and it will be retrieved from _pslinux.py # via sys.modules. PROCFS_PATH = "/proc" from . import _pslinux as _psplatform from ._pslinux import IOPRIO_CLASS_BE # noqa: F401 from ._pslinux import IOPRIO_CLASS_IDLE # noqa: F401 from ._pslinux import IOPRIO_CLASS_NONE # noqa: F401 from ._pslinux import IOPRIO_CLASS_RT # noqa: F401 elif WINDOWS: from . import _pswindows as _psplatform from ._psutil_windows import ABOVE_NORMAL_PRIORITY_CLASS # noqa: F401 from ._psutil_windows import BELOW_NORMAL_PRIORITY_CLASS # noqa: F401 from ._psutil_windows import HIGH_PRIORITY_CLASS # noqa: F401 from ._psutil_windows import IDLE_PRIORITY_CLASS # noqa: F401 from ._psutil_windows import NORMAL_PRIORITY_CLASS # noqa: F401 from ._psutil_windows import REALTIME_PRIORITY_CLASS # noqa: F401 from ._pswindows import CONN_DELETE_TCB # noqa: F401 from ._pswindows import IOPRIO_HIGH # noqa: F401 from ._pswindows import IOPRIO_LOW # noqa: F401 from ._pswindows import IOPRIO_NORMAL # noqa: F401 from ._pswindows import IOPRIO_VERYLOW # noqa: F401 elif MACOS: from . import _psosx as _psplatform elif BSD: from . import _psbsd as _psplatform elif SUNOS: from . import _pssunos as _psplatform from ._pssunos import CONN_BOUND # noqa: F401 from ._pssunos import CONN_IDLE # noqa: F401 # This is public writable API which is read from _pslinux.py and # _pssunos.py via sys.modules. PROCFS_PATH = "/proc" elif AIX: from . import _psaix as _psplatform # This is public API and it will be retrieved from _pslinux.py # via sys.modules. PROCFS_PATH = "/proc" else: # pragma: no cover msg = f"platform {sys.platform} is not supported" raise NotImplementedError(msg) # fmt: off __all__ = [ # exceptions "Error", "NoSuchProcess", "ZombieProcess", "AccessDenied", "TimeoutExpired", # constants "version_info", "__version__", "STATUS_RUNNING", "STATUS_IDLE", "STATUS_SLEEPING", "STATUS_DISK_SLEEP", "STATUS_STOPPED", "STATUS_TRACING_STOP", "STATUS_ZOMBIE", "STATUS_DEAD", "STATUS_WAKING", "STATUS_LOCKED", "STATUS_WAITING", "STATUS_LOCKED", "STATUS_PARKED", "CONN_ESTABLISHED", "CONN_SYN_SENT", "CONN_SYN_RECV", "CONN_FIN_WAIT1", "CONN_FIN_WAIT2", "CONN_TIME_WAIT", "CONN_CLOSE", "CONN_CLOSE_WAIT", "CONN_LAST_ACK", "CONN_LISTEN", "CONN_CLOSING", "CONN_NONE", # "CONN_IDLE", "CONN_BOUND", "AF_LINK", "NIC_DUPLEX_FULL", "NIC_DUPLEX_HALF", "NIC_DUPLEX_UNKNOWN", "POWER_TIME_UNKNOWN", "POWER_TIME_UNLIMITED", "BSD", "FREEBSD", "LINUX", "NETBSD", "OPENBSD", "MACOS", "OSX", "POSIX", "SUNOS", "WINDOWS", "AIX", # "RLIM_INFINITY", "RLIMIT_AS", "RLIMIT_CORE", "RLIMIT_CPU", "RLIMIT_DATA", # "RLIMIT_FSIZE", "RLIMIT_LOCKS", "RLIMIT_MEMLOCK", "RLIMIT_NOFILE", # "RLIMIT_NPROC", "RLIMIT_RSS", "RLIMIT_STACK", "RLIMIT_MSGQUEUE", # "RLIMIT_NICE", "RLIMIT_RTPRIO", "RLIMIT_RTTIME", "RLIMIT_SIGPENDING", # classes "Process", "Popen", # functions "pid_exists", "pids", "process_iter", "wait_procs", # proc "virtual_memory", "swap_memory", # memory "cpu_times", "cpu_percent", "cpu_times_percent", "cpu_count", # cpu "cpu_stats", # "cpu_freq", "getloadavg" "net_io_counters", "net_connections", "net_if_addrs", # network "net_if_stats", "disk_io_counters", "disk_partitions", "disk_usage", # disk # "sensors_temperatures", "sensors_battery", "sensors_fans" # sensors "users", "boot_time", # others ] # fmt: on __all__.extend(_psplatform.__extra__all__) # Linux, FreeBSD if hasattr(_psplatform.Process, "rlimit"): # Populate global namespace with RLIM* constants. from . import _psutil_posix _globals = globals() _name = None for _name in dir(_psutil_posix): if _name.startswith('RLIM') and _name.isupper(): _globals[_name] = getattr(_psutil_posix, _name) __all__.append(_name) del _globals, _name AF_LINK = _psplatform.AF_LINK __author__ = "Giampaolo Rodola'" __version__ = "7.0.0" version_info = tuple(int(num) for num in __version__.split('.')) _timer = getattr(time, 'monotonic', time.time) _TOTAL_PHYMEM = None _LOWEST_PID = None _SENTINEL = object() # Sanity check in case the user messed up with psutil installation # or did something weird with sys.path. In this case we might end # up importing a python module using a C extension module which # was compiled for a different version of psutil. # We want to prevent that by failing sooner rather than later. # See: https://github.com/giampaolo/psutil/issues/564 if int(__version__.replace('.', '')) != getattr( _psplatform.cext, 'version', None ): msg = f"version conflict: {_psplatform.cext.__file__!r} C extension " msg += "module was built for another version of psutil" if hasattr(_psplatform.cext, 'version'): v = ".".join(list(str(_psplatform.cext.version))) msg += f" ({v} instead of {__version__})" else: msg += f" (different than {__version__})" what = getattr( _psplatform.cext, "__file__", "the existing psutil install directory", ) msg += f"; you may try to 'pip uninstall psutil', manually remove {what}" msg += " or clean the virtual env somehow, then reinstall" raise ImportError(msg) # ===================================================================== # --- Utils # ===================================================================== if hasattr(_psplatform, 'ppid_map'): # Faster version (Windows and Linux). _ppid_map = _psplatform.ppid_map else: # pragma: no cover def _ppid_map(): """Return a {pid: ppid, ...} dict for all running processes in one shot. Used to speed up Process.children(). """ ret = {} for pid in pids(): try: ret[pid] = _psplatform.Process(pid).ppid() except (NoSuchProcess, ZombieProcess): pass return ret def _pprint_secs(secs): """Format seconds in a human readable form.""" now = time.time() secs_ago = int(now - secs) fmt = "%H:%M:%S" if secs_ago < 60 * 60 * 24 else "%Y-%m-%d %H:%M:%S" return datetime.datetime.fromtimestamp(secs).strftime(fmt) def _check_conn_kind(kind): """Check net_connections()'s `kind` parameter.""" kinds = tuple(_common.conn_tmap) if kind not in kinds: msg = f"invalid kind argument {kind!r}; valid ones are: {kinds}" raise ValueError(msg) # ===================================================================== # --- Process class # ===================================================================== class Process: """Represents an OS process with the given PID. If PID is omitted current process PID (os.getpid()) is used. Raise NoSuchProcess if PID does not exist. Note that most of the methods of this class do not make sure that the PID of the process being queried has been reused. That means that you may end up retrieving information for another process. The only exceptions for which process identity is pre-emptively checked and guaranteed are: - parent() - children() - nice() (set) - ionice() (set) - rlimit() (set) - cpu_affinity (set) - suspend() - resume() - send_signal() - terminate() - kill() To prevent this problem for all other methods you can use is_running() before querying the process. """ def __init__(self, pid=None): self._init(pid) def _init(self, pid, _ignore_nsp=False): if pid is None: pid = os.getpid() else: if pid < 0: msg = f"pid must be a positive integer (got {pid})" raise ValueError(msg) try: _psplatform.cext.check_pid_range(pid) except OverflowError as err: msg = "process PID out of range" raise NoSuchProcess(pid, msg=msg) from err self._pid = pid self._name = None self._exe = None self._create_time = None self._gone = False self._pid_reused = False self._hash = None self._lock = threading.RLock() # used for caching on Windows only (on POSIX ppid may change) self._ppid = None # platform-specific modules define an _psplatform.Process # implementation class self._proc = _psplatform.Process(pid) self._last_sys_cpu_times = None self._last_proc_cpu_times = None self._exitcode = _SENTINEL self._ident = (self.pid, None) try: self._ident = self._get_ident() except AccessDenied: # This should happen on Windows only, since we use the fast # create time method. AFAIK, on all other platforms we are # able to get create time for all PIDs. pass except ZombieProcess: # Zombies can still be queried by this class (although # not always) and pids() return them so just go on. pass except NoSuchProcess: if not _ignore_nsp: msg = "process PID not found" raise NoSuchProcess(pid, msg=msg) from None self._gone = True def _get_ident(self): """Return a (pid, uid) tuple which is supposed to identify a Process instance univocally over time. The PID alone is not enough, as it can be assigned to a new process after this one terminates, so we add process creation time to the mix. We need this in order to prevent killing the wrong process later on. This is also known as PID reuse or PID recycling problem. The reliability of this strategy mostly depends on create_time() precision, which is 0.01 secs on Linux. The assumption is that, after a process terminates, the kernel won't reuse the same PID after such a short period of time (0.01 secs). Technically this is inherently racy, but practically it should be good enough. """ if WINDOWS: # Use create_time() fast method in order to speedup # `process_iter()`. This means we'll get AccessDenied for # most ADMIN processes, but that's fine since it means # we'll also get AccessDenied on kill(). # https://github.com/giampaolo/psutil/issues/2366#issuecomment-2381646555 self._create_time = self._proc.create_time(fast_only=True) return (self.pid, self._create_time) else: return (self.pid, self.create_time()) def __str__(self): info = collections.OrderedDict() info["pid"] = self.pid if self._name: info['name'] = self._name with self.oneshot(): if self._pid_reused: info["status"] = "terminated + PID reused" else: try: info["name"] = self.name() info["status"] = self.status() except ZombieProcess: info["status"] = "zombie" except NoSuchProcess: info["status"] = "terminated" except AccessDenied: pass if self._exitcode not in {_SENTINEL, None}: info["exitcode"] = self._exitcode if self._create_time is not None: info['started'] = _pprint_secs(self._create_time) return "{}.{}({})".format( self.__class__.__module__, self.__class__.__name__, ", ".join([f"{k}={v!r}" for k, v in info.items()]), ) __repr__ = __str__ def __eq__(self, other): # Test for equality with another Process object based # on PID and creation time. if not isinstance(other, Process): return NotImplemented if OPENBSD or NETBSD: # pragma: no cover # Zombie processes on Open/NetBSD have a creation time of # 0.0. This covers the case when a process started normally # (so it has a ctime), then it turned into a zombie. It's # important to do this because is_running() depends on # __eq__. pid1, ident1 = self._ident pid2, ident2 = other._ident if pid1 == pid2: if ident1 and not ident2: try: return self.status() == STATUS_ZOMBIE except Error: pass return self._ident == other._ident def __ne__(self, other): return not self == other def __hash__(self): if self._hash is None: self._hash = hash(self._ident) return self._hash def _raise_if_pid_reused(self): """Raises NoSuchProcess in case process PID has been reused.""" if self._pid_reused or (not self.is_running() and self._pid_reused): # We may directly raise NSP in here already if PID is just # not running, but I prefer NSP to be raised naturally by # the actual Process API call. This way unit tests will tell # us if the API is broken (aka don't raise NSP when it # should). We also remain consistent with all other "get" # APIs which don't use _raise_if_pid_reused(). msg = "process no longer exists and its PID has been reused" raise NoSuchProcess(self.pid, self._name, msg=msg) @property def pid(self): """The process PID.""" return self._pid # --- utility methods @contextlib.contextmanager def oneshot(self): """Utility context manager which considerably speeds up the retrieval of multiple process information at the same time. Internally different process info (e.g. name, ppid, uids, gids, ...) may be fetched by using the same routine, but only one information is returned and the others are discarded. When using this context manager the internal routine is executed once (in the example below on name()) and the other info are cached. The cache is cleared when exiting the context manager block. The advice is to use this every time you retrieve more than one information about the process. If you're lucky, you'll get a hell of a speedup. >>> import psutil >>> p = psutil.Process() >>> with p.oneshot(): ... p.name() # collect multiple info ... p.cpu_times() # return cached value ... p.cpu_percent() # return cached value ... p.create_time() # return cached value ... >>> """ with self._lock: if hasattr(self, "_cache"): # NOOP: this covers the use case where the user enters the # context twice: # # >>> with p.oneshot(): # ... with p.oneshot(): # ... # # Also, since as_dict() internally uses oneshot() # I expect that the code below will be a pretty common # "mistake" that the user will make, so let's guard # against that: # # >>> with p.oneshot(): # ... p.as_dict() # ... yield else: try: # cached in case cpu_percent() is used self.cpu_times.cache_activate(self) # cached in case memory_percent() is used self.memory_info.cache_activate(self) # cached in case parent() is used self.ppid.cache_activate(self) # cached in case username() is used if POSIX: self.uids.cache_activate(self) # specific implementation cache self._proc.oneshot_enter() yield finally: self.cpu_times.cache_deactivate(self) self.memory_info.cache_deactivate(self) self.ppid.cache_deactivate(self) if POSIX: self.uids.cache_deactivate(self) self._proc.oneshot_exit() def as_dict(self, attrs=None, ad_value=None): """Utility method returning process information as a hashable dictionary. If *attrs* is specified it must be a list of strings reflecting available Process class' attribute names (e.g. ['cpu_times', 'name']) else all public (read only) attributes are assumed. *ad_value* is the value which gets assigned in case AccessDenied or ZombieProcess exception is raised when retrieving that particular process information. """ valid_names = _as_dict_attrnames if attrs is not None: if not isinstance(attrs, (list, tuple, set, frozenset)): msg = f"invalid attrs type {type(attrs)}" raise TypeError(msg) attrs = set(attrs) invalid_names = attrs - valid_names if invalid_names: msg = "invalid attr name{} {}".format( "s" if len(invalid_names) > 1 else "", ", ".join(map(repr, invalid_names)), ) raise ValueError(msg) retdict = {} ls = attrs or valid_names with self.oneshot(): for name in ls: try: if name == 'pid': ret = self.pid else: meth = getattr(self, name) ret = meth() except (AccessDenied, ZombieProcess): ret = ad_value except NotImplementedError: # in case of not implemented functionality (may happen # on old or exotic systems) we want to crash only if # the user explicitly asked for that particular attr if attrs: raise continue retdict[name] = ret return retdict def parent(self): """Return the parent process as a Process object pre-emptively checking whether PID has been reused. If no parent is known return None. """ lowest_pid = _LOWEST_PID if _LOWEST_PID is not None else pids()[0] if self.pid == lowest_pid: return None ppid = self.ppid() if ppid is not None: ctime = self.create_time() try: parent = Process(ppid) if parent.create_time() <= ctime: return parent # ...else ppid has been reused by another process except NoSuchProcess: pass def parents(self): """Return the parents of this process as a list of Process instances. If no parents are known return an empty list. """ parents = [] proc = self.parent() while proc is not None: parents.append(proc) proc = proc.parent() return parents def is_running(self): """Return whether this process is running. It also checks if PID has been reused by another process, in which case it will remove the process from `process_iter()` internal cache and return False. """ if self._gone or self._pid_reused: return False try: # Checking if PID is alive is not enough as the PID might # have been reused by another process. Process identity / # uniqueness over time is guaranteed by (PID + creation # time) and that is verified in __eq__. self._pid_reused = self != Process(self.pid) if self._pid_reused: _pids_reused.add(self.pid) raise NoSuchProcess(self.pid) return True except ZombieProcess: # We should never get here as it's already handled in # Process.__init__; here just for extra safety. return True except NoSuchProcess: self._gone = True return False # --- actual API @memoize_when_activated def ppid(self): """The process parent PID. On Windows the return value is cached after first call. """ # On POSIX we don't want to cache the ppid as it may unexpectedly # change to 1 (init) in case this process turns into a zombie: # https://github.com/giampaolo/psutil/issues/321 # http://stackoverflow.com/questions/356722/ # XXX should we check creation time here rather than in # Process.parent()? self._raise_if_pid_reused() if POSIX: return self._proc.ppid() else: # pragma: no cover self._ppid = self._ppid or self._proc.ppid() return self._ppid def name(self): """The process name. The return value is cached after first call.""" # Process name is only cached on Windows as on POSIX it may # change, see: # https://github.com/giampaolo/psutil/issues/692 if WINDOWS and self._name is not None: return self._name name = self._proc.name() if POSIX and len(name) >= 15: # On UNIX the name gets truncated to the first 15 characters. # If it matches the first part of the cmdline we return that # one instead because it's usually more explicative. # Examples are "gnome-keyring-d" vs. "gnome-keyring-daemon". try: cmdline = self.cmdline() except (AccessDenied, ZombieProcess): # Just pass and return the truncated name: it's better # than nothing. Note: there are actual cases where a # zombie process can return a name() but not a # cmdline(), see: # https://github.com/giampaolo/psutil/issues/2239 pass else: if cmdline: extended_name = os.path.basename(cmdline[0]) if extended_name.startswith(name): name = extended_name self._name = name self._proc._name = name return name def exe(self): """The process executable as an absolute path. May also be an empty string. The return value is cached after first call. """ def guess_it(fallback): # try to guess exe from cmdline[0] in absence of a native # exe representation cmdline = self.cmdline() if cmdline and hasattr(os, 'access') and hasattr(os, 'X_OK'): exe = cmdline[0] # the possible exe # Attempt to guess only in case of an absolute path. # It is not safe otherwise as the process might have # changed cwd. if ( os.path.isabs(exe) and os.path.isfile(exe) and os.access(exe, os.X_OK) ): return exe if isinstance(fallback, AccessDenied): raise fallback return fallback if self._exe is None: try: exe = self._proc.exe() except AccessDenied as err: return guess_it(fallback=err) else: if not exe: # underlying implementation can legitimately return an # empty string; if that's the case we don't want to # raise AD while guessing from the cmdline try: exe = guess_it(fallback=exe) except AccessDenied: pass self._exe = exe return self._exe def cmdline(self): """The command line this process has been called with.""" return self._proc.cmdline() def status(self): """The process current status as a STATUS_* constant.""" try: return self._proc.status() except ZombieProcess: return STATUS_ZOMBIE def username(self): """The name of the user that owns the process. On UNIX this is calculated by using *real* process uid. """ if POSIX: if pwd is None: # might happen if python was installed from sources msg = "requires pwd module shipped with standard python" raise ImportError(msg) real_uid = self.uids().real try: return pwd.getpwuid(real_uid).pw_name except KeyError: # the uid can't be resolved by the system return str(real_uid) else: return self._proc.username() def create_time(self): """The process creation time as a floating point number expressed in seconds since the epoch. The return value is cached after first call. """ if self._create_time is None: self._create_time = self._proc.create_time() return self._create_time def cwd(self): """Process current working directory as an absolute path.""" return self._proc.cwd() def nice(self, value=None): """Get or set process niceness (priority).""" if value is None: return self._proc.nice_get() else: self._raise_if_pid_reused() self._proc.nice_set(value) if POSIX: @memoize_when_activated def uids(self): """Return process UIDs as a (real, effective, saved) namedtuple. """ return self._proc.uids() def gids(self): """Return process GIDs as a (real, effective, saved) namedtuple. """ return self._proc.gids() def terminal(self): """The terminal associated with this process, if any, else None. """ return self._proc.terminal() def num_fds(self): """Return the number of file descriptors opened by this process (POSIX only). """ return self._proc.num_fds() # Linux, BSD, AIX and Windows only if hasattr(_psplatform.Process, "io_counters"): def io_counters(self): """Return process I/O statistics as a (read_count, write_count, read_bytes, write_bytes) namedtuple. Those are the number of read/write calls performed and the amount of bytes read and written by the process. """ return self._proc.io_counters() # Linux and Windows if hasattr(_psplatform.Process, "ionice_get"): def ionice(self, ioclass=None, value=None): """Get or set process I/O niceness (priority). On Linux *ioclass* is one of the IOPRIO_CLASS_* constants. *value* is a number which goes from 0 to 7. The higher the value, the lower the I/O priority of the process. On Windows only *ioclass* is used and it can be set to 2 (normal), 1 (low) or 0 (very low). Available on Linux and Windows > Vista only. """ if ioclass is None: if value is not None: msg = "'ioclass' argument must be specified" raise ValueError(msg) return self._proc.ionice_get() else: self._raise_if_pid_reused() return self._proc.ionice_set(ioclass, value) # Linux / FreeBSD only if hasattr(_psplatform.Process, "rlimit"): def rlimit(self, resource, limits=None): """Get or set process resource limits as a (soft, hard) tuple. *resource* is one of the RLIMIT_* constants. *limits* is supposed to be a (soft, hard) tuple. See "man prlimit" for further info. Available on Linux and FreeBSD only. """ if limits is not None: self._raise_if_pid_reused() return self._proc.rlimit(resource, limits) # Windows, Linux and FreeBSD only if hasattr(_psplatform.Process, "cpu_affinity_get"): def cpu_affinity(self, cpus=None): """Get or set process CPU affinity. If specified, *cpus* must be a list of CPUs for which you want to set the affinity (e.g. [0, 1]). If an empty list is passed, all egible CPUs are assumed (and set). (Windows, Linux and BSD only). """ if cpus is None: return sorted(set(self._proc.cpu_affinity_get())) else: self._raise_if_pid_reused() if not cpus: if hasattr(self._proc, "_get_eligible_cpus"): cpus = self._proc._get_eligible_cpus() else: cpus = tuple(range(len(cpu_times(percpu=True)))) self._proc.cpu_affinity_set(list(set(cpus))) # Linux, FreeBSD, SunOS if hasattr(_psplatform.Process, "cpu_num"): def cpu_num(self): """Return what CPU this process is currently running on. The returned number should be <= psutil.cpu_count() and <= len(psutil.cpu_percent(percpu=True)). It may be used in conjunction with psutil.cpu_percent(percpu=True) to observe the system workload distributed across CPUs. """ return self._proc.cpu_num() # All platforms has it, but maybe not in the future. if hasattr(_psplatform.Process, "environ"): def environ(self): """The environment variables of the process as a dict. Note: this might not reflect changes made after the process started. """ return self._proc.environ() if WINDOWS: def num_handles(self): """Return the number of handles opened by this process (Windows only). """ return self._proc.num_handles() def num_ctx_switches(self): """Return the number of voluntary and involuntary context switches performed by this process. """ return self._proc.num_ctx_switches() def num_threads(self): """Return the number of threads used by this process.""" return self._proc.num_threads() if hasattr(_psplatform.Process, "threads"): def threads(self): """Return threads opened by process as a list of (id, user_time, system_time) namedtuples representing thread id and thread CPU times (user/system). On OpenBSD this method requires root access. """ return self._proc.threads() def children(self, recursive=False): """Return the children of this process as a list of Process instances, pre-emptively checking whether PID has been reused. If *recursive* is True return all the parent descendants. Example (A == this process): A ─┐ │ ├─ B (child) ─┐ │ └─ X (grandchild) ─┐ │ └─ Y (great grandchild) ├─ C (child) └─ D (child) >>> import psutil >>> p = psutil.Process() >>> p.children() B, C, D >>> p.children(recursive=True) B, X, Y, C, D Note that in the example above if process X disappears process Y won't be listed as the reference to process A is lost. """ self._raise_if_pid_reused() ppid_map = _ppid_map() ret = [] if not recursive: for pid, ppid in ppid_map.items(): if ppid == self.pid: try: child = Process(pid) # if child happens to be older than its parent # (self) it means child's PID has been reused if self.create_time() <= child.create_time(): ret.append(child) except (NoSuchProcess, ZombieProcess): pass else: # Construct a {pid: [child pids]} dict reverse_ppid_map = collections.defaultdict(list) for pid, ppid in ppid_map.items(): reverse_ppid_map[ppid].append(pid) # Recursively traverse that dict, starting from self.pid, # such that we only call Process() on actual children seen = set() stack = [self.pid] while stack: pid = stack.pop() if pid in seen: # Since pids can be reused while the ppid_map is # constructed, there may be rare instances where # there's a cycle in the recorded process "tree". continue seen.add(pid) for child_pid in reverse_ppid_map[pid]: try: child = Process(child_pid) # if child happens to be older than its parent # (self) it means child's PID has been reused intime = self.create_time() <= child.create_time() if intime: ret.append(child) stack.append(child_pid) except (NoSuchProcess, ZombieProcess): pass return ret def cpu_percent(self, interval=None): """Return a float representing the current process CPU utilization as a percentage. When *interval* is 0.0 or None (default) compares process times to system CPU times elapsed since last call, returning immediately (non-blocking). That means that the first time this is called it will return a meaningful 0.0 value. When *interval* is > 0.0 compares process times to system CPU times elapsed before and after the interval (blocking). In this case is recommended for accuracy that this function be called with at least 0.1 seconds between calls. A value > 100.0 can be returned in case of processes running multiple threads on different CPU cores. The returned value is explicitly NOT split evenly between all available logical CPUs. This means that a busy loop process running on a system with 2 logical CPUs will be reported as having 100% CPU utilization instead of 50%. Examples: >>> import psutil >>> p = psutil.Process(os.getpid()) >>> # blocking >>> p.cpu_percent(interval=1) 2.0 >>> # non-blocking (percentage since last call) >>> p.cpu_percent(interval=None) 2.9 >>> """ blocking = interval is not None and interval > 0.0 if interval is not None and interval < 0: msg = f"interval is not positive (got {interval!r})" raise ValueError(msg) num_cpus = cpu_count() or 1 def timer(): return _timer() * num_cpus if blocking: st1 = timer() pt1 = self._proc.cpu_times() time.sleep(interval) st2 = timer() pt2 = self._proc.cpu_times() else: st1 = self._last_sys_cpu_times pt1 = self._last_proc_cpu_times st2 = timer() pt2 = self._proc.cpu_times() if st1 is None or pt1 is None: self._last_sys_cpu_times = st2 self._last_proc_cpu_times = pt2 return 0.0 delta_proc = (pt2.user - pt1.user) + (pt2.system - pt1.system) delta_time = st2 - st1 # reset values for next call in case of interval == None self._last_sys_cpu_times = st2 self._last_proc_cpu_times = pt2 try: # This is the utilization split evenly between all CPUs. # E.g. a busy loop process on a 2-CPU-cores system at this # point is reported as 50% instead of 100%. overall_cpus_percent = (delta_proc / delta_time) * 100 except ZeroDivisionError: # interval was too low return 0.0 else: # Note 1: # in order to emulate "top" we multiply the value for the num # of CPU cores. This way the busy process will be reported as # having 100% (or more) usage. # # Note 2: # taskmgr.exe on Windows differs in that it will show 50% # instead. # # Note 3: # a percentage > 100 is legitimate as it can result from a # process with multiple threads running on different CPU # cores (top does the same), see: # http://stackoverflow.com/questions/1032357 # https://github.com/giampaolo/psutil/issues/474 single_cpu_percent = overall_cpus_percent * num_cpus return round(single_cpu_percent, 1) @memoize_when_activated def cpu_times(self): """Return a (user, system, children_user, children_system) namedtuple representing the accumulated process time, in seconds. This is similar to os.times() but per-process. On macOS and Windows children_user and children_system are always set to 0. """ return self._proc.cpu_times() @memoize_when_activated def memory_info(self): """Return a namedtuple with variable fields depending on the platform, representing memory information about the process. The "portable" fields available on all platforms are `rss` and `vms`. All numbers are expressed in bytes. """ return self._proc.memory_info() def memory_full_info(self): """This method returns the same information as memory_info(), plus, on some platform (Linux, macOS, Windows), also provides additional metrics (USS, PSS and swap). The additional metrics provide a better representation of actual process memory usage. Namely USS is the memory which is unique to a process and which would be freed if the process was terminated right now. It does so by passing through the whole process address. As such it usually requires higher user privileges than memory_info() and is considerably slower. """ return self._proc.memory_full_info() def memory_percent(self, memtype="rss"): """Compare process memory to total physical system memory and calculate process memory utilization as a percentage. *memtype* argument is a string that dictates what type of process memory you want to compare against (defaults to "rss"). The list of available strings can be obtained like this: >>> psutil.Process().memory_info()._fields ('rss', 'vms', 'shared', 'text', 'lib', 'data', 'dirty', 'uss', 'pss') """ valid_types = list(_psplatform.pfullmem._fields) if memtype not in valid_types: msg = ( f"invalid memtype {memtype!r}; valid types are" f" {tuple(valid_types)!r}" ) raise ValueError(msg) fun = ( self.memory_info if memtype in _psplatform.pmem._fields else self.memory_full_info ) metrics = fun() value = getattr(metrics, memtype) # use cached value if available total_phymem = _TOTAL_PHYMEM or virtual_memory().total if not total_phymem > 0: # we should never get here msg = ( "can't calculate process memory percent because total physical" f" system memory is not positive ({total_phymem!r})" ) raise ValueError(msg) return (value / float(total_phymem)) * 100 if hasattr(_psplatform.Process, "memory_maps"): def memory_maps(self, grouped=True): """Return process' mapped memory regions as a list of namedtuples whose fields are variable depending on the platform. If *grouped* is True the mapped regions with the same 'path' are grouped together and the different memory fields are summed. If *grouped* is False every mapped region is shown as a single entity and the namedtuple will also include the mapped region's address space ('addr') and permission set ('perms'). """ it = self._proc.memory_maps() if grouped: d = {} for tupl in it: path = tupl[2] nums = tupl[3:] try: d[path] = list(map(lambda x, y: x + y, d[path], nums)) except KeyError: d[path] = nums nt = _psplatform.pmmap_grouped return [nt(path, *d[path]) for path in d] else: nt = _psplatform.pmmap_ext return [nt(*x) for x in it] def open_files(self): """Return files opened by process as a list of (path, fd) namedtuples including the absolute file name and file descriptor number. """ return self._proc.open_files() def net_connections(self, kind='inet'): """Return socket connections opened by process as a list of (fd, family, type, laddr, raddr, status) namedtuples. The *kind* parameter filters for connections that match the following criteria: +------------+----------------------------------------------------+ | Kind Value | Connections using | +------------+----------------------------------------------------+ | inet | IPv4 and IPv6 | | inet4 | IPv4 | | inet6 | IPv6 | | tcp | TCP | | tcp4 | TCP over IPv4 | | tcp6 | TCP over IPv6 | | udp | UDP | | udp4 | UDP over IPv4 | | udp6 | UDP over IPv6 | | unix | UNIX socket (both UDP and TCP protocols) | | all | the sum of all the possible families and protocols | +------------+----------------------------------------------------+ """ _check_conn_kind(kind) return self._proc.net_connections(kind) @_common.deprecated_method(replacement="net_connections") def connections(self, kind="inet"): return self.net_connections(kind=kind) # --- signals if POSIX: def _send_signal(self, sig): assert not self.pid < 0, self.pid self._raise_if_pid_reused() pid, ppid, name = self.pid, self._ppid, self._name if pid == 0: # see "man 2 kill" msg = ( "preventing sending signal to process with PID 0 as it " "would affect every process in the process group of the " "calling process (os.getpid()) instead of PID 0" ) raise ValueError(msg) try: os.kill(pid, sig) except ProcessLookupError as err: if OPENBSD and pid_exists(pid): # We do this because os.kill() lies in case of # zombie processes. raise ZombieProcess(pid, name, ppid) from err self._gone = True raise NoSuchProcess(pid, name) from err except PermissionError as err: raise AccessDenied(pid, name) from err def send_signal(self, sig): """Send a signal *sig* to process pre-emptively checking whether PID has been reused (see signal module constants) . On Windows only SIGTERM is valid and is treated as an alias for kill(). """ if POSIX: self._send_signal(sig) else: # pragma: no cover self._raise_if_pid_reused() if sig != signal.SIGTERM and not self.is_running(): msg = "process no longer exists" raise NoSuchProcess(self.pid, self._name, msg=msg) self._proc.send_signal(sig) def suspend(self): """Suspend process execution with SIGSTOP pre-emptively checking whether PID has been reused. On Windows this has the effect of suspending all process threads. """ if POSIX: self._send_signal(signal.SIGSTOP) else: # pragma: no cover self._raise_if_pid_reused() self._proc.suspend() def resume(self): """Resume process execution with SIGCONT pre-emptively checking whether PID has been reused. On Windows this has the effect of resuming all process threads. """ if POSIX: self._send_signal(signal.SIGCONT) else: # pragma: no cover self._raise_if_pid_reused() self._proc.resume() def terminate(self): """Terminate the process with SIGTERM pre-emptively checking whether PID has been reused. On Windows this is an alias for kill(). """ if POSIX: self._send_signal(signal.SIGTERM) else: # pragma: no cover self._raise_if_pid_reused() self._proc.kill() def kill(self): """Kill the current process with SIGKILL pre-emptively checking whether PID has been reused. """ if POSIX: self._send_signal(signal.SIGKILL) else: # pragma: no cover self._raise_if_pid_reused() self._proc.kill() def wait(self, timeout=None): """Wait for process to terminate and, if process is a children of os.getpid(), also return its exit code, else None. On Windows there's no such limitation (exit code is always returned). If the process is already terminated immediately return None instead of raising NoSuchProcess. If *timeout* (in seconds) is specified and process is still alive raise TimeoutExpired. To wait for multiple Process(es) use psutil.wait_procs(). """ if timeout is not None and not timeout >= 0: msg = "timeout must be a positive integer" raise ValueError(msg) if self._exitcode is not _SENTINEL: return self._exitcode self._exitcode = self._proc.wait(timeout) return self._exitcode # The valid attr names which can be processed by Process.as_dict(). # fmt: off _as_dict_attrnames = { x for x in dir(Process) if not x.startswith("_") and x not in {'send_signal', 'suspend', 'resume', 'terminate', 'kill', 'wait', 'is_running', 'as_dict', 'parent', 'parents', 'children', 'rlimit', 'connections', 'oneshot'} } # fmt: on # ===================================================================== # --- Popen class # ===================================================================== class Popen(Process): """Same as subprocess.Popen, but in addition it provides all psutil.Process methods in a single class. For the following methods which are common to both classes, psutil implementation takes precedence: * send_signal() * terminate() * kill() This is done in order to avoid killing another process in case its PID has been reused, fixing BPO-6973. >>> import psutil >>> from subprocess import PIPE >>> p = psutil.Popen(["python", "-c", "print 'hi'"], stdout=PIPE) >>> p.name() 'python' >>> p.uids() user(real=1000, effective=1000, saved=1000) >>> p.username() 'giampaolo' >>> p.communicate() ('hi', None) >>> p.terminate() >>> p.wait(timeout=2) 0 >>> """ def __init__(self, *args, **kwargs): # Explicitly avoid to raise NoSuchProcess in case the process # spawned by subprocess.Popen terminates too quickly, see: # https://github.com/giampaolo/psutil/issues/193 self.__subproc = subprocess.Popen(*args, **kwargs) self._init(self.__subproc.pid, _ignore_nsp=True) def __dir__(self): return sorted(set(dir(Popen) + dir(subprocess.Popen))) def __enter__(self): if hasattr(self.__subproc, '__enter__'): self.__subproc.__enter__() return self def __exit__(self, *args, **kwargs): if hasattr(self.__subproc, '__exit__'): return self.__subproc.__exit__(*args, **kwargs) else: if self.stdout: self.stdout.close() if self.stderr: self.stderr.close() try: # Flushing a BufferedWriter may raise an error. if self.stdin: self.stdin.close() finally: # Wait for the process to terminate, to avoid zombies. self.wait() def __getattribute__(self, name): try: return object.__getattribute__(self, name) except AttributeError: try: return object.__getattribute__(self.__subproc, name) except AttributeError: msg = f"{self.__class__!r} has no attribute {name!r}" raise AttributeError(msg) from None def wait(self, timeout=None): if self.__subproc.returncode is not None: return self.__subproc.returncode ret = super().wait(timeout) self.__subproc.returncode = ret return ret # ===================================================================== # --- system processes related functions # ===================================================================== def pids(): """Return a list of current running PIDs.""" global _LOWEST_PID ret = sorted(_psplatform.pids()) _LOWEST_PID = ret[0] return ret def pid_exists(pid): """Return True if given PID exists in the current process list. This is faster than doing "pid in psutil.pids()" and should be preferred. """ if pid < 0: return False elif pid == 0 and POSIX: # On POSIX we use os.kill() to determine PID existence. # According to "man 2 kill" PID 0 has a special meaning # though: it refers to <<every process in the process # group of the calling process>> and that is not we want # to do here. return pid in pids() else: return _psplatform.pid_exists(pid) _pmap = {} _pids_reused = set() def process_iter(attrs=None, ad_value=None): """Return a generator yielding a Process instance for all running processes. Every new Process instance is only created once and then cached into an internal table which is updated every time this is used. Cache can optionally be cleared via `process_iter.clear_cache()`. The sorting order in which processes are yielded is based on their PIDs. *attrs* and *ad_value* have the same meaning as in Process.as_dict(). If *attrs* is specified as_dict() is called and the resulting dict is stored as a 'info' attribute attached to returned Process instance. If *attrs* is an empty list it will retrieve all process info (slow). """ global _pmap def add(pid): proc = Process(pid) pmap[proc.pid] = proc return proc def remove(pid): pmap.pop(pid, None) pmap = _pmap.copy() a = set(pids()) b = set(pmap.keys()) new_pids = a - b gone_pids = b - a for pid in gone_pids: remove(pid) while _pids_reused: pid = _pids_reused.pop() debug(f"refreshing Process instance for reused PID {pid}") remove(pid) try: ls = sorted(list(pmap.items()) + list(dict.fromkeys(new_pids).items())) for pid, proc in ls: try: if proc is None: # new process proc = add(pid) if attrs is not None: proc.info = proc.as_dict(attrs=attrs, ad_value=ad_value) yield proc except NoSuchProcess: remove(pid) finally: _pmap = pmap process_iter.cache_clear = lambda: _pmap.clear() # noqa: PLW0108 process_iter.cache_clear.__doc__ = "Clear process_iter() internal cache." def wait_procs(procs, timeout=None, callback=None): """Convenience function which waits for a list of processes to terminate. Return a (gone, alive) tuple indicating which processes are gone and which ones are still alive. The gone ones will have a new *returncode* attribute indicating process exit status (may be None). *callback* is a function which gets called every time a process terminates (a Process instance is passed as callback argument). Function will return as soon as all processes terminate or when *timeout* occurs. Differently from Process.wait() it will not raise TimeoutExpired if *timeout* occurs. Typical use case is: - send SIGTERM to a list of processes - give them some time to terminate - send SIGKILL to those ones which are still alive Example: >>> def on_terminate(proc): ... print("process {} terminated".format(proc)) ... >>> for p in procs: ... p.terminate() ... >>> gone, alive = wait_procs(procs, timeout=3, callback=on_terminate) >>> for p in alive: ... p.kill() """ def check_gone(proc, timeout): try: returncode = proc.wait(timeout=timeout) except (TimeoutExpired, subprocess.TimeoutExpired): pass else: if returncode is not None or not proc.is_running(): # Set new Process instance attribute. proc.returncode = returncode gone.add(proc) if callback is not None: callback(proc) if timeout is not None and not timeout >= 0: msg = f"timeout must be a positive integer, got {timeout}" raise ValueError(msg) gone = set() alive = set(procs) if callback is not None and not callable(callback): msg = f"callback {callback!r} is not a callable" raise TypeError(msg) if timeout is not None: deadline = _timer() + timeout while alive: if timeout is not None and timeout <= 0: break for proc in alive: # Make sure that every complete iteration (all processes) # will last max 1 sec. # We do this because we don't want to wait too long on a # single process: in case it terminates too late other # processes may disappear in the meantime and their PID # reused. max_timeout = 1.0 / len(alive) if timeout is not None: timeout = min((deadline - _timer()), max_timeout) if timeout <= 0: break check_gone(proc, timeout) else: check_gone(proc, max_timeout) alive = alive - gone # noqa: PLR6104 if alive: # Last attempt over processes survived so far. # timeout == 0 won't make this function wait any further. for proc in alive: check_gone(proc, 0) alive = alive - gone # noqa: PLR6104 return (list(gone), list(alive)) # ===================================================================== # --- CPU related functions # ===================================================================== def cpu_count(logical=True): """Return the number of logical CPUs in the system (same as os.cpu_count()). If *logical* is False return the number of physical cores only (e.g. hyper thread CPUs are excluded). Return None if undetermined. The return value is cached after first call. If desired cache can be cleared like this: >>> psutil.cpu_count.cache_clear() """ if logical: ret = _psplatform.cpu_count_logical() else: ret = _psplatform.cpu_count_cores() if ret is not None and ret < 1: ret = None return ret def cpu_times(percpu=False): """Return system-wide CPU times as a namedtuple. Every CPU time represents the seconds the CPU has spent in the given mode. The namedtuple's fields availability varies depending on the platform: - user - system - idle - nice (UNIX) - iowait (Linux) - irq (Linux, FreeBSD) - softirq (Linux) - steal (Linux >= 2.6.11) - guest (Linux >= 2.6.24) - guest_nice (Linux >= 3.2.0) When *percpu* is True return a list of namedtuples for each CPU. First element of the list refers to first CPU, second element to second CPU and so on. The order of the list is consistent across calls. """ if not percpu: return _psplatform.cpu_times() else: return _psplatform.per_cpu_times() try: _last_cpu_times = {threading.current_thread().ident: cpu_times()} except Exception: # noqa: BLE001 # Don't want to crash at import time. _last_cpu_times = {} try: _last_per_cpu_times = { threading.current_thread().ident: cpu_times(percpu=True) } except Exception: # noqa: BLE001 # Don't want to crash at import time. _last_per_cpu_times = {} def _cpu_tot_time(times): """Given a cpu_time() ntuple calculates the total CPU time (including idle time). """ tot = sum(times) if LINUX: # On Linux guest times are already accounted in "user" or # "nice" times, so we subtract them from total. # Htop does the same. References: # https://github.com/giampaolo/psutil/pull/940 # http://unix.stackexchange.com/questions/178045 # https://github.com/torvalds/linux/blob/ # 447976ef4fd09b1be88b316d1a81553f1aa7cd07/kernel/sched/ # cputime.c#L158 tot -= getattr(times, "guest", 0) # Linux 2.6.24+ tot -= getattr(times, "guest_nice", 0) # Linux 3.2.0+ return tot def _cpu_busy_time(times): """Given a cpu_time() ntuple calculates the busy CPU time. We do so by subtracting all idle CPU times. """ busy = _cpu_tot_time(times) busy -= times.idle # Linux: "iowait" is time during which the CPU does not do anything # (waits for IO to complete). On Linux IO wait is *not* accounted # in "idle" time so we subtract it. Htop does the same. # References: # https://github.com/torvalds/linux/blob/ # 447976ef4fd09b1be88b316d1a81553f1aa7cd07/kernel/sched/cputime.c#L244 busy -= getattr(times, "iowait", 0) return busy def _cpu_times_deltas(t1, t2): assert t1._fields == t2._fields, (t1, t2) field_deltas = [] for field in _psplatform.scputimes._fields: field_delta = getattr(t2, field) - getattr(t1, field) # CPU times are always supposed to increase over time # or at least remain the same and that's because time # cannot go backwards. # Surprisingly sometimes this might not be the case (at # least on Windows and Linux), see: # https://github.com/giampaolo/psutil/issues/392 # https://github.com/giampaolo/psutil/issues/645 # https://github.com/giampaolo/psutil/issues/1210 # Trim negative deltas to zero to ignore decreasing fields. # top does the same. Reference: # https://gitlab.com/procps-ng/procps/blob/v3.3.12/top/top.c#L5063 field_delta = max(0, field_delta) field_deltas.append(field_delta) return _psplatform.scputimes(*field_deltas) def cpu_percent(interval=None, percpu=False): """Return a float representing the current system-wide CPU utilization as a percentage. When *interval* is > 0.0 compares system CPU times elapsed before and after the interval (blocking). When *interval* is 0.0 or None compares system CPU times elapsed since last call or module import, returning immediately (non blocking). That means the first time this is called it will return a meaningless 0.0 value which you should ignore. In this case is recommended for accuracy that this function be called with at least 0.1 seconds between calls. When *percpu* is True returns a list of floats representing the utilization as a percentage for each CPU. First element of the list refers to first CPU, second element to second CPU and so on. The order of the list is consistent across calls. Examples: >>> # blocking, system-wide >>> psutil.cpu_percent(interval=1) 2.0 >>> >>> # blocking, per-cpu >>> psutil.cpu_percent(interval=1, percpu=True) [2.0, 1.0] >>> >>> # non-blocking (percentage since last call) >>> psutil.cpu_percent(interval=None) 2.9 >>> """ tid = threading.current_thread().ident blocking = interval is not None and interval > 0.0 if interval is not None and interval < 0: msg = f"interval is not positive (got {interval})" raise ValueError(msg) def calculate(t1, t2): times_delta = _cpu_times_deltas(t1, t2) all_delta = _cpu_tot_time(times_delta) busy_delta = _cpu_busy_time(times_delta) try: busy_perc = (busy_delta / all_delta) * 100 except ZeroDivisionError: return 0.0 else: return round(busy_perc, 1) # system-wide usage if not percpu: if blocking: t1 = cpu_times() time.sleep(interval) else: t1 = _last_cpu_times.get(tid) or cpu_times() _last_cpu_times[tid] = cpu_times() return calculate(t1, _last_cpu_times[tid]) # per-cpu usage else: ret = [] if blocking: tot1 = cpu_times(percpu=True) time.sleep(interval) else: tot1 = _last_per_cpu_times.get(tid) or cpu_times(percpu=True) _last_per_cpu_times[tid] = cpu_times(percpu=True) for t1, t2 in zip(tot1, _last_per_cpu_times[tid]): ret.append(calculate(t1, t2)) return ret # Use a separate dict for cpu_times_percent(), so it's independent from # cpu_percent() and they can both be used within the same program. _last_cpu_times_2 = _last_cpu_times.copy() _last_per_cpu_times_2 = _last_per_cpu_times.copy() def cpu_times_percent(interval=None, percpu=False): """Same as cpu_percent() but provides utilization percentages for each specific CPU time as is returned by cpu_times(). For instance, on Linux we'll get: >>> cpu_times_percent() cpupercent(user=4.8, nice=0.0, system=4.8, idle=90.5, iowait=0.0, irq=0.0, softirq=0.0, steal=0.0, guest=0.0, guest_nice=0.0) >>> *interval* and *percpu* arguments have the same meaning as in cpu_percent(). """ tid = threading.current_thread().ident blocking = interval is not None and interval > 0.0 if interval is not None and interval < 0: msg = f"interval is not positive (got {interval!r})" raise ValueError(msg) def calculate(t1, t2): nums = [] times_delta = _cpu_times_deltas(t1, t2) all_delta = _cpu_tot_time(times_delta) # "scale" is the value to multiply each delta with to get percentages. # We use "max" to avoid division by zero (if all_delta is 0, then all # fields are 0 so percentages will be 0 too. all_delta cannot be a # fraction because cpu times are integers) scale = 100.0 / max(1, all_delta) for field_delta in times_delta: field_perc = field_delta * scale field_perc = round(field_perc, 1) # make sure we don't return negative values or values over 100% field_perc = min(max(0.0, field_perc), 100.0) nums.append(field_perc) return _psplatform.scputimes(*nums) # system-wide usage if not percpu: if blocking: t1 = cpu_times() time.sleep(interval) else: t1 = _last_cpu_times_2.get(tid) or cpu_times() _last_cpu_times_2[tid] = cpu_times() return calculate(t1, _last_cpu_times_2[tid]) # per-cpu usage else: ret = [] if blocking: tot1 = cpu_times(percpu=True) time.sleep(interval) else: tot1 = _last_per_cpu_times_2.get(tid) or cpu_times(percpu=True) _last_per_cpu_times_2[tid] = cpu_times(percpu=True) for t1, t2 in zip(tot1, _last_per_cpu_times_2[tid]): ret.append(calculate(t1, t2)) return ret def cpu_stats(): """Return CPU statistics.""" return _psplatform.cpu_stats() if hasattr(_psplatform, "cpu_freq"): def cpu_freq(percpu=False): """Return CPU frequency as a namedtuple including current, min and max frequency expressed in Mhz. If *percpu* is True and the system supports per-cpu frequency retrieval (Linux only) a list of frequencies is returned for each CPU. If not a list with one element is returned. """ ret = _psplatform.cpu_freq() if percpu: return ret else: num_cpus = float(len(ret)) if num_cpus == 0: return None elif num_cpus == 1: return ret[0] else: currs, mins, maxs = 0.0, 0.0, 0.0 set_none = False for cpu in ret: currs += cpu.current # On Linux if /proc/cpuinfo is used min/max are set # to None. if LINUX and cpu.min is None: set_none = True continue mins += cpu.min maxs += cpu.max current = currs / num_cpus if set_none: min_ = max_ = None else: min_ = mins / num_cpus max_ = maxs / num_cpus return _common.scpufreq(current, min_, max_) __all__.append("cpu_freq") if hasattr(os, "getloadavg") or hasattr(_psplatform, "getloadavg"): # Perform this hasattr check once on import time to either use the # platform based code or proxy straight from the os module. if hasattr(os, "getloadavg"): getloadavg = os.getloadavg else: getloadavg = _psplatform.getloadavg __all__.append("getloadavg") # ===================================================================== # --- system memory related functions # ===================================================================== def virtual_memory(): """Return statistics about system memory usage as a namedtuple including the following fields, expressed in bytes: - total: total physical memory available. - available: the memory that can be given instantly to processes without the system going into swap. This is calculated by summing different memory values depending on the platform and it is supposed to be used to monitor actual memory usage in a cross platform fashion. - percent: the percentage usage calculated as (total - available) / total * 100 - used: memory used, calculated differently depending on the platform and designed for informational purposes only: macOS: active + wired BSD: active + wired + cached Linux: total - free - free: memory not being used at all (zeroed) that is readily available; note that this doesn't reflect the actual memory available (use 'available' instead) Platform-specific fields: - active (UNIX): memory currently in use or very recently used, and so it is in RAM. - inactive (UNIX): memory that is marked as not used. - buffers (BSD, Linux): cache for things like file system metadata. - cached (BSD, macOS): cache for various things. - wired (macOS, BSD): memory that is marked to always stay in RAM. It is never moved to disk. - shared (BSD): memory that may be simultaneously accessed by multiple processes. The sum of 'used' and 'available' does not necessarily equal total. On Windows 'available' and 'free' are the same. """ global _TOTAL_PHYMEM ret = _psplatform.virtual_memory() # cached for later use in Process.memory_percent() _TOTAL_PHYMEM = ret.total return ret def swap_memory(): """Return system swap memory statistics as a namedtuple including the following fields: - total: total swap memory in bytes - used: used swap memory in bytes - free: free swap memory in bytes - percent: the percentage usage - sin: no. of bytes the system has swapped in from disk (cumulative) - sout: no. of bytes the system has swapped out from disk (cumulative) 'sin' and 'sout' on Windows are meaningless and always set to 0. """ return _psplatform.swap_memory() # ===================================================================== # --- disks/partitions related functions # ===================================================================== def disk_usage(path): """Return disk usage statistics about the given *path* as a namedtuple including total, used and free space expressed in bytes plus the percentage usage. """ return _psplatform.disk_usage(path) def disk_partitions(all=False): """Return mounted partitions as a list of (device, mountpoint, fstype, opts) namedtuple. 'opts' field is a raw string separated by commas indicating mount options which may vary depending on the platform. If *all* parameter is False return physical devices only and ignore all others. """ return _psplatform.disk_partitions(all) def disk_io_counters(perdisk=False, nowrap=True): """Return system disk I/O statistics as a namedtuple including the following fields: - read_count: number of reads - write_count: number of writes - read_bytes: number of bytes read - write_bytes: number of bytes written - read_time: time spent reading from disk (in ms) - write_time: time spent writing to disk (in ms) Platform specific: - busy_time: (Linux, FreeBSD) time spent doing actual I/Os (in ms) - read_merged_count (Linux): number of merged reads - write_merged_count (Linux): number of merged writes If *perdisk* is True return the same information for every physical disk installed on the system as a dictionary with partition names as the keys and the namedtuple described above as the values. If *nowrap* is True it detects and adjust the numbers which overflow and wrap (restart from 0) and add "old value" to "new value" so that the returned numbers will always be increasing or remain the same, but never decrease. "disk_io_counters.cache_clear()" can be used to invalidate the cache. On recent Windows versions 'diskperf -y' command may need to be executed first otherwise this function won't find any disk. """ kwargs = dict(perdisk=perdisk) if LINUX else {} rawdict = _psplatform.disk_io_counters(**kwargs) if not rawdict: return {} if perdisk else None if nowrap: rawdict = _wrap_numbers(rawdict, 'psutil.disk_io_counters') nt = getattr(_psplatform, "sdiskio", _common.sdiskio) if perdisk: for disk, fields in rawdict.items(): rawdict[disk] = nt(*fields) return rawdict else: return nt(*(sum(x) for x in zip(*rawdict.values()))) disk_io_counters.cache_clear = functools.partial( _wrap_numbers.cache_clear, 'psutil.disk_io_counters' ) disk_io_counters.cache_clear.__doc__ = "Clears nowrap argument cache" # ===================================================================== # --- network related functions # ===================================================================== def net_io_counters(pernic=False, nowrap=True): """Return network I/O statistics as a namedtuple including the following fields: - bytes_sent: number of bytes sent - bytes_recv: number of bytes received - packets_sent: number of packets sent - packets_recv: number of packets received - errin: total number of errors while receiving - errout: total number of errors while sending - dropin: total number of incoming packets which were dropped - dropout: total number of outgoing packets which were dropped (always 0 on macOS and BSD) If *pernic* is True return the same information for every network interface installed on the system as a dictionary with network interface names as the keys and the namedtuple described above as the values. If *nowrap* is True it detects and adjust the numbers which overflow and wrap (restart from 0) and add "old value" to "new value" so that the returned numbers will always be increasing or remain the same, but never decrease. "net_io_counters.cache_clear()" can be used to invalidate the cache. """ rawdict = _psplatform.net_io_counters() if not rawdict: return {} if pernic else None if nowrap: rawdict = _wrap_numbers(rawdict, 'psutil.net_io_counters') if pernic: for nic, fields in rawdict.items(): rawdict[nic] = _common.snetio(*fields) return rawdict else: return _common.snetio(*[sum(x) for x in zip(*rawdict.values())]) net_io_counters.cache_clear = functools.partial( _wrap_numbers.cache_clear, 'psutil.net_io_counters' ) net_io_counters.cache_clear.__doc__ = "Clears nowrap argument cache" def net_connections(kind='inet'): """Return system-wide socket connections as a list of (fd, family, type, laddr, raddr, status, pid) namedtuples. In case of limited privileges 'fd' and 'pid' may be set to -1 and None respectively. The *kind* parameter filters for connections that fit the following criteria: +------------+----------------------------------------------------+ | Kind Value | Connections using | +------------+----------------------------------------------------+ | inet | IPv4 and IPv6 | | inet4 | IPv4 | | inet6 | IPv6 | | tcp | TCP | | tcp4 | TCP over IPv4 | | tcp6 | TCP over IPv6 | | udp | UDP | | udp4 | UDP over IPv4 | | udp6 | UDP over IPv6 | | unix | UNIX socket (both UDP and TCP protocols) | | all | the sum of all the possible families and protocols | +------------+----------------------------------------------------+ On macOS this function requires root privileges. """ _check_conn_kind(kind) return _psplatform.net_connections(kind) def net_if_addrs(): """Return the addresses associated to each NIC (network interface card) installed on the system as a dictionary whose keys are the NIC names and value is a list of namedtuples for each address assigned to the NIC. Each namedtuple includes 5 fields: - family: can be either socket.AF_INET, socket.AF_INET6 or psutil.AF_LINK, which refers to a MAC address. - address: is the primary address and it is always set. - netmask: and 'broadcast' and 'ptp' may be None. - ptp: stands for "point to point" and references the destination address on a point to point interface (typically a VPN). - broadcast: and *ptp* are mutually exclusive. Note: you can have more than one address of the same family associated with each interface. """ rawlist = _psplatform.net_if_addrs() rawlist.sort(key=lambda x: x[1]) # sort by family ret = collections.defaultdict(list) for name, fam, addr, mask, broadcast, ptp in rawlist: try: fam = socket.AddressFamily(fam) except ValueError: if WINDOWS and fam == -1: fam = _psplatform.AF_LINK elif ( hasattr(_psplatform, "AF_LINK") and fam == _psplatform.AF_LINK ): # Linux defines AF_LINK as an alias for AF_PACKET. # We re-set the family here so that repr(family) # will show AF_LINK rather than AF_PACKET fam = _psplatform.AF_LINK if fam == _psplatform.AF_LINK: # The underlying C function may return an incomplete MAC # address in which case we fill it with null bytes, see: # https://github.com/giampaolo/psutil/issues/786 separator = ":" if POSIX else "-" while addr.count(separator) < 5: addr += f"{separator}00" nt = _common.snicaddr(fam, addr, mask, broadcast, ptp) # On Windows broadcast is None, so we determine it via # ipaddress module. if WINDOWS and fam in {socket.AF_INET, socket.AF_INET6}: try: broadcast = _common.broadcast_addr(nt) except Exception as err: # noqa: BLE001 debug(err) else: if broadcast is not None: nt._replace(broadcast=broadcast) ret[name].append(nt) return dict(ret) def net_if_stats(): """Return information about each NIC (network interface card) installed on the system as a dictionary whose keys are the NIC names and value is a namedtuple with the following fields: - isup: whether the interface is up (bool) - duplex: can be either NIC_DUPLEX_FULL, NIC_DUPLEX_HALF or NIC_DUPLEX_UNKNOWN - speed: the NIC speed expressed in mega bits (MB); if it can't be determined (e.g. 'localhost') it will be set to 0. - mtu: the maximum transmission unit expressed in bytes. """ return _psplatform.net_if_stats() # ===================================================================== # --- sensors # ===================================================================== # Linux, macOS if hasattr(_psplatform, "sensors_temperatures"): def sensors_temperatures(fahrenheit=False): """Return hardware temperatures. Each entry is a namedtuple representing a certain hardware sensor (it may be a CPU, an hard disk or something else, depending on the OS and its configuration). All temperatures are expressed in celsius unless *fahrenheit* is set to True. """ def convert(n): if n is not None: return (float(n) * 9 / 5) + 32 if fahrenheit else n ret = collections.defaultdict(list) rawdict = _psplatform.sensors_temperatures() for name, values in rawdict.items(): while values: label, current, high, critical = values.pop(0) current = convert(current) high = convert(high) critical = convert(critical) if high and not critical: critical = high elif critical and not high: high = critical ret[name].append( _common.shwtemp(label, current, high, critical) ) return dict(ret) __all__.append("sensors_temperatures") # Linux if hasattr(_psplatform, "sensors_fans"): def sensors_fans(): """Return fans speed. Each entry is a namedtuple representing a certain hardware sensor. All speed are expressed in RPM (rounds per minute). """ return _psplatform.sensors_fans() __all__.append("sensors_fans") # Linux, Windows, FreeBSD, macOS if hasattr(_psplatform, "sensors_battery"): def sensors_battery(): """Return battery information. If no battery is installed returns None. - percent: battery power left as a percentage. - secsleft: a rough approximation of how many seconds are left before the battery runs out of power. May be POWER_TIME_UNLIMITED or POWER_TIME_UNLIMITED. - power_plugged: True if the AC power cable is connected. """ return _psplatform.sensors_battery() __all__.append("sensors_battery") # ===================================================================== # --- other system related functions # ===================================================================== def boot_time(): """Return the system boot time expressed in seconds since the epoch.""" # Note: we are not caching this because it is subject to # system clock updates. return _psplatform.boot_time() def users(): """Return users currently connected on the system as a list of namedtuples including the following fields. - user: the name of the user - terminal: the tty or pseudo-tty associated with the user, if any. - host: the host name associated with the entry, if any. - started: the creation time as a floating point number expressed in seconds since the epoch. """ return _psplatform.users() # ===================================================================== # --- Windows services # ===================================================================== if WINDOWS: def win_service_iter(): """Return a generator yielding a WindowsService instance for all Windows services installed. """ return _psplatform.win_service_iter() def win_service_get(name): """Get a Windows service by *name*. Raise NoSuchProcess if no service with such name exists. """ return _psplatform.win_service_get(name) # ===================================================================== def _set_debug(value): """Enable or disable PSUTIL_DEBUG option, which prints debugging messages to stderr. """ import psutil._common psutil._common.PSUTIL_DEBUG = bool(value) _psplatform.cext.set_debug(bool(value)) del memoize_when_activated
| ver. 1.4 |
Github
|
.
| PHP 8.2.28 | Generation time: 0.06 |
proxy
|
phpinfo
|
Settings