mirror of
https://github.com/hakril/PythonForWindows
synced 2026-06-08 14:31:45 +00:00
hakril
116 lines
3.6 KiB
Python
116 lines
3.6 KiB
Python
import ctypes
|
|
import windows
|
|
|
|
from collections import namedtuple
|
|
|
|
from windows import utils
|
|
from windows.generated_def import *
|
|
|
|
|
|
SERVICE_TYPE = {x:x for x in [SERVICE_KERNEL_DRIVER, SERVICE_FILE_SYSTEM_DRIVER, SERVICE_WIN32_OWN_PROCESS, SERVICE_WIN32_SHARE_PROCESS, SERVICE_INTERACTIVE_PROCESS]}
|
|
SERVICE_STATE = {x:x for x in [SERVICE_STOPPED, SERVICE_START_PENDING, SERVICE_STOP_PENDING, SERVICE_RUNNING, SERVICE_CONTINUE_PENDING, SERVICE_PAUSE_PENDING, SERVICE_PAUSED]}
|
|
SERVICE_CONTROLE_ACCEPTED = {x:x for x in []}
|
|
SERVICE_FLAGS = {x:x for x in [SERVICE_RUNS_IN_SYSTEM_PROCESS]}
|
|
|
|
|
|
ServiceStatus = namedtuple("ServiceStatus", ["type", "state", "control_accepted", "flags"])
|
|
"""
|
|
``type`` might be one of:
|
|
|
|
* ``SERVICE_KERNEL_DRIVER(0x1L)``
|
|
* ``SERVICE_FILE_SYSTEM_DRIVER(0x2L)``
|
|
* ``SERVICE_WIN32_OWN_PROCESS(0x10L)``
|
|
* ``SERVICE_WIN32_SHARE_PROCESS(0x20L)``
|
|
* ``SERVICE_INTERACTIVE_PROCESS(0x100L)``
|
|
|
|
``state`` might be one of:
|
|
|
|
* ``SERVICE_STOPPED(0x1L)``
|
|
* ``SERVICE_START_PENDING(0x2L)``
|
|
* ``SERVICE_STOP_PENDING(0x3L)``
|
|
* ``SERVICE_RUNNING(0x4L)``
|
|
* ``SERVICE_CONTINUE_PENDING(0x5L)``
|
|
* ``SERVICE_PAUSE_PENDING(0x6L)``
|
|
* ``SERVICE_PAUSED(0x7L)``
|
|
|
|
``flags`` might be one of:
|
|
|
|
* ``0``
|
|
* ``SERVICE_RUNS_IN_SYSTEM_PROCESS(0x1L)``
|
|
|
|
"""
|
|
|
|
class Service(object):
|
|
def __repr__(self):
|
|
return '<{0} "{1}">'.format(type(self).__name__, self.name)
|
|
|
|
@utils.fixedpropety
|
|
def name(self):
|
|
"""The name of the service
|
|
|
|
:type: :class:`str`
|
|
"""
|
|
return self.lpServiceName
|
|
|
|
@utils.fixedpropety
|
|
def description(self):
|
|
"""The description of the service
|
|
|
|
:type: :class:`str`
|
|
"""
|
|
return self.lpDisplayName
|
|
|
|
@utils.fixedpropety
|
|
def status(self):
|
|
"""The status of the service
|
|
|
|
:type: :class:`ServiceStatus`
|
|
"""
|
|
status = self.ServiceStatusProcess
|
|
stype = SERVICE_TYPE.get(status.dwServiceType, status.dwServiceType)
|
|
sstate = SERVICE_STATE.get(status.dwCurrentState, status.dwCurrentState)
|
|
scontrol = status.dwControlsAccepted
|
|
sflags = SERVICE_FLAGS.get(status.dwServiceFlags, status.dwServiceFlags)
|
|
return ServiceStatus(stype, sstate, scontrol, sflags)
|
|
|
|
@utils.fixedpropety
|
|
def process(self):
|
|
"""The process running the service (if any)
|
|
|
|
:type: :class:`WinProcess <windows.winobject.process.WinProcess>` or ``None``
|
|
"""
|
|
pid = self.ServiceStatusProcess.dwProcessId
|
|
if not pid:
|
|
return None
|
|
l = [p for p in windows.system.processes if p.pid == pid]
|
|
if not l:
|
|
return None # Other thing ?
|
|
return l[0]
|
|
|
|
|
|
class ServiceA(Service, ENUM_SERVICE_STATUS_PROCESSA):
|
|
"""A Service object with ascii data"""
|
|
pass
|
|
|
|
def enumerate_services():
|
|
scmanager = windows.winproxy.OpenSCManagerA(dwDesiredAccess=SC_MANAGER_ENUMERATE_SERVICE)
|
|
|
|
size_needed = DWORD()
|
|
nb_services = DWORD()
|
|
counter = DWORD()
|
|
try:
|
|
windows.winproxy.EnumServicesStatusExA(scmanager, SC_ENUM_PROCESS_INFO, SERVICE_TYPE_ALL, SERVICE_ACTIVE, None, 0, ctypes.byref(size_needed), ctypes.byref(nb_services), byref(counter), None)
|
|
except WindowsError:
|
|
pass
|
|
|
|
while True:
|
|
size = size_needed.value
|
|
buffer = (ctypes.c_byte * size)()
|
|
|
|
try:
|
|
windows.winproxy.EnumServicesStatusExA(scmanager, SC_ENUM_PROCESS_INFO, SERVICE_TYPE_ALL, SERVICE_ACTIVE, buffer, size, ctypes.byref(size_needed), ctypes.byref(nb_services), byref(counter), None)
|
|
except WindowsError as e:
|
|
continue
|
|
|
|
return_type = (ServiceA * nb_services.value)
|
|
return list(return_type.from_buffer(buffer)) |