1#!/bin/bash 2# 3# Copyright (c) 2020 Fuzhou Rockchip Electronics Co., Ltd 4# 5# SPDX-License-Identifier: GPL-2.0 6# 7set -e 8 9FIT_DIR="fit" 10IMG_UBOOT="uboot.img" 11IMG_BOOT="boot.img" 12IMG_RECOVERY="recovery.img" 13ITB_UBOOT="${FIT_DIR}/uboot.itb" 14ITB_BOOT="${FIT_DIR}/boot.itb" 15ITB_RECOVERY="${FIT_DIR}/recovery.itb" 16SIG_BIN="data2sign.bin" 17SIG_UBOOT="${FIT_DIR}/uboot.data2sign" 18SIG_BOOT="${FIT_DIR}/boot.data2sign" 19SIG_RECOVERY="${FIT_DIR}/recovery.data2sign" 20# offs 21OFFS_DATA="0xE00" 22# file 23CHIP_FILE="arch/arm/lib/.asm-offsets.s.cmd" 24# placeholder address 25FDT_ADDR_PLACEHOLDER="0xffffff00" 26KERNEL_ADDR_PLACEHOLDER="0xffffff01" 27RAMDISK_ADDR_PLACEHOLDER="0xffffff02" 28# tools 29MKIMAGE="./tools/mkimage" 30RK_SIGN_TOOL="../rkbin/tools/rk_sign_tool" 31FIT_UNPACK="./scripts/fit-unpack.sh" 32CHECK_SIGN="./tools/fit_check_sign" 33# key 34KEY_DIR="keys/" 35RSA_PRI_KEY="keys/dev.key" 36RSA_PUB_KEY="keys/dev.crt" 37SIGNATURE_KEY_NODE="/signature/key-dev" 38SPL_DTB="spl/u-boot-spl.dtb" 39UBOOT_DTB="u-boot.dtb" 40# its 41ITS_UBOOT="u-boot.its" 42ITS_BOOT="boot.its" 43ITS_RECOVERY="recovery.its" 44ARG_VER_UBOOT="0" 45ARG_VER_BOOT="0" 46ARG_VER_RECOVERY="0" 47 48function help() 49{ 50 echo 51 echo "usage:" 52 echo " $0 [args]" 53 echo 54 echo "args:" 55 echo " --rollback-index-recovery <decimal integer>" 56 echo " --rollback-index-boot <decimal integer>" 57 echo " --rollback-index-uboot <decimal integer>" 58 echo " --version-recovery <decimal integer>" 59 echo " --version-boot <decimal integer>" 60 echo " --version-uboot <decimal integer>" 61 echo " --boot_img <boot image>" 62 echo " --recovery_img <recovery image>" 63 echo " --args <arg>" 64 echo " --ini-loader <loader ini file>" 65 echo " --ini-trust <trust ini file>" 66 echo " --no-check" 67 echo " --spl-new" 68 echo 69} 70 71function arg_check_decimal() 72{ 73 if [ -z $1 ]; then 74 help 75 exit 1 76 fi 77 78 decimal=`echo $1 |sed 's/[0-9]//g'` 79 if [ ! -z ${decimal} ]; then 80 echo "ERROR: $1 is not decimal integer" 81 help 82 exit 1 83 fi 84} 85 86function check_its() 87{ 88 cat $1 | while read line 89 do 90 file=`echo ${line} | sed -n "/incbin/p" | awk -F '"' '{ printf $2 }' | tr -d ' '` 91 if [ ! -f ${file} ]; then 92 echo "ERROR: No ${file}" 93 exit 1 94 fi 95 done 96} 97 98function validate_arg() 99{ 100 case $1 in 101 --no-check|--spl-new|--burn-key-hash) 102 shift=1 103 ;; 104 --ini-trust|--ini-loader|--rollback-index-boot|--rollback-index-recovery|--rollback-index-uboot|--boot_img|--recovery_img|--version-uboot|--version-boot|--version-recovery|--chip) 105 shift=2 106 ;; 107 *) 108 shift=0 109 ;; 110 esac 111 echo ${shift} 112} 113 114function fit_process_args() 115{ 116 if [ $# -eq 0 ]; then 117 help 118 exit 0 119 fi 120 121 while [ $# -gt 0 ]; do 122 case $1 in 123 --args) 124 ARG_VALIDATE=$2 125 shift 2 126 ;; 127 --boot_img) # boot.img 128 ARG_BOOT_IMG=$2 129 shift 2 130 ;; 131 --chip) 132 ARG_CHIP=$2 133 shift 2 134 ;; 135 --recovery_img) # recovery.img 136 ARG_RECOVERY_IMG=$2 137 shift 2 138 ;; 139 --boot_img_dir) # boot.img components directory 140 ARG_BOOT_IMG_DIR=$2 141 shift 2 142 ;; 143 --no-check) # No hostcc fit signature check 144 ARG_NO_CHECK="y" 145 shift 1 146 ;; 147 --ini-trust) # Assign trust ini file 148 ARG_INI_TRUST=$2 149 shift 2 150 ;; 151 --ini-loader) # Assign loader ini file 152 ARG_INI_LOADER=$2 153 shift 2 154 ;; 155 --spl-new) # Use current build u-boot-spl.bin to pack loader 156 ARG_SPL_NEW="y" 157 shift 1 158 ;; 159 --rollback-index-boot) 160 ARG_ROLLBACK_IDX_BOOT=$2 161 arg_check_decimal $2 162 shift 2 163 ;; 164 --rollback-index-recovery) 165 ARG_ROLLBACK_IDX_RECOVERY=$2 166 arg_check_decimal $2 167 shift 2 168 ;; 169 --rollback-index-uboot) 170 ARG_ROLLBACK_IDX_UBOOT=$2 171 arg_check_decimal $2 172 shift 2 173 ;; 174 --version-uboot) 175 ARG_VER_UBOOT=$2 176 arg_check_decimal $2 177 shift 2 178 ;; 179 --version-boot) 180 ARG_VER_BOOT=$2 181 arg_check_decimal $2 182 shift 2 183 ;; 184 --version-recovery) 185 ARG_VER_RECOVERY=$2 186 arg_check_decimal $2 187 shift 2 188 ;; 189 --burn-key-hash) 190 ARG_BURN_KEY_HASH="y" 191 shift 1 192 ;; 193 *) 194 help 195 exit 1 196 ;; 197 esac 198 done 199 200 if grep -q '^CONFIG_FIT_SIGNATURE=y' .config ; then 201 ARG_SIGN="y" 202 fi 203} 204 205function fit_raw_compile() 206{ 207 # Verified-boot: should rebuild code but don't need to repack images. 208 if [ "${ARG_SIGN}" == "y" ]; then 209 ./make.sh --raw-compile 210 fi 211 rm ${FIT_DIR} -rf && mkdir -p ${FIT_DIR} 212} 213 214function fit_gen_uboot_itb() 215{ 216 # generate u-boot.its file 217 ./make.sh itb ${ARG_INI_TRUST} 218 219 # check existance of file in its 220 check_its ${ITS_UBOOT} 221 222 if [ "${ARG_SIGN}" != "y" ]; then 223 ${MKIMAGE} -f ${ITS_UBOOT} -E -p ${OFFS_DATA} ${ITB_UBOOT} -v ${ARG_VER_UBOOT} 224 if [ "${ARG_SPL_NEW}" == "y" ]; then 225 ./make.sh --spl ${ARG_INI_LOADER} 226 echo "pack loader with new: spl/u-boot-spl.bin" 227 else 228 ./make.sh loader ${ARG_INI_LOADER} 229 fi 230 else 231 if [ ! -f ${RSA_PRI_KEY} ]; then 232 echo "ERROR: No ${RSA_PRI_KEY} " 233 exit 1 234 elif [ ! -f ${RSA_PUB_KEY} ]; then 235 echo "ERROR: No ${RSA_PUB_KEY} " 236 exit 1 237 fi 238 239 if ! grep -q '^CONFIG_SPL_FIT_SIGNATURE=y' .config ; then 240 echo "ERROR: CONFIG_SPL_FIT_SIGNATURE is disabled" 241 exit 1 242 fi 243 244 # rollback-index 245 if grep -q '^CONFIG_SPL_FIT_ROLLBACK_PROTECT=y' .config ; then 246 ARG_SPL_ROLLBACK_PROTECT="y" 247 if [ -z ${ARG_ROLLBACK_IDX_UBOOT} ]; then 248 echo "ERROR: No arg \"--rollback-index-uboot <n>\"" 249 exit 1 250 fi 251 fi 252 253 if [ "${ARG_SPL_ROLLBACK_PROTECT}" == "y" ]; then 254 VERSION=`grep 'rollback-index' ${ITS_UBOOT} | awk -F '=' '{ printf $2 }' | tr -d ' '` 255 sed -i "s/rollback-index = ${VERSION}/rollback-index = <${ARG_ROLLBACK_IDX_UBOOT}>;/g" ${ITS_UBOOT} 256 fi 257 258 # Generally, boot.img is signed before uboot.img, so the ras key can be found 259 # in u-boot.dtb. If not found, let's insert rsa key anyway. 260 if ! fdtget -l ${UBOOT_DTB} /signature >/dev/null 2>&1 ; then 261 ${MKIMAGE} -f ${ITS_UBOOT} -k ${KEY_DIR} -K ${UBOOT_DTB} -E -p ${OFFS_DATA} -r ${ITB_UBOOT} -v ${ARG_VER_UBOOT} 262 echo "## Adding RSA public key into ${UBOOT_DTB}" 263 fi 264 265 # Pack 266 ${MKIMAGE} -f ${ITS_UBOOT} -k ${KEY_DIR} -K ${SPL_DTB} -E -p ${OFFS_DATA} -r ${ITB_UBOOT} -v ${ARG_VER_UBOOT} 267 mv ${SIG_BIN} ${SIG_UBOOT} 268 269 # burn-key-hash 270 if [ "${ARG_BURN_KEY_HASH}" == "y" ]; then 271 fdtput -tx ${SPL_DTB} ${SIGNATURE_KEY_NODE} burn-key-hash 0x1 272 fi 273 274 # rollback-index read back check 275 if [ "${ARG_SPL_ROLLBACK_PROTECT}" == "y" ]; then 276 VERSION=`fdtget -ti ${ITB_UBOOT} /configurations/conf rollback-index` 277 if [ "${VERSION}" != "${ARG_ROLLBACK_IDX_UBOOT}" ]; then 278 echo "ERROR: Failed to set rollback-index for ${ITB_UBOOT}"; 279 exit 1 280 fi 281 fi 282 283 # burn-key-hash read back check 284 if [ "${ARG_BURN_KEY_HASH}" == "y" ]; then 285 if [ "`fdtget -ti ${SPL_DTB} ${SIGNATURE_KEY_NODE} burn-key-hash`" != "1" ]; then 286 echo "ERROR: Failed to set burn-key-hash for ${SPL_DTB}"; 287 exit 1 288 fi 289 fi 290 291 # host check signature 292 if [ "${ARG_NO_CHECK}" != "y" ]; then 293 if [ "${ARG_SPL_NEW}" == "y" ]; then 294 ${CHECK_SIGN} -f ${ITB_UBOOT} -k ${SPL_DTB} -s 295 else 296 spl_file="../rkbin/"`sed -n "/FlashBoot=/s/FlashBoot=//p" ${ARG_INI_LOADER} |tr -d '\r'` 297 offs=`fdtdump -s ${spl_file} | head -1 | awk -F ":" '{ print $2 }' | sed "s/ found fdt at offset //g" | tr -d " "` 298 if [ -z ${offs} ]; then 299 echo "ERROR: invalid ${spl_file} , unable to find fdt blob" 300 fi 301 offs=`printf %d ${offs} ` # hex -> dec 302 dd if=${spl_file} of=spl/u-boot-spl-old.dtb bs=${offs} skip=1 >/dev/null 2>&1 303 ${CHECK_SIGN} -f ${ITB_UBOOT} -k spl/u-boot-spl-old.dtb -s 304 fi 305 fi 306 307 # minimize u-boot-spl.dtb: clear as 0 but not remove property. 308 if grep -q '^CONFIG_SPL_FIT_HW_CRYPTO=y' .config ; then 309 fdtput -tx ${SPL_DTB} ${SIGNATURE_KEY_NODE} rsa,r-squared 0x0 310 if grep -q '^CONFIG_SPL_ROCKCHIP_CRYPTO_V1=y' .config ; then 311 fdtput -tx ${SPL_DTB} ${SIGNATURE_KEY_NODE} rsa,np 0x0 312 fdtput -r ${SPL_DTB} ${SIGNATURE_KEY_NODE}/hash@np 313 else 314 fdtput -tx ${SPL_DTB} ${SIGNATURE_KEY_NODE} rsa,c 0x0 315 fdtput -r ${SPL_DTB} ${SIGNATURE_KEY_NODE}/hash@c 316 fi 317 else 318 fdtput -tx ${SPL_DTB} ${SIGNATURE_KEY_NODE} rsa,c 0x0 319 fdtput -tx ${SPL_DTB} ${SIGNATURE_KEY_NODE} rsa,np 0x0 320 fdtput -tx ${SPL_DTB} ${SIGNATURE_KEY_NODE} rsa,exponent-BN 0x0 321 fdtput -r ${SPL_DTB} ${SIGNATURE_KEY_NODE}/hash@c 322 fdtput -r ${SPL_DTB} ${SIGNATURE_KEY_NODE}/hash@np 323 fi 324 325 # repack spl 326 rm -f *_loader_*.bin 327 if [ "${ARG_SPL_NEW}" == "y" ]; then 328 cat spl/u-boot-spl-nodtb.bin > spl/u-boot-spl.bin 329 if ! grep -q '^CONFIG_SPL_SEPARATE_BSS=y' .config ; then 330 cat spl/u-boot-spl-pad.bin >> spl/u-boot-spl.bin 331 fi 332 cat ${SPL_DTB} >> spl/u-boot-spl.bin 333 334 ./make.sh --spl ${ARG_INI_LOADER} 335 echo "## pack loader with new: spl/u-boot-spl.bin" 336 else 337 ./make.sh loader ${ARG_INI_LOADER} 338 fi 339 340 if [ "${ARG_BURN_KEY_HASH}" == "y" ]; then 341 echo "## ${SPL_DTB}: burn-key-hash=1" 342 fi 343 fi 344 345 rm -f u-boot.itb u-boot.img u-boot-dtb.img 346 mv ${ITS_UBOOT} ${FIT_DIR} 347} 348 349function fit_gen_boot_itb() 350{ 351 if [ ! -z ${ARG_BOOT_IMG} ]; then 352 ${FIT_UNPACK} -f ${ARG_BOOT_IMG} -o ${FIT_DIR}/unpack 353 ITS_BOOT="${FIT_DIR}/unpack/image.its" 354 else 355 compression=`awk -F"," '/COMPRESSION=/ { printf $1 }' ${ARG_INI_TRUST} | tr -d ' ' | cut -c 13-` 356 if [ -z "${compression}" ]; then 357 compression="none" 358 fi 359 ./arch/arm/mach-rockchip/make_fit_boot.sh -c ${compression} > ${ITS_BOOT} 360 check_its ${ITS_BOOT} 361 fi 362 363 if [ "${ARG_SIGN}" != "y" ]; then 364 ${MKIMAGE} -f ${ITS_BOOT} -E -p ${OFFS_DATA} ${ITB_BOOT} -v ${ARG_VER_BOOT} 365 else 366 if [ ! -f ${RSA_PRI_KEY} ]; then 367 echo "ERROR: No ${RSA_PRI_KEY}" 368 exit 1 369 elif [ ! -f ${RSA_PUB_KEY} ]; then 370 echo "ERROR: No ${RSA_PUB_KEY}" 371 exit 1 372 fi 373 374 if ! grep -q '^CONFIG_FIT_SIGNATURE=y' .config ; then 375 echo "ERROR: CONFIG_FIT_SIGNATURE is disabled" 376 exit 1 377 fi 378 379 if grep -q '^CONFIG_FIT_ROLLBACK_PROTECT=y' .config ; then 380 ARG_ROLLBACK_PROTECT="y" 381 if [ -z ${ARG_ROLLBACK_IDX_BOOT} ]; then 382 echo "ERROR: No arg \"--rollback-index-boot <n>\"" 383 exit 1 384 fi 385 fi 386 387 # fixup 388 COMMON_FILE=`sed -n "/_common.h/p" ${CHIP_FILE} | awk '{ print $1 }'` 389 FDT_ADDR_R=`awk /fdt_addr_r/ ${COMMON_FILE} | awk -F '=' '{ print $2 }' | awk -F '\\' '{ print $1 }'` 390 KERNEL_ADDR_R=`awk /kernel_addr_r/ ${COMMON_FILE} | awk -F '=' '{ print $2 }' | awk -F '\\' '{ print $1 }'` 391 RMADISK_ADDR_R=`awk /ramdisk_addr_r/ ${COMMON_FILE} | awk -F '=' '{ print $2 }' | awk -F '\\' '{ print $1 }'` 392 sed -i "s/${FDT_ADDR_PLACEHOLDER}/${FDT_ADDR_R}/g" ${ITS_BOOT} 393 sed -i "s/${KERNEL_ADDR_PLACEHOLDER}/${KERNEL_ADDR_R}/g" ${ITS_BOOT} 394 sed -i "s/${RAMDISK_ADDR_PLACEHOLDER}/${RMADISK_ADDR_R}/g" ${ITS_BOOT} 395 if grep -q '^CONFIG_ARM64=y' .config ; then 396 sed -i 's/arch = "arm";/arch = "arm64";/g' ${ITS_BOOT} 397 fi 398 399 if [ "${ARG_ROLLBACK_PROTECT}" == "y" ]; then 400 VERSION=`grep 'rollback-index' ${ITS_BOOT} | awk -F '=' '{ printf $2 }' | tr -d ' '` 401 sed -i "s/rollback-index = ${VERSION}/rollback-index = <${ARG_ROLLBACK_IDX_BOOT}>;/g" ${ITS_BOOT} 402 fi 403 404 ${MKIMAGE} -f ${ITS_BOOT} -k ${KEY_DIR} -K ${UBOOT_DTB} -E -p ${OFFS_DATA} -r ${ITB_BOOT} -v ${ARG_VER_BOOT} 405 mv ${SIG_BIN} ${SIG_BOOT} 406 407 # rollback-index read back check 408 if [ "${ARG_ROLLBACK_PROTECT}" == "y" ]; then 409 VERSION=`fdtget -ti ${ITB_BOOT} /configurations/conf rollback-index` 410 if [ "${VERSION}" != "${ARG_ROLLBACK_IDX_BOOT}" ]; then 411 echo "ERROR: Failed to set rollback-index for ${ITB_BOOT}"; 412 exit 1 413 fi 414 fi 415 416 # host check signature 417 if [ "${ARG_NO_CHECK}" != "y" ]; then 418 ${CHECK_SIGN} -f ${ITB_BOOT} -k ${UBOOT_DTB} 419 fi 420 421 # minimize u-boot.dtb: clearn as 0 but not remove property. 422 if grep -q '^CONFIG_FIT_HW_CRYPTO=y' .config ; then 423 fdtput -tx ${UBOOT_DTB} ${SIGNATURE_KEY_NODE} rsa,r-squared 0x0 424 if grep -q '^CONFIG_ROCKCHIP_CRYPTO_V1=y' .config ; then 425 fdtput -tx ${UBOOT_DTB} ${SIGNATURE_KEY_NODE} rsa,np 0x0 426 else 427 fdtput -tx ${UBOOT_DTB} ${SIGNATURE_KEY_NODE} rsa,c 0x0 428 fi 429 else 430 fdtput -tx ${UBOOT_DTB} ${SIGNATURE_KEY_NODE} rsa,c 0x0 431 fdtput -tx ${UBOOT_DTB} ${SIGNATURE_KEY_NODE} rsa,np 0x0 432 fdtput -tx ${UBOOT_DTB} ${SIGNATURE_KEY_NODE} rsa,exponent-BN 0x0 433 fi 434 fdtput -r ${UBOOT_DTB} ${SIGNATURE_KEY_NODE}/hash@c 435 fdtput -r ${UBOOT_DTB} ${SIGNATURE_KEY_NODE}/hash@np 436 fi 437 438 mv ${ITS_BOOT} ${FIT_DIR} 439} 440 441function fit_gen_recovery_itb() 442{ 443 if [ ! -z ${ARG_RECOVERY_IMG} ]; then 444 ${FIT_UNPACK} -f ${ARG_RECOVERY_IMG} -o ${FIT_DIR}/unpack 445 ITS_RECOVERY="${FIT_DIR}/unpack/image.its" 446 else 447 echo "ERROR: No recovery.img" 448 exit 1 449 fi 450 451 if [ "${ARG_SIGN}" != "y" ]; then 452 ${MKIMAGE} -f ${ITS_RECOVERY} -E -p ${OFFS_DATA} ${ITB_RECOVERY} -v ${ARG_VER_RECOVERY} 453 else 454 if [ ! -f ${RSA_PRI_KEY} ]; then 455 echo "ERROR: No ${RSA_PRI_KEY}" 456 exit 1 457 elif [ ! -f ${RSA_PUB_KEY} ]; then 458 echo "ERROR: No ${RSA_PUB_KEY}" 459 exit 1 460 fi 461 462 if ! grep -q '^CONFIG_FIT_SIGNATURE=y' .config ; then 463 echo "ERROR: CONFIG_FIT_SIGNATURE is disabled" 464 exit 1 465 fi 466 467 if grep -q '^CONFIG_FIT_ROLLBACK_PROTECT=y' .config ; then 468 ARG_ROLLBACK_PROTECT="y" 469 if [ -z ${ARG_ROLLBACK_IDX_RECOVERY} ]; then 470 echo "ERROR: No arg \"--rollback-index-recovery <n>\"" 471 exit 1 472 fi 473 fi 474 475 # fixup 476 COMMON_FILE=`sed -n "/_common.h/p" ${CHIP_FILE} | awk '{ print $1 }'` 477 FDT_ADDR_R=`awk /fdt_addr_r/ ${COMMON_FILE} | awk -F '=' '{ print $2 }' | awk -F '\\' '{ print $1 }'` 478 KERNEL_ADDR_R=`awk /kernel_addr_r/ ${COMMON_FILE} | awk -F '=' '{ print $2 }' | awk -F '\\' '{ print $1 }'` 479 RMADISK_ADDR_R=`awk /ramdisk_addr_r/ ${COMMON_FILE} | awk -F '=' '{ print $2 }' | awk -F '\\' '{ print $1 }'` 480 sed -i "s/${FDT_ADDR_PLACEHOLDER}/${FDT_ADDR_R}/g" ${ITS_RECOVERY} 481 sed -i "s/${KERNEL_ADDR_PLACEHOLDER}/${KERNEL_ADDR_R}/g" ${ITS_RECOVERY} 482 sed -i "s/${RAMDISK_ADDR_PLACEHOLDER}/${RMADISK_ADDR_R}/g" ${ITS_RECOVERY} 483 if grep -q '^CONFIG_ARM64=y' .config ; then 484 sed -i 's/arch = "arm";/arch = "arm64";/g' ${ITS_RECOVERY} 485 fi 486 487 if [ "${ARG_ROLLBACK_PROTECT}" == "y" ]; then 488 VERSION=`grep 'rollback-index' ${ITS_RECOVERY} | awk -F '=' '{ printf $2 }' | tr -d ' '` 489 sed -i "s/rollback-index = ${VERSION}/rollback-index = <${ARG_ROLLBACK_IDX_RECOVERY}>;/g" ${ITS_RECOVERY} 490 fi 491 492 ${MKIMAGE} -f ${ITS_RECOVERY} -k ${KEY_DIR} -K ${UBOOT_DTB} -E -p ${OFFS_DATA} -r ${ITB_RECOVERY} -v ${ARG_VER_RECOVERY} 493 mv ${SIG_BIN} ${SIG_RECOVERY} 494 495 # rollback-index read back check 496 if [ "${ARG_ROLLBACK_PROTECT}" == "y" ]; then 497 VERSION=`fdtget -ti ${ITB_RECOVERY} /configurations/conf rollback-index` 498 if [ "${VERSION}" != "${ARG_ROLLBACK_IDX_RECOVERY}" ]; then 499 echo "ERROR: Failed to set rollback-index for ${ITB_RECOVERY}"; 500 exit 1 501 fi 502 fi 503 504 # host check signature 505 if [ "${ARG_NO_CHECK}" != "y" ]; then 506 ${CHECK_SIGN} -f ${ITB_RECOVERY} -k ${UBOOT_DTB} 507 fi 508 509 # minimize u-boot.dtb: clearn as 0 but not remove property. 510 if grep -q '^CONFIG_FIT_HW_CRYPTO=y' .config ; then 511 fdtput -tx ${UBOOT_DTB} ${SIGNATURE_KEY_NODE} rsa,r-squared 0x0 512 if grep -q '^CONFIG_ROCKCHIP_CRYPTO_V1=y' .config ; then 513 fdtput -tx ${UBOOT_DTB} ${SIGNATURE_KEY_NODE} rsa,np 0x0 514 else 515 fdtput -tx ${UBOOT_DTB} ${SIGNATURE_KEY_NODE} rsa,c 0x0 516 fi 517 else 518 fdtput -tx ${UBOOT_DTB} ${SIGNATURE_KEY_NODE} rsa,c 0x0 519 fdtput -tx ${UBOOT_DTB} ${SIGNATURE_KEY_NODE} rsa,np 0x0 520 fdtput -tx ${UBOOT_DTB} ${SIGNATURE_KEY_NODE} rsa,exponent-BN 0x0 521 fi 522 fdtput -r ${UBOOT_DTB} ${SIGNATURE_KEY_NODE}/hash@c 523 fdtput -r ${UBOOT_DTB} ${SIGNATURE_KEY_NODE}/hash@np 524 fi 525 526 mv ${ITS_RECOVERY} ${FIT_DIR} 527} 528 529function fit_gen_uboot_img() 530{ 531 ITB=$1 532 533 if [ -z ${ITB} ]; then 534 ITB=${ITB_UBOOT} 535 fi 536 537 ITB_MAX_NUM=`sed -n "/SPL_FIT_IMAGE_MULTIPLE/p" .config | awk -F "=" '{ print $2 }'` 538 ITB_MAX_KB=`sed -n "/SPL_FIT_IMAGE_KB/p" .config | awk -F "=" '{ print $2 }'` 539 ITB_MAX_BS=$((ITB_MAX_KB*1024)) 540 ITB_BS=`ls -l ${ITB} | awk '{ print $5 }'` 541 542 if [ ${ITB_BS} -gt ${ITB_MAX_BS} ]; then 543 echo "ERROR: pack ${IMG_UBOOT} failed! ${ITB} actual: ${ITB_BS} bytes, max limit: ${ITB_MAX_BS} bytes" 544 exit 1 545 fi 546 547 rm -f ${IMG_UBOOT} 548 for ((i = 0; i < ${ITB_MAX_NUM}; i++)); 549 do 550 cat ${ITB} >> ${IMG_UBOOT} 551 truncate -s %${ITB_MAX_KB}K ${IMG_UBOOT} 552 done 553} 554 555function fit_gen_boot_img() 556{ 557 ITB=$1 558 559 if [ -z ${ITB} ]; then 560 ITB=${ITB_BOOT} 561 fi 562 563 if [ "${ITB}" != "${IMG_BOOT}" ]; then 564 cp ${ITB} ${IMG_BOOT} -f 565 fi 566} 567 568function fit_gen_recovery_img() 569{ 570 ITB=$1 571 572 if [ -z ${ITB} ]; then 573 ITB=${ITB_RECOVERY} 574 fi 575 576 if [ "${ITB}" != "${IMG_RECOVERY}" ]; then 577 cp ${ITB} ${IMG_RECOVERY} -f 578 fi 579} 580 581function fit_gen_loader() 582{ 583 if grep -Eq '^CONFIG_FIT_SIGNATURE=y' .config ; then 584 ${RK_SIGN_TOOL} cc --chip ${ARG_CHIP: 2: 6} 585 ${RK_SIGN_TOOL} sl --key ./keys/dev.key --pubkey ./keys/dev.pubkey --loader *_loader_*.bin 586 fi 587} 588 589function fit_msg_uboot() 590{ 591 if [ "${ARG_SIGN}" != "y" ]; then 592 MSG_SIGN="no-signed" 593 else 594 MSG_SIGN="signed" 595 fi 596 597 VERSION=`fdtget -ti ${ITB_UBOOT} / version` 598 if [ "${VERSION}" != "" ]; then 599 MSG_VER=", version=${VERSION}" 600 fi 601 602 if [ "${ARG_SPL_ROLLBACK_PROTECT}" == "y" ]; then 603 echo "Image(${MSG_SIGN}${MSG_VER}, rollback-index=${ARG_ROLLBACK_IDX_UBOOT}): ${IMG_UBOOT} (with uboot, trust...) is ready" 604 else 605 echo "Image(${MSG_SIGN}${MSG_VER}): ${IMG_UBOOT} (FIT with uboot, trust...) is ready" 606 fi 607} 608 609function fit_msg_boot() 610{ 611 if [ -z "${ARG_BOOT_IMG}" ]; then 612 return; 613 fi 614 615 if [ "${ARG_SIGN}" != "y" ]; then 616 MSG_SIGN="no-signed" 617 else 618 MSG_SIGN="signed" 619 fi 620 621 VERSION=`fdtget -ti ${ITB_BOOT} / version` 622 if [ "${VERSION}" != "" ]; then 623 MSG_VER=", version=${VERSION}" 624 fi 625 626 if [ "${ARG_ROLLBACK_PROTECT}" == "y" ]; then 627 echo "Image(${MSG_SIGN}${MSG_VER}, rollback-index=${ARG_ROLLBACK_IDX_BOOT}): ${IMG_BOOT} is ready" 628 else 629 echo "Image(${MSG_SIGN}${MSG_VER}): ${IMG_BOOT} (FIT with kernel, fdt, resource...) is ready" 630 fi 631} 632 633function fit_msg_recovery() 634{ 635 if [ -z "${ARG_RECOVERY_IMG}" ]; then 636 return; 637 fi 638 639 if [ "${ARG_SIGN}" != "y" ]; then 640 MSG_SIGN="no-signed" 641 else 642 MSG_SIGN="signed" 643 fi 644 645 VERSION=`fdtget -ti ${ITB_RECOVERY} / version` 646 if [ "${VERSION}" != "" ]; then 647 MSG_VER=", version=${VERSION}" 648 fi 649 650 if [ "${ARG_ROLLBACK_PROTECT}" == "y" ]; then 651 echo "Image(${MSG_SIGN}${MSG_VER}, rollback-index=${ARG_ROLLBACK_IDX_RECOVERY}): ${IMG_RECOVERY} is ready" 652 else 653 echo "Image(${MSG_SIGN}${MSG_VER}): ${IMG_RECOVERY} (FIT with kernel, fdt, resource...) is ready" 654 fi 655} 656 657function fit_msg_loader() 658{ 659 LOADER=`ls *loader*.bin` 660 661 if grep -q '^CONFIG_FIT_SIGNATURE=y' .config ; then 662 echo "Image(signed): ${LOADER} (with spl, ddr, usbplug) is ready" 663 else 664 echo "Image(no-signed): ${LOADER} (with spl, ddr, usbplug) is ready" 665 fi 666} 667 668fit_process_args $* 669 670if [ ! -z "${ARG_VALIDATE}" ]; then 671 validate_arg ${ARG_VALIDATE} 672else 673 fit_raw_compile 674 if [ ! -z "${ARG_RECOVERY_IMG}" ]; then 675 fit_gen_recovery_itb 676 fit_gen_recovery_img 677 fi 678 # "--boot_img_dir" is for U-Boot debug only 679 if [ ! -z "${ARG_BOOT_IMG}" -o ! -z "${ARG_BOOT_IMG_DIR}" ]; then 680 fit_gen_boot_itb 681 fit_gen_boot_img 682 fi 683 fit_gen_uboot_itb 684 fit_gen_uboot_img 685 fit_gen_loader 686 687 echo 688 fit_msg_uboot 689 fit_msg_recovery 690 fit_msg_boot 691 fit_msg_loader 692fi 693