🚑 Fix partition sizes in the custom partitioning view

Author: mmstick

src/Views/DiskView.vala

@@ -46,18 +46,21 @@ public class Installer.DiskView : OptionsView {
         unowned Distinst.InstallOptions options = install_options.get_updated_options ();
 
         foreach (unowned Distinst.EraseOption disk in options.get_erase_options ()) {
+            uint64 sectors = disk.get_sectors ();
             uint64 sector_size = disk.get_sector_size ();
+
             string logo = Utils.string_from_utf8 (disk.get_linux_icon ());
             string label = Utils.string_from_utf8 (disk.get_model ());
+
             string details = "%s %.1f GiB".printf (
                 Utils.string_from_utf8 (disk.get_device_path ()),
-                (double) disk.get_sectors () / (double) Utils.normalize_sectors(SECTORS_AS_GIB, sector_size)
+                (double) sectors / (double) Utils.normalize_sectors (SECTORS_AS_GIB, sector_size)
             );
 
             // Ensure that the user cannot select a disk that is too large for BIOS installs.
             bool msdos_too_large =
                 Distinst.bootloader_detect () == Distinst.PartitionTable.MSDOS
-                && disk.get_sectors () > MSDOS_MAX_SECTORS;
+                && sectors > MSDOS_MAX_SECTORS;
 
             base.add_option(logo, label, details, (button) => {
                 if (disk.meets_requirements () && !msdos_too_large) {

src/Views/PartitioningView.vala

@@ -192,7 +192,7 @@ public class Installer.PartitioningView : AbstractInstallerView {
                 partitions.add (partition);
             }
 
-            var disk_bar = new DiskBar (model, path, size, (owned) partitions);
+            var disk_bar = new DiskBar (model, path, size, sector_size, (owned) partitions);
             label_sizer.add_widget (disk_bar.label);
             disk_list.pack_start (disk_bar);
         }
@@ -248,7 +248,7 @@ public class Installer.PartitioningView : AbstractInstallerView {
             }
         }
 
-        var disk_bar = new DiskBar (model, path, size, (owned) partitions);
+        var disk_bar = new DiskBar (model, path, size, sector_size, (owned) partitions);
         label_sizer.add_widget (disk_bar.label);
         disk_list.pack_start (disk_bar);
     }

src/Widgets/DiskBar.vala

@@ -22,6 +22,7 @@ public class Installer.DiskBar: Gtk.Grid {
     public string disk_name { get; construct; }
     public string disk_path { get; construct; }
     public uint64 size { get; construct; }
+    public uint64 sector_size { get; construct; }
     public Gee.ArrayList<PartitionBar> partitions { get; construct; }
 
     public Gtk.Box label;
@@ -34,13 +35,15 @@ public class Installer.DiskBar: Gtk.Grid {
         string model,
         string path,
         uint64 size,
+        uint64 sector_size,
         Gee.ArrayList<PartitionBar> partitions
     ) {
         Object (
             disk_name: model,
             disk_path: path,
             partitions: partitions,
-            size: size
+            size: size,
+            sector_size: sector_size
         );
     }
 
@@ -66,7 +69,7 @@ public class Installer.DiskBar: Gtk.Grid {
             used += partition.get_size ();
         }
 
-        return size - (used * 512);
+        return size - (used * sector_size);
     }
 
     private void generate_legend () {
@@ -79,7 +82,7 @@ public class Installer.DiskBar: Gtk.Grid {
         legend.add (legend_container);
 
         foreach (PartitionBar p in partitions) {
-            add_legend (p.path, p.get_size () * 512, Distinst.strfilesys (p.filesystem), p.vg, p.menu);
+            add_legend (p.path, p.get_size () * sector_size, Distinst.strfilesys (p.filesystem), p.vg, p.menu);
         }
 
         if (size / 100 < unused) {
@@ -165,7 +168,7 @@ public class Installer.DiskBar: Gtk.Grid {
 
     private void update_sector_lengths (Gee.ArrayList<PartitionBar> partitions, Gtk.Allocation alloc) {
         var alloc_width = alloc.width;
-        var disk_sectors = this.size / 512;
+        var disk_sectors = this.size / sector_size;
 
         int[] lengths = {};
         for (int x = 0; x < partitions.size; x++) {