mirror of
https://github.com/sqlmapproject/sqlmap.git
synced 2026-02-16 04:10:33 +03:00
160 lines
5.5 KiB
Python
160 lines
5.5 KiB
Python
#!/usr/bin/env python
|
|
|
|
"""
|
|
Copyright (c) 2006-2012 sqlmap developers (http://sqlmap.org/)
|
|
See the file 'doc/COPYING' for copying permission
|
|
"""
|
|
|
|
from lib.core.common import Backend, getLimitRange, isAdminFromPrivileges, isInferenceAvailable, isNoneValue,\
|
|
isNumPosStrValue, isTechniqueAvailable
|
|
from lib.core.data import conf, kb, logger, queries
|
|
from lib.core.enums import CHARSET_TYPE, EXPECTED, PAYLOAD
|
|
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)
|
|
|
|
def getRoles(self, query2=False):
|
|
infoMsg = "fetching database users roles"
|
|
|
|
rootQuery = queries[Backend.getIdentifiedDbms()].roles
|
|
|
|
if conf.user == "CU":
|
|
infoMsg += " for current user"
|
|
conf.user = self.getCurrentUser()
|
|
|
|
logger.info(infoMsg)
|
|
|
|
# Set containing the list of DBMS administrators
|
|
areAdmins = set()
|
|
|
|
if any(isTechniqueAvailable(_) for _ in (
|
|
PAYLOAD.TECHNIQUE.UNION, PAYLOAD.TECHNIQUE.ERROR, PAYLOAD.TECHNIQUE.QUERY)) or conf.direct:
|
|
if query2:
|
|
query = rootQuery.inband.query2
|
|
condition = rootQuery.inband.condition2
|
|
else:
|
|
query = rootQuery.inband.query
|
|
condition = rootQuery.inband.condition
|
|
|
|
if conf.user:
|
|
users = conf.user.split(",")
|
|
query += " WHERE "
|
|
query += " OR ".join("%s = '%s'" % (condition, user) for user in sorted(users))
|
|
|
|
values = inject.getValue(query, blind=False, time=False)
|
|
|
|
if not values and not query2:
|
|
infoMsg = "trying with table USER_ROLE_PRIVS"
|
|
logger.info(infoMsg)
|
|
|
|
return self.getRoles(query2=True)
|
|
|
|
if not isNoneValue(values):
|
|
for value in values:
|
|
user = None
|
|
roles = set()
|
|
|
|
for count in xrange(0, len(value)):
|
|
# The first column is always the username
|
|
if count == 0:
|
|
user = value[count]
|
|
|
|
# The other columns are the roles
|
|
else:
|
|
role = value[count]
|
|
|
|
# In Oracle we get the list of roles as string
|
|
roles.add(role)
|
|
|
|
if isAdminFromPrivileges(roles):
|
|
areAdmins.add(user)
|
|
|
|
if kb.data.cachedUsersRoles.has_key(user):
|
|
kb.data.cachedUsersRoles[user].extend(roles)
|
|
else:
|
|
kb.data.cachedUsersRoles[user] = list(roles)
|
|
|
|
if not kb.data.cachedUsersRoles and isInferenceAvailable() and not conf.direct:
|
|
conditionChar = "="
|
|
|
|
if conf.user:
|
|
users = conf.user.split(",")
|
|
else:
|
|
if not len(kb.data.cachedUsers):
|
|
users = self.getUsers()
|
|
else:
|
|
users = kb.data.cachedUsers
|
|
|
|
retrievedUsers = set()
|
|
|
|
for user in users:
|
|
unescapedUser = None
|
|
|
|
if user in retrievedUsers:
|
|
continue
|
|
|
|
infoMsg = "fetching number of roles "
|
|
infoMsg += "for user '%s'" % user
|
|
logger.info(infoMsg)
|
|
|
|
if unescapedUser:
|
|
queryUser = unescapedUser
|
|
else:
|
|
queryUser = user
|
|
|
|
if query2:
|
|
query = rootQuery.blind.count2 % queryUser
|
|
else:
|
|
query = rootQuery.blind.count % queryUser
|
|
count = inject.getValue(query, union=False, error=False, expected=EXPECTED.INT,
|
|
charsetType=CHARSET_TYPE.DIGITS)
|
|
|
|
if not isNumPosStrValue(count):
|
|
if count != 0 and not query2:
|
|
infoMsg = "trying with table USER_SYS_PRIVS"
|
|
logger.info(infoMsg)
|
|
|
|
return self.getPrivileges(query2=True)
|
|
|
|
warnMsg = "unable to retrieve the number of "
|
|
warnMsg += "roles for user '%s'" % user
|
|
logger.warn(warnMsg)
|
|
continue
|
|
|
|
infoMsg = "fetching roles for user '%s'" % user
|
|
logger.info(infoMsg)
|
|
|
|
roles = set()
|
|
|
|
indexRange = getLimitRange(count, plusOne=True)
|
|
|
|
for index in indexRange:
|
|
if query2:
|
|
query = rootQuery.blind.query2 % (queryUser, index)
|
|
else:
|
|
query = rootQuery.blind.query % (queryUser, index)
|
|
role = inject.getValue(query, union=False, error=False)
|
|
|
|
# In Oracle we get the list of roles as string
|
|
roles.add(role)
|
|
|
|
if roles:
|
|
kb.data.cachedUsersRoles[user] = list(roles)
|
|
else:
|
|
warnMsg = "unable to retrieve the roles "
|
|
warnMsg += "for user '%s'" % user
|
|
logger.warn(warnMsg)
|
|
|
|
retrievedUsers.add(user)
|
|
|
|
if not kb.data.cachedUsersRoles:
|
|
errMsg = "unable to retrieve the roles "
|
|
errMsg += "for the database users"
|
|
raise sqlmapNoneDataException, errMsg
|
|
|
|
return kb.data.cachedUsersRoles, areAdmins
|