mirror of
https://github.com/capstone-engine/llvm-capstone.git
synced 2025-04-09 01:11:53 +00:00
2cd84c905d
This is the Linux counterpart to the sampling support I added
on the macOS side.
This change also introduces zip-file compression if the size of
the sample output is greater than 10 KB. The Linux side can be
quite large and the textual content is averaging over a 10x
compression factor on tests that I force to time out. When
compression takes place, the filename becomes:
{session_dir}/{TestFilename.py}-{pid}.sample.zip
This support relies on the linux 'perf' tool. If it isn't
present, the behavior is to ignore pre-kill processing of
the timed out test process.
Note calling the perf tool under the timeout command appears
to nuke the profiled process. This was causing the timeout
kill logic to fail due to the process having disappeared.
I modified the kill logic to catch the case of the process
not existing, and I have it ignore the kill request in that
case. Any other exception is still raised.
Reviewers: labath
Subscribers: lldb-commits
Differential Revision: https://reviews.llvm.org/D24890
llvm-svn: 282436
77 lines
2.9 KiB
Python
77 lines
2.9 KiB
Python
"""Provides a pre-kill method to run on Linux.
|
|
|
|
This timeout pre-kill method relies on the Linux perf-tools
|
|
distribution. The appropriate way to obtain this set of tools
|
|
will depend on the Linux distribution.
|
|
|
|
For Ubuntu 16.04, the invoke the following command:
|
|
sudo apt-get install perf-tools-unstable
|
|
"""
|
|
from __future__ import print_function
|
|
|
|
# system imports
|
|
import os
|
|
import subprocess
|
|
import sys
|
|
import tempfile
|
|
|
|
|
|
def do_pre_kill(process_id, runner_context, output_stream, sample_time=3):
|
|
"""Samples the given process id, and puts the output to output_stream.
|
|
|
|
@param process_id the local process to sample.
|
|
|
|
@param runner_context a dictionary of details about the architectures
|
|
and platform on which the given process is running. Expected keys are
|
|
archs (array of architectures), platform_name, platform_url, and
|
|
platform_working_dir.
|
|
|
|
@param output_stream file-like object that should be used to write the
|
|
results of sampling.
|
|
|
|
@param sample_time specifies the time in seconds that should be captured.
|
|
"""
|
|
|
|
# Validate args.
|
|
if runner_context is None:
|
|
raise Exception("runner_context argument is required")
|
|
if not isinstance(runner_context, dict):
|
|
raise Exception("runner_context argument must be a dictionary")
|
|
|
|
# We will try to run sample on the local host only if there is no URL
|
|
# to a remote.
|
|
if "platform_url" in runner_context and (
|
|
runner_context["platform_url"] is not None):
|
|
import pprint
|
|
sys.stderr.write(
|
|
"warning: skipping timeout pre-kill sample invocation because we "
|
|
"don't know how to run on a remote yet. runner_context={}\n"
|
|
.format(pprint.pformat(runner_context)))
|
|
|
|
# We're going to create a temp file, and immediately overwrite it with the
|
|
# following command. This just ensures we don't have any races in
|
|
# creation of the temporary sample file.
|
|
fileno, filename = tempfile.mkstemp(suffix='perfdata')
|
|
os.close(fileno)
|
|
fileno = None
|
|
|
|
try:
|
|
with open(os.devnull, 'w') as devnull:
|
|
returncode = subprocess.call(['timeout', str(sample_time), 'perf',
|
|
'record', '-g', '-o', filename, '-p', str(process_id)],
|
|
stdout=devnull, stderr=devnull)
|
|
if returncode == 0 or returncode == 124:
|
|
# This is okay - this is the timeout return code, which is totally
|
|
# expected.
|
|
pass
|
|
else:
|
|
raise Exception("failed to call 'perf record .., error: {}".format(
|
|
returncode))
|
|
|
|
with open(os.devnull, 'w') as devnull:
|
|
output = subprocess.check_output(['perf', 'report', '--call-graph',
|
|
'--stdio', '-i', filename], stderr=devnull)
|
|
output_stream.write(output)
|
|
finally:
|
|
os.remove(filename)
|