feat: add self-test for relocation engine

1 files changed, 46 insertions, 0 deletions

diff --git a/na/test/test_reloc.cc b/na/test/test_reloc.cc
new file mode 100644
index 0000000..54e7144
--- /dev/null
+++ b/na/test/test_reloc.cc
@@ -0,0 +1,46 @@
+/* SPDX-License-Identifier: GPL-2.0-or-later */
+/* Copyright (C) 2026 Richard Qin */
+
+/*
+ * Boot Self-Test: relocation engine
+ */
+
+#include <lume/relocate.h>
+#include <lume/selftest.h>
+
+struct RelocBlob {
+    RelocTableHeader hdr;
+    RelocEntry entries[2];
+};
+
+void selftest_reloc()
+{
+    st_begin("reloc: relative + abs64");
+
+    alignas(8) uint8 buf[64] = {};
+    uint64 base = reinterpret_cast<uint64>(buf);
+    uint64 link_base = base - 0x1000;
+
+    RelocBlob blob = {};
+    blob.hdr.magic = kRelocTableMagic;
+    blob.hdr.version = kRelocTableVersion;
+    blob.hdr.arch = RELOC_ARCH_RISCV;
+    blob.hdr.entry_size = sizeof(RelocEntry);
+    blob.hdr.entry_count = 2;
+
+    blob.entries[0].offset = 8;
+    blob.entries[0].addend = 0x200;
+    blob.entries[0].type = RELOC_RISCV_RELATIVE;
+
+    blob.entries[1].offset = 16;
+    blob.entries[1].addend = link_base + 0x1234;
+    blob.entries[1].type = RELOC_RISCV_64;
+
+    RelocStatus st = relocate_apply(base, sizeof(buf), link_base,
+                                    &blob, sizeof(blob));
+    ST_ASSERT_EQ(st, RELOC_OK);
+    ST_ASSERT_EQ(*reinterpret_cast<uint64*>(base + 8), base + 0x200);
+    ST_ASSERT_EQ(*reinterpret_cast<uint64*>(base + 16), base + 0x1234);
+
+    st_pass();
+}