Minor bug fix

Author: stamparm

lib/core/agent.py

@@ -275,7 +275,7 @@ def cleanupPayload(self, payload, origValue=None):
  inferenceQuery = inference.query
 
  payload = payload.replace("[INFERENCE]", inferenceQuery)
- else:
+ elif not kb.testMode:
  errMsg = "invalid usage of inference payload without "
  errMsg += "knowledge of underlying DBMS"
  raise SqlmapNoneDataException(errMsg)

lib/request/inject.py

@@ -286,13 +286,14 @@ def _goBooleanProxy(expression):
 
  initTechnique(kb.technique)
 
- query = agent.prefixQuery(kb.injection.data[kb.technique].vector)
- query = agent.suffixQuery(query)
- payload = agent.payload(newValue=query)
- output = _goDns(payload, expression)
+ if conf.dnsName:
+ query = agent.prefixQuery(kb.injection.data[kb.technique].vector)
+ query = agent.suffixQuery(query)
+ payload = agent.payload(newValue=query)
+ output = _goDns(payload, expression)
 
- if output is not None:
- return output
+  if output is not None:
+  return output
 
  vector = kb.injection.data[kb.technique].vector
  vector = vector.replace("[INFERENCE]", expression)