Add missing get_partition_id function

Change-Id: Ifad4249bd79c237e045bf40f3b0ec4ce136029f0

diff --git a/src/initrd-recovery/init b/src/initrd-recovery/init
index 21cc1bbe3a6649afd1d9e06e8f6acf22b2c11db1..0b6f354b71877307d2b69c990144964f9a2d42fd 100755 (executable)
--- a/src/initrd-recovery/init
+++ b/src/initrd-recovery/init
@@ -145,6 +145,45 @@ restore_partitions() {
     fi
 }
 
+#------------------------------------------------
+#       get partition id
+#------------------------------------------------
+get_partition_id() {
+    P_SLOT=$([[ $(</proc/cmdline) =~ partition_ab=([ab]) ]]; echo ${BASH_REMATCH[1]})
+    P_SUFFIX=""
+
+    if [ "${P_SLOT}" != "" ]; then
+        P_SUFFIX="_${P_SLOT}"
+        echo "Using A/B slot: ${P_SLOT}"
+    fi
+
+    PART_ROOTFS=`/sbin/blkid -t PARTLABEL=rootfs${P_SUFFIX} -o device`
+    if [ x$PART_ROOTFS = "x" ]
+    then
+        PART_ROOTFS=`/sbin/blkid -L rootfs`
+    fi
+
+    PART_SYSTEM_DATA=`/sbin/blkid -t PARTLABEL=system-data -o device`
+    if [ x$PART_SYSTEM_DATA = "x" ]
+    then
+        PART_SYSTEM_DATA=`/sbin/blkid -L system-data`
+    fi
+
+    PART_RAMDISK=`/sbin/blkid -t PARTLABEL=ramdisk${P_SUFFIX} -o device`
+    if [ x$PART_RAMDISK = "x" ]
+    then
+        PART_RAMDISK=`/sbin/blkid -L ramdisk`
+    fi
+
+    PART_HAL=`/sbin/blkid -t PARTLABEL=hal${P_SUFFIX} -o device`
+    if [ x$PART_HAL = "x" ]
+    then
+        PART_HAL=`/sbin/blkid -L hal`
+    fi
+
+    PART_USER=$("$BLKID" --match-token PARTLABEL=user -o device || "$BLKID" --match-token LABEL=user -o device)
+}
+
 #------------------------------------------------
 #       Main Routine Start
 #------------------------------------------------