add basic mdadm support

Signed-off-by: Zen <[email protected]>

Author: desultory

src/ugrd/fs/mdraid.py

@@ -0,0 +1,5 @@
+__version__ = '0.1.0'
+
+
+def md_init(self):
+    return "mdadm --assemble --scan"

src/ugrd/fs/mdraid.toml

@@ -0,0 +1,4 @@
+binaries = ['mdadm']
+
+[imports.init_early]
+"ugrd.fs.mdadm" = [ "md_init" ]

src/ugrd/fs/mounts.py

@@ -1,5 +1,5 @@
 __author__ = 'desultory'
-__version__ = '4.16.4'
+__version__ = '4.16.5'
 
 from pathlib import Path
 from zenlib.util import contains, pretty_print
@@ -375,6 +375,9 @@ def _autodetect_dm(self, mountpoint, device=None) -> None:
         autodetect_luks(self, source_device, dev_name, dm_info)
     elif dm_info.get('type') == 'LVM2_member':
         autodetect_lvm(self, source_device, dev_name, dm_info)
+    elif dm_info.get('type') == 'linux_raid_member':
+        autodetect_raid(self, source_device, dev_name, dm_info)
+        raise NotImplementedError("RAID autodetection not implemented.")
     else:
         raise RuntimeError("Unknown device mapper device type: %s" % dm_info.get('type'))
 
@@ -388,6 +391,15 @@ def _autodetect_dm(self, mountpoint, device=None) -> None:
             self.logger.debug("Slave does not appear to be a DM device: %s" % slave)
 
 
+@contains('autodetect_root_raid', "Skipping RAID autodetection, autodetect_root_raid is disabled.", log_level=30)
+@contains('hostonly', "Skipping RAID autodetection, hostonly mode is disabled.", log_level=30)
+def autodetect_raid(self, mount_loc, dm_num, dm_info) -> None:
+    """ Autodetects MD RAID mounts and sets the raid config. """
+    if 'ugrd.fs.raid' not in self['modules']:
+        self.logger.info("Autodetected MDRAID mount, enabling the mdraid module.")
+        self['modules'] = 'ugrd.fs.mdraid'
+
+
 @contains('autodetect_root_dm', "Skipping device mapper autodetection, autodetect_root_dm is disabled.", log_level=30)
 @contains('hostonly', "Skipping device mapper autodetection, hostonly mode is disabled.", log_level=30)
 def autodetect_root_dm(self) -> None:

src/ugrd/fs/mounts.toml

@@ -13,6 +13,7 @@ autodetect_root = true
 autodetect_root_dm = true
 autodetect_root_luks = true
 autodetect_root_lvm = true
+autodetec_root_raid = true
 autodetect_init_mount = true
 late_fstab = "/etc/fstab.late"
 
@@ -53,6 +54,7 @@ autodetect_root = "bool"  # Add the autodetect_root property, if defined, the ro
 autodetect_root_dm = "bool"  # Whether or not to try to autodetect device-mapper partitions
 autodetect_root_luks = "bool"  # Whether or not to try to autodetect LUKS partitions
 autodetect_root_lvm = "bool"  # Whether or not to try to autodetect LVM partitions
+autodetec_root_raid = "bool"  # Whether or not to try to autodetect MDRAID partitions
 autodetect_init_mount = "bool"  # Adds a late_mount for the init target if it exists under a mount on the host
 resolve_root_dev = "bool"  # Whether or not to attempt resolve the root device source
 _mounts = "dict"  # The mounts information