aboutsummaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
authorAUTOMATIC1111 <16777216c@gmail.com>2022-11-19 12:31:09 +0000
committerGitHub <noreply@github.com>2022-11-19 12:31:09 +0000
commitccd73fc186603b626d996b888d628ebb6b1a38d8 (patch)
treef08a0c529045f09c4a07bf447d8888fa2bdc904b
parent32718834f3fa4bee0eea35e94da39849a66e788f (diff)
parent3405acc6a4dcef2b73782a04924a9a12422e54f0 (diff)
downloadstable-diffusion-webui-gfx803-ccd73fc186603b626d996b888d628ebb6b1a38d8.tar.gz
stable-diffusion-webui-gfx803-ccd73fc186603b626d996b888d628ebb6b1a38d8.tar.bz2
stable-diffusion-webui-gfx803-ccd73fc186603b626d996b888d628ebb6b1a38d8.zip
Merge pull request #4717 from papuSpartan/security
Add --server-name to the list of arguments considered insecure
-rw-r--r--modules/shared.py2
-rw-r--r--webui.py5
2 files changed, 5 insertions, 2 deletions
diff --git a/modules/shared.py b/modules/shared.py
index 10c15c49..a4457305 100644
--- a/modules/shared.py
+++ b/modules/shared.py
@@ -107,7 +107,7 @@ restricted_opts = {
"outdir_save",
}
-cmd_opts.disable_extension_access = (cmd_opts.share or cmd_opts.listen) and not cmd_opts.enable_insecure_extension_access
+cmd_opts.disable_extension_access = (cmd_opts.share or cmd_opts.listen or cmd_opts.server_name) and not cmd_opts.enable_insecure_extension_access
devices.device, devices.device_interrogate, devices.device_gfpgan, devices.device_swinir, devices.device_esrgan, devices.device_scunet, devices.device_codeformer = \
(devices.cpu if any(y in cmd_opts.use_cpu for y in [x, 'all']) else devices.get_optimal_device() for x in ['sd', 'interrogate', 'gfpgan', 'swinir', 'esrgan', 'scunet', 'codeformer'])
diff --git a/webui.py b/webui.py
index 2cd3bae9..c5e5fe75 100644
--- a/webui.py
+++ b/webui.py
@@ -33,7 +33,10 @@ from modules.shared import cmd_opts
import modules.hypernetworks.hypernetwork
queue_lock = threading.Lock()
-server_name = "0.0.0.0" if cmd_opts.listen else cmd_opts.server_name
+if cmd_opts.server_name:
+ server_name = cmd_opts.server_name
+else:
+ server_name = "0.0.0.0" if cmd_opts.listen else None
def wrap_queued_call(func):
def f(*args, **kwargs):