Procházet zdrojové kódy

update pickle safe filenames

AUTOMATIC1111 před 1 rokem
rodič
revize
07cf95c76e
1 změnil soubory, kde provedl 2 přidání a 2 odebrání
  1. 2 2
      modules/safe.py

+ 2 - 2
modules/safe.py

@@ -64,8 +64,8 @@ class RestrictedUnpickler(pickle.Unpickler):
         raise Exception(f"global '{module}/{name}' is forbidden")
         raise Exception(f"global '{module}/{name}' is forbidden")
 
 
 
 
-# Regular expression that accepts 'dirname/version', 'dirname/data.pkl', and 'dirname/data/<number>'
-allowed_zip_names_re = re.compile(r"^([^/]+)/((data/\d+)|version|(data\.pkl))$")
+# Regular expression that accepts 'dirname/version', 'dirname/byteorder', 'dirname/data.pkl', '.data/serialization_id', and 'dirname/data/<number>'
+allowed_zip_names_re = re.compile(r"^([^/]+)/((data/\d+)|version|byteorder|.data/serialization_id|(data\.pkl))$")
 data_pkl_re = re.compile(r"^([^/]+)/data\.pkl$")
 data_pkl_re = re.compile(r"^([^/]+)/data\.pkl$")
 
 
 def check_zip_filenames(filename, names):
 def check_zip_filenames(filename, names):