Just in case commit related to the aee4c93c8b

This commit is contained in:
Miroslav Stampar 2015-09-10 15:19:33 +02:00
parent 00955a7eb5
commit 7a261ef447

View File

@ -18,6 +18,7 @@ from lib.core.common import readInput
from lib.core.data import conf from lib.core.data import conf
from lib.core.data import logger from lib.core.data import logger
from lib.core.data import paths from lib.core.data import paths
from lib.core.exception import SqlmapDataException
class ICMPsh: class ICMPsh:
""" """
@ -41,6 +42,9 @@ class ICMPsh:
while not address: while not address:
address = readInput(message, default=self.remoteIP) address = readInput(message, default=self.remoteIP)
if conf.batch and not address:
raise SqlmapDataException("remote host address is missing")
return address return address
def _selectLhost(self): def _selectLhost(self):
@ -53,6 +57,9 @@ class ICMPsh:
while not address: while not address:
address = readInput(message, default=self.localIP) address = readInput(message, default=self.localIP)
if conf.batch and not address:
raise SqlmapDataException("local host address is missing")
return address return address
def _prepareIngredients(self, encode=True): def _prepareIngredients(self, encode=True):