557f43ce6fb30530f6245e7a9e9da81c04646b3b
[openwrt/openwrt.git] / target / linux / ixp4xx / base-files / lib / upgrade / platform.sh
1 CI_BLKSZ=65536
2 CI_LDADR=0x00800000
3
4 platform_find_partitions() {
5 local first dev size erasesize name
6 while read dev size erasesize name; do
7 name=${name#'"'}; name=${name%'"'}
8 case "$name" in
9 vmlinux.bin.l7|kernel|linux|rootfs)
10 if [ -z "$first" ]; then
11 first="$name"
12 else
13 echo "$erasesize:$first:$name"
14 break
15 fi
16 ;;
17 esac
18 done < /proc/mtd
19 }
20
21 platform_find_kernelpart() {
22 local part
23 for part in "${1%:*}" "${1#*:}"; do
24 case "$part" in
25 vmlinux.bin.l7|kernel|linux)
26 echo "$part"
27 break
28 ;;
29 esac
30 done
31 }
32
33 platform_find_part_size() {
34 local first dev size erasesize name
35 while read dev size erasesize name; do
36 name=${name#'"'}; name=${name%'"'}
37 [ "$name" = "$1" ] && {
38 echo "$size"
39 break
40 }
41 done < /proc/mtd
42 }
43
44 platform_do_upgrade_combined() {
45 local partitions=$(platform_find_partitions)
46 local kernelpart=$(platform_find_kernelpart "${partitions#*:}")
47 local erase_size=$((0x${partitions%%:*})); partitions="${partitions#*:}"
48 local kern_part_size=0x$(platform_find_part_size "$kernelpart")
49 local kern_part_blocks=$(($kern_part_size / $CI_BLKSZ))
50 local kern_length=0x$(dd if="$1" bs=2 skip=1 count=4 2>/dev/null)
51 local kern_blocks=$(($kern_length / $CI_BLKSZ))
52 local root_blocks=$((0x$(dd if="$1" bs=2 skip=5 count=4 2>/dev/null) / $CI_BLKSZ))
53
54 v "platform_do_upgrade_combined"
55 v "partitions=$partitions"
56 v "kernelpart=$kernelpart"
57 v "kernel_part_size=$kern_part_size"
58 v "kernel_part_blocks=$kern_part_blocks"
59 v "kern_length=$kern_length"
60 v "erase_size=$erase_size"
61 v "kern_blocks=$kern_blocks"
62 v "root_blocks=$root_blocks"
63 v "kern_pad_blocks=$(($kern_part_blocks-$kern_blocks))"
64
65 if [ -n "$partitions" ] && [ -n "$kernelpart" ] && \
66 [ ${kern_blocks:-0} -gt 0 ] && \
67 [ ${root_blocks:-0} -gt 0 ] && \
68 [ ${erase_size:-0} -gt 0 ];
69 then
70 local append=""
71 [ -f "$UPGRADE_BACKUP" -a "$UPGRADE_OPT_SAVE_CONFIG" -eq 1 ] && append="-j $UPGRADE_BACKUP"
72
73 # write the kernel
74 dd if="$1" bs=$CI_BLKSZ skip=1 count=$kern_blocks 2>/dev/null | \
75 mtd -F$kernelpart:$kern_part_size:$CI_LDADR write - $kernelpart
76 # write the rootfs
77 dd if="$1" bs=$CI_BLKSZ skip=$((1+$kern_blocks)) count=$root_blocks 2>/dev/null | \
78 mtd $append write - rootfs
79 else
80 echo "invalid image"
81 fi
82 }
83
84 platform_check_image() {
85 local board=$(board_name)
86 local magic="$(get_magic_word "$1")"
87 local partitions=$(platform_find_partitions)
88 local kernelpart=$(platform_find_kernelpart "${partitions#*:}")
89 local kern_part_size=0x$(platform_find_part_size "$kernelpart")
90 local kern_length=0x$(dd if="$1" bs=2 skip=1 count=4 2>/dev/null)
91
92 [ "$#" -gt 1 ] && return 1
93
94 case "$board" in
95 avila | cambria )
96 [ "$magic" != "4349" ] && {
97 echo "Invalid image. Use *-sysupgrade.bin files on this board"
98 return 1
99 }
100
101 kern_length_b=$(printf '%d' $kern_length)
102 kern_part_size_b=$(printf '%d' $kern_part_size)
103 if [ $kern_length_b -gt $kern_part_size_b ]; then
104 echo "Invalid image. Kernel size ($kern_length) exceeds kernel partition ($kern_part_size)"
105 return 1
106 fi
107
108 local md5_img=$(dd if="$1" bs=2 skip=9 count=16 2>/dev/null)
109 local md5_chk=$(dd if="$1" bs=$CI_BLKSZ skip=1 2>/dev/null | md5sum -); md5_chk="${md5_chk%% *}"
110 if [ -n "$md5_img" -a -n "$md5_chk" ] && [ "$md5_img" = "$md5_chk" ]; then
111 return 0
112 else
113 echo "Invalid image. Contents do not match checksum (image:$md5_img calculated:$md5_chk)"
114 return 1
115 fi
116
117 return 0
118 ;;
119 esac
120
121 echo "Sysupgrade is not yet supported on $board."
122 return 1
123 }
124
125 platform_do_upgrade() {
126 local board=$(board_name)
127
128 v "board=$board"
129 case "$board" in
130 avila | cambria )
131 platform_do_upgrade_combined "$1"
132 ;;
133 *)
134 default_do_upgrade "$1"
135 ;;
136 esac
137 }