[ar71xx] dir825b1: Add image for DIR-825-B1 that uses the whole flash (must first...
[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|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)
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         dir-825-b1 | \
104         dir-825-b1-openwrt | \
105         dir-825-b1-tostock | \
106         ew-dorin | \
107         ew-dorin-router | \
108         mzk-w04nu | \
109         mzk-w300nh | \
110         tew-632brp | \
111         tew-673gru | \
112         tew-712br | \
113         wrt400n | \
114         airrouter | \
115         bullet-m | \
116         nanostation-m | \
117         rocket-m | \
118         rw2458n | \
119         wzr-hp-g300nh2 | \
120         wzr-hp-g300nh | \
121         wzr-hp-g450h | \
122         wzr-hp-ag300h | \
123         whr-g301n | \
124         whr-hp-g300n | \
125         whr-hp-gn | \
126         wlae-ag300n | \
127         nbg460n_550n_550nh | \
128         unifi )
129                 [ "$magic" != "2705" ] && {
130                         echo "Invalid image type."
131                         return 1
132                 }
133                 return 0
134                 ;;
135         om2p | \
136         om2p-lc)
137                 platform_check_image_om2p "$magic_long" "$1" && return 0
138                 return 1
139                 ;;
140         tl-mr11u | \
141         tl-mr3020 | \
142         tl-mr3220 | \
143         tl-mr3420 | \
144         tl-wa901nd | \
145         tl-wa901nd-v2 | \
146         tl-wdr4300 | \
147         tl-wr703n | \
148         tl-wr741nd | \
149         tl-wr741nd-v4 | \
150         tl-wr841n-v1 | \
151         tl-wr841n-v7 | \
152         tl-wr941nd | \
153         tl-wr1041n-v2 | \
154         tl-wr1043nd | \
155         tl-wr2543n)
156                 [ "$magic" != "0100" ] && {
157                         echo "Invalid image type."
158                         return 1
159                 }
160
161                 local hwid
162                 local imageid
163
164                 hwid=$(tplink_get_hwid)
165                 imageid=$(tplink_get_image_hwid "$1")
166
167                 [ "$hwid" != "$imageid" ] && {
168                         echo "Invalid image, hardware ID mismatch, hw:$hwid image:$imageid."
169                         return 1
170                 }
171
172                 return 0
173                 ;;
174         wndr3700)
175                 local hw_magic
176
177                 hw_magic="$(ar71xx_get_mtd_part_magic firmware)"
178                 [ "$magic_long" != "$hw_magic" ] && {
179                         echo "Invalid image, hardware ID mismatch, hw:$hw_magic image:$magic_long."
180                         return 1
181                 }
182                 return 0
183                 ;;
184         wrt160nl)
185                 [ "$magic" != "4e4c" ] && {
186                         echo "Invalid image type."
187                         return 1
188                 }
189                 return 0
190                 ;;
191         routerstation | \
192         routerstation-pro | \
193         ls-sr71 | \
194         pb42 | \
195         pb44 | \
196         all0305 | \
197         eap7660d | \
198         ja76pf | \
199         ja76pf2)
200                 [ "$magic" != "4349" ] && {
201                         echo "Invalid image. Use *-sysupgrade.bin files on this board"
202                         return 1
203                 }
204
205                 local md5_img=$(dd if="$1" bs=2 skip=9 count=16 2>/dev/null)
206                 local md5_chk=$(dd if="$1" bs=$CI_BLKSZ skip=1 2>/dev/null | md5sum -); md5_chk="${md5_chk%% *}"
207
208                 if [ -n "$md5_img" -a -n "$md5_chk" ] && [ "$md5_img" = "$md5_chk" ]; then
209                         return 0
210                 else
211                         echo "Invalid image. Contents do not match checksum (image:$md5_img calculated:$md5_chk)"
212                         return 1
213                 fi
214                 return 0
215                 ;;
216         esac
217
218         echo "Sysupgrade is not yet supported on $board."
219         return 1
220 }
221
222 platform_do_upgrade() {
223         local board=$(ar71xx_board_name)
224
225         case "$board" in
226         routerstation | \
227         routerstation-pro | \
228         ls-sr71 | \
229         all0305 | \
230         eap7660d | \
231         pb42 | \
232         pb44 | \
233         ja76pf | \
234         ja76pf2)
235                 platform_do_upgrade_combined "$ARGV"
236                 ;;
237         all0258n )
238                 platform_do_upgrade_allnet "0x9f050000" "$ARGV"
239                 ;;
240         all0315n )
241                 platform_do_upgrade_allnet "0x9f080000" "$ARGV"
242                 ;;
243         om2p | \
244         om2p-lc)
245                 platform_do_upgrade_om2p "$ARGV"
246                 ;;
247         *)
248                 default_do_upgrade "$ARGV"
249                 ;;
250         esac
251 }
252
253 disable_watchdog() {
254         killall watchdog
255         ( ps | grep -v 'grep' | grep '/dev/watchdog' ) && {
256                 echo 'Could not disable watchdog'
257                 return 1
258         }
259 }
260
261 append sysupgrade_pre_upgrade disable_watchdog