scripts: sd_fusing_rpi4: remove resize2fs command
[platform/kernel/u-boot.git] / scripts / tizen / sd_fusing_rpi4.sh
1 #!/bin/bash
2
3 declare FORMAT=""
4 declare DEVICE=""
5 declare -i OLD_DD=0
6
7 # Binaires array for fusing
8 declare -a FUSING_BINARY_ARRAY
9 declare -i FUSING_BINARY_NUM=0
10
11 declare CONV_ASCII=""
12 declare -i FUS_ENTRY_NUM=0
13 declare -i ab_option=0
14
15 # binary name | part number | bs | label | fs type
16 declare -a PART_TABLE=(
17         "boot.img"                      1       4M      boot_a                  vfat
18         "rootfs.img"                    2       4M      rootfs_a                ext4
19         "system-data.img"               3       4M      system-data             ext4
20         "user.img"                      5       4M      user                    ext4
21         "modules.img"                   6       4M      module_a                ext4
22         "ramdisk.img"                   7       4M      ramdisk_a               ext4
23         "ramdisk-recovery.img"          8       4M      ramdisk-recovery_a      ext4
24         "hal.img"                       10      4M      hal_a                   ext4
25         "boot.img"                      11      4M      boot_b                  vfat
26         "rootfs.img"                    12      4M      rootfs_b                ext4
27         "modules.img"                   13      4M      module_b                ext4
28         "ramdisk.img"                   14      4M      ramdisk_b               ext4
29         "ramdisk-recovery.img"          15      4M      ramdisk-recovery_b      ext4
30         "hal.img"                       16      4M      hal_b                   ext4
31         )
32 declare -a PART_TABLE_B=(
33         "boot.img"                      11      4M      boot_b                  vfat
34         "rootfs.img"                    12      4M      rootfs_b                ext4
35         "modules.img"                   13      4M      module_b                ext4
36         "ramdisk.img"                   14      4M      ramdisk_b               ext4
37         "ramdisk-recovery.img"          15      4M      ramdisk-recovery_b      ext4
38         "hal.img"                       16      4M      hal_b                   ext4
39         )
40
41 declare -r -i PART_TABLE_COL=5
42 declare -r -i PART_TABLE_ROW=${#PART_TABLE[*]}/${PART_TABLE_COL}
43 declare -r -i PART_TABLE_ROW_B=${#PART_TABLE_B[*]}/${PART_TABLE_COL}
44
45 # partition table support
46 function get_index_use_name () {
47         local -r binary_name=$1
48
49         for ((idx=0;idx<$PART_TABLE_ROW;idx++)); do
50                 if [ ${PART_TABLE[idx * ${PART_TABLE_COL} + 0]} == "$binary_name" ]; then
51                         return $idx
52                 fi
53         done
54
55         # return out of bound index
56         return $idx
57 }
58
59 # partition table support
60 function get_index_use_name_to_b () {
61         local -r binary_name=$1
62
63         for ((idx=0;idx<$PART_TABLE_ROW_B;idx++)); do
64                 if [ ${PART_TABLE_B[idx * ${PART_TABLE_COL} + 0]} == "$binary_name" ]; then
65                         return $idx
66                 fi
67         done
68
69         # return out of bound index
70         return $idx
71 }
72
73 function print_message () {
74         local color=$1
75         local message=$2
76
77         tput setaf $color
78         tput bold
79         echo ""
80         echo $message
81         tput sgr 0
82 }
83
84 function check_ddversion () {
85         # NOTE
86         # before coreutils dd 8.24, dd doesn't support "status=progress"
87         # and the option causes fusing failure. For backward compatibility,
88         # do not use the option for old dd
89         local version=(`dd --version | head -1 | grep -o '[0-9]\+'`)
90         local major=${version[0]}
91         local minor=${version[1]}
92
93         if [ $major -lt 8 ];  then
94                 OLD_DD=1
95         elif [ $major -eq 8 -a $minor -lt 24 ];  then
96                 OLD_DD=1
97         fi
98 }
99
100 function fusing_image () {
101         if [ $ab_option == 2 ] ; then
102                 echo "Skip to update Partitoin A"
103                 return
104         fi
105         local -r fusing_img=$1
106
107         # get binary info using basename
108         get_index_use_name $(basename "$fusing_img")
109         local -r -i part_idx=$?
110
111         if [ $part_idx -ne $PART_TABLE_ROW ];then
112                 local -r num=${PART_TABLE[${part_idx} * ${PART_TABLE_COL} + 1]}
113                 if [ "${num}" == "" ]; then
114                         local -r blktype=disk
115                 else
116                         local -r blktype=part
117                 fi
118                 local -r device=/dev/`lsblk ${DEVICE} -o TYPE,KNAME | awk "/^${blktype}.*[a-z]${num}\$/ { print \\\$2 }"`
119                 local -r bs=${PART_TABLE[${part_idx} * ${PART_TABLE_COL} + 2]}
120         else
121                 echo "Not supported binary: $fusing_img"
122                 return
123         fi
124
125         local -r input_size=`du -b $fusing_img | awk '{print $1}'`
126         local -r input_size_mb=`expr $input_size / 1024 / 1024`
127
128         print_message 2 "[Fusing $1 ($input_size_mb MiB)]"
129         if [ "$blktype" == "part" ]; then
130                 local MOUNT_PART=`mount | grep $device`
131                 if [ "$MOUNT_PART" != "" ]; then
132                         umount $device
133                 fi
134         fi
135         if [ $OLD_DD == 1 ]; then
136                 dd if=$fusing_img | pv -s $input_size | dd of=$device bs=$bs
137         else
138                 dd if=$fusing_img of=$device bs=$bs status=progress oflag=direct
139         fi
140 }
141
142 function fusing_image_to_b () {
143         if [ $ab_option == 0 ]  ; then
144                 echo "Skip to update Partitoin B"
145                 return
146         fi
147         local -r fusing_img=$1
148
149         # get binary info using basename
150         get_index_use_name_to_b $(basename "$fusing_img")
151
152         local -r -i part_idx=$?
153
154         if [ $part_idx -ne $PART_TABLE_ROW_B ];then
155                 local -r num=${PART_TABLE_B[${part_idx} * ${PART_TABLE_COL} + 1]}
156                 if [ "${num}" == "" ]; then
157                         local -r blktype=disk
158                 else
159                         local -r blktype=part
160                 fi
161                 local -r device=/dev/`lsblk ${DEVICE} -o TYPE,KNAME | awk "/^${blktype}.*[a-z]${num}\$/ { print \\\$2 }"`
162                 local -r bs=${PART_TABLE_B[${part_idx} * ${PART_TABLE_COL} + 2]}
163         else
164                 echo "Not supported binary: $fusing_img"
165                 return
166         fi
167
168         local -r input_size=`du -b $fusing_img | awk '{print $1}'`
169         local -r input_size_mb=`expr $input_size / 1024 / 1024`
170
171         print_message 2 "[Fusing $1 ($input_size_mb MiB)]"
172         if [ "$blktype" == "part" ]; then
173                 local MOUNT_PART=`mount | grep $device`
174                 if [ "$MOUNT_PART" != "" ]; then
175                         umount $device
176                 fi
177         fi
178         if [ $OLD_DD == 1 ]; then
179                 dd if=$fusing_img | pv -s $input_size | dd of=$device bs=$bs
180         else
181                 dd if=$fusing_img of=$device bs=$bs status=progress oflag=direct
182         fi
183 }
184
185 function fuse_image_tarball () {
186         local -r filepath=$1
187         local -r temp_dir="tar_tmp"
188
189         mkdir -p $temp_dir
190         tar xvf $filepath -C $temp_dir
191         cd $temp_dir
192
193         for file in *
194         do
195                 fusing_image $file
196                 fusing_image_to_b $file
197         done
198
199         cd ..
200         rm -rf $temp_dir
201         eval sync
202 }
203
204 function fuse_image () {
205
206         if [ "$FUSING_BINARY_NUM" == 0 ]; then
207                 return
208         fi
209
210         for ((fuse_idx = 0 ; fuse_idx < $FUSING_BINARY_NUM ; fuse_idx++))
211         do
212                 local filename=${FUSING_BINARY_ARRAY[fuse_idx]}
213
214                 case "$filename" in
215                     *.tar | *.tar.gz)
216                         fuse_image_tarball $filename
217                         ;;
218                     *)
219                         fusing_image $filename
220                         fusing_image_to_b $filename
221                         ;;
222                 esac
223         done
224         echo ""
225 }
226
227 # partition format
228 function mkpart_3 () {
229         # NOTE: if your sfdisk version is less than 2.26.0, then you should use following sfdisk command:
230         # sfdisk --in-order --Linux --unit M $DISK <<-__EOF__
231
232         # NOTE: sfdisk 2.26 doesn't support units other than sectors and marks --unit option as deprecated.
233         # The input data needs to contain multipliers (MiB) instead.
234         local version=(`sfdisk -v | grep -o '[0-9]\+'`)
235         local major=${version[0]}
236         local minor=${version[1]}
237         local sfdisk_new=0
238         local support_delete=0
239
240         if [ $major -gt 2 ];  then
241                 sfdisk_new=1
242                 if [ $major -eq 2 -a $minor -ge 28 ]; then
243                         support_delete=1
244                 fi
245         else
246                 if [ $major -eq 2 -a $minor -ge 26 ];  then
247                         sfdisk_new=1
248                 fi
249         fi
250
251         if [ $sfdisk_new == 0 ]; then
252                 echo "$(tput setaf 3)$(tput bold)NOTICE: Your sfdisk ${version[0]}.${version[1]}  version is too old. Update Latest sfdisk!"
253                 tput sgr 0
254                 exit -1
255         fi
256
257         local -r DISK=$DEVICE
258         local -r SIZE=`sfdisk -s $DISK`
259         local -r SIZE_MB=$((SIZE >> 10))
260
261         local -r BOOT_SZ=64
262         local -r ROOTFS_SZ=3072
263         local -r DATA_SZ=1344
264         local -r MODULE_SZ=32
265         local -r RAMDISK_SZ=32
266         local -r RAMDISK_RECOVERY_SZ=32
267         local -r INFORM_SZ=8
268         local -r HAL_SZ=256
269         local -r PARAM_SZ=4
270         local -r RESERVED1_SZ=64
271         local -r RESERVED2_SZ=125
272         local -r EXTEND_SZ=36
273
274         let "USER_SZ = $SIZE_MB - $BOOT_SZ * 2 - $ROOTFS_SZ * 2 - $DATA_SZ - $MODULE_SZ * 2 - $RAMDISK_SZ * 2 - $RAMDISK_RECOVERY_SZ * 2 - $INFORM_SZ - $EXTEND_SZ - $HAL_SZ * 2 - $RESERVED1_SZ - $RESERVED2_SZ - $PARAM_SZ"
275
276         local -r BOOT_A=${PART_TABLE[0 * ${PART_TABLE_COL} + 3]}
277         local -r ROOTFS_A=${PART_TABLE[1 * ${PART_TABLE_COL} + 3]}
278         local -r SYSTEMDATA=${PART_TABLE[2 * ${PART_TABLE_COL} + 3]}
279         local -r USER=${PART_TABLE[3 * ${PART_TABLE_COL} + 3]}
280         local -r MODULE_A=${PART_TABLE[4 * ${PART_TABLE_COL} + 3]}
281         local -r RAMDISK_A=${PART_TABLE[5 * ${PART_TABLE_COL} + 3]}
282         local -r RAMDISK_RECOVERY_A=${PART_TABLE[6 * ${PART_TABLE_COL} + 3]}
283         local -r INFORM=inform
284         local -r HAL_A=${PART_TABLE[7 * ${PART_TABLE_COL} + 3]}
285         local -r BOOT_B=${PART_TABLE[8 * ${PART_TABLE_COL} + 3]}
286         local -r ROOTFS_B=${PART_TABLE[9 * ${PART_TABLE_COL} + 3]}
287         local -r MODULE_B=${PART_TABLE[10 * ${PART_TABLE_COL} + 3]}
288         local -r RAMDISK_B=${PART_TABLE[11 * ${PART_TABLE_COL} + 3]}
289         local -r RAMDISK_RECOVERY_B=${PART_TABLE[12 * ${PART_TABLE_COL} + 3]}
290         local -r HAL_B=${PART_TABLE[13 * ${PART_TABLE_COL} + 3]}
291         local -r RESERVED0=reserved0
292         local -r RESERVED1=reserved1
293         local -r RESERVED2=reserved2
294
295         if [[ $USER_SZ -le 100 ]]
296         then
297                 echo "We recommend to use more than 8GB disk"
298                 exit 0
299         fi
300
301         echo "================================================"
302         echo "Label                     dev             size"
303         echo "================================================"
304         echo $BOOT_A"                   " $DISK"1       " $BOOT_SZ "MB"
305         echo $ROOTFS_A"                 " $DISK"2       " $ROOTFS_SZ "MB"
306         echo $SYSTEMDATA"               " $DISK"3       " $DATA_SZ "MB"
307         echo "[Extend]""                " $DISK"4"
308         echo " "$USER"                  " $DISK"5       " $USER_SZ "MB"
309         echo " "$MODULE_A"              " $DISK"6       " $MODULE_SZ "MB"
310         echo " "$RAMDISK_A"             " $DISK"7       " $RAMDISK_SZ "MB"
311         echo " "$RAMDISK_RECOVERY_A"    " $DISK"8       " $RAMDISK_RECOVERY_SZ "MB"
312         echo " "$INFORM"                " $DISK"9       " $INFORM_SZ "MB"
313         echo " "$HAL_A"                 " $DISK"10      " $HAL_SZ "MB"
314         echo " "$BOOT_B"                " $DISK"11      " $BOOT_SZ "MB"
315         echo " "$ROOTFS_B"              " $DISK"12      " $ROOTFS_SZ "MB"
316         echo " "$MODULE_B"              " $DISK"13      " $MODULE_SZ "MB"
317         echo " "$RAMDISK_B"             " $DISK"14      " $RAMDISK_SZ "MB"
318         echo " "$RAMDISK_RECOVERY_B"    " $DISK"15      " $RAMDISK_RECOVERY_SZ "MB"
319         echo " "$HAL_B"                 " $DISK"16      " $HAL_SZ "MB"
320         echo " "$RESERVED0"             " $DISK"17      " $PARAM_SZ "MB"
321         echo " "$RESERVED1"             " $DISK"18      " $RESERVED1_SZ "MB"
322         echo " "$RESERVED2"             " $DISK"19      " $RESERVED2_SZ "MB"
323
324         local MOUNT_LIST=`mount | grep $DISK | awk '{print $1}'`
325         for mnt in $MOUNT_LIST
326         do
327                 umount $mnt
328         done
329
330         echo "Remove partition table..."
331         dd if=/dev/zero of=$DISK bs=512 count=32 conv=notrunc
332
333         if [ $support_delete == 1 ]; then
334                 sfdisk --delete $DISK
335         fi
336
337         SCRIPT=""
338         for ((idx=0; idx < $PART_TABLE_ROW; idx++)); do
339                 NR=${PART_TABLE[idx * ${PART_TABLE_COL} + 1]}
340                 eval "PART_LABEL_NR_${NR}=${PART_TABLE[idx * ${PART_TABLE_COL} + 3]}"
341         done
342
343         sfdisk $DISK <<-__EOF__
344         label: gpt
345         ${DEVICE}1 : start=4MiB, size=${BOOT_SZ}MiB, type= C12A7328-F81F-11D2-BA4B-00A0C93EC93B, name=${PART_LABEL_NR_1}
346         ${DEVICE}2 : size=${ROOTFS_SZ}MiB, name=${PART_LABEL_NR_2}
347         ${DEVICE}3 : size=${DATA_SZ}MiB, name=${PART_LABEL_NR_3}
348         ${DEVICE}4 : size=${EXTEND_SZ}MiB, name=none
349         ${DEVICE}5 : size=${USER_SZ}MiB, name=${PART_LABEL_NR_5}
350         ${DEVICE}6 : size=${MODULE_SZ}MiB, name=${PART_LABEL_NR_6}
351         ${DEVICE}7 : size=${RAMDISK_SZ}MiB, name=${PART_LABEL_NR_7}
352         ${DEVICE}8 : size=${RAMDISK_RECOVERY_SZ}MiB, name=${PART_LABEL_NR_8}
353         ${DEVICE}9 : size=${INFORM_SZ}MiB, name=inform
354         ${DEVICE}10: size=${HAL_SZ}MiB, name=${PART_LABEL_NR_10}
355         ${DEVICE}11: size=${BOOT_SZ}MiB, type= C12A7328-F81F-11D2-BA4B-00A0C93EC93B, name=${PART_LABEL_NR_11}
356         ${DEVICE}12: size=${ROOTFS_SZ}MiB, name=${PART_LABEL_NR_12}
357         ${DEVICE}13: size=${MODULE_SZ}MiB, name=${PART_LABEL_NR_13}
358         ${DEVICE}14: size=${RAMDISK_SZ}MiB, name=${PART_LABEL_NR_14}
359         ${DEVICE}15: size=${RAMDISK_RECOVERY_SZ}MiB, name=${PART_LABEL_NR_15}
360         ${DEVICE}16: size=${HAL_SZ}MiB, name=${PART_LABEL_NR_16}
361         ${DEVICE}17: size=${PARAM_SZ}MiB, name=reserved0
362         ${DEVICE}18: size=${RESERVED1_SZ}MiB, name=reserved1
363         ${DEVICE}19: size=${RESERVED2_SZ}MiB, name=reserved2
364         __EOF__
365
366
367         for ((idx=0;idx<$PART_TABLE_ROW;idx++)); do
368                 local PART=/dev/`lsblk ${DISK} -o TYPE,KNAME | awk "/^part.*[a-z]${PART_TABLE[$idx * ${PART_TABLE_COL} + 1]}\$/ { print \\\$2 }"`
369                 if [ "${PART_TABLE[$idx * ${PART_TABLE_COL} + 4]}" == "vfat" ]; then
370                         mkfs.vfat -F 16 ${PART} -n ${PART_TABLE[$idx * ${PART_TABLE_COL} + 3]}
371                         if [ $? -eq 1 ]; then
372                                 echo "Failed to format as FAT filesystem"
373                                 exit -1
374                         fi
375                 elif [ "${PART_TABLE[$idx * ${PART_TABLE_COL} + 4]}" == "ext4" ]; then
376                         mkfs.ext4 -q ${PART} -L ${PART_TABLE[$idx * ${PART_TABLE_COL} + 3]} -F
377                 else
378                         echo "Skip to format for unknown filesystem type ${PART_TABLE[$idx * ${PART_TABLE_COL} + 4]} for part$idx, ${PART_TABLE[$idx * ${PART_TABLE_COL} + 3]}"
379                 fi
380         done
381
382         local -r PART9=/dev/`lsblk ${DISK} -o TYPE,KNAME | grep part | awk '{ print $2 }' | grep -G "[a-z]9\$"`
383         mkfs.ext4 -q ${PART9} -L $INFORM -F -O ^metadata_csum
384
385         # create "reboot-param.bin" file in inform partition for passing reboot parameter
386         # It should be done only once upon partition format.
387         if [ -d mnt_tmp ]; then
388                 echo "Remove the existing mnt_tmp directory!!"
389                 rm -rf mnt_tmp
390         fi
391         mkdir mnt_tmp
392         mount -t ext4 ${PART9} ./mnt_tmp
393         echo "norm" > ./mnt_tmp/reboot-param.bin
394         echo "norm" > ./mnt_tmp/reboot-param.info
395         echo "a" > ./mnt_tmp/partition-ab.info
396         echo "0" > ./mnt_tmp/partition-ab-cloned.info
397         echo "0" > ./mnt_tmp/upgrade-status.info
398
399         sync
400         umount ./mnt_tmp
401         rmdir mnt_tmp
402
403         local -r PART17=/dev/`lsblk ${DISK} -o TYPE,KNAME | grep part | awk '{ print $2 }' | grep -G "[a-z]17\$"`
404         mkfs.ext4 -q ${PART17} -L $RESERVED0 -F
405
406         local -r PART18=/dev/`lsblk ${DISK} -o TYPE,KNAME | grep part | awk '{ print $2 }' | grep -G "[a-z]18\$"`
407         mkfs.ext4 -q ${PART18} -L $RESERVED1 -F
408
409         local -r PART19=/dev/`lsblk ${DISK} -o TYPE,KNAME | grep part | awk '{ print $2 }' | grep -G "[a-z]19\$"`
410         mkfs.ext4 -q ${PART19} -L $RESERVED2 -F
411 }
412
413 function show_usage () {
414         echo "- Usage:"
415         echo "  sudo ./sd_fusing*.sh -d <device> [-b <path> <path> ..] [--format] [--update [b] ]"
416         echo "  -d  : device ndoe "
417         echo "  -b  : binary "
418         echo "  --update : If want to update Image about B Partition, use --update option with b"
419         echo "             Otherwise, it will be updated to both partition"
420 }
421
422 function check_partition_format () {
423         if [ "$FORMAT" != "2" ]; then
424                 echo "-----------------------"
425                 echo "Skip $DEVICE format"
426                 echo "-----------------------"
427                 return 0
428         fi
429
430         echo "-------------------------------"
431         echo "Start $DEVICE format"
432         echo ""
433         mkpart_3
434         echo "End $DEVICE format"
435         echo "-------------------------------"
436         echo ""
437 }
438
439 function check_args () {
440         if [ "$DEVICE" == "" ]; then
441                 echo -n "$(tput setaf 3)$(tput bold)$DEVICE will be formatted, Is it OK? [y/<n>] "
442                 show_usage
443                 tput sgr 0
444                 exit 0
445         fi
446
447         if [ "$DEVICE" != "" ]; then
448                 echo "Device: $DEVICE"
449         fi
450
451         if [ "$FUSING_BINARY_NUM" != 0 ]; then
452                 echo "Fusing binary: "
453                 for ((bid = 0 ; bid < $FUSING_BINARY_NUM ; bid++))
454                 do
455                         echo "  ${FUSING_BINARY_ARRAY[bid]}"
456                 done
457                 echo ""
458         fi
459
460         if [ "$FORMAT" == "1" ]; then
461                 echo ""
462                 echo "$(tput setaf 3)$(tput bold)$DEVICE will be formatted, Is it OK? [y/<n>]"
463                 tput sgr 0
464                 read input
465                 if [ "$input" == "y" ] || [ "$input" == "Y" ]; then
466                         FORMAT=2
467                 else
468                         FORMAT=0
469                 fi
470         fi
471 }
472
473 function check_device () {
474         if [ ! -b "$DEVICE" ]; then
475                 echo "No such device: $DEVICE"
476                 exit 0
477         fi
478
479         local REMOVABLE=`lsblk $DEVICE -nd -o RM | grep 1 | wc -l`
480         if [ "$REMOVABLE" == "0" ]; then
481                 echo ""
482                 echo "$(tput setaf 3)$(tput bold)$DEVICE is not a removable disk, Is it OK? [y/<n>]"
483                 tput sgr 0
484                 read input
485                 if [ "$input" != "y" ] && [ "$input" != "Y" ]; then
486                         exit 0
487                 fi
488         fi
489
490         if [ ! -w "$DEVICE" ]; then
491                 echo "Write not permitted: $DEVICE"
492                 exit 0
493         fi
494 }
495
496 function print_logo () {
497         echo ""
498         echo "Raspberry Pi4 downloader, version 1.0.6"
499         echo ""
500 }
501
502 print_logo
503
504 function add_fusing_binary() {
505         local declare binary_name=$1
506
507         if [ "$binary_name" != "" ]; then
508                 if [ -f "$binary_name" ]; then
509                         FUSING_BINARY_ARRAY[$FUSING_BINARY_NUM]=$binary_name
510
511                         FUSING_BINARY_NUM=$((FUSING_BINARY_NUM + 1))
512                 else
513                         echo "No such file: $binary_name"
514                 fi
515         fi
516 }
517
518
519 declare -i binary_option=0
520
521 while test $# -ne 0; do
522         option=$1
523         shift
524
525         case $option in
526         --f | --format)
527                 FORMAT="1"
528                 binary_option=0
529                 ;;
530         -d)
531                 DEVICE=$1
532                 binary_option=0
533                 shift
534                 ;;
535         -b)
536                 add_fusing_binary $1
537                 binary_option=1
538                 shift
539                 ;;
540         --update)
541                 if [ "$1" == "b" ] ; then
542                         ab_option=2
543                 else
544                         ab_option=1
545                 fi
546                 shift
547                 ;;
548         *)
549                 if [ $binary_option == 1 ];then
550                         add_fusing_binary $option
551                 else
552                         echo "Unkown command: $option"
553                         exit
554                 fi
555                 ;;
556         esac
557 done
558
559 check_args
560 check_device
561 check_partition_format
562 check_ddversion
563 fuse_image