mirror of
https://github.com/sqlmapproject/sqlmap.git
synced 2024-11-22 17:46:37 +03:00
312 lines
11 KiB
Python
312 lines
11 KiB
Python
#!/usr/bin/env python
|
|
|
|
"""
|
|
$Id$
|
|
|
|
Copyright (c) 2006-2010 sqlmap developers (http://sqlmap.sourceforge.net/)
|
|
See the file 'doc/COPYING' for copying permission
|
|
"""
|
|
|
|
from lib.core.agent import agent
|
|
from lib.core.common import arrayizeValue
|
|
from lib.core.common import getRange
|
|
from lib.core.common import isNumPosStrValue
|
|
from lib.core.common import isTechniqueAvailable
|
|
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.enums import DBMS
|
|
from lib.core.enums import EXPECTED
|
|
from lib.core.exception import sqlmapNoneDataException
|
|
from lib.request import inject
|
|
|
|
from plugins.generic.enumeration import Enumeration as GenericEnumeration
|
|
|
|
class Enumeration(GenericEnumeration):
|
|
def __init__(self):
|
|
GenericEnumeration.__init__(self, DBMS.MSSQL)
|
|
|
|
def getPrivileges(self, *args):
|
|
warnMsg = "on Microsoft SQL Server it is not possible to fetch "
|
|
warnMsg += "database users privileges"
|
|
logger.warn(warnMsg)
|
|
|
|
return {}
|
|
|
|
def getTables(self):
|
|
infoMsg = "fetching tables"
|
|
if conf.db:
|
|
infoMsg += " for database '%s'" % conf.db
|
|
logger.info(infoMsg)
|
|
|
|
rootQuery = queries[kb.dbms].tables
|
|
|
|
if not conf.db:
|
|
if not len(kb.data.cachedDbs):
|
|
dbs = self.getDbs()
|
|
else:
|
|
dbs = kb.data.cachedDbs
|
|
else:
|
|
if "," in conf.db:
|
|
dbs = conf.db.split(",")
|
|
else:
|
|
dbs = [conf.db]
|
|
|
|
if isTechniqueAvailable(PAYLOAD.TECHNIQUE.UNION) or conf.direct:
|
|
for db in dbs:
|
|
if conf.excludeSysDbs and db in self.excludeDbsList:
|
|
infoMsg = "skipping system database '%s'" % db
|
|
logger.info(infoMsg)
|
|
|
|
continue
|
|
|
|
query = rootQuery.inband.query % db
|
|
value = inject.getValue(query, blind=False, error=False)
|
|
|
|
if value:
|
|
kb.data.cachedTables[db] = arrayizeValue(value)
|
|
|
|
if not kb.data.cachedTables and not conf.direct:
|
|
for db in dbs:
|
|
if conf.excludeSysDbs and db in self.excludeDbsList:
|
|
infoMsg = "skipping system database '%s'" % db
|
|
logger.info(infoMsg)
|
|
|
|
continue
|
|
|
|
infoMsg = "fetching number of tables for "
|
|
infoMsg += "database '%s'" % db
|
|
logger.info(infoMsg)
|
|
|
|
query = rootQuery.blind.count % db
|
|
count = inject.getValue(query, inband=False, charsetType=2)
|
|
|
|
if not isNumPosStrValue(count):
|
|
warnMsg = "unable to retrieve the number of "
|
|
warnMsg += "tables for database '%s'" % db
|
|
logger.warn(warnMsg)
|
|
continue
|
|
|
|
tables = []
|
|
|
|
for index in range(int(count)):
|
|
query = rootQuery.blind.query % (db, index, db)
|
|
table = inject.getValue(query, inband=False)
|
|
tables.append(table)
|
|
kb.hintValue = table
|
|
|
|
if tables:
|
|
kb.data.cachedTables[db] = tables
|
|
else:
|
|
warnMsg = "unable to retrieve the tables "
|
|
warnMsg += "for database '%s'" % db
|
|
logger.warn(warnMsg)
|
|
|
|
if not kb.data.cachedTables:
|
|
errMsg = "unable to retrieve the tables for any database"
|
|
raise sqlmapNoneDataException(errMsg)
|
|
|
|
return kb.data.cachedTables
|
|
|
|
def searchTable(self):
|
|
rootQuery = queries[kb.dbms].search_table
|
|
foundTbls = {}
|
|
tblList = conf.tbl.split(",")
|
|
tblCond = rootQuery.inband.condition
|
|
dbCond = rootQuery.inband.condition2
|
|
|
|
tblConsider, tblCondParam = self.likeOrExact("table")
|
|
|
|
if not len(kb.data.cachedDbs):
|
|
enumDbs = self.getDbs()
|
|
else:
|
|
enumDbs = kb.data.cachedDbs
|
|
|
|
for db in enumDbs:
|
|
foundTbls[db] = []
|
|
|
|
for tbl in tblList:
|
|
infoMsg = "searching table"
|
|
if tblConsider == "1":
|
|
infoMsg += "s like"
|
|
infoMsg += " '%s'" % tbl
|
|
logger.info(infoMsg)
|
|
|
|
tblQuery = "%s%s" % (tblCond, tblCondParam)
|
|
tblQuery = tblQuery % tbl
|
|
|
|
for db in foundTbls.keys():
|
|
if conf.excludeSysDbs and db in self.excludeDbsList:
|
|
infoMsg = "skipping system database '%s'" % db
|
|
logger.info(infoMsg)
|
|
|
|
continue
|
|
|
|
if isTechniqueAvailable(PAYLOAD.TECHNIQUE.UNION) or conf.direct:
|
|
query = rootQuery.inband.query % db
|
|
query += tblQuery
|
|
values = inject.getValue(query, blind=False, error=False)
|
|
|
|
if values:
|
|
if isinstance(values, basestring):
|
|
values = [ values ]
|
|
|
|
for foundTbl in values:
|
|
foundTbls[db].append(foundTbl)
|
|
else:
|
|
infoMsg = "fetching number of table"
|
|
if tblConsider == "1":
|
|
infoMsg += "s like"
|
|
infoMsg += " '%s' in database '%s'" % (tbl, db)
|
|
logger.info(infoMsg)
|
|
|
|
query = rootQuery.blind.count2
|
|
query = query % db
|
|
query += " AND %s" % tblQuery
|
|
count = inject.getValue(query, inband=False, expected=EXPECTED.INT, charsetType=2)
|
|
|
|
if not isNumPosStrValue(count):
|
|
warnMsg = "no table"
|
|
if tblConsider == "1":
|
|
warnMsg += "s like"
|
|
warnMsg += " '%s' " % tbl
|
|
warnMsg += "in database '%s'" % db
|
|
logger.warn(warnMsg)
|
|
|
|
continue
|
|
|
|
indexRange = getRange(count)
|
|
|
|
for index in indexRange:
|
|
query = rootQuery.blind.query2
|
|
query = query % db
|
|
query += " AND %s" % tblQuery
|
|
query = agent.limitQuery(index, query, tblCond)
|
|
tbl = inject.getValue(query, inband=False)
|
|
kb.hintValue = tbl
|
|
foundTbls[db].append(tbl)
|
|
|
|
for db, tbls in foundTbls.items():
|
|
if len(tbls) == 0:
|
|
foundTbls.pop(db)
|
|
|
|
return foundTbls
|
|
|
|
def searchColumn(self):
|
|
rootQuery = queries[kb.dbms].search_column
|
|
foundCols = {}
|
|
dbs = {}
|
|
colList = conf.col.split(",")
|
|
colCond = rootQuery.inband.condition
|
|
colConsider, colCondParam = self.likeOrExact("column")
|
|
|
|
if not len(kb.data.cachedDbs):
|
|
enumDbs = self.getDbs()
|
|
else:
|
|
enumDbs = kb.data.cachedDbs
|
|
|
|
for db in enumDbs:
|
|
dbs[db] = {}
|
|
|
|
for column in colList:
|
|
infoMsg = "searching column"
|
|
if colConsider == "1":
|
|
infoMsg += "s like"
|
|
infoMsg += " '%s'" % column
|
|
logger.info(infoMsg)
|
|
|
|
foundCols[column] = {}
|
|
|
|
colQuery = "%s%s" % (colCond, colCondParam)
|
|
colQuery = colQuery % column
|
|
|
|
for db in dbs.keys():
|
|
if conf.excludeSysDbs and db in self.excludeDbsList:
|
|
infoMsg = "skipping system database '%s'" % db
|
|
logger.info(infoMsg)
|
|
|
|
continue
|
|
|
|
if isTechniqueAvailable(PAYLOAD.TECHNIQUE.UNION) or conf.direct:
|
|
query = rootQuery.inband.query % (db, db, db, db, db)
|
|
query += " AND %s" % colQuery.replace("[DB]", db)
|
|
values = inject.getValue(query, blind=False, error=False)
|
|
|
|
if values:
|
|
if isinstance(values, basestring):
|
|
values = [ values ]
|
|
|
|
for foundTbl in values:
|
|
if foundTbl not in dbs[db]:
|
|
dbs[db][foundTbl] = {}
|
|
|
|
if colConsider == "1":
|
|
conf.db = db
|
|
conf.tbl = foundTbl
|
|
conf.col = column
|
|
|
|
self.getColumns(onlyColNames=True)
|
|
|
|
dbs[db][foundTbl].update(kb.data.cachedColumns[db][foundTbl])
|
|
kb.data.cachedColumns = {}
|
|
else:
|
|
dbs[db][foundTbl][column] = None
|
|
|
|
if db in foundCols[column]:
|
|
foundCols[column][db].append(foundTbl)
|
|
else:
|
|
foundCols[column][db] = [ foundTbl ]
|
|
else:
|
|
foundCols[column][db] = []
|
|
|
|
infoMsg = "fetching number of tables containing column"
|
|
if colConsider == "1":
|
|
infoMsg += "s like"
|
|
infoMsg += " '%s' in database '%s'" % (column, db)
|
|
logger.info(infoMsg)
|
|
|
|
query = rootQuery.blind.count2
|
|
query = query % (db, db, db, db, db)
|
|
query += " AND %s" % colQuery.replace("[DB]", db)
|
|
count = inject.getValue(query, inband=False, expected=EXPECTED.INT, charsetType=2)
|
|
|
|
if not isNumPosStrValue(count):
|
|
warnMsg = "no tables contain column"
|
|
if colConsider == "1":
|
|
warnMsg += "s like"
|
|
warnMsg += " '%s' " % column
|
|
warnMsg += "in database '%s'" % db
|
|
logger.warn(warnMsg)
|
|
|
|
continue
|
|
|
|
indexRange = getRange(count)
|
|
|
|
for index in indexRange:
|
|
query = rootQuery.blind.query2
|
|
query = query % (db, db, db, db, db)
|
|
query += " AND %s" % colQuery.replace("[DB]", db)
|
|
query = agent.limitQuery(index, query, colCond.replace("[DB]", db))
|
|
tbl = inject.getValue(query, inband=False)
|
|
kb.hintValue = tbl
|
|
|
|
if tbl not in dbs[db]:
|
|
dbs[db][tbl] = {}
|
|
|
|
if colConsider == "1":
|
|
conf.db = db
|
|
conf.tbl = tbl
|
|
conf.col = column
|
|
|
|
self.getColumns(onlyColNames=True)
|
|
|
|
dbs[db][tbl].update(kb.data.cachedColumns[db][tbl])
|
|
kb.data.cachedColumns = {}
|
|
else:
|
|
dbs[db][tbl][column] = None
|
|
|
|
foundCols[column][db].append(tbl)
|
|
|
|
self.dumpFoundColumn(dbs, foundCols, colConsider)
|