Adding support for Amazon Redshift (pgsql fork)

stamparm · stamparm · commit 8d343fc2a6c2 · 2020-01-31T11:39:16.000+01:00
diff --git a/lib/core/enums.py b/lib/core/enums.py
@@ -81,6 +81,7 @@ class FORK(object):
     PERCONA = "Percona"
     COCKROACHDB = "CockroachDB"
     TIDB = "TiDB"
+    REDSHIFT = "Amazon Redshift"
 
 class CUSTOM_LOGGING(object):
     PAYLOAD = 9
diff --git a/lib/core/settings.py b/lib/core/settings.py
@@ -18,7 +18,7 @@
 from thirdparty.six import unichr as _unichr
 
 # sqlmap version (<major>.<minor>.<month>.<monthly commit>)
-VERSION = "1.4.1.62"
+VERSION = "1.4.1.63"
 TYPE = "dev" if VERSION.count('.') > 2 and VERSION.split('.')[-1] != '0' else "stable"
 TYPE_COLORS = {"dev": 33, "stable": 90, "pip": 34}
 VERSION_STRING = "sqlmap/%s#%s" % ('.'.join(VERSION.split('.')[:-1]) if VERSION.count('.') > 2 and VERSION.split('.')[-1] == '0' else VERSION, TYPE)
diff --git a/plugins/dbms/postgresql/fingerprint.py b/plugins/dbms/postgresql/fingerprint.py
@@ -29,7 +29,13 @@ def getFingerprint(self):
         fork = hashDBRetrieve(HASHDB_KEYS.DBMS_FORK)
 
         if fork is None:
-            fork = inject.checkBooleanExpression("VERSION() LIKE '%CockroachDB%'") and FORK.COCKROACHDB or ""
+            if inject.checkBooleanExpression("VERSION() LIKE '%CockroachDB%'"):
+                fork = FORK.COCKROACHDB
+            elif inject.checkBooleanExpression("VERSION() LIKE '%Redshift%'"):  # Reference: https://dataedo.com/kb/query/amazon-redshift/check-server-version
+                fork = FORK.REDSHIFT
+            else:
+                fork = ""
+
             hashDBWrite(HASHDB_KEYS.DBMS_FORK, fork)
 
         value = ""
