Fix for an Issue sqlmapproject#794

Author: stamparm

lib/core/settings.py

@@ -480,7 +480,7 @@
 HASHDB_END_TRANSACTION_RETRIES = 3
 
 # Unique milestone value used for forced deprecation of old HashDB values (e.g. when changing hash/pickle mechanism)
-HASHDB_MILESTONE_VALUE = "OZkQMtwHoP" # r9e02816 "".join(random.sample(string.ascii_letters, 10))
+HASHDB_MILESTONE_VALUE = "nXkbwIURlN" # rd74b803 "".join(random.sample(string.ascii_letters, 10))
 
 # Warn user of possible delay due to large page dump in full UNION query injections
 LARGE_OUTPUT_THRESHOLD = 1024 ** 2

lib/request/inject.py

@@ -367,11 +367,12 @@ def getValue(expression, blind=True, union=True, error=True, time=True, fromUser
  if not conf.forceDns:
  if union and isTechniqueAvailable(PAYLOAD.TECHNIQUE.UNION):
  kb.technique = PAYLOAD.TECHNIQUE.UNION
+ kb.forcePartialUnion = kb.injection.data[PAYLOAD.TECHNIQUE.UNION].vector[8]
  value = _goUnion(forgeCaseExpression if expected == EXPECTED.BOOL else query, unpack, dump)
  count += 1
  found = (value is not None) or (value is None and expectingNone) or count >= MAX_TECHNIQUES_PER_VALUE
 
- if not found and not expected and kb.injection.data[PAYLOAD.TECHNIQUE.UNION].where == PAYLOAD.WHERE.ORIGINAL:
+ if not found and not expected and kb.injection.data[PAYLOAD.TECHNIQUE.UNION].where == PAYLOAD.WHERE.ORIGINAL and not kb.forcePartialUnion:
  warnMsg = "something went wrong with full UNION "
  warnMsg += "technique (could be because of "
  warnMsg += "limitation on retrieved number of entries)"

lib/techniques/union/test.py

@@ -187,7 +187,7 @@ def _unionPosition(comment, place, parameter, prefix, suffix, count, where=PAYLO
  if content and phrase in content:
  validPayload = payload
  kb.unionDuplicates = len(re.findall(phrase, content, re.I)) > 1
- vector = (position, count, comment, prefix, suffix, kb.uChar, where, kb.unionDuplicates)
+ vector = (position, count, comment, prefix, suffix, kb.uChar, where, kb.unionDuplicates, False)
 
  if where == PAYLOAD.WHERE.ORIGINAL:
  # Prepare expression with delimiters
@@ -205,7 +205,7 @@ def _unionPosition(comment, place, parameter, prefix, suffix, count, where=PAYLO
  content = "%s%s".lower() % (page or "", listToStrValue(headers.headers if headers else None) or "")
 
  if not all(_ in content for _ in (phrase, phrase2)):
- vector = (position, count, comment, prefix, suffix, kb.uChar, where, kb.unionDuplicates)
+ vector = (position, count, comment, prefix, suffix, kb.uChar, where, kb.unionDuplicates, True)
  elif not kb.unionDuplicates:
  fromTable = " FROM (%s) AS %s" % (" UNION ".join("SELECT %d%s%s" % (_, FROM_DUMMY_TABLE.get(Backend.getIdentifiedDbms(), ""), " AS %s" % randomStr() if _ == 0 else "") for _ in xrange(LIMITED_ROWS_TEST_NUMBER)), randomStr())
 
@@ -221,7 +221,7 @@ def _unionPosition(comment, place, parameter, prefix, suffix, count, where=PAYLO
  if content.count(phrase) > 0 and content.count(phrase) < LIMITED_ROWS_TEST_NUMBER:
  warnMsg = "output with limited number of rows detected. Switching to partial mode"
  logger.warn(warnMsg)
- vector = (position, count, comment, prefix, suffix, kb.uChar, PAYLOAD.WHERE.NEGATIVE, kb.unionDuplicates)
+ vector = (position, count, comment, prefix, suffix, kb.uChar, PAYLOAD.WHERE.NEGATIVE, kb.unionDuplicates, False)
 
  unionErrorCase = kb.errorIsNone and wasLastResponseDBMSError()
 

lib/techniques/union/use.py

@@ -67,6 +67,7 @@ def _oneShotUnionUse(expression, unpack=True, limited=False):
  # Forge the union SQL injection request
  vector = kb.injection.data[PAYLOAD.TECHNIQUE.UNION].vector
  kb.unionDuplicates = vector[7]
+ kb.forcePartialUnion = vector[8]
  query = agent.forgeUnionQuery(injExpression, vector[0], vector[1], vector[2], vector[3], vector[4], vector[5], vector[6], None, limited)
  where = PAYLOAD.WHERE.NEGATIVE if conf.limitStart or conf.limitStop else vector[6]
  payload = agent.payload(newValue=query, where=where)
@@ -182,12 +183,12 @@ def unionUse(expression, unpack=True, dump=False):
  # NOTE: we assume that only queries that get data from a table can
  # return multiple entries
  if (kb.injection.data[PAYLOAD.TECHNIQUE.UNION].where == PAYLOAD.WHERE.NEGATIVE or \
+ kb.forcePartialUnion or \
  (dump and (conf.limitStart or conf.limitStop)) or "LIMIT " in expression.upper()) and \
  " FROM " in expression.upper() and ((Backend.getIdentifiedDbms() \
  not in FROM_DUMMY_TABLE) or (Backend.getIdentifiedDbms() in FROM_DUMMY_TABLE \
  and not expression.upper().endswith(FROM_DUMMY_TABLE[Backend.getIdentifiedDbms()]))) \
- and not re.search(SQL_SCALAR_REGEX, expression, re.I)\
- or kb.forcePartialUnion:
+ and not re.search(SQL_SCALAR_REGEX, expression, re.I):
  expression, limitCond, topLimit, startLimit, stopLimit = agent.limitCondition(expression, dump)
 
  if limitCond: