disk: validate config

Author: codefiles

archinstall/__init__.py

@@ -274,7 +274,11 @@ def post_process_arguments(args: dict[str, Any]) -> None:
 		path = args['plugin']
 		load_plugin(path)
 
-	load_config()
+	try:
+		load_config()
+	except ValueError as err:
+		warn(str(err))
+		exit(1)
 
 
 define_arguments()

archinstall/lib/disk/device_model.py

@@ -13,6 +13,7 @@
 
 from ..exceptions import DiskError, SysCallError
 from ..general import SysCommand
+from ..hardware import SysInfo
 from ..output import debug
 from ..storage import storage
 
@@ -148,6 +149,43 @@ def parse_arg(cls, disk_config: _DiskLayoutConfigurationSerialization) -> DiskLa
 			device_modification.partitions = device_partitions
 			device_modifications.append(device_modification)
 
+		using_gpt = SysInfo.has_uefi()
+
+		for dev_mod in device_modifications:
+			partitions = sorted(dev_mod.partitions, key=lambda p: p.start)
+
+			for i, current_partition in enumerate(partitions[1:], start=1):
+				previous_partition = partitions[i - 1]
+				if (
+					current_partition.status == ModificationStatus.Create
+					and current_partition.start < previous_partition.end
+				):
+					raise ValueError('Partitions overlap')
+
+			partitions = [
+				part_mod for part_mod in dev_mod.partitions
+				if part_mod.status == ModificationStatus.Create
+			]
+
+			if not partitions:
+				continue
+
+			for part in partitions:
+				if (
+					part.start != part.start.align()
+					or part.length != part.length.align()
+				):
+					raise ValueError('Partition is misaligned')
+
+			total_size = dev_mod.device.device_info.total_size
+
+			if using_gpt:
+				if partitions[-1].end > total_size.gpt_end():
+					raise ValueError('Partition overlaps backup GPT header')
+			else:
+				if partitions[-1].end > total_size.align():
+					raise ValueError('Partition too large for device')
+
 		# Parse LVM configuration from settings
 		if (lvm_arg := disk_config.get('lvm_config', None)) is not None:
 			config.lvm_config = LvmConfiguration.parse_arg(lvm_arg, config)