sqlmapproject/sqlmap
1
1
Fork 0
mirror of https://github.com/sqlmapproject/sqlmap.git synced 2025-01-24 08:14:24 +03:00
Code Issues Packages Projects Releases Wiki Activity

code refactoring

Browse Source
This commit is contained in:
Miroslav Stampar 2010-11-23 21:00:42 +00:00
parent 7877a931d5
commit f9f076ba97
3 changed files with 11 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
lib/core/common.py
View File

@ -1259,8 +1259,8 @@ def getConsoleWidth(default=80):
return width if width else default
def clearConsoleLine():
dataToStdout("\r%s\r" % (" " * (getConsoleWidth() - 1)))
def clearConsoleLine(forceOutput=False):
dataToStdout("\r%s\r" % (" " * (getConsoleWidth() - 1)), forceOutput)
def parseXmlFile(xmlFile, handler):
stream = StringIO(readCachedFileContent(xmlFile))

4
lib/core/testing.py
View File

@ -17,9 +17,9 @@ import tempfile
import time
from lib.controller.controller import start
from lib.core.common import clearConsoleLine
from lib.core.common import dataToStdout
from lib.core.common import getCompiledRegex
from lib.core.common import getConsoleWidth
from lib.core.common import readXmlFile
from lib.core.data import conf
from lib.core.data import logger
@ -65,7 +65,7 @@ def smokeTest():
status = '%d/%d (%d%s) ' % (count, length, round(100.0*count/length), '%')
dataToStdout("\r[%s] [INFO] complete: %s" % (time.strftime("%X"), status))
dataToStdout("\r%s\r" % (" "*(getConsoleWidth()-1)))
clearConsoleLine()
if retVal:
logger.info("smoke test final result: PASSED")
else:

14
lib/techniques/brute/use.py
View File

@ -10,8 +10,8 @@ See the file 'doc/COPYING' for copying permission
import time
from lib.core.agent import agent
from lib.core.common import clearConsoleLine
from lib.core.common import dataToStdout
from lib.core.common import getConsoleWidth
from lib.core.common import getFileItems
from lib.core.common import popValue
from lib.core.common import pushValue
@ -42,8 +42,8 @@ def tableExists(tableFile):
result = Request.queryPage(agent.payload(newValue=query))
if result:
infoMsg = "\r[%s] [INFO] retrieved: %s" % (time.strftime("%X"), table)
infoMsg = "%s%s\n" % (infoMsg, " "*(getConsoleWidth()-1-len(infoMsg)))
clearConsoleLine(True)
infoMsg = "\r[%s] [INFO] retrieved: %s\n" % (time.strftime("%X"), table)
dataToStdout(infoMsg, True)
retVal.append(table)
@ -53,7 +53,7 @@ def tableExists(tableFile):
conf.verbose = popValue()
dataToStdout("\n", True)
clearConsoleLine(True)
if not retVal:
warnMsg = "no table found"
@ -93,8 +93,8 @@ def columnExists(columnFile):
result = Request.queryPage(agent.payload(newValue=query))
if result:
infoMsg = "\r[%s] [INFO] retrieved: %s" % (time.strftime("%X"), column)
infoMsg = "%s%s\n" % (infoMsg, " "*(getConsoleWidth()-1-len(infoMsg)))
clearConsoleLine(True)
infoMsg = "\r[%s] [INFO] retrieved: %s\n" % (time.strftime("%X"), column)
dataToStdout(infoMsg, True)
retVal.append(column)
@ -104,7 +104,7 @@ def columnExists(columnFile):
conf.verbose = popValue()
dataToStdout("\n", True)
clearConsoleLine(True)
if not retVal:
warnMsg = "no column found"