diff --git a/lib/utils/api.py b/lib/utils/api.py index 0be752d2d..1d186a60d 100644 --- a/lib/utils/api.py +++ b/lib/utils/api.py @@ -637,7 +637,7 @@ def download(taskid, target, filename): return jsonize({"success": False, "message": "File does not exist"}) -def server(host=RESTAPI_DEFAULT_ADDRESS, port=RESTAPI_DEFAULT_PORT): +def server(host=RESTAPI_DEFAULT_ADDRESS, port=RESTAPI_DEFAULT_PORT, server_name='wsgiref'): """ REST-JSON API server """ @@ -655,7 +655,14 @@ def server(host=RESTAPI_DEFAULT_ADDRESS, port=RESTAPI_DEFAULT_PORT): # Run RESTful API try: - run(host=host, port=port, quiet=True, debug=False) + if server_name == 'gevent': + from gevent import monkey + monkey.patch_all() + elif server_name == 'eventlet': + import eventlet + eventlet.monkey_patch() + logger.debug('use {0} adapter run bottle'.format(server_name)) + run(host=host, port=port, quiet=True, debug=False, server=server_name) except socket.error, ex: if "already in use" in getSafeExString(ex): logger.error("Address already in use ('%s:%s')" % (host, port)) diff --git a/sqlmapapi.py b/sqlmapapi.py index a4baa6ef3..a1effcc17 100755 --- a/sqlmapapi.py +++ b/sqlmapapi.py @@ -37,11 +37,12 @@ def main(): apiparser.add_option("-c", "--client", help="Act as a REST-JSON API client", default=RESTAPI_DEFAULT_PORT, action="store_true") apiparser.add_option("-H", "--host", help="Host of the REST-JSON API server", default=RESTAPI_DEFAULT_ADDRESS, action="store") apiparser.add_option("-p", "--port", help="Port of the the REST-JSON API server", default=RESTAPI_DEFAULT_PORT, type="int", action="store") + apiparser.add_option("", "--servername", help="bottle Server adapter to use default is wsgiref, see bottle document ", default='wsgiref', action="store") (args, _) = apiparser.parse_args() # Start the client or the server if args.server is True: - server(args.host, args.port) + server(args.host, args.port, server_name=args.servername) elif args.client is True: client(args.host, args.port) else: