File size: 566 Bytes
deb302b | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 | import os, glob, joblib, sys
WORK = sys.argv[1]
# remove any stale markers
for m in glob.glob(os.path.join(WORK, "marker_*")):
os.remove(m)
for c in ["none","zlib","gzip","bz2","lzma","xz","lz4"]:
p = os.path.join(WORK, f"evil_{c}.joblib")
try:
joblib.load(p)
except Exception as ex:
print(f"{c:5s} load-exc {type(ex).__name__}: {ex}")
marker = os.path.join(WORK, f"marker_{c}")
if os.path.exists(marker):
print(f"{c:5s} EXECUTED=True {open(marker).read().strip()}")
else:
print(f"{c:5s} EXECUTED=False")
|