target/linux/x86/base-files/lib/upgrade/platform.sh

   1 RAMFS_COPY_BIN='grub-bios-setup'
   2
   3 platform_check_image() {
   4         local diskdev partdev diff
   5         [ "$#" -gt 1 ] && return 1
   6
   7         case "$(get_magic_word "$1")" in
   8                 eb48|eb63) ;;
   9                 *)
  10                         echo "Invalid image type"
  11                         return 1
  12                 ;;
  13         esac
  14
  15         export_bootdevice && export_partdevice diskdev 0 || {
  16                 echo "Unable to determine upgrade device"
  17                 return 1
  18         }
  19
  20         get_partitions "/dev/$diskdev" bootdisk
  21
  22         #extract the boot sector from the image
  23         get_image "$@" | dd of=/tmp/image.bs count=1 bs=512b 2>/dev/null
  24
  25         get_partitions /tmp/image.bs image
  26
  27         #compare tables
  28         diff="$(grep -F -x -v -f /tmp/partmap.bootdisk /tmp/partmap.image)"
  29
  30         rm -f /tmp/image.bs /tmp/partmap.bootdisk /tmp/partmap.image
  31
  32         if [ -n "$diff" ]; then
  33                 echo "Partition layout has changed. Full image will be written."
  34                 ask_bool 0 "Abort" && exit 1
  35                 return 0
  36         fi
  37 }
  38
  39 platform_copy_config() {
  40         local partdev
  41
  42         if export_partdevice partdev 1; then
  43                 mount -t ext4 -o rw,noatime "/dev/$partdev" /mnt
  44                 cp -af "$UPGRADE_BACKUP" "/mnt/$BACKUP_FILE"
  45                 umount /mnt
  46         fi
  47 }
  48
  49 platform_do_bootloader_upgrade() {
  50         local bootpart
  51         local diskdev="$1"
  52
  53         if export_partdevice bootpart 1; then
  54                 mkdir -p /tmp/boot
  55                 mount -o rw,noatime "/dev/$bootpart" /tmp/boot
  56                 echo "(hd0) /dev/$diskdev" > /tmp/device.map
  57
  58                 echo "Upgrading bootloader on /dev/$diskdev..."
  59                 grub-bios-setup \
  60                         -m "/tmp/device.map" \
  61                         -d "/tmp/boot/boot/grub" \
  62                         -r "hd0,msdos1" \
  63                         "/dev/$diskdev" \
  64                 && touch /boot/grub/upgraded
  65
  66                 umount /tmp/boot
  67         fi
  68 }
  69
  70 platform_do_upgrade() {
  71         local diskdev partdev diff
  72
  73         export_bootdevice && export_partdevice diskdev 0 || {
  74                 echo "Unable to determine upgrade device"
  75                 return 1
  76         }
  77
  78         sync
  79
  80         if [ "$UPGRADE_OPT_SAVE_PARTITIONS" = "1" ]; then
  81                 get_partitions "/dev/$diskdev" bootdisk
  82
  83                 #extract the boot sector from the image
  84                 get_image "$@" | dd of=/tmp/image.bs count=1 bs=512b
  85
  86                 get_partitions /tmp/image.bs image
  87
  88                 #compare tables
  89                 diff="$(grep -F -x -v -f /tmp/partmap.bootdisk /tmp/partmap.image)"
  90         else
  91                 diff=1
  92         fi
  93
  94         if [ -n "$diff" ]; then
  95                 get_image "$@" | dd of="/dev/$diskdev" bs=4096 conv=fsync
  96
  97                 # Separate removal and addtion is necessary; otherwise, partition 1
  98                 # will be missing if it overlaps with the old partition 2
  99                 partx -d - "/dev/$diskdev"
 100                 partx -a - "/dev/$diskdev"
 101
 102                 return 0
 103         fi
 104
 105         #iterate over each partition from the image and write it to the boot disk
 106         while read part start size; do
 107                 if export_partdevice partdev $part; then
 108                         echo "Writing image to /dev/$partdev..."
 109                         get_image "$@" | dd of="/dev/$partdev" ibs="512" obs=1M skip="$start" count="$size" conv=fsync
 110                 else
 111                         echo "Unable to find partition $part device, skipped."
 112                 fi
 113         done < /tmp/partmap.image
 114
 115         #copy partition uuid
 116         echo "Writing new UUID to /dev/$diskdev..."
 117         get_image "$@" | dd of="/dev/$diskdev" bs=1 skip=440 count=4 seek=440 conv=fsync
 118
 119         platform_do_bootloader_upgrade "$diskdev"
 120 }