handle folder removal

SecurityResearchStaff · Nov 24, 2017 · 843ad7b · 843ad7b
1 parent b9f7b4f
commit 843ad7b
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/androguard/decompiler/decompiler.py b/androguard/decompiler/decompiler.py
@@ -586,6 +586,7 @@ def __init__(self, vm, vmx, jadx="jadx", keepfiles=False):
             print(stdout)
 
             if x.returncode != 0:
+                rrmdir(tmpfolder)
                 raise JADXDecompilerError("Could not decompile file. Args: {}".format(" ".join(cmd)))
 
         # Next we parse the folder structure for later lookup
@@ -626,7 +627,8 @@ def __init__(self, vm, vmx, jadx="jadx", keepfiles=False):
             if not os.path.isfile(os.path.join(tmpfolder, cl.replace("/", os.sep))):
                 print("Found a class called {} which is not decompiled by jadx".format(cl), file=sys.stderr)
 
-        rrmdir(tmpfolder)
+        if not keepfiles:
+            rrmdir(tmpfolder)
 
     def get_source_method(self, m):
         """