From 8bd324eab29446a2a4a5d11932ae2414f7c8cdf3 Mon Sep 17 00:00:00 2001 From: Joost VandeVondele Date: Sun, 7 Jul 2024 09:21:58 +0200 Subject: [PATCH] Verify sha before cache write --- worker/games.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/worker/games.py b/worker/games.py index 64463a3c5..3bcdc6a02 100644 --- a/worker/games.py +++ b/worker/games.py @@ -328,7 +328,9 @@ def download_net(remote, testing_dir, net, global_cache): url = remote + "/api/nn/" + net print("Downloading {}".format(net)) content = requests_get(url, allow_redirects=True, timeout=HTTP_TIMEOUT).content - cache_write(global_cache, net, content) + hash = hashlib.sha256(content).hexdigest() + if hash[:12] == net[3:15]: + cache_write(global_cache, net, content) else: print("Using {} from global cache".format(net))