one more just in case fix for safeSQLIdentificator naming on MSSQL --tables

This commit is contained in:
Miroslav Stampar 2012-02-29 14:05:53 +00:00
parent d06182347f
commit 10dd9096f7
2 changed files with 7 additions and 4 deletions

View File

@ -102,7 +102,9 @@ class Enumeration(GenericEnumeration):
break
if not isNoneValue(value):
kb.data.cachedTables[db] = arrayizeValue(value)
value = filter(None, arrayizeValue(value))
value = [safeSQLIdentificatorNaming(_, True) for _ in value]
kb.data.cachedTables[db] = value
if not kb.data.cachedTables and isInferenceAvailable() and not conf.direct:
for db in dbs:

View File

@ -927,9 +927,10 @@ class Enumeration:
query = rootQuery.blind.query % (unsafeSQLIdentificatorNaming(db), index)
table = inject.getValue(query, inband=False, error=False)
kb.hintValue = table
table = safeSQLIdentificatorNaming(table, True)
tables.append(table)
if not isNoneValue(table):
kb.hintValue = table
table = safeSQLIdentificatorNaming(table, True)
tables.append(table)
if tables:
kb.data.cachedTables[db] = tables