minor fix for a bug reported by shaohua pan (argument of type 'NoneType' is not iterable)

This commit is contained in:
Miroslav Stampar 2010-12-27 11:36:36 +00:00
parent 51a492e17d
commit 68462466f2

View File

@ -1117,8 +1117,12 @@ class Enumeration:
if conf.col:
colList = conf.col.split(",")
else:
elif kb.data.cachedColumns and conf.db in kb.data.cachedColumns and conf.tbl in kb.data.cachedColumns[conf.db]:
colList = kb.data.cachedColumns[conf.db][conf.tbl].keys()
else:
errMsg = "missing column names, "
errMsg += "can't dump table"
raise sqlmapNoneDataException, errMsg
if colList in ([None], ['None']):
warnMsg = "unable to retrieve column names"