From 25af33eac0709c3cc055658bf539d0729ceb680e Mon Sep 17 00:00:00 2001
From: HimbeerserverDE <himbeerserverde@gmail.com>
Date: Tue, 11 Apr 2023 22:40:27 +0200
Subject: rpi: install dtbs

---
 src/main.rs | 21 ++++++++++++++++++++-
 1 file changed, 20 insertions(+), 1 deletion(-)

(limited to 'src')

diff --git a/src/main.rs b/src/main.rs
index 8e0930f..3cffa3b 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -228,6 +228,26 @@ fn write_boot(
 
     // We don't need the firmware to boot on other supported architectures.
     if arch == "rpi" {
+        println!("Installing RPi dtbs...");
+
+        let dtbcopy = [
+            "bcm2710-rpi-3-b.dtb",
+            "bcm2710-rpi-3-b-plus.dtb",
+            "bcm2710-rpi-cm3.dtb",
+            "bcm2711-rpi-4-b.dtb",
+            "bcm2710-rpi-zero-2-w.dtb",
+        ];
+
+        for dtb in dtbcopy {
+            println!("Installing RPi dtb: {}", dtb);
+
+            let mut file = root_dir.create_file(dtb)?;
+
+            let mut resp =
+                reqwest::blocking::get(KERNEL_BASE.to_owned() + dtb)?.error_for_status()?;
+            resp.copy_to(&mut file)?;
+        }
+
         println!("Installing RPi firmware...");
 
         let fwcopy = [
@@ -257,7 +277,6 @@ fn write_boot(
 
             let mut resp =
                 reqwest::blocking::get(FIRMWARE_BASE.to_owned() + fw)?.error_for_status()?;
-
             resp.copy_to(&mut file)?;
         }
     }
-- 
cgit v1.2.3