1#!/bin/sh
2#
3# Copyright (C) 2012 OpenWrt.org
4#
5
6. /lib/functions.sh
7. /lib/ar71xx.sh
8
9get_mtd_part_size() {
10	local part_name=$1
11	local first dev size erasesize name
12	while read dev size erasesize name; do
13		name=${name#'"'}; name=${name%'"'}
14		if [ "$name" = "$part_name" ]; then
15			echo $((0x$size))
16			break
17		fi
18	done < /proc/mtd
19}
20
21get_magic_at() {
22	local mtddev=$1
23	local pos=$2
24	dd bs=1 count=2 skip=$pos if=$mtddev 2>/dev/null | hexdump -v -n 4 -e '1/1 "%02x"'
25}
26
27dir825b_is_caldata_valid() {
28	local mtddev=$1
29	local magic
30
31	magic=$(get_magic_at $mtddev 4096)
32	[ "$magic" != "a55a" ] && return 0
33
34	magic=$(get_magic_at $mtddev 20480)
35	[ "$magic" != "a55a" ] && return 0
36
37	return 1
38}
39
40dir825b_copy_caldata() {
41	local cal_src=$1
42	local cal_dst=$2
43	local mtd_src
44	local mtd_dst
45	local md5_src
46	local md5_dst
47
48	mtd_src=$(find_mtd_part $cal_src)
49	[ -z "$mtd_src" ] && {
50		echo "no $cal_src partition found"
51		return 1
52	}
53
54	mtd_dst=$(find_mtd_part $cal_dst)
55	[ -z "$mtd_dst" ] && {
56		echo "no $cal_dst partition found"
57		return 1
58	}
59
60	dir825b_is_caldata_valid "$mtd_src" && {
61		echo "no valid calibration data found in $cal_src"
62		return 1
63	}
64
65	dir825b_is_caldata_valid "$mtd_dst" && {
66		echo "Copying calibration data from $cal_src to $cal_dst..."
67		dd if="$mtd_src" 2>/dev/null | mtd -q -q write - "$cal_dst"
68	}
69
70        md5_src=$(md5sum "$mtd_src") && md5_src="${md5_src%% *}"
71        md5_dst=$(md5sum "$mtd_dst") && md5_dst="${md5_dst%% *}"
72
73	[ "$md5_src" != "$md5_dst" ] && {
74		echo "calibration data mismatch $cal_src:$md5_src $cal_dst:$md5_dst"
75		return 1
76	}
77
78	return 0
79}
80
81dir825b_do_upgrade_combined() {
82	local fw_part=$1
83	local fw_file=$2
84	local fw_mtd=$(find_mtd_part $fw_part)
85	local fw_length=0x$(dd if="$fw_file" bs=2 skip=1 count=4 2>/dev/null)
86	local fw_blocks=$(($fw_length / 65536))
87
88	if [ -n "$fw_mtd" ] &&  [ ${fw_blocks:-0} -gt 0 ]; then
89		local append=""
90		[ -f "$CONF_TAR" -a "$SAVE_CONFIG" -eq 1 ] && append="-j $CONF_TAR"
91
92		sync
93		dd if="$fw_file" bs=64k skip=1 count=$fw_blocks 2>/dev/null | \
94			mtd $append write - "$fw_part"
95	fi
96}
97
98dir825b_check_image() {
99	local magic="$(get_magic_long "$1")"
100	local fw_mtd=$(find_mtd_part "firmware_orig")
101
102	case "$magic" in
103	"27051956")
104		;;
105	"43493030")
106		local md5_img=$(dd if="$1" bs=2 skip=9 count=16 2>/dev/null)
107		local md5_chk=$(dd if="$1" bs=64k skip=1 2>/dev/null | md5sum -); md5_chk="${md5_chk%% *}"
108		local fw_len=$(dd if="$1" bs=2 skip=1 count=4 2>/dev/null)
109		local fw_part_len=$(get_mtd_part_size "firmware")
110
111		if [ -z "$fw_mtd" ]; then
112			ask_bool 0 "Do you have a backup of the caldata partition?" || {
113				echo "Warning, please make sure that you have a backup of the caldata partition."
114				echo "Once you have that, use 'sysupgrade -i' for upgrading to the 'fat' firmware."
115				return 1
116			}
117		fi
118
119		if [ -z "$md5_img" -o -z "$md5_chk" ]; then
120			echo "Unable to get image checksums. Maybe you are using a streamed image?"
121			return 1
122		fi
123
124		if [ "$md5_img" != "$md5_chk" ]; then
125			echo "Invalid image. Contents do not match checksum (image:$md5_img calculated:$md5_chk)"
126			return 1
127		fi
128
129		fw_len=$((0x$fw_len))
130		fw_part_len=${fw_part_len:-0}
131
132		if [ $fw_part_len -lt $fw_len ]; then
133			echo "The upgrade image is too big (size:$fw_len available:$fw_part_len)"
134			return 1
135		fi
136		;;
137	*)
138		echo "Unsupported image format."
139		return 1
140		;;
141	esac
142
143	return 0
144}
145
146platform_do_upgrade_dir825b() {
147	local magic="$(get_magic_long "$1")"
148	local fw_mtd=$(find_mtd_part "firmware_orig")
149
150	case "$magic" in
151	"27051956")
152		if [ -n "$fw_mtd" ]; then
153			# restore calibration data before downgrading to
154			# the normal image
155			dir825b_copy_caldata "caldata" "caldata_orig" || {
156				echo "unable to restore calibration data"
157				exit 1
158			}
159			PART_NAME="firmware_orig"
160		else
161			PART_NAME="firmware"
162		fi
163		default_do_upgrade "$ARGV"
164		;;
165	"43493030")
166		if [ -z "$fw_mtd" ]; then
167			# backup calibration data before upgrading to the
168			# fat image
169			dir825b_copy_caldata "caldata" "caldata_copy" || {
170				echo "unable to backup calibration data"
171				exit 1
172			}
173		fi
174		dir825b_do_upgrade_combined "firmware" "$ARGV"
175		;;
176	esac
177}
178