From 336dfb0aa7abb373b0f7cb82193a6fb1d4c5e64e Mon Sep 17 00:00:00 2001
From: Eileen Yoon <eyn@gmx.com>
Date: Sat, 19 Aug 2023 03:54:54 +0900
Subject: [PATCH] kboot: Reserve ISP firmware

Signed-off-by: Eileen Yoon <eyn@gmx.com>
---
 src/kboot.c | 66 +++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 66 insertions(+)

diff --git a/src/kboot.c b/src/kboot.c
index e37499d07..20a3d5bad 100644
--- a/src/kboot.c
+++ b/src/kboot.c
@@ -1791,6 +1791,68 @@ static int dt_set_sio_fwdata(void)
     return 0;
 }
 
+struct isp_segment_ranges {
+    u64 phys;
+    u64 iova;
+    u64 remap;
+    u32 size;
+    u32 unk;
+} PACKED;
+
+static int dt_set_isp_fwdata(void)
+{
+    const char *path = "isp";
+    const char *adt_path = "/arm-io/isp";
+
+    int adt_node = adt_path_offset(adt, adt_path);
+    if (adt_node < 0) {
+        printf("ADT: '%s' node not found\n", adt_path);
+        return 0;
+    }
+
+    u32 segments_len;
+    struct isp_segment_ranges *segments;
+    segments =
+        (struct isp_segment_ranges *)adt_getprop(adt, adt_node, "segment-ranges", &segments_len);
+    if (!segments || !segments_len)
+        bail("ADT: invalid ISP segment-ranges\n");
+
+    int count = segments_len / sizeof(*segments);
+    for (int i = 0; i < count; i++)
+        segments[i].remap = segments[i].iova; // match sio segment-ranges
+
+    u64 ctrr_size;
+    switch (os_firmware.version) {
+        case V12_1: // haven't checked, probably right
+        case V12_2: // "
+        case V12_3:
+        case V12_3_1:
+        case V12_4:
+        case V12_5:
+            ctrr_size = 0x1800000;
+            break;
+        case V13_5:
+            ctrr_size = 0x1000000;
+            break;
+        default:
+            bail("FDT: couldn't get ISP CTRR size (%d)\n", os_firmware.version);
+    }
+
+    u64 heap_base = segments[count - 1].iova + segments[count - 1].size;
+    u64 heap_size = ctrr_size - heap_base;
+
+    int fdt_node = fdt_path_offset(dt, path);
+    if (fdt_node < 0)
+        bail("FDT: '%s' node not found\n", path);
+
+    if (fdt_appendprop_u64(dt, fdt_node, "apple,isp-heap-base", heap_base))
+        bail("FDT: couldn't set apple,isp-heap-base\n");
+    if (fdt_appendprop_u64(dt, fdt_node, "apple,isp-heap-size", heap_size))
+        bail("FDT: couldn't set apple,isp-heap-size\n");
+
+    return 0;
+}
+
 static int dt_disable_missing_devs(const char *adt_prefix, const char *dt_prefix, int max_devs)
 {
     int ret = -1;
@@ -2101,6 +2163,10 @@ int kboot_prepare_dt(void *fdt)
         return -1;
     if (dt_set_sio_fwdata())
         return -1;
+    if (dt_set_isp_fwdata())
+        return -1;
+    if (dt_reserve_asc_firmware("/arm-io/isp", "isp"))
+        return -1;
 #ifndef RELEASE
     if (dt_transfer_virtios())
         return 1;