Support auto expand for ext4 fs

Author: kvinwang

dstack-util/src/system_setup.rs

@@ -561,16 +561,29 @@ impl<'a> Stage0<'a> {
                     }
                 }
                 FsType::Ext4 => {
-                    if cmd!(mountpoint -q $mount_point).is_err() {
-                        cmd!(mount $fs_dev $mount_point)
-                            .context("Failed to mount ext4 filesystem")?;
-                    }
+                    Self::mount_e2fs(&fs_dev, mount_point)
+                        .context("Failed to mount ext4 filesystem")?;
                 }
             }
         }
         Ok(())
     }
 
+    fn mount_e2fs(dev: &impl AsRef<Path>, mount_point: &impl AsRef<Path>) -> Result<()> {
+        let dev = dev.as_ref();
+        let mount_point = mount_point.as_ref();
+        cmd! {
+            info "Checking filesystem";
+            e2fsck -f -p $dev;
+            info "Trying to resize filesystem if needed";
+            resize2fs $dev;
+            info "Mounting filesystem";
+            mount $dev $mount_point;
+        }
+        .context("Failed to check filesystem")?;
+        Ok(())
+    }
+
     fn luks_setup(&self, disk_crypt_key: &str, name: &str) -> Result<()> {
         let root_hd = &self.args.device;
         let sector_offset = PAYLOAD_OFFSET / 512;