|
|
|
|
|
|
|
|
| """Windows platform implementation."""
|
|
|
| import contextlib
|
| import enum
|
| import functools
|
| import os
|
| import signal
|
| import sys
|
| import threading
|
| 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
|
| ):
|
|
|
|
|
|
|
|
|
| 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
|
|
|
|
|
|
|
|
|
|
|
| __extra__all__ = [
|
| "win_service_iter", "win_service_get",
|
|
|
| "ABOVE_NORMAL_PRIORITY_CLASS", "BELOW_NORMAL_PRIORITY_CLASS",
|
| "HIGH_PRIORITY_CLASS", "IDLE_PRIORITY_CLASS", "NORMAL_PRIORITY_CLASS",
|
| "REALTIME_PRIORITY_CLASS",
|
|
|
| "IOPRIO_VERYLOW", "IOPRIO_LOW", "IOPRIO_NORMAL", "IOPRIO_HIGH",
|
|
|
| "CONN_DELETE_TCB", "AF_LINK",
|
| ]
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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,
|
| )
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| scputimes = namedtuple('scputimes',
|
| ['user', 'system', 'idle', 'interrupt', 'dpc'])
|
|
|
| svmem = namedtuple('svmem', ['total', 'available', 'percent', 'used', 'free'])
|
|
|
| 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'])
|
|
|
| pfullmem = namedtuple('pfullmem', pmem._fields + ('uss', ))
|
|
|
| pmmap_grouped = namedtuple('pmmap_grouped', ['path', 'rss'])
|
|
|
| pmmap_ext = namedtuple(
|
| 'pmmap_ext', 'addr perms ' + ' '.join(pmmap_grouped._fields))
|
|
|
| pio = namedtuple('pio', ['read_count', 'write_count',
|
| 'read_bytes', 'write_bytes',
|
| 'other_count', 'other_bytes'])
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| @functools.lru_cache(maxsize=512)
|
| def convert_dos_path(s):
|
| r"""Convert paths using native DOS format like:
|
| "\Device\HarddiskVolume1\Windows\systemew\file.txt" or
|
| "\??\C:\Windows\systemew\file.txt"
|
| into:
|
| "C:\Windows\systemew\file.txt".
|
| """
|
| if s.startswith('\\\\'):
|
| return s
|
| rawdrive = '\\'.join(s.split('\\')[:3])
|
| if rawdrive in {"\\??\\UNC", "\\Device\\Mup"}:
|
| rawdrive = '\\'.join(s.split('\\')[:5])
|
| driveletter = '\\\\' + '\\'.join(s.split('\\')[3:5])
|
| elif rawdrive.startswith('\\??\\'):
|
| driveletter = s.split('\\')[2]
|
| else:
|
| driveletter = cext.QueryDosDevice(rawdrive)
|
| remainder = s[len(rawdrive) :]
|
| return os.path.join(driveletter, remainder)
|
|
|
|
|
| @memoize
|
| def getpagesize():
|
| return cext.getpagesize()
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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]
|
|
|
|
|
|
|
| total = total_system - total_phys
|
|
|
|
|
|
|
|
|
| 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_io_counters = cext.disk_io_counters
|
|
|
|
|
| def disk_usage(path):
|
| """Return disk usage associated with path."""
|
| if isinstance(path, bytes):
|
|
|
|
|
| path = path.decode(ENCODING, errors="strict")
|
| total, used, free = cext.disk_usage(path)
|
| 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]
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| def cpu_times():
|
| """Return system CPU times as a named tuple."""
|
| user, system, idle = cext.cpu_times()
|
|
|
|
|
|
|
| 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_initialized = False
|
| _lock = threading.Lock()
|
|
|
|
|
| def _getloadavg_impl():
|
|
|
| raw_loads = cext.getloadavg()
|
| return tuple(round(load, 2) for load in raw_loads)
|
|
|
|
|
| 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_initialized
|
|
|
| if _loadavg_initialized:
|
| return _getloadavg_impl()
|
|
|
| with _lock:
|
| if not _loadavg_initialized:
|
| cext.init_loadavg_counter()
|
| _loadavg_initialized = True
|
|
|
| return _getloadavg_impl()
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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()
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| def sensors_battery():
|
| """Return battery information."""
|
|
|
|
|
|
|
| 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)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| _last_btime = 0
|
|
|
|
|
| def boot_time():
|
| """The system boot time expressed in seconds since the epoch. This
|
| also includes the time spent during hybernate / suspend.
|
| """
|
|
|
|
|
|
|
| global _last_btime
|
| ret = time.time() - cext.uptime()
|
| 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
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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:
|
| """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):
|
|
|
|
|
| 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)
|
| )
|
|
|
| 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
|
|
|
|
|
|
|
| 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']
|
|
|
|
|
|
|
| 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())
|
|
|
|
|
|
|
| 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
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| pids = cext.pids
|
| pid_exists = cext.pid_exists
|
| ppid_map = cext.ppid_map
|
|
|
|
|
| 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):
|
| 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
|
|
|
|
|
|
|
| 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.
|
| """
|
|
|
|
|
| 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:
|
|
|
|
|
| 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
|
|
|
| @wrap_exceptions
|
| @retry_error_partial_copy
|
| def cmdline(self):
|
| if cext.WINVER >= cext.WINDOWS_8_1:
|
|
|
|
|
| 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):
|
|
|
|
|
| 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):
|
|
|
|
|
|
|
| t = self._get_raw_meminfo()
|
| rss = t[2]
|
| vms = t[7]
|
| 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:
|
|
|
|
|
| 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:
|
|
|
| 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 = cext.proc_wait(self.pid, cext_timeout)
|
| except cext.TimeoutExpired as err:
|
|
|
| raise TimeoutExpired(timeout, self.pid, self._name) from err
|
| except cext.TimeoutAbandoned:
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| exit_code = None
|
|
|
|
|
|
|
|
|
|
|
| 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)
|
|
|
| @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):
|
|
|
|
|
| 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']]
|
|
|
| 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)
|
|
|
|
|
| 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()
|
|
|
|
|
|
|
|
|
| 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
|
|
|
|
|
|
|
|
|
| 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']]
|
|
|
| return _common.pctxsw(ctx_switches, 0)
|
|
|