mirror of
https://github.com/sqlmapproject/sqlmap.git
synced 2024-11-23 01:56:36 +03:00
203 lines
6.5 KiB
Python
203 lines
6.5 KiB
Python
#!/usr/bin/env python
|
|
|
|
"""
|
|
$Id: postgresql.py 286 2008-07-25 23:09:48Z inquisb $
|
|
|
|
This file is part of the sqlmap project, http://sqlmap.sourceforge.net.
|
|
|
|
Copyright (c) 2006-2008 Bernardo Damele A. G. <bernardo.damele@gmail.com>
|
|
and Daniele Bellucci <daniele.bellucci@gmail.com>
|
|
|
|
sqlmap is free software; you can redistribute it and/or modify it under
|
|
the terms of the GNU General Public License as published by the Free
|
|
Software Foundation version 2 of the License.
|
|
|
|
sqlmap is distributed in the hope that it will be useful, but WITHOUT ANY
|
|
WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
|
|
FOR A PARTICULAR PURPOSE. See the GNU General Public License for more
|
|
details.
|
|
|
|
You should have received a copy of the GNU General Public License along
|
|
with sqlmap; if not, write to the Free Software Foundation, Inc., 51
|
|
Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
|
|
"""
|
|
|
|
|
|
|
|
import re
|
|
|
|
from lib.core.common import formatFingerprint
|
|
from lib.core.common import getHtmlErrorFp
|
|
from lib.core.common import randomInt
|
|
from lib.core.data import conf
|
|
from lib.core.data import kb
|
|
from lib.core.data import logger
|
|
from lib.core.exception import sqlmapSyntaxException
|
|
from lib.core.session import setDbms
|
|
from lib.core.settings import PGSQL_ALIASES
|
|
from lib.core.unescaper import unescaper
|
|
from lib.request import inject
|
|
#from lib.utils.fuzzer import passiveFuzzing
|
|
|
|
from plugins.generic.enumeration import Enumeration
|
|
from plugins.generic.filesystem import Filesystem
|
|
from plugins.generic.fingerprint import Fingerprint
|
|
from plugins.generic.takeover import Takeover
|
|
|
|
|
|
class PostgreSQLMap(Fingerprint, Enumeration, Filesystem, Takeover):
|
|
"""
|
|
This class defines PostgreSQL methods
|
|
"""
|
|
|
|
def __init__(self):
|
|
Enumeration.__init__(self, "PostgreSQL")
|
|
|
|
unescaper.setUnescape(PostgreSQLMap.unescape)
|
|
|
|
|
|
@staticmethod
|
|
def unescape(expression):
|
|
while True:
|
|
index = expression.find("'")
|
|
if index == -1:
|
|
break
|
|
|
|
firstIndex = index + 1
|
|
index = expression[firstIndex:].find("'")
|
|
|
|
if index == -1:
|
|
raise sqlmapSyntaxException, "Unenclosed ' in '%s'" % expression
|
|
|
|
lastIndex = firstIndex + index
|
|
old = "'%s'" % expression[firstIndex:lastIndex]
|
|
unescaped = "("
|
|
|
|
for i in range(firstIndex, lastIndex):
|
|
unescaped += "CHR(%d)" % (ord(expression[i]))
|
|
if i < lastIndex - 1:
|
|
unescaped += "||"
|
|
|
|
unescaped += ")"
|
|
expression = expression.replace(old, unescaped)
|
|
|
|
return expression
|
|
|
|
|
|
@staticmethod
|
|
def escape(expression):
|
|
while True:
|
|
index = expression.find("CHR(")
|
|
if index == -1:
|
|
break
|
|
|
|
firstIndex = index
|
|
index = expression[firstIndex:].find("))")
|
|
|
|
if index == -1:
|
|
raise sqlmapSyntaxException, "Unenclosed ) in '%s'" % expression
|
|
|
|
lastIndex = firstIndex + index + 1
|
|
old = expression[firstIndex:lastIndex]
|
|
oldUpper = old.upper()
|
|
oldUpper = oldUpper.replace("CHR(", "").replace(")", "")
|
|
oldUpper = oldUpper.split("||")
|
|
|
|
escaped = "'%s'" % "".join([chr(int(char)) for char in oldUpper])
|
|
expression = expression.replace(old, escaped)
|
|
|
|
return expression
|
|
|
|
|
|
def getFingerprint(self):
|
|
if not conf.extensiveFp:
|
|
return "PostgreSQL"
|
|
|
|
actVer = formatFingerprint()
|
|
|
|
blank = " " * 16
|
|
value = "active fingerprint: %s" % actVer
|
|
|
|
if self.banner:
|
|
banVer = re.search("^PostgreSQL ([\d\.]+)", self.banner)
|
|
banVer = banVer.groups()[0]
|
|
banVer = formatFingerprint([banVer])
|
|
|
|
value += "\n%sbanner parsing fingerprint: %s" % (blank, banVer)
|
|
|
|
#passiveFuzzing()
|
|
htmlParsed = getHtmlErrorFp()
|
|
|
|
if htmlParsed:
|
|
value += "\n%shtml error message fingerprint: %s" % (blank, htmlParsed)
|
|
|
|
return value
|
|
|
|
|
|
def checkDbms(self):
|
|
if conf.dbms in PGSQL_ALIASES:
|
|
setDbms("PostgreSQL")
|
|
|
|
if not conf.extensiveFp:
|
|
return True
|
|
|
|
logMsg = "testing PostgreSQL"
|
|
logger.info(logMsg)
|
|
|
|
randInt = str(randomInt(1))
|
|
query = "COALESCE(%s, NULL)" % randInt
|
|
|
|
if inject.getValue(query) == randInt:
|
|
logMsg = "confirming PostgreSQL"
|
|
logger.info(logMsg)
|
|
|
|
query = "LENGTH('%s')" % randInt
|
|
|
|
if not inject.getValue(query) == "1":
|
|
warnMsg = "the back-end DMBS is not PostgreSQL"
|
|
logger.warn(warnMsg)
|
|
|
|
return False
|
|
|
|
setDbms("PostgreSQL")
|
|
|
|
if not conf.extensiveFp:
|
|
return True
|
|
|
|
if inject.getValue("SUBSTR(TRANSACTION_TIMESTAMP(), 1, 1)") == "2":
|
|
kb.dbmsVersion = [">= 8.2.0"]
|
|
elif inject.getValue("GREATEST(5, 9, 1)") == "9":
|
|
kb.dbmsVersion = [">= 8.1.0", "< 8.2.0"]
|
|
elif inject.getValue("WIDTH_BUCKET(5.35, 0.024, 10.06, 5)") == "3":
|
|
kb.dbmsVersion = [">= 8.0.0", "< 8.1.0"]
|
|
elif inject.getValue("SUBSTR(MD5('sqlmap'), 1, 1)"):
|
|
kb.dbmsVersion = [">= 7.4.0", "< 8.0.0"]
|
|
elif inject.getValue("SUBSTR(CURRENT_SCHEMA(), 1, 1)") == "p":
|
|
kb.dbmsVersion = [">= 7.3.0", "< 7.4.0"]
|
|
elif inject.getValue("BIT_LENGTH(1)") == "8":
|
|
kb.dbmsVersion = [">= 7.2.0", "< 7.3.0"]
|
|
elif inject.getValue("SUBSTR(QUOTE_LITERAL('a'), 2, 1)") == "a":
|
|
kb.dbmsVersion = [">= 7.1.0", "< 7.2.0"]
|
|
elif inject.getValue("POW(2, 3)") == "8":
|
|
kb.dbmsVersion = [">= 7.0.0", "< 7.1.0"]
|
|
elif inject.getValue("MAX('a')") == "a":
|
|
kb.dbmsVersion = [">= 6.5.0", "< 6.5.3"]
|
|
elif re.search("([\d\.]+)", inject.getValue("SUBSTR(VERSION(), 12, 5)")):
|
|
kb.dbmsVersion = [">= 6.4.0", "< 6.5.0"]
|
|
elif inject.getValue("SUBSTR(CURRENT_DATE, 1, 1)") == "2":
|
|
kb.dbmsVersion = [">= 6.3.0", "< 6.4.0"]
|
|
elif inject.getValue("SUBSTRING('sqlmap', 1, 1)") == "s":
|
|
kb.dbmsVersion = [">= 6.2.0", "< 6.3.0"]
|
|
else:
|
|
kb.dbmsVersion = ["< 6.2.0"]
|
|
|
|
if conf.getBanner:
|
|
self.banner = inject.getValue("VERSION()")
|
|
|
|
return True
|
|
else:
|
|
warnMsg = "the back-end DMBS is not PostgreSQL"
|
|
logger.warn(warnMsg)
|
|
|
|
return False
|