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                         v "Invalid image type"
  11                         return 1
  12                 ;;
  13         esac
  14
  15         export_bootdevice && export_partdevice diskdev 0 || {
  16                 v "Unable to determine upgrade device"
  17                 return 1
  18         }
  19
  20         get_partitions "/dev/$diskdev" bootdisk
  21
  22         v "Extract boot sector from the image"
  23         get_image_dd "$1" of=/tmp/image.bs count=63 bs=512b
  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                 v "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 parttype=ext4
  41
  42         if export_partdevice partdev 1; then
  43                 part_magic_fat "/dev/$partdev" && parttype=vfat
  44                 mount -t $parttype -o rw,noatime "/dev/$partdev" /mnt
  45                 cp -af "$UPGRADE_BACKUP" "/mnt/$BACKUP_FILE"
  46                 umount /mnt
  47         fi
  48 }
  49
  50 platform_do_bootloader_upgrade() {
  51         local bootpart parttable=msdos
  52         local diskdev="$1"
  53
  54         if export_partdevice bootpart 1; then
  55                 mkdir -p /tmp/boot
  56                 mount -o rw,noatime "/dev/$bootpart" /tmp/boot
  57                 echo "(hd0) /dev/$diskdev" > /tmp/device.map
  58                 part_magic_efi "/dev/$diskdev" && parttable=gpt
  59
  60                 v "Upgrading bootloader on /dev/$diskdev..."
  61                 grub-bios-setup \
  62                         -m "/tmp/device.map" \
  63                         -d "/tmp/boot/boot/grub" \
  64                         -r "hd0,${parttable}1" \
  65                         "/dev/$diskdev" \
  66                 && touch /tmp/boot/boot/grub/upgraded
  67
  68         case "$(board_name)" in
  69         cisco-mx100-hw)
  70             # If the MX100 is booted UEFI AND the SATA HDD exists, we need to change
  71             # grub's root= to hd1 for it to boot correctly, otherwise we can keep it hd0.
  72             if [ -d /sys/firmware/efi ] && [ "$(ls -a /dev/sd[a-z] | wc -w)" -gt 1 ] ; then
  73                 sed -i "s|hd0,${parttable}1|hd1,${parttable}1|g" /tmp/boot/boot/grub/grub.cfg
  74             fi
  75             ;;
  76         esac
  77
  78                 umount /tmp/boot
  79         fi
  80 }
  81
  82 platform_do_upgrade() {
  83         local diskdev partdev diff
  84
  85         export_bootdevice && export_partdevice diskdev 0 || {
  86                 v "Unable to determine upgrade device"
  87                 return 1
  88         }
  89
  90         sync
  91
  92         if [ "$UPGRADE_OPT_SAVE_PARTITIONS" = "1" ]; then
  93                 get_partitions "/dev/$diskdev" bootdisk
  94
  95                 v "Extract boot sector from the image"
  96                 get_image_dd "$1" of=/tmp/image.bs count=63 bs=512b
  97
  98                 get_partitions /tmp/image.bs image
  99
 100                 #compare tables
 101                 diff="$(grep -F -x -v -f /tmp/partmap.bootdisk /tmp/partmap.image)"
 102         else
 103                 diff=1
 104         fi
 105
 106         if [ -n "$diff" ]; then
 107                 get_image_dd "$1" of="/dev/$diskdev" bs=4096 conv=fsync
 108
 109                 # Separate removal and addtion is necessary; otherwise, partition 1
 110                 # will be missing if it overlaps with the old partition 2
 111                 partx -d - "/dev/$diskdev"
 112                 partx -a - "/dev/$diskdev"
 113
 114                 return 0
 115         fi
 116
 117         #iterate over each partition from the image and write it to the boot disk
 118         while read part start size; do
 119                 if export_partdevice partdev $part; then
 120                         v "Writing image to /dev/$partdev..."
 121                         get_image_dd "$1" of="/dev/$partdev" ibs=512 obs=1M skip="$start" count="$size" conv=fsync
 122                 else
 123                         v "Unable to find partition $part device, skipped."
 124                 fi
 125         done < /tmp/partmap.image
 126
 127         v "Writing new UUID to /dev/$diskdev..."
 128         get_image_dd "$1" of="/dev/$diskdev" bs=1 skip=440 count=4 seek=440 conv=fsync
 129
 130         platform_do_bootloader_upgrade "$diskdev"
 131         local parttype=ext4
 132         part_magic_efi "/dev/$diskdev" || return 0
 133
 134         if export_partdevice partdev 1; then
 135                 part_magic_fat "/dev/$partdev" && parttype=vfat
 136                 mount -t $parttype -o rw,noatime "/dev/$partdev" /mnt
 137                 set -- $(dd if="/dev/$diskdev" bs=1 skip=1168 count=16 2>/dev/null | hexdump -v -e '8/1 "%02x "" "2/1 "%02x""-"6/1 "%02x"')
 138                 sed -i "s/\(PARTUUID=\)[a-f0-9-]\+/\1$4$3$2$1-$6$5-$8$7-$9/ig" /mnt/boot/grub/grub.cfg
 139                 umount /mnt
 140         fi
 141 }