From 02db66e2f6f65273cdd23db42a07d79b3c2b71a1 Mon Sep 17 00:00:00 2001 From: root Date: Mon, 4 May 2015 09:57:53 +0500 Subject: [PATCH] Rescue connection refused backtrace --- lib/sqlmap/sqlmap_session.rb | 21 ++++++++++++++------- 1 file changed, 14 insertions(+), 7 deletions(-) diff --git a/lib/sqlmap/sqlmap_session.rb b/lib/sqlmap/sqlmap_session.rb index 8d74bde672..380344a578 100644 --- a/lib/sqlmap/sqlmap_session.rb +++ b/lib/sqlmap/sqlmap_session.rb @@ -13,9 +13,13 @@ module Sqlmap args['headers'] = headers if headers args['vars_get'] = params if params - res = c.request_cgi(args) - res = c.send_recv(res) - return res + begin + res = c.request_cgi(args) + res = c.send_recv(res) + return res + rescue Rex::ConnectionRefused + return + end end def post(uri, headers = nil, data = nil, originator_args = nil) @@ -26,12 +30,15 @@ module Sqlmap } args.merge!(originator_args) if originator_args - args['headers'] = headers if headers args['data'] = data if data - res = c.request_cgi(args) - res = c.send_recv(res) - return res + begin + res = c.request_cgi(args) + res = c.send_recv(res) + return res + rescue Rex::ConnectionRefused + return + end end end end