From d2a9c7584fd4479f98a97d005812a4bf6d5a6037 Mon Sep 17 00:00:00 2001 From: Miroslav Stampar Date: Mon, 31 Aug 2015 09:51:35 +0200 Subject: [PATCH] Minor patch --- lib/core/target.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/lib/core/target.py b/lib/core/target.py index 50158817a..983d54541 100644 --- a/lib/core/target.py +++ b/lib/core/target.py @@ -223,11 +223,11 @@ def _setRequestParams(): message += "in the target URL itself? [Y/n/q] " test = readInput(message, default="Y") - if not test or test[0] not in ("n", "N"): + if test and test[0] in ("q", "Q"): + raise SqlmapUserQuitException + elif not test or test[0] not in ("n", "N"): conf.url = "%s%s" % (conf.url, CUSTOM_INJECTION_MARK_CHAR) kb.processUserMarks = True - elif test[0] in ("q", "Q"): - raise SqlmapUserQuitException for place, value in ((PLACE.URI, conf.url), (PLACE.CUSTOM_POST, conf.data), (PLACE.CUSTOM_HEADER, str(conf.httpHeaders))): _ = re.sub(PROBLEMATIC_CUSTOM_INJECTION_PATTERNS, "", value or "") if place == PLACE.CUSTOM_HEADER else value or ""