mk_delta/common/bin/mk_delta.sh

   1 #!/bin/bash
   2
   3
   4 #==================
   5 # funtions
   6 #==================
   7
   8 #------------------------------------------------------------------------------
   9 # Function :
  10 #       fn_gen_update_cfg
  11 #
  12 # Description :
  13 #       read delta configuration file for generating update configuration file to be used during update.
  14 #       delta config file format : PART_NAME PART_BIN DELTA_BIN UPDATE_TYPE BLK_DEV BLK_OFFSET (deliminator : TAB)
  15
  16 fn_gen_update_cfg()
  17 {
  18         in_file=$1
  19         out_file=$2
  20
  21         if [ ! -r ${in_file} ]; then
  22                 echo "file ${in_file} cannot be read"
  23                 return 1
  24         fi
  25
  26         line_num=0
  27         while read line
  28         do
  29                 # skip comment line
  30                 if [ "${line:0:1}" = "#" ]; then
  31                         continue
  32                 fi
  33
  34                 # skip empty line
  35                 line_len=${#line}
  36                 if [ "${line_len}" = "0" ]; then
  37                         continue
  38                 fi
  39
  40
  41                 fields=(${line})
  42                 PART_NAME[${line_num}]=${fields[0]}
  43                 PART_BIN[${line_num}]=${fields[1]}
  44                 DELTA_BIN[${line_num}]=${fields[2]}
  45                 UPDATE_TYPE[${line_num}]=${fields[3]}
  46                 BLK_DEV[${line_num}]=${fields[4]}
  47                 BLK_OFFSET[${line_num}]=${fields[5]}
  48
  49                 line_num=$(($line_num + 1))
  50
  51         done < ${in_file}
  52
  53         PART_NUM=${line_num}
  54
  55         if [ -f ${out_file} ]; then
  56                 sudo rm -fr ${out_file}
  57         fi
  58
  59         for ((update_idx=0;update_idx<PART_NUM;update_idx++))
  60         do
  61                 f1=${PART_NAME[${update_idx}]}
  62                 f2=${DELTA_BIN[${update_idx}]}
  63                 f3=${UPDATE_TYPE[${update_idx}]}
  64                 f4=${BLK_DEV[${update_idx}]}
  65                 f5=${BLK_OFFSET[${update_idx}]}
  66
  67                 echo -e "$f1\t\t$f2\t\t$f3\t\t$f4\t\t$f5" >> ${out_file}
  68         done
  69
  70         return 0
  71 }
  72
  73 #------------------------------------------------------------------------------
  74 # Function :
  75 #       fn_set_default_params
  76 #
  77 # Description :
  78 #       set default params used globally in this script
  79 #
  80
  81 fn_set_default_params()
  82 {
  83         MY_ID=`whoami`
  84         MK_PART_DELTA=${COMMON_BINDIR}/mk_part_delta.sh
  85         DELTA_CFG_PATH=./cfg/delta.cfg
  86         UPDATE_CFG_PATH=./cfg/update.cfg
  87         DELTA_UA=./data/delta.ua
  88         UPDATE_MANAGER=./data/upgrade-trigger.sh
  89
  90         LOG_PATH=./data/Delta.log
  91         # Getting date and minor version
  92         MONDATE=$(date +%m%d)
  93         i=1
  94         while [ -d "result/${MONDATE}_${i}" ]
  95         do
  96                 i=$(($i + 1))
  97         done
  98         MONDATE="${MONDATE}_${i}"
  99         echo "MONDATE = ${MONDATE}"
 100
 101         if [ "z${OLD_TAR_DIR}" = "z" ]; then
 102                 export OLD_TAR_DIR="old_tar"
 103         fi
 104         if [ "z${NEW_TAR_DIR}" = "z" ]; then
 105                 export NEW_TAR_DIR="new_tar"
 106         fi
 107 }
 108
 109 #------------------------------------------------------------------------------
 110 # Function :
 111 #       fn_swap_binary_files
 112 #
 113 # Description :
 114 #       change old & new for making backward delta
 115 #
 116
 117 fn_swap_binary_files()
 118 {
 119         TEMP_TAR_DIR=${OLD_TAR_DIR}
 120         export OLD_TAR_DIR=${NEW_TAR_DIR}
 121         export NEW_TAR_DIR=${TEMP_TAR_DIR}
 122 }
 123
 124 #------------------------------------------------------------------------------
 125 # Function :
 126 #       fn_print_update_cfg
 127
 128 fn_print_update_cfg()
 129 {
 130         for ((update_idx=0;update_idx<PART_NUM;update_idx++))
 131         do
 132                         echo "------------------------------"
 133                         echo "[PART_NAME]    ${PART_NAME[${update_idx}]}"
 134                         echo "[DELTA_BIN]    ${DELTA_BIN[${update_idx}]}"
 135                         echo "[UPDATE_TYPE]  ${UPDATE_TYPE[${update_idx}]}"
 136                         echo "[BLK_DEV]      ${BLK_DEV[${update_idx}]}"
 137                         echo "[BLK_OFFSET]   ${BLK_OFFSET[${update_idx}]}"
 138
 139         done
 140         echo "------------------------------"
 141 }
 142
 143 #------------------------------------------------------------------------------
 144 # Function :
 145 #       fn_process_extra_delta
 146 #
 147
 148 fn_process_extra_delta()
 149 {
 150         RAMDISK2_FULL_IMG=0
 151         if [ "${part_name}" = "RAMDISK2" ]; then
 152                 if [ "${RAMDISK2_FULL_IMG}" = "0" ]; then
 153                         #--- delta recovery partition ---
 154                         ${MK_PART_DELTA} ${part_name} DELTA_IMG ${part_bin} ${delta_bin}
 155                         if [ "$?" = "0" ]; then
 156                                 sudo mv data/${PART_OUT} ${RESULT_DIR}/${part_bin}
 157                                 sudo mv ${RESULT_DIR}/${part_bin}/${delta_bin} ${DELTA_DIR}
 158                         fi
 159                 fi
 160                 if [ "${RAMDISK2_FULL_IMG}" = "1" ]; then
 161                         #--- full recovery partition ---
 162                         ${MK_PART_DELTA} ${part_name} FULL_IMG ${part_bin} ${delta_bin}
 163                         if [ "$?" = "0" ]; then
 164                                 sudo mv data/${PART_OUT} ${RESULT_DIR}/${part_bin}
 165                                 sudo mv ${RESULT_DIR}/${part_bin}/${part_bin} ${DELTA_DIR}
 166                         fi
 167                 fi
 168         elif [ "${part_name}" = "RECOVERY" ]; then
 169                 #--- recovery kernel ---
 170                 sudo echo "1" > ${DELTA_DIR}/${delta_bin}
 171         elif [ "${part_name}" = "EXTRA" ]; then
 172                 echo "---"
 173         fi
 174 }
 175
 176 #------------------------------------------------------------------------------
 177 # Function :
 178 #       fn_get_tar_file_names
 179 #
 180 # Description :
 181 #       find tar file name (without path) that contains given partition binary
 182 #
 183
 184 fn_get_tar_file_names()
 185 {
 186         BIN_TO_FIND=$1
 187
 188         cd ${DATA_DIR}/${OLD_TAR_DIR}
 189         OLD_TAR_LIST=`ls *.tar *.tar.gz *.tgz *.tar.md5 2>/dev/null`
 190         cd ${CUR_DIR}
 191
 192         cd ${DATA_DIR}/${NEW_TAR_DIR}
 193         NEW_TAR_LIST=`ls *.tar *.tar.gz *.tgz *.tar.md5 2>/dev/null`
 194         cd ${CUR_DIR}
 195
 196         #--- find which tar contains the binary ---
 197         for tar_file in ${OLD_TAR_LIST}
 198         do
 199                 result=`tar -tvf ${DATA_DIR}/${OLD_TAR_DIR}/${tar_file} | grep " ${BIN_TO_FIND}$"`
 200                 if [ "$?" = "0" ]; then
 201                         OLD_TAR_FILE=${tar_file}
 202                         break
 203                 fi
 204         done
 205
 206         for tar_file in ${NEW_TAR_LIST}
 207         do
 208                 result=`tar -tvf ${DATA_DIR}/${NEW_TAR_DIR}/${tar_file} | grep " ${BIN_TO_FIND}$"`
 209                 if [ "$?" = "0" ]; then
 210                         NEW_TAR_FILE=${tar_file}
 211                         break
 212                 fi
 213         done
 214
 215 }
 216
 217 #------------------------------------------------------------------------------
 218 # Function :
 219 #       fn_extract_from_image
 220 #
 221 # Description :
 222 #       extract some files from given image
 223 #
 224
 225 fn_extract_from_image()
 226 {
 227         CUR_DIR=`pwd`
 228         DATA_DIR=./data
 229         IMAGE=$1
 230         TARGET_FILE=$2
 231         MNT_PNT=TMP_MNT
 232         mkdir -p ${MNT_PNT}
 233
 234         fn_get_tar_file_names $IMAGE
 235         tar xvf ${DATA_DIR}/${NEW_TAR_DIR}/${NEW_TAR_FILE} ${IMAGE}
 236         if [ "$?" != "0" ]; then
 237                 return 1;
 238         fi
 239
 240         sudo mount -t ext4 -o loop ${IMAGE} ${MNT_PNT}
 241         if [ "$?" != "0" ]; then
 242                 return 1;
 243         fi
 244
 245         EXTRACT_FILE="${MNT_PNT}${TARGET_FILE}"
 246         if [ -e ${EXTRACT_FILE} ]; then
 247                 sudo cp ${EXTRACT_FILE} ${DELTA_DIR}
 248         else
 249                 echo "There is no ${TARGET_FILE} in ${IMAGE}"
 250         fi
 251
 252         sudo umount ${MNT_PNT}
 253         sudo rm -rf ${MNT_PNT}
 254         sudo rm -f ${IMAGE}
 255 }
 256
 257 ###############################################################################
 258 #==================
 259 # Main Start
 260 #==================
 261
 262 COMMON_BINDIR=${PWD}/../common/bin
 263
 264 fn_set_default_params
 265
 266 fn_gen_update_cfg ${DELTA_CFG_PATH} ${UPDATE_CFG_PATH}
 267
 268 ################### Make delta Forward and Backward #####################
 269 if [ "z${FW_BW}" = "z" ]; then
 270         #FW_BW="FW_DELTA BW_DELTA"
 271         FW_BW="FW_DELTA"
 272 fi
 273
 274 START_TIMESTAMP="$(date +%T)"
 275
 276 for Make_Delta in $FW_BW
 277 do
 278         #--- archive result directory ---
 279         echo "Make $Make_Delta Delta file"
 280         RESULT_DIR=result/$MONDATE/$Make_Delta
 281         DELTA_DIR=${RESULT_DIR}/DELTA
 282         mkdir -p ${DELTA_DIR}
 283
 284         if [ "$Make_Delta" = "BW_DELTA" ]; then
 285                 fn_swap_binary_files
 286         fi
 287
 288
 289         #--- make delta for each partition ---
 290         for ((part_num=0;part_num<PART_NUM;part_num++))
 291         do
 292                 part_name=${PART_NAME[${part_num}]}
 293                 update_type=${UPDATE_TYPE[${part_num}]}
 294                 part_bin=${PART_BIN[${part_num}]}
 295                 delta_bin=${DELTA_BIN[${part_num}]}
 296
 297                 PART_OUT=${part_name}_OUT
 298
 299                 if [ "${update_type}" = "EXTRA" ]; then
 300                         fn_process_extra_delta
 301                 else
 302                         ${MK_PART_DELTA} ${part_name} ${update_type} ${part_bin} ${delta_bin}
 303                         if [ "$?" = "0" ]; then
 304                                 if [ "${update_type}" = "DELTA_FS" ]; then
 305                                         sudo mv data/${PART_OUT} ${RESULT_DIR}/${part_bin}
 306                                         sudo mv ${RESULT_DIR}/${part_bin} ${DELTA_DIR}
 307                                 else
 308                                         sudo mv data/${PART_OUT} ${RESULT_DIR}/${part_bin}
 309                                         sudo mv ${RESULT_DIR}/${part_bin}/${delta_bin} ${DELTA_DIR}
 310                                 fi
 311                         else
 312                                 sudo rm -rf ${RESULT_DIR}/${part_bin}
 313                                 echo "Error: Abort Delta Generation"
 314                                 exit 1
 315                         fi
 316                 fi
 317                 sudo rm -rf ${RESULT_DIR}/${part_bin}
 318         done
 319
 320         #move update.cfg to delta directory
 321         if [ -r ${UPDATE_CFG_PATH} ]; then
 322                 sudo cp ${UPDATE_CFG_PATH} ${DELTA_DIR}/update.cfg
 323                 sudo rm ${UPDATE_CFG_PATH}
 324         fi
 325         if [ -r ${LOG_PATH} ]; then
 326                 sudo cp ${LOG_PATH} ${RESULT_DIR}/Delta.log
 327                 sudo rm ${LOG_PATH}
 328         fi
 329
 330         #--- extract files which would be appended to delta.tar ---
 331         echo "Extract binaries for update from images"
 332         fn_extract_from_image rootfs.img /usr/bin/upgrade-trigger.sh
 333         fn_extract_from_image ramdisk-recovery.img /usr/bin/delta.ua
 334
 335         #--- archive result directory ---
 336         cd result/$MONDATE
 337         echo "tar result directory"
 338         sudo tar cf ${MONDATE}_${Make_Delta}.tar $Make_Delta
 339         cd ../..
 340
 341         echo "Making delta binary is End!! ($Make_Delta)"
 342 done
 343
 344 if [ -r ${UPDATE_CFG_PATH} ]; then
 345         sudo rm ${UPDATE_CFG_PATH}
 346 fi
 347 cd ${DELTA_DIR}
 348 sudo cp ${COMMON_BINDIR}/unpack.sh ./
 349 sudo tar --overwrite -cf ../delta.tar *
 350 cd -
 351
 352 END_TIMESTAMP="$(date +%T)"
 353
 354 echo "OVERALL TIMESTAMP : [START] ${START_TIMESTAMP} ~ [END] ${END_TIMESTAMP}"
 355