Minor bug fix to correctly deal with unicode queries with -d

bdamele · bdamele · commit f26de8921631 · 2010-05-28T11:32:10.000Z
diff --git a/lib/core/convert.py b/lib/core/convert.py
@@ -100,3 +100,9 @@ def urlencode(string, safe=":/?%&=", convall=False):
         result = urllib.quote(string, safe)
 
     return result
+
+def utf8encode(string):
+    return string.encode("utf-8")
+
+def utf8decode(string):
+    return string.decode("utf-8")
diff --git a/lib/request/direct.py b/lib/request/direct.py
@@ -26,6 +26,7 @@
 from lib.core.common import dataToSessionFile
 from lib.core.convert import base64pickle
 from lib.core.convert import base64unpickle
+from lib.core.convert import utf8decode
 from lib.core.data import conf
 from lib.core.data import kb
 from lib.core.data import logger
@@ -35,6 +36,7 @@
 def direct(query, content=True):
     output = None
     select = False
+    query = utf8decode(query)
     query = agent.payloadDirect(query)
 
     if kb.dbms == "Oracle" and query.startswith("SELECT ") and " FROM " not in query:
