fix: reject stale first-boot scripts in OTA release payloads

2026-10-05 18:44:46 -04:00
parent d9775ac144
commit ccf823590a
6 changed files with 193 additions and 0 deletions
@@ -0,0 +1,43 @@
+#!/usr/bin/env python3
+"""Exercise the release payload gate with real archives, including stale bytes."""
+import importlib.util
+import io
+import pathlib
+import tarfile
+import tempfile
+import unittest
+
+ROOT = pathlib.Path(__file__).resolve().parents[2]
+spec = importlib.util.spec_from_file_location('check_ota_runtime', ROOT / 'scripts/check-ota-runtime.py')
+validator = importlib.util.module_from_spec(spec)
+spec.loader.exec_module(validator)
+
+
+class RuntimePayloadTests(unittest.TestCase):
+    def test_actual_archives(self):
+        with tempfile.TemporaryDirectory() as directory:
+            base = pathlib.Path(directory)
+            source = base / 'first-boot-containers.sh'
+            source.write_bytes(b'#!/bin/sh\necho qualified\n')
+            for case, contents, valid in [
+                ('qualified', [source.read_bytes()], True),
+                ('stale', [b'#!/bin/sh\necho obsolete\n'], False),
+                ('missing', [], False),
+                ('duplicate', [source.read_bytes(), source.read_bytes()], False),
+            ]:
+                with self.subTest(case=case):
+                    archive = base / (case + '.tar.gz')
+                    with tarfile.open(archive, 'w:gz') as output:
+                        for content in contents:
+                            member = tarfile.TarInfo('./archipelago-runtime/scripts/first-boot-containers.sh')
+                            member.size = len(content)
+                            output.addfile(member, io.BytesIO(content))
+                    if valid:
+                        validator.check(archive, source)
+                    else:
+                        with self.assertRaises(ValueError):
+                            validator.check(archive, source)
+
+
+if __name__ == '__main__':
+    unittest.main()