From 659f0d364616af63e712bb0b883adaf1bb7e4e7a Mon Sep 17 00:00:00 2001
From: Jacob Pease <jacobpease@protonmail.com>
Date: Tue, 23 Jul 2024 16:32:52 -0500
Subject: [PATCH] Added some minor error checking to gpt.c.

---
 fpga/zsbl/gpt.c | 17 +++++++++++++++++
 1 file changed, 17 insertions(+)

diff --git a/fpga/zsbl/gpt.c b/fpga/zsbl/gpt.c
index 65f56924b..90948935b 100644
--- a/fpga/zsbl/gpt.c
+++ b/fpga/zsbl/gpt.c
@@ -38,9 +38,26 @@ int gpt_load_partitions() {
   partition_entries_t *opensbi = (partition_entries_t *)(lba2_buf + 128);
   partition_entries_t *kernel = (partition_entries_t *)(lba2_buf + 256);
 
+  // Load device tree
   ret = disk_read((BYTE *)FDT_ADDRESS, fdt->first_lba, fdt->last_lba - fdt->first_lba + 1);
+  if (ret < 0) {
+    print_uart("Failed to load device tree!");
+    return -1;
+  }
+
+  // Load OpenSBI
   ret = disk_read((BYTE *)OPENSBI_ADDRESS, opensbi->first_lba, opensbi->last_lba - opensbi->first_lba + 1);
+  if (ret < 0) {
+    print_uart("Failed to load OpenSBI!");
+    return -1;
+  }
+
+  // Load Linux
   ret = disk_read((BYTE *)KERNEL_ADDRESS, kernel->first_lba,kernel->last_lba - kernel->first_lba + 1);
+  if (ret < 0) {
+    print_uart("Failed to load Linux!");
+    return -1;
+  }
 
   return 0;
 }