sqlmapproject/sqlmap
1
1
Fork 0
mirror of https://github.com/sqlmapproject/sqlmap.git synced 2024-11-22 17:46:37 +03:00
Code Issues Packages Projects Releases Wiki Activity

Fix for Issue #177

Browse Source
This commit is contained in:
Miroslav Stampar 2010-03-16 13:11:44 +00:00
parent 323cf2b7f2
commit 1dfe558d3d
4 changed files with 21 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
lib/controller/controller.py
View File

@ -31,6 +31,8 @@ from lib.controller.checks import checkRegexp
from lib.controller.checks import checkConnection
from lib.core.common import paramToDict
from lib.core.common import parseTargetUrl
from lib.core.common import pop
from lib.core.common import push
from lib.core.common import readInput
from lib.core.data import conf
from lib.core.data import kb
@ -103,6 +105,10 @@ def start():
if kb.targetUrls and len(kb.targetUrls) > 1:
infoMsg = "sqlmap got a total of %d targets" % len(kb.targetUrls)
logger.info(infoMsg)
if conf.multipleTargets:
push(conf.raise404)
conf.raise404 = False
hostCount = 0
cookieStr = ""
@ -267,7 +273,12 @@ def start():
logger.error(e)
else:
logger.error(e)
if conf.multipleTargets:
conf.raise404 = pop()
return
if conf.multipleTargets:
conf.raise404 = pop()
if conf.loggedToOut:
logger.info("Fetched data logged to text files under '%s'" % conf.outputPath)

6
lib/core/common.py
View File

@ -939,3 +939,9 @@ def posixToNtSlashes(filepath):
def ntToPosixSlashes(filepath):
return filepath.replace('\\', '/')
def push(value):
conf.stack.append(value)
def pop():
return conf.stack.pop()

2
lib/core/option.py
View File

@ -957,12 +957,14 @@ def __setConfAttributes():
conf.path = None
conf.port = None
conf.progressWidth = 54
conf.raise404 = True
conf.retriesCount = 0
conf.scheme = None
#conf.seqMatcher = difflib.SequenceMatcher(lambda x: x in " \t")
conf.seqMatcher = difflib.SequenceMatcher(None)
conf.seqLock = None
conf.sessionFP = None
conf.stack = []
conf.start = True
conf.threadContinue = True
conf.threadException = False

4
lib/request/connect.py
View File

@ -70,7 +70,7 @@ class Connect:
direct = kwargs.get('direct', False)
multipart = kwargs.get('multipart', False)
silent = kwargs.get('silent', False)
raise404 = kwargs.get('raise404', True)
raise404 = kwargs.get('raise404', None)
page = ""
cookieStr = ""
@ -181,7 +181,7 @@ class Connect:
errMsg = "not authorized, try to provide right HTTP "
errMsg += "authentication type and valid credentials"
raise sqlmapConnectionException, errMsg
elif e.code == 404 and raise404:
elif e.code == 404 and (raise404 or (raise404 is None and conf.raise404)):
errMsg = "page not found"
raise sqlmapConnectionException, errMsg
else: