From 1cfd6a6b9d05ba9b3fe6fc6a70d11311e820c212 Mon Sep 17 00:00:00 2001 From: Bernardo Damele Date: Fri, 14 Jan 2011 15:16:34 +0000 Subject: [PATCH] Code cleanup --- lib/core/common.py | 2 ++ lib/core/datatype.py | 6 +++--- 2 files changed, 5 insertions(+), 3 deletions(-) diff --git a/lib/core/common.py b/lib/core/common.py index 5459a56f7..0fc69fd2e 100644 --- a/lib/core/common.py +++ b/lib/core/common.py @@ -1950,10 +1950,12 @@ def initTechnique(technique=None): conf.textOnly = True debugMsg = "restoring switch --text-only" logger.debug(debugMsg) + if kb.injection.conf.string: conf.string = kb.injection.conf.string debugMsg = "restoring option --string '%s'" % conf.string logger.debug(debugMsg) + if kb.injection.conf.regexp: conf.regexp = kb.injection.conf.regexp debugMsg = "restoring option --regexp '%s'" % conf.regexp diff --git a/lib/core/datatype.py b/lib/core/datatype.py index 4baa321ae..e60cab473 100644 --- a/lib/core/datatype.py +++ b/lib/core/datatype.py @@ -73,11 +73,11 @@ class injectionDict(advancedDict): self.suffix = None self.clause = None - # data is a dict with stype as key and a tuple as value with - # title, where, comment and reqPayload + # data is a dict with various stype, each which is a dict with + # all the information specific for that stype self.data = advancedDict() - # conf is a dict with stores current snapshot of important + # conf is a dict which stores current snapshot of important # options used during detection self.conf = advancedDict()