diff --git a/extra/vulnserver/vulnserver.py b/extra/vulnserver/vulnserver.py
index be07f4e7b..520652435 100644
--- a/extra/vulnserver/vulnserver.py
+++ b/extra/vulnserver/vulnserver.py
@@ -129,7 +129,6 @@ class ReqHandler(BaseHTTPRequestHandler):
self.url, self.params = path, params
if self.url == '/':
-
if not any(_ in self.params for _ in ("id", "query")):
self.send_response(OK)
self.send_header("Content-type", "text/html; charset=%s" % UNICODE_ENCODING)
@@ -158,18 +157,22 @@ class ReqHandler(BaseHTTPRequestHandler):
output += "SQL results:
\n"
- if results:
- output += "
\n"
-
- for row in results:
- output += ""
- for value in row:
- output += "| %s | " % value
- output += "
\n"
-
- output += "
\n"
+ if self.params.get("code", ""):
+ if not results:
+ code = INTERNAL_SERVER_ERROR
else:
- output += "no results found"
+ if results:
+ output += "\n"
+
+ for row in results:
+ output += ""
+ for value in row:
+ output += "| %s | " % value
+ output += "
\n"
+
+ output += "
\n"
+ else:
+ output += "no results found"
output += "