mirror of
https://github.com/sqlmapproject/sqlmap.git
synced 2026-02-04 04:26:34 +00:00
Merge pull request #1678 from dozysun/servername-option
add --adapter option to support various of bottle server adapter
This commit is contained in:
@@ -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, adapter='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 adapter == 'gevent':
|
||||
from gevent import monkey
|
||||
monkey.patch_all()
|
||||
elif adapter == 'eventlet':
|
||||
import eventlet
|
||||
eventlet.monkey_patch()
|
||||
logger.debug('use {0} adapter run bottle'.format(adapter))
|
||||
run(host=host, port=port, quiet=True, debug=False, server=adapter)
|
||||
except socket.error, ex:
|
||||
if "already in use" in getSafeExString(ex):
|
||||
logger.error("Address already in use ('%s:%s')" % (host, port))
|
||||
|
||||
Reference in New Issue
Block a user