mirror of
https://github.com/sqlmapproject/sqlmap.git
synced 2024-11-23 01:56:36 +03:00
629 lines
28 KiB
Python
629 lines
28 KiB
Python
#!/usr/bin/env python2
|
|
|
|
"""
|
|
Copyright (c) 2006-2019 sqlmap developers (http://sqlmap.org/)
|
|
See the file 'LICENSE' for copying permission
|
|
"""
|
|
|
|
import re
|
|
|
|
from lib.core.agent import agent
|
|
from lib.core.bigarray import BigArray
|
|
from lib.core.common import Backend
|
|
from lib.core.common import clearConsoleLine
|
|
from lib.core.common import getLimitRange
|
|
from lib.core.common import getSafeExString
|
|
from lib.core.common import isInferenceAvailable
|
|
from lib.core.common import isListLike
|
|
from lib.core.common import isNoneValue
|
|
from lib.core.common import isNumPosStrValue
|
|
from lib.core.common import isTechniqueAvailable
|
|
from lib.core.common import prioritySortColumns
|
|
from lib.core.common import readInput
|
|
from lib.core.common import safeSQLIdentificatorNaming
|
|
from lib.core.common import singleTimeLogMessage
|
|
from lib.core.common import singleTimeWarnMessage
|
|
from lib.core.common import unArrayizeValue
|
|
from lib.core.common import unsafeSQLIdentificatorNaming
|
|
from lib.core.convert import getUnicode
|
|
from lib.core.data import conf
|
|
from lib.core.data import kb
|
|
from lib.core.data import logger
|
|
from lib.core.data import queries
|
|
from lib.core.dicts import DUMP_REPLACEMENTS
|
|
from lib.core.enums import CHARSET_TYPE
|
|
from lib.core.enums import DBMS
|
|
from lib.core.enums import EXPECTED
|
|
from lib.core.enums import PAYLOAD
|
|
from lib.core.exception import SqlmapConnectionException
|
|
from lib.core.exception import SqlmapMissingMandatoryOptionException
|
|
from lib.core.exception import SqlmapNoneDataException
|
|
from lib.core.exception import SqlmapUnsupportedFeatureException
|
|
from lib.core.settings import CHECK_ZERO_COLUMNS_THRESHOLD
|
|
from lib.core.settings import CURRENT_DB
|
|
from lib.core.settings import NULL
|
|
from lib.request import inject
|
|
from lib.utils.hash import attackDumpedTable
|
|
from lib.utils.pivotdumptable import pivotDumpTable
|
|
from thirdparty import six
|
|
from thirdparty.six.moves import zip as _zip
|
|
|
|
class Entries:
|
|
"""
|
|
This class defines entries' enumeration functionalities for plugins.
|
|
"""
|
|
|
|
def __init__(self):
|
|
pass
|
|
|
|
def dumpTable(self, foundData=None):
|
|
self.forceDbmsEnum()
|
|
|
|
if conf.db is None or conf.db == CURRENT_DB:
|
|
if conf.db is None:
|
|
warnMsg = "missing database parameter. sqlmap is going "
|
|
warnMsg += "to use the current database to enumerate "
|
|
warnMsg += "table(s) entries"
|
|
logger.warn(warnMsg)
|
|
|
|
conf.db = self.getCurrentDb()
|
|
|
|
elif conf.db is not None:
|
|
if Backend.getIdentifiedDbms() in (DBMS.ORACLE, DBMS.DB2, DBMS.HSQLDB, DBMS.H2):
|
|
conf.db = conf.db.upper()
|
|
|
|
if ',' in conf.db:
|
|
errMsg = "only one database name is allowed when enumerating "
|
|
errMsg += "the tables' columns"
|
|
raise SqlmapMissingMandatoryOptionException(errMsg)
|
|
|
|
if conf.exclude and conf.db in conf.exclude.split(','):
|
|
infoMsg = "skipping database '%s'" % unsafeSQLIdentificatorNaming(conf.db)
|
|
singleTimeLogMessage(infoMsg)
|
|
return
|
|
|
|
conf.db = safeSQLIdentificatorNaming(conf.db)
|
|
|
|
if conf.tbl:
|
|
if Backend.getIdentifiedDbms() in (DBMS.ORACLE, DBMS.DB2, DBMS.HSQLDB, DBMS.H2):
|
|
conf.tbl = conf.tbl.upper()
|
|
|
|
tblList = conf.tbl.split(',')
|
|
else:
|
|
self.getTables()
|
|
|
|
if len(kb.data.cachedTables) > 0:
|
|
tblList = kb.data.cachedTables.values()
|
|
|
|
if isinstance(tblList[0], (set, tuple, list)):
|
|
tblList = tblList[0]
|
|
elif not conf.search:
|
|
errMsg = "unable to retrieve the tables "
|
|
errMsg += "in database '%s'" % unsafeSQLIdentificatorNaming(conf.db)
|
|
raise SqlmapNoneDataException(errMsg)
|
|
else:
|
|
return
|
|
|
|
for tbl in tblList:
|
|
tblList[tblList.index(tbl)] = safeSQLIdentificatorNaming(tbl, True)
|
|
|
|
for tbl in tblList:
|
|
if kb.dumpKeyboardInterrupt:
|
|
break
|
|
|
|
if conf.exclude and tbl in conf.exclude.split(','):
|
|
infoMsg = "skipping table '%s'" % unsafeSQLIdentificatorNaming(tbl)
|
|
singleTimeLogMessage(infoMsg)
|
|
continue
|
|
|
|
conf.tbl = tbl
|
|
kb.data.dumpedTable = {}
|
|
|
|
if foundData is None:
|
|
kb.data.cachedColumns = {}
|
|
self.getColumns(onlyColNames=True, dumpMode=True)
|
|
else:
|
|
kb.data.cachedColumns = foundData
|
|
|
|
try:
|
|
if Backend.isDbms(DBMS.INFORMIX):
|
|
kb.dumpTable = "%s:%s" % (conf.db, tbl)
|
|
else:
|
|
kb.dumpTable = "%s.%s" % (conf.db, tbl)
|
|
|
|
if safeSQLIdentificatorNaming(conf.db) not in kb.data.cachedColumns or safeSQLIdentificatorNaming(tbl, True) not in kb.data.cachedColumns[safeSQLIdentificatorNaming(conf.db)] or not kb.data.cachedColumns[safeSQLIdentificatorNaming(conf.db)][safeSQLIdentificatorNaming(tbl, True)]:
|
|
warnMsg = "unable to enumerate the columns for table "
|
|
warnMsg += "'%s' in database" % unsafeSQLIdentificatorNaming(tbl)
|
|
warnMsg += " '%s'" % unsafeSQLIdentificatorNaming(conf.db)
|
|
warnMsg += ", skipping" if len(tblList) > 1 else ""
|
|
logger.warn(warnMsg)
|
|
|
|
continue
|
|
|
|
columns = kb.data.cachedColumns[safeSQLIdentificatorNaming(conf.db)][safeSQLIdentificatorNaming(tbl, True)]
|
|
colList = sorted(column for column in columns if column)
|
|
|
|
if conf.exclude:
|
|
colList = [_ for _ in colList if _ not in conf.exclude.split(',')]
|
|
|
|
if not colList:
|
|
warnMsg = "skipping table '%s'" % unsafeSQLIdentificatorNaming(tbl)
|
|
warnMsg += " in database '%s'" % unsafeSQLIdentificatorNaming(conf.db)
|
|
warnMsg += " (no usable column names)"
|
|
logger.warn(warnMsg)
|
|
continue
|
|
|
|
kb.dumpColumns = colList
|
|
colNames = colString = ", ".join(column for column in colList)
|
|
rootQuery = queries[Backend.getIdentifiedDbms()].dump_table
|
|
|
|
infoMsg = "fetching entries"
|
|
if conf.col:
|
|
infoMsg += " of column(s) '%s'" % colNames
|
|
infoMsg += " for table '%s'" % unsafeSQLIdentificatorNaming(tbl)
|
|
infoMsg += " in database '%s'" % unsafeSQLIdentificatorNaming(conf.db)
|
|
logger.info(infoMsg)
|
|
|
|
for column in colList:
|
|
_ = agent.preprocessField(tbl, column)
|
|
if _ != column:
|
|
colString = re.sub(r"\b%s\b" % re.escape(column), _, colString)
|
|
|
|
entriesCount = 0
|
|
|
|
if any(isTechniqueAvailable(_) for _ in (PAYLOAD.TECHNIQUE.UNION, PAYLOAD.TECHNIQUE.ERROR, PAYLOAD.TECHNIQUE.QUERY)) or conf.direct:
|
|
entries = []
|
|
query = None
|
|
|
|
if Backend.getIdentifiedDbms() in (DBMS.ORACLE, DBMS.DB2):
|
|
query = rootQuery.inband.query % (colString, tbl.upper() if not conf.db else ("%s.%s" % (conf.db.upper(), tbl.upper())))
|
|
elif Backend.getIdentifiedDbms() in (DBMS.SQLITE, DBMS.ACCESS, DBMS.FIREBIRD, DBMS.MAXDB):
|
|
query = rootQuery.inband.query % (colString, tbl)
|
|
elif Backend.getIdentifiedDbms() in (DBMS.SYBASE, DBMS.MSSQL):
|
|
# Partial inband and error
|
|
if not (isTechniqueAvailable(PAYLOAD.TECHNIQUE.UNION) and kb.injection.data[PAYLOAD.TECHNIQUE.UNION].where == PAYLOAD.WHERE.ORIGINAL):
|
|
table = "%s.%s" % (conf.db, tbl)
|
|
|
|
if Backend.isDbms(DBMS.MSSQL) and not conf.forcePivoting:
|
|
warnMsg = "in case of table dumping problems (e.g. column entry order) "
|
|
warnMsg += "you are advised to rerun with '--force-pivoting'"
|
|
singleTimeWarnMessage(warnMsg)
|
|
|
|
query = rootQuery.blind.count % table
|
|
query = agent.whereQuery(query)
|
|
|
|
count = inject.getValue(query, blind=False, time=False, expected=EXPECTED.INT, charsetType=CHARSET_TYPE.DIGITS)
|
|
if isNumPosStrValue(count):
|
|
try:
|
|
indexRange = getLimitRange(count, plusOne=True)
|
|
|
|
for index in indexRange:
|
|
row = []
|
|
for column in colList:
|
|
query = rootQuery.blind.query3 % (column, column, table, index)
|
|
query = agent.whereQuery(query)
|
|
value = inject.getValue(query, blind=False, time=False, dump=True) or ""
|
|
row.append(value)
|
|
|
|
entries.append(row)
|
|
|
|
except KeyboardInterrupt:
|
|
kb.dumpKeyboardInterrupt = True
|
|
clearConsoleLine()
|
|
warnMsg = "Ctrl+C detected in dumping phase"
|
|
logger.warn(warnMsg)
|
|
|
|
if not entries and not kb.dumpKeyboardInterrupt:
|
|
try:
|
|
retVal = pivotDumpTable(table, colList, blind=False)
|
|
except KeyboardInterrupt:
|
|
retVal = None
|
|
kb.dumpKeyboardInterrupt = True
|
|
clearConsoleLine()
|
|
warnMsg = "Ctrl+C detected in dumping phase"
|
|
logger.warn(warnMsg)
|
|
|
|
if retVal:
|
|
entries, _ = retVal
|
|
entries = _zip(*[entries[colName] for colName in colList])
|
|
else:
|
|
query = rootQuery.inband.query % (colString, conf.db, tbl)
|
|
elif Backend.getIdentifiedDbms() in (DBMS.MYSQL, DBMS.PGSQL, DBMS.HSQLDB, DBMS.H2):
|
|
query = rootQuery.inband.query % (colString, conf.db, tbl, prioritySortColumns(colList)[0])
|
|
else:
|
|
query = rootQuery.inband.query % (colString, conf.db, tbl)
|
|
|
|
query = agent.whereQuery(query)
|
|
|
|
if not entries and query and not kb.dumpKeyboardInterrupt:
|
|
try:
|
|
entries = inject.getValue(query, blind=False, time=False, dump=True)
|
|
except KeyboardInterrupt:
|
|
entries = None
|
|
kb.dumpKeyboardInterrupt = True
|
|
clearConsoleLine()
|
|
warnMsg = "Ctrl+C detected in dumping phase"
|
|
logger.warn(warnMsg)
|
|
|
|
if not isNoneValue(entries):
|
|
if isinstance(entries, six.string_types):
|
|
entries = [entries]
|
|
elif not isListLike(entries):
|
|
entries = []
|
|
|
|
entriesCount = len(entries)
|
|
|
|
for index, column in enumerate(colList):
|
|
if column not in kb.data.dumpedTable:
|
|
kb.data.dumpedTable[column] = {"length": len(column), "values": BigArray()}
|
|
|
|
for entry in entries:
|
|
if entry is None or len(entry) == 0:
|
|
continue
|
|
|
|
if isinstance(entry, six.string_types):
|
|
colEntry = entry
|
|
else:
|
|
colEntry = unArrayizeValue(entry[index]) if index < len(entry) else u''
|
|
|
|
maxLen = max(len(column), len(DUMP_REPLACEMENTS.get(getUnicode(colEntry), getUnicode(colEntry))))
|
|
|
|
if maxLen > kb.data.dumpedTable[column]["length"]:
|
|
kb.data.dumpedTable[column]["length"] = maxLen
|
|
|
|
kb.data.dumpedTable[column]["values"].append(colEntry)
|
|
|
|
if not kb.data.dumpedTable and isInferenceAvailable() and not conf.direct:
|
|
infoMsg = "fetching number of "
|
|
if conf.col:
|
|
infoMsg += "column(s) '%s' " % colNames
|
|
infoMsg += "entries for table '%s' " % unsafeSQLIdentificatorNaming(tbl)
|
|
infoMsg += "in database '%s'" % unsafeSQLIdentificatorNaming(conf.db)
|
|
logger.info(infoMsg)
|
|
|
|
if Backend.getIdentifiedDbms() in (DBMS.ORACLE, DBMS.DB2):
|
|
query = rootQuery.blind.count % (tbl.upper() if not conf.db else ("%s.%s" % (conf.db.upper(), tbl.upper())))
|
|
elif Backend.getIdentifiedDbms() in (DBMS.SQLITE, DBMS.ACCESS, DBMS.FIREBIRD):
|
|
query = rootQuery.blind.count % tbl
|
|
elif Backend.getIdentifiedDbms() in (DBMS.SYBASE, DBMS.MSSQL):
|
|
query = rootQuery.blind.count % ("%s.%s" % (conf.db, tbl))
|
|
elif Backend.isDbms(DBMS.MAXDB):
|
|
query = rootQuery.blind.count % tbl
|
|
elif Backend.isDbms(DBMS.INFORMIX):
|
|
query = rootQuery.blind.count % (conf.db, tbl)
|
|
else:
|
|
query = rootQuery.blind.count % (conf.db, tbl)
|
|
|
|
query = agent.whereQuery(query)
|
|
|
|
count = inject.getValue(query, union=False, error=False, expected=EXPECTED.INT, charsetType=CHARSET_TYPE.DIGITS)
|
|
|
|
lengths = {}
|
|
entries = {}
|
|
|
|
if count == 0:
|
|
warnMsg = "table '%s' " % unsafeSQLIdentificatorNaming(tbl)
|
|
warnMsg += "in database '%s' " % unsafeSQLIdentificatorNaming(conf.db)
|
|
warnMsg += "appears to be empty"
|
|
logger.warn(warnMsg)
|
|
|
|
for column in colList:
|
|
lengths[column] = len(column)
|
|
entries[column] = []
|
|
|
|
elif not isNumPosStrValue(count):
|
|
warnMsg = "unable to retrieve the number of "
|
|
if conf.col:
|
|
warnMsg += "column(s) '%s' " % colNames
|
|
warnMsg += "entries for table '%s' " % unsafeSQLIdentificatorNaming(tbl)
|
|
warnMsg += "in database '%s'" % unsafeSQLIdentificatorNaming(conf.db)
|
|
logger.warn(warnMsg)
|
|
|
|
continue
|
|
|
|
elif Backend.getIdentifiedDbms() in (DBMS.ACCESS, DBMS.SYBASE, DBMS.MAXDB, DBMS.MSSQL, DBMS.INFORMIX):
|
|
if Backend.isDbms(DBMS.ACCESS):
|
|
table = tbl
|
|
elif Backend.getIdentifiedDbms() in (DBMS.SYBASE, DBMS.MSSQL):
|
|
table = "%s.%s" % (conf.db, tbl)
|
|
elif Backend.isDbms(DBMS.MAXDB):
|
|
table = "%s.%s" % (conf.db, tbl)
|
|
elif Backend.isDbms(DBMS.INFORMIX):
|
|
table = "%s:%s" % (conf.db, tbl)
|
|
|
|
if Backend.isDbms(DBMS.MSSQL) and not conf.forcePivoting:
|
|
warnMsg = "in case of table dumping problems (e.g. column entry order) "
|
|
warnMsg += "you are advised to rerun with '--force-pivoting'"
|
|
singleTimeWarnMessage(warnMsg)
|
|
|
|
try:
|
|
indexRange = getLimitRange(count, plusOne=True)
|
|
|
|
for index in indexRange:
|
|
for column in colList:
|
|
query = rootQuery.blind.query3 % (column, column, table, index)
|
|
query = agent.whereQuery(query)
|
|
|
|
value = inject.getValue(query, union=False, error=False, dump=True) or ""
|
|
|
|
if column not in lengths:
|
|
lengths[column] = 0
|
|
|
|
if column not in entries:
|
|
entries[column] = BigArray()
|
|
|
|
lengths[column] = max(lengths[column], len(DUMP_REPLACEMENTS.get(getUnicode(value), getUnicode(value))))
|
|
entries[column].append(value)
|
|
|
|
except KeyboardInterrupt:
|
|
kb.dumpKeyboardInterrupt = True
|
|
clearConsoleLine()
|
|
warnMsg = "Ctrl+C detected in dumping phase"
|
|
logger.warn(warnMsg)
|
|
|
|
if not entries and not kb.dumpKeyboardInterrupt:
|
|
try:
|
|
retVal = pivotDumpTable(table, colList, count, blind=True)
|
|
except KeyboardInterrupt:
|
|
retVal = None
|
|
kb.dumpKeyboardInterrupt = True
|
|
clearConsoleLine()
|
|
warnMsg = "Ctrl+C detected in dumping phase"
|
|
logger.warn(warnMsg)
|
|
|
|
if retVal:
|
|
entries, lengths = retVal
|
|
|
|
else:
|
|
emptyColumns = []
|
|
plusOne = Backend.getIdentifiedDbms() in (DBMS.ORACLE, DBMS.DB2)
|
|
indexRange = getLimitRange(count, plusOne=plusOne)
|
|
|
|
if len(colList) < len(indexRange) > CHECK_ZERO_COLUMNS_THRESHOLD:
|
|
debugMsg = "checking for empty columns"
|
|
logger.debug(infoMsg)
|
|
|
|
for column in colList:
|
|
if not inject.checkBooleanExpression("(SELECT COUNT(%s) FROM %s)>0" % (column, kb.dumpTable)):
|
|
emptyColumns.append(column)
|
|
debugMsg = "column '%s' of table '%s' will not be " % (column, kb.dumpTable)
|
|
debugMsg += "dumped as it appears to be empty"
|
|
logger.debug(debugMsg)
|
|
|
|
try:
|
|
for index in indexRange:
|
|
for column in colList:
|
|
value = ""
|
|
|
|
if column not in lengths:
|
|
lengths[column] = 0
|
|
|
|
if column not in entries:
|
|
entries[column] = BigArray()
|
|
|
|
if Backend.getIdentifiedDbms() in (DBMS.MYSQL, DBMS.PGSQL, DBMS.HSQLDB, DBMS.H2):
|
|
query = rootQuery.blind.query % (agent.preprocessField(tbl, column), conf.db, conf.tbl, sorted(colList, key=len)[0], index)
|
|
elif Backend.getIdentifiedDbms() in (DBMS.ORACLE, DBMS.DB2):
|
|
query = rootQuery.blind.query % (agent.preprocessField(tbl, column), tbl.upper() if not conf.db else ("%s.%s" % (conf.db.upper(), tbl.upper())), index)
|
|
elif Backend.isDbms(DBMS.SQLITE):
|
|
query = rootQuery.blind.query % (agent.preprocessField(tbl, column), tbl, index)
|
|
elif Backend.isDbms(DBMS.FIREBIRD):
|
|
query = rootQuery.blind.query % (index, agent.preprocessField(tbl, column), tbl)
|
|
elif Backend.isDbms(DBMS.INFORMIX):
|
|
query = rootQuery.blind.query % (index, agent.preprocessField(tbl, column), conf.db, tbl, sorted(colList, key=len)[0])
|
|
|
|
query = agent.whereQuery(query)
|
|
|
|
value = NULL if column in emptyColumns else inject.getValue(query, union=False, error=False, dump=True)
|
|
value = '' if value is None else value
|
|
|
|
lengths[column] = max(lengths[column], len(DUMP_REPLACEMENTS.get(getUnicode(value), getUnicode(value))))
|
|
entries[column].append(value)
|
|
|
|
except KeyboardInterrupt:
|
|
kb.dumpKeyboardInterrupt = True
|
|
clearConsoleLine()
|
|
warnMsg = "Ctrl+C detected in dumping phase"
|
|
logger.warn(warnMsg)
|
|
|
|
for column, columnEntries in entries.items():
|
|
length = max(lengths[column], len(column))
|
|
|
|
kb.data.dumpedTable[column] = {"length": length, "values": columnEntries}
|
|
|
|
entriesCount = len(columnEntries)
|
|
|
|
if len(kb.data.dumpedTable) == 0 or (entriesCount == 0 and kb.permissionFlag):
|
|
warnMsg = "unable to retrieve the entries "
|
|
if conf.col:
|
|
warnMsg += "of columns '%s' " % colNames
|
|
warnMsg += "for table '%s' " % unsafeSQLIdentificatorNaming(tbl)
|
|
warnMsg += "in database '%s'%s" % (unsafeSQLIdentificatorNaming(conf.db), " (permission denied)" if kb.permissionFlag else "")
|
|
logger.warn(warnMsg)
|
|
else:
|
|
kb.data.dumpedTable["__infos__"] = {"count": entriesCount,
|
|
"table": safeSQLIdentificatorNaming(tbl, True),
|
|
"db": safeSQLIdentificatorNaming(conf.db)}
|
|
try:
|
|
attackDumpedTable()
|
|
except (IOError, OSError) as ex:
|
|
errMsg = "an error occurred while attacking "
|
|
errMsg += "table dump ('%s')" % getSafeExString(ex)
|
|
logger.critical(errMsg)
|
|
conf.dumper.dbTableValues(kb.data.dumpedTable)
|
|
|
|
except SqlmapConnectionException as ex:
|
|
errMsg = "connection exception detected in dumping phase "
|
|
errMsg += "('%s')" % getSafeExString(ex)
|
|
logger.critical(errMsg)
|
|
|
|
finally:
|
|
kb.dumpColumns = None
|
|
kb.dumpTable = None
|
|
|
|
def dumpAll(self):
|
|
if conf.db is not None and conf.tbl is None:
|
|
self.dumpTable()
|
|
return
|
|
|
|
if Backend.isDbms(DBMS.MYSQL) and not kb.data.has_information_schema:
|
|
errMsg = "information_schema not available, "
|
|
errMsg += "back-end DBMS is MySQL < 5.0"
|
|
raise SqlmapUnsupportedFeatureException(errMsg)
|
|
|
|
infoMsg = "sqlmap will dump entries of all tables from all databases now"
|
|
logger.info(infoMsg)
|
|
|
|
conf.tbl = None
|
|
conf.col = None
|
|
|
|
self.getTables()
|
|
|
|
if kb.data.cachedTables:
|
|
if isinstance(kb.data.cachedTables, list):
|
|
kb.data.cachedTables = {None: kb.data.cachedTables}
|
|
|
|
for db, tables in kb.data.cachedTables.items():
|
|
conf.db = db
|
|
|
|
for table in tables:
|
|
if conf.exclude and table in conf.exclude.split(','):
|
|
infoMsg = "skipping table '%s'" % unsafeSQLIdentificatorNaming(table)
|
|
logger.info(infoMsg)
|
|
continue
|
|
|
|
try:
|
|
conf.tbl = table
|
|
kb.data.cachedColumns = {}
|
|
kb.data.dumpedTable = {}
|
|
|
|
self.dumpTable()
|
|
except SqlmapNoneDataException:
|
|
infoMsg = "skipping table '%s'" % unsafeSQLIdentificatorNaming(table)
|
|
logger.info(infoMsg)
|
|
|
|
def dumpFoundColumn(self, dbs, foundCols, colConsider):
|
|
message = "do you want to dump entries? [Y/n] "
|
|
|
|
if not readInput(message, default='Y', boolean=True):
|
|
return
|
|
|
|
dumpFromDbs = []
|
|
message = "which database(s)?\n[a]ll (default)\n"
|
|
|
|
for db, tblData in dbs.items():
|
|
if tblData:
|
|
message += "[%s]\n" % unsafeSQLIdentificatorNaming(db)
|
|
|
|
message += "[q]uit"
|
|
choice = readInput(message, default='a')
|
|
|
|
if not choice or choice in ('a', 'A'):
|
|
dumpFromDbs = list(dbs.keys())
|
|
elif choice in ('q', 'Q'):
|
|
return
|
|
else:
|
|
dumpFromDbs = choice.replace(" ", "").split(',')
|
|
|
|
for db, tblData in dbs.items():
|
|
if db not in dumpFromDbs or not tblData:
|
|
continue
|
|
|
|
conf.db = db
|
|
dumpFromTbls = []
|
|
message = "which table(s) of database '%s'?\n" % unsafeSQLIdentificatorNaming(db)
|
|
message += "[a]ll (default)\n"
|
|
|
|
for tbl in tblData:
|
|
message += "[%s]\n" % tbl
|
|
|
|
message += "[s]kip\n"
|
|
message += "[q]uit"
|
|
choice = readInput(message, default='a')
|
|
|
|
if not choice or choice in ('a', 'A'):
|
|
dumpFromTbls = tblData
|
|
elif choice in ('s', 'S'):
|
|
continue
|
|
elif choice in ('q', 'Q'):
|
|
return
|
|
else:
|
|
dumpFromTbls = choice.replace(" ", "").split(',')
|
|
|
|
for table, columns in tblData.items():
|
|
if table not in dumpFromTbls:
|
|
continue
|
|
|
|
conf.tbl = table
|
|
colList = [_ for _ in columns if _]
|
|
|
|
if conf.exclude:
|
|
colList = [_ for _ in colList if _ not in conf.exclude.split(',')]
|
|
|
|
conf.col = ','.join(colList)
|
|
kb.data.cachedColumns = {}
|
|
kb.data.dumpedTable = {}
|
|
|
|
data = self.dumpTable(dbs)
|
|
|
|
if data:
|
|
conf.dumper.dbTableValues(data)
|
|
|
|
def dumpFoundTables(self, tables):
|
|
message = "do you want to dump tables' entries? [Y/n] "
|
|
|
|
if not readInput(message, default='Y', boolean=True):
|
|
return
|
|
|
|
dumpFromDbs = []
|
|
message = "which database(s)?\n[a]ll (default)\n"
|
|
|
|
for db, tablesList in tables.items():
|
|
if tablesList:
|
|
message += "[%s]\n" % unsafeSQLIdentificatorNaming(db)
|
|
|
|
message += "[q]uit"
|
|
choice = readInput(message, default='a')
|
|
|
|
if not choice or choice.lower() == 'a':
|
|
dumpFromDbs = list(tables.keys())
|
|
elif choice.lower() == 'q':
|
|
return
|
|
else:
|
|
dumpFromDbs = choice.replace(" ", "").split(',')
|
|
|
|
for db, tablesList in tables.items():
|
|
if db not in dumpFromDbs or not tablesList:
|
|
continue
|
|
|
|
conf.db = db
|
|
dumpFromTbls = []
|
|
message = "which table(s) of database '%s'?\n" % unsafeSQLIdentificatorNaming(db)
|
|
message += "[a]ll (default)\n"
|
|
|
|
for tbl in tablesList:
|
|
message += "[%s]\n" % unsafeSQLIdentificatorNaming(tbl)
|
|
|
|
message += "[s]kip\n"
|
|
message += "[q]uit"
|
|
choice = readInput(message, default='a')
|
|
|
|
if not choice or choice.lower() == 'a':
|
|
dumpFromTbls = tablesList
|
|
elif choice.lower() == 's':
|
|
continue
|
|
elif choice.lower() == 'q':
|
|
return
|
|
else:
|
|
dumpFromTbls = choice.replace(" ", "").split(',')
|
|
|
|
for table in dumpFromTbls:
|
|
conf.tbl = table
|
|
kb.data.cachedColumns = {}
|
|
kb.data.dumpedTable = {}
|
|
|
|
data = self.dumpTable()
|
|
|
|
if data:
|
|
conf.dumper.dbTableValues(data)
|