Another patch for #3389 (Fixes #3397)

stamparm · stamparm · commit 070e17306721 · 2018-12-10T12:59:13.000+01:00
diff --git a/lib/core/common.py b/lib/core/common.py
@@ -997,11 +997,6 @@ def readInput(message, default=None, checkBatch=True, boolean=False):
     retVal = None
     kb.stickyLevel = None
 
-    if kb.lastInputMessage == message:
-        checkBatch = False
-    else:
-        kb.lastInputMessage = message
-
     message = getUnicode(message)
 
     if "\n" in message:
diff --git a/lib/core/option.py b/lib/core/option.py
@@ -1792,7 +1792,6 @@ def _setKnowledgeBaseAttributes(flushAll=True):
     kb.injection = InjectionDict()
     kb.injections = []
     kb.laggingChecked = False
-    kb.lastInputMessage = None
     kb.lastParserStatus = None
 
     kb.locks = AttribDict()
diff --git a/lib/takeover/metasploit.py b/lib/takeover/metasploit.py
@@ -168,19 +168,8 @@ def _skeletonSelection(self, msg, lst=None, maxValue=1, default=1):
 
         choice = readInput(message, default="%d" % default)
 
-        if not choice:
-            if lst:
-                choice = getUnicode(default, UNICODE_ENCODING)
-            else:
-                return default
-
-        elif not choice.isdigit():
-            logger.warn("invalid value, only digits are allowed")
-            return self._skeletonSelection(msg, lst, maxValue, default)
-
-        elif int(choice) > maxValue or int(choice) < 1:
-            logger.warn("invalid value, it must be a digit between 1 and %d" % maxValue)
-            return self._skeletonSelection(msg, lst, maxValue, default)
+        if not choice or not choice.isdigit() or int(choice) > maxValue or int(choice) < 1:
+            choice = default
 
         choice = int(choice)
 
