Remove testcase_dir if it is empty

sslab-gatech · Aug 6, 2019 · 4e0c8f5 · 4e0c8f5
1 parent 6f00c3d
commit 4e0c8f5
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/qsym/afl.py b/qsym/afl.py
@@ -387,6 +387,12 @@ def run_file(self, fp):
         if os.path.exists(q.log_file):
             os.unlink(q.log_file)
 
+        # Remove testcase_dir if it`s empty
+        try:
+            os.rmdir(q.testcase_directory)
+        except Exception:
+            pass
+
         logger.debug("Generate %d testcases" % num_testcase)
         logger.debug("%d testcases are new" % (self.state.index - old_idx))
 

diff --git a/qsym/executor.py b/qsym/executor.py
@@ -65,6 +65,10 @@ def status_file(self):
     def log_file(self):
         return os.path.join(self.testcase_dir, "pin.log")
 
+    @property
+    def testcase_directory(self):
+        return self.testcase_dir
+
     def check_elf32(self):
         # assume cmd[0] is always the target binary (?)
         if os.path.exists(self.cmd[0]):