Merge ~andersson123/ubuntu-archive-tools:torrent-testing-script into ubuntu-archive-tools:main

[Tim Andersson]

[Tim Andersson (andersson123)]

Example output:

```
~/canonical/code/ubuntu-archive-tools$ ./check-torrents -r noble
INFO:root:checking hash: 0b33930759bb30a47663952af9c95f1fdd9c820b For image: ubuntu-24.04-live-server-arm64+largemem.iso.torrent
INFO:root:Found code: 0b33930759bb30a47663952af9c95f1fdd9c820b
INFO:root:checking hash: a958368d56536a1b568607171dca92b155e3994b For image: ubuntu-24.04-live-server-arm64.iso.torrent
INFO:root:Found code: a958368d56536a1b568607171dca92b155e3994b
INFO:root:checking hash: ead8c4756e522e8441818ff5ec397d6168f165f6 For image: ubuntu-24.04-live-server-ppc64el.iso.torrent
INFO:root:Found code: ead8c4756e522e8441818ff5ec397d6168f165f6
INFO:root:checking hash: 0f93fc8e83ec3b01865e4e1466d0a3f2889d5345 For image: ubuntu-24.04-live-server-s390x.iso.torrent
INFO:root:Found code: 0f93fc8e83ec3b01865e4e1466d0a3f2889d5345
INFO:root:{
  "ubuntu-24.04-live-server-arm64+largemem.iso.torrent": true,
  "ubuntu-24.04-live-server-arm64.iso.torrent": true,
  "ubuntu-24.04-live-server-ppc64el.iso.torrent": true,
  "ubuntu-24.04-live-server-s390x.iso.torrent": true
}
```

Unmerged commits

7fdda85... by Tim Andersson on 2024-04-25

feat: add torrent testing script

This script checks all the available .torrent images for a given
release.

It downloads the .torrent images, extracts the hash, and then check
that the image name and hash matches up with:
https://torrent.ubuntu.com/tracker_index

diff --git a/check-torrents b/check-torrents
new file mode 100755
index 0000000..6b11e57
--- /dev/null
+++ b/check-torrents
@@ -0,0 +1,91 @@
+#!/usr/bin/python3
+
+import argparse
+import requests
+import subprocess
+import re
+import json
+import logging
+import os
+
+
+def parse_args():
+    parser = argparse.ArgumentParser()
+    parser.add_argument(
+        "--release",
+        "-r",
+        required=True,
+        help="Release to test torrents for (codename)",
+    )
+    return parser.parse_args()
+
+
+def get_url(release):
+    return f"rsync://cdimage.ubuntu.com/cdimage/releases/{release}/release/"
+
+
+def rsync_list_torrent_images(walk_me_url):
+    format = ".iso.torrent"
+    rsync = subprocess.check_output(
+        [
+            "rsync",
+            "--dry-run",
+            "--out-format='%f'",
+            f"{walk_me_url}*{format}",
+        ],
+        text=True,
+    )
+    t_files = []
+    for t_file in rsync.splitlines():
+        t_files.append(t_file.split(" ")[-1])
+    return t_files
+
+
+def get_torrent_hash(base_url, torrent_image_name):
+    _ = subprocess.check_call(
+        [
+            "rsync",
+            f"{base_url}{torrent_image_name}",
+            f"/tmp/{torrent_image_name}"
+        ]
+    )
+    show_transmission = subprocess.check_output(
+        [
+            "transmission-show",
+            f"/tmp/{torrent_image_name}",
+        ],
+        text=True,
+    )
+    return re.search(r'Hash: (.*)\n', show_transmission).group(1)
+
+
+def check_hash_is_correct(hash, image_name):
+    logging.info("checking hash: %s For image: %s" % (hash, image_name))
+    r = requests.get("https://torrent.ubuntu.com/tracker_index").content.decode("utf-8")
+    for line in r.splitlines():
+        if image_name.replace(".torrent", "") in line:
+            code = re.search(r'<code>(.*)</code>', line).group(1)
+            logging.info("Found code: %s" % code)
+            return code == hash
+    return False
+
+
+def check_torrent(base_url, torrent_image_name):
+    hash = get_torrent_hash(base_url, torrent_image_name)
+    return check_hash_is_correct(hash, torrent_image_name)
+
+
+def main():
+    logging.basicConfig(level=logging.INFO)
+    args = parse_args()
+    url_to_walk = get_url(args.release)
+    images_to_check = rsync_list_torrent_images(url_to_walk)
+    torrent_results = {}
+    for torrent_img in images_to_check:
+        torrent_results[torrent_img] = check_torrent(url_to_walk, torrent_img)
+        os.remove(f"/tmp/{torrent_img}")
+    logging.info(json.dumps(torrent_results, indent=2))
+
+
+if __name__ == "__main__":
+    main()