105 lines
3.1 KiB
Python
105 lines
3.1 KiB
Python
"""Miscellaneous utility functions used by the hang analyzer."""
|
|
|
|
import os
|
|
import sys
|
|
import time
|
|
import signal
|
|
import logging
|
|
import subprocess
|
|
from distutils import spawn # pylint: disable=no-name-in-module
|
|
|
|
from buildscripts.resmokelib import core
|
|
|
|
_IS_WINDOWS = (sys.platform == "win32")
|
|
|
|
if _IS_WINDOWS:
|
|
import win32event
|
|
import win32api
|
|
|
|
|
|
def call(args, logger):
|
|
"""Call subprocess on args list."""
|
|
logger.info(str(args))
|
|
|
|
# Use a common pipe for stdout & stderr for logging.
|
|
process = subprocess.Popen(args, stdout=subprocess.PIPE, stderr=subprocess.STDOUT)
|
|
logger_pipe = core.pipe.LoggerPipe(logger, logging.INFO, process.stdout)
|
|
logger_pipe.wait_until_started()
|
|
|
|
ret = process.wait()
|
|
logger_pipe.wait_until_finished()
|
|
|
|
if ret != 0:
|
|
logger.error("Bad exit code %d", ret)
|
|
raise Exception("Bad exit code %d from %s" % (ret, " ".join(args)))
|
|
|
|
|
|
def find_program(prog, paths):
|
|
"""Find the specified program in env PATH, or tries a set of paths."""
|
|
for loc in paths:
|
|
full_prog = os.path.join(loc, prog)
|
|
if os.path.exists(full_prog):
|
|
return full_prog
|
|
|
|
return spawn.find_executable(prog)
|
|
|
|
|
|
def callo(args, logger):
|
|
"""Call subprocess on args string."""
|
|
logger.info("%s", str(args))
|
|
return subprocess.check_output(args).decode('utf-8', 'replace')
|
|
|
|
|
|
def signal_python(logger, pinfo):
|
|
"""Send appropriate dumping signal to python processes."""
|
|
|
|
# On Windows, we set up an event object to wait on a signal. For Cygwin, we register
|
|
# a signal handler to wait for the signal since it supports POSIX signals.
|
|
if _IS_WINDOWS:
|
|
logger.info("Calling SetEvent to signal python process %s with PID %d", pinfo.name,
|
|
pinfo.pid)
|
|
signal_event_object(logger, pinfo.pid)
|
|
else:
|
|
logger.info("Sending signal SIGUSR1 to python process %s with PID %d", pinfo.name,
|
|
pinfo.pid)
|
|
signal_process(logger, pinfo.pid, signal.SIGUSR1)
|
|
|
|
|
|
def signal_event_object(logger, pid):
|
|
"""Signal the Windows event object."""
|
|
|
|
# Use unique event_name created.
|
|
event_name = "Global\\Mongo_Python_" + str(pid)
|
|
|
|
try:
|
|
desired_access = win32event.EVENT_MODIFY_STATE
|
|
inherit_handle = False
|
|
task_timeout_handle = win32event.OpenEvent(desired_access, inherit_handle, event_name)
|
|
except win32event.error as err:
|
|
logger.info("Exception from win32event.OpenEvent with error: %s", err)
|
|
return
|
|
|
|
try:
|
|
win32event.SetEvent(task_timeout_handle)
|
|
except win32event.error as err:
|
|
logger.info("Exception from win32event.SetEvent with error: %s", err)
|
|
finally:
|
|
win32api.CloseHandle(task_timeout_handle)
|
|
|
|
logger.info("Waiting for process to report")
|
|
time.sleep(5)
|
|
|
|
|
|
def signal_process(logger, pid, signalnum):
|
|
"""Signal process with signal, N/A on Windows."""
|
|
try:
|
|
os.kill(pid, signalnum)
|
|
|
|
logger.info("Waiting for process to report")
|
|
time.sleep(5)
|
|
except OSError as err:
|
|
logger.error("Hit OS error trying to signal process: %s", err)
|
|
|
|
except AttributeError:
|
|
logger.error("Cannot send signal to a process on Windows")
|