1 #!/bin/bash 2 ## Author: Guohua.Wu <Guohua.Wu@nufront.com> 3 ## Rev: v0.1, Mon Jun 11 16:47:20 CST 2018 4 ## Description: Shell script to install SD card with images. 5 6 ## Open it for debug 7 set -x 8 ## partition size in MB 9 BOOTLOAD_RESERVE=16 10 BOOT_ROM_SIZE=16 11 EXT4_FS_SIZE=512 12 ##FAT32=TOTAL - EXT 13 FAT32_FS_SIZE=1024 14 15 help() { 16 17 bn=`basename $0` 18 cat << EOF 19 usage $bn <option> device_node 20 21 options: 22 -h displays this help message 23 -s only get partition size 24 -np not partition. 25 -f flash image. 26 EOF 27 } 28 29 echo " check the if root?" 30 userid=`id -u` 31 if [ $userid -ne "0" ]; then 32 echo "you're not root?" 33 exit 34 fi 35 36 37 # parse command line 38 moreoptions=1 39 node="na" 40 cal_only=0 41 flash_images=0 42 not_partition=0 43 while [ "$moreoptions" = 1 -a $# -gt 0 ]; do 44 case $1 in 45 -h) help; exit ;; 46 -s) cal_only=1 ;; 47 -f) flash_images=1 ;; 48 -np) not_partition=1 ;; 49 *) moreoptions=0; node=$1 ;; 50 esac 51 [ "$moreoptions" = 0 ] && [ $# -gt 1 ] && help && exit 52 [ "$moreoptions" = 1 ] && shift 53 done 54 55 if [ ! -e ${node} ]; then 56 help 57 exit 58 fi 59 60 61 # call sfdisk to create partition table 62 # get total card size 63 seprate=0 64 total_size=`sfdisk -s ${node}` 65 total_size=`expr ${total_size} / 1024` 66 boot_rom_sizeb=`expr ${BOOT_ROM_SIZE} + ${BOOTLOAD_RESERVE}` 67 fat32_size=`expr ${total_size} - ${boot_rom_sizeb} - ${EXT4_FS_SIZE} + ${seprate}` 68 69 # create partitions 70 if [ "${cal_only}" -eq "1" ]; then 71 cat << EOF 72 BOOT : ${boot_rom_sizeb}MB 73 FAT32 : ${fat32_size}MB 74 EXT4 : ${EXT4_FS_SIZE}MB 75 EOF 76 exit 77 fi 78 79 function format_partitions 80 { 81 echo "formating partitions" 82 mkfs.vfat ${node}${part}1 -n"vfat" 83 mkfs.ext4 ${node}${part}2 -L"rootfs" 84 # mkdir /media/tmp 85 # mount ${node}${part}2 /media/tmp 86 # amount=$(df -k | grep ${node}${part}2 | awk '{print $2}') 87 # stag=$amount 88 # stag=$((stag-32)) 89 # kilo=K 90 # amountkilo=$stag$kilo 91 # sleep 1s 92 # umount /media/tmp 93 # rm -rf /media/tmp 94 # e2fsck -f ${node}${part}2 95 # resize2fs ${node}${part}2 $amountkilo 96 } 97 98 function flash_filesystem 99 { 100 if [ "${flash_images}" -eq "1" ]; then 101 #TODO:: unpack the ROM image 102 OUTDIR=/tmp/nufront-sdcard-provision 103 POSTFIX='.img' 104 [ -d ${OUTDIR} ] && rm -rf ${OUTDIR} 105 mkdir ${OUTDIR} 106 romfile=$(find ./ -name "*.ROM" | sort -r | head -n 1) 107 unzip ${romfile} -d ${OUTDIR} 108 if [ $? == '0' ]; then 109 POSTFIX='' 110 cd ${OUTDIR} 111 fi 112 113 echo "flashing images..." 114 dd if=u-boot-spl${POSTFIX} of=${node} bs=512 seek=4096 115 if [ ${POSTFIX}"FROMROM" != "FROMROM" ]; then 116 dd if=u-boot${POSTFIX} of=${node} bs=512 seek=4608 117 else 118 dd if=uboot${POSTFIX} of=${node} bs=512 seek=4608 119 fi 120 dd if=/dev/zero of=${node} bs=512 seek=1536 count=16 121 122 #dd if=rootfs.img of=${node}${part}2 123 echo "Unpack the rootfs" 124 mkdir /media/tmp 125 mount -t ext4 ${node}${part}2 /media/tmp 126 tar zxvf rootfs.tar.gz -C /media/tmp 127 sync 128 umount -f ${node}${part}2 129 rm -rf /media/tmp 130 131 echo "Install kernel" 132 mkdir /media/tmp 133 mount -t vfat ${node}${part}1 /media/tmp 134 cp -rf uImage /media/tmp 135 sync 136 umount -f ${node}${part}1 137 rm -rf /media/tmp 138 139 if [ ${POSTFIX}"FROMROM" == "FROMROM" ]; then 140 cd - 141 fi 142 rm -rf ${OUTDIR} 143 fi 144 } 145 146 #try to umount the filesystem 147 df | grep ${node} 148 if [ $? == 0 ]; then 149 echo "Umounting the nodes" 150 umount -f ${node}* 151 fi 152 #umount -f ${node}* 153 154 if [[ "${not_partition}" -eq "1" && "${flash_images}" -eq "1" ]] ; then 155 flash_filesystem 156 exit 157 fi 158 159 echo "Destroy the partition table" 160 dd if=/dev/zero of=${node} bs=1024 count=1 161 162 boot_start=0 163 fat32_start=`expr ${boot_start} + ${boot_rom_sizeb} * 2048` 164 ext4_start=`expr ${fat32_start} + ${fat32_size} * 2048` 165 166 echo "Making the Partition" 167 sfdisk --force ${node} << EOF 168 ${fat32_start},${fat32_size}M,83 169 ${ext4_start},${EXT4_FS_SIZE}M,83 170 EOF 171 172 echo "Format the partitions" 173 part="" 174 echo ${node} | grep mmcblk > /dev/null 175 if [ "$?" -eq "0" ]; then 176 part="p" 177 fi 178 179 format_partitions 180 flash_filesystem 181 182 sync 183 sleep 1 184 echo "Reload the partition table" 185 partprobe 186 echo "Finished!!!"