Firepup650/bitbot-3.11-fork
2
0
Fork 0
Code Issues Pull requests Activity Actions

Change REST API to listen on IPv6

Browse source
This commit is contained in:
jesopo 2019-08-13 12:58:23 +01:00
parent fc5e3d3936
commit 2af428cf2f
1 changed files with 5 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
modules/rest_api.py
View file

@ -4,7 +4,7 @@
#--require-config tls-api-key
#--require-config tls-api-certificate
import http.server, json, ssl, threading, uuid, urllib.parse
import http.server, json, socket, ssl, threading, uuid, urllib.parse
from src import ModuleManager, utils
_bot = None
@ -110,6 +110,9 @@ class Handler(http.server.BaseHTTPRequestHandler):
def log_message(self, format, *args):
return
class BitBotIPv6HTTPd(http.server.HTTPServer):
address_family = socket.AF_INET6
@utils.export("botset",
utils.BoolSetting("rest-api", "Enable/disable REST API"))
@utils.export("botset",
@ -128,7 +131,7 @@ class Module(ModuleManager.BaseModule):
self.httpd = None
if self.bot.get_setting("rest-api", False):
port = int(self.bot.config.get("api-port", "5000"))
self.httpd = http.server.HTTPServer(("", port), Handler)
self.httpd = BitBotIPv6HTTPd(("", port), Handler)
self.httpd.socket = utils.security.ssl_wrap(self.httpd.socket,
cert=self.bot.config["tls-api-certificate"],