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 break
if not isNoneValue(value): 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: if not kb.data.cachedTables and isInferenceAvailable() and not conf.direct:
for db in dbs: for db in dbs:

View File

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