flasher/rootfs/etc/updater.sh

   1 #/bin/sh
   2 #
   3 # on-target updater script for Pandora
   4 # Copyright (c) 2010, Gražvydas Ignotas
   5
   6 set -e
   7
   8 . /etc/updater_funcs.sh
   9
  10 status_uboot="missing"
  11 status_uboot_env="already empty"
  12 status_uimage="missing"
  13 status_rootfs="missing"
  14
  15 cleanup()
  16 {
  17         cd /tmp
  18         if [ -n "$finalize_mpoint" ]
  19         then
  20                 umount $finalize_mpoint 2> /dev/null || true
  21         fi
  22         if [ -n "$rootfs_mpoint" ]
  23         then
  24                 umount $rootfs_mpoint 2> /dev/null || true
  25         fi
  26 }
  27
  28 mkdir /mnt/ubifs 2> /dev/null || true
  29
  30 if [ ! -e /dev/mmcblk0p1 ]
  31 then
  32         # hmh what's up with the SD card?
  33         log "Can't access SD card, please reinsert and press enter"
  34         read a < /dev/tty3
  35         test -e /dev/mmcblk0p1
  36 fi
  37
  38 mdevs="/dev/mmcblk0p1 /dev/mmcblk0p2"
  39 bootfiles="bootf.tgz bootfiles.tar.gz uimage.tar.gz"
  40
  41 cd /tmp
  42
  43 for md in $mdevs
  44 do
  45         mpoint="/mnt/`basename $md`"
  46         keep_mpoint="no"
  47         mkdir $mpoint 2> /dev/null || true
  48         if ! mount $md $mpoint
  49         then
  50                 continue;
  51         fi
  52
  53         if [ -x $mpoint/initialize.sh ]
  54         then
  55                 log ""
  56                 log "running init script.."
  57                 cd $mpoint
  58                 if ! ./initialize.sh
  59                 then
  60                         log "initialize.sh failed."
  61                         cleanup
  62                         false
  63                 fi
  64                 log "done."
  65                 cd /tmp
  66         fi
  67
  68         for bf in $bootfiles
  69         do
  70                 if [ -e $mpoint/$bf ]
  71                 then
  72                         tar xzvf $mpoint/$bf
  73                 fi
  74         done
  75
  76         if [ -x $mpoint/finalize.sh ]
  77         then
  78                 finalize_mpoint=$mpoint
  79                 keep_mpoint="yes"
  80         fi
  81
  82         if [ -e $mpoint/rootfs.img -a -e $mpoint/rootfs.md5 -a -z "$rootfs_file" ]
  83         then
  84                 rootfs_file=$mpoint/rootfs.img
  85                 rootfs_md5=`head -n 1 $mpoint/rootfs.md5 | cut -c 1-32`
  86                 rootfs_mpoint=$mpoint
  87                 keep_mpoint="yes"
  88         fi
  89
  90         if [ "$keep_mpoint" != "yes" ]
  91         then
  92                 umount $mpoint
  93         fi
  94 done
  95
  96 if [ -e u-boot.bin ]
  97 then
  98         size=`stat -c %s u-boot.bin`
  99         dd if=/dev/mtdblock1 of=u-boot.bin.flashed bs=$size count=1
 100         if ! cmp u-boot.bin u-boot.bin.flashed
 101         then
 102                 flash_eraseall /dev/mtd1
 103                 nandwrite -p /dev/mtd1 u-boot.bin
 104                 log "u-boot.bin flashed."
 105                 status_uboot="flashed"
 106         else
 107                 log "u-boot.bin already flashed, skipped."
 108                 status_uboot="already there"
 109         fi
 110 else
 111         log "u-boot is missing, skipped."
 112 fi
 113
 114 if ! cmp /dev/mtdblock2 /usr/local/erasednandblk
 115 then
 116         flash_eraseall /dev/mtd2
 117         status_uboot_env="cleared"
 118         log "u-boot environment cleared."
 119 fi
 120
 121 if [ -e uImage ]
 122 then
 123         # attempt to mount and check
 124         ubiattach /dev/ubi_ctrl -m 3 || true
 125         mount -t ubifs ubi0:boot /mnt/ubifs  2> /dev/null || true
 126         if ! cmp uImage /mnt/ubifs/uImage  2> /dev/null
 127         then
 128                 log "flashing uImage.."
 129                 umount /mnt/ubifs  2> /dev/null || true
 130                 ubidetach /dev/ubi_ctrl -d 0  2> /dev/null || true
 131                 # format just in case, should be harmless even it's formated already
 132                 ubiformat /dev/mtd3 -s 512 -y
 133                 ubiattach /dev/ubi_ctrl -m 3
 134                 ubimkvol /dev/ubi0 -m -N boot
 135                 mount -t ubifs ubi0:boot /mnt/ubifs
 136                 cp uImage /mnt/ubifs/uImage
 137                 status_uimage="flashed"
 138                 log "done"
 139         else
 140                 log "uImage already flashed, skipped."
 141                 status_uimage="already there"
 142         fi
 143         umount /mnt/ubifs
 144         ubidetach /dev/ubi_ctrl -d 0
 145 else
 146         log "uImage is missing, skipped."
 147 fi
 148
 149 if [ -n "$rootfs_file" ]
 150 then
 151         log "Calculating checksum, this might take a few minutes.."
 152         real_md5=`md5sum $rootfs_file | cut -c 1-32`
 153         if [ "$real_md5" = "$rootfs_md5" ]
 154         then
 155                 log $real_md5
 156         else
 157                 log "INCORRECT"
 158                 cleanup
 159                 false
 160         fi
 161
 162         # be destructive here, too hard to validate rootfs to bother
 163         log "formatting.."
 164         ubiformat /dev/mtd4 -s 512 -y
 165         ubiattach /dev/ubi_ctrl -m 4
 166         ubimkvol /dev/ubi0 -m -N rootfs
 167         log "done."
 168         log "Writing rootfs, please wait. This may take up to 15 minutes.."
 169         ubiupdatevol /dev/ubi0_0 $rootfs_file
 170         ubidetach /dev/ubi_ctrl -d 0
 171         status_rootfs="flashed"
 172         log "done."
 173 else
 174         log "root filesystem is missing, skipped."
 175 fi
 176
 177 log "======================="
 178 log "results:"
 179 log "u-boot:     $status_uboot"
 180 log "u-boot-env: $status_uboot_env"
 181 log "kernel:     $status_uimage"
 182 log "rootfs:     $status_rootfs $rootfs_file"
 183
 184 if [ -n "$finalize_mpoint" ]
 185 then
 186         log ""
 187         log "running finalize script.."
 188         cd $finalize_mpoint
 189         if ! ./finalize.sh
 190         then
 191                 log "finalize.sh failed."
 192                 cleanup
 193                 false
 194         fi
 195         log "done."
 196 fi
 197
 198 cleanup
 199
 200 exit 0