1 files changed, 34 insertions, 15 deletions
diff --git a/Scripts/mount_ubifs_rcc.sh b/Scripts/mount_ubifs_rcc.sh
index 57d41cc..1cd22ed 100644
--- a/Scripts/mount_ubifs_rcc.sh
+++ b/Scripts/mount_ubifs_rcc.sh
@@ -1,5 +1,5 @@
 #!/bin/sh
-# Copyright 2020, Ludwig V. <https://github.com/ludwig-v>
+# Copyright 2020-2021, Ludwig V. <https://github.com/ludwig-v>
 
 # This program is free software: you can redistribute it and/or modify
 # it under the terms of the GNU General Public License as published by
@@ -17,20 +17,39 @@
 
 UBIFS_IMAGE=$1
 
-mkdir -p /mnt/ubifs
-if grep -qs '/dev/ubi' /proc/mounts; then
-	umount /mnt/ubifs
-	ubidetach -p /dev/mtd0
-	rmmod nandsim
+if ! [ -x "$(command -v ubiformat)" ]; then
+	echo 'Error: mtd-utils is not installed.' >&2
+	exit 1
 fi
 
-modprobe ubi
-modprobe nandsim first_id_byte=0xec second_id_byte=0xd7 third_id_byte=0x00 fourth_id_byte=0x36
-flash_erase /dev/mtd0 0 0
-ubiformat /dev/mtd0 -s 4096 -O 4096
-ubiattach -m 0 -d 0 -O 4096
-ubimkvol /dev/ubi0 -N NAC_UBIFS -s 1024MiB
-ubiupdatevol /dev/ubi0_0 $UBIFS_IMAGE
-mount /dev/ubi0_0 /mnt/ubifs
+if modinfo ubi | grep -wq "filename:"; then
+	if modinfo nandsim | grep -wq "filename:"; then
+		mkdir -p /mnt/ubifs
+		if lsmod | grep -wq "^nandsim"; then
+			umount /mnt/ubifs
+			ubidetach -p /dev/mtd0
+			rmmod nandsim
+		fi
 
-echo "${UBIFS_IMAGE} mounted on /mnt/ubifs"
-\ No newline at end of file
+		modprobe ubi
+		modprobe nandsim first_id_byte=0xec second_id_byte=0xd7 third_id_byte=0x00 fourth_id_byte=0x36
+		flash_erase /dev/mtd0 0 0
+		ubiformat /dev/mtd0 -s 4096 -O 4096
+		ubiattach -m 0 -d 0 -O 4096
+		ubimkvol /dev/ubi0 -N RCCC_UBIFS -s 1024MiB
+		ubiupdatevol /dev/ubi0_0 $UBIFS_IMAGE
+		mount /dev/ubi0_0 /mnt/ubifs
+
+		echo "${UBIFS_IMAGE} mounted on /mnt/ubifs"
+		
+		exit 0
+	else
+		echo 'Error: nandsim kernel module is not installed.' >&2
+		
+		exit 1
+	fi
+else
+	echo 'Error: ubi kernel module is not installed.' >&2
+	
+	exit 1
+fi
+\ No newline at end of file