whitelist CLientStatus

This commit is contained in:
Fabian Dill 2021-03-19 04:17:26 +01:00
parent 218fb0298f
commit 38a0a3f15e
1 changed files with 1 additions and 1 deletions

View File

@ -359,7 +359,7 @@ class RestrictedUnpickler(pickle.Unpickler):
def find_class(self, module, name):
if module == "builtins" and name in safe_builtins:
return getattr(builtins, name)
if module == "NetUtils" and name in {"NetworkItem"}:
if module == "NetUtils" and name in {"NetworkItem", "ClientStatus"}:
import NetUtils
return getattr(NetUtils, name)
# Forbid everything else.