Update sql_cli/scanner.py

Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>
This commit is contained in:
Wilbert Chandra 2026-01-07 20:18:55 +07:00 committed by GitHub
parent 5ddf749620
commit ae3920f0b3
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -126,23 +126,29 @@ class SQLMapScanner:
pass pass
return process.returncode == 0, full_output return process.returncode == 0, full_output
else: except Exception as cleanup_error:
# Standard blocking run console.log(
result = subprocess.run(cmd, capture_output=True, text=True) f"Failed to remove temporary sqlmap output directory {tmp_output_dir!r}: {cleanup_error}"
)
# Cleanup temporary output directory
try:
shutil.rmtree(tmp_output_dir)
except:
pass
return result.returncode == 0, result.stdout + result.stderr return result.returncode == 0, result.stdout + result.stderr
except subprocess.TimeoutExpired:
# Cleanup on timeout
try:
shutil.rmtree(tmp_output_dir)
except Exception as cleanup_error:
console.log(
f"Failed to remove temporary sqlmap output directory {tmp_output_dir!r} after timeout: {cleanup_error}"
)
return False, "Test timed out after 10 minutes"
except Exception as e: except Exception as e:
# Cleanup on error # Cleanup on error
try: try:
shutil.rmtree(tmp_output_dir) shutil.rmtree(tmp_output_dir)
except: except Exception as cleanup_error:
pass console.log(
f"Failed to remove temporary sqlmap output directory {tmp_output_dir!r} after error: {cleanup_error}"
)
return False, str(e) return False, str(e)
def parse_results(self, output: str) -> Dict[str, Any]: def parse_results(self, output: str) -> Dict[str, Any]: