From 49915f3c3362b565d28e5134f040974971e4851f Mon Sep 17 00:00:00 2001
From: Miroslav Stampar <miroslav.stampar@gmail.com>
Date: Thu, 30 Sep 2010 19:49:14 +0000
Subject: [PATCH] minor update

---
 lib/takeover/udf.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/lib/takeover/udf.py b/lib/takeover/udf.py
index e848dd83e..08895ebee 100644
--- a/lib/takeover/udf.py
+++ b/lib/takeover/udf.py
@@ -34,6 +34,7 @@ from lib.core.data import queries
 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 @@ class UDF:
         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 = []