myelin-accelerator 0.2.0

Rust/CUDA acceleration primitives for spiking neural networks and neuromorphic workloads
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85

# Copyright 2026 Raul Montoya Cardenas
# SPDX-License-Identifier: MIT OR Apache-2.0
"""Archive boundary checks for the v0.2.0 Cargo release preflight."""

import io
import sys
import tarfile
import tempfile
import unittest
from pathlib import Path

sys.path.insert(0, str(Path(__file__).resolve().parents[1] / "scripts"))
from prepare_crate import REQUIRED_PATHS, check_tracked_inventory, inspect_archive  # noqa: E402


class ArchiveBoundaryTests(unittest.TestCase):
    def archive(self, root: Path, paths: set[str], symlink: str | None = None) -> Path:
        output = root / "myelin-accelerator-0.2.0.crate"
        with tarfile.open(output, "w:gz") as archive:
            for path in sorted(paths):
                payload = b"fixture\n"
                info = tarfile.TarInfo(f"myelin-accelerator-0.2.0/{path}")
                if path == symlink:
                    info.type = tarfile.SYMTYPE
                    info.linkname = "../outside"
                    archive.addfile(info)
                    continue
                info.size = len(payload)
                archive.addfile(info, io.BytesIO(payload))
        return output

    def test_accepts_minimal_required_source_archive(self):
        with tempfile.TemporaryDirectory() as temp:
            files = inspect_archive(self.archive(Path(temp), set(REQUIRED_PATHS)))
        self.assertEqual(files, set(REQUIRED_PATHS))

    def test_rejects_missing_cuda_header(self):
        with tempfile.TemporaryDirectory() as temp:
            archive = self.archive(Path(temp), set(REQUIRED_PATHS) - {"cu/common.cuh"})
            with self.assertRaisesRegex(ValueError, "cu/common.cuh"):
                inspect_archive(archive)

    def test_rejects_credential_shaped_file(self):
        with tempfile.TemporaryDirectory() as temp:
            archive = self.archive(Path(temp), set(REQUIRED_PATHS) | {"src/.env"})
            with self.assertRaisesRegex(ValueError, "src/.env"):
                inspect_archive(archive)

    def test_rejects_path_outside_expected_crate_root(self):
        with tempfile.TemporaryDirectory() as temp:
            archive = self.archive(Path(temp), set(REQUIRED_PATHS) | {"../outside.txt"})
            with self.assertRaisesRegex(ValueError, "outside crate root"):
                inspect_archive(archive)

    def test_rejects_absolute_path_inside_archive_root(self):
        with tempfile.TemporaryDirectory() as temp:
            archive = self.archive(Path(temp), set(REQUIRED_PATHS) | {"/outside.txt"})
            with self.assertRaisesRegex(ValueError, "outside crate root"):
                inspect_archive(archive)

    def test_rejects_untracked_file_even_under_an_included_directory(self):
        with self.assertRaisesRegex(ValueError, "src/ignored.rs"):
            check_tracked_inventory(set(REQUIRED_PATHS) | {"src/ignored.rs"}, set(REQUIRED_PATHS))

    def test_allows_cargo_generated_metadata(self):
        check_tracked_inventory(
            set(REQUIRED_PATHS) | {".cargo_vcs_info.json", "Cargo.toml.orig"},
            set(REQUIRED_PATHS),
        )

    def test_rejects_common_private_key_name(self):
        with tempfile.TemporaryDirectory() as temp:
            archive = self.archive(Path(temp), set(REQUIRED_PATHS) | {"src/id_ed25519"})
            with self.assertRaisesRegex(ValueError, "id_ed25519"):
                inspect_archive(archive)

    def test_rejects_symlink_member(self):
        with tempfile.TemporaryDirectory() as temp:
            archive = self.archive(Path(temp), set(REQUIRED_PATHS) | {"src/linked.rs"}, "src/linked.rs")
            with self.assertRaisesRegex(ValueError, "non-file archive member"):
                inspect_archive(archive)


if __name__ == "__main__":
    unittest.main()