minor update

stamparm · stamparm · commit 49915f3c3362 · 2010-09-30T19:49:14.000Z
diff --git a/lib/takeover/udf.py b/lib/takeover/udf.py
@@ -34,6 +34,7 @@
 from lib.core.exception import sqlmapFilePathException
 from lib.core.exception import sqlmapMissingMandatoryOptionException
 from lib.core.exception import sqlmapUnsupportedFeatureException
+from lib.core.exception import sqlmapUserQuitException
 from lib.core.unescaper import unescaper
 from lib.request import inject
 from lib.techniques.outband.stacked import stackedTest
@@ -308,9 +309,12 @@ def udfInjectCustom(self):
         msg   += "functions now? [Y/n/q] "
         choice = readInput(msg, default="Y")
 
-        if choice[0] not in ( "y", "Y" ):
+        if choice[0] in ( "n", "N" ):
             self.cleanup(udfDict=self.udfs)
             return
+        elif choice[0] in ( "q", "Q" ):
+            self.cleanup(udfDict=self.udfs)
+            raise sqlmapUserQuitException
 
         while True:
             udfList = []
