mirror of
https://github.com/sqlmapproject/sqlmap.git
synced 2026-01-21 15:54:27 +03:00
Update sql_cli/scanner.py
Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>
This commit is contained in:
parent
5ddf749620
commit
ae3920f0b3
|
|
@ -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]:
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue
Block a user