85 lines
3.0 KiB
Python
85 lines
3.0 KiB
Python
import win32process
|
||
import win32job
|
||
import win32api
|
||
import win32con
|
||
import subprocess
|
||
import win32event
|
||
|
||
import time
|
||
|
||
class ChildProcessManager:
|
||
def __init__(self):
|
||
self.HJob = None
|
||
self.ProcessId = None
|
||
self.ProcessHandle = None
|
||
|
||
def StartChildProcess(self, exe_path):
|
||
self._Cleanup()
|
||
try:
|
||
self.HJob = win32job.CreateJobObject(None, "")
|
||
if not self.HJob: raise Exception("Failed To Create Job Object")
|
||
except Exception as e:
|
||
print(f"Failed To Create Job Object: {e}")
|
||
self._Cleanup()
|
||
return False
|
||
|
||
try:
|
||
info = win32job.QueryInformationJobObject(self.HJob, win32job.JobObjectExtendedLimitInformation)
|
||
info['BasicLimitInformation']['LimitFlags'] |= win32job.JOB_OBJECT_LIMIT_KILL_ON_JOB_CLOSE
|
||
win32job.SetInformationJobObject(self.HJob, win32job.JobObjectExtendedLimitInformation, info)
|
||
except Exception as e:
|
||
print(f"Failed To Set Job Object Information: {e}")
|
||
self._Cleanup()
|
||
return False
|
||
|
||
startUpInfo = win32process.STARTUPINFO()
|
||
try:
|
||
hProcess, hThread, dwProcessId, dwThreadId = win32process.CreateProcess(
|
||
None, exe_path, None, None, False,
|
||
win32process.CREATE_NEW_PROCESS_GROUP | win32process.CREATE_SUSPENDED,
|
||
None, None, startUpInfo
|
||
)
|
||
self.ProcessId = dwProcessId
|
||
self.ProcessHandle = hProcess
|
||
win32job.AssignProcessToJobObject(self.HJob, hProcess)
|
||
win32process.ResumeThread(hThread)
|
||
win32api.CloseHandle(hThread)
|
||
print(f"Successfully Started Child Process, PID: {dwProcessId}")
|
||
return True
|
||
except Exception as e:
|
||
print(f"Failed To Start Child Process: {e}")
|
||
self._Cleanup()
|
||
return False
|
||
|
||
def IsProcessAlive(self):
|
||
if not self.ProcessId or not self.ProcessHandle: return False
|
||
try:
|
||
result = win32event.WaitForSingleObject(self.ProcessHandle, 0)
|
||
return result == win32con.WAIT_TIMEOUT
|
||
except Exception as e:
|
||
print(f"Failed To Check Process Status: {e}")
|
||
return False
|
||
def _Cleanup(self):
|
||
if self.ProcessHandle:
|
||
try: win32api.CloseHandle(self.ProcessHandle)
|
||
except: pass
|
||
self.ProcessHandle = None
|
||
if self.HJob:
|
||
try: win32api.CloseHandle(self.HJob)
|
||
except: pass
|
||
self.HJob = None
|
||
self.ProcessId = None
|
||
def Close(self):
|
||
self._Cleanup()
|
||
|
||
if __name__ == "__main__":
|
||
manager = ChildProcessManager()
|
||
if manager.StartChildProcess("notepad.exe"):
|
||
try:
|
||
while True:
|
||
status = "存活" if manager.IsProcessAlive() else "死亡"
|
||
print(f"子进程状态: {status}，PID: {manager.ProcessId}")
|
||
time.sleep(1)
|
||
finally:
|
||
manager.Close()
|