Verify sha before cache write

vondele · Jul 7, 2024 · 8bd324e · 8bd324e
1 parent 12981ff
commit 8bd324e
Showing 1 changed file with 3 additions and 1 deletion.
diff --git 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))