ar71xx: add support for TP-Link TL-MR11U
[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)
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         all0258n )
77                 platform_check_image_all0258n "$1" && return 0
78                 return 1
79                 ;;
80         alfa-ap96 | \
81         alfa-nx | \
82         ap113 | \
83         ap121 | \
84         ap121-mini | \
85         ap96 | \
86         db120 | \
87         hornet-ub | \
88         zcn-1523h-2 | \
89         zcn-1523h-5)
90                 [ "$magic_long" != "68737173" -a "$magic_long" != "19852003" ] && {
91                         echo "Invalid image type."
92                         return 1
93                 }
94                 return 0
95                 ;;
96         ap81 | \
97         ap83 | \
98         dir-600-a1 | \
99         dir-615-c1 | \
100         dir-825-b1 | \
101         mzk-w04nu | \
102         mzk-w300nh | \
103         tew-632brp | \
104         wrt400n | \
105         airrouter | \
106         bullet-m | \
107         nanostation-m | \
108         rocket-m | \
109         rw2458n | \
110         wzr-hp-g300nh2 | \
111         wzr-hp-g300nh | \
112         wzr-hp-g450h | \
113         wzr-hp-ag300h | \
114         whr-g301n | \
115         whr-hp-g300n | \
116         whr-hp-gn | \
117         nbg460n_550n_550nh | \
118         unifi )
119                 [ "$magic" != "2705" ] && {
120                         echo "Invalid image type."
121                         return 1
122                 }
123                 return 0
124                 ;;
125         tl-mr11u | \
126         tl-mr3020 | \
127         tl-mr3220 | \
128         tl-mr3420 | \
129         tl-wa901nd | \
130         tl-wa901nd-v2 | \
131         tl-wr703n | \
132         tl-wr741nd | \
133         tl-wr741nd-v4 | \
134         tl-wr841n-v1 | \
135         tl-wr841n-v7 | \
136         tl-wr941nd | \
137         tl-wr1043nd | \
138         tl-wr2543n)
139                 [ "$magic" != "0100" ] && {
140                         echo "Invalid image type."
141                         return 1
142                 }
143
144                 local hwid
145                 local imageid
146
147                 hwid=$(tplink_get_hwid)
148                 imageid=$(tplink_get_image_hwid "$1")
149
150                 [ "$hwid" != "$imageid" ] && {
151                         echo "Invalid image, hardware ID mismatch, hw:$hwid image:$imageid."
152                         return 1
153                 }
154
155                 return 0
156                 ;;
157         wndr3700)
158                 local hw_magic
159
160                 hw_magic="$(ar71xx_get_mtd_part_magic firmware)"
161                 [ "$magic_long" != "$hw_magic" ] && {
162                         echo "Invalid image, hardware ID mismatch, hw:$hw_magic image:$magic_long."
163                         return 1
164                 }
165                 return 0
166                 ;;
167         wrt160nl)
168                 [ "$magic" != "4e4c" ] && {
169                         echo "Invalid image type."
170                         return 1
171                 }
172                 return 0
173                 ;;
174         routerstation | \
175         routerstation-pro | \
176         ls-sr71 | \
177         pb42 | \
178         pb44 | \
179         eap7660d | \
180         ja76pf )
181                 [ "$magic" != "4349" ] && {
182                         echo "Invalid image. Use *-sysupgrade.bin files on this board"
183                         return 1
184                 }
185
186                 local md5_img=$(dd if="$1" bs=2 skip=9 count=16 2>/dev/null)
187                 local md5_chk=$(dd if="$1" bs=$CI_BLKSZ skip=1 2>/dev/null | md5sum -); md5_chk="${md5_chk%% *}"
188
189                 if [ -n "$md5_img" -a -n "$md5_chk" ] && [ "$md5_img" = "$md5_chk" ]; then
190                         return 0
191                 else
192                         echo "Invalid image. Contents do not match checksum (image:$md5_img calculated:$md5_chk)"
193                         return 1
194                 fi
195                 return 0
196                 ;;
197         esac
198
199         echo "Sysupgrade is not yet supported on $board."
200         return 1
201 }
202
203 platform_do_upgrade() {
204         local board=$(ar71xx_board_name)
205
206         case "$board" in
207         routerstation | \
208         routerstation-pro | \
209         ls-sr71 | \
210         eap7660d | \
211         pb42 | \
212         pb44 | \
213         ja76pf)
214                 platform_do_upgrade_combined "$ARGV"
215                 ;;
216         all0258n )
217                 platform_do_upgrade_all0258n "$ARGV"
218                 ;;
219         *)
220                 default_do_upgrade "$ARGV"
221                 ;;
222         esac
223 }
224
225 disable_watchdog() {
226         killall watchdog
227         ( ps | grep -v 'grep' | grep '/dev/watchdog' ) && {
228                 echo 'Could not disable watchdog'
229                 return 1
230         }
231 }
232
233 append sysupgrade_pre_upgrade disable_watchdog