summaryrefslogtreecommitdiffstats
path: root/target/linux/ar71xx/base-files/lib/upgrade/openmesh.sh
blob: 5179875fe04bfbf3f2603a85e9be5ceba8954073 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
# The U-Boot loader of the OpenMesh devices requires image sizes and
# checksums to be provided in the U-Boot environment.
# The OpenMesh devices come with 2 main partitions - while one is active
# sysupgrade will flash the other. The boot order is changed to boot the
# newly flashed partition. If the new partition can't be booted due to
# upgrade failures the previously used partition is loaded.

trim()
{
	echo $1
}

cfg_value_get()
{
	local cfg=$1 cfg_opt
	local section=$2 our_section=0
	local param=$3 our_param=

	for cfg_opt in $cfg
		do
			[ "$cfg_opt" = "[$section]" ] && our_section=1 && continue
			[ "$our_section" = "1" ] || continue

			our_param=$(echo ${cfg_opt%%=*})
			[ "$param" = "$our_param" ] && echo ${cfg_opt##*=} && break
		done
}

# make sure we got uboot-envtools and fw_env.config copied over to the ramfs
platform_add_ramfs_ubootenv()
{
	[ -e /usr/sbin/fw_printenv ] && install_bin /usr/sbin/fw_printenv /usr/sbin/fw_setenv
	[ -e /etc/fw_env.config ] && install_file /etc/fw_env.config
}
append sysupgrade_pre_upgrade platform_add_ramfs_ubootenv

platform_check_image_openmesh()
{
	local img_magic=$1
	local img_path=$2
	local fw_printenv=/usr/sbin/fw_printenv
	local img_board_target= img_num_files= i=0
	local cfg_name= kernel_name= rootfs_name=

	case "$img_magic" in
		# Combined Extended Image v1
		43453031)
			img_board_target=$(trim $(dd if="$img_path" bs=4 skip=1 count=8 2>/dev/null))
			img_num_files=$(trim $(dd if="$img_path" bs=2 skip=18 count=1 2>/dev/null))
			;;
		*)
			echo "Invalid image ($img_magic). Use combined extended images on this platform"
			return 1
			;;
	esac

	case "$img_board_target" in
		OM2P)
			[ "$board" = "om2p" ] && break
			[ "$board" = "om2p-lc" ] && break
			[ "$board" = "om2p-hs" ] && break
			echo "Invalid image board target ($img_board_target) for this platform: $board. Use the correct image for this platform"
			return 1
			;;
		MR600)
			[ "$board" = "mr600" ] && break
			[ "$board" = "mr600v2" ] && break
			echo "Invalid image board target ($img_board_target) for this platform: $board. Use the correct image for this platform"
			return 1
			;;
		*)
			echo "Invalid board target ($img_board_target). Use the correct image for this platform"
			return 1
			;;
	esac

	[ $img_num_files -ne 3 ] && {
		echo "Invalid number of embedded images ($img_num_files). Use the correct image for this platform"
		return 1
	}

	cfg_name=$(trim $(dd if="$img_path" bs=2 skip=19 count=16 2>/dev/null))

	[ "$cfg_name" != "fwupgrade.cfg" ] && {
		echo "Invalid embedded config file ($cfg_name). Use the correct image for this platform"
		return 1
	}

	kernel_name=$(trim $(dd if="$img_path" bs=2 skip=55 count=16 2>/dev/null))

	[ "$kernel_name" != "kernel" ] && {
		echo "Invalid embedded kernel file ($kernel_name). Use the correct image for this platform"
		return 1
	}

	rootfs_name=$(trim $(dd if="$img_path" bs=2 skip=91 count=16 2>/dev/null))

	[ "$rootfs_name" != "rootfs" ] && {
		echo "Invalid embedded kernel file ($rootfs_name). Use the correct image for this platform"
		return 1
	}

	[ ! -x "$fw_printenv" ] && {
		echo "Please install uboot-envtools!"
		return 1
	}

	[ ! -r "/etc/fw_env.config" ] && {
		echo "/etc/fw_env.config is missing"
		return 1
	}

	return 0
}

