ar71xx: fix sysupgrade for jjPlus JWAP003
[openwrt.git] / target / linux / ar71xx / base-files / lib / upgrade / platform.sh
1 #
2 # Copyright (C) 2011 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|vmlinux|kernel|linux|linux.bin|rootfs|filesystem)
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|vmlinux|kernel|linux|linux.bin)
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 tplink_get_image_hwid() {
65         get_image "$@" | dd bs=4 count=1 skip=16 2>/dev/null | hexdump -v -n 4 -e '1/1 "%02x"'
66 }
67
68 platform_check_image() {
69         local board=$(ar71xx_board_name)
70         local magic="$(get_magic_word "$1")"
71         local magic_long="$(get_magic_long "$1")"
72
73         [ "$ARGC" -gt 1 ] && return 1
74
75         case "$board" in
76         all0315n | \
77         all0258n )
78                 platform_check_image_allnet "$1" && return 0
79                 return 1
80                 ;;
81         alfa-ap96 | \
82         alfa-nx | \
83         ap113 | \
84         ap121 | \
85         ap121-mini | \
86         ap136 | \
87         ap96 | \
88         db120 | \
89         hornet-ub | \
90         zcn-1523h-2 | \
91         zcn-1523h-5)
92                 [ "$magic_long" != "68737173" -a "$magic_long" != "19852003" ] && {
93                         echo "Invalid image type."
94                         return 1
95                 }
96                 return 0
97                 ;;
98         ap81 | \
99         ap83 | \
100         dir-600-a1 | \
101         dir-615-c1 | \
102         dir-615-e4 | \
103         ew-dorin | \
104         ew-dorin-router | \
105         mzk-w04nu | \
106         mzk-w300nh | \
107         tew-632brp | \
108         tew-712br | \
109         wrt400n | \
110         airrouter | \
111         bullet-m | \
112         nanostation-m | \
113         rocket-m | \
114         rw2458n | \
115         wzr-hp-g300nh2 | \
116         wzr-hp-g300nh | \
117         wzr-hp-g450h | \
118         wzr-hp-ag300h | \
119         whr-g301n | \
120         whr-hp-g300n | \
121         whr-hp-gn | \
122         wlae-ag300n | \
123         nbg460n_550n_550nh | \
124         unifi )
125                 [ "$magic" != "2705" ] && {
126                         echo "Invalid image type."
127                         return 1
128                 }
129                 return 0
130                 ;;
131
132         dir-825-b1 | \
133         tew-673gru)
134                 dir825b_check_image "$1" && return 0
135                 ;;
136
137         om2p | \
138         om2p-lc)
139                 platform_check_image_om2p "$magic_long" "$1" && return 0
140                 return 1
141                 ;;
142         tl-mr11u | \
143         tl-mr3020 | \
144         tl-mr3040 | \
145         tl-mr3220 | \
146         tl-mr3420 | \
147         tl-wa901nd | \
148         tl-wa901nd-v2 | \
149         tl-wdr4300 | \
150         tl-wr703n | \
151         tl-wr741nd | \
152         tl-wr741nd-v4 | \
153         tl-wr841n-v1 | \
154         tl-wr841n-v7 | \
155         tl-wr841n-v8 | \
156         tl-wr941nd | \
157         tl-wr1041n-v2 | \
158         tl-wr1043nd | \
159         tl-wr2543n)
160                 [ "$magic" != "0100" ] && {
161                         echo "Invalid image type."
162                         return 1
163                 }
164
165                 local hwid
166                 local imageid
167
168                 hwid=$(tplink_get_hwid)
169                 imageid=$(tplink_get_image_hwid "$1")
170
171                 [ "$hwid" != "$imageid" ] && {
172                         echo "Invalid image, hardware ID mismatch, hw:$hwid image:$imageid."
173                         return 1
174                 }
175
176                 return 0
177                 ;;
178         wndr3700)
179                 local hw_magic
180
181                 hw_magic="$(ar71xx_get_mtd_part_magic firmware)"
182                 [ "$magic_long" != "$hw_magic" ] && {
183                         echo "Invalid image, hardware ID mismatch, hw:$hw_magic image:$magic_long."
184                         return 1
185                 }
186                 return 0
187                 ;;
188         wrt160nl)
189                 [ "$magic" != "4e4c" ] && {
190                         echo "Invalid image type."
191                         return 1
192                 }
193                 return 0
194                 ;;
195         routerstation | \
196         routerstation-pro | \
197         ls-sr71 | \
198         pb42 | \
199         pb44 | \
200         all0305 | \
201         eap7660d | \
202         ja76pf | \
203         ja76pf2 | \
204         jwap003)
205                 [ "$magic" != "4349" ] && {
206                         echo "Invalid image. Use *-sysupgrade.bin files on this board"
207                         return 1
208                 }
209
210                 local md5_img=$(dd if="$1" bs=2 skip=9 count=16 2>/dev/null)
211                 local md5_chk=$(dd if="$1" bs=$CI_BLKSZ skip=1 2>/dev/null | md5sum -); md5_chk="${md5_chk%% *}"
212
213                 if [ -n "$md5_img" -a -n "$md5_chk" ] && [ "$md5_img" = "$md5_chk" ]; then
214                         return 0
215                 else
216                         echo "Invalid image. Contents do not match checksum (image:$md5_img calculated:$md5_chk)"
217                         return 1
218                 fi
219                 return 0
220                 ;;
221         esac
222
223         echo "Sysupgrade is not yet supported on $board."
224         return 1
225 }
226
227 platform_do_upgrade() {
228         local board=$(ar71xx_board_name)
229
230         case "$board" in
231         routerstation | \
232         routerstation-pro | \
233         ls-sr71 | \
234         all0305 | \
235         eap7660d | \
236         pb42 | \
237         pb44 | \
238         ja76pf | \
239         ja76pf2 | \
240         jwap003)
241                 platform_do_upgrade_combined "$ARGV"
242                 ;;
243         all0258n )
244                 platform_do_upgrade_allnet "0x9f050000" "$ARGV"
245                 ;;
246         all0315n )
247                 platform_do_upgrade_allnet "0x9f080000" "$ARGV"
248                 ;;
249         dir-825-b1 |\
250         tew-673gru)
251                 platform_do_upgrade_dir825b "$ARGV"
252                 ;;
253         om2p | \
254         om2p-lc)
255                 platform_do_upgrade_om2p "$ARGV"
256                 ;;
257         *)
258                 default_do_upgrade "$ARGV"
259                 ;;
260         esac
261 }
262
263 disable_watchdog() {
264         killall watchdog
265         ( ps | grep -v 'grep' | grep '/dev/watchdog' ) && {
266                 echo 'Could not disable watchdog'
267                 return 1
268         }
269 }
270
271 append sysupgrade_pre_upgrade disable_watchdog