platform: generic-arm64: Add dtb boot support

Add support for boot protocol with dtb in x0 instead of ram size. Find
gic (v3) registers in device tree instead of making an smc call.

TODO: Add gicv2 support

Bug: 297275002
Change-Id: Ie30b71cb6b7897438bf953b04ad1f41583ce6d97
diff --git a/platform/generic-arm64/platform.c b/platform/generic-arm64/platform.c
index 22dff25..f6343cc 100644
--- a/platform/generic-arm64/platform.c
+++ b/platform/generic-arm64/platform.c
@@ -26,12 +26,15 @@
 #include <dev/timer/arm_generic.h>
 #include <inttypes.h>
 #include <kernel/vm.h>
+#include <lib/device_tree/libfdt_helpers.h>
 #include <lk/init.h>
 #include <platform/gic.h>
 #include <string.h>
 #include <sys/types.h>
 
+#if ARM64_BOOT_PROTOCOL_X0_MEMSIZE || ARCH_ARM
 #include "smc.h"
+#endif
 
 #define ARM_GENERIC_TIMER_INT_CNTV 27
 #define ARM_GENERIC_TIMER_INT_CNTPS 29
@@ -57,6 +60,35 @@
 #endif
 #endif
 
+extern ulong lk_boot_args[4];
+
+#if ARM64_BOOT_PROTOCOL_X0_DTB
+static void generic_arm64_reserve_device_tree(paddr_t ram_base,
+                                              size_t ram_size) {
+    struct list_node list;
+    list_initialize(&list);
+
+    paddr_t fdt_paddr = lk_boot_args[0];
+    if (fdt_paddr < ram_base || fdt_paddr - ram_base >= ram_size) {
+        /* fdt address is outside ram_arena, no need to reserve it */
+        return;
+    }
+    const void* fdt = paddr_to_kvaddr(fdt_paddr);
+    if (fdt_check_header(fdt)) {
+        return;
+    }
+    size_t fdt_size = fdt_totalsize(fdt);
+    /* if fdt_paddr is not page aligned add offset in first page to size */
+    fdt_size += fdt_paddr & (PAGE_SIZE - 1);
+    uint fdt_page_count = DIV_ROUND_UP(fdt_size, PAGE_SIZE);
+    uint fdt_reserved_page_count =
+            pmm_alloc_range(fdt_paddr, fdt_page_count, &list);
+    if (fdt_page_count != fdt_reserved_page_count) {
+        panic("failed to reserve memory for device tree");
+    }
+}
+#endif
+
 /* initial memory mappings. parsed by start.S */
 struct mmu_initial_mapping mmu_initial_mappings[] = {
         /* Mark next entry as dynamic as it might be updated
@@ -95,8 +127,13 @@
         }
     }
     pmm_add_arena(&ram_arena);
+#if ARM64_BOOT_PROTOCOL_X0_DTB
+    generic_arm64_reserve_device_tree(ram_arena.base, ram_arena.size);
+#endif
 }
 
+#if ARM64_BOOT_PROTOCOL_X0_MEMSIZE || ARCH_ARM
+
 static paddr_t generic_arm64_get_reg_base(int reg) {
 #if ARCH_ARM64
     return generic_arm64_smc(SMC_FC64_GET_REG_BASE, reg, 0, 0);
@@ -105,11 +142,74 @@
 #endif
 }
 
+#endif
+
 static void platform_after_vm_init(uint level) {
+#if ARM64_BOOT_PROTOCOL_X0_MEMSIZE || ARCH_ARM
     paddr_t gicc = generic_arm64_get_reg_base(SMC_GET_GIC_BASE_GICC);
     paddr_t gicd = generic_arm64_get_reg_base(SMC_GET_GIC_BASE_GICD);
     paddr_t gicr = generic_arm64_get_reg_base(SMC_GET_GIC_BASE_GICR);
+#elif ARM64_BOOT_PROTOCOL_X0_DTB
+    int ret;
+    void* fdt;
+    size_t fdt_size;
+    paddr_t fdt_paddr = lk_boot_args[0];
+    ret = vmm_alloc_physical(
+            vmm_get_kernel_aspace(), "device_tree_probe", PAGE_SIZE, &fdt, 0,
+            fdt_paddr, 0, ARCH_MMU_FLAG_PERM_NO_EXECUTE | ARCH_MMU_FLAG_CACHED);
+    if (ret) {
+        dprintf(CRITICAL,
+                "failed to map device tree page at 0x%" PRIxPADDR ": %d\n",
+                fdt_paddr, ret);
+        return;
+    }
+    if (fdt_check_header(fdt)) {
+        dprintf(CRITICAL, "invalid device tree at 0x%" PRIxPADDR ": %d\n",
+                fdt_paddr, ret);
+        return;
+    }
+    fdt_size = fdt_totalsize(fdt);
+    if (fdt_size > PAGE_SIZE) {
+        dprintf(INFO, "remapping device tree with size 0x%zx\n", fdt_size);
+        vmm_free_region(vmm_get_kernel_aspace(), (vaddr_t)fdt);
+        ret = vmm_alloc_physical(
+                vmm_get_kernel_aspace(), "device_tree_full", fdt_size, &fdt, 0,
+                fdt_paddr, 0,
+                ARCH_MMU_FLAG_PERM_NO_EXECUTE | ARCH_MMU_FLAG_CACHED);
+        if (ret) {
+            dprintf(CRITICAL,
+                    "failed to map device tree at 0x%" PRIxPADDR
+                    " sz 0x%zx: %d\n",
+                    fdt_paddr, fdt_size, ret);
+            return;
+        }
+    }
 
+    int fdt_gic_offset = fdt_node_offset_by_compatible(fdt, 0, "arm,gic-v3");
+    paddr_t gicc = 0; /* gic-v3 does not need a memory mapped gicc */
+    paddr_t gicd, gicr;
+    size_t gicd_size, gicr_size;
+    if (fdt_helper_get_reg(fdt, fdt_gic_offset, 0, &gicd, &gicd_size)) {
+        dprintf(CRITICAL, "failed get gicd regs, offset %d\n", fdt_gic_offset);
+        return;
+    }
+    if (fdt_helper_get_reg(fdt, fdt_gic_offset, 1, &gicr, &gicr_size)) {
+        dprintf(CRITICAL, "failed get gicr regs, offset %d\n", fdt_gic_offset);
+        return;
+    }
+    if (gicd_size != GICD_SIZE) {
+        dprintf(CRITICAL, "unexpected gicd_size %zd != %d\n", gicd_size,
+                GICD_SIZE);
+        return;
+    }
+    if (gicr_size < GICR_SIZE) {
+        dprintf(CRITICAL, "unexpected gicr_size %zd < %d\n", gicr_size,
+                GICR_SIZE);
+        return;
+    }
+#else
+#error "Unknown ARM64_BOOT_PROTOCOL"
+#endif
     dprintf(INFO,
             "gicc 0x%" PRIxPADDR ", gicd 0x%" PRIxPADDR ", gicr 0x%" PRIxPADDR
             "\n",
diff --git a/platform/generic-arm64/rules.mk b/platform/generic-arm64/rules.mk
index 2eb6d36..d90b9de 100644
--- a/platform/generic-arm64/rules.mk
+++ b/platform/generic-arm64/rules.mk
@@ -78,9 +78,9 @@
 	$(LOCAL_DIR)/smc_service_access_policy.c \
 
 MODULE_DEPS += \
+	$(LKROOT)/lib/device_tree \
 	trusty/kernel/lib/dtb_embedded \
 	trusty/kernel/lib/dtb_service \
 	trusty/kernel/lib/ktipc \
-	external/dtc/libfdt \
 
 include make/module.mk