platform_do_upgrade_openmesh()
{
	local img_path=$1 img_board_target=
	local kernel_start_addr= kernel_start_addr1= kernel_start_addr2=
	local kernel_size= kernel_md5=
	local rootfs_size= rootfs_checksize= rootfs_md5=
	local kernel_bsize= total_size=
	local data_offset=$((64 * 1024)) block_size= offset=
	local uboot_env_upgrade="/tmp/fw_env_upgrade"
	local cfg_size= kernel_size= rootfs_size=
	local append=""

	[ -f "$CONF_TAR" -a "$SAVE_CONFIG" -eq 1 ] && append="-j $CONF_TAR"

	cfg_size=$(dd if="$img_path" bs=2 skip=35 count=4 2>/dev/null)
	kernel_size=$(dd if="$img_path" bs=2 skip=71 count=4 2>/dev/null)
	rootfs_size=$(dd if="$img_path" bs=2 skip=107 count=4 2>/dev/null)

	img_board_target=$(trim $(dd if="$img_path" bs=4 skip=1 count=8 2>/dev/null))
	cfg_content=$(dd if="$img_path" bs=1 skip=$data_offset count=$(echo $((0x$cfg_size))) 2>/dev/null)

	case $img_board_target in
		OM2P)
			block_size=$((256 * 1024))
			total_size=7340032
			kernel_start_addr1=0x9f1c0000
			kernel_start_addr2=0x9f8c0000
			;;
		MR600)
			block_size=$((64 * 1024))
			total_size=7995392
			kernel_start_addr1=0x9f0b0000
			kernel_start_addr2=0x9f850000
			;;
	esac

	kernel_md5=$(cfg_value_get "$cfg_content" "vmlinux" "md5sum")
	rootfs_md5=$(cfg_value_get "$cfg_content" "rootfs" "md5sum")
	rootfs_checksize=$(cfg_value_get "$cfg_content" "rootfs" "checksize")

	if [ "$((0x$kernel_size % $block_size))" = "0" ]
		then
			kernel_bsize=$(echo $((0x$kernel_size)))
		else
			kernel_bsize=$((0x$kernel_size + ($block_size - (0x$kernel_size % $block_size))))
	fi

	mtd -q erase inactive

	offset=$(echo $(($data_offset + 0x$cfg_size + 0x$kernel_size)))
	dd if="$img_path" bs=1 skip=$offset count=$(echo $((0x$rootfs_size))) 2>&- | mtd -n -p $kernel_bsize $append write - "inactive"

	offset=$(echo $(($data_offset + 0x$cfg_size)))
	dd if="$img_path" bs=1 skip=$offset count=$(echo $((0x$kernel_size))) 2>&- | mtd -n write - "inactive"

	rm $uboot_env_upgrade 2>&-

	if [ "$(grep 'mtd3:.*inactive' /proc/mtd)" ]
		then
			printf "kernel_size_1 %u\n" $(($kernel_bsize / 1024)) >> $uboot_env_upgrade
			printf "rootfs_size_1 %u\n" $((($total_size - $kernel_bsize) / 1024)) >> $uboot_env_upgrade
			printf "bootseq 1,2\n" >> $uboot_env_upgrade
			kernel_start_addr=$kernel_start_addr1
		else
			printf "kernel_size_2 %u\n" $(($kernel_bsize / 1024)) >> $uboot_env_upgrade
			printf "rootfs_size_2 %u\n" $((($total_size - $kernel_bsize) / 1024)) >> $uboot_env_upgrade
			printf "bootseq 2,1\n" >> $uboot_env_upgrade
			kernel_start_addr=$kernel_start_addr2
	fi

	printf "vmlinux_start_addr %s\n" $kernel_start_addr >> $uboot_env_upgrade
	printf "vmlinux_size 0x%s\n" $kernel_size >> $uboot_env_upgrade
	printf "vmlinux_checksum %s\n" $kernel_md5 >> $uboot_env_upgrade
	printf "rootfs_start_addr 0x%x\n" $(($kernel_start_addr + $kernel_bsize)) >> $uboot_env_upgrade
	printf "rootfs_size %s\n" $rootfs_checksize >> $uboot_env_upgrade
	printf "rootfs_checksum %s\n" $rootfs_md5 >> $uboot_env_upgrade

	fw_setenv -s $uboot_env_upgrade || {
		echo "failed to update U-Boot environment"
		return 1
	}
}