From 71ff081fde9f35d5519a412146daffe22badc720 Mon Sep 17 00:00:00 2001
From: Miroslav Stampar <miroslav.stampar@gmail.com>
Date: Sun, 27 May 2012 09:11:19 +0000
Subject: [PATCH] minor update

---
 lib/core/option.py | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

diff --git a/lib/core/option.py b/lib/core/option.py
index 94f24da98..229842c31 100644
--- a/lib/core/option.py
+++ b/lib/core/option.py
@@ -1730,8 +1730,13 @@ def __setDNSServer():
     isAdmin = runningAsAdmin()
 
     if isAdmin:
-        conf.dnsServer = DNSServer()
-        conf.dnsServer.run()
+        try:
+            conf.dnsServer = DNSServer()
+            conf.dnsServer.run()
+        except socket.error, msg:
+            errMsg = "there was an error while setting up "
+            errMsg += "DNS server instance ('%s')" % msg
+            raise sqlmapGenericException, errMsg
     else:
         errMsg = "you need to run sqlmap as an administrator "
         errMsg += "if you want to perform a DNS data exfiltration attack "