| 1 | # |
| 2 | # Copyright (C) 2009 OpenWrt.org |
| 3 | # |
| 4 | |
| 5 | . /lib/ar71xx.sh |
| 6 | |
| 7 | PART_NAME=firmware |
| 8 | RAMFS_COPY_DATA=/lib/ar71xx.sh |
| 9 | |
| 10 | CI_BLKSZ=65536 |
| 11 | CI_LDADR=0x80060000 |
| 12 | |
| 13 | platform_find_partitions() { |
| 14 | local first dev size erasesize name |
| 15 | while read dev size erasesize name; do |
| 16 | name=${name#'"'}; name=${name%'"'} |
| 17 | case "$name" in |
| 18 | vmlinux.bin.l7|kernel|linux|rootfs) |
| 19 | if [ -z "$first" ]; then |
| 20 | first="$name" |
| 21 | else |
| 22 | echo "$erasesize:$first:$name" |
| 23 | break |
| 24 | fi |
| 25 | ;; |
| 26 | esac |
| 27 | done < /proc/mtd |
| 28 | } |
| 29 | |
| 30 | platform_find_kernelpart() { |
| 31 | local part |
| 32 | for part in "${1%:*}" "${1#*:}"; do |
| 33 | case "$part" in |
| 34 | vmlinux.bin.l7|kernel|linux) |
| 35 | echo "$part" |
| 36 | break |
| 37 | ;; |
| 38 | esac |
| 39 | done |
| 40 | } |
| 41 | |
| 42 | platform_do_upgrade_combined() { |
| 43 | local partitions=$(platform_find_partitions) |
| 44 | local kernelpart=$(platform_find_kernelpart "${partitions#*:}") |
| 45 | local erase_size=$((0x${partitions%%:*})); partitions="${partitions#*:}" |
| 46 | local kern_length=0x$(dd if="$1" bs=2 skip=1 count=4 2>/dev/null) |
| 47 | local kern_blocks=$(($kern_length / $CI_BLKSZ)) |
| 48 | local root_blocks=$((0x$(dd if="$1" bs=2 skip=5 count=4 2>/dev/null) / $CI_BLKSZ)) |
| 49 | |
| 50 | if [ -n "$partitions" ] && [ -n "$kernelpart" ] && \ |
| 51 | [ ${kern_blocks:-0} -gt 0 ] && \ |
| 52 | [ ${root_blocks:-0} -gt ${kern_blocks:-0} ] && \ |
| 53 | [ ${erase_size:-0} -gt 0 ]; |
| 54 | then |
| 55 | local append="" |
| 56 | [ -f "$CONF_TAR" -a "$SAVE_CONFIG" -eq 1 ] && append="-j $CONF_TAR" |
| 57 | |
| 58 | ( dd if="$1" bs=$CI_BLKSZ skip=1 count=$kern_blocks 2>/dev/null; \ |
| 59 | dd if="$1" bs=$CI_BLKSZ skip=$((1+$kern_blocks)) count=$root_blocks 2>/dev/null ) | \ |
| 60 | mtd -r $append -F$kernelpart:$kern_length:$CI_LDADR,rootfs write - $partitions |
| 61 | fi |
| 62 | } |
| 63 | |
| 64 | platform_check_image() { |
| 65 | local board=$(ar71xx_board_name) |
| 66 | local magic="$(get_magic_word "$1")" |
| 67 | |
| 68 | [ "$ARGC" -gt 1 ] && return 1 |
| 69 | |
| 70 | case "$board" in |
| 71 | ap81 | ap83 | dir-600-a1 | dir-615-c1 | dir-825-b1 | mzk-w04nu | mzk-w300nh | tew-632brp | wrt400n | bullet-m | nanostation-m | rocket-m | wzr-hp-g300nh ) |
| 72 | [ "$magic" != "2705" ] && { |
| 73 | echo "Invalid image type." |
| 74 | return 1 |
| 75 | } |
| 76 | return 0 |
| 77 | ;; |
| 78 | tl-wr741nd | tl-wr841n-v1 | tl-wr941nd | tl-wr1043nd) |
| 79 | [ "$magic" != "0100" ] && { |
| 80 | echo "Invalid image type." |
| 81 | return 1 |
| 82 | } |
| 83 | return 0 |
| 84 | ;; |
| 85 | wndr3700) |
| 86 | [ "$magic" != "3337" ] && { |
| 87 | echo "Invalid image type." |
| 88 | return 1 |
| 89 | } |
| 90 | return 0 |
| 91 | ;; |
| 92 | wrt160nl) |
| 93 | [ "$magic" != "4e4c" ] && { |
| 94 | echo "Invalid image type." |
| 95 | return 1 |
| 96 | } |
| 97 | return 0 |
| 98 | ;; |
| 99 | routerstation | routerstation-pro | ls-sr71 | pb42 | pb44) |
| 100 | [ "$magic" != "4349" ] && { |
| 101 | echo "Invalid image. Use *-sysupgrade.bin files on this board" |
| 102 | return 1 |
| 103 | } |
| 104 | |
| 105 | local md5_img=$(dd if="$1" bs=2 skip=9 count=16 2>/dev/null) |
| 106 | local md5_chk=$(dd if="$1" bs=$CI_BLKSZ skip=1 2>/dev/null | md5sum -); md5_chk="${md5_chk%% *}" |
| 107 | |
| 108 | if [ -n "$md5_img" -a -n "$md5_chk" ] && [ "$md5_img" = "$md5_chk" ]; then |
| 109 | return 0 |
| 110 | else |
| 111 | echo "Invalid image. Contents do not match checksum (image:$md5_img calculated:$md5_chk)" |
| 112 | return 1 |
| 113 | fi |
| 114 | return 0 |
| 115 | ;; |
| 116 | esac |
| 117 | |
| 118 | echo "Sysupgrade is not yet supported on $board." |
| 119 | return 1 |
| 120 | } |
| 121 | |
| 122 | platform_do_upgrade() { |
| 123 | local board=$(ar71xx_board_name) |
| 124 | |
| 125 | case "$board" in |
| 126 | routerstation | routerstation-pro | ls-sr71) |
| 127 | platform_do_upgrade_combined "$ARGV" |
| 128 | ;; |
| 129 | *) |
| 130 | default_do_upgrade "$ARGV" |
| 131 | ;; |
| 132 | esac |
| 133 | } |
| 134 | |
| 135 | disable_watchdog() { |
| 136 | killall watchdog |
| 137 | ( ps | grep -v 'grep' | grep '/dev/watchdog' ) && { |
| 138 | echo 'Could not disable watchdog' |
| 139 | return 1 |
| 140 | } |
| 141 | } |
| 142 | |
| 143 | append sysupgrade_pre_upgrade disable_watchdog |
| 144 | |
