Andrew Hamilton
72b5f3750e
- It wasn't worth the complexity. It was only useful when running the python profile tool, which executes python scripts. But that tool isn't very useful when run against normal scripts. It has been removed too. - The sandbox also wasn't working inside snaps, appimages, systemd-nspawn or chroots.
91 lines
3 KiB
Python
Executable file
91 lines
3 KiB
Python
Executable file
#!/usr/bin/env python3
|
|
|
|
# Copyright (C) 2015-2017 Andrew Hamilton. All rights reserved.
|
|
# Licensed under the Artistic License 2.0.
|
|
|
|
import asyncio
|
|
import os
|
|
import signal
|
|
|
|
import tools
|
|
|
|
|
|
class Worker:
|
|
|
|
def __init__(self, is_already_paused, is_being_tested):
|
|
self.is_already_paused = is_already_paused
|
|
self.is_being_tested = is_being_tested
|
|
self.result = None
|
|
self.process = None
|
|
self.child_pgid = None
|
|
|
|
@asyncio.coroutine
|
|
def create_process(self):
|
|
create = asyncio.create_subprocess_exec(
|
|
*[__file__], stdin=asyncio.subprocess.PIPE,
|
|
stdout=asyncio.subprocess.PIPE, stderr=asyncio.subprocess.PIPE,
|
|
preexec_fn=os.setsid)
|
|
self.process = yield from create
|
|
pid_line = yield from self.process.stdout.readline()
|
|
self.child_pgid = int(pid_line.strip())
|
|
os.setpriority(os.PRIO_PGRP, self.child_pgid, 19)
|
|
|
|
@asyncio.coroutine
|
|
def run_tool(self, path, tool):
|
|
self.process.stdin.write(("%s\n%s\n" %
|
|
(tool.__qualname__, path)).encode("utf-8"))
|
|
data = yield from self.process.stdout.readline()
|
|
return tools.Status(int(data))
|
|
|
|
@asyncio.coroutine
|
|
def job_runner(self, summary, log, jobs_added_event,
|
|
appearance_changed_event):
|
|
yield from self.create_process()
|
|
while True:
|
|
yield from jobs_added_event.wait()
|
|
while True:
|
|
try:
|
|
self.result = summary.get_closest_placeholder()
|
|
except StopIteration:
|
|
self.result = None
|
|
if summary.result_total == summary.completed_total:
|
|
log.log_message("All results are up to date.")
|
|
if self.is_being_tested:
|
|
os.kill(os.getpid(), signal.SIGINT)
|
|
break
|
|
yield from self.result.run(log, appearance_changed_event, self)
|
|
summary.completed_total += 1
|
|
jobs_added_event.clear()
|
|
|
|
def pause(self):
|
|
if self.result is not None and \
|
|
self.result.status == tools.Status.running:
|
|
os.killpg(self.child_pgid, signal.SIGSTOP)
|
|
self.result.set_status(tools.Status.paused)
|
|
|
|
def continue_(self):
|
|
if self.result is not None and \
|
|
self.result.status == tools.Status.paused:
|
|
self.result.set_status(tools.Status.running)
|
|
os.killpg(self.child_pgid, signal.SIGCONT)
|
|
|
|
def kill(self):
|
|
if self.child_pgid is not None:
|
|
os.killpg(self.child_pgid, signal.SIGKILL)
|
|
|
|
|
|
def main():
|
|
print(os.getpgid(os.getpid()), flush=True)
|
|
try:
|
|
while True:
|
|
tool_name, path = input(), input()
|
|
tool = getattr(tools, tool_name)
|
|
result = tools.Result(path, tool)
|
|
status, result.result = tools.run_tool_no_error(path, tool)
|
|
print(status.value, flush=True)
|
|
except:
|
|
tools.log_error()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|