From 3afb007442321acf9d1dbfd9362457aec504e434 Mon Sep 17 00:00:00 2001
From: 2Shirt <2xShirt@gmail.com>
Date: Sun, 17 Nov 2024 14:56:21 -0800
Subject: [PATCH] Skip MBR extended partition containers

---
 src/system/diskpart.rs | 16 ++++++++++------
 1 file changed, 10 insertions(+), 6 deletions(-)

diff --git a/src/system/diskpart.rs b/src/system/diskpart.rs
index 04612c9..4475f93 100644
--- a/src/system/diskpart.rs
+++ b/src/system/diskpart.rs
@@ -89,12 +89,16 @@ pub fn get_partition_details(
         contents = run_script(&script);
     };
     for (_, [number, size]) in RE_LIS.captures_iter(&contents).map(|c| c.extract()) {
-        let part = Partition {
-            id: number.parse().unwrap(),
-            size: string_to_bytes(size),
-            ..Default::default()
-        };
-        parts.push(part);
+        let part_num = number.parse().unwrap();
+        if part_num != 0 {
+            // part_num == 0 is reserved for extended partition "containers" so we can exclude them
+            let part = Partition {
+                id: part_num,
+                size: string_to_bytes(size),
+                ..Default::default()
+            };
+            parts.push(part);
+        }
     }
 
     // Detail parititon