minor update

stamparm · stamparm · commit af226796055b · 2010-12-08T13:09:27.000Z
diff --git a/lib/core/agent.py b/lib/core/agent.py
@@ -20,6 +20,7 @@
 from lib.core.data import queries
 from lib.core.datatype import advancedDict
 from lib.core.enums import DBMS
+from lib.core.enums import PAYLOAD
 from lib.core.enums import PLACE
 from lib.core.exception import sqlmapNoneDataException
 from lib.core.settings import PAYLOAD_DELIMITER
@@ -151,10 +152,10 @@ def prefixQuery(self, string):
         # Either if the technique is stacked queries (<stype>) or we are
         # replacing (<where>) the parameter original value with our
         # payload, do not put a space after the prefix
-        if kb.technique == 4:
+        if kb.technique == PAYLOAD.TECHNIQUE.STACKED:
             query = kb.injection.prefix
         elif kb.injection.clause == [2, 3] or kb.injection.clause == [ 3 ]:
-            if kb.technique != 3:
+            if kb.technique != PAYLOAD.TECHNIQUE.UNION:
                 query = kb.injection.prefix
         elif kb.technique and kb.technique in kb.injection.data:
             where = kb.injection.data[kb.technique].where
diff --git a/lib/techniques/inband/union/test.py b/lib/techniques/inband/union/test.py
@@ -14,6 +14,7 @@
 from lib.core.data import logger
 from lib.core.data import queries
 from lib.core.enums import DBMS
+from lib.core.enums import PAYLOAD
 from lib.core.session import setUnion
 from lib.core.unescaper import unescaper
 from lib.parse.html import htmlParser
@@ -118,7 +119,7 @@ def unionTest():
         return kb.unionTest
 
     oldTechnique = kb.technique
-    kb.technique = 3
+    kb.technique = PAYLOAD.TECHNIQUE.UNION
 
     if conf.uChar == "NULL":
         technique = "NULL bruteforcing"
