From f444ad851ff518328b672aa518fe795fb1c37e20 Mon Sep 17 00:00:00 2001 From: ebolam Date: Tue, 14 Jun 2022 19:30:01 -0400 Subject: [PATCH] Potential catch for if somehow a user sends a delete model with a .. in it. --- aiserver.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/aiserver.py b/aiserver.py index be11305a..1b16b7ab 100644 --- a/aiserver.py +++ b/aiserver.py @@ -3097,7 +3097,7 @@ def get_message(msg): else: get_model_info(vars.model) elif(msg['cmd'] == 'delete_model'): - if "{}/models".format(os.getcwd()) in msg['data'] or "{}\\models".format(os.getcwd()) in msg['data']: + if "{}/models".format(os.getcwd()) in os.path.abspath(msg['data']) or "{}\\models".format(os.getcwd()) in os.path.abspath(msg['data']): if check_if_dir_is_model(msg['data']): print("It's a model, now we really will kill it") import shutil