mirror of
https://github.com/sqlmapproject/sqlmap.git
synced 2025-02-03 05:04:11 +03:00
fix for Bug #183 (--threads dot output)
This commit is contained in:
parent
1aeaa5db47
commit
938a3ab0b9
|
@ -32,6 +32,7 @@ import time
|
|||
import urlparse
|
||||
import ntpath
|
||||
import posixpath
|
||||
import subprocess
|
||||
|
||||
from tempfile import NamedTemporaryFile
|
||||
from tempfile import mkstemp
|
||||
|
@ -1062,3 +1063,25 @@ def isBase64EncodedString(subject):
|
|||
|
||||
def isHexEncodedString(subject):
|
||||
return re.match(r"\A[0-9a-fA-F]+\Z", subject) is not None
|
||||
|
||||
def getConsoleWidth(default=80):
|
||||
width = None
|
||||
|
||||
if 'COLUMNS' in os.environ and os.environ['COLUMNS'].isdigit():
|
||||
width = int(os.environ['COLUMNS'])
|
||||
else:
|
||||
output=subprocess.Popen('stty size', shell=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE).stdout.read()
|
||||
items = output.split()
|
||||
if len(items) == 2 and items[1].isdigit():
|
||||
width = int(items[1])
|
||||
|
||||
if width is None:
|
||||
try:
|
||||
import curses
|
||||
stdscr = curses.initscr()
|
||||
_, width = stdscr.getmaxyx()
|
||||
curses.endwin()
|
||||
except:
|
||||
pass
|
||||
|
||||
return width if width else default
|
|
@ -34,6 +34,7 @@ import urlparse
|
|||
|
||||
from ConfigParser import ConfigParser
|
||||
|
||||
from lib.core.common import getConsoleWidth
|
||||
from lib.core.common import getFileType
|
||||
from lib.core.common import normalizePath
|
||||
from lib.core.common import ntToPosixSlashes
|
||||
|
@ -904,10 +905,12 @@ def __setConfAttributes():
|
|||
conf.threadException = False
|
||||
conf.wFileType = None
|
||||
|
||||
width = getConsoleWidth()
|
||||
|
||||
if conf.eta:
|
||||
conf.progressWidth = 54
|
||||
conf.progressWidth = width-26
|
||||
else:
|
||||
conf.progressWidth = 34
|
||||
conf.progressWidth = width-46
|
||||
|
||||
def __setKnowledgeBaseAttributes():
|
||||
"""
|
||||
|
|
Loading…
Reference in New Issue
Block a user