retrace server: add more logic to cleanup script

1 files changed, 76 insertions, 7 deletions

diff --git a/src/retrace/retrace.py b/src/retrace/retrace.py
index 33488bba..dfebcb62 100644
--- a/src/retrace/retrace.py
+++ b/src/retrace/retrace.py
@@ -22,13 +22,7 @@ PACKAGE_PARSER = re.compile("^(.+)-([0-9]+(\.[0-9]+)*-[0-9]+)\.([^-]+)$")
 DF_OUTPUT_PARSER = re.compile("^([^ ^\t]*)[ \t]+([0-9]+)[ \t]+([0-9]+)[ \t]+([0-9]+)[ \t]+([0-9]+%)[ \t]+(.*)$")
 DU_OUTPUT_PARSER = re.compile("^([0-9]+)")
 URL_PARSER = re.compile("^/([0-9]+)/?")
-RELEASE_PARSERS = {
-  "fedora": re.compile("^Fedora[^0-9]+([0-9]+)[^\(]\(([^\)]+)\)$"),
-}
-
-GUESS_RELEASE_PARSERS = {
-  "fedora": re.compile("\.fc([0-9]+)"),
-}
+WORKER_RUNNING_PARSER = re.compile("^([0-9]+)[ \t]+[0-9]+[ \t]+([^ ^\t]+)[ \t]+.*abrt-retrace-worker ([0-9]+)$")
 
 HANDLE_ARCHIVE = {
   "application/x-xz-compressed-tar": {
@@ -260,6 +254,81 @@ def get_active_tasks():
 
     return tasks
 
+def run_ps():
+    pipe = Popen(["ps", "-eo", "pid,ppid,etime,cmd"], stdout=PIPE).stdout
+    lines = pipe.readlines()
+    pipe.close()
+
+    return lines
+
+def get_running_tasks(ps_output=None):
+    if not ps_output:
+        ps_output = run_ps()
+
+    result = []
+
+    for line in ps_output:
+        match = WORKER_RUNNING_PARSER.match(line)
+        if match:
+            result.append((int(match.group(1)), int(match.group(3)), match.group(2)))
+
+    return result
+
+def get_process_tree(pid, ps_output):
+    result = [pid]
+
+    parser = re.compile("^([0-9]+)[ \t]+(%d).*$" % pid)
+
+    for line in ps_output:
+        match = parser.match(line)
+        if match:
+            pid = int(match.group(1))
+            result.extend(get_process_tree(pid, ps_output))
+
+    return result
+
+def kill_process_and_childs(process_id, ps_output=None):
+    result = True
+
+    if not ps_output:
+        ps_output = run_ps()
+
+    for pid in get_process_tree(process_id, ps_output):
+        try:
+            os.kill(pid, 9)
+        except OSError, ex:
+            result = False
+
+    return result
+
+def cleanup_task(taskid):
+    null = open("/dev/null", "w")
+
+    savedir = "%s/%d" % (CONFIG["SaveDir"], taskid)
+    if os.path.isfile("%s/mock.cfg" % savedir):
+        call(["mock", "-r", "../../%s/mock" % savedir, "--scrub=all"],
+             stdout=null, stderr=null)
+
+    call(["rm", "-rf", "%s/crash" % savedir, "%s/mock.cfg" % savedir],
+         stdout=null, stderr=null)
+
+    rawlog = "%s/log" % savedir
+    newlog = "%s/retrace_log" % savedir
+    if os.path.isfile(rawlog):
+        try:
+            os.rename(rawlog, newlog)
+        except:
+            pass
+
+    try:
+        log = open(newlog, "a")
+        log.write("Killed by garbage collector\n")
+        log.close()
+    except:
+        pass
+
+    null.close()
+
 def init_crashstats_db():
     try:
         con = sqlite3.connect("%s/%s" % (CONFIG["SaveDir"], CONFIG["DBFile"]))