ar71xx: add sysupgrade support for OpenMesh MR900v2
[openwrt.git] / target / linux / ar71xx / base-files / lib / upgrade / openmesh.sh
1 # The U-Boot loader of the OpenMesh devices requires image sizes and
2 # checksums to be provided in the U-Boot environment.
3 # The OpenMesh devices come with 2 main partitions - while one is active
4 # sysupgrade will flash the other. The boot order is changed to boot the
5 # newly flashed partition. If the new partition can't be booted due to
6 # upgrade failures the previously used partition is loaded.
7
8 trim()
9 {
10         echo $1
11 }
12
13 cfg_value_get()
14 {
15         local cfg=$1 cfg_opt
16         local section=$2 our_section=0
17         local param=$3 our_param=
18
19         for cfg_opt in $cfg
20                 do
21                         [ "$cfg_opt" = "[$section]" ] && our_section=1 && continue
22                         [ "$our_section" = "1" ] || continue
23
24                         our_param=$(echo ${cfg_opt%%=*})
25                         [ "$param" = "$our_param" ] && echo ${cfg_opt##*=} && break
26                 done
27 }
28
29 # make sure we got uboot-envtools and fw_env.config copied over to the ramfs
30 # create /var/lock for the lock "fw_setenv.lock" of fw_setenv
31 platform_add_ramfs_ubootenv()
32 {
33         [ -e /usr/sbin/fw_printenv ] && install_bin /usr/sbin/fw_printenv /usr/sbin/fw_setenv
34         [ -e /etc/fw_env.config ] && install_file /etc/fw_env.config
35         mkdir -p $RAM_ROOT/var/lock
36 }
37 append sysupgrade_pre_upgrade platform_add_ramfs_ubootenv
38
39 platform_check_image_openmesh()
40 {
41         local img_magic=$1
42         local img_path=$2
43         local fw_printenv=/usr/sbin/fw_printenv
44         local img_board_target= img_num_files= i=0
45         local cfg_name= kernel_name= rootfs_name=
46
47         case "$img_magic" in
48                 # Combined Extended Image v1
49                 43453031)
50                         img_board_target=$(trim $(dd if="$img_path" bs=4 skip=1 count=8 2>/dev/null))
51                         img_num_files=$(trim $(dd if="$img_path" bs=2 skip=18 count=1 2>/dev/null))
52                         ;;
53                 *)
54                         echo "Invalid image ($img_magic). Use combined extended images on this platform"
55                         return 1
56                         ;;
57         esac
58
59         case "$img_board_target" in
60                 OM2P)
61                         [ "$board" = "om2p" ] && break
62                         [ "$board" = "om2pv2" ] && break
63                         [ "$board" = "om2p-lc" ] && break
64                         [ "$board" = "om2p-hs" ] && break
65                         [ "$board" = "om2p-hsv2" ] && break
66                         echo "Invalid image board target ($img_board_target) for this platform: $board. Use the correct image for this platform"
67                         return 1
68                         ;;
69                 OM5P)
70                         [ "$board" = "om5p" ] && break
71                         echo "Invalid image board target ($img_board_target) for this platform: $board. Use the correct image for this platform"
72                         return 1
73                         ;;
74                 MR600)
75                         [ "$board" = "mr600" ] && break
76                         [ "$board" = "mr600v2" ] && break
77                         echo "Invalid image board target ($img_board_target) for this platform: $board. Use the correct image for this platform"
78                         return 1
79                         ;;
80                 MR900)
81                         [ "$board" = "mr900" ] && break
82                         [ "$board" = "mr900v2" ] && break
83                         echo "Invalid image board target ($img_board_target) for this platform: $board. Use the correct image for this platform"
84                         return 1
85                         ;;
86                 *)
87                         echo "Invalid board target ($img_board_target). Use the correct image for this platform"
88                         return 1
89                         ;;
90         esac
91
92         [ $img_num_files -ne 3 ] && {
93                 echo "Invalid number of embedded images ($img_num_files). Use the correct image for this platform"
94                 return 1
95         }
96
97         cfg_name=$(trim $(dd if="$img_path" bs=2 skip=19 count=16 2>/dev/null))
98
99         [ "$cfg_name" != "fwupgrade.cfg" ] && {
100                 echo "Invalid embedded config file ($cfg_name). Use the correct image for this platform"
101                 return 1
102         }
103
104         kernel_name=$(trim $(dd if="$img_path" bs=2 skip=55 count=16 2>/dev/null))
105
106         [ "$kernel_name" != "kernel" ] && {
107                 echo "Invalid embedded kernel file ($kernel_name). Use the correct image for this platform"
108                 return 1
109         }
110
111         rootfs_name=$(trim $(dd if="$img_path" bs=2 skip=91 count=16 2>/dev/null))
112
113         [ "$rootfs_name" != "rootfs" ] && {
114                 echo "Invalid embedded kernel file ($rootfs_name). Use the correct image for this platform"
115                 return 1
116         }
117
118         [ ! -x "$fw_printenv" ] && {
119                 echo "Please install uboot-envtools!"
120                 return 1
121         }
122
123         [ ! -r "/etc/fw_env.config" ] && {
124                 echo "/etc/fw_env.config is missing"
125                 return 1
126         }
127
128         return 0
129 }
130
131 platform_do_upgrade_openmesh()
132 {
133         local img_path=$1 img_board_target=
134         local kernel_start_addr= kernel_start_addr1= kernel_start_addr2=
135         local kernel_size= kernel_md5=
136         local rootfs_size= rootfs_checksize= rootfs_md5=
137         local kernel_bsize= total_size=
138         local data_offset=$((64 * 1024)) block_size= offset=
139         local uboot_env_upgrade="/tmp/fw_env_upgrade"
140         local cfg_size= kernel_size= rootfs_size=
141         local append=""
142
143         [ -f "$CONF_TAR" -a "$SAVE_CONFIG" -eq 1 ] && append="-j $CONF_TAR"
144
145         cfg_size=$(dd if="$img_path" bs=2 skip=35 count=4 2>/dev/null)
146         kernel_size=$(dd if="$img_path" bs=2 skip=71 count=4 2>/dev/null)
147         rootfs_size=$(dd if="$img_path" bs=2 skip=107 count=4 2>/dev/null)
148
149         img_board_target=$(trim $(dd if="$img_path" bs=4 skip=1 count=8 2>/dev/null))
150         cfg_content=$(dd if="$img_path" bs=1 skip=$data_offset count=$(echo $((0x$cfg_size))) 2>/dev/null)
151
152         case $img_board_target in
153                 OM2P)
154                         block_size=$((256 * 1024))
155                         total_size=7340032
156                         kernel_start_addr1=0x9f1c0000
157                         kernel_start_addr2=0x9f8c0000
158                         ;;
159                 OM5P|MR600|MR900)
160                         block_size=$((64 * 1024))
161                         total_size=7995392
162                         kernel_start_addr1=0x9f0b0000
163                         kernel_start_addr2=0x9f850000
164                         ;;
165         esac
166
167         kernel_md5=$(cfg_value_get "$cfg_content" "vmlinux" "md5sum")
168         rootfs_md5=$(cfg_value_get "$cfg_content" "rootfs" "md5sum")
169         rootfs_checksize=$(cfg_value_get "$cfg_content" "rootfs" "checksize")
170
171         if [ "$((0x$kernel_size % $block_size))" = "0" ]
172                 then
173                         kernel_bsize=$(echo $((0x$kernel_size)))
174                 else
175                         kernel_bsize=$((0x$kernel_size + ($block_size - (0x$kernel_size % $block_size))))
176         fi
177
178         mtd -q erase inactive
179
180         offset=$(echo $(($data_offset + 0x$cfg_size + 0x$kernel_size)))
181         dd if="$img_path" bs=1 skip=$offset count=$(echo $((0x$rootfs_size))) 2>&- | mtd -n -p $kernel_bsize $append write - "inactive"
182
183         offset=$(echo $(($data_offset + 0x$cfg_size)))
184         dd if="$img_path" bs=1 skip=$offset count=$(echo $((0x$kernel_size))) 2>&- | mtd -n write - "inactive"
185
186         rm $uboot_env_upgrade 2>&-
187
188         if [ "$(grep 'mtd3:.*inactive' /proc/mtd)" ]
189                 then
190                         printf "kernel_size_1 %u\n" $(($kernel_bsize / 1024)) >> $uboot_env_upgrade
191                         printf "rootfs_size_1 %u\n" $((($total_size - $kernel_bsize) / 1024)) >> $uboot_env_upgrade
192                         printf "bootseq 1,2\n" >> $uboot_env_upgrade
193                         kernel_start_addr=$kernel_start_addr1
194                 else
195                         printf "kernel_size_2 %u\n" $(($kernel_bsize / 1024)) >> $uboot_env_upgrade
196                         printf "rootfs_size_2 %u\n" $((($total_size - $kernel_bsize) / 1024)) >> $uboot_env_upgrade
197                         printf "bootseq 2,1\n" >> $uboot_env_upgrade
198                         kernel_start_addr=$kernel_start_addr2
199         fi
200
201         printf "vmlinux_start_addr %s\n" $kernel_start_addr >> $uboot_env_upgrade
202         printf "vmlinux_size 0x%s\n" $kernel_size >> $uboot_env_upgrade
203         printf "vmlinux_checksum %s\n" $kernel_md5 >> $uboot_env_upgrade
204         printf "rootfs_start_addr 0x%x\n" $(($kernel_start_addr + $kernel_bsize)) >> $uboot_env_upgrade
205         printf "rootfs_size %s\n" $rootfs_checksize >> $uboot_env_upgrade
206         printf "rootfs_checksum %s\n" $rootfs_md5 >> $uboot_env_upgrade
207
208         fw_setenv -s $uboot_env_upgrade || {
209                 echo "failed to update U-Boot environment"
210                 return 1
211         }
212 }