update /boot/grub/device.map if needed

1 files changed, 8 insertions, 0 deletions

diff --git a/rescue/restore-image.sh b/rescue/restore-image.sh
index c5fa5726e..70a7cab89 100755
--- a/rescue/restore-image.sh
+++ b/rescue/restore-image.sh
@@ -11,6 +11,7 @@ fi
 
 export PATH="/sbin:/bin:/usr/sbin:/usr/bin"
 
+mnt_dir="/tmp/mnt"
 restore_media="/tmp/media"
 images_dir="$restore_media/images"
 images="$images_dir/list"
@@ -184,6 +185,13 @@ function expand_fs()
 		    parted $disk -- mkpartfs primary linux-swap ${main_part_sectors}s -1s yes
 		    mkswap -L swap $swap_part
 		fi
+                mkdir -p $mnt_dir
+                mount $main_part $mnt_dir
+                grub_dir="$mnt_dir/boot/grub"
+                if [ -d "$grub_dir" ]; then
+                    echo "(hd0) $disk" > "$grub_dir/device.map"
+                fi
+                umount $mnt_dir
 	fi
 }