PenTest/sqlmap
1
0
Fork 0
mirror of https://github.com/sqlmapproject/sqlmap.git synced 2025-12-06 20:51:31 +00:00
Code Issues Projects Releases Wiki Activity

major enhancement, code refactoring for issue #297

Browse Source
This commit is contained in:
Bernardo Damele
2013-01-29 01:39:27 +00:00
parent d07881b6c3
commit f1ab887c55
6 changed files with 281 additions and 168 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
_sqlmap.py
View File

@@ -59,11 +59,11 @@ def main():
cmdLineOptions.update(cmdLineParser().__dict__)
init(cmdLineOptions)
if hasattr(conf, "ipc_database"):
if hasattr(conf, "api"):
# Overwrite system standard output and standard error to write
# to a temporary I/O database
sys.stdout = StdDbOut(type_="stdout")
sys.stderr = StdDbOut(type_="stderr")
# to an IPC database
sys.stdout = StdDbOut(conf.taskid, messagetype="stdout")
sys.stderr = StdDbOut(conf.taskid, messagetype="stderr")
banner()
@@ -122,10 +122,10 @@ def main():
except KeyboardInterrupt:
pass
if hasattr(conf, "ipc_database"):
if hasattr(conf, "api"):
try:
conf.ipc_database_cursor.close()
conf.ipc_database_connection.close()
conf.database_cursor.close()
conf.database_connection.close()
except KeyboardInterrupt:
pass